{
  "best_global_step": null,
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 2.47955322265625e-05,
  "eval_steps": 500,
  "global_step": 100000,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0,
      "model_forward_time": 5.135016441345215,
      "step": 0
    },
    {
      "dynamics_loss_avg": 0.23728200793266296,
      "epoch": 0,
      "step": 0
    },
    {
      "action_loss_avg": 0.362386554479599,
      "epoch": 0,
      "step": 0
    },
    {
      "epoch": 0,
      "step": 0,
      "training_step_time": 13.505264282226562
    },
    {
      "epoch": 9.5367431640625e-10,
      "model_forward_time": 1.6850941181182861,
      "step": 1
    },
    {
      "epoch": 9.5367431640625e-10,
      "step": 1,
      "training_step_time": 8.191267013549805
    },
    {
      "epoch": 1.9073486328125e-09,
      "model_forward_time": 1.6901895999908447,
      "step": 2
    },
    {
      "epoch": 1.9073486328125e-09,
      "step": 2,
      "training_step_time": 7.2892937660217285
    },
    {
      "epoch": 2.86102294921875e-09,
      "model_forward_time": 2.7646026611328125,
      "step": 3
    },
    {
      "epoch": 2.86102294921875e-09,
      "step": 3,
      "training_step_time": 8.047422647476196
    },
    {
      "epoch": 3.814697265625e-09,
      "model_forward_time": 1.7311489582061768,
      "step": 4
    },
    {
      "epoch": 3.814697265625e-09,
      "step": 4,
      "training_step_time": 8.28611969947815
    },
    {
      "epoch": 4.76837158203125e-09,
      "model_forward_time": 1.7303462028503418,
      "step": 5
    },
    {
      "epoch": 4.76837158203125e-09,
      "step": 5,
      "training_step_time": 8.201030254364014
    },
    {
      "epoch": 5.7220458984375e-09,
      "model_forward_time": 1.7282004356384277,
      "step": 6
    },
    {
      "epoch": 5.7220458984375e-09,
      "step": 6,
      "training_step_time": 8.321105241775513
    },
    {
      "epoch": 6.67572021484375e-09,
      "model_forward_time": 1.7307872772216797,
      "step": 7
    },
    {
      "epoch": 6.67572021484375e-09,
      "step": 7,
      "training_step_time": 7.0273566246032715
    },
    {
      "epoch": 7.62939453125e-09,
      "model_forward_time": 2.826185464859009,
      "step": 8
    },
    {
      "epoch": 7.62939453125e-09,
      "step": 8,
      "training_step_time": 8.013866662979126
    },
    {
      "epoch": 8.58306884765625e-09,
      "model_forward_time": 1.7348554134368896,
      "step": 9
    },
    {
      "epoch": 8.58306884765625e-09,
      "step": 9,
      "training_step_time": 8.438541173934937
    },
    {
      "epoch": 9.5367431640625e-09,
      "grad_norm": 13.033432006835938,
      "learning_rate": 1.8000000000000002e-08,
      "loss": 0.6812,
      "step": 10
    },
    {
      "epoch": 9.5367431640625e-09,
      "model_forward_time": 1.8534460067749023,
      "step": 10
    },
    {
      "dynamics_loss_avg": 0.24329907149076463,
      "epoch": 9.5367431640625e-09,
      "step": 10
    },
    {
      "action_loss_avg": 0.4641735181212425,
      "epoch": 9.5367431640625e-09,
      "step": 10
    },
    {
      "epoch": 9.5367431640625e-09,
      "step": 10,
      "training_step_time": 8.416142463684082
    },
    {
      "epoch": 1.049041748046875e-08,
      "model_forward_time": 1.867192268371582,
      "step": 11
    },
    {
      "epoch": 1.049041748046875e-08,
      "step": 11,
      "training_step_time": 8.436052322387695
    },
    {
      "epoch": 1.1444091796875e-08,
      "model_forward_time": 1.885650396347046,
      "step": 12
    },
    {
      "epoch": 1.1444091796875e-08,
      "step": 12,
      "training_step_time": 8.419640302658081
    },
    {
      "epoch": 1.239776611328125e-08,
      "model_forward_time": 1.1715929508209229,
      "step": 13
    },
    {
      "epoch": 1.239776611328125e-08,
      "step": 13,
      "training_step_time": 7.188096761703491
    },
    {
      "epoch": 1.33514404296875e-08,
      "model_forward_time": 2.274608612060547,
      "step": 14
    },
    {
      "epoch": 1.33514404296875e-08,
      "step": 14,
      "training_step_time": 7.936015605926514
    },
    {
      "epoch": 1.430511474609375e-08,
      "model_forward_time": 1.6978886127471924,
      "step": 15
    },
    {
      "epoch": 1.430511474609375e-08,
      "step": 15,
      "training_step_time": 8.441263437271118
    },
    {
      "epoch": 1.52587890625e-08,
      "model_forward_time": 1.67873215675354,
      "step": 16
    },
    {
      "epoch": 1.52587890625e-08,
      "step": 16,
      "training_step_time": 8.454991579055786
    },
    {
      "epoch": 1.621246337890625e-08,
      "model_forward_time": 1.6740126609802246,
      "step": 17
    },
    {
      "epoch": 1.621246337890625e-08,
      "step": 17,
      "training_step_time": 8.328298091888428
    },
    {
      "epoch": 1.71661376953125e-08,
      "model_forward_time": 1.6891160011291504,
      "step": 18
    },
    {
      "epoch": 1.71661376953125e-08,
      "step": 18,
      "training_step_time": 8.254902362823486
    },
    {
      "epoch": 1.811981201171875e-08,
      "model_forward_time": 0.9647080898284912,
      "step": 19
    },
    {
      "epoch": 1.811981201171875e-08,
      "step": 19,
      "training_step_time": 7.152583360671997
    },
    {
      "epoch": 1.9073486328125e-08,
      "grad_norm": 31.34088706970215,
      "learning_rate": 3.8e-08,
      "loss": 0.6239,
      "step": 20
    },
    {
      "epoch": 1.9073486328125e-08,
      "model_forward_time": 2.229238271713257,
      "step": 20
    },
    {
      "dynamics_loss_avg": 0.20507694110274316,
      "epoch": 1.9073486328125e-08,
      "step": 20
    },
    {
      "action_loss_avg": 0.3871685653924942,
      "epoch": 1.9073486328125e-08,
      "step": 20
    },
    {
      "epoch": 1.9073486328125e-08,
      "step": 20,
      "training_step_time": 7.926627159118652
    },
    {
      "epoch": 2.002716064453125e-08,
      "model_forward_time": 1.678215503692627,
      "step": 21
    },
    {
      "epoch": 2.002716064453125e-08,
      "step": 21,
      "training_step_time": 8.209725379943848
    },
    {
      "epoch": 2.09808349609375e-08,
      "model_forward_time": 1.6784188747406006,
      "step": 22
    },
    {
      "epoch": 2.09808349609375e-08,
      "step": 22,
      "training_step_time": 8.188854455947876
    },
    {
      "epoch": 2.193450927734375e-08,
      "model_forward_time": 1.6874701976776123,
      "step": 23
    },
    {
      "epoch": 2.193450927734375e-08,
      "step": 23,
      "training_step_time": 7.100697994232178
    },
    {
      "epoch": 2.288818359375e-08,
      "model_forward_time": 2.2213449478149414,
      "step": 24
    },
    {
      "epoch": 2.288818359375e-08,
      "step": 24,
      "training_step_time": 7.828748941421509
    },
    {
      "epoch": 2.384185791015625e-08,
      "model_forward_time": 1.689148187637329,
      "step": 25
    },
    {
      "epoch": 2.384185791015625e-08,
      "step": 25,
      "training_step_time": 7.917466878890991
    },
    {
      "epoch": 2.47955322265625e-08,
      "model_forward_time": 1.6792418956756592,
      "step": 26
    },
    {
      "epoch": 2.47955322265625e-08,
      "step": 26,
      "training_step_time": 8.008623838424683
    },
    {
      "epoch": 2.574920654296875e-08,
      "model_forward_time": 1.694936752319336,
      "step": 27
    },
    {
      "epoch": 2.574920654296875e-08,
      "step": 27,
      "training_step_time": 8.086255073547363
    },
    {
      "epoch": 2.6702880859375e-08,
      "model_forward_time": 0.9759626388549805,
      "step": 28
    },
    {
      "epoch": 2.6702880859375e-08,
      "step": 28,
      "training_step_time": 6.931939363479614
    },
    {
      "epoch": 2.765655517578125e-08,
      "model_forward_time": 2.223451614379883,
      "step": 29
    },
    {
      "epoch": 2.765655517578125e-08,
      "step": 29,
      "training_step_time": 7.916088581085205
    },
    {
      "epoch": 2.86102294921875e-08,
      "grad_norm": 19.874494552612305,
      "learning_rate": 5.8e-08,
      "loss": 0.6887,
      "step": 30
    },
    {
      "epoch": 2.86102294921875e-08,
      "model_forward_time": 1.6881537437438965,
      "step": 30
    },
    {
      "dynamics_loss_avg": 0.24114589095115663,
      "epoch": 2.86102294921875e-08,
      "step": 30
    },
    {
      "action_loss_avg": 0.48411210924386977,
      "epoch": 2.86102294921875e-08,
      "step": 30
    },
    {
      "epoch": 2.86102294921875e-08,
      "step": 30,
      "training_step_time": 8.26917839050293
    },
    {
      "epoch": 2.956390380859375e-08,
      "model_forward_time": 1.6908483505249023,
      "step": 31
    },
    {
      "epoch": 2.956390380859375e-08,
      "step": 31,
      "training_step_time": 8.288066148757935
    },
    {
      "epoch": 3.0517578125e-08,
      "model_forward_time": 1.6935186386108398,
      "step": 32
    },
    {
      "epoch": 3.0517578125e-08,
      "step": 32,
      "training_step_time": 8.334256172180176
    },
    {
      "epoch": 3.147125244140625e-08,
      "model_forward_time": 1.6895575523376465,
      "step": 33
    },
    {
      "epoch": 3.147125244140625e-08,
      "step": 33,
      "training_step_time": 6.947652578353882
    },
    {
      "epoch": 3.24249267578125e-08,
      "model_forward_time": 2.2140700817108154,
      "step": 34
    },
    {
      "epoch": 3.24249267578125e-08,
      "step": 34,
      "training_step_time": 7.858170747756958
    },
    {
      "epoch": 3.337860107421875e-08,
      "model_forward_time": 1.690141201019287,
      "step": 35
    },
    {
      "epoch": 3.337860107421875e-08,
      "step": 35,
      "training_step_time": 8.253374814987183
    },
    {
      "epoch": 3.4332275390625e-08,
      "model_forward_time": 1.6882658004760742,
      "step": 36
    },
    {
      "epoch": 3.4332275390625e-08,
      "step": 36,
      "training_step_time": 8.357186794281006
    },
    {
      "epoch": 3.528594970703125e-08,
      "model_forward_time": 1.6945109367370605,
      "step": 37
    },
    {
      "epoch": 3.528594970703125e-08,
      "step": 37,
      "training_step_time": 8.36316466331482
    },
    {
      "epoch": 3.62396240234375e-08,
      "model_forward_time": 1.6844394207000732,
      "step": 38
    },
    {
      "epoch": 3.62396240234375e-08,
      "step": 38,
      "training_step_time": 6.99097204208374
    },
    {
      "epoch": 3.719329833984375e-08,
      "model_forward_time": 2.8652689456939697,
      "step": 39
    },
    {
      "epoch": 3.719329833984375e-08,
      "step": 39,
      "training_step_time": 7.898171663284302
    },
    {
      "epoch": 3.814697265625e-08,
      "grad_norm": 26.16657066345215,
      "learning_rate": 7.8e-08,
      "loss": 0.7859,
      "step": 40
    },
    {
      "epoch": 3.814697265625e-08,
      "model_forward_time": 1.6905288696289062,
      "step": 40
    },
    {
      "dynamics_loss_avg": 0.26339799612760545,
      "epoch": 3.814697265625e-08,
      "step": 40
    },
    {
      "action_loss_avg": 0.5664175897836685,
      "epoch": 3.814697265625e-08,
      "step": 40
    },
    {
      "epoch": 3.814697265625e-08,
      "step": 40,
      "training_step_time": 8.454554319381714
    },
    {
      "epoch": 3.910064697265625e-08,
      "model_forward_time": 1.6863670349121094,
      "step": 41
    },
    {
      "epoch": 3.910064697265625e-08,
      "step": 41,
      "training_step_time": 8.344207763671875
    },
    {
      "epoch": 4.00543212890625e-08,
      "model_forward_time": 1.6862866878509521,
      "step": 42
    },
    {
      "epoch": 4.00543212890625e-08,
      "step": 42,
      "training_step_time": 8.526909828186035
    },
    {
      "epoch": 4.100799560546875e-08,
      "model_forward_time": 1.682875633239746,
      "step": 43
    },
    {
      "epoch": 4.100799560546875e-08,
      "step": 43,
      "training_step_time": 6.9590232372283936
    },
    {
      "epoch": 4.1961669921875e-08,
      "model_forward_time": 2.843092203140259,
      "step": 44
    },
    {
      "epoch": 4.1961669921875e-08,
      "step": 44,
      "training_step_time": 7.864465236663818
    },
    {
      "epoch": 4.291534423828125e-08,
      "model_forward_time": 1.689107894897461,
      "step": 45
    },
    {
      "epoch": 4.291534423828125e-08,
      "step": 45,
      "training_step_time": 8.386337757110596
    },
    {
      "epoch": 4.38690185546875e-08,
      "model_forward_time": 1.677030086517334,
      "step": 46
    },
    {
      "epoch": 4.38690185546875e-08,
      "step": 46,
      "training_step_time": 8.32478642463684
    },
    {
      "epoch": 4.482269287109375e-08,
      "model_forward_time": 1.6967144012451172,
      "step": 47
    },
    {
      "epoch": 4.482269287109375e-08,
      "step": 47,
      "training_step_time": 8.473295211791992
    },
    {
      "epoch": 4.57763671875e-08,
      "model_forward_time": 1.6925649642944336,
      "step": 48
    },
    {
      "epoch": 4.57763671875e-08,
      "step": 48,
      "training_step_time": 6.952227830886841
    },
    {
      "epoch": 4.673004150390625e-08,
      "model_forward_time": 2.786001443862915,
      "step": 49
    },
    {
      "epoch": 4.673004150390625e-08,
      "step": 49,
      "training_step_time": 7.826355457305908
    },
    {
      "epoch": 4.76837158203125e-08,
      "grad_norm": 10.389887809753418,
      "learning_rate": 9.8e-08,
      "loss": 0.6765,
      "step": 50
    },
    {
      "epoch": 4.76837158203125e-08,
      "model_forward_time": 1.7000172138214111,
      "step": 50
    },
    {
      "dynamics_loss_avg": 0.23899742141366004,
      "epoch": 4.76837158203125e-08,
      "step": 50
    },
    {
      "action_loss_avg": 0.4405762813985348,
      "epoch": 4.76837158203125e-08,
      "step": 50
    },
    {
      "epoch": 4.76837158203125e-08,
      "step": 50,
      "training_step_time": 8.365705490112305
    },
    {
      "epoch": 4.863739013671875e-08,
      "model_forward_time": 1.7003731727600098,
      "step": 51
    },
    {
      "epoch": 4.863739013671875e-08,
      "step": 51,
      "training_step_time": 8.362471103668213
    },
    {
      "epoch": 4.9591064453125e-08,
      "model_forward_time": 1.704549789428711,
      "step": 52
    },
    {
      "epoch": 4.9591064453125e-08,
      "step": 52,
      "training_step_time": 8.42310357093811
    },
    {
      "epoch": 5.054473876953125e-08,
      "model_forward_time": 1.716766357421875,
      "step": 53
    },
    {
      "epoch": 5.054473876953125e-08,
      "step": 53,
      "training_step_time": 6.958104848861694
    },
    {
      "epoch": 5.14984130859375e-08,
      "model_forward_time": 2.23427677154541,
      "step": 54
    },
    {
      "epoch": 5.14984130859375e-08,
      "step": 54,
      "training_step_time": 7.733570337295532
    },
    {
      "epoch": 5.245208740234375e-08,
      "model_forward_time": 1.6982176303863525,
      "step": 55
    },
    {
      "epoch": 5.245208740234375e-08,
      "step": 55,
      "training_step_time": 8.366776704788208
    },
    {
      "epoch": 5.340576171875e-08,
      "model_forward_time": 1.690398931503296,
      "step": 56
    },
    {
      "epoch": 5.340576171875e-08,
      "step": 56,
      "training_step_time": 8.541640758514404
    },
    {
      "epoch": 5.435943603515625e-08,
      "model_forward_time": 1.6947729587554932,
      "step": 57
    },
    {
      "epoch": 5.435943603515625e-08,
      "step": 57,
      "training_step_time": 8.4315927028656
    },
    {
      "epoch": 5.53131103515625e-08,
      "model_forward_time": 1.6975111961364746,
      "step": 58
    },
    {
      "epoch": 5.53131103515625e-08,
      "step": 58,
      "training_step_time": 6.925417900085449
    },
    {
      "epoch": 5.626678466796875e-08,
      "model_forward_time": 2.2303829193115234,
      "step": 59
    },
    {
      "epoch": 5.626678466796875e-08,
      "step": 59,
      "training_step_time": 7.86589503288269
    },
    {
      "epoch": 5.7220458984375e-08,
      "grad_norm": 17.413827896118164,
      "learning_rate": 1.1800000000000001e-07,
      "loss": 0.7144,
      "step": 60
    },
    {
      "epoch": 5.7220458984375e-08,
      "model_forward_time": 1.6976492404937744,
      "step": 60
    },
    {
      "dynamics_loss_avg": 0.26920486316084863,
      "epoch": 5.7220458984375e-08,
      "step": 60
    },
    {
      "action_loss_avg": 0.4802272379398346,
      "epoch": 5.7220458984375e-08,
      "step": 60
    },
    {
      "epoch": 5.7220458984375e-08,
      "step": 60,
      "training_step_time": 8.462333917617798
    },
    {
      "epoch": 5.817413330078125e-08,
      "model_forward_time": 1.6947991847991943,
      "step": 61
    },
    {
      "epoch": 5.817413330078125e-08,
      "step": 61,
      "training_step_time": 8.467076301574707
    },
    {
      "epoch": 5.91278076171875e-08,
      "model_forward_time": 1.6966345310211182,
      "step": 62
    },
    {
      "epoch": 5.91278076171875e-08,
      "step": 62,
      "training_step_time": 8.451696872711182
    },
    {
      "epoch": 6.008148193359375e-08,
      "model_forward_time": 1.6965396404266357,
      "step": 63
    },
    {
      "epoch": 6.008148193359375e-08,
      "step": 63,
      "training_step_time": 6.927457332611084
    },
    {
      "epoch": 6.103515625e-08,
      "model_forward_time": 2.206165075302124,
      "step": 64
    },
    {
      "epoch": 6.103515625e-08,
      "step": 64,
      "training_step_time": 7.858572006225586
    },
    {
      "epoch": 6.198883056640625e-08,
      "model_forward_time": 1.705596685409546,
      "step": 65
    },
    {
      "epoch": 6.198883056640625e-08,
      "step": 65,
      "training_step_time": 8.409224510192871
    },
    {
      "epoch": 6.29425048828125e-08,
      "model_forward_time": 1.6939640045166016,
      "step": 66
    },
    {
      "epoch": 6.29425048828125e-08,
      "step": 66,
      "training_step_time": 8.479732513427734
    },
    {
      "epoch": 6.389617919921875e-08,
      "model_forward_time": 1.6963541507720947,
      "step": 67
    },
    {
      "epoch": 6.389617919921875e-08,
      "step": 67,
      "training_step_time": 8.473066806793213
    },
    {
      "epoch": 6.4849853515625e-08,
      "model_forward_time": 1.3462252616882324,
      "step": 68
    },
    {
      "epoch": 6.4849853515625e-08,
      "step": 68,
      "training_step_time": 8.455634355545044
    },
    {
      "epoch": 6.580352783203126e-08,
      "model_forward_time": 1.6954350471496582,
      "step": 69
    },
    {
      "epoch": 6.580352783203126e-08,
      "step": 69,
      "training_step_time": 6.956857204437256
    },
    {
      "epoch": 6.67572021484375e-08,
      "grad_norm": 5.721103668212891,
      "learning_rate": 1.3800000000000002e-07,
      "loss": 0.6522,
      "step": 70
    },
    {
      "epoch": 6.67572021484375e-08,
      "model_forward_time": 1.3296120166778564,
      "step": 70
    },
    {
      "dynamics_loss_avg": 0.24098846167325974,
      "epoch": 6.67572021484375e-08,
      "step": 70
    },
    {
      "action_loss_avg": 0.35007002055644987,
      "epoch": 6.67572021484375e-08,
      "step": 70
    },
    {
      "epoch": 6.67572021484375e-08,
      "step": 70,
      "training_step_time": 7.959063291549683
    },
    {
      "epoch": 6.771087646484374e-08,
      "model_forward_time": 1.6907989978790283,
      "step": 71
    },
    {
      "epoch": 6.771087646484374e-08,
      "step": 71,
      "training_step_time": 8.289921045303345
    },
    {
      "epoch": 6.866455078125e-08,
      "model_forward_time": 1.6978607177734375,
      "step": 72
    },
    {
      "epoch": 6.866455078125e-08,
      "step": 72,
      "training_step_time": 8.21355152130127
    },
    {
      "epoch": 6.961822509765625e-08,
      "model_forward_time": 1.7044587135314941,
      "step": 73
    },
    {
      "epoch": 6.961822509765625e-08,
      "step": 73,
      "training_step_time": 8.248126029968262
    },
    {
      "epoch": 7.05718994140625e-08,
      "model_forward_time": 1.6964805126190186,
      "step": 74
    },
    {
      "epoch": 7.05718994140625e-08,
      "step": 74,
      "training_step_time": 7.020855903625488
    },
    {
      "epoch": 7.152557373046875e-08,
      "model_forward_time": 1.7014751434326172,
      "step": 75
    },
    {
      "epoch": 7.152557373046875e-08,
      "step": 75,
      "training_step_time": 7.908069372177124
    },
    {
      "epoch": 7.2479248046875e-08,
      "model_forward_time": 1.7020683288574219,
      "step": 76
    },
    {
      "epoch": 7.2479248046875e-08,
      "step": 76,
      "training_step_time": 8.231433868408203
    },
    {
      "epoch": 7.343292236328125e-08,
      "model_forward_time": 1.7006738185882568,
      "step": 77
    },
    {
      "epoch": 7.343292236328125e-08,
      "step": 77,
      "training_step_time": 8.230119943618774
    },
    {
      "epoch": 7.43865966796875e-08,
      "model_forward_time": 1.7035369873046875,
      "step": 78
    },
    {
      "epoch": 7.43865966796875e-08,
      "step": 78,
      "training_step_time": 6.825886249542236
    },
    {
      "epoch": 7.534027099609375e-08,
      "model_forward_time": 1.8903865814208984,
      "step": 79
    },
    {
      "epoch": 7.534027099609375e-08,
      "step": 79,
      "training_step_time": 7.809800386428833
    },
    {
      "epoch": 7.62939453125e-08,
      "grad_norm": 7.469008445739746,
      "learning_rate": 1.5800000000000004e-07,
      "loss": 0.5528,
      "step": 80
    },
    {
      "epoch": 7.62939453125e-08,
      "model_forward_time": 1.6949431896209717,
      "step": 80
    },
    {
      "dynamics_loss_avg": 0.2395896330475807,
      "epoch": 7.62939453125e-08,
      "step": 80
    },
    {
      "action_loss_avg": 0.35480891913175583,
      "epoch": 7.62939453125e-08,
      "step": 80
    },
    {
      "epoch": 7.62939453125e-08,
      "step": 80,
      "training_step_time": 8.32582712173462
    },
    {
      "epoch": 7.724761962890625e-08,
      "model_forward_time": 1.7050418853759766,
      "step": 81
    },
    {
      "epoch": 7.724761962890625e-08,
      "step": 81,
      "training_step_time": 8.487173080444336
    },
    {
      "epoch": 7.82012939453125e-08,
      "model_forward_time": 1.6971559524536133,
      "step": 82
    },
    {
      "epoch": 7.82012939453125e-08,
      "step": 82,
      "training_step_time": 8.421865224838257
    },
    {
      "epoch": 7.915496826171875e-08,
      "model_forward_time": 1.6998937129974365,
      "step": 83
    },
    {
      "epoch": 7.915496826171875e-08,
      "step": 83,
      "training_step_time": 6.7835142612457275
    },
    {
      "epoch": 8.0108642578125e-08,
      "model_forward_time": 2.2862401008605957,
      "step": 84
    },
    {
      "epoch": 8.0108642578125e-08,
      "step": 84,
      "training_step_time": 7.815289497375488
    },
    {
      "epoch": 8.106231689453126e-08,
      "model_forward_time": 1.70979642868042,
      "step": 85
    },
    {
      "epoch": 8.106231689453126e-08,
      "step": 85,
      "training_step_time": 8.366561651229858
    },
    {
      "epoch": 8.20159912109375e-08,
      "model_forward_time": 1.7246930599212646,
      "step": 86
    },
    {
      "epoch": 8.20159912109375e-08,
      "step": 86,
      "training_step_time": 8.518652200698853
    },
    {
      "epoch": 8.296966552734374e-08,
      "model_forward_time": 1.7129838466644287,
      "step": 87
    },
    {
      "epoch": 8.296966552734374e-08,
      "step": 87,
      "training_step_time": 8.5013427734375
    },
    {
      "epoch": 8.392333984375e-08,
      "model_forward_time": 1.7159254550933838,
      "step": 88
    },
    {
      "epoch": 8.392333984375e-08,
      "step": 88,
      "training_step_time": 6.755397319793701
    },
    {
      "epoch": 8.487701416015625e-08,
      "model_forward_time": 2.292405605316162,
      "step": 89
    },
    {
      "epoch": 8.487701416015625e-08,
      "step": 89,
      "training_step_time": 7.878424882888794
    },
    {
      "epoch": 8.58306884765625e-08,
      "grad_norm": 4.558644771575928,
      "learning_rate": 1.78e-07,
      "loss": 0.528,
      "step": 90
    },
    {
      "epoch": 8.58306884765625e-08,
      "model_forward_time": 1.7251992225646973,
      "step": 90
    },
    {
      "dynamics_loss_avg": 0.17385129109025002,
      "epoch": 8.58306884765625e-08,
      "step": 90
    },
    {
      "action_loss_avg": 0.2503506027162075,
      "epoch": 8.58306884765625e-08,
      "step": 90
    },
    {
      "epoch": 8.58306884765625e-08,
      "step": 90,
      "training_step_time": 8.324567556381226
    },
    {
      "epoch": 8.678436279296875e-08,
      "model_forward_time": 1.7243921756744385,
      "step": 91
    },
    {
      "epoch": 8.678436279296875e-08,
      "step": 91,
      "training_step_time": 8.512906074523926
    },
    {
      "epoch": 8.7738037109375e-08,
      "model_forward_time": 1.7266206741333008,
      "step": 92
    },
    {
      "epoch": 8.7738037109375e-08,
      "step": 92,
      "training_step_time": 8.47740364074707
    },
    {
      "epoch": 8.869171142578125e-08,
      "model_forward_time": 1.7191901206970215,
      "step": 93
    },
    {
      "epoch": 8.869171142578125e-08,
      "step": 93,
      "training_step_time": 8.487781286239624
    },
    {
      "epoch": 8.96453857421875e-08,
      "model_forward_time": 1.7241549491882324,
      "step": 94
    },
    {
      "epoch": 8.96453857421875e-08,
      "step": 94,
      "training_step_time": 6.770693302154541
    },
    {
      "epoch": 9.059906005859375e-08,
      "model_forward_time": 1.7223844528198242,
      "step": 95
    },
    {
      "epoch": 9.059906005859375e-08,
      "step": 95,
      "training_step_time": 7.880565404891968
    },
    {
      "epoch": 9.1552734375e-08,
      "model_forward_time": 1.726104497909546,
      "step": 96
    },
    {
      "epoch": 9.1552734375e-08,
      "step": 96,
      "training_step_time": 8.459637880325317
    },
    {
      "epoch": 9.250640869140625e-08,
      "model_forward_time": 1.7262334823608398,
      "step": 97
    },
    {
      "epoch": 9.250640869140625e-08,
      "step": 97,
      "training_step_time": 8.50673794746399
    },
    {
      "epoch": 9.34600830078125e-08,
      "model_forward_time": 1.7226557731628418,
      "step": 98
    },
    {
      "epoch": 9.34600830078125e-08,
      "step": 98,
      "training_step_time": 8.57677173614502
    },
    {
      "epoch": 9.441375732421875e-08,
      "model_forward_time": 1.7214884757995605,
      "step": 99
    },
    {
      "epoch": 9.441375732421875e-08,
      "step": 99,
      "training_step_time": 6.760815620422363
    },
    {
      "epoch": 9.5367431640625e-08,
      "grad_norm": 7.993073463439941,
      "learning_rate": 1.9800000000000003e-07,
      "loss": 0.5264,
      "step": 100
    },
    {
      "epoch": 9.5367431640625e-08,
      "model_forward_time": 2.3032519817352295,
      "step": 100
    },
    {
      "dynamics_loss_avg": 0.15312132984399796,
      "epoch": 9.5367431640625e-08,
      "step": 100
    },
    {
      "action_loss_avg": 0.255811121314764,
      "epoch": 9.5367431640625e-08,
      "step": 100
    },
    {
      "epoch": 9.5367431640625e-08,
      "step": 100,
      "training_step_time": 7.934019327163696
    },
    {
      "epoch": 9.632110595703126e-08,
      "model_forward_time": 1.7174086570739746,
      "step": 101
    },
    {
      "epoch": 9.632110595703126e-08,
      "step": 101,
      "training_step_time": 8.611235857009888
    },
    {
      "epoch": 9.72747802734375e-08,
      "model_forward_time": 1.7097022533416748,
      "step": 102
    },
    {
      "epoch": 9.72747802734375e-08,
      "step": 102,
      "training_step_time": 8.603151798248291
    },
    {
      "epoch": 9.822845458984374e-08,
      "model_forward_time": 1.7160303592681885,
      "step": 103
    },
    {
      "epoch": 9.822845458984374e-08,
      "step": 103,
      "training_step_time": 8.615766048431396
    },
    {
      "epoch": 9.918212890625e-08,
      "model_forward_time": 1.7211949825286865,
      "step": 104
    },
    {
      "epoch": 9.918212890625e-08,
      "step": 104,
      "training_step_time": 6.774665355682373
    },
    {
      "epoch": 1.0013580322265625e-07,
      "model_forward_time": 2.8308560848236084,
      "step": 105
    },
    {
      "epoch": 1.0013580322265625e-07,
      "step": 105,
      "training_step_time": 8.033710718154907
    },
    {
      "epoch": 1.010894775390625e-07,
      "model_forward_time": 1.7206077575683594,
      "step": 106
    },
    {
      "epoch": 1.010894775390625e-07,
      "step": 106,
      "training_step_time": 8.589106798171997
    },
    {
      "epoch": 1.0204315185546875e-07,
      "model_forward_time": 1.7237699031829834,
      "step": 107
    },
    {
      "epoch": 1.0204315185546875e-07,
      "step": 107,
      "training_step_time": 8.744105339050293
    },
    {
      "epoch": 1.02996826171875e-07,
      "model_forward_time": 1.717940330505371,
      "step": 108
    },
    {
      "epoch": 1.02996826171875e-07,
      "step": 108,
      "training_step_time": 8.67017650604248
    },
    {
      "epoch": 1.0395050048828125e-07,
      "model_forward_time": 1.715195655822754,
      "step": 109
    },
    {
      "epoch": 1.0395050048828125e-07,
      "step": 109,
      "training_step_time": 8.719998598098755
    },
    {
      "epoch": 1.049041748046875e-07,
      "grad_norm": 8.497740745544434,
      "learning_rate": 2.1800000000000002e-07,
      "loss": 0.5313,
      "step": 110
    },
    {
      "epoch": 1.049041748046875e-07,
      "model_forward_time": 1.7140541076660156,
      "step": 110
    },
    {
      "dynamics_loss_avg": 0.18245785310864449,
      "epoch": 1.049041748046875e-07,
      "step": 110
    },
    {
      "action_loss_avg": 0.37309360653162005,
      "epoch": 1.049041748046875e-07,
      "step": 110
    },
    {
      "epoch": 1.049041748046875e-07,
      "step": 110,
      "training_step_time": 6.825489521026611
    },
    {
      "epoch": 1.0585784912109375e-07,
      "model_forward_time": 2.807817220687866,
      "step": 111
    },
    {
      "epoch": 1.0585784912109375e-07,
      "step": 111,
      "training_step_time": 7.9566919803619385
    },
    {
      "epoch": 1.068115234375e-07,
      "model_forward_time": 1.71785306930542,
      "step": 112
    },
    {
      "epoch": 1.068115234375e-07,
      "step": 112,
      "training_step_time": 8.83371901512146
    },
    {
      "epoch": 1.0776519775390625e-07,
      "model_forward_time": 1.0083611011505127,
      "step": 113
    },
    {
      "epoch": 1.0776519775390625e-07,
      "step": 113,
      "training_step_time": 8.79776120185852
    },
    {
      "epoch": 1.087188720703125e-07,
      "model_forward_time": 1.713697910308838,
      "step": 114
    },
    {
      "epoch": 1.087188720703125e-07,
      "step": 114,
      "training_step_time": 8.736467361450195
    },
    {
      "epoch": 1.0967254638671875e-07,
      "model_forward_time": 1.7162652015686035,
      "step": 115
    },
    {
      "epoch": 1.0967254638671875e-07,
      "step": 115,
      "training_step_time": 8.813260316848755
    },
    {
      "epoch": 1.10626220703125e-07,
      "model_forward_time": 1.7133142948150635,
      "step": 116
    },
    {
      "epoch": 1.10626220703125e-07,
      "step": 116,
      "training_step_time": 6.807996034622192
    },
    {
      "epoch": 1.1157989501953126e-07,
      "model_forward_time": 2.8317480087280273,
      "step": 117
    },
    {
      "epoch": 1.1157989501953126e-07,
      "step": 117,
      "training_step_time": 7.943705320358276
    },
    {
      "epoch": 1.125335693359375e-07,
      "model_forward_time": 1.730661153793335,
      "step": 118
    },
    {
      "epoch": 1.125335693359375e-07,
      "step": 118,
      "training_step_time": 8.754091501235962
    },
    {
      "epoch": 1.1348724365234374e-07,
      "model_forward_time": 1.7299237251281738,
      "step": 119
    },
    {
      "epoch": 1.1348724365234374e-07,
      "step": 119,
      "training_step_time": 8.794931173324585
    },
    {
      "epoch": 1.1444091796875e-07,
      "grad_norm": 5.151562690734863,
      "learning_rate": 2.3800000000000004e-07,
      "loss": 0.5186,
      "step": 120
    },
    {
      "epoch": 1.1444091796875e-07,
      "model_forward_time": 1.7223801612854004,
      "step": 120
    },
    {
      "dynamics_loss_avg": 0.15792640075087547,
      "epoch": 1.1444091796875e-07,
      "step": 120
    },
    {
      "action_loss_avg": 0.26478165835142137,
      "epoch": 1.1444091796875e-07,
      "step": 120
    },
    {
      "epoch": 1.1444091796875e-07,
      "step": 120,
      "training_step_time": 8.816375017166138
    },
    {
      "epoch": 1.1539459228515625e-07,
      "model_forward_time": 1.7226371765136719,
      "step": 121
    },
    {
      "epoch": 1.1539459228515625e-07,
      "step": 121,
      "training_step_time": 8.830787897109985
    },
    {
      "epoch": 1.163482666015625e-07,
      "model_forward_time": 1.7198526859283447,
      "step": 122
    },
    {
      "epoch": 1.163482666015625e-07,
      "step": 122,
      "training_step_time": 6.8096489906311035
    },
    {
      "epoch": 1.1730194091796875e-07,
      "model_forward_time": 2.8648416996002197,
      "step": 123
    },
    {
      "epoch": 1.1730194091796875e-07,
      "step": 123,
      "training_step_time": 7.908997058868408
    },
    {
      "epoch": 1.18255615234375e-07,
      "model_forward_time": 1.7146148681640625,
      "step": 124
    },
    {
      "epoch": 1.18255615234375e-07,
      "step": 124,
      "training_step_time": 8.701417446136475
    },
    {
      "epoch": 1.1920928955078125e-07,
      "model_forward_time": 1.7195537090301514,
      "step": 125
    },
    {
      "epoch": 1.1920928955078125e-07,
      "step": 125,
      "training_step_time": 8.883916139602661
    },
    {
      "epoch": 1.201629638671875e-07,
      "model_forward_time": 1.735550880432129,
      "step": 126
    },
    {
      "epoch": 1.201629638671875e-07,
      "step": 126,
      "training_step_time": 8.9260835647583
    },
    {
      "epoch": 1.2111663818359375e-07,
      "model_forward_time": 1.73842453956604,
      "step": 127
    },
    {
      "epoch": 1.2111663818359375e-07,
      "step": 127,
      "training_step_time": 8.779427289962769
    },
    {
      "epoch": 1.220703125e-07,
      "model_forward_time": 1.7397360801696777,
      "step": 128
    },
    {
      "epoch": 1.220703125e-07,
      "step": 128,
      "training_step_time": 8.799763917922974
    },
    {
      "epoch": 1.2302398681640625e-07,
      "model_forward_time": 1.7331039905548096,
      "step": 129
    },
    {
      "epoch": 1.2302398681640625e-07,
      "step": 129,
      "training_step_time": 6.827985048294067
    },
    {
      "epoch": 1.239776611328125e-07,
      "grad_norm": 2.3432183265686035,
      "learning_rate": 2.58e-07,
      "loss": 0.3878,
      "step": 130
    },
    {
      "epoch": 1.239776611328125e-07,
      "model_forward_time": 1.7359881401062012,
      "step": 130
    },
    {
      "dynamics_loss_avg": 0.11907150782644749,
      "epoch": 1.239776611328125e-07,
      "step": 130
    },
    {
      "action_loss_avg": 0.21213946640491485,
      "epoch": 1.239776611328125e-07,
      "step": 130
    },
    {
      "epoch": 1.239776611328125e-07,
      "step": 130,
      "training_step_time": 7.925016641616821
    },
    {
      "epoch": 1.2493133544921876e-07,
      "model_forward_time": 1.7480361461639404,
      "step": 131
    },
    {
      "epoch": 1.2493133544921876e-07,
      "step": 131,
      "training_step_time": 8.985440015792847
    },
    {
      "epoch": 1.25885009765625e-07,
      "model_forward_time": 1.7254810333251953,
      "step": 132
    },
    {
      "epoch": 1.25885009765625e-07,
      "step": 132,
      "training_step_time": 8.944493293762207
    },
    {
      "epoch": 1.2683868408203126e-07,
      "model_forward_time": 1.7250864505767822,
      "step": 133
    },
    {
      "epoch": 1.2683868408203126e-07,
      "step": 133,
      "training_step_time": 8.82885193824768
    },
    {
      "epoch": 1.277923583984375e-07,
      "model_forward_time": 1.733820915222168,
      "step": 134
    },
    {
      "epoch": 1.277923583984375e-07,
      "step": 134,
      "training_step_time": 8.81927227973938
    },
    {
      "epoch": 1.2874603271484376e-07,
      "model_forward_time": 1.7309660911560059,
      "step": 135
    },
    {
      "epoch": 1.2874603271484376e-07,
      "step": 135,
      "training_step_time": 6.824461460113525
    },
    {
      "epoch": 1.2969970703125e-07,
      "model_forward_time": 2.8570122718811035,
      "step": 136
    },
    {
      "epoch": 1.2969970703125e-07,
      "step": 136,
      "training_step_time": 7.933849573135376
    },
    {
      "epoch": 1.3065338134765626e-07,
      "model_forward_time": 1.7318310737609863,
      "step": 137
    },
    {
      "epoch": 1.3065338134765626e-07,
      "step": 137,
      "training_step_time": 8.929311037063599
    },
    {
      "epoch": 1.316070556640625e-07,
      "model_forward_time": 1.7219960689544678,
      "step": 138
    },
    {
      "epoch": 1.316070556640625e-07,
      "step": 138,
      "training_step_time": 8.972900629043579
    },
    {
      "epoch": 1.3256072998046876e-07,
      "model_forward_time": 1.7347040176391602,
      "step": 139
    },
    {
      "epoch": 1.3256072998046876e-07,
      "step": 139,
      "training_step_time": 8.91817593574524
    },
    {
      "epoch": 1.33514404296875e-07,
      "grad_norm": 1.7478151321411133,
      "learning_rate": 2.7800000000000003e-07,
      "loss": 0.4293,
      "step": 140
    },
    {
      "epoch": 1.33514404296875e-07,
      "model_forward_time": 1.7383387088775635,
      "step": 140
    },
    {
      "dynamics_loss_avg": 0.11116686016321183,
      "epoch": 1.33514404296875e-07,
      "step": 140
    },
    {
      "action_loss_avg": 0.2742048852145672,
      "epoch": 1.33514404296875e-07,
      "step": 140
    },
    {
      "epoch": 1.33514404296875e-07,
      "step": 140,
      "training_step_time": 8.903992176055908
    },
    {
      "epoch": 1.3446807861328124e-07,
      "model_forward_time": 1.737027883529663,
      "step": 141
    },
    {
      "epoch": 1.3446807861328124e-07,
      "step": 141,
      "training_step_time": 8.950124263763428
    },
    {
      "epoch": 1.354217529296875e-07,
      "model_forward_time": 1.7210993766784668,
      "step": 142
    },
    {
      "epoch": 1.354217529296875e-07,
      "step": 142,
      "training_step_time": 6.823861122131348
    },
    {
      "epoch": 1.3637542724609374e-07,
      "model_forward_time": 2.8787574768066406,
      "step": 143
    },
    {
      "epoch": 1.3637542724609374e-07,
      "step": 143,
      "training_step_time": 7.896075963973999
    },
    {
      "epoch": 1.373291015625e-07,
      "model_forward_time": 1.7654290199279785,
      "step": 144
    },
    {
      "epoch": 1.373291015625e-07,
      "step": 144,
      "training_step_time": 8.87618374824524
    },
    {
      "epoch": 1.3828277587890624e-07,
      "model_forward_time": 1.3937418460845947,
      "step": 145
    },
    {
      "epoch": 1.3828277587890624e-07,
      "step": 145,
      "training_step_time": 8.92690896987915
    },
    {
      "epoch": 1.392364501953125e-07,
      "model_forward_time": 1.7620811462402344,
      "step": 146
    },
    {
      "epoch": 1.392364501953125e-07,
      "step": 146,
      "training_step_time": 8.948724269866943
    },
    {
      "epoch": 1.4019012451171874e-07,
      "model_forward_time": 1.7496538162231445,
      "step": 147
    },
    {
      "epoch": 1.4019012451171874e-07,
      "step": 147,
      "training_step_time": 8.950151920318604
    },
    {
      "epoch": 1.41143798828125e-07,
      "model_forward_time": 1.7510251998901367,
      "step": 148
    },
    {
      "epoch": 1.41143798828125e-07,
      "step": 148,
      "training_step_time": 6.793968915939331
    },
    {
      "epoch": 1.4209747314453124e-07,
      "model_forward_time": 2.8865966796875,
      "step": 149
    },
    {
      "epoch": 1.4209747314453124e-07,
      "step": 149,
      "training_step_time": 7.922501087188721
    },
    {
      "epoch": 1.430511474609375e-07,
      "grad_norm": 1.8044108152389526,
      "learning_rate": 2.9800000000000005e-07,
      "loss": 0.4229,
      "step": 150
    },
    {
      "epoch": 1.430511474609375e-07,
      "model_forward_time": 1.7285573482513428,
      "step": 150
    },
    {
      "dynamics_loss_avg": 0.09928785599768161,
      "epoch": 1.430511474609375e-07,
      "step": 150
    },
    {
      "action_loss_avg": 0.25437909960746763,
      "epoch": 1.430511474609375e-07,
      "step": 150
    },
    {
      "epoch": 1.430511474609375e-07,
      "step": 150,
      "training_step_time": 8.928001165390015
    },
    {
      "epoch": 1.4400482177734375e-07,
      "model_forward_time": 1.739372730255127,
      "step": 151
    },
    {
      "epoch": 1.4400482177734375e-07,
      "step": 151,
      "training_step_time": 9.045226097106934
    },
    {
      "epoch": 1.4495849609375e-07,
      "model_forward_time": 1.7287466526031494,
      "step": 152
    },
    {
      "epoch": 1.4495849609375e-07,
      "step": 152,
      "training_step_time": 8.996089935302734
    },
    {
      "epoch": 1.4591217041015625e-07,
      "model_forward_time": 1.7245714664459229,
      "step": 153
    },
    {
      "epoch": 1.4591217041015625e-07,
      "step": 153,
      "training_step_time": 9.004014015197754
    },
    {
      "epoch": 1.468658447265625e-07,
      "model_forward_time": 1.7402589321136475,
      "step": 154
    },
    {
      "epoch": 1.468658447265625e-07,
      "step": 154,
      "training_step_time": 8.958032846450806
    },
    {
      "epoch": 1.4781951904296875e-07,
      "model_forward_time": 1.7376580238342285,
      "step": 155
    },
    {
      "epoch": 1.4781951904296875e-07,
      "step": 155,
      "training_step_time": 6.825499057769775
    },
    {
      "epoch": 1.48773193359375e-07,
      "model_forward_time": 2.901076078414917,
      "step": 156
    },
    {
      "epoch": 1.48773193359375e-07,
      "step": 156,
      "training_step_time": 8.021576404571533
    },
    {
      "epoch": 1.4972686767578125e-07,
      "model_forward_time": 1.7480216026306152,
      "step": 157
    },
    {
      "epoch": 1.4972686767578125e-07,
      "step": 157,
      "training_step_time": 9.052130222320557
    },
    {
      "epoch": 1.506805419921875e-07,
      "model_forward_time": 1.746366024017334,
      "step": 158
    },
    {
      "epoch": 1.506805419921875e-07,
      "step": 158,
      "training_step_time": 9.045068979263306
    },
    {
      "epoch": 1.5163421630859375e-07,
      "model_forward_time": 1.7484614849090576,
      "step": 159
    },
    {
      "epoch": 1.5163421630859375e-07,
      "step": 159,
      "training_step_time": 9.015072584152222
    },
    {
      "epoch": 1.52587890625e-07,
      "grad_norm": 2.555778980255127,
      "learning_rate": 3.1800000000000007e-07,
      "loss": 0.4814,
      "step": 160
    },
    {
      "epoch": 1.52587890625e-07,
      "model_forward_time": 1.748063087463379,
      "step": 160
    },
    {
      "dynamics_loss_avg": 0.1333631880581379,
      "epoch": 1.52587890625e-07,
      "step": 160
    },
    {
      "action_loss_avg": 0.33043704479932784,
      "epoch": 1.52587890625e-07,
      "step": 160
    },
    {
      "epoch": 1.52587890625e-07,
      "step": 160,
      "training_step_time": 9.031559467315674
    },
    {
      "epoch": 1.5354156494140625e-07,
      "model_forward_time": 1.7446436882019043,
      "step": 161
    },
    {
      "epoch": 1.5354156494140625e-07,
      "step": 161,
      "training_step_time": 8.938046216964722
    },
    {
      "epoch": 1.544952392578125e-07,
      "model_forward_time": 1.781606912612915,
      "step": 162
    },
    {
      "epoch": 1.544952392578125e-07,
      "step": 162,
      "training_step_time": 6.777897596359253
    },
    {
      "epoch": 1.5544891357421876e-07,
      "model_forward_time": 2.9005234241485596,
      "step": 163
    },
    {
      "epoch": 1.5544891357421876e-07,
      "step": 163,
      "training_step_time": 7.852183103561401
    },
    {
      "epoch": 1.56402587890625e-07,
      "model_forward_time": 1.7358720302581787,
      "step": 164
    },
    {
      "epoch": 1.56402587890625e-07,
      "step": 164,
      "training_step_time": 9.047269821166992
    },
    {
      "epoch": 1.5735626220703126e-07,
      "model_forward_time": 1.4000959396362305,
      "step": 165
    },
    {
      "epoch": 1.5735626220703126e-07,
      "step": 165,
      "training_step_time": 9.035094261169434
    },
    {
      "epoch": 1.583099365234375e-07,
      "model_forward_time": 1.7298393249511719,
      "step": 166
    },
    {
      "epoch": 1.583099365234375e-07,
      "step": 166,
      "training_step_time": 8.969271659851074
    },
    {
      "epoch": 1.5926361083984376e-07,
      "model_forward_time": 1.7481136322021484,
      "step": 167
    },
    {
      "epoch": 1.5926361083984376e-07,
      "step": 167,
      "training_step_time": 8.874678134918213
    },
    {
      "epoch": 1.6021728515625e-07,
      "model_forward_time": 1.7489938735961914,
      "step": 168
    },
    {
      "epoch": 1.6021728515625e-07,
      "step": 168,
      "training_step_time": 8.975467443466187
    },
    {
      "epoch": 1.6117095947265626e-07,
      "model_forward_time": 1.7518343925476074,
      "step": 169
    },
    {
      "epoch": 1.6117095947265626e-07,
      "step": 169,
      "training_step_time": 6.7663490772247314
    },
    {
      "epoch": 1.621246337890625e-07,
      "grad_norm": 1.677869200706482,
      "learning_rate": 3.38e-07,
      "loss": 0.4111,
      "step": 170
    },
    {
      "epoch": 1.621246337890625e-07,
      "model_forward_time": 2.4576587677001953,
      "step": 170
    },
    {
      "dynamics_loss_avg": 0.12005617171525955,
      "epoch": 1.621246337890625e-07,
      "step": 170
    },
    {
      "action_loss_avg": 0.29364987909793855,
      "epoch": 1.621246337890625e-07,
      "step": 170
    },
    {
      "epoch": 1.621246337890625e-07,
      "step": 170,
      "training_step_time": 7.275170087814331
    },
    {
      "epoch": 1.6307830810546876e-07,
      "model_forward_time": 1.7734622955322266,
      "step": 171
    },
    {
      "epoch": 1.6307830810546876e-07,
      "step": 171,
      "training_step_time": 8.552801370620728
    },
    {
      "epoch": 1.64031982421875e-07,
      "model_forward_time": 1.762495994567871,
      "step": 172
    },
    {
      "epoch": 1.64031982421875e-07,
      "step": 172,
      "training_step_time": 8.957066535949707
    },
    {
      "epoch": 1.6498565673828124e-07,
      "model_forward_time": 1.7690181732177734,
      "step": 173
    },
    {
      "epoch": 1.6498565673828124e-07,
      "step": 173,
      "training_step_time": 8.936213493347168
    },
    {
      "epoch": 1.659393310546875e-07,
      "model_forward_time": 1.7589852809906006,
      "step": 174
    },
    {
      "epoch": 1.659393310546875e-07,
      "step": 174,
      "training_step_time": 8.934358596801758
    },
    {
      "epoch": 1.6689300537109374e-07,
      "model_forward_time": 1.7511184215545654,
      "step": 175
    },
    {
      "epoch": 1.6689300537109374e-07,
      "step": 175,
      "training_step_time": 8.849007368087769
    },
    {
      "epoch": 1.678466796875e-07,
      "model_forward_time": 1.7442171573638916,
      "step": 176
    },
    {
      "epoch": 1.678466796875e-07,
      "step": 176,
      "training_step_time": 6.766932725906372
    },
    {
      "epoch": 1.6880035400390624e-07,
      "model_forward_time": 2.8670706748962402,
      "step": 177
    },
    {
      "epoch": 1.6880035400390624e-07,
      "step": 177,
      "training_step_time": 7.8146889209747314
    },
    {
      "epoch": 1.697540283203125e-07,
      "model_forward_time": 1.3806333541870117,
      "step": 178
    },
    {
      "epoch": 1.697540283203125e-07,
      "step": 178,
      "training_step_time": 8.963376522064209
    },
    {
      "epoch": 1.7070770263671874e-07,
      "model_forward_time": 1.758101224899292,
      "step": 179
    },
    {
      "epoch": 1.7070770263671874e-07,
      "step": 179,
      "training_step_time": 8.97921085357666
    },
    {
      "epoch": 1.71661376953125e-07,
      "grad_norm": 2.1630210876464844,
      "learning_rate": 3.58e-07,
      "loss": 0.447,
      "step": 180
    },
    {
      "epoch": 1.71661376953125e-07,
      "model_forward_time": 1.7000939846038818,
      "step": 180
    },
    {
      "dynamics_loss_avg": 0.10979956723749637,
      "epoch": 1.71661376953125e-07,
      "step": 180
    },
    {
      "action_loss_avg": 0.2625324919819832,
      "epoch": 1.71661376953125e-07,
      "step": 180
    },
    {
      "epoch": 1.71661376953125e-07,
      "step": 180,
      "training_step_time": 8.93427848815918
    },
    {
      "epoch": 1.7261505126953124e-07,
      "model_forward_time": 1.711193561553955,
      "step": 181
    },
    {
      "epoch": 1.7261505126953124e-07,
      "step": 181,
      "training_step_time": 9.000589847564697
    },
    {
      "epoch": 1.735687255859375e-07,
      "model_forward_time": 1.7072153091430664,
      "step": 182
    },
    {
      "epoch": 1.735687255859375e-07,
      "step": 182,
      "training_step_time": 9.010826587677002
    },
    {
      "epoch": 1.7452239990234375e-07,
      "model_forward_time": 1.7292110919952393,
      "step": 183
    },
    {
      "epoch": 1.7452239990234375e-07,
      "step": 183,
      "training_step_time": 6.5502095222473145
    },
    {
      "epoch": 1.7547607421875e-07,
      "model_forward_time": 1.7323625087738037,
      "step": 184
    },
    {
      "epoch": 1.7547607421875e-07,
      "step": 184,
      "training_step_time": 8.017616748809814
    },
    {
      "epoch": 1.7642974853515625e-07,
      "model_forward_time": 1.7301077842712402,
      "step": 185
    },
    {
      "epoch": 1.7642974853515625e-07,
      "step": 185,
      "training_step_time": 9.076311349868774
    },
    {
      "epoch": 1.773834228515625e-07,
      "model_forward_time": 1.7297656536102295,
      "step": 186
    },
    {
      "epoch": 1.773834228515625e-07,
      "step": 186,
      "training_step_time": 9.033447742462158
    },
    {
      "epoch": 1.7833709716796875e-07,
      "model_forward_time": 1.7328150272369385,
      "step": 187
    },
    {
      "epoch": 1.7833709716796875e-07,
      "step": 187,
      "training_step_time": 9.02099895477295
    },
    {
      "epoch": 1.79290771484375e-07,
      "model_forward_time": 1.7321033477783203,
      "step": 188
    },
    {
      "epoch": 1.79290771484375e-07,
      "step": 188,
      "training_step_time": 8.997132062911987
    },
    {
      "epoch": 1.8024444580078125e-07,
      "model_forward_time": 1.7226943969726562,
      "step": 189
    },
    {
      "epoch": 1.8024444580078125e-07,
      "step": 189,
      "training_step_time": 7.902080535888672
    },
    {
      "epoch": 1.811981201171875e-07,
      "grad_norm": 1.4010969400405884,
      "learning_rate": 3.78e-07,
      "loss": 0.3855,
      "step": 190
    },
    {
      "epoch": 1.811981201171875e-07,
      "model_forward_time": 1.7047836780548096,
      "step": 190
    },
    {
      "dynamics_loss_avg": 0.09764364417642354,
      "epoch": 1.811981201171875e-07,
      "step": 190
    },
    {
      "action_loss_avg": 0.2760950274765491,
      "epoch": 1.811981201171875e-07,
      "step": 190
    },
    {
      "epoch": 1.811981201171875e-07,
      "step": 190,
      "training_step_time": 7.762536287307739
    },
    {
      "epoch": 1.8215179443359375e-07,
      "model_forward_time": 1.7176940441131592,
      "step": 191
    },
    {
      "epoch": 1.8215179443359375e-07,
      "step": 191,
      "training_step_time": 8.082483530044556
    },
    {
      "epoch": 1.8310546875e-07,
      "model_forward_time": 1.7201194763183594,
      "step": 192
    },
    {
      "epoch": 1.8310546875e-07,
      "step": 192,
      "training_step_time": 9.064448595046997
    },
    {
      "epoch": 1.8405914306640625e-07,
      "model_forward_time": 1.7239456176757812,
      "step": 193
    },
    {
      "epoch": 1.8405914306640625e-07,
      "step": 193,
      "training_step_time": 9.027938604354858
    },
    {
      "epoch": 1.850128173828125e-07,
      "model_forward_time": 1.7234675884246826,
      "step": 194
    },
    {
      "epoch": 1.850128173828125e-07,
      "step": 194,
      "training_step_time": 8.966609716415405
    },
    {
      "epoch": 1.8596649169921876e-07,
      "model_forward_time": 1.732248306274414,
      "step": 195
    },
    {
      "epoch": 1.8596649169921876e-07,
      "step": 195,
      "training_step_time": 9.006797313690186
    },
    {
      "epoch": 1.86920166015625e-07,
      "model_forward_time": 1.6899349689483643,
      "step": 196
    },
    {
      "epoch": 1.86920166015625e-07,
      "step": 196,
      "training_step_time": 7.872893333435059
    },
    {
      "epoch": 1.8787384033203126e-07,
      "model_forward_time": 2.8526549339294434,
      "step": 197
    },
    {
      "epoch": 1.8787384033203126e-07,
      "step": 197,
      "training_step_time": 7.62344765663147
    },
    {
      "epoch": 1.888275146484375e-07,
      "model_forward_time": 1.7196524143218994,
      "step": 198
    },
    {
      "epoch": 1.888275146484375e-07,
      "step": 198,
      "training_step_time": 7.849677324295044
    },
    {
      "epoch": 1.8978118896484376e-07,
      "model_forward_time": 1.7131927013397217,
      "step": 199
    },
    {
      "epoch": 1.8978118896484376e-07,
      "step": 199,
      "training_step_time": 8.601621627807617
    },
    {
      "epoch": 1.9073486328125e-07,
      "grad_norm": 3.142336130142212,
      "learning_rate": 3.9800000000000004e-07,
      "loss": 0.3914,
      "step": 200
    },
    {
      "epoch": 1.9073486328125e-07,
      "model_forward_time": 1.6907598972320557,
      "step": 200
    },
    {
      "dynamics_loss_avg": 0.08556851241737604,
      "epoch": 1.9073486328125e-07,
      "step": 200
    },
    {
      "action_loss_avg": 0.27804946303367617,
      "epoch": 1.9073486328125e-07,
      "step": 200
    },
    {
      "epoch": 1.9073486328125e-07,
      "step": 200,
      "training_step_time": 8.521296262741089
    },
    {
      "epoch": 1.9168853759765626e-07,
      "model_forward_time": 1.7095115184783936,
      "step": 201
    },
    {
      "epoch": 1.9168853759765626e-07,
      "step": 201,
      "training_step_time": 8.915602445602417
    },
    {
      "epoch": 1.926422119140625e-07,
      "model_forward_time": 0.9753479957580566,
      "step": 202
    },
    {
      "epoch": 1.926422119140625e-07,
      "step": 202,
      "training_step_time": 6.7636823654174805
    },
    {
      "epoch": 1.9359588623046876e-07,
      "model_forward_time": 2.268258571624756,
      "step": 203
    },
    {
      "epoch": 1.9359588623046876e-07,
      "step": 203,
      "training_step_time": 7.908738851547241
    },
    {
      "epoch": 1.94549560546875e-07,
      "model_forward_time": 1.3847930431365967,
      "step": 204
    },
    {
      "epoch": 1.94549560546875e-07,
      "step": 204,
      "training_step_time": 8.632330179214478
    },
    {
      "epoch": 1.9550323486328124e-07,
      "model_forward_time": 1.7385780811309814,
      "step": 205
    },
    {
      "epoch": 1.9550323486328124e-07,
      "step": 205,
      "training_step_time": 8.573870182037354
    },
    {
      "epoch": 1.964569091796875e-07,
      "model_forward_time": 1.730618953704834,
      "step": 206
    },
    {
      "epoch": 1.964569091796875e-07,
      "step": 206,
      "training_step_time": 8.602568864822388
    },
    {
      "epoch": 1.9741058349609374e-07,
      "model_forward_time": 1.7242977619171143,
      "step": 207
    },
    {
      "epoch": 1.9741058349609374e-07,
      "step": 207,
      "training_step_time": 8.420811891555786
    },
    {
      "epoch": 1.983642578125e-07,
      "model_forward_time": 1.7854926586151123,
      "step": 208
    },
    {
      "epoch": 1.983642578125e-07,
      "step": 208,
      "training_step_time": 6.74628472328186
    },
    {
      "epoch": 1.9931793212890624e-07,
      "model_forward_time": 1.3903748989105225,
      "step": 209
    },
    {
      "epoch": 1.9931793212890624e-07,
      "step": 209,
      "training_step_time": 7.976067543029785
    },
    {
      "epoch": 2.002716064453125e-07,
      "grad_norm": 1.7091124057769775,
      "learning_rate": 4.18e-07,
      "loss": 0.3643,
      "step": 210
    },
    {
      "epoch": 2.002716064453125e-07,
      "model_forward_time": 1.752152681350708,
      "step": 210
    },
    {
      "dynamics_loss_avg": 0.10914816074073315,
      "epoch": 2.002716064453125e-07,
      "step": 210
    },
    {
      "action_loss_avg": 0.31749114096164704,
      "epoch": 2.002716064453125e-07,
      "step": 210
    },
    {
      "epoch": 2.002716064453125e-07,
      "step": 210,
      "training_step_time": 8.661190748214722
    },
    {
      "epoch": 2.0122528076171874e-07,
      "model_forward_time": 1.750257968902588,
      "step": 211
    },
    {
      "epoch": 2.0122528076171874e-07,
      "step": 211,
      "training_step_time": 8.517831087112427
    },
    {
      "epoch": 2.02178955078125e-07,
      "model_forward_time": 1.7468571662902832,
      "step": 212
    },
    {
      "epoch": 2.02178955078125e-07,
      "step": 212,
      "training_step_time": 8.603549480438232
    },
    {
      "epoch": 2.0313262939453125e-07,
      "model_forward_time": 1.756141185760498,
      "step": 213
    },
    {
      "epoch": 2.0313262939453125e-07,
      "step": 213,
      "training_step_time": 6.683689832687378
    },
    {
      "epoch": 2.040863037109375e-07,
      "model_forward_time": 1.7447905540466309,
      "step": 214
    },
    {
      "epoch": 2.040863037109375e-07,
      "step": 214,
      "training_step_time": 7.810766220092773
    },
    {
      "epoch": 2.0503997802734375e-07,
      "model_forward_time": 1.741164207458496,
      "step": 215
    },
    {
      "epoch": 2.0503997802734375e-07,
      "step": 215,
      "training_step_time": 8.554214239120483
    },
    {
      "epoch": 2.0599365234375e-07,
      "model_forward_time": 1.7441833019256592,
      "step": 216
    },
    {
      "epoch": 2.0599365234375e-07,
      "step": 216,
      "training_step_time": 8.586291313171387
    },
    {
      "epoch": 2.0694732666015625e-07,
      "model_forward_time": 1.765014410018921,
      "step": 217
    },
    {
      "epoch": 2.0694732666015625e-07,
      "step": 217,
      "training_step_time": 8.944851636886597
    },
    {
      "epoch": 2.079010009765625e-07,
      "model_forward_time": 1.753671646118164,
      "step": 218
    },
    {
      "epoch": 2.079010009765625e-07,
      "step": 218,
      "training_step_time": 7.813552141189575
    },
    {
      "epoch": 2.0885467529296875e-07,
      "model_forward_time": 2.86262583732605,
      "step": 219
    },
    {
      "epoch": 2.0885467529296875e-07,
      "step": 219,
      "training_step_time": 7.646305799484253
    },
    {
      "epoch": 2.09808349609375e-07,
      "grad_norm": 2.5078225135803223,
      "learning_rate": 4.3800000000000003e-07,
      "loss": 0.3552,
      "step": 220
    },
    {
      "epoch": 2.09808349609375e-07,
      "model_forward_time": 1.7289879322052002,
      "step": 220
    },
    {
      "dynamics_loss_avg": 0.08561347275972367,
      "epoch": 2.09808349609375e-07,
      "step": 220
    },
    {
      "action_loss_avg": 0.27049022912979126,
      "epoch": 2.09808349609375e-07,
      "step": 220
    },
    {
      "epoch": 2.09808349609375e-07,
      "step": 220,
      "training_step_time": 8.17796516418457
    },
    {
      "epoch": 2.1076202392578125e-07,
      "model_forward_time": 1.7290987968444824,
      "step": 221
    },
    {
      "epoch": 2.1076202392578125e-07,
      "step": 221,
      "training_step_time": 9.017020225524902
    },
    {
      "epoch": 2.117156982421875e-07,
      "model_forward_time": 1.735032558441162,
      "step": 222
    },
    {
      "epoch": 2.117156982421875e-07,
      "step": 222,
      "training_step_time": 8.886026382446289
    },
    {
      "epoch": 2.1266937255859375e-07,
      "model_forward_time": 1.731844425201416,
      "step": 223
    },
    {
      "epoch": 2.1266937255859375e-07,
      "step": 223,
      "training_step_time": 8.977138042449951
    },
    {
      "epoch": 2.13623046875e-07,
      "model_forward_time": 1.7456951141357422,
      "step": 224
    },
    {
      "epoch": 2.13623046875e-07,
      "step": 224,
      "training_step_time": 9.054963827133179
    },
    {
      "epoch": 2.1457672119140626e-07,
      "model_forward_time": 1.7237813472747803,
      "step": 225
    },
    {
      "epoch": 2.1457672119140626e-07,
      "step": 225,
      "training_step_time": 7.903213739395142
    },
    {
      "epoch": 2.155303955078125e-07,
      "model_forward_time": 2.8455810546875,
      "step": 226
    },
    {
      "epoch": 2.155303955078125e-07,
      "step": 226,
      "training_step_time": 7.802817106246948
    },
    {
      "epoch": 2.1648406982421876e-07,
      "model_forward_time": 1.7364895343780518,
      "step": 227
    },
    {
      "epoch": 2.1648406982421876e-07,
      "step": 227,
      "training_step_time": 8.25540804862976
    },
    {
      "epoch": 2.17437744140625e-07,
      "model_forward_time": 1.7405996322631836,
      "step": 228
    },
    {
      "epoch": 2.17437744140625e-07,
      "step": 228,
      "training_step_time": 9.00835132598877
    },
    {
      "epoch": 2.1839141845703126e-07,
      "model_forward_time": 1.7450108528137207,
      "step": 229
    },
    {
      "epoch": 2.1839141845703126e-07,
      "step": 229,
      "training_step_time": 9.041968584060669
    },
    {
      "epoch": 2.193450927734375e-07,
      "grad_norm": 1.7288017272949219,
      "learning_rate": 4.5800000000000005e-07,
      "loss": 0.3887,
      "step": 230
    },
    {
      "epoch": 2.193450927734375e-07,
      "model_forward_time": 1.731623649597168,
      "step": 230
    },
    {
      "dynamics_loss_avg": 0.1048770509660244,
      "epoch": 2.193450927734375e-07,
      "step": 230
    },
    {
      "action_loss_avg": 0.2947812110185623,
      "epoch": 2.193450927734375e-07,
      "step": 230
    },
    {
      "epoch": 2.193450927734375e-07,
      "step": 230,
      "training_step_time": 9.158458471298218
    },
    {
      "epoch": 2.2029876708984376e-07,
      "model_forward_time": 1.7445933818817139,
      "step": 231
    },
    {
      "epoch": 2.2029876708984376e-07,
      "step": 231,
      "training_step_time": 9.045623779296875
    },
    {
      "epoch": 2.2125244140625e-07,
      "model_forward_time": 1.7974603176116943,
      "step": 232
    },
    {
      "epoch": 2.2125244140625e-07,
      "step": 232,
      "training_step_time": 8.304693222045898
    },
    {
      "epoch": 2.2220611572265626e-07,
      "model_forward_time": 1.7814316749572754,
      "step": 233
    },
    {
      "epoch": 2.2220611572265626e-07,
      "step": 233,
      "training_step_time": 7.257607936859131
    },
    {
      "epoch": 2.231597900390625e-07,
      "model_forward_time": 1.7228975296020508,
      "step": 234
    },
    {
      "epoch": 2.231597900390625e-07,
      "step": 234,
      "training_step_time": 7.960320472717285
    },
    {
      "epoch": 2.2411346435546874e-07,
      "model_forward_time": 1.7438158988952637,
      "step": 235
    },
    {
      "epoch": 2.2411346435546874e-07,
      "step": 235,
      "training_step_time": 9.023059129714966
    },
    {
      "epoch": 2.25067138671875e-07,
      "model_forward_time": 1.8529226779937744,
      "step": 236
    },
    {
      "epoch": 2.25067138671875e-07,
      "step": 236,
      "training_step_time": 9.024430513381958
    },
    {
      "epoch": 2.2602081298828124e-07,
      "model_forward_time": 1.7331507205963135,
      "step": 237
    },
    {
      "epoch": 2.2602081298828124e-07,
      "step": 237,
      "training_step_time": 9.083913803100586
    },
    {
      "epoch": 2.269744873046875e-07,
      "model_forward_time": 1.7299067974090576,
      "step": 238
    },
    {
      "epoch": 2.269744873046875e-07,
      "step": 238,
      "training_step_time": 9.08345365524292
    },
    {
      "epoch": 2.2792816162109374e-07,
      "model_forward_time": 1.7268359661102295,
      "step": 239
    },
    {
      "epoch": 2.2792816162109374e-07,
      "step": 239,
      "training_step_time": 8.351972341537476
    },
    {
      "epoch": 2.288818359375e-07,
      "grad_norm": 2.7014882564544678,
      "learning_rate": 4.78e-07,
      "loss": 0.3764,
      "step": 240
    },
    {
      "epoch": 2.288818359375e-07,
      "model_forward_time": 1.7281153202056885,
      "step": 240
    },
    {
      "dynamics_loss_avg": 0.08891693986952305,
      "epoch": 2.288818359375e-07,
      "step": 240
    },
    {
      "action_loss_avg": 0.26362094432115557,
      "epoch": 2.288818359375e-07,
      "step": 240
    },
    {
      "epoch": 2.288818359375e-07,
      "step": 240,
      "training_step_time": 6.9855124950408936
    },
    {
      "epoch": 2.2983551025390624e-07,
      "model_forward_time": 1.7264292240142822,
      "step": 241
    },
    {
      "epoch": 2.2983551025390624e-07,
      "step": 241,
      "training_step_time": 7.893686771392822
    },
    {
      "epoch": 2.307891845703125e-07,
      "model_forward_time": 1.7300055027008057,
      "step": 242
    },
    {
      "epoch": 2.307891845703125e-07,
      "step": 242,
      "training_step_time": 9.02092719078064
    },
    {
      "epoch": 2.3174285888671874e-07,
      "model_forward_time": 1.7410023212432861,
      "step": 243
    },
    {
      "epoch": 2.3174285888671874e-07,
      "step": 243,
      "training_step_time": 9.051255702972412
    },
    {
      "epoch": 2.32696533203125e-07,
      "model_forward_time": 1.726330041885376,
      "step": 244
    },
    {
      "epoch": 2.32696533203125e-07,
      "step": 244,
      "training_step_time": 8.992882490158081
    },
    {
      "epoch": 2.3365020751953125e-07,
      "model_forward_time": 1.7299885749816895,
      "step": 245
    },
    {
      "epoch": 2.3365020751953125e-07,
      "step": 245,
      "training_step_time": 8.97418761253357
    },
    {
      "epoch": 2.346038818359375e-07,
      "model_forward_time": 1.7265405654907227,
      "step": 246
    },
    {
      "epoch": 2.346038818359375e-07,
      "step": 246,
      "training_step_time": 9.06604552268982
    },
    {
      "epoch": 2.3555755615234375e-07,
      "model_forward_time": 1.7280268669128418,
      "step": 247
    },
    {
      "epoch": 2.3555755615234375e-07,
      "step": 247,
      "training_step_time": 8.179302215576172
    },
    {
      "epoch": 2.3651123046875e-07,
      "model_forward_time": 2.251966953277588,
      "step": 248
    },
    {
      "epoch": 2.3651123046875e-07,
      "step": 248,
      "training_step_time": 7.710681438446045
    },
    {
      "epoch": 2.3746490478515625e-07,
      "model_forward_time": 1.739800214767456,
      "step": 249
    },
    {
      "epoch": 2.3746490478515625e-07,
      "step": 249,
      "training_step_time": 8.614752531051636
    },
    {
      "epoch": 2.384185791015625e-07,
      "grad_norm": 3.791624069213867,
      "learning_rate": 4.98e-07,
      "loss": 0.4006,
      "step": 250
    },
    {
      "epoch": 2.384185791015625e-07,
      "model_forward_time": 1.7295844554901123,
      "step": 250
    },
    {
      "dynamics_loss_avg": 0.09331002458930016,
      "epoch": 2.384185791015625e-07,
      "step": 250
    },
    {
      "action_loss_avg": 0.2907704286277294,
      "epoch": 2.384185791015625e-07,
      "step": 250
    },
    {
      "epoch": 2.384185791015625e-07,
      "step": 250,
      "training_step_time": 9.04326868057251
    },
    {
      "epoch": 2.3937225341796875e-07,
      "model_forward_time": 1.7345502376556396,
      "step": 251
    },
    {
      "epoch": 2.3937225341796875e-07,
      "step": 251,
      "training_step_time": 9.097055673599243
    },
    {
      "epoch": 2.40325927734375e-07,
      "model_forward_time": 1.7494285106658936,
      "step": 252
    },
    {
      "epoch": 2.40325927734375e-07,
      "step": 252,
      "training_step_time": 8.974069356918335
    },
    {
      "epoch": 2.4127960205078125e-07,
      "model_forward_time": 1.735612154006958,
      "step": 253
    },
    {
      "epoch": 2.4127960205078125e-07,
      "step": 253,
      "training_step_time": 9.051035642623901
    },
    {
      "epoch": 2.422332763671875e-07,
      "model_forward_time": 1.7240948677062988,
      "step": 254
    },
    {
      "epoch": 2.422332763671875e-07,
      "step": 254,
      "training_step_time": 8.34707760810852
    },
    {
      "epoch": 2.4318695068359375e-07,
      "model_forward_time": 1.8086090087890625,
      "step": 255
    },
    {
      "epoch": 2.4318695068359375e-07,
      "step": 255,
      "training_step_time": 7.2632246017456055
    },
    {
      "epoch": 2.44140625e-07,
      "model_forward_time": 1.7395198345184326,
      "step": 256
    },
    {
      "epoch": 2.44140625e-07,
      "step": 256,
      "training_step_time": 7.9907166957855225
    },
    {
      "epoch": 2.4509429931640626e-07,
      "model_forward_time": 1.7386131286621094,
      "step": 257
    },
    {
      "epoch": 2.4509429931640626e-07,
      "step": 257,
      "training_step_time": 9.050305366516113
    },
    {
      "epoch": 2.460479736328125e-07,
      "model_forward_time": 1.7435188293457031,
      "step": 258
    },
    {
      "epoch": 2.460479736328125e-07,
      "step": 258,
      "training_step_time": 9.05015754699707
    },
    {
      "epoch": 2.4700164794921876e-07,
      "model_forward_time": 1.745879888534546,
      "step": 259
    },
    {
      "epoch": 2.4700164794921876e-07,
      "step": 259,
      "training_step_time": 9.035665035247803
    },
    {
      "epoch": 2.47955322265625e-07,
      "grad_norm": 3.7327256202697754,
      "learning_rate": 5.180000000000001e-07,
      "loss": 0.3765,
      "step": 260
    },
    {
      "epoch": 2.47955322265625e-07,
      "model_forward_time": 1.7506113052368164,
      "step": 260
    },
    {
      "dynamics_loss_avg": 0.10259649641811848,
      "epoch": 2.47955322265625e-07,
      "step": 260
    },
    {
      "action_loss_avg": 0.30325575470924376,
      "epoch": 2.47955322265625e-07,
      "step": 260
    },
    {
      "epoch": 2.47955322265625e-07,
      "step": 260,
      "training_step_time": 8.988666296005249
    },
    {
      "epoch": 2.4890899658203126e-07,
      "model_forward_time": 1.7420408725738525,
      "step": 261
    },
    {
      "epoch": 2.4890899658203126e-07,
      "step": 261,
      "training_step_time": 8.403944492340088
    },
    {
      "epoch": 2.498626708984375e-07,
      "model_forward_time": 1.793360710144043,
      "step": 262
    },
    {
      "epoch": 2.498626708984375e-07,
      "step": 262,
      "training_step_time": 7.186270236968994
    },
    {
      "epoch": 2.5081634521484376e-07,
      "model_forward_time": 1.726327657699585,
      "step": 263
    },
    {
      "epoch": 2.5081634521484376e-07,
      "step": 263,
      "training_step_time": 7.9883363246917725
    },
    {
      "epoch": 2.5177001953125e-07,
      "model_forward_time": 1.723296880722046,
      "step": 264
    },
    {
      "epoch": 2.5177001953125e-07,
      "step": 264,
      "training_step_time": 9.019120693206787
    },
    {
      "epoch": 2.5272369384765626e-07,
      "model_forward_time": 1.734086513519287,
      "step": 265
    },
    {
      "epoch": 2.5272369384765626e-07,
      "step": 265,
      "training_step_time": 9.048922300338745
    },
    {
      "epoch": 2.536773681640625e-07,
      "model_forward_time": 1.723168134689331,
      "step": 266
    },
    {
      "epoch": 2.536773681640625e-07,
      "step": 266,
      "training_step_time": 9.059417247772217
    },
    {
      "epoch": 2.5463104248046876e-07,
      "model_forward_time": 1.743408203125,
      "step": 267
    },
    {
      "epoch": 2.5463104248046876e-07,
      "step": 267,
      "training_step_time": 9.058916330337524
    },
    {
      "epoch": 2.55584716796875e-07,
      "model_forward_time": 1.7391891479492188,
      "step": 268
    },
    {
      "epoch": 2.55584716796875e-07,
      "step": 268,
      "training_step_time": 8.912872552871704
    },
    {
      "epoch": 2.5653839111328127e-07,
      "model_forward_time": 1.7542898654937744,
      "step": 269
    },
    {
      "epoch": 2.5653839111328127e-07,
      "step": 269,
      "training_step_time": 6.856276988983154
    },
    {
      "epoch": 2.574920654296875e-07,
      "grad_norm": 1.7370169162750244,
      "learning_rate": 5.380000000000001e-07,
      "loss": 0.361,
      "step": 270
    },
    {
      "epoch": 2.574920654296875e-07,
      "model_forward_time": 2.3454668521881104,
      "step": 270
    },
    {
      "dynamics_loss_avg": 0.08291096184402705,
      "epoch": 2.574920654296875e-07,
      "step": 270
    },
    {
      "action_loss_avg": 0.2396176666021347,
      "epoch": 2.574920654296875e-07,
      "step": 270
    },
    {
      "epoch": 2.574920654296875e-07,
      "step": 270,
      "training_step_time": 7.5091776847839355
    },
    {
      "epoch": 2.5844573974609377e-07,
      "model_forward_time": 1.7392077445983887,
      "step": 271
    },
    {
      "epoch": 2.5844573974609377e-07,
      "step": 271,
      "training_step_time": 8.65517258644104
    },
    {
      "epoch": 2.593994140625e-07,
      "model_forward_time": 1.7298669815063477,
      "step": 272
    },
    {
      "epoch": 2.593994140625e-07,
      "step": 272,
      "training_step_time": 8.826265096664429
    },
    {
      "epoch": 2.6035308837890627e-07,
      "model_forward_time": 1.7320568561553955,
      "step": 273
    },
    {
      "epoch": 2.6035308837890627e-07,
      "step": 273,
      "training_step_time": 8.890555381774902
    },
    {
      "epoch": 2.613067626953125e-07,
      "model_forward_time": 1.727872610092163,
      "step": 274
    },
    {
      "epoch": 2.613067626953125e-07,
      "step": 274,
      "training_step_time": 8.7730131149292
    },
    {
      "epoch": 2.6226043701171877e-07,
      "model_forward_time": 1.723893165588379,
      "step": 275
    },
    {
      "epoch": 2.6226043701171877e-07,
      "step": 275,
      "training_step_time": 7.952938795089722
    },
    {
      "epoch": 2.63214111328125e-07,
      "model_forward_time": 2.2827982902526855,
      "step": 276
    },
    {
      "epoch": 2.63214111328125e-07,
      "step": 276,
      "training_step_time": 7.580907821655273
    },
    {
      "epoch": 2.6416778564453127e-07,
      "model_forward_time": 1.3650200366973877,
      "step": 277
    },
    {
      "epoch": 2.6416778564453127e-07,
      "step": 277,
      "training_step_time": 8.044291734695435
    },
    {
      "epoch": 2.651214599609375e-07,
      "model_forward_time": 1.3706843852996826,
      "step": 278
    },
    {
      "epoch": 2.651214599609375e-07,
      "step": 278,
      "training_step_time": 8.821063041687012
    },
    {
      "epoch": 2.660751342773438e-07,
      "model_forward_time": 1.7334215641021729,
      "step": 279
    },
    {
      "epoch": 2.660751342773438e-07,
      "step": 279,
      "training_step_time": 8.810692071914673
    },
    {
      "epoch": 2.6702880859375e-07,
      "grad_norm": 3.224087953567505,
      "learning_rate": 5.580000000000001e-07,
      "loss": 0.3775,
      "step": 280
    },
    {
      "epoch": 2.6702880859375e-07,
      "model_forward_time": 1.7373385429382324,
      "step": 280
    },
    {
      "dynamics_loss_avg": 0.09444050844758749,
      "epoch": 2.6702880859375e-07,
      "step": 280
    },
    {
      "action_loss_avg": 0.2788097470998764,
      "epoch": 2.6702880859375e-07,
      "step": 280
    },
    {
      "epoch": 2.6702880859375e-07,
      "step": 280,
      "training_step_time": 9.070829153060913
    },
    {
      "epoch": 2.679824829101563e-07,
      "model_forward_time": 1.7238006591796875,
      "step": 281
    },
    {
      "epoch": 2.679824829101563e-07,
      "step": 281,
      "training_step_time": 9.146883487701416
    },
    {
      "epoch": 2.689361572265625e-07,
      "model_forward_time": 1.7377638816833496,
      "step": 282
    },
    {
      "epoch": 2.689361572265625e-07,
      "step": 282,
      "training_step_time": 7.914579153060913
    },
    {
      "epoch": 2.698898315429687e-07,
      "model_forward_time": 2.8185174465179443,
      "step": 283
    },
    {
      "epoch": 2.698898315429687e-07,
      "step": 283,
      "training_step_time": 7.623045206069946
    },
    {
      "epoch": 2.70843505859375e-07,
      "model_forward_time": 1.7142045497894287,
      "step": 284
    },
    {
      "epoch": 2.70843505859375e-07,
      "step": 284,
      "training_step_time": 8.54141116142273
    },
    {
      "epoch": 2.7179718017578123e-07,
      "model_forward_time": 1.6946709156036377,
      "step": 285
    },
    {
      "epoch": 2.7179718017578123e-07,
      "step": 285,
      "training_step_time": 9.082579612731934
    },
    {
      "epoch": 2.727508544921875e-07,
      "model_forward_time": 1.7065308094024658,
      "step": 286
    },
    {
      "epoch": 2.727508544921875e-07,
      "step": 286,
      "training_step_time": 8.091939449310303
    },
    {
      "epoch": 2.7370452880859373e-07,
      "model_forward_time": 1.7024445533752441,
      "step": 287
    },
    {
      "epoch": 2.7370452880859373e-07,
      "step": 287,
      "training_step_time": 6.876184701919556
    },
    {
      "epoch": 2.74658203125e-07,
      "model_forward_time": 1.706920862197876,
      "step": 288
    },
    {
      "epoch": 2.74658203125e-07,
      "step": 288,
      "training_step_time": 6.868406534194946
    },
    {
      "epoch": 2.7561187744140623e-07,
      "model_forward_time": 1.7048664093017578,
      "step": 289
    },
    {
      "epoch": 2.7561187744140623e-07,
      "step": 289,
      "training_step_time": 6.8676488399505615
    },
    {
      "epoch": 2.765655517578125e-07,
      "grad_norm": 1.8819568157196045,
      "learning_rate": 5.78e-07,
      "loss": 0.3819,
      "step": 290
    },
    {
      "epoch": 2.765655517578125e-07,
      "model_forward_time": 1.7036092281341553,
      "step": 290
    },
    {
      "dynamics_loss_avg": 0.09286041408777237,
      "epoch": 2.765655517578125e-07,
      "step": 290
    },
    {
      "action_loss_avg": 0.3179521709680557,
      "epoch": 2.765655517578125e-07,
      "step": 290
    },
    {
      "epoch": 2.765655517578125e-07,
      "step": 290,
      "training_step_time": 6.893681287765503
    },
    {
      "epoch": 2.7751922607421873e-07,
      "model_forward_time": 1.7066128253936768,
      "step": 291
    },
    {
      "epoch": 2.7751922607421873e-07,
      "step": 291,
      "training_step_time": 6.892820119857788
    },
    {
      "epoch": 2.78472900390625e-07,
      "model_forward_time": 1.7047221660614014,
      "step": 292
    },
    {
      "epoch": 2.78472900390625e-07,
      "step": 292,
      "training_step_time": 6.894377708435059
    },
    {
      "epoch": 2.7942657470703123e-07,
      "model_forward_time": 1.7102515697479248,
      "step": 293
    },
    {
      "epoch": 2.7942657470703123e-07,
      "step": 293,
      "training_step_time": 6.859716892242432
    },
    {
      "epoch": 2.803802490234375e-07,
      "model_forward_time": 1.7019314765930176,
      "step": 294
    },
    {
      "epoch": 2.803802490234375e-07,
      "step": 294,
      "training_step_time": 6.786156892776489
    },
    {
      "epoch": 2.8133392333984374e-07,
      "model_forward_time": 2.2585980892181396,
      "step": 295
    },
    {
      "epoch": 2.8133392333984374e-07,
      "step": 295,
      "training_step_time": 7.783655166625977
    },
    {
      "epoch": 2.8228759765625e-07,
      "model_forward_time": 1.707798719406128,
      "step": 296
    },
    {
      "epoch": 2.8228759765625e-07,
      "step": 296,
      "training_step_time": 7.938898324966431
    },
    {
      "epoch": 2.8324127197265624e-07,
      "model_forward_time": 1.7110707759857178,
      "step": 297
    },
    {
      "epoch": 2.8324127197265624e-07,
      "step": 297,
      "training_step_time": 6.8673765659332275
    },
    {
      "epoch": 2.841949462890625e-07,
      "model_forward_time": 1.694300651550293,
      "step": 298
    },
    {
      "epoch": 2.841949462890625e-07,
      "step": 298,
      "training_step_time": 6.850127220153809
    },
    {
      "epoch": 2.8514862060546874e-07,
      "model_forward_time": 1.7186129093170166,
      "step": 299
    },
    {
      "epoch": 2.8514862060546874e-07,
      "step": 299,
      "training_step_time": 6.895806312561035
    },
    {
      "epoch": 2.86102294921875e-07,
      "grad_norm": 1.4566606283187866,
      "learning_rate": 5.98e-07,
      "loss": 0.304,
      "step": 300
    },
    {
      "epoch": 2.86102294921875e-07,
      "model_forward_time": 1.3384215831756592,
      "step": 300
    },
    {
      "dynamics_loss_avg": 0.07839864790439606,
      "epoch": 2.86102294921875e-07,
      "step": 300
    },
    {
      "action_loss_avg": 0.21521787494421005,
      "epoch": 2.86102294921875e-07,
      "step": 300
    },
    {
      "epoch": 2.86102294921875e-07,
      "step": 300,
      "training_step_time": 6.851764440536499
    },
    {
      "epoch": 2.8705596923828124e-07,
      "model_forward_time": 1.7137401103973389,
      "step": 301
    },
    {
      "epoch": 2.8705596923828124e-07,
      "step": 301,
      "training_step_time": 6.860678434371948
    },
    {
      "epoch": 2.880096435546875e-07,
      "model_forward_time": 1.7094995975494385,
      "step": 302
    },
    {
      "epoch": 2.880096435546875e-07,
      "step": 302,
      "training_step_time": 6.921575546264648
    },
    {
      "epoch": 2.8896331787109374e-07,
      "model_forward_time": 1.7176876068115234,
      "step": 303
    },
    {
      "epoch": 2.8896331787109374e-07,
      "step": 303,
      "training_step_time": 6.864233016967773
    },
    {
      "epoch": 2.899169921875e-07,
      "model_forward_time": 2.2725281715393066,
      "step": 304
    },
    {
      "epoch": 2.899169921875e-07,
      "step": 304,
      "training_step_time": 7.769463777542114
    },
    {
      "epoch": 2.9087066650390624e-07,
      "model_forward_time": 1.7052297592163086,
      "step": 305
    },
    {
      "epoch": 2.9087066650390624e-07,
      "step": 305,
      "training_step_time": 7.871997356414795
    },
    {
      "epoch": 2.918243408203125e-07,
      "model_forward_time": 1.7081835269927979,
      "step": 306
    },
    {
      "epoch": 2.918243408203125e-07,
      "step": 306,
      "training_step_time": 7.656690359115601
    },
    {
      "epoch": 2.9277801513671875e-07,
      "model_forward_time": 1.7068052291870117,
      "step": 307
    },
    {
      "epoch": 2.9277801513671875e-07,
      "step": 307,
      "training_step_time": 6.874790668487549
    },
    {
      "epoch": 2.93731689453125e-07,
      "model_forward_time": 1.7074215412139893,
      "step": 308
    },
    {
      "epoch": 2.93731689453125e-07,
      "step": 308,
      "training_step_time": 6.860036611557007
    },
    {
      "epoch": 2.9468536376953125e-07,
      "model_forward_time": 1.7126390933990479,
      "step": 309
    },
    {
      "epoch": 2.9468536376953125e-07,
      "step": 309,
      "training_step_time": 6.868224382400513
    },
    {
      "epoch": 2.956390380859375e-07,
      "grad_norm": 2.4572103023529053,
      "learning_rate": 6.180000000000001e-07,
      "loss": 0.366,
      "step": 310
    },
    {
      "epoch": 2.956390380859375e-07,
      "model_forward_time": 1.3293261528015137,
      "step": 310
    },
    {
      "dynamics_loss_avg": 0.09042210690677166,
      "epoch": 2.956390380859375e-07,
      "step": 310
    },
    {
      "action_loss_avg": 0.20640290565788746,
      "epoch": 2.956390380859375e-07,
      "step": 310
    },
    {
      "epoch": 2.956390380859375e-07,
      "step": 310,
      "training_step_time": 6.85309100151062
    },
    {
      "epoch": 2.9659271240234375e-07,
      "model_forward_time": 1.714186668395996,
      "step": 311
    },
    {
      "epoch": 2.9659271240234375e-07,
      "step": 311,
      "training_step_time": 6.846043109893799
    },
    {
      "epoch": 2.9754638671875e-07,
      "model_forward_time": 1.7059133052825928,
      "step": 312
    },
    {
      "epoch": 2.9754638671875e-07,
      "step": 312,
      "training_step_time": 6.85648250579834
    },
    {
      "epoch": 2.9850006103515625e-07,
      "model_forward_time": 2.2706499099731445,
      "step": 313
    },
    {
      "epoch": 2.9850006103515625e-07,
      "step": 313,
      "training_step_time": 7.755636692047119
    },
    {
      "epoch": 2.994537353515625e-07,
      "model_forward_time": 0.695105791091919,
      "step": 314
    },
    {
      "epoch": 2.994537353515625e-07,
      "step": 314,
      "training_step_time": 8.889237642288208
    },
    {
      "epoch": 3.0040740966796875e-07,
      "model_forward_time": 1.7047431468963623,
      "step": 315
    },
    {
      "epoch": 3.0040740966796875e-07,
      "step": 315,
      "training_step_time": 7.616554498672485
    },
    {
      "epoch": 3.01361083984375e-07,
      "model_forward_time": 1.711437702178955,
      "step": 316
    },
    {
      "epoch": 3.01361083984375e-07,
      "step": 316,
      "training_step_time": 6.846878290176392
    },
    {
      "epoch": 3.0231475830078125e-07,
      "model_forward_time": 1.7034492492675781,
      "step": 317
    },
    {
      "epoch": 3.0231475830078125e-07,
      "step": 317,
      "training_step_time": 6.736553192138672
    },
    {
      "epoch": 3.032684326171875e-07,
      "model_forward_time": 1.7147316932678223,
      "step": 318
    },
    {
      "epoch": 3.032684326171875e-07,
      "step": 318,
      "training_step_time": 6.905012130737305
    },
    {
      "epoch": 3.0422210693359376e-07,
      "model_forward_time": 1.3384068012237549,
      "step": 319
    },
    {
      "epoch": 3.0422210693359376e-07,
      "step": 319,
      "training_step_time": 6.8606157302856445
    },
    {
      "epoch": 3.0517578125e-07,
      "grad_norm": 1.4175788164138794,
      "learning_rate": 6.38e-07,
      "loss": 0.3412,
      "step": 320
    },
    {
      "epoch": 3.0517578125e-07,
      "model_forward_time": 1.714479923248291,
      "step": 320
    },
    {
      "dynamics_loss_avg": 0.08611302115023137,
      "epoch": 3.0517578125e-07,
      "step": 320
    },
    {
      "action_loss_avg": 0.2279242567718029,
      "epoch": 3.0517578125e-07,
      "step": 320
    },
    {
      "epoch": 3.0517578125e-07,
      "step": 320,
      "training_step_time": 6.849871397018433
    },
    {
      "epoch": 3.0612945556640626e-07,
      "model_forward_time": 1.688216209411621,
      "step": 321
    },
    {
      "epoch": 3.0612945556640626e-07,
      "step": 321,
      "training_step_time": 6.817328929901123
    },
    {
      "epoch": 3.070831298828125e-07,
      "model_forward_time": 1.7186005115509033,
      "step": 322
    },
    {
      "epoch": 3.070831298828125e-07,
      "step": 322,
      "training_step_time": 8.21769666671753
    },
    {
      "epoch": 3.0803680419921876e-07,
      "model_forward_time": 2.235898971557617,
      "step": 323
    },
    {
      "epoch": 3.0803680419921876e-07,
      "step": 323,
      "training_step_time": 11.7734694480896
    },
    {
      "epoch": 3.08990478515625e-07,
      "model_forward_time": 1.7035541534423828,
      "step": 324
    },
    {
      "epoch": 3.08990478515625e-07,
      "step": 324,
      "training_step_time": 7.664727449417114
    },
    {
      "epoch": 3.0994415283203126e-07,
      "model_forward_time": 1.7095129489898682,
      "step": 325
    },
    {
      "epoch": 3.0994415283203126e-07,
      "step": 325,
      "training_step_time": 6.86543607711792
    },
    {
      "epoch": 3.108978271484375e-07,
      "model_forward_time": 1.6970806121826172,
      "step": 326
    },
    {
      "epoch": 3.108978271484375e-07,
      "step": 326,
      "training_step_time": 6.863741397857666
    },
    {
      "epoch": 3.1185150146484376e-07,
      "model_forward_time": 1.7103807926177979,
      "step": 327
    },
    {
      "epoch": 3.1185150146484376e-07,
      "step": 327,
      "training_step_time": 6.855975151062012
    },
    {
      "epoch": 3.1280517578125e-07,
      "model_forward_time": 1.7050564289093018,
      "step": 328
    },
    {
      "epoch": 3.1280517578125e-07,
      "step": 328,
      "training_step_time": 6.908839702606201
    },
    {
      "epoch": 3.1375885009765626e-07,
      "model_forward_time": 1.7038097381591797,
      "step": 329
    },
    {
      "epoch": 3.1375885009765626e-07,
      "step": 329,
      "training_step_time": 6.847212314605713
    },
    {
      "epoch": 3.147125244140625e-07,
      "grad_norm": 2.906271457672119,
      "learning_rate": 6.58e-07,
      "loss": 0.3812,
      "step": 330
    },
    {
      "epoch": 3.147125244140625e-07,
      "model_forward_time": 1.7014195919036865,
      "step": 330
    },
    {
      "dynamics_loss_avg": 0.10248811393976212,
      "epoch": 3.147125244140625e-07,
      "step": 330
    },
    {
      "action_loss_avg": 0.3192122936248779,
      "epoch": 3.147125244140625e-07,
      "step": 330
    },
    {
      "epoch": 3.147125244140625e-07,
      "step": 330,
      "training_step_time": 6.829528093338013
    },
    {
      "epoch": 3.1566619873046877e-07,
      "model_forward_time": 1.7056443691253662,
      "step": 331
    },
    {
      "epoch": 3.1566619873046877e-07,
      "step": 331,
      "training_step_time": 7.686388254165649
    },
    {
      "epoch": 3.16619873046875e-07,
      "model_forward_time": 2.2494945526123047,
      "step": 332
    },
    {
      "epoch": 3.16619873046875e-07,
      "step": 332,
      "training_step_time": 10.629199266433716
    },
    {
      "epoch": 3.1757354736328127e-07,
      "model_forward_time": 1.718130350112915,
      "step": 333
    },
    {
      "epoch": 3.1757354736328127e-07,
      "step": 333,
      "training_step_time": 7.931608200073242
    },
    {
      "epoch": 3.185272216796875e-07,
      "model_forward_time": 1.7154483795166016,
      "step": 334
    },
    {
      "epoch": 3.185272216796875e-07,
      "step": 334,
      "training_step_time": 7.666808128356934
    },
    {
      "epoch": 3.1948089599609377e-07,
      "model_forward_time": 1.7174279689788818,
      "step": 335
    },
    {
      "epoch": 3.1948089599609377e-07,
      "step": 335,
      "training_step_time": 6.842065334320068
    },
    {
      "epoch": 3.204345703125e-07,
      "model_forward_time": 1.7146494388580322,
      "step": 336
    },
    {
      "epoch": 3.204345703125e-07,
      "step": 336,
      "training_step_time": 6.905775785446167
    },
    {
      "epoch": 3.2138824462890627e-07,
      "model_forward_time": 1.7129864692687988,
      "step": 337
    },
    {
      "epoch": 3.2138824462890627e-07,
      "step": 337,
      "training_step_time": 6.856630325317383
    },
    {
      "epoch": 3.223419189453125e-07,
      "model_forward_time": 1.7273669242858887,
      "step": 338
    },
    {
      "epoch": 3.223419189453125e-07,
      "step": 338,
      "training_step_time": 6.849541902542114
    },
    {
      "epoch": 3.2329559326171877e-07,
      "model_forward_time": 1.7248039245605469,
      "step": 339
    },
    {
      "epoch": 3.2329559326171877e-07,
      "step": 339,
      "training_step_time": 6.860366582870483
    },
    {
      "epoch": 3.24249267578125e-07,
      "grad_norm": 1.6156812906265259,
      "learning_rate": 6.78e-07,
      "loss": 0.3635,
      "step": 340
    },
    {
      "epoch": 3.24249267578125e-07,
      "model_forward_time": 1.7285945415496826,
      "step": 340
    },
    {
      "dynamics_loss_avg": 0.08467992581427097,
      "epoch": 3.24249267578125e-07,
      "step": 340
    },
    {
      "action_loss_avg": 0.23825230151414872,
      "epoch": 3.24249267578125e-07,
      "step": 340
    },
    {
      "epoch": 3.24249267578125e-07,
      "step": 340,
      "training_step_time": 6.859499454498291
    },
    {
      "epoch": 3.252029418945313e-07,
      "model_forward_time": 1.7557265758514404,
      "step": 341
    },
    {
      "epoch": 3.252029418945313e-07,
      "step": 341,
      "training_step_time": 8.558354139328003
    },
    {
      "epoch": 3.261566162109375e-07,
      "model_forward_time": 1.7276887893676758,
      "step": 342
    },
    {
      "epoch": 3.261566162109375e-07,
      "step": 342,
      "training_step_time": 8.549750804901123
    },
    {
      "epoch": 3.271102905273438e-07,
      "model_forward_time": 1.7166292667388916,
      "step": 343
    },
    {
      "epoch": 3.271102905273438e-07,
      "step": 343,
      "training_step_time": 7.589193344116211
    },
    {
      "epoch": 3.2806396484375e-07,
      "model_forward_time": 1.723330020904541,
      "step": 344
    },
    {
      "epoch": 3.2806396484375e-07,
      "step": 344,
      "training_step_time": 6.87442684173584
    },
    {
      "epoch": 3.290176391601562e-07,
      "model_forward_time": 1.7139668464660645,
      "step": 345
    },
    {
      "epoch": 3.290176391601562e-07,
      "step": 345,
      "training_step_time": 6.855194807052612
    },
    {
      "epoch": 3.299713134765625e-07,
      "model_forward_time": 1.7247703075408936,
      "step": 346
    },
    {
      "epoch": 3.299713134765625e-07,
      "step": 346,
      "training_step_time": 6.857741832733154
    },
    {
      "epoch": 3.309249877929687e-07,
      "model_forward_time": 1.7340044975280762,
      "step": 347
    },
    {
      "epoch": 3.309249877929687e-07,
      "step": 347,
      "training_step_time": 6.881190776824951
    },
    {
      "epoch": 3.31878662109375e-07,
      "model_forward_time": 1.7234601974487305,
      "step": 348
    },
    {
      "epoch": 3.31878662109375e-07,
      "step": 348,
      "training_step_time": 6.852786064147949
    },
    {
      "epoch": 3.3283233642578123e-07,
      "model_forward_time": 1.7326469421386719,
      "step": 349
    },
    {
      "epoch": 3.3283233642578123e-07,
      "step": 349,
      "training_step_time": 6.8478968143463135
    },
    {
      "epoch": 3.337860107421875e-07,
      "grad_norm": 2.2447850704193115,
      "learning_rate": 6.98e-07,
      "loss": 0.4028,
      "step": 350
    },
    {
      "epoch": 3.337860107421875e-07,
      "model_forward_time": 1.7453160285949707,
      "step": 350
    },
    {
      "dynamics_loss_avg": 0.09385906495153903,
      "epoch": 3.337860107421875e-07,
      "step": 350
    },
    {
      "action_loss_avg": 0.27478640154004097,
      "epoch": 3.337860107421875e-07,
      "step": 350
    },
    {
      "epoch": 3.337860107421875e-07,
      "step": 350,
      "training_step_time": 7.752504110336304
    },
    {
      "epoch": 3.3473968505859373e-07,
      "model_forward_time": 2.262648105621338,
      "step": 351
    },
    {
      "epoch": 3.3473968505859373e-07,
      "step": 351,
      "training_step_time": 10.479159832000732
    },
    {
      "epoch": 3.35693359375e-07,
      "model_forward_time": 1.7249853610992432,
      "step": 352
    },
    {
      "epoch": 3.35693359375e-07,
      "step": 352,
      "training_step_time": 7.75569748878479
    },
    {
      "epoch": 3.3664703369140623e-07,
      "model_forward_time": 1.7293038368225098,
      "step": 353
    },
    {
      "epoch": 3.3664703369140623e-07,
      "step": 353,
      "training_step_time": 6.829831123352051
    },
    {
      "epoch": 3.376007080078125e-07,
      "model_forward_time": 1.7253434658050537,
      "step": 354
    },
    {
      "epoch": 3.376007080078125e-07,
      "step": 354,
      "training_step_time": 6.749956369400024
    },
    {
      "epoch": 3.3855438232421873e-07,
      "model_forward_time": 1.728459358215332,
      "step": 355
    },
    {
      "epoch": 3.3855438232421873e-07,
      "step": 355,
      "training_step_time": 6.838656663894653
    },
    {
      "epoch": 3.39508056640625e-07,
      "model_forward_time": 1.7239911556243896,
      "step": 356
    },
    {
      "epoch": 3.39508056640625e-07,
      "step": 356,
      "training_step_time": 6.825570583343506
    },
    {
      "epoch": 3.4046173095703123e-07,
      "model_forward_time": 1.7211039066314697,
      "step": 357
    },
    {
      "epoch": 3.4046173095703123e-07,
      "step": 357,
      "training_step_time": 6.8368494510650635
    },
    {
      "epoch": 3.414154052734375e-07,
      "model_forward_time": 1.3470990657806396,
      "step": 358
    },
    {
      "epoch": 3.414154052734375e-07,
      "step": 358,
      "training_step_time": 6.884784936904907
    },
    {
      "epoch": 3.4236907958984374e-07,
      "model_forward_time": 1.768777847290039,
      "step": 359
    },
    {
      "epoch": 3.4236907958984374e-07,
      "step": 359,
      "training_step_time": 7.836175918579102
    },
    {
      "epoch": 3.4332275390625e-07,
      "grad_norm": 3.481602907180786,
      "learning_rate": 7.18e-07,
      "loss": 0.3584,
      "step": 360
    },
    {
      "epoch": 3.4332275390625e-07,
      "model_forward_time": 2.25860595703125,
      "step": 360
    },
    {
      "dynamics_loss_avg": 0.09422270450741052,
      "epoch": 3.4332275390625e-07,
      "step": 360
    },
    {
      "action_loss_avg": 0.26617863923311236,
      "epoch": 3.4332275390625e-07,
      "step": 360
    },
    {
      "epoch": 3.4332275390625e-07,
      "step": 360,
      "training_step_time": 9.761659145355225
    },
    {
      "epoch": 3.4427642822265624e-07,
      "model_forward_time": 1.7287085056304932,
      "step": 361
    },
    {
      "epoch": 3.4427642822265624e-07,
      "step": 361,
      "training_step_time": 7.901093244552612
    },
    {
      "epoch": 3.452301025390625e-07,
      "model_forward_time": 1.7336907386779785,
      "step": 362
    },
    {
      "epoch": 3.452301025390625e-07,
      "step": 362,
      "training_step_time": 7.603902578353882
    },
    {
      "epoch": 3.4618377685546874e-07,
      "model_forward_time": 1.732496738433838,
      "step": 363
    },
    {
      "epoch": 3.4618377685546874e-07,
      "step": 363,
      "training_step_time": 6.924683570861816
    },
    {
      "epoch": 3.47137451171875e-07,
      "model_forward_time": 1.707261085510254,
      "step": 364
    },
    {
      "epoch": 3.47137451171875e-07,
      "step": 364,
      "training_step_time": 6.902386665344238
    },
    {
      "epoch": 3.4809112548828124e-07,
      "model_forward_time": 1.7331147193908691,
      "step": 365
    },
    {
      "epoch": 3.4809112548828124e-07,
      "step": 365,
      "training_step_time": 6.970314979553223
    },
    {
      "epoch": 3.490447998046875e-07,
      "model_forward_time": 1.726210355758667,
      "step": 366
    },
    {
      "epoch": 3.490447998046875e-07,
      "step": 366,
      "training_step_time": 6.915976524353027
    },
    {
      "epoch": 3.4999847412109374e-07,
      "model_forward_time": 1.733487844467163,
      "step": 367
    },
    {
      "epoch": 3.4999847412109374e-07,
      "step": 367,
      "training_step_time": 6.945109128952026
    },
    {
      "epoch": 3.509521484375e-07,
      "model_forward_time": 1.739478588104248,
      "step": 368
    },
    {
      "epoch": 3.509521484375e-07,
      "step": 368,
      "training_step_time": 9.54306674003601
    },
    {
      "epoch": 3.5190582275390624e-07,
      "model_forward_time": 2.25618577003479,
      "step": 369
    },
    {
      "epoch": 3.5190582275390624e-07,
      "step": 369,
      "training_step_time": 10.001971006393433
    },
    {
      "epoch": 3.528594970703125e-07,
      "grad_norm": 3.33528995513916,
      "learning_rate": 7.380000000000001e-07,
      "loss": 0.3505,
      "step": 370
    },
    {
      "epoch": 3.528594970703125e-07,
      "model_forward_time": 1.7203609943389893,
      "step": 370
    },
    {
      "dynamics_loss_avg": 0.08345930390059948,
      "epoch": 3.528594970703125e-07,
      "step": 370
    },
    {
      "action_loss_avg": 0.26328696608543395,
      "epoch": 3.528594970703125e-07,
      "step": 370
    },
    {
      "epoch": 3.528594970703125e-07,
      "step": 370,
      "training_step_time": 8.366231679916382
    },
    {
      "epoch": 3.5381317138671875e-07,
      "model_forward_time": 1.7161593437194824,
      "step": 371
    },
    {
      "epoch": 3.5381317138671875e-07,
      "step": 371,
      "training_step_time": 7.567530155181885
    },
    {
      "epoch": 3.54766845703125e-07,
      "model_forward_time": 1.7102141380310059,
      "step": 372
    },
    {
      "epoch": 3.54766845703125e-07,
      "step": 372,
      "training_step_time": 6.789988994598389
    },
    {
      "epoch": 3.5572052001953125e-07,
      "model_forward_time": 1.7236690521240234,
      "step": 373
    },
    {
      "epoch": 3.5572052001953125e-07,
      "step": 373,
      "training_step_time": 6.80803370475769
    },
    {
      "epoch": 3.566741943359375e-07,
      "model_forward_time": 1.7237262725830078,
      "step": 374
    },
    {
      "epoch": 3.566741943359375e-07,
      "step": 374,
      "training_step_time": 6.835459232330322
    },
    {
      "epoch": 3.5762786865234375e-07,
      "model_forward_time": 1.7219429016113281,
      "step": 375
    },
    {
      "epoch": 3.5762786865234375e-07,
      "step": 375,
      "training_step_time": 6.827622413635254
    },
    {
      "epoch": 3.5858154296875e-07,
      "model_forward_time": 1.7253541946411133,
      "step": 376
    },
    {
      "epoch": 3.5858154296875e-07,
      "step": 376,
      "training_step_time": 6.787758111953735
    },
    {
      "epoch": 3.5953521728515625e-07,
      "model_forward_time": 1.7254326343536377,
      "step": 377
    },
    {
      "epoch": 3.5953521728515625e-07,
      "step": 377,
      "training_step_time": 8.218990564346313
    },
    {
      "epoch": 3.604888916015625e-07,
      "model_forward_time": 2.249437093734741,
      "step": 378
    },
    {
      "epoch": 3.604888916015625e-07,
      "step": 378,
      "training_step_time": 9.548735857009888
    },
    {
      "epoch": 3.6144256591796875e-07,
      "model_forward_time": 1.7213149070739746,
      "step": 379
    },
    {
      "epoch": 3.6144256591796875e-07,
      "step": 379,
      "training_step_time": 7.80207633972168
    },
    {
      "epoch": 3.62396240234375e-07,
      "grad_norm": 2.7662789821624756,
      "learning_rate": 7.580000000000001e-07,
      "loss": 0.3664,
      "step": 380
    },
    {
      "epoch": 3.62396240234375e-07,
      "model_forward_time": 1.7253377437591553,
      "step": 380
    },
    {
      "dynamics_loss_avg": 0.09142470434308052,
      "epoch": 3.62396240234375e-07,
      "step": 380
    },
    {
      "action_loss_avg": 0.2615880861878395,
      "epoch": 3.62396240234375e-07,
      "step": 380
    },
    {
      "epoch": 3.62396240234375e-07,
      "step": 380,
      "training_step_time": 7.528285980224609
    },
    {
      "epoch": 3.6334991455078126e-07,
      "model_forward_time": 1.724783182144165,
      "step": 381
    },
    {
      "epoch": 3.6334991455078126e-07,
      "step": 381,
      "training_step_time": 6.8128814697265625
    },
    {
      "epoch": 3.643035888671875e-07,
      "model_forward_time": 1.722480297088623,
      "step": 382
    },
    {
      "epoch": 3.643035888671875e-07,
      "step": 382,
      "training_step_time": 6.812298059463501
    },
    {
      "epoch": 3.6525726318359376e-07,
      "model_forward_time": 1.7179617881774902,
      "step": 383
    },
    {
      "epoch": 3.6525726318359376e-07,
      "step": 383,
      "training_step_time": 6.8769917488098145
    },
    {
      "epoch": 3.662109375e-07,
      "model_forward_time": 1.7279956340789795,
      "step": 384
    },
    {
      "epoch": 3.662109375e-07,
      "step": 384,
      "training_step_time": 6.816347599029541
    },
    {
      "epoch": 3.6716461181640626e-07,
      "model_forward_time": 1.3476393222808838,
      "step": 385
    },
    {
      "epoch": 3.6716461181640626e-07,
      "step": 385,
      "training_step_time": 6.771726608276367
    },
    {
      "epoch": 3.681182861328125e-07,
      "model_forward_time": 1.7255966663360596,
      "step": 386
    },
    {
      "epoch": 3.681182861328125e-07,
      "step": 386,
      "training_step_time": 8.131870985031128
    },
    {
      "epoch": 3.6907196044921876e-07,
      "model_forward_time": 2.2675137519836426,
      "step": 387
    },
    {
      "epoch": 3.6907196044921876e-07,
      "step": 387,
      "training_step_time": 9.755908489227295
    },
    {
      "epoch": 3.70025634765625e-07,
      "model_forward_time": 1.7291481494903564,
      "step": 388
    },
    {
      "epoch": 3.70025634765625e-07,
      "step": 388,
      "training_step_time": 7.817093133926392
    },
    {
      "epoch": 3.7097930908203126e-07,
      "model_forward_time": 1.7247450351715088,
      "step": 389
    },
    {
      "epoch": 3.7097930908203126e-07,
      "step": 389,
      "training_step_time": 7.755523204803467
    },
    {
      "epoch": 3.719329833984375e-07,
      "grad_norm": 2.5819144248962402,
      "learning_rate": 7.78e-07,
      "loss": 0.3307,
      "step": 390
    },
    {
      "epoch": 3.719329833984375e-07,
      "model_forward_time": 1.7135183811187744,
      "step": 390
    },
    {
      "dynamics_loss_avg": 0.07676812019199133,
      "epoch": 3.719329833984375e-07,
      "step": 390
    },
    {
      "action_loss_avg": 0.2305485002696514,
      "epoch": 3.719329833984375e-07,
      "step": 390
    },
    {
      "epoch": 3.719329833984375e-07,
      "step": 390,
      "training_step_time": 7.534842014312744
    },
    {
      "epoch": 3.7288665771484376e-07,
      "model_forward_time": 1.7250046730041504,
      "step": 391
    },
    {
      "epoch": 3.7288665771484376e-07,
      "step": 391,
      "training_step_time": 6.778962135314941
    },
    {
      "epoch": 3.7384033203125e-07,
      "model_forward_time": 1.3515517711639404,
      "step": 392
    },
    {
      "epoch": 3.7384033203125e-07,
      "step": 392,
      "training_step_time": 6.769145727157593
    },
    {
      "epoch": 3.7479400634765627e-07,
      "model_forward_time": 1.725330114364624,
      "step": 393
    },
    {
      "epoch": 3.7479400634765627e-07,
      "step": 393,
      "training_step_time": 6.77265477180481
    },
    {
      "epoch": 3.757476806640625e-07,
      "model_forward_time": 1.726597547531128,
      "step": 394
    },
    {
      "epoch": 3.757476806640625e-07,
      "step": 394,
      "training_step_time": 6.793349504470825
    },
    {
      "epoch": 3.7670135498046877e-07,
      "model_forward_time": 1.7323107719421387,
      "step": 395
    },
    {
      "epoch": 3.7670135498046877e-07,
      "step": 395,
      "training_step_time": 7.293731927871704
    },
    {
      "epoch": 3.77655029296875e-07,
      "model_forward_time": 1.7586581707000732,
      "step": 396
    },
    {
      "epoch": 3.77655029296875e-07,
      "step": 396,
      "training_step_time": 7.661494493484497
    },
    {
      "epoch": 3.7860870361328127e-07,
      "model_forward_time": 2.2430481910705566,
      "step": 397
    },
    {
      "epoch": 3.7860870361328127e-07,
      "step": 397,
      "training_step_time": 9.595630168914795
    },
    {
      "epoch": 3.795623779296875e-07,
      "model_forward_time": 1.7162673473358154,
      "step": 398
    },
    {
      "epoch": 3.795623779296875e-07,
      "step": 398,
      "training_step_time": 7.745737075805664
    },
    {
      "epoch": 3.8051605224609377e-07,
      "model_forward_time": 1.7260193824768066,
      "step": 399
    },
    {
      "epoch": 3.8051605224609377e-07,
      "step": 399,
      "training_step_time": 7.504984378814697
    },
    {
      "epoch": 3.814697265625e-07,
      "grad_norm": 3.2008588314056396,
      "learning_rate": 7.98e-07,
      "loss": 0.3446,
      "step": 400
    },
    {
      "epoch": 3.814697265625e-07,
      "model_forward_time": 1.3498246669769287,
      "step": 400
    },
    {
      "dynamics_loss_avg": 0.08794236574321986,
      "epoch": 3.814697265625e-07,
      "step": 400
    },
    {
      "action_loss_avg": 0.2463346768170595,
      "epoch": 3.814697265625e-07,
      "step": 400
    },
    {
      "epoch": 3.814697265625e-07,
      "step": 400,
      "training_step_time": 6.754846096038818
    },
    {
      "epoch": 3.8242340087890627e-07,
      "model_forward_time": 1.7250168323516846,
      "step": 401
    },
    {
      "epoch": 3.8242340087890627e-07,
      "step": 401,
      "training_step_time": 6.793964385986328
    },
    {
      "epoch": 3.833770751953125e-07,
      "model_forward_time": 1.7329280376434326,
      "step": 402
    },
    {
      "epoch": 3.833770751953125e-07,
      "step": 402,
      "training_step_time": 6.821590900421143
    },
    {
      "epoch": 3.843307495117188e-07,
      "model_forward_time": 1.724034309387207,
      "step": 403
    },
    {
      "epoch": 3.843307495117188e-07,
      "step": 403,
      "training_step_time": 6.798497438430786
    },
    {
      "epoch": 3.85284423828125e-07,
      "model_forward_time": 1.7273316383361816,
      "step": 404
    },
    {
      "epoch": 3.85284423828125e-07,
      "step": 404,
      "training_step_time": 6.767481803894043
    },
    {
      "epoch": 3.862380981445313e-07,
      "model_forward_time": 1.7461986541748047,
      "step": 405
    },
    {
      "epoch": 3.862380981445313e-07,
      "step": 405,
      "training_step_time": 8.744632005691528
    },
    {
      "epoch": 3.871917724609375e-07,
      "model_forward_time": 2.2505996227264404,
      "step": 406
    },
    {
      "epoch": 3.871917724609375e-07,
      "step": 406,
      "training_step_time": 10.151307582855225
    },
    {
      "epoch": 3.881454467773437e-07,
      "model_forward_time": 1.7210063934326172,
      "step": 407
    },
    {
      "epoch": 3.881454467773437e-07,
      "step": 407,
      "training_step_time": 7.84825873374939
    },
    {
      "epoch": 3.8909912109375e-07,
      "model_forward_time": 1.718977689743042,
      "step": 408
    },
    {
      "epoch": 3.8909912109375e-07,
      "step": 408,
      "training_step_time": 7.5424230098724365
    },
    {
      "epoch": 3.900527954101562e-07,
      "model_forward_time": 1.7162692546844482,
      "step": 409
    },
    {
      "epoch": 3.900527954101562e-07,
      "step": 409,
      "training_step_time": 6.733654975891113
    },
    {
      "epoch": 3.910064697265625e-07,
      "grad_norm": 2.6271350383758545,
      "learning_rate": 8.18e-07,
      "loss": 0.3482,
      "step": 410
    },
    {
      "epoch": 3.910064697265625e-07,
      "model_forward_time": 1.7155427932739258,
      "step": 410
    },
    {
      "dynamics_loss_avg": 0.08541057985275984,
      "epoch": 3.910064697265625e-07,
      "step": 410
    },
    {
      "action_loss_avg": 0.25225976631045344,
      "epoch": 3.910064697265625e-07,
      "step": 410
    },
    {
      "epoch": 3.910064697265625e-07,
      "step": 410,
      "training_step_time": 6.764976978302002
    },
    {
      "epoch": 3.9196014404296873e-07,
      "model_forward_time": 1.7216811180114746,
      "step": 411
    },
    {
      "epoch": 3.9196014404296873e-07,
      "step": 411,
      "training_step_time": 6.805744171142578
    },
    {
      "epoch": 3.92913818359375e-07,
      "model_forward_time": 1.7149994373321533,
      "step": 412
    },
    {
      "epoch": 3.92913818359375e-07,
      "step": 412,
      "training_step_time": 6.736318349838257
    },
    {
      "epoch": 3.9386749267578123e-07,
      "model_forward_time": 1.7180306911468506,
      "step": 413
    },
    {
      "epoch": 3.9386749267578123e-07,
      "step": 413,
      "training_step_time": 6.748216867446899
    },
    {
      "epoch": 3.948211669921875e-07,
      "model_forward_time": 1.7518248558044434,
      "step": 414
    },
    {
      "epoch": 3.948211669921875e-07,
      "step": 414,
      "training_step_time": 7.769893407821655
    },
    {
      "epoch": 3.9577484130859373e-07,
      "model_forward_time": 2.2619478702545166,
      "step": 415
    },
    {
      "epoch": 3.9577484130859373e-07,
      "step": 415,
      "training_step_time": 10.353297710418701
    },
    {
      "epoch": 3.96728515625e-07,
      "model_forward_time": 1.7053554058074951,
      "step": 416
    },
    {
      "epoch": 3.96728515625e-07,
      "step": 416,
      "training_step_time": 8.729155778884888
    },
    {
      "epoch": 3.9768218994140623e-07,
      "model_forward_time": 1.719297170639038,
      "step": 417
    },
    {
      "epoch": 3.9768218994140623e-07,
      "step": 417,
      "training_step_time": 7.688624382019043
    },
    {
      "epoch": 3.986358642578125e-07,
      "model_forward_time": 1.709101676940918,
      "step": 418
    },
    {
      "epoch": 3.986358642578125e-07,
      "step": 418,
      "training_step_time": 6.792285203933716
    },
    {
      "epoch": 3.9958953857421873e-07,
      "model_forward_time": 1.6908915042877197,
      "step": 419
    },
    {
      "epoch": 3.9958953857421873e-07,
      "step": 419,
      "training_step_time": 6.8225579261779785
    },
    {
      "epoch": 4.00543212890625e-07,
      "grad_norm": 3.3027184009552,
      "learning_rate": 8.380000000000001e-07,
      "loss": 0.319,
      "step": 420
    },
    {
      "epoch": 4.00543212890625e-07,
      "model_forward_time": 1.7007896900177002,
      "step": 420
    },
    {
      "dynamics_loss_avg": 0.08047780767083168,
      "epoch": 4.00543212890625e-07,
      "step": 420
    },
    {
      "action_loss_avg": 0.19904664382338524,
      "epoch": 4.00543212890625e-07,
      "step": 420
    },
    {
      "epoch": 4.00543212890625e-07,
      "step": 420,
      "training_step_time": 6.776312589645386
    },
    {
      "epoch": 4.0149688720703124e-07,
      "model_forward_time": 1.702728271484375,
      "step": 421
    },
    {
      "epoch": 4.0149688720703124e-07,
      "step": 421,
      "training_step_time": 6.7358880043029785
    },
    {
      "epoch": 4.024505615234375e-07,
      "model_forward_time": 1.6998059749603271,
      "step": 422
    },
    {
      "epoch": 4.024505615234375e-07,
      "step": 422,
      "training_step_time": 6.737339019775391
    },
    {
      "epoch": 4.0340423583984374e-07,
      "model_forward_time": 1.73978853225708,
      "step": 423
    },
    {
      "epoch": 4.0340423583984374e-07,
      "step": 423,
      "training_step_time": 8.30200982093811
    },
    {
      "epoch": 4.0435791015625e-07,
      "model_forward_time": 1.6834731101989746,
      "step": 424
    },
    {
      "epoch": 4.0435791015625e-07,
      "step": 424,
      "training_step_time": 11.911878824234009
    },
    {
      "epoch": 4.0531158447265624e-07,
      "model_forward_time": 1.7006769180297852,
      "step": 425
    },
    {
      "epoch": 4.0531158447265624e-07,
      "step": 425,
      "training_step_time": 9.721922159194946
    },
    {
      "epoch": 4.062652587890625e-07,
      "model_forward_time": 1.707493782043457,
      "step": 426
    },
    {
      "epoch": 4.062652587890625e-07,
      "step": 426,
      "training_step_time": 7.722834587097168
    },
    {
      "epoch": 4.0721893310546874e-07,
      "model_forward_time": 1.7130355834960938,
      "step": 427
    },
    {
      "epoch": 4.0721893310546874e-07,
      "step": 427,
      "training_step_time": 7.173260927200317
    },
    {
      "epoch": 4.08172607421875e-07,
      "model_forward_time": 1.7037031650543213,
      "step": 428
    },
    {
      "epoch": 4.08172607421875e-07,
      "step": 428,
      "training_step_time": 6.823720693588257
    },
    {
      "epoch": 4.0912628173828124e-07,
      "model_forward_time": 1.6928534507751465,
      "step": 429
    },
    {
      "epoch": 4.0912628173828124e-07,
      "step": 429,
      "training_step_time": 6.822588920593262
    },
    {
      "epoch": 4.100799560546875e-07,
      "grad_norm": 2.0963151454925537,
      "learning_rate": 8.580000000000001e-07,
      "loss": 0.2925,
      "step": 430
    },
    {
      "epoch": 4.100799560546875e-07,
      "model_forward_time": 1.7046778202056885,
      "step": 430
    },
    {
      "dynamics_loss_avg": 0.0836631640791893,
      "epoch": 4.100799560546875e-07,
      "step": 430
    },
    {
      "action_loss_avg": 0.20418504551053046,
      "epoch": 4.100799560546875e-07,
      "step": 430
    },
    {
      "epoch": 4.100799560546875e-07,
      "step": 430,
      "training_step_time": 6.792514801025391
    },
    {
      "epoch": 4.1103363037109374e-07,
      "model_forward_time": 1.6905784606933594,
      "step": 431
    },
    {
      "epoch": 4.1103363037109374e-07,
      "step": 431,
      "training_step_time": 6.837872743606567
    },
    {
      "epoch": 4.119873046875e-07,
      "model_forward_time": 1.7088549137115479,
      "step": 432
    },
    {
      "epoch": 4.119873046875e-07,
      "step": 432,
      "training_step_time": 7.746506690979004
    },
    {
      "epoch": 4.1294097900390625e-07,
      "model_forward_time": 1.7025361061096191,
      "step": 433
    },
    {
      "epoch": 4.1294097900390625e-07,
      "step": 433,
      "training_step_time": 9.792988538742065
    },
    {
      "epoch": 4.138946533203125e-07,
      "model_forward_time": 2.2440927028656006,
      "step": 434
    },
    {
      "epoch": 4.138946533203125e-07,
      "step": 434,
      "training_step_time": 8.561196327209473
    },
    {
      "epoch": 4.1484832763671875e-07,
      "model_forward_time": 1.7100272178649902,
      "step": 435
    },
    {
      "epoch": 4.1484832763671875e-07,
      "step": 435,
      "training_step_time": 7.72945499420166
    },
    {
      "epoch": 4.15802001953125e-07,
      "model_forward_time": 1.704742193222046,
      "step": 436
    },
    {
      "epoch": 4.15802001953125e-07,
      "step": 436,
      "training_step_time": 7.558996200561523
    },
    {
      "epoch": 4.1675567626953125e-07,
      "model_forward_time": 1.7039964199066162,
      "step": 437
    },
    {
      "epoch": 4.1675567626953125e-07,
      "step": 437,
      "training_step_time": 6.864949941635132
    },
    {
      "epoch": 4.177093505859375e-07,
      "model_forward_time": 1.7097291946411133,
      "step": 438
    },
    {
      "epoch": 4.177093505859375e-07,
      "step": 438,
      "training_step_time": 6.860205888748169
    },
    {
      "epoch": 4.1866302490234375e-07,
      "model_forward_time": 1.714893102645874,
      "step": 439
    },
    {
      "epoch": 4.1866302490234375e-07,
      "step": 439,
      "training_step_time": 6.854872226715088
    },
    {
      "epoch": 4.1961669921875e-07,
      "grad_norm": 2.562819004058838,
      "learning_rate": 8.780000000000001e-07,
      "loss": 0.303,
      "step": 440
    },
    {
      "epoch": 4.1961669921875e-07,
      "model_forward_time": 1.7073578834533691,
      "step": 440
    },
    {
      "dynamics_loss_avg": 0.07163209049031138,
      "epoch": 4.1961669921875e-07,
      "step": 440
    },
    {
      "action_loss_avg": 0.19895894527435304,
      "epoch": 4.1961669921875e-07,
      "step": 440
    },
    {
      "epoch": 4.1961669921875e-07,
      "step": 440,
      "training_step_time": 6.803474426269531
    },
    {
      "epoch": 4.2057037353515625e-07,
      "model_forward_time": 1.7149262428283691,
      "step": 441
    },
    {
      "epoch": 4.2057037353515625e-07,
      "step": 441,
      "training_step_time": 7.742277383804321
    },
    {
      "epoch": 4.215240478515625e-07,
      "model_forward_time": 1.713477611541748,
      "step": 442
    },
    {
      "epoch": 4.215240478515625e-07,
      "step": 442,
      "training_step_time": 8.38550591468811
    },
    {
      "epoch": 4.2247772216796875e-07,
      "model_forward_time": 2.2240095138549805,
      "step": 443
    },
    {
      "epoch": 4.2247772216796875e-07,
      "step": 443,
      "training_step_time": 8.281894445419312
    },
    {
      "epoch": 4.23431396484375e-07,
      "model_forward_time": 0.9722244739532471,
      "step": 444
    },
    {
      "epoch": 4.23431396484375e-07,
      "step": 444,
      "training_step_time": 7.742002010345459
    },
    {
      "epoch": 4.2438507080078126e-07,
      "model_forward_time": 1.7009046077728271,
      "step": 445
    },
    {
      "epoch": 4.2438507080078126e-07,
      "step": 445,
      "training_step_time": 7.189014196395874
    },
    {
      "epoch": 4.253387451171875e-07,
      "model_forward_time": 1.6923151016235352,
      "step": 446
    },
    {
      "epoch": 4.253387451171875e-07,
      "step": 446,
      "training_step_time": 6.881716251373291
    },
    {
      "epoch": 4.2629241943359376e-07,
      "model_forward_time": 1.7114489078521729,
      "step": 447
    },
    {
      "epoch": 4.2629241943359376e-07,
      "step": 447,
      "training_step_time": 6.784024715423584
    },
    {
      "epoch": 4.2724609375e-07,
      "model_forward_time": 1.707763910293579,
      "step": 448
    },
    {
      "epoch": 4.2724609375e-07,
      "step": 448,
      "training_step_time": 6.782080173492432
    },
    {
      "epoch": 4.2819976806640626e-07,
      "model_forward_time": 1.7111499309539795,
      "step": 449
    },
    {
      "epoch": 4.2819976806640626e-07,
      "step": 449,
      "training_step_time": 6.7845048904418945
    },
    {
      "epoch": 4.291534423828125e-07,
      "grad_norm": 3.0788378715515137,
      "learning_rate": 8.980000000000001e-07,
      "loss": 0.3216,
      "step": 450
    },
    {
      "epoch": 4.291534423828125e-07,
      "model_forward_time": 1.708988904953003,
      "step": 450
    },
    {
      "dynamics_loss_avg": 0.10699505619704723,
      "epoch": 4.291534423828125e-07,
      "step": 450
    },
    {
      "action_loss_avg": 0.2634680584073067,
      "epoch": 4.291534423828125e-07,
      "step": 450
    },
    {
      "epoch": 4.291534423828125e-07,
      "step": 450,
      "training_step_time": 8.499824523925781
    },
    {
      "epoch": 4.3010711669921876e-07,
      "model_forward_time": 1.7175238132476807,
      "step": 451
    },
    {
      "epoch": 4.3010711669921876e-07,
      "step": 451,
      "training_step_time": 8.694570541381836
    },
    {
      "epoch": 4.31060791015625e-07,
      "model_forward_time": 0.9890930652618408,
      "step": 452
    },
    {
      "epoch": 4.31060791015625e-07,
      "step": 452,
      "training_step_time": 8.598142862319946
    },
    {
      "epoch": 4.3201446533203126e-07,
      "model_forward_time": 2.2350337505340576,
      "step": 453
    },
    {
      "epoch": 4.3201446533203126e-07,
      "step": 453,
      "training_step_time": 8.702686786651611
    },
    {
      "epoch": 4.329681396484375e-07,
      "model_forward_time": 1.7092201709747314,
      "step": 454
    },
    {
      "epoch": 4.329681396484375e-07,
      "step": 454,
      "training_step_time": 7.616062641143799
    },
    {
      "epoch": 4.3392181396484376e-07,
      "model_forward_time": 1.7030208110809326,
      "step": 455
    },
    {
      "epoch": 4.3392181396484376e-07,
      "step": 455,
      "training_step_time": 6.817190885543823
    },
    {
      "epoch": 4.3487548828125e-07,
      "model_forward_time": 1.710866928100586,
      "step": 456
    },
    {
      "epoch": 4.3487548828125e-07,
      "step": 456,
      "training_step_time": 6.813313961029053
    },
    {
      "epoch": 4.3582916259765627e-07,
      "model_forward_time": 1.7072598934173584,
      "step": 457
    },
    {
      "epoch": 4.3582916259765627e-07,
      "step": 457,
      "training_step_time": 6.837791204452515
    },
    {
      "epoch": 4.367828369140625e-07,
      "model_forward_time": 1.7134947776794434,
      "step": 458
    },
    {
      "epoch": 4.367828369140625e-07,
      "step": 458,
      "training_step_time": 6.820586442947388
    },
    {
      "epoch": 4.3773651123046877e-07,
      "model_forward_time": 1.723146677017212,
      "step": 459
    },
    {
      "epoch": 4.3773651123046877e-07,
      "step": 459,
      "training_step_time": 7.679547309875488
    },
    {
      "epoch": 4.38690185546875e-07,
      "grad_norm": 2.5981054306030273,
      "learning_rate": 9.180000000000001e-07,
      "loss": 0.3323,
      "step": 460
    },
    {
      "epoch": 4.38690185546875e-07,
      "model_forward_time": 1.702878475189209,
      "step": 460
    },
    {
      "dynamics_loss_avg": 0.09555726051330567,
      "epoch": 4.38690185546875e-07,
      "step": 460
    },
    {
      "action_loss_avg": 0.21719306707382202,
      "epoch": 4.38690185546875e-07,
      "step": 460
    },
    {
      "epoch": 4.38690185546875e-07,
      "step": 460,
      "training_step_time": 7.805405616760254
    },
    {
      "epoch": 4.3964385986328127e-07,
      "model_forward_time": 1.7063205242156982,
      "step": 461
    },
    {
      "epoch": 4.3964385986328127e-07,
      "step": 461,
      "training_step_time": 9.523587703704834
    },
    {
      "epoch": 4.405975341796875e-07,
      "model_forward_time": 1.8490946292877197,
      "step": 462
    },
    {
      "epoch": 4.405975341796875e-07,
      "step": 462,
      "training_step_time": 8.393055200576782
    },
    {
      "epoch": 4.4155120849609377e-07,
      "model_forward_time": 1.701735496520996,
      "step": 463
    },
    {
      "epoch": 4.4155120849609377e-07,
      "step": 463,
      "training_step_time": 8.77202033996582
    },
    {
      "epoch": 4.425048828125e-07,
      "model_forward_time": 1.3377902507781982,
      "step": 464
    },
    {
      "epoch": 4.425048828125e-07,
      "step": 464,
      "training_step_time": 6.76562762260437
    },
    {
      "epoch": 4.4345855712890627e-07,
      "model_forward_time": 1.7095820903778076,
      "step": 465
    },
    {
      "epoch": 4.4345855712890627e-07,
      "step": 465,
      "training_step_time": 6.821110010147095
    },
    {
      "epoch": 4.444122314453125e-07,
      "model_forward_time": 1.7103509902954102,
      "step": 466
    },
    {
      "epoch": 4.444122314453125e-07,
      "step": 466,
      "training_step_time": 6.8627424240112305
    },
    {
      "epoch": 4.453659057617188e-07,
      "model_forward_time": 1.7083868980407715,
      "step": 467
    },
    {
      "epoch": 4.453659057617188e-07,
      "step": 467,
      "training_step_time": 6.790939569473267
    },
    {
      "epoch": 4.46319580078125e-07,
      "model_forward_time": 1.7286458015441895,
      "step": 468
    },
    {
      "epoch": 4.46319580078125e-07,
      "step": 468,
      "training_step_time": 7.8456690311431885
    },
    {
      "epoch": 4.472732543945312e-07,
      "model_forward_time": 1.7206206321716309,
      "step": 469
    },
    {
      "epoch": 4.472732543945312e-07,
      "step": 469,
      "training_step_time": 9.31859564781189
    },
    {
      "epoch": 4.482269287109375e-07,
      "grad_norm": 1.5847197771072388,
      "learning_rate": 9.380000000000001e-07,
      "loss": 0.2757,
      "step": 470
    },
    {
      "epoch": 4.482269287109375e-07,
      "model_forward_time": 1.6981759071350098,
      "step": 470
    },
    {
      "dynamics_loss_avg": 0.08257863521575928,
      "epoch": 4.482269287109375e-07,
      "step": 470
    },
    {
      "action_loss_avg": 0.2251536011695862,
      "epoch": 4.482269287109375e-07,
      "step": 470
    },
    {
      "epoch": 4.482269287109375e-07,
      "step": 470,
      "training_step_time": 7.76340126991272
    },
    {
      "epoch": 4.491806030273437e-07,
      "model_forward_time": 2.25361967086792,
      "step": 471
    },
    {
      "epoch": 4.491806030273437e-07,
      "step": 471,
      "training_step_time": 7.744183778762817
    },
    {
      "epoch": 4.5013427734375e-07,
      "model_forward_time": 1.7077648639678955,
      "step": 472
    },
    {
      "epoch": 4.5013427734375e-07,
      "step": 472,
      "training_step_time": 7.637266635894775
    },
    {
      "epoch": 4.5108795166015623e-07,
      "model_forward_time": 1.7054908275604248,
      "step": 473
    },
    {
      "epoch": 4.5108795166015623e-07,
      "step": 473,
      "training_step_time": 7.484712362289429
    },
    {
      "epoch": 4.520416259765625e-07,
      "model_forward_time": 1.711993932723999,
      "step": 474
    },
    {
      "epoch": 4.520416259765625e-07,
      "step": 474,
      "training_step_time": 6.725938558578491
    },
    {
      "epoch": 4.5299530029296873e-07,
      "model_forward_time": 0.9903440475463867,
      "step": 475
    },
    {
      "epoch": 4.5299530029296873e-07,
      "step": 475,
      "training_step_time": 6.737310886383057
    },
    {
      "epoch": 4.53948974609375e-07,
      "model_forward_time": 1.7119083404541016,
      "step": 476
    },
    {
      "epoch": 4.53948974609375e-07,
      "step": 476,
      "training_step_time": 6.759819507598877
    },
    {
      "epoch": 4.5490264892578123e-07,
      "model_forward_time": 1.7092700004577637,
      "step": 477
    },
    {
      "epoch": 4.5490264892578123e-07,
      "step": 477,
      "training_step_time": 8.216017723083496
    },
    {
      "epoch": 4.558563232421875e-07,
      "model_forward_time": 1.7086386680603027,
      "step": 478
    },
    {
      "epoch": 4.558563232421875e-07,
      "step": 478,
      "training_step_time": 7.6721203327178955
    },
    {
      "epoch": 4.5680999755859373e-07,
      "model_forward_time": 1.7041096687316895,
      "step": 479
    },
    {
      "epoch": 4.5680999755859373e-07,
      "step": 479,
      "training_step_time": 9.719369411468506
    },
    {
      "epoch": 4.57763671875e-07,
      "grad_norm": 2.5469539165496826,
      "learning_rate": 9.58e-07,
      "loss": 0.3019,
      "step": 480
    },
    {
      "epoch": 4.57763671875e-07,
      "model_forward_time": 1.710763931274414,
      "step": 480
    },
    {
      "dynamics_loss_avg": 0.08362894486635923,
      "epoch": 4.57763671875e-07,
      "step": 480
    },
    {
      "action_loss_avg": 0.22118504717946053,
      "epoch": 4.57763671875e-07,
      "step": 480
    },
    {
      "epoch": 4.57763671875e-07,
      "step": 480,
      "training_step_time": 8.557679653167725
    },
    {
      "epoch": 4.5871734619140623e-07,
      "model_forward_time": 2.2475621700286865,
      "step": 481
    },
    {
      "epoch": 4.5871734619140623e-07,
      "step": 481,
      "training_step_time": 7.557703733444214
    },
    {
      "epoch": 4.596710205078125e-07,
      "model_forward_time": 1.7162485122680664,
      "step": 482
    },
    {
      "epoch": 4.596710205078125e-07,
      "step": 482,
      "training_step_time": 7.509838342666626
    },
    {
      "epoch": 4.6062469482421874e-07,
      "model_forward_time": 1.713433027267456,
      "step": 483
    },
    {
      "epoch": 4.6062469482421874e-07,
      "step": 483,
      "training_step_time": 6.7178168296813965
    },
    {
      "epoch": 4.61578369140625e-07,
      "model_forward_time": 1.7132995128631592,
      "step": 484
    },
    {
      "epoch": 4.61578369140625e-07,
      "step": 484,
      "training_step_time": 6.711557626724243
    },
    {
      "epoch": 4.6253204345703124e-07,
      "model_forward_time": 1.7114131450653076,
      "step": 485
    },
    {
      "epoch": 4.6253204345703124e-07,
      "step": 485,
      "training_step_time": 6.694762229919434
    },
    {
      "epoch": 4.634857177734375e-07,
      "model_forward_time": 1.7066431045532227,
      "step": 486
    },
    {
      "epoch": 4.634857177734375e-07,
      "step": 486,
      "training_step_time": 8.122011184692383
    },
    {
      "epoch": 4.6443939208984374e-07,
      "model_forward_time": 1.709500789642334,
      "step": 487
    },
    {
      "epoch": 4.6443939208984374e-07,
      "step": 487,
      "training_step_time": 8.265133142471313
    },
    {
      "epoch": 4.6539306640625e-07,
      "model_forward_time": 1.7069652080535889,
      "step": 488
    },
    {
      "epoch": 4.6539306640625e-07,
      "step": 488,
      "training_step_time": 8.316223859786987
    },
    {
      "epoch": 4.6634674072265624e-07,
      "model_forward_time": 1.7024426460266113,
      "step": 489
    },
    {
      "epoch": 4.6634674072265624e-07,
      "step": 489,
      "training_step_time": 9.99090576171875
    },
    {
      "epoch": 4.673004150390625e-07,
      "grad_norm": 2.1974408626556396,
      "learning_rate": 9.78e-07,
      "loss": 0.3113,
      "step": 490
    },
    {
      "epoch": 4.673004150390625e-07,
      "model_forward_time": 2.2473971843719482,
      "step": 490
    },
    {
      "dynamics_loss_avg": 0.09583833813667297,
      "epoch": 4.673004150390625e-07,
      "step": 490
    },
    {
      "action_loss_avg": 0.2225711762905121,
      "epoch": 4.673004150390625e-07,
      "step": 490
    },
    {
      "epoch": 4.673004150390625e-07,
      "step": 490,
      "training_step_time": 8.650310039520264
    },
    {
      "epoch": 4.6825408935546874e-07,
      "model_forward_time": 1.7158653736114502,
      "step": 491
    },
    {
      "epoch": 4.6825408935546874e-07,
      "step": 491,
      "training_step_time": 7.554400444030762
    },
    {
      "epoch": 4.69207763671875e-07,
      "model_forward_time": 1.714202880859375,
      "step": 492
    },
    {
      "epoch": 4.69207763671875e-07,
      "step": 492,
      "training_step_time": 6.72940468788147
    },
    {
      "epoch": 4.7016143798828124e-07,
      "model_forward_time": 1.7064361572265625,
      "step": 493
    },
    {
      "epoch": 4.7016143798828124e-07,
      "step": 493,
      "training_step_time": 6.69468355178833
    },
    {
      "epoch": 4.711151123046875e-07,
      "model_forward_time": 1.7071070671081543,
      "step": 494
    },
    {
      "epoch": 4.711151123046875e-07,
      "step": 494,
      "training_step_time": 6.707387685775757
    },
    {
      "epoch": 4.7206878662109375e-07,
      "model_forward_time": 1.6845297813415527,
      "step": 495
    },
    {
      "epoch": 4.7206878662109375e-07,
      "step": 495,
      "training_step_time": 7.606265544891357
    },
    {
      "epoch": 4.730224609375e-07,
      "model_forward_time": 1.7065534591674805,
      "step": 496
    },
    {
      "epoch": 4.730224609375e-07,
      "step": 496,
      "training_step_time": 7.234346389770508
    },
    {
      "epoch": 4.7397613525390625e-07,
      "model_forward_time": 1.7071399688720703,
      "step": 497
    },
    {
      "epoch": 4.7397613525390625e-07,
      "step": 497,
      "training_step_time": 9.553518295288086
    },
    {
      "epoch": 4.749298095703125e-07,
      "model_forward_time": 1.698152780532837,
      "step": 498
    },
    {
      "epoch": 4.749298095703125e-07,
      "step": 498,
      "training_step_time": 10.400643825531006
    },
    {
      "epoch": 4.7588348388671875e-07,
      "model_forward_time": 1.8398280143737793,
      "step": 499
    },
    {
      "epoch": 4.7588348388671875e-07,
      "step": 499,
      "training_step_time": 7.596942663192749
    },
    {
      "epoch": 4.76837158203125e-07,
      "grad_norm": 3.3228225708007812,
      "learning_rate": 9.98e-07,
      "loss": 0.2658,
      "step": 500
    },
    {
      "epoch": 4.76837158203125e-07,
      "model_forward_time": 1.6963109970092773,
      "step": 500
    },
    {
      "dynamics_loss_avg": 0.0740045353770256,
      "epoch": 4.76837158203125e-07,
      "step": 500
    },
    {
      "action_loss_avg": 0.21135902255773545,
      "epoch": 4.76837158203125e-07,
      "step": 500
    },
    {
      "epoch": 4.76837158203125e-07,
      "step": 500,
      "training_step_time": 7.744238376617432
    },
    {
      "epoch": 4.777908325195313e-07,
      "model_forward_time": 1.7047696113586426,
      "step": 501
    },
    {
      "epoch": 4.777908325195313e-07,
      "step": 501,
      "training_step_time": 7.017990827560425
    },
    {
      "epoch": 4.787445068359375e-07,
      "model_forward_time": 1.691932201385498,
      "step": 502
    },
    {
      "epoch": 4.787445068359375e-07,
      "step": 502,
      "training_step_time": 6.791388034820557
    },
    {
      "epoch": 4.796981811523438e-07,
      "model_forward_time": 1.7042250633239746,
      "step": 503
    },
    {
      "epoch": 4.796981811523438e-07,
      "step": 503,
      "training_step_time": 6.7656614780426025
    },
    {
      "epoch": 4.8065185546875e-07,
      "model_forward_time": 1.6968159675598145,
      "step": 504
    },
    {
      "epoch": 4.8065185546875e-07,
      "step": 504,
      "training_step_time": 8.213029384613037
    },
    {
      "epoch": 4.816055297851563e-07,
      "model_forward_time": 1.7070116996765137,
      "step": 505
    },
    {
      "epoch": 4.816055297851563e-07,
      "step": 505,
      "training_step_time": 8.23396921157837
    },
    {
      "epoch": 4.825592041015625e-07,
      "model_forward_time": 1.7095680236816406,
      "step": 506
    },
    {
      "epoch": 4.825592041015625e-07,
      "step": 506,
      "training_step_time": 8.53795075416565
    },
    {
      "epoch": 4.835128784179688e-07,
      "model_forward_time": 1.7150144577026367,
      "step": 507
    },
    {
      "epoch": 4.835128784179688e-07,
      "step": 507,
      "training_step_time": 9.334635496139526
    },
    {
      "epoch": 4.84466552734375e-07,
      "model_forward_time": 2.214301586151123,
      "step": 508
    },
    {
      "epoch": 4.84466552734375e-07,
      "step": 508,
      "training_step_time": 7.621165037155151
    },
    {
      "epoch": 4.854202270507813e-07,
      "model_forward_time": 1.7159686088562012,
      "step": 509
    },
    {
      "epoch": 4.854202270507813e-07,
      "step": 509,
      "training_step_time": 7.625494718551636
    },
    {
      "epoch": 4.863739013671875e-07,
      "grad_norm": 4.480686664581299,
      "learning_rate": 1.018e-06,
      "loss": 0.2445,
      "step": 510
    },
    {
      "epoch": 4.863739013671875e-07,
      "model_forward_time": 1.7250981330871582,
      "step": 510
    },
    {
      "dynamics_loss_avg": 0.0761227797716856,
      "epoch": 4.863739013671875e-07,
      "step": 510
    },
    {
      "action_loss_avg": 0.14940378852188588,
      "epoch": 4.863739013671875e-07,
      "step": 510
    },
    {
      "epoch": 4.863739013671875e-07,
      "step": 510,
      "training_step_time": 7.477079153060913
    },
    {
      "epoch": 4.873275756835938e-07,
      "model_forward_time": 1.71626615524292,
      "step": 511
    },
    {
      "epoch": 4.873275756835938e-07,
      "step": 511,
      "training_step_time": 6.757155179977417
    },
    {
      "epoch": 4.8828125e-07,
      "model_forward_time": 1.7207238674163818,
      "step": 512
    },
    {
      "epoch": 4.8828125e-07,
      "step": 512,
      "training_step_time": 6.725782632827759
    },
    {
      "epoch": 4.892349243164063e-07,
      "model_forward_time": 1.7173266410827637,
      "step": 513
    },
    {
      "epoch": 4.892349243164063e-07,
      "step": 513,
      "training_step_time": 7.611977815628052
    },
    {
      "epoch": 4.901885986328125e-07,
      "model_forward_time": 1.7569849491119385,
      "step": 514
    },
    {
      "epoch": 4.901885986328125e-07,
      "step": 514,
      "training_step_time": 7.719993591308594
    },
    {
      "epoch": 4.911422729492188e-07,
      "model_forward_time": 1.7140908241271973,
      "step": 515
    },
    {
      "epoch": 4.911422729492188e-07,
      "step": 515,
      "training_step_time": 8.814945697784424
    },
    {
      "epoch": 4.92095947265625e-07,
      "model_forward_time": 1.7217226028442383,
      "step": 516
    },
    {
      "epoch": 4.92095947265625e-07,
      "step": 516,
      "training_step_time": 10.3101487159729
    },
    {
      "epoch": 4.930496215820313e-07,
      "model_forward_time": 1.7178843021392822,
      "step": 517
    },
    {
      "epoch": 4.930496215820313e-07,
      "step": 517,
      "training_step_time": 9.607831239700317
    },
    {
      "epoch": 4.940032958984375e-07,
      "model_forward_time": 2.264224052429199,
      "step": 518
    },
    {
      "epoch": 4.940032958984375e-07,
      "step": 518,
      "training_step_time": 7.578117847442627
    },
    {
      "epoch": 4.949569702148438e-07,
      "model_forward_time": 1.7202491760253906,
      "step": 519
    },
    {
      "epoch": 4.949569702148438e-07,
      "step": 519,
      "training_step_time": 7.656305313110352
    },
    {
      "epoch": 4.9591064453125e-07,
      "grad_norm": 5.200476169586182,
      "learning_rate": 1.038e-06,
      "loss": 0.2271,
      "step": 520
    },
    {
      "epoch": 4.9591064453125e-07,
      "model_forward_time": 1.7147481441497803,
      "step": 520
    },
    {
      "dynamics_loss_avg": 0.0880697388201952,
      "epoch": 4.9591064453125e-07,
      "step": 520
    },
    {
      "action_loss_avg": 0.15409714840352534,
      "epoch": 4.9591064453125e-07,
      "step": 520
    },
    {
      "epoch": 4.9591064453125e-07,
      "step": 520,
      "training_step_time": 6.7641870975494385
    },
    {
      "epoch": 4.968643188476563e-07,
      "model_forward_time": 1.7161650657653809,
      "step": 521
    },
    {
      "epoch": 4.968643188476563e-07,
      "step": 521,
      "training_step_time": 6.764609336853027
    },
    {
      "epoch": 4.978179931640625e-07,
      "model_forward_time": 1.7020313739776611,
      "step": 522
    },
    {
      "epoch": 4.978179931640625e-07,
      "step": 522,
      "training_step_time": 7.638510227203369
    },
    {
      "epoch": 4.987716674804688e-07,
      "model_forward_time": 1.7536022663116455,
      "step": 523
    },
    {
      "epoch": 4.987716674804688e-07,
      "step": 523,
      "training_step_time": 7.702029228210449
    },
    {
      "epoch": 4.99725341796875e-07,
      "model_forward_time": 1.6999239921569824,
      "step": 524
    },
    {
      "epoch": 4.99725341796875e-07,
      "step": 524,
      "training_step_time": 8.727388143539429
    },
    {
      "epoch": 5.006790161132813e-07,
      "model_forward_time": 1.7182271480560303,
      "step": 525
    },
    {
      "epoch": 5.006790161132813e-07,
      "step": 525,
      "training_step_time": 8.772384405136108
    },
    {
      "epoch": 5.016326904296875e-07,
      "model_forward_time": 1.7330899238586426,
      "step": 526
    },
    {
      "epoch": 5.016326904296875e-07,
      "step": 526,
      "training_step_time": 9.428514003753662
    },
    {
      "epoch": 5.025863647460938e-07,
      "model_forward_time": 2.2358813285827637,
      "step": 527
    },
    {
      "epoch": 5.025863647460938e-07,
      "step": 527,
      "training_step_time": 7.679673194885254
    },
    {
      "epoch": 5.035400390625e-07,
      "model_forward_time": 1.6896352767944336,
      "step": 528
    },
    {
      "epoch": 5.035400390625e-07,
      "step": 528,
      "training_step_time": 7.52596640586853
    },
    {
      "epoch": 5.044937133789063e-07,
      "model_forward_time": 1.7140781879425049,
      "step": 529
    },
    {
      "epoch": 5.044937133789063e-07,
      "step": 529,
      "training_step_time": 6.704188346862793
    },
    {
      "epoch": 5.054473876953125e-07,
      "grad_norm": 3.184798240661621,
      "learning_rate": 1.058e-06,
      "loss": 0.2212,
      "step": 530
    },
    {
      "epoch": 5.054473876953125e-07,
      "model_forward_time": 1.714094877243042,
      "step": 530
    },
    {
      "dynamics_loss_avg": 0.07387355864048004,
      "epoch": 5.054473876953125e-07,
      "step": 530
    },
    {
      "action_loss_avg": 0.13758794888854026,
      "epoch": 5.054473876953125e-07,
      "step": 530
    },
    {
      "epoch": 5.054473876953125e-07,
      "step": 530,
      "training_step_time": 6.704007148742676
    },
    {
      "epoch": 5.064010620117188e-07,
      "model_forward_time": 1.7209513187408447,
      "step": 531
    },
    {
      "epoch": 5.064010620117188e-07,
      "step": 531,
      "training_step_time": 7.6211302280426025
    },
    {
      "epoch": 5.07354736328125e-07,
      "model_forward_time": 1.7509582042694092,
      "step": 532
    },
    {
      "epoch": 5.07354736328125e-07,
      "step": 532,
      "training_step_time": 7.6346659660339355
    },
    {
      "epoch": 5.083084106445313e-07,
      "model_forward_time": 1.712517261505127,
      "step": 533
    },
    {
      "epoch": 5.083084106445313e-07,
      "step": 533,
      "training_step_time": 7.687558650970459
    },
    {
      "epoch": 5.092620849609375e-07,
      "model_forward_time": 1.3414521217346191,
      "step": 534
    },
    {
      "epoch": 5.092620849609375e-07,
      "step": 534,
      "training_step_time": 9.050418615341187
    },
    {
      "epoch": 5.102157592773438e-07,
      "model_forward_time": 1.7205402851104736,
      "step": 535
    },
    {
      "epoch": 5.102157592773438e-07,
      "step": 535,
      "training_step_time": 8.597058773040771
    },
    {
      "epoch": 5.1116943359375e-07,
      "model_forward_time": 2.255642890930176,
      "step": 536
    },
    {
      "epoch": 5.1116943359375e-07,
      "step": 536,
      "training_step_time": 8.676592111587524
    },
    {
      "epoch": 5.121231079101563e-07,
      "model_forward_time": 1.715395212173462,
      "step": 537
    },
    {
      "epoch": 5.121231079101563e-07,
      "step": 537,
      "training_step_time": 7.590056896209717
    },
    {
      "epoch": 5.130767822265625e-07,
      "model_forward_time": 1.7156610488891602,
      "step": 538
    },
    {
      "epoch": 5.130767822265625e-07,
      "step": 538,
      "training_step_time": 6.686715364456177
    },
    {
      "epoch": 5.140304565429688e-07,
      "model_forward_time": 1.7071280479431152,
      "step": 539
    },
    {
      "epoch": 5.140304565429688e-07,
      "step": 539,
      "training_step_time": 6.721839666366577
    },
    {
      "epoch": 5.14984130859375e-07,
      "grad_norm": 4.689679145812988,
      "learning_rate": 1.078e-06,
      "loss": 0.2037,
      "step": 540
    },
    {
      "epoch": 5.14984130859375e-07,
      "model_forward_time": 1.7236104011535645,
      "step": 540
    },
    {
      "dynamics_loss_avg": 0.0663435086607933,
      "epoch": 5.14984130859375e-07,
      "step": 540
    },
    {
      "action_loss_avg": 0.11076994445174933,
      "epoch": 5.14984130859375e-07,
      "step": 540
    },
    {
      "epoch": 5.14984130859375e-07,
      "step": 540,
      "training_step_time": 7.60939884185791
    },
    {
      "epoch": 5.159378051757813e-07,
      "model_forward_time": 1.7502071857452393,
      "step": 541
    },
    {
      "epoch": 5.159378051757813e-07,
      "step": 541,
      "training_step_time": 7.570281744003296
    },
    {
      "epoch": 5.168914794921875e-07,
      "model_forward_time": 1.7173693180084229,
      "step": 542
    },
    {
      "epoch": 5.168914794921875e-07,
      "step": 542,
      "training_step_time": 7.710428953170776
    },
    {
      "epoch": 5.178451538085938e-07,
      "model_forward_time": 1.7128050327301025,
      "step": 543
    },
    {
      "epoch": 5.178451538085938e-07,
      "step": 543,
      "training_step_time": 10.415135145187378
    },
    {
      "epoch": 5.18798828125e-07,
      "model_forward_time": 1.7189607620239258,
      "step": 544
    },
    {
      "epoch": 5.18798828125e-07,
      "step": 544,
      "training_step_time": 8.872424125671387
    },
    {
      "epoch": 5.197525024414063e-07,
      "model_forward_time": 1.3322336673736572,
      "step": 545
    },
    {
      "epoch": 5.197525024414063e-07,
      "step": 545,
      "training_step_time": 8.771512508392334
    },
    {
      "epoch": 5.207061767578125e-07,
      "model_forward_time": 1.707510232925415,
      "step": 546
    },
    {
      "epoch": 5.207061767578125e-07,
      "step": 546,
      "training_step_time": 7.691835880279541
    },
    {
      "epoch": 5.216598510742188e-07,
      "model_forward_time": 1.338118076324463,
      "step": 547
    },
    {
      "epoch": 5.216598510742188e-07,
      "step": 547,
      "training_step_time": 7.116771459579468
    },
    {
      "epoch": 5.22613525390625e-07,
      "model_forward_time": 1.3431642055511475,
      "step": 548
    },
    {
      "epoch": 5.22613525390625e-07,
      "step": 548,
      "training_step_time": 6.766011476516724
    },
    {
      "epoch": 5.235671997070313e-07,
      "model_forward_time": 1.7094478607177734,
      "step": 549
    },
    {
      "epoch": 5.235671997070313e-07,
      "step": 549,
      "training_step_time": 7.3758704662323
    },
    {
      "epoch": 5.245208740234375e-07,
      "grad_norm": 3.0917892456054688,
      "learning_rate": 1.0980000000000001e-06,
      "loss": 0.2326,
      "step": 550
    },
    {
      "epoch": 5.245208740234375e-07,
      "model_forward_time": 1.7461283206939697,
      "step": 550
    },
    {
      "dynamics_loss_avg": 0.07251195330172777,
      "epoch": 5.245208740234375e-07,
      "step": 550
    },
    {
      "action_loss_avg": 0.1288819570094347,
      "epoch": 5.245208740234375e-07,
      "step": 550
    },
    {
      "epoch": 5.245208740234375e-07,
      "step": 550,
      "training_step_time": 7.640328884124756
    },
    {
      "epoch": 5.254745483398438e-07,
      "model_forward_time": 1.7099764347076416,
      "step": 551
    },
    {
      "epoch": 5.254745483398438e-07,
      "step": 551,
      "training_step_time": 7.722843885421753
    },
    {
      "epoch": 5.2642822265625e-07,
      "model_forward_time": 1.7256100177764893,
      "step": 552
    },
    {
      "epoch": 5.2642822265625e-07,
      "step": 552,
      "training_step_time": 8.277631044387817
    },
    {
      "epoch": 5.273818969726563e-07,
      "model_forward_time": 1.7176990509033203,
      "step": 553
    },
    {
      "epoch": 5.273818969726563e-07,
      "step": 553,
      "training_step_time": 8.531565189361572
    },
    {
      "epoch": 5.283355712890625e-07,
      "model_forward_time": 1.716538667678833,
      "step": 554
    },
    {
      "epoch": 5.283355712890625e-07,
      "step": 554,
      "training_step_time": 8.905240058898926
    },
    {
      "epoch": 5.292892456054688e-07,
      "model_forward_time": 2.2587249279022217,
      "step": 555
    },
    {
      "epoch": 5.292892456054688e-07,
      "step": 555,
      "training_step_time": 9.58755111694336
    },
    {
      "epoch": 5.30242919921875e-07,
      "model_forward_time": 1.7167840003967285,
      "step": 556
    },
    {
      "epoch": 5.30242919921875e-07,
      "step": 556,
      "training_step_time": 7.496184587478638
    },
    {
      "epoch": 5.311965942382813e-07,
      "model_forward_time": 1.7256693840026855,
      "step": 557
    },
    {
      "epoch": 5.311965942382813e-07,
      "step": 557,
      "training_step_time": 6.734807014465332
    },
    {
      "epoch": 5.321502685546875e-07,
      "model_forward_time": 1.7207961082458496,
      "step": 558
    },
    {
      "epoch": 5.321502685546875e-07,
      "step": 558,
      "training_step_time": 7.414226770401001
    },
    {
      "epoch": 5.331039428710938e-07,
      "model_forward_time": 1.7213149070739746,
      "step": 559
    },
    {
      "epoch": 5.331039428710938e-07,
      "step": 559,
      "training_step_time": 7.332615375518799
    },
    {
      "epoch": 5.340576171875e-07,
      "grad_norm": 2.420917510986328,
      "learning_rate": 1.1180000000000001e-06,
      "loss": 0.2208,
      "step": 560
    },
    {
      "epoch": 5.340576171875e-07,
      "model_forward_time": 1.753389835357666,
      "step": 560
    },
    {
      "dynamics_loss_avg": 0.08935407884418964,
      "epoch": 5.340576171875e-07,
      "step": 560
    },
    {
      "action_loss_avg": 0.14797514379024507,
      "epoch": 5.340576171875e-07,
      "step": 560
    },
    {
      "epoch": 5.340576171875e-07,
      "step": 560,
      "training_step_time": 7.811438322067261
    },
    {
      "epoch": 5.350112915039063e-07,
      "model_forward_time": 1.7432653903961182,
      "step": 561
    },
    {
      "epoch": 5.350112915039063e-07,
      "step": 561,
      "training_step_time": 7.597124338150024
    },
    {
      "epoch": 5.359649658203126e-07,
      "model_forward_time": 1.7451283931732178,
      "step": 562
    },
    {
      "epoch": 5.359649658203126e-07,
      "step": 562,
      "training_step_time": 8.750847816467285
    },
    {
      "epoch": 5.369186401367187e-07,
      "model_forward_time": 1.723201036453247,
      "step": 563
    },
    {
      "epoch": 5.369186401367187e-07,
      "step": 563,
      "training_step_time": 8.504651546478271
    },
    {
      "epoch": 5.37872314453125e-07,
      "model_forward_time": 2.833686590194702,
      "step": 564
    },
    {
      "epoch": 5.37872314453125e-07,
      "step": 564,
      "training_step_time": 7.701440811157227
    },
    {
      "epoch": 5.388259887695312e-07,
      "model_forward_time": 1.7241666316986084,
      "step": 565
    },
    {
      "epoch": 5.388259887695312e-07,
      "step": 565,
      "training_step_time": 7.652592420578003
    },
    {
      "epoch": 5.397796630859375e-07,
      "model_forward_time": 1.0219378471374512,
      "step": 566
    },
    {
      "epoch": 5.397796630859375e-07,
      "step": 566,
      "training_step_time": 6.683156490325928
    },
    {
      "epoch": 5.407333374023437e-07,
      "model_forward_time": 1.7321386337280273,
      "step": 567
    },
    {
      "epoch": 5.407333374023437e-07,
      "step": 567,
      "training_step_time": 7.4053919315338135
    },
    {
      "epoch": 5.4168701171875e-07,
      "model_forward_time": 1.7157981395721436,
      "step": 568
    },
    {
      "epoch": 5.4168701171875e-07,
      "step": 568,
      "training_step_time": 7.620771646499634
    },
    {
      "epoch": 5.426406860351562e-07,
      "model_forward_time": 1.7243154048919678,
      "step": 569
    },
    {
      "epoch": 5.426406860351562e-07,
      "step": 569,
      "training_step_time": 7.803140163421631
    },
    {
      "epoch": 5.435943603515625e-07,
      "grad_norm": 5.5335259437561035,
      "learning_rate": 1.1380000000000002e-06,
      "loss": 0.2225,
      "step": 570
    },
    {
      "epoch": 5.435943603515625e-07,
      "model_forward_time": 1.7240898609161377,
      "step": 570
    },
    {
      "dynamics_loss_avg": 0.08560980558395385,
      "epoch": 5.435943603515625e-07,
      "step": 570
    },
    {
      "action_loss_avg": 0.1347634270787239,
      "epoch": 5.435943603515625e-07,
      "step": 570
    },
    {
      "epoch": 5.435943603515625e-07,
      "step": 570,
      "training_step_time": 8.336438417434692
    },
    {
      "epoch": 5.445480346679687e-07,
      "model_forward_time": 1.7445900440216064,
      "step": 571
    },
    {
      "epoch": 5.445480346679687e-07,
      "step": 571,
      "training_step_time": 9.420020341873169
    },
    {
      "epoch": 5.45501708984375e-07,
      "model_forward_time": 1.7282404899597168,
      "step": 572
    },
    {
      "epoch": 5.45501708984375e-07,
      "step": 572,
      "training_step_time": 7.806726694107056
    },
    {
      "epoch": 5.464553833007812e-07,
      "model_forward_time": 1.7262535095214844,
      "step": 573
    },
    {
      "epoch": 5.464553833007812e-07,
      "step": 573,
      "training_step_time": 7.742629289627075
    },
    {
      "epoch": 5.474090576171875e-07,
      "model_forward_time": 2.9303486347198486,
      "step": 574
    },
    {
      "epoch": 5.474090576171875e-07,
      "step": 574,
      "training_step_time": 7.7200376987457275
    },
    {
      "epoch": 5.483627319335937e-07,
      "model_forward_time": 1.7239429950714111,
      "step": 575
    },
    {
      "epoch": 5.483627319335937e-07,
      "step": 575,
      "training_step_time": 6.748235702514648
    },
    {
      "epoch": 5.4931640625e-07,
      "model_forward_time": 1.732325792312622,
      "step": 576
    },
    {
      "epoch": 5.4931640625e-07,
      "step": 576,
      "training_step_time": 7.365375518798828
    },
    {
      "epoch": 5.502700805664062e-07,
      "model_forward_time": 1.724428415298462,
      "step": 577
    },
    {
      "epoch": 5.502700805664062e-07,
      "step": 577,
      "training_step_time": 7.275146484375
    },
    {
      "epoch": 5.512237548828125e-07,
      "model_forward_time": 1.7233803272247314,
      "step": 578
    },
    {
      "epoch": 5.512237548828125e-07,
      "step": 578,
      "training_step_time": 7.736884117126465
    },
    {
      "epoch": 5.521774291992187e-07,
      "model_forward_time": 1.726097583770752,
      "step": 579
    },
    {
      "epoch": 5.521774291992187e-07,
      "step": 579,
      "training_step_time": 7.627735376358032
    },
    {
      "epoch": 5.53131103515625e-07,
      "grad_norm": 3.401325225830078,
      "learning_rate": 1.1580000000000002e-06,
      "loss": 0.2096,
      "step": 580
    },
    {
      "epoch": 5.53131103515625e-07,
      "model_forward_time": 1.8037521839141846,
      "step": 580
    },
    {
      "dynamics_loss_avg": 0.07608464416116476,
      "epoch": 5.53131103515625e-07,
      "step": 580
    },
    {
      "action_loss_avg": 0.14591020625084639,
      "epoch": 5.53131103515625e-07,
      "step": 580
    },
    {
      "epoch": 5.53131103515625e-07,
      "step": 580,
      "training_step_time": 9.469701051712036
    },
    {
      "epoch": 5.540847778320312e-07,
      "model_forward_time": 1.728367805480957,
      "step": 581
    },
    {
      "epoch": 5.540847778320312e-07,
      "step": 581,
      "training_step_time": 8.566266059875488
    },
    {
      "epoch": 5.550384521484375e-07,
      "model_forward_time": 1.7284789085388184,
      "step": 582
    },
    {
      "epoch": 5.550384521484375e-07,
      "step": 582,
      "training_step_time": 7.827998161315918
    },
    {
      "epoch": 5.559921264648437e-07,
      "model_forward_time": 2.8845713138580322,
      "step": 583
    },
    {
      "epoch": 5.559921264648437e-07,
      "step": 583,
      "training_step_time": 7.721912145614624
    },
    {
      "epoch": 5.5694580078125e-07,
      "model_forward_time": 1.7243373394012451,
      "step": 584
    },
    {
      "epoch": 5.5694580078125e-07,
      "step": 584,
      "training_step_time": 7.555585145950317
    },
    {
      "epoch": 5.578994750976562e-07,
      "model_forward_time": 1.7340943813323975,
      "step": 585
    },
    {
      "epoch": 5.578994750976562e-07,
      "step": 585,
      "training_step_time": 7.382938861846924
    },
    {
      "epoch": 5.588531494140625e-07,
      "model_forward_time": 1.7266342639923096,
      "step": 586
    },
    {
      "epoch": 5.588531494140625e-07,
      "step": 586,
      "training_step_time": 7.317123174667358
    },
    {
      "epoch": 5.598068237304687e-07,
      "model_forward_time": 1.725665807723999,
      "step": 587
    },
    {
      "epoch": 5.598068237304687e-07,
      "step": 587,
      "training_step_time": 9.283560514450073
    },
    {
      "epoch": 5.60760498046875e-07,
      "model_forward_time": 1.7246572971343994,
      "step": 588
    },
    {
      "epoch": 5.60760498046875e-07,
      "step": 588,
      "training_step_time": 7.629165410995483
    },
    {
      "epoch": 5.617141723632812e-07,
      "model_forward_time": 1.7464094161987305,
      "step": 589
    },
    {
      "epoch": 5.617141723632812e-07,
      "step": 589,
      "training_step_time": 9.298720359802246
    },
    {
      "epoch": 5.626678466796875e-07,
      "grad_norm": 2.6369056701660156,
      "learning_rate": 1.1780000000000002e-06,
      "loss": 0.2236,
      "step": 590
    },
    {
      "epoch": 5.626678466796875e-07,
      "model_forward_time": 1.7326996326446533,
      "step": 590
    },
    {
      "dynamics_loss_avg": 0.07005206793546677,
      "epoch": 5.626678466796875e-07,
      "step": 590
    },
    {
      "action_loss_avg": 0.08541158605366946,
      "epoch": 5.626678466796875e-07,
      "step": 590
    },
    {
      "epoch": 5.626678466796875e-07,
      "step": 590,
      "training_step_time": 9.274857759475708
    },
    {
      "epoch": 5.636215209960937e-07,
      "model_forward_time": 1.728334665298462,
      "step": 591
    },
    {
      "epoch": 5.636215209960937e-07,
      "step": 591,
      "training_step_time": 7.775325059890747
    },
    {
      "epoch": 5.645751953125e-07,
      "model_forward_time": 2.89408278465271,
      "step": 592
    },
    {
      "epoch": 5.645751953125e-07,
      "step": 592,
      "training_step_time": 7.735386610031128
    },
    {
      "epoch": 5.655288696289062e-07,
      "model_forward_time": 1.7318990230560303,
      "step": 593
    },
    {
      "epoch": 5.655288696289062e-07,
      "step": 593,
      "training_step_time": 7.569200754165649
    },
    {
      "epoch": 5.664825439453125e-07,
      "model_forward_time": 1.722550392150879,
      "step": 594
    },
    {
      "epoch": 5.664825439453125e-07,
      "step": 594,
      "training_step_time": 7.437907457351685
    },
    {
      "epoch": 5.674362182617187e-07,
      "model_forward_time": 1.7235891819000244,
      "step": 595
    },
    {
      "epoch": 5.674362182617187e-07,
      "step": 595,
      "training_step_time": 7.268556594848633
    },
    {
      "epoch": 5.68389892578125e-07,
      "model_forward_time": 1.7223377227783203,
      "step": 596
    },
    {
      "epoch": 5.68389892578125e-07,
      "step": 596,
      "training_step_time": 8.995145082473755
    },
    {
      "epoch": 5.693435668945312e-07,
      "model_forward_time": 1.7260065078735352,
      "step": 597
    },
    {
      "epoch": 5.693435668945312e-07,
      "step": 597,
      "training_step_time": 7.849384784698486
    },
    {
      "epoch": 5.702972412109375e-07,
      "model_forward_time": 1.7403275966644287,
      "step": 598
    },
    {
      "epoch": 5.702972412109375e-07,
      "step": 598,
      "training_step_time": 8.687133073806763
    },
    {
      "epoch": 5.712509155273437e-07,
      "model_forward_time": 1.738161325454712,
      "step": 599
    },
    {
      "epoch": 5.712509155273437e-07,
      "step": 599,
      "training_step_time": 8.578685522079468
    },
    {
      "epoch": 5.7220458984375e-07,
      "grad_norm": 4.262243270874023,
      "learning_rate": 1.1980000000000002e-06,
      "loss": 0.2392,
      "step": 600
    },
    {
      "epoch": 5.7220458984375e-07,
      "model_forward_time": 1.7168428897857666,
      "step": 600
    },
    {
      "dynamics_loss_avg": 0.09572584331035613,
      "epoch": 5.7220458984375e-07,
      "step": 600
    },
    {
      "action_loss_avg": 0.12345724850893021,
      "epoch": 5.7220458984375e-07,
      "step": 600
    },
    {
      "epoch": 5.7220458984375e-07,
      "step": 600,
      "training_step_time": 8.54109239578247
    },
    {
      "epoch": 5.731582641601562e-07,
      "model_forward_time": 2.4551050662994385,
      "step": 601
    },
    {
      "epoch": 5.731582641601562e-07,
      "step": 601,
      "training_step_time": 8.206552743911743
    },
    {
      "epoch": 5.741119384765625e-07,
      "model_forward_time": 1.7324192523956299,
      "step": 602
    },
    {
      "epoch": 5.741119384765625e-07,
      "step": 602,
      "training_step_time": 7.566995143890381
    },
    {
      "epoch": 5.750656127929687e-07,
      "model_forward_time": 1.724869728088379,
      "step": 603
    },
    {
      "epoch": 5.750656127929687e-07,
      "step": 603,
      "training_step_time": 7.380269765853882
    },
    {
      "epoch": 5.76019287109375e-07,
      "model_forward_time": 1.7264573574066162,
      "step": 604
    },
    {
      "epoch": 5.76019287109375e-07,
      "step": 604,
      "training_step_time": 7.259525537490845
    },
    {
      "epoch": 5.769729614257812e-07,
      "model_forward_time": 1.7243285179138184,
      "step": 605
    },
    {
      "epoch": 5.769729614257812e-07,
      "step": 605,
      "training_step_time": 8.698610305786133
    },
    {
      "epoch": 5.779266357421875e-07,
      "model_forward_time": 1.3633525371551514,
      "step": 606
    },
    {
      "epoch": 5.779266357421875e-07,
      "step": 606,
      "training_step_time": 7.698895215988159
    },
    {
      "epoch": 5.788803100585937e-07,
      "model_forward_time": 1.7311193943023682,
      "step": 607
    },
    {
      "epoch": 5.788803100585937e-07,
      "step": 607,
      "training_step_time": 8.29383134841919
    },
    {
      "epoch": 5.79833984375e-07,
      "model_forward_time": 1.7213568687438965,
      "step": 608
    },
    {
      "epoch": 5.79833984375e-07,
      "step": 608,
      "training_step_time": 8.283480644226074
    },
    {
      "epoch": 5.807876586914062e-07,
      "model_forward_time": 1.7249689102172852,
      "step": 609
    },
    {
      "epoch": 5.807876586914062e-07,
      "step": 609,
      "training_step_time": 7.626710653305054
    },
    {
      "epoch": 5.817413330078125e-07,
      "grad_norm": 2.658095121383667,
      "learning_rate": 1.2180000000000002e-06,
      "loss": 0.2066,
      "step": 610
    },
    {
      "epoch": 5.817413330078125e-07,
      "model_forward_time": 1.7195475101470947,
      "step": 610
    },
    {
      "dynamics_loss_avg": 0.08622294142842293,
      "epoch": 5.817413330078125e-07,
      "step": 610
    },
    {
      "action_loss_avg": 0.09966701678931714,
      "epoch": 5.817413330078125e-07,
      "step": 610
    },
    {
      "epoch": 5.817413330078125e-07,
      "step": 610,
      "training_step_time": 8.783174276351929
    },
    {
      "epoch": 5.826950073242187e-07,
      "model_forward_time": 2.919680118560791,
      "step": 611
    },
    {
      "epoch": 5.826950073242187e-07,
      "step": 611,
      "training_step_time": 7.7456138134002686
    },
    {
      "epoch": 5.83648681640625e-07,
      "model_forward_time": 1.7242379188537598,
      "step": 612
    },
    {
      "epoch": 5.83648681640625e-07,
      "step": 612,
      "training_step_time": 7.391110181808472
    },
    {
      "epoch": 5.846023559570312e-07,
      "model_forward_time": 1.7211792469024658,
      "step": 613
    },
    {
      "epoch": 5.846023559570312e-07,
      "step": 613,
      "training_step_time": 7.269721984863281
    },
    {
      "epoch": 5.855560302734375e-07,
      "model_forward_time": 1.7199676036834717,
      "step": 614
    },
    {
      "epoch": 5.855560302734375e-07,
      "step": 614,
      "training_step_time": 6.703808546066284
    },
    {
      "epoch": 5.865097045898437e-07,
      "model_forward_time": 1.7300317287445068,
      "step": 615
    },
    {
      "epoch": 5.865097045898437e-07,
      "step": 615,
      "training_step_time": 7.76751446723938
    },
    {
      "epoch": 5.8746337890625e-07,
      "model_forward_time": 1.7321975231170654,
      "step": 616
    },
    {
      "epoch": 5.8746337890625e-07,
      "step": 616,
      "training_step_time": 8.651460886001587
    },
    {
      "epoch": 5.884170532226562e-07,
      "model_forward_time": 1.7250850200653076,
      "step": 617
    },
    {
      "epoch": 5.884170532226562e-07,
      "step": 617,
      "training_step_time": 9.42465591430664
    },
    {
      "epoch": 5.893707275390625e-07,
      "model_forward_time": 1.7417514324188232,
      "step": 618
    },
    {
      "epoch": 5.893707275390625e-07,
      "step": 618,
      "training_step_time": 9.254010915756226
    },
    {
      "epoch": 5.903244018554687e-07,
      "model_forward_time": 1.7291135787963867,
      "step": 619
    },
    {
      "epoch": 5.903244018554687e-07,
      "step": 619,
      "training_step_time": 7.861593723297119
    },
    {
      "epoch": 5.91278076171875e-07,
      "grad_norm": 5.343528747558594,
      "learning_rate": 1.238e-06,
      "loss": 0.181,
      "step": 620
    },
    {
      "epoch": 5.91278076171875e-07,
      "model_forward_time": 2.8789775371551514,
      "step": 620
    },
    {
      "dynamics_loss_avg": 0.0991497978568077,
      "epoch": 5.91278076171875e-07,
      "step": 620
    },
    {
      "action_loss_avg": 0.10658937245607376,
      "epoch": 5.91278076171875e-07,
      "step": 620
    },
    {
      "epoch": 5.91278076171875e-07,
      "step": 620,
      "training_step_time": 7.710394859313965
    },
    {
      "epoch": 5.922317504882812e-07,
      "model_forward_time": 1.725830316543579,
      "step": 621
    },
    {
      "epoch": 5.922317504882812e-07,
      "step": 621,
      "training_step_time": 7.495280027389526
    },
    {
      "epoch": 5.931854248046875e-07,
      "model_forward_time": 1.7313804626464844,
      "step": 622
    },
    {
      "epoch": 5.931854248046875e-07,
      "step": 622,
      "training_step_time": 7.2391581535339355
    },
    {
      "epoch": 5.941390991210937e-07,
      "model_forward_time": 1.7234435081481934,
      "step": 623
    },
    {
      "epoch": 5.941390991210937e-07,
      "step": 623,
      "training_step_time": 7.08789849281311
    },
    {
      "epoch": 5.950927734375e-07,
      "model_forward_time": 1.7217693328857422,
      "step": 624
    },
    {
      "epoch": 5.950927734375e-07,
      "step": 624,
      "training_step_time": 7.777331113815308
    },
    {
      "epoch": 5.960464477539062e-07,
      "model_forward_time": 1.7302563190460205,
      "step": 625
    },
    {
      "epoch": 5.960464477539062e-07,
      "step": 625,
      "training_step_time": 7.578465938568115
    },
    {
      "epoch": 5.970001220703125e-07,
      "model_forward_time": 1.7453110218048096,
      "step": 626
    },
    {
      "epoch": 5.970001220703125e-07,
      "step": 626,
      "training_step_time": 8.48749303817749
    },
    {
      "epoch": 5.979537963867188e-07,
      "model_forward_time": 1.7506911754608154,
      "step": 627
    },
    {
      "epoch": 5.979537963867188e-07,
      "step": 627,
      "training_step_time": 9.578720092773438
    },
    {
      "epoch": 5.98907470703125e-07,
      "model_forward_time": 1.7253592014312744,
      "step": 628
    },
    {
      "epoch": 5.98907470703125e-07,
      "step": 628,
      "training_step_time": 7.893633127212524
    },
    {
      "epoch": 5.998611450195313e-07,
      "model_forward_time": 2.8325884342193604,
      "step": 629
    },
    {
      "epoch": 5.998611450195313e-07,
      "step": 629,
      "training_step_time": 7.686512231826782
    },
    {
      "epoch": 6.008148193359375e-07,
      "grad_norm": 2.5795681476593018,
      "learning_rate": 1.258e-06,
      "loss": 0.1709,
      "step": 630
    },
    {
      "epoch": 6.008148193359375e-07,
      "model_forward_time": 1.7232458591461182,
      "step": 630
    },
    {
      "dynamics_loss_avg": 0.07634822558611631,
      "epoch": 6.008148193359375e-07,
      "step": 630
    },
    {
      "action_loss_avg": 0.09399538673460484,
      "epoch": 6.008148193359375e-07,
      "step": 630
    },
    {
      "epoch": 6.008148193359375e-07,
      "step": 630,
      "training_step_time": 7.6020190715789795
    },
    {
      "epoch": 6.017684936523438e-07,
      "model_forward_time": 1.721292495727539,
      "step": 631
    },
    {
      "epoch": 6.017684936523438e-07,
      "step": 631,
      "training_step_time": 7.290177822113037
    },
    {
      "epoch": 6.0272216796875e-07,
      "model_forward_time": 1.725325584411621,
      "step": 632
    },
    {
      "epoch": 6.0272216796875e-07,
      "step": 632,
      "training_step_time": 6.7842724323272705
    },
    {
      "epoch": 6.036758422851563e-07,
      "model_forward_time": 1.725900650024414,
      "step": 633
    },
    {
      "epoch": 6.036758422851563e-07,
      "step": 633,
      "training_step_time": 7.795313835144043
    },
    {
      "epoch": 6.046295166015625e-07,
      "model_forward_time": 1.720670223236084,
      "step": 634
    },
    {
      "epoch": 6.046295166015625e-07,
      "step": 634,
      "training_step_time": 8.554425239562988
    },
    {
      "epoch": 6.055831909179688e-07,
      "model_forward_time": 1.7467622756958008,
      "step": 635
    },
    {
      "epoch": 6.055831909179688e-07,
      "step": 635,
      "training_step_time": 8.291113138198853
    },
    {
      "epoch": 6.06536865234375e-07,
      "model_forward_time": 1.7417259216308594,
      "step": 636
    },
    {
      "epoch": 6.06536865234375e-07,
      "step": 636,
      "training_step_time": 8.501298189163208
    },
    {
      "epoch": 6.074905395507813e-07,
      "model_forward_time": 1.7325351238250732,
      "step": 637
    },
    {
      "epoch": 6.074905395507813e-07,
      "step": 637,
      "training_step_time": 8.657968521118164
    },
    {
      "epoch": 6.084442138671875e-07,
      "model_forward_time": 2.8260512351989746,
      "step": 638
    },
    {
      "epoch": 6.084442138671875e-07,
      "step": 638,
      "training_step_time": 8.512786865234375
    },
    {
      "epoch": 6.093978881835938e-07,
      "model_forward_time": 1.7303667068481445,
      "step": 639
    },
    {
      "epoch": 6.093978881835938e-07,
      "step": 639,
      "training_step_time": 7.544106483459473
    },
    {
      "epoch": 6.103515625e-07,
      "grad_norm": 6.689511775970459,
      "learning_rate": 1.278e-06,
      "loss": 0.1776,
      "step": 640
    },
    {
      "epoch": 6.103515625e-07,
      "model_forward_time": 1.730053424835205,
      "step": 640
    },
    {
      "dynamics_loss_avg": 0.07357973624020815,
      "epoch": 6.103515625e-07,
      "step": 640
    },
    {
      "action_loss_avg": 0.0869475096464157,
      "epoch": 6.103515625e-07,
      "step": 640
    },
    {
      "epoch": 6.103515625e-07,
      "step": 640,
      "training_step_time": 7.255635023117065
    },
    {
      "epoch": 6.113052368164063e-07,
      "model_forward_time": 1.7323129177093506,
      "step": 641
    },
    {
      "epoch": 6.113052368164063e-07,
      "step": 641,
      "training_step_time": 6.780760765075684
    },
    {
      "epoch": 6.122589111328125e-07,
      "model_forward_time": 1.7264387607574463,
      "step": 642
    },
    {
      "epoch": 6.122589111328125e-07,
      "step": 642,
      "training_step_time": 7.792999982833862
    },
    {
      "epoch": 6.132125854492188e-07,
      "model_forward_time": 1.724048137664795,
      "step": 643
    },
    {
      "epoch": 6.132125854492188e-07,
      "step": 643,
      "training_step_time": 7.566452503204346
    },
    {
      "epoch": 6.14166259765625e-07,
      "model_forward_time": 1.7424473762512207,
      "step": 644
    },
    {
      "epoch": 6.14166259765625e-07,
      "step": 644,
      "training_step_time": 7.737737655639648
    },
    {
      "epoch": 6.151199340820313e-07,
      "model_forward_time": 1.7268803119659424,
      "step": 645
    },
    {
      "epoch": 6.151199340820313e-07,
      "step": 645,
      "training_step_time": 7.705058336257935
    },
    {
      "epoch": 6.160736083984375e-07,
      "model_forward_time": 1.7242071628570557,
      "step": 646
    },
    {
      "epoch": 6.160736083984375e-07,
      "step": 646,
      "training_step_time": 7.741744756698608
    },
    {
      "epoch": 6.170272827148438e-07,
      "model_forward_time": 1.720808744430542,
      "step": 647
    },
    {
      "epoch": 6.170272827148438e-07,
      "step": 647,
      "training_step_time": 8.65336012840271
    },
    {
      "epoch": 6.1798095703125e-07,
      "model_forward_time": 1.726837158203125,
      "step": 648
    },
    {
      "epoch": 6.1798095703125e-07,
      "step": 648,
      "training_step_time": 8.689261674880981
    },
    {
      "epoch": 6.189346313476563e-07,
      "model_forward_time": 1.7253313064575195,
      "step": 649
    },
    {
      "epoch": 6.189346313476563e-07,
      "step": 649,
      "training_step_time": 6.776116609573364
    },
    {
      "epoch": 6.198883056640625e-07,
      "grad_norm": 3.325247049331665,
      "learning_rate": 1.2980000000000001e-06,
      "loss": 0.1975,
      "step": 650
    },
    {
      "epoch": 6.198883056640625e-07,
      "model_forward_time": 1.723935604095459,
      "step": 650
    },
    {
      "dynamics_loss_avg": 0.07549346722662449,
      "epoch": 6.198883056640625e-07,
      "step": 650
    },
    {
      "action_loss_avg": 0.0748559108003974,
      "epoch": 6.198883056640625e-07,
      "step": 650
    },
    {
      "epoch": 6.198883056640625e-07,
      "step": 650,
      "training_step_time": 7.3096442222595215
    },
    {
      "epoch": 6.208419799804688e-07,
      "model_forward_time": 1.7228384017944336,
      "step": 651
    },
    {
      "epoch": 6.208419799804688e-07,
      "step": 651,
      "training_step_time": 8.40431833267212
    },
    {
      "epoch": 6.21795654296875e-07,
      "model_forward_time": 1.7205970287322998,
      "step": 652
    },
    {
      "epoch": 6.21795654296875e-07,
      "step": 652,
      "training_step_time": 7.586063623428345
    },
    {
      "epoch": 6.227493286132813e-07,
      "model_forward_time": 1.7503349781036377,
      "step": 653
    },
    {
      "epoch": 6.227493286132813e-07,
      "step": 653,
      "training_step_time": 7.74100136756897
    },
    {
      "epoch": 6.237030029296875e-07,
      "model_forward_time": 1.7259001731872559,
      "step": 654
    },
    {
      "epoch": 6.237030029296875e-07,
      "step": 654,
      "training_step_time": 9.143535852432251
    },
    {
      "epoch": 6.246566772460938e-07,
      "model_forward_time": 1.7518596649169922,
      "step": 655
    },
    {
      "epoch": 6.246566772460938e-07,
      "step": 655,
      "training_step_time": 8.56629228591919
    },
    {
      "epoch": 6.256103515625e-07,
      "model_forward_time": 1.7253754138946533,
      "step": 656
    },
    {
      "epoch": 6.256103515625e-07,
      "step": 656,
      "training_step_time": 7.752116680145264
    },
    {
      "epoch": 6.265640258789063e-07,
      "model_forward_time": 2.9470129013061523,
      "step": 657
    },
    {
      "epoch": 6.265640258789063e-07,
      "step": 657,
      "training_step_time": 7.813941240310669
    },
    {
      "epoch": 6.275177001953125e-07,
      "model_forward_time": 1.7292630672454834,
      "step": 658
    },
    {
      "epoch": 6.275177001953125e-07,
      "step": 658,
      "training_step_time": 7.671237468719482
    },
    {
      "epoch": 6.284713745117188e-07,
      "model_forward_time": 1.723644495010376,
      "step": 659
    },
    {
      "epoch": 6.284713745117188e-07,
      "step": 659,
      "training_step_time": 7.296071529388428
    },
    {
      "epoch": 6.29425048828125e-07,
      "grad_norm": 7.85638952255249,
      "learning_rate": 1.3180000000000001e-06,
      "loss": 0.1832,
      "step": 660
    },
    {
      "epoch": 6.29425048828125e-07,
      "model_forward_time": 1.7200839519500732,
      "step": 660
    },
    {
      "dynamics_loss_avg": 0.09324922263622284,
      "epoch": 6.29425048828125e-07,
      "step": 660
    },
    {
      "action_loss_avg": 0.1052811861038208,
      "epoch": 6.29425048828125e-07,
      "step": 660
    },
    {
      "epoch": 6.29425048828125e-07,
      "step": 660,
      "training_step_time": 8.107866287231445
    },
    {
      "epoch": 6.303787231445313e-07,
      "model_forward_time": 1.7269566059112549,
      "step": 661
    },
    {
      "epoch": 6.303787231445313e-07,
      "step": 661,
      "training_step_time": 8.884450197219849
    },
    {
      "epoch": 6.313323974609375e-07,
      "model_forward_time": 1.741048812866211,
      "step": 662
    },
    {
      "epoch": 6.313323974609375e-07,
      "step": 662,
      "training_step_time": 8.24821662902832
    },
    {
      "epoch": 6.322860717773438e-07,
      "model_forward_time": 1.7254493236541748,
      "step": 663
    },
    {
      "epoch": 6.322860717773438e-07,
      "step": 663,
      "training_step_time": 8.342106342315674
    },
    {
      "epoch": 6.3323974609375e-07,
      "model_forward_time": 1.7409968376159668,
      "step": 664
    },
    {
      "epoch": 6.3323974609375e-07,
      "step": 664,
      "training_step_time": 8.76207685470581
    },
    {
      "epoch": 6.341934204101563e-07,
      "model_forward_time": 1.7207162380218506,
      "step": 665
    },
    {
      "epoch": 6.341934204101563e-07,
      "step": 665,
      "training_step_time": 8.334365129470825
    },
    {
      "epoch": 6.351470947265625e-07,
      "model_forward_time": 2.90909743309021,
      "step": 666
    },
    {
      "epoch": 6.351470947265625e-07,
      "step": 666,
      "training_step_time": 7.8171727657318115
    },
    {
      "epoch": 6.361007690429688e-07,
      "model_forward_time": 1.725325584411621,
      "step": 667
    },
    {
      "epoch": 6.361007690429688e-07,
      "step": 667,
      "training_step_time": 7.726878881454468
    },
    {
      "epoch": 6.37054443359375e-07,
      "model_forward_time": 1.7149405479431152,
      "step": 668
    },
    {
      "epoch": 6.37054443359375e-07,
      "step": 668,
      "training_step_time": 7.344696998596191
    },
    {
      "epoch": 6.380081176757813e-07,
      "model_forward_time": 1.726698637008667,
      "step": 669
    },
    {
      "epoch": 6.380081176757813e-07,
      "step": 669,
      "training_step_time": 7.215234994888306
    },
    {
      "epoch": 6.389617919921875e-07,
      "grad_norm": 2.3712210655212402,
      "learning_rate": 1.3380000000000001e-06,
      "loss": 0.1787,
      "step": 670
    },
    {
      "epoch": 6.389617919921875e-07,
      "model_forward_time": 1.7106621265411377,
      "step": 670
    },
    {
      "dynamics_loss_avg": 0.08276114240288734,
      "epoch": 6.389617919921875e-07,
      "step": 670
    },
    {
      "action_loss_avg": 0.06926674763672054,
      "epoch": 6.389617919921875e-07,
      "step": 670
    },
    {
      "epoch": 6.389617919921875e-07,
      "step": 670,
      "training_step_time": 7.868170261383057
    },
    {
      "epoch": 6.399154663085938e-07,
      "model_forward_time": 1.7418053150177002,
      "step": 671
    },
    {
      "epoch": 6.399154663085938e-07,
      "step": 671,
      "training_step_time": 8.621953010559082
    },
    {
      "epoch": 6.40869140625e-07,
      "model_forward_time": 1.716928243637085,
      "step": 672
    },
    {
      "epoch": 6.40869140625e-07,
      "step": 672,
      "training_step_time": 10.071876287460327
    },
    {
      "epoch": 6.418228149414063e-07,
      "model_forward_time": 1.7314999103546143,
      "step": 673
    },
    {
      "epoch": 6.418228149414063e-07,
      "step": 673,
      "training_step_time": 8.410414457321167
    },
    {
      "epoch": 6.427764892578125e-07,
      "model_forward_time": 1.7265594005584717,
      "step": 674
    },
    {
      "epoch": 6.427764892578125e-07,
      "step": 674,
      "training_step_time": 8.45708179473877
    },
    {
      "epoch": 6.437301635742188e-07,
      "model_forward_time": 2.8800623416900635,
      "step": 675
    },
    {
      "epoch": 6.437301635742188e-07,
      "step": 675,
      "training_step_time": 7.845203638076782
    },
    {
      "epoch": 6.44683837890625e-07,
      "model_forward_time": 1.7120022773742676,
      "step": 676
    },
    {
      "epoch": 6.44683837890625e-07,
      "step": 676,
      "training_step_time": 7.720349073410034
    },
    {
      "epoch": 6.456375122070313e-07,
      "model_forward_time": 1.7375857830047607,
      "step": 677
    },
    {
      "epoch": 6.456375122070313e-07,
      "step": 677,
      "training_step_time": 7.354663610458374
    },
    {
      "epoch": 6.465911865234375e-07,
      "model_forward_time": 1.722278118133545,
      "step": 678
    },
    {
      "epoch": 6.465911865234375e-07,
      "step": 678,
      "training_step_time": 7.35792350769043
    },
    {
      "epoch": 6.475448608398438e-07,
      "model_forward_time": 1.71706223487854,
      "step": 679
    },
    {
      "epoch": 6.475448608398438e-07,
      "step": 679,
      "training_step_time": 7.788946151733398
    },
    {
      "epoch": 6.4849853515625e-07,
      "grad_norm": 4.931527614593506,
      "learning_rate": 1.3580000000000002e-06,
      "loss": 0.1821,
      "step": 680
    },
    {
      "epoch": 6.4849853515625e-07,
      "model_forward_time": 1.7546532154083252,
      "step": 680
    },
    {
      "dynamics_loss_avg": 0.0930489581078291,
      "epoch": 6.4849853515625e-07,
      "step": 680
    },
    {
      "action_loss_avg": 0.090859554708004,
      "epoch": 6.4849853515625e-07,
      "step": 680
    },
    {
      "epoch": 6.4849853515625e-07,
      "step": 680,
      "training_step_time": 8.56687331199646
    },
    {
      "epoch": 6.494522094726563e-07,
      "model_forward_time": 1.7251243591308594,
      "step": 681
    },
    {
      "epoch": 6.494522094726563e-07,
      "step": 681,
      "training_step_time": 8.876119375228882
    },
    {
      "epoch": 6.504058837890625e-07,
      "model_forward_time": 1.7347047328948975,
      "step": 682
    },
    {
      "epoch": 6.504058837890625e-07,
      "step": 682,
      "training_step_time": 7.713012456893921
    },
    {
      "epoch": 6.513595581054688e-07,
      "model_forward_time": 1.7250289916992188,
      "step": 683
    },
    {
      "epoch": 6.513595581054688e-07,
      "step": 683,
      "training_step_time": 8.351670980453491
    },
    {
      "epoch": 6.52313232421875e-07,
      "model_forward_time": 2.8466737270355225,
      "step": 684
    },
    {
      "epoch": 6.52313232421875e-07,
      "step": 684,
      "training_step_time": 8.413408041000366
    },
    {
      "epoch": 6.532669067382813e-07,
      "model_forward_time": 1.724571704864502,
      "step": 685
    },
    {
      "epoch": 6.532669067382813e-07,
      "step": 685,
      "training_step_time": 7.6782097816467285
    },
    {
      "epoch": 6.542205810546876e-07,
      "model_forward_time": 1.7158093452453613,
      "step": 686
    },
    {
      "epoch": 6.542205810546876e-07,
      "step": 686,
      "training_step_time": 7.467078924179077
    },
    {
      "epoch": 6.551742553710938e-07,
      "model_forward_time": 1.7195689678192139,
      "step": 687
    },
    {
      "epoch": 6.551742553710938e-07,
      "step": 687,
      "training_step_time": 6.813044309616089
    },
    {
      "epoch": 6.561279296875e-07,
      "model_forward_time": 1.0210521221160889,
      "step": 688
    },
    {
      "epoch": 6.561279296875e-07,
      "step": 688,
      "training_step_time": 7.7669453620910645
    },
    {
      "epoch": 6.570816040039062e-07,
      "model_forward_time": 1.3830475807189941,
      "step": 689
    },
    {
      "epoch": 6.570816040039062e-07,
      "step": 689,
      "training_step_time": 7.68363356590271
    },
    {
      "epoch": 6.580352783203124e-07,
      "grad_norm": 5.586204528808594,
      "learning_rate": 1.3780000000000002e-06,
      "loss": 0.1815,
      "step": 690
    },
    {
      "epoch": 6.580352783203124e-07,
      "model_forward_time": 1.727998971939087,
      "step": 690
    },
    {
      "dynamics_loss_avg": 0.1079309731721878,
      "epoch": 6.580352783203124e-07,
      "step": 690
    },
    {
      "action_loss_avg": 0.0873658873140812,
      "epoch": 6.580352783203124e-07,
      "step": 690
    },
    {
      "epoch": 6.580352783203124e-07,
      "step": 690,
      "training_step_time": 9.33120322227478
    },
    {
      "epoch": 6.589889526367187e-07,
      "model_forward_time": 1.726147174835205,
      "step": 691
    },
    {
      "epoch": 6.589889526367187e-07,
      "step": 691,
      "training_step_time": 9.05370306968689
    },
    {
      "epoch": 6.59942626953125e-07,
      "model_forward_time": 1.7289795875549316,
      "step": 692
    },
    {
      "epoch": 6.59942626953125e-07,
      "step": 692,
      "training_step_time": 8.774030208587646
    },
    {
      "epoch": 6.608963012695312e-07,
      "model_forward_time": 1.7273797988891602,
      "step": 693
    },
    {
      "epoch": 6.608963012695312e-07,
      "step": 693,
      "training_step_time": 7.712717533111572
    },
    {
      "epoch": 6.618499755859375e-07,
      "model_forward_time": 2.8653597831726074,
      "step": 694
    },
    {
      "epoch": 6.618499755859375e-07,
      "step": 694,
      "training_step_time": 7.723679542541504
    },
    {
      "epoch": 6.628036499023437e-07,
      "model_forward_time": 1.7259438037872314,
      "step": 695
    },
    {
      "epoch": 6.628036499023437e-07,
      "step": 695,
      "training_step_time": 7.685678720474243
    },
    {
      "epoch": 6.6375732421875e-07,
      "model_forward_time": 1.7312297821044922,
      "step": 696
    },
    {
      "epoch": 6.6375732421875e-07,
      "step": 696,
      "training_step_time": 6.764790058135986
    },
    {
      "epoch": 6.647109985351562e-07,
      "model_forward_time": 1.726228952407837,
      "step": 697
    },
    {
      "epoch": 6.647109985351562e-07,
      "step": 697,
      "training_step_time": 7.787076234817505
    },
    {
      "epoch": 6.656646728515625e-07,
      "model_forward_time": 1.761843204498291,
      "step": 698
    },
    {
      "epoch": 6.656646728515625e-07,
      "step": 698,
      "training_step_time": 8.199320077896118
    },
    {
      "epoch": 6.666183471679687e-07,
      "model_forward_time": 1.7237701416015625,
      "step": 699
    },
    {
      "epoch": 6.666183471679687e-07,
      "step": 699,
      "training_step_time": 7.782749176025391
    },
    {
      "epoch": 6.67572021484375e-07,
      "grad_norm": 5.366738319396973,
      "learning_rate": 1.3980000000000002e-06,
      "loss": 0.163,
      "step": 700
    },
    {
      "epoch": 6.67572021484375e-07,
      "model_forward_time": 1.366661548614502,
      "step": 700
    },
    {
      "dynamics_loss_avg": 0.07850152738392353,
      "epoch": 6.67572021484375e-07,
      "step": 700
    },
    {
      "action_loss_avg": 0.06844356525689363,
      "epoch": 6.67572021484375e-07,
      "step": 700
    },
    {
      "epoch": 6.67572021484375e-07,
      "step": 700,
      "training_step_time": 7.6743855476379395
    },
    {
      "epoch": 6.685256958007812e-07,
      "model_forward_time": 1.7455875873565674,
      "step": 701
    },
    {
      "epoch": 6.685256958007812e-07,
      "step": 701,
      "training_step_time": 8.73367190361023
    },
    {
      "epoch": 6.694793701171875e-07,
      "model_forward_time": 1.719348669052124,
      "step": 702
    },
    {
      "epoch": 6.694793701171875e-07,
      "step": 702,
      "training_step_time": 7.72322678565979
    },
    {
      "epoch": 6.704330444335937e-07,
      "model_forward_time": 2.8609659671783447,
      "step": 703
    },
    {
      "epoch": 6.704330444335937e-07,
      "step": 703,
      "training_step_time": 7.769454479217529
    },
    {
      "epoch": 6.7138671875e-07,
      "model_forward_time": 1.7265710830688477,
      "step": 704
    },
    {
      "epoch": 6.7138671875e-07,
      "step": 704,
      "training_step_time": 8.673089504241943
    },
    {
      "epoch": 6.723403930664062e-07,
      "model_forward_time": 1.732783317565918,
      "step": 705
    },
    {
      "epoch": 6.723403930664062e-07,
      "step": 705,
      "training_step_time": 6.789691925048828
    },
    {
      "epoch": 6.732940673828125e-07,
      "model_forward_time": 1.3676235675811768,
      "step": 706
    },
    {
      "epoch": 6.732940673828125e-07,
      "step": 706,
      "training_step_time": 7.747331857681274
    },
    {
      "epoch": 6.742477416992187e-07,
      "model_forward_time": 1.7409238815307617,
      "step": 707
    },
    {
      "epoch": 6.742477416992187e-07,
      "step": 707,
      "training_step_time": 8.351013422012329
    },
    {
      "epoch": 6.75201416015625e-07,
      "model_forward_time": 1.7248289585113525,
      "step": 708
    },
    {
      "epoch": 6.75201416015625e-07,
      "step": 708,
      "training_step_time": 8.355957508087158
    },
    {
      "epoch": 6.761550903320312e-07,
      "model_forward_time": 1.7234439849853516,
      "step": 709
    },
    {
      "epoch": 6.761550903320312e-07,
      "step": 709,
      "training_step_time": 7.7517781257629395
    },
    {
      "epoch": 6.771087646484375e-07,
      "grad_norm": 4.55839204788208,
      "learning_rate": 1.4180000000000002e-06,
      "loss": 0.1687,
      "step": 710
    },
    {
      "epoch": 6.771087646484375e-07,
      "model_forward_time": 1.7390146255493164,
      "step": 710
    },
    {
      "dynamics_loss_avg": 0.07293650098145008,
      "epoch": 6.771087646484375e-07,
      "step": 710
    },
    {
      "action_loss_avg": 0.06957529447972774,
      "epoch": 6.771087646484375e-07,
      "step": 710
    },
    {
      "epoch": 6.771087646484375e-07,
      "step": 710,
      "training_step_time": 7.818583726882935
    },
    {
      "epoch": 6.780624389648437e-07,
      "model_forward_time": 1.728907823562622,
      "step": 711
    },
    {
      "epoch": 6.780624389648437e-07,
      "step": 711,
      "training_step_time": 7.652775764465332
    },
    {
      "epoch": 6.7901611328125e-07,
      "model_forward_time": 1.723036527633667,
      "step": 712
    },
    {
      "epoch": 6.7901611328125e-07,
      "step": 712,
      "training_step_time": 8.629453420639038
    },
    {
      "epoch": 6.799697875976562e-07,
      "model_forward_time": 2.9419264793395996,
      "step": 713
    },
    {
      "epoch": 6.799697875976562e-07,
      "step": 713,
      "training_step_time": 7.780092477798462
    },
    {
      "epoch": 6.809234619140625e-07,
      "model_forward_time": 1.7221190929412842,
      "step": 714
    },
    {
      "epoch": 6.809234619140625e-07,
      "step": 714,
      "training_step_time": 7.313182830810547
    },
    {
      "epoch": 6.818771362304687e-07,
      "model_forward_time": 1.7271463871002197,
      "step": 715
    },
    {
      "epoch": 6.818771362304687e-07,
      "step": 715,
      "training_step_time": 8.203861951828003
    },
    {
      "epoch": 6.82830810546875e-07,
      "model_forward_time": 1.7404563426971436,
      "step": 716
    },
    {
      "epoch": 6.82830810546875e-07,
      "step": 716,
      "training_step_time": 8.56937551498413
    },
    {
      "epoch": 6.837844848632812e-07,
      "model_forward_time": 1.7208359241485596,
      "step": 717
    },
    {
      "epoch": 6.837844848632812e-07,
      "step": 717,
      "training_step_time": 10.402000904083252
    },
    {
      "epoch": 6.847381591796875e-07,
      "model_forward_time": 1.7328553199768066,
      "step": 718
    },
    {
      "epoch": 6.847381591796875e-07,
      "step": 718,
      "training_step_time": 8.391886234283447
    },
    {
      "epoch": 6.856918334960937e-07,
      "model_forward_time": 1.7316384315490723,
      "step": 719
    },
    {
      "epoch": 6.856918334960937e-07,
      "step": 719,
      "training_step_time": 8.424952030181885
    },
    {
      "epoch": 6.866455078125e-07,
      "grad_norm": 4.2549147605896,
      "learning_rate": 1.4380000000000003e-06,
      "loss": 0.1677,
      "step": 720
    },
    {
      "epoch": 6.866455078125e-07,
      "model_forward_time": 1.7366209030151367,
      "step": 720
    },
    {
      "dynamics_loss_avg": 0.08232584893703461,
      "epoch": 6.866455078125e-07,
      "step": 720
    },
    {
      "action_loss_avg": 0.07317008636891842,
      "epoch": 6.866455078125e-07,
      "step": 720
    },
    {
      "epoch": 6.866455078125e-07,
      "step": 720,
      "training_step_time": 7.729484558105469
    },
    {
      "epoch": 6.875991821289062e-07,
      "model_forward_time": 1.7249572277069092,
      "step": 721
    },
    {
      "epoch": 6.875991821289062e-07,
      "step": 721,
      "training_step_time": 8.482111692428589
    },
    {
      "epoch": 6.885528564453125e-07,
      "model_forward_time": 2.931358575820923,
      "step": 722
    },
    {
      "epoch": 6.885528564453125e-07,
      "step": 722,
      "training_step_time": 7.749956369400024
    },
    {
      "epoch": 6.895065307617187e-07,
      "model_forward_time": 1.7275471687316895,
      "step": 723
    },
    {
      "epoch": 6.895065307617187e-07,
      "step": 723,
      "training_step_time": 7.438521862030029
    },
    {
      "epoch": 6.90460205078125e-07,
      "model_forward_time": 1.7270870208740234,
      "step": 724
    },
    {
      "epoch": 6.90460205078125e-07,
      "step": 724,
      "training_step_time": 6.771076917648315
    },
    {
      "epoch": 6.914138793945312e-07,
      "model_forward_time": 1.7224626541137695,
      "step": 725
    },
    {
      "epoch": 6.914138793945312e-07,
      "step": 725,
      "training_step_time": 8.490796327590942
    },
    {
      "epoch": 6.923675537109375e-07,
      "model_forward_time": 1.7211415767669678,
      "step": 726
    },
    {
      "epoch": 6.923675537109375e-07,
      "step": 726,
      "training_step_time": 8.659234285354614
    },
    {
      "epoch": 6.933212280273437e-07,
      "model_forward_time": 1.7174193859100342,
      "step": 727
    },
    {
      "epoch": 6.933212280273437e-07,
      "step": 727,
      "training_step_time": 8.755615949630737
    },
    {
      "epoch": 6.9427490234375e-07,
      "model_forward_time": 1.7192268371582031,
      "step": 728
    },
    {
      "epoch": 6.9427490234375e-07,
      "step": 728,
      "training_step_time": 8.604788064956665
    },
    {
      "epoch": 6.952285766601562e-07,
      "model_forward_time": 1.740900993347168,
      "step": 729
    },
    {
      "epoch": 6.952285766601562e-07,
      "step": 729,
      "training_step_time": 9.72344708442688
    },
    {
      "epoch": 6.961822509765625e-07,
      "grad_norm": 4.728267192840576,
      "learning_rate": 1.4580000000000003e-06,
      "loss": 0.172,
      "step": 730
    },
    {
      "epoch": 6.961822509765625e-07,
      "model_forward_time": 1.724839687347412,
      "step": 730
    },
    {
      "dynamics_loss_avg": 0.0872889693826437,
      "epoch": 6.961822509765625e-07,
      "step": 730
    },
    {
      "action_loss_avg": 0.07267152592539787,
      "epoch": 6.961822509765625e-07,
      "step": 730
    },
    {
      "epoch": 6.961822509765625e-07,
      "step": 730,
      "training_step_time": 7.826940298080444
    },
    {
      "epoch": 6.971359252929687e-07,
      "model_forward_time": 2.9333343505859375,
      "step": 731
    },
    {
      "epoch": 6.971359252929687e-07,
      "step": 731,
      "training_step_time": 7.75358510017395
    },
    {
      "epoch": 6.98089599609375e-07,
      "model_forward_time": 1.7310755252838135,
      "step": 732
    },
    {
      "epoch": 6.98089599609375e-07,
      "step": 732,
      "training_step_time": 7.681265830993652
    },
    {
      "epoch": 6.990432739257812e-07,
      "model_forward_time": 1.723008632659912,
      "step": 733
    },
    {
      "epoch": 6.990432739257812e-07,
      "step": 733,
      "training_step_time": 7.303362846374512
    },
    {
      "epoch": 6.999969482421875e-07,
      "model_forward_time": 1.724125862121582,
      "step": 734
    },
    {
      "epoch": 6.999969482421875e-07,
      "step": 734,
      "training_step_time": 8.822530508041382
    },
    {
      "epoch": 7.009506225585937e-07,
      "model_forward_time": 1.7485120296478271,
      "step": 735
    },
    {
      "epoch": 7.009506225585937e-07,
      "step": 735,
      "training_step_time": 8.461051225662231
    },
    {
      "epoch": 7.01904296875e-07,
      "model_forward_time": 1.7277472019195557,
      "step": 736
    },
    {
      "epoch": 7.01904296875e-07,
      "step": 736,
      "training_step_time": 8.569167137145996
    },
    {
      "epoch": 7.028579711914062e-07,
      "model_forward_time": 1.7287395000457764,
      "step": 737
    },
    {
      "epoch": 7.028579711914062e-07,
      "step": 737,
      "training_step_time": 8.712619066238403
    },
    {
      "epoch": 7.038116455078125e-07,
      "model_forward_time": 1.7425315380096436,
      "step": 738
    },
    {
      "epoch": 7.038116455078125e-07,
      "step": 738,
      "training_step_time": 7.662738800048828
    },
    {
      "epoch": 7.047653198242187e-07,
      "model_forward_time": 1.7250022888183594,
      "step": 739
    },
    {
      "epoch": 7.047653198242187e-07,
      "step": 739,
      "training_step_time": 7.797189950942993
    },
    {
      "epoch": 7.05718994140625e-07,
      "grad_norm": 3.799577474594116,
      "learning_rate": 1.478e-06,
      "loss": 0.1677,
      "step": 740
    },
    {
      "epoch": 7.05718994140625e-07,
      "model_forward_time": 2.9082119464874268,
      "step": 740
    },
    {
      "dynamics_loss_avg": 0.0834349125623703,
      "epoch": 7.05718994140625e-07,
      "step": 740
    },
    {
      "action_loss_avg": 0.0798565313220024,
      "epoch": 7.05718994140625e-07,
      "step": 740
    },
    {
      "epoch": 7.05718994140625e-07,
      "step": 740,
      "training_step_time": 7.746774673461914
    },
    {
      "epoch": 7.066726684570312e-07,
      "model_forward_time": 1.7238593101501465,
      "step": 741
    },
    {
      "epoch": 7.066726684570312e-07,
      "step": 741,
      "training_step_time": 7.73736834526062
    },
    {
      "epoch": 7.076263427734375e-07,
      "model_forward_time": 1.7235989570617676,
      "step": 742
    },
    {
      "epoch": 7.076263427734375e-07,
      "step": 742,
      "training_step_time": 7.311419486999512
    },
    {
      "epoch": 7.085800170898437e-07,
      "model_forward_time": 1.724255084991455,
      "step": 743
    },
    {
      "epoch": 7.085800170898437e-07,
      "step": 743,
      "training_step_time": 8.734184503555298
    },
    {
      "epoch": 7.0953369140625e-07,
      "model_forward_time": 1.7487537860870361,
      "step": 744
    },
    {
      "epoch": 7.0953369140625e-07,
      "step": 744,
      "training_step_time": 7.560187578201294
    },
    {
      "epoch": 7.104873657226562e-07,
      "model_forward_time": 1.7249853610992432,
      "step": 745
    },
    {
      "epoch": 7.104873657226562e-07,
      "step": 745,
      "training_step_time": 8.330888986587524
    },
    {
      "epoch": 7.114410400390625e-07,
      "model_forward_time": 1.7298474311828613,
      "step": 746
    },
    {
      "epoch": 7.114410400390625e-07,
      "step": 746,
      "training_step_time": 7.908678293228149
    },
    {
      "epoch": 7.123947143554687e-07,
      "model_forward_time": 1.7544243335723877,
      "step": 747
    },
    {
      "epoch": 7.123947143554687e-07,
      "step": 747,
      "training_step_time": 9.303585290908813
    },
    {
      "epoch": 7.13348388671875e-07,
      "model_forward_time": 1.7190678119659424,
      "step": 748
    },
    {
      "epoch": 7.13348388671875e-07,
      "step": 748,
      "training_step_time": 7.796051025390625
    },
    {
      "epoch": 7.143020629882812e-07,
      "model_forward_time": 2.9008967876434326,
      "step": 749
    },
    {
      "epoch": 7.143020629882812e-07,
      "step": 749,
      "training_step_time": 7.750531196594238
    },
    {
      "epoch": 7.152557373046875e-07,
      "grad_norm": 2.002215623855591,
      "learning_rate": 1.498e-06,
      "loss": 0.1476,
      "step": 750
    },
    {
      "epoch": 7.152557373046875e-07,
      "model_forward_time": 1.7308905124664307,
      "step": 750
    },
    {
      "dynamics_loss_avg": 0.09138642866164445,
      "epoch": 7.152557373046875e-07,
      "step": 750
    },
    {
      "action_loss_avg": 0.06920404871925712,
      "epoch": 7.152557373046875e-07,
      "step": 750
    },
    {
      "epoch": 7.152557373046875e-07,
      "step": 750,
      "training_step_time": 7.782378911972046
    },
    {
      "epoch": 7.162094116210938e-07,
      "model_forward_time": 1.7295024394989014,
      "step": 751
    },
    {
      "epoch": 7.162094116210938e-07,
      "step": 751,
      "training_step_time": 7.579120635986328
    },
    {
      "epoch": 7.171630859375e-07,
      "model_forward_time": 1.7213623523712158,
      "step": 752
    },
    {
      "epoch": 7.171630859375e-07,
      "step": 752,
      "training_step_time": 7.680031776428223
    },
    {
      "epoch": 7.181167602539063e-07,
      "model_forward_time": 1.7422122955322266,
      "step": 753
    },
    {
      "epoch": 7.181167602539063e-07,
      "step": 753,
      "training_step_time": 9.909511089324951
    },
    {
      "epoch": 7.190704345703125e-07,
      "model_forward_time": 1.7188444137573242,
      "step": 754
    },
    {
      "epoch": 7.190704345703125e-07,
      "step": 754,
      "training_step_time": 7.809394598007202
    },
    {
      "epoch": 7.200241088867188e-07,
      "model_forward_time": 1.7246525287628174,
      "step": 755
    },
    {
      "epoch": 7.200241088867188e-07,
      "step": 755,
      "training_step_time": 8.639861345291138
    },
    {
      "epoch": 7.20977783203125e-07,
      "model_forward_time": 1.741579294204712,
      "step": 756
    },
    {
      "epoch": 7.20977783203125e-07,
      "step": 756,
      "training_step_time": 8.312880277633667
    },
    {
      "epoch": 7.219314575195313e-07,
      "model_forward_time": 1.730729341506958,
      "step": 757
    },
    {
      "epoch": 7.219314575195313e-07,
      "step": 757,
      "training_step_time": 7.741264343261719
    },
    {
      "epoch": 7.228851318359375e-07,
      "model_forward_time": 2.9077658653259277,
      "step": 758
    },
    {
      "epoch": 7.228851318359375e-07,
      "step": 758,
      "training_step_time": 7.7564966678619385
    },
    {
      "epoch": 7.238388061523438e-07,
      "model_forward_time": 1.7277612686157227,
      "step": 759
    },
    {
      "epoch": 7.238388061523438e-07,
      "step": 759,
      "training_step_time": 7.733920097351074
    },
    {
      "epoch": 7.2479248046875e-07,
      "grad_norm": 5.540458679199219,
      "learning_rate": 1.5180000000000001e-06,
      "loss": 0.1577,
      "step": 760
    },
    {
      "epoch": 7.2479248046875e-07,
      "model_forward_time": 1.7212259769439697,
      "step": 760
    },
    {
      "dynamics_loss_avg": 0.07599171437323093,
      "epoch": 7.2479248046875e-07,
      "step": 760
    },
    {
      "action_loss_avg": 0.057177900522947314,
      "epoch": 7.2479248046875e-07,
      "step": 760
    },
    {
      "epoch": 7.2479248046875e-07,
      "step": 760,
      "training_step_time": 7.559107065200806
    },
    {
      "epoch": 7.257461547851563e-07,
      "model_forward_time": 1.715343952178955,
      "step": 761
    },
    {
      "epoch": 7.257461547851563e-07,
      "step": 761,
      "training_step_time": 8.804036617279053
    },
    {
      "epoch": 7.266998291015625e-07,
      "model_forward_time": 1.7376422882080078,
      "step": 762
    },
    {
      "epoch": 7.266998291015625e-07,
      "step": 762,
      "training_step_time": 7.773665189743042
    },
    {
      "epoch": 7.276535034179688e-07,
      "model_forward_time": 1.7306606769561768,
      "step": 763
    },
    {
      "epoch": 7.276535034179688e-07,
      "step": 763,
      "training_step_time": 8.345402956008911
    },
    {
      "epoch": 7.28607177734375e-07,
      "model_forward_time": 1.7260842323303223,
      "step": 764
    },
    {
      "epoch": 7.28607177734375e-07,
      "step": 764,
      "training_step_time": 8.536304712295532
    },
    {
      "epoch": 7.295608520507813e-07,
      "model_forward_time": 1.7445242404937744,
      "step": 765
    },
    {
      "epoch": 7.295608520507813e-07,
      "step": 765,
      "training_step_time": 8.302820682525635
    },
    {
      "epoch": 7.305145263671875e-07,
      "model_forward_time": 1.7175102233886719,
      "step": 766
    },
    {
      "epoch": 7.305145263671875e-07,
      "step": 766,
      "training_step_time": 8.75619387626648
    },
    {
      "epoch": 7.314682006835938e-07,
      "model_forward_time": 2.929555892944336,
      "step": 767
    },
    {
      "epoch": 7.314682006835938e-07,
      "step": 767,
      "training_step_time": 7.744078874588013
    },
    {
      "epoch": 7.32421875e-07,
      "model_forward_time": 1.7242450714111328,
      "step": 768
    },
    {
      "epoch": 7.32421875e-07,
      "step": 768,
      "training_step_time": 7.7637619972229
    },
    {
      "epoch": 7.333755493164063e-07,
      "model_forward_time": 1.72676420211792,
      "step": 769
    },
    {
      "epoch": 7.333755493164063e-07,
      "step": 769,
      "training_step_time": 7.77192497253418
    },
    {
      "epoch": 7.343292236328125e-07,
      "grad_norm": 3.973489284515381,
      "learning_rate": 1.5380000000000001e-06,
      "loss": 0.1557,
      "step": 770
    },
    {
      "epoch": 7.343292236328125e-07,
      "model_forward_time": 1.730295181274414,
      "step": 770
    },
    {
      "dynamics_loss_avg": 0.08288074508309365,
      "epoch": 7.343292236328125e-07,
      "step": 770
    },
    {
      "action_loss_avg": 0.06600346667692066,
      "epoch": 7.343292236328125e-07,
      "step": 770
    },
    {
      "epoch": 7.343292236328125e-07,
      "step": 770,
      "training_step_time": 7.701236009597778
    },
    {
      "epoch": 7.352828979492188e-07,
      "model_forward_time": 1.7291285991668701,
      "step": 771
    },
    {
      "epoch": 7.352828979492188e-07,
      "step": 771,
      "training_step_time": 7.849156141281128
    },
    {
      "epoch": 7.36236572265625e-07,
      "model_forward_time": 1.7251358032226562,
      "step": 772
    },
    {
      "epoch": 7.36236572265625e-07,
      "step": 772,
      "training_step_time": 8.66262435913086
    },
    {
      "epoch": 7.371902465820313e-07,
      "model_forward_time": 1.7194204330444336,
      "step": 773
    },
    {
      "epoch": 7.371902465820313e-07,
      "step": 773,
      "training_step_time": 8.656986474990845
    },
    {
      "epoch": 7.381439208984375e-07,
      "model_forward_time": 1.7450997829437256,
      "step": 774
    },
    {
      "epoch": 7.381439208984375e-07,
      "step": 774,
      "training_step_time": 9.479766368865967
    },
    {
      "epoch": 7.390975952148438e-07,
      "model_forward_time": 1.7239809036254883,
      "step": 775
    },
    {
      "epoch": 7.390975952148438e-07,
      "step": 775,
      "training_step_time": 8.78959608078003
    },
    {
      "epoch": 7.4005126953125e-07,
      "model_forward_time": 2.9328513145446777,
      "step": 776
    },
    {
      "epoch": 7.4005126953125e-07,
      "step": 776,
      "training_step_time": 7.783502101898193
    },
    {
      "epoch": 7.410049438476563e-07,
      "model_forward_time": 1.725344181060791,
      "step": 777
    },
    {
      "epoch": 7.410049438476563e-07,
      "step": 777,
      "training_step_time": 7.733405113220215
    },
    {
      "epoch": 7.419586181640625e-07,
      "model_forward_time": 1.7246880531311035,
      "step": 778
    },
    {
      "epoch": 7.419586181640625e-07,
      "step": 778,
      "training_step_time": 7.740410327911377
    },
    {
      "epoch": 7.429122924804688e-07,
      "model_forward_time": 1.7425413131713867,
      "step": 779
    },
    {
      "epoch": 7.429122924804688e-07,
      "step": 779,
      "training_step_time": 8.35841178894043
    },
    {
      "epoch": 7.43865966796875e-07,
      "grad_norm": 4.211995601654053,
      "learning_rate": 1.5580000000000002e-06,
      "loss": 0.1471,
      "step": 780
    },
    {
      "epoch": 7.43865966796875e-07,
      "model_forward_time": 1.7239489555358887,
      "step": 780
    },
    {
      "dynamics_loss_avg": 0.08257119804620743,
      "epoch": 7.43865966796875e-07,
      "step": 780
    },
    {
      "action_loss_avg": 0.06722223460674286,
      "epoch": 7.43865966796875e-07,
      "step": 780
    },
    {
      "epoch": 7.43865966796875e-07,
      "step": 780,
      "training_step_time": 7.779482364654541
    },
    {
      "epoch": 7.448196411132813e-07,
      "model_forward_time": 1.7221391201019287,
      "step": 781
    },
    {
      "epoch": 7.448196411132813e-07,
      "step": 781,
      "training_step_time": 8.239349603652954
    },
    {
      "epoch": 7.457733154296875e-07,
      "model_forward_time": 1.7249503135681152,
      "step": 782
    },
    {
      "epoch": 7.457733154296875e-07,
      "step": 782,
      "training_step_time": 9.622958898544312
    },
    {
      "epoch": 7.467269897460938e-07,
      "model_forward_time": 1.7414679527282715,
      "step": 783
    },
    {
      "epoch": 7.467269897460938e-07,
      "step": 783,
      "training_step_time": 7.895970821380615
    },
    {
      "epoch": 7.476806640625e-07,
      "model_forward_time": 1.7327322959899902,
      "step": 784
    },
    {
      "epoch": 7.476806640625e-07,
      "step": 784,
      "training_step_time": 8.874228715896606
    },
    {
      "epoch": 7.486343383789063e-07,
      "model_forward_time": 2.9725522994995117,
      "step": 785
    },
    {
      "epoch": 7.486343383789063e-07,
      "step": 785,
      "training_step_time": 8.86412525177002
    },
    {
      "epoch": 7.495880126953125e-07,
      "model_forward_time": 1.725595474243164,
      "step": 786
    },
    {
      "epoch": 7.495880126953125e-07,
      "step": 786,
      "training_step_time": 7.726297616958618
    },
    {
      "epoch": 7.505416870117188e-07,
      "model_forward_time": 1.7214083671569824,
      "step": 787
    },
    {
      "epoch": 7.505416870117188e-07,
      "step": 787,
      "training_step_time": 7.811821222305298
    },
    {
      "epoch": 7.51495361328125e-07,
      "model_forward_time": 1.7211806774139404,
      "step": 788
    },
    {
      "epoch": 7.51495361328125e-07,
      "step": 788,
      "training_step_time": 7.477633714675903
    },
    {
      "epoch": 7.524490356445313e-07,
      "model_forward_time": 1.7255191802978516,
      "step": 789
    },
    {
      "epoch": 7.524490356445313e-07,
      "step": 789,
      "training_step_time": 7.7753822803497314
    },
    {
      "epoch": 7.534027099609375e-07,
      "grad_norm": 4.010906219482422,
      "learning_rate": 1.5780000000000002e-06,
      "loss": 0.1355,
      "step": 790
    },
    {
      "epoch": 7.534027099609375e-07,
      "model_forward_time": 1.7593204975128174,
      "step": 790
    },
    {
      "dynamics_loss_avg": 0.06886830795556306,
      "epoch": 7.534027099609375e-07,
      "step": 790
    },
    {
      "action_loss_avg": 0.05609360877424478,
      "epoch": 7.534027099609375e-07,
      "step": 790
    },
    {
      "epoch": 7.534027099609375e-07,
      "step": 790,
      "training_step_time": 7.746504306793213
    },
    {
      "epoch": 7.543563842773438e-07,
      "model_forward_time": 1.722376823425293,
      "step": 791
    },
    {
      "epoch": 7.543563842773438e-07,
      "step": 791,
      "training_step_time": 8.881154775619507
    },
    {
      "epoch": 7.5531005859375e-07,
      "model_forward_time": 1.738861083984375,
      "step": 792
    },
    {
      "epoch": 7.5531005859375e-07,
      "step": 792,
      "training_step_time": 7.902942419052124
    },
    {
      "epoch": 7.562637329101563e-07,
      "model_forward_time": 1.725327968597412,
      "step": 793
    },
    {
      "epoch": 7.562637329101563e-07,
      "step": 793,
      "training_step_time": 7.81976318359375
    },
    {
      "epoch": 7.572174072265625e-07,
      "model_forward_time": 2.922410011291504,
      "step": 794
    },
    {
      "epoch": 7.572174072265625e-07,
      "step": 794,
      "training_step_time": 7.820811748504639
    },
    {
      "epoch": 7.581710815429688e-07,
      "model_forward_time": 1.7279574871063232,
      "step": 795
    },
    {
      "epoch": 7.581710815429688e-07,
      "step": 795,
      "training_step_time": 7.7394492626190186
    },
    {
      "epoch": 7.59124755859375e-07,
      "model_forward_time": 1.0258071422576904,
      "step": 796
    },
    {
      "epoch": 7.59124755859375e-07,
      "step": 796,
      "training_step_time": 7.808601140975952
    },
    {
      "epoch": 7.600784301757813e-07,
      "model_forward_time": 1.727334976196289,
      "step": 797
    },
    {
      "epoch": 7.600784301757813e-07,
      "step": 797,
      "training_step_time": 7.811727046966553
    },
    {
      "epoch": 7.610321044921875e-07,
      "model_forward_time": 1.7285945415496826,
      "step": 798
    },
    {
      "epoch": 7.610321044921875e-07,
      "step": 798,
      "training_step_time": 7.806084156036377
    },
    {
      "epoch": 7.619857788085938e-07,
      "model_forward_time": 1.7487308979034424,
      "step": 799
    },
    {
      "epoch": 7.619857788085938e-07,
      "step": 799,
      "training_step_time": 7.717027187347412
    },
    {
      "epoch": 7.62939453125e-07,
      "grad_norm": 2.9160826206207275,
      "learning_rate": 1.5980000000000002e-06,
      "loss": 0.1617,
      "step": 800
    },
    {
      "epoch": 7.62939453125e-07,
      "model_forward_time": 1.7212932109832764,
      "step": 800
    },
    {
      "dynamics_loss_avg": 0.07431471571326256,
      "epoch": 7.62939453125e-07,
      "step": 800
    },
    {
      "action_loss_avg": 0.061739105731248856,
      "epoch": 7.62939453125e-07,
      "step": 800
    },
    {
      "epoch": 7.62939453125e-07,
      "step": 800,
      "training_step_time": 9.967210531234741
    },
    {
      "epoch": 7.638931274414063e-07,
      "model_forward_time": 1.7366504669189453,
      "step": 801
    },
    {
      "epoch": 7.638931274414063e-07,
      "step": 801,
      "training_step_time": 8.495183944702148
    },
    {
      "epoch": 7.648468017578125e-07,
      "model_forward_time": 1.727628231048584,
      "step": 802
    },
    {
      "epoch": 7.648468017578125e-07,
      "step": 802,
      "training_step_time": 8.511553525924683
    },
    {
      "epoch": 7.658004760742188e-07,
      "model_forward_time": 2.180454730987549,
      "step": 803
    },
    {
      "epoch": 7.658004760742188e-07,
      "step": 803,
      "training_step_time": 7.691481590270996
    },
    {
      "epoch": 7.66754150390625e-07,
      "model_forward_time": 1.7369048595428467,
      "step": 804
    },
    {
      "epoch": 7.66754150390625e-07,
      "step": 804,
      "training_step_time": 7.880978345870972
    },
    {
      "epoch": 7.677078247070313e-07,
      "model_forward_time": 1.726938009262085,
      "step": 805
    },
    {
      "epoch": 7.677078247070313e-07,
      "step": 805,
      "training_step_time": 8.37354850769043
    },
    {
      "epoch": 7.686614990234375e-07,
      "model_forward_time": 1.7200720310211182,
      "step": 806
    },
    {
      "epoch": 7.686614990234375e-07,
      "step": 806,
      "training_step_time": 7.8058922290802
    },
    {
      "epoch": 7.696151733398438e-07,
      "model_forward_time": 1.7304012775421143,
      "step": 807
    },
    {
      "epoch": 7.696151733398438e-07,
      "step": 807,
      "training_step_time": 7.799496650695801
    },
    {
      "epoch": 7.7056884765625e-07,
      "model_forward_time": 1.761277675628662,
      "step": 808
    },
    {
      "epoch": 7.7056884765625e-07,
      "step": 808,
      "training_step_time": 7.777338743209839
    },
    {
      "epoch": 7.715225219726563e-07,
      "model_forward_time": 1.7216572761535645,
      "step": 809
    },
    {
      "epoch": 7.715225219726563e-07,
      "step": 809,
      "training_step_time": 8.83825159072876
    },
    {
      "epoch": 7.724761962890626e-07,
      "grad_norm": 3.191779136657715,
      "learning_rate": 1.6180000000000002e-06,
      "loss": 0.1535,
      "step": 810
    },
    {
      "epoch": 7.724761962890626e-07,
      "model_forward_time": 1.736473798751831,
      "step": 810
    },
    {
      "dynamics_loss_avg": 0.09756693802773952,
      "epoch": 7.724761962890626e-07,
      "step": 810
    },
    {
      "action_loss_avg": 0.08639632975682617,
      "epoch": 7.724761962890626e-07,
      "step": 810
    },
    {
      "epoch": 7.724761962890626e-07,
      "step": 810,
      "training_step_time": 7.824240207672119
    },
    {
      "epoch": 7.734298706054688e-07,
      "model_forward_time": 1.7301998138427734,
      "step": 811
    },
    {
      "epoch": 7.734298706054688e-07,
      "step": 811,
      "training_step_time": 7.710338115692139
    },
    {
      "epoch": 7.74383544921875e-07,
      "model_forward_time": 2.801668643951416,
      "step": 812
    },
    {
      "epoch": 7.74383544921875e-07,
      "step": 812,
      "training_step_time": 8.600331544876099
    },
    {
      "epoch": 7.753372192382812e-07,
      "model_forward_time": 1.7287909984588623,
      "step": 813
    },
    {
      "epoch": 7.753372192382812e-07,
      "step": 813,
      "training_step_time": 7.716078996658325
    },
    {
      "epoch": 7.762908935546874e-07,
      "model_forward_time": 1.73587965965271,
      "step": 814
    },
    {
      "epoch": 7.762908935546874e-07,
      "step": 814,
      "training_step_time": 7.661688566207886
    },
    {
      "epoch": 7.772445678710937e-07,
      "model_forward_time": 1.7217159271240234,
      "step": 815
    },
    {
      "epoch": 7.772445678710937e-07,
      "step": 815,
      "training_step_time": 7.741850852966309
    },
    {
      "epoch": 7.781982421875e-07,
      "model_forward_time": 1.7291579246520996,
      "step": 816
    },
    {
      "epoch": 7.781982421875e-07,
      "step": 816,
      "training_step_time": 7.625367641448975
    },
    {
      "epoch": 7.791519165039062e-07,
      "model_forward_time": 1.7369842529296875,
      "step": 817
    },
    {
      "epoch": 7.791519165039062e-07,
      "step": 817,
      "training_step_time": 8.44495177268982
    },
    {
      "epoch": 7.801055908203125e-07,
      "model_forward_time": 1.7241768836975098,
      "step": 818
    },
    {
      "epoch": 7.801055908203125e-07,
      "step": 818,
      "training_step_time": 7.818295955657959
    },
    {
      "epoch": 7.810592651367187e-07,
      "model_forward_time": 1.746821641921997,
      "step": 819
    },
    {
      "epoch": 7.810592651367187e-07,
      "step": 819,
      "training_step_time": 8.70760440826416
    },
    {
      "epoch": 7.82012939453125e-07,
      "grad_norm": 3.5267527103424072,
      "learning_rate": 1.6380000000000002e-06,
      "loss": 0.1563,
      "step": 820
    },
    {
      "epoch": 7.82012939453125e-07,
      "model_forward_time": 1.7173123359680176,
      "step": 820
    },
    {
      "dynamics_loss_avg": 0.0999838300049305,
      "epoch": 7.82012939453125e-07,
      "step": 820
    },
    {
      "action_loss_avg": 0.0677394350990653,
      "epoch": 7.82012939453125e-07,
      "step": 820
    },
    {
      "epoch": 7.82012939453125e-07,
      "step": 820,
      "training_step_time": 8.63644003868103
    },
    {
      "epoch": 7.829666137695312e-07,
      "model_forward_time": 1.7246925830841064,
      "step": 821
    },
    {
      "epoch": 7.829666137695312e-07,
      "step": 821,
      "training_step_time": 9.196517705917358
    },
    {
      "epoch": 7.839202880859375e-07,
      "model_forward_time": 2.5255677700042725,
      "step": 822
    },
    {
      "epoch": 7.839202880859375e-07,
      "step": 822,
      "training_step_time": 8.701982498168945
    },
    {
      "epoch": 7.848739624023437e-07,
      "model_forward_time": 1.725944995880127,
      "step": 823
    },
    {
      "epoch": 7.848739624023437e-07,
      "step": 823,
      "training_step_time": 7.672717571258545
    },
    {
      "epoch": 7.8582763671875e-07,
      "model_forward_time": 1.7204127311706543,
      "step": 824
    },
    {
      "epoch": 7.8582763671875e-07,
      "step": 824,
      "training_step_time": 7.718768119812012
    },
    {
      "epoch": 7.867813110351562e-07,
      "model_forward_time": 1.7209081649780273,
      "step": 825
    },
    {
      "epoch": 7.867813110351562e-07,
      "step": 825,
      "training_step_time": 8.18721079826355
    },
    {
      "epoch": 7.877349853515625e-07,
      "model_forward_time": 1.7299747467041016,
      "step": 826
    },
    {
      "epoch": 7.877349853515625e-07,
      "step": 826,
      "training_step_time": 8.699161291122437
    },
    {
      "epoch": 7.886886596679687e-07,
      "model_forward_time": 1.7233076095581055,
      "step": 827
    },
    {
      "epoch": 7.886886596679687e-07,
      "step": 827,
      "training_step_time": 7.838068246841431
    },
    {
      "epoch": 7.89642333984375e-07,
      "model_forward_time": 1.7290019989013672,
      "step": 828
    },
    {
      "epoch": 7.89642333984375e-07,
      "step": 828,
      "training_step_time": 7.6683995723724365
    },
    {
      "epoch": 7.905960083007812e-07,
      "model_forward_time": 1.7304635047912598,
      "step": 829
    },
    {
      "epoch": 7.905960083007812e-07,
      "step": 829,
      "training_step_time": 9.389987707138062
    },
    {
      "epoch": 7.915496826171875e-07,
      "grad_norm": 3.8808183670043945,
      "learning_rate": 1.6580000000000003e-06,
      "loss": 0.1347,
      "step": 830
    },
    {
      "epoch": 7.915496826171875e-07,
      "model_forward_time": 1.7269272804260254,
      "step": 830
    },
    {
      "dynamics_loss_avg": 0.0753917682915926,
      "epoch": 7.915496826171875e-07,
      "step": 830
    },
    {
      "action_loss_avg": 0.05854531414806843,
      "epoch": 7.915496826171875e-07,
      "step": 830
    },
    {
      "epoch": 7.915496826171875e-07,
      "step": 830,
      "training_step_time": 9.155998706817627
    },
    {
      "epoch": 7.925033569335937e-07,
      "model_forward_time": 2.873837471008301,
      "step": 831
    },
    {
      "epoch": 7.925033569335937e-07,
      "step": 831,
      "training_step_time": 7.7013773918151855
    },
    {
      "epoch": 7.9345703125e-07,
      "model_forward_time": 1.7176084518432617,
      "step": 832
    },
    {
      "epoch": 7.9345703125e-07,
      "step": 832,
      "training_step_time": 7.735933065414429
    },
    {
      "epoch": 7.944107055664062e-07,
      "model_forward_time": 1.7194819450378418,
      "step": 833
    },
    {
      "epoch": 7.944107055664062e-07,
      "step": 833,
      "training_step_time": 7.6629931926727295
    },
    {
      "epoch": 7.953643798828125e-07,
      "model_forward_time": 1.7131848335266113,
      "step": 834
    },
    {
      "epoch": 7.953643798828125e-07,
      "step": 834,
      "training_step_time": 7.756222724914551
    },
    {
      "epoch": 7.963180541992187e-07,
      "model_forward_time": 1.719834804534912,
      "step": 835
    },
    {
      "epoch": 7.963180541992187e-07,
      "step": 835,
      "training_step_time": 7.939650774002075
    },
    {
      "epoch": 7.97271728515625e-07,
      "model_forward_time": 1.7318286895751953,
      "step": 836
    },
    {
      "epoch": 7.97271728515625e-07,
      "step": 836,
      "training_step_time": 7.764951705932617
    },
    {
      "epoch": 7.982254028320312e-07,
      "model_forward_time": 1.7127785682678223,
      "step": 837
    },
    {
      "epoch": 7.982254028320312e-07,
      "step": 837,
      "training_step_time": 9.756340742111206
    },
    {
      "epoch": 7.991790771484375e-07,
      "model_forward_time": 1.7362749576568604,
      "step": 838
    },
    {
      "epoch": 7.991790771484375e-07,
      "step": 838,
      "training_step_time": 7.7958433628082275
    },
    {
      "epoch": 8.001327514648437e-07,
      "model_forward_time": 1.7117547988891602,
      "step": 839
    },
    {
      "epoch": 8.001327514648437e-07,
      "step": 839,
      "training_step_time": 8.238409757614136
    },
    {
      "epoch": 8.0108642578125e-07,
      "grad_norm": 3.7012252807617188,
      "learning_rate": 1.6780000000000003e-06,
      "loss": 0.1392,
      "step": 840
    },
    {
      "epoch": 8.0108642578125e-07,
      "model_forward_time": 2.8733227252960205,
      "step": 840
    },
    {
      "dynamics_loss_avg": 0.07516168858855962,
      "epoch": 8.0108642578125e-07,
      "step": 840
    },
    {
      "action_loss_avg": 0.062009716499596836,
      "epoch": 8.0108642578125e-07,
      "step": 840
    },
    {
      "epoch": 8.0108642578125e-07,
      "step": 840,
      "training_step_time": 8.73236608505249
    },
    {
      "epoch": 8.020401000976562e-07,
      "model_forward_time": 1.7162623405456543,
      "step": 841
    },
    {
      "epoch": 8.020401000976562e-07,
      "step": 841,
      "training_step_time": 8.604928970336914
    },
    {
      "epoch": 8.029937744140625e-07,
      "model_forward_time": 1.7154426574707031,
      "step": 842
    },
    {
      "epoch": 8.029937744140625e-07,
      "step": 842,
      "training_step_time": 7.743588209152222
    },
    {
      "epoch": 8.039474487304687e-07,
      "model_forward_time": 1.7205708026885986,
      "step": 843
    },
    {
      "epoch": 8.039474487304687e-07,
      "step": 843,
      "training_step_time": 7.7921247482299805
    },
    {
      "epoch": 8.04901123046875e-07,
      "model_forward_time": 1.7198905944824219,
      "step": 844
    },
    {
      "epoch": 8.04901123046875e-07,
      "step": 844,
      "training_step_time": 7.949703931808472
    },
    {
      "epoch": 8.058547973632812e-07,
      "model_forward_time": 1.7383062839508057,
      "step": 845
    },
    {
      "epoch": 8.058547973632812e-07,
      "step": 845,
      "training_step_time": 7.684479236602783
    },
    {
      "epoch": 8.068084716796875e-07,
      "model_forward_time": 1.7198717594146729,
      "step": 846
    },
    {
      "epoch": 8.068084716796875e-07,
      "step": 846,
      "training_step_time": 7.6690356731414795
    },
    {
      "epoch": 8.077621459960937e-07,
      "model_forward_time": 1.7338988780975342,
      "step": 847
    },
    {
      "epoch": 8.077621459960937e-07,
      "step": 847,
      "training_step_time": 7.825591325759888
    },
    {
      "epoch": 8.087158203125e-07,
      "model_forward_time": 1.715538501739502,
      "step": 848
    },
    {
      "epoch": 8.087158203125e-07,
      "step": 848,
      "training_step_time": 7.684153318405151
    },
    {
      "epoch": 8.096694946289062e-07,
      "model_forward_time": 2.829003095626831,
      "step": 849
    },
    {
      "epoch": 8.096694946289062e-07,
      "step": 849,
      "training_step_time": 7.7397496700286865
    },
    {
      "epoch": 8.106231689453125e-07,
      "grad_norm": 1.9545193910598755,
      "learning_rate": 1.6980000000000003e-06,
      "loss": 0.1405,
      "step": 850
    },
    {
      "epoch": 8.106231689453125e-07,
      "model_forward_time": 1.7209081649780273,
      "step": 850
    },
    {
      "dynamics_loss_avg": 0.08181364350020885,
      "epoch": 8.106231689453125e-07,
      "step": 850
    },
    {
      "action_loss_avg": 0.05539990002289415,
      "epoch": 8.106231689453125e-07,
      "step": 850
    },
    {
      "epoch": 8.106231689453125e-07,
      "step": 850,
      "training_step_time": 7.700869083404541
    },
    {
      "epoch": 8.115768432617187e-07,
      "model_forward_time": 1.7158253192901611,
      "step": 851
    },
    {
      "epoch": 8.115768432617187e-07,
      "step": 851,
      "training_step_time": 7.740008354187012
    },
    {
      "epoch": 8.12530517578125e-07,
      "model_forward_time": 1.7170922756195068,
      "step": 852
    },
    {
      "epoch": 8.12530517578125e-07,
      "step": 852,
      "training_step_time": 7.803992033004761
    },
    {
      "epoch": 8.134841918945312e-07,
      "model_forward_time": 1.7190608978271484,
      "step": 853
    },
    {
      "epoch": 8.134841918945312e-07,
      "step": 853,
      "training_step_time": 7.715133905410767
    },
    {
      "epoch": 8.144378662109375e-07,
      "model_forward_time": 1.7644178867340088,
      "step": 854
    },
    {
      "epoch": 8.144378662109375e-07,
      "step": 854,
      "training_step_time": 7.969412326812744
    },
    {
      "epoch": 8.153915405273437e-07,
      "model_forward_time": 1.7096915245056152,
      "step": 855
    },
    {
      "epoch": 8.153915405273437e-07,
      "step": 855,
      "training_step_time": 8.216439485549927
    },
    {
      "epoch": 8.1634521484375e-07,
      "model_forward_time": 1.7203607559204102,
      "step": 856
    },
    {
      "epoch": 8.1634521484375e-07,
      "step": 856,
      "training_step_time": 7.904008150100708
    },
    {
      "epoch": 8.172988891601562e-07,
      "model_forward_time": 1.3462355136871338,
      "step": 857
    },
    {
      "epoch": 8.172988891601562e-07,
      "step": 857,
      "training_step_time": 7.6531853675842285
    },
    {
      "epoch": 8.182525634765625e-07,
      "model_forward_time": 1.7135632038116455,
      "step": 858
    },
    {
      "epoch": 8.182525634765625e-07,
      "step": 858,
      "training_step_time": 7.649683713912964
    },
    {
      "epoch": 8.192062377929687e-07,
      "model_forward_time": 2.89164662361145,
      "step": 859
    },
    {
      "epoch": 8.192062377929687e-07,
      "step": 859,
      "training_step_time": 7.723957777023315
    },
    {
      "epoch": 8.20159912109375e-07,
      "grad_norm": 2.881168842315674,
      "learning_rate": 1.7180000000000003e-06,
      "loss": 0.1299,
      "step": 860
    },
    {
      "epoch": 8.20159912109375e-07,
      "model_forward_time": 1.7138020992279053,
      "step": 860
    },
    {
      "dynamics_loss_avg": 0.08283034674823284,
      "epoch": 8.20159912109375e-07,
      "step": 860
    },
    {
      "action_loss_avg": 0.057131876982748506,
      "epoch": 8.20159912109375e-07,
      "step": 860
    },
    {
      "epoch": 8.20159912109375e-07,
      "step": 860,
      "training_step_time": 7.684075593948364
    },
    {
      "epoch": 8.211135864257812e-07,
      "model_forward_time": 1.7133407592773438,
      "step": 861
    },
    {
      "epoch": 8.211135864257812e-07,
      "step": 861,
      "training_step_time": 8.557006359100342
    },
    {
      "epoch": 8.220672607421875e-07,
      "model_forward_time": 1.7158753871917725,
      "step": 862
    },
    {
      "epoch": 8.220672607421875e-07,
      "step": 862,
      "training_step_time": 8.600617408752441
    },
    {
      "epoch": 8.230209350585937e-07,
      "model_forward_time": 1.721825122833252,
      "step": 863
    },
    {
      "epoch": 8.230209350585937e-07,
      "step": 863,
      "training_step_time": 7.910278081893921
    },
    {
      "epoch": 8.23974609375e-07,
      "model_forward_time": 1.7055225372314453,
      "step": 864
    },
    {
      "epoch": 8.23974609375e-07,
      "step": 864,
      "training_step_time": 8.294506072998047
    },
    {
      "epoch": 8.249282836914062e-07,
      "model_forward_time": 1.71882963180542,
      "step": 865
    },
    {
      "epoch": 8.249282836914062e-07,
      "step": 865,
      "training_step_time": 8.575125455856323
    },
    {
      "epoch": 8.258819580078125e-07,
      "model_forward_time": 1.7465505599975586,
      "step": 866
    },
    {
      "epoch": 8.258819580078125e-07,
      "step": 866,
      "training_step_time": 7.5307135581970215
    },
    {
      "epoch": 8.268356323242187e-07,
      "model_forward_time": 1.7171742916107178,
      "step": 867
    },
    {
      "epoch": 8.268356323242187e-07,
      "step": 867,
      "training_step_time": 8.356821775436401
    },
    {
      "epoch": 8.27789306640625e-07,
      "model_forward_time": 2.8396856784820557,
      "step": 868
    },
    {
      "epoch": 8.27789306640625e-07,
      "step": 868,
      "training_step_time": 7.664866209030151
    },
    {
      "epoch": 8.287429809570312e-07,
      "model_forward_time": 1.7075066566467285,
      "step": 869
    },
    {
      "epoch": 8.287429809570312e-07,
      "step": 869,
      "training_step_time": 7.682347774505615
    },
    {
      "epoch": 8.296966552734375e-07,
      "grad_norm": 1.823087215423584,
      "learning_rate": 1.7380000000000003e-06,
      "loss": 0.1349,
      "step": 870
    },
    {
      "epoch": 8.296966552734375e-07,
      "model_forward_time": 1.7093160152435303,
      "step": 870
    },
    {
      "dynamics_loss_avg": 0.07603428661823272,
      "epoch": 8.296966552734375e-07,
      "step": 870
    },
    {
      "action_loss_avg": 0.05204085311852395,
      "epoch": 8.296966552734375e-07,
      "step": 870
    },
    {
      "epoch": 8.296966552734375e-07,
      "step": 870,
      "training_step_time": 7.572297811508179
    },
    {
      "epoch": 8.306503295898437e-07,
      "model_forward_time": 1.7137646675109863,
      "step": 871
    },
    {
      "epoch": 8.306503295898437e-07,
      "step": 871,
      "training_step_time": 8.364821195602417
    },
    {
      "epoch": 8.3160400390625e-07,
      "model_forward_time": 1.7352066040039062,
      "step": 872
    },
    {
      "epoch": 8.3160400390625e-07,
      "step": 872,
      "training_step_time": 7.911390542984009
    },
    {
      "epoch": 8.325576782226562e-07,
      "model_forward_time": 1.712970495223999,
      "step": 873
    },
    {
      "epoch": 8.325576782226562e-07,
      "step": 873,
      "training_step_time": 9.893961429595947
    },
    {
      "epoch": 8.335113525390625e-07,
      "model_forward_time": 1.717026710510254,
      "step": 874
    },
    {
      "epoch": 8.335113525390625e-07,
      "step": 874,
      "training_step_time": 8.678845405578613
    },
    {
      "epoch": 8.344650268554688e-07,
      "model_forward_time": 1.7399744987487793,
      "step": 875
    },
    {
      "epoch": 8.344650268554688e-07,
      "step": 875,
      "training_step_time": 8.701780080795288
    },
    {
      "epoch": 8.35418701171875e-07,
      "model_forward_time": 1.720628261566162,
      "step": 876
    },
    {
      "epoch": 8.35418701171875e-07,
      "step": 876,
      "training_step_time": 7.668762683868408
    },
    {
      "epoch": 8.363723754882813e-07,
      "model_forward_time": 2.837111234664917,
      "step": 877
    },
    {
      "epoch": 8.363723754882813e-07,
      "step": 877,
      "training_step_time": 7.686865329742432
    },
    {
      "epoch": 8.373260498046875e-07,
      "model_forward_time": 1.714839220046997,
      "step": 878
    },
    {
      "epoch": 8.373260498046875e-07,
      "step": 878,
      "training_step_time": 8.747641324996948
    },
    {
      "epoch": 8.382797241210938e-07,
      "model_forward_time": 1.7123522758483887,
      "step": 879
    },
    {
      "epoch": 8.382797241210938e-07,
      "step": 879,
      "training_step_time": 7.633001089096069
    },
    {
      "epoch": 8.392333984375e-07,
      "grad_norm": 2.2337539196014404,
      "learning_rate": 1.7580000000000004e-06,
      "loss": 0.1311,
      "step": 880
    },
    {
      "epoch": 8.392333984375e-07,
      "model_forward_time": 1.7153139114379883,
      "step": 880
    },
    {
      "dynamics_loss_avg": 0.07373355105519294,
      "epoch": 8.392333984375e-07,
      "step": 880
    },
    {
      "action_loss_avg": 0.046897981129586694,
      "epoch": 8.392333984375e-07,
      "step": 880
    },
    {
      "epoch": 8.392333984375e-07,
      "step": 880,
      "training_step_time": 7.778600215911865
    },
    {
      "epoch": 8.401870727539063e-07,
      "model_forward_time": 1.7097346782684326,
      "step": 881
    },
    {
      "epoch": 8.401870727539063e-07,
      "step": 881,
      "training_step_time": 7.9266297817230225
    },
    {
      "epoch": 8.411407470703125e-07,
      "model_forward_time": 1.7451200485229492,
      "step": 882
    },
    {
      "epoch": 8.411407470703125e-07,
      "step": 882,
      "training_step_time": 8.266149520874023
    },
    {
      "epoch": 8.420944213867188e-07,
      "model_forward_time": 1.3396172523498535,
      "step": 883
    },
    {
      "epoch": 8.420944213867188e-07,
      "step": 883,
      "training_step_time": 8.55782961845398
    },
    {
      "epoch": 8.43048095703125e-07,
      "model_forward_time": 1.7336905002593994,
      "step": 884
    },
    {
      "epoch": 8.43048095703125e-07,
      "step": 884,
      "training_step_time": 7.929900407791138
    },
    {
      "epoch": 8.440017700195313e-07,
      "model_forward_time": 1.714428424835205,
      "step": 885
    },
    {
      "epoch": 8.440017700195313e-07,
      "step": 885,
      "training_step_time": 8.36234450340271
    },
    {
      "epoch": 8.449554443359375e-07,
      "model_forward_time": 2.799255847930908,
      "step": 886
    },
    {
      "epoch": 8.449554443359375e-07,
      "step": 886,
      "training_step_time": 8.380393266677856
    },
    {
      "epoch": 8.459091186523438e-07,
      "model_forward_time": 1.714855670928955,
      "step": 887
    },
    {
      "epoch": 8.459091186523438e-07,
      "step": 887,
      "training_step_time": 7.545702695846558
    },
    {
      "epoch": 8.4686279296875e-07,
      "model_forward_time": 1.7172293663024902,
      "step": 888
    },
    {
      "epoch": 8.4686279296875e-07,
      "step": 888,
      "training_step_time": 7.691821813583374
    },
    {
      "epoch": 8.478164672851563e-07,
      "model_forward_time": 1.7230424880981445,
      "step": 889
    },
    {
      "epoch": 8.478164672851563e-07,
      "step": 889,
      "training_step_time": 8.764364957809448
    },
    {
      "epoch": 8.487701416015625e-07,
      "grad_norm": 2.9033374786376953,
      "learning_rate": 1.7780000000000004e-06,
      "loss": 0.1417,
      "step": 890
    },
    {
      "epoch": 8.487701416015625e-07,
      "model_forward_time": 1.7132766246795654,
      "step": 890
    },
    {
      "dynamics_loss_avg": 0.07572070881724358,
      "epoch": 8.487701416015625e-07,
      "step": 890
    },
    {
      "action_loss_avg": 0.04467863226309419,
      "epoch": 8.487701416015625e-07,
      "step": 890
    },
    {
      "epoch": 8.487701416015625e-07,
      "step": 890,
      "training_step_time": 7.910924434661865
    },
    {
      "epoch": 8.497238159179688e-07,
      "model_forward_time": 1.7393746376037598,
      "step": 891
    },
    {
      "epoch": 8.497238159179688e-07,
      "step": 891,
      "training_step_time": 7.822301864624023
    },
    {
      "epoch": 8.50677490234375e-07,
      "model_forward_time": 1.7156789302825928,
      "step": 892
    },
    {
      "epoch": 8.50677490234375e-07,
      "step": 892,
      "training_step_time": 7.687561988830566
    },
    {
      "epoch": 8.516311645507813e-07,
      "model_forward_time": 1.725527048110962,
      "step": 893
    },
    {
      "epoch": 8.516311645507813e-07,
      "step": 893,
      "training_step_time": 7.86064600944519
    },
    {
      "epoch": 8.525848388671875e-07,
      "model_forward_time": 1.7141788005828857,
      "step": 894
    },
    {
      "epoch": 8.525848388671875e-07,
      "step": 894,
      "training_step_time": 8.735146045684814
    },
    {
      "epoch": 8.535385131835938e-07,
      "model_forward_time": 1.715726375579834,
      "step": 895
    },
    {
      "epoch": 8.535385131835938e-07,
      "step": 895,
      "training_step_time": 9.204359292984009
    },
    {
      "epoch": 8.544921875e-07,
      "model_forward_time": 2.8266494274139404,
      "step": 896
    },
    {
      "epoch": 8.544921875e-07,
      "step": 896,
      "training_step_time": 7.713206768035889
    },
    {
      "epoch": 8.554458618164063e-07,
      "model_forward_time": 1.7189390659332275,
      "step": 897
    },
    {
      "epoch": 8.554458618164063e-07,
      "step": 897,
      "training_step_time": 7.794064044952393
    },
    {
      "epoch": 8.563995361328125e-07,
      "model_forward_time": 1.7190752029418945,
      "step": 898
    },
    {
      "epoch": 8.563995361328125e-07,
      "step": 898,
      "training_step_time": 7.81798791885376
    },
    {
      "epoch": 8.573532104492188e-07,
      "model_forward_time": 1.7160687446594238,
      "step": 899
    },
    {
      "epoch": 8.573532104492188e-07,
      "step": 899,
      "training_step_time": 8.949146747589111
    },
    {
      "epoch": 8.58306884765625e-07,
      "grad_norm": 4.132716655731201,
      "learning_rate": 1.798e-06,
      "loss": 0.1327,
      "step": 900
    },
    {
      "epoch": 8.58306884765625e-07,
      "model_forward_time": 1.7435777187347412,
      "step": 900
    },
    {
      "dynamics_loss_avg": 0.09050294682383538,
      "epoch": 8.58306884765625e-07,
      "step": 900
    },
    {
      "action_loss_avg": 0.04136559143662453,
      "epoch": 8.58306884765625e-07,
      "step": 900
    },
    {
      "epoch": 8.58306884765625e-07,
      "step": 900,
      "training_step_time": 7.6422905921936035
    },
    {
      "epoch": 8.592605590820313e-07,
      "model_forward_time": 1.7198855876922607,
      "step": 901
    },
    {
      "epoch": 8.592605590820313e-07,
      "step": 901,
      "training_step_time": 7.785876512527466
    },
    {
      "epoch": 8.602142333984375e-07,
      "model_forward_time": 1.7079899311065674,
      "step": 902
    },
    {
      "epoch": 8.602142333984375e-07,
      "step": 902,
      "training_step_time": 7.674105882644653
    },
    {
      "epoch": 8.611679077148438e-07,
      "model_forward_time": 1.7393343448638916,
      "step": 903
    },
    {
      "epoch": 8.611679077148438e-07,
      "step": 903,
      "training_step_time": 8.494426012039185
    },
    {
      "epoch": 8.6212158203125e-07,
      "model_forward_time": 1.718139410018921,
      "step": 904
    },
    {
      "epoch": 8.6212158203125e-07,
      "step": 904,
      "training_step_time": 7.588921308517456
    },
    {
      "epoch": 8.630752563476563e-07,
      "model_forward_time": 2.91251277923584,
      "step": 905
    },
    {
      "epoch": 8.630752563476563e-07,
      "step": 905,
      "training_step_time": 7.710063457489014
    },
    {
      "epoch": 8.640289306640625e-07,
      "model_forward_time": 1.717634916305542,
      "step": 906
    },
    {
      "epoch": 8.640289306640625e-07,
      "step": 906,
      "training_step_time": 7.702049016952515
    },
    {
      "epoch": 8.649826049804688e-07,
      "model_forward_time": 1.70902419090271,
      "step": 907
    },
    {
      "epoch": 8.649826049804688e-07,
      "step": 907,
      "training_step_time": 7.751503229141235
    },
    {
      "epoch": 8.65936279296875e-07,
      "model_forward_time": 1.7156310081481934,
      "step": 908
    },
    {
      "epoch": 8.65936279296875e-07,
      "step": 908,
      "training_step_time": 7.953050136566162
    },
    {
      "epoch": 8.668899536132813e-07,
      "model_forward_time": 1.729966402053833,
      "step": 909
    },
    {
      "epoch": 8.668899536132813e-07,
      "step": 909,
      "training_step_time": 8.286208868026733
    },
    {
      "epoch": 8.678436279296875e-07,
      "grad_norm": 4.135775566101074,
      "learning_rate": 1.818e-06,
      "loss": 0.1319,
      "step": 910
    },
    {
      "epoch": 8.678436279296875e-07,
      "model_forward_time": 1.7081267833709717,
      "step": 910
    },
    {
      "dynamics_loss_avg": 0.08479868322610855,
      "epoch": 8.678436279296875e-07,
      "step": 910
    },
    {
      "action_loss_avg": 0.05348649527877569,
      "epoch": 8.678436279296875e-07,
      "step": 910
    },
    {
      "epoch": 8.678436279296875e-07,
      "step": 910,
      "training_step_time": 8.701900482177734
    },
    {
      "epoch": 8.687973022460938e-07,
      "model_forward_time": 1.7218787670135498,
      "step": 911
    },
    {
      "epoch": 8.687973022460938e-07,
      "step": 911,
      "training_step_time": 7.793251037597656
    },
    {
      "epoch": 8.697509765625e-07,
      "model_forward_time": 1.7456071376800537,
      "step": 912
    },
    {
      "epoch": 8.697509765625e-07,
      "step": 912,
      "training_step_time": 9.76960039138794
    },
    {
      "epoch": 8.707046508789063e-07,
      "model_forward_time": 1.7185842990875244,
      "step": 913
    },
    {
      "epoch": 8.707046508789063e-07,
      "step": 913,
      "training_step_time": 9.25711178779602
    },
    {
      "epoch": 8.716583251953125e-07,
      "model_forward_time": 2.8672266006469727,
      "step": 914
    },
    {
      "epoch": 8.716583251953125e-07,
      "step": 914,
      "training_step_time": 7.712743282318115
    },
    {
      "epoch": 8.726119995117188e-07,
      "model_forward_time": 1.7185616493225098,
      "step": 915
    },
    {
      "epoch": 8.726119995117188e-07,
      "step": 915,
      "training_step_time": 7.59987211227417
    },
    {
      "epoch": 8.73565673828125e-07,
      "model_forward_time": 1.7195167541503906,
      "step": 916
    },
    {
      "epoch": 8.73565673828125e-07,
      "step": 916,
      "training_step_time": 8.640414953231812
    },
    {
      "epoch": 8.745193481445313e-07,
      "model_forward_time": 1.7013778686523438,
      "step": 917
    },
    {
      "epoch": 8.745193481445313e-07,
      "step": 917,
      "training_step_time": 7.9189722537994385
    },
    {
      "epoch": 8.754730224609375e-07,
      "model_forward_time": 1.7213149070739746,
      "step": 918
    },
    {
      "epoch": 8.754730224609375e-07,
      "step": 918,
      "training_step_time": 7.663952827453613
    },
    {
      "epoch": 8.764266967773438e-07,
      "model_forward_time": 1.7388441562652588,
      "step": 919
    },
    {
      "epoch": 8.764266967773438e-07,
      "step": 919,
      "training_step_time": 8.329184770584106
    },
    {
      "epoch": 8.7738037109375e-07,
      "grad_norm": 3.76987361907959,
      "learning_rate": 1.838e-06,
      "loss": 0.152,
      "step": 920
    },
    {
      "epoch": 8.7738037109375e-07,
      "model_forward_time": 1.712153673171997,
      "step": 920
    },
    {
      "dynamics_loss_avg": 0.10281871296465397,
      "epoch": 8.7738037109375e-07,
      "step": 920
    },
    {
      "action_loss_avg": 0.05900832805782556,
      "epoch": 8.7738037109375e-07,
      "step": 920
    },
    {
      "epoch": 8.7738037109375e-07,
      "step": 920,
      "training_step_time": 7.653105020523071
    },
    {
      "epoch": 8.783340454101563e-07,
      "model_forward_time": 1.7455143928527832,
      "step": 921
    },
    {
      "epoch": 8.783340454101563e-07,
      "step": 921,
      "training_step_time": 9.353513717651367
    },
    {
      "epoch": 8.792877197265625e-07,
      "model_forward_time": 1.7228691577911377,
      "step": 922
    },
    {
      "epoch": 8.792877197265625e-07,
      "step": 922,
      "training_step_time": 7.733316898345947
    },
    {
      "epoch": 8.802413940429688e-07,
      "model_forward_time": 2.8596017360687256,
      "step": 923
    },
    {
      "epoch": 8.802413940429688e-07,
      "step": 923,
      "training_step_time": 7.687209367752075
    },
    {
      "epoch": 8.81195068359375e-07,
      "model_forward_time": 1.7179741859436035,
      "step": 924
    },
    {
      "epoch": 8.81195068359375e-07,
      "step": 924,
      "training_step_time": 8.441798686981201
    },
    {
      "epoch": 8.821487426757813e-07,
      "model_forward_time": 1.7227997779846191,
      "step": 925
    },
    {
      "epoch": 8.821487426757813e-07,
      "step": 925,
      "training_step_time": 7.693136692047119
    },
    {
      "epoch": 8.831024169921875e-07,
      "model_forward_time": 1.7167658805847168,
      "step": 926
    },
    {
      "epoch": 8.831024169921875e-07,
      "step": 926,
      "training_step_time": 7.9276933670043945
    },
    {
      "epoch": 8.840560913085938e-07,
      "model_forward_time": 1.716341257095337,
      "step": 927
    },
    {
      "epoch": 8.840560913085938e-07,
      "step": 927,
      "training_step_time": 7.701953411102295
    },
    {
      "epoch": 8.85009765625e-07,
      "model_forward_time": 1.7323415279388428,
      "step": 928
    },
    {
      "epoch": 8.85009765625e-07,
      "step": 928,
      "training_step_time": 8.659721374511719
    },
    {
      "epoch": 8.859634399414063e-07,
      "model_forward_time": 1.725393295288086,
      "step": 929
    },
    {
      "epoch": 8.859634399414063e-07,
      "step": 929,
      "training_step_time": 8.747065782546997
    },
    {
      "epoch": 8.869171142578125e-07,
      "grad_norm": 3.3685553073883057,
      "learning_rate": 1.858e-06,
      "loss": 0.1506,
      "step": 930
    },
    {
      "epoch": 8.869171142578125e-07,
      "model_forward_time": 1.721731424331665,
      "step": 930
    },
    {
      "dynamics_loss_avg": 0.09680515602231025,
      "epoch": 8.869171142578125e-07,
      "step": 930
    },
    {
      "action_loss_avg": 0.06126527152955532,
      "epoch": 8.869171142578125e-07,
      "step": 930
    },
    {
      "epoch": 8.869171142578125e-07,
      "step": 930,
      "training_step_time": 7.82196569442749
    },
    {
      "epoch": 8.878707885742188e-07,
      "model_forward_time": 1.7154715061187744,
      "step": 931
    },
    {
      "epoch": 8.878707885742188e-07,
      "step": 931,
      "training_step_time": 8.438887357711792
    },
    {
      "epoch": 8.88824462890625e-07,
      "model_forward_time": 2.8692872524261475,
      "step": 932
    },
    {
      "epoch": 8.88824462890625e-07,
      "step": 932,
      "training_step_time": 9.170904159545898
    },
    {
      "epoch": 8.897781372070313e-07,
      "model_forward_time": 1.7150485515594482,
      "step": 933
    },
    {
      "epoch": 8.897781372070313e-07,
      "step": 933,
      "training_step_time": 7.448299407958984
    },
    {
      "epoch": 8.907318115234375e-07,
      "model_forward_time": 1.712022066116333,
      "step": 934
    },
    {
      "epoch": 8.907318115234375e-07,
      "step": 934,
      "training_step_time": 7.731033563613892
    },
    {
      "epoch": 8.916854858398438e-07,
      "model_forward_time": 1.729816198348999,
      "step": 935
    },
    {
      "epoch": 8.916854858398438e-07,
      "step": 935,
      "training_step_time": 8.845345735549927
    },
    {
      "epoch": 8.9263916015625e-07,
      "model_forward_time": 1.7208154201507568,
      "step": 936
    },
    {
      "epoch": 8.9263916015625e-07,
      "step": 936,
      "training_step_time": 7.66467022895813
    },
    {
      "epoch": 8.935928344726563e-07,
      "model_forward_time": 1.7308878898620605,
      "step": 937
    },
    {
      "epoch": 8.935928344726563e-07,
      "step": 937,
      "training_step_time": 7.778536796569824
    },
    {
      "epoch": 8.945465087890624e-07,
      "model_forward_time": 1.7141094207763672,
      "step": 938
    },
    {
      "epoch": 8.945465087890624e-07,
      "step": 938,
      "training_step_time": 7.787133693695068
    },
    {
      "epoch": 8.955001831054687e-07,
      "model_forward_time": 1.7171761989593506,
      "step": 939
    },
    {
      "epoch": 8.955001831054687e-07,
      "step": 939,
      "training_step_time": 9.051968097686768
    },
    {
      "epoch": 8.96453857421875e-07,
      "grad_norm": 2.9535038471221924,
      "learning_rate": 1.878e-06,
      "loss": 0.1354,
      "step": 940
    },
    {
      "epoch": 8.96453857421875e-07,
      "model_forward_time": 1.7108805179595947,
      "step": 940
    },
    {
      "dynamics_loss_avg": 0.06976192705333233,
      "epoch": 8.96453857421875e-07,
      "step": 940
    },
    {
      "action_loss_avg": 0.046373847499489786,
      "epoch": 8.96453857421875e-07,
      "step": 940
    },
    {
      "epoch": 8.96453857421875e-07,
      "step": 940,
      "training_step_time": 7.754448652267456
    },
    {
      "epoch": 8.974075317382812e-07,
      "model_forward_time": 2.7748355865478516,
      "step": 941
    },
    {
      "epoch": 8.974075317382812e-07,
      "step": 941,
      "training_step_time": 8.625950574874878
    },
    {
      "epoch": 8.983612060546875e-07,
      "model_forward_time": 1.7225215435028076,
      "step": 942
    },
    {
      "epoch": 8.983612060546875e-07,
      "step": 942,
      "training_step_time": 7.64017128944397
    },
    {
      "epoch": 8.993148803710937e-07,
      "model_forward_time": 1.717621088027954,
      "step": 943
    },
    {
      "epoch": 8.993148803710937e-07,
      "step": 943,
      "training_step_time": 8.683181047439575
    },
    {
      "epoch": 9.002685546875e-07,
      "model_forward_time": 1.720597505569458,
      "step": 944
    },
    {
      "epoch": 9.002685546875e-07,
      "step": 944,
      "training_step_time": 8.696845769882202
    },
    {
      "epoch": 9.012222290039062e-07,
      "model_forward_time": 1.7124249935150146,
      "step": 945
    },
    {
      "epoch": 9.012222290039062e-07,
      "step": 945,
      "training_step_time": 8.569155931472778
    },
    {
      "epoch": 9.021759033203125e-07,
      "model_forward_time": 1.7185027599334717,
      "step": 946
    },
    {
      "epoch": 9.021759033203125e-07,
      "step": 946,
      "training_step_time": 8.531668424606323
    },
    {
      "epoch": 9.031295776367187e-07,
      "model_forward_time": 1.7208068370819092,
      "step": 947
    },
    {
      "epoch": 9.031295776367187e-07,
      "step": 947,
      "training_step_time": 8.503244400024414
    },
    {
      "epoch": 9.04083251953125e-07,
      "model_forward_time": 1.7147362232208252,
      "step": 948
    },
    {
      "epoch": 9.04083251953125e-07,
      "step": 948,
      "training_step_time": 7.538259744644165
    },
    {
      "epoch": 9.050369262695312e-07,
      "model_forward_time": 1.7424705028533936,
      "step": 949
    },
    {
      "epoch": 9.050369262695312e-07,
      "step": 949,
      "training_step_time": 8.466831684112549
    },
    {
      "epoch": 9.059906005859375e-07,
      "grad_norm": 3.455247163772583,
      "learning_rate": 1.898e-06,
      "loss": 0.1521,
      "step": 950
    },
    {
      "epoch": 9.059906005859375e-07,
      "model_forward_time": 1.7111048698425293,
      "step": 950
    },
    {
      "dynamics_loss_avg": 0.08260140866041184,
      "epoch": 9.059906005859375e-07,
      "step": 950
    },
    {
      "action_loss_avg": 0.03618793366476893,
      "epoch": 9.059906005859375e-07,
      "step": 950
    },
    {
      "epoch": 9.059906005859375e-07,
      "step": 950,
      "training_step_time": 7.614393711090088
    },
    {
      "epoch": 9.069442749023437e-07,
      "model_forward_time": 2.930798292160034,
      "step": 951
    },
    {
      "epoch": 9.069442749023437e-07,
      "step": 951,
      "training_step_time": 8.76399040222168
    },
    {
      "epoch": 9.0789794921875e-07,
      "model_forward_time": 1.7195119857788086,
      "step": 952
    },
    {
      "epoch": 9.0789794921875e-07,
      "step": 952,
      "training_step_time": 9.520845413208008
    },
    {
      "epoch": 9.088516235351562e-07,
      "model_forward_time": 1.718513011932373,
      "step": 953
    },
    {
      "epoch": 9.088516235351562e-07,
      "step": 953,
      "training_step_time": 7.755998134613037
    },
    {
      "epoch": 9.098052978515625e-07,
      "model_forward_time": 1.718780279159546,
      "step": 954
    },
    {
      "epoch": 9.098052978515625e-07,
      "step": 954,
      "training_step_time": 7.921518564224243
    },
    {
      "epoch": 9.107589721679687e-07,
      "model_forward_time": 1.7196407318115234,
      "step": 955
    },
    {
      "epoch": 9.107589721679687e-07,
      "step": 955,
      "training_step_time": 7.678781986236572
    },
    {
      "epoch": 9.11712646484375e-07,
      "model_forward_time": 1.710597276687622,
      "step": 956
    },
    {
      "epoch": 9.11712646484375e-07,
      "step": 956,
      "training_step_time": 7.7817535400390625
    },
    {
      "epoch": 9.126663208007812e-07,
      "model_forward_time": 1.7176008224487305,
      "step": 957
    },
    {
      "epoch": 9.126663208007812e-07,
      "step": 957,
      "training_step_time": 7.66107439994812
    },
    {
      "epoch": 9.136199951171875e-07,
      "model_forward_time": 1.3629159927368164,
      "step": 958
    },
    {
      "epoch": 9.136199951171875e-07,
      "step": 958,
      "training_step_time": 7.790975093841553
    },
    {
      "epoch": 9.145736694335937e-07,
      "model_forward_time": 1.7153582572937012,
      "step": 959
    },
    {
      "epoch": 9.145736694335937e-07,
      "step": 959,
      "training_step_time": 8.607600927352905
    },
    {
      "epoch": 9.1552734375e-07,
      "grad_norm": 2.5387916564941406,
      "learning_rate": 1.918e-06,
      "loss": 0.1266,
      "step": 960
    },
    {
      "epoch": 9.1552734375e-07,
      "model_forward_time": 2.8318748474121094,
      "step": 960
    },
    {
      "dynamics_loss_avg": 0.06991430092602968,
      "epoch": 9.1552734375e-07,
      "step": 960
    },
    {
      "action_loss_avg": 0.04034458650276065,
      "epoch": 9.1552734375e-07,
      "step": 960
    },
    {
      "epoch": 9.1552734375e-07,
      "step": 960,
      "training_step_time": 7.701399564743042
    },
    {
      "epoch": 9.164810180664062e-07,
      "model_forward_time": 1.7198116779327393,
      "step": 961
    },
    {
      "epoch": 9.164810180664062e-07,
      "step": 961,
      "training_step_time": 8.485939741134644
    },
    {
      "epoch": 9.174346923828125e-07,
      "model_forward_time": 1.723332405090332,
      "step": 962
    },
    {
      "epoch": 9.174346923828125e-07,
      "step": 962,
      "training_step_time": 8.734177827835083
    },
    {
      "epoch": 9.183883666992187e-07,
      "model_forward_time": 1.7210493087768555,
      "step": 963
    },
    {
      "epoch": 9.183883666992187e-07,
      "step": 963,
      "training_step_time": 9.058452606201172
    },
    {
      "epoch": 9.19342041015625e-07,
      "model_forward_time": 1.7228546142578125,
      "step": 964
    },
    {
      "epoch": 9.19342041015625e-07,
      "step": 964,
      "training_step_time": 7.691207408905029
    },
    {
      "epoch": 9.202957153320312e-07,
      "model_forward_time": 1.744358777999878,
      "step": 965
    },
    {
      "epoch": 9.202957153320312e-07,
      "step": 965,
      "training_step_time": 8.673874855041504
    },
    {
      "epoch": 9.212493896484375e-07,
      "model_forward_time": 1.714977741241455,
      "step": 966
    },
    {
      "epoch": 9.212493896484375e-07,
      "step": 966,
      "training_step_time": 11.647830963134766
    },
    {
      "epoch": 9.222030639648437e-07,
      "model_forward_time": 1.726038932800293,
      "step": 967
    },
    {
      "epoch": 9.222030639648437e-07,
      "step": 967,
      "training_step_time": 8.609747409820557
    },
    {
      "epoch": 9.2315673828125e-07,
      "model_forward_time": 1.7196624279022217,
      "step": 968
    },
    {
      "epoch": 9.2315673828125e-07,
      "step": 968,
      "training_step_time": 6.880950212478638
    },
    {
      "epoch": 9.241104125976562e-07,
      "model_forward_time": 1.7196481227874756,
      "step": 969
    },
    {
      "epoch": 9.241104125976562e-07,
      "step": 969,
      "training_step_time": 6.826159238815308
    },
    {
      "epoch": 9.250640869140625e-07,
      "grad_norm": 2.727336883544922,
      "learning_rate": 1.9380000000000003e-06,
      "loss": 0.1423,
      "step": 970
    },
    {
      "epoch": 9.250640869140625e-07,
      "model_forward_time": 1.7044434547424316,
      "step": 970
    },
    {
      "dynamics_loss_avg": 0.06380363572388888,
      "epoch": 9.250640869140625e-07,
      "step": 970
    },
    {
      "action_loss_avg": 0.039738962426781654,
      "epoch": 9.250640869140625e-07,
      "step": 970
    },
    {
      "epoch": 9.250640869140625e-07,
      "step": 970,
      "training_step_time": 6.860019683837891
    },
    {
      "epoch": 9.260177612304687e-07,
      "model_forward_time": 1.727971076965332,
      "step": 971
    },
    {
      "epoch": 9.260177612304687e-07,
      "step": 971,
      "training_step_time": 6.87732720375061
    },
    {
      "epoch": 9.26971435546875e-07,
      "model_forward_time": 1.7244198322296143,
      "step": 972
    },
    {
      "epoch": 9.26971435546875e-07,
      "step": 972,
      "training_step_time": 6.857078552246094
    },
    {
      "epoch": 9.279251098632812e-07,
      "model_forward_time": 1.7238969802856445,
      "step": 973
    },
    {
      "epoch": 9.279251098632812e-07,
      "step": 973,
      "training_step_time": 6.846999168395996
    },
    {
      "epoch": 9.288787841796875e-07,
      "model_forward_time": 1.7144017219543457,
      "step": 974
    },
    {
      "epoch": 9.288787841796875e-07,
      "step": 974,
      "training_step_time": 6.864513158798218
    },
    {
      "epoch": 9.298324584960937e-07,
      "model_forward_time": 1.7145302295684814,
      "step": 975
    },
    {
      "epoch": 9.298324584960937e-07,
      "step": 975,
      "training_step_time": 6.804215669631958
    },
    {
      "epoch": 9.307861328125e-07,
      "model_forward_time": 2.8407723903656006,
      "step": 976
    },
    {
      "epoch": 9.307861328125e-07,
      "step": 976,
      "training_step_time": 8.750728130340576
    },
    {
      "epoch": 9.317398071289062e-07,
      "model_forward_time": 1.7232537269592285,
      "step": 977
    },
    {
      "epoch": 9.317398071289062e-07,
      "step": 977,
      "training_step_time": 7.257093191146851
    },
    {
      "epoch": 9.326934814453125e-07,
      "model_forward_time": 1.7308695316314697,
      "step": 978
    },
    {
      "epoch": 9.326934814453125e-07,
      "step": 978,
      "training_step_time": 6.821022272109985
    },
    {
      "epoch": 9.336471557617187e-07,
      "model_forward_time": 1.7249181270599365,
      "step": 979
    },
    {
      "epoch": 9.336471557617187e-07,
      "step": 979,
      "training_step_time": 6.810015916824341
    },
    {
      "epoch": 9.34600830078125e-07,
      "grad_norm": 5.152181625366211,
      "learning_rate": 1.958e-06,
      "loss": 0.1385,
      "step": 980
    },
    {
      "epoch": 9.34600830078125e-07,
      "model_forward_time": 1.7116725444793701,
      "step": 980
    },
    {
      "dynamics_loss_avg": 0.0976649634540081,
      "epoch": 9.34600830078125e-07,
      "step": 980
    },
    {
      "action_loss_avg": 0.06734657231718302,
      "epoch": 9.34600830078125e-07,
      "step": 980
    },
    {
      "epoch": 9.34600830078125e-07,
      "step": 980,
      "training_step_time": 6.828380823135376
    },
    {
      "epoch": 9.355545043945312e-07,
      "model_forward_time": 1.7287318706512451,
      "step": 981
    },
    {
      "epoch": 9.355545043945312e-07,
      "step": 981,
      "training_step_time": 6.803689956665039
    },
    {
      "epoch": 9.365081787109375e-07,
      "model_forward_time": 1.716705322265625,
      "step": 982
    },
    {
      "epoch": 9.365081787109375e-07,
      "step": 982,
      "training_step_time": 6.830366134643555
    },
    {
      "epoch": 9.374618530273437e-07,
      "model_forward_time": 1.7133958339691162,
      "step": 983
    },
    {
      "epoch": 9.374618530273437e-07,
      "step": 983,
      "training_step_time": 6.8325817584991455
    },
    {
      "epoch": 9.3841552734375e-07,
      "model_forward_time": 1.7204315662384033,
      "step": 984
    },
    {
      "epoch": 9.3841552734375e-07,
      "step": 984,
      "training_step_time": 6.836839199066162
    },
    {
      "epoch": 9.393692016601562e-07,
      "model_forward_time": 2.8488216400146484,
      "step": 985
    },
    {
      "epoch": 9.393692016601562e-07,
      "step": 985,
      "training_step_time": 11.179688930511475
    },
    {
      "epoch": 9.403228759765625e-07,
      "model_forward_time": 1.7254295349121094,
      "step": 986
    },
    {
      "epoch": 9.403228759765625e-07,
      "step": 986,
      "training_step_time": 7.704755783081055
    },
    {
      "epoch": 9.412765502929687e-07,
      "model_forward_time": 1.7243249416351318,
      "step": 987
    },
    {
      "epoch": 9.412765502929687e-07,
      "step": 987,
      "training_step_time": 6.818444013595581
    },
    {
      "epoch": 9.42230224609375e-07,
      "model_forward_time": 1.722196102142334,
      "step": 988
    },
    {
      "epoch": 9.42230224609375e-07,
      "step": 988,
      "training_step_time": 6.84975790977478
    },
    {
      "epoch": 9.431838989257812e-07,
      "model_forward_time": 1.7239668369293213,
      "step": 989
    },
    {
      "epoch": 9.431838989257812e-07,
      "step": 989,
      "training_step_time": 6.867578983306885
    },
    {
      "epoch": 9.441375732421875e-07,
      "grad_norm": 2.7368714809417725,
      "learning_rate": 1.9780000000000004e-06,
      "loss": 0.1333,
      "step": 990
    },
    {
      "epoch": 9.441375732421875e-07,
      "model_forward_time": 1.7179179191589355,
      "step": 990
    },
    {
      "dynamics_loss_avg": 0.07275562155991792,
      "epoch": 9.441375732421875e-07,
      "step": 990
    },
    {
      "action_loss_avg": 0.041397635079920295,
      "epoch": 9.441375732421875e-07,
      "step": 990
    },
    {
      "epoch": 9.441375732421875e-07,
      "step": 990,
      "training_step_time": 6.8214545249938965
    },
    {
      "epoch": 9.450912475585937e-07,
      "model_forward_time": 1.715599536895752,
      "step": 991
    },
    {
      "epoch": 9.450912475585937e-07,
      "step": 991,
      "training_step_time": 6.840726375579834
    },
    {
      "epoch": 9.46044921875e-07,
      "model_forward_time": 1.7203028202056885,
      "step": 992
    },
    {
      "epoch": 9.46044921875e-07,
      "step": 992,
      "training_step_time": 6.826893091201782
    },
    {
      "epoch": 9.469985961914062e-07,
      "model_forward_time": 1.7153193950653076,
      "step": 993
    },
    {
      "epoch": 9.469985961914062e-07,
      "step": 993,
      "training_step_time": 6.860549211502075
    },
    {
      "epoch": 9.479522705078125e-07,
      "model_forward_time": 2.8804099559783936,
      "step": 994
    },
    {
      "epoch": 9.479522705078125e-07,
      "step": 994,
      "training_step_time": 10.794715404510498
    },
    {
      "epoch": 9.489059448242187e-07,
      "model_forward_time": 1.35951566696167,
      "step": 995
    },
    {
      "epoch": 9.489059448242187e-07,
      "step": 995,
      "training_step_time": 7.708569049835205
    },
    {
      "epoch": 9.49859619140625e-07,
      "model_forward_time": 1.715334177017212,
      "step": 996
    },
    {
      "epoch": 9.49859619140625e-07,
      "step": 996,
      "training_step_time": 6.804772615432739
    },
    {
      "epoch": 9.508132934570312e-07,
      "model_forward_time": 1.7192323207855225,
      "step": 997
    },
    {
      "epoch": 9.508132934570312e-07,
      "step": 997,
      "training_step_time": 6.819971799850464
    },
    {
      "epoch": 9.517669677734375e-07,
      "model_forward_time": 1.7367689609527588,
      "step": 998
    },
    {
      "epoch": 9.517669677734375e-07,
      "step": 998,
      "training_step_time": 6.981008768081665
    },
    {
      "epoch": 9.527206420898437e-07,
      "model_forward_time": 1.8078501224517822,
      "step": 999
    },
    {
      "epoch": 9.527206420898437e-07,
      "step": 999,
      "training_step_time": 6.8531653881073
    },
    {
      "epoch": 9.5367431640625e-07,
      "grad_norm": 4.776866912841797,
      "learning_rate": 1.998e-06,
      "loss": 0.1373,
      "step": 1000
    },
    {
      "epoch": 9.5367431640625e-07,
      "model_forward_time": 1.7273101806640625,
      "step": 1000
    },
    {
      "dynamics_loss_avg": 0.09030454233288765,
      "epoch": 9.5367431640625e-07,
      "step": 1000
    },
    {
      "action_loss_avg": 0.058996882289648056,
      "epoch": 9.5367431640625e-07,
      "step": 1000
    },
    {
      "epoch": 9.5367431640625e-07,
      "step": 1000,
      "training_step_time": 6.901028633117676
    },
    {
      "epoch": 9.546279907226561e-07,
      "model_forward_time": 1.7289018630981445,
      "step": 1001
    },
    {
      "epoch": 9.546279907226561e-07,
      "step": 1001,
      "training_step_time": 6.881777286529541
    },
    {
      "epoch": 9.555816650390625e-07,
      "model_forward_time": 1.7324779033660889,
      "step": 1002
    },
    {
      "epoch": 9.555816650390625e-07,
      "step": 1002,
      "training_step_time": 6.909353733062744
    },
    {
      "epoch": 9.565353393554686e-07,
      "model_forward_time": 2.717836380004883,
      "step": 1003
    },
    {
      "epoch": 9.565353393554686e-07,
      "step": 1003,
      "training_step_time": 10.11046290397644
    },
    {
      "epoch": 9.57489013671875e-07,
      "model_forward_time": 1.7168684005737305,
      "step": 1004
    },
    {
      "epoch": 9.57489013671875e-07,
      "step": 1004,
      "training_step_time": 8.553043603897095
    },
    {
      "epoch": 9.584426879882811e-07,
      "model_forward_time": 1.7386677265167236,
      "step": 1005
    },
    {
      "epoch": 9.584426879882811e-07,
      "step": 1005,
      "training_step_time": 7.116193056106567
    },
    {
      "epoch": 9.593963623046875e-07,
      "model_forward_time": 1.7245237827301025,
      "step": 1006
    },
    {
      "epoch": 9.593963623046875e-07,
      "step": 1006,
      "training_step_time": 6.883649110794067
    },
    {
      "epoch": 9.603500366210937e-07,
      "model_forward_time": 1.721224069595337,
      "step": 1007
    },
    {
      "epoch": 9.603500366210937e-07,
      "step": 1007,
      "training_step_time": 6.858428955078125
    },
    {
      "epoch": 9.613037109375e-07,
      "model_forward_time": 1.7294690608978271,
      "step": 1008
    },
    {
      "epoch": 9.613037109375e-07,
      "step": 1008,
      "training_step_time": 6.865239858627319
    },
    {
      "epoch": 9.622573852539062e-07,
      "model_forward_time": 1.7296218872070312,
      "step": 1009
    },
    {
      "epoch": 9.622573852539062e-07,
      "step": 1009,
      "training_step_time": 6.883192539215088
    },
    {
      "epoch": 9.632110595703125e-07,
      "grad_norm": 2.52067232131958,
      "learning_rate": 2.0180000000000004e-06,
      "loss": 0.1137,
      "step": 1010
    },
    {
      "epoch": 9.632110595703125e-07,
      "model_forward_time": 1.736788272857666,
      "step": 1010
    },
    {
      "dynamics_loss_avg": 0.07387881446629763,
      "epoch": 9.632110595703125e-07,
      "step": 1010
    },
    {
      "action_loss_avg": 0.04722910728305578,
      "epoch": 9.632110595703125e-07,
      "step": 1010
    },
    {
      "epoch": 9.632110595703125e-07,
      "step": 1010,
      "training_step_time": 6.85505747795105
    },
    {
      "epoch": 9.641647338867187e-07,
      "model_forward_time": 1.7261855602264404,
      "step": 1011
    },
    {
      "epoch": 9.641647338867187e-07,
      "step": 1011,
      "training_step_time": 6.888763904571533
    },
    {
      "epoch": 9.65118408203125e-07,
      "model_forward_time": 2.2497522830963135,
      "step": 1012
    },
    {
      "epoch": 9.65118408203125e-07,
      "step": 1012,
      "training_step_time": 8.418315410614014
    },
    {
      "epoch": 9.660720825195312e-07,
      "model_forward_time": 1.7546465396881104,
      "step": 1013
    },
    {
      "epoch": 9.660720825195312e-07,
      "step": 1013,
      "training_step_time": 9.467488527297974
    },
    {
      "epoch": 9.670257568359375e-07,
      "model_forward_time": 1.7241981029510498,
      "step": 1014
    },
    {
      "epoch": 9.670257568359375e-07,
      "step": 1014,
      "training_step_time": 7.615995407104492
    },
    {
      "epoch": 9.679794311523437e-07,
      "model_forward_time": 1.7236928939819336,
      "step": 1015
    },
    {
      "epoch": 9.679794311523437e-07,
      "step": 1015,
      "training_step_time": 6.898115634918213
    },
    {
      "epoch": 9.6893310546875e-07,
      "model_forward_time": 1.730229377746582,
      "step": 1016
    },
    {
      "epoch": 9.6893310546875e-07,
      "step": 1016,
      "training_step_time": 6.905230283737183
    },
    {
      "epoch": 9.698867797851562e-07,
      "model_forward_time": 1.7270622253417969,
      "step": 1017
    },
    {
      "epoch": 9.698867797851562e-07,
      "step": 1017,
      "training_step_time": 6.829963684082031
    },
    {
      "epoch": 9.708404541015625e-07,
      "model_forward_time": 1.7272391319274902,
      "step": 1018
    },
    {
      "epoch": 9.708404541015625e-07,
      "step": 1018,
      "training_step_time": 6.912939071655273
    },
    {
      "epoch": 9.717941284179687e-07,
      "model_forward_time": 1.7265410423278809,
      "step": 1019
    },
    {
      "epoch": 9.717941284179687e-07,
      "step": 1019,
      "training_step_time": 6.895693778991699
    },
    {
      "epoch": 9.72747802734375e-07,
      "grad_norm": 4.7687883377075195,
      "learning_rate": 2.0380000000000002e-06,
      "loss": 0.1277,
      "step": 1020
    },
    {
      "epoch": 9.72747802734375e-07,
      "model_forward_time": 1.7257275581359863,
      "step": 1020
    },
    {
      "dynamics_loss_avg": 0.08344086036086082,
      "epoch": 9.72747802734375e-07,
      "step": 1020
    },
    {
      "action_loss_avg": 0.046363506373018025,
      "epoch": 9.72747802734375e-07,
      "step": 1020
    },
    {
      "epoch": 9.72747802734375e-07,
      "step": 1020,
      "training_step_time": 6.969858169555664
    },
    {
      "epoch": 9.737014770507812e-07,
      "model_forward_time": 2.278247356414795,
      "step": 1021
    },
    {
      "epoch": 9.737014770507812e-07,
      "step": 1021,
      "training_step_time": 7.29604434967041
    },
    {
      "epoch": 9.746551513671875e-07,
      "model_forward_time": 1.7525827884674072,
      "step": 1022
    },
    {
      "epoch": 9.746551513671875e-07,
      "step": 1022,
      "training_step_time": 10.429097414016724
    },
    {
      "epoch": 9.756088256835937e-07,
      "model_forward_time": 1.7283062934875488,
      "step": 1023
    },
    {
      "epoch": 9.756088256835937e-07,
      "step": 1023,
      "training_step_time": 7.693947076797485
    },
    {
      "epoch": 9.765625e-07,
      "model_forward_time": 1.7250616550445557,
      "step": 1024
    },
    {
      "epoch": 9.765625e-07,
      "step": 1024,
      "training_step_time": 6.917737245559692
    },
    {
      "epoch": 9.775161743164062e-07,
      "model_forward_time": 1.720855712890625,
      "step": 1025
    },
    {
      "epoch": 9.775161743164062e-07,
      "step": 1025,
      "training_step_time": 6.9082067012786865
    },
    {
      "epoch": 9.784698486328125e-07,
      "model_forward_time": 1.7355315685272217,
      "step": 1026
    },
    {
      "epoch": 9.784698486328125e-07,
      "step": 1026,
      "training_step_time": 6.889965534210205
    },
    {
      "epoch": 9.794235229492187e-07,
      "model_forward_time": 1.733799695968628,
      "step": 1027
    },
    {
      "epoch": 9.794235229492187e-07,
      "step": 1027,
      "training_step_time": 6.883948087692261
    },
    {
      "epoch": 9.80377197265625e-07,
      "model_forward_time": 1.7356605529785156,
      "step": 1028
    },
    {
      "epoch": 9.80377197265625e-07,
      "step": 1028,
      "training_step_time": 6.899972200393677
    },
    {
      "epoch": 9.813308715820312e-07,
      "model_forward_time": 1.7299714088439941,
      "step": 1029
    },
    {
      "epoch": 9.813308715820312e-07,
      "step": 1029,
      "training_step_time": 6.916033744812012
    },
    {
      "epoch": 9.822845458984375e-07,
      "grad_norm": 4.328577995300293,
      "learning_rate": 2.0580000000000005e-06,
      "loss": 0.1384,
      "step": 1030
    },
    {
      "epoch": 9.822845458984375e-07,
      "model_forward_time": 2.271796703338623,
      "step": 1030
    },
    {
      "dynamics_loss_avg": 0.08096640147268772,
      "epoch": 9.822845458984375e-07,
      "step": 1030
    },
    {
      "action_loss_avg": 0.05921840686351061,
      "epoch": 9.822845458984375e-07,
      "step": 1030
    },
    {
      "epoch": 9.822845458984375e-07,
      "step": 1030,
      "training_step_time": 7.3076701164245605
    },
    {
      "epoch": 9.832382202148437e-07,
      "model_forward_time": 1.759495496749878,
      "step": 1031
    },
    {
      "epoch": 9.832382202148437e-07,
      "step": 1031,
      "training_step_time": 12.425719499588013
    },
    {
      "epoch": 9.8419189453125e-07,
      "model_forward_time": 1.7303297519683838,
      "step": 1032
    },
    {
      "epoch": 9.8419189453125e-07,
      "step": 1032,
      "training_step_time": 7.879986763000488
    },
    {
      "epoch": 9.851455688476562e-07,
      "model_forward_time": 1.7337226867675781,
      "step": 1033
    },
    {
      "epoch": 9.851455688476562e-07,
      "step": 1033,
      "training_step_time": 6.892273187637329
    },
    {
      "epoch": 9.860992431640625e-07,
      "model_forward_time": 1.738699197769165,
      "step": 1034
    },
    {
      "epoch": 9.860992431640625e-07,
      "step": 1034,
      "training_step_time": 6.872711896896362
    },
    {
      "epoch": 9.870529174804687e-07,
      "model_forward_time": 1.732661247253418,
      "step": 1035
    },
    {
      "epoch": 9.870529174804687e-07,
      "step": 1035,
      "training_step_time": 6.931665420532227
    },
    {
      "epoch": 9.88006591796875e-07,
      "model_forward_time": 1.729938268661499,
      "step": 1036
    },
    {
      "epoch": 9.88006591796875e-07,
      "step": 1036,
      "training_step_time": 6.889222621917725
    },
    {
      "epoch": 9.889602661132812e-07,
      "model_forward_time": 1.7219212055206299,
      "step": 1037
    },
    {
      "epoch": 9.889602661132812e-07,
      "step": 1037,
      "training_step_time": 6.940487623214722
    },
    {
      "epoch": 9.899139404296875e-07,
      "model_forward_time": 1.7294678688049316,
      "step": 1038
    },
    {
      "epoch": 9.899139404296875e-07,
      "step": 1038,
      "training_step_time": 6.933857679367065
    },
    {
      "epoch": 9.908676147460937e-07,
      "model_forward_time": 2.2420034408569336,
      "step": 1039
    },
    {
      "epoch": 9.908676147460937e-07,
      "step": 1039,
      "training_step_time": 7.303634166717529
    },
    {
      "epoch": 9.918212890625e-07,
      "grad_norm": 4.654504299163818,
      "learning_rate": 2.0780000000000003e-06,
      "loss": 0.131,
      "step": 1040
    },
    {
      "epoch": 9.918212890625e-07,
      "model_forward_time": 1.7497859001159668,
      "step": 1040
    },
    {
      "dynamics_loss_avg": 0.09003510810434819,
      "epoch": 9.918212890625e-07,
      "step": 1040
    },
    {
      "action_loss_avg": 0.06262095151469112,
      "epoch": 9.918212890625e-07,
      "step": 1040
    },
    {
      "epoch": 9.918212890625e-07,
      "step": 1040,
      "training_step_time": 10.488535642623901
    },
    {
      "epoch": 9.927749633789062e-07,
      "model_forward_time": 1.7197816371917725,
      "step": 1041
    },
    {
      "epoch": 9.927749633789062e-07,
      "step": 1041,
      "training_step_time": 8.102550268173218
    },
    {
      "epoch": 9.937286376953125e-07,
      "model_forward_time": 1.745478630065918,
      "step": 1042
    },
    {
      "epoch": 9.937286376953125e-07,
      "step": 1042,
      "training_step_time": 7.694082021713257
    },
    {
      "epoch": 9.946823120117187e-07,
      "model_forward_time": 1.7389628887176514,
      "step": 1043
    },
    {
      "epoch": 9.946823120117187e-07,
      "step": 1043,
      "training_step_time": 7.07087516784668
    },
    {
      "epoch": 9.95635986328125e-07,
      "model_forward_time": 1.7290325164794922,
      "step": 1044
    },
    {
      "epoch": 9.95635986328125e-07,
      "step": 1044,
      "training_step_time": 6.934597015380859
    },
    {
      "epoch": 9.965896606445312e-07,
      "model_forward_time": 1.7291958332061768,
      "step": 1045
    },
    {
      "epoch": 9.965896606445312e-07,
      "step": 1045,
      "training_step_time": 7.099797010421753
    },
    {
      "epoch": 9.975433349609375e-07,
      "model_forward_time": 1.735776662826538,
      "step": 1046
    },
    {
      "epoch": 9.975433349609375e-07,
      "step": 1046,
      "training_step_time": 7.0691587924957275
    },
    {
      "epoch": 9.984970092773437e-07,
      "model_forward_time": 1.734339714050293,
      "step": 1047
    },
    {
      "epoch": 9.984970092773437e-07,
      "step": 1047,
      "training_step_time": 7.032320261001587
    },
    {
      "epoch": 9.9945068359375e-07,
      "model_forward_time": 1.3605742454528809,
      "step": 1048
    },
    {
      "epoch": 9.9945068359375e-07,
      "step": 1048,
      "training_step_time": 7.474637031555176
    },
    {
      "epoch": 1.0004043579101562e-06,
      "model_forward_time": 2.2843453884124756,
      "step": 1049
    },
    {
      "epoch": 1.0004043579101562e-06,
      "step": 1049,
      "training_step_time": 9.350382566452026
    },
    {
      "epoch": 1.0013580322265625e-06,
      "grad_norm": 4.433798789978027,
      "learning_rate": 2.098e-06,
      "loss": 0.131,
      "step": 1050
    },
    {
      "epoch": 1.0013580322265625e-06,
      "model_forward_time": 1.7019178867340088,
      "step": 1050
    },
    {
      "dynamics_loss_avg": 0.06711968071758748,
      "epoch": 1.0013580322265625e-06,
      "step": 1050
    },
    {
      "action_loss_avg": 0.049013863503932956,
      "epoch": 1.0013580322265625e-06,
      "step": 1050
    },
    {
      "epoch": 1.0013580322265625e-06,
      "step": 1050,
      "training_step_time": 8.722011089324951
    },
    {
      "epoch": 1.0023117065429687e-06,
      "model_forward_time": 1.707571268081665,
      "step": 1051
    },
    {
      "epoch": 1.0023117065429687e-06,
      "step": 1051,
      "training_step_time": 7.6735711097717285
    },
    {
      "epoch": 1.003265380859375e-06,
      "model_forward_time": 1.7183034420013428,
      "step": 1052
    },
    {
      "epoch": 1.003265380859375e-06,
      "step": 1052,
      "training_step_time": 7.042793273925781
    },
    {
      "epoch": 1.0042190551757812e-06,
      "model_forward_time": 1.7126452922821045,
      "step": 1053
    },
    {
      "epoch": 1.0042190551757812e-06,
      "step": 1053,
      "training_step_time": 7.054816007614136
    },
    {
      "epoch": 1.0051727294921875e-06,
      "model_forward_time": 1.718738079071045,
      "step": 1054
    },
    {
      "epoch": 1.0051727294921875e-06,
      "step": 1054,
      "training_step_time": 7.079203844070435
    },
    {
      "epoch": 1.0061264038085937e-06,
      "model_forward_time": 1.711620569229126,
      "step": 1055
    },
    {
      "epoch": 1.0061264038085937e-06,
      "step": 1055,
      "training_step_time": 7.039037466049194
    },
    {
      "epoch": 1.007080078125e-06,
      "model_forward_time": 1.7101938724517822,
      "step": 1056
    },
    {
      "epoch": 1.007080078125e-06,
      "step": 1056,
      "training_step_time": 7.0571160316467285
    },
    {
      "epoch": 1.0080337524414062e-06,
      "model_forward_time": 1.6940746307373047,
      "step": 1057
    },
    {
      "epoch": 1.0080337524414062e-06,
      "step": 1057,
      "training_step_time": 7.0522027015686035
    },
    {
      "epoch": 1.0089874267578125e-06,
      "model_forward_time": 1.8599557876586914,
      "step": 1058
    },
    {
      "epoch": 1.0089874267578125e-06,
      "step": 1058,
      "training_step_time": 9.663388729095459
    },
    {
      "epoch": 1.0099411010742187e-06,
      "model_forward_time": 1.7151682376861572,
      "step": 1059
    },
    {
      "epoch": 1.0099411010742187e-06,
      "step": 1059,
      "training_step_time": 10.063592672348022
    },
    {
      "epoch": 1.010894775390625e-06,
      "grad_norm": 3.1790904998779297,
      "learning_rate": 2.118e-06,
      "loss": 0.1316,
      "step": 1060
    },
    {
      "epoch": 1.010894775390625e-06,
      "model_forward_time": 1.715574026107788,
      "step": 1060
    },
    {
      "dynamics_loss_avg": 0.07002630084753036,
      "epoch": 1.010894775390625e-06,
      "step": 1060
    },
    {
      "action_loss_avg": 0.043095100577920674,
      "epoch": 1.010894775390625e-06,
      "step": 1060
    },
    {
      "epoch": 1.010894775390625e-06,
      "step": 1060,
      "training_step_time": 7.881439685821533
    },
    {
      "epoch": 1.0118484497070312e-06,
      "model_forward_time": 1.6916587352752686,
      "step": 1061
    },
    {
      "epoch": 1.0118484497070312e-06,
      "step": 1061,
      "training_step_time": 7.020718336105347
    },
    {
      "epoch": 1.0128021240234376e-06,
      "model_forward_time": 1.708827257156372,
      "step": 1062
    },
    {
      "epoch": 1.0128021240234376e-06,
      "step": 1062,
      "training_step_time": 7.094346761703491
    },
    {
      "epoch": 1.0137557983398437e-06,
      "model_forward_time": 1.713916540145874,
      "step": 1063
    },
    {
      "epoch": 1.0137557983398437e-06,
      "step": 1063,
      "training_step_time": 7.115983486175537
    },
    {
      "epoch": 1.01470947265625e-06,
      "model_forward_time": 1.7122788429260254,
      "step": 1064
    },
    {
      "epoch": 1.01470947265625e-06,
      "step": 1064,
      "training_step_time": 7.123759984970093
    },
    {
      "epoch": 1.0156631469726562e-06,
      "model_forward_time": 1.7133285999298096,
      "step": 1065
    },
    {
      "epoch": 1.0156631469726562e-06,
      "step": 1065,
      "training_step_time": 7.0781450271606445
    },
    {
      "epoch": 1.0166168212890626e-06,
      "model_forward_time": 1.7156412601470947,
      "step": 1066
    },
    {
      "epoch": 1.0166168212890626e-06,
      "step": 1066,
      "training_step_time": 6.9193196296691895
    },
    {
      "epoch": 1.0175704956054687e-06,
      "model_forward_time": 2.224339008331299,
      "step": 1067
    },
    {
      "epoch": 1.0175704956054687e-06,
      "step": 1067,
      "training_step_time": 8.695593357086182
    },
    {
      "epoch": 1.018524169921875e-06,
      "model_forward_time": 1.756967544555664,
      "step": 1068
    },
    {
      "epoch": 1.018524169921875e-06,
      "step": 1068,
      "training_step_time": 8.575236320495605
    },
    {
      "epoch": 1.0194778442382812e-06,
      "model_forward_time": 1.3038966655731201,
      "step": 1069
    },
    {
      "epoch": 1.0194778442382812e-06,
      "step": 1069,
      "training_step_time": 8.727528095245361
    },
    {
      "epoch": 1.0204315185546876e-06,
      "grad_norm": 1.985026478767395,
      "learning_rate": 2.138e-06,
      "loss": 0.1331,
      "step": 1070
    },
    {
      "epoch": 1.0204315185546876e-06,
      "model_forward_time": 1.718989610671997,
      "step": 1070
    },
    {
      "dynamics_loss_avg": 0.08543792106211186,
      "epoch": 1.0204315185546876e-06,
      "step": 1070
    },
    {
      "action_loss_avg": 0.059133071824908255,
      "epoch": 1.0204315185546876e-06,
      "step": 1070
    },
    {
      "epoch": 1.0204315185546876e-06,
      "step": 1070,
      "training_step_time": 7.683042526245117
    },
    {
      "epoch": 1.0213851928710937e-06,
      "model_forward_time": 0.9917452335357666,
      "step": 1071
    },
    {
      "epoch": 1.0213851928710937e-06,
      "step": 1071,
      "training_step_time": 6.990738868713379
    },
    {
      "epoch": 1.0223388671875e-06,
      "model_forward_time": 1.7103281021118164,
      "step": 1072
    },
    {
      "epoch": 1.0223388671875e-06,
      "step": 1072,
      "training_step_time": 7.084283828735352
    },
    {
      "epoch": 1.0232925415039062e-06,
      "model_forward_time": 1.7105011940002441,
      "step": 1073
    },
    {
      "epoch": 1.0232925415039062e-06,
      "step": 1073,
      "training_step_time": 7.074907302856445
    },
    {
      "epoch": 1.0242462158203126e-06,
      "model_forward_time": 1.7168450355529785,
      "step": 1074
    },
    {
      "epoch": 1.0242462158203126e-06,
      "step": 1074,
      "training_step_time": 7.149727821350098
    },
    {
      "epoch": 1.0251998901367187e-06,
      "model_forward_time": 1.7081208229064941,
      "step": 1075
    },
    {
      "epoch": 1.0251998901367187e-06,
      "step": 1075,
      "training_step_time": 7.07678747177124
    },
    {
      "epoch": 1.026153564453125e-06,
      "model_forward_time": 1.7090442180633545,
      "step": 1076
    },
    {
      "epoch": 1.026153564453125e-06,
      "step": 1076,
      "training_step_time": 9.47693157196045
    },
    {
      "epoch": 1.0271072387695312e-06,
      "model_forward_time": 2.2488105297088623,
      "step": 1077
    },
    {
      "epoch": 1.0271072387695312e-06,
      "step": 1077,
      "training_step_time": 14.5789213180542
    },
    {
      "epoch": 1.0280609130859376e-06,
      "model_forward_time": 1.7093312740325928,
      "step": 1078
    },
    {
      "epoch": 1.0280609130859376e-06,
      "step": 1078,
      "training_step_time": 8.977971315383911
    },
    {
      "epoch": 1.0290145874023437e-06,
      "model_forward_time": 1.718986988067627,
      "step": 1079
    },
    {
      "epoch": 1.0290145874023437e-06,
      "step": 1079,
      "training_step_time": 7.6675262451171875
    },
    {
      "epoch": 1.02996826171875e-06,
      "grad_norm": 3.1328110694885254,
      "learning_rate": 2.158e-06,
      "loss": 0.1313,
      "step": 1080
    },
    {
      "epoch": 1.02996826171875e-06,
      "model_forward_time": 1.7076337337493896,
      "step": 1080
    },
    {
      "dynamics_loss_avg": 0.0687230221927166,
      "epoch": 1.02996826171875e-06,
      "step": 1080
    },
    {
      "action_loss_avg": 0.04522304553538561,
      "epoch": 1.02996826171875e-06,
      "step": 1080
    },
    {
      "epoch": 1.02996826171875e-06,
      "step": 1080,
      "training_step_time": 7.04494047164917
    },
    {
      "epoch": 1.0309219360351562e-06,
      "model_forward_time": 1.7086341381072998,
      "step": 1081
    },
    {
      "epoch": 1.0309219360351562e-06,
      "step": 1081,
      "training_step_time": 7.0805723667144775
    },
    {
      "epoch": 1.0318756103515626e-06,
      "model_forward_time": 1.7172260284423828,
      "step": 1082
    },
    {
      "epoch": 1.0318756103515626e-06,
      "step": 1082,
      "training_step_time": 7.023071527481079
    },
    {
      "epoch": 1.0328292846679687e-06,
      "model_forward_time": 1.7150475978851318,
      "step": 1083
    },
    {
      "epoch": 1.0328292846679687e-06,
      "step": 1083,
      "training_step_time": 7.1060099601745605
    },
    {
      "epoch": 1.033782958984375e-06,
      "model_forward_time": 1.7109534740447998,
      "step": 1084
    },
    {
      "epoch": 1.033782958984375e-06,
      "step": 1084,
      "training_step_time": 7.093188047409058
    },
    {
      "epoch": 1.0347366333007812e-06,
      "model_forward_time": 1.7067968845367432,
      "step": 1085
    },
    {
      "epoch": 1.0347366333007812e-06,
      "step": 1085,
      "training_step_time": 8.648254632949829
    },
    {
      "epoch": 1.0356903076171876e-06,
      "model_forward_time": 2.2561981678009033,
      "step": 1086
    },
    {
      "epoch": 1.0356903076171876e-06,
      "step": 1086,
      "training_step_time": 10.902220249176025
    },
    {
      "epoch": 1.0366439819335937e-06,
      "model_forward_time": 1.7142064571380615,
      "step": 1087
    },
    {
      "epoch": 1.0366439819335937e-06,
      "step": 1087,
      "training_step_time": 9.002283811569214
    },
    {
      "epoch": 1.03759765625e-06,
      "model_forward_time": 1.711775302886963,
      "step": 1088
    },
    {
      "epoch": 1.03759765625e-06,
      "step": 1088,
      "training_step_time": 7.874554634094238
    },
    {
      "epoch": 1.0385513305664062e-06,
      "model_forward_time": 1.7245256900787354,
      "step": 1089
    },
    {
      "epoch": 1.0385513305664062e-06,
      "step": 1089,
      "training_step_time": 7.080361366271973
    },
    {
      "epoch": 1.0395050048828126e-06,
      "grad_norm": 1.9917091131210327,
      "learning_rate": 2.178e-06,
      "loss": 0.1394,
      "step": 1090
    },
    {
      "epoch": 1.0395050048828126e-06,
      "model_forward_time": 1.708420753479004,
      "step": 1090
    },
    {
      "dynamics_loss_avg": 0.09374657534062862,
      "epoch": 1.0395050048828126e-06,
      "step": 1090
    },
    {
      "action_loss_avg": 0.0657986331731081,
      "epoch": 1.0395050048828126e-06,
      "step": 1090
    },
    {
      "epoch": 1.0395050048828126e-06,
      "step": 1090,
      "training_step_time": 7.07318902015686
    },
    {
      "epoch": 1.0404586791992187e-06,
      "model_forward_time": 1.709242582321167,
      "step": 1091
    },
    {
      "epoch": 1.0404586791992187e-06,
      "step": 1091,
      "training_step_time": 7.084691524505615
    },
    {
      "epoch": 1.041412353515625e-06,
      "model_forward_time": 1.7079479694366455,
      "step": 1092
    },
    {
      "epoch": 1.041412353515625e-06,
      "step": 1092,
      "training_step_time": 7.066466569900513
    },
    {
      "epoch": 1.0423660278320312e-06,
      "model_forward_time": 1.7119865417480469,
      "step": 1093
    },
    {
      "epoch": 1.0423660278320312e-06,
      "step": 1093,
      "training_step_time": 6.996655225753784
    },
    {
      "epoch": 1.0433197021484376e-06,
      "model_forward_time": 1.7179312705993652,
      "step": 1094
    },
    {
      "epoch": 1.0433197021484376e-06,
      "step": 1094,
      "training_step_time": 8.384002685546875
    },
    {
      "epoch": 1.0442733764648437e-06,
      "model_forward_time": 2.2399613857269287,
      "step": 1095
    },
    {
      "epoch": 1.0442733764648437e-06,
      "step": 1095,
      "training_step_time": 9.87287712097168
    },
    {
      "epoch": 1.04522705078125e-06,
      "model_forward_time": 1.7100470066070557,
      "step": 1096
    },
    {
      "epoch": 1.04522705078125e-06,
      "step": 1096,
      "training_step_time": 7.989272832870483
    },
    {
      "epoch": 1.0461807250976562e-06,
      "model_forward_time": 1.7047679424285889,
      "step": 1097
    },
    {
      "epoch": 1.0461807250976562e-06,
      "step": 1097,
      "training_step_time": 7.928864002227783
    },
    {
      "epoch": 1.0471343994140626e-06,
      "model_forward_time": 1.7085556983947754,
      "step": 1098
    },
    {
      "epoch": 1.0471343994140626e-06,
      "step": 1098,
      "training_step_time": 7.0953919887542725
    },
    {
      "epoch": 1.0480880737304687e-06,
      "model_forward_time": 1.7111916542053223,
      "step": 1099
    },
    {
      "epoch": 1.0480880737304687e-06,
      "step": 1099,
      "training_step_time": 7.09211277961731
    },
    {
      "epoch": 1.049041748046875e-06,
      "grad_norm": 2.0338094234466553,
      "learning_rate": 2.198e-06,
      "loss": 0.1329,
      "step": 1100
    },
    {
      "epoch": 1.049041748046875e-06,
      "model_forward_time": 1.707690954208374,
      "step": 1100
    },
    {
      "dynamics_loss_avg": 0.09237723834812642,
      "epoch": 1.049041748046875e-06,
      "step": 1100
    },
    {
      "action_loss_avg": 0.058837201446294785,
      "epoch": 1.049041748046875e-06,
      "step": 1100
    },
    {
      "epoch": 1.049041748046875e-06,
      "step": 1100,
      "training_step_time": 7.056131362915039
    },
    {
      "epoch": 1.0499954223632812e-06,
      "model_forward_time": 1.7084617614746094,
      "step": 1101
    },
    {
      "epoch": 1.0499954223632812e-06,
      "step": 1101,
      "training_step_time": 7.068253040313721
    },
    {
      "epoch": 1.0509490966796876e-06,
      "model_forward_time": 1.7169055938720703,
      "step": 1102
    },
    {
      "epoch": 1.0509490966796876e-06,
      "step": 1102,
      "training_step_time": 7.100076913833618
    },
    {
      "epoch": 1.0519027709960937e-06,
      "model_forward_time": 1.7135372161865234,
      "step": 1103
    },
    {
      "epoch": 1.0519027709960937e-06,
      "step": 1103,
      "training_step_time": 8.177530288696289
    },
    {
      "epoch": 1.0528564453125e-06,
      "model_forward_time": 1.7187983989715576,
      "step": 1104
    },
    {
      "epoch": 1.0528564453125e-06,
      "step": 1104,
      "training_step_time": 11.149811506271362
    },
    {
      "epoch": 1.0538101196289062e-06,
      "model_forward_time": 2.234382152557373,
      "step": 1105
    },
    {
      "epoch": 1.0538101196289062e-06,
      "step": 1105,
      "training_step_time": 8.06184983253479
    },
    {
      "epoch": 1.0547637939453126e-06,
      "model_forward_time": 1.7134885787963867,
      "step": 1106
    },
    {
      "epoch": 1.0547637939453126e-06,
      "step": 1106,
      "training_step_time": 7.940118789672852
    },
    {
      "epoch": 1.0557174682617187e-06,
      "model_forward_time": 1.717271089553833,
      "step": 1107
    },
    {
      "epoch": 1.0557174682617187e-06,
      "step": 1107,
      "training_step_time": 7.1034629344940186
    },
    {
      "epoch": 1.056671142578125e-06,
      "model_forward_time": 1.7111928462982178,
      "step": 1108
    },
    {
      "epoch": 1.056671142578125e-06,
      "step": 1108,
      "training_step_time": 7.122272253036499
    },
    {
      "epoch": 1.0576248168945312e-06,
      "model_forward_time": 1.7104673385620117,
      "step": 1109
    },
    {
      "epoch": 1.0576248168945312e-06,
      "step": 1109,
      "training_step_time": 7.057352542877197
    },
    {
      "epoch": 1.0585784912109376e-06,
      "grad_norm": 3.643786907196045,
      "learning_rate": 2.218e-06,
      "loss": 0.1351,
      "step": 1110
    },
    {
      "epoch": 1.0585784912109376e-06,
      "model_forward_time": 1.7032511234283447,
      "step": 1110
    },
    {
      "dynamics_loss_avg": 0.09372815638780593,
      "epoch": 1.0585784912109376e-06,
      "step": 1110
    },
    {
      "action_loss_avg": 0.044246313720941545,
      "epoch": 1.0585784912109376e-06,
      "step": 1110
    },
    {
      "epoch": 1.0585784912109376e-06,
      "step": 1110,
      "training_step_time": 7.1042962074279785
    },
    {
      "epoch": 1.0595321655273437e-06,
      "model_forward_time": 1.72432279586792,
      "step": 1111
    },
    {
      "epoch": 1.0595321655273437e-06,
      "step": 1111,
      "training_step_time": 7.051634311676025
    },
    {
      "epoch": 1.06048583984375e-06,
      "model_forward_time": 1.7115669250488281,
      "step": 1112
    },
    {
      "epoch": 1.06048583984375e-06,
      "step": 1112,
      "training_step_time": 7.604922771453857
    },
    {
      "epoch": 1.0614395141601562e-06,
      "model_forward_time": 1.7098989486694336,
      "step": 1113
    },
    {
      "epoch": 1.0614395141601562e-06,
      "step": 1113,
      "training_step_time": 10.118630170822144
    },
    {
      "epoch": 1.0623931884765626e-06,
      "model_forward_time": 2.2405271530151367,
      "step": 1114
    },
    {
      "epoch": 1.0623931884765626e-06,
      "step": 1114,
      "training_step_time": 10.24716567993164
    },
    {
      "epoch": 1.0633468627929687e-06,
      "model_forward_time": 1.7197024822235107,
      "step": 1115
    },
    {
      "epoch": 1.0633468627929687e-06,
      "step": 1115,
      "training_step_time": 8.724365711212158
    },
    {
      "epoch": 1.064300537109375e-06,
      "model_forward_time": 1.7127649784088135,
      "step": 1116
    },
    {
      "epoch": 1.064300537109375e-06,
      "step": 1116,
      "training_step_time": 7.734246492385864
    },
    {
      "epoch": 1.0652542114257812e-06,
      "model_forward_time": 1.709639072418213,
      "step": 1117
    },
    {
      "epoch": 1.0652542114257812e-06,
      "step": 1117,
      "training_step_time": 7.115553140640259
    },
    {
      "epoch": 1.0662078857421876e-06,
      "model_forward_time": 1.718663215637207,
      "step": 1118
    },
    {
      "epoch": 1.0662078857421876e-06,
      "step": 1118,
      "training_step_time": 7.04020357131958
    },
    {
      "epoch": 1.0671615600585937e-06,
      "model_forward_time": 1.7044572830200195,
      "step": 1119
    },
    {
      "epoch": 1.0671615600585937e-06,
      "step": 1119,
      "training_step_time": 7.103213310241699
    },
    {
      "epoch": 1.068115234375e-06,
      "grad_norm": 2.909770965576172,
      "learning_rate": 2.238e-06,
      "loss": 0.1197,
      "step": 1120
    },
    {
      "epoch": 1.068115234375e-06,
      "model_forward_time": 1.7107431888580322,
      "step": 1120
    },
    {
      "dynamics_loss_avg": 0.07464651223272085,
      "epoch": 1.068115234375e-06,
      "step": 1120
    },
    {
      "action_loss_avg": 0.03360494421795011,
      "epoch": 1.068115234375e-06,
      "step": 1120
    },
    {
      "epoch": 1.068115234375e-06,
      "step": 1120,
      "training_step_time": 7.089945077896118
    },
    {
      "epoch": 1.0690689086914062e-06,
      "model_forward_time": 1.3416368961334229,
      "step": 1121
    },
    {
      "epoch": 1.0690689086914062e-06,
      "step": 1121,
      "training_step_time": 7.5420145988464355
    },
    {
      "epoch": 1.0700225830078126e-06,
      "model_forward_time": 1.7115178108215332,
      "step": 1122
    },
    {
      "epoch": 1.0700225830078126e-06,
      "step": 1122,
      "training_step_time": 8.14531397819519
    },
    {
      "epoch": 1.0709762573242187e-06,
      "model_forward_time": 2.2309892177581787,
      "step": 1123
    },
    {
      "epoch": 1.0709762573242187e-06,
      "step": 1123,
      "training_step_time": 9.972176313400269
    },
    {
      "epoch": 1.071929931640625e-06,
      "model_forward_time": 1.728663444519043,
      "step": 1124
    },
    {
      "epoch": 1.071929931640625e-06,
      "step": 1124,
      "training_step_time": 8.001549243927002
    },
    {
      "epoch": 1.0728836059570312e-06,
      "model_forward_time": 1.7083725929260254,
      "step": 1125
    },
    {
      "epoch": 1.0728836059570312e-06,
      "step": 1125,
      "training_step_time": 7.743587970733643
    },
    {
      "epoch": 1.0738372802734374e-06,
      "model_forward_time": 1.7196850776672363,
      "step": 1126
    },
    {
      "epoch": 1.0738372802734374e-06,
      "step": 1126,
      "training_step_time": 7.002805948257446
    },
    {
      "epoch": 1.0747909545898438e-06,
      "model_forward_time": 1.713385820388794,
      "step": 1127
    },
    {
      "epoch": 1.0747909545898438e-06,
      "step": 1127,
      "training_step_time": 6.924760103225708
    },
    {
      "epoch": 1.07574462890625e-06,
      "model_forward_time": 1.709247350692749,
      "step": 1128
    },
    {
      "epoch": 1.07574462890625e-06,
      "step": 1128,
      "training_step_time": 6.9574198722839355
    },
    {
      "epoch": 1.0766983032226563e-06,
      "model_forward_time": 1.7127923965454102,
      "step": 1129
    },
    {
      "epoch": 1.0766983032226563e-06,
      "step": 1129,
      "training_step_time": 6.896714210510254
    },
    {
      "epoch": 1.0776519775390624e-06,
      "grad_norm": 3.631967544555664,
      "learning_rate": 2.2580000000000002e-06,
      "loss": 0.1243,
      "step": 1130
    },
    {
      "epoch": 1.0776519775390624e-06,
      "model_forward_time": 1.702770709991455,
      "step": 1130
    },
    {
      "dynamics_loss_avg": 0.05630621369928122,
      "epoch": 1.0776519775390624e-06,
      "step": 1130
    },
    {
      "action_loss_avg": 0.038895906903780995,
      "epoch": 1.0776519775390624e-06,
      "step": 1130
    },
    {
      "epoch": 1.0776519775390624e-06,
      "step": 1130,
      "training_step_time": 7.407947778701782
    },
    {
      "epoch": 1.0786056518554688e-06,
      "model_forward_time": 1.709010362625122,
      "step": 1131
    },
    {
      "epoch": 1.0786056518554688e-06,
      "step": 1131,
      "training_step_time": 8.806004047393799
    },
    {
      "epoch": 1.079559326171875e-06,
      "model_forward_time": 2.2462852001190186,
      "step": 1132
    },
    {
      "epoch": 1.079559326171875e-06,
      "step": 1132,
      "training_step_time": 12.020473718643188
    },
    {
      "epoch": 1.0805130004882813e-06,
      "model_forward_time": 1.7052268981933594,
      "step": 1133
    },
    {
      "epoch": 1.0805130004882813e-06,
      "step": 1133,
      "training_step_time": 8.964171409606934
    },
    {
      "epoch": 1.0814666748046874e-06,
      "model_forward_time": 1.7017171382904053,
      "step": 1134
    },
    {
      "epoch": 1.0814666748046874e-06,
      "step": 1134,
      "training_step_time": 7.753251314163208
    },
    {
      "epoch": 1.0824203491210938e-06,
      "model_forward_time": 1.69980788230896,
      "step": 1135
    },
    {
      "epoch": 1.0824203491210938e-06,
      "step": 1135,
      "training_step_time": 6.984439373016357
    },
    {
      "epoch": 1.0833740234375e-06,
      "model_forward_time": 1.7046804428100586,
      "step": 1136
    },
    {
      "epoch": 1.0833740234375e-06,
      "step": 1136,
      "training_step_time": 6.9133758544921875
    },
    {
      "epoch": 1.0843276977539063e-06,
      "model_forward_time": 1.711754560470581,
      "step": 1137
    },
    {
      "epoch": 1.0843276977539063e-06,
      "step": 1137,
      "training_step_time": 6.895111322402954
    },
    {
      "epoch": 1.0852813720703124e-06,
      "model_forward_time": 1.7050666809082031,
      "step": 1138
    },
    {
      "epoch": 1.0852813720703124e-06,
      "step": 1138,
      "training_step_time": 6.955523490905762
    },
    {
      "epoch": 1.0862350463867188e-06,
      "model_forward_time": 1.7060492038726807,
      "step": 1139
    },
    {
      "epoch": 1.0862350463867188e-06,
      "step": 1139,
      "training_step_time": 7.4130823612213135
    },
    {
      "epoch": 1.087188720703125e-06,
      "grad_norm": 2.5957846641540527,
      "learning_rate": 2.278e-06,
      "loss": 0.1354,
      "step": 1140
    },
    {
      "epoch": 1.087188720703125e-06,
      "model_forward_time": 1.7030251026153564,
      "step": 1140
    },
    {
      "dynamics_loss_avg": 0.06819954812526703,
      "epoch": 1.087188720703125e-06,
      "step": 1140
    },
    {
      "action_loss_avg": 0.0448031960055232,
      "epoch": 1.087188720703125e-06,
      "step": 1140
    },
    {
      "epoch": 1.087188720703125e-06,
      "step": 1140,
      "training_step_time": 7.810365438461304
    },
    {
      "epoch": 1.0881423950195313e-06,
      "model_forward_time": 2.2432942390441895,
      "step": 1141
    },
    {
      "epoch": 1.0881423950195313e-06,
      "step": 1141,
      "training_step_time": 9.419824600219727
    },
    {
      "epoch": 1.0890960693359374e-06,
      "model_forward_time": 1.7047805786132812,
      "step": 1142
    },
    {
      "epoch": 1.0890960693359374e-06,
      "step": 1142,
      "training_step_time": 8.564908027648926
    },
    {
      "epoch": 1.0900497436523438e-06,
      "model_forward_time": 1.7070972919464111,
      "step": 1143
    },
    {
      "epoch": 1.0900497436523438e-06,
      "step": 1143,
      "training_step_time": 7.7907233238220215
    },
    {
      "epoch": 1.09100341796875e-06,
      "model_forward_time": 1.7067043781280518,
      "step": 1144
    },
    {
      "epoch": 1.09100341796875e-06,
      "step": 1144,
      "training_step_time": 6.8585052490234375
    },
    {
      "epoch": 1.0919570922851563e-06,
      "model_forward_time": 1.7062313556671143,
      "step": 1145
    },
    {
      "epoch": 1.0919570922851563e-06,
      "step": 1145,
      "training_step_time": 6.925754547119141
    },
    {
      "epoch": 1.0929107666015624e-06,
      "model_forward_time": 1.702892780303955,
      "step": 1146
    },
    {
      "epoch": 1.0929107666015624e-06,
      "step": 1146,
      "training_step_time": 6.937241792678833
    },
    {
      "epoch": 1.0938644409179688e-06,
      "model_forward_time": 1.709841251373291,
      "step": 1147
    },
    {
      "epoch": 1.0938644409179688e-06,
      "step": 1147,
      "training_step_time": 6.8958940505981445
    },
    {
      "epoch": 1.094818115234375e-06,
      "model_forward_time": 1.703230381011963,
      "step": 1148
    },
    {
      "epoch": 1.094818115234375e-06,
      "step": 1148,
      "training_step_time": 7.358911752700806
    },
    {
      "epoch": 1.0957717895507813e-06,
      "model_forward_time": 1.7053477764129639,
      "step": 1149
    },
    {
      "epoch": 1.0957717895507813e-06,
      "step": 1149,
      "training_step_time": 7.913794994354248
    },
    {
      "epoch": 1.0967254638671874e-06,
      "grad_norm": 2.1958587169647217,
      "learning_rate": 2.2980000000000003e-06,
      "loss": 0.1176,
      "step": 1150
    },
    {
      "epoch": 1.0967254638671874e-06,
      "model_forward_time": 1.708003282546997,
      "step": 1150
    },
    {
      "dynamics_loss_avg": 0.08139044381678104,
      "epoch": 1.0967254638671874e-06,
      "step": 1150
    },
    {
      "action_loss_avg": 0.05134138017892838,
      "epoch": 1.0967254638671874e-06,
      "step": 1150
    },
    {
      "epoch": 1.0967254638671874e-06,
      "step": 1150,
      "training_step_time": 7.618534088134766
    },
    {
      "epoch": 1.0976791381835938e-06,
      "model_forward_time": 2.2275524139404297,
      "step": 1151
    },
    {
      "epoch": 1.0976791381835938e-06,
      "step": 1151,
      "training_step_time": 10.955403327941895
    },
    {
      "epoch": 1.0986328125e-06,
      "model_forward_time": 1.7066161632537842,
      "step": 1152
    },
    {
      "epoch": 1.0986328125e-06,
      "step": 1152,
      "training_step_time": 8.820752382278442
    },
    {
      "epoch": 1.0995864868164063e-06,
      "model_forward_time": 1.703942060470581,
      "step": 1153
    },
    {
      "epoch": 1.0995864868164063e-06,
      "step": 1153,
      "training_step_time": 6.934662342071533
    },
    {
      "epoch": 1.1005401611328124e-06,
      "model_forward_time": 1.3341989517211914,
      "step": 1154
    },
    {
      "epoch": 1.1005401611328124e-06,
      "step": 1154,
      "training_step_time": 6.883335113525391
    },
    {
      "epoch": 1.1014938354492188e-06,
      "model_forward_time": 1.710702657699585,
      "step": 1155
    },
    {
      "epoch": 1.1014938354492188e-06,
      "step": 1155,
      "training_step_time": 6.92609167098999
    },
    {
      "epoch": 1.102447509765625e-06,
      "model_forward_time": 1.712151050567627,
      "step": 1156
    },
    {
      "epoch": 1.102447509765625e-06,
      "step": 1156,
      "training_step_time": 6.946538209915161
    },
    {
      "epoch": 1.1034011840820313e-06,
      "model_forward_time": 1.703763723373413,
      "step": 1157
    },
    {
      "epoch": 1.1034011840820313e-06,
      "step": 1157,
      "training_step_time": 7.43876051902771
    },
    {
      "epoch": 1.1043548583984374e-06,
      "model_forward_time": 1.7033581733703613,
      "step": 1158
    },
    {
      "epoch": 1.1043548583984374e-06,
      "step": 1158,
      "training_step_time": 8.804364919662476
    },
    {
      "epoch": 1.1053085327148438e-06,
      "model_forward_time": 1.7036733627319336,
      "step": 1159
    },
    {
      "epoch": 1.1053085327148438e-06,
      "step": 1159,
      "training_step_time": 7.6422343254089355
    },
    {
      "epoch": 1.10626220703125e-06,
      "grad_norm": 1.9721763134002686,
      "learning_rate": 2.318e-06,
      "loss": 0.1145,
      "step": 1160
    },
    {
      "epoch": 1.10626220703125e-06,
      "model_forward_time": 2.248666763305664,
      "step": 1160
    },
    {
      "dynamics_loss_avg": 0.08208473809063435,
      "epoch": 1.10626220703125e-06,
      "step": 1160
    },
    {
      "action_loss_avg": 0.046285297721624374,
      "epoch": 1.10626220703125e-06,
      "step": 1160
    },
    {
      "epoch": 1.10626220703125e-06,
      "step": 1160,
      "training_step_time": 9.369356155395508
    },
    {
      "epoch": 1.1072158813476563e-06,
      "model_forward_time": 1.7058286666870117,
      "step": 1161
    },
    {
      "epoch": 1.1072158813476563e-06,
      "step": 1161,
      "training_step_time": 7.900902271270752
    },
    {
      "epoch": 1.1081695556640624e-06,
      "model_forward_time": 1.70285964012146,
      "step": 1162
    },
    {
      "epoch": 1.1081695556640624e-06,
      "step": 1162,
      "training_step_time": 7.736220836639404
    },
    {
      "epoch": 1.1091232299804688e-06,
      "model_forward_time": 1.7114932537078857,
      "step": 1163
    },
    {
      "epoch": 1.1091232299804688e-06,
      "step": 1163,
      "training_step_time": 6.902868986129761
    },
    {
      "epoch": 1.110076904296875e-06,
      "model_forward_time": 1.7102088928222656,
      "step": 1164
    },
    {
      "epoch": 1.110076904296875e-06,
      "step": 1164,
      "training_step_time": 6.924161672592163
    },
    {
      "epoch": 1.1110305786132813e-06,
      "model_forward_time": 1.7033581733703613,
      "step": 1165
    },
    {
      "epoch": 1.1110305786132813e-06,
      "step": 1165,
      "training_step_time": 6.923652410507202
    },
    {
      "epoch": 1.1119842529296874e-06,
      "model_forward_time": 1.7099359035491943,
      "step": 1166
    },
    {
      "epoch": 1.1119842529296874e-06,
      "step": 1166,
      "training_step_time": 7.431460618972778
    },
    {
      "epoch": 1.1129379272460938e-06,
      "model_forward_time": 1.7070283889770508,
      "step": 1167
    },
    {
      "epoch": 1.1129379272460938e-06,
      "step": 1167,
      "training_step_time": 8.891589641571045
    },
    {
      "epoch": 1.1138916015625e-06,
      "model_forward_time": 1.6913292407989502,
      "step": 1168
    },
    {
      "epoch": 1.1138916015625e-06,
      "step": 1168,
      "training_step_time": 7.451446771621704
    },
    {
      "epoch": 1.1148452758789063e-06,
      "model_forward_time": 2.249951124191284,
      "step": 1169
    },
    {
      "epoch": 1.1148452758789063e-06,
      "step": 1169,
      "training_step_time": 9.038321256637573
    },
    {
      "epoch": 1.1157989501953124e-06,
      "grad_norm": 2.4219019412994385,
      "learning_rate": 2.3380000000000003e-06,
      "loss": 0.1228,
      "step": 1170
    },
    {
      "epoch": 1.1157989501953124e-06,
      "model_forward_time": 1.7033770084381104,
      "step": 1170
    },
    {
      "dynamics_loss_avg": 0.07291396446526051,
      "epoch": 1.1157989501953124e-06,
      "step": 1170
    },
    {
      "action_loss_avg": 0.042213368508964774,
      "epoch": 1.1157989501953124e-06,
      "step": 1170
    },
    {
      "epoch": 1.1157989501953124e-06,
      "step": 1170,
      "training_step_time": 8.584645509719849
    },
    {
      "epoch": 1.1167526245117188e-06,
      "model_forward_time": 1.7068791389465332,
      "step": 1171
    },
    {
      "epoch": 1.1167526245117188e-06,
      "step": 1171,
      "training_step_time": 7.838610649108887
    },
    {
      "epoch": 1.117706298828125e-06,
      "model_forward_time": 1.7127048969268799,
      "step": 1172
    },
    {
      "epoch": 1.117706298828125e-06,
      "step": 1172,
      "training_step_time": 7.059045314788818
    },
    {
      "epoch": 1.1186599731445313e-06,
      "model_forward_time": 1.7070555686950684,
      "step": 1173
    },
    {
      "epoch": 1.1186599731445313e-06,
      "step": 1173,
      "training_step_time": 7.0555102825164795
    },
    {
      "epoch": 1.1196136474609374e-06,
      "model_forward_time": 1.7059924602508545,
      "step": 1174
    },
    {
      "epoch": 1.1196136474609374e-06,
      "step": 1174,
      "training_step_time": 7.046316385269165
    },
    {
      "epoch": 1.1205673217773438e-06,
      "model_forward_time": 1.706397533416748,
      "step": 1175
    },
    {
      "epoch": 1.1205673217773438e-06,
      "step": 1175,
      "training_step_time": 7.549703598022461
    },
    {
      "epoch": 1.12152099609375e-06,
      "model_forward_time": 1.7158992290496826,
      "step": 1176
    },
    {
      "epoch": 1.12152099609375e-06,
      "step": 1176,
      "training_step_time": 8.12691879272461
    },
    {
      "epoch": 1.1224746704101563e-06,
      "model_forward_time": 1.7025797367095947,
      "step": 1177
    },
    {
      "epoch": 1.1224746704101563e-06,
      "step": 1177,
      "training_step_time": 7.755654573440552
    },
    {
      "epoch": 1.1234283447265624e-06,
      "model_forward_time": 2.2553322315216064,
      "step": 1178
    },
    {
      "epoch": 1.1234283447265624e-06,
      "step": 1178,
      "training_step_time": 9.400723934173584
    },
    {
      "epoch": 1.1243820190429688e-06,
      "model_forward_time": 1.7063045501708984,
      "step": 1179
    },
    {
      "epoch": 1.1243820190429688e-06,
      "step": 1179,
      "training_step_time": 8.067142486572266
    },
    {
      "epoch": 1.125335693359375e-06,
      "grad_norm": 2.2104666233062744,
      "learning_rate": 2.358e-06,
      "loss": 0.1115,
      "step": 1180
    },
    {
      "epoch": 1.125335693359375e-06,
      "model_forward_time": 1.701564073562622,
      "step": 1180
    },
    {
      "dynamics_loss_avg": 0.08130561746656895,
      "epoch": 1.125335693359375e-06,
      "step": 1180
    },
    {
      "action_loss_avg": 0.06315772458910943,
      "epoch": 1.125335693359375e-06,
      "step": 1180
    },
    {
      "epoch": 1.125335693359375e-06,
      "step": 1180,
      "training_step_time": 7.883309841156006
    },
    {
      "epoch": 1.1262893676757813e-06,
      "model_forward_time": 1.7095916271209717,
      "step": 1181
    },
    {
      "epoch": 1.1262893676757813e-06,
      "step": 1181,
      "training_step_time": 7.101574182510376
    },
    {
      "epoch": 1.1272430419921874e-06,
      "model_forward_time": 1.7029576301574707,
      "step": 1182
    },
    {
      "epoch": 1.1272430419921874e-06,
      "step": 1182,
      "training_step_time": 7.086329936981201
    },
    {
      "epoch": 1.1281967163085938e-06,
      "model_forward_time": 1.7033369541168213,
      "step": 1183
    },
    {
      "epoch": 1.1281967163085938e-06,
      "step": 1183,
      "training_step_time": 7.097941160202026
    },
    {
      "epoch": 1.129150390625e-06,
      "model_forward_time": 1.7002654075622559,
      "step": 1184
    },
    {
      "epoch": 1.129150390625e-06,
      "step": 1184,
      "training_step_time": 7.565277576446533
    },
    {
      "epoch": 1.1301040649414063e-06,
      "model_forward_time": 1.7099802494049072,
      "step": 1185
    },
    {
      "epoch": 1.1301040649414063e-06,
      "step": 1185,
      "training_step_time": 7.996001243591309
    },
    {
      "epoch": 1.1310577392578124e-06,
      "model_forward_time": 1.7004799842834473,
      "step": 1186
    },
    {
      "epoch": 1.1310577392578124e-06,
      "step": 1186,
      "training_step_time": 7.66818904876709
    },
    {
      "epoch": 1.1320114135742188e-06,
      "model_forward_time": 1.7426347732543945,
      "step": 1187
    },
    {
      "epoch": 1.1320114135742188e-06,
      "step": 1187,
      "training_step_time": 10.457153081893921
    },
    {
      "epoch": 1.132965087890625e-06,
      "model_forward_time": 1.6990199089050293,
      "step": 1188
    },
    {
      "epoch": 1.132965087890625e-06,
      "step": 1188,
      "training_step_time": 8.544282674789429
    },
    {
      "epoch": 1.1339187622070313e-06,
      "model_forward_time": 1.7152109146118164,
      "step": 1189
    },
    {
      "epoch": 1.1339187622070313e-06,
      "step": 1189,
      "training_step_time": 7.977982521057129
    },
    {
      "epoch": 1.1348724365234374e-06,
      "grad_norm": 2.1997315883636475,
      "learning_rate": 2.3780000000000004e-06,
      "loss": 0.1102,
      "step": 1190
    },
    {
      "epoch": 1.1348724365234374e-06,
      "model_forward_time": 1.6971909999847412,
      "step": 1190
    },
    {
      "dynamics_loss_avg": 0.07622184474021196,
      "epoch": 1.1348724365234374e-06,
      "step": 1190
    },
    {
      "action_loss_avg": 0.04800773551687598,
      "epoch": 1.1348724365234374e-06,
      "step": 1190
    },
    {
      "epoch": 1.1348724365234374e-06,
      "step": 1190,
      "training_step_time": 7.47684383392334
    },
    {
      "epoch": 1.1358261108398438e-06,
      "model_forward_time": 1.706758975982666,
      "step": 1191
    },
    {
      "epoch": 1.1358261108398438e-06,
      "step": 1191,
      "training_step_time": 7.1014628410339355
    },
    {
      "epoch": 1.13677978515625e-06,
      "model_forward_time": 1.7111403942108154,
      "step": 1192
    },
    {
      "epoch": 1.13677978515625e-06,
      "step": 1192,
      "training_step_time": 7.031283855438232
    },
    {
      "epoch": 1.1377334594726563e-06,
      "model_forward_time": 1.7040724754333496,
      "step": 1193
    },
    {
      "epoch": 1.1377334594726563e-06,
      "step": 1193,
      "training_step_time": 7.534799337387085
    },
    {
      "epoch": 1.1386871337890625e-06,
      "model_forward_time": 1.7022063732147217,
      "step": 1194
    },
    {
      "epoch": 1.1386871337890625e-06,
      "step": 1194,
      "training_step_time": 7.190501928329468
    },
    {
      "epoch": 1.1396408081054688e-06,
      "model_forward_time": 1.7114841938018799,
      "step": 1195
    },
    {
      "epoch": 1.1396408081054688e-06,
      "step": 1195,
      "training_step_time": 8.10670256614685
    },
    {
      "epoch": 1.140594482421875e-06,
      "model_forward_time": 1.7415289878845215,
      "step": 1196
    },
    {
      "epoch": 1.140594482421875e-06,
      "step": 1196,
      "training_step_time": 9.428551435470581
    },
    {
      "epoch": 1.1415481567382813e-06,
      "model_forward_time": 1.7124512195587158,
      "step": 1197
    },
    {
      "epoch": 1.1415481567382813e-06,
      "step": 1197,
      "training_step_time": 9.090019702911377
    },
    {
      "epoch": 1.1425018310546875e-06,
      "model_forward_time": 1.7158353328704834,
      "step": 1198
    },
    {
      "epoch": 1.1425018310546875e-06,
      "step": 1198,
      "training_step_time": 7.952972650527954
    },
    {
      "epoch": 1.1434555053710938e-06,
      "model_forward_time": 1.718224287033081,
      "step": 1199
    },
    {
      "epoch": 1.1434555053710938e-06,
      "step": 1199,
      "training_step_time": 7.762944221496582
    },
    {
      "epoch": 1.1444091796875e-06,
      "grad_norm": 2.7025187015533447,
      "learning_rate": 2.398e-06,
      "loss": 0.1245,
      "step": 1200
    },
    {
      "epoch": 1.1444091796875e-06,
      "model_forward_time": 1.7245151996612549,
      "step": 1200
    },
    {
      "dynamics_loss_avg": 0.07505987696349621,
      "epoch": 1.1444091796875e-06,
      "step": 1200
    },
    {
      "action_loss_avg": 0.045589125156402587,
      "epoch": 1.1444091796875e-06,
      "step": 1200
    },
    {
      "epoch": 1.1444091796875e-06,
      "step": 1200,
      "training_step_time": 7.015176057815552
    },
    {
      "epoch": 1.1453628540039063e-06,
      "model_forward_time": 1.7182211875915527,
      "step": 1201
    },
    {
      "epoch": 1.1453628540039063e-06,
      "step": 1201,
      "training_step_time": 6.979884624481201
    },
    {
      "epoch": 1.1463165283203125e-06,
      "model_forward_time": 1.7216100692749023,
      "step": 1202
    },
    {
      "epoch": 1.1463165283203125e-06,
      "step": 1202,
      "training_step_time": 7.554502725601196
    },
    {
      "epoch": 1.1472702026367188e-06,
      "model_forward_time": 1.7223989963531494,
      "step": 1203
    },
    {
      "epoch": 1.1472702026367188e-06,
      "step": 1203,
      "training_step_time": 7.297762870788574
    },
    {
      "epoch": 1.148223876953125e-06,
      "model_forward_time": 1.7197144031524658,
      "step": 1204
    },
    {
      "epoch": 1.148223876953125e-06,
      "step": 1204,
      "training_step_time": 8.805509567260742
    },
    {
      "epoch": 1.1491775512695313e-06,
      "model_forward_time": 1.7185750007629395,
      "step": 1205
    },
    {
      "epoch": 1.1491775512695313e-06,
      "step": 1205,
      "training_step_time": 8.889040470123291
    },
    {
      "epoch": 1.1501312255859375e-06,
      "model_forward_time": 1.720362901687622,
      "step": 1206
    },
    {
      "epoch": 1.1501312255859375e-06,
      "step": 1206,
      "training_step_time": 10.335408926010132
    },
    {
      "epoch": 1.1510848999023438e-06,
      "model_forward_time": 1.7219593524932861,
      "step": 1207
    },
    {
      "epoch": 1.1510848999023438e-06,
      "step": 1207,
      "training_step_time": 7.971012592315674
    },
    {
      "epoch": 1.15203857421875e-06,
      "model_forward_time": 1.7378206253051758,
      "step": 1208
    },
    {
      "epoch": 1.15203857421875e-06,
      "step": 1208,
      "training_step_time": 7.821977853775024
    },
    {
      "epoch": 1.1529922485351563e-06,
      "model_forward_time": 1.7191181182861328,
      "step": 1209
    },
    {
      "epoch": 1.1529922485351563e-06,
      "step": 1209,
      "training_step_time": 6.952715635299683
    },
    {
      "epoch": 1.1539459228515625e-06,
      "grad_norm": 3.710550308227539,
      "learning_rate": 2.418e-06,
      "loss": 0.128,
      "step": 1210
    },
    {
      "epoch": 1.1539459228515625e-06,
      "model_forward_time": 1.7205700874328613,
      "step": 1210
    },
    {
      "dynamics_loss_avg": 0.0804579684510827,
      "epoch": 1.1539459228515625e-06,
      "step": 1210
    },
    {
      "action_loss_avg": 0.05517721716314554,
      "epoch": 1.1539459228515625e-06,
      "step": 1210
    },
    {
      "epoch": 1.1539459228515625e-06,
      "step": 1210,
      "training_step_time": 6.9305267333984375
    },
    {
      "epoch": 1.1548995971679688e-06,
      "model_forward_time": 1.7285418510437012,
      "step": 1211
    },
    {
      "epoch": 1.1548995971679688e-06,
      "step": 1211,
      "training_step_time": 7.4579198360443115
    },
    {
      "epoch": 1.155853271484375e-06,
      "model_forward_time": 1.7169301509857178,
      "step": 1212
    },
    {
      "epoch": 1.155853271484375e-06,
      "step": 1212,
      "training_step_time": 7.176608324050903
    },
    {
      "epoch": 1.1568069458007813e-06,
      "model_forward_time": 1.7176358699798584,
      "step": 1213
    },
    {
      "epoch": 1.1568069458007813e-06,
      "step": 1213,
      "training_step_time": 8.032090663909912
    },
    {
      "epoch": 1.1577606201171875e-06,
      "model_forward_time": 1.722325325012207,
      "step": 1214
    },
    {
      "epoch": 1.1577606201171875e-06,
      "step": 1214,
      "training_step_time": 9.087180137634277
    },
    {
      "epoch": 1.1587142944335938e-06,
      "model_forward_time": 2.2886879444122314,
      "step": 1215
    },
    {
      "epoch": 1.1587142944335938e-06,
      "step": 1215,
      "training_step_time": 10.728930234909058
    },
    {
      "epoch": 1.15966796875e-06,
      "model_forward_time": 1.7193543910980225,
      "step": 1216
    },
    {
      "epoch": 1.15966796875e-06,
      "step": 1216,
      "training_step_time": 8.837169885635376
    },
    {
      "epoch": 1.1606216430664063e-06,
      "model_forward_time": 1.719956636428833,
      "step": 1217
    },
    {
      "epoch": 1.1606216430664063e-06,
      "step": 1217,
      "training_step_time": 7.835310935974121
    },
    {
      "epoch": 1.1615753173828125e-06,
      "model_forward_time": 1.7152154445648193,
      "step": 1218
    },
    {
      "epoch": 1.1615753173828125e-06,
      "step": 1218,
      "training_step_time": 6.939758062362671
    },
    {
      "epoch": 1.1625289916992188e-06,
      "model_forward_time": 1.7185993194580078,
      "step": 1219
    },
    {
      "epoch": 1.1625289916992188e-06,
      "step": 1219,
      "training_step_time": 7.031326532363892
    },
    {
      "epoch": 1.163482666015625e-06,
      "grad_norm": 3.1906051635742188,
      "learning_rate": 2.438e-06,
      "loss": 0.1306,
      "step": 1220
    },
    {
      "epoch": 1.163482666015625e-06,
      "model_forward_time": 1.7306346893310547,
      "step": 1220
    },
    {
      "dynamics_loss_avg": 0.08559233881533146,
      "epoch": 1.163482666015625e-06,
      "step": 1220
    },
    {
      "action_loss_avg": 0.049556266888976096,
      "epoch": 1.163482666015625e-06,
      "step": 1220
    },
    {
      "epoch": 1.163482666015625e-06,
      "step": 1220,
      "training_step_time": 7.4498021602630615
    },
    {
      "epoch": 1.1644363403320313e-06,
      "model_forward_time": 1.7194528579711914,
      "step": 1221
    },
    {
      "epoch": 1.1644363403320313e-06,
      "step": 1221,
      "training_step_time": 7.011692523956299
    },
    {
      "epoch": 1.1653900146484375e-06,
      "model_forward_time": 1.7159013748168945,
      "step": 1222
    },
    {
      "epoch": 1.1653900146484375e-06,
      "step": 1222,
      "training_step_time": 7.508603572845459
    },
    {
      "epoch": 1.1663436889648438e-06,
      "model_forward_time": 1.717726707458496,
      "step": 1223
    },
    {
      "epoch": 1.1663436889648438e-06,
      "step": 1223,
      "training_step_time": 8.146760940551758
    },
    {
      "epoch": 1.16729736328125e-06,
      "model_forward_time": 1.946669101715088,
      "step": 1224
    },
    {
      "epoch": 1.16729736328125e-06,
      "step": 1224,
      "training_step_time": 10.720806121826172
    },
    {
      "epoch": 1.1682510375976563e-06,
      "model_forward_time": 1.7158634662628174,
      "step": 1225
    },
    {
      "epoch": 1.1682510375976563e-06,
      "step": 1225,
      "training_step_time": 8.042975664138794
    },
    {
      "epoch": 1.1692047119140625e-06,
      "model_forward_time": 1.727018117904663,
      "step": 1226
    },
    {
      "epoch": 1.1692047119140625e-06,
      "step": 1226,
      "training_step_time": 8.4832444190979
    },
    {
      "epoch": 1.1701583862304688e-06,
      "model_forward_time": 1.7175259590148926,
      "step": 1227
    },
    {
      "epoch": 1.1701583862304688e-06,
      "step": 1227,
      "training_step_time": 7.6436450481414795
    },
    {
      "epoch": 1.171112060546875e-06,
      "model_forward_time": 1.7149622440338135,
      "step": 1228
    },
    {
      "epoch": 1.171112060546875e-06,
      "step": 1228,
      "training_step_time": 7.025122404098511
    },
    {
      "epoch": 1.1720657348632813e-06,
      "model_forward_time": 1.72263503074646,
      "step": 1229
    },
    {
      "epoch": 1.1720657348632813e-06,
      "step": 1229,
      "training_step_time": 7.489471435546875
    },
    {
      "epoch": 1.1730194091796875e-06,
      "grad_norm": 3.193929672241211,
      "learning_rate": 2.458e-06,
      "loss": 0.1335,
      "step": 1230
    },
    {
      "epoch": 1.1730194091796875e-06,
      "model_forward_time": 1.7175781726837158,
      "step": 1230
    },
    {
      "dynamics_loss_avg": 0.08020945098251105,
      "epoch": 1.1730194091796875e-06,
      "step": 1230
    },
    {
      "action_loss_avg": 0.049327298812568185,
      "epoch": 1.1730194091796875e-06,
      "step": 1230
    },
    {
      "epoch": 1.1730194091796875e-06,
      "step": 1230,
      "training_step_time": 6.939690828323364
    },
    {
      "epoch": 1.1739730834960938e-06,
      "model_forward_time": 1.7112939357757568,
      "step": 1231
    },
    {
      "epoch": 1.1739730834960938e-06,
      "step": 1231,
      "training_step_time": 7.53481388092041
    },
    {
      "epoch": 1.1749267578125e-06,
      "model_forward_time": 1.7234327793121338,
      "step": 1232
    },
    {
      "epoch": 1.1749267578125e-06,
      "step": 1232,
      "training_step_time": 8.045853614807129
    },
    {
      "epoch": 1.1758804321289063e-06,
      "model_forward_time": 2.3238914012908936,
      "step": 1233
    },
    {
      "epoch": 1.1758804321289063e-06,
      "step": 1233,
      "training_step_time": 9.260710000991821
    },
    {
      "epoch": 1.1768341064453125e-06,
      "model_forward_time": 1.7231807708740234,
      "step": 1234
    },
    {
      "epoch": 1.1768341064453125e-06,
      "step": 1234,
      "training_step_time": 8.948975086212158
    },
    {
      "epoch": 1.1777877807617188e-06,
      "model_forward_time": 1.724900245666504,
      "step": 1235
    },
    {
      "epoch": 1.1777877807617188e-06,
      "step": 1235,
      "training_step_time": 9.480973243713379
    },
    {
      "epoch": 1.178741455078125e-06,
      "model_forward_time": 1.7292745113372803,
      "step": 1236
    },
    {
      "epoch": 1.178741455078125e-06,
      "step": 1236,
      "training_step_time": 7.628130912780762
    },
    {
      "epoch": 1.1796951293945313e-06,
      "model_forward_time": 1.7270119190216064,
      "step": 1237
    },
    {
      "epoch": 1.1796951293945313e-06,
      "step": 1237,
      "training_step_time": 6.923339366912842
    },
    {
      "epoch": 1.1806488037109375e-06,
      "model_forward_time": 1.7232837677001953,
      "step": 1238
    },
    {
      "epoch": 1.1806488037109375e-06,
      "step": 1238,
      "training_step_time": 7.571263074874878
    },
    {
      "epoch": 1.1816024780273438e-06,
      "model_forward_time": 1.7205095291137695,
      "step": 1239
    },
    {
      "epoch": 1.1816024780273438e-06,
      "step": 1239,
      "training_step_time": 6.998308897018433
    },
    {
      "epoch": 1.18255615234375e-06,
      "grad_norm": 2.7324349880218506,
      "learning_rate": 2.4780000000000002e-06,
      "loss": 0.1263,
      "step": 1240
    },
    {
      "epoch": 1.18255615234375e-06,
      "model_forward_time": 1.7250525951385498,
      "step": 1240
    },
    {
      "dynamics_loss_avg": 0.08051232565194369,
      "epoch": 1.18255615234375e-06,
      "step": 1240
    },
    {
      "action_loss_avg": 0.04362300299108028,
      "epoch": 1.18255615234375e-06,
      "step": 1240
    },
    {
      "epoch": 1.18255615234375e-06,
      "step": 1240,
      "training_step_time": 7.542438507080078
    },
    {
      "epoch": 1.1835098266601563e-06,
      "model_forward_time": 1.7237741947174072,
      "step": 1241
    },
    {
      "epoch": 1.1835098266601563e-06,
      "step": 1241,
      "training_step_time": 8.874189138412476
    },
    {
      "epoch": 1.1844635009765625e-06,
      "model_forward_time": 2.3168838024139404,
      "step": 1242
    },
    {
      "epoch": 1.1844635009765625e-06,
      "step": 1242,
      "training_step_time": 8.545127868652344
    },
    {
      "epoch": 1.1854171752929688e-06,
      "model_forward_time": 1.7188012599945068,
      "step": 1243
    },
    {
      "epoch": 1.1854171752929688e-06,
      "step": 1243,
      "training_step_time": 8.05363130569458
    },
    {
      "epoch": 1.186370849609375e-06,
      "model_forward_time": 1.7219984531402588,
      "step": 1244
    },
    {
      "epoch": 1.186370849609375e-06,
      "step": 1244,
      "training_step_time": 7.833555221557617
    },
    {
      "epoch": 1.1873245239257814e-06,
      "model_forward_time": 1.731837511062622,
      "step": 1245
    },
    {
      "epoch": 1.1873245239257814e-06,
      "step": 1245,
      "training_step_time": 7.783994674682617
    },
    {
      "epoch": 1.1882781982421875e-06,
      "model_forward_time": 1.725874662399292,
      "step": 1246
    },
    {
      "epoch": 1.1882781982421875e-06,
      "step": 1246,
      "training_step_time": 6.861609220504761
    },
    {
      "epoch": 1.1892318725585939e-06,
      "model_forward_time": 1.7272932529449463,
      "step": 1247
    },
    {
      "epoch": 1.1892318725585939e-06,
      "step": 1247,
      "training_step_time": 7.477009296417236
    },
    {
      "epoch": 1.190185546875e-06,
      "model_forward_time": 1.7306907176971436,
      "step": 1248
    },
    {
      "epoch": 1.190185546875e-06,
      "step": 1248,
      "training_step_time": 6.893150806427002
    },
    {
      "epoch": 1.1911392211914064e-06,
      "model_forward_time": 1.721980094909668,
      "step": 1249
    },
    {
      "epoch": 1.1911392211914064e-06,
      "step": 1249,
      "training_step_time": 7.474268913269043
    },
    {
      "epoch": 1.1920928955078125e-06,
      "grad_norm": 1.7823868989944458,
      "learning_rate": 2.498e-06,
      "loss": 0.1366,
      "step": 1250
    },
    {
      "epoch": 1.1920928955078125e-06,
      "model_forward_time": 1.7165179252624512,
      "step": 1250
    },
    {
      "dynamics_loss_avg": 0.06345563940703869,
      "epoch": 1.1920928955078125e-06,
      "step": 1250
    },
    {
      "action_loss_avg": 0.03054672833532095,
      "epoch": 1.1920928955078125e-06,
      "step": 1250
    },
    {
      "epoch": 1.1920928955078125e-06,
      "step": 1250,
      "training_step_time": 8.023988246917725
    },
    {
      "epoch": 1.1930465698242186e-06,
      "model_forward_time": 2.2821505069732666,
      "step": 1251
    },
    {
      "epoch": 1.1930465698242186e-06,
      "step": 1251,
      "training_step_time": 8.71035361289978
    },
    {
      "epoch": 1.194000244140625e-06,
      "model_forward_time": 1.366769552230835,
      "step": 1252
    },
    {
      "epoch": 1.194000244140625e-06,
      "step": 1252,
      "training_step_time": 8.794033288955688
    },
    {
      "epoch": 1.1949539184570311e-06,
      "model_forward_time": 1.7083959579467773,
      "step": 1253
    },
    {
      "epoch": 1.1949539184570311e-06,
      "step": 1253,
      "training_step_time": 8.805039167404175
    },
    {
      "epoch": 1.1959075927734375e-06,
      "model_forward_time": 1.705613374710083,
      "step": 1254
    },
    {
      "epoch": 1.1959075927734375e-06,
      "step": 1254,
      "training_step_time": 7.75605583190918
    },
    {
      "epoch": 1.1968612670898436e-06,
      "model_forward_time": 1.7055554389953613,
      "step": 1255
    },
    {
      "epoch": 1.1968612670898436e-06,
      "step": 1255,
      "training_step_time": 7.218443155288696
    },
    {
      "epoch": 1.19781494140625e-06,
      "model_forward_time": 1.7138926982879639,
      "step": 1256
    },
    {
      "epoch": 1.19781494140625e-06,
      "step": 1256,
      "training_step_time": 7.447510480880737
    },
    {
      "epoch": 1.1987686157226562e-06,
      "model_forward_time": 1.688509225845337,
      "step": 1257
    },
    {
      "epoch": 1.1987686157226562e-06,
      "step": 1257,
      "training_step_time": 6.849602460861206
    },
    {
      "epoch": 1.1997222900390625e-06,
      "model_forward_time": 1.7016427516937256,
      "step": 1258
    },
    {
      "epoch": 1.1997222900390625e-06,
      "step": 1258,
      "training_step_time": 7.483015298843384
    },
    {
      "epoch": 1.2006759643554687e-06,
      "model_forward_time": 1.7037341594696045,
      "step": 1259
    },
    {
      "epoch": 1.2006759643554687e-06,
      "step": 1259,
      "training_step_time": 8.757381200790405
    },
    {
      "epoch": 1.201629638671875e-06,
      "grad_norm": 3.2492740154266357,
      "learning_rate": 2.5180000000000003e-06,
      "loss": 0.132,
      "step": 1260
    },
    {
      "epoch": 1.201629638671875e-06,
      "model_forward_time": 1.7190923690795898,
      "step": 1260
    },
    {
      "dynamics_loss_avg": 0.10655555352568627,
      "epoch": 1.201629638671875e-06,
      "step": 1260
    },
    {
      "action_loss_avg": 0.0735254032537341,
      "epoch": 1.201629638671875e-06,
      "step": 1260
    },
    {
      "epoch": 1.201629638671875e-06,
      "step": 1260,
      "training_step_time": 9.722773790359497
    },
    {
      "epoch": 1.2025833129882812e-06,
      "model_forward_time": 2.270535707473755,
      "step": 1261
    },
    {
      "epoch": 1.2025833129882812e-06,
      "step": 1261,
      "training_step_time": 8.585324048995972
    },
    {
      "epoch": 1.2035369873046875e-06,
      "model_forward_time": 1.705599069595337,
      "step": 1262
    },
    {
      "epoch": 1.2035369873046875e-06,
      "step": 1262,
      "training_step_time": 7.851413249969482
    },
    {
      "epoch": 1.2044906616210937e-06,
      "model_forward_time": 1.7065258026123047,
      "step": 1263
    },
    {
      "epoch": 1.2044906616210937e-06,
      "step": 1263,
      "training_step_time": 8.751783847808838
    },
    {
      "epoch": 1.2054443359375e-06,
      "model_forward_time": 1.7057034969329834,
      "step": 1264
    },
    {
      "epoch": 1.2054443359375e-06,
      "step": 1264,
      "training_step_time": 7.20267391204834
    },
    {
      "epoch": 1.2063980102539062e-06,
      "model_forward_time": 1.7059593200683594,
      "step": 1265
    },
    {
      "epoch": 1.2063980102539062e-06,
      "step": 1265,
      "training_step_time": 7.4793922901153564
    },
    {
      "epoch": 1.2073516845703125e-06,
      "model_forward_time": 1.705979347229004,
      "step": 1266
    },
    {
      "epoch": 1.2073516845703125e-06,
      "step": 1266,
      "training_step_time": 6.816619634628296
    },
    {
      "epoch": 1.2083053588867187e-06,
      "model_forward_time": 1.7106094360351562,
      "step": 1267
    },
    {
      "epoch": 1.2083053588867187e-06,
      "step": 1267,
      "training_step_time": 7.466755628585815
    },
    {
      "epoch": 1.209259033203125e-06,
      "model_forward_time": 1.7051081657409668,
      "step": 1268
    },
    {
      "epoch": 1.209259033203125e-06,
      "step": 1268,
      "training_step_time": 7.704553604125977
    },
    {
      "epoch": 1.2102127075195312e-06,
      "model_forward_time": 1.699148178100586,
      "step": 1269
    },
    {
      "epoch": 1.2102127075195312e-06,
      "step": 1269,
      "training_step_time": 8.680124282836914
    },
    {
      "epoch": 1.2111663818359375e-06,
      "grad_norm": 3.263881206512451,
      "learning_rate": 2.5380000000000005e-06,
      "loss": 0.1126,
      "step": 1270
    },
    {
      "epoch": 1.2111663818359375e-06,
      "model_forward_time": 2.301774740219116,
      "step": 1270
    },
    {
      "dynamics_loss_avg": 0.0862166803330183,
      "epoch": 1.2111663818359375e-06,
      "step": 1270
    },
    {
      "action_loss_avg": 0.04669231865555048,
      "epoch": 1.2111663818359375e-06,
      "step": 1270
    },
    {
      "epoch": 1.2111663818359375e-06,
      "step": 1270,
      "training_step_time": 9.530534505844116
    },
    {
      "epoch": 1.2121200561523437e-06,
      "model_forward_time": 1.703423023223877,
      "step": 1271
    },
    {
      "epoch": 1.2121200561523437e-06,
      "step": 1271,
      "training_step_time": 7.893686294555664
    },
    {
      "epoch": 1.21307373046875e-06,
      "model_forward_time": 1.7065863609313965,
      "step": 1272
    },
    {
      "epoch": 1.21307373046875e-06,
      "step": 1272,
      "training_step_time": 8.654603719711304
    },
    {
      "epoch": 1.2140274047851562e-06,
      "model_forward_time": 1.7018413543701172,
      "step": 1273
    },
    {
      "epoch": 1.2140274047851562e-06,
      "step": 1273,
      "training_step_time": 7.577804327011108
    },
    {
      "epoch": 1.2149810791015625e-06,
      "model_forward_time": 1.705664873123169,
      "step": 1274
    },
    {
      "epoch": 1.2149810791015625e-06,
      "step": 1274,
      "training_step_time": 7.5064778327941895
    },
    {
      "epoch": 1.2159347534179687e-06,
      "model_forward_time": 1.7021965980529785,
      "step": 1275
    },
    {
      "epoch": 1.2159347534179687e-06,
      "step": 1275,
      "training_step_time": 6.910125732421875
    },
    {
      "epoch": 1.216888427734375e-06,
      "model_forward_time": 1.7005186080932617,
      "step": 1276
    },
    {
      "epoch": 1.216888427734375e-06,
      "step": 1276,
      "training_step_time": 7.474613189697266
    },
    {
      "epoch": 1.2178421020507812e-06,
      "model_forward_time": 1.706028699874878,
      "step": 1277
    },
    {
      "epoch": 1.2178421020507812e-06,
      "step": 1277,
      "training_step_time": 7.750981092453003
    },
    {
      "epoch": 1.2187957763671875e-06,
      "model_forward_time": 1.7177824974060059,
      "step": 1278
    },
    {
      "epoch": 1.2187957763671875e-06,
      "step": 1278,
      "training_step_time": 7.746114015579224
    },
    {
      "epoch": 1.2197494506835937e-06,
      "model_forward_time": 2.2954254150390625,
      "step": 1279
    },
    {
      "epoch": 1.2197494506835937e-06,
      "step": 1279,
      "training_step_time": 8.677101373672485
    },
    {
      "epoch": 1.220703125e-06,
      "grad_norm": 2.493624210357666,
      "learning_rate": 2.5580000000000003e-06,
      "loss": 0.104,
      "step": 1280
    },
    {
      "epoch": 1.220703125e-06,
      "model_forward_time": 1.701784372329712,
      "step": 1280
    },
    {
      "dynamics_loss_avg": 0.06955887712538242,
      "epoch": 1.220703125e-06,
      "step": 1280
    },
    {
      "action_loss_avg": 0.031834392342716455,
      "epoch": 1.220703125e-06,
      "step": 1280
    },
    {
      "epoch": 1.220703125e-06,
      "step": 1280,
      "training_step_time": 7.736363887786865
    },
    {
      "epoch": 1.2216567993164062e-06,
      "model_forward_time": 1.7055954933166504,
      "step": 1281
    },
    {
      "epoch": 1.2216567993164062e-06,
      "step": 1281,
      "training_step_time": 8.86108136177063
    },
    {
      "epoch": 1.2226104736328125e-06,
      "model_forward_time": 1.703536033630371,
      "step": 1282
    },
    {
      "epoch": 1.2226104736328125e-06,
      "step": 1282,
      "training_step_time": 7.728076457977295
    },
    {
      "epoch": 1.2235641479492187e-06,
      "model_forward_time": 1.7045807838439941,
      "step": 1283
    },
    {
      "epoch": 1.2235641479492187e-06,
      "step": 1283,
      "training_step_time": 7.481274366378784
    },
    {
      "epoch": 1.224517822265625e-06,
      "model_forward_time": 1.7034759521484375,
      "step": 1284
    },
    {
      "epoch": 1.224517822265625e-06,
      "step": 1284,
      "training_step_time": 6.8440916538238525
    },
    {
      "epoch": 1.2254714965820312e-06,
      "model_forward_time": 1.7085938453674316,
      "step": 1285
    },
    {
      "epoch": 1.2254714965820312e-06,
      "step": 1285,
      "training_step_time": 7.502007722854614
    },
    {
      "epoch": 1.2264251708984375e-06,
      "model_forward_time": 1.7202661037445068,
      "step": 1286
    },
    {
      "epoch": 1.2264251708984375e-06,
      "step": 1286,
      "training_step_time": 7.75129508972168
    },
    {
      "epoch": 1.2273788452148437e-06,
      "model_forward_time": 1.7275207042694092,
      "step": 1287
    },
    {
      "epoch": 1.2273788452148437e-06,
      "step": 1287,
      "training_step_time": 8.410444021224976
    },
    {
      "epoch": 1.22833251953125e-06,
      "model_forward_time": 2.2707529067993164,
      "step": 1288
    },
    {
      "epoch": 1.22833251953125e-06,
      "step": 1288,
      "training_step_time": 9.447555780410767
    },
    {
      "epoch": 1.2292861938476562e-06,
      "model_forward_time": 1.7264986038208008,
      "step": 1289
    },
    {
      "epoch": 1.2292861938476562e-06,
      "step": 1289,
      "training_step_time": 7.969900846481323
    },
    {
      "epoch": 1.2302398681640625e-06,
      "grad_norm": 1.9508192539215088,
      "learning_rate": 2.578e-06,
      "loss": 0.111,
      "step": 1290
    },
    {
      "epoch": 1.2302398681640625e-06,
      "model_forward_time": 1.724989414215088,
      "step": 1290
    },
    {
      "dynamics_loss_avg": 0.07400027103722095,
      "epoch": 1.2302398681640625e-06,
      "step": 1290
    },
    {
      "action_loss_avg": 0.04503412991762161,
      "epoch": 1.2302398681640625e-06,
      "step": 1290
    },
    {
      "epoch": 1.2302398681640625e-06,
      "step": 1290,
      "training_step_time": 7.848452091217041
    },
    {
      "epoch": 1.2311935424804687e-06,
      "model_forward_time": 1.7282817363739014,
      "step": 1291
    },
    {
      "epoch": 1.2311935424804687e-06,
      "step": 1291,
      "training_step_time": 7.858336925506592
    },
    {
      "epoch": 1.232147216796875e-06,
      "model_forward_time": 1.7187881469726562,
      "step": 1292
    },
    {
      "epoch": 1.232147216796875e-06,
      "step": 1292,
      "training_step_time": 7.6662678718566895
    },
    {
      "epoch": 1.2331008911132812e-06,
      "model_forward_time": 1.7203547954559326,
      "step": 1293
    },
    {
      "epoch": 1.2331008911132812e-06,
      "step": 1293,
      "training_step_time": 6.864579677581787
    },
    {
      "epoch": 1.2340545654296875e-06,
      "model_forward_time": 1.716874122619629,
      "step": 1294
    },
    {
      "epoch": 1.2340545654296875e-06,
      "step": 1294,
      "training_step_time": 7.524617671966553
    },
    {
      "epoch": 1.2350082397460937e-06,
      "model_forward_time": 1.729691982269287,
      "step": 1295
    },
    {
      "epoch": 1.2350082397460937e-06,
      "step": 1295,
      "training_step_time": 7.701087713241577
    },
    {
      "epoch": 1.2359619140625e-06,
      "model_forward_time": 1.7333860397338867,
      "step": 1296
    },
    {
      "epoch": 1.2359619140625e-06,
      "step": 1296,
      "training_step_time": 8.6263747215271
    },
    {
      "epoch": 1.2369155883789062e-06,
      "model_forward_time": 2.2877793312072754,
      "step": 1297
    },
    {
      "epoch": 1.2369155883789062e-06,
      "step": 1297,
      "training_step_time": 8.762592315673828
    },
    {
      "epoch": 1.2378692626953125e-06,
      "model_forward_time": 1.7633812427520752,
      "step": 1298
    },
    {
      "epoch": 1.2378692626953125e-06,
      "step": 1298,
      "training_step_time": 9.79216742515564
    },
    {
      "epoch": 1.2388229370117187e-06,
      "model_forward_time": 1.728311538696289,
      "step": 1299
    },
    {
      "epoch": 1.2388229370117187e-06,
      "step": 1299,
      "training_step_time": 8.772258281707764
    },
    {
      "epoch": 1.239776611328125e-06,
      "grad_norm": 2.662904739379883,
      "learning_rate": 2.598e-06,
      "loss": 0.1237,
      "step": 1300
    },
    {
      "epoch": 1.239776611328125e-06,
      "model_forward_time": 1.7250289916992188,
      "step": 1300
    },
    {
      "dynamics_loss_avg": 0.07426668629050255,
      "epoch": 1.239776611328125e-06,
      "step": 1300
    },
    {
      "action_loss_avg": 0.045106634078547356,
      "epoch": 1.239776611328125e-06,
      "step": 1300
    },
    {
      "epoch": 1.239776611328125e-06,
      "step": 1300,
      "training_step_time": 7.701044797897339
    },
    {
      "epoch": 1.2407302856445312e-06,
      "model_forward_time": 1.7317020893096924,
      "step": 1301
    },
    {
      "epoch": 1.2407302856445312e-06,
      "step": 1301,
      "training_step_time": 7.670301914215088
    },
    {
      "epoch": 1.2416839599609375e-06,
      "model_forward_time": 1.7242538928985596,
      "step": 1302
    },
    {
      "epoch": 1.2416839599609375e-06,
      "step": 1302,
      "training_step_time": 6.825326442718506
    },
    {
      "epoch": 1.2426376342773437e-06,
      "model_forward_time": 1.723374605178833,
      "step": 1303
    },
    {
      "epoch": 1.2426376342773437e-06,
      "step": 1303,
      "training_step_time": 7.485827445983887
    },
    {
      "epoch": 1.24359130859375e-06,
      "model_forward_time": 1.720160961151123,
      "step": 1304
    },
    {
      "epoch": 1.24359130859375e-06,
      "step": 1304,
      "training_step_time": 7.683786869049072
    },
    {
      "epoch": 1.2445449829101562e-06,
      "model_forward_time": 1.7219977378845215,
      "step": 1305
    },
    {
      "epoch": 1.2445449829101562e-06,
      "step": 1305,
      "training_step_time": 7.775456428527832
    },
    {
      "epoch": 1.2454986572265625e-06,
      "model_forward_time": 1.7117314338684082,
      "step": 1306
    },
    {
      "epoch": 1.2454986572265625e-06,
      "step": 1306,
      "training_step_time": 9.39878225326538
    },
    {
      "epoch": 1.2464523315429687e-06,
      "model_forward_time": 1.7660939693450928,
      "step": 1307
    },
    {
      "epoch": 1.2464523315429687e-06,
      "step": 1307,
      "training_step_time": 10.614105939865112
    },
    {
      "epoch": 1.247406005859375e-06,
      "model_forward_time": 1.0105774402618408,
      "step": 1308
    },
    {
      "epoch": 1.247406005859375e-06,
      "step": 1308,
      "training_step_time": 9.911931276321411
    },
    {
      "epoch": 1.2483596801757812e-06,
      "model_forward_time": 1.7228856086730957,
      "step": 1309
    },
    {
      "epoch": 1.2483596801757812e-06,
      "step": 1309,
      "training_step_time": 8.832072973251343
    },
    {
      "epoch": 1.2493133544921876e-06,
      "grad_norm": 2.1854498386383057,
      "learning_rate": 2.618e-06,
      "loss": 0.1125,
      "step": 1310
    },
    {
      "epoch": 1.2493133544921876e-06,
      "model_forward_time": 1.7267944812774658,
      "step": 1310
    },
    {
      "dynamics_loss_avg": 0.0961328437551856,
      "epoch": 1.2493133544921876e-06,
      "step": 1310
    },
    {
      "action_loss_avg": 0.03641289193183184,
      "epoch": 1.2493133544921876e-06,
      "step": 1310
    },
    {
      "epoch": 1.2493133544921876e-06,
      "step": 1310,
      "training_step_time": 7.669582366943359
    },
    {
      "epoch": 1.2502670288085937e-06,
      "model_forward_time": 1.7216534614562988,
      "step": 1311
    },
    {
      "epoch": 1.2502670288085937e-06,
      "step": 1311,
      "training_step_time": 6.860955476760864
    },
    {
      "epoch": 1.251220703125e-06,
      "model_forward_time": 1.7185537815093994,
      "step": 1312
    },
    {
      "epoch": 1.251220703125e-06,
      "step": 1312,
      "training_step_time": 7.498663663864136
    },
    {
      "epoch": 1.2521743774414062e-06,
      "model_forward_time": 1.733489990234375,
      "step": 1313
    },
    {
      "epoch": 1.2521743774414062e-06,
      "step": 1313,
      "training_step_time": 7.361007213592529
    },
    {
      "epoch": 1.2531280517578126e-06,
      "model_forward_time": 1.7230803966522217,
      "step": 1314
    },
    {
      "epoch": 1.2531280517578126e-06,
      "step": 1314,
      "training_step_time": 7.897451400756836
    },
    {
      "epoch": 1.2540817260742187e-06,
      "model_forward_time": 1.7143869400024414,
      "step": 1315
    },
    {
      "epoch": 1.2540817260742187e-06,
      "step": 1315,
      "training_step_time": 7.8501996994018555
    },
    {
      "epoch": 1.255035400390625e-06,
      "model_forward_time": 2.310297966003418,
      "step": 1316
    },
    {
      "epoch": 1.255035400390625e-06,
      "step": 1316,
      "training_step_time": 9.898930311203003
    },
    {
      "epoch": 1.2559890747070312e-06,
      "model_forward_time": 1.3578381538391113,
      "step": 1317
    },
    {
      "epoch": 1.2559890747070312e-06,
      "step": 1317,
      "training_step_time": 9.037007093429565
    },
    {
      "epoch": 1.2569427490234376e-06,
      "model_forward_time": 1.7230849266052246,
      "step": 1318
    },
    {
      "epoch": 1.2569427490234376e-06,
      "step": 1318,
      "training_step_time": 8.48895001411438
    },
    {
      "epoch": 1.2578964233398437e-06,
      "model_forward_time": 1.7216837406158447,
      "step": 1319
    },
    {
      "epoch": 1.2578964233398437e-06,
      "step": 1319,
      "training_step_time": 7.703393936157227
    },
    {
      "epoch": 1.25885009765625e-06,
      "grad_norm": 2.2713232040405273,
      "learning_rate": 2.638e-06,
      "loss": 0.1189,
      "step": 1320
    },
    {
      "epoch": 1.25885009765625e-06,
      "model_forward_time": 1.7269742488861084,
      "step": 1320
    },
    {
      "dynamics_loss_avg": 0.0792810145765543,
      "epoch": 1.25885009765625e-06,
      "step": 1320
    },
    {
      "action_loss_avg": 0.05138655118644238,
      "epoch": 1.25885009765625e-06,
      "step": 1320
    },
    {
      "epoch": 1.25885009765625e-06,
      "step": 1320,
      "training_step_time": 7.657706260681152
    },
    {
      "epoch": 1.2598037719726562e-06,
      "model_forward_time": 1.7207510471343994,
      "step": 1321
    },
    {
      "epoch": 1.2598037719726562e-06,
      "step": 1321,
      "training_step_time": 7.471009016036987
    },
    {
      "epoch": 1.2607574462890626e-06,
      "model_forward_time": 1.7280380725860596,
      "step": 1322
    },
    {
      "epoch": 1.2607574462890626e-06,
      "step": 1322,
      "training_step_time": 7.2890684604644775
    },
    {
      "epoch": 1.2617111206054687e-06,
      "model_forward_time": 1.7290704250335693,
      "step": 1323
    },
    {
      "epoch": 1.2617111206054687e-06,
      "step": 1323,
      "training_step_time": 7.799257755279541
    },
    {
      "epoch": 1.262664794921875e-06,
      "model_forward_time": 1.7209067344665527,
      "step": 1324
    },
    {
      "epoch": 1.262664794921875e-06,
      "step": 1324,
      "training_step_time": 7.951004266738892
    },
    {
      "epoch": 1.2636184692382812e-06,
      "model_forward_time": 2.28533673286438,
      "step": 1325
    },
    {
      "epoch": 1.2636184692382812e-06,
      "step": 1325,
      "training_step_time": 8.92936635017395
    },
    {
      "epoch": 1.2645721435546876e-06,
      "model_forward_time": 1.7229831218719482,
      "step": 1326
    },
    {
      "epoch": 1.2645721435546876e-06,
      "step": 1326,
      "training_step_time": 8.012485980987549
    },
    {
      "epoch": 1.2655258178710937e-06,
      "model_forward_time": 1.7207672595977783,
      "step": 1327
    },
    {
      "epoch": 1.2655258178710937e-06,
      "step": 1327,
      "training_step_time": 7.808453559875488
    },
    {
      "epoch": 1.2664794921875e-06,
      "model_forward_time": 1.7266168594360352,
      "step": 1328
    },
    {
      "epoch": 1.2664794921875e-06,
      "step": 1328,
      "training_step_time": 7.7442052364349365
    },
    {
      "epoch": 1.2674331665039062e-06,
      "model_forward_time": 1.7232651710510254,
      "step": 1329
    },
    {
      "epoch": 1.2674331665039062e-06,
      "step": 1329,
      "training_step_time": 7.647116422653198
    },
    {
      "epoch": 1.2683868408203126e-06,
      "grad_norm": 1.4718186855316162,
      "learning_rate": 2.6580000000000002e-06,
      "loss": 0.121,
      "step": 1330
    },
    {
      "epoch": 1.2683868408203126e-06,
      "model_forward_time": 1.7220678329467773,
      "step": 1330
    },
    {
      "dynamics_loss_avg": 0.10273189917206764,
      "epoch": 1.2683868408203126e-06,
      "step": 1330
    },
    {
      "action_loss_avg": 0.04927161894738674,
      "epoch": 1.2683868408203126e-06,
      "step": 1330
    },
    {
      "epoch": 1.2683868408203126e-06,
      "step": 1330,
      "training_step_time": 7.436778545379639
    },
    {
      "epoch": 1.2693405151367187e-06,
      "model_forward_time": 1.7247092723846436,
      "step": 1331
    },
    {
      "epoch": 1.2693405151367187e-06,
      "step": 1331,
      "training_step_time": 7.447235345840454
    },
    {
      "epoch": 1.270294189453125e-06,
      "model_forward_time": 1.7263269424438477,
      "step": 1332
    },
    {
      "epoch": 1.270294189453125e-06,
      "step": 1332,
      "training_step_time": 7.781463623046875
    },
    {
      "epoch": 1.2712478637695312e-06,
      "model_forward_time": 1.0156612396240234,
      "step": 1333
    },
    {
      "epoch": 1.2712478637695312e-06,
      "step": 1333,
      "training_step_time": 7.878623962402344
    },
    {
      "epoch": 1.2722015380859376e-06,
      "model_forward_time": 1.7282629013061523,
      "step": 1334
    },
    {
      "epoch": 1.2722015380859376e-06,
      "step": 1334,
      "training_step_time": 8.825084209442139
    },
    {
      "epoch": 1.2731552124023437e-06,
      "model_forward_time": 2.2579188346862793,
      "step": 1335
    },
    {
      "epoch": 1.2731552124023437e-06,
      "step": 1335,
      "training_step_time": 8.004027843475342
    },
    {
      "epoch": 1.27410888671875e-06,
      "model_forward_time": 1.7231123447418213,
      "step": 1336
    },
    {
      "epoch": 1.27410888671875e-06,
      "step": 1336,
      "training_step_time": 7.843466520309448
    },
    {
      "epoch": 1.2750625610351562e-06,
      "model_forward_time": 1.725550651550293,
      "step": 1337
    },
    {
      "epoch": 1.2750625610351562e-06,
      "step": 1337,
      "training_step_time": 8.727347612380981
    },
    {
      "epoch": 1.2760162353515626e-06,
      "model_forward_time": 1.7243106365203857,
      "step": 1338
    },
    {
      "epoch": 1.2760162353515626e-06,
      "step": 1338,
      "training_step_time": 8.636238813400269
    },
    {
      "epoch": 1.2769699096679687e-06,
      "model_forward_time": 1.7260668277740479,
      "step": 1339
    },
    {
      "epoch": 1.2769699096679687e-06,
      "step": 1339,
      "training_step_time": 7.404662132263184
    },
    {
      "epoch": 1.277923583984375e-06,
      "grad_norm": 2.651655912399292,
      "learning_rate": 2.678e-06,
      "loss": 0.1227,
      "step": 1340
    },
    {
      "epoch": 1.277923583984375e-06,
      "model_forward_time": 1.715775728225708,
      "step": 1340
    },
    {
      "dynamics_loss_avg": 0.09166927933692932,
      "epoch": 1.277923583984375e-06,
      "step": 1340
    },
    {
      "action_loss_avg": 0.05589608680456877,
      "epoch": 1.277923583984375e-06,
      "step": 1340
    },
    {
      "epoch": 1.277923583984375e-06,
      "step": 1340,
      "training_step_time": 7.5677385330200195
    },
    {
      "epoch": 1.2788772583007812e-06,
      "model_forward_time": 1.735779047012329,
      "step": 1341
    },
    {
      "epoch": 1.2788772583007812e-06,
      "step": 1341,
      "training_step_time": 7.734973430633545
    },
    {
      "epoch": 1.2798309326171876e-06,
      "model_forward_time": 1.7309198379516602,
      "step": 1342
    },
    {
      "epoch": 1.2798309326171876e-06,
      "step": 1342,
      "training_step_time": 8.623464822769165
    },
    {
      "epoch": 1.2807846069335937e-06,
      "model_forward_time": 1.7224256992340088,
      "step": 1343
    },
    {
      "epoch": 1.2807846069335937e-06,
      "step": 1343,
      "training_step_time": 9.440216064453125
    },
    {
      "epoch": 1.28173828125e-06,
      "model_forward_time": 2.2791748046875,
      "step": 1344
    },
    {
      "epoch": 1.28173828125e-06,
      "step": 1344,
      "training_step_time": 8.868791580200195
    },
    {
      "epoch": 1.2826919555664062e-06,
      "model_forward_time": 1.724259853363037,
      "step": 1345
    },
    {
      "epoch": 1.2826919555664062e-06,
      "step": 1345,
      "training_step_time": 8.709743976593018
    },
    {
      "epoch": 1.2836456298828126e-06,
      "model_forward_time": 1.7199325561523438,
      "step": 1346
    },
    {
      "epoch": 1.2836456298828126e-06,
      "step": 1346,
      "training_step_time": 7.666716575622559
    },
    {
      "epoch": 1.2845993041992187e-06,
      "model_forward_time": 1.727416753768921,
      "step": 1347
    },
    {
      "epoch": 1.2845993041992187e-06,
      "step": 1347,
      "training_step_time": 7.7302422523498535
    },
    {
      "epoch": 1.285552978515625e-06,
      "model_forward_time": 1.721956491470337,
      "step": 1348
    },
    {
      "epoch": 1.285552978515625e-06,
      "step": 1348,
      "training_step_time": 8.402578830718994
    },
    {
      "epoch": 1.2865066528320312e-06,
      "model_forward_time": 1.7136428356170654,
      "step": 1349
    },
    {
      "epoch": 1.2865066528320312e-06,
      "step": 1349,
      "training_step_time": 7.512679576873779
    },
    {
      "epoch": 1.2874603271484376e-06,
      "grad_norm": 2.0724873542785645,
      "learning_rate": 2.6980000000000003e-06,
      "loss": 0.1155,
      "step": 1350
    },
    {
      "epoch": 1.2874603271484376e-06,
      "model_forward_time": 1.7245142459869385,
      "step": 1350
    },
    {
      "dynamics_loss_avg": 0.07566445507109165,
      "epoch": 1.2874603271484376e-06,
      "step": 1350
    },
    {
      "action_loss_avg": 0.037582595366984606,
      "epoch": 1.2874603271484376e-06,
      "step": 1350
    },
    {
      "epoch": 1.2874603271484376e-06,
      "step": 1350,
      "training_step_time": 7.656893014907837
    },
    {
      "epoch": 1.2884140014648437e-06,
      "model_forward_time": 1.011723279953003,
      "step": 1351
    },
    {
      "epoch": 1.2884140014648437e-06,
      "step": 1351,
      "training_step_time": 8.573049783706665
    },
    {
      "epoch": 1.28936767578125e-06,
      "model_forward_time": 1.7222063541412354,
      "step": 1352
    },
    {
      "epoch": 1.28936767578125e-06,
      "step": 1352,
      "training_step_time": 7.843617916107178
    },
    {
      "epoch": 1.2903213500976562e-06,
      "model_forward_time": 2.2928059101104736,
      "step": 1353
    },
    {
      "epoch": 1.2903213500976562e-06,
      "step": 1353,
      "training_step_time": 10.077973365783691
    },
    {
      "epoch": 1.2912750244140626e-06,
      "model_forward_time": 1.723088026046753,
      "step": 1354
    },
    {
      "epoch": 1.2912750244140626e-06,
      "step": 1354,
      "training_step_time": 8.568717241287231
    },
    {
      "epoch": 1.2922286987304687e-06,
      "model_forward_time": 1.7291195392608643,
      "step": 1355
    },
    {
      "epoch": 1.2922286987304687e-06,
      "step": 1355,
      "training_step_time": 7.808084487915039
    },
    {
      "epoch": 1.293182373046875e-06,
      "model_forward_time": 1.7207329273223877,
      "step": 1356
    },
    {
      "epoch": 1.293182373046875e-06,
      "step": 1356,
      "training_step_time": 7.665665864944458
    },
    {
      "epoch": 1.2941360473632812e-06,
      "model_forward_time": 1.7207841873168945,
      "step": 1357
    },
    {
      "epoch": 1.2941360473632812e-06,
      "step": 1357,
      "training_step_time": 7.202222108840942
    },
    {
      "epoch": 1.2950897216796876e-06,
      "model_forward_time": 1.7252066135406494,
      "step": 1358
    },
    {
      "epoch": 1.2950897216796876e-06,
      "step": 1358,
      "training_step_time": 7.596287965774536
    },
    {
      "epoch": 1.2960433959960937e-06,
      "model_forward_time": 1.7280139923095703,
      "step": 1359
    },
    {
      "epoch": 1.2960433959960937e-06,
      "step": 1359,
      "training_step_time": 7.585788011550903
    },
    {
      "epoch": 1.2969970703125e-06,
      "grad_norm": 1.7992548942565918,
      "learning_rate": 2.718e-06,
      "loss": 0.1071,
      "step": 1360
    },
    {
      "epoch": 1.2969970703125e-06,
      "model_forward_time": 1.7245967388153076,
      "step": 1360
    },
    {
      "dynamics_loss_avg": 0.06338249947875738,
      "epoch": 1.2969970703125e-06,
      "step": 1360
    },
    {
      "action_loss_avg": 0.03284999513998628,
      "epoch": 1.2969970703125e-06,
      "step": 1360
    },
    {
      "epoch": 1.2969970703125e-06,
      "step": 1360,
      "training_step_time": 7.7828850746154785
    },
    {
      "epoch": 1.2979507446289062e-06,
      "model_forward_time": 1.7185797691345215,
      "step": 1361
    },
    {
      "epoch": 1.2979507446289062e-06,
      "step": 1361,
      "training_step_time": 7.757296323776245
    },
    {
      "epoch": 1.2989044189453126e-06,
      "model_forward_time": 1.7286462783813477,
      "step": 1362
    },
    {
      "epoch": 1.2989044189453126e-06,
      "step": 1362,
      "training_step_time": 8.827347040176392
    },
    {
      "epoch": 1.2998580932617187e-06,
      "model_forward_time": 2.2592716217041016,
      "step": 1363
    },
    {
      "epoch": 1.2998580932617187e-06,
      "step": 1363,
      "training_step_time": 7.7994701862335205
    },
    {
      "epoch": 1.300811767578125e-06,
      "model_forward_time": 1.7288670539855957,
      "step": 1364
    },
    {
      "epoch": 1.300811767578125e-06,
      "step": 1364,
      "training_step_time": 7.865737676620483
    },
    {
      "epoch": 1.3017654418945312e-06,
      "model_forward_time": 1.7144381999969482,
      "step": 1365
    },
    {
      "epoch": 1.3017654418945312e-06,
      "step": 1365,
      "training_step_time": 7.765187978744507
    },
    {
      "epoch": 1.3027191162109376e-06,
      "model_forward_time": 1.721524953842163,
      "step": 1366
    },
    {
      "epoch": 1.3027191162109376e-06,
      "step": 1366,
      "training_step_time": 7.440539598464966
    },
    {
      "epoch": 1.3036727905273437e-06,
      "model_forward_time": 1.014831304550171,
      "step": 1367
    },
    {
      "epoch": 1.3036727905273437e-06,
      "step": 1367,
      "training_step_time": 7.693942070007324
    },
    {
      "epoch": 1.30462646484375e-06,
      "model_forward_time": 1.7251365184783936,
      "step": 1368
    },
    {
      "epoch": 1.30462646484375e-06,
      "step": 1368,
      "training_step_time": 7.183284759521484
    },
    {
      "epoch": 1.3055801391601562e-06,
      "model_forward_time": 1.7203750610351562,
      "step": 1369
    },
    {
      "epoch": 1.3055801391601562e-06,
      "step": 1369,
      "training_step_time": 7.839219331741333
    },
    {
      "epoch": 1.3065338134765626e-06,
      "grad_norm": 2.0063955783843994,
      "learning_rate": 2.7380000000000003e-06,
      "loss": 0.1123,
      "step": 1370
    },
    {
      "epoch": 1.3065338134765626e-06,
      "model_forward_time": 1.7312722206115723,
      "step": 1370
    },
    {
      "dynamics_loss_avg": 0.06882018968462944,
      "epoch": 1.3065338134765626e-06,
      "step": 1370
    },
    {
      "action_loss_avg": 0.04821349922567606,
      "epoch": 1.3065338134765626e-06,
      "step": 1370
    },
    {
      "epoch": 1.3065338134765626e-06,
      "step": 1370,
      "training_step_time": 8.672221899032593
    },
    {
      "epoch": 1.3074874877929687e-06,
      "model_forward_time": 1.7577595710754395,
      "step": 1371
    },
    {
      "epoch": 1.3074874877929687e-06,
      "step": 1371,
      "training_step_time": 8.744978904724121
    },
    {
      "epoch": 1.308441162109375e-06,
      "model_forward_time": 2.269709825515747,
      "step": 1372
    },
    {
      "epoch": 1.308441162109375e-06,
      "step": 1372,
      "training_step_time": 8.023743629455566
    },
    {
      "epoch": 1.3093948364257812e-06,
      "model_forward_time": 1.7248480319976807,
      "step": 1373
    },
    {
      "epoch": 1.3093948364257812e-06,
      "step": 1373,
      "training_step_time": 7.743240833282471
    },
    {
      "epoch": 1.3103485107421876e-06,
      "model_forward_time": 1.7227470874786377,
      "step": 1374
    },
    {
      "epoch": 1.3103485107421876e-06,
      "step": 1374,
      "training_step_time": 7.655200242996216
    },
    {
      "epoch": 1.3113021850585938e-06,
      "model_forward_time": 1.7308812141418457,
      "step": 1375
    },
    {
      "epoch": 1.3113021850585938e-06,
      "step": 1375,
      "training_step_time": 7.762330532073975
    },
    {
      "epoch": 1.312255859375e-06,
      "model_forward_time": 1.723822832107544,
      "step": 1376
    },
    {
      "epoch": 1.312255859375e-06,
      "step": 1376,
      "training_step_time": 8.277475357055664
    },
    {
      "epoch": 1.3132095336914063e-06,
      "model_forward_time": 1.72292160987854,
      "step": 1377
    },
    {
      "epoch": 1.3132095336914063e-06,
      "step": 1377,
      "training_step_time": 7.489450931549072
    },
    {
      "epoch": 1.3141632080078124e-06,
      "model_forward_time": 1.7312695980072021,
      "step": 1378
    },
    {
      "epoch": 1.3141632080078124e-06,
      "step": 1378,
      "training_step_time": 7.797584772109985
    },
    {
      "epoch": 1.3151168823242188e-06,
      "model_forward_time": 1.7206883430480957,
      "step": 1379
    },
    {
      "epoch": 1.3151168823242188e-06,
      "step": 1379,
      "training_step_time": 7.610769510269165
    },
    {
      "epoch": 1.316070556640625e-06,
      "grad_norm": 1.8060375452041626,
      "learning_rate": 2.758e-06,
      "loss": 0.1114,
      "step": 1380
    },
    {
      "epoch": 1.316070556640625e-06,
      "model_forward_time": 1.7356159687042236,
      "step": 1380
    },
    {
      "dynamics_loss_avg": 0.06810841113328933,
      "epoch": 1.316070556640625e-06,
      "step": 1380
    },
    {
      "action_loss_avg": 0.04086309429258108,
      "epoch": 1.316070556640625e-06,
      "step": 1380
    },
    {
      "epoch": 1.316070556640625e-06,
      "step": 1380,
      "training_step_time": 7.829660415649414
    },
    {
      "epoch": 1.3170242309570313e-06,
      "model_forward_time": 2.2770445346832275,
      "step": 1381
    },
    {
      "epoch": 1.3170242309570313e-06,
      "step": 1381,
      "training_step_time": 9.48564338684082
    },
    {
      "epoch": 1.3179779052734374e-06,
      "model_forward_time": 1.7248799800872803,
      "step": 1382
    },
    {
      "epoch": 1.3179779052734374e-06,
      "step": 1382,
      "training_step_time": 7.815686941146851
    },
    {
      "epoch": 1.3189315795898438e-06,
      "model_forward_time": 1.730949878692627,
      "step": 1383
    },
    {
      "epoch": 1.3189315795898438e-06,
      "step": 1383,
      "training_step_time": 7.708716154098511
    },
    {
      "epoch": 1.31988525390625e-06,
      "model_forward_time": 1.7221992015838623,
      "step": 1384
    },
    {
      "epoch": 1.31988525390625e-06,
      "step": 1384,
      "training_step_time": 7.835003852844238
    },
    {
      "epoch": 1.3208389282226563e-06,
      "model_forward_time": 1.6948127746582031,
      "step": 1385
    },
    {
      "epoch": 1.3208389282226563e-06,
      "step": 1385,
      "training_step_time": 7.323206424713135
    },
    {
      "epoch": 1.3217926025390624e-06,
      "model_forward_time": 1.7107653617858887,
      "step": 1386
    },
    {
      "epoch": 1.3217926025390624e-06,
      "step": 1386,
      "training_step_time": 7.576199769973755
    },
    {
      "epoch": 1.3227462768554688e-06,
      "model_forward_time": 1.7077751159667969,
      "step": 1387
    },
    {
      "epoch": 1.3227462768554688e-06,
      "step": 1387,
      "training_step_time": 7.773715496063232
    },
    {
      "epoch": 1.323699951171875e-06,
      "model_forward_time": 1.704451084136963,
      "step": 1388
    },
    {
      "epoch": 1.323699951171875e-06,
      "step": 1388,
      "training_step_time": 8.714268445968628
    },
    {
      "epoch": 1.3246536254882813e-06,
      "model_forward_time": 1.702657699584961,
      "step": 1389
    },
    {
      "epoch": 1.3246536254882813e-06,
      "step": 1389,
      "training_step_time": 8.329996585845947
    },
    {
      "epoch": 1.3256072998046874e-06,
      "grad_norm": 2.0034353733062744,
      "learning_rate": 2.7780000000000003e-06,
      "loss": 0.118,
      "step": 1390
    },
    {
      "epoch": 1.3256072998046874e-06,
      "model_forward_time": 1.7027089595794678,
      "step": 1390
    },
    {
      "dynamics_loss_avg": 0.06632257662713528,
      "epoch": 1.3256072998046874e-06,
      "step": 1390
    },
    {
      "action_loss_avg": 0.033321940992027524,
      "epoch": 1.3256072998046874e-06,
      "step": 1390
    },
    {
      "epoch": 1.3256072998046874e-06,
      "step": 1390,
      "training_step_time": 9.71194052696228
    },
    {
      "epoch": 1.3265609741210938e-06,
      "model_forward_time": 1.7103755474090576,
      "step": 1391
    },
    {
      "epoch": 1.3265609741210938e-06,
      "step": 1391,
      "training_step_time": 7.78074312210083
    },
    {
      "epoch": 1.3275146484375e-06,
      "model_forward_time": 1.724712610244751,
      "step": 1392
    },
    {
      "epoch": 1.3275146484375e-06,
      "step": 1392,
      "training_step_time": 8.62710428237915
    },
    {
      "epoch": 1.3284683227539063e-06,
      "model_forward_time": 1.7336022853851318,
      "step": 1393
    },
    {
      "epoch": 1.3284683227539063e-06,
      "step": 1393,
      "training_step_time": 7.7217183113098145
    },
    {
      "epoch": 1.3294219970703124e-06,
      "model_forward_time": 1.7237591743469238,
      "step": 1394
    },
    {
      "epoch": 1.3294219970703124e-06,
      "step": 1394,
      "training_step_time": 7.274535179138184
    },
    {
      "epoch": 1.3303756713867188e-06,
      "model_forward_time": 1.7229275703430176,
      "step": 1395
    },
    {
      "epoch": 1.3303756713867188e-06,
      "step": 1395,
      "training_step_time": 7.594438791275024
    },
    {
      "epoch": 1.331329345703125e-06,
      "model_forward_time": 1.7299950122833252,
      "step": 1396
    },
    {
      "epoch": 1.331329345703125e-06,
      "step": 1396,
      "training_step_time": 8.237984895706177
    },
    {
      "epoch": 1.3322830200195313e-06,
      "model_forward_time": 1.7238492965698242,
      "step": 1397
    },
    {
      "epoch": 1.3322830200195313e-06,
      "step": 1397,
      "training_step_time": 7.482832431793213
    },
    {
      "epoch": 1.3332366943359374e-06,
      "model_forward_time": 1.7208006381988525,
      "step": 1398
    },
    {
      "epoch": 1.3332366943359374e-06,
      "step": 1398,
      "training_step_time": 7.795963764190674
    },
    {
      "epoch": 1.3341903686523438e-06,
      "model_forward_time": 1.7754924297332764,
      "step": 1399
    },
    {
      "epoch": 1.3341903686523438e-06,
      "step": 1399,
      "training_step_time": 7.987060070037842
    },
    {
      "epoch": 1.33514404296875e-06,
      "grad_norm": 1.6436835527420044,
      "learning_rate": 2.798e-06,
      "loss": 0.1167,
      "step": 1400
    },
    {
      "epoch": 1.33514404296875e-06,
      "model_forward_time": 2.252725124359131,
      "step": 1400
    },
    {
      "dynamics_loss_avg": 0.08498592935502529,
      "epoch": 1.33514404296875e-06,
      "step": 1400
    },
    {
      "action_loss_avg": 0.046587765030562875,
      "epoch": 1.33514404296875e-06,
      "step": 1400
    },
    {
      "epoch": 1.33514404296875e-06,
      "step": 1400,
      "training_step_time": 8.824867963790894
    },
    {
      "epoch": 1.3360977172851563e-06,
      "model_forward_time": 1.7240166664123535,
      "step": 1401
    },
    {
      "epoch": 1.3360977172851563e-06,
      "step": 1401,
      "training_step_time": 7.812275409698486
    },
    {
      "epoch": 1.3370513916015624e-06,
      "model_forward_time": 1.7240865230560303,
      "step": 1402
    },
    {
      "epoch": 1.3370513916015624e-06,
      "step": 1402,
      "training_step_time": 7.642657995223999
    },
    {
      "epoch": 1.3380050659179688e-06,
      "model_forward_time": 1.7239313125610352,
      "step": 1403
    },
    {
      "epoch": 1.3380050659179688e-06,
      "step": 1403,
      "training_step_time": 8.733442783355713
    },
    {
      "epoch": 1.338958740234375e-06,
      "model_forward_time": 1.7232425212860107,
      "step": 1404
    },
    {
      "epoch": 1.338958740234375e-06,
      "step": 1404,
      "training_step_time": 7.612992525100708
    },
    {
      "epoch": 1.3399124145507813e-06,
      "model_forward_time": 1.725327491760254,
      "step": 1405
    },
    {
      "epoch": 1.3399124145507813e-06,
      "step": 1405,
      "training_step_time": 7.774935960769653
    },
    {
      "epoch": 1.3408660888671874e-06,
      "model_forward_time": 1.7234618663787842,
      "step": 1406
    },
    {
      "epoch": 1.3408660888671874e-06,
      "step": 1406,
      "training_step_time": 7.71883487701416
    },
    {
      "epoch": 1.3418197631835938e-06,
      "model_forward_time": 1.717406988143921,
      "step": 1407
    },
    {
      "epoch": 1.3418197631835938e-06,
      "step": 1407,
      "training_step_time": 7.752138614654541
    },
    {
      "epoch": 1.3427734375e-06,
      "model_forward_time": 1.76951265335083,
      "step": 1408
    },
    {
      "epoch": 1.3427734375e-06,
      "step": 1408,
      "training_step_time": 9.544710636138916
    },
    {
      "epoch": 1.3437271118164063e-06,
      "model_forward_time": 2.2641351222991943,
      "step": 1409
    },
    {
      "epoch": 1.3437271118164063e-06,
      "step": 1409,
      "training_step_time": 7.873124599456787
    },
    {
      "epoch": 1.3446807861328124e-06,
      "grad_norm": 2.4819531440734863,
      "learning_rate": 2.8180000000000004e-06,
      "loss": 0.1269,
      "step": 1410
    },
    {
      "epoch": 1.3446807861328124e-06,
      "model_forward_time": 1.71834135055542,
      "step": 1410
    },
    {
      "dynamics_loss_avg": 0.09447596333920956,
      "epoch": 1.3446807861328124e-06,
      "step": 1410
    },
    {
      "action_loss_avg": 0.04359881184063852,
      "epoch": 1.3446807861328124e-06,
      "step": 1410
    },
    {
      "epoch": 1.3446807861328124e-06,
      "step": 1410,
      "training_step_time": 8.564261198043823
    },
    {
      "epoch": 1.3456344604492188e-06,
      "model_forward_time": 1.7218825817108154,
      "step": 1411
    },
    {
      "epoch": 1.3456344604492188e-06,
      "step": 1411,
      "training_step_time": 7.704607009887695
    },
    {
      "epoch": 1.346588134765625e-06,
      "model_forward_time": 1.7211592197418213,
      "step": 1412
    },
    {
      "epoch": 1.346588134765625e-06,
      "step": 1412,
      "training_step_time": 7.798751592636108
    },
    {
      "epoch": 1.3475418090820313e-06,
      "model_forward_time": 1.7190232276916504,
      "step": 1413
    },
    {
      "epoch": 1.3475418090820313e-06,
      "step": 1413,
      "training_step_time": 7.5748608112335205
    },
    {
      "epoch": 1.3484954833984374e-06,
      "model_forward_time": 1.7214107513427734,
      "step": 1414
    },
    {
      "epoch": 1.3484954833984374e-06,
      "step": 1414,
      "training_step_time": 7.730133771896362
    },
    {
      "epoch": 1.3494491577148438e-06,
      "model_forward_time": 1.7262518405914307,
      "step": 1415
    },
    {
      "epoch": 1.3494491577148438e-06,
      "step": 1415,
      "training_step_time": 7.689327001571655
    },
    {
      "epoch": 1.35040283203125e-06,
      "model_forward_time": 1.7208380699157715,
      "step": 1416
    },
    {
      "epoch": 1.35040283203125e-06,
      "step": 1416,
      "training_step_time": 7.620495319366455
    },
    {
      "epoch": 1.3513565063476563e-06,
      "model_forward_time": 1.7693195343017578,
      "step": 1417
    },
    {
      "epoch": 1.3513565063476563e-06,
      "step": 1417,
      "training_step_time": 7.770063161849976
    },
    {
      "epoch": 1.3523101806640624e-06,
      "model_forward_time": 2.2067110538482666,
      "step": 1418
    },
    {
      "epoch": 1.3523101806640624e-06,
      "step": 1418,
      "training_step_time": 10.200493574142456
    },
    {
      "epoch": 1.3532638549804688e-06,
      "model_forward_time": 1.7054073810577393,
      "step": 1419
    },
    {
      "epoch": 1.3532638549804688e-06,
      "step": 1419,
      "training_step_time": 10.16861343383789
    },
    {
      "epoch": 1.354217529296875e-06,
      "grad_norm": 3.37483549118042,
      "learning_rate": 2.838e-06,
      "loss": 0.1248,
      "step": 1420
    },
    {
      "epoch": 1.354217529296875e-06,
      "model_forward_time": 1.702197551727295,
      "step": 1420
    },
    {
      "dynamics_loss_avg": 0.06691961288452149,
      "epoch": 1.354217529296875e-06,
      "step": 1420
    },
    {
      "action_loss_avg": 0.03451351635158062,
      "epoch": 1.354217529296875e-06,
      "step": 1420
    },
    {
      "epoch": 1.354217529296875e-06,
      "step": 1420,
      "training_step_time": 7.684845209121704
    },
    {
      "epoch": 1.3551712036132813e-06,
      "model_forward_time": 1.7043352127075195,
      "step": 1421
    },
    {
      "epoch": 1.3551712036132813e-06,
      "step": 1421,
      "training_step_time": 7.704246282577515
    },
    {
      "epoch": 1.3561248779296874e-06,
      "model_forward_time": 1.7028160095214844,
      "step": 1422
    },
    {
      "epoch": 1.3561248779296874e-06,
      "step": 1422,
      "training_step_time": 7.32514762878418
    },
    {
      "epoch": 1.3570785522460938e-06,
      "model_forward_time": 1.7016115188598633,
      "step": 1423
    },
    {
      "epoch": 1.3570785522460938e-06,
      "step": 1423,
      "training_step_time": 7.796316623687744
    },
    {
      "epoch": 1.3580322265625e-06,
      "model_forward_time": 1.7063257694244385,
      "step": 1424
    },
    {
      "epoch": 1.3580322265625e-06,
      "step": 1424,
      "training_step_time": 8.699646711349487
    },
    {
      "epoch": 1.3589859008789063e-06,
      "model_forward_time": 1.705394983291626,
      "step": 1425
    },
    {
      "epoch": 1.3589859008789063e-06,
      "step": 1425,
      "training_step_time": 8.161287069320679
    },
    {
      "epoch": 1.3599395751953124e-06,
      "model_forward_time": 1.7412679195404053,
      "step": 1426
    },
    {
      "epoch": 1.3599395751953124e-06,
      "step": 1426,
      "training_step_time": 7.805900812149048
    },
    {
      "epoch": 1.3608932495117188e-06,
      "model_forward_time": 2.226518154144287,
      "step": 1427
    },
    {
      "epoch": 1.3608932495117188e-06,
      "step": 1427,
      "training_step_time": 8.860555648803711
    },
    {
      "epoch": 1.361846923828125e-06,
      "model_forward_time": 1.706982135772705,
      "step": 1428
    },
    {
      "epoch": 1.361846923828125e-06,
      "step": 1428,
      "training_step_time": 8.762319564819336
    },
    {
      "epoch": 1.3628005981445313e-06,
      "model_forward_time": 1.6854047775268555,
      "step": 1429
    },
    {
      "epoch": 1.3628005981445313e-06,
      "step": 1429,
      "training_step_time": 7.748960256576538
    },
    {
      "epoch": 1.3637542724609374e-06,
      "grad_norm": 1.4556183815002441,
      "learning_rate": 2.8580000000000004e-06,
      "loss": 0.1133,
      "step": 1430
    },
    {
      "epoch": 1.3637542724609374e-06,
      "model_forward_time": 1.6902813911437988,
      "step": 1430
    },
    {
      "dynamics_loss_avg": 0.07774076983332634,
      "epoch": 1.3637542724609374e-06,
      "step": 1430
    },
    {
      "action_loss_avg": 0.049103420600295065,
      "epoch": 1.3637542724609374e-06,
      "step": 1430
    },
    {
      "epoch": 1.3637542724609374e-06,
      "step": 1430,
      "training_step_time": 8.726173639297485
    },
    {
      "epoch": 1.3647079467773438e-06,
      "model_forward_time": 1.692319631576538,
      "step": 1431
    },
    {
      "epoch": 1.3647079467773438e-06,
      "step": 1431,
      "training_step_time": 7.319429636001587
    },
    {
      "epoch": 1.36566162109375e-06,
      "model_forward_time": 1.6846449375152588,
      "step": 1432
    },
    {
      "epoch": 1.36566162109375e-06,
      "step": 1432,
      "training_step_time": 7.743297100067139
    },
    {
      "epoch": 1.3666152954101563e-06,
      "model_forward_time": 1.6918740272521973,
      "step": 1433
    },
    {
      "epoch": 1.3666152954101563e-06,
      "step": 1433,
      "training_step_time": 7.64794659614563
    },
    {
      "epoch": 1.3675689697265624e-06,
      "model_forward_time": 1.6956682205200195,
      "step": 1434
    },
    {
      "epoch": 1.3675689697265624e-06,
      "step": 1434,
      "training_step_time": 8.340548276901245
    },
    {
      "epoch": 1.3685226440429688e-06,
      "model_forward_time": 1.6984100341796875,
      "step": 1435
    },
    {
      "epoch": 1.3685226440429688e-06,
      "step": 1435,
      "training_step_time": 8.80721926689148
    },
    {
      "epoch": 1.369476318359375e-06,
      "model_forward_time": 2.2198610305786133,
      "step": 1436
    },
    {
      "epoch": 1.369476318359375e-06,
      "step": 1436,
      "training_step_time": 9.162054061889648
    },
    {
      "epoch": 1.3704299926757813e-06,
      "model_forward_time": 1.690478801727295,
      "step": 1437
    },
    {
      "epoch": 1.3704299926757813e-06,
      "step": 1437,
      "training_step_time": 7.798983812332153
    },
    {
      "epoch": 1.3713836669921874e-06,
      "model_forward_time": 1.6887812614440918,
      "step": 1438
    },
    {
      "epoch": 1.3713836669921874e-06,
      "step": 1438,
      "training_step_time": 7.735146760940552
    },
    {
      "epoch": 1.3723373413085938e-06,
      "model_forward_time": 1.685868740081787,
      "step": 1439
    },
    {
      "epoch": 1.3723373413085938e-06,
      "step": 1439,
      "training_step_time": 7.704649209976196
    },
    {
      "epoch": 1.373291015625e-06,
      "grad_norm": 2.237949848175049,
      "learning_rate": 2.8780000000000002e-06,
      "loss": 0.1164,
      "step": 1440
    },
    {
      "epoch": 1.373291015625e-06,
      "model_forward_time": 1.6852574348449707,
      "step": 1440
    },
    {
      "dynamics_loss_avg": 0.08071233034133911,
      "epoch": 1.373291015625e-06,
      "step": 1440
    },
    {
      "action_loss_avg": 0.05592996878549457,
      "epoch": 1.373291015625e-06,
      "step": 1440
    },
    {
      "epoch": 1.373291015625e-06,
      "step": 1440,
      "training_step_time": 7.5864269733428955
    },
    {
      "epoch": 1.3742446899414063e-06,
      "model_forward_time": 1.6882069110870361,
      "step": 1441
    },
    {
      "epoch": 1.3742446899414063e-06,
      "step": 1441,
      "training_step_time": 7.742096185684204
    },
    {
      "epoch": 1.3751983642578125e-06,
      "model_forward_time": 1.6869511604309082,
      "step": 1442
    },
    {
      "epoch": 1.3751983642578125e-06,
      "step": 1442,
      "training_step_time": 7.665125846862793
    },
    {
      "epoch": 1.3761520385742188e-06,
      "model_forward_time": 1.6936943531036377,
      "step": 1443
    },
    {
      "epoch": 1.3761520385742188e-06,
      "step": 1443,
      "training_step_time": 9.23410940170288
    },
    {
      "epoch": 1.377105712890625e-06,
      "model_forward_time": 1.7080869674682617,
      "step": 1444
    },
    {
      "epoch": 1.377105712890625e-06,
      "step": 1444,
      "training_step_time": 7.776195764541626
    },
    {
      "epoch": 1.3780593872070313e-06,
      "model_forward_time": 2.207888126373291,
      "step": 1445
    },
    {
      "epoch": 1.3780593872070313e-06,
      "step": 1445,
      "training_step_time": 9.751443147659302
    },
    {
      "epoch": 1.3790130615234375e-06,
      "model_forward_time": 1.6945297718048096,
      "step": 1446
    },
    {
      "epoch": 1.3790130615234375e-06,
      "step": 1446,
      "training_step_time": 9.600330829620361
    },
    {
      "epoch": 1.3799667358398438e-06,
      "model_forward_time": 0.9788212776184082,
      "step": 1447
    },
    {
      "epoch": 1.3799667358398438e-06,
      "step": 1447,
      "training_step_time": 7.751277446746826
    },
    {
      "epoch": 1.38092041015625e-06,
      "model_forward_time": 1.6889722347259521,
      "step": 1448
    },
    {
      "epoch": 1.38092041015625e-06,
      "step": 1448,
      "training_step_time": 7.634974718093872
    },
    {
      "epoch": 1.3818740844726563e-06,
      "model_forward_time": 1.6910114288330078,
      "step": 1449
    },
    {
      "epoch": 1.3818740844726563e-06,
      "step": 1449,
      "training_step_time": 7.776273250579834
    },
    {
      "epoch": 1.3828277587890625e-06,
      "grad_norm": 1.8016228675842285,
      "learning_rate": 2.8980000000000005e-06,
      "loss": 0.1146,
      "step": 1450
    },
    {
      "epoch": 1.3828277587890625e-06,
      "model_forward_time": 1.6856787204742432,
      "step": 1450
    },
    {
      "dynamics_loss_avg": 0.09039479456841945,
      "epoch": 1.3828277587890625e-06,
      "step": 1450
    },
    {
      "action_loss_avg": 0.0568555885925889,
      "epoch": 1.3828277587890625e-06,
      "step": 1450
    },
    {
      "epoch": 1.3828277587890625e-06,
      "step": 1450,
      "training_step_time": 7.752326250076294
    },
    {
      "epoch": 1.3837814331054688e-06,
      "model_forward_time": 1.6914422512054443,
      "step": 1451
    },
    {
      "epoch": 1.3837814331054688e-06,
      "step": 1451,
      "training_step_time": 8.366526126861572
    },
    {
      "epoch": 1.384735107421875e-06,
      "model_forward_time": 1.6874217987060547,
      "step": 1452
    },
    {
      "epoch": 1.384735107421875e-06,
      "step": 1452,
      "training_step_time": 8.692860841751099
    },
    {
      "epoch": 1.3856887817382813e-06,
      "model_forward_time": 1.7391393184661865,
      "step": 1453
    },
    {
      "epoch": 1.3856887817382813e-06,
      "step": 1453,
      "training_step_time": 8.759940147399902
    },
    {
      "epoch": 1.3866424560546875e-06,
      "model_forward_time": 2.2232251167297363,
      "step": 1454
    },
    {
      "epoch": 1.3866424560546875e-06,
      "step": 1454,
      "training_step_time": 8.657510042190552
    },
    {
      "epoch": 1.3875961303710938e-06,
      "model_forward_time": 1.6967906951904297,
      "step": 1455
    },
    {
      "epoch": 1.3875961303710938e-06,
      "step": 1455,
      "training_step_time": 7.839918613433838
    },
    {
      "epoch": 1.3885498046875e-06,
      "model_forward_time": 1.6861944198608398,
      "step": 1456
    },
    {
      "epoch": 1.3885498046875e-06,
      "step": 1456,
      "training_step_time": 7.769624471664429
    },
    {
      "epoch": 1.3895034790039063e-06,
      "model_forward_time": 1.6844727993011475,
      "step": 1457
    },
    {
      "epoch": 1.3895034790039063e-06,
      "step": 1457,
      "training_step_time": 7.708942890167236
    },
    {
      "epoch": 1.3904571533203125e-06,
      "model_forward_time": 1.699012279510498,
      "step": 1458
    },
    {
      "epoch": 1.3904571533203125e-06,
      "step": 1458,
      "training_step_time": 8.013077974319458
    },
    {
      "epoch": 1.3914108276367188e-06,
      "model_forward_time": 1.687732219696045,
      "step": 1459
    },
    {
      "epoch": 1.3914108276367188e-06,
      "step": 1459,
      "training_step_time": 8.299984455108643
    },
    {
      "epoch": 1.392364501953125e-06,
      "grad_norm": 2.3339571952819824,
      "learning_rate": 2.9180000000000003e-06,
      "loss": 0.1216,
      "step": 1460
    },
    {
      "epoch": 1.392364501953125e-06,
      "model_forward_time": 1.6843242645263672,
      "step": 1460
    },
    {
      "dynamics_loss_avg": 0.07764823511242866,
      "epoch": 1.392364501953125e-06,
      "step": 1460
    },
    {
      "action_loss_avg": 0.04517902210354805,
      "epoch": 1.392364501953125e-06,
      "step": 1460
    },
    {
      "epoch": 1.392364501953125e-06,
      "step": 1460,
      "training_step_time": 7.626959323883057
    },
    {
      "epoch": 1.3933181762695313e-06,
      "model_forward_time": 1.6904025077819824,
      "step": 1461
    },
    {
      "epoch": 1.3933181762695313e-06,
      "step": 1461,
      "training_step_time": 7.7260847091674805
    },
    {
      "epoch": 1.3942718505859375e-06,
      "model_forward_time": 1.732271671295166,
      "step": 1462
    },
    {
      "epoch": 1.3942718505859375e-06,
      "step": 1462,
      "training_step_time": 7.761354684829712
    },
    {
      "epoch": 1.3952255249023438e-06,
      "model_forward_time": 2.2526955604553223,
      "step": 1463
    },
    {
      "epoch": 1.3952255249023438e-06,
      "step": 1463,
      "training_step_time": 8.547966718673706
    },
    {
      "epoch": 1.39617919921875e-06,
      "model_forward_time": 1.688596487045288,
      "step": 1464
    },
    {
      "epoch": 1.39617919921875e-06,
      "step": 1464,
      "training_step_time": 8.772592782974243
    },
    {
      "epoch": 1.3971328735351563e-06,
      "model_forward_time": 1.6872217655181885,
      "step": 1465
    },
    {
      "epoch": 1.3971328735351563e-06,
      "step": 1465,
      "training_step_time": 7.81185507774353
    },
    {
      "epoch": 1.3980865478515625e-06,
      "model_forward_time": 1.6799464225769043,
      "step": 1466
    },
    {
      "epoch": 1.3980865478515625e-06,
      "step": 1466,
      "training_step_time": 7.725491523742676
    },
    {
      "epoch": 1.3990402221679688e-06,
      "model_forward_time": 1.6866660118103027,
      "step": 1467
    },
    {
      "epoch": 1.3990402221679688e-06,
      "step": 1467,
      "training_step_time": 8.05305552482605
    },
    {
      "epoch": 1.399993896484375e-06,
      "model_forward_time": 0.9778501987457275,
      "step": 1468
    },
    {
      "epoch": 1.399993896484375e-06,
      "step": 1468,
      "training_step_time": 7.794013261795044
    },
    {
      "epoch": 1.4009475708007813e-06,
      "model_forward_time": 1.689864158630371,
      "step": 1469
    },
    {
      "epoch": 1.4009475708007813e-06,
      "step": 1469,
      "training_step_time": 7.5183000564575195
    },
    {
      "epoch": 1.4019012451171875e-06,
      "grad_norm": 3.7034459114074707,
      "learning_rate": 2.9380000000000005e-06,
      "loss": 0.1309,
      "step": 1470
    },
    {
      "epoch": 1.4019012451171875e-06,
      "model_forward_time": 1.6921355724334717,
      "step": 1470
    },
    {
      "dynamics_loss_avg": 0.09888758957386017,
      "epoch": 1.4019012451171875e-06,
      "step": 1470
    },
    {
      "action_loss_avg": 0.08343789279460907,
      "epoch": 1.4019012451171875e-06,
      "step": 1470
    },
    {
      "epoch": 1.4019012451171875e-06,
      "step": 1470,
      "training_step_time": 7.744152545928955
    },
    {
      "epoch": 1.4028549194335938e-06,
      "model_forward_time": 1.746619701385498,
      "step": 1471
    },
    {
      "epoch": 1.4028549194335938e-06,
      "step": 1471,
      "training_step_time": 7.766065359115601
    },
    {
      "epoch": 1.40380859375e-06,
      "model_forward_time": 1.7152986526489258,
      "step": 1472
    },
    {
      "epoch": 1.40380859375e-06,
      "step": 1472,
      "training_step_time": 8.291868925094604
    },
    {
      "epoch": 1.4047622680664063e-06,
      "model_forward_time": 2.244328260421753,
      "step": 1473
    },
    {
      "epoch": 1.4047622680664063e-06,
      "step": 1473,
      "training_step_time": 9.40184760093689
    },
    {
      "epoch": 1.4057159423828125e-06,
      "model_forward_time": 1.7112693786621094,
      "step": 1474
    },
    {
      "epoch": 1.4057159423828125e-06,
      "step": 1474,
      "training_step_time": 8.759032011032104
    },
    {
      "epoch": 1.4066696166992188e-06,
      "model_forward_time": 1.7100026607513428,
      "step": 1475
    },
    {
      "epoch": 1.4066696166992188e-06,
      "step": 1475,
      "training_step_time": 7.778141736984253
    },
    {
      "epoch": 1.407623291015625e-06,
      "model_forward_time": 1.7080085277557373,
      "step": 1476
    },
    {
      "epoch": 1.407623291015625e-06,
      "step": 1476,
      "training_step_time": 7.982346057891846
    },
    {
      "epoch": 1.4085769653320313e-06,
      "model_forward_time": 1.7134804725646973,
      "step": 1477
    },
    {
      "epoch": 1.4085769653320313e-06,
      "step": 1477,
      "training_step_time": 7.763259649276733
    },
    {
      "epoch": 1.4095306396484375e-06,
      "model_forward_time": 1.7067556381225586,
      "step": 1478
    },
    {
      "epoch": 1.4095306396484375e-06,
      "step": 1478,
      "training_step_time": 8.044649600982666
    },
    {
      "epoch": 1.4104843139648438e-06,
      "model_forward_time": 1.7138020992279053,
      "step": 1479
    },
    {
      "epoch": 1.4104843139648438e-06,
      "step": 1479,
      "training_step_time": 9.06879210472107
    },
    {
      "epoch": 1.41143798828125e-06,
      "grad_norm": 2.6542019844055176,
      "learning_rate": 2.9580000000000003e-06,
      "loss": 0.118,
      "step": 1480
    },
    {
      "epoch": 1.41143798828125e-06,
      "model_forward_time": 1.730891466140747,
      "step": 1480
    },
    {
      "dynamics_loss_avg": 0.07608223874121904,
      "epoch": 1.41143798828125e-06,
      "step": 1480
    },
    {
      "action_loss_avg": 0.04972158446907997,
      "epoch": 1.41143798828125e-06,
      "step": 1480
    },
    {
      "epoch": 1.41143798828125e-06,
      "step": 1480,
      "training_step_time": 7.8617753982543945
    },
    {
      "epoch": 1.4123916625976563e-06,
      "model_forward_time": 1.708831548690796,
      "step": 1481
    },
    {
      "epoch": 1.4123916625976563e-06,
      "step": 1481,
      "training_step_time": 7.761728048324585
    },
    {
      "epoch": 1.4133453369140625e-06,
      "model_forward_time": 2.2540085315704346,
      "step": 1482
    },
    {
      "epoch": 1.4133453369140625e-06,
      "step": 1482,
      "training_step_time": 8.00828242301941
    },
    {
      "epoch": 1.4142990112304688e-06,
      "model_forward_time": 1.7147338390350342,
      "step": 1483
    },
    {
      "epoch": 1.4142990112304688e-06,
      "step": 1483,
      "training_step_time": 7.841876268386841
    },
    {
      "epoch": 1.415252685546875e-06,
      "model_forward_time": 1.7108700275421143,
      "step": 1484
    },
    {
      "epoch": 1.415252685546875e-06,
      "step": 1484,
      "training_step_time": 7.831114292144775
    },
    {
      "epoch": 1.4162063598632813e-06,
      "model_forward_time": 1.7109277248382568,
      "step": 1485
    },
    {
      "epoch": 1.4162063598632813e-06,
      "step": 1485,
      "training_step_time": 8.098682403564453
    },
    {
      "epoch": 1.4171600341796875e-06,
      "model_forward_time": 1.679701805114746,
      "step": 1486
    },
    {
      "epoch": 1.4171600341796875e-06,
      "step": 1486,
      "training_step_time": 7.791704177856445
    },
    {
      "epoch": 1.4181137084960938e-06,
      "model_forward_time": 1.672194004058838,
      "step": 1487
    },
    {
      "epoch": 1.4181137084960938e-06,
      "step": 1487,
      "training_step_time": 7.574789047241211
    },
    {
      "epoch": 1.4190673828125e-06,
      "model_forward_time": 1.690565586090088,
      "step": 1488
    },
    {
      "epoch": 1.4190673828125e-06,
      "step": 1488,
      "training_step_time": 7.700024843215942
    },
    {
      "epoch": 1.4200210571289063e-06,
      "model_forward_time": 1.6902458667755127,
      "step": 1489
    },
    {
      "epoch": 1.4200210571289063e-06,
      "step": 1489,
      "training_step_time": 8.388180017471313
    },
    {
      "epoch": 1.4209747314453125e-06,
      "grad_norm": 1.4351179599761963,
      "learning_rate": 2.9780000000000005e-06,
      "loss": 0.1044,
      "step": 1490
    },
    {
      "epoch": 1.4209747314453125e-06,
      "model_forward_time": 1.688232183456421,
      "step": 1490
    },
    {
      "dynamics_loss_avg": 0.06495290696620941,
      "epoch": 1.4209747314453125e-06,
      "step": 1490
    },
    {
      "action_loss_avg": 0.0320509915240109,
      "epoch": 1.4209747314453125e-06,
      "step": 1490
    },
    {
      "epoch": 1.4209747314453125e-06,
      "step": 1490,
      "training_step_time": 8.660478830337524
    },
    {
      "epoch": 1.4219284057617188e-06,
      "model_forward_time": 2.2499144077301025,
      "step": 1491
    },
    {
      "epoch": 1.4219284057617188e-06,
      "step": 1491,
      "training_step_time": 8.759564399719238
    },
    {
      "epoch": 1.422882080078125e-06,
      "model_forward_time": 1.6827645301818848,
      "step": 1492
    },
    {
      "epoch": 1.422882080078125e-06,
      "step": 1492,
      "training_step_time": 7.865157842636108
    },
    {
      "epoch": 1.4238357543945313e-06,
      "model_forward_time": 1.6900403499603271,
      "step": 1493
    },
    {
      "epoch": 1.4238357543945313e-06,
      "step": 1493,
      "training_step_time": 8.45451545715332
    },
    {
      "epoch": 1.4247894287109375e-06,
      "model_forward_time": 1.6898319721221924,
      "step": 1494
    },
    {
      "epoch": 1.4247894287109375e-06,
      "step": 1494,
      "training_step_time": 8.48460865020752
    },
    {
      "epoch": 1.4257431030273439e-06,
      "model_forward_time": 1.6912105083465576,
      "step": 1495
    },
    {
      "epoch": 1.4257431030273439e-06,
      "step": 1495,
      "training_step_time": 7.929513692855835
    },
    {
      "epoch": 1.42669677734375e-06,
      "model_forward_time": 1.6965172290802002,
      "step": 1496
    },
    {
      "epoch": 1.42669677734375e-06,
      "step": 1496,
      "training_step_time": 7.472958564758301
    },
    {
      "epoch": 1.4276504516601564e-06,
      "model_forward_time": 1.6882407665252686,
      "step": 1497
    },
    {
      "epoch": 1.4276504516601564e-06,
      "step": 1497,
      "training_step_time": 7.572083473205566
    },
    {
      "epoch": 1.4286041259765625e-06,
      "model_forward_time": 1.6930651664733887,
      "step": 1498
    },
    {
      "epoch": 1.4286041259765625e-06,
      "step": 1498,
      "training_step_time": 7.790558576583862
    },
    {
      "epoch": 1.4295578002929689e-06,
      "model_forward_time": 1.7463574409484863,
      "step": 1499
    },
    {
      "epoch": 1.4295578002929689e-06,
      "step": 1499,
      "training_step_time": 7.663993835449219
    },
    {
      "epoch": 1.430511474609375e-06,
      "grad_norm": 2.108052968978882,
      "learning_rate": 2.9980000000000003e-06,
      "loss": 0.1096,
      "step": 1500
    },
    {
      "epoch": 1.430511474609375e-06,
      "model_forward_time": 2.229104995727539,
      "step": 1500
    },
    {
      "dynamics_loss_avg": 0.07273178771138192,
      "epoch": 1.430511474609375e-06,
      "step": 1500
    },
    {
      "action_loss_avg": 0.0325996246188879,
      "epoch": 1.430511474609375e-06,
      "step": 1500
    },
    {
      "epoch": 1.430511474609375e-06,
      "step": 1500,
      "training_step_time": 9.5269615650177
    },
    {
      "epoch": 1.4314651489257811e-06,
      "model_forward_time": 1.6905694007873535,
      "step": 1501
    },
    {
      "epoch": 1.4314651489257811e-06,
      "step": 1501,
      "training_step_time": 7.825380802154541
    },
    {
      "epoch": 1.4324188232421875e-06,
      "model_forward_time": 1.6940605640411377,
      "step": 1502
    },
    {
      "epoch": 1.4324188232421875e-06,
      "step": 1502,
      "training_step_time": 8.7193603515625
    },
    {
      "epoch": 1.4333724975585936e-06,
      "model_forward_time": 1.6819260120391846,
      "step": 1503
    },
    {
      "epoch": 1.4333724975585936e-06,
      "step": 1503,
      "training_step_time": 8.330385684967041
    },
    {
      "epoch": 1.434326171875e-06,
      "model_forward_time": 1.6900529861450195,
      "step": 1504
    },
    {
      "epoch": 1.434326171875e-06,
      "step": 1504,
      "training_step_time": 8.049509048461914
    },
    {
      "epoch": 1.4352798461914061e-06,
      "model_forward_time": 0.9775948524475098,
      "step": 1505
    },
    {
      "epoch": 1.4352798461914061e-06,
      "step": 1505,
      "training_step_time": 7.331044435501099
    },
    {
      "epoch": 1.4362335205078125e-06,
      "model_forward_time": 1.6884005069732666,
      "step": 1506
    },
    {
      "epoch": 1.4362335205078125e-06,
      "step": 1506,
      "training_step_time": 7.677265167236328
    },
    {
      "epoch": 1.4371871948242187e-06,
      "model_forward_time": 1.690354824066162,
      "step": 1507
    },
    {
      "epoch": 1.4371871948242187e-06,
      "step": 1507,
      "training_step_time": 7.880717754364014
    },
    {
      "epoch": 1.438140869140625e-06,
      "model_forward_time": 1.7378203868865967,
      "step": 1508
    },
    {
      "epoch": 1.438140869140625e-06,
      "step": 1508,
      "training_step_time": 7.727215766906738
    },
    {
      "epoch": 1.4390945434570312e-06,
      "model_forward_time": 1.6905851364135742,
      "step": 1509
    },
    {
      "epoch": 1.4390945434570312e-06,
      "step": 1509,
      "training_step_time": 7.8811187744140625
    },
    {
      "epoch": 1.4400482177734375e-06,
      "grad_norm": 1.7517611980438232,
      "learning_rate": 3.0180000000000006e-06,
      "loss": 0.1138,
      "step": 1510
    },
    {
      "epoch": 1.4400482177734375e-06,
      "model_forward_time": 2.227815628051758,
      "step": 1510
    },
    {
      "dynamics_loss_avg": 0.08619731441140174,
      "epoch": 1.4400482177734375e-06,
      "step": 1510
    },
    {
      "action_loss_avg": 0.03761579785495996,
      "epoch": 1.4400482177734375e-06,
      "step": 1510
    },
    {
      "epoch": 1.4400482177734375e-06,
      "step": 1510,
      "training_step_time": 9.646219730377197
    },
    {
      "epoch": 1.4410018920898437e-06,
      "model_forward_time": 1.6931538581848145,
      "step": 1511
    },
    {
      "epoch": 1.4410018920898437e-06,
      "step": 1511,
      "training_step_time": 7.826681137084961
    },
    {
      "epoch": 1.44195556640625e-06,
      "model_forward_time": 1.689993143081665,
      "step": 1512
    },
    {
      "epoch": 1.44195556640625e-06,
      "step": 1512,
      "training_step_time": 7.711989641189575
    },
    {
      "epoch": 1.4429092407226562e-06,
      "model_forward_time": 1.6863501071929932,
      "step": 1513
    },
    {
      "epoch": 1.4429092407226562e-06,
      "step": 1513,
      "training_step_time": 8.041580438613892
    },
    {
      "epoch": 1.4438629150390625e-06,
      "model_forward_time": 1.6916224956512451,
      "step": 1514
    },
    {
      "epoch": 1.4438629150390625e-06,
      "step": 1514,
      "training_step_time": 7.652088165283203
    },
    {
      "epoch": 1.4448165893554687e-06,
      "model_forward_time": 1.683229923248291,
      "step": 1515
    },
    {
      "epoch": 1.4448165893554687e-06,
      "step": 1515,
      "training_step_time": 7.677513122558594
    },
    {
      "epoch": 1.445770263671875e-06,
      "model_forward_time": 1.6913793087005615,
      "step": 1516
    },
    {
      "epoch": 1.445770263671875e-06,
      "step": 1516,
      "training_step_time": 7.733108043670654
    },
    {
      "epoch": 1.4467239379882812e-06,
      "model_forward_time": 1.7343437671661377,
      "step": 1517
    },
    {
      "epoch": 1.4467239379882812e-06,
      "step": 1517,
      "training_step_time": 10.619350671768188
    },
    {
      "epoch": 1.4476776123046875e-06,
      "model_forward_time": 1.6920619010925293,
      "step": 1518
    },
    {
      "epoch": 1.4476776123046875e-06,
      "step": 1518,
      "training_step_time": 7.874832391738892
    },
    {
      "epoch": 1.4486312866210937e-06,
      "model_forward_time": 2.2361652851104736,
      "step": 1519
    },
    {
      "epoch": 1.4486312866210937e-06,
      "step": 1519,
      "training_step_time": 9.712347030639648
    },
    {
      "epoch": 1.4495849609375e-06,
      "grad_norm": 1.7627562284469604,
      "learning_rate": 3.0380000000000004e-06,
      "loss": 0.1161,
      "step": 1520
    },
    {
      "epoch": 1.4495849609375e-06,
      "model_forward_time": 1.7090668678283691,
      "step": 1520
    },
    {
      "dynamics_loss_avg": 0.0802441582083702,
      "epoch": 1.4495849609375e-06,
      "step": 1520
    },
    {
      "action_loss_avg": 0.03923049662262201,
      "epoch": 1.4495849609375e-06,
      "step": 1520
    },
    {
      "epoch": 1.4495849609375e-06,
      "step": 1520,
      "training_step_time": 7.766649484634399
    },
    {
      "epoch": 1.4505386352539062e-06,
      "model_forward_time": 1.7084529399871826,
      "step": 1521
    },
    {
      "epoch": 1.4505386352539062e-06,
      "step": 1521,
      "training_step_time": 7.733700513839722
    },
    {
      "epoch": 1.4514923095703125e-06,
      "model_forward_time": 1.7090682983398438,
      "step": 1522
    },
    {
      "epoch": 1.4514923095703125e-06,
      "step": 1522,
      "training_step_time": 8.56942081451416
    },
    {
      "epoch": 1.4524459838867187e-06,
      "model_forward_time": 1.7136244773864746,
      "step": 1523
    },
    {
      "epoch": 1.4524459838867187e-06,
      "step": 1523,
      "training_step_time": 7.628777742385864
    },
    {
      "epoch": 1.453399658203125e-06,
      "model_forward_time": 1.703582525253296,
      "step": 1524
    },
    {
      "epoch": 1.453399658203125e-06,
      "step": 1524,
      "training_step_time": 7.655423641204834
    },
    {
      "epoch": 1.4543533325195312e-06,
      "model_forward_time": 1.7072618007659912,
      "step": 1525
    },
    {
      "epoch": 1.4543533325195312e-06,
      "step": 1525,
      "training_step_time": 7.657905101776123
    },
    {
      "epoch": 1.4553070068359375e-06,
      "model_forward_time": 1.7422707080841064,
      "step": 1526
    },
    {
      "epoch": 1.4553070068359375e-06,
      "step": 1526,
      "training_step_time": 8.570138454437256
    },
    {
      "epoch": 1.4562606811523437e-06,
      "model_forward_time": 1.7086386680603027,
      "step": 1527
    },
    {
      "epoch": 1.4562606811523437e-06,
      "step": 1527,
      "training_step_time": 8.266557693481445
    },
    {
      "epoch": 1.45721435546875e-06,
      "model_forward_time": 2.2460343837738037,
      "step": 1528
    },
    {
      "epoch": 1.45721435546875e-06,
      "step": 1528,
      "training_step_time": 7.809039354324341
    },
    {
      "epoch": 1.4581680297851562e-06,
      "model_forward_time": 1.7053868770599365,
      "step": 1529
    },
    {
      "epoch": 1.4581680297851562e-06,
      "step": 1529,
      "training_step_time": 7.927404165267944
    },
    {
      "epoch": 1.4591217041015625e-06,
      "grad_norm": 2.6750328540802,
      "learning_rate": 3.0580000000000006e-06,
      "loss": 0.1122,
      "step": 1530
    },
    {
      "epoch": 1.4591217041015625e-06,
      "model_forward_time": 1.702049970626831,
      "step": 1530
    },
    {
      "dynamics_loss_avg": 0.06951109766960144,
      "epoch": 1.4591217041015625e-06,
      "step": 1530
    },
    {
      "action_loss_avg": 0.0313365425914526,
      "epoch": 1.4591217041015625e-06,
      "step": 1530
    },
    {
      "epoch": 1.4591217041015625e-06,
      "step": 1530,
      "training_step_time": 7.839235067367554
    },
    {
      "epoch": 1.4600753784179687e-06,
      "model_forward_time": 1.7098309993743896,
      "step": 1531
    },
    {
      "epoch": 1.4600753784179687e-06,
      "step": 1531,
      "training_step_time": 7.784769058227539
    },
    {
      "epoch": 1.461029052734375e-06,
      "model_forward_time": 1.7080051898956299,
      "step": 1532
    },
    {
      "epoch": 1.461029052734375e-06,
      "step": 1532,
      "training_step_time": 8.005121946334839
    },
    {
      "epoch": 1.4619827270507812e-06,
      "model_forward_time": 1.703150987625122,
      "step": 1533
    },
    {
      "epoch": 1.4619827270507812e-06,
      "step": 1533,
      "training_step_time": 7.69072961807251
    },
    {
      "epoch": 1.4629364013671875e-06,
      "model_forward_time": 1.712010145187378,
      "step": 1534
    },
    {
      "epoch": 1.4629364013671875e-06,
      "step": 1534,
      "training_step_time": 7.675028085708618
    },
    {
      "epoch": 1.4638900756835937e-06,
      "model_forward_time": 1.7106902599334717,
      "step": 1535
    },
    {
      "epoch": 1.4638900756835937e-06,
      "step": 1535,
      "training_step_time": 9.347749948501587
    },
    {
      "epoch": 1.46484375e-06,
      "model_forward_time": 1.7125670909881592,
      "step": 1536
    },
    {
      "epoch": 1.46484375e-06,
      "step": 1536,
      "training_step_time": 7.689697504043579
    },
    {
      "epoch": 1.4657974243164062e-06,
      "model_forward_time": 2.2399966716766357,
      "step": 1537
    },
    {
      "epoch": 1.4657974243164062e-06,
      "step": 1537,
      "training_step_time": 8.879074096679688
    },
    {
      "epoch": 1.4667510986328125e-06,
      "model_forward_time": 1.713883876800537,
      "step": 1538
    },
    {
      "epoch": 1.4667510986328125e-06,
      "step": 1538,
      "training_step_time": 7.9349589347839355
    },
    {
      "epoch": 1.4677047729492187e-06,
      "model_forward_time": 1.7105226516723633,
      "step": 1539
    },
    {
      "epoch": 1.4677047729492187e-06,
      "step": 1539,
      "training_step_time": 9.401280641555786
    },
    {
      "epoch": 1.468658447265625e-06,
      "grad_norm": 1.903999924659729,
      "learning_rate": 3.0780000000000004e-06,
      "loss": 0.1233,
      "step": 1540
    },
    {
      "epoch": 1.468658447265625e-06,
      "model_forward_time": 1.7066986560821533,
      "step": 1540
    },
    {
      "dynamics_loss_avg": 0.08998305611312389,
      "epoch": 1.468658447265625e-06,
      "step": 1540
    },
    {
      "action_loss_avg": 0.038461006432771686,
      "epoch": 1.468658447265625e-06,
      "step": 1540
    },
    {
      "epoch": 1.468658447265625e-06,
      "step": 1540,
      "training_step_time": 8.7267324924469
    },
    {
      "epoch": 1.4696121215820312e-06,
      "model_forward_time": 1.7021856307983398,
      "step": 1541
    },
    {
      "epoch": 1.4696121215820312e-06,
      "step": 1541,
      "training_step_time": 9.062034845352173
    },
    {
      "epoch": 1.4705657958984375e-06,
      "model_forward_time": 1.7079436779022217,
      "step": 1542
    },
    {
      "epoch": 1.4705657958984375e-06,
      "step": 1542,
      "training_step_time": 7.704840421676636
    },
    {
      "epoch": 1.4715194702148437e-06,
      "model_forward_time": 1.3340375423431396,
      "step": 1543
    },
    {
      "epoch": 1.4715194702148437e-06,
      "step": 1543,
      "training_step_time": 7.636517286300659
    },
    {
      "epoch": 1.47247314453125e-06,
      "model_forward_time": 1.7122302055358887,
      "step": 1544
    },
    {
      "epoch": 1.47247314453125e-06,
      "step": 1544,
      "training_step_time": 9.55258321762085
    },
    {
      "epoch": 1.4734268188476562e-06,
      "model_forward_time": 1.733273983001709,
      "step": 1545
    },
    {
      "epoch": 1.4734268188476562e-06,
      "step": 1545,
      "training_step_time": 7.705962181091309
    },
    {
      "epoch": 1.4743804931640625e-06,
      "model_forward_time": 1.6995141506195068,
      "step": 1546
    },
    {
      "epoch": 1.4743804931640625e-06,
      "step": 1546,
      "training_step_time": 9.141878604888916
    },
    {
      "epoch": 1.4753341674804687e-06,
      "model_forward_time": 2.227130174636841,
      "step": 1547
    },
    {
      "epoch": 1.4753341674804687e-06,
      "step": 1547,
      "training_step_time": 8.031821727752686
    },
    {
      "epoch": 1.476287841796875e-06,
      "model_forward_time": 1.7066471576690674,
      "step": 1548
    },
    {
      "epoch": 1.476287841796875e-06,
      "step": 1548,
      "training_step_time": 7.938592195510864
    },
    {
      "epoch": 1.4772415161132812e-06,
      "model_forward_time": 1.7195377349853516,
      "step": 1549
    },
    {
      "epoch": 1.4772415161132812e-06,
      "step": 1549,
      "training_step_time": 7.70804762840271
    },
    {
      "epoch": 1.4781951904296875e-06,
      "grad_norm": 1.3798060417175293,
      "learning_rate": 3.0980000000000007e-06,
      "loss": 0.1159,
      "step": 1550
    },
    {
      "epoch": 1.4781951904296875e-06,
      "model_forward_time": 1.6979429721832275,
      "step": 1550
    },
    {
      "dynamics_loss_avg": 0.0791710700839758,
      "epoch": 1.4781951904296875e-06,
      "step": 1550
    },
    {
      "action_loss_avg": 0.05064977202564478,
      "epoch": 1.4781951904296875e-06,
      "step": 1550
    },
    {
      "epoch": 1.4781951904296875e-06,
      "step": 1550,
      "training_step_time": 9.092475652694702
    },
    {
      "epoch": 1.4791488647460937e-06,
      "model_forward_time": 1.71305513381958,
      "step": 1551
    },
    {
      "epoch": 1.4791488647460937e-06,
      "step": 1551,
      "training_step_time": 7.719259738922119
    },
    {
      "epoch": 1.4801025390625e-06,
      "model_forward_time": 1.7159075736999512,
      "step": 1552
    },
    {
      "epoch": 1.4801025390625e-06,
      "step": 1552,
      "training_step_time": 7.540290832519531
    },
    {
      "epoch": 1.4810562133789062e-06,
      "model_forward_time": 1.715510368347168,
      "step": 1553
    },
    {
      "epoch": 1.4810562133789062e-06,
      "step": 1553,
      "training_step_time": 8.81085991859436
    },
    {
      "epoch": 1.4820098876953125e-06,
      "model_forward_time": 1.7541604042053223,
      "step": 1554
    },
    {
      "epoch": 1.4820098876953125e-06,
      "step": 1554,
      "training_step_time": 8.344801425933838
    },
    {
      "epoch": 1.4829635620117187e-06,
      "model_forward_time": 1.7059822082519531,
      "step": 1555
    },
    {
      "epoch": 1.4829635620117187e-06,
      "step": 1555,
      "training_step_time": 8.596367359161377
    },
    {
      "epoch": 1.483917236328125e-06,
      "model_forward_time": 2.2550227642059326,
      "step": 1556
    },
    {
      "epoch": 1.483917236328125e-06,
      "step": 1556,
      "training_step_time": 7.835200548171997
    },
    {
      "epoch": 1.4848709106445312e-06,
      "model_forward_time": 1.3400180339813232,
      "step": 1557
    },
    {
      "epoch": 1.4848709106445312e-06,
      "step": 1557,
      "training_step_time": 9.479917526245117
    },
    {
      "epoch": 1.4858245849609375e-06,
      "model_forward_time": 1.720367431640625,
      "step": 1558
    },
    {
      "epoch": 1.4858245849609375e-06,
      "step": 1558,
      "training_step_time": 7.942259311676025
    },
    {
      "epoch": 1.4867782592773437e-06,
      "model_forward_time": 1.7181005477905273,
      "step": 1559
    },
    {
      "epoch": 1.4867782592773437e-06,
      "step": 1559,
      "training_step_time": 8.05812931060791
    },
    {
      "epoch": 1.48773193359375e-06,
      "grad_norm": 1.8160886764526367,
      "learning_rate": 3.1180000000000005e-06,
      "loss": 0.1117,
      "step": 1560
    },
    {
      "epoch": 1.48773193359375e-06,
      "model_forward_time": 1.7090694904327393,
      "step": 1560
    },
    {
      "dynamics_loss_avg": 0.07579473406076431,
      "epoch": 1.48773193359375e-06,
      "step": 1560
    },
    {
      "action_loss_avg": 0.03869277872145176,
      "epoch": 1.48773193359375e-06,
      "step": 1560
    },
    {
      "epoch": 1.48773193359375e-06,
      "step": 1560,
      "training_step_time": 7.701178789138794
    },
    {
      "epoch": 1.4886856079101562e-06,
      "model_forward_time": 1.7164723873138428,
      "step": 1561
    },
    {
      "epoch": 1.4886856079101562e-06,
      "step": 1561,
      "training_step_time": 7.581380128860474
    },
    {
      "epoch": 1.4896392822265626e-06,
      "model_forward_time": 1.7136743068695068,
      "step": 1562
    },
    {
      "epoch": 1.4896392822265626e-06,
      "step": 1562,
      "training_step_time": 7.733064413070679
    },
    {
      "epoch": 1.4905929565429687e-06,
      "model_forward_time": 1.7565786838531494,
      "step": 1563
    },
    {
      "epoch": 1.4905929565429687e-06,
      "step": 1563,
      "training_step_time": 7.9803900718688965
    },
    {
      "epoch": 1.491546630859375e-06,
      "model_forward_time": 1.7131106853485107,
      "step": 1564
    },
    {
      "epoch": 1.491546630859375e-06,
      "step": 1564,
      "training_step_time": 8.937859773635864
    },
    {
      "epoch": 1.4925003051757812e-06,
      "model_forward_time": 2.24448823928833,
      "step": 1565
    },
    {
      "epoch": 1.4925003051757812e-06,
      "step": 1565,
      "training_step_time": 9.245810508728027
    },
    {
      "epoch": 1.4934539794921876e-06,
      "model_forward_time": 1.7128736972808838,
      "step": 1566
    },
    {
      "epoch": 1.4934539794921876e-06,
      "step": 1566,
      "training_step_time": 8.488790512084961
    },
    {
      "epoch": 1.4944076538085937e-06,
      "model_forward_time": 1.7130794525146484,
      "step": 1567
    },
    {
      "epoch": 1.4944076538085937e-06,
      "step": 1567,
      "training_step_time": 8.594344854354858
    },
    {
      "epoch": 1.495361328125e-06,
      "model_forward_time": 1.6917071342468262,
      "step": 1568
    },
    {
      "epoch": 1.495361328125e-06,
      "step": 1568,
      "training_step_time": 8.320809125900269
    },
    {
      "epoch": 1.4963150024414062e-06,
      "model_forward_time": 1.7130448818206787,
      "step": 1569
    },
    {
      "epoch": 1.4963150024414062e-06,
      "step": 1569,
      "training_step_time": 7.757186412811279
    },
    {
      "epoch": 1.4972686767578126e-06,
      "grad_norm": 1.6207380294799805,
      "learning_rate": 3.1380000000000003e-06,
      "loss": 0.1182,
      "step": 1570
    },
    {
      "epoch": 1.4972686767578126e-06,
      "model_forward_time": 1.7114408016204834,
      "step": 1570
    },
    {
      "dynamics_loss_avg": 0.0681289505213499,
      "epoch": 1.4972686767578126e-06,
      "step": 1570
    },
    {
      "action_loss_avg": 0.03077774392440915,
      "epoch": 1.4972686767578126e-06,
      "step": 1570
    },
    {
      "epoch": 1.4972686767578126e-06,
      "step": 1570,
      "training_step_time": 7.605692386627197
    },
    {
      "epoch": 1.4982223510742187e-06,
      "model_forward_time": 1.716381549835205,
      "step": 1571
    },
    {
      "epoch": 1.4982223510742187e-06,
      "step": 1571,
      "training_step_time": 7.795737028121948
    },
    {
      "epoch": 1.499176025390625e-06,
      "model_forward_time": 1.7590250968933105,
      "step": 1572
    },
    {
      "epoch": 1.499176025390625e-06,
      "step": 1572,
      "training_step_time": 9.514140605926514
    },
    {
      "epoch": 1.5001296997070312e-06,
      "model_forward_time": 1.6972742080688477,
      "step": 1573
    },
    {
      "epoch": 1.5001296997070312e-06,
      "step": 1573,
      "training_step_time": 7.92767333984375
    },
    {
      "epoch": 1.5010833740234376e-06,
      "model_forward_time": 2.2382125854492188,
      "step": 1574
    },
    {
      "epoch": 1.5010833740234376e-06,
      "step": 1574,
      "training_step_time": 7.84693455696106
    },
    {
      "epoch": 1.5020370483398437e-06,
      "model_forward_time": 1.7112047672271729,
      "step": 1575
    },
    {
      "epoch": 1.5020370483398437e-06,
      "step": 1575,
      "training_step_time": 8.81523084640503
    },
    {
      "epoch": 1.50299072265625e-06,
      "model_forward_time": 1.7091422080993652,
      "step": 1576
    },
    {
      "epoch": 1.50299072265625e-06,
      "step": 1576,
      "training_step_time": 7.88356876373291
    },
    {
      "epoch": 1.5039443969726562e-06,
      "model_forward_time": 1.7149393558502197,
      "step": 1577
    },
    {
      "epoch": 1.5039443969726562e-06,
      "step": 1577,
      "training_step_time": 8.790143489837646
    },
    {
      "epoch": 1.5048980712890626e-06,
      "model_forward_time": 1.7104666233062744,
      "step": 1578
    },
    {
      "epoch": 1.5048980712890626e-06,
      "step": 1578,
      "training_step_time": 7.742221117019653
    },
    {
      "epoch": 1.5058517456054687e-06,
      "model_forward_time": 1.7105011940002441,
      "step": 1579
    },
    {
      "epoch": 1.5058517456054687e-06,
      "step": 1579,
      "training_step_time": 7.714502573013306
    },
    {
      "epoch": 1.506805419921875e-06,
      "grad_norm": 2.114297866821289,
      "learning_rate": 3.1580000000000005e-06,
      "loss": 0.1102,
      "step": 1580
    },
    {
      "epoch": 1.506805419921875e-06,
      "model_forward_time": 1.7139744758605957,
      "step": 1580
    },
    {
      "dynamics_loss_avg": 0.09301335923373699,
      "epoch": 1.506805419921875e-06,
      "step": 1580
    },
    {
      "action_loss_avg": 0.04510025484487414,
      "epoch": 1.506805419921875e-06,
      "step": 1580
    },
    {
      "epoch": 1.506805419921875e-06,
      "step": 1580,
      "training_step_time": 7.8384833335876465
    },
    {
      "epoch": 1.5077590942382812e-06,
      "model_forward_time": 1.7505402565002441,
      "step": 1581
    },
    {
      "epoch": 1.5077590942382812e-06,
      "step": 1581,
      "training_step_time": 8.558334827423096
    },
    {
      "epoch": 1.5087127685546876e-06,
      "model_forward_time": 1.7132773399353027,
      "step": 1582
    },
    {
      "epoch": 1.5087127685546876e-06,
      "step": 1582,
      "training_step_time": 7.846640110015869
    },
    {
      "epoch": 1.5096664428710937e-06,
      "model_forward_time": 1.7066941261291504,
      "step": 1583
    },
    {
      "epoch": 1.5096664428710937e-06,
      "step": 1583,
      "training_step_time": 8.505318403244019
    },
    {
      "epoch": 1.5106201171875e-06,
      "model_forward_time": 2.2441928386688232,
      "step": 1584
    },
    {
      "epoch": 1.5106201171875e-06,
      "step": 1584,
      "training_step_time": 8.881876945495605
    },
    {
      "epoch": 1.5115737915039062e-06,
      "model_forward_time": 1.7084922790527344,
      "step": 1585
    },
    {
      "epoch": 1.5115737915039062e-06,
      "step": 1585,
      "training_step_time": 7.962121248245239
    },
    {
      "epoch": 1.5125274658203126e-06,
      "model_forward_time": 1.7097768783569336,
      "step": 1586
    },
    {
      "epoch": 1.5125274658203126e-06,
      "step": 1586,
      "training_step_time": 7.904957294464111
    },
    {
      "epoch": 1.5134811401367187e-06,
      "model_forward_time": 1.7145087718963623,
      "step": 1587
    },
    {
      "epoch": 1.5134811401367187e-06,
      "step": 1587,
      "training_step_time": 8.033093452453613
    },
    {
      "epoch": 1.514434814453125e-06,
      "model_forward_time": 1.7044739723205566,
      "step": 1588
    },
    {
      "epoch": 1.514434814453125e-06,
      "step": 1588,
      "training_step_time": 8.649942874908447
    },
    {
      "epoch": 1.5153884887695312e-06,
      "model_forward_time": 1.7159230709075928,
      "step": 1589
    },
    {
      "epoch": 1.5153884887695312e-06,
      "step": 1589,
      "training_step_time": 7.771721839904785
    },
    {
      "epoch": 1.5163421630859376e-06,
      "grad_norm": 2.933586835861206,
      "learning_rate": 3.1780000000000003e-06,
      "loss": 0.1095,
      "step": 1590
    },
    {
      "epoch": 1.5163421630859376e-06,
      "model_forward_time": 1.7350640296936035,
      "step": 1590
    },
    {
      "dynamics_loss_avg": 0.05672355405986309,
      "epoch": 1.5163421630859376e-06,
      "step": 1590
    },
    {
      "action_loss_avg": 0.022040956374257802,
      "epoch": 1.5163421630859376e-06,
      "step": 1590
    },
    {
      "epoch": 1.5163421630859376e-06,
      "step": 1590,
      "training_step_time": 7.981479644775391
    },
    {
      "epoch": 1.5172958374023437e-06,
      "model_forward_time": 1.7021408081054688,
      "step": 1591
    },
    {
      "epoch": 1.5172958374023437e-06,
      "step": 1591,
      "training_step_time": 7.816952466964722
    },
    {
      "epoch": 1.51824951171875e-06,
      "model_forward_time": 1.721038579940796,
      "step": 1592
    },
    {
      "epoch": 1.51824951171875e-06,
      "step": 1592,
      "training_step_time": 9.660035848617554
    },
    {
      "epoch": 1.5192031860351562e-06,
      "model_forward_time": 1.5330314636230469,
      "step": 1593
    },
    {
      "epoch": 1.5192031860351562e-06,
      "step": 1593,
      "training_step_time": 9.418675661087036
    },
    {
      "epoch": 1.5201568603515626e-06,
      "model_forward_time": 1.7038824558258057,
      "step": 1594
    },
    {
      "epoch": 1.5201568603515626e-06,
      "step": 1594,
      "training_step_time": 8.973529577255249
    },
    {
      "epoch": 1.5211105346679687e-06,
      "model_forward_time": 1.715155839920044,
      "step": 1595
    },
    {
      "epoch": 1.5211105346679687e-06,
      "step": 1595,
      "training_step_time": 7.8319525718688965
    },
    {
      "epoch": 1.522064208984375e-06,
      "model_forward_time": 1.711327075958252,
      "step": 1596
    },
    {
      "epoch": 1.522064208984375e-06,
      "step": 1596,
      "training_step_time": 7.991617441177368
    },
    {
      "epoch": 1.5230178833007812e-06,
      "model_forward_time": 1.7045705318450928,
      "step": 1597
    },
    {
      "epoch": 1.5230178833007812e-06,
      "step": 1597,
      "training_step_time": 8.745157957077026
    },
    {
      "epoch": 1.5239715576171876e-06,
      "model_forward_time": 1.7119147777557373,
      "step": 1598
    },
    {
      "epoch": 1.5239715576171876e-06,
      "step": 1598,
      "training_step_time": 7.749162673950195
    },
    {
      "epoch": 1.5249252319335937e-06,
      "model_forward_time": 1.7200160026550293,
      "step": 1599
    },
    {
      "epoch": 1.5249252319335937e-06,
      "step": 1599,
      "training_step_time": 7.961718797683716
    },
    {
      "epoch": 1.52587890625e-06,
      "grad_norm": 1.724165439605713,
      "learning_rate": 3.198e-06,
      "loss": 0.1078,
      "step": 1600
    },
    {
      "epoch": 1.52587890625e-06,
      "model_forward_time": 1.702829360961914,
      "step": 1600
    },
    {
      "dynamics_loss_avg": 0.08216851949691772,
      "epoch": 1.52587890625e-06,
      "step": 1600
    },
    {
      "action_loss_avg": 0.02969840317964554,
      "epoch": 1.52587890625e-06,
      "step": 1600
    },
    {
      "epoch": 1.52587890625e-06,
      "step": 1600,
      "training_step_time": 7.848381519317627
    },
    {
      "epoch": 1.5268325805664062e-06,
      "model_forward_time": 1.7036640644073486,
      "step": 1601
    },
    {
      "epoch": 1.5268325805664062e-06,
      "step": 1601,
      "training_step_time": 7.937057018280029
    },
    {
      "epoch": 1.5277862548828126e-06,
      "model_forward_time": 2.234563112258911,
      "step": 1602
    },
    {
      "epoch": 1.5277862548828126e-06,
      "step": 1602,
      "training_step_time": 8.916605472564697
    },
    {
      "epoch": 1.5287399291992187e-06,
      "model_forward_time": 1.7179980278015137,
      "step": 1603
    },
    {
      "epoch": 1.5287399291992187e-06,
      "step": 1603,
      "training_step_time": 7.856621026992798
    },
    {
      "epoch": 1.529693603515625e-06,
      "model_forward_time": 1.7120413780212402,
      "step": 1604
    },
    {
      "epoch": 1.529693603515625e-06,
      "step": 1604,
      "training_step_time": 8.595603466033936
    },
    {
      "epoch": 1.5306472778320312e-06,
      "model_forward_time": 1.7171947956085205,
      "step": 1605
    },
    {
      "epoch": 1.5306472778320312e-06,
      "step": 1605,
      "training_step_time": 7.882075309753418
    },
    {
      "epoch": 1.5316009521484376e-06,
      "model_forward_time": 1.7058746814727783,
      "step": 1606
    },
    {
      "epoch": 1.5316009521484376e-06,
      "step": 1606,
      "training_step_time": 8.0408616065979
    },
    {
      "epoch": 1.5325546264648437e-06,
      "model_forward_time": 1.7161178588867188,
      "step": 1607
    },
    {
      "epoch": 1.5325546264648437e-06,
      "step": 1607,
      "training_step_time": 7.755718946456909
    },
    {
      "epoch": 1.53350830078125e-06,
      "model_forward_time": 1.7094473838806152,
      "step": 1608
    },
    {
      "epoch": 1.53350830078125e-06,
      "step": 1608,
      "training_step_time": 7.962294816970825
    },
    {
      "epoch": 1.5344619750976562e-06,
      "model_forward_time": 1.7099335193634033,
      "step": 1609
    },
    {
      "epoch": 1.5344619750976562e-06,
      "step": 1609,
      "training_step_time": 9.576327800750732
    },
    {
      "epoch": 1.5354156494140626e-06,
      "grad_norm": 1.5201350450515747,
      "learning_rate": 3.218e-06,
      "loss": 0.1149,
      "step": 1610
    },
    {
      "epoch": 1.5354156494140626e-06,
      "model_forward_time": 0.9960677623748779,
      "step": 1610
    },
    {
      "dynamics_loss_avg": 0.06840770840644836,
      "epoch": 1.5354156494140626e-06,
      "step": 1610
    },
    {
      "action_loss_avg": 0.030939529463648797,
      "epoch": 1.5354156494140626e-06,
      "step": 1610
    },
    {
      "epoch": 1.5354156494140626e-06,
      "step": 1610,
      "training_step_time": 8.01408338546753
    },
    {
      "epoch": 1.5363693237304687e-06,
      "model_forward_time": 1.7158699035644531,
      "step": 1611
    },
    {
      "epoch": 1.5363693237304687e-06,
      "step": 1611,
      "training_step_time": 9.87850022315979
    },
    {
      "epoch": 1.537322998046875e-06,
      "model_forward_time": 1.8574550151824951,
      "step": 1612
    },
    {
      "epoch": 1.537322998046875e-06,
      "step": 1612,
      "training_step_time": 8.124160289764404
    },
    {
      "epoch": 1.5382766723632812e-06,
      "model_forward_time": 1.7119457721710205,
      "step": 1613
    },
    {
      "epoch": 1.5382766723632812e-06,
      "step": 1613,
      "training_step_time": 7.840737581253052
    },
    {
      "epoch": 1.5392303466796876e-06,
      "model_forward_time": 1.7231979370117188,
      "step": 1614
    },
    {
      "epoch": 1.5392303466796876e-06,
      "step": 1614,
      "training_step_time": 8.975059509277344
    },
    {
      "epoch": 1.5401840209960937e-06,
      "model_forward_time": 1.7100515365600586,
      "step": 1615
    },
    {
      "epoch": 1.5401840209960937e-06,
      "step": 1615,
      "training_step_time": 10.094064235687256
    },
    {
      "epoch": 1.5411376953125e-06,
      "model_forward_time": 1.7127494812011719,
      "step": 1616
    },
    {
      "epoch": 1.5411376953125e-06,
      "step": 1616,
      "training_step_time": 7.011090993881226
    },
    {
      "epoch": 1.5420913696289062e-06,
      "model_forward_time": 1.717714786529541,
      "step": 1617
    },
    {
      "epoch": 1.5420913696289062e-06,
      "step": 1617,
      "training_step_time": 6.96546196937561
    },
    {
      "epoch": 1.5430450439453126e-06,
      "model_forward_time": 1.710078477859497,
      "step": 1618
    },
    {
      "epoch": 1.5430450439453126e-06,
      "step": 1618,
      "training_step_time": 6.954319715499878
    },
    {
      "epoch": 1.5439987182617187e-06,
      "model_forward_time": 1.698453426361084,
      "step": 1619
    },
    {
      "epoch": 1.5439987182617187e-06,
      "step": 1619,
      "training_step_time": 6.96852970123291
    },
    {
      "epoch": 1.544952392578125e-06,
      "grad_norm": 1.060290813446045,
      "learning_rate": 3.238e-06,
      "loss": 0.1098,
      "step": 1620
    },
    {
      "epoch": 1.544952392578125e-06,
      "model_forward_time": 1.71266508102417,
      "step": 1620
    },
    {
      "dynamics_loss_avg": 0.08117041513323783,
      "epoch": 1.544952392578125e-06,
      "step": 1620
    },
    {
      "action_loss_avg": 0.037264634110033515,
      "epoch": 1.544952392578125e-06,
      "step": 1620
    },
    {
      "epoch": 1.544952392578125e-06,
      "step": 1620,
      "training_step_time": 6.936599016189575
    },
    {
      "epoch": 1.5459060668945312e-06,
      "model_forward_time": 1.7096312046051025,
      "step": 1621
    },
    {
      "epoch": 1.5459060668945312e-06,
      "step": 1621,
      "training_step_time": 6.940028429031372
    },
    {
      "epoch": 1.5468597412109376e-06,
      "model_forward_time": 1.706679344177246,
      "step": 1622
    },
    {
      "epoch": 1.5468597412109376e-06,
      "step": 1622,
      "training_step_time": 6.918571949005127
    },
    {
      "epoch": 1.5478134155273437e-06,
      "model_forward_time": 1.700300693511963,
      "step": 1623
    },
    {
      "epoch": 1.5478134155273437e-06,
      "step": 1623,
      "training_step_time": 6.967610597610474
    },
    {
      "epoch": 1.54876708984375e-06,
      "model_forward_time": 2.2347733974456787,
      "step": 1624
    },
    {
      "epoch": 1.54876708984375e-06,
      "step": 1624,
      "training_step_time": 7.596039295196533
    },
    {
      "epoch": 1.5497207641601562e-06,
      "model_forward_time": 1.7042772769927979,
      "step": 1625
    },
    {
      "epoch": 1.5497207641601562e-06,
      "step": 1625,
      "training_step_time": 8.008153676986694
    },
    {
      "epoch": 1.5506744384765624e-06,
      "model_forward_time": 1.704580307006836,
      "step": 1626
    },
    {
      "epoch": 1.5506744384765624e-06,
      "step": 1626,
      "training_step_time": 6.9808290004730225
    },
    {
      "epoch": 1.5516281127929688e-06,
      "model_forward_time": 1.7071821689605713,
      "step": 1627
    },
    {
      "epoch": 1.5516281127929688e-06,
      "step": 1627,
      "training_step_time": 6.961753845214844
    },
    {
      "epoch": 1.552581787109375e-06,
      "model_forward_time": 1.7005367279052734,
      "step": 1628
    },
    {
      "epoch": 1.552581787109375e-06,
      "step": 1628,
      "training_step_time": 6.952253103256226
    },
    {
      "epoch": 1.5535354614257813e-06,
      "model_forward_time": 1.7037723064422607,
      "step": 1629
    },
    {
      "epoch": 1.5535354614257813e-06,
      "step": 1629,
      "training_step_time": 6.9794065952301025
    },
    {
      "epoch": 1.5544891357421874e-06,
      "grad_norm": 2.2477920055389404,
      "learning_rate": 3.258e-06,
      "loss": 0.1033,
      "step": 1630
    },
    {
      "epoch": 1.5544891357421874e-06,
      "model_forward_time": 1.331371784210205,
      "step": 1630
    },
    {
      "dynamics_loss_avg": 0.07372984308749438,
      "epoch": 1.5544891357421874e-06,
      "step": 1630
    },
    {
      "action_loss_avg": 0.046476914826780555,
      "epoch": 1.5544891357421874e-06,
      "step": 1630
    },
    {
      "epoch": 1.5544891357421874e-06,
      "step": 1630,
      "training_step_time": 6.982332229614258
    },
    {
      "epoch": 1.5554428100585938e-06,
      "model_forward_time": 1.699622631072998,
      "step": 1631
    },
    {
      "epoch": 1.5554428100585938e-06,
      "step": 1631,
      "training_step_time": 6.966156721115112
    },
    {
      "epoch": 1.556396484375e-06,
      "model_forward_time": 1.713148593902588,
      "step": 1632
    },
    {
      "epoch": 1.556396484375e-06,
      "step": 1632,
      "training_step_time": 6.9209229946136475
    },
    {
      "epoch": 1.5573501586914063e-06,
      "model_forward_time": 2.23854923248291,
      "step": 1633
    },
    {
      "epoch": 1.5573501586914063e-06,
      "step": 1633,
      "training_step_time": 7.644574403762817
    },
    {
      "epoch": 1.5583038330078124e-06,
      "model_forward_time": 1.7061617374420166,
      "step": 1634
    },
    {
      "epoch": 1.5583038330078124e-06,
      "step": 1634,
      "training_step_time": 8.020072937011719
    },
    {
      "epoch": 1.5592575073242188e-06,
      "model_forward_time": 1.713789701461792,
      "step": 1635
    },
    {
      "epoch": 1.5592575073242188e-06,
      "step": 1635,
      "training_step_time": 7.372762680053711
    },
    {
      "epoch": 1.560211181640625e-06,
      "model_forward_time": 1.7069902420043945,
      "step": 1636
    },
    {
      "epoch": 1.560211181640625e-06,
      "step": 1636,
      "training_step_time": 6.924696207046509
    },
    {
      "epoch": 1.5611648559570313e-06,
      "model_forward_time": 1.7103800773620605,
      "step": 1637
    },
    {
      "epoch": 1.5611648559570313e-06,
      "step": 1637,
      "training_step_time": 6.960899114608765
    },
    {
      "epoch": 1.5621185302734374e-06,
      "model_forward_time": 1.7055244445800781,
      "step": 1638
    },
    {
      "epoch": 1.5621185302734374e-06,
      "step": 1638,
      "training_step_time": 6.904224872589111
    },
    {
      "epoch": 1.5630722045898438e-06,
      "model_forward_time": 1.7063429355621338,
      "step": 1639
    },
    {
      "epoch": 1.5630722045898438e-06,
      "step": 1639,
      "training_step_time": 6.98886775970459
    },
    {
      "epoch": 1.56402587890625e-06,
      "grad_norm": 2.112579345703125,
      "learning_rate": 3.278e-06,
      "loss": 0.1012,
      "step": 1640
    },
    {
      "epoch": 1.56402587890625e-06,
      "model_forward_time": 1.7166297435760498,
      "step": 1640
    },
    {
      "dynamics_loss_avg": 0.059893027693033216,
      "epoch": 1.56402587890625e-06,
      "step": 1640
    },
    {
      "action_loss_avg": 0.026419718656688927,
      "epoch": 1.56402587890625e-06,
      "step": 1640
    },
    {
      "epoch": 1.56402587890625e-06,
      "step": 1640,
      "training_step_time": 6.945155382156372
    },
    {
      "epoch": 1.5649795532226563e-06,
      "model_forward_time": 1.7124965190887451,
      "step": 1641
    },
    {
      "epoch": 1.5649795532226563e-06,
      "step": 1641,
      "training_step_time": 6.939344644546509
    },
    {
      "epoch": 1.5659332275390624e-06,
      "model_forward_time": 2.242002487182617,
      "step": 1642
    },
    {
      "epoch": 1.5659332275390624e-06,
      "step": 1642,
      "training_step_time": 7.633488416671753
    },
    {
      "epoch": 1.5668869018554688e-06,
      "model_forward_time": 1.7105183601379395,
      "step": 1643
    },
    {
      "epoch": 1.5668869018554688e-06,
      "step": 1643,
      "training_step_time": 8.870602130889893
    },
    {
      "epoch": 1.567840576171875e-06,
      "model_forward_time": 1.7119503021240234,
      "step": 1644
    },
    {
      "epoch": 1.567840576171875e-06,
      "step": 1644,
      "training_step_time": 7.720573663711548
    },
    {
      "epoch": 1.5687942504882813e-06,
      "model_forward_time": 1.706160306930542,
      "step": 1645
    },
    {
      "epoch": 1.5687942504882813e-06,
      "step": 1645,
      "training_step_time": 6.9896392822265625
    },
    {
      "epoch": 1.5697479248046874e-06,
      "model_forward_time": 1.7065849304199219,
      "step": 1646
    },
    {
      "epoch": 1.5697479248046874e-06,
      "step": 1646,
      "training_step_time": 6.956559658050537
    },
    {
      "epoch": 1.5707015991210938e-06,
      "model_forward_time": 1.7187774181365967,
      "step": 1647
    },
    {
      "epoch": 1.5707015991210938e-06,
      "step": 1647,
      "training_step_time": 6.945791482925415
    },
    {
      "epoch": 1.5716552734375e-06,
      "model_forward_time": 1.7143328189849854,
      "step": 1648
    },
    {
      "epoch": 1.5716552734375e-06,
      "step": 1648,
      "training_step_time": 6.930038690567017
    },
    {
      "epoch": 1.5726089477539063e-06,
      "model_forward_time": 1.709587574005127,
      "step": 1649
    },
    {
      "epoch": 1.5726089477539063e-06,
      "step": 1649,
      "training_step_time": 7.014553546905518
    },
    {
      "epoch": 1.5735626220703124e-06,
      "grad_norm": 1.2478783130645752,
      "learning_rate": 3.298e-06,
      "loss": 0.1084,
      "step": 1650
    },
    {
      "epoch": 1.5735626220703124e-06,
      "model_forward_time": 1.701655626296997,
      "step": 1650
    },
    {
      "dynamics_loss_avg": 0.07929955199360847,
      "epoch": 1.5735626220703124e-06,
      "step": 1650
    },
    {
      "action_loss_avg": 0.02501370422542095,
      "epoch": 1.5735626220703124e-06,
      "step": 1650
    },
    {
      "epoch": 1.5735626220703124e-06,
      "step": 1650,
      "training_step_time": 6.936467885971069
    },
    {
      "epoch": 1.5745162963867188e-06,
      "model_forward_time": 2.2555184364318848,
      "step": 1651
    },
    {
      "epoch": 1.5745162963867188e-06,
      "step": 1651,
      "training_step_time": 7.642589330673218
    },
    {
      "epoch": 1.575469970703125e-06,
      "model_forward_time": 1.7101736068725586,
      "step": 1652
    },
    {
      "epoch": 1.575469970703125e-06,
      "step": 1652,
      "training_step_time": 8.998317956924438
    },
    {
      "epoch": 1.5764236450195313e-06,
      "model_forward_time": 1.709057092666626,
      "step": 1653
    },
    {
      "epoch": 1.5764236450195313e-06,
      "step": 1653,
      "training_step_time": 7.647222518920898
    },
    {
      "epoch": 1.5773773193359374e-06,
      "model_forward_time": 1.722606897354126,
      "step": 1654
    },
    {
      "epoch": 1.5773773193359374e-06,
      "step": 1654,
      "training_step_time": 6.902855157852173
    },
    {
      "epoch": 1.5783309936523438e-06,
      "model_forward_time": 1.7197132110595703,
      "step": 1655
    },
    {
      "epoch": 1.5783309936523438e-06,
      "step": 1655,
      "training_step_time": 6.8393166065216064
    },
    {
      "epoch": 1.57928466796875e-06,
      "model_forward_time": 1.716904640197754,
      "step": 1656
    },
    {
      "epoch": 1.57928466796875e-06,
      "step": 1656,
      "training_step_time": 6.919893264770508
    },
    {
      "epoch": 1.5802383422851563e-06,
      "model_forward_time": 1.7151925563812256,
      "step": 1657
    },
    {
      "epoch": 1.5802383422851563e-06,
      "step": 1657,
      "training_step_time": 6.877021074295044
    },
    {
      "epoch": 1.5811920166015624e-06,
      "model_forward_time": 1.7191264629364014,
      "step": 1658
    },
    {
      "epoch": 1.5811920166015624e-06,
      "step": 1658,
      "training_step_time": 6.87123703956604
    },
    {
      "epoch": 1.5821456909179688e-06,
      "model_forward_time": 1.7192976474761963,
      "step": 1659
    },
    {
      "epoch": 1.5821456909179688e-06,
      "step": 1659,
      "training_step_time": 6.9112420082092285
    },
    {
      "epoch": 1.583099365234375e-06,
      "grad_norm": 1.26596999168396,
      "learning_rate": 3.3180000000000003e-06,
      "loss": 0.1156,
      "step": 1660
    },
    {
      "epoch": 1.583099365234375e-06,
      "model_forward_time": 2.270735025405884,
      "step": 1660
    },
    {
      "dynamics_loss_avg": 0.0936010241508484,
      "epoch": 1.583099365234375e-06,
      "step": 1660
    },
    {
      "action_loss_avg": 0.03647156907245517,
      "epoch": 1.583099365234375e-06,
      "step": 1660
    },
    {
      "epoch": 1.583099365234375e-06,
      "step": 1660,
      "training_step_time": 7.571079969406128
    },
    {
      "epoch": 1.5840530395507813e-06,
      "model_forward_time": 1.7172398567199707,
      "step": 1661
    },
    {
      "epoch": 1.5840530395507813e-06,
      "step": 1661,
      "training_step_time": 7.941053152084351
    },
    {
      "epoch": 1.5850067138671874e-06,
      "model_forward_time": 1.717606782913208,
      "step": 1662
    },
    {
      "epoch": 1.5850067138671874e-06,
      "step": 1662,
      "training_step_time": 7.878732919692993
    },
    {
      "epoch": 1.5859603881835938e-06,
      "model_forward_time": 1.72043776512146,
      "step": 1663
    },
    {
      "epoch": 1.5859603881835938e-06,
      "step": 1663,
      "training_step_time": 6.912540912628174
    },
    {
      "epoch": 1.5869140625e-06,
      "model_forward_time": 1.7181081771850586,
      "step": 1664
    },
    {
      "epoch": 1.5869140625e-06,
      "step": 1664,
      "training_step_time": 6.8682684898376465
    },
    {
      "epoch": 1.5878677368164063e-06,
      "model_forward_time": 1.71744966506958,
      "step": 1665
    },
    {
      "epoch": 1.5878677368164063e-06,
      "step": 1665,
      "training_step_time": 6.917657136917114
    },
    {
      "epoch": 1.5888214111328124e-06,
      "model_forward_time": 1.7191171646118164,
      "step": 1666
    },
    {
      "epoch": 1.5888214111328124e-06,
      "step": 1666,
      "training_step_time": 6.8856401443481445
    },
    {
      "epoch": 1.5897750854492188e-06,
      "model_forward_time": 1.721101999282837,
      "step": 1667
    },
    {
      "epoch": 1.5897750854492188e-06,
      "step": 1667,
      "training_step_time": 6.897261142730713
    },
    {
      "epoch": 1.590728759765625e-06,
      "model_forward_time": 1.7293329238891602,
      "step": 1668
    },
    {
      "epoch": 1.590728759765625e-06,
      "step": 1668,
      "training_step_time": 6.868791341781616
    },
    {
      "epoch": 1.5916824340820313e-06,
      "model_forward_time": 2.2557079792022705,
      "step": 1669
    },
    {
      "epoch": 1.5916824340820313e-06,
      "step": 1669,
      "training_step_time": 7.265801429748535
    },
    {
      "epoch": 1.5926361083984374e-06,
      "grad_norm": 1.6653127670288086,
      "learning_rate": 3.338e-06,
      "loss": 0.1006,
      "step": 1670
    },
    {
      "epoch": 1.5926361083984374e-06,
      "model_forward_time": 1.6987190246582031,
      "step": 1670
    },
    {
      "dynamics_loss_avg": 0.07815249934792519,
      "epoch": 1.5926361083984374e-06,
      "step": 1670
    },
    {
      "action_loss_avg": 0.03482058551162481,
      "epoch": 1.5926361083984374e-06,
      "step": 1670
    },
    {
      "epoch": 1.5926361083984374e-06,
      "step": 1670,
      "training_step_time": 8.61813497543335
    },
    {
      "epoch": 1.5935897827148438e-06,
      "model_forward_time": 1.7225596904754639,
      "step": 1671
    },
    {
      "epoch": 1.5935897827148438e-06,
      "step": 1671,
      "training_step_time": 8.596207857131958
    },
    {
      "epoch": 1.59454345703125e-06,
      "model_forward_time": 1.701871395111084,
      "step": 1672
    },
    {
      "epoch": 1.59454345703125e-06,
      "step": 1672,
      "training_step_time": 7.652401924133301
    },
    {
      "epoch": 1.5954971313476563e-06,
      "model_forward_time": 1.7185492515563965,
      "step": 1673
    },
    {
      "epoch": 1.5954971313476563e-06,
      "step": 1673,
      "training_step_time": 6.898043870925903
    },
    {
      "epoch": 1.5964508056640624e-06,
      "model_forward_time": 1.7220311164855957,
      "step": 1674
    },
    {
      "epoch": 1.5964508056640624e-06,
      "step": 1674,
      "training_step_time": 6.879989147186279
    },
    {
      "epoch": 1.5974044799804688e-06,
      "model_forward_time": 1.7145705223083496,
      "step": 1675
    },
    {
      "epoch": 1.5974044799804688e-06,
      "step": 1675,
      "training_step_time": 6.8987627029418945
    },
    {
      "epoch": 1.598358154296875e-06,
      "model_forward_time": 1.7145607471466064,
      "step": 1676
    },
    {
      "epoch": 1.598358154296875e-06,
      "step": 1676,
      "training_step_time": 6.897907257080078
    },
    {
      "epoch": 1.5993118286132813e-06,
      "model_forward_time": 1.7142653465270996,
      "step": 1677
    },
    {
      "epoch": 1.5993118286132813e-06,
      "step": 1677,
      "training_step_time": 6.871697425842285
    },
    {
      "epoch": 1.6002655029296874e-06,
      "model_forward_time": 1.746213436126709,
      "step": 1678
    },
    {
      "epoch": 1.6002655029296874e-06,
      "step": 1678,
      "training_step_time": 7.213402271270752
    },
    {
      "epoch": 1.6012191772460938e-06,
      "model_forward_time": 2.238543748855591,
      "step": 1679
    },
    {
      "epoch": 1.6012191772460938e-06,
      "step": 1679,
      "training_step_time": 8.951204776763916
    },
    {
      "epoch": 1.6021728515625e-06,
      "grad_norm": 1.541358232498169,
      "learning_rate": 3.3580000000000003e-06,
      "loss": 0.1027,
      "step": 1680
    },
    {
      "epoch": 1.6021728515625e-06,
      "model_forward_time": 1.7075245380401611,
      "step": 1680
    },
    {
      "dynamics_loss_avg": 0.056646345928311345,
      "epoch": 1.6021728515625e-06,
      "step": 1680
    },
    {
      "action_loss_avg": 0.02885879431851208,
      "epoch": 1.6021728515625e-06,
      "step": 1680
    },
    {
      "epoch": 1.6021728515625e-06,
      "step": 1680,
      "training_step_time": 7.926802396774292
    },
    {
      "epoch": 1.6031265258789063e-06,
      "model_forward_time": 1.7218248844146729,
      "step": 1681
    },
    {
      "epoch": 1.6031265258789063e-06,
      "step": 1681,
      "training_step_time": 7.663618564605713
    },
    {
      "epoch": 1.6040802001953124e-06,
      "model_forward_time": 1.7110555171966553,
      "step": 1682
    },
    {
      "epoch": 1.6040802001953124e-06,
      "step": 1682,
      "training_step_time": 7.250133275985718
    },
    {
      "epoch": 1.6050338745117188e-06,
      "model_forward_time": 1.7057723999023438,
      "step": 1683
    },
    {
      "epoch": 1.6050338745117188e-06,
      "step": 1683,
      "training_step_time": 6.888403415679932
    },
    {
      "epoch": 1.605987548828125e-06,
      "model_forward_time": 1.7084221839904785,
      "step": 1684
    },
    {
      "epoch": 1.605987548828125e-06,
      "step": 1684,
      "training_step_time": 6.86582350730896
    },
    {
      "epoch": 1.6069412231445313e-06,
      "model_forward_time": 1.7162292003631592,
      "step": 1685
    },
    {
      "epoch": 1.6069412231445313e-06,
      "step": 1685,
      "training_step_time": 6.893885612487793
    },
    {
      "epoch": 1.6078948974609374e-06,
      "model_forward_time": 1.708205223083496,
      "step": 1686
    },
    {
      "epoch": 1.6078948974609374e-06,
      "step": 1686,
      "training_step_time": 6.820958375930786
    },
    {
      "epoch": 1.6088485717773438e-06,
      "model_forward_time": 1.7436425685882568,
      "step": 1687
    },
    {
      "epoch": 1.6088485717773438e-06,
      "step": 1687,
      "training_step_time": 7.223682403564453
    },
    {
      "epoch": 1.60980224609375e-06,
      "model_forward_time": 2.2199831008911133,
      "step": 1688
    },
    {
      "epoch": 1.60980224609375e-06,
      "step": 1688,
      "training_step_time": 8.887652397155762
    },
    {
      "epoch": 1.6107559204101563e-06,
      "model_forward_time": 1.7088253498077393,
      "step": 1689
    },
    {
      "epoch": 1.6107559204101563e-06,
      "step": 1689,
      "training_step_time": 9.652629613876343
    },
    {
      "epoch": 1.6117095947265624e-06,
      "grad_norm": 1.1339448690414429,
      "learning_rate": 3.378e-06,
      "loss": 0.1053,
      "step": 1690
    },
    {
      "epoch": 1.6117095947265624e-06,
      "model_forward_time": 1.7074015140533447,
      "step": 1690
    },
    {
      "dynamics_loss_avg": 0.08150692638009786,
      "epoch": 1.6117095947265624e-06,
      "step": 1690
    },
    {
      "action_loss_avg": 0.02945122104138136,
      "epoch": 1.6117095947265624e-06,
      "step": 1690
    },
    {
      "epoch": 1.6117095947265624e-06,
      "step": 1690,
      "training_step_time": 7.713178396224976
    },
    {
      "epoch": 1.6126632690429688e-06,
      "model_forward_time": 1.7107527256011963,
      "step": 1691
    },
    {
      "epoch": 1.6126632690429688e-06,
      "step": 1691,
      "training_step_time": 7.224529027938843
    },
    {
      "epoch": 1.613616943359375e-06,
      "model_forward_time": 1.705042839050293,
      "step": 1692
    },
    {
      "epoch": 1.613616943359375e-06,
      "step": 1692,
      "training_step_time": 6.8641698360443115
    },
    {
      "epoch": 1.6145706176757813e-06,
      "model_forward_time": 1.7087409496307373,
      "step": 1693
    },
    {
      "epoch": 1.6145706176757813e-06,
      "step": 1693,
      "training_step_time": 6.865780830383301
    },
    {
      "epoch": 1.6155242919921875e-06,
      "model_forward_time": 1.715324878692627,
      "step": 1694
    },
    {
      "epoch": 1.6155242919921875e-06,
      "step": 1694,
      "training_step_time": 6.884241819381714
    },
    {
      "epoch": 1.6164779663085938e-06,
      "model_forward_time": 1.7093830108642578,
      "step": 1695
    },
    {
      "epoch": 1.6164779663085938e-06,
      "step": 1695,
      "training_step_time": 6.853266000747681
    },
    {
      "epoch": 1.617431640625e-06,
      "model_forward_time": 1.7402985095977783,
      "step": 1696
    },
    {
      "epoch": 1.617431640625e-06,
      "step": 1696,
      "training_step_time": 7.216189384460449
    },
    {
      "epoch": 1.6183853149414063e-06,
      "model_forward_time": 2.2250754833221436,
      "step": 1697
    },
    {
      "epoch": 1.6183853149414063e-06,
      "step": 1697,
      "training_step_time": 7.871677398681641
    },
    {
      "epoch": 1.6193389892578125e-06,
      "model_forward_time": 0.9941854476928711,
      "step": 1698
    },
    {
      "epoch": 1.6193389892578125e-06,
      "step": 1698,
      "training_step_time": 9.726319074630737
    },
    {
      "epoch": 1.6202926635742188e-06,
      "model_forward_time": 1.7055730819702148,
      "step": 1699
    },
    {
      "epoch": 1.6202926635742188e-06,
      "step": 1699,
      "training_step_time": 7.76373028755188
    },
    {
      "epoch": 1.621246337890625e-06,
      "grad_norm": 1.1933029890060425,
      "learning_rate": 3.3980000000000003e-06,
      "loss": 0.1048,
      "step": 1700
    },
    {
      "epoch": 1.621246337890625e-06,
      "model_forward_time": 1.6990232467651367,
      "step": 1700
    },
    {
      "dynamics_loss_avg": 0.0893986277282238,
      "epoch": 1.621246337890625e-06,
      "step": 1700
    },
    {
      "action_loss_avg": 0.03312426321208477,
      "epoch": 1.621246337890625e-06,
      "step": 1700
    },
    {
      "epoch": 1.621246337890625e-06,
      "step": 1700,
      "training_step_time": 7.224796772003174
    },
    {
      "epoch": 1.6222000122070313e-06,
      "model_forward_time": 1.7123253345489502,
      "step": 1701
    },
    {
      "epoch": 1.6222000122070313e-06,
      "step": 1701,
      "training_step_time": 6.883967876434326
    },
    {
      "epoch": 1.6231536865234375e-06,
      "model_forward_time": 1.7072272300720215,
      "step": 1702
    },
    {
      "epoch": 1.6231536865234375e-06,
      "step": 1702,
      "training_step_time": 6.901011943817139
    },
    {
      "epoch": 1.6241073608398438e-06,
      "model_forward_time": 1.7079029083251953,
      "step": 1703
    },
    {
      "epoch": 1.6241073608398438e-06,
      "step": 1703,
      "training_step_time": 6.872542142868042
    },
    {
      "epoch": 1.62506103515625e-06,
      "model_forward_time": 1.7073321342468262,
      "step": 1704
    },
    {
      "epoch": 1.62506103515625e-06,
      "step": 1704,
      "training_step_time": 6.886523723602295
    },
    {
      "epoch": 1.6260147094726563e-06,
      "model_forward_time": 1.7459163665771484,
      "step": 1705
    },
    {
      "epoch": 1.6260147094726563e-06,
      "step": 1705,
      "training_step_time": 7.2338621616363525
    },
    {
      "epoch": 1.6269683837890625e-06,
      "model_forward_time": 2.2161004543304443,
      "step": 1706
    },
    {
      "epoch": 1.6269683837890625e-06,
      "step": 1706,
      "training_step_time": 8.515265226364136
    },
    {
      "epoch": 1.6279220581054688e-06,
      "model_forward_time": 1.7076694965362549,
      "step": 1707
    },
    {
      "epoch": 1.6279220581054688e-06,
      "step": 1707,
      "training_step_time": 10.476715564727783
    },
    {
      "epoch": 1.628875732421875e-06,
      "model_forward_time": 1.70973539352417,
      "step": 1708
    },
    {
      "epoch": 1.628875732421875e-06,
      "step": 1708,
      "training_step_time": 7.83812952041626
    },
    {
      "epoch": 1.6298294067382813e-06,
      "model_forward_time": 1.7097790241241455,
      "step": 1709
    },
    {
      "epoch": 1.6298294067382813e-06,
      "step": 1709,
      "training_step_time": 7.473866939544678
    },
    {
      "epoch": 1.6307830810546875e-06,
      "grad_norm": 1.6072914600372314,
      "learning_rate": 3.418e-06,
      "loss": 0.099,
      "step": 1710
    },
    {
      "epoch": 1.6307830810546875e-06,
      "model_forward_time": 1.7015018463134766,
      "step": 1710
    },
    {
      "dynamics_loss_avg": 0.06624244395643472,
      "epoch": 1.6307830810546875e-06,
      "step": 1710
    },
    {
      "action_loss_avg": 0.03381995460949838,
      "epoch": 1.6307830810546875e-06,
      "step": 1710
    },
    {
      "epoch": 1.6307830810546875e-06,
      "step": 1710,
      "training_step_time": 6.8572678565979
    },
    {
      "epoch": 1.6317367553710938e-06,
      "model_forward_time": 1.7128026485443115,
      "step": 1711
    },
    {
      "epoch": 1.6317367553710938e-06,
      "step": 1711,
      "training_step_time": 6.89589262008667
    },
    {
      "epoch": 1.6326904296875e-06,
      "model_forward_time": 0.9890055656433105,
      "step": 1712
    },
    {
      "epoch": 1.6326904296875e-06,
      "step": 1712,
      "training_step_time": 6.887006759643555
    },
    {
      "epoch": 1.6336441040039063e-06,
      "model_forward_time": 1.7009401321411133,
      "step": 1713
    },
    {
      "epoch": 1.6336441040039063e-06,
      "step": 1713,
      "training_step_time": 6.874167442321777
    },
    {
      "epoch": 1.6345977783203125e-06,
      "model_forward_time": 1.730146884918213,
      "step": 1714
    },
    {
      "epoch": 1.6345977783203125e-06,
      "step": 1714,
      "training_step_time": 7.224518775939941
    },
    {
      "epoch": 1.6355514526367188e-06,
      "model_forward_time": 1.7146122455596924,
      "step": 1715
    },
    {
      "epoch": 1.6355514526367188e-06,
      "step": 1715,
      "training_step_time": 7.894715785980225
    },
    {
      "epoch": 1.636505126953125e-06,
      "model_forward_time": 2.233220338821411,
      "step": 1716
    },
    {
      "epoch": 1.636505126953125e-06,
      "step": 1716,
      "training_step_time": 9.50688910484314
    },
    {
      "epoch": 1.6374588012695313e-06,
      "model_forward_time": 1.7117202281951904,
      "step": 1717
    },
    {
      "epoch": 1.6374588012695313e-06,
      "step": 1717,
      "training_step_time": 7.913179874420166
    },
    {
      "epoch": 1.6384124755859375e-06,
      "model_forward_time": 1.7093555927276611,
      "step": 1718
    },
    {
      "epoch": 1.6384124755859375e-06,
      "step": 1718,
      "training_step_time": 7.712420463562012
    },
    {
      "epoch": 1.6393661499023438e-06,
      "model_forward_time": 1.7075109481811523,
      "step": 1719
    },
    {
      "epoch": 1.6393661499023438e-06,
      "step": 1719,
      "training_step_time": 7.219266176223755
    },
    {
      "epoch": 1.64031982421875e-06,
      "grad_norm": 1.0731602907180786,
      "learning_rate": 3.4380000000000004e-06,
      "loss": 0.1007,
      "step": 1720
    },
    {
      "epoch": 1.64031982421875e-06,
      "model_forward_time": 1.7061645984649658,
      "step": 1720
    },
    {
      "dynamics_loss_avg": 0.07560733929276467,
      "epoch": 1.64031982421875e-06,
      "step": 1720
    },
    {
      "action_loss_avg": 0.03344972152262926,
      "epoch": 1.64031982421875e-06,
      "step": 1720
    },
    {
      "epoch": 1.64031982421875e-06,
      "step": 1720,
      "training_step_time": 6.874334096908569
    },
    {
      "epoch": 1.6412734985351563e-06,
      "model_forward_time": 1.712376594543457,
      "step": 1721
    },
    {
      "epoch": 1.6412734985351563e-06,
      "step": 1721,
      "training_step_time": 6.869632959365845
    },
    {
      "epoch": 1.6422271728515625e-06,
      "model_forward_time": 1.7022614479064941,
      "step": 1722
    },
    {
      "epoch": 1.6422271728515625e-06,
      "step": 1722,
      "training_step_time": 6.8797760009765625
    },
    {
      "epoch": 1.6431808471679688e-06,
      "model_forward_time": 1.7050764560699463,
      "step": 1723
    },
    {
      "epoch": 1.6431808471679688e-06,
      "step": 1723,
      "training_step_time": 8.165769338607788
    },
    {
      "epoch": 1.644134521484375e-06,
      "model_forward_time": 1.7129101753234863,
      "step": 1724
    },
    {
      "epoch": 1.644134521484375e-06,
      "step": 1724,
      "training_step_time": 8.689977884292603
    },
    {
      "epoch": 1.6450881958007813e-06,
      "model_forward_time": 2.224886417388916,
      "step": 1725
    },
    {
      "epoch": 1.6450881958007813e-06,
      "step": 1725,
      "training_step_time": 8.99118185043335
    },
    {
      "epoch": 1.6460418701171875e-06,
      "model_forward_time": 1.7136082649230957,
      "step": 1726
    },
    {
      "epoch": 1.6460418701171875e-06,
      "step": 1726,
      "training_step_time": 9.573232650756836
    },
    {
      "epoch": 1.6469955444335938e-06,
      "model_forward_time": 1.7102365493774414,
      "step": 1727
    },
    {
      "epoch": 1.6469955444335938e-06,
      "step": 1727,
      "training_step_time": 7.720898389816284
    },
    {
      "epoch": 1.64794921875e-06,
      "model_forward_time": 1.7146704196929932,
      "step": 1728
    },
    {
      "epoch": 1.64794921875e-06,
      "step": 1728,
      "training_step_time": 7.222667694091797
    },
    {
      "epoch": 1.6489028930664063e-06,
      "model_forward_time": 1.708693265914917,
      "step": 1729
    },
    {
      "epoch": 1.6489028930664063e-06,
      "step": 1729,
      "training_step_time": 6.877777576446533
    },
    {
      "epoch": 1.6498565673828125e-06,
      "grad_norm": 1.279050350189209,
      "learning_rate": 3.458e-06,
      "loss": 0.1066,
      "step": 1730
    },
    {
      "epoch": 1.6498565673828125e-06,
      "model_forward_time": 1.699920892715454,
      "step": 1730
    },
    {
      "dynamics_loss_avg": 0.08063988462090492,
      "epoch": 1.6498565673828125e-06,
      "step": 1730
    },
    {
      "action_loss_avg": 0.030720270238816738,
      "epoch": 1.6498565673828125e-06,
      "step": 1730
    },
    {
      "epoch": 1.6498565673828125e-06,
      "step": 1730,
      "training_step_time": 6.86897349357605
    },
    {
      "epoch": 1.6508102416992188e-06,
      "model_forward_time": 1.7036166191101074,
      "step": 1731
    },
    {
      "epoch": 1.6508102416992188e-06,
      "step": 1731,
      "training_step_time": 6.88738489151001
    },
    {
      "epoch": 1.651763916015625e-06,
      "model_forward_time": 1.7030205726623535,
      "step": 1732
    },
    {
      "epoch": 1.651763916015625e-06,
      "step": 1732,
      "training_step_time": 6.882816553115845
    },
    {
      "epoch": 1.6527175903320313e-06,
      "model_forward_time": 1.7495441436767578,
      "step": 1733
    },
    {
      "epoch": 1.6527175903320313e-06,
      "step": 1733,
      "training_step_time": 7.878002882003784
    },
    {
      "epoch": 1.6536712646484375e-06,
      "model_forward_time": 2.2235829830169678,
      "step": 1734
    },
    {
      "epoch": 1.6536712646484375e-06,
      "step": 1734,
      "training_step_time": 8.983902931213379
    },
    {
      "epoch": 1.6546249389648438e-06,
      "model_forward_time": 1.709808349609375,
      "step": 1735
    },
    {
      "epoch": 1.6546249389648438e-06,
      "step": 1735,
      "training_step_time": 8.74153995513916
    },
    {
      "epoch": 1.65557861328125e-06,
      "model_forward_time": 1.703702688217163,
      "step": 1736
    },
    {
      "epoch": 1.65557861328125e-06,
      "step": 1736,
      "training_step_time": 7.7994749546051025
    },
    {
      "epoch": 1.6565322875976563e-06,
      "model_forward_time": 1.7103981971740723,
      "step": 1737
    },
    {
      "epoch": 1.6565322875976563e-06,
      "step": 1737,
      "training_step_time": 7.244235277175903
    },
    {
      "epoch": 1.6574859619140625e-06,
      "model_forward_time": 1.7180695533752441,
      "step": 1738
    },
    {
      "epoch": 1.6574859619140625e-06,
      "step": 1738,
      "training_step_time": 6.9413604736328125
    },
    {
      "epoch": 1.6584396362304688e-06,
      "model_forward_time": 1.7155723571777344,
      "step": 1739
    },
    {
      "epoch": 1.6584396362304688e-06,
      "step": 1739,
      "training_step_time": 6.888542890548706
    },
    {
      "epoch": 1.659393310546875e-06,
      "grad_norm": 1.7270054817199707,
      "learning_rate": 3.4780000000000004e-06,
      "loss": 0.0973,
      "step": 1740
    },
    {
      "epoch": 1.659393310546875e-06,
      "model_forward_time": 1.7006206512451172,
      "step": 1740
    },
    {
      "dynamics_loss_avg": 0.0639527639374137,
      "epoch": 1.659393310546875e-06,
      "step": 1740
    },
    {
      "action_loss_avg": 0.031172178871929645,
      "epoch": 1.659393310546875e-06,
      "step": 1740
    },
    {
      "epoch": 1.659393310546875e-06,
      "step": 1740,
      "training_step_time": 6.8935792446136475
    },
    {
      "epoch": 1.6603469848632813e-06,
      "model_forward_time": 1.7093727588653564,
      "step": 1741
    },
    {
      "epoch": 1.6603469848632813e-06,
      "step": 1741,
      "training_step_time": 6.906351804733276
    },
    {
      "epoch": 1.6613006591796875e-06,
      "model_forward_time": 1.753890037536621,
      "step": 1742
    },
    {
      "epoch": 1.6613006591796875e-06,
      "step": 1742,
      "training_step_time": 7.91204571723938
    },
    {
      "epoch": 1.6622543334960938e-06,
      "model_forward_time": 1.718446969985962,
      "step": 1743
    },
    {
      "epoch": 1.6622543334960938e-06,
      "step": 1743,
      "training_step_time": 8.667402744293213
    },
    {
      "epoch": 1.6632080078125e-06,
      "model_forward_time": 2.2293806076049805,
      "step": 1744
    },
    {
      "epoch": 1.6632080078125e-06,
      "step": 1744,
      "training_step_time": 9.100719928741455
    },
    {
      "epoch": 1.6641616821289064e-06,
      "model_forward_time": 1.7055690288543701,
      "step": 1745
    },
    {
      "epoch": 1.6641616821289064e-06,
      "step": 1745,
      "training_step_time": 9.278393030166626
    },
    {
      "epoch": 1.6651153564453125e-06,
      "model_forward_time": 1.7102046012878418,
      "step": 1746
    },
    {
      "epoch": 1.6651153564453125e-06,
      "step": 1746,
      "training_step_time": 6.892803430557251
    },
    {
      "epoch": 1.6660690307617189e-06,
      "model_forward_time": 1.7066993713378906,
      "step": 1747
    },
    {
      "epoch": 1.6660690307617189e-06,
      "step": 1747,
      "training_step_time": 7.3043131828308105
    },
    {
      "epoch": 1.667022705078125e-06,
      "model_forward_time": 1.7076785564422607,
      "step": 1748
    },
    {
      "epoch": 1.667022705078125e-06,
      "step": 1748,
      "training_step_time": 6.925687074661255
    },
    {
      "epoch": 1.6679763793945314e-06,
      "model_forward_time": 1.6998746395111084,
      "step": 1749
    },
    {
      "epoch": 1.6679763793945314e-06,
      "step": 1749,
      "training_step_time": 6.93553352355957
    },
    {
      "epoch": 1.6689300537109375e-06,
      "grad_norm": 2.320312023162842,
      "learning_rate": 3.4980000000000002e-06,
      "loss": 0.1114,
      "step": 1750
    },
    {
      "epoch": 1.6689300537109375e-06,
      "model_forward_time": 1.700573205947876,
      "step": 1750
    },
    {
      "dynamics_loss_avg": 0.06324283797293902,
      "epoch": 1.6689300537109375e-06,
      "step": 1750
    },
    {
      "action_loss_avg": 0.03475711247883737,
      "epoch": 1.6689300537109375e-06,
      "step": 1750
    },
    {
      "epoch": 1.6689300537109375e-06,
      "step": 1750,
      "training_step_time": 6.8670947551727295
    },
    {
      "epoch": 1.6698837280273436e-06,
      "model_forward_time": 1.755568265914917,
      "step": 1751
    },
    {
      "epoch": 1.6698837280273436e-06,
      "step": 1751,
      "training_step_time": 8.438457250595093
    },
    {
      "epoch": 1.67083740234375e-06,
      "model_forward_time": 1.6976096630096436,
      "step": 1752
    },
    {
      "epoch": 1.67083740234375e-06,
      "step": 1752,
      "training_step_time": 9.753973007202148
    },
    {
      "epoch": 1.6717910766601561e-06,
      "model_forward_time": 2.237621545791626,
      "step": 1753
    },
    {
      "epoch": 1.6717910766601561e-06,
      "step": 1753,
      "training_step_time": 9.713894605636597
    },
    {
      "epoch": 1.6727447509765625e-06,
      "model_forward_time": 1.7104310989379883,
      "step": 1754
    },
    {
      "epoch": 1.6727447509765625e-06,
      "step": 1754,
      "training_step_time": 7.927874565124512
    },
    {
      "epoch": 1.6736984252929686e-06,
      "model_forward_time": 1.706521987915039,
      "step": 1755
    },
    {
      "epoch": 1.6736984252929686e-06,
      "step": 1755,
      "training_step_time": 7.56848669052124
    },
    {
      "epoch": 1.674652099609375e-06,
      "model_forward_time": 1.7124605178833008,
      "step": 1756
    },
    {
      "epoch": 1.674652099609375e-06,
      "step": 1756,
      "training_step_time": 6.931652545928955
    },
    {
      "epoch": 1.6756057739257812e-06,
      "model_forward_time": 1.7029411792755127,
      "step": 1757
    },
    {
      "epoch": 1.6756057739257812e-06,
      "step": 1757,
      "training_step_time": 6.903754711151123
    },
    {
      "epoch": 1.6765594482421875e-06,
      "model_forward_time": 1.700103998184204,
      "step": 1758
    },
    {
      "epoch": 1.6765594482421875e-06,
      "step": 1758,
      "training_step_time": 6.958166599273682
    },
    {
      "epoch": 1.6775131225585937e-06,
      "model_forward_time": 1.739591121673584,
      "step": 1759
    },
    {
      "epoch": 1.6775131225585937e-06,
      "step": 1759,
      "training_step_time": 6.958980560302734
    },
    {
      "epoch": 1.678466796875e-06,
      "grad_norm": 1.6609339714050293,
      "learning_rate": 3.5180000000000005e-06,
      "loss": 0.109,
      "step": 1760
    },
    {
      "epoch": 1.678466796875e-06,
      "model_forward_time": 1.727524757385254,
      "step": 1760
    },
    {
      "dynamics_loss_avg": 0.06852809563279152,
      "epoch": 1.678466796875e-06,
      "step": 1760
    },
    {
      "action_loss_avg": 0.029779915511608124,
      "epoch": 1.678466796875e-06,
      "step": 1760
    },
    {
      "epoch": 1.678466796875e-06,
      "step": 1760,
      "training_step_time": 8.089641094207764
    },
    {
      "epoch": 1.6794204711914062e-06,
      "model_forward_time": 1.7141737937927246,
      "step": 1761
    },
    {
      "epoch": 1.6794204711914062e-06,
      "step": 1761,
      "training_step_time": 9.494258165359497
    },
    {
      "epoch": 1.6803741455078125e-06,
      "model_forward_time": 2.254277229309082,
      "step": 1762
    },
    {
      "epoch": 1.6803741455078125e-06,
      "step": 1762,
      "training_step_time": 10.456714391708374
    },
    {
      "epoch": 1.6813278198242187e-06,
      "model_forward_time": 1.7213387489318848,
      "step": 1763
    },
    {
      "epoch": 1.6813278198242187e-06,
      "step": 1763,
      "training_step_time": 10.207223653793335
    },
    {
      "epoch": 1.682281494140625e-06,
      "model_forward_time": 1.7184233665466309,
      "step": 1764
    },
    {
      "epoch": 1.682281494140625e-06,
      "step": 1764,
      "training_step_time": 7.826127052307129
    },
    {
      "epoch": 1.6832351684570312e-06,
      "model_forward_time": 1.720463514328003,
      "step": 1765
    },
    {
      "epoch": 1.6832351684570312e-06,
      "step": 1765,
      "training_step_time": 7.3254804611206055
    },
    {
      "epoch": 1.6841888427734375e-06,
      "model_forward_time": 1.824129581451416,
      "step": 1766
    },
    {
      "epoch": 1.6841888427734375e-06,
      "step": 1766,
      "training_step_time": 6.9585630893707275
    },
    {
      "epoch": 1.6851425170898437e-06,
      "model_forward_time": 1.8254292011260986,
      "step": 1767
    },
    {
      "epoch": 1.6851425170898437e-06,
      "step": 1767,
      "training_step_time": 7.041790246963501
    },
    {
      "epoch": 1.68609619140625e-06,
      "model_forward_time": 1.8156843185424805,
      "step": 1768
    },
    {
      "epoch": 1.68609619140625e-06,
      "step": 1768,
      "training_step_time": 7.003015518188477
    },
    {
      "epoch": 1.6870498657226562e-06,
      "model_forward_time": 1.8294763565063477,
      "step": 1769
    },
    {
      "epoch": 1.6870498657226562e-06,
      "step": 1769,
      "training_step_time": 9.04147458076477
    },
    {
      "epoch": 1.6880035400390625e-06,
      "grad_norm": 1.0654371976852417,
      "learning_rate": 3.5380000000000003e-06,
      "loss": 0.1082,
      "step": 1770
    },
    {
      "epoch": 1.6880035400390625e-06,
      "model_forward_time": 1.8153390884399414,
      "step": 1770
    },
    {
      "dynamics_loss_avg": 0.08542204014956951,
      "epoch": 1.6880035400390625e-06,
      "step": 1770
    },
    {
      "action_loss_avg": 0.05185302086174488,
      "epoch": 1.6880035400390625e-06,
      "step": 1770
    },
    {
      "epoch": 1.6880035400390625e-06,
      "step": 1770,
      "training_step_time": 8.674459218978882
    },
    {
      "epoch": 1.6889572143554687e-06,
      "model_forward_time": 2.3554728031158447,
      "step": 1771
    },
    {
      "epoch": 1.6889572143554687e-06,
      "step": 1771,
      "training_step_time": 8.88321042060852
    },
    {
      "epoch": 1.689910888671875e-06,
      "model_forward_time": 1.7513072490692139,
      "step": 1772
    },
    {
      "epoch": 1.689910888671875e-06,
      "step": 1772,
      "training_step_time": 9.64108657836914
    },
    {
      "epoch": 1.6908645629882812e-06,
      "model_forward_time": 1.7210941314697266,
      "step": 1773
    },
    {
      "epoch": 1.6908645629882812e-06,
      "step": 1773,
      "training_step_time": 7.758037567138672
    },
    {
      "epoch": 1.6918182373046875e-06,
      "model_forward_time": 1.348109245300293,
      "step": 1774
    },
    {
      "epoch": 1.6918182373046875e-06,
      "step": 1774,
      "training_step_time": 7.52554988861084
    },
    {
      "epoch": 1.6927719116210937e-06,
      "model_forward_time": 1.7273340225219727,
      "step": 1775
    },
    {
      "epoch": 1.6927719116210937e-06,
      "step": 1775,
      "training_step_time": 6.917619466781616
    },
    {
      "epoch": 1.6937255859375e-06,
      "model_forward_time": 1.7254841327667236,
      "step": 1776
    },
    {
      "epoch": 1.6937255859375e-06,
      "step": 1776,
      "training_step_time": 6.9262824058532715
    },
    {
      "epoch": 1.6946792602539062e-06,
      "model_forward_time": 1.7214038372039795,
      "step": 1777
    },
    {
      "epoch": 1.6946792602539062e-06,
      "step": 1777,
      "training_step_time": 7.046168088912964
    },
    {
      "epoch": 1.6956329345703125e-06,
      "model_forward_time": 1.7147748470306396,
      "step": 1778
    },
    {
      "epoch": 1.6956329345703125e-06,
      "step": 1778,
      "training_step_time": 9.107438802719116
    },
    {
      "epoch": 1.6965866088867187e-06,
      "model_forward_time": 1.7399814128875732,
      "step": 1779
    },
    {
      "epoch": 1.6965866088867187e-06,
      "step": 1779,
      "training_step_time": 8.74755859375
    },
    {
      "epoch": 1.697540283203125e-06,
      "grad_norm": 1.1308664083480835,
      "learning_rate": 3.5580000000000005e-06,
      "loss": 0.0997,
      "step": 1780
    },
    {
      "epoch": 1.697540283203125e-06,
      "model_forward_time": 2.218221664428711,
      "step": 1780
    },
    {
      "dynamics_loss_avg": 0.055114491283893584,
      "epoch": 1.697540283203125e-06,
      "step": 1780
    },
    {
      "action_loss_avg": 0.03354180483147502,
      "epoch": 1.697540283203125e-06,
      "step": 1780
    },
    {
      "epoch": 1.697540283203125e-06,
      "step": 1780,
      "training_step_time": 10.537163972854614
    },
    {
      "epoch": 1.6984939575195312e-06,
      "model_forward_time": 1.7136361598968506,
      "step": 1781
    },
    {
      "epoch": 1.6984939575195312e-06,
      "step": 1781,
      "training_step_time": 9.554723739624023
    },
    {
      "epoch": 1.6994476318359375e-06,
      "model_forward_time": 1.728602647781372,
      "step": 1782
    },
    {
      "epoch": 1.6994476318359375e-06,
      "step": 1782,
      "training_step_time": 8.631261587142944
    },
    {
      "epoch": 1.7004013061523437e-06,
      "model_forward_time": 1.7183258533477783,
      "step": 1783
    },
    {
      "epoch": 1.7004013061523437e-06,
      "step": 1783,
      "training_step_time": 7.542189836502075
    },
    {
      "epoch": 1.70135498046875e-06,
      "model_forward_time": 1.7298367023468018,
      "step": 1784
    },
    {
      "epoch": 1.70135498046875e-06,
      "step": 1784,
      "training_step_time": 7.25623631477356
    },
    {
      "epoch": 1.7023086547851562e-06,
      "model_forward_time": 1.7222309112548828,
      "step": 1785
    },
    {
      "epoch": 1.7023086547851562e-06,
      "step": 1785,
      "training_step_time": 6.947351694107056
    },
    {
      "epoch": 1.7032623291015625e-06,
      "model_forward_time": 1.7217357158660889,
      "step": 1786
    },
    {
      "epoch": 1.7032623291015625e-06,
      "step": 1786,
      "training_step_time": 6.920607566833496
    },
    {
      "epoch": 1.7042160034179687e-06,
      "model_forward_time": 1.720937967300415,
      "step": 1787
    },
    {
      "epoch": 1.7042160034179687e-06,
      "step": 1787,
      "training_step_time": 7.990607023239136
    },
    {
      "epoch": 1.705169677734375e-06,
      "model_forward_time": 1.752619981765747,
      "step": 1788
    },
    {
      "epoch": 1.705169677734375e-06,
      "step": 1788,
      "training_step_time": 8.559877157211304
    },
    {
      "epoch": 1.7061233520507812e-06,
      "model_forward_time": 1.7198398113250732,
      "step": 1789
    },
    {
      "epoch": 1.7061233520507812e-06,
      "step": 1789,
      "training_step_time": 8.940722703933716
    },
    {
      "epoch": 1.7070770263671875e-06,
      "grad_norm": 1.734910011291504,
      "learning_rate": 3.5780000000000003e-06,
      "loss": 0.1025,
      "step": 1790
    },
    {
      "epoch": 1.7070770263671875e-06,
      "model_forward_time": 2.268706798553467,
      "step": 1790
    },
    {
      "dynamics_loss_avg": 0.07539090737700463,
      "epoch": 1.7070770263671875e-06,
      "step": 1790
    },
    {
      "action_loss_avg": 0.03415700355544686,
      "epoch": 1.7070770263671875e-06,
      "step": 1790
    },
    {
      "epoch": 1.7070770263671875e-06,
      "step": 1790,
      "training_step_time": 10.275572299957275
    },
    {
      "epoch": 1.7080307006835937e-06,
      "model_forward_time": 1.7190296649932861,
      "step": 1791
    },
    {
      "epoch": 1.7080307006835937e-06,
      "step": 1791,
      "training_step_time": 7.853640794754028
    },
    {
      "epoch": 1.708984375e-06,
      "model_forward_time": 1.7285957336425781,
      "step": 1792
    },
    {
      "epoch": 1.708984375e-06,
      "step": 1792,
      "training_step_time": 7.708830833435059
    },
    {
      "epoch": 1.7099380493164062e-06,
      "model_forward_time": 1.7193925380706787,
      "step": 1793
    },
    {
      "epoch": 1.7099380493164062e-06,
      "step": 1793,
      "training_step_time": 7.753998041152954
    },
    {
      "epoch": 1.7108917236328125e-06,
      "model_forward_time": 1.726100206375122,
      "step": 1794
    },
    {
      "epoch": 1.7108917236328125e-06,
      "step": 1794,
      "training_step_time": 6.9517600536346436
    },
    {
      "epoch": 1.7118453979492187e-06,
      "model_forward_time": 1.7349762916564941,
      "step": 1795
    },
    {
      "epoch": 1.7118453979492187e-06,
      "step": 1795,
      "training_step_time": 6.925755977630615
    },
    {
      "epoch": 1.712799072265625e-06,
      "model_forward_time": 1.7185428142547607,
      "step": 1796
    },
    {
      "epoch": 1.712799072265625e-06,
      "step": 1796,
      "training_step_time": 7.6794373989105225
    },
    {
      "epoch": 1.7137527465820312e-06,
      "model_forward_time": 1.7436630725860596,
      "step": 1797
    },
    {
      "epoch": 1.7137527465820312e-06,
      "step": 1797,
      "training_step_time": 8.008350610733032
    },
    {
      "epoch": 1.7147064208984375e-06,
      "model_forward_time": 1.3534812927246094,
      "step": 1798
    },
    {
      "epoch": 1.7147064208984375e-06,
      "step": 1798,
      "training_step_time": 7.839354038238525
    },
    {
      "epoch": 1.7156600952148437e-06,
      "model_forward_time": 2.243544101715088,
      "step": 1799
    },
    {
      "epoch": 1.7156600952148437e-06,
      "step": 1799,
      "training_step_time": 10.345028400421143
    },
    {
      "epoch": 1.71661376953125e-06,
      "grad_norm": 1.4229283332824707,
      "learning_rate": 3.5980000000000005e-06,
      "loss": 0.1005,
      "step": 1800
    },
    {
      "epoch": 1.71661376953125e-06,
      "model_forward_time": 1.7252590656280518,
      "step": 1800
    },
    {
      "dynamics_loss_avg": 0.062382595427334306,
      "epoch": 1.71661376953125e-06,
      "step": 1800
    },
    {
      "action_loss_avg": 0.03253088695928454,
      "epoch": 1.71661376953125e-06,
      "step": 1800
    },
    {
      "epoch": 1.71661376953125e-06,
      "step": 1800,
      "training_step_time": 8.695443868637085
    },
    {
      "epoch": 1.7175674438476562e-06,
      "model_forward_time": 1.720947504043579,
      "step": 1801
    },
    {
      "epoch": 1.7175674438476562e-06,
      "step": 1801,
      "training_step_time": 8.39023733139038
    },
    {
      "epoch": 1.7185211181640625e-06,
      "model_forward_time": 1.737654685974121,
      "step": 1802
    },
    {
      "epoch": 1.7185211181640625e-06,
      "step": 1802,
      "training_step_time": 8.264861345291138
    },
    {
      "epoch": 1.7194747924804687e-06,
      "model_forward_time": 1.7240090370178223,
      "step": 1803
    },
    {
      "epoch": 1.7194747924804687e-06,
      "step": 1803,
      "training_step_time": 6.932937145233154
    },
    {
      "epoch": 1.720428466796875e-06,
      "model_forward_time": 1.7259647846221924,
      "step": 1804
    },
    {
      "epoch": 1.720428466796875e-06,
      "step": 1804,
      "training_step_time": 6.900425910949707
    },
    {
      "epoch": 1.7213821411132812e-06,
      "model_forward_time": 1.7378063201904297,
      "step": 1805
    },
    {
      "epoch": 1.7213821411132812e-06,
      "step": 1805,
      "training_step_time": 7.795264959335327
    },
    {
      "epoch": 1.7223358154296875e-06,
      "model_forward_time": 1.7323386669158936,
      "step": 1806
    },
    {
      "epoch": 1.7223358154296875e-06,
      "step": 1806,
      "training_step_time": 8.11966323852539
    },
    {
      "epoch": 1.7232894897460937e-06,
      "model_forward_time": 1.727825403213501,
      "step": 1807
    },
    {
      "epoch": 1.7232894897460937e-06,
      "step": 1807,
      "training_step_time": 9.584659814834595
    },
    {
      "epoch": 1.7242431640625e-06,
      "model_forward_time": 2.2816896438598633,
      "step": 1808
    },
    {
      "epoch": 1.7242431640625e-06,
      "step": 1808,
      "training_step_time": 7.971454381942749
    },
    {
      "epoch": 1.7251968383789062e-06,
      "model_forward_time": 1.7227187156677246,
      "step": 1809
    },
    {
      "epoch": 1.7251968383789062e-06,
      "step": 1809,
      "training_step_time": 10.388067483901978
    },
    {
      "epoch": 1.7261505126953126e-06,
      "grad_norm": 0.9596677422523499,
      "learning_rate": 3.6180000000000003e-06,
      "loss": 0.1067,
      "step": 1810
    },
    {
      "epoch": 1.7261505126953126e-06,
      "model_forward_time": 1.7098102569580078,
      "step": 1810
    },
    {
      "dynamics_loss_avg": 0.08042270187288522,
      "epoch": 1.7261505126953126e-06,
      "step": 1810
    },
    {
      "action_loss_avg": 0.036147922091186044,
      "epoch": 1.7261505126953126e-06,
      "step": 1810
    },
    {
      "epoch": 1.7261505126953126e-06,
      "step": 1810,
      "training_step_time": 7.766258955001831
    },
    {
      "epoch": 1.7271041870117187e-06,
      "model_forward_time": 1.7148466110229492,
      "step": 1811
    },
    {
      "epoch": 1.7271041870117187e-06,
      "step": 1811,
      "training_step_time": 7.677281141281128
    },
    {
      "epoch": 1.728057861328125e-06,
      "model_forward_time": 1.7215733528137207,
      "step": 1812
    },
    {
      "epoch": 1.728057861328125e-06,
      "step": 1812,
      "training_step_time": 7.242839813232422
    },
    {
      "epoch": 1.7290115356445312e-06,
      "model_forward_time": 1.7071175575256348,
      "step": 1813
    },
    {
      "epoch": 1.7290115356445312e-06,
      "step": 1813,
      "training_step_time": 6.967567443847656
    },
    {
      "epoch": 1.7299652099609376e-06,
      "model_forward_time": 1.7140727043151855,
      "step": 1814
    },
    {
      "epoch": 1.7299652099609376e-06,
      "step": 1814,
      "training_step_time": 8.352097272872925
    },
    {
      "epoch": 1.7309188842773437e-06,
      "model_forward_time": 0.999431848526001,
      "step": 1815
    },
    {
      "epoch": 1.7309188842773437e-06,
      "step": 1815,
      "training_step_time": 8.918096780776978
    },
    {
      "epoch": 1.73187255859375e-06,
      "model_forward_time": 1.7114112377166748,
      "step": 1816
    },
    {
      "epoch": 1.73187255859375e-06,
      "step": 1816,
      "training_step_time": 8.181561470031738
    },
    {
      "epoch": 1.7328262329101562e-06,
      "model_forward_time": 1.7198140621185303,
      "step": 1817
    },
    {
      "epoch": 1.7328262329101562e-06,
      "step": 1817,
      "training_step_time": 10.73172926902771
    },
    {
      "epoch": 1.7337799072265626e-06,
      "model_forward_time": 1.7030463218688965,
      "step": 1818
    },
    {
      "epoch": 1.7337799072265626e-06,
      "step": 1818,
      "training_step_time": 7.835407495498657
    },
    {
      "epoch": 1.7347335815429687e-06,
      "model_forward_time": 1.710982084274292,
      "step": 1819
    },
    {
      "epoch": 1.7347335815429687e-06,
      "step": 1819,
      "training_step_time": 9.183685541152954
    },
    {
      "epoch": 1.735687255859375e-06,
      "grad_norm": 1.287767767906189,
      "learning_rate": 3.6380000000000006e-06,
      "loss": 0.0958,
      "step": 1820
    },
    {
      "epoch": 1.735687255859375e-06,
      "model_forward_time": 1.3440964221954346,
      "step": 1820
    },
    {
      "dynamics_loss_avg": 0.08251254893839359,
      "epoch": 1.735687255859375e-06,
      "step": 1820
    },
    {
      "action_loss_avg": 0.03989528492093086,
      "epoch": 1.735687255859375e-06,
      "step": 1820
    },
    {
      "epoch": 1.735687255859375e-06,
      "step": 1820,
      "training_step_time": 7.656191349029541
    },
    {
      "epoch": 1.7366409301757812e-06,
      "model_forward_time": 1.7180571556091309,
      "step": 1821
    },
    {
      "epoch": 1.7366409301757812e-06,
      "step": 1821,
      "training_step_time": 7.178743362426758
    },
    {
      "epoch": 1.7375946044921876e-06,
      "model_forward_time": 1.7189152240753174,
      "step": 1822
    },
    {
      "epoch": 1.7375946044921876e-06,
      "step": 1822,
      "training_step_time": 6.8746538162231445
    },
    {
      "epoch": 1.7385482788085937e-06,
      "model_forward_time": 1.7152085304260254,
      "step": 1823
    },
    {
      "epoch": 1.7385482788085937e-06,
      "step": 1823,
      "training_step_time": 7.310131549835205
    },
    {
      "epoch": 1.739501953125e-06,
      "model_forward_time": 1.7117269039154053,
      "step": 1824
    },
    {
      "epoch": 1.739501953125e-06,
      "step": 1824,
      "training_step_time": 7.8936450481414795
    },
    {
      "epoch": 1.7404556274414062e-06,
      "model_forward_time": 1.7599084377288818,
      "step": 1825
    },
    {
      "epoch": 1.7404556274414062e-06,
      "step": 1825,
      "training_step_time": 8.91373062133789
    },
    {
      "epoch": 1.7414093017578126e-06,
      "model_forward_time": 1.3282582759857178,
      "step": 1826
    },
    {
      "epoch": 1.7414093017578126e-06,
      "step": 1826,
      "training_step_time": 10.169573783874512
    },
    {
      "epoch": 1.7423629760742187e-06,
      "model_forward_time": 2.2430169582366943,
      "step": 1827
    },
    {
      "epoch": 1.7423629760742187e-06,
      "step": 1827,
      "training_step_time": 9.837067365646362
    },
    {
      "epoch": 1.743316650390625e-06,
      "model_forward_time": 1.3431954383850098,
      "step": 1828
    },
    {
      "epoch": 1.743316650390625e-06,
      "step": 1828,
      "training_step_time": 8.434216976165771
    },
    {
      "epoch": 1.7442703247070312e-06,
      "model_forward_time": 1.7065939903259277,
      "step": 1829
    },
    {
      "epoch": 1.7442703247070312e-06,
      "step": 1829,
      "training_step_time": 7.761269569396973
    },
    {
      "epoch": 1.7452239990234376e-06,
      "grad_norm": 1.5783599615097046,
      "learning_rate": 3.6580000000000004e-06,
      "loss": 0.0935,
      "step": 1830
    },
    {
      "epoch": 1.7452239990234376e-06,
      "model_forward_time": 1.7235569953918457,
      "step": 1830
    },
    {
      "dynamics_loss_avg": 0.06367423720657825,
      "epoch": 1.7452239990234376e-06,
      "step": 1830
    },
    {
      "action_loss_avg": 0.027674997178837656,
      "epoch": 1.7452239990234376e-06,
      "step": 1830
    },
    {
      "epoch": 1.7452239990234376e-06,
      "step": 1830,
      "training_step_time": 7.303511381149292
    },
    {
      "epoch": 1.7461776733398437e-06,
      "model_forward_time": 1.6968319416046143,
      "step": 1831
    },
    {
      "epoch": 1.7461776733398437e-06,
      "step": 1831,
      "training_step_time": 6.724670171737671
    },
    {
      "epoch": 1.74713134765625e-06,
      "model_forward_time": 1.704801321029663,
      "step": 1832
    },
    {
      "epoch": 1.74713134765625e-06,
      "step": 1832,
      "training_step_time": 7.324135780334473
    },
    {
      "epoch": 1.7480850219726562e-06,
      "model_forward_time": 1.3479328155517578,
      "step": 1833
    },
    {
      "epoch": 1.7480850219726562e-06,
      "step": 1833,
      "training_step_time": 7.701253414154053
    },
    {
      "epoch": 1.7490386962890626e-06,
      "model_forward_time": 1.7092132568359375,
      "step": 1834
    },
    {
      "epoch": 1.7490386962890626e-06,
      "step": 1834,
      "training_step_time": 7.759549379348755
    },
    {
      "epoch": 1.7499923706054687e-06,
      "model_forward_time": 1.7087640762329102,
      "step": 1835
    },
    {
      "epoch": 1.7499923706054687e-06,
      "step": 1835,
      "training_step_time": 8.910027265548706
    },
    {
      "epoch": 1.750946044921875e-06,
      "model_forward_time": 2.2637581825256348,
      "step": 1836
    },
    {
      "epoch": 1.750946044921875e-06,
      "step": 1836,
      "training_step_time": 8.80488896369934
    },
    {
      "epoch": 1.7518997192382812e-06,
      "model_forward_time": 1.7105634212493896,
      "step": 1837
    },
    {
      "epoch": 1.7518997192382812e-06,
      "step": 1837,
      "training_step_time": 7.9130189418792725
    },
    {
      "epoch": 1.7528533935546876e-06,
      "model_forward_time": 1.7062368392944336,
      "step": 1838
    },
    {
      "epoch": 1.7528533935546876e-06,
      "step": 1838,
      "training_step_time": 7.695897579193115
    },
    {
      "epoch": 1.7538070678710937e-06,
      "model_forward_time": 1.6892430782318115,
      "step": 1839
    },
    {
      "epoch": 1.7538070678710937e-06,
      "step": 1839,
      "training_step_time": 7.434950828552246
    },
    {
      "epoch": 1.7547607421875e-06,
      "grad_norm": 1.2135953903198242,
      "learning_rate": 3.6780000000000006e-06,
      "loss": 0.0959,
      "step": 1840
    },
    {
      "epoch": 1.7547607421875e-06,
      "model_forward_time": 1.6936688423156738,
      "step": 1840
    },
    {
      "dynamics_loss_avg": 0.06744288932532072,
      "epoch": 1.7547607421875e-06,
      "step": 1840
    },
    {
      "action_loss_avg": 0.02812618496827781,
      "epoch": 1.7547607421875e-06,
      "step": 1840
    },
    {
      "epoch": 1.7547607421875e-06,
      "step": 1840,
      "training_step_time": 6.880413770675659
    },
    {
      "epoch": 1.7557144165039062e-06,
      "model_forward_time": 1.6946280002593994,
      "step": 1841
    },
    {
      "epoch": 1.7557144165039062e-06,
      "step": 1841,
      "training_step_time": 7.361713409423828
    },
    {
      "epoch": 1.7566680908203126e-06,
      "model_forward_time": 1.7079203128814697,
      "step": 1842
    },
    {
      "epoch": 1.7566680908203126e-06,
      "step": 1842,
      "training_step_time": 8.757042407989502
    },
    {
      "epoch": 1.7576217651367187e-06,
      "model_forward_time": 1.701514720916748,
      "step": 1843
    },
    {
      "epoch": 1.7576217651367187e-06,
      "step": 1843,
      "training_step_time": 7.63273286819458
    },
    {
      "epoch": 1.758575439453125e-06,
      "model_forward_time": 1.6966867446899414,
      "step": 1844
    },
    {
      "epoch": 1.758575439453125e-06,
      "step": 1844,
      "training_step_time": 8.972367286682129
    },
    {
      "epoch": 1.7595291137695312e-06,
      "model_forward_time": 1.7089762687683105,
      "step": 1845
    },
    {
      "epoch": 1.7595291137695312e-06,
      "step": 1845,
      "training_step_time": 11.365226030349731
    },
    {
      "epoch": 1.7604827880859376e-06,
      "model_forward_time": 1.703364610671997,
      "step": 1846
    },
    {
      "epoch": 1.7604827880859376e-06,
      "step": 1846,
      "training_step_time": 9.387302160263062
    },
    {
      "epoch": 1.7614364624023437e-06,
      "model_forward_time": 1.703763484954834,
      "step": 1847
    },
    {
      "epoch": 1.7614364624023437e-06,
      "step": 1847,
      "training_step_time": 7.698357105255127
    },
    {
      "epoch": 1.76239013671875e-06,
      "model_forward_time": 1.702254295349121,
      "step": 1848
    },
    {
      "epoch": 1.76239013671875e-06,
      "step": 1848,
      "training_step_time": 7.667237043380737
    },
    {
      "epoch": 1.7633438110351562e-06,
      "model_forward_time": 1.7054245471954346,
      "step": 1849
    },
    {
      "epoch": 1.7633438110351562e-06,
      "step": 1849,
      "training_step_time": 6.944429874420166
    },
    {
      "epoch": 1.7642974853515626e-06,
      "grad_norm": 1.4253369569778442,
      "learning_rate": 3.6980000000000004e-06,
      "loss": 0.1029,
      "step": 1850
    },
    {
      "epoch": 1.7642974853515626e-06,
      "model_forward_time": 1.697309970855713,
      "step": 1850
    },
    {
      "dynamics_loss_avg": 0.0742376696318388,
      "epoch": 1.7642974853515626e-06,
      "step": 1850
    },
    {
      "action_loss_avg": 0.028127282578498124,
      "epoch": 1.7642974853515626e-06,
      "step": 1850
    },
    {
      "epoch": 1.7642974853515626e-06,
      "step": 1850,
      "training_step_time": 7.339660406112671
    },
    {
      "epoch": 1.7652511596679687e-06,
      "model_forward_time": 1.7207038402557373,
      "step": 1851
    },
    {
      "epoch": 1.7652511596679687e-06,
      "step": 1851,
      "training_step_time": 7.716115236282349
    },
    {
      "epoch": 1.766204833984375e-06,
      "model_forward_time": 1.6997241973876953,
      "step": 1852
    },
    {
      "epoch": 1.766204833984375e-06,
      "step": 1852,
      "training_step_time": 7.787493705749512
    },
    {
      "epoch": 1.7671585083007812e-06,
      "model_forward_time": 1.7061386108398438,
      "step": 1853
    },
    {
      "epoch": 1.7671585083007812e-06,
      "step": 1853,
      "training_step_time": 9.792744398117065
    },
    {
      "epoch": 1.7681121826171876e-06,
      "model_forward_time": 1.702906608581543,
      "step": 1854
    },
    {
      "epoch": 1.7681121826171876e-06,
      "step": 1854,
      "training_step_time": 8.780231714248657
    },
    {
      "epoch": 1.7690658569335937e-06,
      "model_forward_time": 2.239189624786377,
      "step": 1855
    },
    {
      "epoch": 1.7690658569335937e-06,
      "step": 1855,
      "training_step_time": 8.451956510543823
    },
    {
      "epoch": 1.77001953125e-06,
      "model_forward_time": 1.70029616355896,
      "step": 1856
    },
    {
      "epoch": 1.77001953125e-06,
      "step": 1856,
      "training_step_time": 7.7254638671875
    },
    {
      "epoch": 1.7709732055664062e-06,
      "model_forward_time": 1.7047972679138184,
      "step": 1857
    },
    {
      "epoch": 1.7709732055664062e-06,
      "step": 1857,
      "training_step_time": 7.692148923873901
    },
    {
      "epoch": 1.7719268798828126e-06,
      "model_forward_time": 1.7092137336730957,
      "step": 1858
    },
    {
      "epoch": 1.7719268798828126e-06,
      "step": 1858,
      "training_step_time": 7.471444845199585
    },
    {
      "epoch": 1.7728805541992187e-06,
      "model_forward_time": 1.7041537761688232,
      "step": 1859
    },
    {
      "epoch": 1.7728805541992187e-06,
      "step": 1859,
      "training_step_time": 7.391286373138428
    },
    {
      "epoch": 1.773834228515625e-06,
      "grad_norm": 1.960524320602417,
      "learning_rate": 3.7180000000000007e-06,
      "loss": 0.1018,
      "step": 1860
    },
    {
      "epoch": 1.773834228515625e-06,
      "model_forward_time": 1.7073159217834473,
      "step": 1860
    },
    {
      "dynamics_loss_avg": 0.08069926593452692,
      "epoch": 1.773834228515625e-06,
      "step": 1860
    },
    {
      "action_loss_avg": 0.04973173290491104,
      "epoch": 1.773834228515625e-06,
      "step": 1860
    },
    {
      "epoch": 1.773834228515625e-06,
      "step": 1860,
      "training_step_time": 7.755844593048096
    },
    {
      "epoch": 1.7747879028320312e-06,
      "model_forward_time": 1.7134172916412354,
      "step": 1861
    },
    {
      "epoch": 1.7747879028320312e-06,
      "step": 1861,
      "training_step_time": 7.830845832824707
    },
    {
      "epoch": 1.7757415771484376e-06,
      "model_forward_time": 1.7221534252166748,
      "step": 1862
    },
    {
      "epoch": 1.7757415771484376e-06,
      "step": 1862,
      "training_step_time": 8.732080698013306
    },
    {
      "epoch": 1.7766952514648437e-06,
      "model_forward_time": 1.6983096599578857,
      "step": 1863
    },
    {
      "epoch": 1.7766952514648437e-06,
      "step": 1863,
      "training_step_time": 8.037151336669922
    },
    {
      "epoch": 1.77764892578125e-06,
      "model_forward_time": 2.238678455352783,
      "step": 1864
    },
    {
      "epoch": 1.77764892578125e-06,
      "step": 1864,
      "training_step_time": 9.094361066818237
    },
    {
      "epoch": 1.7786026000976562e-06,
      "model_forward_time": 0.989332914352417,
      "step": 1865
    },
    {
      "epoch": 1.7786026000976562e-06,
      "step": 1865,
      "training_step_time": 10.29996132850647
    },
    {
      "epoch": 1.7795562744140626e-06,
      "model_forward_time": 1.695671558380127,
      "step": 1866
    },
    {
      "epoch": 1.7795562744140626e-06,
      "step": 1866,
      "training_step_time": 7.6811912059783936
    },
    {
      "epoch": 1.7805099487304687e-06,
      "model_forward_time": 1.7022218704223633,
      "step": 1867
    },
    {
      "epoch": 1.7805099487304687e-06,
      "step": 1867,
      "training_step_time": 7.434553146362305
    },
    {
      "epoch": 1.781463623046875e-06,
      "model_forward_time": 1.7058088779449463,
      "step": 1868
    },
    {
      "epoch": 1.781463623046875e-06,
      "step": 1868,
      "training_step_time": 7.347726821899414
    },
    {
      "epoch": 1.7824172973632812e-06,
      "model_forward_time": 1.7015702724456787,
      "step": 1869
    },
    {
      "epoch": 1.7824172973632812e-06,
      "step": 1869,
      "training_step_time": 7.685417413711548
    },
    {
      "epoch": 1.7833709716796876e-06,
      "grad_norm": 1.0052192211151123,
      "learning_rate": 3.7380000000000005e-06,
      "loss": 0.1101,
      "step": 1870
    },
    {
      "epoch": 1.7833709716796876e-06,
      "model_forward_time": 1.7231323719024658,
      "step": 1870
    },
    {
      "dynamics_loss_avg": 0.09162686578929424,
      "epoch": 1.7833709716796876e-06,
      "step": 1870
    },
    {
      "action_loss_avg": 0.029767604544758797,
      "epoch": 1.7833709716796876e-06,
      "step": 1870
    },
    {
      "epoch": 1.7833709716796876e-06,
      "step": 1870,
      "training_step_time": 7.786253452301025
    },
    {
      "epoch": 1.7843246459960937e-06,
      "model_forward_time": 1.7022716999053955,
      "step": 1871
    },
    {
      "epoch": 1.7843246459960937e-06,
      "step": 1871,
      "training_step_time": 8.047576665878296
    },
    {
      "epoch": 1.7852783203125e-06,
      "model_forward_time": 1.7066943645477295,
      "step": 1872
    },
    {
      "epoch": 1.7852783203125e-06,
      "step": 1872,
      "training_step_time": 7.9777350425720215
    },
    {
      "epoch": 1.7862319946289062e-06,
      "model_forward_time": 1.7067196369171143,
      "step": 1873
    },
    {
      "epoch": 1.7862319946289062e-06,
      "step": 1873,
      "training_step_time": 10.249606847763062
    },
    {
      "epoch": 1.7871856689453126e-06,
      "model_forward_time": 1.7065997123718262,
      "step": 1874
    },
    {
      "epoch": 1.7871856689453126e-06,
      "step": 1874,
      "training_step_time": 8.446453332901001
    },
    {
      "epoch": 1.7881393432617188e-06,
      "model_forward_time": 1.7051756381988525,
      "step": 1875
    },
    {
      "epoch": 1.7881393432617188e-06,
      "step": 1875,
      "training_step_time": 7.7011330127716064
    },
    {
      "epoch": 1.789093017578125e-06,
      "model_forward_time": 1.7054197788238525,
      "step": 1876
    },
    {
      "epoch": 1.789093017578125e-06,
      "step": 1876,
      "training_step_time": 7.491146564483643
    },
    {
      "epoch": 1.7900466918945313e-06,
      "model_forward_time": 1.7025761604309082,
      "step": 1877
    },
    {
      "epoch": 1.7900466918945313e-06,
      "step": 1877,
      "training_step_time": 7.38555121421814
    },
    {
      "epoch": 1.7910003662109374e-06,
      "model_forward_time": 1.7126150131225586,
      "step": 1878
    },
    {
      "epoch": 1.7910003662109374e-06,
      "step": 1878,
      "training_step_time": 7.735025882720947
    },
    {
      "epoch": 1.7919540405273438e-06,
      "model_forward_time": 1.716355323791504,
      "step": 1879
    },
    {
      "epoch": 1.7919540405273438e-06,
      "step": 1879,
      "training_step_time": 8.88134241104126
    },
    {
      "epoch": 1.79290771484375e-06,
      "grad_norm": 1.4438077211380005,
      "learning_rate": 3.7580000000000007e-06,
      "loss": 0.1016,
      "step": 1880
    },
    {
      "epoch": 1.79290771484375e-06,
      "model_forward_time": 1.713447093963623,
      "step": 1880
    },
    {
      "dynamics_loss_avg": 0.0717793520539999,
      "epoch": 1.79290771484375e-06,
      "step": 1880
    },
    {
      "action_loss_avg": 0.031001051515340806,
      "epoch": 1.79290771484375e-06,
      "step": 1880
    },
    {
      "epoch": 1.79290771484375e-06,
      "step": 1880,
      "training_step_time": 8.479729652404785
    },
    {
      "epoch": 1.7938613891601563e-06,
      "model_forward_time": 1.7092270851135254,
      "step": 1881
    },
    {
      "epoch": 1.7938613891601563e-06,
      "step": 1881,
      "training_step_time": 11.155540704727173
    },
    {
      "epoch": 1.7948150634765624e-06,
      "model_forward_time": 1.7047455310821533,
      "step": 1882
    },
    {
      "epoch": 1.7948150634765624e-06,
      "step": 1882,
      "training_step_time": 8.370716571807861
    },
    {
      "epoch": 1.7957687377929688e-06,
      "model_forward_time": 2.2337095737457275,
      "step": 1883
    },
    {
      "epoch": 1.7957687377929688e-06,
      "step": 1883,
      "training_step_time": 7.818879842758179
    },
    {
      "epoch": 1.796722412109375e-06,
      "model_forward_time": 1.7198975086212158,
      "step": 1884
    },
    {
      "epoch": 1.796722412109375e-06,
      "step": 1884,
      "training_step_time": 7.636486768722534
    },
    {
      "epoch": 1.7976760864257813e-06,
      "model_forward_time": 1.7093911170959473,
      "step": 1885
    },
    {
      "epoch": 1.7976760864257813e-06,
      "step": 1885,
      "training_step_time": 7.500631809234619
    },
    {
      "epoch": 1.7986297607421874e-06,
      "model_forward_time": 1.708998680114746,
      "step": 1886
    },
    {
      "epoch": 1.7986297607421874e-06,
      "step": 1886,
      "training_step_time": 7.317418336868286
    },
    {
      "epoch": 1.7995834350585938e-06,
      "model_forward_time": 1.7061467170715332,
      "step": 1887
    },
    {
      "epoch": 1.7995834350585938e-06,
      "step": 1887,
      "training_step_time": 7.691909313201904
    },
    {
      "epoch": 1.800537109375e-06,
      "model_forward_time": 1.7110309600830078,
      "step": 1888
    },
    {
      "epoch": 1.800537109375e-06,
      "step": 1888,
      "training_step_time": 7.80090856552124
    },
    {
      "epoch": 1.8014907836914063e-06,
      "model_forward_time": 1.712146282196045,
      "step": 1889
    },
    {
      "epoch": 1.8014907836914063e-06,
      "step": 1889,
      "training_step_time": 7.965388298034668
    },
    {
      "epoch": 1.8024444580078124e-06,
      "grad_norm": 1.2174203395843506,
      "learning_rate": 3.7780000000000005e-06,
      "loss": 0.102,
      "step": 1890
    },
    {
      "epoch": 1.8024444580078124e-06,
      "model_forward_time": 1.7115278244018555,
      "step": 1890
    },
    {
      "dynamics_loss_avg": 0.07583578117191792,
      "epoch": 1.8024444580078124e-06,
      "step": 1890
    },
    {
      "action_loss_avg": 0.0439912976231426,
      "epoch": 1.8024444580078124e-06,
      "step": 1890
    },
    {
      "epoch": 1.8024444580078124e-06,
      "step": 1890,
      "training_step_time": 9.707630395889282
    },
    {
      "epoch": 1.8033981323242188e-06,
      "model_forward_time": 1.7065701484680176,
      "step": 1891
    },
    {
      "epoch": 1.8033981323242188e-06,
      "step": 1891,
      "training_step_time": 7.899055480957031
    },
    {
      "epoch": 1.804351806640625e-06,
      "model_forward_time": 2.2377772331237793,
      "step": 1892
    },
    {
      "epoch": 1.804351806640625e-06,
      "step": 1892,
      "training_step_time": 10.440418243408203
    },
    {
      "epoch": 1.8053054809570313e-06,
      "model_forward_time": 1.703458309173584,
      "step": 1893
    },
    {
      "epoch": 1.8053054809570313e-06,
      "step": 1893,
      "training_step_time": 7.715783596038818
    },
    {
      "epoch": 1.8062591552734374e-06,
      "model_forward_time": 1.7002511024475098,
      "step": 1894
    },
    {
      "epoch": 1.8062591552734374e-06,
      "step": 1894,
      "training_step_time": 7.312264442443848
    },
    {
      "epoch": 1.8072128295898438e-06,
      "model_forward_time": 1.7033541202545166,
      "step": 1895
    },
    {
      "epoch": 1.8072128295898438e-06,
      "step": 1895,
      "training_step_time": 7.334190130233765
    },
    {
      "epoch": 1.80816650390625e-06,
      "model_forward_time": 1.7080175876617432,
      "step": 1896
    },
    {
      "epoch": 1.80816650390625e-06,
      "step": 1896,
      "training_step_time": 7.696857929229736
    },
    {
      "epoch": 1.8091201782226563e-06,
      "model_forward_time": 1.7052900791168213,
      "step": 1897
    },
    {
      "epoch": 1.8091201782226563e-06,
      "step": 1897,
      "training_step_time": 8.49535846710205
    },
    {
      "epoch": 1.8100738525390624e-06,
      "model_forward_time": 1.7147455215454102,
      "step": 1898
    },
    {
      "epoch": 1.8100738525390624e-06,
      "step": 1898,
      "training_step_time": 7.959665060043335
    },
    {
      "epoch": 1.8110275268554688e-06,
      "model_forward_time": 1.7150325775146484,
      "step": 1899
    },
    {
      "epoch": 1.8110275268554688e-06,
      "step": 1899,
      "training_step_time": 9.710551738739014
    },
    {
      "epoch": 1.811981201171875e-06,
      "grad_norm": 0.8543697595596313,
      "learning_rate": 3.7980000000000007e-06,
      "loss": 0.1027,
      "step": 1900
    },
    {
      "epoch": 1.811981201171875e-06,
      "model_forward_time": 1.7008495330810547,
      "step": 1900
    },
    {
      "dynamics_loss_avg": 0.08737555705010891,
      "epoch": 1.811981201171875e-06,
      "step": 1900
    },
    {
      "action_loss_avg": 0.03294555926695466,
      "epoch": 1.811981201171875e-06,
      "step": 1900
    },
    {
      "epoch": 1.811981201171875e-06,
      "step": 1900,
      "training_step_time": 9.988831043243408
    },
    {
      "epoch": 1.8129348754882813e-06,
      "model_forward_time": 2.244122266769409,
      "step": 1901
    },
    {
      "epoch": 1.8129348754882813e-06,
      "step": 1901,
      "training_step_time": 7.924933195114136
    },
    {
      "epoch": 1.8138885498046874e-06,
      "model_forward_time": 1.6936910152435303,
      "step": 1902
    },
    {
      "epoch": 1.8138885498046874e-06,
      "step": 1902,
      "training_step_time": 7.72395133972168
    },
    {
      "epoch": 1.8148422241210938e-06,
      "model_forward_time": 1.7066495418548584,
      "step": 1903
    },
    {
      "epoch": 1.8148422241210938e-06,
      "step": 1903,
      "training_step_time": 7.508084297180176
    },
    {
      "epoch": 1.8157958984375e-06,
      "model_forward_time": 1.6971688270568848,
      "step": 1904
    },
    {
      "epoch": 1.8157958984375e-06,
      "step": 1904,
      "training_step_time": 7.31285834312439
    },
    {
      "epoch": 1.8167495727539063e-06,
      "model_forward_time": 1.6985821723937988,
      "step": 1905
    },
    {
      "epoch": 1.8167495727539063e-06,
      "step": 1905,
      "training_step_time": 7.397487163543701
    },
    {
      "epoch": 1.8177032470703124e-06,
      "model_forward_time": 1.7033135890960693,
      "step": 1906
    },
    {
      "epoch": 1.8177032470703124e-06,
      "step": 1906,
      "training_step_time": 7.859167098999023
    },
    {
      "epoch": 1.8186569213867188e-06,
      "model_forward_time": 1.745232343673706,
      "step": 1907
    },
    {
      "epoch": 1.8186569213867188e-06,
      "step": 1907,
      "training_step_time": 8.850210666656494
    },
    {
      "epoch": 1.819610595703125e-06,
      "model_forward_time": 1.6938154697418213,
      "step": 1908
    },
    {
      "epoch": 1.819610595703125e-06,
      "step": 1908,
      "training_step_time": 9.588165998458862
    },
    {
      "epoch": 1.8205642700195313e-06,
      "model_forward_time": 1.6968016624450684,
      "step": 1909
    },
    {
      "epoch": 1.8205642700195313e-06,
      "step": 1909,
      "training_step_time": 9.761609077453613
    },
    {
      "epoch": 1.8215179443359374e-06,
      "grad_norm": 1.3820527791976929,
      "learning_rate": 3.818e-06,
      "loss": 0.1004,
      "step": 1910
    },
    {
      "epoch": 1.8215179443359374e-06,
      "model_forward_time": 2.2279956340789795,
      "step": 1910
    },
    {
      "dynamics_loss_avg": 0.06974578388035298,
      "epoch": 1.8215179443359374e-06,
      "step": 1910
    },
    {
      "action_loss_avg": 0.0226653840392828,
      "epoch": 1.8215179443359374e-06,
      "step": 1910
    },
    {
      "epoch": 1.8215179443359374e-06,
      "step": 1910,
      "training_step_time": 9.440135717391968
    },
    {
      "epoch": 1.8224716186523438e-06,
      "model_forward_time": 1.7009429931640625,
      "step": 1911
    },
    {
      "epoch": 1.8224716186523438e-06,
      "step": 1911,
      "training_step_time": 10.682600498199463
    },
    {
      "epoch": 1.82342529296875e-06,
      "model_forward_time": 1.697697639465332,
      "step": 1912
    },
    {
      "epoch": 1.82342529296875e-06,
      "step": 1912,
      "training_step_time": 8.588958024978638
    },
    {
      "epoch": 1.8243789672851563e-06,
      "model_forward_time": 1.6928446292877197,
      "step": 1913
    },
    {
      "epoch": 1.8243789672851563e-06,
      "step": 1913,
      "training_step_time": 7.308257579803467
    },
    {
      "epoch": 1.8253326416015624e-06,
      "model_forward_time": 1.7013585567474365,
      "step": 1914
    },
    {
      "epoch": 1.8253326416015624e-06,
      "step": 1914,
      "training_step_time": 7.406840562820435
    },
    {
      "epoch": 1.8262863159179688e-06,
      "model_forward_time": 0.6545107364654541,
      "step": 1915
    },
    {
      "epoch": 1.8262863159179688e-06,
      "step": 1915,
      "training_step_time": 7.8801429271698
    },
    {
      "epoch": 1.827239990234375e-06,
      "model_forward_time": 1.7332911491394043,
      "step": 1916
    },
    {
      "epoch": 1.827239990234375e-06,
      "step": 1916,
      "training_step_time": 7.982748031616211
    },
    {
      "epoch": 1.8281936645507813e-06,
      "model_forward_time": 1.7058696746826172,
      "step": 1917
    },
    {
      "epoch": 1.8281936645507813e-06,
      "step": 1917,
      "training_step_time": 7.692162752151489
    },
    {
      "epoch": 1.8291473388671874e-06,
      "model_forward_time": 1.703974723815918,
      "step": 1918
    },
    {
      "epoch": 1.8291473388671874e-06,
      "step": 1918,
      "training_step_time": 7.752989292144775
    },
    {
      "epoch": 1.8301010131835938e-06,
      "model_forward_time": 1.6855485439300537,
      "step": 1919
    },
    {
      "epoch": 1.8301010131835938e-06,
      "step": 1919,
      "training_step_time": 8.958554744720459
    },
    {
      "epoch": 1.8310546875e-06,
      "grad_norm": 1.2571197748184204,
      "learning_rate": 3.838e-06,
      "loss": 0.1016,
      "step": 1920
    },
    {
      "epoch": 1.8310546875e-06,
      "model_forward_time": 2.2349023818969727,
      "step": 1920
    },
    {
      "dynamics_loss_avg": 0.08220961149781943,
      "epoch": 1.8310546875e-06,
      "step": 1920
    },
    {
      "action_loss_avg": 0.037947350088506936,
      "epoch": 1.8310546875e-06,
      "step": 1920
    },
    {
      "epoch": 1.8310546875e-06,
      "step": 1920,
      "training_step_time": 7.669407606124878
    },
    {
      "epoch": 1.8320083618164063e-06,
      "model_forward_time": 1.7024526596069336,
      "step": 1921
    },
    {
      "epoch": 1.8320083618164063e-06,
      "step": 1921,
      "training_step_time": 7.67911171913147
    },
    {
      "epoch": 1.8329620361328124e-06,
      "model_forward_time": 1.69879150390625,
      "step": 1922
    },
    {
      "epoch": 1.8329620361328124e-06,
      "step": 1922,
      "training_step_time": 7.715807199478149
    },
    {
      "epoch": 1.8339157104492188e-06,
      "model_forward_time": 1.7002687454223633,
      "step": 1923
    },
    {
      "epoch": 1.8339157104492188e-06,
      "step": 1923,
      "training_step_time": 7.364199638366699
    },
    {
      "epoch": 1.834869384765625e-06,
      "model_forward_time": 1.702005386352539,
      "step": 1924
    },
    {
      "epoch": 1.834869384765625e-06,
      "step": 1924,
      "training_step_time": 7.977789402008057
    },
    {
      "epoch": 1.8358230590820313e-06,
      "model_forward_time": 1.7140476703643799,
      "step": 1925
    },
    {
      "epoch": 1.8358230590820313e-06,
      "step": 1925,
      "training_step_time": 7.415532827377319
    },
    {
      "epoch": 1.8367767333984374e-06,
      "model_forward_time": 1.7019479274749756,
      "step": 1926
    },
    {
      "epoch": 1.8367767333984374e-06,
      "step": 1926,
      "training_step_time": 7.883440971374512
    },
    {
      "epoch": 1.8377304077148438e-06,
      "model_forward_time": 1.6935615539550781,
      "step": 1927
    },
    {
      "epoch": 1.8377304077148438e-06,
      "step": 1927,
      "training_step_time": 7.961730480194092
    },
    {
      "epoch": 1.83868408203125e-06,
      "model_forward_time": 1.710627555847168,
      "step": 1928
    },
    {
      "epoch": 1.83868408203125e-06,
      "step": 1928,
      "training_step_time": 8.960004091262817
    },
    {
      "epoch": 1.8396377563476563e-06,
      "model_forward_time": 2.271390676498413,
      "step": 1929
    },
    {
      "epoch": 1.8396377563476563e-06,
      "step": 1929,
      "training_step_time": 7.738167762756348
    },
    {
      "epoch": 1.8405914306640624e-06,
      "grad_norm": 1.663201093673706,
      "learning_rate": 3.858e-06,
      "loss": 0.108,
      "step": 1930
    },
    {
      "epoch": 1.8405914306640624e-06,
      "model_forward_time": 1.7060680389404297,
      "step": 1930
    },
    {
      "dynamics_loss_avg": 0.06707594236359,
      "epoch": 1.8405914306640624e-06,
      "step": 1930
    },
    {
      "action_loss_avg": 0.03858061702921987,
      "epoch": 1.8405914306640624e-06,
      "step": 1930
    },
    {
      "epoch": 1.8405914306640624e-06,
      "step": 1930,
      "training_step_time": 7.758549213409424
    },
    {
      "epoch": 1.8415451049804688e-06,
      "model_forward_time": 1.691234827041626,
      "step": 1931
    },
    {
      "epoch": 1.8415451049804688e-06,
      "step": 1931,
      "training_step_time": 7.72666335105896
    },
    {
      "epoch": 1.842498779296875e-06,
      "model_forward_time": 1.7115752696990967,
      "step": 1932
    },
    {
      "epoch": 1.842498779296875e-06,
      "step": 1932,
      "training_step_time": 7.833171367645264
    },
    {
      "epoch": 1.8434524536132813e-06,
      "model_forward_time": 0.9950869083404541,
      "step": 1933
    },
    {
      "epoch": 1.8434524536132813e-06,
      "step": 1933,
      "training_step_time": 7.90650486946106
    },
    {
      "epoch": 1.8444061279296874e-06,
      "model_forward_time": 1.7202727794647217,
      "step": 1934
    },
    {
      "epoch": 1.8444061279296874e-06,
      "step": 1934,
      "training_step_time": 7.3089916706085205
    },
    {
      "epoch": 1.8453598022460938e-06,
      "model_forward_time": 1.7092244625091553,
      "step": 1935
    },
    {
      "epoch": 1.8453598022460938e-06,
      "step": 1935,
      "training_step_time": 7.818740129470825
    },
    {
      "epoch": 1.8463134765625e-06,
      "model_forward_time": 1.7014663219451904,
      "step": 1936
    },
    {
      "epoch": 1.8463134765625e-06,
      "step": 1936,
      "training_step_time": 7.817823171615601
    },
    {
      "epoch": 1.8472671508789063e-06,
      "model_forward_time": 1.706895351409912,
      "step": 1937
    },
    {
      "epoch": 1.8472671508789063e-06,
      "step": 1937,
      "training_step_time": 9.479928731918335
    },
    {
      "epoch": 1.8482208251953124e-06,
      "model_forward_time": 1.7137207984924316,
      "step": 1938
    },
    {
      "epoch": 1.8482208251953124e-06,
      "step": 1938,
      "training_step_time": 7.825977087020874
    },
    {
      "epoch": 1.8491744995117188e-06,
      "model_forward_time": 2.2378647327423096,
      "step": 1939
    },
    {
      "epoch": 1.8491744995117188e-06,
      "step": 1939,
      "training_step_time": 8.161601305007935
    },
    {
      "epoch": 1.850128173828125e-06,
      "grad_norm": 1.5414650440216064,
      "learning_rate": 3.878e-06,
      "loss": 0.1031,
      "step": 1940
    },
    {
      "epoch": 1.850128173828125e-06,
      "model_forward_time": 1.7040095329284668,
      "step": 1940
    },
    {
      "dynamics_loss_avg": 0.08441524095833301,
      "epoch": 1.850128173828125e-06,
      "step": 1940
    },
    {
      "action_loss_avg": 0.040145224798470734,
      "epoch": 1.850128173828125e-06,
      "step": 1940
    },
    {
      "epoch": 1.850128173828125e-06,
      "step": 1940,
      "training_step_time": 7.713013648986816
    },
    {
      "epoch": 1.8510818481445313e-06,
      "model_forward_time": 1.6992096900939941,
      "step": 1941
    },
    {
      "epoch": 1.8510818481445313e-06,
      "step": 1941,
      "training_step_time": 7.280917406082153
    },
    {
      "epoch": 1.8520355224609375e-06,
      "model_forward_time": 1.6987197399139404,
      "step": 1942
    },
    {
      "epoch": 1.8520355224609375e-06,
      "step": 1942,
      "training_step_time": 7.889579772949219
    },
    {
      "epoch": 1.8529891967773438e-06,
      "model_forward_time": 1.3515489101409912,
      "step": 1943
    },
    {
      "epoch": 1.8529891967773438e-06,
      "step": 1943,
      "training_step_time": 7.344116449356079
    },
    {
      "epoch": 1.85394287109375e-06,
      "model_forward_time": 1.7171955108642578,
      "step": 1944
    },
    {
      "epoch": 1.85394287109375e-06,
      "step": 1944,
      "training_step_time": 7.8768980503082275
    },
    {
      "epoch": 1.8548965454101563e-06,
      "model_forward_time": 1.710496187210083,
      "step": 1945
    },
    {
      "epoch": 1.8548965454101563e-06,
      "step": 1945,
      "training_step_time": 9.415502548217773
    },
    {
      "epoch": 1.8558502197265625e-06,
      "model_forward_time": 1.704719066619873,
      "step": 1946
    },
    {
      "epoch": 1.8558502197265625e-06,
      "step": 1946,
      "training_step_time": 9.67112135887146
    },
    {
      "epoch": 1.8568038940429688e-06,
      "model_forward_time": 1.7023403644561768,
      "step": 1947
    },
    {
      "epoch": 1.8568038940429688e-06,
      "step": 1947,
      "training_step_time": 7.836457014083862
    },
    {
      "epoch": 1.857757568359375e-06,
      "model_forward_time": 2.215862989425659,
      "step": 1948
    },
    {
      "epoch": 1.857757568359375e-06,
      "step": 1948,
      "training_step_time": 8.395814657211304
    },
    {
      "epoch": 1.8587112426757813e-06,
      "model_forward_time": 1.7009680271148682,
      "step": 1949
    },
    {
      "epoch": 1.8587112426757813e-06,
      "step": 1949,
      "training_step_time": 7.650402545928955
    },
    {
      "epoch": 1.8596649169921875e-06,
      "grad_norm": 1.0582560300827026,
      "learning_rate": 3.898e-06,
      "loss": 0.0892,
      "step": 1950
    },
    {
      "epoch": 1.8596649169921875e-06,
      "model_forward_time": 1.69895601272583,
      "step": 1950
    },
    {
      "dynamics_loss_avg": 0.05343831367790699,
      "epoch": 1.8596649169921875e-06,
      "step": 1950
    },
    {
      "action_loss_avg": 0.021848983876407147,
      "epoch": 1.8596649169921875e-06,
      "step": 1950
    },
    {
      "epoch": 1.8596649169921875e-06,
      "step": 1950,
      "training_step_time": 7.712458610534668
    },
    {
      "epoch": 1.8606185913085938e-06,
      "model_forward_time": 1.6986379623413086,
      "step": 1951
    },
    {
      "epoch": 1.8606185913085938e-06,
      "step": 1951,
      "training_step_time": 7.985953330993652
    },
    {
      "epoch": 1.861572265625e-06,
      "model_forward_time": 1.7144172191619873,
      "step": 1952
    },
    {
      "epoch": 1.861572265625e-06,
      "step": 1952,
      "training_step_time": 8.735383033752441
    },
    {
      "epoch": 1.8625259399414063e-06,
      "model_forward_time": 1.7148475646972656,
      "step": 1953
    },
    {
      "epoch": 1.8625259399414063e-06,
      "step": 1953,
      "training_step_time": 7.917253017425537
    },
    {
      "epoch": 1.8634796142578125e-06,
      "model_forward_time": 1.6943902969360352,
      "step": 1954
    },
    {
      "epoch": 1.8634796142578125e-06,
      "step": 1954,
      "training_step_time": 7.843427658081055
    },
    {
      "epoch": 1.8644332885742188e-06,
      "model_forward_time": 1.7008652687072754,
      "step": 1955
    },
    {
      "epoch": 1.8644332885742188e-06,
      "step": 1955,
      "training_step_time": 9.129482746124268
    },
    {
      "epoch": 1.865386962890625e-06,
      "model_forward_time": 1.7024085521697998,
      "step": 1956
    },
    {
      "epoch": 1.865386962890625e-06,
      "step": 1956,
      "training_step_time": 8.82168984413147
    },
    {
      "epoch": 1.8663406372070313e-06,
      "model_forward_time": 2.2503206729888916,
      "step": 1957
    },
    {
      "epoch": 1.8663406372070313e-06,
      "step": 1957,
      "training_step_time": 8.512006044387817
    },
    {
      "epoch": 1.8672943115234375e-06,
      "model_forward_time": 1.6904239654541016,
      "step": 1958
    },
    {
      "epoch": 1.8672943115234375e-06,
      "step": 1958,
      "training_step_time": 7.551542043685913
    },
    {
      "epoch": 1.8682479858398438e-06,
      "model_forward_time": 1.6961419582366943,
      "step": 1959
    },
    {
      "epoch": 1.8682479858398438e-06,
      "step": 1959,
      "training_step_time": 7.447845697402954
    },
    {
      "epoch": 1.86920166015625e-06,
      "grad_norm": 1.558945894241333,
      "learning_rate": 3.9180000000000004e-06,
      "loss": 0.0885,
      "step": 1960
    },
    {
      "epoch": 1.86920166015625e-06,
      "model_forward_time": 0.9829533100128174,
      "step": 1960
    },
    {
      "dynamics_loss_avg": 0.06691500241868198,
      "epoch": 1.86920166015625e-06,
      "step": 1960
    },
    {
      "action_loss_avg": 0.02936691043432802,
      "epoch": 1.86920166015625e-06,
      "step": 1960
    },
    {
      "epoch": 1.86920166015625e-06,
      "step": 1960,
      "training_step_time": 7.830655813217163
    },
    {
      "epoch": 1.8701553344726563e-06,
      "model_forward_time": 1.7074694633483887,
      "step": 1961
    },
    {
      "epoch": 1.8701553344726563e-06,
      "step": 1961,
      "training_step_time": 7.6200549602508545
    },
    {
      "epoch": 1.8711090087890625e-06,
      "model_forward_time": 1.7056241035461426,
      "step": 1962
    },
    {
      "epoch": 1.8711090087890625e-06,
      "step": 1962,
      "training_step_time": 7.802377462387085
    },
    {
      "epoch": 1.8720626831054688e-06,
      "model_forward_time": 1.6980762481689453,
      "step": 1963
    },
    {
      "epoch": 1.8720626831054688e-06,
      "step": 1963,
      "training_step_time": 8.461242914199829
    },
    {
      "epoch": 1.873016357421875e-06,
      "model_forward_time": 1.6975915431976318,
      "step": 1964
    },
    {
      "epoch": 1.873016357421875e-06,
      "step": 1964,
      "training_step_time": 7.904613733291626
    },
    {
      "epoch": 1.8739700317382813e-06,
      "model_forward_time": 1.7037770748138428,
      "step": 1965
    },
    {
      "epoch": 1.8739700317382813e-06,
      "step": 1965,
      "training_step_time": 8.361194849014282
    },
    {
      "epoch": 1.8749237060546875e-06,
      "model_forward_time": 1.700148344039917,
      "step": 1966
    },
    {
      "epoch": 1.8749237060546875e-06,
      "step": 1966,
      "training_step_time": 9.485468864440918
    },
    {
      "epoch": 1.8758773803710938e-06,
      "model_forward_time": 1.7012784481048584,
      "step": 1967
    },
    {
      "epoch": 1.8758773803710938e-06,
      "step": 1967,
      "training_step_time": 8.572736740112305
    },
    {
      "epoch": 1.8768310546875e-06,
      "model_forward_time": 1.6970000267028809,
      "step": 1968
    },
    {
      "epoch": 1.8768310546875e-06,
      "step": 1968,
      "training_step_time": 7.527328252792358
    },
    {
      "epoch": 1.8777847290039063e-06,
      "model_forward_time": 1.696631908416748,
      "step": 1969
    },
    {
      "epoch": 1.8777847290039063e-06,
      "step": 1969,
      "training_step_time": 7.844756603240967
    },
    {
      "epoch": 1.8787384033203125e-06,
      "grad_norm": 0.9500861167907715,
      "learning_rate": 3.938e-06,
      "loss": 0.1037,
      "step": 1970
    },
    {
      "epoch": 1.8787384033203125e-06,
      "model_forward_time": 1.7098662853240967,
      "step": 1970
    },
    {
      "dynamics_loss_avg": 0.07099382001906633,
      "epoch": 1.8787384033203125e-06,
      "step": 1970
    },
    {
      "action_loss_avg": 0.03677974585443735,
      "epoch": 1.8787384033203125e-06,
      "step": 1970
    },
    {
      "epoch": 1.8787384033203125e-06,
      "step": 1970,
      "training_step_time": 8.72641110420227
    },
    {
      "epoch": 1.8796920776367188e-06,
      "model_forward_time": 1.6975724697113037,
      "step": 1971
    },
    {
      "epoch": 1.8796920776367188e-06,
      "step": 1971,
      "training_step_time": 7.8147056102752686
    },
    {
      "epoch": 1.880645751953125e-06,
      "model_forward_time": 1.7144606113433838,
      "step": 1972
    },
    {
      "epoch": 1.880645751953125e-06,
      "step": 1972,
      "training_step_time": 7.444073677062988
    },
    {
      "epoch": 1.8815994262695313e-06,
      "model_forward_time": 1.7083442211151123,
      "step": 1973
    },
    {
      "epoch": 1.8815994262695313e-06,
      "step": 1973,
      "training_step_time": 10.435687780380249
    },
    {
      "epoch": 1.8825531005859375e-06,
      "model_forward_time": 1.7038228511810303,
      "step": 1974
    },
    {
      "epoch": 1.8825531005859375e-06,
      "step": 1974,
      "training_step_time": 8.82605242729187
    },
    {
      "epoch": 1.8835067749023438e-06,
      "model_forward_time": 1.7096762657165527,
      "step": 1975
    },
    {
      "epoch": 1.8835067749023438e-06,
      "step": 1975,
      "training_step_time": 8.596054315567017
    },
    {
      "epoch": 1.88446044921875e-06,
      "model_forward_time": 2.256011962890625,
      "step": 1976
    },
    {
      "epoch": 1.88446044921875e-06,
      "step": 1976,
      "training_step_time": 8.41785216331482
    },
    {
      "epoch": 1.8854141235351563e-06,
      "model_forward_time": 1.7044343948364258,
      "step": 1977
    },
    {
      "epoch": 1.8854141235351563e-06,
      "step": 1977,
      "training_step_time": 7.590896368026733
    },
    {
      "epoch": 1.8863677978515625e-06,
      "model_forward_time": 1.7091774940490723,
      "step": 1978
    },
    {
      "epoch": 1.8863677978515625e-06,
      "step": 1978,
      "training_step_time": 7.729785203933716
    },
    {
      "epoch": 1.8873214721679688e-06,
      "model_forward_time": 1.707819938659668,
      "step": 1979
    },
    {
      "epoch": 1.8873214721679688e-06,
      "step": 1979,
      "training_step_time": 7.6179282665252686
    },
    {
      "epoch": 1.888275146484375e-06,
      "grad_norm": 1.749429702758789,
      "learning_rate": 3.958e-06,
      "loss": 0.1081,
      "step": 1980
    },
    {
      "epoch": 1.888275146484375e-06,
      "model_forward_time": 1.7068607807159424,
      "step": 1980
    },
    {
      "dynamics_loss_avg": 0.0912324383854866,
      "epoch": 1.888275146484375e-06,
      "step": 1980
    },
    {
      "action_loss_avg": 0.026696481555700303,
      "epoch": 1.888275146484375e-06,
      "step": 1980
    },
    {
      "epoch": 1.888275146484375e-06,
      "step": 1980,
      "training_step_time": 8.520829439163208
    },
    {
      "epoch": 1.8892288208007813e-06,
      "model_forward_time": 1.6975641250610352,
      "step": 1981
    },
    {
      "epoch": 1.8892288208007813e-06,
      "step": 1981,
      "training_step_time": 7.941294193267822
    },
    {
      "epoch": 1.8901824951171875e-06,
      "model_forward_time": 1.69637131690979,
      "step": 1982
    },
    {
      "epoch": 1.8901824951171875e-06,
      "step": 1982,
      "training_step_time": 9.977771282196045
    },
    {
      "epoch": 1.8911361694335938e-06,
      "model_forward_time": 1.6978886127471924,
      "step": 1983
    },
    {
      "epoch": 1.8911361694335938e-06,
      "step": 1983,
      "training_step_time": 8.143629312515259
    },
    {
      "epoch": 1.89208984375e-06,
      "model_forward_time": 1.6941215991973877,
      "step": 1984
    },
    {
      "epoch": 1.89208984375e-06,
      "step": 1984,
      "training_step_time": 7.568014144897461
    },
    {
      "epoch": 1.8930435180664063e-06,
      "model_forward_time": 2.2251696586608887,
      "step": 1985
    },
    {
      "epoch": 1.8930435180664063e-06,
      "step": 1985,
      "training_step_time": 7.508641719818115
    },
    {
      "epoch": 1.8939971923828125e-06,
      "model_forward_time": 1.696183204650879,
      "step": 1986
    },
    {
      "epoch": 1.8939971923828125e-06,
      "step": 1986,
      "training_step_time": 7.524261474609375
    },
    {
      "epoch": 1.8949508666992188e-06,
      "model_forward_time": 1.6943233013153076,
      "step": 1987
    },
    {
      "epoch": 1.8949508666992188e-06,
      "step": 1987,
      "training_step_time": 7.477163791656494
    },
    {
      "epoch": 1.895904541015625e-06,
      "model_forward_time": 1.7044029235839844,
      "step": 1988
    },
    {
      "epoch": 1.895904541015625e-06,
      "step": 1988,
      "training_step_time": 7.7354724407196045
    },
    {
      "epoch": 1.8968582153320313e-06,
      "model_forward_time": 1.6939351558685303,
      "step": 1989
    },
    {
      "epoch": 1.8968582153320313e-06,
      "step": 1989,
      "training_step_time": 7.875049352645874
    },
    {
      "epoch": 1.8978118896484375e-06,
      "grad_norm": 1.4446111917495728,
      "learning_rate": 3.978e-06,
      "loss": 0.1056,
      "step": 1990
    },
    {
      "epoch": 1.8978118896484375e-06,
      "model_forward_time": 1.7272632122039795,
      "step": 1990
    },
    {
      "dynamics_loss_avg": 0.07094864547252655,
      "epoch": 1.8978118896484375e-06,
      "step": 1990
    },
    {
      "action_loss_avg": 0.026979812979698183,
      "epoch": 1.8978118896484375e-06,
      "step": 1990
    },
    {
      "epoch": 1.8978118896484375e-06,
      "step": 1990,
      "training_step_time": 7.444366216659546
    },
    {
      "epoch": 1.8987655639648438e-06,
      "model_forward_time": 1.7022252082824707,
      "step": 1991
    },
    {
      "epoch": 1.8987655639648438e-06,
      "step": 1991,
      "training_step_time": 8.383188486099243
    },
    {
      "epoch": 1.89971923828125e-06,
      "model_forward_time": 1.7052855491638184,
      "step": 1992
    },
    {
      "epoch": 1.89971923828125e-06,
      "step": 1992,
      "training_step_time": 9.72122073173523
    },
    {
      "epoch": 1.9006729125976563e-06,
      "model_forward_time": 1.7280232906341553,
      "step": 1993
    },
    {
      "epoch": 1.9006729125976563e-06,
      "step": 1993,
      "training_step_time": 9.637014627456665
    },
    {
      "epoch": 1.9016265869140625e-06,
      "model_forward_time": 2.2243783473968506,
      "step": 1994
    },
    {
      "epoch": 1.9016265869140625e-06,
      "step": 1994,
      "training_step_time": 7.579223155975342
    },
    {
      "epoch": 1.9025802612304689e-06,
      "model_forward_time": 1.705683708190918,
      "step": 1995
    },
    {
      "epoch": 1.9025802612304689e-06,
      "step": 1995,
      "training_step_time": 7.606031179428101
    },
    {
      "epoch": 1.903533935546875e-06,
      "model_forward_time": 1.3427507877349854,
      "step": 1996
    },
    {
      "epoch": 1.903533935546875e-06,
      "step": 1996,
      "training_step_time": 7.430707216262817
    },
    {
      "epoch": 1.9044876098632814e-06,
      "model_forward_time": 1.70294189453125,
      "step": 1997
    },
    {
      "epoch": 1.9044876098632814e-06,
      "step": 1997,
      "training_step_time": 7.826635360717773
    },
    {
      "epoch": 1.9054412841796875e-06,
      "model_forward_time": 1.715451955795288,
      "step": 1998
    },
    {
      "epoch": 1.9054412841796875e-06,
      "step": 1998,
      "training_step_time": 8.064966201782227
    },
    {
      "epoch": 1.9063949584960939e-06,
      "model_forward_time": 1.7189106941223145,
      "step": 1999
    },
    {
      "epoch": 1.9063949584960939e-06,
      "step": 1999,
      "training_step_time": 7.510100841522217
    },
    {
      "epoch": 1.9073486328125e-06,
      "grad_norm": 1.107646107673645,
      "learning_rate": 3.9980000000000005e-06,
      "loss": 0.1105,
      "step": 2000
    },
    {
      "epoch": 1.9073486328125e-06,
      "model_forward_time": 1.6932518482208252,
      "step": 2000
    },
    {
      "dynamics_loss_avg": 0.0776527252048254,
      "epoch": 1.9073486328125e-06,
      "step": 2000
    },
    {
      "action_loss_avg": 0.02978597367182374,
      "epoch": 1.9073486328125e-06,
      "step": 2000
    },
    {
      "epoch": 1.9073486328125e-06,
      "step": 2000,
      "training_step_time": 10.167771339416504
    },
    {
      "epoch": 1.908302307128906e-06,
      "model_forward_time": 1.7169990539550781,
      "step": 2001
    },
    {
      "epoch": 1.908302307128906e-06,
      "step": 2001,
      "training_step_time": 8.862201690673828
    },
    {
      "epoch": 1.9092559814453123e-06,
      "model_forward_time": 1.7068710327148438,
      "step": 2002
    },
    {
      "epoch": 1.9092559814453123e-06,
      "step": 2002,
      "training_step_time": 9.55755615234375
    },
    {
      "epoch": 1.910209655761719e-06,
      "model_forward_time": 2.75675106048584,
      "step": 2003
    },
    {
      "epoch": 1.910209655761719e-06,
      "step": 2003,
      "training_step_time": 7.596719264984131
    },
    {
      "epoch": 1.911163330078125e-06,
      "model_forward_time": 1.7058584690093994,
      "step": 2004
    },
    {
      "epoch": 1.911163330078125e-06,
      "step": 2004,
      "training_step_time": 7.691846609115601
    },
    {
      "epoch": 1.912117004394531e-06,
      "model_forward_time": 1.7079482078552246,
      "step": 2005
    },
    {
      "epoch": 1.912117004394531e-06,
      "step": 2005,
      "training_step_time": 7.6439971923828125
    },
    {
      "epoch": 1.9130706787109373e-06,
      "model_forward_time": 1.7144551277160645,
      "step": 2006
    },
    {
      "epoch": 1.9130706787109373e-06,
      "step": 2006,
      "training_step_time": 7.971351623535156
    },
    {
      "epoch": 1.914024353027344e-06,
      "model_forward_time": 1.7513177394866943,
      "step": 2007
    },
    {
      "epoch": 1.914024353027344e-06,
      "step": 2007,
      "training_step_time": 7.6420464515686035
    },
    {
      "epoch": 1.91497802734375e-06,
      "model_forward_time": 1.7041046619415283,
      "step": 2008
    },
    {
      "epoch": 1.91497802734375e-06,
      "step": 2008,
      "training_step_time": 9.714282989501953
    },
    {
      "epoch": 1.915931701660156e-06,
      "model_forward_time": 1.7247428894042969,
      "step": 2009
    },
    {
      "epoch": 1.915931701660156e-06,
      "step": 2009,
      "training_step_time": 8.324375629425049
    },
    {
      "epoch": 1.9168853759765623e-06,
      "grad_norm": 1.426952838897705,
      "learning_rate": 4.018e-06,
      "loss": 0.1094,
      "step": 2010
    },
    {
      "epoch": 1.9168853759765623e-06,
      "model_forward_time": 1.6875901222229004,
      "step": 2010
    },
    {
      "dynamics_loss_avg": 0.07893378548324108,
      "epoch": 1.9168853759765623e-06,
      "step": 2010
    },
    {
      "action_loss_avg": 0.02011995706707239,
      "epoch": 1.9168853759765623e-06,
      "step": 2010
    },
    {
      "epoch": 1.9168853759765623e-06,
      "step": 2010,
      "training_step_time": 9.093353033065796
    },
    {
      "epoch": 1.917839050292969e-06,
      "model_forward_time": 1.7066099643707275,
      "step": 2011
    },
    {
      "epoch": 1.917839050292969e-06,
      "step": 2011,
      "training_step_time": 8.36152172088623
    },
    {
      "epoch": 1.918792724609375e-06,
      "model_forward_time": 1.8547728061676025,
      "step": 2012
    },
    {
      "epoch": 1.918792724609375e-06,
      "step": 2012,
      "training_step_time": 7.810027837753296
    },
    {
      "epoch": 1.919746398925781e-06,
      "model_forward_time": 1.7106742858886719,
      "step": 2013
    },
    {
      "epoch": 1.919746398925781e-06,
      "step": 2013,
      "training_step_time": 8.296902894973755
    },
    {
      "epoch": 1.9207000732421873e-06,
      "model_forward_time": 1.721278190612793,
      "step": 2014
    },
    {
      "epoch": 1.9207000732421873e-06,
      "step": 2014,
      "training_step_time": 7.600422143936157
    },
    {
      "epoch": 1.921653747558594e-06,
      "model_forward_time": 1.7079534530639648,
      "step": 2015
    },
    {
      "epoch": 1.921653747558594e-06,
      "step": 2015,
      "training_step_time": 7.602323532104492
    },
    {
      "epoch": 1.922607421875e-06,
      "model_forward_time": 1.3971302509307861,
      "step": 2016
    },
    {
      "epoch": 1.922607421875e-06,
      "step": 2016,
      "training_step_time": 7.890103101730347
    },
    {
      "epoch": 1.923561096191406e-06,
      "model_forward_time": 1.6964573860168457,
      "step": 2017
    },
    {
      "epoch": 1.923561096191406e-06,
      "step": 2017,
      "training_step_time": 8.21110987663269
    },
    {
      "epoch": 1.9245147705078123e-06,
      "model_forward_time": 1.71382737159729,
      "step": 2018
    },
    {
      "epoch": 1.9245147705078123e-06,
      "step": 2018,
      "training_step_time": 7.657314300537109
    },
    {
      "epoch": 1.925468444824219e-06,
      "model_forward_time": 1.7134954929351807,
      "step": 2019
    },
    {
      "epoch": 1.925468444824219e-06,
      "step": 2019,
      "training_step_time": 9.64762020111084
    },
    {
      "epoch": 1.926422119140625e-06,
      "grad_norm": 1.1469478607177734,
      "learning_rate": 4.038e-06,
      "loss": 0.1056,
      "step": 2020
    },
    {
      "epoch": 1.926422119140625e-06,
      "model_forward_time": 1.7081806659698486,
      "step": 2020
    },
    {
      "dynamics_loss_avg": 0.08248401284217835,
      "epoch": 1.926422119140625e-06,
      "step": 2020
    },
    {
      "action_loss_avg": 0.02816497937310487,
      "epoch": 1.926422119140625e-06,
      "step": 2020
    },
    {
      "epoch": 1.926422119140625e-06,
      "step": 2020,
      "training_step_time": 7.720660448074341
    },
    {
      "epoch": 1.927375793457031e-06,
      "model_forward_time": 2.231438636779785,
      "step": 2021
    },
    {
      "epoch": 1.927375793457031e-06,
      "step": 2021,
      "training_step_time": 7.682055234909058
    },
    {
      "epoch": 1.9283294677734373e-06,
      "model_forward_time": 1.7132062911987305,
      "step": 2022
    },
    {
      "epoch": 1.9283294677734373e-06,
      "step": 2022,
      "training_step_time": 7.646483659744263
    },
    {
      "epoch": 1.929283142089844e-06,
      "model_forward_time": 1.7272346019744873,
      "step": 2023
    },
    {
      "epoch": 1.929283142089844e-06,
      "step": 2023,
      "training_step_time": 8.570712327957153
    },
    {
      "epoch": 1.93023681640625e-06,
      "model_forward_time": 1.709601640701294,
      "step": 2024
    },
    {
      "epoch": 1.93023681640625e-06,
      "step": 2024,
      "training_step_time": 7.641615152359009
    },
    {
      "epoch": 1.931190490722656e-06,
      "model_forward_time": 1.749558925628662,
      "step": 2025
    },
    {
      "epoch": 1.931190490722656e-06,
      "step": 2025,
      "training_step_time": 7.757762908935547
    },
    {
      "epoch": 1.9321441650390623e-06,
      "model_forward_time": 1.7205479145050049,
      "step": 2026
    },
    {
      "epoch": 1.9321441650390623e-06,
      "step": 2026,
      "training_step_time": 7.812304496765137
    },
    {
      "epoch": 1.933097839355469e-06,
      "model_forward_time": 1.331254005432129,
      "step": 2027
    },
    {
      "epoch": 1.933097839355469e-06,
      "step": 2027,
      "training_step_time": 8.551742553710938
    },
    {
      "epoch": 1.934051513671875e-06,
      "model_forward_time": 1.7009506225585938,
      "step": 2028
    },
    {
      "epoch": 1.934051513671875e-06,
      "step": 2028,
      "training_step_time": 10.248889207839966
    },
    {
      "epoch": 1.935005187988281e-06,
      "model_forward_time": 1.7134532928466797,
      "step": 2029
    },
    {
      "epoch": 1.935005187988281e-06,
      "step": 2029,
      "training_step_time": 8.876657247543335
    },
    {
      "epoch": 1.9359588623046873e-06,
      "grad_norm": 0.9496733546257019,
      "learning_rate": 4.058e-06,
      "loss": 0.091,
      "step": 2030
    },
    {
      "epoch": 1.9359588623046873e-06,
      "model_forward_time": 1.3461077213287354,
      "step": 2030
    },
    {
      "dynamics_loss_avg": 0.07755245808511972,
      "epoch": 1.9359588623046873e-06,
      "step": 2030
    },
    {
      "action_loss_avg": 0.022588610369712114,
      "epoch": 1.9359588623046873e-06,
      "step": 2030
    },
    {
      "epoch": 1.9359588623046873e-06,
      "step": 2030,
      "training_step_time": 8.425298690795898
    },
    {
      "epoch": 1.936912536621094e-06,
      "model_forward_time": 2.2175257205963135,
      "step": 2031
    },
    {
      "epoch": 1.936912536621094e-06,
      "step": 2031,
      "training_step_time": 9.393222570419312
    },
    {
      "epoch": 1.9378662109375e-06,
      "model_forward_time": 1.7099440097808838,
      "step": 2032
    },
    {
      "epoch": 1.9378662109375e-06,
      "step": 2032,
      "training_step_time": 7.763279676437378
    },
    {
      "epoch": 1.938819885253906e-06,
      "model_forward_time": 1.7049169540405273,
      "step": 2033
    },
    {
      "epoch": 1.938819885253906e-06,
      "step": 2033,
      "training_step_time": 7.6451215744018555
    },
    {
      "epoch": 1.9397735595703123e-06,
      "model_forward_time": 1.7360560894012451,
      "step": 2034
    },
    {
      "epoch": 1.9397735595703123e-06,
      "step": 2034,
      "training_step_time": 7.891238212585449
    },
    {
      "epoch": 1.940727233886719e-06,
      "model_forward_time": 1.7260332107543945,
      "step": 2035
    },
    {
      "epoch": 1.940727233886719e-06,
      "step": 2035,
      "training_step_time": 7.844683885574341
    },
    {
      "epoch": 1.941680908203125e-06,
      "model_forward_time": 1.706110954284668,
      "step": 2036
    },
    {
      "epoch": 1.941680908203125e-06,
      "step": 2036,
      "training_step_time": 7.936805725097656
    },
    {
      "epoch": 1.942634582519531e-06,
      "model_forward_time": 1.7167906761169434,
      "step": 2037
    },
    {
      "epoch": 1.942634582519531e-06,
      "step": 2037,
      "training_step_time": 10.327790260314941
    },
    {
      "epoch": 1.9435882568359373e-06,
      "model_forward_time": 1.7134642601013184,
      "step": 2038
    },
    {
      "epoch": 1.9435882568359373e-06,
      "step": 2038,
      "training_step_time": 10.907235860824585
    },
    {
      "epoch": 1.944541931152344e-06,
      "model_forward_time": 1.7095730304718018,
      "step": 2039
    },
    {
      "epoch": 1.944541931152344e-06,
      "step": 2039,
      "training_step_time": 8.800988912582397
    },
    {
      "epoch": 1.94549560546875e-06,
      "grad_norm": 1.4792685508728027,
      "learning_rate": 4.078000000000001e-06,
      "loss": 0.0903,
      "step": 2040
    },
    {
      "epoch": 1.94549560546875e-06,
      "model_forward_time": 2.234673023223877,
      "step": 2040
    },
    {
      "dynamics_loss_avg": 0.0573857806622982,
      "epoch": 1.94549560546875e-06,
      "step": 2040
    },
    {
      "action_loss_avg": 0.017186229676008226,
      "epoch": 1.94549560546875e-06,
      "step": 2040
    },
    {
      "epoch": 1.94549560546875e-06,
      "step": 2040,
      "training_step_time": 7.742586374282837
    },
    {
      "epoch": 1.946449279785156e-06,
      "model_forward_time": 1.715118169784546,
      "step": 2041
    },
    {
      "epoch": 1.946449279785156e-06,
      "step": 2041,
      "training_step_time": 8.560015439987183
    },
    {
      "epoch": 1.9474029541015623e-06,
      "model_forward_time": 1.7185981273651123,
      "step": 2042
    },
    {
      "epoch": 1.9474029541015623e-06,
      "step": 2042,
      "training_step_time": 7.6508543491363525
    },
    {
      "epoch": 1.948356628417969e-06,
      "model_forward_time": 1.7036621570587158,
      "step": 2043
    },
    {
      "epoch": 1.948356628417969e-06,
      "step": 2043,
      "training_step_time": 7.907764911651611
    },
    {
      "epoch": 1.949310302734375e-06,
      "model_forward_time": 1.7570273876190186,
      "step": 2044
    },
    {
      "epoch": 1.949310302734375e-06,
      "step": 2044,
      "training_step_time": 7.922988653182983
    },
    {
      "epoch": 1.950263977050781e-06,
      "model_forward_time": 1.7115557193756104,
      "step": 2045
    },
    {
      "epoch": 1.950263977050781e-06,
      "step": 2045,
      "training_step_time": 7.490424871444702
    },
    {
      "epoch": 1.9512176513671873e-06,
      "model_forward_time": 1.7066986560821533,
      "step": 2046
    },
    {
      "epoch": 1.9512176513671873e-06,
      "step": 2046,
      "training_step_time": 9.644337892532349
    },
    {
      "epoch": 1.952171325683594e-06,
      "model_forward_time": 1.7238049507141113,
      "step": 2047
    },
    {
      "epoch": 1.952171325683594e-06,
      "step": 2047,
      "training_step_time": 8.751278638839722
    },
    {
      "epoch": 1.953125e-06,
      "model_forward_time": 1.7121155261993408,
      "step": 2048
    },
    {
      "epoch": 1.953125e-06,
      "step": 2048,
      "training_step_time": 7.744904279708862
    },
    {
      "epoch": 1.954078674316406e-06,
      "model_forward_time": 2.2414259910583496,
      "step": 2049
    },
    {
      "epoch": 1.954078674316406e-06,
      "step": 2049,
      "training_step_time": 7.640071392059326
    },
    {
      "epoch": 1.9550323486328123e-06,
      "grad_norm": 1.3332175016403198,
      "learning_rate": 4.098e-06,
      "loss": 0.1044,
      "step": 2050
    },
    {
      "epoch": 1.9550323486328123e-06,
      "model_forward_time": 1.7148613929748535,
      "step": 2050
    },
    {
      "dynamics_loss_avg": 0.06114501040428877,
      "epoch": 1.9550323486328123e-06,
      "step": 2050
    },
    {
      "action_loss_avg": 0.02152840541675687,
      "epoch": 1.9550323486328123e-06,
      "step": 2050
    },
    {
      "epoch": 1.9550323486328123e-06,
      "step": 2050,
      "training_step_time": 8.430651426315308
    },
    {
      "epoch": 1.955986022949219e-06,
      "model_forward_time": 1.7126860618591309,
      "step": 2051
    },
    {
      "epoch": 1.955986022949219e-06,
      "step": 2051,
      "training_step_time": 7.63989782333374
    },
    {
      "epoch": 1.956939697265625e-06,
      "model_forward_time": 1.7114484310150146,
      "step": 2052
    },
    {
      "epoch": 1.956939697265625e-06,
      "step": 2052,
      "training_step_time": 7.886930465698242
    },
    {
      "epoch": 1.957893371582031e-06,
      "model_forward_time": 1.765099287033081,
      "step": 2053
    },
    {
      "epoch": 1.957893371582031e-06,
      "step": 2053,
      "training_step_time": 7.950522422790527
    },
    {
      "epoch": 1.9588470458984373e-06,
      "model_forward_time": 1.7050564289093018,
      "step": 2054
    },
    {
      "epoch": 1.9588470458984373e-06,
      "step": 2054,
      "training_step_time": 7.454832315444946
    },
    {
      "epoch": 1.959800720214844e-06,
      "model_forward_time": 1.705711841583252,
      "step": 2055
    },
    {
      "epoch": 1.959800720214844e-06,
      "step": 2055,
      "training_step_time": 8.34731149673462
    },
    {
      "epoch": 1.96075439453125e-06,
      "model_forward_time": 1.7082993984222412,
      "step": 2056
    },
    {
      "epoch": 1.96075439453125e-06,
      "step": 2056,
      "training_step_time": 8.88919973373413
    },
    {
      "epoch": 1.961708068847656e-06,
      "model_forward_time": 1.706808090209961,
      "step": 2057
    },
    {
      "epoch": 1.961708068847656e-06,
      "step": 2057,
      "training_step_time": 8.3882155418396
    },
    {
      "epoch": 1.9626617431640623e-06,
      "model_forward_time": 2.2453620433807373,
      "step": 2058
    },
    {
      "epoch": 1.9626617431640623e-06,
      "step": 2058,
      "training_step_time": 7.679620027542114
    },
    {
      "epoch": 1.963615417480469e-06,
      "model_forward_time": 1.6895394325256348,
      "step": 2059
    },
    {
      "epoch": 1.963615417480469e-06,
      "step": 2059,
      "training_step_time": 7.628751277923584
    },
    {
      "epoch": 1.964569091796875e-06,
      "grad_norm": 2.704709529876709,
      "learning_rate": 4.118e-06,
      "loss": 0.1121,
      "step": 2060
    },
    {
      "epoch": 1.964569091796875e-06,
      "model_forward_time": 1.6950139999389648,
      "step": 2060
    },
    {
      "dynamics_loss_avg": 0.09341713301837444,
      "epoch": 1.964569091796875e-06,
      "step": 2060
    },
    {
      "action_loss_avg": 0.03632458113133907,
      "epoch": 1.964569091796875e-06,
      "step": 2060
    },
    {
      "epoch": 1.964569091796875e-06,
      "step": 2060,
      "training_step_time": 7.722568035125732
    },
    {
      "epoch": 1.965522766113281e-06,
      "model_forward_time": 1.6927011013031006,
      "step": 2061
    },
    {
      "epoch": 1.965522766113281e-06,
      "step": 2061,
      "training_step_time": 7.88005518913269
    },
    {
      "epoch": 1.9664764404296873e-06,
      "model_forward_time": 1.7066848278045654,
      "step": 2062
    },
    {
      "epoch": 1.9664764404296873e-06,
      "step": 2062,
      "training_step_time": 7.935410499572754
    },
    {
      "epoch": 1.967430114746094e-06,
      "model_forward_time": 1.7063140869140625,
      "step": 2063
    },
    {
      "epoch": 1.967430114746094e-06,
      "step": 2063,
      "training_step_time": 8.413818359375
    },
    {
      "epoch": 1.9683837890625e-06,
      "model_forward_time": 1.7230937480926514,
      "step": 2064
    },
    {
      "epoch": 1.9683837890625e-06,
      "step": 2064,
      "training_step_time": 7.886459112167358
    },
    {
      "epoch": 1.969337463378906e-06,
      "model_forward_time": 1.7064192295074463,
      "step": 2065
    },
    {
      "epoch": 1.969337463378906e-06,
      "step": 2065,
      "training_step_time": 8.537032127380371
    },
    {
      "epoch": 1.9702911376953123e-06,
      "model_forward_time": 1.7015411853790283,
      "step": 2066
    },
    {
      "epoch": 1.9702911376953123e-06,
      "step": 2066,
      "training_step_time": 8.790063619613647
    },
    {
      "epoch": 1.971244812011719e-06,
      "model_forward_time": 2.238668203353882,
      "step": 2067
    },
    {
      "epoch": 1.971244812011719e-06,
      "step": 2067,
      "training_step_time": 7.736811876296997
    },
    {
      "epoch": 1.972198486328125e-06,
      "model_forward_time": 1.7119786739349365,
      "step": 2068
    },
    {
      "epoch": 1.972198486328125e-06,
      "step": 2068,
      "training_step_time": 7.756235599517822
    },
    {
      "epoch": 1.973152160644531e-06,
      "model_forward_time": 1.7130725383758545,
      "step": 2069
    },
    {
      "epoch": 1.973152160644531e-06,
      "step": 2069,
      "training_step_time": 7.79140043258667
    },
    {
      "epoch": 1.9741058349609373e-06,
      "grad_norm": 1.1118547916412354,
      "learning_rate": 4.138e-06,
      "loss": 0.0947,
      "step": 2070
    },
    {
      "epoch": 1.9741058349609373e-06,
      "model_forward_time": 1.7115726470947266,
      "step": 2070
    },
    {
      "dynamics_loss_avg": 0.08019212409853935,
      "epoch": 1.9741058349609373e-06,
      "step": 2070
    },
    {
      "action_loss_avg": 0.021598212141543626,
      "epoch": 1.9741058349609373e-06,
      "step": 2070
    },
    {
      "epoch": 1.9741058349609373e-06,
      "step": 2070,
      "training_step_time": 9.03831171989441
    },
    {
      "epoch": 1.975059509277344e-06,
      "model_forward_time": 1.7278180122375488,
      "step": 2071
    },
    {
      "epoch": 1.975059509277344e-06,
      "step": 2071,
      "training_step_time": 7.947620868682861
    },
    {
      "epoch": 1.97601318359375e-06,
      "model_forward_time": 1.7057793140411377,
      "step": 2072
    },
    {
      "epoch": 1.97601318359375e-06,
      "step": 2072,
      "training_step_time": 7.362224340438843
    },
    {
      "epoch": 1.976966857910156e-06,
      "model_forward_time": 1.7276389598846436,
      "step": 2073
    },
    {
      "epoch": 1.976966857910156e-06,
      "step": 2073,
      "training_step_time": 7.524152994155884
    },
    {
      "epoch": 1.9779205322265624e-06,
      "model_forward_time": 1.7307591438293457,
      "step": 2074
    },
    {
      "epoch": 1.9779205322265624e-06,
      "step": 2074,
      "training_step_time": 9.444872379302979
    },
    {
      "epoch": 1.978874206542969e-06,
      "model_forward_time": 1.728757619857788,
      "step": 2075
    },
    {
      "epoch": 1.978874206542969e-06,
      "step": 2075,
      "training_step_time": 8.631635189056396
    },
    {
      "epoch": 1.979827880859375e-06,
      "model_forward_time": 1.7251851558685303,
      "step": 2076
    },
    {
      "epoch": 1.979827880859375e-06,
      "step": 2076,
      "training_step_time": 8.434016227722168
    },
    {
      "epoch": 1.980781555175781e-06,
      "model_forward_time": 2.261190176010132,
      "step": 2077
    },
    {
      "epoch": 1.980781555175781e-06,
      "step": 2077,
      "training_step_time": 8.365924835205078
    },
    {
      "epoch": 1.9817352294921874e-06,
      "model_forward_time": 1.7277553081512451,
      "step": 2078
    },
    {
      "epoch": 1.9817352294921874e-06,
      "step": 2078,
      "training_step_time": 7.84098482131958
    },
    {
      "epoch": 1.982688903808594e-06,
      "model_forward_time": 1.36281418800354,
      "step": 2079
    },
    {
      "epoch": 1.982688903808594e-06,
      "step": 2079,
      "training_step_time": 8.06536054611206
    },
    {
      "epoch": 1.983642578125e-06,
      "grad_norm": 0.680354654788971,
      "learning_rate": 4.158000000000001e-06,
      "loss": 0.101,
      "step": 2080
    },
    {
      "epoch": 1.983642578125e-06,
      "model_forward_time": 1.7452809810638428,
      "step": 2080
    },
    {
      "dynamics_loss_avg": 0.061321061290800574,
      "epoch": 1.983642578125e-06,
      "step": 2080
    },
    {
      "action_loss_avg": 0.019663850869983436,
      "epoch": 1.983642578125e-06,
      "step": 2080
    },
    {
      "epoch": 1.983642578125e-06,
      "step": 2080,
      "training_step_time": 8.612361431121826
    },
    {
      "epoch": 1.9845962524414062e-06,
      "model_forward_time": 1.7103357315063477,
      "step": 2081
    },
    {
      "epoch": 1.9845962524414062e-06,
      "step": 2081,
      "training_step_time": 7.891975402832031
    },
    {
      "epoch": 1.9855499267578124e-06,
      "model_forward_time": 1.7377650737762451,
      "step": 2082
    },
    {
      "epoch": 1.9855499267578124e-06,
      "step": 2082,
      "training_step_time": 9.441460847854614
    },
    {
      "epoch": 1.986503601074219e-06,
      "model_forward_time": 1.729419231414795,
      "step": 2083
    },
    {
      "epoch": 1.986503601074219e-06,
      "step": 2083,
      "training_step_time": 7.889155626296997
    },
    {
      "epoch": 1.987457275390625e-06,
      "model_forward_time": 1.7357256412506104,
      "step": 2084
    },
    {
      "epoch": 1.987457275390625e-06,
      "step": 2084,
      "training_step_time": 8.88152003288269
    },
    {
      "epoch": 1.9884109497070312e-06,
      "model_forward_time": 1.6903042793273926,
      "step": 2085
    },
    {
      "epoch": 1.9884109497070312e-06,
      "step": 2085,
      "training_step_time": 9.51119875907898
    },
    {
      "epoch": 1.9893646240234374e-06,
      "model_forward_time": 2.241525411605835,
      "step": 2086
    },
    {
      "epoch": 1.9893646240234374e-06,
      "step": 2086,
      "training_step_time": 7.662349224090576
    },
    {
      "epoch": 1.990318298339844e-06,
      "model_forward_time": 1.7113466262817383,
      "step": 2087
    },
    {
      "epoch": 1.990318298339844e-06,
      "step": 2087,
      "training_step_time": 7.76189398765564
    },
    {
      "epoch": 1.99127197265625e-06,
      "model_forward_time": 1.706197738647461,
      "step": 2088
    },
    {
      "epoch": 1.99127197265625e-06,
      "step": 2088,
      "training_step_time": 7.918027877807617
    },
    {
      "epoch": 1.9922256469726562e-06,
      "model_forward_time": 1.7128772735595703,
      "step": 2089
    },
    {
      "epoch": 1.9922256469726562e-06,
      "step": 2089,
      "training_step_time": 8.710749626159668
    },
    {
      "epoch": 1.9931793212890624e-06,
      "grad_norm": 0.923542320728302,
      "learning_rate": 4.1780000000000005e-06,
      "loss": 0.108,
      "step": 2090
    },
    {
      "epoch": 1.9931793212890624e-06,
      "model_forward_time": 1.7044360637664795,
      "step": 2090
    },
    {
      "dynamics_loss_avg": 0.06314112842082978,
      "epoch": 1.9931793212890624e-06,
      "step": 2090
    },
    {
      "action_loss_avg": 0.017218409525230526,
      "epoch": 1.9931793212890624e-06,
      "step": 2090
    },
    {
      "epoch": 1.9931793212890624e-06,
      "step": 2090,
      "training_step_time": 7.760956048965454
    },
    {
      "epoch": 1.994132995605469e-06,
      "model_forward_time": 1.7145051956176758,
      "step": 2091
    },
    {
      "epoch": 1.994132995605469e-06,
      "step": 2091,
      "training_step_time": 8.024527072906494
    },
    {
      "epoch": 1.995086669921875e-06,
      "model_forward_time": 1.7064459323883057,
      "step": 2092
    },
    {
      "epoch": 1.995086669921875e-06,
      "step": 2092,
      "training_step_time": 8.495619297027588
    },
    {
      "epoch": 1.9960403442382812e-06,
      "model_forward_time": 1.7034003734588623,
      "step": 2093
    },
    {
      "epoch": 1.9960403442382812e-06,
      "step": 2093,
      "training_step_time": 7.766770601272583
    },
    {
      "epoch": 1.9969940185546874e-06,
      "model_forward_time": 1.7131567001342773,
      "step": 2094
    },
    {
      "epoch": 1.9969940185546874e-06,
      "step": 2094,
      "training_step_time": 7.8161561489105225
    },
    {
      "epoch": 1.997947692871094e-06,
      "model_forward_time": 1.4882400035858154,
      "step": 2095
    },
    {
      "epoch": 1.997947692871094e-06,
      "step": 2095,
      "training_step_time": 7.687823534011841
    },
    {
      "epoch": 1.9989013671875e-06,
      "model_forward_time": 1.7025282382965088,
      "step": 2096
    },
    {
      "epoch": 1.9989013671875e-06,
      "step": 2096,
      "training_step_time": 7.742549657821655
    },
    {
      "epoch": 1.9998550415039062e-06,
      "model_forward_time": 1.7271852493286133,
      "step": 2097
    },
    {
      "epoch": 1.9998550415039062e-06,
      "step": 2097,
      "training_step_time": 7.575917720794678
    },
    {
      "epoch": 2.0008087158203124e-06,
      "model_forward_time": 1.7091288566589355,
      "step": 2098
    },
    {
      "epoch": 2.0008087158203124e-06,
      "step": 2098,
      "training_step_time": 7.937615394592285
    },
    {
      "epoch": 2.001762390136719e-06,
      "model_forward_time": 1.7435276508331299,
      "step": 2099
    },
    {
      "epoch": 2.001762390136719e-06,
      "step": 2099,
      "training_step_time": 7.757485389709473
    },
    {
      "epoch": 2.002716064453125e-06,
      "grad_norm": 1.43766188621521,
      "learning_rate": 4.198e-06,
      "loss": 0.1044,
      "step": 2100
    },
    {
      "epoch": 2.002716064453125e-06,
      "model_forward_time": 1.7122993469238281,
      "step": 2100
    },
    {
      "dynamics_loss_avg": 0.07066967450082302,
      "epoch": 2.002716064453125e-06,
      "step": 2100
    },
    {
      "action_loss_avg": 0.018870839197188616,
      "epoch": 2.002716064453125e-06,
      "step": 2100
    },
    {
      "epoch": 2.002716064453125e-06,
      "step": 2100,
      "training_step_time": 7.151448488235474
    },
    {
      "epoch": 2.0036697387695312e-06,
      "model_forward_time": 1.3285975456237793,
      "step": 2101
    },
    {
      "epoch": 2.0036697387695312e-06,
      "step": 2101,
      "training_step_time": 9.148372888565063
    },
    {
      "epoch": 2.0046234130859374e-06,
      "model_forward_time": 1.708583116531372,
      "step": 2102
    },
    {
      "epoch": 2.0046234130859374e-06,
      "step": 2102,
      "training_step_time": 7.79626727104187
    },
    {
      "epoch": 2.005577087402344e-06,
      "model_forward_time": 1.7075669765472412,
      "step": 2103
    },
    {
      "epoch": 2.005577087402344e-06,
      "step": 2103,
      "training_step_time": 7.795585870742798
    },
    {
      "epoch": 2.00653076171875e-06,
      "model_forward_time": 1.7072412967681885,
      "step": 2104
    },
    {
      "epoch": 2.00653076171875e-06,
      "step": 2104,
      "training_step_time": 8.706549882888794
    },
    {
      "epoch": 2.0074844360351562e-06,
      "model_forward_time": 2.207717180252075,
      "step": 2105
    },
    {
      "epoch": 2.0074844360351562e-06,
      "step": 2105,
      "training_step_time": 7.691911697387695
    },
    {
      "epoch": 2.0084381103515624e-06,
      "model_forward_time": 1.6928260326385498,
      "step": 2106
    },
    {
      "epoch": 2.0084381103515624e-06,
      "step": 2106,
      "training_step_time": 8.543097257614136
    },
    {
      "epoch": 2.009391784667969e-06,
      "model_forward_time": 1.6975598335266113,
      "step": 2107
    },
    {
      "epoch": 2.009391784667969e-06,
      "step": 2107,
      "training_step_time": 7.968254804611206
    },
    {
      "epoch": 2.010345458984375e-06,
      "model_forward_time": 1.707343339920044,
      "step": 2108
    },
    {
      "epoch": 2.010345458984375e-06,
      "step": 2108,
      "training_step_time": 7.7921671867370605
    },
    {
      "epoch": 2.0112991333007812e-06,
      "model_forward_time": 1.6832540035247803,
      "step": 2109
    },
    {
      "epoch": 2.0112991333007812e-06,
      "step": 2109,
      "training_step_time": 7.722655534744263
    },
    {
      "epoch": 2.0122528076171874e-06,
      "grad_norm": 1.0103217363357544,
      "learning_rate": 4.218e-06,
      "loss": 0.1002,
      "step": 2110
    },
    {
      "epoch": 2.0122528076171874e-06,
      "model_forward_time": 1.7065985202789307,
      "step": 2110
    },
    {
      "dynamics_loss_avg": 0.061679482832551,
      "epoch": 2.0122528076171874e-06,
      "step": 2110
    },
    {
      "action_loss_avg": 0.019810167187824846,
      "epoch": 2.0122528076171874e-06,
      "step": 2110
    },
    {
      "epoch": 2.0122528076171874e-06,
      "step": 2110,
      "training_step_time": 7.615749835968018
    },
    {
      "epoch": 2.013206481933594e-06,
      "model_forward_time": 1.6911544799804688,
      "step": 2111
    },
    {
      "epoch": 2.013206481933594e-06,
      "step": 2111,
      "training_step_time": 9.919934749603271
    },
    {
      "epoch": 2.01416015625e-06,
      "model_forward_time": 1.689922571182251,
      "step": 2112
    },
    {
      "epoch": 2.01416015625e-06,
      "step": 2112,
      "training_step_time": 7.799763202667236
    },
    {
      "epoch": 2.0151138305664062e-06,
      "model_forward_time": 1.6878845691680908,
      "step": 2113
    },
    {
      "epoch": 2.0151138305664062e-06,
      "step": 2113,
      "training_step_time": 8.43955373764038
    },
    {
      "epoch": 2.0160675048828124e-06,
      "model_forward_time": 2.237048387527466,
      "step": 2114
    },
    {
      "epoch": 2.0160675048828124e-06,
      "step": 2114,
      "training_step_time": 9.302842140197754
    },
    {
      "epoch": 2.017021179199219e-06,
      "model_forward_time": 1.6890413761138916,
      "step": 2115
    },
    {
      "epoch": 2.017021179199219e-06,
      "step": 2115,
      "training_step_time": 7.646969318389893
    },
    {
      "epoch": 2.017974853515625e-06,
      "model_forward_time": 1.6963310241699219,
      "step": 2116
    },
    {
      "epoch": 2.017974853515625e-06,
      "step": 2116,
      "training_step_time": 8.56899118423462
    },
    {
      "epoch": 2.0189285278320312e-06,
      "model_forward_time": 1.7023744583129883,
      "step": 2117
    },
    {
      "epoch": 2.0189285278320312e-06,
      "step": 2117,
      "training_step_time": 8.262853622436523
    },
    {
      "epoch": 2.0198822021484374e-06,
      "model_forward_time": 1.6889193058013916,
      "step": 2118
    },
    {
      "epoch": 2.0198822021484374e-06,
      "step": 2118,
      "training_step_time": 7.325445890426636
    },
    {
      "epoch": 2.020835876464844e-06,
      "model_forward_time": 1.710268497467041,
      "step": 2119
    },
    {
      "epoch": 2.020835876464844e-06,
      "step": 2119,
      "training_step_time": 8.32298493385315
    },
    {
      "epoch": 2.02178955078125e-06,
      "grad_norm": 0.9589949250221252,
      "learning_rate": 4.238000000000001e-06,
      "loss": 0.1047,
      "step": 2120
    },
    {
      "epoch": 2.02178955078125e-06,
      "model_forward_time": 1.6884751319885254,
      "step": 2120
    },
    {
      "dynamics_loss_avg": 0.09112454168498516,
      "epoch": 2.02178955078125e-06,
      "step": 2120
    },
    {
      "action_loss_avg": 0.023014398012310266,
      "epoch": 2.02178955078125e-06,
      "step": 2120
    },
    {
      "epoch": 2.02178955078125e-06,
      "step": 2120,
      "training_step_time": 9.245913982391357
    },
    {
      "epoch": 2.0227432250976562e-06,
      "model_forward_time": 1.7006356716156006,
      "step": 2121
    },
    {
      "epoch": 2.0227432250976562e-06,
      "step": 2121,
      "training_step_time": 7.802175283432007
    },
    {
      "epoch": 2.0236968994140624e-06,
      "model_forward_time": 1.6985511779785156,
      "step": 2122
    },
    {
      "epoch": 2.0236968994140624e-06,
      "step": 2122,
      "training_step_time": 8.578377723693848
    },
    {
      "epoch": 2.024650573730469e-06,
      "model_forward_time": 2.2534806728363037,
      "step": 2123
    },
    {
      "epoch": 2.024650573730469e-06,
      "step": 2123,
      "training_step_time": 7.74745774269104
    },
    {
      "epoch": 2.025604248046875e-06,
      "model_forward_time": 1.690458059310913,
      "step": 2124
    },
    {
      "epoch": 2.025604248046875e-06,
      "step": 2124,
      "training_step_time": 7.64414644241333
    },
    {
      "epoch": 2.0265579223632812e-06,
      "model_forward_time": 1.695190668106079,
      "step": 2125
    },
    {
      "epoch": 2.0265579223632812e-06,
      "step": 2125,
      "training_step_time": 7.875758171081543
    },
    {
      "epoch": 2.0275115966796874e-06,
      "model_forward_time": 1.6979484558105469,
      "step": 2126
    },
    {
      "epoch": 2.0275115966796874e-06,
      "step": 2126,
      "training_step_time": 7.728906869888306
    },
    {
      "epoch": 2.0284652709960935e-06,
      "model_forward_time": 1.6887390613555908,
      "step": 2127
    },
    {
      "epoch": 2.0284652709960935e-06,
      "step": 2127,
      "training_step_time": 7.780534982681274
    },
    {
      "epoch": 2.0294189453125e-06,
      "model_forward_time": 1.6961770057678223,
      "step": 2128
    },
    {
      "epoch": 2.0294189453125e-06,
      "step": 2128,
      "training_step_time": 7.727710247039795
    },
    {
      "epoch": 2.0303726196289063e-06,
      "model_forward_time": 1.6933045387268066,
      "step": 2129
    },
    {
      "epoch": 2.0303726196289063e-06,
      "step": 2129,
      "training_step_time": 11.499256372451782
    },
    {
      "epoch": 2.0313262939453124e-06,
      "grad_norm": 0.8312352895736694,
      "learning_rate": 4.2580000000000006e-06,
      "loss": 0.0974,
      "step": 2130
    },
    {
      "epoch": 2.0313262939453124e-06,
      "model_forward_time": 1.7241144180297852,
      "step": 2130
    },
    {
      "dynamics_loss_avg": 0.06663514748215675,
      "epoch": 2.0313262939453124e-06,
      "step": 2130
    },
    {
      "action_loss_avg": 0.021360671473667027,
      "epoch": 2.0313262939453124e-06,
      "step": 2130
    },
    {
      "epoch": 2.0313262939453124e-06,
      "step": 2130,
      "training_step_time": 8.549168348312378
    },
    {
      "epoch": 2.0322799682617185e-06,
      "model_forward_time": 1.3347458839416504,
      "step": 2131
    },
    {
      "epoch": 2.0322799682617185e-06,
      "step": 2131,
      "training_step_time": 8.871984720230103
    },
    {
      "epoch": 2.033233642578125e-06,
      "model_forward_time": 2.2215821743011475,
      "step": 2132
    },
    {
      "epoch": 2.033233642578125e-06,
      "step": 2132,
      "training_step_time": 8.656193017959595
    },
    {
      "epoch": 2.0341873168945313e-06,
      "model_forward_time": 1.7094075679779053,
      "step": 2133
    },
    {
      "epoch": 2.0341873168945313e-06,
      "step": 2133,
      "training_step_time": 7.673518896102905
    },
    {
      "epoch": 2.0351409912109374e-06,
      "model_forward_time": 1.708177089691162,
      "step": 2134
    },
    {
      "epoch": 2.0351409912109374e-06,
      "step": 2134,
      "training_step_time": 7.762649059295654
    },
    {
      "epoch": 2.0360946655273435e-06,
      "model_forward_time": 1.7092792987823486,
      "step": 2135
    },
    {
      "epoch": 2.0360946655273435e-06,
      "step": 2135,
      "training_step_time": 8.097452640533447
    },
    {
      "epoch": 2.03704833984375e-06,
      "model_forward_time": 1.7202422618865967,
      "step": 2136
    },
    {
      "epoch": 2.03704833984375e-06,
      "step": 2136,
      "training_step_time": 8.882154703140259
    },
    {
      "epoch": 2.0380020141601563e-06,
      "model_forward_time": 1.7153453826904297,
      "step": 2137
    },
    {
      "epoch": 2.0380020141601563e-06,
      "step": 2137,
      "training_step_time": 7.686847686767578
    },
    {
      "epoch": 2.0389556884765624e-06,
      "model_forward_time": 1.7244741916656494,
      "step": 2138
    },
    {
      "epoch": 2.0389556884765624e-06,
      "step": 2138,
      "training_step_time": 8.854335308074951
    },
    {
      "epoch": 2.0399093627929686e-06,
      "model_forward_time": 1.7098581790924072,
      "step": 2139
    },
    {
      "epoch": 2.0399093627929686e-06,
      "step": 2139,
      "training_step_time": 8.922451972961426
    },
    {
      "epoch": 2.040863037109375e-06,
      "grad_norm": 0.9330998659133911,
      "learning_rate": 4.278e-06,
      "loss": 0.0929,
      "step": 2140
    },
    {
      "epoch": 2.040863037109375e-06,
      "model_forward_time": 1.7091388702392578,
      "step": 2140
    },
    {
      "dynamics_loss_avg": 0.06007766854017973,
      "epoch": 2.040863037109375e-06,
      "step": 2140
    },
    {
      "action_loss_avg": 0.01787739647552371,
      "epoch": 2.040863037109375e-06,
      "step": 2140
    },
    {
      "epoch": 2.040863037109375e-06,
      "step": 2140,
      "training_step_time": 7.951577186584473
    },
    {
      "epoch": 2.0418167114257813e-06,
      "model_forward_time": 1.7108428478240967,
      "step": 2141
    },
    {
      "epoch": 2.0418167114257813e-06,
      "step": 2141,
      "training_step_time": 7.779580116271973
    },
    {
      "epoch": 2.0427703857421874e-06,
      "model_forward_time": 2.24251389503479,
      "step": 2142
    },
    {
      "epoch": 2.0427703857421874e-06,
      "step": 2142,
      "training_step_time": 8.33101511001587
    },
    {
      "epoch": 2.0437240600585936e-06,
      "model_forward_time": 1.7034790515899658,
      "step": 2143
    },
    {
      "epoch": 2.0437240600585936e-06,
      "step": 2143,
      "training_step_time": 7.6217262744903564
    },
    {
      "epoch": 2.044677734375e-06,
      "model_forward_time": 1.7104883193969727,
      "step": 2144
    },
    {
      "epoch": 2.044677734375e-06,
      "step": 2144,
      "training_step_time": 8.06575083732605
    },
    {
      "epoch": 2.0456314086914063e-06,
      "model_forward_time": 1.716101884841919,
      "step": 2145
    },
    {
      "epoch": 2.0456314086914063e-06,
      "step": 2145,
      "training_step_time": 7.87990140914917
    },
    {
      "epoch": 2.0465850830078124e-06,
      "model_forward_time": 1.7074439525604248,
      "step": 2146
    },
    {
      "epoch": 2.0465850830078124e-06,
      "step": 2146,
      "training_step_time": 7.510106325149536
    },
    {
      "epoch": 2.0475387573242186e-06,
      "model_forward_time": 1.7276055812835693,
      "step": 2147
    },
    {
      "epoch": 2.0475387573242186e-06,
      "step": 2147,
      "training_step_time": 8.887414693832397
    },
    {
      "epoch": 2.048492431640625e-06,
      "model_forward_time": 1.7122418880462646,
      "step": 2148
    },
    {
      "epoch": 2.048492431640625e-06,
      "step": 2148,
      "training_step_time": 7.8734869956970215
    },
    {
      "epoch": 2.0494461059570313e-06,
      "model_forward_time": 1.7086467742919922,
      "step": 2149
    },
    {
      "epoch": 2.0494461059570313e-06,
      "step": 2149,
      "training_step_time": 7.735761642456055
    },
    {
      "epoch": 2.0503997802734374e-06,
      "grad_norm": 1.086315393447876,
      "learning_rate": 4.298e-06,
      "loss": 0.1001,
      "step": 2150
    },
    {
      "epoch": 2.0503997802734374e-06,
      "model_forward_time": 1.7089478969573975,
      "step": 2150
    },
    {
      "dynamics_loss_avg": 0.06565698329359293,
      "epoch": 2.0503997802734374e-06,
      "step": 2150
    },
    {
      "action_loss_avg": 0.017529558343812823,
      "epoch": 2.0503997802734374e-06,
      "step": 2150
    },
    {
      "epoch": 2.0503997802734374e-06,
      "step": 2150,
      "training_step_time": 7.658941030502319
    },
    {
      "epoch": 2.0513534545898436e-06,
      "model_forward_time": 2.203855514526367,
      "step": 2151
    },
    {
      "epoch": 2.0513534545898436e-06,
      "step": 2151,
      "training_step_time": 7.5655927658081055
    },
    {
      "epoch": 2.05230712890625e-06,
      "model_forward_time": 1.6973145008087158,
      "step": 2152
    },
    {
      "epoch": 2.05230712890625e-06,
      "step": 2152,
      "training_step_time": 7.546090841293335
    },
    {
      "epoch": 2.0532608032226563e-06,
      "model_forward_time": 1.6982612609863281,
      "step": 2153
    },
    {
      "epoch": 2.0532608032226563e-06,
      "step": 2153,
      "training_step_time": 8.07828664779663
    },
    {
      "epoch": 2.0542144775390624e-06,
      "model_forward_time": 1.7059249877929688,
      "step": 2154
    },
    {
      "epoch": 2.0542144775390624e-06,
      "step": 2154,
      "training_step_time": 7.776607275009155
    },
    {
      "epoch": 2.0551681518554686e-06,
      "model_forward_time": 1.6905930042266846,
      "step": 2155
    },
    {
      "epoch": 2.0551681518554686e-06,
      "step": 2155,
      "training_step_time": 7.312621831893921
    },
    {
      "epoch": 2.056121826171875e-06,
      "model_forward_time": 1.6934092044830322,
      "step": 2156
    },
    {
      "epoch": 2.056121826171875e-06,
      "step": 2156,
      "training_step_time": 8.434422016143799
    },
    {
      "epoch": 2.0570755004882813e-06,
      "model_forward_time": 1.6949646472930908,
      "step": 2157
    },
    {
      "epoch": 2.0570755004882813e-06,
      "step": 2157,
      "training_step_time": 7.789772033691406
    },
    {
      "epoch": 2.0580291748046874e-06,
      "model_forward_time": 1.325911283493042,
      "step": 2158
    },
    {
      "epoch": 2.0580291748046874e-06,
      "step": 2158,
      "training_step_time": 7.835089206695557
    },
    {
      "epoch": 2.0589828491210936e-06,
      "model_forward_time": 1.6930294036865234,
      "step": 2159
    },
    {
      "epoch": 2.0589828491210936e-06,
      "step": 2159,
      "training_step_time": 7.820077657699585
    },
    {
      "epoch": 2.0599365234375e-06,
      "grad_norm": 0.9051061272621155,
      "learning_rate": 4.318000000000001e-06,
      "loss": 0.0967,
      "step": 2160
    },
    {
      "epoch": 2.0599365234375e-06,
      "model_forward_time": 2.2227959632873535,
      "step": 2160
    },
    {
      "dynamics_loss_avg": 0.06843006201088428,
      "epoch": 2.0599365234375e-06,
      "step": 2160
    },
    {
      "action_loss_avg": 0.022866456722840666,
      "epoch": 2.0599365234375e-06,
      "step": 2160
    },
    {
      "epoch": 2.0599365234375e-06,
      "step": 2160,
      "training_step_time": 7.693899154663086
    },
    {
      "epoch": 2.0608901977539063e-06,
      "model_forward_time": 1.7040925025939941,
      "step": 2161
    },
    {
      "epoch": 2.0608901977539063e-06,
      "step": 2161,
      "training_step_time": 8.762456893920898
    },
    {
      "epoch": 2.0618438720703124e-06,
      "model_forward_time": 1.6998882293701172,
      "step": 2162
    },
    {
      "epoch": 2.0618438720703124e-06,
      "step": 2162,
      "training_step_time": 9.042460203170776
    },
    {
      "epoch": 2.0627975463867186e-06,
      "model_forward_time": 1.7037487030029297,
      "step": 2163
    },
    {
      "epoch": 2.0627975463867186e-06,
      "step": 2163,
      "training_step_time": 8.798856735229492
    },
    {
      "epoch": 2.063751220703125e-06,
      "model_forward_time": 1.6951391696929932,
      "step": 2164
    },
    {
      "epoch": 2.063751220703125e-06,
      "step": 2164,
      "training_step_time": 7.740864276885986
    },
    {
      "epoch": 2.0647048950195313e-06,
      "model_forward_time": 1.7061913013458252,
      "step": 2165
    },
    {
      "epoch": 2.0647048950195313e-06,
      "step": 2165,
      "training_step_time": 8.426529884338379
    },
    {
      "epoch": 2.0656585693359374e-06,
      "model_forward_time": 1.6988868713378906,
      "step": 2166
    },
    {
      "epoch": 2.0656585693359374e-06,
      "step": 2166,
      "training_step_time": 9.812926530838013
    },
    {
      "epoch": 2.0666122436523436e-06,
      "model_forward_time": 1.6981914043426514,
      "step": 2167
    },
    {
      "epoch": 2.0666122436523436e-06,
      "step": 2167,
      "training_step_time": 7.785599231719971
    },
    {
      "epoch": 2.06756591796875e-06,
      "model_forward_time": 1.6990411281585693,
      "step": 2168
    },
    {
      "epoch": 2.06756591796875e-06,
      "step": 2168,
      "training_step_time": 7.868952512741089
    },
    {
      "epoch": 2.0685195922851563e-06,
      "model_forward_time": 1.6995329856872559,
      "step": 2169
    },
    {
      "epoch": 2.0685195922851563e-06,
      "step": 2169,
      "training_step_time": 9.524454355239868
    },
    {
      "epoch": 2.0694732666015624e-06,
      "grad_norm": 1.0170928239822388,
      "learning_rate": 4.338000000000001e-06,
      "loss": 0.0943,
      "step": 2170
    },
    {
      "epoch": 2.0694732666015624e-06,
      "model_forward_time": 1.6990351676940918,
      "step": 2170
    },
    {
      "dynamics_loss_avg": 0.07254066839814186,
      "epoch": 2.0694732666015624e-06,
      "step": 2170
    },
    {
      "action_loss_avg": 0.029407746437937022,
      "epoch": 2.0694732666015624e-06,
      "step": 2170
    },
    {
      "epoch": 2.0694732666015624e-06,
      "step": 2170,
      "training_step_time": 7.543058633804321
    },
    {
      "epoch": 2.0704269409179686e-06,
      "model_forward_time": 1.6980171203613281,
      "step": 2171
    },
    {
      "epoch": 2.0704269409179686e-06,
      "step": 2171,
      "training_step_time": 8.657873630523682
    },
    {
      "epoch": 2.071380615234375e-06,
      "model_forward_time": 1.701714277267456,
      "step": 2172
    },
    {
      "epoch": 2.071380615234375e-06,
      "step": 2172,
      "training_step_time": 7.746339559555054
    },
    {
      "epoch": 2.0723342895507813e-06,
      "model_forward_time": 1.7138903141021729,
      "step": 2173
    },
    {
      "epoch": 2.0723342895507813e-06,
      "step": 2173,
      "training_step_time": 7.608894348144531
    },
    {
      "epoch": 2.0732879638671874e-06,
      "model_forward_time": 1.7071759700775146,
      "step": 2174
    },
    {
      "epoch": 2.0732879638671874e-06,
      "step": 2174,
      "training_step_time": 8.417409896850586
    },
    {
      "epoch": 2.0742416381835936e-06,
      "model_forward_time": 1.3460063934326172,
      "step": 2175
    },
    {
      "epoch": 2.0742416381835936e-06,
      "step": 2175,
      "training_step_time": 10.000600814819336
    },
    {
      "epoch": 2.0751953125e-06,
      "model_forward_time": 1.703526496887207,
      "step": 2176
    },
    {
      "epoch": 2.0751953125e-06,
      "step": 2176,
      "training_step_time": 8.577632904052734
    },
    {
      "epoch": 2.0761489868164063e-06,
      "model_forward_time": 1.7055976390838623,
      "step": 2177
    },
    {
      "epoch": 2.0761489868164063e-06,
      "step": 2177,
      "training_step_time": 8.688194036483765
    },
    {
      "epoch": 2.0771026611328124e-06,
      "model_forward_time": 1.3353135585784912,
      "step": 2178
    },
    {
      "epoch": 2.0771026611328124e-06,
      "step": 2178,
      "training_step_time": 7.621989727020264
    },
    {
      "epoch": 2.0780563354492186e-06,
      "model_forward_time": 2.2402610778808594,
      "step": 2179
    },
    {
      "epoch": 2.0780563354492186e-06,
      "step": 2179,
      "training_step_time": 7.491436958312988
    },
    {
      "epoch": 2.079010009765625e-06,
      "grad_norm": 1.1209053993225098,
      "learning_rate": 4.3580000000000005e-06,
      "loss": 0.0954,
      "step": 2180
    },
    {
      "epoch": 2.079010009765625e-06,
      "model_forward_time": 1.7111024856567383,
      "step": 2180
    },
    {
      "dynamics_loss_avg": 0.07181648351252079,
      "epoch": 2.079010009765625e-06,
      "step": 2180
    },
    {
      "action_loss_avg": 0.01685849274508655,
      "epoch": 2.079010009765625e-06,
      "step": 2180
    },
    {
      "epoch": 2.079010009765625e-06,
      "step": 2180,
      "training_step_time": 7.5701868534088135
    },
    {
      "epoch": 2.0799636840820313e-06,
      "model_forward_time": 1.7110776901245117,
      "step": 2181
    },
    {
      "epoch": 2.0799636840820313e-06,
      "step": 2181,
      "training_step_time": 7.714295864105225
    },
    {
      "epoch": 2.0809173583984374e-06,
      "model_forward_time": 1.7199678421020508,
      "step": 2182
    },
    {
      "epoch": 2.0809173583984374e-06,
      "step": 2182,
      "training_step_time": 8.559575319290161
    },
    {
      "epoch": 2.0818710327148436e-06,
      "model_forward_time": 1.7072951793670654,
      "step": 2183
    },
    {
      "epoch": 2.0818710327148436e-06,
      "step": 2183,
      "training_step_time": 7.500351905822754
    },
    {
      "epoch": 2.08282470703125e-06,
      "model_forward_time": 1.7257678508758545,
      "step": 2184
    },
    {
      "epoch": 2.08282470703125e-06,
      "step": 2184,
      "training_step_time": 7.6336870193481445
    },
    {
      "epoch": 2.0837783813476563e-06,
      "model_forward_time": 1.700256586074829,
      "step": 2185
    },
    {
      "epoch": 2.0837783813476563e-06,
      "step": 2185,
      "training_step_time": 7.639940023422241
    },
    {
      "epoch": 2.0847320556640624e-06,
      "model_forward_time": 1.7093286514282227,
      "step": 2186
    },
    {
      "epoch": 2.0847320556640624e-06,
      "step": 2186,
      "training_step_time": 7.744218349456787
    },
    {
      "epoch": 2.0856857299804686e-06,
      "model_forward_time": 1.713313102722168,
      "step": 2187
    },
    {
      "epoch": 2.0856857299804686e-06,
      "step": 2187,
      "training_step_time": 7.540428876876831
    },
    {
      "epoch": 2.086639404296875e-06,
      "model_forward_time": 2.9096295833587646,
      "step": 2188
    },
    {
      "epoch": 2.086639404296875e-06,
      "step": 2188,
      "training_step_time": 7.803143262863159
    },
    {
      "epoch": 2.0875930786132813e-06,
      "model_forward_time": 1.762282371520996,
      "step": 2189
    },
    {
      "epoch": 2.0875930786132813e-06,
      "step": 2189,
      "training_step_time": 8.503530740737915
    },
    {
      "epoch": 2.0885467529296874e-06,
      "grad_norm": 1.2765332460403442,
      "learning_rate": 4.378e-06,
      "loss": 0.1032,
      "step": 2190
    },
    {
      "epoch": 2.0885467529296874e-06,
      "model_forward_time": 1.7645843029022217,
      "step": 2190
    },
    {
      "dynamics_loss_avg": 0.0692599942907691,
      "epoch": 2.0885467529296874e-06,
      "step": 2190
    },
    {
      "action_loss_avg": 0.019203066965565084,
      "epoch": 2.0885467529296874e-06,
      "step": 2190
    },
    {
      "epoch": 2.0885467529296874e-06,
      "step": 2190,
      "training_step_time": 7.748314142227173
    },
    {
      "epoch": 2.0895004272460936e-06,
      "model_forward_time": 1.7946140766143799,
      "step": 2191
    },
    {
      "epoch": 2.0895004272460936e-06,
      "step": 2191,
      "training_step_time": 7.599822998046875
    },
    {
      "epoch": 2.0904541015625e-06,
      "model_forward_time": 1.7583866119384766,
      "step": 2192
    },
    {
      "epoch": 2.0904541015625e-06,
      "step": 2192,
      "training_step_time": 8.14232873916626
    },
    {
      "epoch": 2.0914077758789063e-06,
      "model_forward_time": 1.756434440612793,
      "step": 2193
    },
    {
      "epoch": 2.0914077758789063e-06,
      "step": 2193,
      "training_step_time": 7.564115285873413
    },
    {
      "epoch": 2.0923614501953125e-06,
      "model_forward_time": 1.7579960823059082,
      "step": 2194
    },
    {
      "epoch": 2.0923614501953125e-06,
      "step": 2194,
      "training_step_time": 7.630343914031982
    },
    {
      "epoch": 2.0933151245117186e-06,
      "model_forward_time": 1.7470567226409912,
      "step": 2195
    },
    {
      "epoch": 2.0933151245117186e-06,
      "step": 2195,
      "training_step_time": 7.8021533489227295
    },
    {
      "epoch": 2.094268798828125e-06,
      "model_forward_time": 1.758631944656372,
      "step": 2196
    },
    {
      "epoch": 2.094268798828125e-06,
      "step": 2196,
      "training_step_time": 8.516132593154907
    },
    {
      "epoch": 2.0952224731445313e-06,
      "model_forward_time": 2.8820302486419678,
      "step": 2197
    },
    {
      "epoch": 2.0952224731445313e-06,
      "step": 2197,
      "training_step_time": 8.339269161224365
    },
    {
      "epoch": 2.0961761474609375e-06,
      "model_forward_time": 1.7510297298431396,
      "step": 2198
    },
    {
      "epoch": 2.0961761474609375e-06,
      "step": 2198,
      "training_step_time": 7.708009481430054
    },
    {
      "epoch": 2.0971298217773436e-06,
      "model_forward_time": 1.7581870555877686,
      "step": 2199
    },
    {
      "epoch": 2.0971298217773436e-06,
      "step": 2199,
      "training_step_time": 8.067881107330322
    },
    {
      "epoch": 2.09808349609375e-06,
      "grad_norm": 1.0908769369125366,
      "learning_rate": 4.398000000000001e-06,
      "loss": 0.1082,
      "step": 2200
    },
    {
      "epoch": 2.09808349609375e-06,
      "model_forward_time": 1.7678499221801758,
      "step": 2200
    },
    {
      "dynamics_loss_avg": 0.06890528295189143,
      "epoch": 2.09808349609375e-06,
      "step": 2200
    },
    {
      "action_loss_avg": 0.015821115835569798,
      "epoch": 2.09808349609375e-06,
      "step": 2200
    },
    {
      "epoch": 2.09808349609375e-06,
      "step": 2200,
      "training_step_time": 7.650946617126465
    },
    {
      "epoch": 2.0990371704101563e-06,
      "model_forward_time": 1.756866216659546,
      "step": 2201
    },
    {
      "epoch": 2.0990371704101563e-06,
      "step": 2201,
      "training_step_time": 7.56411600112915
    },
    {
      "epoch": 2.0999908447265625e-06,
      "model_forward_time": 1.7555983066558838,
      "step": 2202
    },
    {
      "epoch": 2.0999908447265625e-06,
      "step": 2202,
      "training_step_time": 9.385132312774658
    },
    {
      "epoch": 2.1009445190429686e-06,
      "model_forward_time": 1.7557051181793213,
      "step": 2203
    },
    {
      "epoch": 2.1009445190429686e-06,
      "step": 2203,
      "training_step_time": 7.695677280426025
    },
    {
      "epoch": 2.101898193359375e-06,
      "model_forward_time": 1.752577304840088,
      "step": 2204
    },
    {
      "epoch": 2.101898193359375e-06,
      "step": 2204,
      "training_step_time": 7.6143622398376465
    },
    {
      "epoch": 2.1028518676757813e-06,
      "model_forward_time": 1.7704570293426514,
      "step": 2205
    },
    {
      "epoch": 2.1028518676757813e-06,
      "step": 2205,
      "training_step_time": 7.578096151351929
    },
    {
      "epoch": 2.1038055419921875e-06,
      "model_forward_time": 1.7494313716888428,
      "step": 2206
    },
    {
      "epoch": 2.1038055419921875e-06,
      "step": 2206,
      "training_step_time": 8.687883138656616
    },
    {
      "epoch": 2.1047592163085936e-06,
      "model_forward_time": 1.7615070343017578,
      "step": 2207
    },
    {
      "epoch": 2.1047592163085936e-06,
      "step": 2207,
      "training_step_time": 8.535101175308228
    },
    {
      "epoch": 2.105712890625e-06,
      "model_forward_time": 1.7644336223602295,
      "step": 2208
    },
    {
      "epoch": 2.105712890625e-06,
      "step": 2208,
      "training_step_time": 7.800759792327881
    },
    {
      "epoch": 2.1066665649414063e-06,
      "model_forward_time": 1.7738139629364014,
      "step": 2209
    },
    {
      "epoch": 2.1066665649414063e-06,
      "step": 2209,
      "training_step_time": 7.712970733642578
    },
    {
      "epoch": 2.1076202392578125e-06,
      "grad_norm": 0.61407071352005,
      "learning_rate": 4.418000000000001e-06,
      "loss": 0.0944,
      "step": 2210
    },
    {
      "epoch": 2.1076202392578125e-06,
      "model_forward_time": 1.8163599967956543,
      "step": 2210
    },
    {
      "dynamics_loss_avg": 0.06931456662714482,
      "epoch": 2.1076202392578125e-06,
      "step": 2210
    },
    {
      "action_loss_avg": 0.024019783455878498,
      "epoch": 2.1076202392578125e-06,
      "step": 2210
    },
    {
      "epoch": 2.1076202392578125e-06,
      "step": 2210,
      "training_step_time": 7.635533094406128
    },
    {
      "epoch": 2.1085739135742186e-06,
      "model_forward_time": 1.7597949504852295,
      "step": 2211
    },
    {
      "epoch": 2.1085739135742186e-06,
      "step": 2211,
      "training_step_time": 8.523617267608643
    },
    {
      "epoch": 2.109527587890625e-06,
      "model_forward_time": 1.7578785419464111,
      "step": 2212
    },
    {
      "epoch": 2.109527587890625e-06,
      "step": 2212,
      "training_step_time": 9.53313946723938
    },
    {
      "epoch": 2.1104812622070313e-06,
      "model_forward_time": 1.7397029399871826,
      "step": 2213
    },
    {
      "epoch": 2.1104812622070313e-06,
      "step": 2213,
      "training_step_time": 8.988474130630493
    },
    {
      "epoch": 2.1114349365234375e-06,
      "model_forward_time": 1.7478744983673096,
      "step": 2214
    },
    {
      "epoch": 2.1114349365234375e-06,
      "step": 2214,
      "training_step_time": 7.684710502624512
    },
    {
      "epoch": 2.1123886108398436e-06,
      "model_forward_time": 1.7533454895019531,
      "step": 2215
    },
    {
      "epoch": 2.1123886108398436e-06,
      "step": 2215,
      "training_step_time": 7.7122626304626465
    },
    {
      "epoch": 2.11334228515625e-06,
      "model_forward_time": 2.5692343711853027,
      "step": 2216
    },
    {
      "epoch": 2.11334228515625e-06,
      "step": 2216,
      "training_step_time": 7.56510853767395
    },
    {
      "epoch": 2.1142959594726563e-06,
      "model_forward_time": 1.7814233303070068,
      "step": 2217
    },
    {
      "epoch": 2.1142959594726563e-06,
      "step": 2217,
      "training_step_time": 7.826121091842651
    },
    {
      "epoch": 2.1152496337890625e-06,
      "model_forward_time": 1.7460541725158691,
      "step": 2218
    },
    {
      "epoch": 2.1152496337890625e-06,
      "step": 2218,
      "training_step_time": 8.66572880744934
    },
    {
      "epoch": 2.1162033081054686e-06,
      "model_forward_time": 1.816007137298584,
      "step": 2219
    },
    {
      "epoch": 2.1162033081054686e-06,
      "step": 2219,
      "training_step_time": 7.529781818389893
    },
    {
      "epoch": 2.117156982421875e-06,
      "grad_norm": 1.0023049116134644,
      "learning_rate": 4.438e-06,
      "loss": 0.0933,
      "step": 2220
    },
    {
      "epoch": 2.117156982421875e-06,
      "model_forward_time": 1.4066271781921387,
      "step": 2220
    },
    {
      "dynamics_loss_avg": 0.09337139539420605,
      "epoch": 2.117156982421875e-06,
      "step": 2220
    },
    {
      "action_loss_avg": 0.0266107308678329,
      "epoch": 2.117156982421875e-06,
      "step": 2220
    },
    {
      "epoch": 2.117156982421875e-06,
      "step": 2220,
      "training_step_time": 7.5534257888793945
    },
    {
      "epoch": 2.1181106567382813e-06,
      "model_forward_time": 1.7623896598815918,
      "step": 2221
    },
    {
      "epoch": 2.1181106567382813e-06,
      "step": 2221,
      "training_step_time": 9.603868246078491
    },
    {
      "epoch": 2.1190643310546875e-06,
      "model_forward_time": 1.7587859630584717,
      "step": 2222
    },
    {
      "epoch": 2.1190643310546875e-06,
      "step": 2222,
      "training_step_time": 8.872128963470459
    },
    {
      "epoch": 2.1200180053710936e-06,
      "model_forward_time": 1.758957862854004,
      "step": 2223
    },
    {
      "epoch": 2.1200180053710936e-06,
      "step": 2223,
      "training_step_time": 7.572413444519043
    },
    {
      "epoch": 2.1209716796875e-06,
      "model_forward_time": 1.7937343120574951,
      "step": 2224
    },
    {
      "epoch": 2.1209716796875e-06,
      "step": 2224,
      "training_step_time": 8.213810920715332
    },
    {
      "epoch": 2.1219253540039063e-06,
      "model_forward_time": 2.8782100677490234,
      "step": 2225
    },
    {
      "epoch": 2.1219253540039063e-06,
      "step": 2225,
      "training_step_time": 8.878312826156616
    },
    {
      "epoch": 2.1228790283203125e-06,
      "model_forward_time": 1.6812145709991455,
      "step": 2226
    },
    {
      "epoch": 2.1228790283203125e-06,
      "step": 2226,
      "training_step_time": 8.477837085723877
    },
    {
      "epoch": 2.1238327026367186e-06,
      "model_forward_time": 1.709212064743042,
      "step": 2227
    },
    {
      "epoch": 2.1238327026367186e-06,
      "step": 2227,
      "training_step_time": 7.580866813659668
    },
    {
      "epoch": 2.124786376953125e-06,
      "model_forward_time": 1.7003540992736816,
      "step": 2228
    },
    {
      "epoch": 2.124786376953125e-06,
      "step": 2228,
      "training_step_time": 7.521532297134399
    },
    {
      "epoch": 2.1257400512695313e-06,
      "model_forward_time": 1.6973977088928223,
      "step": 2229
    },
    {
      "epoch": 2.1257400512695313e-06,
      "step": 2229,
      "training_step_time": 7.520661354064941
    },
    {
      "epoch": 2.1266937255859375e-06,
      "grad_norm": 0.7470768690109253,
      "learning_rate": 4.458e-06,
      "loss": 0.0966,
      "step": 2230
    },
    {
      "epoch": 2.1266937255859375e-06,
      "model_forward_time": 1.7169148921966553,
      "step": 2230
    },
    {
      "dynamics_loss_avg": 0.06473308261483908,
      "epoch": 2.1266937255859375e-06,
      "step": 2230
    },
    {
      "action_loss_avg": 0.016545588010922076,
      "epoch": 2.1266937255859375e-06,
      "step": 2230
    },
    {
      "epoch": 2.1266937255859375e-06,
      "step": 2230,
      "training_step_time": 8.099299907684326
    },
    {
      "epoch": 2.1276473999023436e-06,
      "model_forward_time": 1.7044036388397217,
      "step": 2231
    },
    {
      "epoch": 2.1276473999023436e-06,
      "step": 2231,
      "training_step_time": 7.640249252319336
    },
    {
      "epoch": 2.12860107421875e-06,
      "model_forward_time": 1.6987597942352295,
      "step": 2232
    },
    {
      "epoch": 2.12860107421875e-06,
      "step": 2232,
      "training_step_time": 8.474294662475586
    },
    {
      "epoch": 2.1295547485351563e-06,
      "model_forward_time": 1.6963379383087158,
      "step": 2233
    },
    {
      "epoch": 2.1295547485351563e-06,
      "step": 2233,
      "training_step_time": 7.69331693649292
    },
    {
      "epoch": 2.1305084228515625e-06,
      "model_forward_time": 2.2372961044311523,
      "step": 2234
    },
    {
      "epoch": 2.1305084228515625e-06,
      "step": 2234,
      "training_step_time": 7.680875778198242
    },
    {
      "epoch": 2.1314620971679686e-06,
      "model_forward_time": 1.7037169933319092,
      "step": 2235
    },
    {
      "epoch": 2.1314620971679686e-06,
      "step": 2235,
      "training_step_time": 8.42949628829956
    },
    {
      "epoch": 2.132415771484375e-06,
      "model_forward_time": 1.6992816925048828,
      "step": 2236
    },
    {
      "epoch": 2.132415771484375e-06,
      "step": 2236,
      "training_step_time": 7.535857677459717
    },
    {
      "epoch": 2.1333694458007813e-06,
      "model_forward_time": 1.6913599967956543,
      "step": 2237
    },
    {
      "epoch": 2.1333694458007813e-06,
      "step": 2237,
      "training_step_time": 8.637500524520874
    },
    {
      "epoch": 2.1343231201171875e-06,
      "model_forward_time": 1.6986281871795654,
      "step": 2238
    },
    {
      "epoch": 2.1343231201171875e-06,
      "step": 2238,
      "training_step_time": 7.52474308013916
    },
    {
      "epoch": 2.1352767944335936e-06,
      "model_forward_time": 1.7162833213806152,
      "step": 2239
    },
    {
      "epoch": 2.1352767944335936e-06,
      "step": 2239,
      "training_step_time": 7.6106507778167725
    },
    {
      "epoch": 2.13623046875e-06,
      "grad_norm": 0.8600204586982727,
      "learning_rate": 4.478e-06,
      "loss": 0.092,
      "step": 2240
    },
    {
      "epoch": 2.13623046875e-06,
      "model_forward_time": 1.6951210498809814,
      "step": 2240
    },
    {
      "dynamics_loss_avg": 0.0613824862986803,
      "epoch": 2.13623046875e-06,
      "step": 2240
    },
    {
      "action_loss_avg": 0.016384853282943368,
      "epoch": 2.13623046875e-06,
      "step": 2240
    },
    {
      "epoch": 2.13623046875e-06,
      "step": 2240,
      "training_step_time": 8.654149532318115
    },
    {
      "epoch": 2.1371841430664063e-06,
      "model_forward_time": 1.6993458271026611,
      "step": 2241
    },
    {
      "epoch": 2.1371841430664063e-06,
      "step": 2241,
      "training_step_time": 7.6089842319488525
    },
    {
      "epoch": 2.1381378173828125e-06,
      "model_forward_time": 1.7023322582244873,
      "step": 2242
    },
    {
      "epoch": 2.1381378173828125e-06,
      "step": 2242,
      "training_step_time": 7.551892042160034
    },
    {
      "epoch": 2.1390914916992186e-06,
      "model_forward_time": 1.7015347480773926,
      "step": 2243
    },
    {
      "epoch": 2.1390914916992186e-06,
      "step": 2243,
      "training_step_time": 8.632198095321655
    },
    {
      "epoch": 2.140045166015625e-06,
      "model_forward_time": 1.7008867263793945,
      "step": 2244
    },
    {
      "epoch": 2.140045166015625e-06,
      "step": 2244,
      "training_step_time": 7.514513969421387
    },
    {
      "epoch": 2.1409988403320314e-06,
      "model_forward_time": 1.698559284210205,
      "step": 2245
    },
    {
      "epoch": 2.1409988403320314e-06,
      "step": 2245,
      "training_step_time": 7.519185543060303
    },
    {
      "epoch": 2.1419525146484375e-06,
      "model_forward_time": 1.716200590133667,
      "step": 2246
    },
    {
      "epoch": 2.1419525146484375e-06,
      "step": 2246,
      "training_step_time": 8.702370882034302
    },
    {
      "epoch": 2.1429061889648436e-06,
      "model_forward_time": 1.7020769119262695,
      "step": 2247
    },
    {
      "epoch": 2.1429061889648436e-06,
      "step": 2247,
      "training_step_time": 7.5770182609558105
    },
    {
      "epoch": 2.14385986328125e-06,
      "model_forward_time": 1.7050673961639404,
      "step": 2248
    },
    {
      "epoch": 2.14385986328125e-06,
      "step": 2248,
      "training_step_time": 7.601752996444702
    },
    {
      "epoch": 2.1448135375976564e-06,
      "model_forward_time": 1.694932222366333,
      "step": 2249
    },
    {
      "epoch": 2.1448135375976564e-06,
      "step": 2249,
      "training_step_time": 7.632952451705933
    },
    {
      "epoch": 2.1457672119140625e-06,
      "grad_norm": 0.7408762574195862,
      "learning_rate": 4.498e-06,
      "loss": 0.097,
      "step": 2250
    },
    {
      "epoch": 2.1457672119140625e-06,
      "model_forward_time": 1.6931593418121338,
      "step": 2250
    },
    {
      "dynamics_loss_avg": 0.08050927929580212,
      "epoch": 2.1457672119140625e-06,
      "step": 2250
    },
    {
      "action_loss_avg": 0.01402065074071288,
      "epoch": 2.1457672119140625e-06,
      "step": 2250
    },
    {
      "epoch": 2.1457672119140625e-06,
      "step": 2250,
      "training_step_time": 10.056137800216675
    },
    {
      "epoch": 2.1467208862304686e-06,
      "model_forward_time": 1.6979851722717285,
      "step": 2251
    },
    {
      "epoch": 2.1467208862304686e-06,
      "step": 2251,
      "training_step_time": 7.70944619178772
    },
    {
      "epoch": 2.147674560546875e-06,
      "model_forward_time": 1.6962363719940186,
      "step": 2252
    },
    {
      "epoch": 2.147674560546875e-06,
      "step": 2252,
      "training_step_time": 7.582386016845703
    },
    {
      "epoch": 2.1486282348632814e-06,
      "model_forward_time": 2.229295492172241,
      "step": 2253
    },
    {
      "epoch": 2.1486282348632814e-06,
      "step": 2253,
      "training_step_time": 8.153476476669312
    },
    {
      "epoch": 2.1495819091796875e-06,
      "model_forward_time": 1.696382999420166,
      "step": 2254
    },
    {
      "epoch": 2.1495819091796875e-06,
      "step": 2254,
      "training_step_time": 7.467446804046631
    },
    {
      "epoch": 2.1505355834960936e-06,
      "model_forward_time": 1.7210354804992676,
      "step": 2255
    },
    {
      "epoch": 2.1505355834960936e-06,
      "step": 2255,
      "training_step_time": 7.627664566040039
    },
    {
      "epoch": 2.1514892578125e-06,
      "model_forward_time": 1.6883139610290527,
      "step": 2256
    },
    {
      "epoch": 2.1514892578125e-06,
      "step": 2256,
      "training_step_time": 7.583941698074341
    },
    {
      "epoch": 2.1524429321289064e-06,
      "model_forward_time": 1.705977439880371,
      "step": 2257
    },
    {
      "epoch": 2.1524429321289064e-06,
      "step": 2257,
      "training_step_time": 7.652299642562866
    },
    {
      "epoch": 2.1533966064453125e-06,
      "model_forward_time": 1.6969807147979736,
      "step": 2258
    },
    {
      "epoch": 2.1533966064453125e-06,
      "step": 2258,
      "training_step_time": 8.617209672927856
    },
    {
      "epoch": 2.1543502807617187e-06,
      "model_forward_time": 1.6963729858398438,
      "step": 2259
    },
    {
      "epoch": 2.1543502807617187e-06,
      "step": 2259,
      "training_step_time": 9.335065603256226
    },
    {
      "epoch": 2.155303955078125e-06,
      "grad_norm": 0.7744852900505066,
      "learning_rate": 4.518e-06,
      "loss": 0.1012,
      "step": 2260
    },
    {
      "epoch": 2.155303955078125e-06,
      "model_forward_time": 1.7004830837249756,
      "step": 2260
    },
    {
      "dynamics_loss_avg": 0.06740138921886682,
      "epoch": 2.155303955078125e-06,
      "step": 2260
    },
    {
      "action_loss_avg": 0.014989531831815839,
      "epoch": 2.155303955078125e-06,
      "step": 2260
    },
    {
      "epoch": 2.155303955078125e-06,
      "step": 2260,
      "training_step_time": 7.644894123077393
    },
    {
      "epoch": 2.1562576293945314e-06,
      "model_forward_time": 1.697317123413086,
      "step": 2261
    },
    {
      "epoch": 2.1562576293945314e-06,
      "step": 2261,
      "training_step_time": 7.530215740203857
    },
    {
      "epoch": 2.1572113037109375e-06,
      "model_forward_time": 2.234063148498535,
      "step": 2262
    },
    {
      "epoch": 2.1572113037109375e-06,
      "step": 2262,
      "training_step_time": 8.450401782989502
    },
    {
      "epoch": 2.1581649780273437e-06,
      "model_forward_time": 1.7015275955200195,
      "step": 2263
    },
    {
      "epoch": 2.1581649780273437e-06,
      "step": 2263,
      "training_step_time": 7.6458563804626465
    },
    {
      "epoch": 2.15911865234375e-06,
      "model_forward_time": 1.7166540622711182,
      "step": 2264
    },
    {
      "epoch": 2.15911865234375e-06,
      "step": 2264,
      "training_step_time": 8.625306129455566
    },
    {
      "epoch": 2.1600723266601564e-06,
      "model_forward_time": 1.700671672821045,
      "step": 2265
    },
    {
      "epoch": 2.1600723266601564e-06,
      "step": 2265,
      "training_step_time": 7.613373279571533
    },
    {
      "epoch": 2.1610260009765625e-06,
      "model_forward_time": 1.6962573528289795,
      "step": 2266
    },
    {
      "epoch": 2.1610260009765625e-06,
      "step": 2266,
      "training_step_time": 8.273061037063599
    },
    {
      "epoch": 2.1619796752929687e-06,
      "model_forward_time": 1.7044415473937988,
      "step": 2267
    },
    {
      "epoch": 2.1619796752929687e-06,
      "step": 2267,
      "training_step_time": 9.091230869293213
    },
    {
      "epoch": 2.162933349609375e-06,
      "model_forward_time": 1.7022643089294434,
      "step": 2268
    },
    {
      "epoch": 2.162933349609375e-06,
      "step": 2268,
      "training_step_time": 8.298171043395996
    },
    {
      "epoch": 2.1638870239257814e-06,
      "model_forward_time": 1.706362247467041,
      "step": 2269
    },
    {
      "epoch": 2.1638870239257814e-06,
      "step": 2269,
      "training_step_time": 6.680307626724243
    },
    {
      "epoch": 2.1648406982421875e-06,
      "grad_norm": 0.9233662486076355,
      "learning_rate": 4.5380000000000004e-06,
      "loss": 0.0934,
      "step": 2270
    },
    {
      "epoch": 2.1648406982421875e-06,
      "model_forward_time": 1.694460391998291,
      "step": 2270
    },
    {
      "dynamics_loss_avg": 0.07114879060536623,
      "epoch": 2.1648406982421875e-06,
      "step": 2270
    },
    {
      "action_loss_avg": 0.017290712846443058,
      "epoch": 2.1648406982421875e-06,
      "step": 2270
    },
    {
      "epoch": 2.1648406982421875e-06,
      "step": 2270,
      "training_step_time": 6.656255483627319
    },
    {
      "epoch": 2.1657943725585937e-06,
      "model_forward_time": 1.7012038230895996,
      "step": 2271
    },
    {
      "epoch": 2.1657943725585937e-06,
      "step": 2271,
      "training_step_time": 6.671896934509277
    },
    {
      "epoch": 2.166748046875e-06,
      "model_forward_time": 1.7082583904266357,
      "step": 2272
    },
    {
      "epoch": 2.166748046875e-06,
      "step": 2272,
      "training_step_time": 6.699005126953125
    },
    {
      "epoch": 2.1677017211914064e-06,
      "model_forward_time": 1.7015743255615234,
      "step": 2273
    },
    {
      "epoch": 2.1677017211914064e-06,
      "step": 2273,
      "training_step_time": 6.685292720794678
    },
    {
      "epoch": 2.1686553955078125e-06,
      "model_forward_time": 1.704458236694336,
      "step": 2274
    },
    {
      "epoch": 2.1686553955078125e-06,
      "step": 2274,
      "training_step_time": 6.689416170120239
    },
    {
      "epoch": 2.1696090698242187e-06,
      "model_forward_time": 1.704448938369751,
      "step": 2275
    },
    {
      "epoch": 2.1696090698242187e-06,
      "step": 2275,
      "training_step_time": 6.68651819229126
    },
    {
      "epoch": 2.170562744140625e-06,
      "model_forward_time": 1.7017171382904053,
      "step": 2276
    },
    {
      "epoch": 2.170562744140625e-06,
      "step": 2276,
      "training_step_time": 6.6530468463897705
    },
    {
      "epoch": 2.1715164184570314e-06,
      "model_forward_time": 2.2428102493286133,
      "step": 2277
    },
    {
      "epoch": 2.1715164184570314e-06,
      "step": 2277,
      "training_step_time": 7.725332260131836
    },
    {
      "epoch": 2.1724700927734375e-06,
      "model_forward_time": 1.6968610286712646,
      "step": 2278
    },
    {
      "epoch": 2.1724700927734375e-06,
      "step": 2278,
      "training_step_time": 7.702809572219849
    },
    {
      "epoch": 2.1734237670898437e-06,
      "model_forward_time": 1.702650785446167,
      "step": 2279
    },
    {
      "epoch": 2.1734237670898437e-06,
      "step": 2279,
      "training_step_time": 6.7283034324646
    },
    {
      "epoch": 2.17437744140625e-06,
      "grad_norm": 1.3957021236419678,
      "learning_rate": 4.558e-06,
      "loss": 0.0923,
      "step": 2280
    },
    {
      "epoch": 2.17437744140625e-06,
      "model_forward_time": 0.981738805770874,
      "step": 2280
    },
    {
      "dynamics_loss_avg": 0.06113290805369616,
      "epoch": 2.17437744140625e-06,
      "step": 2280
    },
    {
      "action_loss_avg": 0.014878050284460187,
      "epoch": 2.17437744140625e-06,
      "step": 2280
    },
    {
      "epoch": 2.17437744140625e-06,
      "step": 2280,
      "training_step_time": 6.688253402709961
    },
    {
      "epoch": 2.1753311157226564e-06,
      "model_forward_time": 1.699354648590088,
      "step": 2281
    },
    {
      "epoch": 2.1753311157226564e-06,
      "step": 2281,
      "training_step_time": 6.6844322681427
    },
    {
      "epoch": 2.1762847900390625e-06,
      "model_forward_time": 1.70343017578125,
      "step": 2282
    },
    {
      "epoch": 2.1762847900390625e-06,
      "step": 2282,
      "training_step_time": 6.685734510421753
    },
    {
      "epoch": 2.1772384643554687e-06,
      "model_forward_time": 1.6796488761901855,
      "step": 2283
    },
    {
      "epoch": 2.1772384643554687e-06,
      "step": 2283,
      "training_step_time": 6.6899943351745605
    },
    {
      "epoch": 2.178192138671875e-06,
      "model_forward_time": 1.6959302425384521,
      "step": 2284
    },
    {
      "epoch": 2.178192138671875e-06,
      "step": 2284,
      "training_step_time": 6.665664911270142
    },
    {
      "epoch": 2.1791458129882814e-06,
      "model_forward_time": 0.9844470024108887,
      "step": 2285
    },
    {
      "epoch": 2.1791458129882814e-06,
      "step": 2285,
      "training_step_time": 6.692981481552124
    },
    {
      "epoch": 2.1800994873046875e-06,
      "model_forward_time": 1.3365180492401123,
      "step": 2286
    },
    {
      "epoch": 2.1800994873046875e-06,
      "step": 2286,
      "training_step_time": 11.69659948348999
    },
    {
      "epoch": 2.1810531616210937e-06,
      "model_forward_time": 2.2116622924804688,
      "step": 2287
    },
    {
      "epoch": 2.1810531616210937e-06,
      "step": 2287,
      "training_step_time": 7.738968372344971
    },
    {
      "epoch": 2.1820068359375e-06,
      "model_forward_time": 1.3258659839630127,
      "step": 2288
    },
    {
      "epoch": 2.1820068359375e-06,
      "step": 2288,
      "training_step_time": 6.639309406280518
    },
    {
      "epoch": 2.1829605102539064e-06,
      "model_forward_time": 1.7016055583953857,
      "step": 2289
    },
    {
      "epoch": 2.1829605102539064e-06,
      "step": 2289,
      "training_step_time": 6.678015470504761
    },
    {
      "epoch": 2.1839141845703125e-06,
      "grad_norm": 0.8409633636474609,
      "learning_rate": 4.578e-06,
      "loss": 0.091,
      "step": 2290
    },
    {
      "epoch": 2.1839141845703125e-06,
      "model_forward_time": 1.6981940269470215,
      "step": 2290
    },
    {
      "dynamics_loss_avg": 0.08229154646396637,
      "epoch": 2.1839141845703125e-06,
      "step": 2290
    },
    {
      "action_loss_avg": 0.021344706043601037,
      "epoch": 2.1839141845703125e-06,
      "step": 2290
    },
    {
      "epoch": 2.1839141845703125e-06,
      "step": 2290,
      "training_step_time": 6.6798834800720215
    },
    {
      "epoch": 2.1848678588867187e-06,
      "model_forward_time": 1.697754144668579,
      "step": 2291
    },
    {
      "epoch": 2.1848678588867187e-06,
      "step": 2291,
      "training_step_time": 6.651883602142334
    },
    {
      "epoch": 2.185821533203125e-06,
      "model_forward_time": 1.6913342475891113,
      "step": 2292
    },
    {
      "epoch": 2.185821533203125e-06,
      "step": 2292,
      "training_step_time": 6.664705276489258
    },
    {
      "epoch": 2.1867752075195314e-06,
      "model_forward_time": 1.6997036933898926,
      "step": 2293
    },
    {
      "epoch": 2.1867752075195314e-06,
      "step": 2293,
      "training_step_time": 6.736725807189941
    },
    {
      "epoch": 2.1877288818359375e-06,
      "model_forward_time": 1.6986687183380127,
      "step": 2294
    },
    {
      "epoch": 2.1877288818359375e-06,
      "step": 2294,
      "training_step_time": 6.708811283111572
    },
    {
      "epoch": 2.1886825561523437e-06,
      "model_forward_time": 1.7102882862091064,
      "step": 2295
    },
    {
      "epoch": 2.1886825561523437e-06,
      "step": 2295,
      "training_step_time": 11.396700143814087
    },
    {
      "epoch": 2.18963623046875e-06,
      "model_forward_time": 2.2184269428253174,
      "step": 2296
    },
    {
      "epoch": 2.18963623046875e-06,
      "step": 2296,
      "training_step_time": 7.783223628997803
    },
    {
      "epoch": 2.1905899047851564e-06,
      "model_forward_time": 1.699430227279663,
      "step": 2297
    },
    {
      "epoch": 2.1905899047851564e-06,
      "step": 2297,
      "training_step_time": 6.670883655548096
    },
    {
      "epoch": 2.1915435791015625e-06,
      "model_forward_time": 1.699592113494873,
      "step": 2298
    },
    {
      "epoch": 2.1915435791015625e-06,
      "step": 2298,
      "training_step_time": 6.6996049880981445
    },
    {
      "epoch": 2.1924972534179687e-06,
      "model_forward_time": 1.7010021209716797,
      "step": 2299
    },
    {
      "epoch": 2.1924972534179687e-06,
      "step": 2299,
      "training_step_time": 6.6439690589904785
    },
    {
      "epoch": 2.193450927734375e-06,
      "grad_norm": 1.1689757108688354,
      "learning_rate": 4.598e-06,
      "loss": 0.0895,
      "step": 2300
    },
    {
      "epoch": 2.193450927734375e-06,
      "model_forward_time": 1.688654899597168,
      "step": 2300
    },
    {
      "dynamics_loss_avg": 0.05894334316253662,
      "epoch": 2.193450927734375e-06,
      "step": 2300
    },
    {
      "action_loss_avg": 0.01521861795336008,
      "epoch": 2.193450927734375e-06,
      "step": 2300
    },
    {
      "epoch": 2.193450927734375e-06,
      "step": 2300,
      "training_step_time": 6.652124881744385
    },
    {
      "epoch": 2.1944046020507814e-06,
      "model_forward_time": 1.6979334354400635,
      "step": 2301
    },
    {
      "epoch": 2.1944046020507814e-06,
      "step": 2301,
      "training_step_time": 6.6204445362091064
    },
    {
      "epoch": 2.1953582763671875e-06,
      "model_forward_time": 1.6975362300872803,
      "step": 2302
    },
    {
      "epoch": 2.1953582763671875e-06,
      "step": 2302,
      "training_step_time": 6.677781820297241
    },
    {
      "epoch": 2.1963119506835937e-06,
      "model_forward_time": 1.7039077281951904,
      "step": 2303
    },
    {
      "epoch": 2.1963119506835937e-06,
      "step": 2303,
      "training_step_time": 6.6429362297058105
    },
    {
      "epoch": 2.197265625e-06,
      "model_forward_time": 1.7042045593261719,
      "step": 2304
    },
    {
      "epoch": 2.197265625e-06,
      "step": 2304,
      "training_step_time": 10.134063243865967
    },
    {
      "epoch": 2.1982192993164064e-06,
      "model_forward_time": 2.2371456623077393,
      "step": 2305
    },
    {
      "epoch": 2.1982192993164064e-06,
      "step": 2305,
      "training_step_time": 7.7447357177734375
    },
    {
      "epoch": 2.1991729736328125e-06,
      "model_forward_time": 1.702709436416626,
      "step": 2306
    },
    {
      "epoch": 2.1991729736328125e-06,
      "step": 2306,
      "training_step_time": 7.445881605148315
    },
    {
      "epoch": 2.2001266479492187e-06,
      "model_forward_time": 1.6937429904937744,
      "step": 2307
    },
    {
      "epoch": 2.2001266479492187e-06,
      "step": 2307,
      "training_step_time": 6.649582386016846
    },
    {
      "epoch": 2.201080322265625e-06,
      "model_forward_time": 1.6901490688323975,
      "step": 2308
    },
    {
      "epoch": 2.201080322265625e-06,
      "step": 2308,
      "training_step_time": 6.666254043579102
    },
    {
      "epoch": 2.2020339965820314e-06,
      "model_forward_time": 1.6997127532958984,
      "step": 2309
    },
    {
      "epoch": 2.2020339965820314e-06,
      "step": 2309,
      "training_step_time": 6.648587703704834
    },
    {
      "epoch": 2.2029876708984376e-06,
      "grad_norm": 1.4089045524597168,
      "learning_rate": 4.6180000000000005e-06,
      "loss": 0.0977,
      "step": 2310
    },
    {
      "epoch": 2.2029876708984376e-06,
      "model_forward_time": 1.687082290649414,
      "step": 2310
    },
    {
      "dynamics_loss_avg": 0.06390250883996487,
      "epoch": 2.2029876708984376e-06,
      "step": 2310
    },
    {
      "action_loss_avg": 0.018775473302230238,
      "epoch": 2.2029876708984376e-06,
      "step": 2310
    },
    {
      "epoch": 2.2029876708984376e-06,
      "step": 2310,
      "training_step_time": 6.723396301269531
    },
    {
      "epoch": 2.2039413452148437e-06,
      "model_forward_time": 1.696549654006958,
      "step": 2311
    },
    {
      "epoch": 2.2039413452148437e-06,
      "step": 2311,
      "training_step_time": 6.7746570110321045
    },
    {
      "epoch": 2.20489501953125e-06,
      "model_forward_time": 1.700366735458374,
      "step": 2312
    },
    {
      "epoch": 2.20489501953125e-06,
      "step": 2312,
      "training_step_time": 6.692917346954346
    },
    {
      "epoch": 2.2058486938476564e-06,
      "model_forward_time": 1.700289249420166,
      "step": 2313
    },
    {
      "epoch": 2.2058486938476564e-06,
      "step": 2313,
      "training_step_time": 8.474185705184937
    },
    {
      "epoch": 2.2068023681640626e-06,
      "model_forward_time": 2.264765501022339,
      "step": 2314
    },
    {
      "epoch": 2.2068023681640626e-06,
      "step": 2314,
      "training_step_time": 7.705424547195435
    },
    {
      "epoch": 2.2077560424804687e-06,
      "model_forward_time": 1.6978585720062256,
      "step": 2315
    },
    {
      "epoch": 2.2077560424804687e-06,
      "step": 2315,
      "training_step_time": 7.5930259227752686
    },
    {
      "epoch": 2.208709716796875e-06,
      "model_forward_time": 1.695744276046753,
      "step": 2316
    },
    {
      "epoch": 2.208709716796875e-06,
      "step": 2316,
      "training_step_time": 6.799237966537476
    },
    {
      "epoch": 2.2096633911132814e-06,
      "model_forward_time": 1.6998419761657715,
      "step": 2317
    },
    {
      "epoch": 2.2096633911132814e-06,
      "step": 2317,
      "training_step_time": 6.78218150138855
    },
    {
      "epoch": 2.2106170654296876e-06,
      "model_forward_time": 1.6977019309997559,
      "step": 2318
    },
    {
      "epoch": 2.2106170654296876e-06,
      "step": 2318,
      "training_step_time": 6.819195985794067
    },
    {
      "epoch": 2.2115707397460937e-06,
      "model_forward_time": 1.7001924514770508,
      "step": 2319
    },
    {
      "epoch": 2.2115707397460937e-06,
      "step": 2319,
      "training_step_time": 6.827911853790283
    },
    {
      "epoch": 2.2125244140625e-06,
      "grad_norm": 1.117006778717041,
      "learning_rate": 4.638e-06,
      "loss": 0.0965,
      "step": 2320
    },
    {
      "epoch": 2.2125244140625e-06,
      "model_forward_time": 1.6942667961120605,
      "step": 2320
    },
    {
      "dynamics_loss_avg": 0.06191593986004591,
      "epoch": 2.2125244140625e-06,
      "step": 2320
    },
    {
      "action_loss_avg": 0.02053746595047414,
      "epoch": 2.2125244140625e-06,
      "step": 2320
    },
    {
      "epoch": 2.2125244140625e-06,
      "step": 2320,
      "training_step_time": 6.744811058044434
    },
    {
      "epoch": 2.2134780883789064e-06,
      "model_forward_time": 1.6975891590118408,
      "step": 2321
    },
    {
      "epoch": 2.2134780883789064e-06,
      "step": 2321,
      "training_step_time": 6.754526853561401
    },
    {
      "epoch": 2.2144317626953126e-06,
      "model_forward_time": 1.6973681449890137,
      "step": 2322
    },
    {
      "epoch": 2.2144317626953126e-06,
      "step": 2322,
      "training_step_time": 7.48306941986084
    },
    {
      "epoch": 2.2153854370117187e-06,
      "model_forward_time": 2.2523550987243652,
      "step": 2323
    },
    {
      "epoch": 2.2153854370117187e-06,
      "step": 2323,
      "training_step_time": 9.059237957000732
    },
    {
      "epoch": 2.216339111328125e-06,
      "model_forward_time": 1.697995901107788,
      "step": 2324
    },
    {
      "epoch": 2.216339111328125e-06,
      "step": 2324,
      "training_step_time": 7.757984638214111
    },
    {
      "epoch": 2.2172927856445314e-06,
      "model_forward_time": 1.698328971862793,
      "step": 2325
    },
    {
      "epoch": 2.2172927856445314e-06,
      "step": 2325,
      "training_step_time": 7.4807515144348145
    },
    {
      "epoch": 2.2182464599609376e-06,
      "model_forward_time": 1.704272985458374,
      "step": 2326
    },
    {
      "epoch": 2.2182464599609376e-06,
      "step": 2326,
      "training_step_time": 6.760544538497925
    },
    {
      "epoch": 2.2192001342773437e-06,
      "model_forward_time": 1.7011983394622803,
      "step": 2327
    },
    {
      "epoch": 2.2192001342773437e-06,
      "step": 2327,
      "training_step_time": 6.7386462688446045
    },
    {
      "epoch": 2.22015380859375e-06,
      "model_forward_time": 1.6969640254974365,
      "step": 2328
    },
    {
      "epoch": 2.22015380859375e-06,
      "step": 2328,
      "training_step_time": 6.764706373214722
    },
    {
      "epoch": 2.2211074829101564e-06,
      "model_forward_time": 1.6933515071868896,
      "step": 2329
    },
    {
      "epoch": 2.2211074829101564e-06,
      "step": 2329,
      "training_step_time": 6.7905402183532715
    },
    {
      "epoch": 2.2220611572265626e-06,
      "grad_norm": 0.8246415257453918,
      "learning_rate": 4.658e-06,
      "loss": 0.0911,
      "step": 2330
    },
    {
      "epoch": 2.2220611572265626e-06,
      "model_forward_time": 1.6974592208862305,
      "step": 2330
    },
    {
      "dynamics_loss_avg": 0.07564187049865723,
      "epoch": 2.2220611572265626e-06,
      "step": 2330
    },
    {
      "action_loss_avg": 0.015006580017507076,
      "epoch": 2.2220611572265626e-06,
      "step": 2330
    },
    {
      "epoch": 2.2220611572265626e-06,
      "step": 2330,
      "training_step_time": 6.762501001358032
    },
    {
      "epoch": 2.2230148315429687e-06,
      "model_forward_time": 1.7012665271759033,
      "step": 2331
    },
    {
      "epoch": 2.2230148315429687e-06,
      "step": 2331,
      "training_step_time": 7.420962333679199
    },
    {
      "epoch": 2.223968505859375e-06,
      "model_forward_time": 2.2307658195495605,
      "step": 2332
    },
    {
      "epoch": 2.223968505859375e-06,
      "step": 2332,
      "training_step_time": 8.885601997375488
    },
    {
      "epoch": 2.2249221801757814e-06,
      "model_forward_time": 1.6862053871154785,
      "step": 2333
    },
    {
      "epoch": 2.2249221801757814e-06,
      "step": 2333,
      "training_step_time": 7.7710371017456055
    },
    {
      "epoch": 2.2258758544921876e-06,
      "model_forward_time": 1.7085375785827637,
      "step": 2334
    },
    {
      "epoch": 2.2258758544921876e-06,
      "step": 2334,
      "training_step_time": 7.4718804359436035
    },
    {
      "epoch": 2.2268295288085937e-06,
      "model_forward_time": 1.710754632949829,
      "step": 2335
    },
    {
      "epoch": 2.2268295288085937e-06,
      "step": 2335,
      "training_step_time": 6.719618082046509
    },
    {
      "epoch": 2.227783203125e-06,
      "model_forward_time": 1.3321356773376465,
      "step": 2336
    },
    {
      "epoch": 2.227783203125e-06,
      "step": 2336,
      "training_step_time": 6.707974910736084
    },
    {
      "epoch": 2.2287368774414064e-06,
      "model_forward_time": 1.7153925895690918,
      "step": 2337
    },
    {
      "epoch": 2.2287368774414064e-06,
      "step": 2337,
      "training_step_time": 6.753990650177002
    },
    {
      "epoch": 2.2296905517578126e-06,
      "model_forward_time": 1.7181522846221924,
      "step": 2338
    },
    {
      "epoch": 2.2296905517578126e-06,
      "step": 2338,
      "training_step_time": 6.761300325393677
    },
    {
      "epoch": 2.2306442260742187e-06,
      "model_forward_time": 1.7149171829223633,
      "step": 2339
    },
    {
      "epoch": 2.2306442260742187e-06,
      "step": 2339,
      "training_step_time": 6.70053243637085
    },
    {
      "epoch": 2.231597900390625e-06,
      "grad_norm": 0.9005067348480225,
      "learning_rate": 4.678e-06,
      "loss": 0.0923,
      "step": 2340
    },
    {
      "epoch": 2.231597900390625e-06,
      "model_forward_time": 1.7066116333007812,
      "step": 2340
    },
    {
      "dynamics_loss_avg": 0.08675014749169349,
      "epoch": 2.231597900390625e-06,
      "step": 2340
    },
    {
      "action_loss_avg": 0.021616704855114223,
      "epoch": 2.231597900390625e-06,
      "step": 2340
    },
    {
      "epoch": 2.231597900390625e-06,
      "step": 2340,
      "training_step_time": 7.466123342514038
    },
    {
      "epoch": 2.2325515747070314e-06,
      "model_forward_time": 1.721315860748291,
      "step": 2341
    },
    {
      "epoch": 2.2325515747070314e-06,
      "step": 2341,
      "training_step_time": 10.148537158966064
    },
    {
      "epoch": 2.2335052490234376e-06,
      "model_forward_time": 2.262859344482422,
      "step": 2342
    },
    {
      "epoch": 2.2335052490234376e-06,
      "step": 2342,
      "training_step_time": 7.768521308898926
    },
    {
      "epoch": 2.2344589233398437e-06,
      "model_forward_time": 1.7068476676940918,
      "step": 2343
    },
    {
      "epoch": 2.2344589233398437e-06,
      "step": 2343,
      "training_step_time": 7.473788022994995
    },
    {
      "epoch": 2.23541259765625e-06,
      "model_forward_time": 1.7098686695098877,
      "step": 2344
    },
    {
      "epoch": 2.23541259765625e-06,
      "step": 2344,
      "training_step_time": 6.674938678741455
    },
    {
      "epoch": 2.2363662719726564e-06,
      "model_forward_time": 1.7212474346160889,
      "step": 2345
    },
    {
      "epoch": 2.2363662719726564e-06,
      "step": 2345,
      "training_step_time": 6.745998382568359
    },
    {
      "epoch": 2.2373199462890626e-06,
      "model_forward_time": 1.715238094329834,
      "step": 2346
    },
    {
      "epoch": 2.2373199462890626e-06,
      "step": 2346,
      "training_step_time": 6.708019971847534
    },
    {
      "epoch": 2.2382736206054687e-06,
      "model_forward_time": 1.703354835510254,
      "step": 2347
    },
    {
      "epoch": 2.2382736206054687e-06,
      "step": 2347,
      "training_step_time": 6.709009885787964
    },
    {
      "epoch": 2.239227294921875e-06,
      "model_forward_time": 1.692744255065918,
      "step": 2348
    },
    {
      "epoch": 2.239227294921875e-06,
      "step": 2348,
      "training_step_time": 6.780796051025391
    },
    {
      "epoch": 2.2401809692382814e-06,
      "model_forward_time": 1.6983838081359863,
      "step": 2349
    },
    {
      "epoch": 2.2401809692382814e-06,
      "step": 2349,
      "training_step_time": 7.504781484603882
    },
    {
      "epoch": 2.2411346435546876e-06,
      "grad_norm": 0.664935827255249,
      "learning_rate": 4.698000000000001e-06,
      "loss": 0.0987,
      "step": 2350
    },
    {
      "epoch": 2.2411346435546876e-06,
      "model_forward_time": 1.3226559162139893,
      "step": 2350
    },
    {
      "dynamics_loss_avg": 0.056525101326406,
      "epoch": 2.2411346435546876e-06,
      "step": 2350
    },
    {
      "action_loss_avg": 0.011274210410192609,
      "epoch": 2.2411346435546876e-06,
      "step": 2350
    },
    {
      "epoch": 2.2411346435546876e-06,
      "step": 2350,
      "training_step_time": 8.729157447814941
    },
    {
      "epoch": 2.2420883178710937e-06,
      "model_forward_time": 2.229641914367676,
      "step": 2351
    },
    {
      "epoch": 2.2420883178710937e-06,
      "step": 2351,
      "training_step_time": 8.32664680480957
    },
    {
      "epoch": 2.2430419921875e-06,
      "model_forward_time": 1.701099157333374,
      "step": 2352
    },
    {
      "epoch": 2.2430419921875e-06,
      "step": 2352,
      "training_step_time": 7.694504737854004
    },
    {
      "epoch": 2.2439956665039064e-06,
      "model_forward_time": 1.694462776184082,
      "step": 2353
    },
    {
      "epoch": 2.2439956665039064e-06,
      "step": 2353,
      "training_step_time": 7.626894950866699
    },
    {
      "epoch": 2.2449493408203126e-06,
      "model_forward_time": 1.7121617794036865,
      "step": 2354
    },
    {
      "epoch": 2.2449493408203126e-06,
      "step": 2354,
      "training_step_time": 6.68930721282959
    },
    {
      "epoch": 2.2459030151367187e-06,
      "model_forward_time": 1.3393006324768066,
      "step": 2355
    },
    {
      "epoch": 2.2459030151367187e-06,
      "step": 2355,
      "training_step_time": 6.730976581573486
    },
    {
      "epoch": 2.246856689453125e-06,
      "model_forward_time": 1.7117977142333984,
      "step": 2356
    },
    {
      "epoch": 2.246856689453125e-06,
      "step": 2356,
      "training_step_time": 6.775120496749878
    },
    {
      "epoch": 2.2478103637695314e-06,
      "model_forward_time": 1.7170202732086182,
      "step": 2357
    },
    {
      "epoch": 2.2478103637695314e-06,
      "step": 2357,
      "training_step_time": 6.767805337905884
    },
    {
      "epoch": 2.2487640380859376e-06,
      "model_forward_time": 1.7180249691009521,
      "step": 2358
    },
    {
      "epoch": 2.2487640380859376e-06,
      "step": 2358,
      "training_step_time": 7.465162754058838
    },
    {
      "epoch": 2.2497177124023437e-06,
      "model_forward_time": 1.7134582996368408,
      "step": 2359
    },
    {
      "epoch": 2.2497177124023437e-06,
      "step": 2359,
      "training_step_time": 8.490391969680786
    },
    {
      "epoch": 2.25067138671875e-06,
      "grad_norm": 0.8596805334091187,
      "learning_rate": 4.718e-06,
      "loss": 0.0961,
      "step": 2360
    },
    {
      "epoch": 2.25067138671875e-06,
      "model_forward_time": 1.7401371002197266,
      "step": 2360
    },
    {
      "dynamics_loss_avg": 0.06707405243068934,
      "epoch": 2.25067138671875e-06,
      "step": 2360
    },
    {
      "action_loss_avg": 0.01627875054255128,
      "epoch": 2.25067138671875e-06,
      "step": 2360
    },
    {
      "epoch": 2.25067138671875e-06,
      "step": 2360,
      "training_step_time": 10.039944887161255
    },
    {
      "epoch": 2.2516250610351564e-06,
      "model_forward_time": 1.7108170986175537,
      "step": 2361
    },
    {
      "epoch": 2.2516250610351564e-06,
      "step": 2361,
      "training_step_time": 7.954291343688965
    },
    {
      "epoch": 2.2525787353515626e-06,
      "model_forward_time": 1.7167627811431885,
      "step": 2362
    },
    {
      "epoch": 2.2525787353515626e-06,
      "step": 2362,
      "training_step_time": 7.270139455795288
    },
    {
      "epoch": 2.2535324096679687e-06,
      "model_forward_time": 1.7120699882507324,
      "step": 2363
    },
    {
      "epoch": 2.2535324096679687e-06,
      "step": 2363,
      "training_step_time": 6.843847751617432
    },
    {
      "epoch": 2.254486083984375e-06,
      "model_forward_time": 1.7078936100006104,
      "step": 2364
    },
    {
      "epoch": 2.254486083984375e-06,
      "step": 2364,
      "training_step_time": 6.8057990074157715
    },
    {
      "epoch": 2.2554397583007815e-06,
      "model_forward_time": 1.7124195098876953,
      "step": 2365
    },
    {
      "epoch": 2.2554397583007815e-06,
      "step": 2365,
      "training_step_time": 6.837428569793701
    },
    {
      "epoch": 2.2563934326171876e-06,
      "model_forward_time": 1.7166011333465576,
      "step": 2366
    },
    {
      "epoch": 2.2563934326171876e-06,
      "step": 2366,
      "training_step_time": 6.794787883758545
    },
    {
      "epoch": 2.2573471069335937e-06,
      "model_forward_time": 1.7128105163574219,
      "step": 2367
    },
    {
      "epoch": 2.2573471069335937e-06,
      "step": 2367,
      "training_step_time": 6.834569692611694
    },
    {
      "epoch": 2.25830078125e-06,
      "model_forward_time": 1.7185359001159668,
      "step": 2368
    },
    {
      "epoch": 2.25830078125e-06,
      "step": 2368,
      "training_step_time": 7.653083562850952
    },
    {
      "epoch": 2.2592544555664065e-06,
      "model_forward_time": 1.752349615097046,
      "step": 2369
    },
    {
      "epoch": 2.2592544555664065e-06,
      "step": 2369,
      "training_step_time": 10.553043365478516
    },
    {
      "epoch": 2.2602081298828126e-06,
      "grad_norm": 0.7623290419578552,
      "learning_rate": 4.738e-06,
      "loss": 0.0991,
      "step": 2370
    },
    {
      "epoch": 2.2602081298828126e-06,
      "model_forward_time": 2.242957353591919,
      "step": 2370
    },
    {
      "dynamics_loss_avg": 0.08201719783246517,
      "epoch": 2.2602081298828126e-06,
      "step": 2370
    },
    {
      "action_loss_avg": 0.01751863076351583,
      "epoch": 2.2602081298828126e-06,
      "step": 2370
    },
    {
      "epoch": 2.2602081298828126e-06,
      "step": 2370,
      "training_step_time": 8.919898986816406
    },
    {
      "epoch": 2.2611618041992187e-06,
      "model_forward_time": 1.7114970684051514,
      "step": 2371
    },
    {
      "epoch": 2.2611618041992187e-06,
      "step": 2371,
      "training_step_time": 7.229633569717407
    },
    {
      "epoch": 2.262115478515625e-06,
      "model_forward_time": 1.709876537322998,
      "step": 2372
    },
    {
      "epoch": 2.262115478515625e-06,
      "step": 2372,
      "training_step_time": 6.77867579460144
    },
    {
      "epoch": 2.2630691528320315e-06,
      "model_forward_time": 1.7101223468780518,
      "step": 2373
    },
    {
      "epoch": 2.2630691528320315e-06,
      "step": 2373,
      "training_step_time": 6.720191717147827
    },
    {
      "epoch": 2.2640228271484376e-06,
      "model_forward_time": 1.7140679359436035,
      "step": 2374
    },
    {
      "epoch": 2.2640228271484376e-06,
      "step": 2374,
      "training_step_time": 6.844709157943726
    },
    {
      "epoch": 2.2649765014648438e-06,
      "model_forward_time": 1.7136940956115723,
      "step": 2375
    },
    {
      "epoch": 2.2649765014648438e-06,
      "step": 2375,
      "training_step_time": 6.8439531326293945
    },
    {
      "epoch": 2.26593017578125e-06,
      "model_forward_time": 1.7078006267547607,
      "step": 2376
    },
    {
      "epoch": 2.26593017578125e-06,
      "step": 2376,
      "training_step_time": 6.840709924697876
    },
    {
      "epoch": 2.266883850097656e-06,
      "model_forward_time": 1.7087113857269287,
      "step": 2377
    },
    {
      "epoch": 2.266883850097656e-06,
      "step": 2377,
      "training_step_time": 7.661358118057251
    },
    {
      "epoch": 2.2678375244140626e-06,
      "model_forward_time": 1.7162306308746338,
      "step": 2378
    },
    {
      "epoch": 2.2678375244140626e-06,
      "step": 2378,
      "training_step_time": 9.316259145736694
    },
    {
      "epoch": 2.2687911987304688e-06,
      "model_forward_time": 2.251244068145752,
      "step": 2379
    },
    {
      "epoch": 2.2687911987304688e-06,
      "step": 2379,
      "training_step_time": 8.899842023849487
    },
    {
      "epoch": 2.269744873046875e-06,
      "grad_norm": 1.2447607517242432,
      "learning_rate": 4.758e-06,
      "loss": 0.1042,
      "step": 2380
    },
    {
      "epoch": 2.269744873046875e-06,
      "model_forward_time": 1.698828935623169,
      "step": 2380
    },
    {
      "dynamics_loss_avg": 0.08173255473375321,
      "epoch": 2.269744873046875e-06,
      "step": 2380
    },
    {
      "action_loss_avg": 0.020628237817436456,
      "epoch": 2.269744873046875e-06,
      "step": 2380
    },
    {
      "epoch": 2.269744873046875e-06,
      "step": 2380,
      "training_step_time": 7.274297714233398
    },
    {
      "epoch": 2.270698547363281e-06,
      "model_forward_time": 1.712397813796997,
      "step": 2381
    },
    {
      "epoch": 2.270698547363281e-06,
      "step": 2381,
      "training_step_time": 6.8124589920043945
    },
    {
      "epoch": 2.2716522216796876e-06,
      "model_forward_time": 0.9909458160400391,
      "step": 2382
    },
    {
      "epoch": 2.2716522216796876e-06,
      "step": 2382,
      "training_step_time": 6.827076435089111
    },
    {
      "epoch": 2.2726058959960938e-06,
      "model_forward_time": 1.710737943649292,
      "step": 2383
    },
    {
      "epoch": 2.2726058959960938e-06,
      "step": 2383,
      "training_step_time": 6.850473403930664
    },
    {
      "epoch": 2.2735595703125e-06,
      "model_forward_time": 1.7176837921142578,
      "step": 2384
    },
    {
      "epoch": 2.2735595703125e-06,
      "step": 2384,
      "training_step_time": 6.908430099487305
    },
    {
      "epoch": 2.274513244628906e-06,
      "model_forward_time": 1.7097883224487305,
      "step": 2385
    },
    {
      "epoch": 2.274513244628906e-06,
      "step": 2385,
      "training_step_time": 6.870637655258179
    },
    {
      "epoch": 2.2754669189453126e-06,
      "model_forward_time": 1.7126781940460205,
      "step": 2386
    },
    {
      "epoch": 2.2754669189453126e-06,
      "step": 2386,
      "training_step_time": 7.738839387893677
    },
    {
      "epoch": 2.2764205932617188e-06,
      "model_forward_time": 1.7258529663085938,
      "step": 2387
    },
    {
      "epoch": 2.2764205932617188e-06,
      "step": 2387,
      "training_step_time": 9.381788969039917
    },
    {
      "epoch": 2.277374267578125e-06,
      "model_forward_time": 2.221726179122925,
      "step": 2388
    },
    {
      "epoch": 2.277374267578125e-06,
      "step": 2388,
      "training_step_time": 9.446074485778809
    },
    {
      "epoch": 2.278327941894531e-06,
      "model_forward_time": 1.7114121913909912,
      "step": 2389
    },
    {
      "epoch": 2.278327941894531e-06,
      "step": 2389,
      "training_step_time": 7.727452754974365
    },
    {
      "epoch": 2.2792816162109376e-06,
      "grad_norm": 0.842903196811676,
      "learning_rate": 4.778000000000001e-06,
      "loss": 0.0957,
      "step": 2390
    },
    {
      "epoch": 2.2792816162109376e-06,
      "model_forward_time": 1.3350913524627686,
      "step": 2390
    },
    {
      "dynamics_loss_avg": 0.06705475468188524,
      "epoch": 2.2792816162109376e-06,
      "step": 2390
    },
    {
      "action_loss_avg": 0.01861022221855819,
      "epoch": 2.2792816162109376e-06,
      "step": 2390
    },
    {
      "epoch": 2.2792816162109376e-06,
      "step": 2390,
      "training_step_time": 7.128232479095459
    },
    {
      "epoch": 2.2802352905273438e-06,
      "model_forward_time": 1.6834416389465332,
      "step": 2391
    },
    {
      "epoch": 2.2802352905273438e-06,
      "step": 2391,
      "training_step_time": 6.72954535484314
    },
    {
      "epoch": 2.28118896484375e-06,
      "model_forward_time": 1.6996161937713623,
      "step": 2392
    },
    {
      "epoch": 2.28118896484375e-06,
      "step": 2392,
      "training_step_time": 6.741552829742432
    },
    {
      "epoch": 2.282142639160156e-06,
      "model_forward_time": 1.6806533336639404,
      "step": 2393
    },
    {
      "epoch": 2.282142639160156e-06,
      "step": 2393,
      "training_step_time": 6.728445529937744
    },
    {
      "epoch": 2.2830963134765626e-06,
      "model_forward_time": 1.6902515888214111,
      "step": 2394
    },
    {
      "epoch": 2.2830963134765626e-06,
      "step": 2394,
      "training_step_time": 6.756926774978638
    },
    {
      "epoch": 2.2840499877929688e-06,
      "model_forward_time": 1.6881098747253418,
      "step": 2395
    },
    {
      "epoch": 2.2840499877929688e-06,
      "step": 2395,
      "training_step_time": 8.57385540008545
    },
    {
      "epoch": 2.285003662109375e-06,
      "model_forward_time": 1.6933045387268066,
      "step": 2396
    },
    {
      "epoch": 2.285003662109375e-06,
      "step": 2396,
      "training_step_time": 9.5520658493042
    },
    {
      "epoch": 2.285957336425781e-06,
      "model_forward_time": 1.7458598613739014,
      "step": 2397
    },
    {
      "epoch": 2.285957336425781e-06,
      "step": 2397,
      "training_step_time": 8.810672998428345
    },
    {
      "epoch": 2.2869110107421876e-06,
      "model_forward_time": 2.204094171524048,
      "step": 2398
    },
    {
      "epoch": 2.2869110107421876e-06,
      "step": 2398,
      "training_step_time": 7.669694185256958
    },
    {
      "epoch": 2.2878646850585938e-06,
      "model_forward_time": 1.7034378051757812,
      "step": 2399
    },
    {
      "epoch": 2.2878646850585938e-06,
      "step": 2399,
      "training_step_time": 7.176367521286011
    },
    {
      "epoch": 2.288818359375e-06,
      "grad_norm": 0.9522948265075684,
      "learning_rate": 4.7980000000000005e-06,
      "loss": 0.092,
      "step": 2400
    },
    {
      "epoch": 2.288818359375e-06,
      "model_forward_time": 1.6864290237426758,
      "step": 2400
    },
    {
      "dynamics_loss_avg": 0.07004248201847077,
      "epoch": 2.288818359375e-06,
      "step": 2400
    },
    {
      "action_loss_avg": 0.012594981724396348,
      "epoch": 2.288818359375e-06,
      "step": 2400
    },
    {
      "epoch": 2.288818359375e-06,
      "step": 2400,
      "training_step_time": 6.75780725479126
    },
    {
      "epoch": 2.289772033691406e-06,
      "model_forward_time": 1.687330961227417,
      "step": 2401
    },
    {
      "epoch": 2.289772033691406e-06,
      "step": 2401,
      "training_step_time": 6.7479798793792725
    },
    {
      "epoch": 2.2907257080078126e-06,
      "model_forward_time": 1.6874003410339355,
      "step": 2402
    },
    {
      "epoch": 2.2907257080078126e-06,
      "step": 2402,
      "training_step_time": 6.785500526428223
    },
    {
      "epoch": 2.2916793823242188e-06,
      "model_forward_time": 1.686363697052002,
      "step": 2403
    },
    {
      "epoch": 2.2916793823242188e-06,
      "step": 2403,
      "training_step_time": 6.793493270874023
    },
    {
      "epoch": 2.292633056640625e-06,
      "model_forward_time": 1.6966030597686768,
      "step": 2404
    },
    {
      "epoch": 2.292633056640625e-06,
      "step": 2404,
      "training_step_time": 7.710700988769531
    },
    {
      "epoch": 2.293586730957031e-06,
      "model_forward_time": 1.687957763671875,
      "step": 2405
    },
    {
      "epoch": 2.293586730957031e-06,
      "step": 2405,
      "training_step_time": 10.228371620178223
    },
    {
      "epoch": 2.2945404052734376e-06,
      "model_forward_time": 1.7126929759979248,
      "step": 2406
    },
    {
      "epoch": 2.2945404052734376e-06,
      "step": 2406,
      "training_step_time": 8.865253925323486
    },
    {
      "epoch": 2.2954940795898438e-06,
      "model_forward_time": 2.22407603263855,
      "step": 2407
    },
    {
      "epoch": 2.2954940795898438e-06,
      "step": 2407,
      "training_step_time": 7.801012754440308
    },
    {
      "epoch": 2.29644775390625e-06,
      "model_forward_time": 1.692126750946045,
      "step": 2408
    },
    {
      "epoch": 2.29644775390625e-06,
      "step": 2408,
      "training_step_time": 7.184037446975708
    },
    {
      "epoch": 2.297401428222656e-06,
      "model_forward_time": 1.689345359802246,
      "step": 2409
    },
    {
      "epoch": 2.297401428222656e-06,
      "step": 2409,
      "training_step_time": 6.725556135177612
    },
    {
      "epoch": 2.2983551025390626e-06,
      "grad_norm": 0.7480741143226624,
      "learning_rate": 4.818e-06,
      "loss": 0.0939,
      "step": 2410
    },
    {
      "epoch": 2.2983551025390626e-06,
      "model_forward_time": 1.6930372714996338,
      "step": 2410
    },
    {
      "dynamics_loss_avg": 0.054166828468441966,
      "epoch": 2.2983551025390626e-06,
      "step": 2410
    },
    {
      "action_loss_avg": 0.014290714589878916,
      "epoch": 2.2983551025390626e-06,
      "step": 2410
    },
    {
      "epoch": 2.2983551025390626e-06,
      "step": 2410,
      "training_step_time": 6.7517852783203125
    },
    {
      "epoch": 2.2993087768554688e-06,
      "model_forward_time": 1.695028305053711,
      "step": 2411
    },
    {
      "epoch": 2.2993087768554688e-06,
      "step": 2411,
      "training_step_time": 6.826766729354858
    },
    {
      "epoch": 2.300262451171875e-06,
      "model_forward_time": 1.689457654953003,
      "step": 2412
    },
    {
      "epoch": 2.300262451171875e-06,
      "step": 2412,
      "training_step_time": 6.831456184387207
    },
    {
      "epoch": 2.301216125488281e-06,
      "model_forward_time": 1.6863141059875488,
      "step": 2413
    },
    {
      "epoch": 2.301216125488281e-06,
      "step": 2413,
      "training_step_time": 8.005025148391724
    },
    {
      "epoch": 2.3021697998046876e-06,
      "model_forward_time": 1.6887247562408447,
      "step": 2414
    },
    {
      "epoch": 2.3021697998046876e-06,
      "step": 2414,
      "training_step_time": 8.57095193862915
    },
    {
      "epoch": 2.3031234741210938e-06,
      "model_forward_time": 1.7187108993530273,
      "step": 2415
    },
    {
      "epoch": 2.3031234741210938e-06,
      "step": 2415,
      "training_step_time": 9.238855123519897
    },
    {
      "epoch": 2.3040771484375e-06,
      "model_forward_time": 2.2393412590026855,
      "step": 2416
    },
    {
      "epoch": 2.3040771484375e-06,
      "step": 2416,
      "training_step_time": 9.224754810333252
    },
    {
      "epoch": 2.305030822753906e-06,
      "model_forward_time": 1.686586856842041,
      "step": 2417
    },
    {
      "epoch": 2.305030822753906e-06,
      "step": 2417,
      "training_step_time": 7.030702590942383
    },
    {
      "epoch": 2.3059844970703126e-06,
      "model_forward_time": 1.6903676986694336,
      "step": 2418
    },
    {
      "epoch": 2.3059844970703126e-06,
      "step": 2418,
      "training_step_time": 7.11496376991272
    },
    {
      "epoch": 2.3069381713867188e-06,
      "model_forward_time": 1.6846179962158203,
      "step": 2419
    },
    {
      "epoch": 2.3069381713867188e-06,
      "step": 2419,
      "training_step_time": 6.742968797683716
    },
    {
      "epoch": 2.307891845703125e-06,
      "grad_norm": 0.8091705441474915,
      "learning_rate": 4.838e-06,
      "loss": 0.0973,
      "step": 2420
    },
    {
      "epoch": 2.307891845703125e-06,
      "model_forward_time": 1.6967964172363281,
      "step": 2420
    },
    {
      "dynamics_loss_avg": 0.06014033667743206,
      "epoch": 2.307891845703125e-06,
      "step": 2420
    },
    {
      "action_loss_avg": 0.014365414623171091,
      "epoch": 2.307891845703125e-06,
      "step": 2420
    },
    {
      "epoch": 2.307891845703125e-06,
      "step": 2420,
      "training_step_time": 6.79573655128479
    },
    {
      "epoch": 2.308845520019531e-06,
      "model_forward_time": 1.692502737045288,
      "step": 2421
    },
    {
      "epoch": 2.308845520019531e-06,
      "step": 2421,
      "training_step_time": 6.814342260360718
    },
    {
      "epoch": 2.3097991943359376e-06,
      "model_forward_time": 1.6838247776031494,
      "step": 2422
    },
    {
      "epoch": 2.3097991943359376e-06,
      "step": 2422,
      "training_step_time": 7.074040412902832
    },
    {
      "epoch": 2.310752868652344e-06,
      "model_forward_time": 1.6924958229064941,
      "step": 2423
    },
    {
      "epoch": 2.310752868652344e-06,
      "step": 2423,
      "training_step_time": 7.677203178405762
    },
    {
      "epoch": 2.31170654296875e-06,
      "model_forward_time": 1.6945726871490479,
      "step": 2424
    },
    {
      "epoch": 2.31170654296875e-06,
      "step": 2424,
      "training_step_time": 7.89625883102417
    },
    {
      "epoch": 2.312660217285156e-06,
      "model_forward_time": 2.214369058609009,
      "step": 2425
    },
    {
      "epoch": 2.312660217285156e-06,
      "step": 2425,
      "training_step_time": 9.792151927947998
    },
    {
      "epoch": 2.3136138916015626e-06,
      "model_forward_time": 1.6884875297546387,
      "step": 2426
    },
    {
      "epoch": 2.3136138916015626e-06,
      "step": 2426,
      "training_step_time": 7.725016355514526
    },
    {
      "epoch": 2.314567565917969e-06,
      "model_forward_time": 1.3126797676086426,
      "step": 2427
    },
    {
      "epoch": 2.314567565917969e-06,
      "step": 2427,
      "training_step_time": 7.252157211303711
    },
    {
      "epoch": 2.315521240234375e-06,
      "model_forward_time": 1.6927030086517334,
      "step": 2428
    },
    {
      "epoch": 2.315521240234375e-06,
      "step": 2428,
      "training_step_time": 6.898264646530151
    },
    {
      "epoch": 2.316474914550781e-06,
      "model_forward_time": 1.6897952556610107,
      "step": 2429
    },
    {
      "epoch": 2.316474914550781e-06,
      "step": 2429,
      "training_step_time": 6.944296598434448
    },
    {
      "epoch": 2.3174285888671877e-06,
      "grad_norm": 0.8773289918899536,
      "learning_rate": 4.858000000000001e-06,
      "loss": 0.0994,
      "step": 2430
    },
    {
      "epoch": 2.3174285888671877e-06,
      "model_forward_time": 1.6888644695281982,
      "step": 2430
    },
    {
      "dynamics_loss_avg": 0.06757515631616115,
      "epoch": 2.3174285888671877e-06,
      "step": 2430
    },
    {
      "action_loss_avg": 0.02191650401800871,
      "epoch": 2.3174285888671877e-06,
      "step": 2430
    },
    {
      "epoch": 2.3174285888671877e-06,
      "step": 2430,
      "training_step_time": 6.8718414306640625
    },
    {
      "epoch": 2.318382263183594e-06,
      "model_forward_time": 1.6867048740386963,
      "step": 2431
    },
    {
      "epoch": 2.318382263183594e-06,
      "step": 2431,
      "training_step_time": 7.191287040710449
    },
    {
      "epoch": 2.3193359375e-06,
      "model_forward_time": 1.687549114227295,
      "step": 2432
    },
    {
      "epoch": 2.3193359375e-06,
      "step": 2432,
      "training_step_time": 8.60007619857788
    },
    {
      "epoch": 2.320289611816406e-06,
      "model_forward_time": 1.6941604614257812,
      "step": 2433
    },
    {
      "epoch": 2.320289611816406e-06,
      "step": 2433,
      "training_step_time": 9.500861167907715
    },
    {
      "epoch": 2.3212432861328127e-06,
      "model_forward_time": 1.7066152095794678,
      "step": 2434
    },
    {
      "epoch": 2.3212432861328127e-06,
      "step": 2434,
      "training_step_time": 9.653607845306396
    },
    {
      "epoch": 2.322196960449219e-06,
      "model_forward_time": 2.2169811725616455,
      "step": 2435
    },
    {
      "epoch": 2.322196960449219e-06,
      "step": 2435,
      "training_step_time": 7.842163324356079
    },
    {
      "epoch": 2.323150634765625e-06,
      "model_forward_time": 1.685701608657837,
      "step": 2436
    },
    {
      "epoch": 2.323150634765625e-06,
      "step": 2436,
      "training_step_time": 7.21258807182312
    },
    {
      "epoch": 2.324104309082031e-06,
      "model_forward_time": 1.6872272491455078,
      "step": 2437
    },
    {
      "epoch": 2.324104309082031e-06,
      "step": 2437,
      "training_step_time": 6.8878936767578125
    },
    {
      "epoch": 2.3250579833984377e-06,
      "model_forward_time": 1.6903750896453857,
      "step": 2438
    },
    {
      "epoch": 2.3250579833984377e-06,
      "step": 2438,
      "training_step_time": 6.843102216720581
    },
    {
      "epoch": 2.326011657714844e-06,
      "model_forward_time": 1.6938395500183105,
      "step": 2439
    },
    {
      "epoch": 2.326011657714844e-06,
      "step": 2439,
      "training_step_time": 6.848852872848511
    },
    {
      "epoch": 2.32696533203125e-06,
      "grad_norm": 1.0727872848510742,
      "learning_rate": 4.8780000000000006e-06,
      "loss": 0.092,
      "step": 2440
    },
    {
      "epoch": 2.32696533203125e-06,
      "model_forward_time": 1.68788480758667,
      "step": 2440
    },
    {
      "dynamics_loss_avg": 0.06939710211008787,
      "epoch": 2.32696533203125e-06,
      "step": 2440
    },
    {
      "action_loss_avg": 0.017569845262914895,
      "epoch": 2.32696533203125e-06,
      "step": 2440
    },
    {
      "epoch": 2.32696533203125e-06,
      "step": 2440,
      "training_step_time": 7.186986446380615
    },
    {
      "epoch": 2.327919006347656e-06,
      "model_forward_time": 1.6918790340423584,
      "step": 2441
    },
    {
      "epoch": 2.327919006347656e-06,
      "step": 2441,
      "training_step_time": 10.160691499710083
    },
    {
      "epoch": 2.3288726806640627e-06,
      "model_forward_time": 1.6952776908874512,
      "step": 2442
    },
    {
      "epoch": 2.3288726806640627e-06,
      "step": 2442,
      "training_step_time": 10.36400842666626
    },
    {
      "epoch": 2.329826354980469e-06,
      "model_forward_time": 1.693002462387085,
      "step": 2443
    },
    {
      "epoch": 2.329826354980469e-06,
      "step": 2443,
      "training_step_time": 8.530940294265747
    },
    {
      "epoch": 2.330780029296875e-06,
      "model_forward_time": 2.219188690185547,
      "step": 2444
    },
    {
      "epoch": 2.330780029296875e-06,
      "step": 2444,
      "training_step_time": 7.836572885513306
    },
    {
      "epoch": 2.331733703613281e-06,
      "model_forward_time": 1.688979148864746,
      "step": 2445
    },
    {
      "epoch": 2.331733703613281e-06,
      "step": 2445,
      "training_step_time": 7.147313594818115
    },
    {
      "epoch": 2.3326873779296877e-06,
      "model_forward_time": 1.6899263858795166,
      "step": 2446
    },
    {
      "epoch": 2.3326873779296877e-06,
      "step": 2446,
      "training_step_time": 6.7970709800720215
    },
    {
      "epoch": 2.333641052246094e-06,
      "model_forward_time": 1.6874206066131592,
      "step": 2447
    },
    {
      "epoch": 2.333641052246094e-06,
      "step": 2447,
      "training_step_time": 6.805238723754883
    },
    {
      "epoch": 2.3345947265625e-06,
      "model_forward_time": 1.6873440742492676,
      "step": 2448
    },
    {
      "epoch": 2.3345947265625e-06,
      "step": 2448,
      "training_step_time": 6.815779447555542
    },
    {
      "epoch": 2.335548400878906e-06,
      "model_forward_time": 1.713346242904663,
      "step": 2449
    },
    {
      "epoch": 2.335548400878906e-06,
      "step": 2449,
      "training_step_time": 7.094177722930908
    },
    {
      "epoch": 2.3365020751953127e-06,
      "grad_norm": 0.9143434762954712,
      "learning_rate": 4.898e-06,
      "loss": 0.1011,
      "step": 2450
    },
    {
      "epoch": 2.3365020751953127e-06,
      "model_forward_time": 1.6870207786560059,
      "step": 2450
    },
    {
      "dynamics_loss_avg": 0.06155174244195223,
      "epoch": 2.3365020751953127e-06,
      "step": 2450
    },
    {
      "action_loss_avg": 0.013212808594107627,
      "epoch": 2.3365020751953127e-06,
      "step": 2450
    },
    {
      "epoch": 2.3365020751953127e-06,
      "step": 2450,
      "training_step_time": 7.502503395080566
    },
    {
      "epoch": 2.337455749511719e-06,
      "model_forward_time": 1.686701774597168,
      "step": 2451
    },
    {
      "epoch": 2.337455749511719e-06,
      "step": 2451,
      "training_step_time": 9.54768443107605
    },
    {
      "epoch": 2.338409423828125e-06,
      "model_forward_time": 1.7037694454193115,
      "step": 2452
    },
    {
      "epoch": 2.338409423828125e-06,
      "step": 2452,
      "training_step_time": 9.194746017456055
    },
    {
      "epoch": 2.339363098144531e-06,
      "model_forward_time": 2.2271335124969482,
      "step": 2453
    },
    {
      "epoch": 2.339363098144531e-06,
      "step": 2453,
      "training_step_time": 7.853610277175903
    },
    {
      "epoch": 2.3403167724609377e-06,
      "model_forward_time": 1.6895966529846191,
      "step": 2454
    },
    {
      "epoch": 2.3403167724609377e-06,
      "step": 2454,
      "training_step_time": 7.4075634479522705
    },
    {
      "epoch": 2.341270446777344e-06,
      "model_forward_time": 1.6891767978668213,
      "step": 2455
    },
    {
      "epoch": 2.341270446777344e-06,
      "step": 2455,
      "training_step_time": 6.83087158203125
    },
    {
      "epoch": 2.34222412109375e-06,
      "model_forward_time": 1.6913056373596191,
      "step": 2456
    },
    {
      "epoch": 2.34222412109375e-06,
      "step": 2456,
      "training_step_time": 6.883641242980957
    },
    {
      "epoch": 2.343177795410156e-06,
      "model_forward_time": 1.6870617866516113,
      "step": 2457
    },
    {
      "epoch": 2.343177795410156e-06,
      "step": 2457,
      "training_step_time": 6.837433576583862
    },
    {
      "epoch": 2.3441314697265627e-06,
      "model_forward_time": 1.6850602626800537,
      "step": 2458
    },
    {
      "epoch": 2.3441314697265627e-06,
      "step": 2458,
      "training_step_time": 7.078970909118652
    },
    {
      "epoch": 2.345085144042969e-06,
      "model_forward_time": 1.6854352951049805,
      "step": 2459
    },
    {
      "epoch": 2.345085144042969e-06,
      "step": 2459,
      "training_step_time": 7.2873992919921875
    },
    {
      "epoch": 2.346038818359375e-06,
      "grad_norm": 0.7305988669395447,
      "learning_rate": 4.918e-06,
      "loss": 0.0937,
      "step": 2460
    },
    {
      "epoch": 2.346038818359375e-06,
      "model_forward_time": 1.6917078495025635,
      "step": 2460
    },
    {
      "dynamics_loss_avg": 0.06921304408460856,
      "epoch": 2.346038818359375e-06,
      "step": 2460
    },
    {
      "action_loss_avg": 0.01845975313335657,
      "epoch": 2.346038818359375e-06,
      "step": 2460
    },
    {
      "epoch": 2.346038818359375e-06,
      "step": 2460,
      "training_step_time": 9.087255001068115
    },
    {
      "epoch": 2.346992492675781e-06,
      "model_forward_time": 1.6983716487884521,
      "step": 2461
    },
    {
      "epoch": 2.346992492675781e-06,
      "step": 2461,
      "training_step_time": 7.926048040390015
    },
    {
      "epoch": 2.3479461669921877e-06,
      "model_forward_time": 2.2213568687438965,
      "step": 2462
    },
    {
      "epoch": 2.3479461669921877e-06,
      "step": 2462,
      "training_step_time": 8.673662185668945
    },
    {
      "epoch": 2.348899841308594e-06,
      "model_forward_time": 1.6931798458099365,
      "step": 2463
    },
    {
      "epoch": 2.348899841308594e-06,
      "step": 2463,
      "training_step_time": 7.4416821002960205
    },
    {
      "epoch": 2.349853515625e-06,
      "model_forward_time": 1.6885616779327393,
      "step": 2464
    },
    {
      "epoch": 2.349853515625e-06,
      "step": 2464,
      "training_step_time": 6.780418395996094
    },
    {
      "epoch": 2.350807189941406e-06,
      "model_forward_time": 1.6875452995300293,
      "step": 2465
    },
    {
      "epoch": 2.350807189941406e-06,
      "step": 2465,
      "training_step_time": 6.816130876541138
    },
    {
      "epoch": 2.3517608642578127e-06,
      "model_forward_time": 1.6898398399353027,
      "step": 2466
    },
    {
      "epoch": 2.3517608642578127e-06,
      "step": 2466,
      "training_step_time": 6.780057430267334
    },
    {
      "epoch": 2.352714538574219e-06,
      "model_forward_time": 1.6894984245300293,
      "step": 2467
    },
    {
      "epoch": 2.352714538574219e-06,
      "step": 2467,
      "training_step_time": 7.081234693527222
    },
    {
      "epoch": 2.353668212890625e-06,
      "model_forward_time": 1.6865406036376953,
      "step": 2468
    },
    {
      "epoch": 2.353668212890625e-06,
      "step": 2468,
      "training_step_time": 7.863491058349609
    },
    {
      "epoch": 2.354621887207031e-06,
      "model_forward_time": 1.6934380531311035,
      "step": 2469
    },
    {
      "epoch": 2.354621887207031e-06,
      "step": 2469,
      "training_step_time": 9.387346744537354
    },
    {
      "epoch": 2.3555755615234377e-06,
      "grad_norm": 0.5765329003334045,
      "learning_rate": 4.938000000000001e-06,
      "loss": 0.0918,
      "step": 2470
    },
    {
      "epoch": 2.3555755615234377e-06,
      "model_forward_time": 1.6896519660949707,
      "step": 2470
    },
    {
      "dynamics_loss_avg": 0.07790741994976998,
      "epoch": 2.3555755615234377e-06,
      "step": 2470
    },
    {
      "action_loss_avg": 0.013240313669666649,
      "epoch": 2.3555755615234377e-06,
      "step": 2470
    },
    {
      "epoch": 2.3555755615234377e-06,
      "step": 2470,
      "training_step_time": 8.744411706924438
    },
    {
      "epoch": 2.356529235839844e-06,
      "model_forward_time": 2.2088968753814697,
      "step": 2471
    },
    {
      "epoch": 2.356529235839844e-06,
      "step": 2471,
      "training_step_time": 9.588491916656494
    },
    {
      "epoch": 2.35748291015625e-06,
      "model_forward_time": 1.6881446838378906,
      "step": 2472
    },
    {
      "epoch": 2.35748291015625e-06,
      "step": 2472,
      "training_step_time": 7.653618812561035
    },
    {
      "epoch": 2.358436584472656e-06,
      "model_forward_time": 1.6786282062530518,
      "step": 2473
    },
    {
      "epoch": 2.358436584472656e-06,
      "step": 2473,
      "training_step_time": 6.791238069534302
    },
    {
      "epoch": 2.3593902587890627e-06,
      "model_forward_time": 1.6932878494262695,
      "step": 2474
    },
    {
      "epoch": 2.3593902587890627e-06,
      "step": 2474,
      "training_step_time": 6.829521417617798
    },
    {
      "epoch": 2.360343933105469e-06,
      "model_forward_time": 1.6872563362121582,
      "step": 2475
    },
    {
      "epoch": 2.360343933105469e-06,
      "step": 2475,
      "training_step_time": 6.775264739990234
    },
    {
      "epoch": 2.361297607421875e-06,
      "model_forward_time": 1.6874406337738037,
      "step": 2476
    },
    {
      "epoch": 2.361297607421875e-06,
      "step": 2476,
      "training_step_time": 7.17496657371521
    },
    {
      "epoch": 2.362251281738281e-06,
      "model_forward_time": 1.689147710800171,
      "step": 2477
    },
    {
      "epoch": 2.362251281738281e-06,
      "step": 2477,
      "training_step_time": 7.034054279327393
    },
    {
      "epoch": 2.3632049560546877e-06,
      "model_forward_time": 1.6914901733398438,
      "step": 2478
    },
    {
      "epoch": 2.3632049560546877e-06,
      "step": 2478,
      "training_step_time": 7.741094589233398
    },
    {
      "epoch": 2.364158630371094e-06,
      "model_forward_time": 1.6888134479522705,
      "step": 2479
    },
    {
      "epoch": 2.364158630371094e-06,
      "step": 2479,
      "training_step_time": 9.357220888137817
    },
    {
      "epoch": 2.3651123046875e-06,
      "grad_norm": 0.87713623046875,
      "learning_rate": 4.958000000000001e-06,
      "loss": 0.0983,
      "step": 2480
    },
    {
      "epoch": 2.3651123046875e-06,
      "model_forward_time": 1.7227551937103271,
      "step": 2480
    },
    {
      "dynamics_loss_avg": 0.06779609080404043,
      "epoch": 2.3651123046875e-06,
      "step": 2480
    },
    {
      "action_loss_avg": 0.01502970433793962,
      "epoch": 2.3651123046875e-06,
      "step": 2480
    },
    {
      "epoch": 2.3651123046875e-06,
      "step": 2480,
      "training_step_time": 8.412980556488037
    },
    {
      "epoch": 2.366065979003906e-06,
      "model_forward_time": 2.222003698348999,
      "step": 2481
    },
    {
      "epoch": 2.366065979003906e-06,
      "step": 2481,
      "training_step_time": 7.637818098068237
    },
    {
      "epoch": 2.3670196533203127e-06,
      "model_forward_time": 1.6820976734161377,
      "step": 2482
    },
    {
      "epoch": 2.3670196533203127e-06,
      "step": 2482,
      "training_step_time": 7.393907308578491
    },
    {
      "epoch": 2.367973327636719e-06,
      "model_forward_time": 1.6890602111816406,
      "step": 2483
    },
    {
      "epoch": 2.367973327636719e-06,
      "step": 2483,
      "training_step_time": 6.814559698104858
    },
    {
      "epoch": 2.368927001953125e-06,
      "model_forward_time": 0.9708094596862793,
      "step": 2484
    },
    {
      "epoch": 2.368927001953125e-06,
      "step": 2484,
      "training_step_time": 6.744801044464111
    },
    {
      "epoch": 2.369880676269531e-06,
      "model_forward_time": 1.6886699199676514,
      "step": 2485
    },
    {
      "epoch": 2.369880676269531e-06,
      "step": 2485,
      "training_step_time": 7.06838321685791
    },
    {
      "epoch": 2.3708343505859377e-06,
      "model_forward_time": 1.6855907440185547,
      "step": 2486
    },
    {
      "epoch": 2.3708343505859377e-06,
      "step": 2486,
      "training_step_time": 7.030566453933716
    },
    {
      "epoch": 2.371788024902344e-06,
      "model_forward_time": 1.687589406967163,
      "step": 2487
    },
    {
      "epoch": 2.371788024902344e-06,
      "step": 2487,
      "training_step_time": 7.698974370956421
    },
    {
      "epoch": 2.37274169921875e-06,
      "model_forward_time": 1.6908988952636719,
      "step": 2488
    },
    {
      "epoch": 2.37274169921875e-06,
      "step": 2488,
      "training_step_time": 7.689242362976074
    },
    {
      "epoch": 2.373695373535156e-06,
      "model_forward_time": 1.7027857303619385,
      "step": 2489
    },
    {
      "epoch": 2.373695373535156e-06,
      "step": 2489,
      "training_step_time": 8.365062713623047
    },
    {
      "epoch": 2.3746490478515627e-06,
      "grad_norm": 1.0051597356796265,
      "learning_rate": 4.9780000000000005e-06,
      "loss": 0.1008,
      "step": 2490
    },
    {
      "epoch": 2.3746490478515627e-06,
      "model_forward_time": 2.235934019088745,
      "step": 2490
    },
    {
      "dynamics_loss_avg": 0.06736342422664165,
      "epoch": 2.3746490478515627e-06,
      "step": 2490
    },
    {
      "action_loss_avg": 0.015013945242390036,
      "epoch": 2.3746490478515627e-06,
      "step": 2490
    },
    {
      "epoch": 2.3746490478515627e-06,
      "step": 2490,
      "training_step_time": 7.662414073944092
    },
    {
      "epoch": 2.375602722167969e-06,
      "model_forward_time": 1.6878385543823242,
      "step": 2491
    },
    {
      "epoch": 2.375602722167969e-06,
      "step": 2491,
      "training_step_time": 7.4658732414245605
    },
    {
      "epoch": 2.376556396484375e-06,
      "model_forward_time": 1.6887199878692627,
      "step": 2492
    },
    {
      "epoch": 2.376556396484375e-06,
      "step": 2492,
      "training_step_time": 6.756598949432373
    },
    {
      "epoch": 2.377510070800781e-06,
      "model_forward_time": 1.6917455196380615,
      "step": 2493
    },
    {
      "epoch": 2.377510070800781e-06,
      "step": 2493,
      "training_step_time": 6.7676100730896
    },
    {
      "epoch": 2.3784637451171877e-06,
      "model_forward_time": 1.6825759410858154,
      "step": 2494
    },
    {
      "epoch": 2.3784637451171877e-06,
      "step": 2494,
      "training_step_time": 7.080331325531006
    },
    {
      "epoch": 2.379417419433594e-06,
      "model_forward_time": 1.6908855438232422,
      "step": 2495
    },
    {
      "epoch": 2.379417419433594e-06,
      "step": 2495,
      "training_step_time": 7.040821552276611
    },
    {
      "epoch": 2.38037109375e-06,
      "model_forward_time": 1.6854448318481445,
      "step": 2496
    },
    {
      "epoch": 2.38037109375e-06,
      "step": 2496,
      "training_step_time": 7.68916392326355
    },
    {
      "epoch": 2.381324768066406e-06,
      "model_forward_time": 1.690554141998291,
      "step": 2497
    },
    {
      "epoch": 2.381324768066406e-06,
      "step": 2497,
      "training_step_time": 9.497703552246094
    },
    {
      "epoch": 2.3822784423828127e-06,
      "model_forward_time": 1.696692705154419,
      "step": 2498
    },
    {
      "epoch": 2.3822784423828127e-06,
      "step": 2498,
      "training_step_time": 8.843961954116821
    },
    {
      "epoch": 2.383232116699219e-06,
      "model_forward_time": 2.218524694442749,
      "step": 2499
    },
    {
      "epoch": 2.383232116699219e-06,
      "step": 2499,
      "training_step_time": 8.833115577697754
    },
    {
      "epoch": 2.384185791015625e-06,
      "grad_norm": 0.903207004070282,
      "learning_rate": 4.998e-06,
      "loss": 0.0886,
      "step": 2500
    },
    {
      "epoch": 2.384185791015625e-06,
      "model_forward_time": 1.686884880065918,
      "step": 2500
    },
    {
      "dynamics_loss_avg": 0.07109237164258957,
      "epoch": 2.384185791015625e-06,
      "step": 2500
    },
    {
      "action_loss_avg": 0.015334866475313902,
      "epoch": 2.384185791015625e-06,
      "step": 2500
    },
    {
      "epoch": 2.384185791015625e-06,
      "step": 2500,
      "training_step_time": 7.486871957778931
    },
    {
      "epoch": 2.385139465332031e-06,
      "model_forward_time": 1.6935808658599854,
      "step": 2501
    },
    {
      "epoch": 2.385139465332031e-06,
      "step": 2501,
      "training_step_time": 6.774590730667114
    },
    {
      "epoch": 2.3860931396484373e-06,
      "model_forward_time": 1.6831002235412598,
      "step": 2502
    },
    {
      "epoch": 2.3860931396484373e-06,
      "step": 2502,
      "training_step_time": 6.784469127655029
    },
    {
      "epoch": 2.387046813964844e-06,
      "model_forward_time": 1.6835546493530273,
      "step": 2503
    },
    {
      "epoch": 2.387046813964844e-06,
      "step": 2503,
      "training_step_time": 7.094947338104248
    },
    {
      "epoch": 2.38800048828125e-06,
      "model_forward_time": 1.6943175792694092,
      "step": 2504
    },
    {
      "epoch": 2.38800048828125e-06,
      "step": 2504,
      "training_step_time": 6.994097471237183
    },
    {
      "epoch": 2.388954162597656e-06,
      "model_forward_time": 1.6877446174621582,
      "step": 2505
    },
    {
      "epoch": 2.388954162597656e-06,
      "step": 2505,
      "training_step_time": 7.690143585205078
    },
    {
      "epoch": 2.3899078369140623e-06,
      "model_forward_time": 1.6873104572296143,
      "step": 2506
    },
    {
      "epoch": 2.3899078369140623e-06,
      "step": 2506,
      "training_step_time": 7.637756586074829
    },
    {
      "epoch": 2.390861511230469e-06,
      "model_forward_time": 1.707240343093872,
      "step": 2507
    },
    {
      "epoch": 2.390861511230469e-06,
      "step": 2507,
      "training_step_time": 8.35244107246399
    },
    {
      "epoch": 2.391815185546875e-06,
      "model_forward_time": 1.7061655521392822,
      "step": 2508
    },
    {
      "epoch": 2.391815185546875e-06,
      "step": 2508,
      "training_step_time": 9.310014724731445
    },
    {
      "epoch": 2.392768859863281e-06,
      "model_forward_time": 1.689497470855713,
      "step": 2509
    },
    {
      "epoch": 2.392768859863281e-06,
      "step": 2509,
      "training_step_time": 7.488399028778076
    },
    {
      "epoch": 2.3937225341796873e-06,
      "grad_norm": 0.758589506149292,
      "learning_rate": 5.018000000000001e-06,
      "loss": 0.0946,
      "step": 2510
    },
    {
      "epoch": 2.3937225341796873e-06,
      "model_forward_time": 1.6908340454101562,
      "step": 2510
    },
    {
      "dynamics_loss_avg": 0.0718554962426424,
      "epoch": 2.3937225341796873e-06,
      "step": 2510
    },
    {
      "action_loss_avg": 0.024354901630431414,
      "epoch": 2.3937225341796873e-06,
      "step": 2510
    },
    {
      "epoch": 2.3937225341796873e-06,
      "step": 2510,
      "training_step_time": 7.426095008850098
    },
    {
      "epoch": 2.394676208496094e-06,
      "model_forward_time": 1.3152692317962646,
      "step": 2511
    },
    {
      "epoch": 2.394676208496094e-06,
      "step": 2511,
      "training_step_time": 6.772656679153442
    },
    {
      "epoch": 2.3956298828125e-06,
      "model_forward_time": 1.688000202178955,
      "step": 2512
    },
    {
      "epoch": 2.3956298828125e-06,
      "step": 2512,
      "training_step_time": 7.075721979141235
    },
    {
      "epoch": 2.396583557128906e-06,
      "model_forward_time": 1.6944830417633057,
      "step": 2513
    },
    {
      "epoch": 2.396583557128906e-06,
      "step": 2513,
      "training_step_time": 7.03496241569519
    },
    {
      "epoch": 2.3975372314453123e-06,
      "model_forward_time": 1.6849515438079834,
      "step": 2514
    },
    {
      "epoch": 2.3975372314453123e-06,
      "step": 2514,
      "training_step_time": 7.620010852813721
    },
    {
      "epoch": 2.398490905761719e-06,
      "model_forward_time": 1.692962646484375,
      "step": 2515
    },
    {
      "epoch": 2.398490905761719e-06,
      "step": 2515,
      "training_step_time": 9.721871614456177
    },
    {
      "epoch": 2.399444580078125e-06,
      "model_forward_time": 1.6881639957427979,
      "step": 2516
    },
    {
      "epoch": 2.399444580078125e-06,
      "step": 2516,
      "training_step_time": 9.286022186279297
    },
    {
      "epoch": 2.400398254394531e-06,
      "model_forward_time": 1.6914443969726562,
      "step": 2517
    },
    {
      "epoch": 2.400398254394531e-06,
      "step": 2517,
      "training_step_time": 9.471407890319824
    },
    {
      "epoch": 2.4013519287109373e-06,
      "model_forward_time": 2.224130153656006,
      "step": 2518
    },
    {
      "epoch": 2.4013519287109373e-06,
      "step": 2518,
      "training_step_time": 7.67045521736145
    },
    {
      "epoch": 2.402305603027344e-06,
      "model_forward_time": 1.6883938312530518,
      "step": 2519
    },
    {
      "epoch": 2.402305603027344e-06,
      "step": 2519,
      "training_step_time": 7.602341175079346
    },
    {
      "epoch": 2.40325927734375e-06,
      "grad_norm": 0.6857231855392456,
      "learning_rate": 5.038000000000001e-06,
      "loss": 0.1042,
      "step": 2520
    },
    {
      "epoch": 2.40325927734375e-06,
      "model_forward_time": 1.6880011558532715,
      "step": 2520
    },
    {
      "dynamics_loss_avg": 0.07907262779772281,
      "epoch": 2.40325927734375e-06,
      "step": 2520
    },
    {
      "action_loss_avg": 0.01841652486473322,
      "epoch": 2.40325927734375e-06,
      "step": 2520
    },
    {
      "epoch": 2.40325927734375e-06,
      "step": 2520,
      "training_step_time": 6.801347017288208
    },
    {
      "epoch": 2.404212951660156e-06,
      "model_forward_time": 1.6937580108642578,
      "step": 2521
    },
    {
      "epoch": 2.404212951660156e-06,
      "step": 2521,
      "training_step_time": 7.090354919433594
    },
    {
      "epoch": 2.4051666259765623e-06,
      "model_forward_time": 1.68888521194458,
      "step": 2522
    },
    {
      "epoch": 2.4051666259765623e-06,
      "step": 2522,
      "training_step_time": 7.0462729930877686
    },
    {
      "epoch": 2.406120300292969e-06,
      "model_forward_time": 1.6843748092651367,
      "step": 2523
    },
    {
      "epoch": 2.406120300292969e-06,
      "step": 2523,
      "training_step_time": 7.7289323806762695
    },
    {
      "epoch": 2.407073974609375e-06,
      "model_forward_time": 1.6925828456878662,
      "step": 2524
    },
    {
      "epoch": 2.407073974609375e-06,
      "step": 2524,
      "training_step_time": 9.111886262893677
    },
    {
      "epoch": 2.408027648925781e-06,
      "model_forward_time": 1.314133882522583,
      "step": 2525
    },
    {
      "epoch": 2.408027648925781e-06,
      "step": 2525,
      "training_step_time": 9.160108089447021
    },
    {
      "epoch": 2.4089813232421873e-06,
      "model_forward_time": 1.7032487392425537,
      "step": 2526
    },
    {
      "epoch": 2.4089813232421873e-06,
      "step": 2526,
      "training_step_time": 8.907060384750366
    },
    {
      "epoch": 2.409934997558594e-06,
      "model_forward_time": 2.231119155883789,
      "step": 2527
    },
    {
      "epoch": 2.409934997558594e-06,
      "step": 2527,
      "training_step_time": 8.57314419746399
    },
    {
      "epoch": 2.410888671875e-06,
      "model_forward_time": 1.6924188137054443,
      "step": 2528
    },
    {
      "epoch": 2.410888671875e-06,
      "step": 2528,
      "training_step_time": 7.696964740753174
    },
    {
      "epoch": 2.411842346191406e-06,
      "model_forward_time": 1.6834921836853027,
      "step": 2529
    },
    {
      "epoch": 2.411842346191406e-06,
      "step": 2529,
      "training_step_time": 6.802539110183716
    },
    {
      "epoch": 2.4127960205078123e-06,
      "grad_norm": 0.6423655152320862,
      "learning_rate": 5.0580000000000005e-06,
      "loss": 0.0997,
      "step": 2530
    },
    {
      "epoch": 2.4127960205078123e-06,
      "model_forward_time": 1.6981267929077148,
      "step": 2530
    },
    {
      "dynamics_loss_avg": 0.07604212947189808,
      "epoch": 2.4127960205078123e-06,
      "step": 2530
    },
    {
      "action_loss_avg": 0.014600456575863063,
      "epoch": 2.4127960205078123e-06,
      "step": 2530
    },
    {
      "epoch": 2.4127960205078123e-06,
      "step": 2530,
      "training_step_time": 7.122112989425659
    },
    {
      "epoch": 2.413749694824219e-06,
      "model_forward_time": 0.9801070690155029,
      "step": 2531
    },
    {
      "epoch": 2.413749694824219e-06,
      "step": 2531,
      "training_step_time": 7.06283712387085
    },
    {
      "epoch": 2.414703369140625e-06,
      "model_forward_time": 1.7001171112060547,
      "step": 2532
    },
    {
      "epoch": 2.414703369140625e-06,
      "step": 2532,
      "training_step_time": 7.789651155471802
    },
    {
      "epoch": 2.415657043457031e-06,
      "model_forward_time": 1.6956803798675537,
      "step": 2533
    },
    {
      "epoch": 2.415657043457031e-06,
      "step": 2533,
      "training_step_time": 8.264181137084961
    },
    {
      "epoch": 2.4166107177734373e-06,
      "model_forward_time": 1.6855759620666504,
      "step": 2534
    },
    {
      "epoch": 2.4166107177734373e-06,
      "step": 2534,
      "training_step_time": 8.559978485107422
    },
    {
      "epoch": 2.417564392089844e-06,
      "model_forward_time": 1.7003211975097656,
      "step": 2535
    },
    {
      "epoch": 2.417564392089844e-06,
      "step": 2535,
      "training_step_time": 9.77940034866333
    },
    {
      "epoch": 2.41851806640625e-06,
      "model_forward_time": 1.319413661956787,
      "step": 2536
    },
    {
      "epoch": 2.41851806640625e-06,
      "step": 2536,
      "training_step_time": 7.70776629447937
    },
    {
      "epoch": 2.419471740722656e-06,
      "model_forward_time": 2.235623598098755,
      "step": 2537
    },
    {
      "epoch": 2.419471740722656e-06,
      "step": 2537,
      "training_step_time": 7.644877195358276
    },
    {
      "epoch": 2.4204254150390623e-06,
      "model_forward_time": 1.6896617412567139,
      "step": 2538
    },
    {
      "epoch": 2.4204254150390623e-06,
      "step": 2538,
      "training_step_time": 6.759624481201172
    },
    {
      "epoch": 2.421379089355469e-06,
      "model_forward_time": 1.6873929500579834,
      "step": 2539
    },
    {
      "epoch": 2.421379089355469e-06,
      "step": 2539,
      "training_step_time": 7.0876383781433105
    },
    {
      "epoch": 2.422332763671875e-06,
      "grad_norm": 0.8925269246101379,
      "learning_rate": 5.078e-06,
      "loss": 0.0918,
      "step": 2540
    },
    {
      "epoch": 2.422332763671875e-06,
      "model_forward_time": 1.6894450187683105,
      "step": 2540
    },
    {
      "dynamics_loss_avg": 0.07539274487644435,
      "epoch": 2.422332763671875e-06,
      "step": 2540
    },
    {
      "action_loss_avg": 0.018569172825664282,
      "epoch": 2.422332763671875e-06,
      "step": 2540
    },
    {
      "epoch": 2.422332763671875e-06,
      "step": 2540,
      "training_step_time": 7.016802787780762
    },
    {
      "epoch": 2.423286437988281e-06,
      "model_forward_time": 1.6940689086914062,
      "step": 2541
    },
    {
      "epoch": 2.423286437988281e-06,
      "step": 2541,
      "training_step_time": 8.321984052658081
    },
    {
      "epoch": 2.4242401123046873e-06,
      "model_forward_time": 1.69236159324646,
      "step": 2542
    },
    {
      "epoch": 2.4242401123046873e-06,
      "step": 2542,
      "training_step_time": 8.267212390899658
    },
    {
      "epoch": 2.425193786621094e-06,
      "model_forward_time": 1.6966288089752197,
      "step": 2543
    },
    {
      "epoch": 2.425193786621094e-06,
      "step": 2543,
      "training_step_time": 8.668389081954956
    },
    {
      "epoch": 2.4261474609375e-06,
      "model_forward_time": 1.700059175491333,
      "step": 2544
    },
    {
      "epoch": 2.4261474609375e-06,
      "step": 2544,
      "training_step_time": 8.477009057998657
    },
    {
      "epoch": 2.427101135253906e-06,
      "model_forward_time": 1.6926379203796387,
      "step": 2545
    },
    {
      "epoch": 2.427101135253906e-06,
      "step": 2545,
      "training_step_time": 8.523765802383423
    },
    {
      "epoch": 2.4280548095703123e-06,
      "model_forward_time": 2.216265916824341,
      "step": 2546
    },
    {
      "epoch": 2.4280548095703123e-06,
      "step": 2546,
      "training_step_time": 7.641583204269409
    },
    {
      "epoch": 2.429008483886719e-06,
      "model_forward_time": 1.6906239986419678,
      "step": 2547
    },
    {
      "epoch": 2.429008483886719e-06,
      "step": 2547,
      "training_step_time": 7.554527997970581
    },
    {
      "epoch": 2.429962158203125e-06,
      "model_forward_time": 1.314880609512329,
      "step": 2548
    },
    {
      "epoch": 2.429962158203125e-06,
      "step": 2548,
      "training_step_time": 7.138483047485352
    },
    {
      "epoch": 2.430915832519531e-06,
      "model_forward_time": 1.6909148693084717,
      "step": 2549
    },
    {
      "epoch": 2.430915832519531e-06,
      "step": 2549,
      "training_step_time": 7.06586766242981
    },
    {
      "epoch": 2.4318695068359373e-06,
      "grad_norm": 0.7042389512062073,
      "learning_rate": 5.098000000000001e-06,
      "loss": 0.084,
      "step": 2550
    },
    {
      "epoch": 2.4318695068359373e-06,
      "model_forward_time": 1.6907083988189697,
      "step": 2550
    },
    {
      "dynamics_loss_avg": 0.06463074907660485,
      "epoch": 2.4318695068359373e-06,
      "step": 2550
    },
    {
      "action_loss_avg": 0.010930280899628997,
      "epoch": 2.4318695068359373e-06,
      "step": 2550
    },
    {
      "epoch": 2.4318695068359373e-06,
      "step": 2550,
      "training_step_time": 7.643922567367554
    },
    {
      "epoch": 2.432823181152344e-06,
      "model_forward_time": 1.689863681793213,
      "step": 2551
    },
    {
      "epoch": 2.432823181152344e-06,
      "step": 2551,
      "training_step_time": 7.609598636627197
    },
    {
      "epoch": 2.43377685546875e-06,
      "model_forward_time": 1.69173002243042,
      "step": 2552
    },
    {
      "epoch": 2.43377685546875e-06,
      "step": 2552,
      "training_step_time": 8.682353019714355
    },
    {
      "epoch": 2.434730529785156e-06,
      "model_forward_time": 1.69834303855896,
      "step": 2553
    },
    {
      "epoch": 2.434730529785156e-06,
      "step": 2553,
      "training_step_time": 10.974292755126953
    },
    {
      "epoch": 2.4356842041015623e-06,
      "model_forward_time": 1.688208818435669,
      "step": 2554
    },
    {
      "epoch": 2.4356842041015623e-06,
      "step": 2554,
      "training_step_time": 7.848376989364624
    },
    {
      "epoch": 2.436637878417969e-06,
      "model_forward_time": 2.2262399196624756,
      "step": 2555
    },
    {
      "epoch": 2.436637878417969e-06,
      "step": 2555,
      "training_step_time": 8.814478635787964
    },
    {
      "epoch": 2.437591552734375e-06,
      "model_forward_time": 1.690382719039917,
      "step": 2556
    },
    {
      "epoch": 2.437591552734375e-06,
      "step": 2556,
      "training_step_time": 7.652013063430786
    },
    {
      "epoch": 2.438545227050781e-06,
      "model_forward_time": 1.696094274520874,
      "step": 2557
    },
    {
      "epoch": 2.438545227050781e-06,
      "step": 2557,
      "training_step_time": 6.766810655593872
    },
    {
      "epoch": 2.4394989013671873e-06,
      "model_forward_time": 1.685065746307373,
      "step": 2558
    },
    {
      "epoch": 2.4394989013671873e-06,
      "step": 2558,
      "training_step_time": 7.089544296264648
    },
    {
      "epoch": 2.440452575683594e-06,
      "model_forward_time": 1.6925814151763916,
      "step": 2559
    },
    {
      "epoch": 2.440452575683594e-06,
      "step": 2559,
      "training_step_time": 7.6860692501068115
    },
    {
      "epoch": 2.44140625e-06,
      "grad_norm": 0.8160154223442078,
      "learning_rate": 5.118000000000001e-06,
      "loss": 0.0789,
      "step": 2560
    },
    {
      "epoch": 2.44140625e-06,
      "model_forward_time": 1.6898479461669922,
      "step": 2560
    },
    {
      "dynamics_loss_avg": 0.0620355786755681,
      "epoch": 2.44140625e-06,
      "step": 2560
    },
    {
      "action_loss_avg": 0.012186298752203584,
      "epoch": 2.44140625e-06,
      "step": 2560
    },
    {
      "epoch": 2.44140625e-06,
      "step": 2560,
      "training_step_time": 7.6094276905059814
    },
    {
      "epoch": 2.442359924316406e-06,
      "model_forward_time": 1.6794238090515137,
      "step": 2561
    },
    {
      "epoch": 2.442359924316406e-06,
      "step": 2561,
      "training_step_time": 7.6997480392456055
    },
    {
      "epoch": 2.4433135986328123e-06,
      "model_forward_time": 1.6980485916137695,
      "step": 2562
    },
    {
      "epoch": 2.4433135986328123e-06,
      "step": 2562,
      "training_step_time": 10.517239809036255
    },
    {
      "epoch": 2.444267272949219e-06,
      "model_forward_time": 1.6834051609039307,
      "step": 2563
    },
    {
      "epoch": 2.444267272949219e-06,
      "step": 2563,
      "training_step_time": 7.795189380645752
    },
    {
      "epoch": 2.445220947265625e-06,
      "model_forward_time": 2.2179956436157227,
      "step": 2564
    },
    {
      "epoch": 2.445220947265625e-06,
      "step": 2564,
      "training_step_time": 7.6734514236450195
    },
    {
      "epoch": 2.446174621582031e-06,
      "model_forward_time": 1.6928865909576416,
      "step": 2565
    },
    {
      "epoch": 2.446174621582031e-06,
      "step": 2565,
      "training_step_time": 7.6630449295043945
    },
    {
      "epoch": 2.4471282958984373e-06,
      "model_forward_time": 1.6876220703125,
      "step": 2566
    },
    {
      "epoch": 2.4471282958984373e-06,
      "step": 2566,
      "training_step_time": 6.773109674453735
    },
    {
      "epoch": 2.448081970214844e-06,
      "model_forward_time": 1.693464756011963,
      "step": 2567
    },
    {
      "epoch": 2.448081970214844e-06,
      "step": 2567,
      "training_step_time": 7.034149169921875
    },
    {
      "epoch": 2.44903564453125e-06,
      "model_forward_time": 1.6889371871948242,
      "step": 2568
    },
    {
      "epoch": 2.44903564453125e-06,
      "step": 2568,
      "training_step_time": 7.561504602432251
    },
    {
      "epoch": 2.449989318847656e-06,
      "model_forward_time": 1.693521499633789,
      "step": 2569
    },
    {
      "epoch": 2.449989318847656e-06,
      "step": 2569,
      "training_step_time": 7.595430850982666
    },
    {
      "epoch": 2.4509429931640623e-06,
      "grad_norm": 1.0553957223892212,
      "learning_rate": 5.138000000000001e-06,
      "loss": 0.088,
      "step": 2570
    },
    {
      "epoch": 2.4509429931640623e-06,
      "model_forward_time": 1.673522710800171,
      "step": 2570
    },
    {
      "dynamics_loss_avg": 0.03922051228582859,
      "epoch": 2.4509429931640623e-06,
      "step": 2570
    },
    {
      "action_loss_avg": 0.007931359903886914,
      "epoch": 2.4509429931640623e-06,
      "step": 2570
    },
    {
      "epoch": 2.4509429931640623e-06,
      "step": 2570,
      "training_step_time": 8.378236055374146
    },
    {
      "epoch": 2.451896667480469e-06,
      "model_forward_time": 1.6908137798309326,
      "step": 2571
    },
    {
      "epoch": 2.451896667480469e-06,
      "step": 2571,
      "training_step_time": 9.351406574249268
    },
    {
      "epoch": 2.452850341796875e-06,
      "model_forward_time": 1.3302550315856934,
      "step": 2572
    },
    {
      "epoch": 2.452850341796875e-06,
      "step": 2572,
      "training_step_time": 8.336785316467285
    },
    {
      "epoch": 2.453804016113281e-06,
      "model_forward_time": 1.7041733264923096,
      "step": 2573
    },
    {
      "epoch": 2.453804016113281e-06,
      "step": 2573,
      "training_step_time": 7.605603933334351
    },
    {
      "epoch": 2.4547576904296874e-06,
      "model_forward_time": 2.2269632816314697,
      "step": 2574
    },
    {
      "epoch": 2.4547576904296874e-06,
      "step": 2574,
      "training_step_time": 7.643052339553833
    },
    {
      "epoch": 2.455711364746094e-06,
      "model_forward_time": 1.6962542533874512,
      "step": 2575
    },
    {
      "epoch": 2.455711364746094e-06,
      "step": 2575,
      "training_step_time": 6.81380295753479
    },
    {
      "epoch": 2.4566650390625e-06,
      "model_forward_time": 1.6936876773834229,
      "step": 2576
    },
    {
      "epoch": 2.4566650390625e-06,
      "step": 2576,
      "training_step_time": 7.119609117507935
    },
    {
      "epoch": 2.457618713378906e-06,
      "model_forward_time": 1.690849781036377,
      "step": 2577
    },
    {
      "epoch": 2.457618713378906e-06,
      "step": 2577,
      "training_step_time": 7.279149055480957
    },
    {
      "epoch": 2.4585723876953124e-06,
      "model_forward_time": 1.690476894378662,
      "step": 2578
    },
    {
      "epoch": 2.4585723876953124e-06,
      "step": 2578,
      "training_step_time": 8.499873161315918
    },
    {
      "epoch": 2.459526062011719e-06,
      "model_forward_time": 1.703601598739624,
      "step": 2579
    },
    {
      "epoch": 2.459526062011719e-06,
      "step": 2579,
      "training_step_time": 7.590351581573486
    },
    {
      "epoch": 2.460479736328125e-06,
      "grad_norm": 0.3835853934288025,
      "learning_rate": 5.158e-06,
      "loss": 0.0885,
      "step": 2580
    },
    {
      "epoch": 2.460479736328125e-06,
      "model_forward_time": 1.689629316329956,
      "step": 2580
    },
    {
      "dynamics_loss_avg": 0.07649397812783718,
      "epoch": 2.460479736328125e-06,
      "step": 2580
    },
    {
      "action_loss_avg": 0.014080674387514591,
      "epoch": 2.460479736328125e-06,
      "step": 2580
    },
    {
      "epoch": 2.460479736328125e-06,
      "step": 2580,
      "training_step_time": 9.668680667877197
    },
    {
      "epoch": 2.4614334106445312e-06,
      "model_forward_time": 1.7066361904144287,
      "step": 2581
    },
    {
      "epoch": 2.4614334106445312e-06,
      "step": 2581,
      "training_step_time": 8.30357313156128
    },
    {
      "epoch": 2.4623870849609374e-06,
      "model_forward_time": 1.7001993656158447,
      "step": 2582
    },
    {
      "epoch": 2.4623870849609374e-06,
      "step": 2582,
      "training_step_time": 9.352259397506714
    },
    {
      "epoch": 2.463340759277344e-06,
      "model_forward_time": 2.2195496559143066,
      "step": 2583
    },
    {
      "epoch": 2.463340759277344e-06,
      "step": 2583,
      "training_step_time": 8.583394527435303
    },
    {
      "epoch": 2.46429443359375e-06,
      "model_forward_time": 1.6885249614715576,
      "step": 2584
    },
    {
      "epoch": 2.46429443359375e-06,
      "step": 2584,
      "training_step_time": 7.627896308898926
    },
    {
      "epoch": 2.4652481079101562e-06,
      "model_forward_time": 0.9693601131439209,
      "step": 2585
    },
    {
      "epoch": 2.4652481079101562e-06,
      "step": 2585,
      "training_step_time": 8.112190961837769
    },
    {
      "epoch": 2.4662017822265624e-06,
      "model_forward_time": 1.6919033527374268,
      "step": 2586
    },
    {
      "epoch": 2.4662017822265624e-06,
      "step": 2586,
      "training_step_time": 7.335798740386963
    },
    {
      "epoch": 2.467155456542969e-06,
      "model_forward_time": 1.6917533874511719,
      "step": 2587
    },
    {
      "epoch": 2.467155456542969e-06,
      "step": 2587,
      "training_step_time": 7.740149736404419
    },
    {
      "epoch": 2.468109130859375e-06,
      "model_forward_time": 1.6921021938323975,
      "step": 2588
    },
    {
      "epoch": 2.468109130859375e-06,
      "step": 2588,
      "training_step_time": 7.610990524291992
    },
    {
      "epoch": 2.4690628051757812e-06,
      "model_forward_time": 1.693704605102539,
      "step": 2589
    },
    {
      "epoch": 2.4690628051757812e-06,
      "step": 2589,
      "training_step_time": 9.504944801330566
    },
    {
      "epoch": 2.4700164794921874e-06,
      "grad_norm": 0.37051406502723694,
      "learning_rate": 5.178000000000001e-06,
      "loss": 0.0843,
      "step": 2590
    },
    {
      "epoch": 2.4700164794921874e-06,
      "model_forward_time": 1.7061808109283447,
      "step": 2590
    },
    {
      "dynamics_loss_avg": 0.06300594955682755,
      "epoch": 2.4700164794921874e-06,
      "step": 2590
    },
    {
      "action_loss_avg": 0.01438421248458326,
      "epoch": 2.4700164794921874e-06,
      "step": 2590
    },
    {
      "epoch": 2.4700164794921874e-06,
      "step": 2590,
      "training_step_time": 10.034068822860718
    },
    {
      "epoch": 2.470970153808594e-06,
      "model_forward_time": 1.6899089813232422,
      "step": 2591
    },
    {
      "epoch": 2.470970153808594e-06,
      "step": 2591,
      "training_step_time": 8.573951482772827
    },
    {
      "epoch": 2.471923828125e-06,
      "model_forward_time": 1.6908442974090576,
      "step": 2592
    },
    {
      "epoch": 2.471923828125e-06,
      "step": 2592,
      "training_step_time": 8.280604839324951
    },
    {
      "epoch": 2.4728775024414062e-06,
      "model_forward_time": 1.6900076866149902,
      "step": 2593
    },
    {
      "epoch": 2.4728775024414062e-06,
      "step": 2593,
      "training_step_time": 7.668097257614136
    },
    {
      "epoch": 2.4738311767578124e-06,
      "model_forward_time": 1.6898488998413086,
      "step": 2594
    },
    {
      "epoch": 2.4738311767578124e-06,
      "step": 2594,
      "training_step_time": 6.777582883834839
    },
    {
      "epoch": 2.474784851074219e-06,
      "model_forward_time": 1.6875743865966797,
      "step": 2595
    },
    {
      "epoch": 2.474784851074219e-06,
      "step": 2595,
      "training_step_time": 7.340139389038086
    },
    {
      "epoch": 2.475738525390625e-06,
      "model_forward_time": 1.680866003036499,
      "step": 2596
    },
    {
      "epoch": 2.475738525390625e-06,
      "step": 2596,
      "training_step_time": 7.658099889755249
    },
    {
      "epoch": 2.4766921997070312e-06,
      "model_forward_time": 1.6891076564788818,
      "step": 2597
    },
    {
      "epoch": 2.4766921997070312e-06,
      "step": 2597,
      "training_step_time": 8.515058517456055
    },
    {
      "epoch": 2.4776458740234374e-06,
      "model_forward_time": 1.688873529434204,
      "step": 2598
    },
    {
      "epoch": 2.4776458740234374e-06,
      "step": 2598,
      "training_step_time": 8.081358194351196
    },
    {
      "epoch": 2.478599548339844e-06,
      "model_forward_time": 1.6916358470916748,
      "step": 2599
    },
    {
      "epoch": 2.478599548339844e-06,
      "step": 2599,
      "training_step_time": 9.423574924468994
    },
    {
      "epoch": 2.47955322265625e-06,
      "grad_norm": 0.6743723750114441,
      "learning_rate": 5.198000000000001e-06,
      "loss": 0.0869,
      "step": 2600
    },
    {
      "epoch": 2.47955322265625e-06,
      "model_forward_time": 1.6872541904449463,
      "step": 2600
    },
    {
      "dynamics_loss_avg": 0.08325716480612755,
      "epoch": 2.47955322265625e-06,
      "step": 2600
    },
    {
      "action_loss_avg": 0.01865784772671759,
      "epoch": 2.47955322265625e-06,
      "step": 2600
    },
    {
      "epoch": 2.47955322265625e-06,
      "step": 2600,
      "training_step_time": 8.380911350250244
    },
    {
      "epoch": 2.4805068969726562e-06,
      "model_forward_time": 1.702261209487915,
      "step": 2601
    },
    {
      "epoch": 2.4805068969726562e-06,
      "step": 2601,
      "training_step_time": 7.706118822097778
    },
    {
      "epoch": 2.4814605712890624e-06,
      "model_forward_time": 2.216097116470337,
      "step": 2602
    },
    {
      "epoch": 2.4814605712890624e-06,
      "step": 2602,
      "training_step_time": 7.55897331237793
    },
    {
      "epoch": 2.482414245605469e-06,
      "model_forward_time": 1.688697099685669,
      "step": 2603
    },
    {
      "epoch": 2.482414245605469e-06,
      "step": 2603,
      "training_step_time": 6.784624814987183
    },
    {
      "epoch": 2.483367919921875e-06,
      "model_forward_time": 1.6979317665100098,
      "step": 2604
    },
    {
      "epoch": 2.483367919921875e-06,
      "step": 2604,
      "training_step_time": 7.346060276031494
    },
    {
      "epoch": 2.4843215942382812e-06,
      "model_forward_time": 1.6898510456085205,
      "step": 2605
    },
    {
      "epoch": 2.4843215942382812e-06,
      "step": 2605,
      "training_step_time": 7.692766427993774
    },
    {
      "epoch": 2.4852752685546874e-06,
      "model_forward_time": 1.6871428489685059,
      "step": 2606
    },
    {
      "epoch": 2.4852752685546874e-06,
      "step": 2606,
      "training_step_time": 7.629308462142944
    },
    {
      "epoch": 2.486228942871094e-06,
      "model_forward_time": 1.6893589496612549,
      "step": 2607
    },
    {
      "epoch": 2.486228942871094e-06,
      "step": 2607,
      "training_step_time": 9.2377188205719
    },
    {
      "epoch": 2.4871826171875e-06,
      "model_forward_time": 1.6862521171569824,
      "step": 2608
    },
    {
      "epoch": 2.4871826171875e-06,
      "step": 2608,
      "training_step_time": 9.913532733917236
    },
    {
      "epoch": 2.4881362915039062e-06,
      "model_forward_time": 1.690732479095459,
      "step": 2609
    },
    {
      "epoch": 2.4881362915039062e-06,
      "step": 2609,
      "training_step_time": 7.82021689414978
    },
    {
      "epoch": 2.4890899658203124e-06,
      "grad_norm": 0.8195958137512207,
      "learning_rate": 5.218000000000001e-06,
      "loss": 0.097,
      "step": 2610
    },
    {
      "epoch": 2.4890899658203124e-06,
      "model_forward_time": 1.7026879787445068,
      "step": 2610
    },
    {
      "dynamics_loss_avg": 0.06019201241433621,
      "epoch": 2.4890899658203124e-06,
      "step": 2610
    },
    {
      "action_loss_avg": 0.016471588239073753,
      "epoch": 2.4890899658203124e-06,
      "step": 2610
    },
    {
      "epoch": 2.4890899658203124e-06,
      "step": 2610,
      "training_step_time": 7.654978513717651
    },
    {
      "epoch": 2.490043640136719e-06,
      "model_forward_time": 2.224095344543457,
      "step": 2611
    },
    {
      "epoch": 2.490043640136719e-06,
      "step": 2611,
      "training_step_time": 7.90584135055542
    },
    {
      "epoch": 2.490997314453125e-06,
      "model_forward_time": 1.6883952617645264,
      "step": 2612
    },
    {
      "epoch": 2.490997314453125e-06,
      "step": 2612,
      "training_step_time": 6.878453493118286
    },
    {
      "epoch": 2.4919509887695312e-06,
      "model_forward_time": 1.685136318206787,
      "step": 2613
    },
    {
      "epoch": 2.4919509887695312e-06,
      "step": 2613,
      "training_step_time": 7.4552366733551025
    },
    {
      "epoch": 2.4929046630859374e-06,
      "model_forward_time": 1.6871745586395264,
      "step": 2614
    },
    {
      "epoch": 2.4929046630859374e-06,
      "step": 2614,
      "training_step_time": 7.282031059265137
    },
    {
      "epoch": 2.493858337402344e-06,
      "model_forward_time": 1.6897883415222168,
      "step": 2615
    },
    {
      "epoch": 2.493858337402344e-06,
      "step": 2615,
      "training_step_time": 7.7295849323272705
    },
    {
      "epoch": 2.49481201171875e-06,
      "model_forward_time": 1.6910786628723145,
      "step": 2616
    },
    {
      "epoch": 2.49481201171875e-06,
      "step": 2616,
      "training_step_time": 7.6415934562683105
    },
    {
      "epoch": 2.4957656860351562e-06,
      "model_forward_time": 1.6874830722808838,
      "step": 2617
    },
    {
      "epoch": 2.4957656860351562e-06,
      "step": 2617,
      "training_step_time": 8.717785120010376
    },
    {
      "epoch": 2.4967193603515624e-06,
      "model_forward_time": 1.6940219402313232,
      "step": 2618
    },
    {
      "epoch": 2.4967193603515624e-06,
      "step": 2618,
      "training_step_time": 7.777311325073242
    },
    {
      "epoch": 2.497673034667969e-06,
      "model_forward_time": 1.6943199634552002,
      "step": 2619
    },
    {
      "epoch": 2.497673034667969e-06,
      "step": 2619,
      "training_step_time": 7.656343936920166
    },
    {
      "epoch": 2.498626708984375e-06,
      "grad_norm": 0.7897319197654724,
      "learning_rate": 5.2380000000000005e-06,
      "loss": 0.1,
      "step": 2620
    },
    {
      "epoch": 2.498626708984375e-06,
      "model_forward_time": 2.2181687355041504,
      "step": 2620
    },
    {
      "dynamics_loss_avg": 0.07306436710059643,
      "epoch": 2.498626708984375e-06,
      "step": 2620
    },
    {
      "action_loss_avg": 0.012081410037353635,
      "epoch": 2.498626708984375e-06,
      "step": 2620
    },
    {
      "epoch": 2.498626708984375e-06,
      "step": 2620,
      "training_step_time": 7.708298921585083
    },
    {
      "epoch": 2.4995803833007812e-06,
      "model_forward_time": 1.7027575969696045,
      "step": 2621
    },
    {
      "epoch": 2.4995803833007812e-06,
      "step": 2621,
      "training_step_time": 7.502511024475098
    },
    {
      "epoch": 2.5005340576171874e-06,
      "model_forward_time": 1.6967792510986328,
      "step": 2622
    },
    {
      "epoch": 2.5005340576171874e-06,
      "step": 2622,
      "training_step_time": 7.335740804672241
    },
    {
      "epoch": 2.501487731933594e-06,
      "model_forward_time": 1.6980810165405273,
      "step": 2623
    },
    {
      "epoch": 2.501487731933594e-06,
      "step": 2623,
      "training_step_time": 7.045049428939819
    },
    {
      "epoch": 2.50244140625e-06,
      "model_forward_time": 1.698645830154419,
      "step": 2624
    },
    {
      "epoch": 2.50244140625e-06,
      "step": 2624,
      "training_step_time": 8.70746397972107
    },
    {
      "epoch": 2.5033950805664062e-06,
      "model_forward_time": 1.6980743408203125,
      "step": 2625
    },
    {
      "epoch": 2.5033950805664062e-06,
      "step": 2625,
      "training_step_time": 9.339348554611206
    },
    {
      "epoch": 2.5043487548828124e-06,
      "model_forward_time": 1.6961312294006348,
      "step": 2626
    },
    {
      "epoch": 2.5043487548828124e-06,
      "step": 2626,
      "training_step_time": 8.630478382110596
    },
    {
      "epoch": 2.5053024291992185e-06,
      "model_forward_time": 1.692624807357788,
      "step": 2627
    },
    {
      "epoch": 2.5053024291992185e-06,
      "step": 2627,
      "training_step_time": 8.789400577545166
    },
    {
      "epoch": 2.506256103515625e-06,
      "model_forward_time": 1.7081372737884521,
      "step": 2628
    },
    {
      "epoch": 2.506256103515625e-06,
      "step": 2628,
      "training_step_time": 7.691805124282837
    },
    {
      "epoch": 2.5072097778320313e-06,
      "model_forward_time": 2.237610101699829,
      "step": 2629
    },
    {
      "epoch": 2.5072097778320313e-06,
      "step": 2629,
      "training_step_time": 7.5834267139434814
    },
    {
      "epoch": 2.5081634521484374e-06,
      "grad_norm": 0.5741244554519653,
      "learning_rate": 5.258000000000001e-06,
      "loss": 0.0884,
      "step": 2630
    },
    {
      "epoch": 2.5081634521484374e-06,
      "model_forward_time": 1.6928215026855469,
      "step": 2630
    },
    {
      "dynamics_loss_avg": 0.0604500662535429,
      "epoch": 2.5081634521484374e-06,
      "step": 2630
    },
    {
      "action_loss_avg": 0.011743199080228806,
      "epoch": 2.5081634521484374e-06,
      "step": 2630
    },
    {
      "epoch": 2.5081634521484374e-06,
      "step": 2630,
      "training_step_time": 7.662482023239136
    },
    {
      "epoch": 2.5091171264648435e-06,
      "model_forward_time": 1.6985840797424316,
      "step": 2631
    },
    {
      "epoch": 2.5091171264648435e-06,
      "step": 2631,
      "training_step_time": 7.861359357833862
    },
    {
      "epoch": 2.51007080078125e-06,
      "model_forward_time": 1.6924731731414795,
      "step": 2632
    },
    {
      "epoch": 2.51007080078125e-06,
      "step": 2632,
      "training_step_time": 7.04446268081665
    },
    {
      "epoch": 2.5110244750976563e-06,
      "model_forward_time": 1.6946392059326172,
      "step": 2633
    },
    {
      "epoch": 2.5110244750976563e-06,
      "step": 2633,
      "training_step_time": 7.800657510757446
    },
    {
      "epoch": 2.5119781494140624e-06,
      "model_forward_time": 1.694037914276123,
      "step": 2634
    },
    {
      "epoch": 2.5119781494140624e-06,
      "step": 2634,
      "training_step_time": 7.589033842086792
    },
    {
      "epoch": 2.5129318237304685e-06,
      "model_forward_time": 1.6961438655853271,
      "step": 2635
    },
    {
      "epoch": 2.5129318237304685e-06,
      "step": 2635,
      "training_step_time": 8.593721866607666
    },
    {
      "epoch": 2.513885498046875e-06,
      "model_forward_time": 1.6882312297821045,
      "step": 2636
    },
    {
      "epoch": 2.513885498046875e-06,
      "step": 2636,
      "training_step_time": 9.243239164352417
    },
    {
      "epoch": 2.5148391723632813e-06,
      "model_forward_time": 1.6871206760406494,
      "step": 2637
    },
    {
      "epoch": 2.5148391723632813e-06,
      "step": 2637,
      "training_step_time": 7.81547737121582
    },
    {
      "epoch": 2.5157928466796874e-06,
      "model_forward_time": 1.748485803604126,
      "step": 2638
    },
    {
      "epoch": 2.5157928466796874e-06,
      "step": 2638,
      "training_step_time": 8.111199855804443
    },
    {
      "epoch": 2.5167465209960936e-06,
      "model_forward_time": 1.7164430618286133,
      "step": 2639
    },
    {
      "epoch": 2.5167465209960936e-06,
      "step": 2639,
      "training_step_time": 7.667152643203735
    },
    {
      "epoch": 2.5177001953125e-06,
      "grad_norm": 0.5137100219726562,
      "learning_rate": 5.278000000000001e-06,
      "loss": 0.0841,
      "step": 2640
    },
    {
      "epoch": 2.5177001953125e-06,
      "model_forward_time": 1.6922564506530762,
      "step": 2640
    },
    {
      "dynamics_loss_avg": 0.06366793923079968,
      "epoch": 2.5177001953125e-06,
      "step": 2640
    },
    {
      "action_loss_avg": 0.011154521629214287,
      "epoch": 2.5177001953125e-06,
      "step": 2640
    },
    {
      "epoch": 2.5177001953125e-06,
      "step": 2640,
      "training_step_time": 7.306114673614502
    },
    {
      "epoch": 2.5186538696289063e-06,
      "model_forward_time": 1.6918692588806152,
      "step": 2641
    },
    {
      "epoch": 2.5186538696289063e-06,
      "step": 2641,
      "training_step_time": 7.101043939590454
    },
    {
      "epoch": 2.5196075439453124e-06,
      "model_forward_time": 1.6889426708221436,
      "step": 2642
    },
    {
      "epoch": 2.5196075439453124e-06,
      "step": 2642,
      "training_step_time": 7.795217752456665
    },
    {
      "epoch": 2.5205612182617186e-06,
      "model_forward_time": 1.6930465698242188,
      "step": 2643
    },
    {
      "epoch": 2.5205612182617186e-06,
      "step": 2643,
      "training_step_time": 7.60092830657959
    },
    {
      "epoch": 2.521514892578125e-06,
      "model_forward_time": 1.3288779258728027,
      "step": 2644
    },
    {
      "epoch": 2.521514892578125e-06,
      "step": 2644,
      "training_step_time": 9.480119705200195
    },
    {
      "epoch": 2.5224685668945313e-06,
      "model_forward_time": 1.6951844692230225,
      "step": 2645
    },
    {
      "epoch": 2.5224685668945313e-06,
      "step": 2645,
      "training_step_time": 8.974517107009888
    },
    {
      "epoch": 2.5234222412109374e-06,
      "model_forward_time": 1.688232421875,
      "step": 2646
    },
    {
      "epoch": 2.5234222412109374e-06,
      "step": 2646,
      "training_step_time": 9.224376678466797
    },
    {
      "epoch": 2.5243759155273436e-06,
      "model_forward_time": 1.7506515979766846,
      "step": 2647
    },
    {
      "epoch": 2.5243759155273436e-06,
      "step": 2647,
      "training_step_time": 7.661852836608887
    },
    {
      "epoch": 2.52532958984375e-06,
      "model_forward_time": 2.2468807697296143,
      "step": 2648
    },
    {
      "epoch": 2.52532958984375e-06,
      "step": 2648,
      "training_step_time": 7.562711000442505
    },
    {
      "epoch": 2.5262832641601563e-06,
      "model_forward_time": 1.6956024169921875,
      "step": 2649
    },
    {
      "epoch": 2.5262832641601563e-06,
      "step": 2649,
      "training_step_time": 7.335357666015625
    },
    {
      "epoch": 2.5272369384765624e-06,
      "grad_norm": 0.6608676314353943,
      "learning_rate": 5.298000000000001e-06,
      "loss": 0.0839,
      "step": 2650
    },
    {
      "epoch": 2.5272369384765624e-06,
      "model_forward_time": 1.6971473693847656,
      "step": 2650
    },
    {
      "dynamics_loss_avg": 0.07364092003554105,
      "epoch": 2.5272369384765624e-06,
      "step": 2650
    },
    {
      "action_loss_avg": 0.013097899779677392,
      "epoch": 2.5272369384765624e-06,
      "step": 2650
    },
    {
      "epoch": 2.5272369384765624e-06,
      "step": 2650,
      "training_step_time": 7.080629348754883
    },
    {
      "epoch": 2.5281906127929686e-06,
      "model_forward_time": 1.6934750080108643,
      "step": 2651
    },
    {
      "epoch": 2.5281906127929686e-06,
      "step": 2651,
      "training_step_time": 7.760936498641968
    },
    {
      "epoch": 2.529144287109375e-06,
      "model_forward_time": 1.691795825958252,
      "step": 2652
    },
    {
      "epoch": 2.529144287109375e-06,
      "step": 2652,
      "training_step_time": 7.5406389236450195
    },
    {
      "epoch": 2.5300979614257813e-06,
      "model_forward_time": 1.6945600509643555,
      "step": 2653
    },
    {
      "epoch": 2.5300979614257813e-06,
      "step": 2653,
      "training_step_time": 8.112545490264893
    },
    {
      "epoch": 2.5310516357421874e-06,
      "model_forward_time": 1.6899762153625488,
      "step": 2654
    },
    {
      "epoch": 2.5310516357421874e-06,
      "step": 2654,
      "training_step_time": 8.169728755950928
    },
    {
      "epoch": 2.5320053100585936e-06,
      "model_forward_time": 1.6939880847930908,
      "step": 2655
    },
    {
      "epoch": 2.5320053100585936e-06,
      "step": 2655,
      "training_step_time": 7.852781295776367
    },
    {
      "epoch": 2.532958984375e-06,
      "model_forward_time": 1.7358770370483398,
      "step": 2656
    },
    {
      "epoch": 2.532958984375e-06,
      "step": 2656,
      "training_step_time": 8.524857521057129
    },
    {
      "epoch": 2.5339126586914063e-06,
      "model_forward_time": 2.222180128097534,
      "step": 2657
    },
    {
      "epoch": 2.5339126586914063e-06,
      "step": 2657,
      "training_step_time": 7.762476921081543
    },
    {
      "epoch": 2.5348663330078124e-06,
      "model_forward_time": 1.6902024745941162,
      "step": 2658
    },
    {
      "epoch": 2.5348663330078124e-06,
      "step": 2658,
      "training_step_time": 7.3998236656188965
    },
    {
      "epoch": 2.5358200073242186e-06,
      "model_forward_time": 1.707442045211792,
      "step": 2659
    },
    {
      "epoch": 2.5358200073242186e-06,
      "step": 2659,
      "training_step_time": 7.086309194564819
    },
    {
      "epoch": 2.536773681640625e-06,
      "grad_norm": 0.8986786007881165,
      "learning_rate": 5.318000000000001e-06,
      "loss": 0.0847,
      "step": 2660
    },
    {
      "epoch": 2.536773681640625e-06,
      "model_forward_time": 1.6976606845855713,
      "step": 2660
    },
    {
      "dynamics_loss_avg": 0.04748948998749256,
      "epoch": 2.536773681640625e-06,
      "step": 2660
    },
    {
      "action_loss_avg": 0.008183235675096512,
      "epoch": 2.536773681640625e-06,
      "step": 2660
    },
    {
      "epoch": 2.536773681640625e-06,
      "step": 2660,
      "training_step_time": 7.742777347564697
    },
    {
      "epoch": 2.5377273559570313e-06,
      "model_forward_time": 1.690117359161377,
      "step": 2661
    },
    {
      "epoch": 2.5377273559570313e-06,
      "step": 2661,
      "training_step_time": 7.560842990875244
    },
    {
      "epoch": 2.5386810302734374e-06,
      "model_forward_time": 1.6985414028167725,
      "step": 2662
    },
    {
      "epoch": 2.5386810302734374e-06,
      "step": 2662,
      "training_step_time": 8.543603897094727
    },
    {
      "epoch": 2.5396347045898436e-06,
      "model_forward_time": 1.6910293102264404,
      "step": 2663
    },
    {
      "epoch": 2.5396347045898436e-06,
      "step": 2663,
      "training_step_time": 7.607834577560425
    },
    {
      "epoch": 2.54058837890625e-06,
      "model_forward_time": 1.6963622570037842,
      "step": 2664
    },
    {
      "epoch": 2.54058837890625e-06,
      "step": 2664,
      "training_step_time": 7.84669828414917
    },
    {
      "epoch": 2.5415420532226563e-06,
      "model_forward_time": 1.7203402519226074,
      "step": 2665
    },
    {
      "epoch": 2.5415420532226563e-06,
      "step": 2665,
      "training_step_time": 7.683104515075684
    },
    {
      "epoch": 2.5424957275390624e-06,
      "model_forward_time": 2.2271924018859863,
      "step": 2666
    },
    {
      "epoch": 2.5424957275390624e-06,
      "step": 2666,
      "training_step_time": 7.6905457973480225
    },
    {
      "epoch": 2.5434494018554686e-06,
      "model_forward_time": 1.6890368461608887,
      "step": 2667
    },
    {
      "epoch": 2.5434494018554686e-06,
      "step": 2667,
      "training_step_time": 7.570209741592407
    },
    {
      "epoch": 2.544403076171875e-06,
      "model_forward_time": 1.7081718444824219,
      "step": 2668
    },
    {
      "epoch": 2.544403076171875e-06,
      "step": 2668,
      "training_step_time": 7.0577003955841064
    },
    {
      "epoch": 2.5453567504882813e-06,
      "model_forward_time": 1.693819522857666,
      "step": 2669
    },
    {
      "epoch": 2.5453567504882813e-06,
      "step": 2669,
      "training_step_time": 7.776609897613525
    },
    {
      "epoch": 2.5463104248046874e-06,
      "grad_norm": 0.6572678685188293,
      "learning_rate": 5.338000000000001e-06,
      "loss": 0.0811,
      "step": 2670
    },
    {
      "epoch": 2.5463104248046874e-06,
      "model_forward_time": 1.6867620944976807,
      "step": 2670
    },
    {
      "dynamics_loss_avg": 0.07020233925431967,
      "epoch": 2.5463104248046874e-06,
      "step": 2670
    },
    {
      "action_loss_avg": 0.02305629588663578,
      "epoch": 2.5463104248046874e-06,
      "step": 2670
    },
    {
      "epoch": 2.5463104248046874e-06,
      "step": 2670,
      "training_step_time": 7.5142982006073
    },
    {
      "epoch": 2.5472640991210936e-06,
      "model_forward_time": 1.691103458404541,
      "step": 2671
    },
    {
      "epoch": 2.5472640991210936e-06,
      "step": 2671,
      "training_step_time": 8.396859645843506
    },
    {
      "epoch": 2.5482177734375e-06,
      "model_forward_time": 1.6851530075073242,
      "step": 2672
    },
    {
      "epoch": 2.5482177734375e-06,
      "step": 2672,
      "training_step_time": 7.639802694320679
    },
    {
      "epoch": 2.5491714477539063e-06,
      "model_forward_time": 1.691509485244751,
      "step": 2673
    },
    {
      "epoch": 2.5491714477539063e-06,
      "step": 2673,
      "training_step_time": 7.77980899810791
    },
    {
      "epoch": 2.5501251220703124e-06,
      "model_forward_time": 1.6917531490325928,
      "step": 2674
    },
    {
      "epoch": 2.5501251220703124e-06,
      "step": 2674,
      "training_step_time": 7.742663145065308
    },
    {
      "epoch": 2.5510787963867186e-06,
      "model_forward_time": 2.248774766921997,
      "step": 2675
    },
    {
      "epoch": 2.5510787963867186e-06,
      "step": 2675,
      "training_step_time": 7.749938726425171
    },
    {
      "epoch": 2.552032470703125e-06,
      "model_forward_time": 1.6887950897216797,
      "step": 2676
    },
    {
      "epoch": 2.552032470703125e-06,
      "step": 2676,
      "training_step_time": 7.482177257537842
    },
    {
      "epoch": 2.5529861450195313e-06,
      "model_forward_time": 1.6904845237731934,
      "step": 2677
    },
    {
      "epoch": 2.5529861450195313e-06,
      "step": 2677,
      "training_step_time": 7.18264627456665
    },
    {
      "epoch": 2.5539398193359374e-06,
      "model_forward_time": 1.6902010440826416,
      "step": 2678
    },
    {
      "epoch": 2.5539398193359374e-06,
      "step": 2678,
      "training_step_time": 7.711159706115723
    },
    {
      "epoch": 2.5548934936523436e-06,
      "model_forward_time": 1.6882145404815674,
      "step": 2679
    },
    {
      "epoch": 2.5548934936523436e-06,
      "step": 2679,
      "training_step_time": 7.610551357269287
    },
    {
      "epoch": 2.55584716796875e-06,
      "grad_norm": 0.8066829442977905,
      "learning_rate": 5.358000000000001e-06,
      "loss": 0.0849,
      "step": 2680
    },
    {
      "epoch": 2.55584716796875e-06,
      "model_forward_time": 1.6874451637268066,
      "step": 2680
    },
    {
      "dynamics_loss_avg": 0.060015632398426534,
      "epoch": 2.55584716796875e-06,
      "step": 2680
    },
    {
      "action_loss_avg": 0.0099337887018919,
      "epoch": 2.55584716796875e-06,
      "step": 2680
    },
    {
      "epoch": 2.55584716796875e-06,
      "step": 2680,
      "training_step_time": 8.21215271949768
    },
    {
      "epoch": 2.5568008422851563e-06,
      "model_forward_time": 1.68572998046875,
      "step": 2681
    },
    {
      "epoch": 2.5568008422851563e-06,
      "step": 2681,
      "training_step_time": 8.446295976638794
    },
    {
      "epoch": 2.5577545166015624e-06,
      "model_forward_time": 1.6932175159454346,
      "step": 2682
    },
    {
      "epoch": 2.5577545166015624e-06,
      "step": 2682,
      "training_step_time": 8.894026041030884
    },
    {
      "epoch": 2.5587081909179686e-06,
      "model_forward_time": 1.6911406517028809,
      "step": 2683
    },
    {
      "epoch": 2.5587081909179686e-06,
      "step": 2683,
      "training_step_time": 8.875834465026855
    },
    {
      "epoch": 2.559661865234375e-06,
      "model_forward_time": 1.7364041805267334,
      "step": 2684
    },
    {
      "epoch": 2.559661865234375e-06,
      "step": 2684,
      "training_step_time": 9.112343549728394
    },
    {
      "epoch": 2.5606155395507813e-06,
      "model_forward_time": 1.693979024887085,
      "step": 2685
    },
    {
      "epoch": 2.5606155395507813e-06,
      "step": 2685,
      "training_step_time": 8.179602861404419
    },
    {
      "epoch": 2.5615692138671874e-06,
      "model_forward_time": 1.689497470855713,
      "step": 2686
    },
    {
      "epoch": 2.5615692138671874e-06,
      "step": 2686,
      "training_step_time": 8.192521095275879
    },
    {
      "epoch": 2.5625228881835936e-06,
      "model_forward_time": 1.6906781196594238,
      "step": 2687
    },
    {
      "epoch": 2.5625228881835936e-06,
      "step": 2687,
      "training_step_time": 7.646423816680908
    },
    {
      "epoch": 2.5634765625e-06,
      "model_forward_time": 1.691877841949463,
      "step": 2688
    },
    {
      "epoch": 2.5634765625e-06,
      "step": 2688,
      "training_step_time": 7.6150689125061035
    },
    {
      "epoch": 2.5644302368164063e-06,
      "model_forward_time": 1.6904733180999756,
      "step": 2689
    },
    {
      "epoch": 2.5644302368164063e-06,
      "step": 2689,
      "training_step_time": 7.565166711807251
    },
    {
      "epoch": 2.5653839111328124e-06,
      "grad_norm": 0.732093334197998,
      "learning_rate": 5.378e-06,
      "loss": 0.0999,
      "step": 2690
    },
    {
      "epoch": 2.5653839111328124e-06,
      "model_forward_time": 1.688035488128662,
      "step": 2690
    },
    {
      "dynamics_loss_avg": 0.07938721738755702,
      "epoch": 2.5653839111328124e-06,
      "step": 2690
    },
    {
      "action_loss_avg": 0.017509748204611243,
      "epoch": 2.5653839111328124e-06,
      "step": 2690
    },
    {
      "epoch": 2.5653839111328124e-06,
      "step": 2690,
      "training_step_time": 9.176563024520874
    },
    {
      "epoch": 2.5663375854492186e-06,
      "model_forward_time": 1.6816251277923584,
      "step": 2691
    },
    {
      "epoch": 2.5663375854492186e-06,
      "step": 2691,
      "training_step_time": 9.101060628890991
    },
    {
      "epoch": 2.567291259765625e-06,
      "model_forward_time": 1.702402114868164,
      "step": 2692
    },
    {
      "epoch": 2.567291259765625e-06,
      "step": 2692,
      "training_step_time": 7.794558763504028
    },
    {
      "epoch": 2.5682449340820313e-06,
      "model_forward_time": 1.3970115184783936,
      "step": 2693
    },
    {
      "epoch": 2.5682449340820313e-06,
      "step": 2693,
      "training_step_time": 9.616552352905273
    },
    {
      "epoch": 2.5691986083984375e-06,
      "model_forward_time": 2.2142064571380615,
      "step": 2694
    },
    {
      "epoch": 2.5691986083984375e-06,
      "step": 2694,
      "training_step_time": 7.624058723449707
    },
    {
      "epoch": 2.5701522827148436e-06,
      "model_forward_time": 1.698002576828003,
      "step": 2695
    },
    {
      "epoch": 2.5701522827148436e-06,
      "step": 2695,
      "training_step_time": 7.467102289199829
    },
    {
      "epoch": 2.57110595703125e-06,
      "model_forward_time": 1.6895253658294678,
      "step": 2696
    },
    {
      "epoch": 2.57110595703125e-06,
      "step": 2696,
      "training_step_time": 7.644928932189941
    },
    {
      "epoch": 2.5720596313476563e-06,
      "model_forward_time": 1.6854209899902344,
      "step": 2697
    },
    {
      "epoch": 2.5720596313476563e-06,
      "step": 2697,
      "training_step_time": 7.555819272994995
    },
    {
      "epoch": 2.5730133056640625e-06,
      "model_forward_time": 1.6919670104980469,
      "step": 2698
    },
    {
      "epoch": 2.5730133056640625e-06,
      "step": 2698,
      "training_step_time": 7.395795822143555
    },
    {
      "epoch": 2.5739669799804686e-06,
      "model_forward_time": 0.9819297790527344,
      "step": 2699
    },
    {
      "epoch": 2.5739669799804686e-06,
      "step": 2699,
      "training_step_time": 8.485690116882324
    },
    {
      "epoch": 2.574920654296875e-06,
      "grad_norm": 0.3908555805683136,
      "learning_rate": 5.398e-06,
      "loss": 0.0894,
      "step": 2700
    },
    {
      "epoch": 2.574920654296875e-06,
      "model_forward_time": 1.6976490020751953,
      "step": 2700
    },
    {
      "dynamics_loss_avg": 0.07430585250258445,
      "epoch": 2.574920654296875e-06,
      "step": 2700
    },
    {
      "action_loss_avg": 0.016011396329849957,
      "epoch": 2.574920654296875e-06,
      "step": 2700
    },
    {
      "epoch": 2.574920654296875e-06,
      "step": 2700,
      "training_step_time": 7.671178579330444
    },
    {
      "epoch": 2.5758743286132813e-06,
      "model_forward_time": 1.6858503818511963,
      "step": 2701
    },
    {
      "epoch": 2.5758743286132813e-06,
      "step": 2701,
      "training_step_time": 8.907032489776611
    },
    {
      "epoch": 2.5768280029296875e-06,
      "model_forward_time": 1.7501935958862305,
      "step": 2702
    },
    {
      "epoch": 2.5768280029296875e-06,
      "step": 2702,
      "training_step_time": 7.687009811401367
    },
    {
      "epoch": 2.5777816772460936e-06,
      "model_forward_time": 2.212390899658203,
      "step": 2703
    },
    {
      "epoch": 2.5777816772460936e-06,
      "step": 2703,
      "training_step_time": 7.730337858200073
    },
    {
      "epoch": 2.5787353515625e-06,
      "model_forward_time": 1.6867802143096924,
      "step": 2704
    },
    {
      "epoch": 2.5787353515625e-06,
      "step": 2704,
      "training_step_time": 7.484586477279663
    },
    {
      "epoch": 2.5796890258789063e-06,
      "model_forward_time": 1.691260814666748,
      "step": 2705
    },
    {
      "epoch": 2.5796890258789063e-06,
      "step": 2705,
      "training_step_time": 7.469064950942993
    },
    {
      "epoch": 2.5806427001953125e-06,
      "model_forward_time": 1.7032110691070557,
      "step": 2706
    },
    {
      "epoch": 2.5806427001953125e-06,
      "step": 2706,
      "training_step_time": 8.766443729400635
    },
    {
      "epoch": 2.5815963745117186e-06,
      "model_forward_time": 1.6922833919525146,
      "step": 2707
    },
    {
      "epoch": 2.5815963745117186e-06,
      "step": 2707,
      "training_step_time": 7.35604190826416
    },
    {
      "epoch": 2.582550048828125e-06,
      "model_forward_time": 1.688481092453003,
      "step": 2708
    },
    {
      "epoch": 2.582550048828125e-06,
      "step": 2708,
      "training_step_time": 7.667860984802246
    },
    {
      "epoch": 2.5835037231445313e-06,
      "model_forward_time": 1.6939260959625244,
      "step": 2709
    },
    {
      "epoch": 2.5835037231445313e-06,
      "step": 2709,
      "training_step_time": 8.517714738845825
    },
    {
      "epoch": 2.5844573974609375e-06,
      "grad_norm": 0.8186837434768677,
      "learning_rate": 5.418e-06,
      "loss": 0.0845,
      "step": 2710
    },
    {
      "epoch": 2.5844573974609375e-06,
      "model_forward_time": 1.6875450611114502,
      "step": 2710
    },
    {
      "dynamics_loss_avg": 0.08029174767434596,
      "epoch": 2.5844573974609375e-06,
      "step": 2710
    },
    {
      "action_loss_avg": 0.01836090609431267,
      "epoch": 2.5844573974609375e-06,
      "step": 2710
    },
    {
      "epoch": 2.5844573974609375e-06,
      "step": 2710,
      "training_step_time": 8.396810531616211
    },
    {
      "epoch": 2.5854110717773436e-06,
      "model_forward_time": 1.7408695220947266,
      "step": 2711
    },
    {
      "epoch": 2.5854110717773436e-06,
      "step": 2711,
      "training_step_time": 7.893707990646362
    },
    {
      "epoch": 2.58636474609375e-06,
      "model_forward_time": 1.6920835971832275,
      "step": 2712
    },
    {
      "epoch": 2.58636474609375e-06,
      "step": 2712,
      "training_step_time": 7.478705406188965
    },
    {
      "epoch": 2.5873184204101563e-06,
      "model_forward_time": 2.219205141067505,
      "step": 2713
    },
    {
      "epoch": 2.5873184204101563e-06,
      "step": 2713,
      "training_step_time": 8.379734992980957
    },
    {
      "epoch": 2.5882720947265625e-06,
      "model_forward_time": 1.6916821002960205,
      "step": 2714
    },
    {
      "epoch": 2.5882720947265625e-06,
      "step": 2714,
      "training_step_time": 7.370423793792725
    },
    {
      "epoch": 2.5892257690429686e-06,
      "model_forward_time": 1.6954054832458496,
      "step": 2715
    },
    {
      "epoch": 2.5892257690429686e-06,
      "step": 2715,
      "training_step_time": 8.618199586868286
    },
    {
      "epoch": 2.590179443359375e-06,
      "model_forward_time": 1.7065565586090088,
      "step": 2716
    },
    {
      "epoch": 2.590179443359375e-06,
      "step": 2716,
      "training_step_time": 8.0832998752594
    },
    {
      "epoch": 2.5911331176757813e-06,
      "model_forward_time": 1.7008395195007324,
      "step": 2717
    },
    {
      "epoch": 2.5911331176757813e-06,
      "step": 2717,
      "training_step_time": 7.522273778915405
    },
    {
      "epoch": 2.5920867919921875e-06,
      "model_forward_time": 1.7029399871826172,
      "step": 2718
    },
    {
      "epoch": 2.5920867919921875e-06,
      "step": 2718,
      "training_step_time": 8.593828678131104
    },
    {
      "epoch": 2.5930404663085936e-06,
      "model_forward_time": 1.711411476135254,
      "step": 2719
    },
    {
      "epoch": 2.5930404663085936e-06,
      "step": 2719,
      "training_step_time": 8.717852354049683
    },
    {
      "epoch": 2.593994140625e-06,
      "grad_norm": 0.45514923334121704,
      "learning_rate": 5.438e-06,
      "loss": 0.0928,
      "step": 2720
    },
    {
      "epoch": 2.593994140625e-06,
      "model_forward_time": 1.7535276412963867,
      "step": 2720
    },
    {
      "dynamics_loss_avg": 0.06354275345802307,
      "epoch": 2.593994140625e-06,
      "step": 2720
    },
    {
      "action_loss_avg": 0.012738521350547672,
      "epoch": 2.593994140625e-06,
      "step": 2720
    },
    {
      "epoch": 2.593994140625e-06,
      "step": 2720,
      "training_step_time": 7.598744630813599
    },
    {
      "epoch": 2.5949478149414063e-06,
      "model_forward_time": 1.7066092491149902,
      "step": 2721
    },
    {
      "epoch": 2.5949478149414063e-06,
      "step": 2721,
      "training_step_time": 8.546828746795654
    },
    {
      "epoch": 2.5959014892578125e-06,
      "model_forward_time": 2.2217488288879395,
      "step": 2722
    },
    {
      "epoch": 2.5959014892578125e-06,
      "step": 2722,
      "training_step_time": 8.114519119262695
    },
    {
      "epoch": 2.5968551635742186e-06,
      "model_forward_time": 1.7009170055389404,
      "step": 2723
    },
    {
      "epoch": 2.5968551635742186e-06,
      "step": 2723,
      "training_step_time": 7.326435327529907
    },
    {
      "epoch": 2.597808837890625e-06,
      "model_forward_time": 1.6998274326324463,
      "step": 2724
    },
    {
      "epoch": 2.597808837890625e-06,
      "step": 2724,
      "training_step_time": 7.649052619934082
    },
    {
      "epoch": 2.5987625122070313e-06,
      "model_forward_time": 1.703537940979004,
      "step": 2725
    },
    {
      "epoch": 2.5987625122070313e-06,
      "step": 2725,
      "training_step_time": 8.28355360031128
    },
    {
      "epoch": 2.5997161865234375e-06,
      "model_forward_time": 1.69746732711792,
      "step": 2726
    },
    {
      "epoch": 2.5997161865234375e-06,
      "step": 2726,
      "training_step_time": 8.327428102493286
    },
    {
      "epoch": 2.6006698608398436e-06,
      "model_forward_time": 1.696079969406128,
      "step": 2727
    },
    {
      "epoch": 2.6006698608398436e-06,
      "step": 2727,
      "training_step_time": 10.409088373184204
    },
    {
      "epoch": 2.60162353515625e-06,
      "model_forward_time": 1.7029695510864258,
      "step": 2728
    },
    {
      "epoch": 2.60162353515625e-06,
      "step": 2728,
      "training_step_time": 7.692970275878906
    },
    {
      "epoch": 2.6025772094726563e-06,
      "model_forward_time": 1.740429401397705,
      "step": 2729
    },
    {
      "epoch": 2.6025772094726563e-06,
      "step": 2729,
      "training_step_time": 8.645593404769897
    },
    {
      "epoch": 2.6035308837890625e-06,
      "grad_norm": 0.8694458603858948,
      "learning_rate": 5.458e-06,
      "loss": 0.0977,
      "step": 2730
    },
    {
      "epoch": 2.6035308837890625e-06,
      "model_forward_time": 1.6959424018859863,
      "step": 2730
    },
    {
      "dynamics_loss_avg": 0.07240321785211563,
      "epoch": 2.6035308837890625e-06,
      "step": 2730
    },
    {
      "action_loss_avg": 0.011937703518196941,
      "epoch": 2.6035308837890625e-06,
      "step": 2730
    },
    {
      "epoch": 2.6035308837890625e-06,
      "step": 2730,
      "training_step_time": 7.555109262466431
    },
    {
      "epoch": 2.6044845581054686e-06,
      "model_forward_time": 2.2401607036590576,
      "step": 2731
    },
    {
      "epoch": 2.6044845581054686e-06,
      "step": 2731,
      "training_step_time": 7.588192939758301
    },
    {
      "epoch": 2.605438232421875e-06,
      "model_forward_time": 1.7021205425262451,
      "step": 2732
    },
    {
      "epoch": 2.605438232421875e-06,
      "step": 2732,
      "training_step_time": 7.465249538421631
    },
    {
      "epoch": 2.6063919067382813e-06,
      "model_forward_time": 1.6993441581726074,
      "step": 2733
    },
    {
      "epoch": 2.6063919067382813e-06,
      "step": 2733,
      "training_step_time": 7.024631977081299
    },
    {
      "epoch": 2.6073455810546875e-06,
      "model_forward_time": 1.7033066749572754,
      "step": 2734
    },
    {
      "epoch": 2.6073455810546875e-06,
      "step": 2734,
      "training_step_time": 7.6913816928863525
    },
    {
      "epoch": 2.6082992553710936e-06,
      "model_forward_time": 1.690901517868042,
      "step": 2735
    },
    {
      "epoch": 2.6082992553710936e-06,
      "step": 2735,
      "training_step_time": 7.560538291931152
    },
    {
      "epoch": 2.6092529296875e-06,
      "model_forward_time": 1.706296682357788,
      "step": 2736
    },
    {
      "epoch": 2.6092529296875e-06,
      "step": 2736,
      "training_step_time": 8.328360795974731
    },
    {
      "epoch": 2.6102066040039063e-06,
      "model_forward_time": 1.701207160949707,
      "step": 2737
    },
    {
      "epoch": 2.6102066040039063e-06,
      "step": 2737,
      "training_step_time": 8.98133397102356
    },
    {
      "epoch": 2.6111602783203125e-06,
      "model_forward_time": 1.7198364734649658,
      "step": 2738
    },
    {
      "epoch": 2.6111602783203125e-06,
      "step": 2738,
      "training_step_time": 7.871206998825073
    },
    {
      "epoch": 2.6121139526367186e-06,
      "model_forward_time": 1.6993818283081055,
      "step": 2739
    },
    {
      "epoch": 2.6121139526367186e-06,
      "step": 2739,
      "training_step_time": 8.552951335906982
    },
    {
      "epoch": 2.613067626953125e-06,
      "grad_norm": 0.584118127822876,
      "learning_rate": 5.478e-06,
      "loss": 0.0991,
      "step": 2740
    },
    {
      "epoch": 2.613067626953125e-06,
      "model_forward_time": 2.2346324920654297,
      "step": 2740
    },
    {
      "dynamics_loss_avg": 0.07686330135911704,
      "epoch": 2.613067626953125e-06,
      "step": 2740
    },
    {
      "action_loss_avg": 0.023274481249973177,
      "epoch": 2.613067626953125e-06,
      "step": 2740
    },
    {
      "epoch": 2.613067626953125e-06,
      "step": 2740,
      "training_step_time": 8.414441585540771
    },
    {
      "epoch": 2.6140213012695313e-06,
      "model_forward_time": 1.701758861541748,
      "step": 2741
    },
    {
      "epoch": 2.6140213012695313e-06,
      "step": 2741,
      "training_step_time": 7.531540870666504
    },
    {
      "epoch": 2.6149749755859375e-06,
      "model_forward_time": 1.7099685668945312,
      "step": 2742
    },
    {
      "epoch": 2.6149749755859375e-06,
      "step": 2742,
      "training_step_time": 7.432114124298096
    },
    {
      "epoch": 2.6159286499023436e-06,
      "model_forward_time": 1.6967277526855469,
      "step": 2743
    },
    {
      "epoch": 2.6159286499023436e-06,
      "step": 2743,
      "training_step_time": 7.748766899108887
    },
    {
      "epoch": 2.61688232421875e-06,
      "model_forward_time": 1.699805736541748,
      "step": 2744
    },
    {
      "epoch": 2.61688232421875e-06,
      "step": 2744,
      "training_step_time": 7.563387632369995
    },
    {
      "epoch": 2.6178359985351564e-06,
      "model_forward_time": 1.7027878761291504,
      "step": 2745
    },
    {
      "epoch": 2.6178359985351564e-06,
      "step": 2745,
      "training_step_time": 10.399075031280518
    },
    {
      "epoch": 2.6187896728515625e-06,
      "model_forward_time": 1.689445972442627,
      "step": 2746
    },
    {
      "epoch": 2.6187896728515625e-06,
      "step": 2746,
      "training_step_time": 9.66025996208191
    },
    {
      "epoch": 2.6197433471679686e-06,
      "model_forward_time": 1.6945891380310059,
      "step": 2747
    },
    {
      "epoch": 2.6197433471679686e-06,
      "step": 2747,
      "training_step_time": 8.458614349365234
    },
    {
      "epoch": 2.620697021484375e-06,
      "model_forward_time": 1.6954829692840576,
      "step": 2748
    },
    {
      "epoch": 2.620697021484375e-06,
      "step": 2748,
      "training_step_time": 7.918112277984619
    },
    {
      "epoch": 2.6216506958007814e-06,
      "model_forward_time": 2.2327144145965576,
      "step": 2749
    },
    {
      "epoch": 2.6216506958007814e-06,
      "step": 2749,
      "training_step_time": 7.742948293685913
    },
    {
      "epoch": 2.6226043701171875e-06,
      "grad_norm": 0.5321668386459351,
      "learning_rate": 5.498e-06,
      "loss": 0.103,
      "step": 2750
    },
    {
      "epoch": 2.6226043701171875e-06,
      "model_forward_time": 1.6904029846191406,
      "step": 2750
    },
    {
      "dynamics_loss_avg": 0.09019850250333547,
      "epoch": 2.6226043701171875e-06,
      "step": 2750
    },
    {
      "action_loss_avg": 0.018040104815736414,
      "epoch": 2.6226043701171875e-06,
      "step": 2750
    },
    {
      "epoch": 2.6226043701171875e-06,
      "step": 2750,
      "training_step_time": 7.753096342086792
    },
    {
      "epoch": 2.6235580444335936e-06,
      "model_forward_time": 1.697178840637207,
      "step": 2751
    },
    {
      "epoch": 2.6235580444335936e-06,
      "step": 2751,
      "training_step_time": 7.605758428573608
    },
    {
      "epoch": 2.62451171875e-06,
      "model_forward_time": 1.6929688453674316,
      "step": 2752
    },
    {
      "epoch": 2.62451171875e-06,
      "step": 2752,
      "training_step_time": 8.408223867416382
    },
    {
      "epoch": 2.6254653930664064e-06,
      "model_forward_time": 1.696730136871338,
      "step": 2753
    },
    {
      "epoch": 2.6254653930664064e-06,
      "step": 2753,
      "training_step_time": 7.781646251678467
    },
    {
      "epoch": 2.6264190673828125e-06,
      "model_forward_time": 1.7044010162353516,
      "step": 2754
    },
    {
      "epoch": 2.6264190673828125e-06,
      "step": 2754,
      "training_step_time": 8.29163384437561
    },
    {
      "epoch": 2.6273727416992186e-06,
      "model_forward_time": 1.699068546295166,
      "step": 2755
    },
    {
      "epoch": 2.6273727416992186e-06,
      "step": 2755,
      "training_step_time": 7.88713264465332
    },
    {
      "epoch": 2.628326416015625e-06,
      "model_forward_time": 1.6826190948486328,
      "step": 2756
    },
    {
      "epoch": 2.628326416015625e-06,
      "step": 2756,
      "training_step_time": 9.321751594543457
    },
    {
      "epoch": 2.6292800903320314e-06,
      "model_forward_time": 1.7429113388061523,
      "step": 2757
    },
    {
      "epoch": 2.6292800903320314e-06,
      "step": 2757,
      "training_step_time": 8.39179539680481
    },
    {
      "epoch": 2.6302337646484375e-06,
      "model_forward_time": 1.746875286102295,
      "step": 2758
    },
    {
      "epoch": 2.6302337646484375e-06,
      "step": 2758,
      "training_step_time": 8.291247844696045
    },
    {
      "epoch": 2.6311874389648437e-06,
      "model_forward_time": 1.7470746040344238,
      "step": 2759
    },
    {
      "epoch": 2.6311874389648437e-06,
      "step": 2759,
      "training_step_time": 7.836441278457642
    },
    {
      "epoch": 2.63214111328125e-06,
      "grad_norm": 1.059245228767395,
      "learning_rate": 5.518e-06,
      "loss": 0.1015,
      "step": 2760
    },
    {
      "epoch": 2.63214111328125e-06,
      "model_forward_time": 1.768352746963501,
      "step": 2760
    },
    {
      "dynamics_loss_avg": 0.0899201650172472,
      "epoch": 2.63214111328125e-06,
      "step": 2760
    },
    {
      "action_loss_avg": 0.01506914203055203,
      "epoch": 2.63214111328125e-06,
      "step": 2760
    },
    {
      "epoch": 2.63214111328125e-06,
      "step": 2760,
      "training_step_time": 7.569856643676758
    },
    {
      "epoch": 2.6330947875976564e-06,
      "model_forward_time": 1.7460317611694336,
      "step": 2761
    },
    {
      "epoch": 2.6330947875976564e-06,
      "step": 2761,
      "training_step_time": 7.754801511764526
    },
    {
      "epoch": 2.6340484619140625e-06,
      "model_forward_time": 1.727337121963501,
      "step": 2762
    },
    {
      "epoch": 2.6340484619140625e-06,
      "step": 2762,
      "training_step_time": 7.7701942920684814
    },
    {
      "epoch": 2.6350021362304687e-06,
      "model_forward_time": 1.7588019371032715,
      "step": 2763
    },
    {
      "epoch": 2.6350021362304687e-06,
      "step": 2763,
      "training_step_time": 7.728443384170532
    },
    {
      "epoch": 2.635955810546875e-06,
      "model_forward_time": 1.7446491718292236,
      "step": 2764
    },
    {
      "epoch": 2.635955810546875e-06,
      "step": 2764,
      "training_step_time": 7.812859535217285
    },
    {
      "epoch": 2.6369094848632814e-06,
      "model_forward_time": 1.7490386962890625,
      "step": 2765
    },
    {
      "epoch": 2.6369094848632814e-06,
      "step": 2765,
      "training_step_time": 9.33054780960083
    },
    {
      "epoch": 2.6378631591796875e-06,
      "model_forward_time": 1.799407720565796,
      "step": 2766
    },
    {
      "epoch": 2.6378631591796875e-06,
      "step": 2766,
      "training_step_time": 8.623242855072021
    },
    {
      "epoch": 2.6388168334960937e-06,
      "model_forward_time": 1.7448525428771973,
      "step": 2767
    },
    {
      "epoch": 2.6388168334960937e-06,
      "step": 2767,
      "training_step_time": 7.7276201248168945
    },
    {
      "epoch": 2.6397705078125e-06,
      "model_forward_time": 2.305952310562134,
      "step": 2768
    },
    {
      "epoch": 2.6397705078125e-06,
      "step": 2768,
      "training_step_time": 7.638382434844971
    },
    {
      "epoch": 2.6407241821289064e-06,
      "model_forward_time": 1.7513680458068848,
      "step": 2769
    },
    {
      "epoch": 2.6407241821289064e-06,
      "step": 2769,
      "training_step_time": 7.310598134994507
    },
    {
      "epoch": 2.6416778564453125e-06,
      "grad_norm": 0.6342085003852844,
      "learning_rate": 5.538e-06,
      "loss": 0.0997,
      "step": 2770
    },
    {
      "epoch": 2.6416778564453125e-06,
      "model_forward_time": 1.7546429634094238,
      "step": 2770
    },
    {
      "dynamics_loss_avg": 0.09235086739063263,
      "epoch": 2.6416778564453125e-06,
      "step": 2770
    },
    {
      "action_loss_avg": 0.021648307982832193,
      "epoch": 2.6416778564453125e-06,
      "step": 2770
    },
    {
      "epoch": 2.6416778564453125e-06,
      "step": 2770,
      "training_step_time": 7.691647529602051
    },
    {
      "epoch": 2.6426315307617187e-06,
      "model_forward_time": 1.741577386856079,
      "step": 2771
    },
    {
      "epoch": 2.6426315307617187e-06,
      "step": 2771,
      "training_step_time": 8.232075691223145
    },
    {
      "epoch": 2.643585205078125e-06,
      "model_forward_time": 1.7452998161315918,
      "step": 2772
    },
    {
      "epoch": 2.643585205078125e-06,
      "step": 2772,
      "training_step_time": 9.25190281867981
    },
    {
      "epoch": 2.6445388793945314e-06,
      "model_forward_time": 1.7510883808135986,
      "step": 2773
    },
    {
      "epoch": 2.6445388793945314e-06,
      "step": 2773,
      "training_step_time": 8.22834062576294
    },
    {
      "epoch": 2.6454925537109375e-06,
      "model_forward_time": 1.7598721981048584,
      "step": 2774
    },
    {
      "epoch": 2.6454925537109375e-06,
      "step": 2774,
      "training_step_time": 7.842958211898804
    },
    {
      "epoch": 2.6464462280273437e-06,
      "model_forward_time": 1.775937795639038,
      "step": 2775
    },
    {
      "epoch": 2.6464462280273437e-06,
      "step": 2775,
      "training_step_time": 8.862954139709473
    },
    {
      "epoch": 2.64739990234375e-06,
      "model_forward_time": 1.7412359714508057,
      "step": 2776
    },
    {
      "epoch": 2.64739990234375e-06,
      "step": 2776,
      "training_step_time": 7.687204122543335
    },
    {
      "epoch": 2.6483535766601564e-06,
      "model_forward_time": 2.3109307289123535,
      "step": 2777
    },
    {
      "epoch": 2.6483535766601564e-06,
      "step": 2777,
      "training_step_time": 7.728714466094971
    },
    {
      "epoch": 2.6493072509765625e-06,
      "model_forward_time": 1.7578904628753662,
      "step": 2778
    },
    {
      "epoch": 2.6493072509765625e-06,
      "step": 2778,
      "training_step_time": 7.6753644943237305
    },
    {
      "epoch": 2.6502609252929687e-06,
      "model_forward_time": 1.7541999816894531,
      "step": 2779
    },
    {
      "epoch": 2.6502609252929687e-06,
      "step": 2779,
      "training_step_time": 8.294359683990479
    },
    {
      "epoch": 2.651214599609375e-06,
      "grad_norm": 0.842586874961853,
      "learning_rate": 5.558e-06,
      "loss": 0.0942,
      "step": 2780
    },
    {
      "epoch": 2.651214599609375e-06,
      "model_forward_time": 1.7543210983276367,
      "step": 2780
    },
    {
      "dynamics_loss_avg": 0.08498630747199058,
      "epoch": 2.651214599609375e-06,
      "step": 2780
    },
    {
      "action_loss_avg": 0.0214082517195493,
      "epoch": 2.651214599609375e-06,
      "step": 2780
    },
    {
      "epoch": 2.651214599609375e-06,
      "step": 2780,
      "training_step_time": 7.747192621231079
    },
    {
      "epoch": 2.6521682739257814e-06,
      "model_forward_time": 1.7504470348358154,
      "step": 2781
    },
    {
      "epoch": 2.6521682739257814e-06,
      "step": 2781,
      "training_step_time": 8.48751187324524
    },
    {
      "epoch": 2.6531219482421875e-06,
      "model_forward_time": 1.7461440563201904,
      "step": 2782
    },
    {
      "epoch": 2.6531219482421875e-06,
      "step": 2782,
      "training_step_time": 8.40138840675354
    },
    {
      "epoch": 2.6540756225585937e-06,
      "model_forward_time": 1.754333257675171,
      "step": 2783
    },
    {
      "epoch": 2.6540756225585937e-06,
      "step": 2783,
      "training_step_time": 9.343740224838257
    },
    {
      "epoch": 2.655029296875e-06,
      "model_forward_time": 1.7574689388275146,
      "step": 2784
    },
    {
      "epoch": 2.655029296875e-06,
      "step": 2784,
      "training_step_time": 7.875373840332031
    },
    {
      "epoch": 2.6559829711914064e-06,
      "model_forward_time": 1.7522833347320557,
      "step": 2785
    },
    {
      "epoch": 2.6559829711914064e-06,
      "step": 2785,
      "training_step_time": 8.795226097106934
    },
    {
      "epoch": 2.6569366455078125e-06,
      "model_forward_time": 2.3170828819274902,
      "step": 2786
    },
    {
      "epoch": 2.6569366455078125e-06,
      "step": 2786,
      "training_step_time": 7.690286159515381
    },
    {
      "epoch": 2.6578903198242187e-06,
      "model_forward_time": 1.755425214767456,
      "step": 2787
    },
    {
      "epoch": 2.6578903198242187e-06,
      "step": 2787,
      "training_step_time": 8.29064130783081
    },
    {
      "epoch": 2.658843994140625e-06,
      "model_forward_time": 1.7532105445861816,
      "step": 2788
    },
    {
      "epoch": 2.658843994140625e-06,
      "step": 2788,
      "training_step_time": 8.2352774143219
    },
    {
      "epoch": 2.6597976684570314e-06,
      "model_forward_time": 1.741328477859497,
      "step": 2789
    },
    {
      "epoch": 2.6597976684570314e-06,
      "step": 2789,
      "training_step_time": 7.600100755691528
    },
    {
      "epoch": 2.6607513427734375e-06,
      "grad_norm": 1.200789213180542,
      "learning_rate": 5.578e-06,
      "loss": 0.0939,
      "step": 2790
    },
    {
      "epoch": 2.6607513427734375e-06,
      "model_forward_time": 1.7518465518951416,
      "step": 2790
    },
    {
      "dynamics_loss_avg": 0.06892589014023542,
      "epoch": 2.6607513427734375e-06,
      "step": 2790
    },
    {
      "action_loss_avg": 0.023082710197195412,
      "epoch": 2.6607513427734375e-06,
      "step": 2790
    },
    {
      "epoch": 2.6607513427734375e-06,
      "step": 2790,
      "training_step_time": 7.543862342834473
    },
    {
      "epoch": 2.6617050170898437e-06,
      "model_forward_time": 1.7477209568023682,
      "step": 2791
    },
    {
      "epoch": 2.6617050170898437e-06,
      "step": 2791,
      "training_step_time": 8.489808320999146
    },
    {
      "epoch": 2.66265869140625e-06,
      "model_forward_time": 1.7528550624847412,
      "step": 2792
    },
    {
      "epoch": 2.66265869140625e-06,
      "step": 2792,
      "training_step_time": 8.781983375549316
    },
    {
      "epoch": 2.6636123657226564e-06,
      "model_forward_time": 1.7484066486358643,
      "step": 2793
    },
    {
      "epoch": 2.6636123657226564e-06,
      "step": 2793,
      "training_step_time": 8.631153583526611
    },
    {
      "epoch": 2.6645660400390625e-06,
      "model_forward_time": 1.7953228950500488,
      "step": 2794
    },
    {
      "epoch": 2.6645660400390625e-06,
      "step": 2794,
      "training_step_time": 7.7305588722229
    },
    {
      "epoch": 2.6655197143554687e-06,
      "model_forward_time": 2.3233585357666016,
      "step": 2795
    },
    {
      "epoch": 2.6655197143554687e-06,
      "step": 2795,
      "training_step_time": 8.396608829498291
    },
    {
      "epoch": 2.666473388671875e-06,
      "model_forward_time": 1.3759946823120117,
      "step": 2796
    },
    {
      "epoch": 2.666473388671875e-06,
      "step": 2796,
      "training_step_time": 7.092221975326538
    },
    {
      "epoch": 2.6674270629882814e-06,
      "model_forward_time": 1.7358958721160889,
      "step": 2797
    },
    {
      "epoch": 2.6674270629882814e-06,
      "step": 2797,
      "training_step_time": 9.427963256835938
    },
    {
      "epoch": 2.6683807373046875e-06,
      "model_forward_time": 1.7507946491241455,
      "step": 2798
    },
    {
      "epoch": 2.6683807373046875e-06,
      "step": 2798,
      "training_step_time": 7.659005165100098
    },
    {
      "epoch": 2.6693344116210937e-06,
      "model_forward_time": 1.6925935745239258,
      "step": 2799
    },
    {
      "epoch": 2.6693344116210937e-06,
      "step": 2799,
      "training_step_time": 8.1459321975708
    },
    {
      "epoch": 2.6702880859375e-06,
      "grad_norm": 0.9714585542678833,
      "learning_rate": 5.5980000000000004e-06,
      "loss": 0.0925,
      "step": 2800
    },
    {
      "epoch": 2.6702880859375e-06,
      "model_forward_time": 1.6943533420562744,
      "step": 2800
    },
    {
      "dynamics_loss_avg": 0.06180726382881403,
      "epoch": 2.6702880859375e-06,
      "step": 2800
    },
    {
      "action_loss_avg": 0.01004903125576675,
      "epoch": 2.6702880859375e-06,
      "step": 2800
    },
    {
      "epoch": 2.6702880859375e-06,
      "step": 2800,
      "training_step_time": 7.6215293407440186
    },
    {
      "epoch": 2.6712417602539064e-06,
      "model_forward_time": 1.688610315322876,
      "step": 2801
    },
    {
      "epoch": 2.6712417602539064e-06,
      "step": 2801,
      "training_step_time": 7.642322778701782
    },
    {
      "epoch": 2.6721954345703125e-06,
      "model_forward_time": 1.3318426609039307,
      "step": 2802
    },
    {
      "epoch": 2.6721954345703125e-06,
      "step": 2802,
      "training_step_time": 8.704677104949951
    },
    {
      "epoch": 2.6731491088867187e-06,
      "model_forward_time": 1.0296318531036377,
      "step": 2803
    },
    {
      "epoch": 2.6731491088867187e-06,
      "step": 2803,
      "training_step_time": 8.492090463638306
    },
    {
      "epoch": 2.674102783203125e-06,
      "model_forward_time": 1.6957612037658691,
      "step": 2804
    },
    {
      "epoch": 2.674102783203125e-06,
      "step": 2804,
      "training_step_time": 7.7424023151397705
    },
    {
      "epoch": 2.6750564575195314e-06,
      "model_forward_time": 2.2510592937469482,
      "step": 2805
    },
    {
      "epoch": 2.6750564575195314e-06,
      "step": 2805,
      "training_step_time": 7.440141916275024
    },
    {
      "epoch": 2.6760101318359375e-06,
      "model_forward_time": 1.326097011566162,
      "step": 2806
    },
    {
      "epoch": 2.6760101318359375e-06,
      "step": 2806,
      "training_step_time": 8.057589530944824
    },
    {
      "epoch": 2.6769638061523437e-06,
      "model_forward_time": 1.7043592929840088,
      "step": 2807
    },
    {
      "epoch": 2.6769638061523437e-06,
      "step": 2807,
      "training_step_time": 7.749769687652588
    },
    {
      "epoch": 2.67791748046875e-06,
      "model_forward_time": 1.6987719535827637,
      "step": 2808
    },
    {
      "epoch": 2.67791748046875e-06,
      "step": 2808,
      "training_step_time": 7.589893341064453
    },
    {
      "epoch": 2.6788711547851564e-06,
      "model_forward_time": 1.7078783512115479,
      "step": 2809
    },
    {
      "epoch": 2.6788711547851564e-06,
      "step": 2809,
      "training_step_time": 7.599881410598755
    },
    {
      "epoch": 2.6798248291015626e-06,
      "grad_norm": 0.7344852089881897,
      "learning_rate": 5.618e-06,
      "loss": 0.0868,
      "step": 2810
    },
    {
      "epoch": 2.6798248291015626e-06,
      "model_forward_time": 1.7049477100372314,
      "step": 2810
    },
    {
      "dynamics_loss_avg": 0.06619388721883297,
      "epoch": 2.6798248291015626e-06,
      "step": 2810
    },
    {
      "action_loss_avg": 0.017544955434277653,
      "epoch": 2.6798248291015626e-06,
      "step": 2810
    },
    {
      "epoch": 2.6798248291015626e-06,
      "step": 2810,
      "training_step_time": 8.26762056350708
    },
    {
      "epoch": 2.6807785034179687e-06,
      "model_forward_time": 1.697859287261963,
      "step": 2811
    },
    {
      "epoch": 2.6807785034179687e-06,
      "step": 2811,
      "training_step_time": 7.774896144866943
    },
    {
      "epoch": 2.681732177734375e-06,
      "model_forward_time": 1.7551684379577637,
      "step": 2812
    },
    {
      "epoch": 2.681732177734375e-06,
      "step": 2812,
      "training_step_time": 9.837124586105347
    },
    {
      "epoch": 2.6826858520507814e-06,
      "model_forward_time": 1.7031593322753906,
      "step": 2813
    },
    {
      "epoch": 2.6826858520507814e-06,
      "step": 2813,
      "training_step_time": 9.025816202163696
    },
    {
      "epoch": 2.6836395263671876e-06,
      "model_forward_time": 1.7084681987762451,
      "step": 2814
    },
    {
      "epoch": 2.6836395263671876e-06,
      "step": 2814,
      "training_step_time": 7.9795005321502686
    },
    {
      "epoch": 2.6845932006835937e-06,
      "model_forward_time": 1.6965742111206055,
      "step": 2815
    },
    {
      "epoch": 2.6845932006835937e-06,
      "step": 2815,
      "training_step_time": 7.5931947231292725
    },
    {
      "epoch": 2.685546875e-06,
      "model_forward_time": 1.7043709754943848,
      "step": 2816
    },
    {
      "epoch": 2.685546875e-06,
      "step": 2816,
      "training_step_time": 7.650606870651245
    },
    {
      "epoch": 2.6865005493164064e-06,
      "model_forward_time": 1.705808162689209,
      "step": 2817
    },
    {
      "epoch": 2.6865005493164064e-06,
      "step": 2817,
      "training_step_time": 7.640721321105957
    },
    {
      "epoch": 2.6874542236328126e-06,
      "model_forward_time": 1.698749303817749,
      "step": 2818
    },
    {
      "epoch": 2.6874542236328126e-06,
      "step": 2818,
      "training_step_time": 7.613001823425293
    },
    {
      "epoch": 2.6884078979492187e-06,
      "model_forward_time": 1.7047719955444336,
      "step": 2819
    },
    {
      "epoch": 2.6884078979492187e-06,
      "step": 2819,
      "training_step_time": 7.73119044303894
    },
    {
      "epoch": 2.689361572265625e-06,
      "grad_norm": 0.6780060529708862,
      "learning_rate": 5.638e-06,
      "loss": 0.0869,
      "step": 2820
    },
    {
      "epoch": 2.689361572265625e-06,
      "model_forward_time": 1.723097801208496,
      "step": 2820
    },
    {
      "dynamics_loss_avg": 0.0743276134133339,
      "epoch": 2.689361572265625e-06,
      "step": 2820
    },
    {
      "action_loss_avg": 0.011276244977489113,
      "epoch": 2.689361572265625e-06,
      "step": 2820
    },
    {
      "epoch": 2.689361572265625e-06,
      "step": 2820,
      "training_step_time": 8.538052558898926
    },
    {
      "epoch": 2.6903152465820314e-06,
      "model_forward_time": 1.7710683345794678,
      "step": 2821
    },
    {
      "epoch": 2.6903152465820314e-06,
      "step": 2821,
      "training_step_time": 8.551121473312378
    },
    {
      "epoch": 2.6912689208984376e-06,
      "model_forward_time": 1.7222986221313477,
      "step": 2822
    },
    {
      "epoch": 2.6912689208984376e-06,
      "step": 2822,
      "training_step_time": 7.8506975173950195
    },
    {
      "epoch": 2.6922225952148437e-06,
      "model_forward_time": 1.7152385711669922,
      "step": 2823
    },
    {
      "epoch": 2.6922225952148437e-06,
      "step": 2823,
      "training_step_time": 7.584460496902466
    },
    {
      "epoch": 2.69317626953125e-06,
      "model_forward_time": 2.2719504833221436,
      "step": 2824
    },
    {
      "epoch": 2.69317626953125e-06,
      "step": 2824,
      "training_step_time": 7.338914632797241
    },
    {
      "epoch": 2.6941299438476564e-06,
      "model_forward_time": 1.7173340320587158,
      "step": 2825
    },
    {
      "epoch": 2.6941299438476564e-06,
      "step": 2825,
      "training_step_time": 7.600128650665283
    },
    {
      "epoch": 2.6950836181640626e-06,
      "model_forward_time": 1.7199254035949707,
      "step": 2826
    },
    {
      "epoch": 2.6950836181640626e-06,
      "step": 2826,
      "training_step_time": 7.730874538421631
    },
    {
      "epoch": 2.6960372924804687e-06,
      "model_forward_time": 1.7241096496582031,
      "step": 2827
    },
    {
      "epoch": 2.6960372924804687e-06,
      "step": 2827,
      "training_step_time": 8.403621673583984
    },
    {
      "epoch": 2.696990966796875e-06,
      "model_forward_time": 1.7235143184661865,
      "step": 2828
    },
    {
      "epoch": 2.696990966796875e-06,
      "step": 2828,
      "training_step_time": 8.022340059280396
    },
    {
      "epoch": 2.6979446411132814e-06,
      "model_forward_time": 1.711122751235962,
      "step": 2829
    },
    {
      "epoch": 2.6979446411132814e-06,
      "step": 2829,
      "training_step_time": 8.549588680267334
    },
    {
      "epoch": 2.6988983154296876e-06,
      "grad_norm": 0.8469730019569397,
      "learning_rate": 5.658e-06,
      "loss": 0.0904,
      "step": 2830
    },
    {
      "epoch": 2.6988983154296876e-06,
      "model_forward_time": 1.7655932903289795,
      "step": 2830
    },
    {
      "dynamics_loss_avg": 0.06123376786708832,
      "epoch": 2.6988983154296876e-06,
      "step": 2830
    },
    {
      "action_loss_avg": 0.009718886145856231,
      "epoch": 2.6988983154296876e-06,
      "step": 2830
    },
    {
      "epoch": 2.6988983154296876e-06,
      "step": 2830,
      "training_step_time": 8.030166149139404
    },
    {
      "epoch": 2.6998519897460937e-06,
      "model_forward_time": 1.7160065174102783,
      "step": 2831
    },
    {
      "epoch": 2.6998519897460937e-06,
      "step": 2831,
      "training_step_time": 7.860347032546997
    },
    {
      "epoch": 2.7008056640625e-06,
      "model_forward_time": 1.7186577320098877,
      "step": 2832
    },
    {
      "epoch": 2.7008056640625e-06,
      "step": 2832,
      "training_step_time": 7.800147294998169
    },
    {
      "epoch": 2.7017593383789064e-06,
      "model_forward_time": 2.255763530731201,
      "step": 2833
    },
    {
      "epoch": 2.7017593383789064e-06,
      "step": 2833,
      "training_step_time": 7.773798227310181
    },
    {
      "epoch": 2.7027130126953126e-06,
      "model_forward_time": 1.711291790008545,
      "step": 2834
    },
    {
      "epoch": 2.7027130126953126e-06,
      "step": 2834,
      "training_step_time": 8.634584903717041
    },
    {
      "epoch": 2.7036666870117187e-06,
      "model_forward_time": 1.7205462455749512,
      "step": 2835
    },
    {
      "epoch": 2.7036666870117187e-06,
      "step": 2835,
      "training_step_time": 7.8021814823150635
    },
    {
      "epoch": 2.704620361328125e-06,
      "model_forward_time": 1.737623929977417,
      "step": 2836
    },
    {
      "epoch": 2.704620361328125e-06,
      "step": 2836,
      "training_step_time": 7.692122459411621
    },
    {
      "epoch": 2.7055740356445314e-06,
      "model_forward_time": 1.7231690883636475,
      "step": 2837
    },
    {
      "epoch": 2.7055740356445314e-06,
      "step": 2837,
      "training_step_time": 8.939599990844727
    },
    {
      "epoch": 2.7065277099609376e-06,
      "model_forward_time": 1.7136523723602295,
      "step": 2838
    },
    {
      "epoch": 2.7065277099609376e-06,
      "step": 2838,
      "training_step_time": 7.8865721225738525
    },
    {
      "epoch": 2.7074813842773437e-06,
      "model_forward_time": 1.771629810333252,
      "step": 2839
    },
    {
      "epoch": 2.7074813842773437e-06,
      "step": 2839,
      "training_step_time": 9.063680410385132
    },
    {
      "epoch": 2.70843505859375e-06,
      "grad_norm": 0.5476734042167664,
      "learning_rate": 5.6780000000000005e-06,
      "loss": 0.0852,
      "step": 2840
    },
    {
      "epoch": 2.70843505859375e-06,
      "model_forward_time": 1.7139294147491455,
      "step": 2840
    },
    {
      "dynamics_loss_avg": 0.08862900547683239,
      "epoch": 2.70843505859375e-06,
      "step": 2840
    },
    {
      "action_loss_avg": 0.01652568285353482,
      "epoch": 2.70843505859375e-06,
      "step": 2840
    },
    {
      "epoch": 2.70843505859375e-06,
      "step": 2840,
      "training_step_time": 8.85326099395752
    },
    {
      "epoch": 2.7093887329101564e-06,
      "model_forward_time": 1.7182374000549316,
      "step": 2841
    },
    {
      "epoch": 2.7093887329101564e-06,
      "step": 2841,
      "training_step_time": 7.7952234745025635
    },
    {
      "epoch": 2.7103424072265626e-06,
      "model_forward_time": 2.262101888656616,
      "step": 2842
    },
    {
      "epoch": 2.7103424072265626e-06,
      "step": 2842,
      "training_step_time": 7.715730428695679
    },
    {
      "epoch": 2.7112960815429687e-06,
      "model_forward_time": 1.7207913398742676,
      "step": 2843
    },
    {
      "epoch": 2.7112960815429687e-06,
      "step": 2843,
      "training_step_time": 7.608044147491455
    },
    {
      "epoch": 2.712249755859375e-06,
      "model_forward_time": 1.7174937725067139,
      "step": 2844
    },
    {
      "epoch": 2.712249755859375e-06,
      "step": 2844,
      "training_step_time": 7.789675951004028
    },
    {
      "epoch": 2.7132034301757814e-06,
      "model_forward_time": 1.726775884628296,
      "step": 2845
    },
    {
      "epoch": 2.7132034301757814e-06,
      "step": 2845,
      "training_step_time": 9.235397100448608
    },
    {
      "epoch": 2.7141571044921876e-06,
      "model_forward_time": 1.6986029148101807,
      "step": 2846
    },
    {
      "epoch": 2.7141571044921876e-06,
      "step": 2846,
      "training_step_time": 7.888747692108154
    },
    {
      "epoch": 2.7151107788085937e-06,
      "model_forward_time": 1.712355136871338,
      "step": 2847
    },
    {
      "epoch": 2.7151107788085937e-06,
      "step": 2847,
      "training_step_time": 9.98577618598938
    },
    {
      "epoch": 2.716064453125e-06,
      "model_forward_time": 1.7349367141723633,
      "step": 2848
    },
    {
      "epoch": 2.716064453125e-06,
      "step": 2848,
      "training_step_time": 7.924743413925171
    },
    {
      "epoch": 2.7170181274414064e-06,
      "model_forward_time": 1.6972401142120361,
      "step": 2849
    },
    {
      "epoch": 2.7170181274414064e-06,
      "step": 2849,
      "training_step_time": 9.19619107246399
    },
    {
      "epoch": 2.7179718017578126e-06,
      "grad_norm": 0.608555257320404,
      "learning_rate": 5.698e-06,
      "loss": 0.0827,
      "step": 2850
    },
    {
      "epoch": 2.7179718017578126e-06,
      "model_forward_time": 1.6918563842773438,
      "step": 2850
    },
    {
      "dynamics_loss_avg": 0.06850316897034645,
      "epoch": 2.7179718017578126e-06,
      "step": 2850
    },
    {
      "action_loss_avg": 0.010556384548544883,
      "epoch": 2.7179718017578126e-06,
      "step": 2850
    },
    {
      "epoch": 2.7179718017578126e-06,
      "step": 2850,
      "training_step_time": 8.37583613395691
    },
    {
      "epoch": 2.7189254760742187e-06,
      "model_forward_time": 2.2611453533172607,
      "step": 2851
    },
    {
      "epoch": 2.7189254760742187e-06,
      "step": 2851,
      "training_step_time": 7.72033953666687
    },
    {
      "epoch": 2.719879150390625e-06,
      "model_forward_time": 1.707376480102539,
      "step": 2852
    },
    {
      "epoch": 2.719879150390625e-06,
      "step": 2852,
      "training_step_time": 8.419633150100708
    },
    {
      "epoch": 2.7208328247070314e-06,
      "model_forward_time": 1.7042474746704102,
      "step": 2853
    },
    {
      "epoch": 2.7208328247070314e-06,
      "step": 2853,
      "training_step_time": 7.792055368423462
    },
    {
      "epoch": 2.7217864990234376e-06,
      "model_forward_time": 1.706726312637329,
      "step": 2854
    },
    {
      "epoch": 2.7217864990234376e-06,
      "step": 2854,
      "training_step_time": 7.759710788726807
    },
    {
      "epoch": 2.7227401733398437e-06,
      "model_forward_time": 1.7033746242523193,
      "step": 2855
    },
    {
      "epoch": 2.7227401733398437e-06,
      "step": 2855,
      "training_step_time": 7.96776008605957
    },
    {
      "epoch": 2.72369384765625e-06,
      "model_forward_time": 1.7014775276184082,
      "step": 2856
    },
    {
      "epoch": 2.72369384765625e-06,
      "step": 2856,
      "training_step_time": 9.524832963943481
    },
    {
      "epoch": 2.7246475219726564e-06,
      "model_forward_time": 1.7261700630187988,
      "step": 2857
    },
    {
      "epoch": 2.7246475219726564e-06,
      "step": 2857,
      "training_step_time": 8.780658483505249
    },
    {
      "epoch": 2.7256011962890626e-06,
      "model_forward_time": 1.7034943103790283,
      "step": 2858
    },
    {
      "epoch": 2.7256011962890626e-06,
      "step": 2858,
      "training_step_time": 7.918789625167847
    },
    {
      "epoch": 2.7265548706054687e-06,
      "model_forward_time": 1.7055137157440186,
      "step": 2859
    },
    {
      "epoch": 2.7265548706054687e-06,
      "step": 2859,
      "training_step_time": 9.354858875274658
    },
    {
      "epoch": 2.727508544921875e-06,
      "grad_norm": 0.538773238658905,
      "learning_rate": 5.718e-06,
      "loss": 0.0795,
      "step": 2860
    },
    {
      "epoch": 2.727508544921875e-06,
      "model_forward_time": 2.242027997970581,
      "step": 2860
    },
    {
      "dynamics_loss_avg": 0.06606311369687319,
      "epoch": 2.727508544921875e-06,
      "step": 2860
    },
    {
      "action_loss_avg": 0.011928336042910815,
      "epoch": 2.727508544921875e-06,
      "step": 2860
    },
    {
      "epoch": 2.727508544921875e-06,
      "step": 2860,
      "training_step_time": 8.760563850402832
    },
    {
      "epoch": 2.7284622192382814e-06,
      "model_forward_time": 1.695373296737671,
      "step": 2861
    },
    {
      "epoch": 2.7284622192382814e-06,
      "step": 2861,
      "training_step_time": 7.819508790969849
    },
    {
      "epoch": 2.7294158935546876e-06,
      "model_forward_time": 1.7042946815490723,
      "step": 2862
    },
    {
      "epoch": 2.7294158935546876e-06,
      "step": 2862,
      "training_step_time": 7.731414794921875
    },
    {
      "epoch": 2.7303695678710937e-06,
      "model_forward_time": 1.7100348472595215,
      "step": 2863
    },
    {
      "epoch": 2.7303695678710937e-06,
      "step": 2863,
      "training_step_time": 7.716907262802124
    },
    {
      "epoch": 2.7313232421875e-06,
      "model_forward_time": 1.7087492942810059,
      "step": 2864
    },
    {
      "epoch": 2.7313232421875e-06,
      "step": 2864,
      "training_step_time": 7.965062856674194
    },
    {
      "epoch": 2.7322769165039065e-06,
      "model_forward_time": 1.7033674716949463,
      "step": 2865
    },
    {
      "epoch": 2.7322769165039065e-06,
      "step": 2865,
      "training_step_time": 9.757139921188354
    },
    {
      "epoch": 2.7332305908203126e-06,
      "model_forward_time": 1.7261996269226074,
      "step": 2866
    },
    {
      "epoch": 2.7332305908203126e-06,
      "step": 2866,
      "training_step_time": 8.793139696121216
    },
    {
      "epoch": 2.7341842651367187e-06,
      "model_forward_time": 0.9960618019104004,
      "step": 2867
    },
    {
      "epoch": 2.7341842651367187e-06,
      "step": 2867,
      "training_step_time": 8.0262610912323
    },
    {
      "epoch": 2.735137939453125e-06,
      "model_forward_time": 1.735133171081543,
      "step": 2868
    },
    {
      "epoch": 2.735137939453125e-06,
      "step": 2868,
      "training_step_time": 8.638164043426514
    },
    {
      "epoch": 2.7360916137695315e-06,
      "model_forward_time": 1.708578109741211,
      "step": 2869
    },
    {
      "epoch": 2.7360916137695315e-06,
      "step": 2869,
      "training_step_time": 8.415970802307129
    },
    {
      "epoch": 2.7370452880859376e-06,
      "grad_norm": 0.7461087107658386,
      "learning_rate": 5.738e-06,
      "loss": 0.0887,
      "step": 2870
    },
    {
      "epoch": 2.7370452880859376e-06,
      "model_forward_time": 2.2560060024261475,
      "step": 2870
    },
    {
      "dynamics_loss_avg": 0.06799757797271014,
      "epoch": 2.7370452880859376e-06,
      "step": 2870
    },
    {
      "action_loss_avg": 0.011011721566319466,
      "epoch": 2.7370452880859376e-06,
      "step": 2870
    },
    {
      "epoch": 2.7370452880859376e-06,
      "step": 2870,
      "training_step_time": 7.741071701049805
    },
    {
      "epoch": 2.7379989624023437e-06,
      "model_forward_time": 1.705277442932129,
      "step": 2871
    },
    {
      "epoch": 2.7379989624023437e-06,
      "step": 2871,
      "training_step_time": 8.05303168296814
    },
    {
      "epoch": 2.73895263671875e-06,
      "model_forward_time": 1.7095439434051514,
      "step": 2872
    },
    {
      "epoch": 2.73895263671875e-06,
      "step": 2872,
      "training_step_time": 7.6268134117126465
    },
    {
      "epoch": 2.7399063110351565e-06,
      "model_forward_time": 1.714799404144287,
      "step": 2873
    },
    {
      "epoch": 2.7399063110351565e-06,
      "step": 2873,
      "training_step_time": 7.866044044494629
    },
    {
      "epoch": 2.7408599853515626e-06,
      "model_forward_time": 1.7087161540985107,
      "step": 2874
    },
    {
      "epoch": 2.7408599853515626e-06,
      "step": 2874,
      "training_step_time": 7.5739898681640625
    },
    {
      "epoch": 2.7418136596679688e-06,
      "model_forward_time": 1.7167916297912598,
      "step": 2875
    },
    {
      "epoch": 2.7418136596679688e-06,
      "step": 2875,
      "training_step_time": 8.792927742004395
    },
    {
      "epoch": 2.742767333984375e-06,
      "model_forward_time": 1.762619972229004,
      "step": 2876
    },
    {
      "epoch": 2.742767333984375e-06,
      "step": 2876,
      "training_step_time": 8.608506679534912
    },
    {
      "epoch": 2.743721008300781e-06,
      "model_forward_time": 1.7089147567749023,
      "step": 2877
    },
    {
      "epoch": 2.743721008300781e-06,
      "step": 2877,
      "training_step_time": 9.517991781234741
    },
    {
      "epoch": 2.7446746826171876e-06,
      "model_forward_time": 1.7016880512237549,
      "step": 2878
    },
    {
      "epoch": 2.7446746826171876e-06,
      "step": 2878,
      "training_step_time": 7.700068950653076
    },
    {
      "epoch": 2.7456283569335938e-06,
      "model_forward_time": 2.2688918113708496,
      "step": 2879
    },
    {
      "epoch": 2.7456283569335938e-06,
      "step": 2879,
      "training_step_time": 7.678727149963379
    },
    {
      "epoch": 2.74658203125e-06,
      "grad_norm": 0.8592420816421509,
      "learning_rate": 5.758000000000001e-06,
      "loss": 0.0866,
      "step": 2880
    },
    {
      "epoch": 2.74658203125e-06,
      "model_forward_time": 1.7043485641479492,
      "step": 2880
    },
    {
      "dynamics_loss_avg": 0.06413273103535175,
      "epoch": 2.74658203125e-06,
      "step": 2880
    },
    {
      "action_loss_avg": 0.010874362802132964,
      "epoch": 2.74658203125e-06,
      "step": 2880
    },
    {
      "epoch": 2.74658203125e-06,
      "step": 2880,
      "training_step_time": 7.51873779296875
    },
    {
      "epoch": 2.747535705566406e-06,
      "model_forward_time": 1.713660478591919,
      "step": 2881
    },
    {
      "epoch": 2.747535705566406e-06,
      "step": 2881,
      "training_step_time": 8.094780921936035
    },
    {
      "epoch": 2.7484893798828126e-06,
      "model_forward_time": 1.7056090831756592,
      "step": 2882
    },
    {
      "epoch": 2.7484893798828126e-06,
      "step": 2882,
      "training_step_time": 7.8965044021606445
    },
    {
      "epoch": 2.7494430541992188e-06,
      "model_forward_time": 1.710341453552246,
      "step": 2883
    },
    {
      "epoch": 2.7494430541992188e-06,
      "step": 2883,
      "training_step_time": 8.254584312438965
    },
    {
      "epoch": 2.750396728515625e-06,
      "model_forward_time": 1.7072343826293945,
      "step": 2884
    },
    {
      "epoch": 2.750396728515625e-06,
      "step": 2884,
      "training_step_time": 8.632640600204468
    },
    {
      "epoch": 2.751350402832031e-06,
      "model_forward_time": 1.7381253242492676,
      "step": 2885
    },
    {
      "epoch": 2.751350402832031e-06,
      "step": 2885,
      "training_step_time": 7.701629877090454
    },
    {
      "epoch": 2.7523040771484376e-06,
      "model_forward_time": 1.6999809741973877,
      "step": 2886
    },
    {
      "epoch": 2.7523040771484376e-06,
      "step": 2886,
      "training_step_time": 7.677538871765137
    },
    {
      "epoch": 2.7532577514648438e-06,
      "model_forward_time": 0.9976844787597656,
      "step": 2887
    },
    {
      "epoch": 2.7532577514648438e-06,
      "step": 2887,
      "training_step_time": 7.6927618980407715
    },
    {
      "epoch": 2.75421142578125e-06,
      "model_forward_time": 1.714794397354126,
      "step": 2888
    },
    {
      "epoch": 2.75421142578125e-06,
      "step": 2888,
      "training_step_time": 8.1149320602417
    },
    {
      "epoch": 2.755165100097656e-06,
      "model_forward_time": 1.7014575004577637,
      "step": 2889
    },
    {
      "epoch": 2.755165100097656e-06,
      "step": 2889,
      "training_step_time": 7.524476051330566
    },
    {
      "epoch": 2.7561187744140626e-06,
      "grad_norm": 0.7784494757652283,
      "learning_rate": 5.778e-06,
      "loss": 0.0915,
      "step": 2890
    },
    {
      "epoch": 2.7561187744140626e-06,
      "model_forward_time": 1.7049434185028076,
      "step": 2890
    },
    {
      "dynamics_loss_avg": 0.0670313449576497,
      "epoch": 2.7561187744140626e-06,
      "step": 2890
    },
    {
      "action_loss_avg": 0.011293287761509419,
      "epoch": 2.7561187744140626e-06,
      "step": 2890
    },
    {
      "epoch": 2.7561187744140626e-06,
      "step": 2890,
      "training_step_time": 7.577879190444946
    },
    {
      "epoch": 2.7570724487304688e-06,
      "model_forward_time": 1.7330663204193115,
      "step": 2891
    },
    {
      "epoch": 2.7570724487304688e-06,
      "step": 2891,
      "training_step_time": 7.807665824890137
    },
    {
      "epoch": 2.758026123046875e-06,
      "model_forward_time": 1.712181806564331,
      "step": 2892
    },
    {
      "epoch": 2.758026123046875e-06,
      "step": 2892,
      "training_step_time": 8.089876174926758
    },
    {
      "epoch": 2.758979797363281e-06,
      "model_forward_time": 1.709639072418213,
      "step": 2893
    },
    {
      "epoch": 2.758979797363281e-06,
      "step": 2893,
      "training_step_time": 7.62067174911499
    },
    {
      "epoch": 2.7599334716796876e-06,
      "model_forward_time": 1.7124354839324951,
      "step": 2894
    },
    {
      "epoch": 2.7599334716796876e-06,
      "step": 2894,
      "training_step_time": 9.180183172225952
    },
    {
      "epoch": 2.7608871459960938e-06,
      "model_forward_time": 1.7127883434295654,
      "step": 2895
    },
    {
      "epoch": 2.7608871459960938e-06,
      "step": 2895,
      "training_step_time": 8.446897029876709
    },
    {
      "epoch": 2.7618408203125e-06,
      "model_forward_time": 1.3445918560028076,
      "step": 2896
    },
    {
      "epoch": 2.7618408203125e-06,
      "step": 2896,
      "training_step_time": 8.727464199066162
    },
    {
      "epoch": 2.762794494628906e-06,
      "model_forward_time": 1.7119765281677246,
      "step": 2897
    },
    {
      "epoch": 2.762794494628906e-06,
      "step": 2897,
      "training_step_time": 7.60750150680542
    },
    {
      "epoch": 2.7637481689453126e-06,
      "model_forward_time": 2.2539520263671875,
      "step": 2898
    },
    {
      "epoch": 2.7637481689453126e-06,
      "step": 2898,
      "training_step_time": 7.712289571762085
    },
    {
      "epoch": 2.7647018432617188e-06,
      "model_forward_time": 1.7184007167816162,
      "step": 2899
    },
    {
      "epoch": 2.7647018432617188e-06,
      "step": 2899,
      "training_step_time": 7.581622123718262
    },
    {
      "epoch": 2.765655517578125e-06,
      "grad_norm": 0.7200440764427185,
      "learning_rate": 5.798e-06,
      "loss": 0.087,
      "step": 2900
    },
    {
      "epoch": 2.765655517578125e-06,
      "model_forward_time": 1.706144094467163,
      "step": 2900
    },
    {
      "dynamics_loss_avg": 0.07103471085429192,
      "epoch": 2.765655517578125e-06,
      "step": 2900
    },
    {
      "action_loss_avg": 0.013111055758781732,
      "epoch": 2.765655517578125e-06,
      "step": 2900
    },
    {
      "epoch": 2.765655517578125e-06,
      "step": 2900,
      "training_step_time": 7.54326605796814
    },
    {
      "epoch": 2.766609191894531e-06,
      "model_forward_time": 1.7148442268371582,
      "step": 2901
    },
    {
      "epoch": 2.766609191894531e-06,
      "step": 2901,
      "training_step_time": 9.005077123641968
    },
    {
      "epoch": 2.7675628662109376e-06,
      "model_forward_time": 1.7159507274627686,
      "step": 2902
    },
    {
      "epoch": 2.7675628662109376e-06,
      "step": 2902,
      "training_step_time": 7.650520324707031
    },
    {
      "epoch": 2.7685165405273438e-06,
      "model_forward_time": 1.7123565673828125,
      "step": 2903
    },
    {
      "epoch": 2.7685165405273438e-06,
      "step": 2903,
      "training_step_time": 8.475589513778687
    },
    {
      "epoch": 2.76947021484375e-06,
      "model_forward_time": 1.7671546936035156,
      "step": 2904
    },
    {
      "epoch": 2.76947021484375e-06,
      "step": 2904,
      "training_step_time": 7.837588310241699
    },
    {
      "epoch": 2.770423889160156e-06,
      "model_forward_time": 1.701369285583496,
      "step": 2905
    },
    {
      "epoch": 2.770423889160156e-06,
      "step": 2905,
      "training_step_time": 8.715500116348267
    },
    {
      "epoch": 2.7713775634765626e-06,
      "model_forward_time": 1.7035691738128662,
      "step": 2906
    },
    {
      "epoch": 2.7713775634765626e-06,
      "step": 2906,
      "training_step_time": 7.631449460983276
    },
    {
      "epoch": 2.7723312377929688e-06,
      "model_forward_time": 2.260866641998291,
      "step": 2907
    },
    {
      "epoch": 2.7723312377929688e-06,
      "step": 2907,
      "training_step_time": 8.616719484329224
    },
    {
      "epoch": 2.773284912109375e-06,
      "model_forward_time": 1.7102363109588623,
      "step": 2908
    },
    {
      "epoch": 2.773284912109375e-06,
      "step": 2908,
      "training_step_time": 7.630119562149048
    },
    {
      "epoch": 2.774238586425781e-06,
      "model_forward_time": 1.713080644607544,
      "step": 2909
    },
    {
      "epoch": 2.774238586425781e-06,
      "step": 2909,
      "training_step_time": 7.583975315093994
    },
    {
      "epoch": 2.7751922607421876e-06,
      "grad_norm": 0.5491101741790771,
      "learning_rate": 5.818e-06,
      "loss": 0.0835,
      "step": 2910
    },
    {
      "epoch": 2.7751922607421876e-06,
      "model_forward_time": 1.7054312229156494,
      "step": 2910
    },
    {
      "dynamics_loss_avg": 0.06936246920377016,
      "epoch": 2.7751922607421876e-06,
      "step": 2910
    },
    {
      "action_loss_avg": 0.014743001945316792,
      "epoch": 2.7751922607421876e-06,
      "step": 2910
    },
    {
      "epoch": 2.7751922607421876e-06,
      "step": 2910,
      "training_step_time": 7.835715055465698
    },
    {
      "epoch": 2.7761459350585938e-06,
      "model_forward_time": 1.7096221446990967,
      "step": 2911
    },
    {
      "epoch": 2.7761459350585938e-06,
      "step": 2911,
      "training_step_time": 9.01813292503357
    },
    {
      "epoch": 2.777099609375e-06,
      "model_forward_time": 1.715000867843628,
      "step": 2912
    },
    {
      "epoch": 2.777099609375e-06,
      "step": 2912,
      "training_step_time": 8.669054508209229
    },
    {
      "epoch": 2.778053283691406e-06,
      "model_forward_time": 1.7568809986114502,
      "step": 2913
    },
    {
      "epoch": 2.778053283691406e-06,
      "step": 2913,
      "training_step_time": 9.006573677062988
    },
    {
      "epoch": 2.7790069580078126e-06,
      "model_forward_time": 1.7092313766479492,
      "step": 2914
    },
    {
      "epoch": 2.7790069580078126e-06,
      "step": 2914,
      "training_step_time": 8.735036373138428
    },
    {
      "epoch": 2.7799606323242188e-06,
      "model_forward_time": 1.7058401107788086,
      "step": 2915
    },
    {
      "epoch": 2.7799606323242188e-06,
      "step": 2915,
      "training_step_time": 7.7320921421051025
    },
    {
      "epoch": 2.780914306640625e-06,
      "model_forward_time": 2.2602663040161133,
      "step": 2916
    },
    {
      "epoch": 2.780914306640625e-06,
      "step": 2916,
      "training_step_time": 7.754678249359131
    },
    {
      "epoch": 2.781867980957031e-06,
      "model_forward_time": 1.7079424858093262,
      "step": 2917
    },
    {
      "epoch": 2.781867980957031e-06,
      "step": 2917,
      "training_step_time": 7.630662202835083
    },
    {
      "epoch": 2.7828216552734376e-06,
      "model_forward_time": 1.7179529666900635,
      "step": 2918
    },
    {
      "epoch": 2.7828216552734376e-06,
      "step": 2918,
      "training_step_time": 7.608977317810059
    },
    {
      "epoch": 2.7837753295898438e-06,
      "model_forward_time": 1.7161085605621338,
      "step": 2919
    },
    {
      "epoch": 2.7837753295898438e-06,
      "step": 2919,
      "training_step_time": 8.600749731063843
    },
    {
      "epoch": 2.78472900390625e-06,
      "grad_norm": 0.6052378416061401,
      "learning_rate": 5.838000000000001e-06,
      "loss": 0.0862,
      "step": 2920
    },
    {
      "epoch": 2.78472900390625e-06,
      "model_forward_time": 1.70310378074646,
      "step": 2920
    },
    {
      "dynamics_loss_avg": 0.08396063819527626,
      "epoch": 2.78472900390625e-06,
      "step": 2920
    },
    {
      "action_loss_avg": 0.015846102219074964,
      "epoch": 2.78472900390625e-06,
      "step": 2920
    },
    {
      "epoch": 2.78472900390625e-06,
      "step": 2920,
      "training_step_time": 9.38492465019226
    },
    {
      "epoch": 2.785682678222656e-06,
      "model_forward_time": 1.722125768661499,
      "step": 2921
    },
    {
      "epoch": 2.785682678222656e-06,
      "step": 2921,
      "training_step_time": 8.190180540084839
    },
    {
      "epoch": 2.7866363525390626e-06,
      "model_forward_time": 1.7604069709777832,
      "step": 2922
    },
    {
      "epoch": 2.7866363525390626e-06,
      "step": 2922,
      "training_step_time": 7.853098154067993
    },
    {
      "epoch": 2.787590026855469e-06,
      "model_forward_time": 1.7134852409362793,
      "step": 2923
    },
    {
      "epoch": 2.787590026855469e-06,
      "step": 2923,
      "training_step_time": 7.756376028060913
    },
    {
      "epoch": 2.788543701171875e-06,
      "model_forward_time": 1.7049975395202637,
      "step": 2924
    },
    {
      "epoch": 2.788543701171875e-06,
      "step": 2924,
      "training_step_time": 8.733667612075806
    },
    {
      "epoch": 2.789497375488281e-06,
      "model_forward_time": 2.252761125564575,
      "step": 2925
    },
    {
      "epoch": 2.789497375488281e-06,
      "step": 2925,
      "training_step_time": 7.751163482666016
    },
    {
      "epoch": 2.7904510498046876e-06,
      "model_forward_time": 1.712939977645874,
      "step": 2926
    },
    {
      "epoch": 2.7904510498046876e-06,
      "step": 2926,
      "training_step_time": 7.6086907386779785
    },
    {
      "epoch": 2.791404724121094e-06,
      "model_forward_time": 1.710632085800171,
      "step": 2927
    },
    {
      "epoch": 2.791404724121094e-06,
      "step": 2927,
      "training_step_time": 7.546727180480957
    },
    {
      "epoch": 2.7923583984375e-06,
      "model_forward_time": 1.7228562831878662,
      "step": 2928
    },
    {
      "epoch": 2.7923583984375e-06,
      "step": 2928,
      "training_step_time": 7.907615423202515
    },
    {
      "epoch": 2.793312072753906e-06,
      "model_forward_time": 1.7215452194213867,
      "step": 2929
    },
    {
      "epoch": 2.793312072753906e-06,
      "step": 2929,
      "training_step_time": 7.699388027191162
    },
    {
      "epoch": 2.7942657470703127e-06,
      "grad_norm": 0.5102879405021667,
      "learning_rate": 5.8580000000000005e-06,
      "loss": 0.0793,
      "step": 2930
    },
    {
      "epoch": 2.7942657470703127e-06,
      "model_forward_time": 1.710733413696289,
      "step": 2930
    },
    {
      "dynamics_loss_avg": 0.0647917116060853,
      "epoch": 2.7942657470703127e-06,
      "step": 2930
    },
    {
      "action_loss_avg": 0.012060756236314774,
      "epoch": 2.7942657470703127e-06,
      "step": 2930
    },
    {
      "epoch": 2.7942657470703127e-06,
      "step": 2930,
      "training_step_time": 7.645444393157959
    },
    {
      "epoch": 2.795219421386719e-06,
      "model_forward_time": 1.7738192081451416,
      "step": 2931
    },
    {
      "epoch": 2.795219421386719e-06,
      "step": 2931,
      "training_step_time": 7.789735317230225
    },
    {
      "epoch": 2.796173095703125e-06,
      "model_forward_time": 1.7217516899108887,
      "step": 2932
    },
    {
      "epoch": 2.796173095703125e-06,
      "step": 2932,
      "training_step_time": 7.704691171646118
    },
    {
      "epoch": 2.797126770019531e-06,
      "model_forward_time": 1.7377662658691406,
      "step": 2933
    },
    {
      "epoch": 2.797126770019531e-06,
      "step": 2933,
      "training_step_time": 9.267850160598755
    },
    {
      "epoch": 2.7980804443359377e-06,
      "model_forward_time": 1.72586989402771,
      "step": 2934
    },
    {
      "epoch": 2.7980804443359377e-06,
      "step": 2934,
      "training_step_time": 7.6828882694244385
    },
    {
      "epoch": 2.799034118652344e-06,
      "model_forward_time": 2.272134304046631,
      "step": 2935
    },
    {
      "epoch": 2.799034118652344e-06,
      "step": 2935,
      "training_step_time": 7.5819010734558105
    },
    {
      "epoch": 2.79998779296875e-06,
      "model_forward_time": 1.717099905014038,
      "step": 2936
    },
    {
      "epoch": 2.79998779296875e-06,
      "step": 2936,
      "training_step_time": 7.437811374664307
    },
    {
      "epoch": 2.800941467285156e-06,
      "model_forward_time": 1.7172789573669434,
      "step": 2937
    },
    {
      "epoch": 2.800941467285156e-06,
      "step": 2937,
      "training_step_time": 8.363149881362915
    },
    {
      "epoch": 2.8018951416015627e-06,
      "model_forward_time": 1.7193963527679443,
      "step": 2938
    },
    {
      "epoch": 2.8018951416015627e-06,
      "step": 2938,
      "training_step_time": 7.6647868156433105
    },
    {
      "epoch": 2.802848815917969e-06,
      "model_forward_time": 1.71533203125,
      "step": 2939
    },
    {
      "epoch": 2.802848815917969e-06,
      "step": 2939,
      "training_step_time": 7.651494741439819
    },
    {
      "epoch": 2.803802490234375e-06,
      "grad_norm": 0.7018288969993591,
      "learning_rate": 5.878e-06,
      "loss": 0.0849,
      "step": 2940
    },
    {
      "epoch": 2.803802490234375e-06,
      "model_forward_time": 1.7542505264282227,
      "step": 2940
    },
    {
      "dynamics_loss_avg": 0.06800001263618469,
      "epoch": 2.803802490234375e-06,
      "step": 2940
    },
    {
      "action_loss_avg": 0.015590117918327451,
      "epoch": 2.803802490234375e-06,
      "step": 2940
    },
    {
      "epoch": 2.803802490234375e-06,
      "step": 2940,
      "training_step_time": 8.73773980140686
    },
    {
      "epoch": 2.804756164550781e-06,
      "model_forward_time": 1.7236700057983398,
      "step": 2941
    },
    {
      "epoch": 2.804756164550781e-06,
      "step": 2941,
      "training_step_time": 7.674607753753662
    },
    {
      "epoch": 2.8057098388671877e-06,
      "model_forward_time": 1.719421148300171,
      "step": 2942
    },
    {
      "epoch": 2.8057098388671877e-06,
      "step": 2942,
      "training_step_time": 8.525238513946533
    },
    {
      "epoch": 2.806663513183594e-06,
      "model_forward_time": 1.732337236404419,
      "step": 2943
    },
    {
      "epoch": 2.806663513183594e-06,
      "step": 2943,
      "training_step_time": 9.328893423080444
    },
    {
      "epoch": 2.8076171875e-06,
      "model_forward_time": 1.5360195636749268,
      "step": 2944
    },
    {
      "epoch": 2.8076171875e-06,
      "step": 2944,
      "training_step_time": 7.516772985458374
    },
    {
      "epoch": 2.808570861816406e-06,
      "model_forward_time": 1.7325847148895264,
      "step": 2945
    },
    {
      "epoch": 2.808570861816406e-06,
      "step": 2945,
      "training_step_time": 7.708778142929077
    },
    {
      "epoch": 2.8095245361328127e-06,
      "model_forward_time": 1.6942849159240723,
      "step": 2946
    },
    {
      "epoch": 2.8095245361328127e-06,
      "step": 2946,
      "training_step_time": 7.476863145828247
    },
    {
      "epoch": 2.810478210449219e-06,
      "model_forward_time": 1.7254743576049805,
      "step": 2947
    },
    {
      "epoch": 2.810478210449219e-06,
      "step": 2947,
      "training_step_time": 8.60790467262268
    },
    {
      "epoch": 2.811431884765625e-06,
      "model_forward_time": 1.7290873527526855,
      "step": 2948
    },
    {
      "epoch": 2.811431884765625e-06,
      "step": 2948,
      "training_step_time": 7.642696142196655
    },
    {
      "epoch": 2.812385559082031e-06,
      "model_forward_time": 1.7730987071990967,
      "step": 2949
    },
    {
      "epoch": 2.812385559082031e-06,
      "step": 2949,
      "training_step_time": 7.746987581253052
    },
    {
      "epoch": 2.8133392333984377e-06,
      "grad_norm": 0.6359254121780396,
      "learning_rate": 5.898e-06,
      "loss": 0.0846,
      "step": 2950
    },
    {
      "epoch": 2.8133392333984377e-06,
      "model_forward_time": 1.7177865505218506,
      "step": 2950
    },
    {
      "dynamics_loss_avg": 0.05663506090641022,
      "epoch": 2.8133392333984377e-06,
      "step": 2950
    },
    {
      "action_loss_avg": 0.014240727573633195,
      "epoch": 2.8133392333984377e-06,
      "step": 2950
    },
    {
      "epoch": 2.8133392333984377e-06,
      "step": 2950,
      "training_step_time": 7.664091110229492
    },
    {
      "epoch": 2.814292907714844e-06,
      "model_forward_time": 1.726269245147705,
      "step": 2951
    },
    {
      "epoch": 2.814292907714844e-06,
      "step": 2951,
      "training_step_time": 10.02431845664978
    },
    {
      "epoch": 2.81524658203125e-06,
      "model_forward_time": 1.7162189483642578,
      "step": 2952
    },
    {
      "epoch": 2.81524658203125e-06,
      "step": 2952,
      "training_step_time": 8.814639329910278
    },
    {
      "epoch": 2.816200256347656e-06,
      "model_forward_time": 1.7172868251800537,
      "step": 2953
    },
    {
      "epoch": 2.816200256347656e-06,
      "step": 2953,
      "training_step_time": 6.771222114562988
    },
    {
      "epoch": 2.8171539306640627e-06,
      "model_forward_time": 1.7233192920684814,
      "step": 2954
    },
    {
      "epoch": 2.8171539306640627e-06,
      "step": 2954,
      "training_step_time": 6.8757710456848145
    },
    {
      "epoch": 2.818107604980469e-06,
      "model_forward_time": 1.725928544998169,
      "step": 2955
    },
    {
      "epoch": 2.818107604980469e-06,
      "step": 2955,
      "training_step_time": 6.885734796524048
    },
    {
      "epoch": 2.819061279296875e-06,
      "model_forward_time": 1.7164764404296875,
      "step": 2956
    },
    {
      "epoch": 2.819061279296875e-06,
      "step": 2956,
      "training_step_time": 6.8987438678741455
    },
    {
      "epoch": 2.820014953613281e-06,
      "model_forward_time": 1.7065601348876953,
      "step": 2957
    },
    {
      "epoch": 2.820014953613281e-06,
      "step": 2957,
      "training_step_time": 6.911817312240601
    },
    {
      "epoch": 2.8209686279296877e-06,
      "model_forward_time": 1.7307589054107666,
      "step": 2958
    },
    {
      "epoch": 2.8209686279296877e-06,
      "step": 2958,
      "training_step_time": 6.854657173156738
    },
    {
      "epoch": 2.821922302246094e-06,
      "model_forward_time": 1.7192749977111816,
      "step": 2959
    },
    {
      "epoch": 2.821922302246094e-06,
      "step": 2959,
      "training_step_time": 6.85644006729126
    },
    {
      "epoch": 2.8228759765625e-06,
      "grad_norm": 0.5663124918937683,
      "learning_rate": 5.918000000000001e-06,
      "loss": 0.0806,
      "step": 2960
    },
    {
      "epoch": 2.8228759765625e-06,
      "model_forward_time": 1.7152454853057861,
      "step": 2960
    },
    {
      "dynamics_loss_avg": 0.05684359073638916,
      "epoch": 2.8228759765625e-06,
      "step": 2960
    },
    {
      "action_loss_avg": 0.010736136906780302,
      "epoch": 2.8228759765625e-06,
      "step": 2960
    },
    {
      "epoch": 2.8228759765625e-06,
      "step": 2960,
      "training_step_time": 7.846283674240112
    },
    {
      "epoch": 2.823829650878906e-06,
      "model_forward_time": 1.776270866394043,
      "step": 2961
    },
    {
      "epoch": 2.823829650878906e-06,
      "step": 2961,
      "training_step_time": 8.20987606048584
    },
    {
      "epoch": 2.8247833251953127e-06,
      "model_forward_time": 1.7139439582824707,
      "step": 2962
    },
    {
      "epoch": 2.8247833251953127e-06,
      "step": 2962,
      "training_step_time": 7.926054954528809
    },
    {
      "epoch": 2.825736999511719e-06,
      "model_forward_time": 1.7264485359191895,
      "step": 2963
    },
    {
      "epoch": 2.825736999511719e-06,
      "step": 2963,
      "training_step_time": 6.862133979797363
    },
    {
      "epoch": 2.826690673828125e-06,
      "model_forward_time": 1.7188491821289062,
      "step": 2964
    },
    {
      "epoch": 2.826690673828125e-06,
      "step": 2964,
      "training_step_time": 6.862546682357788
    },
    {
      "epoch": 2.827644348144531e-06,
      "model_forward_time": 1.7310938835144043,
      "step": 2965
    },
    {
      "epoch": 2.827644348144531e-06,
      "step": 2965,
      "training_step_time": 6.84391713142395
    },
    {
      "epoch": 2.8285980224609377e-06,
      "model_forward_time": 1.7462189197540283,
      "step": 2966
    },
    {
      "epoch": 2.8285980224609377e-06,
      "step": 2966,
      "training_step_time": 6.851826906204224
    },
    {
      "epoch": 2.829551696777344e-06,
      "model_forward_time": 1.7248220443725586,
      "step": 2967
    },
    {
      "epoch": 2.829551696777344e-06,
      "step": 2967,
      "training_step_time": 6.844001770019531
    },
    {
      "epoch": 2.83050537109375e-06,
      "model_forward_time": 1.354628562927246,
      "step": 2968
    },
    {
      "epoch": 2.83050537109375e-06,
      "step": 2968,
      "training_step_time": 6.83829140663147
    },
    {
      "epoch": 2.831459045410156e-06,
      "model_forward_time": 1.728445291519165,
      "step": 2969
    },
    {
      "epoch": 2.831459045410156e-06,
      "step": 2969,
      "training_step_time": 6.8566460609436035
    },
    {
      "epoch": 2.8324127197265627e-06,
      "grad_norm": 0.4140446186065674,
      "learning_rate": 5.9380000000000006e-06,
      "loss": 0.0917,
      "step": 2970
    },
    {
      "epoch": 2.8324127197265627e-06,
      "model_forward_time": 2.2921903133392334,
      "step": 2970
    },
    {
      "dynamics_loss_avg": 0.06328570637851953,
      "epoch": 2.8324127197265627e-06,
      "step": 2970
    },
    {
      "action_loss_avg": 0.008168570220004768,
      "epoch": 2.8324127197265627e-06,
      "step": 2970
    },
    {
      "epoch": 2.8324127197265627e-06,
      "step": 2970,
      "training_step_time": 8.818434476852417
    },
    {
      "epoch": 2.833366394042969e-06,
      "model_forward_time": 1.7166039943695068,
      "step": 2971
    },
    {
      "epoch": 2.833366394042969e-06,
      "step": 2971,
      "training_step_time": 7.824826002120972
    },
    {
      "epoch": 2.834320068359375e-06,
      "model_forward_time": 1.715113878250122,
      "step": 2972
    },
    {
      "epoch": 2.834320068359375e-06,
      "step": 2972,
      "training_step_time": 7.297509431838989
    },
    {
      "epoch": 2.835273742675781e-06,
      "model_forward_time": 1.7248272895812988,
      "step": 2973
    },
    {
      "epoch": 2.835273742675781e-06,
      "step": 2973,
      "training_step_time": 6.82917857170105
    },
    {
      "epoch": 2.8362274169921877e-06,
      "model_forward_time": 1.7170159816741943,
      "step": 2974
    },
    {
      "epoch": 2.8362274169921877e-06,
      "step": 2974,
      "training_step_time": 6.831103563308716
    },
    {
      "epoch": 2.837181091308594e-06,
      "model_forward_time": 1.7182443141937256,
      "step": 2975
    },
    {
      "epoch": 2.837181091308594e-06,
      "step": 2975,
      "training_step_time": 6.880865097045898
    },
    {
      "epoch": 2.838134765625e-06,
      "model_forward_time": 1.7379169464111328,
      "step": 2976
    },
    {
      "epoch": 2.838134765625e-06,
      "step": 2976,
      "training_step_time": 6.869351387023926
    },
    {
      "epoch": 2.839088439941406e-06,
      "model_forward_time": 1.7182188034057617,
      "step": 2977
    },
    {
      "epoch": 2.839088439941406e-06,
      "step": 2977,
      "training_step_time": 6.897244930267334
    },
    {
      "epoch": 2.8400421142578127e-06,
      "model_forward_time": 1.7201547622680664,
      "step": 2978
    },
    {
      "epoch": 2.8400421142578127e-06,
      "step": 2978,
      "training_step_time": 6.890779495239258
    },
    {
      "epoch": 2.840995788574219e-06,
      "model_forward_time": 2.278125047683716,
      "step": 2979
    },
    {
      "epoch": 2.840995788574219e-06,
      "step": 2979,
      "training_step_time": 7.776012182235718
    },
    {
      "epoch": 2.841949462890625e-06,
      "grad_norm": 0.8295871615409851,
      "learning_rate": 5.958e-06,
      "loss": 0.0928,
      "step": 2980
    },
    {
      "epoch": 2.841949462890625e-06,
      "model_forward_time": 1.707183837890625,
      "step": 2980
    },
    {
      "dynamics_loss_avg": 0.0743509043008089,
      "epoch": 2.841949462890625e-06,
      "step": 2980
    },
    {
      "action_loss_avg": 0.013148440979421139,
      "epoch": 2.841949462890625e-06,
      "step": 2980
    },
    {
      "epoch": 2.841949462890625e-06,
      "step": 2980,
      "training_step_time": 7.958928108215332
    },
    {
      "epoch": 2.842903137207031e-06,
      "model_forward_time": 1.7121226787567139,
      "step": 2981
    },
    {
      "epoch": 2.842903137207031e-06,
      "step": 2981,
      "training_step_time": 7.7255637645721436
    },
    {
      "epoch": 2.8438568115234377e-06,
      "model_forward_time": 1.7221310138702393,
      "step": 2982
    },
    {
      "epoch": 2.8438568115234377e-06,
      "step": 2982,
      "training_step_time": 6.746142387390137
    },
    {
      "epoch": 2.844810485839844e-06,
      "model_forward_time": 1.7101855278015137,
      "step": 2983
    },
    {
      "epoch": 2.844810485839844e-06,
      "step": 2983,
      "training_step_time": 6.858870267868042
    },
    {
      "epoch": 2.84576416015625e-06,
      "model_forward_time": 1.7124426364898682,
      "step": 2984
    },
    {
      "epoch": 2.84576416015625e-06,
      "step": 2984,
      "training_step_time": 6.892436265945435
    },
    {
      "epoch": 2.846717834472656e-06,
      "model_forward_time": 1.717087984085083,
      "step": 2985
    },
    {
      "epoch": 2.846717834472656e-06,
      "step": 2985,
      "training_step_time": 6.861675024032593
    },
    {
      "epoch": 2.8476715087890627e-06,
      "model_forward_time": 1.7198207378387451,
      "step": 2986
    },
    {
      "epoch": 2.8476715087890627e-06,
      "step": 2986,
      "training_step_time": 6.8917083740234375
    },
    {
      "epoch": 2.848625183105469e-06,
      "model_forward_time": 1.7176883220672607,
      "step": 2987
    },
    {
      "epoch": 2.848625183105469e-06,
      "step": 2987,
      "training_step_time": 6.819646835327148
    },
    {
      "epoch": 2.849578857421875e-06,
      "model_forward_time": 2.2660348415374756,
      "step": 2988
    },
    {
      "epoch": 2.849578857421875e-06,
      "step": 2988,
      "training_step_time": 8.607621192932129
    },
    {
      "epoch": 2.850532531738281e-06,
      "model_forward_time": 1.3466532230377197,
      "step": 2989
    },
    {
      "epoch": 2.850532531738281e-06,
      "step": 2989,
      "training_step_time": 9.68768572807312
    },
    {
      "epoch": 2.8514862060546877e-06,
      "grad_norm": 0.7007991671562195,
      "learning_rate": 5.978e-06,
      "loss": 0.0847,
      "step": 2990
    },
    {
      "epoch": 2.8514862060546877e-06,
      "model_forward_time": 1.7068471908569336,
      "step": 2990
    },
    {
      "dynamics_loss_avg": 0.07774941325187683,
      "epoch": 2.8514862060546877e-06,
      "step": 2990
    },
    {
      "action_loss_avg": 0.012162189721129835,
      "epoch": 2.8514862060546877e-06,
      "step": 2990
    },
    {
      "epoch": 2.8514862060546877e-06,
      "step": 2990,
      "training_step_time": 7.720518589019775
    },
    {
      "epoch": 2.852439880371094e-06,
      "model_forward_time": 1.715198278427124,
      "step": 2991
    },
    {
      "epoch": 2.852439880371094e-06,
      "step": 2991,
      "training_step_time": 6.889345169067383
    },
    {
      "epoch": 2.8533935546875e-06,
      "model_forward_time": 1.713475227355957,
      "step": 2992
    },
    {
      "epoch": 2.8533935546875e-06,
      "step": 2992,
      "training_step_time": 6.8639235496521
    },
    {
      "epoch": 2.854347229003906e-06,
      "model_forward_time": 1.7123475074768066,
      "step": 2993
    },
    {
      "epoch": 2.854347229003906e-06,
      "step": 2993,
      "training_step_time": 6.829169988632202
    },
    {
      "epoch": 2.8553009033203127e-06,
      "model_forward_time": 1.7124204635620117,
      "step": 2994
    },
    {
      "epoch": 2.8553009033203127e-06,
      "step": 2994,
      "training_step_time": 6.870806932449341
    },
    {
      "epoch": 2.856254577636719e-06,
      "model_forward_time": 1.7042162418365479,
      "step": 2995
    },
    {
      "epoch": 2.856254577636719e-06,
      "step": 2995,
      "training_step_time": 6.857646942138672
    },
    {
      "epoch": 2.857208251953125e-06,
      "model_forward_time": 1.7200617790222168,
      "step": 2996
    },
    {
      "epoch": 2.857208251953125e-06,
      "step": 2996,
      "training_step_time": 6.838196039199829
    },
    {
      "epoch": 2.858161926269531e-06,
      "model_forward_time": 1.7151143550872803,
      "step": 2997
    },
    {
      "epoch": 2.858161926269531e-06,
      "step": 2997,
      "training_step_time": 8.236162424087524
    },
    {
      "epoch": 2.8591156005859377e-06,
      "model_forward_time": 1.7966089248657227,
      "step": 2998
    },
    {
      "epoch": 2.8591156005859377e-06,
      "step": 2998,
      "training_step_time": 8.86854362487793
    },
    {
      "epoch": 2.860069274902344e-06,
      "model_forward_time": 1.7553648948669434,
      "step": 2999
    },
    {
      "epoch": 2.860069274902344e-06,
      "step": 2999,
      "training_step_time": 7.8201000690460205
    },
    {
      "epoch": 2.86102294921875e-06,
      "grad_norm": 0.45285549759864807,
      "learning_rate": 5.998000000000001e-06,
      "loss": 0.0958,
      "step": 3000
    },
    {
      "epoch": 2.86102294921875e-06,
      "model_forward_time": 1.7130112648010254,
      "step": 3000
    },
    {
      "dynamics_loss_avg": 0.0736676000058651,
      "epoch": 2.86102294921875e-06,
      "step": 3000
    },
    {
      "action_loss_avg": 0.009021738381125032,
      "epoch": 2.86102294921875e-06,
      "step": 3000
    },
    {
      "epoch": 2.86102294921875e-06,
      "step": 3000,
      "training_step_time": 7.370667219161987
    },
    {
      "epoch": 2.861976623535156e-06,
      "model_forward_time": 1.7121050357818604,
      "step": 3001
    },
    {
      "epoch": 2.861976623535156e-06,
      "step": 3001,
      "training_step_time": 6.805949687957764
    },
    {
      "epoch": 2.8629302978515623e-06,
      "model_forward_time": 1.7142395973205566,
      "step": 3002
    },
    {
      "epoch": 2.8629302978515623e-06,
      "step": 3002,
      "training_step_time": 6.770607233047485
    },
    {
      "epoch": 2.863883972167969e-06,
      "model_forward_time": 1.7222232818603516,
      "step": 3003
    },
    {
      "epoch": 2.863883972167969e-06,
      "step": 3003,
      "training_step_time": 6.7620849609375
    },
    {
      "epoch": 2.864837646484375e-06,
      "model_forward_time": 1.72514009475708,
      "step": 3004
    },
    {
      "epoch": 2.864837646484375e-06,
      "step": 3004,
      "training_step_time": 6.841155052185059
    },
    {
      "epoch": 2.865791320800781e-06,
      "model_forward_time": 1.721482753753662,
      "step": 3005
    },
    {
      "epoch": 2.865791320800781e-06,
      "step": 3005,
      "training_step_time": 6.8021697998046875
    },
    {
      "epoch": 2.8667449951171873e-06,
      "model_forward_time": 2.652510404586792,
      "step": 3006
    },
    {
      "epoch": 2.8667449951171873e-06,
      "step": 3006,
      "training_step_time": 9.653651237487793
    },
    {
      "epoch": 2.867698669433594e-06,
      "model_forward_time": 1.774811029434204,
      "step": 3007
    },
    {
      "epoch": 2.867698669433594e-06,
      "step": 3007,
      "training_step_time": 13.96295714378357
    },
    {
      "epoch": 2.86865234375e-06,
      "model_forward_time": 1.7135624885559082,
      "step": 3008
    },
    {
      "epoch": 2.86865234375e-06,
      "step": 3008,
      "training_step_time": 7.797932147979736
    },
    {
      "epoch": 2.869606018066406e-06,
      "model_forward_time": 1.7111420631408691,
      "step": 3009
    },
    {
      "epoch": 2.869606018066406e-06,
      "step": 3009,
      "training_step_time": 7.487242698669434
    },
    {
      "epoch": 2.8705596923828123e-06,
      "grad_norm": 0.6073870658874512,
      "learning_rate": 6.018000000000001e-06,
      "loss": 0.093,
      "step": 3010
    },
    {
      "epoch": 2.8705596923828123e-06,
      "model_forward_time": 1.7118580341339111,
      "step": 3010
    },
    {
      "dynamics_loss_avg": 0.0670097216963768,
      "epoch": 2.8705596923828123e-06,
      "step": 3010
    },
    {
      "action_loss_avg": 0.020519908424466847,
      "epoch": 2.8705596923828123e-06,
      "step": 3010
    },
    {
      "epoch": 2.8705596923828123e-06,
      "step": 3010,
      "training_step_time": 6.774132013320923
    },
    {
      "epoch": 2.871513366699219e-06,
      "model_forward_time": 1.7304952144622803,
      "step": 3011
    },
    {
      "epoch": 2.871513366699219e-06,
      "step": 3011,
      "training_step_time": 6.833173751831055
    },
    {
      "epoch": 2.872467041015625e-06,
      "model_forward_time": 1.7155969142913818,
      "step": 3012
    },
    {
      "epoch": 2.872467041015625e-06,
      "step": 3012,
      "training_step_time": 6.838920593261719
    },
    {
      "epoch": 2.873420715332031e-06,
      "model_forward_time": 1.7063648700714111,
      "step": 3013
    },
    {
      "epoch": 2.873420715332031e-06,
      "step": 3013,
      "training_step_time": 6.816821575164795
    },
    {
      "epoch": 2.8743743896484373e-06,
      "model_forward_time": 1.7138197422027588,
      "step": 3014
    },
    {
      "epoch": 2.8743743896484373e-06,
      "step": 3014,
      "training_step_time": 6.869497776031494
    },
    {
      "epoch": 2.875328063964844e-06,
      "model_forward_time": 1.0101807117462158,
      "step": 3015
    },
    {
      "epoch": 2.875328063964844e-06,
      "step": 3015,
      "training_step_time": 7.549097537994385
    },
    {
      "epoch": 2.87628173828125e-06,
      "model_forward_time": 2.2983558177948,
      "step": 3016
    },
    {
      "epoch": 2.87628173828125e-06,
      "step": 3016,
      "training_step_time": 12.391932249069214
    },
    {
      "epoch": 2.877235412597656e-06,
      "model_forward_time": 1.7134857177734375,
      "step": 3017
    },
    {
      "epoch": 2.877235412597656e-06,
      "step": 3017,
      "training_step_time": 7.8176891803741455
    },
    {
      "epoch": 2.8781890869140623e-06,
      "model_forward_time": 1.7168607711791992,
      "step": 3018
    },
    {
      "epoch": 2.8781890869140623e-06,
      "step": 3018,
      "training_step_time": 7.629801273345947
    },
    {
      "epoch": 2.879142761230469e-06,
      "model_forward_time": 1.721095323562622,
      "step": 3019
    },
    {
      "epoch": 2.879142761230469e-06,
      "step": 3019,
      "training_step_time": 6.78348708152771
    },
    {
      "epoch": 2.880096435546875e-06,
      "grad_norm": 0.6296072006225586,
      "learning_rate": 6.0380000000000005e-06,
      "loss": 0.087,
      "step": 3020
    },
    {
      "epoch": 2.880096435546875e-06,
      "model_forward_time": 1.7109978199005127,
      "step": 3020
    },
    {
      "dynamics_loss_avg": 0.07553216032683849,
      "epoch": 2.880096435546875e-06,
      "step": 3020
    },
    {
      "action_loss_avg": 0.01607248152140528,
      "epoch": 2.880096435546875e-06,
      "step": 3020
    },
    {
      "epoch": 2.880096435546875e-06,
      "step": 3020,
      "training_step_time": 6.83202600479126
    },
    {
      "epoch": 2.881050109863281e-06,
      "model_forward_time": 1.7137093544006348,
      "step": 3021
    },
    {
      "epoch": 2.881050109863281e-06,
      "step": 3021,
      "training_step_time": 6.773812770843506
    },
    {
      "epoch": 2.8820037841796873e-06,
      "model_forward_time": 1.7188842296600342,
      "step": 3022
    },
    {
      "epoch": 2.8820037841796873e-06,
      "step": 3022,
      "training_step_time": 6.759171724319458
    },
    {
      "epoch": 2.882957458496094e-06,
      "model_forward_time": 1.713313341140747,
      "step": 3023
    },
    {
      "epoch": 2.882957458496094e-06,
      "step": 3023,
      "training_step_time": 6.773576498031616
    },
    {
      "epoch": 2.8839111328125e-06,
      "model_forward_time": 1.71915864944458,
      "step": 3024
    },
    {
      "epoch": 2.8839111328125e-06,
      "step": 3024,
      "training_step_time": 6.813970565795898
    },
    {
      "epoch": 2.884864807128906e-06,
      "model_forward_time": 2.2725770473480225,
      "step": 3025
    },
    {
      "epoch": 2.884864807128906e-06,
      "step": 3025,
      "training_step_time": 10.75642466545105
    },
    {
      "epoch": 2.8858184814453123e-06,
      "model_forward_time": 1.7177481651306152,
      "step": 3026
    },
    {
      "epoch": 2.8858184814453123e-06,
      "step": 3026,
      "training_step_time": 8.511122465133667
    },
    {
      "epoch": 2.886772155761719e-06,
      "model_forward_time": 1.7099175453186035,
      "step": 3027
    },
    {
      "epoch": 2.886772155761719e-06,
      "step": 3027,
      "training_step_time": 7.705794811248779
    },
    {
      "epoch": 2.887725830078125e-06,
      "model_forward_time": 1.7197062969207764,
      "step": 3028
    },
    {
      "epoch": 2.887725830078125e-06,
      "step": 3028,
      "training_step_time": 6.885381460189819
    },
    {
      "epoch": 2.888679504394531e-06,
      "model_forward_time": 1.7211828231811523,
      "step": 3029
    },
    {
      "epoch": 2.888679504394531e-06,
      "step": 3029,
      "training_step_time": 6.8140857219696045
    },
    {
      "epoch": 2.8896331787109373e-06,
      "grad_norm": 0.5617853999137878,
      "learning_rate": 6.058e-06,
      "loss": 0.085,
      "step": 3030
    },
    {
      "epoch": 2.8896331787109373e-06,
      "model_forward_time": 1.7116074562072754,
      "step": 3030
    },
    {
      "dynamics_loss_avg": 0.07888477295637131,
      "epoch": 2.8896331787109373e-06,
      "step": 3030
    },
    {
      "action_loss_avg": 0.016246466618031263,
      "epoch": 2.8896331787109373e-06,
      "step": 3030
    },
    {
      "epoch": 2.8896331787109373e-06,
      "step": 3030,
      "training_step_time": 6.794842958450317
    },
    {
      "epoch": 2.890586853027344e-06,
      "model_forward_time": 1.7162277698516846,
      "step": 3031
    },
    {
      "epoch": 2.890586853027344e-06,
      "step": 3031,
      "training_step_time": 6.863133668899536
    },
    {
      "epoch": 2.89154052734375e-06,
      "model_forward_time": 1.713231086730957,
      "step": 3032
    },
    {
      "epoch": 2.89154052734375e-06,
      "step": 3032,
      "training_step_time": 6.818930387496948
    },
    {
      "epoch": 2.892494201660156e-06,
      "model_forward_time": 1.7136921882629395,
      "step": 3033
    },
    {
      "epoch": 2.892494201660156e-06,
      "step": 3033,
      "training_step_time": 6.850935220718384
    },
    {
      "epoch": 2.8934478759765623e-06,
      "model_forward_time": 2.277907133102417,
      "step": 3034
    },
    {
      "epoch": 2.8934478759765623e-06,
      "step": 3034,
      "training_step_time": 8.598812103271484
    },
    {
      "epoch": 2.894401550292969e-06,
      "model_forward_time": 1.7120709419250488,
      "step": 3035
    },
    {
      "epoch": 2.894401550292969e-06,
      "step": 3035,
      "training_step_time": 8.929741144180298
    },
    {
      "epoch": 2.895355224609375e-06,
      "model_forward_time": 1.7149815559387207,
      "step": 3036
    },
    {
      "epoch": 2.895355224609375e-06,
      "step": 3036,
      "training_step_time": 7.865635871887207
    },
    {
      "epoch": 2.896308898925781e-06,
      "model_forward_time": 1.7225406169891357,
      "step": 3037
    },
    {
      "epoch": 2.896308898925781e-06,
      "step": 3037,
      "training_step_time": 6.840042591094971
    },
    {
      "epoch": 2.8972625732421873e-06,
      "model_forward_time": 1.7099547386169434,
      "step": 3038
    },
    {
      "epoch": 2.8972625732421873e-06,
      "step": 3038,
      "training_step_time": 6.849361181259155
    },
    {
      "epoch": 2.898216247558594e-06,
      "model_forward_time": 1.7152068614959717,
      "step": 3039
    },
    {
      "epoch": 2.898216247558594e-06,
      "step": 3039,
      "training_step_time": 6.83927583694458
    },
    {
      "epoch": 2.899169921875e-06,
      "grad_norm": 0.44357529282569885,
      "learning_rate": 6.078000000000001e-06,
      "loss": 0.083,
      "step": 3040
    },
    {
      "epoch": 2.899169921875e-06,
      "model_forward_time": 1.7120096683502197,
      "step": 3040
    },
    {
      "dynamics_loss_avg": 0.04962478056550026,
      "epoch": 2.899169921875e-06,
      "step": 3040
    },
    {
      "action_loss_avg": 0.010884755058214068,
      "epoch": 2.899169921875e-06,
      "step": 3040
    },
    {
      "epoch": 2.899169921875e-06,
      "step": 3040,
      "training_step_time": 6.838853120803833
    },
    {
      "epoch": 2.900123596191406e-06,
      "model_forward_time": 1.7153441905975342,
      "step": 3041
    },
    {
      "epoch": 2.900123596191406e-06,
      "step": 3041,
      "training_step_time": 6.822314023971558
    },
    {
      "epoch": 2.9010772705078123e-06,
      "model_forward_time": 1.7189898490905762,
      "step": 3042
    },
    {
      "epoch": 2.9010772705078123e-06,
      "step": 3042,
      "training_step_time": 6.837883234024048
    },
    {
      "epoch": 2.902030944824219e-06,
      "model_forward_time": 2.239835739135742,
      "step": 3043
    },
    {
      "epoch": 2.902030944824219e-06,
      "step": 3043,
      "training_step_time": 7.72314977645874
    },
    {
      "epoch": 2.902984619140625e-06,
      "model_forward_time": 1.7667250633239746,
      "step": 3044
    },
    {
      "epoch": 2.902984619140625e-06,
      "step": 3044,
      "training_step_time": 11.561999559402466
    },
    {
      "epoch": 2.903938293457031e-06,
      "model_forward_time": 1.7287371158599854,
      "step": 3045
    },
    {
      "epoch": 2.903938293457031e-06,
      "step": 3045,
      "training_step_time": 7.8773157596588135
    },
    {
      "epoch": 2.9048919677734373e-06,
      "model_forward_time": 1.3610267639160156,
      "step": 3046
    },
    {
      "epoch": 2.9048919677734373e-06,
      "step": 3046,
      "training_step_time": 7.421383619308472
    },
    {
      "epoch": 2.905845642089844e-06,
      "model_forward_time": 1.736924648284912,
      "step": 3047
    },
    {
      "epoch": 2.905845642089844e-06,
      "step": 3047,
      "training_step_time": 6.917378664016724
    },
    {
      "epoch": 2.90679931640625e-06,
      "model_forward_time": 1.7237486839294434,
      "step": 3048
    },
    {
      "epoch": 2.90679931640625e-06,
      "step": 3048,
      "training_step_time": 6.8667802810668945
    },
    {
      "epoch": 2.907752990722656e-06,
      "model_forward_time": 1.7284328937530518,
      "step": 3049
    },
    {
      "epoch": 2.907752990722656e-06,
      "step": 3049,
      "training_step_time": 6.849333763122559
    },
    {
      "epoch": 2.9087066650390623e-06,
      "grad_norm": 0.7444144487380981,
      "learning_rate": 6.098000000000001e-06,
      "loss": 0.0886,
      "step": 3050
    },
    {
      "epoch": 2.9087066650390623e-06,
      "model_forward_time": 1.7230935096740723,
      "step": 3050
    },
    {
      "dynamics_loss_avg": 0.05422552637755871,
      "epoch": 2.9087066650390623e-06,
      "step": 3050
    },
    {
      "action_loss_avg": 0.008000605436973274,
      "epoch": 2.9087066650390623e-06,
      "step": 3050
    },
    {
      "epoch": 2.9087066650390623e-06,
      "step": 3050,
      "training_step_time": 6.82204794883728
    },
    {
      "epoch": 2.909660339355469e-06,
      "model_forward_time": 1.7325890064239502,
      "step": 3051
    },
    {
      "epoch": 2.909660339355469e-06,
      "step": 3051,
      "training_step_time": 6.887925386428833
    },
    {
      "epoch": 2.910614013671875e-06,
      "model_forward_time": 1.7331087589263916,
      "step": 3052
    },
    {
      "epoch": 2.910614013671875e-06,
      "step": 3052,
      "training_step_time": 7.798830509185791
    },
    {
      "epoch": 2.911567687988281e-06,
      "model_forward_time": 2.3106508255004883,
      "step": 3053
    },
    {
      "epoch": 2.911567687988281e-06,
      "step": 3053,
      "training_step_time": 9.92766809463501
    },
    {
      "epoch": 2.9125213623046873e-06,
      "model_forward_time": 1.7198944091796875,
      "step": 3054
    },
    {
      "epoch": 2.9125213623046873e-06,
      "step": 3054,
      "training_step_time": 8.83034896850586
    },
    {
      "epoch": 2.913475036621094e-06,
      "model_forward_time": 1.7443304061889648,
      "step": 3055
    },
    {
      "epoch": 2.913475036621094e-06,
      "step": 3055,
      "training_step_time": 7.416200637817383
    },
    {
      "epoch": 2.9144287109375e-06,
      "model_forward_time": 1.7203662395477295,
      "step": 3056
    },
    {
      "epoch": 2.9144287109375e-06,
      "step": 3056,
      "training_step_time": 6.872997999191284
    },
    {
      "epoch": 2.915382385253906e-06,
      "model_forward_time": 1.7201063632965088,
      "step": 3057
    },
    {
      "epoch": 2.915382385253906e-06,
      "step": 3057,
      "training_step_time": 6.87749171257019
    },
    {
      "epoch": 2.9163360595703123e-06,
      "model_forward_time": 1.7221100330352783,
      "step": 3058
    },
    {
      "epoch": 2.9163360595703123e-06,
      "step": 3058,
      "training_step_time": 6.866112947463989
    },
    {
      "epoch": 2.917289733886719e-06,
      "model_forward_time": 1.7177674770355225,
      "step": 3059
    },
    {
      "epoch": 2.917289733886719e-06,
      "step": 3059,
      "training_step_time": 6.854041576385498
    },
    {
      "epoch": 2.918243408203125e-06,
      "grad_norm": 0.36162012815475464,
      "learning_rate": 6.1180000000000005e-06,
      "loss": 0.0886,
      "step": 3060
    },
    {
      "epoch": 2.918243408203125e-06,
      "model_forward_time": 1.7111916542053223,
      "step": 3060
    },
    {
      "dynamics_loss_avg": 0.08730459362268447,
      "epoch": 2.918243408203125e-06,
      "step": 3060
    },
    {
      "action_loss_avg": 0.013548508030362427,
      "epoch": 2.918243408203125e-06,
      "step": 3060
    },
    {
      "epoch": 2.918243408203125e-06,
      "step": 3060,
      "training_step_time": 6.880528211593628
    },
    {
      "epoch": 2.919197082519531e-06,
      "model_forward_time": 1.7128715515136719,
      "step": 3061
    },
    {
      "epoch": 2.919197082519531e-06,
      "step": 3061,
      "training_step_time": 7.722660779953003
    },
    {
      "epoch": 2.9201507568359373e-06,
      "model_forward_time": 2.267991304397583,
      "step": 3062
    },
    {
      "epoch": 2.9201507568359373e-06,
      "step": 3062,
      "training_step_time": 8.579712629318237
    },
    {
      "epoch": 2.921104431152344e-06,
      "model_forward_time": 1.7228960990905762,
      "step": 3063
    },
    {
      "epoch": 2.921104431152344e-06,
      "step": 3063,
      "training_step_time": 8.438040971755981
    },
    {
      "epoch": 2.92205810546875e-06,
      "model_forward_time": 1.7249095439910889,
      "step": 3064
    },
    {
      "epoch": 2.92205810546875e-06,
      "step": 3064,
      "training_step_time": 7.63434362411499
    },
    {
      "epoch": 2.923011779785156e-06,
      "model_forward_time": 1.710386037826538,
      "step": 3065
    },
    {
      "epoch": 2.923011779785156e-06,
      "step": 3065,
      "training_step_time": 6.815455436706543
    },
    {
      "epoch": 2.9239654541015623e-06,
      "model_forward_time": 1.7180516719818115,
      "step": 3066
    },
    {
      "epoch": 2.9239654541015623e-06,
      "step": 3066,
      "training_step_time": 6.8280029296875
    },
    {
      "epoch": 2.924919128417969e-06,
      "model_forward_time": 1.72023606300354,
      "step": 3067
    },
    {
      "epoch": 2.924919128417969e-06,
      "step": 3067,
      "training_step_time": 6.800769567489624
    },
    {
      "epoch": 2.925872802734375e-06,
      "model_forward_time": 1.3452143669128418,
      "step": 3068
    },
    {
      "epoch": 2.925872802734375e-06,
      "step": 3068,
      "training_step_time": 6.827928304672241
    },
    {
      "epoch": 2.926826477050781e-06,
      "model_forward_time": 1.726698875427246,
      "step": 3069
    },
    {
      "epoch": 2.926826477050781e-06,
      "step": 3069,
      "training_step_time": 6.856481313705444
    },
    {
      "epoch": 2.9277801513671873e-06,
      "grad_norm": 0.3220575153827667,
      "learning_rate": 6.138e-06,
      "loss": 0.0845,
      "step": 3070
    },
    {
      "epoch": 2.9277801513671873e-06,
      "model_forward_time": 1.7204043865203857,
      "step": 3070
    },
    {
      "dynamics_loss_avg": 0.058553498797118665,
      "epoch": 2.9277801513671873e-06,
      "step": 3070
    },
    {
      "action_loss_avg": 0.009789923089556396,
      "epoch": 2.9277801513671873e-06,
      "step": 3070
    },
    {
      "epoch": 2.9277801513671873e-06,
      "step": 3070,
      "training_step_time": 7.711093902587891
    },
    {
      "epoch": 2.928733825683594e-06,
      "model_forward_time": 2.305583953857422,
      "step": 3071
    },
    {
      "epoch": 2.928733825683594e-06,
      "step": 3071,
      "training_step_time": 9.409262418746948
    },
    {
      "epoch": 2.9296875e-06,
      "model_forward_time": 1.7336688041687012,
      "step": 3072
    },
    {
      "epoch": 2.9296875e-06,
      "step": 3072,
      "training_step_time": 8.953672647476196
    },
    {
      "epoch": 2.930641174316406e-06,
      "model_forward_time": 1.7242438793182373,
      "step": 3073
    },
    {
      "epoch": 2.930641174316406e-06,
      "step": 3073,
      "training_step_time": 7.593730211257935
    },
    {
      "epoch": 2.9315948486328124e-06,
      "model_forward_time": 1.7250750064849854,
      "step": 3074
    },
    {
      "epoch": 2.9315948486328124e-06,
      "step": 3074,
      "training_step_time": 6.693201303482056
    },
    {
      "epoch": 2.932548522949219e-06,
      "model_forward_time": 1.7249360084533691,
      "step": 3075
    },
    {
      "epoch": 2.932548522949219e-06,
      "step": 3075,
      "training_step_time": 6.837391138076782
    },
    {
      "epoch": 2.933502197265625e-06,
      "model_forward_time": 1.7162585258483887,
      "step": 3076
    },
    {
      "epoch": 2.933502197265625e-06,
      "step": 3076,
      "training_step_time": 6.827170133590698
    },
    {
      "epoch": 2.934455871582031e-06,
      "model_forward_time": 1.726912021636963,
      "step": 3077
    },
    {
      "epoch": 2.934455871582031e-06,
      "step": 3077,
      "training_step_time": 6.8514134883880615
    },
    {
      "epoch": 2.9354095458984374e-06,
      "model_forward_time": 1.7251951694488525,
      "step": 3078
    },
    {
      "epoch": 2.9354095458984374e-06,
      "step": 3078,
      "training_step_time": 6.821655750274658
    },
    {
      "epoch": 2.936363220214844e-06,
      "model_forward_time": 1.714484691619873,
      "step": 3079
    },
    {
      "epoch": 2.936363220214844e-06,
      "step": 3079,
      "training_step_time": 7.5840394496917725
    },
    {
      "epoch": 2.93731689453125e-06,
      "grad_norm": 0.44703853130340576,
      "learning_rate": 6.158000000000001e-06,
      "loss": 0.0813,
      "step": 3080
    },
    {
      "epoch": 2.93731689453125e-06,
      "model_forward_time": 2.2990520000457764,
      "step": 3080
    },
    {
      "dynamics_loss_avg": 0.057034398429095745,
      "epoch": 2.93731689453125e-06,
      "step": 3080
    },
    {
      "action_loss_avg": 0.008457150403410196,
      "epoch": 2.93731689453125e-06,
      "step": 3080
    },
    {
      "epoch": 2.93731689453125e-06,
      "step": 3080,
      "training_step_time": 8.64554500579834
    },
    {
      "epoch": 2.9382705688476562e-06,
      "model_forward_time": 1.7227387428283691,
      "step": 3081
    },
    {
      "epoch": 2.9382705688476562e-06,
      "step": 3081,
      "training_step_time": 9.93874192237854
    },
    {
      "epoch": 2.9392242431640624e-06,
      "model_forward_time": 1.7191379070281982,
      "step": 3082
    },
    {
      "epoch": 2.9392242431640624e-06,
      "step": 3082,
      "training_step_time": 7.6550843715667725
    },
    {
      "epoch": 2.940177917480469e-06,
      "model_forward_time": 1.7250208854675293,
      "step": 3083
    },
    {
      "epoch": 2.940177917480469e-06,
      "step": 3083,
      "training_step_time": 7.58802342414856
    },
    {
      "epoch": 2.941131591796875e-06,
      "model_forward_time": 1.724518060684204,
      "step": 3084
    },
    {
      "epoch": 2.941131591796875e-06,
      "step": 3084,
      "training_step_time": 6.730858564376831
    },
    {
      "epoch": 2.9420852661132812e-06,
      "model_forward_time": 1.7278108596801758,
      "step": 3085
    },
    {
      "epoch": 2.9420852661132812e-06,
      "step": 3085,
      "training_step_time": 6.708095073699951
    },
    {
      "epoch": 2.9430389404296874e-06,
      "model_forward_time": 1.7187423706054688,
      "step": 3086
    },
    {
      "epoch": 2.9430389404296874e-06,
      "step": 3086,
      "training_step_time": 6.798240900039673
    },
    {
      "epoch": 2.943992614746094e-06,
      "model_forward_time": 1.7232091426849365,
      "step": 3087
    },
    {
      "epoch": 2.943992614746094e-06,
      "step": 3087,
      "training_step_time": 6.791628360748291
    },
    {
      "epoch": 2.9449462890625e-06,
      "model_forward_time": 1.7208213806152344,
      "step": 3088
    },
    {
      "epoch": 2.9449462890625e-06,
      "step": 3088,
      "training_step_time": 7.306391954421997
    },
    {
      "epoch": 2.9458999633789062e-06,
      "model_forward_time": 2.2991044521331787,
      "step": 3089
    },
    {
      "epoch": 2.9458999633789062e-06,
      "step": 3089,
      "training_step_time": 8.170997381210327
    },
    {
      "epoch": 2.9468536376953124e-06,
      "grad_norm": 0.34621870517730713,
      "learning_rate": 6.178000000000001e-06,
      "loss": 0.0854,
      "step": 3090
    },
    {
      "epoch": 2.9468536376953124e-06,
      "model_forward_time": 1.696375846862793,
      "step": 3090
    },
    {
      "dynamics_loss_avg": 0.046642627008259295,
      "epoch": 2.9468536376953124e-06,
      "step": 3090
    },
    {
      "action_loss_avg": 0.009564357018098235,
      "epoch": 2.9468536376953124e-06,
      "step": 3090
    },
    {
      "epoch": 2.9468536376953124e-06,
      "step": 3090,
      "training_step_time": 10.277771234512329
    },
    {
      "epoch": 2.947807312011719e-06,
      "model_forward_time": 1.7161734104156494,
      "step": 3091
    },
    {
      "epoch": 2.947807312011719e-06,
      "step": 3091,
      "training_step_time": 7.706646203994751
    },
    {
      "epoch": 2.948760986328125e-06,
      "model_forward_time": 1.7112090587615967,
      "step": 3092
    },
    {
      "epoch": 2.948760986328125e-06,
      "step": 3092,
      "training_step_time": 7.5185706615448
    },
    {
      "epoch": 2.9497146606445312e-06,
      "model_forward_time": 1.7073030471801758,
      "step": 3093
    },
    {
      "epoch": 2.9497146606445312e-06,
      "step": 3093,
      "training_step_time": 6.703812837600708
    },
    {
      "epoch": 2.9506683349609374e-06,
      "model_forward_time": 1.7161667346954346,
      "step": 3094
    },
    {
      "epoch": 2.9506683349609374e-06,
      "step": 3094,
      "training_step_time": 6.755074739456177
    },
    {
      "epoch": 2.951622009277344e-06,
      "model_forward_time": 1.7203819751739502,
      "step": 3095
    },
    {
      "epoch": 2.951622009277344e-06,
      "step": 3095,
      "training_step_time": 6.734292268753052
    },
    {
      "epoch": 2.95257568359375e-06,
      "model_forward_time": 1.7144396305084229,
      "step": 3096
    },
    {
      "epoch": 2.95257568359375e-06,
      "step": 3096,
      "training_step_time": 6.752782344818115
    },
    {
      "epoch": 2.9535293579101562e-06,
      "model_forward_time": 1.706681251525879,
      "step": 3097
    },
    {
      "epoch": 2.9535293579101562e-06,
      "step": 3097,
      "training_step_time": 7.274866104125977
    },
    {
      "epoch": 2.9544830322265624e-06,
      "model_forward_time": 1.7126891613006592,
      "step": 3098
    },
    {
      "epoch": 2.9544830322265624e-06,
      "step": 3098,
      "training_step_time": 8.087992191314697
    },
    {
      "epoch": 2.955436706542969e-06,
      "model_forward_time": 1.752918004989624,
      "step": 3099
    },
    {
      "epoch": 2.955436706542969e-06,
      "step": 3099,
      "training_step_time": 11.365686655044556
    },
    {
      "epoch": 2.956390380859375e-06,
      "grad_norm": 0.5796746015548706,
      "learning_rate": 6.198000000000001e-06,
      "loss": 0.088,
      "step": 3100
    },
    {
      "epoch": 2.956390380859375e-06,
      "model_forward_time": 1.343982458114624,
      "step": 3100
    },
    {
      "dynamics_loss_avg": 0.06603314876556396,
      "epoch": 2.956390380859375e-06,
      "step": 3100
    },
    {
      "action_loss_avg": 0.009017279371619225,
      "epoch": 2.956390380859375e-06,
      "step": 3100
    },
    {
      "epoch": 2.956390380859375e-06,
      "step": 3100,
      "training_step_time": 8.352476596832275
    },
    {
      "epoch": 2.9573440551757812e-06,
      "model_forward_time": 1.7123146057128906,
      "step": 3101
    },
    {
      "epoch": 2.9573440551757812e-06,
      "step": 3101,
      "training_step_time": 7.5801286697387695
    },
    {
      "epoch": 2.9582977294921874e-06,
      "model_forward_time": 1.7167513370513916,
      "step": 3102
    },
    {
      "epoch": 2.9582977294921874e-06,
      "step": 3102,
      "training_step_time": 6.812522888183594
    },
    {
      "epoch": 2.959251403808594e-06,
      "model_forward_time": 1.711038589477539,
      "step": 3103
    },
    {
      "epoch": 2.959251403808594e-06,
      "step": 3103,
      "training_step_time": 6.763324975967407
    },
    {
      "epoch": 2.960205078125e-06,
      "model_forward_time": 1.7183465957641602,
      "step": 3104
    },
    {
      "epoch": 2.960205078125e-06,
      "step": 3104,
      "training_step_time": 6.7149670124053955
    },
    {
      "epoch": 2.9611587524414062e-06,
      "model_forward_time": 1.715423345565796,
      "step": 3105
    },
    {
      "epoch": 2.9611587524414062e-06,
      "step": 3105,
      "training_step_time": 6.715869665145874
    },
    {
      "epoch": 2.9621124267578124e-06,
      "model_forward_time": 1.7066528797149658,
      "step": 3106
    },
    {
      "epoch": 2.9621124267578124e-06,
      "step": 3106,
      "training_step_time": 7.222988605499268
    },
    {
      "epoch": 2.963066101074219e-06,
      "model_forward_time": 1.7178843021392822,
      "step": 3107
    },
    {
      "epoch": 2.963066101074219e-06,
      "step": 3107,
      "training_step_time": 7.575603246688843
    },
    {
      "epoch": 2.964019775390625e-06,
      "model_forward_time": 1.9637572765350342,
      "step": 3108
    },
    {
      "epoch": 2.964019775390625e-06,
      "step": 3108,
      "training_step_time": 12.297405004501343
    },
    {
      "epoch": 2.9649734497070312e-06,
      "model_forward_time": 1.7215087413787842,
      "step": 3109
    },
    {
      "epoch": 2.9649734497070312e-06,
      "step": 3109,
      "training_step_time": 7.665103912353516
    },
    {
      "epoch": 2.9659271240234374e-06,
      "grad_norm": 0.3892357051372528,
      "learning_rate": 6.2180000000000004e-06,
      "loss": 0.0877,
      "step": 3110
    },
    {
      "epoch": 2.9659271240234374e-06,
      "model_forward_time": 1.7104253768920898,
      "step": 3110
    },
    {
      "dynamics_loss_avg": 0.08015019334852695,
      "epoch": 2.9659271240234374e-06,
      "step": 3110
    },
    {
      "action_loss_avg": 0.015872697299346327,
      "epoch": 2.9659271240234374e-06,
      "step": 3110
    },
    {
      "epoch": 2.9659271240234374e-06,
      "step": 3110,
      "training_step_time": 7.589301824569702
    },
    {
      "epoch": 2.966880798339844e-06,
      "model_forward_time": 1.7180325984954834,
      "step": 3111
    },
    {
      "epoch": 2.966880798339844e-06,
      "step": 3111,
      "training_step_time": 7.524266004562378
    },
    {
      "epoch": 2.96783447265625e-06,
      "model_forward_time": 1.7139520645141602,
      "step": 3112
    },
    {
      "epoch": 2.96783447265625e-06,
      "step": 3112,
      "training_step_time": 6.745237350463867
    },
    {
      "epoch": 2.9687881469726562e-06,
      "model_forward_time": 1.720609426498413,
      "step": 3113
    },
    {
      "epoch": 2.9687881469726562e-06,
      "step": 3113,
      "training_step_time": 6.7682106494903564
    },
    {
      "epoch": 2.9697418212890624e-06,
      "model_forward_time": 1.708186149597168,
      "step": 3114
    },
    {
      "epoch": 2.9697418212890624e-06,
      "step": 3114,
      "training_step_time": 6.759345054626465
    },
    {
      "epoch": 2.970695495605469e-06,
      "model_forward_time": 1.7101964950561523,
      "step": 3115
    },
    {
      "epoch": 2.970695495605469e-06,
      "step": 3115,
      "training_step_time": 6.7573864459991455
    },
    {
      "epoch": 2.971649169921875e-06,
      "model_forward_time": 1.704390287399292,
      "step": 3116
    },
    {
      "epoch": 2.971649169921875e-06,
      "step": 3116,
      "training_step_time": 7.559602499008179
    },
    {
      "epoch": 2.9726028442382812e-06,
      "model_forward_time": 2.2773003578186035,
      "step": 3117
    },
    {
      "epoch": 2.9726028442382812e-06,
      "step": 3117,
      "training_step_time": 9.654489517211914
    },
    {
      "epoch": 2.9735565185546874e-06,
      "model_forward_time": 1.7132434844970703,
      "step": 3118
    },
    {
      "epoch": 2.9735565185546874e-06,
      "step": 3118,
      "training_step_time": 10.869190216064453
    },
    {
      "epoch": 2.974510192871094e-06,
      "model_forward_time": 1.715552568435669,
      "step": 3119
    },
    {
      "epoch": 2.974510192871094e-06,
      "step": 3119,
      "training_step_time": 7.600360155105591
    },
    {
      "epoch": 2.9754638671875e-06,
      "grad_norm": 0.5903337001800537,
      "learning_rate": 6.238000000000001e-06,
      "loss": 0.0833,
      "step": 3120
    },
    {
      "epoch": 2.9754638671875e-06,
      "model_forward_time": 1.7100679874420166,
      "step": 3120
    },
    {
      "dynamics_loss_avg": 0.06768479477614164,
      "epoch": 2.9754638671875e-06,
      "step": 3120
    },
    {
      "action_loss_avg": 0.013240938680246473,
      "epoch": 2.9754638671875e-06,
      "step": 3120
    },
    {
      "epoch": 2.9754638671875e-06,
      "step": 3120,
      "training_step_time": 6.795110702514648
    },
    {
      "epoch": 2.9764175415039062e-06,
      "model_forward_time": 1.713808298110962,
      "step": 3121
    },
    {
      "epoch": 2.9764175415039062e-06,
      "step": 3121,
      "training_step_time": 6.767925262451172
    },
    {
      "epoch": 2.9773712158203124e-06,
      "model_forward_time": 1.7126235961914062,
      "step": 3122
    },
    {
      "epoch": 2.9773712158203124e-06,
      "step": 3122,
      "training_step_time": 6.760679483413696
    },
    {
      "epoch": 2.978324890136719e-06,
      "model_forward_time": 1.7253592014312744,
      "step": 3123
    },
    {
      "epoch": 2.978324890136719e-06,
      "step": 3123,
      "training_step_time": 6.752290964126587
    },
    {
      "epoch": 2.979278564453125e-06,
      "model_forward_time": 1.7278215885162354,
      "step": 3124
    },
    {
      "epoch": 2.979278564453125e-06,
      "step": 3124,
      "training_step_time": 6.7699408531188965
    },
    {
      "epoch": 2.9802322387695312e-06,
      "model_forward_time": 1.7161226272583008,
      "step": 3125
    },
    {
      "epoch": 2.9802322387695312e-06,
      "step": 3125,
      "training_step_time": 7.526260137557983
    },
    {
      "epoch": 2.9811859130859374e-06,
      "model_forward_time": 2.2764816284179688,
      "step": 3126
    },
    {
      "epoch": 2.9811859130859374e-06,
      "step": 3126,
      "training_step_time": 9.454021692276001
    },
    {
      "epoch": 2.9821395874023435e-06,
      "model_forward_time": 1.7108213901519775,
      "step": 3127
    },
    {
      "epoch": 2.9821395874023435e-06,
      "step": 3127,
      "training_step_time": 8.535335063934326
    },
    {
      "epoch": 2.98309326171875e-06,
      "model_forward_time": 1.7183587551116943,
      "step": 3128
    },
    {
      "epoch": 2.98309326171875e-06,
      "step": 3128,
      "training_step_time": 7.592503786087036
    },
    {
      "epoch": 2.9840469360351563e-06,
      "model_forward_time": 1.719675064086914,
      "step": 3129
    },
    {
      "epoch": 2.9840469360351563e-06,
      "step": 3129,
      "training_step_time": 7.619678497314453
    },
    {
      "epoch": 2.9850006103515624e-06,
      "grad_norm": 0.37413761019706726,
      "learning_rate": 6.258000000000001e-06,
      "loss": 0.0834,
      "step": 3130
    },
    {
      "epoch": 2.9850006103515624e-06,
      "model_forward_time": 1.7126986980438232,
      "step": 3130
    },
    {
      "dynamics_loss_avg": 0.061618888191878794,
      "epoch": 2.9850006103515624e-06,
      "step": 3130
    },
    {
      "action_loss_avg": 0.009940921212546527,
      "epoch": 2.9850006103515624e-06,
      "step": 3130
    },
    {
      "epoch": 2.9850006103515624e-06,
      "step": 3130,
      "training_step_time": 6.753148078918457
    },
    {
      "epoch": 2.9859542846679685e-06,
      "model_forward_time": 1.71954345703125,
      "step": 3131
    },
    {
      "epoch": 2.9859542846679685e-06,
      "step": 3131,
      "training_step_time": 6.7896037101745605
    },
    {
      "epoch": 2.986907958984375e-06,
      "model_forward_time": 1.7211956977844238,
      "step": 3132
    },
    {
      "epoch": 2.986907958984375e-06,
      "step": 3132,
      "training_step_time": 6.72526478767395
    },
    {
      "epoch": 2.9878616333007813e-06,
      "model_forward_time": 1.7201554775238037,
      "step": 3133
    },
    {
      "epoch": 2.9878616333007813e-06,
      "step": 3133,
      "training_step_time": 6.791717290878296
    },
    {
      "epoch": 2.9888153076171874e-06,
      "model_forward_time": 1.7234923839569092,
      "step": 3134
    },
    {
      "epoch": 2.9888153076171874e-06,
      "step": 3134,
      "training_step_time": 7.2975687980651855
    },
    {
      "epoch": 2.9897689819335935e-06,
      "model_forward_time": 1.7037041187286377,
      "step": 3135
    },
    {
      "epoch": 2.9897689819335935e-06,
      "step": 3135,
      "training_step_time": 8.980590343475342
    },
    {
      "epoch": 2.99072265625e-06,
      "model_forward_time": 1.7133960723876953,
      "step": 3136
    },
    {
      "epoch": 2.99072265625e-06,
      "step": 3136,
      "training_step_time": 9.830415964126587
    },
    {
      "epoch": 2.9916763305664063e-06,
      "model_forward_time": 1.7067325115203857,
      "step": 3137
    },
    {
      "epoch": 2.9916763305664063e-06,
      "step": 3137,
      "training_step_time": 8.285929918289185
    },
    {
      "epoch": 2.9926300048828124e-06,
      "model_forward_time": 1.728261947631836,
      "step": 3138
    },
    {
      "epoch": 2.9926300048828124e-06,
      "step": 3138,
      "training_step_time": 7.60083794593811
    },
    {
      "epoch": 2.9935836791992186e-06,
      "model_forward_time": 1.718165397644043,
      "step": 3139
    },
    {
      "epoch": 2.9935836791992186e-06,
      "step": 3139,
      "training_step_time": 6.7877357006073
    },
    {
      "epoch": 2.994537353515625e-06,
      "grad_norm": 0.6174045205116272,
      "learning_rate": 6.278000000000001e-06,
      "loss": 0.0846,
      "step": 3140
    },
    {
      "epoch": 2.994537353515625e-06,
      "model_forward_time": 1.7114241123199463,
      "step": 3140
    },
    {
      "dynamics_loss_avg": 0.06207503415644169,
      "epoch": 2.994537353515625e-06,
      "step": 3140
    },
    {
      "action_loss_avg": 0.012189675541594624,
      "epoch": 2.994537353515625e-06,
      "step": 3140
    },
    {
      "epoch": 2.994537353515625e-06,
      "step": 3140,
      "training_step_time": 6.775171995162964
    },
    {
      "epoch": 2.9954910278320313e-06,
      "model_forward_time": 1.7186744213104248,
      "step": 3141
    },
    {
      "epoch": 2.9954910278320313e-06,
      "step": 3141,
      "training_step_time": 6.806804180145264
    },
    {
      "epoch": 2.9964447021484374e-06,
      "model_forward_time": 1.7121729850769043,
      "step": 3142
    },
    {
      "epoch": 2.9964447021484374e-06,
      "step": 3142,
      "training_step_time": 6.7961554527282715
    },
    {
      "epoch": 2.9973983764648436e-06,
      "model_forward_time": 1.7226343154907227,
      "step": 3143
    },
    {
      "epoch": 2.9973983764648436e-06,
      "step": 3143,
      "training_step_time": 7.31252646446228
    },
    {
      "epoch": 2.99835205078125e-06,
      "model_forward_time": 1.7053172588348389,
      "step": 3144
    },
    {
      "epoch": 2.99835205078125e-06,
      "step": 3144,
      "training_step_time": 7.62891149520874
    },
    {
      "epoch": 2.9993057250976563e-06,
      "model_forward_time": 2.267725944519043,
      "step": 3145
    },
    {
      "epoch": 2.9993057250976563e-06,
      "step": 3145,
      "training_step_time": 9.98839545249939
    },
    {
      "epoch": 3.0002593994140624e-06,
      "model_forward_time": 1.7165184020996094,
      "step": 3146
    },
    {
      "epoch": 3.0002593994140624e-06,
      "step": 3146,
      "training_step_time": 7.66980767250061
    },
    {
      "epoch": 3.0012130737304686e-06,
      "model_forward_time": 1.7158658504486084,
      "step": 3147
    },
    {
      "epoch": 3.0012130737304686e-06,
      "step": 3147,
      "training_step_time": 7.505699396133423
    },
    {
      "epoch": 3.002166748046875e-06,
      "model_forward_time": 1.7185077667236328,
      "step": 3148
    },
    {
      "epoch": 3.002166748046875e-06,
      "step": 3148,
      "training_step_time": 7.570618152618408
    },
    {
      "epoch": 3.0031204223632813e-06,
      "model_forward_time": 1.7120695114135742,
      "step": 3149
    },
    {
      "epoch": 3.0031204223632813e-06,
      "step": 3149,
      "training_step_time": 6.751596212387085
    },
    {
      "epoch": 3.0040740966796874e-06,
      "grad_norm": 0.46520406007766724,
      "learning_rate": 6.2980000000000005e-06,
      "loss": 0.0875,
      "step": 3150
    },
    {
      "epoch": 3.0040740966796874e-06,
      "model_forward_time": 1.7090351581573486,
      "step": 3150
    },
    {
      "dynamics_loss_avg": 0.07435129601508379,
      "epoch": 3.0040740966796874e-06,
      "step": 3150
    },
    {
      "action_loss_avg": 0.010658555093687027,
      "epoch": 3.0040740966796874e-06,
      "step": 3150
    },
    {
      "epoch": 3.0040740966796874e-06,
      "step": 3150,
      "training_step_time": 6.757356643676758
    },
    {
      "epoch": 3.0050277709960936e-06,
      "model_forward_time": 1.7154958248138428,
      "step": 3151
    },
    {
      "epoch": 3.0050277709960936e-06,
      "step": 3151,
      "training_step_time": 6.747749328613281
    },
    {
      "epoch": 3.0059814453125e-06,
      "model_forward_time": 1.345529556274414,
      "step": 3152
    },
    {
      "epoch": 3.0059814453125e-06,
      "step": 3152,
      "training_step_time": 7.182913541793823
    },
    {
      "epoch": 3.0069351196289063e-06,
      "model_forward_time": 1.7193734645843506,
      "step": 3153
    },
    {
      "epoch": 3.0069351196289063e-06,
      "step": 3153,
      "training_step_time": 7.541845321655273
    },
    {
      "epoch": 3.0078887939453124e-06,
      "model_forward_time": 2.271366596221924,
      "step": 3154
    },
    {
      "epoch": 3.0078887939453124e-06,
      "step": 3154,
      "training_step_time": 8.978341817855835
    },
    {
      "epoch": 3.0088424682617186e-06,
      "model_forward_time": 1.7163341045379639,
      "step": 3155
    },
    {
      "epoch": 3.0088424682617186e-06,
      "step": 3155,
      "training_step_time": 8.86550498008728
    },
    {
      "epoch": 3.009796142578125e-06,
      "model_forward_time": 1.717712163925171,
      "step": 3156
    },
    {
      "epoch": 3.009796142578125e-06,
      "step": 3156,
      "training_step_time": 7.6340789794921875
    },
    {
      "epoch": 3.0107498168945313e-06,
      "model_forward_time": 1.714372158050537,
      "step": 3157
    },
    {
      "epoch": 3.0107498168945313e-06,
      "step": 3157,
      "training_step_time": 7.5335235595703125
    },
    {
      "epoch": 3.0117034912109374e-06,
      "model_forward_time": 1.7200279235839844,
      "step": 3158
    },
    {
      "epoch": 3.0117034912109374e-06,
      "step": 3158,
      "training_step_time": 6.791680574417114
    },
    {
      "epoch": 3.0126571655273436e-06,
      "model_forward_time": 1.7211198806762695,
      "step": 3159
    },
    {
      "epoch": 3.0126571655273436e-06,
      "step": 3159,
      "training_step_time": 6.726256370544434
    },
    {
      "epoch": 3.01361083984375e-06,
      "grad_norm": 0.43390703201293945,
      "learning_rate": 6.318000000000001e-06,
      "loss": 0.0794,
      "step": 3160
    },
    {
      "epoch": 3.01361083984375e-06,
      "model_forward_time": 1.7065784931182861,
      "step": 3160
    },
    {
      "dynamics_loss_avg": 0.06151476614177227,
      "epoch": 3.01361083984375e-06,
      "step": 3160
    },
    {
      "action_loss_avg": 0.006976993870921433,
      "epoch": 3.01361083984375e-06,
      "step": 3160
    },
    {
      "epoch": 3.01361083984375e-06,
      "step": 3160,
      "training_step_time": 6.736843585968018
    },
    {
      "epoch": 3.0145645141601563e-06,
      "model_forward_time": 1.7107155323028564,
      "step": 3161
    },
    {
      "epoch": 3.0145645141601563e-06,
      "step": 3161,
      "training_step_time": 7.139681339263916
    },
    {
      "epoch": 3.0155181884765624e-06,
      "model_forward_time": 1.709524393081665,
      "step": 3162
    },
    {
      "epoch": 3.0155181884765624e-06,
      "step": 3162,
      "training_step_time": 8.546880006790161
    },
    {
      "epoch": 3.0164718627929686e-06,
      "model_forward_time": 2.282804012298584,
      "step": 3163
    },
    {
      "epoch": 3.0164718627929686e-06,
      "step": 3163,
      "training_step_time": 9.757024765014648
    },
    {
      "epoch": 3.017425537109375e-06,
      "model_forward_time": 1.7152233123779297,
      "step": 3164
    },
    {
      "epoch": 3.017425537109375e-06,
      "step": 3164,
      "training_step_time": 9.3271164894104
    },
    {
      "epoch": 3.0183792114257813e-06,
      "model_forward_time": 1.7121267318725586,
      "step": 3165
    },
    {
      "epoch": 3.0183792114257813e-06,
      "step": 3165,
      "training_step_time": 7.604353666305542
    },
    {
      "epoch": 3.0193328857421874e-06,
      "model_forward_time": 1.716095209121704,
      "step": 3166
    },
    {
      "epoch": 3.0193328857421874e-06,
      "step": 3166,
      "training_step_time": 7.559510707855225
    },
    {
      "epoch": 3.0202865600585936e-06,
      "model_forward_time": 1.717156171798706,
      "step": 3167
    },
    {
      "epoch": 3.0202865600585936e-06,
      "step": 3167,
      "training_step_time": 6.747525691986084
    },
    {
      "epoch": 3.021240234375e-06,
      "model_forward_time": 1.7089293003082275,
      "step": 3168
    },
    {
      "epoch": 3.021240234375e-06,
      "step": 3168,
      "training_step_time": 6.789583444595337
    },
    {
      "epoch": 3.0221939086914063e-06,
      "model_forward_time": 1.7134287357330322,
      "step": 3169
    },
    {
      "epoch": 3.0221939086914063e-06,
      "step": 3169,
      "training_step_time": 6.725290060043335
    },
    {
      "epoch": 3.0231475830078124e-06,
      "grad_norm": 0.5523387789726257,
      "learning_rate": 6.338000000000001e-06,
      "loss": 0.0806,
      "step": 3170
    },
    {
      "epoch": 3.0231475830078124e-06,
      "model_forward_time": 1.713413953781128,
      "step": 3170
    },
    {
      "dynamics_loss_avg": 0.05910355020314455,
      "epoch": 3.0231475830078124e-06,
      "step": 3170
    },
    {
      "action_loss_avg": 0.009311811649240553,
      "epoch": 3.0231475830078124e-06,
      "step": 3170
    },
    {
      "epoch": 3.0231475830078124e-06,
      "step": 3170,
      "training_step_time": 7.085115909576416
    },
    {
      "epoch": 3.0241012573242186e-06,
      "model_forward_time": 1.7139582633972168,
      "step": 3171
    },
    {
      "epoch": 3.0241012573242186e-06,
      "step": 3171,
      "training_step_time": 7.543804407119751
    },
    {
      "epoch": 3.025054931640625e-06,
      "model_forward_time": 1.7733337879180908,
      "step": 3172
    },
    {
      "epoch": 3.025054931640625e-06,
      "step": 3172,
      "training_step_time": 8.985817909240723
    },
    {
      "epoch": 3.0260086059570313e-06,
      "model_forward_time": 1.7112350463867188,
      "step": 3173
    },
    {
      "epoch": 3.0260086059570313e-06,
      "step": 3173,
      "training_step_time": 8.602218866348267
    },
    {
      "epoch": 3.0269622802734374e-06,
      "model_forward_time": 1.71474027633667,
      "step": 3174
    },
    {
      "epoch": 3.0269622802734374e-06,
      "step": 3174,
      "training_step_time": 7.680612087249756
    },
    {
      "epoch": 3.0279159545898436e-06,
      "model_forward_time": 1.7198364734649658,
      "step": 3175
    },
    {
      "epoch": 3.0279159545898436e-06,
      "step": 3175,
      "training_step_time": 7.474264621734619
    },
    {
      "epoch": 3.02886962890625e-06,
      "model_forward_time": 1.7151217460632324,
      "step": 3176
    },
    {
      "epoch": 3.02886962890625e-06,
      "step": 3176,
      "training_step_time": 7.156277418136597
    },
    {
      "epoch": 3.0298233032226563e-06,
      "model_forward_time": 1.712653636932373,
      "step": 3177
    },
    {
      "epoch": 3.0298233032226563e-06,
      "step": 3177,
      "training_step_time": 6.761276483535767
    },
    {
      "epoch": 3.0307769775390624e-06,
      "model_forward_time": 1.7200407981872559,
      "step": 3178
    },
    {
      "epoch": 3.0307769775390624e-06,
      "step": 3178,
      "training_step_time": 6.751325845718384
    },
    {
      "epoch": 3.0317306518554686e-06,
      "model_forward_time": 1.712548017501831,
      "step": 3179
    },
    {
      "epoch": 3.0317306518554686e-06,
      "step": 3179,
      "training_step_time": 7.021381378173828
    },
    {
      "epoch": 3.032684326171875e-06,
      "grad_norm": 0.45200392603874207,
      "learning_rate": 6.358000000000001e-06,
      "loss": 0.0872,
      "step": 3180
    },
    {
      "epoch": 3.032684326171875e-06,
      "model_forward_time": 1.7119436264038086,
      "step": 3180
    },
    {
      "dynamics_loss_avg": 0.06983041763305664,
      "epoch": 3.032684326171875e-06,
      "step": 3180
    },
    {
      "action_loss_avg": 0.011739968648180365,
      "epoch": 3.032684326171875e-06,
      "step": 3180
    },
    {
      "epoch": 3.032684326171875e-06,
      "step": 3180,
      "training_step_time": 7.500689744949341
    },
    {
      "epoch": 3.0336380004882813e-06,
      "model_forward_time": 1.7657749652862549,
      "step": 3181
    },
    {
      "epoch": 3.0336380004882813e-06,
      "step": 3181,
      "training_step_time": 8.951380491256714
    },
    {
      "epoch": 3.0345916748046874e-06,
      "model_forward_time": 2.2481911182403564,
      "step": 3182
    },
    {
      "epoch": 3.0345916748046874e-06,
      "step": 3182,
      "training_step_time": 9.214169979095459
    },
    {
      "epoch": 3.0355453491210936e-06,
      "model_forward_time": 1.706103801727295,
      "step": 3183
    },
    {
      "epoch": 3.0355453491210936e-06,
      "step": 3183,
      "training_step_time": 9.593308687210083
    },
    {
      "epoch": 3.0364990234375e-06,
      "model_forward_time": 1.7053635120391846,
      "step": 3184
    },
    {
      "epoch": 3.0364990234375e-06,
      "step": 3184,
      "training_step_time": 7.315284729003906
    },
    {
      "epoch": 3.0374526977539063e-06,
      "model_forward_time": 1.716038465499878,
      "step": 3185
    },
    {
      "epoch": 3.0374526977539063e-06,
      "step": 3185,
      "training_step_time": 7.573500871658325
    },
    {
      "epoch": 3.0384063720703124e-06,
      "model_forward_time": 1.3497719764709473,
      "step": 3186
    },
    {
      "epoch": 3.0384063720703124e-06,
      "step": 3186,
      "training_step_time": 7.028538703918457
    },
    {
      "epoch": 3.0393600463867186e-06,
      "model_forward_time": 1.7136127948760986,
      "step": 3187
    },
    {
      "epoch": 3.0393600463867186e-06,
      "step": 3187,
      "training_step_time": 6.734438896179199
    },
    {
      "epoch": 3.040313720703125e-06,
      "model_forward_time": 1.7129592895507812,
      "step": 3188
    },
    {
      "epoch": 3.040313720703125e-06,
      "step": 3188,
      "training_step_time": 6.999847888946533
    },
    {
      "epoch": 3.0412673950195313e-06,
      "model_forward_time": 1.71122407913208,
      "step": 3189
    },
    {
      "epoch": 3.0412673950195313e-06,
      "step": 3189,
      "training_step_time": 7.98695707321167
    },
    {
      "epoch": 3.0422210693359374e-06,
      "grad_norm": 0.6448268890380859,
      "learning_rate": 6.378000000000001e-06,
      "loss": 0.0889,
      "step": 3190
    },
    {
      "epoch": 3.0422210693359374e-06,
      "model_forward_time": 1.7622957229614258,
      "step": 3190
    },
    {
      "dynamics_loss_avg": 0.06878645494580268,
      "epoch": 3.0422210693359374e-06,
      "step": 3190
    },
    {
      "action_loss_avg": 0.011331770243123174,
      "epoch": 3.0422210693359374e-06,
      "step": 3190
    },
    {
      "epoch": 3.0422210693359374e-06,
      "step": 3190,
      "training_step_time": 7.5996668338775635
    },
    {
      "epoch": 3.0431747436523436e-06,
      "model_forward_time": 2.2437469959259033,
      "step": 3191
    },
    {
      "epoch": 3.0431747436523436e-06,
      "step": 3191,
      "training_step_time": 9.28769564628601
    },
    {
      "epoch": 3.04412841796875e-06,
      "model_forward_time": 1.7066490650177002,
      "step": 3192
    },
    {
      "epoch": 3.04412841796875e-06,
      "step": 3192,
      "training_step_time": 8.315875053405762
    },
    {
      "epoch": 3.0450820922851563e-06,
      "model_forward_time": 1.7061948776245117,
      "step": 3193
    },
    {
      "epoch": 3.0450820922851563e-06,
      "step": 3193,
      "training_step_time": 7.585211992263794
    },
    {
      "epoch": 3.0460357666015625e-06,
      "model_forward_time": 1.7154150009155273,
      "step": 3194
    },
    {
      "epoch": 3.0460357666015625e-06,
      "step": 3194,
      "training_step_time": 7.635906219482422
    },
    {
      "epoch": 3.0469894409179686e-06,
      "model_forward_time": 1.7122058868408203,
      "step": 3195
    },
    {
      "epoch": 3.0469894409179686e-06,
      "step": 3195,
      "training_step_time": 7.077661514282227
    },
    {
      "epoch": 3.047943115234375e-06,
      "model_forward_time": 1.709557294845581,
      "step": 3196
    },
    {
      "epoch": 3.047943115234375e-06,
      "step": 3196,
      "training_step_time": 6.7204954624176025
    },
    {
      "epoch": 3.0488967895507813e-06,
      "model_forward_time": 1.7008967399597168,
      "step": 3197
    },
    {
      "epoch": 3.0488967895507813e-06,
      "step": 3197,
      "training_step_time": 6.788081169128418
    },
    {
      "epoch": 3.0498504638671875e-06,
      "model_forward_time": 1.330873966217041,
      "step": 3198
    },
    {
      "epoch": 3.0498504638671875e-06,
      "step": 3198,
      "training_step_time": 8.126003503799438
    },
    {
      "epoch": 3.0508041381835936e-06,
      "model_forward_time": 1.7639119625091553,
      "step": 3199
    },
    {
      "epoch": 3.0508041381835936e-06,
      "step": 3199,
      "training_step_time": 8.07793927192688
    },
    {
      "epoch": 3.0517578125e-06,
      "grad_norm": 0.6631144285202026,
      "learning_rate": 6.398000000000001e-06,
      "loss": 0.0919,
      "step": 3200
    },
    {
      "epoch": 3.0517578125e-06,
      "model_forward_time": 2.260232448577881,
      "step": 3200
    },
    {
      "dynamics_loss_avg": 0.06136277057230473,
      "epoch": 3.0517578125e-06,
      "step": 3200
    },
    {
      "action_loss_avg": 0.009157403343124315,
      "epoch": 3.0517578125e-06,
      "step": 3200
    },
    {
      "epoch": 3.0517578125e-06,
      "step": 3200,
      "training_step_time": 7.607076644897461
    },
    {
      "epoch": 3.0527114868164063e-06,
      "model_forward_time": 1.699582576751709,
      "step": 3201
    },
    {
      "epoch": 3.0527114868164063e-06,
      "step": 3201,
      "training_step_time": 7.738626718521118
    },
    {
      "epoch": 3.0536651611328125e-06,
      "model_forward_time": 1.338214635848999,
      "step": 3202
    },
    {
      "epoch": 3.0536651611328125e-06,
      "step": 3202,
      "training_step_time": 8.663570642471313
    },
    {
      "epoch": 3.0546188354492186e-06,
      "model_forward_time": 1.6855530738830566,
      "step": 3203
    },
    {
      "epoch": 3.0546188354492186e-06,
      "step": 3203,
      "training_step_time": 7.290525436401367
    },
    {
      "epoch": 3.055572509765625e-06,
      "model_forward_time": 1.7019431591033936,
      "step": 3204
    },
    {
      "epoch": 3.055572509765625e-06,
      "step": 3204,
      "training_step_time": 7.557599306106567
    },
    {
      "epoch": 3.0565261840820313e-06,
      "model_forward_time": 1.704303503036499,
      "step": 3205
    },
    {
      "epoch": 3.0565261840820313e-06,
      "step": 3205,
      "training_step_time": 7.025259733200073
    },
    {
      "epoch": 3.0574798583984375e-06,
      "model_forward_time": 1.698540449142456,
      "step": 3206
    },
    {
      "epoch": 3.0574798583984375e-06,
      "step": 3206,
      "training_step_time": 6.72403883934021
    },
    {
      "epoch": 3.0584335327148436e-06,
      "model_forward_time": 1.6993906497955322,
      "step": 3207
    },
    {
      "epoch": 3.0584335327148436e-06,
      "step": 3207,
      "training_step_time": 7.374090671539307
    },
    {
      "epoch": 3.05938720703125e-06,
      "model_forward_time": 1.7609879970550537,
      "step": 3208
    },
    {
      "epoch": 3.05938720703125e-06,
      "step": 3208,
      "training_step_time": 7.560774803161621
    },
    {
      "epoch": 3.0603408813476563e-06,
      "model_forward_time": 2.2131781578063965,
      "step": 3209
    },
    {
      "epoch": 3.0603408813476563e-06,
      "step": 3209,
      "training_step_time": 9.666139602661133
    },
    {
      "epoch": 3.0612945556640625e-06,
      "grad_norm": 0.5051037669181824,
      "learning_rate": 6.418000000000001e-06,
      "loss": 0.0851,
      "step": 3210
    },
    {
      "epoch": 3.0612945556640625e-06,
      "model_forward_time": 1.694192886352539,
      "step": 3210
    },
    {
      "dynamics_loss_avg": 0.0834938257932663,
      "epoch": 3.0612945556640625e-06,
      "step": 3210
    },
    {
      "action_loss_avg": 0.014691640809178352,
      "epoch": 3.0612945556640625e-06,
      "step": 3210
    },
    {
      "epoch": 3.0612945556640625e-06,
      "step": 3210,
      "training_step_time": 8.354911088943481
    },
    {
      "epoch": 3.0622482299804686e-06,
      "model_forward_time": 1.7325901985168457,
      "step": 3211
    },
    {
      "epoch": 3.0622482299804686e-06,
      "step": 3211,
      "training_step_time": 7.622900009155273
    },
    {
      "epoch": 3.063201904296875e-06,
      "model_forward_time": 1.7022626399993896,
      "step": 3212
    },
    {
      "epoch": 3.063201904296875e-06,
      "step": 3212,
      "training_step_time": 7.557897090911865
    },
    {
      "epoch": 3.0641555786132813e-06,
      "model_forward_time": 1.7336513996124268,
      "step": 3213
    },
    {
      "epoch": 3.0641555786132813e-06,
      "step": 3213,
      "training_step_time": 7.61444354057312
    },
    {
      "epoch": 3.0651092529296875e-06,
      "model_forward_time": 1.7034156322479248,
      "step": 3214
    },
    {
      "epoch": 3.0651092529296875e-06,
      "step": 3214,
      "training_step_time": 7.121702432632446
    },
    {
      "epoch": 3.0660629272460936e-06,
      "model_forward_time": 1.721327781677246,
      "step": 3215
    },
    {
      "epoch": 3.0660629272460936e-06,
      "step": 3215,
      "training_step_time": 6.772127866744995
    },
    {
      "epoch": 3.0670166015625e-06,
      "model_forward_time": 1.7169227600097656,
      "step": 3216
    },
    {
      "epoch": 3.0670166015625e-06,
      "step": 3216,
      "training_step_time": 7.0643980503082275
    },
    {
      "epoch": 3.0679702758789063e-06,
      "model_forward_time": 1.7776408195495605,
      "step": 3217
    },
    {
      "epoch": 3.0679702758789063e-06,
      "step": 3217,
      "training_step_time": 9.058321952819824
    },
    {
      "epoch": 3.0689239501953125e-06,
      "model_forward_time": 1.7155523300170898,
      "step": 3218
    },
    {
      "epoch": 3.0689239501953125e-06,
      "step": 3218,
      "training_step_time": 8.212249755859375
    },
    {
      "epoch": 3.0698776245117186e-06,
      "model_forward_time": 2.27528715133667,
      "step": 3219
    },
    {
      "epoch": 3.0698776245117186e-06,
      "step": 3219,
      "training_step_time": 8.468968868255615
    },
    {
      "epoch": 3.070831298828125e-06,
      "grad_norm": 0.6336765885353088,
      "learning_rate": 6.438000000000001e-06,
      "loss": 0.0835,
      "step": 3220
    },
    {
      "epoch": 3.070831298828125e-06,
      "model_forward_time": 1.7196757793426514,
      "step": 3220
    },
    {
      "dynamics_loss_avg": 0.05737768225371838,
      "epoch": 3.070831298828125e-06,
      "step": 3220
    },
    {
      "action_loss_avg": 0.011088831350207329,
      "epoch": 3.070831298828125e-06,
      "step": 3220
    },
    {
      "epoch": 3.070831298828125e-06,
      "step": 3220,
      "training_step_time": 7.70491361618042
    },
    {
      "epoch": 3.0717849731445313e-06,
      "model_forward_time": 1.718949794769287,
      "step": 3221
    },
    {
      "epoch": 3.0717849731445313e-06,
      "step": 3221,
      "training_step_time": 7.304858207702637
    },
    {
      "epoch": 3.0727386474609375e-06,
      "model_forward_time": 1.7159843444824219,
      "step": 3222
    },
    {
      "epoch": 3.0727386474609375e-06,
      "step": 3222,
      "training_step_time": 7.6762425899505615
    },
    {
      "epoch": 3.0736923217773436e-06,
      "model_forward_time": 1.7235355377197266,
      "step": 3223
    },
    {
      "epoch": 3.0736923217773436e-06,
      "step": 3223,
      "training_step_time": 7.732400894165039
    },
    {
      "epoch": 3.07464599609375e-06,
      "model_forward_time": 1.7454934120178223,
      "step": 3224
    },
    {
      "epoch": 3.07464599609375e-06,
      "step": 3224,
      "training_step_time": 6.807711362838745
    },
    {
      "epoch": 3.0755996704101563e-06,
      "model_forward_time": 1.7243986129760742,
      "step": 3225
    },
    {
      "epoch": 3.0755996704101563e-06,
      "step": 3225,
      "training_step_time": 7.095849514007568
    },
    {
      "epoch": 3.0765533447265625e-06,
      "model_forward_time": 1.7637591361999512,
      "step": 3226
    },
    {
      "epoch": 3.0765533447265625e-06,
      "step": 3226,
      "training_step_time": 7.575284481048584
    },
    {
      "epoch": 3.0775070190429686e-06,
      "model_forward_time": 1.8340144157409668,
      "step": 3227
    },
    {
      "epoch": 3.0775070190429686e-06,
      "step": 3227,
      "training_step_time": 7.626684665679932
    },
    {
      "epoch": 3.078460693359375e-06,
      "model_forward_time": 2.384186267852783,
      "step": 3228
    },
    {
      "epoch": 3.078460693359375e-06,
      "step": 3228,
      "training_step_time": 11.648139715194702
    },
    {
      "epoch": 3.0794143676757813e-06,
      "model_forward_time": 1.828596830368042,
      "step": 3229
    },
    {
      "epoch": 3.0794143676757813e-06,
      "step": 3229,
      "training_step_time": 8.787828207015991
    },
    {
      "epoch": 3.0803680419921875e-06,
      "grad_norm": 0.5203779935836792,
      "learning_rate": 6.458000000000001e-06,
      "loss": 0.0829,
      "step": 3230
    },
    {
      "epoch": 3.0803680419921875e-06,
      "model_forward_time": 1.836721420288086,
      "step": 3230
    },
    {
      "dynamics_loss_avg": 0.07770322076976299,
      "epoch": 3.0803680419921875e-06,
      "step": 3230
    },
    {
      "action_loss_avg": 0.018087345454841853,
      "epoch": 3.0803680419921875e-06,
      "step": 3230
    },
    {
      "epoch": 3.0803680419921875e-06,
      "step": 3230,
      "training_step_time": 7.615550994873047
    },
    {
      "epoch": 3.0813217163085936e-06,
      "model_forward_time": 1.8197021484375,
      "step": 3231
    },
    {
      "epoch": 3.0813217163085936e-06,
      "step": 3231,
      "training_step_time": 7.588228464126587
    },
    {
      "epoch": 3.082275390625e-06,
      "model_forward_time": 1.73592209815979,
      "step": 3232
    },
    {
      "epoch": 3.082275390625e-06,
      "step": 3232,
      "training_step_time": 7.189867258071899
    },
    {
      "epoch": 3.0832290649414063e-06,
      "model_forward_time": 1.7267744541168213,
      "step": 3233
    },
    {
      "epoch": 3.0832290649414063e-06,
      "step": 3233,
      "training_step_time": 7.038134336471558
    },
    {
      "epoch": 3.0841827392578125e-06,
      "model_forward_time": 1.736572027206421,
      "step": 3234
    },
    {
      "epoch": 3.0841827392578125e-06,
      "step": 3234,
      "training_step_time": 6.7592761516571045
    },
    {
      "epoch": 3.0851364135742186e-06,
      "model_forward_time": 1.754298210144043,
      "step": 3235
    },
    {
      "epoch": 3.0851364135742186e-06,
      "step": 3235,
      "training_step_time": 7.5009987354278564
    },
    {
      "epoch": 3.086090087890625e-06,
      "model_forward_time": 1.734264850616455,
      "step": 3236
    },
    {
      "epoch": 3.086090087890625e-06,
      "step": 3236,
      "training_step_time": 8.30982494354248
    },
    {
      "epoch": 3.0870437622070313e-06,
      "model_forward_time": 2.3255083560943604,
      "step": 3237
    },
    {
      "epoch": 3.0870437622070313e-06,
      "step": 3237,
      "training_step_time": 9.158113718032837
    },
    {
      "epoch": 3.0879974365234375e-06,
      "model_forward_time": 1.7293179035186768,
      "step": 3238
    },
    {
      "epoch": 3.0879974365234375e-06,
      "step": 3238,
      "training_step_time": 8.614994049072266
    },
    {
      "epoch": 3.0889511108398436e-06,
      "model_forward_time": 1.7309365272521973,
      "step": 3239
    },
    {
      "epoch": 3.0889511108398436e-06,
      "step": 3239,
      "training_step_time": 7.63934063911438
    },
    {
      "epoch": 3.08990478515625e-06,
      "grad_norm": 0.3783767521381378,
      "learning_rate": 6.478000000000001e-06,
      "loss": 0.0846,
      "step": 3240
    },
    {
      "epoch": 3.08990478515625e-06,
      "model_forward_time": 1.7360851764678955,
      "step": 3240
    },
    {
      "dynamics_loss_avg": 0.04698328021913767,
      "epoch": 3.08990478515625e-06,
      "step": 3240
    },
    {
      "action_loss_avg": 0.010712511790916323,
      "epoch": 3.08990478515625e-06,
      "step": 3240
    },
    {
      "epoch": 3.08990478515625e-06,
      "step": 3240,
      "training_step_time": 7.536470651626587
    },
    {
      "epoch": 3.0908584594726563e-06,
      "model_forward_time": 1.7377450466156006,
      "step": 3241
    },
    {
      "epoch": 3.0908584594726563e-06,
      "step": 3241,
      "training_step_time": 7.597336769104004
    },
    {
      "epoch": 3.0918121337890625e-06,
      "model_forward_time": 1.7302360534667969,
      "step": 3242
    },
    {
      "epoch": 3.0918121337890625e-06,
      "step": 3242,
      "training_step_time": 7.063556909561157
    },
    {
      "epoch": 3.0927658081054686e-06,
      "model_forward_time": 1.7207176685333252,
      "step": 3243
    },
    {
      "epoch": 3.0927658081054686e-06,
      "step": 3243,
      "training_step_time": 6.734522342681885
    },
    {
      "epoch": 3.093719482421875e-06,
      "model_forward_time": 1.7363331317901611,
      "step": 3244
    },
    {
      "epoch": 3.093719482421875e-06,
      "step": 3244,
      "training_step_time": 7.561209201812744
    },
    {
      "epoch": 3.0946731567382814e-06,
      "model_forward_time": 1.7891252040863037,
      "step": 3245
    },
    {
      "epoch": 3.0946731567382814e-06,
      "step": 3245,
      "training_step_time": 8.38471508026123
    },
    {
      "epoch": 3.0956268310546875e-06,
      "model_forward_time": 2.265014886856079,
      "step": 3246
    },
    {
      "epoch": 3.0956268310546875e-06,
      "step": 3246,
      "training_step_time": 8.368805646896362
    },
    {
      "epoch": 3.0965805053710936e-06,
      "model_forward_time": 1.7302806377410889,
      "step": 3247
    },
    {
      "epoch": 3.0965805053710936e-06,
      "step": 3247,
      "training_step_time": 9.573571681976318
    },
    {
      "epoch": 3.0975341796875e-06,
      "model_forward_time": 1.7312183380126953,
      "step": 3248
    },
    {
      "epoch": 3.0975341796875e-06,
      "step": 3248,
      "training_step_time": 7.686400651931763
    },
    {
      "epoch": 3.0984878540039064e-06,
      "model_forward_time": 1.7346487045288086,
      "step": 3249
    },
    {
      "epoch": 3.0984878540039064e-06,
      "step": 3249,
      "training_step_time": 7.317521095275879
    },
    {
      "epoch": 3.0994415283203125e-06,
      "grad_norm": 0.5768354535102844,
      "learning_rate": 6.498000000000001e-06,
      "loss": 0.0856,
      "step": 3250
    },
    {
      "epoch": 3.0994415283203125e-06,
      "model_forward_time": 1.726475715637207,
      "step": 3250
    },
    {
      "dynamics_loss_avg": 0.06883449107408524,
      "epoch": 3.0994415283203125e-06,
      "step": 3250
    },
    {
      "action_loss_avg": 0.011543247639201582,
      "epoch": 3.0994415283203125e-06,
      "step": 3250
    },
    {
      "epoch": 3.0994415283203125e-06,
      "step": 3250,
      "training_step_time": 7.5792152881622314
    },
    {
      "epoch": 3.1003952026367186e-06,
      "model_forward_time": 1.696141242980957,
      "step": 3251
    },
    {
      "epoch": 3.1003952026367186e-06,
      "step": 3251,
      "training_step_time": 7.086066961288452
    },
    {
      "epoch": 3.101348876953125e-06,
      "model_forward_time": 1.7157835960388184,
      "step": 3252
    },
    {
      "epoch": 3.101348876953125e-06,
      "step": 3252,
      "training_step_time": 6.7738401889801025
    },
    {
      "epoch": 3.1023025512695314e-06,
      "model_forward_time": 1.6957156658172607,
      "step": 3253
    },
    {
      "epoch": 3.1023025512695314e-06,
      "step": 3253,
      "training_step_time": 7.5351269245147705
    },
    {
      "epoch": 3.1032562255859375e-06,
      "model_forward_time": 1.7530245780944824,
      "step": 3254
    },
    {
      "epoch": 3.1032562255859375e-06,
      "step": 3254,
      "training_step_time": 7.6212170124053955
    },
    {
      "epoch": 3.1042098999023436e-06,
      "model_forward_time": 2.2193217277526855,
      "step": 3255
    },
    {
      "epoch": 3.1042098999023436e-06,
      "step": 3255,
      "training_step_time": 9.577853441238403
    },
    {
      "epoch": 3.10516357421875e-06,
      "model_forward_time": 1.7139325141906738,
      "step": 3256
    },
    {
      "epoch": 3.10516357421875e-06,
      "step": 3256,
      "training_step_time": 7.728087425231934
    },
    {
      "epoch": 3.1061172485351564e-06,
      "model_forward_time": 1.7119908332824707,
      "step": 3257
    },
    {
      "epoch": 3.1061172485351564e-06,
      "step": 3257,
      "training_step_time": 8.64716911315918
    },
    {
      "epoch": 3.1070709228515625e-06,
      "model_forward_time": 1.7122321128845215,
      "step": 3258
    },
    {
      "epoch": 3.1070709228515625e-06,
      "step": 3258,
      "training_step_time": 8.578726291656494
    },
    {
      "epoch": 3.1080245971679687e-06,
      "model_forward_time": 1.7145650386810303,
      "step": 3259
    },
    {
      "epoch": 3.1080245971679687e-06,
      "step": 3259,
      "training_step_time": 7.5363404750823975
    },
    {
      "epoch": 3.108978271484375e-06,
      "grad_norm": 0.429656058549881,
      "learning_rate": 6.518000000000001e-06,
      "loss": 0.0799,
      "step": 3260
    },
    {
      "epoch": 3.108978271484375e-06,
      "model_forward_time": 1.7166922092437744,
      "step": 3260
    },
    {
      "dynamics_loss_avg": 0.08304686471819878,
      "epoch": 3.108978271484375e-06,
      "step": 3260
    },
    {
      "action_loss_avg": 0.014502661023288965,
      "epoch": 3.108978271484375e-06,
      "step": 3260
    },
    {
      "epoch": 3.108978271484375e-06,
      "step": 3260,
      "training_step_time": 6.771023273468018
    },
    {
      "epoch": 3.1099319458007814e-06,
      "model_forward_time": 1.70562744140625,
      "step": 3261
    },
    {
      "epoch": 3.1099319458007814e-06,
      "step": 3261,
      "training_step_time": 6.7971789836883545
    },
    {
      "epoch": 3.1108856201171875e-06,
      "model_forward_time": 1.7128987312316895,
      "step": 3262
    },
    {
      "epoch": 3.1108856201171875e-06,
      "step": 3262,
      "training_step_time": 7.367306232452393
    },
    {
      "epoch": 3.1118392944335937e-06,
      "model_forward_time": 1.7470653057098389,
      "step": 3263
    },
    {
      "epoch": 3.1118392944335937e-06,
      "step": 3263,
      "training_step_time": 8.596691131591797
    },
    {
      "epoch": 3.11279296875e-06,
      "model_forward_time": 1.7093133926391602,
      "step": 3264
    },
    {
      "epoch": 3.11279296875e-06,
      "step": 3264,
      "training_step_time": 8.256340265274048
    },
    {
      "epoch": 3.1137466430664064e-06,
      "model_forward_time": 1.7046394348144531,
      "step": 3265
    },
    {
      "epoch": 3.1137466430664064e-06,
      "step": 3265,
      "training_step_time": 9.588726997375488
    },
    {
      "epoch": 3.1147003173828125e-06,
      "model_forward_time": 1.72868013381958,
      "step": 3266
    },
    {
      "epoch": 3.1147003173828125e-06,
      "step": 3266,
      "training_step_time": 8.603321313858032
    },
    {
      "epoch": 3.1156539916992187e-06,
      "model_forward_time": 1.7092039585113525,
      "step": 3267
    },
    {
      "epoch": 3.1156539916992187e-06,
      "step": 3267,
      "training_step_time": 8.686256885528564
    },
    {
      "epoch": 3.116607666015625e-06,
      "model_forward_time": 1.7105233669281006,
      "step": 3268
    },
    {
      "epoch": 3.116607666015625e-06,
      "step": 3268,
      "training_step_time": 7.56301736831665
    },
    {
      "epoch": 3.1175613403320314e-06,
      "model_forward_time": 1.7141478061676025,
      "step": 3269
    },
    {
      "epoch": 3.1175613403320314e-06,
      "step": 3269,
      "training_step_time": 6.738791227340698
    },
    {
      "epoch": 3.1185150146484375e-06,
      "grad_norm": 0.46175113320350647,
      "learning_rate": 6.538000000000001e-06,
      "loss": 0.0764,
      "step": 3270
    },
    {
      "epoch": 3.1185150146484375e-06,
      "model_forward_time": 1.7099411487579346,
      "step": 3270
    },
    {
      "dynamics_loss_avg": 0.0684265648946166,
      "epoch": 3.1185150146484375e-06,
      "step": 3270
    },
    {
      "action_loss_avg": 0.013508666306734085,
      "epoch": 3.1185150146484375e-06,
      "step": 3270
    },
    {
      "epoch": 3.1185150146484375e-06,
      "step": 3270,
      "training_step_time": 7.090924978256226
    },
    {
      "epoch": 3.1194686889648437e-06,
      "model_forward_time": 1.7117421627044678,
      "step": 3271
    },
    {
      "epoch": 3.1194686889648437e-06,
      "step": 3271,
      "training_step_time": 7.108937740325928
    },
    {
      "epoch": 3.12042236328125e-06,
      "model_forward_time": 1.7299175262451172,
      "step": 3272
    },
    {
      "epoch": 3.12042236328125e-06,
      "step": 3272,
      "training_step_time": 9.482843399047852
    },
    {
      "epoch": 3.1213760375976564e-06,
      "model_forward_time": 1.7115256786346436,
      "step": 3273
    },
    {
      "epoch": 3.1213760375976564e-06,
      "step": 3273,
      "training_step_time": 7.776581048965454
    },
    {
      "epoch": 3.1223297119140625e-06,
      "model_forward_time": 2.2458999156951904,
      "step": 3274
    },
    {
      "epoch": 3.1223297119140625e-06,
      "step": 3274,
      "training_step_time": 10.041152238845825
    },
    {
      "epoch": 3.1232833862304687e-06,
      "model_forward_time": 1.7077281475067139,
      "step": 3275
    },
    {
      "epoch": 3.1232833862304687e-06,
      "step": 3275,
      "training_step_time": 7.667965650558472
    },
    {
      "epoch": 3.124237060546875e-06,
      "model_forward_time": 1.7111873626708984,
      "step": 3276
    },
    {
      "epoch": 3.124237060546875e-06,
      "step": 3276,
      "training_step_time": 7.633256673812866
    },
    {
      "epoch": 3.1251907348632814e-06,
      "model_forward_time": 1.711437463760376,
      "step": 3277
    },
    {
      "epoch": 3.1251907348632814e-06,
      "step": 3277,
      "training_step_time": 7.648298978805542
    },
    {
      "epoch": 3.1261444091796875e-06,
      "model_forward_time": 1.7095282077789307,
      "step": 3278
    },
    {
      "epoch": 3.1261444091796875e-06,
      "step": 3278,
      "training_step_time": 7.578953742980957
    },
    {
      "epoch": 3.1270980834960937e-06,
      "model_forward_time": 1.710097074508667,
      "step": 3279
    },
    {
      "epoch": 3.1270980834960937e-06,
      "step": 3279,
      "training_step_time": 7.155958890914917
    },
    {
      "epoch": 3.1280517578125e-06,
      "grad_norm": 0.4312073588371277,
      "learning_rate": 6.558000000000001e-06,
      "loss": 0.0751,
      "step": 3280
    },
    {
      "epoch": 3.1280517578125e-06,
      "model_forward_time": 1.7120394706726074,
      "step": 3280
    },
    {
      "dynamics_loss_avg": 0.06659470330923796,
      "epoch": 3.1280517578125e-06,
      "step": 3280
    },
    {
      "action_loss_avg": 0.010534435883164406,
      "epoch": 3.1280517578125e-06,
      "step": 3280
    },
    {
      "epoch": 3.1280517578125e-06,
      "step": 3280,
      "training_step_time": 7.602741956710815
    },
    {
      "epoch": 3.1290054321289064e-06,
      "model_forward_time": 1.710334300994873,
      "step": 3281
    },
    {
      "epoch": 3.1290054321289064e-06,
      "step": 3281,
      "training_step_time": 8.306073665618896
    },
    {
      "epoch": 3.1299591064453125e-06,
      "model_forward_time": 1.7094173431396484,
      "step": 3282
    },
    {
      "epoch": 3.1299591064453125e-06,
      "step": 3282,
      "training_step_time": 8.788240432739258
    },
    {
      "epoch": 3.1309127807617187e-06,
      "model_forward_time": 2.236633539199829,
      "step": 3283
    },
    {
      "epoch": 3.1309127807617187e-06,
      "step": 3283,
      "training_step_time": 8.238797664642334
    },
    {
      "epoch": 3.131866455078125e-06,
      "model_forward_time": 1.7120418548583984,
      "step": 3284
    },
    {
      "epoch": 3.131866455078125e-06,
      "step": 3284,
      "training_step_time": 7.773312091827393
    },
    {
      "epoch": 3.1328201293945314e-06,
      "model_forward_time": 1.709507703781128,
      "step": 3285
    },
    {
      "epoch": 3.1328201293945314e-06,
      "step": 3285,
      "training_step_time": 7.70429539680481
    },
    {
      "epoch": 3.1337738037109375e-06,
      "model_forward_time": 1.7055137157440186,
      "step": 3286
    },
    {
      "epoch": 3.1337738037109375e-06,
      "step": 3286,
      "training_step_time": 7.68054723739624
    },
    {
      "epoch": 3.1347274780273437e-06,
      "model_forward_time": 1.7071812152862549,
      "step": 3287
    },
    {
      "epoch": 3.1347274780273437e-06,
      "step": 3287,
      "training_step_time": 7.70632791519165
    },
    {
      "epoch": 3.13568115234375e-06,
      "model_forward_time": 1.7034945487976074,
      "step": 3288
    },
    {
      "epoch": 3.13568115234375e-06,
      "step": 3288,
      "training_step_time": 7.0743327140808105
    },
    {
      "epoch": 3.1366348266601564e-06,
      "model_forward_time": 1.7260510921478271,
      "step": 3289
    },
    {
      "epoch": 3.1366348266601564e-06,
      "step": 3289,
      "training_step_time": 7.032280206680298
    },
    {
      "epoch": 3.1375885009765625e-06,
      "grad_norm": 0.6714953184127808,
      "learning_rate": 6.578000000000001e-06,
      "loss": 0.0813,
      "step": 3290
    },
    {
      "epoch": 3.1375885009765625e-06,
      "model_forward_time": 1.7043731212615967,
      "step": 3290
    },
    {
      "dynamics_loss_avg": 0.05453764330595732,
      "epoch": 3.1375885009765625e-06,
      "step": 3290
    },
    {
      "action_loss_avg": 0.006563161360099911,
      "epoch": 3.1375885009765625e-06,
      "step": 3290
    },
    {
      "epoch": 3.1375885009765625e-06,
      "step": 3290,
      "training_step_time": 8.568605422973633
    },
    {
      "epoch": 3.1385421752929687e-06,
      "model_forward_time": 1.764944314956665,
      "step": 3291
    },
    {
      "epoch": 3.1385421752929687e-06,
      "step": 3291,
      "training_step_time": 7.700159072875977
    },
    {
      "epoch": 3.139495849609375e-06,
      "model_forward_time": 2.2384495735168457,
      "step": 3292
    },
    {
      "epoch": 3.139495849609375e-06,
      "step": 3292,
      "training_step_time": 8.133536577224731
    },
    {
      "epoch": 3.1404495239257814e-06,
      "model_forward_time": 1.705676555633545,
      "step": 3293
    },
    {
      "epoch": 3.1404495239257814e-06,
      "step": 3293,
      "training_step_time": 9.06803297996521
    },
    {
      "epoch": 3.1414031982421875e-06,
      "model_forward_time": 1.7052114009857178,
      "step": 3294
    },
    {
      "epoch": 3.1414031982421875e-06,
      "step": 3294,
      "training_step_time": 7.672242879867554
    },
    {
      "epoch": 3.1423568725585937e-06,
      "model_forward_time": 1.7040131092071533,
      "step": 3295
    },
    {
      "epoch": 3.1423568725585937e-06,
      "step": 3295,
      "training_step_time": 7.591580152511597
    },
    {
      "epoch": 3.143310546875e-06,
      "model_forward_time": 1.7101266384124756,
      "step": 3296
    },
    {
      "epoch": 3.143310546875e-06,
      "step": 3296,
      "training_step_time": 7.582594156265259
    },
    {
      "epoch": 3.1442642211914064e-06,
      "model_forward_time": 1.7043347358703613,
      "step": 3297
    },
    {
      "epoch": 3.1442642211914064e-06,
      "step": 3297,
      "training_step_time": 6.714930057525635
    },
    {
      "epoch": 3.1452178955078125e-06,
      "model_forward_time": 1.7019262313842773,
      "step": 3298
    },
    {
      "epoch": 3.1452178955078125e-06,
      "step": 3298,
      "training_step_time": 7.0833892822265625
    },
    {
      "epoch": 3.1461715698242187e-06,
      "model_forward_time": 1.7002131938934326,
      "step": 3299
    },
    {
      "epoch": 3.1461715698242187e-06,
      "step": 3299,
      "training_step_time": 7.5945775508880615
    },
    {
      "epoch": 3.147125244140625e-06,
      "grad_norm": 0.5578919649124146,
      "learning_rate": 6.598000000000001e-06,
      "loss": 0.0811,
      "step": 3300
    },
    {
      "epoch": 3.147125244140625e-06,
      "model_forward_time": 1.767364740371704,
      "step": 3300
    },
    {
      "dynamics_loss_avg": 0.05666815862059593,
      "epoch": 3.147125244140625e-06,
      "step": 3300
    },
    {
      "action_loss_avg": 0.010522531857714058,
      "epoch": 3.147125244140625e-06,
      "step": 3300
    },
    {
      "epoch": 3.147125244140625e-06,
      "step": 3300,
      "training_step_time": 7.5564963817596436
    },
    {
      "epoch": 3.1480789184570314e-06,
      "model_forward_time": 2.2469329833984375,
      "step": 3301
    },
    {
      "epoch": 3.1480789184570314e-06,
      "step": 3301,
      "training_step_time": 7.683832883834839
    },
    {
      "epoch": 3.1490325927734375e-06,
      "model_forward_time": 1.7058558464050293,
      "step": 3302
    },
    {
      "epoch": 3.1490325927734375e-06,
      "step": 3302,
      "training_step_time": 8.425782680511475
    },
    {
      "epoch": 3.1499862670898437e-06,
      "model_forward_time": 1.7070717811584473,
      "step": 3303
    },
    {
      "epoch": 3.1499862670898437e-06,
      "step": 3303,
      "training_step_time": 8.283798694610596
    },
    {
      "epoch": 3.15093994140625e-06,
      "model_forward_time": 1.7070112228393555,
      "step": 3304
    },
    {
      "epoch": 3.15093994140625e-06,
      "step": 3304,
      "training_step_time": 7.6283347606658936
    },
    {
      "epoch": 3.1518936157226564e-06,
      "model_forward_time": 1.7144644260406494,
      "step": 3305
    },
    {
      "epoch": 3.1518936157226564e-06,
      "step": 3305,
      "training_step_time": 7.499189138412476
    },
    {
      "epoch": 3.1528472900390625e-06,
      "model_forward_time": 1.7169809341430664,
      "step": 3306
    },
    {
      "epoch": 3.1528472900390625e-06,
      "step": 3306,
      "training_step_time": 6.743375778198242
    },
    {
      "epoch": 3.1538009643554687e-06,
      "model_forward_time": 1.7056772708892822,
      "step": 3307
    },
    {
      "epoch": 3.1538009643554687e-06,
      "step": 3307,
      "training_step_time": 7.138723134994507
    },
    {
      "epoch": 3.154754638671875e-06,
      "model_forward_time": 1.7116260528564453,
      "step": 3308
    },
    {
      "epoch": 3.154754638671875e-06,
      "step": 3308,
      "training_step_time": 7.423245191574097
    },
    {
      "epoch": 3.1557083129882814e-06,
      "model_forward_time": 1.7470149993896484,
      "step": 3309
    },
    {
      "epoch": 3.1557083129882814e-06,
      "step": 3309,
      "training_step_time": 7.555297136306763
    },
    {
      "epoch": 3.1566619873046876e-06,
      "grad_norm": 0.5749732255935669,
      "learning_rate": 6.618000000000001e-06,
      "loss": 0.0825,
      "step": 3310
    },
    {
      "epoch": 3.1566619873046876e-06,
      "model_forward_time": 2.2557260990142822,
      "step": 3310
    },
    {
      "dynamics_loss_avg": 0.07330585792660713,
      "epoch": 3.1566619873046876e-06,
      "step": 3310
    },
    {
      "action_loss_avg": 0.010805806703865528,
      "epoch": 3.1566619873046876e-06,
      "step": 3310
    },
    {
      "epoch": 3.1566619873046876e-06,
      "step": 3310,
      "training_step_time": 9.35636043548584
    },
    {
      "epoch": 3.1576156616210937e-06,
      "model_forward_time": 1.339858055114746,
      "step": 3311
    },
    {
      "epoch": 3.1576156616210937e-06,
      "step": 3311,
      "training_step_time": 10.912752866744995
    },
    {
      "epoch": 3.1585693359375e-06,
      "model_forward_time": 1.7062044143676758,
      "step": 3312
    },
    {
      "epoch": 3.1585693359375e-06,
      "step": 3312,
      "training_step_time": 8.559966325759888
    },
    {
      "epoch": 3.1595230102539064e-06,
      "model_forward_time": 1.7087161540985107,
      "step": 3313
    },
    {
      "epoch": 3.1595230102539064e-06,
      "step": 3313,
      "training_step_time": 9.781939029693604
    },
    {
      "epoch": 3.1604766845703126e-06,
      "model_forward_time": 1.6909191608428955,
      "step": 3314
    },
    {
      "epoch": 3.1604766845703126e-06,
      "step": 3314,
      "training_step_time": 7.203949451446533
    },
    {
      "epoch": 3.1614303588867187e-06,
      "model_forward_time": 1.7095134258270264,
      "step": 3315
    },
    {
      "epoch": 3.1614303588867187e-06,
      "step": 3315,
      "training_step_time": 7.546405553817749
    },
    {
      "epoch": 3.162384033203125e-06,
      "model_forward_time": 1.7013297080993652,
      "step": 3316
    },
    {
      "epoch": 3.162384033203125e-06,
      "step": 3316,
      "training_step_time": 7.085105895996094
    },
    {
      "epoch": 3.1633377075195314e-06,
      "model_forward_time": 1.709728717803955,
      "step": 3317
    },
    {
      "epoch": 3.1633377075195314e-06,
      "step": 3317,
      "training_step_time": 7.472639322280884
    },
    {
      "epoch": 3.1642913818359376e-06,
      "model_forward_time": 1.7357122898101807,
      "step": 3318
    },
    {
      "epoch": 3.1642913818359376e-06,
      "step": 3318,
      "training_step_time": 7.49657678604126
    },
    {
      "epoch": 3.1652450561523437e-06,
      "model_forward_time": 1.7103323936462402,
      "step": 3319
    },
    {
      "epoch": 3.1652450561523437e-06,
      "step": 3319,
      "training_step_time": 8.168777227401733
    },
    {
      "epoch": 3.16619873046875e-06,
      "grad_norm": 0.49448636174201965,
      "learning_rate": 6.638e-06,
      "loss": 0.0897,
      "step": 3320
    },
    {
      "epoch": 3.16619873046875e-06,
      "model_forward_time": 2.2351973056793213,
      "step": 3320
    },
    {
      "dynamics_loss_avg": 0.06830191481858491,
      "epoch": 3.16619873046875e-06,
      "step": 3320
    },
    {
      "action_loss_avg": 0.01082023037597537,
      "epoch": 3.16619873046875e-06,
      "step": 3320
    },
    {
      "epoch": 3.16619873046875e-06,
      "step": 3320,
      "training_step_time": 9.757916927337646
    },
    {
      "epoch": 3.1671524047851564e-06,
      "model_forward_time": 1.720853328704834,
      "step": 3321
    },
    {
      "epoch": 3.1671524047851564e-06,
      "step": 3321,
      "training_step_time": 8.190461874008179
    },
    {
      "epoch": 3.1681060791015626e-06,
      "model_forward_time": 1.7204749584197998,
      "step": 3322
    },
    {
      "epoch": 3.1681060791015626e-06,
      "step": 3322,
      "training_step_time": 7.5948872566223145
    },
    {
      "epoch": 3.1690597534179687e-06,
      "model_forward_time": 1.7190616130828857,
      "step": 3323
    },
    {
      "epoch": 3.1690597534179687e-06,
      "step": 3323,
      "training_step_time": 7.58307671546936
    },
    {
      "epoch": 3.170013427734375e-06,
      "model_forward_time": 1.7198984622955322,
      "step": 3324
    },
    {
      "epoch": 3.170013427734375e-06,
      "step": 3324,
      "training_step_time": 7.476538419723511
    },
    {
      "epoch": 3.1709671020507814e-06,
      "model_forward_time": 1.715254783630371,
      "step": 3325
    },
    {
      "epoch": 3.1709671020507814e-06,
      "step": 3325,
      "training_step_time": 7.013628959655762
    },
    {
      "epoch": 3.1719207763671876e-06,
      "model_forward_time": 1.7187819480895996,
      "step": 3326
    },
    {
      "epoch": 3.1719207763671876e-06,
      "step": 3326,
      "training_step_time": 7.473530292510986
    },
    {
      "epoch": 3.1728744506835937e-06,
      "model_forward_time": 1.7158000469207764,
      "step": 3327
    },
    {
      "epoch": 3.1728744506835937e-06,
      "step": 3327,
      "training_step_time": 8.836848497390747
    },
    {
      "epoch": 3.173828125e-06,
      "model_forward_time": 1.707810640335083,
      "step": 3328
    },
    {
      "epoch": 3.173828125e-06,
      "step": 3328,
      "training_step_time": 8.811731100082397
    },
    {
      "epoch": 3.1747817993164064e-06,
      "model_forward_time": 2.222170829772949,
      "step": 3329
    },
    {
      "epoch": 3.1747817993164064e-06,
      "step": 3329,
      "training_step_time": 7.6014244556427
    },
    {
      "epoch": 3.1757354736328126e-06,
      "grad_norm": 0.41280490159988403,
      "learning_rate": 6.658e-06,
      "loss": 0.0861,
      "step": 3330
    },
    {
      "epoch": 3.1757354736328126e-06,
      "model_forward_time": 1.7066328525543213,
      "step": 3330
    },
    {
      "dynamics_loss_avg": 0.06733545325696469,
      "epoch": 3.1757354736328126e-06,
      "step": 3330
    },
    {
      "action_loss_avg": 0.010868690186180174,
      "epoch": 3.1757354736328126e-06,
      "step": 3330
    },
    {
      "epoch": 3.1757354736328126e-06,
      "step": 3330,
      "training_step_time": 7.7070159912109375
    },
    {
      "epoch": 3.1766891479492187e-06,
      "model_forward_time": 1.7073156833648682,
      "step": 3331
    },
    {
      "epoch": 3.1766891479492187e-06,
      "step": 3331,
      "training_step_time": 7.68382453918457
    },
    {
      "epoch": 3.177642822265625e-06,
      "model_forward_time": 1.7099800109863281,
      "step": 3332
    },
    {
      "epoch": 3.177642822265625e-06,
      "step": 3332,
      "training_step_time": 7.623546123504639
    },
    {
      "epoch": 3.1785964965820314e-06,
      "model_forward_time": 1.7124950885772705,
      "step": 3333
    },
    {
      "epoch": 3.1785964965820314e-06,
      "step": 3333,
      "training_step_time": 7.5131261348724365
    },
    {
      "epoch": 3.1795501708984376e-06,
      "model_forward_time": 1.7174127101898193,
      "step": 3334
    },
    {
      "epoch": 3.1795501708984376e-06,
      "step": 3334,
      "training_step_time": 7.006189346313477
    },
    {
      "epoch": 3.1805038452148437e-06,
      "model_forward_time": 1.7227234840393066,
      "step": 3335
    },
    {
      "epoch": 3.1805038452148437e-06,
      "step": 3335,
      "training_step_time": 7.608276605606079
    },
    {
      "epoch": 3.18145751953125e-06,
      "model_forward_time": 1.7208757400512695,
      "step": 3336
    },
    {
      "epoch": 3.18145751953125e-06,
      "step": 3336,
      "training_step_time": 7.520207405090332
    },
    {
      "epoch": 3.1824111938476564e-06,
      "model_forward_time": 1.7574658393859863,
      "step": 3337
    },
    {
      "epoch": 3.1824111938476564e-06,
      "step": 3337,
      "training_step_time": 8.472048282623291
    },
    {
      "epoch": 3.1833648681640626e-06,
      "model_forward_time": 2.2265214920043945,
      "step": 3338
    },
    {
      "epoch": 3.1833648681640626e-06,
      "step": 3338,
      "training_step_time": 9.559287548065186
    },
    {
      "epoch": 3.1843185424804687e-06,
      "model_forward_time": 1.7145330905914307,
      "step": 3339
    },
    {
      "epoch": 3.1843185424804687e-06,
      "step": 3339,
      "training_step_time": 7.769688606262207
    },
    {
      "epoch": 3.185272216796875e-06,
      "grad_norm": 0.3762357234954834,
      "learning_rate": 6.678e-06,
      "loss": 0.0813,
      "step": 3340
    },
    {
      "epoch": 3.185272216796875e-06,
      "model_forward_time": 1.705815315246582,
      "step": 3340
    },
    {
      "dynamics_loss_avg": 0.06736543253064156,
      "epoch": 3.185272216796875e-06,
      "step": 3340
    },
    {
      "action_loss_avg": 0.010330428183078767,
      "epoch": 3.185272216796875e-06,
      "step": 3340
    },
    {
      "epoch": 3.185272216796875e-06,
      "step": 3340,
      "training_step_time": 8.323737621307373
    },
    {
      "epoch": 3.1862258911132814e-06,
      "model_forward_time": 1.7043006420135498,
      "step": 3341
    },
    {
      "epoch": 3.1862258911132814e-06,
      "step": 3341,
      "training_step_time": 7.640918731689453
    },
    {
      "epoch": 3.1871795654296876e-06,
      "model_forward_time": 1.7126028537750244,
      "step": 3342
    },
    {
      "epoch": 3.1871795654296876e-06,
      "step": 3342,
      "training_step_time": 7.175761699676514
    },
    {
      "epoch": 3.1881332397460937e-06,
      "model_forward_time": 1.7162787914276123,
      "step": 3343
    },
    {
      "epoch": 3.1881332397460937e-06,
      "step": 3343,
      "training_step_time": 7.515726089477539
    },
    {
      "epoch": 3.1890869140625e-06,
      "model_forward_time": 1.7151172161102295,
      "step": 3344
    },
    {
      "epoch": 3.1890869140625e-06,
      "step": 3344,
      "training_step_time": 7.5593626499176025
    },
    {
      "epoch": 3.1900405883789064e-06,
      "model_forward_time": 1.7083570957183838,
      "step": 3345
    },
    {
      "epoch": 3.1900405883789064e-06,
      "step": 3345,
      "training_step_time": 7.501418352127075
    },
    {
      "epoch": 3.1909942626953126e-06,
      "model_forward_time": 1.7679147720336914,
      "step": 3346
    },
    {
      "epoch": 3.1909942626953126e-06,
      "step": 3346,
      "training_step_time": 8.53443694114685
    },
    {
      "epoch": 3.1919479370117187e-06,
      "model_forward_time": 2.2808101177215576,
      "step": 3347
    },
    {
      "epoch": 3.1919479370117187e-06,
      "step": 3347,
      "training_step_time": 7.660415410995483
    },
    {
      "epoch": 3.192901611328125e-06,
      "model_forward_time": 1.704317331314087,
      "step": 3348
    },
    {
      "epoch": 3.192901611328125e-06,
      "step": 3348,
      "training_step_time": 9.232184886932373
    },
    {
      "epoch": 3.1938552856445314e-06,
      "model_forward_time": 1.7125961780548096,
      "step": 3349
    },
    {
      "epoch": 3.1938552856445314e-06,
      "step": 3349,
      "training_step_time": 8.78554630279541
    },
    {
      "epoch": 3.1948089599609376e-06,
      "grad_norm": 0.4272921681404114,
      "learning_rate": 6.698e-06,
      "loss": 0.0831,
      "step": 3350
    },
    {
      "epoch": 3.1948089599609376e-06,
      "model_forward_time": 1.7034051418304443,
      "step": 3350
    },
    {
      "dynamics_loss_avg": 0.06657842509448528,
      "epoch": 3.1948089599609376e-06,
      "step": 3350
    },
    {
      "action_loss_avg": 0.008717884588986636,
      "epoch": 3.1948089599609376e-06,
      "step": 3350
    },
    {
      "epoch": 3.1948089599609376e-06,
      "step": 3350,
      "training_step_time": 7.670840740203857
    },
    {
      "epoch": 3.1957626342773437e-06,
      "model_forward_time": 1.710188627243042,
      "step": 3351
    },
    {
      "epoch": 3.1957626342773437e-06,
      "step": 3351,
      "training_step_time": 7.687103033065796
    },
    {
      "epoch": 3.19671630859375e-06,
      "model_forward_time": 1.7046387195587158,
      "step": 3352
    },
    {
      "epoch": 3.19671630859375e-06,
      "step": 3352,
      "training_step_time": 7.654187917709351
    },
    {
      "epoch": 3.1976699829101564e-06,
      "model_forward_time": 1.7015774250030518,
      "step": 3353
    },
    {
      "epoch": 3.1976699829101564e-06,
      "step": 3353,
      "training_step_time": 7.4880523681640625
    },
    {
      "epoch": 3.1986236572265626e-06,
      "model_forward_time": 1.7055232524871826,
      "step": 3354
    },
    {
      "epoch": 3.1986236572265626e-06,
      "step": 3354,
      "training_step_time": 7.689178705215454
    },
    {
      "epoch": 3.1995773315429687e-06,
      "model_forward_time": 1.7510614395141602,
      "step": 3355
    },
    {
      "epoch": 3.1995773315429687e-06,
      "step": 3355,
      "training_step_time": 8.141839981079102
    },
    {
      "epoch": 3.200531005859375e-06,
      "model_forward_time": 2.2406094074249268,
      "step": 3356
    },
    {
      "epoch": 3.200531005859375e-06,
      "step": 3356,
      "training_step_time": 8.292034387588501
    },
    {
      "epoch": 3.2014846801757814e-06,
      "model_forward_time": 1.7028346061706543,
      "step": 3357
    },
    {
      "epoch": 3.2014846801757814e-06,
      "step": 3357,
      "training_step_time": 10.626453161239624
    },
    {
      "epoch": 3.2024383544921876e-06,
      "model_forward_time": 1.7045137882232666,
      "step": 3358
    },
    {
      "epoch": 3.2024383544921876e-06,
      "step": 3358,
      "training_step_time": 7.622988939285278
    },
    {
      "epoch": 3.2033920288085937e-06,
      "model_forward_time": 1.7108821868896484,
      "step": 3359
    },
    {
      "epoch": 3.2033920288085937e-06,
      "step": 3359,
      "training_step_time": 7.6660544872283936
    },
    {
      "epoch": 3.204345703125e-06,
      "grad_norm": 0.3921871781349182,
      "learning_rate": 6.718e-06,
      "loss": 0.0779,
      "step": 3360
    },
    {
      "epoch": 3.204345703125e-06,
      "model_forward_time": 1.7038030624389648,
      "step": 3360
    },
    {
      "dynamics_loss_avg": 0.057579382136464116,
      "epoch": 3.204345703125e-06,
      "step": 3360
    },
    {
      "action_loss_avg": 0.012211105786263943,
      "epoch": 3.204345703125e-06,
      "step": 3360
    },
    {
      "epoch": 3.204345703125e-06,
      "step": 3360,
      "training_step_time": 7.578778028488159
    },
    {
      "epoch": 3.2052993774414064e-06,
      "model_forward_time": 1.7090766429901123,
      "step": 3361
    },
    {
      "epoch": 3.2052993774414064e-06,
      "step": 3361,
      "training_step_time": 7.581478834152222
    },
    {
      "epoch": 3.2062530517578126e-06,
      "model_forward_time": 1.7033295631408691,
      "step": 3362
    },
    {
      "epoch": 3.2062530517578126e-06,
      "step": 3362,
      "training_step_time": 7.458370208740234
    },
    {
      "epoch": 3.2072067260742187e-06,
      "model_forward_time": 1.6960515975952148,
      "step": 3363
    },
    {
      "epoch": 3.2072067260742187e-06,
      "step": 3363,
      "training_step_time": 7.143628120422363
    },
    {
      "epoch": 3.208160400390625e-06,
      "model_forward_time": 1.7270619869232178,
      "step": 3364
    },
    {
      "epoch": 3.208160400390625e-06,
      "step": 3364,
      "training_step_time": 7.5658204555511475
    },
    {
      "epoch": 3.2091140747070315e-06,
      "model_forward_time": 1.7029311656951904,
      "step": 3365
    },
    {
      "epoch": 3.2091140747070315e-06,
      "step": 3365,
      "training_step_time": 8.637115955352783
    },
    {
      "epoch": 3.2100677490234376e-06,
      "model_forward_time": 1.698500394821167,
      "step": 3366
    },
    {
      "epoch": 3.2100677490234376e-06,
      "step": 3366,
      "training_step_time": 9.061713695526123
    },
    {
      "epoch": 3.2110214233398437e-06,
      "model_forward_time": 1.706010103225708,
      "step": 3367
    },
    {
      "epoch": 3.2110214233398437e-06,
      "step": 3367,
      "training_step_time": 8.17093539237976
    },
    {
      "epoch": 3.21197509765625e-06,
      "model_forward_time": 1.7127645015716553,
      "step": 3368
    },
    {
      "epoch": 3.21197509765625e-06,
      "step": 3368,
      "training_step_time": 7.565164804458618
    },
    {
      "epoch": 3.2129287719726565e-06,
      "model_forward_time": 1.7091164588928223,
      "step": 3369
    },
    {
      "epoch": 3.2129287719726565e-06,
      "step": 3369,
      "training_step_time": 7.648008108139038
    },
    {
      "epoch": 3.2138824462890626e-06,
      "grad_norm": 0.5905706286430359,
      "learning_rate": 6.738e-06,
      "loss": 0.0803,
      "step": 3370
    },
    {
      "epoch": 3.2138824462890626e-06,
      "model_forward_time": 1.7035932540893555,
      "step": 3370
    },
    {
      "dynamics_loss_avg": 0.0662631580606103,
      "epoch": 3.2138824462890626e-06,
      "step": 3370
    },
    {
      "action_loss_avg": 0.009838063828647136,
      "epoch": 3.2138824462890626e-06,
      "step": 3370
    },
    {
      "epoch": 3.2138824462890626e-06,
      "step": 3370,
      "training_step_time": 7.479678392410278
    },
    {
      "epoch": 3.2148361206054687e-06,
      "model_forward_time": 1.6980953216552734,
      "step": 3371
    },
    {
      "epoch": 3.2148361206054687e-06,
      "step": 3371,
      "training_step_time": 7.277953386306763
    },
    {
      "epoch": 3.215789794921875e-06,
      "model_forward_time": 1.697779893875122,
      "step": 3372
    },
    {
      "epoch": 3.215789794921875e-06,
      "step": 3372,
      "training_step_time": 8.371520757675171
    },
    {
      "epoch": 3.2167434692382815e-06,
      "model_forward_time": 1.7123148441314697,
      "step": 3373
    },
    {
      "epoch": 3.2167434692382815e-06,
      "step": 3373,
      "training_step_time": 7.578061103820801
    },
    {
      "epoch": 3.2176971435546876e-06,
      "model_forward_time": 1.7067179679870605,
      "step": 3374
    },
    {
      "epoch": 3.2176971435546876e-06,
      "step": 3374,
      "training_step_time": 8.552290439605713
    },
    {
      "epoch": 3.2186508178710938e-06,
      "model_forward_time": 1.696882724761963,
      "step": 3375
    },
    {
      "epoch": 3.2186508178710938e-06,
      "step": 3375,
      "training_step_time": 10.618813276290894
    },
    {
      "epoch": 3.2196044921875e-06,
      "model_forward_time": 1.6985411643981934,
      "step": 3376
    },
    {
      "epoch": 3.2196044921875e-06,
      "step": 3376,
      "training_step_time": 7.642845630645752
    },
    {
      "epoch": 3.220558166503906e-06,
      "model_forward_time": 1.7003474235534668,
      "step": 3377
    },
    {
      "epoch": 3.220558166503906e-06,
      "step": 3377,
      "training_step_time": 8.473265409469604
    },
    {
      "epoch": 3.2215118408203126e-06,
      "model_forward_time": 1.714874267578125,
      "step": 3378
    },
    {
      "epoch": 3.2215118408203126e-06,
      "step": 3378,
      "training_step_time": 7.5916907787323
    },
    {
      "epoch": 3.2224655151367188e-06,
      "model_forward_time": 1.7014548778533936,
      "step": 3379
    },
    {
      "epoch": 3.2224655151367188e-06,
      "step": 3379,
      "training_step_time": 7.550916910171509
    },
    {
      "epoch": 3.223419189453125e-06,
      "grad_norm": 0.680046021938324,
      "learning_rate": 6.758e-06,
      "loss": 0.086,
      "step": 3380
    },
    {
      "epoch": 3.223419189453125e-06,
      "model_forward_time": 1.7003602981567383,
      "step": 3380
    },
    {
      "dynamics_loss_avg": 0.054119284823536874,
      "epoch": 3.223419189453125e-06,
      "step": 3380
    },
    {
      "action_loss_avg": 0.008589906129054725,
      "epoch": 3.223419189453125e-06,
      "step": 3380
    },
    {
      "epoch": 3.223419189453125e-06,
      "step": 3380,
      "training_step_time": 7.004690885543823
    },
    {
      "epoch": 3.224372863769531e-06,
      "model_forward_time": 1.7137541770935059,
      "step": 3381
    },
    {
      "epoch": 3.224372863769531e-06,
      "step": 3381,
      "training_step_time": 7.536731243133545
    },
    {
      "epoch": 3.2253265380859376e-06,
      "model_forward_time": 1.7002341747283936,
      "step": 3382
    },
    {
      "epoch": 3.2253265380859376e-06,
      "step": 3382,
      "training_step_time": 7.5777740478515625
    },
    {
      "epoch": 3.2262802124023438e-06,
      "model_forward_time": 1.7592997550964355,
      "step": 3383
    },
    {
      "epoch": 3.2262802124023438e-06,
      "step": 3383,
      "training_step_time": 8.075854778289795
    },
    {
      "epoch": 3.22723388671875e-06,
      "model_forward_time": 1.7140612602233887,
      "step": 3384
    },
    {
      "epoch": 3.22723388671875e-06,
      "step": 3384,
      "training_step_time": 9.262813091278076
    },
    {
      "epoch": 3.228187561035156e-06,
      "model_forward_time": 1.710047960281372,
      "step": 3385
    },
    {
      "epoch": 3.228187561035156e-06,
      "step": 3385,
      "training_step_time": 7.604505777359009
    },
    {
      "epoch": 3.2291412353515626e-06,
      "model_forward_time": 1.711535930633545,
      "step": 3386
    },
    {
      "epoch": 3.2291412353515626e-06,
      "step": 3386,
      "training_step_time": 7.660934209823608
    },
    {
      "epoch": 3.2300949096679688e-06,
      "model_forward_time": 1.6993207931518555,
      "step": 3387
    },
    {
      "epoch": 3.2300949096679688e-06,
      "step": 3387,
      "training_step_time": 7.441322565078735
    },
    {
      "epoch": 3.231048583984375e-06,
      "model_forward_time": 1.7086172103881836,
      "step": 3388
    },
    {
      "epoch": 3.231048583984375e-06,
      "step": 3388,
      "training_step_time": 7.542400598526001
    },
    {
      "epoch": 3.232002258300781e-06,
      "model_forward_time": 1.7084400653839111,
      "step": 3389
    },
    {
      "epoch": 3.232002258300781e-06,
      "step": 3389,
      "training_step_time": 7.513100862503052
    },
    {
      "epoch": 3.2329559326171876e-06,
      "grad_norm": 0.44037753343582153,
      "learning_rate": 6.778e-06,
      "loss": 0.0899,
      "step": 3390
    },
    {
      "epoch": 3.2329559326171876e-06,
      "model_forward_time": 1.7062146663665771,
      "step": 3390
    },
    {
      "dynamics_loss_avg": 0.06309255883097649,
      "epoch": 3.2329559326171876e-06,
      "step": 3390
    },
    {
      "action_loss_avg": 0.008162356400862335,
      "epoch": 3.2329559326171876e-06,
      "step": 3390
    },
    {
      "epoch": 3.2329559326171876e-06,
      "step": 3390,
      "training_step_time": 7.556218385696411
    },
    {
      "epoch": 3.2339096069335938e-06,
      "model_forward_time": 1.7068254947662354,
      "step": 3391
    },
    {
      "epoch": 3.2339096069335938e-06,
      "step": 3391,
      "training_step_time": 7.541245937347412
    },
    {
      "epoch": 3.23486328125e-06,
      "model_forward_time": 1.7519497871398926,
      "step": 3392
    },
    {
      "epoch": 3.23486328125e-06,
      "step": 3392,
      "training_step_time": 7.677206039428711
    },
    {
      "epoch": 3.235816955566406e-06,
      "model_forward_time": 2.2405030727386475,
      "step": 3393
    },
    {
      "epoch": 3.235816955566406e-06,
      "step": 3393,
      "training_step_time": 9.666256666183472
    },
    {
      "epoch": 3.2367706298828126e-06,
      "model_forward_time": 1.7010078430175781,
      "step": 3394
    },
    {
      "epoch": 3.2367706298828126e-06,
      "step": 3394,
      "training_step_time": 8.546013355255127
    },
    {
      "epoch": 3.2377243041992188e-06,
      "model_forward_time": 1.7065813541412354,
      "step": 3395
    },
    {
      "epoch": 3.2377243041992188e-06,
      "step": 3395,
      "training_step_time": 9.26489520072937
    },
    {
      "epoch": 3.238677978515625e-06,
      "model_forward_time": 1.7103524208068848,
      "step": 3396
    },
    {
      "epoch": 3.238677978515625e-06,
      "step": 3396,
      "training_step_time": 7.530005693435669
    },
    {
      "epoch": 3.239631652832031e-06,
      "model_forward_time": 1.7061662673950195,
      "step": 3397
    },
    {
      "epoch": 3.239631652832031e-06,
      "step": 3397,
      "training_step_time": 7.6003196239471436
    },
    {
      "epoch": 3.2405853271484376e-06,
      "model_forward_time": 1.7061192989349365,
      "step": 3398
    },
    {
      "epoch": 3.2405853271484376e-06,
      "step": 3398,
      "training_step_time": 7.572829246520996
    },
    {
      "epoch": 3.2415390014648438e-06,
      "model_forward_time": 1.7122764587402344,
      "step": 3399
    },
    {
      "epoch": 3.2415390014648438e-06,
      "step": 3399,
      "training_step_time": 7.53685736656189
    },
    {
      "epoch": 3.24249267578125e-06,
      "grad_norm": 0.5830209851264954,
      "learning_rate": 6.798e-06,
      "loss": 0.0895,
      "step": 3400
    },
    {
      "epoch": 3.24249267578125e-06,
      "model_forward_time": 1.7008097171783447,
      "step": 3400
    },
    {
      "dynamics_loss_avg": 0.08088923580944538,
      "epoch": 3.24249267578125e-06,
      "step": 3400
    },
    {
      "action_loss_avg": 0.011545699462294579,
      "epoch": 3.24249267578125e-06,
      "step": 3400
    },
    {
      "epoch": 3.24249267578125e-06,
      "step": 3400,
      "training_step_time": 7.4079413414001465
    },
    {
      "epoch": 3.243446350097656e-06,
      "model_forward_time": 1.750797986984253,
      "step": 3401
    },
    {
      "epoch": 3.243446350097656e-06,
      "step": 3401,
      "training_step_time": 7.642116546630859
    },
    {
      "epoch": 3.2444000244140626e-06,
      "model_forward_time": 2.2414629459381104,
      "step": 3402
    },
    {
      "epoch": 3.2444000244140626e-06,
      "step": 3402,
      "training_step_time": 7.632700204849243
    },
    {
      "epoch": 3.2453536987304688e-06,
      "model_forward_time": 1.6997172832489014,
      "step": 3403
    },
    {
      "epoch": 3.2453536987304688e-06,
      "step": 3403,
      "training_step_time": 7.653895378112793
    },
    {
      "epoch": 3.246307373046875e-06,
      "model_forward_time": 1.7190980911254883,
      "step": 3404
    },
    {
      "epoch": 3.246307373046875e-06,
      "step": 3404,
      "training_step_time": 8.467397451400757
    },
    {
      "epoch": 3.247261047363281e-06,
      "model_forward_time": 1.6938929557800293,
      "step": 3405
    },
    {
      "epoch": 3.247261047363281e-06,
      "step": 3405,
      "training_step_time": 7.657926321029663
    },
    {
      "epoch": 3.2482147216796876e-06,
      "model_forward_time": 1.7147154808044434,
      "step": 3406
    },
    {
      "epoch": 3.2482147216796876e-06,
      "step": 3406,
      "training_step_time": 7.576287269592285
    },
    {
      "epoch": 3.2491683959960938e-06,
      "model_forward_time": 1.7030301094055176,
      "step": 3407
    },
    {
      "epoch": 3.2491683959960938e-06,
      "step": 3407,
      "training_step_time": 7.531614542007446
    },
    {
      "epoch": 3.2501220703125e-06,
      "model_forward_time": 1.7008411884307861,
      "step": 3408
    },
    {
      "epoch": 3.2501220703125e-06,
      "step": 3408,
      "training_step_time": 7.327740907669067
    },
    {
      "epoch": 3.251075744628906e-06,
      "model_forward_time": 1.7083055973052979,
      "step": 3409
    },
    {
      "epoch": 3.251075744628906e-06,
      "step": 3409,
      "training_step_time": 7.417615175247192
    },
    {
      "epoch": 3.2520294189453126e-06,
      "grad_norm": 0.6393888592720032,
      "learning_rate": 6.818e-06,
      "loss": 0.0844,
      "step": 3410
    },
    {
      "epoch": 3.2520294189453126e-06,
      "model_forward_time": 1.7295379638671875,
      "step": 3410
    },
    {
      "dynamics_loss_avg": 0.07382374927401543,
      "epoch": 3.2520294189453126e-06,
      "step": 3410
    },
    {
      "action_loss_avg": 0.013302926113829017,
      "epoch": 3.2520294189453126e-06,
      "step": 3410
    },
    {
      "epoch": 3.2520294189453126e-06,
      "step": 3410,
      "training_step_time": 7.646054267883301
    },
    {
      "epoch": 3.2529830932617188e-06,
      "model_forward_time": 2.2314350605010986,
      "step": 3411
    },
    {
      "epoch": 3.2529830932617188e-06,
      "step": 3411,
      "training_step_time": 8.8176589012146
    },
    {
      "epoch": 3.253936767578125e-06,
      "model_forward_time": 1.705350637435913,
      "step": 3412
    },
    {
      "epoch": 3.253936767578125e-06,
      "step": 3412,
      "training_step_time": 9.643203496932983
    },
    {
      "epoch": 3.254890441894531e-06,
      "model_forward_time": 1.703657627105713,
      "step": 3413
    },
    {
      "epoch": 3.254890441894531e-06,
      "step": 3413,
      "training_step_time": 7.62513279914856
    },
    {
      "epoch": 3.2558441162109376e-06,
      "model_forward_time": 1.7047486305236816,
      "step": 3414
    },
    {
      "epoch": 3.2558441162109376e-06,
      "step": 3414,
      "training_step_time": 9.29344367980957
    },
    {
      "epoch": 3.2567977905273438e-06,
      "model_forward_time": 1.7061891555786133,
      "step": 3415
    },
    {
      "epoch": 3.2567977905273438e-06,
      "step": 3415,
      "training_step_time": 7.6245787143707275
    },
    {
      "epoch": 3.25775146484375e-06,
      "model_forward_time": 1.706305742263794,
      "step": 3416
    },
    {
      "epoch": 3.25775146484375e-06,
      "step": 3416,
      "training_step_time": 8.571867227554321
    },
    {
      "epoch": 3.258705139160156e-06,
      "model_forward_time": 1.7018587589263916,
      "step": 3417
    },
    {
      "epoch": 3.258705139160156e-06,
      "step": 3417,
      "training_step_time": 8.285967350006104
    },
    {
      "epoch": 3.2596588134765626e-06,
      "model_forward_time": 1.7058134078979492,
      "step": 3418
    },
    {
      "epoch": 3.2596588134765626e-06,
      "step": 3418,
      "training_step_time": 7.59885311126709
    },
    {
      "epoch": 3.2606124877929688e-06,
      "model_forward_time": 1.720005989074707,
      "step": 3419
    },
    {
      "epoch": 3.2606124877929688e-06,
      "step": 3419,
      "training_step_time": 7.649187088012695
    },
    {
      "epoch": 3.261566162109375e-06,
      "grad_norm": 0.4759685695171356,
      "learning_rate": 6.8380000000000004e-06,
      "loss": 0.0853,
      "step": 3420
    },
    {
      "epoch": 3.261566162109375e-06,
      "model_forward_time": 2.236213207244873,
      "step": 3420
    },
    {
      "dynamics_loss_avg": 0.05834326334297657,
      "epoch": 3.261566162109375e-06,
      "step": 3420
    },
    {
      "action_loss_avg": 0.009048012387938797,
      "epoch": 3.261566162109375e-06,
      "step": 3420
    },
    {
      "epoch": 3.261566162109375e-06,
      "step": 3420,
      "training_step_time": 7.628567934036255
    },
    {
      "epoch": 3.262519836425781e-06,
      "model_forward_time": 1.6884725093841553,
      "step": 3421
    },
    {
      "epoch": 3.262519836425781e-06,
      "step": 3421,
      "training_step_time": 8.589938402175903
    },
    {
      "epoch": 3.2634735107421876e-06,
      "model_forward_time": 1.7073826789855957,
      "step": 3422
    },
    {
      "epoch": 3.2634735107421876e-06,
      "step": 3422,
      "training_step_time": 8.53053593635559
    },
    {
      "epoch": 3.264427185058594e-06,
      "model_forward_time": 1.7061936855316162,
      "step": 3423
    },
    {
      "epoch": 3.264427185058594e-06,
      "step": 3423,
      "training_step_time": 8.646794080734253
    },
    {
      "epoch": 3.265380859375e-06,
      "model_forward_time": 1.709230899810791,
      "step": 3424
    },
    {
      "epoch": 3.265380859375e-06,
      "step": 3424,
      "training_step_time": 7.584131240844727
    },
    {
      "epoch": 3.266334533691406e-06,
      "model_forward_time": 1.7042741775512695,
      "step": 3425
    },
    {
      "epoch": 3.266334533691406e-06,
      "step": 3425,
      "training_step_time": 7.545289993286133
    },
    {
      "epoch": 3.2672882080078126e-06,
      "model_forward_time": 1.7048156261444092,
      "step": 3426
    },
    {
      "epoch": 3.2672882080078126e-06,
      "step": 3426,
      "training_step_time": 7.520801782608032
    },
    {
      "epoch": 3.268241882324219e-06,
      "model_forward_time": 1.7058448791503906,
      "step": 3427
    },
    {
      "epoch": 3.268241882324219e-06,
      "step": 3427,
      "training_step_time": 7.522846221923828
    },
    {
      "epoch": 3.269195556640625e-06,
      "model_forward_time": 1.699958086013794,
      "step": 3428
    },
    {
      "epoch": 3.269195556640625e-06,
      "step": 3428,
      "training_step_time": 8.930116891860962
    },
    {
      "epoch": 3.270149230957031e-06,
      "model_forward_time": 2.2500152587890625,
      "step": 3429
    },
    {
      "epoch": 3.270149230957031e-06,
      "step": 3429,
      "training_step_time": 9.236580610275269
    },
    {
      "epoch": 3.2711029052734377e-06,
      "grad_norm": 0.6962510347366333,
      "learning_rate": 6.858e-06,
      "loss": 0.0839,
      "step": 3430
    },
    {
      "epoch": 3.2711029052734377e-06,
      "model_forward_time": 1.7204859256744385,
      "step": 3430
    },
    {
      "dynamics_loss_avg": 0.07056623101234435,
      "epoch": 3.2711029052734377e-06,
      "step": 3430
    },
    {
      "action_loss_avg": 0.010420613852329553,
      "epoch": 3.2711029052734377e-06,
      "step": 3430
    },
    {
      "epoch": 3.2711029052734377e-06,
      "step": 3430,
      "training_step_time": 7.751891613006592
    },
    {
      "epoch": 3.272056579589844e-06,
      "model_forward_time": 1.7185685634613037,
      "step": 3431
    },
    {
      "epoch": 3.272056579589844e-06,
      "step": 3431,
      "training_step_time": 8.554720163345337
    },
    {
      "epoch": 3.27301025390625e-06,
      "model_forward_time": 1.7298712730407715,
      "step": 3432
    },
    {
      "epoch": 3.27301025390625e-06,
      "step": 3432,
      "training_step_time": 7.702244758605957
    },
    {
      "epoch": 3.273963928222656e-06,
      "model_forward_time": 1.7227022647857666,
      "step": 3433
    },
    {
      "epoch": 3.273963928222656e-06,
      "step": 3433,
      "training_step_time": 7.6716835498809814
    },
    {
      "epoch": 3.2749176025390627e-06,
      "model_forward_time": 1.7173707485198975,
      "step": 3434
    },
    {
      "epoch": 3.2749176025390627e-06,
      "step": 3434,
      "training_step_time": 7.692919492721558
    },
    {
      "epoch": 3.275871276855469e-06,
      "model_forward_time": 1.725006103515625,
      "step": 3435
    },
    {
      "epoch": 3.275871276855469e-06,
      "step": 3435,
      "training_step_time": 7.271413564682007
    },
    {
      "epoch": 3.276824951171875e-06,
      "model_forward_time": 1.727914810180664,
      "step": 3436
    },
    {
      "epoch": 3.276824951171875e-06,
      "step": 3436,
      "training_step_time": 7.5814008712768555
    },
    {
      "epoch": 3.277778625488281e-06,
      "model_forward_time": 1.7225239276885986,
      "step": 3437
    },
    {
      "epoch": 3.277778625488281e-06,
      "step": 3437,
      "training_step_time": 7.576311349868774
    },
    {
      "epoch": 3.2787322998046877e-06,
      "model_forward_time": 2.2798240184783936,
      "step": 3438
    },
    {
      "epoch": 3.2787322998046877e-06,
      "step": 3438,
      "training_step_time": 8.632949590682983
    },
    {
      "epoch": 3.279685974121094e-06,
      "model_forward_time": 1.7244658470153809,
      "step": 3439
    },
    {
      "epoch": 3.279685974121094e-06,
      "step": 3439,
      "training_step_time": 8.57115364074707
    },
    {
      "epoch": 3.2806396484375e-06,
      "grad_norm": 0.40925952792167664,
      "learning_rate": 6.878e-06,
      "loss": 0.0827,
      "step": 3440
    },
    {
      "epoch": 3.2806396484375e-06,
      "model_forward_time": 1.3474934101104736,
      "step": 3440
    },
    {
      "dynamics_loss_avg": 0.06375932712107897,
      "epoch": 3.2806396484375e-06,
      "step": 3440
    },
    {
      "action_loss_avg": 0.011772919073700904,
      "epoch": 3.2806396484375e-06,
      "step": 3440
    },
    {
      "epoch": 3.2806396484375e-06,
      "step": 3440,
      "training_step_time": 8.399402141571045
    },
    {
      "epoch": 3.281593322753906e-06,
      "model_forward_time": 1.7238860130310059,
      "step": 3441
    },
    {
      "epoch": 3.281593322753906e-06,
      "step": 3441,
      "training_step_time": 8.651996374130249
    },
    {
      "epoch": 3.2825469970703127e-06,
      "model_forward_time": 1.720449686050415,
      "step": 3442
    },
    {
      "epoch": 3.2825469970703127e-06,
      "step": 3442,
      "training_step_time": 7.622012376785278
    },
    {
      "epoch": 3.283500671386719e-06,
      "model_forward_time": 1.7203080654144287,
      "step": 3443
    },
    {
      "epoch": 3.283500671386719e-06,
      "step": 3443,
      "training_step_time": 7.603266954421997
    },
    {
      "epoch": 3.284454345703125e-06,
      "model_forward_time": 1.7219955921173096,
      "step": 3444
    },
    {
      "epoch": 3.284454345703125e-06,
      "step": 3444,
      "training_step_time": 7.727315902709961
    },
    {
      "epoch": 3.285408020019531e-06,
      "model_forward_time": 1.7222747802734375,
      "step": 3445
    },
    {
      "epoch": 3.285408020019531e-06,
      "step": 3445,
      "training_step_time": 7.607872009277344
    },
    {
      "epoch": 3.2863616943359377e-06,
      "model_forward_time": 1.7303643226623535,
      "step": 3446
    },
    {
      "epoch": 3.2863616943359377e-06,
      "step": 3446,
      "training_step_time": 7.562861680984497
    },
    {
      "epoch": 3.287315368652344e-06,
      "model_forward_time": 2.2922589778900146,
      "step": 3447
    },
    {
      "epoch": 3.287315368652344e-06,
      "step": 3447,
      "training_step_time": 7.699752569198608
    },
    {
      "epoch": 3.28826904296875e-06,
      "model_forward_time": 1.7243354320526123,
      "step": 3448
    },
    {
      "epoch": 3.28826904296875e-06,
      "step": 3448,
      "training_step_time": 8.54296064376831
    },
    {
      "epoch": 3.289222717285156e-06,
      "model_forward_time": 1.7388947010040283,
      "step": 3449
    },
    {
      "epoch": 3.289222717285156e-06,
      "step": 3449,
      "training_step_time": 7.791820764541626
    },
    {
      "epoch": 3.2901763916015627e-06,
      "grad_norm": 0.5890548825263977,
      "learning_rate": 6.898e-06,
      "loss": 0.0877,
      "step": 3450
    },
    {
      "epoch": 3.2901763916015627e-06,
      "model_forward_time": 1.7149810791015625,
      "step": 3450
    },
    {
      "dynamics_loss_avg": 0.06671169493347406,
      "epoch": 3.2901763916015627e-06,
      "step": 3450
    },
    {
      "action_loss_avg": 0.01273432767484337,
      "epoch": 3.2901763916015627e-06,
      "step": 3450
    },
    {
      "epoch": 3.2901763916015627e-06,
      "step": 3450,
      "training_step_time": 8.635318994522095
    },
    {
      "epoch": 3.291130065917969e-06,
      "model_forward_time": 1.7248625755310059,
      "step": 3451
    },
    {
      "epoch": 3.291130065917969e-06,
      "step": 3451,
      "training_step_time": 7.566535711288452
    },
    {
      "epoch": 3.292083740234375e-06,
      "model_forward_time": 1.7210948467254639,
      "step": 3452
    },
    {
      "epoch": 3.292083740234375e-06,
      "step": 3452,
      "training_step_time": 7.6719279289245605
    },
    {
      "epoch": 3.293037414550781e-06,
      "model_forward_time": 1.7242910861968994,
      "step": 3453
    },
    {
      "epoch": 3.293037414550781e-06,
      "step": 3453,
      "training_step_time": 7.585700511932373
    },
    {
      "epoch": 3.2939910888671877e-06,
      "model_forward_time": 1.7215781211853027,
      "step": 3454
    },
    {
      "epoch": 3.2939910888671877e-06,
      "step": 3454,
      "training_step_time": 7.655560255050659
    },
    {
      "epoch": 3.294944763183594e-06,
      "model_forward_time": 1.7154951095581055,
      "step": 3455
    },
    {
      "epoch": 3.294944763183594e-06,
      "step": 3455,
      "training_step_time": 7.526895761489868
    },
    {
      "epoch": 3.2958984375e-06,
      "model_forward_time": 2.287834405899048,
      "step": 3456
    },
    {
      "epoch": 3.2958984375e-06,
      "step": 3456,
      "training_step_time": 8.704936027526855
    },
    {
      "epoch": 3.296852111816406e-06,
      "model_forward_time": 1.7278475761413574,
      "step": 3457
    },
    {
      "epoch": 3.296852111816406e-06,
      "step": 3457,
      "training_step_time": 9.454967737197876
    },
    {
      "epoch": 3.2978057861328127e-06,
      "model_forward_time": 1.7202930450439453,
      "step": 3458
    },
    {
      "epoch": 3.2978057861328127e-06,
      "step": 3458,
      "training_step_time": 8.479034900665283
    },
    {
      "epoch": 3.298759460449219e-06,
      "model_forward_time": 1.7262146472930908,
      "step": 3459
    },
    {
      "epoch": 3.298759460449219e-06,
      "step": 3459,
      "training_step_time": 8.12113332748413
    },
    {
      "epoch": 3.299713134765625e-06,
      "grad_norm": 0.33782759308815,
      "learning_rate": 6.9180000000000005e-06,
      "loss": 0.0872,
      "step": 3460
    },
    {
      "epoch": 3.299713134765625e-06,
      "model_forward_time": 1.717581033706665,
      "step": 3460
    },
    {
      "dynamics_loss_avg": 0.058375397324562074,
      "epoch": 3.299713134765625e-06,
      "step": 3460
    },
    {
      "action_loss_avg": 0.009155166917480528,
      "epoch": 3.299713134765625e-06,
      "step": 3460
    },
    {
      "epoch": 3.299713134765625e-06,
      "step": 3460,
      "training_step_time": 8.673648118972778
    },
    {
      "epoch": 3.300666809082031e-06,
      "model_forward_time": 1.7256219387054443,
      "step": 3461
    },
    {
      "epoch": 3.300666809082031e-06,
      "step": 3461,
      "training_step_time": 7.616020441055298
    },
    {
      "epoch": 3.3016204833984377e-06,
      "model_forward_time": 1.7023820877075195,
      "step": 3462
    },
    {
      "epoch": 3.3016204833984377e-06,
      "step": 3462,
      "training_step_time": 7.635831117630005
    },
    {
      "epoch": 3.302574157714844e-06,
      "model_forward_time": 1.7128355503082275,
      "step": 3463
    },
    {
      "epoch": 3.302574157714844e-06,
      "step": 3463,
      "training_step_time": 8.500293493270874
    },
    {
      "epoch": 3.30352783203125e-06,
      "model_forward_time": 1.7246253490447998,
      "step": 3464
    },
    {
      "epoch": 3.30352783203125e-06,
      "step": 3464,
      "training_step_time": 8.44064736366272
    },
    {
      "epoch": 3.304481506347656e-06,
      "model_forward_time": 1.3583452701568604,
      "step": 3465
    },
    {
      "epoch": 3.304481506347656e-06,
      "step": 3465,
      "training_step_time": 9.628390312194824
    },
    {
      "epoch": 3.3054351806640627e-06,
      "model_forward_time": 2.3069276809692383,
      "step": 3466
    },
    {
      "epoch": 3.3054351806640627e-06,
      "step": 3466,
      "training_step_time": 7.604621887207031
    },
    {
      "epoch": 3.306388854980469e-06,
      "model_forward_time": 1.7243564128875732,
      "step": 3467
    },
    {
      "epoch": 3.306388854980469e-06,
      "step": 3467,
      "training_step_time": 9.690715789794922
    },
    {
      "epoch": 3.307342529296875e-06,
      "model_forward_time": 1.727701187133789,
      "step": 3468
    },
    {
      "epoch": 3.307342529296875e-06,
      "step": 3468,
      "training_step_time": 7.626884698867798
    },
    {
      "epoch": 3.308296203613281e-06,
      "model_forward_time": 1.7217960357666016,
      "step": 3469
    },
    {
      "epoch": 3.308296203613281e-06,
      "step": 3469,
      "training_step_time": 8.55658483505249
    },
    {
      "epoch": 3.3092498779296877e-06,
      "grad_norm": 0.5496275424957275,
      "learning_rate": 6.938e-06,
      "loss": 0.0818,
      "step": 3470
    },
    {
      "epoch": 3.3092498779296877e-06,
      "model_forward_time": 1.720400333404541,
      "step": 3470
    },
    {
      "dynamics_loss_avg": 0.05957014784216881,
      "epoch": 3.3092498779296877e-06,
      "step": 3470
    },
    {
      "action_loss_avg": 0.012372076092287899,
      "epoch": 3.3092498779296877e-06,
      "step": 3470
    },
    {
      "epoch": 3.3092498779296877e-06,
      "step": 3470,
      "training_step_time": 7.527903318405151
    },
    {
      "epoch": 3.310203552246094e-06,
      "model_forward_time": 1.7240042686462402,
      "step": 3471
    },
    {
      "epoch": 3.310203552246094e-06,
      "step": 3471,
      "training_step_time": 7.650141477584839
    },
    {
      "epoch": 3.3111572265625e-06,
      "model_forward_time": 1.7261383533477783,
      "step": 3472
    },
    {
      "epoch": 3.3111572265625e-06,
      "step": 3472,
      "training_step_time": 7.619893789291382
    },
    {
      "epoch": 3.312110900878906e-06,
      "model_forward_time": 1.7211320400238037,
      "step": 3473
    },
    {
      "epoch": 3.312110900878906e-06,
      "step": 3473,
      "training_step_time": 7.49775767326355
    },
    {
      "epoch": 3.3130645751953127e-06,
      "model_forward_time": 1.7460787296295166,
      "step": 3474
    },
    {
      "epoch": 3.3130645751953127e-06,
      "step": 3474,
      "training_step_time": 7.839449644088745
    },
    {
      "epoch": 3.314018249511719e-06,
      "model_forward_time": 2.289308547973633,
      "step": 3475
    },
    {
      "epoch": 3.314018249511719e-06,
      "step": 3475,
      "training_step_time": 8.611358165740967
    },
    {
      "epoch": 3.314971923828125e-06,
      "model_forward_time": 1.717181921005249,
      "step": 3476
    },
    {
      "epoch": 3.314971923828125e-06,
      "step": 3476,
      "training_step_time": 7.653323411941528
    },
    {
      "epoch": 3.315925598144531e-06,
      "model_forward_time": 1.345822811126709,
      "step": 3477
    },
    {
      "epoch": 3.315925598144531e-06,
      "step": 3477,
      "training_step_time": 10.281328916549683
    },
    {
      "epoch": 3.3168792724609377e-06,
      "model_forward_time": 1.7162675857543945,
      "step": 3478
    },
    {
      "epoch": 3.3168792724609377e-06,
      "step": 3478,
      "training_step_time": 7.65951681137085
    },
    {
      "epoch": 3.317832946777344e-06,
      "model_forward_time": 1.7226860523223877,
      "step": 3479
    },
    {
      "epoch": 3.317832946777344e-06,
      "step": 3479,
      "training_step_time": 7.5667994022369385
    },
    {
      "epoch": 3.31878662109375e-06,
      "grad_norm": 0.41667601466178894,
      "learning_rate": 6.958e-06,
      "loss": 0.0815,
      "step": 3480
    },
    {
      "epoch": 3.31878662109375e-06,
      "model_forward_time": 1.7211337089538574,
      "step": 3480
    },
    {
      "dynamics_loss_avg": 0.05263077113777399,
      "epoch": 3.31878662109375e-06,
      "step": 3480
    },
    {
      "action_loss_avg": 0.007541183778084814,
      "epoch": 3.31878662109375e-06,
      "step": 3480
    },
    {
      "epoch": 3.31878662109375e-06,
      "step": 3480,
      "training_step_time": 7.661818981170654
    },
    {
      "epoch": 3.319740295410156e-06,
      "model_forward_time": 1.7191417217254639,
      "step": 3481
    },
    {
      "epoch": 3.319740295410156e-06,
      "step": 3481,
      "training_step_time": 7.625290632247925
    },
    {
      "epoch": 3.3206939697265627e-06,
      "model_forward_time": 1.718013048171997,
      "step": 3482
    },
    {
      "epoch": 3.3206939697265627e-06,
      "step": 3482,
      "training_step_time": 7.543860673904419
    },
    {
      "epoch": 3.321647644042969e-06,
      "model_forward_time": 1.717979907989502,
      "step": 3483
    },
    {
      "epoch": 3.321647644042969e-06,
      "step": 3483,
      "training_step_time": 8.61556100845337
    },
    {
      "epoch": 3.322601318359375e-06,
      "model_forward_time": 2.302062511444092,
      "step": 3484
    },
    {
      "epoch": 3.322601318359375e-06,
      "step": 3484,
      "training_step_time": 7.628232002258301
    },
    {
      "epoch": 3.323554992675781e-06,
      "model_forward_time": 1.722895622253418,
      "step": 3485
    },
    {
      "epoch": 3.323554992675781e-06,
      "step": 3485,
      "training_step_time": 9.72419786453247
    },
    {
      "epoch": 3.3245086669921877e-06,
      "model_forward_time": 1.7180652618408203,
      "step": 3486
    },
    {
      "epoch": 3.3245086669921877e-06,
      "step": 3486,
      "training_step_time": 7.7873029708862305
    },
    {
      "epoch": 3.325462341308594e-06,
      "model_forward_time": 1.7056312561035156,
      "step": 3487
    },
    {
      "epoch": 3.325462341308594e-06,
      "step": 3487,
      "training_step_time": 7.626777172088623
    },
    {
      "epoch": 3.326416015625e-06,
      "model_forward_time": 1.7258059978485107,
      "step": 3488
    },
    {
      "epoch": 3.326416015625e-06,
      "step": 3488,
      "training_step_time": 7.5413854122161865
    },
    {
      "epoch": 3.327369689941406e-06,
      "model_forward_time": 1.7153534889221191,
      "step": 3489
    },
    {
      "epoch": 3.327369689941406e-06,
      "step": 3489,
      "training_step_time": 7.622679948806763
    },
    {
      "epoch": 3.3283233642578127e-06,
      "grad_norm": 0.3653847575187683,
      "learning_rate": 6.978e-06,
      "loss": 0.0789,
      "step": 3490
    },
    {
      "epoch": 3.3283233642578127e-06,
      "model_forward_time": 1.6966652870178223,
      "step": 3490
    },
    {
      "dynamics_loss_avg": 0.07020456567406655,
      "epoch": 3.3283233642578127e-06,
      "step": 3490
    },
    {
      "action_loss_avg": 0.01357202574145049,
      "epoch": 3.3283233642578127e-06,
      "step": 3490
    },
    {
      "epoch": 3.3283233642578127e-06,
      "step": 3490,
      "training_step_time": 7.565037488937378
    },
    {
      "epoch": 3.329277038574219e-06,
      "model_forward_time": 1.7056519985198975,
      "step": 3491
    },
    {
      "epoch": 3.329277038574219e-06,
      "step": 3491,
      "training_step_time": 7.4173688888549805
    },
    {
      "epoch": 3.330230712890625e-06,
      "model_forward_time": 1.714763879776001,
      "step": 3492
    },
    {
      "epoch": 3.330230712890625e-06,
      "step": 3492,
      "training_step_time": 8.504636287689209
    },
    {
      "epoch": 3.331184387207031e-06,
      "model_forward_time": 1.7439086437225342,
      "step": 3493
    },
    {
      "epoch": 3.331184387207031e-06,
      "step": 3493,
      "training_step_time": 9.25743293762207
    },
    {
      "epoch": 3.3321380615234377e-06,
      "model_forward_time": 1.6988532543182373,
      "step": 3494
    },
    {
      "epoch": 3.3321380615234377e-06,
      "step": 3494,
      "training_step_time": 8.665708780288696
    },
    {
      "epoch": 3.333091735839844e-06,
      "model_forward_time": 1.7030155658721924,
      "step": 3495
    },
    {
      "epoch": 3.333091735839844e-06,
      "step": 3495,
      "training_step_time": 7.723829030990601
    },
    {
      "epoch": 3.33404541015625e-06,
      "model_forward_time": 1.7104897499084473,
      "step": 3496
    },
    {
      "epoch": 3.33404541015625e-06,
      "step": 3496,
      "training_step_time": 7.642739534378052
    },
    {
      "epoch": 3.334999084472656e-06,
      "model_forward_time": 1.7115099430084229,
      "step": 3497
    },
    {
      "epoch": 3.334999084472656e-06,
      "step": 3497,
      "training_step_time": 7.436060905456543
    },
    {
      "epoch": 3.3359527587890627e-06,
      "model_forward_time": 1.7040202617645264,
      "step": 3498
    },
    {
      "epoch": 3.3359527587890627e-06,
      "step": 3498,
      "training_step_time": 7.516341924667358
    },
    {
      "epoch": 3.336906433105469e-06,
      "model_forward_time": 1.7110207080841064,
      "step": 3499
    },
    {
      "epoch": 3.336906433105469e-06,
      "step": 3499,
      "training_step_time": 8.095073938369751
    },
    {
      "epoch": 3.337860107421875e-06,
      "grad_norm": 0.5008355379104614,
      "learning_rate": 6.998000000000001e-06,
      "loss": 0.0761,
      "step": 3500
    },
    {
      "epoch": 3.337860107421875e-06,
      "model_forward_time": 1.7006454467773438,
      "step": 3500
    },
    {
      "dynamics_loss_avg": 0.04730861857533455,
      "epoch": 3.337860107421875e-06,
      "step": 3500
    },
    {
      "action_loss_avg": 0.008322147931903601,
      "epoch": 3.337860107421875e-06,
      "step": 3500
    },
    {
      "epoch": 3.337860107421875e-06,
      "step": 3500,
      "training_step_time": 8.266700983047485
    },
    {
      "epoch": 3.338813781738281e-06,
      "model_forward_time": 1.7070176601409912,
      "step": 3501
    },
    {
      "epoch": 3.338813781738281e-06,
      "step": 3501,
      "training_step_time": 7.4229395389556885
    },
    {
      "epoch": 3.3397674560546873e-06,
      "model_forward_time": 1.723379135131836,
      "step": 3502
    },
    {
      "epoch": 3.3397674560546873e-06,
      "step": 3502,
      "training_step_time": 8.299269914627075
    },
    {
      "epoch": 3.340721130371094e-06,
      "model_forward_time": 2.26531720161438,
      "step": 3503
    },
    {
      "epoch": 3.340721130371094e-06,
      "step": 3503,
      "training_step_time": 8.561538934707642
    },
    {
      "epoch": 3.3416748046875e-06,
      "model_forward_time": 1.8197977542877197,
      "step": 3504
    },
    {
      "epoch": 3.3416748046875e-06,
      "step": 3504,
      "training_step_time": 9.59958791732788
    },
    {
      "epoch": 3.342628479003906e-06,
      "model_forward_time": 1.756650686264038,
      "step": 3505
    },
    {
      "epoch": 3.342628479003906e-06,
      "step": 3505,
      "training_step_time": 9.585003137588501
    },
    {
      "epoch": 3.3435821533203123e-06,
      "model_forward_time": 1.7471802234649658,
      "step": 3506
    },
    {
      "epoch": 3.3435821533203123e-06,
      "step": 3506,
      "training_step_time": 8.61262845993042
    },
    {
      "epoch": 3.344535827636719e-06,
      "model_forward_time": 1.7569048404693604,
      "step": 3507
    },
    {
      "epoch": 3.344535827636719e-06,
      "step": 3507,
      "training_step_time": 7.5357983112335205
    },
    {
      "epoch": 3.345489501953125e-06,
      "model_forward_time": 1.7483623027801514,
      "step": 3508
    },
    {
      "epoch": 3.345489501953125e-06,
      "step": 3508,
      "training_step_time": 7.666558027267456
    },
    {
      "epoch": 3.346443176269531e-06,
      "model_forward_time": 1.7785260677337646,
      "step": 3509
    },
    {
      "epoch": 3.346443176269531e-06,
      "step": 3509,
      "training_step_time": 8.70767092704773
    },
    {
      "epoch": 3.3473968505859373e-06,
      "grad_norm": 0.6006207466125488,
      "learning_rate": 7.018e-06,
      "loss": 0.0813,
      "step": 3510
    },
    {
      "epoch": 3.3473968505859373e-06,
      "model_forward_time": 1.8140323162078857,
      "step": 3510
    },
    {
      "dynamics_loss_avg": 0.07865564152598381,
      "epoch": 3.3473968505859373e-06,
      "step": 3510
    },
    {
      "action_loss_avg": 0.010740024363622069,
      "epoch": 3.3473968505859373e-06,
      "step": 3510
    },
    {
      "epoch": 3.3473968505859373e-06,
      "step": 3510,
      "training_step_time": 7.474530935287476
    },
    {
      "epoch": 3.348350524902344e-06,
      "model_forward_time": 1.7775774002075195,
      "step": 3511
    },
    {
      "epoch": 3.348350524902344e-06,
      "step": 3511,
      "training_step_time": 9.510260820388794
    },
    {
      "epoch": 3.34930419921875e-06,
      "model_forward_time": 2.9076178073883057,
      "step": 3512
    },
    {
      "epoch": 3.34930419921875e-06,
      "step": 3512,
      "training_step_time": 9.030743598937988
    },
    {
      "epoch": 3.350257873535156e-06,
      "model_forward_time": 1.7560455799102783,
      "step": 3513
    },
    {
      "epoch": 3.350257873535156e-06,
      "step": 3513,
      "training_step_time": 8.59479808807373
    },
    {
      "epoch": 3.3512115478515623e-06,
      "model_forward_time": 1.8647668361663818,
      "step": 3514
    },
    {
      "epoch": 3.3512115478515623e-06,
      "step": 3514,
      "training_step_time": 8.76282787322998
    },
    {
      "epoch": 3.352165222167969e-06,
      "model_forward_time": 1.7532050609588623,
      "step": 3515
    },
    {
      "epoch": 3.352165222167969e-06,
      "step": 3515,
      "training_step_time": 7.813978910446167
    },
    {
      "epoch": 3.353118896484375e-06,
      "model_forward_time": 1.7336218357086182,
      "step": 3516
    },
    {
      "epoch": 3.353118896484375e-06,
      "step": 3516,
      "training_step_time": 7.482654333114624
    },
    {
      "epoch": 3.354072570800781e-06,
      "model_forward_time": 1.7369780540466309,
      "step": 3517
    },
    {
      "epoch": 3.354072570800781e-06,
      "step": 3517,
      "training_step_time": 7.763854503631592
    },
    {
      "epoch": 3.3550262451171873e-06,
      "model_forward_time": 1.734832525253296,
      "step": 3518
    },
    {
      "epoch": 3.3550262451171873e-06,
      "step": 3518,
      "training_step_time": 8.501230239868164
    },
    {
      "epoch": 3.355979919433594e-06,
      "model_forward_time": 1.731684684753418,
      "step": 3519
    },
    {
      "epoch": 3.355979919433594e-06,
      "step": 3519,
      "training_step_time": 7.662678956985474
    },
    {
      "epoch": 3.35693359375e-06,
      "grad_norm": 0.5457996129989624,
      "learning_rate": 7.038e-06,
      "loss": 0.089,
      "step": 3520
    },
    {
      "epoch": 3.35693359375e-06,
      "model_forward_time": 1.7335929870605469,
      "step": 3520
    },
    {
      "dynamics_loss_avg": 0.07177421394735575,
      "epoch": 3.35693359375e-06,
      "step": 3520
    },
    {
      "action_loss_avg": 0.011153505183756351,
      "epoch": 3.35693359375e-06,
      "step": 3520
    },
    {
      "epoch": 3.35693359375e-06,
      "step": 3520,
      "training_step_time": 7.812612056732178
    },
    {
      "epoch": 3.357887268066406e-06,
      "model_forward_time": 2.892413377761841,
      "step": 3521
    },
    {
      "epoch": 3.357887268066406e-06,
      "step": 3521,
      "training_step_time": 7.711878299713135
    },
    {
      "epoch": 3.3588409423828123e-06,
      "model_forward_time": 1.7374639511108398,
      "step": 3522
    },
    {
      "epoch": 3.3588409423828123e-06,
      "step": 3522,
      "training_step_time": 8.644685506820679
    },
    {
      "epoch": 3.359794616699219e-06,
      "model_forward_time": 1.7655308246612549,
      "step": 3523
    },
    {
      "epoch": 3.359794616699219e-06,
      "step": 3523,
      "training_step_time": 9.035882472991943
    },
    {
      "epoch": 3.360748291015625e-06,
      "model_forward_time": 1.7379541397094727,
      "step": 3524
    },
    {
      "epoch": 3.360748291015625e-06,
      "step": 3524,
      "training_step_time": 7.705333709716797
    },
    {
      "epoch": 3.361701965332031e-06,
      "model_forward_time": 1.7137446403503418,
      "step": 3525
    },
    {
      "epoch": 3.361701965332031e-06,
      "step": 3525,
      "training_step_time": 7.547595262527466
    },
    {
      "epoch": 3.3626556396484373e-06,
      "model_forward_time": 1.7251505851745605,
      "step": 3526
    },
    {
      "epoch": 3.3626556396484373e-06,
      "step": 3526,
      "training_step_time": 7.659245729446411
    },
    {
      "epoch": 3.363609313964844e-06,
      "model_forward_time": 1.7279536724090576,
      "step": 3527
    },
    {
      "epoch": 3.363609313964844e-06,
      "step": 3527,
      "training_step_time": 7.60891318321228
    },
    {
      "epoch": 3.36456298828125e-06,
      "model_forward_time": 1.7332792282104492,
      "step": 3528
    },
    {
      "epoch": 3.36456298828125e-06,
      "step": 3528,
      "training_step_time": 7.536757946014404
    },
    {
      "epoch": 3.365516662597656e-06,
      "model_forward_time": 1.7365546226501465,
      "step": 3529
    },
    {
      "epoch": 3.365516662597656e-06,
      "step": 3529,
      "training_step_time": 7.801709175109863
    },
    {
      "epoch": 3.3664703369140623e-06,
      "grad_norm": 0.7067126035690308,
      "learning_rate": 7.058e-06,
      "loss": 0.0853,
      "step": 3530
    },
    {
      "epoch": 3.3664703369140623e-06,
      "model_forward_time": 2.8961942195892334,
      "step": 3530
    },
    {
      "dynamics_loss_avg": 0.062158288434147835,
      "epoch": 3.3664703369140623e-06,
      "step": 3530
    },
    {
      "action_loss_avg": 0.007753709843382239,
      "epoch": 3.3664703369140623e-06,
      "step": 3530
    },
    {
      "epoch": 3.3664703369140623e-06,
      "step": 3530,
      "training_step_time": 7.6578192710876465
    },
    {
      "epoch": 3.367424011230469e-06,
      "model_forward_time": 1.7360131740570068,
      "step": 3531
    },
    {
      "epoch": 3.367424011230469e-06,
      "step": 3531,
      "training_step_time": 8.97955322265625
    },
    {
      "epoch": 3.368377685546875e-06,
      "model_forward_time": 1.7576332092285156,
      "step": 3532
    },
    {
      "epoch": 3.368377685546875e-06,
      "step": 3532,
      "training_step_time": 8.584851264953613
    },
    {
      "epoch": 3.369331359863281e-06,
      "model_forward_time": 1.7394990921020508,
      "step": 3533
    },
    {
      "epoch": 3.369331359863281e-06,
      "step": 3533,
      "training_step_time": 7.689433574676514
    },
    {
      "epoch": 3.3702850341796873e-06,
      "model_forward_time": 1.7367298603057861,
      "step": 3534
    },
    {
      "epoch": 3.3702850341796873e-06,
      "step": 3534,
      "training_step_time": 8.5274658203125
    },
    {
      "epoch": 3.371238708496094e-06,
      "model_forward_time": 1.740635871887207,
      "step": 3535
    },
    {
      "epoch": 3.371238708496094e-06,
      "step": 3535,
      "training_step_time": 7.584183216094971
    },
    {
      "epoch": 3.3721923828125e-06,
      "model_forward_time": 1.7396354675292969,
      "step": 3536
    },
    {
      "epoch": 3.3721923828125e-06,
      "step": 3536,
      "training_step_time": 8.426727056503296
    },
    {
      "epoch": 3.373146057128906e-06,
      "model_forward_time": 1.7347354888916016,
      "step": 3537
    },
    {
      "epoch": 3.373146057128906e-06,
      "step": 3537,
      "training_step_time": 7.465537071228027
    },
    {
      "epoch": 3.3740997314453123e-06,
      "model_forward_time": 1.735142469406128,
      "step": 3538
    },
    {
      "epoch": 3.3740997314453123e-06,
      "step": 3538,
      "training_step_time": 8.653119325637817
    },
    {
      "epoch": 3.375053405761719e-06,
      "model_forward_time": 2.9435970783233643,
      "step": 3539
    },
    {
      "epoch": 3.375053405761719e-06,
      "step": 3539,
      "training_step_time": 7.695586919784546
    },
    {
      "epoch": 3.376007080078125e-06,
      "grad_norm": 0.4587980806827545,
      "learning_rate": 7.078000000000001e-06,
      "loss": 0.0848,
      "step": 3540
    },
    {
      "epoch": 3.376007080078125e-06,
      "model_forward_time": 1.7322206497192383,
      "step": 3540
    },
    {
      "dynamics_loss_avg": 0.05895399879664183,
      "epoch": 3.376007080078125e-06,
      "step": 3540
    },
    {
      "action_loss_avg": 0.012346576515119523,
      "epoch": 3.376007080078125e-06,
      "step": 3540
    },
    {
      "epoch": 3.376007080078125e-06,
      "step": 3540,
      "training_step_time": 7.575607538223267
    },
    {
      "epoch": 3.376960754394531e-06,
      "model_forward_time": 1.7443201541900635,
      "step": 3541
    },
    {
      "epoch": 3.376960754394531e-06,
      "step": 3541,
      "training_step_time": 7.762927532196045
    },
    {
      "epoch": 3.3779144287109373e-06,
      "model_forward_time": 1.7333240509033203,
      "step": 3542
    },
    {
      "epoch": 3.3779144287109373e-06,
      "step": 3542,
      "training_step_time": 7.617291688919067
    },
    {
      "epoch": 3.378868103027344e-06,
      "model_forward_time": 1.737074851989746,
      "step": 3543
    },
    {
      "epoch": 3.378868103027344e-06,
      "step": 3543,
      "training_step_time": 8.34913969039917
    },
    {
      "epoch": 3.37982177734375e-06,
      "model_forward_time": 1.7366878986358643,
      "step": 3544
    },
    {
      "epoch": 3.37982177734375e-06,
      "step": 3544,
      "training_step_time": 8.501302480697632
    },
    {
      "epoch": 3.380775451660156e-06,
      "model_forward_time": 1.744807481765747,
      "step": 3545
    },
    {
      "epoch": 3.380775451660156e-06,
      "step": 3545,
      "training_step_time": 7.396009683609009
    },
    {
      "epoch": 3.3817291259765623e-06,
      "model_forward_time": 1.7342021465301514,
      "step": 3546
    },
    {
      "epoch": 3.3817291259765623e-06,
      "step": 3546,
      "training_step_time": 8.662943840026855
    },
    {
      "epoch": 3.382682800292969e-06,
      "model_forward_time": 1.7299144268035889,
      "step": 3547
    },
    {
      "epoch": 3.382682800292969e-06,
      "step": 3547,
      "training_step_time": 7.458653211593628
    },
    {
      "epoch": 3.383636474609375e-06,
      "model_forward_time": 2.4959397315979004,
      "step": 3548
    },
    {
      "epoch": 3.383636474609375e-06,
      "step": 3548,
      "training_step_time": 7.524183988571167
    },
    {
      "epoch": 3.384590148925781e-06,
      "model_forward_time": 1.7381179332733154,
      "step": 3549
    },
    {
      "epoch": 3.384590148925781e-06,
      "step": 3549,
      "training_step_time": 8.728443384170532
    },
    {
      "epoch": 3.3855438232421873e-06,
      "grad_norm": 0.6002715826034546,
      "learning_rate": 7.0980000000000005e-06,
      "loss": 0.0846,
      "step": 3550
    },
    {
      "epoch": 3.3855438232421873e-06,
      "model_forward_time": 1.7288455963134766,
      "step": 3550
    },
    {
      "dynamics_loss_avg": 0.06493359655141831,
      "epoch": 3.3855438232421873e-06,
      "step": 3550
    },
    {
      "action_loss_avg": 0.008519295509904623,
      "epoch": 3.3855438232421873e-06,
      "step": 3550
    },
    {
      "epoch": 3.3855438232421873e-06,
      "step": 3550,
      "training_step_time": 9.241521120071411
    },
    {
      "epoch": 3.386497497558594e-06,
      "model_forward_time": 1.7199351787567139,
      "step": 3551
    },
    {
      "epoch": 3.386497497558594e-06,
      "step": 3551,
      "training_step_time": 8.448365926742554
    },
    {
      "epoch": 3.387451171875e-06,
      "model_forward_time": 1.729985237121582,
      "step": 3552
    },
    {
      "epoch": 3.387451171875e-06,
      "step": 3552,
      "training_step_time": 7.51066517829895
    },
    {
      "epoch": 3.388404846191406e-06,
      "model_forward_time": 1.7229211330413818,
      "step": 3553
    },
    {
      "epoch": 3.388404846191406e-06,
      "step": 3553,
      "training_step_time": 7.415014743804932
    },
    {
      "epoch": 3.3893585205078123e-06,
      "model_forward_time": 1.7380146980285645,
      "step": 3554
    },
    {
      "epoch": 3.3893585205078123e-06,
      "step": 3554,
      "training_step_time": 7.610172510147095
    },
    {
      "epoch": 3.390312194824219e-06,
      "model_forward_time": 1.7421882152557373,
      "step": 3555
    },
    {
      "epoch": 3.390312194824219e-06,
      "step": 3555,
      "training_step_time": 7.589996814727783
    },
    {
      "epoch": 3.391265869140625e-06,
      "model_forward_time": 1.7459847927093506,
      "step": 3556
    },
    {
      "epoch": 3.391265869140625e-06,
      "step": 3556,
      "training_step_time": 7.76461386680603
    },
    {
      "epoch": 3.392219543457031e-06,
      "model_forward_time": 2.2790374755859375,
      "step": 3557
    },
    {
      "epoch": 3.392219543457031e-06,
      "step": 3557,
      "training_step_time": 7.547600269317627
    },
    {
      "epoch": 3.3931732177734373e-06,
      "model_forward_time": 1.795508861541748,
      "step": 3558
    },
    {
      "epoch": 3.3931732177734373e-06,
      "step": 3558,
      "training_step_time": 7.597649097442627
    },
    {
      "epoch": 3.394126892089844e-06,
      "model_forward_time": 1.7414886951446533,
      "step": 3559
    },
    {
      "epoch": 3.394126892089844e-06,
      "step": 3559,
      "training_step_time": 8.134562253952026
    },
    {
      "epoch": 3.39508056640625e-06,
      "grad_norm": 0.3578817844390869,
      "learning_rate": 7.118e-06,
      "loss": 0.0767,
      "step": 3560
    },
    {
      "epoch": 3.39508056640625e-06,
      "model_forward_time": 1.7218725681304932,
      "step": 3560
    },
    {
      "dynamics_loss_avg": 0.061337411403656006,
      "epoch": 3.39508056640625e-06,
      "step": 3560
    },
    {
      "action_loss_avg": 0.010243230685591697,
      "epoch": 3.39508056640625e-06,
      "step": 3560
    },
    {
      "epoch": 3.39508056640625e-06,
      "step": 3560,
      "training_step_time": 7.710005283355713
    },
    {
      "epoch": 3.396034240722656e-06,
      "model_forward_time": 1.7419757843017578,
      "step": 3561
    },
    {
      "epoch": 3.396034240722656e-06,
      "step": 3561,
      "training_step_time": 8.69592022895813
    },
    {
      "epoch": 3.3969879150390623e-06,
      "model_forward_time": 1.738487958908081,
      "step": 3562
    },
    {
      "epoch": 3.3969879150390623e-06,
      "step": 3562,
      "training_step_time": 7.559283494949341
    },
    {
      "epoch": 3.397941589355469e-06,
      "model_forward_time": 1.7404253482818604,
      "step": 3563
    },
    {
      "epoch": 3.397941589355469e-06,
      "step": 3563,
      "training_step_time": 7.698395729064941
    },
    {
      "epoch": 3.398895263671875e-06,
      "model_forward_time": 1.7561275959014893,
      "step": 3564
    },
    {
      "epoch": 3.398895263671875e-06,
      "step": 3564,
      "training_step_time": 8.11031198501587
    },
    {
      "epoch": 3.399848937988281e-06,
      "model_forward_time": 1.7672514915466309,
      "step": 3565
    },
    {
      "epoch": 3.399848937988281e-06,
      "step": 3565,
      "training_step_time": 7.7600274085998535
    },
    {
      "epoch": 3.4008026123046873e-06,
      "model_forward_time": 2.3188438415527344,
      "step": 3566
    },
    {
      "epoch": 3.4008026123046873e-06,
      "step": 3566,
      "training_step_time": 7.503092050552368
    },
    {
      "epoch": 3.401756286621094e-06,
      "model_forward_time": 1.8111979961395264,
      "step": 3567
    },
    {
      "epoch": 3.401756286621094e-06,
      "step": 3567,
      "training_step_time": 9.142732858657837
    },
    {
      "epoch": 3.4027099609375e-06,
      "model_forward_time": 1.7526450157165527,
      "step": 3568
    },
    {
      "epoch": 3.4027099609375e-06,
      "step": 3568,
      "training_step_time": 7.5537428855896
    },
    {
      "epoch": 3.403663635253906e-06,
      "model_forward_time": 1.7521922588348389,
      "step": 3569
    },
    {
      "epoch": 3.403663635253906e-06,
      "step": 3569,
      "training_step_time": 7.790741920471191
    },
    {
      "epoch": 3.4046173095703123e-06,
      "grad_norm": 0.4527120292186737,
      "learning_rate": 7.138e-06,
      "loss": 0.0795,
      "step": 3570
    },
    {
      "epoch": 3.4046173095703123e-06,
      "model_forward_time": 1.750194787979126,
      "step": 3570
    },
    {
      "dynamics_loss_avg": 0.05106478165835142,
      "epoch": 3.4046173095703123e-06,
      "step": 3570
    },
    {
      "action_loss_avg": 0.008616452780552208,
      "epoch": 3.4046173095703123e-06,
      "step": 3570
    },
    {
      "epoch": 3.4046173095703123e-06,
      "step": 3570,
      "training_step_time": 8.13706111907959
    },
    {
      "epoch": 3.405570983886719e-06,
      "model_forward_time": 1.7632596492767334,
      "step": 3571
    },
    {
      "epoch": 3.405570983886719e-06,
      "step": 3571,
      "training_step_time": 7.5097973346710205
    },
    {
      "epoch": 3.406524658203125e-06,
      "model_forward_time": 1.7576427459716797,
      "step": 3572
    },
    {
      "epoch": 3.406524658203125e-06,
      "step": 3572,
      "training_step_time": 7.549531936645508
    },
    {
      "epoch": 3.407478332519531e-06,
      "model_forward_time": 1.7620394229888916,
      "step": 3573
    },
    {
      "epoch": 3.407478332519531e-06,
      "step": 3573,
      "training_step_time": 7.549400329589844
    },
    {
      "epoch": 3.4084320068359374e-06,
      "model_forward_time": 1.7556753158569336,
      "step": 3574
    },
    {
      "epoch": 3.4084320068359374e-06,
      "step": 3574,
      "training_step_time": 7.752102851867676
    },
    {
      "epoch": 3.409385681152344e-06,
      "model_forward_time": 2.321183681488037,
      "step": 3575
    },
    {
      "epoch": 3.409385681152344e-06,
      "step": 3575,
      "training_step_time": 7.506448745727539
    },
    {
      "epoch": 3.41033935546875e-06,
      "model_forward_time": 1.8110673427581787,
      "step": 3576
    },
    {
      "epoch": 3.41033935546875e-06,
      "step": 3576,
      "training_step_time": 7.640058994293213
    },
    {
      "epoch": 3.411293029785156e-06,
      "model_forward_time": 1.7555851936340332,
      "step": 3577
    },
    {
      "epoch": 3.411293029785156e-06,
      "step": 3577,
      "training_step_time": 8.561746835708618
    },
    {
      "epoch": 3.4122467041015624e-06,
      "model_forward_time": 1.7494566440582275,
      "step": 3578
    },
    {
      "epoch": 3.4122467041015624e-06,
      "step": 3578,
      "training_step_time": 8.654195547103882
    },
    {
      "epoch": 3.413200378417969e-06,
      "model_forward_time": 1.764101266860962,
      "step": 3579
    },
    {
      "epoch": 3.413200378417969e-06,
      "step": 3579,
      "training_step_time": 8.598560333251953
    },
    {
      "epoch": 3.414154052734375e-06,
      "grad_norm": 0.250066876411438,
      "learning_rate": 7.158000000000001e-06,
      "loss": 0.0857,
      "step": 3580
    },
    {
      "epoch": 3.414154052734375e-06,
      "model_forward_time": 1.7485463619232178,
      "step": 3580
    },
    {
      "dynamics_loss_avg": 0.06047487985342741,
      "epoch": 3.414154052734375e-06,
      "step": 3580
    },
    {
      "action_loss_avg": 0.007321776077151299,
      "epoch": 3.414154052734375e-06,
      "step": 3580
    },
    {
      "epoch": 3.414154052734375e-06,
      "step": 3580,
      "training_step_time": 7.693924903869629
    },
    {
      "epoch": 3.4151077270507812e-06,
      "model_forward_time": 1.757178544998169,
      "step": 3581
    },
    {
      "epoch": 3.4151077270507812e-06,
      "step": 3581,
      "training_step_time": 7.592257499694824
    },
    {
      "epoch": 3.4160614013671874e-06,
      "model_forward_time": 1.7614707946777344,
      "step": 3582
    },
    {
      "epoch": 3.4160614013671874e-06,
      "step": 3582,
      "training_step_time": 7.519948482513428
    },
    {
      "epoch": 3.417015075683594e-06,
      "model_forward_time": 1.7588717937469482,
      "step": 3583
    },
    {
      "epoch": 3.417015075683594e-06,
      "step": 3583,
      "training_step_time": 8.000144481658936
    },
    {
      "epoch": 3.41796875e-06,
      "model_forward_time": 2.3313097953796387,
      "step": 3584
    },
    {
      "epoch": 3.41796875e-06,
      "step": 3584,
      "training_step_time": 7.920513153076172
    },
    {
      "epoch": 3.4189224243164062e-06,
      "model_forward_time": 1.7922399044036865,
      "step": 3585
    },
    {
      "epoch": 3.4189224243164062e-06,
      "step": 3585,
      "training_step_time": 7.678837537765503
    },
    {
      "epoch": 3.4198760986328124e-06,
      "model_forward_time": 1.7631347179412842,
      "step": 3586
    },
    {
      "epoch": 3.4198760986328124e-06,
      "step": 3586,
      "training_step_time": 8.469230890274048
    },
    {
      "epoch": 3.420829772949219e-06,
      "model_forward_time": 1.7245161533355713,
      "step": 3587
    },
    {
      "epoch": 3.420829772949219e-06,
      "step": 3587,
      "training_step_time": 9.723927974700928
    },
    {
      "epoch": 3.421783447265625e-06,
      "model_forward_time": 1.740140438079834,
      "step": 3588
    },
    {
      "epoch": 3.421783447265625e-06,
      "step": 3588,
      "training_step_time": 8.165455341339111
    },
    {
      "epoch": 3.4227371215820312e-06,
      "model_forward_time": 1.7305324077606201,
      "step": 3589
    },
    {
      "epoch": 3.4227371215820312e-06,
      "step": 3589,
      "training_step_time": 8.688422441482544
    },
    {
      "epoch": 3.4236907958984374e-06,
      "grad_norm": 0.4422224760055542,
      "learning_rate": 7.1780000000000006e-06,
      "loss": 0.0843,
      "step": 3590
    },
    {
      "epoch": 3.4236907958984374e-06,
      "model_forward_time": 1.7320971488952637,
      "step": 3590
    },
    {
      "dynamics_loss_avg": 0.07698571123182774,
      "epoch": 3.4236907958984374e-06,
      "step": 3590
    },
    {
      "action_loss_avg": 0.010520174587145447,
      "epoch": 3.4236907958984374e-06,
      "step": 3590
    },
    {
      "epoch": 3.4236907958984374e-06,
      "step": 3590,
      "training_step_time": 7.593925714492798
    },
    {
      "epoch": 3.424644470214844e-06,
      "model_forward_time": 1.7452325820922852,
      "step": 3591
    },
    {
      "epoch": 3.424644470214844e-06,
      "step": 3591,
      "training_step_time": 7.495694637298584
    },
    {
      "epoch": 3.42559814453125e-06,
      "model_forward_time": 1.741654634475708,
      "step": 3592
    },
    {
      "epoch": 3.42559814453125e-06,
      "step": 3592,
      "training_step_time": 8.477049112319946
    },
    {
      "epoch": 3.4265518188476562e-06,
      "model_forward_time": 2.2873706817626953,
      "step": 3593
    },
    {
      "epoch": 3.4265518188476562e-06,
      "step": 3593,
      "training_step_time": 7.645591735839844
    },
    {
      "epoch": 3.4275054931640624e-06,
      "model_forward_time": 1.7636771202087402,
      "step": 3594
    },
    {
      "epoch": 3.4275054931640624e-06,
      "step": 3594,
      "training_step_time": 7.613862991333008
    },
    {
      "epoch": 3.428459167480469e-06,
      "model_forward_time": 1.7316100597381592,
      "step": 3595
    },
    {
      "epoch": 3.428459167480469e-06,
      "step": 3595,
      "training_step_time": 8.695855140686035
    },
    {
      "epoch": 3.429412841796875e-06,
      "model_forward_time": 1.7366559505462646,
      "step": 3596
    },
    {
      "epoch": 3.429412841796875e-06,
      "step": 3596,
      "training_step_time": 9.450881481170654
    },
    {
      "epoch": 3.4303665161132812e-06,
      "model_forward_time": 1.7027416229248047,
      "step": 3597
    },
    {
      "epoch": 3.4303665161132812e-06,
      "step": 3597,
      "training_step_time": 9.758206844329834
    },
    {
      "epoch": 3.4313201904296874e-06,
      "model_forward_time": 1.7151143550872803,
      "step": 3598
    },
    {
      "epoch": 3.4313201904296874e-06,
      "step": 3598,
      "training_step_time": 7.607663154602051
    },
    {
      "epoch": 3.432273864746094e-06,
      "model_forward_time": 1.7290427684783936,
      "step": 3599
    },
    {
      "epoch": 3.432273864746094e-06,
      "step": 3599,
      "training_step_time": 8.8075430393219
    },
    {
      "epoch": 3.4332275390625e-06,
      "grad_norm": 0.3750140964984894,
      "learning_rate": 7.198e-06,
      "loss": 0.0814,
      "step": 3600
    },
    {
      "epoch": 3.4332275390625e-06,
      "model_forward_time": 1.721177101135254,
      "step": 3600
    },
    {
      "dynamics_loss_avg": 0.05895934477448463,
      "epoch": 3.4332275390625e-06,
      "step": 3600
    },
    {
      "action_loss_avg": 0.008481295127421618,
      "epoch": 3.4332275390625e-06,
      "step": 3600
    },
    {
      "epoch": 3.4332275390625e-06,
      "step": 3600,
      "training_step_time": 8.252690076828003
    },
    {
      "epoch": 3.4341812133789062e-06,
      "model_forward_time": 1.7248988151550293,
      "step": 3601
    },
    {
      "epoch": 3.4341812133789062e-06,
      "step": 3601,
      "training_step_time": 9.337460279464722
    },
    {
      "epoch": 3.4351348876953124e-06,
      "model_forward_time": 2.2592785358428955,
      "step": 3602
    },
    {
      "epoch": 3.4351348876953124e-06,
      "step": 3602,
      "training_step_time": 8.770821332931519
    },
    {
      "epoch": 3.436088562011719e-06,
      "model_forward_time": 1.7255141735076904,
      "step": 3603
    },
    {
      "epoch": 3.436088562011719e-06,
      "step": 3603,
      "training_step_time": 8.581377029418945
    },
    {
      "epoch": 3.437042236328125e-06,
      "model_forward_time": 1.7096078395843506,
      "step": 3604
    },
    {
      "epoch": 3.437042236328125e-06,
      "step": 3604,
      "training_step_time": 7.645388841629028
    },
    {
      "epoch": 3.4379959106445312e-06,
      "model_forward_time": 1.7248411178588867,
      "step": 3605
    },
    {
      "epoch": 3.4379959106445312e-06,
      "step": 3605,
      "training_step_time": 8.664657831192017
    },
    {
      "epoch": 3.4389495849609374e-06,
      "model_forward_time": 1.7397065162658691,
      "step": 3606
    },
    {
      "epoch": 3.4389495849609374e-06,
      "step": 3606,
      "training_step_time": 8.396649360656738
    },
    {
      "epoch": 3.439903259277344e-06,
      "model_forward_time": 1.7183005809783936,
      "step": 3607
    },
    {
      "epoch": 3.439903259277344e-06,
      "step": 3607,
      "training_step_time": 8.092135906219482
    },
    {
      "epoch": 3.44085693359375e-06,
      "model_forward_time": 1.725478172302246,
      "step": 3608
    },
    {
      "epoch": 3.44085693359375e-06,
      "step": 3608,
      "training_step_time": 7.7336745262146
    },
    {
      "epoch": 3.4418106079101562e-06,
      "model_forward_time": 1.7254064083099365,
      "step": 3609
    },
    {
      "epoch": 3.4418106079101562e-06,
      "step": 3609,
      "training_step_time": 7.7348949909210205
    },
    {
      "epoch": 3.4427642822265624e-06,
      "grad_norm": 0.3259921371936798,
      "learning_rate": 7.218e-06,
      "loss": 0.0862,
      "step": 3610
    },
    {
      "epoch": 3.4427642822265624e-06,
      "model_forward_time": 1.7244281768798828,
      "step": 3610
    },
    {
      "dynamics_loss_avg": 0.05384471695870161,
      "epoch": 3.4427642822265624e-06,
      "step": 3610
    },
    {
      "action_loss_avg": 0.0067250464577227834,
      "epoch": 3.4427642822265624e-06,
      "step": 3610
    },
    {
      "epoch": 3.4427642822265624e-06,
      "step": 3610,
      "training_step_time": 7.58072566986084
    },
    {
      "epoch": 3.443717956542969e-06,
      "model_forward_time": 2.2636125087738037,
      "step": 3611
    },
    {
      "epoch": 3.443717956542969e-06,
      "step": 3611,
      "training_step_time": 7.777242660522461
    },
    {
      "epoch": 3.444671630859375e-06,
      "model_forward_time": 1.732846736907959,
      "step": 3612
    },
    {
      "epoch": 3.444671630859375e-06,
      "step": 3612,
      "training_step_time": 7.74914026260376
    },
    {
      "epoch": 3.4456253051757812e-06,
      "model_forward_time": 1.7579102516174316,
      "step": 3613
    },
    {
      "epoch": 3.4456253051757812e-06,
      "step": 3613,
      "training_step_time": 9.014893054962158
    },
    {
      "epoch": 3.4465789794921874e-06,
      "model_forward_time": 1.7175498008728027,
      "step": 3614
    },
    {
      "epoch": 3.4465789794921874e-06,
      "step": 3614,
      "training_step_time": 7.733751058578491
    },
    {
      "epoch": 3.447532653808594e-06,
      "model_forward_time": 1.7076404094696045,
      "step": 3615
    },
    {
      "epoch": 3.447532653808594e-06,
      "step": 3615,
      "training_step_time": 8.432743072509766
    },
    {
      "epoch": 3.448486328125e-06,
      "model_forward_time": 1.718541145324707,
      "step": 3616
    },
    {
      "epoch": 3.448486328125e-06,
      "step": 3616,
      "training_step_time": 8.815345764160156
    },
    {
      "epoch": 3.4494400024414062e-06,
      "model_forward_time": 1.728888750076294,
      "step": 3617
    },
    {
      "epoch": 3.4494400024414062e-06,
      "step": 3617,
      "training_step_time": 7.570794105529785
    },
    {
      "epoch": 3.4503936767578124e-06,
      "model_forward_time": 1.7198834419250488,
      "step": 3618
    },
    {
      "epoch": 3.4503936767578124e-06,
      "step": 3618,
      "training_step_time": 7.579500675201416
    },
    {
      "epoch": 3.451347351074219e-06,
      "model_forward_time": 2.268441677093506,
      "step": 3619
    },
    {
      "epoch": 3.451347351074219e-06,
      "step": 3619,
      "training_step_time": 7.618569612503052
    },
    {
      "epoch": 3.452301025390625e-06,
      "grad_norm": 0.33901530504226685,
      "learning_rate": 7.238000000000001e-06,
      "loss": 0.0824,
      "step": 3620
    },
    {
      "epoch": 3.452301025390625e-06,
      "model_forward_time": 1.701826810836792,
      "step": 3620
    },
    {
      "dynamics_loss_avg": 0.06793496757745743,
      "epoch": 3.452301025390625e-06,
      "step": 3620
    },
    {
      "action_loss_avg": 0.010625974321737886,
      "epoch": 3.452301025390625e-06,
      "step": 3620
    },
    {
      "epoch": 3.452301025390625e-06,
      "step": 3620,
      "training_step_time": 8.216084003448486
    },
    {
      "epoch": 3.4532546997070312e-06,
      "model_forward_time": 1.7125542163848877,
      "step": 3621
    },
    {
      "epoch": 3.4532546997070312e-06,
      "step": 3621,
      "training_step_time": 6.743131399154663
    },
    {
      "epoch": 3.4542083740234374e-06,
      "model_forward_time": 1.3314461708068848,
      "step": 3622
    },
    {
      "epoch": 3.4542083740234374e-06,
      "step": 3622,
      "training_step_time": 6.920266628265381
    },
    {
      "epoch": 3.455162048339844e-06,
      "model_forward_time": 1.7092196941375732,
      "step": 3623
    },
    {
      "epoch": 3.455162048339844e-06,
      "step": 3623,
      "training_step_time": 6.899674415588379
    },
    {
      "epoch": 3.45611572265625e-06,
      "model_forward_time": 1.7132997512817383,
      "step": 3624
    },
    {
      "epoch": 3.45611572265625e-06,
      "step": 3624,
      "training_step_time": 6.861594200134277
    },
    {
      "epoch": 3.4570693969726562e-06,
      "model_forward_time": 1.7017192840576172,
      "step": 3625
    },
    {
      "epoch": 3.4570693969726562e-06,
      "step": 3625,
      "training_step_time": 6.859053373336792
    },
    {
      "epoch": 3.4580230712890624e-06,
      "model_forward_time": 1.7250957489013672,
      "step": 3626
    },
    {
      "epoch": 3.4580230712890624e-06,
      "step": 3626,
      "training_step_time": 6.873987197875977
    },
    {
      "epoch": 3.4589767456054685e-06,
      "model_forward_time": 1.6940429210662842,
      "step": 3627
    },
    {
      "epoch": 3.4589767456054685e-06,
      "step": 3627,
      "training_step_time": 6.8965065479278564
    },
    {
      "epoch": 3.459930419921875e-06,
      "model_forward_time": 1.7036993503570557,
      "step": 3628
    },
    {
      "epoch": 3.459930419921875e-06,
      "step": 3628,
      "training_step_time": 7.730332374572754
    },
    {
      "epoch": 3.4608840942382813e-06,
      "model_forward_time": 2.2722442150115967,
      "step": 3629
    },
    {
      "epoch": 3.4608840942382813e-06,
      "step": 3629,
      "training_step_time": 10.599156856536865
    },
    {
      "epoch": 3.4618377685546874e-06,
      "grad_norm": 0.40535420179367065,
      "learning_rate": 7.258000000000001e-06,
      "loss": 0.08,
      "step": 3630
    },
    {
      "epoch": 3.4618377685546874e-06,
      "model_forward_time": 1.7036159038543701,
      "step": 3630
    },
    {
      "dynamics_loss_avg": 0.05252504535019398,
      "epoch": 3.4618377685546874e-06,
      "step": 3630
    },
    {
      "action_loss_avg": 0.006190582865383476,
      "epoch": 3.4618377685546874e-06,
      "step": 3630
    },
    {
      "epoch": 3.4618377685546874e-06,
      "step": 3630,
      "training_step_time": 7.770726203918457
    },
    {
      "epoch": 3.4627914428710935e-06,
      "model_forward_time": 1.7033360004425049,
      "step": 3631
    },
    {
      "epoch": 3.4627914428710935e-06,
      "step": 3631,
      "training_step_time": 6.898173809051514
    },
    {
      "epoch": 3.4637451171875e-06,
      "model_forward_time": 1.689638376235962,
      "step": 3632
    },
    {
      "epoch": 3.4637451171875e-06,
      "step": 3632,
      "training_step_time": 6.910612344741821
    },
    {
      "epoch": 3.4646987915039063e-06,
      "model_forward_time": 1.7016820907592773,
      "step": 3633
    },
    {
      "epoch": 3.4646987915039063e-06,
      "step": 3633,
      "training_step_time": 6.9085681438446045
    },
    {
      "epoch": 3.4656524658203124e-06,
      "model_forward_time": 0.99637770652771,
      "step": 3634
    },
    {
      "epoch": 3.4656524658203124e-06,
      "step": 3634,
      "training_step_time": 6.908178329467773
    },
    {
      "epoch": 3.4666061401367185e-06,
      "model_forward_time": 1.7319839000701904,
      "step": 3635
    },
    {
      "epoch": 3.4666061401367185e-06,
      "step": 3635,
      "training_step_time": 6.924778699874878
    },
    {
      "epoch": 3.467559814453125e-06,
      "model_forward_time": 1.7058522701263428,
      "step": 3636
    },
    {
      "epoch": 3.467559814453125e-06,
      "step": 3636,
      "training_step_time": 6.87802791595459
    },
    {
      "epoch": 3.4685134887695313e-06,
      "model_forward_time": 1.7051541805267334,
      "step": 3637
    },
    {
      "epoch": 3.4685134887695313e-06,
      "step": 3637,
      "training_step_time": 6.892863512039185
    },
    {
      "epoch": 3.4694671630859374e-06,
      "model_forward_time": 2.256626605987549,
      "step": 3638
    },
    {
      "epoch": 3.4694671630859374e-06,
      "step": 3638,
      "training_step_time": 8.44544792175293
    },
    {
      "epoch": 3.4704208374023436e-06,
      "model_forward_time": 1.708585262298584,
      "step": 3639
    },
    {
      "epoch": 3.4704208374023436e-06,
      "step": 3639,
      "training_step_time": 7.958240032196045
    },
    {
      "epoch": 3.47137451171875e-06,
      "grad_norm": 0.43605759739875793,
      "learning_rate": 7.2780000000000005e-06,
      "loss": 0.0808,
      "step": 3640
    },
    {
      "epoch": 3.47137451171875e-06,
      "model_forward_time": 1.7051584720611572,
      "step": 3640
    },
    {
      "dynamics_loss_avg": 0.06745039857923985,
      "epoch": 3.47137451171875e-06,
      "step": 3640
    },
    {
      "action_loss_avg": 0.012200554157607258,
      "epoch": 3.47137451171875e-06,
      "step": 3640
    },
    {
      "epoch": 3.47137451171875e-06,
      "step": 3640,
      "training_step_time": 6.881330251693726
    },
    {
      "epoch": 3.4723281860351563e-06,
      "model_forward_time": 1.7104201316833496,
      "step": 3641
    },
    {
      "epoch": 3.4723281860351563e-06,
      "step": 3641,
      "training_step_time": 6.915319204330444
    },
    {
      "epoch": 3.4732818603515624e-06,
      "model_forward_time": 1.7028567790985107,
      "step": 3642
    },
    {
      "epoch": 3.4732818603515624e-06,
      "step": 3642,
      "training_step_time": 6.959673881530762
    },
    {
      "epoch": 3.4742355346679686e-06,
      "model_forward_time": 1.7109708786010742,
      "step": 3643
    },
    {
      "epoch": 3.4742355346679686e-06,
      "step": 3643,
      "training_step_time": 6.860992670059204
    },
    {
      "epoch": 3.475189208984375e-06,
      "model_forward_time": 1.7059099674224854,
      "step": 3644
    },
    {
      "epoch": 3.475189208984375e-06,
      "step": 3644,
      "training_step_time": 6.876092910766602
    },
    {
      "epoch": 3.4761428833007813e-06,
      "model_forward_time": 1.7062835693359375,
      "step": 3645
    },
    {
      "epoch": 3.4761428833007813e-06,
      "step": 3645,
      "training_step_time": 6.933290004730225
    },
    {
      "epoch": 3.4770965576171874e-06,
      "model_forward_time": 1.7169671058654785,
      "step": 3646
    },
    {
      "epoch": 3.4770965576171874e-06,
      "step": 3646,
      "training_step_time": 6.901106595993042
    },
    {
      "epoch": 3.4780502319335936e-06,
      "model_forward_time": 2.270555019378662,
      "step": 3647
    },
    {
      "epoch": 3.4780502319335936e-06,
      "step": 3647,
      "training_step_time": 7.825778245925903
    },
    {
      "epoch": 3.47900390625e-06,
      "model_forward_time": 1.7085235118865967,
      "step": 3648
    },
    {
      "epoch": 3.47900390625e-06,
      "step": 3648,
      "training_step_time": 9.093740701675415
    },
    {
      "epoch": 3.4799575805664063e-06,
      "model_forward_time": 1.7068719863891602,
      "step": 3649
    },
    {
      "epoch": 3.4799575805664063e-06,
      "step": 3649,
      "training_step_time": 7.815152883529663
    },
    {
      "epoch": 3.4809112548828124e-06,
      "grad_norm": 0.31234103441238403,
      "learning_rate": 7.298e-06,
      "loss": 0.0734,
      "step": 3650
    },
    {
      "epoch": 3.4809112548828124e-06,
      "model_forward_time": 1.7069118022918701,
      "step": 3650
    },
    {
      "dynamics_loss_avg": 0.05331712123006582,
      "epoch": 3.4809112548828124e-06,
      "step": 3650
    },
    {
      "action_loss_avg": 0.009263394307345152,
      "epoch": 3.4809112548828124e-06,
      "step": 3650
    },
    {
      "epoch": 3.4809112548828124e-06,
      "step": 3650,
      "training_step_time": 6.9186296463012695
    },
    {
      "epoch": 3.4818649291992186e-06,
      "model_forward_time": 1.7034518718719482,
      "step": 3651
    },
    {
      "epoch": 3.4818649291992186e-06,
      "step": 3651,
      "training_step_time": 6.968975782394409
    },
    {
      "epoch": 3.482818603515625e-06,
      "model_forward_time": 1.7055151462554932,
      "step": 3652
    },
    {
      "epoch": 3.482818603515625e-06,
      "step": 3652,
      "training_step_time": 6.870930433273315
    },
    {
      "epoch": 3.4837722778320313e-06,
      "model_forward_time": 1.7058837413787842,
      "step": 3653
    },
    {
      "epoch": 3.4837722778320313e-06,
      "step": 3653,
      "training_step_time": 6.798654556274414
    },
    {
      "epoch": 3.4847259521484374e-06,
      "model_forward_time": 1.7082045078277588,
      "step": 3654
    },
    {
      "epoch": 3.4847259521484374e-06,
      "step": 3654,
      "training_step_time": 6.91189432144165
    },
    {
      "epoch": 3.4856796264648436e-06,
      "model_forward_time": 1.7067911624908447,
      "step": 3655
    },
    {
      "epoch": 3.4856796264648436e-06,
      "step": 3655,
      "training_step_time": 6.9365928173065186
    },
    {
      "epoch": 3.48663330078125e-06,
      "model_forward_time": 1.710404872894287,
      "step": 3656
    },
    {
      "epoch": 3.48663330078125e-06,
      "step": 3656,
      "training_step_time": 8.293959617614746
    },
    {
      "epoch": 3.4875869750976563e-06,
      "model_forward_time": 1.7056829929351807,
      "step": 3657
    },
    {
      "epoch": 3.4875869750976563e-06,
      "step": 3657,
      "training_step_time": 9.836096048355103
    },
    {
      "epoch": 3.4885406494140624e-06,
      "model_forward_time": 1.7075541019439697,
      "step": 3658
    },
    {
      "epoch": 3.4885406494140624e-06,
      "step": 3658,
      "training_step_time": 7.767420530319214
    },
    {
      "epoch": 3.4894943237304686e-06,
      "model_forward_time": 1.7134671211242676,
      "step": 3659
    },
    {
      "epoch": 3.4894943237304686e-06,
      "step": 3659,
      "training_step_time": 6.878677606582642
    },
    {
      "epoch": 3.490447998046875e-06,
      "grad_norm": 0.4448428153991699,
      "learning_rate": 7.318000000000001e-06,
      "loss": 0.0834,
      "step": 3660
    },
    {
      "epoch": 3.490447998046875e-06,
      "model_forward_time": 1.698272705078125,
      "step": 3660
    },
    {
      "dynamics_loss_avg": 0.0676528276875615,
      "epoch": 3.490447998046875e-06,
      "step": 3660
    },
    {
      "action_loss_avg": 0.01020844760350883,
      "epoch": 3.490447998046875e-06,
      "step": 3660
    },
    {
      "epoch": 3.490447998046875e-06,
      "step": 3660,
      "training_step_time": 6.887038707733154
    },
    {
      "epoch": 3.4914016723632813e-06,
      "model_forward_time": 1.706883192062378,
      "step": 3661
    },
    {
      "epoch": 3.4914016723632813e-06,
      "step": 3661,
      "training_step_time": 6.919942140579224
    },
    {
      "epoch": 3.4923553466796874e-06,
      "model_forward_time": 1.7120134830474854,
      "step": 3662
    },
    {
      "epoch": 3.4923553466796874e-06,
      "step": 3662,
      "training_step_time": 6.902409076690674
    },
    {
      "epoch": 3.4933090209960936e-06,
      "model_forward_time": 1.7032558917999268,
      "step": 3663
    },
    {
      "epoch": 3.4933090209960936e-06,
      "step": 3663,
      "training_step_time": 6.960216522216797
    },
    {
      "epoch": 3.4942626953125e-06,
      "model_forward_time": 1.7022480964660645,
      "step": 3664
    },
    {
      "epoch": 3.4942626953125e-06,
      "step": 3664,
      "training_step_time": 6.906069993972778
    },
    {
      "epoch": 3.4952163696289063e-06,
      "model_forward_time": 1.7011442184448242,
      "step": 3665
    },
    {
      "epoch": 3.4952163696289063e-06,
      "step": 3665,
      "training_step_time": 8.68071699142456
    },
    {
      "epoch": 3.4961700439453124e-06,
      "model_forward_time": 2.252161979675293,
      "step": 3666
    },
    {
      "epoch": 3.4961700439453124e-06,
      "step": 3666,
      "training_step_time": 12.522002458572388
    },
    {
      "epoch": 3.4971237182617186e-06,
      "model_forward_time": 1.7014873027801514,
      "step": 3667
    },
    {
      "epoch": 3.4971237182617186e-06,
      "step": 3667,
      "training_step_time": 7.975662708282471
    },
    {
      "epoch": 3.498077392578125e-06,
      "model_forward_time": 1.7093582153320312,
      "step": 3668
    },
    {
      "epoch": 3.498077392578125e-06,
      "step": 3668,
      "training_step_time": 6.864525318145752
    },
    {
      "epoch": 3.4990310668945313e-06,
      "model_forward_time": 1.7113983631134033,
      "step": 3669
    },
    {
      "epoch": 3.4990310668945313e-06,
      "step": 3669,
      "training_step_time": 6.9002697467803955
    },
    {
      "epoch": 3.4999847412109374e-06,
      "grad_norm": 0.3954758048057556,
      "learning_rate": 7.338000000000001e-06,
      "loss": 0.0878,
      "step": 3670
    },
    {
      "epoch": 3.4999847412109374e-06,
      "model_forward_time": 1.7095131874084473,
      "step": 3670
    },
    {
      "dynamics_loss_avg": 0.06889848355203868,
      "epoch": 3.4999847412109374e-06,
      "step": 3670
    },
    {
      "action_loss_avg": 0.00802849738392979,
      "epoch": 3.4999847412109374e-06,
      "step": 3670
    },
    {
      "epoch": 3.4999847412109374e-06,
      "step": 3670,
      "training_step_time": 6.904718637466431
    },
    {
      "epoch": 3.5009384155273436e-06,
      "model_forward_time": 1.7055280208587646,
      "step": 3671
    },
    {
      "epoch": 3.5009384155273436e-06,
      "step": 3671,
      "training_step_time": 6.911357879638672
    },
    {
      "epoch": 3.50189208984375e-06,
      "model_forward_time": 1.7074997425079346,
      "step": 3672
    },
    {
      "epoch": 3.50189208984375e-06,
      "step": 3672,
      "training_step_time": 6.890897035598755
    },
    {
      "epoch": 3.5028457641601563e-06,
      "model_forward_time": 1.6992576122283936,
      "step": 3673
    },
    {
      "epoch": 3.5028457641601563e-06,
      "step": 3673,
      "training_step_time": 6.916651725769043
    },
    {
      "epoch": 3.5037994384765624e-06,
      "model_forward_time": 1.7122352123260498,
      "step": 3674
    },
    {
      "epoch": 3.5037994384765624e-06,
      "step": 3674,
      "training_step_time": 7.7855751514434814
    },
    {
      "epoch": 3.5047531127929686e-06,
      "model_forward_time": 2.295574903488159,
      "step": 3675
    },
    {
      "epoch": 3.5047531127929686e-06,
      "step": 3675,
      "training_step_time": 10.11085295677185
    },
    {
      "epoch": 3.505706787109375e-06,
      "model_forward_time": 1.7148747444152832,
      "step": 3676
    },
    {
      "epoch": 3.505706787109375e-06,
      "step": 3676,
      "training_step_time": 8.97695803642273
    },
    {
      "epoch": 3.5066604614257813e-06,
      "model_forward_time": 1.7143959999084473,
      "step": 3677
    },
    {
      "epoch": 3.5066604614257813e-06,
      "step": 3677,
      "training_step_time": 7.810672998428345
    },
    {
      "epoch": 3.5076141357421874e-06,
      "model_forward_time": 1.7051024436950684,
      "step": 3678
    },
    {
      "epoch": 3.5076141357421874e-06,
      "step": 3678,
      "training_step_time": 6.938199758529663
    },
    {
      "epoch": 3.5085678100585936e-06,
      "model_forward_time": 1.7157220840454102,
      "step": 3679
    },
    {
      "epoch": 3.5085678100585936e-06,
      "step": 3679,
      "training_step_time": 6.91566276550293
    },
    {
      "epoch": 3.509521484375e-06,
      "grad_norm": 0.509283185005188,
      "learning_rate": 7.3580000000000005e-06,
      "loss": 0.0802,
      "step": 3680
    },
    {
      "epoch": 3.509521484375e-06,
      "model_forward_time": 1.7110555171966553,
      "step": 3680
    },
    {
      "dynamics_loss_avg": 0.06607998833060265,
      "epoch": 3.509521484375e-06,
      "step": 3680
    },
    {
      "action_loss_avg": 0.011309322784654796,
      "epoch": 3.509521484375e-06,
      "step": 3680
    },
    {
      "epoch": 3.509521484375e-06,
      "step": 3680,
      "training_step_time": 6.89485502243042
    },
    {
      "epoch": 3.5104751586914063e-06,
      "model_forward_time": 1.7142713069915771,
      "step": 3681
    },
    {
      "epoch": 3.5104751586914063e-06,
      "step": 3681,
      "training_step_time": 6.915806770324707
    },
    {
      "epoch": 3.5114288330078124e-06,
      "model_forward_time": 1.713498592376709,
      "step": 3682
    },
    {
      "epoch": 3.5114288330078124e-06,
      "step": 3682,
      "training_step_time": 6.928795337677002
    },
    {
      "epoch": 3.5123825073242186e-06,
      "model_forward_time": 1.7098219394683838,
      "step": 3683
    },
    {
      "epoch": 3.5123825073242186e-06,
      "step": 3683,
      "training_step_time": 7.940819025039673
    },
    {
      "epoch": 3.513336181640625e-06,
      "model_forward_time": 2.3194077014923096,
      "step": 3684
    },
    {
      "epoch": 3.513336181640625e-06,
      "step": 3684,
      "training_step_time": 8.576698303222656
    },
    {
      "epoch": 3.5142898559570313e-06,
      "model_forward_time": 1.7176403999328613,
      "step": 3685
    },
    {
      "epoch": 3.5142898559570313e-06,
      "step": 3685,
      "training_step_time": 11.047605276107788
    },
    {
      "epoch": 3.5152435302734374e-06,
      "model_forward_time": 1.7078495025634766,
      "step": 3686
    },
    {
      "epoch": 3.5152435302734374e-06,
      "step": 3686,
      "training_step_time": 7.897802114486694
    },
    {
      "epoch": 3.5161972045898436e-06,
      "model_forward_time": 1.7150657176971436,
      "step": 3687
    },
    {
      "epoch": 3.5161972045898436e-06,
      "step": 3687,
      "training_step_time": 6.930432081222534
    },
    {
      "epoch": 3.51715087890625e-06,
      "model_forward_time": 1.7148475646972656,
      "step": 3688
    },
    {
      "epoch": 3.51715087890625e-06,
      "step": 3688,
      "training_step_time": 6.988293886184692
    },
    {
      "epoch": 3.5181045532226563e-06,
      "model_forward_time": 1.7058031558990479,
      "step": 3689
    },
    {
      "epoch": 3.5181045532226563e-06,
      "step": 3689,
      "training_step_time": 6.849256753921509
    },
    {
      "epoch": 3.5190582275390624e-06,
      "grad_norm": 0.6163653135299683,
      "learning_rate": 7.378e-06,
      "loss": 0.0855,
      "step": 3690
    },
    {
      "epoch": 3.5190582275390624e-06,
      "model_forward_time": 1.705784559249878,
      "step": 3690
    },
    {
      "dynamics_loss_avg": 0.07194918040186167,
      "epoch": 3.5190582275390624e-06,
      "step": 3690
    },
    {
      "action_loss_avg": 0.009599701152183115,
      "epoch": 3.5190582275390624e-06,
      "step": 3690
    },
    {
      "epoch": 3.5190582275390624e-06,
      "step": 3690,
      "training_step_time": 6.937834024429321
    },
    {
      "epoch": 3.5200119018554686e-06,
      "model_forward_time": 1.7054238319396973,
      "step": 3691
    },
    {
      "epoch": 3.5200119018554686e-06,
      "step": 3691,
      "training_step_time": 6.939205169677734
    },
    {
      "epoch": 3.520965576171875e-06,
      "model_forward_time": 1.7045495510101318,
      "step": 3692
    },
    {
      "epoch": 3.520965576171875e-06,
      "step": 3692,
      "training_step_time": 7.007702112197876
    },
    {
      "epoch": 3.5219192504882813e-06,
      "model_forward_time": 2.284917116165161,
      "step": 3693
    },
    {
      "epoch": 3.5219192504882813e-06,
      "step": 3693,
      "training_step_time": 8.470180034637451
    },
    {
      "epoch": 3.5228729248046875e-06,
      "model_forward_time": 1.700767993927002,
      "step": 3694
    },
    {
      "epoch": 3.5228729248046875e-06,
      "step": 3694,
      "training_step_time": 9.311620950698853
    },
    {
      "epoch": 3.5238265991210936e-06,
      "model_forward_time": 1.709287166595459,
      "step": 3695
    },
    {
      "epoch": 3.5238265991210936e-06,
      "step": 3695,
      "training_step_time": 8.029362678527832
    },
    {
      "epoch": 3.5247802734375e-06,
      "model_forward_time": 1.7085988521575928,
      "step": 3696
    },
    {
      "epoch": 3.5247802734375e-06,
      "step": 3696,
      "training_step_time": 6.922749757766724
    },
    {
      "epoch": 3.5257339477539063e-06,
      "model_forward_time": 1.7052984237670898,
      "step": 3697
    },
    {
      "epoch": 3.5257339477539063e-06,
      "step": 3697,
      "training_step_time": 6.919342517852783
    },
    {
      "epoch": 3.5266876220703125e-06,
      "model_forward_time": 1.3303956985473633,
      "step": 3698
    },
    {
      "epoch": 3.5266876220703125e-06,
      "step": 3698,
      "training_step_time": 6.909597396850586
    },
    {
      "epoch": 3.5276412963867186e-06,
      "model_forward_time": 1.7108347415924072,
      "step": 3699
    },
    {
      "epoch": 3.5276412963867186e-06,
      "step": 3699,
      "training_step_time": 6.9033043384552
    },
    {
      "epoch": 3.528594970703125e-06,
      "grad_norm": 0.3562106192111969,
      "learning_rate": 7.398000000000001e-06,
      "loss": 0.0824,
      "step": 3700
    },
    {
      "epoch": 3.528594970703125e-06,
      "model_forward_time": 1.710306167602539,
      "step": 3700
    },
    {
      "dynamics_loss_avg": 0.05841680858284235,
      "epoch": 3.528594970703125e-06,
      "step": 3700
    },
    {
      "action_loss_avg": 0.006854036473669112,
      "epoch": 3.528594970703125e-06,
      "step": 3700
    },
    {
      "epoch": 3.528594970703125e-06,
      "step": 3700,
      "training_step_time": 6.902719259262085
    },
    {
      "epoch": 3.5295486450195313e-06,
      "model_forward_time": 1.7022051811218262,
      "step": 3701
    },
    {
      "epoch": 3.5295486450195313e-06,
      "step": 3701,
      "training_step_time": 6.884505271911621
    },
    {
      "epoch": 3.5305023193359375e-06,
      "model_forward_time": 2.292832136154175,
      "step": 3702
    },
    {
      "epoch": 3.5305023193359375e-06,
      "step": 3702,
      "training_step_time": 8.86012601852417
    },
    {
      "epoch": 3.5314559936523436e-06,
      "model_forward_time": 1.7056632041931152,
      "step": 3703
    },
    {
      "epoch": 3.5314559936523436e-06,
      "step": 3703,
      "training_step_time": 9.920581102371216
    },
    {
      "epoch": 3.53240966796875e-06,
      "model_forward_time": 1.709139108657837,
      "step": 3704
    },
    {
      "epoch": 3.53240966796875e-06,
      "step": 3704,
      "training_step_time": 7.991959810256958
    },
    {
      "epoch": 3.5333633422851563e-06,
      "model_forward_time": 1.7063651084899902,
      "step": 3705
    },
    {
      "epoch": 3.5333633422851563e-06,
      "step": 3705,
      "training_step_time": 7.821429967880249
    },
    {
      "epoch": 3.5343170166015625e-06,
      "model_forward_time": 1.709571123123169,
      "step": 3706
    },
    {
      "epoch": 3.5343170166015625e-06,
      "step": 3706,
      "training_step_time": 6.854566335678101
    },
    {
      "epoch": 3.5352706909179686e-06,
      "model_forward_time": 1.7147951126098633,
      "step": 3707
    },
    {
      "epoch": 3.5352706909179686e-06,
      "step": 3707,
      "training_step_time": 6.877976417541504
    },
    {
      "epoch": 3.536224365234375e-06,
      "model_forward_time": 1.7107009887695312,
      "step": 3708
    },
    {
      "epoch": 3.536224365234375e-06,
      "step": 3708,
      "training_step_time": 6.870184421539307
    },
    {
      "epoch": 3.5371780395507813e-06,
      "model_forward_time": 1.7207140922546387,
      "step": 3709
    },
    {
      "epoch": 3.5371780395507813e-06,
      "step": 3709,
      "training_step_time": 6.893730640411377
    },
    {
      "epoch": 3.5381317138671875e-06,
      "grad_norm": 0.4921303689479828,
      "learning_rate": 7.418000000000001e-06,
      "loss": 0.0858,
      "step": 3710
    },
    {
      "epoch": 3.5381317138671875e-06,
      "model_forward_time": 1.7105693817138672,
      "step": 3710
    },
    {
      "dynamics_loss_avg": 0.05726806921884418,
      "epoch": 3.5381317138671875e-06,
      "step": 3710
    },
    {
      "action_loss_avg": 0.006551367742940783,
      "epoch": 3.5381317138671875e-06,
      "step": 3710
    },
    {
      "epoch": 3.5381317138671875e-06,
      "step": 3710,
      "training_step_time": 6.733749866485596
    },
    {
      "epoch": 3.5390853881835936e-06,
      "model_forward_time": 2.2367827892303467,
      "step": 3711
    },
    {
      "epoch": 3.5390853881835936e-06,
      "step": 3711,
      "training_step_time": 9.72347092628479
    },
    {
      "epoch": 3.5400390625e-06,
      "model_forward_time": 1.7067739963531494,
      "step": 3712
    },
    {
      "epoch": 3.5400390625e-06,
      "step": 3712,
      "training_step_time": 9.84530782699585
    },
    {
      "epoch": 3.5409927368164063e-06,
      "model_forward_time": 1.7067549228668213,
      "step": 3713
    },
    {
      "epoch": 3.5409927368164063e-06,
      "step": 3713,
      "training_step_time": 8.965091705322266
    },
    {
      "epoch": 3.5419464111328125e-06,
      "model_forward_time": 1.7020347118377686,
      "step": 3714
    },
    {
      "epoch": 3.5419464111328125e-06,
      "step": 3714,
      "training_step_time": 7.8192644119262695
    },
    {
      "epoch": 3.5429000854492186e-06,
      "model_forward_time": 1.705104112625122,
      "step": 3715
    },
    {
      "epoch": 3.5429000854492186e-06,
      "step": 3715,
      "training_step_time": 6.861292123794556
    },
    {
      "epoch": 3.543853759765625e-06,
      "model_forward_time": 1.7087583541870117,
      "step": 3716
    },
    {
      "epoch": 3.543853759765625e-06,
      "step": 3716,
      "training_step_time": 6.883191823959351
    },
    {
      "epoch": 3.5448074340820313e-06,
      "model_forward_time": 1.7097477912902832,
      "step": 3717
    },
    {
      "epoch": 3.5448074340820313e-06,
      "step": 3717,
      "training_step_time": 6.7430419921875
    },
    {
      "epoch": 3.5457611083984375e-06,
      "model_forward_time": 1.708181619644165,
      "step": 3718
    },
    {
      "epoch": 3.5457611083984375e-06,
      "step": 3718,
      "training_step_time": 6.9047980308532715
    },
    {
      "epoch": 3.5467147827148436e-06,
      "model_forward_time": 1.7026703357696533,
      "step": 3719
    },
    {
      "epoch": 3.5467147827148436e-06,
      "step": 3719,
      "training_step_time": 6.875671625137329
    },
    {
      "epoch": 3.54766845703125e-06,
      "grad_norm": 0.4916965067386627,
      "learning_rate": 7.438000000000001e-06,
      "loss": 0.0774,
      "step": 3720
    },
    {
      "epoch": 3.54766845703125e-06,
      "model_forward_time": 2.2432963848114014,
      "step": 3720
    },
    {
      "dynamics_loss_avg": 0.07022178489714861,
      "epoch": 3.54766845703125e-06,
      "step": 3720
    },
    {
      "action_loss_avg": 0.01060195784084499,
      "epoch": 3.54766845703125e-06,
      "step": 3720
    },
    {
      "epoch": 3.54766845703125e-06,
      "step": 3720,
      "training_step_time": 7.79790472984314
    },
    {
      "epoch": 3.5486221313476563e-06,
      "model_forward_time": 1.7663819789886475,
      "step": 3721
    },
    {
      "epoch": 3.5486221313476563e-06,
      "step": 3721,
      "training_step_time": 10.096450567245483
    },
    {
      "epoch": 3.5495758056640625e-06,
      "model_forward_time": 1.7052271366119385,
      "step": 3722
    },
    {
      "epoch": 3.5495758056640625e-06,
      "step": 3722,
      "training_step_time": 8.055285453796387
    },
    {
      "epoch": 3.5505294799804686e-06,
      "model_forward_time": 1.7121353149414062,
      "step": 3723
    },
    {
      "epoch": 3.5505294799804686e-06,
      "step": 3723,
      "training_step_time": 7.7958502769470215
    },
    {
      "epoch": 3.551483154296875e-06,
      "model_forward_time": 1.7058393955230713,
      "step": 3724
    },
    {
      "epoch": 3.551483154296875e-06,
      "step": 3724,
      "training_step_time": 6.911158800125122
    },
    {
      "epoch": 3.5524368286132813e-06,
      "model_forward_time": 1.706028938293457,
      "step": 3725
    },
    {
      "epoch": 3.5524368286132813e-06,
      "step": 3725,
      "training_step_time": 6.924426078796387
    },
    {
      "epoch": 3.5533905029296875e-06,
      "model_forward_time": 1.7054343223571777,
      "step": 3726
    },
    {
      "epoch": 3.5533905029296875e-06,
      "step": 3726,
      "training_step_time": 6.86103081703186
    },
    {
      "epoch": 3.5543441772460936e-06,
      "model_forward_time": 1.7008943557739258,
      "step": 3727
    },
    {
      "epoch": 3.5543441772460936e-06,
      "step": 3727,
      "training_step_time": 6.920954942703247
    },
    {
      "epoch": 3.5552978515625e-06,
      "model_forward_time": 1.7102017402648926,
      "step": 3728
    },
    {
      "epoch": 3.5552978515625e-06,
      "step": 3728,
      "training_step_time": 6.897092580795288
    },
    {
      "epoch": 3.5562515258789063e-06,
      "model_forward_time": 2.2258217334747314,
      "step": 3729
    },
    {
      "epoch": 3.5562515258789063e-06,
      "step": 3729,
      "training_step_time": 7.748029947280884
    },
    {
      "epoch": 3.5572052001953125e-06,
      "grad_norm": 0.3553106188774109,
      "learning_rate": 7.458e-06,
      "loss": 0.079,
      "step": 3730
    },
    {
      "epoch": 3.5572052001953125e-06,
      "model_forward_time": 1.760617733001709,
      "step": 3730
    },
    {
      "dynamics_loss_avg": 0.062307719700038434,
      "epoch": 3.5572052001953125e-06,
      "step": 3730
    },
    {
      "action_loss_avg": 0.010924934642389417,
      "epoch": 3.5572052001953125e-06,
      "step": 3730
    },
    {
      "epoch": 3.5572052001953125e-06,
      "step": 3730,
      "training_step_time": 8.419029474258423
    },
    {
      "epoch": 3.5581588745117186e-06,
      "model_forward_time": 1.7013304233551025,
      "step": 3731
    },
    {
      "epoch": 3.5581588745117186e-06,
      "step": 3731,
      "training_step_time": 9.545246601104736
    },
    {
      "epoch": 3.559112548828125e-06,
      "model_forward_time": 1.7164618968963623,
      "step": 3732
    },
    {
      "epoch": 3.559112548828125e-06,
      "step": 3732,
      "training_step_time": 7.789932727813721
    },
    {
      "epoch": 3.5600662231445313e-06,
      "model_forward_time": 1.7103197574615479,
      "step": 3733
    },
    {
      "epoch": 3.5600662231445313e-06,
      "step": 3733,
      "training_step_time": 7.803632020950317
    },
    {
      "epoch": 3.5610198974609375e-06,
      "model_forward_time": 1.7104740142822266,
      "step": 3734
    },
    {
      "epoch": 3.5610198974609375e-06,
      "step": 3734,
      "training_step_time": 6.925698757171631
    },
    {
      "epoch": 3.5619735717773436e-06,
      "model_forward_time": 1.7132248878479004,
      "step": 3735
    },
    {
      "epoch": 3.5619735717773436e-06,
      "step": 3735,
      "training_step_time": 6.951402425765991
    },
    {
      "epoch": 3.56292724609375e-06,
      "model_forward_time": 1.7174174785614014,
      "step": 3736
    },
    {
      "epoch": 3.56292724609375e-06,
      "step": 3736,
      "training_step_time": 6.914415121078491
    },
    {
      "epoch": 3.5638809204101563e-06,
      "model_forward_time": 1.7111241817474365,
      "step": 3737
    },
    {
      "epoch": 3.5638809204101563e-06,
      "step": 3737,
      "training_step_time": 6.945871591567993
    },
    {
      "epoch": 3.5648345947265625e-06,
      "model_forward_time": 1.7065966129302979,
      "step": 3738
    },
    {
      "epoch": 3.5648345947265625e-06,
      "step": 3738,
      "training_step_time": 7.771691799163818
    },
    {
      "epoch": 3.5657882690429686e-06,
      "model_forward_time": 2.2889156341552734,
      "step": 3739
    },
    {
      "epoch": 3.5657882690429686e-06,
      "step": 3739,
      "training_step_time": 8.893627405166626
    },
    {
      "epoch": 3.566741943359375e-06,
      "grad_norm": 0.43537306785583496,
      "learning_rate": 7.478000000000001e-06,
      "loss": 0.0879,
      "step": 3740
    },
    {
      "epoch": 3.566741943359375e-06,
      "model_forward_time": 1.7178065776824951,
      "step": 3740
    },
    {
      "dynamics_loss_avg": 0.05414696298539638,
      "epoch": 3.566741943359375e-06,
      "step": 3740
    },
    {
      "action_loss_avg": 0.01098471663426608,
      "epoch": 3.566741943359375e-06,
      "step": 3740
    },
    {
      "epoch": 3.566741943359375e-06,
      "step": 3740,
      "training_step_time": 9.807136058807373
    },
    {
      "epoch": 3.5676956176757813e-06,
      "model_forward_time": 1.722534418106079,
      "step": 3741
    },
    {
      "epoch": 3.5676956176757813e-06,
      "step": 3741,
      "training_step_time": 9.06850266456604
    },
    {
      "epoch": 3.5686492919921875e-06,
      "model_forward_time": 1.7202017307281494,
      "step": 3742
    },
    {
      "epoch": 3.5686492919921875e-06,
      "step": 3742,
      "training_step_time": 7.871659278869629
    },
    {
      "epoch": 3.5696029663085936e-06,
      "model_forward_time": 1.7223784923553467,
      "step": 3743
    },
    {
      "epoch": 3.5696029663085936e-06,
      "step": 3743,
      "training_step_time": 6.979534387588501
    },
    {
      "epoch": 3.570556640625e-06,
      "model_forward_time": 1.723562240600586,
      "step": 3744
    },
    {
      "epoch": 3.570556640625e-06,
      "step": 3744,
      "training_step_time": 6.924943208694458
    },
    {
      "epoch": 3.5715103149414064e-06,
      "model_forward_time": 1.6968719959259033,
      "step": 3745
    },
    {
      "epoch": 3.5715103149414064e-06,
      "step": 3745,
      "training_step_time": 6.975992679595947
    },
    {
      "epoch": 3.5724639892578125e-06,
      "model_forward_time": 1.7104241847991943,
      "step": 3746
    },
    {
      "epoch": 3.5724639892578125e-06,
      "step": 3746,
      "training_step_time": 6.954748630523682
    },
    {
      "epoch": 3.5734176635742186e-06,
      "model_forward_time": 1.7102322578430176,
      "step": 3747
    },
    {
      "epoch": 3.5734176635742186e-06,
      "step": 3747,
      "training_step_time": 8.793177604675293
    },
    {
      "epoch": 3.574371337890625e-06,
      "model_forward_time": 2.278578042984009,
      "step": 3748
    },
    {
      "epoch": 3.574371337890625e-06,
      "step": 3748,
      "training_step_time": 7.899808883666992
    },
    {
      "epoch": 3.5753250122070314e-06,
      "model_forward_time": 1.7027194499969482,
      "step": 3749
    },
    {
      "epoch": 3.5753250122070314e-06,
      "step": 3749,
      "training_step_time": 9.783738851547241
    },
    {
      "epoch": 3.5762786865234375e-06,
      "grad_norm": 0.5123917460441589,
      "learning_rate": 7.498000000000001e-06,
      "loss": 0.0883,
      "step": 3750
    },
    {
      "epoch": 3.5762786865234375e-06,
      "model_forward_time": 1.6912693977355957,
      "step": 3750
    },
    {
      "dynamics_loss_avg": 0.05889129713177681,
      "epoch": 3.5762786865234375e-06,
      "step": 3750
    },
    {
      "action_loss_avg": 0.005414095893502236,
      "epoch": 3.5762786865234375e-06,
      "step": 3750
    },
    {
      "epoch": 3.5762786865234375e-06,
      "step": 3750,
      "training_step_time": 8.021098375320435
    },
    {
      "epoch": 3.5772323608398436e-06,
      "model_forward_time": 1.7165887355804443,
      "step": 3751
    },
    {
      "epoch": 3.5772323608398436e-06,
      "step": 3751,
      "training_step_time": 7.870972394943237
    },
    {
      "epoch": 3.57818603515625e-06,
      "model_forward_time": 1.7149322032928467,
      "step": 3752
    },
    {
      "epoch": 3.57818603515625e-06,
      "step": 3752,
      "training_step_time": 6.939297676086426
    },
    {
      "epoch": 3.5791397094726564e-06,
      "model_forward_time": 1.7092387676239014,
      "step": 3753
    },
    {
      "epoch": 3.5791397094726564e-06,
      "step": 3753,
      "training_step_time": 6.942394971847534
    },
    {
      "epoch": 3.5800933837890625e-06,
      "model_forward_time": 1.7191686630249023,
      "step": 3754
    },
    {
      "epoch": 3.5800933837890625e-06,
      "step": 3754,
      "training_step_time": 6.961244583129883
    },
    {
      "epoch": 3.5810470581054686e-06,
      "model_forward_time": 1.698338270187378,
      "step": 3755
    },
    {
      "epoch": 3.5810470581054686e-06,
      "step": 3755,
      "training_step_time": 6.869343042373657
    },
    {
      "epoch": 3.582000732421875e-06,
      "model_forward_time": 1.7237143516540527,
      "step": 3756
    },
    {
      "epoch": 3.582000732421875e-06,
      "step": 3756,
      "training_step_time": 7.930762529373169
    },
    {
      "epoch": 3.5829544067382814e-06,
      "model_forward_time": 2.2496628761291504,
      "step": 3757
    },
    {
      "epoch": 3.5829544067382814e-06,
      "step": 3757,
      "training_step_time": 9.135292530059814
    },
    {
      "epoch": 3.5839080810546875e-06,
      "model_forward_time": 1.7128808498382568,
      "step": 3758
    },
    {
      "epoch": 3.5839080810546875e-06,
      "step": 3758,
      "training_step_time": 9.714689016342163
    },
    {
      "epoch": 3.5848617553710937e-06,
      "model_forward_time": 1.7214937210083008,
      "step": 3759
    },
    {
      "epoch": 3.5848617553710937e-06,
      "step": 3759,
      "training_step_time": 8.558281183242798
    },
    {
      "epoch": 3.5858154296875e-06,
      "grad_norm": 0.3447176516056061,
      "learning_rate": 7.518000000000001e-06,
      "loss": 0.0896,
      "step": 3760
    },
    {
      "epoch": 3.5858154296875e-06,
      "model_forward_time": 1.7193982601165771,
      "step": 3760
    },
    {
      "dynamics_loss_avg": 0.07599443085491657,
      "epoch": 3.5858154296875e-06,
      "step": 3760
    },
    {
      "action_loss_avg": 0.012710383115336298,
      "epoch": 3.5858154296875e-06,
      "step": 3760
    },
    {
      "epoch": 3.5858154296875e-06,
      "step": 3760,
      "training_step_time": 7.750940561294556
    },
    {
      "epoch": 3.5867691040039064e-06,
      "model_forward_time": 1.7153010368347168,
      "step": 3761
    },
    {
      "epoch": 3.5867691040039064e-06,
      "step": 3761,
      "training_step_time": 6.8521363735198975
    },
    {
      "epoch": 3.5877227783203125e-06,
      "model_forward_time": 1.724480390548706,
      "step": 3762
    },
    {
      "epoch": 3.5877227783203125e-06,
      "step": 3762,
      "training_step_time": 6.738354921340942
    },
    {
      "epoch": 3.5886764526367187e-06,
      "model_forward_time": 1.717087984085083,
      "step": 3763
    },
    {
      "epoch": 3.5886764526367187e-06,
      "step": 3763,
      "training_step_time": 6.924635410308838
    },
    {
      "epoch": 3.589630126953125e-06,
      "model_forward_time": 1.7206289768218994,
      "step": 3764
    },
    {
      "epoch": 3.589630126953125e-06,
      "step": 3764,
      "training_step_time": 6.8932764530181885
    },
    {
      "epoch": 3.5905838012695314e-06,
      "model_forward_time": 1.7183003425598145,
      "step": 3765
    },
    {
      "epoch": 3.5905838012695314e-06,
      "step": 3765,
      "training_step_time": 6.90286922454834
    },
    {
      "epoch": 3.5915374755859375e-06,
      "model_forward_time": 2.2623250484466553,
      "step": 3766
    },
    {
      "epoch": 3.5915374755859375e-06,
      "step": 3766,
      "training_step_time": 7.840364694595337
    },
    {
      "epoch": 3.5924911499023437e-06,
      "model_forward_time": 1.7823126316070557,
      "step": 3767
    },
    {
      "epoch": 3.5924911499023437e-06,
      "step": 3767,
      "training_step_time": 10.985503435134888
    },
    {
      "epoch": 3.59344482421875e-06,
      "model_forward_time": 1.7101504802703857,
      "step": 3768
    },
    {
      "epoch": 3.59344482421875e-06,
      "step": 3768,
      "training_step_time": 8.801422834396362
    },
    {
      "epoch": 3.5943984985351564e-06,
      "model_forward_time": 1.7233493328094482,
      "step": 3769
    },
    {
      "epoch": 3.5943984985351564e-06,
      "step": 3769,
      "training_step_time": 7.853559255599976
    },
    {
      "epoch": 3.5953521728515625e-06,
      "grad_norm": 0.454418420791626,
      "learning_rate": 7.5380000000000005e-06,
      "loss": 0.0923,
      "step": 3770
    },
    {
      "epoch": 3.5953521728515625e-06,
      "model_forward_time": 1.7291889190673828,
      "step": 3770
    },
    {
      "dynamics_loss_avg": 0.07432845830917359,
      "epoch": 3.5953521728515625e-06,
      "step": 3770
    },
    {
      "action_loss_avg": 0.011747391463723033,
      "epoch": 3.5953521728515625e-06,
      "step": 3770
    },
    {
      "epoch": 3.5953521728515625e-06,
      "step": 3770,
      "training_step_time": 6.876243352890015
    },
    {
      "epoch": 3.5963058471679687e-06,
      "model_forward_time": 1.719419240951538,
      "step": 3771
    },
    {
      "epoch": 3.5963058471679687e-06,
      "step": 3771,
      "training_step_time": 6.947238445281982
    },
    {
      "epoch": 3.597259521484375e-06,
      "model_forward_time": 1.717538833618164,
      "step": 3772
    },
    {
      "epoch": 3.597259521484375e-06,
      "step": 3772,
      "training_step_time": 6.972352504730225
    },
    {
      "epoch": 3.5982131958007814e-06,
      "model_forward_time": 1.7205471992492676,
      "step": 3773
    },
    {
      "epoch": 3.5982131958007814e-06,
      "step": 3773,
      "training_step_time": 6.916036367416382
    },
    {
      "epoch": 3.5991668701171875e-06,
      "model_forward_time": 1.7151246070861816,
      "step": 3774
    },
    {
      "epoch": 3.5991668701171875e-06,
      "step": 3774,
      "training_step_time": 6.95063591003418
    },
    {
      "epoch": 3.6001205444335937e-06,
      "model_forward_time": 2.2618091106414795,
      "step": 3775
    },
    {
      "epoch": 3.6001205444335937e-06,
      "step": 3775,
      "training_step_time": 7.862166881561279
    },
    {
      "epoch": 3.60107421875e-06,
      "model_forward_time": 1.7601912021636963,
      "step": 3776
    },
    {
      "epoch": 3.60107421875e-06,
      "step": 3776,
      "training_step_time": 9.839974403381348
    },
    {
      "epoch": 3.6020278930664064e-06,
      "model_forward_time": 1.7149417400360107,
      "step": 3777
    },
    {
      "epoch": 3.6020278930664064e-06,
      "step": 3777,
      "training_step_time": 10.83525538444519
    },
    {
      "epoch": 3.6029815673828125e-06,
      "model_forward_time": 1.716965675354004,
      "step": 3778
    },
    {
      "epoch": 3.6029815673828125e-06,
      "step": 3778,
      "training_step_time": 8.969387292861938
    },
    {
      "epoch": 3.6039352416992187e-06,
      "model_forward_time": 1.7150530815124512,
      "step": 3779
    },
    {
      "epoch": 3.6039352416992187e-06,
      "step": 3779,
      "training_step_time": 7.863074779510498
    },
    {
      "epoch": 3.604888916015625e-06,
      "grad_norm": 0.37457895278930664,
      "learning_rate": 7.558000000000001e-06,
      "loss": 0.0783,
      "step": 3780
    },
    {
      "epoch": 3.604888916015625e-06,
      "model_forward_time": 1.339507818222046,
      "step": 3780
    },
    {
      "dynamics_loss_avg": 0.07194112557917834,
      "epoch": 3.604888916015625e-06,
      "step": 3780
    },
    {
      "action_loss_avg": 0.012215628585545346,
      "epoch": 3.604888916015625e-06,
      "step": 3780
    },
    {
      "epoch": 3.604888916015625e-06,
      "step": 3780,
      "training_step_time": 6.689844131469727
    },
    {
      "epoch": 3.6058425903320314e-06,
      "model_forward_time": 1.338482141494751,
      "step": 3781
    },
    {
      "epoch": 3.6058425903320314e-06,
      "step": 3781,
      "training_step_time": 6.873574733734131
    },
    {
      "epoch": 3.6067962646484375e-06,
      "model_forward_time": 1.722707748413086,
      "step": 3782
    },
    {
      "epoch": 3.6067962646484375e-06,
      "step": 3782,
      "training_step_time": 6.90730094909668
    },
    {
      "epoch": 3.6077499389648437e-06,
      "model_forward_time": 1.333953619003296,
      "step": 3783
    },
    {
      "epoch": 3.6077499389648437e-06,
      "step": 3783,
      "training_step_time": 6.859284162521362
    },
    {
      "epoch": 3.60870361328125e-06,
      "model_forward_time": 1.7202212810516357,
      "step": 3784
    },
    {
      "epoch": 3.60870361328125e-06,
      "step": 3784,
      "training_step_time": 8.627665042877197
    },
    {
      "epoch": 3.6096572875976564e-06,
      "model_forward_time": 2.3066296577453613,
      "step": 3785
    },
    {
      "epoch": 3.6096572875976564e-06,
      "step": 3785,
      "training_step_time": 8.377676010131836
    },
    {
      "epoch": 3.6106109619140625e-06,
      "model_forward_time": 1.7168915271759033,
      "step": 3786
    },
    {
      "epoch": 3.6106109619140625e-06,
      "step": 3786,
      "training_step_time": 10.960012197494507
    },
    {
      "epoch": 3.6115646362304687e-06,
      "model_forward_time": 1.7258520126342773,
      "step": 3787
    },
    {
      "epoch": 3.6115646362304687e-06,
      "step": 3787,
      "training_step_time": 8.00604772567749
    },
    {
      "epoch": 3.612518310546875e-06,
      "model_forward_time": 1.719961404800415,
      "step": 3788
    },
    {
      "epoch": 3.612518310546875e-06,
      "step": 3788,
      "training_step_time": 7.889848232269287
    },
    {
      "epoch": 3.6134719848632814e-06,
      "model_forward_time": 1.721541404724121,
      "step": 3789
    },
    {
      "epoch": 3.6134719848632814e-06,
      "step": 3789,
      "training_step_time": 6.926802396774292
    },
    {
      "epoch": 3.6144256591796875e-06,
      "grad_norm": 0.6152052879333496,
      "learning_rate": 7.578000000000001e-06,
      "loss": 0.0789,
      "step": 3790
    },
    {
      "epoch": 3.6144256591796875e-06,
      "model_forward_time": 1.7218763828277588,
      "step": 3790
    },
    {
      "dynamics_loss_avg": 0.06709224469959736,
      "epoch": 3.6144256591796875e-06,
      "step": 3790
    },
    {
      "action_loss_avg": 0.010343800554983318,
      "epoch": 3.6144256591796875e-06,
      "step": 3790
    },
    {
      "epoch": 3.6144256591796875e-06,
      "step": 3790,
      "training_step_time": 6.896541118621826
    },
    {
      "epoch": 3.6153793334960937e-06,
      "model_forward_time": 1.7191567420959473,
      "step": 3791
    },
    {
      "epoch": 3.6153793334960937e-06,
      "step": 3791,
      "training_step_time": 6.91756272315979
    },
    {
      "epoch": 3.6163330078125e-06,
      "model_forward_time": 1.7198755741119385,
      "step": 3792
    },
    {
      "epoch": 3.6163330078125e-06,
      "step": 3792,
      "training_step_time": 6.714489698410034
    },
    {
      "epoch": 3.6172866821289064e-06,
      "model_forward_time": 1.7248075008392334,
      "step": 3793
    },
    {
      "epoch": 3.6172866821289064e-06,
      "step": 3793,
      "training_step_time": 7.735537052154541
    },
    {
      "epoch": 3.6182403564453125e-06,
      "model_forward_time": 2.30438232421875,
      "step": 3794
    },
    {
      "epoch": 3.6182403564453125e-06,
      "step": 3794,
      "training_step_time": 8.303115367889404
    },
    {
      "epoch": 3.6191940307617187e-06,
      "model_forward_time": 1.6904664039611816,
      "step": 3795
    },
    {
      "epoch": 3.6191940307617187e-06,
      "step": 3795,
      "training_step_time": 9.438464641571045
    },
    {
      "epoch": 3.620147705078125e-06,
      "model_forward_time": 1.7070858478546143,
      "step": 3796
    },
    {
      "epoch": 3.620147705078125e-06,
      "step": 3796,
      "training_step_time": 10.696257829666138
    },
    {
      "epoch": 3.6211013793945314e-06,
      "model_forward_time": 1.7123944759368896,
      "step": 3797
    },
    {
      "epoch": 3.6211013793945314e-06,
      "step": 3797,
      "training_step_time": 7.974357604980469
    },
    {
      "epoch": 3.6220550537109375e-06,
      "model_forward_time": 1.7011797428131104,
      "step": 3798
    },
    {
      "epoch": 3.6220550537109375e-06,
      "step": 3798,
      "training_step_time": 6.908603668212891
    },
    {
      "epoch": 3.6230087280273437e-06,
      "model_forward_time": 1.7061829566955566,
      "step": 3799
    },
    {
      "epoch": 3.6230087280273437e-06,
      "step": 3799,
      "training_step_time": 6.884830474853516
    },
    {
      "epoch": 3.62396240234375e-06,
      "grad_norm": 0.5471476316452026,
      "learning_rate": 7.598000000000001e-06,
      "loss": 0.0791,
      "step": 3800
    },
    {
      "epoch": 3.62396240234375e-06,
      "model_forward_time": 1.6987814903259277,
      "step": 3800
    },
    {
      "dynamics_loss_avg": 0.07204963769763709,
      "epoch": 3.62396240234375e-06,
      "step": 3800
    },
    {
      "action_loss_avg": 0.009675523289479316,
      "epoch": 3.62396240234375e-06,
      "step": 3800
    },
    {
      "epoch": 3.62396240234375e-06,
      "step": 3800,
      "training_step_time": 6.963799476623535
    },
    {
      "epoch": 3.6249160766601564e-06,
      "model_forward_time": 1.7074415683746338,
      "step": 3801
    },
    {
      "epoch": 3.6249160766601564e-06,
      "step": 3801,
      "training_step_time": 6.911706209182739
    },
    {
      "epoch": 3.6258697509765625e-06,
      "model_forward_time": 1.7011098861694336,
      "step": 3802
    },
    {
      "epoch": 3.6258697509765625e-06,
      "step": 3802,
      "training_step_time": 7.4569385051727295
    },
    {
      "epoch": 3.6268234252929687e-06,
      "model_forward_time": 2.2227494716644287,
      "step": 3803
    },
    {
      "epoch": 3.6268234252929687e-06,
      "step": 3803,
      "training_step_time": 8.6808762550354
    },
    {
      "epoch": 3.627777099609375e-06,
      "model_forward_time": 1.7192857265472412,
      "step": 3804
    },
    {
      "epoch": 3.627777099609375e-06,
      "step": 3804,
      "training_step_time": 9.794145584106445
    },
    {
      "epoch": 3.6287307739257814e-06,
      "model_forward_time": 1.700085163116455,
      "step": 3805
    },
    {
      "epoch": 3.6287307739257814e-06,
      "step": 3805,
      "training_step_time": 8.456658363342285
    },
    {
      "epoch": 3.6296844482421875e-06,
      "model_forward_time": 1.71425461769104,
      "step": 3806
    },
    {
      "epoch": 3.6296844482421875e-06,
      "step": 3806,
      "training_step_time": 7.6464550495147705
    },
    {
      "epoch": 3.6306381225585937e-06,
      "model_forward_time": 1.7034170627593994,
      "step": 3807
    },
    {
      "epoch": 3.6306381225585937e-06,
      "step": 3807,
      "training_step_time": 6.871682167053223
    },
    {
      "epoch": 3.631591796875e-06,
      "model_forward_time": 1.7031686305999756,
      "step": 3808
    },
    {
      "epoch": 3.631591796875e-06,
      "step": 3808,
      "training_step_time": 6.880990743637085
    },
    {
      "epoch": 3.6325454711914064e-06,
      "model_forward_time": 1.7025177478790283,
      "step": 3809
    },
    {
      "epoch": 3.6325454711914064e-06,
      "step": 3809,
      "training_step_time": 6.9115259647369385
    },
    {
      "epoch": 3.6334991455078126e-06,
      "grad_norm": 0.6878056526184082,
      "learning_rate": 7.618000000000001e-06,
      "loss": 0.083,
      "step": 3810
    },
    {
      "epoch": 3.6334991455078126e-06,
      "model_forward_time": 1.700925588607788,
      "step": 3810
    },
    {
      "dynamics_loss_avg": 0.06629642210900784,
      "epoch": 3.6334991455078126e-06,
      "step": 3810
    },
    {
      "action_loss_avg": 0.01017203542869538,
      "epoch": 3.6334991455078126e-06,
      "step": 3810
    },
    {
      "epoch": 3.6334991455078126e-06,
      "step": 3810,
      "training_step_time": 6.889358997344971
    },
    {
      "epoch": 3.6344528198242187e-06,
      "model_forward_time": 1.6999790668487549,
      "step": 3811
    },
    {
      "epoch": 3.6344528198242187e-06,
      "step": 3811,
      "training_step_time": 7.602218389511108
    },
    {
      "epoch": 3.635406494140625e-06,
      "model_forward_time": 1.7004711627960205,
      "step": 3812
    },
    {
      "epoch": 3.635406494140625e-06,
      "step": 3812,
      "training_step_time": 8.348071336746216
    },
    {
      "epoch": 3.6363601684570314e-06,
      "model_forward_time": 1.756338119506836,
      "step": 3813
    },
    {
      "epoch": 3.6363601684570314e-06,
      "step": 3813,
      "training_step_time": 9.749064922332764
    },
    {
      "epoch": 3.6373138427734376e-06,
      "model_forward_time": 1.705747127532959,
      "step": 3814
    },
    {
      "epoch": 3.6373138427734376e-06,
      "step": 3814,
      "training_step_time": 8.662959814071655
    },
    {
      "epoch": 3.6382675170898437e-06,
      "model_forward_time": 1.7045753002166748,
      "step": 3815
    },
    {
      "epoch": 3.6382675170898437e-06,
      "step": 3815,
      "training_step_time": 8.024221420288086
    },
    {
      "epoch": 3.63922119140625e-06,
      "model_forward_time": 1.7018818855285645,
      "step": 3816
    },
    {
      "epoch": 3.63922119140625e-06,
      "step": 3816,
      "training_step_time": 7.842714786529541
    },
    {
      "epoch": 3.6401748657226564e-06,
      "model_forward_time": 1.7049272060394287,
      "step": 3817
    },
    {
      "epoch": 3.6401748657226564e-06,
      "step": 3817,
      "training_step_time": 6.92279839515686
    },
    {
      "epoch": 3.6411285400390626e-06,
      "model_forward_time": 1.7065002918243408,
      "step": 3818
    },
    {
      "epoch": 3.6411285400390626e-06,
      "step": 3818,
      "training_step_time": 6.940068006515503
    },
    {
      "epoch": 3.6420822143554687e-06,
      "model_forward_time": 1.7035298347473145,
      "step": 3819
    },
    {
      "epoch": 3.6420822143554687e-06,
      "step": 3819,
      "training_step_time": 6.900421619415283
    },
    {
      "epoch": 3.643035888671875e-06,
      "grad_norm": 0.36519861221313477,
      "learning_rate": 7.638e-06,
      "loss": 0.0804,
      "step": 3820
    },
    {
      "epoch": 3.643035888671875e-06,
      "model_forward_time": 1.705960750579834,
      "step": 3820
    },
    {
      "dynamics_loss_avg": 0.07059589140117169,
      "epoch": 3.643035888671875e-06,
      "step": 3820
    },
    {
      "action_loss_avg": 0.012158370390534402,
      "epoch": 3.643035888671875e-06,
      "step": 3820
    },
    {
      "epoch": 3.643035888671875e-06,
      "step": 3820,
      "training_step_time": 7.356274843215942
    },
    {
      "epoch": 3.6439895629882814e-06,
      "model_forward_time": 1.6941804885864258,
      "step": 3821
    },
    {
      "epoch": 3.6439895629882814e-06,
      "step": 3821,
      "training_step_time": 8.194044351577759
    },
    {
      "epoch": 3.6449432373046876e-06,
      "model_forward_time": 2.314894914627075,
      "step": 3822
    },
    {
      "epoch": 3.6449432373046876e-06,
      "step": 3822,
      "training_step_time": 9.383261680603027
    },
    {
      "epoch": 3.6458969116210937e-06,
      "model_forward_time": 0.9832980632781982,
      "step": 3823
    },
    {
      "epoch": 3.6458969116210937e-06,
      "step": 3823,
      "training_step_time": 9.067429780960083
    },
    {
      "epoch": 3.6468505859375e-06,
      "model_forward_time": 1.7128305435180664,
      "step": 3824
    },
    {
      "epoch": 3.6468505859375e-06,
      "step": 3824,
      "training_step_time": 9.13203239440918
    },
    {
      "epoch": 3.6478042602539064e-06,
      "model_forward_time": 1.7011282444000244,
      "step": 3825
    },
    {
      "epoch": 3.6478042602539064e-06,
      "step": 3825,
      "training_step_time": 7.680366516113281
    },
    {
      "epoch": 3.6487579345703126e-06,
      "model_forward_time": 1.7079811096191406,
      "step": 3826
    },
    {
      "epoch": 3.6487579345703126e-06,
      "step": 3826,
      "training_step_time": 7.829753637313843
    },
    {
      "epoch": 3.6497116088867187e-06,
      "model_forward_time": 1.702113389968872,
      "step": 3827
    },
    {
      "epoch": 3.6497116088867187e-06,
      "step": 3827,
      "training_step_time": 6.978284597396851
    },
    {
      "epoch": 3.650665283203125e-06,
      "model_forward_time": 1.7036714553833008,
      "step": 3828
    },
    {
      "epoch": 3.650665283203125e-06,
      "step": 3828,
      "training_step_time": 6.88904595375061
    },
    {
      "epoch": 3.6516189575195314e-06,
      "model_forward_time": 1.7089033126831055,
      "step": 3829
    },
    {
      "epoch": 3.6516189575195314e-06,
      "step": 3829,
      "training_step_time": 7.599971532821655
    },
    {
      "epoch": 3.6525726318359376e-06,
      "grad_norm": 0.44845691323280334,
      "learning_rate": 7.658e-06,
      "loss": 0.085,
      "step": 3830
    },
    {
      "epoch": 3.6525726318359376e-06,
      "model_forward_time": 1.7001945972442627,
      "step": 3830
    },
    {
      "dynamics_loss_avg": 0.06568097043782473,
      "epoch": 3.6525726318359376e-06,
      "step": 3830
    },
    {
      "action_loss_avg": 0.010044202487915755,
      "epoch": 3.6525726318359376e-06,
      "step": 3830
    },
    {
      "epoch": 3.6525726318359376e-06,
      "step": 3830,
      "training_step_time": 7.817485094070435
    },
    {
      "epoch": 3.6535263061523437e-06,
      "model_forward_time": 2.2907135486602783,
      "step": 3831
    },
    {
      "epoch": 3.6535263061523437e-06,
      "step": 3831,
      "training_step_time": 10.306449890136719
    },
    {
      "epoch": 3.65447998046875e-06,
      "model_forward_time": 1.7040092945098877,
      "step": 3832
    },
    {
      "epoch": 3.65447998046875e-06,
      "step": 3832,
      "training_step_time": 8.862571716308594
    },
    {
      "epoch": 3.6554336547851564e-06,
      "model_forward_time": 1.7058207988739014,
      "step": 3833
    },
    {
      "epoch": 3.6554336547851564e-06,
      "step": 3833,
      "training_step_time": 9.012496948242188
    },
    {
      "epoch": 3.6563873291015626e-06,
      "model_forward_time": 1.7116658687591553,
      "step": 3834
    },
    {
      "epoch": 3.6563873291015626e-06,
      "step": 3834,
      "training_step_time": 7.959527969360352
    },
    {
      "epoch": 3.6573410034179687e-06,
      "model_forward_time": 1.7037935256958008,
      "step": 3835
    },
    {
      "epoch": 3.6573410034179687e-06,
      "step": 3835,
      "training_step_time": 7.835317850112915
    },
    {
      "epoch": 3.658294677734375e-06,
      "model_forward_time": 1.7115094661712646,
      "step": 3836
    },
    {
      "epoch": 3.658294677734375e-06,
      "step": 3836,
      "training_step_time": 6.97118353843689
    },
    {
      "epoch": 3.6592483520507814e-06,
      "model_forward_time": 1.7030153274536133,
      "step": 3837
    },
    {
      "epoch": 3.6592483520507814e-06,
      "step": 3837,
      "training_step_time": 6.956812143325806
    },
    {
      "epoch": 3.6602020263671876e-06,
      "model_forward_time": 1.7113091945648193,
      "step": 3838
    },
    {
      "epoch": 3.6602020263671876e-06,
      "step": 3838,
      "training_step_time": 7.356993913650513
    },
    {
      "epoch": 3.6611557006835937e-06,
      "model_forward_time": 1.695807933807373,
      "step": 3839
    },
    {
      "epoch": 3.6611557006835937e-06,
      "step": 3839,
      "training_step_time": 7.838536024093628
    },
    {
      "epoch": 3.662109375e-06,
      "grad_norm": 0.3136507570743561,
      "learning_rate": 7.678000000000002e-06,
      "loss": 0.0816,
      "step": 3840
    },
    {
      "epoch": 3.662109375e-06,
      "model_forward_time": 2.256085157394409,
      "step": 3840
    },
    {
      "dynamics_loss_avg": 0.05256224051117897,
      "epoch": 3.662109375e-06,
      "step": 3840
    },
    {
      "action_loss_avg": 0.006385555397719145,
      "epoch": 3.662109375e-06,
      "step": 3840
    },
    {
      "epoch": 3.662109375e-06,
      "step": 3840,
      "training_step_time": 8.42143988609314
    },
    {
      "epoch": 3.6630630493164064e-06,
      "model_forward_time": 1.7039382457733154,
      "step": 3841
    },
    {
      "epoch": 3.6630630493164064e-06,
      "step": 3841,
      "training_step_time": 9.234641313552856
    },
    {
      "epoch": 3.6640167236328126e-06,
      "model_forward_time": 1.7079741954803467,
      "step": 3842
    },
    {
      "epoch": 3.6640167236328126e-06,
      "step": 3842,
      "training_step_time": 8.068049669265747
    },
    {
      "epoch": 3.6649703979492187e-06,
      "model_forward_time": 1.7007474899291992,
      "step": 3843
    },
    {
      "epoch": 3.6649703979492187e-06,
      "step": 3843,
      "training_step_time": 7.989848375320435
    },
    {
      "epoch": 3.665924072265625e-06,
      "model_forward_time": 1.70988130569458,
      "step": 3844
    },
    {
      "epoch": 3.665924072265625e-06,
      "step": 3844,
      "training_step_time": 7.905717134475708
    },
    {
      "epoch": 3.6668777465820314e-06,
      "model_forward_time": 1.703617811203003,
      "step": 3845
    },
    {
      "epoch": 3.6668777465820314e-06,
      "step": 3845,
      "training_step_time": 6.978130578994751
    },
    {
      "epoch": 3.6678314208984376e-06,
      "model_forward_time": 1.7113966941833496,
      "step": 3846
    },
    {
      "epoch": 3.6678314208984376e-06,
      "step": 3846,
      "training_step_time": 6.964917182922363
    },
    {
      "epoch": 3.6687850952148437e-06,
      "model_forward_time": 1.6977863311767578,
      "step": 3847
    },
    {
      "epoch": 3.6687850952148437e-06,
      "step": 3847,
      "training_step_time": 6.9248206615448
    },
    {
      "epoch": 3.66973876953125e-06,
      "model_forward_time": 1.7054996490478516,
      "step": 3848
    },
    {
      "epoch": 3.66973876953125e-06,
      "step": 3848,
      "training_step_time": 8.568607091903687
    },
    {
      "epoch": 3.6706924438476564e-06,
      "model_forward_time": 2.2477381229400635,
      "step": 3849
    },
    {
      "epoch": 3.6706924438476564e-06,
      "step": 3849,
      "training_step_time": 9.252790927886963
    },
    {
      "epoch": 3.6716461181640626e-06,
      "grad_norm": 0.42784368991851807,
      "learning_rate": 7.698000000000002e-06,
      "loss": 0.0854,
      "step": 3850
    },
    {
      "epoch": 3.6716461181640626e-06,
      "model_forward_time": 1.6997044086456299,
      "step": 3850
    },
    {
      "dynamics_loss_avg": 0.07174902949482202,
      "epoch": 3.6716461181640626e-06,
      "step": 3850
    },
    {
      "action_loss_avg": 0.011187193333171308,
      "epoch": 3.6716461181640626e-06,
      "step": 3850
    },
    {
      "epoch": 3.6716461181640626e-06,
      "step": 3850,
      "training_step_time": 8.577849388122559
    },
    {
      "epoch": 3.6725997924804687e-06,
      "model_forward_time": 1.706346035003662,
      "step": 3851
    },
    {
      "epoch": 3.6725997924804687e-06,
      "step": 3851,
      "training_step_time": 8.913405179977417
    },
    {
      "epoch": 3.673553466796875e-06,
      "model_forward_time": 0.9941797256469727,
      "step": 3852
    },
    {
      "epoch": 3.673553466796875e-06,
      "step": 3852,
      "training_step_time": 8.060529708862305
    },
    {
      "epoch": 3.6745071411132814e-06,
      "model_forward_time": 1.7075598239898682,
      "step": 3853
    },
    {
      "epoch": 3.6745071411132814e-06,
      "step": 3853,
      "training_step_time": 7.8967673778533936
    },
    {
      "epoch": 3.6754608154296876e-06,
      "model_forward_time": 1.70143461227417,
      "step": 3854
    },
    {
      "epoch": 3.6754608154296876e-06,
      "step": 3854,
      "training_step_time": 6.944085121154785
    },
    {
      "epoch": 3.6764144897460937e-06,
      "model_forward_time": 1.7046098709106445,
      "step": 3855
    },
    {
      "epoch": 3.6764144897460937e-06,
      "step": 3855,
      "training_step_time": 6.929959774017334
    },
    {
      "epoch": 3.6773681640625e-06,
      "model_forward_time": 1.677586555480957,
      "step": 3856
    },
    {
      "epoch": 3.6773681640625e-06,
      "step": 3856,
      "training_step_time": 6.952286958694458
    },
    {
      "epoch": 3.6783218383789064e-06,
      "model_forward_time": 1.6966493129730225,
      "step": 3857
    },
    {
      "epoch": 3.6783218383789064e-06,
      "step": 3857,
      "training_step_time": 7.758943557739258
    },
    {
      "epoch": 3.6792755126953126e-06,
      "model_forward_time": 1.3313493728637695,
      "step": 3858
    },
    {
      "epoch": 3.6792755126953126e-06,
      "step": 3858,
      "training_step_time": 7.82561469078064
    },
    {
      "epoch": 3.6802291870117187e-06,
      "model_forward_time": 2.265500068664551,
      "step": 3859
    },
    {
      "epoch": 3.6802291870117187e-06,
      "step": 3859,
      "training_step_time": 10.173846244812012
    },
    {
      "epoch": 3.681182861328125e-06,
      "grad_norm": 0.47052454948425293,
      "learning_rate": 7.718000000000001e-06,
      "loss": 0.0878,
      "step": 3860
    },
    {
      "epoch": 3.681182861328125e-06,
      "model_forward_time": 1.6944632530212402,
      "step": 3860
    },
    {
      "dynamics_loss_avg": 0.05503144860267639,
      "epoch": 3.681182861328125e-06,
      "step": 3860
    },
    {
      "action_loss_avg": 0.006580460024997592,
      "epoch": 3.681182861328125e-06,
      "step": 3860
    },
    {
      "epoch": 3.681182861328125e-06,
      "step": 3860,
      "training_step_time": 7.913609743118286
    },
    {
      "epoch": 3.6821365356445314e-06,
      "model_forward_time": 1.7007148265838623,
      "step": 3861
    },
    {
      "epoch": 3.6821365356445314e-06,
      "step": 3861,
      "training_step_time": 8.027677059173584
    },
    {
      "epoch": 3.6830902099609376e-06,
      "model_forward_time": 1.697723627090454,
      "step": 3862
    },
    {
      "epoch": 3.6830902099609376e-06,
      "step": 3862,
      "training_step_time": 7.913298845291138
    },
    {
      "epoch": 3.6840438842773437e-06,
      "model_forward_time": 1.692314624786377,
      "step": 3863
    },
    {
      "epoch": 3.6840438842773437e-06,
      "step": 3863,
      "training_step_time": 6.730189800262451
    },
    {
      "epoch": 3.68499755859375e-06,
      "model_forward_time": 1.6963045597076416,
      "step": 3864
    },
    {
      "epoch": 3.68499755859375e-06,
      "step": 3864,
      "training_step_time": 6.872108697891235
    },
    {
      "epoch": 3.6859512329101565e-06,
      "model_forward_time": 1.705981969833374,
      "step": 3865
    },
    {
      "epoch": 3.6859512329101565e-06,
      "step": 3865,
      "training_step_time": 6.900649070739746
    },
    {
      "epoch": 3.6869049072265626e-06,
      "model_forward_time": 1.6924569606781006,
      "step": 3866
    },
    {
      "epoch": 3.6869049072265626e-06,
      "step": 3866,
      "training_step_time": 7.67808985710144
    },
    {
      "epoch": 3.6878585815429687e-06,
      "model_forward_time": 1.700395107269287,
      "step": 3867
    },
    {
      "epoch": 3.6878585815429687e-06,
      "step": 3867,
      "training_step_time": 8.285271644592285
    },
    {
      "epoch": 3.688812255859375e-06,
      "model_forward_time": 2.262247323989868,
      "step": 3868
    },
    {
      "epoch": 3.688812255859375e-06,
      "step": 3868,
      "training_step_time": 8.622347831726074
    },
    {
      "epoch": 3.6897659301757815e-06,
      "model_forward_time": 1.702373743057251,
      "step": 3869
    },
    {
      "epoch": 3.6897659301757815e-06,
      "step": 3869,
      "training_step_time": 8.97520923614502
    },
    {
      "epoch": 3.6907196044921876e-06,
      "grad_norm": 0.5243707895278931,
      "learning_rate": 7.738000000000001e-06,
      "loss": 0.0841,
      "step": 3870
    },
    {
      "epoch": 3.6907196044921876e-06,
      "model_forward_time": 1.69417142868042,
      "step": 3870
    },
    {
      "dynamics_loss_avg": 0.08501752763986588,
      "epoch": 3.6907196044921876e-06,
      "step": 3870
    },
    {
      "action_loss_avg": 0.01389152999036014,
      "epoch": 3.6907196044921876e-06,
      "step": 3870
    },
    {
      "epoch": 3.6907196044921876e-06,
      "step": 3870,
      "training_step_time": 8.239237308502197
    },
    {
      "epoch": 3.6916732788085937e-06,
      "model_forward_time": 1.7041971683502197,
      "step": 3871
    },
    {
      "epoch": 3.6916732788085937e-06,
      "step": 3871,
      "training_step_time": 7.952699661254883
    },
    {
      "epoch": 3.692626953125e-06,
      "model_forward_time": 1.6987433433532715,
      "step": 3872
    },
    {
      "epoch": 3.692626953125e-06,
      "step": 3872,
      "training_step_time": 7.6856842041015625
    },
    {
      "epoch": 3.6935806274414065e-06,
      "model_forward_time": 1.6992835998535156,
      "step": 3873
    },
    {
      "epoch": 3.6935806274414065e-06,
      "step": 3873,
      "training_step_time": 6.916664123535156
    },
    {
      "epoch": 3.6945343017578126e-06,
      "model_forward_time": 1.6923279762268066,
      "step": 3874
    },
    {
      "epoch": 3.6945343017578126e-06,
      "step": 3874,
      "training_step_time": 6.867555141448975
    },
    {
      "epoch": 3.6954879760742188e-06,
      "model_forward_time": 1.6941571235656738,
      "step": 3875
    },
    {
      "epoch": 3.6954879760742188e-06,
      "step": 3875,
      "training_step_time": 7.742270231246948
    },
    {
      "epoch": 3.696441650390625e-06,
      "model_forward_time": 1.7217752933502197,
      "step": 3876
    },
    {
      "epoch": 3.696441650390625e-06,
      "step": 3876,
      "training_step_time": 7.737267732620239
    },
    {
      "epoch": 3.697395324707031e-06,
      "model_forward_time": 2.270380973815918,
      "step": 3877
    },
    {
      "epoch": 3.697395324707031e-06,
      "step": 3877,
      "training_step_time": 9.341562509536743
    },
    {
      "epoch": 3.6983489990234376e-06,
      "model_forward_time": 1.7046079635620117,
      "step": 3878
    },
    {
      "epoch": 3.6983489990234376e-06,
      "step": 3878,
      "training_step_time": 8.986557245254517
    },
    {
      "epoch": 3.6993026733398438e-06,
      "model_forward_time": 1.6931374073028564,
      "step": 3879
    },
    {
      "epoch": 3.6993026733398438e-06,
      "step": 3879,
      "training_step_time": 8.287126302719116
    },
    {
      "epoch": 3.70025634765625e-06,
      "grad_norm": 0.35334035754203796,
      "learning_rate": 7.758000000000001e-06,
      "loss": 0.0868,
      "step": 3880
    },
    {
      "epoch": 3.70025634765625e-06,
      "model_forward_time": 1.6908905506134033,
      "step": 3880
    },
    {
      "dynamics_loss_avg": 0.07730155885219574,
      "epoch": 3.70025634765625e-06,
      "step": 3880
    },
    {
      "action_loss_avg": 0.009383353078737855,
      "epoch": 3.70025634765625e-06,
      "step": 3880
    },
    {
      "epoch": 3.70025634765625e-06,
      "step": 3880,
      "training_step_time": 9.583593845367432
    },
    {
      "epoch": 3.701210021972656e-06,
      "model_forward_time": 1.7008771896362305,
      "step": 3881
    },
    {
      "epoch": 3.701210021972656e-06,
      "step": 3881,
      "training_step_time": 7.795164346694946
    },
    {
      "epoch": 3.7021636962890626e-06,
      "model_forward_time": 1.7265443801879883,
      "step": 3882
    },
    {
      "epoch": 3.7021636962890626e-06,
      "step": 3882,
      "training_step_time": 6.835883617401123
    },
    {
      "epoch": 3.7031173706054688e-06,
      "model_forward_time": 1.696821928024292,
      "step": 3883
    },
    {
      "epoch": 3.7031173706054688e-06,
      "step": 3883,
      "training_step_time": 6.890115737915039
    },
    {
      "epoch": 3.704071044921875e-06,
      "model_forward_time": 1.6932249069213867,
      "step": 3884
    },
    {
      "epoch": 3.704071044921875e-06,
      "step": 3884,
      "training_step_time": 8.435992956161499
    },
    {
      "epoch": 3.705024719238281e-06,
      "model_forward_time": 1.7065644264221191,
      "step": 3885
    },
    {
      "epoch": 3.705024719238281e-06,
      "step": 3885,
      "training_step_time": 8.287444829940796
    },
    {
      "epoch": 3.7059783935546876e-06,
      "model_forward_time": 1.6969220638275146,
      "step": 3886
    },
    {
      "epoch": 3.7059783935546876e-06,
      "step": 3886,
      "training_step_time": 8.606729745864868
    },
    {
      "epoch": 3.7069320678710938e-06,
      "model_forward_time": 2.2631618976593018,
      "step": 3887
    },
    {
      "epoch": 3.7069320678710938e-06,
      "step": 3887,
      "training_step_time": 7.978029727935791
    },
    {
      "epoch": 3.7078857421875e-06,
      "model_forward_time": 1.701322317123413,
      "step": 3888
    },
    {
      "epoch": 3.7078857421875e-06,
      "step": 3888,
      "training_step_time": 8.74730634689331
    },
    {
      "epoch": 3.708839416503906e-06,
      "model_forward_time": 1.6998093128204346,
      "step": 3889
    },
    {
      "epoch": 3.708839416503906e-06,
      "step": 3889,
      "training_step_time": 7.981032133102417
    },
    {
      "epoch": 3.7097930908203126e-06,
      "grad_norm": 0.29542139172554016,
      "learning_rate": 7.778e-06,
      "loss": 0.0836,
      "step": 3890
    },
    {
      "epoch": 3.7097930908203126e-06,
      "model_forward_time": 1.6901378631591797,
      "step": 3890
    },
    {
      "dynamics_loss_avg": 0.07500817999243736,
      "epoch": 3.7097930908203126e-06,
      "step": 3890
    },
    {
      "action_loss_avg": 0.009094458259642123,
      "epoch": 3.7097930908203126e-06,
      "step": 3890
    },
    {
      "epoch": 3.7097930908203126e-06,
      "step": 3890,
      "training_step_time": 8.740610361099243
    },
    {
      "epoch": 3.7107467651367188e-06,
      "model_forward_time": 1.6992099285125732,
      "step": 3891
    },
    {
      "epoch": 3.7107467651367188e-06,
      "step": 3891,
      "training_step_time": 6.859388828277588
    },
    {
      "epoch": 3.711700439453125e-06,
      "model_forward_time": 1.698462724685669,
      "step": 3892
    },
    {
      "epoch": 3.711700439453125e-06,
      "step": 3892,
      "training_step_time": 6.826121091842651
    },
    {
      "epoch": 3.712654113769531e-06,
      "model_forward_time": 1.7072792053222656,
      "step": 3893
    },
    {
      "epoch": 3.712654113769531e-06,
      "step": 3893,
      "training_step_time": 7.9126198291778564
    },
    {
      "epoch": 3.7136077880859376e-06,
      "model_forward_time": 1.3248822689056396,
      "step": 3894
    },
    {
      "epoch": 3.7136077880859376e-06,
      "step": 3894,
      "training_step_time": 8.435035705566406
    },
    {
      "epoch": 3.7145614624023438e-06,
      "model_forward_time": 1.7028224468231201,
      "step": 3895
    },
    {
      "epoch": 3.7145614624023438e-06,
      "step": 3895,
      "training_step_time": 7.7713704109191895
    },
    {
      "epoch": 3.71551513671875e-06,
      "model_forward_time": 1.9241673946380615,
      "step": 3896
    },
    {
      "epoch": 3.71551513671875e-06,
      "step": 3896,
      "training_step_time": 9.059799194335938
    },
    {
      "epoch": 3.716468811035156e-06,
      "model_forward_time": 1.6974520683288574,
      "step": 3897
    },
    {
      "epoch": 3.716468811035156e-06,
      "step": 3897,
      "training_step_time": 8.794343709945679
    },
    {
      "epoch": 3.7174224853515626e-06,
      "model_forward_time": 1.6974165439605713,
      "step": 3898
    },
    {
      "epoch": 3.7174224853515626e-06,
      "step": 3898,
      "training_step_time": 7.7828004360198975
    },
    {
      "epoch": 3.7183761596679688e-06,
      "model_forward_time": 1.7016606330871582,
      "step": 3899
    },
    {
      "epoch": 3.7183761596679688e-06,
      "step": 3899,
      "training_step_time": 7.848342657089233
    },
    {
      "epoch": 3.719329833984375e-06,
      "grad_norm": 0.3277905285358429,
      "learning_rate": 7.798e-06,
      "loss": 0.0795,
      "step": 3900
    },
    {
      "epoch": 3.719329833984375e-06,
      "model_forward_time": 1.707831621170044,
      "step": 3900
    },
    {
      "dynamics_loss_avg": 0.06123728565871715,
      "epoch": 3.719329833984375e-06,
      "step": 3900
    },
    {
      "action_loss_avg": 0.007628993212711066,
      "epoch": 3.719329833984375e-06,
      "step": 3900
    },
    {
      "epoch": 3.719329833984375e-06,
      "step": 3900,
      "training_step_time": 7.720779657363892
    },
    {
      "epoch": 3.720283508300781e-06,
      "model_forward_time": 1.7028717994689941,
      "step": 3901
    },
    {
      "epoch": 3.720283508300781e-06,
      "step": 3901,
      "training_step_time": 6.881331443786621
    },
    {
      "epoch": 3.7212371826171876e-06,
      "model_forward_time": 1.7067551612854004,
      "step": 3902
    },
    {
      "epoch": 3.7212371826171876e-06,
      "step": 3902,
      "training_step_time": 6.876275062561035
    },
    {
      "epoch": 3.7221908569335938e-06,
      "model_forward_time": 1.701204538345337,
      "step": 3903
    },
    {
      "epoch": 3.7221908569335938e-06,
      "step": 3903,
      "training_step_time": 8.73233151435852
    },
    {
      "epoch": 3.72314453125e-06,
      "model_forward_time": 1.70745849609375,
      "step": 3904
    },
    {
      "epoch": 3.72314453125e-06,
      "step": 3904,
      "training_step_time": 8.81914758682251
    },
    {
      "epoch": 3.724098205566406e-06,
      "model_forward_time": 2.2681634426116943,
      "step": 3905
    },
    {
      "epoch": 3.724098205566406e-06,
      "step": 3905,
      "training_step_time": 10.703383207321167
    },
    {
      "epoch": 3.7250518798828126e-06,
      "model_forward_time": 1.7002055644989014,
      "step": 3906
    },
    {
      "epoch": 3.7250518798828126e-06,
      "step": 3906,
      "training_step_time": 7.910017728805542
    },
    {
      "epoch": 3.7260055541992188e-06,
      "model_forward_time": 1.6962063312530518,
      "step": 3907
    },
    {
      "epoch": 3.7260055541992188e-06,
      "step": 3907,
      "training_step_time": 7.7894287109375
    },
    {
      "epoch": 3.726959228515625e-06,
      "model_forward_time": 1.6992230415344238,
      "step": 3908
    },
    {
      "epoch": 3.726959228515625e-06,
      "step": 3908,
      "training_step_time": 7.950376749038696
    },
    {
      "epoch": 3.727912902832031e-06,
      "model_forward_time": 1.7029697895050049,
      "step": 3909
    },
    {
      "epoch": 3.727912902832031e-06,
      "step": 3909,
      "training_step_time": 7.299394130706787
    },
    {
      "epoch": 3.7288665771484376e-06,
      "grad_norm": 0.2969738841056824,
      "learning_rate": 7.818e-06,
      "loss": 0.0802,
      "step": 3910
    },
    {
      "epoch": 3.7288665771484376e-06,
      "model_forward_time": 1.7005228996276855,
      "step": 3910
    },
    {
      "dynamics_loss_avg": 0.05865680631250143,
      "epoch": 3.7288665771484376e-06,
      "step": 3910
    },
    {
      "action_loss_avg": 0.008111032471060754,
      "epoch": 3.7288665771484376e-06,
      "step": 3910
    },
    {
      "epoch": 3.7288665771484376e-06,
      "step": 3910,
      "training_step_time": 6.882179498672485
    },
    {
      "epoch": 3.7298202514648438e-06,
      "model_forward_time": 1.7053651809692383,
      "step": 3911
    },
    {
      "epoch": 3.7298202514648438e-06,
      "step": 3911,
      "training_step_time": 6.860804319381714
    },
    {
      "epoch": 3.73077392578125e-06,
      "model_forward_time": 1.7065834999084473,
      "step": 3912
    },
    {
      "epoch": 3.73077392578125e-06,
      "step": 3912,
      "training_step_time": 7.664060592651367
    },
    {
      "epoch": 3.731727600097656e-06,
      "model_forward_time": 1.7034590244293213,
      "step": 3913
    },
    {
      "epoch": 3.731727600097656e-06,
      "step": 3913,
      "training_step_time": 7.766787528991699
    },
    {
      "epoch": 3.7326812744140626e-06,
      "model_forward_time": 2.2607572078704834,
      "step": 3914
    },
    {
      "epoch": 3.7326812744140626e-06,
      "step": 3914,
      "training_step_time": 9.808279275894165
    },
    {
      "epoch": 3.7336349487304688e-06,
      "model_forward_time": 1.7060585021972656,
      "step": 3915
    },
    {
      "epoch": 3.7336349487304688e-06,
      "step": 3915,
      "training_step_time": 8.055073738098145
    },
    {
      "epoch": 3.734588623046875e-06,
      "model_forward_time": 1.7030181884765625,
      "step": 3916
    },
    {
      "epoch": 3.734588623046875e-06,
      "step": 3916,
      "training_step_time": 8.510530233383179
    },
    {
      "epoch": 3.735542297363281e-06,
      "model_forward_time": 1.709597110748291,
      "step": 3917
    },
    {
      "epoch": 3.735542297363281e-06,
      "step": 3917,
      "training_step_time": 7.903265476226807
    },
    {
      "epoch": 3.7364959716796876e-06,
      "model_forward_time": 1.6947851181030273,
      "step": 3918
    },
    {
      "epoch": 3.7364959716796876e-06,
      "step": 3918,
      "training_step_time": 7.851867198944092
    },
    {
      "epoch": 3.7374496459960938e-06,
      "model_forward_time": 1.705308437347412,
      "step": 3919
    },
    {
      "epoch": 3.7374496459960938e-06,
      "step": 3919,
      "training_step_time": 6.7641518115997314
    },
    {
      "epoch": 3.7384033203125e-06,
      "grad_norm": 0.36670368909835815,
      "learning_rate": 7.838000000000002e-06,
      "loss": 0.0817,
      "step": 3920
    },
    {
      "epoch": 3.7384033203125e-06,
      "model_forward_time": 1.6987652778625488,
      "step": 3920
    },
    {
      "dynamics_loss_avg": 0.06374284848570824,
      "epoch": 3.7384033203125e-06,
      "step": 3920
    },
    {
      "action_loss_avg": 0.0075236099306494,
      "epoch": 3.7384033203125e-06,
      "step": 3920
    },
    {
      "epoch": 3.7384033203125e-06,
      "step": 3920,
      "training_step_time": 6.8911402225494385
    },
    {
      "epoch": 3.739356994628906e-06,
      "model_forward_time": 1.698519229888916,
      "step": 3921
    },
    {
      "epoch": 3.739356994628906e-06,
      "step": 3921,
      "training_step_time": 7.703741073608398
    },
    {
      "epoch": 3.7403106689453126e-06,
      "model_forward_time": 1.7041399478912354,
      "step": 3922
    },
    {
      "epoch": 3.7403106689453126e-06,
      "step": 3922,
      "training_step_time": 8.778222560882568
    },
    {
      "epoch": 3.741264343261719e-06,
      "model_forward_time": 1.713533639907837,
      "step": 3923
    },
    {
      "epoch": 3.741264343261719e-06,
      "step": 3923,
      "training_step_time": 8.29975152015686
    },
    {
      "epoch": 3.742218017578125e-06,
      "model_forward_time": 1.701589584350586,
      "step": 3924
    },
    {
      "epoch": 3.742218017578125e-06,
      "step": 3924,
      "training_step_time": 11.094879150390625
    },
    {
      "epoch": 3.743171691894531e-06,
      "model_forward_time": 1.7036216259002686,
      "step": 3925
    },
    {
      "epoch": 3.743171691894531e-06,
      "step": 3925,
      "training_step_time": 7.7115747928619385
    },
    {
      "epoch": 3.7441253662109376e-06,
      "model_forward_time": 1.6965343952178955,
      "step": 3926
    },
    {
      "epoch": 3.7441253662109376e-06,
      "step": 3926,
      "training_step_time": 8.55811071395874
    },
    {
      "epoch": 3.745079040527344e-06,
      "model_forward_time": 1.6944684982299805,
      "step": 3927
    },
    {
      "epoch": 3.745079040527344e-06,
      "step": 3927,
      "training_step_time": 7.743468999862671
    },
    {
      "epoch": 3.74603271484375e-06,
      "model_forward_time": 1.6973273754119873,
      "step": 3928
    },
    {
      "epoch": 3.74603271484375e-06,
      "step": 3928,
      "training_step_time": 7.795341968536377
    },
    {
      "epoch": 3.746986389160156e-06,
      "model_forward_time": 1.7056846618652344,
      "step": 3929
    },
    {
      "epoch": 3.746986389160156e-06,
      "step": 3929,
      "training_step_time": 6.8842833042144775
    },
    {
      "epoch": 3.7479400634765627e-06,
      "grad_norm": 0.3557077944278717,
      "learning_rate": 7.858000000000002e-06,
      "loss": 0.0825,
      "step": 3930
    },
    {
      "epoch": 3.7479400634765627e-06,
      "model_forward_time": 1.695507526397705,
      "step": 3930
    },
    {
      "dynamics_loss_avg": 0.06263492591679096,
      "epoch": 3.7479400634765627e-06,
      "step": 3930
    },
    {
      "action_loss_avg": 0.0065383943496271964,
      "epoch": 3.7479400634765627e-06,
      "step": 3930
    },
    {
      "epoch": 3.7479400634765627e-06,
      "step": 3930,
      "training_step_time": 7.7271034717559814
    },
    {
      "epoch": 3.748893737792969e-06,
      "model_forward_time": 1.693011999130249,
      "step": 3931
    },
    {
      "epoch": 3.748893737792969e-06,
      "step": 3931,
      "training_step_time": 7.837344646453857
    },
    {
      "epoch": 3.749847412109375e-06,
      "model_forward_time": 1.7025346755981445,
      "step": 3932
    },
    {
      "epoch": 3.749847412109375e-06,
      "step": 3932,
      "training_step_time": 7.874413251876831
    },
    {
      "epoch": 3.750801086425781e-06,
      "model_forward_time": 2.248680830001831,
      "step": 3933
    },
    {
      "epoch": 3.750801086425781e-06,
      "step": 3933,
      "training_step_time": 9.722063302993774
    },
    {
      "epoch": 3.7517547607421877e-06,
      "model_forward_time": 1.69696044921875,
      "step": 3934
    },
    {
      "epoch": 3.7517547607421877e-06,
      "step": 3934,
      "training_step_time": 9.665566205978394
    },
    {
      "epoch": 3.752708435058594e-06,
      "model_forward_time": 1.6985247135162354,
      "step": 3935
    },
    {
      "epoch": 3.752708435058594e-06,
      "step": 3935,
      "training_step_time": 9.391477346420288
    },
    {
      "epoch": 3.753662109375e-06,
      "model_forward_time": 1.6974174976348877,
      "step": 3936
    },
    {
      "epoch": 3.753662109375e-06,
      "step": 3936,
      "training_step_time": 8.009545803070068
    },
    {
      "epoch": 3.754615783691406e-06,
      "model_forward_time": 1.6943132877349854,
      "step": 3937
    },
    {
      "epoch": 3.754615783691406e-06,
      "step": 3937,
      "training_step_time": 7.3851318359375
    },
    {
      "epoch": 3.7555694580078127e-06,
      "model_forward_time": 1.701655387878418,
      "step": 3938
    },
    {
      "epoch": 3.7555694580078127e-06,
      "step": 3938,
      "training_step_time": 6.8674163818359375
    },
    {
      "epoch": 3.756523132324219e-06,
      "model_forward_time": 1.002105951309204,
      "step": 3939
    },
    {
      "epoch": 3.756523132324219e-06,
      "step": 3939,
      "training_step_time": 7.7384560108184814
    },
    {
      "epoch": 3.757476806640625e-06,
      "grad_norm": 0.41617581248283386,
      "learning_rate": 7.878e-06,
      "loss": 0.0795,
      "step": 3940
    },
    {
      "epoch": 3.757476806640625e-06,
      "model_forward_time": 1.702725887298584,
      "step": 3940
    },
    {
      "dynamics_loss_avg": 0.054443178977817294,
      "epoch": 3.757476806640625e-06,
      "step": 3940
    },
    {
      "action_loss_avg": 0.0078951460076496,
      "epoch": 3.757476806640625e-06,
      "step": 3940
    },
    {
      "epoch": 3.757476806640625e-06,
      "step": 3940,
      "training_step_time": 7.878393650054932
    },
    {
      "epoch": 3.758430480957031e-06,
      "model_forward_time": 1.705249309539795,
      "step": 3941
    },
    {
      "epoch": 3.758430480957031e-06,
      "step": 3941,
      "training_step_time": 8.755555629730225
    },
    {
      "epoch": 3.7593841552734377e-06,
      "model_forward_time": 2.2288196086883545,
      "step": 3942
    },
    {
      "epoch": 3.7593841552734377e-06,
      "step": 3942,
      "training_step_time": 8.988962888717651
    },
    {
      "epoch": 3.760337829589844e-06,
      "model_forward_time": 1.702185869216919,
      "step": 3943
    },
    {
      "epoch": 3.760337829589844e-06,
      "step": 3943,
      "training_step_time": 7.755190372467041
    },
    {
      "epoch": 3.76129150390625e-06,
      "model_forward_time": 1.700526475906372,
      "step": 3944
    },
    {
      "epoch": 3.76129150390625e-06,
      "step": 3944,
      "training_step_time": 8.289451122283936
    },
    {
      "epoch": 3.762245178222656e-06,
      "model_forward_time": 1.6995501518249512,
      "step": 3945
    },
    {
      "epoch": 3.762245178222656e-06,
      "step": 3945,
      "training_step_time": 7.838370084762573
    },
    {
      "epoch": 3.7631988525390627e-06,
      "model_forward_time": 1.7016143798828125,
      "step": 3946
    },
    {
      "epoch": 3.7631988525390627e-06,
      "step": 3946,
      "training_step_time": 7.811323642730713
    },
    {
      "epoch": 3.764152526855469e-06,
      "model_forward_time": 1.721472978591919,
      "step": 3947
    },
    {
      "epoch": 3.764152526855469e-06,
      "step": 3947,
      "training_step_time": 6.911561965942383
    },
    {
      "epoch": 3.765106201171875e-06,
      "model_forward_time": 1.6997613906860352,
      "step": 3948
    },
    {
      "epoch": 3.765106201171875e-06,
      "step": 3948,
      "training_step_time": 7.65102744102478
    },
    {
      "epoch": 3.766059875488281e-06,
      "model_forward_time": 1.7009902000427246,
      "step": 3949
    },
    {
      "epoch": 3.766059875488281e-06,
      "step": 3949,
      "training_step_time": 7.794898986816406
    },
    {
      "epoch": 3.7670135498046877e-06,
      "grad_norm": 0.6001616716384888,
      "learning_rate": 7.898e-06,
      "loss": 0.0756,
      "step": 3950
    },
    {
      "epoch": 3.7670135498046877e-06,
      "model_forward_time": 1.702824354171753,
      "step": 3950
    },
    {
      "dynamics_loss_avg": 0.0668702121824026,
      "epoch": 3.7670135498046877e-06,
      "step": 3950
    },
    {
      "action_loss_avg": 0.011959259887225926,
      "epoch": 3.7670135498046877e-06,
      "step": 3950
    },
    {
      "epoch": 3.7670135498046877e-06,
      "step": 3950,
      "training_step_time": 8.317447423934937
    },
    {
      "epoch": 3.767967224121094e-06,
      "model_forward_time": 2.245121717453003,
      "step": 3951
    },
    {
      "epoch": 3.767967224121094e-06,
      "step": 3951,
      "training_step_time": 8.93828296661377
    },
    {
      "epoch": 3.7689208984375e-06,
      "model_forward_time": 1.6990771293640137,
      "step": 3952
    },
    {
      "epoch": 3.7689208984375e-06,
      "step": 3952,
      "training_step_time": 8.601695775985718
    },
    {
      "epoch": 3.769874572753906e-06,
      "model_forward_time": 1.700380802154541,
      "step": 3953
    },
    {
      "epoch": 3.769874572753906e-06,
      "step": 3953,
      "training_step_time": 7.759730577468872
    },
    {
      "epoch": 3.7708282470703127e-06,
      "model_forward_time": 1.7008349895477295,
      "step": 3954
    },
    {
      "epoch": 3.7708282470703127e-06,
      "step": 3954,
      "training_step_time": 8.56066370010376
    },
    {
      "epoch": 3.771781921386719e-06,
      "model_forward_time": 1.7025001049041748,
      "step": 3955
    },
    {
      "epoch": 3.771781921386719e-06,
      "step": 3955,
      "training_step_time": 7.848565101623535
    },
    {
      "epoch": 3.772735595703125e-06,
      "model_forward_time": 1.6973628997802734,
      "step": 3956
    },
    {
      "epoch": 3.772735595703125e-06,
      "step": 3956,
      "training_step_time": 7.523543119430542
    },
    {
      "epoch": 3.773689270019531e-06,
      "model_forward_time": 1.6985177993774414,
      "step": 3957
    },
    {
      "epoch": 3.773689270019531e-06,
      "step": 3957,
      "training_step_time": 7.659363269805908
    },
    {
      "epoch": 3.7746429443359377e-06,
      "model_forward_time": 1.6985735893249512,
      "step": 3958
    },
    {
      "epoch": 3.7746429443359377e-06,
      "step": 3958,
      "training_step_time": 7.767143249511719
    },
    {
      "epoch": 3.775596618652344e-06,
      "model_forward_time": 1.7092797756195068,
      "step": 3959
    },
    {
      "epoch": 3.775596618652344e-06,
      "step": 3959,
      "training_step_time": 7.75165581703186
    },
    {
      "epoch": 3.77655029296875e-06,
      "grad_norm": 0.44839298725128174,
      "learning_rate": 7.918e-06,
      "loss": 0.0775,
      "step": 3960
    },
    {
      "epoch": 3.77655029296875e-06,
      "model_forward_time": 1.7106833457946777,
      "step": 3960
    },
    {
      "dynamics_loss_avg": 0.05263620484620333,
      "epoch": 3.77655029296875e-06,
      "step": 3960
    },
    {
      "action_loss_avg": 0.007060374994762242,
      "epoch": 3.77655029296875e-06,
      "step": 3960
    },
    {
      "epoch": 3.77655029296875e-06,
      "step": 3960,
      "training_step_time": 7.92311429977417
    },
    {
      "epoch": 3.777503967285156e-06,
      "model_forward_time": 2.231956958770752,
      "step": 3961
    },
    {
      "epoch": 3.777503967285156e-06,
      "step": 3961,
      "training_step_time": 9.771438837051392
    },
    {
      "epoch": 3.7784576416015627e-06,
      "model_forward_time": 1.70597505569458,
      "step": 3962
    },
    {
      "epoch": 3.7784576416015627e-06,
      "step": 3962,
      "training_step_time": 7.74172568321228
    },
    {
      "epoch": 3.779411315917969e-06,
      "model_forward_time": 1.7012827396392822,
      "step": 3963
    },
    {
      "epoch": 3.779411315917969e-06,
      "step": 3963,
      "training_step_time": 7.733329772949219
    },
    {
      "epoch": 3.780364990234375e-06,
      "model_forward_time": 1.7000420093536377,
      "step": 3964
    },
    {
      "epoch": 3.780364990234375e-06,
      "step": 3964,
      "training_step_time": 7.935141086578369
    },
    {
      "epoch": 3.781318664550781e-06,
      "model_forward_time": 1.7081694602966309,
      "step": 3965
    },
    {
      "epoch": 3.781318664550781e-06,
      "step": 3965,
      "training_step_time": 7.7974865436553955
    },
    {
      "epoch": 3.7822723388671877e-06,
      "model_forward_time": 1.6993179321289062,
      "step": 3966
    },
    {
      "epoch": 3.7822723388671877e-06,
      "step": 3966,
      "training_step_time": 7.68505072593689
    },
    {
      "epoch": 3.783226013183594e-06,
      "model_forward_time": 1.7069740295410156,
      "step": 3967
    },
    {
      "epoch": 3.783226013183594e-06,
      "step": 3967,
      "training_step_time": 7.667966365814209
    },
    {
      "epoch": 3.7841796875e-06,
      "model_forward_time": 1.7099401950836182,
      "step": 3968
    },
    {
      "epoch": 3.7841796875e-06,
      "step": 3968,
      "training_step_time": 8.46536922454834
    },
    {
      "epoch": 3.785133361816406e-06,
      "model_forward_time": 1.708261251449585,
      "step": 3969
    },
    {
      "epoch": 3.785133361816406e-06,
      "step": 3969,
      "training_step_time": 7.80384087562561
    },
    {
      "epoch": 3.7860870361328127e-06,
      "grad_norm": 0.4715409576892853,
      "learning_rate": 7.938000000000001e-06,
      "loss": 0.0823,
      "step": 3970
    },
    {
      "epoch": 3.7860870361328127e-06,
      "model_forward_time": 2.2401516437530518,
      "step": 3970
    },
    {
      "dynamics_loss_avg": 0.07272825390100479,
      "epoch": 3.7860870361328127e-06,
      "step": 3970
    },
    {
      "action_loss_avg": 0.00796726723201573,
      "epoch": 3.7860870361328127e-06,
      "step": 3970
    },
    {
      "epoch": 3.7860870361328127e-06,
      "step": 3970,
      "training_step_time": 7.9304914474487305
    },
    {
      "epoch": 3.787040710449219e-06,
      "model_forward_time": 1.7042500972747803,
      "step": 3971
    },
    {
      "epoch": 3.787040710449219e-06,
      "step": 3971,
      "training_step_time": 7.683601379394531
    },
    {
      "epoch": 3.787994384765625e-06,
      "model_forward_time": 1.6952097415924072,
      "step": 3972
    },
    {
      "epoch": 3.787994384765625e-06,
      "step": 3972,
      "training_step_time": 8.883159637451172
    },
    {
      "epoch": 3.788948059082031e-06,
      "model_forward_time": 1.7004215717315674,
      "step": 3973
    },
    {
      "epoch": 3.788948059082031e-06,
      "step": 3973,
      "training_step_time": 7.931782484054565
    },
    {
      "epoch": 3.7899017333984377e-06,
      "model_forward_time": 1.6943483352661133,
      "step": 3974
    },
    {
      "epoch": 3.7899017333984377e-06,
      "step": 3974,
      "training_step_time": 7.782222747802734
    },
    {
      "epoch": 3.790855407714844e-06,
      "model_forward_time": 1.699892282485962,
      "step": 3975
    },
    {
      "epoch": 3.790855407714844e-06,
      "step": 3975,
      "training_step_time": 7.82922101020813
    },
    {
      "epoch": 3.79180908203125e-06,
      "model_forward_time": 1.7005441188812256,
      "step": 3976
    },
    {
      "epoch": 3.79180908203125e-06,
      "step": 3976,
      "training_step_time": 7.554685592651367
    },
    {
      "epoch": 3.792762756347656e-06,
      "model_forward_time": 1.6904652118682861,
      "step": 3977
    },
    {
      "epoch": 3.792762756347656e-06,
      "step": 3977,
      "training_step_time": 7.736686706542969
    },
    {
      "epoch": 3.7937164306640627e-06,
      "model_forward_time": 1.7014052867889404,
      "step": 3978
    },
    {
      "epoch": 3.7937164306640627e-06,
      "step": 3978,
      "training_step_time": 8.636234760284424
    },
    {
      "epoch": 3.794670104980469e-06,
      "model_forward_time": 2.299962043762207,
      "step": 3979
    },
    {
      "epoch": 3.794670104980469e-06,
      "step": 3979,
      "training_step_time": 9.099959135055542
    },
    {
      "epoch": 3.795623779296875e-06,
      "grad_norm": 0.4145854413509369,
      "learning_rate": 7.958e-06,
      "loss": 0.0861,
      "step": 3980
    },
    {
      "epoch": 3.795623779296875e-06,
      "model_forward_time": 1.69840669631958,
      "step": 3980
    },
    {
      "dynamics_loss_avg": 0.08707502968609333,
      "epoch": 3.795623779296875e-06,
      "step": 3980
    },
    {
      "action_loss_avg": 0.011599013302475213,
      "epoch": 3.795623779296875e-06,
      "step": 3980
    },
    {
      "epoch": 3.795623779296875e-06,
      "step": 3980,
      "training_step_time": 8.541266202926636
    },
    {
      "epoch": 3.796577453613281e-06,
      "model_forward_time": 1.698160171508789,
      "step": 3981
    },
    {
      "epoch": 3.796577453613281e-06,
      "step": 3981,
      "training_step_time": 7.812636613845825
    },
    {
      "epoch": 3.7975311279296877e-06,
      "model_forward_time": 1.7014222145080566,
      "step": 3982
    },
    {
      "epoch": 3.7975311279296877e-06,
      "step": 3982,
      "training_step_time": 7.854687690734863
    },
    {
      "epoch": 3.798484802246094e-06,
      "model_forward_time": 1.6976768970489502,
      "step": 3983
    },
    {
      "epoch": 3.798484802246094e-06,
      "step": 3983,
      "training_step_time": 7.823651552200317
    },
    {
      "epoch": 3.7994384765625e-06,
      "model_forward_time": 1.7024257183074951,
      "step": 3984
    },
    {
      "epoch": 3.7994384765625e-06,
      "step": 3984,
      "training_step_time": 7.705144882202148
    },
    {
      "epoch": 3.800392150878906e-06,
      "model_forward_time": 1.6950223445892334,
      "step": 3985
    },
    {
      "epoch": 3.800392150878906e-06,
      "step": 3985,
      "training_step_time": 7.688713550567627
    },
    {
      "epoch": 3.8013458251953127e-06,
      "model_forward_time": 1.6972877979278564,
      "step": 3986
    },
    {
      "epoch": 3.8013458251953127e-06,
      "step": 3986,
      "training_step_time": 7.7306506633758545
    },
    {
      "epoch": 3.802299499511719e-06,
      "model_forward_time": 1.7023377418518066,
      "step": 3987
    },
    {
      "epoch": 3.802299499511719e-06,
      "step": 3987,
      "training_step_time": 8.70075011253357
    },
    {
      "epoch": 3.803253173828125e-06,
      "model_forward_time": 2.2422657012939453,
      "step": 3988
    },
    {
      "epoch": 3.803253173828125e-06,
      "step": 3988,
      "training_step_time": 9.529807567596436
    },
    {
      "epoch": 3.804206848144531e-06,
      "model_forward_time": 1.6982142925262451,
      "step": 3989
    },
    {
      "epoch": 3.804206848144531e-06,
      "step": 3989,
      "training_step_time": 7.733962535858154
    },
    {
      "epoch": 3.8051605224609377e-06,
      "grad_norm": 0.5902666449546814,
      "learning_rate": 7.978e-06,
      "loss": 0.0792,
      "step": 3990
    },
    {
      "epoch": 3.8051605224609377e-06,
      "model_forward_time": 1.6913366317749023,
      "step": 3990
    },
    {
      "dynamics_loss_avg": 0.04827895723283291,
      "epoch": 3.8051605224609377e-06,
      "step": 3990
    },
    {
      "action_loss_avg": 0.007004457281436771,
      "epoch": 3.8051605224609377e-06,
      "step": 3990
    },
    {
      "epoch": 3.8051605224609377e-06,
      "step": 3990,
      "training_step_time": 7.700275182723999
    },
    {
      "epoch": 3.806114196777344e-06,
      "model_forward_time": 1.701336145401001,
      "step": 3991
    },
    {
      "epoch": 3.806114196777344e-06,
      "step": 3991,
      "training_step_time": 8.675254344940186
    },
    {
      "epoch": 3.80706787109375e-06,
      "model_forward_time": 1.692258358001709,
      "step": 3992
    },
    {
      "epoch": 3.80706787109375e-06,
      "step": 3992,
      "training_step_time": 7.717987775802612
    },
    {
      "epoch": 3.808021545410156e-06,
      "model_forward_time": 1.7043516635894775,
      "step": 3993
    },
    {
      "epoch": 3.808021545410156e-06,
      "step": 3993,
      "training_step_time": 7.531984806060791
    },
    {
      "epoch": 3.8089752197265627e-06,
      "model_forward_time": 0.9840481281280518,
      "step": 3994
    },
    {
      "epoch": 3.8089752197265627e-06,
      "step": 3994,
      "training_step_time": 7.622970104217529
    },
    {
      "epoch": 3.809928894042969e-06,
      "model_forward_time": 1.703157663345337,
      "step": 3995
    },
    {
      "epoch": 3.809928894042969e-06,
      "step": 3995,
      "training_step_time": 7.643033981323242
    },
    {
      "epoch": 3.810882568359375e-06,
      "model_forward_time": 1.7048125267028809,
      "step": 3996
    },
    {
      "epoch": 3.810882568359375e-06,
      "step": 3996,
      "training_step_time": 8.176459074020386
    },
    {
      "epoch": 3.811836242675781e-06,
      "model_forward_time": 1.737565040588379,
      "step": 3997
    },
    {
      "epoch": 3.811836242675781e-06,
      "step": 3997,
      "training_step_time": 9.645676136016846
    },
    {
      "epoch": 3.8127899169921877e-06,
      "model_forward_time": 2.310786247253418,
      "step": 3998
    },
    {
      "epoch": 3.8127899169921877e-06,
      "step": 3998,
      "training_step_time": 10.204851865768433
    },
    {
      "epoch": 3.813743591308594e-06,
      "model_forward_time": 1.6946113109588623,
      "step": 3999
    },
    {
      "epoch": 3.813743591308594e-06,
      "step": 3999,
      "training_step_time": 7.6864588260650635
    },
    {
      "epoch": 3.814697265625e-06,
      "grad_norm": 0.33807137608528137,
      "learning_rate": 7.998e-06,
      "loss": 0.0845,
      "step": 4000
    },
    {
      "epoch": 3.814697265625e-06,
      "model_forward_time": 1.721569299697876,
      "step": 4000
    },
    {
      "dynamics_loss_avg": 0.05728658102452755,
      "epoch": 3.814697265625e-06,
      "step": 4000
    },
    {
      "action_loss_avg": 0.008648838684894144,
      "epoch": 3.814697265625e-06,
      "step": 4000
    },
    {
      "epoch": 3.814697265625e-06,
      "step": 4000,
      "training_step_time": 7.77866005897522
    },
    {
      "epoch": 3.8156509399414066e-06,
      "model_forward_time": 1.7651219367980957,
      "step": 4001
    },
    {
      "epoch": 3.8156509399414066e-06,
      "step": 4001,
      "training_step_time": 8.785697221755981
    },
    {
      "epoch": 3.816604614257812e-06,
      "model_forward_time": 1.784153938293457,
      "step": 4002
    },
    {
      "epoch": 3.816604614257812e-06,
      "step": 4002,
      "training_step_time": 7.912251234054565
    },
    {
      "epoch": 3.817558288574219e-06,
      "model_forward_time": 1.7446296215057373,
      "step": 4003
    },
    {
      "epoch": 3.817558288574219e-06,
      "step": 4003,
      "training_step_time": 7.741229295730591
    },
    {
      "epoch": 3.818511962890625e-06,
      "model_forward_time": 1.3673546314239502,
      "step": 4004
    },
    {
      "epoch": 3.818511962890625e-06,
      "step": 4004,
      "training_step_time": 8.950706481933594
    },
    {
      "epoch": 3.819465637207031e-06,
      "model_forward_time": 1.7470333576202393,
      "step": 4005
    },
    {
      "epoch": 3.819465637207031e-06,
      "step": 4005,
      "training_step_time": 8.737301349639893
    },
    {
      "epoch": 3.820419311523438e-06,
      "model_forward_time": 2.724491596221924,
      "step": 4006
    },
    {
      "epoch": 3.820419311523438e-06,
      "step": 4006,
      "training_step_time": 14.369249105453491
    },
    {
      "epoch": 3.8213729858398434e-06,
      "model_forward_time": 1.782015323638916,
      "step": 4007
    },
    {
      "epoch": 3.8213729858398434e-06,
      "step": 4007,
      "training_step_time": 9.542353868484497
    },
    {
      "epoch": 3.82232666015625e-06,
      "model_forward_time": 1.7402362823486328,
      "step": 4008
    },
    {
      "epoch": 3.82232666015625e-06,
      "step": 4008,
      "training_step_time": 7.68706750869751
    },
    {
      "epoch": 3.8232803344726566e-06,
      "model_forward_time": 1.0288472175598145,
      "step": 4009
    },
    {
      "epoch": 3.8232803344726566e-06,
      "step": 4009,
      "training_step_time": 7.679608345031738
    },
    {
      "epoch": 3.824234008789062e-06,
      "grad_norm": 0.35937169194221497,
      "learning_rate": 8.018e-06,
      "loss": 0.0861,
      "step": 4010
    },
    {
      "epoch": 3.824234008789062e-06,
      "model_forward_time": 1.7327044010162354,
      "step": 4010
    },
    {
      "dynamics_loss_avg": 0.08353890739381313,
      "epoch": 3.824234008789062e-06,
      "step": 4010
    },
    {
      "action_loss_avg": 0.010080888355150818,
      "epoch": 3.824234008789062e-06,
      "step": 4010
    },
    {
      "epoch": 3.824234008789062e-06,
      "step": 4010,
      "training_step_time": 8.801584005355835
    },
    {
      "epoch": 3.825187683105469e-06,
      "model_forward_time": 1.7344913482666016,
      "step": 4011
    },
    {
      "epoch": 3.825187683105469e-06,
      "step": 4011,
      "training_step_time": 7.797706842422485
    },
    {
      "epoch": 3.826141357421875e-06,
      "model_forward_time": 1.7368812561035156,
      "step": 4012
    },
    {
      "epoch": 3.826141357421875e-06,
      "step": 4012,
      "training_step_time": 7.713318347930908
    },
    {
      "epoch": 3.827095031738281e-06,
      "model_forward_time": 1.7448046207427979,
      "step": 4013
    },
    {
      "epoch": 3.827095031738281e-06,
      "step": 4013,
      "training_step_time": 7.503305196762085
    },
    {
      "epoch": 3.828048706054688e-06,
      "model_forward_time": 1.732541799545288,
      "step": 4014
    },
    {
      "epoch": 3.828048706054688e-06,
      "step": 4014,
      "training_step_time": 7.62397837638855
    },
    {
      "epoch": 3.8290023803710934e-06,
      "model_forward_time": 1.7376530170440674,
      "step": 4015
    },
    {
      "epoch": 3.8290023803710934e-06,
      "step": 4015,
      "training_step_time": 10.600388050079346
    },
    {
      "epoch": 3.8299560546875e-06,
      "model_forward_time": 2.3006536960601807,
      "step": 4016
    },
    {
      "epoch": 3.8299560546875e-06,
      "step": 4016,
      "training_step_time": 8.519391775131226
    },
    {
      "epoch": 3.830909729003907e-06,
      "model_forward_time": 1.730433702468872,
      "step": 4017
    },
    {
      "epoch": 3.830909729003907e-06,
      "step": 4017,
      "training_step_time": 9.229449033737183
    },
    {
      "epoch": 3.831863403320312e-06,
      "model_forward_time": 1.738980770111084,
      "step": 4018
    },
    {
      "epoch": 3.831863403320312e-06,
      "step": 4018,
      "training_step_time": 7.590046644210815
    },
    {
      "epoch": 3.832817077636719e-06,
      "model_forward_time": 1.7364256381988525,
      "step": 4019
    },
    {
      "epoch": 3.832817077636719e-06,
      "step": 4019,
      "training_step_time": 8.715484619140625
    },
    {
      "epoch": 3.833770751953125e-06,
      "grad_norm": 0.34873566031455994,
      "learning_rate": 8.038e-06,
      "loss": 0.082,
      "step": 4020
    },
    {
      "epoch": 3.833770751953125e-06,
      "model_forward_time": 1.3343737125396729,
      "step": 4020
    },
    {
      "dynamics_loss_avg": 0.07841092012822629,
      "epoch": 3.833770751953125e-06,
      "step": 4020
    },
    {
      "action_loss_avg": 0.009983652643859387,
      "epoch": 3.833770751953125e-06,
      "step": 4020
    },
    {
      "epoch": 3.833770751953125e-06,
      "step": 4020,
      "training_step_time": 7.648451566696167
    },
    {
      "epoch": 3.834724426269531e-06,
      "model_forward_time": 1.7262792587280273,
      "step": 4021
    },
    {
      "epoch": 3.834724426269531e-06,
      "step": 4021,
      "training_step_time": 7.546127080917358
    },
    {
      "epoch": 3.835678100585938e-06,
      "model_forward_time": 1.7153520584106445,
      "step": 4022
    },
    {
      "epoch": 3.835678100585938e-06,
      "step": 4022,
      "training_step_time": 7.476118564605713
    },
    {
      "epoch": 3.8366317749023435e-06,
      "model_forward_time": 1.7256145477294922,
      "step": 4023
    },
    {
      "epoch": 3.8366317749023435e-06,
      "step": 4023,
      "training_step_time": 7.571971654891968
    },
    {
      "epoch": 3.83758544921875e-06,
      "model_forward_time": 1.7311604022979736,
      "step": 4024
    },
    {
      "epoch": 3.83758544921875e-06,
      "step": 4024,
      "training_step_time": 8.46269416809082
    },
    {
      "epoch": 3.838539123535157e-06,
      "model_forward_time": 2.275176525115967,
      "step": 4025
    },
    {
      "epoch": 3.838539123535157e-06,
      "step": 4025,
      "training_step_time": 8.507535934448242
    },
    {
      "epoch": 3.839492797851562e-06,
      "model_forward_time": 1.725144863128662,
      "step": 4026
    },
    {
      "epoch": 3.839492797851562e-06,
      "step": 4026,
      "training_step_time": 8.304294109344482
    },
    {
      "epoch": 3.840446472167969e-06,
      "model_forward_time": 1.724776029586792,
      "step": 4027
    },
    {
      "epoch": 3.840446472167969e-06,
      "step": 4027,
      "training_step_time": 8.139603853225708
    },
    {
      "epoch": 3.841400146484375e-06,
      "model_forward_time": 1.7206315994262695,
      "step": 4028
    },
    {
      "epoch": 3.841400146484375e-06,
      "step": 4028,
      "training_step_time": 7.665637731552124
    },
    {
      "epoch": 3.842353820800781e-06,
      "model_forward_time": 1.7178733348846436,
      "step": 4029
    },
    {
      "epoch": 3.842353820800781e-06,
      "step": 4029,
      "training_step_time": 7.694242477416992
    },
    {
      "epoch": 3.843307495117188e-06,
      "grad_norm": 0.4305875301361084,
      "learning_rate": 8.058e-06,
      "loss": 0.0834,
      "step": 4030
    },
    {
      "epoch": 3.843307495117188e-06,
      "model_forward_time": 1.716994047164917,
      "step": 4030
    },
    {
      "dynamics_loss_avg": 0.07700981795787812,
      "epoch": 3.843307495117188e-06,
      "step": 4030
    },
    {
      "action_loss_avg": 0.009624824998900295,
      "epoch": 3.843307495117188e-06,
      "step": 4030
    },
    {
      "epoch": 3.843307495117188e-06,
      "step": 4030,
      "training_step_time": 7.235868453979492
    },
    {
      "epoch": 3.8442611694335935e-06,
      "model_forward_time": 1.7170405387878418,
      "step": 4031
    },
    {
      "epoch": 3.8442611694335935e-06,
      "step": 4031,
      "training_step_time": 8.260817050933838
    },
    {
      "epoch": 3.84521484375e-06,
      "model_forward_time": 1.720954418182373,
      "step": 4032
    },
    {
      "epoch": 3.84521484375e-06,
      "step": 4032,
      "training_step_time": 7.558790683746338
    },
    {
      "epoch": 3.846168518066407e-06,
      "model_forward_time": 1.347386121749878,
      "step": 4033
    },
    {
      "epoch": 3.846168518066407e-06,
      "step": 4033,
      "training_step_time": 8.349371194839478
    },
    {
      "epoch": 3.847122192382812e-06,
      "model_forward_time": 1.7650158405303955,
      "step": 4034
    },
    {
      "epoch": 3.847122192382812e-06,
      "step": 4034,
      "training_step_time": 9.130433797836304
    },
    {
      "epoch": 3.848075866699219e-06,
      "model_forward_time": 1.7319927215576172,
      "step": 4035
    },
    {
      "epoch": 3.848075866699219e-06,
      "step": 4035,
      "training_step_time": 8.6308012008667
    },
    {
      "epoch": 3.849029541015625e-06,
      "model_forward_time": 1.7291951179504395,
      "step": 4036
    },
    {
      "epoch": 3.849029541015625e-06,
      "step": 4036,
      "training_step_time": 7.739121675491333
    },
    {
      "epoch": 3.849983215332031e-06,
      "model_forward_time": 1.715787410736084,
      "step": 4037
    },
    {
      "epoch": 3.849983215332031e-06,
      "step": 4037,
      "training_step_time": 7.665705442428589
    },
    {
      "epoch": 3.850936889648438e-06,
      "model_forward_time": 1.7167820930480957,
      "step": 4038
    },
    {
      "epoch": 3.850936889648438e-06,
      "step": 4038,
      "training_step_time": 7.7187323570251465
    },
    {
      "epoch": 3.8518905639648435e-06,
      "model_forward_time": 1.7410986423492432,
      "step": 4039
    },
    {
      "epoch": 3.8518905639648435e-06,
      "step": 4039,
      "training_step_time": 7.1062726974487305
    },
    {
      "epoch": 3.85284423828125e-06,
      "grad_norm": 0.3177298903465271,
      "learning_rate": 8.078e-06,
      "loss": 0.079,
      "step": 4040
    },
    {
      "epoch": 3.85284423828125e-06,
      "model_forward_time": 1.734926462173462,
      "step": 4040
    },
    {
      "dynamics_loss_avg": 0.07704833634197712,
      "epoch": 3.85284423828125e-06,
      "step": 4040
    },
    {
      "action_loss_avg": 0.0117072272580117,
      "epoch": 3.85284423828125e-06,
      "step": 4040
    },
    {
      "epoch": 3.85284423828125e-06,
      "step": 4040,
      "training_step_time": 7.580004930496216
    },
    {
      "epoch": 3.853797912597657e-06,
      "model_forward_time": 1.72487473487854,
      "step": 4041
    },
    {
      "epoch": 3.853797912597657e-06,
      "step": 4041,
      "training_step_time": 8.563354015350342
    },
    {
      "epoch": 3.854751586914062e-06,
      "model_forward_time": 1.7248585224151611,
      "step": 4042
    },
    {
      "epoch": 3.854751586914062e-06,
      "step": 4042,
      "training_step_time": 8.883896827697754
    },
    {
      "epoch": 3.855705261230469e-06,
      "model_forward_time": 1.7422411441802979,
      "step": 4043
    },
    {
      "epoch": 3.855705261230469e-06,
      "step": 4043,
      "training_step_time": 7.625934600830078
    },
    {
      "epoch": 3.856658935546875e-06,
      "model_forward_time": 2.8384485244750977,
      "step": 4044
    },
    {
      "epoch": 3.856658935546875e-06,
      "step": 4044,
      "training_step_time": 9.474733352661133
    },
    {
      "epoch": 3.857612609863281e-06,
      "model_forward_time": 1.356018304824829,
      "step": 4045
    },
    {
      "epoch": 3.857612609863281e-06,
      "step": 4045,
      "training_step_time": 8.727745056152344
    },
    {
      "epoch": 3.858566284179688e-06,
      "model_forward_time": 1.7269818782806396,
      "step": 4046
    },
    {
      "epoch": 3.858566284179688e-06,
      "step": 4046,
      "training_step_time": 7.6289591789245605
    },
    {
      "epoch": 3.8595199584960935e-06,
      "model_forward_time": 1.7182774543762207,
      "step": 4047
    },
    {
      "epoch": 3.8595199584960935e-06,
      "step": 4047,
      "training_step_time": 7.79410457611084
    },
    {
      "epoch": 3.8604736328125e-06,
      "model_forward_time": 1.734739065170288,
      "step": 4048
    },
    {
      "epoch": 3.8604736328125e-06,
      "step": 4048,
      "training_step_time": 7.637522459030151
    },
    {
      "epoch": 3.861427307128907e-06,
      "model_forward_time": 1.7193517684936523,
      "step": 4049
    },
    {
      "epoch": 3.861427307128907e-06,
      "step": 4049,
      "training_step_time": 7.54715633392334
    },
    {
      "epoch": 3.862380981445312e-06,
      "grad_norm": 0.497057169675827,
      "learning_rate": 8.098000000000001e-06,
      "loss": 0.0835,
      "step": 4050
    },
    {
      "epoch": 3.862380981445312e-06,
      "model_forward_time": 1.3373310565948486,
      "step": 4050
    },
    {
      "dynamics_loss_avg": 0.08330181650817395,
      "epoch": 3.862380981445312e-06,
      "step": 4050
    },
    {
      "action_loss_avg": 0.010611152928322553,
      "epoch": 3.862380981445312e-06,
      "step": 4050
    },
    {
      "epoch": 3.862380981445312e-06,
      "step": 4050,
      "training_step_time": 8.647511005401611
    },
    {
      "epoch": 3.863334655761719e-06,
      "model_forward_time": 1.7315211296081543,
      "step": 4051
    },
    {
      "epoch": 3.863334655761719e-06,
      "step": 4051,
      "training_step_time": 7.565186977386475
    },
    {
      "epoch": 3.864288330078125e-06,
      "model_forward_time": 1.3619391918182373,
      "step": 4052
    },
    {
      "epoch": 3.864288330078125e-06,
      "step": 4052,
      "training_step_time": 8.503430128097534
    },
    {
      "epoch": 3.865242004394531e-06,
      "model_forward_time": 2.862614870071411,
      "step": 4053
    },
    {
      "epoch": 3.865242004394531e-06,
      "step": 4053,
      "training_step_time": 7.763103246688843
    },
    {
      "epoch": 3.866195678710938e-06,
      "model_forward_time": 1.7324175834655762,
      "step": 4054
    },
    {
      "epoch": 3.866195678710938e-06,
      "step": 4054,
      "training_step_time": 7.688743829727173
    },
    {
      "epoch": 3.8671493530273435e-06,
      "model_forward_time": 1.731855869293213,
      "step": 4055
    },
    {
      "epoch": 3.8671493530273435e-06,
      "step": 4055,
      "training_step_time": 7.7201409339904785
    },
    {
      "epoch": 3.86810302734375e-06,
      "model_forward_time": 1.7305819988250732,
      "step": 4056
    },
    {
      "epoch": 3.86810302734375e-06,
      "step": 4056,
      "training_step_time": 8.66532278060913
    },
    {
      "epoch": 3.869056701660157e-06,
      "model_forward_time": 1.730971097946167,
      "step": 4057
    },
    {
      "epoch": 3.869056701660157e-06,
      "step": 4057,
      "training_step_time": 7.591692924499512
    },
    {
      "epoch": 3.870010375976562e-06,
      "model_forward_time": 1.7312209606170654,
      "step": 4058
    },
    {
      "epoch": 3.870010375976562e-06,
      "step": 4058,
      "training_step_time": 7.510183572769165
    },
    {
      "epoch": 3.870964050292969e-06,
      "model_forward_time": 1.733813762664795,
      "step": 4059
    },
    {
      "epoch": 3.870964050292969e-06,
      "step": 4059,
      "training_step_time": 7.555316925048828
    },
    {
      "epoch": 3.871917724609375e-06,
      "grad_norm": 0.4405576288700104,
      "learning_rate": 8.118000000000001e-06,
      "loss": 0.0834,
      "step": 4060
    },
    {
      "epoch": 3.871917724609375e-06,
      "model_forward_time": 1.7343201637268066,
      "step": 4060
    },
    {
      "dynamics_loss_avg": 0.07403013072907924,
      "epoch": 3.871917724609375e-06,
      "step": 4060
    },
    {
      "action_loss_avg": 0.014076989935711026,
      "epoch": 3.871917724609375e-06,
      "step": 4060
    },
    {
      "epoch": 3.871917724609375e-06,
      "step": 4060,
      "training_step_time": 8.383488416671753
    },
    {
      "epoch": 3.872871398925781e-06,
      "model_forward_time": 1.7382111549377441,
      "step": 4061
    },
    {
      "epoch": 3.872871398925781e-06,
      "step": 4061,
      "training_step_time": 8.553966283798218
    },
    {
      "epoch": 3.873825073242188e-06,
      "model_forward_time": 2.8370070457458496,
      "step": 4062
    },
    {
      "epoch": 3.873825073242188e-06,
      "step": 4062,
      "training_step_time": 8.741389274597168
    },
    {
      "epoch": 3.8747787475585935e-06,
      "model_forward_time": 1.7337279319763184,
      "step": 4063
    },
    {
      "epoch": 3.8747787475585935e-06,
      "step": 4063,
      "training_step_time": 7.661894083023071
    },
    {
      "epoch": 3.875732421875e-06,
      "model_forward_time": 1.7323246002197266,
      "step": 4064
    },
    {
      "epoch": 3.875732421875e-06,
      "step": 4064,
      "training_step_time": 8.244579553604126
    },
    {
      "epoch": 3.876686096191407e-06,
      "model_forward_time": 1.7309167385101318,
      "step": 4065
    },
    {
      "epoch": 3.876686096191407e-06,
      "step": 4065,
      "training_step_time": 8.676011323928833
    },
    {
      "epoch": 3.877639770507812e-06,
      "model_forward_time": 1.728898286819458,
      "step": 4066
    },
    {
      "epoch": 3.877639770507812e-06,
      "step": 4066,
      "training_step_time": 7.650701999664307
    },
    {
      "epoch": 3.878593444824219e-06,
      "model_forward_time": 1.7331516742706299,
      "step": 4067
    },
    {
      "epoch": 3.878593444824219e-06,
      "step": 4067,
      "training_step_time": 7.478303670883179
    },
    {
      "epoch": 3.879547119140625e-06,
      "model_forward_time": 1.7240502834320068,
      "step": 4068
    },
    {
      "epoch": 3.879547119140625e-06,
      "step": 4068,
      "training_step_time": 7.530390739440918
    },
    {
      "epoch": 3.880500793457031e-06,
      "model_forward_time": 1.7334134578704834,
      "step": 4069
    },
    {
      "epoch": 3.880500793457031e-06,
      "step": 4069,
      "training_step_time": 8.035293102264404
    },
    {
      "epoch": 3.881454467773438e-06,
      "grad_norm": 0.24160978198051453,
      "learning_rate": 8.138e-06,
      "loss": 0.0861,
      "step": 4070
    },
    {
      "epoch": 3.881454467773438e-06,
      "model_forward_time": 1.7308270931243896,
      "step": 4070
    },
    {
      "dynamics_loss_avg": 0.07958160266280175,
      "epoch": 3.881454467773438e-06,
      "step": 4070
    },
    {
      "action_loss_avg": 0.010123849869705737,
      "epoch": 3.881454467773438e-06,
      "step": 4070
    },
    {
      "epoch": 3.881454467773438e-06,
      "step": 4070,
      "training_step_time": 8.373750925064087
    },
    {
      "epoch": 3.8824081420898435e-06,
      "model_forward_time": 1.7337985038757324,
      "step": 4071
    },
    {
      "epoch": 3.8824081420898435e-06,
      "step": 4071,
      "training_step_time": 8.464423418045044
    },
    {
      "epoch": 3.88336181640625e-06,
      "model_forward_time": 1.7197132110595703,
      "step": 4072
    },
    {
      "epoch": 3.88336181640625e-06,
      "step": 4072,
      "training_step_time": 8.422233819961548
    },
    {
      "epoch": 3.884315490722657e-06,
      "model_forward_time": 1.7300982475280762,
      "step": 4073
    },
    {
      "epoch": 3.884315490722657e-06,
      "step": 4073,
      "training_step_time": 7.709407091140747
    },
    {
      "epoch": 3.885269165039062e-06,
      "model_forward_time": 1.7262613773345947,
      "step": 4074
    },
    {
      "epoch": 3.885269165039062e-06,
      "step": 4074,
      "training_step_time": 8.701494693756104
    },
    {
      "epoch": 3.886222839355469e-06,
      "model_forward_time": 1.7223279476165771,
      "step": 4075
    },
    {
      "epoch": 3.886222839355469e-06,
      "step": 4075,
      "training_step_time": 7.530325889587402
    },
    {
      "epoch": 3.887176513671875e-06,
      "model_forward_time": 1.7111573219299316,
      "step": 4076
    },
    {
      "epoch": 3.887176513671875e-06,
      "step": 4076,
      "training_step_time": 7.530023813247681
    },
    {
      "epoch": 3.888130187988281e-06,
      "model_forward_time": 1.7284603118896484,
      "step": 4077
    },
    {
      "epoch": 3.888130187988281e-06,
      "step": 4077,
      "training_step_time": 8.210492372512817
    },
    {
      "epoch": 3.889083862304688e-06,
      "model_forward_time": 1.7242152690887451,
      "step": 4078
    },
    {
      "epoch": 3.889083862304688e-06,
      "step": 4078,
      "training_step_time": 7.517359495162964
    },
    {
      "epoch": 3.8900375366210935e-06,
      "model_forward_time": 1.7275960445404053,
      "step": 4079
    },
    {
      "epoch": 3.8900375366210935e-06,
      "step": 4079,
      "training_step_time": 7.650998830795288
    },
    {
      "epoch": 3.8909912109375e-06,
      "grad_norm": 0.4045204818248749,
      "learning_rate": 8.158e-06,
      "loss": 0.0842,
      "step": 4080
    },
    {
      "epoch": 3.8909912109375e-06,
      "model_forward_time": 1.7516329288482666,
      "step": 4080
    },
    {
      "dynamics_loss_avg": 0.07540772762149572,
      "epoch": 3.8909912109375e-06,
      "step": 4080
    },
    {
      "action_loss_avg": 0.010498751420527697,
      "epoch": 3.8909912109375e-06,
      "step": 4080
    },
    {
      "epoch": 3.8909912109375e-06,
      "step": 4080,
      "training_step_time": 9.402406692504883
    },
    {
      "epoch": 3.891944885253907e-06,
      "model_forward_time": 2.9539952278137207,
      "step": 4081
    },
    {
      "epoch": 3.891944885253907e-06,
      "step": 4081,
      "training_step_time": 8.498417615890503
    },
    {
      "epoch": 3.892898559570312e-06,
      "model_forward_time": 1.7276349067687988,
      "step": 4082
    },
    {
      "epoch": 3.892898559570312e-06,
      "step": 4082,
      "training_step_time": 7.629589080810547
    },
    {
      "epoch": 3.893852233886719e-06,
      "model_forward_time": 1.7223918437957764,
      "step": 4083
    },
    {
      "epoch": 3.893852233886719e-06,
      "step": 4083,
      "training_step_time": 7.6948723793029785
    },
    {
      "epoch": 3.894805908203125e-06,
      "model_forward_time": 1.7272908687591553,
      "step": 4084
    },
    {
      "epoch": 3.894805908203125e-06,
      "step": 4084,
      "training_step_time": 7.5983662605285645
    },
    {
      "epoch": 3.895759582519531e-06,
      "model_forward_time": 1.7276477813720703,
      "step": 4085
    },
    {
      "epoch": 3.895759582519531e-06,
      "step": 4085,
      "training_step_time": 7.61202073097229
    },
    {
      "epoch": 3.896713256835938e-06,
      "model_forward_time": 1.7242980003356934,
      "step": 4086
    },
    {
      "epoch": 3.896713256835938e-06,
      "step": 4086,
      "training_step_time": 7.4551780223846436
    },
    {
      "epoch": 3.8976669311523435e-06,
      "model_forward_time": 1.730759620666504,
      "step": 4087
    },
    {
      "epoch": 3.8976669311523435e-06,
      "step": 4087,
      "training_step_time": 7.488781452178955
    },
    {
      "epoch": 3.89862060546875e-06,
      "model_forward_time": 1.7255480289459229,
      "step": 4088
    },
    {
      "epoch": 3.89862060546875e-06,
      "step": 4088,
      "training_step_time": 8.82036828994751
    },
    {
      "epoch": 3.899574279785157e-06,
      "model_forward_time": 1.740818977355957,
      "step": 4089
    },
    {
      "epoch": 3.899574279785157e-06,
      "step": 4089,
      "training_step_time": 11.077631711959839
    },
    {
      "epoch": 3.900527954101562e-06,
      "grad_norm": 0.38544589281082153,
      "learning_rate": 8.178e-06,
      "loss": 0.0774,
      "step": 4090
    },
    {
      "epoch": 3.900527954101562e-06,
      "model_forward_time": 2.899690628051758,
      "step": 4090
    },
    {
      "dynamics_loss_avg": 0.06972417840734124,
      "epoch": 3.900527954101562e-06,
      "step": 4090
    },
    {
      "action_loss_avg": 0.00885790425236337,
      "epoch": 3.900527954101562e-06,
      "step": 4090
    },
    {
      "epoch": 3.900527954101562e-06,
      "step": 4090,
      "training_step_time": 7.797935724258423
    },
    {
      "epoch": 3.901481628417969e-06,
      "model_forward_time": 1.7278892993927002,
      "step": 4091
    },
    {
      "epoch": 3.901481628417969e-06,
      "step": 4091,
      "training_step_time": 9.215572118759155
    },
    {
      "epoch": 3.902435302734375e-06,
      "model_forward_time": 1.7282702922821045,
      "step": 4092
    },
    {
      "epoch": 3.902435302734375e-06,
      "step": 4092,
      "training_step_time": 8.719401359558105
    },
    {
      "epoch": 3.903388977050781e-06,
      "model_forward_time": 1.7215802669525146,
      "step": 4093
    },
    {
      "epoch": 3.903388977050781e-06,
      "step": 4093,
      "training_step_time": 7.73137354850769
    },
    {
      "epoch": 3.904342651367188e-06,
      "model_forward_time": 1.7311859130859375,
      "step": 4094
    },
    {
      "epoch": 3.904342651367188e-06,
      "step": 4094,
      "training_step_time": 7.542788982391357
    },
    {
      "epoch": 3.9052963256835935e-06,
      "model_forward_time": 1.7264986038208008,
      "step": 4095
    },
    {
      "epoch": 3.9052963256835935e-06,
      "step": 4095,
      "training_step_time": 7.491963863372803
    },
    {
      "epoch": 3.90625e-06,
      "model_forward_time": 1.721552848815918,
      "step": 4096
    },
    {
      "epoch": 3.90625e-06,
      "step": 4096,
      "training_step_time": 8.176254510879517
    },
    {
      "epoch": 3.907203674316407e-06,
      "model_forward_time": 1.7288522720336914,
      "step": 4097
    },
    {
      "epoch": 3.907203674316407e-06,
      "step": 4097,
      "training_step_time": 7.556198358535767
    },
    {
      "epoch": 3.908157348632812e-06,
      "model_forward_time": 1.73478364944458,
      "step": 4098
    },
    {
      "epoch": 3.908157348632812e-06,
      "step": 4098,
      "training_step_time": 9.643275022506714
    },
    {
      "epoch": 3.909111022949219e-06,
      "model_forward_time": 2.8436899185180664,
      "step": 4099
    },
    {
      "epoch": 3.909111022949219e-06,
      "step": 4099,
      "training_step_time": 7.702743053436279
    },
    {
      "epoch": 3.910064697265625e-06,
      "grad_norm": 0.510468602180481,
      "learning_rate": 8.198e-06,
      "loss": 0.0867,
      "step": 4100
    },
    {
      "epoch": 3.910064697265625e-06,
      "model_forward_time": 1.7274723052978516,
      "step": 4100
    },
    {
      "dynamics_loss_avg": 0.06317772082984448,
      "epoch": 3.910064697265625e-06,
      "step": 4100
    },
    {
      "action_loss_avg": 0.00941896135918796,
      "epoch": 3.910064697265625e-06,
      "step": 4100
    },
    {
      "epoch": 3.910064697265625e-06,
      "step": 4100,
      "training_step_time": 8.647033214569092
    },
    {
      "epoch": 3.911018371582031e-06,
      "model_forward_time": 1.7261810302734375,
      "step": 4101
    },
    {
      "epoch": 3.911018371582031e-06,
      "step": 4101,
      "training_step_time": 8.778380870819092
    },
    {
      "epoch": 3.911972045898438e-06,
      "model_forward_time": 1.7263476848602295,
      "step": 4102
    },
    {
      "epoch": 3.911972045898438e-06,
      "step": 4102,
      "training_step_time": 7.764567852020264
    },
    {
      "epoch": 3.9129257202148435e-06,
      "model_forward_time": 1.7357759475708008,
      "step": 4103
    },
    {
      "epoch": 3.9129257202148435e-06,
      "step": 4103,
      "training_step_time": 7.640098333358765
    },
    {
      "epoch": 3.91387939453125e-06,
      "model_forward_time": 1.3510658740997314,
      "step": 4104
    },
    {
      "epoch": 3.91387939453125e-06,
      "step": 4104,
      "training_step_time": 7.511404037475586
    },
    {
      "epoch": 3.914833068847657e-06,
      "model_forward_time": 1.722914218902588,
      "step": 4105
    },
    {
      "epoch": 3.914833068847657e-06,
      "step": 4105,
      "training_step_time": 7.59867262840271
    },
    {
      "epoch": 3.915786743164062e-06,
      "model_forward_time": 1.725968360900879,
      "step": 4106
    },
    {
      "epoch": 3.915786743164062e-06,
      "step": 4106,
      "training_step_time": 7.647314548492432
    },
    {
      "epoch": 3.916740417480469e-06,
      "model_forward_time": 1.7303404808044434,
      "step": 4107
    },
    {
      "epoch": 3.916740417480469e-06,
      "step": 4107,
      "training_step_time": 9.201003551483154
    },
    {
      "epoch": 3.917694091796875e-06,
      "model_forward_time": 2.819185495376587,
      "step": 4108
    },
    {
      "epoch": 3.917694091796875e-06,
      "step": 4108,
      "training_step_time": 8.664279699325562
    },
    {
      "epoch": 3.918647766113281e-06,
      "model_forward_time": 1.7268421649932861,
      "step": 4109
    },
    {
      "epoch": 3.918647766113281e-06,
      "step": 4109,
      "training_step_time": 8.211956262588501
    },
    {
      "epoch": 3.919601440429688e-06,
      "grad_norm": 0.2513463795185089,
      "learning_rate": 8.218e-06,
      "loss": 0.0854,
      "step": 4110
    },
    {
      "epoch": 3.919601440429688e-06,
      "model_forward_time": 1.7244455814361572,
      "step": 4110
    },
    {
      "dynamics_loss_avg": 0.07274903990328312,
      "epoch": 3.919601440429688e-06,
      "step": 4110
    },
    {
      "action_loss_avg": 0.008718867227435112,
      "epoch": 3.919601440429688e-06,
      "step": 4110
    },
    {
      "epoch": 3.919601440429688e-06,
      "step": 4110,
      "training_step_time": 8.686337232589722
    },
    {
      "epoch": 3.9205551147460935e-06,
      "model_forward_time": 1.342125415802002,
      "step": 4111
    },
    {
      "epoch": 3.9205551147460935e-06,
      "step": 4111,
      "training_step_time": 7.725948810577393
    },
    {
      "epoch": 3.9215087890625e-06,
      "model_forward_time": 1.7201786041259766,
      "step": 4112
    },
    {
      "epoch": 3.9215087890625e-06,
      "step": 4112,
      "training_step_time": 8.209004640579224
    },
    {
      "epoch": 3.922462463378907e-06,
      "model_forward_time": 1.725632667541504,
      "step": 4113
    },
    {
      "epoch": 3.922462463378907e-06,
      "step": 4113,
      "training_step_time": 7.495889663696289
    },
    {
      "epoch": 3.923416137695312e-06,
      "model_forward_time": 1.3524436950683594,
      "step": 4114
    },
    {
      "epoch": 3.923416137695312e-06,
      "step": 4114,
      "training_step_time": 7.546279668807983
    },
    {
      "epoch": 3.924369812011719e-06,
      "model_forward_time": 1.721515417098999,
      "step": 4115
    },
    {
      "epoch": 3.924369812011719e-06,
      "step": 4115,
      "training_step_time": 7.514469623565674
    },
    {
      "epoch": 3.925323486328125e-06,
      "model_forward_time": 1.7326586246490479,
      "step": 4116
    },
    {
      "epoch": 3.925323486328125e-06,
      "step": 4116,
      "training_step_time": 8.511585474014282
    },
    {
      "epoch": 3.926277160644531e-06,
      "model_forward_time": 1.7970654964447021,
      "step": 4117
    },
    {
      "epoch": 3.926277160644531e-06,
      "step": 4117,
      "training_step_time": 8.664501667022705
    },
    {
      "epoch": 3.927230834960938e-06,
      "model_forward_time": 2.89673113822937,
      "step": 4118
    },
    {
      "epoch": 3.927230834960938e-06,
      "step": 4118,
      "training_step_time": 7.774749279022217
    },
    {
      "epoch": 3.9281845092773435e-06,
      "model_forward_time": 1.351909875869751,
      "step": 4119
    },
    {
      "epoch": 3.9281845092773435e-06,
      "step": 4119,
      "training_step_time": 7.641606330871582
    },
    {
      "epoch": 3.92913818359375e-06,
      "grad_norm": 0.5987195372581482,
      "learning_rate": 8.238e-06,
      "loss": 0.0869,
      "step": 4120
    },
    {
      "epoch": 3.92913818359375e-06,
      "model_forward_time": 1.7319114208221436,
      "step": 4120
    },
    {
      "dynamics_loss_avg": 0.07286386750638485,
      "epoch": 3.92913818359375e-06,
      "step": 4120
    },
    {
      "action_loss_avg": 0.00893893128959462,
      "epoch": 3.92913818359375e-06,
      "step": 4120
    },
    {
      "epoch": 3.92913818359375e-06,
      "step": 4120,
      "training_step_time": 8.602226495742798
    },
    {
      "epoch": 3.930091857910157e-06,
      "model_forward_time": 1.724994421005249,
      "step": 4121
    },
    {
      "epoch": 3.930091857910157e-06,
      "step": 4121,
      "training_step_time": 7.546886682510376
    },
    {
      "epoch": 3.931045532226562e-06,
      "model_forward_time": 1.7249112129211426,
      "step": 4122
    },
    {
      "epoch": 3.931045532226562e-06,
      "step": 4122,
      "training_step_time": 8.390603303909302
    },
    {
      "epoch": 3.931999206542969e-06,
      "model_forward_time": 1.7242507934570312,
      "step": 4123
    },
    {
      "epoch": 3.931999206542969e-06,
      "step": 4123,
      "training_step_time": 7.534296035766602
    },
    {
      "epoch": 3.932952880859375e-06,
      "model_forward_time": 1.721235990524292,
      "step": 4124
    },
    {
      "epoch": 3.932952880859375e-06,
      "step": 4124,
      "training_step_time": 7.482269287109375
    },
    {
      "epoch": 3.933906555175781e-06,
      "model_forward_time": 1.721602201461792,
      "step": 4125
    },
    {
      "epoch": 3.933906555175781e-06,
      "step": 4125,
      "training_step_time": 7.57124662399292
    },
    {
      "epoch": 3.934860229492188e-06,
      "model_forward_time": 1.7944788932800293,
      "step": 4126
    },
    {
      "epoch": 3.934860229492188e-06,
      "step": 4126,
      "training_step_time": 8.606208801269531
    },
    {
      "epoch": 3.9358139038085935e-06,
      "model_forward_time": 2.8153059482574463,
      "step": 4127
    },
    {
      "epoch": 3.9358139038085935e-06,
      "step": 4127,
      "training_step_time": 8.732398509979248
    },
    {
      "epoch": 3.936767578125e-06,
      "model_forward_time": 1.3476331233978271,
      "step": 4128
    },
    {
      "epoch": 3.936767578125e-06,
      "step": 4128,
      "training_step_time": 8.171571254730225
    },
    {
      "epoch": 3.937721252441406e-06,
      "model_forward_time": 1.7242751121520996,
      "step": 4129
    },
    {
      "epoch": 3.937721252441406e-06,
      "step": 4129,
      "training_step_time": 7.6820454597473145
    },
    {
      "epoch": 3.938674926757812e-06,
      "grad_norm": 0.3401491940021515,
      "learning_rate": 8.258000000000001e-06,
      "loss": 0.0772,
      "step": 4130
    },
    {
      "epoch": 3.938674926757812e-06,
      "model_forward_time": 1.7262749671936035,
      "step": 4130
    },
    {
      "dynamics_loss_avg": 0.07967427745461464,
      "epoch": 3.938674926757812e-06,
      "step": 4130
    },
    {
      "action_loss_avg": 0.01153154717758298,
      "epoch": 3.938674926757812e-06,
      "step": 4130
    },
    {
      "epoch": 3.938674926757812e-06,
      "step": 4130,
      "training_step_time": 7.625168561935425
    },
    {
      "epoch": 3.939628601074219e-06,
      "model_forward_time": 1.7294037342071533,
      "step": 4131
    },
    {
      "epoch": 3.939628601074219e-06,
      "step": 4131,
      "training_step_time": 7.621141672134399
    },
    {
      "epoch": 3.940582275390625e-06,
      "model_forward_time": 1.7215218544006348,
      "step": 4132
    },
    {
      "epoch": 3.940582275390625e-06,
      "step": 4132,
      "training_step_time": 7.511961936950684
    },
    {
      "epoch": 3.941535949707031e-06,
      "model_forward_time": 1.7282803058624268,
      "step": 4133
    },
    {
      "epoch": 3.941535949707031e-06,
      "step": 4133,
      "training_step_time": 7.4568774700164795
    },
    {
      "epoch": 3.942489624023438e-06,
      "model_forward_time": 1.7275631427764893,
      "step": 4134
    },
    {
      "epoch": 3.942489624023438e-06,
      "step": 4134,
      "training_step_time": 8.387116432189941
    },
    {
      "epoch": 3.9434432983398435e-06,
      "model_forward_time": 1.7339413166046143,
      "step": 4135
    },
    {
      "epoch": 3.9434432983398435e-06,
      "step": 4135,
      "training_step_time": 9.67647123336792
    },
    {
      "epoch": 3.94439697265625e-06,
      "model_forward_time": 1.7273592948913574,
      "step": 4136
    },
    {
      "epoch": 3.94439697265625e-06,
      "step": 4136,
      "training_step_time": 9.036099910736084
    },
    {
      "epoch": 3.945350646972656e-06,
      "model_forward_time": 2.912457227706909,
      "step": 4137
    },
    {
      "epoch": 3.945350646972656e-06,
      "step": 4137,
      "training_step_time": 7.751856088638306
    },
    {
      "epoch": 3.946304321289062e-06,
      "model_forward_time": 1.7293410301208496,
      "step": 4138
    },
    {
      "epoch": 3.946304321289062e-06,
      "step": 4138,
      "training_step_time": 7.817303419113159
    },
    {
      "epoch": 3.947257995605469e-06,
      "model_forward_time": 1.7277324199676514,
      "step": 4139
    },
    {
      "epoch": 3.947257995605469e-06,
      "step": 4139,
      "training_step_time": 8.77766466140747
    },
    {
      "epoch": 3.948211669921875e-06,
      "grad_norm": 0.39062920212745667,
      "learning_rate": 8.278000000000001e-06,
      "loss": 0.0722,
      "step": 4140
    },
    {
      "epoch": 3.948211669921875e-06,
      "model_forward_time": 1.7304232120513916,
      "step": 4140
    },
    {
      "dynamics_loss_avg": 0.0735827786847949,
      "epoch": 3.948211669921875e-06,
      "step": 4140
    },
    {
      "action_loss_avg": 0.00896763326600194,
      "epoch": 3.948211669921875e-06,
      "step": 4140
    },
    {
      "epoch": 3.948211669921875e-06,
      "step": 4140,
      "training_step_time": 7.612333059310913
    },
    {
      "epoch": 3.949165344238281e-06,
      "model_forward_time": 1.7303216457366943,
      "step": 4141
    },
    {
      "epoch": 3.949165344238281e-06,
      "step": 4141,
      "training_step_time": 7.516678094863892
    },
    {
      "epoch": 3.950119018554688e-06,
      "model_forward_time": 1.7294433116912842,
      "step": 4142
    },
    {
      "epoch": 3.950119018554688e-06,
      "step": 4142,
      "training_step_time": 7.648778915405273
    },
    {
      "epoch": 3.9510726928710936e-06,
      "model_forward_time": 1.7290921211242676,
      "step": 4143
    },
    {
      "epoch": 3.9510726928710936e-06,
      "step": 4143,
      "training_step_time": 7.6780853271484375
    },
    {
      "epoch": 3.9520263671875e-06,
      "model_forward_time": 1.728590488433838,
      "step": 4144
    },
    {
      "epoch": 3.9520263671875e-06,
      "step": 4144,
      "training_step_time": 7.679297208786011
    },
    {
      "epoch": 3.952980041503906e-06,
      "model_forward_time": 1.7724850177764893,
      "step": 4145
    },
    {
      "epoch": 3.952980041503906e-06,
      "step": 4145,
      "training_step_time": 9.562150239944458
    },
    {
      "epoch": 3.953933715820312e-06,
      "model_forward_time": 2.900224447250366,
      "step": 4146
    },
    {
      "epoch": 3.953933715820312e-06,
      "step": 4146,
      "training_step_time": 7.841049432754517
    },
    {
      "epoch": 3.954887390136719e-06,
      "model_forward_time": 1.3592846393585205,
      "step": 4147
    },
    {
      "epoch": 3.954887390136719e-06,
      "step": 4147,
      "training_step_time": 7.903514623641968
    },
    {
      "epoch": 3.955841064453125e-06,
      "model_forward_time": 1.7310242652893066,
      "step": 4148
    },
    {
      "epoch": 3.955841064453125e-06,
      "step": 4148,
      "training_step_time": 7.721240758895874
    },
    {
      "epoch": 3.956794738769531e-06,
      "model_forward_time": 1.7123353481292725,
      "step": 4149
    },
    {
      "epoch": 3.956794738769531e-06,
      "step": 4149,
      "training_step_time": 7.695816516876221
    },
    {
      "epoch": 3.957748413085938e-06,
      "grad_norm": 0.5486428737640381,
      "learning_rate": 8.298000000000001e-06,
      "loss": 0.0722,
      "step": 4150
    },
    {
      "epoch": 3.957748413085938e-06,
      "model_forward_time": 1.720479965209961,
      "step": 4150
    },
    {
      "dynamics_loss_avg": 0.05985929369926453,
      "epoch": 3.957748413085938e-06,
      "step": 4150
    },
    {
      "action_loss_avg": 0.008924154681153595,
      "epoch": 3.957748413085938e-06,
      "step": 4150
    },
    {
      "epoch": 3.957748413085938e-06,
      "step": 4150,
      "training_step_time": 7.59993839263916
    },
    {
      "epoch": 3.9587020874023436e-06,
      "model_forward_time": 1.7282824516296387,
      "step": 4151
    },
    {
      "epoch": 3.9587020874023436e-06,
      "step": 4151,
      "training_step_time": 7.62138819694519
    },
    {
      "epoch": 3.95965576171875e-06,
      "model_forward_time": 1.7183914184570312,
      "step": 4152
    },
    {
      "epoch": 3.95965576171875e-06,
      "step": 4152,
      "training_step_time": 7.5780744552612305
    },
    {
      "epoch": 3.960609436035156e-06,
      "model_forward_time": 1.7221205234527588,
      "step": 4153
    },
    {
      "epoch": 3.960609436035156e-06,
      "step": 4153,
      "training_step_time": 8.447798490524292
    },
    {
      "epoch": 3.961563110351562e-06,
      "model_forward_time": 1.7864253520965576,
      "step": 4154
    },
    {
      "epoch": 3.961563110351562e-06,
      "step": 4154,
      "training_step_time": 8.51682448387146
    },
    {
      "epoch": 3.962516784667969e-06,
      "model_forward_time": 2.91810941696167,
      "step": 4155
    },
    {
      "epoch": 3.962516784667969e-06,
      "step": 4155,
      "training_step_time": 7.759708642959595
    },
    {
      "epoch": 3.963470458984375e-06,
      "model_forward_time": 1.7293004989624023,
      "step": 4156
    },
    {
      "epoch": 3.963470458984375e-06,
      "step": 4156,
      "training_step_time": 7.928187847137451
    },
    {
      "epoch": 3.964424133300781e-06,
      "model_forward_time": 1.7238707542419434,
      "step": 4157
    },
    {
      "epoch": 3.964424133300781e-06,
      "step": 4157,
      "training_step_time": 7.678435564041138
    },
    {
      "epoch": 3.965377807617188e-06,
      "model_forward_time": 1.7211990356445312,
      "step": 4158
    },
    {
      "epoch": 3.965377807617188e-06,
      "step": 4158,
      "training_step_time": 7.657457590103149
    },
    {
      "epoch": 3.9663314819335936e-06,
      "model_forward_time": 1.7169742584228516,
      "step": 4159
    },
    {
      "epoch": 3.9663314819335936e-06,
      "step": 4159,
      "training_step_time": 7.590217113494873
    },
    {
      "epoch": 3.96728515625e-06,
      "grad_norm": 0.7716153860092163,
      "learning_rate": 8.318e-06,
      "loss": 0.0778,
      "step": 4160
    },
    {
      "epoch": 3.96728515625e-06,
      "model_forward_time": 1.722630500793457,
      "step": 4160
    },
    {
      "dynamics_loss_avg": 0.055438807047903535,
      "epoch": 3.96728515625e-06,
      "step": 4160
    },
    {
      "action_loss_avg": 0.0088627265766263,
      "epoch": 3.96728515625e-06,
      "step": 4160
    },
    {
      "epoch": 3.96728515625e-06,
      "step": 4160,
      "training_step_time": 7.601159572601318
    },
    {
      "epoch": 3.968238830566406e-06,
      "model_forward_time": 1.7298202514648438,
      "step": 4161
    },
    {
      "epoch": 3.968238830566406e-06,
      "step": 4161,
      "training_step_time": 8.890551567077637
    },
    {
      "epoch": 3.9691925048828124e-06,
      "model_forward_time": 1.712982416152954,
      "step": 4162
    },
    {
      "epoch": 3.9691925048828124e-06,
      "step": 4162,
      "training_step_time": 9.332351446151733
    },
    {
      "epoch": 3.970146179199219e-06,
      "model_forward_time": 1.7706263065338135,
      "step": 4163
    },
    {
      "epoch": 3.970146179199219e-06,
      "step": 4163,
      "training_step_time": 8.150210857391357
    },
    {
      "epoch": 3.971099853515625e-06,
      "model_forward_time": 2.8731446266174316,
      "step": 4164
    },
    {
      "epoch": 3.971099853515625e-06,
      "step": 4164,
      "training_step_time": 8.22855019569397
    },
    {
      "epoch": 3.972053527832031e-06,
      "model_forward_time": 1.7273542881011963,
      "step": 4165
    },
    {
      "epoch": 3.972053527832031e-06,
      "step": 4165,
      "training_step_time": 7.905261516571045
    },
    {
      "epoch": 3.973007202148438e-06,
      "model_forward_time": 1.3493454456329346,
      "step": 4166
    },
    {
      "epoch": 3.973007202148438e-06,
      "step": 4166,
      "training_step_time": 7.606060743331909
    },
    {
      "epoch": 3.9739608764648436e-06,
      "model_forward_time": 1.7219691276550293,
      "step": 4167
    },
    {
      "epoch": 3.9739608764648436e-06,
      "step": 4167,
      "training_step_time": 8.626409530639648
    },
    {
      "epoch": 3.97491455078125e-06,
      "model_forward_time": 1.7283596992492676,
      "step": 4168
    },
    {
      "epoch": 3.97491455078125e-06,
      "step": 4168,
      "training_step_time": 7.830142021179199
    },
    {
      "epoch": 3.975868225097656e-06,
      "model_forward_time": 1.725773811340332,
      "step": 4169
    },
    {
      "epoch": 3.975868225097656e-06,
      "step": 4169,
      "training_step_time": 7.516385316848755
    },
    {
      "epoch": 3.9768218994140624e-06,
      "grad_norm": 0.4759925901889801,
      "learning_rate": 8.338e-06,
      "loss": 0.0791,
      "step": 4170
    },
    {
      "epoch": 3.9768218994140624e-06,
      "model_forward_time": 1.7271509170532227,
      "step": 4170
    },
    {
      "dynamics_loss_avg": 0.047218582779169085,
      "epoch": 3.9768218994140624e-06,
      "step": 4170
    },
    {
      "action_loss_avg": 0.008661206648685037,
      "epoch": 3.9768218994140624e-06,
      "step": 4170
    },
    {
      "epoch": 3.9768218994140624e-06,
      "step": 4170,
      "training_step_time": 7.503079175949097
    },
    {
      "epoch": 3.977775573730469e-06,
      "model_forward_time": 1.72505784034729,
      "step": 4171
    },
    {
      "epoch": 3.977775573730469e-06,
      "step": 4171,
      "training_step_time": 8.686230182647705
    },
    {
      "epoch": 3.978729248046875e-06,
      "model_forward_time": 1.7262508869171143,
      "step": 4172
    },
    {
      "epoch": 3.978729248046875e-06,
      "step": 4172,
      "training_step_time": 8.590308427810669
    },
    {
      "epoch": 3.979682922363281e-06,
      "model_forward_time": 2.8554890155792236,
      "step": 4173
    },
    {
      "epoch": 3.979682922363281e-06,
      "step": 4173,
      "training_step_time": 7.662840843200684
    },
    {
      "epoch": 3.980636596679688e-06,
      "model_forward_time": 1.7198615074157715,
      "step": 4174
    },
    {
      "epoch": 3.980636596679688e-06,
      "step": 4174,
      "training_step_time": 8.591037034988403
    },
    {
      "epoch": 3.981590270996094e-06,
      "model_forward_time": 1.719663381576538,
      "step": 4175
    },
    {
      "epoch": 3.981590270996094e-06,
      "step": 4175,
      "training_step_time": 8.774829387664795
    },
    {
      "epoch": 3.9825439453125e-06,
      "model_forward_time": 1.7248001098632812,
      "step": 4176
    },
    {
      "epoch": 3.9825439453125e-06,
      "step": 4176,
      "training_step_time": 8.685679197311401
    },
    {
      "epoch": 3.983497619628906e-06,
      "model_forward_time": 1.7243852615356445,
      "step": 4177
    },
    {
      "epoch": 3.983497619628906e-06,
      "step": 4177,
      "training_step_time": 7.75794529914856
    },
    {
      "epoch": 3.9844512939453124e-06,
      "model_forward_time": 1.72117280960083,
      "step": 4178
    },
    {
      "epoch": 3.9844512939453124e-06,
      "step": 4178,
      "training_step_time": 8.282801866531372
    },
    {
      "epoch": 3.985404968261719e-06,
      "model_forward_time": 1.7231628894805908,
      "step": 4179
    },
    {
      "epoch": 3.985404968261719e-06,
      "step": 4179,
      "training_step_time": 8.563491106033325
    },
    {
      "epoch": 3.986358642578125e-06,
      "grad_norm": 0.5601908564567566,
      "learning_rate": 8.358e-06,
      "loss": 0.0792,
      "step": 4180
    },
    {
      "epoch": 3.986358642578125e-06,
      "model_forward_time": 1.723344326019287,
      "step": 4180
    },
    {
      "dynamics_loss_avg": 0.0626709870994091,
      "epoch": 3.986358642578125e-06,
      "step": 4180
    },
    {
      "action_loss_avg": 0.0068619341356679795,
      "epoch": 3.986358642578125e-06,
      "step": 4180
    },
    {
      "epoch": 3.986358642578125e-06,
      "step": 4180,
      "training_step_time": 7.780654668807983
    },
    {
      "epoch": 3.987312316894531e-06,
      "model_forward_time": 1.7218217849731445,
      "step": 4181
    },
    {
      "epoch": 3.987312316894531e-06,
      "step": 4181,
      "training_step_time": 8.44687294960022
    },
    {
      "epoch": 3.988265991210938e-06,
      "model_forward_time": 1.7826802730560303,
      "step": 4182
    },
    {
      "epoch": 3.988265991210938e-06,
      "step": 4182,
      "training_step_time": 8.12259030342102
    },
    {
      "epoch": 3.989219665527344e-06,
      "model_forward_time": 1.726207971572876,
      "step": 4183
    },
    {
      "epoch": 3.989219665527344e-06,
      "step": 4183,
      "training_step_time": 8.507256507873535
    },
    {
      "epoch": 3.99017333984375e-06,
      "model_forward_time": 1.7267119884490967,
      "step": 4184
    },
    {
      "epoch": 3.99017333984375e-06,
      "step": 4184,
      "training_step_time": 7.84390926361084
    },
    {
      "epoch": 3.991127014160156e-06,
      "model_forward_time": 1.7265326976776123,
      "step": 4185
    },
    {
      "epoch": 3.991127014160156e-06,
      "step": 4185,
      "training_step_time": 8.435628414154053
    },
    {
      "epoch": 3.9920806884765624e-06,
      "model_forward_time": 1.7215807437896729,
      "step": 4186
    },
    {
      "epoch": 3.9920806884765624e-06,
      "step": 4186,
      "training_step_time": 9.50537919998169
    },
    {
      "epoch": 3.993034362792969e-06,
      "model_forward_time": 1.3475935459136963,
      "step": 4187
    },
    {
      "epoch": 3.993034362792969e-06,
      "step": 4187,
      "training_step_time": 7.563013315200806
    },
    {
      "epoch": 3.993988037109375e-06,
      "model_forward_time": 1.7320921421051025,
      "step": 4188
    },
    {
      "epoch": 3.993988037109375e-06,
      "step": 4188,
      "training_step_time": 7.505063056945801
    },
    {
      "epoch": 3.994941711425781e-06,
      "model_forward_time": 1.7232911586761475,
      "step": 4189
    },
    {
      "epoch": 3.994941711425781e-06,
      "step": 4189,
      "training_step_time": 7.674090385437012
    },
    {
      "epoch": 3.995895385742188e-06,
      "grad_norm": 0.6331287026405334,
      "learning_rate": 8.378e-06,
      "loss": 0.0894,
      "step": 4190
    },
    {
      "epoch": 3.995895385742188e-06,
      "model_forward_time": 1.7286159992218018,
      "step": 4190
    },
    {
      "dynamics_loss_avg": 0.06670845560729503,
      "epoch": 3.995895385742188e-06,
      "step": 4190
    },
    {
      "action_loss_avg": 0.007971351454034447,
      "epoch": 3.995895385742188e-06,
      "step": 4190
    },
    {
      "epoch": 3.995895385742188e-06,
      "step": 4190,
      "training_step_time": 7.650599956512451
    },
    {
      "epoch": 3.996849060058594e-06,
      "model_forward_time": 1.7699897289276123,
      "step": 4191
    },
    {
      "epoch": 3.996849060058594e-06,
      "step": 4191,
      "training_step_time": 9.131510734558105
    },
    {
      "epoch": 3.997802734375e-06,
      "model_forward_time": 2.879152774810791,
      "step": 4192
    },
    {
      "epoch": 3.997802734375e-06,
      "step": 4192,
      "training_step_time": 7.682862997055054
    },
    {
      "epoch": 3.998756408691406e-06,
      "model_forward_time": 1.725860834121704,
      "step": 4193
    },
    {
      "epoch": 3.998756408691406e-06,
      "step": 4193,
      "training_step_time": 8.9198579788208
    },
    {
      "epoch": 3.9997100830078125e-06,
      "model_forward_time": 1.7230987548828125,
      "step": 4194
    },
    {
      "epoch": 3.9997100830078125e-06,
      "step": 4194,
      "training_step_time": 8.445589303970337
    },
    {
      "epoch": 4.000663757324219e-06,
      "model_forward_time": 1.7239270210266113,
      "step": 4195
    },
    {
      "epoch": 4.000663757324219e-06,
      "step": 4195,
      "training_step_time": 7.624312877655029
    },
    {
      "epoch": 4.001617431640625e-06,
      "model_forward_time": 1.7221298217773438,
      "step": 4196
    },
    {
      "epoch": 4.001617431640625e-06,
      "step": 4196,
      "training_step_time": 7.723114728927612
    },
    {
      "epoch": 4.002571105957031e-06,
      "model_forward_time": 1.7179174423217773,
      "step": 4197
    },
    {
      "epoch": 4.002571105957031e-06,
      "step": 4197,
      "training_step_time": 7.495458364486694
    },
    {
      "epoch": 4.003524780273438e-06,
      "model_forward_time": 1.719207525253296,
      "step": 4198
    },
    {
      "epoch": 4.003524780273438e-06,
      "step": 4198,
      "training_step_time": 7.6411073207855225
    },
    {
      "epoch": 4.004478454589844e-06,
      "model_forward_time": 1.7265746593475342,
      "step": 4199
    },
    {
      "epoch": 4.004478454589844e-06,
      "step": 4199,
      "training_step_time": 7.604791164398193
    },
    {
      "epoch": 4.00543212890625e-06,
      "grad_norm": 0.33353784680366516,
      "learning_rate": 8.398e-06,
      "loss": 0.0834,
      "step": 4200
    },
    {
      "epoch": 4.00543212890625e-06,
      "model_forward_time": 1.7416350841522217,
      "step": 4200
    },
    {
      "dynamics_loss_avg": 0.07672350704669953,
      "epoch": 4.00543212890625e-06,
      "step": 4200
    },
    {
      "action_loss_avg": 0.012513699755072594,
      "epoch": 4.00543212890625e-06,
      "step": 4200
    },
    {
      "epoch": 4.00543212890625e-06,
      "step": 4200,
      "training_step_time": 9.494436025619507
    },
    {
      "epoch": 4.006385803222656e-06,
      "model_forward_time": 2.8816983699798584,
      "step": 4201
    },
    {
      "epoch": 4.006385803222656e-06,
      "step": 4201,
      "training_step_time": 8.524438381195068
    },
    {
      "epoch": 4.0073394775390625e-06,
      "model_forward_time": 1.7284440994262695,
      "step": 4202
    },
    {
      "epoch": 4.0073394775390625e-06,
      "step": 4202,
      "training_step_time": 8.403701782226562
    },
    {
      "epoch": 4.008293151855469e-06,
      "model_forward_time": 1.734290599822998,
      "step": 4203
    },
    {
      "epoch": 4.008293151855469e-06,
      "step": 4203,
      "training_step_time": 7.866582870483398
    },
    {
      "epoch": 4.009246826171875e-06,
      "model_forward_time": 1.7386386394500732,
      "step": 4204
    },
    {
      "epoch": 4.009246826171875e-06,
      "step": 4204,
      "training_step_time": 7.661416053771973
    },
    {
      "epoch": 4.010200500488281e-06,
      "model_forward_time": 1.730600357055664,
      "step": 4205
    },
    {
      "epoch": 4.010200500488281e-06,
      "step": 4205,
      "training_step_time": 7.780356407165527
    },
    {
      "epoch": 4.011154174804688e-06,
      "model_forward_time": 1.7261242866516113,
      "step": 4206
    },
    {
      "epoch": 4.011154174804688e-06,
      "step": 4206,
      "training_step_time": 8.207921028137207
    },
    {
      "epoch": 4.012107849121094e-06,
      "model_forward_time": 1.7402710914611816,
      "step": 4207
    },
    {
      "epoch": 4.012107849121094e-06,
      "step": 4207,
      "training_step_time": 7.618268251419067
    },
    {
      "epoch": 4.0130615234375e-06,
      "model_forward_time": 1.7387893199920654,
      "step": 4208
    },
    {
      "epoch": 4.0130615234375e-06,
      "step": 4208,
      "training_step_time": 9.467196226119995
    },
    {
      "epoch": 4.014015197753906e-06,
      "model_forward_time": 1.752842903137207,
      "step": 4209
    },
    {
      "epoch": 4.014015197753906e-06,
      "step": 4209,
      "training_step_time": 7.604707717895508
    },
    {
      "epoch": 4.0149688720703125e-06,
      "grad_norm": 0.4100812077522278,
      "learning_rate": 8.418000000000001e-06,
      "loss": 0.0824,
      "step": 4210
    },
    {
      "epoch": 4.0149688720703125e-06,
      "model_forward_time": 2.8978097438812256,
      "step": 4210
    },
    {
      "dynamics_loss_avg": 0.07921968810260296,
      "epoch": 4.0149688720703125e-06,
      "step": 4210
    },
    {
      "action_loss_avg": 0.012060332298278808,
      "epoch": 4.0149688720703125e-06,
      "step": 4210
    },
    {
      "epoch": 4.0149688720703125e-06,
      "step": 4210,
      "training_step_time": 8.53179144859314
    },
    {
      "epoch": 4.015922546386719e-06,
      "model_forward_time": 1.7383768558502197,
      "step": 4211
    },
    {
      "epoch": 4.015922546386719e-06,
      "step": 4211,
      "training_step_time": 7.821243524551392
    },
    {
      "epoch": 4.016876220703125e-06,
      "model_forward_time": 1.3764777183532715,
      "step": 4212
    },
    {
      "epoch": 4.016876220703125e-06,
      "step": 4212,
      "training_step_time": 7.79163384437561
    },
    {
      "epoch": 4.017829895019531e-06,
      "model_forward_time": 1.7434043884277344,
      "step": 4213
    },
    {
      "epoch": 4.017829895019531e-06,
      "step": 4213,
      "training_step_time": 7.672246694564819
    },
    {
      "epoch": 4.018783569335938e-06,
      "model_forward_time": 1.739232063293457,
      "step": 4214
    },
    {
      "epoch": 4.018783569335938e-06,
      "step": 4214,
      "training_step_time": 7.579708814620972
    },
    {
      "epoch": 4.019737243652344e-06,
      "model_forward_time": 1.744187355041504,
      "step": 4215
    },
    {
      "epoch": 4.019737243652344e-06,
      "step": 4215,
      "training_step_time": 7.415815114974976
    },
    {
      "epoch": 4.02069091796875e-06,
      "model_forward_time": 1.7399344444274902,
      "step": 4216
    },
    {
      "epoch": 4.02069091796875e-06,
      "step": 4216,
      "training_step_time": 7.573660850524902
    },
    {
      "epoch": 4.021644592285156e-06,
      "model_forward_time": 1.740466833114624,
      "step": 4217
    },
    {
      "epoch": 4.021644592285156e-06,
      "step": 4217,
      "training_step_time": 8.243509531021118
    },
    {
      "epoch": 4.0225982666015625e-06,
      "model_forward_time": 1.735438585281372,
      "step": 4218
    },
    {
      "epoch": 4.0225982666015625e-06,
      "step": 4218,
      "training_step_time": 8.784461975097656
    },
    {
      "epoch": 4.023551940917969e-06,
      "model_forward_time": 2.840174913406372,
      "step": 4219
    },
    {
      "epoch": 4.023551940917969e-06,
      "step": 4219,
      "training_step_time": 7.6147942543029785
    },
    {
      "epoch": 4.024505615234375e-06,
      "grad_norm": 0.34624943137168884,
      "learning_rate": 8.438000000000001e-06,
      "loss": 0.0778,
      "step": 4220
    },
    {
      "epoch": 4.024505615234375e-06,
      "model_forward_time": 1.741001844406128,
      "step": 4220
    },
    {
      "dynamics_loss_avg": 0.06605427488684654,
      "epoch": 4.024505615234375e-06,
      "step": 4220
    },
    {
      "action_loss_avg": 0.011520304251462221,
      "epoch": 4.024505615234375e-06,
      "step": 4220
    },
    {
      "epoch": 4.024505615234375e-06,
      "step": 4220,
      "training_step_time": 7.7640416622161865
    },
    {
      "epoch": 4.025459289550781e-06,
      "model_forward_time": 1.7413833141326904,
      "step": 4221
    },
    {
      "epoch": 4.025459289550781e-06,
      "step": 4221,
      "training_step_time": 7.7384655475616455
    },
    {
      "epoch": 4.026412963867188e-06,
      "model_forward_time": 1.7399587631225586,
      "step": 4222
    },
    {
      "epoch": 4.026412963867188e-06,
      "step": 4222,
      "training_step_time": 7.634680509567261
    },
    {
      "epoch": 4.027366638183594e-06,
      "model_forward_time": 1.7403075695037842,
      "step": 4223
    },
    {
      "epoch": 4.027366638183594e-06,
      "step": 4223,
      "training_step_time": 7.507575988769531
    },
    {
      "epoch": 4.0283203125e-06,
      "model_forward_time": 1.7412030696868896,
      "step": 4224
    },
    {
      "epoch": 4.0283203125e-06,
      "step": 4224,
      "training_step_time": 7.520948648452759
    },
    {
      "epoch": 4.029273986816406e-06,
      "model_forward_time": 1.7428104877471924,
      "step": 4225
    },
    {
      "epoch": 4.029273986816406e-06,
      "step": 4225,
      "training_step_time": 7.5477235317230225
    },
    {
      "epoch": 4.0302276611328125e-06,
      "model_forward_time": 1.7357587814331055,
      "step": 4226
    },
    {
      "epoch": 4.0302276611328125e-06,
      "step": 4226,
      "training_step_time": 10.159672498703003
    },
    {
      "epoch": 4.031181335449219e-06,
      "model_forward_time": 1.7398226261138916,
      "step": 4227
    },
    {
      "epoch": 4.031181335449219e-06,
      "step": 4227,
      "training_step_time": 8.251388788223267
    },
    {
      "epoch": 4.032135009765625e-06,
      "model_forward_time": 1.7834064960479736,
      "step": 4228
    },
    {
      "epoch": 4.032135009765625e-06,
      "step": 4228,
      "training_step_time": 8.411160707473755
    },
    {
      "epoch": 4.033088684082031e-06,
      "model_forward_time": 2.9098093509674072,
      "step": 4229
    },
    {
      "epoch": 4.033088684082031e-06,
      "step": 4229,
      "training_step_time": 9.73345947265625
    },
    {
      "epoch": 4.034042358398438e-06,
      "grad_norm": 0.33236172795295715,
      "learning_rate": 8.458000000000001e-06,
      "loss": 0.0824,
      "step": 4230
    },
    {
      "epoch": 4.034042358398438e-06,
      "model_forward_time": 1.7296552658081055,
      "step": 4230
    },
    {
      "dynamics_loss_avg": 0.08147498164325953,
      "epoch": 4.034042358398438e-06,
      "step": 4230
    },
    {
      "action_loss_avg": 0.013346009864471852,
      "epoch": 4.034042358398438e-06,
      "step": 4230
    },
    {
      "epoch": 4.034042358398438e-06,
      "step": 4230,
      "training_step_time": 9.050238847732544
    },
    {
      "epoch": 4.034996032714844e-06,
      "model_forward_time": 1.7354297637939453,
      "step": 4231
    },
    {
      "epoch": 4.034996032714844e-06,
      "step": 4231,
      "training_step_time": 7.631353378295898
    },
    {
      "epoch": 4.03594970703125e-06,
      "model_forward_time": 1.7419722080230713,
      "step": 4232
    },
    {
      "epoch": 4.03594970703125e-06,
      "step": 4232,
      "training_step_time": 8.422794103622437
    },
    {
      "epoch": 4.036903381347656e-06,
      "model_forward_time": 1.7429490089416504,
      "step": 4233
    },
    {
      "epoch": 4.036903381347656e-06,
      "step": 4233,
      "training_step_time": 7.621340274810791
    },
    {
      "epoch": 4.0378570556640625e-06,
      "model_forward_time": 1.7434043884277344,
      "step": 4234
    },
    {
      "epoch": 4.0378570556640625e-06,
      "step": 4234,
      "training_step_time": 7.543652057647705
    },
    {
      "epoch": 4.038810729980469e-06,
      "model_forward_time": 1.7389888763427734,
      "step": 4235
    },
    {
      "epoch": 4.038810729980469e-06,
      "step": 4235,
      "training_step_time": 7.650893688201904
    },
    {
      "epoch": 4.039764404296875e-06,
      "model_forward_time": 1.7350668907165527,
      "step": 4236
    },
    {
      "epoch": 4.039764404296875e-06,
      "step": 4236,
      "training_step_time": 8.537062883377075
    },
    {
      "epoch": 4.040718078613281e-06,
      "model_forward_time": 1.7710955142974854,
      "step": 4237
    },
    {
      "epoch": 4.040718078613281e-06,
      "step": 4237,
      "training_step_time": 9.247580766677856
    },
    {
      "epoch": 4.041671752929688e-06,
      "model_forward_time": 2.52736496925354,
      "step": 4238
    },
    {
      "epoch": 4.041671752929688e-06,
      "step": 4238,
      "training_step_time": 8.682003498077393
    },
    {
      "epoch": 4.042625427246094e-06,
      "model_forward_time": 1.3685050010681152,
      "step": 4239
    },
    {
      "epoch": 4.042625427246094e-06,
      "step": 4239,
      "training_step_time": 8.497973442077637
    },
    {
      "epoch": 4.0435791015625e-06,
      "grad_norm": 0.35165324807167053,
      "learning_rate": 8.478e-06,
      "loss": 0.0727,
      "step": 4240
    },
    {
      "epoch": 4.0435791015625e-06,
      "model_forward_time": 1.7314965724945068,
      "step": 4240
    },
    {
      "dynamics_loss_avg": 0.07252785339951515,
      "epoch": 4.0435791015625e-06,
      "step": 4240
    },
    {
      "action_loss_avg": 0.009806287894025444,
      "epoch": 4.0435791015625e-06,
      "step": 4240
    },
    {
      "epoch": 4.0435791015625e-06,
      "step": 4240,
      "training_step_time": 7.8842833042144775
    },
    {
      "epoch": 4.044532775878906e-06,
      "model_forward_time": 1.7456226348876953,
      "step": 4241
    },
    {
      "epoch": 4.044532775878906e-06,
      "step": 4241,
      "training_step_time": 7.575827360153198
    },
    {
      "epoch": 4.0454864501953125e-06,
      "model_forward_time": 1.7423086166381836,
      "step": 4242
    },
    {
      "epoch": 4.0454864501953125e-06,
      "step": 4242,
      "training_step_time": 8.420580625534058
    },
    {
      "epoch": 4.046440124511719e-06,
      "model_forward_time": 1.7395644187927246,
      "step": 4243
    },
    {
      "epoch": 4.046440124511719e-06,
      "step": 4243,
      "training_step_time": 7.516289472579956
    },
    {
      "epoch": 4.047393798828125e-06,
      "model_forward_time": 1.3697600364685059,
      "step": 4244
    },
    {
      "epoch": 4.047393798828125e-06,
      "step": 4244,
      "training_step_time": 8.463682651519775
    },
    {
      "epoch": 4.048347473144531e-06,
      "model_forward_time": 1.727973461151123,
      "step": 4245
    },
    {
      "epoch": 4.048347473144531e-06,
      "step": 4245,
      "training_step_time": 8.64105772972107
    },
    {
      "epoch": 4.049301147460938e-06,
      "model_forward_time": 1.7517173290252686,
      "step": 4246
    },
    {
      "epoch": 4.049301147460938e-06,
      "step": 4246,
      "training_step_time": 9.61653184890747
    },
    {
      "epoch": 4.050254821777344e-06,
      "model_forward_time": 2.8389317989349365,
      "step": 4247
    },
    {
      "epoch": 4.050254821777344e-06,
      "step": 4247,
      "training_step_time": 7.6402411460876465
    },
    {
      "epoch": 4.05120849609375e-06,
      "model_forward_time": 1.7490770816802979,
      "step": 4248
    },
    {
      "epoch": 4.05120849609375e-06,
      "step": 4248,
      "training_step_time": 7.8494553565979
    },
    {
      "epoch": 4.052162170410156e-06,
      "model_forward_time": 1.7425782680511475,
      "step": 4249
    },
    {
      "epoch": 4.052162170410156e-06,
      "step": 4249,
      "training_step_time": 7.808977365493774
    },
    {
      "epoch": 4.0531158447265625e-06,
      "grad_norm": 0.5119920969009399,
      "learning_rate": 8.498e-06,
      "loss": 0.074,
      "step": 4250
    },
    {
      "epoch": 4.0531158447265625e-06,
      "model_forward_time": 1.7356300354003906,
      "step": 4250
    },
    {
      "dynamics_loss_avg": 0.07060401365160943,
      "epoch": 4.0531158447265625e-06,
      "step": 4250
    },
    {
      "action_loss_avg": 0.01031596064567566,
      "epoch": 4.0531158447265625e-06,
      "step": 4250
    },
    {
      "epoch": 4.0531158447265625e-06,
      "step": 4250,
      "training_step_time": 7.568209648132324
    },
    {
      "epoch": 4.054069519042969e-06,
      "model_forward_time": 1.7322206497192383,
      "step": 4251
    },
    {
      "epoch": 4.054069519042969e-06,
      "step": 4251,
      "training_step_time": 7.61101222038269
    },
    {
      "epoch": 4.055023193359375e-06,
      "model_forward_time": 1.7397305965423584,
      "step": 4252
    },
    {
      "epoch": 4.055023193359375e-06,
      "step": 4252,
      "training_step_time": 7.493772983551025
    },
    {
      "epoch": 4.055976867675781e-06,
      "model_forward_time": 1.7370166778564453,
      "step": 4253
    },
    {
      "epoch": 4.055976867675781e-06,
      "step": 4253,
      "training_step_time": 8.112150192260742
    },
    {
      "epoch": 4.056930541992187e-06,
      "model_forward_time": 1.7350826263427734,
      "step": 4254
    },
    {
      "epoch": 4.056930541992187e-06,
      "step": 4254,
      "training_step_time": 7.572576999664307
    },
    {
      "epoch": 4.057884216308594e-06,
      "model_forward_time": 1.7503035068511963,
      "step": 4255
    },
    {
      "epoch": 4.057884216308594e-06,
      "step": 4255,
      "training_step_time": 7.9475953578948975
    },
    {
      "epoch": 4.058837890625e-06,
      "model_forward_time": 1.3617780208587646,
      "step": 4256
    },
    {
      "epoch": 4.058837890625e-06,
      "step": 4256,
      "training_step_time": 8.469939708709717
    },
    {
      "epoch": 4.059791564941406e-06,
      "model_forward_time": 2.9244489669799805,
      "step": 4257
    },
    {
      "epoch": 4.059791564941406e-06,
      "step": 4257,
      "training_step_time": 8.345845699310303
    },
    {
      "epoch": 4.0607452392578125e-06,
      "model_forward_time": 1.7365918159484863,
      "step": 4258
    },
    {
      "epoch": 4.0607452392578125e-06,
      "step": 4258,
      "training_step_time": 7.86235237121582
    },
    {
      "epoch": 4.061698913574219e-06,
      "model_forward_time": 1.7412338256835938,
      "step": 4259
    },
    {
      "epoch": 4.061698913574219e-06,
      "step": 4259,
      "training_step_time": 7.660055160522461
    },
    {
      "epoch": 4.062652587890625e-06,
      "grad_norm": 0.5958037972450256,
      "learning_rate": 8.518e-06,
      "loss": 0.0805,
      "step": 4260
    },
    {
      "epoch": 4.062652587890625e-06,
      "model_forward_time": 1.368746042251587,
      "step": 4260
    },
    {
      "dynamics_loss_avg": 0.07112456001341343,
      "epoch": 4.062652587890625e-06,
      "step": 4260
    },
    {
      "action_loss_avg": 0.009543130220845342,
      "epoch": 4.062652587890625e-06,
      "step": 4260
    },
    {
      "epoch": 4.062652587890625e-06,
      "step": 4260,
      "training_step_time": 8.604286193847656
    },
    {
      "epoch": 4.063606262207031e-06,
      "model_forward_time": 1.7305614948272705,
      "step": 4261
    },
    {
      "epoch": 4.063606262207031e-06,
      "step": 4261,
      "training_step_time": 7.476656436920166
    },
    {
      "epoch": 4.064559936523437e-06,
      "model_forward_time": 1.7385063171386719,
      "step": 4262
    },
    {
      "epoch": 4.064559936523437e-06,
      "step": 4262,
      "training_step_time": 7.576055288314819
    },
    {
      "epoch": 4.065513610839844e-06,
      "model_forward_time": 1.735896348953247,
      "step": 4263
    },
    {
      "epoch": 4.065513610839844e-06,
      "step": 4263,
      "training_step_time": 7.488518476486206
    },
    {
      "epoch": 4.06646728515625e-06,
      "model_forward_time": 1.7491238117218018,
      "step": 4264
    },
    {
      "epoch": 4.06646728515625e-06,
      "step": 4264,
      "training_step_time": 8.696111917495728
    },
    {
      "epoch": 4.067420959472656e-06,
      "model_forward_time": 1.7385859489440918,
      "step": 4265
    },
    {
      "epoch": 4.067420959472656e-06,
      "step": 4265,
      "training_step_time": 8.44691777229309
    },
    {
      "epoch": 4.0683746337890625e-06,
      "model_forward_time": 2.8917086124420166,
      "step": 4266
    },
    {
      "epoch": 4.0683746337890625e-06,
      "step": 4266,
      "training_step_time": 7.688863515853882
    },
    {
      "epoch": 4.069328308105469e-06,
      "model_forward_time": 1.7368247509002686,
      "step": 4267
    },
    {
      "epoch": 4.069328308105469e-06,
      "step": 4267,
      "training_step_time": 7.824733257293701
    },
    {
      "epoch": 4.070281982421875e-06,
      "model_forward_time": 1.7383129596710205,
      "step": 4268
    },
    {
      "epoch": 4.070281982421875e-06,
      "step": 4268,
      "training_step_time": 8.568657636642456
    },
    {
      "epoch": 4.071235656738281e-06,
      "model_forward_time": 1.7317709922790527,
      "step": 4269
    },
    {
      "epoch": 4.071235656738281e-06,
      "step": 4269,
      "training_step_time": 7.595002889633179
    },
    {
      "epoch": 4.072189331054687e-06,
      "grad_norm": 0.3210153579711914,
      "learning_rate": 8.538e-06,
      "loss": 0.0801,
      "step": 4270
    },
    {
      "epoch": 4.072189331054687e-06,
      "model_forward_time": 1.7312126159667969,
      "step": 4270
    },
    {
      "dynamics_loss_avg": 0.07453901320695877,
      "epoch": 4.072189331054687e-06,
      "step": 4270
    },
    {
      "action_loss_avg": 0.01217005206272006,
      "epoch": 4.072189331054687e-06,
      "step": 4270
    },
    {
      "epoch": 4.072189331054687e-06,
      "step": 4270,
      "training_step_time": 8.470804214477539
    },
    {
      "epoch": 4.073143005371094e-06,
      "model_forward_time": 1.7386431694030762,
      "step": 4271
    },
    {
      "epoch": 4.073143005371094e-06,
      "step": 4271,
      "training_step_time": 8.331526041030884
    },
    {
      "epoch": 4.0740966796875e-06,
      "model_forward_time": 1.7412619590759277,
      "step": 4272
    },
    {
      "epoch": 4.0740966796875e-06,
      "step": 4272,
      "training_step_time": 8.353654861450195
    },
    {
      "epoch": 4.075050354003906e-06,
      "model_forward_time": 1.7433350086212158,
      "step": 4273
    },
    {
      "epoch": 4.075050354003906e-06,
      "step": 4273,
      "training_step_time": 7.877366781234741
    },
    {
      "epoch": 4.0760040283203125e-06,
      "model_forward_time": 1.7684974670410156,
      "step": 4274
    },
    {
      "epoch": 4.0760040283203125e-06,
      "step": 4274,
      "training_step_time": 7.580066204071045
    },
    {
      "epoch": 4.076957702636719e-06,
      "model_forward_time": 4.021726369857788,
      "step": 4275
    },
    {
      "epoch": 4.076957702636719e-06,
      "step": 4275,
      "training_step_time": 9.33955430984497
    },
    {
      "epoch": 4.077911376953125e-06,
      "model_forward_time": 1.7286605834960938,
      "step": 4276
    },
    {
      "epoch": 4.077911376953125e-06,
      "step": 4276,
      "training_step_time": 7.695327520370483
    },
    {
      "epoch": 4.078865051269531e-06,
      "model_forward_time": 1.737978219985962,
      "step": 4277
    },
    {
      "epoch": 4.078865051269531e-06,
      "step": 4277,
      "training_step_time": 6.790482044219971
    },
    {
      "epoch": 4.079818725585937e-06,
      "model_forward_time": 1.7412753105163574,
      "step": 4278
    },
    {
      "epoch": 4.079818725585937e-06,
      "step": 4278,
      "training_step_time": 6.780508279800415
    },
    {
      "epoch": 4.080772399902344e-06,
      "model_forward_time": 1.7414026260375977,
      "step": 4279
    },
    {
      "epoch": 4.080772399902344e-06,
      "step": 4279,
      "training_step_time": 6.817795276641846
    },
    {
      "epoch": 4.08172607421875e-06,
      "grad_norm": 0.4516667127609253,
      "learning_rate": 8.558e-06,
      "loss": 0.075,
      "step": 4280
    },
    {
      "epoch": 4.08172607421875e-06,
      "model_forward_time": 1.7319591045379639,
      "step": 4280
    },
    {
      "dynamics_loss_avg": 0.06800317503511906,
      "epoch": 4.08172607421875e-06,
      "step": 4280
    },
    {
      "action_loss_avg": 0.008622566470876336,
      "epoch": 4.08172607421875e-06,
      "step": 4280
    },
    {
      "epoch": 4.08172607421875e-06,
      "step": 4280,
      "training_step_time": 6.829601764678955
    },
    {
      "epoch": 4.082679748535156e-06,
      "model_forward_time": 1.746685266494751,
      "step": 4281
    },
    {
      "epoch": 4.082679748535156e-06,
      "step": 4281,
      "training_step_time": 6.760883808135986
    },
    {
      "epoch": 4.0836334228515625e-06,
      "model_forward_time": 1.741872787475586,
      "step": 4282
    },
    {
      "epoch": 4.0836334228515625e-06,
      "step": 4282,
      "training_step_time": 6.812396287918091
    },
    {
      "epoch": 4.084587097167969e-06,
      "model_forward_time": 1.7399981021881104,
      "step": 4283
    },
    {
      "epoch": 4.084587097167969e-06,
      "step": 4283,
      "training_step_time": 6.79223895072937
    },
    {
      "epoch": 4.085540771484375e-06,
      "model_forward_time": 1.7414064407348633,
      "step": 4284
    },
    {
      "epoch": 4.085540771484375e-06,
      "step": 4284,
      "training_step_time": 11.896700859069824
    },
    {
      "epoch": 4.086494445800781e-06,
      "model_forward_time": 1.8067002296447754,
      "step": 4285
    },
    {
      "epoch": 4.086494445800781e-06,
      "step": 4285,
      "training_step_time": 7.832306623458862
    },
    {
      "epoch": 4.087448120117187e-06,
      "model_forward_time": 1.7388477325439453,
      "step": 4286
    },
    {
      "epoch": 4.087448120117187e-06,
      "step": 4286,
      "training_step_time": 6.755538702011108
    },
    {
      "epoch": 4.088401794433594e-06,
      "model_forward_time": 1.7458512783050537,
      "step": 4287
    },
    {
      "epoch": 4.088401794433594e-06,
      "step": 4287,
      "training_step_time": 6.84914231300354
    },
    {
      "epoch": 4.08935546875e-06,
      "model_forward_time": 1.7411534786224365,
      "step": 4288
    },
    {
      "epoch": 4.08935546875e-06,
      "step": 4288,
      "training_step_time": 6.806827068328857
    },
    {
      "epoch": 4.090309143066406e-06,
      "model_forward_time": 1.740483045578003,
      "step": 4289
    },
    {
      "epoch": 4.090309143066406e-06,
      "step": 4289,
      "training_step_time": 6.771245956420898
    },
    {
      "epoch": 4.0912628173828125e-06,
      "grad_norm": 0.43368467688560486,
      "learning_rate": 8.578000000000002e-06,
      "loss": 0.0731,
      "step": 4290
    },
    {
      "epoch": 4.0912628173828125e-06,
      "model_forward_time": 1.363344430923462,
      "step": 4290
    },
    {
      "dynamics_loss_avg": 0.0543397031724453,
      "epoch": 4.0912628173828125e-06,
      "step": 4290
    },
    {
      "action_loss_avg": 0.009384067845530808,
      "epoch": 4.0912628173828125e-06,
      "step": 4290
    },
    {
      "epoch": 4.0912628173828125e-06,
      "step": 4290,
      "training_step_time": 6.847381830215454
    },
    {
      "epoch": 4.092216491699219e-06,
      "model_forward_time": 1.7440178394317627,
      "step": 4291
    },
    {
      "epoch": 4.092216491699219e-06,
      "step": 4291,
      "training_step_time": 6.786088705062866
    },
    {
      "epoch": 4.093170166015625e-06,
      "model_forward_time": 1.7392661571502686,
      "step": 4292
    },
    {
      "epoch": 4.093170166015625e-06,
      "step": 4292,
      "training_step_time": 6.791685581207275
    },
    {
      "epoch": 4.094123840332031e-06,
      "model_forward_time": 1.7397539615631104,
      "step": 4293
    },
    {
      "epoch": 4.094123840332031e-06,
      "step": 4293,
      "training_step_time": 6.756498098373413
    },
    {
      "epoch": 4.095077514648437e-06,
      "model_forward_time": 2.924922466278076,
      "step": 4294
    },
    {
      "epoch": 4.095077514648437e-06,
      "step": 4294,
      "training_step_time": 8.41744875907898
    },
    {
      "epoch": 4.096031188964844e-06,
      "model_forward_time": 1.7434399127960205,
      "step": 4295
    },
    {
      "epoch": 4.096031188964844e-06,
      "step": 4295,
      "training_step_time": 6.806708335876465
    },
    {
      "epoch": 4.09698486328125e-06,
      "model_forward_time": 1.7402243614196777,
      "step": 4296
    },
    {
      "epoch": 4.09698486328125e-06,
      "step": 4296,
      "training_step_time": 6.748136043548584
    },
    {
      "epoch": 4.097938537597656e-06,
      "model_forward_time": 1.734849214553833,
      "step": 4297
    },
    {
      "epoch": 4.097938537597656e-06,
      "step": 4297,
      "training_step_time": 6.842440605163574
    },
    {
      "epoch": 4.0988922119140625e-06,
      "model_forward_time": 1.740415096282959,
      "step": 4298
    },
    {
      "epoch": 4.0988922119140625e-06,
      "step": 4298,
      "training_step_time": 6.837609052658081
    },
    {
      "epoch": 4.099845886230469e-06,
      "model_forward_time": 1.734999418258667,
      "step": 4299
    },
    {
      "epoch": 4.099845886230469e-06,
      "step": 4299,
      "training_step_time": 6.775452375411987
    },
    {
      "epoch": 4.100799560546875e-06,
      "grad_norm": 0.4164873957633972,
      "learning_rate": 8.598000000000001e-06,
      "loss": 0.0793,
      "step": 4300
    },
    {
      "epoch": 4.100799560546875e-06,
      "model_forward_time": 1.7238476276397705,
      "step": 4300
    },
    {
      "dynamics_loss_avg": 0.06113612018525601,
      "epoch": 4.100799560546875e-06,
      "step": 4300
    },
    {
      "action_loss_avg": 0.00910059385932982,
      "epoch": 4.100799560546875e-06,
      "step": 4300
    },
    {
      "epoch": 4.100799560546875e-06,
      "step": 4300,
      "training_step_time": 6.844098806381226
    },
    {
      "epoch": 4.101753234863281e-06,
      "model_forward_time": 1.734076976776123,
      "step": 4301
    },
    {
      "epoch": 4.101753234863281e-06,
      "step": 4301,
      "training_step_time": 6.7115888595581055
    },
    {
      "epoch": 4.102706909179687e-06,
      "model_forward_time": 1.7514135837554932,
      "step": 4302
    },
    {
      "epoch": 4.102706909179687e-06,
      "step": 4302,
      "training_step_time": 6.803466796875
    },
    {
      "epoch": 4.103660583496094e-06,
      "model_forward_time": 2.9460036754608154,
      "step": 4303
    },
    {
      "epoch": 4.103660583496094e-06,
      "step": 4303,
      "training_step_time": 8.870028972625732
    },
    {
      "epoch": 4.1046142578125e-06,
      "model_forward_time": 1.7382104396820068,
      "step": 4304
    },
    {
      "epoch": 4.1046142578125e-06,
      "step": 4304,
      "training_step_time": 7.646136045455933
    },
    {
      "epoch": 4.105567932128906e-06,
      "model_forward_time": 1.7442691326141357,
      "step": 4305
    },
    {
      "epoch": 4.105567932128906e-06,
      "step": 4305,
      "training_step_time": 6.828268766403198
    },
    {
      "epoch": 4.1065216064453125e-06,
      "model_forward_time": 1.7531142234802246,
      "step": 4306
    },
    {
      "epoch": 4.1065216064453125e-06,
      "step": 4306,
      "training_step_time": 6.8182549476623535
    },
    {
      "epoch": 4.107475280761719e-06,
      "model_forward_time": 1.7499065399169922,
      "step": 4307
    },
    {
      "epoch": 4.107475280761719e-06,
      "step": 4307,
      "training_step_time": 6.823870897293091
    },
    {
      "epoch": 4.108428955078125e-06,
      "model_forward_time": 1.7518510818481445,
      "step": 4308
    },
    {
      "epoch": 4.108428955078125e-06,
      "step": 4308,
      "training_step_time": 6.834947109222412
    },
    {
      "epoch": 4.109382629394531e-06,
      "model_forward_time": 1.751817226409912,
      "step": 4309
    },
    {
      "epoch": 4.109382629394531e-06,
      "step": 4309,
      "training_step_time": 6.885076522827148
    },
    {
      "epoch": 4.110336303710937e-06,
      "grad_norm": 0.3225063979625702,
      "learning_rate": 8.618000000000001e-06,
      "loss": 0.0822,
      "step": 4310
    },
    {
      "epoch": 4.110336303710937e-06,
      "model_forward_time": 1.7566359043121338,
      "step": 4310
    },
    {
      "dynamics_loss_avg": 0.07279934603720903,
      "epoch": 4.110336303710937e-06,
      "step": 4310
    },
    {
      "action_loss_avg": 0.00957236853428185,
      "epoch": 4.110336303710937e-06,
      "step": 4310
    },
    {
      "epoch": 4.110336303710937e-06,
      "step": 4310,
      "training_step_time": 6.851258277893066
    },
    {
      "epoch": 4.111289978027344e-06,
      "model_forward_time": 1.759889841079712,
      "step": 4311
    },
    {
      "epoch": 4.111289978027344e-06,
      "step": 4311,
      "training_step_time": 6.875285625457764
    },
    {
      "epoch": 4.11224365234375e-06,
      "model_forward_time": 2.8624093532562256,
      "step": 4312
    },
    {
      "epoch": 4.11224365234375e-06,
      "step": 4312,
      "training_step_time": 11.449899673461914
    },
    {
      "epoch": 4.113197326660156e-06,
      "model_forward_time": 1.748307704925537,
      "step": 4313
    },
    {
      "epoch": 4.113197326660156e-06,
      "step": 4313,
      "training_step_time": 7.950360298156738
    },
    {
      "epoch": 4.1141510009765626e-06,
      "model_forward_time": 1.7520015239715576,
      "step": 4314
    },
    {
      "epoch": 4.1141510009765626e-06,
      "step": 4314,
      "training_step_time": 6.845867872238159
    },
    {
      "epoch": 4.115104675292969e-06,
      "model_forward_time": 1.7503764629364014,
      "step": 4315
    },
    {
      "epoch": 4.115104675292969e-06,
      "step": 4315,
      "training_step_time": 6.786200046539307
    },
    {
      "epoch": 4.116058349609375e-06,
      "model_forward_time": 1.7533090114593506,
      "step": 4316
    },
    {
      "epoch": 4.116058349609375e-06,
      "step": 4316,
      "training_step_time": 6.777696371078491
    },
    {
      "epoch": 4.117012023925781e-06,
      "model_forward_time": 1.7497546672821045,
      "step": 4317
    },
    {
      "epoch": 4.117012023925781e-06,
      "step": 4317,
      "training_step_time": 6.796561241149902
    },
    {
      "epoch": 4.117965698242187e-06,
      "model_forward_time": 1.7586867809295654,
      "step": 4318
    },
    {
      "epoch": 4.117965698242187e-06,
      "step": 4318,
      "training_step_time": 6.784781217575073
    },
    {
      "epoch": 4.118919372558594e-06,
      "model_forward_time": 1.7302882671356201,
      "step": 4319
    },
    {
      "epoch": 4.118919372558594e-06,
      "step": 4319,
      "training_step_time": 6.728896379470825
    },
    {
      "epoch": 4.119873046875e-06,
      "grad_norm": 0.4428849220275879,
      "learning_rate": 8.638000000000001e-06,
      "loss": 0.0794,
      "step": 4320
    },
    {
      "epoch": 4.119873046875e-06,
      "model_forward_time": 1.7491931915283203,
      "step": 4320
    },
    {
      "dynamics_loss_avg": 0.0732277300208807,
      "epoch": 4.119873046875e-06,
      "step": 4320
    },
    {
      "action_loss_avg": 0.009945789678022265,
      "epoch": 4.119873046875e-06,
      "step": 4320
    },
    {
      "epoch": 4.119873046875e-06,
      "step": 4320,
      "training_step_time": 6.764249563217163
    },
    {
      "epoch": 4.120826721191406e-06,
      "model_forward_time": 2.813204050064087,
      "step": 4321
    },
    {
      "epoch": 4.120826721191406e-06,
      "step": 4321,
      "training_step_time": 9.227244138717651
    },
    {
      "epoch": 4.1217803955078126e-06,
      "model_forward_time": 1.805058479309082,
      "step": 4322
    },
    {
      "epoch": 4.1217803955078126e-06,
      "step": 4322,
      "training_step_time": 8.419233798980713
    },
    {
      "epoch": 4.122734069824219e-06,
      "model_forward_time": 1.7467985153198242,
      "step": 4323
    },
    {
      "epoch": 4.122734069824219e-06,
      "step": 4323,
      "training_step_time": 6.7171711921691895
    },
    {
      "epoch": 4.123687744140625e-06,
      "model_forward_time": 1.7521209716796875,
      "step": 4324
    },
    {
      "epoch": 4.123687744140625e-06,
      "step": 4324,
      "training_step_time": 6.763817071914673
    },
    {
      "epoch": 4.124641418457031e-06,
      "model_forward_time": 1.7446320056915283,
      "step": 4325
    },
    {
      "epoch": 4.124641418457031e-06,
      "step": 4325,
      "training_step_time": 6.763993978500366
    },
    {
      "epoch": 4.125595092773437e-06,
      "model_forward_time": 1.7455804347991943,
      "step": 4326
    },
    {
      "epoch": 4.125595092773437e-06,
      "step": 4326,
      "training_step_time": 6.726180076599121
    },
    {
      "epoch": 4.126548767089844e-06,
      "model_forward_time": 1.7430641651153564,
      "step": 4327
    },
    {
      "epoch": 4.126548767089844e-06,
      "step": 4327,
      "training_step_time": 6.761111736297607
    },
    {
      "epoch": 4.12750244140625e-06,
      "model_forward_time": 1.741166114807129,
      "step": 4328
    },
    {
      "epoch": 4.12750244140625e-06,
      "step": 4328,
      "training_step_time": 6.691680192947388
    },
    {
      "epoch": 4.128456115722656e-06,
      "model_forward_time": 1.744800090789795,
      "step": 4329
    },
    {
      "epoch": 4.128456115722656e-06,
      "step": 4329,
      "training_step_time": 6.716500520706177
    },
    {
      "epoch": 4.1294097900390626e-06,
      "grad_norm": 0.37591251730918884,
      "learning_rate": 8.658e-06,
      "loss": 0.0819,
      "step": 4330
    },
    {
      "epoch": 4.1294097900390626e-06,
      "model_forward_time": 1.7330245971679688,
      "step": 4330
    },
    {
      "dynamics_loss_avg": 0.055879545770585536,
      "epoch": 4.1294097900390626e-06,
      "step": 4330
    },
    {
      "action_loss_avg": 0.007116533792577684,
      "epoch": 4.1294097900390626e-06,
      "step": 4330
    },
    {
      "epoch": 4.1294097900390626e-06,
      "step": 4330,
      "training_step_time": 11.568967819213867
    },
    {
      "epoch": 4.130363464355469e-06,
      "model_forward_time": 2.855997323989868,
      "step": 4331
    },
    {
      "epoch": 4.130363464355469e-06,
      "step": 4331,
      "training_step_time": 7.793816089630127
    },
    {
      "epoch": 4.131317138671875e-06,
      "model_forward_time": 1.744542121887207,
      "step": 4332
    },
    {
      "epoch": 4.131317138671875e-06,
      "step": 4332,
      "training_step_time": 7.664379835128784
    },
    {
      "epoch": 4.1322708129882814e-06,
      "model_forward_time": 1.7462012767791748,
      "step": 4333
    },
    {
      "epoch": 4.1322708129882814e-06,
      "step": 4333,
      "training_step_time": 6.6952338218688965
    },
    {
      "epoch": 4.133224487304687e-06,
      "model_forward_time": 1.751765251159668,
      "step": 4334
    },
    {
      "epoch": 4.133224487304687e-06,
      "step": 4334,
      "training_step_time": 6.704745769500732
    },
    {
      "epoch": 4.134178161621094e-06,
      "model_forward_time": 1.7582380771636963,
      "step": 4335
    },
    {
      "epoch": 4.134178161621094e-06,
      "step": 4335,
      "training_step_time": 6.770752429962158
    },
    {
      "epoch": 4.1351318359375e-06,
      "model_forward_time": 1.7453362941741943,
      "step": 4336
    },
    {
      "epoch": 4.1351318359375e-06,
      "step": 4336,
      "training_step_time": 6.729342937469482
    },
    {
      "epoch": 4.136085510253906e-06,
      "model_forward_time": 1.7493774890899658,
      "step": 4337
    },
    {
      "epoch": 4.136085510253906e-06,
      "step": 4337,
      "training_step_time": 6.718738317489624
    },
    {
      "epoch": 4.1370391845703126e-06,
      "model_forward_time": 1.7481441497802734,
      "step": 4338
    },
    {
      "epoch": 4.1370391845703126e-06,
      "step": 4338,
      "training_step_time": 6.75627875328064
    },
    {
      "epoch": 4.137992858886719e-06,
      "model_forward_time": 1.7436330318450928,
      "step": 4339
    },
    {
      "epoch": 4.137992858886719e-06,
      "step": 4339,
      "training_step_time": 9.600495100021362
    },
    {
      "epoch": 4.138946533203125e-06,
      "grad_norm": 0.3845762610435486,
      "learning_rate": 8.678e-06,
      "loss": 0.085,
      "step": 4340
    },
    {
      "epoch": 4.138946533203125e-06,
      "model_forward_time": 2.8674561977386475,
      "step": 4340
    },
    {
      "dynamics_loss_avg": 0.07299749292433262,
      "epoch": 4.138946533203125e-06,
      "step": 4340
    },
    {
      "action_loss_avg": 0.008318888954818249,
      "epoch": 4.138946533203125e-06,
      "step": 4340
    },
    {
      "epoch": 4.138946533203125e-06,
      "step": 4340,
      "training_step_time": 8.578104972839355
    },
    {
      "epoch": 4.1399002075195314e-06,
      "model_forward_time": 1.7289774417877197,
      "step": 4341
    },
    {
      "epoch": 4.1399002075195314e-06,
      "step": 4341,
      "training_step_time": 7.5521252155303955
    },
    {
      "epoch": 4.140853881835937e-06,
      "model_forward_time": 1.7465600967407227,
      "step": 4342
    },
    {
      "epoch": 4.140853881835937e-06,
      "step": 4342,
      "training_step_time": 6.827881574630737
    },
    {
      "epoch": 4.141807556152344e-06,
      "model_forward_time": 1.7784385681152344,
      "step": 4343
    },
    {
      "epoch": 4.141807556152344e-06,
      "step": 4343,
      "training_step_time": 6.706207752227783
    },
    {
      "epoch": 4.14276123046875e-06,
      "model_forward_time": 1.7425217628479004,
      "step": 4344
    },
    {
      "epoch": 4.14276123046875e-06,
      "step": 4344,
      "training_step_time": 6.727233409881592
    },
    {
      "epoch": 4.143714904785156e-06,
      "model_forward_time": 1.752580165863037,
      "step": 4345
    },
    {
      "epoch": 4.143714904785156e-06,
      "step": 4345,
      "training_step_time": 6.748441457748413
    },
    {
      "epoch": 4.144668579101563e-06,
      "model_forward_time": 1.7451167106628418,
      "step": 4346
    },
    {
      "epoch": 4.144668579101563e-06,
      "step": 4346,
      "training_step_time": 6.777743339538574
    },
    {
      "epoch": 4.145622253417969e-06,
      "model_forward_time": 1.74912428855896,
      "step": 4347
    },
    {
      "epoch": 4.145622253417969e-06,
      "step": 4347,
      "training_step_time": 6.791325092315674
    },
    {
      "epoch": 4.146575927734375e-06,
      "model_forward_time": 1.7524750232696533,
      "step": 4348
    },
    {
      "epoch": 4.146575927734375e-06,
      "step": 4348,
      "training_step_time": 8.704963445663452
    },
    {
      "epoch": 4.1475296020507814e-06,
      "model_forward_time": 2.9211318492889404,
      "step": 4349
    },
    {
      "epoch": 4.1475296020507814e-06,
      "step": 4349,
      "training_step_time": 10.635698556900024
    },
    {
      "epoch": 4.148483276367187e-06,
      "grad_norm": 0.40621936321258545,
      "learning_rate": 8.698e-06,
      "loss": 0.0882,
      "step": 4350
    },
    {
      "epoch": 4.148483276367187e-06,
      "model_forward_time": 1.7452261447906494,
      "step": 4350
    },
    {
      "dynamics_loss_avg": 0.09384296424686908,
      "epoch": 4.148483276367187e-06,
      "step": 4350
    },
    {
      "action_loss_avg": 0.011571353627368807,
      "epoch": 4.148483276367187e-06,
      "step": 4350
    },
    {
      "epoch": 4.148483276367187e-06,
      "step": 4350,
      "training_step_time": 8.000606536865234
    },
    {
      "epoch": 4.149436950683594e-06,
      "model_forward_time": 1.7430377006530762,
      "step": 4351
    },
    {
      "epoch": 4.149436950683594e-06,
      "step": 4351,
      "training_step_time": 6.761173725128174
    },
    {
      "epoch": 4.150390625e-06,
      "model_forward_time": 1.744612693786621,
      "step": 4352
    },
    {
      "epoch": 4.150390625e-06,
      "step": 4352,
      "training_step_time": 6.876781702041626
    },
    {
      "epoch": 4.151344299316406e-06,
      "model_forward_time": 1.7493579387664795,
      "step": 4353
    },
    {
      "epoch": 4.151344299316406e-06,
      "step": 4353,
      "training_step_time": 6.805650949478149
    },
    {
      "epoch": 4.152297973632813e-06,
      "model_forward_time": 1.3656995296478271,
      "step": 4354
    },
    {
      "epoch": 4.152297973632813e-06,
      "step": 4354,
      "training_step_time": 6.792703866958618
    },
    {
      "epoch": 4.153251647949219e-06,
      "model_forward_time": 1.7412915229797363,
      "step": 4355
    },
    {
      "epoch": 4.153251647949219e-06,
      "step": 4355,
      "training_step_time": 6.762728691101074
    },
    {
      "epoch": 4.154205322265625e-06,
      "model_forward_time": 1.7457430362701416,
      "step": 4356
    },
    {
      "epoch": 4.154205322265625e-06,
      "step": 4356,
      "training_step_time": 6.791064739227295
    },
    {
      "epoch": 4.1551589965820314e-06,
      "model_forward_time": 1.7396411895751953,
      "step": 4357
    },
    {
      "epoch": 4.1551589965820314e-06,
      "step": 4357,
      "training_step_time": 8.358819007873535
    },
    {
      "epoch": 4.156112670898437e-06,
      "model_forward_time": 3.020770311355591,
      "step": 4358
    },
    {
      "epoch": 4.156112670898437e-06,
      "step": 4358,
      "training_step_time": 10.857114553451538
    },
    {
      "epoch": 4.157066345214844e-06,
      "model_forward_time": 1.884300708770752,
      "step": 4359
    },
    {
      "epoch": 4.157066345214844e-06,
      "step": 4359,
      "training_step_time": 7.947883129119873
    },
    {
      "epoch": 4.15802001953125e-06,
      "grad_norm": 0.46110716462135315,
      "learning_rate": 8.718e-06,
      "loss": 0.0861,
      "step": 4360
    },
    {
      "epoch": 4.15802001953125e-06,
      "model_forward_time": 1.9075202941894531,
      "step": 4360
    },
    {
      "dynamics_loss_avg": 0.07042150162160396,
      "epoch": 4.15802001953125e-06,
      "step": 4360
    },
    {
      "action_loss_avg": 0.007422038551885635,
      "epoch": 4.15802001953125e-06,
      "step": 4360
    },
    {
      "epoch": 4.15802001953125e-06,
      "step": 4360,
      "training_step_time": 7.6412353515625
    },
    {
      "epoch": 4.158973693847656e-06,
      "model_forward_time": 1.9243438243865967,
      "step": 4361
    },
    {
      "epoch": 4.158973693847656e-06,
      "step": 4361,
      "training_step_time": 6.879166126251221
    },
    {
      "epoch": 4.159927368164063e-06,
      "model_forward_time": 1.9109234809875488,
      "step": 4362
    },
    {
      "epoch": 4.159927368164063e-06,
      "step": 4362,
      "training_step_time": 6.801363706588745
    },
    {
      "epoch": 4.160881042480469e-06,
      "model_forward_time": 1.9240164756774902,
      "step": 4363
    },
    {
      "epoch": 4.160881042480469e-06,
      "step": 4363,
      "training_step_time": 6.849538564682007
    },
    {
      "epoch": 4.161834716796875e-06,
      "model_forward_time": 1.9114513397216797,
      "step": 4364
    },
    {
      "epoch": 4.161834716796875e-06,
      "step": 4364,
      "training_step_time": 6.843568325042725
    },
    {
      "epoch": 4.1627883911132815e-06,
      "model_forward_time": 1.898132085800171,
      "step": 4365
    },
    {
      "epoch": 4.1627883911132815e-06,
      "step": 4365,
      "training_step_time": 6.804872035980225
    },
    {
      "epoch": 4.163742065429687e-06,
      "model_forward_time": 1.8983173370361328,
      "step": 4366
    },
    {
      "epoch": 4.163742065429687e-06,
      "step": 4366,
      "training_step_time": 8.535395622253418
    },
    {
      "epoch": 4.164695739746094e-06,
      "model_forward_time": 2.9740819931030273,
      "step": 4367
    },
    {
      "epoch": 4.164695739746094e-06,
      "step": 4367,
      "training_step_time": 8.801072835922241
    },
    {
      "epoch": 4.1656494140625e-06,
      "model_forward_time": 1.9069373607635498,
      "step": 4368
    },
    {
      "epoch": 4.1656494140625e-06,
      "step": 4368,
      "training_step_time": 8.316584348678589
    },
    {
      "epoch": 4.166603088378906e-06,
      "model_forward_time": 1.552626609802246,
      "step": 4369
    },
    {
      "epoch": 4.166603088378906e-06,
      "step": 4369,
      "training_step_time": 7.631017208099365
    },
    {
      "epoch": 4.167556762695313e-06,
      "grad_norm": 0.3595951795578003,
      "learning_rate": 8.738000000000002e-06,
      "loss": 0.0766,
      "step": 4370
    },
    {
      "epoch": 4.167556762695313e-06,
      "model_forward_time": 1.9033586978912354,
      "step": 4370
    },
    {
      "dynamics_loss_avg": 0.07475094916298985,
      "epoch": 4.167556762695313e-06,
      "step": 4370
    },
    {
      "action_loss_avg": 0.008323882217518985,
      "epoch": 4.167556762695313e-06,
      "step": 4370
    },
    {
      "epoch": 4.167556762695313e-06,
      "step": 4370,
      "training_step_time": 6.790708780288696
    },
    {
      "epoch": 4.168510437011719e-06,
      "model_forward_time": 1.871960163116455,
      "step": 4371
    },
    {
      "epoch": 4.168510437011719e-06,
      "step": 4371,
      "training_step_time": 6.7952351570129395
    },
    {
      "epoch": 4.169464111328125e-06,
      "model_forward_time": 1.875387191772461,
      "step": 4372
    },
    {
      "epoch": 4.169464111328125e-06,
      "step": 4372,
      "training_step_time": 6.826409816741943
    },
    {
      "epoch": 4.1704177856445315e-06,
      "model_forward_time": 1.890122413635254,
      "step": 4373
    },
    {
      "epoch": 4.1704177856445315e-06,
      "step": 4373,
      "training_step_time": 6.835860252380371
    },
    {
      "epoch": 4.171371459960937e-06,
      "model_forward_time": 1.8823022842407227,
      "step": 4374
    },
    {
      "epoch": 4.171371459960937e-06,
      "step": 4374,
      "training_step_time": 6.893300533294678
    },
    {
      "epoch": 4.172325134277344e-06,
      "model_forward_time": 1.910212755203247,
      "step": 4375
    },
    {
      "epoch": 4.172325134277344e-06,
      "step": 4375,
      "training_step_time": 8.76953935623169
    },
    {
      "epoch": 4.17327880859375e-06,
      "model_forward_time": 1.8787055015563965,
      "step": 4376
    },
    {
      "epoch": 4.17327880859375e-06,
      "step": 4376,
      "training_step_time": 9.668429613113403
    },
    {
      "epoch": 4.174232482910156e-06,
      "model_forward_time": 2.977640151977539,
      "step": 4377
    },
    {
      "epoch": 4.174232482910156e-06,
      "step": 4377,
      "training_step_time": 9.965078115463257
    },
    {
      "epoch": 4.175186157226563e-06,
      "model_forward_time": 1.8779268264770508,
      "step": 4378
    },
    {
      "epoch": 4.175186157226563e-06,
      "step": 4378,
      "training_step_time": 7.949240446090698
    },
    {
      "epoch": 4.176139831542968e-06,
      "model_forward_time": 1.877397060394287,
      "step": 4379
    },
    {
      "epoch": 4.176139831542968e-06,
      "step": 4379,
      "training_step_time": 6.857848405838013
    },
    {
      "epoch": 4.177093505859375e-06,
      "grad_norm": 0.3521474301815033,
      "learning_rate": 8.758000000000002e-06,
      "loss": 0.0728,
      "step": 4380
    },
    {
      "epoch": 4.177093505859375e-06,
      "model_forward_time": 1.1557397842407227,
      "step": 4380
    },
    {
      "dynamics_loss_avg": 0.05969156101346016,
      "epoch": 4.177093505859375e-06,
      "step": 4380
    },
    {
      "action_loss_avg": 0.009982401144225151,
      "epoch": 4.177093505859375e-06,
      "step": 4380
    },
    {
      "epoch": 4.177093505859375e-06,
      "step": 4380,
      "training_step_time": 6.811006546020508
    },
    {
      "epoch": 4.1780471801757815e-06,
      "model_forward_time": 1.8855299949645996,
      "step": 4381
    },
    {
      "epoch": 4.1780471801757815e-06,
      "step": 4381,
      "training_step_time": 6.86585807800293
    },
    {
      "epoch": 4.179000854492187e-06,
      "model_forward_time": 1.860478162765503,
      "step": 4382
    },
    {
      "epoch": 4.179000854492187e-06,
      "step": 4382,
      "training_step_time": 6.847817420959473
    },
    {
      "epoch": 4.179954528808594e-06,
      "model_forward_time": 1.8997914791107178,
      "step": 4383
    },
    {
      "epoch": 4.179954528808594e-06,
      "step": 4383,
      "training_step_time": 6.833284139633179
    },
    {
      "epoch": 4.180908203125e-06,
      "model_forward_time": 1.8955168724060059,
      "step": 4384
    },
    {
      "epoch": 4.180908203125e-06,
      "step": 4384,
      "training_step_time": 7.831850051879883
    },
    {
      "epoch": 4.181861877441406e-06,
      "model_forward_time": 1.9103448390960693,
      "step": 4385
    },
    {
      "epoch": 4.181861877441406e-06,
      "step": 4385,
      "training_step_time": 9.863231420516968
    },
    {
      "epoch": 4.182815551757813e-06,
      "model_forward_time": 3.0063576698303223,
      "step": 4386
    },
    {
      "epoch": 4.182815551757813e-06,
      "step": 4386,
      "training_step_time": 9.902474641799927
    },
    {
      "epoch": 4.183769226074218e-06,
      "model_forward_time": 1.8830208778381348,
      "step": 4387
    },
    {
      "epoch": 4.183769226074218e-06,
      "step": 4387,
      "training_step_time": 8.000426530838013
    },
    {
      "epoch": 4.184722900390625e-06,
      "model_forward_time": 1.884148359298706,
      "step": 4388
    },
    {
      "epoch": 4.184722900390625e-06,
      "step": 4388,
      "training_step_time": 7.151056528091431
    },
    {
      "epoch": 4.1856765747070315e-06,
      "model_forward_time": 1.8883862495422363,
      "step": 4389
    },
    {
      "epoch": 4.1856765747070315e-06,
      "step": 4389,
      "training_step_time": 6.90924596786499
    },
    {
      "epoch": 4.186630249023437e-06,
      "grad_norm": 0.37171563506126404,
      "learning_rate": 8.778000000000001e-06,
      "loss": 0.0779,
      "step": 4390
    },
    {
      "epoch": 4.186630249023437e-06,
      "model_forward_time": 1.8962891101837158,
      "step": 4390
    },
    {
      "dynamics_loss_avg": 0.05619326364248991,
      "epoch": 4.186630249023437e-06,
      "step": 4390
    },
    {
      "action_loss_avg": 0.007567422231659293,
      "epoch": 4.186630249023437e-06,
      "step": 4390
    },
    {
      "epoch": 4.186630249023437e-06,
      "step": 4390,
      "training_step_time": 6.957696914672852
    },
    {
      "epoch": 4.187583923339844e-06,
      "model_forward_time": 1.8868427276611328,
      "step": 4391
    },
    {
      "epoch": 4.187583923339844e-06,
      "step": 4391,
      "training_step_time": 6.916227340698242
    },
    {
      "epoch": 4.18853759765625e-06,
      "model_forward_time": 1.9028148651123047,
      "step": 4392
    },
    {
      "epoch": 4.18853759765625e-06,
      "step": 4392,
      "training_step_time": 6.920626401901245
    },
    {
      "epoch": 4.189491271972656e-06,
      "model_forward_time": 1.8975236415863037,
      "step": 4393
    },
    {
      "epoch": 4.189491271972656e-06,
      "step": 4393,
      "training_step_time": 7.849299669265747
    },
    {
      "epoch": 4.190444946289063e-06,
      "model_forward_time": 1.8768815994262695,
      "step": 4394
    },
    {
      "epoch": 4.190444946289063e-06,
      "step": 4394,
      "training_step_time": 7.8347766399383545
    },
    {
      "epoch": 4.191398620605468e-06,
      "model_forward_time": 1.8878004550933838,
      "step": 4395
    },
    {
      "epoch": 4.191398620605468e-06,
      "step": 4395,
      "training_step_time": 10.491382598876953
    },
    {
      "epoch": 4.192352294921875e-06,
      "model_forward_time": 1.893831491470337,
      "step": 4396
    },
    {
      "epoch": 4.192352294921875e-06,
      "step": 4396,
      "training_step_time": 8.974700212478638
    },
    {
      "epoch": 4.1933059692382815e-06,
      "model_forward_time": 1.8808650970458984,
      "step": 4397
    },
    {
      "epoch": 4.1933059692382815e-06,
      "step": 4397,
      "training_step_time": 7.113343000411987
    },
    {
      "epoch": 4.194259643554687e-06,
      "model_forward_time": 1.7976346015930176,
      "step": 4398
    },
    {
      "epoch": 4.194259643554687e-06,
      "step": 4398,
      "training_step_time": 6.842174053192139
    },
    {
      "epoch": 4.195213317871094e-06,
      "model_forward_time": 1.7315833568572998,
      "step": 4399
    },
    {
      "epoch": 4.195213317871094e-06,
      "step": 4399,
      "training_step_time": 6.868170499801636
    },
    {
      "epoch": 4.1961669921875e-06,
      "grad_norm": 0.5131518244743347,
      "learning_rate": 8.798000000000001e-06,
      "loss": 0.081,
      "step": 4400
    },
    {
      "epoch": 4.1961669921875e-06,
      "model_forward_time": 1.7258460521697998,
      "step": 4400
    },
    {
      "dynamics_loss_avg": 0.07611658871173858,
      "epoch": 4.1961669921875e-06,
      "step": 4400
    },
    {
      "action_loss_avg": 0.008969120495021344,
      "epoch": 4.1961669921875e-06,
      "step": 4400
    },
    {
      "epoch": 4.1961669921875e-06,
      "step": 4400,
      "training_step_time": 6.938100099563599
    },
    {
      "epoch": 4.197120666503906e-06,
      "model_forward_time": 1.7369251251220703,
      "step": 4401
    },
    {
      "epoch": 4.197120666503906e-06,
      "step": 4401,
      "training_step_time": 6.840214490890503
    },
    {
      "epoch": 4.198074340820313e-06,
      "model_forward_time": 1.7379534244537354,
      "step": 4402
    },
    {
      "epoch": 4.198074340820313e-06,
      "step": 4402,
      "training_step_time": 7.916236877441406
    },
    {
      "epoch": 4.199028015136718e-06,
      "model_forward_time": 1.728586196899414,
      "step": 4403
    },
    {
      "epoch": 4.199028015136718e-06,
      "step": 4403,
      "training_step_time": 8.520659685134888
    },
    {
      "epoch": 4.199981689453125e-06,
      "model_forward_time": 1.7432599067687988,
      "step": 4404
    },
    {
      "epoch": 4.199981689453125e-06,
      "step": 4404,
      "training_step_time": 8.563588619232178
    },
    {
      "epoch": 4.2009353637695315e-06,
      "model_forward_time": 2.85622501373291,
      "step": 4405
    },
    {
      "epoch": 4.2009353637695315e-06,
      "step": 4405,
      "training_step_time": 7.945209264755249
    },
    {
      "epoch": 4.201889038085937e-06,
      "model_forward_time": 1.7369484901428223,
      "step": 4406
    },
    {
      "epoch": 4.201889038085937e-06,
      "step": 4406,
      "training_step_time": 7.605431079864502
    },
    {
      "epoch": 4.202842712402344e-06,
      "model_forward_time": 1.040243148803711,
      "step": 4407
    },
    {
      "epoch": 4.202842712402344e-06,
      "step": 4407,
      "training_step_time": 6.858254671096802
    },
    {
      "epoch": 4.20379638671875e-06,
      "model_forward_time": 1.7371141910552979,
      "step": 4408
    },
    {
      "epoch": 4.20379638671875e-06,
      "step": 4408,
      "training_step_time": 6.870908260345459
    },
    {
      "epoch": 4.204750061035156e-06,
      "model_forward_time": 1.7302765846252441,
      "step": 4409
    },
    {
      "epoch": 4.204750061035156e-06,
      "step": 4409,
      "training_step_time": 6.847522497177124
    },
    {
      "epoch": 4.205703735351563e-06,
      "grad_norm": 0.35105594992637634,
      "learning_rate": 8.818000000000001e-06,
      "loss": 0.0853,
      "step": 4410
    },
    {
      "epoch": 4.205703735351563e-06,
      "model_forward_time": 1.728208065032959,
      "step": 4410
    },
    {
      "dynamics_loss_avg": 0.0685547849163413,
      "epoch": 4.205703735351563e-06,
      "step": 4410
    },
    {
      "action_loss_avg": 0.008583384519442915,
      "epoch": 4.205703735351563e-06,
      "step": 4410
    },
    {
      "epoch": 4.205703735351563e-06,
      "step": 4410,
      "training_step_time": 6.88425874710083
    },
    {
      "epoch": 4.206657409667968e-06,
      "model_forward_time": 1.7335591316223145,
      "step": 4411
    },
    {
      "epoch": 4.206657409667968e-06,
      "step": 4411,
      "training_step_time": 7.875861883163452
    },
    {
      "epoch": 4.207611083984375e-06,
      "model_forward_time": 1.73093581199646,
      "step": 4412
    },
    {
      "epoch": 4.207611083984375e-06,
      "step": 4412,
      "training_step_time": 7.7369959354400635
    },
    {
      "epoch": 4.2085647583007815e-06,
      "model_forward_time": 1.7314631938934326,
      "step": 4413
    },
    {
      "epoch": 4.2085647583007815e-06,
      "step": 4413,
      "training_step_time": 9.456491947174072
    },
    {
      "epoch": 4.209518432617187e-06,
      "model_forward_time": 2.887874126434326,
      "step": 4414
    },
    {
      "epoch": 4.209518432617187e-06,
      "step": 4414,
      "training_step_time": 8.453434705734253
    },
    {
      "epoch": 4.210472106933594e-06,
      "model_forward_time": 1.7333788871765137,
      "step": 4415
    },
    {
      "epoch": 4.210472106933594e-06,
      "step": 4415,
      "training_step_time": 7.701237201690674
    },
    {
      "epoch": 4.21142578125e-06,
      "model_forward_time": 1.824488878250122,
      "step": 4416
    },
    {
      "epoch": 4.21142578125e-06,
      "step": 4416,
      "training_step_time": 6.886897087097168
    },
    {
      "epoch": 4.212379455566406e-06,
      "model_forward_time": 1.722449541091919,
      "step": 4417
    },
    {
      "epoch": 4.212379455566406e-06,
      "step": 4417,
      "training_step_time": 6.890086889266968
    },
    {
      "epoch": 4.213333129882813e-06,
      "model_forward_time": 1.7214446067810059,
      "step": 4418
    },
    {
      "epoch": 4.213333129882813e-06,
      "step": 4418,
      "training_step_time": 6.861336946487427
    },
    {
      "epoch": 4.214286804199218e-06,
      "model_forward_time": 1.707934856414795,
      "step": 4419
    },
    {
      "epoch": 4.214286804199218e-06,
      "step": 4419,
      "training_step_time": 6.863682746887207
    },
    {
      "epoch": 4.215240478515625e-06,
      "grad_norm": 0.6379911303520203,
      "learning_rate": 8.838e-06,
      "loss": 0.0832,
      "step": 4420
    },
    {
      "epoch": 4.215240478515625e-06,
      "model_forward_time": 1.7087032794952393,
      "step": 4420
    },
    {
      "dynamics_loss_avg": 0.07218193337321281,
      "epoch": 4.215240478515625e-06,
      "step": 4420
    },
    {
      "action_loss_avg": 0.011766647780314088,
      "epoch": 4.215240478515625e-06,
      "step": 4420
    },
    {
      "epoch": 4.215240478515625e-06,
      "step": 4420,
      "training_step_time": 7.9372477531433105
    },
    {
      "epoch": 4.2161941528320315e-06,
      "model_forward_time": 1.7302179336547852,
      "step": 4421
    },
    {
      "epoch": 4.2161941528320315e-06,
      "step": 4421,
      "training_step_time": 8.71833062171936
    },
    {
      "epoch": 4.217147827148437e-06,
      "model_forward_time": 1.7839725017547607,
      "step": 4422
    },
    {
      "epoch": 4.217147827148437e-06,
      "step": 4422,
      "training_step_time": 8.92195439338684
    },
    {
      "epoch": 4.218101501464844e-06,
      "model_forward_time": 2.802307367324829,
      "step": 4423
    },
    {
      "epoch": 4.218101501464844e-06,
      "step": 4423,
      "training_step_time": 10.680073499679565
    },
    {
      "epoch": 4.21905517578125e-06,
      "model_forward_time": 1.7236390113830566,
      "step": 4424
    },
    {
      "epoch": 4.21905517578125e-06,
      "step": 4424,
      "training_step_time": 7.784624099731445
    },
    {
      "epoch": 4.220008850097656e-06,
      "model_forward_time": 1.7509353160858154,
      "step": 4425
    },
    {
      "epoch": 4.220008850097656e-06,
      "step": 4425,
      "training_step_time": 7.13417911529541
    },
    {
      "epoch": 4.220962524414063e-06,
      "model_forward_time": 1.7215802669525146,
      "step": 4426
    },
    {
      "epoch": 4.220962524414063e-06,
      "step": 4426,
      "training_step_time": 6.9002790451049805
    },
    {
      "epoch": 4.221916198730468e-06,
      "model_forward_time": 1.7286982536315918,
      "step": 4427
    },
    {
      "epoch": 4.221916198730468e-06,
      "step": 4427,
      "training_step_time": 6.913377046585083
    },
    {
      "epoch": 4.222869873046875e-06,
      "model_forward_time": 1.3569293022155762,
      "step": 4428
    },
    {
      "epoch": 4.222869873046875e-06,
      "step": 4428,
      "training_step_time": 6.891481161117554
    },
    {
      "epoch": 4.2238235473632815e-06,
      "model_forward_time": 1.7250511646270752,
      "step": 4429
    },
    {
      "epoch": 4.2238235473632815e-06,
      "step": 4429,
      "training_step_time": 8.007490634918213
    },
    {
      "epoch": 4.224777221679687e-06,
      "grad_norm": 0.4901767373085022,
      "learning_rate": 8.858e-06,
      "loss": 0.0858,
      "step": 4430
    },
    {
      "epoch": 4.224777221679687e-06,
      "model_forward_time": 1.7207810878753662,
      "step": 4430
    },
    {
      "dynamics_loss_avg": 0.06992506105452775,
      "epoch": 4.224777221679687e-06,
      "step": 4430
    },
    {
      "action_loss_avg": 0.010840311087667942,
      "epoch": 4.224777221679687e-06,
      "step": 4430
    },
    {
      "epoch": 4.224777221679687e-06,
      "step": 4430,
      "training_step_time": 8.510261535644531
    },
    {
      "epoch": 4.225730895996094e-06,
      "model_forward_time": 1.353398084640503,
      "step": 4431
    },
    {
      "epoch": 4.225730895996094e-06,
      "step": 4431,
      "training_step_time": 10.349765539169312
    },
    {
      "epoch": 4.2266845703125e-06,
      "model_forward_time": 1.7925424575805664,
      "step": 4432
    },
    {
      "epoch": 4.2266845703125e-06,
      "step": 4432,
      "training_step_time": 8.518449306488037
    },
    {
      "epoch": 4.227638244628906e-06,
      "model_forward_time": 2.8886795043945312,
      "step": 4433
    },
    {
      "epoch": 4.227638244628906e-06,
      "step": 4433,
      "training_step_time": 7.832021951675415
    },
    {
      "epoch": 4.228591918945313e-06,
      "model_forward_time": 1.7268648147583008,
      "step": 4434
    },
    {
      "epoch": 4.228591918945313e-06,
      "step": 4434,
      "training_step_time": 7.507198095321655
    },
    {
      "epoch": 4.229545593261718e-06,
      "model_forward_time": 1.7205703258514404,
      "step": 4435
    },
    {
      "epoch": 4.229545593261718e-06,
      "step": 4435,
      "training_step_time": 6.786955833435059
    },
    {
      "epoch": 4.230499267578125e-06,
      "model_forward_time": 1.726806640625,
      "step": 4436
    },
    {
      "epoch": 4.230499267578125e-06,
      "step": 4436,
      "training_step_time": 6.788132667541504
    },
    {
      "epoch": 4.2314529418945315e-06,
      "model_forward_time": 1.7531542778015137,
      "step": 4437
    },
    {
      "epoch": 4.2314529418945315e-06,
      "step": 4437,
      "training_step_time": 6.7883172035217285
    },
    {
      "epoch": 4.232406616210937e-06,
      "model_forward_time": 1.7283744812011719,
      "step": 4438
    },
    {
      "epoch": 4.232406616210937e-06,
      "step": 4438,
      "training_step_time": 7.832904100418091
    },
    {
      "epoch": 4.233360290527344e-06,
      "model_forward_time": 1.743793249130249,
      "step": 4439
    },
    {
      "epoch": 4.233360290527344e-06,
      "step": 4439,
      "training_step_time": 7.487989664077759
    },
    {
      "epoch": 4.23431396484375e-06,
      "grad_norm": 0.34454402327537537,
      "learning_rate": 8.878e-06,
      "loss": 0.0796,
      "step": 4440
    },
    {
      "epoch": 4.23431396484375e-06,
      "model_forward_time": 1.7295920848846436,
      "step": 4440
    },
    {
      "dynamics_loss_avg": 0.06523956004530192,
      "epoch": 4.23431396484375e-06,
      "step": 4440
    },
    {
      "action_loss_avg": 0.009736232389695943,
      "epoch": 4.23431396484375e-06,
      "step": 4440
    },
    {
      "epoch": 4.23431396484375e-06,
      "step": 4440,
      "training_step_time": 9.49097228050232
    },
    {
      "epoch": 4.235267639160156e-06,
      "model_forward_time": 1.3793399333953857,
      "step": 4441
    },
    {
      "epoch": 4.235267639160156e-06,
      "step": 4441,
      "training_step_time": 8.258599996566772
    },
    {
      "epoch": 4.236221313476563e-06,
      "model_forward_time": 2.4640636444091797,
      "step": 4442
    },
    {
      "epoch": 4.236221313476563e-06,
      "step": 4442,
      "training_step_time": 10.085484981536865
    },
    {
      "epoch": 4.237174987792968e-06,
      "model_forward_time": 1.737584114074707,
      "step": 4443
    },
    {
      "epoch": 4.237174987792968e-06,
      "step": 4443,
      "training_step_time": 7.443178415298462
    },
    {
      "epoch": 4.238128662109375e-06,
      "model_forward_time": 1.7344415187835693,
      "step": 4444
    },
    {
      "epoch": 4.238128662109375e-06,
      "step": 4444,
      "training_step_time": 6.73262882232666
    },
    {
      "epoch": 4.2390823364257815e-06,
      "model_forward_time": 1.7302112579345703,
      "step": 4445
    },
    {
      "epoch": 4.2390823364257815e-06,
      "step": 4445,
      "training_step_time": 6.73715877532959
    },
    {
      "epoch": 4.240036010742187e-06,
      "model_forward_time": 1.7262794971466064,
      "step": 4446
    },
    {
      "epoch": 4.240036010742187e-06,
      "step": 4446,
      "training_step_time": 6.708816766738892
    },
    {
      "epoch": 4.240989685058594e-06,
      "model_forward_time": 1.7295937538146973,
      "step": 4447
    },
    {
      "epoch": 4.240989685058594e-06,
      "step": 4447,
      "training_step_time": 7.8446080684661865
    },
    {
      "epoch": 4.241943359375e-06,
      "model_forward_time": 1.344036340713501,
      "step": 4448
    },
    {
      "epoch": 4.241943359375e-06,
      "step": 4448,
      "training_step_time": 7.4959681034088135
    },
    {
      "epoch": 4.242897033691406e-06,
      "model_forward_time": 1.3502018451690674,
      "step": 4449
    },
    {
      "epoch": 4.242897033691406e-06,
      "step": 4449,
      "training_step_time": 8.385224103927612
    },
    {
      "epoch": 4.243850708007813e-06,
      "grad_norm": 0.33095666766166687,
      "learning_rate": 8.898000000000002e-06,
      "loss": 0.078,
      "step": 4450
    },
    {
      "epoch": 4.243850708007813e-06,
      "model_forward_time": 1.749150037765503,
      "step": 4450
    },
    {
      "dynamics_loss_avg": 0.06905667996034026,
      "epoch": 4.243850708007813e-06,
      "step": 4450
    },
    {
      "action_loss_avg": 0.011459096171893179,
      "epoch": 4.243850708007813e-06,
      "step": 4450
    },
    {
      "epoch": 4.243850708007813e-06,
      "step": 4450,
      "training_step_time": 7.9335973262786865
    },
    {
      "epoch": 4.244804382324218e-06,
      "model_forward_time": 1.7222228050231934,
      "step": 4451
    },
    {
      "epoch": 4.244804382324218e-06,
      "step": 4451,
      "training_step_time": 8.78687858581543
    },
    {
      "epoch": 4.245758056640625e-06,
      "model_forward_time": 2.8961782455444336,
      "step": 4452
    },
    {
      "epoch": 4.245758056640625e-06,
      "step": 4452,
      "training_step_time": 8.683071851730347
    },
    {
      "epoch": 4.2467117309570315e-06,
      "model_forward_time": 1.7190561294555664,
      "step": 4453
    },
    {
      "epoch": 4.2467117309570315e-06,
      "step": 4453,
      "training_step_time": 6.749858140945435
    },
    {
      "epoch": 4.247665405273437e-06,
      "model_forward_time": 1.7177972793579102,
      "step": 4454
    },
    {
      "epoch": 4.247665405273437e-06,
      "step": 4454,
      "training_step_time": 6.78549337387085
    },
    {
      "epoch": 4.248619079589844e-06,
      "model_forward_time": 1.7207365036010742,
      "step": 4455
    },
    {
      "epoch": 4.248619079589844e-06,
      "step": 4455,
      "training_step_time": 6.76483941078186
    },
    {
      "epoch": 4.24957275390625e-06,
      "model_forward_time": 1.7415661811828613,
      "step": 4456
    },
    {
      "epoch": 4.24957275390625e-06,
      "step": 4456,
      "training_step_time": 7.929868698120117
    },
    {
      "epoch": 4.250526428222656e-06,
      "model_forward_time": 1.7229840755462646,
      "step": 4457
    },
    {
      "epoch": 4.250526428222656e-06,
      "step": 4457,
      "training_step_time": 7.486706018447876
    },
    {
      "epoch": 4.251480102539063e-06,
      "model_forward_time": 1.720475196838379,
      "step": 4458
    },
    {
      "epoch": 4.251480102539063e-06,
      "step": 4458,
      "training_step_time": 7.697203636169434
    },
    {
      "epoch": 4.252433776855468e-06,
      "model_forward_time": 1.7213969230651855,
      "step": 4459
    },
    {
      "epoch": 4.252433776855468e-06,
      "step": 4459,
      "training_step_time": 9.703954696655273
    },
    {
      "epoch": 4.253387451171875e-06,
      "grad_norm": 0.4614342153072357,
      "learning_rate": 8.918000000000002e-06,
      "loss": 0.0797,
      "step": 4460
    },
    {
      "epoch": 4.253387451171875e-06,
      "model_forward_time": 1.0591611862182617,
      "step": 4460
    },
    {
      "dynamics_loss_avg": 0.06407370958477258,
      "epoch": 4.253387451171875e-06,
      "step": 4460
    },
    {
      "action_loss_avg": 0.008558225282467902,
      "epoch": 4.253387451171875e-06,
      "step": 4460
    },
    {
      "epoch": 4.253387451171875e-06,
      "step": 4460,
      "training_step_time": 9.252861022949219
    },
    {
      "epoch": 4.2543411254882815e-06,
      "model_forward_time": 2.859318494796753,
      "step": 4461
    },
    {
      "epoch": 4.2543411254882815e-06,
      "step": 4461,
      "training_step_time": 7.647726774215698
    },
    {
      "epoch": 4.255294799804687e-06,
      "model_forward_time": 1.7268083095550537,
      "step": 4462
    },
    {
      "epoch": 4.255294799804687e-06,
      "step": 4462,
      "training_step_time": 7.5301759243011475
    },
    {
      "epoch": 4.256248474121094e-06,
      "model_forward_time": 1.732560396194458,
      "step": 4463
    },
    {
      "epoch": 4.256248474121094e-06,
      "step": 4463,
      "training_step_time": 6.85148549079895
    },
    {
      "epoch": 4.2572021484375e-06,
      "model_forward_time": 1.7256195545196533,
      "step": 4464
    },
    {
      "epoch": 4.2572021484375e-06,
      "step": 4464,
      "training_step_time": 6.88037371635437
    },
    {
      "epoch": 4.258155822753906e-06,
      "model_forward_time": 1.7326791286468506,
      "step": 4465
    },
    {
      "epoch": 4.258155822753906e-06,
      "step": 4465,
      "training_step_time": 7.912186622619629
    },
    {
      "epoch": 4.259109497070313e-06,
      "model_forward_time": 1.7243518829345703,
      "step": 4466
    },
    {
      "epoch": 4.259109497070313e-06,
      "step": 4466,
      "training_step_time": 8.173346757888794
    },
    {
      "epoch": 4.260063171386718e-06,
      "model_forward_time": 1.7332866191864014,
      "step": 4467
    },
    {
      "epoch": 4.260063171386718e-06,
      "step": 4467,
      "training_step_time": 7.657388210296631
    },
    {
      "epoch": 4.261016845703125e-06,
      "model_forward_time": 1.7315196990966797,
      "step": 4468
    },
    {
      "epoch": 4.261016845703125e-06,
      "step": 4468,
      "training_step_time": 7.954072713851929
    },
    {
      "epoch": 4.2619705200195315e-06,
      "model_forward_time": 1.8113117218017578,
      "step": 4469
    },
    {
      "epoch": 4.2619705200195315e-06,
      "step": 4469,
      "training_step_time": 9.632088899612427
    },
    {
      "epoch": 4.262924194335937e-06,
      "grad_norm": 0.27389708161354065,
      "learning_rate": 8.938000000000001e-06,
      "loss": 0.0774,
      "step": 4470
    },
    {
      "epoch": 4.262924194335937e-06,
      "model_forward_time": 2.8313100337982178,
      "step": 4470
    },
    {
      "dynamics_loss_avg": 0.08832083903253078,
      "epoch": 4.262924194335937e-06,
      "step": 4470
    },
    {
      "action_loss_avg": 0.012137290090322495,
      "epoch": 4.262924194335937e-06,
      "step": 4470
    },
    {
      "epoch": 4.262924194335937e-06,
      "step": 4470,
      "training_step_time": 8.468734741210938
    },
    {
      "epoch": 4.263877868652344e-06,
      "model_forward_time": 1.7298173904418945,
      "step": 4471
    },
    {
      "epoch": 4.263877868652344e-06,
      "step": 4471,
      "training_step_time": 7.675827741622925
    },
    {
      "epoch": 4.26483154296875e-06,
      "model_forward_time": 1.7302124500274658,
      "step": 4472
    },
    {
      "epoch": 4.26483154296875e-06,
      "step": 4472,
      "training_step_time": 6.732940196990967
    },
    {
      "epoch": 4.265785217285156e-06,
      "model_forward_time": 1.7352309226989746,
      "step": 4473
    },
    {
      "epoch": 4.265785217285156e-06,
      "step": 4473,
      "training_step_time": 6.787978649139404
    },
    {
      "epoch": 4.266738891601563e-06,
      "model_forward_time": 1.7270162105560303,
      "step": 4474
    },
    {
      "epoch": 4.266738891601563e-06,
      "step": 4474,
      "training_step_time": 7.321463108062744
    },
    {
      "epoch": 4.267692565917968e-06,
      "model_forward_time": 1.7314538955688477,
      "step": 4475
    },
    {
      "epoch": 4.267692565917968e-06,
      "step": 4475,
      "training_step_time": 7.83929181098938
    },
    {
      "epoch": 4.268646240234375e-06,
      "model_forward_time": 1.730813980102539,
      "step": 4476
    },
    {
      "epoch": 4.268646240234375e-06,
      "step": 4476,
      "training_step_time": 9.104260921478271
    },
    {
      "epoch": 4.2695999145507815e-06,
      "model_forward_time": 1.7239422798156738,
      "step": 4477
    },
    {
      "epoch": 4.2695999145507815e-06,
      "step": 4477,
      "training_step_time": 8.724311351776123
    },
    {
      "epoch": 4.270553588867187e-06,
      "model_forward_time": 1.7950491905212402,
      "step": 4478
    },
    {
      "epoch": 4.270553588867187e-06,
      "step": 4478,
      "training_step_time": 8.225072860717773
    },
    {
      "epoch": 4.271507263183594e-06,
      "model_forward_time": 1.360297441482544,
      "step": 4479
    },
    {
      "epoch": 4.271507263183594e-06,
      "step": 4479,
      "training_step_time": 7.63640570640564
    },
    {
      "epoch": 4.2724609375e-06,
      "grad_norm": 0.416875958442688,
      "learning_rate": 8.958000000000001e-06,
      "loss": 0.0848,
      "step": 4480
    },
    {
      "epoch": 4.2724609375e-06,
      "model_forward_time": 2.858715057373047,
      "step": 4480
    },
    {
      "dynamics_loss_avg": 0.06466865614056587,
      "epoch": 4.2724609375e-06,
      "step": 4480
    },
    {
      "action_loss_avg": 0.009683319018222392,
      "epoch": 4.2724609375e-06,
      "step": 4480
    },
    {
      "epoch": 4.2724609375e-06,
      "step": 4480,
      "training_step_time": 7.774590015411377
    },
    {
      "epoch": 4.273414611816406e-06,
      "model_forward_time": 1.7236149311065674,
      "step": 4481
    },
    {
      "epoch": 4.273414611816406e-06,
      "step": 4481,
      "training_step_time": 7.528314113616943
    },
    {
      "epoch": 4.274368286132813e-06,
      "model_forward_time": 1.716156005859375,
      "step": 4482
    },
    {
      "epoch": 4.274368286132813e-06,
      "step": 4482,
      "training_step_time": 6.805461168289185
    },
    {
      "epoch": 4.275321960449218e-06,
      "model_forward_time": 1.726893424987793,
      "step": 4483
    },
    {
      "epoch": 4.275321960449218e-06,
      "step": 4483,
      "training_step_time": 7.384175062179565
    },
    {
      "epoch": 4.276275634765625e-06,
      "model_forward_time": 1.3937525749206543,
      "step": 4484
    },
    {
      "epoch": 4.276275634765625e-06,
      "step": 4484,
      "training_step_time": 8.09294581413269
    },
    {
      "epoch": 4.2772293090820316e-06,
      "model_forward_time": 1.3510806560516357,
      "step": 4485
    },
    {
      "epoch": 4.2772293090820316e-06,
      "step": 4485,
      "training_step_time": 7.512735605239868
    },
    {
      "epoch": 4.278182983398437e-06,
      "model_forward_time": 1.7308003902435303,
      "step": 4486
    },
    {
      "epoch": 4.278182983398437e-06,
      "step": 4486,
      "training_step_time": 8.765348672866821
    },
    {
      "epoch": 4.279136657714844e-06,
      "model_forward_time": 1.725644826889038,
      "step": 4487
    },
    {
      "epoch": 4.279136657714844e-06,
      "step": 4487,
      "training_step_time": 9.842952728271484
    },
    {
      "epoch": 4.28009033203125e-06,
      "model_forward_time": 1.72804594039917,
      "step": 4488
    },
    {
      "epoch": 4.28009033203125e-06,
      "step": 4488,
      "training_step_time": 10.87569785118103
    },
    {
      "epoch": 4.281044006347656e-06,
      "model_forward_time": 2.8501901626586914,
      "step": 4489
    },
    {
      "epoch": 4.281044006347656e-06,
      "step": 4489,
      "training_step_time": 7.861036062240601
    },
    {
      "epoch": 4.281997680664063e-06,
      "grad_norm": 0.36648184061050415,
      "learning_rate": 8.978000000000001e-06,
      "loss": 0.0815,
      "step": 4490
    },
    {
      "epoch": 4.281997680664063e-06,
      "model_forward_time": 1.720130205154419,
      "step": 4490
    },
    {
      "dynamics_loss_avg": 0.08313130773603916,
      "epoch": 4.281997680664063e-06,
      "step": 4490
    },
    {
      "action_loss_avg": 0.014160024700686336,
      "epoch": 4.281997680664063e-06,
      "step": 4490
    },
    {
      "epoch": 4.281997680664063e-06,
      "step": 4490,
      "training_step_time": 7.743027210235596
    },
    {
      "epoch": 4.2829513549804684e-06,
      "model_forward_time": 1.7306339740753174,
      "step": 4491
    },
    {
      "epoch": 4.2829513549804684e-06,
      "step": 4491,
      "training_step_time": 6.961089611053467
    },
    {
      "epoch": 4.283905029296875e-06,
      "model_forward_time": 1.7279613018035889,
      "step": 4492
    },
    {
      "epoch": 4.283905029296875e-06,
      "step": 4492,
      "training_step_time": 7.556676864624023
    },
    {
      "epoch": 4.2848587036132816e-06,
      "model_forward_time": 1.7242956161499023,
      "step": 4493
    },
    {
      "epoch": 4.2848587036132816e-06,
      "step": 4493,
      "training_step_time": 8.032946348190308
    },
    {
      "epoch": 4.285812377929687e-06,
      "model_forward_time": 1.7236764430999756,
      "step": 4494
    },
    {
      "epoch": 4.285812377929687e-06,
      "step": 4494,
      "training_step_time": 7.759426593780518
    },
    {
      "epoch": 4.286766052246094e-06,
      "model_forward_time": 1.728663682937622,
      "step": 4495
    },
    {
      "epoch": 4.286766052246094e-06,
      "step": 4495,
      "training_step_time": 8.007987260818481
    },
    {
      "epoch": 4.2877197265625e-06,
      "model_forward_time": 1.7281413078308105,
      "step": 4496
    },
    {
      "epoch": 4.2877197265625e-06,
      "step": 4496,
      "training_step_time": 9.470278024673462
    },
    {
      "epoch": 4.288673400878906e-06,
      "model_forward_time": 1.4365358352661133,
      "step": 4497
    },
    {
      "epoch": 4.288673400878906e-06,
      "step": 4497,
      "training_step_time": 8.911039352416992
    },
    {
      "epoch": 4.289627075195313e-06,
      "model_forward_time": 2.7878453731536865,
      "step": 4498
    },
    {
      "epoch": 4.289627075195313e-06,
      "step": 4498,
      "training_step_time": 7.7923948764801025
    },
    {
      "epoch": 4.2905807495117184e-06,
      "model_forward_time": 1.733290433883667,
      "step": 4499
    },
    {
      "epoch": 4.2905807495117184e-06,
      "step": 4499,
      "training_step_time": 7.711281061172485
    },
    {
      "epoch": 4.291534423828125e-06,
      "grad_norm": 0.42919421195983887,
      "learning_rate": 8.998000000000001e-06,
      "loss": 0.0844,
      "step": 4500
    },
    {
      "epoch": 4.291534423828125e-06,
      "model_forward_time": 1.7304954528808594,
      "step": 4500
    },
    {
      "dynamics_loss_avg": 0.08774562664330006,
      "epoch": 4.291534423828125e-06,
      "step": 4500
    },
    {
      "action_loss_avg": 0.010839397693052888,
      "epoch": 4.291534423828125e-06,
      "step": 4500
    },
    {
      "epoch": 4.291534423828125e-06,
      "step": 4500,
      "training_step_time": 7.00489068031311
    },
    {
      "epoch": 4.2924880981445316e-06,
      "model_forward_time": 1.737072467803955,
      "step": 4501
    },
    {
      "epoch": 4.2924880981445316e-06,
      "step": 4501,
      "training_step_time": 7.5439534187316895
    },
    {
      "epoch": 4.293441772460937e-06,
      "model_forward_time": 1.7388334274291992,
      "step": 4502
    },
    {
      "epoch": 4.293441772460937e-06,
      "step": 4502,
      "training_step_time": 6.926443099975586
    },
    {
      "epoch": 4.294395446777344e-06,
      "model_forward_time": 1.7394297122955322,
      "step": 4503
    },
    {
      "epoch": 4.294395446777344e-06,
      "step": 4503,
      "training_step_time": 8.697624921798706
    },
    {
      "epoch": 4.29534912109375e-06,
      "model_forward_time": 1.731421947479248,
      "step": 4504
    },
    {
      "epoch": 4.29534912109375e-06,
      "step": 4504,
      "training_step_time": 8.40701961517334
    },
    {
      "epoch": 4.296302795410156e-06,
      "model_forward_time": 1.733564853668213,
      "step": 4505
    },
    {
      "epoch": 4.296302795410156e-06,
      "step": 4505,
      "training_step_time": 10.956472635269165
    },
    {
      "epoch": 4.297256469726563e-06,
      "model_forward_time": 1.463120460510254,
      "step": 4506
    },
    {
      "epoch": 4.297256469726563e-06,
      "step": 4506,
      "training_step_time": 9.756016254425049
    },
    {
      "epoch": 4.2982101440429684e-06,
      "model_forward_time": 1.7389981746673584,
      "step": 4507
    },
    {
      "epoch": 4.2982101440429684e-06,
      "step": 4507,
      "training_step_time": 7.818544387817383
    },
    {
      "epoch": 4.299163818359375e-06,
      "model_forward_time": 2.9134624004364014,
      "step": 4508
    },
    {
      "epoch": 4.299163818359375e-06,
      "step": 4508,
      "training_step_time": 7.957449674606323
    },
    {
      "epoch": 4.3001174926757816e-06,
      "model_forward_time": 1.3902959823608398,
      "step": 4509
    },
    {
      "epoch": 4.3001174926757816e-06,
      "step": 4509,
      "training_step_time": 6.943271160125732
    },
    {
      "epoch": 4.301071166992187e-06,
      "grad_norm": 0.278099000453949,
      "learning_rate": 9.018e-06,
      "loss": 0.0831,
      "step": 4510
    },
    {
      "epoch": 4.301071166992187e-06,
      "model_forward_time": 1.7500386238098145,
      "step": 4510
    },
    {
      "dynamics_loss_avg": 0.06927069686353207,
      "epoch": 4.301071166992187e-06,
      "step": 4510
    },
    {
      "action_loss_avg": 0.009780608536675572,
      "epoch": 4.301071166992187e-06,
      "step": 4510
    },
    {
      "epoch": 4.301071166992187e-06,
      "step": 4510,
      "training_step_time": 7.512888669967651
    },
    {
      "epoch": 4.302024841308594e-06,
      "model_forward_time": 1.7482116222381592,
      "step": 4511
    },
    {
      "epoch": 4.302024841308594e-06,
      "step": 4511,
      "training_step_time": 7.578786849975586
    },
    {
      "epoch": 4.302978515625e-06,
      "model_forward_time": 1.7472302913665771,
      "step": 4512
    },
    {
      "epoch": 4.302978515625e-06,
      "step": 4512,
      "training_step_time": 9.120387315750122
    },
    {
      "epoch": 4.303932189941406e-06,
      "model_forward_time": 1.7333993911743164,
      "step": 4513
    },
    {
      "epoch": 4.303932189941406e-06,
      "step": 4513,
      "training_step_time": 8.258632898330688
    },
    {
      "epoch": 4.304885864257813e-06,
      "model_forward_time": 1.7330222129821777,
      "step": 4514
    },
    {
      "epoch": 4.304885864257813e-06,
      "step": 4514,
      "training_step_time": 8.333064079284668
    },
    {
      "epoch": 4.3058395385742184e-06,
      "model_forward_time": 1.807999610900879,
      "step": 4515
    },
    {
      "epoch": 4.3058395385742184e-06,
      "step": 4515,
      "training_step_time": 8.669113874435425
    },
    {
      "epoch": 4.306793212890625e-06,
      "model_forward_time": 1.737853765487671,
      "step": 4516
    },
    {
      "epoch": 4.306793212890625e-06,
      "step": 4516,
      "training_step_time": 9.33324408531189
    },
    {
      "epoch": 4.307746887207032e-06,
      "model_forward_time": 2.8435425758361816,
      "step": 4517
    },
    {
      "epoch": 4.307746887207032e-06,
      "step": 4517,
      "training_step_time": 8.026180505752563
    },
    {
      "epoch": 4.308700561523437e-06,
      "model_forward_time": 1.739555835723877,
      "step": 4518
    },
    {
      "epoch": 4.308700561523437e-06,
      "step": 4518,
      "training_step_time": 7.026665925979614
    },
    {
      "epoch": 4.309654235839844e-06,
      "model_forward_time": 1.734776258468628,
      "step": 4519
    },
    {
      "epoch": 4.309654235839844e-06,
      "step": 4519,
      "training_step_time": 7.508944511413574
    },
    {
      "epoch": 4.31060791015625e-06,
      "grad_norm": 0.4806819260120392,
      "learning_rate": 9.038e-06,
      "loss": 0.0767,
      "step": 4520
    },
    {
      "epoch": 4.31060791015625e-06,
      "model_forward_time": 1.744095802307129,
      "step": 4520
    },
    {
      "dynamics_loss_avg": 0.06296107545495033,
      "epoch": 4.31060791015625e-06,
      "step": 4520
    },
    {
      "action_loss_avg": 0.008850140892900527,
      "epoch": 4.31060791015625e-06,
      "step": 4520
    },
    {
      "epoch": 4.31060791015625e-06,
      "step": 4520,
      "training_step_time": 7.608163595199585
    },
    {
      "epoch": 4.311561584472656e-06,
      "model_forward_time": 1.3843984603881836,
      "step": 4521
    },
    {
      "epoch": 4.311561584472656e-06,
      "step": 4521,
      "training_step_time": 8.330279111862183
    },
    {
      "epoch": 4.312515258789063e-06,
      "model_forward_time": 1.7377076148986816,
      "step": 4522
    },
    {
      "epoch": 4.312515258789063e-06,
      "step": 4522,
      "training_step_time": 9.130732536315918
    },
    {
      "epoch": 4.3134689331054685e-06,
      "model_forward_time": 1.738389492034912,
      "step": 4523
    },
    {
      "epoch": 4.3134689331054685e-06,
      "step": 4523,
      "training_step_time": 10.051643133163452
    },
    {
      "epoch": 4.314422607421875e-06,
      "model_forward_time": 1.7989144325256348,
      "step": 4524
    },
    {
      "epoch": 4.314422607421875e-06,
      "step": 4524,
      "training_step_time": 9.57587480545044
    },
    {
      "epoch": 4.315376281738282e-06,
      "model_forward_time": 1.7506098747253418,
      "step": 4525
    },
    {
      "epoch": 4.315376281738282e-06,
      "step": 4525,
      "training_step_time": 8.379354476928711
    },
    {
      "epoch": 4.316329956054687e-06,
      "model_forward_time": 2.789158582687378,
      "step": 4526
    },
    {
      "epoch": 4.316329956054687e-06,
      "step": 4526,
      "training_step_time": 7.79125714302063
    },
    {
      "epoch": 4.317283630371094e-06,
      "model_forward_time": 1.748833179473877,
      "step": 4527
    },
    {
      "epoch": 4.317283630371094e-06,
      "step": 4527,
      "training_step_time": 7.225458145141602
    },
    {
      "epoch": 4.3182373046875e-06,
      "model_forward_time": 1.752744436264038,
      "step": 4528
    },
    {
      "epoch": 4.3182373046875e-06,
      "step": 4528,
      "training_step_time": 7.019994020462036
    },
    {
      "epoch": 4.319190979003906e-06,
      "model_forward_time": 1.7637853622436523,
      "step": 4529
    },
    {
      "epoch": 4.319190979003906e-06,
      "step": 4529,
      "training_step_time": 7.605886459350586
    },
    {
      "epoch": 4.320144653320313e-06,
      "grad_norm": 0.38204434514045715,
      "learning_rate": 9.058000000000002e-06,
      "loss": 0.0718,
      "step": 4530
    },
    {
      "epoch": 4.320144653320313e-06,
      "model_forward_time": 1.7555782794952393,
      "step": 4530
    },
    {
      "dynamics_loss_avg": 0.07373173087835312,
      "epoch": 4.320144653320313e-06,
      "step": 4530
    },
    {
      "action_loss_avg": 0.01030555651523173,
      "epoch": 4.320144653320313e-06,
      "step": 4530
    },
    {
      "epoch": 4.320144653320313e-06,
      "step": 4530,
      "training_step_time": 7.775609254837036
    },
    {
      "epoch": 4.3210983276367185e-06,
      "model_forward_time": 1.7510809898376465,
      "step": 4531
    },
    {
      "epoch": 4.3210983276367185e-06,
      "step": 4531,
      "training_step_time": 8.04759693145752
    },
    {
      "epoch": 4.322052001953125e-06,
      "model_forward_time": 1.7537577152252197,
      "step": 4532
    },
    {
      "epoch": 4.322052001953125e-06,
      "step": 4532,
      "training_step_time": 7.881655693054199
    },
    {
      "epoch": 4.323005676269532e-06,
      "model_forward_time": 1.819995641708374,
      "step": 4533
    },
    {
      "epoch": 4.323005676269532e-06,
      "step": 4533,
      "training_step_time": 8.031218767166138
    },
    {
      "epoch": 4.323959350585937e-06,
      "model_forward_time": 1.7492430210113525,
      "step": 4534
    },
    {
      "epoch": 4.323959350585937e-06,
      "step": 4534,
      "training_step_time": 8.666505575180054
    },
    {
      "epoch": 4.324913024902344e-06,
      "model_forward_time": 1.752922534942627,
      "step": 4535
    },
    {
      "epoch": 4.324913024902344e-06,
      "step": 4535,
      "training_step_time": 7.8230016231536865
    },
    {
      "epoch": 4.32586669921875e-06,
      "model_forward_time": 2.931077003479004,
      "step": 4536
    },
    {
      "epoch": 4.32586669921875e-06,
      "step": 4536,
      "training_step_time": 7.921051979064941
    },
    {
      "epoch": 4.326820373535156e-06,
      "model_forward_time": 1.7422847747802734,
      "step": 4537
    },
    {
      "epoch": 4.326820373535156e-06,
      "step": 4537,
      "training_step_time": 6.971745252609253
    },
    {
      "epoch": 4.327774047851563e-06,
      "model_forward_time": 1.7502927780151367,
      "step": 4538
    },
    {
      "epoch": 4.327774047851563e-06,
      "step": 4538,
      "training_step_time": 7.571978807449341
    },
    {
      "epoch": 4.3287277221679685e-06,
      "model_forward_time": 1.3829259872436523,
      "step": 4539
    },
    {
      "epoch": 4.3287277221679685e-06,
      "step": 4539,
      "training_step_time": 7.626699447631836
    },
    {
      "epoch": 4.329681396484375e-06,
      "grad_norm": 0.3815934360027313,
      "learning_rate": 9.078000000000002e-06,
      "loss": 0.0711,
      "step": 4540
    },
    {
      "epoch": 4.329681396484375e-06,
      "model_forward_time": 1.7532110214233398,
      "step": 4540
    },
    {
      "dynamics_loss_avg": 0.06856811922043562,
      "epoch": 4.329681396484375e-06,
      "step": 4540
    },
    {
      "action_loss_avg": 0.009070069063454866,
      "epoch": 4.329681396484375e-06,
      "step": 4540
    },
    {
      "epoch": 4.329681396484375e-06,
      "step": 4540,
      "training_step_time": 8.131545782089233
    },
    {
      "epoch": 4.330635070800782e-06,
      "model_forward_time": 1.7536628246307373,
      "step": 4541
    },
    {
      "epoch": 4.330635070800782e-06,
      "step": 4541,
      "training_step_time": 8.461738109588623
    },
    {
      "epoch": 4.331588745117187e-06,
      "model_forward_time": 1.787423849105835,
      "step": 4542
    },
    {
      "epoch": 4.331588745117187e-06,
      "step": 4542,
      "training_step_time": 9.433967590332031
    },
    {
      "epoch": 4.332542419433594e-06,
      "model_forward_time": 1.7379405498504639,
      "step": 4543
    },
    {
      "epoch": 4.332542419433594e-06,
      "step": 4543,
      "training_step_time": 10.728878498077393
    },
    {
      "epoch": 4.33349609375e-06,
      "model_forward_time": 1.3834729194641113,
      "step": 4544
    },
    {
      "epoch": 4.33349609375e-06,
      "step": 4544,
      "training_step_time": 7.815356969833374
    },
    {
      "epoch": 4.334449768066406e-06,
      "model_forward_time": 2.84134840965271,
      "step": 4545
    },
    {
      "epoch": 4.334449768066406e-06,
      "step": 4545,
      "training_step_time": 7.800535440444946
    },
    {
      "epoch": 4.335403442382813e-06,
      "model_forward_time": 1.7292323112487793,
      "step": 4546
    },
    {
      "epoch": 4.335403442382813e-06,
      "step": 4546,
      "training_step_time": 6.934513092041016
    },
    {
      "epoch": 4.3363571166992185e-06,
      "model_forward_time": 1.7401227951049805,
      "step": 4547
    },
    {
      "epoch": 4.3363571166992185e-06,
      "step": 4547,
      "training_step_time": 7.561950922012329
    },
    {
      "epoch": 4.337310791015625e-06,
      "model_forward_time": 1.3739454746246338,
      "step": 4548
    },
    {
      "epoch": 4.337310791015625e-06,
      "step": 4548,
      "training_step_time": 7.6429197788238525
    },
    {
      "epoch": 4.338264465332032e-06,
      "model_forward_time": 1.733198642730713,
      "step": 4549
    },
    {
      "epoch": 4.338264465332032e-06,
      "step": 4549,
      "training_step_time": 8.276973962783813
    },
    {
      "epoch": 4.339218139648437e-06,
      "grad_norm": 0.3478084206581116,
      "learning_rate": 9.098000000000002e-06,
      "loss": 0.0688,
      "step": 4550
    },
    {
      "epoch": 4.339218139648437e-06,
      "model_forward_time": 1.7495262622833252,
      "step": 4550
    },
    {
      "dynamics_loss_avg": 0.04597004912793636,
      "epoch": 4.339218139648437e-06,
      "step": 4550
    },
    {
      "action_loss_avg": 0.0064587606582790615,
      "epoch": 4.339218139648437e-06,
      "step": 4550
    },
    {
      "epoch": 4.339218139648437e-06,
      "step": 4550,
      "training_step_time": 8.729634761810303
    },
    {
      "epoch": 4.340171813964844e-06,
      "model_forward_time": 1.7924728393554688,
      "step": 4551
    },
    {
      "epoch": 4.340171813964844e-06,
      "step": 4551,
      "training_step_time": 8.026987791061401
    },
    {
      "epoch": 4.34112548828125e-06,
      "model_forward_time": 1.739518165588379,
      "step": 4552
    },
    {
      "epoch": 4.34112548828125e-06,
      "step": 4552,
      "training_step_time": 8.965287208557129
    },
    {
      "epoch": 4.342079162597656e-06,
      "model_forward_time": 1.7418851852416992,
      "step": 4553
    },
    {
      "epoch": 4.342079162597656e-06,
      "step": 4553,
      "training_step_time": 7.822818040847778
    },
    {
      "epoch": 4.343032836914063e-06,
      "model_forward_time": 2.7923965454101562,
      "step": 4554
    },
    {
      "epoch": 4.343032836914063e-06,
      "step": 4554,
      "training_step_time": 7.829370021820068
    },
    {
      "epoch": 4.3439865112304685e-06,
      "model_forward_time": 1.7283315658569336,
      "step": 4555
    },
    {
      "epoch": 4.3439865112304685e-06,
      "step": 4555,
      "training_step_time": 7.2496418952941895
    },
    {
      "epoch": 4.344940185546875e-06,
      "model_forward_time": 1.7397620677947998,
      "step": 4556
    },
    {
      "epoch": 4.344940185546875e-06,
      "step": 4556,
      "training_step_time": 7.562504529953003
    },
    {
      "epoch": 4.345893859863282e-06,
      "model_forward_time": 1.7398028373718262,
      "step": 4557
    },
    {
      "epoch": 4.345893859863282e-06,
      "step": 4557,
      "training_step_time": 7.61230206489563
    },
    {
      "epoch": 4.346847534179687e-06,
      "model_forward_time": 1.7259690761566162,
      "step": 4558
    },
    {
      "epoch": 4.346847534179687e-06,
      "step": 4558,
      "training_step_time": 7.705949544906616
    },
    {
      "epoch": 4.347801208496094e-06,
      "model_forward_time": 1.7343719005584717,
      "step": 4559
    },
    {
      "epoch": 4.347801208496094e-06,
      "step": 4559,
      "training_step_time": 8.619115591049194
    },
    {
      "epoch": 4.3487548828125e-06,
      "grad_norm": 0.3706601858139038,
      "learning_rate": 9.118000000000001e-06,
      "loss": 0.0717,
      "step": 4560
    },
    {
      "epoch": 4.3487548828125e-06,
      "model_forward_time": 1.7968053817749023,
      "step": 4560
    },
    {
      "dynamics_loss_avg": 0.06155950743705034,
      "epoch": 4.3487548828125e-06,
      "step": 4560
    },
    {
      "action_loss_avg": 0.007724713697098195,
      "epoch": 4.3487548828125e-06,
      "step": 4560
    },
    {
      "epoch": 4.3487548828125e-06,
      "step": 4560,
      "training_step_time": 9.368506669998169
    },
    {
      "epoch": 4.349708557128906e-06,
      "model_forward_time": 1.7411391735076904,
      "step": 4561
    },
    {
      "epoch": 4.349708557128906e-06,
      "step": 4561,
      "training_step_time": 8.706172943115234
    },
    {
      "epoch": 4.350662231445313e-06,
      "model_forward_time": 1.73930025100708,
      "step": 4562
    },
    {
      "epoch": 4.350662231445313e-06,
      "step": 4562,
      "training_step_time": 7.865904808044434
    },
    {
      "epoch": 4.3516159057617185e-06,
      "model_forward_time": 1.727769374847412,
      "step": 4563
    },
    {
      "epoch": 4.3516159057617185e-06,
      "step": 4563,
      "training_step_time": 8.037886142730713
    },
    {
      "epoch": 4.352569580078125e-06,
      "model_forward_time": 2.8778269290924072,
      "step": 4564
    },
    {
      "epoch": 4.352569580078125e-06,
      "step": 4564,
      "training_step_time": 7.890916347503662
    },
    {
      "epoch": 4.353523254394532e-06,
      "model_forward_time": 1.3749949932098389,
      "step": 4565
    },
    {
      "epoch": 4.353523254394532e-06,
      "step": 4565,
      "training_step_time": 7.522444486618042
    },
    {
      "epoch": 4.354476928710937e-06,
      "model_forward_time": 1.7524113655090332,
      "step": 4566
    },
    {
      "epoch": 4.354476928710937e-06,
      "step": 4566,
      "training_step_time": 7.590947151184082
    },
    {
      "epoch": 4.355430603027344e-06,
      "model_forward_time": 1.3708369731903076,
      "step": 4567
    },
    {
      "epoch": 4.355430603027344e-06,
      "step": 4567,
      "training_step_time": 8.806350231170654
    },
    {
      "epoch": 4.35638427734375e-06,
      "model_forward_time": 1.7234342098236084,
      "step": 4568
    },
    {
      "epoch": 4.35638427734375e-06,
      "step": 4568,
      "training_step_time": 8.112740278244019
    },
    {
      "epoch": 4.357337951660156e-06,
      "model_forward_time": 1.795243740081787,
      "step": 4569
    },
    {
      "epoch": 4.357337951660156e-06,
      "step": 4569,
      "training_step_time": 9.39735460281372
    },
    {
      "epoch": 4.358291625976563e-06,
      "grad_norm": 0.3732761740684509,
      "learning_rate": 9.138e-06,
      "loss": 0.077,
      "step": 4570
    },
    {
      "epoch": 4.358291625976563e-06,
      "model_forward_time": 1.3793299198150635,
      "step": 4570
    },
    {
      "dynamics_loss_avg": 0.06306672543287277,
      "epoch": 4.358291625976563e-06,
      "step": 4570
    },
    {
      "action_loss_avg": 0.006287807715125382,
      "epoch": 4.358291625976563e-06,
      "step": 4570
    },
    {
      "epoch": 4.358291625976563e-06,
      "step": 4570,
      "training_step_time": 8.141579627990723
    },
    {
      "epoch": 4.3592453002929685e-06,
      "model_forward_time": 1.745208740234375,
      "step": 4571
    },
    {
      "epoch": 4.3592453002929685e-06,
      "step": 4571,
      "training_step_time": 8.996271848678589
    },
    {
      "epoch": 4.360198974609375e-06,
      "model_forward_time": 1.7395570278167725,
      "step": 4572
    },
    {
      "epoch": 4.360198974609375e-06,
      "step": 4572,
      "training_step_time": 7.9375669956207275
    },
    {
      "epoch": 4.361152648925782e-06,
      "model_forward_time": 2.8119144439697266,
      "step": 4573
    },
    {
      "epoch": 4.361152648925782e-06,
      "step": 4573,
      "training_step_time": 7.817825794219971
    },
    {
      "epoch": 4.362106323242187e-06,
      "model_forward_time": 1.7291679382324219,
      "step": 4574
    },
    {
      "epoch": 4.362106323242187e-06,
      "step": 4574,
      "training_step_time": 7.009796857833862
    },
    {
      "epoch": 4.363059997558594e-06,
      "model_forward_time": 1.7361154556274414,
      "step": 4575
    },
    {
      "epoch": 4.363059997558594e-06,
      "step": 4575,
      "training_step_time": 7.640810966491699
    },
    {
      "epoch": 4.364013671875e-06,
      "model_forward_time": 1.8947968482971191,
      "step": 4576
    },
    {
      "epoch": 4.364013671875e-06,
      "step": 4576,
      "training_step_time": 7.773299932479858
    },
    {
      "epoch": 4.364967346191406e-06,
      "model_forward_time": 1.8673899173736572,
      "step": 4577
    },
    {
      "epoch": 4.364967346191406e-06,
      "step": 4577,
      "training_step_time": 7.80277943611145
    },
    {
      "epoch": 4.365921020507813e-06,
      "model_forward_time": 1.898695468902588,
      "step": 4578
    },
    {
      "epoch": 4.365921020507813e-06,
      "step": 4578,
      "training_step_time": 7.925158977508545
    },
    {
      "epoch": 4.3668746948242185e-06,
      "model_forward_time": 1.8871064186096191,
      "step": 4579
    },
    {
      "epoch": 4.3668746948242185e-06,
      "step": 4579,
      "training_step_time": 8.917623519897461
    },
    {
      "epoch": 4.367828369140625e-06,
      "grad_norm": 0.3370128571987152,
      "learning_rate": 9.158e-06,
      "loss": 0.0731,
      "step": 4580
    },
    {
      "epoch": 4.367828369140625e-06,
      "model_forward_time": 1.906076431274414,
      "step": 4580
    },
    {
      "dynamics_loss_avg": 0.05413112808018923,
      "epoch": 4.367828369140625e-06,
      "step": 4580
    },
    {
      "action_loss_avg": 0.005977596482262015,
      "epoch": 4.367828369140625e-06,
      "step": 4580
    },
    {
      "epoch": 4.367828369140625e-06,
      "step": 4580,
      "training_step_time": 10.063488721847534
    },
    {
      "epoch": 4.368782043457032e-06,
      "model_forward_time": 1.883091926574707,
      "step": 4581
    },
    {
      "epoch": 4.368782043457032e-06,
      "step": 4581,
      "training_step_time": 7.916476488113403
    },
    {
      "epoch": 4.369735717773437e-06,
      "model_forward_time": 1.8950119018554688,
      "step": 4582
    },
    {
      "epoch": 4.369735717773437e-06,
      "step": 4582,
      "training_step_time": 8.217475175857544
    },
    {
      "epoch": 4.370689392089844e-06,
      "model_forward_time": 1.7343385219573975,
      "step": 4583
    },
    {
      "epoch": 4.370689392089844e-06,
      "step": 4583,
      "training_step_time": 7.264934062957764
    },
    {
      "epoch": 4.37164306640625e-06,
      "model_forward_time": 1.756958246231079,
      "step": 4584
    },
    {
      "epoch": 4.37164306640625e-06,
      "step": 4584,
      "training_step_time": 7.615309953689575
    },
    {
      "epoch": 4.372596740722656e-06,
      "model_forward_time": 1.7368359565734863,
      "step": 4585
    },
    {
      "epoch": 4.372596740722656e-06,
      "step": 4585,
      "training_step_time": 7.7755725383758545
    },
    {
      "epoch": 4.373550415039063e-06,
      "model_forward_time": 1.7431871891021729,
      "step": 4586
    },
    {
      "epoch": 4.373550415039063e-06,
      "step": 4586,
      "training_step_time": 7.6892406940460205
    },
    {
      "epoch": 4.3745040893554685e-06,
      "model_forward_time": 1.7546091079711914,
      "step": 4587
    },
    {
      "epoch": 4.3745040893554685e-06,
      "step": 4587,
      "training_step_time": 8.591070890426636
    },
    {
      "epoch": 4.375457763671875e-06,
      "model_forward_time": 1.726804494857788,
      "step": 4588
    },
    {
      "epoch": 4.375457763671875e-06,
      "step": 4588,
      "training_step_time": 8.720855236053467
    },
    {
      "epoch": 4.376411437988282e-06,
      "model_forward_time": 1.753636121749878,
      "step": 4589
    },
    {
      "epoch": 4.376411437988282e-06,
      "step": 4589,
      "training_step_time": 8.921411037445068
    },
    {
      "epoch": 4.377365112304687e-06,
      "grad_norm": 0.28066790103912354,
      "learning_rate": 9.178000000000001e-06,
      "loss": 0.0805,
      "step": 4590
    },
    {
      "epoch": 4.377365112304687e-06,
      "model_forward_time": 1.7328686714172363,
      "step": 4590
    },
    {
      "dynamics_loss_avg": 0.06141809299588204,
      "epoch": 4.377365112304687e-06,
      "step": 4590
    },
    {
      "action_loss_avg": 0.007340638991445303,
      "epoch": 4.377365112304687e-06,
      "step": 4590
    },
    {
      "epoch": 4.377365112304687e-06,
      "step": 4590,
      "training_step_time": 7.768951892852783
    },
    {
      "epoch": 4.378318786621094e-06,
      "model_forward_time": 1.7847199440002441,
      "step": 4591
    },
    {
      "epoch": 4.378318786621094e-06,
      "step": 4591,
      "training_step_time": 7.792311191558838
    },
    {
      "epoch": 4.3792724609375e-06,
      "model_forward_time": 2.8853514194488525,
      "step": 4592
    },
    {
      "epoch": 4.3792724609375e-06,
      "step": 4592,
      "training_step_time": 7.844095468521118
    },
    {
      "epoch": 4.380226135253906e-06,
      "model_forward_time": 1.739135503768921,
      "step": 4593
    },
    {
      "epoch": 4.380226135253906e-06,
      "step": 4593,
      "training_step_time": 7.577112197875977
    },
    {
      "epoch": 4.381179809570313e-06,
      "model_forward_time": 1.731044054031372,
      "step": 4594
    },
    {
      "epoch": 4.381179809570313e-06,
      "step": 4594,
      "training_step_time": 7.7746100425720215
    },
    {
      "epoch": 4.3821334838867185e-06,
      "model_forward_time": 1.7260870933532715,
      "step": 4595
    },
    {
      "epoch": 4.3821334838867185e-06,
      "step": 4595,
      "training_step_time": 7.809534311294556
    },
    {
      "epoch": 4.383087158203125e-06,
      "model_forward_time": 1.3618550300598145,
      "step": 4596
    },
    {
      "epoch": 4.383087158203125e-06,
      "step": 4596,
      "training_step_time": 8.76089596748352
    },
    {
      "epoch": 4.384040832519532e-06,
      "model_forward_time": 1.7448081970214844,
      "step": 4597
    },
    {
      "epoch": 4.384040832519532e-06,
      "step": 4597,
      "training_step_time": 8.833444356918335
    },
    {
      "epoch": 4.384994506835937e-06,
      "model_forward_time": 1.7372844219207764,
      "step": 4598
    },
    {
      "epoch": 4.384994506835937e-06,
      "step": 4598,
      "training_step_time": 9.9189293384552
    },
    {
      "epoch": 4.385948181152344e-06,
      "model_forward_time": 1.7542593479156494,
      "step": 4599
    },
    {
      "epoch": 4.385948181152344e-06,
      "step": 4599,
      "training_step_time": 7.808941602706909
    },
    {
      "epoch": 4.38690185546875e-06,
      "grad_norm": 0.4010827839374542,
      "learning_rate": 9.198e-06,
      "loss": 0.0822,
      "step": 4600
    },
    {
      "epoch": 4.38690185546875e-06,
      "model_forward_time": 1.7486228942871094,
      "step": 4600
    },
    {
      "dynamics_loss_avg": 0.08435162231326103,
      "epoch": 4.38690185546875e-06,
      "step": 4600
    },
    {
      "action_loss_avg": 0.008527452452108264,
      "epoch": 4.38690185546875e-06,
      "step": 4600
    },
    {
      "epoch": 4.38690185546875e-06,
      "step": 4600,
      "training_step_time": 7.97067928314209
    },
    {
      "epoch": 4.387855529785156e-06,
      "model_forward_time": 2.838378429412842,
      "step": 4601
    },
    {
      "epoch": 4.387855529785156e-06,
      "step": 4601,
      "training_step_time": 7.867171287536621
    },
    {
      "epoch": 4.388809204101563e-06,
      "model_forward_time": 1.7258813381195068,
      "step": 4602
    },
    {
      "epoch": 4.388809204101563e-06,
      "step": 4602,
      "training_step_time": 8.234047651290894
    },
    {
      "epoch": 4.3897628784179685e-06,
      "model_forward_time": 1.7354896068572998,
      "step": 4603
    },
    {
      "epoch": 4.3897628784179685e-06,
      "step": 4603,
      "training_step_time": 7.773036241531372
    },
    {
      "epoch": 4.390716552734375e-06,
      "model_forward_time": 1.754251480102539,
      "step": 4604
    },
    {
      "epoch": 4.390716552734375e-06,
      "step": 4604,
      "training_step_time": 8.57056736946106
    },
    {
      "epoch": 4.391670227050782e-06,
      "model_forward_time": 1.74955415725708,
      "step": 4605
    },
    {
      "epoch": 4.391670227050782e-06,
      "step": 4605,
      "training_step_time": 9.520539999008179
    },
    {
      "epoch": 4.392623901367187e-06,
      "model_forward_time": 1.7517094612121582,
      "step": 4606
    },
    {
      "epoch": 4.392623901367187e-06,
      "step": 4606,
      "training_step_time": 8.020729064941406
    },
    {
      "epoch": 4.393577575683594e-06,
      "model_forward_time": 1.733696460723877,
      "step": 4607
    },
    {
      "epoch": 4.393577575683594e-06,
      "step": 4607,
      "training_step_time": 9.032639503479004
    },
    {
      "epoch": 4.39453125e-06,
      "model_forward_time": 1.7651863098144531,
      "step": 4608
    },
    {
      "epoch": 4.39453125e-06,
      "step": 4608,
      "training_step_time": 8.876038551330566
    },
    {
      "epoch": 4.395484924316406e-06,
      "model_forward_time": 1.731386661529541,
      "step": 4609
    },
    {
      "epoch": 4.395484924316406e-06,
      "step": 4609,
      "training_step_time": 8.711389064788818
    },
    {
      "epoch": 4.396438598632813e-06,
      "grad_norm": 0.382968932390213,
      "learning_rate": 9.218e-06,
      "loss": 0.0737,
      "step": 4610
    },
    {
      "epoch": 4.396438598632813e-06,
      "model_forward_time": 2.8347818851470947,
      "step": 4610
    },
    {
      "dynamics_loss_avg": 0.0727352224290371,
      "epoch": 4.396438598632813e-06,
      "step": 4610
    },
    {
      "action_loss_avg": 0.011832028115168213,
      "epoch": 4.396438598632813e-06,
      "step": 4610
    },
    {
      "epoch": 4.396438598632813e-06,
      "step": 4610,
      "training_step_time": 7.890960693359375
    },
    {
      "epoch": 4.3973922729492185e-06,
      "model_forward_time": 1.736691951751709,
      "step": 4611
    },
    {
      "epoch": 4.3973922729492185e-06,
      "step": 4611,
      "training_step_time": 7.494027137756348
    },
    {
      "epoch": 4.398345947265625e-06,
      "model_forward_time": 1.3809168338775635,
      "step": 4612
    },
    {
      "epoch": 4.398345947265625e-06,
      "step": 4612,
      "training_step_time": 7.795493841171265
    },
    {
      "epoch": 4.399299621582032e-06,
      "model_forward_time": 1.7482407093048096,
      "step": 4613
    },
    {
      "epoch": 4.399299621582032e-06,
      "step": 4613,
      "training_step_time": 7.869264841079712
    },
    {
      "epoch": 4.400253295898437e-06,
      "model_forward_time": 1.7451879978179932,
      "step": 4614
    },
    {
      "epoch": 4.400253295898437e-06,
      "step": 4614,
      "training_step_time": 8.81603717803955
    },
    {
      "epoch": 4.401206970214844e-06,
      "model_forward_time": 1.7167420387268066,
      "step": 4615
    },
    {
      "epoch": 4.401206970214844e-06,
      "step": 4615,
      "training_step_time": 10.629472494125366
    },
    {
      "epoch": 4.40216064453125e-06,
      "model_forward_time": 1.7422552108764648,
      "step": 4616
    },
    {
      "epoch": 4.40216064453125e-06,
      "step": 4616,
      "training_step_time": 7.936514139175415
    },
    {
      "epoch": 4.403114318847656e-06,
      "model_forward_time": 1.7244689464569092,
      "step": 4617
    },
    {
      "epoch": 4.403114318847656e-06,
      "step": 4617,
      "training_step_time": 7.952004432678223
    },
    {
      "epoch": 4.404067993164063e-06,
      "model_forward_time": 1.7273962497711182,
      "step": 4618
    },
    {
      "epoch": 4.404067993164063e-06,
      "step": 4618,
      "training_step_time": 7.82349157333374
    },
    {
      "epoch": 4.4050216674804685e-06,
      "model_forward_time": 1.7323322296142578,
      "step": 4619
    },
    {
      "epoch": 4.4050216674804685e-06,
      "step": 4619,
      "training_step_time": 9.302276849746704
    },
    {
      "epoch": 4.405975341796875e-06,
      "grad_norm": 0.1727057844400406,
      "learning_rate": 9.238e-06,
      "loss": 0.0758,
      "step": 4620
    },
    {
      "epoch": 4.405975341796875e-06,
      "model_forward_time": 1.794111728668213,
      "step": 4620
    },
    {
      "dynamics_loss_avg": 0.062095524929463866,
      "epoch": 4.405975341796875e-06,
      "step": 4620
    },
    {
      "action_loss_avg": 0.006771438987925649,
      "epoch": 4.405975341796875e-06,
      "step": 4620
    },
    {
      "epoch": 4.405975341796875e-06,
      "step": 4620,
      "training_step_time": 6.987359523773193
    },
    {
      "epoch": 4.406929016113282e-06,
      "model_forward_time": 1.7541708946228027,
      "step": 4621
    },
    {
      "epoch": 4.406929016113282e-06,
      "step": 4621,
      "training_step_time": 7.593853235244751
    },
    {
      "epoch": 4.407882690429687e-06,
      "model_forward_time": 1.7314229011535645,
      "step": 4622
    },
    {
      "epoch": 4.407882690429687e-06,
      "step": 4622,
      "training_step_time": 8.492196559906006
    },
    {
      "epoch": 4.408836364746094e-06,
      "model_forward_time": 1.7416930198669434,
      "step": 4623
    },
    {
      "epoch": 4.408836364746094e-06,
      "step": 4623,
      "training_step_time": 7.7823169231414795
    },
    {
      "epoch": 4.4097900390625e-06,
      "model_forward_time": 1.7404778003692627,
      "step": 4624
    },
    {
      "epoch": 4.4097900390625e-06,
      "step": 4624,
      "training_step_time": 9.116005659103394
    },
    {
      "epoch": 4.410743713378906e-06,
      "model_forward_time": 1.7518315315246582,
      "step": 4625
    },
    {
      "epoch": 4.410743713378906e-06,
      "step": 4625,
      "training_step_time": 9.904269218444824
    },
    {
      "epoch": 4.411697387695313e-06,
      "model_forward_time": 1.7369894981384277,
      "step": 4626
    },
    {
      "epoch": 4.411697387695313e-06,
      "step": 4626,
      "training_step_time": 9.667228937149048
    },
    {
      "epoch": 4.4126510620117185e-06,
      "model_forward_time": 1.7290997505187988,
      "step": 4627
    },
    {
      "epoch": 4.4126510620117185e-06,
      "step": 4627,
      "training_step_time": 7.785032033920288
    },
    {
      "epoch": 4.413604736328125e-06,
      "model_forward_time": 1.7310991287231445,
      "step": 4628
    },
    {
      "epoch": 4.413604736328125e-06,
      "step": 4628,
      "training_step_time": 7.659606456756592
    },
    {
      "epoch": 4.414558410644531e-06,
      "model_forward_time": 2.9119338989257812,
      "step": 4629
    },
    {
      "epoch": 4.414558410644531e-06,
      "step": 4629,
      "training_step_time": 7.87883186340332
    },
    {
      "epoch": 4.415512084960937e-06,
      "grad_norm": 0.4150242507457733,
      "learning_rate": 9.258e-06,
      "loss": 0.0749,
      "step": 4630
    },
    {
      "epoch": 4.415512084960937e-06,
      "model_forward_time": 1.733567476272583,
      "step": 4630
    },
    {
      "dynamics_loss_avg": 0.07303621862083673,
      "epoch": 4.415512084960937e-06,
      "step": 4630
    },
    {
      "action_loss_avg": 0.00725254041608423,
      "epoch": 4.415512084960937e-06,
      "step": 4630
    },
    {
      "epoch": 4.415512084960937e-06,
      "step": 4630,
      "training_step_time": 7.6032960414886475
    },
    {
      "epoch": 4.416465759277344e-06,
      "model_forward_time": 1.7242941856384277,
      "step": 4631
    },
    {
      "epoch": 4.416465759277344e-06,
      "step": 4631,
      "training_step_time": 7.707355499267578
    },
    {
      "epoch": 4.41741943359375e-06,
      "model_forward_time": 1.7368028163909912,
      "step": 4632
    },
    {
      "epoch": 4.41741943359375e-06,
      "step": 4632,
      "training_step_time": 8.492542505264282
    },
    {
      "epoch": 4.418373107910156e-06,
      "model_forward_time": 1.3611576557159424,
      "step": 4633
    },
    {
      "epoch": 4.418373107910156e-06,
      "step": 4633,
      "training_step_time": 9.509187698364258
    },
    {
      "epoch": 4.419326782226563e-06,
      "model_forward_time": 1.7264881134033203,
      "step": 4634
    },
    {
      "epoch": 4.419326782226563e-06,
      "step": 4634,
      "training_step_time": 8.531363010406494
    },
    {
      "epoch": 4.4202804565429685e-06,
      "model_forward_time": 1.7420830726623535,
      "step": 4635
    },
    {
      "epoch": 4.4202804565429685e-06,
      "step": 4635,
      "training_step_time": 7.83698296546936
    },
    {
      "epoch": 4.421234130859375e-06,
      "model_forward_time": 1.7364344596862793,
      "step": 4636
    },
    {
      "epoch": 4.421234130859375e-06,
      "step": 4636,
      "training_step_time": 9.395080089569092
    },
    {
      "epoch": 4.422187805175781e-06,
      "model_forward_time": 1.7582569122314453,
      "step": 4637
    },
    {
      "epoch": 4.422187805175781e-06,
      "step": 4637,
      "training_step_time": 8.698198795318604
    },
    {
      "epoch": 4.423141479492187e-06,
      "model_forward_time": 2.8690624237060547,
      "step": 4638
    },
    {
      "epoch": 4.423141479492187e-06,
      "step": 4638,
      "training_step_time": 7.8347718715667725
    },
    {
      "epoch": 4.424095153808594e-06,
      "model_forward_time": 1.729651927947998,
      "step": 4639
    },
    {
      "epoch": 4.424095153808594e-06,
      "step": 4639,
      "training_step_time": 7.57049036026001
    },
    {
      "epoch": 4.425048828125e-06,
      "grad_norm": 0.38244062662124634,
      "learning_rate": 9.278e-06,
      "loss": 0.0749,
      "step": 4640
    },
    {
      "epoch": 4.425048828125e-06,
      "model_forward_time": 1.727461576461792,
      "step": 4640
    },
    {
      "dynamics_loss_avg": 0.050647514127194884,
      "epoch": 4.425048828125e-06,
      "step": 4640
    },
    {
      "action_loss_avg": 0.0070940795470960435,
      "epoch": 4.425048828125e-06,
      "step": 4640
    },
    {
      "epoch": 4.425048828125e-06,
      "step": 4640,
      "training_step_time": 8.417145490646362
    },
    {
      "epoch": 4.426002502441406e-06,
      "model_forward_time": 1.7295846939086914,
      "step": 4641
    },
    {
      "epoch": 4.426002502441406e-06,
      "step": 4641,
      "training_step_time": 8.538954734802246
    },
    {
      "epoch": 4.426956176757813e-06,
      "model_forward_time": 1.7394540309906006,
      "step": 4642
    },
    {
      "epoch": 4.426956176757813e-06,
      "step": 4642,
      "training_step_time": 8.951328992843628
    },
    {
      "epoch": 4.4279098510742186e-06,
      "model_forward_time": 1.7497971057891846,
      "step": 4643
    },
    {
      "epoch": 4.4279098510742186e-06,
      "step": 4643,
      "training_step_time": 9.010555505752563
    },
    {
      "epoch": 4.428863525390625e-06,
      "model_forward_time": 1.7313861846923828,
      "step": 4644
    },
    {
      "epoch": 4.428863525390625e-06,
      "step": 4644,
      "training_step_time": 8.379384279251099
    },
    {
      "epoch": 4.429817199707031e-06,
      "model_forward_time": 1.735163688659668,
      "step": 4645
    },
    {
      "epoch": 4.429817199707031e-06,
      "step": 4645,
      "training_step_time": 8.396315336227417
    },
    {
      "epoch": 4.430770874023437e-06,
      "model_forward_time": 1.7258474826812744,
      "step": 4646
    },
    {
      "epoch": 4.430770874023437e-06,
      "step": 4646,
      "training_step_time": 7.79354190826416
    },
    {
      "epoch": 4.431724548339844e-06,
      "model_forward_time": 2.551051139831543,
      "step": 4647
    },
    {
      "epoch": 4.431724548339844e-06,
      "step": 4647,
      "training_step_time": 7.8066301345825195
    },
    {
      "epoch": 4.43267822265625e-06,
      "model_forward_time": 1.7355360984802246,
      "step": 4648
    },
    {
      "epoch": 4.43267822265625e-06,
      "step": 4648,
      "training_step_time": 7.568262577056885
    },
    {
      "epoch": 4.433631896972656e-06,
      "model_forward_time": 1.7308804988861084,
      "step": 4649
    },
    {
      "epoch": 4.433631896972656e-06,
      "step": 4649,
      "training_step_time": 7.736067056655884
    },
    {
      "epoch": 4.434585571289063e-06,
      "grad_norm": 0.3489099442958832,
      "learning_rate": 9.298e-06,
      "loss": 0.0785,
      "step": 4650
    },
    {
      "epoch": 4.434585571289063e-06,
      "model_forward_time": 1.7206361293792725,
      "step": 4650
    },
    {
      "dynamics_loss_avg": 0.06787772104144096,
      "epoch": 4.434585571289063e-06,
      "step": 4650
    },
    {
      "action_loss_avg": 0.008381085796281695,
      "epoch": 4.434585571289063e-06,
      "step": 4650
    },
    {
      "epoch": 4.434585571289063e-06,
      "step": 4650,
      "training_step_time": 7.8327155113220215
    },
    {
      "epoch": 4.4355392456054686e-06,
      "model_forward_time": 1.7410247325897217,
      "step": 4651
    },
    {
      "epoch": 4.4355392456054686e-06,
      "step": 4651,
      "training_step_time": 8.578579664230347
    },
    {
      "epoch": 4.436492919921875e-06,
      "model_forward_time": 1.7802550792694092,
      "step": 4652
    },
    {
      "epoch": 4.436492919921875e-06,
      "step": 4652,
      "training_step_time": 8.547616004943848
    },
    {
      "epoch": 4.437446594238281e-06,
      "model_forward_time": 1.7352502346038818,
      "step": 4653
    },
    {
      "epoch": 4.437446594238281e-06,
      "step": 4653,
      "training_step_time": 8.076118230819702
    },
    {
      "epoch": 4.438400268554687e-06,
      "model_forward_time": 1.7372822761535645,
      "step": 4654
    },
    {
      "epoch": 4.438400268554687e-06,
      "step": 4654,
      "training_step_time": 8.422949314117432
    },
    {
      "epoch": 4.439353942871094e-06,
      "model_forward_time": 1.7345564365386963,
      "step": 4655
    },
    {
      "epoch": 4.439353942871094e-06,
      "step": 4655,
      "training_step_time": 7.800279378890991
    },
    {
      "epoch": 4.4403076171875e-06,
      "model_forward_time": 1.735205888748169,
      "step": 4656
    },
    {
      "epoch": 4.4403076171875e-06,
      "step": 4656,
      "training_step_time": 8.393377542495728
    },
    {
      "epoch": 4.441261291503906e-06,
      "model_forward_time": 1.8193469047546387,
      "step": 4657
    },
    {
      "epoch": 4.441261291503906e-06,
      "step": 4657,
      "training_step_time": 7.620840311050415
    },
    {
      "epoch": 4.442214965820313e-06,
      "model_forward_time": 1.7312407493591309,
      "step": 4658
    },
    {
      "epoch": 4.442214965820313e-06,
      "step": 4658,
      "training_step_time": 7.728581428527832
    },
    {
      "epoch": 4.4431686401367186e-06,
      "model_forward_time": 1.7376983165740967,
      "step": 4659
    },
    {
      "epoch": 4.4431686401367186e-06,
      "step": 4659,
      "training_step_time": 8.653386354446411
    },
    {
      "epoch": 4.444122314453125e-06,
      "grad_norm": 0.2661701738834381,
      "learning_rate": 9.318e-06,
      "loss": 0.0801,
      "step": 4660
    },
    {
      "epoch": 4.444122314453125e-06,
      "model_forward_time": 1.7256903648376465,
      "step": 4660
    },
    {
      "dynamics_loss_avg": 0.0600889328867197,
      "epoch": 4.444122314453125e-06,
      "step": 4660
    },
    {
      "action_loss_avg": 0.00680974202696234,
      "epoch": 4.444122314453125e-06,
      "step": 4660
    },
    {
      "epoch": 4.444122314453125e-06,
      "step": 4660,
      "training_step_time": 8.795114994049072
    },
    {
      "epoch": 4.445075988769531e-06,
      "model_forward_time": 1.7447113990783691,
      "step": 4661
    },
    {
      "epoch": 4.445075988769531e-06,
      "step": 4661,
      "training_step_time": 8.294433355331421
    },
    {
      "epoch": 4.4460296630859374e-06,
      "model_forward_time": 1.3730101585388184,
      "step": 4662
    },
    {
      "epoch": 4.4460296630859374e-06,
      "step": 4662,
      "training_step_time": 9.523758172988892
    },
    {
      "epoch": 4.446983337402344e-06,
      "model_forward_time": 1.734367847442627,
      "step": 4663
    },
    {
      "epoch": 4.446983337402344e-06,
      "step": 4663,
      "training_step_time": 10.839266777038574
    },
    {
      "epoch": 4.44793701171875e-06,
      "model_forward_time": 1.7337052822113037,
      "step": 4664
    },
    {
      "epoch": 4.44793701171875e-06,
      "step": 4664,
      "training_step_time": 7.799900054931641
    },
    {
      "epoch": 4.448890686035156e-06,
      "model_forward_time": 1.730720043182373,
      "step": 4665
    },
    {
      "epoch": 4.448890686035156e-06,
      "step": 4665,
      "training_step_time": 7.870699882507324
    },
    {
      "epoch": 4.449844360351563e-06,
      "model_forward_time": 2.943964719772339,
      "step": 4666
    },
    {
      "epoch": 4.449844360351563e-06,
      "step": 4666,
      "training_step_time": 7.94677209854126
    },
    {
      "epoch": 4.4507980346679686e-06,
      "model_forward_time": 1.7577579021453857,
      "step": 4667
    },
    {
      "epoch": 4.4507980346679686e-06,
      "step": 4667,
      "training_step_time": 7.767683744430542
    },
    {
      "epoch": 4.451751708984375e-06,
      "model_forward_time": 1.7427797317504883,
      "step": 4668
    },
    {
      "epoch": 4.451751708984375e-06,
      "step": 4668,
      "training_step_time": 7.683604001998901
    },
    {
      "epoch": 4.452705383300781e-06,
      "model_forward_time": 1.737865924835205,
      "step": 4669
    },
    {
      "epoch": 4.452705383300781e-06,
      "step": 4669,
      "training_step_time": 8.07207727432251
    },
    {
      "epoch": 4.4536590576171874e-06,
      "grad_norm": 0.3187793493270874,
      "learning_rate": 9.338000000000001e-06,
      "loss": 0.0835,
      "step": 4670
    },
    {
      "epoch": 4.4536590576171874e-06,
      "model_forward_time": 1.7259652614593506,
      "step": 4670
    },
    {
      "dynamics_loss_avg": 0.07412923388183117,
      "epoch": 4.4536590576171874e-06,
      "step": 4670
    },
    {
      "action_loss_avg": 0.00982555621303618,
      "epoch": 4.4536590576171874e-06,
      "step": 4670
    },
    {
      "epoch": 4.4536590576171874e-06,
      "step": 4670,
      "training_step_time": 8.772008180618286
    },
    {
      "epoch": 4.454612731933594e-06,
      "model_forward_time": 1.7312095165252686,
      "step": 4671
    },
    {
      "epoch": 4.454612731933594e-06,
      "step": 4671,
      "training_step_time": 8.103448867797852
    },
    {
      "epoch": 4.45556640625e-06,
      "model_forward_time": 1.7697956562042236,
      "step": 4672
    },
    {
      "epoch": 4.45556640625e-06,
      "step": 4672,
      "training_step_time": 8.626692056655884
    },
    {
      "epoch": 4.456520080566406e-06,
      "model_forward_time": 1.7593994140625,
      "step": 4673
    },
    {
      "epoch": 4.456520080566406e-06,
      "step": 4673,
      "training_step_time": 7.8000242710113525
    },
    {
      "epoch": 4.457473754882813e-06,
      "model_forward_time": 1.7265164852142334,
      "step": 4674
    },
    {
      "epoch": 4.457473754882813e-06,
      "step": 4674,
      "training_step_time": 7.818162441253662
    },
    {
      "epoch": 4.458427429199219e-06,
      "model_forward_time": 2.891139268875122,
      "step": 4675
    },
    {
      "epoch": 4.458427429199219e-06,
      "step": 4675,
      "training_step_time": 7.977042198181152
    },
    {
      "epoch": 4.459381103515625e-06,
      "model_forward_time": 1.495985746383667,
      "step": 4676
    },
    {
      "epoch": 4.459381103515625e-06,
      "step": 4676,
      "training_step_time": 7.727930307388306
    },
    {
      "epoch": 4.460334777832031e-06,
      "model_forward_time": 1.3889875411987305,
      "step": 4677
    },
    {
      "epoch": 4.460334777832031e-06,
      "step": 4677,
      "training_step_time": 8.610787630081177
    },
    {
      "epoch": 4.4612884521484374e-06,
      "model_forward_time": 1.731924295425415,
      "step": 4678
    },
    {
      "epoch": 4.4612884521484374e-06,
      "step": 4678,
      "training_step_time": 7.79203200340271
    },
    {
      "epoch": 4.462242126464844e-06,
      "model_forward_time": 1.7264137268066406,
      "step": 4679
    },
    {
      "epoch": 4.462242126464844e-06,
      "step": 4679,
      "training_step_time": 8.700703144073486
    },
    {
      "epoch": 4.46319580078125e-06,
      "grad_norm": 0.4012905955314636,
      "learning_rate": 9.358000000000001e-06,
      "loss": 0.0754,
      "step": 4680
    },
    {
      "epoch": 4.46319580078125e-06,
      "model_forward_time": 1.7304763793945312,
      "step": 4680
    },
    {
      "dynamics_loss_avg": 0.08168029449880124,
      "epoch": 4.46319580078125e-06,
      "step": 4680
    },
    {
      "action_loss_avg": 0.011511838249862193,
      "epoch": 4.46319580078125e-06,
      "step": 4680
    },
    {
      "epoch": 4.46319580078125e-06,
      "step": 4680,
      "training_step_time": 10.151906967163086
    },
    {
      "epoch": 4.464149475097656e-06,
      "model_forward_time": 1.7416229248046875,
      "step": 4681
    },
    {
      "epoch": 4.464149475097656e-06,
      "step": 4681,
      "training_step_time": 7.839147567749023
    },
    {
      "epoch": 4.465103149414063e-06,
      "model_forward_time": 1.735243320465088,
      "step": 4682
    },
    {
      "epoch": 4.465103149414063e-06,
      "step": 4682,
      "training_step_time": 8.79918384552002
    },
    {
      "epoch": 4.466056823730469e-06,
      "model_forward_time": 1.7335548400878906,
      "step": 4683
    },
    {
      "epoch": 4.466056823730469e-06,
      "step": 4683,
      "training_step_time": 9.248697996139526
    },
    {
      "epoch": 4.467010498046875e-06,
      "model_forward_time": 2.818638801574707,
      "step": 4684
    },
    {
      "epoch": 4.467010498046875e-06,
      "step": 4684,
      "training_step_time": 8.72182846069336
    },
    {
      "epoch": 4.467964172363281e-06,
      "model_forward_time": 1.7302775382995605,
      "step": 4685
    },
    {
      "epoch": 4.467964172363281e-06,
      "step": 4685,
      "training_step_time": 7.909214973449707
    },
    {
      "epoch": 4.4689178466796874e-06,
      "model_forward_time": 1.740527868270874,
      "step": 4686
    },
    {
      "epoch": 4.4689178466796874e-06,
      "step": 4686,
      "training_step_time": 7.686171054840088
    },
    {
      "epoch": 4.469871520996094e-06,
      "model_forward_time": 1.7329697608947754,
      "step": 4687
    },
    {
      "epoch": 4.469871520996094e-06,
      "step": 4687,
      "training_step_time": 8.478060007095337
    },
    {
      "epoch": 4.4708251953125e-06,
      "model_forward_time": 1.7270267009735107,
      "step": 4688
    },
    {
      "epoch": 4.4708251953125e-06,
      "step": 4688,
      "training_step_time": 8.97753620147705
    },
    {
      "epoch": 4.471778869628906e-06,
      "model_forward_time": 1.7447936534881592,
      "step": 4689
    },
    {
      "epoch": 4.471778869628906e-06,
      "step": 4689,
      "training_step_time": 9.066894292831421
    },
    {
      "epoch": 4.472732543945313e-06,
      "grad_norm": 0.29325082898139954,
      "learning_rate": 9.378e-06,
      "loss": 0.0735,
      "step": 4690
    },
    {
      "epoch": 4.472732543945313e-06,
      "model_forward_time": 1.761763572692871,
      "step": 4690
    },
    {
      "dynamics_loss_avg": 0.0851132769137621,
      "epoch": 4.472732543945313e-06,
      "step": 4690
    },
    {
      "action_loss_avg": 0.00882570412941277,
      "epoch": 4.472732543945313e-06,
      "step": 4690
    },
    {
      "epoch": 4.472732543945313e-06,
      "step": 4690,
      "training_step_time": 9.494062662124634
    },
    {
      "epoch": 4.473686218261719e-06,
      "model_forward_time": 1.7273635864257812,
      "step": 4691
    },
    {
      "epoch": 4.473686218261719e-06,
      "step": 4691,
      "training_step_time": 7.8561365604400635
    },
    {
      "epoch": 4.474639892578125e-06,
      "model_forward_time": 1.7322077751159668,
      "step": 4692
    },
    {
      "epoch": 4.474639892578125e-06,
      "step": 4692,
      "training_step_time": 8.397159814834595
    },
    {
      "epoch": 4.475593566894531e-06,
      "model_forward_time": 2.8050291538238525,
      "step": 4693
    },
    {
      "epoch": 4.475593566894531e-06,
      "step": 4693,
      "training_step_time": 8.828792572021484
    },
    {
      "epoch": 4.4765472412109375e-06,
      "model_forward_time": 1.4981863498687744,
      "step": 4694
    },
    {
      "epoch": 4.4765472412109375e-06,
      "step": 4694,
      "training_step_time": 7.927923679351807
    },
    {
      "epoch": 4.477500915527344e-06,
      "model_forward_time": 1.7376902103424072,
      "step": 4695
    },
    {
      "epoch": 4.477500915527344e-06,
      "step": 4695,
      "training_step_time": 7.715261936187744
    },
    {
      "epoch": 4.47845458984375e-06,
      "model_forward_time": 1.7291767597198486,
      "step": 4696
    },
    {
      "epoch": 4.47845458984375e-06,
      "step": 4696,
      "training_step_time": 9.37171459197998
    },
    {
      "epoch": 4.479408264160156e-06,
      "model_forward_time": 1.7262263298034668,
      "step": 4697
    },
    {
      "epoch": 4.479408264160156e-06,
      "step": 4697,
      "training_step_time": 8.980637788772583
    },
    {
      "epoch": 4.480361938476563e-06,
      "model_forward_time": 1.7341554164886475,
      "step": 4698
    },
    {
      "epoch": 4.480361938476563e-06,
      "step": 4698,
      "training_step_time": 7.990520238876343
    },
    {
      "epoch": 4.481315612792969e-06,
      "model_forward_time": 1.7484605312347412,
      "step": 4699
    },
    {
      "epoch": 4.481315612792969e-06,
      "step": 4699,
      "training_step_time": 8.89192509651184
    },
    {
      "epoch": 4.482269287109375e-06,
      "grad_norm": 0.36391451954841614,
      "learning_rate": 9.398e-06,
      "loss": 0.0755,
      "step": 4700
    },
    {
      "epoch": 4.482269287109375e-06,
      "model_forward_time": 1.7359237670898438,
      "step": 4700
    },
    {
      "dynamics_loss_avg": 0.06837501302361489,
      "epoch": 4.482269287109375e-06,
      "step": 4700
    },
    {
      "action_loss_avg": 0.009400479297619313,
      "epoch": 4.482269287109375e-06,
      "step": 4700
    },
    {
      "epoch": 4.482269287109375e-06,
      "step": 4700,
      "training_step_time": 8.96028208732605
    },
    {
      "epoch": 4.483222961425781e-06,
      "model_forward_time": 1.734910488128662,
      "step": 4701
    },
    {
      "epoch": 4.483222961425781e-06,
      "step": 4701,
      "training_step_time": 8.86471700668335
    },
    {
      "epoch": 4.4841766357421875e-06,
      "model_forward_time": 1.759310007095337,
      "step": 4702
    },
    {
      "epoch": 4.4841766357421875e-06,
      "step": 4702,
      "training_step_time": 7.6304404735565186
    },
    {
      "epoch": 4.485130310058594e-06,
      "model_forward_time": 2.830518960952759,
      "step": 4703
    },
    {
      "epoch": 4.485130310058594e-06,
      "step": 4703,
      "training_step_time": 7.96028208732605
    },
    {
      "epoch": 4.486083984375e-06,
      "model_forward_time": 1.7684173583984375,
      "step": 4704
    },
    {
      "epoch": 4.486083984375e-06,
      "step": 4704,
      "training_step_time": 7.664838790893555
    },
    {
      "epoch": 4.487037658691406e-06,
      "model_forward_time": 1.7532703876495361,
      "step": 4705
    },
    {
      "epoch": 4.487037658691406e-06,
      "step": 4705,
      "training_step_time": 7.751196622848511
    },
    {
      "epoch": 4.487991333007813e-06,
      "model_forward_time": 1.738067388534546,
      "step": 4706
    },
    {
      "epoch": 4.487991333007813e-06,
      "step": 4706,
      "training_step_time": 9.010815382003784
    },
    {
      "epoch": 4.488945007324219e-06,
      "model_forward_time": 1.7918589115142822,
      "step": 4707
    },
    {
      "epoch": 4.488945007324219e-06,
      "step": 4707,
      "training_step_time": 9.161027431488037
    },
    {
      "epoch": 4.489898681640625e-06,
      "model_forward_time": 1.744494915008545,
      "step": 4708
    },
    {
      "epoch": 4.489898681640625e-06,
      "step": 4708,
      "training_step_time": 8.026838541030884
    },
    {
      "epoch": 4.490852355957031e-06,
      "model_forward_time": 1.7317759990692139,
      "step": 4709
    },
    {
      "epoch": 4.490852355957031e-06,
      "step": 4709,
      "training_step_time": 7.9947028160095215
    },
    {
      "epoch": 4.4918060302734375e-06,
      "grad_norm": 0.3193359971046448,
      "learning_rate": 9.418e-06,
      "loss": 0.0767,
      "step": 4710
    },
    {
      "epoch": 4.4918060302734375e-06,
      "model_forward_time": 1.7365190982818604,
      "step": 4710
    },
    {
      "dynamics_loss_avg": 0.0636225501075387,
      "epoch": 4.4918060302734375e-06,
      "step": 4710
    },
    {
      "action_loss_avg": 0.007663826271891594,
      "epoch": 4.4918060302734375e-06,
      "step": 4710
    },
    {
      "epoch": 4.4918060302734375e-06,
      "step": 4710,
      "training_step_time": 7.90098762512207
    },
    {
      "epoch": 4.492759704589844e-06,
      "model_forward_time": 1.766920804977417,
      "step": 4711
    },
    {
      "epoch": 4.492759704589844e-06,
      "step": 4711,
      "training_step_time": 7.791910409927368
    },
    {
      "epoch": 4.49371337890625e-06,
      "model_forward_time": 2.9513096809387207,
      "step": 4712
    },
    {
      "epoch": 4.49371337890625e-06,
      "step": 4712,
      "training_step_time": 8.000443458557129
    },
    {
      "epoch": 4.494667053222656e-06,
      "model_forward_time": 1.763495922088623,
      "step": 4713
    },
    {
      "epoch": 4.494667053222656e-06,
      "step": 4713,
      "training_step_time": 7.591383218765259
    },
    {
      "epoch": 4.495620727539063e-06,
      "model_forward_time": 1.7296619415283203,
      "step": 4714
    },
    {
      "epoch": 4.495620727539063e-06,
      "step": 4714,
      "training_step_time": 7.784600019454956
    },
    {
      "epoch": 4.496574401855469e-06,
      "model_forward_time": 1.7385144233703613,
      "step": 4715
    },
    {
      "epoch": 4.496574401855469e-06,
      "step": 4715,
      "training_step_time": 8.823848009109497
    },
    {
      "epoch": 4.497528076171875e-06,
      "model_forward_time": 1.7582101821899414,
      "step": 4716
    },
    {
      "epoch": 4.497528076171875e-06,
      "step": 4716,
      "training_step_time": 7.875758647918701
    },
    {
      "epoch": 4.498481750488281e-06,
      "model_forward_time": 1.7293510437011719,
      "step": 4717
    },
    {
      "epoch": 4.498481750488281e-06,
      "step": 4717,
      "training_step_time": 9.559412956237793
    },
    {
      "epoch": 4.4994354248046875e-06,
      "model_forward_time": 1.3726234436035156,
      "step": 4718
    },
    {
      "epoch": 4.4994354248046875e-06,
      "step": 4718,
      "training_step_time": 7.922073602676392
    },
    {
      "epoch": 4.500389099121094e-06,
      "model_forward_time": 1.7331881523132324,
      "step": 4719
    },
    {
      "epoch": 4.500389099121094e-06,
      "step": 4719,
      "training_step_time": 8.53201675415039
    },
    {
      "epoch": 4.5013427734375e-06,
      "grad_norm": 0.3118096590042114,
      "learning_rate": 9.438e-06,
      "loss": 0.0826,
      "step": 4720
    },
    {
      "epoch": 4.5013427734375e-06,
      "model_forward_time": 1.724992275238037,
      "step": 4720
    },
    {
      "dynamics_loss_avg": 0.0650635689496994,
      "epoch": 4.5013427734375e-06,
      "step": 4720
    },
    {
      "action_loss_avg": 0.005521739844698459,
      "epoch": 4.5013427734375e-06,
      "step": 4720
    },
    {
      "epoch": 4.5013427734375e-06,
      "step": 4720,
      "training_step_time": 7.7640204429626465
    },
    {
      "epoch": 4.502296447753906e-06,
      "model_forward_time": 2.834719657897949,
      "step": 4721
    },
    {
      "epoch": 4.502296447753906e-06,
      "step": 4721,
      "training_step_time": 8.98207974433899
    },
    {
      "epoch": 4.503250122070313e-06,
      "model_forward_time": 1.5068495273590088,
      "step": 4722
    },
    {
      "epoch": 4.503250122070313e-06,
      "step": 4722,
      "training_step_time": 7.821834564208984
    },
    {
      "epoch": 4.504203796386719e-06,
      "model_forward_time": 1.7226266860961914,
      "step": 4723
    },
    {
      "epoch": 4.504203796386719e-06,
      "step": 4723,
      "training_step_time": 7.804659843444824
    },
    {
      "epoch": 4.505157470703125e-06,
      "model_forward_time": 1.739433765411377,
      "step": 4724
    },
    {
      "epoch": 4.505157470703125e-06,
      "step": 4724,
      "training_step_time": 8.29398798942566
    },
    {
      "epoch": 4.506111145019531e-06,
      "model_forward_time": 1.0311024188995361,
      "step": 4725
    },
    {
      "epoch": 4.506111145019531e-06,
      "step": 4725,
      "training_step_time": 8.702162981033325
    },
    {
      "epoch": 4.5070648193359375e-06,
      "model_forward_time": 1.7255558967590332,
      "step": 4726
    },
    {
      "epoch": 4.5070648193359375e-06,
      "step": 4726,
      "training_step_time": 7.874354600906372
    },
    {
      "epoch": 4.508018493652344e-06,
      "model_forward_time": 1.748861312866211,
      "step": 4727
    },
    {
      "epoch": 4.508018493652344e-06,
      "step": 4727,
      "training_step_time": 8.713006496429443
    },
    {
      "epoch": 4.50897216796875e-06,
      "model_forward_time": 1.7328369617462158,
      "step": 4728
    },
    {
      "epoch": 4.50897216796875e-06,
      "step": 4728,
      "training_step_time": 8.589895486831665
    },
    {
      "epoch": 4.509925842285156e-06,
      "model_forward_time": 1.7413098812103271,
      "step": 4729
    },
    {
      "epoch": 4.509925842285156e-06,
      "step": 4729,
      "training_step_time": 7.575206995010376
    },
    {
      "epoch": 4.510879516601563e-06,
      "grad_norm": 0.4514041543006897,
      "learning_rate": 9.458e-06,
      "loss": 0.0757,
      "step": 4730
    },
    {
      "epoch": 4.510879516601563e-06,
      "model_forward_time": 1.726494550704956,
      "step": 4730
    },
    {
      "dynamics_loss_avg": 0.06422876566648483,
      "epoch": 4.510879516601563e-06,
      "step": 4730
    },
    {
      "action_loss_avg": 0.006557067716494202,
      "epoch": 4.510879516601563e-06,
      "step": 4730
    },
    {
      "epoch": 4.510879516601563e-06,
      "step": 4730,
      "training_step_time": 7.6397998332977295
    },
    {
      "epoch": 4.511833190917969e-06,
      "model_forward_time": 2.894249439239502,
      "step": 4731
    },
    {
      "epoch": 4.511833190917969e-06,
      "step": 4731,
      "training_step_time": 7.685873508453369
    },
    {
      "epoch": 4.512786865234375e-06,
      "model_forward_time": 1.806114912033081,
      "step": 4732
    },
    {
      "epoch": 4.512786865234375e-06,
      "step": 4732,
      "training_step_time": 7.495072364807129
    },
    {
      "epoch": 4.513740539550781e-06,
      "model_forward_time": 1.7376461029052734,
      "step": 4733
    },
    {
      "epoch": 4.513740539550781e-06,
      "step": 4733,
      "training_step_time": 7.55661940574646
    },
    {
      "epoch": 4.5146942138671875e-06,
      "model_forward_time": 1.7367970943450928,
      "step": 4734
    },
    {
      "epoch": 4.5146942138671875e-06,
      "step": 4734,
      "training_step_time": 8.093963861465454
    },
    {
      "epoch": 4.515647888183594e-06,
      "model_forward_time": 1.7682437896728516,
      "step": 4735
    },
    {
      "epoch": 4.515647888183594e-06,
      "step": 4735,
      "training_step_time": 8.588184833526611
    },
    {
      "epoch": 4.5166015625e-06,
      "model_forward_time": 1.7473177909851074,
      "step": 4736
    },
    {
      "epoch": 4.5166015625e-06,
      "step": 4736,
      "training_step_time": 7.87368369102478
    },
    {
      "epoch": 4.517555236816406e-06,
      "model_forward_time": 1.7378370761871338,
      "step": 4737
    },
    {
      "epoch": 4.517555236816406e-06,
      "step": 4737,
      "training_step_time": 8.495671510696411
    },
    {
      "epoch": 4.518508911132813e-06,
      "model_forward_time": 1.7457480430603027,
      "step": 4738
    },
    {
      "epoch": 4.518508911132813e-06,
      "step": 4738,
      "training_step_time": 7.637365341186523
    },
    {
      "epoch": 4.519462585449219e-06,
      "model_forward_time": 1.739701509475708,
      "step": 4739
    },
    {
      "epoch": 4.519462585449219e-06,
      "step": 4739,
      "training_step_time": 8.540016889572144
    },
    {
      "epoch": 4.520416259765625e-06,
      "grad_norm": 0.4084186553955078,
      "learning_rate": 9.478e-06,
      "loss": 0.0662,
      "step": 4740
    },
    {
      "epoch": 4.520416259765625e-06,
      "model_forward_time": 2.8985235691070557,
      "step": 4740
    },
    {
      "dynamics_loss_avg": 0.07558174803853035,
      "epoch": 4.520416259765625e-06,
      "step": 4740
    },
    {
      "action_loss_avg": 0.008958217361941934,
      "epoch": 4.520416259765625e-06,
      "step": 4740
    },
    {
      "epoch": 4.520416259765625e-06,
      "step": 4740,
      "training_step_time": 7.679733514785767
    },
    {
      "epoch": 4.521369934082031e-06,
      "model_forward_time": 1.749108076095581,
      "step": 4741
    },
    {
      "epoch": 4.521369934082031e-06,
      "step": 4741,
      "training_step_time": 7.410330772399902
    },
    {
      "epoch": 4.5223236083984375e-06,
      "model_forward_time": 1.722599744796753,
      "step": 4742
    },
    {
      "epoch": 4.5223236083984375e-06,
      "step": 4742,
      "training_step_time": 7.609203815460205
    },
    {
      "epoch": 4.523277282714844e-06,
      "model_forward_time": 1.0439751148223877,
      "step": 4743
    },
    {
      "epoch": 4.523277282714844e-06,
      "step": 4743,
      "training_step_time": 8.54834771156311
    },
    {
      "epoch": 4.52423095703125e-06,
      "model_forward_time": 1.3729634284973145,
      "step": 4744
    },
    {
      "epoch": 4.52423095703125e-06,
      "step": 4744,
      "training_step_time": 7.892473220825195
    },
    {
      "epoch": 4.525184631347656e-06,
      "model_forward_time": 1.7209579944610596,
      "step": 4745
    },
    {
      "epoch": 4.525184631347656e-06,
      "step": 4745,
      "training_step_time": 7.718881368637085
    },
    {
      "epoch": 4.526138305664063e-06,
      "model_forward_time": 1.7290318012237549,
      "step": 4746
    },
    {
      "epoch": 4.526138305664063e-06,
      "step": 4746,
      "training_step_time": 8.247224807739258
    },
    {
      "epoch": 4.527091979980469e-06,
      "model_forward_time": 1.7303838729858398,
      "step": 4747
    },
    {
      "epoch": 4.527091979980469e-06,
      "step": 4747,
      "training_step_time": 7.654173135757446
    },
    {
      "epoch": 4.528045654296875e-06,
      "model_forward_time": 1.7325413227081299,
      "step": 4748
    },
    {
      "epoch": 4.528045654296875e-06,
      "step": 4748,
      "training_step_time": 7.145575761795044
    },
    {
      "epoch": 4.528999328613281e-06,
      "model_forward_time": 1.7312052249908447,
      "step": 4749
    },
    {
      "epoch": 4.528999328613281e-06,
      "step": 4749,
      "training_step_time": 8.408798933029175
    },
    {
      "epoch": 4.5299530029296875e-06,
      "grad_norm": 0.3059926927089691,
      "learning_rate": 9.498000000000001e-06,
      "loss": 0.067,
      "step": 4750
    },
    {
      "epoch": 4.5299530029296875e-06,
      "model_forward_time": 2.8361735343933105,
      "step": 4750
    },
    {
      "dynamics_loss_avg": 0.08415843965485692,
      "epoch": 4.5299530029296875e-06,
      "step": 4750
    },
    {
      "action_loss_avg": 0.01146205272525549,
      "epoch": 4.5299530029296875e-06,
      "step": 4750
    },
    {
      "epoch": 4.5299530029296875e-06,
      "step": 4750,
      "training_step_time": 7.586540222167969
    },
    {
      "epoch": 4.530906677246094e-06,
      "model_forward_time": 1.7296514511108398,
      "step": 4751
    },
    {
      "epoch": 4.530906677246094e-06,
      "step": 4751,
      "training_step_time": 7.581182956695557
    },
    {
      "epoch": 4.5318603515625e-06,
      "model_forward_time": 1.7340905666351318,
      "step": 4752
    },
    {
      "epoch": 4.5318603515625e-06,
      "step": 4752,
      "training_step_time": 7.809372663497925
    },
    {
      "epoch": 4.532814025878906e-06,
      "model_forward_time": 1.7425909042358398,
      "step": 4753
    },
    {
      "epoch": 4.532814025878906e-06,
      "step": 4753,
      "training_step_time": 7.792868375778198
    },
    {
      "epoch": 4.533767700195312e-06,
      "model_forward_time": 1.7287981510162354,
      "step": 4754
    },
    {
      "epoch": 4.533767700195312e-06,
      "step": 4754,
      "training_step_time": 7.655599117279053
    },
    {
      "epoch": 4.534721374511719e-06,
      "model_forward_time": 1.7308688163757324,
      "step": 4755
    },
    {
      "epoch": 4.534721374511719e-06,
      "step": 4755,
      "training_step_time": 8.333928108215332
    },
    {
      "epoch": 4.535675048828125e-06,
      "model_forward_time": 1.3718371391296387,
      "step": 4756
    },
    {
      "epoch": 4.535675048828125e-06,
      "step": 4756,
      "training_step_time": 7.672118186950684
    },
    {
      "epoch": 4.536628723144531e-06,
      "model_forward_time": 1.726393461227417,
      "step": 4757
    },
    {
      "epoch": 4.536628723144531e-06,
      "step": 4757,
      "training_step_time": 7.589778661727905
    },
    {
      "epoch": 4.5375823974609375e-06,
      "model_forward_time": 1.742150068283081,
      "step": 4758
    },
    {
      "epoch": 4.5375823974609375e-06,
      "step": 4758,
      "training_step_time": 8.787687540054321
    },
    {
      "epoch": 4.538536071777344e-06,
      "model_forward_time": 2.896913766860962,
      "step": 4759
    },
    {
      "epoch": 4.538536071777344e-06,
      "step": 4759,
      "training_step_time": 8.347220420837402
    },
    {
      "epoch": 4.53948974609375e-06,
      "grad_norm": 0.40068715810775757,
      "learning_rate": 9.518000000000001e-06,
      "loss": 0.0729,
      "step": 4760
    },
    {
      "epoch": 4.53948974609375e-06,
      "model_forward_time": 1.7312164306640625,
      "step": 4760
    },
    {
      "dynamics_loss_avg": 0.06042736144736409,
      "epoch": 4.53948974609375e-06,
      "step": 4760
    },
    {
      "action_loss_avg": 0.007740000740159303,
      "epoch": 4.53948974609375e-06,
      "step": 4760
    },
    {
      "epoch": 4.53948974609375e-06,
      "step": 4760,
      "training_step_time": 8.297748804092407
    },
    {
      "epoch": 4.540443420410156e-06,
      "model_forward_time": 1.7242331504821777,
      "step": 4761
    },
    {
      "epoch": 4.540443420410156e-06,
      "step": 4761,
      "training_step_time": 8.09167218208313
    },
    {
      "epoch": 4.541397094726562e-06,
      "model_forward_time": 1.7301161289215088,
      "step": 4762
    },
    {
      "epoch": 4.541397094726562e-06,
      "step": 4762,
      "training_step_time": 7.779304504394531
    },
    {
      "epoch": 4.542350769042969e-06,
      "model_forward_time": 1.368687629699707,
      "step": 4763
    },
    {
      "epoch": 4.542350769042969e-06,
      "step": 4763,
      "training_step_time": 7.6132941246032715
    },
    {
      "epoch": 4.543304443359375e-06,
      "model_forward_time": 1.380812644958496,
      "step": 4764
    },
    {
      "epoch": 4.543304443359375e-06,
      "step": 4764,
      "training_step_time": 8.620730638504028
    },
    {
      "epoch": 4.544258117675781e-06,
      "model_forward_time": 1.7320241928100586,
      "step": 4765
    },
    {
      "epoch": 4.544258117675781e-06,
      "step": 4765,
      "training_step_time": 7.615253925323486
    },
    {
      "epoch": 4.5452117919921875e-06,
      "model_forward_time": 1.7259957790374756,
      "step": 4766
    },
    {
      "epoch": 4.5452117919921875e-06,
      "step": 4766,
      "training_step_time": 8.265787601470947
    },
    {
      "epoch": 4.546165466308594e-06,
      "model_forward_time": 1.7292039394378662,
      "step": 4767
    },
    {
      "epoch": 4.546165466308594e-06,
      "step": 4767,
      "training_step_time": 8.348819494247437
    },
    {
      "epoch": 4.547119140625e-06,
      "model_forward_time": 1.4579100608825684,
      "step": 4768
    },
    {
      "epoch": 4.547119140625e-06,
      "step": 4768,
      "training_step_time": 7.582248210906982
    },
    {
      "epoch": 4.548072814941406e-06,
      "model_forward_time": 2.488821506500244,
      "step": 4769
    },
    {
      "epoch": 4.548072814941406e-06,
      "step": 4769,
      "training_step_time": 7.56145453453064
    },
    {
      "epoch": 4.549026489257812e-06,
      "grad_norm": 0.38774573802948,
      "learning_rate": 9.538e-06,
      "loss": 0.0709,
      "step": 4770
    },
    {
      "epoch": 4.549026489257812e-06,
      "model_forward_time": 1.3865220546722412,
      "step": 4770
    },
    {
      "dynamics_loss_avg": 0.06259005088359118,
      "epoch": 4.549026489257812e-06,
      "step": 4770
    },
    {
      "action_loss_avg": 0.008460265351459384,
      "epoch": 4.549026489257812e-06,
      "step": 4770
    },
    {
      "epoch": 4.549026489257812e-06,
      "step": 4770,
      "training_step_time": 8.585893154144287
    },
    {
      "epoch": 4.549980163574219e-06,
      "model_forward_time": 1.7242412567138672,
      "step": 4771
    },
    {
      "epoch": 4.549980163574219e-06,
      "step": 4771,
      "training_step_time": 9.890427827835083
    },
    {
      "epoch": 4.550933837890625e-06,
      "model_forward_time": 1.7229156494140625,
      "step": 4772
    },
    {
      "epoch": 4.550933837890625e-06,
      "step": 4772,
      "training_step_time": 8.829731702804565
    },
    {
      "epoch": 4.551887512207031e-06,
      "model_forward_time": 1.7184679508209229,
      "step": 4773
    },
    {
      "epoch": 4.551887512207031e-06,
      "step": 4773,
      "training_step_time": 7.644617557525635
    },
    {
      "epoch": 4.5528411865234375e-06,
      "model_forward_time": 1.7174041271209717,
      "step": 4774
    },
    {
      "epoch": 4.5528411865234375e-06,
      "step": 4774,
      "training_step_time": 8.665649652481079
    },
    {
      "epoch": 4.553794860839844e-06,
      "model_forward_time": 1.7350351810455322,
      "step": 4775
    },
    {
      "epoch": 4.553794860839844e-06,
      "step": 4775,
      "training_step_time": 8.721450328826904
    },
    {
      "epoch": 4.55474853515625e-06,
      "model_forward_time": 1.7289845943450928,
      "step": 4776
    },
    {
      "epoch": 4.55474853515625e-06,
      "step": 4776,
      "training_step_time": 8.601709604263306
    },
    {
      "epoch": 4.555702209472656e-06,
      "model_forward_time": 1.3419101238250732,
      "step": 4777
    },
    {
      "epoch": 4.555702209472656e-06,
      "step": 4777,
      "training_step_time": 7.546329736709595
    },
    {
      "epoch": 4.556655883789062e-06,
      "model_forward_time": 2.8104474544525146,
      "step": 4778
    },
    {
      "epoch": 4.556655883789062e-06,
      "step": 4778,
      "training_step_time": 8.62582015991211
    },
    {
      "epoch": 4.557609558105469e-06,
      "model_forward_time": 1.7321057319641113,
      "step": 4779
    },
    {
      "epoch": 4.557609558105469e-06,
      "step": 4779,
      "training_step_time": 7.605280160903931
    },
    {
      "epoch": 4.558563232421875e-06,
      "grad_norm": 0.5677909851074219,
      "learning_rate": 9.558e-06,
      "loss": 0.0799,
      "step": 4780
    },
    {
      "epoch": 4.558563232421875e-06,
      "model_forward_time": 1.722294569015503,
      "step": 4780
    },
    {
      "dynamics_loss_avg": 0.07238261699676514,
      "epoch": 4.558563232421875e-06,
      "step": 4780
    },
    {
      "action_loss_avg": 0.011400158843025565,
      "epoch": 4.558563232421875e-06,
      "step": 4780
    },
    {
      "epoch": 4.558563232421875e-06,
      "step": 4780,
      "training_step_time": 8.667604684829712
    },
    {
      "epoch": 4.559516906738281e-06,
      "model_forward_time": 1.737107515335083,
      "step": 4781
    },
    {
      "epoch": 4.559516906738281e-06,
      "step": 4781,
      "training_step_time": 7.900391578674316
    },
    {
      "epoch": 4.5604705810546875e-06,
      "model_forward_time": 1.7500495910644531,
      "step": 4782
    },
    {
      "epoch": 4.5604705810546875e-06,
      "step": 4782,
      "training_step_time": 8.232000589370728
    },
    {
      "epoch": 4.561424255371094e-06,
      "model_forward_time": 1.7320904731750488,
      "step": 4783
    },
    {
      "epoch": 4.561424255371094e-06,
      "step": 4783,
      "training_step_time": 9.146701574325562
    },
    {
      "epoch": 4.5623779296875e-06,
      "model_forward_time": 1.7185051441192627,
      "step": 4784
    },
    {
      "epoch": 4.5623779296875e-06,
      "step": 4784,
      "training_step_time": 7.697177886962891
    },
    {
      "epoch": 4.563331604003906e-06,
      "model_forward_time": 1.753159999847412,
      "step": 4785
    },
    {
      "epoch": 4.563331604003906e-06,
      "step": 4785,
      "training_step_time": 7.585883378982544
    },
    {
      "epoch": 4.564285278320312e-06,
      "model_forward_time": 1.024451494216919,
      "step": 4786
    },
    {
      "epoch": 4.564285278320312e-06,
      "step": 4786,
      "training_step_time": 7.536612272262573
    },
    {
      "epoch": 4.565238952636719e-06,
      "model_forward_time": 1.8370413780212402,
      "step": 4787
    },
    {
      "epoch": 4.565238952636719e-06,
      "step": 4787,
      "training_step_time": 8.370316743850708
    },
    {
      "epoch": 4.566192626953125e-06,
      "model_forward_time": 2.826495885848999,
      "step": 4788
    },
    {
      "epoch": 4.566192626953125e-06,
      "step": 4788,
      "training_step_time": 8.175395965576172
    },
    {
      "epoch": 4.567146301269531e-06,
      "model_forward_time": 1.722853660583496,
      "step": 4789
    },
    {
      "epoch": 4.567146301269531e-06,
      "step": 4789,
      "training_step_time": 8.76221513748169
    },
    {
      "epoch": 4.5680999755859375e-06,
      "grad_norm": 0.3506760895252228,
      "learning_rate": 9.578e-06,
      "loss": 0.0756,
      "step": 4790
    },
    {
      "epoch": 4.5680999755859375e-06,
      "model_forward_time": 1.722447395324707,
      "step": 4790
    },
    {
      "dynamics_loss_avg": 0.0818344745784998,
      "epoch": 4.5680999755859375e-06,
      "step": 4790
    },
    {
      "action_loss_avg": 0.009835397731512785,
      "epoch": 4.5680999755859375e-06,
      "step": 4790
    },
    {
      "epoch": 4.5680999755859375e-06,
      "step": 4790,
      "training_step_time": 9.933508396148682
    },
    {
      "epoch": 4.569053649902344e-06,
      "model_forward_time": 1.7298402786254883,
      "step": 4791
    },
    {
      "epoch": 4.569053649902344e-06,
      "step": 4791,
      "training_step_time": 7.560396671295166
    },
    {
      "epoch": 4.57000732421875e-06,
      "model_forward_time": 1.7271924018859863,
      "step": 4792
    },
    {
      "epoch": 4.57000732421875e-06,
      "step": 4792,
      "training_step_time": 8.174131393432617
    },
    {
      "epoch": 4.570960998535156e-06,
      "model_forward_time": 1.714167594909668,
      "step": 4793
    },
    {
      "epoch": 4.570960998535156e-06,
      "step": 4793,
      "training_step_time": 7.675223112106323
    },
    {
      "epoch": 4.571914672851562e-06,
      "model_forward_time": 1.726294755935669,
      "step": 4794
    },
    {
      "epoch": 4.571914672851562e-06,
      "step": 4794,
      "training_step_time": 7.544965505599976
    },
    {
      "epoch": 4.572868347167969e-06,
      "model_forward_time": 1.0223619937896729,
      "step": 4795
    },
    {
      "epoch": 4.572868347167969e-06,
      "step": 4795,
      "training_step_time": 7.566938161849976
    },
    {
      "epoch": 4.573822021484375e-06,
      "model_forward_time": 1.8424744606018066,
      "step": 4796
    },
    {
      "epoch": 4.573822021484375e-06,
      "step": 4796,
      "training_step_time": 7.5980987548828125
    },
    {
      "epoch": 4.574775695800781e-06,
      "model_forward_time": 1.721233606338501,
      "step": 4797
    },
    {
      "epoch": 4.574775695800781e-06,
      "step": 4797,
      "training_step_time": 8.213395595550537
    },
    {
      "epoch": 4.5757293701171875e-06,
      "model_forward_time": 1.367804765701294,
      "step": 4798
    },
    {
      "epoch": 4.5757293701171875e-06,
      "step": 4798,
      "training_step_time": 8.458742380142212
    },
    {
      "epoch": 4.576683044433594e-06,
      "model_forward_time": 1.3647618293762207,
      "step": 4799
    },
    {
      "epoch": 4.576683044433594e-06,
      "step": 4799,
      "training_step_time": 7.752070665359497
    },
    {
      "epoch": 4.57763671875e-06,
      "grad_norm": 0.3748522996902466,
      "learning_rate": 9.598e-06,
      "loss": 0.0724,
      "step": 4800
    },
    {
      "epoch": 4.57763671875e-06,
      "model_forward_time": 1.7262117862701416,
      "step": 4800
    },
    {
      "dynamics_loss_avg": 0.0661506725475192,
      "epoch": 4.57763671875e-06,
      "step": 4800
    },
    {
      "action_loss_avg": 0.009814345021732151,
      "epoch": 4.57763671875e-06,
      "step": 4800
    },
    {
      "epoch": 4.57763671875e-06,
      "step": 4800,
      "training_step_time": 8.434757471084595
    },
    {
      "epoch": 4.578590393066406e-06,
      "model_forward_time": 1.7327706813812256,
      "step": 4801
    },
    {
      "epoch": 4.578590393066406e-06,
      "step": 4801,
      "training_step_time": 8.518864870071411
    },
    {
      "epoch": 4.579544067382812e-06,
      "model_forward_time": 1.7367444038391113,
      "step": 4802
    },
    {
      "epoch": 4.579544067382812e-06,
      "step": 4802,
      "training_step_time": 7.664815664291382
    },
    {
      "epoch": 4.580497741699219e-06,
      "model_forward_time": 1.7411365509033203,
      "step": 4803
    },
    {
      "epoch": 4.580497741699219e-06,
      "step": 4803,
      "training_step_time": 7.552953481674194
    },
    {
      "epoch": 4.581451416015625e-06,
      "model_forward_time": 1.7142088413238525,
      "step": 4804
    },
    {
      "epoch": 4.581451416015625e-06,
      "step": 4804,
      "training_step_time": 8.173219680786133
    },
    {
      "epoch": 4.582405090332031e-06,
      "model_forward_time": 1.8238987922668457,
      "step": 4805
    },
    {
      "epoch": 4.582405090332031e-06,
      "step": 4805,
      "training_step_time": 7.6063807010650635
    },
    {
      "epoch": 4.5833587646484375e-06,
      "model_forward_time": 1.371150255203247,
      "step": 4806
    },
    {
      "epoch": 4.5833587646484375e-06,
      "step": 4806,
      "training_step_time": 8.137681007385254
    },
    {
      "epoch": 4.584312438964844e-06,
      "model_forward_time": 2.8500583171844482,
      "step": 4807
    },
    {
      "epoch": 4.584312438964844e-06,
      "step": 4807,
      "training_step_time": 8.977113246917725
    },
    {
      "epoch": 4.58526611328125e-06,
      "model_forward_time": 1.7259020805358887,
      "step": 4808
    },
    {
      "epoch": 4.58526611328125e-06,
      "step": 4808,
      "training_step_time": 8.711508750915527
    },
    {
      "epoch": 4.586219787597656e-06,
      "model_forward_time": 1.7218208312988281,
      "step": 4809
    },
    {
      "epoch": 4.586219787597656e-06,
      "step": 4809,
      "training_step_time": 8.419021606445312
    },
    {
      "epoch": 4.587173461914062e-06,
      "grad_norm": 0.3410625457763672,
      "learning_rate": 9.618e-06,
      "loss": 0.0704,
      "step": 4810
    },
    {
      "epoch": 4.587173461914062e-06,
      "model_forward_time": 1.724017858505249,
      "step": 4810
    },
    {
      "dynamics_loss_avg": 0.06796432118862868,
      "epoch": 4.587173461914062e-06,
      "step": 4810
    },
    {
      "action_loss_avg": 0.01088202721439302,
      "epoch": 4.587173461914062e-06,
      "step": 4810
    },
    {
      "epoch": 4.587173461914062e-06,
      "step": 4810,
      "training_step_time": 7.6584930419921875
    },
    {
      "epoch": 4.588127136230469e-06,
      "model_forward_time": 1.7394404411315918,
      "step": 4811
    },
    {
      "epoch": 4.588127136230469e-06,
      "step": 4811,
      "training_step_time": 8.06952452659607
    },
    {
      "epoch": 4.589080810546875e-06,
      "model_forward_time": 1.7270653247833252,
      "step": 4812
    },
    {
      "epoch": 4.589080810546875e-06,
      "step": 4812,
      "training_step_time": 7.682790040969849
    },
    {
      "epoch": 4.590034484863281e-06,
      "model_forward_time": 1.7302496433258057,
      "step": 4813
    },
    {
      "epoch": 4.590034484863281e-06,
      "step": 4813,
      "training_step_time": 8.132673978805542
    },
    {
      "epoch": 4.5909881591796876e-06,
      "model_forward_time": 1.7860968112945557,
      "step": 4814
    },
    {
      "epoch": 4.5909881591796876e-06,
      "step": 4814,
      "training_step_time": 8.55133318901062
    },
    {
      "epoch": 4.591941833496094e-06,
      "model_forward_time": 1.7249205112457275,
      "step": 4815
    },
    {
      "epoch": 4.591941833496094e-06,
      "step": 4815,
      "training_step_time": 8.29842495918274
    },
    {
      "epoch": 4.5928955078125e-06,
      "model_forward_time": 2.816206216812134,
      "step": 4816
    },
    {
      "epoch": 4.5928955078125e-06,
      "step": 4816,
      "training_step_time": 7.593827247619629
    },
    {
      "epoch": 4.593849182128906e-06,
      "model_forward_time": 1.7291655540466309,
      "step": 4817
    },
    {
      "epoch": 4.593849182128906e-06,
      "step": 4817,
      "training_step_time": 8.292980194091797
    },
    {
      "epoch": 4.594802856445312e-06,
      "model_forward_time": 1.72271728515625,
      "step": 4818
    },
    {
      "epoch": 4.594802856445312e-06,
      "step": 4818,
      "training_step_time": 8.68575143814087
    },
    {
      "epoch": 4.595756530761719e-06,
      "model_forward_time": 1.7324788570404053,
      "step": 4819
    },
    {
      "epoch": 4.595756530761719e-06,
      "step": 4819,
      "training_step_time": 9.39417028427124
    },
    {
      "epoch": 4.596710205078125e-06,
      "grad_norm": 0.476610004901886,
      "learning_rate": 9.638e-06,
      "loss": 0.0759,
      "step": 4820
    },
    {
      "epoch": 4.596710205078125e-06,
      "model_forward_time": 1.7516851425170898,
      "step": 4820
    },
    {
      "dynamics_loss_avg": 0.06703849248588085,
      "epoch": 4.596710205078125e-06,
      "step": 4820
    },
    {
      "action_loss_avg": 0.007796897133812308,
      "epoch": 4.596710205078125e-06,
      "step": 4820
    },
    {
      "epoch": 4.596710205078125e-06,
      "step": 4820,
      "training_step_time": 8.622761011123657
    },
    {
      "epoch": 4.597663879394531e-06,
      "model_forward_time": 1.7411201000213623,
      "step": 4821
    },
    {
      "epoch": 4.597663879394531e-06,
      "step": 4821,
      "training_step_time": 7.669016599655151
    },
    {
      "epoch": 4.5986175537109376e-06,
      "model_forward_time": 1.722907304763794,
      "step": 4822
    },
    {
      "epoch": 4.5986175537109376e-06,
      "step": 4822,
      "training_step_time": 7.477270841598511
    },
    {
      "epoch": 4.599571228027344e-06,
      "model_forward_time": 1.7360210418701172,
      "step": 4823
    },
    {
      "epoch": 4.599571228027344e-06,
      "step": 4823,
      "training_step_time": 7.6237077713012695
    },
    {
      "epoch": 4.60052490234375e-06,
      "model_forward_time": 1.761253833770752,
      "step": 4824
    },
    {
      "epoch": 4.60052490234375e-06,
      "step": 4824,
      "training_step_time": 7.546473741531372
    },
    {
      "epoch": 4.601478576660156e-06,
      "model_forward_time": 1.354229211807251,
      "step": 4825
    },
    {
      "epoch": 4.601478576660156e-06,
      "step": 4825,
      "training_step_time": 8.863879680633545
    },
    {
      "epoch": 4.602432250976562e-06,
      "model_forward_time": 2.8734664916992188,
      "step": 4826
    },
    {
      "epoch": 4.602432250976562e-06,
      "step": 4826,
      "training_step_time": 9.553361177444458
    },
    {
      "epoch": 4.603385925292969e-06,
      "model_forward_time": 1.3531384468078613,
      "step": 4827
    },
    {
      "epoch": 4.603385925292969e-06,
      "step": 4827,
      "training_step_time": 8.604281663894653
    },
    {
      "epoch": 4.604339599609375e-06,
      "model_forward_time": 1.7276651859283447,
      "step": 4828
    },
    {
      "epoch": 4.604339599609375e-06,
      "step": 4828,
      "training_step_time": 7.649776935577393
    },
    {
      "epoch": 4.605293273925781e-06,
      "model_forward_time": 1.7300617694854736,
      "step": 4829
    },
    {
      "epoch": 4.605293273925781e-06,
      "step": 4829,
      "training_step_time": 8.645006895065308
    },
    {
      "epoch": 4.6062469482421876e-06,
      "grad_norm": 0.5184007287025452,
      "learning_rate": 9.658000000000001e-06,
      "loss": 0.0753,
      "step": 4830
    },
    {
      "epoch": 4.6062469482421876e-06,
      "model_forward_time": 1.730806589126587,
      "step": 4830
    },
    {
      "dynamics_loss_avg": 0.06404553689062595,
      "epoch": 4.6062469482421876e-06,
      "step": 4830
    },
    {
      "action_loss_avg": 0.007242418266832829,
      "epoch": 4.6062469482421876e-06,
      "step": 4830
    },
    {
      "epoch": 4.6062469482421876e-06,
      "step": 4830,
      "training_step_time": 8.66875433921814
    },
    {
      "epoch": 4.607200622558594e-06,
      "model_forward_time": 1.758697271347046,
      "step": 4831
    },
    {
      "epoch": 4.607200622558594e-06,
      "step": 4831,
      "training_step_time": 7.572845697402954
    },
    {
      "epoch": 4.608154296875e-06,
      "model_forward_time": 1.7338435649871826,
      "step": 4832
    },
    {
      "epoch": 4.608154296875e-06,
      "step": 4832,
      "training_step_time": 7.473708391189575
    },
    {
      "epoch": 4.6091079711914064e-06,
      "model_forward_time": 1.778961420059204,
      "step": 4833
    },
    {
      "epoch": 4.6091079711914064e-06,
      "step": 4833,
      "training_step_time": 7.552870273590088
    },
    {
      "epoch": 4.610061645507812e-06,
      "model_forward_time": 1.7250871658325195,
      "step": 4834
    },
    {
      "epoch": 4.610061645507812e-06,
      "step": 4834,
      "training_step_time": 7.576761722564697
    },
    {
      "epoch": 4.611015319824219e-06,
      "model_forward_time": 2.8690693378448486,
      "step": 4835
    },
    {
      "epoch": 4.611015319824219e-06,
      "step": 4835,
      "training_step_time": 7.6158177852630615
    },
    {
      "epoch": 4.611968994140625e-06,
      "model_forward_time": 1.7239305973052979,
      "step": 4836
    },
    {
      "epoch": 4.611968994140625e-06,
      "step": 4836,
      "training_step_time": 7.844453811645508
    },
    {
      "epoch": 4.612922668457031e-06,
      "model_forward_time": 1.7023465633392334,
      "step": 4837
    },
    {
      "epoch": 4.612922668457031e-06,
      "step": 4837,
      "training_step_time": 7.579176425933838
    },
    {
      "epoch": 4.6138763427734376e-06,
      "model_forward_time": 1.722801923751831,
      "step": 4838
    },
    {
      "epoch": 4.6138763427734376e-06,
      "step": 4838,
      "training_step_time": 7.619034051895142
    },
    {
      "epoch": 4.614830017089844e-06,
      "model_forward_time": 1.3635411262512207,
      "step": 4839
    },
    {
      "epoch": 4.614830017089844e-06,
      "step": 4839,
      "training_step_time": 7.662123680114746
    },
    {
      "epoch": 4.61578369140625e-06,
      "grad_norm": 0.4443095326423645,
      "learning_rate": 9.678000000000001e-06,
      "loss": 0.0808,
      "step": 4840
    },
    {
      "epoch": 4.61578369140625e-06,
      "model_forward_time": 1.7323665618896484,
      "step": 4840
    },
    {
      "dynamics_loss_avg": 0.066552047803998,
      "epoch": 4.61578369140625e-06,
      "step": 4840
    },
    {
      "action_loss_avg": 0.007146036881022155,
      "epoch": 4.61578369140625e-06,
      "step": 4840
    },
    {
      "epoch": 4.61578369140625e-06,
      "step": 4840,
      "training_step_time": 7.623446941375732
    },
    {
      "epoch": 4.6167373657226564e-06,
      "model_forward_time": 1.7281742095947266,
      "step": 4841
    },
    {
      "epoch": 4.6167373657226564e-06,
      "step": 4841,
      "training_step_time": 7.508068323135376
    },
    {
      "epoch": 4.617691040039062e-06,
      "model_forward_time": 1.7723312377929688,
      "step": 4842
    },
    {
      "epoch": 4.617691040039062e-06,
      "step": 4842,
      "training_step_time": 7.6129820346832275
    },
    {
      "epoch": 4.618644714355469e-06,
      "model_forward_time": 1.7072608470916748,
      "step": 4843
    },
    {
      "epoch": 4.618644714355469e-06,
      "step": 4843,
      "training_step_time": 7.555332183837891
    },
    {
      "epoch": 4.619598388671875e-06,
      "model_forward_time": 2.7860946655273438,
      "step": 4844
    },
    {
      "epoch": 4.619598388671875e-06,
      "step": 4844,
      "training_step_time": 8.249275922775269
    },
    {
      "epoch": 4.620552062988281e-06,
      "model_forward_time": 1.3493247032165527,
      "step": 4845
    },
    {
      "epoch": 4.620552062988281e-06,
      "step": 4845,
      "training_step_time": 9.54571008682251
    },
    {
      "epoch": 4.621505737304688e-06,
      "model_forward_time": 1.713318109512329,
      "step": 4846
    },
    {
      "epoch": 4.621505737304688e-06,
      "step": 4846,
      "training_step_time": 9.426697731018066
    },
    {
      "epoch": 4.622459411621094e-06,
      "model_forward_time": 1.7204735279083252,
      "step": 4847
    },
    {
      "epoch": 4.622459411621094e-06,
      "step": 4847,
      "training_step_time": 8.486510753631592
    },
    {
      "epoch": 4.6234130859375e-06,
      "model_forward_time": 1.7152979373931885,
      "step": 4848
    },
    {
      "epoch": 4.6234130859375e-06,
      "step": 4848,
      "training_step_time": 8.765962362289429
    },
    {
      "epoch": 4.6243667602539064e-06,
      "model_forward_time": 1.7106201648712158,
      "step": 4849
    },
    {
      "epoch": 4.6243667602539064e-06,
      "step": 4849,
      "training_step_time": 8.395120620727539
    },
    {
      "epoch": 4.625320434570312e-06,
      "grad_norm": 0.41206324100494385,
      "learning_rate": 9.698000000000001e-06,
      "loss": 0.0786,
      "step": 4850
    },
    {
      "epoch": 4.625320434570312e-06,
      "model_forward_time": 1.7158746719360352,
      "step": 4850
    },
    {
      "dynamics_loss_avg": 0.06894316710531712,
      "epoch": 4.625320434570312e-06,
      "step": 4850
    },
    {
      "action_loss_avg": 0.006142487050965428,
      "epoch": 4.625320434570312e-06,
      "step": 4850
    },
    {
      "epoch": 4.625320434570312e-06,
      "step": 4850,
      "training_step_time": 7.520077466964722
    },
    {
      "epoch": 4.626274108886719e-06,
      "model_forward_time": 1.3438725471496582,
      "step": 4851
    },
    {
      "epoch": 4.626274108886719e-06,
      "step": 4851,
      "training_step_time": 7.5374109745025635
    },
    {
      "epoch": 4.627227783203125e-06,
      "model_forward_time": 1.712714433670044,
      "step": 4852
    },
    {
      "epoch": 4.627227783203125e-06,
      "step": 4852,
      "training_step_time": 7.567773103713989
    },
    {
      "epoch": 4.628181457519531e-06,
      "model_forward_time": 1.3560585975646973,
      "step": 4853
    },
    {
      "epoch": 4.628181457519531e-06,
      "step": 4853,
      "training_step_time": 7.507607698440552
    },
    {
      "epoch": 4.629135131835938e-06,
      "model_forward_time": 2.8669326305389404,
      "step": 4854
    },
    {
      "epoch": 4.629135131835938e-06,
      "step": 4854,
      "training_step_time": 7.565937519073486
    },
    {
      "epoch": 4.630088806152344e-06,
      "model_forward_time": 1.7231364250183105,
      "step": 4855
    },
    {
      "epoch": 4.630088806152344e-06,
      "step": 4855,
      "training_step_time": 8.554325819015503
    },
    {
      "epoch": 4.63104248046875e-06,
      "model_forward_time": 1.7224702835083008,
      "step": 4856
    },
    {
      "epoch": 4.63104248046875e-06,
      "step": 4856,
      "training_step_time": 8.610932350158691
    },
    {
      "epoch": 4.6319961547851564e-06,
      "model_forward_time": 1.7185630798339844,
      "step": 4857
    },
    {
      "epoch": 4.6319961547851564e-06,
      "step": 4857,
      "training_step_time": 7.644474983215332
    },
    {
      "epoch": 4.632949829101562e-06,
      "model_forward_time": 1.7098901271820068,
      "step": 4858
    },
    {
      "epoch": 4.632949829101562e-06,
      "step": 4858,
      "training_step_time": 7.63265585899353
    },
    {
      "epoch": 4.633903503417969e-06,
      "model_forward_time": 1.7190806865692139,
      "step": 4859
    },
    {
      "epoch": 4.633903503417969e-06,
      "step": 4859,
      "training_step_time": 7.526336908340454
    },
    {
      "epoch": 4.634857177734375e-06,
      "grad_norm": 0.43559083342552185,
      "learning_rate": 9.718e-06,
      "loss": 0.0789,
      "step": 4860
    },
    {
      "epoch": 4.634857177734375e-06,
      "model_forward_time": 1.7260558605194092,
      "step": 4860
    },
    {
      "dynamics_loss_avg": 0.07828259728848934,
      "epoch": 4.634857177734375e-06,
      "step": 4860
    },
    {
      "action_loss_avg": 0.009589137742295861,
      "epoch": 4.634857177734375e-06,
      "step": 4860
    },
    {
      "epoch": 4.634857177734375e-06,
      "step": 4860,
      "training_step_time": 7.556249618530273
    },
    {
      "epoch": 4.635810852050781e-06,
      "model_forward_time": 1.448108434677124,
      "step": 4861
    },
    {
      "epoch": 4.635810852050781e-06,
      "step": 4861,
      "training_step_time": 7.559732913970947
    },
    {
      "epoch": 4.636764526367188e-06,
      "model_forward_time": 1.7285542488098145,
      "step": 4862
    },
    {
      "epoch": 4.636764526367188e-06,
      "step": 4862,
      "training_step_time": 7.55575704574585
    },
    {
      "epoch": 4.637718200683594e-06,
      "model_forward_time": 2.8309366703033447,
      "step": 4863
    },
    {
      "epoch": 4.637718200683594e-06,
      "step": 4863,
      "training_step_time": 7.560536861419678
    },
    {
      "epoch": 4.638671875e-06,
      "model_forward_time": 1.7207393646240234,
      "step": 4864
    },
    {
      "epoch": 4.638671875e-06,
      "step": 4864,
      "training_step_time": 8.302128076553345
    },
    {
      "epoch": 4.6396255493164065e-06,
      "model_forward_time": 1.7254102230072021,
      "step": 4865
    },
    {
      "epoch": 4.6396255493164065e-06,
      "step": 4865,
      "training_step_time": 7.591267347335815
    },
    {
      "epoch": 4.640579223632812e-06,
      "model_forward_time": 1.7202928066253662,
      "step": 4866
    },
    {
      "epoch": 4.640579223632812e-06,
      "step": 4866,
      "training_step_time": 7.657010793685913
    },
    {
      "epoch": 4.641532897949219e-06,
      "model_forward_time": 1.7169766426086426,
      "step": 4867
    },
    {
      "epoch": 4.641532897949219e-06,
      "step": 4867,
      "training_step_time": 8.631276369094849
    },
    {
      "epoch": 4.642486572265625e-06,
      "model_forward_time": 1.7194783687591553,
      "step": 4868
    },
    {
      "epoch": 4.642486572265625e-06,
      "step": 4868,
      "training_step_time": 7.564945936203003
    },
    {
      "epoch": 4.643440246582031e-06,
      "model_forward_time": 1.7198724746704102,
      "step": 4869
    },
    {
      "epoch": 4.643440246582031e-06,
      "step": 4869,
      "training_step_time": 8.35691523551941
    },
    {
      "epoch": 4.644393920898438e-06,
      "grad_norm": 0.43639102578163147,
      "learning_rate": 9.738e-06,
      "loss": 0.0857,
      "step": 4870
    },
    {
      "epoch": 4.644393920898438e-06,
      "model_forward_time": 1.8287427425384521,
      "step": 4870
    },
    {
      "dynamics_loss_avg": 0.07134243436157703,
      "epoch": 4.644393920898438e-06,
      "step": 4870
    },
    {
      "action_loss_avg": 0.007955440320074558,
      "epoch": 4.644393920898438e-06,
      "step": 4870
    },
    {
      "epoch": 4.644393920898438e-06,
      "step": 4870,
      "training_step_time": 7.6461145877838135
    },
    {
      "epoch": 4.645347595214844e-06,
      "model_forward_time": 1.714327335357666,
      "step": 4871
    },
    {
      "epoch": 4.645347595214844e-06,
      "step": 4871,
      "training_step_time": 7.581497669219971
    },
    {
      "epoch": 4.64630126953125e-06,
      "model_forward_time": 1.7119066715240479,
      "step": 4872
    },
    {
      "epoch": 4.64630126953125e-06,
      "step": 4872,
      "training_step_time": 8.919196128845215
    },
    {
      "epoch": 4.6472549438476565e-06,
      "model_forward_time": 1.7103111743927002,
      "step": 4873
    },
    {
      "epoch": 4.6472549438476565e-06,
      "step": 4873,
      "training_step_time": 7.872720003128052
    },
    {
      "epoch": 4.648208618164062e-06,
      "model_forward_time": 1.7161338329315186,
      "step": 4874
    },
    {
      "epoch": 4.648208618164062e-06,
      "step": 4874,
      "training_step_time": 8.16551947593689
    },
    {
      "epoch": 4.649162292480469e-06,
      "model_forward_time": 1.7309846878051758,
      "step": 4875
    },
    {
      "epoch": 4.649162292480469e-06,
      "step": 4875,
      "training_step_time": 7.524693250656128
    },
    {
      "epoch": 4.650115966796875e-06,
      "model_forward_time": 1.7123584747314453,
      "step": 4876
    },
    {
      "epoch": 4.650115966796875e-06,
      "step": 4876,
      "training_step_time": 7.6791603565216064
    },
    {
      "epoch": 4.651069641113281e-06,
      "model_forward_time": 1.7106695175170898,
      "step": 4877
    },
    {
      "epoch": 4.651069641113281e-06,
      "step": 4877,
      "training_step_time": 7.5777809619903564
    },
    {
      "epoch": 4.652023315429688e-06,
      "model_forward_time": 1.711388111114502,
      "step": 4878
    },
    {
      "epoch": 4.652023315429688e-06,
      "step": 4878,
      "training_step_time": 7.608496427536011
    },
    {
      "epoch": 4.652976989746093e-06,
      "model_forward_time": 1.3452749252319336,
      "step": 4879
    },
    {
      "epoch": 4.652976989746093e-06,
      "step": 4879,
      "training_step_time": 7.604255437850952
    },
    {
      "epoch": 4.6539306640625e-06,
      "grad_norm": 0.31564420461654663,
      "learning_rate": 9.758e-06,
      "loss": 0.0791,
      "step": 4880
    },
    {
      "epoch": 4.6539306640625e-06,
      "model_forward_time": 1.708019733428955,
      "step": 4880
    },
    {
      "dynamics_loss_avg": 0.0614389942958951,
      "epoch": 4.6539306640625e-06,
      "step": 4880
    },
    {
      "action_loss_avg": 0.0072058812016621236,
      "epoch": 4.6539306640625e-06,
      "step": 4880
    },
    {
      "epoch": 4.6539306640625e-06,
      "step": 4880,
      "training_step_time": 8.469555377960205
    },
    {
      "epoch": 4.6548843383789065e-06,
      "model_forward_time": 1.6997649669647217,
      "step": 4881
    },
    {
      "epoch": 4.6548843383789065e-06,
      "step": 4881,
      "training_step_time": 9.919004201889038
    },
    {
      "epoch": 4.655838012695312e-06,
      "model_forward_time": 2.8941080570220947,
      "step": 4882
    },
    {
      "epoch": 4.655838012695312e-06,
      "step": 4882,
      "training_step_time": 7.837148666381836
    },
    {
      "epoch": 4.656791687011719e-06,
      "model_forward_time": 1.7112493515014648,
      "step": 4883
    },
    {
      "epoch": 4.656791687011719e-06,
      "step": 4883,
      "training_step_time": 7.715681314468384
    },
    {
      "epoch": 4.657745361328125e-06,
      "model_forward_time": 1.7242188453674316,
      "step": 4884
    },
    {
      "epoch": 4.657745361328125e-06,
      "step": 4884,
      "training_step_time": 8.603025913238525
    },
    {
      "epoch": 4.658699035644531e-06,
      "model_forward_time": 1.7112867832183838,
      "step": 4885
    },
    {
      "epoch": 4.658699035644531e-06,
      "step": 4885,
      "training_step_time": 7.769111633300781
    },
    {
      "epoch": 4.659652709960938e-06,
      "model_forward_time": 1.7156617641448975,
      "step": 4886
    },
    {
      "epoch": 4.659652709960938e-06,
      "step": 4886,
      "training_step_time": 7.659048080444336
    },
    {
      "epoch": 4.660606384277343e-06,
      "model_forward_time": 1.7166814804077148,
      "step": 4887
    },
    {
      "epoch": 4.660606384277343e-06,
      "step": 4887,
      "training_step_time": 7.634516954421997
    },
    {
      "epoch": 4.66156005859375e-06,
      "model_forward_time": 1.7097105979919434,
      "step": 4888
    },
    {
      "epoch": 4.66156005859375e-06,
      "step": 4888,
      "training_step_time": 7.514580011367798
    },
    {
      "epoch": 4.6625137329101565e-06,
      "model_forward_time": 1.7172362804412842,
      "step": 4889
    },
    {
      "epoch": 4.6625137329101565e-06,
      "step": 4889,
      "training_step_time": 7.5801637172698975
    },
    {
      "epoch": 4.663467407226562e-06,
      "grad_norm": 0.4170455038547516,
      "learning_rate": 9.778e-06,
      "loss": 0.0752,
      "step": 4890
    },
    {
      "epoch": 4.663467407226562e-06,
      "model_forward_time": 1.7133798599243164,
      "step": 4890
    },
    {
      "dynamics_loss_avg": 0.06196139231324196,
      "epoch": 4.663467407226562e-06,
      "step": 4890
    },
    {
      "action_loss_avg": 0.006414019642397762,
      "epoch": 4.663467407226562e-06,
      "step": 4890
    },
    {
      "epoch": 4.663467407226562e-06,
      "step": 4890,
      "training_step_time": 7.483951091766357
    },
    {
      "epoch": 4.664421081542969e-06,
      "model_forward_time": 2.880260944366455,
      "step": 4891
    },
    {
      "epoch": 4.664421081542969e-06,
      "step": 4891,
      "training_step_time": 7.74004602432251
    },
    {
      "epoch": 4.665374755859375e-06,
      "model_forward_time": 1.7106804847717285,
      "step": 4892
    },
    {
      "epoch": 4.665374755859375e-06,
      "step": 4892,
      "training_step_time": 8.212018013000488
    },
    {
      "epoch": 4.666328430175781e-06,
      "model_forward_time": 1.7143821716308594,
      "step": 4893
    },
    {
      "epoch": 4.666328430175781e-06,
      "step": 4893,
      "training_step_time": 7.5868306159973145
    },
    {
      "epoch": 4.667282104492188e-06,
      "model_forward_time": 1.7277522087097168,
      "step": 4894
    },
    {
      "epoch": 4.667282104492188e-06,
      "step": 4894,
      "training_step_time": 9.499786376953125
    },
    {
      "epoch": 4.668235778808593e-06,
      "model_forward_time": 1.7132277488708496,
      "step": 4895
    },
    {
      "epoch": 4.668235778808593e-06,
      "step": 4895,
      "training_step_time": 8.670179605484009
    },
    {
      "epoch": 4.669189453125e-06,
      "model_forward_time": 1.7157576084136963,
      "step": 4896
    },
    {
      "epoch": 4.669189453125e-06,
      "step": 4896,
      "training_step_time": 7.640293836593628
    },
    {
      "epoch": 4.6701431274414065e-06,
      "model_forward_time": 1.709223985671997,
      "step": 4897
    },
    {
      "epoch": 4.6701431274414065e-06,
      "step": 4897,
      "training_step_time": 7.445984840393066
    },
    {
      "epoch": 4.671096801757812e-06,
      "model_forward_time": 1.7372405529022217,
      "step": 4898
    },
    {
      "epoch": 4.671096801757812e-06,
      "step": 4898,
      "training_step_time": 8.141727924346924
    },
    {
      "epoch": 4.672050476074219e-06,
      "model_forward_time": 1.7094595432281494,
      "step": 4899
    },
    {
      "epoch": 4.672050476074219e-06,
      "step": 4899,
      "training_step_time": 7.540932655334473
    },
    {
      "epoch": 4.673004150390625e-06,
      "grad_norm": 0.30404132604599,
      "learning_rate": 9.798e-06,
      "loss": 0.0726,
      "step": 4900
    },
    {
      "epoch": 4.673004150390625e-06,
      "model_forward_time": 2.811770439147949,
      "step": 4900
    },
    {
      "dynamics_loss_avg": 0.06369080822914838,
      "epoch": 4.673004150390625e-06,
      "step": 4900
    },
    {
      "action_loss_avg": 0.0075384496944025155,
      "epoch": 4.673004150390625e-06,
      "step": 4900
    },
    {
      "epoch": 4.673004150390625e-06,
      "step": 4900,
      "training_step_time": 7.840301036834717
    },
    {
      "epoch": 4.673957824707031e-06,
      "model_forward_time": 1.7139253616333008,
      "step": 4901
    },
    {
      "epoch": 4.673957824707031e-06,
      "step": 4901,
      "training_step_time": 9.24704098701477
    },
    {
      "epoch": 4.674911499023438e-06,
      "model_forward_time": 1.713714599609375,
      "step": 4902
    },
    {
      "epoch": 4.674911499023438e-06,
      "step": 4902,
      "training_step_time": 8.340622186660767
    },
    {
      "epoch": 4.675865173339843e-06,
      "model_forward_time": 1.726827621459961,
      "step": 4903
    },
    {
      "epoch": 4.675865173339843e-06,
      "step": 4903,
      "training_step_time": 8.078781127929688
    },
    {
      "epoch": 4.67681884765625e-06,
      "model_forward_time": 1.7125468254089355,
      "step": 4904
    },
    {
      "epoch": 4.67681884765625e-06,
      "step": 4904,
      "training_step_time": 7.601854562759399
    },
    {
      "epoch": 4.6777725219726565e-06,
      "model_forward_time": 1.7152044773101807,
      "step": 4905
    },
    {
      "epoch": 4.6777725219726565e-06,
      "step": 4905,
      "training_step_time": 8.311139822006226
    },
    {
      "epoch": 4.678726196289062e-06,
      "model_forward_time": 1.7099804878234863,
      "step": 4906
    },
    {
      "epoch": 4.678726196289062e-06,
      "step": 4906,
      "training_step_time": 7.44271993637085
    },
    {
      "epoch": 4.679679870605469e-06,
      "model_forward_time": 1.7212886810302734,
      "step": 4907
    },
    {
      "epoch": 4.679679870605469e-06,
      "step": 4907,
      "training_step_time": 8.394427061080933
    },
    {
      "epoch": 4.680633544921875e-06,
      "model_forward_time": 1.7096588611602783,
      "step": 4908
    },
    {
      "epoch": 4.680633544921875e-06,
      "step": 4908,
      "training_step_time": 7.613782644271851
    },
    {
      "epoch": 4.681587219238281e-06,
      "model_forward_time": 1.3420014381408691,
      "step": 4909
    },
    {
      "epoch": 4.681587219238281e-06,
      "step": 4909,
      "training_step_time": 8.672954082489014
    },
    {
      "epoch": 4.682540893554688e-06,
      "grad_norm": 0.35632187128067017,
      "learning_rate": 9.818000000000002e-06,
      "loss": 0.076,
      "step": 4910
    },
    {
      "epoch": 4.682540893554688e-06,
      "model_forward_time": 1.7098348140716553,
      "step": 4910
    },
    {
      "dynamics_loss_avg": 0.07068647779524326,
      "epoch": 4.682540893554688e-06,
      "step": 4910
    },
    {
      "action_loss_avg": 0.007047709124162793,
      "epoch": 4.682540893554688e-06,
      "step": 4910
    },
    {
      "epoch": 4.682540893554688e-06,
      "step": 4910,
      "training_step_time": 7.6733503341674805
    },
    {
      "epoch": 4.683494567871093e-06,
      "model_forward_time": 1.7118582725524902,
      "step": 4911
    },
    {
      "epoch": 4.683494567871093e-06,
      "step": 4911,
      "training_step_time": 9.05049443244934
    },
    {
      "epoch": 4.6844482421875e-06,
      "model_forward_time": 1.7163658142089844,
      "step": 4912
    },
    {
      "epoch": 4.6844482421875e-06,
      "step": 4912,
      "training_step_time": 7.494293928146362
    },
    {
      "epoch": 4.6854019165039065e-06,
      "model_forward_time": 1.7254958152770996,
      "step": 4913
    },
    {
      "epoch": 4.6854019165039065e-06,
      "step": 4913,
      "training_step_time": 7.617090463638306
    },
    {
      "epoch": 4.686355590820312e-06,
      "model_forward_time": 1.7067553997039795,
      "step": 4914
    },
    {
      "epoch": 4.686355590820312e-06,
      "step": 4914,
      "training_step_time": 7.56291389465332
    },
    {
      "epoch": 4.687309265136719e-06,
      "model_forward_time": 1.706636667251587,
      "step": 4915
    },
    {
      "epoch": 4.687309265136719e-06,
      "step": 4915,
      "training_step_time": 7.558115482330322
    },
    {
      "epoch": 4.688262939453125e-06,
      "model_forward_time": 1.3422458171844482,
      "step": 4916
    },
    {
      "epoch": 4.688262939453125e-06,
      "step": 4916,
      "training_step_time": 7.504836320877075
    },
    {
      "epoch": 4.689216613769531e-06,
      "model_forward_time": 1.7105326652526855,
      "step": 4917
    },
    {
      "epoch": 4.689216613769531e-06,
      "step": 4917,
      "training_step_time": 8.155236005783081
    },
    {
      "epoch": 4.690170288085938e-06,
      "model_forward_time": 1.7075426578521729,
      "step": 4918
    },
    {
      "epoch": 4.690170288085938e-06,
      "step": 4918,
      "training_step_time": 9.443841457366943
    },
    {
      "epoch": 4.691123962402343e-06,
      "model_forward_time": 2.901412010192871,
      "step": 4919
    },
    {
      "epoch": 4.691123962402343e-06,
      "step": 4919,
      "training_step_time": 8.652950763702393
    },
    {
      "epoch": 4.69207763671875e-06,
      "grad_norm": 0.26568540930747986,
      "learning_rate": 9.838000000000001e-06,
      "loss": 0.0767,
      "step": 4920
    },
    {
      "epoch": 4.69207763671875e-06,
      "model_forward_time": 1.711545467376709,
      "step": 4920
    },
    {
      "dynamics_loss_avg": 0.06716246195137501,
      "epoch": 4.69207763671875e-06,
      "step": 4920
    },
    {
      "action_loss_avg": 0.007712400075979531,
      "epoch": 4.69207763671875e-06,
      "step": 4920
    },
    {
      "epoch": 4.69207763671875e-06,
      "step": 4920,
      "training_step_time": 7.601178169250488
    },
    {
      "epoch": 4.6930313110351565e-06,
      "model_forward_time": 1.7151663303375244,
      "step": 4921
    },
    {
      "epoch": 4.6930313110351565e-06,
      "step": 4921,
      "training_step_time": 8.517032384872437
    },
    {
      "epoch": 4.693984985351562e-06,
      "model_forward_time": 1.7275769710540771,
      "step": 4922
    },
    {
      "epoch": 4.693984985351562e-06,
      "step": 4922,
      "training_step_time": 8.501046895980835
    },
    {
      "epoch": 4.694938659667969e-06,
      "model_forward_time": 1.7091765403747559,
      "step": 4923
    },
    {
      "epoch": 4.694938659667969e-06,
      "step": 4923,
      "training_step_time": 8.331184387207031
    },
    {
      "epoch": 4.695892333984375e-06,
      "model_forward_time": 1.7144997119903564,
      "step": 4924
    },
    {
      "epoch": 4.695892333984375e-06,
      "step": 4924,
      "training_step_time": 7.649869918823242
    },
    {
      "epoch": 4.696846008300781e-06,
      "model_forward_time": 1.3447000980377197,
      "step": 4925
    },
    {
      "epoch": 4.696846008300781e-06,
      "step": 4925,
      "training_step_time": 7.526523113250732
    },
    {
      "epoch": 4.697799682617188e-06,
      "model_forward_time": 1.7139368057250977,
      "step": 4926
    },
    {
      "epoch": 4.697799682617188e-06,
      "step": 4926,
      "training_step_time": 8.339864492416382
    },
    {
      "epoch": 4.698753356933593e-06,
      "model_forward_time": 1.7161896228790283,
      "step": 4927
    },
    {
      "epoch": 4.698753356933593e-06,
      "step": 4927,
      "training_step_time": 7.497125864028931
    },
    {
      "epoch": 4.69970703125e-06,
      "model_forward_time": 2.876816987991333,
      "step": 4928
    },
    {
      "epoch": 4.69970703125e-06,
      "step": 4928,
      "training_step_time": 7.801164627075195
    },
    {
      "epoch": 4.7006607055664065e-06,
      "model_forward_time": 1.7172224521636963,
      "step": 4929
    },
    {
      "epoch": 4.7006607055664065e-06,
      "step": 4929,
      "training_step_time": 7.615394592285156
    },
    {
      "epoch": 4.701614379882812e-06,
      "grad_norm": 0.2838245630264282,
      "learning_rate": 9.858000000000001e-06,
      "loss": 0.0732,
      "step": 4930
    },
    {
      "epoch": 4.701614379882812e-06,
      "model_forward_time": 1.7179906368255615,
      "step": 4930
    },
    {
      "dynamics_loss_avg": 0.07029545158147812,
      "epoch": 4.701614379882812e-06,
      "step": 4930
    },
    {
      "action_loss_avg": 0.006287408247590065,
      "epoch": 4.701614379882812e-06,
      "step": 4930
    },
    {
      "epoch": 4.701614379882812e-06,
      "step": 4930,
      "training_step_time": 7.618365049362183
    },
    {
      "epoch": 4.702568054199219e-06,
      "model_forward_time": 1.7182135581970215,
      "step": 4931
    },
    {
      "epoch": 4.702568054199219e-06,
      "step": 4931,
      "training_step_time": 7.646427154541016
    },
    {
      "epoch": 4.703521728515625e-06,
      "model_forward_time": 1.7269630432128906,
      "step": 4932
    },
    {
      "epoch": 4.703521728515625e-06,
      "step": 4932,
      "training_step_time": 7.661858797073364
    },
    {
      "epoch": 4.704475402832031e-06,
      "model_forward_time": 1.7151658535003662,
      "step": 4933
    },
    {
      "epoch": 4.704475402832031e-06,
      "step": 4933,
      "training_step_time": 7.647940397262573
    },
    {
      "epoch": 4.705429077148438e-06,
      "model_forward_time": 1.3471136093139648,
      "step": 4934
    },
    {
      "epoch": 4.705429077148438e-06,
      "step": 4934,
      "training_step_time": 8.010875701904297
    },
    {
      "epoch": 4.706382751464843e-06,
      "model_forward_time": 1.7376141548156738,
      "step": 4935
    },
    {
      "epoch": 4.706382751464843e-06,
      "step": 4935,
      "training_step_time": 7.600618124008179
    },
    {
      "epoch": 4.70733642578125e-06,
      "model_forward_time": 1.7150459289550781,
      "step": 4936
    },
    {
      "epoch": 4.70733642578125e-06,
      "step": 4936,
      "training_step_time": 7.527374505996704
    },
    {
      "epoch": 4.7082901000976565e-06,
      "model_forward_time": 2.8225741386413574,
      "step": 4937
    },
    {
      "epoch": 4.7082901000976565e-06,
      "step": 4937,
      "training_step_time": 7.760192632675171
    },
    {
      "epoch": 4.709243774414062e-06,
      "model_forward_time": 1.711913824081421,
      "step": 4938
    },
    {
      "epoch": 4.709243774414062e-06,
      "step": 4938,
      "training_step_time": 7.735595464706421
    },
    {
      "epoch": 4.710197448730469e-06,
      "model_forward_time": 1.7150442600250244,
      "step": 4939
    },
    {
      "epoch": 4.710197448730469e-06,
      "step": 4939,
      "training_step_time": 7.593500852584839
    },
    {
      "epoch": 4.711151123046875e-06,
      "grad_norm": 0.2385847121477127,
      "learning_rate": 9.878000000000001e-06,
      "loss": 0.0783,
      "step": 4940
    },
    {
      "epoch": 4.711151123046875e-06,
      "model_forward_time": 1.7182908058166504,
      "step": 4940
    },
    {
      "dynamics_loss_avg": 0.07054486460983753,
      "epoch": 4.711151123046875e-06,
      "step": 4940
    },
    {
      "action_loss_avg": 0.006419477169401944,
      "epoch": 4.711151123046875e-06,
      "step": 4940
    },
    {
      "epoch": 4.711151123046875e-06,
      "step": 4940,
      "training_step_time": 9.592747211456299
    },
    {
      "epoch": 4.712104797363281e-06,
      "model_forward_time": 1.7239899635314941,
      "step": 4941
    },
    {
      "epoch": 4.712104797363281e-06,
      "step": 4941,
      "training_step_time": 7.680055856704712
    },
    {
      "epoch": 4.713058471679688e-06,
      "model_forward_time": 1.710869312286377,
      "step": 4942
    },
    {
      "epoch": 4.713058471679688e-06,
      "step": 4942,
      "training_step_time": 7.54599928855896
    },
    {
      "epoch": 4.714012145996093e-06,
      "model_forward_time": 1.7135460376739502,
      "step": 4943
    },
    {
      "epoch": 4.714012145996093e-06,
      "step": 4943,
      "training_step_time": 8.338306188583374
    },
    {
      "epoch": 4.7149658203125e-06,
      "model_forward_time": 1.7238337993621826,
      "step": 4944
    },
    {
      "epoch": 4.7149658203125e-06,
      "step": 4944,
      "training_step_time": 8.387065410614014
    },
    {
      "epoch": 4.7159194946289065e-06,
      "model_forward_time": 1.7135274410247803,
      "step": 4945
    },
    {
      "epoch": 4.7159194946289065e-06,
      "step": 4945,
      "training_step_time": 7.582752227783203
    },
    {
      "epoch": 4.716873168945312e-06,
      "model_forward_time": 1.7169535160064697,
      "step": 4946
    },
    {
      "epoch": 4.716873168945312e-06,
      "step": 4946,
      "training_step_time": 8.716195344924927
    },
    {
      "epoch": 4.717826843261719e-06,
      "model_forward_time": 1.708932638168335,
      "step": 4947
    },
    {
      "epoch": 4.717826843261719e-06,
      "step": 4947,
      "training_step_time": 7.8530097007751465
    },
    {
      "epoch": 4.718780517578125e-06,
      "model_forward_time": 1.7070932388305664,
      "step": 4948
    },
    {
      "epoch": 4.718780517578125e-06,
      "step": 4948,
      "training_step_time": 9.713130950927734
    },
    {
      "epoch": 4.719734191894531e-06,
      "model_forward_time": 1.7104458808898926,
      "step": 4949
    },
    {
      "epoch": 4.719734191894531e-06,
      "step": 4949,
      "training_step_time": 8.902556419372559
    },
    {
      "epoch": 4.720687866210938e-06,
      "grad_norm": 0.34516435861587524,
      "learning_rate": 9.898e-06,
      "loss": 0.0755,
      "step": 4950
    },
    {
      "epoch": 4.720687866210938e-06,
      "model_forward_time": 1.713364601135254,
      "step": 4950
    },
    {
      "dynamics_loss_avg": 0.07638778872787952,
      "epoch": 4.720687866210938e-06,
      "step": 4950
    },
    {
      "action_loss_avg": 0.008862071367911995,
      "epoch": 4.720687866210938e-06,
      "step": 4950
    },
    {
      "epoch": 4.720687866210938e-06,
      "step": 4950,
      "training_step_time": 6.827663421630859
    },
    {
      "epoch": 4.721641540527343e-06,
      "model_forward_time": 1.7040348052978516,
      "step": 4951
    },
    {
      "epoch": 4.721641540527343e-06,
      "step": 4951,
      "training_step_time": 6.8219685554504395
    },
    {
      "epoch": 4.72259521484375e-06,
      "model_forward_time": 1.7145802974700928,
      "step": 4952
    },
    {
      "epoch": 4.72259521484375e-06,
      "step": 4952,
      "training_step_time": 6.79996395111084
    },
    {
      "epoch": 4.7235488891601565e-06,
      "model_forward_time": 1.7159335613250732,
      "step": 4953
    },
    {
      "epoch": 4.7235488891601565e-06,
      "step": 4953,
      "training_step_time": 6.787871599197388
    },
    {
      "epoch": 4.724502563476562e-06,
      "model_forward_time": 1.7134814262390137,
      "step": 4954
    },
    {
      "epoch": 4.724502563476562e-06,
      "step": 4954,
      "training_step_time": 6.7534003257751465
    },
    {
      "epoch": 4.725456237792969e-06,
      "model_forward_time": 1.7092361450195312,
      "step": 4955
    },
    {
      "epoch": 4.725456237792969e-06,
      "step": 4955,
      "training_step_time": 6.83763575553894
    },
    {
      "epoch": 4.726409912109375e-06,
      "model_forward_time": 1.710275650024414,
      "step": 4956
    },
    {
      "epoch": 4.726409912109375e-06,
      "step": 4956,
      "training_step_time": 6.779946327209473
    },
    {
      "epoch": 4.727363586425781e-06,
      "model_forward_time": 1.7204463481903076,
      "step": 4957
    },
    {
      "epoch": 4.727363586425781e-06,
      "step": 4957,
      "training_step_time": 6.778233766555786
    },
    {
      "epoch": 4.728317260742188e-06,
      "model_forward_time": 2.886352777481079,
      "step": 4958
    },
    {
      "epoch": 4.728317260742188e-06,
      "step": 4958,
      "training_step_time": 7.647170066833496
    },
    {
      "epoch": 4.729270935058593e-06,
      "model_forward_time": 1.7145047187805176,
      "step": 4959
    },
    {
      "epoch": 4.729270935058593e-06,
      "step": 4959,
      "training_step_time": 7.876110076904297
    },
    {
      "epoch": 4.730224609375e-06,
      "grad_norm": 0.3537176251411438,
      "learning_rate": 9.918e-06,
      "loss": 0.0716,
      "step": 4960
    },
    {
      "epoch": 4.730224609375e-06,
      "model_forward_time": 1.7106318473815918,
      "step": 4960
    },
    {
      "dynamics_loss_avg": 0.07923212721943855,
      "epoch": 4.730224609375e-06,
      "step": 4960
    },
    {
      "action_loss_avg": 0.008187684393487871,
      "epoch": 4.730224609375e-06,
      "step": 4960
    },
    {
      "epoch": 4.730224609375e-06,
      "step": 4960,
      "training_step_time": 6.787602186203003
    },
    {
      "epoch": 4.7311782836914065e-06,
      "model_forward_time": 1.7108139991760254,
      "step": 4961
    },
    {
      "epoch": 4.7311782836914065e-06,
      "step": 4961,
      "training_step_time": 6.77925443649292
    },
    {
      "epoch": 4.732131958007812e-06,
      "model_forward_time": 1.708817958831787,
      "step": 4962
    },
    {
      "epoch": 4.732131958007812e-06,
      "step": 4962,
      "training_step_time": 6.817594766616821
    },
    {
      "epoch": 4.733085632324219e-06,
      "model_forward_time": 1.720740556716919,
      "step": 4963
    },
    {
      "epoch": 4.733085632324219e-06,
      "step": 4963,
      "training_step_time": 6.792302370071411
    },
    {
      "epoch": 4.734039306640625e-06,
      "model_forward_time": 1.7072713375091553,
      "step": 4964
    },
    {
      "epoch": 4.734039306640625e-06,
      "step": 4964,
      "training_step_time": 6.776142835617065
    },
    {
      "epoch": 4.734992980957031e-06,
      "model_forward_time": 1.712939739227295,
      "step": 4965
    },
    {
      "epoch": 4.734992980957031e-06,
      "step": 4965,
      "training_step_time": 6.8148088455200195
    },
    {
      "epoch": 4.735946655273438e-06,
      "model_forward_time": 1.7106904983520508,
      "step": 4966
    },
    {
      "epoch": 4.735946655273438e-06,
      "step": 4966,
      "training_step_time": 6.764078617095947
    },
    {
      "epoch": 4.736900329589843e-06,
      "model_forward_time": 2.894099235534668,
      "step": 4967
    },
    {
      "epoch": 4.736900329589843e-06,
      "step": 4967,
      "training_step_time": 7.690463066101074
    },
    {
      "epoch": 4.73785400390625e-06,
      "model_forward_time": 1.718156099319458,
      "step": 4968
    },
    {
      "epoch": 4.73785400390625e-06,
      "step": 4968,
      "training_step_time": 7.869561672210693
    },
    {
      "epoch": 4.7388076782226565e-06,
      "model_forward_time": 1.7144486904144287,
      "step": 4969
    },
    {
      "epoch": 4.7388076782226565e-06,
      "step": 4969,
      "training_step_time": 6.818028450012207
    },
    {
      "epoch": 4.739761352539062e-06,
      "grad_norm": 0.2943851351737976,
      "learning_rate": 9.938e-06,
      "loss": 0.0739,
      "step": 4970
    },
    {
      "epoch": 4.739761352539062e-06,
      "model_forward_time": 1.704190731048584,
      "step": 4970
    },
    {
      "dynamics_loss_avg": 0.0740693338215351,
      "epoch": 4.739761352539062e-06,
      "step": 4970
    },
    {
      "action_loss_avg": 0.008258097525686025,
      "epoch": 4.739761352539062e-06,
      "step": 4970
    },
    {
      "epoch": 4.739761352539062e-06,
      "step": 4970,
      "training_step_time": 6.820272445678711
    },
    {
      "epoch": 4.740715026855469e-06,
      "model_forward_time": 1.7199687957763672,
      "step": 4971
    },
    {
      "epoch": 4.740715026855469e-06,
      "step": 4971,
      "training_step_time": 6.779932975769043
    },
    {
      "epoch": 4.741668701171875e-06,
      "model_forward_time": 1.34456467628479,
      "step": 4972
    },
    {
      "epoch": 4.741668701171875e-06,
      "step": 4972,
      "training_step_time": 6.767172813415527
    },
    {
      "epoch": 4.742622375488281e-06,
      "model_forward_time": 1.7097642421722412,
      "step": 4973
    },
    {
      "epoch": 4.742622375488281e-06,
      "step": 4973,
      "training_step_time": 6.810044288635254
    },
    {
      "epoch": 4.743576049804688e-06,
      "model_forward_time": 1.7157936096191406,
      "step": 4974
    },
    {
      "epoch": 4.743576049804688e-06,
      "step": 4974,
      "training_step_time": 6.775352478027344
    },
    {
      "epoch": 4.744529724121093e-06,
      "model_forward_time": 1.7114098072052002,
      "step": 4975
    },
    {
      "epoch": 4.744529724121093e-06,
      "step": 4975,
      "training_step_time": 6.782281875610352
    },
    {
      "epoch": 4.7454833984375e-06,
      "model_forward_time": 2.8886101245880127,
      "step": 4976
    },
    {
      "epoch": 4.7454833984375e-06,
      "step": 4976,
      "training_step_time": 7.6646082401275635
    },
    {
      "epoch": 4.7464370727539065e-06,
      "model_forward_time": 1.7122299671173096,
      "step": 4977
    },
    {
      "epoch": 4.7464370727539065e-06,
      "step": 4977,
      "training_step_time": 9.336949348449707
    },
    {
      "epoch": 4.747390747070312e-06,
      "model_forward_time": 1.7255795001983643,
      "step": 4978
    },
    {
      "epoch": 4.747390747070312e-06,
      "step": 4978,
      "training_step_time": 7.771327018737793
    },
    {
      "epoch": 4.748344421386719e-06,
      "model_forward_time": 1.351592779159546,
      "step": 4979
    },
    {
      "epoch": 4.748344421386719e-06,
      "step": 4979,
      "training_step_time": 6.755879640579224
    },
    {
      "epoch": 4.749298095703125e-06,
      "grad_norm": 0.3938795030117035,
      "learning_rate": 9.958e-06,
      "loss": 0.0668,
      "step": 4980
    },
    {
      "epoch": 4.749298095703125e-06,
      "model_forward_time": 1.727698564529419,
      "step": 4980
    },
    {
      "dynamics_loss_avg": 0.05596264861524105,
      "epoch": 4.749298095703125e-06,
      "step": 4980
    },
    {
      "action_loss_avg": 0.005089464620687068,
      "epoch": 4.749298095703125e-06,
      "step": 4980
    },
    {
      "epoch": 4.749298095703125e-06,
      "step": 4980,
      "training_step_time": 6.781733751296997
    },
    {
      "epoch": 4.750251770019531e-06,
      "model_forward_time": 1.7246060371398926,
      "step": 4981
    },
    {
      "epoch": 4.750251770019531e-06,
      "step": 4981,
      "training_step_time": 6.816448211669922
    },
    {
      "epoch": 4.751205444335938e-06,
      "model_forward_time": 1.7232041358947754,
      "step": 4982
    },
    {
      "epoch": 4.751205444335938e-06,
      "step": 4982,
      "training_step_time": 6.817411661148071
    },
    {
      "epoch": 4.752159118652343e-06,
      "model_forward_time": 1.7238562107086182,
      "step": 4983
    },
    {
      "epoch": 4.752159118652343e-06,
      "step": 4983,
      "training_step_time": 6.85263466835022
    },
    {
      "epoch": 4.75311279296875e-06,
      "model_forward_time": 1.7307465076446533,
      "step": 4984
    },
    {
      "epoch": 4.75311279296875e-06,
      "step": 4984,
      "training_step_time": 6.788706302642822
    },
    {
      "epoch": 4.7540664672851566e-06,
      "model_forward_time": 2.876455783843994,
      "step": 4985
    },
    {
      "epoch": 4.7540664672851566e-06,
      "step": 4985,
      "training_step_time": 8.53014326095581
    },
    {
      "epoch": 4.755020141601562e-06,
      "model_forward_time": 1.726320505142212,
      "step": 4986
    },
    {
      "epoch": 4.755020141601562e-06,
      "step": 4986,
      "training_step_time": 10.581741333007812
    },
    {
      "epoch": 4.755973815917969e-06,
      "model_forward_time": 1.7282352447509766,
      "step": 4987
    },
    {
      "epoch": 4.755973815917969e-06,
      "step": 4987,
      "training_step_time": 7.7413718700408936
    },
    {
      "epoch": 4.756927490234375e-06,
      "model_forward_time": 1.7248907089233398,
      "step": 4988
    },
    {
      "epoch": 4.756927490234375e-06,
      "step": 4988,
      "training_step_time": 6.7596681118011475
    },
    {
      "epoch": 4.757881164550781e-06,
      "model_forward_time": 1.7185485363006592,
      "step": 4989
    },
    {
      "epoch": 4.757881164550781e-06,
      "step": 4989,
      "training_step_time": 6.802389860153198
    },
    {
      "epoch": 4.758834838867188e-06,
      "grad_norm": 0.2967859208583832,
      "learning_rate": 9.978000000000002e-06,
      "loss": 0.0705,
      "step": 4990
    },
    {
      "epoch": 4.758834838867188e-06,
      "model_forward_time": 1.7248005867004395,
      "step": 4990
    },
    {
      "dynamics_loss_avg": 0.04833215363323688,
      "epoch": 4.758834838867188e-06,
      "step": 4990
    },
    {
      "action_loss_avg": 0.005167466506827622,
      "epoch": 4.758834838867188e-06,
      "step": 4990
    },
    {
      "epoch": 4.758834838867188e-06,
      "step": 4990,
      "training_step_time": 6.8109986782073975
    },
    {
      "epoch": 4.7597885131835934e-06,
      "model_forward_time": 1.733731746673584,
      "step": 4991
    },
    {
      "epoch": 4.7597885131835934e-06,
      "step": 4991,
      "training_step_time": 6.804029703140259
    },
    {
      "epoch": 4.7607421875e-06,
      "model_forward_time": 1.7271993160247803,
      "step": 4992
    },
    {
      "epoch": 4.7607421875e-06,
      "step": 4992,
      "training_step_time": 6.757490396499634
    },
    {
      "epoch": 4.7616958618164066e-06,
      "model_forward_time": 1.724581241607666,
      "step": 4993
    },
    {
      "epoch": 4.7616958618164066e-06,
      "step": 4993,
      "training_step_time": 6.825143337249756
    },
    {
      "epoch": 4.762649536132812e-06,
      "model_forward_time": 2.8603525161743164,
      "step": 4994
    },
    {
      "epoch": 4.762649536132812e-06,
      "step": 4994,
      "training_step_time": 8.162492036819458
    },
    {
      "epoch": 4.763603210449219e-06,
      "model_forward_time": 1.741210699081421,
      "step": 4995
    },
    {
      "epoch": 4.763603210449219e-06,
      "step": 4995,
      "training_step_time": 10.851438283920288
    },
    {
      "epoch": 4.764556884765625e-06,
      "model_forward_time": 1.7186665534973145,
      "step": 4996
    },
    {
      "epoch": 4.764556884765625e-06,
      "step": 4996,
      "training_step_time": 7.877602577209473
    },
    {
      "epoch": 4.765510559082031e-06,
      "model_forward_time": 1.720560073852539,
      "step": 4997
    },
    {
      "epoch": 4.765510559082031e-06,
      "step": 4997,
      "training_step_time": 6.81362247467041
    },
    {
      "epoch": 4.766464233398438e-06,
      "model_forward_time": 1.7913620471954346,
      "step": 4998
    },
    {
      "epoch": 4.766464233398438e-06,
      "step": 4998,
      "training_step_time": 6.972959280014038
    },
    {
      "epoch": 4.7674179077148434e-06,
      "model_forward_time": 1.7639517784118652,
      "step": 4999
    },
    {
      "epoch": 4.7674179077148434e-06,
      "step": 4999,
      "training_step_time": 6.957680940628052
    },
    {
      "epoch": 4.76837158203125e-06,
      "grad_norm": 0.29561787843704224,
      "learning_rate": 9.998000000000002e-06,
      "loss": 0.0766,
      "step": 5000
    },
    {
      "epoch": 4.76837158203125e-06,
      "model_forward_time": 1.7442524433135986,
      "step": 5000
    },
    {
      "dynamics_loss_avg": 0.06442425828427076,
      "epoch": 4.76837158203125e-06,
      "step": 5000
    },
    {
      "action_loss_avg": 0.005403555414522998,
      "epoch": 4.76837158203125e-06,
      "step": 5000
    },
    {
      "epoch": 4.76837158203125e-06,
      "step": 5000,
      "training_step_time": 6.689849138259888
    },
    {
      "epoch": 4.7693252563476566e-06,
      "model_forward_time": 1.7563631534576416,
      "step": 5001
    },
    {
      "epoch": 4.7693252563476566e-06,
      "step": 5001,
      "training_step_time": 6.805801153182983
    },
    {
      "epoch": 4.770278930664062e-06,
      "model_forward_time": 1.7459723949432373,
      "step": 5002
    },
    {
      "epoch": 4.770278930664062e-06,
      "step": 5002,
      "training_step_time": 7.358229875564575
    },
    {
      "epoch": 4.771232604980469e-06,
      "model_forward_time": 3.3599865436553955,
      "step": 5003
    },
    {
      "epoch": 4.771232604980469e-06,
      "step": 5003,
      "training_step_time": 8.733021020889282
    },
    {
      "epoch": 4.772186279296875e-06,
      "model_forward_time": 1.807856559753418,
      "step": 5004
    },
    {
      "epoch": 4.772186279296875e-06,
      "step": 5004,
      "training_step_time": 11.421250581741333
    },
    {
      "epoch": 4.773139953613281e-06,
      "model_forward_time": 1.7283737659454346,
      "step": 5005
    },
    {
      "epoch": 4.773139953613281e-06,
      "step": 5005,
      "training_step_time": 7.8666346073150635
    },
    {
      "epoch": 4.774093627929688e-06,
      "model_forward_time": 1.734184741973877,
      "step": 5006
    },
    {
      "epoch": 4.774093627929688e-06,
      "step": 5006,
      "training_step_time": 6.800997734069824
    },
    {
      "epoch": 4.7750473022460934e-06,
      "model_forward_time": 1.732837200164795,
      "step": 5007
    },
    {
      "epoch": 4.7750473022460934e-06,
      "step": 5007,
      "training_step_time": 6.750338792800903
    },
    {
      "epoch": 4.7760009765625e-06,
      "model_forward_time": 1.7356901168823242,
      "step": 5008
    },
    {
      "epoch": 4.7760009765625e-06,
      "step": 5008,
      "training_step_time": 6.788820743560791
    },
    {
      "epoch": 4.7769546508789066e-06,
      "model_forward_time": 1.7323243618011475,
      "step": 5009
    },
    {
      "epoch": 4.7769546508789066e-06,
      "step": 5009,
      "training_step_time": 6.80419135093689
    },
    {
      "epoch": 4.777908325195312e-06,
      "grad_norm": 0.33644992113113403,
      "learning_rate": 9.999999778549046e-06,
      "loss": 0.0773,
      "step": 5010
    },
    {
      "epoch": 4.777908325195312e-06,
      "model_forward_time": 1.7400016784667969,
      "step": 5010
    },
    {
      "dynamics_loss_avg": 0.061872022412717345,
      "epoch": 4.777908325195312e-06,
      "step": 5010
    },
    {
      "action_loss_avg": 0.007028865744359791,
      "epoch": 4.777908325195312e-06,
      "step": 5010
    },
    {
      "epoch": 4.777908325195312e-06,
      "step": 5010,
      "training_step_time": 6.764687538146973
    },
    {
      "epoch": 4.778861999511719e-06,
      "model_forward_time": 1.7352592945098877,
      "step": 5011
    },
    {
      "epoch": 4.778861999511719e-06,
      "step": 5011,
      "training_step_time": 6.765196084976196
    },
    {
      "epoch": 4.779815673828125e-06,
      "model_forward_time": 2.8731911182403564,
      "step": 5012
    },
    {
      "epoch": 4.779815673828125e-06,
      "step": 5012,
      "training_step_time": 7.592128276824951
    },
    {
      "epoch": 4.780769348144531e-06,
      "model_forward_time": 1.8081309795379639,
      "step": 5013
    },
    {
      "epoch": 4.780769348144531e-06,
      "step": 5013,
      "training_step_time": 9.915610790252686
    },
    {
      "epoch": 4.781723022460938e-06,
      "model_forward_time": 1.7315177917480469,
      "step": 5014
    },
    {
      "epoch": 4.781723022460938e-06,
      "step": 5014,
      "training_step_time": 7.707399129867554
    },
    {
      "epoch": 4.7826766967773434e-06,
      "model_forward_time": 1.74476957321167,
      "step": 5015
    },
    {
      "epoch": 4.7826766967773434e-06,
      "step": 5015,
      "training_step_time": 7.424760580062866
    },
    {
      "epoch": 4.78363037109375e-06,
      "model_forward_time": 1.7332451343536377,
      "step": 5016
    },
    {
      "epoch": 4.78363037109375e-06,
      "step": 5016,
      "training_step_time": 6.7356414794921875
    },
    {
      "epoch": 4.784584045410157e-06,
      "model_forward_time": 1.7345349788665771,
      "step": 5017
    },
    {
      "epoch": 4.784584045410157e-06,
      "step": 5017,
      "training_step_time": 6.753112077713013
    },
    {
      "epoch": 4.785537719726562e-06,
      "model_forward_time": 1.7322323322296143,
      "step": 5018
    },
    {
      "epoch": 4.785537719726562e-06,
      "step": 5018,
      "training_step_time": 6.77461576461792
    },
    {
      "epoch": 4.786491394042969e-06,
      "model_forward_time": 1.7326061725616455,
      "step": 5019
    },
    {
      "epoch": 4.786491394042969e-06,
      "step": 5019,
      "training_step_time": 6.781363010406494
    },
    {
      "epoch": 4.787445068359375e-06,
      "grad_norm": 0.27877068519592285,
      "learning_rate": 9.999999013039593e-06,
      "loss": 0.076,
      "step": 5020
    },
    {
      "epoch": 4.787445068359375e-06,
      "model_forward_time": 1.7464756965637207,
      "step": 5020
    },
    {
      "dynamics_loss_avg": 0.05903477277606726,
      "epoch": 4.787445068359375e-06,
      "step": 5020
    },
    {
      "action_loss_avg": 0.009068593138363213,
      "epoch": 4.787445068359375e-06,
      "step": 5020
    },
    {
      "epoch": 4.787445068359375e-06,
      "step": 5020,
      "training_step_time": 6.71320366859436
    },
    {
      "epoch": 4.788398742675781e-06,
      "model_forward_time": 2.859081983566284,
      "step": 5021
    },
    {
      "epoch": 4.788398742675781e-06,
      "step": 5021,
      "training_step_time": 7.597134828567505
    },
    {
      "epoch": 4.789352416992188e-06,
      "model_forward_time": 1.8046953678131104,
      "step": 5022
    },
    {
      "epoch": 4.789352416992188e-06,
      "step": 5022,
      "training_step_time": 8.633934020996094
    },
    {
      "epoch": 4.7903060913085935e-06,
      "model_forward_time": 1.7304449081420898,
      "step": 5023
    },
    {
      "epoch": 4.7903060913085935e-06,
      "step": 5023,
      "training_step_time": 7.695075273513794
    },
    {
      "epoch": 4.791259765625e-06,
      "model_forward_time": 1.73240065574646,
      "step": 5024
    },
    {
      "epoch": 4.791259765625e-06,
      "step": 5024,
      "training_step_time": 8.406339406967163
    },
    {
      "epoch": 4.792213439941407e-06,
      "model_forward_time": 1.7649030685424805,
      "step": 5025
    },
    {
      "epoch": 4.792213439941407e-06,
      "step": 5025,
      "training_step_time": 6.7286376953125
    },
    {
      "epoch": 4.793167114257812e-06,
      "model_forward_time": 1.7420134544372559,
      "step": 5026
    },
    {
      "epoch": 4.793167114257812e-06,
      "step": 5026,
      "training_step_time": 6.762989282608032
    },
    {
      "epoch": 4.794120788574219e-06,
      "model_forward_time": 1.7485630512237549,
      "step": 5027
    },
    {
      "epoch": 4.794120788574219e-06,
      "step": 5027,
      "training_step_time": 6.747618198394775
    },
    {
      "epoch": 4.795074462890625e-06,
      "model_forward_time": 1.7593646049499512,
      "step": 5028
    },
    {
      "epoch": 4.795074462890625e-06,
      "step": 5028,
      "training_step_time": 6.726922988891602
    },
    {
      "epoch": 4.796028137207031e-06,
      "model_forward_time": 1.384845495223999,
      "step": 5029
    },
    {
      "epoch": 4.796028137207031e-06,
      "step": 5029,
      "training_step_time": 6.72963547706604
    },
    {
      "epoch": 4.796981811523438e-06,
      "grad_norm": 0.37726640701293945,
      "learning_rate": 9.999997700737766e-06,
      "loss": 0.0729,
      "step": 5030
    },
    {
      "epoch": 4.796981811523438e-06,
      "model_forward_time": 1.7491559982299805,
      "step": 5030
    },
    {
      "dynamics_loss_avg": 0.06746060214936733,
      "epoch": 4.796981811523438e-06,
      "step": 5030
    },
    {
      "action_loss_avg": 0.0083067249157466,
      "epoch": 4.796981811523438e-06,
      "step": 5030
    },
    {
      "epoch": 4.796981811523438e-06,
      "step": 5030,
      "training_step_time": 7.097747802734375
    },
    {
      "epoch": 4.7979354858398435e-06,
      "model_forward_time": 2.836550712585449,
      "step": 5031
    },
    {
      "epoch": 4.7979354858398435e-06,
      "step": 5031,
      "training_step_time": 7.613817930221558
    },
    {
      "epoch": 4.79888916015625e-06,
      "model_forward_time": 1.7682805061340332,
      "step": 5032
    },
    {
      "epoch": 4.79888916015625e-06,
      "step": 5032,
      "training_step_time": 9.920687437057495
    },
    {
      "epoch": 4.799842834472657e-06,
      "model_forward_time": 1.74558687210083,
      "step": 5033
    },
    {
      "epoch": 4.799842834472657e-06,
      "step": 5033,
      "training_step_time": 7.597099304199219
    },
    {
      "epoch": 4.800796508789062e-06,
      "model_forward_time": 1.7488563060760498,
      "step": 5034
    },
    {
      "epoch": 4.800796508789062e-06,
      "step": 5034,
      "training_step_time": 7.655807256698608
    },
    {
      "epoch": 4.801750183105469e-06,
      "model_forward_time": 1.7571461200714111,
      "step": 5035
    },
    {
      "epoch": 4.801750183105469e-06,
      "step": 5035,
      "training_step_time": 6.838984251022339
    },
    {
      "epoch": 4.802703857421875e-06,
      "model_forward_time": 1.749213695526123,
      "step": 5036
    },
    {
      "epoch": 4.802703857421875e-06,
      "step": 5036,
      "training_step_time": 6.779305458068848
    },
    {
      "epoch": 4.803657531738281e-06,
      "model_forward_time": 1.7423551082611084,
      "step": 5037
    },
    {
      "epoch": 4.803657531738281e-06,
      "step": 5037,
      "training_step_time": 6.800827264785767
    },
    {
      "epoch": 4.804611206054688e-06,
      "model_forward_time": 1.7569410800933838,
      "step": 5038
    },
    {
      "epoch": 4.804611206054688e-06,
      "step": 5038,
      "training_step_time": 6.733168840408325
    },
    {
      "epoch": 4.8055648803710935e-06,
      "model_forward_time": 1.7593026161193848,
      "step": 5039
    },
    {
      "epoch": 4.8055648803710935e-06,
      "step": 5039,
      "training_step_time": 7.37253212928772
    },
    {
      "epoch": 4.8065185546875e-06,
      "grad_norm": 0.21716712415218353,
      "learning_rate": 9.999995841643709e-06,
      "loss": 0.0732,
      "step": 5040
    },
    {
      "epoch": 4.8065185546875e-06,
      "model_forward_time": 2.977308511734009,
      "step": 5040
    },
    {
      "dynamics_loss_avg": 0.09133684821426868,
      "epoch": 4.8065185546875e-06,
      "step": 5040
    },
    {
      "action_loss_avg": 0.011420134175568818,
      "epoch": 4.8065185546875e-06,
      "step": 5040
    },
    {
      "epoch": 4.8065185546875e-06,
      "step": 5040,
      "training_step_time": 8.290251731872559
    },
    {
      "epoch": 4.807472229003907e-06,
      "model_forward_time": 1.7537319660186768,
      "step": 5041
    },
    {
      "epoch": 4.807472229003907e-06,
      "step": 5041,
      "training_step_time": 9.235995531082153
    },
    {
      "epoch": 4.808425903320312e-06,
      "model_forward_time": 1.743086814880371,
      "step": 5042
    },
    {
      "epoch": 4.808425903320312e-06,
      "step": 5042,
      "training_step_time": 7.5125203132629395
    },
    {
      "epoch": 4.809379577636719e-06,
      "model_forward_time": 1.747406005859375,
      "step": 5043
    },
    {
      "epoch": 4.809379577636719e-06,
      "step": 5043,
      "training_step_time": 7.5727128982543945
    },
    {
      "epoch": 4.810333251953125e-06,
      "model_forward_time": 1.7418322563171387,
      "step": 5044
    },
    {
      "epoch": 4.810333251953125e-06,
      "step": 5044,
      "training_step_time": 6.787929534912109
    },
    {
      "epoch": 4.811286926269531e-06,
      "model_forward_time": 1.7470722198486328,
      "step": 5045
    },
    {
      "epoch": 4.811286926269531e-06,
      "step": 5045,
      "training_step_time": 6.7707273960113525
    },
    {
      "epoch": 4.812240600585938e-06,
      "model_forward_time": 1.7500865459442139,
      "step": 5046
    },
    {
      "epoch": 4.812240600585938e-06,
      "step": 5046,
      "training_step_time": 6.81735372543335
    },
    {
      "epoch": 4.8131942749023435e-06,
      "model_forward_time": 1.3846755027770996,
      "step": 5047
    },
    {
      "epoch": 4.8131942749023435e-06,
      "step": 5047,
      "training_step_time": 6.768242120742798
    },
    {
      "epoch": 4.81414794921875e-06,
      "model_forward_time": 1.7447693347930908,
      "step": 5048
    },
    {
      "epoch": 4.81414794921875e-06,
      "step": 5048,
      "training_step_time": 6.828215837478638
    },
    {
      "epoch": 4.815101623535157e-06,
      "model_forward_time": 2.9170167446136475,
      "step": 5049
    },
    {
      "epoch": 4.815101623535157e-06,
      "step": 5049,
      "training_step_time": 9.380356788635254
    },
    {
      "epoch": 4.816055297851562e-06,
      "grad_norm": 0.2676977217197418,
      "learning_rate": 9.999993435757623e-06,
      "loss": 0.0712,
      "step": 5050
    },
    {
      "epoch": 4.816055297851562e-06,
      "model_forward_time": 1.744272232055664,
      "step": 5050
    },
    {
      "dynamics_loss_avg": 0.07603609748184681,
      "epoch": 4.816055297851562e-06,
      "step": 5050
    },
    {
      "action_loss_avg": 0.00776148239383474,
      "epoch": 4.816055297851562e-06,
      "step": 5050
    },
    {
      "epoch": 4.816055297851562e-06,
      "step": 5050,
      "training_step_time": 9.974109411239624
    },
    {
      "epoch": 4.817008972167969e-06,
      "model_forward_time": 1.770477294921875,
      "step": 5051
    },
    {
      "epoch": 4.817008972167969e-06,
      "step": 5051,
      "training_step_time": 7.60991644859314
    },
    {
      "epoch": 4.817962646484375e-06,
      "model_forward_time": 1.749237298965454,
      "step": 5052
    },
    {
      "epoch": 4.817962646484375e-06,
      "step": 5052,
      "training_step_time": 7.517919540405273
    },
    {
      "epoch": 4.818916320800781e-06,
      "model_forward_time": 1.7645134925842285,
      "step": 5053
    },
    {
      "epoch": 4.818916320800781e-06,
      "step": 5053,
      "training_step_time": 6.737967491149902
    },
    {
      "epoch": 4.819869995117188e-06,
      "model_forward_time": 1.7466182708740234,
      "step": 5054
    },
    {
      "epoch": 4.819869995117188e-06,
      "step": 5054,
      "training_step_time": 6.735663175582886
    },
    {
      "epoch": 4.8208236694335935e-06,
      "model_forward_time": 1.7509028911590576,
      "step": 5055
    },
    {
      "epoch": 4.8208236694335935e-06,
      "step": 5055,
      "training_step_time": 6.74582839012146
    },
    {
      "epoch": 4.82177734375e-06,
      "model_forward_time": 1.7665395736694336,
      "step": 5056
    },
    {
      "epoch": 4.82177734375e-06,
      "step": 5056,
      "training_step_time": 6.7224953174591064
    },
    {
      "epoch": 4.822731018066407e-06,
      "model_forward_time": 1.774768590927124,
      "step": 5057
    },
    {
      "epoch": 4.822731018066407e-06,
      "step": 5057,
      "training_step_time": 6.69624662399292
    },
    {
      "epoch": 4.823684692382812e-06,
      "model_forward_time": 2.8984720706939697,
      "step": 5058
    },
    {
      "epoch": 4.823684692382812e-06,
      "step": 5058,
      "training_step_time": 8.481878995895386
    },
    {
      "epoch": 4.824638366699219e-06,
      "model_forward_time": 1.749340295791626,
      "step": 5059
    },
    {
      "epoch": 4.824638366699219e-06,
      "step": 5059,
      "training_step_time": 9.898436784744263
    },
    {
      "epoch": 4.825592041015625e-06,
      "grad_norm": 0.2245299369096756,
      "learning_rate": 9.999990483079773e-06,
      "loss": 0.0757,
      "step": 5060
    },
    {
      "epoch": 4.825592041015625e-06,
      "model_forward_time": 1.410205364227295,
      "step": 5060
    },
    {
      "dynamics_loss_avg": 0.04804970072582364,
      "epoch": 4.825592041015625e-06,
      "step": 5060
    },
    {
      "action_loss_avg": 0.006376598647329956,
      "epoch": 4.825592041015625e-06,
      "step": 5060
    },
    {
      "epoch": 4.825592041015625e-06,
      "step": 5060,
      "training_step_time": 8.335378646850586
    },
    {
      "epoch": 4.826545715332031e-06,
      "model_forward_time": 1.7566196918487549,
      "step": 5061
    },
    {
      "epoch": 4.826545715332031e-06,
      "step": 5061,
      "training_step_time": 7.549854040145874
    },
    {
      "epoch": 4.827499389648438e-06,
      "model_forward_time": 1.7529454231262207,
      "step": 5062
    },
    {
      "epoch": 4.827499389648438e-06,
      "step": 5062,
      "training_step_time": 7.536149740219116
    },
    {
      "epoch": 4.8284530639648435e-06,
      "model_forward_time": 1.744807243347168,
      "step": 5063
    },
    {
      "epoch": 4.8284530639648435e-06,
      "step": 5063,
      "training_step_time": 6.738789319992065
    },
    {
      "epoch": 4.82940673828125e-06,
      "model_forward_time": 1.748772144317627,
      "step": 5064
    },
    {
      "epoch": 4.82940673828125e-06,
      "step": 5064,
      "training_step_time": 6.749106168746948
    },
    {
      "epoch": 4.830360412597657e-06,
      "model_forward_time": 1.7616024017333984,
      "step": 5065
    },
    {
      "epoch": 4.830360412597657e-06,
      "step": 5065,
      "training_step_time": 6.777034521102905
    },
    {
      "epoch": 4.831314086914062e-06,
      "model_forward_time": 1.754321575164795,
      "step": 5066
    },
    {
      "epoch": 4.831314086914062e-06,
      "step": 5066,
      "training_step_time": 6.743376731872559
    },
    {
      "epoch": 4.832267761230469e-06,
      "model_forward_time": 1.833221673965454,
      "step": 5067
    },
    {
      "epoch": 4.832267761230469e-06,
      "step": 5067,
      "training_step_time": 7.081220388412476
    },
    {
      "epoch": 4.833221435546875e-06,
      "model_forward_time": 2.8775618076324463,
      "step": 5068
    },
    {
      "epoch": 4.833221435546875e-06,
      "step": 5068,
      "training_step_time": 8.357022047042847
    },
    {
      "epoch": 4.834175109863281e-06,
      "model_forward_time": 1.753896951675415,
      "step": 5069
    },
    {
      "epoch": 4.834175109863281e-06,
      "step": 5069,
      "training_step_time": 8.376379013061523
    },
    {
      "epoch": 4.835128784179688e-06,
      "grad_norm": 0.43044739961624146,
      "learning_rate": 9.999986983610482e-06,
      "loss": 0.0699,
      "step": 5070
    },
    {
      "epoch": 4.835128784179688e-06,
      "model_forward_time": 1.388641595840454,
      "step": 5070
    },
    {
      "dynamics_loss_avg": 0.06511255837976933,
      "epoch": 4.835128784179688e-06,
      "step": 5070
    },
    {
      "action_loss_avg": 0.008203373942524194,
      "epoch": 4.835128784179688e-06,
      "step": 5070
    },
    {
      "epoch": 4.835128784179688e-06,
      "step": 5070,
      "training_step_time": 7.56605339050293
    },
    {
      "epoch": 4.8360824584960935e-06,
      "model_forward_time": 1.7520267963409424,
      "step": 5071
    },
    {
      "epoch": 4.8360824584960935e-06,
      "step": 5071,
      "training_step_time": 7.594663143157959
    },
    {
      "epoch": 4.8370361328125e-06,
      "model_forward_time": 1.755619764328003,
      "step": 5072
    },
    {
      "epoch": 4.8370361328125e-06,
      "step": 5072,
      "training_step_time": 6.782253265380859
    },
    {
      "epoch": 4.837989807128907e-06,
      "model_forward_time": 1.7580277919769287,
      "step": 5073
    },
    {
      "epoch": 4.837989807128907e-06,
      "step": 5073,
      "training_step_time": 6.789710760116577
    },
    {
      "epoch": 4.838943481445312e-06,
      "model_forward_time": 1.7462012767791748,
      "step": 5074
    },
    {
      "epoch": 4.838943481445312e-06,
      "step": 5074,
      "training_step_time": 6.779980421066284
    },
    {
      "epoch": 4.839897155761719e-06,
      "model_forward_time": 1.7553670406341553,
      "step": 5075
    },
    {
      "epoch": 4.839897155761719e-06,
      "step": 5075,
      "training_step_time": 6.7984137535095215
    },
    {
      "epoch": 4.840850830078125e-06,
      "model_forward_time": 1.8194987773895264,
      "step": 5076
    },
    {
      "epoch": 4.840850830078125e-06,
      "step": 5076,
      "training_step_time": 7.9311909675598145
    },
    {
      "epoch": 4.841804504394531e-06,
      "model_forward_time": 2.897482395172119,
      "step": 5077
    },
    {
      "epoch": 4.841804504394531e-06,
      "step": 5077,
      "training_step_time": 9.755261182785034
    },
    {
      "epoch": 4.842758178710938e-06,
      "model_forward_time": 1.7463254928588867,
      "step": 5078
    },
    {
      "epoch": 4.842758178710938e-06,
      "step": 5078,
      "training_step_time": 8.693631887435913
    },
    {
      "epoch": 4.8437118530273435e-06,
      "model_forward_time": 1.7602519989013672,
      "step": 5079
    },
    {
      "epoch": 4.8437118530273435e-06,
      "step": 5079,
      "training_step_time": 8.336113214492798
    },
    {
      "epoch": 4.84466552734375e-06,
      "grad_norm": 0.2501251697540283,
      "learning_rate": 9.99998293735013e-06,
      "loss": 0.0707,
      "step": 5080
    },
    {
      "epoch": 4.84466552734375e-06,
      "model_forward_time": 1.7535269260406494,
      "step": 5080
    },
    {
      "dynamics_loss_avg": 0.05031269527971745,
      "epoch": 4.84466552734375e-06,
      "step": 5080
    },
    {
      "action_loss_avg": 0.00463106952374801,
      "epoch": 4.84466552734375e-06,
      "step": 5080
    },
    {
      "epoch": 4.84466552734375e-06,
      "step": 5080,
      "training_step_time": 7.592858076095581
    },
    {
      "epoch": 4.845619201660157e-06,
      "model_forward_time": 1.7457952499389648,
      "step": 5081
    },
    {
      "epoch": 4.845619201660157e-06,
      "step": 5081,
      "training_step_time": 6.774338722229004
    },
    {
      "epoch": 4.846572875976562e-06,
      "model_forward_time": 1.7681055068969727,
      "step": 5082
    },
    {
      "epoch": 4.846572875976562e-06,
      "step": 5082,
      "training_step_time": 6.795061111450195
    },
    {
      "epoch": 4.847526550292969e-06,
      "model_forward_time": 1.7732534408569336,
      "step": 5083
    },
    {
      "epoch": 4.847526550292969e-06,
      "step": 5083,
      "training_step_time": 6.719597816467285
    },
    {
      "epoch": 4.848480224609375e-06,
      "model_forward_time": 1.7572979927062988,
      "step": 5084
    },
    {
      "epoch": 4.848480224609375e-06,
      "step": 5084,
      "training_step_time": 6.736939191818237
    },
    {
      "epoch": 4.849433898925781e-06,
      "model_forward_time": 1.819359540939331,
      "step": 5085
    },
    {
      "epoch": 4.849433898925781e-06,
      "step": 5085,
      "training_step_time": 6.999394178390503
    },
    {
      "epoch": 4.850387573242188e-06,
      "model_forward_time": 2.9514098167419434,
      "step": 5086
    },
    {
      "epoch": 4.850387573242188e-06,
      "step": 5086,
      "training_step_time": 10.202009201049805
    },
    {
      "epoch": 4.8513412475585935e-06,
      "model_forward_time": 1.7584972381591797,
      "step": 5087
    },
    {
      "epoch": 4.8513412475585935e-06,
      "step": 5087,
      "training_step_time": 9.969737529754639
    },
    {
      "epoch": 4.852294921875e-06,
      "model_forward_time": 1.7558839321136475,
      "step": 5088
    },
    {
      "epoch": 4.852294921875e-06,
      "step": 5088,
      "training_step_time": 7.616586923599243
    },
    {
      "epoch": 4.853248596191407e-06,
      "model_forward_time": 1.7524774074554443,
      "step": 5089
    },
    {
      "epoch": 4.853248596191407e-06,
      "step": 5089,
      "training_step_time": 7.5049216747283936
    },
    {
      "epoch": 4.854202270507812e-06,
      "grad_norm": 0.33826130628585815,
      "learning_rate": 9.999978344299161e-06,
      "loss": 0.0691,
      "step": 5090
    },
    {
      "epoch": 4.854202270507812e-06,
      "model_forward_time": 1.7564616203308105,
      "step": 5090
    },
    {
      "dynamics_loss_avg": 0.0693915638141334,
      "epoch": 4.854202270507812e-06,
      "step": 5090
    },
    {
      "action_loss_avg": 0.010879501095041632,
      "epoch": 4.854202270507812e-06,
      "step": 5090
    },
    {
      "epoch": 4.854202270507812e-06,
      "step": 5090,
      "training_step_time": 6.719892263412476
    },
    {
      "epoch": 4.855155944824219e-06,
      "model_forward_time": 1.7693419456481934,
      "step": 5091
    },
    {
      "epoch": 4.855155944824219e-06,
      "step": 5091,
      "training_step_time": 6.721500396728516
    },
    {
      "epoch": 4.856109619140625e-06,
      "model_forward_time": 1.7518935203552246,
      "step": 5092
    },
    {
      "epoch": 4.856109619140625e-06,
      "step": 5092,
      "training_step_time": 6.6604084968566895
    },
    {
      "epoch": 4.857063293457031e-06,
      "model_forward_time": 1.7524244785308838,
      "step": 5093
    },
    {
      "epoch": 4.857063293457031e-06,
      "step": 5093,
      "training_step_time": 6.6793012619018555
    },
    {
      "epoch": 4.858016967773438e-06,
      "model_forward_time": 1.815669059753418,
      "step": 5094
    },
    {
      "epoch": 4.858016967773438e-06,
      "step": 5094,
      "training_step_time": 6.9795567989349365
    },
    {
      "epoch": 4.8589706420898435e-06,
      "model_forward_time": 2.8699753284454346,
      "step": 5095
    },
    {
      "epoch": 4.8589706420898435e-06,
      "step": 5095,
      "training_step_time": 7.616469621658325
    },
    {
      "epoch": 4.85992431640625e-06,
      "model_forward_time": 1.7611093521118164,
      "step": 5096
    },
    {
      "epoch": 4.85992431640625e-06,
      "step": 5096,
      "training_step_time": 8.27406907081604
    },
    {
      "epoch": 4.860877990722657e-06,
      "model_forward_time": 1.7494392395019531,
      "step": 5097
    },
    {
      "epoch": 4.860877990722657e-06,
      "step": 5097,
      "training_step_time": 7.716807842254639
    },
    {
      "epoch": 4.861831665039062e-06,
      "model_forward_time": 1.7402565479278564,
      "step": 5098
    },
    {
      "epoch": 4.861831665039062e-06,
      "step": 5098,
      "training_step_time": 7.553501844406128
    },
    {
      "epoch": 4.862785339355469e-06,
      "model_forward_time": 1.7678577899932861,
      "step": 5099
    },
    {
      "epoch": 4.862785339355469e-06,
      "step": 5099,
      "training_step_time": 6.764578104019165
    },
    {
      "epoch": 4.863739013671875e-06,
      "grad_norm": 0.2726636230945587,
      "learning_rate": 9.99997320445808e-06,
      "loss": 0.0719,
      "step": 5100
    },
    {
      "epoch": 4.863739013671875e-06,
      "model_forward_time": 1.7461309432983398,
      "step": 5100
    },
    {
      "dynamics_loss_avg": 0.05864730123430491,
      "epoch": 4.863739013671875e-06,
      "step": 5100
    },
    {
      "action_loss_avg": 0.006224510190077126,
      "epoch": 4.863739013671875e-06,
      "step": 5100
    },
    {
      "epoch": 4.863739013671875e-06,
      "step": 5100,
      "training_step_time": 6.736954212188721
    },
    {
      "epoch": 4.864692687988281e-06,
      "model_forward_time": 1.7646028995513916,
      "step": 5101
    },
    {
      "epoch": 4.864692687988281e-06,
      "step": 5101,
      "training_step_time": 6.842705726623535
    },
    {
      "epoch": 4.865646362304688e-06,
      "model_forward_time": 1.3835079669952393,
      "step": 5102
    },
    {
      "epoch": 4.865646362304688e-06,
      "step": 5102,
      "training_step_time": 6.741501808166504
    },
    {
      "epoch": 4.8666000366210935e-06,
      "model_forward_time": 1.8166627883911133,
      "step": 5103
    },
    {
      "epoch": 4.8666000366210935e-06,
      "step": 5103,
      "training_step_time": 6.999483108520508
    },
    {
      "epoch": 4.8675537109375e-06,
      "model_forward_time": 2.8617095947265625,
      "step": 5104
    },
    {
      "epoch": 4.8675537109375e-06,
      "step": 5104,
      "training_step_time": 7.619039297103882
    },
    {
      "epoch": 4.868507385253907e-06,
      "model_forward_time": 1.750554084777832,
      "step": 5105
    },
    {
      "epoch": 4.868507385253907e-06,
      "step": 5105,
      "training_step_time": 9.139781951904297
    },
    {
      "epoch": 4.869461059570312e-06,
      "model_forward_time": 1.7495007514953613,
      "step": 5106
    },
    {
      "epoch": 4.869461059570312e-06,
      "step": 5106,
      "training_step_time": 8.603666067123413
    },
    {
      "epoch": 4.870414733886719e-06,
      "model_forward_time": 1.7570750713348389,
      "step": 5107
    },
    {
      "epoch": 4.870414733886719e-06,
      "step": 5107,
      "training_step_time": 7.71081805229187
    },
    {
      "epoch": 4.871368408203125e-06,
      "model_forward_time": 1.7450120449066162,
      "step": 5108
    },
    {
      "epoch": 4.871368408203125e-06,
      "step": 5108,
      "training_step_time": 7.4351487159729
    },
    {
      "epoch": 4.872322082519531e-06,
      "model_forward_time": 1.7455365657806396,
      "step": 5109
    },
    {
      "epoch": 4.872322082519531e-06,
      "step": 5109,
      "training_step_time": 6.8212175369262695
    },
    {
      "epoch": 4.873275756835938e-06,
      "grad_norm": 0.3800341486930847,
      "learning_rate": 9.999967517827445e-06,
      "loss": 0.0755,
      "step": 5110
    },
    {
      "epoch": 4.873275756835938e-06,
      "model_forward_time": 1.7580499649047852,
      "step": 5110
    },
    {
      "dynamics_loss_avg": 0.05002708341926336,
      "epoch": 4.873275756835938e-06,
      "step": 5110
    },
    {
      "action_loss_avg": 0.007510500983335078,
      "epoch": 4.873275756835938e-06,
      "step": 5110
    },
    {
      "epoch": 4.873275756835938e-06,
      "step": 5110,
      "training_step_time": 6.841939449310303
    },
    {
      "epoch": 4.8742294311523435e-06,
      "model_forward_time": 1.7494828701019287,
      "step": 5111
    },
    {
      "epoch": 4.8742294311523435e-06,
      "step": 5111,
      "training_step_time": 6.824619293212891
    },
    {
      "epoch": 4.87518310546875e-06,
      "model_forward_time": 1.8017494678497314,
      "step": 5112
    },
    {
      "epoch": 4.87518310546875e-06,
      "step": 5112,
      "training_step_time": 7.061255931854248
    },
    {
      "epoch": 4.876136779785157e-06,
      "model_forward_time": 1.3897225856781006,
      "step": 5113
    },
    {
      "epoch": 4.876136779785157e-06,
      "step": 5113,
      "training_step_time": 7.652890682220459
    },
    {
      "epoch": 4.877090454101562e-06,
      "model_forward_time": 2.1359362602233887,
      "step": 5114
    },
    {
      "epoch": 4.877090454101562e-06,
      "step": 5114,
      "training_step_time": 8.326005697250366
    },
    {
      "epoch": 4.878044128417969e-06,
      "model_forward_time": 1.7397778034210205,
      "step": 5115
    },
    {
      "epoch": 4.878044128417969e-06,
      "step": 5115,
      "training_step_time": 7.661966800689697
    },
    {
      "epoch": 4.878997802734375e-06,
      "model_forward_time": 1.7476375102996826,
      "step": 5116
    },
    {
      "epoch": 4.878997802734375e-06,
      "step": 5116,
      "training_step_time": 7.678285598754883
    },
    {
      "epoch": 4.879951477050781e-06,
      "model_forward_time": 1.763404130935669,
      "step": 5117
    },
    {
      "epoch": 4.879951477050781e-06,
      "step": 5117,
      "training_step_time": 7.60288667678833
    },
    {
      "epoch": 4.880905151367188e-06,
      "model_forward_time": 1.7512884140014648,
      "step": 5118
    },
    {
      "epoch": 4.880905151367188e-06,
      "step": 5118,
      "training_step_time": 6.8167383670806885
    },
    {
      "epoch": 4.8818588256835935e-06,
      "model_forward_time": 1.7495567798614502,
      "step": 5119
    },
    {
      "epoch": 4.8818588256835935e-06,
      "step": 5119,
      "training_step_time": 6.712638854980469
    },
    {
      "epoch": 4.8828125e-06,
      "grad_norm": 0.4341500699520111,
      "learning_rate": 9.999961284407879e-06,
      "loss": 0.0812,
      "step": 5120
    },
    {
      "epoch": 4.8828125e-06,
      "model_forward_time": 1.7476146221160889,
      "step": 5120
    },
    {
      "dynamics_loss_avg": 0.05111146941781044,
      "epoch": 4.8828125e-06,
      "step": 5120
    },
    {
      "action_loss_avg": 0.006594840786419809,
      "epoch": 4.8828125e-06,
      "step": 5120
    },
    {
      "epoch": 4.8828125e-06,
      "step": 5120,
      "training_step_time": 6.747222185134888
    },
    {
      "epoch": 4.883766174316407e-06,
      "model_forward_time": 1.8166251182556152,
      "step": 5121
    },
    {
      "epoch": 4.883766174316407e-06,
      "step": 5121,
      "training_step_time": 7.016911506652832
    },
    {
      "epoch": 4.884719848632812e-06,
      "model_forward_time": 1.7491302490234375,
      "step": 5122
    },
    {
      "epoch": 4.884719848632812e-06,
      "step": 5122,
      "training_step_time": 7.53208327293396
    },
    {
      "epoch": 4.885673522949219e-06,
      "model_forward_time": 2.860696315765381,
      "step": 5123
    },
    {
      "epoch": 4.885673522949219e-06,
      "step": 5123,
      "training_step_time": 8.238879919052124
    },
    {
      "epoch": 4.886627197265625e-06,
      "model_forward_time": 1.7455377578735352,
      "step": 5124
    },
    {
      "epoch": 4.886627197265625e-06,
      "step": 5124,
      "training_step_time": 8.732589960098267
    },
    {
      "epoch": 4.887580871582031e-06,
      "model_forward_time": 1.7649366855621338,
      "step": 5125
    },
    {
      "epoch": 4.887580871582031e-06,
      "step": 5125,
      "training_step_time": 8.62938404083252
    },
    {
      "epoch": 4.888534545898438e-06,
      "model_forward_time": 1.744825839996338,
      "step": 5126
    },
    {
      "epoch": 4.888534545898438e-06,
      "step": 5126,
      "training_step_time": 8.32392930984497
    },
    {
      "epoch": 4.8894882202148435e-06,
      "model_forward_time": 1.7644782066345215,
      "step": 5127
    },
    {
      "epoch": 4.8894882202148435e-06,
      "step": 5127,
      "training_step_time": 6.745540142059326
    },
    {
      "epoch": 4.89044189453125e-06,
      "model_forward_time": 1.7403995990753174,
      "step": 5128
    },
    {
      "epoch": 4.89044189453125e-06,
      "step": 5128,
      "training_step_time": 6.840517044067383
    },
    {
      "epoch": 4.891395568847656e-06,
      "model_forward_time": 1.753967046737671,
      "step": 5129
    },
    {
      "epoch": 4.891395568847656e-06,
      "step": 5129,
      "training_step_time": 6.804656982421875
    },
    {
      "epoch": 4.892349243164062e-06,
      "grad_norm": 0.4095786511898041,
      "learning_rate": 9.999954504200067e-06,
      "loss": 0.083,
      "step": 5130
    },
    {
      "epoch": 4.892349243164062e-06,
      "model_forward_time": 1.8175938129425049,
      "step": 5130
    },
    {
      "dynamics_loss_avg": 0.062487238086760044,
      "epoch": 4.892349243164062e-06,
      "step": 5130
    },
    {
      "action_loss_avg": 0.008072247519157827,
      "epoch": 4.892349243164062e-06,
      "step": 5130
    },
    {
      "epoch": 4.892349243164062e-06,
      "step": 5130,
      "training_step_time": 6.782658815383911
    },
    {
      "epoch": 4.893302917480469e-06,
      "model_forward_time": 1.7539963722229004,
      "step": 5131
    },
    {
      "epoch": 4.893302917480469e-06,
      "step": 5131,
      "training_step_time": 8.370668649673462
    },
    {
      "epoch": 4.894256591796875e-06,
      "model_forward_time": 2.816452980041504,
      "step": 5132
    },
    {
      "epoch": 4.894256591796875e-06,
      "step": 5132,
      "training_step_time": 8.688597440719604
    },
    {
      "epoch": 4.895210266113281e-06,
      "model_forward_time": 1.747771978378296,
      "step": 5133
    },
    {
      "epoch": 4.895210266113281e-06,
      "step": 5133,
      "training_step_time": 8.893148183822632
    },
    {
      "epoch": 4.896163940429688e-06,
      "model_forward_time": 1.7545890808105469,
      "step": 5134
    },
    {
      "epoch": 4.896163940429688e-06,
      "step": 5134,
      "training_step_time": 8.957011461257935
    },
    {
      "epoch": 4.8971176147460935e-06,
      "model_forward_time": 1.7549102306365967,
      "step": 5135
    },
    {
      "epoch": 4.8971176147460935e-06,
      "step": 5135,
      "training_step_time": 7.653409957885742
    },
    {
      "epoch": 4.8980712890625e-06,
      "model_forward_time": 1.752105951309204,
      "step": 5136
    },
    {
      "epoch": 4.8980712890625e-06,
      "step": 5136,
      "training_step_time": 7.3450541496276855
    },
    {
      "epoch": 4.899024963378906e-06,
      "model_forward_time": 1.7592179775238037,
      "step": 5137
    },
    {
      "epoch": 4.899024963378906e-06,
      "step": 5137,
      "training_step_time": 6.769556283950806
    },
    {
      "epoch": 4.899978637695312e-06,
      "model_forward_time": 1.7648952007293701,
      "step": 5138
    },
    {
      "epoch": 4.899978637695312e-06,
      "step": 5138,
      "training_step_time": 6.759693622589111
    },
    {
      "epoch": 4.900932312011719e-06,
      "model_forward_time": 1.812962293624878,
      "step": 5139
    },
    {
      "epoch": 4.900932312011719e-06,
      "step": 5139,
      "training_step_time": 7.047273874282837
    },
    {
      "epoch": 4.901885986328125e-06,
      "grad_norm": 0.5315899848937988,
      "learning_rate": 9.999947177204745e-06,
      "loss": 0.0783,
      "step": 5140
    },
    {
      "epoch": 4.901885986328125e-06,
      "model_forward_time": 1.763354778289795,
      "step": 5140
    },
    {
      "dynamics_loss_avg": 0.07465997003018857,
      "epoch": 4.901885986328125e-06,
      "step": 5140
    },
    {
      "action_loss_avg": 0.009158556605689228,
      "epoch": 4.901885986328125e-06,
      "step": 5140
    },
    {
      "epoch": 4.901885986328125e-06,
      "step": 5140,
      "training_step_time": 7.46964955329895
    },
    {
      "epoch": 4.902839660644531e-06,
      "model_forward_time": 1.7699153423309326,
      "step": 5141
    },
    {
      "epoch": 4.902839660644531e-06,
      "step": 5141,
      "training_step_time": 8.425235033035278
    },
    {
      "epoch": 4.903793334960938e-06,
      "model_forward_time": 2.91652250289917,
      "step": 5142
    },
    {
      "epoch": 4.903793334960938e-06,
      "step": 5142,
      "training_step_time": 9.26486849784851
    },
    {
      "epoch": 4.9047470092773436e-06,
      "model_forward_time": 1.7667877674102783,
      "step": 5143
    },
    {
      "epoch": 4.9047470092773436e-06,
      "step": 5143,
      "training_step_time": 8.859632730484009
    },
    {
      "epoch": 4.90570068359375e-06,
      "model_forward_time": 1.7733924388885498,
      "step": 5144
    },
    {
      "epoch": 4.90570068359375e-06,
      "step": 5144,
      "training_step_time": 7.681538343429565
    },
    {
      "epoch": 4.906654357910156e-06,
      "model_forward_time": 1.765542984008789,
      "step": 5145
    },
    {
      "epoch": 4.906654357910156e-06,
      "step": 5145,
      "training_step_time": 7.297915935516357
    },
    {
      "epoch": 4.907608032226562e-06,
      "model_forward_time": 1.7591683864593506,
      "step": 5146
    },
    {
      "epoch": 4.907608032226562e-06,
      "step": 5146,
      "training_step_time": 6.823822498321533
    },
    {
      "epoch": 4.908561706542969e-06,
      "model_forward_time": 1.3948919773101807,
      "step": 5147
    },
    {
      "epoch": 4.908561706542969e-06,
      "step": 5147,
      "training_step_time": 6.749881267547607
    },
    {
      "epoch": 4.909515380859375e-06,
      "model_forward_time": 1.8014132976531982,
      "step": 5148
    },
    {
      "epoch": 4.909515380859375e-06,
      "step": 5148,
      "training_step_time": 7.060056686401367
    },
    {
      "epoch": 4.910469055175781e-06,
      "model_forward_time": 1.7546038627624512,
      "step": 5149
    },
    {
      "epoch": 4.910469055175781e-06,
      "step": 5149,
      "training_step_time": 8.342084407806396
    },
    {
      "epoch": 4.911422729492188e-06,
      "grad_norm": 0.30594056844711304,
      "learning_rate": 9.999939303422718e-06,
      "loss": 0.0753,
      "step": 5150
    },
    {
      "epoch": 4.911422729492188e-06,
      "model_forward_time": 1.756058692932129,
      "step": 5150
    },
    {
      "dynamics_loss_avg": 0.07493102736771107,
      "epoch": 4.911422729492188e-06,
      "step": 5150
    },
    {
      "action_loss_avg": 0.010128299565985798,
      "epoch": 4.911422729492188e-06,
      "step": 5150
    },
    {
      "epoch": 4.911422729492188e-06,
      "step": 5150,
      "training_step_time": 9.650647640228271
    },
    {
      "epoch": 4.9123764038085936e-06,
      "model_forward_time": 2.92812442779541,
      "step": 5151
    },
    {
      "epoch": 4.9123764038085936e-06,
      "step": 5151,
      "training_step_time": 10.39477276802063
    },
    {
      "epoch": 4.913330078125e-06,
      "model_forward_time": 1.7534868717193604,
      "step": 5152
    },
    {
      "epoch": 4.913330078125e-06,
      "step": 5152,
      "training_step_time": 10.573161602020264
    },
    {
      "epoch": 4.914283752441406e-06,
      "model_forward_time": 1.7599704265594482,
      "step": 5153
    },
    {
      "epoch": 4.914283752441406e-06,
      "step": 5153,
      "training_step_time": 7.60426139831543
    },
    {
      "epoch": 4.915237426757812e-06,
      "model_forward_time": 1.7539546489715576,
      "step": 5154
    },
    {
      "epoch": 4.915237426757812e-06,
      "step": 5154,
      "training_step_time": 7.585733652114868
    },
    {
      "epoch": 4.916191101074219e-06,
      "model_forward_time": 1.7634508609771729,
      "step": 5155
    },
    {
      "epoch": 4.916191101074219e-06,
      "step": 5155,
      "training_step_time": 6.761333227157593
    },
    {
      "epoch": 4.917144775390625e-06,
      "model_forward_time": 1.7506451606750488,
      "step": 5156
    },
    {
      "epoch": 4.917144775390625e-06,
      "step": 5156,
      "training_step_time": 6.649357080459595
    },
    {
      "epoch": 4.918098449707031e-06,
      "model_forward_time": 1.7870430946350098,
      "step": 5157
    },
    {
      "epoch": 4.918098449707031e-06,
      "step": 5157,
      "training_step_time": 7.021576404571533
    },
    {
      "epoch": 4.919052124023438e-06,
      "model_forward_time": 1.7612242698669434,
      "step": 5158
    },
    {
      "epoch": 4.919052124023438e-06,
      "step": 5158,
      "training_step_time": 7.447982311248779
    },
    {
      "epoch": 4.9200057983398436e-06,
      "model_forward_time": 1.748664379119873,
      "step": 5159
    },
    {
      "epoch": 4.9200057983398436e-06,
      "step": 5159,
      "training_step_time": 8.58339262008667
    },
    {
      "epoch": 4.92095947265625e-06,
      "grad_norm": 0.3017285168170929,
      "learning_rate": 9.999930882854847e-06,
      "loss": 0.0771,
      "step": 5160
    },
    {
      "epoch": 4.92095947265625e-06,
      "model_forward_time": 2.853823661804199,
      "step": 5160
    },
    {
      "dynamics_loss_avg": 0.0670842032879591,
      "epoch": 4.92095947265625e-06,
      "step": 5160
    },
    {
      "action_loss_avg": 0.009790358296595515,
      "epoch": 4.92095947265625e-06,
      "step": 5160
    },
    {
      "epoch": 4.92095947265625e-06,
      "step": 5160,
      "training_step_time": 8.26566743850708
    },
    {
      "epoch": 4.921913146972656e-06,
      "model_forward_time": 1.7602500915527344,
      "step": 5161
    },
    {
      "epoch": 4.921913146972656e-06,
      "step": 5161,
      "training_step_time": 7.679444313049316
    },
    {
      "epoch": 4.9228668212890624e-06,
      "model_forward_time": 1.7585394382476807,
      "step": 5162
    },
    {
      "epoch": 4.9228668212890624e-06,
      "step": 5162,
      "training_step_time": 7.60319972038269
    },
    {
      "epoch": 4.923820495605469e-06,
      "model_forward_time": 1.7674293518066406,
      "step": 5163
    },
    {
      "epoch": 4.923820495605469e-06,
      "step": 5163,
      "training_step_time": 8.289421319961548
    },
    {
      "epoch": 4.924774169921875e-06,
      "model_forward_time": 1.7482056617736816,
      "step": 5164
    },
    {
      "epoch": 4.924774169921875e-06,
      "step": 5164,
      "training_step_time": 6.790777683258057
    },
    {
      "epoch": 4.925727844238281e-06,
      "model_forward_time": 1.770354986190796,
      "step": 5165
    },
    {
      "epoch": 4.925727844238281e-06,
      "step": 5165,
      "training_step_time": 6.785383462905884
    },
    {
      "epoch": 4.926681518554688e-06,
      "model_forward_time": 1.7583222389221191,
      "step": 5166
    },
    {
      "epoch": 4.926681518554688e-06,
      "step": 5166,
      "training_step_time": 6.978712797164917
    },
    {
      "epoch": 4.9276351928710936e-06,
      "model_forward_time": 1.7506966590881348,
      "step": 5167
    },
    {
      "epoch": 4.9276351928710936e-06,
      "step": 5167,
      "training_step_time": 7.42654275894165
    },
    {
      "epoch": 4.9285888671875e-06,
      "model_forward_time": 1.3918511867523193,
      "step": 5168
    },
    {
      "epoch": 4.9285888671875e-06,
      "step": 5168,
      "training_step_time": 7.489566802978516
    },
    {
      "epoch": 4.929542541503906e-06,
      "model_forward_time": 2.812711715698242,
      "step": 5169
    },
    {
      "epoch": 4.929542541503906e-06,
      "step": 5169,
      "training_step_time": 7.825881242752075
    },
    {
      "epoch": 4.9304962158203124e-06,
      "grad_norm": 0.33811989426612854,
      "learning_rate": 9.99992191550205e-06,
      "loss": 0.0731,
      "step": 5170
    },
    {
      "epoch": 4.9304962158203124e-06,
      "model_forward_time": 1.7465543746948242,
      "step": 5170
    },
    {
      "dynamics_loss_avg": 0.05484360996633768,
      "epoch": 4.9304962158203124e-06,
      "step": 5170
    },
    {
      "action_loss_avg": 0.008179680816829205,
      "epoch": 4.9304962158203124e-06,
      "step": 5170
    },
    {
      "epoch": 4.9304962158203124e-06,
      "step": 5170,
      "training_step_time": 10.038588762283325
    },
    {
      "epoch": 4.931449890136719e-06,
      "model_forward_time": 1.763678789138794,
      "step": 5171
    },
    {
      "epoch": 4.931449890136719e-06,
      "step": 5171,
      "training_step_time": 7.5834715366363525
    },
    {
      "epoch": 4.932403564453125e-06,
      "model_forward_time": 1.7474589347839355,
      "step": 5172
    },
    {
      "epoch": 4.932403564453125e-06,
      "step": 5172,
      "training_step_time": 7.546569108963013
    },
    {
      "epoch": 4.933357238769531e-06,
      "model_forward_time": 1.751784086227417,
      "step": 5173
    },
    {
      "epoch": 4.933357238769531e-06,
      "step": 5173,
      "training_step_time": 7.343362092971802
    },
    {
      "epoch": 4.934310913085938e-06,
      "model_forward_time": 1.7532320022583008,
      "step": 5174
    },
    {
      "epoch": 4.934310913085938e-06,
      "step": 5174,
      "training_step_time": 6.742977142333984
    },
    {
      "epoch": 4.935264587402344e-06,
      "model_forward_time": 1.7642123699188232,
      "step": 5175
    },
    {
      "epoch": 4.935264587402344e-06,
      "step": 5175,
      "training_step_time": 6.752255439758301
    },
    {
      "epoch": 4.93621826171875e-06,
      "model_forward_time": 1.8156521320343018,
      "step": 5176
    },
    {
      "epoch": 4.93621826171875e-06,
      "step": 5176,
      "training_step_time": 7.503960132598877
    },
    {
      "epoch": 4.937171936035156e-06,
      "model_forward_time": 1.7562479972839355,
      "step": 5177
    },
    {
      "epoch": 4.937171936035156e-06,
      "step": 5177,
      "training_step_time": 7.370691776275635
    },
    {
      "epoch": 4.9381256103515624e-06,
      "model_forward_time": 1.7500617504119873,
      "step": 5178
    },
    {
      "epoch": 4.9381256103515624e-06,
      "step": 5178,
      "training_step_time": 9.109019994735718
    },
    {
      "epoch": 4.939079284667969e-06,
      "model_forward_time": 2.9215779304504395,
      "step": 5179
    },
    {
      "epoch": 4.939079284667969e-06,
      "step": 5179,
      "training_step_time": 9.320203065872192
    },
    {
      "epoch": 4.940032958984375e-06,
      "grad_norm": 0.4412403404712677,
      "learning_rate": 9.99991240136531e-06,
      "loss": 0.083,
      "step": 5180
    },
    {
      "epoch": 4.940032958984375e-06,
      "model_forward_time": 1.7467010021209717,
      "step": 5180
    },
    {
      "dynamics_loss_avg": 0.09815755523741246,
      "epoch": 4.940032958984375e-06,
      "step": 5180
    },
    {
      "action_loss_avg": 0.01896064621396363,
      "epoch": 4.940032958984375e-06,
      "step": 5180
    },
    {
      "epoch": 4.940032958984375e-06,
      "step": 5180,
      "training_step_time": 8.495611667633057
    },
    {
      "epoch": 4.940986633300781e-06,
      "model_forward_time": 1.7487878799438477,
      "step": 5181
    },
    {
      "epoch": 4.940986633300781e-06,
      "step": 5181,
      "training_step_time": 7.700170278549194
    },
    {
      "epoch": 4.941940307617188e-06,
      "model_forward_time": 1.7461652755737305,
      "step": 5182
    },
    {
      "epoch": 4.941940307617188e-06,
      "step": 5182,
      "training_step_time": 7.319726228713989
    },
    {
      "epoch": 4.942893981933594e-06,
      "model_forward_time": 1.3775529861450195,
      "step": 5183
    },
    {
      "epoch": 4.942893981933594e-06,
      "step": 5183,
      "training_step_time": 6.807501792907715
    },
    {
      "epoch": 4.94384765625e-06,
      "model_forward_time": 1.763246774673462,
      "step": 5184
    },
    {
      "epoch": 4.94384765625e-06,
      "step": 5184,
      "training_step_time": 6.794340372085571
    },
    {
      "epoch": 4.944801330566406e-06,
      "model_forward_time": 1.8137388229370117,
      "step": 5185
    },
    {
      "epoch": 4.944801330566406e-06,
      "step": 5185,
      "training_step_time": 7.483202219009399
    },
    {
      "epoch": 4.9457550048828124e-06,
      "model_forward_time": 1.7529051303863525,
      "step": 5186
    },
    {
      "epoch": 4.9457550048828124e-06,
      "step": 5186,
      "training_step_time": 7.853827476501465
    },
    {
      "epoch": 4.946708679199219e-06,
      "model_forward_time": 1.761049509048462,
      "step": 5187
    },
    {
      "epoch": 4.946708679199219e-06,
      "step": 5187,
      "training_step_time": 8.471107006072998
    },
    {
      "epoch": 4.947662353515625e-06,
      "model_forward_time": 2.8774843215942383,
      "step": 5188
    },
    {
      "epoch": 4.947662353515625e-06,
      "step": 5188,
      "training_step_time": 9.656391382217407
    },
    {
      "epoch": 4.948616027832031e-06,
      "model_forward_time": 1.771289587020874,
      "step": 5189
    },
    {
      "epoch": 4.948616027832031e-06,
      "step": 5189,
      "training_step_time": 7.580734968185425
    },
    {
      "epoch": 4.949569702148438e-06,
      "grad_norm": 0.37828734517097473,
      "learning_rate": 9.999902340445668e-06,
      "loss": 0.0851,
      "step": 5190
    },
    {
      "epoch": 4.949569702148438e-06,
      "model_forward_time": 1.761718988418579,
      "step": 5190
    },
    {
      "dynamics_loss_avg": 0.07389786280691624,
      "epoch": 4.949569702148438e-06,
      "step": 5190
    },
    {
      "action_loss_avg": 0.010606615571305156,
      "epoch": 4.949569702148438e-06,
      "step": 5190
    },
    {
      "epoch": 4.949569702148438e-06,
      "step": 5190,
      "training_step_time": 7.697062730789185
    },
    {
      "epoch": 4.950523376464844e-06,
      "model_forward_time": 1.7391231060028076,
      "step": 5191
    },
    {
      "epoch": 4.950523376464844e-06,
      "step": 5191,
      "training_step_time": 7.580082893371582
    },
    {
      "epoch": 4.95147705078125e-06,
      "model_forward_time": 1.7630689144134521,
      "step": 5192
    },
    {
      "epoch": 4.95147705078125e-06,
      "step": 5192,
      "training_step_time": 7.298966407775879
    },
    {
      "epoch": 4.952430725097656e-06,
      "model_forward_time": 1.0516462326049805,
      "step": 5193
    },
    {
      "epoch": 4.952430725097656e-06,
      "step": 5193,
      "training_step_time": 6.713517904281616
    },
    {
      "epoch": 4.9533843994140625e-06,
      "model_forward_time": 1.8232781887054443,
      "step": 5194
    },
    {
      "epoch": 4.9533843994140625e-06,
      "step": 5194,
      "training_step_time": 7.518890380859375
    },
    {
      "epoch": 4.954338073730469e-06,
      "model_forward_time": 1.7460947036743164,
      "step": 5195
    },
    {
      "epoch": 4.954338073730469e-06,
      "step": 5195,
      "training_step_time": 7.277558088302612
    },
    {
      "epoch": 4.955291748046875e-06,
      "model_forward_time": 1.7672102451324463,
      "step": 5196
    },
    {
      "epoch": 4.955291748046875e-06,
      "step": 5196,
      "training_step_time": 7.651463031768799
    },
    {
      "epoch": 4.956245422363281e-06,
      "model_forward_time": 1.4101274013519287,
      "step": 5197
    },
    {
      "epoch": 4.956245422363281e-06,
      "step": 5197,
      "training_step_time": 9.206470251083374
    },
    {
      "epoch": 4.957199096679688e-06,
      "model_forward_time": 2.933023452758789,
      "step": 5198
    },
    {
      "epoch": 4.957199096679688e-06,
      "step": 5198,
      "training_step_time": 9.214353799819946
    },
    {
      "epoch": 4.958152770996094e-06,
      "model_forward_time": 1.7649943828582764,
      "step": 5199
    },
    {
      "epoch": 4.958152770996094e-06,
      "step": 5199,
      "training_step_time": 7.648521184921265
    },
    {
      "epoch": 4.9591064453125e-06,
      "grad_norm": 0.48769819736480713,
      "learning_rate": 9.999891732744224e-06,
      "loss": 0.0824,
      "step": 5200
    },
    {
      "epoch": 4.9591064453125e-06,
      "model_forward_time": 1.7580440044403076,
      "step": 5200
    },
    {
      "dynamics_loss_avg": 0.04827052187174559,
      "epoch": 4.9591064453125e-06,
      "step": 5200
    },
    {
      "action_loss_avg": 0.005141600104980171,
      "epoch": 4.9591064453125e-06,
      "step": 5200
    },
    {
      "epoch": 4.9591064453125e-06,
      "step": 5200,
      "training_step_time": 7.650461435317993
    },
    {
      "epoch": 4.960060119628906e-06,
      "model_forward_time": 1.7713112831115723,
      "step": 5201
    },
    {
      "epoch": 4.960060119628906e-06,
      "step": 5201,
      "training_step_time": 7.2206292152404785
    },
    {
      "epoch": 4.9610137939453125e-06,
      "model_forward_time": 1.736569881439209,
      "step": 5202
    },
    {
      "epoch": 4.9610137939453125e-06,
      "step": 5202,
      "training_step_time": 6.690858364105225
    },
    {
      "epoch": 4.961967468261719e-06,
      "model_forward_time": 1.825073480606079,
      "step": 5203
    },
    {
      "epoch": 4.961967468261719e-06,
      "step": 5203,
      "training_step_time": 7.439932584762573
    },
    {
      "epoch": 4.962921142578125e-06,
      "model_forward_time": 1.752939224243164,
      "step": 5204
    },
    {
      "epoch": 4.962921142578125e-06,
      "step": 5204,
      "training_step_time": 7.216504096984863
    },
    {
      "epoch": 4.963874816894531e-06,
      "model_forward_time": 1.758408546447754,
      "step": 5205
    },
    {
      "epoch": 4.963874816894531e-06,
      "step": 5205,
      "training_step_time": 7.4195616245269775
    },
    {
      "epoch": 4.964828491210938e-06,
      "model_forward_time": 1.7447888851165771,
      "step": 5206
    },
    {
      "epoch": 4.964828491210938e-06,
      "step": 5206,
      "training_step_time": 7.734961986541748
    },
    {
      "epoch": 4.965782165527344e-06,
      "model_forward_time": 2.911377191543579,
      "step": 5207
    },
    {
      "epoch": 4.965782165527344e-06,
      "step": 5207,
      "training_step_time": 8.605382919311523
    },
    {
      "epoch": 4.96673583984375e-06,
      "model_forward_time": 1.397090196609497,
      "step": 5208
    },
    {
      "epoch": 4.96673583984375e-06,
      "step": 5208,
      "training_step_time": 7.591027498245239
    },
    {
      "epoch": 4.967689514160156e-06,
      "model_forward_time": 1.7533817291259766,
      "step": 5209
    },
    {
      "epoch": 4.967689514160156e-06,
      "step": 5209,
      "training_step_time": 7.539328575134277
    },
    {
      "epoch": 4.9686431884765625e-06,
      "grad_norm": 0.2896786630153656,
      "learning_rate": 9.999880578262135e-06,
      "loss": 0.0752,
      "step": 5210
    },
    {
      "epoch": 4.9686431884765625e-06,
      "model_forward_time": 1.751058578491211,
      "step": 5210
    },
    {
      "dynamics_loss_avg": 0.08664387837052345,
      "epoch": 4.9686431884765625e-06,
      "step": 5210
    },
    {
      "action_loss_avg": 0.00868439533514902,
      "epoch": 4.9686431884765625e-06,
      "step": 5210
    },
    {
      "epoch": 4.9686431884765625e-06,
      "step": 5210,
      "training_step_time": 7.329350233078003
    },
    {
      "epoch": 4.969596862792969e-06,
      "model_forward_time": 1.7530529499053955,
      "step": 5211
    },
    {
      "epoch": 4.969596862792969e-06,
      "step": 5211,
      "training_step_time": 6.722317457199097
    },
    {
      "epoch": 4.970550537109375e-06,
      "model_forward_time": 1.788856029510498,
      "step": 5212
    },
    {
      "epoch": 4.970550537109375e-06,
      "step": 5212,
      "training_step_time": 7.4296605587005615
    },
    {
      "epoch": 4.971504211425781e-06,
      "model_forward_time": 1.7473692893981934,
      "step": 5213
    },
    {
      "epoch": 4.971504211425781e-06,
      "step": 5213,
      "training_step_time": 7.2214062213897705
    },
    {
      "epoch": 4.972457885742188e-06,
      "model_forward_time": 1.7549021244049072,
      "step": 5214
    },
    {
      "epoch": 4.972457885742188e-06,
      "step": 5214,
      "training_step_time": 7.538832426071167
    },
    {
      "epoch": 4.973411560058594e-06,
      "model_forward_time": 1.7520370483398438,
      "step": 5215
    },
    {
      "epoch": 4.973411560058594e-06,
      "step": 5215,
      "training_step_time": 7.732821702957153
    },
    {
      "epoch": 4.974365234375e-06,
      "model_forward_time": 2.853332996368408,
      "step": 5216
    },
    {
      "epoch": 4.974365234375e-06,
      "step": 5216,
      "training_step_time": 7.654268503189087
    },
    {
      "epoch": 4.975318908691406e-06,
      "model_forward_time": 1.7519423961639404,
      "step": 5217
    },
    {
      "epoch": 4.975318908691406e-06,
      "step": 5217,
      "training_step_time": 7.605981349945068
    },
    {
      "epoch": 4.9762725830078125e-06,
      "model_forward_time": 1.7488927841186523,
      "step": 5218
    },
    {
      "epoch": 4.9762725830078125e-06,
      "step": 5218,
      "training_step_time": 7.66060733795166
    },
    {
      "epoch": 4.977226257324219e-06,
      "model_forward_time": 1.774657964706421,
      "step": 5219
    },
    {
      "epoch": 4.977226257324219e-06,
      "step": 5219,
      "training_step_time": 7.357346534729004
    },
    {
      "epoch": 4.978179931640625e-06,
      "grad_norm": 0.3304315209388733,
      "learning_rate": 9.999868877000625e-06,
      "loss": 0.0737,
      "step": 5220
    },
    {
      "epoch": 4.978179931640625e-06,
      "model_forward_time": 1.7487311363220215,
      "step": 5220
    },
    {
      "dynamics_loss_avg": 0.054933879896998404,
      "epoch": 4.978179931640625e-06,
      "step": 5220
    },
    {
      "action_loss_avg": 0.005564399412833154,
      "epoch": 4.978179931640625e-06,
      "step": 5220
    },
    {
      "epoch": 4.978179931640625e-06,
      "step": 5220,
      "training_step_time": 6.735608339309692
    },
    {
      "epoch": 4.979133605957031e-06,
      "model_forward_time": 1.7517786026000977,
      "step": 5221
    },
    {
      "epoch": 4.979133605957031e-06,
      "step": 5221,
      "training_step_time": 7.385655641555786
    },
    {
      "epoch": 4.980087280273438e-06,
      "model_forward_time": 1.8139703273773193,
      "step": 5222
    },
    {
      "epoch": 4.980087280273438e-06,
      "step": 5222,
      "training_step_time": 7.261947870254517
    },
    {
      "epoch": 4.981040954589844e-06,
      "model_forward_time": 1.767413854598999,
      "step": 5223
    },
    {
      "epoch": 4.981040954589844e-06,
      "step": 5223,
      "training_step_time": 7.518876552581787
    },
    {
      "epoch": 4.98199462890625e-06,
      "model_forward_time": 1.7575857639312744,
      "step": 5224
    },
    {
      "epoch": 4.98199462890625e-06,
      "step": 5224,
      "training_step_time": 9.004960298538208
    },
    {
      "epoch": 4.982948303222656e-06,
      "model_forward_time": 2.8698339462280273,
      "step": 5225
    },
    {
      "epoch": 4.982948303222656e-06,
      "step": 5225,
      "training_step_time": 8.390618801116943
    },
    {
      "epoch": 4.9839019775390625e-06,
      "model_forward_time": 1.7493741512298584,
      "step": 5226
    },
    {
      "epoch": 4.9839019775390625e-06,
      "step": 5226,
      "training_step_time": 8.251153707504272
    },
    {
      "epoch": 4.984855651855469e-06,
      "model_forward_time": 1.7567822933197021,
      "step": 5227
    },
    {
      "epoch": 4.984855651855469e-06,
      "step": 5227,
      "training_step_time": 7.647920608520508
    },
    {
      "epoch": 4.985809326171875e-06,
      "model_forward_time": 1.7763638496398926,
      "step": 5228
    },
    {
      "epoch": 4.985809326171875e-06,
      "step": 5228,
      "training_step_time": 7.333686590194702
    },
    {
      "epoch": 4.986763000488281e-06,
      "model_forward_time": 1.7498292922973633,
      "step": 5229
    },
    {
      "epoch": 4.986763000488281e-06,
      "step": 5229,
      "training_step_time": 7.243778228759766
    },
    {
      "epoch": 4.987716674804688e-06,
      "grad_norm": 0.33990442752838135,
      "learning_rate": 9.999856628960971e-06,
      "loss": 0.0816,
      "step": 5230
    },
    {
      "epoch": 4.987716674804688e-06,
      "model_forward_time": 1.7492454051971436,
      "step": 5230
    },
    {
      "dynamics_loss_avg": 0.07446301635354757,
      "epoch": 4.987716674804688e-06,
      "step": 5230
    },
    {
      "action_loss_avg": 0.008700532466173172,
      "epoch": 4.987716674804688e-06,
      "step": 5230
    },
    {
      "epoch": 4.987716674804688e-06,
      "step": 5230,
      "training_step_time": 6.7142744064331055
    },
    {
      "epoch": 4.988670349121094e-06,
      "model_forward_time": 1.8088433742523193,
      "step": 5231
    },
    {
      "epoch": 4.988670349121094e-06,
      "step": 5231,
      "training_step_time": 7.40579080581665
    },
    {
      "epoch": 4.9896240234375e-06,
      "model_forward_time": 1.7646675109863281,
      "step": 5232
    },
    {
      "epoch": 4.9896240234375e-06,
      "step": 5232,
      "training_step_time": 7.456018924713135
    },
    {
      "epoch": 4.990577697753906e-06,
      "model_forward_time": 1.748748540878296,
      "step": 5233
    },
    {
      "epoch": 4.990577697753906e-06,
      "step": 5233,
      "training_step_time": 8.474804639816284
    },
    {
      "epoch": 4.9915313720703125e-06,
      "model_forward_time": 1.7546255588531494,
      "step": 5234
    },
    {
      "epoch": 4.9915313720703125e-06,
      "step": 5234,
      "training_step_time": 7.775116205215454
    },
    {
      "epoch": 4.992485046386719e-06,
      "model_forward_time": 2.9505834579467773,
      "step": 5235
    },
    {
      "epoch": 4.992485046386719e-06,
      "step": 5235,
      "training_step_time": 9.273295164108276
    },
    {
      "epoch": 4.993438720703125e-06,
      "model_forward_time": 1.752018928527832,
      "step": 5236
    },
    {
      "epoch": 4.993438720703125e-06,
      "step": 5236,
      "training_step_time": 8.112590312957764
    },
    {
      "epoch": 4.994392395019531e-06,
      "model_forward_time": 1.4015684127807617,
      "step": 5237
    },
    {
      "epoch": 4.994392395019531e-06,
      "step": 5237,
      "training_step_time": 7.570596218109131
    },
    {
      "epoch": 4.995346069335938e-06,
      "model_forward_time": 1.7496252059936523,
      "step": 5238
    },
    {
      "epoch": 4.995346069335938e-06,
      "step": 5238,
      "training_step_time": 7.229278802871704
    },
    {
      "epoch": 4.996299743652344e-06,
      "model_forward_time": 1.7641146183013916,
      "step": 5239
    },
    {
      "epoch": 4.996299743652344e-06,
      "step": 5239,
      "training_step_time": 6.70502495765686
    },
    {
      "epoch": 4.99725341796875e-06,
      "grad_norm": 0.3562571108341217,
      "learning_rate": 9.999843834144514e-06,
      "loss": 0.0822,
      "step": 5240
    },
    {
      "epoch": 4.99725341796875e-06,
      "model_forward_time": 1.8217487335205078,
      "step": 5240
    },
    {
      "dynamics_loss_avg": 0.06877470593899489,
      "epoch": 4.99725341796875e-06,
      "step": 5240
    },
    {
      "action_loss_avg": 0.0076241695787757635,
      "epoch": 4.99725341796875e-06,
      "step": 5240
    },
    {
      "epoch": 4.99725341796875e-06,
      "step": 5240,
      "training_step_time": 7.455960035324097
    },
    {
      "epoch": 4.998207092285156e-06,
      "model_forward_time": 1.758962631225586,
      "step": 5241
    },
    {
      "epoch": 4.998207092285156e-06,
      "step": 5241,
      "training_step_time": 7.422964811325073
    },
    {
      "epoch": 4.9991607666015625e-06,
      "model_forward_time": 1.7549829483032227,
      "step": 5242
    },
    {
      "epoch": 4.9991607666015625e-06,
      "step": 5242,
      "training_step_time": 7.636859178543091
    },
    {
      "epoch": 5.000114440917969e-06,
      "model_forward_time": 1.0634026527404785,
      "step": 5243
    },
    {
      "epoch": 5.000114440917969e-06,
      "step": 5243,
      "training_step_time": 9.544029474258423
    },
    {
      "epoch": 5.001068115234375e-06,
      "model_forward_time": 2.886922836303711,
      "step": 5244
    },
    {
      "epoch": 5.001068115234375e-06,
      "step": 5244,
      "training_step_time": 9.600095510482788
    },
    {
      "epoch": 5.002021789550781e-06,
      "model_forward_time": 1.7574069499969482,
      "step": 5245
    },
    {
      "epoch": 5.002021789550781e-06,
      "step": 5245,
      "training_step_time": 7.520296812057495
    },
    {
      "epoch": 5.002975463867188e-06,
      "model_forward_time": 1.7574234008789062,
      "step": 5246
    },
    {
      "epoch": 5.002975463867188e-06,
      "step": 5246,
      "training_step_time": 7.563410997390747
    },
    {
      "epoch": 5.003929138183594e-06,
      "model_forward_time": 1.7502634525299072,
      "step": 5247
    },
    {
      "epoch": 5.003929138183594e-06,
      "step": 5247,
      "training_step_time": 8.063699007034302
    },
    {
      "epoch": 5.0048828125e-06,
      "model_forward_time": 1.754777431488037,
      "step": 5248
    },
    {
      "epoch": 5.0048828125e-06,
      "step": 5248,
      "training_step_time": 7.236002206802368
    },
    {
      "epoch": 5.005836486816406e-06,
      "model_forward_time": 1.8140323162078857,
      "step": 5249
    },
    {
      "epoch": 5.005836486816406e-06,
      "step": 5249,
      "training_step_time": 7.432769298553467
    },
    {
      "epoch": 5.0067901611328125e-06,
      "grad_norm": 0.38459333777427673,
      "learning_rate": 9.999830492552651e-06,
      "loss": 0.0788,
      "step": 5250
    },
    {
      "epoch": 5.0067901611328125e-06,
      "model_forward_time": 1.7485711574554443,
      "step": 5250
    },
    {
      "dynamics_loss_avg": 0.06539219468832017,
      "epoch": 5.0067901611328125e-06,
      "step": 5250
    },
    {
      "action_loss_avg": 0.012110996758565307,
      "epoch": 5.0067901611328125e-06,
      "step": 5250
    },
    {
      "epoch": 5.0067901611328125e-06,
      "step": 5250,
      "training_step_time": 7.268068790435791
    },
    {
      "epoch": 5.007743835449219e-06,
      "model_forward_time": 1.7509961128234863,
      "step": 5251
    },
    {
      "epoch": 5.007743835449219e-06,
      "step": 5251,
      "training_step_time": 7.606315612792969
    },
    {
      "epoch": 5.008697509765625e-06,
      "model_forward_time": 1.7620749473571777,
      "step": 5252
    },
    {
      "epoch": 5.008697509765625e-06,
      "step": 5252,
      "training_step_time": 9.353600263595581
    },
    {
      "epoch": 5.009651184082031e-06,
      "model_forward_time": 2.839913845062256,
      "step": 5253
    },
    {
      "epoch": 5.009651184082031e-06,
      "step": 5253,
      "training_step_time": 11.317031383514404
    },
    {
      "epoch": 5.010604858398437e-06,
      "model_forward_time": 1.7543280124664307,
      "step": 5254
    },
    {
      "epoch": 5.010604858398437e-06,
      "step": 5254,
      "training_step_time": 7.528513431549072
    },
    {
      "epoch": 5.011558532714844e-06,
      "model_forward_time": 1.753840684890747,
      "step": 5255
    },
    {
      "epoch": 5.011558532714844e-06,
      "step": 5255,
      "training_step_time": 7.640708684921265
    },
    {
      "epoch": 5.01251220703125e-06,
      "model_forward_time": 1.757333517074585,
      "step": 5256
    },
    {
      "epoch": 5.01251220703125e-06,
      "step": 5256,
      "training_step_time": 7.4099202156066895
    },
    {
      "epoch": 5.013465881347656e-06,
      "model_forward_time": 1.7571847438812256,
      "step": 5257
    },
    {
      "epoch": 5.013465881347656e-06,
      "step": 5257,
      "training_step_time": 7.229590177536011
    },
    {
      "epoch": 5.0144195556640625e-06,
      "model_forward_time": 1.778998613357544,
      "step": 5258
    },
    {
      "epoch": 5.0144195556640625e-06,
      "step": 5258,
      "training_step_time": 7.424492835998535
    },
    {
      "epoch": 5.015373229980469e-06,
      "model_forward_time": 1.7551805973052979,
      "step": 5259
    },
    {
      "epoch": 5.015373229980469e-06,
      "step": 5259,
      "training_step_time": 7.263925075531006
    },
    {
      "epoch": 5.016326904296875e-06,
      "grad_norm": 0.29885268211364746,
      "learning_rate": 9.999816604186844e-06,
      "loss": 0.0813,
      "step": 5260
    },
    {
      "epoch": 5.016326904296875e-06,
      "model_forward_time": 1.7591636180877686,
      "step": 5260
    },
    {
      "dynamics_loss_avg": 0.07862533256411552,
      "epoch": 5.016326904296875e-06,
      "step": 5260
    },
    {
      "action_loss_avg": 0.009088769485242666,
      "epoch": 5.016326904296875e-06,
      "step": 5260
    },
    {
      "epoch": 5.016326904296875e-06,
      "step": 5260,
      "training_step_time": 8.331915378570557
    },
    {
      "epoch": 5.017280578613281e-06,
      "model_forward_time": 1.7604258060455322,
      "step": 5261
    },
    {
      "epoch": 5.017280578613281e-06,
      "step": 5261,
      "training_step_time": 7.7703330516815186
    },
    {
      "epoch": 5.018234252929687e-06,
      "model_forward_time": 1.7609877586364746,
      "step": 5262
    },
    {
      "epoch": 5.018234252929687e-06,
      "step": 5262,
      "training_step_time": 9.12020492553711
    },
    {
      "epoch": 5.019187927246094e-06,
      "model_forward_time": 2.916721820831299,
      "step": 5263
    },
    {
      "epoch": 5.019187927246094e-06,
      "step": 5263,
      "training_step_time": 7.720221519470215
    },
    {
      "epoch": 5.0201416015625e-06,
      "model_forward_time": 1.7579636573791504,
      "step": 5264
    },
    {
      "epoch": 5.0201416015625e-06,
      "step": 5264,
      "training_step_time": 7.610220670700073
    },
    {
      "epoch": 5.021095275878906e-06,
      "model_forward_time": 1.7556371688842773,
      "step": 5265
    },
    {
      "epoch": 5.021095275878906e-06,
      "step": 5265,
      "training_step_time": 7.352541923522949
    },
    {
      "epoch": 5.0220489501953125e-06,
      "model_forward_time": 1.762183427810669,
      "step": 5266
    },
    {
      "epoch": 5.0220489501953125e-06,
      "step": 5266,
      "training_step_time": 7.297349691390991
    },
    {
      "epoch": 5.023002624511719e-06,
      "model_forward_time": 1.7665510177612305,
      "step": 5267
    },
    {
      "epoch": 5.023002624511719e-06,
      "step": 5267,
      "training_step_time": 7.284810781478882
    },
    {
      "epoch": 5.023956298828125e-06,
      "model_forward_time": 1.754258394241333,
      "step": 5268
    },
    {
      "epoch": 5.023956298828125e-06,
      "step": 5268,
      "training_step_time": 7.476186275482178
    },
    {
      "epoch": 5.024909973144531e-06,
      "model_forward_time": 1.756298303604126,
      "step": 5269
    },
    {
      "epoch": 5.024909973144531e-06,
      "step": 5269,
      "training_step_time": 7.578067302703857
    },
    {
      "epoch": 5.025863647460937e-06,
      "grad_norm": 0.5153310894966125,
      "learning_rate": 9.999802169048609e-06,
      "loss": 0.0734,
      "step": 5270
    },
    {
      "epoch": 5.025863647460937e-06,
      "model_forward_time": 1.7628817558288574,
      "step": 5270
    },
    {
      "dynamics_loss_avg": 0.07928602918982505,
      "epoch": 5.025863647460937e-06,
      "step": 5270
    },
    {
      "action_loss_avg": 0.015558545105159282,
      "epoch": 5.025863647460937e-06,
      "step": 5270
    },
    {
      "epoch": 5.025863647460937e-06,
      "step": 5270,
      "training_step_time": 8.311810970306396
    },
    {
      "epoch": 5.026817321777344e-06,
      "model_forward_time": 1.7667803764343262,
      "step": 5271
    },
    {
      "epoch": 5.026817321777344e-06,
      "step": 5271,
      "training_step_time": 8.629900693893433
    },
    {
      "epoch": 5.02777099609375e-06,
      "model_forward_time": 2.952359676361084,
      "step": 5272
    },
    {
      "epoch": 5.02777099609375e-06,
      "step": 5272,
      "training_step_time": 7.772087574005127
    },
    {
      "epoch": 5.028724670410156e-06,
      "model_forward_time": 1.755056619644165,
      "step": 5273
    },
    {
      "epoch": 5.028724670410156e-06,
      "step": 5273,
      "training_step_time": 7.60073447227478
    },
    {
      "epoch": 5.0296783447265625e-06,
      "model_forward_time": 1.7678842544555664,
      "step": 5274
    },
    {
      "epoch": 5.0296783447265625e-06,
      "step": 5274,
      "training_step_time": 8.0707426071167
    },
    {
      "epoch": 5.030632019042969e-06,
      "model_forward_time": 1.7590439319610596,
      "step": 5275
    },
    {
      "epoch": 5.030632019042969e-06,
      "step": 5275,
      "training_step_time": 7.2164623737335205
    },
    {
      "epoch": 5.031585693359375e-06,
      "model_forward_time": 1.748002052307129,
      "step": 5276
    },
    {
      "epoch": 5.031585693359375e-06,
      "step": 5276,
      "training_step_time": 7.321709156036377
    },
    {
      "epoch": 5.032539367675781e-06,
      "model_forward_time": 1.831878423690796,
      "step": 5277
    },
    {
      "epoch": 5.032539367675781e-06,
      "step": 5277,
      "training_step_time": 7.4742491245269775
    },
    {
      "epoch": 5.033493041992187e-06,
      "model_forward_time": 1.7618865966796875,
      "step": 5278
    },
    {
      "epoch": 5.033493041992187e-06,
      "step": 5278,
      "training_step_time": 7.313713312149048
    },
    {
      "epoch": 5.034446716308594e-06,
      "model_forward_time": 1.746079921722412,
      "step": 5279
    },
    {
      "epoch": 5.034446716308594e-06,
      "step": 5279,
      "training_step_time": 8.718289613723755
    },
    {
      "epoch": 5.035400390625e-06,
      "grad_norm": 0.2226080298423767,
      "learning_rate": 9.999787187139527e-06,
      "loss": 0.0743,
      "step": 5280
    },
    {
      "epoch": 5.035400390625e-06,
      "model_forward_time": 1.7624142169952393,
      "step": 5280
    },
    {
      "dynamics_loss_avg": 0.07901212237775326,
      "epoch": 5.035400390625e-06,
      "step": 5280
    },
    {
      "action_loss_avg": 0.010798739874735475,
      "epoch": 5.035400390625e-06,
      "step": 5280
    },
    {
      "epoch": 5.035400390625e-06,
      "step": 5280,
      "training_step_time": 8.66927719116211
    },
    {
      "epoch": 5.036354064941406e-06,
      "model_forward_time": 2.544976234436035,
      "step": 5281
    },
    {
      "epoch": 5.036354064941406e-06,
      "step": 5281,
      "training_step_time": 7.686333894729614
    },
    {
      "epoch": 5.0373077392578125e-06,
      "model_forward_time": 1.7621326446533203,
      "step": 5282
    },
    {
      "epoch": 5.0373077392578125e-06,
      "step": 5282,
      "training_step_time": 7.5797119140625
    },
    {
      "epoch": 5.038261413574219e-06,
      "model_forward_time": 1.7586376667022705,
      "step": 5283
    },
    {
      "epoch": 5.038261413574219e-06,
      "step": 5283,
      "training_step_time": 7.64078164100647
    },
    {
      "epoch": 5.039215087890625e-06,
      "model_forward_time": 1.7564761638641357,
      "step": 5284
    },
    {
      "epoch": 5.039215087890625e-06,
      "step": 5284,
      "training_step_time": 7.390995025634766
    },
    {
      "epoch": 5.040168762207031e-06,
      "model_forward_time": 1.749190092086792,
      "step": 5285
    },
    {
      "epoch": 5.040168762207031e-06,
      "step": 5285,
      "training_step_time": 7.287273168563843
    },
    {
      "epoch": 5.041122436523437e-06,
      "model_forward_time": 1.8416123390197754,
      "step": 5286
    },
    {
      "epoch": 5.041122436523437e-06,
      "step": 5286,
      "training_step_time": 7.464062690734863
    },
    {
      "epoch": 5.042076110839844e-06,
      "model_forward_time": 1.779707670211792,
      "step": 5287
    },
    {
      "epoch": 5.042076110839844e-06,
      "step": 5287,
      "training_step_time": 7.325199365615845
    },
    {
      "epoch": 5.04302978515625e-06,
      "model_forward_time": 1.7540926933288574,
      "step": 5288
    },
    {
      "epoch": 5.04302978515625e-06,
      "step": 5288,
      "training_step_time": 8.640976190567017
    },
    {
      "epoch": 5.043983459472656e-06,
      "model_forward_time": 1.7720212936401367,
      "step": 5289
    },
    {
      "epoch": 5.043983459472656e-06,
      "step": 5289,
      "training_step_time": 8.78288221359253
    },
    {
      "epoch": 5.0449371337890625e-06,
      "grad_norm": 0.2009151130914688,
      "learning_rate": 9.999771658461236e-06,
      "loss": 0.0731,
      "step": 5290
    },
    {
      "epoch": 5.0449371337890625e-06,
      "model_forward_time": 2.864776849746704,
      "step": 5290
    },
    {
      "dynamics_loss_avg": 0.08411899581551552,
      "epoch": 5.0449371337890625e-06,
      "step": 5290
    },
    {
      "action_loss_avg": 0.010628929873928428,
      "epoch": 5.0449371337890625e-06,
      "step": 5290
    },
    {
      "epoch": 5.0449371337890625e-06,
      "step": 5290,
      "training_step_time": 7.639741659164429
    },
    {
      "epoch": 5.045890808105469e-06,
      "model_forward_time": 1.7548727989196777,
      "step": 5291
    },
    {
      "epoch": 5.045890808105469e-06,
      "step": 5291,
      "training_step_time": 7.696511268615723
    },
    {
      "epoch": 5.046844482421875e-06,
      "model_forward_time": 1.7608873844146729,
      "step": 5292
    },
    {
      "epoch": 5.046844482421875e-06,
      "step": 5292,
      "training_step_time": 7.698640584945679
    },
    {
      "epoch": 5.047798156738281e-06,
      "model_forward_time": 1.7548413276672363,
      "step": 5293
    },
    {
      "epoch": 5.047798156738281e-06,
      "step": 5293,
      "training_step_time": 7.369702100753784
    },
    {
      "epoch": 5.048751831054687e-06,
      "model_forward_time": 1.7478835582733154,
      "step": 5294
    },
    {
      "epoch": 5.048751831054687e-06,
      "step": 5294,
      "training_step_time": 7.339747905731201
    },
    {
      "epoch": 5.049705505371094e-06,
      "model_forward_time": 1.8275468349456787,
      "step": 5295
    },
    {
      "epoch": 5.049705505371094e-06,
      "step": 5295,
      "training_step_time": 7.5127928256988525
    },
    {
      "epoch": 5.0506591796875e-06,
      "model_forward_time": 1.758211612701416,
      "step": 5296
    },
    {
      "epoch": 5.0506591796875e-06,
      "step": 5296,
      "training_step_time": 7.348523139953613
    },
    {
      "epoch": 5.051612854003906e-06,
      "model_forward_time": 1.7586774826049805,
      "step": 5297
    },
    {
      "epoch": 5.051612854003906e-06,
      "step": 5297,
      "training_step_time": 8.391606092453003
    },
    {
      "epoch": 5.0525665283203125e-06,
      "model_forward_time": 1.763542890548706,
      "step": 5298
    },
    {
      "epoch": 5.0525665283203125e-06,
      "step": 5298,
      "training_step_time": 9.474515438079834
    },
    {
      "epoch": 5.053520202636719e-06,
      "model_forward_time": 1.7502186298370361,
      "step": 5299
    },
    {
      "epoch": 5.053520202636719e-06,
      "step": 5299,
      "training_step_time": 9.208105325698853
    },
    {
      "epoch": 5.054473876953125e-06,
      "grad_norm": 0.321389764547348,
      "learning_rate": 9.999755583015431e-06,
      "loss": 0.0713,
      "step": 5300
    },
    {
      "epoch": 5.054473876953125e-06,
      "model_forward_time": 1.748270034790039,
      "step": 5300
    },
    {
      "dynamics_loss_avg": 0.056965972390025855,
      "epoch": 5.054473876953125e-06,
      "step": 5300
    },
    {
      "action_loss_avg": 0.008898949692957103,
      "epoch": 5.054473876953125e-06,
      "step": 5300
    },
    {
      "epoch": 5.054473876953125e-06,
      "step": 5300,
      "training_step_time": 7.760025501251221
    },
    {
      "epoch": 5.055427551269531e-06,
      "model_forward_time": 1.7539482116699219,
      "step": 5301
    },
    {
      "epoch": 5.055427551269531e-06,
      "step": 5301,
      "training_step_time": 7.61578631401062
    },
    {
      "epoch": 5.056381225585937e-06,
      "model_forward_time": 1.7525131702423096,
      "step": 5302
    },
    {
      "epoch": 5.056381225585937e-06,
      "step": 5302,
      "training_step_time": 7.385779857635498
    },
    {
      "epoch": 5.057334899902344e-06,
      "model_forward_time": 1.7739882469177246,
      "step": 5303
    },
    {
      "epoch": 5.057334899902344e-06,
      "step": 5303,
      "training_step_time": 7.343683481216431
    },
    {
      "epoch": 5.05828857421875e-06,
      "model_forward_time": 1.7870495319366455,
      "step": 5304
    },
    {
      "epoch": 5.05828857421875e-06,
      "step": 5304,
      "training_step_time": 7.224423170089722
    },
    {
      "epoch": 5.059242248535156e-06,
      "model_forward_time": 1.746755838394165,
      "step": 5305
    },
    {
      "epoch": 5.059242248535156e-06,
      "step": 5305,
      "training_step_time": 7.3781819343566895
    },
    {
      "epoch": 5.0601959228515625e-06,
      "model_forward_time": 1.748964786529541,
      "step": 5306
    },
    {
      "epoch": 5.0601959228515625e-06,
      "step": 5306,
      "training_step_time": 8.468504428863525
    },
    {
      "epoch": 5.061149597167969e-06,
      "model_forward_time": 1.7549617290496826,
      "step": 5307
    },
    {
      "epoch": 5.061149597167969e-06,
      "step": 5307,
      "training_step_time": 9.091567516326904
    },
    {
      "epoch": 5.062103271484375e-06,
      "model_forward_time": 1.7500042915344238,
      "step": 5308
    },
    {
      "epoch": 5.062103271484375e-06,
      "step": 5308,
      "training_step_time": 7.660557270050049
    },
    {
      "epoch": 5.063056945800781e-06,
      "model_forward_time": 2.9384047985076904,
      "step": 5309
    },
    {
      "epoch": 5.063056945800781e-06,
      "step": 5309,
      "training_step_time": 7.751450777053833
    },
    {
      "epoch": 5.064010620117187e-06,
      "grad_norm": 0.3681463301181793,
      "learning_rate": 9.999738960803874e-06,
      "loss": 0.0727,
      "step": 5310
    },
    {
      "epoch": 5.064010620117187e-06,
      "model_forward_time": 1.753220558166504,
      "step": 5310
    },
    {
      "dynamics_loss_avg": 0.0687549825757742,
      "epoch": 5.064010620117187e-06,
      "step": 5310
    },
    {
      "action_loss_avg": 0.007865138654597103,
      "epoch": 5.064010620117187e-06,
      "step": 5310
    },
    {
      "epoch": 5.064010620117187e-06,
      "step": 5310,
      "training_step_time": 8.071330070495605
    },
    {
      "epoch": 5.064964294433594e-06,
      "model_forward_time": 1.7600035667419434,
      "step": 5311
    },
    {
      "epoch": 5.064964294433594e-06,
      "step": 5311,
      "training_step_time": 7.546021461486816
    },
    {
      "epoch": 5.06591796875e-06,
      "model_forward_time": 1.7497203350067139,
      "step": 5312
    },
    {
      "epoch": 5.06591796875e-06,
      "step": 5312,
      "training_step_time": 7.335839033126831
    },
    {
      "epoch": 5.066871643066406e-06,
      "model_forward_time": 1.7602035999298096,
      "step": 5313
    },
    {
      "epoch": 5.066871643066406e-06,
      "step": 5313,
      "training_step_time": 7.2471394538879395
    },
    {
      "epoch": 5.0678253173828126e-06,
      "model_forward_time": 1.7511892318725586,
      "step": 5314
    },
    {
      "epoch": 5.0678253173828126e-06,
      "step": 5314,
      "training_step_time": 7.315699338912964
    },
    {
      "epoch": 5.068778991699219e-06,
      "model_forward_time": 1.7559547424316406,
      "step": 5315
    },
    {
      "epoch": 5.068778991699219e-06,
      "step": 5315,
      "training_step_time": 7.414587497711182
    },
    {
      "epoch": 5.069732666015625e-06,
      "model_forward_time": 1.7499923706054688,
      "step": 5316
    },
    {
      "epoch": 5.069732666015625e-06,
      "step": 5316,
      "training_step_time": 7.7415220737457275
    },
    {
      "epoch": 5.070686340332031e-06,
      "model_forward_time": 1.7481589317321777,
      "step": 5317
    },
    {
      "epoch": 5.070686340332031e-06,
      "step": 5317,
      "training_step_time": 8.71937870979309
    },
    {
      "epoch": 5.071640014648437e-06,
      "model_forward_time": 2.9191596508026123,
      "step": 5318
    },
    {
      "epoch": 5.071640014648437e-06,
      "step": 5318,
      "training_step_time": 9.325047731399536
    },
    {
      "epoch": 5.072593688964844e-06,
      "model_forward_time": 1.749617576599121,
      "step": 5319
    },
    {
      "epoch": 5.072593688964844e-06,
      "step": 5319,
      "training_step_time": 8.112329244613647
    },
    {
      "epoch": 5.07354736328125e-06,
      "grad_norm": 0.43121641874313354,
      "learning_rate": 9.99972179182838e-06,
      "loss": 0.0791,
      "step": 5320
    },
    {
      "epoch": 5.07354736328125e-06,
      "model_forward_time": 1.7729113101959229,
      "step": 5320
    },
    {
      "dynamics_loss_avg": 0.07072534039616585,
      "epoch": 5.07354736328125e-06,
      "step": 5320
    },
    {
      "action_loss_avg": 0.010155367886181921,
      "epoch": 5.07354736328125e-06,
      "step": 5320
    },
    {
      "epoch": 5.07354736328125e-06,
      "step": 5320,
      "training_step_time": 7.583316326141357
    },
    {
      "epoch": 5.074501037597656e-06,
      "model_forward_time": 1.7518823146820068,
      "step": 5321
    },
    {
      "epoch": 5.074501037597656e-06,
      "step": 5321,
      "training_step_time": 7.309900999069214
    },
    {
      "epoch": 5.0754547119140626e-06,
      "model_forward_time": 1.7542645931243896,
      "step": 5322
    },
    {
      "epoch": 5.0754547119140626e-06,
      "step": 5322,
      "training_step_time": 7.29244327545166
    },
    {
      "epoch": 5.076408386230469e-06,
      "model_forward_time": 1.8121614456176758,
      "step": 5323
    },
    {
      "epoch": 5.076408386230469e-06,
      "step": 5323,
      "training_step_time": 7.334398984909058
    },
    {
      "epoch": 5.077362060546875e-06,
      "model_forward_time": 1.7524170875549316,
      "step": 5324
    },
    {
      "epoch": 5.077362060546875e-06,
      "step": 5324,
      "training_step_time": 7.9626593589782715
    },
    {
      "epoch": 5.078315734863281e-06,
      "model_forward_time": 1.753767967224121,
      "step": 5325
    },
    {
      "epoch": 5.078315734863281e-06,
      "step": 5325,
      "training_step_time": 10.297967910766602
    },
    {
      "epoch": 5.079269409179687e-06,
      "model_forward_time": 1.738218069076538,
      "step": 5326
    },
    {
      "epoch": 5.079269409179687e-06,
      "step": 5326,
      "training_step_time": 9.140950441360474
    },
    {
      "epoch": 5.080223083496094e-06,
      "model_forward_time": 2.9564762115478516,
      "step": 5327
    },
    {
      "epoch": 5.080223083496094e-06,
      "step": 5327,
      "training_step_time": 8.613154649734497
    },
    {
      "epoch": 5.0811767578125e-06,
      "model_forward_time": 1.7508151531219482,
      "step": 5328
    },
    {
      "epoch": 5.0811767578125e-06,
      "step": 5328,
      "training_step_time": 7.737576007843018
    },
    {
      "epoch": 5.082130432128906e-06,
      "model_forward_time": 1.7559492588043213,
      "step": 5329
    },
    {
      "epoch": 5.082130432128906e-06,
      "step": 5329,
      "training_step_time": 8.269543647766113
    },
    {
      "epoch": 5.0830841064453126e-06,
      "grad_norm": 0.5222413539886475,
      "learning_rate": 9.999704076090831e-06,
      "loss": 0.0833,
      "step": 5330
    },
    {
      "epoch": 5.0830841064453126e-06,
      "model_forward_time": 1.7600860595703125,
      "step": 5330
    },
    {
      "dynamics_loss_avg": 0.06658498495817185,
      "epoch": 5.0830841064453126e-06,
      "step": 5330
    },
    {
      "action_loss_avg": 0.00582313728518784,
      "epoch": 5.0830841064453126e-06,
      "step": 5330
    },
    {
      "epoch": 5.0830841064453126e-06,
      "step": 5330,
      "training_step_time": 7.368386507034302
    },
    {
      "epoch": 5.084037780761719e-06,
      "model_forward_time": 1.7494659423828125,
      "step": 5331
    },
    {
      "epoch": 5.084037780761719e-06,
      "step": 5331,
      "training_step_time": 7.273412227630615
    },
    {
      "epoch": 5.084991455078125e-06,
      "model_forward_time": 1.8059172630310059,
      "step": 5332
    },
    {
      "epoch": 5.084991455078125e-06,
      "step": 5332,
      "training_step_time": 7.349077224731445
    },
    {
      "epoch": 5.0859451293945314e-06,
      "model_forward_time": 1.7518963813781738,
      "step": 5333
    },
    {
      "epoch": 5.0859451293945314e-06,
      "step": 5333,
      "training_step_time": 7.412698745727539
    },
    {
      "epoch": 5.086898803710937e-06,
      "model_forward_time": 1.7615115642547607,
      "step": 5334
    },
    {
      "epoch": 5.086898803710937e-06,
      "step": 5334,
      "training_step_time": 7.60675835609436
    },
    {
      "epoch": 5.087852478027344e-06,
      "model_forward_time": 1.7543985843658447,
      "step": 5335
    },
    {
      "epoch": 5.087852478027344e-06,
      "step": 5335,
      "training_step_time": 8.351226568222046
    },
    {
      "epoch": 5.08880615234375e-06,
      "model_forward_time": 2.9704225063323975,
      "step": 5336
    },
    {
      "epoch": 5.08880615234375e-06,
      "step": 5336,
      "training_step_time": 7.790912628173828
    },
    {
      "epoch": 5.089759826660156e-06,
      "model_forward_time": 1.7653965950012207,
      "step": 5337
    },
    {
      "epoch": 5.089759826660156e-06,
      "step": 5337,
      "training_step_time": 8.719873428344727
    },
    {
      "epoch": 5.0907135009765626e-06,
      "model_forward_time": 1.756500482559204,
      "step": 5338
    },
    {
      "epoch": 5.0907135009765626e-06,
      "step": 5338,
      "training_step_time": 8.311890602111816
    },
    {
      "epoch": 5.091667175292969e-06,
      "model_forward_time": 1.7573888301849365,
      "step": 5339
    },
    {
      "epoch": 5.091667175292969e-06,
      "step": 5339,
      "training_step_time": 8.377361536026001
    },
    {
      "epoch": 5.092620849609375e-06,
      "grad_norm": 0.45243462920188904,
      "learning_rate": 9.999685813593158e-06,
      "loss": 0.0797,
      "step": 5340
    },
    {
      "epoch": 5.092620849609375e-06,
      "model_forward_time": 1.7618272304534912,
      "step": 5340
    },
    {
      "dynamics_loss_avg": 0.05605967007577419,
      "epoch": 5.092620849609375e-06,
      "step": 5340
    },
    {
      "action_loss_avg": 0.0044628582429140804,
      "epoch": 5.092620849609375e-06,
      "step": 5340
    },
    {
      "epoch": 5.092620849609375e-06,
      "step": 5340,
      "training_step_time": 7.442206382751465
    },
    {
      "epoch": 5.0935745239257814e-06,
      "model_forward_time": 1.827345848083496,
      "step": 5341
    },
    {
      "epoch": 5.0935745239257814e-06,
      "step": 5341,
      "training_step_time": 8.017101526260376
    },
    {
      "epoch": 5.094528198242187e-06,
      "model_forward_time": 1.7696609497070312,
      "step": 5342
    },
    {
      "epoch": 5.094528198242187e-06,
      "step": 5342,
      "training_step_time": 7.519534111022949
    },
    {
      "epoch": 5.095481872558594e-06,
      "model_forward_time": 1.757737636566162,
      "step": 5343
    },
    {
      "epoch": 5.095481872558594e-06,
      "step": 5343,
      "training_step_time": 8.173413038253784
    },
    {
      "epoch": 5.096435546875e-06,
      "model_forward_time": 1.760085105895996,
      "step": 5344
    },
    {
      "epoch": 5.096435546875e-06,
      "step": 5344,
      "training_step_time": 7.822431564331055
    },
    {
      "epoch": 5.097389221191406e-06,
      "model_forward_time": 2.854403495788574,
      "step": 5345
    },
    {
      "epoch": 5.097389221191406e-06,
      "step": 5345,
      "training_step_time": 8.434256792068481
    },
    {
      "epoch": 5.098342895507813e-06,
      "model_forward_time": 1.7550466060638428,
      "step": 5346
    },
    {
      "epoch": 5.098342895507813e-06,
      "step": 5346,
      "training_step_time": 8.703579664230347
    },
    {
      "epoch": 5.099296569824219e-06,
      "model_forward_time": 1.769120693206787,
      "step": 5347
    },
    {
      "epoch": 5.099296569824219e-06,
      "step": 5347,
      "training_step_time": 7.680501699447632
    },
    {
      "epoch": 5.100250244140625e-06,
      "model_forward_time": 1.762920618057251,
      "step": 5348
    },
    {
      "epoch": 5.100250244140625e-06,
      "step": 5348,
      "training_step_time": 7.647115468978882
    },
    {
      "epoch": 5.1012039184570314e-06,
      "model_forward_time": 1.7548472881317139,
      "step": 5349
    },
    {
      "epoch": 5.1012039184570314e-06,
      "step": 5349,
      "training_step_time": 7.1763856410980225
    },
    {
      "epoch": 5.102157592773437e-06,
      "grad_norm": 0.5982064604759216,
      "learning_rate": 9.999667004337363e-06,
      "loss": 0.0804,
      "step": 5350
    },
    {
      "epoch": 5.102157592773437e-06,
      "model_forward_time": 1.80582594871521,
      "step": 5350
    },
    {
      "dynamics_loss_avg": 0.07519215047359466,
      "epoch": 5.102157592773437e-06,
      "step": 5350
    },
    {
      "action_loss_avg": 0.009396822331473232,
      "epoch": 5.102157592773437e-06,
      "step": 5350
    },
    {
      "epoch": 5.102157592773437e-06,
      "step": 5350,
      "training_step_time": 7.371563911437988
    },
    {
      "epoch": 5.103111267089844e-06,
      "model_forward_time": 1.7481942176818848,
      "step": 5351
    },
    {
      "epoch": 5.103111267089844e-06,
      "step": 5351,
      "training_step_time": 7.50533127784729
    },
    {
      "epoch": 5.10406494140625e-06,
      "model_forward_time": 1.7601423263549805,
      "step": 5352
    },
    {
      "epoch": 5.10406494140625e-06,
      "step": 5352,
      "training_step_time": 8.241393327713013
    },
    {
      "epoch": 5.105018615722656e-06,
      "model_forward_time": 1.7670753002166748,
      "step": 5353
    },
    {
      "epoch": 5.105018615722656e-06,
      "step": 5353,
      "training_step_time": 8.710034370422363
    },
    {
      "epoch": 5.105972290039063e-06,
      "model_forward_time": 1.7490947246551514,
      "step": 5354
    },
    {
      "epoch": 5.105972290039063e-06,
      "step": 5354,
      "training_step_time": 8.649902820587158
    },
    {
      "epoch": 5.106925964355469e-06,
      "model_forward_time": 2.938291072845459,
      "step": 5355
    },
    {
      "epoch": 5.106925964355469e-06,
      "step": 5355,
      "training_step_time": 8.253670454025269
    },
    {
      "epoch": 5.107879638671875e-06,
      "model_forward_time": 1.0614676475524902,
      "step": 5356
    },
    {
      "epoch": 5.107879638671875e-06,
      "step": 5356,
      "training_step_time": 7.720993280410767
    },
    {
      "epoch": 5.1088333129882814e-06,
      "model_forward_time": 1.7555170059204102,
      "step": 5357
    },
    {
      "epoch": 5.1088333129882814e-06,
      "step": 5357,
      "training_step_time": 7.654288291931152
    },
    {
      "epoch": 5.109786987304687e-06,
      "model_forward_time": 1.7713842391967773,
      "step": 5358
    },
    {
      "epoch": 5.109786987304687e-06,
      "step": 5358,
      "training_step_time": 7.2091453075408936
    },
    {
      "epoch": 5.110740661621094e-06,
      "model_forward_time": 1.7776720523834229,
      "step": 5359
    },
    {
      "epoch": 5.110740661621094e-06,
      "step": 5359,
      "training_step_time": 7.387107849121094
    },
    {
      "epoch": 5.1116943359375e-06,
      "grad_norm": 0.4116917848587036,
      "learning_rate": 9.9996476483255e-06,
      "loss": 0.0761,
      "step": 5360
    },
    {
      "epoch": 5.1116943359375e-06,
      "model_forward_time": 1.751530647277832,
      "step": 5360
    },
    {
      "dynamics_loss_avg": 0.07579457797110081,
      "epoch": 5.1116943359375e-06,
      "step": 5360
    },
    {
      "action_loss_avg": 0.008687499305233359,
      "epoch": 5.1116943359375e-06,
      "step": 5360
    },
    {
      "epoch": 5.1116943359375e-06,
      "step": 5360,
      "training_step_time": 7.5240747928619385
    },
    {
      "epoch": 5.112648010253906e-06,
      "model_forward_time": 1.7702867984771729,
      "step": 5361
    },
    {
      "epoch": 5.112648010253906e-06,
      "step": 5361,
      "training_step_time": 7.632327079772949
    },
    {
      "epoch": 5.113601684570313e-06,
      "model_forward_time": 1.7593858242034912,
      "step": 5362
    },
    {
      "epoch": 5.113601684570313e-06,
      "step": 5362,
      "training_step_time": 8.599095582962036
    },
    {
      "epoch": 5.114555358886719e-06,
      "model_forward_time": 1.7496044635772705,
      "step": 5363
    },
    {
      "epoch": 5.114555358886719e-06,
      "step": 5363,
      "training_step_time": 9.673871278762817
    },
    {
      "epoch": 5.115509033203125e-06,
      "model_forward_time": 2.9070632457733154,
      "step": 5364
    },
    {
      "epoch": 5.115509033203125e-06,
      "step": 5364,
      "training_step_time": 7.737246036529541
    },
    {
      "epoch": 5.1164627075195315e-06,
      "model_forward_time": 1.7579689025878906,
      "step": 5365
    },
    {
      "epoch": 5.1164627075195315e-06,
      "step": 5365,
      "training_step_time": 7.692345857620239
    },
    {
      "epoch": 5.117416381835937e-06,
      "model_forward_time": 1.7679963111877441,
      "step": 5366
    },
    {
      "epoch": 5.117416381835937e-06,
      "step": 5366,
      "training_step_time": 7.621823072433472
    },
    {
      "epoch": 5.118370056152344e-06,
      "model_forward_time": 1.753758430480957,
      "step": 5367
    },
    {
      "epoch": 5.118370056152344e-06,
      "step": 5367,
      "training_step_time": 7.105429410934448
    },
    {
      "epoch": 5.11932373046875e-06,
      "model_forward_time": 1.760340929031372,
      "step": 5368
    },
    {
      "epoch": 5.11932373046875e-06,
      "step": 5368,
      "training_step_time": 7.3097639083862305
    },
    {
      "epoch": 5.120277404785156e-06,
      "model_forward_time": 1.8124492168426514,
      "step": 5369
    },
    {
      "epoch": 5.120277404785156e-06,
      "step": 5369,
      "training_step_time": 7.477530241012573
    },
    {
      "epoch": 5.121231079101563e-06,
      "grad_norm": 0.4391654431819916,
      "learning_rate": 9.999627745559688e-06,
      "loss": 0.0738,
      "step": 5370
    },
    {
      "epoch": 5.121231079101563e-06,
      "model_forward_time": 1.7545750141143799,
      "step": 5370
    },
    {
      "dynamics_loss_avg": 0.06510171294212341,
      "epoch": 5.121231079101563e-06,
      "step": 5370
    },
    {
      "action_loss_avg": 0.005921937501989305,
      "epoch": 5.121231079101563e-06,
      "step": 5370
    },
    {
      "epoch": 5.121231079101563e-06,
      "step": 5370,
      "training_step_time": 8.286906480789185
    },
    {
      "epoch": 5.122184753417969e-06,
      "model_forward_time": 1.7558891773223877,
      "step": 5371
    },
    {
      "epoch": 5.122184753417969e-06,
      "step": 5371,
      "training_step_time": 8.639995813369751
    },
    {
      "epoch": 5.123138427734375e-06,
      "model_forward_time": 1.7697947025299072,
      "step": 5372
    },
    {
      "epoch": 5.123138427734375e-06,
      "step": 5372,
      "training_step_time": 7.778256416320801
    },
    {
      "epoch": 5.1240921020507815e-06,
      "model_forward_time": 2.8725152015686035,
      "step": 5373
    },
    {
      "epoch": 5.1240921020507815e-06,
      "step": 5373,
      "training_step_time": 7.711771488189697
    },
    {
      "epoch": 5.125045776367187e-06,
      "model_forward_time": 1.7513635158538818,
      "step": 5374
    },
    {
      "epoch": 5.125045776367187e-06,
      "step": 5374,
      "training_step_time": 7.674211740493774
    },
    {
      "epoch": 5.125999450683594e-06,
      "model_forward_time": 1.776003360748291,
      "step": 5375
    },
    {
      "epoch": 5.125999450683594e-06,
      "step": 5375,
      "training_step_time": 8.173459768295288
    },
    {
      "epoch": 5.126953125e-06,
      "model_forward_time": 1.7534284591674805,
      "step": 5376
    },
    {
      "epoch": 5.126953125e-06,
      "step": 5376,
      "training_step_time": 7.371380805969238
    },
    {
      "epoch": 5.127906799316406e-06,
      "model_forward_time": 1.7537119388580322,
      "step": 5377
    },
    {
      "epoch": 5.127906799316406e-06,
      "step": 5377,
      "training_step_time": 7.338075399398804
    },
    {
      "epoch": 5.128860473632813e-06,
      "model_forward_time": 1.844923734664917,
      "step": 5378
    },
    {
      "epoch": 5.128860473632813e-06,
      "step": 5378,
      "training_step_time": 7.447757005691528
    },
    {
      "epoch": 5.129814147949218e-06,
      "model_forward_time": 1.7421741485595703,
      "step": 5379
    },
    {
      "epoch": 5.129814147949218e-06,
      "step": 5379,
      "training_step_time": 7.39556097984314
    },
    {
      "epoch": 5.130767822265625e-06,
      "grad_norm": 0.2766830027103424,
      "learning_rate": 9.999607296042102e-06,
      "loss": 0.0785,
      "step": 5380
    },
    {
      "epoch": 5.130767822265625e-06,
      "model_forward_time": 1.752194881439209,
      "step": 5380
    },
    {
      "dynamics_loss_avg": 0.06475646756589412,
      "epoch": 5.130767822265625e-06,
      "step": 5380
    },
    {
      "action_loss_avg": 0.008733433485031129,
      "epoch": 5.130767822265625e-06,
      "step": 5380
    },
    {
      "epoch": 5.130767822265625e-06,
      "step": 5380,
      "training_step_time": 8.460562467575073
    },
    {
      "epoch": 5.1317214965820315e-06,
      "model_forward_time": 1.7677178382873535,
      "step": 5381
    },
    {
      "epoch": 5.1317214965820315e-06,
      "step": 5381,
      "training_step_time": 7.6905786991119385
    },
    {
      "epoch": 5.132675170898437e-06,
      "model_forward_time": 1.7493491172790527,
      "step": 5382
    },
    {
      "epoch": 5.132675170898437e-06,
      "step": 5382,
      "training_step_time": 8.148042440414429
    },
    {
      "epoch": 5.133628845214844e-06,
      "model_forward_time": 1.7552053928375244,
      "step": 5383
    },
    {
      "epoch": 5.133628845214844e-06,
      "step": 5383,
      "training_step_time": 7.633542537689209
    },
    {
      "epoch": 5.13458251953125e-06,
      "model_forward_time": 1.7633514404296875,
      "step": 5384
    },
    {
      "epoch": 5.13458251953125e-06,
      "step": 5384,
      "training_step_time": 7.640149116516113
    },
    {
      "epoch": 5.135536193847656e-06,
      "model_forward_time": 1.760080099105835,
      "step": 5385
    },
    {
      "epoch": 5.135536193847656e-06,
      "step": 5385,
      "training_step_time": 8.586884021759033
    },
    {
      "epoch": 5.136489868164063e-06,
      "model_forward_time": 1.7609608173370361,
      "step": 5386
    },
    {
      "epoch": 5.136489868164063e-06,
      "step": 5386,
      "training_step_time": 7.756446838378906
    },
    {
      "epoch": 5.137443542480468e-06,
      "model_forward_time": 1.8168821334838867,
      "step": 5387
    },
    {
      "epoch": 5.137443542480468e-06,
      "step": 5387,
      "training_step_time": 7.379714488983154
    },
    {
      "epoch": 5.138397216796875e-06,
      "model_forward_time": 1.7517132759094238,
      "step": 5388
    },
    {
      "epoch": 5.138397216796875e-06,
      "step": 5388,
      "training_step_time": 8.21432113647461
    },
    {
      "epoch": 5.1393508911132815e-06,
      "model_forward_time": 1.752985954284668,
      "step": 5389
    },
    {
      "epoch": 5.1393508911132815e-06,
      "step": 5389,
      "training_step_time": 7.538835048675537
    },
    {
      "epoch": 5.140304565429687e-06,
      "grad_norm": 0.3032333254814148,
      "learning_rate": 9.999586299774979e-06,
      "loss": 0.0875,
      "step": 5390
    },
    {
      "epoch": 5.140304565429687e-06,
      "model_forward_time": 1.7593438625335693,
      "step": 5390
    },
    {
      "dynamics_loss_avg": 0.06358067505061626,
      "epoch": 5.140304565429687e-06,
      "step": 5390
    },
    {
      "action_loss_avg": 0.007303671725094318,
      "epoch": 5.140304565429687e-06,
      "step": 5390
    },
    {
      "epoch": 5.140304565429687e-06,
      "step": 5390,
      "training_step_time": 9.209500074386597
    },
    {
      "epoch": 5.141258239746094e-06,
      "model_forward_time": 1.7630181312561035,
      "step": 5391
    },
    {
      "epoch": 5.141258239746094e-06,
      "step": 5391,
      "training_step_time": 8.23673939704895
    },
    {
      "epoch": 5.1422119140625e-06,
      "model_forward_time": 2.948962688446045,
      "step": 5392
    },
    {
      "epoch": 5.1422119140625e-06,
      "step": 5392,
      "training_step_time": 7.72992730140686
    },
    {
      "epoch": 5.143165588378906e-06,
      "model_forward_time": 1.7558979988098145,
      "step": 5393
    },
    {
      "epoch": 5.143165588378906e-06,
      "step": 5393,
      "training_step_time": 8.654608011245728
    },
    {
      "epoch": 5.144119262695313e-06,
      "model_forward_time": 1.7695677280426025,
      "step": 5394
    },
    {
      "epoch": 5.144119262695313e-06,
      "step": 5394,
      "training_step_time": 7.556079626083374
    },
    {
      "epoch": 5.145072937011718e-06,
      "model_forward_time": 1.7556226253509521,
      "step": 5395
    },
    {
      "epoch": 5.145072937011718e-06,
      "step": 5395,
      "training_step_time": 7.573970079421997
    },
    {
      "epoch": 5.146026611328125e-06,
      "model_forward_time": 1.811267375946045,
      "step": 5396
    },
    {
      "epoch": 5.146026611328125e-06,
      "step": 5396,
      "training_step_time": 7.430382251739502
    },
    {
      "epoch": 5.1469802856445315e-06,
      "model_forward_time": 1.7491230964660645,
      "step": 5397
    },
    {
      "epoch": 5.1469802856445315e-06,
      "step": 5397,
      "training_step_time": 7.395157814025879
    },
    {
      "epoch": 5.147933959960937e-06,
      "model_forward_time": 1.7594013214111328,
      "step": 5398
    },
    {
      "epoch": 5.147933959960937e-06,
      "step": 5398,
      "training_step_time": 7.5049965381622314
    },
    {
      "epoch": 5.148887634277344e-06,
      "model_forward_time": 1.7562592029571533,
      "step": 5399
    },
    {
      "epoch": 5.148887634277344e-06,
      "step": 5399,
      "training_step_time": 7.713210821151733
    },
    {
      "epoch": 5.14984130859375e-06,
      "grad_norm": 0.3317112326622009,
      "learning_rate": 9.999564756760616e-06,
      "loss": 0.0845,
      "step": 5400
    },
    {
      "epoch": 5.14984130859375e-06,
      "model_forward_time": 1.7627289295196533,
      "step": 5400
    },
    {
      "dynamics_loss_avg": 0.06761581161990762,
      "epoch": 5.14984130859375e-06,
      "step": 5400
    },
    {
      "action_loss_avg": 0.006680008996045217,
      "epoch": 5.14984130859375e-06,
      "step": 5400
    },
    {
      "epoch": 5.14984130859375e-06,
      "step": 5400,
      "training_step_time": 8.7261483669281
    },
    {
      "epoch": 5.150794982910156e-06,
      "model_forward_time": 2.929513454437256,
      "step": 5401
    },
    {
      "epoch": 5.150794982910156e-06,
      "step": 5401,
      "training_step_time": 8.581857204437256
    },
    {
      "epoch": 5.151748657226563e-06,
      "model_forward_time": 1.7523157596588135,
      "step": 5402
    },
    {
      "epoch": 5.151748657226563e-06,
      "step": 5402,
      "training_step_time": 7.674691200256348
    },
    {
      "epoch": 5.152702331542968e-06,
      "model_forward_time": 1.775780200958252,
      "step": 5403
    },
    {
      "epoch": 5.152702331542968e-06,
      "step": 5403,
      "training_step_time": 7.637759447097778
    },
    {
      "epoch": 5.153656005859375e-06,
      "model_forward_time": 1.7516634464263916,
      "step": 5404
    },
    {
      "epoch": 5.153656005859375e-06,
      "step": 5404,
      "training_step_time": 7.564803600311279
    },
    {
      "epoch": 5.1546096801757815e-06,
      "model_forward_time": 1.8087637424468994,
      "step": 5405
    },
    {
      "epoch": 5.1546096801757815e-06,
      "step": 5405,
      "training_step_time": 7.337327718734741
    },
    {
      "epoch": 5.155563354492187e-06,
      "model_forward_time": 1.7680068016052246,
      "step": 5406
    },
    {
      "epoch": 5.155563354492187e-06,
      "step": 5406,
      "training_step_time": 7.405832052230835
    },
    {
      "epoch": 5.156517028808594e-06,
      "model_forward_time": 1.7629005908966064,
      "step": 5407
    },
    {
      "epoch": 5.156517028808594e-06,
      "step": 5407,
      "training_step_time": 8.415247440338135
    },
    {
      "epoch": 5.157470703125e-06,
      "model_forward_time": 1.763686180114746,
      "step": 5408
    },
    {
      "epoch": 5.157470703125e-06,
      "step": 5408,
      "training_step_time": 9.127265453338623
    },
    {
      "epoch": 5.158424377441406e-06,
      "model_forward_time": 1.7671263217926025,
      "step": 5409
    },
    {
      "epoch": 5.158424377441406e-06,
      "step": 5409,
      "training_step_time": 8.543423891067505
    },
    {
      "epoch": 5.159378051757813e-06,
      "grad_norm": 0.23213136196136475,
      "learning_rate": 9.999542667001366e-06,
      "loss": 0.0783,
      "step": 5410
    },
    {
      "epoch": 5.159378051757813e-06,
      "model_forward_time": 2.8940532207489014,
      "step": 5410
    },
    {
      "dynamics_loss_avg": 0.09031310230493546,
      "epoch": 5.159378051757813e-06,
      "step": 5410
    },
    {
      "action_loss_avg": 0.009724772768095136,
      "epoch": 5.159378051757813e-06,
      "step": 5410
    },
    {
      "epoch": 5.159378051757813e-06,
      "step": 5410,
      "training_step_time": 7.620797634124756
    },
    {
      "epoch": 5.160331726074218e-06,
      "model_forward_time": 1.7726190090179443,
      "step": 5411
    },
    {
      "epoch": 5.160331726074218e-06,
      "step": 5411,
      "training_step_time": 7.612279176712036
    },
    {
      "epoch": 5.161285400390625e-06,
      "model_forward_time": 1.757875680923462,
      "step": 5412
    },
    {
      "epoch": 5.161285400390625e-06,
      "step": 5412,
      "training_step_time": 7.636508464813232
    },
    {
      "epoch": 5.1622390747070315e-06,
      "model_forward_time": 1.7631428241729736,
      "step": 5413
    },
    {
      "epoch": 5.1622390747070315e-06,
      "step": 5413,
      "training_step_time": 7.525388479232788
    },
    {
      "epoch": 5.163192749023437e-06,
      "model_forward_time": 1.7697300910949707,
      "step": 5414
    },
    {
      "epoch": 5.163192749023437e-06,
      "step": 5414,
      "training_step_time": 7.297470569610596
    },
    {
      "epoch": 5.164146423339844e-06,
      "model_forward_time": 1.785400152206421,
      "step": 5415
    },
    {
      "epoch": 5.164146423339844e-06,
      "step": 5415,
      "training_step_time": 7.459945917129517
    },
    {
      "epoch": 5.16510009765625e-06,
      "model_forward_time": 1.7571306228637695,
      "step": 5416
    },
    {
      "epoch": 5.16510009765625e-06,
      "step": 5416,
      "training_step_time": 8.20921277999878
    },
    {
      "epoch": 5.166053771972656e-06,
      "model_forward_time": 1.7618634700775146,
      "step": 5417
    },
    {
      "epoch": 5.166053771972656e-06,
      "step": 5417,
      "training_step_time": 8.681864261627197
    },
    {
      "epoch": 5.167007446289063e-06,
      "model_forward_time": 1.75437593460083,
      "step": 5418
    },
    {
      "epoch": 5.167007446289063e-06,
      "step": 5418,
      "training_step_time": 7.646163463592529
    },
    {
      "epoch": 5.167961120605468e-06,
      "model_forward_time": 2.9012362957000732,
      "step": 5419
    },
    {
      "epoch": 5.167961120605468e-06,
      "step": 5419,
      "training_step_time": 9.28989839553833
    },
    {
      "epoch": 5.168914794921875e-06,
      "grad_norm": 0.39109566807746887,
      "learning_rate": 9.999520030499648e-06,
      "loss": 0.0783,
      "step": 5420
    },
    {
      "epoch": 5.168914794921875e-06,
      "model_forward_time": 1.7609643936157227,
      "step": 5420
    },
    {
      "dynamics_loss_avg": 0.08226545676589012,
      "epoch": 5.168914794921875e-06,
      "step": 5420
    },
    {
      "action_loss_avg": 0.009866637201048434,
      "epoch": 5.168914794921875e-06,
      "step": 5420
    },
    {
      "epoch": 5.168914794921875e-06,
      "step": 5420,
      "training_step_time": 8.141603946685791
    },
    {
      "epoch": 5.1698684692382815e-06,
      "model_forward_time": 1.7522737979888916,
      "step": 5421
    },
    {
      "epoch": 5.1698684692382815e-06,
      "step": 5421,
      "training_step_time": 8.329503297805786
    },
    {
      "epoch": 5.170822143554687e-06,
      "model_forward_time": 1.7635219097137451,
      "step": 5422
    },
    {
      "epoch": 5.170822143554687e-06,
      "step": 5422,
      "training_step_time": 7.600075960159302
    },
    {
      "epoch": 5.171775817871094e-06,
      "model_forward_time": 1.758336067199707,
      "step": 5423
    },
    {
      "epoch": 5.171775817871094e-06,
      "step": 5423,
      "training_step_time": 7.5797083377838135
    },
    {
      "epoch": 5.1727294921875e-06,
      "model_forward_time": 1.826876163482666,
      "step": 5424
    },
    {
      "epoch": 5.1727294921875e-06,
      "step": 5424,
      "training_step_time": 9.599188566207886
    },
    {
      "epoch": 5.173683166503906e-06,
      "model_forward_time": 1.7639873027801514,
      "step": 5425
    },
    {
      "epoch": 5.173683166503906e-06,
      "step": 5425,
      "training_step_time": 7.412413835525513
    },
    {
      "epoch": 5.174636840820313e-06,
      "model_forward_time": 1.765892744064331,
      "step": 5426
    },
    {
      "epoch": 5.174636840820313e-06,
      "step": 5426,
      "training_step_time": 9.24212646484375
    },
    {
      "epoch": 5.175590515136718e-06,
      "model_forward_time": 1.748866081237793,
      "step": 5427
    },
    {
      "epoch": 5.175590515136718e-06,
      "step": 5427,
      "training_step_time": 9.097136497497559
    },
    {
      "epoch": 5.176544189453125e-06,
      "model_forward_time": 1.7500789165496826,
      "step": 5428
    },
    {
      "epoch": 5.176544189453125e-06,
      "step": 5428,
      "training_step_time": 9.120897054672241
    },
    {
      "epoch": 5.1774978637695315e-06,
      "model_forward_time": 1.3842246532440186,
      "step": 5429
    },
    {
      "epoch": 5.1774978637695315e-06,
      "step": 5429,
      "training_step_time": 7.446006774902344
    },
    {
      "epoch": 5.178451538085937e-06,
      "grad_norm": 0.29278892278671265,
      "learning_rate": 9.999496847257937e-06,
      "loss": 0.076,
      "step": 5430
    },
    {
      "epoch": 5.178451538085937e-06,
      "model_forward_time": 1.7574834823608398,
      "step": 5430
    },
    {
      "dynamics_loss_avg": 0.077818070538342,
      "epoch": 5.178451538085937e-06,
      "step": 5430
    },
    {
      "action_loss_avg": 0.011517993838060648,
      "epoch": 5.178451538085937e-06,
      "step": 5430
    },
    {
      "epoch": 5.178451538085937e-06,
      "step": 5430,
      "training_step_time": 7.637578964233398
    },
    {
      "epoch": 5.179405212402344e-06,
      "model_forward_time": 1.7738885879516602,
      "step": 5431
    },
    {
      "epoch": 5.179405212402344e-06,
      "step": 5431,
      "training_step_time": 7.554549932479858
    },
    {
      "epoch": 5.18035888671875e-06,
      "model_forward_time": 1.7608616352081299,
      "step": 5432
    },
    {
      "epoch": 5.18035888671875e-06,
      "step": 5432,
      "training_step_time": 7.495850086212158
    },
    {
      "epoch": 5.181312561035156e-06,
      "model_forward_time": 1.8289768695831299,
      "step": 5433
    },
    {
      "epoch": 5.181312561035156e-06,
      "step": 5433,
      "training_step_time": 7.3407440185546875
    },
    {
      "epoch": 5.182266235351563e-06,
      "model_forward_time": 1.74822998046875,
      "step": 5434
    },
    {
      "epoch": 5.182266235351563e-06,
      "step": 5434,
      "training_step_time": 7.357724666595459
    },
    {
      "epoch": 5.183219909667968e-06,
      "model_forward_time": 1.7587361335754395,
      "step": 5435
    },
    {
      "epoch": 5.183219909667968e-06,
      "step": 5435,
      "training_step_time": 7.535501718521118
    },
    {
      "epoch": 5.184173583984375e-06,
      "model_forward_time": 1.76371169090271,
      "step": 5436
    },
    {
      "epoch": 5.184173583984375e-06,
      "step": 5436,
      "training_step_time": 9.73973274230957
    },
    {
      "epoch": 5.1851272583007815e-06,
      "model_forward_time": 1.7765212059020996,
      "step": 5437
    },
    {
      "epoch": 5.1851272583007815e-06,
      "step": 5437,
      "training_step_time": 8.191221237182617
    },
    {
      "epoch": 5.186080932617187e-06,
      "model_forward_time": 2.9511351585388184,
      "step": 5438
    },
    {
      "epoch": 5.186080932617187e-06,
      "step": 5438,
      "training_step_time": 9.3575279712677
    },
    {
      "epoch": 5.187034606933594e-06,
      "model_forward_time": 1.7578461170196533,
      "step": 5439
    },
    {
      "epoch": 5.187034606933594e-06,
      "step": 5439,
      "training_step_time": 8.599996089935303
    },
    {
      "epoch": 5.18798828125e-06,
      "grad_norm": 0.2855420708656311,
      "learning_rate": 9.999473117278765e-06,
      "loss": 0.0762,
      "step": 5440
    },
    {
      "epoch": 5.18798828125e-06,
      "model_forward_time": 1.8857340812683105,
      "step": 5440
    },
    {
      "dynamics_loss_avg": 0.05762611199170351,
      "epoch": 5.18798828125e-06,
      "step": 5440
    },
    {
      "action_loss_avg": 0.007322071399539709,
      "epoch": 5.18798828125e-06,
      "step": 5440
    },
    {
      "epoch": 5.18798828125e-06,
      "step": 5440,
      "training_step_time": 7.553673982620239
    },
    {
      "epoch": 5.188941955566406e-06,
      "model_forward_time": 1.9120891094207764,
      "step": 5441
    },
    {
      "epoch": 5.188941955566406e-06,
      "step": 5441,
      "training_step_time": 7.504538536071777
    },
    {
      "epoch": 5.189895629882813e-06,
      "model_forward_time": 1.9155056476593018,
      "step": 5442
    },
    {
      "epoch": 5.189895629882813e-06,
      "step": 5442,
      "training_step_time": 7.572098731994629
    },
    {
      "epoch": 5.190849304199218e-06,
      "model_forward_time": 1.9095745086669922,
      "step": 5443
    },
    {
      "epoch": 5.190849304199218e-06,
      "step": 5443,
      "training_step_time": 7.358530521392822
    },
    {
      "epoch": 5.191802978515625e-06,
      "model_forward_time": 1.9092342853546143,
      "step": 5444
    },
    {
      "epoch": 5.191802978515625e-06,
      "step": 5444,
      "training_step_time": 7.511417627334595
    },
    {
      "epoch": 5.1927566528320315e-06,
      "model_forward_time": 1.9288623332977295,
      "step": 5445
    },
    {
      "epoch": 5.1927566528320315e-06,
      "step": 5445,
      "training_step_time": 8.296079635620117
    },
    {
      "epoch": 5.193710327148437e-06,
      "model_forward_time": 1.7729337215423584,
      "step": 5446
    },
    {
      "epoch": 5.193710327148437e-06,
      "step": 5446,
      "training_step_time": 9.221639633178711
    },
    {
      "epoch": 5.194664001464844e-06,
      "model_forward_time": 2.9015119075775146,
      "step": 5447
    },
    {
      "epoch": 5.194664001464844e-06,
      "step": 5447,
      "training_step_time": 7.635903596878052
    },
    {
      "epoch": 5.19561767578125e-06,
      "model_forward_time": 1.91465163230896,
      "step": 5448
    },
    {
      "epoch": 5.19561767578125e-06,
      "step": 5448,
      "training_step_time": 7.5816810131073
    },
    {
      "epoch": 5.196571350097656e-06,
      "model_forward_time": 1.9233455657958984,
      "step": 5449
    },
    {
      "epoch": 5.196571350097656e-06,
      "step": 5449,
      "training_step_time": 7.541784286499023
    },
    {
      "epoch": 5.197525024414063e-06,
      "grad_norm": 0.24964483082294464,
      "learning_rate": 9.999448840564732e-06,
      "loss": 0.0726,
      "step": 5450
    },
    {
      "epoch": 5.197525024414063e-06,
      "model_forward_time": 1.9034125804901123,
      "step": 5450
    },
    {
      "dynamics_loss_avg": 0.0718155201524496,
      "epoch": 5.197525024414063e-06,
      "step": 5450
    },
    {
      "action_loss_avg": 0.008075884985737502,
      "epoch": 5.197525024414063e-06,
      "step": 5450
    },
    {
      "epoch": 5.197525024414063e-06,
      "step": 5450,
      "training_step_time": 7.503443956375122
    },
    {
      "epoch": 5.198478698730468e-06,
      "model_forward_time": 1.7551264762878418,
      "step": 5451
    },
    {
      "epoch": 5.198478698730468e-06,
      "step": 5451,
      "training_step_time": 7.577133893966675
    },
    {
      "epoch": 5.199432373046875e-06,
      "model_forward_time": 1.7478961944580078,
      "step": 5452
    },
    {
      "epoch": 5.199432373046875e-06,
      "step": 5452,
      "training_step_time": 8.169904232025146
    },
    {
      "epoch": 5.2003860473632815e-06,
      "model_forward_time": 1.7398738861083984,
      "step": 5453
    },
    {
      "epoch": 5.2003860473632815e-06,
      "step": 5453,
      "training_step_time": 7.505174160003662
    },
    {
      "epoch": 5.201339721679687e-06,
      "model_forward_time": 1.7520253658294678,
      "step": 5454
    },
    {
      "epoch": 5.201339721679687e-06,
      "step": 5454,
      "training_step_time": 7.577075958251953
    },
    {
      "epoch": 5.202293395996094e-06,
      "model_forward_time": 1.735517978668213,
      "step": 5455
    },
    {
      "epoch": 5.202293395996094e-06,
      "step": 5455,
      "training_step_time": 9.12086009979248
    },
    {
      "epoch": 5.2032470703125e-06,
      "model_forward_time": 2.897970199584961,
      "step": 5456
    },
    {
      "epoch": 5.2032470703125e-06,
      "step": 5456,
      "training_step_time": 8.168896675109863
    },
    {
      "epoch": 5.204200744628906e-06,
      "model_forward_time": 1.7637512683868408,
      "step": 5457
    },
    {
      "epoch": 5.204200744628906e-06,
      "step": 5457,
      "training_step_time": 8.428876638412476
    },
    {
      "epoch": 5.205154418945313e-06,
      "model_forward_time": 1.7507221698760986,
      "step": 5458
    },
    {
      "epoch": 5.205154418945313e-06,
      "step": 5458,
      "training_step_time": 8.583398342132568
    },
    {
      "epoch": 5.206108093261718e-06,
      "model_forward_time": 1.736130952835083,
      "step": 5459
    },
    {
      "epoch": 5.206108093261718e-06,
      "step": 5459,
      "training_step_time": 7.5207133293151855
    },
    {
      "epoch": 5.207061767578125e-06,
      "grad_norm": 0.45607295632362366,
      "learning_rate": 9.999424017118489e-06,
      "loss": 0.0799,
      "step": 5460
    },
    {
      "epoch": 5.207061767578125e-06,
      "model_forward_time": 1.72247314453125,
      "step": 5460
    },
    {
      "dynamics_loss_avg": 0.05303787272423506,
      "epoch": 5.207061767578125e-06,
      "step": 5460
    },
    {
      "action_loss_avg": 0.007656021579168737,
      "epoch": 5.207061767578125e-06,
      "step": 5460
    },
    {
      "epoch": 5.207061767578125e-06,
      "step": 5460,
      "training_step_time": 7.5695037841796875
    },
    {
      "epoch": 5.2080154418945315e-06,
      "model_forward_time": 1.7472097873687744,
      "step": 5461
    },
    {
      "epoch": 5.2080154418945315e-06,
      "step": 5461,
      "training_step_time": 7.361064195632935
    },
    {
      "epoch": 5.208969116210937e-06,
      "model_forward_time": 1.7421026229858398,
      "step": 5462
    },
    {
      "epoch": 5.208969116210937e-06,
      "step": 5462,
      "training_step_time": 7.482520341873169
    },
    {
      "epoch": 5.209922790527344e-06,
      "model_forward_time": 1.7428176403045654,
      "step": 5463
    },
    {
      "epoch": 5.209922790527344e-06,
      "step": 5463,
      "training_step_time": 7.561891317367554
    },
    {
      "epoch": 5.21087646484375e-06,
      "model_forward_time": 1.7392237186431885,
      "step": 5464
    },
    {
      "epoch": 5.21087646484375e-06,
      "step": 5464,
      "training_step_time": 8.14680790901184
    },
    {
      "epoch": 5.211830139160156e-06,
      "model_forward_time": 2.9373295307159424,
      "step": 5465
    },
    {
      "epoch": 5.211830139160156e-06,
      "step": 5465,
      "training_step_time": 10.193728685379028
    },
    {
      "epoch": 5.212783813476563e-06,
      "model_forward_time": 1.931039571762085,
      "step": 5466
    },
    {
      "epoch": 5.212783813476563e-06,
      "step": 5466,
      "training_step_time": 7.566697835922241
    },
    {
      "epoch": 5.213737487792968e-06,
      "model_forward_time": 1.929276704788208,
      "step": 5467
    },
    {
      "epoch": 5.213737487792968e-06,
      "step": 5467,
      "training_step_time": 8.207499265670776
    },
    {
      "epoch": 5.214691162109375e-06,
      "model_forward_time": 1.7744286060333252,
      "step": 5468
    },
    {
      "epoch": 5.214691162109375e-06,
      "step": 5468,
      "training_step_time": 7.538349628448486
    },
    {
      "epoch": 5.2156448364257815e-06,
      "model_forward_time": 1.7562541961669922,
      "step": 5469
    },
    {
      "epoch": 5.2156448364257815e-06,
      "step": 5469,
      "training_step_time": 8.025606155395508
    },
    {
      "epoch": 5.216598510742187e-06,
      "grad_norm": 0.30626577138900757,
      "learning_rate": 9.999398646942751e-06,
      "loss": 0.0852,
      "step": 5470
    },
    {
      "epoch": 5.216598510742187e-06,
      "model_forward_time": 1.73417329788208,
      "step": 5470
    },
    {
      "dynamics_loss_avg": 0.07249349802732467,
      "epoch": 5.216598510742187e-06,
      "step": 5470
    },
    {
      "action_loss_avg": 0.0077603754820302125,
      "epoch": 5.216598510742187e-06,
      "step": 5470
    },
    {
      "epoch": 5.216598510742187e-06,
      "step": 5470,
      "training_step_time": 7.641571044921875
    },
    {
      "epoch": 5.217552185058594e-06,
      "model_forward_time": 1.7403614521026611,
      "step": 5471
    },
    {
      "epoch": 5.217552185058594e-06,
      "step": 5471,
      "training_step_time": 7.554497957229614
    },
    {
      "epoch": 5.218505859375e-06,
      "model_forward_time": 1.7593164443969727,
      "step": 5472
    },
    {
      "epoch": 5.218505859375e-06,
      "step": 5472,
      "training_step_time": 8.743354082107544
    },
    {
      "epoch": 5.219459533691406e-06,
      "model_forward_time": 1.7441580295562744,
      "step": 5473
    },
    {
      "epoch": 5.219459533691406e-06,
      "step": 5473,
      "training_step_time": 8.834058284759521
    },
    {
      "epoch": 5.220413208007813e-06,
      "model_forward_time": 1.795605182647705,
      "step": 5474
    },
    {
      "epoch": 5.220413208007813e-06,
      "step": 5474,
      "training_step_time": 9.236919641494751
    },
    {
      "epoch": 5.221366882324218e-06,
      "model_forward_time": 1.732759952545166,
      "step": 5475
    },
    {
      "epoch": 5.221366882324218e-06,
      "step": 5475,
      "training_step_time": 7.582470178604126
    },
    {
      "epoch": 5.222320556640625e-06,
      "model_forward_time": 1.7907216548919678,
      "step": 5476
    },
    {
      "epoch": 5.222320556640625e-06,
      "step": 5476,
      "training_step_time": 7.720078706741333
    },
    {
      "epoch": 5.2232742309570315e-06,
      "model_forward_time": 1.7342758178710938,
      "step": 5477
    },
    {
      "epoch": 5.2232742309570315e-06,
      "step": 5477,
      "training_step_time": 7.611689805984497
    },
    {
      "epoch": 5.224227905273437e-06,
      "model_forward_time": 1.7454438209533691,
      "step": 5478
    },
    {
      "epoch": 5.224227905273437e-06,
      "step": 5478,
      "training_step_time": 7.345474481582642
    },
    {
      "epoch": 5.225181579589844e-06,
      "model_forward_time": 1.733597993850708,
      "step": 5479
    },
    {
      "epoch": 5.225181579589844e-06,
      "step": 5479,
      "training_step_time": 8.668065786361694
    },
    {
      "epoch": 5.22613525390625e-06,
      "grad_norm": 0.27376461029052734,
      "learning_rate": 9.999372730040295e-06,
      "loss": 0.0749,
      "step": 5480
    },
    {
      "epoch": 5.22613525390625e-06,
      "model_forward_time": 1.737896203994751,
      "step": 5480
    },
    {
      "dynamics_loss_avg": 0.0644463762640953,
      "epoch": 5.22613525390625e-06,
      "step": 5480
    },
    {
      "action_loss_avg": 0.006332122650928795,
      "epoch": 5.22613525390625e-06,
      "step": 5480
    },
    {
      "epoch": 5.22613525390625e-06,
      "step": 5480,
      "training_step_time": 7.626405715942383
    },
    {
      "epoch": 5.227088928222656e-06,
      "model_forward_time": 1.7381398677825928,
      "step": 5481
    },
    {
      "epoch": 5.227088928222656e-06,
      "step": 5481,
      "training_step_time": 8.85913896560669
    },
    {
      "epoch": 5.228042602539063e-06,
      "model_forward_time": 1.7463865280151367,
      "step": 5482
    },
    {
      "epoch": 5.228042602539063e-06,
      "step": 5482,
      "training_step_time": 7.7828357219696045
    },
    {
      "epoch": 5.228996276855468e-06,
      "model_forward_time": 1.736788272857666,
      "step": 5483
    },
    {
      "epoch": 5.228996276855468e-06,
      "step": 5483,
      "training_step_time": 8.536349534988403
    },
    {
      "epoch": 5.229949951171875e-06,
      "model_forward_time": 2.9604620933532715,
      "step": 5484
    },
    {
      "epoch": 5.229949951171875e-06,
      "step": 5484,
      "training_step_time": 8.383297443389893
    },
    {
      "epoch": 5.2309036254882816e-06,
      "model_forward_time": 1.7375755310058594,
      "step": 5485
    },
    {
      "epoch": 5.2309036254882816e-06,
      "step": 5485,
      "training_step_time": 7.64326286315918
    },
    {
      "epoch": 5.231857299804687e-06,
      "model_forward_time": 1.7266552448272705,
      "step": 5486
    },
    {
      "epoch": 5.231857299804687e-06,
      "step": 5486,
      "training_step_time": 7.550900220870972
    },
    {
      "epoch": 5.232810974121094e-06,
      "model_forward_time": 1.75498366355896,
      "step": 5487
    },
    {
      "epoch": 5.232810974121094e-06,
      "step": 5487,
      "training_step_time": 7.140806436538696
    },
    {
      "epoch": 5.2337646484375e-06,
      "model_forward_time": 1.7481939792633057,
      "step": 5488
    },
    {
      "epoch": 5.2337646484375e-06,
      "step": 5488,
      "training_step_time": 7.636004447937012
    },
    {
      "epoch": 5.234718322753906e-06,
      "model_forward_time": 1.741947889328003,
      "step": 5489
    },
    {
      "epoch": 5.234718322753906e-06,
      "step": 5489,
      "training_step_time": 7.520052909851074
    },
    {
      "epoch": 5.235671997070313e-06,
      "grad_norm": 0.3475744426250458,
      "learning_rate": 9.999346266413954e-06,
      "loss": 0.0759,
      "step": 5490
    },
    {
      "epoch": 5.235671997070313e-06,
      "model_forward_time": 1.7302396297454834,
      "step": 5490
    },
    {
      "dynamics_loss_avg": 0.088580022752285,
      "epoch": 5.235671997070313e-06,
      "step": 5490
    },
    {
      "action_loss_avg": 0.01596548573579639,
      "epoch": 5.235671997070313e-06,
      "step": 5490
    },
    {
      "epoch": 5.235671997070313e-06,
      "step": 5490,
      "training_step_time": 7.650867938995361
    },
    {
      "epoch": 5.2366256713867184e-06,
      "model_forward_time": 1.7303016185760498,
      "step": 5491
    },
    {
      "epoch": 5.2366256713867184e-06,
      "step": 5491,
      "training_step_time": 8.32088327407837
    },
    {
      "epoch": 5.237579345703125e-06,
      "model_forward_time": 1.744062900543213,
      "step": 5492
    },
    {
      "epoch": 5.237579345703125e-06,
      "step": 5492,
      "training_step_time": 9.265397548675537
    },
    {
      "epoch": 5.2385330200195316e-06,
      "model_forward_time": 2.89186954498291,
      "step": 5493
    },
    {
      "epoch": 5.2385330200195316e-06,
      "step": 5493,
      "training_step_time": 8.687333106994629
    },
    {
      "epoch": 5.239486694335937e-06,
      "model_forward_time": 1.7400522232055664,
      "step": 5494
    },
    {
      "epoch": 5.239486694335937e-06,
      "step": 5494,
      "training_step_time": 7.516781568527222
    },
    {
      "epoch": 5.240440368652344e-06,
      "model_forward_time": 1.7476558685302734,
      "step": 5495
    },
    {
      "epoch": 5.240440368652344e-06,
      "step": 5495,
      "training_step_time": 7.57516622543335
    },
    {
      "epoch": 5.24139404296875e-06,
      "model_forward_time": 1.73207688331604,
      "step": 5496
    },
    {
      "epoch": 5.24139404296875e-06,
      "step": 5496,
      "training_step_time": 7.497022390365601
    },
    {
      "epoch": 5.242347717285156e-06,
      "model_forward_time": 1.731170654296875,
      "step": 5497
    },
    {
      "epoch": 5.242347717285156e-06,
      "step": 5497,
      "training_step_time": 7.552737712860107
    },
    {
      "epoch": 5.243301391601563e-06,
      "model_forward_time": 1.7192766666412354,
      "step": 5498
    },
    {
      "epoch": 5.243301391601563e-06,
      "step": 5498,
      "training_step_time": 7.902941465377808
    },
    {
      "epoch": 5.2442550659179684e-06,
      "model_forward_time": 1.7166500091552734,
      "step": 5499
    },
    {
      "epoch": 5.2442550659179684e-06,
      "step": 5499,
      "training_step_time": 8.174685001373291
    },
    {
      "epoch": 5.245208740234375e-06,
      "grad_norm": 0.3409428298473358,
      "learning_rate": 9.99931925606662e-06,
      "loss": 0.0774,
      "step": 5500
    },
    {
      "epoch": 5.245208740234375e-06,
      "model_forward_time": 1.7231347560882568,
      "step": 5500
    },
    {
      "dynamics_loss_avg": 0.07993022445589304,
      "epoch": 5.245208740234375e-06,
      "step": 5500
    },
    {
      "action_loss_avg": 0.010629966913256794,
      "epoch": 5.245208740234375e-06,
      "step": 5500
    },
    {
      "epoch": 5.245208740234375e-06,
      "step": 5500,
      "training_step_time": 8.160215139389038
    },
    {
      "epoch": 5.2461624145507816e-06,
      "model_forward_time": 1.7241051197052002,
      "step": 5501
    },
    {
      "epoch": 5.2461624145507816e-06,
      "step": 5501,
      "training_step_time": 8.898154735565186
    },
    {
      "epoch": 5.247116088867187e-06,
      "model_forward_time": 2.86775541305542,
      "step": 5502
    },
    {
      "epoch": 5.247116088867187e-06,
      "step": 5502,
      "training_step_time": 7.667259931564331
    },
    {
      "epoch": 5.248069763183594e-06,
      "model_forward_time": 1.736175775527954,
      "step": 5503
    },
    {
      "epoch": 5.248069763183594e-06,
      "step": 5503,
      "training_step_time": 8.513548374176025
    },
    {
      "epoch": 5.2490234375e-06,
      "model_forward_time": 1.7291107177734375,
      "step": 5504
    },
    {
      "epoch": 5.2490234375e-06,
      "step": 5504,
      "training_step_time": 7.596554279327393
    },
    {
      "epoch": 5.249977111816406e-06,
      "model_forward_time": 1.7308349609375,
      "step": 5505
    },
    {
      "epoch": 5.249977111816406e-06,
      "step": 5505,
      "training_step_time": 8.585834741592407
    },
    {
      "epoch": 5.250930786132813e-06,
      "model_forward_time": 1.7393195629119873,
      "step": 5506
    },
    {
      "epoch": 5.250930786132813e-06,
      "step": 5506,
      "training_step_time": 7.633087635040283
    },
    {
      "epoch": 5.2518844604492184e-06,
      "model_forward_time": 1.733264446258545,
      "step": 5507
    },
    {
      "epoch": 5.2518844604492184e-06,
      "step": 5507,
      "training_step_time": 8.089139699935913
    },
    {
      "epoch": 5.252838134765625e-06,
      "model_forward_time": 1.74495530128479,
      "step": 5508
    },
    {
      "epoch": 5.252838134765625e-06,
      "step": 5508,
      "training_step_time": 8.278355360031128
    },
    {
      "epoch": 5.2537918090820316e-06,
      "model_forward_time": 1.7333917617797852,
      "step": 5509
    },
    {
      "epoch": 5.2537918090820316e-06,
      "step": 5509,
      "training_step_time": 8.573952436447144
    },
    {
      "epoch": 5.254745483398437e-06,
      "grad_norm": 0.42978572845458984,
      "learning_rate": 9.99929169900125e-06,
      "loss": 0.0817,
      "step": 5510
    },
    {
      "epoch": 5.254745483398437e-06,
      "model_forward_time": 1.7304694652557373,
      "step": 5510
    },
    {
      "dynamics_loss_avg": 0.06790878400206565,
      "epoch": 5.254745483398437e-06,
      "step": 5510
    },
    {
      "action_loss_avg": 0.008903093077242375,
      "epoch": 5.254745483398437e-06,
      "step": 5510
    },
    {
      "epoch": 5.254745483398437e-06,
      "step": 5510,
      "training_step_time": 9.009219884872437
    },
    {
      "epoch": 5.255699157714844e-06,
      "model_forward_time": 2.95624041557312,
      "step": 5511
    },
    {
      "epoch": 5.255699157714844e-06,
      "step": 5511,
      "training_step_time": 8.753825902938843
    },
    {
      "epoch": 5.25665283203125e-06,
      "model_forward_time": 1.726823091506958,
      "step": 5512
    },
    {
      "epoch": 5.25665283203125e-06,
      "step": 5512,
      "training_step_time": 8.974500179290771
    },
    {
      "epoch": 5.257606506347656e-06,
      "model_forward_time": 1.7541205883026123,
      "step": 5513
    },
    {
      "epoch": 5.257606506347656e-06,
      "step": 5513,
      "training_step_time": 7.586024522781372
    },
    {
      "epoch": 5.258560180664063e-06,
      "model_forward_time": 1.7436590194702148,
      "step": 5514
    },
    {
      "epoch": 5.258560180664063e-06,
      "step": 5514,
      "training_step_time": 7.565841913223267
    },
    {
      "epoch": 5.2595138549804684e-06,
      "model_forward_time": 1.7933149337768555,
      "step": 5515
    },
    {
      "epoch": 5.2595138549804684e-06,
      "step": 5515,
      "training_step_time": 7.54998517036438
    },
    {
      "epoch": 5.260467529296875e-06,
      "model_forward_time": 1.747267723083496,
      "step": 5516
    },
    {
      "epoch": 5.260467529296875e-06,
      "step": 5516,
      "training_step_time": 7.340675354003906
    },
    {
      "epoch": 5.261421203613282e-06,
      "model_forward_time": 1.7475502490997314,
      "step": 5517
    },
    {
      "epoch": 5.261421203613282e-06,
      "step": 5517,
      "training_step_time": 9.042241334915161
    },
    {
      "epoch": 5.262374877929687e-06,
      "model_forward_time": 1.754091739654541,
      "step": 5518
    },
    {
      "epoch": 5.262374877929687e-06,
      "step": 5518,
      "training_step_time": 7.5111706256866455
    },
    {
      "epoch": 5.263328552246094e-06,
      "model_forward_time": 1.746899127960205,
      "step": 5519
    },
    {
      "epoch": 5.263328552246094e-06,
      "step": 5519,
      "training_step_time": 9.018506050109863
    },
    {
      "epoch": 5.2642822265625e-06,
      "grad_norm": 0.24460263550281525,
      "learning_rate": 9.999263595220857e-06,
      "loss": 0.0752,
      "step": 5520
    },
    {
      "epoch": 5.2642822265625e-06,
      "model_forward_time": 2.9472286701202393,
      "step": 5520
    },
    {
      "dynamics_loss_avg": 0.06371672973036765,
      "epoch": 5.2642822265625e-06,
      "step": 5520
    },
    {
      "action_loss_avg": 0.006638486962765455,
      "epoch": 5.2642822265625e-06,
      "step": 5520
    },
    {
      "epoch": 5.2642822265625e-06,
      "step": 5520,
      "training_step_time": 7.723653316497803
    },
    {
      "epoch": 5.265235900878906e-06,
      "model_forward_time": 1.748488187789917,
      "step": 5521
    },
    {
      "epoch": 5.265235900878906e-06,
      "step": 5521,
      "training_step_time": 7.698676586151123
    },
    {
      "epoch": 5.266189575195313e-06,
      "model_forward_time": 1.7433686256408691,
      "step": 5522
    },
    {
      "epoch": 5.266189575195313e-06,
      "step": 5522,
      "training_step_time": 7.678277492523193
    },
    {
      "epoch": 5.2671432495117185e-06,
      "model_forward_time": 1.7395663261413574,
      "step": 5523
    },
    {
      "epoch": 5.2671432495117185e-06,
      "step": 5523,
      "training_step_time": 7.574426651000977
    },
    {
      "epoch": 5.268096923828125e-06,
      "model_forward_time": 1.7339012622833252,
      "step": 5524
    },
    {
      "epoch": 5.268096923828125e-06,
      "step": 5524,
      "training_step_time": 8.169573545455933
    },
    {
      "epoch": 5.269050598144532e-06,
      "model_forward_time": 1.7341465950012207,
      "step": 5525
    },
    {
      "epoch": 5.269050598144532e-06,
      "step": 5525,
      "training_step_time": 7.641247749328613
    },
    {
      "epoch": 5.270004272460937e-06,
      "model_forward_time": 1.738124132156372,
      "step": 5526
    },
    {
      "epoch": 5.270004272460937e-06,
      "step": 5526,
      "training_step_time": 7.557968854904175
    },
    {
      "epoch": 5.270957946777344e-06,
      "model_forward_time": 1.735964059829712,
      "step": 5527
    },
    {
      "epoch": 5.270957946777344e-06,
      "step": 5527,
      "training_step_time": 8.182826519012451
    },
    {
      "epoch": 5.27191162109375e-06,
      "model_forward_time": 1.7327508926391602,
      "step": 5528
    },
    {
      "epoch": 5.27191162109375e-06,
      "step": 5528,
      "training_step_time": 8.66532826423645
    },
    {
      "epoch": 5.272865295410156e-06,
      "model_forward_time": 2.9301629066467285,
      "step": 5529
    },
    {
      "epoch": 5.272865295410156e-06,
      "step": 5529,
      "training_step_time": 9.057218790054321
    },
    {
      "epoch": 5.273818969726563e-06,
      "grad_norm": 0.29018449783325195,
      "learning_rate": 9.999234944728511e-06,
      "loss": 0.0736,
      "step": 5530
    },
    {
      "epoch": 5.273818969726563e-06,
      "model_forward_time": 1.7372348308563232,
      "step": 5530
    },
    {
      "dynamics_loss_avg": 0.07378472909331321,
      "epoch": 5.273818969726563e-06,
      "step": 5530
    },
    {
      "action_loss_avg": 0.006483814143575728,
      "epoch": 5.273818969726563e-06,
      "step": 5530
    },
    {
      "epoch": 5.273818969726563e-06,
      "step": 5530,
      "training_step_time": 7.704147815704346
    },
    {
      "epoch": 5.2747726440429685e-06,
      "model_forward_time": 1.7341365814208984,
      "step": 5531
    },
    {
      "epoch": 5.2747726440429685e-06,
      "step": 5531,
      "training_step_time": 8.716365814208984
    },
    {
      "epoch": 5.275726318359375e-06,
      "model_forward_time": 1.7292940616607666,
      "step": 5532
    },
    {
      "epoch": 5.275726318359375e-06,
      "step": 5532,
      "training_step_time": 8.638367891311646
    },
    {
      "epoch": 5.276679992675782e-06,
      "model_forward_time": 1.735443353652954,
      "step": 5533
    },
    {
      "epoch": 5.276679992675782e-06,
      "step": 5533,
      "training_step_time": 7.6205549240112305
    },
    {
      "epoch": 5.277633666992187e-06,
      "model_forward_time": 1.7398614883422852,
      "step": 5534
    },
    {
      "epoch": 5.277633666992187e-06,
      "step": 5534,
      "training_step_time": 7.692251443862915
    },
    {
      "epoch": 5.278587341308594e-06,
      "model_forward_time": 1.743401288986206,
      "step": 5535
    },
    {
      "epoch": 5.278587341308594e-06,
      "step": 5535,
      "training_step_time": 7.561870574951172
    },
    {
      "epoch": 5.279541015625e-06,
      "model_forward_time": 1.7350618839263916,
      "step": 5536
    },
    {
      "epoch": 5.279541015625e-06,
      "step": 5536,
      "training_step_time": 7.552840709686279
    },
    {
      "epoch": 5.280494689941406e-06,
      "model_forward_time": 1.7728021144866943,
      "step": 5537
    },
    {
      "epoch": 5.280494689941406e-06,
      "step": 5537,
      "training_step_time": 9.967723369598389
    },
    {
      "epoch": 5.281448364257813e-06,
      "model_forward_time": 2.9219765663146973,
      "step": 5538
    },
    {
      "epoch": 5.281448364257813e-06,
      "step": 5538,
      "training_step_time": 8.729076147079468
    },
    {
      "epoch": 5.2824020385742185e-06,
      "model_forward_time": 1.8291940689086914,
      "step": 5539
    },
    {
      "epoch": 5.2824020385742185e-06,
      "step": 5539,
      "training_step_time": 7.701987028121948
    },
    {
      "epoch": 5.283355712890625e-06,
      "grad_norm": 0.21828624606132507,
      "learning_rate": 9.999205747527348e-06,
      "loss": 0.0808,
      "step": 5540
    },
    {
      "epoch": 5.283355712890625e-06,
      "model_forward_time": 1.7109229564666748,
      "step": 5540
    },
    {
      "dynamics_loss_avg": 0.09197787158191204,
      "epoch": 5.283355712890625e-06,
      "step": 5540
    },
    {
      "action_loss_avg": 0.013977239513769746,
      "epoch": 5.283355712890625e-06,
      "step": 5540
    },
    {
      "epoch": 5.283355712890625e-06,
      "step": 5540,
      "training_step_time": 7.750440359115601
    },
    {
      "epoch": 5.284309387207032e-06,
      "model_forward_time": 1.7367656230926514,
      "step": 5541
    },
    {
      "epoch": 5.284309387207032e-06,
      "step": 5541,
      "training_step_time": 7.605417728424072
    },
    {
      "epoch": 5.285263061523437e-06,
      "model_forward_time": 1.7260956764221191,
      "step": 5542
    },
    {
      "epoch": 5.285263061523437e-06,
      "step": 5542,
      "training_step_time": 7.547034502029419
    },
    {
      "epoch": 5.286216735839844e-06,
      "model_forward_time": 1.720289707183838,
      "step": 5543
    },
    {
      "epoch": 5.286216735839844e-06,
      "step": 5543,
      "training_step_time": 8.466799259185791
    },
    {
      "epoch": 5.28717041015625e-06,
      "model_forward_time": 1.7694194316864014,
      "step": 5544
    },
    {
      "epoch": 5.28717041015625e-06,
      "step": 5544,
      "training_step_time": 9.192538976669312
    },
    {
      "epoch": 5.288124084472656e-06,
      "model_forward_time": 1.7216339111328125,
      "step": 5545
    },
    {
      "epoch": 5.288124084472656e-06,
      "step": 5545,
      "training_step_time": 8.508146524429321
    },
    {
      "epoch": 5.289077758789063e-06,
      "model_forward_time": 1.7338736057281494,
      "step": 5546
    },
    {
      "epoch": 5.289077758789063e-06,
      "step": 5546,
      "training_step_time": 8.616663217544556
    },
    {
      "epoch": 5.2900314331054685e-06,
      "model_forward_time": 2.9015772342681885,
      "step": 5547
    },
    {
      "epoch": 5.2900314331054685e-06,
      "step": 5547,
      "training_step_time": 7.834690093994141
    },
    {
      "epoch": 5.290985107421875e-06,
      "model_forward_time": 1.776021957397461,
      "step": 5548
    },
    {
      "epoch": 5.290985107421875e-06,
      "step": 5548,
      "training_step_time": 9.782521963119507
    },
    {
      "epoch": 5.291938781738282e-06,
      "model_forward_time": 1.7239062786102295,
      "step": 5549
    },
    {
      "epoch": 5.291938781738282e-06,
      "step": 5549,
      "training_step_time": 7.6684043407440186
    },
    {
      "epoch": 5.292892456054687e-06,
      "grad_norm": 0.3208986818790436,
      "learning_rate": 9.999176003620563e-06,
      "loss": 0.0787,
      "step": 5550
    },
    {
      "epoch": 5.292892456054687e-06,
      "model_forward_time": 1.7289056777954102,
      "step": 5550
    },
    {
      "dynamics_loss_avg": 0.08326700925827027,
      "epoch": 5.292892456054687e-06,
      "step": 5550
    },
    {
      "action_loss_avg": 0.011125275306403637,
      "epoch": 5.292892456054687e-06,
      "step": 5550
    },
    {
      "epoch": 5.292892456054687e-06,
      "step": 5550,
      "training_step_time": 7.64764666557312
    },
    {
      "epoch": 5.293846130371094e-06,
      "model_forward_time": 1.7231028079986572,
      "step": 5551
    },
    {
      "epoch": 5.293846130371094e-06,
      "step": 5551,
      "training_step_time": 7.561199426651001
    },
    {
      "epoch": 5.2947998046875e-06,
      "model_forward_time": 1.7237460613250732,
      "step": 5552
    },
    {
      "epoch": 5.2947998046875e-06,
      "step": 5552,
      "training_step_time": 7.736846923828125
    },
    {
      "epoch": 5.295753479003906e-06,
      "model_forward_time": 1.7249445915222168,
      "step": 5553
    },
    {
      "epoch": 5.295753479003906e-06,
      "step": 5553,
      "training_step_time": 7.515865325927734
    },
    {
      "epoch": 5.296707153320313e-06,
      "model_forward_time": 1.7318849563598633,
      "step": 5554
    },
    {
      "epoch": 5.296707153320313e-06,
      "step": 5554,
      "training_step_time": 8.906638145446777
    },
    {
      "epoch": 5.2976608276367185e-06,
      "model_forward_time": 1.7228074073791504,
      "step": 5555
    },
    {
      "epoch": 5.2976608276367185e-06,
      "step": 5555,
      "training_step_time": 8.439311742782593
    },
    {
      "epoch": 5.298614501953125e-06,
      "model_forward_time": 2.9205644130706787,
      "step": 5556
    },
    {
      "epoch": 5.298614501953125e-06,
      "step": 5556,
      "training_step_time": 7.803919553756714
    },
    {
      "epoch": 5.299568176269532e-06,
      "model_forward_time": 1.7483434677124023,
      "step": 5557
    },
    {
      "epoch": 5.299568176269532e-06,
      "step": 5557,
      "training_step_time": 7.656544923782349
    },
    {
      "epoch": 5.300521850585937e-06,
      "model_forward_time": 1.7221028804779053,
      "step": 5558
    },
    {
      "epoch": 5.300521850585937e-06,
      "step": 5558,
      "training_step_time": 7.542025566101074
    },
    {
      "epoch": 5.301475524902344e-06,
      "model_forward_time": 1.7354578971862793,
      "step": 5559
    },
    {
      "epoch": 5.301475524902344e-06,
      "step": 5559,
      "training_step_time": 7.6851723194122314
    },
    {
      "epoch": 5.30242919921875e-06,
      "grad_norm": 0.37282684445381165,
      "learning_rate": 9.999145713011405e-06,
      "loss": 0.0729,
      "step": 5560
    },
    {
      "epoch": 5.30242919921875e-06,
      "model_forward_time": 1.7265825271606445,
      "step": 5560
    },
    {
      "dynamics_loss_avg": 0.06687270030379296,
      "epoch": 5.30242919921875e-06,
      "step": 5560
    },
    {
      "action_loss_avg": 0.008737165050115436,
      "epoch": 5.30242919921875e-06,
      "step": 5560
    },
    {
      "epoch": 5.30242919921875e-06,
      "step": 5560,
      "training_step_time": 7.5958497524261475
    },
    {
      "epoch": 5.303382873535156e-06,
      "model_forward_time": 1.7238380908966064,
      "step": 5561
    },
    {
      "epoch": 5.303382873535156e-06,
      "step": 5561,
      "training_step_time": 7.576761484146118
    },
    {
      "epoch": 5.304336547851563e-06,
      "model_forward_time": 1.737602710723877,
      "step": 5562
    },
    {
      "epoch": 5.304336547851563e-06,
      "step": 5562,
      "training_step_time": 7.464587688446045
    },
    {
      "epoch": 5.3052902221679685e-06,
      "model_forward_time": 1.7264437675476074,
      "step": 5563
    },
    {
      "epoch": 5.3052902221679685e-06,
      "step": 5563,
      "training_step_time": 8.617555141448975
    },
    {
      "epoch": 5.306243896484375e-06,
      "model_forward_time": 1.7249717712402344,
      "step": 5564
    },
    {
      "epoch": 5.306243896484375e-06,
      "step": 5564,
      "training_step_time": 8.15206003189087
    },
    {
      "epoch": 5.307197570800782e-06,
      "model_forward_time": 2.964017391204834,
      "step": 5565
    },
    {
      "epoch": 5.307197570800782e-06,
      "step": 5565,
      "training_step_time": 10.138503789901733
    },
    {
      "epoch": 5.308151245117187e-06,
      "model_forward_time": 1.750570297241211,
      "step": 5566
    },
    {
      "epoch": 5.308151245117187e-06,
      "step": 5566,
      "training_step_time": 7.634126663208008
    },
    {
      "epoch": 5.309104919433594e-06,
      "model_forward_time": 1.8505713939666748,
      "step": 5567
    },
    {
      "epoch": 5.309104919433594e-06,
      "step": 5567,
      "training_step_time": 7.598183631896973
    },
    {
      "epoch": 5.31005859375e-06,
      "model_forward_time": 1.7444937229156494,
      "step": 5568
    },
    {
      "epoch": 5.31005859375e-06,
      "step": 5568,
      "training_step_time": 7.624154329299927
    },
    {
      "epoch": 5.311012268066406e-06,
      "model_forward_time": 1.7438976764678955,
      "step": 5569
    },
    {
      "epoch": 5.311012268066406e-06,
      "step": 5569,
      "training_step_time": 7.574918746948242
    },
    {
      "epoch": 5.311965942382813e-06,
      "grad_norm": 0.27009788155555725,
      "learning_rate": 9.999114875703187e-06,
      "loss": 0.0753,
      "step": 5570
    },
    {
      "epoch": 5.311965942382813e-06,
      "model_forward_time": 1.741943120956421,
      "step": 5570
    },
    {
      "dynamics_loss_avg": 0.057853997126221655,
      "epoch": 5.311965942382813e-06,
      "step": 5570
    },
    {
      "action_loss_avg": 0.007440826040692627,
      "epoch": 5.311965942382813e-06,
      "step": 5570
    },
    {
      "epoch": 5.311965942382813e-06,
      "step": 5570,
      "training_step_time": 7.523906469345093
    },
    {
      "epoch": 5.3129196166992185e-06,
      "model_forward_time": 1.7418010234832764,
      "step": 5571
    },
    {
      "epoch": 5.3129196166992185e-06,
      "step": 5571,
      "training_step_time": 7.645601749420166
    },
    {
      "epoch": 5.313873291015625e-06,
      "model_forward_time": 1.744217872619629,
      "step": 5572
    },
    {
      "epoch": 5.313873291015625e-06,
      "step": 5572,
      "training_step_time": 8.284763097763062
    },
    {
      "epoch": 5.314826965332032e-06,
      "model_forward_time": 1.7442874908447266,
      "step": 5573
    },
    {
      "epoch": 5.314826965332032e-06,
      "step": 5573,
      "training_step_time": 8.543675661087036
    },
    {
      "epoch": 5.315780639648437e-06,
      "model_forward_time": 2.900268793106079,
      "step": 5574
    },
    {
      "epoch": 5.315780639648437e-06,
      "step": 5574,
      "training_step_time": 8.686316013336182
    },
    {
      "epoch": 5.316734313964844e-06,
      "model_forward_time": 1.7470016479492188,
      "step": 5575
    },
    {
      "epoch": 5.316734313964844e-06,
      "step": 5575,
      "training_step_time": 7.690901041030884
    },
    {
      "epoch": 5.31768798828125e-06,
      "model_forward_time": 1.8375399112701416,
      "step": 5576
    },
    {
      "epoch": 5.31768798828125e-06,
      "step": 5576,
      "training_step_time": 8.488889455795288
    },
    {
      "epoch": 5.318641662597656e-06,
      "model_forward_time": 1.7502174377441406,
      "step": 5577
    },
    {
      "epoch": 5.318641662597656e-06,
      "step": 5577,
      "training_step_time": 7.647278308868408
    },
    {
      "epoch": 5.319595336914063e-06,
      "model_forward_time": 1.7435901165008545,
      "step": 5578
    },
    {
      "epoch": 5.319595336914063e-06,
      "step": 5578,
      "training_step_time": 7.565060377120972
    },
    {
      "epoch": 5.3205490112304685e-06,
      "model_forward_time": 1.745110034942627,
      "step": 5579
    },
    {
      "epoch": 5.3205490112304685e-06,
      "step": 5579,
      "training_step_time": 7.539077997207642
    },
    {
      "epoch": 5.321502685546875e-06,
      "grad_norm": 0.2998458445072174,
      "learning_rate": 9.999083491699282e-06,
      "loss": 0.07,
      "step": 5580
    },
    {
      "epoch": 5.321502685546875e-06,
      "model_forward_time": 1.7354605197906494,
      "step": 5580
    },
    {
      "dynamics_loss_avg": 0.055295285396277906,
      "epoch": 5.321502685546875e-06,
      "step": 5580
    },
    {
      "action_loss_avg": 0.006494358391501009,
      "epoch": 5.321502685546875e-06,
      "step": 5580
    },
    {
      "epoch": 5.321502685546875e-06,
      "step": 5580,
      "training_step_time": 7.629276990890503
    },
    {
      "epoch": 5.322456359863282e-06,
      "model_forward_time": 1.747570276260376,
      "step": 5581
    },
    {
      "epoch": 5.322456359863282e-06,
      "step": 5581,
      "training_step_time": 7.5228118896484375
    },
    {
      "epoch": 5.323410034179687e-06,
      "model_forward_time": 1.7420625686645508,
      "step": 5582
    },
    {
      "epoch": 5.323410034179687e-06,
      "step": 5582,
      "training_step_time": 7.542390823364258
    },
    {
      "epoch": 5.324363708496094e-06,
      "model_forward_time": 2.8751513957977295,
      "step": 5583
    },
    {
      "epoch": 5.324363708496094e-06,
      "step": 5583,
      "training_step_time": 7.6614179611206055
    },
    {
      "epoch": 5.3253173828125e-06,
      "model_forward_time": 1.7511875629425049,
      "step": 5584
    },
    {
      "epoch": 5.3253173828125e-06,
      "step": 5584,
      "training_step_time": 7.645542860031128
    },
    {
      "epoch": 5.326271057128906e-06,
      "model_forward_time": 1.8027894496917725,
      "step": 5585
    },
    {
      "epoch": 5.326271057128906e-06,
      "step": 5585,
      "training_step_time": 8.6279776096344
    },
    {
      "epoch": 5.327224731445313e-06,
      "model_forward_time": 1.7494075298309326,
      "step": 5586
    },
    {
      "epoch": 5.327224731445313e-06,
      "step": 5586,
      "training_step_time": 7.639611721038818
    },
    {
      "epoch": 5.3281784057617185e-06,
      "model_forward_time": 1.745107650756836,
      "step": 5587
    },
    {
      "epoch": 5.3281784057617185e-06,
      "step": 5587,
      "training_step_time": 7.522565126419067
    },
    {
      "epoch": 5.329132080078125e-06,
      "model_forward_time": 1.7403550148010254,
      "step": 5588
    },
    {
      "epoch": 5.329132080078125e-06,
      "step": 5588,
      "training_step_time": 8.587490320205688
    },
    {
      "epoch": 5.330085754394532e-06,
      "model_forward_time": 1.7515161037445068,
      "step": 5589
    },
    {
      "epoch": 5.330085754394532e-06,
      "step": 5589,
      "training_step_time": 7.630998611450195
    },
    {
      "epoch": 5.331039428710937e-06,
      "grad_norm": 0.31500163674354553,
      "learning_rate": 9.999051561003124e-06,
      "loss": 0.0775,
      "step": 5590
    },
    {
      "epoch": 5.331039428710937e-06,
      "model_forward_time": 1.7437334060668945,
      "step": 5590
    },
    {
      "dynamics_loss_avg": 0.06337961368262768,
      "epoch": 5.331039428710937e-06,
      "step": 5590
    },
    {
      "action_loss_avg": 0.006705674598924816,
      "epoch": 5.331039428710937e-06,
      "step": 5590
    },
    {
      "epoch": 5.331039428710937e-06,
      "step": 5590,
      "training_step_time": 8.0700204372406
    },
    {
      "epoch": 5.331993103027344e-06,
      "model_forward_time": 1.7461776733398438,
      "step": 5591
    },
    {
      "epoch": 5.331993103027344e-06,
      "step": 5591,
      "training_step_time": 8.411093950271606
    },
    {
      "epoch": 5.33294677734375e-06,
      "model_forward_time": 2.8931288719177246,
      "step": 5592
    },
    {
      "epoch": 5.33294677734375e-06,
      "step": 5592,
      "training_step_time": 8.762491226196289
    },
    {
      "epoch": 5.333900451660156e-06,
      "model_forward_time": 1.7514970302581787,
      "step": 5593
    },
    {
      "epoch": 5.333900451660156e-06,
      "step": 5593,
      "training_step_time": 8.664684057235718
    },
    {
      "epoch": 5.334854125976563e-06,
      "model_forward_time": 1.7825076580047607,
      "step": 5594
    },
    {
      "epoch": 5.334854125976563e-06,
      "step": 5594,
      "training_step_time": 8.13498854637146
    },
    {
      "epoch": 5.3358078002929685e-06,
      "model_forward_time": 1.7188060283660889,
      "step": 5595
    },
    {
      "epoch": 5.3358078002929685e-06,
      "step": 5595,
      "training_step_time": 9.127087116241455
    },
    {
      "epoch": 5.336761474609375e-06,
      "model_forward_time": 1.7408826351165771,
      "step": 5596
    },
    {
      "epoch": 5.336761474609375e-06,
      "step": 5596,
      "training_step_time": 7.545818567276001
    },
    {
      "epoch": 5.337715148925782e-06,
      "model_forward_time": 1.7314772605895996,
      "step": 5597
    },
    {
      "epoch": 5.337715148925782e-06,
      "step": 5597,
      "training_step_time": 7.538300514221191
    },
    {
      "epoch": 5.338668823242187e-06,
      "model_forward_time": 1.725599765777588,
      "step": 5598
    },
    {
      "epoch": 5.338668823242187e-06,
      "step": 5598,
      "training_step_time": 7.633988380432129
    },
    {
      "epoch": 5.339622497558594e-06,
      "model_forward_time": 1.7420873641967773,
      "step": 5599
    },
    {
      "epoch": 5.339622497558594e-06,
      "step": 5599,
      "training_step_time": 7.524833679199219
    },
    {
      "epoch": 5.340576171875e-06,
      "grad_norm": 0.3114372789859772,
      "learning_rate": 9.999019083618202e-06,
      "loss": 0.0788,
      "step": 5600
    },
    {
      "epoch": 5.340576171875e-06,
      "model_forward_time": 1.7277553081512451,
      "step": 5600
    },
    {
      "dynamics_loss_avg": 0.055089615378528835,
      "epoch": 5.340576171875e-06,
      "step": 5600
    },
    {
      "action_loss_avg": 0.005280282045714557,
      "epoch": 5.340576171875e-06,
      "step": 5600
    },
    {
      "epoch": 5.340576171875e-06,
      "step": 5600,
      "training_step_time": 7.505022048950195
    },
    {
      "epoch": 5.341529846191406e-06,
      "model_forward_time": 2.8962154388427734,
      "step": 5601
    },
    {
      "epoch": 5.341529846191406e-06,
      "step": 5601,
      "training_step_time": 8.597073554992676
    },
    {
      "epoch": 5.342483520507813e-06,
      "model_forward_time": 1.744323968887329,
      "step": 5602
    },
    {
      "epoch": 5.342483520507813e-06,
      "step": 5602,
      "training_step_time": 9.351240634918213
    },
    {
      "epoch": 5.3434371948242185e-06,
      "model_forward_time": 1.733501672744751,
      "step": 5603
    },
    {
      "epoch": 5.3434371948242185e-06,
      "step": 5603,
      "training_step_time": 9.644506454467773
    },
    {
      "epoch": 5.344390869140625e-06,
      "model_forward_time": 1.7288103103637695,
      "step": 5604
    },
    {
      "epoch": 5.344390869140625e-06,
      "step": 5604,
      "training_step_time": 7.700472354888916
    },
    {
      "epoch": 5.345344543457032e-06,
      "model_forward_time": 1.3620953559875488,
      "step": 5605
    },
    {
      "epoch": 5.345344543457032e-06,
      "step": 5605,
      "training_step_time": 8.127609729766846
    },
    {
      "epoch": 5.346298217773437e-06,
      "model_forward_time": 1.7738962173461914,
      "step": 5606
    },
    {
      "epoch": 5.346298217773437e-06,
      "step": 5606,
      "training_step_time": 8.124590158462524
    },
    {
      "epoch": 5.347251892089844e-06,
      "model_forward_time": 1.7260589599609375,
      "step": 5607
    },
    {
      "epoch": 5.347251892089844e-06,
      "step": 5607,
      "training_step_time": 7.6287713050842285
    },
    {
      "epoch": 5.34820556640625e-06,
      "model_forward_time": 1.7342398166656494,
      "step": 5608
    },
    {
      "epoch": 5.34820556640625e-06,
      "step": 5608,
      "training_step_time": 7.4927873611450195
    },
    {
      "epoch": 5.349159240722656e-06,
      "model_forward_time": 1.738783359527588,
      "step": 5609
    },
    {
      "epoch": 5.349159240722656e-06,
      "step": 5609,
      "training_step_time": 7.49747371673584
    },
    {
      "epoch": 5.350112915039063e-06,
      "grad_norm": 0.31075772643089294,
      "learning_rate": 9.99898605954807e-06,
      "loss": 0.0754,
      "step": 5610
    },
    {
      "epoch": 5.350112915039063e-06,
      "model_forward_time": 2.906687021255493,
      "step": 5610
    },
    {
      "dynamics_loss_avg": 0.08246438056230546,
      "epoch": 5.350112915039063e-06,
      "step": 5610
    },
    {
      "action_loss_avg": 0.008502657385542988,
      "epoch": 5.350112915039063e-06,
      "step": 5610
    },
    {
      "epoch": 5.350112915039063e-06,
      "step": 5610,
      "training_step_time": 7.658169507980347
    },
    {
      "epoch": 5.3510665893554685e-06,
      "model_forward_time": 1.733705997467041,
      "step": 5611
    },
    {
      "epoch": 5.3510665893554685e-06,
      "step": 5611,
      "training_step_time": 8.555889368057251
    },
    {
      "epoch": 5.352020263671875e-06,
      "model_forward_time": 1.7331151962280273,
      "step": 5612
    },
    {
      "epoch": 5.352020263671875e-06,
      "step": 5612,
      "training_step_time": 7.63724946975708
    },
    {
      "epoch": 5.352973937988282e-06,
      "model_forward_time": 1.8339827060699463,
      "step": 5613
    },
    {
      "epoch": 5.352973937988282e-06,
      "step": 5613,
      "training_step_time": 8.663949012756348
    },
    {
      "epoch": 5.353927612304687e-06,
      "model_forward_time": 1.734632968902588,
      "step": 5614
    },
    {
      "epoch": 5.353927612304687e-06,
      "step": 5614,
      "training_step_time": 7.652310132980347
    },
    {
      "epoch": 5.354881286621094e-06,
      "model_forward_time": 1.753612995147705,
      "step": 5615
    },
    {
      "epoch": 5.354881286621094e-06,
      "step": 5615,
      "training_step_time": 7.66295862197876
    },
    {
      "epoch": 5.3558349609375e-06,
      "model_forward_time": 1.7290658950805664,
      "step": 5616
    },
    {
      "epoch": 5.3558349609375e-06,
      "step": 5616,
      "training_step_time": 7.614172697067261
    },
    {
      "epoch": 5.356788635253906e-06,
      "model_forward_time": 1.7300169467926025,
      "step": 5617
    },
    {
      "epoch": 5.356788635253906e-06,
      "step": 5617,
      "training_step_time": 7.665904521942139
    },
    {
      "epoch": 5.357742309570313e-06,
      "model_forward_time": 1.7311720848083496,
      "step": 5618
    },
    {
      "epoch": 5.357742309570313e-06,
      "step": 5618,
      "training_step_time": 8.47241497039795
    },
    {
      "epoch": 5.3586959838867185e-06,
      "model_forward_time": 2.879272699356079,
      "step": 5619
    },
    {
      "epoch": 5.3586959838867185e-06,
      "step": 5619,
      "training_step_time": 9.38984751701355
    },
    {
      "epoch": 5.359649658203125e-06,
      "grad_norm": 0.23580105602741241,
      "learning_rate": 9.998952488796338e-06,
      "loss": 0.0777,
      "step": 5620
    },
    {
      "epoch": 5.359649658203125e-06,
      "model_forward_time": 1.7386891841888428,
      "step": 5620
    },
    {
      "dynamics_loss_avg": 0.07935631945729256,
      "epoch": 5.359649658203125e-06,
      "step": 5620
    },
    {
      "action_loss_avg": 0.009850826510228217,
      "epoch": 5.359649658203125e-06,
      "step": 5620
    },
    {
      "epoch": 5.359649658203125e-06,
      "step": 5620,
      "training_step_time": 10.63846468925476
    },
    {
      "epoch": 5.360603332519532e-06,
      "model_forward_time": 1.7354943752288818,
      "step": 5621
    },
    {
      "epoch": 5.360603332519532e-06,
      "step": 5621,
      "training_step_time": 6.799111843109131
    },
    {
      "epoch": 5.361557006835937e-06,
      "model_forward_time": 1.7615752220153809,
      "step": 5622
    },
    {
      "epoch": 5.361557006835937e-06,
      "step": 5622,
      "training_step_time": 6.795746803283691
    },
    {
      "epoch": 5.362510681152344e-06,
      "model_forward_time": 1.7303147315979004,
      "step": 5623
    },
    {
      "epoch": 5.362510681152344e-06,
      "step": 5623,
      "training_step_time": 6.849312782287598
    },
    {
      "epoch": 5.36346435546875e-06,
      "model_forward_time": 1.7286105155944824,
      "step": 5624
    },
    {
      "epoch": 5.36346435546875e-06,
      "step": 5624,
      "training_step_time": 6.790209531784058
    },
    {
      "epoch": 5.364418029785156e-06,
      "model_forward_time": 1.733858585357666,
      "step": 5625
    },
    {
      "epoch": 5.364418029785156e-06,
      "step": 5625,
      "training_step_time": 6.8051981925964355
    },
    {
      "epoch": 5.365371704101563e-06,
      "model_forward_time": 1.7368102073669434,
      "step": 5626
    },
    {
      "epoch": 5.365371704101563e-06,
      "step": 5626,
      "training_step_time": 6.869690179824829
    },
    {
      "epoch": 5.3663253784179685e-06,
      "model_forward_time": 1.7344906330108643,
      "step": 5627
    },
    {
      "epoch": 5.3663253784179685e-06,
      "step": 5627,
      "training_step_time": 6.806541681289673
    },
    {
      "epoch": 5.367279052734375e-06,
      "model_forward_time": 1.7265098094940186,
      "step": 5628
    },
    {
      "epoch": 5.367279052734375e-06,
      "step": 5628,
      "training_step_time": 7.848891019821167
    },
    {
      "epoch": 5.368232727050781e-06,
      "model_forward_time": 1.7286858558654785,
      "step": 5629
    },
    {
      "epoch": 5.368232727050781e-06,
      "step": 5629,
      "training_step_time": 7.250964641571045
    },
    {
      "epoch": 5.369186401367187e-06,
      "grad_norm": 0.257431298494339,
      "learning_rate": 9.998918371366677e-06,
      "loss": 0.0757,
      "step": 5630
    },
    {
      "epoch": 5.369186401367187e-06,
      "model_forward_time": 1.7398316860198975,
      "step": 5630
    },
    {
      "dynamics_loss_avg": 0.06172054596245289,
      "epoch": 5.369186401367187e-06,
      "step": 5630
    },
    {
      "action_loss_avg": 0.007882288156542927,
      "epoch": 5.369186401367187e-06,
      "step": 5630
    },
    {
      "epoch": 5.369186401367187e-06,
      "step": 5630,
      "training_step_time": 7.7953782081604
    },
    {
      "epoch": 5.370140075683594e-06,
      "model_forward_time": 1.757869005203247,
      "step": 5631
    },
    {
      "epoch": 5.370140075683594e-06,
      "step": 5631,
      "training_step_time": 6.709014892578125
    },
    {
      "epoch": 5.37109375e-06,
      "model_forward_time": 1.7332587242126465,
      "step": 5632
    },
    {
      "epoch": 5.37109375e-06,
      "step": 5632,
      "training_step_time": 6.814020872116089
    },
    {
      "epoch": 5.372047424316406e-06,
      "model_forward_time": 1.7359027862548828,
      "step": 5633
    },
    {
      "epoch": 5.372047424316406e-06,
      "step": 5633,
      "training_step_time": 6.871710300445557
    },
    {
      "epoch": 5.373001098632813e-06,
      "model_forward_time": 1.731914758682251,
      "step": 5634
    },
    {
      "epoch": 5.373001098632813e-06,
      "step": 5634,
      "training_step_time": 6.7921106815338135
    },
    {
      "epoch": 5.3739547729492185e-06,
      "model_forward_time": 1.7362494468688965,
      "step": 5635
    },
    {
      "epoch": 5.3739547729492185e-06,
      "step": 5635,
      "training_step_time": 6.834812164306641
    },
    {
      "epoch": 5.374908447265625e-06,
      "model_forward_time": 1.74320387840271,
      "step": 5636
    },
    {
      "epoch": 5.374908447265625e-06,
      "step": 5636,
      "training_step_time": 6.907430648803711
    },
    {
      "epoch": 5.375862121582031e-06,
      "model_forward_time": 1.7314207553863525,
      "step": 5637
    },
    {
      "epoch": 5.375862121582031e-06,
      "step": 5637,
      "training_step_time": 6.807974100112915
    },
    {
      "epoch": 5.376815795898437e-06,
      "model_forward_time": 2.8480167388916016,
      "step": 5638
    },
    {
      "epoch": 5.376815795898437e-06,
      "step": 5638,
      "training_step_time": 7.7020392417907715
    },
    {
      "epoch": 5.377769470214844e-06,
      "model_forward_time": 1.7320265769958496,
      "step": 5639
    },
    {
      "epoch": 5.377769470214844e-06,
      "step": 5639,
      "training_step_time": 7.78308367729187
    },
    {
      "epoch": 5.37872314453125e-06,
      "grad_norm": 0.26782476902008057,
      "learning_rate": 9.998883707262821e-06,
      "loss": 0.0717,
      "step": 5640
    },
    {
      "epoch": 5.37872314453125e-06,
      "model_forward_time": 1.7359347343444824,
      "step": 5640
    },
    {
      "dynamics_loss_avg": 0.0641085272654891,
      "epoch": 5.37872314453125e-06,
      "step": 5640
    },
    {
      "action_loss_avg": 0.009830715809948742,
      "epoch": 5.37872314453125e-06,
      "step": 5640
    },
    {
      "epoch": 5.37872314453125e-06,
      "step": 5640,
      "training_step_time": 6.830446004867554
    },
    {
      "epoch": 5.379676818847656e-06,
      "model_forward_time": 1.7271699905395508,
      "step": 5641
    },
    {
      "epoch": 5.379676818847656e-06,
      "step": 5641,
      "training_step_time": 6.796442985534668
    },
    {
      "epoch": 5.380630493164063e-06,
      "model_forward_time": 1.7343361377716064,
      "step": 5642
    },
    {
      "epoch": 5.380630493164063e-06,
      "step": 5642,
      "training_step_time": 6.7859206199646
    },
    {
      "epoch": 5.3815841674804686e-06,
      "model_forward_time": 1.728713035583496,
      "step": 5643
    },
    {
      "epoch": 5.3815841674804686e-06,
      "step": 5643,
      "training_step_time": 6.868027210235596
    },
    {
      "epoch": 5.382537841796875e-06,
      "model_forward_time": 1.7293403148651123,
      "step": 5644
    },
    {
      "epoch": 5.382537841796875e-06,
      "step": 5644,
      "training_step_time": 6.761224746704102
    },
    {
      "epoch": 5.383491516113281e-06,
      "model_forward_time": 1.7402207851409912,
      "step": 5645
    },
    {
      "epoch": 5.383491516113281e-06,
      "step": 5645,
      "training_step_time": 6.810192108154297
    },
    {
      "epoch": 5.384445190429687e-06,
      "model_forward_time": 1.7386741638183594,
      "step": 5646
    },
    {
      "epoch": 5.384445190429687e-06,
      "step": 5646,
      "training_step_time": 6.749428749084473
    },
    {
      "epoch": 5.385398864746094e-06,
      "model_forward_time": 2.8572187423706055,
      "step": 5647
    },
    {
      "epoch": 5.385398864746094e-06,
      "step": 5647,
      "training_step_time": 7.726399660110474
    },
    {
      "epoch": 5.3863525390625e-06,
      "model_forward_time": 1.7352333068847656,
      "step": 5648
    },
    {
      "epoch": 5.3863525390625e-06,
      "step": 5648,
      "training_step_time": 9.695201635360718
    },
    {
      "epoch": 5.387306213378906e-06,
      "model_forward_time": 1.731203317642212,
      "step": 5649
    },
    {
      "epoch": 5.387306213378906e-06,
      "step": 5649,
      "training_step_time": 7.499484300613403
    },
    {
      "epoch": 5.388259887695313e-06,
      "grad_norm": 0.36479151248931885,
      "learning_rate": 9.998848496488556e-06,
      "loss": 0.0727,
      "step": 5650
    },
    {
      "epoch": 5.388259887695313e-06,
      "model_forward_time": 1.728989839553833,
      "step": 5650
    },
    {
      "dynamics_loss_avg": 0.0661373419687152,
      "epoch": 5.388259887695313e-06,
      "step": 5650
    },
    {
      "action_loss_avg": 0.00791807733476162,
      "epoch": 5.388259887695313e-06,
      "step": 5650
    },
    {
      "epoch": 5.388259887695313e-06,
      "step": 5650,
      "training_step_time": 6.788781404495239
    },
    {
      "epoch": 5.3892135620117186e-06,
      "model_forward_time": 1.72979736328125,
      "step": 5651
    },
    {
      "epoch": 5.3892135620117186e-06,
      "step": 5651,
      "training_step_time": 6.810488224029541
    },
    {
      "epoch": 5.390167236328125e-06,
      "model_forward_time": 1.7255830764770508,
      "step": 5652
    },
    {
      "epoch": 5.390167236328125e-06,
      "step": 5652,
      "training_step_time": 6.797224998474121
    },
    {
      "epoch": 5.391120910644531e-06,
      "model_forward_time": 1.7394979000091553,
      "step": 5653
    },
    {
      "epoch": 5.391120910644531e-06,
      "step": 5653,
      "training_step_time": 6.779360055923462
    },
    {
      "epoch": 5.392074584960937e-06,
      "model_forward_time": 1.7404570579528809,
      "step": 5654
    },
    {
      "epoch": 5.392074584960937e-06,
      "step": 5654,
      "training_step_time": 6.835885047912598
    },
    {
      "epoch": 5.393028259277344e-06,
      "model_forward_time": 1.7283782958984375,
      "step": 5655
    },
    {
      "epoch": 5.393028259277344e-06,
      "step": 5655,
      "training_step_time": 6.81651759147644
    },
    {
      "epoch": 5.39398193359375e-06,
      "model_forward_time": 2.8836915493011475,
      "step": 5656
    },
    {
      "epoch": 5.39398193359375e-06,
      "step": 5656,
      "training_step_time": 7.7934887409210205
    },
    {
      "epoch": 5.394935607910156e-06,
      "model_forward_time": 1.7437825202941895,
      "step": 5657
    },
    {
      "epoch": 5.394935607910156e-06,
      "step": 5657,
      "training_step_time": 8.753996133804321
    },
    {
      "epoch": 5.395889282226563e-06,
      "model_forward_time": 1.7339105606079102,
      "step": 5658
    },
    {
      "epoch": 5.395889282226563e-06,
      "step": 5658,
      "training_step_time": 7.669546127319336
    },
    {
      "epoch": 5.3968429565429686e-06,
      "model_forward_time": 1.7283196449279785,
      "step": 5659
    },
    {
      "epoch": 5.3968429565429686e-06,
      "step": 5659,
      "training_step_time": 6.7636613845825195
    },
    {
      "epoch": 5.397796630859375e-06,
      "grad_norm": 0.2892424166202545,
      "learning_rate": 9.998812739047737e-06,
      "loss": 0.0706,
      "step": 5660
    },
    {
      "epoch": 5.397796630859375e-06,
      "model_forward_time": 1.741440773010254,
      "step": 5660
    },
    {
      "dynamics_loss_avg": 0.061769567616283896,
      "epoch": 5.397796630859375e-06,
      "step": 5660
    },
    {
      "action_loss_avg": 0.0063337406725622715,
      "epoch": 5.397796630859375e-06,
      "step": 5660
    },
    {
      "epoch": 5.397796630859375e-06,
      "step": 5660,
      "training_step_time": 6.798042058944702
    },
    {
      "epoch": 5.398750305175781e-06,
      "model_forward_time": 1.7401654720306396,
      "step": 5661
    },
    {
      "epoch": 5.398750305175781e-06,
      "step": 5661,
      "training_step_time": 6.792709827423096
    },
    {
      "epoch": 5.3997039794921874e-06,
      "model_forward_time": 1.7628214359283447,
      "step": 5662
    },
    {
      "epoch": 5.3997039794921874e-06,
      "step": 5662,
      "training_step_time": 6.770981788635254
    },
    {
      "epoch": 5.400657653808594e-06,
      "model_forward_time": 1.7315869331359863,
      "step": 5663
    },
    {
      "epoch": 5.400657653808594e-06,
      "step": 5663,
      "training_step_time": 6.823902606964111
    },
    {
      "epoch": 5.401611328125e-06,
      "model_forward_time": 1.7305076122283936,
      "step": 5664
    },
    {
      "epoch": 5.401611328125e-06,
      "step": 5664,
      "training_step_time": 6.76511549949646
    },
    {
      "epoch": 5.402565002441406e-06,
      "model_forward_time": 2.954932928085327,
      "step": 5665
    },
    {
      "epoch": 5.402565002441406e-06,
      "step": 5665,
      "training_step_time": 8.634471654891968
    },
    {
      "epoch": 5.403518676757813e-06,
      "model_forward_time": 1.732151746749878,
      "step": 5666
    },
    {
      "epoch": 5.403518676757813e-06,
      "step": 5666,
      "training_step_time": 8.779541730880737
    },
    {
      "epoch": 5.4044723510742186e-06,
      "model_forward_time": 1.7306315898895264,
      "step": 5667
    },
    {
      "epoch": 5.4044723510742186e-06,
      "step": 5667,
      "training_step_time": 7.857585430145264
    },
    {
      "epoch": 5.405426025390625e-06,
      "model_forward_time": 1.7434992790222168,
      "step": 5668
    },
    {
      "epoch": 5.405426025390625e-06,
      "step": 5668,
      "training_step_time": 6.790924549102783
    },
    {
      "epoch": 5.406379699707031e-06,
      "model_forward_time": 1.7505254745483398,
      "step": 5669
    },
    {
      "epoch": 5.406379699707031e-06,
      "step": 5669,
      "training_step_time": 6.800294399261475
    },
    {
      "epoch": 5.4073333740234374e-06,
      "grad_norm": 0.32697659730911255,
      "learning_rate": 9.99877643494427e-06,
      "loss": 0.0708,
      "step": 5670
    },
    {
      "epoch": 5.4073333740234374e-06,
      "model_forward_time": 1.8089971542358398,
      "step": 5670
    },
    {
      "dynamics_loss_avg": 0.07274168971925973,
      "epoch": 5.4073333740234374e-06,
      "step": 5670
    },
    {
      "action_loss_avg": 0.006241933582350612,
      "epoch": 5.4073333740234374e-06,
      "step": 5670
    },
    {
      "epoch": 5.4073333740234374e-06,
      "step": 5670,
      "training_step_time": 6.792331218719482
    },
    {
      "epoch": 5.408287048339844e-06,
      "model_forward_time": 1.7328567504882812,
      "step": 5671
    },
    {
      "epoch": 5.408287048339844e-06,
      "step": 5671,
      "training_step_time": 6.772626161575317
    },
    {
      "epoch": 5.40924072265625e-06,
      "model_forward_time": 1.7361161708831787,
      "step": 5672
    },
    {
      "epoch": 5.40924072265625e-06,
      "step": 5672,
      "training_step_time": 6.727397203445435
    },
    {
      "epoch": 5.410194396972656e-06,
      "model_forward_time": 1.7373270988464355,
      "step": 5673
    },
    {
      "epoch": 5.410194396972656e-06,
      "step": 5673,
      "training_step_time": 6.7451231479644775
    },
    {
      "epoch": 5.411148071289063e-06,
      "model_forward_time": 2.867581844329834,
      "step": 5674
    },
    {
      "epoch": 5.411148071289063e-06,
      "step": 5674,
      "training_step_time": 7.68433403968811
    },
    {
      "epoch": 5.412101745605469e-06,
      "model_forward_time": 1.71024751663208,
      "step": 5675
    },
    {
      "epoch": 5.412101745605469e-06,
      "step": 5675,
      "training_step_time": 9.573357820510864
    },
    {
      "epoch": 5.413055419921875e-06,
      "model_forward_time": 1.7796871662139893,
      "step": 5676
    },
    {
      "epoch": 5.413055419921875e-06,
      "step": 5676,
      "training_step_time": 7.844663381576538
    },
    {
      "epoch": 5.414009094238281e-06,
      "model_forward_time": 1.737671136856079,
      "step": 5677
    },
    {
      "epoch": 5.414009094238281e-06,
      "step": 5677,
      "training_step_time": 6.695044279098511
    },
    {
      "epoch": 5.4149627685546874e-06,
      "model_forward_time": 1.7238476276397705,
      "step": 5678
    },
    {
      "epoch": 5.4149627685546874e-06,
      "step": 5678,
      "training_step_time": 6.723005771636963
    },
    {
      "epoch": 5.415916442871094e-06,
      "model_forward_time": 1.7327930927276611,
      "step": 5679
    },
    {
      "epoch": 5.415916442871094e-06,
      "step": 5679,
      "training_step_time": 6.693347692489624
    },
    {
      "epoch": 5.4168701171875e-06,
      "grad_norm": 0.26601117849349976,
      "learning_rate": 9.998739584182128e-06,
      "loss": 0.0694,
      "step": 5680
    },
    {
      "epoch": 5.4168701171875e-06,
      "model_forward_time": 1.7355432510375977,
      "step": 5680
    },
    {
      "dynamics_loss_avg": 0.06689053131267428,
      "epoch": 5.4168701171875e-06,
      "step": 5680
    },
    {
      "action_loss_avg": 0.007300976640544831,
      "epoch": 5.4168701171875e-06,
      "step": 5680
    },
    {
      "epoch": 5.4168701171875e-06,
      "step": 5680,
      "training_step_time": 6.701006889343262
    },
    {
      "epoch": 5.417823791503906e-06,
      "model_forward_time": 1.750483751296997,
      "step": 5681
    },
    {
      "epoch": 5.417823791503906e-06,
      "step": 5681,
      "training_step_time": 6.709695100784302
    },
    {
      "epoch": 5.418777465820313e-06,
      "model_forward_time": 1.7414381504058838,
      "step": 5682
    },
    {
      "epoch": 5.418777465820313e-06,
      "step": 5682,
      "training_step_time": 6.6790382862091064
    },
    {
      "epoch": 5.419731140136719e-06,
      "model_forward_time": 1.7565946578979492,
      "step": 5683
    },
    {
      "epoch": 5.419731140136719e-06,
      "step": 5683,
      "training_step_time": 8.656909227371216
    },
    {
      "epoch": 5.420684814453125e-06,
      "model_forward_time": 1.7276198863983154,
      "step": 5684
    },
    {
      "epoch": 5.420684814453125e-06,
      "step": 5684,
      "training_step_time": 8.653468370437622
    },
    {
      "epoch": 5.421638488769531e-06,
      "model_forward_time": 1.7345728874206543,
      "step": 5685
    },
    {
      "epoch": 5.421638488769531e-06,
      "step": 5685,
      "training_step_time": 8.701247930526733
    },
    {
      "epoch": 5.4225921630859374e-06,
      "model_forward_time": 1.7264947891235352,
      "step": 5686
    },
    {
      "epoch": 5.4225921630859374e-06,
      "step": 5686,
      "training_step_time": 7.812938690185547
    },
    {
      "epoch": 5.423545837402344e-06,
      "model_forward_time": 1.7234749794006348,
      "step": 5687
    },
    {
      "epoch": 5.423545837402344e-06,
      "step": 5687,
      "training_step_time": 6.700506210327148
    },
    {
      "epoch": 5.42449951171875e-06,
      "model_forward_time": 1.736746072769165,
      "step": 5688
    },
    {
      "epoch": 5.42449951171875e-06,
      "step": 5688,
      "training_step_time": 6.685062408447266
    },
    {
      "epoch": 5.425453186035156e-06,
      "model_forward_time": 1.7349073886871338,
      "step": 5689
    },
    {
      "epoch": 5.425453186035156e-06,
      "step": 5689,
      "training_step_time": 6.667120933532715
    },
    {
      "epoch": 5.426406860351563e-06,
      "grad_norm": 0.3230317533016205,
      "learning_rate": 9.998702186765342e-06,
      "loss": 0.077,
      "step": 5690
    },
    {
      "epoch": 5.426406860351563e-06,
      "model_forward_time": 1.7231919765472412,
      "step": 5690
    },
    {
      "dynamics_loss_avg": 0.05870835930109024,
      "epoch": 5.426406860351563e-06,
      "step": 5690
    },
    {
      "action_loss_avg": 0.007727855490520597,
      "epoch": 5.426406860351563e-06,
      "step": 5690
    },
    {
      "epoch": 5.426406860351563e-06,
      "step": 5690,
      "training_step_time": 6.717004299163818
    },
    {
      "epoch": 5.427360534667969e-06,
      "model_forward_time": 1.7250761985778809,
      "step": 5691
    },
    {
      "epoch": 5.427360534667969e-06,
      "step": 5691,
      "training_step_time": 6.657999038696289
    },
    {
      "epoch": 5.428314208984375e-06,
      "model_forward_time": 1.7296230792999268,
      "step": 5692
    },
    {
      "epoch": 5.428314208984375e-06,
      "step": 5692,
      "training_step_time": 7.5439677238464355
    },
    {
      "epoch": 5.429267883300781e-06,
      "model_forward_time": 2.8590199947357178,
      "step": 5693
    },
    {
      "epoch": 5.429267883300781e-06,
      "step": 5693,
      "training_step_time": 8.975141763687134
    },
    {
      "epoch": 5.4302215576171875e-06,
      "model_forward_time": 1.735710620880127,
      "step": 5694
    },
    {
      "epoch": 5.4302215576171875e-06,
      "step": 5694,
      "training_step_time": 7.831490516662598
    },
    {
      "epoch": 5.431175231933594e-06,
      "model_forward_time": 1.7492170333862305,
      "step": 5695
    },
    {
      "epoch": 5.431175231933594e-06,
      "step": 5695,
      "training_step_time": 7.640177488327026
    },
    {
      "epoch": 5.43212890625e-06,
      "model_forward_time": 1.7370271682739258,
      "step": 5696
    },
    {
      "epoch": 5.43212890625e-06,
      "step": 5696,
      "training_step_time": 6.717025279998779
    },
    {
      "epoch": 5.433082580566406e-06,
      "model_forward_time": 1.7299888134002686,
      "step": 5697
    },
    {
      "epoch": 5.433082580566406e-06,
      "step": 5697,
      "training_step_time": 6.719522476196289
    },
    {
      "epoch": 5.434036254882813e-06,
      "model_forward_time": 1.7356963157653809,
      "step": 5698
    },
    {
      "epoch": 5.434036254882813e-06,
      "step": 5698,
      "training_step_time": 6.72266697883606
    },
    {
      "epoch": 5.434989929199219e-06,
      "model_forward_time": 1.731447458267212,
      "step": 5699
    },
    {
      "epoch": 5.434989929199219e-06,
      "step": 5699,
      "training_step_time": 6.660107851028442
    },
    {
      "epoch": 5.435943603515625e-06,
      "grad_norm": 0.43787604570388794,
      "learning_rate": 9.998664242698e-06,
      "loss": 0.0807,
      "step": 5700
    },
    {
      "epoch": 5.435943603515625e-06,
      "model_forward_time": 1.7281792163848877,
      "step": 5700
    },
    {
      "dynamics_loss_avg": 0.06284463219344616,
      "epoch": 5.435943603515625e-06,
      "step": 5700
    },
    {
      "action_loss_avg": 0.006077431060839444,
      "epoch": 5.435943603515625e-06,
      "step": 5700
    },
    {
      "epoch": 5.435943603515625e-06,
      "step": 5700,
      "training_step_time": 6.723986387252808
    },
    {
      "epoch": 5.436897277832031e-06,
      "model_forward_time": 1.736915111541748,
      "step": 5701
    },
    {
      "epoch": 5.436897277832031e-06,
      "step": 5701,
      "training_step_time": 7.587650299072266
    },
    {
      "epoch": 5.4378509521484375e-06,
      "model_forward_time": 2.8468871116638184,
      "step": 5702
    },
    {
      "epoch": 5.4378509521484375e-06,
      "step": 5702,
      "training_step_time": 7.5484631061553955
    },
    {
      "epoch": 5.438804626464844e-06,
      "model_forward_time": 1.727853775024414,
      "step": 5703
    },
    {
      "epoch": 5.438804626464844e-06,
      "step": 5703,
      "training_step_time": 9.539505004882812
    },
    {
      "epoch": 5.43975830078125e-06,
      "model_forward_time": 1.7353172302246094,
      "step": 5704
    },
    {
      "epoch": 5.43975830078125e-06,
      "step": 5704,
      "training_step_time": 7.693731307983398
    },
    {
      "epoch": 5.440711975097656e-06,
      "model_forward_time": 1.7195262908935547,
      "step": 5705
    },
    {
      "epoch": 5.440711975097656e-06,
      "step": 5705,
      "training_step_time": 6.731602668762207
    },
    {
      "epoch": 5.441665649414063e-06,
      "model_forward_time": 1.7278661727905273,
      "step": 5706
    },
    {
      "epoch": 5.441665649414063e-06,
      "step": 5706,
      "training_step_time": 6.735844612121582
    },
    {
      "epoch": 5.442619323730469e-06,
      "model_forward_time": 1.7485766410827637,
      "step": 5707
    },
    {
      "epoch": 5.442619323730469e-06,
      "step": 5707,
      "training_step_time": 6.702874422073364
    },
    {
      "epoch": 5.443572998046875e-06,
      "model_forward_time": 1.7313172817230225,
      "step": 5708
    },
    {
      "epoch": 5.443572998046875e-06,
      "step": 5708,
      "training_step_time": 6.7302446365356445
    },
    {
      "epoch": 5.444526672363281e-06,
      "model_forward_time": 1.7639715671539307,
      "step": 5709
    },
    {
      "epoch": 5.444526672363281e-06,
      "step": 5709,
      "training_step_time": 6.715816259384155
    },
    {
      "epoch": 5.4454803466796875e-06,
      "grad_norm": 0.3210074305534363,
      "learning_rate": 9.998625751984252e-06,
      "loss": 0.0799,
      "step": 5710
    },
    {
      "epoch": 5.4454803466796875e-06,
      "model_forward_time": 1.7247307300567627,
      "step": 5710
    },
    {
      "dynamics_loss_avg": 0.05457975082099438,
      "epoch": 5.4454803466796875e-06,
      "step": 5710
    },
    {
      "action_loss_avg": 0.005487447208724916,
      "epoch": 5.4454803466796875e-06,
      "step": 5710
    },
    {
      "epoch": 5.4454803466796875e-06,
      "step": 5710,
      "training_step_time": 7.596562623977661
    },
    {
      "epoch": 5.446434020996094e-06,
      "model_forward_time": 2.9450905323028564,
      "step": 5711
    },
    {
      "epoch": 5.446434020996094e-06,
      "step": 5711,
      "training_step_time": 7.69034481048584
    },
    {
      "epoch": 5.4473876953125e-06,
      "model_forward_time": 1.7398197650909424,
      "step": 5712
    },
    {
      "epoch": 5.4473876953125e-06,
      "step": 5712,
      "training_step_time": 10.397636890411377
    },
    {
      "epoch": 5.448341369628906e-06,
      "model_forward_time": 1.7395892143249512,
      "step": 5713
    },
    {
      "epoch": 5.448341369628906e-06,
      "step": 5713,
      "training_step_time": 7.754159450531006
    },
    {
      "epoch": 5.449295043945313e-06,
      "model_forward_time": 1.7555456161499023,
      "step": 5714
    },
    {
      "epoch": 5.449295043945313e-06,
      "step": 5714,
      "training_step_time": 6.6924426555633545
    },
    {
      "epoch": 5.450248718261719e-06,
      "model_forward_time": 1.7337665557861328,
      "step": 5715
    },
    {
      "epoch": 5.450248718261719e-06,
      "step": 5715,
      "training_step_time": 6.734111785888672
    },
    {
      "epoch": 5.451202392578125e-06,
      "model_forward_time": 1.727543830871582,
      "step": 5716
    },
    {
      "epoch": 5.451202392578125e-06,
      "step": 5716,
      "training_step_time": 6.739394903182983
    },
    {
      "epoch": 5.452156066894531e-06,
      "model_forward_time": 1.7156226634979248,
      "step": 5717
    },
    {
      "epoch": 5.452156066894531e-06,
      "step": 5717,
      "training_step_time": 6.78910756111145
    },
    {
      "epoch": 5.4531097412109375e-06,
      "model_forward_time": 1.7384881973266602,
      "step": 5718
    },
    {
      "epoch": 5.4531097412109375e-06,
      "step": 5718,
      "training_step_time": 6.713891267776489
    },
    {
      "epoch": 5.454063415527344e-06,
      "model_forward_time": 1.743610143661499,
      "step": 5719
    },
    {
      "epoch": 5.454063415527344e-06,
      "step": 5719,
      "training_step_time": 7.603456020355225
    },
    {
      "epoch": 5.45501708984375e-06,
      "grad_norm": 0.3787718117237091,
      "learning_rate": 9.998586714628307e-06,
      "loss": 0.0802,
      "step": 5720
    },
    {
      "epoch": 5.45501708984375e-06,
      "model_forward_time": 2.8710556030273438,
      "step": 5720
    },
    {
      "dynamics_loss_avg": 0.06570151634514332,
      "epoch": 5.45501708984375e-06,
      "step": 5720
    },
    {
      "action_loss_avg": 0.008537333342246711,
      "epoch": 5.45501708984375e-06,
      "step": 5720
    },
    {
      "epoch": 5.45501708984375e-06,
      "step": 5720,
      "training_step_time": 7.604645729064941
    },
    {
      "epoch": 5.455970764160156e-06,
      "model_forward_time": 1.7305023670196533,
      "step": 5721
    },
    {
      "epoch": 5.455970764160156e-06,
      "step": 5721,
      "training_step_time": 11.338366031646729
    },
    {
      "epoch": 5.456924438476563e-06,
      "model_forward_time": 1.7344698905944824,
      "step": 5722
    },
    {
      "epoch": 5.456924438476563e-06,
      "step": 5722,
      "training_step_time": 8.562748193740845
    },
    {
      "epoch": 5.457878112792969e-06,
      "model_forward_time": 1.7437925338745117,
      "step": 5723
    },
    {
      "epoch": 5.457878112792969e-06,
      "step": 5723,
      "training_step_time": 7.658407688140869
    },
    {
      "epoch": 5.458831787109375e-06,
      "model_forward_time": 1.7375116348266602,
      "step": 5724
    },
    {
      "epoch": 5.458831787109375e-06,
      "step": 5724,
      "training_step_time": 6.665257453918457
    },
    {
      "epoch": 5.459785461425781e-06,
      "model_forward_time": 1.7320868968963623,
      "step": 5725
    },
    {
      "epoch": 5.459785461425781e-06,
      "step": 5725,
      "training_step_time": 6.684779644012451
    },
    {
      "epoch": 5.4607391357421875e-06,
      "model_forward_time": 1.7335608005523682,
      "step": 5726
    },
    {
      "epoch": 5.4607391357421875e-06,
      "step": 5726,
      "training_step_time": 6.749269247055054
    },
    {
      "epoch": 5.461692810058594e-06,
      "model_forward_time": 1.7394254207611084,
      "step": 5727
    },
    {
      "epoch": 5.461692810058594e-06,
      "step": 5727,
      "training_step_time": 6.692558526992798
    },
    {
      "epoch": 5.462646484375e-06,
      "model_forward_time": 1.7467575073242188,
      "step": 5728
    },
    {
      "epoch": 5.462646484375e-06,
      "step": 5728,
      "training_step_time": 7.633764982223511
    },
    {
      "epoch": 5.463600158691406e-06,
      "model_forward_time": 2.863696336746216,
      "step": 5729
    },
    {
      "epoch": 5.463600158691406e-06,
      "step": 5729,
      "training_step_time": 7.608543634414673
    },
    {
      "epoch": 5.464553833007813e-06,
      "grad_norm": 0.32262179255485535,
      "learning_rate": 9.998547130634433e-06,
      "loss": 0.0757,
      "step": 5730
    },
    {
      "epoch": 5.464553833007813e-06,
      "model_forward_time": 1.7327916622161865,
      "step": 5730
    },
    {
      "dynamics_loss_avg": 0.07736526001244784,
      "epoch": 5.464553833007813e-06,
      "step": 5730
    },
    {
      "action_loss_avg": 0.009705076937098055,
      "epoch": 5.464553833007813e-06,
      "step": 5730
    },
    {
      "epoch": 5.464553833007813e-06,
      "step": 5730,
      "training_step_time": 8.486757278442383
    },
    {
      "epoch": 5.465507507324219e-06,
      "model_forward_time": 1.3661165237426758,
      "step": 5731
    },
    {
      "epoch": 5.465507507324219e-06,
      "step": 5731,
      "training_step_time": 8.685154676437378
    },
    {
      "epoch": 5.466461181640625e-06,
      "model_forward_time": 1.7349047660827637,
      "step": 5732
    },
    {
      "epoch": 5.466461181640625e-06,
      "step": 5732,
      "training_step_time": 7.744767904281616
    },
    {
      "epoch": 5.467414855957031e-06,
      "model_forward_time": 1.3455157279968262,
      "step": 5733
    },
    {
      "epoch": 5.467414855957031e-06,
      "step": 5733,
      "training_step_time": 6.720203876495361
    },
    {
      "epoch": 5.4683685302734375e-06,
      "model_forward_time": 1.7368042469024658,
      "step": 5734
    },
    {
      "epoch": 5.4683685302734375e-06,
      "step": 5734,
      "training_step_time": 6.769979000091553
    },
    {
      "epoch": 5.469322204589844e-06,
      "model_forward_time": 1.7259013652801514,
      "step": 5735
    },
    {
      "epoch": 5.469322204589844e-06,
      "step": 5735,
      "training_step_time": 6.719554424285889
    },
    {
      "epoch": 5.47027587890625e-06,
      "model_forward_time": 1.7330718040466309,
      "step": 5736
    },
    {
      "epoch": 5.47027587890625e-06,
      "step": 5736,
      "training_step_time": 6.780763387680054
    },
    {
      "epoch": 5.471229553222656e-06,
      "model_forward_time": 1.725947618484497,
      "step": 5737
    },
    {
      "epoch": 5.471229553222656e-06,
      "step": 5737,
      "training_step_time": 7.709710121154785
    },
    {
      "epoch": 5.472183227539063e-06,
      "model_forward_time": 1.751124620437622,
      "step": 5738
    },
    {
      "epoch": 5.472183227539063e-06,
      "step": 5738,
      "training_step_time": 7.689047574996948
    },
    {
      "epoch": 5.473136901855469e-06,
      "model_forward_time": 2.9002885818481445,
      "step": 5739
    },
    {
      "epoch": 5.473136901855469e-06,
      "step": 5739,
      "training_step_time": 8.736251592636108
    },
    {
      "epoch": 5.474090576171875e-06,
      "grad_norm": 0.2607302963733673,
      "learning_rate": 9.998507000006961e-06,
      "loss": 0.0755,
      "step": 5740
    },
    {
      "epoch": 5.474090576171875e-06,
      "model_forward_time": 1.7301523685455322,
      "step": 5740
    },
    {
      "dynamics_loss_avg": 0.07649308890104294,
      "epoch": 5.474090576171875e-06,
      "step": 5740
    },
    {
      "action_loss_avg": 0.007357127848081291,
      "epoch": 5.474090576171875e-06,
      "step": 5740
    },
    {
      "epoch": 5.474090576171875e-06,
      "step": 5740,
      "training_step_time": 9.269964933395386
    },
    {
      "epoch": 5.475044250488281e-06,
      "model_forward_time": 1.7315864562988281,
      "step": 5741
    },
    {
      "epoch": 5.475044250488281e-06,
      "step": 5741,
      "training_step_time": 7.852911710739136
    },
    {
      "epoch": 5.4759979248046875e-06,
      "model_forward_time": 1.7270143032073975,
      "step": 5742
    },
    {
      "epoch": 5.4759979248046875e-06,
      "step": 5742,
      "training_step_time": 7.3845837116241455
    },
    {
      "epoch": 5.476951599121094e-06,
      "model_forward_time": 1.7430672645568848,
      "step": 5743
    },
    {
      "epoch": 5.476951599121094e-06,
      "step": 5743,
      "training_step_time": 6.799890041351318
    },
    {
      "epoch": 5.4779052734375e-06,
      "model_forward_time": 1.7344813346862793,
      "step": 5744
    },
    {
      "epoch": 5.4779052734375e-06,
      "step": 5744,
      "training_step_time": 6.800908803939819
    },
    {
      "epoch": 5.478858947753906e-06,
      "model_forward_time": 1.7609050273895264,
      "step": 5745
    },
    {
      "epoch": 5.478858947753906e-06,
      "step": 5745,
      "training_step_time": 6.808219909667969
    },
    {
      "epoch": 5.479812622070313e-06,
      "model_forward_time": 1.7324471473693848,
      "step": 5746
    },
    {
      "epoch": 5.479812622070313e-06,
      "step": 5746,
      "training_step_time": 7.65223240852356
    },
    {
      "epoch": 5.480766296386719e-06,
      "model_forward_time": 1.7316508293151855,
      "step": 5747
    },
    {
      "epoch": 5.480766296386719e-06,
      "step": 5747,
      "training_step_time": 8.584158658981323
    },
    {
      "epoch": 5.481719970703125e-06,
      "model_forward_time": 2.9230947494506836,
      "step": 5748
    },
    {
      "epoch": 5.481719970703125e-06,
      "step": 5748,
      "training_step_time": 8.722560405731201
    },
    {
      "epoch": 5.482673645019531e-06,
      "model_forward_time": 1.725111961364746,
      "step": 5749
    },
    {
      "epoch": 5.482673645019531e-06,
      "step": 5749,
      "training_step_time": 8.496476173400879
    },
    {
      "epoch": 5.4836273193359375e-06,
      "grad_norm": 0.2274038940668106,
      "learning_rate": 9.99846632275028e-06,
      "loss": 0.0737,
      "step": 5750
    },
    {
      "epoch": 5.4836273193359375e-06,
      "model_forward_time": 1.7748923301696777,
      "step": 5750
    },
    {
      "dynamics_loss_avg": 0.07750081904232502,
      "epoch": 5.4836273193359375e-06,
      "step": 5750
    },
    {
      "action_loss_avg": 0.007863743603229523,
      "epoch": 5.4836273193359375e-06,
      "step": 5750
    },
    {
      "epoch": 5.4836273193359375e-06,
      "step": 5750,
      "training_step_time": 7.732036113739014
    },
    {
      "epoch": 5.484580993652344e-06,
      "model_forward_time": 1.7323119640350342,
      "step": 5751
    },
    {
      "epoch": 5.484580993652344e-06,
      "step": 5751,
      "training_step_time": 7.659313201904297
    },
    {
      "epoch": 5.48553466796875e-06,
      "model_forward_time": 1.723917007446289,
      "step": 5752
    },
    {
      "epoch": 5.48553466796875e-06,
      "step": 5752,
      "training_step_time": 6.7287890911102295
    },
    {
      "epoch": 5.486488342285156e-06,
      "model_forward_time": 1.7507047653198242,
      "step": 5753
    },
    {
      "epoch": 5.486488342285156e-06,
      "step": 5753,
      "training_step_time": 6.6905858516693115
    },
    {
      "epoch": 5.487442016601562e-06,
      "model_forward_time": 1.3706824779510498,
      "step": 5754
    },
    {
      "epoch": 5.487442016601562e-06,
      "step": 5754,
      "training_step_time": 6.725423812866211
    },
    {
      "epoch": 5.488395690917969e-06,
      "model_forward_time": 1.7297871112823486,
      "step": 5755
    },
    {
      "epoch": 5.488395690917969e-06,
      "step": 5755,
      "training_step_time": 7.612290143966675
    },
    {
      "epoch": 5.489349365234375e-06,
      "model_forward_time": 1.7589404582977295,
      "step": 5756
    },
    {
      "epoch": 5.489349365234375e-06,
      "step": 5756,
      "training_step_time": 7.4864466190338135
    },
    {
      "epoch": 5.490303039550781e-06,
      "model_forward_time": 2.914815902709961,
      "step": 5757
    },
    {
      "epoch": 5.490303039550781e-06,
      "step": 5757,
      "training_step_time": 8.520486116409302
    },
    {
      "epoch": 5.4912567138671875e-06,
      "model_forward_time": 1.7389826774597168,
      "step": 5758
    },
    {
      "epoch": 5.4912567138671875e-06,
      "step": 5758,
      "training_step_time": 8.217643976211548
    },
    {
      "epoch": 5.492210388183594e-06,
      "model_forward_time": 1.7363371849060059,
      "step": 5759
    },
    {
      "epoch": 5.492210388183594e-06,
      "step": 5759,
      "training_step_time": 8.39406704902649
    },
    {
      "epoch": 5.4931640625e-06,
      "grad_norm": 0.24070028960704803,
      "learning_rate": 9.998425098868834e-06,
      "loss": 0.0767,
      "step": 5760
    },
    {
      "epoch": 5.4931640625e-06,
      "model_forward_time": 1.7330780029296875,
      "step": 5760
    },
    {
      "dynamics_loss_avg": 0.07992937229573727,
      "epoch": 5.4931640625e-06,
      "step": 5760
    },
    {
      "action_loss_avg": 0.013393984793219715,
      "epoch": 5.4931640625e-06,
      "step": 5760
    },
    {
      "epoch": 5.4931640625e-06,
      "step": 5760,
      "training_step_time": 7.654053211212158
    },
    {
      "epoch": 5.494117736816406e-06,
      "model_forward_time": 1.7323477268218994,
      "step": 5761
    },
    {
      "epoch": 5.494117736816406e-06,
      "step": 5761,
      "training_step_time": 6.700317144393921
    },
    {
      "epoch": 5.495071411132812e-06,
      "model_forward_time": 1.7349767684936523,
      "step": 5762
    },
    {
      "epoch": 5.495071411132812e-06,
      "step": 5762,
      "training_step_time": 6.678081035614014
    },
    {
      "epoch": 5.496025085449219e-06,
      "model_forward_time": 1.7310314178466797,
      "step": 5763
    },
    {
      "epoch": 5.496025085449219e-06,
      "step": 5763,
      "training_step_time": 6.737105369567871
    },
    {
      "epoch": 5.496978759765625e-06,
      "model_forward_time": 1.7446916103363037,
      "step": 5764
    },
    {
      "epoch": 5.496978759765625e-06,
      "step": 5764,
      "training_step_time": 7.547046422958374
    },
    {
      "epoch": 5.497932434082031e-06,
      "model_forward_time": 1.7373201847076416,
      "step": 5765
    },
    {
      "epoch": 5.497932434082031e-06,
      "step": 5765,
      "training_step_time": 7.458936929702759
    },
    {
      "epoch": 5.4988861083984375e-06,
      "model_forward_time": 1.7807867527008057,
      "step": 5766
    },
    {
      "epoch": 5.4988861083984375e-06,
      "step": 5766,
      "training_step_time": 7.5802836418151855
    },
    {
      "epoch": 5.499839782714844e-06,
      "model_forward_time": 2.9225590229034424,
      "step": 5767
    },
    {
      "epoch": 5.499839782714844e-06,
      "step": 5767,
      "training_step_time": 9.960832595825195
    },
    {
      "epoch": 5.50079345703125e-06,
      "model_forward_time": 1.7315869331359863,
      "step": 5768
    },
    {
      "epoch": 5.50079345703125e-06,
      "step": 5768,
      "training_step_time": 8.756445407867432
    },
    {
      "epoch": 5.501747131347656e-06,
      "model_forward_time": 1.7328112125396729,
      "step": 5769
    },
    {
      "epoch": 5.501747131347656e-06,
      "step": 5769,
      "training_step_time": 7.69797682762146
    },
    {
      "epoch": 5.502700805664062e-06,
      "grad_norm": 0.40593138337135315,
      "learning_rate": 9.998383328367136e-06,
      "loss": 0.0819,
      "step": 5770
    },
    {
      "epoch": 5.502700805664062e-06,
      "model_forward_time": 1.729776382446289,
      "step": 5770
    },
    {
      "dynamics_loss_avg": 0.05736964344978333,
      "epoch": 5.502700805664062e-06,
      "step": 5770
    },
    {
      "action_loss_avg": 0.006449479097500444,
      "epoch": 5.502700805664062e-06,
      "step": 5770
    },
    {
      "epoch": 5.502700805664062e-06,
      "step": 5770,
      "training_step_time": 6.713274002075195
    },
    {
      "epoch": 5.503654479980469e-06,
      "model_forward_time": 1.7335827350616455,
      "step": 5771
    },
    {
      "epoch": 5.503654479980469e-06,
      "step": 5771,
      "training_step_time": 6.662365674972534
    },
    {
      "epoch": 5.504608154296875e-06,
      "model_forward_time": 1.7350435256958008,
      "step": 5772
    },
    {
      "epoch": 5.504608154296875e-06,
      "step": 5772,
      "training_step_time": 6.657819032669067
    },
    {
      "epoch": 5.505561828613281e-06,
      "model_forward_time": 1.7363052368164062,
      "step": 5773
    },
    {
      "epoch": 5.505561828613281e-06,
      "step": 5773,
      "training_step_time": 6.659977912902832
    },
    {
      "epoch": 5.5065155029296875e-06,
      "model_forward_time": 1.7442657947540283,
      "step": 5774
    },
    {
      "epoch": 5.5065155029296875e-06,
      "step": 5774,
      "training_step_time": 7.482749938964844
    },
    {
      "epoch": 5.507469177246094e-06,
      "model_forward_time": 1.761056661605835,
      "step": 5775
    },
    {
      "epoch": 5.507469177246094e-06,
      "step": 5775,
      "training_step_time": 8.178344249725342
    },
    {
      "epoch": 5.5084228515625e-06,
      "model_forward_time": 2.9227936267852783,
      "step": 5776
    },
    {
      "epoch": 5.5084228515625e-06,
      "step": 5776,
      "training_step_time": 9.684700727462769
    },
    {
      "epoch": 5.509376525878906e-06,
      "model_forward_time": 1.732717514038086,
      "step": 5777
    },
    {
      "epoch": 5.509376525878906e-06,
      "step": 5777,
      "training_step_time": 8.154826879501343
    },
    {
      "epoch": 5.510330200195312e-06,
      "model_forward_time": 1.7263824939727783,
      "step": 5778
    },
    {
      "epoch": 5.510330200195312e-06,
      "step": 5778,
      "training_step_time": 7.750873804092407
    },
    {
      "epoch": 5.511283874511719e-06,
      "model_forward_time": 1.752917766571045,
      "step": 5779
    },
    {
      "epoch": 5.511283874511719e-06,
      "step": 5779,
      "training_step_time": 6.7203686237335205
    },
    {
      "epoch": 5.512237548828125e-06,
      "grad_norm": 0.40447160601615906,
      "learning_rate": 9.998341011249752e-06,
      "loss": 0.0824,
      "step": 5780
    },
    {
      "epoch": 5.512237548828125e-06,
      "model_forward_time": 1.7356631755828857,
      "step": 5780
    },
    {
      "dynamics_loss_avg": 0.06184784844517708,
      "epoch": 5.512237548828125e-06,
      "step": 5780
    },
    {
      "action_loss_avg": 0.009154825657606124,
      "epoch": 5.512237548828125e-06,
      "step": 5780
    },
    {
      "epoch": 5.512237548828125e-06,
      "step": 5780,
      "training_step_time": 6.717093229293823
    },
    {
      "epoch": 5.513191223144531e-06,
      "model_forward_time": 1.7279233932495117,
      "step": 5781
    },
    {
      "epoch": 5.513191223144531e-06,
      "step": 5781,
      "training_step_time": 6.746354103088379
    },
    {
      "epoch": 5.5141448974609375e-06,
      "model_forward_time": 1.7369554042816162,
      "step": 5782
    },
    {
      "epoch": 5.5141448974609375e-06,
      "step": 5782,
      "training_step_time": 6.714300155639648
    },
    {
      "epoch": 5.515098571777344e-06,
      "model_forward_time": 1.7333714962005615,
      "step": 5783
    },
    {
      "epoch": 5.515098571777344e-06,
      "step": 5783,
      "training_step_time": 8.616246461868286
    },
    {
      "epoch": 5.51605224609375e-06,
      "model_forward_time": 1.7340149879455566,
      "step": 5784
    },
    {
      "epoch": 5.51605224609375e-06,
      "step": 5784,
      "training_step_time": 9.640098094940186
    },
    {
      "epoch": 5.517005920410156e-06,
      "model_forward_time": 2.9195058345794678,
      "step": 5785
    },
    {
      "epoch": 5.517005920410156e-06,
      "step": 5785,
      "training_step_time": 9.968584537506104
    },
    {
      "epoch": 5.517959594726562e-06,
      "model_forward_time": 1.7270793914794922,
      "step": 5786
    },
    {
      "epoch": 5.517959594726562e-06,
      "step": 5786,
      "training_step_time": 9.57033109664917
    },
    {
      "epoch": 5.518913269042969e-06,
      "model_forward_time": 1.743016242980957,
      "step": 5787
    },
    {
      "epoch": 5.518913269042969e-06,
      "step": 5787,
      "training_step_time": 7.85333776473999
    },
    {
      "epoch": 5.519866943359375e-06,
      "model_forward_time": 1.732085943222046,
      "step": 5788
    },
    {
      "epoch": 5.519866943359375e-06,
      "step": 5788,
      "training_step_time": 6.733201265335083
    },
    {
      "epoch": 5.520820617675781e-06,
      "model_forward_time": 1.7351493835449219,
      "step": 5789
    },
    {
      "epoch": 5.520820617675781e-06,
      "step": 5789,
      "training_step_time": 6.747450590133667
    },
    {
      "epoch": 5.5217742919921875e-06,
      "grad_norm": 0.2724761962890625,
      "learning_rate": 9.99829814752131e-06,
      "loss": 0.0779,
      "step": 5790
    },
    {
      "epoch": 5.5217742919921875e-06,
      "model_forward_time": 1.7231693267822266,
      "step": 5790
    },
    {
      "dynamics_loss_avg": 0.06370548065751791,
      "epoch": 5.5217742919921875e-06,
      "step": 5790
    },
    {
      "action_loss_avg": 0.007576352218165994,
      "epoch": 5.5217742919921875e-06,
      "step": 5790
    },
    {
      "epoch": 5.5217742919921875e-06,
      "step": 5790,
      "training_step_time": 6.7825539112091064
    },
    {
      "epoch": 5.522727966308594e-06,
      "model_forward_time": 1.7439484596252441,
      "step": 5791
    },
    {
      "epoch": 5.522727966308594e-06,
      "step": 5791,
      "training_step_time": 6.735527515411377
    },
    {
      "epoch": 5.523681640625e-06,
      "model_forward_time": 1.7425518035888672,
      "step": 5792
    },
    {
      "epoch": 5.523681640625e-06,
      "step": 5792,
      "training_step_time": 7.588110685348511
    },
    {
      "epoch": 5.524635314941406e-06,
      "model_forward_time": 1.7331874370574951,
      "step": 5793
    },
    {
      "epoch": 5.524635314941406e-06,
      "step": 5793,
      "training_step_time": 7.50601053237915
    },
    {
      "epoch": 5.525588989257812e-06,
      "model_forward_time": 2.905468225479126,
      "step": 5794
    },
    {
      "epoch": 5.525588989257812e-06,
      "step": 5794,
      "training_step_time": 9.200115442276001
    },
    {
      "epoch": 5.526542663574219e-06,
      "model_forward_time": 1.735255479812622,
      "step": 5795
    },
    {
      "epoch": 5.526542663574219e-06,
      "step": 5795,
      "training_step_time": 9.774324417114258
    },
    {
      "epoch": 5.527496337890625e-06,
      "model_forward_time": 1.7383217811584473,
      "step": 5796
    },
    {
      "epoch": 5.527496337890625e-06,
      "step": 5796,
      "training_step_time": 8.739753484725952
    },
    {
      "epoch": 5.528450012207031e-06,
      "model_forward_time": 1.7293610572814941,
      "step": 5797
    },
    {
      "epoch": 5.528450012207031e-06,
      "step": 5797,
      "training_step_time": 7.564942121505737
    },
    {
      "epoch": 5.5294036865234375e-06,
      "model_forward_time": 1.7350876331329346,
      "step": 5798
    },
    {
      "epoch": 5.5294036865234375e-06,
      "step": 5798,
      "training_step_time": 6.705850124359131
    },
    {
      "epoch": 5.530357360839844e-06,
      "model_forward_time": 1.7387447357177734,
      "step": 5799
    },
    {
      "epoch": 5.530357360839844e-06,
      "step": 5799,
      "training_step_time": 6.72495174407959
    },
    {
      "epoch": 5.53131103515625e-06,
      "grad_norm": 0.3504405617713928,
      "learning_rate": 9.998254737186497e-06,
      "loss": 0.075,
      "step": 5800
    },
    {
      "epoch": 5.53131103515625e-06,
      "model_forward_time": 1.7490794658660889,
      "step": 5800
    },
    {
      "dynamics_loss_avg": 0.08096212651580573,
      "epoch": 5.53131103515625e-06,
      "step": 5800
    },
    {
      "action_loss_avg": 0.010820277582388372,
      "epoch": 5.53131103515625e-06,
      "step": 5800
    },
    {
      "epoch": 5.53131103515625e-06,
      "step": 5800,
      "training_step_time": 6.759097337722778
    },
    {
      "epoch": 5.532264709472656e-06,
      "model_forward_time": 1.7338154315948486,
      "step": 5801
    },
    {
      "epoch": 5.532264709472656e-06,
      "step": 5801,
      "training_step_time": 7.586176633834839
    },
    {
      "epoch": 5.533218383789062e-06,
      "model_forward_time": 1.7323622703552246,
      "step": 5802
    },
    {
      "epoch": 5.533218383789062e-06,
      "step": 5802,
      "training_step_time": 8.323474645614624
    },
    {
      "epoch": 5.534172058105469e-06,
      "model_forward_time": 2.8983707427978516,
      "step": 5803
    },
    {
      "epoch": 5.534172058105469e-06,
      "step": 5803,
      "training_step_time": 8.60077428817749
    },
    {
      "epoch": 5.535125732421875e-06,
      "model_forward_time": 1.7301867008209229,
      "step": 5804
    },
    {
      "epoch": 5.535125732421875e-06,
      "step": 5804,
      "training_step_time": 8.205147981643677
    },
    {
      "epoch": 5.536079406738281e-06,
      "model_forward_time": 1.7332525253295898,
      "step": 5805
    },
    {
      "epoch": 5.536079406738281e-06,
      "step": 5805,
      "training_step_time": 7.73568320274353
    },
    {
      "epoch": 5.5370330810546875e-06,
      "model_forward_time": 1.7409367561340332,
      "step": 5806
    },
    {
      "epoch": 5.5370330810546875e-06,
      "step": 5806,
      "training_step_time": 7.687772512435913
    },
    {
      "epoch": 5.537986755371094e-06,
      "model_forward_time": 1.7322285175323486,
      "step": 5807
    },
    {
      "epoch": 5.537986755371094e-06,
      "step": 5807,
      "training_step_time": 6.740387916564941
    },
    {
      "epoch": 5.5389404296875e-06,
      "model_forward_time": 1.7314653396606445,
      "step": 5808
    },
    {
      "epoch": 5.5389404296875e-06,
      "step": 5808,
      "training_step_time": 6.69312310218811
    },
    {
      "epoch": 5.539894104003906e-06,
      "model_forward_time": 1.7278354167938232,
      "step": 5809
    },
    {
      "epoch": 5.539894104003906e-06,
      "step": 5809,
      "training_step_time": 6.7198851108551025
    },
    {
      "epoch": 5.540847778320312e-06,
      "grad_norm": 0.3968040943145752,
      "learning_rate": 9.99821078025006e-06,
      "loss": 0.0785,
      "step": 5810
    },
    {
      "epoch": 5.540847778320312e-06,
      "model_forward_time": 1.7462410926818848,
      "step": 5810
    },
    {
      "dynamics_loss_avg": 0.06681622620671987,
      "epoch": 5.540847778320312e-06,
      "step": 5810
    },
    {
      "action_loss_avg": 0.0076985870022326704,
      "epoch": 5.540847778320312e-06,
      "step": 5810
    },
    {
      "epoch": 5.540847778320312e-06,
      "step": 5810,
      "training_step_time": 8.254473209381104
    },
    {
      "epoch": 5.541801452636719e-06,
      "model_forward_time": 1.7317607402801514,
      "step": 5811
    },
    {
      "epoch": 5.541801452636719e-06,
      "step": 5811,
      "training_step_time": 9.249192953109741
    },
    {
      "epoch": 5.542755126953125e-06,
      "model_forward_time": 1.8078784942626953,
      "step": 5812
    },
    {
      "epoch": 5.542755126953125e-06,
      "step": 5812,
      "training_step_time": 8.723724365234375
    },
    {
      "epoch": 5.543708801269531e-06,
      "model_forward_time": 1.7358896732330322,
      "step": 5813
    },
    {
      "epoch": 5.543708801269531e-06,
      "step": 5813,
      "training_step_time": 8.689379692077637
    },
    {
      "epoch": 5.5446624755859376e-06,
      "model_forward_time": 1.7468898296356201,
      "step": 5814
    },
    {
      "epoch": 5.5446624755859376e-06,
      "step": 5814,
      "training_step_time": 8.461012363433838
    },
    {
      "epoch": 5.545616149902344e-06,
      "model_forward_time": 1.7328319549560547,
      "step": 5815
    },
    {
      "epoch": 5.545616149902344e-06,
      "step": 5815,
      "training_step_time": 7.680622100830078
    },
    {
      "epoch": 5.54656982421875e-06,
      "model_forward_time": 1.7338433265686035,
      "step": 5816
    },
    {
      "epoch": 5.54656982421875e-06,
      "step": 5816,
      "training_step_time": 6.703296422958374
    },
    {
      "epoch": 5.547523498535156e-06,
      "model_forward_time": 1.7352635860443115,
      "step": 5817
    },
    {
      "epoch": 5.547523498535156e-06,
      "step": 5817,
      "training_step_time": 6.675084352493286
    },
    {
      "epoch": 5.548477172851562e-06,
      "model_forward_time": 1.7467327117919922,
      "step": 5818
    },
    {
      "epoch": 5.548477172851562e-06,
      "step": 5818,
      "training_step_time": 6.724350690841675
    },
    {
      "epoch": 5.549430847167969e-06,
      "model_forward_time": 1.739804744720459,
      "step": 5819
    },
    {
      "epoch": 5.549430847167969e-06,
      "step": 5819,
      "training_step_time": 7.19331431388855
    },
    {
      "epoch": 5.550384521484375e-06,
      "grad_norm": 0.2879989445209503,
      "learning_rate": 9.998166276716808e-06,
      "loss": 0.0765,
      "step": 5820
    },
    {
      "epoch": 5.550384521484375e-06,
      "model_forward_time": 1.7349326610565186,
      "step": 5820
    },
    {
      "dynamics_loss_avg": 0.06913745384663343,
      "epoch": 5.550384521484375e-06,
      "step": 5820
    },
    {
      "action_loss_avg": 0.005494509113486856,
      "epoch": 5.550384521484375e-06,
      "step": 5820
    },
    {
      "epoch": 5.550384521484375e-06,
      "step": 5820,
      "training_step_time": 7.6999194622039795
    },
    {
      "epoch": 5.551338195800781e-06,
      "model_forward_time": 1.7662429809570312,
      "step": 5821
    },
    {
      "epoch": 5.551338195800781e-06,
      "step": 5821,
      "training_step_time": 7.613613843917847
    },
    {
      "epoch": 5.5522918701171876e-06,
      "model_forward_time": 2.886739730834961,
      "step": 5822
    },
    {
      "epoch": 5.5522918701171876e-06,
      "step": 5822,
      "training_step_time": 9.39273715019226
    },
    {
      "epoch": 5.553245544433594e-06,
      "model_forward_time": 1.7308495044708252,
      "step": 5823
    },
    {
      "epoch": 5.553245544433594e-06,
      "step": 5823,
      "training_step_time": 7.650947570800781
    },
    {
      "epoch": 5.55419921875e-06,
      "model_forward_time": 1.744184970855713,
      "step": 5824
    },
    {
      "epoch": 5.55419921875e-06,
      "step": 5824,
      "training_step_time": 7.69728946685791
    },
    {
      "epoch": 5.555152893066406e-06,
      "model_forward_time": 1.7351150512695312,
      "step": 5825
    },
    {
      "epoch": 5.555152893066406e-06,
      "step": 5825,
      "training_step_time": 7.370462894439697
    },
    {
      "epoch": 5.556106567382812e-06,
      "model_forward_time": 1.7353386878967285,
      "step": 5826
    },
    {
      "epoch": 5.556106567382812e-06,
      "step": 5826,
      "training_step_time": 6.715681314468384
    },
    {
      "epoch": 5.557060241699219e-06,
      "model_forward_time": 1.7299795150756836,
      "step": 5827
    },
    {
      "epoch": 5.557060241699219e-06,
      "step": 5827,
      "training_step_time": 6.720834732055664
    },
    {
      "epoch": 5.558013916015625e-06,
      "model_forward_time": 1.7344882488250732,
      "step": 5828
    },
    {
      "epoch": 5.558013916015625e-06,
      "step": 5828,
      "training_step_time": 7.239084005355835
    },
    {
      "epoch": 5.558967590332031e-06,
      "model_forward_time": 1.7353730201721191,
      "step": 5829
    },
    {
      "epoch": 5.558967590332031e-06,
      "step": 5829,
      "training_step_time": 7.599052429199219
    },
    {
      "epoch": 5.5599212646484376e-06,
      "grad_norm": 0.2906298041343689,
      "learning_rate": 9.998121226591606e-06,
      "loss": 0.0696,
      "step": 5830
    },
    {
      "epoch": 5.5599212646484376e-06,
      "model_forward_time": 1.734891414642334,
      "step": 5830
    },
    {
      "dynamics_loss_avg": 0.06205393858253956,
      "epoch": 5.5599212646484376e-06,
      "step": 5830
    },
    {
      "action_loss_avg": 0.009439721913076937,
      "epoch": 5.5599212646484376e-06,
      "step": 5830
    },
    {
      "epoch": 5.5599212646484376e-06,
      "step": 5830,
      "training_step_time": 9.342580318450928
    },
    {
      "epoch": 5.560874938964844e-06,
      "model_forward_time": 2.888850212097168,
      "step": 5831
    },
    {
      "epoch": 5.560874938964844e-06,
      "step": 5831,
      "training_step_time": 7.6391282081604
    },
    {
      "epoch": 5.56182861328125e-06,
      "model_forward_time": 1.735205888748169,
      "step": 5832
    },
    {
      "epoch": 5.56182861328125e-06,
      "step": 5832,
      "training_step_time": 9.49059271812439
    },
    {
      "epoch": 5.5627822875976564e-06,
      "model_forward_time": 1.738495111465454,
      "step": 5833
    },
    {
      "epoch": 5.5627822875976564e-06,
      "step": 5833,
      "training_step_time": 8.502654790878296
    },
    {
      "epoch": 5.563735961914062e-06,
      "model_forward_time": 1.7256066799163818,
      "step": 5834
    },
    {
      "epoch": 5.563735961914062e-06,
      "step": 5834,
      "training_step_time": 7.685720443725586
    },
    {
      "epoch": 5.564689636230469e-06,
      "model_forward_time": 1.732271432876587,
      "step": 5835
    },
    {
      "epoch": 5.564689636230469e-06,
      "step": 5835,
      "training_step_time": 6.732966184616089
    },
    {
      "epoch": 5.565643310546875e-06,
      "model_forward_time": 1.7382285594940186,
      "step": 5836
    },
    {
      "epoch": 5.565643310546875e-06,
      "step": 5836,
      "training_step_time": 6.753227472305298
    },
    {
      "epoch": 5.566596984863281e-06,
      "model_forward_time": 1.7461047172546387,
      "step": 5837
    },
    {
      "epoch": 5.566596984863281e-06,
      "step": 5837,
      "training_step_time": 7.258739948272705
    },
    {
      "epoch": 5.5675506591796876e-06,
      "model_forward_time": 1.7299160957336426,
      "step": 5838
    },
    {
      "epoch": 5.5675506591796876e-06,
      "step": 5838,
      "training_step_time": 8.611392974853516
    },
    {
      "epoch": 5.568504333496094e-06,
      "model_forward_time": 1.7375879287719727,
      "step": 5839
    },
    {
      "epoch": 5.568504333496094e-06,
      "step": 5839,
      "training_step_time": 8.423134326934814
    },
    {
      "epoch": 5.5694580078125e-06,
      "grad_norm": 0.265684574842453,
      "learning_rate": 9.998075629879382e-06,
      "loss": 0.0696,
      "step": 5840
    },
    {
      "epoch": 5.5694580078125e-06,
      "model_forward_time": 2.893376350402832,
      "step": 5840
    },
    {
      "dynamics_loss_avg": 0.07821417953819036,
      "epoch": 5.5694580078125e-06,
      "step": 5840
    },
    {
      "action_loss_avg": 0.009021652850788087,
      "epoch": 5.5694580078125e-06,
      "step": 5840
    },
    {
      "epoch": 5.5694580078125e-06,
      "step": 5840,
      "training_step_time": 8.23207688331604
    },
    {
      "epoch": 5.5704116821289064e-06,
      "model_forward_time": 1.7312207221984863,
      "step": 5841
    },
    {
      "epoch": 5.5704116821289064e-06,
      "step": 5841,
      "training_step_time": 8.699999809265137
    },
    {
      "epoch": 5.571365356445312e-06,
      "model_forward_time": 1.7386000156402588,
      "step": 5842
    },
    {
      "epoch": 5.571365356445312e-06,
      "step": 5842,
      "training_step_time": 9.28666067123413
    },
    {
      "epoch": 5.572319030761719e-06,
      "model_forward_time": 1.7333383560180664,
      "step": 5843
    },
    {
      "epoch": 5.572319030761719e-06,
      "step": 5843,
      "training_step_time": 7.755331039428711
    },
    {
      "epoch": 5.573272705078125e-06,
      "model_forward_time": 1.7469878196716309,
      "step": 5844
    },
    {
      "epoch": 5.573272705078125e-06,
      "step": 5844,
      "training_step_time": 6.758033990859985
    },
    {
      "epoch": 5.574226379394531e-06,
      "model_forward_time": 1.738877773284912,
      "step": 5845
    },
    {
      "epoch": 5.574226379394531e-06,
      "step": 5845,
      "training_step_time": 6.635817766189575
    },
    {
      "epoch": 5.575180053710938e-06,
      "model_forward_time": 1.7329254150390625,
      "step": 5846
    },
    {
      "epoch": 5.575180053710938e-06,
      "step": 5846,
      "training_step_time": 7.235523223876953
    },
    {
      "epoch": 5.576133728027344e-06,
      "model_forward_time": 1.733673095703125,
      "step": 5847
    },
    {
      "epoch": 5.576133728027344e-06,
      "step": 5847,
      "training_step_time": 7.578854560852051
    },
    {
      "epoch": 5.57708740234375e-06,
      "model_forward_time": 1.7436842918395996,
      "step": 5848
    },
    {
      "epoch": 5.57708740234375e-06,
      "step": 5848,
      "training_step_time": 7.619091749191284
    },
    {
      "epoch": 5.5780410766601564e-06,
      "model_forward_time": 2.890638589859009,
      "step": 5849
    },
    {
      "epoch": 5.5780410766601564e-06,
      "step": 5849,
      "training_step_time": 7.62951397895813
    },
    {
      "epoch": 5.578994750976562e-06,
      "grad_norm": 0.16199347376823425,
      "learning_rate": 9.998029486585121e-06,
      "loss": 0.0722,
      "step": 5850
    },
    {
      "epoch": 5.578994750976562e-06,
      "model_forward_time": 1.720956802368164,
      "step": 5850
    },
    {
      "dynamics_loss_avg": 0.06767927519977093,
      "epoch": 5.578994750976562e-06,
      "step": 5850
    },
    {
      "action_loss_avg": 0.008334704954177142,
      "epoch": 5.578994750976562e-06,
      "step": 5850
    },
    {
      "epoch": 5.578994750976562e-06,
      "step": 5850,
      "training_step_time": 7.669426441192627
    },
    {
      "epoch": 5.579948425292969e-06,
      "model_forward_time": 1.745868444442749,
      "step": 5851
    },
    {
      "epoch": 5.579948425292969e-06,
      "step": 5851,
      "training_step_time": 9.295441150665283
    },
    {
      "epoch": 5.580902099609375e-06,
      "model_forward_time": 1.362384557723999,
      "step": 5852
    },
    {
      "epoch": 5.580902099609375e-06,
      "step": 5852,
      "training_step_time": 7.612912654876709
    },
    {
      "epoch": 5.581855773925781e-06,
      "model_forward_time": 1.7341439723968506,
      "step": 5853
    },
    {
      "epoch": 5.581855773925781e-06,
      "step": 5853,
      "training_step_time": 7.709837913513184
    },
    {
      "epoch": 5.582809448242188e-06,
      "model_forward_time": 1.739464282989502,
      "step": 5854
    },
    {
      "epoch": 5.582809448242188e-06,
      "step": 5854,
      "training_step_time": 6.769468307495117
    },
    {
      "epoch": 5.583763122558594e-06,
      "model_forward_time": 1.728959560394287,
      "step": 5855
    },
    {
      "epoch": 5.583763122558594e-06,
      "step": 5855,
      "training_step_time": 7.255226373672485
    },
    {
      "epoch": 5.584716796875e-06,
      "model_forward_time": 1.7334468364715576,
      "step": 5856
    },
    {
      "epoch": 5.584716796875e-06,
      "step": 5856,
      "training_step_time": 8.081579446792603
    },
    {
      "epoch": 5.5856704711914064e-06,
      "model_forward_time": 1.732909917831421,
      "step": 5857
    },
    {
      "epoch": 5.5856704711914064e-06,
      "step": 5857,
      "training_step_time": 7.643116235733032
    },
    {
      "epoch": 5.586624145507812e-06,
      "model_forward_time": 2.9283387660980225,
      "step": 5858
    },
    {
      "epoch": 5.586624145507812e-06,
      "step": 5858,
      "training_step_time": 8.407496213912964
    },
    {
      "epoch": 5.587577819824219e-06,
      "model_forward_time": 1.7259016036987305,
      "step": 5859
    },
    {
      "epoch": 5.587577819824219e-06,
      "step": 5859,
      "training_step_time": 7.621203422546387
    },
    {
      "epoch": 5.588531494140625e-06,
      "grad_norm": 0.3396676778793335,
      "learning_rate": 9.99798279671387e-06,
      "loss": 0.0744,
      "step": 5860
    },
    {
      "epoch": 5.588531494140625e-06,
      "model_forward_time": 1.7442946434020996,
      "step": 5860
    },
    {
      "dynamics_loss_avg": 0.06764907632023096,
      "epoch": 5.588531494140625e-06,
      "step": 5860
    },
    {
      "action_loss_avg": 0.00768718405161053,
      "epoch": 5.588531494140625e-06,
      "step": 5860
    },
    {
      "epoch": 5.588531494140625e-06,
      "step": 5860,
      "training_step_time": 9.25562310218811
    },
    {
      "epoch": 5.589485168457031e-06,
      "model_forward_time": 1.7297399044036865,
      "step": 5861
    },
    {
      "epoch": 5.589485168457031e-06,
      "step": 5861,
      "training_step_time": 8.662489175796509
    },
    {
      "epoch": 5.590438842773438e-06,
      "model_forward_time": 1.74755859375,
      "step": 5862
    },
    {
      "epoch": 5.590438842773438e-06,
      "step": 5862,
      "training_step_time": 7.647858381271362
    },
    {
      "epoch": 5.591392517089844e-06,
      "model_forward_time": 1.73101806640625,
      "step": 5863
    },
    {
      "epoch": 5.591392517089844e-06,
      "step": 5863,
      "training_step_time": 6.714808464050293
    },
    {
      "epoch": 5.59234619140625e-06,
      "model_forward_time": 1.7288267612457275,
      "step": 5864
    },
    {
      "epoch": 5.59234619140625e-06,
      "step": 5864,
      "training_step_time": 7.237984657287598
    },
    {
      "epoch": 5.5932998657226565e-06,
      "model_forward_time": 1.7309691905975342,
      "step": 5865
    },
    {
      "epoch": 5.5932998657226565e-06,
      "step": 5865,
      "training_step_time": 7.5415403842926025
    },
    {
      "epoch": 5.594253540039062e-06,
      "model_forward_time": 1.7434823513031006,
      "step": 5866
    },
    {
      "epoch": 5.594253540039062e-06,
      "step": 5866,
      "training_step_time": 7.716983079910278
    },
    {
      "epoch": 5.595207214355469e-06,
      "model_forward_time": 1.858560562133789,
      "step": 5867
    },
    {
      "epoch": 5.595207214355469e-06,
      "step": 5867,
      "training_step_time": 8.194755792617798
    },
    {
      "epoch": 5.596160888671875e-06,
      "model_forward_time": 2.8897862434387207,
      "step": 5868
    },
    {
      "epoch": 5.596160888671875e-06,
      "step": 5868,
      "training_step_time": 8.432448625564575
    },
    {
      "epoch": 5.597114562988281e-06,
      "model_forward_time": 1.7453043460845947,
      "step": 5869
    },
    {
      "epoch": 5.597114562988281e-06,
      "step": 5869,
      "training_step_time": 8.33749008178711
    },
    {
      "epoch": 5.598068237304688e-06,
      "grad_norm": 0.4416520893573761,
      "learning_rate": 9.997935560270734e-06,
      "loss": 0.0716,
      "step": 5870
    },
    {
      "epoch": 5.598068237304688e-06,
      "model_forward_time": 1.7362616062164307,
      "step": 5870
    },
    {
      "dynamics_loss_avg": 0.07151610627770424,
      "epoch": 5.598068237304688e-06,
      "step": 5870
    },
    {
      "action_loss_avg": 0.007241192646324635,
      "epoch": 5.598068237304688e-06,
      "step": 5870
    },
    {
      "epoch": 5.598068237304688e-06,
      "step": 5870,
      "training_step_time": 8.3461434841156
    },
    {
      "epoch": 5.599021911621094e-06,
      "model_forward_time": 1.7380859851837158,
      "step": 5871
    },
    {
      "epoch": 5.599021911621094e-06,
      "step": 5871,
      "training_step_time": 7.765544652938843
    },
    {
      "epoch": 5.5999755859375e-06,
      "model_forward_time": 1.7630555629730225,
      "step": 5872
    },
    {
      "epoch": 5.5999755859375e-06,
      "step": 5872,
      "training_step_time": 7.624443054199219
    },
    {
      "epoch": 5.6009292602539065e-06,
      "model_forward_time": 1.7314579486846924,
      "step": 5873
    },
    {
      "epoch": 5.6009292602539065e-06,
      "step": 5873,
      "training_step_time": 7.304106950759888
    },
    {
      "epoch": 5.601882934570312e-06,
      "model_forward_time": 1.7414960861206055,
      "step": 5874
    },
    {
      "epoch": 5.601882934570312e-06,
      "step": 5874,
      "training_step_time": 7.555032253265381
    },
    {
      "epoch": 5.602836608886719e-06,
      "model_forward_time": 1.7346532344818115,
      "step": 5875
    },
    {
      "epoch": 5.602836608886719e-06,
      "step": 5875,
      "training_step_time": 7.744914293289185
    },
    {
      "epoch": 5.603790283203125e-06,
      "model_forward_time": 1.8074114322662354,
      "step": 5876
    },
    {
      "epoch": 5.603790283203125e-06,
      "step": 5876,
      "training_step_time": 7.620502233505249
    },
    {
      "epoch": 5.604743957519531e-06,
      "model_forward_time": 2.984241485595703,
      "step": 5877
    },
    {
      "epoch": 5.604743957519531e-06,
      "step": 5877,
      "training_step_time": 8.71654748916626
    },
    {
      "epoch": 5.605697631835938e-06,
      "model_forward_time": 1.7265088558197021,
      "step": 5878
    },
    {
      "epoch": 5.605697631835938e-06,
      "step": 5878,
      "training_step_time": 9.295788526535034
    },
    {
      "epoch": 5.606651306152343e-06,
      "model_forward_time": 1.7484371662139893,
      "step": 5879
    },
    {
      "epoch": 5.606651306152343e-06,
      "step": 5879,
      "training_step_time": 9.77741265296936
    },
    {
      "epoch": 5.60760498046875e-06,
      "grad_norm": 0.33055567741394043,
      "learning_rate": 9.997887777260879e-06,
      "loss": 0.072,
      "step": 5880
    },
    {
      "epoch": 5.60760498046875e-06,
      "model_forward_time": 1.733428955078125,
      "step": 5880
    },
    {
      "dynamics_loss_avg": 0.07323266994208097,
      "epoch": 5.60760498046875e-06,
      "step": 5880
    },
    {
      "action_loss_avg": 0.009062583476770669,
      "epoch": 5.60760498046875e-06,
      "step": 5880
    },
    {
      "epoch": 5.60760498046875e-06,
      "step": 5880,
      "training_step_time": 7.765254974365234
    },
    {
      "epoch": 5.6085586547851565e-06,
      "model_forward_time": 1.7450566291809082,
      "step": 5881
    },
    {
      "epoch": 5.6085586547851565e-06,
      "step": 5881,
      "training_step_time": 7.615216255187988
    },
    {
      "epoch": 5.609512329101562e-06,
      "model_forward_time": 1.7336621284484863,
      "step": 5882
    },
    {
      "epoch": 5.609512329101562e-06,
      "step": 5882,
      "training_step_time": 7.246808052062988
    },
    {
      "epoch": 5.610466003417969e-06,
      "model_forward_time": 1.7304527759552002,
      "step": 5883
    },
    {
      "epoch": 5.610466003417969e-06,
      "step": 5883,
      "training_step_time": 7.54648494720459
    },
    {
      "epoch": 5.611419677734375e-06,
      "model_forward_time": 1.7380409240722656,
      "step": 5884
    },
    {
      "epoch": 5.611419677734375e-06,
      "step": 5884,
      "training_step_time": 7.70339298248291
    },
    {
      "epoch": 5.612373352050781e-06,
      "model_forward_time": 1.8000497817993164,
      "step": 5885
    },
    {
      "epoch": 5.612373352050781e-06,
      "step": 5885,
      "training_step_time": 8.425944089889526
    },
    {
      "epoch": 5.613327026367188e-06,
      "model_forward_time": 2.917238712310791,
      "step": 5886
    },
    {
      "epoch": 5.613327026367188e-06,
      "step": 5886,
      "training_step_time": 7.657341003417969
    },
    {
      "epoch": 5.614280700683593e-06,
      "model_forward_time": 1.7511796951293945,
      "step": 5887
    },
    {
      "epoch": 5.614280700683593e-06,
      "step": 5887,
      "training_step_time": 8.735551357269287
    },
    {
      "epoch": 5.615234375e-06,
      "model_forward_time": 1.7326958179473877,
      "step": 5888
    },
    {
      "epoch": 5.615234375e-06,
      "step": 5888,
      "training_step_time": 7.6661481857299805
    },
    {
      "epoch": 5.6161880493164065e-06,
      "model_forward_time": 1.7294726371765137,
      "step": 5889
    },
    {
      "epoch": 5.6161880493164065e-06,
      "step": 5889,
      "training_step_time": 8.720744609832764
    },
    {
      "epoch": 5.617141723632812e-06,
      "grad_norm": 0.2879010736942291,
      "learning_rate": 9.997839447689533e-06,
      "loss": 0.079,
      "step": 5890
    },
    {
      "epoch": 5.617141723632812e-06,
      "model_forward_time": 1.7214908599853516,
      "step": 5890
    },
    {
      "dynamics_loss_avg": 0.0751794420182705,
      "epoch": 5.617141723632812e-06,
      "step": 5890
    },
    {
      "action_loss_avg": 0.00814036283409223,
      "epoch": 5.617141723632812e-06,
      "step": 5890
    },
    {
      "epoch": 5.617141723632812e-06,
      "step": 5890,
      "training_step_time": 7.573180913925171
    },
    {
      "epoch": 5.618095397949219e-06,
      "model_forward_time": 1.7346422672271729,
      "step": 5891
    },
    {
      "epoch": 5.618095397949219e-06,
      "step": 5891,
      "training_step_time": 7.209802150726318
    },
    {
      "epoch": 5.619049072265625e-06,
      "model_forward_time": 1.7663559913635254,
      "step": 5892
    },
    {
      "epoch": 5.619049072265625e-06,
      "step": 5892,
      "training_step_time": 7.478838205337524
    },
    {
      "epoch": 5.620002746582031e-06,
      "model_forward_time": 1.7336947917938232,
      "step": 5893
    },
    {
      "epoch": 5.620002746582031e-06,
      "step": 5893,
      "training_step_time": 7.587016582489014
    },
    {
      "epoch": 5.620956420898438e-06,
      "model_forward_time": 1.7273743152618408,
      "step": 5894
    },
    {
      "epoch": 5.620956420898438e-06,
      "step": 5894,
      "training_step_time": 8.36085844039917
    },
    {
      "epoch": 5.621910095214843e-06,
      "model_forward_time": 2.9976024627685547,
      "step": 5895
    },
    {
      "epoch": 5.621910095214843e-06,
      "step": 5895,
      "training_step_time": 9.610029935836792
    },
    {
      "epoch": 5.62286376953125e-06,
      "model_forward_time": 1.7331328392028809,
      "step": 5896
    },
    {
      "epoch": 5.62286376953125e-06,
      "step": 5896,
      "training_step_time": 9.240638017654419
    },
    {
      "epoch": 5.6238174438476565e-06,
      "model_forward_time": 1.7193961143493652,
      "step": 5897
    },
    {
      "epoch": 5.6238174438476565e-06,
      "step": 5897,
      "training_step_time": 8.535263776779175
    },
    {
      "epoch": 5.624771118164062e-06,
      "model_forward_time": 1.7139408588409424,
      "step": 5898
    },
    {
      "epoch": 5.624771118164062e-06,
      "step": 5898,
      "training_step_time": 7.711646318435669
    },
    {
      "epoch": 5.625724792480469e-06,
      "model_forward_time": 1.7151317596435547,
      "step": 5899
    },
    {
      "epoch": 5.625724792480469e-06,
      "step": 5899,
      "training_step_time": 7.9497339725494385
    },
    {
      "epoch": 5.626678466796875e-06,
      "grad_norm": 0.2090018093585968,
      "learning_rate": 9.997790571561978e-06,
      "loss": 0.0775,
      "step": 5900
    },
    {
      "epoch": 5.626678466796875e-06,
      "model_forward_time": 1.7132108211517334,
      "step": 5900
    },
    {
      "dynamics_loss_avg": 0.06408133935183287,
      "epoch": 5.626678466796875e-06,
      "step": 5900
    },
    {
      "action_loss_avg": 0.004416856390889734,
      "epoch": 5.626678466796875e-06,
      "step": 5900
    },
    {
      "epoch": 5.626678466796875e-06,
      "step": 5900,
      "training_step_time": 7.210737943649292
    },
    {
      "epoch": 5.627632141113281e-06,
      "model_forward_time": 1.7258272171020508,
      "step": 5901
    },
    {
      "epoch": 5.627632141113281e-06,
      "step": 5901,
      "training_step_time": 7.497386693954468
    },
    {
      "epoch": 5.628585815429688e-06,
      "model_forward_time": 1.7195477485656738,
      "step": 5902
    },
    {
      "epoch": 5.628585815429688e-06,
      "step": 5902,
      "training_step_time": 7.648348569869995
    },
    {
      "epoch": 5.629539489746093e-06,
      "model_forward_time": 1.7197740077972412,
      "step": 5903
    },
    {
      "epoch": 5.629539489746093e-06,
      "step": 5903,
      "training_step_time": 7.661639213562012
    },
    {
      "epoch": 5.6304931640625e-06,
      "model_forward_time": 2.827162265777588,
      "step": 5904
    },
    {
      "epoch": 5.6304931640625e-06,
      "step": 5904,
      "training_step_time": 9.506932258605957
    },
    {
      "epoch": 5.6314468383789065e-06,
      "model_forward_time": 1.7339627742767334,
      "step": 5905
    },
    {
      "epoch": 5.6314468383789065e-06,
      "step": 5905,
      "training_step_time": 8.69735836982727
    },
    {
      "epoch": 5.632400512695312e-06,
      "model_forward_time": 1.7147712707519531,
      "step": 5906
    },
    {
      "epoch": 5.632400512695312e-06,
      "step": 5906,
      "training_step_time": 8.638093948364258
    },
    {
      "epoch": 5.633354187011719e-06,
      "model_forward_time": 1.720383644104004,
      "step": 5907
    },
    {
      "epoch": 5.633354187011719e-06,
      "step": 5907,
      "training_step_time": 7.643056869506836
    },
    {
      "epoch": 5.634307861328125e-06,
      "model_forward_time": 1.7176930904388428,
      "step": 5908
    },
    {
      "epoch": 5.634307861328125e-06,
      "step": 5908,
      "training_step_time": 7.6804633140563965
    },
    {
      "epoch": 5.635261535644531e-06,
      "model_forward_time": 1.7117810249328613,
      "step": 5909
    },
    {
      "epoch": 5.635261535644531e-06,
      "step": 5909,
      "training_step_time": 7.508632659912109
    },
    {
      "epoch": 5.636215209960938e-06,
      "grad_norm": 0.26452895998954773,
      "learning_rate": 9.99774114888356e-06,
      "loss": 0.0767,
      "step": 5910
    },
    {
      "epoch": 5.636215209960938e-06,
      "model_forward_time": 1.72288179397583,
      "step": 5910
    },
    {
      "dynamics_loss_avg": 0.05325026344507933,
      "epoch": 5.636215209960938e-06,
      "step": 5910
    },
    {
      "action_loss_avg": 0.006115933321416378,
      "epoch": 5.636215209960938e-06,
      "step": 5910
    },
    {
      "epoch": 5.636215209960938e-06,
      "step": 5910,
      "training_step_time": 7.493359088897705
    },
    {
      "epoch": 5.637168884277343e-06,
      "model_forward_time": 1.7187232971191406,
      "step": 5911
    },
    {
      "epoch": 5.637168884277343e-06,
      "step": 5911,
      "training_step_time": 7.617513418197632
    },
    {
      "epoch": 5.63812255859375e-06,
      "model_forward_time": 1.7134308815002441,
      "step": 5912
    },
    {
      "epoch": 5.63812255859375e-06,
      "step": 5912,
      "training_step_time": 7.610525369644165
    },
    {
      "epoch": 5.6390762329101565e-06,
      "model_forward_time": 2.83586049079895,
      "step": 5913
    },
    {
      "epoch": 5.6390762329101565e-06,
      "step": 5913,
      "training_step_time": 8.121331214904785
    },
    {
      "epoch": 5.640029907226562e-06,
      "model_forward_time": 1.70302414894104,
      "step": 5914
    },
    {
      "epoch": 5.640029907226562e-06,
      "step": 5914,
      "training_step_time": 9.598334312438965
    },
    {
      "epoch": 5.640983581542969e-06,
      "model_forward_time": 1.6942577362060547,
      "step": 5915
    },
    {
      "epoch": 5.640983581542969e-06,
      "step": 5915,
      "training_step_time": 8.432392358779907
    },
    {
      "epoch": 5.641937255859375e-06,
      "model_forward_time": 1.6981050968170166,
      "step": 5916
    },
    {
      "epoch": 5.641937255859375e-06,
      "step": 5916,
      "training_step_time": 8.215601444244385
    },
    {
      "epoch": 5.642890930175781e-06,
      "model_forward_time": 1.6966161727905273,
      "step": 5917
    },
    {
      "epoch": 5.642890930175781e-06,
      "step": 5917,
      "training_step_time": 7.703820705413818
    },
    {
      "epoch": 5.643844604492188e-06,
      "model_forward_time": 1.706322431564331,
      "step": 5918
    },
    {
      "epoch": 5.643844604492188e-06,
      "step": 5918,
      "training_step_time": 7.520588636398315
    },
    {
      "epoch": 5.644798278808593e-06,
      "model_forward_time": 1.7017433643341064,
      "step": 5919
    },
    {
      "epoch": 5.644798278808593e-06,
      "step": 5919,
      "training_step_time": 7.527027368545532
    },
    {
      "epoch": 5.645751953125e-06,
      "grad_norm": 0.23835349082946777,
      "learning_rate": 9.997691179659684e-06,
      "loss": 0.0797,
      "step": 5920
    },
    {
      "epoch": 5.645751953125e-06,
      "model_forward_time": 1.6980156898498535,
      "step": 5920
    },
    {
      "dynamics_loss_avg": 0.07103347443044186,
      "epoch": 5.645751953125e-06,
      "step": 5920
    },
    {
      "action_loss_avg": 0.011530378041788936,
      "epoch": 5.645751953125e-06,
      "step": 5920
    },
    {
      "epoch": 5.645751953125e-06,
      "step": 5920,
      "training_step_time": 7.373158931732178
    },
    {
      "epoch": 5.6467056274414065e-06,
      "model_forward_time": 1.706770420074463,
      "step": 5921
    },
    {
      "epoch": 5.6467056274414065e-06,
      "step": 5921,
      "training_step_time": 8.277083396911621
    },
    {
      "epoch": 5.647659301757812e-06,
      "model_forward_time": 2.8072357177734375,
      "step": 5922
    },
    {
      "epoch": 5.647659301757812e-06,
      "step": 5922,
      "training_step_time": 8.566170930862427
    },
    {
      "epoch": 5.648612976074219e-06,
      "model_forward_time": 1.7041058540344238,
      "step": 5923
    },
    {
      "epoch": 5.648612976074219e-06,
      "step": 5923,
      "training_step_time": 7.636211156845093
    },
    {
      "epoch": 5.649566650390625e-06,
      "model_forward_time": 1.7325751781463623,
      "step": 5924
    },
    {
      "epoch": 5.649566650390625e-06,
      "step": 5924,
      "training_step_time": 8.602123498916626
    },
    {
      "epoch": 5.650520324707031e-06,
      "model_forward_time": 1.701054334640503,
      "step": 5925
    },
    {
      "epoch": 5.650520324707031e-06,
      "step": 5925,
      "training_step_time": 8.76039433479309
    },
    {
      "epoch": 5.651473999023438e-06,
      "model_forward_time": 1.705866813659668,
      "step": 5926
    },
    {
      "epoch": 5.651473999023438e-06,
      "step": 5926,
      "training_step_time": 7.87526273727417
    },
    {
      "epoch": 5.652427673339843e-06,
      "model_forward_time": 1.699418067932129,
      "step": 5927
    },
    {
      "epoch": 5.652427673339843e-06,
      "step": 5927,
      "training_step_time": 7.635733366012573
    },
    {
      "epoch": 5.65338134765625e-06,
      "model_forward_time": 1.7010979652404785,
      "step": 5928
    },
    {
      "epoch": 5.65338134765625e-06,
      "step": 5928,
      "training_step_time": 7.6369709968566895
    },
    {
      "epoch": 5.6543350219726565e-06,
      "model_forward_time": 1.710693359375,
      "step": 5929
    },
    {
      "epoch": 5.6543350219726565e-06,
      "step": 5929,
      "training_step_time": 7.459094524383545
    },
    {
      "epoch": 5.655288696289062e-06,
      "grad_norm": 0.26106566190719604,
      "learning_rate": 9.997640663895815e-06,
      "loss": 0.0725,
      "step": 5930
    },
    {
      "epoch": 5.655288696289062e-06,
      "model_forward_time": 1.69809889793396,
      "step": 5930
    },
    {
      "dynamics_loss_avg": 0.0594565961509943,
      "epoch": 5.655288696289062e-06,
      "step": 5930
    },
    {
      "action_loss_avg": 0.005852517765015363,
      "epoch": 5.655288696289062e-06,
      "step": 5930
    },
    {
      "epoch": 5.655288696289062e-06,
      "step": 5930,
      "training_step_time": 7.807108402252197
    },
    {
      "epoch": 5.656242370605469e-06,
      "model_forward_time": 1.7001607418060303,
      "step": 5931
    },
    {
      "epoch": 5.656242370605469e-06,
      "step": 5931,
      "training_step_time": 9.28926968574524
    },
    {
      "epoch": 5.657196044921875e-06,
      "model_forward_time": 1.699293613433838,
      "step": 5932
    },
    {
      "epoch": 5.657196044921875e-06,
      "step": 5932,
      "training_step_time": 8.25114893913269
    },
    {
      "epoch": 5.658149719238281e-06,
      "model_forward_time": 1.7364723682403564,
      "step": 5933
    },
    {
      "epoch": 5.658149719238281e-06,
      "step": 5933,
      "training_step_time": 8.703825235366821
    },
    {
      "epoch": 5.659103393554688e-06,
      "model_forward_time": 1.7219696044921875,
      "step": 5934
    },
    {
      "epoch": 5.659103393554688e-06,
      "step": 5934,
      "training_step_time": 7.708130598068237
    },
    {
      "epoch": 5.660057067871093e-06,
      "model_forward_time": 1.7163844108581543,
      "step": 5935
    },
    {
      "epoch": 5.660057067871093e-06,
      "step": 5935,
      "training_step_time": 7.880728483200073
    },
    {
      "epoch": 5.6610107421875e-06,
      "model_forward_time": 1.698420524597168,
      "step": 5936
    },
    {
      "epoch": 5.6610107421875e-06,
      "step": 5936,
      "training_step_time": 7.661463737487793
    },
    {
      "epoch": 5.6619644165039065e-06,
      "model_forward_time": 1.701263666152954,
      "step": 5937
    },
    {
      "epoch": 5.6619644165039065e-06,
      "step": 5937,
      "training_step_time": 7.6429221630096436
    },
    {
      "epoch": 5.662918090820312e-06,
      "model_forward_time": 1.7193470001220703,
      "step": 5938
    },
    {
      "epoch": 5.662918090820312e-06,
      "step": 5938,
      "training_step_time": 7.500765562057495
    },
    {
      "epoch": 5.663871765136719e-06,
      "model_forward_time": 1.7202246189117432,
      "step": 5939
    },
    {
      "epoch": 5.663871765136719e-06,
      "step": 5939,
      "training_step_time": 7.753199815750122
    },
    {
      "epoch": 5.664825439453125e-06,
      "grad_norm": 0.23483538627624512,
      "learning_rate": 9.997589601597477e-06,
      "loss": 0.0698,
      "step": 5940
    },
    {
      "epoch": 5.664825439453125e-06,
      "model_forward_time": 1.7158968448638916,
      "step": 5940
    },
    {
      "dynamics_loss_avg": 0.08383739665150643,
      "epoch": 5.664825439453125e-06,
      "step": 5940
    },
    {
      "action_loss_avg": 0.006243298749905079,
      "epoch": 5.664825439453125e-06,
      "step": 5940
    },
    {
      "epoch": 5.664825439453125e-06,
      "step": 5940,
      "training_step_time": 8.562860488891602
    },
    {
      "epoch": 5.665779113769531e-06,
      "model_forward_time": 2.875718832015991,
      "step": 5941
    },
    {
      "epoch": 5.665779113769531e-06,
      "step": 5941,
      "training_step_time": 8.734073877334595
    },
    {
      "epoch": 5.666732788085938e-06,
      "model_forward_time": 1.748579978942871,
      "step": 5942
    },
    {
      "epoch": 5.666732788085938e-06,
      "step": 5942,
      "training_step_time": 8.656280994415283
    },
    {
      "epoch": 5.667686462402343e-06,
      "model_forward_time": 1.715118169784546,
      "step": 5943
    },
    {
      "epoch": 5.667686462402343e-06,
      "step": 5943,
      "training_step_time": 7.66682243347168
    },
    {
      "epoch": 5.66864013671875e-06,
      "model_forward_time": 1.7301039695739746,
      "step": 5944
    },
    {
      "epoch": 5.66864013671875e-06,
      "step": 5944,
      "training_step_time": 8.413818120956421
    },
    {
      "epoch": 5.6695938110351565e-06,
      "model_forward_time": 1.7172489166259766,
      "step": 5945
    },
    {
      "epoch": 5.6695938110351565e-06,
      "step": 5945,
      "training_step_time": 7.8157641887664795
    },
    {
      "epoch": 5.670547485351562e-06,
      "model_forward_time": 1.7135560512542725,
      "step": 5946
    },
    {
      "epoch": 5.670547485351562e-06,
      "step": 5946,
      "training_step_time": 7.595091104507446
    },
    {
      "epoch": 5.671501159667969e-06,
      "model_forward_time": 1.6921193599700928,
      "step": 5947
    },
    {
      "epoch": 5.671501159667969e-06,
      "step": 5947,
      "training_step_time": 7.410583257675171
    },
    {
      "epoch": 5.672454833984375e-06,
      "model_forward_time": 1.7070212364196777,
      "step": 5948
    },
    {
      "epoch": 5.672454833984375e-06,
      "step": 5948,
      "training_step_time": 7.8088696002960205
    },
    {
      "epoch": 5.673408508300781e-06,
      "model_forward_time": 1.7023186683654785,
      "step": 5949
    },
    {
      "epoch": 5.673408508300781e-06,
      "step": 5949,
      "training_step_time": 7.711571455001831
    },
    {
      "epoch": 5.674362182617188e-06,
      "grad_norm": 0.2356475591659546,
      "learning_rate": 9.997537992770251e-06,
      "loss": 0.0733,
      "step": 5950
    },
    {
      "epoch": 5.674362182617188e-06,
      "model_forward_time": 2.8456013202667236,
      "step": 5950
    },
    {
      "dynamics_loss_avg": 0.05448809824883938,
      "epoch": 5.674362182617188e-06,
      "step": 5950
    },
    {
      "action_loss_avg": 0.004473156633321196,
      "epoch": 5.674362182617188e-06,
      "step": 5950
    },
    {
      "epoch": 5.674362182617188e-06,
      "step": 5950,
      "training_step_time": 8.855384111404419
    },
    {
      "epoch": 5.675315856933593e-06,
      "model_forward_time": 1.71181321144104,
      "step": 5951
    },
    {
      "epoch": 5.675315856933593e-06,
      "step": 5951,
      "training_step_time": 8.398054122924805
    },
    {
      "epoch": 5.67626953125e-06,
      "model_forward_time": 1.7151939868927002,
      "step": 5952
    },
    {
      "epoch": 5.67626953125e-06,
      "step": 5952,
      "training_step_time": 7.928600788116455
    },
    {
      "epoch": 5.6772232055664065e-06,
      "model_forward_time": 1.7014546394348145,
      "step": 5953
    },
    {
      "epoch": 5.6772232055664065e-06,
      "step": 5953,
      "training_step_time": 7.724700212478638
    },
    {
      "epoch": 5.678176879882812e-06,
      "model_forward_time": 1.7082862854003906,
      "step": 5954
    },
    {
      "epoch": 5.678176879882812e-06,
      "step": 5954,
      "training_step_time": 7.841771602630615
    },
    {
      "epoch": 5.679130554199219e-06,
      "model_forward_time": 1.7114512920379639,
      "step": 5955
    },
    {
      "epoch": 5.679130554199219e-06,
      "step": 5955,
      "training_step_time": 7.248559236526489
    },
    {
      "epoch": 5.680084228515625e-06,
      "model_forward_time": 1.6993732452392578,
      "step": 5956
    },
    {
      "epoch": 5.680084228515625e-06,
      "step": 5956,
      "training_step_time": 7.672857284545898
    },
    {
      "epoch": 5.681037902832031e-06,
      "model_forward_time": 1.7025682926177979,
      "step": 5957
    },
    {
      "epoch": 5.681037902832031e-06,
      "step": 5957,
      "training_step_time": 7.76559591293335
    },
    {
      "epoch": 5.681991577148438e-06,
      "model_forward_time": 1.7119238376617432,
      "step": 5958
    },
    {
      "epoch": 5.681991577148438e-06,
      "step": 5958,
      "training_step_time": 8.558775901794434
    },
    {
      "epoch": 5.682945251464843e-06,
      "model_forward_time": 2.9073328971862793,
      "step": 5959
    },
    {
      "epoch": 5.682945251464843e-06,
      "step": 5959,
      "training_step_time": 7.794230222702026
    },
    {
      "epoch": 5.68389892578125e-06,
      "grad_norm": 0.22076843678951263,
      "learning_rate": 9.997485837419788e-06,
      "loss": 0.0713,
      "step": 5960
    },
    {
      "epoch": 5.68389892578125e-06,
      "model_forward_time": 1.706315517425537,
      "step": 5960
    },
    {
      "dynamics_loss_avg": 0.048859434016048905,
      "epoch": 5.68389892578125e-06,
      "step": 5960
    },
    {
      "action_loss_avg": 0.0038769242586568,
      "epoch": 5.68389892578125e-06,
      "step": 5960
    },
    {
      "epoch": 5.68389892578125e-06,
      "step": 5960,
      "training_step_time": 7.6652445793151855
    },
    {
      "epoch": 5.6848526000976565e-06,
      "model_forward_time": 1.7079203128814697,
      "step": 5961
    },
    {
      "epoch": 5.6848526000976565e-06,
      "step": 5961,
      "training_step_time": 7.714092016220093
    },
    {
      "epoch": 5.685806274414062e-06,
      "model_forward_time": 1.701918601989746,
      "step": 5962
    },
    {
      "epoch": 5.685806274414062e-06,
      "step": 5962,
      "training_step_time": 8.70651888847351
    },
    {
      "epoch": 5.686759948730469e-06,
      "model_forward_time": 1.7137672901153564,
      "step": 5963
    },
    {
      "epoch": 5.686759948730469e-06,
      "step": 5963,
      "training_step_time": 7.844904899597168
    },
    {
      "epoch": 5.687713623046875e-06,
      "model_forward_time": 1.703200340270996,
      "step": 5964
    },
    {
      "epoch": 5.687713623046875e-06,
      "step": 5964,
      "training_step_time": 7.675355434417725
    },
    {
      "epoch": 5.688667297363281e-06,
      "model_forward_time": 1.701996088027954,
      "step": 5965
    },
    {
      "epoch": 5.688667297363281e-06,
      "step": 5965,
      "training_step_time": 7.682131767272949
    },
    {
      "epoch": 5.689620971679688e-06,
      "model_forward_time": 1.7075307369232178,
      "step": 5966
    },
    {
      "epoch": 5.689620971679688e-06,
      "step": 5966,
      "training_step_time": 8.553926944732666
    },
    {
      "epoch": 5.690574645996093e-06,
      "model_forward_time": 1.6879832744598389,
      "step": 5967
    },
    {
      "epoch": 5.690574645996093e-06,
      "step": 5967,
      "training_step_time": 9.180979490280151
    },
    {
      "epoch": 5.6915283203125e-06,
      "model_forward_time": 2.976071357727051,
      "step": 5968
    },
    {
      "epoch": 5.6915283203125e-06,
      "step": 5968,
      "training_step_time": 7.820435285568237
    },
    {
      "epoch": 5.6924819946289065e-06,
      "model_forward_time": 1.704699993133545,
      "step": 5969
    },
    {
      "epoch": 5.6924819946289065e-06,
      "step": 5969,
      "training_step_time": 8.856452465057373
    },
    {
      "epoch": 5.693435668945312e-06,
      "grad_norm": 0.24345414340496063,
      "learning_rate": 9.997433135551786e-06,
      "loss": 0.0658,
      "step": 5970
    },
    {
      "epoch": 5.693435668945312e-06,
      "model_forward_time": 1.7319135665893555,
      "step": 5970
    },
    {
      "dynamics_loss_avg": 0.04796219449490309,
      "epoch": 5.693435668945312e-06,
      "step": 5970
    },
    {
      "action_loss_avg": 0.004603986674919724,
      "epoch": 5.693435668945312e-06,
      "step": 5970
    },
    {
      "epoch": 5.693435668945312e-06,
      "step": 5970,
      "training_step_time": 8.732613563537598
    },
    {
      "epoch": 5.694389343261719e-06,
      "model_forward_time": 1.6876161098480225,
      "step": 5971
    },
    {
      "epoch": 5.694389343261719e-06,
      "step": 5971,
      "training_step_time": 7.837491750717163
    },
    {
      "epoch": 5.695343017578125e-06,
      "model_forward_time": 1.7099597454071045,
      "step": 5972
    },
    {
      "epoch": 5.695343017578125e-06,
      "step": 5972,
      "training_step_time": 7.79862642288208
    },
    {
      "epoch": 5.696296691894531e-06,
      "model_forward_time": 1.7084007263183594,
      "step": 5973
    },
    {
      "epoch": 5.696296691894531e-06,
      "step": 5973,
      "training_step_time": 7.756586074829102
    },
    {
      "epoch": 5.697250366210938e-06,
      "model_forward_time": 1.7079029083251953,
      "step": 5974
    },
    {
      "epoch": 5.697250366210938e-06,
      "step": 5974,
      "training_step_time": 7.632697105407715
    },
    {
      "epoch": 5.698204040527343e-06,
      "model_forward_time": 1.7084667682647705,
      "step": 5975
    },
    {
      "epoch": 5.698204040527343e-06,
      "step": 5975,
      "training_step_time": 7.548982620239258
    },
    {
      "epoch": 5.69915771484375e-06,
      "model_forward_time": 1.7072765827178955,
      "step": 5976
    },
    {
      "epoch": 5.69915771484375e-06,
      "step": 5976,
      "training_step_time": 8.758557558059692
    },
    {
      "epoch": 5.7001113891601565e-06,
      "model_forward_time": 2.8789238929748535,
      "step": 5977
    },
    {
      "epoch": 5.7001113891601565e-06,
      "step": 5977,
      "training_step_time": 7.799872159957886
    },
    {
      "epoch": 5.701065063476562e-06,
      "model_forward_time": 1.7227230072021484,
      "step": 5978
    },
    {
      "epoch": 5.701065063476562e-06,
      "step": 5978,
      "training_step_time": 8.609413146972656
    },
    {
      "epoch": 5.702018737792969e-06,
      "model_forward_time": 1.7250103950500488,
      "step": 5979
    },
    {
      "epoch": 5.702018737792969e-06,
      "step": 5979,
      "training_step_time": 9.522686958312988
    },
    {
      "epoch": 5.702972412109375e-06,
      "grad_norm": 0.2462553232908249,
      "learning_rate": 9.99737988717201e-06,
      "loss": 0.0682,
      "step": 5980
    },
    {
      "epoch": 5.702972412109375e-06,
      "model_forward_time": 1.7030279636383057,
      "step": 5980
    },
    {
      "dynamics_loss_avg": 0.07304980047047138,
      "epoch": 5.702972412109375e-06,
      "step": 5980
    },
    {
      "action_loss_avg": 0.006403922871686518,
      "epoch": 5.702972412109375e-06,
      "step": 5980
    },
    {
      "epoch": 5.702972412109375e-06,
      "step": 5980,
      "training_step_time": 7.8602728843688965
    },
    {
      "epoch": 5.703926086425781e-06,
      "model_forward_time": 1.705397367477417,
      "step": 5981
    },
    {
      "epoch": 5.703926086425781e-06,
      "step": 5981,
      "training_step_time": 7.838793516159058
    },
    {
      "epoch": 5.704879760742188e-06,
      "model_forward_time": 1.7007756233215332,
      "step": 5982
    },
    {
      "epoch": 5.704879760742188e-06,
      "step": 5982,
      "training_step_time": 7.865569591522217
    },
    {
      "epoch": 5.705833435058593e-06,
      "model_forward_time": 1.7113616466522217,
      "step": 5983
    },
    {
      "epoch": 5.705833435058593e-06,
      "step": 5983,
      "training_step_time": 7.673524379730225
    },
    {
      "epoch": 5.706787109375e-06,
      "model_forward_time": 1.6996374130249023,
      "step": 5984
    },
    {
      "epoch": 5.706787109375e-06,
      "step": 5984,
      "training_step_time": 7.445401191711426
    },
    {
      "epoch": 5.7077407836914066e-06,
      "model_forward_time": 1.700899600982666,
      "step": 5985
    },
    {
      "epoch": 5.7077407836914066e-06,
      "step": 5985,
      "training_step_time": 7.788732051849365
    },
    {
      "epoch": 5.708694458007812e-06,
      "model_forward_time": 2.5568900108337402,
      "step": 5986
    },
    {
      "epoch": 5.708694458007812e-06,
      "step": 5986,
      "training_step_time": 7.640767335891724
    },
    {
      "epoch": 5.709648132324219e-06,
      "model_forward_time": 1.7215378284454346,
      "step": 5987
    },
    {
      "epoch": 5.709648132324219e-06,
      "step": 5987,
      "training_step_time": 7.640778541564941
    },
    {
      "epoch": 5.710601806640625e-06,
      "model_forward_time": 1.7235004901885986,
      "step": 5988
    },
    {
      "epoch": 5.710601806640625e-06,
      "step": 5988,
      "training_step_time": 7.768002986907959
    },
    {
      "epoch": 5.711555480957031e-06,
      "model_forward_time": 1.7085864543914795,
      "step": 5989
    },
    {
      "epoch": 5.711555480957031e-06,
      "step": 5989,
      "training_step_time": 8.277711868286133
    },
    {
      "epoch": 5.712509155273438e-06,
      "grad_norm": 0.16857881844043732,
      "learning_rate": 9.997326092286282e-06,
      "loss": 0.0696,
      "step": 5990
    },
    {
      "epoch": 5.712509155273438e-06,
      "model_forward_time": 1.7032670974731445,
      "step": 5990
    },
    {
      "dynamics_loss_avg": 0.06830182783305645,
      "epoch": 5.712509155273438e-06,
      "step": 5990
    },
    {
      "action_loss_avg": 0.005710746208205819,
      "epoch": 5.712509155273438e-06,
      "step": 5990
    },
    {
      "epoch": 5.712509155273438e-06,
      "step": 5990,
      "training_step_time": 9.504826545715332
    },
    {
      "epoch": 5.7134628295898434e-06,
      "model_forward_time": 1.7064225673675537,
      "step": 5991
    },
    {
      "epoch": 5.7134628295898434e-06,
      "step": 5991,
      "training_step_time": 7.858920574188232
    },
    {
      "epoch": 5.71441650390625e-06,
      "model_forward_time": 1.7003259658813477,
      "step": 5992
    },
    {
      "epoch": 5.71441650390625e-06,
      "step": 5992,
      "training_step_time": 7.616412162780762
    },
    {
      "epoch": 5.7153701782226566e-06,
      "model_forward_time": 1.699674367904663,
      "step": 5993
    },
    {
      "epoch": 5.7153701782226566e-06,
      "step": 5993,
      "training_step_time": 7.405824899673462
    },
    {
      "epoch": 5.716323852539062e-06,
      "model_forward_time": 1.701164722442627,
      "step": 5994
    },
    {
      "epoch": 5.716323852539062e-06,
      "step": 5994,
      "training_step_time": 8.765743255615234
    },
    {
      "epoch": 5.717277526855469e-06,
      "model_forward_time": 2.903028964996338,
      "step": 5995
    },
    {
      "epoch": 5.717277526855469e-06,
      "step": 5995,
      "training_step_time": 7.642942905426025
    },
    {
      "epoch": 5.718231201171875e-06,
      "model_forward_time": 1.7093491554260254,
      "step": 5996
    },
    {
      "epoch": 5.718231201171875e-06,
      "step": 5996,
      "training_step_time": 8.627445220947266
    },
    {
      "epoch": 5.719184875488281e-06,
      "model_forward_time": 1.7195255756378174,
      "step": 5997
    },
    {
      "epoch": 5.719184875488281e-06,
      "step": 5997,
      "training_step_time": 10.544981479644775
    },
    {
      "epoch": 5.720138549804688e-06,
      "model_forward_time": 1.7540364265441895,
      "step": 5998
    },
    {
      "epoch": 5.720138549804688e-06,
      "step": 5998,
      "training_step_time": 8.107577800750732
    },
    {
      "epoch": 5.7210922241210934e-06,
      "model_forward_time": 1.7127110958099365,
      "step": 5999
    },
    {
      "epoch": 5.7210922241210934e-06,
      "step": 5999,
      "training_step_time": 7.882641315460205
    },
    {
      "epoch": 5.7220458984375e-06,
      "grad_norm": 0.3488529324531555,
      "learning_rate": 9.997271750900485e-06,
      "loss": 0.0708,
      "step": 6000
    },
    {
      "epoch": 5.7220458984375e-06,
      "model_forward_time": 1.7262754440307617,
      "step": 6000
    },
    {
      "dynamics_loss_avg": 0.05258167888969183,
      "epoch": 5.7220458984375e-06,
      "step": 6000
    },
    {
      "action_loss_avg": 0.004891031538136303,
      "epoch": 5.7220458984375e-06,
      "step": 6000
    },
    {
      "epoch": 5.7220458984375e-06,
      "step": 6000,
      "training_step_time": 7.502185583114624
    },
    {
      "epoch": 5.7229995727539066e-06,
      "model_forward_time": 1.744544267654419,
      "step": 6001
    },
    {
      "epoch": 5.7229995727539066e-06,
      "step": 6001,
      "training_step_time": 7.524776935577393
    },
    {
      "epoch": 5.723953247070312e-06,
      "model_forward_time": 1.7521357536315918,
      "step": 6002
    },
    {
      "epoch": 5.723953247070312e-06,
      "step": 6002,
      "training_step_time": 6.781016111373901
    },
    {
      "epoch": 5.724906921386719e-06,
      "model_forward_time": 1.7840919494628906,
      "step": 6003
    },
    {
      "epoch": 5.724906921386719e-06,
      "step": 6003,
      "training_step_time": 10.235252141952515
    },
    {
      "epoch": 5.725860595703125e-06,
      "model_forward_time": 1.7523512840270996,
      "step": 6004
    },
    {
      "epoch": 5.725860595703125e-06,
      "step": 6004,
      "training_step_time": 9.889899253845215
    },
    {
      "epoch": 5.726814270019531e-06,
      "model_forward_time": 1.7471415996551514,
      "step": 6005
    },
    {
      "epoch": 5.726814270019531e-06,
      "step": 6005,
      "training_step_time": 8.50387191772461
    },
    {
      "epoch": 5.727767944335938e-06,
      "model_forward_time": 1.749220609664917,
      "step": 6006
    },
    {
      "epoch": 5.727767944335938e-06,
      "step": 6006,
      "training_step_time": 9.34091830253601
    },
    {
      "epoch": 5.7287216186523434e-06,
      "model_forward_time": 1.7590844631195068,
      "step": 6007
    },
    {
      "epoch": 5.7287216186523434e-06,
      "step": 6007,
      "training_step_time": 9.70448112487793
    },
    {
      "epoch": 5.72967529296875e-06,
      "model_forward_time": 1.7512142658233643,
      "step": 6008
    },
    {
      "epoch": 5.72967529296875e-06,
      "step": 6008,
      "training_step_time": 8.572508096694946
    },
    {
      "epoch": 5.7306289672851566e-06,
      "model_forward_time": 1.7480649948120117,
      "step": 6009
    },
    {
      "epoch": 5.7306289672851566e-06,
      "step": 6009,
      "training_step_time": 7.718970060348511
    },
    {
      "epoch": 5.731582641601562e-06,
      "grad_norm": 0.3747994303703308,
      "learning_rate": 9.997216863020564e-06,
      "loss": 0.072,
      "step": 6010
    },
    {
      "epoch": 5.731582641601562e-06,
      "model_forward_time": 1.3905563354492188,
      "step": 6010
    },
    {
      "dynamics_loss_avg": 0.06289097107946873,
      "epoch": 5.731582641601562e-06,
      "step": 6010
    },
    {
      "action_loss_avg": 0.00962067455984652,
      "epoch": 5.731582641601562e-06,
      "step": 6010
    },
    {
      "epoch": 5.731582641601562e-06,
      "step": 6010,
      "training_step_time": 7.4614527225494385
    },
    {
      "epoch": 5.732536315917969e-06,
      "model_forward_time": 1.747164011001587,
      "step": 6011
    },
    {
      "epoch": 5.732536315917969e-06,
      "step": 6011,
      "training_step_time": 6.697338581085205
    },
    {
      "epoch": 5.733489990234375e-06,
      "model_forward_time": 1.7550926208496094,
      "step": 6012
    },
    {
      "epoch": 5.733489990234375e-06,
      "step": 6012,
      "training_step_time": 7.558549404144287
    },
    {
      "epoch": 5.734443664550781e-06,
      "model_forward_time": 2.2653801441192627,
      "step": 6013
    },
    {
      "epoch": 5.734443664550781e-06,
      "step": 6013,
      "training_step_time": 8.55239200592041
    },
    {
      "epoch": 5.735397338867188e-06,
      "model_forward_time": 1.737382173538208,
      "step": 6014
    },
    {
      "epoch": 5.735397338867188e-06,
      "step": 6014,
      "training_step_time": 8.299211978912354
    },
    {
      "epoch": 5.7363510131835934e-06,
      "model_forward_time": 1.7399656772613525,
      "step": 6015
    },
    {
      "epoch": 5.7363510131835934e-06,
      "step": 6015,
      "training_step_time": 9.314734697341919
    },
    {
      "epoch": 5.7373046875e-06,
      "model_forward_time": 1.7377164363861084,
      "step": 6016
    },
    {
      "epoch": 5.7373046875e-06,
      "step": 6016,
      "training_step_time": 7.77553653717041
    },
    {
      "epoch": 5.738258361816407e-06,
      "model_forward_time": 1.7379262447357178,
      "step": 6017
    },
    {
      "epoch": 5.738258361816407e-06,
      "step": 6017,
      "training_step_time": 8.694201231002808
    },
    {
      "epoch": 5.739212036132812e-06,
      "model_forward_time": 1.7362854480743408,
      "step": 6018
    },
    {
      "epoch": 5.739212036132812e-06,
      "step": 6018,
      "training_step_time": 7.688854217529297
    },
    {
      "epoch": 5.740165710449219e-06,
      "model_forward_time": 1.7452542781829834,
      "step": 6019
    },
    {
      "epoch": 5.740165710449219e-06,
      "step": 6019,
      "training_step_time": 7.745857238769531
    },
    {
      "epoch": 5.741119384765625e-06,
      "grad_norm": 0.5134737491607666,
      "learning_rate": 9.99716142865252e-06,
      "loss": 0.0759,
      "step": 6020
    },
    {
      "epoch": 5.741119384765625e-06,
      "model_forward_time": 1.73850417137146,
      "step": 6020
    },
    {
      "dynamics_loss_avg": 0.06315434277057648,
      "epoch": 5.741119384765625e-06,
      "step": 6020
    },
    {
      "action_loss_avg": 0.006934662186540663,
      "epoch": 5.741119384765625e-06,
      "step": 6020
    },
    {
      "epoch": 5.741119384765625e-06,
      "step": 6020,
      "training_step_time": 7.533467769622803
    },
    {
      "epoch": 5.742073059082031e-06,
      "model_forward_time": 1.7431974411010742,
      "step": 6021
    },
    {
      "epoch": 5.742073059082031e-06,
      "step": 6021,
      "training_step_time": 8.362308025360107
    },
    {
      "epoch": 5.743026733398438e-06,
      "model_forward_time": 2.265850067138672,
      "step": 6022
    },
    {
      "epoch": 5.743026733398438e-06,
      "step": 6022,
      "training_step_time": 7.69257664680481
    },
    {
      "epoch": 5.7439804077148435e-06,
      "model_forward_time": 1.7426741123199463,
      "step": 6023
    },
    {
      "epoch": 5.7439804077148435e-06,
      "step": 6023,
      "training_step_time": 8.603739976882935
    },
    {
      "epoch": 5.74493408203125e-06,
      "model_forward_time": 1.7330853939056396,
      "step": 6024
    },
    {
      "epoch": 5.74493408203125e-06,
      "step": 6024,
      "training_step_time": 9.193889141082764
    },
    {
      "epoch": 5.745887756347657e-06,
      "model_forward_time": 1.7313625812530518,
      "step": 6025
    },
    {
      "epoch": 5.745887756347657e-06,
      "step": 6025,
      "training_step_time": 8.429490566253662
    },
    {
      "epoch": 5.746841430664062e-06,
      "model_forward_time": 1.7368078231811523,
      "step": 6026
    },
    {
      "epoch": 5.746841430664062e-06,
      "step": 6026,
      "training_step_time": 8.687940835952759
    },
    {
      "epoch": 5.747795104980469e-06,
      "model_forward_time": 1.734816074371338,
      "step": 6027
    },
    {
      "epoch": 5.747795104980469e-06,
      "step": 6027,
      "training_step_time": 7.703805446624756
    },
    {
      "epoch": 5.748748779296875e-06,
      "model_forward_time": 1.7362887859344482,
      "step": 6028
    },
    {
      "epoch": 5.748748779296875e-06,
      "step": 6028,
      "training_step_time": 7.733070611953735
    },
    {
      "epoch": 5.749702453613281e-06,
      "model_forward_time": 1.7390453815460205,
      "step": 6029
    },
    {
      "epoch": 5.749702453613281e-06,
      "step": 6029,
      "training_step_time": 7.582678556442261
    },
    {
      "epoch": 5.750656127929688e-06,
      "grad_norm": 0.4021722674369812,
      "learning_rate": 9.997105447802416e-06,
      "loss": 0.0789,
      "step": 6030
    },
    {
      "epoch": 5.750656127929688e-06,
      "model_forward_time": 1.7367510795593262,
      "step": 6030
    },
    {
      "dynamics_loss_avg": 0.06096797864884138,
      "epoch": 5.750656127929688e-06,
      "step": 6030
    },
    {
      "action_loss_avg": 0.006836471846327186,
      "epoch": 5.750656127929688e-06,
      "step": 6030
    },
    {
      "epoch": 5.750656127929688e-06,
      "step": 6030,
      "training_step_time": 7.260530948638916
    },
    {
      "epoch": 5.7516098022460935e-06,
      "model_forward_time": 2.2972967624664307,
      "step": 6031
    },
    {
      "epoch": 5.7516098022460935e-06,
      "step": 6031,
      "training_step_time": 7.755225419998169
    },
    {
      "epoch": 5.7525634765625e-06,
      "model_forward_time": 1.7308883666992188,
      "step": 6032
    },
    {
      "epoch": 5.7525634765625e-06,
      "step": 6032,
      "training_step_time": 7.565741300582886
    },
    {
      "epoch": 5.753517150878907e-06,
      "model_forward_time": 1.7323048114776611,
      "step": 6033
    },
    {
      "epoch": 5.753517150878907e-06,
      "step": 6033,
      "training_step_time": 8.263721942901611
    },
    {
      "epoch": 5.754470825195312e-06,
      "model_forward_time": 1.7271418571472168,
      "step": 6034
    },
    {
      "epoch": 5.754470825195312e-06,
      "step": 6034,
      "training_step_time": 7.5535008907318115
    },
    {
      "epoch": 5.755424499511719e-06,
      "model_forward_time": 1.7350969314575195,
      "step": 6035
    },
    {
      "epoch": 5.755424499511719e-06,
      "step": 6035,
      "training_step_time": 8.606001377105713
    },
    {
      "epoch": 5.756378173828125e-06,
      "model_forward_time": 1.7387762069702148,
      "step": 6036
    },
    {
      "epoch": 5.756378173828125e-06,
      "step": 6036,
      "training_step_time": 9.744059801101685
    },
    {
      "epoch": 5.757331848144531e-06,
      "model_forward_time": 1.345747709274292,
      "step": 6037
    },
    {
      "epoch": 5.757331848144531e-06,
      "step": 6037,
      "training_step_time": 7.744692802429199
    },
    {
      "epoch": 5.758285522460938e-06,
      "model_forward_time": 1.7381577491760254,
      "step": 6038
    },
    {
      "epoch": 5.758285522460938e-06,
      "step": 6038,
      "training_step_time": 7.5960729122161865
    },
    {
      "epoch": 5.7592391967773435e-06,
      "model_forward_time": 1.7362191677093506,
      "step": 6039
    },
    {
      "epoch": 5.7592391967773435e-06,
      "step": 6039,
      "training_step_time": 7.35142970085144
    },
    {
      "epoch": 5.76019287109375e-06,
      "grad_norm": 0.2804267704486847,
      "learning_rate": 9.997048920476373e-06,
      "loss": 0.0843,
      "step": 6040
    },
    {
      "epoch": 5.76019287109375e-06,
      "model_forward_time": 2.314669132232666,
      "step": 6040
    },
    {
      "dynamics_loss_avg": 0.07178503256291151,
      "epoch": 5.76019287109375e-06,
      "step": 6040
    },
    {
      "action_loss_avg": 0.008517798874527216,
      "epoch": 5.76019287109375e-06,
      "step": 6040
    },
    {
      "epoch": 5.76019287109375e-06,
      "step": 6040,
      "training_step_time": 7.6303935050964355
    },
    {
      "epoch": 5.761146545410157e-06,
      "model_forward_time": 1.7404606342315674,
      "step": 6041
    },
    {
      "epoch": 5.761146545410157e-06,
      "step": 6041,
      "training_step_time": 7.53242039680481
    },
    {
      "epoch": 5.762100219726562e-06,
      "model_forward_time": 1.7373683452606201,
      "step": 6042
    },
    {
      "epoch": 5.762100219726562e-06,
      "step": 6042,
      "training_step_time": 7.673932790756226
    },
    {
      "epoch": 5.763053894042969e-06,
      "model_forward_time": 1.729562759399414,
      "step": 6043
    },
    {
      "epoch": 5.763053894042969e-06,
      "step": 6043,
      "training_step_time": 8.419439315795898
    },
    {
      "epoch": 5.764007568359375e-06,
      "model_forward_time": 1.7390170097351074,
      "step": 6044
    },
    {
      "epoch": 5.764007568359375e-06,
      "step": 6044,
      "training_step_time": 8.348345518112183
    },
    {
      "epoch": 5.764961242675781e-06,
      "model_forward_time": 1.7370002269744873,
      "step": 6045
    },
    {
      "epoch": 5.764961242675781e-06,
      "step": 6045,
      "training_step_time": 9.318127393722534
    },
    {
      "epoch": 5.765914916992188e-06,
      "model_forward_time": 1.7376933097839355,
      "step": 6046
    },
    {
      "epoch": 5.765914916992188e-06,
      "step": 6046,
      "training_step_time": 7.895828485488892
    },
    {
      "epoch": 5.7668685913085935e-06,
      "model_forward_time": 1.7338824272155762,
      "step": 6047
    },
    {
      "epoch": 5.7668685913085935e-06,
      "step": 6047,
      "training_step_time": 7.648473739624023
    },
    {
      "epoch": 5.767822265625e-06,
      "model_forward_time": 1.7335495948791504,
      "step": 6048
    },
    {
      "epoch": 5.767822265625e-06,
      "step": 6048,
      "training_step_time": 7.524720191955566
    },
    {
      "epoch": 5.768775939941407e-06,
      "model_forward_time": 1.8146343231201172,
      "step": 6049
    },
    {
      "epoch": 5.768775939941407e-06,
      "step": 6049,
      "training_step_time": 9.261812686920166
    },
    {
      "epoch": 5.769729614257812e-06,
      "grad_norm": 0.41569575667381287,
      "learning_rate": 9.996991846680573e-06,
      "loss": 0.0793,
      "step": 6050
    },
    {
      "epoch": 5.769729614257812e-06,
      "model_forward_time": 2.267005443572998,
      "step": 6050
    },
    {
      "dynamics_loss_avg": 0.08611148428171873,
      "epoch": 5.769729614257812e-06,
      "step": 6050
    },
    {
      "action_loss_avg": 0.008844701235648244,
      "epoch": 5.769729614257812e-06,
      "step": 6050
    },
    {
      "epoch": 5.769729614257812e-06,
      "step": 6050,
      "training_step_time": 8.557097911834717
    },
    {
      "epoch": 5.770683288574219e-06,
      "model_forward_time": 1.7378079891204834,
      "step": 6051
    },
    {
      "epoch": 5.770683288574219e-06,
      "step": 6051,
      "training_step_time": 7.787277936935425
    },
    {
      "epoch": 5.771636962890625e-06,
      "model_forward_time": 1.728844404220581,
      "step": 6052
    },
    {
      "epoch": 5.771636962890625e-06,
      "step": 6052,
      "training_step_time": 8.233697175979614
    },
    {
      "epoch": 5.772590637207031e-06,
      "model_forward_time": 1.3608148097991943,
      "step": 6053
    },
    {
      "epoch": 5.772590637207031e-06,
      "step": 6053,
      "training_step_time": 7.786156177520752
    },
    {
      "epoch": 5.773544311523438e-06,
      "model_forward_time": 1.7369349002838135,
      "step": 6054
    },
    {
      "epoch": 5.773544311523438e-06,
      "step": 6054,
      "training_step_time": 7.771744251251221
    },
    {
      "epoch": 5.7744979858398435e-06,
      "model_forward_time": 1.7375528812408447,
      "step": 6055
    },
    {
      "epoch": 5.7744979858398435e-06,
      "step": 6055,
      "training_step_time": 8.512756586074829
    },
    {
      "epoch": 5.77545166015625e-06,
      "model_forward_time": 1.7387011051177979,
      "step": 6056
    },
    {
      "epoch": 5.77545166015625e-06,
      "step": 6056,
      "training_step_time": 7.6935412883758545
    },
    {
      "epoch": 5.776405334472657e-06,
      "model_forward_time": 1.7368109226226807,
      "step": 6057
    },
    {
      "epoch": 5.776405334472657e-06,
      "step": 6057,
      "training_step_time": 8.513880729675293
    },
    {
      "epoch": 5.777359008789062e-06,
      "model_forward_time": 1.8157947063446045,
      "step": 6058
    },
    {
      "epoch": 5.777359008789062e-06,
      "step": 6058,
      "training_step_time": 7.767514228820801
    },
    {
      "epoch": 5.778312683105469e-06,
      "model_forward_time": 2.2782301902770996,
      "step": 6059
    },
    {
      "epoch": 5.778312683105469e-06,
      "step": 6059,
      "training_step_time": 7.788790225982666
    },
    {
      "epoch": 5.779266357421875e-06,
      "grad_norm": 0.40251898765563965,
      "learning_rate": 9.996934226421257e-06,
      "loss": 0.08,
      "step": 6060
    },
    {
      "epoch": 5.779266357421875e-06,
      "model_forward_time": 1.7310266494750977,
      "step": 6060
    },
    {
      "dynamics_loss_avg": 0.07239250913262367,
      "epoch": 5.779266357421875e-06,
      "step": 6060
    },
    {
      "action_loss_avg": 0.010364011023193597,
      "epoch": 5.779266357421875e-06,
      "step": 6060
    },
    {
      "epoch": 5.779266357421875e-06,
      "step": 6060,
      "training_step_time": 7.6221020221710205
    },
    {
      "epoch": 5.780220031738281e-06,
      "model_forward_time": 1.7513184547424316,
      "step": 6061
    },
    {
      "epoch": 5.780220031738281e-06,
      "step": 6061,
      "training_step_time": 8.46561861038208
    },
    {
      "epoch": 5.781173706054688e-06,
      "model_forward_time": 1.7378451824188232,
      "step": 6062
    },
    {
      "epoch": 5.781173706054688e-06,
      "step": 6062,
      "training_step_time": 8.561396598815918
    },
    {
      "epoch": 5.7821273803710935e-06,
      "model_forward_time": 1.7429957389831543,
      "step": 6063
    },
    {
      "epoch": 5.7821273803710935e-06,
      "step": 6063,
      "training_step_time": 7.843393325805664
    },
    {
      "epoch": 5.7830810546875e-06,
      "model_forward_time": 1.7433977127075195,
      "step": 6064
    },
    {
      "epoch": 5.7830810546875e-06,
      "step": 6064,
      "training_step_time": 7.856145143508911
    },
    {
      "epoch": 5.784034729003907e-06,
      "model_forward_time": 1.7447826862335205,
      "step": 6065
    },
    {
      "epoch": 5.784034729003907e-06,
      "step": 6065,
      "training_step_time": 8.458667993545532
    },
    {
      "epoch": 5.784988403320312e-06,
      "model_forward_time": 1.7289457321166992,
      "step": 6066
    },
    {
      "epoch": 5.784988403320312e-06,
      "step": 6066,
      "training_step_time": 7.702555894851685
    },
    {
      "epoch": 5.785942077636719e-06,
      "model_forward_time": 1.8072319030761719,
      "step": 6067
    },
    {
      "epoch": 5.785942077636719e-06,
      "step": 6067,
      "training_step_time": 7.791022539138794
    },
    {
      "epoch": 5.786895751953125e-06,
      "model_forward_time": 2.260464668273926,
      "step": 6068
    },
    {
      "epoch": 5.786895751953125e-06,
      "step": 6068,
      "training_step_time": 7.8672730922698975
    },
    {
      "epoch": 5.787849426269531e-06,
      "model_forward_time": 1.7478482723236084,
      "step": 6069
    },
    {
      "epoch": 5.787849426269531e-06,
      "step": 6069,
      "training_step_time": 8.306240558624268
    },
    {
      "epoch": 5.788803100585938e-06,
      "grad_norm": 0.32735782861709595,
      "learning_rate": 9.996876059704727e-06,
      "loss": 0.08,
      "step": 6070
    },
    {
      "epoch": 5.788803100585938e-06,
      "model_forward_time": 1.3653564453125,
      "step": 6070
    },
    {
      "dynamics_loss_avg": 0.06151399500668049,
      "epoch": 5.788803100585938e-06,
      "step": 6070
    },
    {
      "action_loss_avg": 0.008255301439203323,
      "epoch": 5.788803100585938e-06,
      "step": 6070
    },
    {
      "epoch": 5.788803100585938e-06,
      "step": 6070,
      "training_step_time": 8.69342041015625
    },
    {
      "epoch": 5.7897567749023435e-06,
      "model_forward_time": 1.7225606441497803,
      "step": 6071
    },
    {
      "epoch": 5.7897567749023435e-06,
      "step": 6071,
      "training_step_time": 7.629520654678345
    },
    {
      "epoch": 5.79071044921875e-06,
      "model_forward_time": 1.7121596336364746,
      "step": 6072
    },
    {
      "epoch": 5.79071044921875e-06,
      "step": 6072,
      "training_step_time": 9.745951652526855
    },
    {
      "epoch": 5.791664123535157e-06,
      "model_forward_time": 1.330901861190796,
      "step": 6073
    },
    {
      "epoch": 5.791664123535157e-06,
      "step": 6073,
      "training_step_time": 7.700729131698608
    },
    {
      "epoch": 5.792617797851562e-06,
      "model_forward_time": 1.3269598484039307,
      "step": 6074
    },
    {
      "epoch": 5.792617797851562e-06,
      "step": 6074,
      "training_step_time": 7.81177020072937
    },
    {
      "epoch": 5.793571472167969e-06,
      "model_forward_time": 1.7103712558746338,
      "step": 6075
    },
    {
      "epoch": 5.793571472167969e-06,
      "step": 6075,
      "training_step_time": 7.616853475570679
    },
    {
      "epoch": 5.794525146484375e-06,
      "model_forward_time": 1.7147533893585205,
      "step": 6076
    },
    {
      "epoch": 5.794525146484375e-06,
      "step": 6076,
      "training_step_time": 7.764833927154541
    },
    {
      "epoch": 5.795478820800781e-06,
      "model_forward_time": 1.7159457206726074,
      "step": 6077
    },
    {
      "epoch": 5.795478820800781e-06,
      "step": 6077,
      "training_step_time": 8.418158292770386
    },
    {
      "epoch": 5.796432495117188e-06,
      "model_forward_time": 1.7217731475830078,
      "step": 6078
    },
    {
      "epoch": 5.796432495117188e-06,
      "step": 6078,
      "training_step_time": 7.569692373275757
    },
    {
      "epoch": 5.7973861694335935e-06,
      "model_forward_time": 1.7098007202148438,
      "step": 6079
    },
    {
      "epoch": 5.7973861694335935e-06,
      "step": 6079,
      "training_step_time": 8.16543984413147
    },
    {
      "epoch": 5.79833984375e-06,
      "grad_norm": 0.35276806354522705,
      "learning_rate": 9.996817346537342e-06,
      "loss": 0.0809,
      "step": 6080
    },
    {
      "epoch": 5.79833984375e-06,
      "model_forward_time": 1.7092154026031494,
      "step": 6080
    },
    {
      "dynamics_loss_avg": 0.053558955900371075,
      "epoch": 5.79833984375e-06,
      "step": 6080
    },
    {
      "action_loss_avg": 0.007343410653993488,
      "epoch": 5.79833984375e-06,
      "step": 6080
    },
    {
      "epoch": 5.79833984375e-06,
      "step": 6080,
      "training_step_time": 8.629307270050049
    },
    {
      "epoch": 5.799293518066407e-06,
      "model_forward_time": 1.7173078060150146,
      "step": 6081
    },
    {
      "epoch": 5.799293518066407e-06,
      "step": 6081,
      "training_step_time": 9.57706069946289
    },
    {
      "epoch": 5.800247192382812e-06,
      "model_forward_time": 1.7084989547729492,
      "step": 6082
    },
    {
      "epoch": 5.800247192382812e-06,
      "step": 6082,
      "training_step_time": 8.235969305038452
    },
    {
      "epoch": 5.801200866699219e-06,
      "model_forward_time": 1.7186598777770996,
      "step": 6083
    },
    {
      "epoch": 5.801200866699219e-06,
      "step": 6083,
      "training_step_time": 7.836668968200684
    },
    {
      "epoch": 5.802154541015625e-06,
      "model_forward_time": 1.7099885940551758,
      "step": 6084
    },
    {
      "epoch": 5.802154541015625e-06,
      "step": 6084,
      "training_step_time": 8.162933826446533
    },
    {
      "epoch": 5.803108215332031e-06,
      "model_forward_time": 1.7171618938446045,
      "step": 6085
    },
    {
      "epoch": 5.803108215332031e-06,
      "step": 6085,
      "training_step_time": 7.8056275844573975
    },
    {
      "epoch": 5.804061889648438e-06,
      "model_forward_time": 1.712120532989502,
      "step": 6086
    },
    {
      "epoch": 5.804061889648438e-06,
      "step": 6086,
      "training_step_time": 8.37671160697937
    },
    {
      "epoch": 5.8050155639648435e-06,
      "model_forward_time": 1.714630365371704,
      "step": 6087
    },
    {
      "epoch": 5.8050155639648435e-06,
      "step": 6087,
      "training_step_time": 7.61588978767395
    },
    {
      "epoch": 5.80596923828125e-06,
      "model_forward_time": 1.7182602882385254,
      "step": 6088
    },
    {
      "epoch": 5.80596923828125e-06,
      "step": 6088,
      "training_step_time": 8.24020791053772
    },
    {
      "epoch": 5.806922912597657e-06,
      "model_forward_time": 1.7349581718444824,
      "step": 6089
    },
    {
      "epoch": 5.806922912597657e-06,
      "step": 6089,
      "training_step_time": 8.434508800506592
    },
    {
      "epoch": 5.807876586914062e-06,
      "grad_norm": 0.38380324840545654,
      "learning_rate": 9.996758086925527e-06,
      "loss": 0.0772,
      "step": 6090
    },
    {
      "epoch": 5.807876586914062e-06,
      "model_forward_time": 1.7425217628479004,
      "step": 6090
    },
    {
      "dynamics_loss_avg": 0.06345201134681702,
      "epoch": 5.807876586914062e-06,
      "step": 6090
    },
    {
      "action_loss_avg": 0.005994401662610472,
      "epoch": 5.807876586914062e-06,
      "step": 6090
    },
    {
      "epoch": 5.807876586914062e-06,
      "step": 6090,
      "training_step_time": 10.8147292137146
    },
    {
      "epoch": 5.808830261230469e-06,
      "model_forward_time": 1.716214895248413,
      "step": 6091
    },
    {
      "epoch": 5.808830261230469e-06,
      "step": 6091,
      "training_step_time": 8.430991172790527
    },
    {
      "epoch": 5.809783935546875e-06,
      "model_forward_time": 1.7202255725860596,
      "step": 6092
    },
    {
      "epoch": 5.809783935546875e-06,
      "step": 6092,
      "training_step_time": 7.8410797119140625
    },
    {
      "epoch": 5.810737609863281e-06,
      "model_forward_time": 1.7107701301574707,
      "step": 6093
    },
    {
      "epoch": 5.810737609863281e-06,
      "step": 6093,
      "training_step_time": 8.282181739807129
    },
    {
      "epoch": 5.811691284179688e-06,
      "model_forward_time": 1.7253704071044922,
      "step": 6094
    },
    {
      "epoch": 5.811691284179688e-06,
      "step": 6094,
      "training_step_time": 7.770541667938232
    },
    {
      "epoch": 5.8126449584960935e-06,
      "model_forward_time": 1.716038703918457,
      "step": 6095
    },
    {
      "epoch": 5.8126449584960935e-06,
      "step": 6095,
      "training_step_time": 7.565258264541626
    },
    {
      "epoch": 5.8135986328125e-06,
      "model_forward_time": 2.8795347213745117,
      "step": 6096
    },
    {
      "epoch": 5.8135986328125e-06,
      "step": 6096,
      "training_step_time": 7.779237270355225
    },
    {
      "epoch": 5.814552307128907e-06,
      "model_forward_time": 1.7342374324798584,
      "step": 6097
    },
    {
      "epoch": 5.814552307128907e-06,
      "step": 6097,
      "training_step_time": 7.825058937072754
    },
    {
      "epoch": 5.815505981445312e-06,
      "model_forward_time": 1.7280361652374268,
      "step": 6098
    },
    {
      "epoch": 5.815505981445312e-06,
      "step": 6098,
      "training_step_time": 7.712783575057983
    },
    {
      "epoch": 5.816459655761719e-06,
      "model_forward_time": 1.7425105571746826,
      "step": 6099
    },
    {
      "epoch": 5.816459655761719e-06,
      "step": 6099,
      "training_step_time": 7.716329097747803
    },
    {
      "epoch": 5.817413330078125e-06,
      "grad_norm": 0.20384441316127777,
      "learning_rate": 9.99669828087576e-06,
      "loss": 0.0806,
      "step": 6100
    },
    {
      "epoch": 5.817413330078125e-06,
      "model_forward_time": 1.3290555477142334,
      "step": 6100
    },
    {
      "dynamics_loss_avg": 0.06951864063739777,
      "epoch": 5.817413330078125e-06,
      "step": 6100
    },
    {
      "action_loss_avg": 0.01002314630895853,
      "epoch": 5.817413330078125e-06,
      "step": 6100
    },
    {
      "epoch": 5.817413330078125e-06,
      "step": 6100,
      "training_step_time": 8.415280103683472
    },
    {
      "epoch": 5.818367004394531e-06,
      "model_forward_time": 1.7105011940002441,
      "step": 6101
    },
    {
      "epoch": 5.818367004394531e-06,
      "step": 6101,
      "training_step_time": 7.7120444774627686
    },
    {
      "epoch": 5.819320678710938e-06,
      "model_forward_time": 1.7106478214263916,
      "step": 6102
    },
    {
      "epoch": 5.819320678710938e-06,
      "step": 6102,
      "training_step_time": 7.634498596191406
    },
    {
      "epoch": 5.8202743530273435e-06,
      "model_forward_time": 1.7173097133636475,
      "step": 6103
    },
    {
      "epoch": 5.8202743530273435e-06,
      "step": 6103,
      "training_step_time": 7.606084108352661
    },
    {
      "epoch": 5.82122802734375e-06,
      "model_forward_time": 1.70709228515625,
      "step": 6104
    },
    {
      "epoch": 5.82122802734375e-06,
      "step": 6104,
      "training_step_time": 7.7891693115234375
    },
    {
      "epoch": 5.822181701660157e-06,
      "model_forward_time": 2.9043731689453125,
      "step": 6105
    },
    {
      "epoch": 5.822181701660157e-06,
      "step": 6105,
      "training_step_time": 8.753385066986084
    },
    {
      "epoch": 5.823135375976562e-06,
      "model_forward_time": 1.7246415615081787,
      "step": 6106
    },
    {
      "epoch": 5.823135375976562e-06,
      "step": 6106,
      "training_step_time": 7.712260007858276
    },
    {
      "epoch": 5.824089050292969e-06,
      "model_forward_time": 1.7176282405853271,
      "step": 6107
    },
    {
      "epoch": 5.824089050292969e-06,
      "step": 6107,
      "training_step_time": 7.6748528480529785
    },
    {
      "epoch": 5.825042724609375e-06,
      "model_forward_time": 1.730438470840454,
      "step": 6108
    },
    {
      "epoch": 5.825042724609375e-06,
      "step": 6108,
      "training_step_time": 8.54419732093811
    },
    {
      "epoch": 5.825996398925781e-06,
      "model_forward_time": 1.7108819484710693,
      "step": 6109
    },
    {
      "epoch": 5.825996398925781e-06,
      "step": 6109,
      "training_step_time": 8.572462558746338
    },
    {
      "epoch": 5.826950073242188e-06,
      "grad_norm": 0.33129486441612244,
      "learning_rate": 9.99663792839458e-06,
      "loss": 0.0765,
      "step": 6110
    },
    {
      "epoch": 5.826950073242188e-06,
      "model_forward_time": 1.7066690921783447,
      "step": 6110
    },
    {
      "dynamics_loss_avg": 0.053988083451986316,
      "epoch": 5.826950073242188e-06,
      "step": 6110
    },
    {
      "action_loss_avg": 0.005867550638504327,
      "epoch": 5.826950073242188e-06,
      "step": 6110
    },
    {
      "epoch": 5.826950073242188e-06,
      "step": 6110,
      "training_step_time": 8.47204065322876
    },
    {
      "epoch": 5.8279037475585935e-06,
      "model_forward_time": 1.7092602252960205,
      "step": 6111
    },
    {
      "epoch": 5.8279037475585935e-06,
      "step": 6111,
      "training_step_time": 7.770399808883667
    },
    {
      "epoch": 5.828857421875e-06,
      "model_forward_time": 1.7066452503204346,
      "step": 6112
    },
    {
      "epoch": 5.828857421875e-06,
      "step": 6112,
      "training_step_time": 7.634949684143066
    },
    {
      "epoch": 5.829811096191407e-06,
      "model_forward_time": 1.710158348083496,
      "step": 6113
    },
    {
      "epoch": 5.829811096191407e-06,
      "step": 6113,
      "training_step_time": 7.693635940551758
    },
    {
      "epoch": 5.830764770507812e-06,
      "model_forward_time": 2.8704051971435547,
      "step": 6114
    },
    {
      "epoch": 5.830764770507812e-06,
      "step": 6114,
      "training_step_time": 8.393515348434448
    },
    {
      "epoch": 5.831718444824219e-06,
      "model_forward_time": 1.704770803451538,
      "step": 6115
    },
    {
      "epoch": 5.831718444824219e-06,
      "step": 6115,
      "training_step_time": 9.589545965194702
    },
    {
      "epoch": 5.832672119140625e-06,
      "model_forward_time": 1.7013859748840332,
      "step": 6116
    },
    {
      "epoch": 5.832672119140625e-06,
      "step": 6116,
      "training_step_time": 9.097101926803589
    },
    {
      "epoch": 5.833625793457031e-06,
      "model_forward_time": 1.7564518451690674,
      "step": 6117
    },
    {
      "epoch": 5.833625793457031e-06,
      "step": 6117,
      "training_step_time": 7.719645261764526
    },
    {
      "epoch": 5.834579467773438e-06,
      "model_forward_time": 1.719799518585205,
      "step": 6118
    },
    {
      "epoch": 5.834579467773438e-06,
      "step": 6118,
      "training_step_time": 9.786068677902222
    },
    {
      "epoch": 5.8355331420898435e-06,
      "model_forward_time": 1.7122490406036377,
      "step": 6119
    },
    {
      "epoch": 5.8355331420898435e-06,
      "step": 6119,
      "training_step_time": 8.661616086959839
    },
    {
      "epoch": 5.83648681640625e-06,
      "grad_norm": 0.31286805868148804,
      "learning_rate": 9.99657702948859e-06,
      "loss": 0.0809,
      "step": 6120
    },
    {
      "epoch": 5.83648681640625e-06,
      "model_forward_time": 1.3294684886932373,
      "step": 6120
    },
    {
      "dynamics_loss_avg": 0.09149911403656005,
      "epoch": 5.83648681640625e-06,
      "step": 6120
    },
    {
      "action_loss_avg": 0.007281649066135288,
      "epoch": 5.83648681640625e-06,
      "step": 6120
    },
    {
      "epoch": 5.83648681640625e-06,
      "step": 6120,
      "training_step_time": 7.8360819816589355
    },
    {
      "epoch": 5.837440490722657e-06,
      "model_forward_time": 1.707728385925293,
      "step": 6121
    },
    {
      "epoch": 5.837440490722657e-06,
      "step": 6121,
      "training_step_time": 7.65584921836853
    },
    {
      "epoch": 5.838394165039062e-06,
      "model_forward_time": 1.3354318141937256,
      "step": 6122
    },
    {
      "epoch": 5.838394165039062e-06,
      "step": 6122,
      "training_step_time": 7.803860187530518
    },
    {
      "epoch": 5.839347839355469e-06,
      "model_forward_time": 2.7944979667663574,
      "step": 6123
    },
    {
      "epoch": 5.839347839355469e-06,
      "step": 6123,
      "training_step_time": 8.4365394115448
    },
    {
      "epoch": 5.840301513671875e-06,
      "model_forward_time": 1.7133917808532715,
      "step": 6124
    },
    {
      "epoch": 5.840301513671875e-06,
      "step": 6124,
      "training_step_time": 8.599995851516724
    },
    {
      "epoch": 5.841255187988281e-06,
      "model_forward_time": 1.7141954898834229,
      "step": 6125
    },
    {
      "epoch": 5.841255187988281e-06,
      "step": 6125,
      "training_step_time": 7.685842037200928
    },
    {
      "epoch": 5.842208862304688e-06,
      "model_forward_time": 1.7393269538879395,
      "step": 6126
    },
    {
      "epoch": 5.842208862304688e-06,
      "step": 6126,
      "training_step_time": 7.603852987289429
    },
    {
      "epoch": 5.8431625366210935e-06,
      "model_forward_time": 1.7180297374725342,
      "step": 6127
    },
    {
      "epoch": 5.8431625366210935e-06,
      "step": 6127,
      "training_step_time": 8.85736870765686
    },
    {
      "epoch": 5.8441162109375e-06,
      "model_forward_time": 1.7134206295013428,
      "step": 6128
    },
    {
      "epoch": 5.8441162109375e-06,
      "step": 6128,
      "training_step_time": 8.098551034927368
    },
    {
      "epoch": 5.845069885253906e-06,
      "model_forward_time": 1.7125296592712402,
      "step": 6129
    },
    {
      "epoch": 5.845069885253906e-06,
      "step": 6129,
      "training_step_time": 8.530197858810425
    },
    {
      "epoch": 5.846023559570312e-06,
      "grad_norm": 0.3735412061214447,
      "learning_rate": 9.996515584164449e-06,
      "loss": 0.0778,
      "step": 6130
    },
    {
      "epoch": 5.846023559570312e-06,
      "model_forward_time": 1.7108891010284424,
      "step": 6130
    },
    {
      "dynamics_loss_avg": 0.07504316736012698,
      "epoch": 5.846023559570312e-06,
      "step": 6130
    },
    {
      "action_loss_avg": 0.005804730532690883,
      "epoch": 5.846023559570312e-06,
      "step": 6130
    },
    {
      "epoch": 5.846023559570312e-06,
      "step": 6130,
      "training_step_time": 7.615966081619263
    },
    {
      "epoch": 5.846977233886719e-06,
      "model_forward_time": 1.7122881412506104,
      "step": 6131
    },
    {
      "epoch": 5.846977233886719e-06,
      "step": 6131,
      "training_step_time": 7.708680629730225
    },
    {
      "epoch": 5.847930908203125e-06,
      "model_forward_time": 1.7096798419952393,
      "step": 6132
    },
    {
      "epoch": 5.847930908203125e-06,
      "step": 6132,
      "training_step_time": 7.807586669921875
    },
    {
      "epoch": 5.848884582519531e-06,
      "model_forward_time": 2.887805223464966,
      "step": 6133
    },
    {
      "epoch": 5.848884582519531e-06,
      "step": 6133,
      "training_step_time": 7.681391477584839
    },
    {
      "epoch": 5.849838256835938e-06,
      "model_forward_time": 1.7168247699737549,
      "step": 6134
    },
    {
      "epoch": 5.849838256835938e-06,
      "step": 6134,
      "training_step_time": 9.245124578475952
    },
    {
      "epoch": 5.8507919311523435e-06,
      "model_forward_time": 1.7390966415405273,
      "step": 6135
    },
    {
      "epoch": 5.8507919311523435e-06,
      "step": 6135,
      "training_step_time": 8.123275756835938
    },
    {
      "epoch": 5.85174560546875e-06,
      "model_forward_time": 1.738682508468628,
      "step": 6136
    },
    {
      "epoch": 5.85174560546875e-06,
      "step": 6136,
      "training_step_time": 7.785296678543091
    },
    {
      "epoch": 5.852699279785156e-06,
      "model_forward_time": 1.7176635265350342,
      "step": 6137
    },
    {
      "epoch": 5.852699279785156e-06,
      "step": 6137,
      "training_step_time": 7.723931550979614
    },
    {
      "epoch": 5.853652954101562e-06,
      "model_forward_time": 1.7055549621582031,
      "step": 6138
    },
    {
      "epoch": 5.853652954101562e-06,
      "step": 6138,
      "training_step_time": 7.64284610748291
    },
    {
      "epoch": 5.854606628417969e-06,
      "model_forward_time": 1.7071471214294434,
      "step": 6139
    },
    {
      "epoch": 5.854606628417969e-06,
      "step": 6139,
      "training_step_time": 7.845010280609131
    },
    {
      "epoch": 5.855560302734375e-06,
      "grad_norm": 0.3832957446575165,
      "learning_rate": 9.996453592428873e-06,
      "loss": 0.0718,
      "step": 6140
    },
    {
      "epoch": 5.855560302734375e-06,
      "model_forward_time": 1.7092132568359375,
      "step": 6140
    },
    {
      "dynamics_loss_avg": 0.06616627871990204,
      "epoch": 5.855560302734375e-06,
      "step": 6140
    },
    {
      "action_loss_avg": 0.0047902924241498114,
      "epoch": 5.855560302734375e-06,
      "step": 6140
    },
    {
      "epoch": 5.855560302734375e-06,
      "step": 6140,
      "training_step_time": 8.605239629745483
    },
    {
      "epoch": 5.856513977050781e-06,
      "model_forward_time": 1.7122547626495361,
      "step": 6141
    },
    {
      "epoch": 5.856513977050781e-06,
      "step": 6141,
      "training_step_time": 7.711216449737549
    },
    {
      "epoch": 5.857467651367188e-06,
      "model_forward_time": 2.889225959777832,
      "step": 6142
    },
    {
      "epoch": 5.857467651367188e-06,
      "step": 6142,
      "training_step_time": 7.611268997192383
    },
    {
      "epoch": 5.8584213256835936e-06,
      "model_forward_time": 1.7239279747009277,
      "step": 6143
    },
    {
      "epoch": 5.8584213256835936e-06,
      "step": 6143,
      "training_step_time": 8.664067029953003
    },
    {
      "epoch": 5.859375e-06,
      "model_forward_time": 1.7283415794372559,
      "step": 6144
    },
    {
      "epoch": 5.859375e-06,
      "step": 6144,
      "training_step_time": 8.506031036376953
    },
    {
      "epoch": 5.860328674316406e-06,
      "model_forward_time": 1.7563526630401611,
      "step": 6145
    },
    {
      "epoch": 5.860328674316406e-06,
      "step": 6145,
      "training_step_time": 7.712971210479736
    },
    {
      "epoch": 5.861282348632812e-06,
      "model_forward_time": 1.7092792987823486,
      "step": 6146
    },
    {
      "epoch": 5.861282348632812e-06,
      "step": 6146,
      "training_step_time": 8.599169492721558
    },
    {
      "epoch": 5.862236022949219e-06,
      "model_forward_time": 1.7112348079681396,
      "step": 6147
    },
    {
      "epoch": 5.862236022949219e-06,
      "step": 6147,
      "training_step_time": 7.696936368942261
    },
    {
      "epoch": 5.863189697265625e-06,
      "model_forward_time": 1.724400281906128,
      "step": 6148
    },
    {
      "epoch": 5.863189697265625e-06,
      "step": 6148,
      "training_step_time": 8.297558307647705
    },
    {
      "epoch": 5.864143371582031e-06,
      "model_forward_time": 1.7062206268310547,
      "step": 6149
    },
    {
      "epoch": 5.864143371582031e-06,
      "step": 6149,
      "training_step_time": 7.654638051986694
    },
    {
      "epoch": 5.865097045898438e-06,
      "grad_norm": 0.4531986713409424,
      "learning_rate": 9.996391054288646e-06,
      "loss": 0.075,
      "step": 6150
    },
    {
      "epoch": 5.865097045898438e-06,
      "model_forward_time": 1.712031364440918,
      "step": 6150
    },
    {
      "dynamics_loss_avg": 0.0706766590476036,
      "epoch": 5.865097045898438e-06,
      "step": 6150
    },
    {
      "action_loss_avg": 0.008046945626847445,
      "epoch": 5.865097045898438e-06,
      "step": 6150
    },
    {
      "epoch": 5.865097045898438e-06,
      "step": 6150,
      "training_step_time": 7.672349691390991
    },
    {
      "epoch": 5.8660507202148436e-06,
      "model_forward_time": 2.8945131301879883,
      "step": 6151
    },
    {
      "epoch": 5.8660507202148436e-06,
      "step": 6151,
      "training_step_time": 7.924076795578003
    },
    {
      "epoch": 5.86700439453125e-06,
      "model_forward_time": 1.7129993438720703,
      "step": 6152
    },
    {
      "epoch": 5.86700439453125e-06,
      "step": 6152,
      "training_step_time": 7.76588249206543
    },
    {
      "epoch": 5.867958068847656e-06,
      "model_forward_time": 1.7117507457733154,
      "step": 6153
    },
    {
      "epoch": 5.867958068847656e-06,
      "step": 6153,
      "training_step_time": 8.533971071243286
    },
    {
      "epoch": 5.868911743164062e-06,
      "model_forward_time": 1.784343957901001,
      "step": 6154
    },
    {
      "epoch": 5.868911743164062e-06,
      "step": 6154,
      "training_step_time": 7.616539716720581
    },
    {
      "epoch": 5.869865417480469e-06,
      "model_forward_time": 1.7214765548706055,
      "step": 6155
    },
    {
      "epoch": 5.869865417480469e-06,
      "step": 6155,
      "training_step_time": 7.6942524909973145
    },
    {
      "epoch": 5.870819091796875e-06,
      "model_forward_time": 1.716935634613037,
      "step": 6156
    },
    {
      "epoch": 5.870819091796875e-06,
      "step": 6156,
      "training_step_time": 7.584647178649902
    },
    {
      "epoch": 5.871772766113281e-06,
      "model_forward_time": 1.7108004093170166,
      "step": 6157
    },
    {
      "epoch": 5.871772766113281e-06,
      "step": 6157,
      "training_step_time": 7.676999092102051
    },
    {
      "epoch": 5.872726440429688e-06,
      "model_forward_time": 1.715078592300415,
      "step": 6158
    },
    {
      "epoch": 5.872726440429688e-06,
      "step": 6158,
      "training_step_time": 8.179928541183472
    },
    {
      "epoch": 5.8736801147460936e-06,
      "model_forward_time": 1.725085735321045,
      "step": 6159
    },
    {
      "epoch": 5.8736801147460936e-06,
      "step": 6159,
      "training_step_time": 7.742401123046875
    },
    {
      "epoch": 5.8746337890625e-06,
      "grad_norm": 0.3098033666610718,
      "learning_rate": 9.996327969750605e-06,
      "loss": 0.0742,
      "step": 6160
    },
    {
      "epoch": 5.8746337890625e-06,
      "model_forward_time": 2.887946128845215,
      "step": 6160
    },
    {
      "dynamics_loss_avg": 0.060482687503099444,
      "epoch": 5.8746337890625e-06,
      "step": 6160
    },
    {
      "action_loss_avg": 0.00473801635671407,
      "epoch": 5.8746337890625e-06,
      "step": 6160
    },
    {
      "epoch": 5.8746337890625e-06,
      "step": 6160,
      "training_step_time": 7.982487201690674
    },
    {
      "epoch": 5.875587463378906e-06,
      "model_forward_time": 1.6989734172821045,
      "step": 6161
    },
    {
      "epoch": 5.875587463378906e-06,
      "step": 6161,
      "training_step_time": 8.68047571182251
    },
    {
      "epoch": 5.8765411376953124e-06,
      "model_forward_time": 1.7189717292785645,
      "step": 6162
    },
    {
      "epoch": 5.8765411376953124e-06,
      "step": 6162,
      "training_step_time": 8.360850811004639
    },
    {
      "epoch": 5.877494812011719e-06,
      "model_forward_time": 1.7535326480865479,
      "step": 6163
    },
    {
      "epoch": 5.877494812011719e-06,
      "step": 6163,
      "training_step_time": 7.648272275924683
    },
    {
      "epoch": 5.878448486328125e-06,
      "model_forward_time": 1.7132740020751953,
      "step": 6164
    },
    {
      "epoch": 5.878448486328125e-06,
      "step": 6164,
      "training_step_time": 9.283856391906738
    },
    {
      "epoch": 5.879402160644531e-06,
      "model_forward_time": 1.7244925498962402,
      "step": 6165
    },
    {
      "epoch": 5.879402160644531e-06,
      "step": 6165,
      "training_step_time": 9.684887409210205
    },
    {
      "epoch": 5.880355834960938e-06,
      "model_forward_time": 1.7249202728271484,
      "step": 6166
    },
    {
      "epoch": 5.880355834960938e-06,
      "step": 6166,
      "training_step_time": 7.704209804534912
    },
    {
      "epoch": 5.8813095092773436e-06,
      "model_forward_time": 1.7115678787231445,
      "step": 6167
    },
    {
      "epoch": 5.8813095092773436e-06,
      "step": 6167,
      "training_step_time": 7.584359169006348
    },
    {
      "epoch": 5.88226318359375e-06,
      "model_forward_time": 1.734057903289795,
      "step": 6168
    },
    {
      "epoch": 5.88226318359375e-06,
      "step": 6168,
      "training_step_time": 7.644308567047119
    },
    {
      "epoch": 5.883216857910156e-06,
      "model_forward_time": 2.8950726985931396,
      "step": 6169
    },
    {
      "epoch": 5.883216857910156e-06,
      "step": 6169,
      "training_step_time": 7.87781023979187
    },
    {
      "epoch": 5.8841705322265624e-06,
      "grad_norm": 0.3019058406352997,
      "learning_rate": 9.996264338821649e-06,
      "loss": 0.0751,
      "step": 6170
    },
    {
      "epoch": 5.8841705322265624e-06,
      "model_forward_time": 1.716268539428711,
      "step": 6170
    },
    {
      "dynamics_loss_avg": 0.057625258900225164,
      "epoch": 5.8841705322265624e-06,
      "step": 6170
    },
    {
      "action_loss_avg": 0.007354252599179744,
      "epoch": 5.8841705322265624e-06,
      "step": 6170
    },
    {
      "epoch": 5.8841705322265624e-06,
      "step": 6170,
      "training_step_time": 7.729468107223511
    },
    {
      "epoch": 5.885124206542969e-06,
      "model_forward_time": 1.703174114227295,
      "step": 6171
    },
    {
      "epoch": 5.885124206542969e-06,
      "step": 6171,
      "training_step_time": 9.520247459411621
    },
    {
      "epoch": 5.886077880859375e-06,
      "model_forward_time": 1.7273731231689453,
      "step": 6172
    },
    {
      "epoch": 5.886077880859375e-06,
      "step": 6172,
      "training_step_time": 7.623579978942871
    },
    {
      "epoch": 5.887031555175781e-06,
      "model_forward_time": 1.7087395191192627,
      "step": 6173
    },
    {
      "epoch": 5.887031555175781e-06,
      "step": 6173,
      "training_step_time": 8.397370338439941
    },
    {
      "epoch": 5.887985229492188e-06,
      "model_forward_time": 1.7188482284545898,
      "step": 6174
    },
    {
      "epoch": 5.887985229492188e-06,
      "step": 6174,
      "training_step_time": 8.303088188171387
    },
    {
      "epoch": 5.888938903808594e-06,
      "model_forward_time": 1.703429937362671,
      "step": 6175
    },
    {
      "epoch": 5.888938903808594e-06,
      "step": 6175,
      "training_step_time": 7.733519077301025
    },
    {
      "epoch": 5.889892578125e-06,
      "model_forward_time": 1.722358226776123,
      "step": 6176
    },
    {
      "epoch": 5.889892578125e-06,
      "step": 6176,
      "training_step_time": 7.585578441619873
    },
    {
      "epoch": 5.890846252441406e-06,
      "model_forward_time": 1.7117998600006104,
      "step": 6177
    },
    {
      "epoch": 5.890846252441406e-06,
      "step": 6177,
      "training_step_time": 7.6231584548950195
    },
    {
      "epoch": 5.8917999267578124e-06,
      "model_forward_time": 2.8464434146881104,
      "step": 6178
    },
    {
      "epoch": 5.8917999267578124e-06,
      "step": 6178,
      "training_step_time": 8.7403404712677
    },
    {
      "epoch": 5.892753601074219e-06,
      "model_forward_time": 1.7277724742889404,
      "step": 6179
    },
    {
      "epoch": 5.892753601074219e-06,
      "step": 6179,
      "training_step_time": 7.709570407867432
    },
    {
      "epoch": 5.893707275390625e-06,
      "grad_norm": 0.2706964910030365,
      "learning_rate": 9.996200161508735e-06,
      "loss": 0.0755,
      "step": 6180
    },
    {
      "epoch": 5.893707275390625e-06,
      "model_forward_time": 1.714179515838623,
      "step": 6180
    },
    {
      "dynamics_loss_avg": 0.07055078186094761,
      "epoch": 5.893707275390625e-06,
      "step": 6180
    },
    {
      "action_loss_avg": 0.006737623806111514,
      "epoch": 5.893707275390625e-06,
      "step": 6180
    },
    {
      "epoch": 5.893707275390625e-06,
      "step": 6180,
      "training_step_time": 9.660364866256714
    },
    {
      "epoch": 5.894660949707031e-06,
      "model_forward_time": 1.7261898517608643,
      "step": 6181
    },
    {
      "epoch": 5.894660949707031e-06,
      "step": 6181,
      "training_step_time": 8.991788625717163
    },
    {
      "epoch": 5.895614624023438e-06,
      "model_forward_time": 1.7335984706878662,
      "step": 6182
    },
    {
      "epoch": 5.895614624023438e-06,
      "step": 6182,
      "training_step_time": 8.702466011047363
    },
    {
      "epoch": 5.896568298339844e-06,
      "model_forward_time": 1.721893072128296,
      "step": 6183
    },
    {
      "epoch": 5.896568298339844e-06,
      "step": 6183,
      "training_step_time": 9.0562162399292
    },
    {
      "epoch": 5.89752197265625e-06,
      "model_forward_time": 1.7147307395935059,
      "step": 6184
    },
    {
      "epoch": 5.89752197265625e-06,
      "step": 6184,
      "training_step_time": 7.882667064666748
    },
    {
      "epoch": 5.898475646972656e-06,
      "model_forward_time": 1.3397436141967773,
      "step": 6185
    },
    {
      "epoch": 5.898475646972656e-06,
      "step": 6185,
      "training_step_time": 7.865222930908203
    },
    {
      "epoch": 5.8994293212890624e-06,
      "model_forward_time": 1.7123868465423584,
      "step": 6186
    },
    {
      "epoch": 5.8994293212890624e-06,
      "step": 6186,
      "training_step_time": 7.722991228103638
    },
    {
      "epoch": 5.900382995605469e-06,
      "model_forward_time": 2.8687868118286133,
      "step": 6187
    },
    {
      "epoch": 5.900382995605469e-06,
      "step": 6187,
      "training_step_time": 7.826010704040527
    },
    {
      "epoch": 5.901336669921875e-06,
      "model_forward_time": 1.7359423637390137,
      "step": 6188
    },
    {
      "epoch": 5.901336669921875e-06,
      "step": 6188,
      "training_step_time": 8.019044876098633
    },
    {
      "epoch": 5.902290344238281e-06,
      "model_forward_time": 1.7121946811676025,
      "step": 6189
    },
    {
      "epoch": 5.902290344238281e-06,
      "step": 6189,
      "training_step_time": 8.538857221603394
    },
    {
      "epoch": 5.903244018554688e-06,
      "grad_norm": 0.19319111108779907,
      "learning_rate": 9.996135437818886e-06,
      "loss": 0.0867,
      "step": 6190
    },
    {
      "epoch": 5.903244018554688e-06,
      "model_forward_time": 1.7270572185516357,
      "step": 6190
    },
    {
      "dynamics_loss_avg": 0.07245644815266132,
      "epoch": 5.903244018554688e-06,
      "step": 6190
    },
    {
      "action_loss_avg": 0.0058184665744192895,
      "epoch": 5.903244018554688e-06,
      "step": 6190
    },
    {
      "epoch": 5.903244018554688e-06,
      "step": 6190,
      "training_step_time": 7.677786111831665
    },
    {
      "epoch": 5.904197692871094e-06,
      "model_forward_time": 1.723388671875,
      "step": 6191
    },
    {
      "epoch": 5.904197692871094e-06,
      "step": 6191,
      "training_step_time": 7.818110227584839
    },
    {
      "epoch": 5.9051513671875e-06,
      "model_forward_time": 1.7062344551086426,
      "step": 6192
    },
    {
      "epoch": 5.9051513671875e-06,
      "step": 6192,
      "training_step_time": 7.809992074966431
    },
    {
      "epoch": 5.906105041503906e-06,
      "model_forward_time": 1.7311487197875977,
      "step": 6193
    },
    {
      "epoch": 5.906105041503906e-06,
      "step": 6193,
      "training_step_time": 7.72036075592041
    },
    {
      "epoch": 5.9070587158203125e-06,
      "model_forward_time": 1.7202589511871338,
      "step": 6194
    },
    {
      "epoch": 5.9070587158203125e-06,
      "step": 6194,
      "training_step_time": 7.803573846817017
    },
    {
      "epoch": 5.908012390136719e-06,
      "model_forward_time": 1.7092976570129395,
      "step": 6195
    },
    {
      "epoch": 5.908012390136719e-06,
      "step": 6195,
      "training_step_time": 8.346885204315186
    },
    {
      "epoch": 5.908966064453125e-06,
      "model_forward_time": 2.900592088699341,
      "step": 6196
    },
    {
      "epoch": 5.908966064453125e-06,
      "step": 6196,
      "training_step_time": 7.780246019363403
    },
    {
      "epoch": 5.909919738769531e-06,
      "model_forward_time": 1.7028422355651855,
      "step": 6197
    },
    {
      "epoch": 5.909919738769531e-06,
      "step": 6197,
      "training_step_time": 7.9097161293029785
    },
    {
      "epoch": 5.910873413085938e-06,
      "model_forward_time": 1.714359998703003,
      "step": 6198
    },
    {
      "epoch": 5.910873413085938e-06,
      "step": 6198,
      "training_step_time": 9.317676544189453
    },
    {
      "epoch": 5.911827087402344e-06,
      "model_forward_time": 1.7024695873260498,
      "step": 6199
    },
    {
      "epoch": 5.911827087402344e-06,
      "step": 6199,
      "training_step_time": 7.633682489395142
    },
    {
      "epoch": 5.91278076171875e-06,
      "grad_norm": 0.35667845606803894,
      "learning_rate": 9.996070167759175e-06,
      "loss": 0.0744,
      "step": 6200
    },
    {
      "epoch": 5.91278076171875e-06,
      "model_forward_time": 1.7233376502990723,
      "step": 6200
    },
    {
      "dynamics_loss_avg": 0.07081221491098404,
      "epoch": 5.91278076171875e-06,
      "step": 6200
    },
    {
      "action_loss_avg": 0.006819721660576761,
      "epoch": 5.91278076171875e-06,
      "step": 6200
    },
    {
      "epoch": 5.91278076171875e-06,
      "step": 6200,
      "training_step_time": 7.780222654342651
    },
    {
      "epoch": 5.913734436035156e-06,
      "model_forward_time": 1.7420978546142578,
      "step": 6201
    },
    {
      "epoch": 5.913734436035156e-06,
      "step": 6201,
      "training_step_time": 8.682712316513062
    },
    {
      "epoch": 5.9146881103515625e-06,
      "model_forward_time": 1.716033935546875,
      "step": 6202
    },
    {
      "epoch": 5.9146881103515625e-06,
      "step": 6202,
      "training_step_time": 7.646172285079956
    },
    {
      "epoch": 5.915641784667969e-06,
      "model_forward_time": 1.7126514911651611,
      "step": 6203
    },
    {
      "epoch": 5.915641784667969e-06,
      "step": 6203,
      "training_step_time": 7.732231140136719
    },
    {
      "epoch": 5.916595458984375e-06,
      "model_forward_time": 1.729827880859375,
      "step": 6204
    },
    {
      "epoch": 5.916595458984375e-06,
      "step": 6204,
      "training_step_time": 8.28113603591919
    },
    {
      "epoch": 5.917549133300781e-06,
      "model_forward_time": 2.893179416656494,
      "step": 6205
    },
    {
      "epoch": 5.917549133300781e-06,
      "step": 6205,
      "training_step_time": 8.693513631820679
    },
    {
      "epoch": 5.918502807617188e-06,
      "model_forward_time": 1.7258479595184326,
      "step": 6206
    },
    {
      "epoch": 5.918502807617188e-06,
      "step": 6206,
      "training_step_time": 8.335464477539062
    },
    {
      "epoch": 5.919456481933594e-06,
      "model_forward_time": 1.7130703926086426,
      "step": 6207
    },
    {
      "epoch": 5.919456481933594e-06,
      "step": 6207,
      "training_step_time": 8.590096950531006
    },
    {
      "epoch": 5.92041015625e-06,
      "model_forward_time": 1.7350852489471436,
      "step": 6208
    },
    {
      "epoch": 5.92041015625e-06,
      "step": 6208,
      "training_step_time": 7.664348125457764
    },
    {
      "epoch": 5.921363830566406e-06,
      "model_forward_time": 1.7118661403656006,
      "step": 6209
    },
    {
      "epoch": 5.921363830566406e-06,
      "step": 6209,
      "training_step_time": 8.814845085144043
    },
    {
      "epoch": 5.9223175048828125e-06,
      "grad_norm": 0.1912323236465454,
      "learning_rate": 9.996004351336744e-06,
      "loss": 0.0673,
      "step": 6210
    },
    {
      "epoch": 5.9223175048828125e-06,
      "model_forward_time": 1.7264444828033447,
      "step": 6210
    },
    {
      "dynamics_loss_avg": 0.06108395773917437,
      "epoch": 5.9223175048828125e-06,
      "step": 6210
    },
    {
      "action_loss_avg": 0.006496822717599571,
      "epoch": 5.9223175048828125e-06,
      "step": 6210
    },
    {
      "epoch": 5.9223175048828125e-06,
      "step": 6210,
      "training_step_time": 7.6509034633636475
    },
    {
      "epoch": 5.923271179199219e-06,
      "model_forward_time": 1.3451619148254395,
      "step": 6211
    },
    {
      "epoch": 5.923271179199219e-06,
      "step": 6211,
      "training_step_time": 7.6575469970703125
    },
    {
      "epoch": 5.924224853515625e-06,
      "model_forward_time": 1.711867094039917,
      "step": 6212
    },
    {
      "epoch": 5.924224853515625e-06,
      "step": 6212,
      "training_step_time": 8.52069902420044
    },
    {
      "epoch": 5.925178527832031e-06,
      "model_forward_time": 1.7234179973602295,
      "step": 6213
    },
    {
      "epoch": 5.925178527832031e-06,
      "step": 6213,
      "training_step_time": 8.353549480438232
    },
    {
      "epoch": 5.926132202148438e-06,
      "model_forward_time": 2.8278958797454834,
      "step": 6214
    },
    {
      "epoch": 5.926132202148438e-06,
      "step": 6214,
      "training_step_time": 7.639506101608276
    },
    {
      "epoch": 5.927085876464844e-06,
      "model_forward_time": 1.7219011783599854,
      "step": 6215
    },
    {
      "epoch": 5.927085876464844e-06,
      "step": 6215,
      "training_step_time": 8.75673246383667
    },
    {
      "epoch": 5.92803955078125e-06,
      "model_forward_time": 1.7224891185760498,
      "step": 6216
    },
    {
      "epoch": 5.92803955078125e-06,
      "step": 6216,
      "training_step_time": 7.713265895843506
    },
    {
      "epoch": 5.928993225097656e-06,
      "model_forward_time": 1.7122576236724854,
      "step": 6217
    },
    {
      "epoch": 5.928993225097656e-06,
      "step": 6217,
      "training_step_time": 8.601768970489502
    },
    {
      "epoch": 5.9299468994140625e-06,
      "model_forward_time": 1.726205825805664,
      "step": 6218
    },
    {
      "epoch": 5.9299468994140625e-06,
      "step": 6218,
      "training_step_time": 7.745219469070435
    },
    {
      "epoch": 5.930900573730469e-06,
      "model_forward_time": 1.7391583919525146,
      "step": 6219
    },
    {
      "epoch": 5.930900573730469e-06,
      "step": 6219,
      "training_step_time": 7.643115282058716
    },
    {
      "epoch": 5.931854248046875e-06,
      "grad_norm": 0.20742037892341614,
      "learning_rate": 9.995937988558786e-06,
      "loss": 0.0715,
      "step": 6220
    },
    {
      "epoch": 5.931854248046875e-06,
      "model_forward_time": 1.7187011241912842,
      "step": 6220
    },
    {
      "dynamics_loss_avg": 0.06887412276118994,
      "epoch": 5.931854248046875e-06,
      "step": 6220
    },
    {
      "action_loss_avg": 0.008043909166008233,
      "epoch": 5.931854248046875e-06,
      "step": 6220
    },
    {
      "epoch": 5.931854248046875e-06,
      "step": 6220,
      "training_step_time": 7.622163772583008
    },
    {
      "epoch": 5.932807922363281e-06,
      "model_forward_time": 1.7250325679779053,
      "step": 6221
    },
    {
      "epoch": 5.932807922363281e-06,
      "step": 6221,
      "training_step_time": 7.553234100341797
    },
    {
      "epoch": 5.933761596679688e-06,
      "model_forward_time": 1.718597173690796,
      "step": 6222
    },
    {
      "epoch": 5.933761596679688e-06,
      "step": 6222,
      "training_step_time": 8.650219917297363
    },
    {
      "epoch": 5.934715270996094e-06,
      "model_forward_time": 1.7170422077178955,
      "step": 6223
    },
    {
      "epoch": 5.934715270996094e-06,
      "step": 6223,
      "training_step_time": 8.775281429290771
    },
    {
      "epoch": 5.9356689453125e-06,
      "model_forward_time": 2.896728515625,
      "step": 6224
    },
    {
      "epoch": 5.9356689453125e-06,
      "step": 6224,
      "training_step_time": 8.381337881088257
    },
    {
      "epoch": 5.936622619628906e-06,
      "model_forward_time": 1.7139151096343994,
      "step": 6225
    },
    {
      "epoch": 5.936622619628906e-06,
      "step": 6225,
      "training_step_time": 8.655982494354248
    },
    {
      "epoch": 5.9375762939453125e-06,
      "model_forward_time": 1.7245643138885498,
      "step": 6226
    },
    {
      "epoch": 5.9375762939453125e-06,
      "step": 6226,
      "training_step_time": 8.139607429504395
    },
    {
      "epoch": 5.938529968261719e-06,
      "model_forward_time": 1.7205002307891846,
      "step": 6227
    },
    {
      "epoch": 5.938529968261719e-06,
      "step": 6227,
      "training_step_time": 7.6085264682769775
    },
    {
      "epoch": 5.939483642578125e-06,
      "model_forward_time": 1.738217830657959,
      "step": 6228
    },
    {
      "epoch": 5.939483642578125e-06,
      "step": 6228,
      "training_step_time": 7.78064751625061
    },
    {
      "epoch": 5.940437316894531e-06,
      "model_forward_time": 1.7240967750549316,
      "step": 6229
    },
    {
      "epoch": 5.940437316894531e-06,
      "step": 6229,
      "training_step_time": 7.739360094070435
    },
    {
      "epoch": 5.941390991210938e-06,
      "grad_norm": 0.19303566217422485,
      "learning_rate": 9.995871079432562e-06,
      "loss": 0.0657,
      "step": 6230
    },
    {
      "epoch": 5.941390991210938e-06,
      "model_forward_time": 1.7215707302093506,
      "step": 6230
    },
    {
      "dynamics_loss_avg": 0.05624043196439743,
      "epoch": 5.941390991210938e-06,
      "step": 6230
    },
    {
      "action_loss_avg": 0.006083447695709765,
      "epoch": 5.941390991210938e-06,
      "step": 6230
    },
    {
      "epoch": 5.941390991210938e-06,
      "step": 6230,
      "training_step_time": 7.647464990615845
    },
    {
      "epoch": 5.942344665527344e-06,
      "model_forward_time": 1.7423486709594727,
      "step": 6231
    },
    {
      "epoch": 5.942344665527344e-06,
      "step": 6231,
      "training_step_time": 9.837594985961914
    },
    {
      "epoch": 5.94329833984375e-06,
      "model_forward_time": 1.727872371673584,
      "step": 6232
    },
    {
      "epoch": 5.94329833984375e-06,
      "step": 6232,
      "training_step_time": 7.601660966873169
    },
    {
      "epoch": 5.944252014160156e-06,
      "model_forward_time": 2.918734312057495,
      "step": 6233
    },
    {
      "epoch": 5.944252014160156e-06,
      "step": 6233,
      "training_step_time": 7.720944881439209
    },
    {
      "epoch": 5.9452056884765625e-06,
      "model_forward_time": 1.720712423324585,
      "step": 6234
    },
    {
      "epoch": 5.9452056884765625e-06,
      "step": 6234,
      "training_step_time": 8.83820915222168
    },
    {
      "epoch": 5.946159362792969e-06,
      "model_forward_time": 1.7122657299041748,
      "step": 6235
    },
    {
      "epoch": 5.946159362792969e-06,
      "step": 6235,
      "training_step_time": 8.594058275222778
    },
    {
      "epoch": 5.947113037109375e-06,
      "model_forward_time": 1.722104787826538,
      "step": 6236
    },
    {
      "epoch": 5.947113037109375e-06,
      "step": 6236,
      "training_step_time": 9.44983172416687
    },
    {
      "epoch": 5.948066711425781e-06,
      "model_forward_time": 1.7281715869903564,
      "step": 6237
    },
    {
      "epoch": 5.948066711425781e-06,
      "step": 6237,
      "training_step_time": 7.680219411849976
    },
    {
      "epoch": 5.949020385742188e-06,
      "model_forward_time": 1.7264623641967773,
      "step": 6238
    },
    {
      "epoch": 5.949020385742188e-06,
      "step": 6238,
      "training_step_time": 7.586847305297852
    },
    {
      "epoch": 5.949974060058594e-06,
      "model_forward_time": 1.7253236770629883,
      "step": 6239
    },
    {
      "epoch": 5.949974060058594e-06,
      "step": 6239,
      "training_step_time": 7.570056200027466
    },
    {
      "epoch": 5.950927734375e-06,
      "grad_norm": 0.19107572734355927,
      "learning_rate": 9.995803623965389e-06,
      "loss": 0.0672,
      "step": 6240
    },
    {
      "epoch": 5.950927734375e-06,
      "model_forward_time": 1.720978021621704,
      "step": 6240
    },
    {
      "dynamics_loss_avg": 0.07799041252583265,
      "epoch": 5.950927734375e-06,
      "step": 6240
    },
    {
      "action_loss_avg": 0.005996319325640797,
      "epoch": 5.950927734375e-06,
      "step": 6240
    },
    {
      "epoch": 5.950927734375e-06,
      "step": 6240,
      "training_step_time": 8.532910346984863
    },
    {
      "epoch": 5.951881408691406e-06,
      "model_forward_time": 1.7222511768341064,
      "step": 6241
    },
    {
      "epoch": 5.951881408691406e-06,
      "step": 6241,
      "training_step_time": 7.542586803436279
    },
    {
      "epoch": 5.9528350830078125e-06,
      "model_forward_time": 2.8982889652252197,
      "step": 6242
    },
    {
      "epoch": 5.9528350830078125e-06,
      "step": 6242,
      "training_step_time": 7.625320672988892
    },
    {
      "epoch": 5.953788757324219e-06,
      "model_forward_time": 1.7258362770080566,
      "step": 6243
    },
    {
      "epoch": 5.953788757324219e-06,
      "step": 6243,
      "training_step_time": 9.202247142791748
    },
    {
      "epoch": 5.954742431640625e-06,
      "model_forward_time": 1.7153215408325195,
      "step": 6244
    },
    {
      "epoch": 5.954742431640625e-06,
      "step": 6244,
      "training_step_time": 7.65377402305603
    },
    {
      "epoch": 5.955696105957031e-06,
      "model_forward_time": 1.7279152870178223,
      "step": 6245
    },
    {
      "epoch": 5.955696105957031e-06,
      "step": 6245,
      "training_step_time": 8.11070442199707
    },
    {
      "epoch": 5.956649780273438e-06,
      "model_forward_time": 1.7177085876464844,
      "step": 6246
    },
    {
      "epoch": 5.956649780273438e-06,
      "step": 6246,
      "training_step_time": 8.343995332717896
    },
    {
      "epoch": 5.957603454589844e-06,
      "model_forward_time": 1.750072717666626,
      "step": 6247
    },
    {
      "epoch": 5.957603454589844e-06,
      "step": 6247,
      "training_step_time": 8.242427110671997
    },
    {
      "epoch": 5.95855712890625e-06,
      "model_forward_time": 1.713515281677246,
      "step": 6248
    },
    {
      "epoch": 5.95855712890625e-06,
      "step": 6248,
      "training_step_time": 8.621464729309082
    },
    {
      "epoch": 5.959510803222656e-06,
      "model_forward_time": 1.7268092632293701,
      "step": 6249
    },
    {
      "epoch": 5.959510803222656e-06,
      "step": 6249,
      "training_step_time": 8.254884481430054
    },
    {
      "epoch": 5.9604644775390625e-06,
      "grad_norm": 0.1965319961309433,
      "learning_rate": 9.995735622164642e-06,
      "loss": 0.0658,
      "step": 6250
    },
    {
      "epoch": 5.9604644775390625e-06,
      "model_forward_time": 1.7350685596466064,
      "step": 6250
    },
    {
      "dynamics_loss_avg": 0.06548236142843962,
      "epoch": 5.9604644775390625e-06,
      "step": 6250
    },
    {
      "action_loss_avg": 0.007102224347181618,
      "epoch": 5.9604644775390625e-06,
      "step": 6250
    },
    {
      "epoch": 5.9604644775390625e-06,
      "step": 6250,
      "training_step_time": 7.615254163742065
    },
    {
      "epoch": 5.961418151855469e-06,
      "model_forward_time": 2.916597604751587,
      "step": 6251
    },
    {
      "epoch": 5.961418151855469e-06,
      "step": 6251,
      "training_step_time": 8.503140449523926
    },
    {
      "epoch": 5.962371826171875e-06,
      "model_forward_time": 1.7187151908874512,
      "step": 6252
    },
    {
      "epoch": 5.962371826171875e-06,
      "step": 6252,
      "training_step_time": 8.76269006729126
    },
    {
      "epoch": 5.963325500488281e-06,
      "model_forward_time": 1.729851484298706,
      "step": 6253
    },
    {
      "epoch": 5.963325500488281e-06,
      "step": 6253,
      "training_step_time": 7.648299694061279
    },
    {
      "epoch": 5.964279174804687e-06,
      "model_forward_time": 1.7292888164520264,
      "step": 6254
    },
    {
      "epoch": 5.964279174804687e-06,
      "step": 6254,
      "training_step_time": 7.594198942184448
    },
    {
      "epoch": 5.965232849121094e-06,
      "model_forward_time": 1.7195384502410889,
      "step": 6255
    },
    {
      "epoch": 5.965232849121094e-06,
      "step": 6255,
      "training_step_time": 8.68031930923462
    },
    {
      "epoch": 5.9661865234375e-06,
      "model_forward_time": 1.742563247680664,
      "step": 6256
    },
    {
      "epoch": 5.9661865234375e-06,
      "step": 6256,
      "training_step_time": 7.527467966079712
    },
    {
      "epoch": 5.967140197753906e-06,
      "model_forward_time": 1.7162344455718994,
      "step": 6257
    },
    {
      "epoch": 5.967140197753906e-06,
      "step": 6257,
      "training_step_time": 7.605241537094116
    },
    {
      "epoch": 5.9680938720703125e-06,
      "model_forward_time": 1.7249433994293213,
      "step": 6258
    },
    {
      "epoch": 5.9680938720703125e-06,
      "step": 6258,
      "training_step_time": 7.485796213150024
    },
    {
      "epoch": 5.969047546386719e-06,
      "model_forward_time": 1.7311997413635254,
      "step": 6259
    },
    {
      "epoch": 5.969047546386719e-06,
      "step": 6259,
      "training_step_time": 7.675002336502075
    },
    {
      "epoch": 5.970001220703125e-06,
      "grad_norm": 0.24430540204048157,
      "learning_rate": 9.995667074037759e-06,
      "loss": 0.0683,
      "step": 6260
    },
    {
      "epoch": 5.970001220703125e-06,
      "model_forward_time": 2.8488779067993164,
      "step": 6260
    },
    {
      "dynamics_loss_avg": 0.05548473764210939,
      "epoch": 5.970001220703125e-06,
      "step": 6260
    },
    {
      "action_loss_avg": 0.006561685027554632,
      "epoch": 5.970001220703125e-06,
      "step": 6260
    },
    {
      "epoch": 5.970001220703125e-06,
      "step": 6260,
      "training_step_time": 7.5800111293792725
    },
    {
      "epoch": 5.970954895019531e-06,
      "model_forward_time": 1.7264833450317383,
      "step": 6261
    },
    {
      "epoch": 5.970954895019531e-06,
      "step": 6261,
      "training_step_time": 8.71744155883789
    },
    {
      "epoch": 5.971908569335937e-06,
      "model_forward_time": 1.721052646636963,
      "step": 6262
    },
    {
      "epoch": 5.971908569335937e-06,
      "step": 6262,
      "training_step_time": 7.6196043491363525
    },
    {
      "epoch": 5.972862243652344e-06,
      "model_forward_time": 1.7121057510375977,
      "step": 6263
    },
    {
      "epoch": 5.972862243652344e-06,
      "step": 6263,
      "training_step_time": 8.64906096458435
    },
    {
      "epoch": 5.97381591796875e-06,
      "model_forward_time": 1.719043254852295,
      "step": 6264
    },
    {
      "epoch": 5.97381591796875e-06,
      "step": 6264,
      "training_step_time": 7.6015260219573975
    },
    {
      "epoch": 5.974769592285156e-06,
      "model_forward_time": 1.324436902999878,
      "step": 6265
    },
    {
      "epoch": 5.974769592285156e-06,
      "step": 6265,
      "training_step_time": 9.201704502105713
    },
    {
      "epoch": 5.9757232666015625e-06,
      "model_forward_time": 1.7226874828338623,
      "step": 6266
    },
    {
      "epoch": 5.9757232666015625e-06,
      "step": 6266,
      "training_step_time": 7.678735733032227
    },
    {
      "epoch": 5.976676940917969e-06,
      "model_forward_time": 1.7152636051177979,
      "step": 6267
    },
    {
      "epoch": 5.976676940917969e-06,
      "step": 6267,
      "training_step_time": 7.551596164703369
    },
    {
      "epoch": 5.977630615234375e-06,
      "model_forward_time": 1.7123737335205078,
      "step": 6268
    },
    {
      "epoch": 5.977630615234375e-06,
      "step": 6268,
      "training_step_time": 7.645719766616821
    },
    {
      "epoch": 5.978584289550781e-06,
      "model_forward_time": 1.7292025089263916,
      "step": 6269
    },
    {
      "epoch": 5.978584289550781e-06,
      "step": 6269,
      "training_step_time": 8.391581773757935
    },
    {
      "epoch": 5.979537963867187e-06,
      "grad_norm": 0.37844589352607727,
      "learning_rate": 9.995597979592233e-06,
      "loss": 0.0704,
      "step": 6270
    },
    {
      "epoch": 5.979537963867187e-06,
      "model_forward_time": 2.8790738582611084,
      "step": 6270
    },
    {
      "dynamics_loss_avg": 0.06186682637780905,
      "epoch": 5.979537963867187e-06,
      "step": 6270
    },
    {
      "action_loss_avg": 0.005888858181424439,
      "epoch": 5.979537963867187e-06,
      "step": 6270
    },
    {
      "epoch": 5.979537963867187e-06,
      "step": 6270,
      "training_step_time": 8.287445545196533
    },
    {
      "epoch": 5.980491638183594e-06,
      "model_forward_time": 1.7210063934326172,
      "step": 6271
    },
    {
      "epoch": 5.980491638183594e-06,
      "step": 6271,
      "training_step_time": 7.730334520339966
    },
    {
      "epoch": 5.9814453125e-06,
      "model_forward_time": 1.7185091972351074,
      "step": 6272
    },
    {
      "epoch": 5.9814453125e-06,
      "step": 6272,
      "training_step_time": 8.343790054321289
    },
    {
      "epoch": 5.982398986816406e-06,
      "model_forward_time": 1.7200720310211182,
      "step": 6273
    },
    {
      "epoch": 5.982398986816406e-06,
      "step": 6273,
      "training_step_time": 8.068331480026245
    },
    {
      "epoch": 5.9833526611328125e-06,
      "model_forward_time": 1.7260568141937256,
      "step": 6274
    },
    {
      "epoch": 5.9833526611328125e-06,
      "step": 6274,
      "training_step_time": 8.732608556747437
    },
    {
      "epoch": 5.984306335449219e-06,
      "model_forward_time": 1.7265052795410156,
      "step": 6275
    },
    {
      "epoch": 5.984306335449219e-06,
      "step": 6275,
      "training_step_time": 7.687716722488403
    },
    {
      "epoch": 5.985260009765625e-06,
      "model_forward_time": 1.7250361442565918,
      "step": 6276
    },
    {
      "epoch": 5.985260009765625e-06,
      "step": 6276,
      "training_step_time": 7.5586769580841064
    },
    {
      "epoch": 5.986213684082031e-06,
      "model_forward_time": 1.723013162612915,
      "step": 6277
    },
    {
      "epoch": 5.986213684082031e-06,
      "step": 6277,
      "training_step_time": 7.714527368545532
    },
    {
      "epoch": 5.987167358398437e-06,
      "model_forward_time": 1.7188069820404053,
      "step": 6278
    },
    {
      "epoch": 5.987167358398437e-06,
      "step": 6278,
      "training_step_time": 7.546934604644775
    },
    {
      "epoch": 5.988121032714844e-06,
      "model_forward_time": 2.974486827850342,
      "step": 6279
    },
    {
      "epoch": 5.988121032714844e-06,
      "step": 6279,
      "training_step_time": 8.41876482963562
    },
    {
      "epoch": 5.98907470703125e-06,
      "grad_norm": 0.3836172819137573,
      "learning_rate": 9.995528338835625e-06,
      "loss": 0.076,
      "step": 6280
    },
    {
      "epoch": 5.98907470703125e-06,
      "model_forward_time": 1.7179515361785889,
      "step": 6280
    },
    {
      "dynamics_loss_avg": 0.08189480304718018,
      "epoch": 5.98907470703125e-06,
      "step": 6280
    },
    {
      "action_loss_avg": 0.005784620530903339,
      "epoch": 5.98907470703125e-06,
      "step": 6280
    },
    {
      "epoch": 5.98907470703125e-06,
      "step": 6280,
      "training_step_time": 7.737139940261841
    },
    {
      "epoch": 5.990028381347656e-06,
      "model_forward_time": 1.7218894958496094,
      "step": 6281
    },
    {
      "epoch": 5.990028381347656e-06,
      "step": 6281,
      "training_step_time": 8.69778299331665
    },
    {
      "epoch": 5.9909820556640625e-06,
      "model_forward_time": 1.729259967803955,
      "step": 6282
    },
    {
      "epoch": 5.9909820556640625e-06,
      "step": 6282,
      "training_step_time": 8.827236890792847
    },
    {
      "epoch": 5.991935729980469e-06,
      "model_forward_time": 1.7140109539031982,
      "step": 6283
    },
    {
      "epoch": 5.991935729980469e-06,
      "step": 6283,
      "training_step_time": 6.7027060985565186
    },
    {
      "epoch": 5.992889404296875e-06,
      "model_forward_time": 1.7194371223449707,
      "step": 6284
    },
    {
      "epoch": 5.992889404296875e-06,
      "step": 6284,
      "training_step_time": 6.7197980880737305
    },
    {
      "epoch": 5.993843078613281e-06,
      "model_forward_time": 1.722999095916748,
      "step": 6285
    },
    {
      "epoch": 5.993843078613281e-06,
      "step": 6285,
      "training_step_time": 6.7044267654418945
    },
    {
      "epoch": 5.994796752929687e-06,
      "model_forward_time": 1.7279341220855713,
      "step": 6286
    },
    {
      "epoch": 5.994796752929687e-06,
      "step": 6286,
      "training_step_time": 6.698151588439941
    },
    {
      "epoch": 5.995750427246094e-06,
      "model_forward_time": 1.7184488773345947,
      "step": 6287
    },
    {
      "epoch": 5.995750427246094e-06,
      "step": 6287,
      "training_step_time": 6.687569856643677
    },
    {
      "epoch": 5.9967041015625e-06,
      "model_forward_time": 1.714709758758545,
      "step": 6288
    },
    {
      "epoch": 5.9967041015625e-06,
      "step": 6288,
      "training_step_time": 6.718180179595947
    },
    {
      "epoch": 5.997657775878906e-06,
      "model_forward_time": 1.7218873500823975,
      "step": 6289
    },
    {
      "epoch": 5.997657775878906e-06,
      "step": 6289,
      "training_step_time": 6.6737565994262695
    },
    {
      "epoch": 5.9986114501953125e-06,
      "grad_norm": 0.28784945607185364,
      "learning_rate": 9.995458151775547e-06,
      "loss": 0.0744,
      "step": 6290
    },
    {
      "epoch": 5.9986114501953125e-06,
      "model_forward_time": 1.7182133197784424,
      "step": 6290
    },
    {
      "dynamics_loss_avg": 0.05155780026689172,
      "epoch": 5.9986114501953125e-06,
      "step": 6290
    },
    {
      "action_loss_avg": 0.004557813145220279,
      "epoch": 5.9986114501953125e-06,
      "step": 6290
    },
    {
      "epoch": 5.9986114501953125e-06,
      "step": 6290,
      "training_step_time": 6.678388833999634
    },
    {
      "epoch": 5.999565124511719e-06,
      "model_forward_time": 2.90639591217041,
      "step": 6291
    },
    {
      "epoch": 5.999565124511719e-06,
      "step": 6291,
      "training_step_time": 7.602755069732666
    },
    {
      "epoch": 6.000518798828125e-06,
      "model_forward_time": 1.724073886871338,
      "step": 6292
    },
    {
      "epoch": 6.000518798828125e-06,
      "step": 6292,
      "training_step_time": 7.6636669635772705
    },
    {
      "epoch": 6.001472473144531e-06,
      "model_forward_time": 1.7148044109344482,
      "step": 6293
    },
    {
      "epoch": 6.001472473144531e-06,
      "step": 6293,
      "training_step_time": 6.704820394515991
    },
    {
      "epoch": 6.002426147460937e-06,
      "model_forward_time": 1.7221388816833496,
      "step": 6294
    },
    {
      "epoch": 6.002426147460937e-06,
      "step": 6294,
      "training_step_time": 6.706481456756592
    },
    {
      "epoch": 6.003379821777344e-06,
      "model_forward_time": 1.3366408348083496,
      "step": 6295
    },
    {
      "epoch": 6.003379821777344e-06,
      "step": 6295,
      "training_step_time": 6.667900562286377
    },
    {
      "epoch": 6.00433349609375e-06,
      "model_forward_time": 1.7315309047698975,
      "step": 6296
    },
    {
      "epoch": 6.00433349609375e-06,
      "step": 6296,
      "training_step_time": 6.710780382156372
    },
    {
      "epoch": 6.005287170410156e-06,
      "model_forward_time": 1.732459306716919,
      "step": 6297
    },
    {
      "epoch": 6.005287170410156e-06,
      "step": 6297,
      "training_step_time": 6.703327894210815
    },
    {
      "epoch": 6.0062408447265625e-06,
      "model_forward_time": 1.7155933380126953,
      "step": 6298
    },
    {
      "epoch": 6.0062408447265625e-06,
      "step": 6298,
      "training_step_time": 6.690086841583252
    },
    {
      "epoch": 6.007194519042969e-06,
      "model_forward_time": 1.7191474437713623,
      "step": 6299
    },
    {
      "epoch": 6.007194519042969e-06,
      "step": 6299,
      "training_step_time": 6.6982433795928955
    },
    {
      "epoch": 6.008148193359375e-06,
      "grad_norm": 0.3220636248588562,
      "learning_rate": 9.995387418419677e-06,
      "loss": 0.074,
      "step": 6300
    },
    {
      "epoch": 6.008148193359375e-06,
      "model_forward_time": 2.8498852252960205,
      "step": 6300
    },
    {
      "dynamics_loss_avg": 0.07512420751154422,
      "epoch": 6.008148193359375e-06,
      "step": 6300
    },
    {
      "action_loss_avg": 0.009054942848160863,
      "epoch": 6.008148193359375e-06,
      "step": 6300
    },
    {
      "epoch": 6.008148193359375e-06,
      "step": 6300,
      "training_step_time": 7.563336610794067
    },
    {
      "epoch": 6.009101867675781e-06,
      "model_forward_time": 1.7216792106628418,
      "step": 6301
    },
    {
      "epoch": 6.009101867675781e-06,
      "step": 6301,
      "training_step_time": 7.736745119094849
    },
    {
      "epoch": 6.010055541992187e-06,
      "model_forward_time": 1.3372602462768555,
      "step": 6302
    },
    {
      "epoch": 6.010055541992187e-06,
      "step": 6302,
      "training_step_time": 6.6908392906188965
    },
    {
      "epoch": 6.011009216308594e-06,
      "model_forward_time": 1.7107441425323486,
      "step": 6303
    },
    {
      "epoch": 6.011009216308594e-06,
      "step": 6303,
      "training_step_time": 6.7382330894470215
    },
    {
      "epoch": 6.011962890625e-06,
      "model_forward_time": 1.7101869583129883,
      "step": 6304
    },
    {
      "epoch": 6.011962890625e-06,
      "step": 6304,
      "training_step_time": 6.73223876953125
    },
    {
      "epoch": 6.012916564941406e-06,
      "model_forward_time": 1.7110881805419922,
      "step": 6305
    },
    {
      "epoch": 6.012916564941406e-06,
      "step": 6305,
      "training_step_time": 6.664315462112427
    },
    {
      "epoch": 6.0138702392578125e-06,
      "model_forward_time": 1.714341163635254,
      "step": 6306
    },
    {
      "epoch": 6.0138702392578125e-06,
      "step": 6306,
      "training_step_time": 6.707002878189087
    },
    {
      "epoch": 6.014823913574219e-06,
      "model_forward_time": 1.7343449592590332,
      "step": 6307
    },
    {
      "epoch": 6.014823913574219e-06,
      "step": 6307,
      "training_step_time": 6.7332775592803955
    },
    {
      "epoch": 6.015777587890625e-06,
      "model_forward_time": 1.7216904163360596,
      "step": 6308
    },
    {
      "epoch": 6.015777587890625e-06,
      "step": 6308,
      "training_step_time": 6.698842287063599
    },
    {
      "epoch": 6.016731262207031e-06,
      "model_forward_time": 2.8666086196899414,
      "step": 6309
    },
    {
      "epoch": 6.016731262207031e-06,
      "step": 6309,
      "training_step_time": 7.577880859375
    },
    {
      "epoch": 6.017684936523437e-06,
      "grad_norm": 0.37075337767601013,
      "learning_rate": 9.99531613877575e-06,
      "loss": 0.0752,
      "step": 6310
    },
    {
      "epoch": 6.017684936523437e-06,
      "model_forward_time": 1.711592197418213,
      "step": 6310
    },
    {
      "dynamics_loss_avg": 0.06942760944366455,
      "epoch": 6.017684936523437e-06,
      "step": 6310
    },
    {
      "action_loss_avg": 0.005854804674163461,
      "epoch": 6.017684936523437e-06,
      "step": 6310
    },
    {
      "epoch": 6.017684936523437e-06,
      "step": 6310,
      "training_step_time": 10.10424256324768
    },
    {
      "epoch": 6.018638610839844e-06,
      "model_forward_time": 1.7127735614776611,
      "step": 6311
    },
    {
      "epoch": 6.018638610839844e-06,
      "step": 6311,
      "training_step_time": 7.373843669891357
    },
    {
      "epoch": 6.01959228515625e-06,
      "model_forward_time": 1.7106218338012695,
      "step": 6312
    },
    {
      "epoch": 6.01959228515625e-06,
      "step": 6312,
      "training_step_time": 6.732754707336426
    },
    {
      "epoch": 6.020545959472656e-06,
      "model_forward_time": 1.7126116752624512,
      "step": 6313
    },
    {
      "epoch": 6.020545959472656e-06,
      "step": 6313,
      "training_step_time": 6.712656736373901
    },
    {
      "epoch": 6.0214996337890626e-06,
      "model_forward_time": 1.7210309505462646,
      "step": 6314
    },
    {
      "epoch": 6.0214996337890626e-06,
      "step": 6314,
      "training_step_time": 6.72344708442688
    },
    {
      "epoch": 6.022453308105469e-06,
      "model_forward_time": 1.7142612934112549,
      "step": 6315
    },
    {
      "epoch": 6.022453308105469e-06,
      "step": 6315,
      "training_step_time": 6.735344648361206
    },
    {
      "epoch": 6.023406982421875e-06,
      "model_forward_time": 1.7131173610687256,
      "step": 6316
    },
    {
      "epoch": 6.023406982421875e-06,
      "step": 6316,
      "training_step_time": 6.675228595733643
    },
    {
      "epoch": 6.024360656738281e-06,
      "model_forward_time": 1.712820053100586,
      "step": 6317
    },
    {
      "epoch": 6.024360656738281e-06,
      "step": 6317,
      "training_step_time": 6.714691638946533
    },
    {
      "epoch": 6.025314331054687e-06,
      "model_forward_time": 2.853961706161499,
      "step": 6318
    },
    {
      "epoch": 6.025314331054687e-06,
      "step": 6318,
      "training_step_time": 7.605532646179199
    },
    {
      "epoch": 6.026268005371094e-06,
      "model_forward_time": 1.7099330425262451,
      "step": 6319
    },
    {
      "epoch": 6.026268005371094e-06,
      "step": 6319,
      "training_step_time": 10.038993120193481
    },
    {
      "epoch": 6.0272216796875e-06,
      "grad_norm": 0.370898962020874,
      "learning_rate": 9.995244312851559e-06,
      "loss": 0.0723,
      "step": 6320
    },
    {
      "epoch": 6.0272216796875e-06,
      "model_forward_time": 1.7099342346191406,
      "step": 6320
    },
    {
      "dynamics_loss_avg": 0.06453957483172416,
      "epoch": 6.0272216796875e-06,
      "step": 6320
    },
    {
      "action_loss_avg": 0.005948678334243595,
      "epoch": 6.0272216796875e-06,
      "step": 6320
    },
    {
      "epoch": 6.0272216796875e-06,
      "step": 6320,
      "training_step_time": 7.70875358581543
    },
    {
      "epoch": 6.028175354003906e-06,
      "model_forward_time": 1.715270757675171,
      "step": 6321
    },
    {
      "epoch": 6.028175354003906e-06,
      "step": 6321,
      "training_step_time": 6.772143363952637
    },
    {
      "epoch": 6.0291290283203126e-06,
      "model_forward_time": 1.7175638675689697,
      "step": 6322
    },
    {
      "epoch": 6.0291290283203126e-06,
      "step": 6322,
      "training_step_time": 6.7499284744262695
    },
    {
      "epoch": 6.030082702636719e-06,
      "model_forward_time": 1.7099299430847168,
      "step": 6323
    },
    {
      "epoch": 6.030082702636719e-06,
      "step": 6323,
      "training_step_time": 6.819176435470581
    },
    {
      "epoch": 6.031036376953125e-06,
      "model_forward_time": 1.7105097770690918,
      "step": 6324
    },
    {
      "epoch": 6.031036376953125e-06,
      "step": 6324,
      "training_step_time": 6.732787132263184
    },
    {
      "epoch": 6.031990051269531e-06,
      "model_forward_time": 1.710047960281372,
      "step": 6325
    },
    {
      "epoch": 6.031990051269531e-06,
      "step": 6325,
      "training_step_time": 6.800275564193726
    },
    {
      "epoch": 6.032943725585937e-06,
      "model_forward_time": 1.7072527408599854,
      "step": 6326
    },
    {
      "epoch": 6.032943725585937e-06,
      "step": 6326,
      "training_step_time": 6.843870162963867
    },
    {
      "epoch": 6.033897399902344e-06,
      "model_forward_time": 1.7537024021148682,
      "step": 6327
    },
    {
      "epoch": 6.033897399902344e-06,
      "step": 6327,
      "training_step_time": 8.039551973342896
    },
    {
      "epoch": 6.03485107421875e-06,
      "model_forward_time": 1.7108583450317383,
      "step": 6328
    },
    {
      "epoch": 6.03485107421875e-06,
      "step": 6328,
      "training_step_time": 9.118728160858154
    },
    {
      "epoch": 6.035804748535156e-06,
      "model_forward_time": 1.7290160655975342,
      "step": 6329
    },
    {
      "epoch": 6.035804748535156e-06,
      "step": 6329,
      "training_step_time": 7.778648376464844
    },
    {
      "epoch": 6.0367584228515626e-06,
      "grad_norm": 0.31058892607688904,
      "learning_rate": 9.995171940654961e-06,
      "loss": 0.0746,
      "step": 6330
    },
    {
      "epoch": 6.0367584228515626e-06,
      "model_forward_time": 1.3515245914459229,
      "step": 6330
    },
    {
      "dynamics_loss_avg": 0.047321312315762046,
      "epoch": 6.0367584228515626e-06,
      "step": 6330
    },
    {
      "action_loss_avg": 0.004700300609692931,
      "epoch": 6.0367584228515626e-06,
      "step": 6330
    },
    {
      "epoch": 6.0367584228515626e-06,
      "step": 6330,
      "training_step_time": 6.777417421340942
    },
    {
      "epoch": 6.037712097167969e-06,
      "model_forward_time": 1.7131662368774414,
      "step": 6331
    },
    {
      "epoch": 6.037712097167969e-06,
      "step": 6331,
      "training_step_time": 6.779261112213135
    },
    {
      "epoch": 6.038665771484375e-06,
      "model_forward_time": 1.725261926651001,
      "step": 6332
    },
    {
      "epoch": 6.038665771484375e-06,
      "step": 6332,
      "training_step_time": 6.793781995773315
    },
    {
      "epoch": 6.0396194458007814e-06,
      "model_forward_time": 1.714615821838379,
      "step": 6333
    },
    {
      "epoch": 6.0396194458007814e-06,
      "step": 6333,
      "training_step_time": 6.76269006729126
    },
    {
      "epoch": 6.040573120117187e-06,
      "model_forward_time": 1.7150137424468994,
      "step": 6334
    },
    {
      "epoch": 6.040573120117187e-06,
      "step": 6334,
      "training_step_time": 6.784755229949951
    },
    {
      "epoch": 6.041526794433594e-06,
      "model_forward_time": 1.7205767631530762,
      "step": 6335
    },
    {
      "epoch": 6.041526794433594e-06,
      "step": 6335,
      "training_step_time": 6.791745662689209
    },
    {
      "epoch": 6.04248046875e-06,
      "model_forward_time": 1.7180752754211426,
      "step": 6336
    },
    {
      "epoch": 6.04248046875e-06,
      "step": 6336,
      "training_step_time": 7.0330939292907715
    },
    {
      "epoch": 6.043434143066406e-06,
      "model_forward_time": 2.810227394104004,
      "step": 6337
    },
    {
      "epoch": 6.043434143066406e-06,
      "step": 6337,
      "training_step_time": 8.333709239959717
    },
    {
      "epoch": 6.0443878173828126e-06,
      "model_forward_time": 1.3313894271850586,
      "step": 6338
    },
    {
      "epoch": 6.0443878173828126e-06,
      "step": 6338,
      "training_step_time": 8.667537212371826
    },
    {
      "epoch": 6.045341491699219e-06,
      "model_forward_time": 1.7119324207305908,
      "step": 6339
    },
    {
      "epoch": 6.045341491699219e-06,
      "step": 6339,
      "training_step_time": 6.776172399520874
    },
    {
      "epoch": 6.046295166015625e-06,
      "grad_norm": 0.3035619854927063,
      "learning_rate": 9.995099022193872e-06,
      "loss": 0.0724,
      "step": 6340
    },
    {
      "epoch": 6.046295166015625e-06,
      "model_forward_time": 1.7119135856628418,
      "step": 6340
    },
    {
      "dynamics_loss_avg": 0.06559288501739502,
      "epoch": 6.046295166015625e-06,
      "step": 6340
    },
    {
      "action_loss_avg": 0.007593162963166833,
      "epoch": 6.046295166015625e-06,
      "step": 6340
    },
    {
      "epoch": 6.046295166015625e-06,
      "step": 6340,
      "training_step_time": 6.7883970737457275
    },
    {
      "epoch": 6.0472488403320314e-06,
      "model_forward_time": 1.7124760150909424,
      "step": 6341
    },
    {
      "epoch": 6.0472488403320314e-06,
      "step": 6341,
      "training_step_time": 6.800588846206665
    },
    {
      "epoch": 6.048202514648437e-06,
      "model_forward_time": 1.7132129669189453,
      "step": 6342
    },
    {
      "epoch": 6.048202514648437e-06,
      "step": 6342,
      "training_step_time": 6.804921627044678
    },
    {
      "epoch": 6.049156188964844e-06,
      "model_forward_time": 1.7173423767089844,
      "step": 6343
    },
    {
      "epoch": 6.049156188964844e-06,
      "step": 6343,
      "training_step_time": 6.753372430801392
    },
    {
      "epoch": 6.05010986328125e-06,
      "model_forward_time": 1.7106363773345947,
      "step": 6344
    },
    {
      "epoch": 6.05010986328125e-06,
      "step": 6344,
      "training_step_time": 6.749078273773193
    },
    {
      "epoch": 6.051063537597656e-06,
      "model_forward_time": 1.7100472450256348,
      "step": 6345
    },
    {
      "epoch": 6.051063537597656e-06,
      "step": 6345,
      "training_step_time": 6.757940053939819
    },
    {
      "epoch": 6.052017211914063e-06,
      "model_forward_time": 2.866103172302246,
      "step": 6346
    },
    {
      "epoch": 6.052017211914063e-06,
      "step": 6346,
      "training_step_time": 9.302666664123535
    },
    {
      "epoch": 6.052970886230469e-06,
      "model_forward_time": 1.7176318168640137,
      "step": 6347
    },
    {
      "epoch": 6.052970886230469e-06,
      "step": 6347,
      "training_step_time": 7.871265888214111
    },
    {
      "epoch": 6.053924560546875e-06,
      "model_forward_time": 1.7198586463928223,
      "step": 6348
    },
    {
      "epoch": 6.053924560546875e-06,
      "step": 6348,
      "training_step_time": 7.541987419128418
    },
    {
      "epoch": 6.0548782348632814e-06,
      "model_forward_time": 1.7102713584899902,
      "step": 6349
    },
    {
      "epoch": 6.0548782348632814e-06,
      "step": 6349,
      "training_step_time": 6.740378379821777
    },
    {
      "epoch": 6.055831909179687e-06,
      "grad_norm": 0.28998249769210815,
      "learning_rate": 9.99502555747626e-06,
      "loss": 0.0771,
      "step": 6350
    },
    {
      "epoch": 6.055831909179687e-06,
      "model_forward_time": 1.7095189094543457,
      "step": 6350
    },
    {
      "dynamics_loss_avg": 0.09184718355536461,
      "epoch": 6.055831909179687e-06,
      "step": 6350
    },
    {
      "action_loss_avg": 0.008902645180933177,
      "epoch": 6.055831909179687e-06,
      "step": 6350
    },
    {
      "epoch": 6.055831909179687e-06,
      "step": 6350,
      "training_step_time": 6.783858299255371
    },
    {
      "epoch": 6.056785583496094e-06,
      "model_forward_time": 1.7167022228240967,
      "step": 6351
    },
    {
      "epoch": 6.056785583496094e-06,
      "step": 6351,
      "training_step_time": 6.737114429473877
    },
    {
      "epoch": 6.0577392578125e-06,
      "model_forward_time": 1.7098329067230225,
      "step": 6352
    },
    {
      "epoch": 6.0577392578125e-06,
      "step": 6352,
      "training_step_time": 6.764631509780884
    },
    {
      "epoch": 6.058692932128906e-06,
      "model_forward_time": 1.7134006023406982,
      "step": 6353
    },
    {
      "epoch": 6.058692932128906e-06,
      "step": 6353,
      "training_step_time": 6.760787487030029
    },
    {
      "epoch": 6.059646606445313e-06,
      "model_forward_time": 1.7092440128326416,
      "step": 6354
    },
    {
      "epoch": 6.059646606445313e-06,
      "step": 6354,
      "training_step_time": 6.757094860076904
    },
    {
      "epoch": 6.060600280761719e-06,
      "model_forward_time": 2.918534278869629,
      "step": 6355
    },
    {
      "epoch": 6.060600280761719e-06,
      "step": 6355,
      "training_step_time": 8.368910551071167
    },
    {
      "epoch": 6.061553955078125e-06,
      "model_forward_time": 1.7138214111328125,
      "step": 6356
    },
    {
      "epoch": 6.061553955078125e-06,
      "step": 6356,
      "training_step_time": 7.818996429443359
    },
    {
      "epoch": 6.0625076293945314e-06,
      "model_forward_time": 1.7083723545074463,
      "step": 6357
    },
    {
      "epoch": 6.0625076293945314e-06,
      "step": 6357,
      "training_step_time": 7.705489635467529
    },
    {
      "epoch": 6.063461303710937e-06,
      "model_forward_time": 1.7142119407653809,
      "step": 6358
    },
    {
      "epoch": 6.063461303710937e-06,
      "step": 6358,
      "training_step_time": 6.877138376235962
    },
    {
      "epoch": 6.064414978027344e-06,
      "model_forward_time": 1.7181098461151123,
      "step": 6359
    },
    {
      "epoch": 6.064414978027344e-06,
      "step": 6359,
      "training_step_time": 6.960561037063599
    },
    {
      "epoch": 6.06536865234375e-06,
      "grad_norm": 0.3018434941768646,
      "learning_rate": 9.994951546510165e-06,
      "loss": 0.0731,
      "step": 6360
    },
    {
      "epoch": 6.06536865234375e-06,
      "model_forward_time": 1.7121062278747559,
      "step": 6360
    },
    {
      "dynamics_loss_avg": 0.08474710173904895,
      "epoch": 6.06536865234375e-06,
      "step": 6360
    },
    {
      "action_loss_avg": 0.00829502467531711,
      "epoch": 6.06536865234375e-06,
      "step": 6360
    },
    {
      "epoch": 6.06536865234375e-06,
      "step": 6360,
      "training_step_time": 6.920307874679565
    },
    {
      "epoch": 6.066322326660156e-06,
      "model_forward_time": 1.7131173610687256,
      "step": 6361
    },
    {
      "epoch": 6.066322326660156e-06,
      "step": 6361,
      "training_step_time": 6.882028341293335
    },
    {
      "epoch": 6.067276000976563e-06,
      "model_forward_time": 1.7181570529937744,
      "step": 6362
    },
    {
      "epoch": 6.067276000976563e-06,
      "step": 6362,
      "training_step_time": 6.9067018032073975
    },
    {
      "epoch": 6.068229675292969e-06,
      "model_forward_time": 1.7067241668701172,
      "step": 6363
    },
    {
      "epoch": 6.068229675292969e-06,
      "step": 6363,
      "training_step_time": 6.89060115814209
    },
    {
      "epoch": 6.069183349609375e-06,
      "model_forward_time": 1.3789265155792236,
      "step": 6364
    },
    {
      "epoch": 6.069183349609375e-06,
      "step": 6364,
      "training_step_time": 10.347440719604492
    },
    {
      "epoch": 6.0701370239257815e-06,
      "model_forward_time": 2.8709256649017334,
      "step": 6365
    },
    {
      "epoch": 6.0701370239257815e-06,
      "step": 6365,
      "training_step_time": 10.36656665802002
    },
    {
      "epoch": 6.071090698242187e-06,
      "model_forward_time": 1.3290696144104004,
      "step": 6366
    },
    {
      "epoch": 6.071090698242187e-06,
      "step": 6366,
      "training_step_time": 7.831675291061401
    },
    {
      "epoch": 6.072044372558594e-06,
      "model_forward_time": 1.7150399684906006,
      "step": 6367
    },
    {
      "epoch": 6.072044372558594e-06,
      "step": 6367,
      "training_step_time": 6.898644685745239
    },
    {
      "epoch": 6.072998046875e-06,
      "model_forward_time": 1.7232959270477295,
      "step": 6368
    },
    {
      "epoch": 6.072998046875e-06,
      "step": 6368,
      "training_step_time": 6.877812147140503
    },
    {
      "epoch": 6.073951721191406e-06,
      "model_forward_time": 1.715364694595337,
      "step": 6369
    },
    {
      "epoch": 6.073951721191406e-06,
      "step": 6369,
      "training_step_time": 6.908349514007568
    },
    {
      "epoch": 6.074905395507813e-06,
      "grad_norm": 0.3926290273666382,
      "learning_rate": 9.994876989303678e-06,
      "loss": 0.0756,
      "step": 6370
    },
    {
      "epoch": 6.074905395507813e-06,
      "model_forward_time": 1.714202880859375,
      "step": 6370
    },
    {
      "dynamics_loss_avg": 0.07720746323466302,
      "epoch": 6.074905395507813e-06,
      "step": 6370
    },
    {
      "action_loss_avg": 0.008274012012407184,
      "epoch": 6.074905395507813e-06,
      "step": 6370
    },
    {
      "epoch": 6.074905395507813e-06,
      "step": 6370,
      "training_step_time": 6.918300151824951
    },
    {
      "epoch": 6.075859069824219e-06,
      "model_forward_time": 1.7160439491271973,
      "step": 6371
    },
    {
      "epoch": 6.075859069824219e-06,
      "step": 6371,
      "training_step_time": 6.894752025604248
    },
    {
      "epoch": 6.076812744140625e-06,
      "model_forward_time": 1.7111282348632812,
      "step": 6372
    },
    {
      "epoch": 6.076812744140625e-06,
      "step": 6372,
      "training_step_time": 6.911395788192749
    },
    {
      "epoch": 6.0777664184570315e-06,
      "model_forward_time": 1.3435184955596924,
      "step": 6373
    },
    {
      "epoch": 6.0777664184570315e-06,
      "step": 6373,
      "training_step_time": 8.489217281341553
    },
    {
      "epoch": 6.078720092773437e-06,
      "model_forward_time": 2.846496105194092,
      "step": 6374
    },
    {
      "epoch": 6.078720092773437e-06,
      "step": 6374,
      "training_step_time": 9.625733613967896
    },
    {
      "epoch": 6.079673767089844e-06,
      "model_forward_time": 1.7136156558990479,
      "step": 6375
    },
    {
      "epoch": 6.079673767089844e-06,
      "step": 6375,
      "training_step_time": 8.633432626724243
    },
    {
      "epoch": 6.08062744140625e-06,
      "model_forward_time": 1.7183425426483154,
      "step": 6376
    },
    {
      "epoch": 6.08062744140625e-06,
      "step": 6376,
      "training_step_time": 7.7028632164001465
    },
    {
      "epoch": 6.081581115722656e-06,
      "model_forward_time": 1.7149124145507812,
      "step": 6377
    },
    {
      "epoch": 6.081581115722656e-06,
      "step": 6377,
      "training_step_time": 6.946047782897949
    },
    {
      "epoch": 6.082534790039063e-06,
      "model_forward_time": 1.7164192199707031,
      "step": 6378
    },
    {
      "epoch": 6.082534790039063e-06,
      "step": 6378,
      "training_step_time": 6.888148069381714
    },
    {
      "epoch": 6.083488464355468e-06,
      "model_forward_time": 1.715730905532837,
      "step": 6379
    },
    {
      "epoch": 6.083488464355468e-06,
      "step": 6379,
      "training_step_time": 6.857772588729858
    },
    {
      "epoch": 6.084442138671875e-06,
      "grad_norm": 0.27453693747520447,
      "learning_rate": 9.994801885864955e-06,
      "loss": 0.0736,
      "step": 6380
    },
    {
      "epoch": 6.084442138671875e-06,
      "model_forward_time": 1.7204043865203857,
      "step": 6380
    },
    {
      "dynamics_loss_avg": 0.07872613687068224,
      "epoch": 6.084442138671875e-06,
      "step": 6380
    },
    {
      "action_loss_avg": 0.006720122485421598,
      "epoch": 6.084442138671875e-06,
      "step": 6380
    },
    {
      "epoch": 6.084442138671875e-06,
      "step": 6380,
      "training_step_time": 6.90803337097168
    },
    {
      "epoch": 6.0853958129882815e-06,
      "model_forward_time": 1.71466064453125,
      "step": 6381
    },
    {
      "epoch": 6.0853958129882815e-06,
      "step": 6381,
      "training_step_time": 6.889161109924316
    },
    {
      "epoch": 6.086349487304687e-06,
      "model_forward_time": 1.7309041023254395,
      "step": 6382
    },
    {
      "epoch": 6.086349487304687e-06,
      "step": 6382,
      "training_step_time": 9.3251953125
    },
    {
      "epoch": 6.087303161621094e-06,
      "model_forward_time": 2.7919015884399414,
      "step": 6383
    },
    {
      "epoch": 6.087303161621094e-06,
      "step": 6383,
      "training_step_time": 8.896689653396606
    },
    {
      "epoch": 6.0882568359375e-06,
      "model_forward_time": 1.7131898403167725,
      "step": 6384
    },
    {
      "epoch": 6.0882568359375e-06,
      "step": 6384,
      "training_step_time": 7.84979772567749
    },
    {
      "epoch": 6.089210510253906e-06,
      "model_forward_time": 1.7157530784606934,
      "step": 6385
    },
    {
      "epoch": 6.089210510253906e-06,
      "step": 6385,
      "training_step_time": 7.67643141746521
    },
    {
      "epoch": 6.090164184570313e-06,
      "model_forward_time": 1.7076895236968994,
      "step": 6386
    },
    {
      "epoch": 6.090164184570313e-06,
      "step": 6386,
      "training_step_time": 7.12833833694458
    },
    {
      "epoch": 6.091117858886718e-06,
      "model_forward_time": 1.7182631492614746,
      "step": 6387
    },
    {
      "epoch": 6.091117858886718e-06,
      "step": 6387,
      "training_step_time": 6.8313820362091064
    },
    {
      "epoch": 6.092071533203125e-06,
      "model_forward_time": 1.7187378406524658,
      "step": 6388
    },
    {
      "epoch": 6.092071533203125e-06,
      "step": 6388,
      "training_step_time": 6.79443359375
    },
    {
      "epoch": 6.0930252075195315e-06,
      "model_forward_time": 1.7132999897003174,
      "step": 6389
    },
    {
      "epoch": 6.0930252075195315e-06,
      "step": 6389,
      "training_step_time": 6.8574748039245605
    },
    {
      "epoch": 6.093978881835937e-06,
      "grad_norm": 0.23004648089408875,
      "learning_rate": 9.994726236202205e-06,
      "loss": 0.0749,
      "step": 6390
    },
    {
      "epoch": 6.093978881835937e-06,
      "model_forward_time": 1.3329689502716064,
      "step": 6390
    },
    {
      "dynamics_loss_avg": 0.05866950619965792,
      "epoch": 6.093978881835937e-06,
      "step": 6390
    },
    {
      "action_loss_avg": 0.006122928892727942,
      "epoch": 6.093978881835937e-06,
      "step": 6390
    },
    {
      "epoch": 6.093978881835937e-06,
      "step": 6390,
      "training_step_time": 6.844078302383423
    },
    {
      "epoch": 6.094932556152344e-06,
      "model_forward_time": 1.7245121002197266,
      "step": 6391
    },
    {
      "epoch": 6.094932556152344e-06,
      "step": 6391,
      "training_step_time": 7.798603773117065
    },
    {
      "epoch": 6.09588623046875e-06,
      "model_forward_time": 1.741074562072754,
      "step": 6392
    },
    {
      "epoch": 6.09588623046875e-06,
      "step": 6392,
      "training_step_time": 8.903995990753174
    },
    {
      "epoch": 6.096839904785156e-06,
      "model_forward_time": 2.8856356143951416,
      "step": 6393
    },
    {
      "epoch": 6.096839904785156e-06,
      "step": 6393,
      "training_step_time": 9.192702054977417
    },
    {
      "epoch": 6.097793579101563e-06,
      "model_forward_time": 1.719996452331543,
      "step": 6394
    },
    {
      "epoch": 6.097793579101563e-06,
      "step": 6394,
      "training_step_time": 7.79489541053772
    },
    {
      "epoch": 6.098747253417968e-06,
      "model_forward_time": 1.7130253314971924,
      "step": 6395
    },
    {
      "epoch": 6.098747253417968e-06,
      "step": 6395,
      "training_step_time": 7.639488458633423
    },
    {
      "epoch": 6.099700927734375e-06,
      "model_forward_time": 1.7209603786468506,
      "step": 6396
    },
    {
      "epoch": 6.099700927734375e-06,
      "step": 6396,
      "training_step_time": 6.867396354675293
    },
    {
      "epoch": 6.1006546020507815e-06,
      "model_forward_time": 1.7251532077789307,
      "step": 6397
    },
    {
      "epoch": 6.1006546020507815e-06,
      "step": 6397,
      "training_step_time": 6.815683603286743
    },
    {
      "epoch": 6.101608276367187e-06,
      "model_forward_time": 1.7223737239837646,
      "step": 6398
    },
    {
      "epoch": 6.101608276367187e-06,
      "step": 6398,
      "training_step_time": 6.824060678482056
    },
    {
      "epoch": 6.102561950683594e-06,
      "model_forward_time": 1.7126843929290771,
      "step": 6399
    },
    {
      "epoch": 6.102561950683594e-06,
      "step": 6399,
      "training_step_time": 6.7810797691345215
    },
    {
      "epoch": 6.103515625e-06,
      "grad_norm": 0.31447067856788635,
      "learning_rate": 9.994650040323704e-06,
      "loss": 0.07,
      "step": 6400
    },
    {
      "epoch": 6.103515625e-06,
      "model_forward_time": 1.7198920249938965,
      "step": 6400
    },
    {
      "dynamics_loss_avg": 0.07246627509593964,
      "epoch": 6.103515625e-06,
      "step": 6400
    },
    {
      "action_loss_avg": 0.006316935375798494,
      "epoch": 6.103515625e-06,
      "step": 6400
    },
    {
      "epoch": 6.103515625e-06,
      "step": 6400,
      "training_step_time": 7.777105808258057
    },
    {
      "epoch": 6.104469299316406e-06,
      "model_forward_time": 1.7954761981964111,
      "step": 6401
    },
    {
      "epoch": 6.104469299316406e-06,
      "step": 6401,
      "training_step_time": 7.831674098968506
    },
    {
      "epoch": 6.105422973632813e-06,
      "model_forward_time": 2.9058964252471924,
      "step": 6402
    },
    {
      "epoch": 6.105422973632813e-06,
      "step": 6402,
      "training_step_time": 9.612074613571167
    },
    {
      "epoch": 6.106376647949218e-06,
      "model_forward_time": 1.7210235595703125,
      "step": 6403
    },
    {
      "epoch": 6.106376647949218e-06,
      "step": 6403,
      "training_step_time": 7.739972352981567
    },
    {
      "epoch": 6.107330322265625e-06,
      "model_forward_time": 1.3340072631835938,
      "step": 6404
    },
    {
      "epoch": 6.107330322265625e-06,
      "step": 6404,
      "training_step_time": 7.59576940536499
    },
    {
      "epoch": 6.1082839965820315e-06,
      "model_forward_time": 1.7214486598968506,
      "step": 6405
    },
    {
      "epoch": 6.1082839965820315e-06,
      "step": 6405,
      "training_step_time": 6.843262672424316
    },
    {
      "epoch": 6.109237670898437e-06,
      "model_forward_time": 1.3391942977905273,
      "step": 6406
    },
    {
      "epoch": 6.109237670898437e-06,
      "step": 6406,
      "training_step_time": 6.778834819793701
    },
    {
      "epoch": 6.110191345214844e-06,
      "model_forward_time": 1.718529462814331,
      "step": 6407
    },
    {
      "epoch": 6.110191345214844e-06,
      "step": 6407,
      "training_step_time": 6.828445196151733
    },
    {
      "epoch": 6.11114501953125e-06,
      "model_forward_time": 1.7183237075805664,
      "step": 6408
    },
    {
      "epoch": 6.11114501953125e-06,
      "step": 6408,
      "training_step_time": 6.808581829071045
    },
    {
      "epoch": 6.112098693847656e-06,
      "model_forward_time": 1.7302050590515137,
      "step": 6409
    },
    {
      "epoch": 6.112098693847656e-06,
      "step": 6409,
      "training_step_time": 7.771718978881836
    },
    {
      "epoch": 6.113052368164063e-06,
      "grad_norm": 0.2882234752178192,
      "learning_rate": 9.994573298237784e-06,
      "loss": 0.0771,
      "step": 6410
    },
    {
      "epoch": 6.113052368164063e-06,
      "model_forward_time": 1.7758746147155762,
      "step": 6410
    },
    {
      "dynamics_loss_avg": 0.06845835261046887,
      "epoch": 6.113052368164063e-06,
      "step": 6410
    },
    {
      "action_loss_avg": 0.00487291794270277,
      "epoch": 6.113052368164063e-06,
      "step": 6410
    },
    {
      "epoch": 6.113052368164063e-06,
      "step": 6410,
      "training_step_time": 9.108284711837769
    },
    {
      "epoch": 6.114006042480468e-06,
      "model_forward_time": 2.8061649799346924,
      "step": 6411
    },
    {
      "epoch": 6.114006042480468e-06,
      "step": 6411,
      "training_step_time": 9.942235946655273
    },
    {
      "epoch": 6.114959716796875e-06,
      "model_forward_time": 1.7315762042999268,
      "step": 6412
    },
    {
      "epoch": 6.114959716796875e-06,
      "step": 6412,
      "training_step_time": 7.750070333480835
    },
    {
      "epoch": 6.1159133911132815e-06,
      "model_forward_time": 1.7319142818450928,
      "step": 6413
    },
    {
      "epoch": 6.1159133911132815e-06,
      "step": 6413,
      "training_step_time": 7.666042804718018
    },
    {
      "epoch": 6.116867065429687e-06,
      "model_forward_time": 1.7240731716156006,
      "step": 6414
    },
    {
      "epoch": 6.116867065429687e-06,
      "step": 6414,
      "training_step_time": 7.066664218902588
    },
    {
      "epoch": 6.117820739746094e-06,
      "model_forward_time": 1.7312712669372559,
      "step": 6415
    },
    {
      "epoch": 6.117820739746094e-06,
      "step": 6415,
      "training_step_time": 6.76482892036438
    },
    {
      "epoch": 6.1187744140625e-06,
      "model_forward_time": 1.7377903461456299,
      "step": 6416
    },
    {
      "epoch": 6.1187744140625e-06,
      "step": 6416,
      "training_step_time": 6.81210994720459
    },
    {
      "epoch": 6.119728088378906e-06,
      "model_forward_time": 1.733253002166748,
      "step": 6417
    },
    {
      "epoch": 6.119728088378906e-06,
      "step": 6417,
      "training_step_time": 6.838222026824951
    },
    {
      "epoch": 6.120681762695313e-06,
      "model_forward_time": 1.729748249053955,
      "step": 6418
    },
    {
      "epoch": 6.120681762695313e-06,
      "step": 6418,
      "training_step_time": 8.430556058883667
    },
    {
      "epoch": 6.121635437011718e-06,
      "model_forward_time": 1.7476227283477783,
      "step": 6419
    },
    {
      "epoch": 6.121635437011718e-06,
      "step": 6419,
      "training_step_time": 8.68358564376831
    },
    {
      "epoch": 6.122589111328125e-06,
      "grad_norm": 0.2990635335445404,
      "learning_rate": 9.994496009952838e-06,
      "loss": 0.0731,
      "step": 6420
    },
    {
      "epoch": 6.122589111328125e-06,
      "model_forward_time": 1.7275691032409668,
      "step": 6420
    },
    {
      "dynamics_loss_avg": 0.054728870093822476,
      "epoch": 6.122589111328125e-06,
      "step": 6420
    },
    {
      "action_loss_avg": 0.004480508004780859,
      "epoch": 6.122589111328125e-06,
      "step": 6420
    },
    {
      "epoch": 6.122589111328125e-06,
      "step": 6420,
      "training_step_time": 7.624009370803833
    },
    {
      "epoch": 6.1235427856445315e-06,
      "model_forward_time": 2.856750011444092,
      "step": 6421
    },
    {
      "epoch": 6.1235427856445315e-06,
      "step": 6421,
      "training_step_time": 8.242889165878296
    },
    {
      "epoch": 6.124496459960937e-06,
      "model_forward_time": 1.7297368049621582,
      "step": 6422
    },
    {
      "epoch": 6.124496459960937e-06,
      "step": 6422,
      "training_step_time": 7.526393890380859
    },
    {
      "epoch": 6.125450134277344e-06,
      "model_forward_time": 1.7330000400543213,
      "step": 6423
    },
    {
      "epoch": 6.125450134277344e-06,
      "step": 6423,
      "training_step_time": 7.014593124389648
    },
    {
      "epoch": 6.12640380859375e-06,
      "model_forward_time": 1.7233054637908936,
      "step": 6424
    },
    {
      "epoch": 6.12640380859375e-06,
      "step": 6424,
      "training_step_time": 6.7413129806518555
    },
    {
      "epoch": 6.127357482910156e-06,
      "model_forward_time": 1.7297649383544922,
      "step": 6425
    },
    {
      "epoch": 6.127357482910156e-06,
      "step": 6425,
      "training_step_time": 6.772902250289917
    },
    {
      "epoch": 6.128311157226563e-06,
      "model_forward_time": 1.3592643737792969,
      "step": 6426
    },
    {
      "epoch": 6.128311157226563e-06,
      "step": 6426,
      "training_step_time": 6.726651668548584
    },
    {
      "epoch": 6.129264831542968e-06,
      "model_forward_time": 1.7314317226409912,
      "step": 6427
    },
    {
      "epoch": 6.129264831542968e-06,
      "step": 6427,
      "training_step_time": 8.461153984069824
    },
    {
      "epoch": 6.130218505859375e-06,
      "model_forward_time": 1.738480806350708,
      "step": 6428
    },
    {
      "epoch": 6.130218505859375e-06,
      "step": 6428,
      "training_step_time": 8.679145574569702
    },
    {
      "epoch": 6.1311721801757815e-06,
      "model_forward_time": 1.7168147563934326,
      "step": 6429
    },
    {
      "epoch": 6.1311721801757815e-06,
      "step": 6429,
      "training_step_time": 7.651017904281616
    },
    {
      "epoch": 6.132125854492187e-06,
      "grad_norm": 0.3498379588127136,
      "learning_rate": 9.994418175477317e-06,
      "loss": 0.0763,
      "step": 6430
    },
    {
      "epoch": 6.132125854492187e-06,
      "model_forward_time": 2.8910610675811768,
      "step": 6430
    },
    {
      "dynamics_loss_avg": 0.06829015482217074,
      "epoch": 6.132125854492187e-06,
      "step": 6430
    },
    {
      "action_loss_avg": 0.008584776846691965,
      "epoch": 6.132125854492187e-06,
      "step": 6430
    },
    {
      "epoch": 6.132125854492187e-06,
      "step": 6430,
      "training_step_time": 8.48905873298645
    },
    {
      "epoch": 6.133079528808594e-06,
      "model_forward_time": 1.7269880771636963,
      "step": 6431
    },
    {
      "epoch": 6.133079528808594e-06,
      "step": 6431,
      "training_step_time": 7.612207889556885
    },
    {
      "epoch": 6.134033203125e-06,
      "model_forward_time": 1.726264476776123,
      "step": 6432
    },
    {
      "epoch": 6.134033203125e-06,
      "step": 6432,
      "training_step_time": 7.024423360824585
    },
    {
      "epoch": 6.134986877441406e-06,
      "model_forward_time": 1.7225136756896973,
      "step": 6433
    },
    {
      "epoch": 6.134986877441406e-06,
      "step": 6433,
      "training_step_time": 6.745673894882202
    },
    {
      "epoch": 6.135940551757813e-06,
      "model_forward_time": 1.7270934581756592,
      "step": 6434
    },
    {
      "epoch": 6.135940551757813e-06,
      "step": 6434,
      "training_step_time": 6.789764404296875
    },
    {
      "epoch": 6.136894226074218e-06,
      "model_forward_time": 1.7203772068023682,
      "step": 6435
    },
    {
      "epoch": 6.136894226074218e-06,
      "step": 6435,
      "training_step_time": 6.800829172134399
    },
    {
      "epoch": 6.137847900390625e-06,
      "model_forward_time": 1.7293305397033691,
      "step": 6436
    },
    {
      "epoch": 6.137847900390625e-06,
      "step": 6436,
      "training_step_time": 7.706634998321533
    },
    {
      "epoch": 6.1388015747070315e-06,
      "model_forward_time": 1.7358918190002441,
      "step": 6437
    },
    {
      "epoch": 6.1388015747070315e-06,
      "step": 6437,
      "training_step_time": 9.225330591201782
    },
    {
      "epoch": 6.139755249023437e-06,
      "model_forward_time": 1.7437236309051514,
      "step": 6438
    },
    {
      "epoch": 6.139755249023437e-06,
      "step": 6438,
      "training_step_time": 9.157304286956787
    },
    {
      "epoch": 6.140708923339844e-06,
      "model_forward_time": 2.880023241043091,
      "step": 6439
    },
    {
      "epoch": 6.140708923339844e-06,
      "step": 6439,
      "training_step_time": 9.456763982772827
    },
    {
      "epoch": 6.14166259765625e-06,
      "grad_norm": 0.39880990982055664,
      "learning_rate": 9.994339794819733e-06,
      "loss": 0.0735,
      "step": 6440
    },
    {
      "epoch": 6.14166259765625e-06,
      "model_forward_time": 1.7316577434539795,
      "step": 6440
    },
    {
      "dynamics_loss_avg": 0.08141424078494311,
      "epoch": 6.14166259765625e-06,
      "step": 6440
    },
    {
      "action_loss_avg": 0.009648819640278816,
      "epoch": 6.14166259765625e-06,
      "step": 6440
    },
    {
      "epoch": 6.14166259765625e-06,
      "step": 6440,
      "training_step_time": 7.702378749847412
    },
    {
      "epoch": 6.142616271972656e-06,
      "model_forward_time": 1.7230665683746338,
      "step": 6441
    },
    {
      "epoch": 6.142616271972656e-06,
      "step": 6441,
      "training_step_time": 7.59403395652771
    },
    {
      "epoch": 6.143569946289063e-06,
      "model_forward_time": 1.717923879623413,
      "step": 6442
    },
    {
      "epoch": 6.143569946289063e-06,
      "step": 6442,
      "training_step_time": 7.072266578674316
    },
    {
      "epoch": 6.144523620605468e-06,
      "model_forward_time": 1.7282462120056152,
      "step": 6443
    },
    {
      "epoch": 6.144523620605468e-06,
      "step": 6443,
      "training_step_time": 6.843011140823364
    },
    {
      "epoch": 6.145477294921875e-06,
      "model_forward_time": 1.7240805625915527,
      "step": 6444
    },
    {
      "epoch": 6.145477294921875e-06,
      "step": 6444,
      "training_step_time": 6.848323822021484
    },
    {
      "epoch": 6.1464309692382815e-06,
      "model_forward_time": 1.3489446640014648,
      "step": 6445
    },
    {
      "epoch": 6.1464309692382815e-06,
      "step": 6445,
      "training_step_time": 7.759895324707031
    },
    {
      "epoch": 6.147384643554687e-06,
      "model_forward_time": 1.740534782409668,
      "step": 6446
    },
    {
      "epoch": 6.147384643554687e-06,
      "step": 6446,
      "training_step_time": 7.847504138946533
    },
    {
      "epoch": 6.148338317871094e-06,
      "model_forward_time": 1.739680528640747,
      "step": 6447
    },
    {
      "epoch": 6.148338317871094e-06,
      "step": 6447,
      "training_step_time": 9.582087993621826
    },
    {
      "epoch": 6.1492919921875e-06,
      "model_forward_time": 2.7905662059783936,
      "step": 6448
    },
    {
      "epoch": 6.1492919921875e-06,
      "step": 6448,
      "training_step_time": 10.410691261291504
    },
    {
      "epoch": 6.150245666503906e-06,
      "model_forward_time": 1.7220368385314941,
      "step": 6449
    },
    {
      "epoch": 6.150245666503906e-06,
      "step": 6449,
      "training_step_time": 7.821137189865112
    },
    {
      "epoch": 6.151199340820313e-06,
      "grad_norm": 0.21309468150138855,
      "learning_rate": 9.994260867988658e-06,
      "loss": 0.0762,
      "step": 6450
    },
    {
      "epoch": 6.151199340820313e-06,
      "model_forward_time": 1.7248880863189697,
      "step": 6450
    },
    {
      "dynamics_loss_avg": 0.08115276731550694,
      "epoch": 6.151199340820313e-06,
      "step": 6450
    },
    {
      "action_loss_avg": 0.0072328755632042885,
      "epoch": 6.151199340820313e-06,
      "step": 6450
    },
    {
      "epoch": 6.151199340820313e-06,
      "step": 6450,
      "training_step_time": 7.702172756195068
    },
    {
      "epoch": 6.152153015136718e-06,
      "model_forward_time": 1.7267982959747314,
      "step": 6451
    },
    {
      "epoch": 6.152153015136718e-06,
      "step": 6451,
      "training_step_time": 7.1329731941223145
    },
    {
      "epoch": 6.153106689453125e-06,
      "model_forward_time": 1.7270112037658691,
      "step": 6452
    },
    {
      "epoch": 6.153106689453125e-06,
      "step": 6452,
      "training_step_time": 6.906515836715698
    },
    {
      "epoch": 6.1540603637695315e-06,
      "model_forward_time": 1.7165911197662354,
      "step": 6453
    },
    {
      "epoch": 6.1540603637695315e-06,
      "step": 6453,
      "training_step_time": 6.911338806152344
    },
    {
      "epoch": 6.155014038085937e-06,
      "model_forward_time": 1.727614164352417,
      "step": 6454
    },
    {
      "epoch": 6.155014038085937e-06,
      "step": 6454,
      "training_step_time": 7.791309833526611
    },
    {
      "epoch": 6.155967712402344e-06,
      "model_forward_time": 1.3698077201843262,
      "step": 6455
    },
    {
      "epoch": 6.155967712402344e-06,
      "step": 6455,
      "training_step_time": 7.774792671203613
    },
    {
      "epoch": 6.15692138671875e-06,
      "model_forward_time": 1.7336077690124512,
      "step": 6456
    },
    {
      "epoch": 6.15692138671875e-06,
      "step": 6456,
      "training_step_time": 7.821441650390625
    },
    {
      "epoch": 6.157875061035156e-06,
      "model_forward_time": 1.7165546417236328,
      "step": 6457
    },
    {
      "epoch": 6.157875061035156e-06,
      "step": 6457,
      "training_step_time": 8.622289180755615
    },
    {
      "epoch": 6.158828735351563e-06,
      "model_forward_time": 2.869203567504883,
      "step": 6458
    },
    {
      "epoch": 6.158828735351563e-06,
      "step": 6458,
      "training_step_time": 9.703859567642212
    },
    {
      "epoch": 6.159782409667968e-06,
      "model_forward_time": 1.7219393253326416,
      "step": 6459
    },
    {
      "epoch": 6.159782409667968e-06,
      "step": 6459,
      "training_step_time": 7.653257608413696
    },
    {
      "epoch": 6.160736083984375e-06,
      "grad_norm": 0.27745023369789124,
      "learning_rate": 9.994181394992723e-06,
      "loss": 0.0779,
      "step": 6460
    },
    {
      "epoch": 6.160736083984375e-06,
      "model_forward_time": 1.7277841567993164,
      "step": 6460
    },
    {
      "dynamics_loss_avg": 0.07378020454198123,
      "epoch": 6.160736083984375e-06,
      "step": 6460
    },
    {
      "action_loss_avg": 0.006876553688198328,
      "epoch": 6.160736083984375e-06,
      "step": 6460
    },
    {
      "epoch": 6.160736083984375e-06,
      "step": 6460,
      "training_step_time": 7.1715004444122314
    },
    {
      "epoch": 6.1616897583007815e-06,
      "model_forward_time": 1.7145235538482666,
      "step": 6461
    },
    {
      "epoch": 6.1616897583007815e-06,
      "step": 6461,
      "training_step_time": 6.894352912902832
    },
    {
      "epoch": 6.162643432617187e-06,
      "model_forward_time": 1.7134060859680176,
      "step": 6462
    },
    {
      "epoch": 6.162643432617187e-06,
      "step": 6462,
      "training_step_time": 6.879798412322998
    },
    {
      "epoch": 6.163597106933594e-06,
      "model_forward_time": 1.3486299514770508,
      "step": 6463
    },
    {
      "epoch": 6.163597106933594e-06,
      "step": 6463,
      "training_step_time": 7.792975902557373
    },
    {
      "epoch": 6.16455078125e-06,
      "model_forward_time": 1.3767356872558594,
      "step": 6464
    },
    {
      "epoch": 6.16455078125e-06,
      "step": 6464,
      "training_step_time": 9.384557723999023
    },
    {
      "epoch": 6.165504455566406e-06,
      "model_forward_time": 1.7182865142822266,
      "step": 6465
    },
    {
      "epoch": 6.165504455566406e-06,
      "step": 6465,
      "training_step_time": 8.856332302093506
    },
    {
      "epoch": 6.166458129882813e-06,
      "model_forward_time": 1.722304105758667,
      "step": 6466
    },
    {
      "epoch": 6.166458129882813e-06,
      "step": 6466,
      "training_step_time": 8.819912910461426
    },
    {
      "epoch": 6.167411804199218e-06,
      "model_forward_time": 2.883363723754883,
      "step": 6467
    },
    {
      "epoch": 6.167411804199218e-06,
      "step": 6467,
      "training_step_time": 8.806588172912598
    },
    {
      "epoch": 6.168365478515625e-06,
      "model_forward_time": 1.7202224731445312,
      "step": 6468
    },
    {
      "epoch": 6.168365478515625e-06,
      "step": 6468,
      "training_step_time": 7.778620958328247
    },
    {
      "epoch": 6.1693191528320315e-06,
      "model_forward_time": 1.7172534465789795,
      "step": 6469
    },
    {
      "epoch": 6.1693191528320315e-06,
      "step": 6469,
      "training_step_time": 7.1919474601745605
    },
    {
      "epoch": 6.170272827148437e-06,
      "grad_norm": 0.23417054116725922,
      "learning_rate": 9.994101375840619e-06,
      "loss": 0.0738,
      "step": 6470
    },
    {
      "epoch": 6.170272827148437e-06,
      "model_forward_time": 1.7309250831604004,
      "step": 6470
    },
    {
      "dynamics_loss_avg": 0.065069211833179,
      "epoch": 6.170272827148437e-06,
      "step": 6470
    },
    {
      "action_loss_avg": 0.007008750177919865,
      "epoch": 6.170272827148437e-06,
      "step": 6470
    },
    {
      "epoch": 6.170272827148437e-06,
      "step": 6470,
      "training_step_time": 6.883742332458496
    },
    {
      "epoch": 6.171226501464844e-06,
      "model_forward_time": 1.719634771347046,
      "step": 6471
    },
    {
      "epoch": 6.171226501464844e-06,
      "step": 6471,
      "training_step_time": 6.849809169769287
    },
    {
      "epoch": 6.17218017578125e-06,
      "model_forward_time": 1.7231457233428955,
      "step": 6472
    },
    {
      "epoch": 6.17218017578125e-06,
      "step": 6472,
      "training_step_time": 7.825294256210327
    },
    {
      "epoch": 6.173133850097656e-06,
      "model_forward_time": 1.7477636337280273,
      "step": 6473
    },
    {
      "epoch": 6.173133850097656e-06,
      "step": 6473,
      "training_step_time": 7.767857551574707
    },
    {
      "epoch": 6.174087524414063e-06,
      "model_forward_time": 1.7242331504821777,
      "step": 6474
    },
    {
      "epoch": 6.174087524414063e-06,
      "step": 6474,
      "training_step_time": 7.73615837097168
    },
    {
      "epoch": 6.175041198730468e-06,
      "model_forward_time": 1.7393033504486084,
      "step": 6475
    },
    {
      "epoch": 6.175041198730468e-06,
      "step": 6475,
      "training_step_time": 8.558525562286377
    },
    {
      "epoch": 6.175994873046875e-06,
      "model_forward_time": 2.843648672103882,
      "step": 6476
    },
    {
      "epoch": 6.175994873046875e-06,
      "step": 6476,
      "training_step_time": 7.877028465270996
    },
    {
      "epoch": 6.1769485473632815e-06,
      "model_forward_time": 1.723883867263794,
      "step": 6477
    },
    {
      "epoch": 6.1769485473632815e-06,
      "step": 6477,
      "training_step_time": 7.674567937850952
    },
    {
      "epoch": 6.177902221679687e-06,
      "model_forward_time": 1.729661464691162,
      "step": 6478
    },
    {
      "epoch": 6.177902221679687e-06,
      "step": 6478,
      "training_step_time": 7.493892192840576
    },
    {
      "epoch": 6.178855895996094e-06,
      "model_forward_time": 1.7215001583099365,
      "step": 6479
    },
    {
      "epoch": 6.178855895996094e-06,
      "step": 6479,
      "training_step_time": 6.823426246643066
    },
    {
      "epoch": 6.1798095703125e-06,
      "grad_norm": 0.32281967997550964,
      "learning_rate": 9.994020810541098e-06,
      "loss": 0.0725,
      "step": 6480
    },
    {
      "epoch": 6.1798095703125e-06,
      "model_forward_time": 1.7220916748046875,
      "step": 6480
    },
    {
      "dynamics_loss_avg": 0.06214981358498335,
      "epoch": 6.1798095703125e-06,
      "step": 6480
    },
    {
      "action_loss_avg": 0.0062836242839694025,
      "epoch": 6.1798095703125e-06,
      "step": 6480
    },
    {
      "epoch": 6.1798095703125e-06,
      "step": 6480,
      "training_step_time": 6.802203893661499
    },
    {
      "epoch": 6.180763244628906e-06,
      "model_forward_time": 1.7223255634307861,
      "step": 6481
    },
    {
      "epoch": 6.180763244628906e-06,
      "step": 6481,
      "training_step_time": 7.72061824798584
    },
    {
      "epoch": 6.181716918945313e-06,
      "model_forward_time": 1.7520275115966797,
      "step": 6482
    },
    {
      "epoch": 6.181716918945313e-06,
      "step": 6482,
      "training_step_time": 7.823112487792969
    },
    {
      "epoch": 6.182670593261718e-06,
      "model_forward_time": 1.725341558456421,
      "step": 6483
    },
    {
      "epoch": 6.182670593261718e-06,
      "step": 6483,
      "training_step_time": 8.40659213066101
    },
    {
      "epoch": 6.183624267578125e-06,
      "model_forward_time": 1.7373769283294678,
      "step": 6484
    },
    {
      "epoch": 6.183624267578125e-06,
      "step": 6484,
      "training_step_time": 7.661065578460693
    },
    {
      "epoch": 6.1845779418945316e-06,
      "model_forward_time": 2.8037164211273193,
      "step": 6485
    },
    {
      "epoch": 6.1845779418945316e-06,
      "step": 6485,
      "training_step_time": 8.587543487548828
    },
    {
      "epoch": 6.185531616210937e-06,
      "model_forward_time": 1.7171251773834229,
      "step": 6486
    },
    {
      "epoch": 6.185531616210937e-06,
      "step": 6486,
      "training_step_time": 7.661808729171753
    },
    {
      "epoch": 6.186485290527344e-06,
      "model_forward_time": 1.7398347854614258,
      "step": 6487
    },
    {
      "epoch": 6.186485290527344e-06,
      "step": 6487,
      "training_step_time": 8.216078281402588
    },
    {
      "epoch": 6.18743896484375e-06,
      "model_forward_time": 1.3413012027740479,
      "step": 6488
    },
    {
      "epoch": 6.18743896484375e-06,
      "step": 6488,
      "training_step_time": 6.847877264022827
    },
    {
      "epoch": 6.188392639160156e-06,
      "model_forward_time": 1.7340705394744873,
      "step": 6489
    },
    {
      "epoch": 6.188392639160156e-06,
      "step": 6489,
      "training_step_time": 6.770171642303467
    },
    {
      "epoch": 6.189346313476563e-06,
      "grad_norm": 0.3806267976760864,
      "learning_rate": 9.99393969910297e-06,
      "loss": 0.0761,
      "step": 6490
    },
    {
      "epoch": 6.189346313476563e-06,
      "model_forward_time": 1.7245430946350098,
      "step": 6490
    },
    {
      "dynamics_loss_avg": 0.07589104995131493,
      "epoch": 6.189346313476563e-06,
      "step": 6490
    },
    {
      "action_loss_avg": 0.007782040722668171,
      "epoch": 6.189346313476563e-06,
      "step": 6490
    },
    {
      "epoch": 6.189346313476563e-06,
      "step": 6490,
      "training_step_time": 7.725012540817261
    },
    {
      "epoch": 6.1902999877929684e-06,
      "model_forward_time": 1.7553973197937012,
      "step": 6491
    },
    {
      "epoch": 6.1902999877929684e-06,
      "step": 6491,
      "training_step_time": 7.736825466156006
    },
    {
      "epoch": 6.191253662109375e-06,
      "model_forward_time": 1.7225525379180908,
      "step": 6492
    },
    {
      "epoch": 6.191253662109375e-06,
      "step": 6492,
      "training_step_time": 9.23454213142395
    },
    {
      "epoch": 6.1922073364257816e-06,
      "model_forward_time": 1.743086576461792,
      "step": 6493
    },
    {
      "epoch": 6.1922073364257816e-06,
      "step": 6493,
      "training_step_time": 8.346667766571045
    },
    {
      "epoch": 6.193161010742187e-06,
      "model_forward_time": 1.7258515357971191,
      "step": 6494
    },
    {
      "epoch": 6.193161010742187e-06,
      "step": 6494,
      "training_step_time": 7.729306221008301
    },
    {
      "epoch": 6.194114685058594e-06,
      "model_forward_time": 2.890225648880005,
      "step": 6495
    },
    {
      "epoch": 6.194114685058594e-06,
      "step": 6495,
      "training_step_time": 7.692483425140381
    },
    {
      "epoch": 6.195068359375e-06,
      "model_forward_time": 1.7264249324798584,
      "step": 6496
    },
    {
      "epoch": 6.195068359375e-06,
      "step": 6496,
      "training_step_time": 7.6119019985198975
    },
    {
      "epoch": 6.196022033691406e-06,
      "model_forward_time": 1.356623888015747,
      "step": 6497
    },
    {
      "epoch": 6.196022033691406e-06,
      "step": 6497,
      "training_step_time": 7.05353045463562
    },
    {
      "epoch": 6.196975708007813e-06,
      "model_forward_time": 1.7263298034667969,
      "step": 6498
    },
    {
      "epoch": 6.196975708007813e-06,
      "step": 6498,
      "training_step_time": 6.743189573287964
    },
    {
      "epoch": 6.1979293823242184e-06,
      "model_forward_time": 1.7332587242126465,
      "step": 6499
    },
    {
      "epoch": 6.1979293823242184e-06,
      "step": 6499,
      "training_step_time": 7.700135946273804
    },
    {
      "epoch": 6.198883056640625e-06,
      "grad_norm": 0.274658203125,
      "learning_rate": 9.993858041535104e-06,
      "loss": 0.0771,
      "step": 6500
    },
    {
      "epoch": 6.198883056640625e-06,
      "model_forward_time": 1.7370879650115967,
      "step": 6500
    },
    {
      "dynamics_loss_avg": 0.0813639547675848,
      "epoch": 6.198883056640625e-06,
      "step": 6500
    },
    {
      "action_loss_avg": 0.0077706980984658,
      "epoch": 6.198883056640625e-06,
      "step": 6500
    },
    {
      "epoch": 6.198883056640625e-06,
      "step": 6500,
      "training_step_time": 7.708531141281128
    },
    {
      "epoch": 6.1998367309570316e-06,
      "model_forward_time": 1.7196192741394043,
      "step": 6501
    },
    {
      "epoch": 6.1998367309570316e-06,
      "step": 6501,
      "training_step_time": 9.190446853637695
    },
    {
      "epoch": 6.200790405273437e-06,
      "model_forward_time": 1.7356500625610352,
      "step": 6502
    },
    {
      "epoch": 6.200790405273437e-06,
      "step": 6502,
      "training_step_time": 7.78613543510437
    },
    {
      "epoch": 6.201744079589844e-06,
      "model_forward_time": 1.3504066467285156,
      "step": 6503
    },
    {
      "epoch": 6.201744079589844e-06,
      "step": 6503,
      "training_step_time": 8.793901681900024
    },
    {
      "epoch": 6.20269775390625e-06,
      "model_forward_time": 2.8668222427368164,
      "step": 6504
    },
    {
      "epoch": 6.20269775390625e-06,
      "step": 6504,
      "training_step_time": 7.812387466430664
    },
    {
      "epoch": 6.203651428222656e-06,
      "model_forward_time": 1.7238049507141113,
      "step": 6505
    },
    {
      "epoch": 6.203651428222656e-06,
      "step": 6505,
      "training_step_time": 7.62361478805542
    },
    {
      "epoch": 6.204605102539063e-06,
      "model_forward_time": 1.7161712646484375,
      "step": 6506
    },
    {
      "epoch": 6.204605102539063e-06,
      "step": 6506,
      "training_step_time": 7.170828104019165
    },
    {
      "epoch": 6.2055587768554684e-06,
      "model_forward_time": 1.7205982208251953,
      "step": 6507
    },
    {
      "epoch": 6.2055587768554684e-06,
      "step": 6507,
      "training_step_time": 6.879647970199585
    },
    {
      "epoch": 6.206512451171875e-06,
      "model_forward_time": 1.3478963375091553,
      "step": 6508
    },
    {
      "epoch": 6.206512451171875e-06,
      "step": 6508,
      "training_step_time": 7.746680021286011
    },
    {
      "epoch": 6.2074661254882816e-06,
      "model_forward_time": 1.7247459888458252,
      "step": 6509
    },
    {
      "epoch": 6.2074661254882816e-06,
      "step": 6509,
      "training_step_time": 7.76685094833374
    },
    {
      "epoch": 6.208419799804687e-06,
      "grad_norm": 0.2455664724111557,
      "learning_rate": 9.993775837846432e-06,
      "loss": 0.0697,
      "step": 6510
    },
    {
      "epoch": 6.208419799804687e-06,
      "model_forward_time": 1.7460606098175049,
      "step": 6510
    },
    {
      "dynamics_loss_avg": 0.07694596610963345,
      "epoch": 6.208419799804687e-06,
      "step": 6510
    },
    {
      "action_loss_avg": 0.0062833928503096105,
      "epoch": 6.208419799804687e-06,
      "step": 6510
    },
    {
      "epoch": 6.208419799804687e-06,
      "step": 6510,
      "training_step_time": 8.592876195907593
    },
    {
      "epoch": 6.209373474121094e-06,
      "model_forward_time": 1.7175416946411133,
      "step": 6511
    },
    {
      "epoch": 6.209373474121094e-06,
      "step": 6511,
      "training_step_time": 8.73935317993164
    },
    {
      "epoch": 6.2103271484375e-06,
      "model_forward_time": 1.720036506652832,
      "step": 6512
    },
    {
      "epoch": 6.2103271484375e-06,
      "step": 6512,
      "training_step_time": 7.749066114425659
    },
    {
      "epoch": 6.211280822753906e-06,
      "model_forward_time": 2.7951698303222656,
      "step": 6513
    },
    {
      "epoch": 6.211280822753906e-06,
      "step": 6513,
      "training_step_time": 7.9193644523620605
    },
    {
      "epoch": 6.212234497070313e-06,
      "model_forward_time": 1.7294068336486816,
      "step": 6514
    },
    {
      "epoch": 6.212234497070313e-06,
      "step": 6514,
      "training_step_time": 8.749535083770752
    },
    {
      "epoch": 6.2131881713867184e-06,
      "model_forward_time": 1.729672908782959,
      "step": 6515
    },
    {
      "epoch": 6.2131881713867184e-06,
      "step": 6515,
      "training_step_time": 7.458223342895508
    },
    {
      "epoch": 6.214141845703125e-06,
      "model_forward_time": 1.7256903648376465,
      "step": 6516
    },
    {
      "epoch": 6.214141845703125e-06,
      "step": 6516,
      "training_step_time": 6.930519342422485
    },
    {
      "epoch": 6.215095520019532e-06,
      "model_forward_time": 1.7265715599060059,
      "step": 6517
    },
    {
      "epoch": 6.215095520019532e-06,
      "step": 6517,
      "training_step_time": 7.855737686157227
    },
    {
      "epoch": 6.216049194335937e-06,
      "model_forward_time": 1.7273895740509033,
      "step": 6518
    },
    {
      "epoch": 6.216049194335937e-06,
      "step": 6518,
      "training_step_time": 7.7764105796813965
    },
    {
      "epoch": 6.217002868652344e-06,
      "model_forward_time": 1.7416789531707764,
      "step": 6519
    },
    {
      "epoch": 6.217002868652344e-06,
      "step": 6519,
      "training_step_time": 7.904595136642456
    },
    {
      "epoch": 6.21795654296875e-06,
      "grad_norm": 0.20910429954528809,
      "learning_rate": 9.993693088045939e-06,
      "loss": 0.0709,
      "step": 6520
    },
    {
      "epoch": 6.21795654296875e-06,
      "model_forward_time": 1.722369909286499,
      "step": 6520
    },
    {
      "dynamics_loss_avg": 0.07030002772808075,
      "epoch": 6.21795654296875e-06,
      "step": 6520
    },
    {
      "action_loss_avg": 0.005411000736057758,
      "epoch": 6.21795654296875e-06,
      "step": 6520
    },
    {
      "epoch": 6.21795654296875e-06,
      "step": 6520,
      "training_step_time": 8.569932222366333
    },
    {
      "epoch": 6.218910217285156e-06,
      "model_forward_time": 1.752471923828125,
      "step": 6521
    },
    {
      "epoch": 6.218910217285156e-06,
      "step": 6521,
      "training_step_time": 9.170864343643188
    },
    {
      "epoch": 6.219863891601563e-06,
      "model_forward_time": 2.811408519744873,
      "step": 6522
    },
    {
      "epoch": 6.219863891601563e-06,
      "step": 6522,
      "training_step_time": 9.21937608718872
    },
    {
      "epoch": 6.2208175659179685e-06,
      "model_forward_time": 1.7205524444580078,
      "step": 6523
    },
    {
      "epoch": 6.2208175659179685e-06,
      "step": 6523,
      "training_step_time": 7.81537389755249
    },
    {
      "epoch": 6.221771240234375e-06,
      "model_forward_time": 1.723951816558838,
      "step": 6524
    },
    {
      "epoch": 6.221771240234375e-06,
      "step": 6524,
      "training_step_time": 7.567022085189819
    },
    {
      "epoch": 6.222724914550782e-06,
      "model_forward_time": 1.347815990447998,
      "step": 6525
    },
    {
      "epoch": 6.222724914550782e-06,
      "step": 6525,
      "training_step_time": 6.909096956253052
    },
    {
      "epoch": 6.223678588867187e-06,
      "model_forward_time": 1.7144770622253418,
      "step": 6526
    },
    {
      "epoch": 6.223678588867187e-06,
      "step": 6526,
      "training_step_time": 7.836064577102661
    },
    {
      "epoch": 6.224632263183594e-06,
      "model_forward_time": 1.7274792194366455,
      "step": 6527
    },
    {
      "epoch": 6.224632263183594e-06,
      "step": 6527,
      "training_step_time": 7.6895341873168945
    },
    {
      "epoch": 6.2255859375e-06,
      "model_forward_time": 1.7431671619415283,
      "step": 6528
    },
    {
      "epoch": 6.2255859375e-06,
      "step": 6528,
      "training_step_time": 9.000911235809326
    },
    {
      "epoch": 6.226539611816406e-06,
      "model_forward_time": 1.7239620685577393,
      "step": 6529
    },
    {
      "epoch": 6.226539611816406e-06,
      "step": 6529,
      "training_step_time": 7.7635393142700195
    },
    {
      "epoch": 6.227493286132813e-06,
      "grad_norm": 0.31044334173202515,
      "learning_rate": 9.99360979214268e-06,
      "loss": 0.073,
      "step": 6530
    },
    {
      "epoch": 6.227493286132813e-06,
      "model_forward_time": 1.7430450916290283,
      "step": 6530
    },
    {
      "dynamics_loss_avg": 0.07080449927598238,
      "epoch": 6.227493286132813e-06,
      "step": 6530
    },
    {
      "action_loss_avg": 0.007688987208530307,
      "epoch": 6.227493286132813e-06,
      "step": 6530
    },
    {
      "epoch": 6.227493286132813e-06,
      "step": 6530,
      "training_step_time": 11.196310043334961
    },
    {
      "epoch": 6.2284469604492185e-06,
      "model_forward_time": 1.725207805633545,
      "step": 6531
    },
    {
      "epoch": 6.2284469604492185e-06,
      "step": 6531,
      "training_step_time": 10.224279642105103
    },
    {
      "epoch": 6.229400634765625e-06,
      "model_forward_time": 2.9475817680358887,
      "step": 6532
    },
    {
      "epoch": 6.229400634765625e-06,
      "step": 6532,
      "training_step_time": 7.819916486740112
    },
    {
      "epoch": 6.230354309082032e-06,
      "model_forward_time": 1.7316739559173584,
      "step": 6533
    },
    {
      "epoch": 6.230354309082032e-06,
      "step": 6533,
      "training_step_time": 7.751014947891235
    },
    {
      "epoch": 6.231307983398437e-06,
      "model_forward_time": 1.7216393947601318,
      "step": 6534
    },
    {
      "epoch": 6.231307983398437e-06,
      "step": 6534,
      "training_step_time": 6.889102935791016
    },
    {
      "epoch": 6.232261657714844e-06,
      "model_forward_time": 1.72670578956604,
      "step": 6535
    },
    {
      "epoch": 6.232261657714844e-06,
      "step": 6535,
      "training_step_time": 7.741834878921509
    },
    {
      "epoch": 6.23321533203125e-06,
      "model_forward_time": 1.724790096282959,
      "step": 6536
    },
    {
      "epoch": 6.23321533203125e-06,
      "step": 6536,
      "training_step_time": 8.322286605834961
    },
    {
      "epoch": 6.234169006347656e-06,
      "model_forward_time": 1.7389435768127441,
      "step": 6537
    },
    {
      "epoch": 6.234169006347656e-06,
      "step": 6537,
      "training_step_time": 7.915511608123779
    },
    {
      "epoch": 6.235122680664063e-06,
      "model_forward_time": 1.3569581508636475,
      "step": 6538
    },
    {
      "epoch": 6.235122680664063e-06,
      "step": 6538,
      "training_step_time": 8.19314455986023
    },
    {
      "epoch": 6.2360763549804685e-06,
      "model_forward_time": 1.7434120178222656,
      "step": 6539
    },
    {
      "epoch": 6.2360763549804685e-06,
      "step": 6539,
      "training_step_time": 8.501810550689697
    },
    {
      "epoch": 6.237030029296875e-06,
      "grad_norm": 0.23469386994838715,
      "learning_rate": 9.993525950145761e-06,
      "loss": 0.0715,
      "step": 6540
    },
    {
      "epoch": 6.237030029296875e-06,
      "model_forward_time": 1.72831392288208,
      "step": 6540
    },
    {
      "dynamics_loss_avg": 0.06027372907847166,
      "epoch": 6.237030029296875e-06,
      "step": 6540
    },
    {
      "action_loss_avg": 0.004563474410679192,
      "epoch": 6.237030029296875e-06,
      "step": 6540
    },
    {
      "epoch": 6.237030029296875e-06,
      "step": 6540,
      "training_step_time": 10.017611026763916
    },
    {
      "epoch": 6.237983703613282e-06,
      "model_forward_time": 2.853076457977295,
      "step": 6541
    },
    {
      "epoch": 6.237983703613282e-06,
      "step": 6541,
      "training_step_time": 8.294956684112549
    },
    {
      "epoch": 6.238937377929687e-06,
      "model_forward_time": 1.724170207977295,
      "step": 6542
    },
    {
      "epoch": 6.238937377929687e-06,
      "step": 6542,
      "training_step_time": 7.769903898239136
    },
    {
      "epoch": 6.239891052246094e-06,
      "model_forward_time": 1.7232608795166016,
      "step": 6543
    },
    {
      "epoch": 6.239891052246094e-06,
      "step": 6543,
      "training_step_time": 6.866185426712036
    },
    {
      "epoch": 6.2408447265625e-06,
      "model_forward_time": 1.7226128578186035,
      "step": 6544
    },
    {
      "epoch": 6.2408447265625e-06,
      "step": 6544,
      "training_step_time": 7.720502138137817
    },
    {
      "epoch": 6.241798400878906e-06,
      "model_forward_time": 1.7256171703338623,
      "step": 6545
    },
    {
      "epoch": 6.241798400878906e-06,
      "step": 6545,
      "training_step_time": 7.66241192817688
    },
    {
      "epoch": 6.242752075195313e-06,
      "model_forward_time": 1.7279796600341797,
      "step": 6546
    },
    {
      "epoch": 6.242752075195313e-06,
      "step": 6546,
      "training_step_time": 8.00225043296814
    },
    {
      "epoch": 6.2437057495117185e-06,
      "model_forward_time": 1.7259235382080078,
      "step": 6547
    },
    {
      "epoch": 6.2437057495117185e-06,
      "step": 6547,
      "training_step_time": 7.65268874168396
    },
    {
      "epoch": 6.244659423828125e-06,
      "model_forward_time": 1.7450690269470215,
      "step": 6548
    },
    {
      "epoch": 6.244659423828125e-06,
      "step": 6548,
      "training_step_time": 8.412831783294678
    },
    {
      "epoch": 6.245613098144532e-06,
      "model_forward_time": 1.7203702926635742,
      "step": 6549
    },
    {
      "epoch": 6.245613098144532e-06,
      "step": 6549,
      "training_step_time": 10.364903926849365
    },
    {
      "epoch": 6.246566772460937e-06,
      "grad_norm": 0.3731256425380707,
      "learning_rate": 9.993441562064354e-06,
      "loss": 0.0727,
      "step": 6550
    },
    {
      "epoch": 6.246566772460937e-06,
      "model_forward_time": 2.8412749767303467,
      "step": 6550
    },
    {
      "dynamics_loss_avg": 0.07437116988003253,
      "epoch": 6.246566772460937e-06,
      "step": 6550
    },
    {
      "action_loss_avg": 0.011176082212477923,
      "epoch": 6.246566772460937e-06,
      "step": 6550
    },
    {
      "epoch": 6.246566772460937e-06,
      "step": 6550,
      "training_step_time": 8.5961332321167
    },
    {
      "epoch": 6.247520446777344e-06,
      "model_forward_time": 1.3465361595153809,
      "step": 6551
    },
    {
      "epoch": 6.247520446777344e-06,
      "step": 6551,
      "training_step_time": 8.661320924758911
    },
    {
      "epoch": 6.24847412109375e-06,
      "model_forward_time": 1.3530964851379395,
      "step": 6552
    },
    {
      "epoch": 6.24847412109375e-06,
      "step": 6552,
      "training_step_time": 7.819189071655273
    },
    {
      "epoch": 6.249427795410156e-06,
      "model_forward_time": 1.7206757068634033,
      "step": 6553
    },
    {
      "epoch": 6.249427795410156e-06,
      "step": 6553,
      "training_step_time": 7.737177610397339
    },
    {
      "epoch": 6.250381469726563e-06,
      "model_forward_time": 1.7195549011230469,
      "step": 6554
    },
    {
      "epoch": 6.250381469726563e-06,
      "step": 6554,
      "training_step_time": 7.689260959625244
    },
    {
      "epoch": 6.2513351440429685e-06,
      "model_forward_time": 1.7345709800720215,
      "step": 6555
    },
    {
      "epoch": 6.2513351440429685e-06,
      "step": 6555,
      "training_step_time": 8.303325414657593
    },
    {
      "epoch": 6.252288818359375e-06,
      "model_forward_time": 1.742802619934082,
      "step": 6556
    },
    {
      "epoch": 6.252288818359375e-06,
      "step": 6556,
      "training_step_time": 7.388408184051514
    },
    {
      "epoch": 6.253242492675782e-06,
      "model_forward_time": 1.7301316261291504,
      "step": 6557
    },
    {
      "epoch": 6.253242492675782e-06,
      "step": 6557,
      "training_step_time": 8.6003999710083
    },
    {
      "epoch": 6.254196166992187e-06,
      "model_forward_time": 1.7234139442443848,
      "step": 6558
    },
    {
      "epoch": 6.254196166992187e-06,
      "step": 6558,
      "training_step_time": 9.454894542694092
    },
    {
      "epoch": 6.255149841308594e-06,
      "model_forward_time": 1.7189631462097168,
      "step": 6559
    },
    {
      "epoch": 6.255149841308594e-06,
      "step": 6559,
      "training_step_time": 8.32167387008667
    },
    {
      "epoch": 6.256103515625e-06,
      "grad_norm": 0.34667515754699707,
      "learning_rate": 9.993356627907685e-06,
      "loss": 0.0701,
      "step": 6560
    },
    {
      "epoch": 6.256103515625e-06,
      "model_forward_time": 2.914639949798584,
      "step": 6560
    },
    {
      "dynamics_loss_avg": 0.07395300939679146,
      "epoch": 6.256103515625e-06,
      "step": 6560
    },
    {
      "action_loss_avg": 0.007456238614395261,
      "epoch": 6.256103515625e-06,
      "step": 6560
    },
    {
      "epoch": 6.256103515625e-06,
      "step": 6560,
      "training_step_time": 7.8274266719818115
    },
    {
      "epoch": 6.257057189941406e-06,
      "model_forward_time": 1.347529411315918,
      "step": 6561
    },
    {
      "epoch": 6.257057189941406e-06,
      "step": 6561,
      "training_step_time": 7.740951299667358
    },
    {
      "epoch": 6.258010864257813e-06,
      "model_forward_time": 1.3491003513336182,
      "step": 6562
    },
    {
      "epoch": 6.258010864257813e-06,
      "step": 6562,
      "training_step_time": 7.9005303382873535
    },
    {
      "epoch": 6.2589645385742185e-06,
      "model_forward_time": 1.7246906757354736,
      "step": 6563
    },
    {
      "epoch": 6.2589645385742185e-06,
      "step": 6563,
      "training_step_time": 7.724368572235107
    },
    {
      "epoch": 6.259918212890625e-06,
      "model_forward_time": 1.718071699142456,
      "step": 6564
    },
    {
      "epoch": 6.259918212890625e-06,
      "step": 6564,
      "training_step_time": 8.115313529968262
    },
    {
      "epoch": 6.260871887207032e-06,
      "model_forward_time": 1.7583856582641602,
      "step": 6565
    },
    {
      "epoch": 6.260871887207032e-06,
      "step": 6565,
      "training_step_time": 8.836447954177856
    },
    {
      "epoch": 6.261825561523437e-06,
      "model_forward_time": 1.7354686260223389,
      "step": 6566
    },
    {
      "epoch": 6.261825561523437e-06,
      "step": 6566,
      "training_step_time": 7.661772966384888
    },
    {
      "epoch": 6.262779235839844e-06,
      "model_forward_time": 1.7650139331817627,
      "step": 6567
    },
    {
      "epoch": 6.262779235839844e-06,
      "step": 6567,
      "training_step_time": 8.808046817779541
    },
    {
      "epoch": 6.26373291015625e-06,
      "model_forward_time": 1.7349467277526855,
      "step": 6568
    },
    {
      "epoch": 6.26373291015625e-06,
      "step": 6568,
      "training_step_time": 9.212233543395996
    },
    {
      "epoch": 6.264686584472656e-06,
      "model_forward_time": 2.457214593887329,
      "step": 6569
    },
    {
      "epoch": 6.264686584472656e-06,
      "step": 6569,
      "training_step_time": 8.607537746429443
    },
    {
      "epoch": 6.265640258789063e-06,
      "grad_norm": 0.3128010034561157,
      "learning_rate": 9.99327114768504e-06,
      "loss": 0.0721,
      "step": 6570
    },
    {
      "epoch": 6.265640258789063e-06,
      "model_forward_time": 1.7347230911254883,
      "step": 6570
    },
    {
      "dynamics_loss_avg": 0.07177467234432697,
      "epoch": 6.265640258789063e-06,
      "step": 6570
    },
    {
      "action_loss_avg": 0.007721851975657046,
      "epoch": 6.265640258789063e-06,
      "step": 6570
    },
    {
      "epoch": 6.265640258789063e-06,
      "step": 6570,
      "training_step_time": 8.08987283706665
    },
    {
      "epoch": 6.2665939331054685e-06,
      "model_forward_time": 1.7358057498931885,
      "step": 6571
    },
    {
      "epoch": 6.2665939331054685e-06,
      "step": 6571,
      "training_step_time": 7.853940963745117
    },
    {
      "epoch": 6.267547607421875e-06,
      "model_forward_time": 1.7420496940612793,
      "step": 6572
    },
    {
      "epoch": 6.267547607421875e-06,
      "step": 6572,
      "training_step_time": 8.535971879959106
    },
    {
      "epoch": 6.268501281738282e-06,
      "model_forward_time": 1.3641963005065918,
      "step": 6573
    },
    {
      "epoch": 6.268501281738282e-06,
      "step": 6573,
      "training_step_time": 7.477544069290161
    },
    {
      "epoch": 6.269454956054687e-06,
      "model_forward_time": 1.3604977130889893,
      "step": 6574
    },
    {
      "epoch": 6.269454956054687e-06,
      "step": 6574,
      "training_step_time": 8.694027185440063
    },
    {
      "epoch": 6.270408630371094e-06,
      "model_forward_time": 1.7373673915863037,
      "step": 6575
    },
    {
      "epoch": 6.270408630371094e-06,
      "step": 6575,
      "training_step_time": 7.71684455871582
    },
    {
      "epoch": 6.2713623046875e-06,
      "model_forward_time": 1.399582862854004,
      "step": 6576
    },
    {
      "epoch": 6.2713623046875e-06,
      "step": 6576,
      "training_step_time": 8.793437957763672
    },
    {
      "epoch": 6.272315979003906e-06,
      "model_forward_time": 1.723529577255249,
      "step": 6577
    },
    {
      "epoch": 6.272315979003906e-06,
      "step": 6577,
      "training_step_time": 9.11792254447937
    },
    {
      "epoch": 6.273269653320313e-06,
      "model_forward_time": 1.741612195968628,
      "step": 6578
    },
    {
      "epoch": 6.273269653320313e-06,
      "step": 6578,
      "training_step_time": 8.51990008354187
    },
    {
      "epoch": 6.2742233276367185e-06,
      "model_forward_time": 2.520392656326294,
      "step": 6579
    },
    {
      "epoch": 6.2742233276367185e-06,
      "step": 6579,
      "training_step_time": 7.70647931098938
    },
    {
      "epoch": 6.275177001953125e-06,
      "grad_norm": 0.2715633511543274,
      "learning_rate": 9.993185121405771e-06,
      "loss": 0.0714,
      "step": 6580
    },
    {
      "epoch": 6.275177001953125e-06,
      "model_forward_time": 1.7332870960235596,
      "step": 6580
    },
    {
      "dynamics_loss_avg": 0.0648851128295064,
      "epoch": 6.275177001953125e-06,
      "step": 6580
    },
    {
      "action_loss_avg": 0.007859713793732227,
      "epoch": 6.275177001953125e-06,
      "step": 6580
    },
    {
      "epoch": 6.275177001953125e-06,
      "step": 6580,
      "training_step_time": 7.883185386657715
    },
    {
      "epoch": 6.276130676269532e-06,
      "model_forward_time": 1.741445779800415,
      "step": 6581
    },
    {
      "epoch": 6.276130676269532e-06,
      "step": 6581,
      "training_step_time": 7.764260292053223
    },
    {
      "epoch": 6.277084350585937e-06,
      "model_forward_time": 1.7368879318237305,
      "step": 6582
    },
    {
      "epoch": 6.277084350585937e-06,
      "step": 6582,
      "training_step_time": 7.49152684211731
    },
    {
      "epoch": 6.278038024902344e-06,
      "model_forward_time": 1.7432563304901123,
      "step": 6583
    },
    {
      "epoch": 6.278038024902344e-06,
      "step": 6583,
      "training_step_time": 9.005804777145386
    },
    {
      "epoch": 6.27899169921875e-06,
      "model_forward_time": 1.7607698440551758,
      "step": 6584
    },
    {
      "epoch": 6.27899169921875e-06,
      "step": 6584,
      "training_step_time": 9.454013109207153
    },
    {
      "epoch": 6.279945373535156e-06,
      "model_forward_time": 1.7637083530426025,
      "step": 6585
    },
    {
      "epoch": 6.279945373535156e-06,
      "step": 6585,
      "training_step_time": 9.246950149536133
    },
    {
      "epoch": 6.280899047851563e-06,
      "model_forward_time": 1.7343111038208008,
      "step": 6586
    },
    {
      "epoch": 6.280899047851563e-06,
      "step": 6586,
      "training_step_time": 8.447091579437256
    },
    {
      "epoch": 6.2818527221679685e-06,
      "model_forward_time": 1.7347502708435059,
      "step": 6587
    },
    {
      "epoch": 6.2818527221679685e-06,
      "step": 6587,
      "training_step_time": 8.991815090179443
    },
    {
      "epoch": 6.282806396484375e-06,
      "model_forward_time": 2.8100671768188477,
      "step": 6588
    },
    {
      "epoch": 6.282806396484375e-06,
      "step": 6588,
      "training_step_time": 7.707852840423584
    },
    {
      "epoch": 6.283760070800782e-06,
      "model_forward_time": 1.3657610416412354,
      "step": 6589
    },
    {
      "epoch": 6.283760070800782e-06,
      "step": 6589,
      "training_step_time": 7.838467597961426
    },
    {
      "epoch": 6.284713745117187e-06,
      "grad_norm": 0.2668231725692749,
      "learning_rate": 9.993098549079285e-06,
      "loss": 0.0708,
      "step": 6590
    },
    {
      "epoch": 6.284713745117187e-06,
      "model_forward_time": 1.7397973537445068,
      "step": 6590
    },
    {
      "dynamics_loss_avg": 0.07184217981994152,
      "epoch": 6.284713745117187e-06,
      "step": 6590
    },
    {
      "action_loss_avg": 0.005899390997365117,
      "epoch": 6.284713745117187e-06,
      "step": 6590
    },
    {
      "epoch": 6.284713745117187e-06,
      "step": 6590,
      "training_step_time": 8.883704423904419
    },
    {
      "epoch": 6.285667419433594e-06,
      "model_forward_time": 1.737492561340332,
      "step": 6591
    },
    {
      "epoch": 6.285667419433594e-06,
      "step": 6591,
      "training_step_time": 7.82691216468811
    },
    {
      "epoch": 6.28662109375e-06,
      "model_forward_time": 1.734663963317871,
      "step": 6592
    },
    {
      "epoch": 6.28662109375e-06,
      "step": 6592,
      "training_step_time": 7.953234434127808
    },
    {
      "epoch": 6.287574768066406e-06,
      "model_forward_time": 1.7497613430023193,
      "step": 6593
    },
    {
      "epoch": 6.287574768066406e-06,
      "step": 6593,
      "training_step_time": 7.556741237640381
    },
    {
      "epoch": 6.288528442382813e-06,
      "model_forward_time": 1.7483243942260742,
      "step": 6594
    },
    {
      "epoch": 6.288528442382813e-06,
      "step": 6594,
      "training_step_time": 7.692777156829834
    },
    {
      "epoch": 6.2894821166992185e-06,
      "model_forward_time": 1.7345304489135742,
      "step": 6595
    },
    {
      "epoch": 6.2894821166992185e-06,
      "step": 6595,
      "training_step_time": 8.927704334259033
    },
    {
      "epoch": 6.290435791015625e-06,
      "model_forward_time": 1.7347533702850342,
      "step": 6596
    },
    {
      "epoch": 6.290435791015625e-06,
      "step": 6596,
      "training_step_time": 8.984218835830688
    },
    {
      "epoch": 6.291389465332032e-06,
      "model_forward_time": 1.7364425659179688,
      "step": 6597
    },
    {
      "epoch": 6.291389465332032e-06,
      "step": 6597,
      "training_step_time": 8.466119766235352
    },
    {
      "epoch": 6.292343139648437e-06,
      "model_forward_time": 2.87335467338562,
      "step": 6598
    },
    {
      "epoch": 6.292343139648437e-06,
      "step": 6598,
      "training_step_time": 7.65848708152771
    },
    {
      "epoch": 6.293296813964844e-06,
      "model_forward_time": 1.7343089580535889,
      "step": 6599
    },
    {
      "epoch": 6.293296813964844e-06,
      "step": 6599,
      "training_step_time": 7.758417367935181
    },
    {
      "epoch": 6.29425048828125e-06,
      "grad_norm": 0.23830103874206543,
      "learning_rate": 9.993011430715047e-06,
      "loss": 0.0721,
      "step": 6600
    },
    {
      "epoch": 6.29425048828125e-06,
      "model_forward_time": 1.741060495376587,
      "step": 6600
    },
    {
      "dynamics_loss_avg": 0.0568078676238656,
      "epoch": 6.29425048828125e-06,
      "step": 6600
    },
    {
      "action_loss_avg": 0.005407141719479114,
      "epoch": 6.29425048828125e-06,
      "step": 6600
    },
    {
      "epoch": 6.29425048828125e-06,
      "step": 6600,
      "training_step_time": 7.6383373737335205
    },
    {
      "epoch": 6.295204162597656e-06,
      "model_forward_time": 1.7290334701538086,
      "step": 6601
    },
    {
      "epoch": 6.295204162597656e-06,
      "step": 6601,
      "training_step_time": 7.7552430629730225
    },
    {
      "epoch": 6.296157836914063e-06,
      "model_forward_time": 1.7364470958709717,
      "step": 6602
    },
    {
      "epoch": 6.296157836914063e-06,
      "step": 6602,
      "training_step_time": 8.428640365600586
    },
    {
      "epoch": 6.2971115112304685e-06,
      "model_forward_time": 1.7561604976654053,
      "step": 6603
    },
    {
      "epoch": 6.2971115112304685e-06,
      "step": 6603,
      "training_step_time": 9.61107873916626
    },
    {
      "epoch": 6.298065185546875e-06,
      "model_forward_time": 1.7322838306427002,
      "step": 6604
    },
    {
      "epoch": 6.298065185546875e-06,
      "step": 6604,
      "training_step_time": 8.393155336380005
    },
    {
      "epoch": 6.299018859863282e-06,
      "model_forward_time": 1.7345070838928223,
      "step": 6605
    },
    {
      "epoch": 6.299018859863282e-06,
      "step": 6605,
      "training_step_time": 9.910547018051147
    },
    {
      "epoch": 6.299972534179687e-06,
      "model_forward_time": 1.7345752716064453,
      "step": 6606
    },
    {
      "epoch": 6.299972534179687e-06,
      "step": 6606,
      "training_step_time": 8.555580854415894
    },
    {
      "epoch": 6.300926208496094e-06,
      "model_forward_time": 2.897622585296631,
      "step": 6607
    },
    {
      "epoch": 6.300926208496094e-06,
      "step": 6607,
      "training_step_time": 7.670629501342773
    },
    {
      "epoch": 6.3018798828125e-06,
      "model_forward_time": 1.7344310283660889,
      "step": 6608
    },
    {
      "epoch": 6.3018798828125e-06,
      "step": 6608,
      "training_step_time": 7.82693886756897
    },
    {
      "epoch": 6.302833557128906e-06,
      "model_forward_time": 1.7271089553833008,
      "step": 6609
    },
    {
      "epoch": 6.302833557128906e-06,
      "step": 6609,
      "training_step_time": 7.689907789230347
    },
    {
      "epoch": 6.303787231445313e-06,
      "grad_norm": 0.25262701511383057,
      "learning_rate": 9.992923766322586e-06,
      "loss": 0.0747,
      "step": 6610
    },
    {
      "epoch": 6.303787231445313e-06,
      "model_forward_time": 1.7326228618621826,
      "step": 6610
    },
    {
      "dynamics_loss_avg": 0.07136254534125328,
      "epoch": 6.303787231445313e-06,
      "step": 6610
    },
    {
      "action_loss_avg": 0.007767544174566865,
      "epoch": 6.303787231445313e-06,
      "step": 6610
    },
    {
      "epoch": 6.303787231445313e-06,
      "step": 6610,
      "training_step_time": 7.735502004623413
    },
    {
      "epoch": 6.3047409057617185e-06,
      "model_forward_time": 1.7352709770202637,
      "step": 6611
    },
    {
      "epoch": 6.3047409057617185e-06,
      "step": 6611,
      "training_step_time": 7.877904653549194
    },
    {
      "epoch": 6.305694580078125e-06,
      "model_forward_time": 1.7654330730438232,
      "step": 6612
    },
    {
      "epoch": 6.305694580078125e-06,
      "step": 6612,
      "training_step_time": 9.416752576828003
    },
    {
      "epoch": 6.306648254394532e-06,
      "model_forward_time": 1.7322673797607422,
      "step": 6613
    },
    {
      "epoch": 6.306648254394532e-06,
      "step": 6613,
      "training_step_time": 8.31845235824585
    },
    {
      "epoch": 6.307601928710937e-06,
      "model_forward_time": 1.7382667064666748,
      "step": 6614
    },
    {
      "epoch": 6.307601928710937e-06,
      "step": 6614,
      "training_step_time": 7.8274242877960205
    },
    {
      "epoch": 6.308555603027344e-06,
      "model_forward_time": 1.731896162033081,
      "step": 6615
    },
    {
      "epoch": 6.308555603027344e-06,
      "step": 6615,
      "training_step_time": 7.895160913467407
    },
    {
      "epoch": 6.30950927734375e-06,
      "model_forward_time": 2.8734140396118164,
      "step": 6616
    },
    {
      "epoch": 6.30950927734375e-06,
      "step": 6616,
      "training_step_time": 7.876169919967651
    },
    {
      "epoch": 6.310462951660156e-06,
      "model_forward_time": 1.3653030395507812,
      "step": 6617
    },
    {
      "epoch": 6.310462951660156e-06,
      "step": 6617,
      "training_step_time": 7.935234069824219
    },
    {
      "epoch": 6.311416625976563e-06,
      "model_forward_time": 1.73874831199646,
      "step": 6618
    },
    {
      "epoch": 6.311416625976563e-06,
      "step": 6618,
      "training_step_time": 7.828319787979126
    },
    {
      "epoch": 6.3123703002929685e-06,
      "model_forward_time": 1.7429730892181396,
      "step": 6619
    },
    {
      "epoch": 6.3123703002929685e-06,
      "step": 6619,
      "training_step_time": 8.355281591415405
    },
    {
      "epoch": 6.313323974609375e-06,
      "grad_norm": 0.2265031784772873,
      "learning_rate": 9.99283555591149e-06,
      "loss": 0.0767,
      "step": 6620
    },
    {
      "epoch": 6.313323974609375e-06,
      "model_forward_time": 1.7405791282653809,
      "step": 6620
    },
    {
      "dynamics_loss_avg": 0.06275535998865962,
      "epoch": 6.313323974609375e-06,
      "step": 6620
    },
    {
      "action_loss_avg": 0.00834253573557362,
      "epoch": 6.313323974609375e-06,
      "step": 6620
    },
    {
      "epoch": 6.313323974609375e-06,
      "step": 6620,
      "training_step_time": 7.963148832321167
    },
    {
      "epoch": 6.314277648925782e-06,
      "model_forward_time": 1.7585031986236572,
      "step": 6621
    },
    {
      "epoch": 6.314277648925782e-06,
      "step": 6621,
      "training_step_time": 7.769948959350586
    },
    {
      "epoch": 6.315231323242187e-06,
      "model_forward_time": 1.7568941116333008,
      "step": 6622
    },
    {
      "epoch": 6.315231323242187e-06,
      "step": 6622,
      "training_step_time": 7.810893774032593
    },
    {
      "epoch": 6.316184997558594e-06,
      "model_forward_time": 1.732553243637085,
      "step": 6623
    },
    {
      "epoch": 6.316184997558594e-06,
      "step": 6623,
      "training_step_time": 8.665745258331299
    },
    {
      "epoch": 6.317138671875e-06,
      "model_forward_time": 1.3722245693206787,
      "step": 6624
    },
    {
      "epoch": 6.317138671875e-06,
      "step": 6624,
      "training_step_time": 7.819227695465088
    },
    {
      "epoch": 6.318092346191406e-06,
      "model_forward_time": 2.8188443183898926,
      "step": 6625
    },
    {
      "epoch": 6.318092346191406e-06,
      "step": 6625,
      "training_step_time": 7.685112237930298
    },
    {
      "epoch": 6.319046020507813e-06,
      "model_forward_time": 1.7402873039245605,
      "step": 6626
    },
    {
      "epoch": 6.319046020507813e-06,
      "step": 6626,
      "training_step_time": 7.822714567184448
    },
    {
      "epoch": 6.3199996948242185e-06,
      "model_forward_time": 1.7367215156555176,
      "step": 6627
    },
    {
      "epoch": 6.3199996948242185e-06,
      "step": 6627,
      "training_step_time": 7.716257333755493
    },
    {
      "epoch": 6.320953369140625e-06,
      "model_forward_time": 1.7378766536712646,
      "step": 6628
    },
    {
      "epoch": 6.320953369140625e-06,
      "step": 6628,
      "training_step_time": 7.543308258056641
    },
    {
      "epoch": 6.321907043457031e-06,
      "model_forward_time": 1.743727445602417,
      "step": 6629
    },
    {
      "epoch": 6.321907043457031e-06,
      "step": 6629,
      "training_step_time": 7.952876567840576
    },
    {
      "epoch": 6.322860717773437e-06,
      "grad_norm": 0.2504456341266632,
      "learning_rate": 9.992746799491405e-06,
      "loss": 0.0725,
      "step": 6630
    },
    {
      "epoch": 6.322860717773437e-06,
      "model_forward_time": 1.7639400959014893,
      "step": 6630
    },
    {
      "dynamics_loss_avg": 0.07191462256014347,
      "epoch": 6.322860717773437e-06,
      "step": 6630
    },
    {
      "action_loss_avg": 0.007775389426387847,
      "epoch": 6.322860717773437e-06,
      "step": 6630
    },
    {
      "epoch": 6.322860717773437e-06,
      "step": 6630,
      "training_step_time": 9.361106157302856
    },
    {
      "epoch": 6.323814392089844e-06,
      "model_forward_time": 1.767824411392212,
      "step": 6631
    },
    {
      "epoch": 6.323814392089844e-06,
      "step": 6631,
      "training_step_time": 8.260229110717773
    },
    {
      "epoch": 6.32476806640625e-06,
      "model_forward_time": 1.742823839187622,
      "step": 6632
    },
    {
      "epoch": 6.32476806640625e-06,
      "step": 6632,
      "training_step_time": 9.85338282585144
    },
    {
      "epoch": 6.325721740722656e-06,
      "model_forward_time": 1.7331278324127197,
      "step": 6633
    },
    {
      "epoch": 6.325721740722656e-06,
      "step": 6633,
      "training_step_time": 9.229501962661743
    },
    {
      "epoch": 6.326675415039063e-06,
      "model_forward_time": 1.728464126586914,
      "step": 6634
    },
    {
      "epoch": 6.326675415039063e-06,
      "step": 6634,
      "training_step_time": 7.679886817932129
    },
    {
      "epoch": 6.3276290893554685e-06,
      "model_forward_time": 2.929694175720215,
      "step": 6635
    },
    {
      "epoch": 6.3276290893554685e-06,
      "step": 6635,
      "training_step_time": 8.763050079345703
    },
    {
      "epoch": 6.328582763671875e-06,
      "model_forward_time": 1.7436919212341309,
      "step": 6636
    },
    {
      "epoch": 6.328582763671875e-06,
      "step": 6636,
      "training_step_time": 7.674922227859497
    },
    {
      "epoch": 6.329536437988281e-06,
      "model_forward_time": 1.7421197891235352,
      "step": 6637
    },
    {
      "epoch": 6.329536437988281e-06,
      "step": 6637,
      "training_step_time": 7.378174543380737
    },
    {
      "epoch": 6.330490112304687e-06,
      "model_forward_time": 1.732104778289795,
      "step": 6638
    },
    {
      "epoch": 6.330490112304687e-06,
      "step": 6638,
      "training_step_time": 7.587811231613159
    },
    {
      "epoch": 6.331443786621094e-06,
      "model_forward_time": 1.7577235698699951,
      "step": 6639
    },
    {
      "epoch": 6.331443786621094e-06,
      "step": 6639,
      "training_step_time": 7.591867923736572
    },
    {
      "epoch": 6.3323974609375e-06,
      "grad_norm": 0.269997775554657,
      "learning_rate": 9.992657497072033e-06,
      "loss": 0.0712,
      "step": 6640
    },
    {
      "epoch": 6.3323974609375e-06,
      "model_forward_time": 1.7496562004089355,
      "step": 6640
    },
    {
      "dynamics_loss_avg": 0.0874441109597683,
      "epoch": 6.3323974609375e-06,
      "step": 6640
    },
    {
      "action_loss_avg": 0.007368244568351656,
      "epoch": 6.3323974609375e-06,
      "step": 6640
    },
    {
      "epoch": 6.3323974609375e-06,
      "step": 6640,
      "training_step_time": 8.188165426254272
    },
    {
      "epoch": 6.333351135253906e-06,
      "model_forward_time": 1.7423477172851562,
      "step": 6641
    },
    {
      "epoch": 6.333351135253906e-06,
      "step": 6641,
      "training_step_time": 7.715185880661011
    },
    {
      "epoch": 6.334304809570313e-06,
      "model_forward_time": 1.7322616577148438,
      "step": 6642
    },
    {
      "epoch": 6.334304809570313e-06,
      "step": 6642,
      "training_step_time": 8.178727865219116
    },
    {
      "epoch": 6.3352584838867186e-06,
      "model_forward_time": 1.7469158172607422,
      "step": 6643
    },
    {
      "epoch": 6.3352584838867186e-06,
      "step": 6643,
      "training_step_time": 7.718256950378418
    },
    {
      "epoch": 6.336212158203125e-06,
      "model_forward_time": 2.899874210357666,
      "step": 6644
    },
    {
      "epoch": 6.336212158203125e-06,
      "step": 6644,
      "training_step_time": 7.6857404708862305
    },
    {
      "epoch": 6.337165832519531e-06,
      "model_forward_time": 1.7338268756866455,
      "step": 6645
    },
    {
      "epoch": 6.337165832519531e-06,
      "step": 6645,
      "training_step_time": 7.668944597244263
    },
    {
      "epoch": 6.338119506835937e-06,
      "model_forward_time": 1.7465765476226807,
      "step": 6646
    },
    {
      "epoch": 6.338119506835937e-06,
      "step": 6646,
      "training_step_time": 7.2098894119262695
    },
    {
      "epoch": 6.339073181152344e-06,
      "model_forward_time": 1.7428803443908691,
      "step": 6647
    },
    {
      "epoch": 6.339073181152344e-06,
      "step": 6647,
      "training_step_time": 8.368355989456177
    },
    {
      "epoch": 6.34002685546875e-06,
      "model_forward_time": 1.7520067691802979,
      "step": 6648
    },
    {
      "epoch": 6.34002685546875e-06,
      "step": 6648,
      "training_step_time": 8.22404956817627
    },
    {
      "epoch": 6.340980529785156e-06,
      "model_forward_time": 1.750762701034546,
      "step": 6649
    },
    {
      "epoch": 6.340980529785156e-06,
      "step": 6649,
      "training_step_time": 7.667839050292969
    },
    {
      "epoch": 6.341934204101563e-06,
      "grad_norm": 0.2519463300704956,
      "learning_rate": 9.992567648663147e-06,
      "loss": 0.0672,
      "step": 6650
    },
    {
      "epoch": 6.341934204101563e-06,
      "model_forward_time": 1.74491286277771,
      "step": 6650
    },
    {
      "dynamics_loss_avg": 0.06440868489444256,
      "epoch": 6.341934204101563e-06,
      "step": 6650
    },
    {
      "action_loss_avg": 0.0051023963605985045,
      "epoch": 6.341934204101563e-06,
      "step": 6650
    },
    {
      "epoch": 6.341934204101563e-06,
      "step": 6650,
      "training_step_time": 8.952622413635254
    },
    {
      "epoch": 6.3428878784179686e-06,
      "model_forward_time": 1.7271595001220703,
      "step": 6651
    },
    {
      "epoch": 6.3428878784179686e-06,
      "step": 6651,
      "training_step_time": 7.839211702346802
    },
    {
      "epoch": 6.343841552734375e-06,
      "model_forward_time": 1.7347826957702637,
      "step": 6652
    },
    {
      "epoch": 6.343841552734375e-06,
      "step": 6652,
      "training_step_time": 7.812782287597656
    },
    {
      "epoch": 6.344795227050781e-06,
      "model_forward_time": 2.8486921787261963,
      "step": 6653
    },
    {
      "epoch": 6.344795227050781e-06,
      "step": 6653,
      "training_step_time": 7.660367965698242
    },
    {
      "epoch": 6.345748901367187e-06,
      "model_forward_time": 1.73496675491333,
      "step": 6654
    },
    {
      "epoch": 6.345748901367187e-06,
      "step": 6654,
      "training_step_time": 7.821882009506226
    },
    {
      "epoch": 6.346702575683594e-06,
      "model_forward_time": 1.7475402355194092,
      "step": 6655
    },
    {
      "epoch": 6.346702575683594e-06,
      "step": 6655,
      "training_step_time": 8.04364562034607
    },
    {
      "epoch": 6.34765625e-06,
      "model_forward_time": 1.7349684238433838,
      "step": 6656
    },
    {
      "epoch": 6.34765625e-06,
      "step": 6656,
      "training_step_time": 8.515550374984741
    },
    {
      "epoch": 6.348609924316406e-06,
      "model_forward_time": 1.7510349750518799,
      "step": 6657
    },
    {
      "epoch": 6.348609924316406e-06,
      "step": 6657,
      "training_step_time": 7.593930006027222
    },
    {
      "epoch": 6.349563598632813e-06,
      "model_forward_time": 1.7335550785064697,
      "step": 6658
    },
    {
      "epoch": 6.349563598632813e-06,
      "step": 6658,
      "training_step_time": 8.199858665466309
    },
    {
      "epoch": 6.3505172729492186e-06,
      "model_forward_time": 1.7699556350708008,
      "step": 6659
    },
    {
      "epoch": 6.3505172729492186e-06,
      "step": 6659,
      "training_step_time": 8.271847248077393
    },
    {
      "epoch": 6.351470947265625e-06,
      "grad_norm": 0.16593120992183685,
      "learning_rate": 9.992477254274568e-06,
      "loss": 0.0703,
      "step": 6660
    },
    {
      "epoch": 6.351470947265625e-06,
      "model_forward_time": 1.7388486862182617,
      "step": 6660
    },
    {
      "dynamics_loss_avg": 0.04736769758164883,
      "epoch": 6.351470947265625e-06,
      "step": 6660
    },
    {
      "action_loss_avg": 0.0047858251258730885,
      "epoch": 6.351470947265625e-06,
      "step": 6660
    },
    {
      "epoch": 6.351470947265625e-06,
      "step": 6660,
      "training_step_time": 7.837296009063721
    },
    {
      "epoch": 6.352424621582031e-06,
      "model_forward_time": 1.7332541942596436,
      "step": 6661
    },
    {
      "epoch": 6.352424621582031e-06,
      "step": 6661,
      "training_step_time": 8.263146162033081
    },
    {
      "epoch": 6.3533782958984374e-06,
      "model_forward_time": 2.8640525341033936,
      "step": 6662
    },
    {
      "epoch": 6.3533782958984374e-06,
      "step": 6662,
      "training_step_time": 7.783334732055664
    },
    {
      "epoch": 6.354331970214844e-06,
      "model_forward_time": 1.74000883102417,
      "step": 6663
    },
    {
      "epoch": 6.354331970214844e-06,
      "step": 6663,
      "training_step_time": 7.835118770599365
    },
    {
      "epoch": 6.35528564453125e-06,
      "model_forward_time": 1.732440710067749,
      "step": 6664
    },
    {
      "epoch": 6.35528564453125e-06,
      "step": 6664,
      "training_step_time": 7.321149587631226
    },
    {
      "epoch": 6.356239318847656e-06,
      "model_forward_time": 1.7377638816833496,
      "step": 6665
    },
    {
      "epoch": 6.356239318847656e-06,
      "step": 6665,
      "training_step_time": 7.6847922801971436
    },
    {
      "epoch": 6.357192993164063e-06,
      "model_forward_time": 1.762214183807373,
      "step": 6666
    },
    {
      "epoch": 6.357192993164063e-06,
      "step": 6666,
      "training_step_time": 7.819777727127075
    },
    {
      "epoch": 6.3581466674804686e-06,
      "model_forward_time": 1.3670086860656738,
      "step": 6667
    },
    {
      "epoch": 6.3581466674804686e-06,
      "step": 6667,
      "training_step_time": 8.561100482940674
    },
    {
      "epoch": 6.359100341796875e-06,
      "model_forward_time": 1.7523627281188965,
      "step": 6668
    },
    {
      "epoch": 6.359100341796875e-06,
      "step": 6668,
      "training_step_time": 7.83835768699646
    },
    {
      "epoch": 6.360054016113281e-06,
      "model_forward_time": 1.7362017631530762,
      "step": 6669
    },
    {
      "epoch": 6.360054016113281e-06,
      "step": 6669,
      "training_step_time": 9.244897842407227
    },
    {
      "epoch": 6.3610076904296874e-06,
      "grad_norm": 0.21062567830085754,
      "learning_rate": 9.992386313916183e-06,
      "loss": 0.0739,
      "step": 6670
    },
    {
      "epoch": 6.3610076904296874e-06,
      "model_forward_time": 1.7366929054260254,
      "step": 6670
    },
    {
      "dynamics_loss_avg": 0.07086035329848528,
      "epoch": 6.3610076904296874e-06,
      "step": 6670
    },
    {
      "action_loss_avg": 0.0058433237252756955,
      "epoch": 6.3610076904296874e-06,
      "step": 6670
    },
    {
      "epoch": 6.3610076904296874e-06,
      "step": 6670,
      "training_step_time": 9.094935178756714
    },
    {
      "epoch": 6.361961364746094e-06,
      "model_forward_time": 1.7443552017211914,
      "step": 6671
    },
    {
      "epoch": 6.361961364746094e-06,
      "step": 6671,
      "training_step_time": 8.74127745628357
    },
    {
      "epoch": 6.3629150390625e-06,
      "model_forward_time": 2.927492380142212,
      "step": 6672
    },
    {
      "epoch": 6.3629150390625e-06,
      "step": 6672,
      "training_step_time": 7.885207653045654
    },
    {
      "epoch": 6.363868713378906e-06,
      "model_forward_time": 1.7408978939056396,
      "step": 6673
    },
    {
      "epoch": 6.363868713378906e-06,
      "step": 6673,
      "training_step_time": 7.8129706382751465
    },
    {
      "epoch": 6.364822387695313e-06,
      "model_forward_time": 1.7369015216827393,
      "step": 6674
    },
    {
      "epoch": 6.364822387695313e-06,
      "step": 6674,
      "training_step_time": 7.692227363586426
    },
    {
      "epoch": 6.365776062011719e-06,
      "model_forward_time": 1.752455234527588,
      "step": 6675
    },
    {
      "epoch": 6.365776062011719e-06,
      "step": 6675,
      "training_step_time": 10.547540664672852
    },
    {
      "epoch": 6.366729736328125e-06,
      "model_forward_time": 1.7364764213562012,
      "step": 6676
    },
    {
      "epoch": 6.366729736328125e-06,
      "step": 6676,
      "training_step_time": 10.056361675262451
    },
    {
      "epoch": 6.367683410644531e-06,
      "model_forward_time": 1.7599785327911377,
      "step": 6677
    },
    {
      "epoch": 6.367683410644531e-06,
      "step": 6677,
      "training_step_time": 7.764352321624756
    },
    {
      "epoch": 6.3686370849609374e-06,
      "model_forward_time": 1.7323627471923828,
      "step": 6678
    },
    {
      "epoch": 6.3686370849609374e-06,
      "step": 6678,
      "training_step_time": 7.811295747756958
    },
    {
      "epoch": 6.369590759277344e-06,
      "model_forward_time": 1.7425851821899414,
      "step": 6679
    },
    {
      "epoch": 6.369590759277344e-06,
      "step": 6679,
      "training_step_time": 8.797356605529785
    },
    {
      "epoch": 6.37054443359375e-06,
      "grad_norm": 0.29730224609375,
      "learning_rate": 9.992294827597935e-06,
      "loss": 0.0725,
      "step": 6680
    },
    {
      "epoch": 6.37054443359375e-06,
      "model_forward_time": 1.7358624935150146,
      "step": 6680
    },
    {
      "dynamics_loss_avg": 0.06225806213915348,
      "epoch": 6.37054443359375e-06,
      "step": 6680
    },
    {
      "action_loss_avg": 0.006203052657656372,
      "epoch": 6.37054443359375e-06,
      "step": 6680
    },
    {
      "epoch": 6.37054443359375e-06,
      "step": 6680,
      "training_step_time": 7.7914557456970215
    },
    {
      "epoch": 6.371498107910156e-06,
      "model_forward_time": 2.916978359222412,
      "step": 6681
    },
    {
      "epoch": 6.371498107910156e-06,
      "step": 6681,
      "training_step_time": 7.961892366409302
    },
    {
      "epoch": 6.372451782226563e-06,
      "model_forward_time": 1.7387800216674805,
      "step": 6682
    },
    {
      "epoch": 6.372451782226563e-06,
      "step": 6682,
      "training_step_time": 7.799616575241089
    },
    {
      "epoch": 6.373405456542969e-06,
      "model_forward_time": 1.7338008880615234,
      "step": 6683
    },
    {
      "epoch": 6.373405456542969e-06,
      "step": 6683,
      "training_step_time": 7.564833641052246
    },
    {
      "epoch": 6.374359130859375e-06,
      "model_forward_time": 1.756904125213623,
      "step": 6684
    },
    {
      "epoch": 6.374359130859375e-06,
      "step": 6684,
      "training_step_time": 9.338797092437744
    },
    {
      "epoch": 6.375312805175781e-06,
      "model_forward_time": 1.739133596420288,
      "step": 6685
    },
    {
      "epoch": 6.375312805175781e-06,
      "step": 6685,
      "training_step_time": 8.552303552627563
    },
    {
      "epoch": 6.3762664794921874e-06,
      "model_forward_time": 1.744159460067749,
      "step": 6686
    },
    {
      "epoch": 6.3762664794921874e-06,
      "step": 6686,
      "training_step_time": 9.514951229095459
    },
    {
      "epoch": 6.377220153808594e-06,
      "model_forward_time": 1.731177568435669,
      "step": 6687
    },
    {
      "epoch": 6.377220153808594e-06,
      "step": 6687,
      "training_step_time": 7.88554048538208
    },
    {
      "epoch": 6.378173828125e-06,
      "model_forward_time": 1.7369515895843506,
      "step": 6688
    },
    {
      "epoch": 6.378173828125e-06,
      "step": 6688,
      "training_step_time": 8.945421934127808
    },
    {
      "epoch": 6.379127502441406e-06,
      "model_forward_time": 1.7339134216308594,
      "step": 6689
    },
    {
      "epoch": 6.379127502441406e-06,
      "step": 6689,
      "training_step_time": 7.919777154922485
    },
    {
      "epoch": 6.380081176757813e-06,
      "grad_norm": 0.34734687209129333,
      "learning_rate": 9.992202795329832e-06,
      "loss": 0.0753,
      "step": 6690
    },
    {
      "epoch": 6.380081176757813e-06,
      "model_forward_time": 2.8740391731262207,
      "step": 6690
    },
    {
      "dynamics_loss_avg": 0.06769142989069224,
      "epoch": 6.380081176757813e-06,
      "step": 6690
    },
    {
      "action_loss_avg": 0.004723838088102639,
      "epoch": 6.380081176757813e-06,
      "step": 6690
    },
    {
      "epoch": 6.380081176757813e-06,
      "step": 6690,
      "training_step_time": 7.9573657512664795
    },
    {
      "epoch": 6.381034851074219e-06,
      "model_forward_time": 1.7395668029785156,
      "step": 6691
    },
    {
      "epoch": 6.381034851074219e-06,
      "step": 6691,
      "training_step_time": 7.852558612823486
    },
    {
      "epoch": 6.381988525390625e-06,
      "model_forward_time": 1.7362489700317383,
      "step": 6692
    },
    {
      "epoch": 6.381988525390625e-06,
      "step": 6692,
      "training_step_time": 7.630576848983765
    },
    {
      "epoch": 6.382942199707031e-06,
      "model_forward_time": 1.7635440826416016,
      "step": 6693
    },
    {
      "epoch": 6.382942199707031e-06,
      "step": 6693,
      "training_step_time": 9.792109966278076
    },
    {
      "epoch": 6.3838958740234375e-06,
      "model_forward_time": 1.7393436431884766,
      "step": 6694
    },
    {
      "epoch": 6.3838958740234375e-06,
      "step": 6694,
      "training_step_time": 7.819063663482666
    },
    {
      "epoch": 6.384849548339844e-06,
      "model_forward_time": 1.738727331161499,
      "step": 6695
    },
    {
      "epoch": 6.384849548339844e-06,
      "step": 6695,
      "training_step_time": 9.427973985671997
    },
    {
      "epoch": 6.38580322265625e-06,
      "model_forward_time": 1.7376022338867188,
      "step": 6696
    },
    {
      "epoch": 6.38580322265625e-06,
      "step": 6696,
      "training_step_time": 8.482864379882812
    },
    {
      "epoch": 6.386756896972656e-06,
      "model_forward_time": 1.7550604343414307,
      "step": 6697
    },
    {
      "epoch": 6.386756896972656e-06,
      "step": 6697,
      "training_step_time": 9.06774115562439
    },
    {
      "epoch": 6.387710571289063e-06,
      "model_forward_time": 1.741621732711792,
      "step": 6698
    },
    {
      "epoch": 6.387710571289063e-06,
      "step": 6698,
      "training_step_time": 7.903826951980591
    },
    {
      "epoch": 6.388664245605469e-06,
      "model_forward_time": 2.839052200317383,
      "step": 6699
    },
    {
      "epoch": 6.388664245605469e-06,
      "step": 6699,
      "training_step_time": 7.867427110671997
    },
    {
      "epoch": 6.389617919921875e-06,
      "grad_norm": 0.38203248381614685,
      "learning_rate": 9.992110217121935e-06,
      "loss": 0.074,
      "step": 6700
    },
    {
      "epoch": 6.389617919921875e-06,
      "model_forward_time": 1.7363348007202148,
      "step": 6700
    },
    {
      "dynamics_loss_avg": 0.08607855625450611,
      "epoch": 6.389617919921875e-06,
      "step": 6700
    },
    {
      "action_loss_avg": 0.006218689866364003,
      "epoch": 6.389617919921875e-06,
      "step": 6700
    },
    {
      "epoch": 6.389617919921875e-06,
      "step": 6700,
      "training_step_time": 7.955731153488159
    },
    {
      "epoch": 6.390571594238281e-06,
      "model_forward_time": 1.7371258735656738,
      "step": 6701
    },
    {
      "epoch": 6.390571594238281e-06,
      "step": 6701,
      "training_step_time": 7.529855251312256
    },
    {
      "epoch": 6.3915252685546875e-06,
      "model_forward_time": 1.7505171298980713,
      "step": 6702
    },
    {
      "epoch": 6.3915252685546875e-06,
      "step": 6702,
      "training_step_time": 7.841313600540161
    },
    {
      "epoch": 6.392478942871094e-06,
      "model_forward_time": 1.7351107597351074,
      "step": 6703
    },
    {
      "epoch": 6.392478942871094e-06,
      "step": 6703,
      "training_step_time": 8.476226091384888
    },
    {
      "epoch": 6.3934326171875e-06,
      "model_forward_time": 1.7376830577850342,
      "step": 6704
    },
    {
      "epoch": 6.3934326171875e-06,
      "step": 6704,
      "training_step_time": 7.861609697341919
    },
    {
      "epoch": 6.394386291503906e-06,
      "model_forward_time": 1.7630770206451416,
      "step": 6705
    },
    {
      "epoch": 6.394386291503906e-06,
      "step": 6705,
      "training_step_time": 8.319313764572144
    },
    {
      "epoch": 6.395339965820313e-06,
      "model_forward_time": 1.7398526668548584,
      "step": 6706
    },
    {
      "epoch": 6.395339965820313e-06,
      "step": 6706,
      "training_step_time": 8.558165311813354
    },
    {
      "epoch": 6.396293640136719e-06,
      "model_forward_time": 1.7437336444854736,
      "step": 6707
    },
    {
      "epoch": 6.396293640136719e-06,
      "step": 6707,
      "training_step_time": 7.914015531539917
    },
    {
      "epoch": 6.397247314453125e-06,
      "model_forward_time": 2.872453451156616,
      "step": 6708
    },
    {
      "epoch": 6.397247314453125e-06,
      "step": 6708,
      "training_step_time": 7.847405195236206
    },
    {
      "epoch": 6.398200988769531e-06,
      "model_forward_time": 1.7404060363769531,
      "step": 6709
    },
    {
      "epoch": 6.398200988769531e-06,
      "step": 6709,
      "training_step_time": 8.442337274551392
    },
    {
      "epoch": 6.3991546630859375e-06,
      "grad_norm": 0.2904253304004669,
      "learning_rate": 9.992017092984372e-06,
      "loss": 0.0688,
      "step": 6710
    },
    {
      "epoch": 6.3991546630859375e-06,
      "model_forward_time": 1.7308173179626465,
      "step": 6710
    },
    {
      "dynamics_loss_avg": 0.05812083669006825,
      "epoch": 6.3991546630859375e-06,
      "step": 6710
    },
    {
      "action_loss_avg": 0.004596870497334748,
      "epoch": 6.3991546630859375e-06,
      "step": 6710
    },
    {
      "epoch": 6.3991546630859375e-06,
      "step": 6710,
      "training_step_time": 7.428218126296997
    },
    {
      "epoch": 6.400108337402344e-06,
      "model_forward_time": 1.7402033805847168,
      "step": 6711
    },
    {
      "epoch": 6.400108337402344e-06,
      "step": 6711,
      "training_step_time": 8.826091527938843
    },
    {
      "epoch": 6.40106201171875e-06,
      "model_forward_time": 1.755866289138794,
      "step": 6712
    },
    {
      "epoch": 6.40106201171875e-06,
      "step": 6712,
      "training_step_time": 7.987393617630005
    },
    {
      "epoch": 6.402015686035156e-06,
      "model_forward_time": 1.7428481578826904,
      "step": 6713
    },
    {
      "epoch": 6.402015686035156e-06,
      "step": 6713,
      "training_step_time": 8.334752798080444
    },
    {
      "epoch": 6.402969360351563e-06,
      "model_forward_time": 1.7660768032073975,
      "step": 6714
    },
    {
      "epoch": 6.402969360351563e-06,
      "step": 6714,
      "training_step_time": 8.66326117515564
    },
    {
      "epoch": 6.403923034667969e-06,
      "model_forward_time": 1.7407753467559814,
      "step": 6715
    },
    {
      "epoch": 6.403923034667969e-06,
      "step": 6715,
      "training_step_time": 8.722097635269165
    },
    {
      "epoch": 6.404876708984375e-06,
      "model_forward_time": 1.7424278259277344,
      "step": 6716
    },
    {
      "epoch": 6.404876708984375e-06,
      "step": 6716,
      "training_step_time": 7.927700042724609
    },
    {
      "epoch": 6.405830383300781e-06,
      "model_forward_time": 2.8309199810028076,
      "step": 6717
    },
    {
      "epoch": 6.405830383300781e-06,
      "step": 6717,
      "training_step_time": 8.497375965118408
    },
    {
      "epoch": 6.4067840576171875e-06,
      "model_forward_time": 1.7369239330291748,
      "step": 6718
    },
    {
      "epoch": 6.4067840576171875e-06,
      "step": 6718,
      "training_step_time": 8.446134328842163
    },
    {
      "epoch": 6.407737731933594e-06,
      "model_forward_time": 1.7402567863464355,
      "step": 6719
    },
    {
      "epoch": 6.407737731933594e-06,
      "step": 6719,
      "training_step_time": 7.793500661849976
    },
    {
      "epoch": 6.40869140625e-06,
      "grad_norm": 0.2671971917152405,
      "learning_rate": 9.991923422927326e-06,
      "loss": 0.0694,
      "step": 6720
    },
    {
      "epoch": 6.40869140625e-06,
      "model_forward_time": 1.742598295211792,
      "step": 6720
    },
    {
      "dynamics_loss_avg": 0.06789430603384972,
      "epoch": 6.40869140625e-06,
      "step": 6720
    },
    {
      "action_loss_avg": 0.008258049609139562,
      "epoch": 6.40869140625e-06,
      "step": 6720
    },
    {
      "epoch": 6.40869140625e-06,
      "step": 6720,
      "training_step_time": 7.710932493209839
    },
    {
      "epoch": 6.409645080566406e-06,
      "model_forward_time": 1.7619094848632812,
      "step": 6721
    },
    {
      "epoch": 6.409645080566406e-06,
      "step": 6721,
      "training_step_time": 7.840024471282959
    },
    {
      "epoch": 6.410598754882813e-06,
      "model_forward_time": 1.728926658630371,
      "step": 6722
    },
    {
      "epoch": 6.410598754882813e-06,
      "step": 6722,
      "training_step_time": 8.346100330352783
    },
    {
      "epoch": 6.411552429199219e-06,
      "model_forward_time": 1.7511658668518066,
      "step": 6723
    },
    {
      "epoch": 6.411552429199219e-06,
      "step": 6723,
      "training_step_time": 8.330348253250122
    },
    {
      "epoch": 6.412506103515625e-06,
      "model_forward_time": 1.368393898010254,
      "step": 6724
    },
    {
      "epoch": 6.412506103515625e-06,
      "step": 6724,
      "training_step_time": 8.230062484741211
    },
    {
      "epoch": 6.413459777832031e-06,
      "model_forward_time": 1.7616190910339355,
      "step": 6725
    },
    {
      "epoch": 6.413459777832031e-06,
      "step": 6725,
      "training_step_time": 9.00067925453186
    },
    {
      "epoch": 6.4144134521484375e-06,
      "model_forward_time": 1.7469539642333984,
      "step": 6726
    },
    {
      "epoch": 6.4144134521484375e-06,
      "step": 6726,
      "training_step_time": 7.884634494781494
    },
    {
      "epoch": 6.415367126464844e-06,
      "model_forward_time": 2.929980516433716,
      "step": 6727
    },
    {
      "epoch": 6.415367126464844e-06,
      "step": 6727,
      "training_step_time": 7.913188695907593
    },
    {
      "epoch": 6.41632080078125e-06,
      "model_forward_time": 1.7387981414794922,
      "step": 6728
    },
    {
      "epoch": 6.41632080078125e-06,
      "step": 6728,
      "training_step_time": 7.848175287246704
    },
    {
      "epoch": 6.417274475097656e-06,
      "model_forward_time": 1.7423663139343262,
      "step": 6729
    },
    {
      "epoch": 6.417274475097656e-06,
      "step": 6729,
      "training_step_time": 8.274112701416016
    },
    {
      "epoch": 6.418228149414063e-06,
      "grad_norm": 0.44241225719451904,
      "learning_rate": 9.991829206961038e-06,
      "loss": 0.0718,
      "step": 6730
    },
    {
      "epoch": 6.418228149414063e-06,
      "model_forward_time": 1.7499973773956299,
      "step": 6730
    },
    {
      "dynamics_loss_avg": 0.06967852711677551,
      "epoch": 6.418228149414063e-06,
      "step": 6730
    },
    {
      "action_loss_avg": 0.006058186758309603,
      "epoch": 6.418228149414063e-06,
      "step": 6730
    },
    {
      "epoch": 6.418228149414063e-06,
      "step": 6730,
      "training_step_time": 8.681750059127808
    },
    {
      "epoch": 6.419181823730469e-06,
      "model_forward_time": 1.7398416996002197,
      "step": 6731
    },
    {
      "epoch": 6.419181823730469e-06,
      "step": 6731,
      "training_step_time": 8.932498693466187
    },
    {
      "epoch": 6.420135498046875e-06,
      "model_forward_time": 1.749452829360962,
      "step": 6732
    },
    {
      "epoch": 6.420135498046875e-06,
      "step": 6732,
      "training_step_time": 7.814966917037964
    },
    {
      "epoch": 6.421089172363281e-06,
      "model_forward_time": 1.7350037097930908,
      "step": 6733
    },
    {
      "epoch": 6.421089172363281e-06,
      "step": 6733,
      "training_step_time": 7.776073455810547
    },
    {
      "epoch": 6.4220428466796875e-06,
      "model_forward_time": 1.7401611804962158,
      "step": 6734
    },
    {
      "epoch": 6.4220428466796875e-06,
      "step": 6734,
      "training_step_time": 8.451276302337646
    },
    {
      "epoch": 6.422996520996094e-06,
      "model_forward_time": 1.739393949508667,
      "step": 6735
    },
    {
      "epoch": 6.422996520996094e-06,
      "step": 6735,
      "training_step_time": 7.921267032623291
    },
    {
      "epoch": 6.4239501953125e-06,
      "model_forward_time": 2.894397497177124,
      "step": 6736
    },
    {
      "epoch": 6.4239501953125e-06,
      "step": 6736,
      "training_step_time": 7.898814678192139
    },
    {
      "epoch": 6.424903869628906e-06,
      "model_forward_time": 1.742213487625122,
      "step": 6737
    },
    {
      "epoch": 6.424903869628906e-06,
      "step": 6737,
      "training_step_time": 7.889278888702393
    },
    {
      "epoch": 6.425857543945313e-06,
      "model_forward_time": 1.7388761043548584,
      "step": 6738
    },
    {
      "epoch": 6.425857543945313e-06,
      "step": 6738,
      "training_step_time": 7.865375995635986
    },
    {
      "epoch": 6.426811218261719e-06,
      "model_forward_time": 1.7594869136810303,
      "step": 6739
    },
    {
      "epoch": 6.426811218261719e-06,
      "step": 6739,
      "training_step_time": 8.503344297409058
    },
    {
      "epoch": 6.427764892578125e-06,
      "grad_norm": 0.3326249122619629,
      "learning_rate": 9.991734445095813e-06,
      "loss": 0.0715,
      "step": 6740
    },
    {
      "epoch": 6.427764892578125e-06,
      "model_forward_time": 1.7392966747283936,
      "step": 6740
    },
    {
      "dynamics_loss_avg": 0.05932295098900795,
      "epoch": 6.427764892578125e-06,
      "step": 6740
    },
    {
      "action_loss_avg": 0.005513324425555766,
      "epoch": 6.427764892578125e-06,
      "step": 6740
    },
    {
      "epoch": 6.427764892578125e-06,
      "step": 6740,
      "training_step_time": 8.62731671333313
    },
    {
      "epoch": 6.428718566894531e-06,
      "model_forward_time": 1.7374556064605713,
      "step": 6741
    },
    {
      "epoch": 6.428718566894531e-06,
      "step": 6741,
      "training_step_time": 7.650305509567261
    },
    {
      "epoch": 6.4296722412109375e-06,
      "model_forward_time": 1.7724640369415283,
      "step": 6742
    },
    {
      "epoch": 6.4296722412109375e-06,
      "step": 6742,
      "training_step_time": 8.717385053634644
    },
    {
      "epoch": 6.430625915527344e-06,
      "model_forward_time": 1.7423298358917236,
      "step": 6743
    },
    {
      "epoch": 6.430625915527344e-06,
      "step": 6743,
      "training_step_time": 9.215280294418335
    },
    {
      "epoch": 6.43157958984375e-06,
      "model_forward_time": 1.7319889068603516,
      "step": 6744
    },
    {
      "epoch": 6.43157958984375e-06,
      "step": 6744,
      "training_step_time": 7.806387186050415
    },
    {
      "epoch": 6.432533264160156e-06,
      "model_forward_time": 2.8719658851623535,
      "step": 6745
    },
    {
      "epoch": 6.432533264160156e-06,
      "step": 6745,
      "training_step_time": 7.7348809242248535
    },
    {
      "epoch": 6.433486938476563e-06,
      "model_forward_time": 1.0448451042175293,
      "step": 6746
    },
    {
      "epoch": 6.433486938476563e-06,
      "step": 6746,
      "training_step_time": 7.841889381408691
    },
    {
      "epoch": 6.434440612792969e-06,
      "model_forward_time": 1.7372469902038574,
      "step": 6747
    },
    {
      "epoch": 6.434440612792969e-06,
      "step": 6747,
      "training_step_time": 7.649114370346069
    },
    {
      "epoch": 6.435394287109375e-06,
      "model_forward_time": 1.7574057579040527,
      "step": 6748
    },
    {
      "epoch": 6.435394287109375e-06,
      "step": 6748,
      "training_step_time": 7.680149078369141
    },
    {
      "epoch": 6.436347961425781e-06,
      "model_forward_time": 1.7380955219268799,
      "step": 6749
    },
    {
      "epoch": 6.436347961425781e-06,
      "step": 6749,
      "training_step_time": 8.754940748214722
    },
    {
      "epoch": 6.4373016357421875e-06,
      "grad_norm": 0.4044330418109894,
      "learning_rate": 9.991639137342016e-06,
      "loss": 0.0737,
      "step": 6750
    },
    {
      "epoch": 6.4373016357421875e-06,
      "model_forward_time": 1.735412836074829,
      "step": 6750
    },
    {
      "dynamics_loss_avg": 0.06464599668979645,
      "epoch": 6.4373016357421875e-06,
      "step": 6750
    },
    {
      "action_loss_avg": 0.004359679715707898,
      "epoch": 6.4373016357421875e-06,
      "step": 6750
    },
    {
      "epoch": 6.4373016357421875e-06,
      "step": 6750,
      "training_step_time": 7.665207147598267
    },
    {
      "epoch": 6.438255310058594e-06,
      "model_forward_time": 1.7625656127929688,
      "step": 6751
    },
    {
      "epoch": 6.438255310058594e-06,
      "step": 6751,
      "training_step_time": 8.249502897262573
    },
    {
      "epoch": 6.439208984375e-06,
      "model_forward_time": 1.7351880073547363,
      "step": 6752
    },
    {
      "epoch": 6.439208984375e-06,
      "step": 6752,
      "training_step_time": 8.70609450340271
    },
    {
      "epoch": 6.440162658691406e-06,
      "model_forward_time": 1.743549108505249,
      "step": 6753
    },
    {
      "epoch": 6.440162658691406e-06,
      "step": 6753,
      "training_step_time": 8.674084901809692
    },
    {
      "epoch": 6.441116333007812e-06,
      "model_forward_time": 1.7440035343170166,
      "step": 6754
    },
    {
      "epoch": 6.441116333007812e-06,
      "step": 6754,
      "training_step_time": 7.645685434341431
    },
    {
      "epoch": 6.442070007324219e-06,
      "model_forward_time": 2.8981497287750244,
      "step": 6755
    },
    {
      "epoch": 6.442070007324219e-06,
      "step": 6755,
      "training_step_time": 7.836745738983154
    },
    {
      "epoch": 6.443023681640625e-06,
      "model_forward_time": 1.7369229793548584,
      "step": 6756
    },
    {
      "epoch": 6.443023681640625e-06,
      "step": 6756,
      "training_step_time": 7.597433805465698
    },
    {
      "epoch": 6.443977355957031e-06,
      "model_forward_time": 1.7476310729980469,
      "step": 6757
    },
    {
      "epoch": 6.443977355957031e-06,
      "step": 6757,
      "training_step_time": 7.644043684005737
    },
    {
      "epoch": 6.4449310302734375e-06,
      "model_forward_time": 1.747718334197998,
      "step": 6758
    },
    {
      "epoch": 6.4449310302734375e-06,
      "step": 6758,
      "training_step_time": 8.676863431930542
    },
    {
      "epoch": 6.445884704589844e-06,
      "model_forward_time": 1.7473256587982178,
      "step": 6759
    },
    {
      "epoch": 6.445884704589844e-06,
      "step": 6759,
      "training_step_time": 7.664808988571167
    },
    {
      "epoch": 6.44683837890625e-06,
      "grad_norm": 0.3028794527053833,
      "learning_rate": 9.991543283710064e-06,
      "loss": 0.0754,
      "step": 6760
    },
    {
      "epoch": 6.44683837890625e-06,
      "model_forward_time": 1.813537359237671,
      "step": 6760
    },
    {
      "dynamics_loss_avg": 0.06469541527330876,
      "epoch": 6.44683837890625e-06,
      "step": 6760
    },
    {
      "action_loss_avg": 0.005305812740698457,
      "epoch": 6.44683837890625e-06,
      "step": 6760
    },
    {
      "epoch": 6.44683837890625e-06,
      "step": 6760,
      "training_step_time": 7.64912486076355
    },
    {
      "epoch": 6.447792053222656e-06,
      "model_forward_time": 1.7407634258270264,
      "step": 6761
    },
    {
      "epoch": 6.447792053222656e-06,
      "step": 6761,
      "training_step_time": 7.692952871322632
    },
    {
      "epoch": 6.448745727539062e-06,
      "model_forward_time": 1.7491214275360107,
      "step": 6762
    },
    {
      "epoch": 6.448745727539062e-06,
      "step": 6762,
      "training_step_time": 8.48597502708435
    },
    {
      "epoch": 6.449699401855469e-06,
      "model_forward_time": 1.7691409587860107,
      "step": 6763
    },
    {
      "epoch": 6.449699401855469e-06,
      "step": 6763,
      "training_step_time": 7.678783655166626
    },
    {
      "epoch": 6.450653076171875e-06,
      "model_forward_time": 2.9043691158294678,
      "step": 6764
    },
    {
      "epoch": 6.450653076171875e-06,
      "step": 6764,
      "training_step_time": 8.727806329727173
    },
    {
      "epoch": 6.451606750488281e-06,
      "model_forward_time": 1.7476494312286377,
      "step": 6765
    },
    {
      "epoch": 6.451606750488281e-06,
      "step": 6765,
      "training_step_time": 7.328334093093872
    },
    {
      "epoch": 6.4525604248046875e-06,
      "model_forward_time": 1.7474377155303955,
      "step": 6766
    },
    {
      "epoch": 6.4525604248046875e-06,
      "step": 6766,
      "training_step_time": 7.7532429695129395
    },
    {
      "epoch": 6.453514099121094e-06,
      "model_forward_time": 1.745520830154419,
      "step": 6767
    },
    {
      "epoch": 6.453514099121094e-06,
      "step": 6767,
      "training_step_time": 8.530681133270264
    },
    {
      "epoch": 6.4544677734375e-06,
      "model_forward_time": 1.7479071617126465,
      "step": 6768
    },
    {
      "epoch": 6.4544677734375e-06,
      "step": 6768,
      "training_step_time": 8.413462400436401
    },
    {
      "epoch": 6.455421447753906e-06,
      "model_forward_time": 1.7608327865600586,
      "step": 6769
    },
    {
      "epoch": 6.455421447753906e-06,
      "step": 6769,
      "training_step_time": 7.712681293487549
    },
    {
      "epoch": 6.456375122070312e-06,
      "grad_norm": 0.23335173726081848,
      "learning_rate": 9.991446884210446e-06,
      "loss": 0.0715,
      "step": 6770
    },
    {
      "epoch": 6.456375122070312e-06,
      "model_forward_time": 1.7329163551330566,
      "step": 6770
    },
    {
      "dynamics_loss_avg": 0.06471772715449334,
      "epoch": 6.456375122070312e-06,
      "step": 6770
    },
    {
      "action_loss_avg": 0.007808367779944092,
      "epoch": 6.456375122070312e-06,
      "step": 6770
    },
    {
      "epoch": 6.456375122070312e-06,
      "step": 6770,
      "training_step_time": 7.728669166564941
    },
    {
      "epoch": 6.457328796386719e-06,
      "model_forward_time": 1.7718982696533203,
      "step": 6771
    },
    {
      "epoch": 6.457328796386719e-06,
      "step": 6771,
      "training_step_time": 7.7043938636779785
    },
    {
      "epoch": 6.458282470703125e-06,
      "model_forward_time": 1.3824076652526855,
      "step": 6772
    },
    {
      "epoch": 6.458282470703125e-06,
      "step": 6772,
      "training_step_time": 7.876176357269287
    },
    {
      "epoch": 6.459236145019531e-06,
      "model_forward_time": 2.881183624267578,
      "step": 6773
    },
    {
      "epoch": 6.459236145019531e-06,
      "step": 6773,
      "training_step_time": 7.785022974014282
    },
    {
      "epoch": 6.4601898193359375e-06,
      "model_forward_time": 1.768406629562378,
      "step": 6774
    },
    {
      "epoch": 6.4601898193359375e-06,
      "step": 6774,
      "training_step_time": 7.47426962852478
    },
    {
      "epoch": 6.461143493652344e-06,
      "model_forward_time": 1.7487750053405762,
      "step": 6775
    },
    {
      "epoch": 6.461143493652344e-06,
      "step": 6775,
      "training_step_time": 7.71169114112854
    },
    {
      "epoch": 6.46209716796875e-06,
      "model_forward_time": 1.7402336597442627,
      "step": 6776
    },
    {
      "epoch": 6.46209716796875e-06,
      "step": 6776,
      "training_step_time": 8.287034034729004
    },
    {
      "epoch": 6.463050842285156e-06,
      "model_forward_time": 1.739147663116455,
      "step": 6777
    },
    {
      "epoch": 6.463050842285156e-06,
      "step": 6777,
      "training_step_time": 7.604049921035767
    },
    {
      "epoch": 6.464004516601562e-06,
      "model_forward_time": 1.750685453414917,
      "step": 6778
    },
    {
      "epoch": 6.464004516601562e-06,
      "step": 6778,
      "training_step_time": 9.459466218948364
    },
    {
      "epoch": 6.464958190917969e-06,
      "model_forward_time": 1.852874517440796,
      "step": 6779
    },
    {
      "epoch": 6.464958190917969e-06,
      "step": 6779,
      "training_step_time": 8.132800579071045
    },
    {
      "epoch": 6.465911865234375e-06,
      "grad_norm": 0.29480570554733276,
      "learning_rate": 9.9913499388537e-06,
      "loss": 0.0734,
      "step": 6780
    },
    {
      "epoch": 6.465911865234375e-06,
      "model_forward_time": 1.7441480159759521,
      "step": 6780
    },
    {
      "dynamics_loss_avg": 0.07155182007700205,
      "epoch": 6.465911865234375e-06,
      "step": 6780
    },
    {
      "action_loss_avg": 0.007269265339709818,
      "epoch": 6.465911865234375e-06,
      "step": 6780
    },
    {
      "epoch": 6.465911865234375e-06,
      "step": 6780,
      "training_step_time": 9.076171875
    },
    {
      "epoch": 6.466865539550781e-06,
      "model_forward_time": 1.379504919052124,
      "step": 6781
    },
    {
      "epoch": 6.466865539550781e-06,
      "step": 6781,
      "training_step_time": 7.875398635864258
    },
    {
      "epoch": 6.4678192138671875e-06,
      "model_forward_time": 1.7460556030273438,
      "step": 6782
    },
    {
      "epoch": 6.4678192138671875e-06,
      "step": 6782,
      "training_step_time": 8.606192827224731
    },
    {
      "epoch": 6.468772888183594e-06,
      "model_forward_time": 1.7364647388458252,
      "step": 6783
    },
    {
      "epoch": 6.468772888183594e-06,
      "step": 6783,
      "training_step_time": 7.780174493789673
    },
    {
      "epoch": 6.4697265625e-06,
      "model_forward_time": 1.7525262832641602,
      "step": 6784
    },
    {
      "epoch": 6.4697265625e-06,
      "step": 6784,
      "training_step_time": 7.7566492557525635
    },
    {
      "epoch": 6.470680236816406e-06,
      "model_forward_time": 1.7422499656677246,
      "step": 6785
    },
    {
      "epoch": 6.470680236816406e-06,
      "step": 6785,
      "training_step_time": 7.782772541046143
    },
    {
      "epoch": 6.471633911132812e-06,
      "model_forward_time": 1.7481765747070312,
      "step": 6786
    },
    {
      "epoch": 6.471633911132812e-06,
      "step": 6786,
      "training_step_time": 7.694204807281494
    },
    {
      "epoch": 6.472587585449219e-06,
      "model_forward_time": 1.7494924068450928,
      "step": 6787
    },
    {
      "epoch": 6.472587585449219e-06,
      "step": 6787,
      "training_step_time": 7.703188180923462
    },
    {
      "epoch": 6.473541259765625e-06,
      "model_forward_time": 1.7879326343536377,
      "step": 6788
    },
    {
      "epoch": 6.473541259765625e-06,
      "step": 6788,
      "training_step_time": 7.721036672592163
    },
    {
      "epoch": 6.474494934082031e-06,
      "model_forward_time": 1.7481892108917236,
      "step": 6789
    },
    {
      "epoch": 6.474494934082031e-06,
      "step": 6789,
      "training_step_time": 9.232032775878906
    },
    {
      "epoch": 6.4754486083984375e-06,
      "grad_norm": 0.2809803783893585,
      "learning_rate": 9.99125244765043e-06,
      "loss": 0.0638,
      "step": 6790
    },
    {
      "epoch": 6.4754486083984375e-06,
      "model_forward_time": 1.767754316329956,
      "step": 6790
    },
    {
      "dynamics_loss_avg": 0.061622066982090475,
      "epoch": 6.4754486083984375e-06,
      "step": 6790
    },
    {
      "action_loss_avg": 0.006668119574896991,
      "epoch": 6.4754486083984375e-06,
      "step": 6790
    },
    {
      "epoch": 6.4754486083984375e-06,
      "step": 6790,
      "training_step_time": 8.67369270324707
    },
    {
      "epoch": 6.476402282714844e-06,
      "model_forward_time": 1.7497968673706055,
      "step": 6791
    },
    {
      "epoch": 6.476402282714844e-06,
      "step": 6791,
      "training_step_time": 7.716200351715088
    },
    {
      "epoch": 6.47735595703125e-06,
      "model_forward_time": 2.9048891067504883,
      "step": 6792
    },
    {
      "epoch": 6.47735595703125e-06,
      "step": 6792,
      "training_step_time": 7.875613689422607
    },
    {
      "epoch": 6.478309631347656e-06,
      "model_forward_time": 1.7477302551269531,
      "step": 6793
    },
    {
      "epoch": 6.478309631347656e-06,
      "step": 6793,
      "training_step_time": 7.729443788528442
    },
    {
      "epoch": 6.479263305664062e-06,
      "model_forward_time": 1.760314702987671,
      "step": 6794
    },
    {
      "epoch": 6.479263305664062e-06,
      "step": 6794,
      "training_step_time": 8.781375885009766
    },
    {
      "epoch": 6.480216979980469e-06,
      "model_forward_time": 1.747809648513794,
      "step": 6795
    },
    {
      "epoch": 6.480216979980469e-06,
      "step": 6795,
      "training_step_time": 7.716390132904053
    },
    {
      "epoch": 6.481170654296875e-06,
      "model_forward_time": 1.383498191833496,
      "step": 6796
    },
    {
      "epoch": 6.481170654296875e-06,
      "step": 6796,
      "training_step_time": 8.212106704711914
    },
    {
      "epoch": 6.482124328613281e-06,
      "model_forward_time": 1.7444283962249756,
      "step": 6797
    },
    {
      "epoch": 6.482124328613281e-06,
      "step": 6797,
      "training_step_time": 7.842819929122925
    },
    {
      "epoch": 6.4830780029296875e-06,
      "model_forward_time": 1.8470120429992676,
      "step": 6798
    },
    {
      "epoch": 6.4830780029296875e-06,
      "step": 6798,
      "training_step_time": 8.920777082443237
    },
    {
      "epoch": 6.484031677246094e-06,
      "model_forward_time": 1.7483689785003662,
      "step": 6799
    },
    {
      "epoch": 6.484031677246094e-06,
      "step": 6799,
      "training_step_time": 8.575966119766235
    },
    {
      "epoch": 6.4849853515625e-06,
      "grad_norm": 0.336193323135376,
      "learning_rate": 9.991154410611297e-06,
      "loss": 0.0682,
      "step": 6800
    },
    {
      "epoch": 6.4849853515625e-06,
      "model_forward_time": 1.7509355545043945,
      "step": 6800
    },
    {
      "dynamics_loss_avg": 0.0676057480275631,
      "epoch": 6.4849853515625e-06,
      "step": 6800
    },
    {
      "action_loss_avg": 0.004543066048063338,
      "epoch": 6.4849853515625e-06,
      "step": 6800
    },
    {
      "epoch": 6.4849853515625e-06,
      "step": 6800,
      "training_step_time": 8.995554208755493
    },
    {
      "epoch": 6.485939025878906e-06,
      "model_forward_time": 2.9457755088806152,
      "step": 6801
    },
    {
      "epoch": 6.485939025878906e-06,
      "step": 6801,
      "training_step_time": 8.776941776275635
    },
    {
      "epoch": 6.486892700195312e-06,
      "model_forward_time": 1.7462763786315918,
      "step": 6802
    },
    {
      "epoch": 6.486892700195312e-06,
      "step": 6802,
      "training_step_time": 7.736691951751709
    },
    {
      "epoch": 6.487846374511719e-06,
      "model_forward_time": 1.747302770614624,
      "step": 6803
    },
    {
      "epoch": 6.487846374511719e-06,
      "step": 6803,
      "training_step_time": 7.7715466022491455
    },
    {
      "epoch": 6.488800048828125e-06,
      "model_forward_time": 1.7709918022155762,
      "step": 6804
    },
    {
      "epoch": 6.488800048828125e-06,
      "step": 6804,
      "training_step_time": 8.568969011306763
    },
    {
      "epoch": 6.489753723144531e-06,
      "model_forward_time": 1.7473936080932617,
      "step": 6805
    },
    {
      "epoch": 6.489753723144531e-06,
      "step": 6805,
      "training_step_time": 9.052872896194458
    },
    {
      "epoch": 6.4907073974609375e-06,
      "model_forward_time": 1.7664906978607178,
      "step": 6806
    },
    {
      "epoch": 6.4907073974609375e-06,
      "step": 6806,
      "training_step_time": 8.43137788772583
    },
    {
      "epoch": 6.491661071777344e-06,
      "model_forward_time": 1.8447377681732178,
      "step": 6807
    },
    {
      "epoch": 6.491661071777344e-06,
      "step": 6807,
      "training_step_time": 7.900167942047119
    },
    {
      "epoch": 6.49261474609375e-06,
      "model_forward_time": 1.7501451969146729,
      "step": 6808
    },
    {
      "epoch": 6.49261474609375e-06,
      "step": 6808,
      "training_step_time": 7.70037579536438
    },
    {
      "epoch": 6.493568420410156e-06,
      "model_forward_time": 1.7548167705535889,
      "step": 6809
    },
    {
      "epoch": 6.493568420410156e-06,
      "step": 6809,
      "training_step_time": 7.866983652114868
    },
    {
      "epoch": 6.494522094726562e-06,
      "grad_norm": 0.29818883538246155,
      "learning_rate": 9.99105582774702e-06,
      "loss": 0.0691,
      "step": 6810
    },
    {
      "epoch": 6.494522094726562e-06,
      "model_forward_time": 2.8898448944091797,
      "step": 6810
    },
    {
      "dynamics_loss_avg": 0.055607385374605654,
      "epoch": 6.494522094726562e-06,
      "step": 6810
    },
    {
      "action_loss_avg": 0.004716183908749371,
      "epoch": 6.494522094726562e-06,
      "step": 6810
    },
    {
      "epoch": 6.494522094726562e-06,
      "step": 6810,
      "training_step_time": 7.7898900508880615
    },
    {
      "epoch": 6.495475769042969e-06,
      "model_forward_time": 1.7486164569854736,
      "step": 6811
    },
    {
      "epoch": 6.495475769042969e-06,
      "step": 6811,
      "training_step_time": 7.830970525741577
    },
    {
      "epoch": 6.496429443359375e-06,
      "model_forward_time": 1.7421908378601074,
      "step": 6812
    },
    {
      "epoch": 6.496429443359375e-06,
      "step": 6812,
      "training_step_time": 7.784176349639893
    },
    {
      "epoch": 6.497383117675781e-06,
      "model_forward_time": 1.7469511032104492,
      "step": 6813
    },
    {
      "epoch": 6.497383117675781e-06,
      "step": 6813,
      "training_step_time": 7.686792850494385
    },
    {
      "epoch": 6.4983367919921876e-06,
      "model_forward_time": 1.7476084232330322,
      "step": 6814
    },
    {
      "epoch": 6.4983367919921876e-06,
      "step": 6814,
      "training_step_time": 8.268029689788818
    },
    {
      "epoch": 6.499290466308594e-06,
      "model_forward_time": 1.7549176216125488,
      "step": 6815
    },
    {
      "epoch": 6.499290466308594e-06,
      "step": 6815,
      "training_step_time": 7.686798810958862
    },
    {
      "epoch": 6.500244140625e-06,
      "model_forward_time": 1.3994078636169434,
      "step": 6816
    },
    {
      "epoch": 6.500244140625e-06,
      "step": 6816,
      "training_step_time": 7.858583927154541
    },
    {
      "epoch": 6.501197814941406e-06,
      "model_forward_time": 1.758584976196289,
      "step": 6817
    },
    {
      "epoch": 6.501197814941406e-06,
      "step": 6817,
      "training_step_time": 7.758285045623779
    },
    {
      "epoch": 6.502151489257812e-06,
      "model_forward_time": 1.7513427734375,
      "step": 6818
    },
    {
      "epoch": 6.502151489257812e-06,
      "step": 6818,
      "training_step_time": 7.853883504867554
    },
    {
      "epoch": 6.503105163574219e-06,
      "model_forward_time": 2.8725497722625732,
      "step": 6819
    },
    {
      "epoch": 6.503105163574219e-06,
      "step": 6819,
      "training_step_time": 9.422694444656372
    },
    {
      "epoch": 6.504058837890625e-06,
      "grad_norm": 0.336690753698349,
      "learning_rate": 9.990956699068383e-06,
      "loss": 0.0705,
      "step": 6820
    },
    {
      "epoch": 6.504058837890625e-06,
      "model_forward_time": 1.7632579803466797,
      "step": 6820
    },
    {
      "dynamics_loss_avg": 0.06887802332639695,
      "epoch": 6.504058837890625e-06,
      "step": 6820
    },
    {
      "action_loss_avg": 0.00668578885961324,
      "epoch": 6.504058837890625e-06,
      "step": 6820
    },
    {
      "epoch": 6.504058837890625e-06,
      "step": 6820,
      "training_step_time": 7.826097726821899
    },
    {
      "epoch": 6.505012512207031e-06,
      "model_forward_time": 1.752350091934204,
      "step": 6821
    },
    {
      "epoch": 6.505012512207031e-06,
      "step": 6821,
      "training_step_time": 8.357533931732178
    },
    {
      "epoch": 6.5059661865234376e-06,
      "model_forward_time": 1.756098747253418,
      "step": 6822
    },
    {
      "epoch": 6.5059661865234376e-06,
      "step": 6822,
      "training_step_time": 7.737144470214844
    },
    {
      "epoch": 6.506919860839844e-06,
      "model_forward_time": 1.741983413696289,
      "step": 6823
    },
    {
      "epoch": 6.506919860839844e-06,
      "step": 6823,
      "training_step_time": 8.599239587783813
    },
    {
      "epoch": 6.50787353515625e-06,
      "model_forward_time": 1.7505512237548828,
      "step": 6824
    },
    {
      "epoch": 6.50787353515625e-06,
      "step": 6824,
      "training_step_time": 7.703645944595337
    },
    {
      "epoch": 6.508827209472656e-06,
      "model_forward_time": 1.7498698234558105,
      "step": 6825
    },
    {
      "epoch": 6.508827209472656e-06,
      "step": 6825,
      "training_step_time": 10.456696271896362
    },
    {
      "epoch": 6.509780883789062e-06,
      "model_forward_time": 1.75388503074646,
      "step": 6826
    },
    {
      "epoch": 6.509780883789062e-06,
      "step": 6826,
      "training_step_time": 8.204134702682495
    },
    {
      "epoch": 6.510734558105469e-06,
      "model_forward_time": 1.7454845905303955,
      "step": 6827
    },
    {
      "epoch": 6.510734558105469e-06,
      "step": 6827,
      "training_step_time": 7.841822385787964
    },
    {
      "epoch": 6.511688232421875e-06,
      "model_forward_time": 2.846898078918457,
      "step": 6828
    },
    {
      "epoch": 6.511688232421875e-06,
      "step": 6828,
      "training_step_time": 8.717302799224854
    },
    {
      "epoch": 6.512641906738281e-06,
      "model_forward_time": 1.752699375152588,
      "step": 6829
    },
    {
      "epoch": 6.512641906738281e-06,
      "step": 6829,
      "training_step_time": 7.775423526763916
    },
    {
      "epoch": 6.5135955810546876e-06,
      "grad_norm": 0.29424962401390076,
      "learning_rate": 9.990857024586224e-06,
      "loss": 0.0714,
      "step": 6830
    },
    {
      "epoch": 6.5135955810546876e-06,
      "model_forward_time": 1.7297353744506836,
      "step": 6830
    },
    {
      "dynamics_loss_avg": 0.05496537219733,
      "epoch": 6.5135955810546876e-06,
      "step": 6830
    },
    {
      "action_loss_avg": 0.0055783214047551155,
      "epoch": 6.5135955810546876e-06,
      "step": 6830
    },
    {
      "epoch": 6.5135955810546876e-06,
      "step": 6830,
      "training_step_time": 7.662785530090332
    },
    {
      "epoch": 6.514549255371094e-06,
      "model_forward_time": 1.744412899017334,
      "step": 6831
    },
    {
      "epoch": 6.514549255371094e-06,
      "step": 6831,
      "training_step_time": 8.309762716293335
    },
    {
      "epoch": 6.5155029296875e-06,
      "model_forward_time": 1.7418198585510254,
      "step": 6832
    },
    {
      "epoch": 6.5155029296875e-06,
      "step": 6832,
      "training_step_time": 8.535689115524292
    },
    {
      "epoch": 6.5164566040039064e-06,
      "model_forward_time": 1.7429919242858887,
      "step": 6833
    },
    {
      "epoch": 6.5164566040039064e-06,
      "step": 6833,
      "training_step_time": 9.034116983413696
    },
    {
      "epoch": 6.517410278320312e-06,
      "model_forward_time": 1.7391736507415771,
      "step": 6834
    },
    {
      "epoch": 6.517410278320312e-06,
      "step": 6834,
      "training_step_time": 10.949560165405273
    },
    {
      "epoch": 6.518363952636719e-06,
      "model_forward_time": 1.8644769191741943,
      "step": 6835
    },
    {
      "epoch": 6.518363952636719e-06,
      "step": 6835,
      "training_step_time": 8.812034845352173
    },
    {
      "epoch": 6.519317626953125e-06,
      "model_forward_time": 1.7501773834228516,
      "step": 6836
    },
    {
      "epoch": 6.519317626953125e-06,
      "step": 6836,
      "training_step_time": 7.732758045196533
    },
    {
      "epoch": 6.520271301269531e-06,
      "model_forward_time": 1.7405505180358887,
      "step": 6837
    },
    {
      "epoch": 6.520271301269531e-06,
      "step": 6837,
      "training_step_time": 8.402294635772705
    },
    {
      "epoch": 6.5212249755859376e-06,
      "model_forward_time": 2.921278476715088,
      "step": 6838
    },
    {
      "epoch": 6.5212249755859376e-06,
      "step": 6838,
      "training_step_time": 7.80061411857605
    },
    {
      "epoch": 6.522178649902344e-06,
      "model_forward_time": 1.7420589923858643,
      "step": 6839
    },
    {
      "epoch": 6.522178649902344e-06,
      "step": 6839,
      "training_step_time": 7.60979437828064
    },
    {
      "epoch": 6.52313232421875e-06,
      "grad_norm": 0.24001969397068024,
      "learning_rate": 9.990756804311447e-06,
      "loss": 0.0725,
      "step": 6840
    },
    {
      "epoch": 6.52313232421875e-06,
      "model_forward_time": 1.7473092079162598,
      "step": 6840
    },
    {
      "dynamics_loss_avg": 0.06585121415555477,
      "epoch": 6.52313232421875e-06,
      "step": 6840
    },
    {
      "action_loss_avg": 0.004698763519991189,
      "epoch": 6.52313232421875e-06,
      "step": 6840
    },
    {
      "epoch": 6.52313232421875e-06,
      "step": 6840,
      "training_step_time": 7.7051920890808105
    },
    {
      "epoch": 6.5240859985351564e-06,
      "model_forward_time": 1.7608792781829834,
      "step": 6841
    },
    {
      "epoch": 6.5240859985351564e-06,
      "step": 6841,
      "training_step_time": 7.74176025390625
    },
    {
      "epoch": 6.525039672851562e-06,
      "model_forward_time": 1.744312047958374,
      "step": 6842
    },
    {
      "epoch": 6.525039672851562e-06,
      "step": 6842,
      "training_step_time": 8.341456890106201
    },
    {
      "epoch": 6.525993347167969e-06,
      "model_forward_time": 1.7475244998931885,
      "step": 6843
    },
    {
      "epoch": 6.525993347167969e-06,
      "step": 6843,
      "training_step_time": 7.701559543609619
    },
    {
      "epoch": 6.526947021484375e-06,
      "model_forward_time": 1.8121473789215088,
      "step": 6844
    },
    {
      "epoch": 6.526947021484375e-06,
      "step": 6844,
      "training_step_time": 7.798580646514893
    },
    {
      "epoch": 6.527900695800781e-06,
      "model_forward_time": 1.7363076210021973,
      "step": 6845
    },
    {
      "epoch": 6.527900695800781e-06,
      "step": 6845,
      "training_step_time": 7.6353394985198975
    },
    {
      "epoch": 6.528854370117188e-06,
      "model_forward_time": 1.7560651302337646,
      "step": 6846
    },
    {
      "epoch": 6.528854370117188e-06,
      "step": 6846,
      "training_step_time": 8.591985702514648
    },
    {
      "epoch": 6.529808044433594e-06,
      "model_forward_time": 2.4941065311431885,
      "step": 6847
    },
    {
      "epoch": 6.529808044433594e-06,
      "step": 6847,
      "training_step_time": 7.733394622802734
    },
    {
      "epoch": 6.53076171875e-06,
      "model_forward_time": 1.7447013854980469,
      "step": 6848
    },
    {
      "epoch": 6.53076171875e-06,
      "step": 6848,
      "training_step_time": 8.049318313598633
    },
    {
      "epoch": 6.5317153930664064e-06,
      "model_forward_time": 1.7440705299377441,
      "step": 6849
    },
    {
      "epoch": 6.5317153930664064e-06,
      "step": 6849,
      "training_step_time": 8.193354606628418
    },
    {
      "epoch": 6.532669067382812e-06,
      "grad_norm": 0.25863969326019287,
      "learning_rate": 9.990656038255007e-06,
      "loss": 0.0703,
      "step": 6850
    },
    {
      "epoch": 6.532669067382812e-06,
      "model_forward_time": 1.7251343727111816,
      "step": 6850
    },
    {
      "dynamics_loss_avg": 0.07691346630454063,
      "epoch": 6.532669067382812e-06,
      "step": 6850
    },
    {
      "action_loss_avg": 0.008093419717624784,
      "epoch": 6.532669067382812e-06,
      "step": 6850
    },
    {
      "epoch": 6.532669067382812e-06,
      "step": 6850,
      "training_step_time": 7.705603361129761
    },
    {
      "epoch": 6.533622741699219e-06,
      "model_forward_time": 1.7603154182434082,
      "step": 6851
    },
    {
      "epoch": 6.533622741699219e-06,
      "step": 6851,
      "training_step_time": 7.605666875839233
    },
    {
      "epoch": 6.534576416015625e-06,
      "model_forward_time": 1.7422220706939697,
      "step": 6852
    },
    {
      "epoch": 6.534576416015625e-06,
      "step": 6852,
      "training_step_time": 8.262506484985352
    },
    {
      "epoch": 6.535530090332031e-06,
      "model_forward_time": 1.7350072860717773,
      "step": 6853
    },
    {
      "epoch": 6.535530090332031e-06,
      "step": 6853,
      "training_step_time": 7.862618923187256
    },
    {
      "epoch": 6.536483764648438e-06,
      "model_forward_time": 1.3701982498168945,
      "step": 6854
    },
    {
      "epoch": 6.536483764648438e-06,
      "step": 6854,
      "training_step_time": 7.750074148178101
    },
    {
      "epoch": 6.537437438964844e-06,
      "model_forward_time": 1.732692003250122,
      "step": 6855
    },
    {
      "epoch": 6.537437438964844e-06,
      "step": 6855,
      "training_step_time": 7.847663164138794
    },
    {
      "epoch": 6.53839111328125e-06,
      "model_forward_time": 2.868562698364258,
      "step": 6856
    },
    {
      "epoch": 6.53839111328125e-06,
      "step": 6856,
      "training_step_time": 8.24930477142334
    },
    {
      "epoch": 6.5393447875976564e-06,
      "model_forward_time": 1.7388451099395752,
      "step": 6857
    },
    {
      "epoch": 6.5393447875976564e-06,
      "step": 6857,
      "training_step_time": 7.745029449462891
    },
    {
      "epoch": 6.540298461914062e-06,
      "model_forward_time": 1.7327625751495361,
      "step": 6858
    },
    {
      "epoch": 6.540298461914062e-06,
      "step": 6858,
      "training_step_time": 7.707668781280518
    },
    {
      "epoch": 6.541252136230469e-06,
      "model_forward_time": 1.7663989067077637,
      "step": 6859
    },
    {
      "epoch": 6.541252136230469e-06,
      "step": 6859,
      "training_step_time": 7.705389022827148
    },
    {
      "epoch": 6.542205810546875e-06,
      "grad_norm": 0.298892080783844,
      "learning_rate": 9.990554726427927e-06,
      "loss": 0.0676,
      "step": 6860
    },
    {
      "epoch": 6.542205810546875e-06,
      "model_forward_time": 1.7407996654510498,
      "step": 6860
    },
    {
      "dynamics_loss_avg": 0.05798728987574577,
      "epoch": 6.542205810546875e-06,
      "step": 6860
    },
    {
      "action_loss_avg": 0.005414825666230172,
      "epoch": 6.542205810546875e-06,
      "step": 6860
    },
    {
      "epoch": 6.542205810546875e-06,
      "step": 6860,
      "training_step_time": 8.32702374458313
    },
    {
      "epoch": 6.543159484863281e-06,
      "model_forward_time": 1.7412939071655273,
      "step": 6861
    },
    {
      "epoch": 6.543159484863281e-06,
      "step": 6861,
      "training_step_time": 8.604013442993164
    },
    {
      "epoch": 6.544113159179688e-06,
      "model_forward_time": 1.388115644454956,
      "step": 6862
    },
    {
      "epoch": 6.544113159179688e-06,
      "step": 6862,
      "training_step_time": 8.76801347732544
    },
    {
      "epoch": 6.545066833496094e-06,
      "model_forward_time": 1.8447201251983643,
      "step": 6863
    },
    {
      "epoch": 6.545066833496094e-06,
      "step": 6863,
      "training_step_time": 8.220762729644775
    },
    {
      "epoch": 6.5460205078125e-06,
      "model_forward_time": 1.737476110458374,
      "step": 6864
    },
    {
      "epoch": 6.5460205078125e-06,
      "step": 6864,
      "training_step_time": 8.743672847747803
    },
    {
      "epoch": 6.5469741821289065e-06,
      "model_forward_time": 2.8108327388763428,
      "step": 6865
    },
    {
      "epoch": 6.5469741821289065e-06,
      "step": 6865,
      "training_step_time": 7.72093939781189
    },
    {
      "epoch": 6.547927856445312e-06,
      "model_forward_time": 1.7368619441986084,
      "step": 6866
    },
    {
      "epoch": 6.547927856445312e-06,
      "step": 6866,
      "training_step_time": 9.620541095733643
    },
    {
      "epoch": 6.548881530761719e-06,
      "model_forward_time": 1.7337987422943115,
      "step": 6867
    },
    {
      "epoch": 6.548881530761719e-06,
      "step": 6867,
      "training_step_time": 8.432488203048706
    },
    {
      "epoch": 6.549835205078125e-06,
      "model_forward_time": 1.739267349243164,
      "step": 6868
    },
    {
      "epoch": 6.549835205078125e-06,
      "step": 6868,
      "training_step_time": 8.289743661880493
    },
    {
      "epoch": 6.550788879394531e-06,
      "model_forward_time": 1.7328486442565918,
      "step": 6869
    },
    {
      "epoch": 6.550788879394531e-06,
      "step": 6869,
      "training_step_time": 7.63460898399353
    },
    {
      "epoch": 6.551742553710938e-06,
      "grad_norm": 0.3203345239162445,
      "learning_rate": 9.990452868841285e-06,
      "loss": 0.0656,
      "step": 6870
    },
    {
      "epoch": 6.551742553710938e-06,
      "model_forward_time": 1.7262828350067139,
      "step": 6870
    },
    {
      "dynamics_loss_avg": 0.07897525615990161,
      "epoch": 6.551742553710938e-06,
      "step": 6870
    },
    {
      "action_loss_avg": 0.009097945317626,
      "epoch": 6.551742553710938e-06,
      "step": 6870
    },
    {
      "epoch": 6.551742553710938e-06,
      "step": 6870,
      "training_step_time": 9.729930400848389
    },
    {
      "epoch": 6.552696228027344e-06,
      "model_forward_time": 1.7338933944702148,
      "step": 6871
    },
    {
      "epoch": 6.552696228027344e-06,
      "step": 6871,
      "training_step_time": 7.9596734046936035
    },
    {
      "epoch": 6.55364990234375e-06,
      "model_forward_time": 1.8098974227905273,
      "step": 6872
    },
    {
      "epoch": 6.55364990234375e-06,
      "step": 6872,
      "training_step_time": 9.166000604629517
    },
    {
      "epoch": 6.5546035766601565e-06,
      "model_forward_time": 1.7324652671813965,
      "step": 6873
    },
    {
      "epoch": 6.5546035766601565e-06,
      "step": 6873,
      "training_step_time": 7.85164475440979
    },
    {
      "epoch": 6.555557250976562e-06,
      "model_forward_time": 1.7388947010040283,
      "step": 6874
    },
    {
      "epoch": 6.555557250976562e-06,
      "step": 6874,
      "training_step_time": 7.711070775985718
    },
    {
      "epoch": 6.556510925292969e-06,
      "model_forward_time": 2.9300589561462402,
      "step": 6875
    },
    {
      "epoch": 6.556510925292969e-06,
      "step": 6875,
      "training_step_time": 7.780027627944946
    },
    {
      "epoch": 6.557464599609375e-06,
      "model_forward_time": 1.739577293395996,
      "step": 6876
    },
    {
      "epoch": 6.557464599609375e-06,
      "step": 6876,
      "training_step_time": 7.583854675292969
    },
    {
      "epoch": 6.558418273925781e-06,
      "model_forward_time": 1.737394094467163,
      "step": 6877
    },
    {
      "epoch": 6.558418273925781e-06,
      "step": 6877,
      "training_step_time": 7.655614852905273
    },
    {
      "epoch": 6.559371948242188e-06,
      "model_forward_time": 1.7350211143493652,
      "step": 6878
    },
    {
      "epoch": 6.559371948242188e-06,
      "step": 6878,
      "training_step_time": 7.619874954223633
    },
    {
      "epoch": 6.560325622558593e-06,
      "model_forward_time": 1.7406659126281738,
      "step": 6879
    },
    {
      "epoch": 6.560325622558593e-06,
      "step": 6879,
      "training_step_time": 7.66630220413208
    },
    {
      "epoch": 6.561279296875e-06,
      "grad_norm": 0.17964056134223938,
      "learning_rate": 9.99035046550622e-06,
      "loss": 0.0689,
      "step": 6880
    },
    {
      "epoch": 6.561279296875e-06,
      "model_forward_time": 1.724151611328125,
      "step": 6880
    },
    {
      "dynamics_loss_avg": 0.05889364145696163,
      "epoch": 6.561279296875e-06,
      "step": 6880
    },
    {
      "action_loss_avg": 0.003473788604605943,
      "epoch": 6.561279296875e-06,
      "step": 6880
    },
    {
      "epoch": 6.561279296875e-06,
      "step": 6880,
      "training_step_time": 7.844526767730713
    },
    {
      "epoch": 6.5622329711914065e-06,
      "model_forward_time": 1.7876994609832764,
      "step": 6881
    },
    {
      "epoch": 6.5622329711914065e-06,
      "step": 6881,
      "training_step_time": 7.637818098068237
    },
    {
      "epoch": 6.563186645507812e-06,
      "model_forward_time": 1.7298448085784912,
      "step": 6882
    },
    {
      "epoch": 6.563186645507812e-06,
      "step": 6882,
      "training_step_time": 7.681481122970581
    },
    {
      "epoch": 6.564140319824219e-06,
      "model_forward_time": 1.733572006225586,
      "step": 6883
    },
    {
      "epoch": 6.564140319824219e-06,
      "step": 6883,
      "training_step_time": 7.730289697647095
    },
    {
      "epoch": 6.565093994140625e-06,
      "model_forward_time": 2.903123378753662,
      "step": 6884
    },
    {
      "epoch": 6.565093994140625e-06,
      "step": 6884,
      "training_step_time": 7.726799249649048
    },
    {
      "epoch": 6.566047668457031e-06,
      "model_forward_time": 1.73622727394104,
      "step": 6885
    },
    {
      "epoch": 6.566047668457031e-06,
      "step": 6885,
      "training_step_time": 8.22341012954712
    },
    {
      "epoch": 6.567001342773438e-06,
      "model_forward_time": 1.7328071594238281,
      "step": 6886
    },
    {
      "epoch": 6.567001342773438e-06,
      "step": 6886,
      "training_step_time": 7.706793546676636
    },
    {
      "epoch": 6.567955017089843e-06,
      "model_forward_time": 1.734236717224121,
      "step": 6887
    },
    {
      "epoch": 6.567955017089843e-06,
      "step": 6887,
      "training_step_time": 7.62781834602356
    },
    {
      "epoch": 6.56890869140625e-06,
      "model_forward_time": 1.730128288269043,
      "step": 6888
    },
    {
      "epoch": 6.56890869140625e-06,
      "step": 6888,
      "training_step_time": 7.6280128955841064
    },
    {
      "epoch": 6.5698623657226565e-06,
      "model_forward_time": 1.7256059646606445,
      "step": 6889
    },
    {
      "epoch": 6.5698623657226565e-06,
      "step": 6889,
      "training_step_time": 10.246238708496094
    },
    {
      "epoch": 6.570816040039062e-06,
      "grad_norm": 0.2529807686805725,
      "learning_rate": 9.990247516433931e-06,
      "loss": 0.0653,
      "step": 6890
    },
    {
      "epoch": 6.570816040039062e-06,
      "model_forward_time": 1.3777225017547607,
      "step": 6890
    },
    {
      "dynamics_loss_avg": 0.06056267218664289,
      "epoch": 6.570816040039062e-06,
      "step": 6890
    },
    {
      "action_loss_avg": 0.004915982589591294,
      "epoch": 6.570816040039062e-06,
      "step": 6890
    },
    {
      "epoch": 6.570816040039062e-06,
      "step": 6890,
      "training_step_time": 7.609828233718872
    },
    {
      "epoch": 6.571769714355469e-06,
      "model_forward_time": 1.7527165412902832,
      "step": 6891
    },
    {
      "epoch": 6.571769714355469e-06,
      "step": 6891,
      "training_step_time": 7.689839601516724
    },
    {
      "epoch": 6.572723388671875e-06,
      "model_forward_time": 1.7392237186431885,
      "step": 6892
    },
    {
      "epoch": 6.572723388671875e-06,
      "step": 6892,
      "training_step_time": 7.837795257568359
    },
    {
      "epoch": 6.573677062988281e-06,
      "model_forward_time": 2.8832287788391113,
      "step": 6893
    },
    {
      "epoch": 6.573677062988281e-06,
      "step": 6893,
      "training_step_time": 7.771035671234131
    },
    {
      "epoch": 6.574630737304688e-06,
      "model_forward_time": 1.755084753036499,
      "step": 6894
    },
    {
      "epoch": 6.574630737304688e-06,
      "step": 6894,
      "training_step_time": 7.592583179473877
    },
    {
      "epoch": 6.575584411621093e-06,
      "model_forward_time": 1.7362556457519531,
      "step": 6895
    },
    {
      "epoch": 6.575584411621093e-06,
      "step": 6895,
      "training_step_time": 8.292529106140137
    },
    {
      "epoch": 6.5765380859375e-06,
      "model_forward_time": 1.7343299388885498,
      "step": 6896
    },
    {
      "epoch": 6.5765380859375e-06,
      "step": 6896,
      "training_step_time": 8.575779438018799
    },
    {
      "epoch": 6.5774917602539065e-06,
      "model_forward_time": 1.7373576164245605,
      "step": 6897
    },
    {
      "epoch": 6.5774917602539065e-06,
      "step": 6897,
      "training_step_time": 7.5559775829315186
    },
    {
      "epoch": 6.578445434570312e-06,
      "model_forward_time": 1.7446765899658203,
      "step": 6898
    },
    {
      "epoch": 6.578445434570312e-06,
      "step": 6898,
      "training_step_time": 9.72764778137207
    },
    {
      "epoch": 6.579399108886719e-06,
      "model_forward_time": 1.7343926429748535,
      "step": 6899
    },
    {
      "epoch": 6.579399108886719e-06,
      "step": 6899,
      "training_step_time": 7.789164304733276
    },
    {
      "epoch": 6.580352783203125e-06,
      "grad_norm": 0.19403420388698578,
      "learning_rate": 9.990144021635678e-06,
      "loss": 0.0725,
      "step": 6900
    },
    {
      "epoch": 6.580352783203125e-06,
      "model_forward_time": 1.8403065204620361,
      "step": 6900
    },
    {
      "dynamics_loss_avg": 0.06963560096919537,
      "epoch": 6.580352783203125e-06,
      "step": 6900
    },
    {
      "action_loss_avg": 0.0060266811633482575,
      "epoch": 6.580352783203125e-06,
      "step": 6900
    },
    {
      "epoch": 6.580352783203125e-06,
      "step": 6900,
      "training_step_time": 8.578549146652222
    },
    {
      "epoch": 6.581306457519531e-06,
      "model_forward_time": 1.7361924648284912,
      "step": 6901
    },
    {
      "epoch": 6.581306457519531e-06,
      "step": 6901,
      "training_step_time": 7.813207387924194
    },
    {
      "epoch": 6.582260131835938e-06,
      "model_forward_time": 2.9201061725616455,
      "step": 6902
    },
    {
      "epoch": 6.582260131835938e-06,
      "step": 6902,
      "training_step_time": 9.318357706069946
    },
    {
      "epoch": 6.583213806152343e-06,
      "model_forward_time": 1.7417123317718506,
      "step": 6903
    },
    {
      "epoch": 6.583213806152343e-06,
      "step": 6903,
      "training_step_time": 7.6326963901519775
    },
    {
      "epoch": 6.58416748046875e-06,
      "model_forward_time": 1.7344064712524414,
      "step": 6904
    },
    {
      "epoch": 6.58416748046875e-06,
      "step": 6904,
      "training_step_time": 8.27922534942627
    },
    {
      "epoch": 6.5851211547851565e-06,
      "model_forward_time": 1.7380352020263672,
      "step": 6905
    },
    {
      "epoch": 6.5851211547851565e-06,
      "step": 6905,
      "training_step_time": 7.707483530044556
    },
    {
      "epoch": 6.586074829101562e-06,
      "model_forward_time": 1.7402231693267822,
      "step": 6906
    },
    {
      "epoch": 6.586074829101562e-06,
      "step": 6906,
      "training_step_time": 7.573148250579834
    },
    {
      "epoch": 6.587028503417969e-06,
      "model_forward_time": 1.7451233863830566,
      "step": 6907
    },
    {
      "epoch": 6.587028503417969e-06,
      "step": 6907,
      "training_step_time": 10.2728910446167
    },
    {
      "epoch": 6.587982177734375e-06,
      "model_forward_time": 1.7367501258850098,
      "step": 6908
    },
    {
      "epoch": 6.587982177734375e-06,
      "step": 6908,
      "training_step_time": 8.284387826919556
    },
    {
      "epoch": 6.588935852050781e-06,
      "model_forward_time": 1.8215556144714355,
      "step": 6909
    },
    {
      "epoch": 6.588935852050781e-06,
      "step": 6909,
      "training_step_time": 8.858102321624756
    },
    {
      "epoch": 6.589889526367188e-06,
      "grad_norm": 0.26823827624320984,
      "learning_rate": 9.990039981122775e-06,
      "loss": 0.0798,
      "step": 6910
    },
    {
      "epoch": 6.589889526367188e-06,
      "model_forward_time": 1.754634141921997,
      "step": 6910
    },
    {
      "dynamics_loss_avg": 0.047532458882778884,
      "epoch": 6.589889526367188e-06,
      "step": 6910
    },
    {
      "action_loss_avg": 0.004783552244771272,
      "epoch": 6.589889526367188e-06,
      "step": 6910
    },
    {
      "epoch": 6.589889526367188e-06,
      "step": 6910,
      "training_step_time": 7.825668096542358
    },
    {
      "epoch": 6.590843200683593e-06,
      "model_forward_time": 2.863147258758545,
      "step": 6911
    },
    {
      "epoch": 6.590843200683593e-06,
      "step": 6911,
      "training_step_time": 8.246529340744019
    },
    {
      "epoch": 6.591796875e-06,
      "model_forward_time": 1.7518749237060547,
      "step": 6912
    },
    {
      "epoch": 6.591796875e-06,
      "step": 6912,
      "training_step_time": 7.6909685134887695
    },
    {
      "epoch": 6.5927505493164065e-06,
      "model_forward_time": 1.7518434524536133,
      "step": 6913
    },
    {
      "epoch": 6.5927505493164065e-06,
      "step": 6913,
      "training_step_time": 7.580885410308838
    },
    {
      "epoch": 6.593704223632812e-06,
      "model_forward_time": 1.7595319747924805,
      "step": 6914
    },
    {
      "epoch": 6.593704223632812e-06,
      "step": 6914,
      "training_step_time": 7.597898721694946
    },
    {
      "epoch": 6.594657897949219e-06,
      "model_forward_time": 1.7854957580566406,
      "step": 6915
    },
    {
      "epoch": 6.594657897949219e-06,
      "step": 6915,
      "training_step_time": 8.529948711395264
    },
    {
      "epoch": 6.595611572265625e-06,
      "model_forward_time": 1.7495405673980713,
      "step": 6916
    },
    {
      "epoch": 6.595611572265625e-06,
      "step": 6916,
      "training_step_time": 7.551678657531738
    },
    {
      "epoch": 6.596565246582031e-06,
      "model_forward_time": 1.7606334686279297,
      "step": 6917
    },
    {
      "epoch": 6.596565246582031e-06,
      "step": 6917,
      "training_step_time": 7.668328523635864
    },
    {
      "epoch": 6.597518920898438e-06,
      "model_forward_time": 1.7635807991027832,
      "step": 6918
    },
    {
      "epoch": 6.597518920898438e-06,
      "step": 6918,
      "training_step_time": 7.5956950187683105
    },
    {
      "epoch": 6.598472595214843e-06,
      "model_forward_time": 1.7488336563110352,
      "step": 6919
    },
    {
      "epoch": 6.598472595214843e-06,
      "step": 6919,
      "training_step_time": 8.50910496711731
    },
    {
      "epoch": 6.59942626953125e-06,
      "grad_norm": 0.2112305462360382,
      "learning_rate": 9.989935394906602e-06,
      "loss": 0.079,
      "step": 6920
    },
    {
      "epoch": 6.59942626953125e-06,
      "model_forward_time": 1.7555606365203857,
      "step": 6920
    },
    {
      "dynamics_loss_avg": 0.06333582233637572,
      "epoch": 6.59942626953125e-06,
      "step": 6920
    },
    {
      "action_loss_avg": 0.005692028626799584,
      "epoch": 6.59942626953125e-06,
      "step": 6920
    },
    {
      "epoch": 6.59942626953125e-06,
      "step": 6920,
      "training_step_time": 9.231024980545044
    },
    {
      "epoch": 6.6003799438476565e-06,
      "model_forward_time": 1.7284276485443115,
      "step": 6921
    },
    {
      "epoch": 6.6003799438476565e-06,
      "step": 6921,
      "training_step_time": 7.700456857681274
    },
    {
      "epoch": 6.601333618164062e-06,
      "model_forward_time": 1.748061180114746,
      "step": 6922
    },
    {
      "epoch": 6.601333618164062e-06,
      "step": 6922,
      "training_step_time": 7.394546747207642
    },
    {
      "epoch": 6.602287292480469e-06,
      "model_forward_time": 1.7408792972564697,
      "step": 6923
    },
    {
      "epoch": 6.602287292480469e-06,
      "step": 6923,
      "training_step_time": 7.696723699569702
    },
    {
      "epoch": 6.603240966796875e-06,
      "model_forward_time": 1.7442002296447754,
      "step": 6924
    },
    {
      "epoch": 6.603240966796875e-06,
      "step": 6924,
      "training_step_time": 7.66936469078064
    },
    {
      "epoch": 6.604194641113281e-06,
      "model_forward_time": 1.3835699558258057,
      "step": 6925
    },
    {
      "epoch": 6.604194641113281e-06,
      "step": 6925,
      "training_step_time": 7.606658697128296
    },
    {
      "epoch": 6.605148315429688e-06,
      "model_forward_time": 1.7465109825134277,
      "step": 6926
    },
    {
      "epoch": 6.605148315429688e-06,
      "step": 6926,
      "training_step_time": 9.258871078491211
    },
    {
      "epoch": 6.606101989746093e-06,
      "model_forward_time": 1.7399585247039795,
      "step": 6927
    },
    {
      "epoch": 6.606101989746093e-06,
      "step": 6927,
      "training_step_time": 8.296372652053833
    },
    {
      "epoch": 6.6070556640625e-06,
      "model_forward_time": 1.867938756942749,
      "step": 6928
    },
    {
      "epoch": 6.6070556640625e-06,
      "step": 6928,
      "training_step_time": 7.60364294052124
    },
    {
      "epoch": 6.6080093383789065e-06,
      "model_forward_time": 1.742959976196289,
      "step": 6929
    },
    {
      "epoch": 6.6080093383789065e-06,
      "step": 6929,
      "training_step_time": 7.761590480804443
    },
    {
      "epoch": 6.608963012695312e-06,
      "grad_norm": 0.2627109885215759,
      "learning_rate": 9.989830262998598e-06,
      "loss": 0.0714,
      "step": 6930
    },
    {
      "epoch": 6.608963012695312e-06,
      "model_forward_time": 2.9361135959625244,
      "step": 6930
    },
    {
      "dynamics_loss_avg": 0.060198495723307134,
      "epoch": 6.608963012695312e-06,
      "step": 6930
    },
    {
      "action_loss_avg": 0.004406173212919385,
      "epoch": 6.608963012695312e-06,
      "step": 6930
    },
    {
      "epoch": 6.608963012695312e-06,
      "step": 6930,
      "training_step_time": 7.7711710929870605
    },
    {
      "epoch": 6.609916687011719e-06,
      "model_forward_time": 1.7396125793457031,
      "step": 6931
    },
    {
      "epoch": 6.609916687011719e-06,
      "step": 6931,
      "training_step_time": 7.905553102493286
    },
    {
      "epoch": 6.610870361328125e-06,
      "model_forward_time": 1.7415344715118408,
      "step": 6932
    },
    {
      "epoch": 6.610870361328125e-06,
      "step": 6932,
      "training_step_time": 8.27250599861145
    },
    {
      "epoch": 6.611824035644531e-06,
      "model_forward_time": 1.745241403579712,
      "step": 6933
    },
    {
      "epoch": 6.611824035644531e-06,
      "step": 6933,
      "training_step_time": 7.580312967300415
    },
    {
      "epoch": 6.612777709960938e-06,
      "model_forward_time": 1.7400338649749756,
      "step": 6934
    },
    {
      "epoch": 6.612777709960938e-06,
      "step": 6934,
      "training_step_time": 7.557140111923218
    },
    {
      "epoch": 6.613731384277343e-06,
      "model_forward_time": 1.7428498268127441,
      "step": 6935
    },
    {
      "epoch": 6.613731384277343e-06,
      "step": 6935,
      "training_step_time": 8.203319549560547
    },
    {
      "epoch": 6.61468505859375e-06,
      "model_forward_time": 1.7452950477600098,
      "step": 6936
    },
    {
      "epoch": 6.61468505859375e-06,
      "step": 6936,
      "training_step_time": 8.250441074371338
    },
    {
      "epoch": 6.6156387329101565e-06,
      "model_forward_time": 1.849578857421875,
      "step": 6937
    },
    {
      "epoch": 6.6156387329101565e-06,
      "step": 6937,
      "training_step_time": 9.268753051757812
    },
    {
      "epoch": 6.616592407226562e-06,
      "model_forward_time": 1.923485279083252,
      "step": 6938
    },
    {
      "epoch": 6.616592407226562e-06,
      "step": 6938,
      "training_step_time": 7.874166250228882
    },
    {
      "epoch": 6.617546081542969e-06,
      "model_forward_time": 2.9777565002441406,
      "step": 6939
    },
    {
      "epoch": 6.617546081542969e-06,
      "step": 6939,
      "training_step_time": 7.861886024475098
    },
    {
      "epoch": 6.618499755859375e-06,
      "grad_norm": 0.2250174880027771,
      "learning_rate": 9.98972458541026e-06,
      "loss": 0.0658,
      "step": 6940
    },
    {
      "epoch": 6.618499755859375e-06,
      "model_forward_time": 1.9114537239074707,
      "step": 6940
    },
    {
      "dynamics_loss_avg": 0.08897492699325085,
      "epoch": 6.618499755859375e-06,
      "step": 6940
    },
    {
      "action_loss_avg": 0.009049502061679959,
      "epoch": 6.618499755859375e-06,
      "step": 6940
    },
    {
      "epoch": 6.618499755859375e-06,
      "step": 6940,
      "training_step_time": 7.6329123973846436
    },
    {
      "epoch": 6.619453430175781e-06,
      "model_forward_time": 1.8754608631134033,
      "step": 6941
    },
    {
      "epoch": 6.619453430175781e-06,
      "step": 6941,
      "training_step_time": 7.813615798950195
    },
    {
      "epoch": 6.620407104492188e-06,
      "model_forward_time": 1.9016039371490479,
      "step": 6942
    },
    {
      "epoch": 6.620407104492188e-06,
      "step": 6942,
      "training_step_time": 8.550052642822266
    },
    {
      "epoch": 6.621360778808593e-06,
      "model_forward_time": 1.9303936958312988,
      "step": 6943
    },
    {
      "epoch": 6.621360778808593e-06,
      "step": 6943,
      "training_step_time": 7.603363275527954
    },
    {
      "epoch": 6.622314453125e-06,
      "model_forward_time": 1.914543867111206,
      "step": 6944
    },
    {
      "epoch": 6.622314453125e-06,
      "step": 6944,
      "training_step_time": 8.776768922805786
    },
    {
      "epoch": 6.6232681274414065e-06,
      "model_forward_time": 1.8968887329101562,
      "step": 6945
    },
    {
      "epoch": 6.6232681274414065e-06,
      "step": 6945,
      "training_step_time": 8.124775648117065
    },
    {
      "epoch": 6.624221801757812e-06,
      "model_forward_time": 1.8818111419677734,
      "step": 6946
    },
    {
      "epoch": 6.624221801757812e-06,
      "step": 6946,
      "training_step_time": 8.668954849243164
    },
    {
      "epoch": 6.625175476074219e-06,
      "model_forward_time": 1.8979952335357666,
      "step": 6947
    },
    {
      "epoch": 6.625175476074219e-06,
      "step": 6947,
      "training_step_time": 9.500611782073975
    },
    {
      "epoch": 6.626129150390625e-06,
      "model_forward_time": 3.0088963508605957,
      "step": 6948
    },
    {
      "epoch": 6.626129150390625e-06,
      "step": 6948,
      "training_step_time": 8.470429420471191
    },
    {
      "epoch": 6.627082824707031e-06,
      "model_forward_time": 1.9175138473510742,
      "step": 6949
    },
    {
      "epoch": 6.627082824707031e-06,
      "step": 6949,
      "training_step_time": 7.496002435684204
    },
    {
      "epoch": 6.628036499023438e-06,
      "grad_norm": 0.1851583868265152,
      "learning_rate": 9.989618362153139e-06,
      "loss": 0.0753,
      "step": 6950
    },
    {
      "epoch": 6.628036499023438e-06,
      "model_forward_time": 1.9158542156219482,
      "step": 6950
    },
    {
      "dynamics_loss_avg": 0.07874930147081613,
      "epoch": 6.628036499023438e-06,
      "step": 6950
    },
    {
      "action_loss_avg": 0.0062934249639511105,
      "epoch": 6.628036499023438e-06,
      "step": 6950
    },
    {
      "epoch": 6.628036499023438e-06,
      "step": 6950,
      "training_step_time": 8.186070442199707
    },
    {
      "epoch": 6.628990173339843e-06,
      "model_forward_time": 1.9184043407440186,
      "step": 6951
    },
    {
      "epoch": 6.628990173339843e-06,
      "step": 6951,
      "training_step_time": 7.616132736206055
    },
    {
      "epoch": 6.62994384765625e-06,
      "model_forward_time": 1.9058513641357422,
      "step": 6952
    },
    {
      "epoch": 6.62994384765625e-06,
      "step": 6952,
      "training_step_time": 7.55269193649292
    },
    {
      "epoch": 6.6308975219726565e-06,
      "model_forward_time": 1.923356294631958,
      "step": 6953
    },
    {
      "epoch": 6.6308975219726565e-06,
      "step": 6953,
      "training_step_time": 7.7761900424957275
    },
    {
      "epoch": 6.631851196289062e-06,
      "model_forward_time": 1.9029457569122314,
      "step": 6954
    },
    {
      "epoch": 6.631851196289062e-06,
      "step": 6954,
      "training_step_time": 7.606918096542358
    },
    {
      "epoch": 6.632804870605469e-06,
      "model_forward_time": 1.8982722759246826,
      "step": 6955
    },
    {
      "epoch": 6.632804870605469e-06,
      "step": 6955,
      "training_step_time": 7.812275409698486
    },
    {
      "epoch": 6.633758544921875e-06,
      "model_forward_time": 1.9039716720581055,
      "step": 6956
    },
    {
      "epoch": 6.633758544921875e-06,
      "step": 6956,
      "training_step_time": 7.832340717315674
    },
    {
      "epoch": 6.634712219238281e-06,
      "model_forward_time": 2.9802708625793457,
      "step": 6957
    },
    {
      "epoch": 6.634712219238281e-06,
      "step": 6957,
      "training_step_time": 9.612765312194824
    },
    {
      "epoch": 6.635665893554688e-06,
      "model_forward_time": 1.9136180877685547,
      "step": 6958
    },
    {
      "epoch": 6.635665893554688e-06,
      "step": 6958,
      "training_step_time": 9.023082256317139
    },
    {
      "epoch": 6.636619567871093e-06,
      "model_forward_time": 1.5446231365203857,
      "step": 6959
    },
    {
      "epoch": 6.636619567871093e-06,
      "step": 6959,
      "training_step_time": 6.836950302124023
    },
    {
      "epoch": 6.6375732421875e-06,
      "grad_norm": 0.22776755690574646,
      "learning_rate": 9.989511593238859e-06,
      "loss": 0.075,
      "step": 6960
    },
    {
      "epoch": 6.6375732421875e-06,
      "model_forward_time": 1.904353141784668,
      "step": 6960
    },
    {
      "dynamics_loss_avg": 0.06844945773482322,
      "epoch": 6.6375732421875e-06,
      "step": 6960
    },
    {
      "action_loss_avg": 0.004808889207197353,
      "epoch": 6.6375732421875e-06,
      "step": 6960
    },
    {
      "epoch": 6.6375732421875e-06,
      "step": 6960,
      "training_step_time": 6.880366086959839
    },
    {
      "epoch": 6.6385269165039065e-06,
      "model_forward_time": 1.8988394737243652,
      "step": 6961
    },
    {
      "epoch": 6.6385269165039065e-06,
      "step": 6961,
      "training_step_time": 6.851885557174683
    },
    {
      "epoch": 6.639480590820312e-06,
      "model_forward_time": 1.908515453338623,
      "step": 6962
    },
    {
      "epoch": 6.639480590820312e-06,
      "step": 6962,
      "training_step_time": 6.813194513320923
    },
    {
      "epoch": 6.640434265136719e-06,
      "model_forward_time": 1.916471242904663,
      "step": 6963
    },
    {
      "epoch": 6.640434265136719e-06,
      "step": 6963,
      "training_step_time": 6.874460220336914
    },
    {
      "epoch": 6.641387939453125e-06,
      "model_forward_time": 1.9123668670654297,
      "step": 6964
    },
    {
      "epoch": 6.641387939453125e-06,
      "step": 6964,
      "training_step_time": 6.8317952156066895
    },
    {
      "epoch": 6.642341613769531e-06,
      "model_forward_time": 1.905660629272461,
      "step": 6965
    },
    {
      "epoch": 6.642341613769531e-06,
      "step": 6965,
      "training_step_time": 6.861737012863159
    },
    {
      "epoch": 6.643295288085938e-06,
      "model_forward_time": 1.8798668384552002,
      "step": 6966
    },
    {
      "epoch": 6.643295288085938e-06,
      "step": 6966,
      "training_step_time": 7.869842052459717
    },
    {
      "epoch": 6.644248962402343e-06,
      "model_forward_time": 2.9414072036743164,
      "step": 6967
    },
    {
      "epoch": 6.644248962402343e-06,
      "step": 6967,
      "training_step_time": 7.830315351486206
    },
    {
      "epoch": 6.64520263671875e-06,
      "model_forward_time": 1.8836963176727295,
      "step": 6968
    },
    {
      "epoch": 6.64520263671875e-06,
      "step": 6968,
      "training_step_time": 7.909470319747925
    },
    {
      "epoch": 6.6461563110351565e-06,
      "model_forward_time": 1.9202613830566406,
      "step": 6969
    },
    {
      "epoch": 6.6461563110351565e-06,
      "step": 6969,
      "training_step_time": 6.888326168060303
    },
    {
      "epoch": 6.647109985351562e-06,
      "grad_norm": 0.25791001319885254,
      "learning_rate": 9.989404278679092e-06,
      "loss": 0.071,
      "step": 6970
    },
    {
      "epoch": 6.647109985351562e-06,
      "model_forward_time": 1.899634599685669,
      "step": 6970
    },
    {
      "dynamics_loss_avg": 0.06509614121168852,
      "epoch": 6.647109985351562e-06,
      "step": 6970
    },
    {
      "action_loss_avg": 0.005085169145604596,
      "epoch": 6.647109985351562e-06,
      "step": 6970
    },
    {
      "epoch": 6.647109985351562e-06,
      "step": 6970,
      "training_step_time": 6.941642999649048
    },
    {
      "epoch": 6.648063659667969e-06,
      "model_forward_time": 1.899397373199463,
      "step": 6971
    },
    {
      "epoch": 6.648063659667969e-06,
      "step": 6971,
      "training_step_time": 6.897016286849976
    },
    {
      "epoch": 6.649017333984375e-06,
      "model_forward_time": 1.8927454948425293,
      "step": 6972
    },
    {
      "epoch": 6.649017333984375e-06,
      "step": 6972,
      "training_step_time": 6.881694078445435
    },
    {
      "epoch": 6.649971008300781e-06,
      "model_forward_time": 1.919043779373169,
      "step": 6973
    },
    {
      "epoch": 6.649971008300781e-06,
      "step": 6973,
      "training_step_time": 6.892231464385986
    },
    {
      "epoch": 6.650924682617188e-06,
      "model_forward_time": 1.9121530055999756,
      "step": 6974
    },
    {
      "epoch": 6.650924682617188e-06,
      "step": 6974,
      "training_step_time": 6.8745293617248535
    },
    {
      "epoch": 6.651878356933593e-06,
      "model_forward_time": 1.9134719371795654,
      "step": 6975
    },
    {
      "epoch": 6.651878356933593e-06,
      "step": 6975,
      "training_step_time": 6.886914253234863
    },
    {
      "epoch": 6.65283203125e-06,
      "model_forward_time": 2.966860771179199,
      "step": 6976
    },
    {
      "epoch": 6.65283203125e-06,
      "step": 6976,
      "training_step_time": 7.847556114196777
    },
    {
      "epoch": 6.6537857055664065e-06,
      "model_forward_time": 1.8953616619110107,
      "step": 6977
    },
    {
      "epoch": 6.6537857055664065e-06,
      "step": 6977,
      "training_step_time": 7.92469596862793
    },
    {
      "epoch": 6.654739379882812e-06,
      "model_forward_time": 1.9176669120788574,
      "step": 6978
    },
    {
      "epoch": 6.654739379882812e-06,
      "step": 6978,
      "training_step_time": 6.898351430892944
    },
    {
      "epoch": 6.655693054199219e-06,
      "model_forward_time": 1.917820692062378,
      "step": 6979
    },
    {
      "epoch": 6.655693054199219e-06,
      "step": 6979,
      "training_step_time": 6.870274543762207
    },
    {
      "epoch": 6.656646728515625e-06,
      "grad_norm": 0.25439491868019104,
      "learning_rate": 9.989296418485574e-06,
      "loss": 0.0682,
      "step": 6980
    },
    {
      "epoch": 6.656646728515625e-06,
      "model_forward_time": 1.9012188911437988,
      "step": 6980
    },
    {
      "dynamics_loss_avg": 0.05764131732285023,
      "epoch": 6.656646728515625e-06,
      "step": 6980
    },
    {
      "action_loss_avg": 0.004621970083098859,
      "epoch": 6.656646728515625e-06,
      "step": 6980
    },
    {
      "epoch": 6.656646728515625e-06,
      "step": 6980,
      "training_step_time": 6.869070529937744
    },
    {
      "epoch": 6.657600402832031e-06,
      "model_forward_time": 1.9035933017730713,
      "step": 6981
    },
    {
      "epoch": 6.657600402832031e-06,
      "step": 6981,
      "training_step_time": 6.91949200630188
    },
    {
      "epoch": 6.658554077148438e-06,
      "model_forward_time": 1.8982188701629639,
      "step": 6982
    },
    {
      "epoch": 6.658554077148438e-06,
      "step": 6982,
      "training_step_time": 6.864164352416992
    },
    {
      "epoch": 6.659507751464843e-06,
      "model_forward_time": 1.910888433456421,
      "step": 6983
    },
    {
      "epoch": 6.659507751464843e-06,
      "step": 6983,
      "training_step_time": 6.851152658462524
    },
    {
      "epoch": 6.66046142578125e-06,
      "model_forward_time": 1.8890464305877686,
      "step": 6984
    },
    {
      "epoch": 6.66046142578125e-06,
      "step": 6984,
      "training_step_time": 6.886338710784912
    },
    {
      "epoch": 6.6614151000976566e-06,
      "model_forward_time": 2.957836389541626,
      "step": 6985
    },
    {
      "epoch": 6.6614151000976566e-06,
      "step": 6985,
      "training_step_time": 7.849487543106079
    },
    {
      "epoch": 6.662368774414062e-06,
      "model_forward_time": 1.8783388137817383,
      "step": 6986
    },
    {
      "epoch": 6.662368774414062e-06,
      "step": 6986,
      "training_step_time": 9.199750185012817
    },
    {
      "epoch": 6.663322448730469e-06,
      "model_forward_time": 1.8903203010559082,
      "step": 6987
    },
    {
      "epoch": 6.663322448730469e-06,
      "step": 6987,
      "training_step_time": 7.524519205093384
    },
    {
      "epoch": 6.664276123046875e-06,
      "model_forward_time": 1.900416612625122,
      "step": 6988
    },
    {
      "epoch": 6.664276123046875e-06,
      "step": 6988,
      "training_step_time": 6.8897600173950195
    },
    {
      "epoch": 6.665229797363281e-06,
      "model_forward_time": 1.9067745208740234,
      "step": 6989
    },
    {
      "epoch": 6.665229797363281e-06,
      "step": 6989,
      "training_step_time": 6.871054410934448
    },
    {
      "epoch": 6.666183471679688e-06,
      "grad_norm": 0.23220016062259674,
      "learning_rate": 9.989188012670101e-06,
      "loss": 0.0668,
      "step": 6990
    },
    {
      "epoch": 6.666183471679688e-06,
      "model_forward_time": 1.9154815673828125,
      "step": 6990
    },
    {
      "dynamics_loss_avg": 0.05992415696382523,
      "epoch": 6.666183471679688e-06,
      "step": 6990
    },
    {
      "action_loss_avg": 0.005693436285946518,
      "epoch": 6.666183471679688e-06,
      "step": 6990
    },
    {
      "epoch": 6.666183471679688e-06,
      "step": 6990,
      "training_step_time": 6.912914514541626
    },
    {
      "epoch": 6.6671371459960934e-06,
      "model_forward_time": 1.9162936210632324,
      "step": 6991
    },
    {
      "epoch": 6.6671371459960934e-06,
      "step": 6991,
      "training_step_time": 6.851433992385864
    },
    {
      "epoch": 6.6680908203125e-06,
      "model_forward_time": 1.900043249130249,
      "step": 6992
    },
    {
      "epoch": 6.6680908203125e-06,
      "step": 6992,
      "training_step_time": 6.848110198974609
    },
    {
      "epoch": 6.6690444946289066e-06,
      "model_forward_time": 1.90561842918396,
      "step": 6993
    },
    {
      "epoch": 6.6690444946289066e-06,
      "step": 6993,
      "training_step_time": 6.863512754440308
    },
    {
      "epoch": 6.669998168945312e-06,
      "model_forward_time": 3.0001773834228516,
      "step": 6994
    },
    {
      "epoch": 6.669998168945312e-06,
      "step": 6994,
      "training_step_time": 7.9114179611206055
    },
    {
      "epoch": 6.670951843261719e-06,
      "model_forward_time": 1.9018425941467285,
      "step": 6995
    },
    {
      "epoch": 6.670951843261719e-06,
      "step": 6995,
      "training_step_time": 11.877775192260742
    },
    {
      "epoch": 6.671905517578125e-06,
      "model_forward_time": 1.9076762199401855,
      "step": 6996
    },
    {
      "epoch": 6.671905517578125e-06,
      "step": 6996,
      "training_step_time": 7.737310171127319
    },
    {
      "epoch": 6.672859191894531e-06,
      "model_forward_time": 1.8993196487426758,
      "step": 6997
    },
    {
      "epoch": 6.672859191894531e-06,
      "step": 6997,
      "training_step_time": 6.951548337936401
    },
    {
      "epoch": 6.673812866210938e-06,
      "model_forward_time": 1.916677713394165,
      "step": 6998
    },
    {
      "epoch": 6.673812866210938e-06,
      "step": 6998,
      "training_step_time": 7.050022125244141
    },
    {
      "epoch": 6.6747665405273434e-06,
      "model_forward_time": 1.929215908050537,
      "step": 6999
    },
    {
      "epoch": 6.6747665405273434e-06,
      "step": 6999,
      "training_step_time": 6.912086725234985
    },
    {
      "epoch": 6.67572021484375e-06,
      "grad_norm": 0.20531709492206573,
      "learning_rate": 9.989079061244529e-06,
      "loss": 0.0697,
      "step": 7000
    },
    {
      "epoch": 6.67572021484375e-06,
      "model_forward_time": 1.3246641159057617,
      "step": 7000
    },
    {
      "dynamics_loss_avg": 0.04938762160018086,
      "epoch": 6.67572021484375e-06,
      "step": 7000
    },
    {
      "action_loss_avg": 0.003045564331114292,
      "epoch": 6.67572021484375e-06,
      "step": 7000
    },
    {
      "epoch": 6.67572021484375e-06,
      "step": 7000,
      "training_step_time": 6.496188163757324
    },
    {
      "epoch": 6.6766738891601566e-06,
      "model_forward_time": 1.3601961135864258,
      "step": 7001
    },
    {
      "epoch": 6.6766738891601566e-06,
      "step": 7001,
      "training_step_time": 6.8432910442352295
    },
    {
      "epoch": 6.677627563476562e-06,
      "model_forward_time": 1.7280359268188477,
      "step": 7002
    },
    {
      "epoch": 6.677627563476562e-06,
      "step": 7002,
      "training_step_time": 6.855125665664673
    },
    {
      "epoch": 6.678581237792969e-06,
      "model_forward_time": 2.8836660385131836,
      "step": 7003
    },
    {
      "epoch": 6.678581237792969e-06,
      "step": 7003,
      "training_step_time": 9.454290866851807
    },
    {
      "epoch": 6.679534912109375e-06,
      "model_forward_time": 1.7204108238220215,
      "step": 7004
    },
    {
      "epoch": 6.679534912109375e-06,
      "step": 7004,
      "training_step_time": 7.80314302444458
    },
    {
      "epoch": 6.680488586425781e-06,
      "model_forward_time": 1.3473050594329834,
      "step": 7005
    },
    {
      "epoch": 6.680488586425781e-06,
      "step": 7005,
      "training_step_time": 7.861603736877441
    },
    {
      "epoch": 6.681442260742188e-06,
      "model_forward_time": 1.7254221439361572,
      "step": 7006
    },
    {
      "epoch": 6.681442260742188e-06,
      "step": 7006,
      "training_step_time": 6.860677003860474
    },
    {
      "epoch": 6.6823959350585934e-06,
      "model_forward_time": 1.7281200885772705,
      "step": 7007
    },
    {
      "epoch": 6.6823959350585934e-06,
      "step": 7007,
      "training_step_time": 6.886652231216431
    },
    {
      "epoch": 6.683349609375e-06,
      "model_forward_time": 1.7247755527496338,
      "step": 7008
    },
    {
      "epoch": 6.683349609375e-06,
      "step": 7008,
      "training_step_time": 6.90909743309021
    },
    {
      "epoch": 6.6843032836914066e-06,
      "model_forward_time": 1.7242143154144287,
      "step": 7009
    },
    {
      "epoch": 6.6843032836914066e-06,
      "step": 7009,
      "training_step_time": 6.915201187133789
    },
    {
      "epoch": 6.685256958007812e-06,
      "grad_norm": 0.2767711579799652,
      "learning_rate": 9.98896956422077e-06,
      "loss": 0.0685,
      "step": 7010
    },
    {
      "epoch": 6.685256958007812e-06,
      "model_forward_time": 1.7209503650665283,
      "step": 7010
    },
    {
      "dynamics_loss_avg": 0.07074410328641534,
      "epoch": 6.685256958007812e-06,
      "step": 7010
    },
    {
      "action_loss_avg": 0.006717211066279561,
      "epoch": 6.685256958007812e-06,
      "step": 7010
    },
    {
      "epoch": 6.685256958007812e-06,
      "step": 7010,
      "training_step_time": 6.948623180389404
    },
    {
      "epoch": 6.686210632324219e-06,
      "model_forward_time": 1.7331535816192627,
      "step": 7011
    },
    {
      "epoch": 6.686210632324219e-06,
      "step": 7011,
      "training_step_time": 6.858856678009033
    },
    {
      "epoch": 6.687164306640625e-06,
      "model_forward_time": 2.270986557006836,
      "step": 7012
    },
    {
      "epoch": 6.687164306640625e-06,
      "step": 7012,
      "training_step_time": 8.031270265579224
    },
    {
      "epoch": 6.688117980957031e-06,
      "model_forward_time": 1.7423381805419922,
      "step": 7013
    },
    {
      "epoch": 6.688117980957031e-06,
      "step": 7013,
      "training_step_time": 9.338656902313232
    },
    {
      "epoch": 6.689071655273438e-06,
      "model_forward_time": 1.7273337841033936,
      "step": 7014
    },
    {
      "epoch": 6.689071655273438e-06,
      "step": 7014,
      "training_step_time": 7.821505784988403
    },
    {
      "epoch": 6.6900253295898434e-06,
      "model_forward_time": 1.7302930355072021,
      "step": 7015
    },
    {
      "epoch": 6.6900253295898434e-06,
      "step": 7015,
      "training_step_time": 7.375072002410889
    },
    {
      "epoch": 6.69097900390625e-06,
      "model_forward_time": 1.7091178894042969,
      "step": 7016
    },
    {
      "epoch": 6.69097900390625e-06,
      "step": 7016,
      "training_step_time": 6.867685317993164
    },
    {
      "epoch": 6.691932678222657e-06,
      "model_forward_time": 1.7336089611053467,
      "step": 7017
    },
    {
      "epoch": 6.691932678222657e-06,
      "step": 7017,
      "training_step_time": 6.869378089904785
    },
    {
      "epoch": 6.692886352539062e-06,
      "model_forward_time": 1.7289490699768066,
      "step": 7018
    },
    {
      "epoch": 6.692886352539062e-06,
      "step": 7018,
      "training_step_time": 6.848607301712036
    },
    {
      "epoch": 6.693840026855469e-06,
      "model_forward_time": 1.702563762664795,
      "step": 7019
    },
    {
      "epoch": 6.693840026855469e-06,
      "step": 7019,
      "training_step_time": 6.868601083755493
    },
    {
      "epoch": 6.694793701171875e-06,
      "grad_norm": 0.22514960169792175,
      "learning_rate": 9.988859521610802e-06,
      "loss": 0.0694,
      "step": 7020
    },
    {
      "epoch": 6.694793701171875e-06,
      "model_forward_time": 1.6978557109832764,
      "step": 7020
    },
    {
      "dynamics_loss_avg": 0.05738854892551899,
      "epoch": 6.694793701171875e-06,
      "step": 7020
    },
    {
      "action_loss_avg": 0.0054096061736345295,
      "epoch": 6.694793701171875e-06,
      "step": 7020
    },
    {
      "epoch": 6.694793701171875e-06,
      "step": 7020,
      "training_step_time": 6.886969089508057
    },
    {
      "epoch": 6.695747375488281e-06,
      "model_forward_time": 1.6974363327026367,
      "step": 7021
    },
    {
      "epoch": 6.695747375488281e-06,
      "step": 7021,
      "training_step_time": 7.35692286491394
    },
    {
      "epoch": 6.696701049804688e-06,
      "model_forward_time": 2.2412092685699463,
      "step": 7022
    },
    {
      "epoch": 6.696701049804688e-06,
      "step": 7022,
      "training_step_time": 10.165233612060547
    },
    {
      "epoch": 6.6976547241210935e-06,
      "model_forward_time": 1.6982455253601074,
      "step": 7023
    },
    {
      "epoch": 6.6976547241210935e-06,
      "step": 7023,
      "training_step_time": 9.080664157867432
    },
    {
      "epoch": 6.6986083984375e-06,
      "model_forward_time": 1.7047054767608643,
      "step": 7024
    },
    {
      "epoch": 6.6986083984375e-06,
      "step": 7024,
      "training_step_time": 7.781811237335205
    },
    {
      "epoch": 6.699562072753907e-06,
      "model_forward_time": 1.6932120323181152,
      "step": 7025
    },
    {
      "epoch": 6.699562072753907e-06,
      "step": 7025,
      "training_step_time": 6.825313568115234
    },
    {
      "epoch": 6.700515747070312e-06,
      "model_forward_time": 1.6930150985717773,
      "step": 7026
    },
    {
      "epoch": 6.700515747070312e-06,
      "step": 7026,
      "training_step_time": 6.897745370864868
    },
    {
      "epoch": 6.701469421386719e-06,
      "model_forward_time": 1.7021410465240479,
      "step": 7027
    },
    {
      "epoch": 6.701469421386719e-06,
      "step": 7027,
      "training_step_time": 6.849097490310669
    },
    {
      "epoch": 6.702423095703125e-06,
      "model_forward_time": 1.6990110874176025,
      "step": 7028
    },
    {
      "epoch": 6.702423095703125e-06,
      "step": 7028,
      "training_step_time": 6.785911560058594
    },
    {
      "epoch": 6.703376770019531e-06,
      "model_forward_time": 1.6989078521728516,
      "step": 7029
    },
    {
      "epoch": 6.703376770019531e-06,
      "step": 7029,
      "training_step_time": 6.854322195053101
    },
    {
      "epoch": 6.704330444335938e-06,
      "grad_norm": 0.2318904846906662,
      "learning_rate": 9.988748933426657e-06,
      "loss": 0.071,
      "step": 7030
    },
    {
      "epoch": 6.704330444335938e-06,
      "model_forward_time": 1.6914708614349365,
      "step": 7030
    },
    {
      "dynamics_loss_avg": 0.07978128511458635,
      "epoch": 6.704330444335938e-06,
      "step": 7030
    },
    {
      "action_loss_avg": 0.009058407158590853,
      "epoch": 6.704330444335938e-06,
      "step": 7030
    },
    {
      "epoch": 6.704330444335938e-06,
      "step": 7030,
      "training_step_time": 7.345685720443726
    },
    {
      "epoch": 6.7052841186523435e-06,
      "model_forward_time": 2.258716106414795,
      "step": 7031
    },
    {
      "epoch": 6.7052841186523435e-06,
      "step": 7031,
      "training_step_time": 9.95850419998169
    },
    {
      "epoch": 6.70623779296875e-06,
      "model_forward_time": 1.6997690200805664,
      "step": 7032
    },
    {
      "epoch": 6.70623779296875e-06,
      "step": 7032,
      "training_step_time": 8.526461124420166
    },
    {
      "epoch": 6.707191467285157e-06,
      "model_forward_time": 1.6940257549285889,
      "step": 7033
    },
    {
      "epoch": 6.707191467285157e-06,
      "step": 7033,
      "training_step_time": 8.600260734558105
    },
    {
      "epoch": 6.708145141601562e-06,
      "model_forward_time": 1.6980304718017578,
      "step": 7034
    },
    {
      "epoch": 6.708145141601562e-06,
      "step": 7034,
      "training_step_time": 6.8470985889434814
    },
    {
      "epoch": 6.709098815917969e-06,
      "model_forward_time": 1.6962347030639648,
      "step": 7035
    },
    {
      "epoch": 6.709098815917969e-06,
      "step": 7035,
      "training_step_time": 6.832996606826782
    },
    {
      "epoch": 6.710052490234375e-06,
      "model_forward_time": 1.698789119720459,
      "step": 7036
    },
    {
      "epoch": 6.710052490234375e-06,
      "step": 7036,
      "training_step_time": 6.847739934921265
    },
    {
      "epoch": 6.711006164550781e-06,
      "model_forward_time": 1.6924912929534912,
      "step": 7037
    },
    {
      "epoch": 6.711006164550781e-06,
      "step": 7037,
      "training_step_time": 6.856986045837402
    },
    {
      "epoch": 6.711959838867188e-06,
      "model_forward_time": 1.6966991424560547,
      "step": 7038
    },
    {
      "epoch": 6.711959838867188e-06,
      "step": 7038,
      "training_step_time": 6.868721008300781
    },
    {
      "epoch": 6.7129135131835935e-06,
      "model_forward_time": 1.7017359733581543,
      "step": 7039
    },
    {
      "epoch": 6.7129135131835935e-06,
      "step": 7039,
      "training_step_time": 6.852435827255249
    },
    {
      "epoch": 6.7138671875e-06,
      "grad_norm": 0.2243587225675583,
      "learning_rate": 9.988637799680428e-06,
      "loss": 0.073,
      "step": 7040
    },
    {
      "epoch": 6.7138671875e-06,
      "model_forward_time": 2.2510077953338623,
      "step": 7040
    },
    {
      "dynamics_loss_avg": 0.06852555982768535,
      "epoch": 6.7138671875e-06,
      "step": 7040
    },
    {
      "action_loss_avg": 0.006764807994477451,
      "epoch": 6.7138671875e-06,
      "step": 7040
    },
    {
      "epoch": 6.7138671875e-06,
      "step": 7040,
      "training_step_time": 9.852218627929688
    },
    {
      "epoch": 6.714820861816407e-06,
      "model_forward_time": 1.721170425415039,
      "step": 7041
    },
    {
      "epoch": 6.714820861816407e-06,
      "step": 7041,
      "training_step_time": 11.350260734558105
    },
    {
      "epoch": 6.715774536132812e-06,
      "model_forward_time": 1.696716547012329,
      "step": 7042
    },
    {
      "epoch": 6.715774536132812e-06,
      "step": 7042,
      "training_step_time": 8.322100400924683
    },
    {
      "epoch": 6.716728210449219e-06,
      "model_forward_time": 1.691992998123169,
      "step": 7043
    },
    {
      "epoch": 6.716728210449219e-06,
      "step": 7043,
      "training_step_time": 7.364950180053711
    },
    {
      "epoch": 6.717681884765625e-06,
      "model_forward_time": 1.6995420455932617,
      "step": 7044
    },
    {
      "epoch": 6.717681884765625e-06,
      "step": 7044,
      "training_step_time": 6.857178211212158
    },
    {
      "epoch": 6.718635559082031e-06,
      "model_forward_time": 1.695770025253296,
      "step": 7045
    },
    {
      "epoch": 6.718635559082031e-06,
      "step": 7045,
      "training_step_time": 6.910686016082764
    },
    {
      "epoch": 6.719589233398438e-06,
      "model_forward_time": 1.700423002243042,
      "step": 7046
    },
    {
      "epoch": 6.719589233398438e-06,
      "step": 7046,
      "training_step_time": 6.910120010375977
    },
    {
      "epoch": 6.7205429077148435e-06,
      "model_forward_time": 1.6953809261322021,
      "step": 7047
    },
    {
      "epoch": 6.7205429077148435e-06,
      "step": 7047,
      "training_step_time": 6.881452322006226
    },
    {
      "epoch": 6.72149658203125e-06,
      "model_forward_time": 1.6943161487579346,
      "step": 7048
    },
    {
      "epoch": 6.72149658203125e-06,
      "step": 7048,
      "training_step_time": 6.918019533157349
    },
    {
      "epoch": 6.722450256347657e-06,
      "model_forward_time": 1.8624284267425537,
      "step": 7049
    },
    {
      "epoch": 6.722450256347657e-06,
      "step": 7049,
      "training_step_time": 8.604143619537354
    },
    {
      "epoch": 6.723403930664062e-06,
      "grad_norm": 0.22840017080307007,
      "learning_rate": 9.98852612038427e-06,
      "loss": 0.0742,
      "step": 7050
    },
    {
      "epoch": 6.723403930664062e-06,
      "model_forward_time": 1.7064626216888428,
      "step": 7050
    },
    {
      "dynamics_loss_avg": 0.06650864183902741,
      "epoch": 6.723403930664062e-06,
      "step": 7050
    },
    {
      "action_loss_avg": 0.007122199051082134,
      "epoch": 6.723403930664062e-06,
      "step": 7050
    },
    {
      "epoch": 6.723403930664062e-06,
      "step": 7050,
      "training_step_time": 8.894694089889526
    },
    {
      "epoch": 6.724357604980469e-06,
      "model_forward_time": 1.7019946575164795,
      "step": 7051
    },
    {
      "epoch": 6.724357604980469e-06,
      "step": 7051,
      "training_step_time": 7.886545181274414
    },
    {
      "epoch": 6.725311279296875e-06,
      "model_forward_time": 1.7005794048309326,
      "step": 7052
    },
    {
      "epoch": 6.725311279296875e-06,
      "step": 7052,
      "training_step_time": 7.553765773773193
    },
    {
      "epoch": 6.726264953613281e-06,
      "model_forward_time": 1.6969926357269287,
      "step": 7053
    },
    {
      "epoch": 6.726264953613281e-06,
      "step": 7053,
      "training_step_time": 6.926673889160156
    },
    {
      "epoch": 6.727218627929688e-06,
      "model_forward_time": 1.6937785148620605,
      "step": 7054
    },
    {
      "epoch": 6.727218627929688e-06,
      "step": 7054,
      "training_step_time": 6.908600330352783
    },
    {
      "epoch": 6.7281723022460935e-06,
      "model_forward_time": 1.6967206001281738,
      "step": 7055
    },
    {
      "epoch": 6.7281723022460935e-06,
      "step": 7055,
      "training_step_time": 6.880324363708496
    },
    {
      "epoch": 6.7291259765625e-06,
      "model_forward_time": 1.6957876682281494,
      "step": 7056
    },
    {
      "epoch": 6.7291259765625e-06,
      "step": 7056,
      "training_step_time": 6.888587236404419
    },
    {
      "epoch": 6.730079650878907e-06,
      "model_forward_time": 1.7025797367095947,
      "step": 7057
    },
    {
      "epoch": 6.730079650878907e-06,
      "step": 7057,
      "training_step_time": 6.940093517303467
    },
    {
      "epoch": 6.731033325195312e-06,
      "model_forward_time": 2.19645357131958,
      "step": 7058
    },
    {
      "epoch": 6.731033325195312e-06,
      "step": 7058,
      "training_step_time": 7.526898384094238
    },
    {
      "epoch": 6.731986999511719e-06,
      "model_forward_time": 1.7329635620117188,
      "step": 7059
    },
    {
      "epoch": 6.731986999511719e-06,
      "step": 7059,
      "training_step_time": 8.786538362503052
    },
    {
      "epoch": 6.732940673828125e-06,
      "grad_norm": 0.27703168988227844,
      "learning_rate": 9.988413895550398e-06,
      "loss": 0.0758,
      "step": 7060
    },
    {
      "epoch": 6.732940673828125e-06,
      "model_forward_time": 1.7016947269439697,
      "step": 7060
    },
    {
      "dynamics_loss_avg": 0.05824509970843792,
      "epoch": 6.732940673828125e-06,
      "step": 7060
    },
    {
      "action_loss_avg": 0.004072233190527186,
      "epoch": 6.732940673828125e-06,
      "step": 7060
    },
    {
      "epoch": 6.732940673828125e-06,
      "step": 7060,
      "training_step_time": 7.9866416454315186
    },
    {
      "epoch": 6.733894348144531e-06,
      "model_forward_time": 1.6960804462432861,
      "step": 7061
    },
    {
      "epoch": 6.733894348144531e-06,
      "step": 7061,
      "training_step_time": 7.513152837753296
    },
    {
      "epoch": 6.734848022460938e-06,
      "model_forward_time": 1.7032701969146729,
      "step": 7062
    },
    {
      "epoch": 6.734848022460938e-06,
      "step": 7062,
      "training_step_time": 6.927311658859253
    },
    {
      "epoch": 6.7358016967773435e-06,
      "model_forward_time": 1.70184326171875,
      "step": 7063
    },
    {
      "epoch": 6.7358016967773435e-06,
      "step": 7063,
      "training_step_time": 6.9530699253082275
    },
    {
      "epoch": 6.73675537109375e-06,
      "model_forward_time": 1.7005164623260498,
      "step": 7064
    },
    {
      "epoch": 6.73675537109375e-06,
      "step": 7064,
      "training_step_time": 6.938363552093506
    },
    {
      "epoch": 6.737709045410157e-06,
      "model_forward_time": 1.32808256149292,
      "step": 7065
    },
    {
      "epoch": 6.737709045410157e-06,
      "step": 7065,
      "training_step_time": 6.927601099014282
    },
    {
      "epoch": 6.738662719726562e-06,
      "model_forward_time": 1.7025341987609863,
      "step": 7066
    },
    {
      "epoch": 6.738662719726562e-06,
      "step": 7066,
      "training_step_time": 6.928313732147217
    },
    {
      "epoch": 6.739616394042969e-06,
      "model_forward_time": 1.7040495872497559,
      "step": 7067
    },
    {
      "epoch": 6.739616394042969e-06,
      "step": 7067,
      "training_step_time": 7.519114971160889
    },
    {
      "epoch": 6.740570068359375e-06,
      "model_forward_time": 2.236485481262207,
      "step": 7068
    },
    {
      "epoch": 6.740570068359375e-06,
      "step": 7068,
      "training_step_time": 8.642101049423218
    },
    {
      "epoch": 6.741523742675781e-06,
      "model_forward_time": 1.7072970867156982,
      "step": 7069
    },
    {
      "epoch": 6.741523742675781e-06,
      "step": 7069,
      "training_step_time": 7.911917209625244
    },
    {
      "epoch": 6.742477416992188e-06,
      "grad_norm": 0.15586146712303162,
      "learning_rate": 9.98830112519108e-06,
      "loss": 0.0737,
      "step": 7070
    },
    {
      "epoch": 6.742477416992188e-06,
      "model_forward_time": 1.7004272937774658,
      "step": 7070
    },
    {
      "dynamics_loss_avg": 0.06568817663937807,
      "epoch": 6.742477416992188e-06,
      "step": 7070
    },
    {
      "action_loss_avg": 0.0055461513809859754,
      "epoch": 6.742477416992188e-06,
      "step": 7070
    },
    {
      "epoch": 6.742477416992188e-06,
      "step": 7070,
      "training_step_time": 7.798747777938843
    },
    {
      "epoch": 6.7434310913085935e-06,
      "model_forward_time": 1.705491065979004,
      "step": 7071
    },
    {
      "epoch": 6.7434310913085935e-06,
      "step": 7071,
      "training_step_time": 6.847477197647095
    },
    {
      "epoch": 6.744384765625e-06,
      "model_forward_time": 1.706427812576294,
      "step": 7072
    },
    {
      "epoch": 6.744384765625e-06,
      "step": 7072,
      "training_step_time": 6.881860256195068
    },
    {
      "epoch": 6.745338439941407e-06,
      "model_forward_time": 1.6934454441070557,
      "step": 7073
    },
    {
      "epoch": 6.745338439941407e-06,
      "step": 7073,
      "training_step_time": 6.923614263534546
    },
    {
      "epoch": 6.746292114257812e-06,
      "model_forward_time": 1.7049310207366943,
      "step": 7074
    },
    {
      "epoch": 6.746292114257812e-06,
      "step": 7074,
      "training_step_time": 6.941462278366089
    },
    {
      "epoch": 6.747245788574219e-06,
      "model_forward_time": 1.6984763145446777,
      "step": 7075
    },
    {
      "epoch": 6.747245788574219e-06,
      "step": 7075,
      "training_step_time": 6.912854909896851
    },
    {
      "epoch": 6.748199462890625e-06,
      "model_forward_time": 1.7023472785949707,
      "step": 7076
    },
    {
      "epoch": 6.748199462890625e-06,
      "step": 7076,
      "training_step_time": 7.510380029678345
    },
    {
      "epoch": 6.749153137207031e-06,
      "model_forward_time": 2.2347617149353027,
      "step": 7077
    },
    {
      "epoch": 6.749153137207031e-06,
      "step": 7077,
      "training_step_time": 9.628318548202515
    },
    {
      "epoch": 6.750106811523438e-06,
      "model_forward_time": 1.7059273719787598,
      "step": 7078
    },
    {
      "epoch": 6.750106811523438e-06,
      "step": 7078,
      "training_step_time": 10.941372871398926
    },
    {
      "epoch": 6.7510604858398435e-06,
      "model_forward_time": 1.697944164276123,
      "step": 7079
    },
    {
      "epoch": 6.7510604858398435e-06,
      "step": 7079,
      "training_step_time": 8.871711730957031
    },
    {
      "epoch": 6.75201416015625e-06,
      "grad_norm": 0.27269428968429565,
      "learning_rate": 9.988187809318651e-06,
      "loss": 0.0785,
      "step": 7080
    },
    {
      "epoch": 6.75201416015625e-06,
      "model_forward_time": 1.69594144821167,
      "step": 7080
    },
    {
      "dynamics_loss_avg": 0.07214552797377109,
      "epoch": 6.75201416015625e-06,
      "step": 7080
    },
    {
      "action_loss_avg": 0.005263800331158563,
      "epoch": 6.75201416015625e-06,
      "step": 7080
    },
    {
      "epoch": 6.75201416015625e-06,
      "step": 7080,
      "training_step_time": 6.862834215164185
    },
    {
      "epoch": 6.752967834472657e-06,
      "model_forward_time": 1.704716682434082,
      "step": 7081
    },
    {
      "epoch": 6.752967834472657e-06,
      "step": 7081,
      "training_step_time": 6.865992069244385
    },
    {
      "epoch": 6.753921508789062e-06,
      "model_forward_time": 1.7116997241973877,
      "step": 7082
    },
    {
      "epoch": 6.753921508789062e-06,
      "step": 7082,
      "training_step_time": 6.893611431121826
    },
    {
      "epoch": 6.754875183105469e-06,
      "model_forward_time": 1.727895975112915,
      "step": 7083
    },
    {
      "epoch": 6.754875183105469e-06,
      "step": 7083,
      "training_step_time": 6.893433094024658
    },
    {
      "epoch": 6.755828857421875e-06,
      "model_forward_time": 1.7250020503997803,
      "step": 7084
    },
    {
      "epoch": 6.755828857421875e-06,
      "step": 7084,
      "training_step_time": 6.870067596435547
    },
    {
      "epoch": 6.756782531738281e-06,
      "model_forward_time": 1.7146086692810059,
      "step": 7085
    },
    {
      "epoch": 6.756782531738281e-06,
      "step": 7085,
      "training_step_time": 7.4794275760650635
    },
    {
      "epoch": 6.757736206054688e-06,
      "model_forward_time": 2.276060104370117,
      "step": 7086
    },
    {
      "epoch": 6.757736206054688e-06,
      "step": 7086,
      "training_step_time": 9.397670269012451
    },
    {
      "epoch": 6.7586898803710935e-06,
      "model_forward_time": 1.709477424621582,
      "step": 7087
    },
    {
      "epoch": 6.7586898803710935e-06,
      "step": 7087,
      "training_step_time": 9.43262267112732
    },
    {
      "epoch": 6.7596435546875e-06,
      "model_forward_time": 1.7168705463409424,
      "step": 7088
    },
    {
      "epoch": 6.7596435546875e-06,
      "step": 7088,
      "training_step_time": 7.847977876663208
    },
    {
      "epoch": 6.760597229003907e-06,
      "model_forward_time": 1.7120943069458008,
      "step": 7089
    },
    {
      "epoch": 6.760597229003907e-06,
      "step": 7089,
      "training_step_time": 7.592818260192871
    },
    {
      "epoch": 6.761550903320312e-06,
      "grad_norm": 0.20836259424686432,
      "learning_rate": 9.988073947945503e-06,
      "loss": 0.0749,
      "step": 7090
    },
    {
      "epoch": 6.761550903320312e-06,
      "model_forward_time": 1.7150421142578125,
      "step": 7090
    },
    {
      "dynamics_loss_avg": 0.0790156863629818,
      "epoch": 6.761550903320312e-06,
      "step": 7090
    },
    {
      "action_loss_avg": 0.006570258922874927,
      "epoch": 6.761550903320312e-06,
      "step": 7090
    },
    {
      "epoch": 6.761550903320312e-06,
      "step": 7090,
      "training_step_time": 6.927080154418945
    },
    {
      "epoch": 6.762504577636719e-06,
      "model_forward_time": 1.7101242542266846,
      "step": 7091
    },
    {
      "epoch": 6.762504577636719e-06,
      "step": 7091,
      "training_step_time": 6.933502435684204
    },
    {
      "epoch": 6.763458251953125e-06,
      "model_forward_time": 1.708740234375,
      "step": 7092
    },
    {
      "epoch": 6.763458251953125e-06,
      "step": 7092,
      "training_step_time": 6.867040157318115
    },
    {
      "epoch": 6.764411926269531e-06,
      "model_forward_time": 1.7216932773590088,
      "step": 7093
    },
    {
      "epoch": 6.764411926269531e-06,
      "step": 7093,
      "training_step_time": 6.834181785583496
    },
    {
      "epoch": 6.765365600585938e-06,
      "model_forward_time": 1.7174692153930664,
      "step": 7094
    },
    {
      "epoch": 6.765365600585938e-06,
      "step": 7094,
      "training_step_time": 7.499879598617554
    },
    {
      "epoch": 6.7663192749023435e-06,
      "model_forward_time": 2.2639427185058594,
      "step": 7095
    },
    {
      "epoch": 6.7663192749023435e-06,
      "step": 7095,
      "training_step_time": 9.571552515029907
    },
    {
      "epoch": 6.76727294921875e-06,
      "model_forward_time": 1.7094969749450684,
      "step": 7096
    },
    {
      "epoch": 6.76727294921875e-06,
      "step": 7096,
      "training_step_time": 7.892390012741089
    },
    {
      "epoch": 6.768226623535157e-06,
      "model_forward_time": 1.7087821960449219,
      "step": 7097
    },
    {
      "epoch": 6.768226623535157e-06,
      "step": 7097,
      "training_step_time": 8.03400206565857
    },
    {
      "epoch": 6.769180297851562e-06,
      "model_forward_time": 1.71244215965271,
      "step": 7098
    },
    {
      "epoch": 6.769180297851562e-06,
      "step": 7098,
      "training_step_time": 7.690414190292358
    },
    {
      "epoch": 6.770133972167969e-06,
      "model_forward_time": 1.7160835266113281,
      "step": 7099
    },
    {
      "epoch": 6.770133972167969e-06,
      "step": 7099,
      "training_step_time": 6.958523273468018
    },
    {
      "epoch": 6.771087646484375e-06,
      "grad_norm": 0.3072334825992584,
      "learning_rate": 9.987959541084087e-06,
      "loss": 0.0713,
      "step": 7100
    },
    {
      "epoch": 6.771087646484375e-06,
      "model_forward_time": 1.7159333229064941,
      "step": 7100
    },
    {
      "dynamics_loss_avg": 0.07571024177595972,
      "epoch": 6.771087646484375e-06,
      "step": 7100
    },
    {
      "action_loss_avg": 0.007607351406477392,
      "epoch": 6.771087646484375e-06,
      "step": 7100
    },
    {
      "epoch": 6.771087646484375e-06,
      "step": 7100,
      "training_step_time": 6.969667196273804
    },
    {
      "epoch": 6.772041320800781e-06,
      "model_forward_time": 1.7167816162109375,
      "step": 7101
    },
    {
      "epoch": 6.772041320800781e-06,
      "step": 7101,
      "training_step_time": 6.986257076263428
    },
    {
      "epoch": 6.772994995117188e-06,
      "model_forward_time": 1.7075462341308594,
      "step": 7102
    },
    {
      "epoch": 6.772994995117188e-06,
      "step": 7102,
      "training_step_time": 6.952919006347656
    },
    {
      "epoch": 6.7739486694335935e-06,
      "model_forward_time": 1.714672327041626,
      "step": 7103
    },
    {
      "epoch": 6.7739486694335935e-06,
      "step": 7103,
      "training_step_time": 7.54790186882019
    },
    {
      "epoch": 6.77490234375e-06,
      "model_forward_time": 1.7448110580444336,
      "step": 7104
    },
    {
      "epoch": 6.77490234375e-06,
      "step": 7104,
      "training_step_time": 7.667121648788452
    },
    {
      "epoch": 6.775856018066407e-06,
      "model_forward_time": 2.2287371158599854,
      "step": 7105
    },
    {
      "epoch": 6.775856018066407e-06,
      "step": 7105,
      "training_step_time": 7.7848639488220215
    },
    {
      "epoch": 6.776809692382812e-06,
      "model_forward_time": 1.7153911590576172,
      "step": 7106
    },
    {
      "epoch": 6.776809692382812e-06,
      "step": 7106,
      "training_step_time": 7.952513694763184
    },
    {
      "epoch": 6.777763366699219e-06,
      "model_forward_time": 1.7102904319763184,
      "step": 7107
    },
    {
      "epoch": 6.777763366699219e-06,
      "step": 7107,
      "training_step_time": 7.627553462982178
    },
    {
      "epoch": 6.778717041015625e-06,
      "model_forward_time": 1.7142114639282227,
      "step": 7108
    },
    {
      "epoch": 6.778717041015625e-06,
      "step": 7108,
      "training_step_time": 6.890803098678589
    },
    {
      "epoch": 6.779670715332031e-06,
      "model_forward_time": 1.7302708625793457,
      "step": 7109
    },
    {
      "epoch": 6.779670715332031e-06,
      "step": 7109,
      "training_step_time": 6.864778280258179
    },
    {
      "epoch": 6.780624389648438e-06,
      "grad_norm": 0.2252437025308609,
      "learning_rate": 9.987844588746916e-06,
      "loss": 0.068,
      "step": 7110
    },
    {
      "epoch": 6.780624389648438e-06,
      "model_forward_time": 1.709291696548462,
      "step": 7110
    },
    {
      "dynamics_loss_avg": 0.08388704303652048,
      "epoch": 6.780624389648438e-06,
      "step": 7110
    },
    {
      "action_loss_avg": 0.007338292920030654,
      "epoch": 6.780624389648438e-06,
      "step": 7110
    },
    {
      "epoch": 6.780624389648438e-06,
      "step": 7110,
      "training_step_time": 6.885366439819336
    },
    {
      "epoch": 6.7815780639648435e-06,
      "model_forward_time": 1.7150263786315918,
      "step": 7111
    },
    {
      "epoch": 6.7815780639648435e-06,
      "step": 7111,
      "training_step_time": 6.907395124435425
    },
    {
      "epoch": 6.78253173828125e-06,
      "model_forward_time": 1.7045562267303467,
      "step": 7112
    },
    {
      "epoch": 6.78253173828125e-06,
      "step": 7112,
      "training_step_time": 7.5083043575286865
    },
    {
      "epoch": 6.783485412597657e-06,
      "model_forward_time": 1.7225422859191895,
      "step": 7113
    },
    {
      "epoch": 6.783485412597657e-06,
      "step": 7113,
      "training_step_time": 8.580385208129883
    },
    {
      "epoch": 6.784439086914062e-06,
      "model_forward_time": 2.244011878967285,
      "step": 7114
    },
    {
      "epoch": 6.784439086914062e-06,
      "step": 7114,
      "training_step_time": 7.682325601577759
    },
    {
      "epoch": 6.785392761230469e-06,
      "model_forward_time": 1.7125663757324219,
      "step": 7115
    },
    {
      "epoch": 6.785392761230469e-06,
      "step": 7115,
      "training_step_time": 8.83285903930664
    },
    {
      "epoch": 6.786346435546875e-06,
      "model_forward_time": 1.712111234664917,
      "step": 7116
    },
    {
      "epoch": 6.786346435546875e-06,
      "step": 7116,
      "training_step_time": 8.630625486373901
    },
    {
      "epoch": 6.787300109863281e-06,
      "model_forward_time": 1.7114510536193848,
      "step": 7117
    },
    {
      "epoch": 6.787300109863281e-06,
      "step": 7117,
      "training_step_time": 6.855120420455933
    },
    {
      "epoch": 6.788253784179688e-06,
      "model_forward_time": 1.7165412902832031,
      "step": 7118
    },
    {
      "epoch": 6.788253784179688e-06,
      "step": 7118,
      "training_step_time": 6.901411056518555
    },
    {
      "epoch": 6.7892074584960935e-06,
      "model_forward_time": 1.7118239402770996,
      "step": 7119
    },
    {
      "epoch": 6.7892074584960935e-06,
      "step": 7119,
      "training_step_time": 6.872669219970703
    },
    {
      "epoch": 6.7901611328125e-06,
      "grad_norm": 0.20356978476047516,
      "learning_rate": 9.987729090946558e-06,
      "loss": 0.0718,
      "step": 7120
    },
    {
      "epoch": 6.7901611328125e-06,
      "model_forward_time": 1.7134311199188232,
      "step": 7120
    },
    {
      "dynamics_loss_avg": 0.06482837460935116,
      "epoch": 6.7901611328125e-06,
      "step": 7120
    },
    {
      "action_loss_avg": 0.005836542462930083,
      "epoch": 6.7901611328125e-06,
      "step": 7120
    },
    {
      "epoch": 6.7901611328125e-06,
      "step": 7120,
      "training_step_time": 6.8708250522613525
    },
    {
      "epoch": 6.791114807128907e-06,
      "model_forward_time": 1.711970329284668,
      "step": 7121
    },
    {
      "epoch": 6.791114807128907e-06,
      "step": 7121,
      "training_step_time": 7.046203136444092
    },
    {
      "epoch": 6.792068481445312e-06,
      "model_forward_time": 1.7122175693511963,
      "step": 7122
    },
    {
      "epoch": 6.792068481445312e-06,
      "step": 7122,
      "training_step_time": 8.559562921524048
    },
    {
      "epoch": 6.793022155761719e-06,
      "model_forward_time": 2.2533223628997803,
      "step": 7123
    },
    {
      "epoch": 6.793022155761719e-06,
      "step": 7123,
      "training_step_time": 8.540796041488647
    },
    {
      "epoch": 6.793975830078125e-06,
      "model_forward_time": 1.705646276473999,
      "step": 7124
    },
    {
      "epoch": 6.793975830078125e-06,
      "step": 7124,
      "training_step_time": 9.496676921844482
    },
    {
      "epoch": 6.794929504394531e-06,
      "model_forward_time": 1.712054967880249,
      "step": 7125
    },
    {
      "epoch": 6.794929504394531e-06,
      "step": 7125,
      "training_step_time": 7.855393886566162
    },
    {
      "epoch": 6.795883178710938e-06,
      "model_forward_time": 1.710632085800171,
      "step": 7126
    },
    {
      "epoch": 6.795883178710938e-06,
      "step": 7126,
      "training_step_time": 7.6223132610321045
    },
    {
      "epoch": 6.7968368530273435e-06,
      "model_forward_time": 1.3451156616210938,
      "step": 7127
    },
    {
      "epoch": 6.7968368530273435e-06,
      "step": 7127,
      "training_step_time": 6.87261438369751
    },
    {
      "epoch": 6.79779052734375e-06,
      "model_forward_time": 1.7107925415039062,
      "step": 7128
    },
    {
      "epoch": 6.79779052734375e-06,
      "step": 7128,
      "training_step_time": 6.87746262550354
    },
    {
      "epoch": 6.798744201660156e-06,
      "model_forward_time": 1.7147583961486816,
      "step": 7129
    },
    {
      "epoch": 6.798744201660156e-06,
      "step": 7129,
      "training_step_time": 6.94636344909668
    },
    {
      "epoch": 6.799697875976562e-06,
      "grad_norm": 0.3750544786453247,
      "learning_rate": 9.987613047695647e-06,
      "loss": 0.0722,
      "step": 7130
    },
    {
      "epoch": 6.799697875976562e-06,
      "model_forward_time": 1.714803695678711,
      "step": 7130
    },
    {
      "dynamics_loss_avg": 0.05805173078551888,
      "epoch": 6.799697875976562e-06,
      "step": 7130
    },
    {
      "action_loss_avg": 0.0049660365562886,
      "epoch": 6.799697875976562e-06,
      "step": 7130
    },
    {
      "epoch": 6.799697875976562e-06,
      "step": 7130,
      "training_step_time": 6.918772459030151
    },
    {
      "epoch": 6.800651550292969e-06,
      "model_forward_time": 1.7132420539855957,
      "step": 7131
    },
    {
      "epoch": 6.800651550292969e-06,
      "step": 7131,
      "training_step_time": 9.093781471252441
    },
    {
      "epoch": 6.801605224609375e-06,
      "model_forward_time": 2.289703369140625,
      "step": 7132
    },
    {
      "epoch": 6.801605224609375e-06,
      "step": 7132,
      "training_step_time": 7.723707675933838
    },
    {
      "epoch": 6.802558898925781e-06,
      "model_forward_time": 1.7118875980377197,
      "step": 7133
    },
    {
      "epoch": 6.802558898925781e-06,
      "step": 7133,
      "training_step_time": 9.391830921173096
    },
    {
      "epoch": 6.803512573242188e-06,
      "model_forward_time": 1.706883430480957,
      "step": 7134
    },
    {
      "epoch": 6.803512573242188e-06,
      "step": 7134,
      "training_step_time": 8.564496278762817
    },
    {
      "epoch": 6.8044662475585935e-06,
      "model_forward_time": 1.7100393772125244,
      "step": 7135
    },
    {
      "epoch": 6.8044662475585935e-06,
      "step": 7135,
      "training_step_time": 7.554983139038086
    },
    {
      "epoch": 6.805419921875e-06,
      "model_forward_time": 1.7124722003936768,
      "step": 7136
    },
    {
      "epoch": 6.805419921875e-06,
      "step": 7136,
      "training_step_time": 6.839395999908447
    },
    {
      "epoch": 6.806373596191406e-06,
      "model_forward_time": 1.7106881141662598,
      "step": 7137
    },
    {
      "epoch": 6.806373596191406e-06,
      "step": 7137,
      "training_step_time": 6.861756801605225
    },
    {
      "epoch": 6.807327270507812e-06,
      "model_forward_time": 1.7160735130310059,
      "step": 7138
    },
    {
      "epoch": 6.807327270507812e-06,
      "step": 7138,
      "training_step_time": 6.871522665023804
    },
    {
      "epoch": 6.808280944824219e-06,
      "model_forward_time": 1.7136108875274658,
      "step": 7139
    },
    {
      "epoch": 6.808280944824219e-06,
      "step": 7139,
      "training_step_time": 6.842881202697754
    },
    {
      "epoch": 6.809234619140625e-06,
      "grad_norm": 0.2828623652458191,
      "learning_rate": 9.98749645900687e-06,
      "loss": 0.0717,
      "step": 7140
    },
    {
      "epoch": 6.809234619140625e-06,
      "model_forward_time": 1.717618465423584,
      "step": 7140
    },
    {
      "dynamics_loss_avg": 0.05157118681818247,
      "epoch": 6.809234619140625e-06,
      "step": 7140
    },
    {
      "action_loss_avg": 0.004512079909909516,
      "epoch": 6.809234619140625e-06,
      "step": 7140
    },
    {
      "epoch": 6.809234619140625e-06,
      "step": 7140,
      "training_step_time": 8.447323322296143
    },
    {
      "epoch": 6.810188293457031e-06,
      "model_forward_time": 2.267488718032837,
      "step": 7141
    },
    {
      "epoch": 6.810188293457031e-06,
      "step": 7141,
      "training_step_time": 9.46895170211792
    },
    {
      "epoch": 6.811141967773438e-06,
      "model_forward_time": 1.696697473526001,
      "step": 7142
    },
    {
      "epoch": 6.811141967773438e-06,
      "step": 7142,
      "training_step_time": 10.126296281814575
    },
    {
      "epoch": 6.8120956420898436e-06,
      "model_forward_time": 1.7154605388641357,
      "step": 7143
    },
    {
      "epoch": 6.8120956420898436e-06,
      "step": 7143,
      "training_step_time": 8.532827854156494
    },
    {
      "epoch": 6.81304931640625e-06,
      "model_forward_time": 1.723442792892456,
      "step": 7144
    },
    {
      "epoch": 6.81304931640625e-06,
      "step": 7144,
      "training_step_time": 7.542874813079834
    },
    {
      "epoch": 6.814002990722656e-06,
      "model_forward_time": 1.7192270755767822,
      "step": 7145
    },
    {
      "epoch": 6.814002990722656e-06,
      "step": 7145,
      "training_step_time": 6.857340574264526
    },
    {
      "epoch": 6.814956665039062e-06,
      "model_forward_time": 1.7198460102081299,
      "step": 7146
    },
    {
      "epoch": 6.814956665039062e-06,
      "step": 7146,
      "training_step_time": 6.8703718185424805
    },
    {
      "epoch": 6.815910339355469e-06,
      "model_forward_time": 1.7182683944702148,
      "step": 7147
    },
    {
      "epoch": 6.815910339355469e-06,
      "step": 7147,
      "training_step_time": 6.872515439987183
    },
    {
      "epoch": 6.816864013671875e-06,
      "model_forward_time": 1.7210338115692139,
      "step": 7148
    },
    {
      "epoch": 6.816864013671875e-06,
      "step": 7148,
      "training_step_time": 6.847261667251587
    },
    {
      "epoch": 6.817817687988281e-06,
      "model_forward_time": 1.7236993312835693,
      "step": 7149
    },
    {
      "epoch": 6.817817687988281e-06,
      "step": 7149,
      "training_step_time": 8.475880861282349
    },
    {
      "epoch": 6.818771362304688e-06,
      "grad_norm": 0.22065266966819763,
      "learning_rate": 9.987379324892983e-06,
      "loss": 0.0708,
      "step": 7150
    },
    {
      "epoch": 6.818771362304688e-06,
      "model_forward_time": 1.7473583221435547,
      "step": 7150
    },
    {
      "dynamics_loss_avg": 0.06440076120197773,
      "epoch": 6.818771362304688e-06,
      "step": 7150
    },
    {
      "action_loss_avg": 0.00657972190529108,
      "epoch": 6.818771362304688e-06,
      "step": 7150
    },
    {
      "epoch": 6.818771362304688e-06,
      "step": 7150,
      "training_step_time": 8.869069814682007
    },
    {
      "epoch": 6.8197250366210936e-06,
      "model_forward_time": 2.2597544193267822,
      "step": 7151
    },
    {
      "epoch": 6.8197250366210936e-06,
      "step": 7151,
      "training_step_time": 9.118450403213501
    },
    {
      "epoch": 6.8206787109375e-06,
      "model_forward_time": 1.3428997993469238,
      "step": 7152
    },
    {
      "epoch": 6.8206787109375e-06,
      "step": 7152,
      "training_step_time": 7.925363540649414
    },
    {
      "epoch": 6.821632385253906e-06,
      "model_forward_time": 1.720005989074707,
      "step": 7153
    },
    {
      "epoch": 6.821632385253906e-06,
      "step": 7153,
      "training_step_time": 8.248673439025879
    },
    {
      "epoch": 6.822586059570312e-06,
      "model_forward_time": 1.724848985671997,
      "step": 7154
    },
    {
      "epoch": 6.822586059570312e-06,
      "step": 7154,
      "training_step_time": 6.8350841999053955
    },
    {
      "epoch": 6.823539733886719e-06,
      "model_forward_time": 1.7210893630981445,
      "step": 7155
    },
    {
      "epoch": 6.823539733886719e-06,
      "step": 7155,
      "training_step_time": 6.828583002090454
    },
    {
      "epoch": 6.824493408203125e-06,
      "model_forward_time": 1.720738172531128,
      "step": 7156
    },
    {
      "epoch": 6.824493408203125e-06,
      "step": 7156,
      "training_step_time": 6.83223557472229
    },
    {
      "epoch": 6.825447082519531e-06,
      "model_forward_time": 1.723139762878418,
      "step": 7157
    },
    {
      "epoch": 6.825447082519531e-06,
      "step": 7157,
      "training_step_time": 6.8654091358184814
    },
    {
      "epoch": 6.826400756835938e-06,
      "model_forward_time": 1.721229076385498,
      "step": 7158
    },
    {
      "epoch": 6.826400756835938e-06,
      "step": 7158,
      "training_step_time": 7.442531108856201
    },
    {
      "epoch": 6.8273544311523436e-06,
      "model_forward_time": 1.7357096672058105,
      "step": 7159
    },
    {
      "epoch": 6.8273544311523436e-06,
      "step": 7159,
      "training_step_time": 7.792952060699463
    },
    {
      "epoch": 6.82830810546875e-06,
      "grad_norm": 0.19720157980918884,
      "learning_rate": 9.987261645366788e-06,
      "loss": 0.0689,
      "step": 7160
    },
    {
      "epoch": 6.82830810546875e-06,
      "model_forward_time": 2.2441353797912598,
      "step": 7160
    },
    {
      "dynamics_loss_avg": 0.06732130981981754,
      "epoch": 6.82830810546875e-06,
      "step": 7160
    },
    {
      "action_loss_avg": 0.005941586778499186,
      "epoch": 6.82830810546875e-06,
      "step": 7160
    },
    {
      "epoch": 6.82830810546875e-06,
      "step": 7160,
      "training_step_time": 10.39553427696228
    },
    {
      "epoch": 6.829261779785156e-06,
      "model_forward_time": 1.7163617610931396,
      "step": 7161
    },
    {
      "epoch": 6.829261779785156e-06,
      "step": 7161,
      "training_step_time": 8.856763124465942
    },
    {
      "epoch": 6.8302154541015624e-06,
      "model_forward_time": 1.7191998958587646,
      "step": 7162
    },
    {
      "epoch": 6.8302154541015624e-06,
      "step": 7162,
      "training_step_time": 7.755919456481934
    },
    {
      "epoch": 6.831169128417969e-06,
      "model_forward_time": 1.721921682357788,
      "step": 7163
    },
    {
      "epoch": 6.831169128417969e-06,
      "step": 7163,
      "training_step_time": 7.496834754943848
    },
    {
      "epoch": 6.832122802734375e-06,
      "model_forward_time": 1.7213222980499268,
      "step": 7164
    },
    {
      "epoch": 6.832122802734375e-06,
      "step": 7164,
      "training_step_time": 6.858753442764282
    },
    {
      "epoch": 6.833076477050781e-06,
      "model_forward_time": 1.7359182834625244,
      "step": 7165
    },
    {
      "epoch": 6.833076477050781e-06,
      "step": 7165,
      "training_step_time": 6.884880065917969
    },
    {
      "epoch": 6.834030151367188e-06,
      "model_forward_time": 1.7204675674438477,
      "step": 7166
    },
    {
      "epoch": 6.834030151367188e-06,
      "step": 7166,
      "training_step_time": 6.856019973754883
    },
    {
      "epoch": 6.8349838256835936e-06,
      "model_forward_time": 1.7178666591644287,
      "step": 7167
    },
    {
      "epoch": 6.8349838256835936e-06,
      "step": 7167,
      "training_step_time": 8.394174337387085
    },
    {
      "epoch": 6.8359375e-06,
      "model_forward_time": 1.7217183113098145,
      "step": 7168
    },
    {
      "epoch": 6.8359375e-06,
      "step": 7168,
      "training_step_time": 7.787570238113403
    },
    {
      "epoch": 6.836891174316406e-06,
      "model_forward_time": 2.2819323539733887,
      "step": 7169
    },
    {
      "epoch": 6.836891174316406e-06,
      "step": 7169,
      "training_step_time": 9.334699869155884
    },
    {
      "epoch": 6.8378448486328124e-06,
      "grad_norm": 0.20906759798526764,
      "learning_rate": 9.987143420441158e-06,
      "loss": 0.0699,
      "step": 7170
    },
    {
      "epoch": 6.8378448486328124e-06,
      "model_forward_time": 1.718674659729004,
      "step": 7170
    },
    {
      "dynamics_loss_avg": 0.05709693506360054,
      "epoch": 6.8378448486328124e-06,
      "step": 7170
    },
    {
      "action_loss_avg": 0.006035194732248783,
      "epoch": 6.8378448486328124e-06,
      "step": 7170
    },
    {
      "epoch": 6.8378448486328124e-06,
      "step": 7170,
      "training_step_time": 11.09057903289795
    },
    {
      "epoch": 6.838798522949219e-06,
      "model_forward_time": 1.73158860206604,
      "step": 7171
    },
    {
      "epoch": 6.838798522949219e-06,
      "step": 7171,
      "training_step_time": 8.317647457122803
    },
    {
      "epoch": 6.839752197265625e-06,
      "model_forward_time": 1.726449966430664,
      "step": 7172
    },
    {
      "epoch": 6.839752197265625e-06,
      "step": 7172,
      "training_step_time": 7.5274341106414795
    },
    {
      "epoch": 6.840705871582031e-06,
      "model_forward_time": 1.720956802368164,
      "step": 7173
    },
    {
      "epoch": 6.840705871582031e-06,
      "step": 7173,
      "training_step_time": 6.851298809051514
    },
    {
      "epoch": 6.841659545898438e-06,
      "model_forward_time": 1.7323250770568848,
      "step": 7174
    },
    {
      "epoch": 6.841659545898438e-06,
      "step": 7174,
      "training_step_time": 6.818564176559448
    },
    {
      "epoch": 6.842613220214844e-06,
      "model_forward_time": 1.7287664413452148,
      "step": 7175
    },
    {
      "epoch": 6.842613220214844e-06,
      "step": 7175,
      "training_step_time": 6.82100772857666
    },
    {
      "epoch": 6.84356689453125e-06,
      "model_forward_time": 1.7241034507751465,
      "step": 7176
    },
    {
      "epoch": 6.84356689453125e-06,
      "step": 7176,
      "training_step_time": 7.390455007553101
    },
    {
      "epoch": 6.844520568847656e-06,
      "model_forward_time": 1.7302424907684326,
      "step": 7177
    },
    {
      "epoch": 6.844520568847656e-06,
      "step": 7177,
      "training_step_time": 8.22753095626831
    },
    {
      "epoch": 6.8454742431640624e-06,
      "model_forward_time": 2.287641763687134,
      "step": 7178
    },
    {
      "epoch": 6.8454742431640624e-06,
      "step": 7178,
      "training_step_time": 8.783485651016235
    },
    {
      "epoch": 6.846427917480469e-06,
      "model_forward_time": 1.7176382541656494,
      "step": 7179
    },
    {
      "epoch": 6.846427917480469e-06,
      "step": 7179,
      "training_step_time": 10.585256338119507
    },
    {
      "epoch": 6.847381591796875e-06,
      "grad_norm": 0.25185921788215637,
      "learning_rate": 9.987024650129022e-06,
      "loss": 0.0668,
      "step": 7180
    },
    {
      "epoch": 6.847381591796875e-06,
      "model_forward_time": 1.7291734218597412,
      "step": 7180
    },
    {
      "dynamics_loss_avg": 0.06245938297361135,
      "epoch": 6.847381591796875e-06,
      "step": 7180
    },
    {
      "action_loss_avg": 0.006266781967133284,
      "epoch": 6.847381591796875e-06,
      "step": 7180
    },
    {
      "epoch": 6.847381591796875e-06,
      "step": 7180,
      "training_step_time": 8.80395793914795
    },
    {
      "epoch": 6.848335266113281e-06,
      "model_forward_time": 1.7247920036315918,
      "step": 7181
    },
    {
      "epoch": 6.848335266113281e-06,
      "step": 7181,
      "training_step_time": 7.589790105819702
    },
    {
      "epoch": 6.849288940429688e-06,
      "model_forward_time": 1.7375009059906006,
      "step": 7182
    },
    {
      "epoch": 6.849288940429688e-06,
      "step": 7182,
      "training_step_time": 6.810762643814087
    },
    {
      "epoch": 6.850242614746094e-06,
      "model_forward_time": 1.724327564239502,
      "step": 7183
    },
    {
      "epoch": 6.850242614746094e-06,
      "step": 7183,
      "training_step_time": 6.845211029052734
    },
    {
      "epoch": 6.8511962890625e-06,
      "model_forward_time": 1.7331535816192627,
      "step": 7184
    },
    {
      "epoch": 6.8511962890625e-06,
      "step": 7184,
      "training_step_time": 6.866867780685425
    },
    {
      "epoch": 6.852149963378906e-06,
      "model_forward_time": 1.7289373874664307,
      "step": 7185
    },
    {
      "epoch": 6.852149963378906e-06,
      "step": 7185,
      "training_step_time": 7.3494322299957275
    },
    {
      "epoch": 6.8531036376953124e-06,
      "model_forward_time": 1.729811191558838,
      "step": 7186
    },
    {
      "epoch": 6.8531036376953124e-06,
      "step": 7186,
      "training_step_time": 7.625569581985474
    },
    {
      "epoch": 6.854057312011719e-06,
      "model_forward_time": 2.2873411178588867,
      "step": 7187
    },
    {
      "epoch": 6.854057312011719e-06,
      "step": 7187,
      "training_step_time": 8.337070226669312
    },
    {
      "epoch": 6.855010986328125e-06,
      "model_forward_time": 1.733098030090332,
      "step": 7188
    },
    {
      "epoch": 6.855010986328125e-06,
      "step": 7188,
      "training_step_time": 10.408298254013062
    },
    {
      "epoch": 6.855964660644531e-06,
      "model_forward_time": 1.7207703590393066,
      "step": 7189
    },
    {
      "epoch": 6.855964660644531e-06,
      "step": 7189,
      "training_step_time": 8.877368688583374
    },
    {
      "epoch": 6.856918334960938e-06,
      "grad_norm": 0.21732580661773682,
      "learning_rate": 9.98690533444337e-06,
      "loss": 0.0699,
      "step": 7190
    },
    {
      "epoch": 6.856918334960938e-06,
      "model_forward_time": 1.72440505027771,
      "step": 7190
    },
    {
      "dynamics_loss_avg": 0.061896799318492415,
      "epoch": 6.856918334960938e-06,
      "step": 7190
    },
    {
      "action_loss_avg": 0.0062539869453758,
      "epoch": 6.856918334960938e-06,
      "step": 7190
    },
    {
      "epoch": 6.856918334960938e-06,
      "step": 7190,
      "training_step_time": 7.792129755020142
    },
    {
      "epoch": 6.857872009277344e-06,
      "model_forward_time": 1.7348401546478271,
      "step": 7191
    },
    {
      "epoch": 6.857872009277344e-06,
      "step": 7191,
      "training_step_time": 7.236888885498047
    },
    {
      "epoch": 6.85882568359375e-06,
      "model_forward_time": 1.7341992855072021,
      "step": 7192
    },
    {
      "epoch": 6.85882568359375e-06,
      "step": 7192,
      "training_step_time": 6.814974784851074
    },
    {
      "epoch": 6.859779357910156e-06,
      "model_forward_time": 1.7294032573699951,
      "step": 7193
    },
    {
      "epoch": 6.859779357910156e-06,
      "step": 7193,
      "training_step_time": 6.838759422302246
    },
    {
      "epoch": 6.8607330322265625e-06,
      "model_forward_time": 1.7300000190734863,
      "step": 7194
    },
    {
      "epoch": 6.8607330322265625e-06,
      "step": 7194,
      "training_step_time": 7.373610019683838
    },
    {
      "epoch": 6.861686706542969e-06,
      "model_forward_time": 1.73356294631958,
      "step": 7195
    },
    {
      "epoch": 6.861686706542969e-06,
      "step": 7195,
      "training_step_time": 7.624644994735718
    },
    {
      "epoch": 6.862640380859375e-06,
      "model_forward_time": 1.745100736618042,
      "step": 7196
    },
    {
      "epoch": 6.862640380859375e-06,
      "step": 7196,
      "training_step_time": 10.147440433502197
    },
    {
      "epoch": 6.863594055175781e-06,
      "model_forward_time": 1.7147014141082764,
      "step": 7197
    },
    {
      "epoch": 6.863594055175781e-06,
      "step": 7197,
      "training_step_time": 9.794887065887451
    },
    {
      "epoch": 6.864547729492188e-06,
      "model_forward_time": 1.7159438133239746,
      "step": 7198
    },
    {
      "epoch": 6.864547729492188e-06,
      "step": 7198,
      "training_step_time": 8.419016122817993
    },
    {
      "epoch": 6.865501403808594e-06,
      "model_forward_time": 1.7261905670166016,
      "step": 7199
    },
    {
      "epoch": 6.865501403808594e-06,
      "step": 7199,
      "training_step_time": 9.086875915527344
    },
    {
      "epoch": 6.866455078125e-06,
      "grad_norm": 0.21353565156459808,
      "learning_rate": 9.986785473397246e-06,
      "loss": 0.0716,
      "step": 7200
    },
    {
      "epoch": 6.866455078125e-06,
      "model_forward_time": 1.7313923835754395,
      "step": 7200
    },
    {
      "dynamics_loss_avg": 0.0641827940940857,
      "epoch": 6.866455078125e-06,
      "step": 7200
    },
    {
      "action_loss_avg": 0.007339728251099587,
      "epoch": 6.866455078125e-06,
      "step": 7200
    },
    {
      "epoch": 6.866455078125e-06,
      "step": 7200,
      "training_step_time": 7.285402536392212
    },
    {
      "epoch": 6.867408752441406e-06,
      "model_forward_time": 1.7257800102233887,
      "step": 7201
    },
    {
      "epoch": 6.867408752441406e-06,
      "step": 7201,
      "training_step_time": 6.864964008331299
    },
    {
      "epoch": 6.8683624267578125e-06,
      "model_forward_time": 1.7229154109954834,
      "step": 7202
    },
    {
      "epoch": 6.8683624267578125e-06,
      "step": 7202,
      "training_step_time": 6.8585734367370605
    },
    {
      "epoch": 6.869316101074219e-06,
      "model_forward_time": 1.7235431671142578,
      "step": 7203
    },
    {
      "epoch": 6.869316101074219e-06,
      "step": 7203,
      "training_step_time": 7.328353643417358
    },
    {
      "epoch": 6.870269775390625e-06,
      "model_forward_time": 1.7170443534851074,
      "step": 7204
    },
    {
      "epoch": 6.870269775390625e-06,
      "step": 7204,
      "training_step_time": 7.656314849853516
    },
    {
      "epoch": 6.871223449707031e-06,
      "model_forward_time": 1.7238070964813232,
      "step": 7205
    },
    {
      "epoch": 6.871223449707031e-06,
      "step": 7205,
      "training_step_time": 8.679956674575806
    },
    {
      "epoch": 6.872177124023438e-06,
      "model_forward_time": 2.260462999343872,
      "step": 7206
    },
    {
      "epoch": 6.872177124023438e-06,
      "step": 7206,
      "training_step_time": 9.800223112106323
    },
    {
      "epoch": 6.873130798339844e-06,
      "model_forward_time": 1.7244603633880615,
      "step": 7207
    },
    {
      "epoch": 6.873130798339844e-06,
      "step": 7207,
      "training_step_time": 9.558182716369629
    },
    {
      "epoch": 6.87408447265625e-06,
      "model_forward_time": 1.7337665557861328,
      "step": 7208
    },
    {
      "epoch": 6.87408447265625e-06,
      "step": 7208,
      "training_step_time": 7.602768182754517
    },
    {
      "epoch": 6.875038146972656e-06,
      "model_forward_time": 1.7224950790405273,
      "step": 7209
    },
    {
      "epoch": 6.875038146972656e-06,
      "step": 7209,
      "training_step_time": 7.717767000198364
    },
    {
      "epoch": 6.8759918212890625e-06,
      "grad_norm": 0.24106977880001068,
      "learning_rate": 9.98666506700376e-06,
      "loss": 0.0673,
      "step": 7210
    },
    {
      "epoch": 6.8759918212890625e-06,
      "model_forward_time": 1.7291138172149658,
      "step": 7210
    },
    {
      "dynamics_loss_avg": 0.05708850026130676,
      "epoch": 6.8759918212890625e-06,
      "step": 7210
    },
    {
      "action_loss_avg": 0.004816740192472935,
      "epoch": 6.8759918212890625e-06,
      "step": 7210
    },
    {
      "epoch": 6.8759918212890625e-06,
      "step": 7210,
      "training_step_time": 6.764706134796143
    },
    {
      "epoch": 6.876945495605469e-06,
      "model_forward_time": 1.7219774723052979,
      "step": 7211
    },
    {
      "epoch": 6.876945495605469e-06,
      "step": 7211,
      "training_step_time": 6.807173013687134
    },
    {
      "epoch": 6.877899169921875e-06,
      "model_forward_time": 1.723001480102539,
      "step": 7212
    },
    {
      "epoch": 6.877899169921875e-06,
      "step": 7212,
      "training_step_time": 7.364088535308838
    },
    {
      "epoch": 6.878852844238281e-06,
      "model_forward_time": 1.7232749462127686,
      "step": 7213
    },
    {
      "epoch": 6.878852844238281e-06,
      "step": 7213,
      "training_step_time": 7.626094579696655
    },
    {
      "epoch": 6.879806518554688e-06,
      "model_forward_time": 1.725083827972412,
      "step": 7214
    },
    {
      "epoch": 6.879806518554688e-06,
      "step": 7214,
      "training_step_time": 7.765493631362915
    },
    {
      "epoch": 6.880760192871094e-06,
      "model_forward_time": 2.304952383041382,
      "step": 7215
    },
    {
      "epoch": 6.880760192871094e-06,
      "step": 7215,
      "training_step_time": 10.607057332992554
    },
    {
      "epoch": 6.8817138671875e-06,
      "model_forward_time": 1.7189714908599854,
      "step": 7216
    },
    {
      "epoch": 6.8817138671875e-06,
      "step": 7216,
      "training_step_time": 8.942561626434326
    },
    {
      "epoch": 6.882667541503906e-06,
      "model_forward_time": 1.7249557971954346,
      "step": 7217
    },
    {
      "epoch": 6.882667541503906e-06,
      "step": 7217,
      "training_step_time": 8.866466522216797
    },
    {
      "epoch": 6.8836212158203125e-06,
      "model_forward_time": 1.715947151184082,
      "step": 7218
    },
    {
      "epoch": 6.8836212158203125e-06,
      "step": 7218,
      "training_step_time": 7.71550989151001
    },
    {
      "epoch": 6.884574890136719e-06,
      "model_forward_time": 1.72568941116333,
      "step": 7219
    },
    {
      "epoch": 6.884574890136719e-06,
      "step": 7219,
      "training_step_time": 6.811519145965576
    },
    {
      "epoch": 6.885528564453125e-06,
      "grad_norm": 0.24765357375144958,
      "learning_rate": 9.986544115276082e-06,
      "loss": 0.0669,
      "step": 7220
    },
    {
      "epoch": 6.885528564453125e-06,
      "model_forward_time": 1.7403554916381836,
      "step": 7220
    },
    {
      "dynamics_loss_avg": 0.06376529447734355,
      "epoch": 6.885528564453125e-06,
      "step": 7220
    },
    {
      "action_loss_avg": 0.005680274171754718,
      "epoch": 6.885528564453125e-06,
      "step": 7220
    },
    {
      "epoch": 6.885528564453125e-06,
      "step": 7220,
      "training_step_time": 6.8550403118133545
    },
    {
      "epoch": 6.886482238769531e-06,
      "model_forward_time": 1.724179983139038,
      "step": 7221
    },
    {
      "epoch": 6.886482238769531e-06,
      "step": 7221,
      "training_step_time": 7.308882713317871
    },
    {
      "epoch": 6.887435913085938e-06,
      "model_forward_time": 1.7226669788360596,
      "step": 7222
    },
    {
      "epoch": 6.887435913085938e-06,
      "step": 7222,
      "training_step_time": 7.618488073348999
    },
    {
      "epoch": 6.888389587402344e-06,
      "model_forward_time": 1.3566529750823975,
      "step": 7223
    },
    {
      "epoch": 6.888389587402344e-06,
      "step": 7223,
      "training_step_time": 8.267680883407593
    },
    {
      "epoch": 6.88934326171875e-06,
      "model_forward_time": 2.31174898147583,
      "step": 7224
    },
    {
      "epoch": 6.88934326171875e-06,
      "step": 7224,
      "training_step_time": 8.32577657699585
    },
    {
      "epoch": 6.890296936035156e-06,
      "model_forward_time": 1.7214009761810303,
      "step": 7225
    },
    {
      "epoch": 6.890296936035156e-06,
      "step": 7225,
      "training_step_time": 9.14683485031128
    },
    {
      "epoch": 6.8912506103515625e-06,
      "model_forward_time": 1.7148115634918213,
      "step": 7226
    },
    {
      "epoch": 6.8912506103515625e-06,
      "step": 7226,
      "training_step_time": 7.616922616958618
    },
    {
      "epoch": 6.892204284667969e-06,
      "model_forward_time": 1.7259817123413086,
      "step": 7227
    },
    {
      "epoch": 6.892204284667969e-06,
      "step": 7227,
      "training_step_time": 8.541027545928955
    },
    {
      "epoch": 6.893157958984375e-06,
      "model_forward_time": 1.7251863479614258,
      "step": 7228
    },
    {
      "epoch": 6.893157958984375e-06,
      "step": 7228,
      "training_step_time": 7.1602208614349365
    },
    {
      "epoch": 6.894111633300781e-06,
      "model_forward_time": 1.7230684757232666,
      "step": 7229
    },
    {
      "epoch": 6.894111633300781e-06,
      "step": 7229,
      "training_step_time": 6.735125541687012
    },
    {
      "epoch": 6.895065307617188e-06,
      "grad_norm": 0.36039701104164124,
      "learning_rate": 9.986422618227434e-06,
      "loss": 0.073,
      "step": 7230
    },
    {
      "epoch": 6.895065307617188e-06,
      "model_forward_time": 1.720914363861084,
      "step": 7230
    },
    {
      "dynamics_loss_avg": 0.0625811342149973,
      "epoch": 6.895065307617188e-06,
      "step": 7230
    },
    {
      "action_loss_avg": 0.0050829153042286634,
      "epoch": 6.895065307617188e-06,
      "step": 7230
    },
    {
      "epoch": 6.895065307617188e-06,
      "step": 7230,
      "training_step_time": 7.622626066207886
    },
    {
      "epoch": 6.896018981933594e-06,
      "model_forward_time": 1.7283177375793457,
      "step": 7231
    },
    {
      "epoch": 6.896018981933594e-06,
      "step": 7231,
      "training_step_time": 7.624525547027588
    },
    {
      "epoch": 6.89697265625e-06,
      "model_forward_time": 1.7186760902404785,
      "step": 7232
    },
    {
      "epoch": 6.89697265625e-06,
      "step": 7232,
      "training_step_time": 7.73054051399231
    },
    {
      "epoch": 6.897926330566406e-06,
      "model_forward_time": 2.3126299381256104,
      "step": 7233
    },
    {
      "epoch": 6.897926330566406e-06,
      "step": 7233,
      "training_step_time": 9.650856018066406
    },
    {
      "epoch": 6.8988800048828125e-06,
      "model_forward_time": 1.7216408252716064,
      "step": 7234
    },
    {
      "epoch": 6.8988800048828125e-06,
      "step": 7234,
      "training_step_time": 7.823099613189697
    },
    {
      "epoch": 6.899833679199219e-06,
      "model_forward_time": 1.7365872859954834,
      "step": 7235
    },
    {
      "epoch": 6.899833679199219e-06,
      "step": 7235,
      "training_step_time": 7.882968425750732
    },
    {
      "epoch": 6.900787353515625e-06,
      "model_forward_time": 1.7219228744506836,
      "step": 7236
    },
    {
      "epoch": 6.900787353515625e-06,
      "step": 7236,
      "training_step_time": 8.66935420036316
    },
    {
      "epoch": 6.901741027832031e-06,
      "model_forward_time": 1.7242331504821777,
      "step": 7237
    },
    {
      "epoch": 6.901741027832031e-06,
      "step": 7237,
      "training_step_time": 7.2522218227386475
    },
    {
      "epoch": 6.902694702148438e-06,
      "model_forward_time": 1.721714973449707,
      "step": 7238
    },
    {
      "epoch": 6.902694702148438e-06,
      "step": 7238,
      "training_step_time": 6.839997053146362
    },
    {
      "epoch": 6.903648376464844e-06,
      "model_forward_time": 1.7196383476257324,
      "step": 7239
    },
    {
      "epoch": 6.903648376464844e-06,
      "step": 7239,
      "training_step_time": 6.811079978942871
    },
    {
      "epoch": 6.90460205078125e-06,
      "grad_norm": 0.25655779242515564,
      "learning_rate": 9.986300575871107e-06,
      "loss": 0.0725,
      "step": 7240
    },
    {
      "epoch": 6.90460205078125e-06,
      "model_forward_time": 1.7228429317474365,
      "step": 7240
    },
    {
      "dynamics_loss_avg": 0.05715606138110161,
      "epoch": 6.90460205078125e-06,
      "step": 7240
    },
    {
      "action_loss_avg": 0.004955300560686738,
      "epoch": 6.90460205078125e-06,
      "step": 7240
    },
    {
      "epoch": 6.90460205078125e-06,
      "step": 7240,
      "training_step_time": 7.55030083656311
    },
    {
      "epoch": 6.905555725097656e-06,
      "model_forward_time": 1.7189338207244873,
      "step": 7241
    },
    {
      "epoch": 6.905555725097656e-06,
      "step": 7241,
      "training_step_time": 8.372157096862793
    },
    {
      "epoch": 6.9065093994140625e-06,
      "model_forward_time": 1.7187190055847168,
      "step": 7242
    },
    {
      "epoch": 6.9065093994140625e-06,
      "step": 7242,
      "training_step_time": 8.251654386520386
    },
    {
      "epoch": 6.907463073730469e-06,
      "model_forward_time": 1.7785000801086426,
      "step": 7243
    },
    {
      "epoch": 6.907463073730469e-06,
      "step": 7243,
      "training_step_time": 9.27010989189148
    },
    {
      "epoch": 6.908416748046875e-06,
      "model_forward_time": 1.7140626907348633,
      "step": 7244
    },
    {
      "epoch": 6.908416748046875e-06,
      "step": 7244,
      "training_step_time": 8.459613800048828
    },
    {
      "epoch": 6.909370422363281e-06,
      "model_forward_time": 1.7173957824707031,
      "step": 7245
    },
    {
      "epoch": 6.909370422363281e-06,
      "step": 7245,
      "training_step_time": 7.747661590576172
    },
    {
      "epoch": 6.910324096679688e-06,
      "model_forward_time": 1.722815752029419,
      "step": 7246
    },
    {
      "epoch": 6.910324096679688e-06,
      "step": 7246,
      "training_step_time": 7.538606405258179
    },
    {
      "epoch": 6.911277770996094e-06,
      "model_forward_time": 1.7250304222106934,
      "step": 7247
    },
    {
      "epoch": 6.911277770996094e-06,
      "step": 7247,
      "training_step_time": 6.810965538024902
    },
    {
      "epoch": 6.9122314453125e-06,
      "model_forward_time": 1.730987310409546,
      "step": 7248
    },
    {
      "epoch": 6.9122314453125e-06,
      "step": 7248,
      "training_step_time": 6.839562654495239
    },
    {
      "epoch": 6.913185119628906e-06,
      "model_forward_time": 1.7216594219207764,
      "step": 7249
    },
    {
      "epoch": 6.913185119628906e-06,
      "step": 7249,
      "training_step_time": 7.582779169082642
    },
    {
      "epoch": 6.9141387939453125e-06,
      "grad_norm": 0.39154481887817383,
      "learning_rate": 9.986177988220444e-06,
      "loss": 0.0733,
      "step": 7250
    },
    {
      "epoch": 6.9141387939453125e-06,
      "model_forward_time": 1.722383975982666,
      "step": 7250
    },
    {
      "dynamics_loss_avg": 0.06339205447584391,
      "epoch": 6.9141387939453125e-06,
      "step": 7250
    },
    {
      "action_loss_avg": 0.007059344428125769,
      "epoch": 6.9141387939453125e-06,
      "step": 7250
    },
    {
      "epoch": 6.9141387939453125e-06,
      "step": 7250,
      "training_step_time": 7.709920167922974
    },
    {
      "epoch": 6.915092468261719e-06,
      "model_forward_time": 1.727447509765625,
      "step": 7251
    },
    {
      "epoch": 6.915092468261719e-06,
      "step": 7251,
      "training_step_time": 9.425797939300537
    },
    {
      "epoch": 6.916046142578125e-06,
      "model_forward_time": 2.3199031352996826,
      "step": 7252
    },
    {
      "epoch": 6.916046142578125e-06,
      "step": 7252,
      "training_step_time": 8.819460153579712
    },
    {
      "epoch": 6.916999816894531e-06,
      "model_forward_time": 1.7210302352905273,
      "step": 7253
    },
    {
      "epoch": 6.916999816894531e-06,
      "step": 7253,
      "training_step_time": 7.910835027694702
    },
    {
      "epoch": 6.917953491210937e-06,
      "model_forward_time": 1.7082054615020752,
      "step": 7254
    },
    {
      "epoch": 6.917953491210937e-06,
      "step": 7254,
      "training_step_time": 7.837352275848389
    },
    {
      "epoch": 6.918907165527344e-06,
      "model_forward_time": 1.722398281097412,
      "step": 7255
    },
    {
      "epoch": 6.918907165527344e-06,
      "step": 7255,
      "training_step_time": 7.76416802406311
    },
    {
      "epoch": 6.91986083984375e-06,
      "model_forward_time": 1.7328910827636719,
      "step": 7256
    },
    {
      "epoch": 6.91986083984375e-06,
      "step": 7256,
      "training_step_time": 7.2350993156433105
    },
    {
      "epoch": 6.920814514160156e-06,
      "model_forward_time": 1.7200493812561035,
      "step": 7257
    },
    {
      "epoch": 6.920814514160156e-06,
      "step": 7257,
      "training_step_time": 6.887665510177612
    },
    {
      "epoch": 6.9217681884765625e-06,
      "model_forward_time": 1.7258386611938477,
      "step": 7258
    },
    {
      "epoch": 6.9217681884765625e-06,
      "step": 7258,
      "training_step_time": 8.401838541030884
    },
    {
      "epoch": 6.922721862792969e-06,
      "model_forward_time": 1.7216930389404297,
      "step": 7259
    },
    {
      "epoch": 6.922721862792969e-06,
      "step": 7259,
      "training_step_time": 8.718696117401123
    },
    {
      "epoch": 6.923675537109375e-06,
      "grad_norm": 0.3691652715206146,
      "learning_rate": 9.986054855288855e-06,
      "loss": 0.0801,
      "step": 7260
    },
    {
      "epoch": 6.923675537109375e-06,
      "model_forward_time": 1.7252352237701416,
      "step": 7260
    },
    {
      "dynamics_loss_avg": 0.06247028559446335,
      "epoch": 6.923675537109375e-06,
      "step": 7260
    },
    {
      "action_loss_avg": 0.00634336294606328,
      "epoch": 6.923675537109375e-06,
      "step": 7260
    },
    {
      "epoch": 6.923675537109375e-06,
      "step": 7260,
      "training_step_time": 7.694605588912964
    },
    {
      "epoch": 6.924629211425781e-06,
      "model_forward_time": 2.313328504562378,
      "step": 7261
    },
    {
      "epoch": 6.924629211425781e-06,
      "step": 7261,
      "training_step_time": 10.913350820541382
    },
    {
      "epoch": 6.925582885742187e-06,
      "model_forward_time": 1.7230863571166992,
      "step": 7262
    },
    {
      "epoch": 6.925582885742187e-06,
      "step": 7262,
      "training_step_time": 7.924599647521973
    },
    {
      "epoch": 6.926536560058594e-06,
      "model_forward_time": 1.6909806728363037,
      "step": 7263
    },
    {
      "epoch": 6.926536560058594e-06,
      "step": 7263,
      "training_step_time": 8.946574687957764
    },
    {
      "epoch": 6.927490234375e-06,
      "model_forward_time": 1.7042012214660645,
      "step": 7264
    },
    {
      "epoch": 6.927490234375e-06,
      "step": 7264,
      "training_step_time": 7.690181732177734
    },
    {
      "epoch": 6.928443908691406e-06,
      "model_forward_time": 1.713465929031372,
      "step": 7265
    },
    {
      "epoch": 6.928443908691406e-06,
      "step": 7265,
      "training_step_time": 7.295568227767944
    },
    {
      "epoch": 6.9293975830078125e-06,
      "model_forward_time": 1.7195055484771729,
      "step": 7266
    },
    {
      "epoch": 6.9293975830078125e-06,
      "step": 7266,
      "training_step_time": 6.859339714050293
    },
    {
      "epoch": 6.930351257324219e-06,
      "model_forward_time": 1.7028603553771973,
      "step": 7267
    },
    {
      "epoch": 6.930351257324219e-06,
      "step": 7267,
      "training_step_time": 7.40447735786438
    },
    {
      "epoch": 6.931304931640625e-06,
      "model_forward_time": 1.7086365222930908,
      "step": 7268
    },
    {
      "epoch": 6.931304931640625e-06,
      "step": 7268,
      "training_step_time": 7.794427394866943
    },
    {
      "epoch": 6.932258605957031e-06,
      "model_forward_time": 1.7076342105865479,
      "step": 7269
    },
    {
      "epoch": 6.932258605957031e-06,
      "step": 7269,
      "training_step_time": 8.729413986206055
    },
    {
      "epoch": 6.933212280273437e-06,
      "grad_norm": 0.32541367411613464,
      "learning_rate": 9.985931177089802e-06,
      "loss": 0.0832,
      "step": 7270
    },
    {
      "epoch": 6.933212280273437e-06,
      "model_forward_time": 2.3027267456054688,
      "step": 7270
    },
    {
      "dynamics_loss_avg": 0.061642466485500334,
      "epoch": 6.933212280273437e-06,
      "step": 7270
    },
    {
      "action_loss_avg": 0.0052474398165941235,
      "epoch": 6.933212280273437e-06,
      "step": 7270
    },
    {
      "epoch": 6.933212280273437e-06,
      "step": 7270,
      "training_step_time": 8.806917667388916
    },
    {
      "epoch": 6.934165954589844e-06,
      "model_forward_time": 1.7081282138824463,
      "step": 7271
    },
    {
      "epoch": 6.934165954589844e-06,
      "step": 7271,
      "training_step_time": 8.893921613693237
    },
    {
      "epoch": 6.93511962890625e-06,
      "model_forward_time": 1.696422815322876,
      "step": 7272
    },
    {
      "epoch": 6.93511962890625e-06,
      "step": 7272,
      "training_step_time": 9.08083701133728
    },
    {
      "epoch": 6.936073303222656e-06,
      "model_forward_time": 1.7186024188995361,
      "step": 7273
    },
    {
      "epoch": 6.936073303222656e-06,
      "step": 7273,
      "training_step_time": 7.866438388824463
    },
    {
      "epoch": 6.9370269775390625e-06,
      "model_forward_time": 1.3430545330047607,
      "step": 7274
    },
    {
      "epoch": 6.9370269775390625e-06,
      "step": 7274,
      "training_step_time": 7.384176969528198
    },
    {
      "epoch": 6.937980651855469e-06,
      "model_forward_time": 1.7156178951263428,
      "step": 7275
    },
    {
      "epoch": 6.937980651855469e-06,
      "step": 7275,
      "training_step_time": 6.929085969924927
    },
    {
      "epoch": 6.938934326171875e-06,
      "model_forward_time": 1.715435266494751,
      "step": 7276
    },
    {
      "epoch": 6.938934326171875e-06,
      "step": 7276,
      "training_step_time": 7.4067206382751465
    },
    {
      "epoch": 6.939888000488281e-06,
      "model_forward_time": 1.7158584594726562,
      "step": 7277
    },
    {
      "epoch": 6.939888000488281e-06,
      "step": 7277,
      "training_step_time": 7.876737356185913
    },
    {
      "epoch": 6.940841674804687e-06,
      "model_forward_time": 1.7080960273742676,
      "step": 7278
    },
    {
      "epoch": 6.940841674804687e-06,
      "step": 7278,
      "training_step_time": 7.753089427947998
    },
    {
      "epoch": 6.941795349121094e-06,
      "model_forward_time": 2.241636037826538,
      "step": 7279
    },
    {
      "epoch": 6.941795349121094e-06,
      "step": 7279,
      "training_step_time": 9.422486066818237
    },
    {
      "epoch": 6.9427490234375e-06,
      "grad_norm": 0.25409936904907227,
      "learning_rate": 9.985806953636814e-06,
      "loss": 0.0728,
      "step": 7280
    },
    {
      "epoch": 6.9427490234375e-06,
      "model_forward_time": 1.7136096954345703,
      "step": 7280
    },
    {
      "dynamics_loss_avg": 0.08143844082951546,
      "epoch": 6.9427490234375e-06,
      "step": 7280
    },
    {
      "action_loss_avg": 0.008297247136943043,
      "epoch": 6.9427490234375e-06,
      "step": 7280
    },
    {
      "epoch": 6.9427490234375e-06,
      "step": 7280,
      "training_step_time": 10.506817817687988
    },
    {
      "epoch": 6.943702697753906e-06,
      "model_forward_time": 1.707230806350708,
      "step": 7281
    },
    {
      "epoch": 6.943702697753906e-06,
      "step": 7281,
      "training_step_time": 8.520827770233154
    },
    {
      "epoch": 6.9446563720703125e-06,
      "model_forward_time": 1.714156150817871,
      "step": 7282
    },
    {
      "epoch": 6.9446563720703125e-06,
      "step": 7282,
      "training_step_time": 7.769767999649048
    },
    {
      "epoch": 6.945610046386719e-06,
      "model_forward_time": 1.7124664783477783,
      "step": 7283
    },
    {
      "epoch": 6.945610046386719e-06,
      "step": 7283,
      "training_step_time": 7.740464448928833
    },
    {
      "epoch": 6.946563720703125e-06,
      "model_forward_time": 1.7133421897888184,
      "step": 7284
    },
    {
      "epoch": 6.946563720703125e-06,
      "step": 7284,
      "training_step_time": 6.803708553314209
    },
    {
      "epoch": 6.947517395019531e-06,
      "model_forward_time": 1.7050774097442627,
      "step": 7285
    },
    {
      "epoch": 6.947517395019531e-06,
      "step": 7285,
      "training_step_time": 6.8921167850494385
    },
    {
      "epoch": 6.948471069335937e-06,
      "model_forward_time": 1.712778091430664,
      "step": 7286
    },
    {
      "epoch": 6.948471069335937e-06,
      "step": 7286,
      "training_step_time": 7.829142808914185
    },
    {
      "epoch": 6.949424743652344e-06,
      "model_forward_time": 1.3469393253326416,
      "step": 7287
    },
    {
      "epoch": 6.949424743652344e-06,
      "step": 7287,
      "training_step_time": 7.72270655632019
    },
    {
      "epoch": 6.95037841796875e-06,
      "model_forward_time": 1.7083377838134766,
      "step": 7288
    },
    {
      "epoch": 6.95037841796875e-06,
      "step": 7288,
      "training_step_time": 9.317242860794067
    },
    {
      "epoch": 6.951332092285156e-06,
      "model_forward_time": 2.2751522064208984,
      "step": 7289
    },
    {
      "epoch": 6.951332092285156e-06,
      "step": 7289,
      "training_step_time": 9.16602373123169
    },
    {
      "epoch": 6.9522857666015625e-06,
      "grad_norm": 0.23710577189922333,
      "learning_rate": 9.985682184943472e-06,
      "loss": 0.0663,
      "step": 7290
    },
    {
      "epoch": 6.9522857666015625e-06,
      "model_forward_time": 1.702355146408081,
      "step": 7290
    },
    {
      "dynamics_loss_avg": 0.06387631800025702,
      "epoch": 6.9522857666015625e-06,
      "step": 7290
    },
    {
      "action_loss_avg": 0.006145408470183611,
      "epoch": 6.9522857666015625e-06,
      "step": 7290
    },
    {
      "epoch": 6.9522857666015625e-06,
      "step": 7290,
      "training_step_time": 7.935651540756226
    },
    {
      "epoch": 6.953239440917969e-06,
      "model_forward_time": 1.7120370864868164,
      "step": 7291
    },
    {
      "epoch": 6.953239440917969e-06,
      "step": 7291,
      "training_step_time": 7.728379011154175
    },
    {
      "epoch": 6.954193115234375e-06,
      "model_forward_time": 1.709582805633545,
      "step": 7292
    },
    {
      "epoch": 6.954193115234375e-06,
      "step": 7292,
      "training_step_time": 7.7103517055511475
    },
    {
      "epoch": 6.955146789550781e-06,
      "model_forward_time": 1.7109777927398682,
      "step": 7293
    },
    {
      "epoch": 6.955146789550781e-06,
      "step": 7293,
      "training_step_time": 7.130914688110352
    },
    {
      "epoch": 6.956100463867187e-06,
      "model_forward_time": 1.7106854915618896,
      "step": 7294
    },
    {
      "epoch": 6.956100463867187e-06,
      "step": 7294,
      "training_step_time": 6.816201448440552
    },
    {
      "epoch": 6.957054138183594e-06,
      "model_forward_time": 1.7146985530853271,
      "step": 7295
    },
    {
      "epoch": 6.957054138183594e-06,
      "step": 7295,
      "training_step_time": 7.825250148773193
    },
    {
      "epoch": 6.9580078125e-06,
      "model_forward_time": 1.7129182815551758,
      "step": 7296
    },
    {
      "epoch": 6.9580078125e-06,
      "step": 7296,
      "training_step_time": 7.645623445510864
    },
    {
      "epoch": 6.958961486816406e-06,
      "model_forward_time": 1.7086591720581055,
      "step": 7297
    },
    {
      "epoch": 6.958961486816406e-06,
      "step": 7297,
      "training_step_time": 9.39536190032959
    },
    {
      "epoch": 6.9599151611328125e-06,
      "model_forward_time": 2.267709255218506,
      "step": 7298
    },
    {
      "epoch": 6.9599151611328125e-06,
      "step": 7298,
      "training_step_time": 7.706358432769775
    },
    {
      "epoch": 6.960868835449219e-06,
      "model_forward_time": 1.725151538848877,
      "step": 7299
    },
    {
      "epoch": 6.960868835449219e-06,
      "step": 7299,
      "training_step_time": 7.946025609970093
    },
    {
      "epoch": 6.961822509765625e-06,
      "grad_norm": 0.5018030405044556,
      "learning_rate": 9.98555687102342e-06,
      "loss": 0.0673,
      "step": 7300
    },
    {
      "epoch": 6.961822509765625e-06,
      "model_forward_time": 1.7099382877349854,
      "step": 7300
    },
    {
      "dynamics_loss_avg": 0.0687928756698966,
      "epoch": 6.961822509765625e-06,
      "step": 7300
    },
    {
      "action_loss_avg": 0.009289941028691828,
      "epoch": 6.961822509765625e-06,
      "step": 7300
    },
    {
      "epoch": 6.961822509765625e-06,
      "step": 7300,
      "training_step_time": 7.7825400829315186
    },
    {
      "epoch": 6.962776184082031e-06,
      "model_forward_time": 1.7002956867218018,
      "step": 7301
    },
    {
      "epoch": 6.962776184082031e-06,
      "step": 7301,
      "training_step_time": 7.760598421096802
    },
    {
      "epoch": 6.963729858398437e-06,
      "model_forward_time": 1.707923412322998,
      "step": 7302
    },
    {
      "epoch": 6.963729858398437e-06,
      "step": 7302,
      "training_step_time": 7.228237152099609
    },
    {
      "epoch": 6.964683532714844e-06,
      "model_forward_time": 1.3481402397155762,
      "step": 7303
    },
    {
      "epoch": 6.964683532714844e-06,
      "step": 7303,
      "training_step_time": 6.9251933097839355
    },
    {
      "epoch": 6.96563720703125e-06,
      "model_forward_time": 1.7150664329528809,
      "step": 7304
    },
    {
      "epoch": 6.96563720703125e-06,
      "step": 7304,
      "training_step_time": 7.842820167541504
    },
    {
      "epoch": 6.966590881347656e-06,
      "model_forward_time": 1.7086246013641357,
      "step": 7305
    },
    {
      "epoch": 6.966590881347656e-06,
      "step": 7305,
      "training_step_time": 7.71164870262146
    },
    {
      "epoch": 6.9675445556640625e-06,
      "model_forward_time": 1.7156562805175781,
      "step": 7306
    },
    {
      "epoch": 6.9675445556640625e-06,
      "step": 7306,
      "training_step_time": 9.63982629776001
    },
    {
      "epoch": 6.968498229980469e-06,
      "model_forward_time": 2.274590492248535,
      "step": 7307
    },
    {
      "epoch": 6.968498229980469e-06,
      "step": 7307,
      "training_step_time": 8.442375421524048
    },
    {
      "epoch": 6.969451904296875e-06,
      "model_forward_time": 1.7098703384399414,
      "step": 7308
    },
    {
      "epoch": 6.969451904296875e-06,
      "step": 7308,
      "training_step_time": 8.480973958969116
    },
    {
      "epoch": 6.970405578613281e-06,
      "model_forward_time": 1.6983025074005127,
      "step": 7309
    },
    {
      "epoch": 6.970405578613281e-06,
      "step": 7309,
      "training_step_time": 8.736598014831543
    },
    {
      "epoch": 6.971359252929687e-06,
      "grad_norm": 0.2943377196788788,
      "learning_rate": 9.985431011890368e-06,
      "loss": 0.0747,
      "step": 7310
    },
    {
      "epoch": 6.971359252929687e-06,
      "model_forward_time": 1.7103760242462158,
      "step": 7310
    },
    {
      "dynamics_loss_avg": 0.051224900875240564,
      "epoch": 6.971359252929687e-06,
      "step": 7310
    },
    {
      "action_loss_avg": 0.0048573857522569595,
      "epoch": 6.971359252929687e-06,
      "step": 7310
    },
    {
      "epoch": 6.971359252929687e-06,
      "step": 7310,
      "training_step_time": 7.859063625335693
    },
    {
      "epoch": 6.972312927246094e-06,
      "model_forward_time": 1.7065041065216064,
      "step": 7311
    },
    {
      "epoch": 6.972312927246094e-06,
      "step": 7311,
      "training_step_time": 6.743630886077881
    },
    {
      "epoch": 6.9732666015625e-06,
      "model_forward_time": 1.7217228412628174,
      "step": 7312
    },
    {
      "epoch": 6.9732666015625e-06,
      "step": 7312,
      "training_step_time": 6.795191287994385
    },
    {
      "epoch": 6.974220275878906e-06,
      "model_forward_time": 1.717825174331665,
      "step": 7313
    },
    {
      "epoch": 6.974220275878906e-06,
      "step": 7313,
      "training_step_time": 7.550976514816284
    },
    {
      "epoch": 6.9751739501953126e-06,
      "model_forward_time": 1.7223904132843018,
      "step": 7314
    },
    {
      "epoch": 6.9751739501953126e-06,
      "step": 7314,
      "training_step_time": 7.758729696273804
    },
    {
      "epoch": 6.976127624511719e-06,
      "model_forward_time": 1.7215478420257568,
      "step": 7315
    },
    {
      "epoch": 6.976127624511719e-06,
      "step": 7315,
      "training_step_time": 8.43517279624939
    },
    {
      "epoch": 6.977081298828125e-06,
      "model_forward_time": 2.2798187732696533,
      "step": 7316
    },
    {
      "epoch": 6.977081298828125e-06,
      "step": 7316,
      "training_step_time": 9.429399490356445
    },
    {
      "epoch": 6.978034973144531e-06,
      "model_forward_time": 1.7069616317749023,
      "step": 7317
    },
    {
      "epoch": 6.978034973144531e-06,
      "step": 7317,
      "training_step_time": 12.277398824691772
    },
    {
      "epoch": 6.978988647460937e-06,
      "model_forward_time": 1.7045462131500244,
      "step": 7318
    },
    {
      "epoch": 6.978988647460937e-06,
      "step": 7318,
      "training_step_time": 8.46177077293396
    },
    {
      "epoch": 6.979942321777344e-06,
      "model_forward_time": 1.7075104713439941,
      "step": 7319
    },
    {
      "epoch": 6.979942321777344e-06,
      "step": 7319,
      "training_step_time": 8.630373239517212
    },
    {
      "epoch": 6.98089599609375e-06,
      "grad_norm": 0.2928094267845154,
      "learning_rate": 9.985304607558075e-06,
      "loss": 0.0742,
      "step": 7320
    },
    {
      "epoch": 6.98089599609375e-06,
      "model_forward_time": 1.7155585289001465,
      "step": 7320
    },
    {
      "dynamics_loss_avg": 0.054999908059835435,
      "epoch": 6.98089599609375e-06,
      "step": 7320
    },
    {
      "action_loss_avg": 0.0059562877519056204,
      "epoch": 6.98089599609375e-06,
      "step": 7320
    },
    {
      "epoch": 6.98089599609375e-06,
      "step": 7320,
      "training_step_time": 7.6764912605285645
    },
    {
      "epoch": 6.981849670410156e-06,
      "model_forward_time": 1.711026906967163,
      "step": 7321
    },
    {
      "epoch": 6.981849670410156e-06,
      "step": 7321,
      "training_step_time": 7.152525424957275
    },
    {
      "epoch": 6.9828033447265626e-06,
      "model_forward_time": 1.7135248184204102,
      "step": 7322
    },
    {
      "epoch": 6.9828033447265626e-06,
      "step": 7322,
      "training_step_time": 7.414255380630493
    },
    {
      "epoch": 6.983757019042969e-06,
      "model_forward_time": 1.7190711498260498,
      "step": 7323
    },
    {
      "epoch": 6.983757019042969e-06,
      "step": 7323,
      "training_step_time": 7.741319417953491
    },
    {
      "epoch": 6.984710693359375e-06,
      "model_forward_time": 1.7163052558898926,
      "step": 7324
    },
    {
      "epoch": 6.984710693359375e-06,
      "step": 7324,
      "training_step_time": 7.552061080932617
    },
    {
      "epoch": 6.985664367675781e-06,
      "model_forward_time": 1.704319953918457,
      "step": 7325
    },
    {
      "epoch": 6.985664367675781e-06,
      "step": 7325,
      "training_step_time": 8.393696069717407
    },
    {
      "epoch": 6.986618041992187e-06,
      "model_forward_time": 2.246227979660034,
      "step": 7326
    },
    {
      "epoch": 6.986618041992187e-06,
      "step": 7326,
      "training_step_time": 9.398026704788208
    },
    {
      "epoch": 6.987571716308594e-06,
      "model_forward_time": 1.7174029350280762,
      "step": 7327
    },
    {
      "epoch": 6.987571716308594e-06,
      "step": 7327,
      "training_step_time": 9.619292259216309
    },
    {
      "epoch": 6.988525390625e-06,
      "model_forward_time": 1.7086896896362305,
      "step": 7328
    },
    {
      "epoch": 6.988525390625e-06,
      "step": 7328,
      "training_step_time": 7.796460866928101
    },
    {
      "epoch": 6.989479064941406e-06,
      "model_forward_time": 1.7137463092803955,
      "step": 7329
    },
    {
      "epoch": 6.989479064941406e-06,
      "step": 7329,
      "training_step_time": 7.754587173461914
    },
    {
      "epoch": 6.9904327392578126e-06,
      "grad_norm": 0.3669138550758362,
      "learning_rate": 9.985177658040363e-06,
      "loss": 0.0709,
      "step": 7330
    },
    {
      "epoch": 6.9904327392578126e-06,
      "model_forward_time": 1.716501235961914,
      "step": 7330
    },
    {
      "dynamics_loss_avg": 0.06040542759001255,
      "epoch": 6.9904327392578126e-06,
      "step": 7330
    },
    {
      "action_loss_avg": 0.008061917230952531,
      "epoch": 6.9904327392578126e-06,
      "step": 7330
    },
    {
      "epoch": 6.9904327392578126e-06,
      "step": 7330,
      "training_step_time": 7.202423095703125
    },
    {
      "epoch": 6.991386413574219e-06,
      "model_forward_time": 1.7066643238067627,
      "step": 7331
    },
    {
      "epoch": 6.991386413574219e-06,
      "step": 7331,
      "training_step_time": 7.551616430282593
    },
    {
      "epoch": 6.992340087890625e-06,
      "model_forward_time": 1.7104666233062744,
      "step": 7332
    },
    {
      "epoch": 6.992340087890625e-06,
      "step": 7332,
      "training_step_time": 7.851708650588989
    },
    {
      "epoch": 6.9932937622070314e-06,
      "model_forward_time": 1.7132978439331055,
      "step": 7333
    },
    {
      "epoch": 6.9932937622070314e-06,
      "step": 7333,
      "training_step_time": 9.391403675079346
    },
    {
      "epoch": 6.994247436523437e-06,
      "model_forward_time": 1.706587553024292,
      "step": 7334
    },
    {
      "epoch": 6.994247436523437e-06,
      "step": 7334,
      "training_step_time": 7.791438341140747
    },
    {
      "epoch": 6.995201110839844e-06,
      "model_forward_time": 2.2974584102630615,
      "step": 7335
    },
    {
      "epoch": 6.995201110839844e-06,
      "step": 7335,
      "training_step_time": 7.821500539779663
    },
    {
      "epoch": 6.99615478515625e-06,
      "model_forward_time": 1.7059152126312256,
      "step": 7336
    },
    {
      "epoch": 6.99615478515625e-06,
      "step": 7336,
      "training_step_time": 7.920671224594116
    },
    {
      "epoch": 6.997108459472656e-06,
      "model_forward_time": 1.700547456741333,
      "step": 7337
    },
    {
      "epoch": 6.997108459472656e-06,
      "step": 7337,
      "training_step_time": 8.283408164978027
    },
    {
      "epoch": 6.9980621337890626e-06,
      "model_forward_time": 1.3476777076721191,
      "step": 7338
    },
    {
      "epoch": 6.9980621337890626e-06,
      "step": 7338,
      "training_step_time": 7.8731255531311035
    },
    {
      "epoch": 6.999015808105469e-06,
      "model_forward_time": 1.7318124771118164,
      "step": 7339
    },
    {
      "epoch": 6.999015808105469e-06,
      "step": 7339,
      "training_step_time": 7.248893737792969
    },
    {
      "epoch": 6.999969482421875e-06,
      "grad_norm": 0.2668606638908386,
      "learning_rate": 9.98505016335112e-06,
      "loss": 0.0674,
      "step": 7340
    },
    {
      "epoch": 6.999969482421875e-06,
      "model_forward_time": 1.7093620300292969,
      "step": 7340
    },
    {
      "dynamics_loss_avg": 0.0693027002736926,
      "epoch": 6.999969482421875e-06,
      "step": 7340
    },
    {
      "action_loss_avg": 0.006357807258609682,
      "epoch": 6.999969482421875e-06,
      "step": 7340
    },
    {
      "epoch": 6.999969482421875e-06,
      "step": 7340,
      "training_step_time": 7.587561845779419
    },
    {
      "epoch": 7.0009231567382814e-06,
      "model_forward_time": 1.7166869640350342,
      "step": 7341
    },
    {
      "epoch": 7.0009231567382814e-06,
      "step": 7341,
      "training_step_time": 7.933967113494873
    },
    {
      "epoch": 7.001876831054687e-06,
      "model_forward_time": 1.7045929431915283,
      "step": 7342
    },
    {
      "epoch": 7.001876831054687e-06,
      "step": 7342,
      "training_step_time": 9.47179365158081
    },
    {
      "epoch": 7.002830505371094e-06,
      "model_forward_time": 1.7153799533843994,
      "step": 7343
    },
    {
      "epoch": 7.002830505371094e-06,
      "step": 7343,
      "training_step_time": 8.850099325180054
    },
    {
      "epoch": 7.0037841796875e-06,
      "model_forward_time": 2.302286386489868,
      "step": 7344
    },
    {
      "epoch": 7.0037841796875e-06,
      "step": 7344,
      "training_step_time": 8.557049036026001
    },
    {
      "epoch": 7.004737854003906e-06,
      "model_forward_time": 1.7049973011016846,
      "step": 7345
    },
    {
      "epoch": 7.004737854003906e-06,
      "step": 7345,
      "training_step_time": 8.006528615951538
    },
    {
      "epoch": 7.005691528320313e-06,
      "model_forward_time": 1.7023284435272217,
      "step": 7346
    },
    {
      "epoch": 7.005691528320313e-06,
      "step": 7346,
      "training_step_time": 8.910393953323364
    },
    {
      "epoch": 7.006645202636719e-06,
      "model_forward_time": 1.701890230178833,
      "step": 7347
    },
    {
      "epoch": 7.006645202636719e-06,
      "step": 7347,
      "training_step_time": 7.904732704162598
    },
    {
      "epoch": 7.007598876953125e-06,
      "model_forward_time": 1.712249755859375,
      "step": 7348
    },
    {
      "epoch": 7.007598876953125e-06,
      "step": 7348,
      "training_step_time": 7.837306976318359
    },
    {
      "epoch": 7.0085525512695314e-06,
      "model_forward_time": 1.704972505569458,
      "step": 7349
    },
    {
      "epoch": 7.0085525512695314e-06,
      "step": 7349,
      "training_step_time": 8.315817594528198
    },
    {
      "epoch": 7.009506225585937e-06,
      "grad_norm": 0.23988083004951477,
      "learning_rate": 9.984922123504285e-06,
      "loss": 0.0697,
      "step": 7350
    },
    {
      "epoch": 7.009506225585937e-06,
      "model_forward_time": 1.7142367362976074,
      "step": 7350
    },
    {
      "dynamics_loss_avg": 0.07235505059361458,
      "epoch": 7.009506225585937e-06,
      "step": 7350
    },
    {
      "action_loss_avg": 0.006187857291661203,
      "epoch": 7.009506225585937e-06,
      "step": 7350
    },
    {
      "epoch": 7.009506225585937e-06,
      "step": 7350,
      "training_step_time": 7.803525924682617
    },
    {
      "epoch": 7.010459899902344e-06,
      "model_forward_time": 1.7153427600860596,
      "step": 7351
    },
    {
      "epoch": 7.010459899902344e-06,
      "step": 7351,
      "training_step_time": 7.77290415763855
    },
    {
      "epoch": 7.01141357421875e-06,
      "model_forward_time": 1.703460693359375,
      "step": 7352
    },
    {
      "epoch": 7.01141357421875e-06,
      "step": 7352,
      "training_step_time": 10.295568466186523
    },
    {
      "epoch": 7.012367248535156e-06,
      "model_forward_time": 2.3038058280944824,
      "step": 7353
    },
    {
      "epoch": 7.012367248535156e-06,
      "step": 7353,
      "training_step_time": 8.462668895721436
    },
    {
      "epoch": 7.013320922851563e-06,
      "model_forward_time": 1.3425817489624023,
      "step": 7354
    },
    {
      "epoch": 7.013320922851563e-06,
      "step": 7354,
      "training_step_time": 7.896755218505859
    },
    {
      "epoch": 7.014274597167969e-06,
      "model_forward_time": 1.7031137943267822,
      "step": 7355
    },
    {
      "epoch": 7.014274597167969e-06,
      "step": 7355,
      "training_step_time": 8.562940835952759
    },
    {
      "epoch": 7.015228271484375e-06,
      "model_forward_time": 1.7217841148376465,
      "step": 7356
    },
    {
      "epoch": 7.015228271484375e-06,
      "step": 7356,
      "training_step_time": 7.7016472816467285
    },
    {
      "epoch": 7.0161819458007814e-06,
      "model_forward_time": 1.72601318359375,
      "step": 7357
    },
    {
      "epoch": 7.0161819458007814e-06,
      "step": 7357,
      "training_step_time": 7.94585919380188
    },
    {
      "epoch": 7.017135620117187e-06,
      "model_forward_time": 1.7231419086456299,
      "step": 7358
    },
    {
      "epoch": 7.017135620117187e-06,
      "step": 7358,
      "training_step_time": 8.046244621276855
    },
    {
      "epoch": 7.018089294433594e-06,
      "model_forward_time": 1.7179574966430664,
      "step": 7359
    },
    {
      "epoch": 7.018089294433594e-06,
      "step": 7359,
      "training_step_time": 7.7912678718566895
    },
    {
      "epoch": 7.01904296875e-06,
      "grad_norm": 0.2681581377983093,
      "learning_rate": 9.984793538513862e-06,
      "loss": 0.0704,
      "step": 7360
    },
    {
      "epoch": 7.01904296875e-06,
      "model_forward_time": 1.715407133102417,
      "step": 7360
    },
    {
      "dynamics_loss_avg": 0.06808481086045504,
      "epoch": 7.01904296875e-06,
      "step": 7360
    },
    {
      "action_loss_avg": 0.009231447183992714,
      "epoch": 7.01904296875e-06,
      "step": 7360
    },
    {
      "epoch": 7.01904296875e-06,
      "step": 7360,
      "training_step_time": 7.742068767547607
    },
    {
      "epoch": 7.019996643066406e-06,
      "model_forward_time": 1.7207527160644531,
      "step": 7361
    },
    {
      "epoch": 7.019996643066406e-06,
      "step": 7361,
      "training_step_time": 7.606449127197266
    },
    {
      "epoch": 7.020950317382813e-06,
      "model_forward_time": 1.7169854640960693,
      "step": 7362
    },
    {
      "epoch": 7.020950317382813e-06,
      "step": 7362,
      "training_step_time": 8.463789224624634
    },
    {
      "epoch": 7.021903991699219e-06,
      "model_forward_time": 2.2886321544647217,
      "step": 7363
    },
    {
      "epoch": 7.021903991699219e-06,
      "step": 7363,
      "training_step_time": 7.866032123565674
    },
    {
      "epoch": 7.022857666015625e-06,
      "model_forward_time": 1.70908522605896,
      "step": 7364
    },
    {
      "epoch": 7.022857666015625e-06,
      "step": 7364,
      "training_step_time": 8.840832710266113
    },
    {
      "epoch": 7.0238113403320315e-06,
      "model_forward_time": 1.7224771976470947,
      "step": 7365
    },
    {
      "epoch": 7.0238113403320315e-06,
      "step": 7365,
      "training_step_time": 7.77315354347229
    },
    {
      "epoch": 7.024765014648437e-06,
      "model_forward_time": 1.7180023193359375,
      "step": 7366
    },
    {
      "epoch": 7.024765014648437e-06,
      "step": 7366,
      "training_step_time": 7.79526162147522
    },
    {
      "epoch": 7.025718688964844e-06,
      "model_forward_time": 1.7241544723510742,
      "step": 7367
    },
    {
      "epoch": 7.025718688964844e-06,
      "step": 7367,
      "training_step_time": 7.711345434188843
    },
    {
      "epoch": 7.02667236328125e-06,
      "model_forward_time": 1.7254180908203125,
      "step": 7368
    },
    {
      "epoch": 7.02667236328125e-06,
      "step": 7368,
      "training_step_time": 7.7292797565460205
    },
    {
      "epoch": 7.027626037597656e-06,
      "model_forward_time": 1.7148737907409668,
      "step": 7369
    },
    {
      "epoch": 7.027626037597656e-06,
      "step": 7369,
      "training_step_time": 8.78411054611206
    },
    {
      "epoch": 7.028579711914063e-06,
      "grad_norm": 0.23773914575576782,
      "learning_rate": 9.984664408393911e-06,
      "loss": 0.0678,
      "step": 7370
    },
    {
      "epoch": 7.028579711914063e-06,
      "model_forward_time": 1.7179288864135742,
      "step": 7370
    },
    {
      "dynamics_loss_avg": 0.06111161168664694,
      "epoch": 7.028579711914063e-06,
      "step": 7370
    },
    {
      "action_loss_avg": 0.004582104342989624,
      "epoch": 7.028579711914063e-06,
      "step": 7370
    },
    {
      "epoch": 7.028579711914063e-06,
      "step": 7370,
      "training_step_time": 8.51311206817627
    },
    {
      "epoch": 7.029533386230469e-06,
      "model_forward_time": 1.7221839427947998,
      "step": 7371
    },
    {
      "epoch": 7.029533386230469e-06,
      "step": 7371,
      "training_step_time": 9.945655107498169
    },
    {
      "epoch": 7.030487060546875e-06,
      "model_forward_time": 2.3101165294647217,
      "step": 7372
    },
    {
      "epoch": 7.030487060546875e-06,
      "step": 7372,
      "training_step_time": 9.587613344192505
    },
    {
      "epoch": 7.0314407348632815e-06,
      "model_forward_time": 1.7199132442474365,
      "step": 7373
    },
    {
      "epoch": 7.0314407348632815e-06,
      "step": 7373,
      "training_step_time": 9.204707860946655
    },
    {
      "epoch": 7.032394409179687e-06,
      "model_forward_time": 1.7171154022216797,
      "step": 7374
    },
    {
      "epoch": 7.032394409179687e-06,
      "step": 7374,
      "training_step_time": 8.848930358886719
    },
    {
      "epoch": 7.033348083496094e-06,
      "model_forward_time": 1.7196314334869385,
      "step": 7375
    },
    {
      "epoch": 7.033348083496094e-06,
      "step": 7375,
      "training_step_time": 7.801715135574341
    },
    {
      "epoch": 7.0343017578125e-06,
      "model_forward_time": 1.724458932876587,
      "step": 7376
    },
    {
      "epoch": 7.0343017578125e-06,
      "step": 7376,
      "training_step_time": 7.736943006515503
    },
    {
      "epoch": 7.035255432128906e-06,
      "model_forward_time": 1.7216496467590332,
      "step": 7377
    },
    {
      "epoch": 7.035255432128906e-06,
      "step": 7377,
      "training_step_time": 7.681400775909424
    },
    {
      "epoch": 7.036209106445313e-06,
      "model_forward_time": 1.719874620437622,
      "step": 7378
    },
    {
      "epoch": 7.036209106445313e-06,
      "step": 7378,
      "training_step_time": 7.804558515548706
    },
    {
      "epoch": 7.037162780761718e-06,
      "model_forward_time": 1.7231659889221191,
      "step": 7379
    },
    {
      "epoch": 7.037162780761718e-06,
      "step": 7379,
      "training_step_time": 7.6696999073028564
    },
    {
      "epoch": 7.038116455078125e-06,
      "grad_norm": 0.3758300840854645,
      "learning_rate": 9.984534733158556e-06,
      "loss": 0.0643,
      "step": 7380
    },
    {
      "epoch": 7.038116455078125e-06,
      "model_forward_time": 1.7208442687988281,
      "step": 7380
    },
    {
      "dynamics_loss_avg": 0.06546855121850967,
      "epoch": 7.038116455078125e-06,
      "step": 7380
    },
    {
      "action_loss_avg": 0.0065433755051344635,
      "epoch": 7.038116455078125e-06,
      "step": 7380
    },
    {
      "epoch": 7.038116455078125e-06,
      "step": 7380,
      "training_step_time": 7.801879644393921
    },
    {
      "epoch": 7.0390701293945315e-06,
      "model_forward_time": 2.303743600845337,
      "step": 7381
    },
    {
      "epoch": 7.0390701293945315e-06,
      "step": 7381,
      "training_step_time": 8.885316848754883
    },
    {
      "epoch": 7.040023803710937e-06,
      "model_forward_time": 1.7249574661254883,
      "step": 7382
    },
    {
      "epoch": 7.040023803710937e-06,
      "step": 7382,
      "training_step_time": 7.856475830078125
    },
    {
      "epoch": 7.040977478027344e-06,
      "model_forward_time": 1.7120165824890137,
      "step": 7383
    },
    {
      "epoch": 7.040977478027344e-06,
      "step": 7383,
      "training_step_time": 7.789894342422485
    },
    {
      "epoch": 7.04193115234375e-06,
      "model_forward_time": 1.7283196449279785,
      "step": 7384
    },
    {
      "epoch": 7.04193115234375e-06,
      "step": 7384,
      "training_step_time": 7.647494554519653
    },
    {
      "epoch": 7.042884826660156e-06,
      "model_forward_time": 1.7236688137054443,
      "step": 7385
    },
    {
      "epoch": 7.042884826660156e-06,
      "step": 7385,
      "training_step_time": 8.815328121185303
    },
    {
      "epoch": 7.043838500976563e-06,
      "model_forward_time": 1.7131128311157227,
      "step": 7386
    },
    {
      "epoch": 7.043838500976563e-06,
      "step": 7386,
      "training_step_time": 7.610554218292236
    },
    {
      "epoch": 7.044792175292968e-06,
      "model_forward_time": 1.7275774478912354,
      "step": 7387
    },
    {
      "epoch": 7.044792175292968e-06,
      "step": 7387,
      "training_step_time": 7.705837965011597
    },
    {
      "epoch": 7.045745849609375e-06,
      "model_forward_time": 1.7154691219329834,
      "step": 7388
    },
    {
      "epoch": 7.045745849609375e-06,
      "step": 7388,
      "training_step_time": 8.62700629234314
    },
    {
      "epoch": 7.0466995239257815e-06,
      "model_forward_time": 1.7267944812774658,
      "step": 7389
    },
    {
      "epoch": 7.0466995239257815e-06,
      "step": 7389,
      "training_step_time": 9.647428035736084
    },
    {
      "epoch": 7.047653198242187e-06,
      "grad_norm": 0.40664777159690857,
      "learning_rate": 9.984404512821978e-06,
      "loss": 0.0668,
      "step": 7390
    },
    {
      "epoch": 7.047653198242187e-06,
      "model_forward_time": 2.301093101501465,
      "step": 7390
    },
    {
      "dynamics_loss_avg": 0.06835300959646702,
      "epoch": 7.047653198242187e-06,
      "step": 7390
    },
    {
      "action_loss_avg": 0.006798728951252997,
      "epoch": 7.047653198242187e-06,
      "step": 7390
    },
    {
      "epoch": 7.047653198242187e-06,
      "step": 7390,
      "training_step_time": 9.371547937393188
    },
    {
      "epoch": 7.048606872558594e-06,
      "model_forward_time": 1.7223408222198486,
      "step": 7391
    },
    {
      "epoch": 7.048606872558594e-06,
      "step": 7391,
      "training_step_time": 9.923060655593872
    },
    {
      "epoch": 7.049560546875e-06,
      "model_forward_time": 1.7233388423919678,
      "step": 7392
    },
    {
      "epoch": 7.049560546875e-06,
      "step": 7392,
      "training_step_time": 7.807422876358032
    },
    {
      "epoch": 7.050514221191406e-06,
      "model_forward_time": 1.721648931503296,
      "step": 7393
    },
    {
      "epoch": 7.050514221191406e-06,
      "step": 7393,
      "training_step_time": 7.609561204910278
    },
    {
      "epoch": 7.051467895507813e-06,
      "model_forward_time": 1.7067749500274658,
      "step": 7394
    },
    {
      "epoch": 7.051467895507813e-06,
      "step": 7394,
      "training_step_time": 7.771816968917847
    },
    {
      "epoch": 7.052421569824218e-06,
      "model_forward_time": 1.7339513301849365,
      "step": 7395
    },
    {
      "epoch": 7.052421569824218e-06,
      "step": 7395,
      "training_step_time": 7.525111436843872
    },
    {
      "epoch": 7.053375244140625e-06,
      "model_forward_time": 1.3639090061187744,
      "step": 7396
    },
    {
      "epoch": 7.053375244140625e-06,
      "step": 7396,
      "training_step_time": 7.760738134384155
    },
    {
      "epoch": 7.0543289184570315e-06,
      "model_forward_time": 1.711960792541504,
      "step": 7397
    },
    {
      "epoch": 7.0543289184570315e-06,
      "step": 7397,
      "training_step_time": 7.78037166595459
    },
    {
      "epoch": 7.055282592773437e-06,
      "model_forward_time": 1.722714900970459,
      "step": 7398
    },
    {
      "epoch": 7.055282592773437e-06,
      "step": 7398,
      "training_step_time": 7.768013000488281
    },
    {
      "epoch": 7.056236267089844e-06,
      "model_forward_time": 1.3458306789398193,
      "step": 7399
    },
    {
      "epoch": 7.056236267089844e-06,
      "step": 7399,
      "training_step_time": 9.775983333587646
    },
    {
      "epoch": 7.05718994140625e-06,
      "grad_norm": 0.1826419085264206,
      "learning_rate": 9.984273747398411e-06,
      "loss": 0.0702,
      "step": 7400
    },
    {
      "epoch": 7.05718994140625e-06,
      "model_forward_time": 2.2616147994995117,
      "step": 7400
    },
    {
      "dynamics_loss_avg": 0.037772237323224545,
      "epoch": 7.05718994140625e-06,
      "step": 7400
    },
    {
      "action_loss_avg": 0.0037201240891590716,
      "epoch": 7.05718994140625e-06,
      "step": 7400
    },
    {
      "epoch": 7.05718994140625e-06,
      "step": 7400,
      "training_step_time": 7.892213344573975
    },
    {
      "epoch": 7.058143615722656e-06,
      "model_forward_time": 1.7145617008209229,
      "step": 7401
    },
    {
      "epoch": 7.058143615722656e-06,
      "step": 7401,
      "training_step_time": 9.094239950180054
    },
    {
      "epoch": 7.059097290039063e-06,
      "model_forward_time": 1.7272710800170898,
      "step": 7402
    },
    {
      "epoch": 7.059097290039063e-06,
      "step": 7402,
      "training_step_time": 7.560733079910278
    },
    {
      "epoch": 7.060050964355468e-06,
      "model_forward_time": 1.738649606704712,
      "step": 7403
    },
    {
      "epoch": 7.060050964355468e-06,
      "step": 7403,
      "training_step_time": 7.836602449417114
    },
    {
      "epoch": 7.061004638671875e-06,
      "model_forward_time": 1.725778579711914,
      "step": 7404
    },
    {
      "epoch": 7.061004638671875e-06,
      "step": 7404,
      "training_step_time": 7.584200382232666
    },
    {
      "epoch": 7.0619583129882815e-06,
      "model_forward_time": 1.7210023403167725,
      "step": 7405
    },
    {
      "epoch": 7.0619583129882815e-06,
      "step": 7405,
      "training_step_time": 7.774415493011475
    },
    {
      "epoch": 7.062911987304687e-06,
      "model_forward_time": 1.7352449893951416,
      "step": 7406
    },
    {
      "epoch": 7.062911987304687e-06,
      "step": 7406,
      "training_step_time": 7.740146636962891
    },
    {
      "epoch": 7.063865661621094e-06,
      "model_forward_time": 1.7162487506866455,
      "step": 7407
    },
    {
      "epoch": 7.063865661621094e-06,
      "step": 7407,
      "training_step_time": 9.476356506347656
    },
    {
      "epoch": 7.0648193359375e-06,
      "model_forward_time": 1.7177317142486572,
      "step": 7408
    },
    {
      "epoch": 7.0648193359375e-06,
      "step": 7408,
      "training_step_time": 7.899419546127319
    },
    {
      "epoch": 7.065773010253906e-06,
      "model_forward_time": 2.271476984024048,
      "step": 7409
    },
    {
      "epoch": 7.065773010253906e-06,
      "step": 7409,
      "training_step_time": 8.674016237258911
    },
    {
      "epoch": 7.066726684570313e-06,
      "grad_norm": 0.30452412366867065,
      "learning_rate": 9.984142436902166e-06,
      "loss": 0.0718,
      "step": 7410
    },
    {
      "epoch": 7.066726684570313e-06,
      "model_forward_time": 1.7176759243011475,
      "step": 7410
    },
    {
      "dynamics_loss_avg": 0.053738939203321935,
      "epoch": 7.066726684570313e-06,
      "step": 7410
    },
    {
      "action_loss_avg": 0.0051972266810480505,
      "epoch": 7.066726684570313e-06,
      "step": 7410
    },
    {
      "epoch": 7.066726684570313e-06,
      "step": 7410,
      "training_step_time": 7.983959674835205
    },
    {
      "epoch": 7.067680358886718e-06,
      "model_forward_time": 1.7078123092651367,
      "step": 7411
    },
    {
      "epoch": 7.067680358886718e-06,
      "step": 7411,
      "training_step_time": 7.5844433307647705
    },
    {
      "epoch": 7.068634033203125e-06,
      "model_forward_time": 1.7164454460144043,
      "step": 7412
    },
    {
      "epoch": 7.068634033203125e-06,
      "step": 7412,
      "training_step_time": 7.787595748901367
    },
    {
      "epoch": 7.0695877075195315e-06,
      "model_forward_time": 1.7238621711730957,
      "step": 7413
    },
    {
      "epoch": 7.0695877075195315e-06,
      "step": 7413,
      "training_step_time": 7.562272548675537
    },
    {
      "epoch": 7.070541381835937e-06,
      "model_forward_time": 1.711113452911377,
      "step": 7414
    },
    {
      "epoch": 7.070541381835937e-06,
      "step": 7414,
      "training_step_time": 7.787888526916504
    },
    {
      "epoch": 7.071495056152344e-06,
      "model_forward_time": 1.7141404151916504,
      "step": 7415
    },
    {
      "epoch": 7.071495056152344e-06,
      "step": 7415,
      "training_step_time": 7.6981201171875
    },
    {
      "epoch": 7.07244873046875e-06,
      "model_forward_time": 1.7291500568389893,
      "step": 7416
    },
    {
      "epoch": 7.07244873046875e-06,
      "step": 7416,
      "training_step_time": 7.881690740585327
    },
    {
      "epoch": 7.073402404785156e-06,
      "model_forward_time": 1.7240686416625977,
      "step": 7417
    },
    {
      "epoch": 7.073402404785156e-06,
      "step": 7417,
      "training_step_time": 8.605952501296997
    },
    {
      "epoch": 7.074356079101563e-06,
      "model_forward_time": 2.286534547805786,
      "step": 7418
    },
    {
      "epoch": 7.074356079101563e-06,
      "step": 7418,
      "training_step_time": 8.439236164093018
    },
    {
      "epoch": 7.075309753417968e-06,
      "model_forward_time": 1.7146673202514648,
      "step": 7419
    },
    {
      "epoch": 7.075309753417968e-06,
      "step": 7419,
      "training_step_time": 7.865498065948486
    },
    {
      "epoch": 7.076263427734375e-06,
      "grad_norm": 0.23466137051582336,
      "learning_rate": 9.984010581347596e-06,
      "loss": 0.072,
      "step": 7420
    },
    {
      "epoch": 7.076263427734375e-06,
      "model_forward_time": 1.7112586498260498,
      "step": 7420
    },
    {
      "dynamics_loss_avg": 0.05760109657421708,
      "epoch": 7.076263427734375e-06,
      "step": 7420
    },
    {
      "action_loss_avg": 0.0051437702844850716,
      "epoch": 7.076263427734375e-06,
      "step": 7420
    },
    {
      "epoch": 7.076263427734375e-06,
      "step": 7420,
      "training_step_time": 7.666320323944092
    },
    {
      "epoch": 7.0772171020507815e-06,
      "model_forward_time": 1.7123064994812012,
      "step": 7421
    },
    {
      "epoch": 7.0772171020507815e-06,
      "step": 7421,
      "training_step_time": 7.8698835372924805
    },
    {
      "epoch": 7.078170776367187e-06,
      "model_forward_time": 1.7211430072784424,
      "step": 7422
    },
    {
      "epoch": 7.078170776367187e-06,
      "step": 7422,
      "training_step_time": 7.609225034713745
    },
    {
      "epoch": 7.079124450683594e-06,
      "model_forward_time": 1.7105591297149658,
      "step": 7423
    },
    {
      "epoch": 7.079124450683594e-06,
      "step": 7423,
      "training_step_time": 8.993738651275635
    },
    {
      "epoch": 7.080078125e-06,
      "model_forward_time": 1.7055730819702148,
      "step": 7424
    },
    {
      "epoch": 7.080078125e-06,
      "step": 7424,
      "training_step_time": 8.398342609405518
    },
    {
      "epoch": 7.081031799316406e-06,
      "model_forward_time": 1.7044568061828613,
      "step": 7425
    },
    {
      "epoch": 7.081031799316406e-06,
      "step": 7425,
      "training_step_time": 7.704026937484741
    },
    {
      "epoch": 7.081985473632813e-06,
      "model_forward_time": 1.7083187103271484,
      "step": 7426
    },
    {
      "epoch": 7.081985473632813e-06,
      "step": 7426,
      "training_step_time": 9.344258546829224
    },
    {
      "epoch": 7.082939147949218e-06,
      "model_forward_time": 2.2910776138305664,
      "step": 7427
    },
    {
      "epoch": 7.082939147949218e-06,
      "step": 7427,
      "training_step_time": 7.845504522323608
    },
    {
      "epoch": 7.083892822265625e-06,
      "model_forward_time": 1.7202091217041016,
      "step": 7428
    },
    {
      "epoch": 7.083892822265625e-06,
      "step": 7428,
      "training_step_time": 7.878633975982666
    },
    {
      "epoch": 7.0848464965820315e-06,
      "model_forward_time": 1.6991245746612549,
      "step": 7429
    },
    {
      "epoch": 7.0848464965820315e-06,
      "step": 7429,
      "training_step_time": 8.511183023452759
    },
    {
      "epoch": 7.085800170898437e-06,
      "grad_norm": 0.25283926725387573,
      "learning_rate": 9.983878180749122e-06,
      "loss": 0.077,
      "step": 7430
    },
    {
      "epoch": 7.085800170898437e-06,
      "model_forward_time": 1.7070729732513428,
      "step": 7430
    },
    {
      "dynamics_loss_avg": 0.06983375288546086,
      "epoch": 7.085800170898437e-06,
      "step": 7430
    },
    {
      "action_loss_avg": 0.006308118416927755,
      "epoch": 7.085800170898437e-06,
      "step": 7430
    },
    {
      "epoch": 7.085800170898437e-06,
      "step": 7430,
      "training_step_time": 9.149983644485474
    },
    {
      "epoch": 7.086753845214844e-06,
      "model_forward_time": 1.7124197483062744,
      "step": 7431
    },
    {
      "epoch": 7.086753845214844e-06,
      "step": 7431,
      "training_step_time": 7.6989359855651855
    },
    {
      "epoch": 7.08770751953125e-06,
      "model_forward_time": 1.7115068435668945,
      "step": 7432
    },
    {
      "epoch": 7.08770751953125e-06,
      "step": 7432,
      "training_step_time": 7.832779169082642
    },
    {
      "epoch": 7.088661193847656e-06,
      "model_forward_time": 1.7067675590515137,
      "step": 7433
    },
    {
      "epoch": 7.088661193847656e-06,
      "step": 7433,
      "training_step_time": 7.699315309524536
    },
    {
      "epoch": 7.089614868164063e-06,
      "model_forward_time": 1.710000991821289,
      "step": 7434
    },
    {
      "epoch": 7.089614868164063e-06,
      "step": 7434,
      "training_step_time": 8.419065713882446
    },
    {
      "epoch": 7.090568542480468e-06,
      "model_forward_time": 1.7164320945739746,
      "step": 7435
    },
    {
      "epoch": 7.090568542480468e-06,
      "step": 7435,
      "training_step_time": 7.778320550918579
    },
    {
      "epoch": 7.091522216796875e-06,
      "model_forward_time": 2.278627872467041,
      "step": 7436
    },
    {
      "epoch": 7.091522216796875e-06,
      "step": 7436,
      "training_step_time": 7.7803544998168945
    },
    {
      "epoch": 7.0924758911132815e-06,
      "model_forward_time": 1.7105951309204102,
      "step": 7437
    },
    {
      "epoch": 7.0924758911132815e-06,
      "step": 7437,
      "training_step_time": 7.866093158721924
    },
    {
      "epoch": 7.093429565429687e-06,
      "model_forward_time": 1.704789638519287,
      "step": 7438
    },
    {
      "epoch": 7.093429565429687e-06,
      "step": 7438,
      "training_step_time": 7.755254507064819
    },
    {
      "epoch": 7.094383239746094e-06,
      "model_forward_time": 1.705925703048706,
      "step": 7439
    },
    {
      "epoch": 7.094383239746094e-06,
      "step": 7439,
      "training_step_time": 8.95264220237732
    },
    {
      "epoch": 7.0953369140625e-06,
      "grad_norm": 0.30956026911735535,
      "learning_rate": 9.983745235121222e-06,
      "loss": 0.071,
      "step": 7440
    },
    {
      "epoch": 7.0953369140625e-06,
      "model_forward_time": 1.7095088958740234,
      "step": 7440
    },
    {
      "dynamics_loss_avg": 0.06402573212981225,
      "epoch": 7.0953369140625e-06,
      "step": 7440
    },
    {
      "action_loss_avg": 0.0067006802884861825,
      "epoch": 7.0953369140625e-06,
      "step": 7440
    },
    {
      "epoch": 7.0953369140625e-06,
      "step": 7440,
      "training_step_time": 7.692552804946899
    },
    {
      "epoch": 7.096290588378906e-06,
      "model_forward_time": 1.721933126449585,
      "step": 7441
    },
    {
      "epoch": 7.096290588378906e-06,
      "step": 7441,
      "training_step_time": 8.82016921043396
    },
    {
      "epoch": 7.097244262695313e-06,
      "model_forward_time": 1.7045085430145264,
      "step": 7442
    },
    {
      "epoch": 7.097244262695313e-06,
      "step": 7442,
      "training_step_time": 7.71743631362915
    },
    {
      "epoch": 7.098197937011718e-06,
      "model_forward_time": 1.7094409465789795,
      "step": 7443
    },
    {
      "epoch": 7.098197937011718e-06,
      "step": 7443,
      "training_step_time": 8.381772518157959
    },
    {
      "epoch": 7.099151611328125e-06,
      "model_forward_time": 1.708784818649292,
      "step": 7444
    },
    {
      "epoch": 7.099151611328125e-06,
      "step": 7444,
      "training_step_time": 8.407691478729248
    },
    {
      "epoch": 7.1001052856445315e-06,
      "model_forward_time": 1.7562000751495361,
      "step": 7445
    },
    {
      "epoch": 7.1001052856445315e-06,
      "step": 7445,
      "training_step_time": 9.894575834274292
    },
    {
      "epoch": 7.101058959960937e-06,
      "model_forward_time": 1.7094273567199707,
      "step": 7446
    },
    {
      "epoch": 7.101058959960937e-06,
      "step": 7446,
      "training_step_time": 7.844977140426636
    },
    {
      "epoch": 7.102012634277344e-06,
      "model_forward_time": 1.7015612125396729,
      "step": 7447
    },
    {
      "epoch": 7.102012634277344e-06,
      "step": 7447,
      "training_step_time": 8.407970428466797
    },
    {
      "epoch": 7.10296630859375e-06,
      "model_forward_time": 1.702331781387329,
      "step": 7448
    },
    {
      "epoch": 7.10296630859375e-06,
      "step": 7448,
      "training_step_time": 8.547440528869629
    },
    {
      "epoch": 7.103919982910156e-06,
      "model_forward_time": 1.707183837890625,
      "step": 7449
    },
    {
      "epoch": 7.103919982910156e-06,
      "step": 7449,
      "training_step_time": 7.747616767883301
    },
    {
      "epoch": 7.104873657226563e-06,
      "grad_norm": 0.22066345810890198,
      "learning_rate": 9.983611744478439e-06,
      "loss": 0.0683,
      "step": 7450
    },
    {
      "epoch": 7.104873657226563e-06,
      "model_forward_time": 1.3501605987548828,
      "step": 7450
    },
    {
      "dynamics_loss_avg": 0.08105855658650399,
      "epoch": 7.104873657226563e-06,
      "step": 7450
    },
    {
      "action_loss_avg": 0.009069442772306502,
      "epoch": 7.104873657226563e-06,
      "step": 7450
    },
    {
      "epoch": 7.104873657226563e-06,
      "step": 7450,
      "training_step_time": 7.8690409660339355
    },
    {
      "epoch": 7.105827331542968e-06,
      "model_forward_time": 1.7054972648620605,
      "step": 7451
    },
    {
      "epoch": 7.105827331542968e-06,
      "step": 7451,
      "training_step_time": 7.728624105453491
    },
    {
      "epoch": 7.106781005859375e-06,
      "model_forward_time": 1.7073192596435547,
      "step": 7452
    },
    {
      "epoch": 7.106781005859375e-06,
      "step": 7452,
      "training_step_time": 7.640146732330322
    },
    {
      "epoch": 7.1077346801757815e-06,
      "model_forward_time": 1.705141305923462,
      "step": 7453
    },
    {
      "epoch": 7.1077346801757815e-06,
      "step": 7453,
      "training_step_time": 8.493610382080078
    },
    {
      "epoch": 7.108688354492187e-06,
      "model_forward_time": 1.7328417301177979,
      "step": 7454
    },
    {
      "epoch": 7.108688354492187e-06,
      "step": 7454,
      "training_step_time": 9.272037267684937
    },
    {
      "epoch": 7.109642028808594e-06,
      "model_forward_time": 2.2909138202667236,
      "step": 7455
    },
    {
      "epoch": 7.109642028808594e-06,
      "step": 7455,
      "training_step_time": 9.370147705078125
    },
    {
      "epoch": 7.110595703125e-06,
      "model_forward_time": 1.7000784873962402,
      "step": 7456
    },
    {
      "epoch": 7.110595703125e-06,
      "step": 7456,
      "training_step_time": 7.767345905303955
    },
    {
      "epoch": 7.111549377441406e-06,
      "model_forward_time": 1.7223608493804932,
      "step": 7457
    },
    {
      "epoch": 7.111549377441406e-06,
      "step": 7457,
      "training_step_time": 7.518227577209473
    },
    {
      "epoch": 7.112503051757813e-06,
      "model_forward_time": 1.7045867443084717,
      "step": 7458
    },
    {
      "epoch": 7.112503051757813e-06,
      "step": 7458,
      "training_step_time": 7.893206357955933
    },
    {
      "epoch": 7.113456726074218e-06,
      "model_forward_time": 1.7041900157928467,
      "step": 7459
    },
    {
      "epoch": 7.113456726074218e-06,
      "step": 7459,
      "training_step_time": 7.7962965965271
    },
    {
      "epoch": 7.114410400390625e-06,
      "grad_norm": 0.33334481716156006,
      "learning_rate": 9.983477708835365e-06,
      "loss": 0.0666,
      "step": 7460
    },
    {
      "epoch": 7.114410400390625e-06,
      "model_forward_time": 1.7194523811340332,
      "step": 7460
    },
    {
      "dynamics_loss_avg": 0.06373544707894326,
      "epoch": 7.114410400390625e-06,
      "step": 7460
    },
    {
      "action_loss_avg": 0.006750912929419428,
      "epoch": 7.114410400390625e-06,
      "step": 7460
    },
    {
      "epoch": 7.114410400390625e-06,
      "step": 7460,
      "training_step_time": 7.716928958892822
    },
    {
      "epoch": 7.1153640747070315e-06,
      "model_forward_time": 1.7200026512145996,
      "step": 7461
    },
    {
      "epoch": 7.1153640747070315e-06,
      "step": 7461,
      "training_step_time": 7.693376064300537
    },
    {
      "epoch": 7.116317749023437e-06,
      "model_forward_time": 1.7117292881011963,
      "step": 7462
    },
    {
      "epoch": 7.116317749023437e-06,
      "step": 7462,
      "training_step_time": 8.551871299743652
    },
    {
      "epoch": 7.117271423339844e-06,
      "model_forward_time": 1.7209157943725586,
      "step": 7463
    },
    {
      "epoch": 7.117271423339844e-06,
      "step": 7463,
      "training_step_time": 8.742750644683838
    },
    {
      "epoch": 7.11822509765625e-06,
      "model_forward_time": 2.291163682937622,
      "step": 7464
    },
    {
      "epoch": 7.11822509765625e-06,
      "step": 7464,
      "training_step_time": 8.925163269042969
    },
    {
      "epoch": 7.119178771972656e-06,
      "model_forward_time": 1.7257790565490723,
      "step": 7465
    },
    {
      "epoch": 7.119178771972656e-06,
      "step": 7465,
      "training_step_time": 8.528690576553345
    },
    {
      "epoch": 7.120132446289063e-06,
      "model_forward_time": 1.7130279541015625,
      "step": 7466
    },
    {
      "epoch": 7.120132446289063e-06,
      "step": 7466,
      "training_step_time": 7.711308717727661
    },
    {
      "epoch": 7.121086120605468e-06,
      "model_forward_time": 1.7275328636169434,
      "step": 7467
    },
    {
      "epoch": 7.121086120605468e-06,
      "step": 7467,
      "training_step_time": 9.966178178787231
    },
    {
      "epoch": 7.122039794921875e-06,
      "model_forward_time": 1.7230806350708008,
      "step": 7468
    },
    {
      "epoch": 7.122039794921875e-06,
      "step": 7468,
      "training_step_time": 7.554202556610107
    },
    {
      "epoch": 7.1229934692382815e-06,
      "model_forward_time": 1.7247710227966309,
      "step": 7469
    },
    {
      "epoch": 7.1229934692382815e-06,
      "step": 7469,
      "training_step_time": 7.725821495056152
    },
    {
      "epoch": 7.123947143554687e-06,
      "grad_norm": 0.26217034459114075,
      "learning_rate": 9.983343128206664e-06,
      "loss": 0.0687,
      "step": 7470
    },
    {
      "epoch": 7.123947143554687e-06,
      "model_forward_time": 1.7336156368255615,
      "step": 7470
    },
    {
      "dynamics_loss_avg": 0.06007835697382689,
      "epoch": 7.123947143554687e-06,
      "step": 7470
    },
    {
      "action_loss_avg": 0.005637088231742382,
      "epoch": 7.123947143554687e-06,
      "step": 7470
    },
    {
      "epoch": 7.123947143554687e-06,
      "step": 7470,
      "training_step_time": 7.636706590652466
    },
    {
      "epoch": 7.124900817871094e-06,
      "model_forward_time": 1.7217795848846436,
      "step": 7471
    },
    {
      "epoch": 7.124900817871094e-06,
      "step": 7471,
      "training_step_time": 7.726570129394531
    },
    {
      "epoch": 7.1258544921875e-06,
      "model_forward_time": 1.7260236740112305,
      "step": 7472
    },
    {
      "epoch": 7.1258544921875e-06,
      "step": 7472,
      "training_step_time": 10.6035475730896
    },
    {
      "epoch": 7.126808166503906e-06,
      "model_forward_time": 2.3037073612213135,
      "step": 7473
    },
    {
      "epoch": 7.126808166503906e-06,
      "step": 7473,
      "training_step_time": 8.950062990188599
    },
    {
      "epoch": 7.127761840820313e-06,
      "model_forward_time": 1.7151718139648438,
      "step": 7474
    },
    {
      "epoch": 7.127761840820313e-06,
      "step": 7474,
      "training_step_time": 7.837718725204468
    },
    {
      "epoch": 7.128715515136718e-06,
      "model_forward_time": 1.7195580005645752,
      "step": 7475
    },
    {
      "epoch": 7.128715515136718e-06,
      "step": 7475,
      "training_step_time": 7.783784627914429
    },
    {
      "epoch": 7.129669189453125e-06,
      "model_forward_time": 1.727952241897583,
      "step": 7476
    },
    {
      "epoch": 7.129669189453125e-06,
      "step": 7476,
      "training_step_time": 8.072190523147583
    },
    {
      "epoch": 7.1306228637695315e-06,
      "model_forward_time": 1.7176663875579834,
      "step": 7477
    },
    {
      "epoch": 7.1306228637695315e-06,
      "step": 7477,
      "training_step_time": 7.725849390029907
    },
    {
      "epoch": 7.131576538085937e-06,
      "model_forward_time": 1.7152016162872314,
      "step": 7478
    },
    {
      "epoch": 7.131576538085937e-06,
      "step": 7478,
      "training_step_time": 7.830159902572632
    },
    {
      "epoch": 7.132530212402344e-06,
      "model_forward_time": 1.721921443939209,
      "step": 7479
    },
    {
      "epoch": 7.132530212402344e-06,
      "step": 7479,
      "training_step_time": 7.664595365524292
    },
    {
      "epoch": 7.13348388671875e-06,
      "grad_norm": 0.2475145012140274,
      "learning_rate": 9.98320800260705e-06,
      "loss": 0.0683,
      "step": 7480
    },
    {
      "epoch": 7.13348388671875e-06,
      "model_forward_time": 1.7207093238830566,
      "step": 7480
    },
    {
      "dynamics_loss_avg": 0.06854884475469589,
      "epoch": 7.13348388671875e-06,
      "step": 7480
    },
    {
      "action_loss_avg": 0.006666231586132198,
      "epoch": 7.13348388671875e-06,
      "step": 7480
    },
    {
      "epoch": 7.13348388671875e-06,
      "step": 7480,
      "training_step_time": 7.6747214794158936
    },
    {
      "epoch": 7.134437561035156e-06,
      "model_forward_time": 1.7173807621002197,
      "step": 7481
    },
    {
      "epoch": 7.134437561035156e-06,
      "step": 7481,
      "training_step_time": 9.770994424819946
    },
    {
      "epoch": 7.135391235351563e-06,
      "model_forward_time": 2.2910845279693604,
      "step": 7482
    },
    {
      "epoch": 7.135391235351563e-06,
      "step": 7482,
      "training_step_time": 7.965829133987427
    },
    {
      "epoch": 7.136344909667968e-06,
      "model_forward_time": 1.7253446578979492,
      "step": 7483
    },
    {
      "epoch": 7.136344909667968e-06,
      "step": 7483,
      "training_step_time": 7.875980854034424
    },
    {
      "epoch": 7.137298583984375e-06,
      "model_forward_time": 1.719071626663208,
      "step": 7484
    },
    {
      "epoch": 7.137298583984375e-06,
      "step": 7484,
      "training_step_time": 7.811490774154663
    },
    {
      "epoch": 7.1382522583007816e-06,
      "model_forward_time": 1.7113494873046875,
      "step": 7485
    },
    {
      "epoch": 7.1382522583007816e-06,
      "step": 7485,
      "training_step_time": 8.66563105583191
    },
    {
      "epoch": 7.139205932617187e-06,
      "model_forward_time": 1.714838981628418,
      "step": 7486
    },
    {
      "epoch": 7.139205932617187e-06,
      "step": 7486,
      "training_step_time": 7.765841245651245
    },
    {
      "epoch": 7.140159606933594e-06,
      "model_forward_time": 1.7118744850158691,
      "step": 7487
    },
    {
      "epoch": 7.140159606933594e-06,
      "step": 7487,
      "training_step_time": 8.88248085975647
    },
    {
      "epoch": 7.14111328125e-06,
      "model_forward_time": 1.7072410583496094,
      "step": 7488
    },
    {
      "epoch": 7.14111328125e-06,
      "step": 7488,
      "training_step_time": 7.73230242729187
    },
    {
      "epoch": 7.142066955566406e-06,
      "model_forward_time": 1.716627836227417,
      "step": 7489
    },
    {
      "epoch": 7.142066955566406e-06,
      "step": 7489,
      "training_step_time": 7.660050868988037
    },
    {
      "epoch": 7.143020629882813e-06,
      "grad_norm": 0.23817788064479828,
      "learning_rate": 9.9830723320513e-06,
      "loss": 0.0689,
      "step": 7490
    },
    {
      "epoch": 7.143020629882813e-06,
      "model_forward_time": 1.7080082893371582,
      "step": 7490
    },
    {
      "dynamics_loss_avg": 0.06115446956828237,
      "epoch": 7.143020629882813e-06,
      "step": 7490
    },
    {
      "action_loss_avg": 0.005092800094280392,
      "epoch": 7.143020629882813e-06,
      "step": 7490
    },
    {
      "epoch": 7.143020629882813e-06,
      "step": 7490,
      "training_step_time": 9.641744613647461
    },
    {
      "epoch": 7.1439743041992184e-06,
      "model_forward_time": 1.751443862915039,
      "step": 7491
    },
    {
      "epoch": 7.1439743041992184e-06,
      "step": 7491,
      "training_step_time": 9.596776962280273
    },
    {
      "epoch": 7.144927978515625e-06,
      "model_forward_time": 2.235680103302002,
      "step": 7492
    },
    {
      "epoch": 7.144927978515625e-06,
      "step": 7492,
      "training_step_time": 8.888510465621948
    },
    {
      "epoch": 7.1458816528320316e-06,
      "model_forward_time": 1.7138328552246094,
      "step": 7493
    },
    {
      "epoch": 7.1458816528320316e-06,
      "step": 7493,
      "training_step_time": 7.768750905990601
    },
    {
      "epoch": 7.146835327148437e-06,
      "model_forward_time": 1.7119498252868652,
      "step": 7494
    },
    {
      "epoch": 7.146835327148437e-06,
      "step": 7494,
      "training_step_time": 8.030070304870605
    },
    {
      "epoch": 7.147789001464844e-06,
      "model_forward_time": 1.7160601615905762,
      "step": 7495
    },
    {
      "epoch": 7.147789001464844e-06,
      "step": 7495,
      "training_step_time": 7.670807361602783
    },
    {
      "epoch": 7.14874267578125e-06,
      "model_forward_time": 1.7020294666290283,
      "step": 7496
    },
    {
      "epoch": 7.14874267578125e-06,
      "step": 7496,
      "training_step_time": 7.804121017456055
    },
    {
      "epoch": 7.149696350097656e-06,
      "model_forward_time": 1.7088899612426758,
      "step": 7497
    },
    {
      "epoch": 7.149696350097656e-06,
      "step": 7497,
      "training_step_time": 7.720120906829834
    },
    {
      "epoch": 7.150650024414063e-06,
      "model_forward_time": 1.7196738719940186,
      "step": 7498
    },
    {
      "epoch": 7.150650024414063e-06,
      "step": 7498,
      "training_step_time": 7.675918102264404
    },
    {
      "epoch": 7.1516036987304684e-06,
      "model_forward_time": 1.7233881950378418,
      "step": 7499
    },
    {
      "epoch": 7.1516036987304684e-06,
      "step": 7499,
      "training_step_time": 7.776317119598389
    },
    {
      "epoch": 7.152557373046875e-06,
      "grad_norm": 0.20251263678073883,
      "learning_rate": 9.982936116554253e-06,
      "loss": 0.0675,
      "step": 7500
    },
    {
      "epoch": 7.152557373046875e-06,
      "model_forward_time": 1.7396254539489746,
      "step": 7500
    },
    {
      "dynamics_loss_avg": 0.06047760918736458,
      "epoch": 7.152557373046875e-06,
      "step": 7500
    },
    {
      "action_loss_avg": 0.006568082980811596,
      "epoch": 7.152557373046875e-06,
      "step": 7500
    },
    {
      "epoch": 7.152557373046875e-06,
      "step": 7500,
      "training_step_time": 9.628532886505127
    },
    {
      "epoch": 7.1535110473632816e-06,
      "model_forward_time": 2.247960329055786,
      "step": 7501
    },
    {
      "epoch": 7.1535110473632816e-06,
      "step": 7501,
      "training_step_time": 7.866696357727051
    },
    {
      "epoch": 7.154464721679687e-06,
      "model_forward_time": 1.7122230529785156,
      "step": 7502
    },
    {
      "epoch": 7.154464721679687e-06,
      "step": 7502,
      "training_step_time": 8.547625303268433
    },
    {
      "epoch": 7.155418395996094e-06,
      "model_forward_time": 1.705932378768921,
      "step": 7503
    },
    {
      "epoch": 7.155418395996094e-06,
      "step": 7503,
      "training_step_time": 8.685787200927734
    },
    {
      "epoch": 7.1563720703125e-06,
      "model_forward_time": 1.7084236145019531,
      "step": 7504
    },
    {
      "epoch": 7.1563720703125e-06,
      "step": 7504,
      "training_step_time": 7.692033052444458
    },
    {
      "epoch": 7.157325744628906e-06,
      "model_forward_time": 1.7194099426269531,
      "step": 7505
    },
    {
      "epoch": 7.157325744628906e-06,
      "step": 7505,
      "training_step_time": 7.747405767440796
    },
    {
      "epoch": 7.158279418945313e-06,
      "model_forward_time": 1.7102787494659424,
      "step": 7506
    },
    {
      "epoch": 7.158279418945313e-06,
      "step": 7506,
      "training_step_time": 7.716346263885498
    },
    {
      "epoch": 7.1592330932617184e-06,
      "model_forward_time": 1.7114362716674805,
      "step": 7507
    },
    {
      "epoch": 7.1592330932617184e-06,
      "step": 7507,
      "training_step_time": 8.320048809051514
    },
    {
      "epoch": 7.160186767578125e-06,
      "model_forward_time": 1.711190938949585,
      "step": 7508
    },
    {
      "epoch": 7.160186767578125e-06,
      "step": 7508,
      "training_step_time": 7.77282190322876
    },
    {
      "epoch": 7.1611404418945316e-06,
      "model_forward_time": 1.7146332263946533,
      "step": 7509
    },
    {
      "epoch": 7.1611404418945316e-06,
      "step": 7509,
      "training_step_time": 8.635510683059692
    },
    {
      "epoch": 7.162094116210937e-06,
      "grad_norm": 0.28969234228134155,
      "learning_rate": 9.982799356130804e-06,
      "loss": 0.0729,
      "step": 7510
    },
    {
      "epoch": 7.162094116210937e-06,
      "model_forward_time": 2.267827272415161,
      "step": 7510
    },
    {
      "dynamics_loss_avg": 0.0825374685227871,
      "epoch": 7.162094116210937e-06,
      "step": 7510
    },
    {
      "action_loss_avg": 0.007146501354873181,
      "epoch": 7.162094116210937e-06,
      "step": 7510
    },
    {
      "epoch": 7.162094116210937e-06,
      "step": 7510,
      "training_step_time": 9.646886110305786
    },
    {
      "epoch": 7.163047790527344e-06,
      "model_forward_time": 1.712536096572876,
      "step": 7511
    },
    {
      "epoch": 7.163047790527344e-06,
      "step": 7511,
      "training_step_time": 8.228675842285156
    },
    {
      "epoch": 7.16400146484375e-06,
      "model_forward_time": 1.7070953845977783,
      "step": 7512
    },
    {
      "epoch": 7.16400146484375e-06,
      "step": 7512,
      "training_step_time": 7.718621492385864
    },
    {
      "epoch": 7.164955139160156e-06,
      "model_forward_time": 1.7127418518066406,
      "step": 7513
    },
    {
      "epoch": 7.164955139160156e-06,
      "step": 7513,
      "training_step_time": 8.827174663543701
    },
    {
      "epoch": 7.165908813476563e-06,
      "model_forward_time": 1.704554557800293,
      "step": 7514
    },
    {
      "epoch": 7.165908813476563e-06,
      "step": 7514,
      "training_step_time": 8.219844102859497
    },
    {
      "epoch": 7.1668624877929684e-06,
      "model_forward_time": 1.7115418910980225,
      "step": 7515
    },
    {
      "epoch": 7.1668624877929684e-06,
      "step": 7515,
      "training_step_time": 7.7389631271362305
    },
    {
      "epoch": 7.167816162109375e-06,
      "model_forward_time": 1.0109055042266846,
      "step": 7516
    },
    {
      "epoch": 7.167816162109375e-06,
      "step": 7516,
      "training_step_time": 7.631924867630005
    },
    {
      "epoch": 7.168769836425782e-06,
      "model_forward_time": 1.7197866439819336,
      "step": 7517
    },
    {
      "epoch": 7.168769836425782e-06,
      "step": 7517,
      "training_step_time": 8.750627517700195
    },
    {
      "epoch": 7.169723510742187e-06,
      "model_forward_time": 1.353865623474121,
      "step": 7518
    },
    {
      "epoch": 7.169723510742187e-06,
      "step": 7518,
      "training_step_time": 8.883933067321777
    },
    {
      "epoch": 7.170677185058594e-06,
      "model_forward_time": 1.7788059711456299,
      "step": 7519
    },
    {
      "epoch": 7.170677185058594e-06,
      "step": 7519,
      "training_step_time": 8.465868711471558
    },
    {
      "epoch": 7.171630859375e-06,
      "grad_norm": 0.3030361831188202,
      "learning_rate": 9.982662050795908e-06,
      "loss": 0.0739,
      "step": 7520
    },
    {
      "epoch": 7.171630859375e-06,
      "model_forward_time": 2.2560901641845703,
      "step": 7520
    },
    {
      "dynamics_loss_avg": 0.06498001981526613,
      "epoch": 7.171630859375e-06,
      "step": 7520
    },
    {
      "action_loss_avg": 0.004394655546639115,
      "epoch": 7.171630859375e-06,
      "step": 7520
    },
    {
      "epoch": 7.171630859375e-06,
      "step": 7520,
      "training_step_time": 8.875908613204956
    },
    {
      "epoch": 7.172584533691406e-06,
      "model_forward_time": 1.708259105682373,
      "step": 7521
    },
    {
      "epoch": 7.172584533691406e-06,
      "step": 7521,
      "training_step_time": 7.774569988250732
    },
    {
      "epoch": 7.173538208007813e-06,
      "model_forward_time": 1.7246019840240479,
      "step": 7522
    },
    {
      "epoch": 7.173538208007813e-06,
      "step": 7522,
      "training_step_time": 9.397700071334839
    },
    {
      "epoch": 7.1744918823242185e-06,
      "model_forward_time": 1.7141058444976807,
      "step": 7523
    },
    {
      "epoch": 7.1744918823242185e-06,
      "step": 7523,
      "training_step_time": 7.770509481430054
    },
    {
      "epoch": 7.175445556640625e-06,
      "model_forward_time": 1.721792459487915,
      "step": 7524
    },
    {
      "epoch": 7.175445556640625e-06,
      "step": 7524,
      "training_step_time": 7.851308345794678
    },
    {
      "epoch": 7.176399230957032e-06,
      "model_forward_time": 1.7207651138305664,
      "step": 7525
    },
    {
      "epoch": 7.176399230957032e-06,
      "step": 7525,
      "training_step_time": 7.700347661972046
    },
    {
      "epoch": 7.177352905273437e-06,
      "model_forward_time": 1.7129950523376465,
      "step": 7526
    },
    {
      "epoch": 7.177352905273437e-06,
      "step": 7526,
      "training_step_time": 7.744738340377808
    },
    {
      "epoch": 7.178306579589844e-06,
      "model_forward_time": 1.726954698562622,
      "step": 7527
    },
    {
      "epoch": 7.178306579589844e-06,
      "step": 7527,
      "training_step_time": 8.68755578994751
    },
    {
      "epoch": 7.17926025390625e-06,
      "model_forward_time": 1.7688329219818115,
      "step": 7528
    },
    {
      "epoch": 7.17926025390625e-06,
      "step": 7528,
      "training_step_time": 9.028785943984985
    },
    {
      "epoch": 7.180213928222656e-06,
      "model_forward_time": 2.254703998565674,
      "step": 7529
    },
    {
      "epoch": 7.180213928222656e-06,
      "step": 7529,
      "training_step_time": 7.905524015426636
    },
    {
      "epoch": 7.181167602539063e-06,
      "grad_norm": 0.2431817650794983,
      "learning_rate": 9.982524200564583e-06,
      "loss": 0.079,
      "step": 7530
    },
    {
      "epoch": 7.181167602539063e-06,
      "model_forward_time": 1.7063934803009033,
      "step": 7530
    },
    {
      "dynamics_loss_avg": 0.06465613394975663,
      "epoch": 7.181167602539063e-06,
      "step": 7530
    },
    {
      "action_loss_avg": 0.004731421009637415,
      "epoch": 7.181167602539063e-06,
      "step": 7530
    },
    {
      "epoch": 7.181167602539063e-06,
      "step": 7530,
      "training_step_time": 8.742761850357056
    },
    {
      "epoch": 7.1821212768554685e-06,
      "model_forward_time": 1.7234313488006592,
      "step": 7531
    },
    {
      "epoch": 7.1821212768554685e-06,
      "step": 7531,
      "training_step_time": 7.9101338386535645
    },
    {
      "epoch": 7.183074951171875e-06,
      "model_forward_time": 1.726203441619873,
      "step": 7532
    },
    {
      "epoch": 7.183074951171875e-06,
      "step": 7532,
      "training_step_time": 8.628884077072144
    },
    {
      "epoch": 7.184028625488282e-06,
      "model_forward_time": 1.7280173301696777,
      "step": 7533
    },
    {
      "epoch": 7.184028625488282e-06,
      "step": 7533,
      "training_step_time": 7.800203561782837
    },
    {
      "epoch": 7.184982299804687e-06,
      "model_forward_time": 1.7308008670806885,
      "step": 7534
    },
    {
      "epoch": 7.184982299804687e-06,
      "step": 7534,
      "training_step_time": 7.6436967849731445
    },
    {
      "epoch": 7.185935974121094e-06,
      "model_forward_time": 1.7296745777130127,
      "step": 7535
    },
    {
      "epoch": 7.185935974121094e-06,
      "step": 7535,
      "training_step_time": 7.642605543136597
    },
    {
      "epoch": 7.1868896484375e-06,
      "model_forward_time": 1.732048511505127,
      "step": 7536
    },
    {
      "epoch": 7.1868896484375e-06,
      "step": 7536,
      "training_step_time": 8.562310695648193
    },
    {
      "epoch": 7.187843322753906e-06,
      "model_forward_time": 1.7580499649047852,
      "step": 7537
    },
    {
      "epoch": 7.187843322753906e-06,
      "step": 7537,
      "training_step_time": 9.305436849594116
    },
    {
      "epoch": 7.188796997070313e-06,
      "model_forward_time": 2.2769229412078857,
      "step": 7538
    },
    {
      "epoch": 7.188796997070313e-06,
      "step": 7538,
      "training_step_time": 7.83905816078186
    },
    {
      "epoch": 7.1897506713867185e-06,
      "model_forward_time": 1.744821548461914,
      "step": 7539
    },
    {
      "epoch": 7.1897506713867185e-06,
      "step": 7539,
      "training_step_time": 8.748093843460083
    },
    {
      "epoch": 7.190704345703125e-06,
      "grad_norm": 0.39501041173934937,
      "learning_rate": 9.9823858054519e-06,
      "loss": 0.0804,
      "step": 7540
    },
    {
      "epoch": 7.190704345703125e-06,
      "model_forward_time": 1.715592622756958,
      "step": 7540
    },
    {
      "dynamics_loss_avg": 0.0764115707948804,
      "epoch": 7.190704345703125e-06,
      "step": 7540
    },
    {
      "action_loss_avg": 0.008890760387293995,
      "epoch": 7.190704345703125e-06,
      "step": 7540
    },
    {
      "epoch": 7.190704345703125e-06,
      "step": 7540,
      "training_step_time": 7.690677642822266
    },
    {
      "epoch": 7.191658020019532e-06,
      "model_forward_time": 1.7339177131652832,
      "step": 7541
    },
    {
      "epoch": 7.191658020019532e-06,
      "step": 7541,
      "training_step_time": 8.876054525375366
    },
    {
      "epoch": 7.192611694335937e-06,
      "model_forward_time": 1.7264866828918457,
      "step": 7542
    },
    {
      "epoch": 7.192611694335937e-06,
      "step": 7542,
      "training_step_time": 7.76652717590332
    },
    {
      "epoch": 7.193565368652344e-06,
      "model_forward_time": 1.7400505542755127,
      "step": 7543
    },
    {
      "epoch": 7.193565368652344e-06,
      "step": 7543,
      "training_step_time": 7.695120096206665
    },
    {
      "epoch": 7.19451904296875e-06,
      "model_forward_time": 1.7307803630828857,
      "step": 7544
    },
    {
      "epoch": 7.19451904296875e-06,
      "step": 7544,
      "training_step_time": 8.589277029037476
    },
    {
      "epoch": 7.195472717285156e-06,
      "model_forward_time": 1.7186567783355713,
      "step": 7545
    },
    {
      "epoch": 7.195472717285156e-06,
      "step": 7545,
      "training_step_time": 9.208474397659302
    },
    {
      "epoch": 7.196426391601563e-06,
      "model_forward_time": 1.7190303802490234,
      "step": 7546
    },
    {
      "epoch": 7.196426391601563e-06,
      "step": 7546,
      "training_step_time": 8.190646409988403
    },
    {
      "epoch": 7.1973800659179685e-06,
      "model_forward_time": 2.2527313232421875,
      "step": 7547
    },
    {
      "epoch": 7.1973800659179685e-06,
      "step": 7547,
      "training_step_time": 8.901618480682373
    },
    {
      "epoch": 7.198333740234375e-06,
      "model_forward_time": 1.7159392833709717,
      "step": 7548
    },
    {
      "epoch": 7.198333740234375e-06,
      "step": 7548,
      "training_step_time": 8.151515245437622
    },
    {
      "epoch": 7.199287414550782e-06,
      "model_forward_time": 1.7376234531402588,
      "step": 7549
    },
    {
      "epoch": 7.199287414550782e-06,
      "step": 7549,
      "training_step_time": 7.49665093421936
    },
    {
      "epoch": 7.200241088867187e-06,
      "grad_norm": 0.29865509271621704,
      "learning_rate": 9.982246865472998e-06,
      "loss": 0.0779,
      "step": 7550
    },
    {
      "epoch": 7.200241088867187e-06,
      "model_forward_time": 1.723203182220459,
      "step": 7550
    },
    {
      "dynamics_loss_avg": 0.0783020343631506,
      "epoch": 7.200241088867187e-06,
      "step": 7550
    },
    {
      "action_loss_avg": 0.008196697197854519,
      "epoch": 7.200241088867187e-06,
      "step": 7550
    },
    {
      "epoch": 7.200241088867187e-06,
      "step": 7550,
      "training_step_time": 7.7577338218688965
    },
    {
      "epoch": 7.201194763183594e-06,
      "model_forward_time": 1.720534324645996,
      "step": 7551
    },
    {
      "epoch": 7.201194763183594e-06,
      "step": 7551,
      "training_step_time": 7.687467575073242
    },
    {
      "epoch": 7.2021484375e-06,
      "model_forward_time": 1.7215569019317627,
      "step": 7552
    },
    {
      "epoch": 7.2021484375e-06,
      "step": 7552,
      "training_step_time": 7.613938093185425
    },
    {
      "epoch": 7.203102111816406e-06,
      "model_forward_time": 1.7280380725860596,
      "step": 7553
    },
    {
      "epoch": 7.203102111816406e-06,
      "step": 7553,
      "training_step_time": 8.36700963973999
    },
    {
      "epoch": 7.204055786132813e-06,
      "model_forward_time": 1.7183098793029785,
      "step": 7554
    },
    {
      "epoch": 7.204055786132813e-06,
      "step": 7554,
      "training_step_time": 7.64983868598938
    },
    {
      "epoch": 7.2050094604492185e-06,
      "model_forward_time": 1.7227857112884521,
      "step": 7555
    },
    {
      "epoch": 7.2050094604492185e-06,
      "step": 7555,
      "training_step_time": 7.584462642669678
    },
    {
      "epoch": 7.205963134765625e-06,
      "model_forward_time": 2.3209707736968994,
      "step": 7556
    },
    {
      "epoch": 7.205963134765625e-06,
      "step": 7556,
      "training_step_time": 8.312494993209839
    },
    {
      "epoch": 7.206916809082032e-06,
      "model_forward_time": 1.7213482856750488,
      "step": 7557
    },
    {
      "epoch": 7.206916809082032e-06,
      "step": 7557,
      "training_step_time": 7.667298793792725
    },
    {
      "epoch": 7.207870483398437e-06,
      "model_forward_time": 1.716973066329956,
      "step": 7558
    },
    {
      "epoch": 7.207870483398437e-06,
      "step": 7558,
      "training_step_time": 7.673654556274414
    },
    {
      "epoch": 7.208824157714844e-06,
      "model_forward_time": 1.7227461338043213,
      "step": 7559
    },
    {
      "epoch": 7.208824157714844e-06,
      "step": 7559,
      "training_step_time": 7.918819904327393
    },
    {
      "epoch": 7.20977783203125e-06,
      "grad_norm": 0.348749577999115,
      "learning_rate": 9.982107380643069e-06,
      "loss": 0.0742,
      "step": 7560
    },
    {
      "epoch": 7.20977783203125e-06,
      "model_forward_time": 1.7363340854644775,
      "step": 7560
    },
    {
      "dynamics_loss_avg": 0.06533885058015584,
      "epoch": 7.20977783203125e-06,
      "step": 7560
    },
    {
      "action_loss_avg": 0.005671273986808956,
      "epoch": 7.20977783203125e-06,
      "step": 7560
    },
    {
      "epoch": 7.20977783203125e-06,
      "step": 7560,
      "training_step_time": 8.59930682182312
    },
    {
      "epoch": 7.210731506347656e-06,
      "model_forward_time": 1.7227907180786133,
      "step": 7561
    },
    {
      "epoch": 7.210731506347656e-06,
      "step": 7561,
      "training_step_time": 7.624409914016724
    },
    {
      "epoch": 7.211685180664063e-06,
      "model_forward_time": 1.7185359001159668,
      "step": 7562
    },
    {
      "epoch": 7.211685180664063e-06,
      "step": 7562,
      "training_step_time": 8.336703777313232
    },
    {
      "epoch": 7.2126388549804685e-06,
      "model_forward_time": 1.720954418182373,
      "step": 7563
    },
    {
      "epoch": 7.2126388549804685e-06,
      "step": 7563,
      "training_step_time": 8.712936162948608
    },
    {
      "epoch": 7.213592529296875e-06,
      "model_forward_time": 1.7265069484710693,
      "step": 7564
    },
    {
      "epoch": 7.213592529296875e-06,
      "step": 7564,
      "training_step_time": 8.60810136795044
    },
    {
      "epoch": 7.214546203613282e-06,
      "model_forward_time": 2.304511308670044,
      "step": 7565
    },
    {
      "epoch": 7.214546203613282e-06,
      "step": 7565,
      "training_step_time": 8.372382640838623
    },
    {
      "epoch": 7.215499877929687e-06,
      "model_forward_time": 1.7266576290130615,
      "step": 7566
    },
    {
      "epoch": 7.215499877929687e-06,
      "step": 7566,
      "training_step_time": 8.883816719055176
    },
    {
      "epoch": 7.216453552246094e-06,
      "model_forward_time": 1.7080883979797363,
      "step": 7567
    },
    {
      "epoch": 7.216453552246094e-06,
      "step": 7567,
      "training_step_time": 8.243035316467285
    },
    {
      "epoch": 7.2174072265625e-06,
      "model_forward_time": 1.7419826984405518,
      "step": 7568
    },
    {
      "epoch": 7.2174072265625e-06,
      "step": 7568,
      "training_step_time": 7.837716579437256
    },
    {
      "epoch": 7.218360900878906e-06,
      "model_forward_time": 1.7195446491241455,
      "step": 7569
    },
    {
      "epoch": 7.218360900878906e-06,
      "step": 7569,
      "training_step_time": 7.680446624755859
    },
    {
      "epoch": 7.219314575195313e-06,
      "grad_norm": 0.23941701650619507,
      "learning_rate": 9.981967350977368e-06,
      "loss": 0.0796,
      "step": 7570
    },
    {
      "epoch": 7.219314575195313e-06,
      "model_forward_time": 1.717907190322876,
      "step": 7570
    },
    {
      "dynamics_loss_avg": 0.09074065834283829,
      "epoch": 7.219314575195313e-06,
      "step": 7570
    },
    {
      "action_loss_avg": 0.007000263687223196,
      "epoch": 7.219314575195313e-06,
      "step": 7570
    },
    {
      "epoch": 7.219314575195313e-06,
      "step": 7570,
      "training_step_time": 8.256035327911377
    },
    {
      "epoch": 7.2202682495117185e-06,
      "model_forward_time": 1.723292350769043,
      "step": 7571
    },
    {
      "epoch": 7.2202682495117185e-06,
      "step": 7571,
      "training_step_time": 7.571485757827759
    },
    {
      "epoch": 7.221221923828125e-06,
      "model_forward_time": 1.7231602668762207,
      "step": 7572
    },
    {
      "epoch": 7.221221923828125e-06,
      "step": 7572,
      "training_step_time": 7.788638353347778
    },
    {
      "epoch": 7.222175598144532e-06,
      "model_forward_time": 1.7255609035491943,
      "step": 7573
    },
    {
      "epoch": 7.222175598144532e-06,
      "step": 7573,
      "training_step_time": 8.12064504623413
    },
    {
      "epoch": 7.223129272460937e-06,
      "model_forward_time": 2.2984440326690674,
      "step": 7574
    },
    {
      "epoch": 7.223129272460937e-06,
      "step": 7574,
      "training_step_time": 8.891406059265137
    },
    {
      "epoch": 7.224082946777344e-06,
      "model_forward_time": 1.7338347434997559,
      "step": 7575
    },
    {
      "epoch": 7.224082946777344e-06,
      "step": 7575,
      "training_step_time": 7.829128265380859
    },
    {
      "epoch": 7.22503662109375e-06,
      "model_forward_time": 1.7324457168579102,
      "step": 7576
    },
    {
      "epoch": 7.22503662109375e-06,
      "step": 7576,
      "training_step_time": 7.7903828620910645
    },
    {
      "epoch": 7.225990295410156e-06,
      "model_forward_time": 1.7127978801727295,
      "step": 7577
    },
    {
      "epoch": 7.225990295410156e-06,
      "step": 7577,
      "training_step_time": 7.9025187492370605
    },
    {
      "epoch": 7.226943969726563e-06,
      "model_forward_time": 1.7163207530975342,
      "step": 7578
    },
    {
      "epoch": 7.226943969726563e-06,
      "step": 7578,
      "training_step_time": 7.671745538711548
    },
    {
      "epoch": 7.2278976440429685e-06,
      "model_forward_time": 1.7152924537658691,
      "step": 7579
    },
    {
      "epoch": 7.2278976440429685e-06,
      "step": 7579,
      "training_step_time": 7.6877851486206055
    },
    {
      "epoch": 7.228851318359375e-06,
      "grad_norm": 0.31447941064834595,
      "learning_rate": 9.981826776491208e-06,
      "loss": 0.073,
      "step": 7580
    },
    {
      "epoch": 7.228851318359375e-06,
      "model_forward_time": 1.723886489868164,
      "step": 7580
    },
    {
      "dynamics_loss_avg": 0.06901810299605131,
      "epoch": 7.228851318359375e-06,
      "step": 7580
    },
    {
      "action_loss_avg": 0.008475601195823402,
      "epoch": 7.228851318359375e-06,
      "step": 7580
    },
    {
      "epoch": 7.228851318359375e-06,
      "step": 7580,
      "training_step_time": 7.657130241394043
    },
    {
      "epoch": 7.229804992675782e-06,
      "model_forward_time": 1.722099781036377,
      "step": 7581
    },
    {
      "epoch": 7.229804992675782e-06,
      "step": 7581,
      "training_step_time": 7.761097431182861
    },
    {
      "epoch": 7.230758666992187e-06,
      "model_forward_time": 1.7222585678100586,
      "step": 7582
    },
    {
      "epoch": 7.230758666992187e-06,
      "step": 7582,
      "training_step_time": 8.501399993896484
    },
    {
      "epoch": 7.231712341308594e-06,
      "model_forward_time": 2.2608513832092285,
      "step": 7583
    },
    {
      "epoch": 7.231712341308594e-06,
      "step": 7583,
      "training_step_time": 8.848933458328247
    },
    {
      "epoch": 7.232666015625e-06,
      "model_forward_time": 1.7761988639831543,
      "step": 7584
    },
    {
      "epoch": 7.232666015625e-06,
      "step": 7584,
      "training_step_time": 8.839277505874634
    },
    {
      "epoch": 7.233619689941406e-06,
      "model_forward_time": 1.7183983325958252,
      "step": 7585
    },
    {
      "epoch": 7.233619689941406e-06,
      "step": 7585,
      "training_step_time": 8.372723579406738
    },
    {
      "epoch": 7.234573364257813e-06,
      "model_forward_time": 1.7217719554901123,
      "step": 7586
    },
    {
      "epoch": 7.234573364257813e-06,
      "step": 7586,
      "training_step_time": 8.708613395690918
    },
    {
      "epoch": 7.2355270385742185e-06,
      "model_forward_time": 1.716799259185791,
      "step": 7587
    },
    {
      "epoch": 7.2355270385742185e-06,
      "step": 7587,
      "training_step_time": 7.631748914718628
    },
    {
      "epoch": 7.236480712890625e-06,
      "model_forward_time": 1.34153151512146,
      "step": 7588
    },
    {
      "epoch": 7.236480712890625e-06,
      "step": 7588,
      "training_step_time": 7.849319219589233
    },
    {
      "epoch": 7.237434387207032e-06,
      "model_forward_time": 1.7040255069732666,
      "step": 7589
    },
    {
      "epoch": 7.237434387207032e-06,
      "step": 7589,
      "training_step_time": 7.712537527084351
    },
    {
      "epoch": 7.238388061523437e-06,
      "grad_norm": 0.2611893117427826,
      "learning_rate": 9.98168565719996e-06,
      "loss": 0.0789,
      "step": 7590
    },
    {
      "epoch": 7.238388061523437e-06,
      "model_forward_time": 1.7131648063659668,
      "step": 7590
    },
    {
      "dynamics_loss_avg": 0.08061427809298038,
      "epoch": 7.238388061523437e-06,
      "step": 7590
    },
    {
      "action_loss_avg": 0.006493702717125416,
      "epoch": 7.238388061523437e-06,
      "step": 7590
    },
    {
      "epoch": 7.238388061523437e-06,
      "step": 7590,
      "training_step_time": 8.822236776351929
    },
    {
      "epoch": 7.239341735839844e-06,
      "model_forward_time": 1.7180397510528564,
      "step": 7591
    },
    {
      "epoch": 7.239341735839844e-06,
      "step": 7591,
      "training_step_time": 7.742711782455444
    },
    {
      "epoch": 7.24029541015625e-06,
      "model_forward_time": 2.244241237640381,
      "step": 7592
    },
    {
      "epoch": 7.24029541015625e-06,
      "step": 7592,
      "training_step_time": 8.399500846862793
    },
    {
      "epoch": 7.241249084472656e-06,
      "model_forward_time": 1.4135735034942627,
      "step": 7593
    },
    {
      "epoch": 7.241249084472656e-06,
      "step": 7593,
      "training_step_time": 8.663223028182983
    },
    {
      "epoch": 7.242202758789063e-06,
      "model_forward_time": 1.7157659530639648,
      "step": 7594
    },
    {
      "epoch": 7.242202758789063e-06,
      "step": 7594,
      "training_step_time": 8.785202264785767
    },
    {
      "epoch": 7.2431564331054685e-06,
      "model_forward_time": 1.7089118957519531,
      "step": 7595
    },
    {
      "epoch": 7.2431564331054685e-06,
      "step": 7595,
      "training_step_time": 7.8655571937561035
    },
    {
      "epoch": 7.244110107421875e-06,
      "model_forward_time": 1.7096917629241943,
      "step": 7596
    },
    {
      "epoch": 7.244110107421875e-06,
      "step": 7596,
      "training_step_time": 8.14055871963501
    },
    {
      "epoch": 7.245063781738282e-06,
      "model_forward_time": 1.734884262084961,
      "step": 7597
    },
    {
      "epoch": 7.245063781738282e-06,
      "step": 7597,
      "training_step_time": 7.834840536117554
    },
    {
      "epoch": 7.246017456054687e-06,
      "model_forward_time": 1.730151653289795,
      "step": 7598
    },
    {
      "epoch": 7.246017456054687e-06,
      "step": 7598,
      "training_step_time": 8.418200254440308
    },
    {
      "epoch": 7.246971130371094e-06,
      "model_forward_time": 1.734757900238037,
      "step": 7599
    },
    {
      "epoch": 7.246971130371094e-06,
      "step": 7599,
      "training_step_time": 7.728704929351807
    },
    {
      "epoch": 7.2479248046875e-06,
      "grad_norm": 0.23886755108833313,
      "learning_rate": 9.981543993119061e-06,
      "loss": 0.0662,
      "step": 7600
    },
    {
      "epoch": 7.2479248046875e-06,
      "model_forward_time": 1.7270822525024414,
      "step": 7600
    },
    {
      "dynamics_loss_avg": 0.06890691854059697,
      "epoch": 7.2479248046875e-06,
      "step": 7600
    },
    {
      "action_loss_avg": 0.006884258822537959,
      "epoch": 7.2479248046875e-06,
      "step": 7600
    },
    {
      "epoch": 7.2479248046875e-06,
      "step": 7600,
      "training_step_time": 7.741518497467041
    },
    {
      "epoch": 7.248878479003906e-06,
      "model_forward_time": 2.2742342948913574,
      "step": 7601
    },
    {
      "epoch": 7.248878479003906e-06,
      "step": 7601,
      "training_step_time": 7.744469881057739
    },
    {
      "epoch": 7.249832153320313e-06,
      "model_forward_time": 1.7846708297729492,
      "step": 7602
    },
    {
      "epoch": 7.249832153320313e-06,
      "step": 7602,
      "training_step_time": 7.838847875595093
    },
    {
      "epoch": 7.2507858276367185e-06,
      "model_forward_time": 1.7318470478057861,
      "step": 7603
    },
    {
      "epoch": 7.2507858276367185e-06,
      "step": 7603,
      "training_step_time": 8.551858901977539
    },
    {
      "epoch": 7.251739501953125e-06,
      "model_forward_time": 1.7263984680175781,
      "step": 7604
    },
    {
      "epoch": 7.251739501953125e-06,
      "step": 7604,
      "training_step_time": 7.700430393218994
    },
    {
      "epoch": 7.252693176269532e-06,
      "model_forward_time": 1.7323882579803467,
      "step": 7605
    },
    {
      "epoch": 7.252693176269532e-06,
      "step": 7605,
      "training_step_time": 7.711196660995483
    },
    {
      "epoch": 7.253646850585937e-06,
      "model_forward_time": 1.744136095046997,
      "step": 7606
    },
    {
      "epoch": 7.253646850585937e-06,
      "step": 7606,
      "training_step_time": 8.818713426589966
    },
    {
      "epoch": 7.254600524902344e-06,
      "model_forward_time": 1.7286021709442139,
      "step": 7607
    },
    {
      "epoch": 7.254600524902344e-06,
      "step": 7607,
      "training_step_time": 7.677662372589111
    },
    {
      "epoch": 7.25555419921875e-06,
      "model_forward_time": 1.7297136783599854,
      "step": 7608
    },
    {
      "epoch": 7.25555419921875e-06,
      "step": 7608,
      "training_step_time": 7.73104453086853
    },
    {
      "epoch": 7.256507873535156e-06,
      "model_forward_time": 1.7427217960357666,
      "step": 7609
    },
    {
      "epoch": 7.256507873535156e-06,
      "step": 7609,
      "training_step_time": 7.753523111343384
    },
    {
      "epoch": 7.257461547851563e-06,
      "grad_norm": 0.24838443100452423,
      "learning_rate": 9.981401784263998e-06,
      "loss": 0.0751,
      "step": 7610
    },
    {
      "epoch": 7.257461547851563e-06,
      "model_forward_time": 2.2577903270721436,
      "step": 7610
    },
    {
      "dynamics_loss_avg": 0.08369631823152304,
      "epoch": 7.257461547851563e-06,
      "step": 7610
    },
    {
      "action_loss_avg": 0.006280830106697976,
      "epoch": 7.257461547851563e-06,
      "step": 7610
    },
    {
      "epoch": 7.257461547851563e-06,
      "step": 7610,
      "training_step_time": 7.811969518661499
    },
    {
      "epoch": 7.2584152221679685e-06,
      "model_forward_time": 1.7885076999664307,
      "step": 7611
    },
    {
      "epoch": 7.2584152221679685e-06,
      "step": 7611,
      "training_step_time": 9.563197135925293
    },
    {
      "epoch": 7.259368896484375e-06,
      "model_forward_time": 1.7306172847747803,
      "step": 7612
    },
    {
      "epoch": 7.259368896484375e-06,
      "step": 7612,
      "training_step_time": 7.868267059326172
    },
    {
      "epoch": 7.260322570800782e-06,
      "model_forward_time": 1.728088617324829,
      "step": 7613
    },
    {
      "epoch": 7.260322570800782e-06,
      "step": 7613,
      "training_step_time": 7.774922132492065
    },
    {
      "epoch": 7.261276245117187e-06,
      "model_forward_time": 1.7207846641540527,
      "step": 7614
    },
    {
      "epoch": 7.261276245117187e-06,
      "step": 7614,
      "training_step_time": 8.953105688095093
    },
    {
      "epoch": 7.262229919433594e-06,
      "model_forward_time": 1.7336664199829102,
      "step": 7615
    },
    {
      "epoch": 7.262229919433594e-06,
      "step": 7615,
      "training_step_time": 8.803450584411621
    },
    {
      "epoch": 7.26318359375e-06,
      "model_forward_time": 1.7424156665802002,
      "step": 7616
    },
    {
      "epoch": 7.26318359375e-06,
      "step": 7616,
      "training_step_time": 9.250594854354858
    },
    {
      "epoch": 7.264137268066406e-06,
      "model_forward_time": 1.7200145721435547,
      "step": 7617
    },
    {
      "epoch": 7.264137268066406e-06,
      "step": 7617,
      "training_step_time": 6.960516929626465
    },
    {
      "epoch": 7.265090942382813e-06,
      "model_forward_time": 1.7239210605621338,
      "step": 7618
    },
    {
      "epoch": 7.265090942382813e-06,
      "step": 7618,
      "training_step_time": 6.9722137451171875
    },
    {
      "epoch": 7.2660446166992185e-06,
      "model_forward_time": 1.3636419773101807,
      "step": 7619
    },
    {
      "epoch": 7.2660446166992185e-06,
      "step": 7619,
      "training_step_time": 6.975646018981934
    },
    {
      "epoch": 7.266998291015625e-06,
      "grad_norm": 0.36961254477500916,
      "learning_rate": 9.981259030650326e-06,
      "loss": 0.075,
      "step": 7620
    },
    {
      "epoch": 7.266998291015625e-06,
      "model_forward_time": 1.7167305946350098,
      "step": 7620
    },
    {
      "dynamics_loss_avg": 0.057053740695118906,
      "epoch": 7.266998291015625e-06,
      "step": 7620
    },
    {
      "action_loss_avg": 0.004226303671021015,
      "epoch": 7.266998291015625e-06,
      "step": 7620
    },
    {
      "epoch": 7.266998291015625e-06,
      "step": 7620,
      "training_step_time": 6.9364173412323
    },
    {
      "epoch": 7.267951965332032e-06,
      "model_forward_time": 1.7319326400756836,
      "step": 7621
    },
    {
      "epoch": 7.267951965332032e-06,
      "step": 7621,
      "training_step_time": 6.9577813148498535
    },
    {
      "epoch": 7.268905639648437e-06,
      "model_forward_time": 1.7295801639556885,
      "step": 7622
    },
    {
      "epoch": 7.268905639648437e-06,
      "step": 7622,
      "training_step_time": 6.904645681381226
    },
    {
      "epoch": 7.269859313964844e-06,
      "model_forward_time": 1.7387685775756836,
      "step": 7623
    },
    {
      "epoch": 7.269859313964844e-06,
      "step": 7623,
      "training_step_time": 6.962606191635132
    },
    {
      "epoch": 7.27081298828125e-06,
      "model_forward_time": 1.7349891662597656,
      "step": 7624
    },
    {
      "epoch": 7.27081298828125e-06,
      "step": 7624,
      "training_step_time": 6.94739556312561
    },
    {
      "epoch": 7.271766662597656e-06,
      "model_forward_time": 2.2613308429718018,
      "step": 7625
    },
    {
      "epoch": 7.271766662597656e-06,
      "step": 7625,
      "training_step_time": 7.234357118606567
    },
    {
      "epoch": 7.272720336914063e-06,
      "model_forward_time": 1.7288634777069092,
      "step": 7626
    },
    {
      "epoch": 7.272720336914063e-06,
      "step": 7626,
      "training_step_time": 7.999382257461548
    },
    {
      "epoch": 7.2736740112304685e-06,
      "model_forward_time": 1.377129077911377,
      "step": 7627
    },
    {
      "epoch": 7.2736740112304685e-06,
      "step": 7627,
      "training_step_time": 6.954609155654907
    },
    {
      "epoch": 7.274627685546875e-06,
      "model_forward_time": 1.7404630184173584,
      "step": 7628
    },
    {
      "epoch": 7.274627685546875e-06,
      "step": 7628,
      "training_step_time": 6.900237083435059
    },
    {
      "epoch": 7.275581359863281e-06,
      "model_forward_time": 1.7335681915283203,
      "step": 7629
    },
    {
      "epoch": 7.275581359863281e-06,
      "step": 7629,
      "training_step_time": 6.916231632232666
    },
    {
      "epoch": 7.276535034179687e-06,
      "grad_norm": 0.27719348669052124,
      "learning_rate": 9.981115732293655e-06,
      "loss": 0.0761,
      "step": 7630
    },
    {
      "epoch": 7.276535034179687e-06,
      "model_forward_time": 1.7398784160614014,
      "step": 7630
    },
    {
      "dynamics_loss_avg": 0.054076645709574225,
      "epoch": 7.276535034179687e-06,
      "step": 7630
    },
    {
      "action_loss_avg": 0.004552621580660343,
      "epoch": 7.276535034179687e-06,
      "step": 7630
    },
    {
      "epoch": 7.276535034179687e-06,
      "step": 7630,
      "training_step_time": 6.879278182983398
    },
    {
      "epoch": 7.277488708496094e-06,
      "model_forward_time": 1.7230660915374756,
      "step": 7631
    },
    {
      "epoch": 7.277488708496094e-06,
      "step": 7631,
      "training_step_time": 6.899582624435425
    },
    {
      "epoch": 7.2784423828125e-06,
      "model_forward_time": 1.719334602355957,
      "step": 7632
    },
    {
      "epoch": 7.2784423828125e-06,
      "step": 7632,
      "training_step_time": 6.957772493362427
    },
    {
      "epoch": 7.279396057128906e-06,
      "model_forward_time": 1.3624799251556396,
      "step": 7633
    },
    {
      "epoch": 7.279396057128906e-06,
      "step": 7633,
      "training_step_time": 6.890766382217407
    },
    {
      "epoch": 7.280349731445313e-06,
      "model_forward_time": 2.268449068069458,
      "step": 7634
    },
    {
      "epoch": 7.280349731445313e-06,
      "step": 7634,
      "training_step_time": 7.731418132781982
    },
    {
      "epoch": 7.2813034057617185e-06,
      "model_forward_time": 1.7447590827941895,
      "step": 7635
    },
    {
      "epoch": 7.2813034057617185e-06,
      "step": 7635,
      "training_step_time": 7.961907625198364
    },
    {
      "epoch": 7.282257080078125e-06,
      "model_forward_time": 1.7259306907653809,
      "step": 7636
    },
    {
      "epoch": 7.282257080078125e-06,
      "step": 7636,
      "training_step_time": 7.374158620834351
    },
    {
      "epoch": 7.283210754394531e-06,
      "model_forward_time": 1.7322986125946045,
      "step": 7637
    },
    {
      "epoch": 7.283210754394531e-06,
      "step": 7637,
      "training_step_time": 6.864874839782715
    },
    {
      "epoch": 7.284164428710937e-06,
      "model_forward_time": 1.7354605197906494,
      "step": 7638
    },
    {
      "epoch": 7.284164428710937e-06,
      "step": 7638,
      "training_step_time": 6.87934136390686
    },
    {
      "epoch": 7.285118103027344e-06,
      "model_forward_time": 1.7407352924346924,
      "step": 7639
    },
    {
      "epoch": 7.285118103027344e-06,
      "step": 7639,
      "training_step_time": 6.923752546310425
    },
    {
      "epoch": 7.28607177734375e-06,
      "grad_norm": 0.41770580410957336,
      "learning_rate": 9.980971889209659e-06,
      "loss": 0.074,
      "step": 7640
    },
    {
      "epoch": 7.28607177734375e-06,
      "model_forward_time": 1.7302732467651367,
      "step": 7640
    },
    {
      "dynamics_loss_avg": 0.08033010140061378,
      "epoch": 7.28607177734375e-06,
      "step": 7640
    },
    {
      "action_loss_avg": 0.007451202243100852,
      "epoch": 7.28607177734375e-06,
      "step": 7640
    },
    {
      "epoch": 7.28607177734375e-06,
      "step": 7640,
      "training_step_time": 6.885351896286011
    },
    {
      "epoch": 7.287025451660156e-06,
      "model_forward_time": 1.739727258682251,
      "step": 7641
    },
    {
      "epoch": 7.287025451660156e-06,
      "step": 7641,
      "training_step_time": 6.873114585876465
    },
    {
      "epoch": 7.287979125976563e-06,
      "model_forward_time": 1.741145133972168,
      "step": 7642
    },
    {
      "epoch": 7.287979125976563e-06,
      "step": 7642,
      "training_step_time": 6.869768142700195
    },
    {
      "epoch": 7.2889328002929686e-06,
      "model_forward_time": 2.2943551540374756,
      "step": 7643
    },
    {
      "epoch": 7.2889328002929686e-06,
      "step": 7643,
      "training_step_time": 7.332430124282837
    },
    {
      "epoch": 7.289886474609375e-06,
      "model_forward_time": 1.7371585369110107,
      "step": 7644
    },
    {
      "epoch": 7.289886474609375e-06,
      "step": 7644,
      "training_step_time": 9.912869930267334
    },
    {
      "epoch": 7.290840148925781e-06,
      "model_forward_time": 1.7320044040679932,
      "step": 7645
    },
    {
      "epoch": 7.290840148925781e-06,
      "step": 7645,
      "training_step_time": 7.462877511978149
    },
    {
      "epoch": 7.291793823242187e-06,
      "model_forward_time": 1.7275316715240479,
      "step": 7646
    },
    {
      "epoch": 7.291793823242187e-06,
      "step": 7646,
      "training_step_time": 6.75544285774231
    },
    {
      "epoch": 7.292747497558594e-06,
      "model_forward_time": 1.7235941886901855,
      "step": 7647
    },
    {
      "epoch": 7.292747497558594e-06,
      "step": 7647,
      "training_step_time": 6.899298906326294
    },
    {
      "epoch": 7.293701171875e-06,
      "model_forward_time": 1.7256312370300293,
      "step": 7648
    },
    {
      "epoch": 7.293701171875e-06,
      "step": 7648,
      "training_step_time": 6.701819658279419
    },
    {
      "epoch": 7.294654846191406e-06,
      "model_forward_time": 1.7197489738464355,
      "step": 7649
    },
    {
      "epoch": 7.294654846191406e-06,
      "step": 7649,
      "training_step_time": 6.94682240486145
    },
    {
      "epoch": 7.295608520507813e-06,
      "grad_norm": 0.35970234870910645,
      "learning_rate": 9.980827501414064e-06,
      "loss": 0.0737,
      "step": 7650
    },
    {
      "epoch": 7.295608520507813e-06,
      "model_forward_time": 1.7359497547149658,
      "step": 7650
    },
    {
      "dynamics_loss_avg": 0.06387761123478412,
      "epoch": 7.295608520507813e-06,
      "step": 7650
    },
    {
      "action_loss_avg": 0.0052734649390913544,
      "epoch": 7.295608520507813e-06,
      "step": 7650
    },
    {
      "epoch": 7.295608520507813e-06,
      "step": 7650,
      "training_step_time": 6.908881902694702
    },
    {
      "epoch": 7.2965621948242186e-06,
      "model_forward_time": 1.7353367805480957,
      "step": 7651
    },
    {
      "epoch": 7.2965621948242186e-06,
      "step": 7651,
      "training_step_time": 6.851308345794678
    },
    {
      "epoch": 7.297515869140625e-06,
      "model_forward_time": 1.7850720882415771,
      "step": 7652
    },
    {
      "epoch": 7.297515869140625e-06,
      "step": 7652,
      "training_step_time": 8.743118286132812
    },
    {
      "epoch": 7.298469543457031e-06,
      "model_forward_time": 1.7259461879730225,
      "step": 7653
    },
    {
      "epoch": 7.298469543457031e-06,
      "step": 7653,
      "training_step_time": 8.9419527053833
    },
    {
      "epoch": 7.299423217773437e-06,
      "model_forward_time": 1.7314720153808594,
      "step": 7654
    },
    {
      "epoch": 7.299423217773437e-06,
      "step": 7654,
      "training_step_time": 7.7646050453186035
    },
    {
      "epoch": 7.300376892089844e-06,
      "model_forward_time": 1.71592378616333,
      "step": 7655
    },
    {
      "epoch": 7.300376892089844e-06,
      "step": 7655,
      "training_step_time": 7.295247554779053
    },
    {
      "epoch": 7.30133056640625e-06,
      "model_forward_time": 1.72281813621521,
      "step": 7656
    },
    {
      "epoch": 7.30133056640625e-06,
      "step": 7656,
      "training_step_time": 6.949411392211914
    },
    {
      "epoch": 7.302284240722656e-06,
      "model_forward_time": 1.3653056621551514,
      "step": 7657
    },
    {
      "epoch": 7.302284240722656e-06,
      "step": 7657,
      "training_step_time": 6.949862003326416
    },
    {
      "epoch": 7.303237915039063e-06,
      "model_forward_time": 1.7236108779907227,
      "step": 7658
    },
    {
      "epoch": 7.303237915039063e-06,
      "step": 7658,
      "training_step_time": 7.018038511276245
    },
    {
      "epoch": 7.3041915893554686e-06,
      "model_forward_time": 1.7290897369384766,
      "step": 7659
    },
    {
      "epoch": 7.3041915893554686e-06,
      "step": 7659,
      "training_step_time": 6.946654558181763
    },
    {
      "epoch": 7.305145263671875e-06,
      "grad_norm": 0.32322174310684204,
      "learning_rate": 9.980682568922662e-06,
      "loss": 0.0698,
      "step": 7660
    },
    {
      "epoch": 7.305145263671875e-06,
      "model_forward_time": 1.747987985610962,
      "step": 7660
    },
    {
      "dynamics_loss_avg": 0.06980005614459514,
      "epoch": 7.305145263671875e-06,
      "step": 7660
    },
    {
      "action_loss_avg": 0.007887775532435626,
      "epoch": 7.305145263671875e-06,
      "step": 7660
    },
    {
      "epoch": 7.305145263671875e-06,
      "step": 7660,
      "training_step_time": 6.895169258117676
    },
    {
      "epoch": 7.306098937988281e-06,
      "model_forward_time": 1.7720637321472168,
      "step": 7661
    },
    {
      "epoch": 7.306098937988281e-06,
      "step": 7661,
      "training_step_time": 8.202655792236328
    },
    {
      "epoch": 7.3070526123046874e-06,
      "model_forward_time": 1.8915600776672363,
      "step": 7662
    },
    {
      "epoch": 7.3070526123046874e-06,
      "step": 7662,
      "training_step_time": 10.283167123794556
    },
    {
      "epoch": 7.308006286621094e-06,
      "model_forward_time": 1.7281403541564941,
      "step": 7663
    },
    {
      "epoch": 7.308006286621094e-06,
      "step": 7663,
      "training_step_time": 7.858815431594849
    },
    {
      "epoch": 7.3089599609375e-06,
      "model_forward_time": 1.747730016708374,
      "step": 7664
    },
    {
      "epoch": 7.3089599609375e-06,
      "step": 7664,
      "training_step_time": 7.440372467041016
    },
    {
      "epoch": 7.309913635253906e-06,
      "model_forward_time": 1.7340760231018066,
      "step": 7665
    },
    {
      "epoch": 7.309913635253906e-06,
      "step": 7665,
      "training_step_time": 6.97683310508728
    },
    {
      "epoch": 7.310867309570313e-06,
      "model_forward_time": 1.728719711303711,
      "step": 7666
    },
    {
      "epoch": 7.310867309570313e-06,
      "step": 7666,
      "training_step_time": 6.920025587081909
    },
    {
      "epoch": 7.3118209838867186e-06,
      "model_forward_time": 1.728663682937622,
      "step": 7667
    },
    {
      "epoch": 7.3118209838867186e-06,
      "step": 7667,
      "training_step_time": 6.938113689422607
    },
    {
      "epoch": 7.312774658203125e-06,
      "model_forward_time": 1.729851484298706,
      "step": 7668
    },
    {
      "epoch": 7.312774658203125e-06,
      "step": 7668,
      "training_step_time": 6.926361560821533
    },
    {
      "epoch": 7.313728332519531e-06,
      "model_forward_time": 1.7442913055419922,
      "step": 7669
    },
    {
      "epoch": 7.313728332519531e-06,
      "step": 7669,
      "training_step_time": 6.966111898422241
    },
    {
      "epoch": 7.3146820068359374e-06,
      "grad_norm": 0.3064670264720917,
      "learning_rate": 9.980537091751303e-06,
      "loss": 0.069,
      "step": 7670
    },
    {
      "epoch": 7.3146820068359374e-06,
      "model_forward_time": 1.7949225902557373,
      "step": 7670
    },
    {
      "dynamics_loss_avg": 0.06206384468823671,
      "epoch": 7.3146820068359374e-06,
      "step": 7670
    },
    {
      "action_loss_avg": 0.0073498481884598735,
      "epoch": 7.3146820068359374e-06,
      "step": 7670
    },
    {
      "epoch": 7.3146820068359374e-06,
      "step": 7670,
      "training_step_time": 7.621886968612671
    },
    {
      "epoch": 7.315635681152344e-06,
      "model_forward_time": 2.2921741008758545,
      "step": 7671
    },
    {
      "epoch": 7.315635681152344e-06,
      "step": 7671,
      "training_step_time": 10.344496011734009
    },
    {
      "epoch": 7.31658935546875e-06,
      "model_forward_time": 1.7307255268096924,
      "step": 7672
    },
    {
      "epoch": 7.31658935546875e-06,
      "step": 7672,
      "training_step_time": 7.796540975570679
    },
    {
      "epoch": 7.317543029785156e-06,
      "model_forward_time": 1.7354249954223633,
      "step": 7673
    },
    {
      "epoch": 7.317543029785156e-06,
      "step": 7673,
      "training_step_time": 7.413912057876587
    },
    {
      "epoch": 7.318496704101563e-06,
      "model_forward_time": 1.7378032207489014,
      "step": 7674
    },
    {
      "epoch": 7.318496704101563e-06,
      "step": 7674,
      "training_step_time": 6.8661651611328125
    },
    {
      "epoch": 7.319450378417969e-06,
      "model_forward_time": 1.7322502136230469,
      "step": 7675
    },
    {
      "epoch": 7.319450378417969e-06,
      "step": 7675,
      "training_step_time": 6.99370265007019
    },
    {
      "epoch": 7.320404052734375e-06,
      "model_forward_time": 1.737250566482544,
      "step": 7676
    },
    {
      "epoch": 7.320404052734375e-06,
      "step": 7676,
      "training_step_time": 6.947138547897339
    },
    {
      "epoch": 7.321357727050781e-06,
      "model_forward_time": 1.7176015377044678,
      "step": 7677
    },
    {
      "epoch": 7.321357727050781e-06,
      "step": 7677,
      "training_step_time": 6.936116695404053
    },
    {
      "epoch": 7.3223114013671874e-06,
      "model_forward_time": 1.7383661270141602,
      "step": 7678
    },
    {
      "epoch": 7.3223114013671874e-06,
      "step": 7678,
      "training_step_time": 6.970654249191284
    },
    {
      "epoch": 7.323265075683594e-06,
      "model_forward_time": 1.7891159057617188,
      "step": 7679
    },
    {
      "epoch": 7.323265075683594e-06,
      "step": 7679,
      "training_step_time": 7.336665153503418
    },
    {
      "epoch": 7.32421875e-06,
      "grad_norm": 0.3502773344516754,
      "learning_rate": 9.980391069915897e-06,
      "loss": 0.0717,
      "step": 7680
    },
    {
      "epoch": 7.32421875e-06,
      "model_forward_time": 2.275662660598755,
      "step": 7680
    },
    {
      "dynamics_loss_avg": 0.05919029526412487,
      "epoch": 7.32421875e-06,
      "step": 7680
    },
    {
      "action_loss_avg": 0.00656556012108922,
      "epoch": 7.32421875e-06,
      "step": 7680
    },
    {
      "epoch": 7.32421875e-06,
      "step": 7680,
      "training_step_time": 7.82168984413147
    },
    {
      "epoch": 7.325172424316406e-06,
      "model_forward_time": 1.713205099105835,
      "step": 7681
    },
    {
      "epoch": 7.325172424316406e-06,
      "step": 7681,
      "training_step_time": 7.964667797088623
    },
    {
      "epoch": 7.326126098632813e-06,
      "model_forward_time": 1.7378380298614502,
      "step": 7682
    },
    {
      "epoch": 7.326126098632813e-06,
      "step": 7682,
      "training_step_time": 7.718854665756226
    },
    {
      "epoch": 7.327079772949219e-06,
      "model_forward_time": 1.7247734069824219,
      "step": 7683
    },
    {
      "epoch": 7.327079772949219e-06,
      "step": 7683,
      "training_step_time": 6.936994552612305
    },
    {
      "epoch": 7.328033447265625e-06,
      "model_forward_time": 1.7220473289489746,
      "step": 7684
    },
    {
      "epoch": 7.328033447265625e-06,
      "step": 7684,
      "training_step_time": 6.958827018737793
    },
    {
      "epoch": 7.328987121582031e-06,
      "model_forward_time": 1.738102912902832,
      "step": 7685
    },
    {
      "epoch": 7.328987121582031e-06,
      "step": 7685,
      "training_step_time": 6.900400161743164
    },
    {
      "epoch": 7.3299407958984374e-06,
      "model_forward_time": 1.7518255710601807,
      "step": 7686
    },
    {
      "epoch": 7.3299407958984374e-06,
      "step": 7686,
      "training_step_time": 6.891477108001709
    },
    {
      "epoch": 7.330894470214844e-06,
      "model_forward_time": 1.7453532218933105,
      "step": 7687
    },
    {
      "epoch": 7.330894470214844e-06,
      "step": 7687,
      "training_step_time": 6.939424991607666
    },
    {
      "epoch": 7.33184814453125e-06,
      "model_forward_time": 1.7758259773254395,
      "step": 7688
    },
    {
      "epoch": 7.33184814453125e-06,
      "step": 7688,
      "training_step_time": 7.2753705978393555
    },
    {
      "epoch": 7.332801818847656e-06,
      "model_forward_time": 1.7245283126831055,
      "step": 7689
    },
    {
      "epoch": 7.332801818847656e-06,
      "step": 7689,
      "training_step_time": 10.804595708847046
    },
    {
      "epoch": 7.333755493164063e-06,
      "grad_norm": 0.24789263308048248,
      "learning_rate": 9.98024450343241e-06,
      "loss": 0.0744,
      "step": 7690
    },
    {
      "epoch": 7.333755493164063e-06,
      "model_forward_time": 1.7450664043426514,
      "step": 7690
    },
    {
      "dynamics_loss_avg": 0.05620157551020384,
      "epoch": 7.333755493164063e-06,
      "step": 7690
    },
    {
      "action_loss_avg": 0.005605498305521905,
      "epoch": 7.333755493164063e-06,
      "step": 7690
    },
    {
      "epoch": 7.333755493164063e-06,
      "step": 7690,
      "training_step_time": 8.506292819976807
    },
    {
      "epoch": 7.334709167480469e-06,
      "model_forward_time": 1.7489492893218994,
      "step": 7691
    },
    {
      "epoch": 7.334709167480469e-06,
      "step": 7691,
      "training_step_time": 7.715191125869751
    },
    {
      "epoch": 7.335662841796875e-06,
      "model_forward_time": 1.7291364669799805,
      "step": 7692
    },
    {
      "epoch": 7.335662841796875e-06,
      "step": 7692,
      "training_step_time": 7.290733098983765
    },
    {
      "epoch": 7.336616516113281e-06,
      "model_forward_time": 1.7352190017700195,
      "step": 7693
    },
    {
      "epoch": 7.336616516113281e-06,
      "step": 7693,
      "training_step_time": 6.87706446647644
    },
    {
      "epoch": 7.3375701904296875e-06,
      "model_forward_time": 1.723520040512085,
      "step": 7694
    },
    {
      "epoch": 7.3375701904296875e-06,
      "step": 7694,
      "training_step_time": 6.943812370300293
    },
    {
      "epoch": 7.338523864746094e-06,
      "model_forward_time": 1.719425916671753,
      "step": 7695
    },
    {
      "epoch": 7.338523864746094e-06,
      "step": 7695,
      "training_step_time": 6.917057275772095
    },
    {
      "epoch": 7.3394775390625e-06,
      "model_forward_time": 1.7378244400024414,
      "step": 7696
    },
    {
      "epoch": 7.3394775390625e-06,
      "step": 7696,
      "training_step_time": 6.913983106613159
    },
    {
      "epoch": 7.340431213378906e-06,
      "model_forward_time": 1.7962021827697754,
      "step": 7697
    },
    {
      "epoch": 7.340431213378906e-06,
      "step": 7697,
      "training_step_time": 7.301069021224976
    },
    {
      "epoch": 7.341384887695313e-06,
      "model_forward_time": 1.736943244934082,
      "step": 7698
    },
    {
      "epoch": 7.341384887695313e-06,
      "step": 7698,
      "training_step_time": 8.678886651992798
    },
    {
      "epoch": 7.342338562011719e-06,
      "model_forward_time": 2.2697665691375732,
      "step": 7699
    },
    {
      "epoch": 7.342338562011719e-06,
      "step": 7699,
      "training_step_time": 9.850228309631348
    },
    {
      "epoch": 7.343292236328125e-06,
      "grad_norm": 0.22087806463241577,
      "learning_rate": 9.980097392316872e-06,
      "loss": 0.0808,
      "step": 7700
    },
    {
      "epoch": 7.343292236328125e-06,
      "model_forward_time": 1.7174203395843506,
      "step": 7700
    },
    {
      "dynamics_loss_avg": 0.07301086690276862,
      "epoch": 7.343292236328125e-06,
      "step": 7700
    },
    {
      "action_loss_avg": 0.006280324188992381,
      "epoch": 7.343292236328125e-06,
      "step": 7700
    },
    {
      "epoch": 7.343292236328125e-06,
      "step": 7700,
      "training_step_time": 8.737565755844116
    },
    {
      "epoch": 7.344245910644531e-06,
      "model_forward_time": 1.7495684623718262,
      "step": 7701
    },
    {
      "epoch": 7.344245910644531e-06,
      "step": 7701,
      "training_step_time": 7.337467432022095
    },
    {
      "epoch": 7.3451995849609375e-06,
      "model_forward_time": 1.7217345237731934,
      "step": 7702
    },
    {
      "epoch": 7.3451995849609375e-06,
      "step": 7702,
      "training_step_time": 6.883106708526611
    },
    {
      "epoch": 7.346153259277344e-06,
      "model_forward_time": 1.7311875820159912,
      "step": 7703
    },
    {
      "epoch": 7.346153259277344e-06,
      "step": 7703,
      "training_step_time": 6.9808735847473145
    },
    {
      "epoch": 7.34710693359375e-06,
      "model_forward_time": 1.7283992767333984,
      "step": 7704
    },
    {
      "epoch": 7.34710693359375e-06,
      "step": 7704,
      "training_step_time": 6.967062711715698
    },
    {
      "epoch": 7.348060607910156e-06,
      "model_forward_time": 1.7215251922607422,
      "step": 7705
    },
    {
      "epoch": 7.348060607910156e-06,
      "step": 7705,
      "training_step_time": 6.938915252685547
    },
    {
      "epoch": 7.349014282226563e-06,
      "model_forward_time": 1.7928986549377441,
      "step": 7706
    },
    {
      "epoch": 7.349014282226563e-06,
      "step": 7706,
      "training_step_time": 7.3623366355896
    },
    {
      "epoch": 7.349967956542969e-06,
      "model_forward_time": 1.7318198680877686,
      "step": 7707
    },
    {
      "epoch": 7.349967956542969e-06,
      "step": 7707,
      "training_step_time": 7.835345268249512
    },
    {
      "epoch": 7.350921630859375e-06,
      "model_forward_time": 2.250593900680542,
      "step": 7708
    },
    {
      "epoch": 7.350921630859375e-06,
      "step": 7708,
      "training_step_time": 9.68306565284729
    },
    {
      "epoch": 7.351875305175781e-06,
      "model_forward_time": 1.720022439956665,
      "step": 7709
    },
    {
      "epoch": 7.351875305175781e-06,
      "step": 7709,
      "training_step_time": 8.530259132385254
    },
    {
      "epoch": 7.3528289794921875e-06,
      "grad_norm": 0.3049639165401459,
      "learning_rate": 9.97994973658537e-06,
      "loss": 0.08,
      "step": 7710
    },
    {
      "epoch": 7.3528289794921875e-06,
      "model_forward_time": 1.7456092834472656,
      "step": 7710
    },
    {
      "dynamics_loss_avg": 0.06436458490788936,
      "epoch": 7.3528289794921875e-06,
      "step": 7710
    },
    {
      "action_loss_avg": 0.004088736511766911,
      "epoch": 7.3528289794921875e-06,
      "step": 7710
    },
    {
      "epoch": 7.3528289794921875e-06,
      "step": 7710,
      "training_step_time": 7.696236848831177
    },
    {
      "epoch": 7.353782653808594e-06,
      "model_forward_time": 1.7315495014190674,
      "step": 7711
    },
    {
      "epoch": 7.353782653808594e-06,
      "step": 7711,
      "training_step_time": 6.972274541854858
    },
    {
      "epoch": 7.354736328125e-06,
      "model_forward_time": 1.734050989151001,
      "step": 7712
    },
    {
      "epoch": 7.354736328125e-06,
      "step": 7712,
      "training_step_time": 6.986360788345337
    },
    {
      "epoch": 7.355690002441406e-06,
      "model_forward_time": 1.7196238040924072,
      "step": 7713
    },
    {
      "epoch": 7.355690002441406e-06,
      "step": 7713,
      "training_step_time": 6.97179102897644
    },
    {
      "epoch": 7.356643676757813e-06,
      "model_forward_time": 1.7240960597991943,
      "step": 7714
    },
    {
      "epoch": 7.356643676757813e-06,
      "step": 7714,
      "training_step_time": 6.9426798820495605
    },
    {
      "epoch": 7.357597351074219e-06,
      "model_forward_time": 1.7841284275054932,
      "step": 7715
    },
    {
      "epoch": 7.357597351074219e-06,
      "step": 7715,
      "training_step_time": 7.340708255767822
    },
    {
      "epoch": 7.358551025390625e-06,
      "model_forward_time": 1.7424473762512207,
      "step": 7716
    },
    {
      "epoch": 7.358551025390625e-06,
      "step": 7716,
      "training_step_time": 7.957165479660034
    },
    {
      "epoch": 7.359504699707031e-06,
      "model_forward_time": 2.267545461654663,
      "step": 7717
    },
    {
      "epoch": 7.359504699707031e-06,
      "step": 7717,
      "training_step_time": 9.934269189834595
    },
    {
      "epoch": 7.3604583740234375e-06,
      "model_forward_time": 1.7328245639801025,
      "step": 7718
    },
    {
      "epoch": 7.3604583740234375e-06,
      "step": 7718,
      "training_step_time": 8.372958898544312
    },
    {
      "epoch": 7.361412048339844e-06,
      "model_forward_time": 1.7228014469146729,
      "step": 7719
    },
    {
      "epoch": 7.361412048339844e-06,
      "step": 7719,
      "training_step_time": 7.648982048034668
    },
    {
      "epoch": 7.36236572265625e-06,
      "grad_norm": 0.2764319181442261,
      "learning_rate": 9.979801536254055e-06,
      "loss": 0.0767,
      "step": 7720
    },
    {
      "epoch": 7.36236572265625e-06,
      "model_forward_time": 1.7546558380126953,
      "step": 7720
    },
    {
      "dynamics_loss_avg": 0.08217865154147148,
      "epoch": 7.36236572265625e-06,
      "step": 7720
    },
    {
      "action_loss_avg": 0.008067146898247302,
      "epoch": 7.36236572265625e-06,
      "step": 7720
    },
    {
      "epoch": 7.36236572265625e-06,
      "step": 7720,
      "training_step_time": 6.896769762039185
    },
    {
      "epoch": 7.363319396972656e-06,
      "model_forward_time": 1.7316515445709229,
      "step": 7721
    },
    {
      "epoch": 7.363319396972656e-06,
      "step": 7721,
      "training_step_time": 6.962032318115234
    },
    {
      "epoch": 7.364273071289063e-06,
      "model_forward_time": 1.734774112701416,
      "step": 7722
    },
    {
      "epoch": 7.364273071289063e-06,
      "step": 7722,
      "training_step_time": 6.911988973617554
    },
    {
      "epoch": 7.365226745605469e-06,
      "model_forward_time": 1.741227388381958,
      "step": 7723
    },
    {
      "epoch": 7.365226745605469e-06,
      "step": 7723,
      "training_step_time": 6.933660268783569
    },
    {
      "epoch": 7.366180419921875e-06,
      "model_forward_time": 1.7403249740600586,
      "step": 7724
    },
    {
      "epoch": 7.366180419921875e-06,
      "step": 7724,
      "training_step_time": 7.3453686237335205
    },
    {
      "epoch": 7.367134094238281e-06,
      "model_forward_time": 1.7454230785369873,
      "step": 7725
    },
    {
      "epoch": 7.367134094238281e-06,
      "step": 7725,
      "training_step_time": 7.9317474365234375
    },
    {
      "epoch": 7.3680877685546875e-06,
      "model_forward_time": 2.262824058532715,
      "step": 7726
    },
    {
      "epoch": 7.3680877685546875e-06,
      "step": 7726,
      "training_step_time": 9.335306406021118
    },
    {
      "epoch": 7.369041442871094e-06,
      "model_forward_time": 1.7382287979125977,
      "step": 7727
    },
    {
      "epoch": 7.369041442871094e-06,
      "step": 7727,
      "training_step_time": 8.731428146362305
    },
    {
      "epoch": 7.3699951171875e-06,
      "model_forward_time": 1.7332262992858887,
      "step": 7728
    },
    {
      "epoch": 7.3699951171875e-06,
      "step": 7728,
      "training_step_time": 7.817296743392944
    },
    {
      "epoch": 7.370948791503906e-06,
      "model_forward_time": 1.7229266166687012,
      "step": 7729
    },
    {
      "epoch": 7.370948791503906e-06,
      "step": 7729,
      "training_step_time": 7.3413496017456055
    },
    {
      "epoch": 7.371902465820313e-06,
      "grad_norm": 0.34104156494140625,
      "learning_rate": 9.979652791339127e-06,
      "loss": 0.0696,
      "step": 7730
    },
    {
      "epoch": 7.371902465820313e-06,
      "model_forward_time": 1.728325366973877,
      "step": 7730
    },
    {
      "dynamics_loss_avg": 0.056579575128853324,
      "epoch": 7.371902465820313e-06,
      "step": 7730
    },
    {
      "action_loss_avg": 0.010667184018529952,
      "epoch": 7.371902465820313e-06,
      "step": 7730
    },
    {
      "epoch": 7.371902465820313e-06,
      "step": 7730,
      "training_step_time": 6.915256977081299
    },
    {
      "epoch": 7.372856140136719e-06,
      "model_forward_time": 1.7332870960235596,
      "step": 7731
    },
    {
      "epoch": 7.372856140136719e-06,
      "step": 7731,
      "training_step_time": 6.94052791595459
    },
    {
      "epoch": 7.373809814453125e-06,
      "model_forward_time": 1.355013132095337,
      "step": 7732
    },
    {
      "epoch": 7.373809814453125e-06,
      "step": 7732,
      "training_step_time": 6.940897703170776
    },
    {
      "epoch": 7.374763488769531e-06,
      "model_forward_time": 1.7428522109985352,
      "step": 7733
    },
    {
      "epoch": 7.374763488769531e-06,
      "step": 7733,
      "training_step_time": 7.336368560791016
    },
    {
      "epoch": 7.3757171630859375e-06,
      "model_forward_time": 1.7238600254058838,
      "step": 7734
    },
    {
      "epoch": 7.3757171630859375e-06,
      "step": 7734,
      "training_step_time": 8.498838663101196
    },
    {
      "epoch": 7.376670837402344e-06,
      "model_forward_time": 2.273658514022827,
      "step": 7735
    },
    {
      "epoch": 7.376670837402344e-06,
      "step": 7735,
      "training_step_time": 8.664733409881592
    },
    {
      "epoch": 7.37762451171875e-06,
      "model_forward_time": 1.7175474166870117,
      "step": 7736
    },
    {
      "epoch": 7.37762451171875e-06,
      "step": 7736,
      "training_step_time": 8.745975255966187
    },
    {
      "epoch": 7.378578186035156e-06,
      "model_forward_time": 1.745999813079834,
      "step": 7737
    },
    {
      "epoch": 7.378578186035156e-06,
      "step": 7737,
      "training_step_time": 8.606916904449463
    },
    {
      "epoch": 7.379531860351563e-06,
      "model_forward_time": 1.7432053089141846,
      "step": 7738
    },
    {
      "epoch": 7.379531860351563e-06,
      "step": 7738,
      "training_step_time": 7.166277170181274
    },
    {
      "epoch": 7.380485534667969e-06,
      "model_forward_time": 1.7535691261291504,
      "step": 7739
    },
    {
      "epoch": 7.380485534667969e-06,
      "step": 7739,
      "training_step_time": 6.965502023696899
    },
    {
      "epoch": 7.381439208984375e-06,
      "grad_norm": 0.4195444881916046,
      "learning_rate": 9.97950350185686e-06,
      "loss": 0.0714,
      "step": 7740
    },
    {
      "epoch": 7.381439208984375e-06,
      "model_forward_time": 1.7377967834472656,
      "step": 7740
    },
    {
      "dynamics_loss_avg": 0.057071524858474734,
      "epoch": 7.381439208984375e-06,
      "step": 7740
    },
    {
      "action_loss_avg": 0.005462298914790153,
      "epoch": 7.381439208984375e-06,
      "step": 7740
    },
    {
      "epoch": 7.381439208984375e-06,
      "step": 7740,
      "training_step_time": 6.940305471420288
    },
    {
      "epoch": 7.382392883300781e-06,
      "model_forward_time": 1.7342679500579834,
      "step": 7741
    },
    {
      "epoch": 7.382392883300781e-06,
      "step": 7741,
      "training_step_time": 6.999446630477905
    },
    {
      "epoch": 7.3833465576171875e-06,
      "model_forward_time": 1.7324106693267822,
      "step": 7742
    },
    {
      "epoch": 7.3833465576171875e-06,
      "step": 7742,
      "training_step_time": 6.908082485198975
    },
    {
      "epoch": 7.384300231933594e-06,
      "model_forward_time": 1.7745938301086426,
      "step": 7743
    },
    {
      "epoch": 7.384300231933594e-06,
      "step": 7743,
      "training_step_time": 7.79684853553772
    },
    {
      "epoch": 7.38525390625e-06,
      "model_forward_time": 1.73744797706604,
      "step": 7744
    },
    {
      "epoch": 7.38525390625e-06,
      "step": 7744,
      "training_step_time": 9.712745904922485
    },
    {
      "epoch": 7.386207580566406e-06,
      "model_forward_time": 2.2638254165649414,
      "step": 7745
    },
    {
      "epoch": 7.386207580566406e-06,
      "step": 7745,
      "training_step_time": 10.008865594863892
    },
    {
      "epoch": 7.387161254882813e-06,
      "model_forward_time": 1.3689231872558594,
      "step": 7746
    },
    {
      "epoch": 7.387161254882813e-06,
      "step": 7746,
      "training_step_time": 8.80347228050232
    },
    {
      "epoch": 7.388114929199219e-06,
      "model_forward_time": 1.7414255142211914,
      "step": 7747
    },
    {
      "epoch": 7.388114929199219e-06,
      "step": 7747,
      "training_step_time": 7.390859127044678
    },
    {
      "epoch": 7.389068603515625e-06,
      "model_forward_time": 1.7356371879577637,
      "step": 7748
    },
    {
      "epoch": 7.389068603515625e-06,
      "step": 7748,
      "training_step_time": 6.842178821563721
    },
    {
      "epoch": 7.390022277832031e-06,
      "model_forward_time": 1.7263994216918945,
      "step": 7749
    },
    {
      "epoch": 7.390022277832031e-06,
      "step": 7749,
      "training_step_time": 6.963125944137573
    },
    {
      "epoch": 7.3909759521484375e-06,
      "grad_norm": 0.209115132689476,
      "learning_rate": 9.979353667823576e-06,
      "loss": 0.0739,
      "step": 7750
    },
    {
      "epoch": 7.3909759521484375e-06,
      "model_forward_time": 1.7366294860839844,
      "step": 7750
    },
    {
      "dynamics_loss_avg": 0.0788014616817236,
      "epoch": 7.3909759521484375e-06,
      "step": 7750
    },
    {
      "action_loss_avg": 0.006848955620080232,
      "epoch": 7.3909759521484375e-06,
      "step": 7750
    },
    {
      "epoch": 7.3909759521484375e-06,
      "step": 7750,
      "training_step_time": 6.954336643218994
    },
    {
      "epoch": 7.391929626464844e-06,
      "model_forward_time": 1.736893892288208,
      "step": 7751
    },
    {
      "epoch": 7.391929626464844e-06,
      "step": 7751,
      "training_step_time": 6.935795068740845
    },
    {
      "epoch": 7.39288330078125e-06,
      "model_forward_time": 1.7944307327270508,
      "step": 7752
    },
    {
      "epoch": 7.39288330078125e-06,
      "step": 7752,
      "training_step_time": 7.838859558105469
    },
    {
      "epoch": 7.393836975097656e-06,
      "model_forward_time": 1.7201809883117676,
      "step": 7753
    },
    {
      "epoch": 7.393836975097656e-06,
      "step": 7753,
      "training_step_time": 7.868516206741333
    },
    {
      "epoch": 7.394790649414062e-06,
      "model_forward_time": 2.26377010345459,
      "step": 7754
    },
    {
      "epoch": 7.394790649414062e-06,
      "step": 7754,
      "training_step_time": 7.848372459411621
    },
    {
      "epoch": 7.395744323730469e-06,
      "model_forward_time": 1.7363674640655518,
      "step": 7755
    },
    {
      "epoch": 7.395744323730469e-06,
      "step": 7755,
      "training_step_time": 7.741652965545654
    },
    {
      "epoch": 7.396697998046875e-06,
      "model_forward_time": 1.7222108840942383,
      "step": 7756
    },
    {
      "epoch": 7.396697998046875e-06,
      "step": 7756,
      "training_step_time": 7.730799436569214
    },
    {
      "epoch": 7.397651672363281e-06,
      "model_forward_time": 1.7431461811065674,
      "step": 7757
    },
    {
      "epoch": 7.397651672363281e-06,
      "step": 7757,
      "training_step_time": 7.078548431396484
    },
    {
      "epoch": 7.3986053466796875e-06,
      "model_forward_time": 1.7400376796722412,
      "step": 7758
    },
    {
      "epoch": 7.3986053466796875e-06,
      "step": 7758,
      "training_step_time": 6.995986700057983
    },
    {
      "epoch": 7.399559020996094e-06,
      "model_forward_time": 1.7322735786437988,
      "step": 7759
    },
    {
      "epoch": 7.399559020996094e-06,
      "step": 7759,
      "training_step_time": 6.829352378845215
    },
    {
      "epoch": 7.4005126953125e-06,
      "grad_norm": 0.20040756464004517,
      "learning_rate": 9.979203289255659e-06,
      "loss": 0.0696,
      "step": 7760
    },
    {
      "epoch": 7.4005126953125e-06,
      "model_forward_time": 1.7349185943603516,
      "step": 7760
    },
    {
      "dynamics_loss_avg": 0.07224729172885418,
      "epoch": 7.4005126953125e-06,
      "step": 7760
    },
    {
      "action_loss_avg": 0.006079535023309291,
      "epoch": 7.4005126953125e-06,
      "step": 7760
    },
    {
      "epoch": 7.4005126953125e-06,
      "step": 7760,
      "training_step_time": 6.936525106430054
    },
    {
      "epoch": 7.401466369628906e-06,
      "model_forward_time": 1.7575657367706299,
      "step": 7761
    },
    {
      "epoch": 7.401466369628906e-06,
      "step": 7761,
      "training_step_time": 7.799215793609619
    },
    {
      "epoch": 7.402420043945312e-06,
      "model_forward_time": 1.7441818714141846,
      "step": 7762
    },
    {
      "epoch": 7.402420043945312e-06,
      "step": 7762,
      "training_step_time": 8.852702140808105
    },
    {
      "epoch": 7.403373718261719e-06,
      "model_forward_time": 2.2565853595733643,
      "step": 7763
    },
    {
      "epoch": 7.403373718261719e-06,
      "step": 7763,
      "training_step_time": 10.619335889816284
    },
    {
      "epoch": 7.404327392578125e-06,
      "model_forward_time": 1.749239206314087,
      "step": 7764
    },
    {
      "epoch": 7.404327392578125e-06,
      "step": 7764,
      "training_step_time": 8.378271341323853
    },
    {
      "epoch": 7.405281066894531e-06,
      "model_forward_time": 1.7455408573150635,
      "step": 7765
    },
    {
      "epoch": 7.405281066894531e-06,
      "step": 7765,
      "training_step_time": 7.745756387710571
    },
    {
      "epoch": 7.4062347412109375e-06,
      "model_forward_time": 1.7275516986846924,
      "step": 7766
    },
    {
      "epoch": 7.4062347412109375e-06,
      "step": 7766,
      "training_step_time": 7.337983131408691
    },
    {
      "epoch": 7.407188415527344e-06,
      "model_forward_time": 1.7263519763946533,
      "step": 7767
    },
    {
      "epoch": 7.407188415527344e-06,
      "step": 7767,
      "training_step_time": 6.902912855148315
    },
    {
      "epoch": 7.40814208984375e-06,
      "model_forward_time": 1.728652000427246,
      "step": 7768
    },
    {
      "epoch": 7.40814208984375e-06,
      "step": 7768,
      "training_step_time": 6.938520908355713
    },
    {
      "epoch": 7.409095764160156e-06,
      "model_forward_time": 1.7302985191345215,
      "step": 7769
    },
    {
      "epoch": 7.409095764160156e-06,
      "step": 7769,
      "training_step_time": 6.9531025886535645
    },
    {
      "epoch": 7.410049438476562e-06,
      "grad_norm": 0.3080426752567291,
      "learning_rate": 9.979052366169557e-06,
      "loss": 0.0715,
      "step": 7770
    },
    {
      "epoch": 7.410049438476562e-06,
      "model_forward_time": 1.7409870624542236,
      "step": 7770
    },
    {
      "dynamics_loss_avg": 0.06747139878571033,
      "epoch": 7.410049438476562e-06,
      "step": 7770
    },
    {
      "action_loss_avg": 0.00721647716127336,
      "epoch": 7.410049438476562e-06,
      "step": 7770
    },
    {
      "epoch": 7.410049438476562e-06,
      "step": 7770,
      "training_step_time": 7.665153741836548
    },
    {
      "epoch": 7.411003112792969e-06,
      "model_forward_time": 1.7930760383605957,
      "step": 7771
    },
    {
      "epoch": 7.411003112792969e-06,
      "step": 7771,
      "training_step_time": 8.702067613601685
    },
    {
      "epoch": 7.411956787109375e-06,
      "model_forward_time": 2.251929521560669,
      "step": 7772
    },
    {
      "epoch": 7.411956787109375e-06,
      "step": 7772,
      "training_step_time": 8.799426794052124
    },
    {
      "epoch": 7.412910461425781e-06,
      "model_forward_time": 1.7235097885131836,
      "step": 7773
    },
    {
      "epoch": 7.412910461425781e-06,
      "step": 7773,
      "training_step_time": 7.869344711303711
    },
    {
      "epoch": 7.4138641357421875e-06,
      "model_forward_time": 1.7222933769226074,
      "step": 7774
    },
    {
      "epoch": 7.4138641357421875e-06,
      "step": 7774,
      "training_step_time": 8.766566753387451
    },
    {
      "epoch": 7.414817810058594e-06,
      "model_forward_time": 1.7455508708953857,
      "step": 7775
    },
    {
      "epoch": 7.414817810058594e-06,
      "step": 7775,
      "training_step_time": 7.837584972381592
    },
    {
      "epoch": 7.415771484375e-06,
      "model_forward_time": 1.7342822551727295,
      "step": 7776
    },
    {
      "epoch": 7.415771484375e-06,
      "step": 7776,
      "training_step_time": 6.972136497497559
    },
    {
      "epoch": 7.416725158691406e-06,
      "model_forward_time": 1.7245051860809326,
      "step": 7777
    },
    {
      "epoch": 7.416725158691406e-06,
      "step": 7777,
      "training_step_time": 6.963578701019287
    },
    {
      "epoch": 7.417678833007812e-06,
      "model_forward_time": 1.7291843891143799,
      "step": 7778
    },
    {
      "epoch": 7.417678833007812e-06,
      "step": 7778,
      "training_step_time": 6.957175016403198
    },
    {
      "epoch": 7.418632507324219e-06,
      "model_forward_time": 1.7402386665344238,
      "step": 7779
    },
    {
      "epoch": 7.418632507324219e-06,
      "step": 7779,
      "training_step_time": 7.682685613632202
    },
    {
      "epoch": 7.419586181640625e-06,
      "grad_norm": 0.1786177009344101,
      "learning_rate": 9.978900898581775e-06,
      "loss": 0.0714,
      "step": 7780
    },
    {
      "epoch": 7.419586181640625e-06,
      "model_forward_time": 1.7763311862945557,
      "step": 7780
    },
    {
      "dynamics_loss_avg": 0.07224992588162422,
      "epoch": 7.419586181640625e-06,
      "step": 7780
    },
    {
      "action_loss_avg": 0.008340221410617233,
      "epoch": 7.419586181640625e-06,
      "step": 7780
    },
    {
      "epoch": 7.419586181640625e-06,
      "step": 7780,
      "training_step_time": 9.338551759719849
    },
    {
      "epoch": 7.420539855957031e-06,
      "model_forward_time": 2.254885673522949,
      "step": 7781
    },
    {
      "epoch": 7.420539855957031e-06,
      "step": 7781,
      "training_step_time": 8.947439432144165
    },
    {
      "epoch": 7.4214935302734375e-06,
      "model_forward_time": 1.730525016784668,
      "step": 7782
    },
    {
      "epoch": 7.4214935302734375e-06,
      "step": 7782,
      "training_step_time": 9.081735610961914
    },
    {
      "epoch": 7.422447204589844e-06,
      "model_forward_time": 1.734339952468872,
      "step": 7783
    },
    {
      "epoch": 7.422447204589844e-06,
      "step": 7783,
      "training_step_time": 7.729418754577637
    },
    {
      "epoch": 7.42340087890625e-06,
      "model_forward_time": 1.7460577487945557,
      "step": 7784
    },
    {
      "epoch": 7.42340087890625e-06,
      "step": 7784,
      "training_step_time": 7.639862537384033
    },
    {
      "epoch": 7.424354553222656e-06,
      "model_forward_time": 1.7408788204193115,
      "step": 7785
    },
    {
      "epoch": 7.424354553222656e-06,
      "step": 7785,
      "training_step_time": 7.082393646240234
    },
    {
      "epoch": 7.425308227539062e-06,
      "model_forward_time": 1.7316370010375977,
      "step": 7786
    },
    {
      "epoch": 7.425308227539062e-06,
      "step": 7786,
      "training_step_time": 6.94521164894104
    },
    {
      "epoch": 7.426261901855469e-06,
      "model_forward_time": 1.7444157600402832,
      "step": 7787
    },
    {
      "epoch": 7.426261901855469e-06,
      "step": 7787,
      "training_step_time": 6.933652877807617
    },
    {
      "epoch": 7.427215576171875e-06,
      "model_forward_time": 1.7211863994598389,
      "step": 7788
    },
    {
      "epoch": 7.427215576171875e-06,
      "step": 7788,
      "training_step_time": 7.664624452590942
    },
    {
      "epoch": 7.428169250488281e-06,
      "model_forward_time": 1.7842020988464355,
      "step": 7789
    },
    {
      "epoch": 7.428169250488281e-06,
      "step": 7789,
      "training_step_time": 7.775584697723389
    },
    {
      "epoch": 7.4291229248046875e-06,
      "grad_norm": 0.2769334018230438,
      "learning_rate": 9.978748886508875e-06,
      "loss": 0.0688,
      "step": 7790
    },
    {
      "epoch": 7.4291229248046875e-06,
      "model_forward_time": 1.7462544441223145,
      "step": 7790
    },
    {
      "dynamics_loss_avg": 0.05222657807171345,
      "epoch": 7.4291229248046875e-06,
      "step": 7790
    },
    {
      "action_loss_avg": 0.004869648162275553,
      "epoch": 7.4291229248046875e-06,
      "step": 7790
    },
    {
      "epoch": 7.4291229248046875e-06,
      "step": 7790,
      "training_step_time": 11.132686138153076
    },
    {
      "epoch": 7.430076599121094e-06,
      "model_forward_time": 1.712425708770752,
      "step": 7791
    },
    {
      "epoch": 7.430076599121094e-06,
      "step": 7791,
      "training_step_time": 8.885160446166992
    },
    {
      "epoch": 7.4310302734375e-06,
      "model_forward_time": 1.7311546802520752,
      "step": 7792
    },
    {
      "epoch": 7.4310302734375e-06,
      "step": 7792,
      "training_step_time": 7.76435112953186
    },
    {
      "epoch": 7.431983947753906e-06,
      "model_forward_time": 1.730011224746704,
      "step": 7793
    },
    {
      "epoch": 7.431983947753906e-06,
      "step": 7793,
      "training_step_time": 7.59498929977417
    },
    {
      "epoch": 7.432937622070312e-06,
      "model_forward_time": 1.717825174331665,
      "step": 7794
    },
    {
      "epoch": 7.432937622070312e-06,
      "step": 7794,
      "training_step_time": 7.060665845870972
    },
    {
      "epoch": 7.433891296386719e-06,
      "model_forward_time": 1.7194881439208984,
      "step": 7795
    },
    {
      "epoch": 7.433891296386719e-06,
      "step": 7795,
      "training_step_time": 6.918825149536133
    },
    {
      "epoch": 7.434844970703125e-06,
      "model_forward_time": 1.727675199508667,
      "step": 7796
    },
    {
      "epoch": 7.434844970703125e-06,
      "step": 7796,
      "training_step_time": 6.922216415405273
    },
    {
      "epoch": 7.435798645019531e-06,
      "model_forward_time": 1.7378926277160645,
      "step": 7797
    },
    {
      "epoch": 7.435798645019531e-06,
      "step": 7797,
      "training_step_time": 7.665366888046265
    },
    {
      "epoch": 7.4367523193359375e-06,
      "model_forward_time": 1.7675864696502686,
      "step": 7798
    },
    {
      "epoch": 7.4367523193359375e-06,
      "step": 7798,
      "training_step_time": 8.538632869720459
    },
    {
      "epoch": 7.437705993652344e-06,
      "model_forward_time": 1.704881191253662,
      "step": 7799
    },
    {
      "epoch": 7.437705993652344e-06,
      "step": 7799,
      "training_step_time": 8.399227142333984
    },
    {
      "epoch": 7.43865966796875e-06,
      "grad_norm": 0.16657604277133942,
      "learning_rate": 9.978596329967485e-06,
      "loss": 0.0733,
      "step": 7800
    },
    {
      "epoch": 7.43865966796875e-06,
      "model_forward_time": 2.2518599033355713,
      "step": 7800
    },
    {
      "dynamics_loss_avg": 0.08691700026392937,
      "epoch": 7.43865966796875e-06,
      "step": 7800
    },
    {
      "action_loss_avg": 0.006624654331244528,
      "epoch": 7.43865966796875e-06,
      "step": 7800
    },
    {
      "epoch": 7.43865966796875e-06,
      "step": 7800,
      "training_step_time": 8.542574644088745
    },
    {
      "epoch": 7.439613342285156e-06,
      "model_forward_time": 1.7414801120758057,
      "step": 7801
    },
    {
      "epoch": 7.439613342285156e-06,
      "step": 7801,
      "training_step_time": 8.623934507369995
    },
    {
      "epoch": 7.440567016601562e-06,
      "model_forward_time": 1.740318775177002,
      "step": 7802
    },
    {
      "epoch": 7.440567016601562e-06,
      "step": 7802,
      "training_step_time": 7.697491645812988
    },
    {
      "epoch": 7.441520690917969e-06,
      "model_forward_time": 1.734755039215088,
      "step": 7803
    },
    {
      "epoch": 7.441520690917969e-06,
      "step": 7803,
      "training_step_time": 7.564420700073242
    },
    {
      "epoch": 7.442474365234375e-06,
      "model_forward_time": 1.7269103527069092,
      "step": 7804
    },
    {
      "epoch": 7.442474365234375e-06,
      "step": 7804,
      "training_step_time": 6.94093132019043
    },
    {
      "epoch": 7.443428039550781e-06,
      "model_forward_time": 1.720191240310669,
      "step": 7805
    },
    {
      "epoch": 7.443428039550781e-06,
      "step": 7805,
      "training_step_time": 6.87989068031311
    },
    {
      "epoch": 7.4443817138671875e-06,
      "model_forward_time": 1.7428772449493408,
      "step": 7806
    },
    {
      "epoch": 7.4443817138671875e-06,
      "step": 7806,
      "training_step_time": 7.5940070152282715
    },
    {
      "epoch": 7.445335388183594e-06,
      "model_forward_time": 1.7708847522735596,
      "step": 7807
    },
    {
      "epoch": 7.445335388183594e-06,
      "step": 7807,
      "training_step_time": 8.774941205978394
    },
    {
      "epoch": 7.4462890625e-06,
      "model_forward_time": 1.7212471961975098,
      "step": 7808
    },
    {
      "epoch": 7.4462890625e-06,
      "step": 7808,
      "training_step_time": 7.916986465454102
    },
    {
      "epoch": 7.447242736816406e-06,
      "model_forward_time": 2.248523712158203,
      "step": 7809
    },
    {
      "epoch": 7.447242736816406e-06,
      "step": 7809,
      "training_step_time": 9.60516357421875
    },
    {
      "epoch": 7.448196411132812e-06,
      "grad_norm": 0.3090384900569916,
      "learning_rate": 9.978443228974283e-06,
      "loss": 0.0724,
      "step": 7810
    },
    {
      "epoch": 7.448196411132812e-06,
      "model_forward_time": 1.7275288105010986,
      "step": 7810
    },
    {
      "dynamics_loss_avg": 0.07808306701481342,
      "epoch": 7.448196411132812e-06,
      "step": 7810
    },
    {
      "action_loss_avg": 0.0058773616561666134,
      "epoch": 7.448196411132812e-06,
      "step": 7810
    },
    {
      "epoch": 7.448196411132812e-06,
      "step": 7810,
      "training_step_time": 7.715553283691406
    },
    {
      "epoch": 7.449150085449219e-06,
      "model_forward_time": 1.7260515689849854,
      "step": 7811
    },
    {
      "epoch": 7.449150085449219e-06,
      "step": 7811,
      "training_step_time": 8.166023969650269
    },
    {
      "epoch": 7.450103759765625e-06,
      "model_forward_time": 1.7266154289245605,
      "step": 7812
    },
    {
      "epoch": 7.450103759765625e-06,
      "step": 7812,
      "training_step_time": 7.534351348876953
    },
    {
      "epoch": 7.451057434082031e-06,
      "model_forward_time": 1.740224838256836,
      "step": 7813
    },
    {
      "epoch": 7.451057434082031e-06,
      "step": 7813,
      "training_step_time": 6.794085264205933
    },
    {
      "epoch": 7.4520111083984376e-06,
      "model_forward_time": 1.734121561050415,
      "step": 7814
    },
    {
      "epoch": 7.4520111083984376e-06,
      "step": 7814,
      "training_step_time": 6.93598484992981
    },
    {
      "epoch": 7.452964782714844e-06,
      "model_forward_time": 1.731454610824585,
      "step": 7815
    },
    {
      "epoch": 7.452964782714844e-06,
      "step": 7815,
      "training_step_time": 7.5341713428497314
    },
    {
      "epoch": 7.45391845703125e-06,
      "model_forward_time": 1.7528131008148193,
      "step": 7816
    },
    {
      "epoch": 7.45391845703125e-06,
      "step": 7816,
      "training_step_time": 8.739733219146729
    },
    {
      "epoch": 7.454872131347656e-06,
      "model_forward_time": 1.736462116241455,
      "step": 7817
    },
    {
      "epoch": 7.454872131347656e-06,
      "step": 7817,
      "training_step_time": 8.359112977981567
    },
    {
      "epoch": 7.455825805664062e-06,
      "model_forward_time": 2.267965793609619,
      "step": 7818
    },
    {
      "epoch": 7.455825805664062e-06,
      "step": 7818,
      "training_step_time": 8.354886531829834
    },
    {
      "epoch": 7.456779479980469e-06,
      "model_forward_time": 1.7225799560546875,
      "step": 7819
    },
    {
      "epoch": 7.456779479980469e-06,
      "step": 7819,
      "training_step_time": 9.244085550308228
    },
    {
      "epoch": 7.457733154296875e-06,
      "grad_norm": 0.24516019225120544,
      "learning_rate": 9.978289583546016e-06,
      "loss": 0.07,
      "step": 7820
    },
    {
      "epoch": 7.457733154296875e-06,
      "model_forward_time": 1.740964412689209,
      "step": 7820
    },
    {
      "dynamics_loss_avg": 0.04440782051533461,
      "epoch": 7.457733154296875e-06,
      "step": 7820
    },
    {
      "action_loss_avg": 0.002800043951719999,
      "epoch": 7.457733154296875e-06,
      "step": 7820
    },
    {
      "epoch": 7.457733154296875e-06,
      "step": 7820,
      "training_step_time": 7.687457084655762
    },
    {
      "epoch": 7.458686828613281e-06,
      "model_forward_time": 1.72821044921875,
      "step": 7821
    },
    {
      "epoch": 7.458686828613281e-06,
      "step": 7821,
      "training_step_time": 7.644695281982422
    },
    {
      "epoch": 7.4596405029296876e-06,
      "model_forward_time": 1.7469701766967773,
      "step": 7822
    },
    {
      "epoch": 7.4596405029296876e-06,
      "step": 7822,
      "training_step_time": 7.327248573303223
    },
    {
      "epoch": 7.460594177246094e-06,
      "model_forward_time": 1.7183287143707275,
      "step": 7823
    },
    {
      "epoch": 7.460594177246094e-06,
      "step": 7823,
      "training_step_time": 6.905875205993652
    },
    {
      "epoch": 7.4615478515625e-06,
      "model_forward_time": 1.7275402545928955,
      "step": 7824
    },
    {
      "epoch": 7.4615478515625e-06,
      "step": 7824,
      "training_step_time": 7.583465576171875
    },
    {
      "epoch": 7.462501525878906e-06,
      "model_forward_time": 1.7316734790802002,
      "step": 7825
    },
    {
      "epoch": 7.462501525878906e-06,
      "step": 7825,
      "training_step_time": 8.515795230865479
    },
    {
      "epoch": 7.463455200195312e-06,
      "model_forward_time": 1.8032207489013672,
      "step": 7826
    },
    {
      "epoch": 7.463455200195312e-06,
      "step": 7826,
      "training_step_time": 8.608043432235718
    },
    {
      "epoch": 7.464408874511719e-06,
      "model_forward_time": 2.261115789413452,
      "step": 7827
    },
    {
      "epoch": 7.464408874511719e-06,
      "step": 7827,
      "training_step_time": 8.799529790878296
    },
    {
      "epoch": 7.465362548828125e-06,
      "model_forward_time": 1.7382099628448486,
      "step": 7828
    },
    {
      "epoch": 7.465362548828125e-06,
      "step": 7828,
      "training_step_time": 9.321758031845093
    },
    {
      "epoch": 7.466316223144531e-06,
      "model_forward_time": 1.7245755195617676,
      "step": 7829
    },
    {
      "epoch": 7.466316223144531e-06,
      "step": 7829,
      "training_step_time": 7.714014768600464
    },
    {
      "epoch": 7.4672698974609376e-06,
      "grad_norm": 0.1940995752811432,
      "learning_rate": 9.978135393699483e-06,
      "loss": 0.0729,
      "step": 7830
    },
    {
      "epoch": 7.4672698974609376e-06,
      "model_forward_time": 1.7323966026306152,
      "step": 7830
    },
    {
      "dynamics_loss_avg": 0.07608781345188617,
      "epoch": 7.4672698974609376e-06,
      "step": 7830
    },
    {
      "action_loss_avg": 0.005187472025863826,
      "epoch": 7.4672698974609376e-06,
      "step": 7830
    },
    {
      "epoch": 7.4672698974609376e-06,
      "step": 7830,
      "training_step_time": 7.614927768707275
    },
    {
      "epoch": 7.468223571777344e-06,
      "model_forward_time": 1.7305200099945068,
      "step": 7831
    },
    {
      "epoch": 7.468223571777344e-06,
      "step": 7831,
      "training_step_time": 7.190172433853149
    },
    {
      "epoch": 7.46917724609375e-06,
      "model_forward_time": 1.745018720626831,
      "step": 7832
    },
    {
      "epoch": 7.46917724609375e-06,
      "step": 7832,
      "training_step_time": 6.885493755340576
    },
    {
      "epoch": 7.4701309204101564e-06,
      "model_forward_time": 1.7255871295928955,
      "step": 7833
    },
    {
      "epoch": 7.4701309204101564e-06,
      "step": 7833,
      "training_step_time": 7.56831693649292
    },
    {
      "epoch": 7.471084594726562e-06,
      "model_forward_time": 1.7298519611358643,
      "step": 7834
    },
    {
      "epoch": 7.471084594726562e-06,
      "step": 7834,
      "training_step_time": 8.435865640640259
    },
    {
      "epoch": 7.472038269042969e-06,
      "model_forward_time": 1.7963066101074219,
      "step": 7835
    },
    {
      "epoch": 7.472038269042969e-06,
      "step": 7835,
      "training_step_time": 9.814933061599731
    },
    {
      "epoch": 7.472991943359375e-06,
      "model_forward_time": 2.276073932647705,
      "step": 7836
    },
    {
      "epoch": 7.472991943359375e-06,
      "step": 7836,
      "training_step_time": 7.747943162918091
    },
    {
      "epoch": 7.473945617675781e-06,
      "model_forward_time": 1.7271747589111328,
      "step": 7837
    },
    {
      "epoch": 7.473945617675781e-06,
      "step": 7837,
      "training_step_time": 8.84364104270935
    },
    {
      "epoch": 7.4748992919921876e-06,
      "model_forward_time": 1.7183308601379395,
      "step": 7838
    },
    {
      "epoch": 7.4748992919921876e-06,
      "step": 7838,
      "training_step_time": 8.590682029724121
    },
    {
      "epoch": 7.475852966308594e-06,
      "model_forward_time": 1.7296264171600342,
      "step": 7839
    },
    {
      "epoch": 7.475852966308594e-06,
      "step": 7839,
      "training_step_time": 7.599545955657959
    },
    {
      "epoch": 7.476806640625e-06,
      "grad_norm": 0.1616901010274887,
      "learning_rate": 9.97798065945155e-06,
      "loss": 0.0703,
      "step": 7840
    },
    {
      "epoch": 7.476806640625e-06,
      "model_forward_time": 1.733027458190918,
      "step": 7840
    },
    {
      "dynamics_loss_avg": 0.07133647650480271,
      "epoch": 7.476806640625e-06,
      "step": 7840
    },
    {
      "action_loss_avg": 0.008458825200796128,
      "epoch": 7.476806640625e-06,
      "step": 7840
    },
    {
      "epoch": 7.476806640625e-06,
      "step": 7840,
      "training_step_time": 7.602203607559204
    },
    {
      "epoch": 7.4777603149414064e-06,
      "model_forward_time": 1.7240948677062988,
      "step": 7841
    },
    {
      "epoch": 7.4777603149414064e-06,
      "step": 7841,
      "training_step_time": 6.9537458419799805
    },
    {
      "epoch": 7.478713989257812e-06,
      "model_forward_time": 1.7371537685394287,
      "step": 7842
    },
    {
      "epoch": 7.478713989257812e-06,
      "step": 7842,
      "training_step_time": 7.619772911071777
    },
    {
      "epoch": 7.479667663574219e-06,
      "model_forward_time": 1.7167060375213623,
      "step": 7843
    },
    {
      "epoch": 7.479667663574219e-06,
      "step": 7843,
      "training_step_time": 7.658053636550903
    },
    {
      "epoch": 7.480621337890625e-06,
      "model_forward_time": 1.777493953704834,
      "step": 7844
    },
    {
      "epoch": 7.480621337890625e-06,
      "step": 7844,
      "training_step_time": 8.492611646652222
    },
    {
      "epoch": 7.481575012207031e-06,
      "model_forward_time": 2.2631402015686035,
      "step": 7845
    },
    {
      "epoch": 7.481575012207031e-06,
      "step": 7845,
      "training_step_time": 8.88900637626648
    },
    {
      "epoch": 7.482528686523438e-06,
      "model_forward_time": 1.3747353553771973,
      "step": 7846
    },
    {
      "epoch": 7.482528686523438e-06,
      "step": 7846,
      "training_step_time": 8.969864130020142
    },
    {
      "epoch": 7.483482360839844e-06,
      "model_forward_time": 1.739039659500122,
      "step": 7847
    },
    {
      "epoch": 7.483482360839844e-06,
      "step": 7847,
      "training_step_time": 8.879976272583008
    },
    {
      "epoch": 7.48443603515625e-06,
      "model_forward_time": 1.7234907150268555,
      "step": 7848
    },
    {
      "epoch": 7.48443603515625e-06,
      "step": 7848,
      "training_step_time": 7.5835089683532715
    },
    {
      "epoch": 7.4853897094726564e-06,
      "model_forward_time": 1.7405321598052979,
      "step": 7849
    },
    {
      "epoch": 7.4853897094726564e-06,
      "step": 7849,
      "training_step_time": 7.58376145362854
    },
    {
      "epoch": 7.486343383789062e-06,
      "grad_norm": 0.2524053454399109,
      "learning_rate": 9.977825380819135e-06,
      "loss": 0.075,
      "step": 7850
    },
    {
      "epoch": 7.486343383789062e-06,
      "model_forward_time": 1.7558205127716064,
      "step": 7850
    },
    {
      "dynamics_loss_avg": 0.0663075476884842,
      "epoch": 7.486343383789062e-06,
      "step": 7850
    },
    {
      "action_loss_avg": 0.006013605697080493,
      "epoch": 7.486343383789062e-06,
      "step": 7850
    },
    {
      "epoch": 7.486343383789062e-06,
      "step": 7850,
      "training_step_time": 6.938995599746704
    },
    {
      "epoch": 7.487297058105469e-06,
      "model_forward_time": 1.7353477478027344,
      "step": 7851
    },
    {
      "epoch": 7.487297058105469e-06,
      "step": 7851,
      "training_step_time": 7.535049676895142
    },
    {
      "epoch": 7.488250732421875e-06,
      "model_forward_time": 1.7382843494415283,
      "step": 7852
    },
    {
      "epoch": 7.488250732421875e-06,
      "step": 7852,
      "training_step_time": 7.6524176597595215
    },
    {
      "epoch": 7.489204406738281e-06,
      "model_forward_time": 1.7700779438018799,
      "step": 7853
    },
    {
      "epoch": 7.489204406738281e-06,
      "step": 7853,
      "training_step_time": 8.844677209854126
    },
    {
      "epoch": 7.490158081054688e-06,
      "model_forward_time": 2.286983013153076,
      "step": 7854
    },
    {
      "epoch": 7.490158081054688e-06,
      "step": 7854,
      "training_step_time": 8.843470335006714
    },
    {
      "epoch": 7.491111755371094e-06,
      "model_forward_time": 1.736079216003418,
      "step": 7855
    },
    {
      "epoch": 7.491111755371094e-06,
      "step": 7855,
      "training_step_time": 9.152551174163818
    },
    {
      "epoch": 7.4920654296875e-06,
      "model_forward_time": 1.7563996315002441,
      "step": 7856
    },
    {
      "epoch": 7.4920654296875e-06,
      "step": 7856,
      "training_step_time": 7.816731691360474
    },
    {
      "epoch": 7.4930191040039064e-06,
      "model_forward_time": 1.7307190895080566,
      "step": 7857
    },
    {
      "epoch": 7.4930191040039064e-06,
      "step": 7857,
      "training_step_time": 8.174821853637695
    },
    {
      "epoch": 7.493972778320312e-06,
      "model_forward_time": 1.7376339435577393,
      "step": 7858
    },
    {
      "epoch": 7.493972778320312e-06,
      "step": 7858,
      "training_step_time": 7.516738414764404
    },
    {
      "epoch": 7.494926452636719e-06,
      "model_forward_time": 1.747185468673706,
      "step": 7859
    },
    {
      "epoch": 7.494926452636719e-06,
      "step": 7859,
      "training_step_time": 7.200251340866089
    },
    {
      "epoch": 7.495880126953125e-06,
      "grad_norm": 0.26183396577835083,
      "learning_rate": 9.97766955781922e-06,
      "loss": 0.0752,
      "step": 7860
    },
    {
      "epoch": 7.495880126953125e-06,
      "model_forward_time": 1.7265901565551758,
      "step": 7860
    },
    {
      "dynamics_loss_avg": 0.06412576250731945,
      "epoch": 7.495880126953125e-06,
      "step": 7860
    },
    {
      "action_loss_avg": 0.005995458120014519,
      "epoch": 7.495880126953125e-06,
      "step": 7860
    },
    {
      "epoch": 7.495880126953125e-06,
      "step": 7860,
      "training_step_time": 6.86834716796875
    },
    {
      "epoch": 7.496833801269531e-06,
      "model_forward_time": 1.7271389961242676,
      "step": 7861
    },
    {
      "epoch": 7.496833801269531e-06,
      "step": 7861,
      "training_step_time": 7.586978912353516
    },
    {
      "epoch": 7.497787475585938e-06,
      "model_forward_time": 1.7403175830841064,
      "step": 7862
    },
    {
      "epoch": 7.497787475585938e-06,
      "step": 7862,
      "training_step_time": 8.579004764556885
    },
    {
      "epoch": 7.498741149902344e-06,
      "model_forward_time": 2.300797462463379,
      "step": 7863
    },
    {
      "epoch": 7.498741149902344e-06,
      "step": 7863,
      "training_step_time": 7.791933536529541
    },
    {
      "epoch": 7.49969482421875e-06,
      "model_forward_time": 1.745546579360962,
      "step": 7864
    },
    {
      "epoch": 7.49969482421875e-06,
      "step": 7864,
      "training_step_time": 7.843997955322266
    },
    {
      "epoch": 7.5006484985351565e-06,
      "model_forward_time": 1.7190043926239014,
      "step": 7865
    },
    {
      "epoch": 7.5006484985351565e-06,
      "step": 7865,
      "training_step_time": 9.789142608642578
    },
    {
      "epoch": 7.501602172851562e-06,
      "model_forward_time": 1.7419140338897705,
      "step": 7866
    },
    {
      "epoch": 7.501602172851562e-06,
      "step": 7866,
      "training_step_time": 9.861778974533081
    },
    {
      "epoch": 7.502555847167969e-06,
      "model_forward_time": 1.743314266204834,
      "step": 7867
    },
    {
      "epoch": 7.502555847167969e-06,
      "step": 7867,
      "training_step_time": 7.28902792930603
    },
    {
      "epoch": 7.503509521484375e-06,
      "model_forward_time": 1.7502334117889404,
      "step": 7868
    },
    {
      "epoch": 7.503509521484375e-06,
      "step": 7868,
      "training_step_time": 7.543528079986572
    },
    {
      "epoch": 7.504463195800781e-06,
      "model_forward_time": 1.733034372329712,
      "step": 7869
    },
    {
      "epoch": 7.504463195800781e-06,
      "step": 7869,
      "training_step_time": 6.909995079040527
    },
    {
      "epoch": 7.505416870117188e-06,
      "grad_norm": 0.17661894857883453,
      "learning_rate": 9.977513190468848e-06,
      "loss": 0.0699,
      "step": 7870
    },
    {
      "epoch": 7.505416870117188e-06,
      "model_forward_time": 1.7274913787841797,
      "step": 7870
    },
    {
      "dynamics_loss_avg": 0.06472467165440321,
      "epoch": 7.505416870117188e-06,
      "step": 7870
    },
    {
      "action_loss_avg": 0.005383881158195436,
      "epoch": 7.505416870117188e-06,
      "step": 7870
    },
    {
      "epoch": 7.505416870117188e-06,
      "step": 7870,
      "training_step_time": 7.624708652496338
    },
    {
      "epoch": 7.506370544433594e-06,
      "model_forward_time": 1.7521655559539795,
      "step": 7871
    },
    {
      "epoch": 7.506370544433594e-06,
      "step": 7871,
      "training_step_time": 8.68529486656189
    },
    {
      "epoch": 7.50732421875e-06,
      "model_forward_time": 2.3128345012664795,
      "step": 7872
    },
    {
      "epoch": 7.50732421875e-06,
      "step": 7872,
      "training_step_time": 8.6795015335083
    },
    {
      "epoch": 7.5082778930664065e-06,
      "model_forward_time": 1.7447082996368408,
      "step": 7873
    },
    {
      "epoch": 7.5082778930664065e-06,
      "step": 7873,
      "training_step_time": 8.519664525985718
    },
    {
      "epoch": 7.509231567382812e-06,
      "model_forward_time": 1.7353050708770752,
      "step": 7874
    },
    {
      "epoch": 7.509231567382812e-06,
      "step": 7874,
      "training_step_time": 9.563466310501099
    },
    {
      "epoch": 7.510185241699219e-06,
      "model_forward_time": 1.7674446105957031,
      "step": 7875
    },
    {
      "epoch": 7.510185241699219e-06,
      "step": 7875,
      "training_step_time": 8.98526668548584
    },
    {
      "epoch": 7.511138916015625e-06,
      "model_forward_time": 1.7345340251922607,
      "step": 7876
    },
    {
      "epoch": 7.511138916015625e-06,
      "step": 7876,
      "training_step_time": 7.532201290130615
    },
    {
      "epoch": 7.512092590332031e-06,
      "model_forward_time": 1.726665735244751,
      "step": 7877
    },
    {
      "epoch": 7.512092590332031e-06,
      "step": 7877,
      "training_step_time": 7.540210723876953
    },
    {
      "epoch": 7.513046264648438e-06,
      "model_forward_time": 1.7329270839691162,
      "step": 7878
    },
    {
      "epoch": 7.513046264648438e-06,
      "step": 7878,
      "training_step_time": 6.97836971282959
    },
    {
      "epoch": 7.513999938964843e-06,
      "model_forward_time": 1.731564998626709,
      "step": 7879
    },
    {
      "epoch": 7.513999938964843e-06,
      "step": 7879,
      "training_step_time": 7.586609840393066
    },
    {
      "epoch": 7.51495361328125e-06,
      "grad_norm": 0.14812786877155304,
      "learning_rate": 9.977356278785116e-06,
      "loss": 0.0754,
      "step": 7880
    },
    {
      "epoch": 7.51495361328125e-06,
      "model_forward_time": 1.7414600849151611,
      "step": 7880
    },
    {
      "dynamics_loss_avg": 0.06828881502151489,
      "epoch": 7.51495361328125e-06,
      "step": 7880
    },
    {
      "action_loss_avg": 0.004567624838091433,
      "epoch": 7.51495361328125e-06,
      "step": 7880
    },
    {
      "epoch": 7.51495361328125e-06,
      "step": 7880,
      "training_step_time": 7.750746965408325
    },
    {
      "epoch": 7.5159072875976565e-06,
      "model_forward_time": 2.285461902618408,
      "step": 7881
    },
    {
      "epoch": 7.5159072875976565e-06,
      "step": 7881,
      "training_step_time": 7.811723947525024
    },
    {
      "epoch": 7.516860961914062e-06,
      "model_forward_time": 1.7602269649505615,
      "step": 7882
    },
    {
      "epoch": 7.516860961914062e-06,
      "step": 7882,
      "training_step_time": 9.687982559204102
    },
    {
      "epoch": 7.517814636230469e-06,
      "model_forward_time": 1.7198429107666016,
      "step": 7883
    },
    {
      "epoch": 7.517814636230469e-06,
      "step": 7883,
      "training_step_time": 8.912033081054688
    },
    {
      "epoch": 7.518768310546875e-06,
      "model_forward_time": 1.7335877418518066,
      "step": 7884
    },
    {
      "epoch": 7.518768310546875e-06,
      "step": 7884,
      "training_step_time": 8.488614320755005
    },
    {
      "epoch": 7.519721984863281e-06,
      "model_forward_time": 1.743879795074463,
      "step": 7885
    },
    {
      "epoch": 7.519721984863281e-06,
      "step": 7885,
      "training_step_time": 7.579191207885742
    },
    {
      "epoch": 7.520675659179688e-06,
      "model_forward_time": 1.7421791553497314,
      "step": 7886
    },
    {
      "epoch": 7.520675659179688e-06,
      "step": 7886,
      "training_step_time": 7.575944185256958
    },
    {
      "epoch": 7.521629333496093e-06,
      "model_forward_time": 1.7267343997955322,
      "step": 7887
    },
    {
      "epoch": 7.521629333496093e-06,
      "step": 7887,
      "training_step_time": 7.260051012039185
    },
    {
      "epoch": 7.5225830078125e-06,
      "model_forward_time": 1.726306676864624,
      "step": 7888
    },
    {
      "epoch": 7.5225830078125e-06,
      "step": 7888,
      "training_step_time": 7.644472599029541
    },
    {
      "epoch": 7.5235366821289065e-06,
      "model_forward_time": 1.748337745666504,
      "step": 7889
    },
    {
      "epoch": 7.5235366821289065e-06,
      "step": 7889,
      "training_step_time": 7.73524284362793
    },
    {
      "epoch": 7.524490356445312e-06,
      "grad_norm": 0.23134833574295044,
      "learning_rate": 9.977198822785185e-06,
      "loss": 0.0738,
      "step": 7890
    },
    {
      "epoch": 7.524490356445312e-06,
      "model_forward_time": 1.749544382095337,
      "step": 7890
    },
    {
      "dynamics_loss_avg": 0.07867912612855435,
      "epoch": 7.524490356445312e-06,
      "step": 7890
    },
    {
      "action_loss_avg": 0.005333625874482095,
      "epoch": 7.524490356445312e-06,
      "step": 7890
    },
    {
      "epoch": 7.524490356445312e-06,
      "step": 7890,
      "training_step_time": 8.443273544311523
    },
    {
      "epoch": 7.525444030761719e-06,
      "model_forward_time": 2.2878923416137695,
      "step": 7891
    },
    {
      "epoch": 7.525444030761719e-06,
      "step": 7891,
      "training_step_time": 9.576149225234985
    },
    {
      "epoch": 7.526397705078125e-06,
      "model_forward_time": 1.7311463356018066,
      "step": 7892
    },
    {
      "epoch": 7.526397705078125e-06,
      "step": 7892,
      "training_step_time": 9.476157903671265
    },
    {
      "epoch": 7.527351379394531e-06,
      "model_forward_time": 1.7361798286437988,
      "step": 7893
    },
    {
      "epoch": 7.527351379394531e-06,
      "step": 7893,
      "training_step_time": 8.492833614349365
    },
    {
      "epoch": 7.528305053710938e-06,
      "model_forward_time": 1.745009422302246,
      "step": 7894
    },
    {
      "epoch": 7.528305053710938e-06,
      "step": 7894,
      "training_step_time": 8.134218692779541
    },
    {
      "epoch": 7.529258728027343e-06,
      "model_forward_time": 1.7363500595092773,
      "step": 7895
    },
    {
      "epoch": 7.529258728027343e-06,
      "step": 7895,
      "training_step_time": 7.558633327484131
    },
    {
      "epoch": 7.53021240234375e-06,
      "model_forward_time": 1.7088754177093506,
      "step": 7896
    },
    {
      "epoch": 7.53021240234375e-06,
      "step": 7896,
      "training_step_time": 6.9879186153411865
    },
    {
      "epoch": 7.5311660766601565e-06,
      "model_forward_time": 1.7407050132751465,
      "step": 7897
    },
    {
      "epoch": 7.5311660766601565e-06,
      "step": 7897,
      "training_step_time": 7.654613733291626
    },
    {
      "epoch": 7.532119750976562e-06,
      "model_forward_time": 1.740680456161499,
      "step": 7898
    },
    {
      "epoch": 7.532119750976562e-06,
      "step": 7898,
      "training_step_time": 7.77014422416687
    },
    {
      "epoch": 7.533073425292969e-06,
      "model_forward_time": 1.7078702449798584,
      "step": 7899
    },
    {
      "epoch": 7.533073425292969e-06,
      "step": 7899,
      "training_step_time": 13.976548671722412
    },
    {
      "epoch": 7.534027099609375e-06,
      "grad_norm": 0.19125565886497498,
      "learning_rate": 9.977040822486274e-06,
      "loss": 0.0715,
      "step": 7900
    },
    {
      "epoch": 7.534027099609375e-06,
      "model_forward_time": 2.266907215118408,
      "step": 7900
    },
    {
      "dynamics_loss_avg": 0.06795847043395042,
      "epoch": 7.534027099609375e-06,
      "step": 7900
    },
    {
      "action_loss_avg": 0.004894051491282881,
      "epoch": 7.534027099609375e-06,
      "step": 7900
    },
    {
      "epoch": 7.534027099609375e-06,
      "step": 7900,
      "training_step_time": 8.35770034790039
    },
    {
      "epoch": 7.534980773925781e-06,
      "model_forward_time": 1.738182544708252,
      "step": 7901
    },
    {
      "epoch": 7.534980773925781e-06,
      "step": 7901,
      "training_step_time": 9.737043142318726
    },
    {
      "epoch": 7.535934448242188e-06,
      "model_forward_time": 1.7206525802612305,
      "step": 7902
    },
    {
      "epoch": 7.535934448242188e-06,
      "step": 7902,
      "training_step_time": 7.855074167251587
    },
    {
      "epoch": 7.536888122558593e-06,
      "model_forward_time": 1.7352092266082764,
      "step": 7903
    },
    {
      "epoch": 7.536888122558593e-06,
      "step": 7903,
      "training_step_time": 7.611919164657593
    },
    {
      "epoch": 7.537841796875e-06,
      "model_forward_time": 1.724665880203247,
      "step": 7904
    },
    {
      "epoch": 7.537841796875e-06,
      "step": 7904,
      "training_step_time": 7.290861129760742
    },
    {
      "epoch": 7.5387954711914065e-06,
      "model_forward_time": 1.7419729232788086,
      "step": 7905
    },
    {
      "epoch": 7.5387954711914065e-06,
      "step": 7905,
      "training_step_time": 7.58653450012207
    },
    {
      "epoch": 7.539749145507812e-06,
      "model_forward_time": 1.7309482097625732,
      "step": 7906
    },
    {
      "epoch": 7.539749145507812e-06,
      "step": 7906,
      "training_step_time": 6.981414794921875
    },
    {
      "epoch": 7.540702819824219e-06,
      "model_forward_time": 1.732041358947754,
      "step": 7907
    },
    {
      "epoch": 7.540702819824219e-06,
      "step": 7907,
      "training_step_time": 7.70305323600769
    },
    {
      "epoch": 7.541656494140625e-06,
      "model_forward_time": 1.726093053817749,
      "step": 7908
    },
    {
      "epoch": 7.541656494140625e-06,
      "step": 7908,
      "training_step_time": 8.873770952224731
    },
    {
      "epoch": 7.542610168457031e-06,
      "model_forward_time": 2.2966256141662598,
      "step": 7909
    },
    {
      "epoch": 7.542610168457031e-06,
      "step": 7909,
      "training_step_time": 7.880686521530151
    },
    {
      "epoch": 7.543563842773438e-06,
      "grad_norm": 0.2518332600593567,
      "learning_rate": 9.976882277905659e-06,
      "loss": 0.0751,
      "step": 7910
    },
    {
      "epoch": 7.543563842773438e-06,
      "model_forward_time": 1.740666389465332,
      "step": 7910
    },
    {
      "dynamics_loss_avg": 0.05527428640052676,
      "epoch": 7.543563842773438e-06,
      "step": 7910
    },
    {
      "action_loss_avg": 0.0040614542085677385,
      "epoch": 7.543563842773438e-06,
      "step": 7910
    },
    {
      "epoch": 7.543563842773438e-06,
      "step": 7910,
      "training_step_time": 8.556087732315063
    },
    {
      "epoch": 7.544517517089843e-06,
      "model_forward_time": 1.719233512878418,
      "step": 7911
    },
    {
      "epoch": 7.544517517089843e-06,
      "step": 7911,
      "training_step_time": 8.875558376312256
    },
    {
      "epoch": 7.54547119140625e-06,
      "model_forward_time": 1.721522331237793,
      "step": 7912
    },
    {
      "epoch": 7.54547119140625e-06,
      "step": 7912,
      "training_step_time": 8.901342153549194
    },
    {
      "epoch": 7.5464248657226565e-06,
      "model_forward_time": 1.734973669052124,
      "step": 7913
    },
    {
      "epoch": 7.5464248657226565e-06,
      "step": 7913,
      "training_step_time": 7.552626371383667
    },
    {
      "epoch": 7.547378540039062e-06,
      "model_forward_time": 1.704744577407837,
      "step": 7914
    },
    {
      "epoch": 7.547378540039062e-06,
      "step": 7914,
      "training_step_time": 7.640212297439575
    },
    {
      "epoch": 7.548332214355469e-06,
      "model_forward_time": 1.7521865367889404,
      "step": 7915
    },
    {
      "epoch": 7.548332214355469e-06,
      "step": 7915,
      "training_step_time": 7.025241136550903
    },
    {
      "epoch": 7.549285888671875e-06,
      "model_forward_time": 1.726614236831665,
      "step": 7916
    },
    {
      "epoch": 7.549285888671875e-06,
      "step": 7916,
      "training_step_time": 7.648890495300293
    },
    {
      "epoch": 7.550239562988281e-06,
      "model_forward_time": 1.7494070529937744,
      "step": 7917
    },
    {
      "epoch": 7.550239562988281e-06,
      "step": 7917,
      "training_step_time": 7.923696279525757
    },
    {
      "epoch": 7.551193237304688e-06,
      "model_forward_time": 2.278684139251709,
      "step": 7918
    },
    {
      "epoch": 7.551193237304688e-06,
      "step": 7918,
      "training_step_time": 8.521758794784546
    },
    {
      "epoch": 7.552146911621093e-06,
      "model_forward_time": 1.7302477359771729,
      "step": 7919
    },
    {
      "epoch": 7.552146911621093e-06,
      "step": 7919,
      "training_step_time": 9.110249757766724
    },
    {
      "epoch": 7.5531005859375e-06,
      "grad_norm": 0.16986681520938873,
      "learning_rate": 9.976723189060685e-06,
      "loss": 0.0685,
      "step": 7920
    },
    {
      "epoch": 7.5531005859375e-06,
      "model_forward_time": 1.7135987281799316,
      "step": 7920
    },
    {
      "dynamics_loss_avg": 0.07005714792758226,
      "epoch": 7.5531005859375e-06,
      "step": 7920
    },
    {
      "action_loss_avg": 0.005831453436985612,
      "epoch": 7.5531005859375e-06,
      "step": 7920
    },
    {
      "epoch": 7.5531005859375e-06,
      "step": 7920,
      "training_step_time": 8.675279140472412
    },
    {
      "epoch": 7.5540542602539065e-06,
      "model_forward_time": 1.7418584823608398,
      "step": 7921
    },
    {
      "epoch": 7.5540542602539065e-06,
      "step": 7921,
      "training_step_time": 7.843184232711792
    },
    {
      "epoch": 7.555007934570312e-06,
      "model_forward_time": 1.7306725978851318,
      "step": 7922
    },
    {
      "epoch": 7.555007934570312e-06,
      "step": 7922,
      "training_step_time": 8.012783765792847
    },
    {
      "epoch": 7.555961608886719e-06,
      "model_forward_time": 1.737851619720459,
      "step": 7923
    },
    {
      "epoch": 7.555961608886719e-06,
      "step": 7923,
      "training_step_time": 7.624800205230713
    },
    {
      "epoch": 7.556915283203125e-06,
      "model_forward_time": 1.7445361614227295,
      "step": 7924
    },
    {
      "epoch": 7.556915283203125e-06,
      "step": 7924,
      "training_step_time": 7.012980937957764
    },
    {
      "epoch": 7.557868957519531e-06,
      "model_forward_time": 1.7179820537567139,
      "step": 7925
    },
    {
      "epoch": 7.557868957519531e-06,
      "step": 7925,
      "training_step_time": 7.6183226108551025
    },
    {
      "epoch": 7.558822631835938e-06,
      "model_forward_time": 1.7272539138793945,
      "step": 7926
    },
    {
      "epoch": 7.558822631835938e-06,
      "step": 7926,
      "training_step_time": 7.924167633056641
    },
    {
      "epoch": 7.559776306152343e-06,
      "model_forward_time": 2.2871100902557373,
      "step": 7927
    },
    {
      "epoch": 7.559776306152343e-06,
      "step": 7927,
      "training_step_time": 8.41203784942627
    },
    {
      "epoch": 7.56072998046875e-06,
      "model_forward_time": 1.3560950756072998,
      "step": 7928
    },
    {
      "epoch": 7.56072998046875e-06,
      "step": 7928,
      "training_step_time": 8.84194016456604
    },
    {
      "epoch": 7.5616836547851565e-06,
      "model_forward_time": 1.7310776710510254,
      "step": 7929
    },
    {
      "epoch": 7.5616836547851565e-06,
      "step": 7929,
      "training_step_time": 10.87009334564209
    },
    {
      "epoch": 7.562637329101562e-06,
      "grad_norm": 0.21470144391059875,
      "learning_rate": 9.976563555968743e-06,
      "loss": 0.0692,
      "step": 7930
    },
    {
      "epoch": 7.562637329101562e-06,
      "model_forward_time": 1.7357769012451172,
      "step": 7930
    },
    {
      "dynamics_loss_avg": 0.07557281292974949,
      "epoch": 7.562637329101562e-06,
      "step": 7930
    },
    {
      "action_loss_avg": 0.0048417166923172775,
      "epoch": 7.562637329101562e-06,
      "step": 7930
    },
    {
      "epoch": 7.562637329101562e-06,
      "step": 7930,
      "training_step_time": 8.57998776435852
    },
    {
      "epoch": 7.563591003417969e-06,
      "model_forward_time": 1.7506930828094482,
      "step": 7931
    },
    {
      "epoch": 7.563591003417969e-06,
      "step": 7931,
      "training_step_time": 7.577056169509888
    },
    {
      "epoch": 7.564544677734375e-06,
      "model_forward_time": 1.7438039779663086,
      "step": 7932
    },
    {
      "epoch": 7.564544677734375e-06,
      "step": 7932,
      "training_step_time": 7.559271812438965
    },
    {
      "epoch": 7.565498352050781e-06,
      "model_forward_time": 1.7136030197143555,
      "step": 7933
    },
    {
      "epoch": 7.565498352050781e-06,
      "step": 7933,
      "training_step_time": 7.249187231063843
    },
    {
      "epoch": 7.566452026367188e-06,
      "model_forward_time": 1.7126643657684326,
      "step": 7934
    },
    {
      "epoch": 7.566452026367188e-06,
      "step": 7934,
      "training_step_time": 8.451022863388062
    },
    {
      "epoch": 7.567405700683593e-06,
      "model_forward_time": 1.7156004905700684,
      "step": 7935
    },
    {
      "epoch": 7.567405700683593e-06,
      "step": 7935,
      "training_step_time": 7.6983256340026855
    },
    {
      "epoch": 7.568359375e-06,
      "model_forward_time": 1.7281336784362793,
      "step": 7936
    },
    {
      "epoch": 7.568359375e-06,
      "step": 7936,
      "training_step_time": 7.838244438171387
    },
    {
      "epoch": 7.5693130493164065e-06,
      "model_forward_time": 2.257197618484497,
      "step": 7937
    },
    {
      "epoch": 7.5693130493164065e-06,
      "step": 7937,
      "training_step_time": 8.785277843475342
    },
    {
      "epoch": 7.570266723632812e-06,
      "model_forward_time": 1.7092328071594238,
      "step": 7938
    },
    {
      "epoch": 7.570266723632812e-06,
      "step": 7938,
      "training_step_time": 8.813960313796997
    },
    {
      "epoch": 7.571220397949219e-06,
      "model_forward_time": 1.6854417324066162,
      "step": 7939
    },
    {
      "epoch": 7.571220397949219e-06,
      "step": 7939,
      "training_step_time": 8.571358919143677
    },
    {
      "epoch": 7.572174072265625e-06,
      "grad_norm": 0.16716314852237701,
      "learning_rate": 9.976403378647292e-06,
      "loss": 0.066,
      "step": 7940
    },
    {
      "epoch": 7.572174072265625e-06,
      "model_forward_time": 1.7147696018218994,
      "step": 7940
    },
    {
      "dynamics_loss_avg": 0.06333110183477401,
      "epoch": 7.572174072265625e-06,
      "step": 7940
    },
    {
      "action_loss_avg": 0.0040751462453044954,
      "epoch": 7.572174072265625e-06,
      "step": 7940
    },
    {
      "epoch": 7.572174072265625e-06,
      "step": 7940,
      "training_step_time": 7.642047166824341
    },
    {
      "epoch": 7.573127746582031e-06,
      "model_forward_time": 1.716015100479126,
      "step": 7941
    },
    {
      "epoch": 7.573127746582031e-06,
      "step": 7941,
      "training_step_time": 7.363878488540649
    },
    {
      "epoch": 7.574081420898438e-06,
      "model_forward_time": 1.722637414932251,
      "step": 7942
    },
    {
      "epoch": 7.574081420898438e-06,
      "step": 7942,
      "training_step_time": 7.602309703826904
    },
    {
      "epoch": 7.575035095214843e-06,
      "model_forward_time": 1.718531847000122,
      "step": 7943
    },
    {
      "epoch": 7.575035095214843e-06,
      "step": 7943,
      "training_step_time": 7.673219442367554
    },
    {
      "epoch": 7.57598876953125e-06,
      "model_forward_time": 1.7229671478271484,
      "step": 7944
    },
    {
      "epoch": 7.57598876953125e-06,
      "step": 7944,
      "training_step_time": 7.702618360519409
    },
    {
      "epoch": 7.5769424438476565e-06,
      "model_forward_time": 1.709836483001709,
      "step": 7945
    },
    {
      "epoch": 7.5769424438476565e-06,
      "step": 7945,
      "training_step_time": 10.218364953994751
    },
    {
      "epoch": 7.577896118164062e-06,
      "model_forward_time": 2.2576777935028076,
      "step": 7946
    },
    {
      "epoch": 7.577896118164062e-06,
      "step": 7946,
      "training_step_time": 8.516255140304565
    },
    {
      "epoch": 7.578849792480469e-06,
      "model_forward_time": 1.7321856021881104,
      "step": 7947
    },
    {
      "epoch": 7.578849792480469e-06,
      "step": 7947,
      "training_step_time": 8.849306106567383
    },
    {
      "epoch": 7.579803466796875e-06,
      "model_forward_time": 1.7237160205841064,
      "step": 7948
    },
    {
      "epoch": 7.579803466796875e-06,
      "step": 7948,
      "training_step_time": 8.516265392303467
    },
    {
      "epoch": 7.580757141113281e-06,
      "model_forward_time": 1.7176916599273682,
      "step": 7949
    },
    {
      "epoch": 7.580757141113281e-06,
      "step": 7949,
      "training_step_time": 7.7715113162994385
    },
    {
      "epoch": 7.581710815429688e-06,
      "grad_norm": 0.21179607510566711,
      "learning_rate": 9.97624265711385e-06,
      "loss": 0.0642,
      "step": 7950
    },
    {
      "epoch": 7.581710815429688e-06,
      "model_forward_time": 1.7045187950134277,
      "step": 7950
    },
    {
      "dynamics_loss_avg": 0.053681754320859906,
      "epoch": 7.581710815429688e-06,
      "step": 7950
    },
    {
      "action_loss_avg": 0.005138794472441077,
      "epoch": 7.581710815429688e-06,
      "step": 7950
    },
    {
      "epoch": 7.581710815429688e-06,
      "step": 7950,
      "training_step_time": 7.539780378341675
    },
    {
      "epoch": 7.582664489746093e-06,
      "model_forward_time": 1.7185497283935547,
      "step": 7951
    },
    {
      "epoch": 7.582664489746093e-06,
      "step": 7951,
      "training_step_time": 7.602722883224487
    },
    {
      "epoch": 7.5836181640625e-06,
      "model_forward_time": 1.7199008464813232,
      "step": 7952
    },
    {
      "epoch": 7.5836181640625e-06,
      "step": 7952,
      "training_step_time": 6.907728672027588
    },
    {
      "epoch": 7.5845718383789065e-06,
      "model_forward_time": 1.7073745727539062,
      "step": 7953
    },
    {
      "epoch": 7.5845718383789065e-06,
      "step": 7953,
      "training_step_time": 7.591426610946655
    },
    {
      "epoch": 7.585525512695312e-06,
      "model_forward_time": 1.7135272026062012,
      "step": 7954
    },
    {
      "epoch": 7.585525512695312e-06,
      "step": 7954,
      "training_step_time": 8.534650802612305
    },
    {
      "epoch": 7.586479187011719e-06,
      "model_forward_time": 2.2686514854431152,
      "step": 7955
    },
    {
      "epoch": 7.586479187011719e-06,
      "step": 7955,
      "training_step_time": 7.774377822875977
    },
    {
      "epoch": 7.587432861328125e-06,
      "model_forward_time": 1.7249484062194824,
      "step": 7956
    },
    {
      "epoch": 7.587432861328125e-06,
      "step": 7956,
      "training_step_time": 8.65068507194519
    },
    {
      "epoch": 7.588386535644531e-06,
      "model_forward_time": 1.7344529628753662,
      "step": 7957
    },
    {
      "epoch": 7.588386535644531e-06,
      "step": 7957,
      "training_step_time": 9.158933162689209
    },
    {
      "epoch": 7.589340209960938e-06,
      "model_forward_time": 1.718249797821045,
      "step": 7958
    },
    {
      "epoch": 7.589340209960938e-06,
      "step": 7958,
      "training_step_time": 8.727589130401611
    },
    {
      "epoch": 7.590293884277343e-06,
      "model_forward_time": 1.7173888683319092,
      "step": 7959
    },
    {
      "epoch": 7.590293884277343e-06,
      "step": 7959,
      "training_step_time": 8.38913083076477
    },
    {
      "epoch": 7.59124755859375e-06,
      "grad_norm": 0.1753171980381012,
      "learning_rate": 9.976081391385994e-06,
      "loss": 0.0661,
      "step": 7960
    },
    {
      "epoch": 7.59124755859375e-06,
      "model_forward_time": 1.714339017868042,
      "step": 7960
    },
    {
      "dynamics_loss_avg": 0.058375464752316475,
      "epoch": 7.59124755859375e-06,
      "step": 7960
    },
    {
      "action_loss_avg": 0.005603703646920621,
      "epoch": 7.59124755859375e-06,
      "step": 7960
    },
    {
      "epoch": 7.59124755859375e-06,
      "step": 7960,
      "training_step_time": 7.57144021987915
    },
    {
      "epoch": 7.5922012329101565e-06,
      "model_forward_time": 1.710026502609253,
      "step": 7961
    },
    {
      "epoch": 7.5922012329101565e-06,
      "step": 7961,
      "training_step_time": 7.2375328540802
    },
    {
      "epoch": 7.593154907226562e-06,
      "model_forward_time": 1.7094171047210693,
      "step": 7962
    },
    {
      "epoch": 7.593154907226562e-06,
      "step": 7962,
      "training_step_time": 7.635523319244385
    },
    {
      "epoch": 7.594108581542969e-06,
      "model_forward_time": 1.729304552078247,
      "step": 7963
    },
    {
      "epoch": 7.594108581542969e-06,
      "step": 7963,
      "training_step_time": 8.693523168563843
    },
    {
      "epoch": 7.595062255859375e-06,
      "model_forward_time": 2.260859727859497,
      "step": 7964
    },
    {
      "epoch": 7.595062255859375e-06,
      "step": 7964,
      "training_step_time": 7.807267665863037
    },
    {
      "epoch": 7.596015930175781e-06,
      "model_forward_time": 1.720641851425171,
      "step": 7965
    },
    {
      "epoch": 7.596015930175781e-06,
      "step": 7965,
      "training_step_time": 9.211322784423828
    },
    {
      "epoch": 7.596969604492188e-06,
      "model_forward_time": 1.7074921131134033,
      "step": 7966
    },
    {
      "epoch": 7.596969604492188e-06,
      "step": 7966,
      "training_step_time": 7.759696006774902
    },
    {
      "epoch": 7.597923278808593e-06,
      "model_forward_time": 1.7241933345794678,
      "step": 7967
    },
    {
      "epoch": 7.597923278808593e-06,
      "step": 7967,
      "training_step_time": 7.669165372848511
    },
    {
      "epoch": 7.598876953125e-06,
      "model_forward_time": 1.7286789417266846,
      "step": 7968
    },
    {
      "epoch": 7.598876953125e-06,
      "step": 7968,
      "training_step_time": 7.551892280578613
    },
    {
      "epoch": 7.5998306274414065e-06,
      "model_forward_time": 1.6979358196258545,
      "step": 7969
    },
    {
      "epoch": 7.5998306274414065e-06,
      "step": 7969,
      "training_step_time": 7.510668754577637
    },
    {
      "epoch": 7.600784301757812e-06,
      "grad_norm": 0.18767572939395905,
      "learning_rate": 9.975919581481356e-06,
      "loss": 0.0625,
      "step": 7970
    },
    {
      "epoch": 7.600784301757812e-06,
      "model_forward_time": 1.6970770359039307,
      "step": 7970
    },
    {
      "dynamics_loss_avg": 0.048904032073915006,
      "epoch": 7.600784301757812e-06,
      "step": 7970
    },
    {
      "action_loss_avg": 0.004248474270571023,
      "epoch": 7.600784301757812e-06,
      "step": 7970
    },
    {
      "epoch": 7.600784301757812e-06,
      "step": 7970,
      "training_step_time": 7.08123779296875
    },
    {
      "epoch": 7.601737976074219e-06,
      "model_forward_time": 1.727299690246582,
      "step": 7971
    },
    {
      "epoch": 7.601737976074219e-06,
      "step": 7971,
      "training_step_time": 7.616254568099976
    },
    {
      "epoch": 7.602691650390625e-06,
      "model_forward_time": 1.7273468971252441,
      "step": 7972
    },
    {
      "epoch": 7.602691650390625e-06,
      "step": 7972,
      "training_step_time": 7.903543949127197
    },
    {
      "epoch": 7.603645324707031e-06,
      "model_forward_time": 1.74871826171875,
      "step": 7973
    },
    {
      "epoch": 7.603645324707031e-06,
      "step": 7973,
      "training_step_time": 9.637687921524048
    },
    {
      "epoch": 7.604598999023438e-06,
      "model_forward_time": 1.7187726497650146,
      "step": 7974
    },
    {
      "epoch": 7.604598999023438e-06,
      "step": 7974,
      "training_step_time": 9.506877660751343
    },
    {
      "epoch": 7.605552673339843e-06,
      "model_forward_time": 1.7119522094726562,
      "step": 7975
    },
    {
      "epoch": 7.605552673339843e-06,
      "step": 7975,
      "training_step_time": 9.465423583984375
    },
    {
      "epoch": 7.60650634765625e-06,
      "model_forward_time": 1.7122907638549805,
      "step": 7976
    },
    {
      "epoch": 7.60650634765625e-06,
      "step": 7976,
      "training_step_time": 7.774153709411621
    },
    {
      "epoch": 7.6074600219726565e-06,
      "model_forward_time": 1.714669942855835,
      "step": 7977
    },
    {
      "epoch": 7.6074600219726565e-06,
      "step": 7977,
      "training_step_time": 7.647144317626953
    },
    {
      "epoch": 7.608413696289062e-06,
      "model_forward_time": 1.3298990726470947,
      "step": 7978
    },
    {
      "epoch": 7.608413696289062e-06,
      "step": 7978,
      "training_step_time": 7.307281970977783
    },
    {
      "epoch": 7.609367370605469e-06,
      "model_forward_time": 1.724074125289917,
      "step": 7979
    },
    {
      "epoch": 7.609367370605469e-06,
      "step": 7979,
      "training_step_time": 7.562083959579468
    },
    {
      "epoch": 7.610321044921875e-06,
      "grad_norm": 0.23687534034252167,
      "learning_rate": 9.975757227417636e-06,
      "loss": 0.0664,
      "step": 7980
    },
    {
      "epoch": 7.610321044921875e-06,
      "model_forward_time": 1.7157349586486816,
      "step": 7980
    },
    {
      "dynamics_loss_avg": 0.047438545152544975,
      "epoch": 7.610321044921875e-06,
      "step": 7980
    },
    {
      "action_loss_avg": 0.0038518515997566284,
      "epoch": 7.610321044921875e-06,
      "step": 7980
    },
    {
      "epoch": 7.610321044921875e-06,
      "step": 7980,
      "training_step_time": 8.443349361419678
    },
    {
      "epoch": 7.611274719238281e-06,
      "model_forward_time": 1.7235910892486572,
      "step": 7981
    },
    {
      "epoch": 7.611274719238281e-06,
      "step": 7981,
      "training_step_time": 7.662112474441528
    },
    {
      "epoch": 7.612228393554688e-06,
      "model_forward_time": 1.7353301048278809,
      "step": 7982
    },
    {
      "epoch": 7.612228393554688e-06,
      "step": 7982,
      "training_step_time": 8.708899736404419
    },
    {
      "epoch": 7.613182067871093e-06,
      "model_forward_time": 2.3125429153442383,
      "step": 7983
    },
    {
      "epoch": 7.613182067871093e-06,
      "step": 7983,
      "training_step_time": 8.581626415252686
    },
    {
      "epoch": 7.6141357421875e-06,
      "model_forward_time": 1.714859962463379,
      "step": 7984
    },
    {
      "epoch": 7.6141357421875e-06,
      "step": 7984,
      "training_step_time": 9.500184535980225
    },
    {
      "epoch": 7.6150894165039066e-06,
      "model_forward_time": 1.723508358001709,
      "step": 7985
    },
    {
      "epoch": 7.6150894165039066e-06,
      "step": 7985,
      "training_step_time": 8.8148832321167
    },
    {
      "epoch": 7.616043090820312e-06,
      "model_forward_time": 1.7056736946105957,
      "step": 7986
    },
    {
      "epoch": 7.616043090820312e-06,
      "step": 7986,
      "training_step_time": 8.62106704711914
    },
    {
      "epoch": 7.616996765136719e-06,
      "model_forward_time": 1.714341402053833,
      "step": 7987
    },
    {
      "epoch": 7.616996765136719e-06,
      "step": 7987,
      "training_step_time": 7.346422433853149
    },
    {
      "epoch": 7.617950439453125e-06,
      "model_forward_time": 1.7044806480407715,
      "step": 7988
    },
    {
      "epoch": 7.617950439453125e-06,
      "step": 7988,
      "training_step_time": 7.598581314086914
    },
    {
      "epoch": 7.618904113769531e-06,
      "model_forward_time": 1.7290027141571045,
      "step": 7989
    },
    {
      "epoch": 7.618904113769531e-06,
      "step": 7989,
      "training_step_time": 7.567103862762451
    },
    {
      "epoch": 7.619857788085938e-06,
      "grad_norm": 0.2004612237215042,
      "learning_rate": 9.975594329212586e-06,
      "loss": 0.0646,
      "step": 7990
    },
    {
      "epoch": 7.619857788085938e-06,
      "model_forward_time": 1.7211596965789795,
      "step": 7990
    },
    {
      "dynamics_loss_avg": 0.0589907344430685,
      "epoch": 7.619857788085938e-06,
      "step": 7990
    },
    {
      "action_loss_avg": 0.004122729366645217,
      "epoch": 7.619857788085938e-06,
      "step": 7990
    },
    {
      "epoch": 7.619857788085938e-06,
      "step": 7990,
      "training_step_time": 7.617110013961792
    },
    {
      "epoch": 7.6208114624023434e-06,
      "model_forward_time": 1.7282941341400146,
      "step": 7991
    },
    {
      "epoch": 7.6208114624023434e-06,
      "step": 7991,
      "training_step_time": 8.858495473861694
    },
    {
      "epoch": 7.62176513671875e-06,
      "model_forward_time": 2.2744791507720947,
      "step": 7992
    },
    {
      "epoch": 7.62176513671875e-06,
      "step": 7992,
      "training_step_time": 8.529513359069824
    },
    {
      "epoch": 7.6227188110351566e-06,
      "model_forward_time": 1.7111585140228271,
      "step": 7993
    },
    {
      "epoch": 7.6227188110351566e-06,
      "step": 7993,
      "training_step_time": 9.611681461334229
    },
    {
      "epoch": 7.623672485351562e-06,
      "model_forward_time": 1.7202363014221191,
      "step": 7994
    },
    {
      "epoch": 7.623672485351562e-06,
      "step": 7994,
      "training_step_time": 7.775788068771362
    },
    {
      "epoch": 7.624626159667969e-06,
      "model_forward_time": 1.7166357040405273,
      "step": 7995
    },
    {
      "epoch": 7.624626159667969e-06,
      "step": 7995,
      "training_step_time": 8.593505144119263
    },
    {
      "epoch": 7.625579833984375e-06,
      "model_forward_time": 1.7175979614257812,
      "step": 7996
    },
    {
      "epoch": 7.625579833984375e-06,
      "step": 7996,
      "training_step_time": 7.302432298660278
    },
    {
      "epoch": 7.626533508300781e-06,
      "model_forward_time": 1.7138557434082031,
      "step": 7997
    },
    {
      "epoch": 7.626533508300781e-06,
      "step": 7997,
      "training_step_time": 7.384113311767578
    },
    {
      "epoch": 7.627487182617188e-06,
      "model_forward_time": 1.7487378120422363,
      "step": 7998
    },
    {
      "epoch": 7.627487182617188e-06,
      "step": 7998,
      "training_step_time": 7.766331911087036
    },
    {
      "epoch": 7.6284408569335934e-06,
      "model_forward_time": 1.7305731773376465,
      "step": 7999
    },
    {
      "epoch": 7.6284408569335934e-06,
      "step": 7999,
      "training_step_time": 7.496451377868652
    },
    {
      "epoch": 7.62939453125e-06,
      "grad_norm": 0.45555052161216736,
      "learning_rate": 9.97543088688402e-06,
      "loss": 0.0646,
      "step": 8000
    },
    {
      "epoch": 7.62939453125e-06,
      "model_forward_time": 1.7588863372802734,
      "step": 8000
    },
    {
      "dynamics_loss_avg": 0.07059901766479015,
      "epoch": 7.62939453125e-06,
      "step": 8000
    },
    {
      "action_loss_avg": 0.004947528266347945,
      "epoch": 7.62939453125e-06,
      "step": 8000
    },
    {
      "epoch": 7.62939453125e-06,
      "step": 8000,
      "training_step_time": 10.281838178634644
    },
    {
      "epoch": 7.630348205566406e-06,
      "model_forward_time": 1.7343568801879883,
      "step": 8001
    },
    {
      "epoch": 7.630348205566406e-06,
      "step": 8001,
      "training_step_time": 10.58604383468628
    },
    {
      "epoch": 7.631301879882813e-06,
      "model_forward_time": 1.7620060443878174,
      "step": 8002
    },
    {
      "epoch": 7.631301879882813e-06,
      "step": 8002,
      "training_step_time": 10.086180686950684
    },
    {
      "epoch": 7.632255554199219e-06,
      "model_forward_time": 1.7432775497436523,
      "step": 8003
    },
    {
      "epoch": 7.632255554199219e-06,
      "step": 8003,
      "training_step_time": 9.01383090019226
    },
    {
      "epoch": 7.633209228515625e-06,
      "model_forward_time": 1.7566258907318115,
      "step": 8004
    },
    {
      "epoch": 7.633209228515625e-06,
      "step": 8004,
      "training_step_time": 9.357642412185669
    },
    {
      "epoch": 7.634162902832032e-06,
      "model_forward_time": 1.7367126941680908,
      "step": 8005
    },
    {
      "epoch": 7.634162902832032e-06,
      "step": 8005,
      "training_step_time": 7.340139389038086
    },
    {
      "epoch": 7.635116577148438e-06,
      "model_forward_time": 1.7385904788970947,
      "step": 8006
    },
    {
      "epoch": 7.635116577148438e-06,
      "step": 8006,
      "training_step_time": 7.180897951126099
    },
    {
      "epoch": 7.636070251464843e-06,
      "model_forward_time": 1.7394967079162598,
      "step": 8007
    },
    {
      "epoch": 7.636070251464843e-06,
      "step": 8007,
      "training_step_time": 7.501365661621094
    },
    {
      "epoch": 7.63702392578125e-06,
      "model_forward_time": 1.7414140701293945,
      "step": 8008
    },
    {
      "epoch": 7.63702392578125e-06,
      "step": 8008,
      "training_step_time": 8.274939775466919
    },
    {
      "epoch": 7.637977600097657e-06,
      "model_forward_time": 1.7397143840789795,
      "step": 8009
    },
    {
      "epoch": 7.637977600097657e-06,
      "step": 8009,
      "training_step_time": 9.069757223129272
    },
    {
      "epoch": 7.638931274414062e-06,
      "grad_norm": 0.13901016116142273,
      "learning_rate": 9.975266900449814e-06,
      "loss": 0.0681,
      "step": 8010
    },
    {
      "epoch": 7.638931274414062e-06,
      "model_forward_time": 2.885650873184204,
      "step": 8010
    },
    {
      "dynamics_loss_avg": 0.0614255765452981,
      "epoch": 7.638931274414062e-06,
      "step": 8010
    },
    {
      "action_loss_avg": 0.004309108410961926,
      "epoch": 7.638931274414062e-06,
      "step": 8010
    },
    {
      "epoch": 7.638931274414062e-06,
      "step": 8010,
      "training_step_time": 9.473660707473755
    },
    {
      "epoch": 7.639884948730468e-06,
      "model_forward_time": 1.7516555786132812,
      "step": 8011
    },
    {
      "epoch": 7.639884948730468e-06,
      "step": 8011,
      "training_step_time": 10.782606363296509
    },
    {
      "epoch": 7.640838623046875e-06,
      "model_forward_time": 1.7410633563995361,
      "step": 8012
    },
    {
      "epoch": 7.640838623046875e-06,
      "step": 8012,
      "training_step_time": 8.37791395187378
    },
    {
      "epoch": 7.641792297363281e-06,
      "model_forward_time": 1.3897490501403809,
      "step": 8013
    },
    {
      "epoch": 7.641792297363281e-06,
      "step": 8013,
      "training_step_time": 7.705017328262329
    },
    {
      "epoch": 7.642745971679687e-06,
      "model_forward_time": 1.7390964031219482,
      "step": 8014
    },
    {
      "epoch": 7.642745971679687e-06,
      "step": 8014,
      "training_step_time": 7.535479545593262
    },
    {
      "epoch": 7.643699645996094e-06,
      "model_forward_time": 1.7417283058166504,
      "step": 8015
    },
    {
      "epoch": 7.643699645996094e-06,
      "step": 8015,
      "training_step_time": 7.146787166595459
    },
    {
      "epoch": 7.6446533203125e-06,
      "model_forward_time": 1.7488212585449219,
      "step": 8016
    },
    {
      "epoch": 7.6446533203125e-06,
      "step": 8016,
      "training_step_time": 6.976592063903809
    },
    {
      "epoch": 7.645606994628906e-06,
      "model_forward_time": 1.6973366737365723,
      "step": 8017
    },
    {
      "epoch": 7.645606994628906e-06,
      "step": 8017,
      "training_step_time": 7.491565942764282
    },
    {
      "epoch": 7.646560668945313e-06,
      "model_forward_time": 1.7101898193359375,
      "step": 8018
    },
    {
      "epoch": 7.646560668945313e-06,
      "step": 8018,
      "training_step_time": 9.07066035270691
    },
    {
      "epoch": 7.647514343261719e-06,
      "model_forward_time": 2.298807382583618,
      "step": 8019
    },
    {
      "epoch": 7.647514343261719e-06,
      "step": 8019,
      "training_step_time": 8.51760196685791
    },
    {
      "epoch": 7.648468017578125e-06,
      "grad_norm": 0.18267522752285004,
      "learning_rate": 9.975102369927898e-06,
      "loss": 0.0709,
      "step": 8020
    },
    {
      "epoch": 7.648468017578125e-06,
      "model_forward_time": 1.7240617275238037,
      "step": 8020
    },
    {
      "dynamics_loss_avg": 0.06279968693852425,
      "epoch": 7.648468017578125e-06,
      "step": 8020
    },
    {
      "action_loss_avg": 0.005731298157479614,
      "epoch": 7.648468017578125e-06,
      "step": 8020
    },
    {
      "epoch": 7.648468017578125e-06,
      "step": 8020,
      "training_step_time": 7.686482906341553
    },
    {
      "epoch": 7.649421691894532e-06,
      "model_forward_time": 1.718799114227295,
      "step": 8021
    },
    {
      "epoch": 7.649421691894532e-06,
      "step": 8021,
      "training_step_time": 8.359386205673218
    },
    {
      "epoch": 7.650375366210938e-06,
      "model_forward_time": 1.731313943862915,
      "step": 8022
    },
    {
      "epoch": 7.650375366210938e-06,
      "step": 8022,
      "training_step_time": 7.763052225112915
    },
    {
      "epoch": 7.651329040527343e-06,
      "model_forward_time": 1.7154357433319092,
      "step": 8023
    },
    {
      "epoch": 7.651329040527343e-06,
      "step": 8023,
      "training_step_time": 7.53818678855896
    },
    {
      "epoch": 7.65228271484375e-06,
      "model_forward_time": 1.7269785404205322,
      "step": 8024
    },
    {
      "epoch": 7.65228271484375e-06,
      "step": 8024,
      "training_step_time": 7.330883979797363
    },
    {
      "epoch": 7.653236389160157e-06,
      "model_forward_time": 1.7308218479156494,
      "step": 8025
    },
    {
      "epoch": 7.653236389160157e-06,
      "step": 8025,
      "training_step_time": 7.109152793884277
    },
    {
      "epoch": 7.654190063476562e-06,
      "model_forward_time": 1.7183828353881836,
      "step": 8026
    },
    {
      "epoch": 7.654190063476562e-06,
      "step": 8026,
      "training_step_time": 7.440191745758057
    },
    {
      "epoch": 7.655143737792968e-06,
      "model_forward_time": 1.7004611492156982,
      "step": 8027
    },
    {
      "epoch": 7.655143737792968e-06,
      "step": 8027,
      "training_step_time": 7.743225574493408
    },
    {
      "epoch": 7.656097412109375e-06,
      "model_forward_time": 2.2696714401245117,
      "step": 8028
    },
    {
      "epoch": 7.656097412109375e-06,
      "step": 8028,
      "training_step_time": 8.352495431900024
    },
    {
      "epoch": 7.657051086425781e-06,
      "model_forward_time": 1.7120368480682373,
      "step": 8029
    },
    {
      "epoch": 7.657051086425781e-06,
      "step": 8029,
      "training_step_time": 7.612245798110962
    },
    {
      "epoch": 7.658004760742187e-06,
      "grad_norm": 0.2186063826084137,
      "learning_rate": 9.974937295336268e-06,
      "loss": 0.0724,
      "step": 8030
    },
    {
      "epoch": 7.658004760742187e-06,
      "model_forward_time": 1.7138941287994385,
      "step": 8030
    },
    {
      "dynamics_loss_avg": 0.06268302462995053,
      "epoch": 7.658004760742187e-06,
      "step": 8030
    },
    {
      "action_loss_avg": 0.005446235882118344,
      "epoch": 7.658004760742187e-06,
      "step": 8030
    },
    {
      "epoch": 7.658004760742187e-06,
      "step": 8030,
      "training_step_time": 8.179916620254517
    },
    {
      "epoch": 7.658958435058594e-06,
      "model_forward_time": 1.7342476844787598,
      "step": 8031
    },
    {
      "epoch": 7.658958435058594e-06,
      "step": 8031,
      "training_step_time": 7.807093381881714
    },
    {
      "epoch": 7.659912109375e-06,
      "model_forward_time": 1.7254912853240967,
      "step": 8032
    },
    {
      "epoch": 7.659912109375e-06,
      "step": 8032,
      "training_step_time": 7.490330934524536
    },
    {
      "epoch": 7.660865783691406e-06,
      "model_forward_time": 1.7282071113586426,
      "step": 8033
    },
    {
      "epoch": 7.660865783691406e-06,
      "step": 8033,
      "training_step_time": 7.310070514678955
    },
    {
      "epoch": 7.661819458007813e-06,
      "model_forward_time": 1.7384483814239502,
      "step": 8034
    },
    {
      "epoch": 7.661819458007813e-06,
      "step": 8034,
      "training_step_time": 7.10572624206543
    },
    {
      "epoch": 7.662773132324219e-06,
      "model_forward_time": 1.7111306190490723,
      "step": 8035
    },
    {
      "epoch": 7.662773132324219e-06,
      "step": 8035,
      "training_step_time": 7.457936763763428
    },
    {
      "epoch": 7.663726806640625e-06,
      "model_forward_time": 1.7309997081756592,
      "step": 8036
    },
    {
      "epoch": 7.663726806640625e-06,
      "step": 8036,
      "training_step_time": 7.690471649169922
    },
    {
      "epoch": 7.664680480957032e-06,
      "model_forward_time": 2.25974440574646,
      "step": 8037
    },
    {
      "epoch": 7.664680480957032e-06,
      "step": 8037,
      "training_step_time": 8.248977422714233
    },
    {
      "epoch": 7.665634155273438e-06,
      "model_forward_time": 1.778029441833496,
      "step": 8038
    },
    {
      "epoch": 7.665634155273438e-06,
      "step": 8038,
      "training_step_time": 7.689236164093018
    },
    {
      "epoch": 7.666587829589843e-06,
      "model_forward_time": 1.715547800064087,
      "step": 8039
    },
    {
      "epoch": 7.666587829589843e-06,
      "step": 8039,
      "training_step_time": 8.384467840194702
    },
    {
      "epoch": 7.66754150390625e-06,
      "grad_norm": 0.24920423328876495,
      "learning_rate": 9.974771676692975e-06,
      "loss": 0.0752,
      "step": 8040
    },
    {
      "epoch": 7.66754150390625e-06,
      "model_forward_time": 1.7146120071411133,
      "step": 8040
    },
    {
      "dynamics_loss_avg": 0.05319725833833218,
      "epoch": 7.66754150390625e-06,
      "step": 8040
    },
    {
      "action_loss_avg": 0.009511616174131632,
      "epoch": 7.66754150390625e-06,
      "step": 8040
    },
    {
      "epoch": 7.66754150390625e-06,
      "step": 8040,
      "training_step_time": 7.808537721633911
    },
    {
      "epoch": 7.668495178222657e-06,
      "model_forward_time": 1.711876392364502,
      "step": 8041
    },
    {
      "epoch": 7.668495178222657e-06,
      "step": 8041,
      "training_step_time": 7.538143873214722
    },
    {
      "epoch": 7.669448852539062e-06,
      "model_forward_time": 1.7292625904083252,
      "step": 8042
    },
    {
      "epoch": 7.669448852539062e-06,
      "step": 8042,
      "training_step_time": 7.357098340988159
    },
    {
      "epoch": 7.670402526855468e-06,
      "model_forward_time": 1.72696852684021,
      "step": 8043
    },
    {
      "epoch": 7.670402526855468e-06,
      "step": 8043,
      "training_step_time": 7.141441106796265
    },
    {
      "epoch": 7.671356201171875e-06,
      "model_forward_time": 1.7225704193115234,
      "step": 8044
    },
    {
      "epoch": 7.671356201171875e-06,
      "step": 8044,
      "training_step_time": 7.2000298500061035
    },
    {
      "epoch": 7.672309875488281e-06,
      "model_forward_time": 1.720890760421753,
      "step": 8045
    },
    {
      "epoch": 7.672309875488281e-06,
      "step": 8045,
      "training_step_time": 7.753607511520386
    },
    {
      "epoch": 7.673263549804687e-06,
      "model_forward_time": 2.274052858352661,
      "step": 8046
    },
    {
      "epoch": 7.673263549804687e-06,
      "step": 8046,
      "training_step_time": 8.571757555007935
    },
    {
      "epoch": 7.674217224121094e-06,
      "model_forward_time": 1.7830042839050293,
      "step": 8047
    },
    {
      "epoch": 7.674217224121094e-06,
      "step": 8047,
      "training_step_time": 7.656555652618408
    },
    {
      "epoch": 7.6751708984375e-06,
      "model_forward_time": 1.7235894203186035,
      "step": 8048
    },
    {
      "epoch": 7.6751708984375e-06,
      "step": 8048,
      "training_step_time": 8.265878200531006
    },
    {
      "epoch": 7.676124572753906e-06,
      "model_forward_time": 1.7075679302215576,
      "step": 8049
    },
    {
      "epoch": 7.676124572753906e-06,
      "step": 8049,
      "training_step_time": 8.650350332260132
    },
    {
      "epoch": 7.677078247070313e-06,
      "grad_norm": 0.3000238835811615,
      "learning_rate": 9.974605514016131e-06,
      "loss": 0.0796,
      "step": 8050
    },
    {
      "epoch": 7.677078247070313e-06,
      "model_forward_time": 1.7100775241851807,
      "step": 8050
    },
    {
      "dynamics_loss_avg": 0.06307216342538595,
      "epoch": 7.677078247070313e-06,
      "step": 8050
    },
    {
      "action_loss_avg": 0.004596764501184225,
      "epoch": 7.677078247070313e-06,
      "step": 8050
    },
    {
      "epoch": 7.677078247070313e-06,
      "step": 8050,
      "training_step_time": 7.514120817184448
    },
    {
      "epoch": 7.678031921386719e-06,
      "model_forward_time": 1.7315032482147217,
      "step": 8051
    },
    {
      "epoch": 7.678031921386719e-06,
      "step": 8051,
      "training_step_time": 7.538655996322632
    },
    {
      "epoch": 7.678985595703125e-06,
      "model_forward_time": 1.734226942062378,
      "step": 8052
    },
    {
      "epoch": 7.678985595703125e-06,
      "step": 8052,
      "training_step_time": 7.173367023468018
    },
    {
      "epoch": 7.679939270019532e-06,
      "model_forward_time": 1.7068192958831787,
      "step": 8053
    },
    {
      "epoch": 7.679939270019532e-06,
      "step": 8053,
      "training_step_time": 7.005494117736816
    },
    {
      "epoch": 7.680892944335938e-06,
      "model_forward_time": 1.7475988864898682,
      "step": 8054
    },
    {
      "epoch": 7.680892944335938e-06,
      "step": 8054,
      "training_step_time": 7.739569902420044
    },
    {
      "epoch": 7.681846618652343e-06,
      "model_forward_time": 2.229525327682495,
      "step": 8055
    },
    {
      "epoch": 7.681846618652343e-06,
      "step": 8055,
      "training_step_time": 8.453471183776855
    },
    {
      "epoch": 7.68280029296875e-06,
      "model_forward_time": 1.717989206314087,
      "step": 8056
    },
    {
      "epoch": 7.68280029296875e-06,
      "step": 8056,
      "training_step_time": 8.275558233261108
    },
    {
      "epoch": 7.683753967285157e-06,
      "model_forward_time": 1.723294734954834,
      "step": 8057
    },
    {
      "epoch": 7.683753967285157e-06,
      "step": 8057,
      "training_step_time": 8.626020193099976
    },
    {
      "epoch": 7.684707641601562e-06,
      "model_forward_time": 1.7178232669830322,
      "step": 8058
    },
    {
      "epoch": 7.684707641601562e-06,
      "step": 8058,
      "training_step_time": 8.770628690719604
    },
    {
      "epoch": 7.685661315917968e-06,
      "model_forward_time": 1.6867356300354004,
      "step": 8059
    },
    {
      "epoch": 7.685661315917968e-06,
      "step": 8059,
      "training_step_time": 8.449203491210938
    },
    {
      "epoch": 7.686614990234375e-06,
      "grad_norm": 0.2748921513557434,
      "learning_rate": 9.974438807323907e-06,
      "loss": 0.0775,
      "step": 8060
    },
    {
      "epoch": 7.686614990234375e-06,
      "model_forward_time": 1.7175467014312744,
      "step": 8060
    },
    {
      "dynamics_loss_avg": 0.07343250457197428,
      "epoch": 7.686614990234375e-06,
      "step": 8060
    },
    {
      "action_loss_avg": 0.007568817003630102,
      "epoch": 7.686614990234375e-06,
      "step": 8060
    },
    {
      "epoch": 7.686614990234375e-06,
      "step": 8060,
      "training_step_time": 7.548013687133789
    },
    {
      "epoch": 7.687568664550781e-06,
      "model_forward_time": 1.7036240100860596,
      "step": 8061
    },
    {
      "epoch": 7.687568664550781e-06,
      "step": 8061,
      "training_step_time": 7.17138147354126
    },
    {
      "epoch": 7.688522338867187e-06,
      "model_forward_time": 1.7216317653656006,
      "step": 8062
    },
    {
      "epoch": 7.688522338867187e-06,
      "step": 8062,
      "training_step_time": 7.12070894241333
    },
    {
      "epoch": 7.689476013183594e-06,
      "model_forward_time": 1.696774959564209,
      "step": 8063
    },
    {
      "epoch": 7.689476013183594e-06,
      "step": 8063,
      "training_step_time": 7.6718480587005615
    },
    {
      "epoch": 7.6904296875e-06,
      "model_forward_time": 1.7065613269805908,
      "step": 8064
    },
    {
      "epoch": 7.6904296875e-06,
      "step": 8064,
      "training_step_time": 8.753939151763916
    },
    {
      "epoch": 7.691383361816406e-06,
      "model_forward_time": 1.706251621246338,
      "step": 8065
    },
    {
      "epoch": 7.691383361816406e-06,
      "step": 8065,
      "training_step_time": 8.285815715789795
    },
    {
      "epoch": 7.692337036132813e-06,
      "model_forward_time": 1.7615787982940674,
      "step": 8066
    },
    {
      "epoch": 7.692337036132813e-06,
      "step": 8066,
      "training_step_time": 8.625020503997803
    },
    {
      "epoch": 7.693290710449219e-06,
      "model_forward_time": 1.7203245162963867,
      "step": 8067
    },
    {
      "epoch": 7.693290710449219e-06,
      "step": 8067,
      "training_step_time": 9.434889554977417
    },
    {
      "epoch": 7.694244384765625e-06,
      "model_forward_time": 1.7013585567474365,
      "step": 8068
    },
    {
      "epoch": 7.694244384765625e-06,
      "step": 8068,
      "training_step_time": 9.015910863876343
    },
    {
      "epoch": 7.695198059082032e-06,
      "model_forward_time": 1.729264736175537,
      "step": 8069
    },
    {
      "epoch": 7.695198059082032e-06,
      "step": 8069,
      "training_step_time": 7.550218105316162
    },
    {
      "epoch": 7.696151733398438e-06,
      "grad_norm": 0.25395530462265015,
      "learning_rate": 9.974271556634535e-06,
      "loss": 0.0724,
      "step": 8070
    },
    {
      "epoch": 7.696151733398438e-06,
      "model_forward_time": 1.7048594951629639,
      "step": 8070
    },
    {
      "dynamics_loss_avg": 0.06365514174103737,
      "epoch": 7.696151733398438e-06,
      "step": 8070
    },
    {
      "action_loss_avg": 0.00518065148498863,
      "epoch": 7.696151733398438e-06,
      "step": 8070
    },
    {
      "epoch": 7.696151733398438e-06,
      "step": 8070,
      "training_step_time": 7.3892621994018555
    },
    {
      "epoch": 7.697105407714843e-06,
      "model_forward_time": 1.7134997844696045,
      "step": 8071
    },
    {
      "epoch": 7.697105407714843e-06,
      "step": 8071,
      "training_step_time": 7.14176869392395
    },
    {
      "epoch": 7.69805908203125e-06,
      "model_forward_time": 1.705202341079712,
      "step": 8072
    },
    {
      "epoch": 7.69805908203125e-06,
      "step": 8072,
      "training_step_time": 7.501292943954468
    },
    {
      "epoch": 7.699012756347657e-06,
      "model_forward_time": 1.7235298156738281,
      "step": 8073
    },
    {
      "epoch": 7.699012756347657e-06,
      "step": 8073,
      "training_step_time": 7.938527822494507
    },
    {
      "epoch": 7.699966430664062e-06,
      "model_forward_time": 2.230402946472168,
      "step": 8074
    },
    {
      "epoch": 7.699966430664062e-06,
      "step": 8074,
      "training_step_time": 8.716726779937744
    },
    {
      "epoch": 7.700920104980468e-06,
      "model_forward_time": 1.7632830142974854,
      "step": 8075
    },
    {
      "epoch": 7.700920104980468e-06,
      "step": 8075,
      "training_step_time": 8.525871276855469
    },
    {
      "epoch": 7.701873779296875e-06,
      "model_forward_time": 1.730700969696045,
      "step": 8076
    },
    {
      "epoch": 7.701873779296875e-06,
      "step": 8076,
      "training_step_time": 7.793477296829224
    },
    {
      "epoch": 7.702827453613281e-06,
      "model_forward_time": 1.7060554027557373,
      "step": 8077
    },
    {
      "epoch": 7.702827453613281e-06,
      "step": 8077,
      "training_step_time": 8.00821852684021
    },
    {
      "epoch": 7.703781127929687e-06,
      "model_forward_time": 1.718682050704956,
      "step": 8078
    },
    {
      "epoch": 7.703781127929687e-06,
      "step": 8078,
      "training_step_time": 8.057844877243042
    },
    {
      "epoch": 7.704734802246094e-06,
      "model_forward_time": 1.7135446071624756,
      "step": 8079
    },
    {
      "epoch": 7.704734802246094e-06,
      "step": 8079,
      "training_step_time": 7.47817850112915
    },
    {
      "epoch": 7.7056884765625e-06,
      "grad_norm": 0.3715246319770813,
      "learning_rate": 9.974103761966302e-06,
      "loss": 0.0688,
      "step": 8080
    },
    {
      "epoch": 7.7056884765625e-06,
      "model_forward_time": 1.7503025531768799,
      "step": 8080
    },
    {
      "dynamics_loss_avg": 0.0678376503288746,
      "epoch": 7.7056884765625e-06,
      "step": 8080
    },
    {
      "action_loss_avg": 0.005648463219404221,
      "epoch": 7.7056884765625e-06,
      "step": 8080
    },
    {
      "epoch": 7.7056884765625e-06,
      "step": 8080,
      "training_step_time": 7.139988899230957
    },
    {
      "epoch": 7.706642150878906e-06,
      "model_forward_time": 1.7242555618286133,
      "step": 8081
    },
    {
      "epoch": 7.706642150878906e-06,
      "step": 8081,
      "training_step_time": 7.468362331390381
    },
    {
      "epoch": 7.707595825195313e-06,
      "model_forward_time": 1.7280752658843994,
      "step": 8082
    },
    {
      "epoch": 7.707595825195313e-06,
      "step": 8082,
      "training_step_time": 8.723040342330933
    },
    {
      "epoch": 7.708549499511719e-06,
      "model_forward_time": 2.2443366050720215,
      "step": 8083
    },
    {
      "epoch": 7.708549499511719e-06,
      "step": 8083,
      "training_step_time": 8.64916467666626
    },
    {
      "epoch": 7.709503173828125e-06,
      "model_forward_time": 1.709326982498169,
      "step": 8084
    },
    {
      "epoch": 7.709503173828125e-06,
      "step": 8084,
      "training_step_time": 8.657632827758789
    },
    {
      "epoch": 7.710456848144532e-06,
      "model_forward_time": 1.3404483795166016,
      "step": 8085
    },
    {
      "epoch": 7.710456848144532e-06,
      "step": 8085,
      "training_step_time": 7.6591362953186035
    },
    {
      "epoch": 7.711410522460938e-06,
      "model_forward_time": 1.7135183811187744,
      "step": 8086
    },
    {
      "epoch": 7.711410522460938e-06,
      "step": 8086,
      "training_step_time": 7.790148973464966
    },
    {
      "epoch": 7.712364196777344e-06,
      "model_forward_time": 1.7128262519836426,
      "step": 8087
    },
    {
      "epoch": 7.712364196777344e-06,
      "step": 8087,
      "training_step_time": 8.63683795928955
    },
    {
      "epoch": 7.71331787109375e-06,
      "model_forward_time": 1.7015905380249023,
      "step": 8088
    },
    {
      "epoch": 7.71331787109375e-06,
      "step": 8088,
      "training_step_time": 8.210069417953491
    },
    {
      "epoch": 7.714271545410157e-06,
      "model_forward_time": 1.7278361320495605,
      "step": 8089
    },
    {
      "epoch": 7.714271545410157e-06,
      "step": 8089,
      "training_step_time": 8.04112434387207
    },
    {
      "epoch": 7.715225219726562e-06,
      "grad_norm": 0.2817200720310211,
      "learning_rate": 9.973935423337564e-06,
      "loss": 0.073,
      "step": 8090
    },
    {
      "epoch": 7.715225219726562e-06,
      "model_forward_time": 1.7084190845489502,
      "step": 8090
    },
    {
      "dynamics_loss_avg": 0.06995413694530725,
      "epoch": 7.715225219726562e-06,
      "step": 8090
    },
    {
      "action_loss_avg": 0.005792877043131739,
      "epoch": 7.715225219726562e-06,
      "step": 8090
    },
    {
      "epoch": 7.715225219726562e-06,
      "step": 8090,
      "training_step_time": 7.259745359420776
    },
    {
      "epoch": 7.716178894042968e-06,
      "model_forward_time": 1.7055273056030273,
      "step": 8091
    },
    {
      "epoch": 7.716178894042968e-06,
      "step": 8091,
      "training_step_time": 8.705626726150513
    },
    {
      "epoch": 7.717132568359376e-06,
      "model_forward_time": 2.223188877105713,
      "step": 8092
    },
    {
      "epoch": 7.717132568359376e-06,
      "step": 8092,
      "training_step_time": 9.305490255355835
    },
    {
      "epoch": 7.718086242675781e-06,
      "model_forward_time": 1.7206416130065918,
      "step": 8093
    },
    {
      "epoch": 7.718086242675781e-06,
      "step": 8093,
      "training_step_time": 7.607465505599976
    },
    {
      "epoch": 7.719039916992187e-06,
      "model_forward_time": 1.7034451961517334,
      "step": 8094
    },
    {
      "epoch": 7.719039916992187e-06,
      "step": 8094,
      "training_step_time": 8.391465425491333
    },
    {
      "epoch": 7.719993591308594e-06,
      "model_forward_time": 1.7190275192260742,
      "step": 8095
    },
    {
      "epoch": 7.719993591308594e-06,
      "step": 8095,
      "training_step_time": 7.88471245765686
    },
    {
      "epoch": 7.720947265625e-06,
      "model_forward_time": 1.343158483505249,
      "step": 8096
    },
    {
      "epoch": 7.720947265625e-06,
      "step": 8096,
      "training_step_time": 8.92652177810669
    },
    {
      "epoch": 7.721900939941406e-06,
      "model_forward_time": 1.719179630279541,
      "step": 8097
    },
    {
      "epoch": 7.721900939941406e-06,
      "step": 8097,
      "training_step_time": 7.608023643493652
    },
    {
      "epoch": 7.722854614257813e-06,
      "model_forward_time": 1.7240524291992188,
      "step": 8098
    },
    {
      "epoch": 7.722854614257813e-06,
      "step": 8098,
      "training_step_time": 7.420523405075073
    },
    {
      "epoch": 7.723808288574219e-06,
      "model_forward_time": 1.7014167308807373,
      "step": 8099
    },
    {
      "epoch": 7.723808288574219e-06,
      "step": 8099,
      "training_step_time": 7.340675592422485
    },
    {
      "epoch": 7.724761962890625e-06,
      "grad_norm": 0.32700788974761963,
      "learning_rate": 9.973766540766723e-06,
      "loss": 0.0701,
      "step": 8100
    },
    {
      "epoch": 7.724761962890625e-06,
      "model_forward_time": 1.708251953125,
      "step": 8100
    },
    {
      "dynamics_loss_avg": 0.0573100846260786,
      "epoch": 7.724761962890625e-06,
      "step": 8100
    },
    {
      "action_loss_avg": 0.005420524231158197,
      "epoch": 7.724761962890625e-06,
      "step": 8100
    },
    {
      "epoch": 7.724761962890625e-06,
      "step": 8100,
      "training_step_time": 7.824541091918945
    },
    {
      "epoch": 7.725715637207032e-06,
      "model_forward_time": 1.875509262084961,
      "step": 8101
    },
    {
      "epoch": 7.725715637207032e-06,
      "step": 8101,
      "training_step_time": 7.725611925125122
    },
    {
      "epoch": 7.726669311523438e-06,
      "model_forward_time": 1.7176411151885986,
      "step": 8102
    },
    {
      "epoch": 7.726669311523438e-06,
      "step": 8102,
      "training_step_time": 7.8635573387146
    },
    {
      "epoch": 7.727622985839844e-06,
      "model_forward_time": 1.7586004734039307,
      "step": 8103
    },
    {
      "epoch": 7.727622985839844e-06,
      "step": 8103,
      "training_step_time": 7.7479002475738525
    },
    {
      "epoch": 7.72857666015625e-06,
      "model_forward_time": 1.7222404479980469,
      "step": 8104
    },
    {
      "epoch": 7.72857666015625e-06,
      "step": 8104,
      "training_step_time": 8.678952693939209
    },
    {
      "epoch": 7.729530334472657e-06,
      "model_forward_time": 1.716092586517334,
      "step": 8105
    },
    {
      "epoch": 7.729530334472657e-06,
      "step": 8105,
      "training_step_time": 7.760488271713257
    },
    {
      "epoch": 7.730484008789062e-06,
      "model_forward_time": 1.7072784900665283,
      "step": 8106
    },
    {
      "epoch": 7.730484008789062e-06,
      "step": 8106,
      "training_step_time": 8.903947591781616
    },
    {
      "epoch": 7.731437683105468e-06,
      "model_forward_time": 1.7104933261871338,
      "step": 8107
    },
    {
      "epoch": 7.731437683105468e-06,
      "step": 8107,
      "training_step_time": 7.484789133071899
    },
    {
      "epoch": 7.732391357421876e-06,
      "model_forward_time": 1.7175052165985107,
      "step": 8108
    },
    {
      "epoch": 7.732391357421876e-06,
      "step": 8108,
      "training_step_time": 7.337002277374268
    },
    {
      "epoch": 7.733345031738281e-06,
      "model_forward_time": 1.7133572101593018,
      "step": 8109
    },
    {
      "epoch": 7.733345031738281e-06,
      "step": 8109,
      "training_step_time": 7.787728786468506
    },
    {
      "epoch": 7.734298706054687e-06,
      "grad_norm": 0.3067004680633545,
      "learning_rate": 9.973597114272252e-06,
      "loss": 0.0738,
      "step": 8110
    },
    {
      "epoch": 7.734298706054687e-06,
      "model_forward_time": 1.7013771533966064,
      "step": 8110
    },
    {
      "dynamics_loss_avg": 0.04567953525111079,
      "epoch": 7.734298706054687e-06,
      "step": 8110
    },
    {
      "action_loss_avg": 0.0044901560177095234,
      "epoch": 7.734298706054687e-06,
      "step": 8110
    },
    {
      "epoch": 7.734298706054687e-06,
      "step": 8110,
      "training_step_time": 8.88509202003479
    },
    {
      "epoch": 7.735252380371094e-06,
      "model_forward_time": 1.714087963104248,
      "step": 8111
    },
    {
      "epoch": 7.735252380371094e-06,
      "step": 8111,
      "training_step_time": 7.8264546394348145
    },
    {
      "epoch": 7.7362060546875e-06,
      "model_forward_time": 1.7493104934692383,
      "step": 8112
    },
    {
      "epoch": 7.7362060546875e-06,
      "step": 8112,
      "training_step_time": 7.7144763469696045
    },
    {
      "epoch": 7.737159729003906e-06,
      "model_forward_time": 1.711205005645752,
      "step": 8113
    },
    {
      "epoch": 7.737159729003906e-06,
      "step": 8113,
      "training_step_time": 8.344445943832397
    },
    {
      "epoch": 7.738113403320313e-06,
      "model_forward_time": 1.7142877578735352,
      "step": 8114
    },
    {
      "epoch": 7.738113403320313e-06,
      "step": 8114,
      "training_step_time": 7.761941432952881
    },
    {
      "epoch": 7.739067077636719e-06,
      "model_forward_time": 1.7165701389312744,
      "step": 8115
    },
    {
      "epoch": 7.739067077636719e-06,
      "step": 8115,
      "training_step_time": 8.204010248184204
    },
    {
      "epoch": 7.740020751953125e-06,
      "model_forward_time": 1.7244718074798584,
      "step": 8116
    },
    {
      "epoch": 7.740020751953125e-06,
      "step": 8116,
      "training_step_time": 7.5153138637542725
    },
    {
      "epoch": 7.740974426269532e-06,
      "model_forward_time": 1.7183680534362793,
      "step": 8117
    },
    {
      "epoch": 7.740974426269532e-06,
      "step": 8117,
      "training_step_time": 7.379500389099121
    },
    {
      "epoch": 7.741928100585938e-06,
      "model_forward_time": 1.7246310710906982,
      "step": 8118
    },
    {
      "epoch": 7.741928100585938e-06,
      "step": 8118,
      "training_step_time": 9.638574123382568
    },
    {
      "epoch": 7.742881774902344e-06,
      "model_forward_time": 1.7118873596191406,
      "step": 8119
    },
    {
      "epoch": 7.742881774902344e-06,
      "step": 8119,
      "training_step_time": 7.97368860244751
    },
    {
      "epoch": 7.74383544921875e-06,
      "grad_norm": 0.2997503876686096,
      "learning_rate": 9.973427143872677e-06,
      "loss": 0.0757,
      "step": 8120
    },
    {
      "epoch": 7.74383544921875e-06,
      "model_forward_time": 2.24224853515625,
      "step": 8120
    },
    {
      "dynamics_loss_avg": 0.06056882794946432,
      "epoch": 7.74383544921875e-06,
      "step": 8120
    },
    {
      "action_loss_avg": 0.00473414504667744,
      "epoch": 7.74383544921875e-06,
      "step": 8120
    },
    {
      "epoch": 7.74383544921875e-06,
      "step": 8120,
      "training_step_time": 8.784733533859253
    },
    {
      "epoch": 7.744789123535157e-06,
      "model_forward_time": 1.7623569965362549,
      "step": 8121
    },
    {
      "epoch": 7.744789123535157e-06,
      "step": 8121,
      "training_step_time": 8.540621757507324
    },
    {
      "epoch": 7.745742797851562e-06,
      "model_forward_time": 1.7252700328826904,
      "step": 8122
    },
    {
      "epoch": 7.745742797851562e-06,
      "step": 8122,
      "training_step_time": 8.507934093475342
    },
    {
      "epoch": 7.746696472167968e-06,
      "model_forward_time": 1.7228171825408936,
      "step": 8123
    },
    {
      "epoch": 7.746696472167968e-06,
      "step": 8123,
      "training_step_time": 7.912690162658691
    },
    {
      "epoch": 7.747650146484376e-06,
      "model_forward_time": 1.7434887886047363,
      "step": 8124
    },
    {
      "epoch": 7.747650146484376e-06,
      "step": 8124,
      "training_step_time": 7.692962408065796
    },
    {
      "epoch": 7.748603820800781e-06,
      "model_forward_time": 1.7227377891540527,
      "step": 8125
    },
    {
      "epoch": 7.748603820800781e-06,
      "step": 8125,
      "training_step_time": 7.483148813247681
    },
    {
      "epoch": 7.749557495117187e-06,
      "model_forward_time": 1.721198558807373,
      "step": 8126
    },
    {
      "epoch": 7.749557495117187e-06,
      "step": 8126,
      "training_step_time": 7.3733906745910645
    },
    {
      "epoch": 7.750511169433594e-06,
      "model_forward_time": 1.7212982177734375,
      "step": 8127
    },
    {
      "epoch": 7.750511169433594e-06,
      "step": 8127,
      "training_step_time": 7.27616810798645
    },
    {
      "epoch": 7.75146484375e-06,
      "model_forward_time": 1.71455979347229,
      "step": 8128
    },
    {
      "epoch": 7.75146484375e-06,
      "step": 8128,
      "training_step_time": 8.667162895202637
    },
    {
      "epoch": 7.752418518066406e-06,
      "model_forward_time": 2.2400472164154053,
      "step": 8129
    },
    {
      "epoch": 7.752418518066406e-06,
      "step": 8129,
      "training_step_time": 7.856398820877075
    },
    {
      "epoch": 7.753372192382813e-06,
      "grad_norm": 0.30860796570777893,
      "learning_rate": 9.973256629586589e-06,
      "loss": 0.0728,
      "step": 8130
    },
    {
      "epoch": 7.753372192382813e-06,
      "model_forward_time": 1.7602922916412354,
      "step": 8130
    },
    {
      "dynamics_loss_avg": 0.06980712972581386,
      "epoch": 7.753372192382813e-06,
      "step": 8130
    },
    {
      "action_loss_avg": 0.00530275721102953,
      "epoch": 7.753372192382813e-06,
      "step": 8130
    },
    {
      "epoch": 7.753372192382813e-06,
      "step": 8130,
      "training_step_time": 8.397989988327026
    },
    {
      "epoch": 7.754325866699219e-06,
      "model_forward_time": 1.724886178970337,
      "step": 8131
    },
    {
      "epoch": 7.754325866699219e-06,
      "step": 8131,
      "training_step_time": 8.855902433395386
    },
    {
      "epoch": 7.755279541015625e-06,
      "model_forward_time": 1.7243576049804688,
      "step": 8132
    },
    {
      "epoch": 7.755279541015625e-06,
      "step": 8132,
      "training_step_time": 8.910212516784668
    },
    {
      "epoch": 7.75623321533203e-06,
      "model_forward_time": 1.7269175052642822,
      "step": 8133
    },
    {
      "epoch": 7.75623321533203e-06,
      "step": 8133,
      "training_step_time": 7.6943113803863525
    },
    {
      "epoch": 7.757186889648438e-06,
      "model_forward_time": 1.7309763431549072,
      "step": 8134
    },
    {
      "epoch": 7.757186889648438e-06,
      "step": 8134,
      "training_step_time": 8.516313076019287
    },
    {
      "epoch": 7.758140563964844e-06,
      "model_forward_time": 1.7285234928131104,
      "step": 8135
    },
    {
      "epoch": 7.758140563964844e-06,
      "step": 8135,
      "training_step_time": 7.3536293506622314
    },
    {
      "epoch": 7.75909423828125e-06,
      "model_forward_time": 1.7274222373962402,
      "step": 8136
    },
    {
      "epoch": 7.75909423828125e-06,
      "step": 8136,
      "training_step_time": 7.282909393310547
    },
    {
      "epoch": 7.760047912597657e-06,
      "model_forward_time": 1.7207486629486084,
      "step": 8137
    },
    {
      "epoch": 7.760047912597657e-06,
      "step": 8137,
      "training_step_time": 8.905319452285767
    },
    {
      "epoch": 7.761001586914062e-06,
      "model_forward_time": 2.2646749019622803,
      "step": 8138
    },
    {
      "epoch": 7.761001586914062e-06,
      "step": 8138,
      "training_step_time": 8.352516889572144
    },
    {
      "epoch": 7.761955261230468e-06,
      "model_forward_time": 1.7250702381134033,
      "step": 8139
    },
    {
      "epoch": 7.761955261230468e-06,
      "step": 8139,
      "training_step_time": 8.704062461853027
    },
    {
      "epoch": 7.762908935546876e-06,
      "grad_norm": 0.24235112965106964,
      "learning_rate": 9.973085571432633e-06,
      "loss": 0.0693,
      "step": 8140
    },
    {
      "epoch": 7.762908935546876e-06,
      "model_forward_time": 1.7172143459320068,
      "step": 8140
    },
    {
      "dynamics_loss_avg": 0.07826349884271622,
      "epoch": 7.762908935546876e-06,
      "step": 8140
    },
    {
      "action_loss_avg": 0.0071454247226938605,
      "epoch": 7.762908935546876e-06,
      "step": 8140
    },
    {
      "epoch": 7.762908935546876e-06,
      "step": 8140,
      "training_step_time": 7.739871263504028
    },
    {
      "epoch": 7.763862609863281e-06,
      "model_forward_time": 1.717334508895874,
      "step": 8141
    },
    {
      "epoch": 7.763862609863281e-06,
      "step": 8141,
      "training_step_time": 7.837003231048584
    },
    {
      "epoch": 7.764816284179687e-06,
      "model_forward_time": 1.7275729179382324,
      "step": 8142
    },
    {
      "epoch": 7.764816284179687e-06,
      "step": 8142,
      "training_step_time": 8.733153581619263
    },
    {
      "epoch": 7.765769958496094e-06,
      "model_forward_time": 1.7161865234375,
      "step": 8143
    },
    {
      "epoch": 7.765769958496094e-06,
      "step": 8143,
      "training_step_time": 7.721751928329468
    },
    {
      "epoch": 7.7667236328125e-06,
      "model_forward_time": 1.7216851711273193,
      "step": 8144
    },
    {
      "epoch": 7.7667236328125e-06,
      "step": 8144,
      "training_step_time": 7.527289390563965
    },
    {
      "epoch": 7.767677307128906e-06,
      "model_forward_time": 1.7235829830169678,
      "step": 8145
    },
    {
      "epoch": 7.767677307128906e-06,
      "step": 8145,
      "training_step_time": 7.291939735412598
    },
    {
      "epoch": 7.768630981445313e-06,
      "model_forward_time": 1.7191338539123535,
      "step": 8146
    },
    {
      "epoch": 7.768630981445313e-06,
      "step": 8146,
      "training_step_time": 7.810557126998901
    },
    {
      "epoch": 7.769584655761719e-06,
      "model_forward_time": 2.2440199851989746,
      "step": 8147
    },
    {
      "epoch": 7.769584655761719e-06,
      "step": 8147,
      "training_step_time": 7.691709995269775
    },
    {
      "epoch": 7.770538330078125e-06,
      "model_forward_time": 1.7158634662628174,
      "step": 8148
    },
    {
      "epoch": 7.770538330078125e-06,
      "step": 8148,
      "training_step_time": 8.553236961364746
    },
    {
      "epoch": 7.77149200439453e-06,
      "model_forward_time": 1.7562546730041504,
      "step": 8149
    },
    {
      "epoch": 7.77149200439453e-06,
      "step": 8149,
      "training_step_time": 7.722292900085449
    },
    {
      "epoch": 7.772445678710938e-06,
      "grad_norm": 0.26621055603027344,
      "learning_rate": 9.972913969429514e-06,
      "loss": 0.0719,
      "step": 8150
    },
    {
      "epoch": 7.772445678710938e-06,
      "model_forward_time": 1.7205066680908203,
      "step": 8150
    },
    {
      "dynamics_loss_avg": 0.06429771445691586,
      "epoch": 7.772445678710938e-06,
      "step": 8150
    },
    {
      "action_loss_avg": 0.007006499799899757,
      "epoch": 7.772445678710938e-06,
      "step": 8150
    },
    {
      "epoch": 7.772445678710938e-06,
      "step": 8150,
      "training_step_time": 8.403663873672485
    },
    {
      "epoch": 7.773399353027344e-06,
      "model_forward_time": 1.7176470756530762,
      "step": 8151
    },
    {
      "epoch": 7.773399353027344e-06,
      "step": 8151,
      "training_step_time": 8.530000448226929
    },
    {
      "epoch": 7.77435302734375e-06,
      "model_forward_time": 1.7211525440216064,
      "step": 8152
    },
    {
      "epoch": 7.77435302734375e-06,
      "step": 8152,
      "training_step_time": 7.6810243129730225
    },
    {
      "epoch": 7.775306701660157e-06,
      "model_forward_time": 1.6982598304748535,
      "step": 8153
    },
    {
      "epoch": 7.775306701660157e-06,
      "step": 8153,
      "training_step_time": 7.4883198738098145
    },
    {
      "epoch": 7.776260375976562e-06,
      "model_forward_time": 1.7213449478149414,
      "step": 8154
    },
    {
      "epoch": 7.776260375976562e-06,
      "step": 8154,
      "training_step_time": 7.102304935455322
    },
    {
      "epoch": 7.777214050292968e-06,
      "model_forward_time": 1.7142002582550049,
      "step": 8155
    },
    {
      "epoch": 7.777214050292968e-06,
      "step": 8155,
      "training_step_time": 7.8053812980651855
    },
    {
      "epoch": 7.778167724609376e-06,
      "model_forward_time": 2.2757019996643066,
      "step": 8156
    },
    {
      "epoch": 7.778167724609376e-06,
      "step": 8156,
      "training_step_time": 7.925078868865967
    },
    {
      "epoch": 7.779121398925781e-06,
      "model_forward_time": 1.7103197574615479,
      "step": 8157
    },
    {
      "epoch": 7.779121398925781e-06,
      "step": 8157,
      "training_step_time": 8.861557483673096
    },
    {
      "epoch": 7.780075073242187e-06,
      "model_forward_time": 1.7150306701660156,
      "step": 8158
    },
    {
      "epoch": 7.780075073242187e-06,
      "step": 8158,
      "training_step_time": 8.65088152885437
    },
    {
      "epoch": 7.781028747558594e-06,
      "model_forward_time": 1.6914551258087158,
      "step": 8159
    },
    {
      "epoch": 7.781028747558594e-06,
      "step": 8159,
      "training_step_time": 8.466528415679932
    },
    {
      "epoch": 7.781982421875e-06,
      "grad_norm": 0.448686420917511,
      "learning_rate": 9.972741823596e-06,
      "loss": 0.0767,
      "step": 8160
    },
    {
      "epoch": 7.781982421875e-06,
      "model_forward_time": 1.6954994201660156,
      "step": 8160
    },
    {
      "dynamics_loss_avg": 0.0686110358685255,
      "epoch": 7.781982421875e-06,
      "step": 8160
    },
    {
      "action_loss_avg": 0.005304216593503952,
      "epoch": 7.781982421875e-06,
      "step": 8160
    },
    {
      "epoch": 7.781982421875e-06,
      "step": 8160,
      "training_step_time": 8.783526182174683
    },
    {
      "epoch": 7.782936096191406e-06,
      "model_forward_time": 1.7104997634887695,
      "step": 8161
    },
    {
      "epoch": 7.782936096191406e-06,
      "step": 8161,
      "training_step_time": 7.691070079803467
    },
    {
      "epoch": 7.783889770507813e-06,
      "model_forward_time": 1.3276009559631348,
      "step": 8162
    },
    {
      "epoch": 7.783889770507813e-06,
      "step": 8162,
      "training_step_time": 7.5420403480529785
    },
    {
      "epoch": 7.784843444824219e-06,
      "model_forward_time": 1.6945819854736328,
      "step": 8163
    },
    {
      "epoch": 7.784843444824219e-06,
      "step": 8163,
      "training_step_time": 7.104655504226685
    },
    {
      "epoch": 7.785797119140625e-06,
      "model_forward_time": 1.6904747486114502,
      "step": 8164
    },
    {
      "epoch": 7.785797119140625e-06,
      "step": 8164,
      "training_step_time": 7.835610628128052
    },
    {
      "epoch": 7.78675079345703e-06,
      "model_forward_time": 1.7014100551605225,
      "step": 8165
    },
    {
      "epoch": 7.78675079345703e-06,
      "step": 8165,
      "training_step_time": 8.621415853500366
    },
    {
      "epoch": 7.787704467773438e-06,
      "model_forward_time": 2.23524808883667,
      "step": 8166
    },
    {
      "epoch": 7.787704467773438e-06,
      "step": 8166,
      "training_step_time": 8.893068552017212
    },
    {
      "epoch": 7.788658142089844e-06,
      "model_forward_time": 1.7137668132781982,
      "step": 8167
    },
    {
      "epoch": 7.788658142089844e-06,
      "step": 8167,
      "training_step_time": 8.491511344909668
    },
    {
      "epoch": 7.78961181640625e-06,
      "model_forward_time": 1.6949238777160645,
      "step": 8168
    },
    {
      "epoch": 7.78961181640625e-06,
      "step": 8168,
      "training_step_time": 9.508021116256714
    },
    {
      "epoch": 7.790565490722657e-06,
      "model_forward_time": 1.3246769905090332,
      "step": 8169
    },
    {
      "epoch": 7.790565490722657e-06,
      "step": 8169,
      "training_step_time": 7.750765562057495
    },
    {
      "epoch": 7.791519165039062e-06,
      "grad_norm": 0.3355375826358795,
      "learning_rate": 9.972569133950917e-06,
      "loss": 0.0773,
      "step": 8170
    },
    {
      "epoch": 7.791519165039062e-06,
      "model_forward_time": 1.7138426303863525,
      "step": 8170
    },
    {
      "dynamics_loss_avg": 0.08506378531455994,
      "epoch": 7.791519165039062e-06,
      "step": 8170
    },
    {
      "action_loss_avg": 0.008511516777798534,
      "epoch": 7.791519165039062e-06,
      "step": 8170
    },
    {
      "epoch": 7.791519165039062e-06,
      "step": 8170,
      "training_step_time": 8.646405220031738
    },
    {
      "epoch": 7.792472839355468e-06,
      "model_forward_time": 1.705479621887207,
      "step": 8171
    },
    {
      "epoch": 7.792472839355468e-06,
      "step": 8171,
      "training_step_time": 7.437392950057983
    },
    {
      "epoch": 7.793426513671876e-06,
      "model_forward_time": 1.7052743434906006,
      "step": 8172
    },
    {
      "epoch": 7.793426513671876e-06,
      "step": 8172,
      "training_step_time": 7.479463815689087
    },
    {
      "epoch": 7.794380187988281e-06,
      "model_forward_time": 1.7055485248565674,
      "step": 8173
    },
    {
      "epoch": 7.794380187988281e-06,
      "step": 8173,
      "training_step_time": 7.283241033554077
    },
    {
      "epoch": 7.795333862304687e-06,
      "model_forward_time": 1.705651044845581,
      "step": 8174
    },
    {
      "epoch": 7.795333862304687e-06,
      "step": 8174,
      "training_step_time": 7.844766616821289
    },
    {
      "epoch": 7.796287536621094e-06,
      "model_forward_time": 2.2309908866882324,
      "step": 8175
    },
    {
      "epoch": 7.796287536621094e-06,
      "step": 8175,
      "training_step_time": 7.9036900997161865
    },
    {
      "epoch": 7.7972412109375e-06,
      "model_forward_time": 1.7125730514526367,
      "step": 8176
    },
    {
      "epoch": 7.7972412109375e-06,
      "step": 8176,
      "training_step_time": 8.447341442108154
    },
    {
      "epoch": 7.798194885253906e-06,
      "model_forward_time": 1.3506848812103271,
      "step": 8177
    },
    {
      "epoch": 7.798194885253906e-06,
      "step": 8177,
      "training_step_time": 8.632425546646118
    },
    {
      "epoch": 7.799148559570313e-06,
      "model_forward_time": 1.742272138595581,
      "step": 8178
    },
    {
      "epoch": 7.799148559570313e-06,
      "step": 8178,
      "training_step_time": 8.788922309875488
    },
    {
      "epoch": 7.800102233886719e-06,
      "model_forward_time": 1.730437994003296,
      "step": 8179
    },
    {
      "epoch": 7.800102233886719e-06,
      "step": 8179,
      "training_step_time": 7.629811525344849
    },
    {
      "epoch": 7.801055908203125e-06,
      "grad_norm": 0.27908608317375183,
      "learning_rate": 9.972395900513152e-06,
      "loss": 0.0755,
      "step": 8180
    },
    {
      "epoch": 7.801055908203125e-06,
      "model_forward_time": 1.7322969436645508,
      "step": 8180
    },
    {
      "dynamics_loss_avg": 0.07238938473165035,
      "epoch": 7.801055908203125e-06,
      "step": 8180
    },
    {
      "action_loss_avg": 0.0064491973724216224,
      "epoch": 7.801055908203125e-06,
      "step": 8180
    },
    {
      "epoch": 7.801055908203125e-06,
      "step": 8180,
      "training_step_time": 7.660485506057739
    },
    {
      "epoch": 7.80200958251953e-06,
      "model_forward_time": 1.7190237045288086,
      "step": 8181
    },
    {
      "epoch": 7.80200958251953e-06,
      "step": 8181,
      "training_step_time": 8.405170679092407
    },
    {
      "epoch": 7.802963256835938e-06,
      "model_forward_time": 1.7293260097503662,
      "step": 8182
    },
    {
      "epoch": 7.802963256835938e-06,
      "step": 8182,
      "training_step_time": 7.276198387145996
    },
    {
      "epoch": 7.803916931152344e-06,
      "model_forward_time": 1.705806016921997,
      "step": 8183
    },
    {
      "epoch": 7.803916931152344e-06,
      "step": 8183,
      "training_step_time": 7.816979646682739
    },
    {
      "epoch": 7.80487060546875e-06,
      "model_forward_time": 2.2654941082000732,
      "step": 8184
    },
    {
      "epoch": 7.80487060546875e-06,
      "step": 8184,
      "training_step_time": 7.83932638168335
    },
    {
      "epoch": 7.805824279785157e-06,
      "model_forward_time": 1.7074496746063232,
      "step": 8185
    },
    {
      "epoch": 7.805824279785157e-06,
      "step": 8185,
      "training_step_time": 8.707871437072754
    },
    {
      "epoch": 7.806777954101562e-06,
      "model_forward_time": 1.7727406024932861,
      "step": 8186
    },
    {
      "epoch": 7.806777954101562e-06,
      "step": 8186,
      "training_step_time": 7.83973503112793
    },
    {
      "epoch": 7.807731628417968e-06,
      "model_forward_time": 1.7146615982055664,
      "step": 8187
    },
    {
      "epoch": 7.807731628417968e-06,
      "step": 8187,
      "training_step_time": 8.949241399765015
    },
    {
      "epoch": 7.808685302734376e-06,
      "model_forward_time": 1.7403864860534668,
      "step": 8188
    },
    {
      "epoch": 7.808685302734376e-06,
      "step": 8188,
      "training_step_time": 7.754958391189575
    },
    {
      "epoch": 7.809638977050781e-06,
      "model_forward_time": 1.7086360454559326,
      "step": 8189
    },
    {
      "epoch": 7.809638977050781e-06,
      "step": 8189,
      "training_step_time": 8.452339172363281
    },
    {
      "epoch": 7.810592651367187e-06,
      "grad_norm": 0.31494319438934326,
      "learning_rate": 9.972222123301645e-06,
      "loss": 0.0714,
      "step": 8190
    },
    {
      "epoch": 7.810592651367187e-06,
      "model_forward_time": 1.7072639465332031,
      "step": 8190
    },
    {
      "dynamics_loss_avg": 0.061270919255912305,
      "epoch": 7.810592651367187e-06,
      "step": 8190
    },
    {
      "action_loss_avg": 0.0050229424145072695,
      "epoch": 7.810592651367187e-06,
      "step": 8190
    },
    {
      "epoch": 7.810592651367187e-06,
      "step": 8190,
      "training_step_time": 8.580199241638184
    },
    {
      "epoch": 7.811546325683594e-06,
      "model_forward_time": 1.708847999572754,
      "step": 8191
    },
    {
      "epoch": 7.811546325683594e-06,
      "step": 8191,
      "training_step_time": 7.315033435821533
    },
    {
      "epoch": 7.8125e-06,
      "model_forward_time": 1.7376220226287842,
      "step": 8192
    },
    {
      "epoch": 7.8125e-06,
      "step": 8192,
      "training_step_time": 7.862743377685547
    },
    {
      "epoch": 7.813453674316406e-06,
      "model_forward_time": 2.250349998474121,
      "step": 8193
    },
    {
      "epoch": 7.813453674316406e-06,
      "step": 8193,
      "training_step_time": 8.392800331115723
    },
    {
      "epoch": 7.814407348632813e-06,
      "model_forward_time": 1.7126517295837402,
      "step": 8194
    },
    {
      "epoch": 7.814407348632813e-06,
      "step": 8194,
      "training_step_time": 7.836182594299316
    },
    {
      "epoch": 7.815361022949219e-06,
      "model_forward_time": 1.7307827472686768,
      "step": 8195
    },
    {
      "epoch": 7.815361022949219e-06,
      "step": 8195,
      "training_step_time": 8.78717851638794
    },
    {
      "epoch": 7.816314697265625e-06,
      "model_forward_time": 1.7047104835510254,
      "step": 8196
    },
    {
      "epoch": 7.816314697265625e-06,
      "step": 8196,
      "training_step_time": 7.8492138385772705
    },
    {
      "epoch": 7.81726837158203e-06,
      "model_forward_time": 1.7009103298187256,
      "step": 8197
    },
    {
      "epoch": 7.81726837158203e-06,
      "step": 8197,
      "training_step_time": 7.8168065547943115
    },
    {
      "epoch": 7.818222045898438e-06,
      "model_forward_time": 1.7122726440429688,
      "step": 8198
    },
    {
      "epoch": 7.818222045898438e-06,
      "step": 8198,
      "training_step_time": 7.63374137878418
    },
    {
      "epoch": 7.819175720214844e-06,
      "model_forward_time": 1.7173688411712646,
      "step": 8199
    },
    {
      "epoch": 7.819175720214844e-06,
      "step": 8199,
      "training_step_time": 8.625016212463379
    },
    {
      "epoch": 7.82012939453125e-06,
      "grad_norm": 0.25306436419487,
      "learning_rate": 9.972047802335402e-06,
      "loss": 0.0759,
      "step": 8200
    },
    {
      "epoch": 7.82012939453125e-06,
      "model_forward_time": 1.6995325088500977,
      "step": 8200
    },
    {
      "dynamics_loss_avg": 0.07379833273589612,
      "epoch": 7.82012939453125e-06,
      "step": 8200
    },
    {
      "action_loss_avg": 0.00634462961461395,
      "epoch": 7.82012939453125e-06,
      "step": 8200
    },
    {
      "epoch": 7.82012939453125e-06,
      "step": 8200,
      "training_step_time": 7.333559036254883
    },
    {
      "epoch": 7.821083068847657e-06,
      "model_forward_time": 1.712268590927124,
      "step": 8201
    },
    {
      "epoch": 7.821083068847657e-06,
      "step": 8201,
      "training_step_time": 9.539550304412842
    },
    {
      "epoch": 7.822036743164062e-06,
      "model_forward_time": 1.7022042274475098,
      "step": 8202
    },
    {
      "epoch": 7.822036743164062e-06,
      "step": 8202,
      "training_step_time": 7.882295370101929
    },
    {
      "epoch": 7.822990417480468e-06,
      "model_forward_time": 2.2657711505889893,
      "step": 8203
    },
    {
      "epoch": 7.822990417480468e-06,
      "step": 8203,
      "training_step_time": 8.513875007629395
    },
    {
      "epoch": 7.823944091796876e-06,
      "model_forward_time": 1.7367231845855713,
      "step": 8204
    },
    {
      "epoch": 7.823944091796876e-06,
      "step": 8204,
      "training_step_time": 7.604807138442993
    },
    {
      "epoch": 7.824897766113281e-06,
      "model_forward_time": 1.7052361965179443,
      "step": 8205
    },
    {
      "epoch": 7.824897766113281e-06,
      "step": 8205,
      "training_step_time": 8.817124843597412
    },
    {
      "epoch": 7.825851440429687e-06,
      "model_forward_time": 1.7038156986236572,
      "step": 8206
    },
    {
      "epoch": 7.825851440429687e-06,
      "step": 8206,
      "training_step_time": 7.723526477813721
    },
    {
      "epoch": 7.826805114746094e-06,
      "model_forward_time": 1.7080285549163818,
      "step": 8207
    },
    {
      "epoch": 7.826805114746094e-06,
      "step": 8207,
      "training_step_time": 7.5317957401275635
    },
    {
      "epoch": 7.8277587890625e-06,
      "model_forward_time": 1.709792137145996,
      "step": 8208
    },
    {
      "epoch": 7.8277587890625e-06,
      "step": 8208,
      "training_step_time": 7.4467785358428955
    },
    {
      "epoch": 7.828712463378906e-06,
      "model_forward_time": 1.7125880718231201,
      "step": 8209
    },
    {
      "epoch": 7.828712463378906e-06,
      "step": 8209,
      "training_step_time": 7.546107053756714
    },
    {
      "epoch": 7.829666137695313e-06,
      "grad_norm": 0.22446797788143158,
      "learning_rate": 9.971872937633489e-06,
      "loss": 0.0757,
      "step": 8210
    },
    {
      "epoch": 7.829666137695313e-06,
      "model_forward_time": 1.705366849899292,
      "step": 8210
    },
    {
      "dynamics_loss_avg": 0.062303471751511096,
      "epoch": 7.829666137695313e-06,
      "step": 8210
    },
    {
      "action_loss_avg": 0.004534079087898135,
      "epoch": 7.829666137695313e-06,
      "step": 8210
    },
    {
      "epoch": 7.829666137695313e-06,
      "step": 8210,
      "training_step_time": 9.099231243133545
    },
    {
      "epoch": 7.830619812011719e-06,
      "model_forward_time": 1.7085545063018799,
      "step": 8211
    },
    {
      "epoch": 7.830619812011719e-06,
      "step": 8211,
      "training_step_time": 7.789883613586426
    },
    {
      "epoch": 7.831573486328125e-06,
      "model_forward_time": 2.2565815448760986,
      "step": 8212
    },
    {
      "epoch": 7.831573486328125e-06,
      "step": 8212,
      "training_step_time": 7.928375482559204
    },
    {
      "epoch": 7.83252716064453e-06,
      "model_forward_time": 1.705806016921997,
      "step": 8213
    },
    {
      "epoch": 7.83252716064453e-06,
      "step": 8213,
      "training_step_time": 9.159059524536133
    },
    {
      "epoch": 7.833480834960938e-06,
      "model_forward_time": 1.710848093032837,
      "step": 8214
    },
    {
      "epoch": 7.833480834960938e-06,
      "step": 8214,
      "training_step_time": 10.602452516555786
    },
    {
      "epoch": 7.834434509277344e-06,
      "model_forward_time": 1.7144231796264648,
      "step": 8215
    },
    {
      "epoch": 7.834434509277344e-06,
      "step": 8215,
      "training_step_time": 7.875826835632324
    },
    {
      "epoch": 7.83538818359375e-06,
      "model_forward_time": 1.7094180583953857,
      "step": 8216
    },
    {
      "epoch": 7.83538818359375e-06,
      "step": 8216,
      "training_step_time": 7.722563743591309
    },
    {
      "epoch": 7.836341857910157e-06,
      "model_forward_time": 1.7160582542419434,
      "step": 8217
    },
    {
      "epoch": 7.836341857910157e-06,
      "step": 8217,
      "training_step_time": 7.571381092071533
    },
    {
      "epoch": 7.837295532226562e-06,
      "model_forward_time": 1.3503644466400146,
      "step": 8218
    },
    {
      "epoch": 7.837295532226562e-06,
      "step": 8218,
      "training_step_time": 7.615539789199829
    },
    {
      "epoch": 7.838249206542968e-06,
      "model_forward_time": 1.7152860164642334,
      "step": 8219
    },
    {
      "epoch": 7.838249206542968e-06,
      "step": 8219,
      "training_step_time": 7.257076025009155
    },
    {
      "epoch": 7.839202880859376e-06,
      "grad_norm": 0.26552116870880127,
      "learning_rate": 9.971697529215025e-06,
      "loss": 0.0757,
      "step": 8220
    },
    {
      "epoch": 7.839202880859376e-06,
      "model_forward_time": 1.7066826820373535,
      "step": 8220
    },
    {
      "dynamics_loss_avg": 0.06864883825182914,
      "epoch": 7.839202880859376e-06,
      "step": 8220
    },
    {
      "action_loss_avg": 0.005513991881161928,
      "epoch": 7.839202880859376e-06,
      "step": 8220
    },
    {
      "epoch": 7.839202880859376e-06,
      "step": 8220,
      "training_step_time": 7.792742967605591
    },
    {
      "epoch": 7.840156555175781e-06,
      "model_forward_time": 2.2720699310302734,
      "step": 8221
    },
    {
      "epoch": 7.840156555175781e-06,
      "step": 8221,
      "training_step_time": 8.346401929855347
    },
    {
      "epoch": 7.841110229492187e-06,
      "model_forward_time": 1.7000768184661865,
      "step": 8222
    },
    {
      "epoch": 7.841110229492187e-06,
      "step": 8222,
      "training_step_time": 8.569462299346924
    },
    {
      "epoch": 7.842063903808594e-06,
      "model_forward_time": 1.7682905197143555,
      "step": 8223
    },
    {
      "epoch": 7.842063903808594e-06,
      "step": 8223,
      "training_step_time": 8.328141689300537
    },
    {
      "epoch": 7.843017578125e-06,
      "model_forward_time": 1.6967129707336426,
      "step": 8224
    },
    {
      "epoch": 7.843017578125e-06,
      "step": 8224,
      "training_step_time": 8.765945196151733
    },
    {
      "epoch": 7.843971252441406e-06,
      "model_forward_time": 1.7088000774383545,
      "step": 8225
    },
    {
      "epoch": 7.843971252441406e-06,
      "step": 8225,
      "training_step_time": 7.677666664123535
    },
    {
      "epoch": 7.844924926757813e-06,
      "model_forward_time": 1.7117774486541748,
      "step": 8226
    },
    {
      "epoch": 7.844924926757813e-06,
      "step": 8226,
      "training_step_time": 7.546120643615723
    },
    {
      "epoch": 7.845878601074219e-06,
      "model_forward_time": 1.705697774887085,
      "step": 8227
    },
    {
      "epoch": 7.845878601074219e-06,
      "step": 8227,
      "training_step_time": 7.648859977722168
    },
    {
      "epoch": 7.846832275390625e-06,
      "model_forward_time": 1.7228834629058838,
      "step": 8228
    },
    {
      "epoch": 7.846832275390625e-06,
      "step": 8228,
      "training_step_time": 7.235690116882324
    },
    {
      "epoch": 7.84778594970703e-06,
      "model_forward_time": 1.7195663452148438,
      "step": 8229
    },
    {
      "epoch": 7.84778594970703e-06,
      "step": 8229,
      "training_step_time": 7.795603036880493
    },
    {
      "epoch": 7.848739624023438e-06,
      "grad_norm": 0.2834405303001404,
      "learning_rate": 9.971521577099192e-06,
      "loss": 0.0714,
      "step": 8230
    },
    {
      "epoch": 7.848739624023438e-06,
      "model_forward_time": 2.2347614765167236,
      "step": 8230
    },
    {
      "dynamics_loss_avg": 0.0684117829427123,
      "epoch": 7.848739624023438e-06,
      "step": 8230
    },
    {
      "action_loss_avg": 0.00689259588252753,
      "epoch": 7.848739624023438e-06,
      "step": 8230
    },
    {
      "epoch": 7.848739624023438e-06,
      "step": 8230,
      "training_step_time": 7.760969877243042
    },
    {
      "epoch": 7.849693298339844e-06,
      "model_forward_time": 1.7150285243988037,
      "step": 8231
    },
    {
      "epoch": 7.849693298339844e-06,
      "step": 8231,
      "training_step_time": 8.612516403198242
    },
    {
      "epoch": 7.85064697265625e-06,
      "model_forward_time": 1.756211519241333,
      "step": 8232
    },
    {
      "epoch": 7.85064697265625e-06,
      "step": 8232,
      "training_step_time": 7.701541900634766
    },
    {
      "epoch": 7.851600646972657e-06,
      "model_forward_time": 1.709233283996582,
      "step": 8233
    },
    {
      "epoch": 7.851600646972657e-06,
      "step": 8233,
      "training_step_time": 10.28787088394165
    },
    {
      "epoch": 7.852554321289062e-06,
      "model_forward_time": 1.7096014022827148,
      "step": 8234
    },
    {
      "epoch": 7.852554321289062e-06,
      "step": 8234,
      "training_step_time": 9.500084161758423
    },
    {
      "epoch": 7.853507995605468e-06,
      "model_forward_time": 1.7127418518066406,
      "step": 8235
    },
    {
      "epoch": 7.853507995605468e-06,
      "step": 8235,
      "training_step_time": 7.440531969070435
    },
    {
      "epoch": 7.854461669921876e-06,
      "model_forward_time": 1.7534313201904297,
      "step": 8236
    },
    {
      "epoch": 7.854461669921876e-06,
      "step": 8236,
      "training_step_time": 7.427147626876831
    },
    {
      "epoch": 7.855415344238281e-06,
      "model_forward_time": 1.7139313220977783,
      "step": 8237
    },
    {
      "epoch": 7.855415344238281e-06,
      "step": 8237,
      "training_step_time": 7.888000726699829
    },
    {
      "epoch": 7.856369018554687e-06,
      "model_forward_time": 1.7114453315734863,
      "step": 8238
    },
    {
      "epoch": 7.856369018554687e-06,
      "step": 8238,
      "training_step_time": 8.19939661026001
    },
    {
      "epoch": 7.857322692871094e-06,
      "model_forward_time": 1.7116835117340088,
      "step": 8239
    },
    {
      "epoch": 7.857322692871094e-06,
      "step": 8239,
      "training_step_time": 10.110037088394165
    },
    {
      "epoch": 7.8582763671875e-06,
      "grad_norm": 0.25587642192840576,
      "learning_rate": 9.971345081305238e-06,
      "loss": 0.074,
      "step": 8240
    },
    {
      "epoch": 7.8582763671875e-06,
      "model_forward_time": 2.254645586013794,
      "step": 8240
    },
    {
      "dynamics_loss_avg": 0.08387916088104248,
      "epoch": 7.8582763671875e-06,
      "step": 8240
    },
    {
      "action_loss_avg": 0.007561936869751662,
      "epoch": 7.8582763671875e-06,
      "step": 8240
    },
    {
      "epoch": 7.8582763671875e-06,
      "step": 8240,
      "training_step_time": 8.266426086425781
    },
    {
      "epoch": 7.859230041503906e-06,
      "model_forward_time": 1.757913589477539,
      "step": 8241
    },
    {
      "epoch": 7.859230041503906e-06,
      "step": 8241,
      "training_step_time": 7.704583644866943
    },
    {
      "epoch": 7.860183715820313e-06,
      "model_forward_time": 1.7146732807159424,
      "step": 8242
    },
    {
      "epoch": 7.860183715820313e-06,
      "step": 8242,
      "training_step_time": 7.7570884227752686
    },
    {
      "epoch": 7.861137390136719e-06,
      "model_forward_time": 1.7145073413848877,
      "step": 8243
    },
    {
      "epoch": 7.861137390136719e-06,
      "step": 8243,
      "training_step_time": 8.478602886199951
    },
    {
      "epoch": 7.862091064453125e-06,
      "model_forward_time": 1.7002146244049072,
      "step": 8244
    },
    {
      "epoch": 7.862091064453125e-06,
      "step": 8244,
      "training_step_time": 7.425788164138794
    },
    {
      "epoch": 7.86304473876953e-06,
      "model_forward_time": 1.7163169384002686,
      "step": 8245
    },
    {
      "epoch": 7.86304473876953e-06,
      "step": 8245,
      "training_step_time": 7.4828782081604
    },
    {
      "epoch": 7.863998413085938e-06,
      "model_forward_time": 1.7070655822753906,
      "step": 8246
    },
    {
      "epoch": 7.863998413085938e-06,
      "step": 8246,
      "training_step_time": 7.023207664489746
    },
    {
      "epoch": 7.864952087402344e-06,
      "model_forward_time": 1.7147126197814941,
      "step": 8247
    },
    {
      "epoch": 7.864952087402344e-06,
      "step": 8247,
      "training_step_time": 7.594390392303467
    },
    {
      "epoch": 7.86590576171875e-06,
      "model_forward_time": 1.7009079456329346,
      "step": 8248
    },
    {
      "epoch": 7.86590576171875e-06,
      "step": 8248,
      "training_step_time": 7.773974180221558
    },
    {
      "epoch": 7.866859436035157e-06,
      "model_forward_time": 2.2443878650665283,
      "step": 8249
    },
    {
      "epoch": 7.866859436035157e-06,
      "step": 8249,
      "training_step_time": 7.677976846694946
    },
    {
      "epoch": 7.867813110351562e-06,
      "grad_norm": 0.2794078290462494,
      "learning_rate": 9.971168041852456e-06,
      "loss": 0.0744,
      "step": 8250
    },
    {
      "epoch": 7.867813110351562e-06,
      "model_forward_time": 1.7433290481567383,
      "step": 8250
    },
    {
      "dynamics_loss_avg": 0.05688280761241913,
      "epoch": 7.867813110351562e-06,
      "step": 8250
    },
    {
      "action_loss_avg": 0.004419613152276724,
      "epoch": 7.867813110351562e-06,
      "step": 8250
    },
    {
      "epoch": 7.867813110351562e-06,
      "step": 8250,
      "training_step_time": 9.203696012496948
    },
    {
      "epoch": 7.868766784667968e-06,
      "model_forward_time": 1.7034146785736084,
      "step": 8251
    },
    {
      "epoch": 7.868766784667968e-06,
      "step": 8251,
      "training_step_time": 7.787979364395142
    },
    {
      "epoch": 7.869720458984376e-06,
      "model_forward_time": 1.697190523147583,
      "step": 8252
    },
    {
      "epoch": 7.869720458984376e-06,
      "step": 8252,
      "training_step_time": 7.697424650192261
    },
    {
      "epoch": 7.870674133300781e-06,
      "model_forward_time": 1.7291233539581299,
      "step": 8253
    },
    {
      "epoch": 7.870674133300781e-06,
      "step": 8253,
      "training_step_time": 7.490018606185913
    },
    {
      "epoch": 7.871627807617187e-06,
      "model_forward_time": 1.711949110031128,
      "step": 8254
    },
    {
      "epoch": 7.871627807617187e-06,
      "step": 8254,
      "training_step_time": 7.988190174102783
    },
    {
      "epoch": 7.872581481933594e-06,
      "model_forward_time": 1.7103068828582764,
      "step": 8255
    },
    {
      "epoch": 7.872581481933594e-06,
      "step": 8255,
      "training_step_time": 7.2322022914886475
    },
    {
      "epoch": 7.87353515625e-06,
      "model_forward_time": 1.7175135612487793,
      "step": 8256
    },
    {
      "epoch": 7.87353515625e-06,
      "step": 8256,
      "training_step_time": 7.376698970794678
    },
    {
      "epoch": 7.874488830566406e-06,
      "model_forward_time": 1.7149896621704102,
      "step": 8257
    },
    {
      "epoch": 7.874488830566406e-06,
      "step": 8257,
      "training_step_time": 7.713392734527588
    },
    {
      "epoch": 7.875442504882812e-06,
      "model_forward_time": 2.2497477531433105,
      "step": 8258
    },
    {
      "epoch": 7.875442504882812e-06,
      "step": 8258,
      "training_step_time": 7.64760160446167
    },
    {
      "epoch": 7.876396179199219e-06,
      "model_forward_time": 1.7140753269195557,
      "step": 8259
    },
    {
      "epoch": 7.876396179199219e-06,
      "step": 8259,
      "training_step_time": 8.34785795211792
    },
    {
      "epoch": 7.877349853515625e-06,
      "grad_norm": 0.3689747750759125,
      "learning_rate": 9.970990458760214e-06,
      "loss": 0.0773,
      "step": 8260
    },
    {
      "epoch": 7.877349853515625e-06,
      "model_forward_time": 1.716064214706421,
      "step": 8260
    },
    {
      "dynamics_loss_avg": 0.0745331471785903,
      "epoch": 7.877349853515625e-06,
      "step": 8260
    },
    {
      "action_loss_avg": 0.007274807139765471,
      "epoch": 7.877349853515625e-06,
      "step": 8260
    },
    {
      "epoch": 7.877349853515625e-06,
      "step": 8260,
      "training_step_time": 7.779231309890747
    },
    {
      "epoch": 7.87830352783203e-06,
      "model_forward_time": 1.7001633644104004,
      "step": 8261
    },
    {
      "epoch": 7.87830352783203e-06,
      "step": 8261,
      "training_step_time": 7.659708499908447
    },
    {
      "epoch": 7.879257202148438e-06,
      "model_forward_time": 1.709770917892456,
      "step": 8262
    },
    {
      "epoch": 7.879257202148438e-06,
      "step": 8262,
      "training_step_time": 9.599818229675293
    },
    {
      "epoch": 7.880210876464844e-06,
      "model_forward_time": 1.7047467231750488,
      "step": 8263
    },
    {
      "epoch": 7.880210876464844e-06,
      "step": 8263,
      "training_step_time": 7.453153848648071
    },
    {
      "epoch": 7.88116455078125e-06,
      "model_forward_time": 1.7202222347259521,
      "step": 8264
    },
    {
      "epoch": 7.88116455078125e-06,
      "step": 8264,
      "training_step_time": 7.25575590133667
    },
    {
      "epoch": 7.882118225097657e-06,
      "model_forward_time": 1.7133612632751465,
      "step": 8265
    },
    {
      "epoch": 7.882118225097657e-06,
      "step": 8265,
      "training_step_time": 7.309154272079468
    },
    {
      "epoch": 7.883071899414063e-06,
      "model_forward_time": 1.7106621265411377,
      "step": 8266
    },
    {
      "epoch": 7.883071899414063e-06,
      "step": 8266,
      "training_step_time": 7.69144082069397
    },
    {
      "epoch": 7.884025573730468e-06,
      "model_forward_time": 2.2457339763641357,
      "step": 8267
    },
    {
      "epoch": 7.884025573730468e-06,
      "step": 8267,
      "training_step_time": 8.61481499671936
    },
    {
      "epoch": 7.884979248046876e-06,
      "model_forward_time": 1.7228620052337646,
      "step": 8268
    },
    {
      "epoch": 7.884979248046876e-06,
      "step": 8268,
      "training_step_time": 7.683724641799927
    },
    {
      "epoch": 7.885932922363281e-06,
      "model_forward_time": 1.7689502239227295,
      "step": 8269
    },
    {
      "epoch": 7.885932922363281e-06,
      "step": 8269,
      "training_step_time": 8.71284031867981
    },
    {
      "epoch": 7.886886596679687e-06,
      "grad_norm": 0.24477826058864594,
      "learning_rate": 9.970812332047929e-06,
      "loss": 0.0686,
      "step": 8270
    },
    {
      "epoch": 7.886886596679687e-06,
      "model_forward_time": 1.7062995433807373,
      "step": 8270
    },
    {
      "dynamics_loss_avg": 0.058471953682601455,
      "epoch": 7.886886596679687e-06,
      "step": 8270
    },
    {
      "action_loss_avg": 0.005884872941533103,
      "epoch": 7.886886596679687e-06,
      "step": 8270
    },
    {
      "epoch": 7.886886596679687e-06,
      "step": 8270,
      "training_step_time": 7.712541580200195
    },
    {
      "epoch": 7.887840270996095e-06,
      "model_forward_time": 1.7222614288330078,
      "step": 8271
    },
    {
      "epoch": 7.887840270996095e-06,
      "step": 8271,
      "training_step_time": 8.617479085922241
    },
    {
      "epoch": 7.8887939453125e-06,
      "model_forward_time": 1.363105297088623,
      "step": 8272
    },
    {
      "epoch": 7.8887939453125e-06,
      "step": 8272,
      "training_step_time": 7.477472543716431
    },
    {
      "epoch": 7.889747619628906e-06,
      "model_forward_time": 2.858966588973999,
      "step": 8273
    },
    {
      "epoch": 7.889747619628906e-06,
      "step": 8273,
      "training_step_time": 7.648298501968384
    },
    {
      "epoch": 7.890701293945312e-06,
      "model_forward_time": 1.7316150665283203,
      "step": 8274
    },
    {
      "epoch": 7.890701293945312e-06,
      "step": 8274,
      "training_step_time": 8.033909797668457
    },
    {
      "epoch": 7.891654968261719e-06,
      "model_forward_time": 1.7266921997070312,
      "step": 8275
    },
    {
      "epoch": 7.891654968261719e-06,
      "step": 8275,
      "training_step_time": 6.740412473678589
    },
    {
      "epoch": 7.892608642578125e-06,
      "model_forward_time": 1.7261748313903809,
      "step": 8276
    },
    {
      "epoch": 7.892608642578125e-06,
      "step": 8276,
      "training_step_time": 6.715636253356934
    },
    {
      "epoch": 7.89356231689453e-06,
      "model_forward_time": 1.7388660907745361,
      "step": 8277
    },
    {
      "epoch": 7.89356231689453e-06,
      "step": 8277,
      "training_step_time": 6.762590169906616
    },
    {
      "epoch": 7.894515991210938e-06,
      "model_forward_time": 1.3654768466949463,
      "step": 8278
    },
    {
      "epoch": 7.894515991210938e-06,
      "step": 8278,
      "training_step_time": 6.676045656204224
    },
    {
      "epoch": 7.895469665527344e-06,
      "model_forward_time": 1.6939058303833008,
      "step": 8279
    },
    {
      "epoch": 7.895469665527344e-06,
      "step": 8279,
      "training_step_time": 6.666090726852417
    },
    {
      "epoch": 7.89642333984375e-06,
      "grad_norm": 0.20508497953414917,
      "learning_rate": 9.97063366173508e-06,
      "loss": 0.0719,
      "step": 8280
    },
    {
      "epoch": 7.89642333984375e-06,
      "model_forward_time": 1.7192800045013428,
      "step": 8280
    },
    {
      "dynamics_loss_avg": 0.07673056162893772,
      "epoch": 7.89642333984375e-06,
      "step": 8280
    },
    {
      "action_loss_avg": 0.006125773466192186,
      "epoch": 7.89642333984375e-06,
      "step": 8280
    },
    {
      "epoch": 7.89642333984375e-06,
      "step": 8280,
      "training_step_time": 6.736693620681763
    },
    {
      "epoch": 7.897377014160157e-06,
      "model_forward_time": 1.7238445281982422,
      "step": 8281
    },
    {
      "epoch": 7.897377014160157e-06,
      "step": 8281,
      "training_step_time": 6.710201740264893
    },
    {
      "epoch": 7.898330688476563e-06,
      "model_forward_time": 1.7159349918365479,
      "step": 8282
    },
    {
      "epoch": 7.898330688476563e-06,
      "step": 8282,
      "training_step_time": 6.681981563568115
    },
    {
      "epoch": 7.899284362792968e-06,
      "model_forward_time": 2.2870469093322754,
      "step": 8283
    },
    {
      "epoch": 7.899284362792968e-06,
      "step": 8283,
      "training_step_time": 8.79214596748352
    },
    {
      "epoch": 7.900238037109376e-06,
      "model_forward_time": 1.7130560874938965,
      "step": 8284
    },
    {
      "epoch": 7.900238037109376e-06,
      "step": 8284,
      "training_step_time": 7.779906749725342
    },
    {
      "epoch": 7.901191711425781e-06,
      "model_forward_time": 1.727473258972168,
      "step": 8285
    },
    {
      "epoch": 7.901191711425781e-06,
      "step": 8285,
      "training_step_time": 6.749385356903076
    },
    {
      "epoch": 7.902145385742187e-06,
      "model_forward_time": 1.7184534072875977,
      "step": 8286
    },
    {
      "epoch": 7.902145385742187e-06,
      "step": 8286,
      "training_step_time": 6.70867133140564
    },
    {
      "epoch": 7.903099060058595e-06,
      "model_forward_time": 1.7198939323425293,
      "step": 8287
    },
    {
      "epoch": 7.903099060058595e-06,
      "step": 8287,
      "training_step_time": 6.695657730102539
    },
    {
      "epoch": 7.904052734375e-06,
      "model_forward_time": 1.7310400009155273,
      "step": 8288
    },
    {
      "epoch": 7.904052734375e-06,
      "step": 8288,
      "training_step_time": 6.647155284881592
    },
    {
      "epoch": 7.905006408691406e-06,
      "model_forward_time": 1.7264010906219482,
      "step": 8289
    },
    {
      "epoch": 7.905006408691406e-06,
      "step": 8289,
      "training_step_time": 6.668371200561523
    },
    {
      "epoch": 7.905960083007812e-06,
      "grad_norm": 0.2540203630924225,
      "learning_rate": 9.970454447841206e-06,
      "loss": 0.0726,
      "step": 8290
    },
    {
      "epoch": 7.905960083007812e-06,
      "model_forward_time": 1.7151999473571777,
      "step": 8290
    },
    {
      "dynamics_loss_avg": 0.05949773918837309,
      "epoch": 7.905960083007812e-06,
      "step": 8290
    },
    {
      "action_loss_avg": 0.008793600637000054,
      "epoch": 7.905960083007812e-06,
      "step": 8290
    },
    {
      "epoch": 7.905960083007812e-06,
      "step": 8290,
      "training_step_time": 6.694025993347168
    },
    {
      "epoch": 7.906913757324219e-06,
      "model_forward_time": 1.725024700164795,
      "step": 8291
    },
    {
      "epoch": 7.906913757324219e-06,
      "step": 8291,
      "training_step_time": 6.682495594024658
    },
    {
      "epoch": 7.907867431640625e-06,
      "model_forward_time": 2.266965866088867,
      "step": 8292
    },
    {
      "epoch": 7.907867431640625e-06,
      "step": 8292,
      "training_step_time": 8.337993621826172
    },
    {
      "epoch": 7.90882110595703e-06,
      "model_forward_time": 1.7237532138824463,
      "step": 8293
    },
    {
      "epoch": 7.90882110595703e-06,
      "step": 8293,
      "training_step_time": 7.6274120807647705
    },
    {
      "epoch": 7.909774780273438e-06,
      "model_forward_time": 1.7284212112426758,
      "step": 8294
    },
    {
      "epoch": 7.909774780273438e-06,
      "step": 8294,
      "training_step_time": 6.677269697189331
    },
    {
      "epoch": 7.910728454589844e-06,
      "model_forward_time": 1.7327189445495605,
      "step": 8295
    },
    {
      "epoch": 7.910728454589844e-06,
      "step": 8295,
      "training_step_time": 6.686644554138184
    },
    {
      "epoch": 7.91168212890625e-06,
      "model_forward_time": 1.7262439727783203,
      "step": 8296
    },
    {
      "epoch": 7.91168212890625e-06,
      "step": 8296,
      "training_step_time": 6.707346677780151
    },
    {
      "epoch": 7.912635803222657e-06,
      "model_forward_time": 1.7430949211120605,
      "step": 8297
    },
    {
      "epoch": 7.912635803222657e-06,
      "step": 8297,
      "training_step_time": 6.6923558712005615
    },
    {
      "epoch": 7.913589477539063e-06,
      "model_forward_time": 1.336144208908081,
      "step": 8298
    },
    {
      "epoch": 7.913589477539063e-06,
      "step": 8298,
      "training_step_time": 6.700113534927368
    },
    {
      "epoch": 7.914543151855468e-06,
      "model_forward_time": 1.7329633235931396,
      "step": 8299
    },
    {
      "epoch": 7.914543151855468e-06,
      "step": 8299,
      "training_step_time": 6.6433024406433105
    },
    {
      "epoch": 7.915496826171876e-06,
      "grad_norm": 0.24017609655857086,
      "learning_rate": 9.970274690385909e-06,
      "loss": 0.0755,
      "step": 8300
    },
    {
      "epoch": 7.915496826171876e-06,
      "model_forward_time": 1.7240099906921387,
      "step": 8300
    },
    {
      "dynamics_loss_avg": 0.0716729823499918,
      "epoch": 7.915496826171876e-06,
      "step": 8300
    },
    {
      "action_loss_avg": 0.007480567484162748,
      "epoch": 7.915496826171876e-06,
      "step": 8300
    },
    {
      "epoch": 7.915496826171876e-06,
      "step": 8300,
      "training_step_time": 6.682411193847656
    },
    {
      "epoch": 7.916450500488281e-06,
      "model_forward_time": 2.290452718734741,
      "step": 8301
    },
    {
      "epoch": 7.916450500488281e-06,
      "step": 8301,
      "training_step_time": 8.031890630722046
    },
    {
      "epoch": 7.917404174804687e-06,
      "model_forward_time": 1.7308287620544434,
      "step": 8302
    },
    {
      "epoch": 7.917404174804687e-06,
      "step": 8302,
      "training_step_time": 8.833963871002197
    },
    {
      "epoch": 7.918357849121095e-06,
      "model_forward_time": 1.732719898223877,
      "step": 8303
    },
    {
      "epoch": 7.918357849121095e-06,
      "step": 8303,
      "training_step_time": 7.4237751960754395
    },
    {
      "epoch": 7.9193115234375e-06,
      "model_forward_time": 1.7299613952636719,
      "step": 8304
    },
    {
      "epoch": 7.9193115234375e-06,
      "step": 8304,
      "training_step_time": 6.6805949211120605
    },
    {
      "epoch": 7.920265197753906e-06,
      "model_forward_time": 1.7417893409729004,
      "step": 8305
    },
    {
      "epoch": 7.920265197753906e-06,
      "step": 8305,
      "training_step_time": 6.704130411148071
    },
    {
      "epoch": 7.921218872070312e-06,
      "model_forward_time": 1.7273192405700684,
      "step": 8306
    },
    {
      "epoch": 7.921218872070312e-06,
      "step": 8306,
      "training_step_time": 6.683713436126709
    },
    {
      "epoch": 7.922172546386719e-06,
      "model_forward_time": 1.7127881050109863,
      "step": 8307
    },
    {
      "epoch": 7.922172546386719e-06,
      "step": 8307,
      "training_step_time": 6.6816465854644775
    },
    {
      "epoch": 7.923126220703125e-06,
      "model_forward_time": 1.7287559509277344,
      "step": 8308
    },
    {
      "epoch": 7.923126220703125e-06,
      "step": 8308,
      "training_step_time": 6.695176839828491
    },
    {
      "epoch": 7.92407989501953e-06,
      "model_forward_time": 1.733609676361084,
      "step": 8309
    },
    {
      "epoch": 7.92407989501953e-06,
      "step": 8309,
      "training_step_time": 6.670682668685913
    },
    {
      "epoch": 7.925033569335938e-06,
      "grad_norm": 0.35370421409606934,
      "learning_rate": 9.970094389388844e-06,
      "loss": 0.073,
      "step": 8310
    },
    {
      "epoch": 7.925033569335938e-06,
      "model_forward_time": 1.7361352443695068,
      "step": 8310
    },
    {
      "dynamics_loss_avg": 0.06999487932771445,
      "epoch": 7.925033569335938e-06,
      "step": 8310
    },
    {
      "action_loss_avg": 0.005413059890270233,
      "epoch": 7.925033569335938e-06,
      "step": 8310
    },
    {
      "epoch": 7.925033569335938e-06,
      "step": 8310,
      "training_step_time": 8.516726016998291
    },
    {
      "epoch": 7.925987243652344e-06,
      "model_forward_time": 1.7251107692718506,
      "step": 8311
    },
    {
      "epoch": 7.925987243652344e-06,
      "step": 8311,
      "training_step_time": 11.523407220840454
    },
    {
      "epoch": 7.92694091796875e-06,
      "model_forward_time": 1.7214722633361816,
      "step": 8312
    },
    {
      "epoch": 7.92694091796875e-06,
      "step": 8312,
      "training_step_time": 7.630685329437256
    },
    {
      "epoch": 7.927894592285157e-06,
      "model_forward_time": 1.7353413105010986,
      "step": 8313
    },
    {
      "epoch": 7.927894592285157e-06,
      "step": 8313,
      "training_step_time": 6.683358669281006
    },
    {
      "epoch": 7.928848266601563e-06,
      "model_forward_time": 1.7364909648895264,
      "step": 8314
    },
    {
      "epoch": 7.928848266601563e-06,
      "step": 8314,
      "training_step_time": 6.699771404266357
    },
    {
      "epoch": 7.929801940917968e-06,
      "model_forward_time": 1.7266621589660645,
      "step": 8315
    },
    {
      "epoch": 7.929801940917968e-06,
      "step": 8315,
      "training_step_time": 6.665918350219727
    },
    {
      "epoch": 7.930755615234376e-06,
      "model_forward_time": 1.739896297454834,
      "step": 8316
    },
    {
      "epoch": 7.930755615234376e-06,
      "step": 8316,
      "training_step_time": 6.678416728973389
    },
    {
      "epoch": 7.931709289550781e-06,
      "model_forward_time": 1.7214672565460205,
      "step": 8317
    },
    {
      "epoch": 7.931709289550781e-06,
      "step": 8317,
      "training_step_time": 6.677414178848267
    },
    {
      "epoch": 7.932662963867187e-06,
      "model_forward_time": 1.7290492057800293,
      "step": 8318
    },
    {
      "epoch": 7.932662963867187e-06,
      "step": 8318,
      "training_step_time": 6.66170334815979
    },
    {
      "epoch": 7.933616638183595e-06,
      "model_forward_time": 1.735774278640747,
      "step": 8319
    },
    {
      "epoch": 7.933616638183595e-06,
      "step": 8319,
      "training_step_time": 7.495700120925903
    },
    {
      "epoch": 7.9345703125e-06,
      "grad_norm": 0.32433220744132996,
      "learning_rate": 9.969913544869728e-06,
      "loss": 0.0751,
      "step": 8320
    },
    {
      "epoch": 7.9345703125e-06,
      "model_forward_time": 2.2769665718078613,
      "step": 8320
    },
    {
      "dynamics_loss_avg": 0.07169583216309547,
      "epoch": 7.9345703125e-06,
      "step": 8320
    },
    {
      "action_loss_avg": 0.005710375751368702,
      "epoch": 7.9345703125e-06,
      "step": 8320
    },
    {
      "epoch": 7.9345703125e-06,
      "step": 8320,
      "training_step_time": 10.325019359588623
    },
    {
      "epoch": 7.935523986816406e-06,
      "model_forward_time": 1.7142541408538818,
      "step": 8321
    },
    {
      "epoch": 7.935523986816406e-06,
      "step": 8321,
      "training_step_time": 7.732133150100708
    },
    {
      "epoch": 7.936477661132812e-06,
      "model_forward_time": 1.715024709701538,
      "step": 8322
    },
    {
      "epoch": 7.936477661132812e-06,
      "step": 8322,
      "training_step_time": 7.533761024475098
    },
    {
      "epoch": 7.937431335449219e-06,
      "model_forward_time": 1.7149338722229004,
      "step": 8323
    },
    {
      "epoch": 7.937431335449219e-06,
      "step": 8323,
      "training_step_time": 6.781029939651489
    },
    {
      "epoch": 7.938385009765625e-06,
      "model_forward_time": 1.71101713180542,
      "step": 8324
    },
    {
      "epoch": 7.938385009765625e-06,
      "step": 8324,
      "training_step_time": 6.778033256530762
    },
    {
      "epoch": 7.93933868408203e-06,
      "model_forward_time": 1.701439380645752,
      "step": 8325
    },
    {
      "epoch": 7.93933868408203e-06,
      "step": 8325,
      "training_step_time": 6.757536172866821
    },
    {
      "epoch": 7.940292358398438e-06,
      "model_forward_time": 1.7025477886199951,
      "step": 8326
    },
    {
      "epoch": 7.940292358398438e-06,
      "step": 8326,
      "training_step_time": 6.752547979354858
    },
    {
      "epoch": 7.941246032714844e-06,
      "model_forward_time": 1.7046844959259033,
      "step": 8327
    },
    {
      "epoch": 7.941246032714844e-06,
      "step": 8327,
      "training_step_time": 6.77591609954834
    },
    {
      "epoch": 7.94219970703125e-06,
      "model_forward_time": 1.7304856777191162,
      "step": 8328
    },
    {
      "epoch": 7.94219970703125e-06,
      "step": 8328,
      "training_step_time": 6.773280143737793
    },
    {
      "epoch": 7.943153381347657e-06,
      "model_forward_time": 2.2682673931121826,
      "step": 8329
    },
    {
      "epoch": 7.943153381347657e-06,
      "step": 8329,
      "training_step_time": 8.253326416015625
    },
    {
      "epoch": 7.944107055664063e-06,
      "grad_norm": 0.39531606435775757,
      "learning_rate": 9.96973215684834e-06,
      "loss": 0.0767,
      "step": 8330
    },
    {
      "epoch": 7.944107055664063e-06,
      "model_forward_time": 1.7092251777648926,
      "step": 8330
    },
    {
      "dynamics_loss_avg": 0.06700912043452263,
      "epoch": 7.944107055664063e-06,
      "step": 8330
    },
    {
      "action_loss_avg": 0.004275035543832928,
      "epoch": 7.944107055664063e-06,
      "step": 8330
    },
    {
      "epoch": 7.944107055664063e-06,
      "step": 8330,
      "training_step_time": 9.245839834213257
    },
    {
      "epoch": 7.945060729980468e-06,
      "model_forward_time": 1.345536708831787,
      "step": 8331
    },
    {
      "epoch": 7.945060729980468e-06,
      "step": 8331,
      "training_step_time": 7.652389764785767
    },
    {
      "epoch": 7.946014404296876e-06,
      "model_forward_time": 1.725043773651123,
      "step": 8332
    },
    {
      "epoch": 7.946014404296876e-06,
      "step": 8332,
      "training_step_time": 6.798535346984863
    },
    {
      "epoch": 7.946968078613281e-06,
      "model_forward_time": 1.722212314605713,
      "step": 8333
    },
    {
      "epoch": 7.946968078613281e-06,
      "step": 8333,
      "training_step_time": 6.800122261047363
    },
    {
      "epoch": 7.947921752929687e-06,
      "model_forward_time": 1.7148444652557373,
      "step": 8334
    },
    {
      "epoch": 7.947921752929687e-06,
      "step": 8334,
      "training_step_time": 6.797849893569946
    },
    {
      "epoch": 7.948875427246095e-06,
      "model_forward_time": 1.7235431671142578,
      "step": 8335
    },
    {
      "epoch": 7.948875427246095e-06,
      "step": 8335,
      "training_step_time": 6.807135343551636
    },
    {
      "epoch": 7.9498291015625e-06,
      "model_forward_time": 1.7062108516693115,
      "step": 8336
    },
    {
      "epoch": 7.9498291015625e-06,
      "step": 8336,
      "training_step_time": 6.677902936935425
    },
    {
      "epoch": 7.950782775878906e-06,
      "model_forward_time": 1.715022325515747,
      "step": 8337
    },
    {
      "epoch": 7.950782775878906e-06,
      "step": 8337,
      "training_step_time": 6.761199235916138
    },
    {
      "epoch": 7.951736450195312e-06,
      "model_forward_time": 2.2754082679748535,
      "step": 8338
    },
    {
      "epoch": 7.951736450195312e-06,
      "step": 8338,
      "training_step_time": 7.696927309036255
    },
    {
      "epoch": 7.952690124511719e-06,
      "model_forward_time": 1.720273494720459,
      "step": 8339
    },
    {
      "epoch": 7.952690124511719e-06,
      "step": 8339,
      "training_step_time": 7.687578916549683
    },
    {
      "epoch": 7.953643798828125e-06,
      "grad_norm": 0.2266470342874527,
      "learning_rate": 9.969550225344513e-06,
      "loss": 0.0778,
      "step": 8340
    },
    {
      "epoch": 7.953643798828125e-06,
      "model_forward_time": 1.7310214042663574,
      "step": 8340
    },
    {
      "dynamics_loss_avg": 0.08138651810586453,
      "epoch": 7.953643798828125e-06,
      "step": 8340
    },
    {
      "action_loss_avg": 0.005865720799192786,
      "epoch": 7.953643798828125e-06,
      "step": 8340
    },
    {
      "epoch": 7.953643798828125e-06,
      "step": 8340,
      "training_step_time": 7.659908056259155
    },
    {
      "epoch": 7.95459747314453e-06,
      "model_forward_time": 1.706660509109497,
      "step": 8341
    },
    {
      "epoch": 7.95459747314453e-06,
      "step": 8341,
      "training_step_time": 6.7322704792022705
    },
    {
      "epoch": 7.955551147460938e-06,
      "model_forward_time": 1.7284135818481445,
      "step": 8342
    },
    {
      "epoch": 7.955551147460938e-06,
      "step": 8342,
      "training_step_time": 6.7962000370025635
    },
    {
      "epoch": 7.956504821777344e-06,
      "model_forward_time": 1.7151999473571777,
      "step": 8343
    },
    {
      "epoch": 7.956504821777344e-06,
      "step": 8343,
      "training_step_time": 6.803683519363403
    },
    {
      "epoch": 7.95745849609375e-06,
      "model_forward_time": 1.7330985069274902,
      "step": 8344
    },
    {
      "epoch": 7.95745849609375e-06,
      "step": 8344,
      "training_step_time": 6.779683828353882
    },
    {
      "epoch": 7.958412170410157e-06,
      "model_forward_time": 1.716921329498291,
      "step": 8345
    },
    {
      "epoch": 7.958412170410157e-06,
      "step": 8345,
      "training_step_time": 6.8320348262786865
    },
    {
      "epoch": 7.959365844726563e-06,
      "model_forward_time": 1.7125558853149414,
      "step": 8346
    },
    {
      "epoch": 7.959365844726563e-06,
      "step": 8346,
      "training_step_time": 6.745564937591553
    },
    {
      "epoch": 7.960319519042968e-06,
      "model_forward_time": 2.255333662033081,
      "step": 8347
    },
    {
      "epoch": 7.960319519042968e-06,
      "step": 8347,
      "training_step_time": 8.774263143539429
    },
    {
      "epoch": 7.961273193359376e-06,
      "model_forward_time": 1.7246201038360596,
      "step": 8348
    },
    {
      "epoch": 7.961273193359376e-06,
      "step": 8348,
      "training_step_time": 10.788277626037598
    },
    {
      "epoch": 7.962226867675781e-06,
      "model_forward_time": 1.705106258392334,
      "step": 8349
    },
    {
      "epoch": 7.962226867675781e-06,
      "step": 8349,
      "training_step_time": 7.706298589706421
    },
    {
      "epoch": 7.963180541992187e-06,
      "grad_norm": 0.41652771830558777,
      "learning_rate": 9.969367750378148e-06,
      "loss": 0.0754,
      "step": 8350
    },
    {
      "epoch": 7.963180541992187e-06,
      "model_forward_time": 1.691742181777954,
      "step": 8350
    },
    {
      "dynamics_loss_avg": 0.07337041608989239,
      "epoch": 7.963180541992187e-06,
      "step": 8350
    },
    {
      "action_loss_avg": 0.005676807626150549,
      "epoch": 7.963180541992187e-06,
      "step": 8350
    },
    {
      "epoch": 7.963180541992187e-06,
      "step": 8350,
      "training_step_time": 6.79511570930481
    },
    {
      "epoch": 7.964134216308595e-06,
      "model_forward_time": 1.7161366939544678,
      "step": 8351
    },
    {
      "epoch": 7.964134216308595e-06,
      "step": 8351,
      "training_step_time": 6.901246786117554
    },
    {
      "epoch": 7.965087890625e-06,
      "model_forward_time": 1.7080416679382324,
      "step": 8352
    },
    {
      "epoch": 7.965087890625e-06,
      "step": 8352,
      "training_step_time": 6.817229509353638
    },
    {
      "epoch": 7.966041564941406e-06,
      "model_forward_time": 1.7117116451263428,
      "step": 8353
    },
    {
      "epoch": 7.966041564941406e-06,
      "step": 8353,
      "training_step_time": 6.8153395652771
    },
    {
      "epoch": 7.966995239257812e-06,
      "model_forward_time": 1.7082252502441406,
      "step": 8354
    },
    {
      "epoch": 7.966995239257812e-06,
      "step": 8354,
      "training_step_time": 6.787961006164551
    },
    {
      "epoch": 7.96794891357422e-06,
      "model_forward_time": 1.7103557586669922,
      "step": 8355
    },
    {
      "epoch": 7.96794891357422e-06,
      "step": 8355,
      "training_step_time": 6.8091490268707275
    },
    {
      "epoch": 7.968902587890625e-06,
      "model_forward_time": 1.7342960834503174,
      "step": 8356
    },
    {
      "epoch": 7.968902587890625e-06,
      "step": 8356,
      "training_step_time": 7.717355251312256
    },
    {
      "epoch": 7.96985626220703e-06,
      "model_forward_time": 2.2412281036376953,
      "step": 8357
    },
    {
      "epoch": 7.96985626220703e-06,
      "step": 8357,
      "training_step_time": 9.810161828994751
    },
    {
      "epoch": 7.970809936523438e-06,
      "model_forward_time": 1.7128138542175293,
      "step": 8358
    },
    {
      "epoch": 7.970809936523438e-06,
      "step": 8358,
      "training_step_time": 7.822987079620361
    },
    {
      "epoch": 7.971763610839844e-06,
      "model_forward_time": 1.710688829421997,
      "step": 8359
    },
    {
      "epoch": 7.971763610839844e-06,
      "step": 8359,
      "training_step_time": 7.270160436630249
    },
    {
      "epoch": 7.97271728515625e-06,
      "grad_norm": 0.3055046796798706,
      "learning_rate": 9.969184731969195e-06,
      "loss": 0.0675,
      "step": 8360
    },
    {
      "epoch": 7.97271728515625e-06,
      "model_forward_time": 1.711991310119629,
      "step": 8360
    },
    {
      "dynamics_loss_avg": 0.05237649008631706,
      "epoch": 7.97271728515625e-06,
      "step": 8360
    },
    {
      "action_loss_avg": 0.0034174898639321325,
      "epoch": 7.97271728515625e-06,
      "step": 8360
    },
    {
      "epoch": 7.97271728515625e-06,
      "step": 8360,
      "training_step_time": 6.783889293670654
    },
    {
      "epoch": 7.973670959472657e-06,
      "model_forward_time": 1.7143564224243164,
      "step": 8361
    },
    {
      "epoch": 7.973670959472657e-06,
      "step": 8361,
      "training_step_time": 6.826286554336548
    },
    {
      "epoch": 7.974624633789063e-06,
      "model_forward_time": 1.7131414413452148,
      "step": 8362
    },
    {
      "epoch": 7.974624633789063e-06,
      "step": 8362,
      "training_step_time": 6.808175086975098
    },
    {
      "epoch": 7.975578308105468e-06,
      "model_forward_time": 1.713714838027954,
      "step": 8363
    },
    {
      "epoch": 7.975578308105468e-06,
      "step": 8363,
      "training_step_time": 6.8218488693237305
    },
    {
      "epoch": 7.976531982421876e-06,
      "model_forward_time": 1.7087364196777344,
      "step": 8364
    },
    {
      "epoch": 7.976531982421876e-06,
      "step": 8364,
      "training_step_time": 6.860321521759033
    },
    {
      "epoch": 7.977485656738281e-06,
      "model_forward_time": 1.721482515335083,
      "step": 8365
    },
    {
      "epoch": 7.977485656738281e-06,
      "step": 8365,
      "training_step_time": 7.655061483383179
    },
    {
      "epoch": 7.978439331054687e-06,
      "model_forward_time": 2.245529890060425,
      "step": 8366
    },
    {
      "epoch": 7.978439331054687e-06,
      "step": 8366,
      "training_step_time": 9.330991268157959
    },
    {
      "epoch": 7.979393005371095e-06,
      "model_forward_time": 1.7111055850982666,
      "step": 8367
    },
    {
      "epoch": 7.979393005371095e-06,
      "step": 8367,
      "training_step_time": 7.802521467208862
    },
    {
      "epoch": 7.9803466796875e-06,
      "model_forward_time": 1.7130630016326904,
      "step": 8368
    },
    {
      "epoch": 7.9803466796875e-06,
      "step": 8368,
      "training_step_time": 7.57006311416626
    },
    {
      "epoch": 7.981300354003906e-06,
      "model_forward_time": 1.7152254581451416,
      "step": 8369
    },
    {
      "epoch": 7.981300354003906e-06,
      "step": 8369,
      "training_step_time": 6.86341118812561
    },
    {
      "epoch": 7.982254028320312e-06,
      "grad_norm": 0.37572532892227173,
      "learning_rate": 9.96900117013767e-06,
      "loss": 0.07,
      "step": 8370
    },
    {
      "epoch": 7.982254028320312e-06,
      "model_forward_time": 1.706707239151001,
      "step": 8370
    },
    {
      "dynamics_loss_avg": 0.05644680978730321,
      "epoch": 7.982254028320312e-06,
      "step": 8370
    },
    {
      "action_loss_avg": 0.003899699996691197,
      "epoch": 7.982254028320312e-06,
      "step": 8370
    },
    {
      "epoch": 7.982254028320312e-06,
      "step": 8370,
      "training_step_time": 6.797548532485962
    },
    {
      "epoch": 7.98320770263672e-06,
      "model_forward_time": 1.7118382453918457,
      "step": 8371
    },
    {
      "epoch": 7.98320770263672e-06,
      "step": 8371,
      "training_step_time": 6.766953706741333
    },
    {
      "epoch": 7.984161376953125e-06,
      "model_forward_time": 1.7165107727050781,
      "step": 8372
    },
    {
      "epoch": 7.984161376953125e-06,
      "step": 8372,
      "training_step_time": 6.84618878364563
    },
    {
      "epoch": 7.98511505126953e-06,
      "model_forward_time": 1.7132959365844727,
      "step": 8373
    },
    {
      "epoch": 7.98511505126953e-06,
      "step": 8373,
      "training_step_time": 6.706453323364258
    },
    {
      "epoch": 7.986068725585938e-06,
      "model_forward_time": 1.7101621627807617,
      "step": 8374
    },
    {
      "epoch": 7.986068725585938e-06,
      "step": 8374,
      "training_step_time": 7.175300359725952
    },
    {
      "epoch": 7.987022399902344e-06,
      "model_forward_time": 2.2650797367095947,
      "step": 8375
    },
    {
      "epoch": 7.987022399902344e-06,
      "step": 8375,
      "training_step_time": 9.20420241355896
    },
    {
      "epoch": 7.98797607421875e-06,
      "model_forward_time": 1.7102794647216797,
      "step": 8376
    },
    {
      "epoch": 7.98797607421875e-06,
      "step": 8376,
      "training_step_time": 9.233950853347778
    },
    {
      "epoch": 7.988929748535157e-06,
      "model_forward_time": 1.7127642631530762,
      "step": 8377
    },
    {
      "epoch": 7.988929748535157e-06,
      "step": 8377,
      "training_step_time": 8.475685358047485
    },
    {
      "epoch": 7.989883422851563e-06,
      "model_forward_time": 1.710484266281128,
      "step": 8378
    },
    {
      "epoch": 7.989883422851563e-06,
      "step": 8378,
      "training_step_time": 6.739468336105347
    },
    {
      "epoch": 7.990837097167968e-06,
      "model_forward_time": 1.7090539932250977,
      "step": 8379
    },
    {
      "epoch": 7.990837097167968e-06,
      "step": 8379,
      "training_step_time": 6.755369424819946
    },
    {
      "epoch": 7.991790771484376e-06,
      "grad_norm": 0.4408678412437439,
      "learning_rate": 9.96881706490365e-06,
      "loss": 0.0668,
      "step": 8380
    },
    {
      "epoch": 7.991790771484376e-06,
      "model_forward_time": 1.705230712890625,
      "step": 8380
    },
    {
      "dynamics_loss_avg": 0.05775104071944952,
      "epoch": 7.991790771484376e-06,
      "step": 8380
    },
    {
      "action_loss_avg": 0.004585814289748668,
      "epoch": 7.991790771484376e-06,
      "step": 8380
    },
    {
      "epoch": 7.991790771484376e-06,
      "step": 8380,
      "training_step_time": 6.755980014801025
    },
    {
      "epoch": 7.992744445800781e-06,
      "model_forward_time": 1.7155628204345703,
      "step": 8381
    },
    {
      "epoch": 7.992744445800781e-06,
      "step": 8381,
      "training_step_time": 6.824408769607544
    },
    {
      "epoch": 7.993698120117187e-06,
      "model_forward_time": 1.713057279586792,
      "step": 8382
    },
    {
      "epoch": 7.993698120117187e-06,
      "step": 8382,
      "training_step_time": 6.756053447723389
    },
    {
      "epoch": 7.994651794433593e-06,
      "model_forward_time": 1.722219467163086,
      "step": 8383
    },
    {
      "epoch": 7.994651794433593e-06,
      "step": 8383,
      "training_step_time": 7.5678160190582275
    },
    {
      "epoch": 7.99560546875e-06,
      "model_forward_time": 2.2647266387939453,
      "step": 8384
    },
    {
      "epoch": 7.99560546875e-06,
      "step": 8384,
      "training_step_time": 8.396686315536499
    },
    {
      "epoch": 7.996559143066406e-06,
      "model_forward_time": 1.7091779708862305,
      "step": 8385
    },
    {
      "epoch": 7.996559143066406e-06,
      "step": 8385,
      "training_step_time": 9.607083082199097
    },
    {
      "epoch": 7.997512817382812e-06,
      "model_forward_time": 1.7115404605865479,
      "step": 8386
    },
    {
      "epoch": 7.997512817382812e-06,
      "step": 8386,
      "training_step_time": 7.62995457649231
    },
    {
      "epoch": 7.99846649169922e-06,
      "model_forward_time": 1.7083723545074463,
      "step": 8387
    },
    {
      "epoch": 7.99846649169922e-06,
      "step": 8387,
      "training_step_time": 7.197626113891602
    },
    {
      "epoch": 7.999420166015625e-06,
      "model_forward_time": 1.722137212753296,
      "step": 8388
    },
    {
      "epoch": 7.999420166015625e-06,
      "step": 8388,
      "training_step_time": 6.703109264373779
    },
    {
      "epoch": 8.00037384033203e-06,
      "model_forward_time": 1.7158565521240234,
      "step": 8389
    },
    {
      "epoch": 8.00037384033203e-06,
      "step": 8389,
      "training_step_time": 6.7181220054626465
    },
    {
      "epoch": 8.001327514648438e-06,
      "grad_norm": 0.300983190536499,
      "learning_rate": 9.968632416287265e-06,
      "loss": 0.0692,
      "step": 8390
    },
    {
      "epoch": 8.001327514648438e-06,
      "model_forward_time": 1.7045986652374268,
      "step": 8390
    },
    {
      "dynamics_loss_avg": 0.05985859800130129,
      "epoch": 8.001327514648438e-06,
      "step": 8390
    },
    {
      "action_loss_avg": 0.00496770185418427,
      "epoch": 8.001327514648438e-06,
      "step": 8390
    },
    {
      "epoch": 8.001327514648438e-06,
      "step": 8390,
      "training_step_time": 6.705929279327393
    },
    {
      "epoch": 8.002281188964844e-06,
      "model_forward_time": 1.7132573127746582,
      "step": 8391
    },
    {
      "epoch": 8.002281188964844e-06,
      "step": 8391,
      "training_step_time": 6.687375783920288
    },
    {
      "epoch": 8.00323486328125e-06,
      "model_forward_time": 1.711812973022461,
      "step": 8392
    },
    {
      "epoch": 8.00323486328125e-06,
      "step": 8392,
      "training_step_time": 7.538904666900635
    },
    {
      "epoch": 8.004188537597657e-06,
      "model_forward_time": 2.274056911468506,
      "step": 8393
    },
    {
      "epoch": 8.004188537597657e-06,
      "step": 8393,
      "training_step_time": 9.487102746963501
    },
    {
      "epoch": 8.005142211914063e-06,
      "model_forward_time": 1.7085790634155273,
      "step": 8394
    },
    {
      "epoch": 8.005142211914063e-06,
      "step": 8394,
      "training_step_time": 9.715360403060913
    },
    {
      "epoch": 8.006095886230468e-06,
      "model_forward_time": 1.706789255142212,
      "step": 8395
    },
    {
      "epoch": 8.006095886230468e-06,
      "step": 8395,
      "training_step_time": 7.668501615524292
    },
    {
      "epoch": 8.007049560546876e-06,
      "model_forward_time": 1.7092199325561523,
      "step": 8396
    },
    {
      "epoch": 8.007049560546876e-06,
      "step": 8396,
      "training_step_time": 7.271815776824951
    },
    {
      "epoch": 8.008003234863281e-06,
      "model_forward_time": 1.7137901782989502,
      "step": 8397
    },
    {
      "epoch": 8.008003234863281e-06,
      "step": 8397,
      "training_step_time": 6.830717086791992
    },
    {
      "epoch": 8.008956909179687e-06,
      "model_forward_time": 1.7216627597808838,
      "step": 8398
    },
    {
      "epoch": 8.008956909179687e-06,
      "step": 8398,
      "training_step_time": 6.806915044784546
    },
    {
      "epoch": 8.009910583496093e-06,
      "model_forward_time": 1.7249667644500732,
      "step": 8399
    },
    {
      "epoch": 8.009910583496093e-06,
      "step": 8399,
      "training_step_time": 6.795175790786743
    },
    {
      "epoch": 8.0108642578125e-06,
      "grad_norm": 0.23882080614566803,
      "learning_rate": 9.96844722430871e-06,
      "loss": 0.0725,
      "step": 8400
    },
    {
      "epoch": 8.0108642578125e-06,
      "model_forward_time": 1.705134630203247,
      "step": 8400
    },
    {
      "dynamics_loss_avg": 0.059916661493480204,
      "epoch": 8.0108642578125e-06,
      "step": 8400
    },
    {
      "action_loss_avg": 0.005447822401765734,
      "epoch": 8.0108642578125e-06,
      "step": 8400
    },
    {
      "epoch": 8.0108642578125e-06,
      "step": 8400,
      "training_step_time": 6.7267961502075195
    },
    {
      "epoch": 8.011817932128906e-06,
      "model_forward_time": 1.715284824371338,
      "step": 8401
    },
    {
      "epoch": 8.011817932128906e-06,
      "step": 8401,
      "training_step_time": 7.5401551723480225
    },
    {
      "epoch": 8.012771606445312e-06,
      "model_forward_time": 1.7377159595489502,
      "step": 8402
    },
    {
      "epoch": 8.012771606445312e-06,
      "step": 8402,
      "training_step_time": 10.023688554763794
    },
    {
      "epoch": 8.01372528076172e-06,
      "model_forward_time": 1.698307752609253,
      "step": 8403
    },
    {
      "epoch": 8.01372528076172e-06,
      "step": 8403,
      "training_step_time": 7.793156862258911
    },
    {
      "epoch": 8.014678955078125e-06,
      "model_forward_time": 1.7201471328735352,
      "step": 8404
    },
    {
      "epoch": 8.014678955078125e-06,
      "step": 8404,
      "training_step_time": 7.606553554534912
    },
    {
      "epoch": 8.01563262939453e-06,
      "model_forward_time": 1.7092583179473877,
      "step": 8405
    },
    {
      "epoch": 8.01563262939453e-06,
      "step": 8405,
      "training_step_time": 7.244799852371216
    },
    {
      "epoch": 8.016586303710938e-06,
      "model_forward_time": 1.7096545696258545,
      "step": 8406
    },
    {
      "epoch": 8.016586303710938e-06,
      "step": 8406,
      "training_step_time": 6.697237730026245
    },
    {
      "epoch": 8.017539978027344e-06,
      "model_forward_time": 1.7177505493164062,
      "step": 8407
    },
    {
      "epoch": 8.017539978027344e-06,
      "step": 8407,
      "training_step_time": 6.745993375778198
    },
    {
      "epoch": 8.01849365234375e-06,
      "model_forward_time": 1.7157068252563477,
      "step": 8408
    },
    {
      "epoch": 8.01849365234375e-06,
      "step": 8408,
      "training_step_time": 6.672191619873047
    },
    {
      "epoch": 8.019447326660157e-06,
      "model_forward_time": 1.7243402004241943,
      "step": 8409
    },
    {
      "epoch": 8.019447326660157e-06,
      "step": 8409,
      "training_step_time": 6.716958522796631
    },
    {
      "epoch": 8.020401000976563e-06,
      "grad_norm": 0.288521945476532,
      "learning_rate": 9.968261488988235e-06,
      "loss": 0.0698,
      "step": 8410
    },
    {
      "epoch": 8.020401000976563e-06,
      "model_forward_time": 1.709155559539795,
      "step": 8410
    },
    {
      "dynamics_loss_avg": 0.08200845420360565,
      "epoch": 8.020401000976563e-06,
      "step": 8410
    },
    {
      "action_loss_avg": 0.007124142930842936,
      "epoch": 8.020401000976563e-06,
      "step": 8410
    },
    {
      "epoch": 8.020401000976563e-06,
      "step": 8410,
      "training_step_time": 7.494633674621582
    },
    {
      "epoch": 8.021354675292968e-06,
      "model_forward_time": 1.7155745029449463,
      "step": 8411
    },
    {
      "epoch": 8.021354675292968e-06,
      "step": 8411,
      "training_step_time": 8.493793249130249
    },
    {
      "epoch": 8.022308349609376e-06,
      "model_forward_time": 2.2208664417266846,
      "step": 8412
    },
    {
      "epoch": 8.022308349609376e-06,
      "step": 8412,
      "training_step_time": 8.964539766311646
    },
    {
      "epoch": 8.023262023925781e-06,
      "model_forward_time": 1.709928274154663,
      "step": 8413
    },
    {
      "epoch": 8.023262023925781e-06,
      "step": 8413,
      "training_step_time": 7.694394111633301
    },
    {
      "epoch": 8.024215698242187e-06,
      "model_forward_time": 1.7076027393341064,
      "step": 8414
    },
    {
      "epoch": 8.024215698242187e-06,
      "step": 8414,
      "training_step_time": 7.550243139266968
    },
    {
      "epoch": 8.025169372558593e-06,
      "model_forward_time": 1.717221975326538,
      "step": 8415
    },
    {
      "epoch": 8.025169372558593e-06,
      "step": 8415,
      "training_step_time": 7.153374910354614
    },
    {
      "epoch": 8.026123046875e-06,
      "model_forward_time": 1.711613655090332,
      "step": 8416
    },
    {
      "epoch": 8.026123046875e-06,
      "step": 8416,
      "training_step_time": 6.664801359176636
    },
    {
      "epoch": 8.027076721191406e-06,
      "model_forward_time": 1.710904836654663,
      "step": 8417
    },
    {
      "epoch": 8.027076721191406e-06,
      "step": 8417,
      "training_step_time": 6.671400785446167
    },
    {
      "epoch": 8.028030395507812e-06,
      "model_forward_time": 1.7096161842346191,
      "step": 8418
    },
    {
      "epoch": 8.028030395507812e-06,
      "step": 8418,
      "training_step_time": 6.69180703163147
    },
    {
      "epoch": 8.02898406982422e-06,
      "model_forward_time": 1.7048094272613525,
      "step": 8419
    },
    {
      "epoch": 8.02898406982422e-06,
      "step": 8419,
      "training_step_time": 7.467108488082886
    },
    {
      "epoch": 8.029937744140625e-06,
      "grad_norm": 0.3991638123989105,
      "learning_rate": 9.968075210346155e-06,
      "loss": 0.0693,
      "step": 8420
    },
    {
      "epoch": 8.029937744140625e-06,
      "model_forward_time": 1.7083525657653809,
      "step": 8420
    },
    {
      "dynamics_loss_avg": 0.06108528785407543,
      "epoch": 8.029937744140625e-06,
      "step": 8420
    },
    {
      "action_loss_avg": 0.007466932688839733,
      "epoch": 8.029937744140625e-06,
      "step": 8420
    },
    {
      "epoch": 8.029937744140625e-06,
      "step": 8420,
      "training_step_time": 7.46203351020813
    },
    {
      "epoch": 8.03089141845703e-06,
      "model_forward_time": 2.2557177543640137,
      "step": 8421
    },
    {
      "epoch": 8.03089141845703e-06,
      "step": 8421,
      "training_step_time": 8.742053031921387
    },
    {
      "epoch": 8.031845092773438e-06,
      "model_forward_time": 1.7153000831604004,
      "step": 8422
    },
    {
      "epoch": 8.031845092773438e-06,
      "step": 8422,
      "training_step_time": 9.463619709014893
    },
    {
      "epoch": 8.032798767089844e-06,
      "model_forward_time": 1.710928201675415,
      "step": 8423
    },
    {
      "epoch": 8.032798767089844e-06,
      "step": 8423,
      "training_step_time": 7.547974109649658
    },
    {
      "epoch": 8.03375244140625e-06,
      "model_forward_time": 1.710400104522705,
      "step": 8424
    },
    {
      "epoch": 8.03375244140625e-06,
      "step": 8424,
      "training_step_time": 7.187968969345093
    },
    {
      "epoch": 8.034706115722657e-06,
      "model_forward_time": 1.7207534313201904,
      "step": 8425
    },
    {
      "epoch": 8.034706115722657e-06,
      "step": 8425,
      "training_step_time": 6.718862771987915
    },
    {
      "epoch": 8.035659790039063e-06,
      "model_forward_time": 1.7091624736785889,
      "step": 8426
    },
    {
      "epoch": 8.035659790039063e-06,
      "step": 8426,
      "training_step_time": 6.716416358947754
    },
    {
      "epoch": 8.036613464355468e-06,
      "model_forward_time": 1.7085723876953125,
      "step": 8427
    },
    {
      "epoch": 8.036613464355468e-06,
      "step": 8427,
      "training_step_time": 6.660641431808472
    },
    {
      "epoch": 8.037567138671876e-06,
      "model_forward_time": 1.7133147716522217,
      "step": 8428
    },
    {
      "epoch": 8.037567138671876e-06,
      "step": 8428,
      "training_step_time": 6.661769866943359
    },
    {
      "epoch": 8.038520812988282e-06,
      "model_forward_time": 1.7107892036437988,
      "step": 8429
    },
    {
      "epoch": 8.038520812988282e-06,
      "step": 8429,
      "training_step_time": 7.523988246917725
    },
    {
      "epoch": 8.039474487304687e-06,
      "grad_norm": 0.20758098363876343,
      "learning_rate": 9.96788838840284e-06,
      "loss": 0.0738,
      "step": 8430
    },
    {
      "epoch": 8.039474487304687e-06,
      "model_forward_time": 2.270515203475952,
      "step": 8430
    },
    {
      "dynamics_loss_avg": 0.061868278123438356,
      "epoch": 8.039474487304687e-06,
      "step": 8430
    },
    {
      "action_loss_avg": 0.0036899049184285103,
      "epoch": 8.039474487304687e-06,
      "step": 8430
    },
    {
      "epoch": 8.039474487304687e-06,
      "step": 8430,
      "training_step_time": 8.891777276992798
    },
    {
      "epoch": 8.040428161621093e-06,
      "model_forward_time": 1.7114405632019043,
      "step": 8431
    },
    {
      "epoch": 8.040428161621093e-06,
      "step": 8431,
      "training_step_time": 9.20491909980774
    },
    {
      "epoch": 8.0413818359375e-06,
      "model_forward_time": 1.714461088180542,
      "step": 8432
    },
    {
      "epoch": 8.0413818359375e-06,
      "step": 8432,
      "training_step_time": 7.584022045135498
    },
    {
      "epoch": 8.042335510253906e-06,
      "model_forward_time": 1.7095186710357666,
      "step": 8433
    },
    {
      "epoch": 8.042335510253906e-06,
      "step": 8433,
      "training_step_time": 7.328404188156128
    },
    {
      "epoch": 8.043289184570312e-06,
      "model_forward_time": 1.335118055343628,
      "step": 8434
    },
    {
      "epoch": 8.043289184570312e-06,
      "step": 8434,
      "training_step_time": 6.666619539260864
    },
    {
      "epoch": 8.04424285888672e-06,
      "model_forward_time": 1.7031745910644531,
      "step": 8435
    },
    {
      "epoch": 8.04424285888672e-06,
      "step": 8435,
      "training_step_time": 6.736207962036133
    },
    {
      "epoch": 8.045196533203125e-06,
      "model_forward_time": 1.7222239971160889,
      "step": 8436
    },
    {
      "epoch": 8.045196533203125e-06,
      "step": 8436,
      "training_step_time": 6.713504791259766
    },
    {
      "epoch": 8.04615020751953e-06,
      "model_forward_time": 1.7112762928009033,
      "step": 8437
    },
    {
      "epoch": 8.04615020751953e-06,
      "step": 8437,
      "training_step_time": 6.655597925186157
    },
    {
      "epoch": 8.047103881835938e-06,
      "model_forward_time": 1.7081151008605957,
      "step": 8438
    },
    {
      "epoch": 8.047103881835938e-06,
      "step": 8438,
      "training_step_time": 7.5374226570129395
    },
    {
      "epoch": 8.048057556152344e-06,
      "model_forward_time": 2.2459075450897217,
      "step": 8439
    },
    {
      "epoch": 8.048057556152344e-06,
      "step": 8439,
      "training_step_time": 8.822299003601074
    },
    {
      "epoch": 8.04901123046875e-06,
      "grad_norm": 0.3446912169456482,
      "learning_rate": 9.967701023178717e-06,
      "loss": 0.0745,
      "step": 8440
    },
    {
      "epoch": 8.04901123046875e-06,
      "model_forward_time": 1.703683614730835,
      "step": 8440
    },
    {
      "dynamics_loss_avg": 0.054359902441501615,
      "epoch": 8.04901123046875e-06,
      "step": 8440
    },
    {
      "action_loss_avg": 0.003909709351137281,
      "epoch": 8.04901123046875e-06,
      "step": 8440
    },
    {
      "epoch": 8.04901123046875e-06,
      "step": 8440,
      "training_step_time": 8.417064905166626
    },
    {
      "epoch": 8.049964904785157e-06,
      "model_forward_time": 1.7133281230926514,
      "step": 8441
    },
    {
      "epoch": 8.049964904785157e-06,
      "step": 8441,
      "training_step_time": 7.651365280151367
    },
    {
      "epoch": 8.050918579101563e-06,
      "model_forward_time": 1.7074873447418213,
      "step": 8442
    },
    {
      "epoch": 8.050918579101563e-06,
      "step": 8442,
      "training_step_time": 7.392640590667725
    },
    {
      "epoch": 8.051872253417968e-06,
      "model_forward_time": 1.7070865631103516,
      "step": 8443
    },
    {
      "epoch": 8.051872253417968e-06,
      "step": 8443,
      "training_step_time": 6.6769280433654785
    },
    {
      "epoch": 8.052825927734376e-06,
      "model_forward_time": 1.7096197605133057,
      "step": 8444
    },
    {
      "epoch": 8.052825927734376e-06,
      "step": 8444,
      "training_step_time": 6.678793668746948
    },
    {
      "epoch": 8.053779602050782e-06,
      "model_forward_time": 1.7063920497894287,
      "step": 8445
    },
    {
      "epoch": 8.053779602050782e-06,
      "step": 8445,
      "training_step_time": 6.66840386390686
    },
    {
      "epoch": 8.054733276367187e-06,
      "model_forward_time": 1.7069787979125977,
      "step": 8446
    },
    {
      "epoch": 8.054733276367187e-06,
      "step": 8446,
      "training_step_time": 6.704851150512695
    },
    {
      "epoch": 8.055686950683593e-06,
      "model_forward_time": 1.7059211730957031,
      "step": 8447
    },
    {
      "epoch": 8.055686950683593e-06,
      "step": 8447,
      "training_step_time": 7.530624628067017
    },
    {
      "epoch": 8.056640625e-06,
      "model_forward_time": 1.7336299419403076,
      "step": 8448
    },
    {
      "epoch": 8.056640625e-06,
      "step": 8448,
      "training_step_time": 8.915038824081421
    },
    {
      "epoch": 8.057594299316406e-06,
      "model_forward_time": 1.334151268005371,
      "step": 8449
    },
    {
      "epoch": 8.057594299316406e-06,
      "step": 8449,
      "training_step_time": 10.036778450012207
    },
    {
      "epoch": 8.058547973632812e-06,
      "grad_norm": 0.22392676770687103,
      "learning_rate": 9.967513114694281e-06,
      "loss": 0.0755,
      "step": 8450
    },
    {
      "epoch": 8.058547973632812e-06,
      "model_forward_time": 1.7013707160949707,
      "step": 8450
    },
    {
      "dynamics_loss_avg": 0.06959527041763067,
      "epoch": 8.058547973632812e-06,
      "step": 8450
    },
    {
      "action_loss_avg": 0.005562468955758959,
      "epoch": 8.058547973632812e-06,
      "step": 8450
    },
    {
      "epoch": 8.058547973632812e-06,
      "step": 8450,
      "training_step_time": 8.214172124862671
    },
    {
      "epoch": 8.05950164794922e-06,
      "model_forward_time": 1.7077388763427734,
      "step": 8451
    },
    {
      "epoch": 8.05950164794922e-06,
      "step": 8451,
      "training_step_time": 7.486403226852417
    },
    {
      "epoch": 8.060455322265625e-06,
      "model_forward_time": 1.717193365097046,
      "step": 8452
    },
    {
      "epoch": 8.060455322265625e-06,
      "step": 8452,
      "training_step_time": 7.2799577713012695
    },
    {
      "epoch": 8.06140899658203e-06,
      "model_forward_time": 1.7069098949432373,
      "step": 8453
    },
    {
      "epoch": 8.06140899658203e-06,
      "step": 8453,
      "training_step_time": 6.704889297485352
    },
    {
      "epoch": 8.062362670898438e-06,
      "model_forward_time": 1.7096643447875977,
      "step": 8454
    },
    {
      "epoch": 8.062362670898438e-06,
      "step": 8454,
      "training_step_time": 6.689133644104004
    },
    {
      "epoch": 8.063316345214844e-06,
      "model_forward_time": 1.7089793682098389,
      "step": 8455
    },
    {
      "epoch": 8.063316345214844e-06,
      "step": 8455,
      "training_step_time": 6.737472772598267
    },
    {
      "epoch": 8.06427001953125e-06,
      "model_forward_time": 1.7062182426452637,
      "step": 8456
    },
    {
      "epoch": 8.06427001953125e-06,
      "step": 8456,
      "training_step_time": 7.675083875656128
    },
    {
      "epoch": 8.065223693847657e-06,
      "model_forward_time": 1.7219741344451904,
      "step": 8457
    },
    {
      "epoch": 8.065223693847657e-06,
      "step": 8457,
      "training_step_time": 8.429415702819824
    },
    {
      "epoch": 8.066177368164063e-06,
      "model_forward_time": 2.234203577041626,
      "step": 8458
    },
    {
      "epoch": 8.066177368164063e-06,
      "step": 8458,
      "training_step_time": 9.150825262069702
    },
    {
      "epoch": 8.067131042480468e-06,
      "model_forward_time": 1.7135906219482422,
      "step": 8459
    },
    {
      "epoch": 8.067131042480468e-06,
      "step": 8459,
      "training_step_time": 9.919864416122437
    },
    {
      "epoch": 8.068084716796876e-06,
      "grad_norm": 0.4335807263851166,
      "learning_rate": 9.96732466297008e-06,
      "loss": 0.0754,
      "step": 8460
    },
    {
      "epoch": 8.068084716796876e-06,
      "model_forward_time": 1.7020020484924316,
      "step": 8460
    },
    {
      "dynamics_loss_avg": 0.0891087580472231,
      "epoch": 8.068084716796876e-06,
      "step": 8460
    },
    {
      "action_loss_avg": 0.005539338500238955,
      "epoch": 8.068084716796876e-06,
      "step": 8460
    },
    {
      "epoch": 8.068084716796876e-06,
      "step": 8460,
      "training_step_time": 7.455868482589722
    },
    {
      "epoch": 8.069038391113282e-06,
      "model_forward_time": 1.7137401103973389,
      "step": 8461
    },
    {
      "epoch": 8.069038391113282e-06,
      "step": 8461,
      "training_step_time": 7.293093204498291
    },
    {
      "epoch": 8.069992065429687e-06,
      "model_forward_time": 1.7073392868041992,
      "step": 8462
    },
    {
      "epoch": 8.069992065429687e-06,
      "step": 8462,
      "training_step_time": 6.656692266464233
    },
    {
      "epoch": 8.070945739746093e-06,
      "model_forward_time": 1.7070891857147217,
      "step": 8463
    },
    {
      "epoch": 8.070945739746093e-06,
      "step": 8463,
      "training_step_time": 6.724515914916992
    },
    {
      "epoch": 8.0718994140625e-06,
      "model_forward_time": 1.706758975982666,
      "step": 8464
    },
    {
      "epoch": 8.0718994140625e-06,
      "step": 8464,
      "training_step_time": 6.695271253585815
    },
    {
      "epoch": 8.072853088378906e-06,
      "model_forward_time": 1.7075140476226807,
      "step": 8465
    },
    {
      "epoch": 8.072853088378906e-06,
      "step": 8465,
      "training_step_time": 7.6185619831085205
    },
    {
      "epoch": 8.073806762695312e-06,
      "model_forward_time": 1.711097002029419,
      "step": 8466
    },
    {
      "epoch": 8.073806762695312e-06,
      "step": 8466,
      "training_step_time": 7.626819849014282
    },
    {
      "epoch": 8.07476043701172e-06,
      "model_forward_time": 2.255709171295166,
      "step": 8467
    },
    {
      "epoch": 8.07476043701172e-06,
      "step": 8467,
      "training_step_time": 8.407450199127197
    },
    {
      "epoch": 8.075714111328125e-06,
      "model_forward_time": 1.711066722869873,
      "step": 8468
    },
    {
      "epoch": 8.075714111328125e-06,
      "step": 8468,
      "training_step_time": 9.237296104431152
    },
    {
      "epoch": 8.07666778564453e-06,
      "model_forward_time": 1.7080247402191162,
      "step": 8469
    },
    {
      "epoch": 8.07666778564453e-06,
      "step": 8469,
      "training_step_time": 7.603720188140869
    },
    {
      "epoch": 8.077621459960938e-06,
      "grad_norm": 0.29517605900764465,
      "learning_rate": 9.96713566802672e-06,
      "loss": 0.0699,
      "step": 8470
    },
    {
      "epoch": 8.077621459960938e-06,
      "model_forward_time": 1.7052326202392578,
      "step": 8470
    },
    {
      "dynamics_loss_avg": 0.08820408098399639,
      "epoch": 8.077621459960938e-06,
      "step": 8470
    },
    {
      "action_loss_avg": 0.007908792560920119,
      "epoch": 8.077621459960938e-06,
      "step": 8470
    },
    {
      "epoch": 8.077621459960938e-06,
      "step": 8470,
      "training_step_time": 7.371444940567017
    },
    {
      "epoch": 8.078575134277344e-06,
      "model_forward_time": 1.71246337890625,
      "step": 8471
    },
    {
      "epoch": 8.078575134277344e-06,
      "step": 8471,
      "training_step_time": 6.756695032119751
    },
    {
      "epoch": 8.07952880859375e-06,
      "model_forward_time": 1.7136421203613281,
      "step": 8472
    },
    {
      "epoch": 8.07952880859375e-06,
      "step": 8472,
      "training_step_time": 6.7532384395599365
    },
    {
      "epoch": 8.080482482910157e-06,
      "model_forward_time": 1.7110519409179688,
      "step": 8473
    },
    {
      "epoch": 8.080482482910157e-06,
      "step": 8473,
      "training_step_time": 6.80716872215271
    },
    {
      "epoch": 8.081436157226563e-06,
      "model_forward_time": 1.7133879661560059,
      "step": 8474
    },
    {
      "epoch": 8.081436157226563e-06,
      "step": 8474,
      "training_step_time": 7.725956439971924
    },
    {
      "epoch": 8.082389831542968e-06,
      "model_forward_time": 1.7095463275909424,
      "step": 8475
    },
    {
      "epoch": 8.082389831542968e-06,
      "step": 8475,
      "training_step_time": 7.6750099658966064
    },
    {
      "epoch": 8.083343505859376e-06,
      "model_forward_time": 2.2629880905151367,
      "step": 8476
    },
    {
      "epoch": 8.083343505859376e-06,
      "step": 8476,
      "training_step_time": 8.780193567276001
    },
    {
      "epoch": 8.084297180175782e-06,
      "model_forward_time": 1.3305153846740723,
      "step": 8477
    },
    {
      "epoch": 8.084297180175782e-06,
      "step": 8477,
      "training_step_time": 9.051288843154907
    },
    {
      "epoch": 8.085250854492187e-06,
      "model_forward_time": 1.7105259895324707,
      "step": 8478
    },
    {
      "epoch": 8.085250854492187e-06,
      "step": 8478,
      "training_step_time": 8.629157304763794
    },
    {
      "epoch": 8.086204528808593e-06,
      "model_forward_time": 1.7097523212432861,
      "step": 8479
    },
    {
      "epoch": 8.086204528808593e-06,
      "step": 8479,
      "training_step_time": 7.392291069030762
    },
    {
      "epoch": 8.087158203125e-06,
      "grad_norm": 0.3259721100330353,
      "learning_rate": 9.966946129884874e-06,
      "loss": 0.0748,
      "step": 8480
    },
    {
      "epoch": 8.087158203125e-06,
      "model_forward_time": 1.7075226306915283,
      "step": 8480
    },
    {
      "dynamics_loss_avg": 0.05508048757910729,
      "epoch": 8.087158203125e-06,
      "step": 8480
    },
    {
      "action_loss_avg": 0.004763400251977146,
      "epoch": 8.087158203125e-06,
      "step": 8480
    },
    {
      "epoch": 8.087158203125e-06,
      "step": 8480,
      "training_step_time": 7.333074331283569
    },
    {
      "epoch": 8.088111877441406e-06,
      "model_forward_time": 1.7138354778289795,
      "step": 8481
    },
    {
      "epoch": 8.088111877441406e-06,
      "step": 8481,
      "training_step_time": 6.732487440109253
    },
    {
      "epoch": 8.089065551757812e-06,
      "model_forward_time": 1.7097549438476562,
      "step": 8482
    },
    {
      "epoch": 8.089065551757812e-06,
      "step": 8482,
      "training_step_time": 6.78357195854187
    },
    {
      "epoch": 8.09001922607422e-06,
      "model_forward_time": 1.705378770828247,
      "step": 8483
    },
    {
      "epoch": 8.09001922607422e-06,
      "step": 8483,
      "training_step_time": 7.5935750007629395
    },
    {
      "epoch": 8.090972900390625e-06,
      "model_forward_time": 1.7234981060028076,
      "step": 8484
    },
    {
      "epoch": 8.090972900390625e-06,
      "step": 8484,
      "training_step_time": 7.665852785110474
    },
    {
      "epoch": 8.09192657470703e-06,
      "model_forward_time": 2.2656185626983643,
      "step": 8485
    },
    {
      "epoch": 8.09192657470703e-06,
      "step": 8485,
      "training_step_time": 8.944669485092163
    },
    {
      "epoch": 8.092880249023438e-06,
      "model_forward_time": 1.7117340564727783,
      "step": 8486
    },
    {
      "epoch": 8.092880249023438e-06,
      "step": 8486,
      "training_step_time": 9.457436561584473
    },
    {
      "epoch": 8.093833923339844e-06,
      "model_forward_time": 1.7141258716583252,
      "step": 8487
    },
    {
      "epoch": 8.093833923339844e-06,
      "step": 8487,
      "training_step_time": 9.086273670196533
    },
    {
      "epoch": 8.09478759765625e-06,
      "model_forward_time": 1.7165875434875488,
      "step": 8488
    },
    {
      "epoch": 8.09478759765625e-06,
      "step": 8488,
      "training_step_time": 7.426218032836914
    },
    {
      "epoch": 8.095741271972657e-06,
      "model_forward_time": 1.7149674892425537,
      "step": 8489
    },
    {
      "epoch": 8.095741271972657e-06,
      "step": 8489,
      "training_step_time": 7.341779470443726
    },
    {
      "epoch": 8.096694946289063e-06,
      "grad_norm": 0.3052823841571808,
      "learning_rate": 9.966756048565265e-06,
      "loss": 0.0737,
      "step": 8490
    },
    {
      "epoch": 8.096694946289063e-06,
      "model_forward_time": 1.7074964046478271,
      "step": 8490
    },
    {
      "dynamics_loss_avg": 0.07259822860360146,
      "epoch": 8.096694946289063e-06,
      "step": 8490
    },
    {
      "action_loss_avg": 0.008045470144134014,
      "epoch": 8.096694946289063e-06,
      "step": 8490
    },
    {
      "epoch": 8.096694946289063e-06,
      "step": 8490,
      "training_step_time": 6.7154340744018555
    },
    {
      "epoch": 8.097648620605468e-06,
      "model_forward_time": 1.7094628810882568,
      "step": 8491
    },
    {
      "epoch": 8.097648620605468e-06,
      "step": 8491,
      "training_step_time": 6.776462554931641
    },
    {
      "epoch": 8.098602294921876e-06,
      "model_forward_time": 1.7185561656951904,
      "step": 8492
    },
    {
      "epoch": 8.098602294921876e-06,
      "step": 8492,
      "training_step_time": 7.6019127368927
    },
    {
      "epoch": 8.099555969238282e-06,
      "model_forward_time": 1.7090380191802979,
      "step": 8493
    },
    {
      "epoch": 8.099555969238282e-06,
      "step": 8493,
      "training_step_time": 7.763185501098633
    },
    {
      "epoch": 8.100509643554687e-06,
      "model_forward_time": 1.7383038997650146,
      "step": 8494
    },
    {
      "epoch": 8.100509643554687e-06,
      "step": 8494,
      "training_step_time": 7.776781797409058
    },
    {
      "epoch": 8.101463317871093e-06,
      "model_forward_time": 2.271489143371582,
      "step": 8495
    },
    {
      "epoch": 8.101463317871093e-06,
      "step": 8495,
      "training_step_time": 8.977176666259766
    },
    {
      "epoch": 8.1024169921875e-06,
      "model_forward_time": 1.7093403339385986,
      "step": 8496
    },
    {
      "epoch": 8.1024169921875e-06,
      "step": 8496,
      "training_step_time": 8.587148189544678
    },
    {
      "epoch": 8.103370666503906e-06,
      "model_forward_time": 1.718414545059204,
      "step": 8497
    },
    {
      "epoch": 8.103370666503906e-06,
      "step": 8497,
      "training_step_time": 7.646564483642578
    },
    {
      "epoch": 8.104324340820312e-06,
      "model_forward_time": 1.7094905376434326,
      "step": 8498
    },
    {
      "epoch": 8.104324340820312e-06,
      "step": 8498,
      "training_step_time": 7.236776351928711
    },
    {
      "epoch": 8.10527801513672e-06,
      "model_forward_time": 1.7115750312805176,
      "step": 8499
    },
    {
      "epoch": 8.10527801513672e-06,
      "step": 8499,
      "training_step_time": 7.30010986328125
    },
    {
      "epoch": 8.106231689453125e-06,
      "grad_norm": 0.3046548664569855,
      "learning_rate": 9.966565424088681e-06,
      "loss": 0.0789,
      "step": 8500
    },
    {
      "epoch": 8.106231689453125e-06,
      "model_forward_time": 1.707787036895752,
      "step": 8500
    },
    {
      "dynamics_loss_avg": 0.07405384704470634,
      "epoch": 8.106231689453125e-06,
      "step": 8500
    },
    {
      "action_loss_avg": 0.0052133635035716,
      "epoch": 8.106231689453125e-06,
      "step": 8500
    },
    {
      "epoch": 8.106231689453125e-06,
      "step": 8500,
      "training_step_time": 6.7281434535980225
    },
    {
      "epoch": 8.10718536376953e-06,
      "model_forward_time": 1.701676845550537,
      "step": 8501
    },
    {
      "epoch": 8.10718536376953e-06,
      "step": 8501,
      "training_step_time": 7.759531497955322
    },
    {
      "epoch": 8.108139038085938e-06,
      "model_forward_time": 1.7119505405426025,
      "step": 8502
    },
    {
      "epoch": 8.108139038085938e-06,
      "step": 8502,
      "training_step_time": 8.623831748962402
    },
    {
      "epoch": 8.109092712402344e-06,
      "model_forward_time": 1.7013351917266846,
      "step": 8503
    },
    {
      "epoch": 8.109092712402344e-06,
      "step": 8503,
      "training_step_time": 8.336479187011719
    },
    {
      "epoch": 8.11004638671875e-06,
      "model_forward_time": 2.2337894439697266,
      "step": 8504
    },
    {
      "epoch": 8.11004638671875e-06,
      "step": 8504,
      "training_step_time": 8.604607582092285
    },
    {
      "epoch": 8.111000061035157e-06,
      "model_forward_time": 1.691964864730835,
      "step": 8505
    },
    {
      "epoch": 8.111000061035157e-06,
      "step": 8505,
      "training_step_time": 10.278483629226685
    },
    {
      "epoch": 8.111953735351563e-06,
      "model_forward_time": 1.7089037895202637,
      "step": 8506
    },
    {
      "epoch": 8.111953735351563e-06,
      "step": 8506,
      "training_step_time": 7.475239515304565
    },
    {
      "epoch": 8.112907409667968e-06,
      "model_forward_time": 1.7089459896087646,
      "step": 8507
    },
    {
      "epoch": 8.112907409667968e-06,
      "step": 8507,
      "training_step_time": 7.250867128372192
    },
    {
      "epoch": 8.113861083984374e-06,
      "model_forward_time": 1.7068698406219482,
      "step": 8508
    },
    {
      "epoch": 8.113861083984374e-06,
      "step": 8508,
      "training_step_time": 7.33404803276062
    },
    {
      "epoch": 8.114814758300782e-06,
      "model_forward_time": 1.7061669826507568,
      "step": 8509
    },
    {
      "epoch": 8.114814758300782e-06,
      "step": 8509,
      "training_step_time": 6.754156827926636
    },
    {
      "epoch": 8.115768432617187e-06,
      "grad_norm": 0.30866122245788574,
      "learning_rate": 9.966374256475971e-06,
      "loss": 0.0727,
      "step": 8510
    },
    {
      "epoch": 8.115768432617187e-06,
      "model_forward_time": 1.3231391906738281,
      "step": 8510
    },
    {
      "dynamics_loss_avg": 0.06452792976051569,
      "epoch": 8.115768432617187e-06,
      "step": 8510
    },
    {
      "action_loss_avg": 0.005087879806524143,
      "epoch": 8.115768432617187e-06,
      "step": 8510
    },
    {
      "epoch": 8.115768432617187e-06,
      "step": 8510,
      "training_step_time": 6.758665323257446
    },
    {
      "epoch": 8.116722106933593e-06,
      "model_forward_time": 1.6914393901824951,
      "step": 8511
    },
    {
      "epoch": 8.116722106933593e-06,
      "step": 8511,
      "training_step_time": 7.615060329437256
    },
    {
      "epoch": 8.11767578125e-06,
      "model_forward_time": 1.7209115028381348,
      "step": 8512
    },
    {
      "epoch": 8.11767578125e-06,
      "step": 8512,
      "training_step_time": 8.242995262145996
    },
    {
      "epoch": 8.118629455566406e-06,
      "model_forward_time": 2.2327635288238525,
      "step": 8513
    },
    {
      "epoch": 8.118629455566406e-06,
      "step": 8513,
      "training_step_time": 8.60025954246521
    },
    {
      "epoch": 8.119583129882812e-06,
      "model_forward_time": 1.714066982269287,
      "step": 8514
    },
    {
      "epoch": 8.119583129882812e-06,
      "step": 8514,
      "training_step_time": 8.840138673782349
    },
    {
      "epoch": 8.12053680419922e-06,
      "model_forward_time": 1.7050292491912842,
      "step": 8515
    },
    {
      "epoch": 8.12053680419922e-06,
      "step": 8515,
      "training_step_time": 7.572124481201172
    },
    {
      "epoch": 8.121490478515625e-06,
      "model_forward_time": 1.7061543464660645,
      "step": 8516
    },
    {
      "epoch": 8.121490478515625e-06,
      "step": 8516,
      "training_step_time": 7.178843259811401
    },
    {
      "epoch": 8.12244415283203e-06,
      "model_forward_time": 1.7193527221679688,
      "step": 8517
    },
    {
      "epoch": 8.12244415283203e-06,
      "step": 8517,
      "training_step_time": 7.326017379760742
    },
    {
      "epoch": 8.123397827148438e-06,
      "model_forward_time": 1.7090530395507812,
      "step": 8518
    },
    {
      "epoch": 8.123397827148438e-06,
      "step": 8518,
      "training_step_time": 6.687580823898315
    },
    {
      "epoch": 8.124351501464844e-06,
      "model_forward_time": 1.705704927444458,
      "step": 8519
    },
    {
      "epoch": 8.124351501464844e-06,
      "step": 8519,
      "training_step_time": 6.6460840702056885
    },
    {
      "epoch": 8.12530517578125e-06,
      "grad_norm": 0.24127086997032166,
      "learning_rate": 9.96618254574804e-06,
      "loss": 0.0686,
      "step": 8520
    },
    {
      "epoch": 8.12530517578125e-06,
      "model_forward_time": 1.7008321285247803,
      "step": 8520
    },
    {
      "dynamics_loss_avg": 0.060515715181827544,
      "epoch": 8.12530517578125e-06,
      "step": 8520
    },
    {
      "action_loss_avg": 0.0059351816307753325,
      "epoch": 8.12530517578125e-06,
      "step": 8520
    },
    {
      "epoch": 8.12530517578125e-06,
      "step": 8520,
      "training_step_time": 7.6162331104278564
    },
    {
      "epoch": 8.126258850097657e-06,
      "model_forward_time": 1.708620548248291,
      "step": 8521
    },
    {
      "epoch": 8.126258850097657e-06,
      "step": 8521,
      "training_step_time": 7.731360197067261
    },
    {
      "epoch": 8.127212524414063e-06,
      "model_forward_time": 2.2659597396850586,
      "step": 8522
    },
    {
      "epoch": 8.127212524414063e-06,
      "step": 8522,
      "training_step_time": 8.223193883895874
    },
    {
      "epoch": 8.128166198730468e-06,
      "model_forward_time": 1.705923080444336,
      "step": 8523
    },
    {
      "epoch": 8.128166198730468e-06,
      "step": 8523,
      "training_step_time": 9.444091558456421
    },
    {
      "epoch": 8.129119873046874e-06,
      "model_forward_time": 1.7080085277557373,
      "step": 8524
    },
    {
      "epoch": 8.129119873046874e-06,
      "step": 8524,
      "training_step_time": 7.671628475189209
    },
    {
      "epoch": 8.130073547363282e-06,
      "model_forward_time": 1.7081522941589355,
      "step": 8525
    },
    {
      "epoch": 8.130073547363282e-06,
      "step": 8525,
      "training_step_time": 7.317748308181763
    },
    {
      "epoch": 8.131027221679687e-06,
      "model_forward_time": 1.7041490077972412,
      "step": 8526
    },
    {
      "epoch": 8.131027221679687e-06,
      "step": 8526,
      "training_step_time": 7.253559827804565
    },
    {
      "epoch": 8.131980895996093e-06,
      "model_forward_time": 1.710233449935913,
      "step": 8527
    },
    {
      "epoch": 8.131980895996093e-06,
      "step": 8527,
      "training_step_time": 6.677334308624268
    },
    {
      "epoch": 8.1329345703125e-06,
      "model_forward_time": 1.7049806118011475,
      "step": 8528
    },
    {
      "epoch": 8.1329345703125e-06,
      "step": 8528,
      "training_step_time": 6.751758813858032
    },
    {
      "epoch": 8.133888244628906e-06,
      "model_forward_time": 1.7069523334503174,
      "step": 8529
    },
    {
      "epoch": 8.133888244628906e-06,
      "step": 8529,
      "training_step_time": 7.657238245010376
    },
    {
      "epoch": 8.134841918945312e-06,
      "grad_norm": 0.3170378804206848,
      "learning_rate": 9.96599029192585e-06,
      "loss": 0.0698,
      "step": 8530
    },
    {
      "epoch": 8.134841918945312e-06,
      "model_forward_time": 1.329463005065918,
      "step": 8530
    },
    {
      "dynamics_loss_avg": 0.08462328761816025,
      "epoch": 8.134841918945312e-06,
      "step": 8530
    },
    {
      "action_loss_avg": 0.0074183880467899145,
      "epoch": 8.134841918945312e-06,
      "step": 8530
    },
    {
      "epoch": 8.134841918945312e-06,
      "step": 8530,
      "training_step_time": 9.356108665466309
    },
    {
      "epoch": 8.13579559326172e-06,
      "model_forward_time": 1.7534828186035156,
      "step": 8531
    },
    {
      "epoch": 8.13579559326172e-06,
      "step": 8531,
      "training_step_time": 9.634507656097412
    },
    {
      "epoch": 8.136749267578125e-06,
      "model_forward_time": 2.23707914352417,
      "step": 8532
    },
    {
      "epoch": 8.136749267578125e-06,
      "step": 8532,
      "training_step_time": 9.574023485183716
    },
    {
      "epoch": 8.13770294189453e-06,
      "model_forward_time": 1.707453727722168,
      "step": 8533
    },
    {
      "epoch": 8.13770294189453e-06,
      "step": 8533,
      "training_step_time": 8.49921202659607
    },
    {
      "epoch": 8.138656616210938e-06,
      "model_forward_time": 1.7104966640472412,
      "step": 8534
    },
    {
      "epoch": 8.138656616210938e-06,
      "step": 8534,
      "training_step_time": 8.39160418510437
    },
    {
      "epoch": 8.139610290527344e-06,
      "model_forward_time": 1.7059972286224365,
      "step": 8535
    },
    {
      "epoch": 8.139610290527344e-06,
      "step": 8535,
      "training_step_time": 7.135359048843384
    },
    {
      "epoch": 8.14056396484375e-06,
      "model_forward_time": 1.7050502300262451,
      "step": 8536
    },
    {
      "epoch": 8.14056396484375e-06,
      "step": 8536,
      "training_step_time": 7.28769326210022
    },
    {
      "epoch": 8.141517639160157e-06,
      "model_forward_time": 1.7114036083221436,
      "step": 8537
    },
    {
      "epoch": 8.141517639160157e-06,
      "step": 8537,
      "training_step_time": 6.688667058944702
    },
    {
      "epoch": 8.142471313476563e-06,
      "model_forward_time": 1.7228643894195557,
      "step": 8538
    },
    {
      "epoch": 8.142471313476563e-06,
      "step": 8538,
      "training_step_time": 7.501028299331665
    },
    {
      "epoch": 8.143424987792968e-06,
      "model_forward_time": 1.7015862464904785,
      "step": 8539
    },
    {
      "epoch": 8.143424987792968e-06,
      "step": 8539,
      "training_step_time": 8.402193307876587
    },
    {
      "epoch": 8.144378662109374e-06,
      "grad_norm": 0.3660831153392792,
      "learning_rate": 9.965797495030429e-06,
      "loss": 0.0727,
      "step": 8540
    },
    {
      "epoch": 8.144378662109374e-06,
      "model_forward_time": 1.3344273567199707,
      "step": 8540
    },
    {
      "dynamics_loss_avg": 0.0716440998017788,
      "epoch": 8.144378662109374e-06,
      "step": 8540
    },
    {
      "action_loss_avg": 0.004977544210851193,
      "epoch": 8.144378662109374e-06,
      "step": 8540
    },
    {
      "epoch": 8.144378662109374e-06,
      "step": 8540,
      "training_step_time": 8.646025657653809
    },
    {
      "epoch": 8.145332336425782e-06,
      "model_forward_time": 2.225827693939209,
      "step": 8541
    },
    {
      "epoch": 8.145332336425782e-06,
      "step": 8541,
      "training_step_time": 7.575725555419922
    },
    {
      "epoch": 8.146286010742187e-06,
      "model_forward_time": 1.707747459411621,
      "step": 8542
    },
    {
      "epoch": 8.146286010742187e-06,
      "step": 8542,
      "training_step_time": 9.893603086471558
    },
    {
      "epoch": 8.147239685058593e-06,
      "model_forward_time": 1.7094154357910156,
      "step": 8543
    },
    {
      "epoch": 8.147239685058593e-06,
      "step": 8543,
      "training_step_time": 8.542019128799438
    },
    {
      "epoch": 8.148193359375e-06,
      "model_forward_time": 1.7147369384765625,
      "step": 8544
    },
    {
      "epoch": 8.148193359375e-06,
      "step": 8544,
      "training_step_time": 7.2791571617126465
    },
    {
      "epoch": 8.149147033691406e-06,
      "model_forward_time": 1.6887640953063965,
      "step": 8545
    },
    {
      "epoch": 8.149147033691406e-06,
      "step": 8545,
      "training_step_time": 7.30138087272644
    },
    {
      "epoch": 8.150100708007812e-06,
      "model_forward_time": 1.707895040512085,
      "step": 8546
    },
    {
      "epoch": 8.150100708007812e-06,
      "step": 8546,
      "training_step_time": 6.71902060508728
    },
    {
      "epoch": 8.15105438232422e-06,
      "model_forward_time": 1.7092835903167725,
      "step": 8547
    },
    {
      "epoch": 8.15105438232422e-06,
      "step": 8547,
      "training_step_time": 7.613835573196411
    },
    {
      "epoch": 8.152008056640625e-06,
      "model_forward_time": 1.7043447494506836,
      "step": 8548
    },
    {
      "epoch": 8.152008056640625e-06,
      "step": 8548,
      "training_step_time": 7.8946287631988525
    },
    {
      "epoch": 8.15296173095703e-06,
      "model_forward_time": 1.7248282432556152,
      "step": 8549
    },
    {
      "epoch": 8.15296173095703e-06,
      "step": 8549,
      "training_step_time": 7.72529673576355
    },
    {
      "epoch": 8.153915405273438e-06,
      "grad_norm": 0.32683178782463074,
      "learning_rate": 9.96560415508286e-06,
      "loss": 0.0793,
      "step": 8550
    },
    {
      "epoch": 8.153915405273438e-06,
      "model_forward_time": 2.2271690368652344,
      "step": 8550
    },
    {
      "dynamics_loss_avg": 0.05948252435773611,
      "epoch": 8.153915405273438e-06,
      "step": 8550
    },
    {
      "action_loss_avg": 0.006004818202927709,
      "epoch": 8.153915405273438e-06,
      "step": 8550
    },
    {
      "epoch": 8.153915405273438e-06,
      "step": 8550,
      "training_step_time": 9.397913932800293
    },
    {
      "epoch": 8.154869079589844e-06,
      "model_forward_time": 1.70701265335083,
      "step": 8551
    },
    {
      "epoch": 8.154869079589844e-06,
      "step": 8551,
      "training_step_time": 9.298019170761108
    },
    {
      "epoch": 8.15582275390625e-06,
      "model_forward_time": 1.6955037117004395,
      "step": 8552
    },
    {
      "epoch": 8.15582275390625e-06,
      "step": 8552,
      "training_step_time": 7.583195686340332
    },
    {
      "epoch": 8.156776428222657e-06,
      "model_forward_time": 1.708467960357666,
      "step": 8553
    },
    {
      "epoch": 8.156776428222657e-06,
      "step": 8553,
      "training_step_time": 7.540464401245117
    },
    {
      "epoch": 8.157730102539063e-06,
      "model_forward_time": 1.7065441608428955,
      "step": 8554
    },
    {
      "epoch": 8.157730102539063e-06,
      "step": 8554,
      "training_step_time": 8.046841621398926
    },
    {
      "epoch": 8.158683776855468e-06,
      "model_forward_time": 1.714245319366455,
      "step": 8555
    },
    {
      "epoch": 8.158683776855468e-06,
      "step": 8555,
      "training_step_time": 6.6756274700164795
    },
    {
      "epoch": 8.159637451171874e-06,
      "model_forward_time": 1.7169713973999023,
      "step": 8556
    },
    {
      "epoch": 8.159637451171874e-06,
      "step": 8556,
      "training_step_time": 8.48601508140564
    },
    {
      "epoch": 8.160591125488282e-06,
      "model_forward_time": 1.7060880661010742,
      "step": 8557
    },
    {
      "epoch": 8.160591125488282e-06,
      "step": 8557,
      "training_step_time": 7.843711614608765
    },
    {
      "epoch": 8.161544799804687e-06,
      "model_forward_time": 1.701864242553711,
      "step": 8558
    },
    {
      "epoch": 8.161544799804687e-06,
      "step": 8558,
      "training_step_time": 8.36216139793396
    },
    {
      "epoch": 8.162498474121093e-06,
      "model_forward_time": 2.234434127807617,
      "step": 8559
    },
    {
      "epoch": 8.162498474121093e-06,
      "step": 8559,
      "training_step_time": 10.003956317901611
    },
    {
      "epoch": 8.1634521484375e-06,
      "grad_norm": 0.3070042133331299,
      "learning_rate": 9.965410272104286e-06,
      "loss": 0.0771,
      "step": 8560
    },
    {
      "epoch": 8.1634521484375e-06,
      "model_forward_time": 1.6968750953674316,
      "step": 8560
    },
    {
      "dynamics_loss_avg": 0.06311405319720506,
      "epoch": 8.1634521484375e-06,
      "step": 8560
    },
    {
      "action_loss_avg": 0.006569653726182878,
      "epoch": 8.1634521484375e-06,
      "step": 8560
    },
    {
      "epoch": 8.1634521484375e-06,
      "step": 8560,
      "training_step_time": 9.076989889144897
    },
    {
      "epoch": 8.164405822753906e-06,
      "model_forward_time": 1.7044098377227783,
      "step": 8561
    },
    {
      "epoch": 8.164405822753906e-06,
      "step": 8561,
      "training_step_time": 7.712201118469238
    },
    {
      "epoch": 8.165359497070312e-06,
      "model_forward_time": 1.7077693939208984,
      "step": 8562
    },
    {
      "epoch": 8.165359497070312e-06,
      "step": 8562,
      "training_step_time": 7.642567873001099
    },
    {
      "epoch": 8.16631317138672e-06,
      "model_forward_time": 1.7105026245117188,
      "step": 8563
    },
    {
      "epoch": 8.16631317138672e-06,
      "step": 8563,
      "training_step_time": 7.254883527755737
    },
    {
      "epoch": 8.167266845703125e-06,
      "model_forward_time": 1.7033216953277588,
      "step": 8564
    },
    {
      "epoch": 8.167266845703125e-06,
      "step": 8564,
      "training_step_time": 7.26870584487915
    },
    {
      "epoch": 8.16822052001953e-06,
      "model_forward_time": 1.7075800895690918,
      "step": 8565
    },
    {
      "epoch": 8.16822052001953e-06,
      "step": 8565,
      "training_step_time": 7.590197801589966
    },
    {
      "epoch": 8.169174194335938e-06,
      "model_forward_time": 1.7084205150604248,
      "step": 8566
    },
    {
      "epoch": 8.169174194335938e-06,
      "step": 8566,
      "training_step_time": 7.737293004989624
    },
    {
      "epoch": 8.170127868652344e-06,
      "model_forward_time": 1.7068886756896973,
      "step": 8567
    },
    {
      "epoch": 8.170127868652344e-06,
      "step": 8567,
      "training_step_time": 7.577822208404541
    },
    {
      "epoch": 8.17108154296875e-06,
      "model_forward_time": 1.7485840320587158,
      "step": 8568
    },
    {
      "epoch": 8.17108154296875e-06,
      "step": 8568,
      "training_step_time": 7.641266345977783
    },
    {
      "epoch": 8.172035217285157e-06,
      "model_forward_time": 2.2301576137542725,
      "step": 8569
    },
    {
      "epoch": 8.172035217285157e-06,
      "step": 8569,
      "training_step_time": 9.2794771194458
    },
    {
      "epoch": 8.172988891601563e-06,
      "grad_norm": 0.32104554772377014,
      "learning_rate": 9.96521584611591e-06,
      "loss": 0.0759,
      "step": 8570
    },
    {
      "epoch": 8.172988891601563e-06,
      "model_forward_time": 1.7029869556427002,
      "step": 8570
    },
    {
      "dynamics_loss_avg": 0.0701593279838562,
      "epoch": 8.172988891601563e-06,
      "step": 8570
    },
    {
      "action_loss_avg": 0.007946506747975946,
      "epoch": 8.172988891601563e-06,
      "step": 8570
    },
    {
      "epoch": 8.172988891601563e-06,
      "step": 8570,
      "training_step_time": 7.680753946304321
    },
    {
      "epoch": 8.173942565917968e-06,
      "model_forward_time": 1.331174373626709,
      "step": 8571
    },
    {
      "epoch": 8.173942565917968e-06,
      "step": 8571,
      "training_step_time": 8.611690282821655
    },
    {
      "epoch": 8.174896240234374e-06,
      "model_forward_time": 1.7132065296173096,
      "step": 8572
    },
    {
      "epoch": 8.174896240234374e-06,
      "step": 8572,
      "training_step_time": 7.218803882598877
    },
    {
      "epoch": 8.175849914550782e-06,
      "model_forward_time": 1.7051091194152832,
      "step": 8573
    },
    {
      "epoch": 8.175849914550782e-06,
      "step": 8573,
      "training_step_time": 7.316360712051392
    },
    {
      "epoch": 8.176803588867187e-06,
      "model_forward_time": 1.7188537120819092,
      "step": 8574
    },
    {
      "epoch": 8.176803588867187e-06,
      "step": 8574,
      "training_step_time": 7.5778772830963135
    },
    {
      "epoch": 8.177757263183593e-06,
      "model_forward_time": 1.7069671154022217,
      "step": 8575
    },
    {
      "epoch": 8.177757263183593e-06,
      "step": 8575,
      "training_step_time": 7.640504837036133
    },
    {
      "epoch": 8.1787109375e-06,
      "model_forward_time": 1.7140452861785889,
      "step": 8576
    },
    {
      "epoch": 8.1787109375e-06,
      "step": 8576,
      "training_step_time": 8.286577224731445
    },
    {
      "epoch": 8.179664611816406e-06,
      "model_forward_time": 1.7608718872070312,
      "step": 8577
    },
    {
      "epoch": 8.179664611816406e-06,
      "step": 8577,
      "training_step_time": 9.633976697921753
    },
    {
      "epoch": 8.180618286132812e-06,
      "model_forward_time": 2.2436201572418213,
      "step": 8578
    },
    {
      "epoch": 8.180618286132812e-06,
      "step": 8578,
      "training_step_time": 8.59032678604126
    },
    {
      "epoch": 8.18157196044922e-06,
      "model_forward_time": 1.7071309089660645,
      "step": 8579
    },
    {
      "epoch": 8.18157196044922e-06,
      "step": 8579,
      "training_step_time": 7.66151237487793
    },
    {
      "epoch": 8.182525634765625e-06,
      "grad_norm": 0.15507566928863525,
      "learning_rate": 9.965020877138995e-06,
      "loss": 0.074,
      "step": 8580
    },
    {
      "epoch": 8.182525634765625e-06,
      "model_forward_time": 1.7025518417358398,
      "step": 8580
    },
    {
      "dynamics_loss_avg": 0.07394606694579124,
      "epoch": 8.182525634765625e-06,
      "step": 8580
    },
    {
      "action_loss_avg": 0.007055127620697021,
      "epoch": 8.182525634765625e-06,
      "step": 8580
    },
    {
      "epoch": 8.182525634765625e-06,
      "step": 8580,
      "training_step_time": 7.5317301750183105
    },
    {
      "epoch": 8.18347930908203e-06,
      "model_forward_time": 1.7074236869812012,
      "step": 8581
    },
    {
      "epoch": 8.18347930908203e-06,
      "step": 8581,
      "training_step_time": 7.23112940788269
    },
    {
      "epoch": 8.184432983398438e-06,
      "model_forward_time": 1.7143609523773193,
      "step": 8582
    },
    {
      "epoch": 8.184432983398438e-06,
      "step": 8582,
      "training_step_time": 7.310236930847168
    },
    {
      "epoch": 8.185386657714844e-06,
      "model_forward_time": 1.7056021690368652,
      "step": 8583
    },
    {
      "epoch": 8.185386657714844e-06,
      "step": 8583,
      "training_step_time": 7.594367504119873
    },
    {
      "epoch": 8.18634033203125e-06,
      "model_forward_time": 1.707658052444458,
      "step": 8584
    },
    {
      "epoch": 8.18634033203125e-06,
      "step": 8584,
      "training_step_time": 8.108532667160034
    },
    {
      "epoch": 8.187294006347657e-06,
      "model_forward_time": 1.7173254489898682,
      "step": 8585
    },
    {
      "epoch": 8.187294006347657e-06,
      "step": 8585,
      "training_step_time": 9.152489185333252
    },
    {
      "epoch": 8.188247680664063e-06,
      "model_forward_time": 1.7217204570770264,
      "step": 8586
    },
    {
      "epoch": 8.188247680664063e-06,
      "step": 8586,
      "training_step_time": 9.025114297866821
    },
    {
      "epoch": 8.189201354980469e-06,
      "model_forward_time": 2.2236838340759277,
      "step": 8587
    },
    {
      "epoch": 8.189201354980469e-06,
      "step": 8587,
      "training_step_time": 8.615234851837158
    },
    {
      "epoch": 8.190155029296874e-06,
      "model_forward_time": 1.709216594696045,
      "step": 8588
    },
    {
      "epoch": 8.190155029296874e-06,
      "step": 8588,
      "training_step_time": 7.635369300842285
    },
    {
      "epoch": 8.191108703613282e-06,
      "model_forward_time": 1.7172303199768066,
      "step": 8589
    },
    {
      "epoch": 8.191108703613282e-06,
      "step": 8589,
      "training_step_time": 7.450956106185913
    },
    {
      "epoch": 8.192062377929687e-06,
      "grad_norm": 0.2036259025335312,
      "learning_rate": 9.964825365194862e-06,
      "loss": 0.0672,
      "step": 8590
    },
    {
      "epoch": 8.192062377929687e-06,
      "model_forward_time": 1.7014031410217285,
      "step": 8590
    },
    {
      "dynamics_loss_avg": 0.06722703091800213,
      "epoch": 8.192062377929687e-06,
      "step": 8590
    },
    {
      "action_loss_avg": 0.007144044060260058,
      "epoch": 8.192062377929687e-06,
      "step": 8590
    },
    {
      "epoch": 8.192062377929687e-06,
      "step": 8590,
      "training_step_time": 7.596695899963379
    },
    {
      "epoch": 8.193016052246093e-06,
      "model_forward_time": 1.7087600231170654,
      "step": 8591
    },
    {
      "epoch": 8.193016052246093e-06,
      "step": 8591,
      "training_step_time": 7.307135105133057
    },
    {
      "epoch": 8.1939697265625e-06,
      "model_forward_time": 1.708263874053955,
      "step": 8592
    },
    {
      "epoch": 8.1939697265625e-06,
      "step": 8592,
      "training_step_time": 7.547429084777832
    },
    {
      "epoch": 8.194923400878906e-06,
      "model_forward_time": 1.7104756832122803,
      "step": 8593
    },
    {
      "epoch": 8.194923400878906e-06,
      "step": 8593,
      "training_step_time": 7.359705448150635
    },
    {
      "epoch": 8.195877075195312e-06,
      "model_forward_time": 1.7073945999145508,
      "step": 8594
    },
    {
      "epoch": 8.195877075195312e-06,
      "step": 8594,
      "training_step_time": 7.833214044570923
    },
    {
      "epoch": 8.19683074951172e-06,
      "model_forward_time": 1.715686321258545,
      "step": 8595
    },
    {
      "epoch": 8.19683074951172e-06,
      "step": 8595,
      "training_step_time": 7.640513181686401
    },
    {
      "epoch": 8.197784423828125e-06,
      "model_forward_time": 2.251530170440674,
      "step": 8596
    },
    {
      "epoch": 8.197784423828125e-06,
      "step": 8596,
      "training_step_time": 8.601435899734497
    },
    {
      "epoch": 8.19873809814453e-06,
      "model_forward_time": 1.6992368698120117,
      "step": 8597
    },
    {
      "epoch": 8.19873809814453e-06,
      "step": 8597,
      "training_step_time": 8.491812467575073
    },
    {
      "epoch": 8.199691772460938e-06,
      "model_forward_time": 1.7175424098968506,
      "step": 8598
    },
    {
      "epoch": 8.199691772460938e-06,
      "step": 8598,
      "training_step_time": 7.532785892486572
    },
    {
      "epoch": 8.200645446777344e-06,
      "model_forward_time": 1.7069718837738037,
      "step": 8599
    },
    {
      "epoch": 8.200645446777344e-06,
      "step": 8599,
      "training_step_time": 7.629709482192993
    },
    {
      "epoch": 8.20159912109375e-06,
      "grad_norm": 0.3513069450855255,
      "learning_rate": 9.96462931030489e-06,
      "loss": 0.0701,
      "step": 8600
    },
    {
      "epoch": 8.20159912109375e-06,
      "model_forward_time": 1.7043297290802002,
      "step": 8600
    },
    {
      "dynamics_loss_avg": 0.08433218114078045,
      "epoch": 8.20159912109375e-06,
      "step": 8600
    },
    {
      "action_loss_avg": 0.0059445192920975385,
      "epoch": 8.20159912109375e-06,
      "step": 8600
    },
    {
      "epoch": 8.20159912109375e-06,
      "step": 8600,
      "training_step_time": 7.333399295806885
    },
    {
      "epoch": 8.202552795410157e-06,
      "model_forward_time": 1.7149930000305176,
      "step": 8601
    },
    {
      "epoch": 8.202552795410157e-06,
      "step": 8601,
      "training_step_time": 7.602685928344727
    },
    {
      "epoch": 8.203506469726563e-06,
      "model_forward_time": 1.7204258441925049,
      "step": 8602
    },
    {
      "epoch": 8.203506469726563e-06,
      "step": 8602,
      "training_step_time": 7.352893829345703
    },
    {
      "epoch": 8.204460144042969e-06,
      "model_forward_time": 1.7046034336090088,
      "step": 8603
    },
    {
      "epoch": 8.204460144042969e-06,
      "step": 8603,
      "training_step_time": 7.947206497192383
    },
    {
      "epoch": 8.205413818359374e-06,
      "model_forward_time": 1.712589979171753,
      "step": 8604
    },
    {
      "epoch": 8.205413818359374e-06,
      "step": 8604,
      "training_step_time": 7.742477893829346
    },
    {
      "epoch": 8.206367492675782e-06,
      "model_forward_time": 2.264848470687866,
      "step": 8605
    },
    {
      "epoch": 8.206367492675782e-06,
      "step": 8605,
      "training_step_time": 8.57405400276184
    },
    {
      "epoch": 8.207321166992187e-06,
      "model_forward_time": 1.7048530578613281,
      "step": 8606
    },
    {
      "epoch": 8.207321166992187e-06,
      "step": 8606,
      "training_step_time": 9.387409687042236
    },
    {
      "epoch": 8.208274841308593e-06,
      "model_forward_time": 1.7105047702789307,
      "step": 8607
    },
    {
      "epoch": 8.208274841308593e-06,
      "step": 8607,
      "training_step_time": 7.615547180175781
    },
    {
      "epoch": 8.209228515625e-06,
      "model_forward_time": 1.7085707187652588,
      "step": 8608
    },
    {
      "epoch": 8.209228515625e-06,
      "step": 8608,
      "training_step_time": 7.751096963882446
    },
    {
      "epoch": 8.210182189941406e-06,
      "model_forward_time": 1.7173805236816406,
      "step": 8609
    },
    {
      "epoch": 8.210182189941406e-06,
      "step": 8609,
      "training_step_time": 7.3750598430633545
    },
    {
      "epoch": 8.211135864257812e-06,
      "grad_norm": 0.2347135692834854,
      "learning_rate": 9.96443271249052e-06,
      "loss": 0.0701,
      "step": 8610
    },
    {
      "epoch": 8.211135864257812e-06,
      "model_forward_time": 1.7095866203308105,
      "step": 8610
    },
    {
      "dynamics_loss_avg": 0.07224603798240423,
      "epoch": 8.211135864257812e-06,
      "step": 8610
    },
    {
      "action_loss_avg": 0.0059069649316370485,
      "epoch": 8.211135864257812e-06,
      "step": 8610
    },
    {
      "epoch": 8.211135864257812e-06,
      "step": 8610,
      "training_step_time": 7.629502296447754
    },
    {
      "epoch": 8.21208953857422e-06,
      "model_forward_time": 1.711418867111206,
      "step": 8611
    },
    {
      "epoch": 8.21208953857422e-06,
      "step": 8611,
      "training_step_time": 7.389590263366699
    },
    {
      "epoch": 8.213043212890625e-06,
      "model_forward_time": 1.7128841876983643,
      "step": 8612
    },
    {
      "epoch": 8.213043212890625e-06,
      "step": 8612,
      "training_step_time": 8.743996620178223
    },
    {
      "epoch": 8.21399688720703e-06,
      "model_forward_time": 1.7128896713256836,
      "step": 8613
    },
    {
      "epoch": 8.21399688720703e-06,
      "step": 8613,
      "training_step_time": 7.635029077529907
    },
    {
      "epoch": 8.214950561523438e-06,
      "model_forward_time": 2.2687668800354004,
      "step": 8614
    },
    {
      "epoch": 8.214950561523438e-06,
      "step": 8614,
      "training_step_time": 9.231207370758057
    },
    {
      "epoch": 8.215904235839844e-06,
      "model_forward_time": 1.7109918594360352,
      "step": 8615
    },
    {
      "epoch": 8.215904235839844e-06,
      "step": 8615,
      "training_step_time": 8.40099811553955
    },
    {
      "epoch": 8.21685791015625e-06,
      "model_forward_time": 1.7113761901855469,
      "step": 8616
    },
    {
      "epoch": 8.21685791015625e-06,
      "step": 8616,
      "training_step_time": 8.958930969238281
    },
    {
      "epoch": 8.217811584472657e-06,
      "model_forward_time": 1.7145311832427979,
      "step": 8617
    },
    {
      "epoch": 8.217811584472657e-06,
      "step": 8617,
      "training_step_time": 8.232502698898315
    },
    {
      "epoch": 8.218765258789063e-06,
      "model_forward_time": 1.7118079662322998,
      "step": 8618
    },
    {
      "epoch": 8.218765258789063e-06,
      "step": 8618,
      "training_step_time": 7.248353481292725
    },
    {
      "epoch": 8.219718933105469e-06,
      "model_forward_time": 1.7120802402496338,
      "step": 8619
    },
    {
      "epoch": 8.219718933105469e-06,
      "step": 8619,
      "training_step_time": 7.569648742675781
    },
    {
      "epoch": 8.220672607421874e-06,
      "grad_norm": 0.2943930923938751,
      "learning_rate": 9.964235571773254e-06,
      "loss": 0.0695,
      "step": 8620
    },
    {
      "epoch": 8.220672607421874e-06,
      "model_forward_time": 1.3343391418457031,
      "step": 8620
    },
    {
      "dynamics_loss_avg": 0.05139335542917252,
      "epoch": 8.220672607421874e-06,
      "step": 8620
    },
    {
      "action_loss_avg": 0.002937520982231945,
      "epoch": 8.220672607421874e-06,
      "step": 8620
    },
    {
      "epoch": 8.220672607421874e-06,
      "step": 8620,
      "training_step_time": 7.174274444580078
    },
    {
      "epoch": 8.221626281738282e-06,
      "model_forward_time": 1.7188704013824463,
      "step": 8621
    },
    {
      "epoch": 8.221626281738282e-06,
      "step": 8621,
      "training_step_time": 7.6460230350494385
    },
    {
      "epoch": 8.222579956054687e-06,
      "model_forward_time": 1.709777593612671,
      "step": 8622
    },
    {
      "epoch": 8.222579956054687e-06,
      "step": 8622,
      "training_step_time": 7.759237766265869
    },
    {
      "epoch": 8.223533630371093e-06,
      "model_forward_time": 1.7351257801055908,
      "step": 8623
    },
    {
      "epoch": 8.223533630371093e-06,
      "step": 8623,
      "training_step_time": 7.654349088668823
    },
    {
      "epoch": 8.2244873046875e-06,
      "model_forward_time": 2.252077579498291,
      "step": 8624
    },
    {
      "epoch": 8.2244873046875e-06,
      "step": 8624,
      "training_step_time": 8.530332088470459
    },
    {
      "epoch": 8.225440979003906e-06,
      "model_forward_time": 1.7139852046966553,
      "step": 8625
    },
    {
      "epoch": 8.225440979003906e-06,
      "step": 8625,
      "training_step_time": 8.537190437316895
    },
    {
      "epoch": 8.226394653320312e-06,
      "model_forward_time": 1.716506004333496,
      "step": 8626
    },
    {
      "epoch": 8.226394653320312e-06,
      "step": 8626,
      "training_step_time": 7.612112045288086
    },
    {
      "epoch": 8.22734832763672e-06,
      "model_forward_time": 1.7140552997589111,
      "step": 8627
    },
    {
      "epoch": 8.22734832763672e-06,
      "step": 8627,
      "training_step_time": 7.469396114349365
    },
    {
      "epoch": 8.228302001953125e-06,
      "model_forward_time": 1.7204627990722656,
      "step": 8628
    },
    {
      "epoch": 8.228302001953125e-06,
      "step": 8628,
      "training_step_time": 8.42619800567627
    },
    {
      "epoch": 8.22925567626953e-06,
      "model_forward_time": 1.7130827903747559,
      "step": 8629
    },
    {
      "epoch": 8.22925567626953e-06,
      "step": 8629,
      "training_step_time": 7.279614210128784
    },
    {
      "epoch": 8.230209350585938e-06,
      "grad_norm": 0.15060870349407196,
      "learning_rate": 9.96403788817465e-06,
      "loss": 0.0673,
      "step": 8630
    },
    {
      "epoch": 8.230209350585938e-06,
      "model_forward_time": 1.7073726654052734,
      "step": 8630
    },
    {
      "dynamics_loss_avg": 0.060955815203487876,
      "epoch": 8.230209350585938e-06,
      "step": 8630
    },
    {
      "action_loss_avg": 0.004713751684175804,
      "epoch": 8.230209350585938e-06,
      "step": 8630
    },
    {
      "epoch": 8.230209350585938e-06,
      "step": 8630,
      "training_step_time": 7.59375262260437
    },
    {
      "epoch": 8.231163024902344e-06,
      "model_forward_time": 1.7019474506378174,
      "step": 8631
    },
    {
      "epoch": 8.231163024902344e-06,
      "step": 8631,
      "training_step_time": 7.941296577453613
    },
    {
      "epoch": 8.23211669921875e-06,
      "model_forward_time": 1.7109472751617432,
      "step": 8632
    },
    {
      "epoch": 8.23211669921875e-06,
      "step": 8632,
      "training_step_time": 9.27334451675415
    },
    {
      "epoch": 8.233070373535155e-06,
      "model_forward_time": 2.2326340675354004,
      "step": 8633
    },
    {
      "epoch": 8.233070373535155e-06,
      "step": 8633,
      "training_step_time": 7.592108488082886
    },
    {
      "epoch": 8.234024047851563e-06,
      "model_forward_time": 1.7055261135101318,
      "step": 8634
    },
    {
      "epoch": 8.234024047851563e-06,
      "step": 8634,
      "training_step_time": 8.59540581703186
    },
    {
      "epoch": 8.234977722167969e-06,
      "model_forward_time": 1.7042734622955322,
      "step": 8635
    },
    {
      "epoch": 8.234977722167969e-06,
      "step": 8635,
      "training_step_time": 8.746954441070557
    },
    {
      "epoch": 8.235931396484374e-06,
      "model_forward_time": 1.707484483718872,
      "step": 8636
    },
    {
      "epoch": 8.235931396484374e-06,
      "step": 8636,
      "training_step_time": 7.3969902992248535
    },
    {
      "epoch": 8.236885070800782e-06,
      "model_forward_time": 1.7071599960327148,
      "step": 8637
    },
    {
      "epoch": 8.236885070800782e-06,
      "step": 8637,
      "training_step_time": 7.553100824356079
    },
    {
      "epoch": 8.237838745117187e-06,
      "model_forward_time": 1.7048885822296143,
      "step": 8638
    },
    {
      "epoch": 8.237838745117187e-06,
      "step": 8638,
      "training_step_time": 7.29445481300354
    },
    {
      "epoch": 8.238792419433593e-06,
      "model_forward_time": 1.709467887878418,
      "step": 8639
    },
    {
      "epoch": 8.238792419433593e-06,
      "step": 8639,
      "training_step_time": 7.311827898025513
    },
    {
      "epoch": 8.23974609375e-06,
      "grad_norm": 0.18893709778785706,
      "learning_rate": 9.963839661716326e-06,
      "loss": 0.0656,
      "step": 8640
    },
    {
      "epoch": 8.23974609375e-06,
      "model_forward_time": 1.7002456188201904,
      "step": 8640
    },
    {
      "dynamics_loss_avg": 0.07852263022214175,
      "epoch": 8.23974609375e-06,
      "step": 8640
    },
    {
      "action_loss_avg": 0.006344418250955641,
      "epoch": 8.23974609375e-06,
      "step": 8640
    },
    {
      "epoch": 8.23974609375e-06,
      "step": 8640,
      "training_step_time": 8.825709819793701
    },
    {
      "epoch": 8.240699768066406e-06,
      "model_forward_time": 1.7095634937286377,
      "step": 8641
    },
    {
      "epoch": 8.240699768066406e-06,
      "step": 8641,
      "training_step_time": 8.554042339324951
    },
    {
      "epoch": 8.241653442382812e-06,
      "model_forward_time": 2.265179395675659,
      "step": 8642
    },
    {
      "epoch": 8.241653442382812e-06,
      "step": 8642,
      "training_step_time": 7.625093221664429
    },
    {
      "epoch": 8.24260711669922e-06,
      "model_forward_time": 1.7003371715545654,
      "step": 8643
    },
    {
      "epoch": 8.24260711669922e-06,
      "step": 8643,
      "training_step_time": 7.600854396820068
    },
    {
      "epoch": 8.243560791015625e-06,
      "model_forward_time": 1.7102088928222656,
      "step": 8644
    },
    {
      "epoch": 8.243560791015625e-06,
      "step": 8644,
      "training_step_time": 7.751784801483154
    },
    {
      "epoch": 8.24451446533203e-06,
      "model_forward_time": 1.7061614990234375,
      "step": 8645
    },
    {
      "epoch": 8.24451446533203e-06,
      "step": 8645,
      "training_step_time": 7.554480791091919
    },
    {
      "epoch": 8.245468139648438e-06,
      "model_forward_time": 1.7043240070343018,
      "step": 8646
    },
    {
      "epoch": 8.245468139648438e-06,
      "step": 8646,
      "training_step_time": 7.571614742279053
    },
    {
      "epoch": 8.246421813964844e-06,
      "model_forward_time": 1.7083160877227783,
      "step": 8647
    },
    {
      "epoch": 8.246421813964844e-06,
      "step": 8647,
      "training_step_time": 7.329873323440552
    },
    {
      "epoch": 8.24737548828125e-06,
      "model_forward_time": 1.7095739841461182,
      "step": 8648
    },
    {
      "epoch": 8.24737548828125e-06,
      "step": 8648,
      "training_step_time": 7.3451478481292725
    },
    {
      "epoch": 8.248329162597655e-06,
      "model_forward_time": 1.708070993423462,
      "step": 8649
    },
    {
      "epoch": 8.248329162597655e-06,
      "step": 8649,
      "training_step_time": 7.8592870235443115
    },
    {
      "epoch": 8.249282836914063e-06,
      "grad_norm": 0.2375025898218155,
      "learning_rate": 9.963640892419958e-06,
      "loss": 0.0659,
      "step": 8650
    },
    {
      "epoch": 8.249282836914063e-06,
      "model_forward_time": 1.6997814178466797,
      "step": 8650
    },
    {
      "dynamics_loss_avg": 0.07163699865341186,
      "epoch": 8.249282836914063e-06,
      "step": 8650
    },
    {
      "action_loss_avg": 0.006017350079491735,
      "epoch": 8.249282836914063e-06,
      "step": 8650
    },
    {
      "epoch": 8.249282836914063e-06,
      "step": 8650,
      "training_step_time": 8.369411945343018
    },
    {
      "epoch": 8.250236511230469e-06,
      "model_forward_time": 2.253758192062378,
      "step": 8651
    },
    {
      "epoch": 8.250236511230469e-06,
      "step": 8651,
      "training_step_time": 7.680807113647461
    },
    {
      "epoch": 8.251190185546874e-06,
      "model_forward_time": 1.7056384086608887,
      "step": 8652
    },
    {
      "epoch": 8.251190185546874e-06,
      "step": 8652,
      "training_step_time": 7.711229085922241
    },
    {
      "epoch": 8.252143859863282e-06,
      "model_forward_time": 1.711125373840332,
      "step": 8653
    },
    {
      "epoch": 8.252143859863282e-06,
      "step": 8653,
      "training_step_time": 8.422541379928589
    },
    {
      "epoch": 8.253097534179687e-06,
      "model_forward_time": 1.7038233280181885,
      "step": 8654
    },
    {
      "epoch": 8.253097534179687e-06,
      "step": 8654,
      "training_step_time": 7.676645278930664
    },
    {
      "epoch": 8.254051208496093e-06,
      "model_forward_time": 1.7100846767425537,
      "step": 8655
    },
    {
      "epoch": 8.254051208496093e-06,
      "step": 8655,
      "training_step_time": 7.668241500854492
    },
    {
      "epoch": 8.2550048828125e-06,
      "model_forward_time": 1.7096939086914062,
      "step": 8656
    },
    {
      "epoch": 8.2550048828125e-06,
      "step": 8656,
      "training_step_time": 7.426929712295532
    },
    {
      "epoch": 8.255958557128906e-06,
      "model_forward_time": 1.741088628768921,
      "step": 8657
    },
    {
      "epoch": 8.255958557128906e-06,
      "step": 8657,
      "training_step_time": 6.678088903427124
    },
    {
      "epoch": 8.256912231445312e-06,
      "model_forward_time": 1.7086186408996582,
      "step": 8658
    },
    {
      "epoch": 8.256912231445312e-06,
      "step": 8658,
      "training_step_time": 8.195300340652466
    },
    {
      "epoch": 8.25786590576172e-06,
      "model_forward_time": 1.7144811153411865,
      "step": 8659
    },
    {
      "epoch": 8.25786590576172e-06,
      "step": 8659,
      "training_step_time": 7.6284754276275635
    },
    {
      "epoch": 8.258819580078125e-06,
      "grad_norm": 0.19895368814468384,
      "learning_rate": 9.963441580307286e-06,
      "loss": 0.0659,
      "step": 8660
    },
    {
      "epoch": 8.258819580078125e-06,
      "model_forward_time": 1.7134392261505127,
      "step": 8660
    },
    {
      "dynamics_loss_avg": 0.05078704208135605,
      "epoch": 8.258819580078125e-06,
      "step": 8660
    },
    {
      "action_loss_avg": 0.004995001951465383,
      "epoch": 8.258819580078125e-06,
      "step": 8660
    },
    {
      "epoch": 8.258819580078125e-06,
      "step": 8660,
      "training_step_time": 8.974977731704712
    },
    {
      "epoch": 8.259773254394531e-06,
      "model_forward_time": 1.7085344791412354,
      "step": 8661
    },
    {
      "epoch": 8.259773254394531e-06,
      "step": 8661,
      "training_step_time": 8.222326755523682
    },
    {
      "epoch": 8.260726928710938e-06,
      "model_forward_time": 1.7236053943634033,
      "step": 8662
    },
    {
      "epoch": 8.260726928710938e-06,
      "step": 8662,
      "training_step_time": 9.423786640167236
    },
    {
      "epoch": 8.261680603027344e-06,
      "model_forward_time": 1.6984338760375977,
      "step": 8663
    },
    {
      "epoch": 8.261680603027344e-06,
      "step": 8663,
      "training_step_time": 7.575145244598389
    },
    {
      "epoch": 8.26263427734375e-06,
      "model_forward_time": 1.7069408893585205,
      "step": 8664
    },
    {
      "epoch": 8.26263427734375e-06,
      "step": 8664,
      "training_step_time": 7.646724224090576
    },
    {
      "epoch": 8.263587951660155e-06,
      "model_forward_time": 1.711268424987793,
      "step": 8665
    },
    {
      "epoch": 8.263587951660155e-06,
      "step": 8665,
      "training_step_time": 8.161643028259277
    },
    {
      "epoch": 8.264541625976563e-06,
      "model_forward_time": 1.7076060771942139,
      "step": 8666
    },
    {
      "epoch": 8.264541625976563e-06,
      "step": 8666,
      "training_step_time": 7.288566589355469
    },
    {
      "epoch": 8.265495300292969e-06,
      "model_forward_time": 1.7049152851104736,
      "step": 8667
    },
    {
      "epoch": 8.265495300292969e-06,
      "step": 8667,
      "training_step_time": 7.326345920562744
    },
    {
      "epoch": 8.266448974609374e-06,
      "model_forward_time": 1.7069604396820068,
      "step": 8668
    },
    {
      "epoch": 8.266448974609374e-06,
      "step": 8668,
      "training_step_time": 8.405809164047241
    },
    {
      "epoch": 8.267402648925782e-06,
      "model_forward_time": 1.7107839584350586,
      "step": 8669
    },
    {
      "epoch": 8.267402648925782e-06,
      "step": 8669,
      "training_step_time": 7.6982293128967285
    },
    {
      "epoch": 8.268356323242187e-06,
      "grad_norm": 0.22596406936645508,
      "learning_rate": 9.963241725400104e-06,
      "loss": 0.0731,
      "step": 8670
    },
    {
      "epoch": 8.268356323242187e-06,
      "model_forward_time": 2.2445082664489746,
      "step": 8670
    },
    {
      "dynamics_loss_avg": 0.057808947004377845,
      "epoch": 8.268356323242187e-06,
      "step": 8670
    },
    {
      "action_loss_avg": 0.003893931373022497,
      "epoch": 8.268356323242187e-06,
      "step": 8670
    },
    {
      "epoch": 8.268356323242187e-06,
      "step": 8670,
      "training_step_time": 8.60569429397583
    },
    {
      "epoch": 8.269309997558593e-06,
      "model_forward_time": 1.7061872482299805,
      "step": 8671
    },
    {
      "epoch": 8.269309997558593e-06,
      "step": 8671,
      "training_step_time": 7.60166072845459
    },
    {
      "epoch": 8.270263671875e-06,
      "model_forward_time": 1.7097833156585693,
      "step": 8672
    },
    {
      "epoch": 8.270263671875e-06,
      "step": 8672,
      "training_step_time": 7.788331031799316
    },
    {
      "epoch": 8.271217346191406e-06,
      "model_forward_time": 1.7087366580963135,
      "step": 8673
    },
    {
      "epoch": 8.271217346191406e-06,
      "step": 8673,
      "training_step_time": 7.632584095001221
    },
    {
      "epoch": 8.272171020507812e-06,
      "model_forward_time": 1.7088358402252197,
      "step": 8674
    },
    {
      "epoch": 8.272171020507812e-06,
      "step": 8674,
      "training_step_time": 7.310920000076294
    },
    {
      "epoch": 8.27312469482422e-06,
      "model_forward_time": 1.7056264877319336,
      "step": 8675
    },
    {
      "epoch": 8.27312469482422e-06,
      "step": 8675,
      "training_step_time": 7.298510313034058
    },
    {
      "epoch": 8.274078369140625e-06,
      "model_forward_time": 1.7038183212280273,
      "step": 8676
    },
    {
      "epoch": 8.274078369140625e-06,
      "step": 8676,
      "training_step_time": 7.299745559692383
    },
    {
      "epoch": 8.275032043457031e-06,
      "model_forward_time": 1.7100207805633545,
      "step": 8677
    },
    {
      "epoch": 8.275032043457031e-06,
      "step": 8677,
      "training_step_time": 7.698534250259399
    },
    {
      "epoch": 8.275985717773438e-06,
      "model_forward_time": 1.7344896793365479,
      "step": 8678
    },
    {
      "epoch": 8.275985717773438e-06,
      "step": 8678,
      "training_step_time": 8.64256477355957
    },
    {
      "epoch": 8.276939392089844e-06,
      "model_forward_time": 2.223397731781006,
      "step": 8679
    },
    {
      "epoch": 8.276939392089844e-06,
      "step": 8679,
      "training_step_time": 7.665038108825684
    },
    {
      "epoch": 8.27789306640625e-06,
      "grad_norm": 0.2169642299413681,
      "learning_rate": 9.963041327720271e-06,
      "loss": 0.0664,
      "step": 8680
    },
    {
      "epoch": 8.27789306640625e-06,
      "model_forward_time": 1.7028450965881348,
      "step": 8680
    },
    {
      "dynamics_loss_avg": 0.07538287155330181,
      "epoch": 8.27789306640625e-06,
      "step": 8680
    },
    {
      "action_loss_avg": 0.006285606464371085,
      "epoch": 8.27789306640625e-06,
      "step": 8680
    },
    {
      "epoch": 8.27789306640625e-06,
      "step": 8680,
      "training_step_time": 7.549192428588867
    },
    {
      "epoch": 8.278846740722655e-06,
      "model_forward_time": 1.7092366218566895,
      "step": 8681
    },
    {
      "epoch": 8.278846740722655e-06,
      "step": 8681,
      "training_step_time": 7.746798753738403
    },
    {
      "epoch": 8.279800415039063e-06,
      "model_forward_time": 1.7068703174591064,
      "step": 8682
    },
    {
      "epoch": 8.279800415039063e-06,
      "step": 8682,
      "training_step_time": 7.657717943191528
    },
    {
      "epoch": 8.280754089355469e-06,
      "model_forward_time": 1.689652442932129,
      "step": 8683
    },
    {
      "epoch": 8.280754089355469e-06,
      "step": 8683,
      "training_step_time": 7.28563117980957
    },
    {
      "epoch": 8.281707763671874e-06,
      "model_forward_time": 1.713853359222412,
      "step": 8684
    },
    {
      "epoch": 8.281707763671874e-06,
      "step": 8684,
      "training_step_time": 7.280892848968506
    },
    {
      "epoch": 8.282661437988282e-06,
      "model_forward_time": 1.7040231227874756,
      "step": 8685
    },
    {
      "epoch": 8.282661437988282e-06,
      "step": 8685,
      "training_step_time": 8.056743621826172
    },
    {
      "epoch": 8.283615112304687e-06,
      "model_forward_time": 1.7102701663970947,
      "step": 8686
    },
    {
      "epoch": 8.283615112304687e-06,
      "step": 8686,
      "training_step_time": 7.642844915390015
    },
    {
      "epoch": 8.284568786621093e-06,
      "model_forward_time": 1.707963228225708,
      "step": 8687
    },
    {
      "epoch": 8.284568786621093e-06,
      "step": 8687,
      "training_step_time": 7.705017805099487
    },
    {
      "epoch": 8.2855224609375e-06,
      "model_forward_time": 2.2438690662384033,
      "step": 8688
    },
    {
      "epoch": 8.2855224609375e-06,
      "step": 8688,
      "training_step_time": 7.650360107421875
    },
    {
      "epoch": 8.286476135253906e-06,
      "model_forward_time": 1.7119958400726318,
      "step": 8689
    },
    {
      "epoch": 8.286476135253906e-06,
      "step": 8689,
      "training_step_time": 7.5623908042907715
    },
    {
      "epoch": 8.287429809570312e-06,
      "grad_norm": 0.24241487681865692,
      "learning_rate": 9.962840387289698e-06,
      "loss": 0.0694,
      "step": 8690
    },
    {
      "epoch": 8.287429809570312e-06,
      "model_forward_time": 1.6892282962799072,
      "step": 8690
    },
    {
      "dynamics_loss_avg": 0.050015689246356486,
      "epoch": 8.287429809570312e-06,
      "step": 8690
    },
    {
      "action_loss_avg": 0.004520086012780667,
      "epoch": 8.287429809570312e-06,
      "step": 8690
    },
    {
      "epoch": 8.287429809570312e-06,
      "step": 8690,
      "training_step_time": 8.341779708862305
    },
    {
      "epoch": 8.28838348388672e-06,
      "model_forward_time": 1.7039599418640137,
      "step": 8691
    },
    {
      "epoch": 8.28838348388672e-06,
      "step": 8691,
      "training_step_time": 7.624390602111816
    },
    {
      "epoch": 8.289337158203125e-06,
      "model_forward_time": 1.7129547595977783,
      "step": 8692
    },
    {
      "epoch": 8.289337158203125e-06,
      "step": 8692,
      "training_step_time": 7.573487281799316
    },
    {
      "epoch": 8.290290832519531e-06,
      "model_forward_time": 1.7037010192871094,
      "step": 8693
    },
    {
      "epoch": 8.290290832519531e-06,
      "step": 8693,
      "training_step_time": 7.340410470962524
    },
    {
      "epoch": 8.291244506835938e-06,
      "model_forward_time": 1.7030391693115234,
      "step": 8694
    },
    {
      "epoch": 8.291244506835938e-06,
      "step": 8694,
      "training_step_time": 7.552358865737915
    },
    {
      "epoch": 8.292198181152344e-06,
      "model_forward_time": 1.7089612483978271,
      "step": 8695
    },
    {
      "epoch": 8.292198181152344e-06,
      "step": 8695,
      "training_step_time": 7.703831672668457
    },
    {
      "epoch": 8.29315185546875e-06,
      "model_forward_time": 1.7117509841918945,
      "step": 8696
    },
    {
      "epoch": 8.29315185546875e-06,
      "step": 8696,
      "training_step_time": 8.576643228530884
    },
    {
      "epoch": 8.294105529785155e-06,
      "model_forward_time": 2.2482805252075195,
      "step": 8697
    },
    {
      "epoch": 8.294105529785155e-06,
      "step": 8697,
      "training_step_time": 8.690858364105225
    },
    {
      "epoch": 8.295059204101563e-06,
      "model_forward_time": 1.73219895362854,
      "step": 8698
    },
    {
      "epoch": 8.295059204101563e-06,
      "step": 8698,
      "training_step_time": 8.64374589920044
    },
    {
      "epoch": 8.296012878417969e-06,
      "model_forward_time": 1.7117838859558105,
      "step": 8699
    },
    {
      "epoch": 8.296012878417969e-06,
      "step": 8699,
      "training_step_time": 9.321927547454834
    },
    {
      "epoch": 8.296966552734374e-06,
      "grad_norm": 0.2657139301300049,
      "learning_rate": 9.962638904130363e-06,
      "loss": 0.0701,
      "step": 8700
    },
    {
      "epoch": 8.296966552734374e-06,
      "model_forward_time": 1.7051794528961182,
      "step": 8700
    },
    {
      "dynamics_loss_avg": 0.053219825588166715,
      "epoch": 8.296966552734374e-06,
      "step": 8700
    },
    {
      "action_loss_avg": 0.005653303442522883,
      "epoch": 8.296966552734374e-06,
      "step": 8700
    },
    {
      "epoch": 8.296966552734374e-06,
      "step": 8700,
      "training_step_time": 8.204649686813354
    },
    {
      "epoch": 8.297920227050782e-06,
      "model_forward_time": 1.707829236984253,
      "step": 8701
    },
    {
      "epoch": 8.297920227050782e-06,
      "step": 8701,
      "training_step_time": 7.609025478363037
    },
    {
      "epoch": 8.298873901367187e-06,
      "model_forward_time": 1.7160325050354004,
      "step": 8702
    },
    {
      "epoch": 8.298873901367187e-06,
      "step": 8702,
      "training_step_time": 8.073464155197144
    },
    {
      "epoch": 8.299827575683593e-06,
      "model_forward_time": 1.7051808834075928,
      "step": 8703
    },
    {
      "epoch": 8.299827575683593e-06,
      "step": 8703,
      "training_step_time": 7.193800926208496
    },
    {
      "epoch": 8.30078125e-06,
      "model_forward_time": 1.7087559700012207,
      "step": 8704
    },
    {
      "epoch": 8.30078125e-06,
      "step": 8704,
      "training_step_time": 8.509324312210083
    },
    {
      "epoch": 8.301734924316406e-06,
      "model_forward_time": 1.7031347751617432,
      "step": 8705
    },
    {
      "epoch": 8.301734924316406e-06,
      "step": 8705,
      "training_step_time": 8.642297983169556
    },
    {
      "epoch": 8.302688598632812e-06,
      "model_forward_time": 2.255006790161133,
      "step": 8706
    },
    {
      "epoch": 8.302688598632812e-06,
      "step": 8706,
      "training_step_time": 9.284549474716187
    },
    {
      "epoch": 8.30364227294922e-06,
      "model_forward_time": 1.7434566020965576,
      "step": 8707
    },
    {
      "epoch": 8.30364227294922e-06,
      "step": 8707,
      "training_step_time": 9.026825428009033
    },
    {
      "epoch": 8.304595947265625e-06,
      "model_forward_time": 1.6892821788787842,
      "step": 8708
    },
    {
      "epoch": 8.304595947265625e-06,
      "step": 8708,
      "training_step_time": 8.94466757774353
    },
    {
      "epoch": 8.305549621582031e-06,
      "model_forward_time": 1.6939797401428223,
      "step": 8709
    },
    {
      "epoch": 8.305549621582031e-06,
      "step": 8709,
      "training_step_time": 7.7487633228302
    },
    {
      "epoch": 8.306503295898438e-06,
      "grad_norm": 0.2050478756427765,
      "learning_rate": 9.962436878264299e-06,
      "loss": 0.0686,
      "step": 8710
    },
    {
      "epoch": 8.306503295898438e-06,
      "model_forward_time": 1.692765474319458,
      "step": 8710
    },
    {
      "dynamics_loss_avg": 0.06513884104788303,
      "epoch": 8.306503295898438e-06,
      "step": 8710
    },
    {
      "action_loss_avg": 0.0036730578722199425,
      "epoch": 8.306503295898438e-06,
      "step": 8710
    },
    {
      "epoch": 8.306503295898438e-06,
      "step": 8710,
      "training_step_time": 7.643671035766602
    },
    {
      "epoch": 8.307456970214844e-06,
      "model_forward_time": 1.3229148387908936,
      "step": 8711
    },
    {
      "epoch": 8.307456970214844e-06,
      "step": 8711,
      "training_step_time": 7.330743312835693
    },
    {
      "epoch": 8.30841064453125e-06,
      "model_forward_time": 1.6864688396453857,
      "step": 8712
    },
    {
      "epoch": 8.30841064453125e-06,
      "step": 8712,
      "training_step_time": 7.1979429721832275
    },
    {
      "epoch": 8.309364318847655e-06,
      "model_forward_time": 1.6946816444396973,
      "step": 8713
    },
    {
      "epoch": 8.309364318847655e-06,
      "step": 8713,
      "training_step_time": 7.705923795700073
    },
    {
      "epoch": 8.310317993164063e-06,
      "model_forward_time": 1.6941120624542236,
      "step": 8714
    },
    {
      "epoch": 8.310317993164063e-06,
      "step": 8714,
      "training_step_time": 8.297393321990967
    },
    {
      "epoch": 8.311271667480469e-06,
      "model_forward_time": 1.6929142475128174,
      "step": 8715
    },
    {
      "epoch": 8.311271667480469e-06,
      "step": 8715,
      "training_step_time": 8.518322467803955
    },
    {
      "epoch": 8.312225341796874e-06,
      "model_forward_time": 2.2876529693603516,
      "step": 8716
    },
    {
      "epoch": 8.312225341796874e-06,
      "step": 8716,
      "training_step_time": 7.69410252571106
    },
    {
      "epoch": 8.313179016113282e-06,
      "model_forward_time": 1.6924302577972412,
      "step": 8717
    },
    {
      "epoch": 8.313179016113282e-06,
      "step": 8717,
      "training_step_time": 8.481466054916382
    },
    {
      "epoch": 8.314132690429687e-06,
      "model_forward_time": 1.6935701370239258,
      "step": 8718
    },
    {
      "epoch": 8.314132690429687e-06,
      "step": 8718,
      "training_step_time": 8.643076658248901
    },
    {
      "epoch": 8.315086364746093e-06,
      "model_forward_time": 1.6941449642181396,
      "step": 8719
    },
    {
      "epoch": 8.315086364746093e-06,
      "step": 8719,
      "training_step_time": 7.580073595046997
    },
    {
      "epoch": 8.3160400390625e-06,
      "grad_norm": 0.2122563123703003,
      "learning_rate": 9.962234309713598e-06,
      "loss": 0.074,
      "step": 8720
    },
    {
      "epoch": 8.3160400390625e-06,
      "model_forward_time": 1.681994915008545,
      "step": 8720
    },
    {
      "dynamics_loss_avg": 0.05685020722448826,
      "epoch": 8.3160400390625e-06,
      "step": 8720
    },
    {
      "action_loss_avg": 0.00315225530648604,
      "epoch": 8.3160400390625e-06,
      "step": 8720
    },
    {
      "epoch": 8.3160400390625e-06,
      "step": 8720,
      "training_step_time": 7.1928486824035645
    },
    {
      "epoch": 8.316993713378906e-06,
      "model_forward_time": 1.6931941509246826,
      "step": 8721
    },
    {
      "epoch": 8.316993713378906e-06,
      "step": 8721,
      "training_step_time": 7.288041591644287
    },
    {
      "epoch": 8.317947387695312e-06,
      "model_forward_time": 1.6893506050109863,
      "step": 8722
    },
    {
      "epoch": 8.317947387695312e-06,
      "step": 8722,
      "training_step_time": 7.662540912628174
    },
    {
      "epoch": 8.31890106201172e-06,
      "model_forward_time": 1.696937084197998,
      "step": 8723
    },
    {
      "epoch": 8.31890106201172e-06,
      "step": 8723,
      "training_step_time": 8.266721963882446
    },
    {
      "epoch": 8.319854736328125e-06,
      "model_forward_time": 1.690185546875,
      "step": 8724
    },
    {
      "epoch": 8.319854736328125e-06,
      "step": 8724,
      "training_step_time": 8.582882404327393
    },
    {
      "epoch": 8.320808410644531e-06,
      "model_forward_time": 2.290379285812378,
      "step": 8725
    },
    {
      "epoch": 8.320808410644531e-06,
      "step": 8725,
      "training_step_time": 7.646812438964844
    },
    {
      "epoch": 8.321762084960938e-06,
      "model_forward_time": 1.6928257942199707,
      "step": 8726
    },
    {
      "epoch": 8.321762084960938e-06,
      "step": 8726,
      "training_step_time": 8.680265188217163
    },
    {
      "epoch": 8.322715759277344e-06,
      "model_forward_time": 1.6955993175506592,
      "step": 8727
    },
    {
      "epoch": 8.322715759277344e-06,
      "step": 8727,
      "training_step_time": 7.77530837059021
    },
    {
      "epoch": 8.32366943359375e-06,
      "model_forward_time": 1.6957998275756836,
      "step": 8728
    },
    {
      "epoch": 8.32366943359375e-06,
      "step": 8728,
      "training_step_time": 7.534575462341309
    },
    {
      "epoch": 8.324623107910155e-06,
      "model_forward_time": 1.689220666885376,
      "step": 8729
    },
    {
      "epoch": 8.324623107910155e-06,
      "step": 8729,
      "training_step_time": 7.305870294570923
    },
    {
      "epoch": 8.325576782226563e-06,
      "grad_norm": 0.14178389310836792,
      "learning_rate": 9.962031198500415e-06,
      "loss": 0.0748,
      "step": 8730
    },
    {
      "epoch": 8.325576782226563e-06,
      "model_forward_time": 1.6874034404754639,
      "step": 8730
    },
    {
      "dynamics_loss_avg": 0.05740875508636236,
      "epoch": 8.325576782226563e-06,
      "step": 8730
    },
    {
      "action_loss_avg": 0.003949396661482751,
      "epoch": 8.325576782226563e-06,
      "step": 8730
    },
    {
      "epoch": 8.325576782226563e-06,
      "step": 8730,
      "training_step_time": 7.347469806671143
    },
    {
      "epoch": 8.326530456542969e-06,
      "model_forward_time": 1.6903536319732666,
      "step": 8731
    },
    {
      "epoch": 8.326530456542969e-06,
      "step": 8731,
      "training_step_time": 7.796961784362793
    },
    {
      "epoch": 8.327484130859374e-06,
      "model_forward_time": 1.3191537857055664,
      "step": 8732
    },
    {
      "epoch": 8.327484130859374e-06,
      "step": 8732,
      "training_step_time": 7.561288118362427
    },
    {
      "epoch": 8.328437805175782e-06,
      "model_forward_time": 1.316173791885376,
      "step": 8733
    },
    {
      "epoch": 8.328437805175782e-06,
      "step": 8733,
      "training_step_time": 7.658651113510132
    },
    {
      "epoch": 8.329391479492187e-06,
      "model_forward_time": 2.243837833404541,
      "step": 8734
    },
    {
      "epoch": 8.329391479492187e-06,
      "step": 8734,
      "training_step_time": 8.443914651870728
    },
    {
      "epoch": 8.330345153808593e-06,
      "model_forward_time": 1.6881663799285889,
      "step": 8735
    },
    {
      "epoch": 8.330345153808593e-06,
      "step": 8735,
      "training_step_time": 9.299238681793213
    },
    {
      "epoch": 8.331298828125e-06,
      "model_forward_time": 1.6881115436553955,
      "step": 8736
    },
    {
      "epoch": 8.331298828125e-06,
      "step": 8736,
      "training_step_time": 7.719208240509033
    },
    {
      "epoch": 8.332252502441406e-06,
      "model_forward_time": 1.6996188163757324,
      "step": 8737
    },
    {
      "epoch": 8.332252502441406e-06,
      "step": 8737,
      "training_step_time": 8.562152624130249
    },
    {
      "epoch": 8.333206176757812e-06,
      "model_forward_time": 1.6930644512176514,
      "step": 8738
    },
    {
      "epoch": 8.333206176757812e-06,
      "step": 8738,
      "training_step_time": 7.335312366485596
    },
    {
      "epoch": 8.33415985107422e-06,
      "model_forward_time": 1.6934690475463867,
      "step": 8739
    },
    {
      "epoch": 8.33415985107422e-06,
      "step": 8739,
      "training_step_time": 7.268270969390869
    },
    {
      "epoch": 8.335113525390625e-06,
      "grad_norm": 0.20381668210029602,
      "learning_rate": 9.961827544646958e-06,
      "loss": 0.0712,
      "step": 8740
    },
    {
      "epoch": 8.335113525390625e-06,
      "model_forward_time": 1.6857967376708984,
      "step": 8740
    },
    {
      "dynamics_loss_avg": 0.05916092824190855,
      "epoch": 8.335113525390625e-06,
      "step": 8740
    },
    {
      "action_loss_avg": 0.0032175528991501777,
      "epoch": 8.335113525390625e-06,
      "step": 8740
    },
    {
      "epoch": 8.335113525390625e-06,
      "step": 8740,
      "training_step_time": 7.778114080429077
    },
    {
      "epoch": 8.336067199707031e-06,
      "model_forward_time": 1.6944198608398438,
      "step": 8741
    },
    {
      "epoch": 8.336067199707031e-06,
      "step": 8741,
      "training_step_time": 7.603937864303589
    },
    {
      "epoch": 8.337020874023438e-06,
      "model_forward_time": 1.693385362625122,
      "step": 8742
    },
    {
      "epoch": 8.337020874023438e-06,
      "step": 8742,
      "training_step_time": 9.192168235778809
    },
    {
      "epoch": 8.337974548339844e-06,
      "model_forward_time": 2.257859468460083,
      "step": 8743
    },
    {
      "epoch": 8.337974548339844e-06,
      "step": 8743,
      "training_step_time": 7.584212779998779
    },
    {
      "epoch": 8.33892822265625e-06,
      "model_forward_time": 1.6942901611328125,
      "step": 8744
    },
    {
      "epoch": 8.33892822265625e-06,
      "step": 8744,
      "training_step_time": 7.60042142868042
    },
    {
      "epoch": 8.339881896972656e-06,
      "model_forward_time": 1.699049711227417,
      "step": 8745
    },
    {
      "epoch": 8.339881896972656e-06,
      "step": 8745,
      "training_step_time": 7.718552589416504
    },
    {
      "epoch": 8.340835571289063e-06,
      "model_forward_time": 1.698392391204834,
      "step": 8746
    },
    {
      "epoch": 8.340835571289063e-06,
      "step": 8746,
      "training_step_time": 8.230488538742065
    },
    {
      "epoch": 8.341789245605469e-06,
      "model_forward_time": 1.6901662349700928,
      "step": 8747
    },
    {
      "epoch": 8.341789245605469e-06,
      "step": 8747,
      "training_step_time": 7.5550432205200195
    },
    {
      "epoch": 8.342742919921874e-06,
      "model_forward_time": 1.6963527202606201,
      "step": 8748
    },
    {
      "epoch": 8.342742919921874e-06,
      "step": 8748,
      "training_step_time": 7.278423309326172
    },
    {
      "epoch": 8.343696594238282e-06,
      "model_forward_time": 1.6907105445861816,
      "step": 8749
    },
    {
      "epoch": 8.343696594238282e-06,
      "step": 8749,
      "training_step_time": 7.862963676452637
    },
    {
      "epoch": 8.344650268554688e-06,
      "grad_norm": 0.21283107995986938,
      "learning_rate": 9.961623348175503e-06,
      "loss": 0.0726,
      "step": 8750
    },
    {
      "epoch": 8.344650268554688e-06,
      "model_forward_time": 1.687898874282837,
      "step": 8750
    },
    {
      "dynamics_loss_avg": 0.0766485856845975,
      "epoch": 8.344650268554688e-06,
      "step": 8750
    },
    {
      "action_loss_avg": 0.005202477448619902,
      "epoch": 8.344650268554688e-06,
      "step": 8750
    },
    {
      "epoch": 8.344650268554688e-06,
      "step": 8750,
      "training_step_time": 7.622337818145752
    },
    {
      "epoch": 8.345603942871093e-06,
      "model_forward_time": 1.692328691482544,
      "step": 8751
    },
    {
      "epoch": 8.345603942871093e-06,
      "step": 8751,
      "training_step_time": 8.576097965240479
    },
    {
      "epoch": 8.3465576171875e-06,
      "model_forward_time": 2.250758409500122,
      "step": 8752
    },
    {
      "epoch": 8.3465576171875e-06,
      "step": 8752,
      "training_step_time": 7.65474534034729
    },
    {
      "epoch": 8.347511291503906e-06,
      "model_forward_time": 1.6883800029754639,
      "step": 8753
    },
    {
      "epoch": 8.347511291503906e-06,
      "step": 8753,
      "training_step_time": 7.646822452545166
    },
    {
      "epoch": 8.348464965820312e-06,
      "model_forward_time": 1.6882591247558594,
      "step": 8754
    },
    {
      "epoch": 8.348464965820312e-06,
      "step": 8754,
      "training_step_time": 9.430830240249634
    },
    {
      "epoch": 8.34941864013672e-06,
      "model_forward_time": 1.6856796741485596,
      "step": 8755
    },
    {
      "epoch": 8.34941864013672e-06,
      "step": 8755,
      "training_step_time": 7.587213516235352
    },
    {
      "epoch": 8.350372314453125e-06,
      "model_forward_time": 1.6942572593688965,
      "step": 8756
    },
    {
      "epoch": 8.350372314453125e-06,
      "step": 8756,
      "training_step_time": 7.663992643356323
    },
    {
      "epoch": 8.351325988769531e-06,
      "model_forward_time": 1.6838722229003906,
      "step": 8757
    },
    {
      "epoch": 8.351325988769531e-06,
      "step": 8757,
      "training_step_time": 7.290880918502808
    },
    {
      "epoch": 8.352279663085937e-06,
      "model_forward_time": 1.6853952407836914,
      "step": 8758
    },
    {
      "epoch": 8.352279663085937e-06,
      "step": 8758,
      "training_step_time": 7.833398818969727
    },
    {
      "epoch": 8.353233337402344e-06,
      "model_forward_time": 1.6846132278442383,
      "step": 8759
    },
    {
      "epoch": 8.353233337402344e-06,
      "step": 8759,
      "training_step_time": 7.649557113647461
    },
    {
      "epoch": 8.35418701171875e-06,
      "grad_norm": 0.23441898822784424,
      "learning_rate": 9.961418609108377e-06,
      "loss": 0.0737,
      "step": 8760
    },
    {
      "epoch": 8.35418701171875e-06,
      "model_forward_time": 1.6823673248291016,
      "step": 8760
    },
    {
      "dynamics_loss_avg": 0.06394406780600548,
      "epoch": 8.35418701171875e-06,
      "step": 8760
    },
    {
      "action_loss_avg": 0.00477164916228503,
      "epoch": 8.35418701171875e-06,
      "step": 8760
    },
    {
      "epoch": 8.35418701171875e-06,
      "step": 8760,
      "training_step_time": 8.633983612060547
    },
    {
      "epoch": 8.355140686035156e-06,
      "model_forward_time": 2.2203128337860107,
      "step": 8761
    },
    {
      "epoch": 8.355140686035156e-06,
      "step": 8761,
      "training_step_time": 8.393013954162598
    },
    {
      "epoch": 8.356094360351563e-06,
      "model_forward_time": 1.70558762550354,
      "step": 8762
    },
    {
      "epoch": 8.356094360351563e-06,
      "step": 8762,
      "training_step_time": 10.41458249092102
    },
    {
      "epoch": 8.357048034667969e-06,
      "model_forward_time": 1.683269739151001,
      "step": 8763
    },
    {
      "epoch": 8.357048034667969e-06,
      "step": 8763,
      "training_step_time": 7.725538969039917
    },
    {
      "epoch": 8.358001708984374e-06,
      "model_forward_time": 1.6840472221374512,
      "step": 8764
    },
    {
      "epoch": 8.358001708984374e-06,
      "step": 8764,
      "training_step_time": 7.609851121902466
    },
    {
      "epoch": 8.358955383300782e-06,
      "model_forward_time": 1.6798396110534668,
      "step": 8765
    },
    {
      "epoch": 8.358955383300782e-06,
      "step": 8765,
      "training_step_time": 7.677274942398071
    },
    {
      "epoch": 8.359909057617188e-06,
      "model_forward_time": 1.6902570724487305,
      "step": 8766
    },
    {
      "epoch": 8.359909057617188e-06,
      "step": 8766,
      "training_step_time": 7.36738395690918
    },
    {
      "epoch": 8.360862731933593e-06,
      "model_forward_time": 1.684067726135254,
      "step": 8767
    },
    {
      "epoch": 8.360862731933593e-06,
      "step": 8767,
      "training_step_time": 7.429055690765381
    },
    {
      "epoch": 8.36181640625e-06,
      "model_forward_time": 1.687192678451538,
      "step": 8768
    },
    {
      "epoch": 8.36181640625e-06,
      "step": 8768,
      "training_step_time": 8.146961688995361
    },
    {
      "epoch": 8.362770080566406e-06,
      "model_forward_time": 1.6834733486175537,
      "step": 8769
    },
    {
      "epoch": 8.362770080566406e-06,
      "step": 8769,
      "training_step_time": 7.575482606887817
    },
    {
      "epoch": 8.363723754882812e-06,
      "grad_norm": 0.19314247369766235,
      "learning_rate": 9.961213327467971e-06,
      "loss": 0.073,
      "step": 8770
    },
    {
      "epoch": 8.363723754882812e-06,
      "model_forward_time": 1.7021543979644775,
      "step": 8770
    },
    {
      "dynamics_loss_avg": 0.05888779014348984,
      "epoch": 8.363723754882812e-06,
      "step": 8770
    },
    {
      "action_loss_avg": 0.0039953059982508424,
      "epoch": 8.363723754882812e-06,
      "step": 8770
    },
    {
      "epoch": 8.363723754882812e-06,
      "step": 8770,
      "training_step_time": 8.707634925842285
    },
    {
      "epoch": 8.36467742919922e-06,
      "model_forward_time": 1.691115140914917,
      "step": 8771
    },
    {
      "epoch": 8.36467742919922e-06,
      "step": 8771,
      "training_step_time": 7.585117816925049
    },
    {
      "epoch": 8.365631103515625e-06,
      "model_forward_time": 1.6911282539367676,
      "step": 8772
    },
    {
      "epoch": 8.365631103515625e-06,
      "step": 8772,
      "training_step_time": 8.257271528244019
    },
    {
      "epoch": 8.366584777832031e-06,
      "model_forward_time": 1.685286283493042,
      "step": 8773
    },
    {
      "epoch": 8.366584777832031e-06,
      "step": 8773,
      "training_step_time": 9.516912460327148
    },
    {
      "epoch": 8.367538452148437e-06,
      "model_forward_time": 1.6888267993927002,
      "step": 8774
    },
    {
      "epoch": 8.367538452148437e-06,
      "step": 8774,
      "training_step_time": 7.56488561630249
    },
    {
      "epoch": 8.368492126464844e-06,
      "model_forward_time": 1.6853899955749512,
      "step": 8775
    },
    {
      "epoch": 8.368492126464844e-06,
      "step": 8775,
      "training_step_time": 7.324544191360474
    },
    {
      "epoch": 8.36944580078125e-06,
      "model_forward_time": 1.6778171062469482,
      "step": 8776
    },
    {
      "epoch": 8.36944580078125e-06,
      "step": 8776,
      "training_step_time": 7.390974760055542
    },
    {
      "epoch": 8.370399475097656e-06,
      "model_forward_time": 1.6822962760925293,
      "step": 8777
    },
    {
      "epoch": 8.370399475097656e-06,
      "step": 8777,
      "training_step_time": 8.285694122314453
    },
    {
      "epoch": 8.371353149414063e-06,
      "model_forward_time": 1.684617280960083,
      "step": 8778
    },
    {
      "epoch": 8.371353149414063e-06,
      "step": 8778,
      "training_step_time": 9.343012809753418
    },
    {
      "epoch": 8.372306823730469e-06,
      "model_forward_time": 1.6973955631256104,
      "step": 8779
    },
    {
      "epoch": 8.372306823730469e-06,
      "step": 8779,
      "training_step_time": 9.221049547195435
    },
    {
      "epoch": 8.373260498046874e-06,
      "grad_norm": 0.286590576171875,
      "learning_rate": 9.961007503276736e-06,
      "loss": 0.0761,
      "step": 8780
    },
    {
      "epoch": 8.373260498046874e-06,
      "model_forward_time": 2.226912260055542,
      "step": 8780
    },
    {
      "dynamics_loss_avg": 0.07635643556714058,
      "epoch": 8.373260498046874e-06,
      "step": 8780
    },
    {
      "action_loss_avg": 0.003920178068801761,
      "epoch": 8.373260498046874e-06,
      "step": 8780
    },
    {
      "epoch": 8.373260498046874e-06,
      "step": 8780,
      "training_step_time": 9.616639852523804
    },
    {
      "epoch": 8.374214172363282e-06,
      "model_forward_time": 1.6827447414398193,
      "step": 8781
    },
    {
      "epoch": 8.374214172363282e-06,
      "step": 8781,
      "training_step_time": 7.716583967208862
    },
    {
      "epoch": 8.375167846679688e-06,
      "model_forward_time": 1.6879355907440186,
      "step": 8782
    },
    {
      "epoch": 8.375167846679688e-06,
      "step": 8782,
      "training_step_time": 9.069823026657104
    },
    {
      "epoch": 8.376121520996093e-06,
      "model_forward_time": 1.68961763381958,
      "step": 8783
    },
    {
      "epoch": 8.376121520996093e-06,
      "step": 8783,
      "training_step_time": 7.539945840835571
    },
    {
      "epoch": 8.3770751953125e-06,
      "model_forward_time": 1.6834406852722168,
      "step": 8784
    },
    {
      "epoch": 8.3770751953125e-06,
      "step": 8784,
      "training_step_time": 7.323462724685669
    },
    {
      "epoch": 8.378028869628906e-06,
      "model_forward_time": 1.6924183368682861,
      "step": 8785
    },
    {
      "epoch": 8.378028869628906e-06,
      "step": 8785,
      "training_step_time": 7.381408452987671
    },
    {
      "epoch": 8.378982543945312e-06,
      "model_forward_time": 1.6851253509521484,
      "step": 8786
    },
    {
      "epoch": 8.378982543945312e-06,
      "step": 8786,
      "training_step_time": 7.762801647186279
    },
    {
      "epoch": 8.37993621826172e-06,
      "model_forward_time": 1.6900010108947754,
      "step": 8787
    },
    {
      "epoch": 8.37993621826172e-06,
      "step": 8787,
      "training_step_time": 8.281470775604248
    },
    {
      "epoch": 8.380889892578125e-06,
      "model_forward_time": 1.6977953910827637,
      "step": 8788
    },
    {
      "epoch": 8.380889892578125e-06,
      "step": 8788,
      "training_step_time": 8.408416271209717
    },
    {
      "epoch": 8.381843566894531e-06,
      "model_forward_time": 2.2460291385650635,
      "step": 8789
    },
    {
      "epoch": 8.381843566894531e-06,
      "step": 8789,
      "training_step_time": 7.617318868637085
    },
    {
      "epoch": 8.382797241210937e-06,
      "grad_norm": 0.20117142796516418,
      "learning_rate": 9.960801136557179e-06,
      "loss": 0.0746,
      "step": 8790
    },
    {
      "epoch": 8.382797241210937e-06,
      "model_forward_time": 1.681370735168457,
      "step": 8790
    },
    {
      "dynamics_loss_avg": 0.07292396184056997,
      "epoch": 8.382797241210937e-06,
      "step": 8790
    },
    {
      "action_loss_avg": 0.0050489978981204334,
      "epoch": 8.382797241210937e-06,
      "step": 8790
    },
    {
      "epoch": 8.382797241210937e-06,
      "step": 8790,
      "training_step_time": 7.836398601531982
    },
    {
      "epoch": 8.383750915527344e-06,
      "model_forward_time": 1.6839241981506348,
      "step": 8791
    },
    {
      "epoch": 8.383750915527344e-06,
      "step": 8791,
      "training_step_time": 8.322461366653442
    },
    {
      "epoch": 8.38470458984375e-06,
      "model_forward_time": 1.6901967525482178,
      "step": 8792
    },
    {
      "epoch": 8.38470458984375e-06,
      "step": 8792,
      "training_step_time": 7.476099967956543
    },
    {
      "epoch": 8.385658264160156e-06,
      "model_forward_time": 1.689432144165039,
      "step": 8793
    },
    {
      "epoch": 8.385658264160156e-06,
      "step": 8793,
      "training_step_time": 8.243163347244263
    },
    {
      "epoch": 8.386611938476563e-06,
      "model_forward_time": 1.684542179107666,
      "step": 8794
    },
    {
      "epoch": 8.386611938476563e-06,
      "step": 8794,
      "training_step_time": 7.372443914413452
    },
    {
      "epoch": 8.387565612792969e-06,
      "model_forward_time": 1.6919691562652588,
      "step": 8795
    },
    {
      "epoch": 8.387565612792969e-06,
      "step": 8795,
      "training_step_time": 7.894670009613037
    },
    {
      "epoch": 8.388519287109374e-06,
      "model_forward_time": 1.6918790340423584,
      "step": 8796
    },
    {
      "epoch": 8.388519287109374e-06,
      "step": 8796,
      "training_step_time": 9.852510452270508
    },
    {
      "epoch": 8.389472961425782e-06,
      "model_forward_time": 1.7162458896636963,
      "step": 8797
    },
    {
      "epoch": 8.389472961425782e-06,
      "step": 8797,
      "training_step_time": 7.574050426483154
    },
    {
      "epoch": 8.390426635742188e-06,
      "model_forward_time": 2.231067419052124,
      "step": 8798
    },
    {
      "epoch": 8.390426635742188e-06,
      "step": 8798,
      "training_step_time": 8.214789152145386
    },
    {
      "epoch": 8.391380310058593e-06,
      "model_forward_time": 1.7080941200256348,
      "step": 8799
    },
    {
      "epoch": 8.391380310058593e-06,
      "step": 8799,
      "training_step_time": 7.72266149520874
    },
    {
      "epoch": 8.392333984375e-06,
      "grad_norm": 0.3088889718055725,
      "learning_rate": 9.960594227331867e-06,
      "loss": 0.0744,
      "step": 8800
    },
    {
      "epoch": 8.392333984375e-06,
      "model_forward_time": 1.6951484680175781,
      "step": 8800
    },
    {
      "dynamics_loss_avg": 0.0727343237027526,
      "epoch": 8.392333984375e-06,
      "step": 8800
    },
    {
      "action_loss_avg": 0.005142632964998483,
      "epoch": 8.392333984375e-06,
      "step": 8800
    },
    {
      "epoch": 8.392333984375e-06,
      "step": 8800,
      "training_step_time": 8.071415424346924
    },
    {
      "epoch": 8.393287658691406e-06,
      "model_forward_time": 1.697056770324707,
      "step": 8801
    },
    {
      "epoch": 8.393287658691406e-06,
      "step": 8801,
      "training_step_time": 8.111206531524658
    },
    {
      "epoch": 8.394241333007812e-06,
      "model_forward_time": 1.6895709037780762,
      "step": 8802
    },
    {
      "epoch": 8.394241333007812e-06,
      "step": 8802,
      "training_step_time": 7.629852533340454
    },
    {
      "epoch": 8.39519500732422e-06,
      "model_forward_time": 1.6906764507293701,
      "step": 8803
    },
    {
      "epoch": 8.39519500732422e-06,
      "step": 8803,
      "training_step_time": 8.255627393722534
    },
    {
      "epoch": 8.396148681640625e-06,
      "model_forward_time": 1.6951706409454346,
      "step": 8804
    },
    {
      "epoch": 8.396148681640625e-06,
      "step": 8804,
      "training_step_time": 7.750741958618164
    },
    {
      "epoch": 8.397102355957031e-06,
      "model_forward_time": 1.6983232498168945,
      "step": 8805
    },
    {
      "epoch": 8.397102355957031e-06,
      "step": 8805,
      "training_step_time": 8.488816976547241
    },
    {
      "epoch": 8.398056030273437e-06,
      "model_forward_time": 1.7090535163879395,
      "step": 8806
    },
    {
      "epoch": 8.398056030273437e-06,
      "step": 8806,
      "training_step_time": 8.309066772460938
    },
    {
      "epoch": 8.399009704589844e-06,
      "model_forward_time": 2.246994972229004,
      "step": 8807
    },
    {
      "epoch": 8.399009704589844e-06,
      "step": 8807,
      "training_step_time": 7.689256429672241
    },
    {
      "epoch": 8.39996337890625e-06,
      "model_forward_time": 1.7108526229858398,
      "step": 8808
    },
    {
      "epoch": 8.39996337890625e-06,
      "step": 8808,
      "training_step_time": 8.527936697006226
    },
    {
      "epoch": 8.400917053222656e-06,
      "model_forward_time": 1.6929755210876465,
      "step": 8809
    },
    {
      "epoch": 8.400917053222656e-06,
      "step": 8809,
      "training_step_time": 8.177175998687744
    },
    {
      "epoch": 8.401870727539063e-06,
      "grad_norm": 0.29728150367736816,
      "learning_rate": 9.96038677562343e-06,
      "loss": 0.0772,
      "step": 8810
    },
    {
      "epoch": 8.401870727539063e-06,
      "model_forward_time": 1.6940555572509766,
      "step": 8810
    },
    {
      "dynamics_loss_avg": 0.05777708888053894,
      "epoch": 8.401870727539063e-06,
      "step": 8810
    },
    {
      "action_loss_avg": 0.004823092394508422,
      "epoch": 8.401870727539063e-06,
      "step": 8810
    },
    {
      "epoch": 8.401870727539063e-06,
      "step": 8810,
      "training_step_time": 9.616945743560791
    },
    {
      "epoch": 8.402824401855469e-06,
      "model_forward_time": 1.693873405456543,
      "step": 8811
    },
    {
      "epoch": 8.402824401855469e-06,
      "step": 8811,
      "training_step_time": 7.571848154067993
    },
    {
      "epoch": 8.403778076171874e-06,
      "model_forward_time": 1.6954364776611328,
      "step": 8812
    },
    {
      "epoch": 8.403778076171874e-06,
      "step": 8812,
      "training_step_time": 7.405926942825317
    },
    {
      "epoch": 8.404731750488282e-06,
      "model_forward_time": 1.6874170303344727,
      "step": 8813
    },
    {
      "epoch": 8.404731750488282e-06,
      "step": 8813,
      "training_step_time": 7.735266447067261
    },
    {
      "epoch": 8.405685424804688e-06,
      "model_forward_time": 1.696227788925171,
      "step": 8814
    },
    {
      "epoch": 8.405685424804688e-06,
      "step": 8814,
      "training_step_time": 7.466269493103027
    },
    {
      "epoch": 8.406639099121093e-06,
      "model_forward_time": 1.7135353088378906,
      "step": 8815
    },
    {
      "epoch": 8.406639099121093e-06,
      "step": 8815,
      "training_step_time": 10.717703342437744
    },
    {
      "epoch": 8.4075927734375e-06,
      "model_forward_time": 2.225741386413574,
      "step": 8816
    },
    {
      "epoch": 8.4075927734375e-06,
      "step": 8816,
      "training_step_time": 7.650706052780151
    },
    {
      "epoch": 8.408546447753906e-06,
      "model_forward_time": 1.707866907119751,
      "step": 8817
    },
    {
      "epoch": 8.408546447753906e-06,
      "step": 8817,
      "training_step_time": 10.292691469192505
    },
    {
      "epoch": 8.409500122070312e-06,
      "model_forward_time": 1.6936328411102295,
      "step": 8818
    },
    {
      "epoch": 8.409500122070312e-06,
      "step": 8818,
      "training_step_time": 7.622899532318115
    },
    {
      "epoch": 8.41045379638672e-06,
      "model_forward_time": 1.69417405128479,
      "step": 8819
    },
    {
      "epoch": 8.41045379638672e-06,
      "step": 8819,
      "training_step_time": 7.48557448387146
    },
    {
      "epoch": 8.411407470703125e-06,
      "grad_norm": 0.2095382958650589,
      "learning_rate": 9.96017878145455e-06,
      "loss": 0.0765,
      "step": 8820
    },
    {
      "epoch": 8.411407470703125e-06,
      "model_forward_time": 1.6866931915283203,
      "step": 8820
    },
    {
      "dynamics_loss_avg": 0.06621305476874113,
      "epoch": 8.411407470703125e-06,
      "step": 8820
    },
    {
      "action_loss_avg": 0.008218298549763858,
      "epoch": 8.411407470703125e-06,
      "step": 8820
    },
    {
      "epoch": 8.411407470703125e-06,
      "step": 8820,
      "training_step_time": 7.544226169586182
    },
    {
      "epoch": 8.412361145019531e-06,
      "model_forward_time": 1.6964588165283203,
      "step": 8821
    },
    {
      "epoch": 8.412361145019531e-06,
      "step": 8821,
      "training_step_time": 7.332054376602173
    },
    {
      "epoch": 8.413314819335937e-06,
      "model_forward_time": 1.6978034973144531,
      "step": 8822
    },
    {
      "epoch": 8.413314819335937e-06,
      "step": 8822,
      "training_step_time": 7.21124792098999
    },
    {
      "epoch": 8.414268493652344e-06,
      "model_forward_time": 1.6932647228240967,
      "step": 8823
    },
    {
      "epoch": 8.414268493652344e-06,
      "step": 8823,
      "training_step_time": 7.686049699783325
    },
    {
      "epoch": 8.41522216796875e-06,
      "model_forward_time": 1.713423490524292,
      "step": 8824
    },
    {
      "epoch": 8.41522216796875e-06,
      "step": 8824,
      "training_step_time": 9.250950336456299
    },
    {
      "epoch": 8.416175842285156e-06,
      "model_forward_time": 1.69533109664917,
      "step": 8825
    },
    {
      "epoch": 8.416175842285156e-06,
      "step": 8825,
      "training_step_time": 7.590776205062866
    },
    {
      "epoch": 8.417129516601563e-06,
      "model_forward_time": 2.247422456741333,
      "step": 8826
    },
    {
      "epoch": 8.417129516601563e-06,
      "step": 8826,
      "training_step_time": 7.568186044692993
    },
    {
      "epoch": 8.418083190917969e-06,
      "model_forward_time": 1.69612455368042,
      "step": 8827
    },
    {
      "epoch": 8.418083190917969e-06,
      "step": 8827,
      "training_step_time": 8.629167556762695
    },
    {
      "epoch": 8.419036865234374e-06,
      "model_forward_time": 1.69368314743042,
      "step": 8828
    },
    {
      "epoch": 8.419036865234374e-06,
      "step": 8828,
      "training_step_time": 7.51258659362793
    },
    {
      "epoch": 8.419990539550782e-06,
      "model_forward_time": 1.6931631565093994,
      "step": 8829
    },
    {
      "epoch": 8.419990539550782e-06,
      "step": 8829,
      "training_step_time": 7.5336737632751465
    },
    {
      "epoch": 8.420944213867188e-06,
      "grad_norm": 0.42173880338668823,
      "learning_rate": 9.959970244847978e-06,
      "loss": 0.0741,
      "step": 8830
    },
    {
      "epoch": 8.420944213867188e-06,
      "model_forward_time": 1.6939973831176758,
      "step": 8830
    },
    {
      "dynamics_loss_avg": 0.06731425104662776,
      "epoch": 8.420944213867188e-06,
      "step": 8830
    },
    {
      "action_loss_avg": 0.005694822638179176,
      "epoch": 8.420944213867188e-06,
      "step": 8830
    },
    {
      "epoch": 8.420944213867188e-06,
      "step": 8830,
      "training_step_time": 7.37067985534668
    },
    {
      "epoch": 8.421897888183593e-06,
      "model_forward_time": 1.6976892948150635,
      "step": 8831
    },
    {
      "epoch": 8.421897888183593e-06,
      "step": 8831,
      "training_step_time": 7.1939537525177
    },
    {
      "epoch": 8.4228515625e-06,
      "model_forward_time": 1.6955063343048096,
      "step": 8832
    },
    {
      "epoch": 8.4228515625e-06,
      "step": 8832,
      "training_step_time": 7.649232387542725
    },
    {
      "epoch": 8.423805236816406e-06,
      "model_forward_time": 1.709390640258789,
      "step": 8833
    },
    {
      "epoch": 8.423805236816406e-06,
      "step": 8833,
      "training_step_time": 8.58353853225708
    },
    {
      "epoch": 8.424758911132812e-06,
      "model_forward_time": 1.6968064308166504,
      "step": 8834
    },
    {
      "epoch": 8.424758911132812e-06,
      "step": 8834,
      "training_step_time": 7.616588115692139
    },
    {
      "epoch": 8.42571258544922e-06,
      "model_forward_time": 2.233898639678955,
      "step": 8835
    },
    {
      "epoch": 8.42571258544922e-06,
      "step": 8835,
      "training_step_time": 8.325203657150269
    },
    {
      "epoch": 8.426666259765625e-06,
      "model_forward_time": 1.691659688949585,
      "step": 8836
    },
    {
      "epoch": 8.426666259765625e-06,
      "step": 8836,
      "training_step_time": 9.241353988647461
    },
    {
      "epoch": 8.427619934082031e-06,
      "model_forward_time": 1.694242000579834,
      "step": 8837
    },
    {
      "epoch": 8.427619934082031e-06,
      "step": 8837,
      "training_step_time": 8.446780681610107
    },
    {
      "epoch": 8.428573608398437e-06,
      "model_forward_time": 1.690415620803833,
      "step": 8838
    },
    {
      "epoch": 8.428573608398437e-06,
      "step": 8838,
      "training_step_time": 7.388900279998779
    },
    {
      "epoch": 8.429527282714844e-06,
      "model_forward_time": 1.694854974746704,
      "step": 8839
    },
    {
      "epoch": 8.429527282714844e-06,
      "step": 8839,
      "training_step_time": 7.544654369354248
    },
    {
      "epoch": 8.43048095703125e-06,
      "grad_norm": 0.300416499376297,
      "learning_rate": 9.959761165826517e-06,
      "loss": 0.0664,
      "step": 8840
    },
    {
      "epoch": 8.43048095703125e-06,
      "model_forward_time": 1.6922457218170166,
      "step": 8840
    },
    {
      "dynamics_loss_avg": 0.06325996275991201,
      "epoch": 8.43048095703125e-06,
      "step": 8840
    },
    {
      "action_loss_avg": 0.0037510045047383757,
      "epoch": 8.43048095703125e-06,
      "step": 8840
    },
    {
      "epoch": 8.43048095703125e-06,
      "step": 8840,
      "training_step_time": 7.229934930801392
    },
    {
      "epoch": 8.431434631347656e-06,
      "model_forward_time": 1.6976521015167236,
      "step": 8841
    },
    {
      "epoch": 8.431434631347656e-06,
      "step": 8841,
      "training_step_time": 7.665682315826416
    },
    {
      "epoch": 8.432388305664063e-06,
      "model_forward_time": 1.7234060764312744,
      "step": 8842
    },
    {
      "epoch": 8.432388305664063e-06,
      "step": 8842,
      "training_step_time": 7.500048637390137
    },
    {
      "epoch": 8.433341979980469e-06,
      "model_forward_time": 1.6936285495758057,
      "step": 8843
    },
    {
      "epoch": 8.433341979980469e-06,
      "step": 8843,
      "training_step_time": 7.6738269329071045
    },
    {
      "epoch": 8.434295654296874e-06,
      "model_forward_time": 2.219862699508667,
      "step": 8844
    },
    {
      "epoch": 8.434295654296874e-06,
      "step": 8844,
      "training_step_time": 9.065927505493164
    },
    {
      "epoch": 8.435249328613282e-06,
      "model_forward_time": 1.6918280124664307,
      "step": 8845
    },
    {
      "epoch": 8.435249328613282e-06,
      "step": 8845,
      "training_step_time": 8.190279006958008
    },
    {
      "epoch": 8.436203002929688e-06,
      "model_forward_time": 1.692011833190918,
      "step": 8846
    },
    {
      "epoch": 8.436203002929688e-06,
      "step": 8846,
      "training_step_time": 7.569056034088135
    },
    {
      "epoch": 8.437156677246093e-06,
      "model_forward_time": 1.7001876831054688,
      "step": 8847
    },
    {
      "epoch": 8.437156677246093e-06,
      "step": 8847,
      "training_step_time": 9.027685642242432
    },
    {
      "epoch": 8.4381103515625e-06,
      "model_forward_time": 1.7108545303344727,
      "step": 8848
    },
    {
      "epoch": 8.4381103515625e-06,
      "step": 8848,
      "training_step_time": 7.584433555603027
    },
    {
      "epoch": 8.439064025878906e-06,
      "model_forward_time": 1.7080187797546387,
      "step": 8849
    },
    {
      "epoch": 8.439064025878906e-06,
      "step": 8849,
      "training_step_time": 7.271142482757568
    },
    {
      "epoch": 8.440017700195312e-06,
      "grad_norm": 0.22586266696453094,
      "learning_rate": 9.959551544413033e-06,
      "loss": 0.0729,
      "step": 8850
    },
    {
      "epoch": 8.440017700195312e-06,
      "model_forward_time": 1.6994874477386475,
      "step": 8850
    },
    {
      "dynamics_loss_avg": 0.06425260305404663,
      "epoch": 8.440017700195312e-06,
      "step": 8850
    },
    {
      "action_loss_avg": 0.006829868291970342,
      "epoch": 8.440017700195312e-06,
      "step": 8850
    },
    {
      "epoch": 8.440017700195312e-06,
      "step": 8850,
      "training_step_time": 7.646366357803345
    },
    {
      "epoch": 8.44097137451172e-06,
      "model_forward_time": 1.7201333045959473,
      "step": 8851
    },
    {
      "epoch": 8.44097137451172e-06,
      "step": 8851,
      "training_step_time": 8.003204584121704
    },
    {
      "epoch": 8.441925048828125e-06,
      "model_forward_time": 1.7085943222045898,
      "step": 8852
    },
    {
      "epoch": 8.441925048828125e-06,
      "step": 8852,
      "training_step_time": 8.646901845932007
    },
    {
      "epoch": 8.442878723144531e-06,
      "model_forward_time": 2.2748191356658936,
      "step": 8853
    },
    {
      "epoch": 8.442878723144531e-06,
      "step": 8853,
      "training_step_time": 8.829390525817871
    },
    {
      "epoch": 8.443832397460937e-06,
      "model_forward_time": 1.7239515781402588,
      "step": 8854
    },
    {
      "epoch": 8.443832397460937e-06,
      "step": 8854,
      "training_step_time": 9.935187339782715
    },
    {
      "epoch": 8.444786071777344e-06,
      "model_forward_time": 1.705402135848999,
      "step": 8855
    },
    {
      "epoch": 8.444786071777344e-06,
      "step": 8855,
      "training_step_time": 7.585086822509766
    },
    {
      "epoch": 8.44573974609375e-06,
      "model_forward_time": 1.7079453468322754,
      "step": 8856
    },
    {
      "epoch": 8.44573974609375e-06,
      "step": 8856,
      "training_step_time": 7.952041387557983
    },
    {
      "epoch": 8.446693420410156e-06,
      "model_forward_time": 1.7057056427001953,
      "step": 8857
    },
    {
      "epoch": 8.446693420410156e-06,
      "step": 8857,
      "training_step_time": 7.591451406478882
    },
    {
      "epoch": 8.447647094726563e-06,
      "model_forward_time": 1.699230432510376,
      "step": 8858
    },
    {
      "epoch": 8.447647094726563e-06,
      "step": 8858,
      "training_step_time": 7.278563499450684
    },
    {
      "epoch": 8.448600769042969e-06,
      "model_forward_time": 1.705885887145996,
      "step": 8859
    },
    {
      "epoch": 8.448600769042969e-06,
      "step": 8859,
      "training_step_time": 7.535823822021484
    },
    {
      "epoch": 8.449554443359374e-06,
      "grad_norm": 0.29330575466156006,
      "learning_rate": 9.959341380630449e-06,
      "loss": 0.0755,
      "step": 8860
    },
    {
      "epoch": 8.449554443359374e-06,
      "model_forward_time": 1.7156267166137695,
      "step": 8860
    },
    {
      "dynamics_loss_avg": 0.05063544511795044,
      "epoch": 8.449554443359374e-06,
      "step": 8860
    },
    {
      "action_loss_avg": 0.0030319127719849346,
      "epoch": 8.449554443359374e-06,
      "step": 8860
    },
    {
      "epoch": 8.449554443359374e-06,
      "step": 8860,
      "training_step_time": 7.258621454238892
    },
    {
      "epoch": 8.450508117675782e-06,
      "model_forward_time": 1.7142410278320312,
      "step": 8861
    },
    {
      "epoch": 8.450508117675782e-06,
      "step": 8861,
      "training_step_time": 7.599849700927734
    },
    {
      "epoch": 8.451461791992188e-06,
      "model_forward_time": 2.2279930114746094,
      "step": 8862
    },
    {
      "epoch": 8.451461791992188e-06,
      "step": 8862,
      "training_step_time": 9.100781440734863
    },
    {
      "epoch": 8.452415466308593e-06,
      "model_forward_time": 1.711719036102295,
      "step": 8863
    },
    {
      "epoch": 8.452415466308593e-06,
      "step": 8863,
      "training_step_time": 7.746369123458862
    },
    {
      "epoch": 8.453369140625e-06,
      "model_forward_time": 1.6924502849578857,
      "step": 8864
    },
    {
      "epoch": 8.453369140625e-06,
      "step": 8864,
      "training_step_time": 7.6026904582977295
    },
    {
      "epoch": 8.454322814941406e-06,
      "model_forward_time": 1.6946661472320557,
      "step": 8865
    },
    {
      "epoch": 8.454322814941406e-06,
      "step": 8865,
      "training_step_time": 7.507558107376099
    },
    {
      "epoch": 8.455276489257812e-06,
      "model_forward_time": 1.6892876625061035,
      "step": 8866
    },
    {
      "epoch": 8.455276489257812e-06,
      "step": 8866,
      "training_step_time": 8.25500750541687
    },
    {
      "epoch": 8.45623016357422e-06,
      "model_forward_time": 1.6889657974243164,
      "step": 8867
    },
    {
      "epoch": 8.45623016357422e-06,
      "step": 8867,
      "training_step_time": 7.312237501144409
    },
    {
      "epoch": 8.457183837890625e-06,
      "model_forward_time": 1.6803510189056396,
      "step": 8868
    },
    {
      "epoch": 8.457183837890625e-06,
      "step": 8868,
      "training_step_time": 8.457321405410767
    },
    {
      "epoch": 8.458137512207031e-06,
      "model_forward_time": 1.7049455642700195,
      "step": 8869
    },
    {
      "epoch": 8.458137512207031e-06,
      "step": 8869,
      "training_step_time": 7.2051520347595215
    },
    {
      "epoch": 8.459091186523437e-06,
      "grad_norm": 0.30788981914520264,
      "learning_rate": 9.959130674501746e-06,
      "loss": 0.0771,
      "step": 8870
    },
    {
      "epoch": 8.459091186523437e-06,
      "model_forward_time": 1.6937243938446045,
      "step": 8870
    },
    {
      "dynamics_loss_avg": 0.07134657837450505,
      "epoch": 8.459091186523437e-06,
      "step": 8870
    },
    {
      "action_loss_avg": 0.005620517983334139,
      "epoch": 8.459091186523437e-06,
      "step": 8870
    },
    {
      "epoch": 8.459091186523437e-06,
      "step": 8870,
      "training_step_time": 8.329197645187378
    },
    {
      "epoch": 8.460044860839844e-06,
      "model_forward_time": 1.694154977798462,
      "step": 8871
    },
    {
      "epoch": 8.460044860839844e-06,
      "step": 8871,
      "training_step_time": 9.747661352157593
    },
    {
      "epoch": 8.46099853515625e-06,
      "model_forward_time": 1.7059366703033447,
      "step": 8872
    },
    {
      "epoch": 8.46099853515625e-06,
      "step": 8872,
      "training_step_time": 10.296207189559937
    },
    {
      "epoch": 8.461952209472656e-06,
      "model_forward_time": 1.6946067810058594,
      "step": 8873
    },
    {
      "epoch": 8.461952209472656e-06,
      "step": 8873,
      "training_step_time": 8.437210321426392
    },
    {
      "epoch": 8.462905883789063e-06,
      "model_forward_time": 1.6929669380187988,
      "step": 8874
    },
    {
      "epoch": 8.462905883789063e-06,
      "step": 8874,
      "training_step_time": 7.500821113586426
    },
    {
      "epoch": 8.463859558105469e-06,
      "model_forward_time": 1.693735122680664,
      "step": 8875
    },
    {
      "epoch": 8.463859558105469e-06,
      "step": 8875,
      "training_step_time": 7.565001726150513
    },
    {
      "epoch": 8.464813232421874e-06,
      "model_forward_time": 1.696887731552124,
      "step": 8876
    },
    {
      "epoch": 8.464813232421874e-06,
      "step": 8876,
      "training_step_time": 7.284929990768433
    },
    {
      "epoch": 8.465766906738282e-06,
      "model_forward_time": 1.6940481662750244,
      "step": 8877
    },
    {
      "epoch": 8.465766906738282e-06,
      "step": 8877,
      "training_step_time": 7.470324277877808
    },
    {
      "epoch": 8.466720581054688e-06,
      "model_forward_time": 1.70786452293396,
      "step": 8878
    },
    {
      "epoch": 8.466720581054688e-06,
      "step": 8878,
      "training_step_time": 7.434867858886719
    },
    {
      "epoch": 8.467674255371093e-06,
      "model_forward_time": 1.6933329105377197,
      "step": 8879
    },
    {
      "epoch": 8.467674255371093e-06,
      "step": 8879,
      "training_step_time": 8.512881994247437
    },
    {
      "epoch": 8.4686279296875e-06,
      "grad_norm": 0.27477359771728516,
      "learning_rate": 9.958919426049969e-06,
      "loss": 0.0758,
      "step": 8880
    },
    {
      "epoch": 8.4686279296875e-06,
      "model_forward_time": 1.6903107166290283,
      "step": 8880
    },
    {
      "dynamics_loss_avg": 0.08179281800985336,
      "epoch": 8.4686279296875e-06,
      "step": 8880
    },
    {
      "action_loss_avg": 0.0049749485217034815,
      "epoch": 8.4686279296875e-06,
      "step": 8880
    },
    {
      "epoch": 8.4686279296875e-06,
      "step": 8880,
      "training_step_time": 7.540848016738892
    },
    {
      "epoch": 8.469581604003906e-06,
      "model_forward_time": 2.238253116607666,
      "step": 8881
    },
    {
      "epoch": 8.469581604003906e-06,
      "step": 8881,
      "training_step_time": 7.653202295303345
    },
    {
      "epoch": 8.470535278320312e-06,
      "model_forward_time": 1.6916351318359375,
      "step": 8882
    },
    {
      "epoch": 8.470535278320312e-06,
      "step": 8882,
      "training_step_time": 8.642273187637329
    },
    {
      "epoch": 8.471488952636718e-06,
      "model_forward_time": 1.6900804042816162,
      "step": 8883
    },
    {
      "epoch": 8.471488952636718e-06,
      "step": 8883,
      "training_step_time": 9.171911239624023
    },
    {
      "epoch": 8.472442626953125e-06,
      "model_forward_time": 1.7028594017028809,
      "step": 8884
    },
    {
      "epoch": 8.472442626953125e-06,
      "step": 8884,
      "training_step_time": 7.467163562774658
    },
    {
      "epoch": 8.473396301269531e-06,
      "model_forward_time": 1.6952085494995117,
      "step": 8885
    },
    {
      "epoch": 8.473396301269531e-06,
      "step": 8885,
      "training_step_time": 7.387153148651123
    },
    {
      "epoch": 8.474349975585937e-06,
      "model_forward_time": 1.6926202774047852,
      "step": 8886
    },
    {
      "epoch": 8.474349975585937e-06,
      "step": 8886,
      "training_step_time": 7.333287477493286
    },
    {
      "epoch": 8.475303649902344e-06,
      "model_forward_time": 1.7072727680206299,
      "step": 8887
    },
    {
      "epoch": 8.475303649902344e-06,
      "step": 8887,
      "training_step_time": 7.4402077198028564
    },
    {
      "epoch": 8.47625732421875e-06,
      "model_forward_time": 1.701220989227295,
      "step": 8888
    },
    {
      "epoch": 8.47625732421875e-06,
      "step": 8888,
      "training_step_time": 9.069729089736938
    },
    {
      "epoch": 8.477210998535156e-06,
      "model_forward_time": 1.3185679912567139,
      "step": 8889
    },
    {
      "epoch": 8.477210998535156e-06,
      "step": 8889,
      "training_step_time": 7.6337573528289795
    },
    {
      "epoch": 8.478164672851563e-06,
      "grad_norm": 0.2465500831604004,
      "learning_rate": 9.95870763529822e-06,
      "loss": 0.0795,
      "step": 8890
    },
    {
      "epoch": 8.478164672851563e-06,
      "model_forward_time": 2.248661756515503,
      "step": 8890
    },
    {
      "dynamics_loss_avg": 0.08699409272521734,
      "epoch": 8.478164672851563e-06,
      "step": 8890
    },
    {
      "action_loss_avg": 0.007468918059021235,
      "epoch": 8.478164672851563e-06,
      "step": 8890
    },
    {
      "epoch": 8.478164672851563e-06,
      "step": 8890,
      "training_step_time": 9.436858415603638
    },
    {
      "epoch": 8.479118347167969e-06,
      "model_forward_time": 1.6979832649230957,
      "step": 8891
    },
    {
      "epoch": 8.479118347167969e-06,
      "step": 8891,
      "training_step_time": 7.632989883422852
    },
    {
      "epoch": 8.480072021484374e-06,
      "model_forward_time": 1.6950569152832031,
      "step": 8892
    },
    {
      "epoch": 8.480072021484374e-06,
      "step": 8892,
      "training_step_time": 7.592907905578613
    },
    {
      "epoch": 8.481025695800782e-06,
      "model_forward_time": 1.695131540298462,
      "step": 8893
    },
    {
      "epoch": 8.481025695800782e-06,
      "step": 8893,
      "training_step_time": 7.413303375244141
    },
    {
      "epoch": 8.481979370117188e-06,
      "model_forward_time": 1.6916272640228271,
      "step": 8894
    },
    {
      "epoch": 8.481979370117188e-06,
      "step": 8894,
      "training_step_time": 7.502065658569336
    },
    {
      "epoch": 8.482933044433593e-06,
      "model_forward_time": 1.693579912185669,
      "step": 8895
    },
    {
      "epoch": 8.482933044433593e-06,
      "step": 8895,
      "training_step_time": 7.307774543762207
    },
    {
      "epoch": 8.48388671875e-06,
      "model_forward_time": 1.7147958278656006,
      "step": 8896
    },
    {
      "epoch": 8.48388671875e-06,
      "step": 8896,
      "training_step_time": 7.480175971984863
    },
    {
      "epoch": 8.484840393066406e-06,
      "model_forward_time": 1.6984546184539795,
      "step": 8897
    },
    {
      "epoch": 8.484840393066406e-06,
      "step": 8897,
      "training_step_time": 8.392475843429565
    },
    {
      "epoch": 8.485794067382812e-06,
      "model_forward_time": 1.6970713138580322,
      "step": 8898
    },
    {
      "epoch": 8.485794067382812e-06,
      "step": 8898,
      "training_step_time": 7.57884955406189
    },
    {
      "epoch": 8.486747741699218e-06,
      "model_forward_time": 2.2476935386657715,
      "step": 8899
    },
    {
      "epoch": 8.486747741699218e-06,
      "step": 8899,
      "training_step_time": 7.6228766441345215
    },
    {
      "epoch": 8.487701416015625e-06,
      "grad_norm": 0.1646726429462433,
      "learning_rate": 9.958495302269658e-06,
      "loss": 0.0771,
      "step": 8900
    },
    {
      "epoch": 8.487701416015625e-06,
      "model_forward_time": 1.6889770030975342,
      "step": 8900
    },
    {
      "dynamics_loss_avg": 0.06779694706201553,
      "epoch": 8.487701416015625e-06,
      "step": 8900
    },
    {
      "action_loss_avg": 0.0068144891993142664,
      "epoch": 8.487701416015625e-06,
      "step": 8900
    },
    {
      "epoch": 8.487701416015625e-06,
      "step": 8900,
      "training_step_time": 7.732735633850098
    },
    {
      "epoch": 8.488655090332031e-06,
      "model_forward_time": 1.6922314167022705,
      "step": 8901
    },
    {
      "epoch": 8.488655090332031e-06,
      "step": 8901,
      "training_step_time": 7.623330354690552
    },
    {
      "epoch": 8.489608764648437e-06,
      "model_forward_time": 1.3157167434692383,
      "step": 8902
    },
    {
      "epoch": 8.489608764648437e-06,
      "step": 8902,
      "training_step_time": 7.442767381668091
    },
    {
      "epoch": 8.490562438964844e-06,
      "model_forward_time": 1.6865811347961426,
      "step": 8903
    },
    {
      "epoch": 8.490562438964844e-06,
      "step": 8903,
      "training_step_time": 8.414390325546265
    },
    {
      "epoch": 8.49151611328125e-06,
      "model_forward_time": 1.711045265197754,
      "step": 8904
    },
    {
      "epoch": 8.49151611328125e-06,
      "step": 8904,
      "training_step_time": 8.084123611450195
    },
    {
      "epoch": 8.492469787597656e-06,
      "model_forward_time": 1.7049775123596191,
      "step": 8905
    },
    {
      "epoch": 8.492469787597656e-06,
      "step": 8905,
      "training_step_time": 7.47652006149292
    },
    {
      "epoch": 8.493423461914063e-06,
      "model_forward_time": 1.698770523071289,
      "step": 8906
    },
    {
      "epoch": 8.493423461914063e-06,
      "step": 8906,
      "training_step_time": 7.88693642616272
    },
    {
      "epoch": 8.494377136230469e-06,
      "model_forward_time": 1.702211618423462,
      "step": 8907
    },
    {
      "epoch": 8.494377136230469e-06,
      "step": 8907,
      "training_step_time": 8.622586488723755
    },
    {
      "epoch": 8.495330810546874e-06,
      "model_forward_time": 2.2406811714172363,
      "step": 8908
    },
    {
      "epoch": 8.495330810546874e-06,
      "step": 8908,
      "training_step_time": 7.570302248001099
    },
    {
      "epoch": 8.496284484863282e-06,
      "model_forward_time": 1.7000720500946045,
      "step": 8909
    },
    {
      "epoch": 8.496284484863282e-06,
      "step": 8909,
      "training_step_time": 7.7479567527771
    },
    {
      "epoch": 8.497238159179688e-06,
      "grad_norm": 0.2403412163257599,
      "learning_rate": 9.958282426987504e-06,
      "loss": 0.0783,
      "step": 8910
    },
    {
      "epoch": 8.497238159179688e-06,
      "model_forward_time": 1.6914198398590088,
      "step": 8910
    },
    {
      "dynamics_loss_avg": 0.08679862227290869,
      "epoch": 8.497238159179688e-06,
      "step": 8910
    },
    {
      "action_loss_avg": 0.005821799952536821,
      "epoch": 8.497238159179688e-06,
      "step": 8910
    },
    {
      "epoch": 8.497238159179688e-06,
      "step": 8910,
      "training_step_time": 8.494754552841187
    },
    {
      "epoch": 8.498191833496093e-06,
      "model_forward_time": 1.7000234127044678,
      "step": 8911
    },
    {
      "epoch": 8.498191833496093e-06,
      "step": 8911,
      "training_step_time": 8.105252981185913
    },
    {
      "epoch": 8.4991455078125e-06,
      "model_forward_time": 1.342947006225586,
      "step": 8912
    },
    {
      "epoch": 8.4991455078125e-06,
      "step": 8912,
      "training_step_time": 8.411623477935791
    },
    {
      "epoch": 8.500099182128906e-06,
      "model_forward_time": 1.709333896636963,
      "step": 8913
    },
    {
      "epoch": 8.500099182128906e-06,
      "step": 8913,
      "training_step_time": 8.153848886489868
    },
    {
      "epoch": 8.501052856445312e-06,
      "model_forward_time": 1.701200246810913,
      "step": 8914
    },
    {
      "epoch": 8.501052856445312e-06,
      "step": 8914,
      "training_step_time": 8.117766618728638
    },
    {
      "epoch": 8.502006530761718e-06,
      "model_forward_time": 1.7045860290527344,
      "step": 8915
    },
    {
      "epoch": 8.502006530761718e-06,
      "step": 8915,
      "training_step_time": 7.380744218826294
    },
    {
      "epoch": 8.502960205078125e-06,
      "model_forward_time": 1.70237398147583,
      "step": 8916
    },
    {
      "epoch": 8.502960205078125e-06,
      "step": 8916,
      "training_step_time": 7.651004076004028
    },
    {
      "epoch": 8.503913879394531e-06,
      "model_forward_time": 1.706202507019043,
      "step": 8917
    },
    {
      "epoch": 8.503913879394531e-06,
      "step": 8917,
      "training_step_time": 9.195008516311646
    },
    {
      "epoch": 8.504867553710937e-06,
      "model_forward_time": 2.25557279586792,
      "step": 8918
    },
    {
      "epoch": 8.504867553710937e-06,
      "step": 8918,
      "training_step_time": 7.744306325912476
    },
    {
      "epoch": 8.505821228027344e-06,
      "model_forward_time": 1.7267611026763916,
      "step": 8919
    },
    {
      "epoch": 8.505821228027344e-06,
      "step": 8919,
      "training_step_time": 10.356802463531494
    },
    {
      "epoch": 8.50677490234375e-06,
      "grad_norm": 0.31807780265808105,
      "learning_rate": 9.958069009475042e-06,
      "loss": 0.0776,
      "step": 8920
    },
    {
      "epoch": 8.50677490234375e-06,
      "model_forward_time": 1.7210924625396729,
      "step": 8920
    },
    {
      "dynamics_loss_avg": 0.07866429015994073,
      "epoch": 8.50677490234375e-06,
      "step": 8920
    },
    {
      "action_loss_avg": 0.00599755224538967,
      "epoch": 8.50677490234375e-06,
      "step": 8920
    },
    {
      "epoch": 8.50677490234375e-06,
      "step": 8920,
      "training_step_time": 7.681190729141235
    },
    {
      "epoch": 8.507728576660156e-06,
      "model_forward_time": 1.728302240371704,
      "step": 8921
    },
    {
      "epoch": 8.507728576660156e-06,
      "step": 8921,
      "training_step_time": 8.613725900650024
    },
    {
      "epoch": 8.508682250976563e-06,
      "model_forward_time": 1.7214245796203613,
      "step": 8922
    },
    {
      "epoch": 8.508682250976563e-06,
      "step": 8922,
      "training_step_time": 7.5493385791778564
    },
    {
      "epoch": 8.509635925292969e-06,
      "model_forward_time": 1.722749948501587,
      "step": 8923
    },
    {
      "epoch": 8.509635925292969e-06,
      "step": 8923,
      "training_step_time": 7.5152428150177
    },
    {
      "epoch": 8.510589599609375e-06,
      "model_forward_time": 1.766014814376831,
      "step": 8924
    },
    {
      "epoch": 8.510589599609375e-06,
      "step": 8924,
      "training_step_time": 7.269840717315674
    },
    {
      "epoch": 8.511543273925782e-06,
      "model_forward_time": 1.7475450038909912,
      "step": 8925
    },
    {
      "epoch": 8.511543273925782e-06,
      "step": 8925,
      "training_step_time": 9.424751281738281
    },
    {
      "epoch": 8.512496948242188e-06,
      "model_forward_time": 1.7131009101867676,
      "step": 8926
    },
    {
      "epoch": 8.512496948242188e-06,
      "step": 8926,
      "training_step_time": 8.485692024230957
    },
    {
      "epoch": 8.513450622558593e-06,
      "model_forward_time": 2.2472527027130127,
      "step": 8927
    },
    {
      "epoch": 8.513450622558593e-06,
      "step": 8927,
      "training_step_time": 7.801553964614868
    },
    {
      "epoch": 8.514404296875e-06,
      "model_forward_time": 1.724125623703003,
      "step": 8928
    },
    {
      "epoch": 8.514404296875e-06,
      "step": 8928,
      "training_step_time": 8.471489667892456
    },
    {
      "epoch": 8.515357971191407e-06,
      "model_forward_time": 1.7229204177856445,
      "step": 8929
    },
    {
      "epoch": 8.515357971191407e-06,
      "step": 8929,
      "training_step_time": 9.14547348022461
    },
    {
      "epoch": 8.516311645507812e-06,
      "grad_norm": 0.3945573568344116,
      "learning_rate": 9.957855049755605e-06,
      "loss": 0.073,
      "step": 8930
    },
    {
      "epoch": 8.516311645507812e-06,
      "model_forward_time": 1.7133634090423584,
      "step": 8930
    },
    {
      "dynamics_loss_avg": 0.07056104391813278,
      "epoch": 8.516311645507812e-06,
      "step": 8930
    },
    {
      "action_loss_avg": 0.005630162206944078,
      "epoch": 8.516311645507812e-06,
      "step": 8930
    },
    {
      "epoch": 8.516311645507812e-06,
      "step": 8930,
      "training_step_time": 8.0466148853302
    },
    {
      "epoch": 8.517265319824218e-06,
      "model_forward_time": 1.720496654510498,
      "step": 8931
    },
    {
      "epoch": 8.517265319824218e-06,
      "step": 8931,
      "training_step_time": 7.472879886627197
    },
    {
      "epoch": 8.518218994140625e-06,
      "model_forward_time": 1.7170209884643555,
      "step": 8932
    },
    {
      "epoch": 8.518218994140625e-06,
      "step": 8932,
      "training_step_time": 7.59318995475769
    },
    {
      "epoch": 8.519172668457031e-06,
      "model_forward_time": 1.7269783020019531,
      "step": 8933
    },
    {
      "epoch": 8.519172668457031e-06,
      "step": 8933,
      "training_step_time": 7.27518630027771
    },
    {
      "epoch": 8.520126342773437e-06,
      "model_forward_time": 1.727022647857666,
      "step": 8934
    },
    {
      "epoch": 8.520126342773437e-06,
      "step": 8934,
      "training_step_time": 8.140661716461182
    },
    {
      "epoch": 8.521080017089844e-06,
      "model_forward_time": 1.7226581573486328,
      "step": 8935
    },
    {
      "epoch": 8.521080017089844e-06,
      "step": 8935,
      "training_step_time": 7.690698862075806
    },
    {
      "epoch": 8.52203369140625e-06,
      "model_forward_time": 2.2676374912261963,
      "step": 8936
    },
    {
      "epoch": 8.52203369140625e-06,
      "step": 8936,
      "training_step_time": 7.703973054885864
    },
    {
      "epoch": 8.522987365722656e-06,
      "model_forward_time": 1.7233362197875977,
      "step": 8937
    },
    {
      "epoch": 8.522987365722656e-06,
      "step": 8937,
      "training_step_time": 7.600322723388672
    },
    {
      "epoch": 8.523941040039063e-06,
      "model_forward_time": 1.7170264720916748,
      "step": 8938
    },
    {
      "epoch": 8.523941040039063e-06,
      "step": 8938,
      "training_step_time": 7.626766920089722
    },
    {
      "epoch": 8.524894714355469e-06,
      "model_forward_time": 1.7287039756774902,
      "step": 8939
    },
    {
      "epoch": 8.524894714355469e-06,
      "step": 8939,
      "training_step_time": 7.400183439254761
    },
    {
      "epoch": 8.525848388671875e-06,
      "grad_norm": 0.4066036343574524,
      "learning_rate": 9.957640547852593e-06,
      "loss": 0.0764,
      "step": 8940
    },
    {
      "epoch": 8.525848388671875e-06,
      "model_forward_time": 1.7182416915893555,
      "step": 8940
    },
    {
      "dynamics_loss_avg": 0.05086687579751015,
      "epoch": 8.525848388671875e-06,
      "step": 8940
    },
    {
      "action_loss_avg": 0.004691463359631598,
      "epoch": 8.525848388671875e-06,
      "step": 8940
    },
    {
      "epoch": 8.525848388671875e-06,
      "step": 8940,
      "training_step_time": 7.602801084518433
    },
    {
      "epoch": 8.526802062988282e-06,
      "model_forward_time": 1.7195451259613037,
      "step": 8941
    },
    {
      "epoch": 8.526802062988282e-06,
      "step": 8941,
      "training_step_time": 7.4409496784210205
    },
    {
      "epoch": 8.527755737304688e-06,
      "model_forward_time": 1.7163405418395996,
      "step": 8942
    },
    {
      "epoch": 8.527755737304688e-06,
      "step": 8942,
      "training_step_time": 8.089231014251709
    },
    {
      "epoch": 8.528709411621093e-06,
      "model_forward_time": 1.7206988334655762,
      "step": 8943
    },
    {
      "epoch": 8.528709411621093e-06,
      "step": 8943,
      "training_step_time": 7.626806259155273
    },
    {
      "epoch": 8.5296630859375e-06,
      "model_forward_time": 1.7211596965789795,
      "step": 8944
    },
    {
      "epoch": 8.5296630859375e-06,
      "step": 8944,
      "training_step_time": 8.287668228149414
    },
    {
      "epoch": 8.530616760253907e-06,
      "model_forward_time": 2.26263689994812,
      "step": 8945
    },
    {
      "epoch": 8.530616760253907e-06,
      "step": 8945,
      "training_step_time": 8.179677724838257
    },
    {
      "epoch": 8.531570434570312e-06,
      "model_forward_time": 1.720188856124878,
      "step": 8946
    },
    {
      "epoch": 8.531570434570312e-06,
      "step": 8946,
      "training_step_time": 8.278931379318237
    },
    {
      "epoch": 8.532524108886718e-06,
      "model_forward_time": 1.3446192741394043,
      "step": 8947
    },
    {
      "epoch": 8.532524108886718e-06,
      "step": 8947,
      "training_step_time": 8.034236431121826
    },
    {
      "epoch": 8.533477783203125e-06,
      "model_forward_time": 1.7021608352661133,
      "step": 8948
    },
    {
      "epoch": 8.533477783203125e-06,
      "step": 8948,
      "training_step_time": 7.934011220932007
    },
    {
      "epoch": 8.534431457519531e-06,
      "model_forward_time": 1.7208900451660156,
      "step": 8949
    },
    {
      "epoch": 8.534431457519531e-06,
      "step": 8949,
      "training_step_time": 8.892595529556274
    },
    {
      "epoch": 8.535385131835937e-06,
      "grad_norm": 0.3502533733844757,
      "learning_rate": 9.957425503789466e-06,
      "loss": 0.07,
      "step": 8950
    },
    {
      "epoch": 8.535385131835937e-06,
      "model_forward_time": 1.7191903591156006,
      "step": 8950
    },
    {
      "dynamics_loss_avg": 0.06197942979633808,
      "epoch": 8.535385131835937e-06,
      "step": 8950
    },
    {
      "action_loss_avg": 0.0043271323665976524,
      "epoch": 8.535385131835937e-06,
      "step": 8950
    },
    {
      "epoch": 8.535385131835937e-06,
      "step": 8950,
      "training_step_time": 7.389989614486694
    },
    {
      "epoch": 8.536338806152344e-06,
      "model_forward_time": 1.7187137603759766,
      "step": 8951
    },
    {
      "epoch": 8.536338806152344e-06,
      "step": 8951,
      "training_step_time": 7.517104148864746
    },
    {
      "epoch": 8.53729248046875e-06,
      "model_forward_time": 1.767200231552124,
      "step": 8952
    },
    {
      "epoch": 8.53729248046875e-06,
      "step": 8952,
      "training_step_time": 8.31011438369751
    },
    {
      "epoch": 8.538246154785156e-06,
      "model_forward_time": 1.346836805343628,
      "step": 8953
    },
    {
      "epoch": 8.538246154785156e-06,
      "step": 8953,
      "training_step_time": 7.583606958389282
    },
    {
      "epoch": 8.539199829101563e-06,
      "model_forward_time": 1.7240972518920898,
      "step": 8954
    },
    {
      "epoch": 8.539199829101563e-06,
      "step": 8954,
      "training_step_time": 8.147308588027954
    },
    {
      "epoch": 8.540153503417969e-06,
      "model_forward_time": 2.24700927734375,
      "step": 8955
    },
    {
      "epoch": 8.540153503417969e-06,
      "step": 8955,
      "training_step_time": 8.671250104904175
    },
    {
      "epoch": 8.541107177734375e-06,
      "model_forward_time": 1.7260355949401855,
      "step": 8956
    },
    {
      "epoch": 8.541107177734375e-06,
      "step": 8956,
      "training_step_time": 7.629793643951416
    },
    {
      "epoch": 8.542060852050782e-06,
      "model_forward_time": 1.719773292541504,
      "step": 8957
    },
    {
      "epoch": 8.542060852050782e-06,
      "step": 8957,
      "training_step_time": 7.68984842300415
    },
    {
      "epoch": 8.543014526367188e-06,
      "model_forward_time": 1.7139606475830078,
      "step": 8958
    },
    {
      "epoch": 8.543014526367188e-06,
      "step": 8958,
      "training_step_time": 7.951164245605469
    },
    {
      "epoch": 8.543968200683593e-06,
      "model_forward_time": 1.73002028465271,
      "step": 8959
    },
    {
      "epoch": 8.543968200683593e-06,
      "step": 8959,
      "training_step_time": 7.413153409957886
    },
    {
      "epoch": 8.544921875e-06,
      "grad_norm": 0.3062564730644226,
      "learning_rate": 9.957209917589738e-06,
      "loss": 0.0704,
      "step": 8960
    },
    {
      "epoch": 8.544921875e-06,
      "model_forward_time": 1.7120296955108643,
      "step": 8960
    },
    {
      "dynamics_loss_avg": 0.06151488609611988,
      "epoch": 8.544921875e-06,
      "step": 8960
    },
    {
      "action_loss_avg": 0.0036139359814114868,
      "epoch": 8.544921875e-06,
      "step": 8960
    },
    {
      "epoch": 8.544921875e-06,
      "step": 8960,
      "training_step_time": 7.602380990982056
    },
    {
      "epoch": 8.545875549316407e-06,
      "model_forward_time": 1.761221170425415,
      "step": 8961
    },
    {
      "epoch": 8.545875549316407e-06,
      "step": 8961,
      "training_step_time": 7.5738654136657715
    },
    {
      "epoch": 8.546829223632812e-06,
      "model_forward_time": 1.343714714050293,
      "step": 8962
    },
    {
      "epoch": 8.546829223632812e-06,
      "step": 8962,
      "training_step_time": 8.627445459365845
    },
    {
      "epoch": 8.547782897949218e-06,
      "model_forward_time": 2.6135740280151367,
      "step": 8963
    },
    {
      "epoch": 8.547782897949218e-06,
      "step": 8963,
      "training_step_time": 8.156383037567139
    },
    {
      "epoch": 8.548736572265625e-06,
      "model_forward_time": 1.7174103260040283,
      "step": 8964
    },
    {
      "epoch": 8.548736572265625e-06,
      "step": 8964,
      "training_step_time": 8.487930297851562
    },
    {
      "epoch": 8.549690246582031e-06,
      "model_forward_time": 1.7165393829345703,
      "step": 8965
    },
    {
      "epoch": 8.549690246582031e-06,
      "step": 8965,
      "training_step_time": 6.7487640380859375
    },
    {
      "epoch": 8.550643920898437e-06,
      "model_forward_time": 1.7209722995758057,
      "step": 8966
    },
    {
      "epoch": 8.550643920898437e-06,
      "step": 8966,
      "training_step_time": 6.735719442367554
    },
    {
      "epoch": 8.551597595214844e-06,
      "model_forward_time": 1.7227575778961182,
      "step": 8967
    },
    {
      "epoch": 8.551597595214844e-06,
      "step": 8967,
      "training_step_time": 6.712550401687622
    },
    {
      "epoch": 8.55255126953125e-06,
      "model_forward_time": 1.7183775901794434,
      "step": 8968
    },
    {
      "epoch": 8.55255126953125e-06,
      "step": 8968,
      "training_step_time": 6.6834259033203125
    },
    {
      "epoch": 8.553504943847656e-06,
      "model_forward_time": 1.724776029586792,
      "step": 8969
    },
    {
      "epoch": 8.553504943847656e-06,
      "step": 8969,
      "training_step_time": 6.700706720352173
    },
    {
      "epoch": 8.554458618164063e-06,
      "grad_norm": 0.314436674118042,
      "learning_rate": 9.956993789276988e-06,
      "loss": 0.0647,
      "step": 8970
    },
    {
      "epoch": 8.554458618164063e-06,
      "model_forward_time": 1.7198128700256348,
      "step": 8970
    },
    {
      "dynamics_loss_avg": 0.06012323796749115,
      "epoch": 8.554458618164063e-06,
      "step": 8970
    },
    {
      "action_loss_avg": 0.004480715957470238,
      "epoch": 8.554458618164063e-06,
      "step": 8970
    },
    {
      "epoch": 8.554458618164063e-06,
      "step": 8970,
      "training_step_time": 6.686079502105713
    },
    {
      "epoch": 8.555412292480469e-06,
      "model_forward_time": 1.7229318618774414,
      "step": 8971
    },
    {
      "epoch": 8.555412292480469e-06,
      "step": 8971,
      "training_step_time": 6.657726526260376
    },
    {
      "epoch": 8.556365966796875e-06,
      "model_forward_time": 1.7236032485961914,
      "step": 8972
    },
    {
      "epoch": 8.556365966796875e-06,
      "step": 8972,
      "training_step_time": 7.220398664474487
    },
    {
      "epoch": 8.557319641113282e-06,
      "model_forward_time": 2.2434608936309814,
      "step": 8973
    },
    {
      "epoch": 8.557319641113282e-06,
      "step": 8973,
      "training_step_time": 9.67923903465271
    },
    {
      "epoch": 8.558273315429688e-06,
      "model_forward_time": 1.7195312976837158,
      "step": 8974
    },
    {
      "epoch": 8.558273315429688e-06,
      "step": 8974,
      "training_step_time": 7.540199518203735
    },
    {
      "epoch": 8.559226989746093e-06,
      "model_forward_time": 1.753162145614624,
      "step": 8975
    },
    {
      "epoch": 8.559226989746093e-06,
      "step": 8975,
      "training_step_time": 6.6887571811676025
    },
    {
      "epoch": 8.5601806640625e-06,
      "model_forward_time": 1.7166314125061035,
      "step": 8976
    },
    {
      "epoch": 8.5601806640625e-06,
      "step": 8976,
      "training_step_time": 6.723024129867554
    },
    {
      "epoch": 8.561134338378907e-06,
      "model_forward_time": 1.7235839366912842,
      "step": 8977
    },
    {
      "epoch": 8.561134338378907e-06,
      "step": 8977,
      "training_step_time": 6.662615060806274
    },
    {
      "epoch": 8.562088012695312e-06,
      "model_forward_time": 1.7163960933685303,
      "step": 8978
    },
    {
      "epoch": 8.562088012695312e-06,
      "step": 8978,
      "training_step_time": 6.704327344894409
    },
    {
      "epoch": 8.563041687011718e-06,
      "model_forward_time": 1.7233035564422607,
      "step": 8979
    },
    {
      "epoch": 8.563041687011718e-06,
      "step": 8979,
      "training_step_time": 6.693239450454712
    },
    {
      "epoch": 8.563995361328125e-06,
      "grad_norm": 0.3928113281726837,
      "learning_rate": 9.956777118874848e-06,
      "loss": 0.071,
      "step": 8980
    },
    {
      "epoch": 8.563995361328125e-06,
      "model_forward_time": 1.7153105735778809,
      "step": 8980
    },
    {
      "dynamics_loss_avg": 0.059511129185557364,
      "epoch": 8.563995361328125e-06,
      "step": 8980
    },
    {
      "action_loss_avg": 0.006221759342588484,
      "epoch": 8.563995361328125e-06,
      "step": 8980
    },
    {
      "epoch": 8.563995361328125e-06,
      "step": 8980,
      "training_step_time": 6.667905807495117
    },
    {
      "epoch": 8.564949035644531e-06,
      "model_forward_time": 1.7218809127807617,
      "step": 8981
    },
    {
      "epoch": 8.564949035644531e-06,
      "step": 8981,
      "training_step_time": 6.673126935958862
    },
    {
      "epoch": 8.565902709960937e-06,
      "model_forward_time": 2.3074636459350586,
      "step": 8982
    },
    {
      "epoch": 8.565902709960937e-06,
      "step": 8982,
      "training_step_time": 10.450850248336792
    },
    {
      "epoch": 8.566856384277344e-06,
      "model_forward_time": 1.7212247848510742,
      "step": 8983
    },
    {
      "epoch": 8.566856384277344e-06,
      "step": 8983,
      "training_step_time": 7.582806825637817
    },
    {
      "epoch": 8.56781005859375e-06,
      "model_forward_time": 1.7181389331817627,
      "step": 8984
    },
    {
      "epoch": 8.56781005859375e-06,
      "step": 8984,
      "training_step_time": 6.696267604827881
    },
    {
      "epoch": 8.568763732910156e-06,
      "model_forward_time": 1.7274432182312012,
      "step": 8985
    },
    {
      "epoch": 8.568763732910156e-06,
      "step": 8985,
      "training_step_time": 6.655123472213745
    },
    {
      "epoch": 8.569717407226563e-06,
      "model_forward_time": 1.718214750289917,
      "step": 8986
    },
    {
      "epoch": 8.569717407226563e-06,
      "step": 8986,
      "training_step_time": 6.654672384262085
    },
    {
      "epoch": 8.570671081542969e-06,
      "model_forward_time": 1.7231590747833252,
      "step": 8987
    },
    {
      "epoch": 8.570671081542969e-06,
      "step": 8987,
      "training_step_time": 6.698918342590332
    },
    {
      "epoch": 8.571624755859375e-06,
      "model_forward_time": 1.723417043685913,
      "step": 8988
    },
    {
      "epoch": 8.571624755859375e-06,
      "step": 8988,
      "training_step_time": 6.694383382797241
    },
    {
      "epoch": 8.572578430175782e-06,
      "model_forward_time": 1.7193832397460938,
      "step": 8989
    },
    {
      "epoch": 8.572578430175782e-06,
      "step": 8989,
      "training_step_time": 6.696056842803955
    },
    {
      "epoch": 8.573532104492188e-06,
      "grad_norm": 0.244352325797081,
      "learning_rate": 9.956559906407016e-06,
      "loss": 0.0707,
      "step": 8990
    },
    {
      "epoch": 8.573532104492188e-06,
      "model_forward_time": 1.7185423374176025,
      "step": 8990
    },
    {
      "dynamics_loss_avg": 0.06219916399568319,
      "epoch": 8.573532104492188e-06,
      "step": 8990
    },
    {
      "action_loss_avg": 0.003281237056944519,
      "epoch": 8.573532104492188e-06,
      "step": 8990
    },
    {
      "epoch": 8.573532104492188e-06,
      "step": 8990,
      "training_step_time": 6.681829452514648
    },
    {
      "epoch": 8.574485778808593e-06,
      "model_forward_time": 2.3080062866210938,
      "step": 8991
    },
    {
      "epoch": 8.574485778808593e-06,
      "step": 8991,
      "training_step_time": 12.7893967628479
    },
    {
      "epoch": 8.575439453125e-06,
      "model_forward_time": 1.7250680923461914,
      "step": 8992
    },
    {
      "epoch": 8.575439453125e-06,
      "step": 8992,
      "training_step_time": 7.72767448425293
    },
    {
      "epoch": 8.576393127441407e-06,
      "model_forward_time": 1.7174739837646484,
      "step": 8993
    },
    {
      "epoch": 8.576393127441407e-06,
      "step": 8993,
      "training_step_time": 6.972100496292114
    },
    {
      "epoch": 8.577346801757812e-06,
      "model_forward_time": 1.715714693069458,
      "step": 8994
    },
    {
      "epoch": 8.577346801757812e-06,
      "step": 8994,
      "training_step_time": 6.68592095375061
    },
    {
      "epoch": 8.578300476074218e-06,
      "model_forward_time": 1.7153799533843994,
      "step": 8995
    },
    {
      "epoch": 8.578300476074218e-06,
      "step": 8995,
      "training_step_time": 6.708414077758789
    },
    {
      "epoch": 8.579254150390625e-06,
      "model_forward_time": 1.714536428451538,
      "step": 8996
    },
    {
      "epoch": 8.579254150390625e-06,
      "step": 8996,
      "training_step_time": 6.702029466629028
    },
    {
      "epoch": 8.580207824707031e-06,
      "model_forward_time": 1.73466157913208,
      "step": 8997
    },
    {
      "epoch": 8.580207824707031e-06,
      "step": 8997,
      "training_step_time": 6.71159029006958
    },
    {
      "epoch": 8.581161499023437e-06,
      "model_forward_time": 1.7643938064575195,
      "step": 8998
    },
    {
      "epoch": 8.581161499023437e-06,
      "step": 8998,
      "training_step_time": 6.7868874073028564
    },
    {
      "epoch": 8.582115173339844e-06,
      "model_forward_time": 1.7310466766357422,
      "step": 8999
    },
    {
      "epoch": 8.582115173339844e-06,
      "step": 8999,
      "training_step_time": 6.782392501831055
    },
    {
      "epoch": 8.58306884765625e-06,
      "grad_norm": 0.2847399115562439,
      "learning_rate": 9.956342151897244e-06,
      "loss": 0.0734,
      "step": 9000
    },
    {
      "epoch": 8.58306884765625e-06,
      "model_forward_time": 1.7460808753967285,
      "step": 9000
    },
    {
      "dynamics_loss_avg": 0.08493267493322491,
      "epoch": 8.58306884765625e-06,
      "step": 9000
    },
    {
      "action_loss_avg": 0.0049378937808796765,
      "epoch": 8.58306884765625e-06,
      "step": 9000
    },
    {
      "epoch": 8.58306884765625e-06,
      "step": 9000,
      "training_step_time": 10.122003316879272
    },
    {
      "epoch": 8.584022521972656e-06,
      "model_forward_time": 1.7329421043395996,
      "step": 9001
    },
    {
      "epoch": 8.584022521972656e-06,
      "step": 9001,
      "training_step_time": 11.530378103256226
    },
    {
      "epoch": 8.584976196289063e-06,
      "model_forward_time": 1.7346844673156738,
      "step": 9002
    },
    {
      "epoch": 8.584976196289063e-06,
      "step": 9002,
      "training_step_time": 7.646372318267822
    },
    {
      "epoch": 8.585929870605469e-06,
      "model_forward_time": 1.7357752323150635,
      "step": 9003
    },
    {
      "epoch": 8.585929870605469e-06,
      "step": 9003,
      "training_step_time": 6.7000274658203125
    },
    {
      "epoch": 8.586883544921875e-06,
      "model_forward_time": 1.8019168376922607,
      "step": 9004
    },
    {
      "epoch": 8.586883544921875e-06,
      "step": 9004,
      "training_step_time": 6.676204204559326
    },
    {
      "epoch": 8.587837219238282e-06,
      "model_forward_time": 1.8284308910369873,
      "step": 9005
    },
    {
      "epoch": 8.587837219238282e-06,
      "step": 9005,
      "training_step_time": 6.640576124191284
    },
    {
      "epoch": 8.588790893554688e-06,
      "model_forward_time": 1.8217549324035645,
      "step": 9006
    },
    {
      "epoch": 8.588790893554688e-06,
      "step": 9006,
      "training_step_time": 6.721406936645508
    },
    {
      "epoch": 8.589744567871093e-06,
      "model_forward_time": 1.824784517288208,
      "step": 9007
    },
    {
      "epoch": 8.589744567871093e-06,
      "step": 9007,
      "training_step_time": 6.7112579345703125
    },
    {
      "epoch": 8.5906982421875e-06,
      "model_forward_time": 2.804971218109131,
      "step": 9008
    },
    {
      "epoch": 8.5906982421875e-06,
      "step": 9008,
      "training_step_time": 7.476195335388184
    },
    {
      "epoch": 8.591651916503907e-06,
      "model_forward_time": 1.8182389736175537,
      "step": 9009
    },
    {
      "epoch": 8.591651916503907e-06,
      "step": 9009,
      "training_step_time": 8.963512182235718
    },
    {
      "epoch": 8.592605590820312e-06,
      "grad_norm": 0.2625865340232849,
      "learning_rate": 9.956123855369346e-06,
      "loss": 0.0707,
      "step": 9010
    },
    {
      "epoch": 8.592605590820312e-06,
      "model_forward_time": 1.8218233585357666,
      "step": 9010
    },
    {
      "dynamics_loss_avg": 0.05956487851217389,
      "epoch": 8.592605590820312e-06,
      "step": 9010
    },
    {
      "action_loss_avg": 0.0038919100537896157,
      "epoch": 8.592605590820312e-06,
      "step": 9010
    },
    {
      "epoch": 8.592605590820312e-06,
      "step": 9010,
      "training_step_time": 8.66847014427185
    },
    {
      "epoch": 8.593559265136718e-06,
      "model_forward_time": 1.8118822574615479,
      "step": 9011
    },
    {
      "epoch": 8.593559265136718e-06,
      "step": 9011,
      "training_step_time": 7.706952095031738
    },
    {
      "epoch": 8.594512939453125e-06,
      "model_forward_time": 1.8263483047485352,
      "step": 9012
    },
    {
      "epoch": 8.594512939453125e-06,
      "step": 9012,
      "training_step_time": 6.80686616897583
    },
    {
      "epoch": 8.595466613769531e-06,
      "model_forward_time": 1.826676607131958,
      "step": 9013
    },
    {
      "epoch": 8.595466613769531e-06,
      "step": 9013,
      "training_step_time": 6.83499813079834
    },
    {
      "epoch": 8.596420288085937e-06,
      "model_forward_time": 1.8160240650177002,
      "step": 9014
    },
    {
      "epoch": 8.596420288085937e-06,
      "step": 9014,
      "training_step_time": 6.702197074890137
    },
    {
      "epoch": 8.597373962402344e-06,
      "model_forward_time": 1.8239543437957764,
      "step": 9015
    },
    {
      "epoch": 8.597373962402344e-06,
      "step": 9015,
      "training_step_time": 6.723177433013916
    },
    {
      "epoch": 8.59832763671875e-06,
      "model_forward_time": 1.823643684387207,
      "step": 9016
    },
    {
      "epoch": 8.59832763671875e-06,
      "step": 9016,
      "training_step_time": 6.729761362075806
    },
    {
      "epoch": 8.599281311035156e-06,
      "model_forward_time": 1.8373644351959229,
      "step": 9017
    },
    {
      "epoch": 8.599281311035156e-06,
      "step": 9017,
      "training_step_time": 6.739679336547852
    },
    {
      "epoch": 8.600234985351563e-06,
      "model_forward_time": 2.3737986087799072,
      "step": 9018
    },
    {
      "epoch": 8.600234985351563e-06,
      "step": 9018,
      "training_step_time": 8.48740530014038
    },
    {
      "epoch": 8.601188659667969e-06,
      "model_forward_time": 1.8262720108032227,
      "step": 9019
    },
    {
      "epoch": 8.601188659667969e-06,
      "step": 9019,
      "training_step_time": 9.33916711807251
    },
    {
      "epoch": 8.602142333984375e-06,
      "grad_norm": 0.2580662965774536,
      "learning_rate": 9.955905016847197e-06,
      "loss": 0.0761,
      "step": 9020
    },
    {
      "epoch": 8.602142333984375e-06,
      "model_forward_time": 1.8270270824432373,
      "step": 9020
    },
    {
      "dynamics_loss_avg": 0.07484334986656904,
      "epoch": 8.602142333984375e-06,
      "step": 9020
    },
    {
      "action_loss_avg": 0.004975090874359012,
      "epoch": 8.602142333984375e-06,
      "step": 9020
    },
    {
      "epoch": 8.602142333984375e-06,
      "step": 9020,
      "training_step_time": 7.697951555252075
    },
    {
      "epoch": 8.603096008300782e-06,
      "model_forward_time": 1.840522289276123,
      "step": 9021
    },
    {
      "epoch": 8.603096008300782e-06,
      "step": 9021,
      "training_step_time": 6.770257234573364
    },
    {
      "epoch": 8.604049682617188e-06,
      "model_forward_time": 1.8399617671966553,
      "step": 9022
    },
    {
      "epoch": 8.604049682617188e-06,
      "step": 9022,
      "training_step_time": 6.776278495788574
    },
    {
      "epoch": 8.605003356933593e-06,
      "model_forward_time": 1.8279571533203125,
      "step": 9023
    },
    {
      "epoch": 8.605003356933593e-06,
      "step": 9023,
      "training_step_time": 6.763867616653442
    },
    {
      "epoch": 8.60595703125e-06,
      "model_forward_time": 1.841869831085205,
      "step": 9024
    },
    {
      "epoch": 8.60595703125e-06,
      "step": 9024,
      "training_step_time": 6.778587579727173
    },
    {
      "epoch": 8.606910705566407e-06,
      "model_forward_time": 1.8335280418395996,
      "step": 9025
    },
    {
      "epoch": 8.606910705566407e-06,
      "step": 9025,
      "training_step_time": 6.771299600601196
    },
    {
      "epoch": 8.607864379882812e-06,
      "model_forward_time": 1.835028886795044,
      "step": 9026
    },
    {
      "epoch": 8.607864379882812e-06,
      "step": 9026,
      "training_step_time": 6.724287271499634
    },
    {
      "epoch": 8.608818054199218e-06,
      "model_forward_time": 2.359232187271118,
      "step": 9027
    },
    {
      "epoch": 8.608818054199218e-06,
      "step": 9027,
      "training_step_time": 7.545799970626831
    },
    {
      "epoch": 8.609771728515625e-06,
      "model_forward_time": 1.833097219467163,
      "step": 9028
    },
    {
      "epoch": 8.609771728515625e-06,
      "step": 9028,
      "training_step_time": 10.213165283203125
    },
    {
      "epoch": 8.610725402832031e-06,
      "model_forward_time": 1.8389744758605957,
      "step": 9029
    },
    {
      "epoch": 8.610725402832031e-06,
      "step": 9029,
      "training_step_time": 7.752967119216919
    },
    {
      "epoch": 8.611679077148437e-06,
      "grad_norm": 0.39914074540138245,
      "learning_rate": 9.955685636354724e-06,
      "loss": 0.0713,
      "step": 9030
    },
    {
      "epoch": 8.611679077148437e-06,
      "model_forward_time": 1.8279294967651367,
      "step": 9030
    },
    {
      "dynamics_loss_avg": 0.06513136550784111,
      "epoch": 8.611679077148437e-06,
      "step": 9030
    },
    {
      "action_loss_avg": 0.00849516475573182,
      "epoch": 8.611679077148437e-06,
      "step": 9030
    },
    {
      "epoch": 8.611679077148437e-06,
      "step": 9030,
      "training_step_time": 7.10747766494751
    },
    {
      "epoch": 8.612632751464844e-06,
      "model_forward_time": 1.8263640403747559,
      "step": 9031
    },
    {
      "epoch": 8.612632751464844e-06,
      "step": 9031,
      "training_step_time": 6.762389898300171
    },
    {
      "epoch": 8.61358642578125e-06,
      "model_forward_time": 1.8378596305847168,
      "step": 9032
    },
    {
      "epoch": 8.61358642578125e-06,
      "step": 9032,
      "training_step_time": 6.782518625259399
    },
    {
      "epoch": 8.614540100097656e-06,
      "model_forward_time": 1.820603847503662,
      "step": 9033
    },
    {
      "epoch": 8.614540100097656e-06,
      "step": 9033,
      "training_step_time": 6.780923843383789
    },
    {
      "epoch": 8.615493774414063e-06,
      "model_forward_time": 1.8255608081817627,
      "step": 9034
    },
    {
      "epoch": 8.615493774414063e-06,
      "step": 9034,
      "training_step_time": 6.820714712142944
    },
    {
      "epoch": 8.616447448730469e-06,
      "model_forward_time": 1.8393208980560303,
      "step": 9035
    },
    {
      "epoch": 8.616447448730469e-06,
      "step": 9035,
      "training_step_time": 6.7798991203308105
    },
    {
      "epoch": 8.617401123046875e-06,
      "model_forward_time": 2.3295791149139404,
      "step": 9036
    },
    {
      "epoch": 8.617401123046875e-06,
      "step": 9036,
      "training_step_time": 8.025544166564941
    },
    {
      "epoch": 8.618354797363282e-06,
      "model_forward_time": 1.8267781734466553,
      "step": 9037
    },
    {
      "epoch": 8.618354797363282e-06,
      "step": 9037,
      "training_step_time": 11.139933109283447
    },
    {
      "epoch": 8.619308471679688e-06,
      "model_forward_time": 1.825566053390503,
      "step": 9038
    },
    {
      "epoch": 8.619308471679688e-06,
      "step": 9038,
      "training_step_time": 10.082842350006104
    },
    {
      "epoch": 8.620262145996093e-06,
      "model_forward_time": 1.8213121891021729,
      "step": 9039
    },
    {
      "epoch": 8.620262145996093e-06,
      "step": 9039,
      "training_step_time": 7.694298267364502
    },
    {
      "epoch": 8.6212158203125e-06,
      "grad_norm": 0.34338825941085815,
      "learning_rate": 9.95546571391592e-06,
      "loss": 0.0702,
      "step": 9040
    },
    {
      "epoch": 8.6212158203125e-06,
      "model_forward_time": 1.8146932125091553,
      "step": 9040
    },
    {
      "dynamics_loss_avg": 0.0704844169318676,
      "epoch": 8.6212158203125e-06,
      "step": 9040
    },
    {
      "action_loss_avg": 0.005071582819800824,
      "epoch": 8.6212158203125e-06,
      "step": 9040
    },
    {
      "epoch": 8.6212158203125e-06,
      "step": 9040,
      "training_step_time": 6.716348648071289
    },
    {
      "epoch": 8.622169494628907e-06,
      "model_forward_time": 1.8183014392852783,
      "step": 9041
    },
    {
      "epoch": 8.622169494628907e-06,
      "step": 9041,
      "training_step_time": 6.770513296127319
    },
    {
      "epoch": 8.623123168945312e-06,
      "model_forward_time": 1.8237667083740234,
      "step": 9042
    },
    {
      "epoch": 8.623123168945312e-06,
      "step": 9042,
      "training_step_time": 6.783904552459717
    },
    {
      "epoch": 8.624076843261718e-06,
      "model_forward_time": 1.829366683959961,
      "step": 9043
    },
    {
      "epoch": 8.624076843261718e-06,
      "step": 9043,
      "training_step_time": 6.748278856277466
    },
    {
      "epoch": 8.625030517578125e-06,
      "model_forward_time": 1.82188081741333,
      "step": 9044
    },
    {
      "epoch": 8.625030517578125e-06,
      "step": 9044,
      "training_step_time": 6.722125053405762
    },
    {
      "epoch": 8.625984191894531e-06,
      "model_forward_time": 2.3634450435638428,
      "step": 9045
    },
    {
      "epoch": 8.625984191894531e-06,
      "step": 9045,
      "training_step_time": 7.536246061325073
    },
    {
      "epoch": 8.626937866210937e-06,
      "model_forward_time": 1.816042184829712,
      "step": 9046
    },
    {
      "epoch": 8.626937866210937e-06,
      "step": 9046,
      "training_step_time": 9.672572612762451
    },
    {
      "epoch": 8.627891540527344e-06,
      "model_forward_time": 1.0952565670013428,
      "step": 9047
    },
    {
      "epoch": 8.627891540527344e-06,
      "step": 9047,
      "training_step_time": 8.939842224121094
    },
    {
      "epoch": 8.62884521484375e-06,
      "model_forward_time": 1.8204028606414795,
      "step": 9048
    },
    {
      "epoch": 8.62884521484375e-06,
      "step": 9048,
      "training_step_time": 7.821206092834473
    },
    {
      "epoch": 8.629798889160156e-06,
      "model_forward_time": 1.8245666027069092,
      "step": 9049
    },
    {
      "epoch": 8.629798889160156e-06,
      "step": 9049,
      "training_step_time": 6.895971059799194
    },
    {
      "epoch": 8.630752563476563e-06,
      "grad_norm": 0.2995575964450836,
      "learning_rate": 9.955245249554836e-06,
      "loss": 0.0654,
      "step": 9050
    },
    {
      "epoch": 8.630752563476563e-06,
      "model_forward_time": 1.8218355178833008,
      "step": 9050
    },
    {
      "dynamics_loss_avg": 0.05389820467680693,
      "epoch": 8.630752563476563e-06,
      "step": 9050
    },
    {
      "action_loss_avg": 0.007980561268050223,
      "epoch": 8.630752563476563e-06,
      "step": 9050
    },
    {
      "epoch": 8.630752563476563e-06,
      "step": 9050,
      "training_step_time": 6.9170145988464355
    },
    {
      "epoch": 8.631706237792969e-06,
      "model_forward_time": 1.8179218769073486,
      "step": 9051
    },
    {
      "epoch": 8.631706237792969e-06,
      "step": 9051,
      "training_step_time": 6.895543336868286
    },
    {
      "epoch": 8.632659912109375e-06,
      "model_forward_time": 1.822373628616333,
      "step": 9052
    },
    {
      "epoch": 8.632659912109375e-06,
      "step": 9052,
      "training_step_time": 6.938841104507446
    },
    {
      "epoch": 8.633613586425782e-06,
      "model_forward_time": 1.8215086460113525,
      "step": 9053
    },
    {
      "epoch": 8.633613586425782e-06,
      "step": 9053,
      "training_step_time": 6.9206671714782715
    },
    {
      "epoch": 8.634567260742188e-06,
      "model_forward_time": 1.8138964176177979,
      "step": 9054
    },
    {
      "epoch": 8.634567260742188e-06,
      "step": 9054,
      "training_step_time": 7.554381370544434
    },
    {
      "epoch": 8.635520935058593e-06,
      "model_forward_time": 2.350799322128296,
      "step": 9055
    },
    {
      "epoch": 8.635520935058593e-06,
      "step": 9055,
      "training_step_time": 8.783696174621582
    },
    {
      "epoch": 8.636474609375e-06,
      "model_forward_time": 1.820497751235962,
      "step": 9056
    },
    {
      "epoch": 8.636474609375e-06,
      "step": 9056,
      "training_step_time": 10.009294748306274
    },
    {
      "epoch": 8.637428283691407e-06,
      "model_forward_time": 1.8092801570892334,
      "step": 9057
    },
    {
      "epoch": 8.637428283691407e-06,
      "step": 9057,
      "training_step_time": 7.837953567504883
    },
    {
      "epoch": 8.638381958007812e-06,
      "model_forward_time": 1.8219103813171387,
      "step": 9058
    },
    {
      "epoch": 8.638381958007812e-06,
      "step": 9058,
      "training_step_time": 6.894080638885498
    },
    {
      "epoch": 8.639335632324218e-06,
      "model_forward_time": 1.8372910022735596,
      "step": 9059
    },
    {
      "epoch": 8.639335632324218e-06,
      "step": 9059,
      "training_step_time": 6.8853676319122314
    },
    {
      "epoch": 8.640289306640625e-06,
      "grad_norm": 0.23015721142292023,
      "learning_rate": 9.955024243295584e-06,
      "loss": 0.0666,
      "step": 9060
    },
    {
      "epoch": 8.640289306640625e-06,
      "model_forward_time": 1.82962965965271,
      "step": 9060
    },
    {
      "dynamics_loss_avg": 0.060802612081170083,
      "epoch": 8.640289306640625e-06,
      "step": 9060
    },
    {
      "action_loss_avg": 0.006227296020369977,
      "epoch": 8.640289306640625e-06,
      "step": 9060
    },
    {
      "epoch": 8.640289306640625e-06,
      "step": 9060,
      "training_step_time": 6.797602653503418
    },
    {
      "epoch": 8.641242980957031e-06,
      "model_forward_time": 1.8111803531646729,
      "step": 9061
    },
    {
      "epoch": 8.641242980957031e-06,
      "step": 9061,
      "training_step_time": 6.893156290054321
    },
    {
      "epoch": 8.642196655273437e-06,
      "model_forward_time": 1.4384362697601318,
      "step": 9062
    },
    {
      "epoch": 8.642196655273437e-06,
      "step": 9062,
      "training_step_time": 6.79716157913208
    },
    {
      "epoch": 8.643150329589844e-06,
      "model_forward_time": 1.8395419120788574,
      "step": 9063
    },
    {
      "epoch": 8.643150329589844e-06,
      "step": 9063,
      "training_step_time": 7.645580768585205
    },
    {
      "epoch": 8.64410400390625e-06,
      "model_forward_time": 2.36398983001709,
      "step": 9064
    },
    {
      "epoch": 8.64410400390625e-06,
      "step": 9064,
      "training_step_time": 8.842546463012695
    },
    {
      "epoch": 8.645057678222656e-06,
      "model_forward_time": 1.8309409618377686,
      "step": 9065
    },
    {
      "epoch": 8.645057678222656e-06,
      "step": 9065,
      "training_step_time": 10.107380867004395
    },
    {
      "epoch": 8.646011352539063e-06,
      "model_forward_time": 1.836660385131836,
      "step": 9066
    },
    {
      "epoch": 8.646011352539063e-06,
      "step": 9066,
      "training_step_time": 7.73321795463562
    },
    {
      "epoch": 8.646965026855469e-06,
      "model_forward_time": 1.8373832702636719,
      "step": 9067
    },
    {
      "epoch": 8.646965026855469e-06,
      "step": 9067,
      "training_step_time": 7.668973922729492
    },
    {
      "epoch": 8.647918701171875e-06,
      "model_forward_time": 1.8370773792266846,
      "step": 9068
    },
    {
      "epoch": 8.647918701171875e-06,
      "step": 9068,
      "training_step_time": 6.846689701080322
    },
    {
      "epoch": 8.648872375488282e-06,
      "model_forward_time": 1.834488868713379,
      "step": 9069
    },
    {
      "epoch": 8.648872375488282e-06,
      "step": 9069,
      "training_step_time": 6.866129398345947
    },
    {
      "epoch": 8.649826049804688e-06,
      "grad_norm": 0.18953241407871246,
      "learning_rate": 9.954802695162329e-06,
      "loss": 0.0704,
      "step": 9070
    },
    {
      "epoch": 8.649826049804688e-06,
      "model_forward_time": 1.8268728256225586,
      "step": 9070
    },
    {
      "dynamics_loss_avg": 0.051808744110167025,
      "epoch": 8.649826049804688e-06,
      "step": 9070
    },
    {
      "action_loss_avg": 0.002714193379506469,
      "epoch": 8.649826049804688e-06,
      "step": 9070
    },
    {
      "epoch": 8.649826049804688e-06,
      "step": 9070,
      "training_step_time": 6.865217208862305
    },
    {
      "epoch": 8.650779724121093e-06,
      "model_forward_time": 1.8265721797943115,
      "step": 9071
    },
    {
      "epoch": 8.650779724121093e-06,
      "step": 9071,
      "training_step_time": 6.792893648147583
    },
    {
      "epoch": 8.6517333984375e-06,
      "model_forward_time": 1.4612445831298828,
      "step": 9072
    },
    {
      "epoch": 8.6517333984375e-06,
      "step": 9072,
      "training_step_time": 7.677822828292847
    },
    {
      "epoch": 8.652687072753907e-06,
      "model_forward_time": 2.3588509559631348,
      "step": 9073
    },
    {
      "epoch": 8.652687072753907e-06,
      "step": 9073,
      "training_step_time": 7.787620544433594
    },
    {
      "epoch": 8.653640747070312e-06,
      "model_forward_time": 1.8358004093170166,
      "step": 9074
    },
    {
      "epoch": 8.653640747070312e-06,
      "step": 9074,
      "training_step_time": 8.840866565704346
    },
    {
      "epoch": 8.654594421386718e-06,
      "model_forward_time": 1.831346035003662,
      "step": 9075
    },
    {
      "epoch": 8.654594421386718e-06,
      "step": 9075,
      "training_step_time": 8.640663385391235
    },
    {
      "epoch": 8.655548095703125e-06,
      "model_forward_time": 1.8261818885803223,
      "step": 9076
    },
    {
      "epoch": 8.655548095703125e-06,
      "step": 9076,
      "training_step_time": 7.681056499481201
    },
    {
      "epoch": 8.656501770019531e-06,
      "model_forward_time": 1.82966947555542,
      "step": 9077
    },
    {
      "epoch": 8.656501770019531e-06,
      "step": 9077,
      "training_step_time": 6.856344223022461
    },
    {
      "epoch": 8.657455444335937e-06,
      "model_forward_time": 1.4689505100250244,
      "step": 9078
    },
    {
      "epoch": 8.657455444335937e-06,
      "step": 9078,
      "training_step_time": 6.855808734893799
    },
    {
      "epoch": 8.658409118652344e-06,
      "model_forward_time": 1.8366429805755615,
      "step": 9079
    },
    {
      "epoch": 8.658409118652344e-06,
      "step": 9079,
      "training_step_time": 6.840940237045288
    },
    {
      "epoch": 8.65936279296875e-06,
      "grad_norm": 0.19479501247406006,
      "learning_rate": 9.954580605179301e-06,
      "loss": 0.0741,
      "step": 9080
    },
    {
      "epoch": 8.65936279296875e-06,
      "model_forward_time": 1.824408769607544,
      "step": 9080
    },
    {
      "dynamics_loss_avg": 0.06223271004855633,
      "epoch": 8.65936279296875e-06,
      "step": 9080
    },
    {
      "action_loss_avg": 0.0034364891704171895,
      "epoch": 8.65936279296875e-06,
      "step": 9080
    },
    {
      "epoch": 8.65936279296875e-06,
      "step": 9080,
      "training_step_time": 6.842088460922241
    },
    {
      "epoch": 8.660316467285156e-06,
      "model_forward_time": 1.8349494934082031,
      "step": 9081
    },
    {
      "epoch": 8.660316467285156e-06,
      "step": 9081,
      "training_step_time": 7.714099645614624
    },
    {
      "epoch": 8.661270141601563e-06,
      "model_forward_time": 1.8366916179656982,
      "step": 9082
    },
    {
      "epoch": 8.661270141601563e-06,
      "step": 9082,
      "training_step_time": 7.697537422180176
    },
    {
      "epoch": 8.662223815917969e-06,
      "model_forward_time": 2.3919789791107178,
      "step": 9083
    },
    {
      "epoch": 8.662223815917969e-06,
      "step": 9083,
      "training_step_time": 8.758986473083496
    },
    {
      "epoch": 8.663177490234375e-06,
      "model_forward_time": 1.8108980655670166,
      "step": 9084
    },
    {
      "epoch": 8.663177490234375e-06,
      "step": 9084,
      "training_step_time": 7.746216535568237
    },
    {
      "epoch": 8.664131164550782e-06,
      "model_forward_time": 1.8232066631317139,
      "step": 9085
    },
    {
      "epoch": 8.664131164550782e-06,
      "step": 9085,
      "training_step_time": 7.6879072189331055
    },
    {
      "epoch": 8.665084838867188e-06,
      "model_forward_time": 1.8244118690490723,
      "step": 9086
    },
    {
      "epoch": 8.665084838867188e-06,
      "step": 9086,
      "training_step_time": 7.078572511672974
    },
    {
      "epoch": 8.666038513183594e-06,
      "model_forward_time": 1.8239052295684814,
      "step": 9087
    },
    {
      "epoch": 8.666038513183594e-06,
      "step": 9087,
      "training_step_time": 6.843229293823242
    },
    {
      "epoch": 8.6669921875e-06,
      "model_forward_time": 1.8215136528015137,
      "step": 9088
    },
    {
      "epoch": 8.6669921875e-06,
      "step": 9088,
      "training_step_time": 6.838992118835449
    },
    {
      "epoch": 8.667945861816407e-06,
      "model_forward_time": 1.8186829090118408,
      "step": 9089
    },
    {
      "epoch": 8.667945861816407e-06,
      "step": 9089,
      "training_step_time": 6.869788646697998
    },
    {
      "epoch": 8.668899536132812e-06,
      "grad_norm": 0.18721389770507812,
      "learning_rate": 9.954357973370788e-06,
      "loss": 0.0714,
      "step": 9090
    },
    {
      "epoch": 8.668899536132812e-06,
      "model_forward_time": 1.8166565895080566,
      "step": 9090
    },
    {
      "dynamics_loss_avg": 0.07371114790439606,
      "epoch": 8.668899536132812e-06,
      "step": 9090
    },
    {
      "action_loss_avg": 0.0076407338725402955,
      "epoch": 8.668899536132812e-06,
      "step": 9090
    },
    {
      "epoch": 8.668899536132812e-06,
      "step": 9090,
      "training_step_time": 7.630011796951294
    },
    {
      "epoch": 8.669853210449218e-06,
      "model_forward_time": 1.815288782119751,
      "step": 9091
    },
    {
      "epoch": 8.669853210449218e-06,
      "step": 9091,
      "training_step_time": 8.483943700790405
    },
    {
      "epoch": 8.670806884765626e-06,
      "model_forward_time": 2.3513031005859375,
      "step": 9092
    },
    {
      "epoch": 8.670806884765626e-06,
      "step": 9092,
      "training_step_time": 8.343307971954346
    },
    {
      "epoch": 8.671760559082031e-06,
      "model_forward_time": 1.8223605155944824,
      "step": 9093
    },
    {
      "epoch": 8.671760559082031e-06,
      "step": 9093,
      "training_step_time": 9.480664491653442
    },
    {
      "epoch": 8.672714233398437e-06,
      "model_forward_time": 1.8271868228912354,
      "step": 9094
    },
    {
      "epoch": 8.672714233398437e-06,
      "step": 9094,
      "training_step_time": 7.693530797958374
    },
    {
      "epoch": 8.673667907714844e-06,
      "model_forward_time": 1.8188509941101074,
      "step": 9095
    },
    {
      "epoch": 8.673667907714844e-06,
      "step": 9095,
      "training_step_time": 7.087047100067139
    },
    {
      "epoch": 8.67462158203125e-06,
      "model_forward_time": 1.8182299137115479,
      "step": 9096
    },
    {
      "epoch": 8.67462158203125e-06,
      "step": 9096,
      "training_step_time": 6.839024066925049
    },
    {
      "epoch": 8.675575256347656e-06,
      "model_forward_time": 1.8186068534851074,
      "step": 9097
    },
    {
      "epoch": 8.675575256347656e-06,
      "step": 9097,
      "training_step_time": 6.9096879959106445
    },
    {
      "epoch": 8.676528930664063e-06,
      "model_forward_time": 1.8225886821746826,
      "step": 9098
    },
    {
      "epoch": 8.676528930664063e-06,
      "step": 9098,
      "training_step_time": 6.878369092941284
    },
    {
      "epoch": 8.677482604980469e-06,
      "model_forward_time": 1.8206233978271484,
      "step": 9099
    },
    {
      "epoch": 8.677482604980469e-06,
      "step": 9099,
      "training_step_time": 7.61220121383667
    },
    {
      "epoch": 8.678436279296875e-06,
      "grad_norm": 0.2128855586051941,
      "learning_rate": 9.954134799761134e-06,
      "loss": 0.0738,
      "step": 9100
    },
    {
      "epoch": 8.678436279296875e-06,
      "model_forward_time": 1.819016933441162,
      "step": 9100
    },
    {
      "dynamics_loss_avg": 0.06541473213583231,
      "epoch": 8.678436279296875e-06,
      "step": 9100
    },
    {
      "action_loss_avg": 0.0069958990905433895,
      "epoch": 8.678436279296875e-06,
      "step": 9100
    },
    {
      "epoch": 8.678436279296875e-06,
      "step": 9100,
      "training_step_time": 7.647746801376343
    },
    {
      "epoch": 8.679389953613282e-06,
      "model_forward_time": 2.3624846935272217,
      "step": 9101
    },
    {
      "epoch": 8.679389953613282e-06,
      "step": 9101,
      "training_step_time": 8.612783193588257
    },
    {
      "epoch": 8.680343627929688e-06,
      "model_forward_time": 1.8211874961853027,
      "step": 9102
    },
    {
      "epoch": 8.680343627929688e-06,
      "step": 9102,
      "training_step_time": 11.292485475540161
    },
    {
      "epoch": 8.681297302246094e-06,
      "model_forward_time": 1.824066162109375,
      "step": 9103
    },
    {
      "epoch": 8.681297302246094e-06,
      "step": 9103,
      "training_step_time": 7.756681442260742
    },
    {
      "epoch": 8.6822509765625e-06,
      "model_forward_time": 1.8321497440338135,
      "step": 9104
    },
    {
      "epoch": 8.6822509765625e-06,
      "step": 9104,
      "training_step_time": 7.619772434234619
    },
    {
      "epoch": 8.683204650878907e-06,
      "model_forward_time": 1.821904182434082,
      "step": 9105
    },
    {
      "epoch": 8.683204650878907e-06,
      "step": 9105,
      "training_step_time": 6.842912435531616
    },
    {
      "epoch": 8.684158325195312e-06,
      "model_forward_time": 1.7975890636444092,
      "step": 9106
    },
    {
      "epoch": 8.684158325195312e-06,
      "step": 9106,
      "training_step_time": 6.918797254562378
    },
    {
      "epoch": 8.685111999511718e-06,
      "model_forward_time": 1.827354907989502,
      "step": 9107
    },
    {
      "epoch": 8.685111999511718e-06,
      "step": 9107,
      "training_step_time": 6.8844380378723145
    },
    {
      "epoch": 8.686065673828126e-06,
      "model_forward_time": 1.8265647888183594,
      "step": 9108
    },
    {
      "epoch": 8.686065673828126e-06,
      "step": 9108,
      "training_step_time": 6.871911525726318
    },
    {
      "epoch": 8.687019348144531e-06,
      "model_forward_time": 1.8213427066802979,
      "step": 9109
    },
    {
      "epoch": 8.687019348144531e-06,
      "step": 9109,
      "training_step_time": 7.630924463272095
    },
    {
      "epoch": 8.687973022460937e-06,
      "grad_norm": 0.2900746166706085,
      "learning_rate": 9.953911084374747e-06,
      "loss": 0.0755,
      "step": 9110
    },
    {
      "epoch": 8.687973022460937e-06,
      "model_forward_time": 2.372999429702759,
      "step": 9110
    },
    {
      "dynamics_loss_avg": 0.07145184837281704,
      "epoch": 8.687973022460937e-06,
      "step": 9110
    },
    {
      "action_loss_avg": 0.0051424498436972495,
      "epoch": 8.687973022460937e-06,
      "step": 9110
    },
    {
      "epoch": 8.687973022460937e-06,
      "step": 9110,
      "training_step_time": 9.51293683052063
    },
    {
      "epoch": 8.688926696777344e-06,
      "model_forward_time": 1.819166660308838,
      "step": 9111
    },
    {
      "epoch": 8.688926696777344e-06,
      "step": 9111,
      "training_step_time": 10.468295812606812
    },
    {
      "epoch": 8.68988037109375e-06,
      "model_forward_time": 1.8186850547790527,
      "step": 9112
    },
    {
      "epoch": 8.68988037109375e-06,
      "step": 9112,
      "training_step_time": 8.559389114379883
    },
    {
      "epoch": 8.690834045410156e-06,
      "model_forward_time": 1.8170480728149414,
      "step": 9113
    },
    {
      "epoch": 8.690834045410156e-06,
      "step": 9113,
      "training_step_time": 7.6026201248168945
    },
    {
      "epoch": 8.691787719726563e-06,
      "model_forward_time": 1.8152668476104736,
      "step": 9114
    },
    {
      "epoch": 8.691787719726563e-06,
      "step": 9114,
      "training_step_time": 6.848677158355713
    },
    {
      "epoch": 8.692741394042969e-06,
      "model_forward_time": 1.8256211280822754,
      "step": 9115
    },
    {
      "epoch": 8.692741394042969e-06,
      "step": 9115,
      "training_step_time": 6.84471583366394
    },
    {
      "epoch": 8.693695068359375e-06,
      "model_forward_time": 1.815493106842041,
      "step": 9116
    },
    {
      "epoch": 8.693695068359375e-06,
      "step": 9116,
      "training_step_time": 6.8340537548065186
    },
    {
      "epoch": 8.694648742675782e-06,
      "model_forward_time": 1.8055591583251953,
      "step": 9117
    },
    {
      "epoch": 8.694648742675782e-06,
      "step": 9117,
      "training_step_time": 6.83577561378479
    },
    {
      "epoch": 8.695602416992188e-06,
      "model_forward_time": 1.8231384754180908,
      "step": 9118
    },
    {
      "epoch": 8.695602416992188e-06,
      "step": 9118,
      "training_step_time": 7.65014910697937
    },
    {
      "epoch": 8.696556091308594e-06,
      "model_forward_time": 1.8178861141204834,
      "step": 9119
    },
    {
      "epoch": 8.696556091308594e-06,
      "step": 9119,
      "training_step_time": 9.501493215560913
    },
    {
      "epoch": 8.697509765625e-06,
      "grad_norm": 0.2991759777069092,
      "learning_rate": 9.953686827236092e-06,
      "loss": 0.0733,
      "step": 9120
    },
    {
      "epoch": 8.697509765625e-06,
      "model_forward_time": 1.815690517425537,
      "step": 9120
    },
    {
      "dynamics_loss_avg": 0.0688159516081214,
      "epoch": 8.697509765625e-06,
      "step": 9120
    },
    {
      "action_loss_avg": 0.004716557508800179,
      "epoch": 8.697509765625e-06,
      "step": 9120
    },
    {
      "epoch": 8.697509765625e-06,
      "step": 9120,
      "training_step_time": 8.799266576766968
    },
    {
      "epoch": 8.698463439941407e-06,
      "model_forward_time": 1.827188491821289,
      "step": 9121
    },
    {
      "epoch": 8.698463439941407e-06,
      "step": 9121,
      "training_step_time": 9.309930562973022
    },
    {
      "epoch": 8.699417114257812e-06,
      "model_forward_time": 1.8199536800384521,
      "step": 9122
    },
    {
      "epoch": 8.699417114257812e-06,
      "step": 9122,
      "training_step_time": 7.6939921379089355
    },
    {
      "epoch": 8.700370788574218e-06,
      "model_forward_time": 1.834285020828247,
      "step": 9123
    },
    {
      "epoch": 8.700370788574218e-06,
      "step": 9123,
      "training_step_time": 6.931525945663452
    },
    {
      "epoch": 8.701324462890626e-06,
      "model_forward_time": 1.8215763568878174,
      "step": 9124
    },
    {
      "epoch": 8.701324462890626e-06,
      "step": 9124,
      "training_step_time": 6.856589078903198
    },
    {
      "epoch": 8.702278137207031e-06,
      "model_forward_time": 1.8055224418640137,
      "step": 9125
    },
    {
      "epoch": 8.702278137207031e-06,
      "step": 9125,
      "training_step_time": 6.8856916427612305
    },
    {
      "epoch": 8.703231811523437e-06,
      "model_forward_time": 1.8103761672973633,
      "step": 9126
    },
    {
      "epoch": 8.703231811523437e-06,
      "step": 9126,
      "training_step_time": 6.904972314834595
    },
    {
      "epoch": 8.704185485839844e-06,
      "model_forward_time": 1.8229527473449707,
      "step": 9127
    },
    {
      "epoch": 8.704185485839844e-06,
      "step": 9127,
      "training_step_time": 7.652283430099487
    },
    {
      "epoch": 8.70513916015625e-06,
      "model_forward_time": 1.8315849304199219,
      "step": 9128
    },
    {
      "epoch": 8.70513916015625e-06,
      "step": 9128,
      "training_step_time": 8.236589431762695
    },
    {
      "epoch": 8.706092834472656e-06,
      "model_forward_time": 2.3500921726226807,
      "step": 9129
    },
    {
      "epoch": 8.706092834472656e-06,
      "step": 9129,
      "training_step_time": 7.806427478790283
    },
    {
      "epoch": 8.707046508789063e-06,
      "grad_norm": 0.23497696220874786,
      "learning_rate": 9.953462028369695e-06,
      "loss": 0.0755,
      "step": 9130
    },
    {
      "epoch": 8.707046508789063e-06,
      "model_forward_time": 1.8132750988006592,
      "step": 9130
    },
    {
      "dynamics_loss_avg": 0.05672171311452985,
      "epoch": 8.707046508789063e-06,
      "step": 9130
    },
    {
      "action_loss_avg": 0.005519369727699086,
      "epoch": 8.707046508789063e-06,
      "step": 9130
    },
    {
      "epoch": 8.707046508789063e-06,
      "step": 9130,
      "training_step_time": 11.264517307281494
    },
    {
      "epoch": 8.708000183105469e-06,
      "model_forward_time": 1.8134779930114746,
      "step": 9131
    },
    {
      "epoch": 8.708000183105469e-06,
      "step": 9131,
      "training_step_time": 7.742802143096924
    },
    {
      "epoch": 8.708953857421875e-06,
      "model_forward_time": 1.8199522495269775,
      "step": 9132
    },
    {
      "epoch": 8.708953857421875e-06,
      "step": 9132,
      "training_step_time": 7.51862645149231
    },
    {
      "epoch": 8.70990753173828e-06,
      "model_forward_time": 1.8196299076080322,
      "step": 9133
    },
    {
      "epoch": 8.70990753173828e-06,
      "step": 9133,
      "training_step_time": 6.950369358062744
    },
    {
      "epoch": 8.710861206054688e-06,
      "model_forward_time": 1.8224267959594727,
      "step": 9134
    },
    {
      "epoch": 8.710861206054688e-06,
      "step": 9134,
      "training_step_time": 6.868119478225708
    },
    {
      "epoch": 8.711814880371094e-06,
      "model_forward_time": 1.8153572082519531,
      "step": 9135
    },
    {
      "epoch": 8.711814880371094e-06,
      "step": 9135,
      "training_step_time": 6.885357856750488
    },
    {
      "epoch": 8.7127685546875e-06,
      "model_forward_time": 1.8277959823608398,
      "step": 9136
    },
    {
      "epoch": 8.7127685546875e-06,
      "step": 9136,
      "training_step_time": 7.7132885456085205
    },
    {
      "epoch": 8.713722229003907e-06,
      "model_forward_time": 1.8157329559326172,
      "step": 9137
    },
    {
      "epoch": 8.713722229003907e-06,
      "step": 9137,
      "training_step_time": 8.364421844482422
    },
    {
      "epoch": 8.714675903320312e-06,
      "model_forward_time": 2.3510677814483643,
      "step": 9138
    },
    {
      "epoch": 8.714675903320312e-06,
      "step": 9138,
      "training_step_time": 9.61945629119873
    },
    {
      "epoch": 8.715629577636718e-06,
      "model_forward_time": 1.8188371658325195,
      "step": 9139
    },
    {
      "epoch": 8.715629577636718e-06,
      "step": 9139,
      "training_step_time": 8.897183418273926
    },
    {
      "epoch": 8.716583251953126e-06,
      "grad_norm": 0.3955620527267456,
      "learning_rate": 9.953236687800137e-06,
      "loss": 0.0773,
      "step": 9140
    },
    {
      "epoch": 8.716583251953126e-06,
      "model_forward_time": 1.8167247772216797,
      "step": 9140
    },
    {
      "dynamics_loss_avg": 0.07308409791439771,
      "epoch": 8.716583251953126e-06,
      "step": 9140
    },
    {
      "action_loss_avg": 0.0072623312880750746,
      "epoch": 8.716583251953126e-06,
      "step": 9140
    },
    {
      "epoch": 8.716583251953126e-06,
      "step": 9140,
      "training_step_time": 8.22561001777649
    },
    {
      "epoch": 8.717536926269531e-06,
      "model_forward_time": 1.8236122131347656,
      "step": 9141
    },
    {
      "epoch": 8.717536926269531e-06,
      "step": 9141,
      "training_step_time": 7.561798810958862
    },
    {
      "epoch": 8.718490600585937e-06,
      "model_forward_time": 1.8173844814300537,
      "step": 9142
    },
    {
      "epoch": 8.718490600585937e-06,
      "step": 9142,
      "training_step_time": 6.819074392318726
    },
    {
      "epoch": 8.719444274902344e-06,
      "model_forward_time": 1.8374717235565186,
      "step": 9143
    },
    {
      "epoch": 8.719444274902344e-06,
      "step": 9143,
      "training_step_time": 6.834112644195557
    },
    {
      "epoch": 8.72039794921875e-06,
      "model_forward_time": 1.8356072902679443,
      "step": 9144
    },
    {
      "epoch": 8.72039794921875e-06,
      "step": 9144,
      "training_step_time": 6.825126886367798
    },
    {
      "epoch": 8.721351623535156e-06,
      "model_forward_time": 1.8051187992095947,
      "step": 9145
    },
    {
      "epoch": 8.721351623535156e-06,
      "step": 9145,
      "training_step_time": 7.686513185501099
    },
    {
      "epoch": 8.722305297851563e-06,
      "model_forward_time": 1.8323616981506348,
      "step": 9146
    },
    {
      "epoch": 8.722305297851563e-06,
      "step": 9146,
      "training_step_time": 7.654895067214966
    },
    {
      "epoch": 8.723258972167969e-06,
      "model_forward_time": 2.337496757507324,
      "step": 9147
    },
    {
      "epoch": 8.723258972167969e-06,
      "step": 9147,
      "training_step_time": 7.724773168563843
    },
    {
      "epoch": 8.724212646484375e-06,
      "model_forward_time": 1.8175077438354492,
      "step": 9148
    },
    {
      "epoch": 8.724212646484375e-06,
      "step": 9148,
      "training_step_time": 9.529601335525513
    },
    {
      "epoch": 8.72516632080078e-06,
      "model_forward_time": 1.8261456489562988,
      "step": 9149
    },
    {
      "epoch": 8.72516632080078e-06,
      "step": 9149,
      "training_step_time": 8.335064172744751
    },
    {
      "epoch": 8.726119995117188e-06,
      "grad_norm": 0.21164673566818237,
      "learning_rate": 9.953010805552062e-06,
      "loss": 0.08,
      "step": 9150
    },
    {
      "epoch": 8.726119995117188e-06,
      "model_forward_time": 1.4378156661987305,
      "step": 9150
    },
    {
      "dynamics_loss_avg": 0.05696840193122625,
      "epoch": 8.726119995117188e-06,
      "step": 9150
    },
    {
      "action_loss_avg": 0.005956746189622208,
      "epoch": 8.726119995117188e-06,
      "step": 9150
    },
    {
      "epoch": 8.726119995117188e-06,
      "step": 9150,
      "training_step_time": 7.677815198898315
    },
    {
      "epoch": 8.727073669433594e-06,
      "model_forward_time": 1.8242473602294922,
      "step": 9151
    },
    {
      "epoch": 8.727073669433594e-06,
      "step": 9151,
      "training_step_time": 6.863683462142944
    },
    {
      "epoch": 8.72802734375e-06,
      "model_forward_time": 1.8305351734161377,
      "step": 9152
    },
    {
      "epoch": 8.72802734375e-06,
      "step": 9152,
      "training_step_time": 6.845573425292969
    },
    {
      "epoch": 8.728981018066407e-06,
      "model_forward_time": 1.8270213603973389,
      "step": 9153
    },
    {
      "epoch": 8.728981018066407e-06,
      "step": 9153,
      "training_step_time": 6.829277276992798
    },
    {
      "epoch": 8.729934692382812e-06,
      "model_forward_time": 1.8227965831756592,
      "step": 9154
    },
    {
      "epoch": 8.729934692382812e-06,
      "step": 9154,
      "training_step_time": 7.617801904678345
    },
    {
      "epoch": 8.730888366699218e-06,
      "model_forward_time": 1.8232858180999756,
      "step": 9155
    },
    {
      "epoch": 8.730888366699218e-06,
      "step": 9155,
      "training_step_time": 7.598994016647339
    },
    {
      "epoch": 8.731842041015626e-06,
      "model_forward_time": 2.3611438274383545,
      "step": 9156
    },
    {
      "epoch": 8.731842041015626e-06,
      "step": 9156,
      "training_step_time": 8.626266241073608
    },
    {
      "epoch": 8.732795715332031e-06,
      "model_forward_time": 1.7952549457550049,
      "step": 9157
    },
    {
      "epoch": 8.732795715332031e-06,
      "step": 9157,
      "training_step_time": 9.507625102996826
    },
    {
      "epoch": 8.733749389648437e-06,
      "model_forward_time": 1.8126637935638428,
      "step": 9158
    },
    {
      "epoch": 8.733749389648437e-06,
      "step": 9158,
      "training_step_time": 8.714643001556396
    },
    {
      "epoch": 8.734703063964844e-06,
      "model_forward_time": 1.8039836883544922,
      "step": 9159
    },
    {
      "epoch": 8.734703063964844e-06,
      "step": 9159,
      "training_step_time": 7.724013805389404
    },
    {
      "epoch": 8.73565673828125e-06,
      "grad_norm": 0.3720652759075165,
      "learning_rate": 9.952784381650171e-06,
      "loss": 0.0775,
      "step": 9160
    },
    {
      "epoch": 8.73565673828125e-06,
      "model_forward_time": 1.084686040878296,
      "step": 9160
    },
    {
      "dynamics_loss_avg": 0.07943685539066792,
      "epoch": 8.73565673828125e-06,
      "step": 9160
    },
    {
      "action_loss_avg": 0.007569929538294673,
      "epoch": 8.73565673828125e-06,
      "step": 9160
    },
    {
      "epoch": 8.73565673828125e-06,
      "step": 9160,
      "training_step_time": 6.86423659324646
    },
    {
      "epoch": 8.736610412597656e-06,
      "model_forward_time": 1.8093211650848389,
      "step": 9161
    },
    {
      "epoch": 8.736610412597656e-06,
      "step": 9161,
      "training_step_time": 6.929413318634033
    },
    {
      "epoch": 8.737564086914063e-06,
      "model_forward_time": 1.8156356811523438,
      "step": 9162
    },
    {
      "epoch": 8.737564086914063e-06,
      "step": 9162,
      "training_step_time": 6.867796182632446
    },
    {
      "epoch": 8.738517761230469e-06,
      "model_forward_time": 1.8157446384429932,
      "step": 9163
    },
    {
      "epoch": 8.738517761230469e-06,
      "step": 9163,
      "training_step_time": 7.627861499786377
    },
    {
      "epoch": 8.739471435546875e-06,
      "model_forward_time": 1.8146488666534424,
      "step": 9164
    },
    {
      "epoch": 8.739471435546875e-06,
      "step": 9164,
      "training_step_time": 7.555302858352661
    },
    {
      "epoch": 8.74042510986328e-06,
      "model_forward_time": 1.8228099346160889,
      "step": 9165
    },
    {
      "epoch": 8.74042510986328e-06,
      "step": 9165,
      "training_step_time": 7.6291162967681885
    },
    {
      "epoch": 8.741378784179688e-06,
      "model_forward_time": 2.3360750675201416,
      "step": 9166
    },
    {
      "epoch": 8.741378784179688e-06,
      "step": 9166,
      "training_step_time": 8.280094623565674
    },
    {
      "epoch": 8.742332458496094e-06,
      "model_forward_time": 1.8193292617797852,
      "step": 9167
    },
    {
      "epoch": 8.742332458496094e-06,
      "step": 9167,
      "training_step_time": 8.67728853225708
    },
    {
      "epoch": 8.7432861328125e-06,
      "model_forward_time": 1.8063247203826904,
      "step": 9168
    },
    {
      "epoch": 8.7432861328125e-06,
      "step": 9168,
      "training_step_time": 7.837411880493164
    },
    {
      "epoch": 8.744239807128907e-06,
      "model_forward_time": 1.810209035873413,
      "step": 9169
    },
    {
      "epoch": 8.744239807128907e-06,
      "step": 9169,
      "training_step_time": 6.892027378082275
    },
    {
      "epoch": 8.745193481445312e-06,
      "grad_norm": 0.34332892298698425,
      "learning_rate": 9.952557416119226e-06,
      "loss": 0.075,
      "step": 9170
    },
    {
      "epoch": 8.745193481445312e-06,
      "model_forward_time": 1.8044562339782715,
      "step": 9170
    },
    {
      "dynamics_loss_avg": 0.05963800735771656,
      "epoch": 8.745193481445312e-06,
      "step": 9170
    },
    {
      "action_loss_avg": 0.005991264153271914,
      "epoch": 8.745193481445312e-06,
      "step": 9170
    },
    {
      "epoch": 8.745193481445312e-06,
      "step": 9170,
      "training_step_time": 6.862802743911743
    },
    {
      "epoch": 8.746147155761718e-06,
      "model_forward_time": 1.8095223903656006,
      "step": 9171
    },
    {
      "epoch": 8.746147155761718e-06,
      "step": 9171,
      "training_step_time": 6.861574172973633
    },
    {
      "epoch": 8.747100830078126e-06,
      "model_forward_time": 1.8034923076629639,
      "step": 9172
    },
    {
      "epoch": 8.747100830078126e-06,
      "step": 9172,
      "training_step_time": 7.35526967048645
    },
    {
      "epoch": 8.748054504394531e-06,
      "model_forward_time": 1.807427167892456,
      "step": 9173
    },
    {
      "epoch": 8.748054504394531e-06,
      "step": 9173,
      "training_step_time": 7.696879148483276
    },
    {
      "epoch": 8.749008178710937e-06,
      "model_forward_time": 1.804274559020996,
      "step": 9174
    },
    {
      "epoch": 8.749008178710937e-06,
      "step": 9174,
      "training_step_time": 8.756288528442383
    },
    {
      "epoch": 8.749961853027344e-06,
      "model_forward_time": 2.3250744342803955,
      "step": 9175
    },
    {
      "epoch": 8.749961853027344e-06,
      "step": 9175,
      "training_step_time": 11.737082242965698
    },
    {
      "epoch": 8.75091552734375e-06,
      "model_forward_time": 1.8159806728363037,
      "step": 9176
    },
    {
      "epoch": 8.75091552734375e-06,
      "step": 9176,
      "training_step_time": 7.867572069168091
    },
    {
      "epoch": 8.751869201660156e-06,
      "model_forward_time": 1.8113329410552979,
      "step": 9177
    },
    {
      "epoch": 8.751869201660156e-06,
      "step": 9177,
      "training_step_time": 8.27201533317566
    },
    {
      "epoch": 8.752822875976563e-06,
      "model_forward_time": 1.7997126579284668,
      "step": 9178
    },
    {
      "epoch": 8.752822875976563e-06,
      "step": 9178,
      "training_step_time": 7.570887804031372
    },
    {
      "epoch": 8.753776550292969e-06,
      "model_forward_time": 1.8264496326446533,
      "step": 9179
    },
    {
      "epoch": 8.753776550292969e-06,
      "step": 9179,
      "training_step_time": 6.855039119720459
    },
    {
      "epoch": 8.754730224609375e-06,
      "grad_norm": 0.36590707302093506,
      "learning_rate": 9.952329908984052e-06,
      "loss": 0.0665,
      "step": 9180
    },
    {
      "epoch": 8.754730224609375e-06,
      "model_forward_time": 1.8036048412322998,
      "step": 9180
    },
    {
      "dynamics_loss_avg": 0.07388156615197658,
      "epoch": 8.754730224609375e-06,
      "step": 9180
    },
    {
      "action_loss_avg": 0.0075855248142033815,
      "epoch": 8.754730224609375e-06,
      "step": 9180
    },
    {
      "epoch": 8.754730224609375e-06,
      "step": 9180,
      "training_step_time": 6.877025604248047
    },
    {
      "epoch": 8.75568389892578e-06,
      "model_forward_time": 1.8212566375732422,
      "step": 9181
    },
    {
      "epoch": 8.75568389892578e-06,
      "step": 9181,
      "training_step_time": 7.3396196365356445
    },
    {
      "epoch": 8.756637573242188e-06,
      "model_forward_time": 1.8208680152893066,
      "step": 9182
    },
    {
      "epoch": 8.756637573242188e-06,
      "step": 9182,
      "training_step_time": 7.665120363235474
    },
    {
      "epoch": 8.757591247558594e-06,
      "model_forward_time": 1.8121864795684814,
      "step": 9183
    },
    {
      "epoch": 8.757591247558594e-06,
      "step": 9183,
      "training_step_time": 7.7284886837005615
    },
    {
      "epoch": 8.758544921875e-06,
      "model_forward_time": 2.357741594314575,
      "step": 9184
    },
    {
      "epoch": 8.758544921875e-06,
      "step": 9184,
      "training_step_time": 7.8581438064575195
    },
    {
      "epoch": 8.759498596191407e-06,
      "model_forward_time": 1.8289618492126465,
      "step": 9185
    },
    {
      "epoch": 8.759498596191407e-06,
      "step": 9185,
      "training_step_time": 10.569305181503296
    },
    {
      "epoch": 8.760452270507812e-06,
      "model_forward_time": 1.8092632293701172,
      "step": 9186
    },
    {
      "epoch": 8.760452270507812e-06,
      "step": 9186,
      "training_step_time": 8.578906774520874
    },
    {
      "epoch": 8.761405944824218e-06,
      "model_forward_time": 1.8217852115631104,
      "step": 9187
    },
    {
      "epoch": 8.761405944824218e-06,
      "step": 9187,
      "training_step_time": 7.66338586807251
    },
    {
      "epoch": 8.762359619140626e-06,
      "model_forward_time": 1.8228161334991455,
      "step": 9188
    },
    {
      "epoch": 8.762359619140626e-06,
      "step": 9188,
      "training_step_time": 6.885669469833374
    },
    {
      "epoch": 8.763313293457031e-06,
      "model_forward_time": 1.8134019374847412,
      "step": 9189
    },
    {
      "epoch": 8.763313293457031e-06,
      "step": 9189,
      "training_step_time": 6.8509180545806885
    },
    {
      "epoch": 8.764266967773437e-06,
      "grad_norm": 0.26486125588417053,
      "learning_rate": 9.952101860269521e-06,
      "loss": 0.07,
      "step": 9190
    },
    {
      "epoch": 8.764266967773437e-06,
      "model_forward_time": 1.8167407512664795,
      "step": 9190
    },
    {
      "dynamics_loss_avg": 0.06780160218477249,
      "epoch": 8.764266967773437e-06,
      "step": 9190
    },
    {
      "action_loss_avg": 0.004004630737472326,
      "epoch": 8.764266967773437e-06,
      "step": 9190
    },
    {
      "epoch": 8.764266967773437e-06,
      "step": 9190,
      "training_step_time": 7.302566289901733
    },
    {
      "epoch": 8.765220642089844e-06,
      "model_forward_time": 1.8055825233459473,
      "step": 9191
    },
    {
      "epoch": 8.765220642089844e-06,
      "step": 9191,
      "training_step_time": 7.57135272026062
    },
    {
      "epoch": 8.76617431640625e-06,
      "model_forward_time": 1.8184189796447754,
      "step": 9192
    },
    {
      "epoch": 8.76617431640625e-06,
      "step": 9192,
      "training_step_time": 7.710114240646362
    },
    {
      "epoch": 8.767127990722656e-06,
      "model_forward_time": 1.8116059303283691,
      "step": 9193
    },
    {
      "epoch": 8.767127990722656e-06,
      "step": 9193,
      "training_step_time": 10.206883907318115
    },
    {
      "epoch": 8.768081665039063e-06,
      "model_forward_time": 1.801914930343628,
      "step": 9194
    },
    {
      "epoch": 8.768081665039063e-06,
      "step": 9194,
      "training_step_time": 8.758943557739258
    },
    {
      "epoch": 8.769035339355469e-06,
      "model_forward_time": 1.8217113018035889,
      "step": 9195
    },
    {
      "epoch": 8.769035339355469e-06,
      "step": 9195,
      "training_step_time": 7.7323691844940186
    },
    {
      "epoch": 8.769989013671875e-06,
      "model_forward_time": 1.8154489994049072,
      "step": 9196
    },
    {
      "epoch": 8.769989013671875e-06,
      "step": 9196,
      "training_step_time": 7.742766618728638
    },
    {
      "epoch": 8.77094268798828e-06,
      "model_forward_time": 1.819448709487915,
      "step": 9197
    },
    {
      "epoch": 8.77094268798828e-06,
      "step": 9197,
      "training_step_time": 7.48533821105957
    },
    {
      "epoch": 8.771896362304688e-06,
      "model_forward_time": 1.8251838684082031,
      "step": 9198
    },
    {
      "epoch": 8.771896362304688e-06,
      "step": 9198,
      "training_step_time": 6.856598138809204
    },
    {
      "epoch": 8.772850036621094e-06,
      "model_forward_time": 1.8155467510223389,
      "step": 9199
    },
    {
      "epoch": 8.772850036621094e-06,
      "step": 9199,
      "training_step_time": 7.316487789154053
    },
    {
      "epoch": 8.7738037109375e-06,
      "grad_norm": 0.2600138485431671,
      "learning_rate": 9.951873270000576e-06,
      "loss": 0.0726,
      "step": 9200
    },
    {
      "epoch": 8.7738037109375e-06,
      "model_forward_time": 1.8171215057373047,
      "step": 9200
    },
    {
      "dynamics_loss_avg": 0.05604172255843878,
      "epoch": 8.7738037109375e-06,
      "step": 9200
    },
    {
      "action_loss_avg": 0.0029650639509782193,
      "epoch": 8.7738037109375e-06,
      "step": 9200
    },
    {
      "epoch": 8.7738037109375e-06,
      "step": 9200,
      "training_step_time": 7.626585245132446
    },
    {
      "epoch": 8.774757385253907e-06,
      "model_forward_time": 1.8231232166290283,
      "step": 9201
    },
    {
      "epoch": 8.774757385253907e-06,
      "step": 9201,
      "training_step_time": 8.540190935134888
    },
    {
      "epoch": 8.775711059570312e-06,
      "model_forward_time": 1.811934471130371,
      "step": 9202
    },
    {
      "epoch": 8.775711059570312e-06,
      "step": 9202,
      "training_step_time": 9.260179281234741
    },
    {
      "epoch": 8.776664733886718e-06,
      "model_forward_time": 2.3731496334075928,
      "step": 9203
    },
    {
      "epoch": 8.776664733886718e-06,
      "step": 9203,
      "training_step_time": 8.770266771316528
    },
    {
      "epoch": 8.777618408203126e-06,
      "model_forward_time": 1.8246493339538574,
      "step": 9204
    },
    {
      "epoch": 8.777618408203126e-06,
      "step": 9204,
      "training_step_time": 8.80907940864563
    },
    {
      "epoch": 8.778572082519531e-06,
      "model_forward_time": 1.8303279876708984,
      "step": 9205
    },
    {
      "epoch": 8.778572082519531e-06,
      "step": 9205,
      "training_step_time": 7.830480098724365
    },
    {
      "epoch": 8.779525756835937e-06,
      "model_forward_time": 1.8257787227630615,
      "step": 9206
    },
    {
      "epoch": 8.779525756835937e-06,
      "step": 9206,
      "training_step_time": 7.574225187301636
    },
    {
      "epoch": 8.780479431152344e-06,
      "model_forward_time": 1.8248059749603271,
      "step": 9207
    },
    {
      "epoch": 8.780479431152344e-06,
      "step": 9207,
      "training_step_time": 6.890540361404419
    },
    {
      "epoch": 8.78143310546875e-06,
      "model_forward_time": 1.8250606060028076,
      "step": 9208
    },
    {
      "epoch": 8.78143310546875e-06,
      "step": 9208,
      "training_step_time": 7.382352113723755
    },
    {
      "epoch": 8.782386779785156e-06,
      "model_forward_time": 1.8285188674926758,
      "step": 9209
    },
    {
      "epoch": 8.782386779785156e-06,
      "step": 9209,
      "training_step_time": 7.637847900390625
    },
    {
      "epoch": 8.783340454101563e-06,
      "grad_norm": 0.2886998951435089,
      "learning_rate": 9.951644138202216e-06,
      "loss": 0.0664,
      "step": 9210
    },
    {
      "epoch": 8.783340454101563e-06,
      "model_forward_time": 1.827223539352417,
      "step": 9210
    },
    {
      "dynamics_loss_avg": 0.0659354854375124,
      "epoch": 8.783340454101563e-06,
      "step": 9210
    },
    {
      "action_loss_avg": 0.006702400208450854,
      "epoch": 8.783340454101563e-06,
      "step": 9210
    },
    {
      "epoch": 8.783340454101563e-06,
      "step": 9210,
      "training_step_time": 7.728230237960815
    },
    {
      "epoch": 8.784294128417969e-06,
      "model_forward_time": 1.8249444961547852,
      "step": 9211
    },
    {
      "epoch": 8.784294128417969e-06,
      "step": 9211,
      "training_step_time": 8.430355072021484
    },
    {
      "epoch": 8.785247802734375e-06,
      "model_forward_time": 2.310286521911621,
      "step": 9212
    },
    {
      "epoch": 8.785247802734375e-06,
      "step": 9212,
      "training_step_time": 8.78836703300476
    },
    {
      "epoch": 8.78620147705078e-06,
      "model_forward_time": 1.8141295909881592,
      "step": 9213
    },
    {
      "epoch": 8.78620147705078e-06,
      "step": 9213,
      "training_step_time": 9.701299667358398
    },
    {
      "epoch": 8.787155151367188e-06,
      "model_forward_time": 1.822465419769287,
      "step": 9214
    },
    {
      "epoch": 8.787155151367188e-06,
      "step": 9214,
      "training_step_time": 7.763322591781616
    },
    {
      "epoch": 8.788108825683594e-06,
      "model_forward_time": 1.831200122833252,
      "step": 9215
    },
    {
      "epoch": 8.788108825683594e-06,
      "step": 9215,
      "training_step_time": 7.533942699432373
    },
    {
      "epoch": 8.7890625e-06,
      "model_forward_time": 1.8193423748016357,
      "step": 9216
    },
    {
      "epoch": 8.7890625e-06,
      "step": 9216,
      "training_step_time": 6.804868698120117
    },
    {
      "epoch": 8.790016174316407e-06,
      "model_forward_time": 1.8155076503753662,
      "step": 9217
    },
    {
      "epoch": 8.790016174316407e-06,
      "step": 9217,
      "training_step_time": 7.196894645690918
    },
    {
      "epoch": 8.790969848632812e-06,
      "model_forward_time": 1.824413537979126,
      "step": 9218
    },
    {
      "epoch": 8.790969848632812e-06,
      "step": 9218,
      "training_step_time": 7.786834955215454
    },
    {
      "epoch": 8.791923522949218e-06,
      "model_forward_time": 1.8238937854766846,
      "step": 9219
    },
    {
      "epoch": 8.791923522949218e-06,
      "step": 9219,
      "training_step_time": 8.74564528465271
    },
    {
      "epoch": 8.792877197265626e-06,
      "grad_norm": 0.17195145785808563,
      "learning_rate": 9.951414464899497e-06,
      "loss": 0.0702,
      "step": 9220
    },
    {
      "epoch": 8.792877197265626e-06,
      "model_forward_time": 1.822340726852417,
      "step": 9220
    },
    {
      "dynamics_loss_avg": 0.05785051789134741,
      "epoch": 8.792877197265626e-06,
      "step": 9220
    },
    {
      "action_loss_avg": 0.0033656657906249166,
      "epoch": 8.792877197265626e-06,
      "step": 9220
    },
    {
      "epoch": 8.792877197265626e-06,
      "step": 9220,
      "training_step_time": 7.649779319763184
    },
    {
      "epoch": 8.793830871582031e-06,
      "model_forward_time": 2.3424551486968994,
      "step": 9221
    },
    {
      "epoch": 8.793830871582031e-06,
      "step": 9221,
      "training_step_time": 8.798806428909302
    },
    {
      "epoch": 8.794784545898437e-06,
      "model_forward_time": 1.825272798538208,
      "step": 9222
    },
    {
      "epoch": 8.794784545898437e-06,
      "step": 9222,
      "training_step_time": 10.257498979568481
    },
    {
      "epoch": 8.795738220214844e-06,
      "model_forward_time": 1.8227481842041016,
      "step": 9223
    },
    {
      "epoch": 8.795738220214844e-06,
      "step": 9223,
      "training_step_time": 8.648123264312744
    },
    {
      "epoch": 8.79669189453125e-06,
      "model_forward_time": 1.824521780014038,
      "step": 9224
    },
    {
      "epoch": 8.79669189453125e-06,
      "step": 9224,
      "training_step_time": 8.574421644210815
    },
    {
      "epoch": 8.797645568847656e-06,
      "model_forward_time": 1.8279752731323242,
      "step": 9225
    },
    {
      "epoch": 8.797645568847656e-06,
      "step": 9225,
      "training_step_time": 7.488689661026001
    },
    {
      "epoch": 8.798599243164063e-06,
      "model_forward_time": 1.8280322551727295,
      "step": 9226
    },
    {
      "epoch": 8.798599243164063e-06,
      "step": 9226,
      "training_step_time": 7.306963205337524
    },
    {
      "epoch": 8.799552917480469e-06,
      "model_forward_time": 1.8197417259216309,
      "step": 9227
    },
    {
      "epoch": 8.799552917480469e-06,
      "step": 9227,
      "training_step_time": 7.143077611923218
    },
    {
      "epoch": 8.800506591796875e-06,
      "model_forward_time": 1.8195278644561768,
      "step": 9228
    },
    {
      "epoch": 8.800506591796875e-06,
      "step": 9228,
      "training_step_time": 7.742905855178833
    },
    {
      "epoch": 8.80146026611328e-06,
      "model_forward_time": 1.8304657936096191,
      "step": 9229
    },
    {
      "epoch": 8.80146026611328e-06,
      "step": 9229,
      "training_step_time": 8.375484228134155
    },
    {
      "epoch": 8.802413940429688e-06,
      "grad_norm": 0.1731761395931244,
      "learning_rate": 9.951184250117538e-06,
      "loss": 0.0675,
      "step": 9230
    },
    {
      "epoch": 8.802413940429688e-06,
      "model_forward_time": 2.368180990219116,
      "step": 9230
    },
    {
      "dynamics_loss_avg": 0.08069139681756496,
      "epoch": 8.802413940429688e-06,
      "step": 9230
    },
    {
      "action_loss_avg": 0.008002871996723115,
      "epoch": 8.802413940429688e-06,
      "step": 9230
    },
    {
      "epoch": 8.802413940429688e-06,
      "step": 9230,
      "training_step_time": 8.524861097335815
    },
    {
      "epoch": 8.803367614746094e-06,
      "model_forward_time": 1.82004714012146,
      "step": 9231
    },
    {
      "epoch": 8.803367614746094e-06,
      "step": 9231,
      "training_step_time": 10.426272630691528
    },
    {
      "epoch": 8.8043212890625e-06,
      "model_forward_time": 1.8255937099456787,
      "step": 9232
    },
    {
      "epoch": 8.8043212890625e-06,
      "step": 9232,
      "training_step_time": 9.739151239395142
    },
    {
      "epoch": 8.805274963378907e-06,
      "model_forward_time": 1.8170380592346191,
      "step": 9233
    },
    {
      "epoch": 8.805274963378907e-06,
      "step": 9233,
      "training_step_time": 7.738617181777954
    },
    {
      "epoch": 8.806228637695312e-06,
      "model_forward_time": 1.823227882385254,
      "step": 9234
    },
    {
      "epoch": 8.806228637695312e-06,
      "step": 9234,
      "training_step_time": 7.500210523605347
    },
    {
      "epoch": 8.807182312011718e-06,
      "model_forward_time": 1.8236234188079834,
      "step": 9235
    },
    {
      "epoch": 8.807182312011718e-06,
      "step": 9235,
      "training_step_time": 7.282764196395874
    },
    {
      "epoch": 8.808135986328126e-06,
      "model_forward_time": 1.8198039531707764,
      "step": 9236
    },
    {
      "epoch": 8.808135986328126e-06,
      "step": 9236,
      "training_step_time": 7.155219078063965
    },
    {
      "epoch": 8.809089660644531e-06,
      "model_forward_time": 1.81941819190979,
      "step": 9237
    },
    {
      "epoch": 8.809089660644531e-06,
      "step": 9237,
      "training_step_time": 7.614685773849487
    },
    {
      "epoch": 8.810043334960937e-06,
      "model_forward_time": 1.8310284614562988,
      "step": 9238
    },
    {
      "epoch": 8.810043334960937e-06,
      "step": 9238,
      "training_step_time": 7.709271430969238
    },
    {
      "epoch": 8.810997009277344e-06,
      "model_forward_time": 2.359877109527588,
      "step": 9239
    },
    {
      "epoch": 8.810997009277344e-06,
      "step": 9239,
      "training_step_time": 9.495458602905273
    },
    {
      "epoch": 8.81195068359375e-06,
      "grad_norm": 0.37116163969039917,
      "learning_rate": 9.950953493881513e-06,
      "loss": 0.0689,
      "step": 9240
    },
    {
      "epoch": 8.81195068359375e-06,
      "model_forward_time": 1.8259060382843018,
      "step": 9240
    },
    {
      "dynamics_loss_avg": 0.06871747281402349,
      "epoch": 8.81195068359375e-06,
      "step": 9240
    },
    {
      "action_loss_avg": 0.0032701366872061044,
      "epoch": 8.81195068359375e-06,
      "step": 9240
    },
    {
      "epoch": 8.81195068359375e-06,
      "step": 9240,
      "training_step_time": 7.744935035705566
    },
    {
      "epoch": 8.812904357910156e-06,
      "model_forward_time": 1.8345801830291748,
      "step": 9241
    },
    {
      "epoch": 8.812904357910156e-06,
      "step": 9241,
      "training_step_time": 8.669275999069214
    },
    {
      "epoch": 8.813858032226563e-06,
      "model_forward_time": 1.8285527229309082,
      "step": 9242
    },
    {
      "epoch": 8.813858032226563e-06,
      "step": 9242,
      "training_step_time": 7.720319986343384
    },
    {
      "epoch": 8.814811706542969e-06,
      "model_forward_time": 1.8357226848602295,
      "step": 9243
    },
    {
      "epoch": 8.814811706542969e-06,
      "step": 9243,
      "training_step_time": 7.537562370300293
    },
    {
      "epoch": 8.815765380859375e-06,
      "model_forward_time": 1.830399990081787,
      "step": 9244
    },
    {
      "epoch": 8.815765380859375e-06,
      "step": 9244,
      "training_step_time": 7.315516948699951
    },
    {
      "epoch": 8.81671905517578e-06,
      "model_forward_time": 1.8309829235076904,
      "step": 9245
    },
    {
      "epoch": 8.81671905517578e-06,
      "step": 9245,
      "training_step_time": 7.078486919403076
    },
    {
      "epoch": 8.817672729492188e-06,
      "model_forward_time": 1.8285484313964844,
      "step": 9246
    },
    {
      "epoch": 8.817672729492188e-06,
      "step": 9246,
      "training_step_time": 7.676392316818237
    },
    {
      "epoch": 8.818626403808594e-06,
      "model_forward_time": 1.835986614227295,
      "step": 9247
    },
    {
      "epoch": 8.818626403808594e-06,
      "step": 9247,
      "training_step_time": 7.31434178352356
    },
    {
      "epoch": 8.819580078125e-06,
      "model_forward_time": 1.8266756534576416,
      "step": 9248
    },
    {
      "epoch": 8.819580078125e-06,
      "step": 9248,
      "training_step_time": 8.182400465011597
    },
    {
      "epoch": 8.820533752441407e-06,
      "model_forward_time": 1.8333582878112793,
      "step": 9249
    },
    {
      "epoch": 8.820533752441407e-06,
      "step": 9249,
      "training_step_time": 8.642847776412964
    },
    {
      "epoch": 8.821487426757812e-06,
      "grad_norm": 0.36317506432533264,
      "learning_rate": 9.950722196216657e-06,
      "loss": 0.0746,
      "step": 9250
    },
    {
      "epoch": 8.821487426757812e-06,
      "model_forward_time": 1.8053083419799805,
      "step": 9250
    },
    {
      "dynamics_loss_avg": 0.06573391761630773,
      "epoch": 8.821487426757812e-06,
      "step": 9250
    },
    {
      "action_loss_avg": 0.004612161102704704,
      "epoch": 8.821487426757812e-06,
      "step": 9250
    },
    {
      "epoch": 8.821487426757812e-06,
      "step": 9250,
      "training_step_time": 7.723125696182251
    },
    {
      "epoch": 8.822441101074218e-06,
      "model_forward_time": 1.8261387348175049,
      "step": 9251
    },
    {
      "epoch": 8.822441101074218e-06,
      "step": 9251,
      "training_step_time": 7.724749803543091
    },
    {
      "epoch": 8.823394775390626e-06,
      "model_forward_time": 1.8329856395721436,
      "step": 9252
    },
    {
      "epoch": 8.823394775390626e-06,
      "step": 9252,
      "training_step_time": 7.6560351848602295
    },
    {
      "epoch": 8.824348449707031e-06,
      "model_forward_time": 1.8290925025939941,
      "step": 9253
    },
    {
      "epoch": 8.824348449707031e-06,
      "step": 9253,
      "training_step_time": 7.241227149963379
    },
    {
      "epoch": 8.825302124023437e-06,
      "model_forward_time": 1.8291800022125244,
      "step": 9254
    },
    {
      "epoch": 8.825302124023437e-06,
      "step": 9254,
      "training_step_time": 7.085473537445068
    },
    {
      "epoch": 8.826255798339844e-06,
      "model_forward_time": 1.8294649124145508,
      "step": 9255
    },
    {
      "epoch": 8.826255798339844e-06,
      "step": 9255,
      "training_step_time": 7.595753192901611
    },
    {
      "epoch": 8.82720947265625e-06,
      "model_forward_time": 1.836561918258667,
      "step": 9256
    },
    {
      "epoch": 8.82720947265625e-06,
      "step": 9256,
      "training_step_time": 7.253351211547852
    },
    {
      "epoch": 8.828163146972656e-06,
      "model_forward_time": 1.8289861679077148,
      "step": 9257
    },
    {
      "epoch": 8.828163146972656e-06,
      "step": 9257,
      "training_step_time": 8.63990831375122
    },
    {
      "epoch": 8.829116821289062e-06,
      "model_forward_time": 2.362682580947876,
      "step": 9258
    },
    {
      "epoch": 8.829116821289062e-06,
      "step": 9258,
      "training_step_time": 9.811553001403809
    },
    {
      "epoch": 8.830070495605469e-06,
      "model_forward_time": 1.8303534984588623,
      "step": 9259
    },
    {
      "epoch": 8.830070495605469e-06,
      "step": 9259,
      "training_step_time": 7.707669496536255
    },
    {
      "epoch": 8.831024169921875e-06,
      "grad_norm": 0.20248202979564667,
      "learning_rate": 9.950490357148266e-06,
      "loss": 0.0754,
      "step": 9260
    },
    {
      "epoch": 8.831024169921875e-06,
      "model_forward_time": 1.8261134624481201,
      "step": 9260
    },
    {
      "dynamics_loss_avg": 0.050182577781379224,
      "epoch": 8.831024169921875e-06,
      "step": 9260
    },
    {
      "action_loss_avg": 0.004415921890176833,
      "epoch": 8.831024169921875e-06,
      "step": 9260
    },
    {
      "epoch": 8.831024169921875e-06,
      "step": 9260,
      "training_step_time": 7.874579191207886
    },
    {
      "epoch": 8.83197784423828e-06,
      "model_forward_time": 1.8102905750274658,
      "step": 9261
    },
    {
      "epoch": 8.83197784423828e-06,
      "step": 9261,
      "training_step_time": 7.7223992347717285
    },
    {
      "epoch": 8.832931518554688e-06,
      "model_forward_time": 1.828733205795288,
      "step": 9262
    },
    {
      "epoch": 8.832931518554688e-06,
      "step": 9262,
      "training_step_time": 7.385540723800659
    },
    {
      "epoch": 8.833885192871094e-06,
      "model_forward_time": 1.8394193649291992,
      "step": 9263
    },
    {
      "epoch": 8.833885192871094e-06,
      "step": 9263,
      "training_step_time": 7.087234735488892
    },
    {
      "epoch": 8.8348388671875e-06,
      "model_forward_time": 1.8333690166473389,
      "step": 9264
    },
    {
      "epoch": 8.8348388671875e-06,
      "step": 9264,
      "training_step_time": 6.716809511184692
    },
    {
      "epoch": 8.835792541503907e-06,
      "model_forward_time": 1.8274247646331787,
      "step": 9265
    },
    {
      "epoch": 8.835792541503907e-06,
      "step": 9265,
      "training_step_time": 7.546147346496582
    },
    {
      "epoch": 8.836746215820313e-06,
      "model_forward_time": 1.8266897201538086,
      "step": 9266
    },
    {
      "epoch": 8.836746215820313e-06,
      "step": 9266,
      "training_step_time": 8.630667209625244
    },
    {
      "epoch": 8.837699890136718e-06,
      "model_forward_time": 2.3755300045013428,
      "step": 9267
    },
    {
      "epoch": 8.837699890136718e-06,
      "step": 9267,
      "training_step_time": 8.709489822387695
    },
    {
      "epoch": 8.838653564453126e-06,
      "model_forward_time": 1.8309180736541748,
      "step": 9268
    },
    {
      "epoch": 8.838653564453126e-06,
      "step": 9268,
      "training_step_time": 8.355597019195557
    },
    {
      "epoch": 8.839607238769531e-06,
      "model_forward_time": 1.8254330158233643,
      "step": 9269
    },
    {
      "epoch": 8.839607238769531e-06,
      "step": 9269,
      "training_step_time": 7.779754638671875
    },
    {
      "epoch": 8.840560913085937e-06,
      "grad_norm": 0.2048889398574829,
      "learning_rate": 9.950257976701692e-06,
      "loss": 0.0766,
      "step": 9270
    },
    {
      "epoch": 8.840560913085937e-06,
      "model_forward_time": 1.8208494186401367,
      "step": 9270
    },
    {
      "dynamics_loss_avg": 0.06887301951646804,
      "epoch": 8.840560913085937e-06,
      "step": 9270
    },
    {
      "action_loss_avg": 0.006448348751291633,
      "epoch": 8.840560913085937e-06,
      "step": 9270
    },
    {
      "epoch": 8.840560913085937e-06,
      "step": 9270,
      "training_step_time": 7.5912768840789795
    },
    {
      "epoch": 8.841514587402345e-06,
      "model_forward_time": 1.826798439025879,
      "step": 9271
    },
    {
      "epoch": 8.841514587402345e-06,
      "step": 9271,
      "training_step_time": 7.4368157386779785
    },
    {
      "epoch": 8.84246826171875e-06,
      "model_forward_time": 1.824986219406128,
      "step": 9272
    },
    {
      "epoch": 8.84246826171875e-06,
      "step": 9272,
      "training_step_time": 7.079399585723877
    },
    {
      "epoch": 8.843421936035156e-06,
      "model_forward_time": 1.8257415294647217,
      "step": 9273
    },
    {
      "epoch": 8.843421936035156e-06,
      "step": 9273,
      "training_step_time": 6.726851463317871
    },
    {
      "epoch": 8.844375610351562e-06,
      "model_forward_time": 1.8227462768554688,
      "step": 9274
    },
    {
      "epoch": 8.844375610351562e-06,
      "step": 9274,
      "training_step_time": 7.5041821002960205
    },
    {
      "epoch": 8.845329284667969e-06,
      "model_forward_time": 1.8217055797576904,
      "step": 9275
    },
    {
      "epoch": 8.845329284667969e-06,
      "step": 9275,
      "training_step_time": 7.480449438095093
    },
    {
      "epoch": 8.846282958984375e-06,
      "model_forward_time": 2.3560421466827393,
      "step": 9276
    },
    {
      "epoch": 8.846282958984375e-06,
      "step": 9276,
      "training_step_time": 7.74602746963501
    },
    {
      "epoch": 8.84723663330078e-06,
      "model_forward_time": 1.8260776996612549,
      "step": 9277
    },
    {
      "epoch": 8.84723663330078e-06,
      "step": 9277,
      "training_step_time": 7.673352241516113
    },
    {
      "epoch": 8.848190307617188e-06,
      "model_forward_time": 1.812915325164795,
      "step": 9278
    },
    {
      "epoch": 8.848190307617188e-06,
      "step": 9278,
      "training_step_time": 7.8355326652526855
    },
    {
      "epoch": 8.849143981933594e-06,
      "model_forward_time": 1.8285744190216064,
      "step": 9279
    },
    {
      "epoch": 8.849143981933594e-06,
      "step": 9279,
      "training_step_time": 7.561896324157715
    },
    {
      "epoch": 8.85009765625e-06,
      "grad_norm": 0.2062665969133377,
      "learning_rate": 9.950025054902348e-06,
      "loss": 0.0659,
      "step": 9280
    },
    {
      "epoch": 8.85009765625e-06,
      "model_forward_time": 1.8257451057434082,
      "step": 9280
    },
    {
      "dynamics_loss_avg": 0.04701474346220493,
      "epoch": 8.85009765625e-06,
      "step": 9280
    },
    {
      "action_loss_avg": 0.003968084906227886,
      "epoch": 8.85009765625e-06,
      "step": 9280
    },
    {
      "epoch": 8.85009765625e-06,
      "step": 9280,
      "training_step_time": 8.161113262176514
    },
    {
      "epoch": 8.851051330566407e-06,
      "model_forward_time": 1.8227686882019043,
      "step": 9281
    },
    {
      "epoch": 8.851051330566407e-06,
      "step": 9281,
      "training_step_time": 7.063354730606079
    },
    {
      "epoch": 8.852005004882813e-06,
      "model_forward_time": 1.825357437133789,
      "step": 9282
    },
    {
      "epoch": 8.852005004882813e-06,
      "step": 9282,
      "training_step_time": 6.691202878952026
    },
    {
      "epoch": 8.852958679199218e-06,
      "model_forward_time": 1.8329036235809326,
      "step": 9283
    },
    {
      "epoch": 8.852958679199218e-06,
      "step": 9283,
      "training_step_time": 8.570656299591064
    },
    {
      "epoch": 8.853912353515626e-06,
      "model_forward_time": 1.8272442817687988,
      "step": 9284
    },
    {
      "epoch": 8.853912353515626e-06,
      "step": 9284,
      "training_step_time": 6.9857566356658936
    },
    {
      "epoch": 8.854866027832031e-06,
      "model_forward_time": 2.3790977001190186,
      "step": 9285
    },
    {
      "epoch": 8.854866027832031e-06,
      "step": 9285,
      "training_step_time": 9.557287454605103
    },
    {
      "epoch": 8.855819702148437e-06,
      "model_forward_time": 1.823854923248291,
      "step": 9286
    },
    {
      "epoch": 8.855819702148437e-06,
      "step": 9286,
      "training_step_time": 8.372931003570557
    },
    {
      "epoch": 8.856773376464845e-06,
      "model_forward_time": 1.8289813995361328,
      "step": 9287
    },
    {
      "epoch": 8.856773376464845e-06,
      "step": 9287,
      "training_step_time": 9.336614847183228
    },
    {
      "epoch": 8.85772705078125e-06,
      "model_forward_time": 1.8278064727783203,
      "step": 9288
    },
    {
      "epoch": 8.85772705078125e-06,
      "step": 9288,
      "training_step_time": 8.05748987197876
    },
    {
      "epoch": 8.858680725097656e-06,
      "model_forward_time": 1.820363998413086,
      "step": 9289
    },
    {
      "epoch": 8.858680725097656e-06,
      "step": 9289,
      "training_step_time": 7.6679112911224365
    },
    {
      "epoch": 8.859634399414062e-06,
      "grad_norm": 0.1770574301481247,
      "learning_rate": 9.949791591775706e-06,
      "loss": 0.0699,
      "step": 9290
    },
    {
      "epoch": 8.859634399414062e-06,
      "model_forward_time": 1.8320729732513428,
      "step": 9290
    },
    {
      "dynamics_loss_avg": 0.0696060074493289,
      "epoch": 8.859634399414062e-06,
      "step": 9290
    },
    {
      "action_loss_avg": 0.006246702431235462,
      "epoch": 8.859634399414062e-06,
      "step": 9290
    },
    {
      "epoch": 8.859634399414062e-06,
      "step": 9290,
      "training_step_time": 7.381186485290527
    },
    {
      "epoch": 8.860588073730469e-06,
      "model_forward_time": 1.4609863758087158,
      "step": 9291
    },
    {
      "epoch": 8.860588073730469e-06,
      "step": 9291,
      "training_step_time": 6.734269380569458
    },
    {
      "epoch": 8.861541748046875e-06,
      "model_forward_time": 1.824997901916504,
      "step": 9292
    },
    {
      "epoch": 8.861541748046875e-06,
      "step": 9292,
      "training_step_time": 8.666768312454224
    },
    {
      "epoch": 8.86249542236328e-06,
      "model_forward_time": 1.8310108184814453,
      "step": 9293
    },
    {
      "epoch": 8.86249542236328e-06,
      "step": 9293,
      "training_step_time": 7.28445029258728
    },
    {
      "epoch": 8.863449096679688e-06,
      "model_forward_time": 2.322746515274048,
      "step": 9294
    },
    {
      "epoch": 8.863449096679688e-06,
      "step": 9294,
      "training_step_time": 8.37265658378601
    },
    {
      "epoch": 8.864402770996094e-06,
      "model_forward_time": 1.8304173946380615,
      "step": 9295
    },
    {
      "epoch": 8.864402770996094e-06,
      "step": 9295,
      "training_step_time": 9.384928703308105
    },
    {
      "epoch": 8.8653564453125e-06,
      "model_forward_time": 1.828861951828003,
      "step": 9296
    },
    {
      "epoch": 8.8653564453125e-06,
      "step": 9296,
      "training_step_time": 8.764610528945923
    },
    {
      "epoch": 8.866310119628907e-06,
      "model_forward_time": 1.8171923160552979,
      "step": 9297
    },
    {
      "epoch": 8.866310119628907e-06,
      "step": 9297,
      "training_step_time": 7.613250255584717
    },
    {
      "epoch": 8.867263793945313e-06,
      "model_forward_time": 1.8258700370788574,
      "step": 9298
    },
    {
      "epoch": 8.867263793945313e-06,
      "step": 9298,
      "training_step_time": 7.671747922897339
    },
    {
      "epoch": 8.868217468261718e-06,
      "model_forward_time": 1.4698243141174316,
      "step": 9299
    },
    {
      "epoch": 8.868217468261718e-06,
      "step": 9299,
      "training_step_time": 7.439796447753906
    },
    {
      "epoch": 8.869171142578126e-06,
      "grad_norm": 0.2701340317726135,
      "learning_rate": 9.949557587347299e-06,
      "loss": 0.0692,
      "step": 9300
    },
    {
      "epoch": 8.869171142578126e-06,
      "model_forward_time": 1.819657564163208,
      "step": 9300
    },
    {
      "dynamics_loss_avg": 0.0686870027333498,
      "epoch": 8.869171142578126e-06,
      "step": 9300
    },
    {
      "action_loss_avg": 0.007311402907362208,
      "epoch": 8.869171142578126e-06,
      "step": 9300
    },
    {
      "epoch": 8.869171142578126e-06,
      "step": 9300,
      "training_step_time": 6.712962627410889
    },
    {
      "epoch": 8.870124816894531e-06,
      "model_forward_time": 1.8285167217254639,
      "step": 9301
    },
    {
      "epoch": 8.870124816894531e-06,
      "step": 9301,
      "training_step_time": 7.601230144500732
    },
    {
      "epoch": 8.871078491210937e-06,
      "model_forward_time": 1.8330180644989014,
      "step": 9302
    },
    {
      "epoch": 8.871078491210937e-06,
      "step": 9302,
      "training_step_time": 7.27555513381958
    },
    {
      "epoch": 8.872032165527345e-06,
      "model_forward_time": 1.8230524063110352,
      "step": 9303
    },
    {
      "epoch": 8.872032165527345e-06,
      "step": 9303,
      "training_step_time": 9.625704765319824
    },
    {
      "epoch": 8.87298583984375e-06,
      "model_forward_time": 2.356644630432129,
      "step": 9304
    },
    {
      "epoch": 8.87298583984375e-06,
      "step": 9304,
      "training_step_time": 7.712326765060425
    },
    {
      "epoch": 8.873939514160156e-06,
      "model_forward_time": 1.82631516456604,
      "step": 9305
    },
    {
      "epoch": 8.873939514160156e-06,
      "step": 9305,
      "training_step_time": 9.02348256111145
    },
    {
      "epoch": 8.874893188476562e-06,
      "model_forward_time": 1.8300528526306152,
      "step": 9306
    },
    {
      "epoch": 8.874893188476562e-06,
      "step": 9306,
      "training_step_time": 8.469763994216919
    },
    {
      "epoch": 8.875846862792969e-06,
      "model_forward_time": 1.8226559162139893,
      "step": 9307
    },
    {
      "epoch": 8.875846862792969e-06,
      "step": 9307,
      "training_step_time": 7.718909502029419
    },
    {
      "epoch": 8.876800537109375e-06,
      "model_forward_time": 1.8304078578948975,
      "step": 9308
    },
    {
      "epoch": 8.876800537109375e-06,
      "step": 9308,
      "training_step_time": 7.388399124145508
    },
    {
      "epoch": 8.87775421142578e-06,
      "model_forward_time": 1.8349103927612305,
      "step": 9309
    },
    {
      "epoch": 8.87775421142578e-06,
      "step": 9309,
      "training_step_time": 6.718557357788086
    },
    {
      "epoch": 8.878707885742188e-06,
      "grad_norm": 0.25028932094573975,
      "learning_rate": 9.949323041642714e-06,
      "loss": 0.0718,
      "step": 9310
    },
    {
      "epoch": 8.878707885742188e-06,
      "model_forward_time": 1.8140661716461182,
      "step": 9310
    },
    {
      "dynamics_loss_avg": 0.0562737287953496,
      "epoch": 8.878707885742188e-06,
      "step": 9310
    },
    {
      "action_loss_avg": 0.004839776665903628,
      "epoch": 8.878707885742188e-06,
      "step": 9310
    },
    {
      "epoch": 8.878707885742188e-06,
      "step": 9310,
      "training_step_time": 7.5704193115234375
    },
    {
      "epoch": 8.879661560058594e-06,
      "model_forward_time": 1.8269994258880615,
      "step": 9311
    },
    {
      "epoch": 8.879661560058594e-06,
      "step": 9311,
      "training_step_time": 7.279355525970459
    },
    {
      "epoch": 8.880615234375e-06,
      "model_forward_time": 1.821467399597168,
      "step": 9312
    },
    {
      "epoch": 8.880615234375e-06,
      "step": 9312,
      "training_step_time": 9.23866319656372
    },
    {
      "epoch": 8.881568908691407e-06,
      "model_forward_time": 2.3434126377105713,
      "step": 9313
    },
    {
      "epoch": 8.881568908691407e-06,
      "step": 9313,
      "training_step_time": 8.698470115661621
    },
    {
      "epoch": 8.882522583007813e-06,
      "model_forward_time": 1.8337934017181396,
      "step": 9314
    },
    {
      "epoch": 8.882522583007813e-06,
      "step": 9314,
      "training_step_time": 9.266982793807983
    },
    {
      "epoch": 8.883476257324218e-06,
      "model_forward_time": 1.8167665004730225,
      "step": 9315
    },
    {
      "epoch": 8.883476257324218e-06,
      "step": 9315,
      "training_step_time": 7.83175253868103
    },
    {
      "epoch": 8.884429931640626e-06,
      "model_forward_time": 1.8250758647918701,
      "step": 9316
    },
    {
      "epoch": 8.884429931640626e-06,
      "step": 9316,
      "training_step_time": 8.704407930374146
    },
    {
      "epoch": 8.885383605957031e-06,
      "model_forward_time": 1.8269367218017578,
      "step": 9317
    },
    {
      "epoch": 8.885383605957031e-06,
      "step": 9317,
      "training_step_time": 7.379332065582275
    },
    {
      "epoch": 8.886337280273437e-06,
      "model_forward_time": 1.8246030807495117,
      "step": 9318
    },
    {
      "epoch": 8.886337280273437e-06,
      "step": 9318,
      "training_step_time": 7.393966436386108
    },
    {
      "epoch": 8.887290954589845e-06,
      "model_forward_time": 1.8258576393127441,
      "step": 9319
    },
    {
      "epoch": 8.887290954589845e-06,
      "step": 9319,
      "training_step_time": 7.60709810256958
    },
    {
      "epoch": 8.88824462890625e-06,
      "grad_norm": 0.14275185763835907,
      "learning_rate": 9.949087954687601e-06,
      "loss": 0.0713,
      "step": 9320
    },
    {
      "epoch": 8.88824462890625e-06,
      "model_forward_time": 1.8255066871643066,
      "step": 9320
    },
    {
      "dynamics_loss_avg": 0.06722042430192232,
      "epoch": 8.88824462890625e-06,
      "step": 9320
    },
    {
      "action_loss_avg": 0.004005936835892498,
      "epoch": 8.88824462890625e-06,
      "step": 9320
    },
    {
      "epoch": 8.88824462890625e-06,
      "step": 9320,
      "training_step_time": 7.359441757202148
    },
    {
      "epoch": 8.889198303222656e-06,
      "model_forward_time": 1.8217945098876953,
      "step": 9321
    },
    {
      "epoch": 8.889198303222656e-06,
      "step": 9321,
      "training_step_time": 7.521830320358276
    },
    {
      "epoch": 8.890151977539062e-06,
      "model_forward_time": 2.3928115367889404,
      "step": 9322
    },
    {
      "epoch": 8.890151977539062e-06,
      "step": 9322,
      "training_step_time": 8.51047945022583
    },
    {
      "epoch": 8.891105651855469e-06,
      "model_forward_time": 1.8276615142822266,
      "step": 9323
    },
    {
      "epoch": 8.891105651855469e-06,
      "step": 9323,
      "training_step_time": 7.725273609161377
    },
    {
      "epoch": 8.892059326171875e-06,
      "model_forward_time": 1.8274412155151367,
      "step": 9324
    },
    {
      "epoch": 8.892059326171875e-06,
      "step": 9324,
      "training_step_time": 8.742012739181519
    },
    {
      "epoch": 8.89301300048828e-06,
      "model_forward_time": 1.7895500659942627,
      "step": 9325
    },
    {
      "epoch": 8.89301300048828e-06,
      "step": 9325,
      "training_step_time": 7.6510772705078125
    },
    {
      "epoch": 8.893966674804688e-06,
      "model_forward_time": 1.8072521686553955,
      "step": 9326
    },
    {
      "epoch": 8.893966674804688e-06,
      "step": 9326,
      "training_step_time": 7.685420513153076
    },
    {
      "epoch": 8.894920349121094e-06,
      "model_forward_time": 1.81526780128479,
      "step": 9327
    },
    {
      "epoch": 8.894920349121094e-06,
      "step": 9327,
      "training_step_time": 7.407553672790527
    },
    {
      "epoch": 8.8958740234375e-06,
      "model_forward_time": 1.8128368854522705,
      "step": 9328
    },
    {
      "epoch": 8.8958740234375e-06,
      "step": 9328,
      "training_step_time": 6.739813566207886
    },
    {
      "epoch": 8.896827697753907e-06,
      "model_forward_time": 1.8117005825042725,
      "step": 9329
    },
    {
      "epoch": 8.896827697753907e-06,
      "step": 9329,
      "training_step_time": 7.602261066436768
    },
    {
      "epoch": 8.897781372070313e-06,
      "grad_norm": 0.2328711748123169,
      "learning_rate": 9.948852326507672e-06,
      "loss": 0.0672,
      "step": 9330
    },
    {
      "epoch": 8.897781372070313e-06,
      "model_forward_time": 1.8103690147399902,
      "step": 9330
    },
    {
      "dynamics_loss_avg": 0.0589525131508708,
      "epoch": 8.897781372070313e-06,
      "step": 9330
    },
    {
      "action_loss_avg": 0.005725507368333637,
      "epoch": 8.897781372070313e-06,
      "step": 9330
    },
    {
      "epoch": 8.897781372070313e-06,
      "step": 9330,
      "training_step_time": 8.480414867401123
    },
    {
      "epoch": 8.898735046386718e-06,
      "model_forward_time": 2.3534791469573975,
      "step": 9331
    },
    {
      "epoch": 8.898735046386718e-06,
      "step": 9331,
      "training_step_time": 8.55235481262207
    },
    {
      "epoch": 8.899688720703126e-06,
      "model_forward_time": 1.8168442249298096,
      "step": 9332
    },
    {
      "epoch": 8.899688720703126e-06,
      "step": 9332,
      "training_step_time": 7.656768798828125
    },
    {
      "epoch": 8.900642395019531e-06,
      "model_forward_time": 1.8139762878417969,
      "step": 9333
    },
    {
      "epoch": 8.900642395019531e-06,
      "step": 9333,
      "training_step_time": 9.791685104370117
    },
    {
      "epoch": 8.901596069335937e-06,
      "model_forward_time": 1.8140788078308105,
      "step": 9334
    },
    {
      "epoch": 8.901596069335937e-06,
      "step": 9334,
      "training_step_time": 8.660855054855347
    },
    {
      "epoch": 8.902549743652345e-06,
      "model_forward_time": 1.8193984031677246,
      "step": 9335
    },
    {
      "epoch": 8.902549743652345e-06,
      "step": 9335,
      "training_step_time": 7.691559076309204
    },
    {
      "epoch": 8.90350341796875e-06,
      "model_forward_time": 1.8139634132385254,
      "step": 9336
    },
    {
      "epoch": 8.90350341796875e-06,
      "step": 9336,
      "training_step_time": 7.457542896270752
    },
    {
      "epoch": 8.904457092285156e-06,
      "model_forward_time": 1.8121099472045898,
      "step": 9337
    },
    {
      "epoch": 8.904457092285156e-06,
      "step": 9337,
      "training_step_time": 6.764478921890259
    },
    {
      "epoch": 8.905410766601562e-06,
      "model_forward_time": 1.8060798645019531,
      "step": 9338
    },
    {
      "epoch": 8.905410766601562e-06,
      "step": 9338,
      "training_step_time": 8.336732625961304
    },
    {
      "epoch": 8.906364440917969e-06,
      "model_forward_time": 1.8119423389434814,
      "step": 9339
    },
    {
      "epoch": 8.906364440917969e-06,
      "step": 9339,
      "training_step_time": 8.24784541130066
    },
    {
      "epoch": 8.907318115234375e-06,
      "grad_norm": 0.23019267618656158,
      "learning_rate": 9.948616157128695e-06,
      "loss": 0.0706,
      "step": 9340
    },
    {
      "epoch": 8.907318115234375e-06,
      "model_forward_time": 1.8081996440887451,
      "step": 9340
    },
    {
      "dynamics_loss_avg": 0.05075568463653326,
      "epoch": 8.907318115234375e-06,
      "step": 9340
    },
    {
      "action_loss_avg": 0.004536552896024659,
      "epoch": 8.907318115234375e-06,
      "step": 9340
    },
    {
      "epoch": 8.907318115234375e-06,
      "step": 9340,
      "training_step_time": 10.394459962844849
    },
    {
      "epoch": 8.90827178955078e-06,
      "model_forward_time": 1.821014165878296,
      "step": 9341
    },
    {
      "epoch": 8.90827178955078e-06,
      "step": 9341,
      "training_step_time": 8.647456645965576
    },
    {
      "epoch": 8.909225463867188e-06,
      "model_forward_time": 1.8166930675506592,
      "step": 9342
    },
    {
      "epoch": 8.909225463867188e-06,
      "step": 9342,
      "training_step_time": 11.291912317276001
    },
    {
      "epoch": 8.910179138183594e-06,
      "model_forward_time": 1.8139233589172363,
      "step": 9343
    },
    {
      "epoch": 8.910179138183594e-06,
      "step": 9343,
      "training_step_time": 8.227955341339111
    },
    {
      "epoch": 8.9111328125e-06,
      "model_forward_time": 1.811249017715454,
      "step": 9344
    },
    {
      "epoch": 8.9111328125e-06,
      "step": 9344,
      "training_step_time": 7.793359279632568
    },
    {
      "epoch": 8.912086486816407e-06,
      "model_forward_time": 1.8133583068847656,
      "step": 9345
    },
    {
      "epoch": 8.912086486816407e-06,
      "step": 9345,
      "training_step_time": 7.579487323760986
    },
    {
      "epoch": 8.913040161132813e-06,
      "model_forward_time": 1.8135898113250732,
      "step": 9346
    },
    {
      "epoch": 8.913040161132813e-06,
      "step": 9346,
      "training_step_time": 6.756864547729492
    },
    {
      "epoch": 8.913993835449218e-06,
      "model_forward_time": 1.7929613590240479,
      "step": 9347
    },
    {
      "epoch": 8.913993835449218e-06,
      "step": 9347,
      "training_step_time": 7.636565208435059
    },
    {
      "epoch": 8.914947509765626e-06,
      "model_forward_time": 1.799466609954834,
      "step": 9348
    },
    {
      "epoch": 8.914947509765626e-06,
      "step": 9348,
      "training_step_time": 7.6665732860565186
    },
    {
      "epoch": 8.915901184082031e-06,
      "model_forward_time": 1.8091154098510742,
      "step": 9349
    },
    {
      "epoch": 8.915901184082031e-06,
      "step": 9349,
      "training_step_time": 8.671108722686768
    },
    {
      "epoch": 8.916854858398437e-06,
      "grad_norm": 0.23569466173648834,
      "learning_rate": 9.948379446576492e-06,
      "loss": 0.0727,
      "step": 9350
    },
    {
      "epoch": 8.916854858398437e-06,
      "model_forward_time": 2.3272745609283447,
      "step": 9350
    },
    {
      "dynamics_loss_avg": 0.05892320442944765,
      "epoch": 8.916854858398437e-06,
      "step": 9350
    },
    {
      "action_loss_avg": 0.006015577609650791,
      "epoch": 8.916854858398437e-06,
      "step": 9350
    },
    {
      "epoch": 8.916854858398437e-06,
      "step": 9350,
      "training_step_time": 9.632730722427368
    },
    {
      "epoch": 8.917808532714845e-06,
      "model_forward_time": 1.8049123287200928,
      "step": 9351
    },
    {
      "epoch": 8.917808532714845e-06,
      "step": 9351,
      "training_step_time": 7.736819744110107
    },
    {
      "epoch": 8.91876220703125e-06,
      "model_forward_time": 1.8045949935913086,
      "step": 9352
    },
    {
      "epoch": 8.91876220703125e-06,
      "step": 9352,
      "training_step_time": 7.877577066421509
    },
    {
      "epoch": 8.919715881347656e-06,
      "model_forward_time": 1.8034217357635498,
      "step": 9353
    },
    {
      "epoch": 8.919715881347656e-06,
      "step": 9353,
      "training_step_time": 7.615053176879883
    },
    {
      "epoch": 8.920669555664062e-06,
      "model_forward_time": 1.8049376010894775,
      "step": 9354
    },
    {
      "epoch": 8.920669555664062e-06,
      "step": 9354,
      "training_step_time": 7.63442325592041
    },
    {
      "epoch": 8.92162322998047e-06,
      "model_forward_time": 1.8043186664581299,
      "step": 9355
    },
    {
      "epoch": 8.92162322998047e-06,
      "step": 9355,
      "training_step_time": 6.800145626068115
    },
    {
      "epoch": 8.922576904296875e-06,
      "model_forward_time": 1.8047311305999756,
      "step": 9356
    },
    {
      "epoch": 8.922576904296875e-06,
      "step": 9356,
      "training_step_time": 7.648023843765259
    },
    {
      "epoch": 8.92353057861328e-06,
      "model_forward_time": 1.8049728870391846,
      "step": 9357
    },
    {
      "epoch": 8.92353057861328e-06,
      "step": 9357,
      "training_step_time": 7.644607067108154
    },
    {
      "epoch": 8.924484252929688e-06,
      "model_forward_time": 1.8077378273010254,
      "step": 9358
    },
    {
      "epoch": 8.924484252929688e-06,
      "step": 9358,
      "training_step_time": 8.699625253677368
    },
    {
      "epoch": 8.925437927246094e-06,
      "model_forward_time": 2.3224058151245117,
      "step": 9359
    },
    {
      "epoch": 8.925437927246094e-06,
      "step": 9359,
      "training_step_time": 7.693570375442505
    },
    {
      "epoch": 8.9263916015625e-06,
      "grad_norm": 0.3124989867210388,
      "learning_rate": 9.948142194876954e-06,
      "loss": 0.073,
      "step": 9360
    },
    {
      "epoch": 8.9263916015625e-06,
      "model_forward_time": 1.8079264163970947,
      "step": 9360
    },
    {
      "dynamics_loss_avg": 0.06910667419433594,
      "epoch": 8.9263916015625e-06,
      "step": 9360
    },
    {
      "action_loss_avg": 0.004440207849256694,
      "epoch": 8.9263916015625e-06,
      "step": 9360
    },
    {
      "epoch": 8.9263916015625e-06,
      "step": 9360,
      "training_step_time": 9.412666082382202
    },
    {
      "epoch": 8.927345275878907e-06,
      "model_forward_time": 1.8081049919128418,
      "step": 9361
    },
    {
      "epoch": 8.927345275878907e-06,
      "step": 9361,
      "training_step_time": 7.890379190444946
    },
    {
      "epoch": 8.928298950195313e-06,
      "model_forward_time": 1.8071744441986084,
      "step": 9362
    },
    {
      "epoch": 8.928298950195313e-06,
      "step": 9362,
      "training_step_time": 7.7752416133880615
    },
    {
      "epoch": 8.929252624511718e-06,
      "model_forward_time": 1.7974772453308105,
      "step": 9363
    },
    {
      "epoch": 8.929252624511718e-06,
      "step": 9363,
      "training_step_time": 7.5254762172698975
    },
    {
      "epoch": 8.930206298828126e-06,
      "model_forward_time": 1.8025765419006348,
      "step": 9364
    },
    {
      "epoch": 8.930206298828126e-06,
      "step": 9364,
      "training_step_time": 7.441787004470825
    },
    {
      "epoch": 8.931159973144531e-06,
      "model_forward_time": 1.7894587516784668,
      "step": 9365
    },
    {
      "epoch": 8.931159973144531e-06,
      "step": 9365,
      "training_step_time": 8.11290192604065
    },
    {
      "epoch": 8.932113647460937e-06,
      "model_forward_time": 1.7808988094329834,
      "step": 9366
    },
    {
      "epoch": 8.932113647460937e-06,
      "step": 9366,
      "training_step_time": 7.457050323486328
    },
    {
      "epoch": 8.933067321777345e-06,
      "model_forward_time": 1.800445556640625,
      "step": 9367
    },
    {
      "epoch": 8.933067321777345e-06,
      "step": 9367,
      "training_step_time": 7.6617348194122314
    },
    {
      "epoch": 8.93402099609375e-06,
      "model_forward_time": 2.3288586139678955,
      "step": 9368
    },
    {
      "epoch": 8.93402099609375e-06,
      "step": 9368,
      "training_step_time": 7.578667640686035
    },
    {
      "epoch": 8.934974670410156e-06,
      "model_forward_time": 1.80643630027771,
      "step": 9369
    },
    {
      "epoch": 8.934974670410156e-06,
      "step": 9369,
      "training_step_time": 7.825787544250488
    },
    {
      "epoch": 8.935928344726562e-06,
      "grad_norm": 0.5265482664108276,
      "learning_rate": 9.947904402056024e-06,
      "loss": 0.0795,
      "step": 9370
    },
    {
      "epoch": 8.935928344726562e-06,
      "model_forward_time": 1.7985715866088867,
      "step": 9370
    },
    {
      "dynamics_loss_avg": 0.07484718766063451,
      "epoch": 8.935928344726562e-06,
      "step": 9370
    },
    {
      "action_loss_avg": 0.009473787539172917,
      "epoch": 8.935928344726562e-06,
      "step": 9370
    },
    {
      "epoch": 8.935928344726562e-06,
      "step": 9370,
      "training_step_time": 7.848014831542969
    },
    {
      "epoch": 8.93688201904297e-06,
      "model_forward_time": 1.8067607879638672,
      "step": 9371
    },
    {
      "epoch": 8.93688201904297e-06,
      "step": 9371,
      "training_step_time": 8.598395586013794
    },
    {
      "epoch": 8.937835693359375e-06,
      "model_forward_time": 1.8034250736236572,
      "step": 9372
    },
    {
      "epoch": 8.937835693359375e-06,
      "step": 9372,
      "training_step_time": 8.750689268112183
    },
    {
      "epoch": 8.93878936767578e-06,
      "model_forward_time": 1.8044118881225586,
      "step": 9373
    },
    {
      "epoch": 8.93878936767578e-06,
      "step": 9373,
      "training_step_time": 7.461995601654053
    },
    {
      "epoch": 8.939743041992188e-06,
      "model_forward_time": 1.8087069988250732,
      "step": 9374
    },
    {
      "epoch": 8.939743041992188e-06,
      "step": 9374,
      "training_step_time": 7.996168851852417
    },
    {
      "epoch": 8.940696716308594e-06,
      "model_forward_time": 1.8049137592315674,
      "step": 9375
    },
    {
      "epoch": 8.940696716308594e-06,
      "step": 9375,
      "training_step_time": 7.363889932632446
    },
    {
      "epoch": 8.941650390625e-06,
      "model_forward_time": 1.8062119483947754,
      "step": 9376
    },
    {
      "epoch": 8.941650390625e-06,
      "step": 9376,
      "training_step_time": 8.189198017120361
    },
    {
      "epoch": 8.942604064941407e-06,
      "model_forward_time": 2.3442671298980713,
      "step": 9377
    },
    {
      "epoch": 8.942604064941407e-06,
      "step": 9377,
      "training_step_time": 8.034796237945557
    },
    {
      "epoch": 8.943557739257813e-06,
      "model_forward_time": 1.435662031173706,
      "step": 9378
    },
    {
      "epoch": 8.943557739257813e-06,
      "step": 9378,
      "training_step_time": 9.639743089675903
    },
    {
      "epoch": 8.944511413574218e-06,
      "model_forward_time": 1.8031885623931885,
      "step": 9379
    },
    {
      "epoch": 8.944511413574218e-06,
      "step": 9379,
      "training_step_time": 8.318277359008789
    },
    {
      "epoch": 8.945465087890626e-06,
      "grad_norm": 0.4645969271659851,
      "learning_rate": 9.94766606813971e-06,
      "loss": 0.072,
      "step": 9380
    },
    {
      "epoch": 8.945465087890626e-06,
      "model_forward_time": 1.8010950088500977,
      "step": 9380
    },
    {
      "dynamics_loss_avg": 0.062128414213657376,
      "epoch": 8.945465087890626e-06,
      "step": 9380
    },
    {
      "action_loss_avg": 0.007514966651797295,
      "epoch": 8.945465087890626e-06,
      "step": 9380
    },
    {
      "epoch": 8.945465087890626e-06,
      "step": 9380,
      "training_step_time": 7.644258737564087
    },
    {
      "epoch": 8.946418762207031e-06,
      "model_forward_time": 1.8092875480651855,
      "step": 9381
    },
    {
      "epoch": 8.946418762207031e-06,
      "step": 9381,
      "training_step_time": 7.644067764282227
    },
    {
      "epoch": 8.947372436523437e-06,
      "model_forward_time": 1.4411962032318115,
      "step": 9382
    },
    {
      "epoch": 8.947372436523437e-06,
      "step": 9382,
      "training_step_time": 7.418744802474976
    },
    {
      "epoch": 8.948326110839843e-06,
      "model_forward_time": 1.8021714687347412,
      "step": 9383
    },
    {
      "epoch": 8.948326110839843e-06,
      "step": 9383,
      "training_step_time": 7.0512471199035645
    },
    {
      "epoch": 8.94927978515625e-06,
      "model_forward_time": 1.7948486804962158,
      "step": 9384
    },
    {
      "epoch": 8.94927978515625e-06,
      "step": 9384,
      "training_step_time": 7.526629686355591
    },
    {
      "epoch": 8.950233459472656e-06,
      "model_forward_time": 1.8038172721862793,
      "step": 9385
    },
    {
      "epoch": 8.950233459472656e-06,
      "step": 9385,
      "training_step_time": 8.389076948165894
    },
    {
      "epoch": 8.951187133789062e-06,
      "model_forward_time": 1.80106782913208,
      "step": 9386
    },
    {
      "epoch": 8.951187133789062e-06,
      "step": 9386,
      "training_step_time": 8.469034433364868
    },
    {
      "epoch": 8.95214080810547e-06,
      "model_forward_time": 2.340754985809326,
      "step": 9387
    },
    {
      "epoch": 8.95214080810547e-06,
      "step": 9387,
      "training_step_time": 8.669922828674316
    },
    {
      "epoch": 8.953094482421875e-06,
      "model_forward_time": 1.804267406463623,
      "step": 9388
    },
    {
      "epoch": 8.953094482421875e-06,
      "step": 9388,
      "training_step_time": 8.798498153686523
    },
    {
      "epoch": 8.95404815673828e-06,
      "model_forward_time": 1.7998309135437012,
      "step": 9389
    },
    {
      "epoch": 8.95404815673828e-06,
      "step": 9389,
      "training_step_time": 7.7225892543792725
    },
    {
      "epoch": 8.955001831054688e-06,
      "grad_norm": 0.34000223875045776,
      "learning_rate": 9.94742719315407e-06,
      "loss": 0.072,
      "step": 9390
    },
    {
      "epoch": 8.955001831054688e-06,
      "model_forward_time": 1.8063287734985352,
      "step": 9390
    },
    {
      "dynamics_loss_avg": 0.08088878765702248,
      "epoch": 8.955001831054688e-06,
      "step": 9390
    },
    {
      "action_loss_avg": 0.008373416471295058,
      "epoch": 8.955001831054688e-06,
      "step": 9390
    },
    {
      "epoch": 8.955001831054688e-06,
      "step": 9390,
      "training_step_time": 7.693528175354004
    },
    {
      "epoch": 8.955955505371094e-06,
      "model_forward_time": 1.8074724674224854,
      "step": 9391
    },
    {
      "epoch": 8.955955505371094e-06,
      "step": 9391,
      "training_step_time": 7.541030645370483
    },
    {
      "epoch": 8.9569091796875e-06,
      "model_forward_time": 1.8034465312957764,
      "step": 9392
    },
    {
      "epoch": 8.9569091796875e-06,
      "step": 9392,
      "training_step_time": 7.076465845108032
    },
    {
      "epoch": 8.957862854003907e-06,
      "model_forward_time": 1.7998332977294922,
      "step": 9393
    },
    {
      "epoch": 8.957862854003907e-06,
      "step": 9393,
      "training_step_time": 7.533084392547607
    },
    {
      "epoch": 8.958816528320313e-06,
      "model_forward_time": 1.8033766746520996,
      "step": 9394
    },
    {
      "epoch": 8.958816528320313e-06,
      "step": 9394,
      "training_step_time": 7.67500901222229
    },
    {
      "epoch": 8.959770202636718e-06,
      "model_forward_time": 1.7974460124969482,
      "step": 9395
    },
    {
      "epoch": 8.959770202636718e-06,
      "step": 9395,
      "training_step_time": 8.148547649383545
    },
    {
      "epoch": 8.960723876953126e-06,
      "model_forward_time": 2.316053628921509,
      "step": 9396
    },
    {
      "epoch": 8.960723876953126e-06,
      "step": 9396,
      "training_step_time": 9.205381393432617
    },
    {
      "epoch": 8.961677551269531e-06,
      "model_forward_time": 1.8040947914123535,
      "step": 9397
    },
    {
      "epoch": 8.961677551269531e-06,
      "step": 9397,
      "training_step_time": 7.714015007019043
    },
    {
      "epoch": 8.962631225585937e-06,
      "model_forward_time": 1.801210880279541,
      "step": 9398
    },
    {
      "epoch": 8.962631225585937e-06,
      "step": 9398,
      "training_step_time": 8.636107206344604
    },
    {
      "epoch": 8.963584899902343e-06,
      "model_forward_time": 1.7993509769439697,
      "step": 9399
    },
    {
      "epoch": 8.963584899902343e-06,
      "step": 9399,
      "training_step_time": 8.699087381362915
    },
    {
      "epoch": 8.96453857421875e-06,
      "grad_norm": 0.18006131052970886,
      "learning_rate": 9.947187777125234e-06,
      "loss": 0.0687,
      "step": 9400
    },
    {
      "epoch": 8.96453857421875e-06,
      "model_forward_time": 1.7965710163116455,
      "step": 9400
    },
    {
      "dynamics_loss_avg": 0.06565685253590345,
      "epoch": 8.96453857421875e-06,
      "step": 9400
    },
    {
      "action_loss_avg": 0.005134954478126019,
      "epoch": 8.96453857421875e-06,
      "step": 9400
    },
    {
      "epoch": 8.96453857421875e-06,
      "step": 9400,
      "training_step_time": 7.527047395706177
    },
    {
      "epoch": 8.965492248535156e-06,
      "model_forward_time": 1.8073656558990479,
      "step": 9401
    },
    {
      "epoch": 8.965492248535156e-06,
      "step": 9401,
      "training_step_time": 7.079567909240723
    },
    {
      "epoch": 8.966445922851562e-06,
      "model_forward_time": 1.806347131729126,
      "step": 9402
    },
    {
      "epoch": 8.966445922851562e-06,
      "step": 9402,
      "training_step_time": 7.543012619018555
    },
    {
      "epoch": 8.96739959716797e-06,
      "model_forward_time": 1.805699110031128,
      "step": 9403
    },
    {
      "epoch": 8.96739959716797e-06,
      "step": 9403,
      "training_step_time": 7.674419164657593
    },
    {
      "epoch": 8.968353271484375e-06,
      "model_forward_time": 1.7989730834960938,
      "step": 9404
    },
    {
      "epoch": 8.968353271484375e-06,
      "step": 9404,
      "training_step_time": 8.587005615234375
    },
    {
      "epoch": 8.96930694580078e-06,
      "model_forward_time": 2.340122699737549,
      "step": 9405
    },
    {
      "epoch": 8.96930694580078e-06,
      "step": 9405,
      "training_step_time": 8.470025300979614
    },
    {
      "epoch": 8.970260620117188e-06,
      "model_forward_time": 1.812885046005249,
      "step": 9406
    },
    {
      "epoch": 8.970260620117188e-06,
      "step": 9406,
      "training_step_time": 8.527126789093018
    },
    {
      "epoch": 8.971214294433594e-06,
      "model_forward_time": 1.8040797710418701,
      "step": 9407
    },
    {
      "epoch": 8.971214294433594e-06,
      "step": 9407,
      "training_step_time": 8.571781158447266
    },
    {
      "epoch": 8.97216796875e-06,
      "model_forward_time": 1.8031835556030273,
      "step": 9408
    },
    {
      "epoch": 8.97216796875e-06,
      "step": 9408,
      "training_step_time": 7.798632621765137
    },
    {
      "epoch": 8.973121643066407e-06,
      "model_forward_time": 1.8064277172088623,
      "step": 9409
    },
    {
      "epoch": 8.973121643066407e-06,
      "step": 9409,
      "training_step_time": 7.667204856872559
    },
    {
      "epoch": 8.974075317382813e-06,
      "grad_norm": 0.2935458719730377,
      "learning_rate": 9.946947820079378e-06,
      "loss": 0.0678,
      "step": 9410
    },
    {
      "epoch": 8.974075317382813e-06,
      "model_forward_time": 1.431164026260376,
      "step": 9410
    },
    {
      "dynamics_loss_avg": 0.060791699588298796,
      "epoch": 8.974075317382813e-06,
      "step": 9410
    },
    {
      "action_loss_avg": 0.007664263376500458,
      "epoch": 8.974075317382813e-06,
      "step": 9410
    },
    {
      "epoch": 8.974075317382813e-06,
      "step": 9410,
      "training_step_time": 7.412224054336548
    },
    {
      "epoch": 8.975028991699218e-06,
      "model_forward_time": 1.8037984371185303,
      "step": 9411
    },
    {
      "epoch": 8.975028991699218e-06,
      "step": 9411,
      "training_step_time": 7.581415414810181
    },
    {
      "epoch": 8.975982666015626e-06,
      "model_forward_time": 1.8026282787322998,
      "step": 9412
    },
    {
      "epoch": 8.975982666015626e-06,
      "step": 9412,
      "training_step_time": 7.663339138031006
    },
    {
      "epoch": 8.976936340332031e-06,
      "model_forward_time": 1.7980141639709473,
      "step": 9413
    },
    {
      "epoch": 8.976936340332031e-06,
      "step": 9413,
      "training_step_time": 7.558151960372925
    },
    {
      "epoch": 8.977890014648437e-06,
      "model_forward_time": 2.330113649368286,
      "step": 9414
    },
    {
      "epoch": 8.977890014648437e-06,
      "step": 9414,
      "training_step_time": 8.271031379699707
    },
    {
      "epoch": 8.978843688964843e-06,
      "model_forward_time": 1.8127241134643555,
      "step": 9415
    },
    {
      "epoch": 8.978843688964843e-06,
      "step": 9415,
      "training_step_time": 8.233285665512085
    },
    {
      "epoch": 8.97979736328125e-06,
      "model_forward_time": 1.8099710941314697,
      "step": 9416
    },
    {
      "epoch": 8.97979736328125e-06,
      "step": 9416,
      "training_step_time": 7.778332710266113
    },
    {
      "epoch": 8.980751037597656e-06,
      "model_forward_time": 1.7879807949066162,
      "step": 9417
    },
    {
      "epoch": 8.980751037597656e-06,
      "step": 9417,
      "training_step_time": 7.631103277206421
    },
    {
      "epoch": 8.981704711914062e-06,
      "model_forward_time": 1.8082270622253418,
      "step": 9418
    },
    {
      "epoch": 8.981704711914062e-06,
      "step": 9418,
      "training_step_time": 8.471035242080688
    },
    {
      "epoch": 8.98265838623047e-06,
      "model_forward_time": 1.813979148864746,
      "step": 9419
    },
    {
      "epoch": 8.98265838623047e-06,
      "step": 9419,
      "training_step_time": 7.496755361557007
    },
    {
      "epoch": 8.983612060546875e-06,
      "grad_norm": 0.24651388823986053,
      "learning_rate": 9.946707322042746e-06,
      "loss": 0.0692,
      "step": 9420
    },
    {
      "epoch": 8.983612060546875e-06,
      "model_forward_time": 1.817694902420044,
      "step": 9420
    },
    {
      "dynamics_loss_avg": 0.05162000218406319,
      "epoch": 8.983612060546875e-06,
      "step": 9420
    },
    {
      "action_loss_avg": 0.00484833000227809,
      "epoch": 8.983612060546875e-06,
      "step": 9420
    },
    {
      "epoch": 8.983612060546875e-06,
      "step": 9420,
      "training_step_time": 7.51688814163208
    },
    {
      "epoch": 8.98456573486328e-06,
      "model_forward_time": 1.8303821086883545,
      "step": 9421
    },
    {
      "epoch": 8.98456573486328e-06,
      "step": 9421,
      "training_step_time": 9.403317213058472
    },
    {
      "epoch": 8.985519409179688e-06,
      "model_forward_time": 1.8274922370910645,
      "step": 9422
    },
    {
      "epoch": 8.985519409179688e-06,
      "step": 9422,
      "training_step_time": 7.599246263504028
    },
    {
      "epoch": 8.986473083496094e-06,
      "model_forward_time": 1.8212428092956543,
      "step": 9423
    },
    {
      "epoch": 8.986473083496094e-06,
      "step": 9423,
      "training_step_time": 8.359593868255615
    },
    {
      "epoch": 8.9874267578125e-06,
      "model_forward_time": 2.3783133029937744,
      "step": 9424
    },
    {
      "epoch": 8.9874267578125e-06,
      "step": 9424,
      "training_step_time": 8.405929565429688
    },
    {
      "epoch": 8.988380432128907e-06,
      "model_forward_time": 1.8252842426300049,
      "step": 9425
    },
    {
      "epoch": 8.988380432128907e-06,
      "step": 9425,
      "training_step_time": 9.218008041381836
    },
    {
      "epoch": 8.989334106445313e-06,
      "model_forward_time": 1.8254516124725342,
      "step": 9426
    },
    {
      "epoch": 8.989334106445313e-06,
      "step": 9426,
      "training_step_time": 7.674936294555664
    },
    {
      "epoch": 8.990287780761718e-06,
      "model_forward_time": 1.8262691497802734,
      "step": 9427
    },
    {
      "epoch": 8.990287780761718e-06,
      "step": 9427,
      "training_step_time": 7.749911546707153
    },
    {
      "epoch": 8.991241455078126e-06,
      "model_forward_time": 1.8315279483795166,
      "step": 9428
    },
    {
      "epoch": 8.991241455078126e-06,
      "step": 9428,
      "training_step_time": 7.51763129234314
    },
    {
      "epoch": 8.992195129394532e-06,
      "model_forward_time": 1.8356781005859375,
      "step": 9429
    },
    {
      "epoch": 8.992195129394532e-06,
      "step": 9429,
      "training_step_time": 7.5725672245025635
    },
    {
      "epoch": 8.993148803710937e-06,
      "grad_norm": 0.2688181698322296,
      "learning_rate": 9.94646628304164e-06,
      "loss": 0.0735,
      "step": 9430
    },
    {
      "epoch": 8.993148803710937e-06,
      "model_forward_time": 1.826920986175537,
      "step": 9430
    },
    {
      "dynamics_loss_avg": 0.0676863556727767,
      "epoch": 8.993148803710937e-06,
      "step": 9430
    },
    {
      "action_loss_avg": 0.0037165005458518864,
      "epoch": 8.993148803710937e-06,
      "step": 9430
    },
    {
      "epoch": 8.993148803710937e-06,
      "step": 9430,
      "training_step_time": 7.642934083938599
    },
    {
      "epoch": 8.994102478027343e-06,
      "model_forward_time": 1.8239471912384033,
      "step": 9431
    },
    {
      "epoch": 8.994102478027343e-06,
      "step": 9431,
      "training_step_time": 7.522336006164551
    },
    {
      "epoch": 8.99505615234375e-06,
      "model_forward_time": 1.8178350925445557,
      "step": 9432
    },
    {
      "epoch": 8.99505615234375e-06,
      "step": 9432,
      "training_step_time": 8.315322399139404
    },
    {
      "epoch": 8.996009826660156e-06,
      "model_forward_time": 2.3577065467834473,
      "step": 9433
    },
    {
      "epoch": 8.996009826660156e-06,
      "step": 9433,
      "training_step_time": 8.205609321594238
    },
    {
      "epoch": 8.996963500976562e-06,
      "model_forward_time": 1.8206052780151367,
      "step": 9434
    },
    {
      "epoch": 8.996963500976562e-06,
      "step": 9434,
      "training_step_time": 7.770261287689209
    },
    {
      "epoch": 8.99791717529297e-06,
      "model_forward_time": 1.8207824230194092,
      "step": 9435
    },
    {
      "epoch": 8.99791717529297e-06,
      "step": 9435,
      "training_step_time": 8.686326742172241
    },
    {
      "epoch": 8.998870849609375e-06,
      "model_forward_time": 1.812523365020752,
      "step": 9436
    },
    {
      "epoch": 8.998870849609375e-06,
      "step": 9436,
      "training_step_time": 7.675368070602417
    },
    {
      "epoch": 8.99982452392578e-06,
      "model_forward_time": 1.814802646636963,
      "step": 9437
    },
    {
      "epoch": 8.99982452392578e-06,
      "step": 9437,
      "training_step_time": 7.418865442276001
    },
    {
      "epoch": 9.000778198242188e-06,
      "model_forward_time": 1.8158369064331055,
      "step": 9438
    },
    {
      "epoch": 9.000778198242188e-06,
      "step": 9438,
      "training_step_time": 7.569131135940552
    },
    {
      "epoch": 9.001731872558594e-06,
      "model_forward_time": 1.8203907012939453,
      "step": 9439
    },
    {
      "epoch": 9.001731872558594e-06,
      "step": 9439,
      "training_step_time": 7.71203875541687
    },
    {
      "epoch": 9.002685546875e-06,
      "grad_norm": 0.2105739712715149,
      "learning_rate": 9.946224703102419e-06,
      "loss": 0.0702,
      "step": 9440
    },
    {
      "epoch": 9.002685546875e-06,
      "model_forward_time": 1.454200029373169,
      "step": 9440
    },
    {
      "dynamics_loss_avg": 0.06298065008595585,
      "epoch": 9.002685546875e-06,
      "step": 9440
    },
    {
      "action_loss_avg": 0.003627348714508116,
      "epoch": 9.002685546875e-06,
      "step": 9440
    },
    {
      "epoch": 9.002685546875e-06,
      "step": 9440,
      "training_step_time": 7.552396535873413
    },
    {
      "epoch": 9.003639221191407e-06,
      "model_forward_time": 1.8201336860656738,
      "step": 9441
    },
    {
      "epoch": 9.003639221191407e-06,
      "step": 9441,
      "training_step_time": 7.6024744510650635
    },
    {
      "epoch": 9.004592895507813e-06,
      "model_forward_time": 2.3602824211120605,
      "step": 9442
    },
    {
      "epoch": 9.004592895507813e-06,
      "step": 9442,
      "training_step_time": 9.255489826202393
    },
    {
      "epoch": 9.005546569824218e-06,
      "model_forward_time": 1.787381649017334,
      "step": 9443
    },
    {
      "epoch": 9.005546569824218e-06,
      "step": 9443,
      "training_step_time": 8.764435052871704
    },
    {
      "epoch": 9.006500244140626e-06,
      "model_forward_time": 1.8062589168548584,
      "step": 9444
    },
    {
      "epoch": 9.006500244140626e-06,
      "step": 9444,
      "training_step_time": 7.669931650161743
    },
    {
      "epoch": 9.007453918457032e-06,
      "model_forward_time": 1.807021141052246,
      "step": 9445
    },
    {
      "epoch": 9.007453918457032e-06,
      "step": 9445,
      "training_step_time": 7.7296953201293945
    },
    {
      "epoch": 9.008407592773437e-06,
      "model_forward_time": 1.805006742477417,
      "step": 9446
    },
    {
      "epoch": 9.008407592773437e-06,
      "step": 9446,
      "training_step_time": 8.630979776382446
    },
    {
      "epoch": 9.009361267089843e-06,
      "model_forward_time": 1.7925045490264893,
      "step": 9447
    },
    {
      "epoch": 9.009361267089843e-06,
      "step": 9447,
      "training_step_time": 7.617865562438965
    },
    {
      "epoch": 9.01031494140625e-06,
      "model_forward_time": 1.807304859161377,
      "step": 9448
    },
    {
      "epoch": 9.01031494140625e-06,
      "step": 9448,
      "training_step_time": 7.709078311920166
    },
    {
      "epoch": 9.011268615722656e-06,
      "model_forward_time": 1.805650234222412,
      "step": 9449
    },
    {
      "epoch": 9.011268615722656e-06,
      "step": 9449,
      "training_step_time": 7.597628116607666
    },
    {
      "epoch": 9.012222290039062e-06,
      "grad_norm": 0.1862318217754364,
      "learning_rate": 9.945982582251498e-06,
      "loss": 0.0706,
      "step": 9450
    },
    {
      "epoch": 9.012222290039062e-06,
      "model_forward_time": 1.8027255535125732,
      "step": 9450
    },
    {
      "dynamics_loss_avg": 0.05668600825592875,
      "epoch": 9.012222290039062e-06,
      "step": 9450
    },
    {
      "action_loss_avg": 0.004690873145591468,
      "epoch": 9.012222290039062e-06,
      "step": 9450
    },
    {
      "epoch": 9.012222290039062e-06,
      "step": 9450,
      "training_step_time": 8.267796754837036
    },
    {
      "epoch": 9.01317596435547e-06,
      "model_forward_time": 2.324392557144165,
      "step": 9451
    },
    {
      "epoch": 9.01317596435547e-06,
      "step": 9451,
      "training_step_time": 8.793487787246704
    },
    {
      "epoch": 9.014129638671875e-06,
      "model_forward_time": 1.8023290634155273,
      "step": 9452
    },
    {
      "epoch": 9.014129638671875e-06,
      "step": 9452,
      "training_step_time": 7.717888116836548
    },
    {
      "epoch": 9.01508331298828e-06,
      "model_forward_time": 1.708390712738037,
      "step": 9453
    },
    {
      "epoch": 9.01508331298828e-06,
      "step": 9453,
      "training_step_time": 7.8070385456085205
    },
    {
      "epoch": 9.016036987304688e-06,
      "model_forward_time": 1.7033717632293701,
      "step": 9454
    },
    {
      "epoch": 9.016036987304688e-06,
      "step": 9454,
      "training_step_time": 7.805210590362549
    },
    {
      "epoch": 9.016990661621094e-06,
      "model_forward_time": 1.7124464511871338,
      "step": 9455
    },
    {
      "epoch": 9.016990661621094e-06,
      "step": 9455,
      "training_step_time": 7.772323131561279
    },
    {
      "epoch": 9.0179443359375e-06,
      "model_forward_time": 1.7145414352416992,
      "step": 9456
    },
    {
      "epoch": 9.0179443359375e-06,
      "step": 9456,
      "training_step_time": 7.67008638381958
    },
    {
      "epoch": 9.018898010253907e-06,
      "model_forward_time": 1.7086286544799805,
      "step": 9457
    },
    {
      "epoch": 9.018898010253907e-06,
      "step": 9457,
      "training_step_time": 7.785810947418213
    },
    {
      "epoch": 9.019851684570313e-06,
      "model_forward_time": 1.7048728466033936,
      "step": 9458
    },
    {
      "epoch": 9.019851684570313e-06,
      "step": 9458,
      "training_step_time": 7.960275650024414
    },
    {
      "epoch": 9.020805358886718e-06,
      "model_forward_time": 1.7120966911315918,
      "step": 9459
    },
    {
      "epoch": 9.020805358886718e-06,
      "step": 9459,
      "training_step_time": 8.534420490264893
    },
    {
      "epoch": 9.021759033203126e-06,
      "grad_norm": 0.23598793148994446,
      "learning_rate": 9.94573992051536e-06,
      "loss": 0.0692,
      "step": 9460
    },
    {
      "epoch": 9.021759033203126e-06,
      "model_forward_time": 1.330596685409546,
      "step": 9460
    },
    {
      "dynamics_loss_avg": 0.05868142992258072,
      "epoch": 9.021759033203126e-06,
      "step": 9460
    },
    {
      "action_loss_avg": 0.006761838076636195,
      "epoch": 9.021759033203126e-06,
      "step": 9460
    },
    {
      "epoch": 9.021759033203126e-06,
      "step": 9460,
      "training_step_time": 9.048516511917114
    },
    {
      "epoch": 9.022712707519532e-06,
      "model_forward_time": 2.2512552738189697,
      "step": 9461
    },
    {
      "epoch": 9.022712707519532e-06,
      "step": 9461,
      "training_step_time": 8.744803190231323
    },
    {
      "epoch": 9.023666381835937e-06,
      "model_forward_time": 1.7497072219848633,
      "step": 9462
    },
    {
      "epoch": 9.023666381835937e-06,
      "step": 9462,
      "training_step_time": 7.603419303894043
    },
    {
      "epoch": 9.024620056152343e-06,
      "model_forward_time": 1.7106401920318604,
      "step": 9463
    },
    {
      "epoch": 9.024620056152343e-06,
      "step": 9463,
      "training_step_time": 7.609800577163696
    },
    {
      "epoch": 9.02557373046875e-06,
      "model_forward_time": 1.7101044654846191,
      "step": 9464
    },
    {
      "epoch": 9.02557373046875e-06,
      "step": 9464,
      "training_step_time": 7.560755014419556
    },
    {
      "epoch": 9.026527404785156e-06,
      "model_forward_time": 1.7095327377319336,
      "step": 9465
    },
    {
      "epoch": 9.026527404785156e-06,
      "step": 9465,
      "training_step_time": 7.457950115203857
    },
    {
      "epoch": 9.027481079101562e-06,
      "model_forward_time": 1.7080540657043457,
      "step": 9466
    },
    {
      "epoch": 9.027481079101562e-06,
      "step": 9466,
      "training_step_time": 7.558683156967163
    },
    {
      "epoch": 9.02843475341797e-06,
      "model_forward_time": 1.7195026874542236,
      "step": 9467
    },
    {
      "epoch": 9.02843475341797e-06,
      "step": 9467,
      "training_step_time": 8.342679738998413
    },
    {
      "epoch": 9.029388427734375e-06,
      "model_forward_time": 1.711418628692627,
      "step": 9468
    },
    {
      "epoch": 9.029388427734375e-06,
      "step": 9468,
      "training_step_time": 7.425450086593628
    },
    {
      "epoch": 9.03034210205078e-06,
      "model_forward_time": 1.6876447200775146,
      "step": 9469
    },
    {
      "epoch": 9.03034210205078e-06,
      "step": 9469,
      "training_step_time": 8.655640602111816
    },
    {
      "epoch": 9.031295776367188e-06,
      "grad_norm": 0.20817042887210846,
      "learning_rate": 9.94549671792054e-06,
      "loss": 0.0714,
      "step": 9470
    },
    {
      "epoch": 9.031295776367188e-06,
      "model_forward_time": 2.224799633026123,
      "step": 9470
    },
    {
      "dynamics_loss_avg": 0.07135848142206669,
      "epoch": 9.031295776367188e-06,
      "step": 9470
    },
    {
      "action_loss_avg": 0.0059415694326162335,
      "epoch": 9.031295776367188e-06,
      "step": 9470
    },
    {
      "epoch": 9.031295776367188e-06,
      "step": 9470,
      "training_step_time": 8.64305591583252
    },
    {
      "epoch": 9.032249450683594e-06,
      "model_forward_time": 1.7017467021942139,
      "step": 9471
    },
    {
      "epoch": 9.032249450683594e-06,
      "step": 9471,
      "training_step_time": 8.234755992889404
    },
    {
      "epoch": 9.033203125e-06,
      "model_forward_time": 1.7016513347625732,
      "step": 9472
    },
    {
      "epoch": 9.033203125e-06,
      "step": 9472,
      "training_step_time": 7.557462930679321
    },
    {
      "epoch": 9.034156799316407e-06,
      "model_forward_time": 1.7020268440246582,
      "step": 9473
    },
    {
      "epoch": 9.034156799316407e-06,
      "step": 9473,
      "training_step_time": 9.899093866348267
    },
    {
      "epoch": 9.035110473632813e-06,
      "model_forward_time": 1.7051630020141602,
      "step": 9474
    },
    {
      "epoch": 9.035110473632813e-06,
      "step": 9474,
      "training_step_time": 8.37403416633606
    },
    {
      "epoch": 9.036064147949218e-06,
      "model_forward_time": 1.699833631515503,
      "step": 9475
    },
    {
      "epoch": 9.036064147949218e-06,
      "step": 9475,
      "training_step_time": 7.567102432250977
    },
    {
      "epoch": 9.037017822265626e-06,
      "model_forward_time": 1.700789451599121,
      "step": 9476
    },
    {
      "epoch": 9.037017822265626e-06,
      "step": 9476,
      "training_step_time": 7.5049262046813965
    },
    {
      "epoch": 9.037971496582032e-06,
      "model_forward_time": 1.6977708339691162,
      "step": 9477
    },
    {
      "epoch": 9.037971496582032e-06,
      "step": 9477,
      "training_step_time": 7.457834720611572
    },
    {
      "epoch": 9.038925170898437e-06,
      "model_forward_time": 1.7009220123291016,
      "step": 9478
    },
    {
      "epoch": 9.038925170898437e-06,
      "step": 9478,
      "training_step_time": 7.776841402053833
    },
    {
      "epoch": 9.039878845214843e-06,
      "model_forward_time": 2.2448601722717285,
      "step": 9479
    },
    {
      "epoch": 9.039878845214843e-06,
      "step": 9479,
      "training_step_time": 8.129292249679565
    },
    {
      "epoch": 9.04083251953125e-06,
      "grad_norm": 0.3256925940513611,
      "learning_rate": 9.945252974493635e-06,
      "loss": 0.0684,
      "step": 9480
    },
    {
      "epoch": 9.04083251953125e-06,
      "model_forward_time": 1.711843729019165,
      "step": 9480
    },
    {
      "dynamics_loss_avg": 0.055854180827736855,
      "epoch": 9.04083251953125e-06,
      "step": 9480
    },
    {
      "action_loss_avg": 0.0039989595301449295,
      "epoch": 9.04083251953125e-06,
      "step": 9480
    },
    {
      "epoch": 9.04083251953125e-06,
      "step": 9480,
      "training_step_time": 8.735878944396973
    },
    {
      "epoch": 9.041786193847656e-06,
      "model_forward_time": 1.3419888019561768,
      "step": 9481
    },
    {
      "epoch": 9.041786193847656e-06,
      "step": 9481,
      "training_step_time": 7.670908212661743
    },
    {
      "epoch": 9.042739868164062e-06,
      "model_forward_time": 1.70635986328125,
      "step": 9482
    },
    {
      "epoch": 9.042739868164062e-06,
      "step": 9482,
      "training_step_time": 7.642070770263672
    },
    {
      "epoch": 9.04369354248047e-06,
      "model_forward_time": 1.7017548084259033,
      "step": 9483
    },
    {
      "epoch": 9.04369354248047e-06,
      "step": 9483,
      "training_step_time": 7.700362682342529
    },
    {
      "epoch": 9.044647216796875e-06,
      "model_forward_time": 1.7019076347351074,
      "step": 9484
    },
    {
      "epoch": 9.044647216796875e-06,
      "step": 9484,
      "training_step_time": 7.654446840286255
    },
    {
      "epoch": 9.04560089111328e-06,
      "model_forward_time": 1.7035109996795654,
      "step": 9485
    },
    {
      "epoch": 9.04560089111328e-06,
      "step": 9485,
      "training_step_time": 7.695599555969238
    },
    {
      "epoch": 9.046554565429688e-06,
      "model_forward_time": 1.6983442306518555,
      "step": 9486
    },
    {
      "epoch": 9.046554565429688e-06,
      "step": 9486,
      "training_step_time": 8.292921304702759
    },
    {
      "epoch": 9.047508239746094e-06,
      "model_forward_time": 1.7008781433105469,
      "step": 9487
    },
    {
      "epoch": 9.047508239746094e-06,
      "step": 9487,
      "training_step_time": 7.640878677368164
    },
    {
      "epoch": 9.0484619140625e-06,
      "model_forward_time": 2.241203546524048,
      "step": 9488
    },
    {
      "epoch": 9.0484619140625e-06,
      "step": 9488,
      "training_step_time": 8.49340271949768
    },
    {
      "epoch": 9.049415588378907e-06,
      "model_forward_time": 1.7054517269134521,
      "step": 9489
    },
    {
      "epoch": 9.049415588378907e-06,
      "step": 9489,
      "training_step_time": 9.491695642471313
    },
    {
      "epoch": 9.050369262695313e-06,
      "grad_norm": 0.21233287453651428,
      "learning_rate": 9.9450086902613e-06,
      "loss": 0.0681,
      "step": 9490
    },
    {
      "epoch": 9.050369262695313e-06,
      "model_forward_time": 1.7219176292419434,
      "step": 9490
    },
    {
      "dynamics_loss_avg": 0.05400148965418339,
      "epoch": 9.050369262695313e-06,
      "step": 9490
    },
    {
      "action_loss_avg": 0.004259068635292351,
      "epoch": 9.050369262695313e-06,
      "step": 9490
    },
    {
      "epoch": 9.050369262695313e-06,
      "step": 9490,
      "training_step_time": 7.6803812980651855
    },
    {
      "epoch": 9.051322937011718e-06,
      "model_forward_time": 1.6973464488983154,
      "step": 9491
    },
    {
      "epoch": 9.051322937011718e-06,
      "step": 9491,
      "training_step_time": 7.532549619674683
    },
    {
      "epoch": 9.052276611328126e-06,
      "model_forward_time": 1.704059362411499,
      "step": 9492
    },
    {
      "epoch": 9.052276611328126e-06,
      "step": 9492,
      "training_step_time": 7.621980905532837
    },
    {
      "epoch": 9.053230285644532e-06,
      "model_forward_time": 1.7066879272460938,
      "step": 9493
    },
    {
      "epoch": 9.053230285644532e-06,
      "step": 9493,
      "training_step_time": 8.579676628112793
    },
    {
      "epoch": 9.054183959960937e-06,
      "model_forward_time": 1.7205021381378174,
      "step": 9494
    },
    {
      "epoch": 9.054183959960937e-06,
      "step": 9494,
      "training_step_time": 7.571249961853027
    },
    {
      "epoch": 9.055137634277343e-06,
      "model_forward_time": 1.7247169017791748,
      "step": 9495
    },
    {
      "epoch": 9.055137634277343e-06,
      "step": 9495,
      "training_step_time": 8.834408044815063
    },
    {
      "epoch": 9.05609130859375e-06,
      "model_forward_time": 1.7168638706207275,
      "step": 9496
    },
    {
      "epoch": 9.05609130859375e-06,
      "step": 9496,
      "training_step_time": 8.632983207702637
    },
    {
      "epoch": 9.057044982910156e-06,
      "model_forward_time": 1.7286837100982666,
      "step": 9497
    },
    {
      "epoch": 9.057044982910156e-06,
      "step": 9497,
      "training_step_time": 8.34108018875122
    },
    {
      "epoch": 9.057998657226562e-06,
      "model_forward_time": 2.2624874114990234,
      "step": 9498
    },
    {
      "epoch": 9.057998657226562e-06,
      "step": 9498,
      "training_step_time": 9.370980978012085
    },
    {
      "epoch": 9.05895233154297e-06,
      "model_forward_time": 1.7741360664367676,
      "step": 9499
    },
    {
      "epoch": 9.05895233154297e-06,
      "step": 9499,
      "training_step_time": 7.772745847702026
    },
    {
      "epoch": 9.059906005859375e-06,
      "grad_norm": 0.1824738085269928,
      "learning_rate": 9.944763865250248e-06,
      "loss": 0.0673,
      "step": 9500
    },
    {
      "epoch": 9.059906005859375e-06,
      "model_forward_time": 1.7178072929382324,
      "step": 9500
    },
    {
      "dynamics_loss_avg": 0.0788472829386592,
      "epoch": 9.059906005859375e-06,
      "step": 9500
    },
    {
      "action_loss_avg": 0.005628000118304044,
      "epoch": 9.059906005859375e-06,
      "step": 9500
    },
    {
      "epoch": 9.059906005859375e-06,
      "step": 9500,
      "training_step_time": 7.446599960327148
    },
    {
      "epoch": 9.06085968017578e-06,
      "model_forward_time": 1.7183856964111328,
      "step": 9501
    },
    {
      "epoch": 9.06085968017578e-06,
      "step": 9501,
      "training_step_time": 7.717038869857788
    },
    {
      "epoch": 9.061813354492188e-06,
      "model_forward_time": 1.7290639877319336,
      "step": 9502
    },
    {
      "epoch": 9.061813354492188e-06,
      "step": 9502,
      "training_step_time": 7.549015283584595
    },
    {
      "epoch": 9.062767028808594e-06,
      "model_forward_time": 1.7190227508544922,
      "step": 9503
    },
    {
      "epoch": 9.062767028808594e-06,
      "step": 9503,
      "training_step_time": 7.553779602050781
    },
    {
      "epoch": 9.063720703125e-06,
      "model_forward_time": 1.729952335357666,
      "step": 9504
    },
    {
      "epoch": 9.063720703125e-06,
      "step": 9504,
      "training_step_time": 7.389595985412598
    },
    {
      "epoch": 9.064674377441407e-06,
      "model_forward_time": 1.720048189163208,
      "step": 9505
    },
    {
      "epoch": 9.064674377441407e-06,
      "step": 9505,
      "training_step_time": 8.81131362915039
    },
    {
      "epoch": 9.065628051757813e-06,
      "model_forward_time": 1.7208333015441895,
      "step": 9506
    },
    {
      "epoch": 9.065628051757813e-06,
      "step": 9506,
      "training_step_time": 7.5680437088012695
    },
    {
      "epoch": 9.066581726074218e-06,
      "model_forward_time": 2.2537026405334473,
      "step": 9507
    },
    {
      "epoch": 9.066581726074218e-06,
      "step": 9507,
      "training_step_time": 8.150603771209717
    },
    {
      "epoch": 9.067535400390624e-06,
      "model_forward_time": 1.7305760383605957,
      "step": 9508
    },
    {
      "epoch": 9.067535400390624e-06,
      "step": 9508,
      "training_step_time": 8.877346754074097
    },
    {
      "epoch": 9.068489074707032e-06,
      "model_forward_time": 1.7121777534484863,
      "step": 9509
    },
    {
      "epoch": 9.068489074707032e-06,
      "step": 9509,
      "training_step_time": 8.512391328811646
    },
    {
      "epoch": 9.069442749023437e-06,
      "grad_norm": 0.1968795657157898,
      "learning_rate": 9.944518499487255e-06,
      "loss": 0.0733,
      "step": 9510
    },
    {
      "epoch": 9.069442749023437e-06,
      "model_forward_time": 1.714115858078003,
      "step": 9510
    },
    {
      "dynamics_loss_avg": 0.05209437422454357,
      "epoch": 9.069442749023437e-06,
      "step": 9510
    },
    {
      "action_loss_avg": 0.0034951514389831573,
      "epoch": 9.069442749023437e-06,
      "step": 9510
    },
    {
      "epoch": 9.069442749023437e-06,
      "step": 9510,
      "training_step_time": 7.667517423629761
    },
    {
      "epoch": 9.070396423339843e-06,
      "model_forward_time": 1.7106335163116455,
      "step": 9511
    },
    {
      "epoch": 9.070396423339843e-06,
      "step": 9511,
      "training_step_time": 7.638895511627197
    },
    {
      "epoch": 9.07135009765625e-06,
      "model_forward_time": 1.709965705871582,
      "step": 9512
    },
    {
      "epoch": 9.07135009765625e-06,
      "step": 9512,
      "training_step_time": 7.602905511856079
    },
    {
      "epoch": 9.072303771972656e-06,
      "model_forward_time": 1.717010259628296,
      "step": 9513
    },
    {
      "epoch": 9.072303771972656e-06,
      "step": 9513,
      "training_step_time": 7.406413316726685
    },
    {
      "epoch": 9.073257446289062e-06,
      "model_forward_time": 1.712763786315918,
      "step": 9514
    },
    {
      "epoch": 9.073257446289062e-06,
      "step": 9514,
      "training_step_time": 7.679795026779175
    },
    {
      "epoch": 9.07421112060547e-06,
      "model_forward_time": 1.709397792816162,
      "step": 9515
    },
    {
      "epoch": 9.07421112060547e-06,
      "step": 9515,
      "training_step_time": 7.719151735305786
    },
    {
      "epoch": 9.075164794921875e-06,
      "model_forward_time": 2.272320032119751,
      "step": 9516
    },
    {
      "epoch": 9.075164794921875e-06,
      "step": 9516,
      "training_step_time": 9.474588394165039
    },
    {
      "epoch": 9.07611846923828e-06,
      "model_forward_time": 1.7166712284088135,
      "step": 9517
    },
    {
      "epoch": 9.07611846923828e-06,
      "step": 9517,
      "training_step_time": 9.703918218612671
    },
    {
      "epoch": 9.077072143554688e-06,
      "model_forward_time": 1.7208096981048584,
      "step": 9518
    },
    {
      "epoch": 9.077072143554688e-06,
      "step": 9518,
      "training_step_time": 7.69268012046814
    },
    {
      "epoch": 9.078025817871094e-06,
      "model_forward_time": 1.717454195022583,
      "step": 9519
    },
    {
      "epoch": 9.078025817871094e-06,
      "step": 9519,
      "training_step_time": 8.008508443832397
    },
    {
      "epoch": 9.0789794921875e-06,
      "grad_norm": 0.26404696702957153,
      "learning_rate": 9.944272592999151e-06,
      "loss": 0.0732,
      "step": 9520
    },
    {
      "epoch": 9.0789794921875e-06,
      "model_forward_time": 1.7178623676300049,
      "step": 9520
    },
    {
      "dynamics_loss_avg": 0.05689262794330716,
      "epoch": 9.0789794921875e-06,
      "step": 9520
    },
    {
      "action_loss_avg": 0.004613736062310636,
      "epoch": 9.0789794921875e-06,
      "step": 9520
    },
    {
      "epoch": 9.0789794921875e-06,
      "step": 9520,
      "training_step_time": 7.665642976760864
    },
    {
      "epoch": 9.079933166503907e-06,
      "model_forward_time": 1.7139251232147217,
      "step": 9521
    },
    {
      "epoch": 9.079933166503907e-06,
      "step": 9521,
      "training_step_time": 7.509707450866699
    },
    {
      "epoch": 9.080886840820313e-06,
      "model_forward_time": 1.7163796424865723,
      "step": 9522
    },
    {
      "epoch": 9.080886840820313e-06,
      "step": 9522,
      "training_step_time": 7.501817941665649
    },
    {
      "epoch": 9.081840515136718e-06,
      "model_forward_time": 1.7188138961791992,
      "step": 9523
    },
    {
      "epoch": 9.081840515136718e-06,
      "step": 9523,
      "training_step_time": 7.63441801071167
    },
    {
      "epoch": 9.082794189453124e-06,
      "model_forward_time": 1.717926025390625,
      "step": 9524
    },
    {
      "epoch": 9.082794189453124e-06,
      "step": 9524,
      "training_step_time": 8.723593711853027
    },
    {
      "epoch": 9.083747863769532e-06,
      "model_forward_time": 2.2793946266174316,
      "step": 9525
    },
    {
      "epoch": 9.083747863769532e-06,
      "step": 9525,
      "training_step_time": 8.309612274169922
    },
    {
      "epoch": 9.084701538085937e-06,
      "model_forward_time": 1.7171258926391602,
      "step": 9526
    },
    {
      "epoch": 9.084701538085937e-06,
      "step": 9526,
      "training_step_time": 7.7578442096710205
    },
    {
      "epoch": 9.085655212402343e-06,
      "model_forward_time": 1.7597570419311523,
      "step": 9527
    },
    {
      "epoch": 9.085655212402343e-06,
      "step": 9527,
      "training_step_time": 8.370144605636597
    },
    {
      "epoch": 9.08660888671875e-06,
      "model_forward_time": 1.726395845413208,
      "step": 9528
    },
    {
      "epoch": 9.08660888671875e-06,
      "step": 9528,
      "training_step_time": 8.67310643196106
    },
    {
      "epoch": 9.087562561035156e-06,
      "model_forward_time": 1.71455717086792,
      "step": 9529
    },
    {
      "epoch": 9.087562561035156e-06,
      "step": 9529,
      "training_step_time": 7.628723621368408
    },
    {
      "epoch": 9.088516235351562e-06,
      "grad_norm": 0.18802709877490997,
      "learning_rate": 9.94402614581283e-06,
      "loss": 0.0707,
      "step": 9530
    },
    {
      "epoch": 9.088516235351562e-06,
      "model_forward_time": 1.7109851837158203,
      "step": 9530
    },
    {
      "dynamics_loss_avg": 0.0680518651381135,
      "epoch": 9.088516235351562e-06,
      "step": 9530
    },
    {
      "action_loss_avg": 0.007513690646737814,
      "epoch": 9.088516235351562e-06,
      "step": 9530
    },
    {
      "epoch": 9.088516235351562e-06,
      "step": 9530,
      "training_step_time": 7.6983959674835205
    },
    {
      "epoch": 9.08946990966797e-06,
      "model_forward_time": 1.7093982696533203,
      "step": 9531
    },
    {
      "epoch": 9.08946990966797e-06,
      "step": 9531,
      "training_step_time": 7.6981353759765625
    },
    {
      "epoch": 9.090423583984375e-06,
      "model_forward_time": 1.7183585166931152,
      "step": 9532
    },
    {
      "epoch": 9.090423583984375e-06,
      "step": 9532,
      "training_step_time": 7.681023836135864
    },
    {
      "epoch": 9.09137725830078e-06,
      "model_forward_time": 1.7144451141357422,
      "step": 9533
    },
    {
      "epoch": 9.09137725830078e-06,
      "step": 9533,
      "training_step_time": 7.627444744110107
    },
    {
      "epoch": 9.092330932617188e-06,
      "model_forward_time": 2.269589424133301,
      "step": 9534
    },
    {
      "epoch": 9.092330932617188e-06,
      "step": 9534,
      "training_step_time": 10.260049819946289
    },
    {
      "epoch": 9.093284606933594e-06,
      "model_forward_time": 1.724785566329956,
      "step": 9535
    },
    {
      "epoch": 9.093284606933594e-06,
      "step": 9535,
      "training_step_time": 7.7230141162872314
    },
    {
      "epoch": 9.09423828125e-06,
      "model_forward_time": 1.7687301635742188,
      "step": 9536
    },
    {
      "epoch": 9.09423828125e-06,
      "step": 9536,
      "training_step_time": 8.758131980895996
    },
    {
      "epoch": 9.095191955566407e-06,
      "model_forward_time": 1.7185347080230713,
      "step": 9537
    },
    {
      "epoch": 9.095191955566407e-06,
      "step": 9537,
      "training_step_time": 7.574001312255859
    },
    {
      "epoch": 9.096145629882813e-06,
      "model_forward_time": 1.7312345504760742,
      "step": 9538
    },
    {
      "epoch": 9.096145629882813e-06,
      "step": 9538,
      "training_step_time": 7.819855690002441
    },
    {
      "epoch": 9.097099304199218e-06,
      "model_forward_time": 1.3360786437988281,
      "step": 9539
    },
    {
      "epoch": 9.097099304199218e-06,
      "step": 9539,
      "training_step_time": 7.668820858001709
    },
    {
      "epoch": 9.098052978515624e-06,
      "grad_norm": 0.28880560398101807,
      "learning_rate": 9.943779157955245e-06,
      "loss": 0.0761,
      "step": 9540
    },
    {
      "epoch": 9.098052978515624e-06,
      "model_forward_time": 1.716897964477539,
      "step": 9540
    },
    {
      "dynamics_loss_avg": 0.055238889344036576,
      "epoch": 9.098052978515624e-06,
      "step": 9540
    },
    {
      "action_loss_avg": 0.004739632143173367,
      "epoch": 9.098052978515624e-06,
      "step": 9540
    },
    {
      "epoch": 9.098052978515624e-06,
      "step": 9540,
      "training_step_time": 7.606328964233398
    },
    {
      "epoch": 9.099006652832032e-06,
      "model_forward_time": 1.7246291637420654,
      "step": 9541
    },
    {
      "epoch": 9.099006652832032e-06,
      "step": 9541,
      "training_step_time": 8.335910558700562
    },
    {
      "epoch": 9.099960327148437e-06,
      "model_forward_time": 1.7144498825073242,
      "step": 9542
    },
    {
      "epoch": 9.099960327148437e-06,
      "step": 9542,
      "training_step_time": 7.623426914215088
    },
    {
      "epoch": 9.100914001464843e-06,
      "model_forward_time": 1.7180659770965576,
      "step": 9543
    },
    {
      "epoch": 9.100914001464843e-06,
      "step": 9543,
      "training_step_time": 7.567939758300781
    },
    {
      "epoch": 9.10186767578125e-06,
      "model_forward_time": 2.260620355606079,
      "step": 9544
    },
    {
      "epoch": 9.10186767578125e-06,
      "step": 9544,
      "training_step_time": 7.709429740905762
    },
    {
      "epoch": 9.102821350097656e-06,
      "model_forward_time": 1.7744407653808594,
      "step": 9545
    },
    {
      "epoch": 9.102821350097656e-06,
      "step": 9545,
      "training_step_time": 7.631672620773315
    },
    {
      "epoch": 9.103775024414062e-06,
      "model_forward_time": 1.341970682144165,
      "step": 9546
    },
    {
      "epoch": 9.103775024414062e-06,
      "step": 9546,
      "training_step_time": 8.2739896774292
    },
    {
      "epoch": 9.10472869873047e-06,
      "model_forward_time": 1.715043544769287,
      "step": 9547
    },
    {
      "epoch": 9.10472869873047e-06,
      "step": 9547,
      "training_step_time": 7.873436689376831
    },
    {
      "epoch": 9.105682373046875e-06,
      "model_forward_time": 1.7175490856170654,
      "step": 9548
    },
    {
      "epoch": 9.105682373046875e-06,
      "step": 9548,
      "training_step_time": 7.680503845214844
    },
    {
      "epoch": 9.10663604736328e-06,
      "model_forward_time": 1.7292397022247314,
      "step": 9549
    },
    {
      "epoch": 9.10663604736328e-06,
      "step": 9549,
      "training_step_time": 7.578083753585815
    },
    {
      "epoch": 9.107589721679688e-06,
      "grad_norm": 0.20746980607509613,
      "learning_rate": 9.943531629453404e-06,
      "loss": 0.0738,
      "step": 9550
    },
    {
      "epoch": 9.107589721679688e-06,
      "model_forward_time": 1.715468406677246,
      "step": 9550
    },
    {
      "dynamics_loss_avg": 0.07151745576411486,
      "epoch": 9.107589721679688e-06,
      "step": 9550
    },
    {
      "action_loss_avg": 0.006567048246506601,
      "epoch": 9.107589721679688e-06,
      "step": 9550
    },
    {
      "epoch": 9.107589721679688e-06,
      "step": 9550,
      "training_step_time": 7.559905052185059
    },
    {
      "epoch": 9.108543395996094e-06,
      "model_forward_time": 1.7122812271118164,
      "step": 9551
    },
    {
      "epoch": 9.108543395996094e-06,
      "step": 9551,
      "training_step_time": 8.504053592681885
    },
    {
      "epoch": 9.1094970703125e-06,
      "model_forward_time": 1.7231760025024414,
      "step": 9552
    },
    {
      "epoch": 9.1094970703125e-06,
      "step": 9552,
      "training_step_time": 7.508932828903198
    },
    {
      "epoch": 9.110450744628907e-06,
      "model_forward_time": 2.2556488513946533,
      "step": 9553
    },
    {
      "epoch": 9.110450744628907e-06,
      "step": 9553,
      "training_step_time": 7.553839206695557
    },
    {
      "epoch": 9.111404418945313e-06,
      "model_forward_time": 1.7615926265716553,
      "step": 9554
    },
    {
      "epoch": 9.111404418945313e-06,
      "step": 9554,
      "training_step_time": 8.557386875152588
    },
    {
      "epoch": 9.112358093261718e-06,
      "model_forward_time": 1.7121148109436035,
      "step": 9555
    },
    {
      "epoch": 9.112358093261718e-06,
      "step": 9555,
      "training_step_time": 7.440953254699707
    },
    {
      "epoch": 9.113311767578124e-06,
      "model_forward_time": 1.71559476852417,
      "step": 9556
    },
    {
      "epoch": 9.113311767578124e-06,
      "step": 9556,
      "training_step_time": 7.763221740722656
    },
    {
      "epoch": 9.114265441894532e-06,
      "model_forward_time": 1.7517311573028564,
      "step": 9557
    },
    {
      "epoch": 9.114265441894532e-06,
      "step": 9557,
      "training_step_time": 8.392285346984863
    },
    {
      "epoch": 9.115219116210937e-06,
      "model_forward_time": 1.6981439590454102,
      "step": 9558
    },
    {
      "epoch": 9.115219116210937e-06,
      "step": 9558,
      "training_step_time": 7.469380617141724
    },
    {
      "epoch": 9.116172790527343e-06,
      "model_forward_time": 1.6987199783325195,
      "step": 9559
    },
    {
      "epoch": 9.116172790527343e-06,
      "step": 9559,
      "training_step_time": 7.445251226425171
    },
    {
      "epoch": 9.11712646484375e-06,
      "grad_norm": 0.18566879630088806,
      "learning_rate": 9.943283560334375e-06,
      "loss": 0.0682,
      "step": 9560
    },
    {
      "epoch": 9.11712646484375e-06,
      "model_forward_time": 1.6984736919403076,
      "step": 9560
    },
    {
      "dynamics_loss_avg": 0.0844688966870308,
      "epoch": 9.11712646484375e-06,
      "step": 9560
    },
    {
      "action_loss_avg": 0.009366122860228643,
      "epoch": 9.11712646484375e-06,
      "step": 9560
    },
    {
      "epoch": 9.11712646484375e-06,
      "step": 9560,
      "training_step_time": 7.485344409942627
    },
    {
      "epoch": 9.118080139160156e-06,
      "model_forward_time": 1.6986818313598633,
      "step": 9561
    },
    {
      "epoch": 9.118080139160156e-06,
      "step": 9561,
      "training_step_time": 8.257568836212158
    },
    {
      "epoch": 9.119033813476562e-06,
      "model_forward_time": 2.2587976455688477,
      "step": 9562
    },
    {
      "epoch": 9.119033813476562e-06,
      "step": 9562,
      "training_step_time": 8.568023681640625
    },
    {
      "epoch": 9.11998748779297e-06,
      "model_forward_time": 1.7166016101837158,
      "step": 9563
    },
    {
      "epoch": 9.11998748779297e-06,
      "step": 9563,
      "training_step_time": 9.557793617248535
    },
    {
      "epoch": 9.120941162109375e-06,
      "model_forward_time": 1.6980397701263428,
      "step": 9564
    },
    {
      "epoch": 9.120941162109375e-06,
      "step": 9564,
      "training_step_time": 7.614910125732422
    },
    {
      "epoch": 9.12189483642578e-06,
      "model_forward_time": 1.704498529434204,
      "step": 9565
    },
    {
      "epoch": 9.12189483642578e-06,
      "step": 9565,
      "training_step_time": 8.395739555358887
    },
    {
      "epoch": 9.122848510742188e-06,
      "model_forward_time": 1.6987133026123047,
      "step": 9566
    },
    {
      "epoch": 9.122848510742188e-06,
      "step": 9566,
      "training_step_time": 8.510463953018188
    },
    {
      "epoch": 9.123802185058594e-06,
      "model_forward_time": 1.703291416168213,
      "step": 9567
    },
    {
      "epoch": 9.123802185058594e-06,
      "step": 9567,
      "training_step_time": 7.49901556968689
    },
    {
      "epoch": 9.124755859375e-06,
      "model_forward_time": 1.698441505432129,
      "step": 9568
    },
    {
      "epoch": 9.124755859375e-06,
      "step": 9568,
      "training_step_time": 7.5717551708221436
    },
    {
      "epoch": 9.125709533691407e-06,
      "model_forward_time": 1.7056984901428223,
      "step": 9569
    },
    {
      "epoch": 9.125709533691407e-06,
      "step": 9569,
      "training_step_time": 7.667397737503052
    },
    {
      "epoch": 9.126663208007813e-06,
      "grad_norm": 0.18662388622760773,
      "learning_rate": 9.94303495062529e-06,
      "loss": 0.0684,
      "step": 9570
    },
    {
      "epoch": 9.126663208007813e-06,
      "model_forward_time": 1.6941967010498047,
      "step": 9570
    },
    {
      "dynamics_loss_avg": 0.07310593333095312,
      "epoch": 9.126663208007813e-06,
      "step": 9570
    },
    {
      "action_loss_avg": 0.0046937704551965,
      "epoch": 9.126663208007813e-06,
      "step": 9570
    },
    {
      "epoch": 9.126663208007813e-06,
      "step": 9570,
      "training_step_time": 8.40610647201538
    },
    {
      "epoch": 9.127616882324218e-06,
      "model_forward_time": 2.2483863830566406,
      "step": 9571
    },
    {
      "epoch": 9.127616882324218e-06,
      "step": 9571,
      "training_step_time": 8.528930187225342
    },
    {
      "epoch": 9.128570556640624e-06,
      "model_forward_time": 1.7044777870178223,
      "step": 9572
    },
    {
      "epoch": 9.128570556640624e-06,
      "step": 9572,
      "training_step_time": 8.565171003341675
    },
    {
      "epoch": 9.129524230957032e-06,
      "model_forward_time": 1.7200226783752441,
      "step": 9573
    },
    {
      "epoch": 9.129524230957032e-06,
      "step": 9573,
      "training_step_time": 7.564671993255615
    },
    {
      "epoch": 9.130477905273437e-06,
      "model_forward_time": 1.700329065322876,
      "step": 9574
    },
    {
      "epoch": 9.130477905273437e-06,
      "step": 9574,
      "training_step_time": 7.626543760299683
    },
    {
      "epoch": 9.131431579589843e-06,
      "model_forward_time": 1.698122262954712,
      "step": 9575
    },
    {
      "epoch": 9.131431579589843e-06,
      "step": 9575,
      "training_step_time": 7.887294054031372
    },
    {
      "epoch": 9.13238525390625e-06,
      "model_forward_time": 1.697770118713379,
      "step": 9576
    },
    {
      "epoch": 9.13238525390625e-06,
      "step": 9576,
      "training_step_time": 7.580177307128906
    },
    {
      "epoch": 9.133338928222656e-06,
      "model_forward_time": 1.6958065032958984,
      "step": 9577
    },
    {
      "epoch": 9.133338928222656e-06,
      "step": 9577,
      "training_step_time": 7.638494968414307
    },
    {
      "epoch": 9.134292602539062e-06,
      "model_forward_time": 1.6963863372802734,
      "step": 9578
    },
    {
      "epoch": 9.134292602539062e-06,
      "step": 9578,
      "training_step_time": 7.643590450286865
    },
    {
      "epoch": 9.13524627685547e-06,
      "model_forward_time": 1.702915906906128,
      "step": 9579
    },
    {
      "epoch": 9.13524627685547e-06,
      "step": 9579,
      "training_step_time": 8.613696575164795
    },
    {
      "epoch": 9.136199951171875e-06,
      "grad_norm": 0.2030915468931198,
      "learning_rate": 9.942785800353332e-06,
      "loss": 0.0628,
      "step": 9580
    },
    {
      "epoch": 9.136199951171875e-06,
      "model_forward_time": 1.6920030117034912,
      "step": 9580
    },
    {
      "dynamics_loss_avg": 0.07255449257791043,
      "epoch": 9.136199951171875e-06,
      "step": 9580
    },
    {
      "action_loss_avg": 0.004214541241526604,
      "epoch": 9.136199951171875e-06,
      "step": 9580
    },
    {
      "epoch": 9.136199951171875e-06,
      "step": 9580,
      "training_step_time": 7.978898048400879
    },
    {
      "epoch": 9.13715362548828e-06,
      "model_forward_time": 1.7085583209991455,
      "step": 9581
    },
    {
      "epoch": 9.13715362548828e-06,
      "step": 9581,
      "training_step_time": 8.24735975265503
    },
    {
      "epoch": 9.138107299804688e-06,
      "model_forward_time": 1.7094223499298096,
      "step": 9582
    },
    {
      "epoch": 9.138107299804688e-06,
      "step": 9582,
      "training_step_time": 7.5903589725494385
    },
    {
      "epoch": 9.139060974121094e-06,
      "model_forward_time": 1.6817266941070557,
      "step": 9583
    },
    {
      "epoch": 9.139060974121094e-06,
      "step": 9583,
      "training_step_time": 7.493567228317261
    },
    {
      "epoch": 9.1400146484375e-06,
      "model_forward_time": 1.318368911743164,
      "step": 9584
    },
    {
      "epoch": 9.1400146484375e-06,
      "step": 9584,
      "training_step_time": 7.448453187942505
    },
    {
      "epoch": 9.140968322753907e-06,
      "model_forward_time": 1.6915297508239746,
      "step": 9585
    },
    {
      "epoch": 9.140968322753907e-06,
      "step": 9585,
      "training_step_time": 7.813042879104614
    },
    {
      "epoch": 9.141921997070313e-06,
      "model_forward_time": 1.6942596435546875,
      "step": 9586
    },
    {
      "epoch": 9.141921997070313e-06,
      "step": 9586,
      "training_step_time": 7.4928107261657715
    },
    {
      "epoch": 9.142875671386719e-06,
      "model_forward_time": 1.6944236755371094,
      "step": 9587
    },
    {
      "epoch": 9.142875671386719e-06,
      "step": 9587,
      "training_step_time": 7.668310642242432
    },
    {
      "epoch": 9.143829345703124e-06,
      "model_forward_time": 1.690558910369873,
      "step": 9588
    },
    {
      "epoch": 9.143829345703124e-06,
      "step": 9588,
      "training_step_time": 8.085699558258057
    },
    {
      "epoch": 9.144783020019532e-06,
      "model_forward_time": 1.6954114437103271,
      "step": 9589
    },
    {
      "epoch": 9.144783020019532e-06,
      "step": 9589,
      "training_step_time": 8.230319261550903
    },
    {
      "epoch": 9.145736694335937e-06,
      "grad_norm": 0.3182251751422882,
      "learning_rate": 9.942536109545751e-06,
      "loss": 0.0663,
      "step": 9590
    },
    {
      "epoch": 9.145736694335937e-06,
      "model_forward_time": 2.258526086807251,
      "step": 9590
    },
    {
      "dynamics_loss_avg": 0.05056168250739575,
      "epoch": 9.145736694335937e-06,
      "step": 9590
    },
    {
      "action_loss_avg": 0.0026301116333343087,
      "epoch": 9.145736694335937e-06,
      "step": 9590
    },
    {
      "epoch": 9.145736694335937e-06,
      "step": 9590,
      "training_step_time": 8.216912984848022
    },
    {
      "epoch": 9.146690368652343e-06,
      "model_forward_time": 1.769791841506958,
      "step": 9591
    },
    {
      "epoch": 9.146690368652343e-06,
      "step": 9591,
      "training_step_time": 7.635601282119751
    },
    {
      "epoch": 9.14764404296875e-06,
      "model_forward_time": 1.717109203338623,
      "step": 9592
    },
    {
      "epoch": 9.14764404296875e-06,
      "step": 9592,
      "training_step_time": 8.632351398468018
    },
    {
      "epoch": 9.148597717285156e-06,
      "model_forward_time": 1.7090437412261963,
      "step": 9593
    },
    {
      "epoch": 9.148597717285156e-06,
      "step": 9593,
      "training_step_time": 7.388237476348877
    },
    {
      "epoch": 9.149551391601562e-06,
      "model_forward_time": 1.7143762111663818,
      "step": 9594
    },
    {
      "epoch": 9.149551391601562e-06,
      "step": 9594,
      "training_step_time": 7.813177824020386
    },
    {
      "epoch": 9.15050506591797e-06,
      "model_forward_time": 1.711423635482788,
      "step": 9595
    },
    {
      "epoch": 9.15050506591797e-06,
      "step": 9595,
      "training_step_time": 7.483545303344727
    },
    {
      "epoch": 9.151458740234375e-06,
      "model_forward_time": 1.7032415866851807,
      "step": 9596
    },
    {
      "epoch": 9.151458740234375e-06,
      "step": 9596,
      "training_step_time": 8.033438682556152
    },
    {
      "epoch": 9.15241241455078e-06,
      "model_forward_time": 1.7133891582489014,
      "step": 9597
    },
    {
      "epoch": 9.15241241455078e-06,
      "step": 9597,
      "training_step_time": 8.053717136383057
    },
    {
      "epoch": 9.153366088867188e-06,
      "model_forward_time": 1.7314624786376953,
      "step": 9598
    },
    {
      "epoch": 9.153366088867188e-06,
      "step": 9598,
      "training_step_time": 8.120513916015625
    },
    {
      "epoch": 9.154319763183594e-06,
      "model_forward_time": 2.249319314956665,
      "step": 9599
    },
    {
      "epoch": 9.154319763183594e-06,
      "step": 9599,
      "training_step_time": 8.66522479057312
    },
    {
      "epoch": 9.1552734375e-06,
      "grad_norm": 0.2510678768157959,
      "learning_rate": 9.942285878229852e-06,
      "loss": 0.0666,
      "step": 9600
    },
    {
      "epoch": 9.1552734375e-06,
      "model_forward_time": 1.767322301864624,
      "step": 9600
    },
    {
      "dynamics_loss_avg": 0.04814170924946666,
      "epoch": 9.1552734375e-06,
      "step": 9600
    },
    {
      "action_loss_avg": 0.003197760018520057,
      "epoch": 9.1552734375e-06,
      "step": 9600
    },
    {
      "epoch": 9.1552734375e-06,
      "step": 9600,
      "training_step_time": 7.656757354736328
    },
    {
      "epoch": 9.156227111816407e-06,
      "model_forward_time": 1.6869313716888428,
      "step": 9601
    },
    {
      "epoch": 9.156227111816407e-06,
      "step": 9601,
      "training_step_time": 8.357054471969604
    },
    {
      "epoch": 9.157180786132813e-06,
      "model_forward_time": 1.7124519348144531,
      "step": 9602
    },
    {
      "epoch": 9.157180786132813e-06,
      "step": 9602,
      "training_step_time": 8.589980840682983
    },
    {
      "epoch": 9.158134460449219e-06,
      "model_forward_time": 1.717731237411499,
      "step": 9603
    },
    {
      "epoch": 9.158134460449219e-06,
      "step": 9603,
      "training_step_time": 7.892849922180176
    },
    {
      "epoch": 9.159088134765624e-06,
      "model_forward_time": 1.7053768634796143,
      "step": 9604
    },
    {
      "epoch": 9.159088134765624e-06,
      "step": 9604,
      "training_step_time": 7.517739295959473
    },
    {
      "epoch": 9.160041809082032e-06,
      "model_forward_time": 1.704150915145874,
      "step": 9605
    },
    {
      "epoch": 9.160041809082032e-06,
      "step": 9605,
      "training_step_time": 8.377135753631592
    },
    {
      "epoch": 9.160995483398437e-06,
      "model_forward_time": 1.3356618881225586,
      "step": 9606
    },
    {
      "epoch": 9.160995483398437e-06,
      "step": 9606,
      "training_step_time": 7.57539439201355
    },
    {
      "epoch": 9.161949157714843e-06,
      "model_forward_time": 1.3323051929473877,
      "step": 9607
    },
    {
      "epoch": 9.161949157714843e-06,
      "step": 9607,
      "training_step_time": 8.61662769317627
    },
    {
      "epoch": 9.16290283203125e-06,
      "model_forward_time": 2.254547357559204,
      "step": 9608
    },
    {
      "epoch": 9.16290283203125e-06,
      "step": 9608,
      "training_step_time": 8.336368322372437
    },
    {
      "epoch": 9.163856506347656e-06,
      "model_forward_time": 1.7640390396118164,
      "step": 9609
    },
    {
      "epoch": 9.163856506347656e-06,
      "step": 9609,
      "training_step_time": 8.277031183242798
    },
    {
      "epoch": 9.164810180664062e-06,
      "grad_norm": 0.1692071408033371,
      "learning_rate": 9.942035106433001e-06,
      "loss": 0.0661,
      "step": 9610
    },
    {
      "epoch": 9.164810180664062e-06,
      "model_forward_time": 1.7159717082977295,
      "step": 9610
    },
    {
      "dynamics_loss_avg": 0.055477718450129034,
      "epoch": 9.164810180664062e-06,
      "step": 9610
    },
    {
      "action_loss_avg": 0.00367762828245759,
      "epoch": 9.164810180664062e-06,
      "step": 9610
    },
    {
      "epoch": 9.164810180664062e-06,
      "step": 9610,
      "training_step_time": 7.628623962402344
    },
    {
      "epoch": 9.16576385498047e-06,
      "model_forward_time": 1.3331153392791748,
      "step": 9611
    },
    {
      "epoch": 9.16576385498047e-06,
      "step": 9611,
      "training_step_time": 9.669126987457275
    },
    {
      "epoch": 9.166717529296875e-06,
      "model_forward_time": 1.7002019882202148,
      "step": 9612
    },
    {
      "epoch": 9.166717529296875e-06,
      "step": 9612,
      "training_step_time": 8.606126308441162
    },
    {
      "epoch": 9.16767120361328e-06,
      "model_forward_time": 1.70758056640625,
      "step": 9613
    },
    {
      "epoch": 9.16767120361328e-06,
      "step": 9613,
      "training_step_time": 7.88272762298584
    },
    {
      "epoch": 9.168624877929688e-06,
      "model_forward_time": 1.7065496444702148,
      "step": 9614
    },
    {
      "epoch": 9.168624877929688e-06,
      "step": 9614,
      "training_step_time": 8.358774185180664
    },
    {
      "epoch": 9.169578552246094e-06,
      "model_forward_time": 1.7116076946258545,
      "step": 9615
    },
    {
      "epoch": 9.169578552246094e-06,
      "step": 9615,
      "training_step_time": 8.152722358703613
    },
    {
      "epoch": 9.1705322265625e-06,
      "model_forward_time": 1.7095491886138916,
      "step": 9616
    },
    {
      "epoch": 9.1705322265625e-06,
      "step": 9616,
      "training_step_time": 8.64319133758545
    },
    {
      "epoch": 9.171485900878907e-06,
      "model_forward_time": 1.7132289409637451,
      "step": 9617
    },
    {
      "epoch": 9.171485900878907e-06,
      "step": 9617,
      "training_step_time": 7.591244220733643
    },
    {
      "epoch": 9.172439575195313e-06,
      "model_forward_time": 2.301420211791992,
      "step": 9618
    },
    {
      "epoch": 9.172439575195313e-06,
      "step": 9618,
      "training_step_time": 7.624792814254761
    },
    {
      "epoch": 9.173393249511719e-06,
      "model_forward_time": 1.7061779499053955,
      "step": 9619
    },
    {
      "epoch": 9.173393249511719e-06,
      "step": 9619,
      "training_step_time": 7.660294532775879
    },
    {
      "epoch": 9.174346923828124e-06,
      "grad_norm": 0.2558600604534149,
      "learning_rate": 9.94178379418262e-06,
      "loss": 0.066,
      "step": 9620
    },
    {
      "epoch": 9.174346923828124e-06,
      "model_forward_time": 1.7191169261932373,
      "step": 9620
    },
    {
      "dynamics_loss_avg": 0.054287468641996385,
      "epoch": 9.174346923828124e-06,
      "step": 9620
    },
    {
      "action_loss_avg": 0.006733266776427627,
      "epoch": 9.174346923828124e-06,
      "step": 9620
    },
    {
      "epoch": 9.174346923828124e-06,
      "step": 9620,
      "training_step_time": 8.063114404678345
    },
    {
      "epoch": 9.175300598144532e-06,
      "model_forward_time": 1.703526258468628,
      "step": 9621
    },
    {
      "epoch": 9.175300598144532e-06,
      "step": 9621,
      "training_step_time": 7.6791462898254395
    },
    {
      "epoch": 9.176254272460937e-06,
      "model_forward_time": 1.7095892429351807,
      "step": 9622
    },
    {
      "epoch": 9.176254272460937e-06,
      "step": 9622,
      "training_step_time": 7.871640682220459
    },
    {
      "epoch": 9.177207946777343e-06,
      "model_forward_time": 1.7113428115844727,
      "step": 9623
    },
    {
      "epoch": 9.177207946777343e-06,
      "step": 9623,
      "training_step_time": 7.7812581062316895
    },
    {
      "epoch": 9.17816162109375e-06,
      "model_forward_time": 1.7115881443023682,
      "step": 9624
    },
    {
      "epoch": 9.17816162109375e-06,
      "step": 9624,
      "training_step_time": 7.636343955993652
    },
    {
      "epoch": 9.179115295410156e-06,
      "model_forward_time": 1.7133760452270508,
      "step": 9625
    },
    {
      "epoch": 9.179115295410156e-06,
      "step": 9625,
      "training_step_time": 7.663568735122681
    },
    {
      "epoch": 9.180068969726562e-06,
      "model_forward_time": 1.7253227233886719,
      "step": 9626
    },
    {
      "epoch": 9.180068969726562e-06,
      "step": 9626,
      "training_step_time": 8.337302923202515
    },
    {
      "epoch": 9.18102264404297e-06,
      "model_forward_time": 2.287645101547241,
      "step": 9627
    },
    {
      "epoch": 9.18102264404297e-06,
      "step": 9627,
      "training_step_time": 8.508643388748169
    },
    {
      "epoch": 9.181976318359375e-06,
      "model_forward_time": 2.896556854248047,
      "step": 9628
    },
    {
      "epoch": 9.181976318359375e-06,
      "step": 9628,
      "training_step_time": 7.7509355545043945
    },
    {
      "epoch": 9.18292999267578e-06,
      "model_forward_time": 1.7408640384674072,
      "step": 9629
    },
    {
      "epoch": 9.18292999267578e-06,
      "step": 9629,
      "training_step_time": 7.698181629180908
    },
    {
      "epoch": 9.183883666992188e-06,
      "grad_norm": 0.13639315962791443,
      "learning_rate": 9.941531941506194e-06,
      "loss": 0.0668,
      "step": 9630
    },
    {
      "epoch": 9.183883666992188e-06,
      "model_forward_time": 1.7043802738189697,
      "step": 9630
    },
    {
      "dynamics_loss_avg": 0.07666763626039028,
      "epoch": 9.183883666992188e-06,
      "step": 9630
    },
    {
      "action_loss_avg": 0.00533227720297873,
      "epoch": 9.183883666992188e-06,
      "step": 9630
    },
    {
      "epoch": 9.183883666992188e-06,
      "step": 9630,
      "training_step_time": 6.825682640075684
    },
    {
      "epoch": 9.184837341308594e-06,
      "model_forward_time": 1.7039294242858887,
      "step": 9631
    },
    {
      "epoch": 9.184837341308594e-06,
      "step": 9631,
      "training_step_time": 6.786309003829956
    },
    {
      "epoch": 9.185791015625e-06,
      "model_forward_time": 1.7161390781402588,
      "step": 9632
    },
    {
      "epoch": 9.185791015625e-06,
      "step": 9632,
      "training_step_time": 6.799221992492676
    },
    {
      "epoch": 9.186744689941405e-06,
      "model_forward_time": 1.7131738662719727,
      "step": 9633
    },
    {
      "epoch": 9.186744689941405e-06,
      "step": 9633,
      "training_step_time": 6.811816215515137
    },
    {
      "epoch": 9.187698364257813e-06,
      "model_forward_time": 1.7074048519134521,
      "step": 9634
    },
    {
      "epoch": 9.187698364257813e-06,
      "step": 9634,
      "training_step_time": 6.81208872795105
    },
    {
      "epoch": 9.188652038574219e-06,
      "model_forward_time": 1.7056117057800293,
      "step": 9635
    },
    {
      "epoch": 9.188652038574219e-06,
      "step": 9635,
      "training_step_time": 6.804879665374756
    },
    {
      "epoch": 9.189605712890624e-06,
      "model_forward_time": 1.3216490745544434,
      "step": 9636
    },
    {
      "epoch": 9.189605712890624e-06,
      "step": 9636,
      "training_step_time": 6.805283069610596
    },
    {
      "epoch": 9.190559387207032e-06,
      "model_forward_time": 2.270836591720581,
      "step": 9637
    },
    {
      "epoch": 9.190559387207032e-06,
      "step": 9637,
      "training_step_time": 11.16575002670288
    },
    {
      "epoch": 9.191513061523437e-06,
      "model_forward_time": 1.6941680908203125,
      "step": 9638
    },
    {
      "epoch": 9.191513061523437e-06,
      "step": 9638,
      "training_step_time": 7.749854803085327
    },
    {
      "epoch": 9.192466735839843e-06,
      "model_forward_time": 1.7208218574523926,
      "step": 9639
    },
    {
      "epoch": 9.192466735839843e-06,
      "step": 9639,
      "training_step_time": 7.585078954696655
    },
    {
      "epoch": 9.19342041015625e-06,
      "grad_norm": 0.19189874827861786,
      "learning_rate": 9.941279548431263e-06,
      "loss": 0.0727,
      "step": 9640
    },
    {
      "epoch": 9.19342041015625e-06,
      "model_forward_time": 1.704272985458374,
      "step": 9640
    },
    {
      "dynamics_loss_avg": 0.05628975089639425,
      "epoch": 9.19342041015625e-06,
      "step": 9640
    },
    {
      "action_loss_avg": 0.0028475071187131105,
      "epoch": 9.19342041015625e-06,
      "step": 9640
    },
    {
      "epoch": 9.19342041015625e-06,
      "step": 9640,
      "training_step_time": 6.757155895233154
    },
    {
      "epoch": 9.194374084472656e-06,
      "model_forward_time": 1.7054359912872314,
      "step": 9641
    },
    {
      "epoch": 9.194374084472656e-06,
      "step": 9641,
      "training_step_time": 6.770852327346802
    },
    {
      "epoch": 9.195327758789062e-06,
      "model_forward_time": 1.70542573928833,
      "step": 9642
    },
    {
      "epoch": 9.195327758789062e-06,
      "step": 9642,
      "training_step_time": 6.813705205917358
    },
    {
      "epoch": 9.19628143310547e-06,
      "model_forward_time": 1.7046866416931152,
      "step": 9643
    },
    {
      "epoch": 9.19628143310547e-06,
      "step": 9643,
      "training_step_time": 6.842768907546997
    },
    {
      "epoch": 9.197235107421875e-06,
      "model_forward_time": 1.711238145828247,
      "step": 9644
    },
    {
      "epoch": 9.197235107421875e-06,
      "step": 9644,
      "training_step_time": 6.802692651748657
    },
    {
      "epoch": 9.19818878173828e-06,
      "model_forward_time": 1.7088346481323242,
      "step": 9645
    },
    {
      "epoch": 9.19818878173828e-06,
      "step": 9645,
      "training_step_time": 6.79985785484314
    },
    {
      "epoch": 9.199142456054688e-06,
      "model_forward_time": 2.2692456245422363,
      "step": 9646
    },
    {
      "epoch": 9.199142456054688e-06,
      "step": 9646,
      "training_step_time": 7.379420757293701
    },
    {
      "epoch": 9.200096130371094e-06,
      "model_forward_time": 1.7052133083343506,
      "step": 9647
    },
    {
      "epoch": 9.200096130371094e-06,
      "step": 9647,
      "training_step_time": 8.39113450050354
    },
    {
      "epoch": 9.2010498046875e-06,
      "model_forward_time": 1.7165429592132568,
      "step": 9648
    },
    {
      "epoch": 9.2010498046875e-06,
      "step": 9648,
      "training_step_time": 7.746016502380371
    },
    {
      "epoch": 9.202003479003905e-06,
      "model_forward_time": 1.7098472118377686,
      "step": 9649
    },
    {
      "epoch": 9.202003479003905e-06,
      "step": 9649,
      "training_step_time": 6.795184135437012
    },
    {
      "epoch": 9.202957153320313e-06,
      "grad_norm": 0.22764630615711212,
      "learning_rate": 9.941026614985432e-06,
      "loss": 0.0706,
      "step": 9650
    },
    {
      "epoch": 9.202957153320313e-06,
      "model_forward_time": 1.7045536041259766,
      "step": 9650
    },
    {
      "dynamics_loss_avg": 0.050627097859978674,
      "epoch": 9.202957153320313e-06,
      "step": 9650
    },
    {
      "action_loss_avg": 0.0032906622160226106,
      "epoch": 9.202957153320313e-06,
      "step": 9650
    },
    {
      "epoch": 9.202957153320313e-06,
      "step": 9650,
      "training_step_time": 6.7585296630859375
    },
    {
      "epoch": 9.203910827636719e-06,
      "model_forward_time": 1.7068500518798828,
      "step": 9651
    },
    {
      "epoch": 9.203910827636719e-06,
      "step": 9651,
      "training_step_time": 6.767191410064697
    },
    {
      "epoch": 9.204864501953124e-06,
      "model_forward_time": 1.7032811641693115,
      "step": 9652
    },
    {
      "epoch": 9.204864501953124e-06,
      "step": 9652,
      "training_step_time": 6.767387390136719
    },
    {
      "epoch": 9.205818176269532e-06,
      "model_forward_time": 1.7158641815185547,
      "step": 9653
    },
    {
      "epoch": 9.205818176269532e-06,
      "step": 9653,
      "training_step_time": 6.817754745483398
    },
    {
      "epoch": 9.206771850585937e-06,
      "model_forward_time": 1.720641851425171,
      "step": 9654
    },
    {
      "epoch": 9.206771850585937e-06,
      "step": 9654,
      "training_step_time": 6.699071645736694
    },
    {
      "epoch": 9.207725524902343e-06,
      "model_forward_time": 2.2931463718414307,
      "step": 9655
    },
    {
      "epoch": 9.207725524902343e-06,
      "step": 9655,
      "training_step_time": 7.388814687728882
    },
    {
      "epoch": 9.20867919921875e-06,
      "model_forward_time": 1.7000529766082764,
      "step": 9656
    },
    {
      "epoch": 9.20867919921875e-06,
      "step": 9656,
      "training_step_time": 9.141069412231445
    },
    {
      "epoch": 9.209632873535156e-06,
      "model_forward_time": 1.7094855308532715,
      "step": 9657
    },
    {
      "epoch": 9.209632873535156e-06,
      "step": 9657,
      "training_step_time": 7.857460975646973
    },
    {
      "epoch": 9.210586547851562e-06,
      "model_forward_time": 1.7037522792816162,
      "step": 9658
    },
    {
      "epoch": 9.210586547851562e-06,
      "step": 9658,
      "training_step_time": 6.805051565170288
    },
    {
      "epoch": 9.21154022216797e-06,
      "model_forward_time": 1.706108808517456,
      "step": 9659
    },
    {
      "epoch": 9.21154022216797e-06,
      "step": 9659,
      "training_step_time": 6.792123794555664
    },
    {
      "epoch": 9.212493896484375e-06,
      "grad_norm": 0.2134198546409607,
      "learning_rate": 9.940773141196356e-06,
      "loss": 0.0709,
      "step": 9660
    },
    {
      "epoch": 9.212493896484375e-06,
      "model_forward_time": 1.7153091430664062,
      "step": 9660
    },
    {
      "dynamics_loss_avg": 0.06931026130914689,
      "epoch": 9.212493896484375e-06,
      "step": 9660
    },
    {
      "action_loss_avg": 0.004495049698743969,
      "epoch": 9.212493896484375e-06,
      "step": 9660
    },
    {
      "epoch": 9.212493896484375e-06,
      "step": 9660,
      "training_step_time": 6.78987717628479
    },
    {
      "epoch": 9.213447570800781e-06,
      "model_forward_time": 1.7123241424560547,
      "step": 9661
    },
    {
      "epoch": 9.213447570800781e-06,
      "step": 9661,
      "training_step_time": 6.828995943069458
    },
    {
      "epoch": 9.214401245117188e-06,
      "model_forward_time": 1.7083969116210938,
      "step": 9662
    },
    {
      "epoch": 9.214401245117188e-06,
      "step": 9662,
      "training_step_time": 6.756500720977783
    },
    {
      "epoch": 9.215354919433594e-06,
      "model_forward_time": 1.7056300640106201,
      "step": 9663
    },
    {
      "epoch": 9.215354919433594e-06,
      "step": 9663,
      "training_step_time": 6.797327995300293
    },
    {
      "epoch": 9.21630859375e-06,
      "model_forward_time": 2.3000831604003906,
      "step": 9664
    },
    {
      "epoch": 9.21630859375e-06,
      "step": 9664,
      "training_step_time": 7.0396153926849365
    },
    {
      "epoch": 9.217262268066405e-06,
      "model_forward_time": 1.711113691329956,
      "step": 9665
    },
    {
      "epoch": 9.217262268066405e-06,
      "step": 9665,
      "training_step_time": 10.721641778945923
    },
    {
      "epoch": 9.218215942382813e-06,
      "model_forward_time": 1.705615758895874,
      "step": 9666
    },
    {
      "epoch": 9.218215942382813e-06,
      "step": 9666,
      "training_step_time": 7.847197532653809
    },
    {
      "epoch": 9.219169616699219e-06,
      "model_forward_time": 1.719191312789917,
      "step": 9667
    },
    {
      "epoch": 9.219169616699219e-06,
      "step": 9667,
      "training_step_time": 6.775583267211914
    },
    {
      "epoch": 9.220123291015624e-06,
      "model_forward_time": 1.7176551818847656,
      "step": 9668
    },
    {
      "epoch": 9.220123291015624e-06,
      "step": 9668,
      "training_step_time": 6.7553534507751465
    },
    {
      "epoch": 9.221076965332032e-06,
      "model_forward_time": 1.7096588611602783,
      "step": 9669
    },
    {
      "epoch": 9.221076965332032e-06,
      "step": 9669,
      "training_step_time": 6.851538419723511
    },
    {
      "epoch": 9.222030639648437e-06,
      "grad_norm": 0.18430915474891663,
      "learning_rate": 9.94051912709176e-06,
      "loss": 0.0682,
      "step": 9670
    },
    {
      "epoch": 9.222030639648437e-06,
      "model_forward_time": 1.706329107284546,
      "step": 9670
    },
    {
      "dynamics_loss_avg": 0.05555581096559763,
      "epoch": 9.222030639648437e-06,
      "step": 9670
    },
    {
      "action_loss_avg": 0.0034266722737811505,
      "epoch": 9.222030639648437e-06,
      "step": 9670
    },
    {
      "epoch": 9.222030639648437e-06,
      "step": 9670,
      "training_step_time": 6.741666078567505
    },
    {
      "epoch": 9.222984313964843e-06,
      "model_forward_time": 1.7066359519958496,
      "step": 9671
    },
    {
      "epoch": 9.222984313964843e-06,
      "step": 9671,
      "training_step_time": 6.787342309951782
    },
    {
      "epoch": 9.22393798828125e-06,
      "model_forward_time": 1.71773099899292,
      "step": 9672
    },
    {
      "epoch": 9.22393798828125e-06,
      "step": 9672,
      "training_step_time": 6.7992103099823
    },
    {
      "epoch": 9.224891662597656e-06,
      "model_forward_time": 2.249851703643799,
      "step": 9673
    },
    {
      "epoch": 9.224891662597656e-06,
      "step": 9673,
      "training_step_time": 8.00612473487854
    },
    {
      "epoch": 9.225845336914062e-06,
      "model_forward_time": 1.712035894393921,
      "step": 9674
    },
    {
      "epoch": 9.225845336914062e-06,
      "step": 9674,
      "training_step_time": 10.69887900352478
    },
    {
      "epoch": 9.22679901123047e-06,
      "model_forward_time": 1.702854871749878,
      "step": 9675
    },
    {
      "epoch": 9.22679901123047e-06,
      "step": 9675,
      "training_step_time": 7.861068487167358
    },
    {
      "epoch": 9.227752685546875e-06,
      "model_forward_time": 1.7081966400146484,
      "step": 9676
    },
    {
      "epoch": 9.227752685546875e-06,
      "step": 9676,
      "training_step_time": 6.844378471374512
    },
    {
      "epoch": 9.228706359863281e-06,
      "model_forward_time": 1.7072515487670898,
      "step": 9677
    },
    {
      "epoch": 9.228706359863281e-06,
      "step": 9677,
      "training_step_time": 6.771941900253296
    },
    {
      "epoch": 9.229660034179688e-06,
      "model_forward_time": 1.7156202793121338,
      "step": 9678
    },
    {
      "epoch": 9.229660034179688e-06,
      "step": 9678,
      "training_step_time": 6.76219916343689
    },
    {
      "epoch": 9.230613708496094e-06,
      "model_forward_time": 1.7130630016326904,
      "step": 9679
    },
    {
      "epoch": 9.230613708496094e-06,
      "step": 9679,
      "training_step_time": 6.74360466003418
    },
    {
      "epoch": 9.2315673828125e-06,
      "grad_norm": 0.23813456296920776,
      "learning_rate": 9.940264572699422e-06,
      "loss": 0.0673,
      "step": 9680
    },
    {
      "epoch": 9.2315673828125e-06,
      "model_forward_time": 1.7030088901519775,
      "step": 9680
    },
    {
      "dynamics_loss_avg": 0.0501217370852828,
      "epoch": 9.2315673828125e-06,
      "step": 9680
    },
    {
      "action_loss_avg": 0.0032311555347405373,
      "epoch": 9.2315673828125e-06,
      "step": 9680
    },
    {
      "epoch": 9.2315673828125e-06,
      "step": 9680,
      "training_step_time": 6.793216228485107
    },
    {
      "epoch": 9.232521057128905e-06,
      "model_forward_time": 1.713219165802002,
      "step": 9681
    },
    {
      "epoch": 9.232521057128905e-06,
      "step": 9681,
      "training_step_time": 6.784515142440796
    },
    {
      "epoch": 9.233474731445313e-06,
      "model_forward_time": 1.7039790153503418,
      "step": 9682
    },
    {
      "epoch": 9.233474731445313e-06,
      "step": 9682,
      "training_step_time": 7.504363059997559
    },
    {
      "epoch": 9.234428405761719e-06,
      "model_forward_time": 1.7587556838989258,
      "step": 9683
    },
    {
      "epoch": 9.234428405761719e-06,
      "step": 9683,
      "training_step_time": 9.382390260696411
    },
    {
      "epoch": 9.235382080078124e-06,
      "model_forward_time": 1.708167552947998,
      "step": 9684
    },
    {
      "epoch": 9.235382080078124e-06,
      "step": 9684,
      "training_step_time": 9.427511215209961
    },
    {
      "epoch": 9.236335754394532e-06,
      "model_forward_time": 1.7044081687927246,
      "step": 9685
    },
    {
      "epoch": 9.236335754394532e-06,
      "step": 9685,
      "training_step_time": 7.490406274795532
    },
    {
      "epoch": 9.237289428710937e-06,
      "model_forward_time": 1.7070016860961914,
      "step": 9686
    },
    {
      "epoch": 9.237289428710937e-06,
      "step": 9686,
      "training_step_time": 6.7242591381073
    },
    {
      "epoch": 9.238243103027343e-06,
      "model_forward_time": 1.7049975395202637,
      "step": 9687
    },
    {
      "epoch": 9.238243103027343e-06,
      "step": 9687,
      "training_step_time": 6.808129787445068
    },
    {
      "epoch": 9.23919677734375e-06,
      "model_forward_time": 1.7083518505096436,
      "step": 9688
    },
    {
      "epoch": 9.23919677734375e-06,
      "step": 9688,
      "training_step_time": 6.753134727478027
    },
    {
      "epoch": 9.240150451660156e-06,
      "model_forward_time": 1.7187199592590332,
      "step": 9689
    },
    {
      "epoch": 9.240150451660156e-06,
      "step": 9689,
      "training_step_time": 6.749624490737915
    },
    {
      "epoch": 9.241104125976562e-06,
      "grad_norm": 0.24206529557704926,
      "learning_rate": 9.940009478047173e-06,
      "loss": 0.0628,
      "step": 9690
    },
    {
      "epoch": 9.241104125976562e-06,
      "model_forward_time": 1.711137056350708,
      "step": 9690
    },
    {
      "dynamics_loss_avg": 0.06600643731653691,
      "epoch": 9.241104125976562e-06,
      "step": 9690
    },
    {
      "action_loss_avg": 0.00618671162519604,
      "epoch": 9.241104125976562e-06,
      "step": 9690
    },
    {
      "epoch": 9.241104125976562e-06,
      "step": 9690,
      "training_step_time": 6.734384298324585
    },
    {
      "epoch": 9.24205780029297e-06,
      "model_forward_time": 1.7070469856262207,
      "step": 9691
    },
    {
      "epoch": 9.24205780029297e-06,
      "step": 9691,
      "training_step_time": 7.796621322631836
    },
    {
      "epoch": 9.243011474609375e-06,
      "model_forward_time": 1.7526373863220215,
      "step": 9692
    },
    {
      "epoch": 9.243011474609375e-06,
      "step": 9692,
      "training_step_time": 8.463884830474854
    },
    {
      "epoch": 9.243965148925781e-06,
      "model_forward_time": 1.7135250568389893,
      "step": 9693
    },
    {
      "epoch": 9.243965148925781e-06,
      "step": 9693,
      "training_step_time": 8.714938163757324
    },
    {
      "epoch": 9.244918823242188e-06,
      "model_forward_time": 1.7094101905822754,
      "step": 9694
    },
    {
      "epoch": 9.244918823242188e-06,
      "step": 9694,
      "training_step_time": 7.821589469909668
    },
    {
      "epoch": 9.245872497558594e-06,
      "model_forward_time": 1.7210845947265625,
      "step": 9695
    },
    {
      "epoch": 9.245872497558594e-06,
      "step": 9695,
      "training_step_time": 6.777726650238037
    },
    {
      "epoch": 9.246826171875e-06,
      "model_forward_time": 1.7096223831176758,
      "step": 9696
    },
    {
      "epoch": 9.246826171875e-06,
      "step": 9696,
      "training_step_time": 6.763081789016724
    },
    {
      "epoch": 9.247779846191405e-06,
      "model_forward_time": 1.7053661346435547,
      "step": 9697
    },
    {
      "epoch": 9.247779846191405e-06,
      "step": 9697,
      "training_step_time": 6.727572917938232
    },
    {
      "epoch": 9.248733520507813e-06,
      "model_forward_time": 1.7112669944763184,
      "step": 9698
    },
    {
      "epoch": 9.248733520507813e-06,
      "step": 9698,
      "training_step_time": 6.744009494781494
    },
    {
      "epoch": 9.249687194824219e-06,
      "model_forward_time": 1.7047741413116455,
      "step": 9699
    },
    {
      "epoch": 9.249687194824219e-06,
      "step": 9699,
      "training_step_time": 6.688161849975586
    },
    {
      "epoch": 9.250640869140624e-06,
      "grad_norm": 0.2518468499183655,
      "learning_rate": 9.939753843162919e-06,
      "loss": 0.071,
      "step": 9700
    },
    {
      "epoch": 9.250640869140624e-06,
      "model_forward_time": 1.6992244720458984,
      "step": 9700
    },
    {
      "dynamics_loss_avg": 0.06472157537937165,
      "epoch": 9.250640869140624e-06,
      "step": 9700
    },
    {
      "action_loss_avg": 0.00414220776874572,
      "epoch": 9.250640869140624e-06,
      "step": 9700
    },
    {
      "epoch": 9.250640869140624e-06,
      "step": 9700,
      "training_step_time": 7.059067010879517
    },
    {
      "epoch": 9.251594543457032e-06,
      "model_forward_time": 1.7473230361938477,
      "step": 9701
    },
    {
      "epoch": 9.251594543457032e-06,
      "step": 9701,
      "training_step_time": 8.19877314567566
    },
    {
      "epoch": 9.252548217773437e-06,
      "model_forward_time": 1.3238019943237305,
      "step": 9702
    },
    {
      "epoch": 9.252548217773437e-06,
      "step": 9702,
      "training_step_time": 9.642099618911743
    },
    {
      "epoch": 9.253501892089843e-06,
      "model_forward_time": 1.7027151584625244,
      "step": 9703
    },
    {
      "epoch": 9.253501892089843e-06,
      "step": 9703,
      "training_step_time": 7.860514402389526
    },
    {
      "epoch": 9.25445556640625e-06,
      "model_forward_time": 1.7069463729858398,
      "step": 9704
    },
    {
      "epoch": 9.25445556640625e-06,
      "step": 9704,
      "training_step_time": 6.752045154571533
    },
    {
      "epoch": 9.255409240722656e-06,
      "model_forward_time": 1.7102198600769043,
      "step": 9705
    },
    {
      "epoch": 9.255409240722656e-06,
      "step": 9705,
      "training_step_time": 6.7307798862457275
    },
    {
      "epoch": 9.256362915039062e-06,
      "model_forward_time": 1.693190097808838,
      "step": 9706
    },
    {
      "epoch": 9.256362915039062e-06,
      "step": 9706,
      "training_step_time": 6.741997003555298
    },
    {
      "epoch": 9.25731658935547e-06,
      "model_forward_time": 1.7099535465240479,
      "step": 9707
    },
    {
      "epoch": 9.25731658935547e-06,
      "step": 9707,
      "training_step_time": 6.788060426712036
    },
    {
      "epoch": 9.258270263671875e-06,
      "model_forward_time": 1.705678939819336,
      "step": 9708
    },
    {
      "epoch": 9.258270263671875e-06,
      "step": 9708,
      "training_step_time": 6.745664358139038
    },
    {
      "epoch": 9.259223937988281e-06,
      "model_forward_time": 1.7001962661743164,
      "step": 9709
    },
    {
      "epoch": 9.259223937988281e-06,
      "step": 9709,
      "training_step_time": 6.787029981613159
    },
    {
      "epoch": 9.260177612304688e-06,
      "grad_norm": 0.2526674270629883,
      "learning_rate": 9.93949766807461e-06,
      "loss": 0.0672,
      "step": 9710
    },
    {
      "epoch": 9.260177612304688e-06,
      "model_forward_time": 2.2687337398529053,
      "step": 9710
    },
    {
      "dynamics_loss_avg": 0.06867488734424114,
      "epoch": 9.260177612304688e-06,
      "step": 9710
    },
    {
      "action_loss_avg": 0.004478600481525064,
      "epoch": 9.260177612304688e-06,
      "step": 9710
    },
    {
      "epoch": 9.260177612304688e-06,
      "step": 9710,
      "training_step_time": 7.653637886047363
    },
    {
      "epoch": 9.261131286621094e-06,
      "model_forward_time": 1.327615737915039,
      "step": 9711
    },
    {
      "epoch": 9.261131286621094e-06,
      "step": 9711,
      "training_step_time": 9.50460147857666
    },
    {
      "epoch": 9.2620849609375e-06,
      "model_forward_time": 1.7094011306762695,
      "step": 9712
    },
    {
      "epoch": 9.2620849609375e-06,
      "step": 9712,
      "training_step_time": 7.746739864349365
    },
    {
      "epoch": 9.263038635253905e-06,
      "model_forward_time": 1.7082126140594482,
      "step": 9713
    },
    {
      "epoch": 9.263038635253905e-06,
      "step": 9713,
      "training_step_time": 7.897090911865234
    },
    {
      "epoch": 9.263992309570313e-06,
      "model_forward_time": 1.7075870037078857,
      "step": 9714
    },
    {
      "epoch": 9.263992309570313e-06,
      "step": 9714,
      "training_step_time": 6.813517808914185
    },
    {
      "epoch": 9.264945983886719e-06,
      "model_forward_time": 1.6993403434753418,
      "step": 9715
    },
    {
      "epoch": 9.264945983886719e-06,
      "step": 9715,
      "training_step_time": 6.771654844284058
    },
    {
      "epoch": 9.265899658203124e-06,
      "model_forward_time": 1.6999688148498535,
      "step": 9716
    },
    {
      "epoch": 9.265899658203124e-06,
      "step": 9716,
      "training_step_time": 6.770122289657593
    },
    {
      "epoch": 9.266853332519532e-06,
      "model_forward_time": 1.697627305984497,
      "step": 9717
    },
    {
      "epoch": 9.266853332519532e-06,
      "step": 9717,
      "training_step_time": 6.7489025592803955
    },
    {
      "epoch": 9.267807006835937e-06,
      "model_forward_time": 1.7121877670288086,
      "step": 9718
    },
    {
      "epoch": 9.267807006835937e-06,
      "step": 9718,
      "training_step_time": 6.801792621612549
    },
    {
      "epoch": 9.268760681152343e-06,
      "model_forward_time": 2.2603397369384766,
      "step": 9719
    },
    {
      "epoch": 9.268760681152343e-06,
      "step": 9719,
      "training_step_time": 8.177383184432983
    },
    {
      "epoch": 9.26971435546875e-06,
      "grad_norm": 0.2981056571006775,
      "learning_rate": 9.939240952810262e-06,
      "loss": 0.0724,
      "step": 9720
    },
    {
      "epoch": 9.26971435546875e-06,
      "model_forward_time": 1.751293420791626,
      "step": 9720
    },
    {
      "dynamics_loss_avg": 0.05624109748750925,
      "epoch": 9.26971435546875e-06,
      "step": 9720
    },
    {
      "action_loss_avg": 0.004755880241282285,
      "epoch": 9.26971435546875e-06,
      "step": 9720
    },
    {
      "epoch": 9.26971435546875e-06,
      "step": 9720,
      "training_step_time": 9.066619634628296
    },
    {
      "epoch": 9.270668029785156e-06,
      "model_forward_time": 1.7161250114440918,
      "step": 9721
    },
    {
      "epoch": 9.270668029785156e-06,
      "step": 9721,
      "training_step_time": 8.667755842208862
    },
    {
      "epoch": 9.271621704101562e-06,
      "model_forward_time": 1.7106289863586426,
      "step": 9722
    },
    {
      "epoch": 9.271621704101562e-06,
      "step": 9722,
      "training_step_time": 7.825612306594849
    },
    {
      "epoch": 9.27257537841797e-06,
      "model_forward_time": 1.7081410884857178,
      "step": 9723
    },
    {
      "epoch": 9.27257537841797e-06,
      "step": 9723,
      "training_step_time": 6.79133415222168
    },
    {
      "epoch": 9.273529052734375e-06,
      "model_forward_time": 1.7186312675476074,
      "step": 9724
    },
    {
      "epoch": 9.273529052734375e-06,
      "step": 9724,
      "training_step_time": 6.75846791267395
    },
    {
      "epoch": 9.274482727050781e-06,
      "model_forward_time": 1.7130472660064697,
      "step": 9725
    },
    {
      "epoch": 9.274482727050781e-06,
      "step": 9725,
      "training_step_time": 6.755834102630615
    },
    {
      "epoch": 9.275436401367188e-06,
      "model_forward_time": 1.7105906009674072,
      "step": 9726
    },
    {
      "epoch": 9.275436401367188e-06,
      "step": 9726,
      "training_step_time": 6.777583360671997
    },
    {
      "epoch": 9.276390075683594e-06,
      "model_forward_time": 1.7094109058380127,
      "step": 9727
    },
    {
      "epoch": 9.276390075683594e-06,
      "step": 9727,
      "training_step_time": 6.746246576309204
    },
    {
      "epoch": 9.27734375e-06,
      "model_forward_time": 2.2475388050079346,
      "step": 9728
    },
    {
      "epoch": 9.27734375e-06,
      "step": 9728,
      "training_step_time": 8.673563718795776
    },
    {
      "epoch": 9.278297424316405e-06,
      "model_forward_time": 1.7604072093963623,
      "step": 9729
    },
    {
      "epoch": 9.278297424316405e-06,
      "step": 9729,
      "training_step_time": 9.012410402297974
    },
    {
      "epoch": 9.279251098632813e-06,
      "grad_norm": 0.28594279289245605,
      "learning_rate": 9.938983697397948e-06,
      "loss": 0.0729,
      "step": 9730
    },
    {
      "epoch": 9.279251098632813e-06,
      "model_forward_time": 1.7057445049285889,
      "step": 9730
    },
    {
      "dynamics_loss_avg": 0.07185812704265118,
      "epoch": 9.279251098632813e-06,
      "step": 9730
    },
    {
      "action_loss_avg": 0.006779302447102964,
      "epoch": 9.279251098632813e-06,
      "step": 9730
    },
    {
      "epoch": 9.279251098632813e-06,
      "step": 9730,
      "training_step_time": 10.075793266296387
    },
    {
      "epoch": 9.280204772949219e-06,
      "model_forward_time": 1.7083320617675781,
      "step": 9731
    },
    {
      "epoch": 9.280204772949219e-06,
      "step": 9731,
      "training_step_time": 7.998459577560425
    },
    {
      "epoch": 9.281158447265624e-06,
      "model_forward_time": 1.7088871002197266,
      "step": 9732
    },
    {
      "epoch": 9.281158447265624e-06,
      "step": 9732,
      "training_step_time": 6.769383192062378
    },
    {
      "epoch": 9.282112121582032e-06,
      "model_forward_time": 1.719609260559082,
      "step": 9733
    },
    {
      "epoch": 9.282112121582032e-06,
      "step": 9733,
      "training_step_time": 6.791182518005371
    },
    {
      "epoch": 9.283065795898437e-06,
      "model_forward_time": 1.7088439464569092,
      "step": 9734
    },
    {
      "epoch": 9.283065795898437e-06,
      "step": 9734,
      "training_step_time": 6.648029565811157
    },
    {
      "epoch": 9.284019470214843e-06,
      "model_forward_time": 1.7065463066101074,
      "step": 9735
    },
    {
      "epoch": 9.284019470214843e-06,
      "step": 9735,
      "training_step_time": 6.809462070465088
    },
    {
      "epoch": 9.28497314453125e-06,
      "model_forward_time": 0.9922263622283936,
      "step": 9736
    },
    {
      "epoch": 9.28497314453125e-06,
      "step": 9736,
      "training_step_time": 6.7916975021362305
    },
    {
      "epoch": 9.285926818847656e-06,
      "model_forward_time": 2.2414798736572266,
      "step": 9737
    },
    {
      "epoch": 9.285926818847656e-06,
      "step": 9737,
      "training_step_time": 8.398458242416382
    },
    {
      "epoch": 9.286880493164062e-06,
      "model_forward_time": 1.7562158107757568,
      "step": 9738
    },
    {
      "epoch": 9.286880493164062e-06,
      "step": 9738,
      "training_step_time": 9.13074016571045
    },
    {
      "epoch": 9.28783416748047e-06,
      "model_forward_time": 1.7051994800567627,
      "step": 9739
    },
    {
      "epoch": 9.28783416748047e-06,
      "step": 9739,
      "training_step_time": 10.726890563964844
    },
    {
      "epoch": 9.288787841796875e-06,
      "grad_norm": 0.2465950846672058,
      "learning_rate": 9.938725901865805e-06,
      "loss": 0.0712,
      "step": 9740
    },
    {
      "epoch": 9.288787841796875e-06,
      "model_forward_time": 1.6965363025665283,
      "step": 9740
    },
    {
      "dynamics_loss_avg": 0.06365934368222952,
      "epoch": 9.288787841796875e-06,
      "step": 9740
    },
    {
      "action_loss_avg": 0.006273023528046906,
      "epoch": 9.288787841796875e-06,
      "step": 9740
    },
    {
      "epoch": 9.288787841796875e-06,
      "step": 9740,
      "training_step_time": 7.842688083648682
    },
    {
      "epoch": 9.289741516113281e-06,
      "model_forward_time": 1.710127592086792,
      "step": 9741
    },
    {
      "epoch": 9.289741516113281e-06,
      "step": 9741,
      "training_step_time": 6.957429647445679
    },
    {
      "epoch": 9.290695190429688e-06,
      "model_forward_time": 1.708179235458374,
      "step": 9742
    },
    {
      "epoch": 9.290695190429688e-06,
      "step": 9742,
      "training_step_time": 6.774629831314087
    },
    {
      "epoch": 9.291648864746094e-06,
      "model_forward_time": 1.7121708393096924,
      "step": 9743
    },
    {
      "epoch": 9.291648864746094e-06,
      "step": 9743,
      "training_step_time": 6.767155408859253
    },
    {
      "epoch": 9.2926025390625e-06,
      "model_forward_time": 1.7113304138183594,
      "step": 9744
    },
    {
      "epoch": 9.2926025390625e-06,
      "step": 9744,
      "training_step_time": 6.778594017028809
    },
    {
      "epoch": 9.293556213378906e-06,
      "model_forward_time": 1.7030704021453857,
      "step": 9745
    },
    {
      "epoch": 9.293556213378906e-06,
      "step": 9745,
      "training_step_time": 6.762661933898926
    },
    {
      "epoch": 9.294509887695313e-06,
      "model_forward_time": 1.7155683040618896,
      "step": 9746
    },
    {
      "epoch": 9.294509887695313e-06,
      "step": 9746,
      "training_step_time": 8.361592531204224
    },
    {
      "epoch": 9.295463562011719e-06,
      "model_forward_time": 2.2710959911346436,
      "step": 9747
    },
    {
      "epoch": 9.295463562011719e-06,
      "step": 9747,
      "training_step_time": 8.4363431930542
    },
    {
      "epoch": 9.296417236328124e-06,
      "model_forward_time": 1.7044644355773926,
      "step": 9748
    },
    {
      "epoch": 9.296417236328124e-06,
      "step": 9748,
      "training_step_time": 10.265358924865723
    },
    {
      "epoch": 9.297370910644532e-06,
      "model_forward_time": 1.707871437072754,
      "step": 9749
    },
    {
      "epoch": 9.297370910644532e-06,
      "step": 9749,
      "training_step_time": 8.675331830978394
    },
    {
      "epoch": 9.298324584960938e-06,
      "grad_norm": 0.2675195634365082,
      "learning_rate": 9.938467566242021e-06,
      "loss": 0.0728,
      "step": 9750
    },
    {
      "epoch": 9.298324584960938e-06,
      "model_forward_time": 1.7053306102752686,
      "step": 9750
    },
    {
      "dynamics_loss_avg": 0.06317232022993266,
      "epoch": 9.298324584960938e-06,
      "step": 9750
    },
    {
      "action_loss_avg": 0.005596927352598868,
      "epoch": 9.298324584960938e-06,
      "step": 9750
    },
    {
      "epoch": 9.298324584960938e-06,
      "step": 9750,
      "training_step_time": 7.533316373825073
    },
    {
      "epoch": 9.299278259277343e-06,
      "model_forward_time": 1.7058665752410889,
      "step": 9751
    },
    {
      "epoch": 9.299278259277343e-06,
      "step": 9751,
      "training_step_time": 6.777748346328735
    },
    {
      "epoch": 9.30023193359375e-06,
      "model_forward_time": 1.7129297256469727,
      "step": 9752
    },
    {
      "epoch": 9.30023193359375e-06,
      "step": 9752,
      "training_step_time": 6.7554826736450195
    },
    {
      "epoch": 9.301185607910156e-06,
      "model_forward_time": 1.7035140991210938,
      "step": 9753
    },
    {
      "epoch": 9.301185607910156e-06,
      "step": 9753,
      "training_step_time": 6.814440488815308
    },
    {
      "epoch": 9.302139282226562e-06,
      "model_forward_time": 1.7057669162750244,
      "step": 9754
    },
    {
      "epoch": 9.302139282226562e-06,
      "step": 9754,
      "training_step_time": 6.788834095001221
    },
    {
      "epoch": 9.30309295654297e-06,
      "model_forward_time": 1.7186198234558105,
      "step": 9755
    },
    {
      "epoch": 9.30309295654297e-06,
      "step": 9755,
      "training_step_time": 8.12990117073059
    },
    {
      "epoch": 9.304046630859375e-06,
      "model_forward_time": 2.2428085803985596,
      "step": 9756
    },
    {
      "epoch": 9.304046630859375e-06,
      "step": 9756,
      "training_step_time": 7.602008819580078
    },
    {
      "epoch": 9.305000305175781e-06,
      "model_forward_time": 1.7444329261779785,
      "step": 9757
    },
    {
      "epoch": 9.305000305175781e-06,
      "step": 9757,
      "training_step_time": 9.30106520652771
    },
    {
      "epoch": 9.305953979492187e-06,
      "model_forward_time": 1.7045440673828125,
      "step": 9758
    },
    {
      "epoch": 9.305953979492187e-06,
      "step": 9758,
      "training_step_time": 8.64810299873352
    },
    {
      "epoch": 9.306907653808594e-06,
      "model_forward_time": 1.7040956020355225,
      "step": 9759
    },
    {
      "epoch": 9.306907653808594e-06,
      "step": 9759,
      "training_step_time": 7.9124367237091064
    },
    {
      "epoch": 9.307861328125e-06,
      "grad_norm": 0.2419251650571823,
      "learning_rate": 9.93820869055485e-06,
      "loss": 0.0764,
      "step": 9760
    },
    {
      "epoch": 9.307861328125e-06,
      "model_forward_time": 1.704772710800171,
      "step": 9760
    },
    {
      "dynamics_loss_avg": 0.061609197221696374,
      "epoch": 9.307861328125e-06,
      "step": 9760
    },
    {
      "action_loss_avg": 0.004153312684502452,
      "epoch": 9.307861328125e-06,
      "step": 9760
    },
    {
      "epoch": 9.307861328125e-06,
      "step": 9760,
      "training_step_time": 6.89816427230835
    },
    {
      "epoch": 9.308815002441406e-06,
      "model_forward_time": 1.7081551551818848,
      "step": 9761
    },
    {
      "epoch": 9.308815002441406e-06,
      "step": 9761,
      "training_step_time": 6.880687952041626
    },
    {
      "epoch": 9.309768676757813e-06,
      "model_forward_time": 1.705915927886963,
      "step": 9762
    },
    {
      "epoch": 9.309768676757813e-06,
      "step": 9762,
      "training_step_time": 6.909920930862427
    },
    {
      "epoch": 9.310722351074219e-06,
      "model_forward_time": 1.7068097591400146,
      "step": 9763
    },
    {
      "epoch": 9.310722351074219e-06,
      "step": 9763,
      "training_step_time": 6.9403908252716064
    },
    {
      "epoch": 9.311676025390624e-06,
      "model_forward_time": 1.7179608345031738,
      "step": 9764
    },
    {
      "epoch": 9.311676025390624e-06,
      "step": 9764,
      "training_step_time": 8.448414087295532
    },
    {
      "epoch": 9.312629699707032e-06,
      "model_forward_time": 2.2381999492645264,
      "step": 9765
    },
    {
      "epoch": 9.312629699707032e-06,
      "step": 9765,
      "training_step_time": 7.769495964050293
    },
    {
      "epoch": 9.313583374023438e-06,
      "model_forward_time": 1.7494878768920898,
      "step": 9766
    },
    {
      "epoch": 9.313583374023438e-06,
      "step": 9766,
      "training_step_time": 10.38615107536316
    },
    {
      "epoch": 9.314537048339843e-06,
      "model_forward_time": 1.7119183540344238,
      "step": 9767
    },
    {
      "epoch": 9.314537048339843e-06,
      "step": 9767,
      "training_step_time": 8.805855989456177
    },
    {
      "epoch": 9.31549072265625e-06,
      "model_forward_time": 1.7210633754730225,
      "step": 9768
    },
    {
      "epoch": 9.31549072265625e-06,
      "step": 9768,
      "training_step_time": 8.081229448318481
    },
    {
      "epoch": 9.316444396972656e-06,
      "model_forward_time": 1.7122302055358887,
      "step": 9769
    },
    {
      "epoch": 9.316444396972656e-06,
      "step": 9769,
      "training_step_time": 6.912153244018555
    },
    {
      "epoch": 9.317398071289062e-06,
      "grad_norm": 0.2884242832660675,
      "learning_rate": 9.9379492748326e-06,
      "loss": 0.0705,
      "step": 9770
    },
    {
      "epoch": 9.317398071289062e-06,
      "model_forward_time": 1.7116649150848389,
      "step": 9770
    },
    {
      "dynamics_loss_avg": 0.06302407756447792,
      "epoch": 9.317398071289062e-06,
      "step": 9770
    },
    {
      "action_loss_avg": 0.004251924436539411,
      "epoch": 9.317398071289062e-06,
      "step": 9770
    },
    {
      "epoch": 9.317398071289062e-06,
      "step": 9770,
      "training_step_time": 6.910522699356079
    },
    {
      "epoch": 9.31835174560547e-06,
      "model_forward_time": 1.7142870426177979,
      "step": 9771
    },
    {
      "epoch": 9.31835174560547e-06,
      "step": 9771,
      "training_step_time": 6.9467339515686035
    },
    {
      "epoch": 9.319305419921875e-06,
      "model_forward_time": 1.7178244590759277,
      "step": 9772
    },
    {
      "epoch": 9.319305419921875e-06,
      "step": 9772,
      "training_step_time": 6.9353978633880615
    },
    {
      "epoch": 9.320259094238281e-06,
      "model_forward_time": 1.7118542194366455,
      "step": 9773
    },
    {
      "epoch": 9.320259094238281e-06,
      "step": 9773,
      "training_step_time": 7.705600738525391
    },
    {
      "epoch": 9.321212768554687e-06,
      "model_forward_time": 2.261874198913574,
      "step": 9774
    },
    {
      "epoch": 9.321212768554687e-06,
      "step": 9774,
      "training_step_time": 7.810502290725708
    },
    {
      "epoch": 9.322166442871094e-06,
      "model_forward_time": 1.750075101852417,
      "step": 9775
    },
    {
      "epoch": 9.322166442871094e-06,
      "step": 9775,
      "training_step_time": 9.498771905899048
    },
    {
      "epoch": 9.3231201171875e-06,
      "model_forward_time": 1.7118101119995117,
      "step": 9776
    },
    {
      "epoch": 9.3231201171875e-06,
      "step": 9776,
      "training_step_time": 7.662245035171509
    },
    {
      "epoch": 9.324073791503906e-06,
      "model_forward_time": 1.7189369201660156,
      "step": 9777
    },
    {
      "epoch": 9.324073791503906e-06,
      "step": 9777,
      "training_step_time": 8.723483324050903
    },
    {
      "epoch": 9.325027465820313e-06,
      "model_forward_time": 1.7165687084197998,
      "step": 9778
    },
    {
      "epoch": 9.325027465820313e-06,
      "step": 9778,
      "training_step_time": 6.7657692432403564
    },
    {
      "epoch": 9.325981140136719e-06,
      "model_forward_time": 1.7190430164337158,
      "step": 9779
    },
    {
      "epoch": 9.325981140136719e-06,
      "step": 9779,
      "training_step_time": 6.725358247756958
    },
    {
      "epoch": 9.326934814453124e-06,
      "grad_norm": 0.19035901129245758,
      "learning_rate": 9.937689319103642e-06,
      "loss": 0.069,
      "step": 9780
    },
    {
      "epoch": 9.326934814453124e-06,
      "model_forward_time": 1.7151997089385986,
      "step": 9780
    },
    {
      "dynamics_loss_avg": 0.07230818355455995,
      "epoch": 9.326934814453124e-06,
      "step": 9780
    },
    {
      "action_loss_avg": 0.004810520069440827,
      "epoch": 9.326934814453124e-06,
      "step": 9780
    },
    {
      "epoch": 9.326934814453124e-06,
      "step": 9780,
      "training_step_time": 6.753799915313721
    },
    {
      "epoch": 9.327888488769532e-06,
      "model_forward_time": 1.7296640872955322,
      "step": 9781
    },
    {
      "epoch": 9.327888488769532e-06,
      "step": 9781,
      "training_step_time": 6.757444620132446
    },
    {
      "epoch": 9.328842163085938e-06,
      "model_forward_time": 1.724461317062378,
      "step": 9782
    },
    {
      "epoch": 9.328842163085938e-06,
      "step": 9782,
      "training_step_time": 7.60713791847229
    },
    {
      "epoch": 9.329795837402343e-06,
      "model_forward_time": 2.274906873703003,
      "step": 9783
    },
    {
      "epoch": 9.329795837402343e-06,
      "step": 9783,
      "training_step_time": 7.670789480209351
    },
    {
      "epoch": 9.33074951171875e-06,
      "model_forward_time": 1.7306580543518066,
      "step": 9784
    },
    {
      "epoch": 9.33074951171875e-06,
      "step": 9784,
      "training_step_time": 9.54411268234253
    },
    {
      "epoch": 9.331703186035156e-06,
      "model_forward_time": 1.718522548675537,
      "step": 9785
    },
    {
      "epoch": 9.331703186035156e-06,
      "step": 9785,
      "training_step_time": 9.44852352142334
    },
    {
      "epoch": 9.332656860351562e-06,
      "model_forward_time": 1.7273342609405518,
      "step": 9786
    },
    {
      "epoch": 9.332656860351562e-06,
      "step": 9786,
      "training_step_time": 7.635135889053345
    },
    {
      "epoch": 9.33361053466797e-06,
      "model_forward_time": 1.720698356628418,
      "step": 9787
    },
    {
      "epoch": 9.33361053466797e-06,
      "step": 9787,
      "training_step_time": 7.787902593612671
    },
    {
      "epoch": 9.334564208984375e-06,
      "model_forward_time": 1.723146677017212,
      "step": 9788
    },
    {
      "epoch": 9.334564208984375e-06,
      "step": 9788,
      "training_step_time": 6.766870021820068
    },
    {
      "epoch": 9.335517883300781e-06,
      "model_forward_time": 1.716646432876587,
      "step": 9789
    },
    {
      "epoch": 9.335517883300781e-06,
      "step": 9789,
      "training_step_time": 6.813894271850586
    },
    {
      "epoch": 9.336471557617187e-06,
      "grad_norm": 0.2135837823152542,
      "learning_rate": 9.937428823396404e-06,
      "loss": 0.062,
      "step": 9790
    },
    {
      "epoch": 9.336471557617187e-06,
      "model_forward_time": 1.7237043380737305,
      "step": 9790
    },
    {
      "dynamics_loss_avg": 0.05081794410943985,
      "epoch": 9.336471557617187e-06,
      "step": 9790
    },
    {
      "action_loss_avg": 0.004209394787903875,
      "epoch": 9.336471557617187e-06,
      "step": 9790
    },
    {
      "epoch": 9.336471557617187e-06,
      "step": 9790,
      "training_step_time": 6.7804787158966064
    },
    {
      "epoch": 9.337425231933594e-06,
      "model_forward_time": 1.7219324111938477,
      "step": 9791
    },
    {
      "epoch": 9.337425231933594e-06,
      "step": 9791,
      "training_step_time": 7.6184241771698
    },
    {
      "epoch": 9.33837890625e-06,
      "model_forward_time": 1.7139496803283691,
      "step": 9792
    },
    {
      "epoch": 9.33837890625e-06,
      "step": 9792,
      "training_step_time": 8.824651718139648
    },
    {
      "epoch": 9.339332580566406e-06,
      "model_forward_time": 1.701730489730835,
      "step": 9793
    },
    {
      "epoch": 9.339332580566406e-06,
      "step": 9793,
      "training_step_time": 8.185275554656982
    },
    {
      "epoch": 9.340286254882813e-06,
      "model_forward_time": 1.713947057723999,
      "step": 9794
    },
    {
      "epoch": 9.340286254882813e-06,
      "step": 9794,
      "training_step_time": 9.27185344696045
    },
    {
      "epoch": 9.341239929199219e-06,
      "model_forward_time": 1.7229516506195068,
      "step": 9795
    },
    {
      "epoch": 9.341239929199219e-06,
      "step": 9795,
      "training_step_time": 7.754085063934326
    },
    {
      "epoch": 9.342193603515624e-06,
      "model_forward_time": 1.7150297164916992,
      "step": 9796
    },
    {
      "epoch": 9.342193603515624e-06,
      "step": 9796,
      "training_step_time": 7.723510980606079
    },
    {
      "epoch": 9.343147277832032e-06,
      "model_forward_time": 1.7144954204559326,
      "step": 9797
    },
    {
      "epoch": 9.343147277832032e-06,
      "step": 9797,
      "training_step_time": 6.791549444198608
    },
    {
      "epoch": 9.344100952148438e-06,
      "model_forward_time": 1.7163951396942139,
      "step": 9798
    },
    {
      "epoch": 9.344100952148438e-06,
      "step": 9798,
      "training_step_time": 6.8139989376068115
    },
    {
      "epoch": 9.345054626464843e-06,
      "model_forward_time": 1.721872091293335,
      "step": 9799
    },
    {
      "epoch": 9.345054626464843e-06,
      "step": 9799,
      "training_step_time": 6.791122198104858
    },
    {
      "epoch": 9.34600830078125e-06,
      "grad_norm": 0.19376574456691742,
      "learning_rate": 9.937167787739372e-06,
      "loss": 0.0643,
      "step": 9800
    },
    {
      "epoch": 9.34600830078125e-06,
      "model_forward_time": 1.7059149742126465,
      "step": 9800
    },
    {
      "dynamics_loss_avg": 0.0496103348210454,
      "epoch": 9.34600830078125e-06,
      "step": 9800
    },
    {
      "action_loss_avg": 0.0047202424262650314,
      "epoch": 9.34600830078125e-06,
      "step": 9800
    },
    {
      "epoch": 9.34600830078125e-06,
      "step": 9800,
      "training_step_time": 8.115003108978271
    },
    {
      "epoch": 9.346961975097656e-06,
      "model_forward_time": 1.715933084487915,
      "step": 9801
    },
    {
      "epoch": 9.346961975097656e-06,
      "step": 9801,
      "training_step_time": 7.649904251098633
    },
    {
      "epoch": 9.347915649414062e-06,
      "model_forward_time": 2.267528533935547,
      "step": 9802
    },
    {
      "epoch": 9.347915649414062e-06,
      "step": 9802,
      "training_step_time": 10.614845275878906
    },
    {
      "epoch": 9.34886932373047e-06,
      "model_forward_time": 1.7558841705322266,
      "step": 9803
    },
    {
      "epoch": 9.34886932373047e-06,
      "step": 9803,
      "training_step_time": 8.294628381729126
    },
    {
      "epoch": 9.349822998046875e-06,
      "model_forward_time": 1.7029054164886475,
      "step": 9804
    },
    {
      "epoch": 9.349822998046875e-06,
      "step": 9804,
      "training_step_time": 7.634284973144531
    },
    {
      "epoch": 9.350776672363281e-06,
      "model_forward_time": 1.7134530544281006,
      "step": 9805
    },
    {
      "epoch": 9.350776672363281e-06,
      "step": 9805,
      "training_step_time": 7.821375846862793
    },
    {
      "epoch": 9.351730346679687e-06,
      "model_forward_time": 1.7038886547088623,
      "step": 9806
    },
    {
      "epoch": 9.351730346679687e-06,
      "step": 9806,
      "training_step_time": 6.750859022140503
    },
    {
      "epoch": 9.352684020996094e-06,
      "model_forward_time": 1.721374273300171,
      "step": 9807
    },
    {
      "epoch": 9.352684020996094e-06,
      "step": 9807,
      "training_step_time": 6.762654781341553
    },
    {
      "epoch": 9.3536376953125e-06,
      "model_forward_time": 1.7217354774475098,
      "step": 9808
    },
    {
      "epoch": 9.3536376953125e-06,
      "step": 9808,
      "training_step_time": 6.75918173789978
    },
    {
      "epoch": 9.354591369628906e-06,
      "model_forward_time": 1.7075426578521729,
      "step": 9809
    },
    {
      "epoch": 9.354591369628906e-06,
      "step": 9809,
      "training_step_time": 7.194660902023315
    },
    {
      "epoch": 9.355545043945313e-06,
      "grad_norm": 0.3051232397556305,
      "learning_rate": 9.936906212161095e-06,
      "loss": 0.0649,
      "step": 9810
    },
    {
      "epoch": 9.355545043945313e-06,
      "model_forward_time": 1.713571310043335,
      "step": 9810
    },
    {
      "dynamics_loss_avg": 0.06256914604455233,
      "epoch": 9.355545043945313e-06,
      "step": 9810
    },
    {
      "action_loss_avg": 0.005027941823936999,
      "epoch": 9.355545043945313e-06,
      "step": 9810
    },
    {
      "epoch": 9.355545043945313e-06,
      "step": 9810,
      "training_step_time": 7.629213571548462
    },
    {
      "epoch": 9.356498718261719e-06,
      "model_forward_time": 2.262761354446411,
      "step": 9811
    },
    {
      "epoch": 9.356498718261719e-06,
      "step": 9811,
      "training_step_time": 7.666720867156982
    },
    {
      "epoch": 9.357452392578124e-06,
      "model_forward_time": 1.7624437808990479,
      "step": 9812
    },
    {
      "epoch": 9.357452392578124e-06,
      "step": 9812,
      "training_step_time": 11.247323513031006
    },
    {
      "epoch": 9.358406066894532e-06,
      "model_forward_time": 1.693042278289795,
      "step": 9813
    },
    {
      "epoch": 9.358406066894532e-06,
      "step": 9813,
      "training_step_time": 10.073421001434326
    },
    {
      "epoch": 9.359359741210938e-06,
      "model_forward_time": 1.7220957279205322,
      "step": 9814
    },
    {
      "epoch": 9.359359741210938e-06,
      "step": 9814,
      "training_step_time": 7.834702968597412
    },
    {
      "epoch": 9.360313415527343e-06,
      "model_forward_time": 1.7137165069580078,
      "step": 9815
    },
    {
      "epoch": 9.360313415527343e-06,
      "step": 9815,
      "training_step_time": 7.185951471328735
    },
    {
      "epoch": 9.36126708984375e-06,
      "model_forward_time": 1.6903226375579834,
      "step": 9816
    },
    {
      "epoch": 9.36126708984375e-06,
      "step": 9816,
      "training_step_time": 6.719346761703491
    },
    {
      "epoch": 9.362220764160156e-06,
      "model_forward_time": 1.7007098197937012,
      "step": 9817
    },
    {
      "epoch": 9.362220764160156e-06,
      "step": 9817,
      "training_step_time": 6.75469708442688
    },
    {
      "epoch": 9.363174438476562e-06,
      "model_forward_time": 1.7081222534179688,
      "step": 9818
    },
    {
      "epoch": 9.363174438476562e-06,
      "step": 9818,
      "training_step_time": 7.191333770751953
    },
    {
      "epoch": 9.36412811279297e-06,
      "model_forward_time": 1.7067079544067383,
      "step": 9819
    },
    {
      "epoch": 9.36412811279297e-06,
      "step": 9819,
      "training_step_time": 7.565728664398193
    },
    {
      "epoch": 9.365081787109375e-06,
      "grad_norm": 0.22802747786045074,
      "learning_rate": 9.936644096690177e-06,
      "loss": 0.0703,
      "step": 9820
    },
    {
      "epoch": 9.365081787109375e-06,
      "model_forward_time": 2.2373745441436768,
      "step": 9820
    },
    {
      "dynamics_loss_avg": 0.05910844579339027,
      "epoch": 9.365081787109375e-06,
      "step": 9820
    },
    {
      "action_loss_avg": 0.0034185568219982088,
      "epoch": 9.365081787109375e-06,
      "step": 9820
    },
    {
      "epoch": 9.365081787109375e-06,
      "step": 9820,
      "training_step_time": 8.320704698562622
    },
    {
      "epoch": 9.366035461425781e-06,
      "model_forward_time": 1.7297792434692383,
      "step": 9821
    },
    {
      "epoch": 9.366035461425781e-06,
      "step": 9821,
      "training_step_time": 9.522584199905396
    },
    {
      "epoch": 9.366989135742187e-06,
      "model_forward_time": 1.7024049758911133,
      "step": 9822
    },
    {
      "epoch": 9.366989135742187e-06,
      "step": 9822,
      "training_step_time": 7.591464519500732
    },
    {
      "epoch": 9.367942810058594e-06,
      "model_forward_time": 1.7011029720306396,
      "step": 9823
    },
    {
      "epoch": 9.367942810058594e-06,
      "step": 9823,
      "training_step_time": 8.814295530319214
    },
    {
      "epoch": 9.368896484375e-06,
      "model_forward_time": 1.6998744010925293,
      "step": 9824
    },
    {
      "epoch": 9.368896484375e-06,
      "step": 9824,
      "training_step_time": 7.154821395874023
    },
    {
      "epoch": 9.369850158691406e-06,
      "model_forward_time": 1.702608346939087,
      "step": 9825
    },
    {
      "epoch": 9.369850158691406e-06,
      "step": 9825,
      "training_step_time": 6.736230850219727
    },
    {
      "epoch": 9.370803833007813e-06,
      "model_forward_time": 1.7004225254058838,
      "step": 9826
    },
    {
      "epoch": 9.370803833007813e-06,
      "step": 9826,
      "training_step_time": 6.783634662628174
    },
    {
      "epoch": 9.371757507324219e-06,
      "model_forward_time": 1.7095365524291992,
      "step": 9827
    },
    {
      "epoch": 9.371757507324219e-06,
      "step": 9827,
      "training_step_time": 7.297377109527588
    },
    {
      "epoch": 9.372711181640624e-06,
      "model_forward_time": 1.6858458518981934,
      "step": 9828
    },
    {
      "epoch": 9.372711181640624e-06,
      "step": 9828,
      "training_step_time": 7.512477874755859
    },
    {
      "epoch": 9.373664855957032e-06,
      "model_forward_time": 2.240891695022583,
      "step": 9829
    },
    {
      "epoch": 9.373664855957032e-06,
      "step": 9829,
      "training_step_time": 8.261175632476807
    },
    {
      "epoch": 9.374618530273438e-06,
      "grad_norm": 0.2445869743824005,
      "learning_rate": 9.936381441355282e-06,
      "loss": 0.0729,
      "step": 9830
    },
    {
      "epoch": 9.374618530273438e-06,
      "model_forward_time": 1.7172315120697021,
      "step": 9830
    },
    {
      "dynamics_loss_avg": 0.05755896139889956,
      "epoch": 9.374618530273438e-06,
      "step": 9830
    },
    {
      "action_loss_avg": 0.003416846683830954,
      "epoch": 9.374618530273438e-06,
      "step": 9830
    },
    {
      "epoch": 9.374618530273438e-06,
      "step": 9830,
      "training_step_time": 8.938398361206055
    },
    {
      "epoch": 9.375572204589843e-06,
      "model_forward_time": 1.7004327774047852,
      "step": 9831
    },
    {
      "epoch": 9.375572204589843e-06,
      "step": 9831,
      "training_step_time": 9.444130659103394
    },
    {
      "epoch": 9.37652587890625e-06,
      "model_forward_time": 1.7051544189453125,
      "step": 9832
    },
    {
      "epoch": 9.37652587890625e-06,
      "step": 9832,
      "training_step_time": 7.651175260543823
    },
    {
      "epoch": 9.377479553222656e-06,
      "model_forward_time": 1.6953778266906738,
      "step": 9833
    },
    {
      "epoch": 9.377479553222656e-06,
      "step": 9833,
      "training_step_time": 7.804555892944336
    },
    {
      "epoch": 9.378433227539062e-06,
      "model_forward_time": 1.698415756225586,
      "step": 9834
    },
    {
      "epoch": 9.378433227539062e-06,
      "step": 9834,
      "training_step_time": 6.740252733230591
    },
    {
      "epoch": 9.37938690185547e-06,
      "model_forward_time": 1.712742567062378,
      "step": 9835
    },
    {
      "epoch": 9.37938690185547e-06,
      "step": 9835,
      "training_step_time": 6.760857343673706
    },
    {
      "epoch": 9.380340576171875e-06,
      "model_forward_time": 1.7034554481506348,
      "step": 9836
    },
    {
      "epoch": 9.380340576171875e-06,
      "step": 9836,
      "training_step_time": 7.185328245162964
    },
    {
      "epoch": 9.381294250488281e-06,
      "model_forward_time": 1.69950532913208,
      "step": 9837
    },
    {
      "epoch": 9.381294250488281e-06,
      "step": 9837,
      "training_step_time": 7.988525629043579
    },
    {
      "epoch": 9.382247924804687e-06,
      "model_forward_time": 2.2175638675689697,
      "step": 9838
    },
    {
      "epoch": 9.382247924804687e-06,
      "step": 9838,
      "training_step_time": 7.710190773010254
    },
    {
      "epoch": 9.383201599121094e-06,
      "model_forward_time": 1.6983721256256104,
      "step": 9839
    },
    {
      "epoch": 9.383201599121094e-06,
      "step": 9839,
      "training_step_time": 8.575397968292236
    },
    {
      "epoch": 9.3841552734375e-06,
      "grad_norm": 0.17890720069408417,
      "learning_rate": 9.936118246185136e-06,
      "loss": 0.0659,
      "step": 9840
    },
    {
      "epoch": 9.3841552734375e-06,
      "model_forward_time": 1.739631175994873,
      "step": 9840
    },
    {
      "dynamics_loss_avg": 0.07392550278455019,
      "epoch": 9.3841552734375e-06,
      "step": 9840
    },
    {
      "action_loss_avg": 0.00496635683812201,
      "epoch": 9.3841552734375e-06,
      "step": 9840
    },
    {
      "epoch": 9.3841552734375e-06,
      "step": 9840,
      "training_step_time": 9.951699256896973
    },
    {
      "epoch": 9.385108947753906e-06,
      "model_forward_time": 1.7011806964874268,
      "step": 9841
    },
    {
      "epoch": 9.385108947753906e-06,
      "step": 9841,
      "training_step_time": 7.658730506896973
    },
    {
      "epoch": 9.386062622070313e-06,
      "model_forward_time": 1.6952145099639893,
      "step": 9842
    },
    {
      "epoch": 9.386062622070313e-06,
      "step": 9842,
      "training_step_time": 7.788925647735596
    },
    {
      "epoch": 9.387016296386719e-06,
      "model_forward_time": 1.7109057903289795,
      "step": 9843
    },
    {
      "epoch": 9.387016296386719e-06,
      "step": 9843,
      "training_step_time": 7.154503107070923
    },
    {
      "epoch": 9.387969970703124e-06,
      "model_forward_time": 1.7066795825958252,
      "step": 9844
    },
    {
      "epoch": 9.387969970703124e-06,
      "step": 9844,
      "training_step_time": 6.707169771194458
    },
    {
      "epoch": 9.388923645019532e-06,
      "model_forward_time": 1.7000184059143066,
      "step": 9845
    },
    {
      "epoch": 9.388923645019532e-06,
      "step": 9845,
      "training_step_time": 7.133852958679199
    },
    {
      "epoch": 9.389877319335938e-06,
      "model_forward_time": 1.7020697593688965,
      "step": 9846
    },
    {
      "epoch": 9.389877319335938e-06,
      "step": 9846,
      "training_step_time": 7.2203123569488525
    },
    {
      "epoch": 9.390830993652343e-06,
      "model_forward_time": 1.7015113830566406,
      "step": 9847
    },
    {
      "epoch": 9.390830993652343e-06,
      "step": 9847,
      "training_step_time": 8.230343103408813
    },
    {
      "epoch": 9.39178466796875e-06,
      "model_forward_time": 2.2314581871032715,
      "step": 9848
    },
    {
      "epoch": 9.39178466796875e-06,
      "step": 9848,
      "training_step_time": 7.481551170349121
    },
    {
      "epoch": 9.392738342285156e-06,
      "model_forward_time": 1.743903636932373,
      "step": 9849
    },
    {
      "epoch": 9.392738342285156e-06,
      "step": 9849,
      "training_step_time": 9.480563163757324
    },
    {
      "epoch": 9.393692016601562e-06,
      "grad_norm": 0.2606235146522522,
      "learning_rate": 9.935854511208517e-06,
      "loss": 0.0695,
      "step": 9850
    },
    {
      "epoch": 9.393692016601562e-06,
      "model_forward_time": 1.6924822330474854,
      "step": 9850
    },
    {
      "dynamics_loss_avg": 0.06023172978311777,
      "epoch": 9.393692016601562e-06,
      "step": 9850
    },
    {
      "action_loss_avg": 0.0040394418174400926,
      "epoch": 9.393692016601562e-06,
      "step": 9850
    },
    {
      "epoch": 9.393692016601562e-06,
      "step": 9850,
      "training_step_time": 8.565091848373413
    },
    {
      "epoch": 9.39464569091797e-06,
      "model_forward_time": 1.6998000144958496,
      "step": 9851
    },
    {
      "epoch": 9.39464569091797e-06,
      "step": 9851,
      "training_step_time": 7.774071931838989
    },
    {
      "epoch": 9.395599365234375e-06,
      "model_forward_time": 1.7034645080566406,
      "step": 9852
    },
    {
      "epoch": 9.395599365234375e-06,
      "step": 9852,
      "training_step_time": 7.151273250579834
    },
    {
      "epoch": 9.396553039550781e-06,
      "model_forward_time": 1.7079198360443115,
      "step": 9853
    },
    {
      "epoch": 9.396553039550781e-06,
      "step": 9853,
      "training_step_time": 6.702523231506348
    },
    {
      "epoch": 9.397506713867187e-06,
      "model_forward_time": 1.7009241580963135,
      "step": 9854
    },
    {
      "epoch": 9.397506713867187e-06,
      "step": 9854,
      "training_step_time": 7.145365476608276
    },
    {
      "epoch": 9.398460388183594e-06,
      "model_forward_time": 1.6971757411956787,
      "step": 9855
    },
    {
      "epoch": 9.398460388183594e-06,
      "step": 9855,
      "training_step_time": 6.736640691757202
    },
    {
      "epoch": 9.3994140625e-06,
      "model_forward_time": 1.7037744522094727,
      "step": 9856
    },
    {
      "epoch": 9.3994140625e-06,
      "step": 9856,
      "training_step_time": 7.465420722961426
    },
    {
      "epoch": 9.400367736816406e-06,
      "model_forward_time": 2.2293171882629395,
      "step": 9857
    },
    {
      "epoch": 9.400367736816406e-06,
      "step": 9857,
      "training_step_time": 7.718651533126831
    },
    {
      "epoch": 9.401321411132813e-06,
      "model_forward_time": 1.345881700515747,
      "step": 9858
    },
    {
      "epoch": 9.401321411132813e-06,
      "step": 9858,
      "training_step_time": 8.56552529335022
    },
    {
      "epoch": 9.402275085449219e-06,
      "model_forward_time": 1.7087523937225342,
      "step": 9859
    },
    {
      "epoch": 9.402275085449219e-06,
      "step": 9859,
      "training_step_time": 9.40267300605774
    },
    {
      "epoch": 9.403228759765624e-06,
      "grad_norm": 0.2617734670639038,
      "learning_rate": 9.935590236454273e-06,
      "loss": 0.0681,
      "step": 9860
    },
    {
      "epoch": 9.403228759765624e-06,
      "model_forward_time": 1.6966149806976318,
      "step": 9860
    },
    {
      "dynamics_loss_avg": 0.05762241240590811,
      "epoch": 9.403228759765624e-06,
      "step": 9860
    },
    {
      "action_loss_avg": 0.0046291837294120345,
      "epoch": 9.403228759765624e-06,
      "step": 9860
    },
    {
      "epoch": 9.403228759765624e-06,
      "step": 9860,
      "training_step_time": 7.596600770950317
    },
    {
      "epoch": 9.404182434082032e-06,
      "model_forward_time": 1.7067842483520508,
      "step": 9861
    },
    {
      "epoch": 9.404182434082032e-06,
      "step": 9861,
      "training_step_time": 7.81976842880249
    },
    {
      "epoch": 9.405136108398438e-06,
      "model_forward_time": 1.7177751064300537,
      "step": 9862
    },
    {
      "epoch": 9.405136108398438e-06,
      "step": 9862,
      "training_step_time": 7.177282333374023
    },
    {
      "epoch": 9.406089782714843e-06,
      "model_forward_time": 1.699842929840088,
      "step": 9863
    },
    {
      "epoch": 9.406089782714843e-06,
      "step": 9863,
      "training_step_time": 7.179956674575806
    },
    {
      "epoch": 9.40704345703125e-06,
      "model_forward_time": 1.6974072456359863,
      "step": 9864
    },
    {
      "epoch": 9.40704345703125e-06,
      "step": 9864,
      "training_step_time": 6.72481369972229
    },
    {
      "epoch": 9.407997131347656e-06,
      "model_forward_time": 1.7068700790405273,
      "step": 9865
    },
    {
      "epoch": 9.407997131347656e-06,
      "step": 9865,
      "training_step_time": 7.5469748973846436
    },
    {
      "epoch": 9.408950805664062e-06,
      "model_forward_time": 2.2224881649017334,
      "step": 9866
    },
    {
      "epoch": 9.408950805664062e-06,
      "step": 9866,
      "training_step_time": 9.379698514938354
    },
    {
      "epoch": 9.40990447998047e-06,
      "model_forward_time": 1.7375030517578125,
      "step": 9867
    },
    {
      "epoch": 9.40990447998047e-06,
      "step": 9867,
      "training_step_time": 9.64152216911316
    },
    {
      "epoch": 9.410858154296875e-06,
      "model_forward_time": 1.7018470764160156,
      "step": 9868
    },
    {
      "epoch": 9.410858154296875e-06,
      "step": 9868,
      "training_step_time": 8.209756135940552
    },
    {
      "epoch": 9.411811828613281e-06,
      "model_forward_time": 1.708855152130127,
      "step": 9869
    },
    {
      "epoch": 9.411811828613281e-06,
      "step": 9869,
      "training_step_time": 7.645512819290161
    },
    {
      "epoch": 9.412765502929687e-06,
      "grad_norm": 0.2634613513946533,
      "learning_rate": 9.935325421951299e-06,
      "loss": 0.0647,
      "step": 9870
    },
    {
      "epoch": 9.412765502929687e-06,
      "model_forward_time": 1.69844388961792,
      "step": 9870
    },
    {
      "dynamics_loss_avg": 0.05899428874254227,
      "epoch": 9.412765502929687e-06,
      "step": 9870
    },
    {
      "action_loss_avg": 0.003054998890729621,
      "epoch": 9.412765502929687e-06,
      "step": 9870
    },
    {
      "epoch": 9.412765502929687e-06,
      "step": 9870,
      "training_step_time": 7.909452676773071
    },
    {
      "epoch": 9.413719177246094e-06,
      "model_forward_time": 1.7040612697601318,
      "step": 9871
    },
    {
      "epoch": 9.413719177246094e-06,
      "step": 9871,
      "training_step_time": 7.250068187713623
    },
    {
      "epoch": 9.4146728515625e-06,
      "model_forward_time": 1.703247308731079,
      "step": 9872
    },
    {
      "epoch": 9.4146728515625e-06,
      "step": 9872,
      "training_step_time": 7.227852821350098
    },
    {
      "epoch": 9.415626525878906e-06,
      "model_forward_time": 1.7117764949798584,
      "step": 9873
    },
    {
      "epoch": 9.415626525878906e-06,
      "step": 9873,
      "training_step_time": 6.7867796421051025
    },
    {
      "epoch": 9.416580200195313e-06,
      "model_forward_time": 1.696650743484497,
      "step": 9874
    },
    {
      "epoch": 9.416580200195313e-06,
      "step": 9874,
      "training_step_time": 7.400039196014404
    },
    {
      "epoch": 9.417533874511719e-06,
      "model_forward_time": 2.2317302227020264,
      "step": 9875
    },
    {
      "epoch": 9.417533874511719e-06,
      "step": 9875,
      "training_step_time": 8.287207126617432
    },
    {
      "epoch": 9.418487548828124e-06,
      "model_forward_time": 1.7022347450256348,
      "step": 9876
    },
    {
      "epoch": 9.418487548828124e-06,
      "step": 9876,
      "training_step_time": 8.38312554359436
    },
    {
      "epoch": 9.419441223144532e-06,
      "model_forward_time": 1.7116074562072754,
      "step": 9877
    },
    {
      "epoch": 9.419441223144532e-06,
      "step": 9877,
      "training_step_time": 9.450882196426392
    },
    {
      "epoch": 9.420394897460938e-06,
      "model_forward_time": 1.7162883281707764,
      "step": 9878
    },
    {
      "epoch": 9.420394897460938e-06,
      "step": 9878,
      "training_step_time": 7.596441984176636
    },
    {
      "epoch": 9.421348571777343e-06,
      "model_forward_time": 1.7028894424438477,
      "step": 9879
    },
    {
      "epoch": 9.421348571777343e-06,
      "step": 9879,
      "training_step_time": 7.832665920257568
    },
    {
      "epoch": 9.42230224609375e-06,
      "grad_norm": 0.264768123626709,
      "learning_rate": 9.935060067728558e-06,
      "loss": 0.0711,
      "step": 9880
    },
    {
      "epoch": 9.42230224609375e-06,
      "model_forward_time": 1.6990439891815186,
      "step": 9880
    },
    {
      "dynamics_loss_avg": 0.04765116441994906,
      "epoch": 9.42230224609375e-06,
      "step": 9880
    },
    {
      "action_loss_avg": 0.004508822894422337,
      "epoch": 9.42230224609375e-06,
      "step": 9880
    },
    {
      "epoch": 9.42230224609375e-06,
      "step": 9880,
      "training_step_time": 7.154262542724609
    },
    {
      "epoch": 9.423255920410156e-06,
      "model_forward_time": 1.702784776687622,
      "step": 9881
    },
    {
      "epoch": 9.423255920410156e-06,
      "step": 9881,
      "training_step_time": 7.154251337051392
    },
    {
      "epoch": 9.424209594726562e-06,
      "model_forward_time": 1.708174467086792,
      "step": 9882
    },
    {
      "epoch": 9.424209594726562e-06,
      "step": 9882,
      "training_step_time": 6.765815496444702
    },
    {
      "epoch": 9.425163269042968e-06,
      "model_forward_time": 1.7070376873016357,
      "step": 9883
    },
    {
      "epoch": 9.425163269042968e-06,
      "step": 9883,
      "training_step_time": 7.398759841918945
    },
    {
      "epoch": 9.426116943359375e-06,
      "model_forward_time": 1.3108313083648682,
      "step": 9884
    },
    {
      "epoch": 9.426116943359375e-06,
      "step": 9884,
      "training_step_time": 7.7979960441589355
    },
    {
      "epoch": 9.427070617675781e-06,
      "model_forward_time": 2.2446835041046143,
      "step": 9885
    },
    {
      "epoch": 9.427070617675781e-06,
      "step": 9885,
      "training_step_time": 7.577075719833374
    },
    {
      "epoch": 9.428024291992187e-06,
      "model_forward_time": 1.7522342205047607,
      "step": 9886
    },
    {
      "epoch": 9.428024291992187e-06,
      "step": 9886,
      "training_step_time": 7.669769048690796
    },
    {
      "epoch": 9.428977966308594e-06,
      "model_forward_time": 1.69398832321167,
      "step": 9887
    },
    {
      "epoch": 9.428977966308594e-06,
      "step": 9887,
      "training_step_time": 7.657635688781738
    },
    {
      "epoch": 9.429931640625e-06,
      "model_forward_time": 1.7131657600402832,
      "step": 9888
    },
    {
      "epoch": 9.429931640625e-06,
      "step": 9888,
      "training_step_time": 7.631250619888306
    },
    {
      "epoch": 9.430885314941406e-06,
      "model_forward_time": 1.7094557285308838,
      "step": 9889
    },
    {
      "epoch": 9.430885314941406e-06,
      "step": 9889,
      "training_step_time": 7.877515077590942
    },
    {
      "epoch": 9.431838989257813e-06,
      "grad_norm": 0.22461768984794617,
      "learning_rate": 9.934794173815068e-06,
      "loss": 0.0677,
      "step": 9890
    },
    {
      "epoch": 9.431838989257813e-06,
      "model_forward_time": 1.7033545970916748,
      "step": 9890
    },
    {
      "dynamics_loss_avg": 0.048441202752292155,
      "epoch": 9.431838989257813e-06,
      "step": 9890
    },
    {
      "action_loss_avg": 0.0031873986124992372,
      "epoch": 9.431838989257813e-06,
      "step": 9890
    },
    {
      "epoch": 9.431838989257813e-06,
      "step": 9890,
      "training_step_time": 7.216228723526001
    },
    {
      "epoch": 9.432792663574219e-06,
      "model_forward_time": 1.706078290939331,
      "step": 9891
    },
    {
      "epoch": 9.432792663574219e-06,
      "step": 9891,
      "training_step_time": 6.7469868659973145
    },
    {
      "epoch": 9.433746337890624e-06,
      "model_forward_time": 1.7139322757720947,
      "step": 9892
    },
    {
      "epoch": 9.433746337890624e-06,
      "step": 9892,
      "training_step_time": 7.42926812171936
    },
    {
      "epoch": 9.434700012207032e-06,
      "model_forward_time": 1.7191767692565918,
      "step": 9893
    },
    {
      "epoch": 9.434700012207032e-06,
      "step": 9893,
      "training_step_time": 7.692350625991821
    },
    {
      "epoch": 9.435653686523438e-06,
      "model_forward_time": 2.2377052307128906,
      "step": 9894
    },
    {
      "epoch": 9.435653686523438e-06,
      "step": 9894,
      "training_step_time": 7.531743049621582
    },
    {
      "epoch": 9.436607360839843e-06,
      "model_forward_time": 1.7331194877624512,
      "step": 9895
    },
    {
      "epoch": 9.436607360839843e-06,
      "step": 9895,
      "training_step_time": 10.681933879852295
    },
    {
      "epoch": 9.43756103515625e-06,
      "model_forward_time": 1.7007081508636475,
      "step": 9896
    },
    {
      "epoch": 9.43756103515625e-06,
      "step": 9896,
      "training_step_time": 8.19288158416748
    },
    {
      "epoch": 9.438514709472656e-06,
      "model_forward_time": 1.7137997150421143,
      "step": 9897
    },
    {
      "epoch": 9.438514709472656e-06,
      "step": 9897,
      "training_step_time": 8.579625606536865
    },
    {
      "epoch": 9.439468383789062e-06,
      "model_forward_time": 1.3271644115447998,
      "step": 9898
    },
    {
      "epoch": 9.439468383789062e-06,
      "step": 9898,
      "training_step_time": 7.851992607116699
    },
    {
      "epoch": 9.440422058105468e-06,
      "model_forward_time": 1.6950454711914062,
      "step": 9899
    },
    {
      "epoch": 9.440422058105468e-06,
      "step": 9899,
      "training_step_time": 7.262174129486084
    },
    {
      "epoch": 9.441375732421875e-06,
      "grad_norm": 0.3278862535953522,
      "learning_rate": 9.934527740239906e-06,
      "loss": 0.0706,
      "step": 9900
    },
    {
      "epoch": 9.441375732421875e-06,
      "model_forward_time": 1.68125319480896,
      "step": 9900
    },
    {
      "dynamics_loss_avg": 0.06941107921302318,
      "epoch": 9.441375732421875e-06,
      "step": 9900
    },
    {
      "action_loss_avg": 0.007597398536745459,
      "epoch": 9.441375732421875e-06,
      "step": 9900
    },
    {
      "epoch": 9.441375732421875e-06,
      "step": 9900,
      "training_step_time": 6.652543067932129
    },
    {
      "epoch": 9.442329406738281e-06,
      "model_forward_time": 1.7297027111053467,
      "step": 9901
    },
    {
      "epoch": 9.442329406738281e-06,
      "step": 9901,
      "training_step_time": 7.886938571929932
    },
    {
      "epoch": 9.443283081054687e-06,
      "model_forward_time": 1.697587251663208,
      "step": 9902
    },
    {
      "epoch": 9.443283081054687e-06,
      "step": 9902,
      "training_step_time": 8.636528253555298
    },
    {
      "epoch": 9.444236755371094e-06,
      "model_forward_time": 2.2504563331604004,
      "step": 9903
    },
    {
      "epoch": 9.444236755371094e-06,
      "step": 9903,
      "training_step_time": 9.252301931381226
    },
    {
      "epoch": 9.4451904296875e-06,
      "model_forward_time": 1.6994657516479492,
      "step": 9904
    },
    {
      "epoch": 9.4451904296875e-06,
      "step": 9904,
      "training_step_time": 8.139748096466064
    },
    {
      "epoch": 9.446144104003906e-06,
      "model_forward_time": 1.7490675449371338,
      "step": 9905
    },
    {
      "epoch": 9.446144104003906e-06,
      "step": 9905,
      "training_step_time": 7.667770862579346
    },
    {
      "epoch": 9.447097778320313e-06,
      "model_forward_time": 1.7055575847625732,
      "step": 9906
    },
    {
      "epoch": 9.447097778320313e-06,
      "step": 9906,
      "training_step_time": 7.60113000869751
    },
    {
      "epoch": 9.448051452636719e-06,
      "model_forward_time": 1.703855276107788,
      "step": 9907
    },
    {
      "epoch": 9.448051452636719e-06,
      "step": 9907,
      "training_step_time": 7.706135988235474
    },
    {
      "epoch": 9.449005126953124e-06,
      "model_forward_time": 1.7097148895263672,
      "step": 9908
    },
    {
      "epoch": 9.449005126953124e-06,
      "step": 9908,
      "training_step_time": 7.808435678482056
    },
    {
      "epoch": 9.449958801269532e-06,
      "model_forward_time": 1.7054026126861572,
      "step": 9909
    },
    {
      "epoch": 9.449958801269532e-06,
      "step": 9909,
      "training_step_time": 6.812691926956177
    },
    {
      "epoch": 9.450912475585938e-06,
      "grad_norm": 0.20192912220954895,
      "learning_rate": 9.93426076703221e-06,
      "loss": 0.0722,
      "step": 9910
    },
    {
      "epoch": 9.450912475585938e-06,
      "model_forward_time": 1.7022318840026855,
      "step": 9910
    },
    {
      "dynamics_loss_avg": 0.04467757809907198,
      "epoch": 9.450912475585938e-06,
      "step": 9910
    },
    {
      "action_loss_avg": 0.004950887535233051,
      "epoch": 9.450912475585938e-06,
      "step": 9910
    },
    {
      "epoch": 9.450912475585938e-06,
      "step": 9910,
      "training_step_time": 7.334951639175415
    },
    {
      "epoch": 9.451866149902343e-06,
      "model_forward_time": 1.708984375,
      "step": 9911
    },
    {
      "epoch": 9.451866149902343e-06,
      "step": 9911,
      "training_step_time": 7.623411417007446
    },
    {
      "epoch": 9.45281982421875e-06,
      "model_forward_time": 2.250706434249878,
      "step": 9912
    },
    {
      "epoch": 9.45281982421875e-06,
      "step": 9912,
      "training_step_time": 8.317854166030884
    },
    {
      "epoch": 9.453773498535156e-06,
      "model_forward_time": 1.7341465950012207,
      "step": 9913
    },
    {
      "epoch": 9.453773498535156e-06,
      "step": 9913,
      "training_step_time": 9.282634496688843
    },
    {
      "epoch": 9.454727172851562e-06,
      "model_forward_time": 1.731722116470337,
      "step": 9914
    },
    {
      "epoch": 9.454727172851562e-06,
      "step": 9914,
      "training_step_time": 8.659112930297852
    },
    {
      "epoch": 9.455680847167968e-06,
      "model_forward_time": 1.702592372894287,
      "step": 9915
    },
    {
      "epoch": 9.455680847167968e-06,
      "step": 9915,
      "training_step_time": 8.588556289672852
    },
    {
      "epoch": 9.456634521484375e-06,
      "model_forward_time": 1.707909107208252,
      "step": 9916
    },
    {
      "epoch": 9.456634521484375e-06,
      "step": 9916,
      "training_step_time": 7.580382823944092
    },
    {
      "epoch": 9.457588195800781e-06,
      "model_forward_time": 1.7050905227661133,
      "step": 9917
    },
    {
      "epoch": 9.457588195800781e-06,
      "step": 9917,
      "training_step_time": 7.844743490219116
    },
    {
      "epoch": 9.458541870117187e-06,
      "model_forward_time": 1.3275518417358398,
      "step": 9918
    },
    {
      "epoch": 9.458541870117187e-06,
      "step": 9918,
      "training_step_time": 6.755362272262573
    },
    {
      "epoch": 9.459495544433594e-06,
      "model_forward_time": 1.7086834907531738,
      "step": 9919
    },
    {
      "epoch": 9.459495544433594e-06,
      "step": 9919,
      "training_step_time": 7.464905261993408
    },
    {
      "epoch": 9.46044921875e-06,
      "grad_norm": 0.18598072230815887,
      "learning_rate": 9.933993254221172e-06,
      "loss": 0.075,
      "step": 9920
    },
    {
      "epoch": 9.46044921875e-06,
      "model_forward_time": 1.7289245128631592,
      "step": 9920
    },
    {
      "dynamics_loss_avg": 0.05849401280283928,
      "epoch": 9.46044921875e-06,
      "step": 9920
    },
    {
      "action_loss_avg": 0.003378273581620306,
      "epoch": 9.46044921875e-06,
      "step": 9920
    },
    {
      "epoch": 9.46044921875e-06,
      "step": 9920,
      "training_step_time": 7.288808345794678
    },
    {
      "epoch": 9.461402893066406e-06,
      "model_forward_time": 1.7004473209381104,
      "step": 9921
    },
    {
      "epoch": 9.461402893066406e-06,
      "step": 9921,
      "training_step_time": 8.563926219940186
    },
    {
      "epoch": 9.462356567382813e-06,
      "model_forward_time": 2.245877981185913,
      "step": 9922
    },
    {
      "epoch": 9.462356567382813e-06,
      "step": 9922,
      "training_step_time": 9.094433069229126
    },
    {
      "epoch": 9.463310241699219e-06,
      "model_forward_time": 1.7034218311309814,
      "step": 9923
    },
    {
      "epoch": 9.463310241699219e-06,
      "step": 9923,
      "training_step_time": 8.441566467285156
    },
    {
      "epoch": 9.464263916015625e-06,
      "model_forward_time": 1.7603743076324463,
      "step": 9924
    },
    {
      "epoch": 9.464263916015625e-06,
      "step": 9924,
      "training_step_time": 7.558586835861206
    },
    {
      "epoch": 9.465217590332032e-06,
      "model_forward_time": 1.7103159427642822,
      "step": 9925
    },
    {
      "epoch": 9.465217590332032e-06,
      "step": 9925,
      "training_step_time": 7.562672138214111
    },
    {
      "epoch": 9.466171264648438e-06,
      "model_forward_time": 1.7120909690856934,
      "step": 9926
    },
    {
      "epoch": 9.466171264648438e-06,
      "step": 9926,
      "training_step_time": 7.755422592163086
    },
    {
      "epoch": 9.467124938964843e-06,
      "model_forward_time": 1.712672233581543,
      "step": 9927
    },
    {
      "epoch": 9.467124938964843e-06,
      "step": 9927,
      "training_step_time": 6.73812198638916
    },
    {
      "epoch": 9.46807861328125e-06,
      "model_forward_time": 1.7060532569885254,
      "step": 9928
    },
    {
      "epoch": 9.46807861328125e-06,
      "step": 9928,
      "training_step_time": 7.444300413131714
    },
    {
      "epoch": 9.469032287597657e-06,
      "model_forward_time": 1.6919145584106445,
      "step": 9929
    },
    {
      "epoch": 9.469032287597657e-06,
      "step": 9929,
      "training_step_time": 7.985663652420044
    },
    {
      "epoch": 9.469985961914062e-06,
      "grad_norm": 0.229180708527565,
      "learning_rate": 9.933725201836053e-06,
      "loss": 0.075,
      "step": 9930
    },
    {
      "epoch": 9.469985961914062e-06,
      "model_forward_time": 1.7061684131622314,
      "step": 9930
    },
    {
      "dynamics_loss_avg": 0.06892237029969692,
      "epoch": 9.469985961914062e-06,
      "step": 9930
    },
    {
      "action_loss_avg": 0.005701458104886115,
      "epoch": 9.469985961914062e-06,
      "step": 9930
    },
    {
      "epoch": 9.469985961914062e-06,
      "step": 9930,
      "training_step_time": 8.50544786453247
    },
    {
      "epoch": 9.470939636230468e-06,
      "model_forward_time": 2.2541584968566895,
      "step": 9931
    },
    {
      "epoch": 9.470939636230468e-06,
      "step": 9931,
      "training_step_time": 8.502159118652344
    },
    {
      "epoch": 9.471893310546875e-06,
      "model_forward_time": 1.7073616981506348,
      "step": 9932
    },
    {
      "epoch": 9.471893310546875e-06,
      "step": 9932,
      "training_step_time": 8.55120062828064
    },
    {
      "epoch": 9.472846984863281e-06,
      "model_forward_time": 1.7400898933410645,
      "step": 9933
    },
    {
      "epoch": 9.472846984863281e-06,
      "step": 9933,
      "training_step_time": 8.087463855743408
    },
    {
      "epoch": 9.473800659179687e-06,
      "model_forward_time": 1.697066307067871,
      "step": 9934
    },
    {
      "epoch": 9.473800659179687e-06,
      "step": 9934,
      "training_step_time": 7.545751094818115
    },
    {
      "epoch": 9.474754333496094e-06,
      "model_forward_time": 1.7149696350097656,
      "step": 9935
    },
    {
      "epoch": 9.474754333496094e-06,
      "step": 9935,
      "training_step_time": 7.637799024581909
    },
    {
      "epoch": 9.4757080078125e-06,
      "model_forward_time": 1.7059953212738037,
      "step": 9936
    },
    {
      "epoch": 9.4757080078125e-06,
      "step": 9936,
      "training_step_time": 8.659335851669312
    },
    {
      "epoch": 9.476661682128906e-06,
      "model_forward_time": 1.7028944492340088,
      "step": 9937
    },
    {
      "epoch": 9.476661682128906e-06,
      "step": 9937,
      "training_step_time": 7.492872714996338
    },
    {
      "epoch": 9.477615356445313e-06,
      "model_forward_time": 1.7105708122253418,
      "step": 9938
    },
    {
      "epoch": 9.477615356445313e-06,
      "step": 9938,
      "training_step_time": 7.782675743103027
    },
    {
      "epoch": 9.478569030761719e-06,
      "model_forward_time": 1.697700023651123,
      "step": 9939
    },
    {
      "epoch": 9.478569030761719e-06,
      "step": 9939,
      "training_step_time": 7.659204959869385
    },
    {
      "epoch": 9.479522705078125e-06,
      "grad_norm": 0.26417475938796997,
      "learning_rate": 9.933456609906162e-06,
      "loss": 0.0728,
      "step": 9940
    },
    {
      "epoch": 9.479522705078125e-06,
      "model_forward_time": 2.220245838165283,
      "step": 9940
    },
    {
      "dynamics_loss_avg": 0.07866208665072918,
      "epoch": 9.479522705078125e-06,
      "step": 9940
    },
    {
      "action_loss_avg": 0.0059800659655593336,
      "epoch": 9.479522705078125e-06,
      "step": 9940
    },
    {
      "epoch": 9.479522705078125e-06,
      "step": 9940,
      "training_step_time": 7.61385440826416
    },
    {
      "epoch": 9.480476379394532e-06,
      "model_forward_time": 1.3328351974487305,
      "step": 9941
    },
    {
      "epoch": 9.480476379394532e-06,
      "step": 9941,
      "training_step_time": 7.577801942825317
    },
    {
      "epoch": 9.481430053710938e-06,
      "model_forward_time": 1.713350772857666,
      "step": 9942
    },
    {
      "epoch": 9.481430053710938e-06,
      "step": 9942,
      "training_step_time": 7.635511875152588
    },
    {
      "epoch": 9.482383728027343e-06,
      "model_forward_time": 1.7062628269195557,
      "step": 9943
    },
    {
      "epoch": 9.482383728027343e-06,
      "step": 9943,
      "training_step_time": 7.5458550453186035
    },
    {
      "epoch": 9.48333740234375e-06,
      "model_forward_time": 1.6953823566436768,
      "step": 9944
    },
    {
      "epoch": 9.48333740234375e-06,
      "step": 9944,
      "training_step_time": 7.683578968048096
    },
    {
      "epoch": 9.484291076660157e-06,
      "model_forward_time": 1.707383394241333,
      "step": 9945
    },
    {
      "epoch": 9.484291076660157e-06,
      "step": 9945,
      "training_step_time": 7.8193628787994385
    },
    {
      "epoch": 9.485244750976562e-06,
      "model_forward_time": 1.7039413452148438,
      "step": 9946
    },
    {
      "epoch": 9.485244750976562e-06,
      "step": 9946,
      "training_step_time": 7.5001795291900635
    },
    {
      "epoch": 9.486198425292968e-06,
      "model_forward_time": 1.7227814197540283,
      "step": 9947
    },
    {
      "epoch": 9.486198425292968e-06,
      "step": 9947,
      "training_step_time": 7.014154434204102
    },
    {
      "epoch": 9.487152099609375e-06,
      "model_forward_time": 1.7079837322235107,
      "step": 9948
    },
    {
      "epoch": 9.487152099609375e-06,
      "step": 9948,
      "training_step_time": 7.61332893371582
    },
    {
      "epoch": 9.488105773925781e-06,
      "model_forward_time": 2.2338922023773193,
      "step": 9949
    },
    {
      "epoch": 9.488105773925781e-06,
      "step": 9949,
      "training_step_time": 8.311975955963135
    },
    {
      "epoch": 9.489059448242187e-06,
      "grad_norm": 0.1379213035106659,
      "learning_rate": 9.933187478460875e-06,
      "loss": 0.0717,
      "step": 9950
    },
    {
      "epoch": 9.489059448242187e-06,
      "model_forward_time": 1.7094218730926514,
      "step": 9950
    },
    {
      "dynamics_loss_avg": 0.07389916367828846,
      "epoch": 9.489059448242187e-06,
      "step": 9950
    },
    {
      "action_loss_avg": 0.005430421454366297,
      "epoch": 9.489059448242187e-06,
      "step": 9950
    },
    {
      "epoch": 9.489059448242187e-06,
      "step": 9950,
      "training_step_time": 8.571291208267212
    },
    {
      "epoch": 9.490013122558594e-06,
      "model_forward_time": 1.707489013671875,
      "step": 9951
    },
    {
      "epoch": 9.490013122558594e-06,
      "step": 9951,
      "training_step_time": 8.648433446884155
    },
    {
      "epoch": 9.490966796875e-06,
      "model_forward_time": 1.7612519264221191,
      "step": 9952
    },
    {
      "epoch": 9.490966796875e-06,
      "step": 9952,
      "training_step_time": 7.583537578582764
    },
    {
      "epoch": 9.491920471191406e-06,
      "model_forward_time": 1.7054743766784668,
      "step": 9953
    },
    {
      "epoch": 9.491920471191406e-06,
      "step": 9953,
      "training_step_time": 7.620095491409302
    },
    {
      "epoch": 9.492874145507813e-06,
      "model_forward_time": 1.7061536312103271,
      "step": 9954
    },
    {
      "epoch": 9.492874145507813e-06,
      "step": 9954,
      "training_step_time": 7.787078857421875
    },
    {
      "epoch": 9.493827819824219e-06,
      "model_forward_time": 1.7142338752746582,
      "step": 9955
    },
    {
      "epoch": 9.493827819824219e-06,
      "step": 9955,
      "training_step_time": 8.45833706855774
    },
    {
      "epoch": 9.494781494140625e-06,
      "model_forward_time": 1.7066941261291504,
      "step": 9956
    },
    {
      "epoch": 9.494781494140625e-06,
      "step": 9956,
      "training_step_time": 7.061672687530518
    },
    {
      "epoch": 9.495735168457032e-06,
      "model_forward_time": 1.7072880268096924,
      "step": 9957
    },
    {
      "epoch": 9.495735168457032e-06,
      "step": 9957,
      "training_step_time": 7.5331337451934814
    },
    {
      "epoch": 9.496688842773438e-06,
      "model_forward_time": 1.6960415840148926,
      "step": 9958
    },
    {
      "epoch": 9.496688842773438e-06,
      "step": 9958,
      "training_step_time": 9.497110605239868
    },
    {
      "epoch": 9.497642517089843e-06,
      "model_forward_time": 2.231539011001587,
      "step": 9959
    },
    {
      "epoch": 9.497642517089843e-06,
      "step": 9959,
      "training_step_time": 7.6279520988464355
    },
    {
      "epoch": 9.49859619140625e-06,
      "grad_norm": 0.20523671805858612,
      "learning_rate": 9.93291780752962e-06,
      "loss": 0.0704,
      "step": 9960
    },
    {
      "epoch": 9.49859619140625e-06,
      "model_forward_time": 1.6984474658966064,
      "step": 9960
    },
    {
      "dynamics_loss_avg": 0.06598925348371268,
      "epoch": 9.49859619140625e-06,
      "step": 9960
    },
    {
      "action_loss_avg": 0.002915569458855316,
      "epoch": 9.49859619140625e-06,
      "step": 9960
    },
    {
      "epoch": 9.49859619140625e-06,
      "step": 9960,
      "training_step_time": 7.657601594924927
    },
    {
      "epoch": 9.499549865722657e-06,
      "model_forward_time": 1.7320761680603027,
      "step": 9961
    },
    {
      "epoch": 9.499549865722657e-06,
      "step": 9961,
      "training_step_time": 8.601919889450073
    },
    {
      "epoch": 9.500503540039062e-06,
      "model_forward_time": 1.6932296752929688,
      "step": 9962
    },
    {
      "epoch": 9.500503540039062e-06,
      "step": 9962,
      "training_step_time": 7.563863515853882
    },
    {
      "epoch": 9.501457214355468e-06,
      "model_forward_time": 1.7128617763519287,
      "step": 9963
    },
    {
      "epoch": 9.501457214355468e-06,
      "step": 9963,
      "training_step_time": 6.951041221618652
    },
    {
      "epoch": 9.502410888671875e-06,
      "model_forward_time": 1.6926555633544922,
      "step": 9964
    },
    {
      "epoch": 9.502410888671875e-06,
      "step": 9964,
      "training_step_time": 9.140208959579468
    },
    {
      "epoch": 9.503364562988281e-06,
      "model_forward_time": 1.7020528316497803,
      "step": 9965
    },
    {
      "epoch": 9.503364562988281e-06,
      "step": 9965,
      "training_step_time": 7.388970613479614
    },
    {
      "epoch": 9.504318237304687e-06,
      "model_forward_time": 1.7049808502197266,
      "step": 9966
    },
    {
      "epoch": 9.504318237304687e-06,
      "step": 9966,
      "training_step_time": 7.517016172409058
    },
    {
      "epoch": 9.505271911621094e-06,
      "model_forward_time": 1.7004687786102295,
      "step": 9967
    },
    {
      "epoch": 9.505271911621094e-06,
      "step": 9967,
      "training_step_time": 7.599404335021973
    },
    {
      "epoch": 9.5062255859375e-06,
      "model_forward_time": 2.2106807231903076,
      "step": 9968
    },
    {
      "epoch": 9.5062255859375e-06,
      "step": 9968,
      "training_step_time": 7.572264194488525
    },
    {
      "epoch": 9.507179260253906e-06,
      "model_forward_time": 1.7293903827667236,
      "step": 9969
    },
    {
      "epoch": 9.507179260253906e-06,
      "step": 9969,
      "training_step_time": 8.4749174118042
    },
    {
      "epoch": 9.508132934570313e-06,
      "grad_norm": 0.17305445671081543,
      "learning_rate": 9.932647597141893e-06,
      "loss": 0.0672,
      "step": 9970
    },
    {
      "epoch": 9.508132934570313e-06,
      "model_forward_time": 1.7287776470184326,
      "step": 9970
    },
    {
      "dynamics_loss_avg": 0.05825639236718416,
      "epoch": 9.508132934570313e-06,
      "step": 9970
    },
    {
      "action_loss_avg": 0.005098388413898647,
      "epoch": 9.508132934570313e-06,
      "step": 9970
    },
    {
      "epoch": 9.508132934570313e-06,
      "step": 9970,
      "training_step_time": 7.651947259902954
    },
    {
      "epoch": 9.509086608886719e-06,
      "model_forward_time": 1.7084193229675293,
      "step": 9971
    },
    {
      "epoch": 9.509086608886719e-06,
      "step": 9971,
      "training_step_time": 7.581809997558594
    },
    {
      "epoch": 9.510040283203125e-06,
      "model_forward_time": 1.3257851600646973,
      "step": 9972
    },
    {
      "epoch": 9.510040283203125e-06,
      "step": 9972,
      "training_step_time": 6.902634382247925
    },
    {
      "epoch": 9.510993957519532e-06,
      "model_forward_time": 1.7081358432769775,
      "step": 9973
    },
    {
      "epoch": 9.510993957519532e-06,
      "step": 9973,
      "training_step_time": 7.146081209182739
    },
    {
      "epoch": 9.511947631835938e-06,
      "model_forward_time": 0.9819936752319336,
      "step": 9974
    },
    {
      "epoch": 9.511947631835938e-06,
      "step": 9974,
      "training_step_time": 7.495941400527954
    },
    {
      "epoch": 9.512901306152343e-06,
      "model_forward_time": 1.706547737121582,
      "step": 9975
    },
    {
      "epoch": 9.512901306152343e-06,
      "step": 9975,
      "training_step_time": 7.489894390106201
    },
    {
      "epoch": 9.51385498046875e-06,
      "model_forward_time": 1.7142126560211182,
      "step": 9976
    },
    {
      "epoch": 9.51385498046875e-06,
      "step": 9976,
      "training_step_time": 7.566909074783325
    },
    {
      "epoch": 9.514808654785157e-06,
      "model_forward_time": 2.2351601123809814,
      "step": 9977
    },
    {
      "epoch": 9.514808654785157e-06,
      "step": 9977,
      "training_step_time": 9.61430311203003
    },
    {
      "epoch": 9.515762329101562e-06,
      "model_forward_time": 1.7133574485778809,
      "step": 9978
    },
    {
      "epoch": 9.515762329101562e-06,
      "step": 9978,
      "training_step_time": 8.603103637695312
    },
    {
      "epoch": 9.516716003417968e-06,
      "model_forward_time": 1.7328994274139404,
      "step": 9979
    },
    {
      "epoch": 9.516716003417968e-06,
      "step": 9979,
      "training_step_time": 7.68615198135376
    },
    {
      "epoch": 9.517669677734375e-06,
      "grad_norm": 0.2690790295600891,
      "learning_rate": 9.932376847327238e-06,
      "loss": 0.0674,
      "step": 9980
    },
    {
      "epoch": 9.517669677734375e-06,
      "model_forward_time": 1.3232576847076416,
      "step": 9980
    },
    {
      "dynamics_loss_avg": 0.05595198292285204,
      "epoch": 9.517669677734375e-06,
      "step": 9980
    },
    {
      "action_loss_avg": 0.005034424911718816,
      "epoch": 9.517669677734375e-06,
      "step": 9980
    },
    {
      "epoch": 9.517669677734375e-06,
      "step": 9980,
      "training_step_time": 7.649726629257202
    },
    {
      "epoch": 9.518623352050781e-06,
      "model_forward_time": 1.7130718231201172,
      "step": 9981
    },
    {
      "epoch": 9.518623352050781e-06,
      "step": 9981,
      "training_step_time": 7.789125919342041
    },
    {
      "epoch": 9.519577026367187e-06,
      "model_forward_time": 1.701232671737671,
      "step": 9982
    },
    {
      "epoch": 9.519577026367187e-06,
      "step": 9982,
      "training_step_time": 7.634460687637329
    },
    {
      "epoch": 9.520530700683594e-06,
      "model_forward_time": 1.704277753829956,
      "step": 9983
    },
    {
      "epoch": 9.520530700683594e-06,
      "step": 9983,
      "training_step_time": 7.502963066101074
    },
    {
      "epoch": 9.521484375e-06,
      "model_forward_time": 1.703749656677246,
      "step": 9984
    },
    {
      "epoch": 9.521484375e-06,
      "step": 9984,
      "training_step_time": 7.63080096244812
    },
    {
      "epoch": 9.522438049316406e-06,
      "model_forward_time": 1.702451467514038,
      "step": 9985
    },
    {
      "epoch": 9.522438049316406e-06,
      "step": 9985,
      "training_step_time": 7.604055881500244
    },
    {
      "epoch": 9.523391723632813e-06,
      "model_forward_time": 1.701972246170044,
      "step": 9986
    },
    {
      "epoch": 9.523391723632813e-06,
      "step": 9986,
      "training_step_time": 8.835691928863525
    },
    {
      "epoch": 9.524345397949219e-06,
      "model_forward_time": 1.7021114826202393,
      "step": 9987
    },
    {
      "epoch": 9.524345397949219e-06,
      "step": 9987,
      "training_step_time": 9.564436674118042
    },
    {
      "epoch": 9.525299072265625e-06,
      "model_forward_time": 1.7102162837982178,
      "step": 9988
    },
    {
      "epoch": 9.525299072265625e-06,
      "step": 9988,
      "training_step_time": 8.131865501403809
    },
    {
      "epoch": 9.526252746582032e-06,
      "model_forward_time": 1.0164752006530762,
      "step": 9989
    },
    {
      "epoch": 9.526252746582032e-06,
      "step": 9989,
      "training_step_time": 7.667959213256836
    },
    {
      "epoch": 9.527206420898438e-06,
      "grad_norm": 0.16579647362232208,
      "learning_rate": 9.932105558115269e-06,
      "loss": 0.0698,
      "step": 9990
    },
    {
      "epoch": 9.527206420898438e-06,
      "model_forward_time": 1.6995787620544434,
      "step": 9990
    },
    {
      "dynamics_loss_avg": 0.054523216746747497,
      "epoch": 9.527206420898438e-06,
      "step": 9990
    },
    {
      "action_loss_avg": 0.0038409703876823185,
      "epoch": 9.527206420898438e-06,
      "step": 9990
    },
    {
      "epoch": 9.527206420898438e-06,
      "step": 9990,
      "training_step_time": 7.742352724075317
    },
    {
      "epoch": 9.528160095214843e-06,
      "model_forward_time": 1.7125859260559082,
      "step": 9991
    },
    {
      "epoch": 9.528160095214843e-06,
      "step": 9991,
      "training_step_time": 7.724223375320435
    },
    {
      "epoch": 9.52911376953125e-06,
      "model_forward_time": 1.7062525749206543,
      "step": 9992
    },
    {
      "epoch": 9.52911376953125e-06,
      "step": 9992,
      "training_step_time": 7.517220497131348
    },
    {
      "epoch": 9.530067443847657e-06,
      "model_forward_time": 1.707782506942749,
      "step": 9993
    },
    {
      "epoch": 9.530067443847657e-06,
      "step": 9993,
      "training_step_time": 7.6124351024627686
    },
    {
      "epoch": 9.531021118164062e-06,
      "model_forward_time": 1.714808464050293,
      "step": 9994
    },
    {
      "epoch": 9.531021118164062e-06,
      "step": 9994,
      "training_step_time": 8.121631622314453
    },
    {
      "epoch": 9.531974792480468e-06,
      "model_forward_time": 1.7025954723358154,
      "step": 9995
    },
    {
      "epoch": 9.531974792480468e-06,
      "step": 9995,
      "training_step_time": 7.66295051574707
    },
    {
      "epoch": 9.532928466796875e-06,
      "model_forward_time": 2.237452745437622,
      "step": 9996
    },
    {
      "epoch": 9.532928466796875e-06,
      "step": 9996,
      "training_step_time": 10.90077018737793
    },
    {
      "epoch": 9.533882141113281e-06,
      "model_forward_time": 1.7087898254394531,
      "step": 9997
    },
    {
      "epoch": 9.533882141113281e-06,
      "step": 9997,
      "training_step_time": 8.313976764678955
    },
    {
      "epoch": 9.534835815429687e-06,
      "model_forward_time": 1.7760212421417236,
      "step": 9998
    },
    {
      "epoch": 9.534835815429687e-06,
      "step": 9998,
      "training_step_time": 9.662659406661987
    },
    {
      "epoch": 9.535789489746094e-06,
      "model_forward_time": 1.7120161056518555,
      "step": 9999
    },
    {
      "epoch": 9.535789489746094e-06,
      "step": 9999,
      "training_step_time": 7.584027051925659
    },
    {
      "epoch": 9.5367431640625e-06,
      "grad_norm": 0.23312433063983917,
      "learning_rate": 9.931833729535651e-06,
      "loss": 0.0731,
      "step": 10000
    },
    {
      "epoch": 9.5367431640625e-06,
      "model_forward_time": 1.7349872589111328,
      "step": 10000
    },
    {
      "dynamics_loss_avg": 0.05932035706937313,
      "epoch": 9.5367431640625e-06,
      "step": 10000
    },
    {
      "action_loss_avg": 0.0032849934068508446,
      "epoch": 9.5367431640625e-06,
      "step": 10000
    },
    {
      "epoch": 9.5367431640625e-06,
      "step": 10000,
      "training_step_time": 6.447371006011963
    },
    {
      "epoch": 9.537696838378906e-06,
      "model_forward_time": 1.7098581790924072,
      "step": 10001
    },
    {
      "epoch": 9.537696838378906e-06,
      "step": 10001,
      "training_step_time": 8.616878747940063
    },
    {
      "epoch": 9.538650512695313e-06,
      "model_forward_time": 1.7167441844940186,
      "step": 10002
    },
    {
      "epoch": 9.538650512695313e-06,
      "step": 10002,
      "training_step_time": 8.444275856018066
    },
    {
      "epoch": 9.539604187011719e-06,
      "model_forward_time": 1.7158441543579102,
      "step": 10003
    },
    {
      "epoch": 9.539604187011719e-06,
      "step": 10003,
      "training_step_time": 8.557658672332764
    },
    {
      "epoch": 9.540557861328125e-06,
      "model_forward_time": 2.7007181644439697,
      "step": 10004
    },
    {
      "epoch": 9.540557861328125e-06,
      "step": 10004,
      "training_step_time": 9.445831537246704
    },
    {
      "epoch": 9.541511535644532e-06,
      "model_forward_time": 1.7088654041290283,
      "step": 10005
    },
    {
      "epoch": 9.541511535644532e-06,
      "step": 10005,
      "training_step_time": 8.440783023834229
    },
    {
      "epoch": 9.542465209960938e-06,
      "model_forward_time": 1.7146823406219482,
      "step": 10006
    },
    {
      "epoch": 9.542465209960938e-06,
      "step": 10006,
      "training_step_time": 9.51405930519104
    },
    {
      "epoch": 9.543418884277343e-06,
      "model_forward_time": 1.7631101608276367,
      "step": 10007
    },
    {
      "epoch": 9.543418884277343e-06,
      "step": 10007,
      "training_step_time": 7.6974592208862305
    },
    {
      "epoch": 9.54437255859375e-06,
      "model_forward_time": 1.3371155261993408,
      "step": 10008
    },
    {
      "epoch": 9.54437255859375e-06,
      "step": 10008,
      "training_step_time": 7.85680365562439
    },
    {
      "epoch": 9.545326232910157e-06,
      "model_forward_time": 1.722299337387085,
      "step": 10009
    },
    {
      "epoch": 9.545326232910157e-06,
      "step": 10009,
      "training_step_time": 7.7533769607543945
    },
    {
      "epoch": 9.546279907226562e-06,
      "grad_norm": 0.25685834884643555,
      "learning_rate": 9.931561361618111e-06,
      "loss": 0.0749,
      "step": 10010
    },
    {
      "epoch": 9.546279907226562e-06,
      "model_forward_time": 1.708202600479126,
      "step": 10010
    },
    {
      "dynamics_loss_avg": 0.05523245446383953,
      "epoch": 9.546279907226562e-06,
      "step": 10010
    },
    {
      "action_loss_avg": 0.0030957153183408083,
      "epoch": 9.546279907226562e-06,
      "step": 10010
    },
    {
      "epoch": 9.546279907226562e-06,
      "step": 10010,
      "training_step_time": 7.850902795791626
    },
    {
      "epoch": 9.547233581542968e-06,
      "model_forward_time": 1.716198205947876,
      "step": 10011
    },
    {
      "epoch": 9.547233581542968e-06,
      "step": 10011,
      "training_step_time": 7.70529580116272
    },
    {
      "epoch": 9.548187255859375e-06,
      "model_forward_time": 1.7145676612854004,
      "step": 10012
    },
    {
      "epoch": 9.548187255859375e-06,
      "step": 10012,
      "training_step_time": 7.56036639213562
    },
    {
      "epoch": 9.549140930175781e-06,
      "model_forward_time": 1.7089722156524658,
      "step": 10013
    },
    {
      "epoch": 9.549140930175781e-06,
      "step": 10013,
      "training_step_time": 8.631921529769897
    },
    {
      "epoch": 9.550094604492187e-06,
      "model_forward_time": 2.2479212284088135,
      "step": 10014
    },
    {
      "epoch": 9.550094604492187e-06,
      "step": 10014,
      "training_step_time": 7.8759446144104
    },
    {
      "epoch": 9.551048278808594e-06,
      "model_forward_time": 1.3412346839904785,
      "step": 10015
    },
    {
      "epoch": 9.551048278808594e-06,
      "step": 10015,
      "training_step_time": 9.357764720916748
    },
    {
      "epoch": 9.552001953125e-06,
      "model_forward_time": 1.737635612487793,
      "step": 10016
    },
    {
      "epoch": 9.552001953125e-06,
      "step": 10016,
      "training_step_time": 8.609310150146484
    },
    {
      "epoch": 9.552955627441406e-06,
      "model_forward_time": 1.7201378345489502,
      "step": 10017
    },
    {
      "epoch": 9.552955627441406e-06,
      "step": 10017,
      "training_step_time": 7.7757556438446045
    },
    {
      "epoch": 9.553909301757813e-06,
      "model_forward_time": 1.7090661525726318,
      "step": 10018
    },
    {
      "epoch": 9.553909301757813e-06,
      "step": 10018,
      "training_step_time": 7.772814989089966
    },
    {
      "epoch": 9.554862976074219e-06,
      "model_forward_time": 1.715052843093872,
      "step": 10019
    },
    {
      "epoch": 9.554862976074219e-06,
      "step": 10019,
      "training_step_time": 7.7871198654174805
    },
    {
      "epoch": 9.555816650390625e-06,
      "grad_norm": 0.2712886333465576,
      "learning_rate": 9.931288454392435e-06,
      "loss": 0.0726,
      "step": 10020
    },
    {
      "epoch": 9.555816650390625e-06,
      "model_forward_time": 1.7084262371063232,
      "step": 10020
    },
    {
      "dynamics_loss_avg": 0.07771665528416634,
      "epoch": 9.555816650390625e-06,
      "step": 10020
    },
    {
      "action_loss_avg": 0.005454201344400644,
      "epoch": 9.555816650390625e-06,
      "step": 10020
    },
    {
      "epoch": 9.555816650390625e-06,
      "step": 10020,
      "training_step_time": 7.747356176376343
    },
    {
      "epoch": 9.556770324707032e-06,
      "model_forward_time": 1.708874225616455,
      "step": 10021
    },
    {
      "epoch": 9.556770324707032e-06,
      "step": 10021,
      "training_step_time": 7.4381492137908936
    },
    {
      "epoch": 9.557723999023438e-06,
      "model_forward_time": 1.703026533126831,
      "step": 10022
    },
    {
      "epoch": 9.557723999023438e-06,
      "step": 10022,
      "training_step_time": 7.791524648666382
    },
    {
      "epoch": 9.558677673339843e-06,
      "model_forward_time": 2.2478396892547607,
      "step": 10023
    },
    {
      "epoch": 9.558677673339843e-06,
      "step": 10023,
      "training_step_time": 8.585744619369507
    },
    {
      "epoch": 9.55963134765625e-06,
      "model_forward_time": 1.7144899368286133,
      "step": 10024
    },
    {
      "epoch": 9.55963134765625e-06,
      "step": 10024,
      "training_step_time": 8.879329681396484
    },
    {
      "epoch": 9.560585021972657e-06,
      "model_forward_time": 1.7183783054351807,
      "step": 10025
    },
    {
      "epoch": 9.560585021972657e-06,
      "step": 10025,
      "training_step_time": 8.284372568130493
    },
    {
      "epoch": 9.561538696289062e-06,
      "model_forward_time": 1.712050199508667,
      "step": 10026
    },
    {
      "epoch": 9.561538696289062e-06,
      "step": 10026,
      "training_step_time": 7.618624448776245
    },
    {
      "epoch": 9.562492370605468e-06,
      "model_forward_time": 1.7156927585601807,
      "step": 10027
    },
    {
      "epoch": 9.562492370605468e-06,
      "step": 10027,
      "training_step_time": 7.711848497390747
    },
    {
      "epoch": 9.563446044921875e-06,
      "model_forward_time": 1.7170543670654297,
      "step": 10028
    },
    {
      "epoch": 9.563446044921875e-06,
      "step": 10028,
      "training_step_time": 7.268054962158203
    },
    {
      "epoch": 9.564399719238281e-06,
      "model_forward_time": 1.7098569869995117,
      "step": 10029
    },
    {
      "epoch": 9.564399719238281e-06,
      "step": 10029,
      "training_step_time": 7.610633134841919
    },
    {
      "epoch": 9.565353393554687e-06,
      "grad_norm": 0.33962541818618774,
      "learning_rate": 9.931015007888467e-06,
      "loss": 0.0733,
      "step": 10030
    },
    {
      "epoch": 9.565353393554687e-06,
      "model_forward_time": 1.7110555171966553,
      "step": 10030
    },
    {
      "dynamics_loss_avg": 0.05193641968071461,
      "epoch": 9.565353393554687e-06,
      "step": 10030
    },
    {
      "action_loss_avg": 0.0038746851612813773,
      "epoch": 9.565353393554687e-06,
      "step": 10030
    },
    {
      "epoch": 9.565353393554687e-06,
      "step": 10030,
      "training_step_time": 8.336757898330688
    },
    {
      "epoch": 9.566307067871094e-06,
      "model_forward_time": 1.7099545001983643,
      "step": 10031
    },
    {
      "epoch": 9.566307067871094e-06,
      "step": 10031,
      "training_step_time": 7.666097640991211
    },
    {
      "epoch": 9.5672607421875e-06,
      "model_forward_time": 2.278296947479248,
      "step": 10032
    },
    {
      "epoch": 9.5672607421875e-06,
      "step": 10032,
      "training_step_time": 8.743009805679321
    },
    {
      "epoch": 9.568214416503906e-06,
      "model_forward_time": 1.708786964416504,
      "step": 10033
    },
    {
      "epoch": 9.568214416503906e-06,
      "step": 10033,
      "training_step_time": 9.680604696273804
    },
    {
      "epoch": 9.569168090820313e-06,
      "model_forward_time": 1.7000939846038818,
      "step": 10034
    },
    {
      "epoch": 9.569168090820313e-06,
      "step": 10034,
      "training_step_time": 8.320658683776855
    },
    {
      "epoch": 9.570121765136719e-06,
      "model_forward_time": 1.7147831916809082,
      "step": 10035
    },
    {
      "epoch": 9.570121765136719e-06,
      "step": 10035,
      "training_step_time": 8.176172494888306
    },
    {
      "epoch": 9.571075439453125e-06,
      "model_forward_time": 1.7135958671569824,
      "step": 10036
    },
    {
      "epoch": 9.571075439453125e-06,
      "step": 10036,
      "training_step_time": 7.69975471496582
    },
    {
      "epoch": 9.572029113769532e-06,
      "model_forward_time": 1.7117431163787842,
      "step": 10037
    },
    {
      "epoch": 9.572029113769532e-06,
      "step": 10037,
      "training_step_time": 7.188740253448486
    },
    {
      "epoch": 9.572982788085938e-06,
      "model_forward_time": 1.7160885334014893,
      "step": 10038
    },
    {
      "epoch": 9.572982788085938e-06,
      "step": 10038,
      "training_step_time": 7.582369804382324
    },
    {
      "epoch": 9.573936462402343e-06,
      "model_forward_time": 1.723067283630371,
      "step": 10039
    },
    {
      "epoch": 9.573936462402343e-06,
      "step": 10039,
      "training_step_time": 7.521775245666504
    },
    {
      "epoch": 9.57489013671875e-06,
      "grad_norm": 0.46082553267478943,
      "learning_rate": 9.930741022136112e-06,
      "loss": 0.0721,
      "step": 10040
    },
    {
      "epoch": 9.57489013671875e-06,
      "model_forward_time": 1.7019970417022705,
      "step": 10040
    },
    {
      "dynamics_loss_avg": 0.07265989072620868,
      "epoch": 9.57489013671875e-06,
      "step": 10040
    },
    {
      "action_loss_avg": 0.004548168822657317,
      "epoch": 9.57489013671875e-06,
      "step": 10040
    },
    {
      "epoch": 9.57489013671875e-06,
      "step": 10040,
      "training_step_time": 8.114009380340576
    },
    {
      "epoch": 9.575843811035157e-06,
      "model_forward_time": 2.2484946250915527,
      "step": 10041
    },
    {
      "epoch": 9.575843811035157e-06,
      "step": 10041,
      "training_step_time": 8.446434259414673
    },
    {
      "epoch": 9.576797485351562e-06,
      "model_forward_time": 1.7165617942810059,
      "step": 10042
    },
    {
      "epoch": 9.576797485351562e-06,
      "step": 10042,
      "training_step_time": 7.635630369186401
    },
    {
      "epoch": 9.577751159667968e-06,
      "model_forward_time": 1.7109038829803467,
      "step": 10043
    },
    {
      "epoch": 9.577751159667968e-06,
      "step": 10043,
      "training_step_time": 8.197783946990967
    },
    {
      "epoch": 9.578704833984375e-06,
      "model_forward_time": 1.7128477096557617,
      "step": 10044
    },
    {
      "epoch": 9.578704833984375e-06,
      "step": 10044,
      "training_step_time": 7.615805625915527
    },
    {
      "epoch": 9.579658508300781e-06,
      "model_forward_time": 1.7137019634246826,
      "step": 10045
    },
    {
      "epoch": 9.579658508300781e-06,
      "step": 10045,
      "training_step_time": 7.566736221313477
    },
    {
      "epoch": 9.580612182617187e-06,
      "model_forward_time": 1.7095434665679932,
      "step": 10046
    },
    {
      "epoch": 9.580612182617187e-06,
      "step": 10046,
      "training_step_time": 7.141760349273682
    },
    {
      "epoch": 9.581565856933594e-06,
      "model_forward_time": 1.7149665355682373,
      "step": 10047
    },
    {
      "epoch": 9.581565856933594e-06,
      "step": 10047,
      "training_step_time": 7.581922769546509
    },
    {
      "epoch": 9.58251953125e-06,
      "model_forward_time": 1.7130539417266846,
      "step": 10048
    },
    {
      "epoch": 9.58251953125e-06,
      "step": 10048,
      "training_step_time": 7.502030372619629
    },
    {
      "epoch": 9.583473205566406e-06,
      "model_forward_time": 1.7090742588043213,
      "step": 10049
    },
    {
      "epoch": 9.583473205566406e-06,
      "step": 10049,
      "training_step_time": 7.589552879333496
    },
    {
      "epoch": 9.584426879882813e-06,
      "grad_norm": 0.29792436957359314,
      "learning_rate": 9.930466497165332e-06,
      "loss": 0.0694,
      "step": 10050
    },
    {
      "epoch": 9.584426879882813e-06,
      "model_forward_time": 1.7164013385772705,
      "step": 10050
    },
    {
      "dynamics_loss_avg": 0.07501439414918423,
      "epoch": 9.584426879882813e-06,
      "step": 10050
    },
    {
      "action_loss_avg": 0.004559397289995104,
      "epoch": 9.584426879882813e-06,
      "step": 10050
    },
    {
      "epoch": 9.584426879882813e-06,
      "step": 10050,
      "training_step_time": 8.660995960235596
    },
    {
      "epoch": 9.585380554199219e-06,
      "model_forward_time": 2.257202625274658,
      "step": 10051
    },
    {
      "epoch": 9.585380554199219e-06,
      "step": 10051,
      "training_step_time": 10.35523271560669
    },
    {
      "epoch": 9.586334228515625e-06,
      "model_forward_time": 1.709784984588623,
      "step": 10052
    },
    {
      "epoch": 9.586334228515625e-06,
      "step": 10052,
      "training_step_time": 7.641066551208496
    },
    {
      "epoch": 9.587287902832032e-06,
      "model_forward_time": 1.760131597518921,
      "step": 10053
    },
    {
      "epoch": 9.587287902832032e-06,
      "step": 10053,
      "training_step_time": 8.68014144897461
    },
    {
      "epoch": 9.588241577148438e-06,
      "model_forward_time": 1.7134127616882324,
      "step": 10054
    },
    {
      "epoch": 9.588241577148438e-06,
      "step": 10054,
      "training_step_time": 7.584643125534058
    },
    {
      "epoch": 9.589195251464843e-06,
      "model_forward_time": 1.7141296863555908,
      "step": 10055
    },
    {
      "epoch": 9.589195251464843e-06,
      "step": 10055,
      "training_step_time": 7.013144254684448
    },
    {
      "epoch": 9.59014892578125e-06,
      "model_forward_time": 1.7135448455810547,
      "step": 10056
    },
    {
      "epoch": 9.59014892578125e-06,
      "step": 10056,
      "training_step_time": 7.534857749938965
    },
    {
      "epoch": 9.591102600097657e-06,
      "model_forward_time": 1.7149434089660645,
      "step": 10057
    },
    {
      "epoch": 9.591102600097657e-06,
      "step": 10057,
      "training_step_time": 6.635253190994263
    },
    {
      "epoch": 9.592056274414062e-06,
      "model_forward_time": 1.7164955139160156,
      "step": 10058
    },
    {
      "epoch": 9.592056274414062e-06,
      "step": 10058,
      "training_step_time": 7.5842390060424805
    },
    {
      "epoch": 9.593009948730468e-06,
      "model_forward_time": 1.709449052810669,
      "step": 10059
    },
    {
      "epoch": 9.593009948730468e-06,
      "step": 10059,
      "training_step_time": 8.431182384490967
    },
    {
      "epoch": 9.593963623046875e-06,
      "grad_norm": 0.39088010787963867,
      "learning_rate": 9.93019143300615e-06,
      "loss": 0.0697,
      "step": 10060
    },
    {
      "epoch": 9.593963623046875e-06,
      "model_forward_time": 2.2399330139160156,
      "step": 10060
    },
    {
      "dynamics_loss_avg": 0.051546637900173664,
      "epoch": 9.593963623046875e-06,
      "step": 10060
    },
    {
      "action_loss_avg": 0.0061960866209119555,
      "epoch": 9.593963623046875e-06,
      "step": 10060
    },
    {
      "epoch": 9.593963623046875e-06,
      "step": 10060,
      "training_step_time": 8.674935817718506
    },
    {
      "epoch": 9.594917297363281e-06,
      "model_forward_time": 1.7153570652008057,
      "step": 10061
    },
    {
      "epoch": 9.594917297363281e-06,
      "step": 10061,
      "training_step_time": 8.144116878509521
    },
    {
      "epoch": 9.595870971679687e-06,
      "model_forward_time": 1.756258487701416,
      "step": 10062
    },
    {
      "epoch": 9.595870971679687e-06,
      "step": 10062,
      "training_step_time": 9.332474946975708
    },
    {
      "epoch": 9.596824645996094e-06,
      "model_forward_time": 1.7087604999542236,
      "step": 10063
    },
    {
      "epoch": 9.596824645996094e-06,
      "step": 10063,
      "training_step_time": 7.562470197677612
    },
    {
      "epoch": 9.5977783203125e-06,
      "model_forward_time": 1.712778091430664,
      "step": 10064
    },
    {
      "epoch": 9.5977783203125e-06,
      "step": 10064,
      "training_step_time": 7.566426038742065
    },
    {
      "epoch": 9.598731994628906e-06,
      "model_forward_time": 1.6843855381011963,
      "step": 10065
    },
    {
      "epoch": 9.598731994628906e-06,
      "step": 10065,
      "training_step_time": 7.502424001693726
    },
    {
      "epoch": 9.599685668945313e-06,
      "model_forward_time": 1.6964993476867676,
      "step": 10066
    },
    {
      "epoch": 9.599685668945313e-06,
      "step": 10066,
      "training_step_time": 7.538935899734497
    },
    {
      "epoch": 9.600639343261719e-06,
      "model_forward_time": 1.7050127983093262,
      "step": 10067
    },
    {
      "epoch": 9.600639343261719e-06,
      "step": 10067,
      "training_step_time": 7.585946083068848
    },
    {
      "epoch": 9.601593017578125e-06,
      "model_forward_time": 1.6911625862121582,
      "step": 10068
    },
    {
      "epoch": 9.601593017578125e-06,
      "step": 10068,
      "training_step_time": 8.330376625061035
    },
    {
      "epoch": 9.602546691894532e-06,
      "model_forward_time": 2.264817714691162,
      "step": 10069
    },
    {
      "epoch": 9.602546691894532e-06,
      "step": 10069,
      "training_step_time": 8.509196043014526
    },
    {
      "epoch": 9.603500366210938e-06,
      "grad_norm": 0.23405642807483673,
      "learning_rate": 9.929915829688644e-06,
      "loss": 0.071,
      "step": 10070
    },
    {
      "epoch": 9.603500366210938e-06,
      "model_forward_time": 1.6930201053619385,
      "step": 10070
    },
    {
      "dynamics_loss_avg": 0.07006201669573783,
      "epoch": 9.603500366210938e-06,
      "step": 10070
    },
    {
      "action_loss_avg": 0.010456004668958485,
      "epoch": 9.603500366210938e-06,
      "step": 10070
    },
    {
      "epoch": 9.603500366210938e-06,
      "step": 10070,
      "training_step_time": 8.107246398925781
    },
    {
      "epoch": 9.604454040527343e-06,
      "model_forward_time": 1.7047150135040283,
      "step": 10071
    },
    {
      "epoch": 9.604454040527343e-06,
      "step": 10071,
      "training_step_time": 8.877787828445435
    },
    {
      "epoch": 9.60540771484375e-06,
      "model_forward_time": 1.6882922649383545,
      "step": 10072
    },
    {
      "epoch": 9.60540771484375e-06,
      "step": 10072,
      "training_step_time": 8.463900804519653
    },
    {
      "epoch": 9.606361389160157e-06,
      "model_forward_time": 1.7010343074798584,
      "step": 10073
    },
    {
      "epoch": 9.606361389160157e-06,
      "step": 10073,
      "training_step_time": 7.600773334503174
    },
    {
      "epoch": 9.607315063476562e-06,
      "model_forward_time": 1.3217434883117676,
      "step": 10074
    },
    {
      "epoch": 9.607315063476562e-06,
      "step": 10074,
      "training_step_time": 7.5453712940216064
    },
    {
      "epoch": 9.608268737792968e-06,
      "model_forward_time": 1.7010645866394043,
      "step": 10075
    },
    {
      "epoch": 9.608268737792968e-06,
      "step": 10075,
      "training_step_time": 7.582143068313599
    },
    {
      "epoch": 9.609222412109375e-06,
      "model_forward_time": 1.6948668956756592,
      "step": 10076
    },
    {
      "epoch": 9.609222412109375e-06,
      "step": 10076,
      "training_step_time": 7.639675140380859
    },
    {
      "epoch": 9.610176086425781e-06,
      "model_forward_time": 1.6972627639770508,
      "step": 10077
    },
    {
      "epoch": 9.610176086425781e-06,
      "step": 10077,
      "training_step_time": 7.548159122467041
    },
    {
      "epoch": 9.611129760742187e-06,
      "model_forward_time": 2.2453205585479736,
      "step": 10078
    },
    {
      "epoch": 9.611129760742187e-06,
      "step": 10078,
      "training_step_time": 8.605020761489868
    },
    {
      "epoch": 9.612083435058594e-06,
      "model_forward_time": 1.6985077857971191,
      "step": 10079
    },
    {
      "epoch": 9.612083435058594e-06,
      "step": 10079,
      "training_step_time": 7.65238618850708
    },
    {
      "epoch": 9.613037109375e-06,
      "grad_norm": 0.32522523403167725,
      "learning_rate": 9.929639687242956e-06,
      "loss": 0.0734,
      "step": 10080
    },
    {
      "epoch": 9.613037109375e-06,
      "model_forward_time": 1.6920483112335205,
      "step": 10080
    },
    {
      "dynamics_loss_avg": 0.049465746991336344,
      "epoch": 9.613037109375e-06,
      "step": 10080
    },
    {
      "action_loss_avg": 0.005439933307934552,
      "epoch": 9.613037109375e-06,
      "step": 10080
    },
    {
      "epoch": 9.613037109375e-06,
      "step": 10080,
      "training_step_time": 8.36620283126831
    },
    {
      "epoch": 9.613990783691406e-06,
      "model_forward_time": 1.7462255954742432,
      "step": 10081
    },
    {
      "epoch": 9.613990783691406e-06,
      "step": 10081,
      "training_step_time": 7.588781118392944
    },
    {
      "epoch": 9.614944458007813e-06,
      "model_forward_time": 1.7003350257873535,
      "step": 10082
    },
    {
      "epoch": 9.614944458007813e-06,
      "step": 10082,
      "training_step_time": 7.6008687019348145
    },
    {
      "epoch": 9.615898132324219e-06,
      "model_forward_time": 1.698753833770752,
      "step": 10083
    },
    {
      "epoch": 9.615898132324219e-06,
      "step": 10083,
      "training_step_time": 7.545020818710327
    },
    {
      "epoch": 9.616851806640625e-06,
      "model_forward_time": 1.69344162940979,
      "step": 10084
    },
    {
      "epoch": 9.616851806640625e-06,
      "step": 10084,
      "training_step_time": 7.572341203689575
    },
    {
      "epoch": 9.617805480957032e-06,
      "model_forward_time": 1.695669412612915,
      "step": 10085
    },
    {
      "epoch": 9.617805480957032e-06,
      "step": 10085,
      "training_step_time": 8.050874710083008
    },
    {
      "epoch": 9.618759155273438e-06,
      "model_forward_time": 1.716411828994751,
      "step": 10086
    },
    {
      "epoch": 9.618759155273438e-06,
      "step": 10086,
      "training_step_time": 7.377018213272095
    },
    {
      "epoch": 9.619712829589844e-06,
      "model_forward_time": 1.6943304538726807,
      "step": 10087
    },
    {
      "epoch": 9.619712829589844e-06,
      "step": 10087,
      "training_step_time": 7.99812650680542
    },
    {
      "epoch": 9.62066650390625e-06,
      "model_forward_time": 1.7086310386657715,
      "step": 10088
    },
    {
      "epoch": 9.62066650390625e-06,
      "step": 10088,
      "training_step_time": 8.168153524398804
    },
    {
      "epoch": 9.621620178222657e-06,
      "model_forward_time": 1.7151799201965332,
      "step": 10089
    },
    {
      "epoch": 9.621620178222657e-06,
      "step": 10089,
      "training_step_time": 8.518927335739136
    },
    {
      "epoch": 9.622573852539062e-06,
      "grad_norm": 0.25861501693725586,
      "learning_rate": 9.929363005699282e-06,
      "loss": 0.0733,
      "step": 10090
    },
    {
      "epoch": 9.622573852539062e-06,
      "model_forward_time": 1.7543227672576904,
      "step": 10090
    },
    {
      "dynamics_loss_avg": 0.05631790356710553,
      "epoch": 9.622573852539062e-06,
      "step": 10090
    },
    {
      "action_loss_avg": 0.003049157082568854,
      "epoch": 9.622573852539062e-06,
      "step": 10090
    },
    {
      "epoch": 9.622573852539062e-06,
      "step": 10090,
      "training_step_time": 8.646770477294922
    },
    {
      "epoch": 9.623527526855468e-06,
      "model_forward_time": 1.7153279781341553,
      "step": 10091
    },
    {
      "epoch": 9.623527526855468e-06,
      "step": 10091,
      "training_step_time": 7.600143194198608
    },
    {
      "epoch": 9.624481201171876e-06,
      "model_forward_time": 1.7134668827056885,
      "step": 10092
    },
    {
      "epoch": 9.624481201171876e-06,
      "step": 10092,
      "training_step_time": 7.529182195663452
    },
    {
      "epoch": 9.625434875488281e-06,
      "model_forward_time": 1.7116994857788086,
      "step": 10093
    },
    {
      "epoch": 9.625434875488281e-06,
      "step": 10093,
      "training_step_time": 7.567790508270264
    },
    {
      "epoch": 9.626388549804687e-06,
      "model_forward_time": 1.7126028537750244,
      "step": 10094
    },
    {
      "epoch": 9.626388549804687e-06,
      "step": 10094,
      "training_step_time": 7.497663974761963
    },
    {
      "epoch": 9.627342224121094e-06,
      "model_forward_time": 1.333066701889038,
      "step": 10095
    },
    {
      "epoch": 9.627342224121094e-06,
      "step": 10095,
      "training_step_time": 7.551485061645508
    },
    {
      "epoch": 9.6282958984375e-06,
      "model_forward_time": 1.7148258686065674,
      "step": 10096
    },
    {
      "epoch": 9.6282958984375e-06,
      "step": 10096,
      "training_step_time": 7.585645914077759
    },
    {
      "epoch": 9.629249572753906e-06,
      "model_forward_time": 2.248563766479492,
      "step": 10097
    },
    {
      "epoch": 9.629249572753906e-06,
      "step": 10097,
      "training_step_time": 9.648932933807373
    },
    {
      "epoch": 9.630203247070313e-06,
      "model_forward_time": 1.7179255485534668,
      "step": 10098
    },
    {
      "epoch": 9.630203247070313e-06,
      "step": 10098,
      "training_step_time": 8.345861196517944
    },
    {
      "epoch": 9.631156921386719e-06,
      "model_forward_time": 1.7585797309875488,
      "step": 10099
    },
    {
      "epoch": 9.631156921386719e-06,
      "step": 10099,
      "training_step_time": 8.565820217132568
    },
    {
      "epoch": 9.632110595703125e-06,
      "grad_norm": 0.18663331866264343,
      "learning_rate": 9.929085785087881e-06,
      "loss": 0.0733,
      "step": 10100
    },
    {
      "epoch": 9.632110595703125e-06,
      "model_forward_time": 1.7067079544067383,
      "step": 10100
    },
    {
      "dynamics_loss_avg": 0.06424383595585823,
      "epoch": 9.632110595703125e-06,
      "step": 10100
    },
    {
      "action_loss_avg": 0.003690581698901951,
      "epoch": 9.632110595703125e-06,
      "step": 10100
    },
    {
      "epoch": 9.632110595703125e-06,
      "step": 10100,
      "training_step_time": 8.322925806045532
    },
    {
      "epoch": 9.633064270019532e-06,
      "model_forward_time": 1.7122812271118164,
      "step": 10101
    },
    {
      "epoch": 9.633064270019532e-06,
      "step": 10101,
      "training_step_time": 7.483981370925903
    },
    {
      "epoch": 9.634017944335938e-06,
      "model_forward_time": 1.7258491516113281,
      "step": 10102
    },
    {
      "epoch": 9.634017944335938e-06,
      "step": 10102,
      "training_step_time": 7.4575035572052
    },
    {
      "epoch": 9.634971618652344e-06,
      "model_forward_time": 1.7114088535308838,
      "step": 10103
    },
    {
      "epoch": 9.634971618652344e-06,
      "step": 10103,
      "training_step_time": 7.323968172073364
    },
    {
      "epoch": 9.63592529296875e-06,
      "model_forward_time": 1.7151219844818115,
      "step": 10104
    },
    {
      "epoch": 9.63592529296875e-06,
      "step": 10104,
      "training_step_time": 7.597128629684448
    },
    {
      "epoch": 9.636878967285157e-06,
      "model_forward_time": 1.7146809101104736,
      "step": 10105
    },
    {
      "epoch": 9.636878967285157e-06,
      "step": 10105,
      "training_step_time": 7.540520429611206
    },
    {
      "epoch": 9.637832641601562e-06,
      "model_forward_time": 2.2026045322418213,
      "step": 10106
    },
    {
      "epoch": 9.637832641601562e-06,
      "step": 10106,
      "training_step_time": 8.166368246078491
    },
    {
      "epoch": 9.638786315917968e-06,
      "model_forward_time": 1.6950397491455078,
      "step": 10107
    },
    {
      "epoch": 9.638786315917968e-06,
      "step": 10107,
      "training_step_time": 9.133308172225952
    },
    {
      "epoch": 9.639739990234376e-06,
      "model_forward_time": 1.7205164432525635,
      "step": 10108
    },
    {
      "epoch": 9.639739990234376e-06,
      "step": 10108,
      "training_step_time": 7.573386192321777
    },
    {
      "epoch": 9.640693664550781e-06,
      "model_forward_time": 1.6925334930419922,
      "step": 10109
    },
    {
      "epoch": 9.640693664550781e-06,
      "step": 10109,
      "training_step_time": 9.022912740707397
    },
    {
      "epoch": 9.641647338867187e-06,
      "grad_norm": 0.19648800790309906,
      "learning_rate": 9.92880802543907e-06,
      "loss": 0.0717,
      "step": 10110
    },
    {
      "epoch": 9.641647338867187e-06,
      "model_forward_time": 1.6962864398956299,
      "step": 10110
    },
    {
      "dynamics_loss_avg": 0.06415572762489319,
      "epoch": 9.641647338867187e-06,
      "step": 10110
    },
    {
      "action_loss_avg": 0.0027430290880147367,
      "epoch": 9.641647338867187e-06,
      "step": 10110
    },
    {
      "epoch": 9.641647338867187e-06,
      "step": 10110,
      "training_step_time": 7.760728597640991
    },
    {
      "epoch": 9.642601013183594e-06,
      "model_forward_time": 1.694988489151001,
      "step": 10111
    },
    {
      "epoch": 9.642601013183594e-06,
      "step": 10111,
      "training_step_time": 7.639216423034668
    },
    {
      "epoch": 9.6435546875e-06,
      "model_forward_time": 1.703232765197754,
      "step": 10112
    },
    {
      "epoch": 9.6435546875e-06,
      "step": 10112,
      "training_step_time": 7.920580863952637
    },
    {
      "epoch": 9.644508361816406e-06,
      "model_forward_time": 1.7091989517211914,
      "step": 10113
    },
    {
      "epoch": 9.644508361816406e-06,
      "step": 10113,
      "training_step_time": 7.919543266296387
    },
    {
      "epoch": 9.645462036132813e-06,
      "model_forward_time": 1.701585292816162,
      "step": 10114
    },
    {
      "epoch": 9.645462036132813e-06,
      "step": 10114,
      "training_step_time": 8.730975151062012
    },
    {
      "epoch": 9.646415710449219e-06,
      "model_forward_time": 2.2282357215881348,
      "step": 10115
    },
    {
      "epoch": 9.646415710449219e-06,
      "step": 10115,
      "training_step_time": 8.73332953453064
    },
    {
      "epoch": 9.647369384765625e-06,
      "model_forward_time": 1.7033910751342773,
      "step": 10116
    },
    {
      "epoch": 9.647369384765625e-06,
      "step": 10116,
      "training_step_time": 9.731081485748291
    },
    {
      "epoch": 9.648323059082032e-06,
      "model_forward_time": 1.717332363128662,
      "step": 10117
    },
    {
      "epoch": 9.648323059082032e-06,
      "step": 10117,
      "training_step_time": 7.885439872741699
    },
    {
      "epoch": 9.649276733398438e-06,
      "model_forward_time": 1.7016770839691162,
      "step": 10118
    },
    {
      "epoch": 9.649276733398438e-06,
      "step": 10118,
      "training_step_time": 7.891404628753662
    },
    {
      "epoch": 9.650230407714844e-06,
      "model_forward_time": 1.7031092643737793,
      "step": 10119
    },
    {
      "epoch": 9.650230407714844e-06,
      "step": 10119,
      "training_step_time": 7.929566860198975
    },
    {
      "epoch": 9.65118408203125e-06,
      "grad_norm": 0.19646312296390533,
      "learning_rate": 9.928529726783223e-06,
      "loss": 0.0661,
      "step": 10120
    },
    {
      "epoch": 9.65118408203125e-06,
      "model_forward_time": 1.696380615234375,
      "step": 10120
    },
    {
      "dynamics_loss_avg": 0.05847766920924187,
      "epoch": 9.65118408203125e-06,
      "step": 10120
    },
    {
      "action_loss_avg": 0.004038087883964181,
      "epoch": 9.65118408203125e-06,
      "step": 10120
    },
    {
      "epoch": 9.65118408203125e-06,
      "step": 10120,
      "training_step_time": 8.619104862213135
    },
    {
      "epoch": 9.652137756347657e-06,
      "model_forward_time": 1.7020094394683838,
      "step": 10121
    },
    {
      "epoch": 9.652137756347657e-06,
      "step": 10121,
      "training_step_time": 7.903999328613281
    },
    {
      "epoch": 9.653091430664062e-06,
      "model_forward_time": 1.7032639980316162,
      "step": 10122
    },
    {
      "epoch": 9.653091430664062e-06,
      "step": 10122,
      "training_step_time": 7.8258867263793945
    },
    {
      "epoch": 9.654045104980468e-06,
      "model_forward_time": 1.704589605331421,
      "step": 10123
    },
    {
      "epoch": 9.654045104980468e-06,
      "step": 10123,
      "training_step_time": 7.794991731643677
    },
    {
      "epoch": 9.654998779296876e-06,
      "model_forward_time": 2.246486186981201,
      "step": 10124
    },
    {
      "epoch": 9.654998779296876e-06,
      "step": 10124,
      "training_step_time": 7.779067039489746
    },
    {
      "epoch": 9.655952453613281e-06,
      "model_forward_time": 1.6981825828552246,
      "step": 10125
    },
    {
      "epoch": 9.655952453613281e-06,
      "step": 10125,
      "training_step_time": 8.763374090194702
    },
    {
      "epoch": 9.656906127929687e-06,
      "model_forward_time": 1.6992721557617188,
      "step": 10126
    },
    {
      "epoch": 9.656906127929687e-06,
      "step": 10126,
      "training_step_time": 8.396108150482178
    },
    {
      "epoch": 9.657859802246094e-06,
      "model_forward_time": 1.703584909439087,
      "step": 10127
    },
    {
      "epoch": 9.657859802246094e-06,
      "step": 10127,
      "training_step_time": 7.92936635017395
    },
    {
      "epoch": 9.6588134765625e-06,
      "model_forward_time": 1.6997058391571045,
      "step": 10128
    },
    {
      "epoch": 9.6588134765625e-06,
      "step": 10128,
      "training_step_time": 7.898071527481079
    },
    {
      "epoch": 9.659767150878906e-06,
      "model_forward_time": 1.7017524242401123,
      "step": 10129
    },
    {
      "epoch": 9.659767150878906e-06,
      "step": 10129,
      "training_step_time": 7.803305625915527
    },
    {
      "epoch": 9.660720825195313e-06,
      "grad_norm": 0.2056490033864975,
      "learning_rate": 9.928250889150774e-06,
      "loss": 0.0665,
      "step": 10130
    },
    {
      "epoch": 9.660720825195313e-06,
      "model_forward_time": 1.696462631225586,
      "step": 10130
    },
    {
      "dynamics_loss_avg": 0.056790762767195704,
      "epoch": 9.660720825195313e-06,
      "step": 10130
    },
    {
      "action_loss_avg": 0.0037537031224928797,
      "epoch": 9.660720825195313e-06,
      "step": 10130
    },
    {
      "epoch": 9.660720825195313e-06,
      "step": 10130,
      "training_step_time": 8.694329261779785
    },
    {
      "epoch": 9.661674499511719e-06,
      "model_forward_time": 1.7033605575561523,
      "step": 10131
    },
    {
      "epoch": 9.661674499511719e-06,
      "step": 10131,
      "training_step_time": 7.837137699127197
    },
    {
      "epoch": 9.662628173828125e-06,
      "model_forward_time": 1.6997816562652588,
      "step": 10132
    },
    {
      "epoch": 9.662628173828125e-06,
      "step": 10132,
      "training_step_time": 8.36711835861206
    },
    {
      "epoch": 9.66358184814453e-06,
      "model_forward_time": 1.7035574913024902,
      "step": 10133
    },
    {
      "epoch": 9.66358184814453e-06,
      "step": 10133,
      "training_step_time": 7.895379066467285
    },
    {
      "epoch": 9.664535522460938e-06,
      "model_forward_time": 2.231327772140503,
      "step": 10134
    },
    {
      "epoch": 9.664535522460938e-06,
      "step": 10134,
      "training_step_time": 8.878207921981812
    },
    {
      "epoch": 9.665489196777344e-06,
      "model_forward_time": 1.7043848037719727,
      "step": 10135
    },
    {
      "epoch": 9.665489196777344e-06,
      "step": 10135,
      "training_step_time": 9.423369646072388
    },
    {
      "epoch": 9.66644287109375e-06,
      "model_forward_time": 1.7467041015625,
      "step": 10136
    },
    {
      "epoch": 9.66644287109375e-06,
      "step": 10136,
      "training_step_time": 7.90198826789856
    },
    {
      "epoch": 9.667396545410157e-06,
      "model_forward_time": 1.7043733596801758,
      "step": 10137
    },
    {
      "epoch": 9.667396545410157e-06,
      "step": 10137,
      "training_step_time": 7.684793710708618
    },
    {
      "epoch": 9.668350219726562e-06,
      "model_forward_time": 1.701296091079712,
      "step": 10138
    },
    {
      "epoch": 9.668350219726562e-06,
      "step": 10138,
      "training_step_time": 7.817806959152222
    },
    {
      "epoch": 9.669303894042968e-06,
      "model_forward_time": 1.3277454376220703,
      "step": 10139
    },
    {
      "epoch": 9.669303894042968e-06,
      "step": 10139,
      "training_step_time": 7.289064407348633
    },
    {
      "epoch": 9.670257568359376e-06,
      "grad_norm": 0.2379663586616516,
      "learning_rate": 9.92797151257222e-06,
      "loss": 0.0682,
      "step": 10140
    },
    {
      "epoch": 9.670257568359376e-06,
      "model_forward_time": 1.704923391342163,
      "step": 10140
    },
    {
      "dynamics_loss_avg": 0.058718277513980864,
      "epoch": 9.670257568359376e-06,
      "step": 10140
    },
    {
      "action_loss_avg": 0.004144855216145516,
      "epoch": 9.670257568359376e-06,
      "step": 10140
    },
    {
      "epoch": 9.670257568359376e-06,
      "step": 10140,
      "training_step_time": 7.834034204483032
    },
    {
      "epoch": 9.671211242675781e-06,
      "model_forward_time": 1.7007756233215332,
      "step": 10141
    },
    {
      "epoch": 9.671211242675781e-06,
      "step": 10141,
      "training_step_time": 9.21743369102478
    },
    {
      "epoch": 9.672164916992187e-06,
      "model_forward_time": 1.7016890048980713,
      "step": 10142
    },
    {
      "epoch": 9.672164916992187e-06,
      "step": 10142,
      "training_step_time": 9.26493215560913
    },
    {
      "epoch": 9.673118591308594e-06,
      "model_forward_time": 2.2287490367889404,
      "step": 10143
    },
    {
      "epoch": 9.673118591308594e-06,
      "step": 10143,
      "training_step_time": 9.983858346939087
    },
    {
      "epoch": 9.674072265625e-06,
      "model_forward_time": 1.7011337280273438,
      "step": 10144
    },
    {
      "epoch": 9.674072265625e-06,
      "step": 10144,
      "training_step_time": 7.9326348304748535
    },
    {
      "epoch": 9.675025939941406e-06,
      "model_forward_time": 1.749788522720337,
      "step": 10145
    },
    {
      "epoch": 9.675025939941406e-06,
      "step": 10145,
      "training_step_time": 8.48038387298584
    },
    {
      "epoch": 9.675979614257813e-06,
      "model_forward_time": 1.701585054397583,
      "step": 10146
    },
    {
      "epoch": 9.675979614257813e-06,
      "step": 10146,
      "training_step_time": 8.03960919380188
    },
    {
      "epoch": 9.676933288574219e-06,
      "model_forward_time": 1.7042574882507324,
      "step": 10147
    },
    {
      "epoch": 9.676933288574219e-06,
      "step": 10147,
      "training_step_time": 7.863013505935669
    },
    {
      "epoch": 9.677886962890625e-06,
      "model_forward_time": 1.7046926021575928,
      "step": 10148
    },
    {
      "epoch": 9.677886962890625e-06,
      "step": 10148,
      "training_step_time": 7.660682201385498
    },
    {
      "epoch": 9.67884063720703e-06,
      "model_forward_time": 1.7062616348266602,
      "step": 10149
    },
    {
      "epoch": 9.67884063720703e-06,
      "step": 10149,
      "training_step_time": 7.869553804397583
    },
    {
      "epoch": 9.679794311523438e-06,
      "grad_norm": 0.19981108605861664,
      "learning_rate": 9.927691597078109e-06,
      "loss": 0.0693,
      "step": 10150
    },
    {
      "epoch": 9.679794311523438e-06,
      "model_forward_time": 1.6959288120269775,
      "step": 10150
    },
    {
      "dynamics_loss_avg": 0.06082743331789971,
      "epoch": 9.679794311523438e-06,
      "step": 10150
    },
    {
      "action_loss_avg": 0.005020693014375866,
      "epoch": 9.679794311523438e-06,
      "step": 10150
    },
    {
      "epoch": 9.679794311523438e-06,
      "step": 10150,
      "training_step_time": 8.93169355392456
    },
    {
      "epoch": 9.680747985839844e-06,
      "model_forward_time": 1.702568531036377,
      "step": 10151
    },
    {
      "epoch": 9.680747985839844e-06,
      "step": 10151,
      "training_step_time": 7.894133567810059
    },
    {
      "epoch": 9.68170166015625e-06,
      "model_forward_time": 1.8373644351959229,
      "step": 10152
    },
    {
      "epoch": 9.68170166015625e-06,
      "step": 10152,
      "training_step_time": 8.392302751541138
    },
    {
      "epoch": 9.682655334472657e-06,
      "model_forward_time": 1.7024002075195312,
      "step": 10153
    },
    {
      "epoch": 9.682655334472657e-06,
      "step": 10153,
      "training_step_time": 8.003646612167358
    },
    {
      "epoch": 9.683609008789062e-06,
      "model_forward_time": 1.7500505447387695,
      "step": 10154
    },
    {
      "epoch": 9.683609008789062e-06,
      "step": 10154,
      "training_step_time": 7.871139287948608
    },
    {
      "epoch": 9.684562683105468e-06,
      "model_forward_time": 1.7015564441680908,
      "step": 10155
    },
    {
      "epoch": 9.684562683105468e-06,
      "step": 10155,
      "training_step_time": 7.958710432052612
    },
    {
      "epoch": 9.685516357421876e-06,
      "model_forward_time": 1.7063651084899902,
      "step": 10156
    },
    {
      "epoch": 9.685516357421876e-06,
      "step": 10156,
      "training_step_time": 7.792058944702148
    },
    {
      "epoch": 9.686470031738281e-06,
      "model_forward_time": 1.6788926124572754,
      "step": 10157
    },
    {
      "epoch": 9.686470031738281e-06,
      "step": 10157,
      "training_step_time": 7.604216575622559
    },
    {
      "epoch": 9.687423706054687e-06,
      "model_forward_time": 1.7027127742767334,
      "step": 10158
    },
    {
      "epoch": 9.687423706054687e-06,
      "step": 10158,
      "training_step_time": 7.959277868270874
    },
    {
      "epoch": 9.688377380371094e-06,
      "model_forward_time": 1.7003045082092285,
      "step": 10159
    },
    {
      "epoch": 9.688377380371094e-06,
      "step": 10159,
      "training_step_time": 7.996761322021484
    },
    {
      "epoch": 9.6893310546875e-06,
      "grad_norm": 0.2688884139060974,
      "learning_rate": 9.927411142699054e-06,
      "loss": 0.0713,
      "step": 10160
    },
    {
      "epoch": 9.6893310546875e-06,
      "model_forward_time": 1.6993460655212402,
      "step": 10160
    },
    {
      "dynamics_loss_avg": 0.06032247319817543,
      "epoch": 9.6893310546875e-06,
      "step": 10160
    },
    {
      "action_loss_avg": 0.005370111193042248,
      "epoch": 9.6893310546875e-06,
      "step": 10160
    },
    {
      "epoch": 9.6893310546875e-06,
      "step": 10160,
      "training_step_time": 7.8977766036987305
    },
    {
      "epoch": 9.690284729003906e-06,
      "model_forward_time": 2.2269484996795654,
      "step": 10161
    },
    {
      "epoch": 9.690284729003906e-06,
      "step": 10161,
      "training_step_time": 7.945490837097168
    },
    {
      "epoch": 9.691238403320313e-06,
      "model_forward_time": 1.707357406616211,
      "step": 10162
    },
    {
      "epoch": 9.691238403320313e-06,
      "step": 10162,
      "training_step_time": 8.554870843887329
    },
    {
      "epoch": 9.692192077636719e-06,
      "model_forward_time": 1.3237247467041016,
      "step": 10163
    },
    {
      "epoch": 9.692192077636719e-06,
      "step": 10163,
      "training_step_time": 8.813717603683472
    },
    {
      "epoch": 9.693145751953125e-06,
      "model_forward_time": 1.710216999053955,
      "step": 10164
    },
    {
      "epoch": 9.693145751953125e-06,
      "step": 10164,
      "training_step_time": 7.756194114685059
    },
    {
      "epoch": 9.69409942626953e-06,
      "model_forward_time": 1.3301923274993896,
      "step": 10165
    },
    {
      "epoch": 9.69409942626953e-06,
      "step": 10165,
      "training_step_time": 8.490006446838379
    },
    {
      "epoch": 9.695053100585938e-06,
      "model_forward_time": 1.7047951221466064,
      "step": 10166
    },
    {
      "epoch": 9.695053100585938e-06,
      "step": 10166,
      "training_step_time": 7.865099668502808
    },
    {
      "epoch": 9.696006774902344e-06,
      "model_forward_time": 1.7093639373779297,
      "step": 10167
    },
    {
      "epoch": 9.696006774902344e-06,
      "step": 10167,
      "training_step_time": 7.841937780380249
    },
    {
      "epoch": 9.69696044921875e-06,
      "model_forward_time": 1.6954715251922607,
      "step": 10168
    },
    {
      "epoch": 9.69696044921875e-06,
      "step": 10168,
      "training_step_time": 7.923415899276733
    },
    {
      "epoch": 9.697914123535157e-06,
      "model_forward_time": 1.7047853469848633,
      "step": 10169
    },
    {
      "epoch": 9.697914123535157e-06,
      "step": 10169,
      "training_step_time": 7.9505228996276855
    },
    {
      "epoch": 9.698867797851562e-06,
      "grad_norm": 0.15659527480602264,
      "learning_rate": 9.927130149465725e-06,
      "loss": 0.069,
      "step": 10170
    },
    {
      "epoch": 9.698867797851562e-06,
      "model_forward_time": 1.6994726657867432,
      "step": 10170
    },
    {
      "dynamics_loss_avg": 0.057258921302855016,
      "epoch": 9.698867797851562e-06,
      "step": 10170
    },
    {
      "action_loss_avg": 0.0036485903896391393,
      "epoch": 9.698867797851562e-06,
      "step": 10170
    },
    {
      "epoch": 9.698867797851562e-06,
      "step": 10170,
      "training_step_time": 9.750388860702515
    },
    {
      "epoch": 9.699821472167968e-06,
      "model_forward_time": 2.2216238975524902,
      "step": 10171
    },
    {
      "epoch": 9.699821472167968e-06,
      "step": 10171,
      "training_step_time": 9.546897888183594
    },
    {
      "epoch": 9.700775146484376e-06,
      "model_forward_time": 1.721634864807129,
      "step": 10172
    },
    {
      "epoch": 9.700775146484376e-06,
      "step": 10172,
      "training_step_time": 7.927958011627197
    },
    {
      "epoch": 9.701728820800781e-06,
      "model_forward_time": 1.6990740299224854,
      "step": 10173
    },
    {
      "epoch": 9.701728820800781e-06,
      "step": 10173,
      "training_step_time": 8.765991449356079
    },
    {
      "epoch": 9.702682495117187e-06,
      "model_forward_time": 1.7040531635284424,
      "step": 10174
    },
    {
      "epoch": 9.702682495117187e-06,
      "step": 10174,
      "training_step_time": 7.935268878936768
    },
    {
      "epoch": 9.703636169433594e-06,
      "model_forward_time": 1.3286175727844238,
      "step": 10175
    },
    {
      "epoch": 9.703636169433594e-06,
      "step": 10175,
      "training_step_time": 7.844279527664185
    },
    {
      "epoch": 9.70458984375e-06,
      "model_forward_time": 1.7122993469238281,
      "step": 10176
    },
    {
      "epoch": 9.70458984375e-06,
      "step": 10176,
      "training_step_time": 7.716033220291138
    },
    {
      "epoch": 9.705543518066406e-06,
      "model_forward_time": 1.7041170597076416,
      "step": 10177
    },
    {
      "epoch": 9.705543518066406e-06,
      "step": 10177,
      "training_step_time": 8.760330200195312
    },
    {
      "epoch": 9.706497192382813e-06,
      "model_forward_time": 1.699350357055664,
      "step": 10178
    },
    {
      "epoch": 9.706497192382813e-06,
      "step": 10178,
      "training_step_time": 9.429971933364868
    },
    {
      "epoch": 9.707450866699219e-06,
      "model_forward_time": 1.7031219005584717,
      "step": 10179
    },
    {
      "epoch": 9.707450866699219e-06,
      "step": 10179,
      "training_step_time": 8.862798690795898
    },
    {
      "epoch": 9.708404541015625e-06,
      "grad_norm": 0.27603092789649963,
      "learning_rate": 9.92684861740885e-06,
      "loss": 0.071,
      "step": 10180
    },
    {
      "epoch": 9.708404541015625e-06,
      "model_forward_time": 2.241502046585083,
      "step": 10180
    },
    {
      "dynamics_loss_avg": 0.07811882551759482,
      "epoch": 9.708404541015625e-06,
      "step": 10180
    },
    {
      "action_loss_avg": 0.004666927095968276,
      "epoch": 9.708404541015625e-06,
      "step": 10180
    },
    {
      "epoch": 9.708404541015625e-06,
      "step": 10180,
      "training_step_time": 8.927777290344238
    },
    {
      "epoch": 9.70935821533203e-06,
      "model_forward_time": 1.6999762058258057,
      "step": 10181
    },
    {
      "epoch": 9.70935821533203e-06,
      "step": 10181,
      "training_step_time": 7.66491436958313
    },
    {
      "epoch": 9.710311889648438e-06,
      "model_forward_time": 1.7051780223846436,
      "step": 10182
    },
    {
      "epoch": 9.710311889648438e-06,
      "step": 10182,
      "training_step_time": 8.387080192565918
    },
    {
      "epoch": 9.711265563964844e-06,
      "model_forward_time": 1.7082350254058838,
      "step": 10183
    },
    {
      "epoch": 9.711265563964844e-06,
      "step": 10183,
      "training_step_time": 7.671867847442627
    },
    {
      "epoch": 9.71221923828125e-06,
      "model_forward_time": 1.7004380226135254,
      "step": 10184
    },
    {
      "epoch": 9.71221923828125e-06,
      "step": 10184,
      "training_step_time": 7.603842496871948
    },
    {
      "epoch": 9.713172912597657e-06,
      "model_forward_time": 1.6983096599578857,
      "step": 10185
    },
    {
      "epoch": 9.713172912597657e-06,
      "step": 10185,
      "training_step_time": 7.604386806488037
    },
    {
      "epoch": 9.714126586914062e-06,
      "model_forward_time": 1.7040338516235352,
      "step": 10186
    },
    {
      "epoch": 9.714126586914062e-06,
      "step": 10186,
      "training_step_time": 7.514697313308716
    },
    {
      "epoch": 9.715080261230468e-06,
      "model_forward_time": 1.7027323246002197,
      "step": 10187
    },
    {
      "epoch": 9.715080261230468e-06,
      "step": 10187,
      "training_step_time": 8.084786176681519
    },
    {
      "epoch": 9.716033935546876e-06,
      "model_forward_time": 1.698094367980957,
      "step": 10188
    },
    {
      "epoch": 9.716033935546876e-06,
      "step": 10188,
      "training_step_time": 9.285070896148682
    },
    {
      "epoch": 9.716987609863281e-06,
      "model_forward_time": 2.248100519180298,
      "step": 10189
    },
    {
      "epoch": 9.716987609863281e-06,
      "step": 10189,
      "training_step_time": 7.684148550033569
    },
    {
      "epoch": 9.717941284179687e-06,
      "grad_norm": 0.21873122453689575,
      "learning_rate": 9.926566546559218e-06,
      "loss": 0.0723,
      "step": 10190
    },
    {
      "epoch": 9.717941284179687e-06,
      "model_forward_time": 1.7031097412109375,
      "step": 10190
    },
    {
      "dynamics_loss_avg": 0.05145542789250612,
      "epoch": 9.717941284179687e-06,
      "step": 10190
    },
    {
      "action_loss_avg": 0.003246846527326852,
      "epoch": 9.717941284179687e-06,
      "step": 10190
    },
    {
      "epoch": 9.717941284179687e-06,
      "step": 10190,
      "training_step_time": 8.401452541351318
    },
    {
      "epoch": 9.718894958496094e-06,
      "model_forward_time": 1.7397592067718506,
      "step": 10191
    },
    {
      "epoch": 9.718894958496094e-06,
      "step": 10191,
      "training_step_time": 10.284306764602661
    },
    {
      "epoch": 9.7198486328125e-06,
      "model_forward_time": 1.705667495727539,
      "step": 10192
    },
    {
      "epoch": 9.7198486328125e-06,
      "step": 10192,
      "training_step_time": 7.454890012741089
    },
    {
      "epoch": 9.720802307128906e-06,
      "model_forward_time": 1.7077805995941162,
      "step": 10193
    },
    {
      "epoch": 9.720802307128906e-06,
      "step": 10193,
      "training_step_time": 7.691116094589233
    },
    {
      "epoch": 9.721755981445313e-06,
      "model_forward_time": 1.6991913318634033,
      "step": 10194
    },
    {
      "epoch": 9.721755981445313e-06,
      "step": 10194,
      "training_step_time": 7.44702410697937
    },
    {
      "epoch": 9.722709655761719e-06,
      "model_forward_time": 1.6939170360565186,
      "step": 10195
    },
    {
      "epoch": 9.722709655761719e-06,
      "step": 10195,
      "training_step_time": 8.31531834602356
    },
    {
      "epoch": 9.723663330078125e-06,
      "model_forward_time": 1.6979033946990967,
      "step": 10196
    },
    {
      "epoch": 9.723663330078125e-06,
      "step": 10196,
      "training_step_time": 10.317471742630005
    },
    {
      "epoch": 9.72461700439453e-06,
      "model_forward_time": 1.7018163204193115,
      "step": 10197
    },
    {
      "epoch": 9.72461700439453e-06,
      "step": 10197,
      "training_step_time": 7.69343113899231
    },
    {
      "epoch": 9.725570678710938e-06,
      "model_forward_time": 2.226006269454956,
      "step": 10198
    },
    {
      "epoch": 9.725570678710938e-06,
      "step": 10198,
      "training_step_time": 8.668197393417358
    },
    {
      "epoch": 9.726524353027344e-06,
      "model_forward_time": 1.7017748355865479,
      "step": 10199
    },
    {
      "epoch": 9.726524353027344e-06,
      "step": 10199,
      "training_step_time": 8.695763349533081
    },
    {
      "epoch": 9.72747802734375e-06,
      "grad_norm": 0.1452968418598175,
      "learning_rate": 9.926283936947673e-06,
      "loss": 0.0734,
      "step": 10200
    },
    {
      "epoch": 9.72747802734375e-06,
      "model_forward_time": 1.7466154098510742,
      "step": 10200
    },
    {
      "dynamics_loss_avg": 0.06620122361928224,
      "epoch": 9.72747802734375e-06,
      "step": 10200
    },
    {
      "action_loss_avg": 0.0035581473202910274,
      "epoch": 9.72747802734375e-06,
      "step": 10200
    },
    {
      "epoch": 9.72747802734375e-06,
      "step": 10200,
      "training_step_time": 8.695362329483032
    },
    {
      "epoch": 9.728431701660157e-06,
      "model_forward_time": 1.697385549545288,
      "step": 10201
    },
    {
      "epoch": 9.728431701660157e-06,
      "step": 10201,
      "training_step_time": 7.721031665802002
    },
    {
      "epoch": 9.729385375976562e-06,
      "model_forward_time": 1.7034518718719482,
      "step": 10202
    },
    {
      "epoch": 9.729385375976562e-06,
      "step": 10202,
      "training_step_time": 7.773820161819458
    },
    {
      "epoch": 9.730339050292968e-06,
      "model_forward_time": 1.7018077373504639,
      "step": 10203
    },
    {
      "epoch": 9.730339050292968e-06,
      "step": 10203,
      "training_step_time": 7.545892953872681
    },
    {
      "epoch": 9.731292724609376e-06,
      "model_forward_time": 1.7032859325408936,
      "step": 10204
    },
    {
      "epoch": 9.731292724609376e-06,
      "step": 10204,
      "training_step_time": 7.597019910812378
    },
    {
      "epoch": 9.732246398925781e-06,
      "model_forward_time": 1.70017671585083,
      "step": 10205
    },
    {
      "epoch": 9.732246398925781e-06,
      "step": 10205,
      "training_step_time": 8.683169603347778
    },
    {
      "epoch": 9.733200073242187e-06,
      "model_forward_time": 1.6999642848968506,
      "step": 10206
    },
    {
      "epoch": 9.733200073242187e-06,
      "step": 10206,
      "training_step_time": 7.61150860786438
    },
    {
      "epoch": 9.734153747558594e-06,
      "model_forward_time": 1.7021303176879883,
      "step": 10207
    },
    {
      "epoch": 9.734153747558594e-06,
      "step": 10207,
      "training_step_time": 7.655606269836426
    },
    {
      "epoch": 9.735107421875e-06,
      "model_forward_time": 2.2396044731140137,
      "step": 10208
    },
    {
      "epoch": 9.735107421875e-06,
      "step": 10208,
      "training_step_time": 7.677917957305908
    },
    {
      "epoch": 9.736061096191406e-06,
      "model_forward_time": 1.725027084350586,
      "step": 10209
    },
    {
      "epoch": 9.736061096191406e-06,
      "step": 10209,
      "training_step_time": 7.663159608840942
    },
    {
      "epoch": 9.737014770507813e-06,
      "grad_norm": 0.20304705202579498,
      "learning_rate": 9.926000788605127e-06,
      "loss": 0.0675,
      "step": 10210
    },
    {
      "epoch": 9.737014770507813e-06,
      "model_forward_time": 1.6921827793121338,
      "step": 10210
    },
    {
      "dynamics_loss_avg": 0.05363369192928076,
      "epoch": 9.737014770507813e-06,
      "step": 10210
    },
    {
      "action_loss_avg": 0.002855587803060189,
      "epoch": 9.737014770507813e-06,
      "step": 10210
    },
    {
      "epoch": 9.737014770507813e-06,
      "step": 10210,
      "training_step_time": 7.674354076385498
    },
    {
      "epoch": 9.737968444824219e-06,
      "model_forward_time": 1.7142794132232666,
      "step": 10211
    },
    {
      "epoch": 9.737968444824219e-06,
      "step": 10211,
      "training_step_time": 7.391781568527222
    },
    {
      "epoch": 9.738922119140625e-06,
      "model_forward_time": 1.699516773223877,
      "step": 10212
    },
    {
      "epoch": 9.738922119140625e-06,
      "step": 10212,
      "training_step_time": 7.578688859939575
    },
    {
      "epoch": 9.73987579345703e-06,
      "model_forward_time": 1.7009162902832031,
      "step": 10213
    },
    {
      "epoch": 9.73987579345703e-06,
      "step": 10213,
      "training_step_time": 7.658796548843384
    },
    {
      "epoch": 9.740829467773438e-06,
      "model_forward_time": 1.7074780464172363,
      "step": 10214
    },
    {
      "epoch": 9.740829467773438e-06,
      "step": 10214,
      "training_step_time": 7.625631809234619
    },
    {
      "epoch": 9.741783142089844e-06,
      "model_forward_time": 1.702150821685791,
      "step": 10215
    },
    {
      "epoch": 9.741783142089844e-06,
      "step": 10215,
      "training_step_time": 7.663121223449707
    },
    {
      "epoch": 9.74273681640625e-06,
      "model_forward_time": 1.7009782791137695,
      "step": 10216
    },
    {
      "epoch": 9.74273681640625e-06,
      "step": 10216,
      "training_step_time": 8.74791932106018
    },
    {
      "epoch": 9.743690490722657e-06,
      "model_forward_time": 2.2287416458129883,
      "step": 10217
    },
    {
      "epoch": 9.743690490722657e-06,
      "step": 10217,
      "training_step_time": 9.730466842651367
    },
    {
      "epoch": 9.744644165039062e-06,
      "model_forward_time": 1.7075474262237549,
      "step": 10218
    },
    {
      "epoch": 9.744644165039062e-06,
      "step": 10218,
      "training_step_time": 7.624853134155273
    },
    {
      "epoch": 9.745597839355468e-06,
      "model_forward_time": 1.6825242042541504,
      "step": 10219
    },
    {
      "epoch": 9.745597839355468e-06,
      "step": 10219,
      "training_step_time": 7.6539740562438965
    },
    {
      "epoch": 9.746551513671876e-06,
      "grad_norm": 0.19317412376403809,
      "learning_rate": 9.925717101562539e-06,
      "loss": 0.0673,
      "step": 10220
    },
    {
      "epoch": 9.746551513671876e-06,
      "model_forward_time": 1.7015526294708252,
      "step": 10220
    },
    {
      "dynamics_loss_avg": 0.05844015516340732,
      "epoch": 9.746551513671876e-06,
      "step": 10220
    },
    {
      "action_loss_avg": 0.0037473135278560223,
      "epoch": 9.746551513671876e-06,
      "step": 10220
    },
    {
      "epoch": 9.746551513671876e-06,
      "step": 10220,
      "training_step_time": 8.373663425445557
    },
    {
      "epoch": 9.747505187988281e-06,
      "model_forward_time": 1.7016351222991943,
      "step": 10221
    },
    {
      "epoch": 9.747505187988281e-06,
      "step": 10221,
      "training_step_time": 7.561824798583984
    },
    {
      "epoch": 9.748458862304687e-06,
      "model_forward_time": 1.7021162509918213,
      "step": 10222
    },
    {
      "epoch": 9.748458862304687e-06,
      "step": 10222,
      "training_step_time": 7.54837441444397
    },
    {
      "epoch": 9.749412536621094e-06,
      "model_forward_time": 1.7017230987548828,
      "step": 10223
    },
    {
      "epoch": 9.749412536621094e-06,
      "step": 10223,
      "training_step_time": 7.418360710144043
    },
    {
      "epoch": 9.7503662109375e-06,
      "model_forward_time": 1.70316481590271,
      "step": 10224
    },
    {
      "epoch": 9.7503662109375e-06,
      "step": 10224,
      "training_step_time": 7.701663255691528
    },
    {
      "epoch": 9.751319885253906e-06,
      "model_forward_time": 1.7009363174438477,
      "step": 10225
    },
    {
      "epoch": 9.751319885253906e-06,
      "step": 10225,
      "training_step_time": 7.7620227336883545
    },
    {
      "epoch": 9.752273559570313e-06,
      "model_forward_time": 2.235887289047241,
      "step": 10226
    },
    {
      "epoch": 9.752273559570313e-06,
      "step": 10226,
      "training_step_time": 9.699156045913696
    },
    {
      "epoch": 9.753227233886719e-06,
      "model_forward_time": 1.7060353755950928,
      "step": 10227
    },
    {
      "epoch": 9.753227233886719e-06,
      "step": 10227,
      "training_step_time": 8.423166275024414
    },
    {
      "epoch": 9.754180908203125e-06,
      "model_forward_time": 1.744852066040039,
      "step": 10228
    },
    {
      "epoch": 9.754180908203125e-06,
      "step": 10228,
      "training_step_time": 7.736312389373779
    },
    {
      "epoch": 9.75513458251953e-06,
      "model_forward_time": 1.7232844829559326,
      "step": 10229
    },
    {
      "epoch": 9.75513458251953e-06,
      "step": 10229,
      "training_step_time": 7.653064727783203
    },
    {
      "epoch": 9.756088256835938e-06,
      "grad_norm": 0.1677783727645874,
      "learning_rate": 9.925432875850935e-06,
      "loss": 0.0669,
      "step": 10230
    },
    {
      "epoch": 9.756088256835938e-06,
      "model_forward_time": 1.7215750217437744,
      "step": 10230
    },
    {
      "dynamics_loss_avg": 0.07850644700229167,
      "epoch": 9.756088256835938e-06,
      "step": 10230
    },
    {
      "action_loss_avg": 0.003419313789345324,
      "epoch": 9.756088256835938e-06,
      "step": 10230
    },
    {
      "epoch": 9.756088256835938e-06,
      "step": 10230,
      "training_step_time": 7.641552448272705
    },
    {
      "epoch": 9.757041931152344e-06,
      "model_forward_time": 1.6952252388000488,
      "step": 10231
    },
    {
      "epoch": 9.757041931152344e-06,
      "step": 10231,
      "training_step_time": 7.683179616928101
    },
    {
      "epoch": 9.75799560546875e-06,
      "model_forward_time": 1.6976220607757568,
      "step": 10232
    },
    {
      "epoch": 9.75799560546875e-06,
      "step": 10232,
      "training_step_time": 7.47258734703064
    },
    {
      "epoch": 9.758949279785157e-06,
      "model_forward_time": 1.70538330078125,
      "step": 10233
    },
    {
      "epoch": 9.758949279785157e-06,
      "step": 10233,
      "training_step_time": 8.180954217910767
    },
    {
      "epoch": 9.759902954101562e-06,
      "model_forward_time": 1.7087032794952393,
      "step": 10234
    },
    {
      "epoch": 9.759902954101562e-06,
      "step": 10234,
      "training_step_time": 8.605305671691895
    },
    {
      "epoch": 9.760856628417968e-06,
      "model_forward_time": 2.2375943660736084,
      "step": 10235
    },
    {
      "epoch": 9.760856628417968e-06,
      "step": 10235,
      "training_step_time": 11.340311050415039
    },
    {
      "epoch": 9.761810302734376e-06,
      "model_forward_time": 1.699920654296875,
      "step": 10236
    },
    {
      "epoch": 9.761810302734376e-06,
      "step": 10236,
      "training_step_time": 8.671976804733276
    },
    {
      "epoch": 9.762763977050781e-06,
      "model_forward_time": 1.7492163181304932,
      "step": 10237
    },
    {
      "epoch": 9.762763977050781e-06,
      "step": 10237,
      "training_step_time": 9.614867687225342
    },
    {
      "epoch": 9.763717651367187e-06,
      "model_forward_time": 1.3260884284973145,
      "step": 10238
    },
    {
      "epoch": 9.763717651367187e-06,
      "step": 10238,
      "training_step_time": 7.590457201004028
    },
    {
      "epoch": 9.764671325683594e-06,
      "model_forward_time": 1.6981532573699951,
      "step": 10239
    },
    {
      "epoch": 9.764671325683594e-06,
      "step": 10239,
      "training_step_time": 8.742080450057983
    },
    {
      "epoch": 9.765625e-06,
      "grad_norm": 0.19439959526062012,
      "learning_rate": 9.925148111501397e-06,
      "loss": 0.0655,
      "step": 10240
    },
    {
      "epoch": 9.765625e-06,
      "model_forward_time": 1.6905086040496826,
      "step": 10240
    },
    {
      "dynamics_loss_avg": 0.07351321689784526,
      "epoch": 9.765625e-06,
      "step": 10240
    },
    {
      "action_loss_avg": 0.0037995176971890032,
      "epoch": 9.765625e-06,
      "step": 10240
    },
    {
      "epoch": 9.765625e-06,
      "step": 10240,
      "training_step_time": 7.6117167472839355
    },
    {
      "epoch": 9.766578674316406e-06,
      "model_forward_time": 1.7052361965179443,
      "step": 10241
    },
    {
      "epoch": 9.766578674316406e-06,
      "step": 10241,
      "training_step_time": 7.68421483039856
    },
    {
      "epoch": 9.767532348632813e-06,
      "model_forward_time": 1.3226776123046875,
      "step": 10242
    },
    {
      "epoch": 9.767532348632813e-06,
      "step": 10242,
      "training_step_time": 7.722403049468994
    },
    {
      "epoch": 9.768486022949219e-06,
      "model_forward_time": 1.700357437133789,
      "step": 10243
    },
    {
      "epoch": 9.768486022949219e-06,
      "step": 10243,
      "training_step_time": 9.218693017959595
    },
    {
      "epoch": 9.769439697265625e-06,
      "model_forward_time": 1.6981635093688965,
      "step": 10244
    },
    {
      "epoch": 9.769439697265625e-06,
      "step": 10244,
      "training_step_time": 9.563756942749023
    },
    {
      "epoch": 9.77039337158203e-06,
      "model_forward_time": 2.232076406478882,
      "step": 10245
    },
    {
      "epoch": 9.77039337158203e-06,
      "step": 10245,
      "training_step_time": 8.657713651657104
    },
    {
      "epoch": 9.771347045898438e-06,
      "model_forward_time": 1.741513729095459,
      "step": 10246
    },
    {
      "epoch": 9.771347045898438e-06,
      "step": 10246,
      "training_step_time": 9.588046073913574
    },
    {
      "epoch": 9.772300720214844e-06,
      "model_forward_time": 1.7112009525299072,
      "step": 10247
    },
    {
      "epoch": 9.772300720214844e-06,
      "step": 10247,
      "training_step_time": 8.229858875274658
    },
    {
      "epoch": 9.77325439453125e-06,
      "model_forward_time": 1.6996145248413086,
      "step": 10248
    },
    {
      "epoch": 9.77325439453125e-06,
      "step": 10248,
      "training_step_time": 7.80639910697937
    },
    {
      "epoch": 9.774208068847657e-06,
      "model_forward_time": 1.706573247909546,
      "step": 10249
    },
    {
      "epoch": 9.774208068847657e-06,
      "step": 10249,
      "training_step_time": 7.72861647605896
    },
    {
      "epoch": 9.775161743164062e-06,
      "grad_norm": 0.17798544466495514,
      "learning_rate": 9.924862808545066e-06,
      "loss": 0.0659,
      "step": 10250
    },
    {
      "epoch": 9.775161743164062e-06,
      "model_forward_time": 1.7026288509368896,
      "step": 10250
    },
    {
      "dynamics_loss_avg": 0.056629805266857146,
      "epoch": 9.775161743164062e-06,
      "step": 10250
    },
    {
      "action_loss_avg": 0.004483841930050403,
      "epoch": 9.775161743164062e-06,
      "step": 10250
    },
    {
      "epoch": 9.775161743164062e-06,
      "step": 10250,
      "training_step_time": 7.838590621948242
    },
    {
      "epoch": 9.776115417480468e-06,
      "model_forward_time": 1.705007791519165,
      "step": 10251
    },
    {
      "epoch": 9.776115417480468e-06,
      "step": 10251,
      "training_step_time": 7.654960632324219
    },
    {
      "epoch": 9.777069091796876e-06,
      "model_forward_time": 1.7228751182556152,
      "step": 10252
    },
    {
      "epoch": 9.777069091796876e-06,
      "step": 10252,
      "training_step_time": 8.471903324127197
    },
    {
      "epoch": 9.778022766113281e-06,
      "model_forward_time": 1.7192273139953613,
      "step": 10253
    },
    {
      "epoch": 9.778022766113281e-06,
      "step": 10253,
      "training_step_time": 7.679470062255859
    },
    {
      "epoch": 9.778976440429687e-06,
      "model_forward_time": 2.2673494815826416,
      "step": 10254
    },
    {
      "epoch": 9.778976440429687e-06,
      "step": 10254,
      "training_step_time": 7.72932767868042
    },
    {
      "epoch": 9.779930114746094e-06,
      "model_forward_time": 1.7516124248504639,
      "step": 10255
    },
    {
      "epoch": 9.779930114746094e-06,
      "step": 10255,
      "training_step_time": 8.596120834350586
    },
    {
      "epoch": 9.7808837890625e-06,
      "model_forward_time": 1.7185368537902832,
      "step": 10256
    },
    {
      "epoch": 9.7808837890625e-06,
      "step": 10256,
      "training_step_time": 7.597817897796631
    },
    {
      "epoch": 9.781837463378906e-06,
      "model_forward_time": 1.7186286449432373,
      "step": 10257
    },
    {
      "epoch": 9.781837463378906e-06,
      "step": 10257,
      "training_step_time": 8.281294822692871
    },
    {
      "epoch": 9.782791137695312e-06,
      "model_forward_time": 1.7216527462005615,
      "step": 10258
    },
    {
      "epoch": 9.782791137695312e-06,
      "step": 10258,
      "training_step_time": 7.583214282989502
    },
    {
      "epoch": 9.783744812011719e-06,
      "model_forward_time": 1.7215712070465088,
      "step": 10259
    },
    {
      "epoch": 9.783744812011719e-06,
      "step": 10259,
      "training_step_time": 7.691785573959351
    },
    {
      "epoch": 9.784698486328125e-06,
      "grad_norm": 0.16961294412612915,
      "learning_rate": 9.924576967013142e-06,
      "loss": 0.0627,
      "step": 10260
    },
    {
      "epoch": 9.784698486328125e-06,
      "model_forward_time": 1.7112600803375244,
      "step": 10260
    },
    {
      "dynamics_loss_avg": 0.0370940089225769,
      "epoch": 9.784698486328125e-06,
      "step": 10260
    },
    {
      "action_loss_avg": 0.002566135325469077,
      "epoch": 9.784698486328125e-06,
      "step": 10260
    },
    {
      "epoch": 9.784698486328125e-06,
      "step": 10260,
      "training_step_time": 7.662601709365845
    },
    {
      "epoch": 9.78565216064453e-06,
      "model_forward_time": 1.6934525966644287,
      "step": 10261
    },
    {
      "epoch": 9.78565216064453e-06,
      "step": 10261,
      "training_step_time": 8.669944524765015
    },
    {
      "epoch": 9.786605834960938e-06,
      "model_forward_time": 1.7160537242889404,
      "step": 10262
    },
    {
      "epoch": 9.786605834960938e-06,
      "step": 10262,
      "training_step_time": 8.515332460403442
    },
    {
      "epoch": 9.787559509277344e-06,
      "model_forward_time": 2.2699122428894043,
      "step": 10263
    },
    {
      "epoch": 9.787559509277344e-06,
      "step": 10263,
      "training_step_time": 10.450754642486572
    },
    {
      "epoch": 9.78851318359375e-06,
      "model_forward_time": 1.7189879417419434,
      "step": 10264
    },
    {
      "epoch": 9.78851318359375e-06,
      "step": 10264,
      "training_step_time": 8.281813383102417
    },
    {
      "epoch": 9.789466857910157e-06,
      "model_forward_time": 1.7223725318908691,
      "step": 10265
    },
    {
      "epoch": 9.789466857910157e-06,
      "step": 10265,
      "training_step_time": 8.378028631210327
    },
    {
      "epoch": 9.790420532226563e-06,
      "model_forward_time": 1.72664213180542,
      "step": 10266
    },
    {
      "epoch": 9.790420532226563e-06,
      "step": 10266,
      "training_step_time": 8.675129890441895
    },
    {
      "epoch": 9.791374206542968e-06,
      "model_forward_time": 1.7137634754180908,
      "step": 10267
    },
    {
      "epoch": 9.791374206542968e-06,
      "step": 10267,
      "training_step_time": 7.580418825149536
    },
    {
      "epoch": 9.792327880859376e-06,
      "model_forward_time": 1.721165418624878,
      "step": 10268
    },
    {
      "epoch": 9.792327880859376e-06,
      "step": 10268,
      "training_step_time": 7.6861186027526855
    },
    {
      "epoch": 9.793281555175781e-06,
      "model_forward_time": 1.716949224472046,
      "step": 10269
    },
    {
      "epoch": 9.793281555175781e-06,
      "step": 10269,
      "training_step_time": 7.607090711593628
    },
    {
      "epoch": 9.794235229492187e-06,
      "grad_norm": 0.2751635015010834,
      "learning_rate": 9.924290586936887e-06,
      "loss": 0.0673,
      "step": 10270
    },
    {
      "epoch": 9.794235229492187e-06,
      "model_forward_time": 1.7320916652679443,
      "step": 10270
    },
    {
      "dynamics_loss_avg": 0.05852182181552053,
      "epoch": 9.794235229492187e-06,
      "step": 10270
    },
    {
      "action_loss_avg": 0.002477023471146822,
      "epoch": 9.794235229492187e-06,
      "step": 10270
    },
    {
      "epoch": 9.794235229492187e-06,
      "step": 10270,
      "training_step_time": 8.219326496124268
    },
    {
      "epoch": 9.795188903808595e-06,
      "model_forward_time": 1.7217066287994385,
      "step": 10271
    },
    {
      "epoch": 9.795188903808595e-06,
      "step": 10271,
      "training_step_time": 7.703320264816284
    },
    {
      "epoch": 9.796142578125e-06,
      "model_forward_time": 2.2737996578216553,
      "step": 10272
    },
    {
      "epoch": 9.796142578125e-06,
      "step": 10272,
      "training_step_time": 8.680006265640259
    },
    {
      "epoch": 9.797096252441406e-06,
      "model_forward_time": 1.7180323600769043,
      "step": 10273
    },
    {
      "epoch": 9.797096252441406e-06,
      "step": 10273,
      "training_step_time": 7.727816343307495
    },
    {
      "epoch": 9.798049926757812e-06,
      "model_forward_time": 1.766082763671875,
      "step": 10274
    },
    {
      "epoch": 9.798049926757812e-06,
      "step": 10274,
      "training_step_time": 8.369912385940552
    },
    {
      "epoch": 9.799003601074219e-06,
      "model_forward_time": 1.6908578872680664,
      "step": 10275
    },
    {
      "epoch": 9.799003601074219e-06,
      "step": 10275,
      "training_step_time": 7.539877891540527
    },
    {
      "epoch": 9.799957275390625e-06,
      "model_forward_time": 1.3246614933013916,
      "step": 10276
    },
    {
      "epoch": 9.799957275390625e-06,
      "step": 10276,
      "training_step_time": 7.354608058929443
    },
    {
      "epoch": 9.80091094970703e-06,
      "model_forward_time": 1.7020854949951172,
      "step": 10277
    },
    {
      "epoch": 9.80091094970703e-06,
      "step": 10277,
      "training_step_time": 7.670863389968872
    },
    {
      "epoch": 9.801864624023438e-06,
      "model_forward_time": 1.7113091945648193,
      "step": 10278
    },
    {
      "epoch": 9.801864624023438e-06,
      "step": 10278,
      "training_step_time": 7.543282747268677
    },
    {
      "epoch": 9.802818298339844e-06,
      "model_forward_time": 1.7019402980804443,
      "step": 10279
    },
    {
      "epoch": 9.802818298339844e-06,
      "step": 10279,
      "training_step_time": 7.509803056716919
    },
    {
      "epoch": 9.80377197265625e-06,
      "grad_norm": 0.25478512048721313,
      "learning_rate": 9.924003668347615e-06,
      "loss": 0.0708,
      "step": 10280
    },
    {
      "epoch": 9.80377197265625e-06,
      "model_forward_time": 1.7005870342254639,
      "step": 10280
    },
    {
      "dynamics_loss_avg": 0.06196454018354416,
      "epoch": 9.80377197265625e-06,
      "step": 10280
    },
    {
      "action_loss_avg": 0.0026626096107065676,
      "epoch": 9.80377197265625e-06,
      "step": 10280
    },
    {
      "epoch": 9.80377197265625e-06,
      "step": 10280,
      "training_step_time": 7.674895286560059
    },
    {
      "epoch": 9.804725646972657e-06,
      "model_forward_time": 2.19740891456604,
      "step": 10281
    },
    {
      "epoch": 9.804725646972657e-06,
      "step": 10281,
      "training_step_time": 8.580226182937622
    },
    {
      "epoch": 9.805679321289063e-06,
      "model_forward_time": 1.7032780647277832,
      "step": 10282
    },
    {
      "epoch": 9.805679321289063e-06,
      "step": 10282,
      "training_step_time": 8.494925498962402
    },
    {
      "epoch": 9.806632995605468e-06,
      "model_forward_time": 1.7439379692077637,
      "step": 10283
    },
    {
      "epoch": 9.806632995605468e-06,
      "step": 10283,
      "training_step_time": 7.608775615692139
    },
    {
      "epoch": 9.807586669921876e-06,
      "model_forward_time": 1.6902401447296143,
      "step": 10284
    },
    {
      "epoch": 9.807586669921876e-06,
      "step": 10284,
      "training_step_time": 8.526686668395996
    },
    {
      "epoch": 9.808540344238281e-06,
      "model_forward_time": 1.6897823810577393,
      "step": 10285
    },
    {
      "epoch": 9.808540344238281e-06,
      "step": 10285,
      "training_step_time": 7.6507408618927
    },
    {
      "epoch": 9.809494018554687e-06,
      "model_forward_time": 1.7031197547912598,
      "step": 10286
    },
    {
      "epoch": 9.809494018554687e-06,
      "step": 10286,
      "training_step_time": 7.673550367355347
    },
    {
      "epoch": 9.810447692871095e-06,
      "model_forward_time": 1.7052130699157715,
      "step": 10287
    },
    {
      "epoch": 9.810447692871095e-06,
      "step": 10287,
      "training_step_time": 7.613040447235107
    },
    {
      "epoch": 9.8114013671875e-06,
      "model_forward_time": 1.6976726055145264,
      "step": 10288
    },
    {
      "epoch": 9.8114013671875e-06,
      "step": 10288,
      "training_step_time": 7.451099157333374
    },
    {
      "epoch": 9.812355041503906e-06,
      "model_forward_time": 1.700906753540039,
      "step": 10289
    },
    {
      "epoch": 9.812355041503906e-06,
      "step": 10289,
      "training_step_time": 7.646056175231934
    },
    {
      "epoch": 9.813308715820312e-06,
      "grad_norm": 0.18656525015830994,
      "learning_rate": 9.923716211276705e-06,
      "loss": 0.0692,
      "step": 10290
    },
    {
      "epoch": 9.813308715820312e-06,
      "model_forward_time": 1.6895413398742676,
      "step": 10290
    },
    {
      "dynamics_loss_avg": 0.06982258744537831,
      "epoch": 9.813308715820312e-06,
      "step": 10290
    },
    {
      "action_loss_avg": 0.005829448299482465,
      "epoch": 9.813308715820312e-06,
      "step": 10290
    },
    {
      "epoch": 9.813308715820312e-06,
      "step": 10290,
      "training_step_time": 8.503625631332397
    },
    {
      "epoch": 9.814262390136719e-06,
      "model_forward_time": 2.225161075592041,
      "step": 10291
    },
    {
      "epoch": 9.814262390136719e-06,
      "step": 10291,
      "training_step_time": 9.58988070487976
    },
    {
      "epoch": 9.815216064453125e-06,
      "model_forward_time": 1.6968984603881836,
      "step": 10292
    },
    {
      "epoch": 9.815216064453125e-06,
      "step": 10292,
      "training_step_time": 8.420562028884888
    },
    {
      "epoch": 9.81616973876953e-06,
      "model_forward_time": 1.6946897506713867,
      "step": 10293
    },
    {
      "epoch": 9.81616973876953e-06,
      "step": 10293,
      "training_step_time": 6.737116575241089
    },
    {
      "epoch": 9.817123413085938e-06,
      "model_forward_time": 1.699305534362793,
      "step": 10294
    },
    {
      "epoch": 9.817123413085938e-06,
      "step": 10294,
      "training_step_time": 6.700053930282593
    },
    {
      "epoch": 9.818077087402344e-06,
      "model_forward_time": 1.7085001468658447,
      "step": 10295
    },
    {
      "epoch": 9.818077087402344e-06,
      "step": 10295,
      "training_step_time": 6.726433038711548
    },
    {
      "epoch": 9.81903076171875e-06,
      "model_forward_time": 1.7138617038726807,
      "step": 10296
    },
    {
      "epoch": 9.81903076171875e-06,
      "step": 10296,
      "training_step_time": 6.715163707733154
    },
    {
      "epoch": 9.819984436035157e-06,
      "model_forward_time": 1.7024405002593994,
      "step": 10297
    },
    {
      "epoch": 9.819984436035157e-06,
      "step": 10297,
      "training_step_time": 6.76462197303772
    },
    {
      "epoch": 9.820938110351563e-06,
      "model_forward_time": 1.7021551132202148,
      "step": 10298
    },
    {
      "epoch": 9.820938110351563e-06,
      "step": 10298,
      "training_step_time": 6.747089385986328
    },
    {
      "epoch": 9.821891784667968e-06,
      "model_forward_time": 1.6992595195770264,
      "step": 10299
    },
    {
      "epoch": 9.821891784667968e-06,
      "step": 10299,
      "training_step_time": 6.64992356300354
    },
    {
      "epoch": 9.822845458984376e-06,
      "grad_norm": 0.22143550217151642,
      "learning_rate": 9.923428215755594e-06,
      "loss": 0.0734,
      "step": 10300
    },
    {
      "epoch": 9.822845458984376e-06,
      "model_forward_time": 1.7185122966766357,
      "step": 10300
    },
    {
      "dynamics_loss_avg": 0.057087904773652554,
      "epoch": 9.822845458984376e-06,
      "step": 10300
    },
    {
      "action_loss_avg": 0.0029515514965169133,
      "epoch": 9.822845458984376e-06,
      "step": 10300
    },
    {
      "epoch": 9.822845458984376e-06,
      "step": 10300,
      "training_step_time": 6.690219163894653
    },
    {
      "epoch": 9.823799133300781e-06,
      "model_forward_time": 2.2735824584960938,
      "step": 10301
    },
    {
      "epoch": 9.823799133300781e-06,
      "step": 10301,
      "training_step_time": 8.162497758865356
    },
    {
      "epoch": 9.824752807617187e-06,
      "model_forward_time": 1.7242984771728516,
      "step": 10302
    },
    {
      "epoch": 9.824752807617187e-06,
      "step": 10302,
      "training_step_time": 7.691733121871948
    },
    {
      "epoch": 9.825706481933595e-06,
      "model_forward_time": 1.7168071269989014,
      "step": 10303
    },
    {
      "epoch": 9.825706481933595e-06,
      "step": 10303,
      "training_step_time": 6.712426424026489
    },
    {
      "epoch": 9.82666015625e-06,
      "model_forward_time": 1.7198514938354492,
      "step": 10304
    },
    {
      "epoch": 9.82666015625e-06,
      "step": 10304,
      "training_step_time": 6.761021137237549
    },
    {
      "epoch": 9.827613830566406e-06,
      "model_forward_time": 1.726625680923462,
      "step": 10305
    },
    {
      "epoch": 9.827613830566406e-06,
      "step": 10305,
      "training_step_time": 6.746046304702759
    },
    {
      "epoch": 9.828567504882812e-06,
      "model_forward_time": 1.7196190357208252,
      "step": 10306
    },
    {
      "epoch": 9.828567504882812e-06,
      "step": 10306,
      "training_step_time": 6.750687599182129
    },
    {
      "epoch": 9.829521179199219e-06,
      "model_forward_time": 1.7215886116027832,
      "step": 10307
    },
    {
      "epoch": 9.829521179199219e-06,
      "step": 10307,
      "training_step_time": 6.740792512893677
    },
    {
      "epoch": 9.830474853515625e-06,
      "model_forward_time": 1.7249016761779785,
      "step": 10308
    },
    {
      "epoch": 9.830474853515625e-06,
      "step": 10308,
      "training_step_time": 6.779435873031616
    },
    {
      "epoch": 9.83142852783203e-06,
      "model_forward_time": 1.721358060836792,
      "step": 10309
    },
    {
      "epoch": 9.83142852783203e-06,
      "step": 10309,
      "training_step_time": 6.685312509536743
    },
    {
      "epoch": 9.832382202148438e-06,
      "grad_norm": 0.2806704342365265,
      "learning_rate": 9.923139681815774e-06,
      "loss": 0.0688,
      "step": 10310
    },
    {
      "epoch": 9.832382202148438e-06,
      "model_forward_time": 2.2792341709136963,
      "step": 10310
    },
    {
      "dynamics_loss_avg": 0.05924908109009266,
      "epoch": 9.832382202148438e-06,
      "step": 10310
    },
    {
      "action_loss_avg": 0.0046835370012559,
      "epoch": 9.832382202148438e-06,
      "step": 10310
    },
    {
      "epoch": 9.832382202148438e-06,
      "step": 10310,
      "training_step_time": 8.326093912124634
    },
    {
      "epoch": 9.833335876464844e-06,
      "model_forward_time": 1.725052833557129,
      "step": 10311
    },
    {
      "epoch": 9.833335876464844e-06,
      "step": 10311,
      "training_step_time": 8.771795988082886
    },
    {
      "epoch": 9.83428955078125e-06,
      "model_forward_time": 1.717815637588501,
      "step": 10312
    },
    {
      "epoch": 9.83428955078125e-06,
      "step": 10312,
      "training_step_time": 7.1356518268585205
    },
    {
      "epoch": 9.835243225097657e-06,
      "model_forward_time": 1.7248761653900146,
      "step": 10313
    },
    {
      "epoch": 9.835243225097657e-06,
      "step": 10313,
      "training_step_time": 6.745878219604492
    },
    {
      "epoch": 9.836196899414063e-06,
      "model_forward_time": 1.3541369438171387,
      "step": 10314
    },
    {
      "epoch": 9.836196899414063e-06,
      "step": 10314,
      "training_step_time": 6.742387533187866
    },
    {
      "epoch": 9.837150573730468e-06,
      "model_forward_time": 1.7229657173156738,
      "step": 10315
    },
    {
      "epoch": 9.837150573730468e-06,
      "step": 10315,
      "training_step_time": 6.7600884437561035
    },
    {
      "epoch": 9.838104248046876e-06,
      "model_forward_time": 1.7229633331298828,
      "step": 10316
    },
    {
      "epoch": 9.838104248046876e-06,
      "step": 10316,
      "training_step_time": 6.763131856918335
    },
    {
      "epoch": 9.839057922363281e-06,
      "model_forward_time": 1.6886839866638184,
      "step": 10317
    },
    {
      "epoch": 9.839057922363281e-06,
      "step": 10317,
      "training_step_time": 6.7286601066589355
    },
    {
      "epoch": 9.840011596679687e-06,
      "model_forward_time": 1.6998636722564697,
      "step": 10318
    },
    {
      "epoch": 9.840011596679687e-06,
      "step": 10318,
      "training_step_time": 6.773107290267944
    },
    {
      "epoch": 9.840965270996095e-06,
      "model_forward_time": 2.245065927505493,
      "step": 10319
    },
    {
      "epoch": 9.840965270996095e-06,
      "step": 10319,
      "training_step_time": 7.488452672958374
    },
    {
      "epoch": 9.8419189453125e-06,
      "grad_norm": 0.2940528094768524,
      "learning_rate": 9.922850609488801e-06,
      "loss": 0.0642,
      "step": 10320
    },
    {
      "epoch": 9.8419189453125e-06,
      "model_forward_time": 1.6923682689666748,
      "step": 10320
    },
    {
      "dynamics_loss_avg": 0.06555248983204365,
      "epoch": 9.8419189453125e-06,
      "step": 10320
    },
    {
      "action_loss_avg": 0.0045199461630545555,
      "epoch": 9.8419189453125e-06,
      "step": 10320
    },
    {
      "epoch": 9.8419189453125e-06,
      "step": 10320,
      "training_step_time": 8.674367904663086
    },
    {
      "epoch": 9.842872619628906e-06,
      "model_forward_time": 1.7023742198944092,
      "step": 10321
    },
    {
      "epoch": 9.842872619628906e-06,
      "step": 10321,
      "training_step_time": 7.703194618225098
    },
    {
      "epoch": 9.843826293945312e-06,
      "model_forward_time": 1.7059004306793213,
      "step": 10322
    },
    {
      "epoch": 9.843826293945312e-06,
      "step": 10322,
      "training_step_time": 6.802875757217407
    },
    {
      "epoch": 9.844779968261719e-06,
      "model_forward_time": 1.695352554321289,
      "step": 10323
    },
    {
      "epoch": 9.844779968261719e-06,
      "step": 10323,
      "training_step_time": 6.797576189041138
    },
    {
      "epoch": 9.845733642578125e-06,
      "model_forward_time": 1.702446699142456,
      "step": 10324
    },
    {
      "epoch": 9.845733642578125e-06,
      "step": 10324,
      "training_step_time": 6.7776806354522705
    },
    {
      "epoch": 9.84668731689453e-06,
      "model_forward_time": 1.698317050933838,
      "step": 10325
    },
    {
      "epoch": 9.84668731689453e-06,
      "step": 10325,
      "training_step_time": 6.738331079483032
    },
    {
      "epoch": 9.847640991210938e-06,
      "model_forward_time": 1.6996653079986572,
      "step": 10326
    },
    {
      "epoch": 9.847640991210938e-06,
      "step": 10326,
      "training_step_time": 6.790343999862671
    },
    {
      "epoch": 9.848594665527344e-06,
      "model_forward_time": 1.7000319957733154,
      "step": 10327
    },
    {
      "epoch": 9.848594665527344e-06,
      "step": 10327,
      "training_step_time": 6.693155527114868
    },
    {
      "epoch": 9.84954833984375e-06,
      "model_forward_time": 2.2458386421203613,
      "step": 10328
    },
    {
      "epoch": 9.84954833984375e-06,
      "step": 10328,
      "training_step_time": 7.361738681793213
    },
    {
      "epoch": 9.850502014160157e-06,
      "model_forward_time": 1.7263336181640625,
      "step": 10329
    },
    {
      "epoch": 9.850502014160157e-06,
      "step": 10329,
      "training_step_time": 9.146684408187866
    },
    {
      "epoch": 9.851455688476563e-06,
      "grad_norm": 0.26515600085258484,
      "learning_rate": 9.922560998806287e-06,
      "loss": 0.0726,
      "step": 10330
    },
    {
      "epoch": 9.851455688476563e-06,
      "model_forward_time": 1.7282273769378662,
      "step": 10330
    },
    {
      "dynamics_loss_avg": 0.06753068286925554,
      "epoch": 9.851455688476563e-06,
      "step": 10330
    },
    {
      "action_loss_avg": 0.0034003338310867547,
      "epoch": 9.851455688476563e-06,
      "step": 10330
    },
    {
      "epoch": 9.851455688476563e-06,
      "step": 10330,
      "training_step_time": 8.5878267288208
    },
    {
      "epoch": 9.852409362792968e-06,
      "model_forward_time": 1.7236926555633545,
      "step": 10331
    },
    {
      "epoch": 9.852409362792968e-06,
      "step": 10331,
      "training_step_time": 6.817239999771118
    },
    {
      "epoch": 9.853363037109376e-06,
      "model_forward_time": 1.7209956645965576,
      "step": 10332
    },
    {
      "epoch": 9.853363037109376e-06,
      "step": 10332,
      "training_step_time": 6.7447991371154785
    },
    {
      "epoch": 9.854316711425781e-06,
      "model_forward_time": 1.7213513851165771,
      "step": 10333
    },
    {
      "epoch": 9.854316711425781e-06,
      "step": 10333,
      "training_step_time": 6.727540969848633
    },
    {
      "epoch": 9.855270385742187e-06,
      "model_forward_time": 1.714691162109375,
      "step": 10334
    },
    {
      "epoch": 9.855270385742187e-06,
      "step": 10334,
      "training_step_time": 6.743057727813721
    },
    {
      "epoch": 9.856224060058595e-06,
      "model_forward_time": 1.7239668369293213,
      "step": 10335
    },
    {
      "epoch": 9.856224060058595e-06,
      "step": 10335,
      "training_step_time": 6.73064661026001
    },
    {
      "epoch": 9.857177734375e-06,
      "model_forward_time": 1.3436548709869385,
      "step": 10336
    },
    {
      "epoch": 9.857177734375e-06,
      "step": 10336,
      "training_step_time": 6.800487518310547
    },
    {
      "epoch": 9.858131408691406e-06,
      "model_forward_time": 1.728133201599121,
      "step": 10337
    },
    {
      "epoch": 9.858131408691406e-06,
      "step": 10337,
      "training_step_time": 7.302105665206909
    },
    {
      "epoch": 9.859085083007812e-06,
      "model_forward_time": 2.260460376739502,
      "step": 10338
    },
    {
      "epoch": 9.859085083007812e-06,
      "step": 10338,
      "training_step_time": 8.401097774505615
    },
    {
      "epoch": 9.860038757324219e-06,
      "model_forward_time": 1.7211153507232666,
      "step": 10339
    },
    {
      "epoch": 9.860038757324219e-06,
      "step": 10339,
      "training_step_time": 8.782322883605957
    },
    {
      "epoch": 9.860992431640625e-06,
      "grad_norm": 0.18157905340194702,
      "learning_rate": 9.922270849799903e-06,
      "loss": 0.0682,
      "step": 10340
    },
    {
      "epoch": 9.860992431640625e-06,
      "model_forward_time": 1.7138159275054932,
      "step": 10340
    },
    {
      "dynamics_loss_avg": 0.061907804757356646,
      "epoch": 9.860992431640625e-06,
      "step": 10340
    },
    {
      "action_loss_avg": 0.004235376243013888,
      "epoch": 9.860992431640625e-06,
      "step": 10340
    },
    {
      "epoch": 9.860992431640625e-06,
      "step": 10340,
      "training_step_time": 7.705000162124634
    },
    {
      "epoch": 9.86194610595703e-06,
      "model_forward_time": 1.7219157218933105,
      "step": 10341
    },
    {
      "epoch": 9.86194610595703e-06,
      "step": 10341,
      "training_step_time": 6.8301780223846436
    },
    {
      "epoch": 9.862899780273438e-06,
      "model_forward_time": 1.7229530811309814,
      "step": 10342
    },
    {
      "epoch": 9.862899780273438e-06,
      "step": 10342,
      "training_step_time": 6.840785503387451
    },
    {
      "epoch": 9.863853454589844e-06,
      "model_forward_time": 1.7251472473144531,
      "step": 10343
    },
    {
      "epoch": 9.863853454589844e-06,
      "step": 10343,
      "training_step_time": 6.8419716358184814
    },
    {
      "epoch": 9.86480712890625e-06,
      "model_forward_time": 1.726355791091919,
      "step": 10344
    },
    {
      "epoch": 9.86480712890625e-06,
      "step": 10344,
      "training_step_time": 6.817612648010254
    },
    {
      "epoch": 9.865760803222657e-06,
      "model_forward_time": 1.6978237628936768,
      "step": 10345
    },
    {
      "epoch": 9.865760803222657e-06,
      "step": 10345,
      "training_step_time": 6.832871437072754
    },
    {
      "epoch": 9.866714477539063e-06,
      "model_forward_time": 1.7099788188934326,
      "step": 10346
    },
    {
      "epoch": 9.866714477539063e-06,
      "step": 10346,
      "training_step_time": 6.84595251083374
    },
    {
      "epoch": 9.867668151855468e-06,
      "model_forward_time": 2.2489962577819824,
      "step": 10347
    },
    {
      "epoch": 9.867668151855468e-06,
      "step": 10347,
      "training_step_time": 8.205344438552856
    },
    {
      "epoch": 9.868621826171876e-06,
      "model_forward_time": 1.7093112468719482,
      "step": 10348
    },
    {
      "epoch": 9.868621826171876e-06,
      "step": 10348,
      "training_step_time": 8.853917598724365
    },
    {
      "epoch": 9.869575500488281e-06,
      "model_forward_time": 1.7141783237457275,
      "step": 10349
    },
    {
      "epoch": 9.869575500488281e-06,
      "step": 10349,
      "training_step_time": 7.823159694671631
    },
    {
      "epoch": 9.870529174804687e-06,
      "grad_norm": 0.3645023703575134,
      "learning_rate": 9.921980162501382e-06,
      "loss": 0.0688,
      "step": 10350
    },
    {
      "epoch": 9.870529174804687e-06,
      "model_forward_time": 1.7050743103027344,
      "step": 10350
    },
    {
      "dynamics_loss_avg": 0.056690454855561255,
      "epoch": 9.870529174804687e-06,
      "step": 10350
    },
    {
      "action_loss_avg": 0.003941112745087594,
      "epoch": 9.870529174804687e-06,
      "step": 10350
    },
    {
      "epoch": 9.870529174804687e-06,
      "step": 10350,
      "training_step_time": 6.822520971298218
    },
    {
      "epoch": 9.871482849121095e-06,
      "model_forward_time": 1.7180542945861816,
      "step": 10351
    },
    {
      "epoch": 9.871482849121095e-06,
      "step": 10351,
      "training_step_time": 6.825663328170776
    },
    {
      "epoch": 9.8724365234375e-06,
      "model_forward_time": 1.7121453285217285,
      "step": 10352
    },
    {
      "epoch": 9.8724365234375e-06,
      "step": 10352,
      "training_step_time": 6.802645444869995
    },
    {
      "epoch": 9.873390197753906e-06,
      "model_forward_time": 1.6981477737426758,
      "step": 10353
    },
    {
      "epoch": 9.873390197753906e-06,
      "step": 10353,
      "training_step_time": 6.811119079589844
    },
    {
      "epoch": 9.874343872070312e-06,
      "model_forward_time": 1.7145755290985107,
      "step": 10354
    },
    {
      "epoch": 9.874343872070312e-06,
      "step": 10354,
      "training_step_time": 6.808243036270142
    },
    {
      "epoch": 9.87529754638672e-06,
      "model_forward_time": 1.7166728973388672,
      "step": 10355
    },
    {
      "epoch": 9.87529754638672e-06,
      "step": 10355,
      "training_step_time": 6.823984384536743
    },
    {
      "epoch": 9.876251220703125e-06,
      "model_forward_time": 2.2568445205688477,
      "step": 10356
    },
    {
      "epoch": 9.876251220703125e-06,
      "step": 10356,
      "training_step_time": 8.486342191696167
    },
    {
      "epoch": 9.87720489501953e-06,
      "model_forward_time": 1.7047832012176514,
      "step": 10357
    },
    {
      "epoch": 9.87720489501953e-06,
      "step": 10357,
      "training_step_time": 8.802667617797852
    },
    {
      "epoch": 9.878158569335938e-06,
      "model_forward_time": 1.706366777420044,
      "step": 10358
    },
    {
      "epoch": 9.878158569335938e-06,
      "step": 10358,
      "training_step_time": 7.8702778816223145
    },
    {
      "epoch": 9.879112243652344e-06,
      "model_forward_time": 1.7085902690887451,
      "step": 10359
    },
    {
      "epoch": 9.879112243652344e-06,
      "step": 10359,
      "training_step_time": 6.780101537704468
    },
    {
      "epoch": 9.88006591796875e-06,
      "grad_norm": 0.2024589329957962,
      "learning_rate": 9.921688936942507e-06,
      "loss": 0.0711,
      "step": 10360
    },
    {
      "epoch": 9.88006591796875e-06,
      "model_forward_time": 1.706648349761963,
      "step": 10360
    },
    {
      "dynamics_loss_avg": 0.050886370800435544,
      "epoch": 9.88006591796875e-06,
      "step": 10360
    },
    {
      "action_loss_avg": 0.004019565379712731,
      "epoch": 9.88006591796875e-06,
      "step": 10360
    },
    {
      "epoch": 9.88006591796875e-06,
      "step": 10360,
      "training_step_time": 6.771747350692749
    },
    {
      "epoch": 9.881019592285157e-06,
      "model_forward_time": 1.7117774486541748,
      "step": 10361
    },
    {
      "epoch": 9.881019592285157e-06,
      "step": 10361,
      "training_step_time": 6.7727210521698
    },
    {
      "epoch": 9.881973266601563e-06,
      "model_forward_time": 1.7179737091064453,
      "step": 10362
    },
    {
      "epoch": 9.881973266601563e-06,
      "step": 10362,
      "training_step_time": 6.794741868972778
    },
    {
      "epoch": 9.882926940917968e-06,
      "model_forward_time": 1.704683542251587,
      "step": 10363
    },
    {
      "epoch": 9.882926940917968e-06,
      "step": 10363,
      "training_step_time": 6.833228349685669
    },
    {
      "epoch": 9.883880615234376e-06,
      "model_forward_time": 1.7136008739471436,
      "step": 10364
    },
    {
      "epoch": 9.883880615234376e-06,
      "step": 10364,
      "training_step_time": 6.77485728263855
    },
    {
      "epoch": 9.884834289550781e-06,
      "model_forward_time": 2.254000663757324,
      "step": 10365
    },
    {
      "epoch": 9.884834289550781e-06,
      "step": 10365,
      "training_step_time": 7.525211811065674
    },
    {
      "epoch": 9.885787963867187e-06,
      "model_forward_time": 1.7084228992462158,
      "step": 10366
    },
    {
      "epoch": 9.885787963867187e-06,
      "step": 10366,
      "training_step_time": 8.839879035949707
    },
    {
      "epoch": 9.886741638183595e-06,
      "model_forward_time": 1.71219801902771,
      "step": 10367
    },
    {
      "epoch": 9.886741638183595e-06,
      "step": 10367,
      "training_step_time": 7.927159309387207
    },
    {
      "epoch": 9.8876953125e-06,
      "model_forward_time": 1.7103641033172607,
      "step": 10368
    },
    {
      "epoch": 9.8876953125e-06,
      "step": 10368,
      "training_step_time": 7.408226728439331
    },
    {
      "epoch": 9.888648986816406e-06,
      "model_forward_time": 1.7124826908111572,
      "step": 10369
    },
    {
      "epoch": 9.888648986816406e-06,
      "step": 10369,
      "training_step_time": 6.7888007164001465
    },
    {
      "epoch": 9.889602661132812e-06,
      "grad_norm": 0.26745253801345825,
      "learning_rate": 9.92139717315513e-06,
      "loss": 0.0671,
      "step": 10370
    },
    {
      "epoch": 9.889602661132812e-06,
      "model_forward_time": 1.7159950733184814,
      "step": 10370
    },
    {
      "dynamics_loss_avg": 0.05394478365778923,
      "epoch": 9.889602661132812e-06,
      "step": 10370
    },
    {
      "action_loss_avg": 0.0037143076187931,
      "epoch": 9.889602661132812e-06,
      "step": 10370
    },
    {
      "epoch": 9.889602661132812e-06,
      "step": 10370,
      "training_step_time": 6.779646873474121
    },
    {
      "epoch": 9.89055633544922e-06,
      "model_forward_time": 1.708890676498413,
      "step": 10371
    },
    {
      "epoch": 9.89055633544922e-06,
      "step": 10371,
      "training_step_time": 6.7332117557525635
    },
    {
      "epoch": 9.891510009765625e-06,
      "model_forward_time": 1.7121405601501465,
      "step": 10372
    },
    {
      "epoch": 9.891510009765625e-06,
      "step": 10372,
      "training_step_time": 6.669553995132446
    },
    {
      "epoch": 9.89246368408203e-06,
      "model_forward_time": 1.7208194732666016,
      "step": 10373
    },
    {
      "epoch": 9.89246368408203e-06,
      "step": 10373,
      "training_step_time": 6.725086212158203
    },
    {
      "epoch": 9.893417358398438e-06,
      "model_forward_time": 2.2697699069976807,
      "step": 10374
    },
    {
      "epoch": 9.893417358398438e-06,
      "step": 10374,
      "training_step_time": 7.457306385040283
    },
    {
      "epoch": 9.894371032714844e-06,
      "model_forward_time": 1.7266206741333008,
      "step": 10375
    },
    {
      "epoch": 9.894371032714844e-06,
      "step": 10375,
      "training_step_time": 9.517711400985718
    },
    {
      "epoch": 9.89532470703125e-06,
      "model_forward_time": 1.726076602935791,
      "step": 10376
    },
    {
      "epoch": 9.89532470703125e-06,
      "step": 10376,
      "training_step_time": 7.75489068031311
    },
    {
      "epoch": 9.896278381347657e-06,
      "model_forward_time": 1.724494218826294,
      "step": 10377
    },
    {
      "epoch": 9.896278381347657e-06,
      "step": 10377,
      "training_step_time": 7.366166353225708
    },
    {
      "epoch": 9.897232055664063e-06,
      "model_forward_time": 1.7293763160705566,
      "step": 10378
    },
    {
      "epoch": 9.897232055664063e-06,
      "step": 10378,
      "training_step_time": 6.756186246871948
    },
    {
      "epoch": 9.898185729980468e-06,
      "model_forward_time": 1.6975464820861816,
      "step": 10379
    },
    {
      "epoch": 9.898185729980468e-06,
      "step": 10379,
      "training_step_time": 6.750121116638184
    },
    {
      "epoch": 9.899139404296876e-06,
      "grad_norm": 0.19295988976955414,
      "learning_rate": 9.921104871171158e-06,
      "loss": 0.0718,
      "step": 10380
    },
    {
      "epoch": 9.899139404296876e-06,
      "model_forward_time": 1.6979856491088867,
      "step": 10380
    },
    {
      "dynamics_loss_avg": 0.06390890702605248,
      "epoch": 9.899139404296876e-06,
      "step": 10380
    },
    {
      "action_loss_avg": 0.004542706056963652,
      "epoch": 9.899139404296876e-06,
      "step": 10380
    },
    {
      "epoch": 9.899139404296876e-06,
      "step": 10380,
      "training_step_time": 6.743358373641968
    },
    {
      "epoch": 9.900093078613281e-06,
      "model_forward_time": 1.7048277854919434,
      "step": 10381
    },
    {
      "epoch": 9.900093078613281e-06,
      "step": 10381,
      "training_step_time": 6.757418394088745
    },
    {
      "epoch": 9.901046752929687e-06,
      "model_forward_time": 1.7117843627929688,
      "step": 10382
    },
    {
      "epoch": 9.901046752929687e-06,
      "step": 10382,
      "training_step_time": 6.7468085289001465
    },
    {
      "epoch": 9.902000427246093e-06,
      "model_forward_time": 1.7069244384765625,
      "step": 10383
    },
    {
      "epoch": 9.902000427246093e-06,
      "step": 10383,
      "training_step_time": 7.940342426300049
    },
    {
      "epoch": 9.9029541015625e-06,
      "model_forward_time": 2.232743740081787,
      "step": 10384
    },
    {
      "epoch": 9.9029541015625e-06,
      "step": 10384,
      "training_step_time": 9.37682557106018
    },
    {
      "epoch": 9.903907775878906e-06,
      "model_forward_time": 1.7130017280578613,
      "step": 10385
    },
    {
      "epoch": 9.903907775878906e-06,
      "step": 10385,
      "training_step_time": 8.696647882461548
    },
    {
      "epoch": 9.904861450195312e-06,
      "model_forward_time": 1.7117865085601807,
      "step": 10386
    },
    {
      "epoch": 9.904861450195312e-06,
      "step": 10386,
      "training_step_time": 7.679747104644775
    },
    {
      "epoch": 9.90581512451172e-06,
      "model_forward_time": 1.7240173816680908,
      "step": 10387
    },
    {
      "epoch": 9.90581512451172e-06,
      "step": 10387,
      "training_step_time": 6.713640451431274
    },
    {
      "epoch": 9.906768798828125e-06,
      "model_forward_time": 1.7255301475524902,
      "step": 10388
    },
    {
      "epoch": 9.906768798828125e-06,
      "step": 10388,
      "training_step_time": 6.723697900772095
    },
    {
      "epoch": 9.90772247314453e-06,
      "model_forward_time": 1.721282720565796,
      "step": 10389
    },
    {
      "epoch": 9.90772247314453e-06,
      "step": 10389,
      "training_step_time": 6.711075067520142
    },
    {
      "epoch": 9.908676147460938e-06,
      "grad_norm": 0.3042967915534973,
      "learning_rate": 9.920812031022556e-06,
      "loss": 0.0679,
      "step": 10390
    },
    {
      "epoch": 9.908676147460938e-06,
      "model_forward_time": 1.3500158786773682,
      "step": 10390
    },
    {
      "dynamics_loss_avg": 0.06935193575918674,
      "epoch": 9.908676147460938e-06,
      "step": 10390
    },
    {
      "action_loss_avg": 0.004608759540133178,
      "epoch": 9.908676147460938e-06,
      "step": 10390
    },
    {
      "epoch": 9.908676147460938e-06,
      "step": 10390,
      "training_step_time": 6.648577451705933
    },
    {
      "epoch": 9.909629821777344e-06,
      "model_forward_time": 1.725386381149292,
      "step": 10391
    },
    {
      "epoch": 9.909629821777344e-06,
      "step": 10391,
      "training_step_time": 6.725449085235596
    },
    {
      "epoch": 9.91058349609375e-06,
      "model_forward_time": 1.7267959117889404,
      "step": 10392
    },
    {
      "epoch": 9.91058349609375e-06,
      "step": 10392,
      "training_step_time": 8.205023050308228
    },
    {
      "epoch": 9.911537170410157e-06,
      "model_forward_time": 2.271818161010742,
      "step": 10393
    },
    {
      "epoch": 9.911537170410157e-06,
      "step": 10393,
      "training_step_time": 9.220462083816528
    },
    {
      "epoch": 9.912490844726563e-06,
      "model_forward_time": 1.729539394378662,
      "step": 10394
    },
    {
      "epoch": 9.912490844726563e-06,
      "step": 10394,
      "training_step_time": 8.316392660140991
    },
    {
      "epoch": 9.913444519042968e-06,
      "model_forward_time": 1.7368354797363281,
      "step": 10395
    },
    {
      "epoch": 9.913444519042968e-06,
      "step": 10395,
      "training_step_time": 7.8243327140808105
    },
    {
      "epoch": 9.914398193359376e-06,
      "model_forward_time": 1.7159817218780518,
      "step": 10396
    },
    {
      "epoch": 9.914398193359376e-06,
      "step": 10396,
      "training_step_time": 7.328526735305786
    },
    {
      "epoch": 9.915351867675781e-06,
      "model_forward_time": 1.7090404033660889,
      "step": 10397
    },
    {
      "epoch": 9.915351867675781e-06,
      "step": 10397,
      "training_step_time": 6.770293235778809
    },
    {
      "epoch": 9.916305541992187e-06,
      "model_forward_time": 1.705270528793335,
      "step": 10398
    },
    {
      "epoch": 9.916305541992187e-06,
      "step": 10398,
      "training_step_time": 6.760514974594116
    },
    {
      "epoch": 9.917259216308593e-06,
      "model_forward_time": 1.7098379135131836,
      "step": 10399
    },
    {
      "epoch": 9.917259216308593e-06,
      "step": 10399,
      "training_step_time": 6.774298906326294
    },
    {
      "epoch": 9.918212890625e-06,
      "grad_norm": 0.19777771830558777,
      "learning_rate": 9.920518652741348e-06,
      "loss": 0.0709,
      "step": 10400
    },
    {
      "epoch": 9.918212890625e-06,
      "model_forward_time": 1.7021028995513916,
      "step": 10400
    },
    {
      "dynamics_loss_avg": 0.05184831079095602,
      "epoch": 9.918212890625e-06,
      "step": 10400
    },
    {
      "action_loss_avg": 0.003122791842906736,
      "epoch": 9.918212890625e-06,
      "step": 10400
    },
    {
      "epoch": 9.918212890625e-06,
      "step": 10400,
      "training_step_time": 6.74381947517395
    },
    {
      "epoch": 9.919166564941406e-06,
      "model_forward_time": 1.7063319683074951,
      "step": 10401
    },
    {
      "epoch": 9.919166564941406e-06,
      "step": 10401,
      "training_step_time": 7.429734945297241
    },
    {
      "epoch": 9.920120239257812e-06,
      "model_forward_time": 2.249749183654785,
      "step": 10402
    },
    {
      "epoch": 9.920120239257812e-06,
      "step": 10402,
      "training_step_time": 8.276050567626953
    },
    {
      "epoch": 9.92107391357422e-06,
      "model_forward_time": 1.7109344005584717,
      "step": 10403
    },
    {
      "epoch": 9.92107391357422e-06,
      "step": 10403,
      "training_step_time": 10.355749130249023
    },
    {
      "epoch": 9.922027587890625e-06,
      "model_forward_time": 1.713153600692749,
      "step": 10404
    },
    {
      "epoch": 9.922027587890625e-06,
      "step": 10404,
      "training_step_time": 7.740217685699463
    },
    {
      "epoch": 9.92298126220703e-06,
      "model_forward_time": 1.7085204124450684,
      "step": 10405
    },
    {
      "epoch": 9.92298126220703e-06,
      "step": 10405,
      "training_step_time": 7.331691026687622
    },
    {
      "epoch": 9.923934936523438e-06,
      "model_forward_time": 1.7089228630065918,
      "step": 10406
    },
    {
      "epoch": 9.923934936523438e-06,
      "step": 10406,
      "training_step_time": 6.733636379241943
    },
    {
      "epoch": 9.924888610839844e-06,
      "model_forward_time": 1.7113409042358398,
      "step": 10407
    },
    {
      "epoch": 9.924888610839844e-06,
      "step": 10407,
      "training_step_time": 6.742316722869873
    },
    {
      "epoch": 9.92584228515625e-06,
      "model_forward_time": 1.710371971130371,
      "step": 10408
    },
    {
      "epoch": 9.92584228515625e-06,
      "step": 10408,
      "training_step_time": 6.744394779205322
    },
    {
      "epoch": 9.926795959472657e-06,
      "model_forward_time": 1.70973801612854,
      "step": 10409
    },
    {
      "epoch": 9.926795959472657e-06,
      "step": 10409,
      "training_step_time": 6.731295824050903
    },
    {
      "epoch": 9.927749633789063e-06,
      "grad_norm": 0.2047884464263916,
      "learning_rate": 9.920224736359618e-06,
      "loss": 0.0694,
      "step": 10410
    },
    {
      "epoch": 9.927749633789063e-06,
      "model_forward_time": 1.7073616981506348,
      "step": 10410
    },
    {
      "dynamics_loss_avg": 0.06504562627524138,
      "epoch": 9.927749633789063e-06,
      "step": 10410
    },
    {
      "action_loss_avg": 0.004870437807403505,
      "epoch": 9.927749633789063e-06,
      "step": 10410
    },
    {
      "epoch": 9.927749633789063e-06,
      "step": 10410,
      "training_step_time": 7.4778220653533936
    },
    {
      "epoch": 9.928703308105468e-06,
      "model_forward_time": 2.262209177017212,
      "step": 10411
    },
    {
      "epoch": 9.928703308105468e-06,
      "step": 10411,
      "training_step_time": 7.616860866546631
    },
    {
      "epoch": 9.929656982421876e-06,
      "model_forward_time": 1.707158088684082,
      "step": 10412
    },
    {
      "epoch": 9.929656982421876e-06,
      "step": 10412,
      "training_step_time": 9.262242794036865
    },
    {
      "epoch": 9.930610656738281e-06,
      "model_forward_time": 1.7136173248291016,
      "step": 10413
    },
    {
      "epoch": 9.930610656738281e-06,
      "step": 10413,
      "training_step_time": 7.781172037124634
    },
    {
      "epoch": 9.931564331054687e-06,
      "model_forward_time": 1.7097408771514893,
      "step": 10414
    },
    {
      "epoch": 9.931564331054687e-06,
      "step": 10414,
      "training_step_time": 7.378901958465576
    },
    {
      "epoch": 9.932518005371093e-06,
      "model_forward_time": 1.7138826847076416,
      "step": 10415
    },
    {
      "epoch": 9.932518005371093e-06,
      "step": 10415,
      "training_step_time": 6.7274558544158936
    },
    {
      "epoch": 9.9334716796875e-06,
      "model_forward_time": 1.7094194889068604,
      "step": 10416
    },
    {
      "epoch": 9.9334716796875e-06,
      "step": 10416,
      "training_step_time": 6.718259572982788
    },
    {
      "epoch": 9.934425354003906e-06,
      "model_forward_time": 1.7094919681549072,
      "step": 10417
    },
    {
      "epoch": 9.934425354003906e-06,
      "step": 10417,
      "training_step_time": 6.736142158508301
    },
    {
      "epoch": 9.935379028320312e-06,
      "model_forward_time": 1.7154083251953125,
      "step": 10418
    },
    {
      "epoch": 9.935379028320312e-06,
      "step": 10418,
      "training_step_time": 6.69878363609314
    },
    {
      "epoch": 9.93633270263672e-06,
      "model_forward_time": 1.707380771636963,
      "step": 10419
    },
    {
      "epoch": 9.93633270263672e-06,
      "step": 10419,
      "training_step_time": 7.376518487930298
    },
    {
      "epoch": 9.937286376953125e-06,
      "grad_norm": 0.22330088913440704,
      "learning_rate": 9.91993028190951e-06,
      "loss": 0.0644,
      "step": 10420
    },
    {
      "epoch": 9.937286376953125e-06,
      "model_forward_time": 1.7102408409118652,
      "step": 10420
    },
    {
      "dynamics_loss_avg": 0.0658145310357213,
      "epoch": 9.937286376953125e-06,
      "step": 10420
    },
    {
      "action_loss_avg": 0.0034162997850216924,
      "epoch": 9.937286376953125e-06,
      "step": 10420
    },
    {
      "epoch": 9.937286376953125e-06,
      "step": 10420,
      "training_step_time": 8.018157243728638
    },
    {
      "epoch": 9.93824005126953e-06,
      "model_forward_time": 1.709294319152832,
      "step": 10421
    },
    {
      "epoch": 9.93824005126953e-06,
      "step": 10421,
      "training_step_time": 8.35606074333191
    },
    {
      "epoch": 9.939193725585938e-06,
      "model_forward_time": 1.7079846858978271,
      "step": 10422
    },
    {
      "epoch": 9.939193725585938e-06,
      "step": 10422,
      "training_step_time": 8.800505638122559
    },
    {
      "epoch": 9.940147399902344e-06,
      "model_forward_time": 1.7161791324615479,
      "step": 10423
    },
    {
      "epoch": 9.940147399902344e-06,
      "step": 10423,
      "training_step_time": 7.569627523422241
    },
    {
      "epoch": 9.94110107421875e-06,
      "model_forward_time": 1.7125916481018066,
      "step": 10424
    },
    {
      "epoch": 9.94110107421875e-06,
      "step": 10424,
      "training_step_time": 6.732660531997681
    },
    {
      "epoch": 9.942054748535157e-06,
      "model_forward_time": 1.7072641849517822,
      "step": 10425
    },
    {
      "epoch": 9.942054748535157e-06,
      "step": 10425,
      "training_step_time": 6.709134817123413
    },
    {
      "epoch": 9.943008422851563e-06,
      "model_forward_time": 1.7113733291625977,
      "step": 10426
    },
    {
      "epoch": 9.943008422851563e-06,
      "step": 10426,
      "training_step_time": 6.7181713581085205
    },
    {
      "epoch": 9.943962097167968e-06,
      "model_forward_time": 1.7122457027435303,
      "step": 10427
    },
    {
      "epoch": 9.943962097167968e-06,
      "step": 10427,
      "training_step_time": 6.751939058303833
    },
    {
      "epoch": 9.944915771484376e-06,
      "model_forward_time": 1.715247631072998,
      "step": 10428
    },
    {
      "epoch": 9.944915771484376e-06,
      "step": 10428,
      "training_step_time": 7.3673717975616455
    },
    {
      "epoch": 9.945869445800782e-06,
      "model_forward_time": 1.7104768753051758,
      "step": 10429
    },
    {
      "epoch": 9.945869445800782e-06,
      "step": 10429,
      "training_step_time": 7.53443717956543
    },
    {
      "epoch": 9.946823120117187e-06,
      "grad_norm": 0.19311538338661194,
      "learning_rate": 9.919635289423221e-06,
      "loss": 0.072,
      "step": 10430
    },
    {
      "epoch": 9.946823120117187e-06,
      "model_forward_time": 2.2376229763031006,
      "step": 10430
    },
    {
      "dynamics_loss_avg": 0.07054400891065597,
      "epoch": 9.946823120117187e-06,
      "step": 10430
    },
    {
      "action_loss_avg": 0.003084555338136852,
      "epoch": 9.946823120117187e-06,
      "step": 10430
    },
    {
      "epoch": 9.946823120117187e-06,
      "step": 10430,
      "training_step_time": 9.103311777114868
    },
    {
      "epoch": 9.947776794433593e-06,
      "model_forward_time": 1.7070040702819824,
      "step": 10431
    },
    {
      "epoch": 9.947776794433593e-06,
      "step": 10431,
      "training_step_time": 10.942139625549316
    },
    {
      "epoch": 9.94873046875e-06,
      "model_forward_time": 1.7077724933624268,
      "step": 10432
    },
    {
      "epoch": 9.94873046875e-06,
      "step": 10432,
      "training_step_time": 7.64979100227356
    },
    {
      "epoch": 9.949684143066406e-06,
      "model_forward_time": 1.7124137878417969,
      "step": 10433
    },
    {
      "epoch": 9.949684143066406e-06,
      "step": 10433,
      "training_step_time": 6.7630393505096436
    },
    {
      "epoch": 9.950637817382812e-06,
      "model_forward_time": 1.7070488929748535,
      "step": 10434
    },
    {
      "epoch": 9.950637817382812e-06,
      "step": 10434,
      "training_step_time": 6.770773887634277
    },
    {
      "epoch": 9.95159149169922e-06,
      "model_forward_time": 1.7108731269836426,
      "step": 10435
    },
    {
      "epoch": 9.95159149169922e-06,
      "step": 10435,
      "training_step_time": 6.771506071090698
    },
    {
      "epoch": 9.952545166015625e-06,
      "model_forward_time": 1.706617832183838,
      "step": 10436
    },
    {
      "epoch": 9.952545166015625e-06,
      "step": 10436,
      "training_step_time": 6.779539585113525
    },
    {
      "epoch": 9.95349884033203e-06,
      "model_forward_time": 1.7051115036010742,
      "step": 10437
    },
    {
      "epoch": 9.95349884033203e-06,
      "step": 10437,
      "training_step_time": 6.7832841873168945
    },
    {
      "epoch": 9.954452514648438e-06,
      "model_forward_time": 1.7095365524291992,
      "step": 10438
    },
    {
      "epoch": 9.954452514648438e-06,
      "step": 10438,
      "training_step_time": 7.569257974624634
    },
    {
      "epoch": 9.955406188964844e-06,
      "model_forward_time": 1.9358890056610107,
      "step": 10439
    },
    {
      "epoch": 9.955406188964844e-06,
      "step": 10439,
      "training_step_time": 8.157660007476807
    },
    {
      "epoch": 9.95635986328125e-06,
      "grad_norm": 0.20749853551387787,
      "learning_rate": 9.919339758933015e-06,
      "loss": 0.0703,
      "step": 10440
    },
    {
      "epoch": 9.95635986328125e-06,
      "model_forward_time": 1.7020339965820312,
      "step": 10440
    },
    {
      "dynamics_loss_avg": 0.06353312339633703,
      "epoch": 9.95635986328125e-06,
      "step": 10440
    },
    {
      "action_loss_avg": 0.0030166568467393517,
      "epoch": 9.95635986328125e-06,
      "step": 10440
    },
    {
      "epoch": 9.95635986328125e-06,
      "step": 10440,
      "training_step_time": 9.268615245819092
    },
    {
      "epoch": 9.957313537597657e-06,
      "model_forward_time": 1.7097337245941162,
      "step": 10441
    },
    {
      "epoch": 9.957313537597657e-06,
      "step": 10441,
      "training_step_time": 7.722432613372803
    },
    {
      "epoch": 9.958267211914063e-06,
      "model_forward_time": 1.7130126953125,
      "step": 10442
    },
    {
      "epoch": 9.958267211914063e-06,
      "step": 10442,
      "training_step_time": 7.4259583950042725
    },
    {
      "epoch": 9.959220886230468e-06,
      "model_forward_time": 1.7136726379394531,
      "step": 10443
    },
    {
      "epoch": 9.959220886230468e-06,
      "step": 10443,
      "training_step_time": 6.798674583435059
    },
    {
      "epoch": 9.960174560546876e-06,
      "model_forward_time": 1.715057373046875,
      "step": 10444
    },
    {
      "epoch": 9.960174560546876e-06,
      "step": 10444,
      "training_step_time": 6.813103675842285
    },
    {
      "epoch": 9.961128234863282e-06,
      "model_forward_time": 1.7070422172546387,
      "step": 10445
    },
    {
      "epoch": 9.961128234863282e-06,
      "step": 10445,
      "training_step_time": 6.777746915817261
    },
    {
      "epoch": 9.962081909179687e-06,
      "model_forward_time": 1.7108128070831299,
      "step": 10446
    },
    {
      "epoch": 9.962081909179687e-06,
      "step": 10446,
      "training_step_time": 6.804595470428467
    },
    {
      "epoch": 9.963035583496093e-06,
      "model_forward_time": 1.7118244171142578,
      "step": 10447
    },
    {
      "epoch": 9.963035583496093e-06,
      "step": 10447,
      "training_step_time": 7.674795389175415
    },
    {
      "epoch": 9.9639892578125e-06,
      "model_forward_time": 2.252382278442383,
      "step": 10448
    },
    {
      "epoch": 9.9639892578125e-06,
      "step": 10448,
      "training_step_time": 8.56453251838684
    },
    {
      "epoch": 9.964942932128906e-06,
      "model_forward_time": 1.7064156532287598,
      "step": 10449
    },
    {
      "epoch": 9.964942932128906e-06,
      "step": 10449,
      "training_step_time": 8.567401647567749
    },
    {
      "epoch": 9.965896606445312e-06,
      "grad_norm": 0.254810094833374,
      "learning_rate": 9.91904369047121e-06,
      "loss": 0.0698,
      "step": 10450
    },
    {
      "epoch": 9.965896606445312e-06,
      "model_forward_time": 1.336918830871582,
      "step": 10450
    },
    {
      "dynamics_loss_avg": 0.058782149106264114,
      "epoch": 9.965896606445312e-06,
      "step": 10450
    },
    {
      "action_loss_avg": 0.004719408135861159,
      "epoch": 9.965896606445312e-06,
      "step": 10450
    },
    {
      "epoch": 9.965896606445312e-06,
      "step": 10450,
      "training_step_time": 7.682440280914307
    },
    {
      "epoch": 9.96685028076172e-06,
      "model_forward_time": 1.7097883224487305,
      "step": 10451
    },
    {
      "epoch": 9.96685028076172e-06,
      "step": 10451,
      "training_step_time": 8.399696588516235
    },
    {
      "epoch": 9.967803955078125e-06,
      "model_forward_time": 1.7046070098876953,
      "step": 10452
    },
    {
      "epoch": 9.967803955078125e-06,
      "step": 10452,
      "training_step_time": 6.755794286727905
    },
    {
      "epoch": 9.96875762939453e-06,
      "model_forward_time": 1.709052562713623,
      "step": 10453
    },
    {
      "epoch": 9.96875762939453e-06,
      "step": 10453,
      "training_step_time": 6.73105525970459
    },
    {
      "epoch": 9.969711303710938e-06,
      "model_forward_time": 1.713111400604248,
      "step": 10454
    },
    {
      "epoch": 9.969711303710938e-06,
      "step": 10454,
      "training_step_time": 6.736587047576904
    },
    {
      "epoch": 9.970664978027344e-06,
      "model_forward_time": 1.7131438255310059,
      "step": 10455
    },
    {
      "epoch": 9.970664978027344e-06,
      "step": 10455,
      "training_step_time": 6.748608589172363
    },
    {
      "epoch": 9.97161865234375e-06,
      "model_forward_time": 1.7124738693237305,
      "step": 10456
    },
    {
      "epoch": 9.97161865234375e-06,
      "step": 10456,
      "training_step_time": 7.434521436691284
    },
    {
      "epoch": 9.972572326660157e-06,
      "model_forward_time": 1.7237892150878906,
      "step": 10457
    },
    {
      "epoch": 9.972572326660157e-06,
      "step": 10457,
      "training_step_time": 8.913088083267212
    },
    {
      "epoch": 9.973526000976563e-06,
      "model_forward_time": 1.6972990036010742,
      "step": 10458
    },
    {
      "epoch": 9.973526000976563e-06,
      "step": 10458,
      "training_step_time": 10.136998176574707
    },
    {
      "epoch": 9.974479675292968e-06,
      "model_forward_time": 1.708690881729126,
      "step": 10459
    },
    {
      "epoch": 9.974479675292968e-06,
      "step": 10459,
      "training_step_time": 8.80734896659851
    },
    {
      "epoch": 9.975433349609376e-06,
      "grad_norm": 0.1929270476102829,
      "learning_rate": 9.91874708407018e-06,
      "loss": 0.0683,
      "step": 10460
    },
    {
      "epoch": 9.975433349609376e-06,
      "model_forward_time": 1.704545021057129,
      "step": 10460
    },
    {
      "dynamics_loss_avg": 0.06326213404536248,
      "epoch": 9.975433349609376e-06,
      "step": 10460
    },
    {
      "action_loss_avg": 0.0042294787825085224,
      "epoch": 9.975433349609376e-06,
      "step": 10460
    },
    {
      "epoch": 9.975433349609376e-06,
      "step": 10460,
      "training_step_time": 7.35695219039917
    },
    {
      "epoch": 9.976387023925782e-06,
      "model_forward_time": 1.7084534168243408,
      "step": 10461
    },
    {
      "epoch": 9.976387023925782e-06,
      "step": 10461,
      "training_step_time": 7.181470155715942
    },
    {
      "epoch": 9.977340698242187e-06,
      "model_forward_time": 1.7084355354309082,
      "step": 10462
    },
    {
      "epoch": 9.977340698242187e-06,
      "step": 10462,
      "training_step_time": 6.66486382484436
    },
    {
      "epoch": 9.978294372558593e-06,
      "model_forward_time": 1.7048237323760986,
      "step": 10463
    },
    {
      "epoch": 9.978294372558593e-06,
      "step": 10463,
      "training_step_time": 6.706410884857178
    },
    {
      "epoch": 9.979248046875e-06,
      "model_forward_time": 1.702329397201538,
      "step": 10464
    },
    {
      "epoch": 9.979248046875e-06,
      "step": 10464,
      "training_step_time": 6.729118824005127
    },
    {
      "epoch": 9.980201721191406e-06,
      "model_forward_time": 1.7106730937957764,
      "step": 10465
    },
    {
      "epoch": 9.980201721191406e-06,
      "step": 10465,
      "training_step_time": 7.367269277572632
    },
    {
      "epoch": 9.981155395507812e-06,
      "model_forward_time": 1.7030248641967773,
      "step": 10466
    },
    {
      "epoch": 9.981155395507812e-06,
      "step": 10466,
      "training_step_time": 8.061230421066284
    },
    {
      "epoch": 9.98210906982422e-06,
      "model_forward_time": 2.225572347640991,
      "step": 10467
    },
    {
      "epoch": 9.98210906982422e-06,
      "step": 10467,
      "training_step_time": 12.105550527572632
    },
    {
      "epoch": 9.983062744140625e-06,
      "model_forward_time": 1.715134859085083,
      "step": 10468
    },
    {
      "epoch": 9.983062744140625e-06,
      "step": 10468,
      "training_step_time": 7.768100023269653
    },
    {
      "epoch": 9.98401641845703e-06,
      "model_forward_time": 1.715144395828247,
      "step": 10469
    },
    {
      "epoch": 9.98401641845703e-06,
      "step": 10469,
      "training_step_time": 7.56273078918457
    },
    {
      "epoch": 9.984970092773438e-06,
      "grad_norm": 0.22513096034526825,
      "learning_rate": 9.918449939762368e-06,
      "loss": 0.0671,
      "step": 10470
    },
    {
      "epoch": 9.984970092773438e-06,
      "model_forward_time": 1.7028045654296875,
      "step": 10470
    },
    {
      "dynamics_loss_avg": 0.0493816701695323,
      "epoch": 9.984970092773438e-06,
      "step": 10470
    },
    {
      "action_loss_avg": 0.0032339604338631036,
      "epoch": 9.984970092773438e-06,
      "step": 10470
    },
    {
      "epoch": 9.984970092773438e-06,
      "step": 10470,
      "training_step_time": 7.465051651000977
    },
    {
      "epoch": 9.985923767089844e-06,
      "model_forward_time": 1.7108819484710693,
      "step": 10471
    },
    {
      "epoch": 9.985923767089844e-06,
      "step": 10471,
      "training_step_time": 6.733527898788452
    },
    {
      "epoch": 9.98687744140625e-06,
      "model_forward_time": 0.9922866821289062,
      "step": 10472
    },
    {
      "epoch": 9.98687744140625e-06,
      "step": 10472,
      "training_step_time": 6.745668411254883
    },
    {
      "epoch": 9.987831115722657e-06,
      "model_forward_time": 1.718735933303833,
      "step": 10473
    },
    {
      "epoch": 9.987831115722657e-06,
      "step": 10473,
      "training_step_time": 6.740720510482788
    },
    {
      "epoch": 9.988784790039063e-06,
      "model_forward_time": 1.73655366897583,
      "step": 10474
    },
    {
      "epoch": 9.988784790039063e-06,
      "step": 10474,
      "training_step_time": 7.3872315883636475
    },
    {
      "epoch": 9.989738464355468e-06,
      "model_forward_time": 1.3510193824768066,
      "step": 10475
    },
    {
      "epoch": 9.989738464355468e-06,
      "step": 10475,
      "training_step_time": 7.520098447799683
    },
    {
      "epoch": 9.990692138671876e-06,
      "model_forward_time": 2.3013479709625244,
      "step": 10476
    },
    {
      "epoch": 9.990692138671876e-06,
      "step": 10476,
      "training_step_time": 10.232473373413086
    },
    {
      "epoch": 9.991645812988282e-06,
      "model_forward_time": 1.729330062866211,
      "step": 10477
    },
    {
      "epoch": 9.991645812988282e-06,
      "step": 10477,
      "training_step_time": 7.816516876220703
    },
    {
      "epoch": 9.992599487304687e-06,
      "model_forward_time": 1.731365442276001,
      "step": 10478
    },
    {
      "epoch": 9.992599487304687e-06,
      "step": 10478,
      "training_step_time": 7.631901025772095
    },
    {
      "epoch": 9.993553161621093e-06,
      "model_forward_time": 1.7070012092590332,
      "step": 10479
    },
    {
      "epoch": 9.993553161621093e-06,
      "step": 10479,
      "training_step_time": 7.230354309082031
    },
    {
      "epoch": 9.9945068359375e-06,
      "grad_norm": 0.1978931576013565,
      "learning_rate": 9.918152257580261e-06,
      "loss": 0.0679,
      "step": 10480
    },
    {
      "epoch": 9.9945068359375e-06,
      "model_forward_time": 1.707150936126709,
      "step": 10480
    },
    {
      "dynamics_loss_avg": 0.06522790361195803,
      "epoch": 9.9945068359375e-06,
      "step": 10480
    },
    {
      "action_loss_avg": 0.0029617862543091177,
      "epoch": 9.9945068359375e-06,
      "step": 10480
    },
    {
      "epoch": 9.9945068359375e-06,
      "step": 10480,
      "training_step_time": 6.756873846054077
    },
    {
      "epoch": 9.995460510253906e-06,
      "model_forward_time": 1.7090575695037842,
      "step": 10481
    },
    {
      "epoch": 9.995460510253906e-06,
      "step": 10481,
      "training_step_time": 6.724897623062134
    },
    {
      "epoch": 9.996414184570312e-06,
      "model_forward_time": 1.7131121158599854,
      "step": 10482
    },
    {
      "epoch": 9.996414184570312e-06,
      "step": 10482,
      "training_step_time": 6.75308895111084
    },
    {
      "epoch": 9.99736785888672e-06,
      "model_forward_time": 1.7106668949127197,
      "step": 10483
    },
    {
      "epoch": 9.99736785888672e-06,
      "step": 10483,
      "training_step_time": 6.725566864013672
    },
    {
      "epoch": 9.998321533203125e-06,
      "model_forward_time": 1.7076876163482666,
      "step": 10484
    },
    {
      "epoch": 9.998321533203125e-06,
      "step": 10484,
      "training_step_time": 7.686141490936279
    },
    {
      "epoch": 9.99927520751953e-06,
      "model_forward_time": 2.264831066131592,
      "step": 10485
    },
    {
      "epoch": 9.99927520751953e-06,
      "step": 10485,
      "training_step_time": 9.348031759262085
    },
    {
      "epoch": 1.0000228881835938e-05,
      "model_forward_time": 1.7111625671386719,
      "step": 10486
    },
    {
      "epoch": 1.0000228881835938e-05,
      "step": 10486,
      "training_step_time": 8.827728986740112
    },
    {
      "epoch": 1.0001182556152344e-05,
      "model_forward_time": 1.7094931602478027,
      "step": 10487
    },
    {
      "epoch": 1.0001182556152344e-05,
      "step": 10487,
      "training_step_time": 7.624938726425171
    },
    {
      "epoch": 1.000213623046875e-05,
      "model_forward_time": 1.7112979888916016,
      "step": 10488
    },
    {
      "epoch": 1.000213623046875e-05,
      "step": 10488,
      "training_step_time": 7.447293519973755
    },
    {
      "epoch": 1.0003089904785157e-05,
      "model_forward_time": 1.7121963500976562,
      "step": 10489
    },
    {
      "epoch": 1.0003089904785157e-05,
      "step": 10489,
      "training_step_time": 6.747017860412598
    },
    {
      "epoch": 1.0004043579101563e-05,
      "grad_norm": 0.18064150214195251,
      "learning_rate": 9.917854037556419e-06,
      "loss": 0.0642,
      "step": 10490
    },
    {
      "epoch": 1.0004043579101563e-05,
      "model_forward_time": 1.7052996158599854,
      "step": 10490
    },
    {
      "dynamics_loss_avg": 0.04749462436884642,
      "epoch": 1.0004043579101563e-05,
      "step": 10490
    },
    {
      "action_loss_avg": 0.0025311089353635906,
      "epoch": 1.0004043579101563e-05,
      "step": 10490
    },
    {
      "epoch": 1.0004043579101563e-05,
      "step": 10490,
      "training_step_time": 6.74463415145874
    },
    {
      "epoch": 1.0004997253417968e-05,
      "model_forward_time": 1.70937180519104,
      "step": 10491
    },
    {
      "epoch": 1.0004997253417968e-05,
      "step": 10491,
      "training_step_time": 6.784616231918335
    },
    {
      "epoch": 1.0005950927734376e-05,
      "model_forward_time": 1.7076544761657715,
      "step": 10492
    },
    {
      "epoch": 1.0005950927734376e-05,
      "step": 10492,
      "training_step_time": 6.714688539505005
    },
    {
      "epoch": 1.0006904602050782e-05,
      "model_forward_time": 1.708373785018921,
      "step": 10493
    },
    {
      "epoch": 1.0006904602050782e-05,
      "step": 10493,
      "training_step_time": 7.584177494049072
    },
    {
      "epoch": 1.0007858276367187e-05,
      "model_forward_time": 1.7563881874084473,
      "step": 10494
    },
    {
      "epoch": 1.0007858276367187e-05,
      "step": 10494,
      "training_step_time": 8.061529397964478
    },
    {
      "epoch": 1.0008811950683593e-05,
      "model_forward_time": 1.8368113040924072,
      "step": 10495
    },
    {
      "epoch": 1.0008811950683593e-05,
      "step": 10495,
      "training_step_time": 10.573985815048218
    },
    {
      "epoch": 1.0009765625e-05,
      "model_forward_time": 1.7091083526611328,
      "step": 10496
    },
    {
      "epoch": 1.0009765625e-05,
      "step": 10496,
      "training_step_time": 7.808662176132202
    },
    {
      "epoch": 1.0010719299316406e-05,
      "model_forward_time": 1.7062742710113525,
      "step": 10497
    },
    {
      "epoch": 1.0010719299316406e-05,
      "step": 10497,
      "training_step_time": 7.4857096672058105
    },
    {
      "epoch": 1.0011672973632812e-05,
      "model_forward_time": 1.7147340774536133,
      "step": 10498
    },
    {
      "epoch": 1.0011672973632812e-05,
      "step": 10498,
      "training_step_time": 7.213657379150391
    },
    {
      "epoch": 1.001262664794922e-05,
      "model_forward_time": 1.7145020961761475,
      "step": 10499
    },
    {
      "epoch": 1.001262664794922e-05,
      "step": 10499,
      "training_step_time": 6.695120334625244
    },
    {
      "epoch": 1.0013580322265625e-05,
      "grad_norm": 0.21349747478961945,
      "learning_rate": 9.917555279723454e-06,
      "loss": 0.0639,
      "step": 10500
    },
    {
      "epoch": 1.0013580322265625e-05,
      "model_forward_time": 1.703202247619629,
      "step": 10500
    },
    {
      "dynamics_loss_avg": 0.06061666142195463,
      "epoch": 1.0013580322265625e-05,
      "step": 10500
    },
    {
      "action_loss_avg": 0.004533399734646082,
      "epoch": 1.0013580322265625e-05,
      "step": 10500
    },
    {
      "epoch": 1.0013580322265625e-05,
      "step": 10500,
      "training_step_time": 6.672126054763794
    },
    {
      "epoch": 1.001453399658203e-05,
      "model_forward_time": 1.7061121463775635,
      "step": 10501
    },
    {
      "epoch": 1.001453399658203e-05,
      "step": 10501,
      "training_step_time": 6.713491916656494
    },
    {
      "epoch": 1.0015487670898438e-05,
      "model_forward_time": 1.7066714763641357,
      "step": 10502
    },
    {
      "epoch": 1.0015487670898438e-05,
      "step": 10502,
      "training_step_time": 9.125003576278687
    },
    {
      "epoch": 1.0016441345214844e-05,
      "model_forward_time": 1.7484607696533203,
      "step": 10503
    },
    {
      "epoch": 1.0016441345214844e-05,
      "step": 10503,
      "training_step_time": 9.223230361938477
    },
    {
      "epoch": 1.001739501953125e-05,
      "model_forward_time": 2.229705572128296,
      "step": 10504
    },
    {
      "epoch": 1.001739501953125e-05,
      "step": 10504,
      "training_step_time": 10.034685373306274
    },
    {
      "epoch": 1.0018348693847657e-05,
      "model_forward_time": 1.7113096714019775,
      "step": 10505
    },
    {
      "epoch": 1.0018348693847657e-05,
      "step": 10505,
      "training_step_time": 7.819546699523926
    },
    {
      "epoch": 1.0019302368164063e-05,
      "model_forward_time": 1.7145047187805176,
      "step": 10506
    },
    {
      "epoch": 1.0019302368164063e-05,
      "step": 10506,
      "training_step_time": 7.5066142082214355
    },
    {
      "epoch": 1.0020256042480468e-05,
      "model_forward_time": 1.713557481765747,
      "step": 10507
    },
    {
      "epoch": 1.0020256042480468e-05,
      "step": 10507,
      "training_step_time": 7.209801435470581
    },
    {
      "epoch": 1.0021209716796874e-05,
      "model_forward_time": 1.7070508003234863,
      "step": 10508
    },
    {
      "epoch": 1.0021209716796874e-05,
      "step": 10508,
      "training_step_time": 6.730403184890747
    },
    {
      "epoch": 1.0022163391113282e-05,
      "model_forward_time": 1.70088529586792,
      "step": 10509
    },
    {
      "epoch": 1.0022163391113282e-05,
      "step": 10509,
      "training_step_time": 6.756080389022827
    },
    {
      "epoch": 1.0023117065429687e-05,
      "grad_norm": 0.20097027719020844,
      "learning_rate": 9.917255984114036e-06,
      "loss": 0.0682,
      "step": 10510
    },
    {
      "epoch": 1.0023117065429687e-05,
      "model_forward_time": 1.7015373706817627,
      "step": 10510
    },
    {
      "dynamics_loss_avg": 0.06784253492951393,
      "epoch": 1.0023117065429687e-05,
      "step": 10510
    },
    {
      "action_loss_avg": 0.003065959922969341,
      "epoch": 1.0023117065429687e-05,
      "step": 10510
    },
    {
      "epoch": 1.0023117065429687e-05,
      "step": 10510,
      "training_step_time": 6.714717149734497
    },
    {
      "epoch": 1.0024070739746093e-05,
      "model_forward_time": 1.7217321395874023,
      "step": 10511
    },
    {
      "epoch": 1.0024070739746093e-05,
      "step": 10511,
      "training_step_time": 7.408888816833496
    },
    {
      "epoch": 1.00250244140625e-05,
      "model_forward_time": 1.7561113834381104,
      "step": 10512
    },
    {
      "epoch": 1.00250244140625e-05,
      "step": 10512,
      "training_step_time": 7.569253206253052
    },
    {
      "epoch": 1.0025978088378906e-05,
      "model_forward_time": 2.248664617538452,
      "step": 10513
    },
    {
      "epoch": 1.0025978088378906e-05,
      "step": 10513,
      "training_step_time": 9.491286277770996
    },
    {
      "epoch": 1.0026931762695312e-05,
      "model_forward_time": 1.7084336280822754,
      "step": 10514
    },
    {
      "epoch": 1.0026931762695312e-05,
      "step": 10514,
      "training_step_time": 7.760991811752319
    },
    {
      "epoch": 1.002788543701172e-05,
      "model_forward_time": 1.7079787254333496,
      "step": 10515
    },
    {
      "epoch": 1.002788543701172e-05,
      "step": 10515,
      "training_step_time": 7.6192381381988525
    },
    {
      "epoch": 1.0028839111328125e-05,
      "model_forward_time": 1.7084190845489502,
      "step": 10516
    },
    {
      "epoch": 1.0028839111328125e-05,
      "step": 10516,
      "training_step_time": 7.191485166549683
    },
    {
      "epoch": 1.002979278564453e-05,
      "model_forward_time": 1.7152109146118164,
      "step": 10517
    },
    {
      "epoch": 1.002979278564453e-05,
      "step": 10517,
      "training_step_time": 6.694137811660767
    },
    {
      "epoch": 1.0030746459960938e-05,
      "model_forward_time": 1.7045013904571533,
      "step": 10518
    },
    {
      "epoch": 1.0030746459960938e-05,
      "step": 10518,
      "training_step_time": 6.693105220794678
    },
    {
      "epoch": 1.0031700134277344e-05,
      "model_forward_time": 1.7126145362854004,
      "step": 10519
    },
    {
      "epoch": 1.0031700134277344e-05,
      "step": 10519,
      "training_step_time": 6.703740119934082
    },
    {
      "epoch": 1.003265380859375e-05,
      "grad_norm": 0.2297293245792389,
      "learning_rate": 9.916956150760897e-06,
      "loss": 0.0669,
      "step": 10520
    },
    {
      "epoch": 1.003265380859375e-05,
      "model_forward_time": 1.7114686965942383,
      "step": 10520
    },
    {
      "dynamics_loss_avg": 0.06348777879029513,
      "epoch": 1.003265380859375e-05,
      "step": 10520
    },
    {
      "action_loss_avg": 0.0029521300108172,
      "epoch": 1.003265380859375e-05,
      "step": 10520
    },
    {
      "epoch": 1.003265380859375e-05,
      "step": 10520,
      "training_step_time": 7.361276865005493
    },
    {
      "epoch": 1.0033607482910157e-05,
      "model_forward_time": 1.7477638721466064,
      "step": 10521
    },
    {
      "epoch": 1.0033607482910157e-05,
      "step": 10521,
      "training_step_time": 9.71640682220459
    },
    {
      "epoch": 1.0034561157226563e-05,
      "model_forward_time": 2.256516695022583,
      "step": 10522
    },
    {
      "epoch": 1.0034561157226563e-05,
      "step": 10522,
      "training_step_time": 8.505091667175293
    },
    {
      "epoch": 1.0035514831542968e-05,
      "model_forward_time": 1.7054617404937744,
      "step": 10523
    },
    {
      "epoch": 1.0035514831542968e-05,
      "step": 10523,
      "training_step_time": 8.420129299163818
    },
    {
      "epoch": 1.0036468505859374e-05,
      "model_forward_time": 1.7159240245819092,
      "step": 10524
    },
    {
      "epoch": 1.0036468505859374e-05,
      "step": 10524,
      "training_step_time": 7.596317768096924
    },
    {
      "epoch": 1.0037422180175782e-05,
      "model_forward_time": 1.7111990451812744,
      "step": 10525
    },
    {
      "epoch": 1.0037422180175782e-05,
      "step": 10525,
      "training_step_time": 7.455568313598633
    },
    {
      "epoch": 1.0038375854492187e-05,
      "model_forward_time": 1.7153327465057373,
      "step": 10526
    },
    {
      "epoch": 1.0038375854492187e-05,
      "step": 10526,
      "training_step_time": 6.743012428283691
    },
    {
      "epoch": 1.0039329528808593e-05,
      "model_forward_time": 1.703902006149292,
      "step": 10527
    },
    {
      "epoch": 1.0039329528808593e-05,
      "step": 10527,
      "training_step_time": 6.764251470565796
    },
    {
      "epoch": 1.0040283203125e-05,
      "model_forward_time": 1.710021734237671,
      "step": 10528
    },
    {
      "epoch": 1.0040283203125e-05,
      "step": 10528,
      "training_step_time": 6.758567810058594
    },
    {
      "epoch": 1.0041236877441406e-05,
      "model_forward_time": 1.7103283405303955,
      "step": 10529
    },
    {
      "epoch": 1.0041236877441406e-05,
      "step": 10529,
      "training_step_time": 7.367069244384766
    },
    {
      "epoch": 1.0042190551757812e-05,
      "grad_norm": 0.20587366819381714,
      "learning_rate": 9.916655779696826e-06,
      "loss": 0.066,
      "step": 10530
    },
    {
      "epoch": 1.0042190551757812e-05,
      "model_forward_time": 1.7450354099273682,
      "step": 10530
    },
    {
      "dynamics_loss_avg": 0.05035325847566128,
      "epoch": 1.0042190551757812e-05,
      "step": 10530
    },
    {
      "action_loss_avg": 0.003246837551705539,
      "epoch": 1.0042190551757812e-05,
      "step": 10530
    },
    {
      "epoch": 1.0042190551757812e-05,
      "step": 10530,
      "training_step_time": 8.522826433181763
    },
    {
      "epoch": 1.004314422607422e-05,
      "model_forward_time": 2.1977176666259766,
      "step": 10531
    },
    {
      "epoch": 1.004314422607422e-05,
      "step": 10531,
      "training_step_time": 7.57391881942749
    },
    {
      "epoch": 1.0044097900390625e-05,
      "model_forward_time": 1.7043890953063965,
      "step": 10532
    },
    {
      "epoch": 1.0044097900390625e-05,
      "step": 10532,
      "training_step_time": 10.169889211654663
    },
    {
      "epoch": 1.004505157470703e-05,
      "model_forward_time": 1.3343608379364014,
      "step": 10533
    },
    {
      "epoch": 1.004505157470703e-05,
      "step": 10533,
      "training_step_time": 9.475384712219238
    },
    {
      "epoch": 1.0046005249023438e-05,
      "model_forward_time": 1.707550287246704,
      "step": 10534
    },
    {
      "epoch": 1.0046005249023438e-05,
      "step": 10534,
      "training_step_time": 7.387561321258545
    },
    {
      "epoch": 1.0046958923339844e-05,
      "model_forward_time": 1.709451675415039,
      "step": 10535
    },
    {
      "epoch": 1.0046958923339844e-05,
      "step": 10535,
      "training_step_time": 7.184746503829956
    },
    {
      "epoch": 1.004791259765625e-05,
      "model_forward_time": 1.7159912586212158,
      "step": 10536
    },
    {
      "epoch": 1.004791259765625e-05,
      "step": 10536,
      "training_step_time": 6.713074445724487
    },
    {
      "epoch": 1.0048866271972657e-05,
      "model_forward_time": 1.6995527744293213,
      "step": 10537
    },
    {
      "epoch": 1.0048866271972657e-05,
      "step": 10537,
      "training_step_time": 6.741821527481079
    },
    {
      "epoch": 1.0049819946289063e-05,
      "model_forward_time": 1.7065508365631104,
      "step": 10538
    },
    {
      "epoch": 1.0049819946289063e-05,
      "step": 10538,
      "training_step_time": 7.324903964996338
    },
    {
      "epoch": 1.0050773620605468e-05,
      "model_forward_time": 1.3767659664154053,
      "step": 10539
    },
    {
      "epoch": 1.0050773620605468e-05,
      "step": 10539,
      "training_step_time": 7.5382184982299805
    },
    {
      "epoch": 1.0051727294921874e-05,
      "grad_norm": 0.20955407619476318,
      "learning_rate": 9.916354870954672e-06,
      "loss": 0.0638,
      "step": 10540
    },
    {
      "epoch": 1.0051727294921874e-05,
      "model_forward_time": 1.6987981796264648,
      "step": 10540
    },
    {
      "dynamics_loss_avg": 0.0529549041762948,
      "epoch": 1.0051727294921874e-05,
      "step": 10540
    },
    {
      "action_loss_avg": 0.003237787145189941,
      "epoch": 1.0051727294921874e-05,
      "step": 10540
    },
    {
      "epoch": 1.0051727294921874e-05,
      "step": 10540,
      "training_step_time": 8.313757419586182
    },
    {
      "epoch": 1.0052680969238282e-05,
      "model_forward_time": 2.237433910369873,
      "step": 10541
    },
    {
      "epoch": 1.0052680969238282e-05,
      "step": 10541,
      "training_step_time": 7.621456623077393
    },
    {
      "epoch": 1.0053634643554687e-05,
      "model_forward_time": 1.6943321228027344,
      "step": 10542
    },
    {
      "epoch": 1.0053634643554687e-05,
      "step": 10542,
      "training_step_time": 7.799102544784546
    },
    {
      "epoch": 1.0054588317871093e-05,
      "model_forward_time": 1.7021539211273193,
      "step": 10543
    },
    {
      "epoch": 1.0054588317871093e-05,
      "step": 10543,
      "training_step_time": 7.618099689483643
    },
    {
      "epoch": 1.00555419921875e-05,
      "model_forward_time": 1.6993069648742676,
      "step": 10544
    },
    {
      "epoch": 1.00555419921875e-05,
      "step": 10544,
      "training_step_time": 7.193413734436035
    },
    {
      "epoch": 1.0056495666503906e-05,
      "model_forward_time": 1.6968424320220947,
      "step": 10545
    },
    {
      "epoch": 1.0056495666503906e-05,
      "step": 10545,
      "training_step_time": 6.7081379890441895
    },
    {
      "epoch": 1.0057449340820312e-05,
      "model_forward_time": 1.7102444171905518,
      "step": 10546
    },
    {
      "epoch": 1.0057449340820312e-05,
      "step": 10546,
      "training_step_time": 6.713227272033691
    },
    {
      "epoch": 1.005840301513672e-05,
      "model_forward_time": 1.6942143440246582,
      "step": 10547
    },
    {
      "epoch": 1.005840301513672e-05,
      "step": 10547,
      "training_step_time": 7.359125375747681
    },
    {
      "epoch": 1.0059356689453125e-05,
      "model_forward_time": 1.7418725490570068,
      "step": 10548
    },
    {
      "epoch": 1.0059356689453125e-05,
      "step": 10548,
      "training_step_time": 7.500047445297241
    },
    {
      "epoch": 1.006031036376953e-05,
      "model_forward_time": 1.6986918449401855,
      "step": 10549
    },
    {
      "epoch": 1.006031036376953e-05,
      "step": 10549,
      "training_step_time": 9.578450679779053
    },
    {
      "epoch": 1.0061264038085938e-05,
      "grad_norm": 0.4028564691543579,
      "learning_rate": 9.91605342456734e-06,
      "loss": 0.0656,
      "step": 10550
    },
    {
      "epoch": 1.0061264038085938e-05,
      "model_forward_time": 2.24049711227417,
      "step": 10550
    },
    {
      "dynamics_loss_avg": 0.05204482730478048,
      "epoch": 1.0061264038085938e-05,
      "step": 10550
    },
    {
      "action_loss_avg": 0.0025413427560124547,
      "epoch": 1.0061264038085938e-05,
      "step": 10550
    },
    {
      "epoch": 1.0061264038085938e-05,
      "step": 10550,
      "training_step_time": 9.30351710319519
    },
    {
      "epoch": 1.0062217712402344e-05,
      "model_forward_time": 1.7057902812957764,
      "step": 10551
    },
    {
      "epoch": 1.0062217712402344e-05,
      "step": 10551,
      "training_step_time": 7.688579559326172
    },
    {
      "epoch": 1.006317138671875e-05,
      "model_forward_time": 1.705899715423584,
      "step": 10552
    },
    {
      "epoch": 1.006317138671875e-05,
      "step": 10552,
      "training_step_time": 7.621632099151611
    },
    {
      "epoch": 1.0064125061035157e-05,
      "model_forward_time": 1.700993299484253,
      "step": 10553
    },
    {
      "epoch": 1.0064125061035157e-05,
      "step": 10553,
      "training_step_time": 7.430306434631348
    },
    {
      "epoch": 1.0065078735351563e-05,
      "model_forward_time": 1.6997523307800293,
      "step": 10554
    },
    {
      "epoch": 1.0065078735351563e-05,
      "step": 10554,
      "training_step_time": 6.69514536857605
    },
    {
      "epoch": 1.0066032409667968e-05,
      "model_forward_time": 1.712920904159546,
      "step": 10555
    },
    {
      "epoch": 1.0066032409667968e-05,
      "step": 10555,
      "training_step_time": 6.702243328094482
    },
    {
      "epoch": 1.0066986083984374e-05,
      "model_forward_time": 1.6939668655395508,
      "step": 10556
    },
    {
      "epoch": 1.0066986083984374e-05,
      "step": 10556,
      "training_step_time": 7.1775758266448975
    },
    {
      "epoch": 1.0067939758300782e-05,
      "model_forward_time": 1.747438907623291,
      "step": 10557
    },
    {
      "epoch": 1.0067939758300782e-05,
      "step": 10557,
      "training_step_time": 7.58397912979126
    },
    {
      "epoch": 1.0068893432617187e-05,
      "model_forward_time": 1.7014172077178955,
      "step": 10558
    },
    {
      "epoch": 1.0068893432617187e-05,
      "step": 10558,
      "training_step_time": 7.708555221557617
    },
    {
      "epoch": 1.0069847106933593e-05,
      "model_forward_time": 2.246755599975586,
      "step": 10559
    },
    {
      "epoch": 1.0069847106933593e-05,
      "step": 10559,
      "training_step_time": 8.304560661315918
    },
    {
      "epoch": 1.007080078125e-05,
      "grad_norm": 0.2818067669868469,
      "learning_rate": 9.915751440567796e-06,
      "loss": 0.0641,
      "step": 10560
    },
    {
      "epoch": 1.007080078125e-05,
      "model_forward_time": 1.6994774341583252,
      "step": 10560
    },
    {
      "dynamics_loss_avg": 0.045529172336682674,
      "epoch": 1.007080078125e-05,
      "step": 10560
    },
    {
      "action_loss_avg": 0.003614985171589069,
      "epoch": 1.007080078125e-05,
      "step": 10560
    },
    {
      "epoch": 1.007080078125e-05,
      "step": 10560,
      "training_step_time": 8.596471548080444
    },
    {
      "epoch": 1.0071754455566406e-05,
      "model_forward_time": 1.7085015773773193,
      "step": 10561
    },
    {
      "epoch": 1.0071754455566406e-05,
      "step": 10561,
      "training_step_time": 7.60914945602417
    },
    {
      "epoch": 1.0072708129882812e-05,
      "model_forward_time": 1.703005313873291,
      "step": 10562
    },
    {
      "epoch": 1.0072708129882812e-05,
      "step": 10562,
      "training_step_time": 7.184467077255249
    },
    {
      "epoch": 1.007366180419922e-05,
      "model_forward_time": 1.7084386348724365,
      "step": 10563
    },
    {
      "epoch": 1.007366180419922e-05,
      "step": 10563,
      "training_step_time": 6.654632329940796
    },
    {
      "epoch": 1.0074615478515625e-05,
      "model_forward_time": 1.7042131423950195,
      "step": 10564
    },
    {
      "epoch": 1.0074615478515625e-05,
      "step": 10564,
      "training_step_time": 6.674824237823486
    },
    {
      "epoch": 1.007556915283203e-05,
      "model_forward_time": 1.699099063873291,
      "step": 10565
    },
    {
      "epoch": 1.007556915283203e-05,
      "step": 10565,
      "training_step_time": 7.1658830642700195
    },
    {
      "epoch": 1.0076522827148438e-05,
      "model_forward_time": 1.7540414333343506,
      "step": 10566
    },
    {
      "epoch": 1.0076522827148438e-05,
      "step": 10566,
      "training_step_time": 7.3961076736450195
    },
    {
      "epoch": 1.0077476501464844e-05,
      "model_forward_time": 1.3261604309082031,
      "step": 10567
    },
    {
      "epoch": 1.0077476501464844e-05,
      "step": 10567,
      "training_step_time": 8.166332721710205
    },
    {
      "epoch": 1.007843017578125e-05,
      "model_forward_time": 1.6970548629760742,
      "step": 10568
    },
    {
      "epoch": 1.007843017578125e-05,
      "step": 10568,
      "training_step_time": 9.665046691894531
    },
    {
      "epoch": 1.0079383850097657e-05,
      "model_forward_time": 2.2526488304138184,
      "step": 10569
    },
    {
      "epoch": 1.0079383850097657e-05,
      "step": 10569,
      "training_step_time": 10.101337432861328
    },
    {
      "epoch": 1.0080337524414063e-05,
      "grad_norm": 0.1786932498216629,
      "learning_rate": 9.915448918989065e-06,
      "loss": 0.0672,
      "step": 10570
    },
    {
      "epoch": 1.0080337524414063e-05,
      "model_forward_time": 1.703577995300293,
      "step": 10570
    },
    {
      "dynamics_loss_avg": 0.060307780466973784,
      "epoch": 1.0080337524414063e-05,
      "step": 10570
    },
    {
      "action_loss_avg": 0.002845410897862166,
      "epoch": 1.0080337524414063e-05,
      "step": 10570
    },
    {
      "epoch": 1.0080337524414063e-05,
      "step": 10570,
      "training_step_time": 7.73416543006897
    },
    {
      "epoch": 1.0081291198730468e-05,
      "model_forward_time": 1.7032692432403564,
      "step": 10571
    },
    {
      "epoch": 1.0081291198730468e-05,
      "step": 10571,
      "training_step_time": 8.196314811706543
    },
    {
      "epoch": 1.0082244873046874e-05,
      "model_forward_time": 1.7016048431396484,
      "step": 10572
    },
    {
      "epoch": 1.0082244873046874e-05,
      "step": 10572,
      "training_step_time": 6.734875202178955
    },
    {
      "epoch": 1.0083198547363282e-05,
      "model_forward_time": 1.7039382457733154,
      "step": 10573
    },
    {
      "epoch": 1.0083198547363282e-05,
      "step": 10573,
      "training_step_time": 6.670208930969238
    },
    {
      "epoch": 1.0084152221679687e-05,
      "model_forward_time": 1.703484296798706,
      "step": 10574
    },
    {
      "epoch": 1.0084152221679687e-05,
      "step": 10574,
      "training_step_time": 7.063217639923096
    },
    {
      "epoch": 1.0085105895996093e-05,
      "model_forward_time": 1.379319667816162,
      "step": 10575
    },
    {
      "epoch": 1.0085105895996093e-05,
      "step": 10575,
      "training_step_time": 7.337440252304077
    },
    {
      "epoch": 1.00860595703125e-05,
      "model_forward_time": 1.708160161972046,
      "step": 10576
    },
    {
      "epoch": 1.00860595703125e-05,
      "step": 10576,
      "training_step_time": 7.548540115356445
    },
    {
      "epoch": 1.0087013244628906e-05,
      "model_forward_time": 1.7024657726287842,
      "step": 10577
    },
    {
      "epoch": 1.0087013244628906e-05,
      "step": 10577,
      "training_step_time": 8.613221406936646
    },
    {
      "epoch": 1.0087966918945312e-05,
      "model_forward_time": 2.2144322395324707,
      "step": 10578
    },
    {
      "epoch": 1.0087966918945312e-05,
      "step": 10578,
      "training_step_time": 8.557697772979736
    },
    {
      "epoch": 1.008892059326172e-05,
      "model_forward_time": 1.7013840675354004,
      "step": 10579
    },
    {
      "epoch": 1.008892059326172e-05,
      "step": 10579,
      "training_step_time": 7.820437908172607
    },
    {
      "epoch": 1.0089874267578125e-05,
      "grad_norm": 0.22723349928855896,
      "learning_rate": 9.915145859864232e-06,
      "loss": 0.0663,
      "step": 10580
    },
    {
      "epoch": 1.0089874267578125e-05,
      "model_forward_time": 1.715620756149292,
      "step": 10580
    },
    {
      "dynamics_loss_avg": 0.06471056174486875,
      "epoch": 1.0089874267578125e-05,
      "step": 10580
    },
    {
      "action_loss_avg": 0.004762570175807923,
      "epoch": 1.0089874267578125e-05,
      "step": 10580
    },
    {
      "epoch": 1.0089874267578125e-05,
      "step": 10580,
      "training_step_time": 7.6932761669158936
    },
    {
      "epoch": 1.009082794189453e-05,
      "model_forward_time": 1.705500602722168,
      "step": 10581
    },
    {
      "epoch": 1.009082794189453e-05,
      "step": 10581,
      "training_step_time": 7.581869840621948
    },
    {
      "epoch": 1.0091781616210938e-05,
      "model_forward_time": 1.7059226036071777,
      "step": 10582
    },
    {
      "epoch": 1.0091781616210938e-05,
      "step": 10582,
      "training_step_time": 6.814979791641235
    },
    {
      "epoch": 1.0092735290527344e-05,
      "model_forward_time": 1.7139956951141357,
      "step": 10583
    },
    {
      "epoch": 1.0092735290527344e-05,
      "step": 10583,
      "training_step_time": 7.1705498695373535
    },
    {
      "epoch": 1.009368896484375e-05,
      "model_forward_time": 1.7488987445831299,
      "step": 10584
    },
    {
      "epoch": 1.009368896484375e-05,
      "step": 10584,
      "training_step_time": 7.392571210861206
    },
    {
      "epoch": 1.0094642639160157e-05,
      "model_forward_time": 1.7032370567321777,
      "step": 10585
    },
    {
      "epoch": 1.0094642639160157e-05,
      "step": 10585,
      "training_step_time": 7.740580320358276
    },
    {
      "epoch": 1.0095596313476563e-05,
      "model_forward_time": 1.6908559799194336,
      "step": 10586
    },
    {
      "epoch": 1.0095596313476563e-05,
      "step": 10586,
      "training_step_time": 8.723634004592896
    },
    {
      "epoch": 1.0096549987792969e-05,
      "model_forward_time": 2.2514469623565674,
      "step": 10587
    },
    {
      "epoch": 1.0096549987792969e-05,
      "step": 10587,
      "training_step_time": 7.767999649047852
    },
    {
      "epoch": 1.0097503662109374e-05,
      "model_forward_time": 1.7101716995239258,
      "step": 10588
    },
    {
      "epoch": 1.0097503662109374e-05,
      "step": 10588,
      "training_step_time": 7.824414253234863
    },
    {
      "epoch": 1.0098457336425782e-05,
      "model_forward_time": 1.7149429321289062,
      "step": 10589
    },
    {
      "epoch": 1.0098457336425782e-05,
      "step": 10589,
      "training_step_time": 7.535714387893677
    },
    {
      "epoch": 1.0099411010742187e-05,
      "grad_norm": 0.1773858219385147,
      "learning_rate": 9.914842263226437e-06,
      "loss": 0.0642,
      "step": 10590
    },
    {
      "epoch": 1.0099411010742187e-05,
      "model_forward_time": 1.7068097591400146,
      "step": 10590
    },
    {
      "dynamics_loss_avg": 0.06633610054850578,
      "epoch": 1.0099411010742187e-05,
      "step": 10590
    },
    {
      "action_loss_avg": 0.004870544699952006,
      "epoch": 1.0099411010742187e-05,
      "step": 10590
    },
    {
      "epoch": 1.0099411010742187e-05,
      "step": 10590,
      "training_step_time": 7.61262845993042
    },
    {
      "epoch": 1.0100364685058593e-05,
      "model_forward_time": 1.7091634273529053,
      "step": 10591
    },
    {
      "epoch": 1.0100364685058593e-05,
      "step": 10591,
      "training_step_time": 6.831884860992432
    },
    {
      "epoch": 1.0101318359375e-05,
      "model_forward_time": 1.70902681350708,
      "step": 10592
    },
    {
      "epoch": 1.0101318359375e-05,
      "step": 10592,
      "training_step_time": 7.224274635314941
    },
    {
      "epoch": 1.0102272033691406e-05,
      "model_forward_time": 1.7366232872009277,
      "step": 10593
    },
    {
      "epoch": 1.0102272033691406e-05,
      "step": 10593,
      "training_step_time": 7.424226999282837
    },
    {
      "epoch": 1.0103225708007812e-05,
      "model_forward_time": 1.7052795886993408,
      "step": 10594
    },
    {
      "epoch": 1.0103225708007812e-05,
      "step": 10594,
      "training_step_time": 7.736818075180054
    },
    {
      "epoch": 1.010417938232422e-05,
      "model_forward_time": 1.7093191146850586,
      "step": 10595
    },
    {
      "epoch": 1.010417938232422e-05,
      "step": 10595,
      "training_step_time": 7.727918386459351
    },
    {
      "epoch": 1.0105133056640625e-05,
      "model_forward_time": 2.249727487564087,
      "step": 10596
    },
    {
      "epoch": 1.0105133056640625e-05,
      "step": 10596,
      "training_step_time": 8.78753924369812
    },
    {
      "epoch": 1.010608673095703e-05,
      "model_forward_time": 1.701988697052002,
      "step": 10597
    },
    {
      "epoch": 1.010608673095703e-05,
      "step": 10597,
      "training_step_time": 7.834733009338379
    },
    {
      "epoch": 1.0107040405273438e-05,
      "model_forward_time": 1.707883358001709,
      "step": 10598
    },
    {
      "epoch": 1.0107040405273438e-05,
      "step": 10598,
      "training_step_time": 7.711909532546997
    },
    {
      "epoch": 1.0107994079589844e-05,
      "model_forward_time": 1.7111632823944092,
      "step": 10599
    },
    {
      "epoch": 1.0107994079589844e-05,
      "step": 10599,
      "training_step_time": 7.495136737823486
    },
    {
      "epoch": 1.010894775390625e-05,
      "grad_norm": 0.21937412023544312,
      "learning_rate": 9.914538129108882e-06,
      "loss": 0.0624,
      "step": 10600
    },
    {
      "epoch": 1.010894775390625e-05,
      "model_forward_time": 1.7058169841766357,
      "step": 10600
    },
    {
      "dynamics_loss_avg": 0.053966707177460195,
      "epoch": 1.010894775390625e-05,
      "step": 10600
    },
    {
      "action_loss_avg": 0.005483774363528937,
      "epoch": 1.010894775390625e-05,
      "step": 10600
    },
    {
      "epoch": 1.010894775390625e-05,
      "step": 10600,
      "training_step_time": 6.750627517700195
    },
    {
      "epoch": 1.0109901428222657e-05,
      "model_forward_time": 1.7074871063232422,
      "step": 10601
    },
    {
      "epoch": 1.0109901428222657e-05,
      "step": 10601,
      "training_step_time": 7.209919691085815
    },
    {
      "epoch": 1.0110855102539063e-05,
      "model_forward_time": 1.3346178531646729,
      "step": 10602
    },
    {
      "epoch": 1.0110855102539063e-05,
      "step": 10602,
      "training_step_time": 7.417891025543213
    },
    {
      "epoch": 1.0111808776855469e-05,
      "model_forward_time": 1.7132830619812012,
      "step": 10603
    },
    {
      "epoch": 1.0111808776855469e-05,
      "step": 10603,
      "training_step_time": 7.591582536697388
    },
    {
      "epoch": 1.0112762451171874e-05,
      "model_forward_time": 1.7075505256652832,
      "step": 10604
    },
    {
      "epoch": 1.0112762451171874e-05,
      "step": 10604,
      "training_step_time": 7.684398174285889
    },
    {
      "epoch": 1.0113716125488282e-05,
      "model_forward_time": 1.7028098106384277,
      "step": 10605
    },
    {
      "epoch": 1.0113716125488282e-05,
      "step": 10605,
      "training_step_time": 10.099202156066895
    },
    {
      "epoch": 1.0114669799804687e-05,
      "model_forward_time": 2.2364044189453125,
      "step": 10606
    },
    {
      "epoch": 1.0114669799804687e-05,
      "step": 10606,
      "training_step_time": 8.834965705871582
    },
    {
      "epoch": 1.0115623474121093e-05,
      "model_forward_time": 1.7143809795379639,
      "step": 10607
    },
    {
      "epoch": 1.0115623474121093e-05,
      "step": 10607,
      "training_step_time": 8.945212602615356
    },
    {
      "epoch": 1.01165771484375e-05,
      "model_forward_time": 1.3315060138702393,
      "step": 10608
    },
    {
      "epoch": 1.01165771484375e-05,
      "step": 10608,
      "training_step_time": 7.51036524772644
    },
    {
      "epoch": 1.0117530822753906e-05,
      "model_forward_time": 1.702974796295166,
      "step": 10609
    },
    {
      "epoch": 1.0117530822753906e-05,
      "step": 10609,
      "training_step_time": 7.321286201477051
    },
    {
      "epoch": 1.0118484497070312e-05,
      "grad_norm": 0.18128816783428192,
      "learning_rate": 9.914233457544826e-06,
      "loss": 0.0681,
      "step": 10610
    },
    {
      "epoch": 1.0118484497070312e-05,
      "model_forward_time": 1.7086033821105957,
      "step": 10610
    },
    {
      "dynamics_loss_avg": 0.055067920871078965,
      "epoch": 1.0118484497070312e-05,
      "step": 10610
    },
    {
      "action_loss_avg": 0.003435622254619375,
      "epoch": 1.0118484497070312e-05,
      "step": 10610
    },
    {
      "epoch": 1.0118484497070312e-05,
      "step": 10610,
      "training_step_time": 7.217928409576416
    },
    {
      "epoch": 1.011943817138672e-05,
      "model_forward_time": 1.3346717357635498,
      "step": 10611
    },
    {
      "epoch": 1.011943817138672e-05,
      "step": 10611,
      "training_step_time": 7.565985202789307
    },
    {
      "epoch": 1.0120391845703125e-05,
      "model_forward_time": 1.7140588760375977,
      "step": 10612
    },
    {
      "epoch": 1.0120391845703125e-05,
      "step": 10612,
      "training_step_time": 7.635816812515259
    },
    {
      "epoch": 1.012134552001953e-05,
      "model_forward_time": 1.7140262126922607,
      "step": 10613
    },
    {
      "epoch": 1.012134552001953e-05,
      "step": 10613,
      "training_step_time": 7.651767015457153
    },
    {
      "epoch": 1.0122299194335938e-05,
      "model_forward_time": 1.7093884944915771,
      "step": 10614
    },
    {
      "epoch": 1.0122299194335938e-05,
      "step": 10614,
      "training_step_time": 9.066288471221924
    },
    {
      "epoch": 1.0123252868652344e-05,
      "model_forward_time": 2.2371912002563477,
      "step": 10615
    },
    {
      "epoch": 1.0123252868652344e-05,
      "step": 10615,
      "training_step_time": 8.776804208755493
    },
    {
      "epoch": 1.012420654296875e-05,
      "model_forward_time": 1.7135260105133057,
      "step": 10616
    },
    {
      "epoch": 1.012420654296875e-05,
      "step": 10616,
      "training_step_time": 7.755737781524658
    },
    {
      "epoch": 1.0125160217285157e-05,
      "model_forward_time": 0.9948098659515381,
      "step": 10617
    },
    {
      "epoch": 1.0125160217285157e-05,
      "step": 10617,
      "training_step_time": 7.645446538925171
    },
    {
      "epoch": 1.0126113891601563e-05,
      "model_forward_time": 1.7120485305786133,
      "step": 10618
    },
    {
      "epoch": 1.0126113891601563e-05,
      "step": 10618,
      "training_step_time": 7.5673792362213135
    },
    {
      "epoch": 1.0127067565917969e-05,
      "model_forward_time": 1.709761619567871,
      "step": 10619
    },
    {
      "epoch": 1.0127067565917969e-05,
      "step": 10619,
      "training_step_time": 7.225715160369873
    },
    {
      "epoch": 1.0128021240234374e-05,
      "grad_norm": 0.1715131551027298,
      "learning_rate": 9.913928248567587e-06,
      "loss": 0.0695,
      "step": 10620
    },
    {
      "epoch": 1.0128021240234374e-05,
      "model_forward_time": 1.7015438079833984,
      "step": 10620
    },
    {
      "dynamics_loss_avg": 0.055240160506218675,
      "epoch": 1.0128021240234374e-05,
      "step": 10620
    },
    {
      "action_loss_avg": 0.0037246471736580135,
      "epoch": 1.0128021240234374e-05,
      "step": 10620
    },
    {
      "epoch": 1.0128021240234374e-05,
      "step": 10620,
      "training_step_time": 7.553753852844238
    },
    {
      "epoch": 1.0128974914550782e-05,
      "model_forward_time": 1.7417337894439697,
      "step": 10621
    },
    {
      "epoch": 1.0128974914550782e-05,
      "step": 10621,
      "training_step_time": 7.490165710449219
    },
    {
      "epoch": 1.0129928588867187e-05,
      "model_forward_time": 1.7037327289581299,
      "step": 10622
    },
    {
      "epoch": 1.0129928588867187e-05,
      "step": 10622,
      "training_step_time": 8.320838689804077
    },
    {
      "epoch": 1.0130882263183593e-05,
      "model_forward_time": 1.7045414447784424,
      "step": 10623
    },
    {
      "epoch": 1.0130882263183593e-05,
      "step": 10623,
      "training_step_time": 8.740157127380371
    },
    {
      "epoch": 1.01318359375e-05,
      "model_forward_time": 1.7111809253692627,
      "step": 10624
    },
    {
      "epoch": 1.01318359375e-05,
      "step": 10624,
      "training_step_time": 7.777944087982178
    },
    {
      "epoch": 1.0132789611816406e-05,
      "model_forward_time": 2.2366440296173096,
      "step": 10625
    },
    {
      "epoch": 1.0132789611816406e-05,
      "step": 10625,
      "training_step_time": 7.846094369888306
    },
    {
      "epoch": 1.0133743286132812e-05,
      "model_forward_time": 1.7016053199768066,
      "step": 10626
    },
    {
      "epoch": 1.0133743286132812e-05,
      "step": 10626,
      "training_step_time": 7.760388135910034
    },
    {
      "epoch": 1.013469696044922e-05,
      "model_forward_time": 1.7115452289581299,
      "step": 10627
    },
    {
      "epoch": 1.013469696044922e-05,
      "step": 10627,
      "training_step_time": 7.536985158920288
    },
    {
      "epoch": 1.0135650634765625e-05,
      "model_forward_time": 1.7072763442993164,
      "step": 10628
    },
    {
      "epoch": 1.0135650634765625e-05,
      "step": 10628,
      "training_step_time": 6.796557903289795
    },
    {
      "epoch": 1.013660430908203e-05,
      "model_forward_time": 1.7065393924713135,
      "step": 10629
    },
    {
      "epoch": 1.013660430908203e-05,
      "step": 10629,
      "training_step_time": 7.641607284545898
    },
    {
      "epoch": 1.0137557983398438e-05,
      "grad_norm": 0.21594953536987305,
      "learning_rate": 9.913622502210542e-06,
      "loss": 0.0673,
      "step": 10630
    },
    {
      "epoch": 1.0137557983398438e-05,
      "model_forward_time": 1.7439615726470947,
      "step": 10630
    },
    {
      "dynamics_loss_avg": 0.06400900799781084,
      "epoch": 1.0137557983398438e-05,
      "step": 10630
    },
    {
      "action_loss_avg": 0.0037319508264772594,
      "epoch": 1.0137557983398438e-05,
      "step": 10630
    },
    {
      "epoch": 1.0137557983398438e-05,
      "step": 10630,
      "training_step_time": 7.436630964279175
    },
    {
      "epoch": 1.0138511657714844e-05,
      "model_forward_time": 1.7052383422851562,
      "step": 10631
    },
    {
      "epoch": 1.0138511657714844e-05,
      "step": 10631,
      "training_step_time": 8.407969236373901
    },
    {
      "epoch": 1.013946533203125e-05,
      "model_forward_time": 1.7052102088928223,
      "step": 10632
    },
    {
      "epoch": 1.013946533203125e-05,
      "step": 10632,
      "training_step_time": 8.528863906860352
    },
    {
      "epoch": 1.0140419006347655e-05,
      "model_forward_time": 1.7100565433502197,
      "step": 10633
    },
    {
      "epoch": 1.0140419006347655e-05,
      "step": 10633,
      "training_step_time": 8.786010503768921
    },
    {
      "epoch": 1.0141372680664063e-05,
      "model_forward_time": 2.2377889156341553,
      "step": 10634
    },
    {
      "epoch": 1.0141372680664063e-05,
      "step": 10634,
      "training_step_time": 8.360470056533813
    },
    {
      "epoch": 1.0142326354980469e-05,
      "model_forward_time": 1.7128055095672607,
      "step": 10635
    },
    {
      "epoch": 1.0142326354980469e-05,
      "step": 10635,
      "training_step_time": 7.693645000457764
    },
    {
      "epoch": 1.0143280029296874e-05,
      "model_forward_time": 1.3268344402313232,
      "step": 10636
    },
    {
      "epoch": 1.0143280029296874e-05,
      "step": 10636,
      "training_step_time": 7.580934524536133
    },
    {
      "epoch": 1.0144233703613282e-05,
      "model_forward_time": 1.71126389503479,
      "step": 10637
    },
    {
      "epoch": 1.0144233703613282e-05,
      "step": 10637,
      "training_step_time": 6.855640649795532
    },
    {
      "epoch": 1.0145187377929687e-05,
      "model_forward_time": 1.707451343536377,
      "step": 10638
    },
    {
      "epoch": 1.0145187377929687e-05,
      "step": 10638,
      "training_step_time": 7.599663972854614
    },
    {
      "epoch": 1.0146141052246093e-05,
      "model_forward_time": 1.7425346374511719,
      "step": 10639
    },
    {
      "epoch": 1.0146141052246093e-05,
      "step": 10639,
      "training_step_time": 7.4896745681762695
    },
    {
      "epoch": 1.01470947265625e-05,
      "grad_norm": 0.19152091443538666,
      "learning_rate": 9.913316218507129e-06,
      "loss": 0.0719,
      "step": 10640
    },
    {
      "epoch": 1.01470947265625e-05,
      "model_forward_time": 1.7053468227386475,
      "step": 10640
    },
    {
      "dynamics_loss_avg": 0.048515144269913435,
      "epoch": 1.01470947265625e-05,
      "step": 10640
    },
    {
      "action_loss_avg": 0.003888866724446416,
      "epoch": 1.01470947265625e-05,
      "step": 10640
    },
    {
      "epoch": 1.01470947265625e-05,
      "step": 10640,
      "training_step_time": 9.34373950958252
    },
    {
      "epoch": 1.0148048400878906e-05,
      "model_forward_time": 1.704625129699707,
      "step": 10641
    },
    {
      "epoch": 1.0148048400878906e-05,
      "step": 10641,
      "training_step_time": 8.939100742340088
    },
    {
      "epoch": 1.0149002075195312e-05,
      "model_forward_time": 1.7027409076690674,
      "step": 10642
    },
    {
      "epoch": 1.0149002075195312e-05,
      "step": 10642,
      "training_step_time": 8.839984655380249
    },
    {
      "epoch": 1.014995574951172e-05,
      "model_forward_time": 2.2529051303863525,
      "step": 10643
    },
    {
      "epoch": 1.014995574951172e-05,
      "step": 10643,
      "training_step_time": 8.832228183746338
    },
    {
      "epoch": 1.0150909423828125e-05,
      "model_forward_time": 1.7036595344543457,
      "step": 10644
    },
    {
      "epoch": 1.0150909423828125e-05,
      "step": 10644,
      "training_step_time": 7.91277003288269
    },
    {
      "epoch": 1.015186309814453e-05,
      "model_forward_time": 1.7011466026306152,
      "step": 10645
    },
    {
      "epoch": 1.015186309814453e-05,
      "step": 10645,
      "training_step_time": 7.65826153755188
    },
    {
      "epoch": 1.0152816772460938e-05,
      "model_forward_time": 1.6813631057739258,
      "step": 10646
    },
    {
      "epoch": 1.0152816772460938e-05,
      "step": 10646,
      "training_step_time": 7.317497253417969
    },
    {
      "epoch": 1.0153770446777344e-05,
      "model_forward_time": 1.7067339420318604,
      "step": 10647
    },
    {
      "epoch": 1.0153770446777344e-05,
      "step": 10647,
      "training_step_time": 7.538337469100952
    },
    {
      "epoch": 1.015472412109375e-05,
      "model_forward_time": 1.7213594913482666,
      "step": 10648
    },
    {
      "epoch": 1.015472412109375e-05,
      "step": 10648,
      "training_step_time": 7.583299875259399
    },
    {
      "epoch": 1.0155677795410155e-05,
      "model_forward_time": 1.707913875579834,
      "step": 10649
    },
    {
      "epoch": 1.0155677795410155e-05,
      "step": 10649,
      "training_step_time": 8.300383567810059
    },
    {
      "epoch": 1.0156631469726563e-05,
      "grad_norm": 0.237782821059227,
      "learning_rate": 9.91300939749084e-06,
      "loss": 0.0729,
      "step": 10650
    },
    {
      "epoch": 1.0156631469726563e-05,
      "model_forward_time": 1.6998679637908936,
      "step": 10650
    },
    {
      "dynamics_loss_avg": 0.06444879807531834,
      "epoch": 1.0156631469726563e-05,
      "step": 10650
    },
    {
      "action_loss_avg": 0.0034357693046331407,
      "epoch": 1.0156631469726563e-05,
      "step": 10650
    },
    {
      "epoch": 1.0156631469726563e-05,
      "step": 10650,
      "training_step_time": 9.468795776367188
    },
    {
      "epoch": 1.0157585144042969e-05,
      "model_forward_time": 1.7103290557861328,
      "step": 10651
    },
    {
      "epoch": 1.0157585144042969e-05,
      "step": 10651,
      "training_step_time": 7.780728101730347
    },
    {
      "epoch": 1.0158538818359374e-05,
      "model_forward_time": 2.245522975921631,
      "step": 10652
    },
    {
      "epoch": 1.0158538818359374e-05,
      "step": 10652,
      "training_step_time": 7.772960662841797
    },
    {
      "epoch": 1.0159492492675782e-05,
      "model_forward_time": 1.704162359237671,
      "step": 10653
    },
    {
      "epoch": 1.0159492492675782e-05,
      "step": 10653,
      "training_step_time": 8.488794565200806
    },
    {
      "epoch": 1.0160446166992187e-05,
      "model_forward_time": 1.7063910961151123,
      "step": 10654
    },
    {
      "epoch": 1.0160446166992187e-05,
      "step": 10654,
      "training_step_time": 7.550863265991211
    },
    {
      "epoch": 1.0161399841308593e-05,
      "model_forward_time": 1.3295929431915283,
      "step": 10655
    },
    {
      "epoch": 1.0161399841308593e-05,
      "step": 10655,
      "training_step_time": 7.624070405960083
    },
    {
      "epoch": 1.0162353515625e-05,
      "model_forward_time": 1.7104103565216064,
      "step": 10656
    },
    {
      "epoch": 1.0162353515625e-05,
      "step": 10656,
      "training_step_time": 7.195154190063477
    },
    {
      "epoch": 1.0163307189941406e-05,
      "model_forward_time": 1.7087795734405518,
      "step": 10657
    },
    {
      "epoch": 1.0163307189941406e-05,
      "step": 10657,
      "training_step_time": 7.602708578109741
    },
    {
      "epoch": 1.0164260864257812e-05,
      "model_forward_time": 1.6936917304992676,
      "step": 10658
    },
    {
      "epoch": 1.0164260864257812e-05,
      "step": 10658,
      "training_step_time": 7.70892333984375
    },
    {
      "epoch": 1.016521453857422e-05,
      "model_forward_time": 1.331693172454834,
      "step": 10659
    },
    {
      "epoch": 1.016521453857422e-05,
      "step": 10659,
      "training_step_time": 7.734258413314819
    },
    {
      "epoch": 1.0166168212890625e-05,
      "grad_norm": 0.2451900839805603,
      "learning_rate": 9.912702039195231e-06,
      "loss": 0.0712,
      "step": 10660
    },
    {
      "epoch": 1.0166168212890625e-05,
      "model_forward_time": 1.710623025894165,
      "step": 10660
    },
    {
      "dynamics_loss_avg": 0.08678298220038413,
      "epoch": 1.0166168212890625e-05,
      "step": 10660
    },
    {
      "action_loss_avg": 0.0036870390875265004,
      "epoch": 1.0166168212890625e-05,
      "step": 10660
    },
    {
      "epoch": 1.0166168212890625e-05,
      "step": 10660,
      "training_step_time": 7.7539472579956055
    },
    {
      "epoch": 1.0167121887207031e-05,
      "model_forward_time": 1.709737777709961,
      "step": 10661
    },
    {
      "epoch": 1.0167121887207031e-05,
      "step": 10661,
      "training_step_time": 9.657711505889893
    },
    {
      "epoch": 1.0168075561523438e-05,
      "model_forward_time": 2.2488853931427,
      "step": 10662
    },
    {
      "epoch": 1.0168075561523438e-05,
      "step": 10662,
      "training_step_time": 7.804140329360962
    },
    {
      "epoch": 1.0169029235839844e-05,
      "model_forward_time": 1.711167335510254,
      "step": 10663
    },
    {
      "epoch": 1.0169029235839844e-05,
      "step": 10663,
      "training_step_time": 7.674460411071777
    },
    {
      "epoch": 1.016998291015625e-05,
      "model_forward_time": 1.71590256690979,
      "step": 10664
    },
    {
      "epoch": 1.016998291015625e-05,
      "step": 10664,
      "training_step_time": 7.54518461227417
    },
    {
      "epoch": 1.0170936584472655e-05,
      "model_forward_time": 1.717005968093872,
      "step": 10665
    },
    {
      "epoch": 1.0170936584472655e-05,
      "step": 10665,
      "training_step_time": 6.7758402824401855
    },
    {
      "epoch": 1.0171890258789063e-05,
      "model_forward_time": 1.7122251987457275,
      "step": 10666
    },
    {
      "epoch": 1.0171890258789063e-05,
      "step": 10666,
      "training_step_time": 7.426007986068726
    },
    {
      "epoch": 1.0172843933105469e-05,
      "model_forward_time": 1.7519114017486572,
      "step": 10667
    },
    {
      "epoch": 1.0172843933105469e-05,
      "step": 10667,
      "training_step_time": 7.665582180023193
    },
    {
      "epoch": 1.0173797607421874e-05,
      "model_forward_time": 1.7044754028320312,
      "step": 10668
    },
    {
      "epoch": 1.0173797607421874e-05,
      "step": 10668,
      "training_step_time": 8.255483627319336
    },
    {
      "epoch": 1.0174751281738282e-05,
      "model_forward_time": 1.715743064880371,
      "step": 10669
    },
    {
      "epoch": 1.0174751281738282e-05,
      "step": 10669,
      "training_step_time": 8.388792753219604
    },
    {
      "epoch": 1.0175704956054687e-05,
      "grad_norm": 0.15515315532684326,
      "learning_rate": 9.912394143653913e-06,
      "loss": 0.0648,
      "step": 10670
    },
    {
      "epoch": 1.0175704956054687e-05,
      "model_forward_time": 1.7084152698516846,
      "step": 10670
    },
    {
      "dynamics_loss_avg": 0.05999655388295651,
      "epoch": 1.0175704956054687e-05,
      "step": 10670
    },
    {
      "action_loss_avg": 0.0026389275561086835,
      "epoch": 1.0175704956054687e-05,
      "step": 10670
    },
    {
      "epoch": 1.0175704956054687e-05,
      "step": 10670,
      "training_step_time": 7.798006296157837
    },
    {
      "epoch": 1.0176658630371093e-05,
      "model_forward_time": 2.2551186084747314,
      "step": 10671
    },
    {
      "epoch": 1.0176658630371093e-05,
      "step": 10671,
      "training_step_time": 7.861338376998901
    },
    {
      "epoch": 1.01776123046875e-05,
      "model_forward_time": 1.714644432067871,
      "step": 10672
    },
    {
      "epoch": 1.01776123046875e-05,
      "step": 10672,
      "training_step_time": 7.673496246337891
    },
    {
      "epoch": 1.0178565979003906e-05,
      "model_forward_time": 1.7192184925079346,
      "step": 10673
    },
    {
      "epoch": 1.0178565979003906e-05,
      "step": 10673,
      "training_step_time": 7.580091238021851
    },
    {
      "epoch": 1.0179519653320312e-05,
      "model_forward_time": 1.7169153690338135,
      "step": 10674
    },
    {
      "epoch": 1.0179519653320312e-05,
      "step": 10674,
      "training_step_time": 7.340078353881836
    },
    {
      "epoch": 1.018047332763672e-05,
      "model_forward_time": 1.710106372833252,
      "step": 10675
    },
    {
      "epoch": 1.018047332763672e-05,
      "step": 10675,
      "training_step_time": 7.499154567718506
    },
    {
      "epoch": 1.0181427001953125e-05,
      "model_forward_time": 1.7593843936920166,
      "step": 10676
    },
    {
      "epoch": 1.0181427001953125e-05,
      "step": 10676,
      "training_step_time": 7.653872728347778
    },
    {
      "epoch": 1.0182380676269531e-05,
      "model_forward_time": 1.7009437084197998,
      "step": 10677
    },
    {
      "epoch": 1.0182380676269531e-05,
      "step": 10677,
      "training_step_time": 7.6752028465271
    },
    {
      "epoch": 1.0183334350585938e-05,
      "model_forward_time": 1.721362590789795,
      "step": 10678
    },
    {
      "epoch": 1.0183334350585938e-05,
      "step": 10678,
      "training_step_time": 7.747400760650635
    },
    {
      "epoch": 1.0184288024902344e-05,
      "model_forward_time": 1.7096078395843506,
      "step": 10679
    },
    {
      "epoch": 1.0184288024902344e-05,
      "step": 10679,
      "training_step_time": 7.712130069732666
    },
    {
      "epoch": 1.018524169921875e-05,
      "grad_norm": 0.17336559295654297,
      "learning_rate": 9.912085710900557e-06,
      "loss": 0.0685,
      "step": 10680
    },
    {
      "epoch": 1.018524169921875e-05,
      "model_forward_time": 2.250084161758423,
      "step": 10680
    },
    {
      "dynamics_loss_avg": 0.05738876983523369,
      "epoch": 1.018524169921875e-05,
      "step": 10680
    },
    {
      "action_loss_avg": 0.0035658425418660044,
      "epoch": 1.018524169921875e-05,
      "step": 10680
    },
    {
      "epoch": 1.018524169921875e-05,
      "step": 10680,
      "training_step_time": 8.705493688583374
    },
    {
      "epoch": 1.0186195373535155e-05,
      "model_forward_time": 1.7104861736297607,
      "step": 10681
    },
    {
      "epoch": 1.0186195373535155e-05,
      "step": 10681,
      "training_step_time": 8.427253484725952
    },
    {
      "epoch": 1.0187149047851563e-05,
      "model_forward_time": 1.7146196365356445,
      "step": 10682
    },
    {
      "epoch": 1.0187149047851563e-05,
      "step": 10682,
      "training_step_time": 7.495342254638672
    },
    {
      "epoch": 1.0188102722167969e-05,
      "model_forward_time": 1.337007761001587,
      "step": 10683
    },
    {
      "epoch": 1.0188102722167969e-05,
      "step": 10683,
      "training_step_time": 7.515477418899536
    },
    {
      "epoch": 1.0189056396484374e-05,
      "model_forward_time": 1.7168166637420654,
      "step": 10684
    },
    {
      "epoch": 1.0189056396484374e-05,
      "step": 10684,
      "training_step_time": 7.473344564437866
    },
    {
      "epoch": 1.0190010070800782e-05,
      "model_forward_time": 1.7511837482452393,
      "step": 10685
    },
    {
      "epoch": 1.0190010070800782e-05,
      "step": 10685,
      "training_step_time": 7.364047050476074
    },
    {
      "epoch": 1.0190963745117187e-05,
      "model_forward_time": 1.7157256603240967,
      "step": 10686
    },
    {
      "epoch": 1.0190963745117187e-05,
      "step": 10686,
      "training_step_time": 8.498790740966797
    },
    {
      "epoch": 1.0191917419433593e-05,
      "model_forward_time": 1.7185311317443848,
      "step": 10687
    },
    {
      "epoch": 1.0191917419433593e-05,
      "step": 10687,
      "training_step_time": 9.687899351119995
    },
    {
      "epoch": 1.019287109375e-05,
      "model_forward_time": 1.7163290977478027,
      "step": 10688
    },
    {
      "epoch": 1.019287109375e-05,
      "step": 10688,
      "training_step_time": 7.653957366943359
    },
    {
      "epoch": 1.0193824768066406e-05,
      "model_forward_time": 1.7178661823272705,
      "step": 10689
    },
    {
      "epoch": 1.0193824768066406e-05,
      "step": 10689,
      "training_step_time": 9.73776125907898
    },
    {
      "epoch": 1.0194778442382812e-05,
      "grad_norm": 0.20439080893993378,
      "learning_rate": 9.911776740968893e-06,
      "loss": 0.0729,
      "step": 10690
    },
    {
      "epoch": 1.0194778442382812e-05,
      "model_forward_time": 1.716242790222168,
      "step": 10690
    },
    {
      "dynamics_loss_avg": 0.06087668351829052,
      "epoch": 1.0194778442382812e-05,
      "step": 10690
    },
    {
      "action_loss_avg": 0.0038127781299408525,
      "epoch": 1.0194778442382812e-05,
      "step": 10690
    },
    {
      "epoch": 1.0194778442382812e-05,
      "step": 10690,
      "training_step_time": 7.791664361953735
    },
    {
      "epoch": 1.019573211669922e-05,
      "model_forward_time": 1.7167763710021973,
      "step": 10691
    },
    {
      "epoch": 1.019573211669922e-05,
      "step": 10691,
      "training_step_time": 7.5319178104400635
    },
    {
      "epoch": 1.0196685791015625e-05,
      "model_forward_time": 1.7175235748291016,
      "step": 10692
    },
    {
      "epoch": 1.0196685791015625e-05,
      "step": 10692,
      "training_step_time": 7.3597252368927
    },
    {
      "epoch": 1.0197639465332031e-05,
      "model_forward_time": 1.714418649673462,
      "step": 10693
    },
    {
      "epoch": 1.0197639465332031e-05,
      "step": 10693,
      "training_step_time": 7.290998935699463
    },
    {
      "epoch": 1.0198593139648438e-05,
      "model_forward_time": 1.7255282402038574,
      "step": 10694
    },
    {
      "epoch": 1.0198593139648438e-05,
      "step": 10694,
      "training_step_time": 7.578571796417236
    },
    {
      "epoch": 1.0199546813964844e-05,
      "model_forward_time": 1.7230777740478516,
      "step": 10695
    },
    {
      "epoch": 1.0199546813964844e-05,
      "step": 10695,
      "training_step_time": 8.085644483566284
    },
    {
      "epoch": 1.020050048828125e-05,
      "model_forward_time": 1.716942548751831,
      "step": 10696
    },
    {
      "epoch": 1.020050048828125e-05,
      "step": 10696,
      "training_step_time": 8.534288167953491
    },
    {
      "epoch": 1.0201454162597655e-05,
      "model_forward_time": 1.718858003616333,
      "step": 10697
    },
    {
      "epoch": 1.0201454162597655e-05,
      "step": 10697,
      "training_step_time": 7.679527997970581
    },
    {
      "epoch": 1.0202407836914063e-05,
      "model_forward_time": 1.7172133922576904,
      "step": 10698
    },
    {
      "epoch": 1.0202407836914063e-05,
      "step": 10698,
      "training_step_time": 8.539669513702393
    },
    {
      "epoch": 1.0203361511230469e-05,
      "model_forward_time": 2.251952648162842,
      "step": 10699
    },
    {
      "epoch": 1.0203361511230469e-05,
      "step": 10699,
      "training_step_time": 9.018213033676147
    },
    {
      "epoch": 1.0204315185546874e-05,
      "grad_norm": 0.19544902443885803,
      "learning_rate": 9.911467233892708e-06,
      "loss": 0.0722,
      "step": 10700
    },
    {
      "epoch": 1.0204315185546874e-05,
      "model_forward_time": 1.7198970317840576,
      "step": 10700
    },
    {
      "dynamics_loss_avg": 0.058112272247672084,
      "epoch": 1.0204315185546874e-05,
      "step": 10700
    },
    {
      "action_loss_avg": 0.003320535866077989,
      "epoch": 1.0204315185546874e-05,
      "step": 10700
    },
    {
      "epoch": 1.0204315185546874e-05,
      "step": 10700,
      "training_step_time": 7.5024168491363525
    },
    {
      "epoch": 1.0205268859863282e-05,
      "model_forward_time": 1.7148942947387695,
      "step": 10701
    },
    {
      "epoch": 1.0205268859863282e-05,
      "step": 10701,
      "training_step_time": 7.3881378173828125
    },
    {
      "epoch": 1.0206222534179687e-05,
      "model_forward_time": 1.7163937091827393,
      "step": 10702
    },
    {
      "epoch": 1.0206222534179687e-05,
      "step": 10702,
      "training_step_time": 7.3023669719696045
    },
    {
      "epoch": 1.0207176208496093e-05,
      "model_forward_time": 1.7174668312072754,
      "step": 10703
    },
    {
      "epoch": 1.0207176208496093e-05,
      "step": 10703,
      "training_step_time": 7.517390012741089
    },
    {
      "epoch": 1.02081298828125e-05,
      "model_forward_time": 1.7634782791137695,
      "step": 10704
    },
    {
      "epoch": 1.02081298828125e-05,
      "step": 10704,
      "training_step_time": 8.514069080352783
    },
    {
      "epoch": 1.0209083557128906e-05,
      "model_forward_time": 1.717299461364746,
      "step": 10705
    },
    {
      "epoch": 1.0209083557128906e-05,
      "step": 10705,
      "training_step_time": 8.14612078666687
    },
    {
      "epoch": 1.0210037231445312e-05,
      "model_forward_time": 1.7122318744659424,
      "step": 10706
    },
    {
      "epoch": 1.0210037231445312e-05,
      "step": 10706,
      "training_step_time": 8.631498336791992
    },
    {
      "epoch": 1.021099090576172e-05,
      "model_forward_time": 1.723736047744751,
      "step": 10707
    },
    {
      "epoch": 1.021099090576172e-05,
      "step": 10707,
      "training_step_time": 9.156841516494751
    },
    {
      "epoch": 1.0211944580078125e-05,
      "model_forward_time": 2.2527849674224854,
      "step": 10708
    },
    {
      "epoch": 1.0211944580078125e-05,
      "step": 10708,
      "training_step_time": 7.71710205078125
    },
    {
      "epoch": 1.0212898254394531e-05,
      "model_forward_time": 1.720569133758545,
      "step": 10709
    },
    {
      "epoch": 1.0212898254394531e-05,
      "step": 10709,
      "training_step_time": 7.579885482788086
    },
    {
      "epoch": 1.0213851928710938e-05,
      "grad_norm": 0.276189923286438,
      "learning_rate": 9.911157189705853e-06,
      "loss": 0.0739,
      "step": 10710
    },
    {
      "epoch": 1.0213851928710938e-05,
      "model_forward_time": 1.7105302810668945,
      "step": 10710
    },
    {
      "dynamics_loss_avg": 0.07860009260475635,
      "epoch": 1.0213851928710938e-05,
      "step": 10710
    },
    {
      "action_loss_avg": 0.004528656462207436,
      "epoch": 1.0213851928710938e-05,
      "step": 10710
    },
    {
      "epoch": 1.0213851928710938e-05,
      "step": 10710,
      "training_step_time": 7.328620195388794
    },
    {
      "epoch": 1.0214805603027344e-05,
      "model_forward_time": 1.719789981842041,
      "step": 10711
    },
    {
      "epoch": 1.0214805603027344e-05,
      "step": 10711,
      "training_step_time": 7.341822147369385
    },
    {
      "epoch": 1.021575927734375e-05,
      "model_forward_time": 1.716808795928955,
      "step": 10712
    },
    {
      "epoch": 1.021575927734375e-05,
      "step": 10712,
      "training_step_time": 8.355716228485107
    },
    {
      "epoch": 1.0216712951660155e-05,
      "model_forward_time": 1.756993293762207,
      "step": 10713
    },
    {
      "epoch": 1.0216712951660155e-05,
      "step": 10713,
      "training_step_time": 8.568756818771362
    },
    {
      "epoch": 1.0217666625976563e-05,
      "model_forward_time": 1.7104454040527344,
      "step": 10714
    },
    {
      "epoch": 1.0217666625976563e-05,
      "step": 10714,
      "training_step_time": 8.291255712509155
    },
    {
      "epoch": 1.0218620300292969e-05,
      "model_forward_time": 1.7209744453430176,
      "step": 10715
    },
    {
      "epoch": 1.0218620300292969e-05,
      "step": 10715,
      "training_step_time": 9.51842975616455
    },
    {
      "epoch": 1.0219573974609374e-05,
      "model_forward_time": 1.733656883239746,
      "step": 10716
    },
    {
      "epoch": 1.0219573974609374e-05,
      "step": 10716,
      "training_step_time": 9.566668033599854
    },
    {
      "epoch": 1.0220527648925782e-05,
      "model_forward_time": 2.2498903274536133,
      "step": 10717
    },
    {
      "epoch": 1.0220527648925782e-05,
      "step": 10717,
      "training_step_time": 8.764055728912354
    },
    {
      "epoch": 1.0221481323242187e-05,
      "model_forward_time": 1.7126147747039795,
      "step": 10718
    },
    {
      "epoch": 1.0221481323242187e-05,
      "step": 10718,
      "training_step_time": 7.555333614349365
    },
    {
      "epoch": 1.0222434997558593e-05,
      "model_forward_time": 1.7175161838531494,
      "step": 10719
    },
    {
      "epoch": 1.0222434997558593e-05,
      "step": 10719,
      "training_step_time": 7.318216562271118
    },
    {
      "epoch": 1.0223388671875e-05,
      "grad_norm": 0.22995829582214355,
      "learning_rate": 9.910846608442229e-06,
      "loss": 0.0742,
      "step": 10720
    },
    {
      "epoch": 1.0223388671875e-05,
      "model_forward_time": 1.7115066051483154,
      "step": 10720
    },
    {
      "dynamics_loss_avg": 0.06954136658459902,
      "epoch": 1.0223388671875e-05,
      "step": 10720
    },
    {
      "action_loss_avg": 0.004820978967472911,
      "epoch": 1.0223388671875e-05,
      "step": 10720
    },
    {
      "epoch": 1.0223388671875e-05,
      "step": 10720,
      "training_step_time": 7.333049535751343
    },
    {
      "epoch": 1.0224342346191406e-05,
      "model_forward_time": 1.7158057689666748,
      "step": 10721
    },
    {
      "epoch": 1.0224342346191406e-05,
      "step": 10721,
      "training_step_time": 7.24569034576416
    },
    {
      "epoch": 1.0225296020507812e-05,
      "model_forward_time": 1.7625865936279297,
      "step": 10722
    },
    {
      "epoch": 1.0225296020507812e-05,
      "step": 10722,
      "training_step_time": 8.119468212127686
    },
    {
      "epoch": 1.022624969482422e-05,
      "model_forward_time": 1.7144489288330078,
      "step": 10723
    },
    {
      "epoch": 1.022624969482422e-05,
      "step": 10723,
      "training_step_time": 8.219848871231079
    },
    {
      "epoch": 1.0227203369140625e-05,
      "model_forward_time": 1.7133831977844238,
      "step": 10724
    },
    {
      "epoch": 1.0227203369140625e-05,
      "step": 10724,
      "training_step_time": 8.334573030471802
    },
    {
      "epoch": 1.0228157043457031e-05,
      "model_forward_time": 1.7130672931671143,
      "step": 10725
    },
    {
      "epoch": 1.0228157043457031e-05,
      "step": 10725,
      "training_step_time": 8.629802942276001
    },
    {
      "epoch": 1.0229110717773438e-05,
      "model_forward_time": 2.246459484100342,
      "step": 10726
    },
    {
      "epoch": 1.0229110717773438e-05,
      "step": 10726,
      "training_step_time": 8.54379653930664
    },
    {
      "epoch": 1.0230064392089844e-05,
      "model_forward_time": 1.7105767726898193,
      "step": 10727
    },
    {
      "epoch": 1.0230064392089844e-05,
      "step": 10727,
      "training_step_time": 7.6959006786346436
    },
    {
      "epoch": 1.023101806640625e-05,
      "model_forward_time": 1.7112982273101807,
      "step": 10728
    },
    {
      "epoch": 1.023101806640625e-05,
      "step": 10728,
      "training_step_time": 7.41083025932312
    },
    {
      "epoch": 1.0231971740722655e-05,
      "model_forward_time": 1.7153644561767578,
      "step": 10729
    },
    {
      "epoch": 1.0231971740722655e-05,
      "step": 10729,
      "training_step_time": 7.4164674282073975
    },
    {
      "epoch": 1.0232925415039063e-05,
      "grad_norm": 0.18820303678512573,
      "learning_rate": 9.910535490135805e-06,
      "loss": 0.0677,
      "step": 10730
    },
    {
      "epoch": 1.0232925415039063e-05,
      "model_forward_time": 1.7167675495147705,
      "step": 10730
    },
    {
      "dynamics_loss_avg": 0.05702707972377539,
      "epoch": 1.0232925415039063e-05,
      "step": 10730
    },
    {
      "action_loss_avg": 0.0038300231972243637,
      "epoch": 1.0232925415039063e-05,
      "step": 10730
    },
    {
      "epoch": 1.0232925415039063e-05,
      "step": 10730,
      "training_step_time": 7.191982984542847
    },
    {
      "epoch": 1.0233879089355469e-05,
      "model_forward_time": 1.7477405071258545,
      "step": 10731
    },
    {
      "epoch": 1.0233879089355469e-05,
      "step": 10731,
      "training_step_time": 7.5488762855529785
    },
    {
      "epoch": 1.0234832763671874e-05,
      "model_forward_time": 1.7101750373840332,
      "step": 10732
    },
    {
      "epoch": 1.0234832763671874e-05,
      "step": 10732,
      "training_step_time": 8.443597078323364
    },
    {
      "epoch": 1.0235786437988282e-05,
      "model_forward_time": 1.720834732055664,
      "step": 10733
    },
    {
      "epoch": 1.0235786437988282e-05,
      "step": 10733,
      "training_step_time": 7.545395851135254
    },
    {
      "epoch": 1.0236740112304687e-05,
      "model_forward_time": 1.7136561870574951,
      "step": 10734
    },
    {
      "epoch": 1.0236740112304687e-05,
      "step": 10734,
      "training_step_time": 8.447066068649292
    },
    {
      "epoch": 1.0237693786621093e-05,
      "model_forward_time": 2.2401952743530273,
      "step": 10735
    },
    {
      "epoch": 1.0237693786621093e-05,
      "step": 10735,
      "training_step_time": 8.30338430404663
    },
    {
      "epoch": 1.02386474609375e-05,
      "model_forward_time": 1.7144036293029785,
      "step": 10736
    },
    {
      "epoch": 1.02386474609375e-05,
      "step": 10736,
      "training_step_time": 7.763489007949829
    },
    {
      "epoch": 1.0239601135253906e-05,
      "model_forward_time": 1.712144374847412,
      "step": 10737
    },
    {
      "epoch": 1.0239601135253906e-05,
      "step": 10737,
      "training_step_time": 7.399523735046387
    },
    {
      "epoch": 1.0240554809570312e-05,
      "model_forward_time": 1.7150554656982422,
      "step": 10738
    },
    {
      "epoch": 1.0240554809570312e-05,
      "step": 10738,
      "training_step_time": 7.5029380321502686
    },
    {
      "epoch": 1.024150848388672e-05,
      "model_forward_time": 1.7186715602874756,
      "step": 10739
    },
    {
      "epoch": 1.024150848388672e-05,
      "step": 10739,
      "training_step_time": 7.209814071655273
    },
    {
      "epoch": 1.0242462158203125e-05,
      "grad_norm": 0.2575720250606537,
      "learning_rate": 9.910223834820602e-06,
      "loss": 0.0706,
      "step": 10740
    },
    {
      "epoch": 1.0242462158203125e-05,
      "model_forward_time": 1.7191479206085205,
      "step": 10740
    },
    {
      "dynamics_loss_avg": 0.05433154869824648,
      "epoch": 1.0242462158203125e-05,
      "step": 10740
    },
    {
      "action_loss_avg": 0.004177764733321964,
      "epoch": 1.0242462158203125e-05,
      "step": 10740
    },
    {
      "epoch": 1.0242462158203125e-05,
      "step": 10740,
      "training_step_time": 8.33990478515625
    },
    {
      "epoch": 1.0243415832519531e-05,
      "model_forward_time": 1.7140514850616455,
      "step": 10741
    },
    {
      "epoch": 1.0243415832519531e-05,
      "step": 10741,
      "training_step_time": 7.417349338531494
    },
    {
      "epoch": 1.0244369506835938e-05,
      "model_forward_time": 1.7204999923706055,
      "step": 10742
    },
    {
      "epoch": 1.0244369506835938e-05,
      "step": 10742,
      "training_step_time": 9.16831660270691
    },
    {
      "epoch": 1.0245323181152344e-05,
      "model_forward_time": 1.7165606021881104,
      "step": 10743
    },
    {
      "epoch": 1.0245323181152344e-05,
      "step": 10743,
      "training_step_time": 8.51531171798706
    },
    {
      "epoch": 1.024627685546875e-05,
      "model_forward_time": 1.7193970680236816,
      "step": 10744
    },
    {
      "epoch": 1.024627685546875e-05,
      "step": 10744,
      "training_step_time": 8.980182886123657
    },
    {
      "epoch": 1.0247230529785156e-05,
      "model_forward_time": 1.7080695629119873,
      "step": 10745
    },
    {
      "epoch": 1.0247230529785156e-05,
      "step": 10745,
      "training_step_time": 7.7461278438568115
    },
    {
      "epoch": 1.0248184204101563e-05,
      "model_forward_time": 1.7176997661590576,
      "step": 10746
    },
    {
      "epoch": 1.0248184204101563e-05,
      "step": 10746,
      "training_step_time": 8.396007061004639
    },
    {
      "epoch": 1.0249137878417969e-05,
      "model_forward_time": 1.7168002128601074,
      "step": 10747
    },
    {
      "epoch": 1.0249137878417969e-05,
      "step": 10747,
      "training_step_time": 7.528002500534058
    },
    {
      "epoch": 1.0250091552734374e-05,
      "model_forward_time": 1.7178795337677002,
      "step": 10748
    },
    {
      "epoch": 1.0250091552734374e-05,
      "step": 10748,
      "training_step_time": 7.182703971862793
    },
    {
      "epoch": 1.0251045227050782e-05,
      "model_forward_time": 1.7171404361724854,
      "step": 10749
    },
    {
      "epoch": 1.0251045227050782e-05,
      "step": 10749,
      "training_step_time": 8.148520231246948
    },
    {
      "epoch": 1.0251998901367188e-05,
      "grad_norm": 0.2685319185256958,
      "learning_rate": 9.909911642530703e-06,
      "loss": 0.0684,
      "step": 10750
    },
    {
      "epoch": 1.0251998901367188e-05,
      "model_forward_time": 1.7100350856781006,
      "step": 10750
    },
    {
      "dynamics_loss_avg": 0.06312107080593705,
      "epoch": 1.0251998901367188e-05,
      "step": 10750
    },
    {
      "action_loss_avg": 0.003324255731422454,
      "epoch": 1.0251998901367188e-05,
      "step": 10750
    },
    {
      "epoch": 1.0251998901367188e-05,
      "step": 10750,
      "training_step_time": 8.090871810913086
    },
    {
      "epoch": 1.0252952575683593e-05,
      "model_forward_time": 1.7087719440460205,
      "step": 10751
    },
    {
      "epoch": 1.0252952575683593e-05,
      "step": 10751,
      "training_step_time": 8.024778127670288
    },
    {
      "epoch": 1.025390625e-05,
      "model_forward_time": 1.7176012992858887,
      "step": 10752
    },
    {
      "epoch": 1.025390625e-05,
      "step": 10752,
      "training_step_time": 8.457748889923096
    },
    {
      "epoch": 1.0254859924316406e-05,
      "model_forward_time": 1.7156853675842285,
      "step": 10753
    },
    {
      "epoch": 1.0254859924316406e-05,
      "step": 10753,
      "training_step_time": 7.594040632247925
    },
    {
      "epoch": 1.0255813598632812e-05,
      "model_forward_time": 2.2458479404449463,
      "step": 10754
    },
    {
      "epoch": 1.0255813598632812e-05,
      "step": 10754,
      "training_step_time": 7.6923277378082275
    },
    {
      "epoch": 1.025676727294922e-05,
      "model_forward_time": 1.7142200469970703,
      "step": 10755
    },
    {
      "epoch": 1.025676727294922e-05,
      "step": 10755,
      "training_step_time": 7.585918664932251
    },
    {
      "epoch": 1.0257720947265625e-05,
      "model_forward_time": 1.7117571830749512,
      "step": 10756
    },
    {
      "epoch": 1.0257720947265625e-05,
      "step": 10756,
      "training_step_time": 8.405052661895752
    },
    {
      "epoch": 1.0258674621582031e-05,
      "model_forward_time": 1.710864782333374,
      "step": 10757
    },
    {
      "epoch": 1.0258674621582031e-05,
      "step": 10757,
      "training_step_time": 7.121185779571533
    },
    {
      "epoch": 1.0259628295898437e-05,
      "model_forward_time": 1.7226650714874268,
      "step": 10758
    },
    {
      "epoch": 1.0259628295898437e-05,
      "step": 10758,
      "training_step_time": 7.667099475860596
    },
    {
      "epoch": 1.0260581970214844e-05,
      "model_forward_time": 1.7575244903564453,
      "step": 10759
    },
    {
      "epoch": 1.0260581970214844e-05,
      "step": 10759,
      "training_step_time": 7.2620484828948975
    },
    {
      "epoch": 1.026153564453125e-05,
      "grad_norm": 0.2815854549407959,
      "learning_rate": 9.909598913300248e-06,
      "loss": 0.0703,
      "step": 10760
    },
    {
      "epoch": 1.026153564453125e-05,
      "model_forward_time": 1.707165002822876,
      "step": 10760
    },
    {
      "dynamics_loss_avg": 0.06896003987640142,
      "epoch": 1.026153564453125e-05,
      "step": 10760
    },
    {
      "action_loss_avg": 0.004107832681620493,
      "epoch": 1.026153564453125e-05,
      "step": 10760
    },
    {
      "epoch": 1.026153564453125e-05,
      "step": 10760,
      "training_step_time": 7.521043300628662
    },
    {
      "epoch": 1.0262489318847656e-05,
      "model_forward_time": 1.337372064590454,
      "step": 10761
    },
    {
      "epoch": 1.0262489318847656e-05,
      "step": 10761,
      "training_step_time": 9.64684796333313
    },
    {
      "epoch": 1.0263442993164063e-05,
      "model_forward_time": 1.7143549919128418,
      "step": 10762
    },
    {
      "epoch": 1.0263442993164063e-05,
      "step": 10762,
      "training_step_time": 7.636881589889526
    },
    {
      "epoch": 1.0264396667480469e-05,
      "model_forward_time": 2.266585350036621,
      "step": 10763
    },
    {
      "epoch": 1.0264396667480469e-05,
      "step": 10763,
      "training_step_time": 8.638975143432617
    },
    {
      "epoch": 1.0265350341796874e-05,
      "model_forward_time": 1.7187848091125488,
      "step": 10764
    },
    {
      "epoch": 1.0265350341796874e-05,
      "step": 10764,
      "training_step_time": 7.556652784347534
    },
    {
      "epoch": 1.0266304016113282e-05,
      "model_forward_time": 1.7114131450653076,
      "step": 10765
    },
    {
      "epoch": 1.0266304016113282e-05,
      "step": 10765,
      "training_step_time": 8.080738306045532
    },
    {
      "epoch": 1.0267257690429688e-05,
      "model_forward_time": 1.7212250232696533,
      "step": 10766
    },
    {
      "epoch": 1.0267257690429688e-05,
      "step": 10766,
      "training_step_time": 8.510007858276367
    },
    {
      "epoch": 1.0268211364746093e-05,
      "model_forward_time": 1.713271141052246,
      "step": 10767
    },
    {
      "epoch": 1.0268211364746093e-05,
      "step": 10767,
      "training_step_time": 7.611086368560791
    },
    {
      "epoch": 1.02691650390625e-05,
      "model_forward_time": 1.7629079818725586,
      "step": 10768
    },
    {
      "epoch": 1.02691650390625e-05,
      "step": 10768,
      "training_step_time": 7.2430150508880615
    },
    {
      "epoch": 1.0270118713378906e-05,
      "model_forward_time": 1.7183873653411865,
      "step": 10769
    },
    {
      "epoch": 1.0270118713378906e-05,
      "step": 10769,
      "training_step_time": 7.646901607513428
    },
    {
      "epoch": 1.0271072387695312e-05,
      "grad_norm": 0.24618428945541382,
      "learning_rate": 9.909285647163439e-06,
      "loss": 0.0694,
      "step": 10770
    },
    {
      "epoch": 1.0271072387695312e-05,
      "model_forward_time": 1.7117164134979248,
      "step": 10770
    },
    {
      "dynamics_loss_avg": 0.06385343819856644,
      "epoch": 1.0271072387695312e-05,
      "step": 10770
    },
    {
      "action_loss_avg": 0.004402280773501843,
      "epoch": 1.0271072387695312e-05,
      "step": 10770
    },
    {
      "epoch": 1.0271072387695312e-05,
      "step": 10770,
      "training_step_time": 8.301391124725342
    },
    {
      "epoch": 1.027202606201172e-05,
      "model_forward_time": 1.7126471996307373,
      "step": 10771
    },
    {
      "epoch": 1.027202606201172e-05,
      "step": 10771,
      "training_step_time": 8.587736129760742
    },
    {
      "epoch": 1.0272979736328125e-05,
      "model_forward_time": 2.272749423980713,
      "step": 10772
    },
    {
      "epoch": 1.0272979736328125e-05,
      "step": 10772,
      "training_step_time": 8.75194239616394
    },
    {
      "epoch": 1.0273933410644531e-05,
      "model_forward_time": 1.7186806201934814,
      "step": 10773
    },
    {
      "epoch": 1.0273933410644531e-05,
      "step": 10773,
      "training_step_time": 9.058677673339844
    },
    {
      "epoch": 1.0274887084960937e-05,
      "model_forward_time": 1.7154760360717773,
      "step": 10774
    },
    {
      "epoch": 1.0274887084960937e-05,
      "step": 10774,
      "training_step_time": 7.710636377334595
    },
    {
      "epoch": 1.0275840759277344e-05,
      "model_forward_time": 1.7119896411895752,
      "step": 10775
    },
    {
      "epoch": 1.0275840759277344e-05,
      "step": 10775,
      "training_step_time": 7.6638476848602295
    },
    {
      "epoch": 1.027679443359375e-05,
      "model_forward_time": 1.7181625366210938,
      "step": 10776
    },
    {
      "epoch": 1.027679443359375e-05,
      "step": 10776,
      "training_step_time": 8.796506643295288
    },
    {
      "epoch": 1.0277748107910156e-05,
      "model_forward_time": 1.760650634765625,
      "step": 10777
    },
    {
      "epoch": 1.0277748107910156e-05,
      "step": 10777,
      "training_step_time": 8.406887531280518
    },
    {
      "epoch": 1.0278701782226563e-05,
      "model_forward_time": 1.7153034210205078,
      "step": 10778
    },
    {
      "epoch": 1.0278701782226563e-05,
      "step": 10778,
      "training_step_time": 7.718479156494141
    },
    {
      "epoch": 1.0279655456542969e-05,
      "model_forward_time": 1.7101669311523438,
      "step": 10779
    },
    {
      "epoch": 1.0279655456542969e-05,
      "step": 10779,
      "training_step_time": 9.391438007354736
    },
    {
      "epoch": 1.0280609130859374e-05,
      "grad_norm": 0.2582237124443054,
      "learning_rate": 9.908971844154531e-06,
      "loss": 0.0676,
      "step": 10780
    },
    {
      "epoch": 1.0280609130859374e-05,
      "model_forward_time": 1.7101404666900635,
      "step": 10780
    },
    {
      "dynamics_loss_avg": 0.05536565333604813,
      "epoch": 1.0280609130859374e-05,
      "step": 10780
    },
    {
      "action_loss_avg": 0.003659424895886332,
      "epoch": 1.0280609130859374e-05,
      "step": 10780
    },
    {
      "epoch": 1.0280609130859374e-05,
      "step": 10780,
      "training_step_time": 8.77850604057312
    },
    {
      "epoch": 1.0281562805175782e-05,
      "model_forward_time": 2.2529139518737793,
      "step": 10781
    },
    {
      "epoch": 1.0281562805175782e-05,
      "step": 10781,
      "training_step_time": 8.845237255096436
    },
    {
      "epoch": 1.0282516479492188e-05,
      "model_forward_time": 1.7150287628173828,
      "step": 10782
    },
    {
      "epoch": 1.0282516479492188e-05,
      "step": 10782,
      "training_step_time": 8.022255420684814
    },
    {
      "epoch": 1.0283470153808593e-05,
      "model_forward_time": 1.7156181335449219,
      "step": 10783
    },
    {
      "epoch": 1.0283470153808593e-05,
      "step": 10783,
      "training_step_time": 7.739825487136841
    },
    {
      "epoch": 1.0284423828125e-05,
      "model_forward_time": 1.7181951999664307,
      "step": 10784
    },
    {
      "epoch": 1.0284423828125e-05,
      "step": 10784,
      "training_step_time": 7.679455041885376
    },
    {
      "epoch": 1.0285377502441406e-05,
      "model_forward_time": 1.7211947441101074,
      "step": 10785
    },
    {
      "epoch": 1.0285377502441406e-05,
      "step": 10785,
      "training_step_time": 7.709603548049927
    },
    {
      "epoch": 1.0286331176757812e-05,
      "model_forward_time": 1.7599704265594482,
      "step": 10786
    },
    {
      "epoch": 1.0286331176757812e-05,
      "step": 10786,
      "training_step_time": 7.821584224700928
    },
    {
      "epoch": 1.028728485107422e-05,
      "model_forward_time": 1.7152392864227295,
      "step": 10787
    },
    {
      "epoch": 1.028728485107422e-05,
      "step": 10787,
      "training_step_time": 8.72844672203064
    },
    {
      "epoch": 1.0288238525390625e-05,
      "model_forward_time": 1.713860273361206,
      "step": 10788
    },
    {
      "epoch": 1.0288238525390625e-05,
      "step": 10788,
      "training_step_time": 9.801066637039185
    },
    {
      "epoch": 1.0289192199707031e-05,
      "model_forward_time": 1.7171411514282227,
      "step": 10789
    },
    {
      "epoch": 1.0289192199707031e-05,
      "step": 10789,
      "training_step_time": 8.306689262390137
    },
    {
      "epoch": 1.0290145874023437e-05,
      "grad_norm": 0.24848344922065735,
      "learning_rate": 9.908657504307844e-06,
      "loss": 0.0682,
      "step": 10790
    },
    {
      "epoch": 1.0290145874023437e-05,
      "model_forward_time": 1.7105040550231934,
      "step": 10790
    },
    {
      "dynamics_loss_avg": 0.06154022663831711,
      "epoch": 1.0290145874023437e-05,
      "step": 10790
    },
    {
      "action_loss_avg": 0.00387903485679999,
      "epoch": 1.0290145874023437e-05,
      "step": 10790
    },
    {
      "epoch": 1.0290145874023437e-05,
      "step": 10790,
      "training_step_time": 8.275855779647827
    },
    {
      "epoch": 1.0291099548339844e-05,
      "model_forward_time": 2.2677805423736572,
      "step": 10791
    },
    {
      "epoch": 1.0291099548339844e-05,
      "step": 10791,
      "training_step_time": 7.972920179367065
    },
    {
      "epoch": 1.029205322265625e-05,
      "model_forward_time": 1.7164306640625,
      "step": 10792
    },
    {
      "epoch": 1.029205322265625e-05,
      "step": 10792,
      "training_step_time": 7.794633865356445
    },
    {
      "epoch": 1.0293006896972656e-05,
      "model_forward_time": 1.7087640762329102,
      "step": 10793
    },
    {
      "epoch": 1.0293006896972656e-05,
      "step": 10793,
      "training_step_time": 7.683588266372681
    },
    {
      "epoch": 1.0293960571289063e-05,
      "model_forward_time": 1.716841220855713,
      "step": 10794
    },
    {
      "epoch": 1.0293960571289063e-05,
      "step": 10794,
      "training_step_time": 7.721951246261597
    },
    {
      "epoch": 1.0294914245605469e-05,
      "model_forward_time": 1.405433177947998,
      "step": 10795
    },
    {
      "epoch": 1.0294914245605469e-05,
      "step": 10795,
      "training_step_time": 8.46687626838684
    },
    {
      "epoch": 1.0295867919921874e-05,
      "model_forward_time": 1.7084126472473145,
      "step": 10796
    },
    {
      "epoch": 1.0295867919921874e-05,
      "step": 10796,
      "training_step_time": 8.129645109176636
    },
    {
      "epoch": 1.0296821594238282e-05,
      "model_forward_time": 1.7116754055023193,
      "step": 10797
    },
    {
      "epoch": 1.0296821594238282e-05,
      "step": 10797,
      "training_step_time": 9.617276191711426
    },
    {
      "epoch": 1.0297775268554688e-05,
      "model_forward_time": 1.7187082767486572,
      "step": 10798
    },
    {
      "epoch": 1.0297775268554688e-05,
      "step": 10798,
      "training_step_time": 7.699835777282715
    },
    {
      "epoch": 1.0298728942871093e-05,
      "model_forward_time": 1.7127680778503418,
      "step": 10799
    },
    {
      "epoch": 1.0298728942871093e-05,
      "step": 10799,
      "training_step_time": 8.76527214050293
    },
    {
      "epoch": 1.02996826171875e-05,
      "grad_norm": 0.28685829043388367,
      "learning_rate": 9.908342627657753e-06,
      "loss": 0.0675,
      "step": 10800
    },
    {
      "epoch": 1.02996826171875e-05,
      "model_forward_time": 2.2452335357666016,
      "step": 10800
    },
    {
      "dynamics_loss_avg": 0.06669383496046066,
      "epoch": 1.02996826171875e-05,
      "step": 10800
    },
    {
      "action_loss_avg": 0.005371724837459624,
      "epoch": 1.02996826171875e-05,
      "step": 10800
    },
    {
      "epoch": 1.02996826171875e-05,
      "step": 10800,
      "training_step_time": 7.806846857070923
    },
    {
      "epoch": 1.0300636291503906e-05,
      "model_forward_time": 1.7156453132629395,
      "step": 10801
    },
    {
      "epoch": 1.0300636291503906e-05,
      "step": 10801,
      "training_step_time": 7.712848424911499
    },
    {
      "epoch": 1.0301589965820312e-05,
      "model_forward_time": 1.33656644821167,
      "step": 10802
    },
    {
      "epoch": 1.0301589965820312e-05,
      "step": 10802,
      "training_step_time": 7.6207826137542725
    },
    {
      "epoch": 1.030254364013672e-05,
      "model_forward_time": 1.7178592681884766,
      "step": 10803
    },
    {
      "epoch": 1.030254364013672e-05,
      "step": 10803,
      "training_step_time": 7.6111297607421875
    },
    {
      "epoch": 1.0303497314453125e-05,
      "model_forward_time": 1.7572283744812012,
      "step": 10804
    },
    {
      "epoch": 1.0303497314453125e-05,
      "step": 10804,
      "training_step_time": 7.770689010620117
    },
    {
      "epoch": 1.0304450988769531e-05,
      "model_forward_time": 1.7076244354248047,
      "step": 10805
    },
    {
      "epoch": 1.0304450988769531e-05,
      "step": 10805,
      "training_step_time": 8.425120115280151
    },
    {
      "epoch": 1.0305404663085937e-05,
      "model_forward_time": 1.7145476341247559,
      "step": 10806
    },
    {
      "epoch": 1.0305404663085937e-05,
      "step": 10806,
      "training_step_time": 9.155977010726929
    },
    {
      "epoch": 1.0306358337402344e-05,
      "model_forward_time": 1.7234714031219482,
      "step": 10807
    },
    {
      "epoch": 1.0306358337402344e-05,
      "step": 10807,
      "training_step_time": 9.446057796478271
    },
    {
      "epoch": 1.030731201171875e-05,
      "model_forward_time": 1.7170398235321045,
      "step": 10808
    },
    {
      "epoch": 1.030731201171875e-05,
      "step": 10808,
      "training_step_time": 8.7960844039917
    },
    {
      "epoch": 1.0308265686035156e-05,
      "model_forward_time": 2.261470317840576,
      "step": 10809
    },
    {
      "epoch": 1.0308265686035156e-05,
      "step": 10809,
      "training_step_time": 7.769757509231567
    },
    {
      "epoch": 1.0309219360351563e-05,
      "grad_norm": 0.2148612141609192,
      "learning_rate": 9.90802721423869e-06,
      "loss": 0.0693,
      "step": 10810
    },
    {
      "epoch": 1.0309219360351563e-05,
      "model_forward_time": 1.70536470413208,
      "step": 10810
    },
    {
      "dynamics_loss_avg": 0.05061306096613407,
      "epoch": 1.0309219360351563e-05,
      "step": 10810
    },
    {
      "action_loss_avg": 0.004085974465124309,
      "epoch": 1.0309219360351563e-05,
      "step": 10810
    },
    {
      "epoch": 1.0309219360351563e-05,
      "step": 10810,
      "training_step_time": 7.843255043029785
    },
    {
      "epoch": 1.0310173034667969e-05,
      "model_forward_time": 1.7137181758880615,
      "step": 10811
    },
    {
      "epoch": 1.0310173034667969e-05,
      "step": 10811,
      "training_step_time": 7.646368503570557
    },
    {
      "epoch": 1.0311126708984374e-05,
      "model_forward_time": 1.7157044410705566,
      "step": 10812
    },
    {
      "epoch": 1.0311126708984374e-05,
      "step": 10812,
      "training_step_time": 7.596606254577637
    },
    {
      "epoch": 1.0312080383300782e-05,
      "model_forward_time": 1.746002197265625,
      "step": 10813
    },
    {
      "epoch": 1.0312080383300782e-05,
      "step": 10813,
      "training_step_time": 7.725163459777832
    },
    {
      "epoch": 1.0313034057617188e-05,
      "model_forward_time": 1.7096052169799805,
      "step": 10814
    },
    {
      "epoch": 1.0313034057617188e-05,
      "step": 10814,
      "training_step_time": 8.234959602355957
    },
    {
      "epoch": 1.0313987731933593e-05,
      "model_forward_time": 1.3468303680419922,
      "step": 10815
    },
    {
      "epoch": 1.0313987731933593e-05,
      "step": 10815,
      "training_step_time": 7.645809173583984
    },
    {
      "epoch": 1.031494140625e-05,
      "model_forward_time": 1.7066550254821777,
      "step": 10816
    },
    {
      "epoch": 1.031494140625e-05,
      "step": 10816,
      "training_step_time": 7.8054258823394775
    },
    {
      "epoch": 1.0315895080566406e-05,
      "model_forward_time": 1.7146720886230469,
      "step": 10817
    },
    {
      "epoch": 1.0315895080566406e-05,
      "step": 10817,
      "training_step_time": 7.904149770736694
    },
    {
      "epoch": 1.0316848754882812e-05,
      "model_forward_time": 2.2480790615081787,
      "step": 10818
    },
    {
      "epoch": 1.0316848754882812e-05,
      "step": 10818,
      "training_step_time": 7.748369216918945
    },
    {
      "epoch": 1.031780242919922e-05,
      "model_forward_time": 1.7123208045959473,
      "step": 10819
    },
    {
      "epoch": 1.031780242919922e-05,
      "step": 10819,
      "training_step_time": 8.012239933013916
    },
    {
      "epoch": 1.0318756103515625e-05,
      "grad_norm": 0.3302849531173706,
      "learning_rate": 9.90771126408515e-06,
      "loss": 0.0721,
      "step": 10820
    },
    {
      "epoch": 1.0318756103515625e-05,
      "model_forward_time": 1.6946783065795898,
      "step": 10820
    },
    {
      "dynamics_loss_avg": 0.040874730423092843,
      "epoch": 1.0318756103515625e-05,
      "step": 10820
    },
    {
      "action_loss_avg": 0.0023129104170948266,
      "epoch": 1.0318756103515625e-05,
      "step": 10820
    },
    {
      "epoch": 1.0318756103515625e-05,
      "step": 10820,
      "training_step_time": 7.770078420639038
    },
    {
      "epoch": 1.0319709777832031e-05,
      "model_forward_time": 1.705883264541626,
      "step": 10821
    },
    {
      "epoch": 1.0319709777832031e-05,
      "step": 10821,
      "training_step_time": 7.4587132930755615
    },
    {
      "epoch": 1.0320663452148437e-05,
      "model_forward_time": 1.7287724018096924,
      "step": 10822
    },
    {
      "epoch": 1.0320663452148437e-05,
      "step": 10822,
      "training_step_time": 7.852245092391968
    },
    {
      "epoch": 1.0321617126464844e-05,
      "model_forward_time": 1.7085931301116943,
      "step": 10823
    },
    {
      "epoch": 1.0321617126464844e-05,
      "step": 10823,
      "training_step_time": 7.811497688293457
    },
    {
      "epoch": 1.032257080078125e-05,
      "model_forward_time": 1.7232582569122314,
      "step": 10824
    },
    {
      "epoch": 1.032257080078125e-05,
      "step": 10824,
      "training_step_time": 8.968013048171997
    },
    {
      "epoch": 1.0323524475097656e-05,
      "model_forward_time": 1.7057185173034668,
      "step": 10825
    },
    {
      "epoch": 1.0323524475097656e-05,
      "step": 10825,
      "training_step_time": 8.807396650314331
    },
    {
      "epoch": 1.0324478149414063e-05,
      "model_forward_time": 1.7186541557312012,
      "step": 10826
    },
    {
      "epoch": 1.0324478149414063e-05,
      "step": 10826,
      "training_step_time": 9.447572469711304
    },
    {
      "epoch": 1.0325431823730469e-05,
      "model_forward_time": 1.7065932750701904,
      "step": 10827
    },
    {
      "epoch": 1.0325431823730469e-05,
      "step": 10827,
      "training_step_time": 7.8156657218933105
    },
    {
      "epoch": 1.0326385498046874e-05,
      "model_forward_time": 2.241196393966675,
      "step": 10828
    },
    {
      "epoch": 1.0326385498046874e-05,
      "step": 10828,
      "training_step_time": 10.611819505691528
    },
    {
      "epoch": 1.0327339172363282e-05,
      "model_forward_time": 1.70723557472229,
      "step": 10829
    },
    {
      "epoch": 1.0327339172363282e-05,
      "step": 10829,
      "training_step_time": 8.217456340789795
    },
    {
      "epoch": 1.0328292846679688e-05,
      "grad_norm": 0.27672258019447327,
      "learning_rate": 9.907394777231686e-06,
      "loss": 0.0752,
      "step": 10830
    },
    {
      "epoch": 1.0328292846679688e-05,
      "model_forward_time": 1.6987977027893066,
      "step": 10830
    },
    {
      "dynamics_loss_avg": 0.07393009327352047,
      "epoch": 1.0328292846679688e-05,
      "step": 10830
    },
    {
      "action_loss_avg": 0.0054156702477484945,
      "epoch": 1.0328292846679688e-05,
      "step": 10830
    },
    {
      "epoch": 1.0328292846679688e-05,
      "step": 10830,
      "training_step_time": 7.4948344230651855
    },
    {
      "epoch": 1.0329246520996093e-05,
      "model_forward_time": 1.7049739360809326,
      "step": 10831
    },
    {
      "epoch": 1.0329246520996093e-05,
      "step": 10831,
      "training_step_time": 7.665390491485596
    },
    {
      "epoch": 1.03302001953125e-05,
      "model_forward_time": 1.752002239227295,
      "step": 10832
    },
    {
      "epoch": 1.03302001953125e-05,
      "step": 10832,
      "training_step_time": 7.709004878997803
    },
    {
      "epoch": 1.0331153869628906e-05,
      "model_forward_time": 1.7033438682556152,
      "step": 10833
    },
    {
      "epoch": 1.0331153869628906e-05,
      "step": 10833,
      "training_step_time": 7.547636032104492
    },
    {
      "epoch": 1.0332107543945312e-05,
      "model_forward_time": 1.7014875411987305,
      "step": 10834
    },
    {
      "epoch": 1.0332107543945312e-05,
      "step": 10834,
      "training_step_time": 7.703828573226929
    },
    {
      "epoch": 1.033306121826172e-05,
      "model_forward_time": 1.7049391269683838,
      "step": 10835
    },
    {
      "epoch": 1.033306121826172e-05,
      "step": 10835,
      "training_step_time": 8.28147268295288
    },
    {
      "epoch": 1.0334014892578125e-05,
      "model_forward_time": 1.3311817646026611,
      "step": 10836
    },
    {
      "epoch": 1.0334014892578125e-05,
      "step": 10836,
      "training_step_time": 7.795581340789795
    },
    {
      "epoch": 1.0334968566894531e-05,
      "model_forward_time": 1.8518791198730469,
      "step": 10837
    },
    {
      "epoch": 1.0334968566894531e-05,
      "step": 10837,
      "training_step_time": 8.55864405632019
    },
    {
      "epoch": 1.0335922241210937e-05,
      "model_forward_time": 1.7156989574432373,
      "step": 10838
    },
    {
      "epoch": 1.0335922241210937e-05,
      "step": 10838,
      "training_step_time": 8.46011757850647
    },
    {
      "epoch": 1.0336875915527344e-05,
      "model_forward_time": 1.7119450569152832,
      "step": 10839
    },
    {
      "epoch": 1.0336875915527344e-05,
      "step": 10839,
      "training_step_time": 8.394794464111328
    },
    {
      "epoch": 1.033782958984375e-05,
      "grad_norm": 0.23663221299648285,
      "learning_rate": 9.907077753712906e-06,
      "loss": 0.0691,
      "step": 10840
    },
    {
      "epoch": 1.033782958984375e-05,
      "model_forward_time": 1.7049469947814941,
      "step": 10840
    },
    {
      "dynamics_loss_avg": 0.06960468180477619,
      "epoch": 1.033782958984375e-05,
      "step": 10840
    },
    {
      "action_loss_avg": 0.004666791181080043,
      "epoch": 1.033782958984375e-05,
      "step": 10840
    },
    {
      "epoch": 1.033782958984375e-05,
      "step": 10840,
      "training_step_time": 8.37124228477478
    },
    {
      "epoch": 1.0338783264160156e-05,
      "model_forward_time": 1.7519679069519043,
      "step": 10841
    },
    {
      "epoch": 1.0338783264160156e-05,
      "step": 10841,
      "training_step_time": 7.682475566864014
    },
    {
      "epoch": 1.0339736938476563e-05,
      "model_forward_time": 1.709913730621338,
      "step": 10842
    },
    {
      "epoch": 1.0339736938476563e-05,
      "step": 10842,
      "training_step_time": 7.368921279907227
    },
    {
      "epoch": 1.0340690612792969e-05,
      "model_forward_time": 1.7082278728485107,
      "step": 10843
    },
    {
      "epoch": 1.0340690612792969e-05,
      "step": 10843,
      "training_step_time": 8.945410013198853
    },
    {
      "epoch": 1.0341644287109374e-05,
      "model_forward_time": 1.7064683437347412,
      "step": 10844
    },
    {
      "epoch": 1.0341644287109374e-05,
      "step": 10844,
      "training_step_time": 7.8212268352508545
    },
    {
      "epoch": 1.0342597961425782e-05,
      "model_forward_time": 1.716270923614502,
      "step": 10845
    },
    {
      "epoch": 1.0342597961425782e-05,
      "step": 10845,
      "training_step_time": 7.691989421844482
    },
    {
      "epoch": 1.0343551635742188e-05,
      "model_forward_time": 2.2473881244659424,
      "step": 10846
    },
    {
      "epoch": 1.0343551635742188e-05,
      "step": 10846,
      "training_step_time": 8.83274221420288
    },
    {
      "epoch": 1.0344505310058593e-05,
      "model_forward_time": 1.7050633430480957,
      "step": 10847
    },
    {
      "epoch": 1.0344505310058593e-05,
      "step": 10847,
      "training_step_time": 7.694200277328491
    },
    {
      "epoch": 1.0345458984375e-05,
      "model_forward_time": 1.7079319953918457,
      "step": 10848
    },
    {
      "epoch": 1.0345458984375e-05,
      "step": 10848,
      "training_step_time": 7.718918561935425
    },
    {
      "epoch": 1.0346412658691406e-05,
      "model_forward_time": 1.707946538925171,
      "step": 10849
    },
    {
      "epoch": 1.0346412658691406e-05,
      "step": 10849,
      "training_step_time": 7.736482620239258
    },
    {
      "epoch": 1.0347366333007812e-05,
      "grad_norm": 0.1594008207321167,
      "learning_rate": 9.906760193563482e-06,
      "loss": 0.0701,
      "step": 10850
    },
    {
      "epoch": 1.0347366333007812e-05,
      "model_forward_time": 1.7475011348724365,
      "step": 10850
    },
    {
      "dynamics_loss_avg": 0.06312701180577278,
      "epoch": 1.0347366333007812e-05,
      "step": 10850
    },
    {
      "action_loss_avg": 0.003978949750307948,
      "epoch": 1.0347366333007812e-05,
      "step": 10850
    },
    {
      "epoch": 1.0347366333007812e-05,
      "step": 10850,
      "training_step_time": 7.677897930145264
    },
    {
      "epoch": 1.034832000732422e-05,
      "model_forward_time": 1.7093291282653809,
      "step": 10851
    },
    {
      "epoch": 1.034832000732422e-05,
      "step": 10851,
      "training_step_time": 7.5910563468933105
    },
    {
      "epoch": 1.0349273681640625e-05,
      "model_forward_time": 1.7062606811523438,
      "step": 10852
    },
    {
      "epoch": 1.0349273681640625e-05,
      "step": 10852,
      "training_step_time": 8.755985021591187
    },
    {
      "epoch": 1.0350227355957031e-05,
      "model_forward_time": 1.707458734512329,
      "step": 10853
    },
    {
      "epoch": 1.0350227355957031e-05,
      "step": 10853,
      "training_step_time": 9.053122520446777
    },
    {
      "epoch": 1.0351181030273437e-05,
      "model_forward_time": 1.7062742710113525,
      "step": 10854
    },
    {
      "epoch": 1.0351181030273437e-05,
      "step": 10854,
      "training_step_time": 7.667913913726807
    },
    {
      "epoch": 1.0352134704589844e-05,
      "model_forward_time": 2.2223823070526123,
      "step": 10855
    },
    {
      "epoch": 1.0352134704589844e-05,
      "step": 10855,
      "training_step_time": 8.752775430679321
    },
    {
      "epoch": 1.035308837890625e-05,
      "model_forward_time": 1.7081620693206787,
      "step": 10856
    },
    {
      "epoch": 1.035308837890625e-05,
      "step": 10856,
      "training_step_time": 7.671060085296631
    },
    {
      "epoch": 1.0354042053222656e-05,
      "model_forward_time": 1.708299160003662,
      "step": 10857
    },
    {
      "epoch": 1.0354042053222656e-05,
      "step": 10857,
      "training_step_time": 8.200240850448608
    },
    {
      "epoch": 1.0354995727539063e-05,
      "model_forward_time": 1.710059404373169,
      "step": 10858
    },
    {
      "epoch": 1.0354995727539063e-05,
      "step": 10858,
      "training_step_time": 7.698638200759888
    },
    {
      "epoch": 1.0355949401855469e-05,
      "model_forward_time": 1.7353737354278564,
      "step": 10859
    },
    {
      "epoch": 1.0355949401855469e-05,
      "step": 10859,
      "training_step_time": 7.636090278625488
    },
    {
      "epoch": 1.0356903076171874e-05,
      "grad_norm": 0.15098968148231506,
      "learning_rate": 9.906442096818138e-06,
      "loss": 0.0651,
      "step": 10860
    },
    {
      "epoch": 1.0356903076171874e-05,
      "model_forward_time": 1.7036242485046387,
      "step": 10860
    },
    {
      "dynamics_loss_avg": 0.0718739502131939,
      "epoch": 1.0356903076171874e-05,
      "step": 10860
    },
    {
      "action_loss_avg": 0.004692568199243396,
      "epoch": 1.0356903076171874e-05,
      "step": 10860
    },
    {
      "epoch": 1.0356903076171874e-05,
      "step": 10860,
      "training_step_time": 7.54401421546936
    },
    {
      "epoch": 1.0357856750488282e-05,
      "model_forward_time": 1.7088024616241455,
      "step": 10861
    },
    {
      "epoch": 1.0357856750488282e-05,
      "step": 10861,
      "training_step_time": 7.747258424758911
    },
    {
      "epoch": 1.0358810424804688e-05,
      "model_forward_time": 1.707137107849121,
      "step": 10862
    },
    {
      "epoch": 1.0358810424804688e-05,
      "step": 10862,
      "training_step_time": 8.476248979568481
    },
    {
      "epoch": 1.0359764099121093e-05,
      "model_forward_time": 1.7074518203735352,
      "step": 10863
    },
    {
      "epoch": 1.0359764099121093e-05,
      "step": 10863,
      "training_step_time": 8.56775951385498
    },
    {
      "epoch": 1.03607177734375e-05,
      "model_forward_time": 2.2424476146698,
      "step": 10864
    },
    {
      "epoch": 1.03607177734375e-05,
      "step": 10864,
      "training_step_time": 7.719677686691284
    },
    {
      "epoch": 1.0361671447753906e-05,
      "model_forward_time": 1.7101290225982666,
      "step": 10865
    },
    {
      "epoch": 1.0361671447753906e-05,
      "step": 10865,
      "training_step_time": 7.76994252204895
    },
    {
      "epoch": 1.0362625122070312e-05,
      "model_forward_time": 1.7112877368927002,
      "step": 10866
    },
    {
      "epoch": 1.0362625122070312e-05,
      "step": 10866,
      "training_step_time": 7.712981939315796
    },
    {
      "epoch": 1.036357879638672e-05,
      "model_forward_time": 1.7125766277313232,
      "step": 10867
    },
    {
      "epoch": 1.036357879638672e-05,
      "step": 10867,
      "training_step_time": 7.697067975997925
    },
    {
      "epoch": 1.0364532470703125e-05,
      "model_forward_time": 1.7116203308105469,
      "step": 10868
    },
    {
      "epoch": 1.0364532470703125e-05,
      "step": 10868,
      "training_step_time": 7.681591987609863
    },
    {
      "epoch": 1.0365486145019531e-05,
      "model_forward_time": 1.7076003551483154,
      "step": 10869
    },
    {
      "epoch": 1.0365486145019531e-05,
      "step": 10869,
      "training_step_time": 8.104747772216797
    },
    {
      "epoch": 1.0366439819335937e-05,
      "grad_norm": 0.17779186367988586,
      "learning_rate": 9.906123463511665e-06,
      "loss": 0.0655,
      "step": 10870
    },
    {
      "epoch": 1.0366439819335937e-05,
      "model_forward_time": 1.7069544792175293,
      "step": 10870
    },
    {
      "dynamics_loss_avg": 0.05884755020961165,
      "epoch": 1.0366439819335937e-05,
      "step": 10870
    },
    {
      "action_loss_avg": 0.006541477120481432,
      "epoch": 1.0366439819335937e-05,
      "step": 10870
    },
    {
      "epoch": 1.0366439819335937e-05,
      "step": 10870,
      "training_step_time": 7.614353895187378
    },
    {
      "epoch": 1.0367393493652344e-05,
      "model_forward_time": 1.7025461196899414,
      "step": 10871
    },
    {
      "epoch": 1.0367393493652344e-05,
      "step": 10871,
      "training_step_time": 8.53373908996582
    },
    {
      "epoch": 1.036834716796875e-05,
      "model_forward_time": 1.7096824645996094,
      "step": 10872
    },
    {
      "epoch": 1.036834716796875e-05,
      "step": 10872,
      "training_step_time": 8.3106689453125
    },
    {
      "epoch": 1.0369300842285156e-05,
      "model_forward_time": 1.7099223136901855,
      "step": 10873
    },
    {
      "epoch": 1.0369300842285156e-05,
      "step": 10873,
      "training_step_time": 7.75468111038208
    },
    {
      "epoch": 1.0370254516601563e-05,
      "model_forward_time": 2.2289633750915527,
      "step": 10874
    },
    {
      "epoch": 1.0370254516601563e-05,
      "step": 10874,
      "training_step_time": 8.890289783477783
    },
    {
      "epoch": 1.0371208190917969e-05,
      "model_forward_time": 1.707592487335205,
      "step": 10875
    },
    {
      "epoch": 1.0371208190917969e-05,
      "step": 10875,
      "training_step_time": 8.754642248153687
    },
    {
      "epoch": 1.0372161865234374e-05,
      "model_forward_time": 1.711350679397583,
      "step": 10876
    },
    {
      "epoch": 1.0372161865234374e-05,
      "step": 10876,
      "training_step_time": 8.769900560379028
    },
    {
      "epoch": 1.0373115539550782e-05,
      "model_forward_time": 1.7124154567718506,
      "step": 10877
    },
    {
      "epoch": 1.0373115539550782e-05,
      "step": 10877,
      "training_step_time": 7.746448040008545
    },
    {
      "epoch": 1.0374069213867188e-05,
      "model_forward_time": 1.7462356090545654,
      "step": 10878
    },
    {
      "epoch": 1.0374069213867188e-05,
      "step": 10878,
      "training_step_time": 7.541615962982178
    },
    {
      "epoch": 1.0375022888183593e-05,
      "model_forward_time": 1.7115583419799805,
      "step": 10879
    },
    {
      "epoch": 1.0375022888183593e-05,
      "step": 10879,
      "training_step_time": 7.609676837921143
    },
    {
      "epoch": 1.03759765625e-05,
      "grad_norm": 0.17240041494369507,
      "learning_rate": 9.905804293678907e-06,
      "loss": 0.0709,
      "step": 10880
    },
    {
      "epoch": 1.03759765625e-05,
      "model_forward_time": 1.7052204608917236,
      "step": 10880
    },
    {
      "dynamics_loss_avg": 0.05655352622270584,
      "epoch": 1.03759765625e-05,
      "step": 10880
    },
    {
      "action_loss_avg": 0.0034141454729251564,
      "epoch": 1.03759765625e-05,
      "step": 10880
    },
    {
      "epoch": 1.03759765625e-05,
      "step": 10880,
      "training_step_time": 8.668135404586792
    },
    {
      "epoch": 1.0376930236816406e-05,
      "model_forward_time": 1.7096116542816162,
      "step": 10881
    },
    {
      "epoch": 1.0376930236816406e-05,
      "step": 10881,
      "training_step_time": 8.752232789993286
    },
    {
      "epoch": 1.0377883911132812e-05,
      "model_forward_time": 1.7155654430389404,
      "step": 10882
    },
    {
      "epoch": 1.0377883911132812e-05,
      "step": 10882,
      "training_step_time": 8.618936538696289
    },
    {
      "epoch": 1.0378837585449218e-05,
      "model_forward_time": 2.237373113632202,
      "step": 10883
    },
    {
      "epoch": 1.0378837585449218e-05,
      "step": 10883,
      "training_step_time": 7.8438873291015625
    },
    {
      "epoch": 1.0379791259765625e-05,
      "model_forward_time": 1.7088735103607178,
      "step": 10884
    },
    {
      "epoch": 1.0379791259765625e-05,
      "step": 10884,
      "training_step_time": 7.711025714874268
    },
    {
      "epoch": 1.0380744934082031e-05,
      "model_forward_time": 1.7089552879333496,
      "step": 10885
    },
    {
      "epoch": 1.0380744934082031e-05,
      "step": 10885,
      "training_step_time": 7.738319635391235
    },
    {
      "epoch": 1.0381698608398437e-05,
      "model_forward_time": 1.7098190784454346,
      "step": 10886
    },
    {
      "epoch": 1.0381698608398437e-05,
      "step": 10886,
      "training_step_time": 9.307454824447632
    },
    {
      "epoch": 1.0382652282714844e-05,
      "model_forward_time": 1.7499303817749023,
      "step": 10887
    },
    {
      "epoch": 1.0382652282714844e-05,
      "step": 10887,
      "training_step_time": 8.245371580123901
    },
    {
      "epoch": 1.038360595703125e-05,
      "model_forward_time": 1.7087512016296387,
      "step": 10888
    },
    {
      "epoch": 1.038360595703125e-05,
      "step": 10888,
      "training_step_time": 8.43167233467102
    },
    {
      "epoch": 1.0384559631347656e-05,
      "model_forward_time": 1.7221989631652832,
      "step": 10889
    },
    {
      "epoch": 1.0384559631347656e-05,
      "step": 10889,
      "training_step_time": 8.731426239013672
    },
    {
      "epoch": 1.0385513305664063e-05,
      "grad_norm": 0.17965398728847504,
      "learning_rate": 9.905484587354765e-06,
      "loss": 0.0687,
      "step": 10890
    },
    {
      "epoch": 1.0385513305664063e-05,
      "model_forward_time": 1.692969560623169,
      "step": 10890
    },
    {
      "dynamics_loss_avg": 0.05656279046088457,
      "epoch": 1.0385513305664063e-05,
      "step": 10890
    },
    {
      "action_loss_avg": 0.0030571660725399854,
      "epoch": 1.0385513305664063e-05,
      "step": 10890
    },
    {
      "epoch": 1.0385513305664063e-05,
      "step": 10890,
      "training_step_time": 8.872992753982544
    },
    {
      "epoch": 1.0386466979980469e-05,
      "model_forward_time": 1.7067275047302246,
      "step": 10891
    },
    {
      "epoch": 1.0386466979980469e-05,
      "step": 10891,
      "training_step_time": 9.761507511138916
    },
    {
      "epoch": 1.0387420654296874e-05,
      "model_forward_time": 2.237499237060547,
      "step": 10892
    },
    {
      "epoch": 1.0387420654296874e-05,
      "step": 10892,
      "training_step_time": 9.669410467147827
    },
    {
      "epoch": 1.0388374328613282e-05,
      "model_forward_time": 1.7013213634490967,
      "step": 10893
    },
    {
      "epoch": 1.0388374328613282e-05,
      "step": 10893,
      "training_step_time": 8.582311153411865
    },
    {
      "epoch": 1.0389328002929688e-05,
      "model_forward_time": 1.7089436054229736,
      "step": 10894
    },
    {
      "epoch": 1.0389328002929688e-05,
      "step": 10894,
      "training_step_time": 7.743184566497803
    },
    {
      "epoch": 1.0390281677246093e-05,
      "model_forward_time": 1.7064251899719238,
      "step": 10895
    },
    {
      "epoch": 1.0390281677246093e-05,
      "step": 10895,
      "training_step_time": 8.692233085632324
    },
    {
      "epoch": 1.03912353515625e-05,
      "model_forward_time": 1.7523276805877686,
      "step": 10896
    },
    {
      "epoch": 1.03912353515625e-05,
      "step": 10896,
      "training_step_time": 8.186604499816895
    },
    {
      "epoch": 1.0392189025878906e-05,
      "model_forward_time": 1.7124683856964111,
      "step": 10897
    },
    {
      "epoch": 1.0392189025878906e-05,
      "step": 10897,
      "training_step_time": 7.5964601039886475
    },
    {
      "epoch": 1.0393142700195312e-05,
      "model_forward_time": 1.7068464756011963,
      "step": 10898
    },
    {
      "epoch": 1.0393142700195312e-05,
      "step": 10898,
      "training_step_time": 9.839672088623047
    },
    {
      "epoch": 1.0394096374511718e-05,
      "model_forward_time": 1.7063405513763428,
      "step": 10899
    },
    {
      "epoch": 1.0394096374511718e-05,
      "step": 10899,
      "training_step_time": 7.807248115539551
    },
    {
      "epoch": 1.0395050048828125e-05,
      "grad_norm": 0.20903395116329193,
      "learning_rate": 9.905164344574205e-06,
      "loss": 0.0675,
      "step": 10900
    },
    {
      "epoch": 1.0395050048828125e-05,
      "model_forward_time": 1.7008640766143799,
      "step": 10900
    },
    {
      "dynamics_loss_avg": 0.06922947950661182,
      "epoch": 1.0395050048828125e-05,
      "step": 10900
    },
    {
      "action_loss_avg": 0.004451407142914832,
      "epoch": 1.0395050048828125e-05,
      "step": 10900
    },
    {
      "epoch": 1.0395050048828125e-05,
      "step": 10900,
      "training_step_time": 7.705693483352661
    },
    {
      "epoch": 1.0396003723144531e-05,
      "model_forward_time": 2.25490403175354,
      "step": 10901
    },
    {
      "epoch": 1.0396003723144531e-05,
      "step": 10901,
      "training_step_time": 7.662318706512451
    },
    {
      "epoch": 1.0396957397460937e-05,
      "model_forward_time": 1.709610939025879,
      "step": 10902
    },
    {
      "epoch": 1.0396957397460937e-05,
      "step": 10902,
      "training_step_time": 7.7477686405181885
    },
    {
      "epoch": 1.0397911071777344e-05,
      "model_forward_time": 1.7026290893554688,
      "step": 10903
    },
    {
      "epoch": 1.0397911071777344e-05,
      "step": 10903,
      "training_step_time": 7.694012403488159
    },
    {
      "epoch": 1.039886474609375e-05,
      "model_forward_time": 1.7129583358764648,
      "step": 10904
    },
    {
      "epoch": 1.039886474609375e-05,
      "step": 10904,
      "training_step_time": 8.382551431655884
    },
    {
      "epoch": 1.0399818420410156e-05,
      "model_forward_time": 1.7472190856933594,
      "step": 10905
    },
    {
      "epoch": 1.0399818420410156e-05,
      "step": 10905,
      "training_step_time": 7.628825902938843
    },
    {
      "epoch": 1.0400772094726563e-05,
      "model_forward_time": 1.7010507583618164,
      "step": 10906
    },
    {
      "epoch": 1.0400772094726563e-05,
      "step": 10906,
      "training_step_time": 7.558279514312744
    },
    {
      "epoch": 1.0401725769042969e-05,
      "model_forward_time": 1.3345777988433838,
      "step": 10907
    },
    {
      "epoch": 1.0401725769042969e-05,
      "step": 10907,
      "training_step_time": 8.56868290901184
    },
    {
      "epoch": 1.0402679443359374e-05,
      "model_forward_time": 1.3276708126068115,
      "step": 10908
    },
    {
      "epoch": 1.0402679443359374e-05,
      "step": 10908,
      "training_step_time": 10.773362159729004
    },
    {
      "epoch": 1.0403633117675782e-05,
      "model_forward_time": 1.7087740898132324,
      "step": 10909
    },
    {
      "epoch": 1.0403633117675782e-05,
      "step": 10909,
      "training_step_time": 7.722111701965332
    },
    {
      "epoch": 1.0404586791992188e-05,
      "grad_norm": 0.24428033828735352,
      "learning_rate": 9.904843565372249e-06,
      "loss": 0.0707,
      "step": 10910
    },
    {
      "epoch": 1.0404586791992188e-05,
      "model_forward_time": 1.7055222988128662,
      "step": 10910
    },
    {
      "dynamics_loss_avg": 0.0788247250020504,
      "epoch": 1.0404586791992188e-05,
      "step": 10910
    },
    {
      "action_loss_avg": 0.003279790008673444,
      "epoch": 1.0404586791992188e-05,
      "step": 10910
    },
    {
      "epoch": 1.0404586791992188e-05,
      "step": 10910,
      "training_step_time": 8.400085926055908
    },
    {
      "epoch": 1.0405540466308593e-05,
      "model_forward_time": 2.237645149230957,
      "step": 10911
    },
    {
      "epoch": 1.0405540466308593e-05,
      "step": 10911,
      "training_step_time": 7.875781536102295
    },
    {
      "epoch": 1.0406494140625e-05,
      "model_forward_time": 1.7108054161071777,
      "step": 10912
    },
    {
      "epoch": 1.0406494140625e-05,
      "step": 10912,
      "training_step_time": 8.21998119354248
    },
    {
      "epoch": 1.0407447814941406e-05,
      "model_forward_time": 1.7088732719421387,
      "step": 10913
    },
    {
      "epoch": 1.0407447814941406e-05,
      "step": 10913,
      "training_step_time": 7.674747705459595
    },
    {
      "epoch": 1.0408401489257812e-05,
      "model_forward_time": 1.7291288375854492,
      "step": 10914
    },
    {
      "epoch": 1.0408401489257812e-05,
      "step": 10914,
      "training_step_time": 7.520052194595337
    },
    {
      "epoch": 1.0409355163574218e-05,
      "model_forward_time": 1.7079010009765625,
      "step": 10915
    },
    {
      "epoch": 1.0409355163574218e-05,
      "step": 10915,
      "training_step_time": 7.441221237182617
    },
    {
      "epoch": 1.0410308837890625e-05,
      "model_forward_time": 1.3311622142791748,
      "step": 10916
    },
    {
      "epoch": 1.0410308837890625e-05,
      "step": 10916,
      "training_step_time": 8.099039793014526
    },
    {
      "epoch": 1.0411262512207031e-05,
      "model_forward_time": 1.706080436706543,
      "step": 10917
    },
    {
      "epoch": 1.0411262512207031e-05,
      "step": 10917,
      "training_step_time": 8.454997062683105
    },
    {
      "epoch": 1.0412216186523437e-05,
      "model_forward_time": 1.7026598453521729,
      "step": 10918
    },
    {
      "epoch": 1.0412216186523437e-05,
      "step": 10918,
      "training_step_time": 8.059019088745117
    },
    {
      "epoch": 1.0413169860839844e-05,
      "model_forward_time": 1.7082140445709229,
      "step": 10919
    },
    {
      "epoch": 1.0413169860839844e-05,
      "step": 10919,
      "training_step_time": 7.621218681335449
    },
    {
      "epoch": 1.041412353515625e-05,
      "grad_norm": 0.25845757126808167,
      "learning_rate": 9.904522249783972e-06,
      "loss": 0.0663,
      "step": 10920
    },
    {
      "epoch": 1.041412353515625e-05,
      "model_forward_time": 2.240689992904663,
      "step": 10920
    },
    {
      "dynamics_loss_avg": 0.043124768789857625,
      "epoch": 1.041412353515625e-05,
      "step": 10920
    },
    {
      "action_loss_avg": 0.0026603958918713032,
      "epoch": 1.041412353515625e-05,
      "step": 10920
    },
    {
      "epoch": 1.041412353515625e-05,
      "step": 10920,
      "training_step_time": 8.427907466888428
    },
    {
      "epoch": 1.0415077209472656e-05,
      "model_forward_time": 1.7045047283172607,
      "step": 10921
    },
    {
      "epoch": 1.0415077209472656e-05,
      "step": 10921,
      "training_step_time": 7.5765485763549805
    },
    {
      "epoch": 1.0416030883789063e-05,
      "model_forward_time": 1.712918758392334,
      "step": 10922
    },
    {
      "epoch": 1.0416030883789063e-05,
      "step": 10922,
      "training_step_time": 7.680943012237549
    },
    {
      "epoch": 1.0416984558105469e-05,
      "model_forward_time": 1.7490723133087158,
      "step": 10923
    },
    {
      "epoch": 1.0416984558105469e-05,
      "step": 10923,
      "training_step_time": 9.575895309448242
    },
    {
      "epoch": 1.0417938232421875e-05,
      "model_forward_time": 1.706895351409912,
      "step": 10924
    },
    {
      "epoch": 1.0417938232421875e-05,
      "step": 10924,
      "training_step_time": 7.570780277252197
    },
    {
      "epoch": 1.0418891906738282e-05,
      "model_forward_time": 1.7138798236846924,
      "step": 10925
    },
    {
      "epoch": 1.0418891906738282e-05,
      "step": 10925,
      "training_step_time": 7.590346574783325
    },
    {
      "epoch": 1.0419845581054688e-05,
      "model_forward_time": 1.330514669418335,
      "step": 10926
    },
    {
      "epoch": 1.0419845581054688e-05,
      "step": 10926,
      "training_step_time": 9.289241790771484
    },
    {
      "epoch": 1.0420799255371093e-05,
      "model_forward_time": 1.705343246459961,
      "step": 10927
    },
    {
      "epoch": 1.0420799255371093e-05,
      "step": 10927,
      "training_step_time": 8.66921854019165
    },
    {
      "epoch": 1.04217529296875e-05,
      "model_forward_time": 1.7026739120483398,
      "step": 10928
    },
    {
      "epoch": 1.04217529296875e-05,
      "step": 10928,
      "training_step_time": 7.594258785247803
    },
    {
      "epoch": 1.0422706604003907e-05,
      "model_forward_time": 2.249464988708496,
      "step": 10929
    },
    {
      "epoch": 1.0422706604003907e-05,
      "step": 10929,
      "training_step_time": 7.548297166824341
    },
    {
      "epoch": 1.0423660278320312e-05,
      "grad_norm": 0.2629391551017761,
      "learning_rate": 9.904200397844517e-06,
      "loss": 0.0703,
      "step": 10930
    },
    {
      "epoch": 1.0423660278320312e-05,
      "model_forward_time": 1.7063031196594238,
      "step": 10930
    },
    {
      "dynamics_loss_avg": 0.05048362584784627,
      "epoch": 1.0423660278320312e-05,
      "step": 10930
    },
    {
      "action_loss_avg": 0.0025223852251656355,
      "epoch": 1.0423660278320312e-05,
      "step": 10930
    },
    {
      "epoch": 1.0423660278320312e-05,
      "step": 10930,
      "training_step_time": 7.643038749694824
    },
    {
      "epoch": 1.0424613952636718e-05,
      "model_forward_time": 1.7083370685577393,
      "step": 10931
    },
    {
      "epoch": 1.0424613952636718e-05,
      "step": 10931,
      "training_step_time": 7.604778289794922
    },
    {
      "epoch": 1.0425567626953125e-05,
      "model_forward_time": 1.7486965656280518,
      "step": 10932
    },
    {
      "epoch": 1.0425567626953125e-05,
      "step": 10932,
      "training_step_time": 8.765965223312378
    },
    {
      "epoch": 1.0426521301269531e-05,
      "model_forward_time": 1.7067310810089111,
      "step": 10933
    },
    {
      "epoch": 1.0426521301269531e-05,
      "step": 10933,
      "training_step_time": 7.5089709758758545
    },
    {
      "epoch": 1.0427474975585937e-05,
      "model_forward_time": 1.7033491134643555,
      "step": 10934
    },
    {
      "epoch": 1.0427474975585937e-05,
      "step": 10934,
      "training_step_time": 7.407514810562134
    },
    {
      "epoch": 1.0428428649902344e-05,
      "model_forward_time": 1.7061560153961182,
      "step": 10935
    },
    {
      "epoch": 1.0428428649902344e-05,
      "step": 10935,
      "training_step_time": 7.65281867980957
    },
    {
      "epoch": 1.042938232421875e-05,
      "model_forward_time": 1.7056007385253906,
      "step": 10936
    },
    {
      "epoch": 1.042938232421875e-05,
      "step": 10936,
      "training_step_time": 8.458991050720215
    },
    {
      "epoch": 1.0430335998535156e-05,
      "model_forward_time": 1.7072997093200684,
      "step": 10937
    },
    {
      "epoch": 1.0430335998535156e-05,
      "step": 10937,
      "training_step_time": 7.5909340381622314
    },
    {
      "epoch": 1.0431289672851563e-05,
      "model_forward_time": 2.2670884132385254,
      "step": 10938
    },
    {
      "epoch": 1.0431289672851563e-05,
      "step": 10938,
      "training_step_time": 9.159600734710693
    },
    {
      "epoch": 1.0432243347167969e-05,
      "model_forward_time": 1.7083096504211426,
      "step": 10939
    },
    {
      "epoch": 1.0432243347167969e-05,
      "step": 10939,
      "training_step_time": 8.455698013305664
    },
    {
      "epoch": 1.0433197021484375e-05,
      "grad_norm": 0.29120925068855286,
      "learning_rate": 9.903878009589079e-06,
      "loss": 0.0684,
      "step": 10940
    },
    {
      "epoch": 1.0433197021484375e-05,
      "model_forward_time": 1.7117280960083008,
      "step": 10940
    },
    {
      "dynamics_loss_avg": 0.0799258353188634,
      "epoch": 1.0433197021484375e-05,
      "step": 10940
    },
    {
      "action_loss_avg": 0.006877544010058046,
      "epoch": 1.0433197021484375e-05,
      "step": 10940
    },
    {
      "epoch": 1.0433197021484375e-05,
      "step": 10940,
      "training_step_time": 8.103448152542114
    },
    {
      "epoch": 1.0434150695800782e-05,
      "model_forward_time": 1.7499849796295166,
      "step": 10941
    },
    {
      "epoch": 1.0434150695800782e-05,
      "step": 10941,
      "training_step_time": 7.917926073074341
    },
    {
      "epoch": 1.0435104370117188e-05,
      "model_forward_time": 1.7083804607391357,
      "step": 10942
    },
    {
      "epoch": 1.0435104370117188e-05,
      "step": 10942,
      "training_step_time": 7.488976240158081
    },
    {
      "epoch": 1.0436058044433593e-05,
      "model_forward_time": 1.7050647735595703,
      "step": 10943
    },
    {
      "epoch": 1.0436058044433593e-05,
      "step": 10943,
      "training_step_time": 7.472477197647095
    },
    {
      "epoch": 1.043701171875e-05,
      "model_forward_time": 1.7103896141052246,
      "step": 10944
    },
    {
      "epoch": 1.043701171875e-05,
      "step": 10944,
      "training_step_time": 8.313846111297607
    },
    {
      "epoch": 1.0437965393066407e-05,
      "model_forward_time": 1.7022230625152588,
      "step": 10945
    },
    {
      "epoch": 1.0437965393066407e-05,
      "step": 10945,
      "training_step_time": 7.656253099441528
    },
    {
      "epoch": 1.0438919067382812e-05,
      "model_forward_time": 1.7075703144073486,
      "step": 10946
    },
    {
      "epoch": 1.0438919067382812e-05,
      "step": 10946,
      "training_step_time": 7.608960151672363
    },
    {
      "epoch": 1.0439872741699218e-05,
      "model_forward_time": 2.2212183475494385,
      "step": 10947
    },
    {
      "epoch": 1.0439872741699218e-05,
      "step": 10947,
      "training_step_time": 7.483150005340576
    },
    {
      "epoch": 1.0440826416015625e-05,
      "model_forward_time": 1.3267302513122559,
      "step": 10948
    },
    {
      "epoch": 1.0440826416015625e-05,
      "step": 10948,
      "training_step_time": 8.265483140945435
    },
    {
      "epoch": 1.0441780090332031e-05,
      "model_forward_time": 1.713493824005127,
      "step": 10949
    },
    {
      "epoch": 1.0441780090332031e-05,
      "step": 10949,
      "training_step_time": 7.592506647109985
    },
    {
      "epoch": 1.0442733764648437e-05,
      "grad_norm": 0.16308996081352234,
      "learning_rate": 9.903555085052916e-06,
      "loss": 0.0667,
      "step": 10950
    },
    {
      "epoch": 1.0442733764648437e-05,
      "model_forward_time": 1.7403171062469482,
      "step": 10950
    },
    {
      "dynamics_loss_avg": 0.05795227736234665,
      "epoch": 1.0442733764648437e-05,
      "step": 10950
    },
    {
      "action_loss_avg": 0.00440953797660768,
      "epoch": 1.0442733764648437e-05,
      "step": 10950
    },
    {
      "epoch": 1.0442733764648437e-05,
      "step": 10950,
      "training_step_time": 8.618723630905151
    },
    {
      "epoch": 1.0443687438964844e-05,
      "model_forward_time": 1.705329418182373,
      "step": 10951
    },
    {
      "epoch": 1.0443687438964844e-05,
      "step": 10951,
      "training_step_time": 7.66039776802063
    },
    {
      "epoch": 1.044464111328125e-05,
      "model_forward_time": 1.7111706733703613,
      "step": 10952
    },
    {
      "epoch": 1.044464111328125e-05,
      "step": 10952,
      "training_step_time": 7.446036100387573
    },
    {
      "epoch": 1.0445594787597656e-05,
      "model_forward_time": 1.702341079711914,
      "step": 10953
    },
    {
      "epoch": 1.0445594787597656e-05,
      "step": 10953,
      "training_step_time": 8.40194034576416
    },
    {
      "epoch": 1.0446548461914063e-05,
      "model_forward_time": 1.7082700729370117,
      "step": 10954
    },
    {
      "epoch": 1.0446548461914063e-05,
      "step": 10954,
      "training_step_time": 8.848438262939453
    },
    {
      "epoch": 1.0447502136230469e-05,
      "model_forward_time": 1.7097969055175781,
      "step": 10955
    },
    {
      "epoch": 1.0447502136230469e-05,
      "step": 10955,
      "training_step_time": 8.819637537002563
    },
    {
      "epoch": 1.0448455810546875e-05,
      "model_forward_time": 1.7062830924987793,
      "step": 10956
    },
    {
      "epoch": 1.0448455810546875e-05,
      "step": 10956,
      "training_step_time": 7.750616073608398
    },
    {
      "epoch": 1.0449409484863282e-05,
      "model_forward_time": 2.2403008937835693,
      "step": 10957
    },
    {
      "epoch": 1.0449409484863282e-05,
      "step": 10957,
      "training_step_time": 8.020841360092163
    },
    {
      "epoch": 1.0450363159179688e-05,
      "model_forward_time": 1.7018699645996094,
      "step": 10958
    },
    {
      "epoch": 1.0450363159179688e-05,
      "step": 10958,
      "training_step_time": 7.755333185195923
    },
    {
      "epoch": 1.0451316833496093e-05,
      "model_forward_time": 1.7604947090148926,
      "step": 10959
    },
    {
      "epoch": 1.0451316833496093e-05,
      "step": 10959,
      "training_step_time": 7.77611780166626
    },
    {
      "epoch": 1.04522705078125e-05,
      "grad_norm": 0.23889373242855072,
      "learning_rate": 9.90323162427134e-06,
      "loss": 0.0666,
      "step": 10960
    },
    {
      "epoch": 1.04522705078125e-05,
      "model_forward_time": 1.7046403884887695,
      "step": 10960
    },
    {
      "dynamics_loss_avg": 0.05462122354656458,
      "epoch": 1.04522705078125e-05,
      "step": 10960
    },
    {
      "action_loss_avg": 0.0035359660512767734,
      "epoch": 1.04522705078125e-05,
      "step": 10960
    },
    {
      "epoch": 1.04522705078125e-05,
      "step": 10960,
      "training_step_time": 7.851471185684204
    },
    {
      "epoch": 1.0453224182128907e-05,
      "model_forward_time": 1.7065236568450928,
      "step": 10961
    },
    {
      "epoch": 1.0453224182128907e-05,
      "step": 10961,
      "training_step_time": 7.64668083190918
    },
    {
      "epoch": 1.0454177856445312e-05,
      "model_forward_time": 1.7051188945770264,
      "step": 10962
    },
    {
      "epoch": 1.0454177856445312e-05,
      "step": 10962,
      "training_step_time": 7.722634553909302
    },
    {
      "epoch": 1.0455131530761718e-05,
      "model_forward_time": 1.712141990661621,
      "step": 10963
    },
    {
      "epoch": 1.0455131530761718e-05,
      "step": 10963,
      "training_step_time": 9.031392097473145
    },
    {
      "epoch": 1.0456085205078125e-05,
      "model_forward_time": 1.326883316040039,
      "step": 10964
    },
    {
      "epoch": 1.0456085205078125e-05,
      "step": 10964,
      "training_step_time": 7.84057354927063
    },
    {
      "epoch": 1.0457038879394531e-05,
      "model_forward_time": 1.7097444534301758,
      "step": 10965
    },
    {
      "epoch": 1.0457038879394531e-05,
      "step": 10965,
      "training_step_time": 8.547403573989868
    },
    {
      "epoch": 1.0457992553710937e-05,
      "model_forward_time": 2.2430365085601807,
      "step": 10966
    },
    {
      "epoch": 1.0457992553710937e-05,
      "step": 10966,
      "training_step_time": 7.970911264419556
    },
    {
      "epoch": 1.0458946228027344e-05,
      "model_forward_time": 1.7044994831085205,
      "step": 10967
    },
    {
      "epoch": 1.0458946228027344e-05,
      "step": 10967,
      "training_step_time": 8.765437602996826
    },
    {
      "epoch": 1.045989990234375e-05,
      "model_forward_time": 1.7348449230194092,
      "step": 10968
    },
    {
      "epoch": 1.045989990234375e-05,
      "step": 10968,
      "training_step_time": 8.77647066116333
    },
    {
      "epoch": 1.0460853576660156e-05,
      "model_forward_time": 1.7076208591461182,
      "step": 10969
    },
    {
      "epoch": 1.0460853576660156e-05,
      "step": 10969,
      "training_step_time": 7.9531261920928955
    },
    {
      "epoch": 1.0461807250976563e-05,
      "grad_norm": 0.27354487776756287,
      "learning_rate": 9.902907627279724e-06,
      "loss": 0.0681,
      "step": 10970
    },
    {
      "epoch": 1.0461807250976563e-05,
      "model_forward_time": 1.712209939956665,
      "step": 10970
    },
    {
      "dynamics_loss_avg": 0.05855498965829611,
      "epoch": 1.0461807250976563e-05,
      "step": 10970
    },
    {
      "action_loss_avg": 0.005591297463979572,
      "epoch": 1.0461807250976563e-05,
      "step": 10970
    },
    {
      "epoch": 1.0461807250976563e-05,
      "step": 10970,
      "training_step_time": 7.666869640350342
    },
    {
      "epoch": 1.0462760925292969e-05,
      "model_forward_time": 1.7046620845794678,
      "step": 10971
    },
    {
      "epoch": 1.0462760925292969e-05,
      "step": 10971,
      "training_step_time": 7.733684539794922
    },
    {
      "epoch": 1.0463714599609375e-05,
      "model_forward_time": 1.7066223621368408,
      "step": 10972
    },
    {
      "epoch": 1.0463714599609375e-05,
      "step": 10972,
      "training_step_time": 9.200176000595093
    },
    {
      "epoch": 1.0464668273925782e-05,
      "model_forward_time": 1.700082540512085,
      "step": 10973
    },
    {
      "epoch": 1.0464668273925782e-05,
      "step": 10973,
      "training_step_time": 7.7986485958099365
    },
    {
      "epoch": 1.0465621948242188e-05,
      "model_forward_time": 1.7068369388580322,
      "step": 10974
    },
    {
      "epoch": 1.0465621948242188e-05,
      "step": 10974,
      "training_step_time": 8.796298265457153
    },
    {
      "epoch": 1.0466575622558593e-05,
      "model_forward_time": 2.2568111419677734,
      "step": 10975
    },
    {
      "epoch": 1.0466575622558593e-05,
      "step": 10975,
      "training_step_time": 8.489887475967407
    },
    {
      "epoch": 1.0467529296875e-05,
      "model_forward_time": 1.3328173160552979,
      "step": 10976
    },
    {
      "epoch": 1.0467529296875e-05,
      "step": 10976,
      "training_step_time": 8.265132427215576
    },
    {
      "epoch": 1.0468482971191407e-05,
      "model_forward_time": 1.7080507278442383,
      "step": 10977
    },
    {
      "epoch": 1.0468482971191407e-05,
      "step": 10977,
      "training_step_time": 7.651586532592773
    },
    {
      "epoch": 1.0469436645507812e-05,
      "model_forward_time": 1.7081608772277832,
      "step": 10978
    },
    {
      "epoch": 1.0469436645507812e-05,
      "step": 10978,
      "training_step_time": 7.70960545539856
    },
    {
      "epoch": 1.0470390319824218e-05,
      "model_forward_time": 1.708754062652588,
      "step": 10979
    },
    {
      "epoch": 1.0470390319824218e-05,
      "step": 10979,
      "training_step_time": 9.479026079177856
    },
    {
      "epoch": 1.0471343994140625e-05,
      "grad_norm": 0.25350484251976013,
      "learning_rate": 9.902583094113505e-06,
      "loss": 0.0709,
      "step": 10980
    },
    {
      "epoch": 1.0471343994140625e-05,
      "model_forward_time": 1.7021586894989014,
      "step": 10980
    },
    {
      "dynamics_loss_avg": 0.0641117237508297,
      "epoch": 1.0471343994140625e-05,
      "step": 10980
    },
    {
      "action_loss_avg": 0.0025576630898285655,
      "epoch": 1.0471343994140625e-05,
      "step": 10980
    },
    {
      "epoch": 1.0471343994140625e-05,
      "step": 10980,
      "training_step_time": 9.110176086425781
    },
    {
      "epoch": 1.0472297668457031e-05,
      "model_forward_time": 1.7095763683319092,
      "step": 10981
    },
    {
      "epoch": 1.0472297668457031e-05,
      "step": 10981,
      "training_step_time": 6.875173807144165
    },
    {
      "epoch": 1.0473251342773437e-05,
      "model_forward_time": 1.7032158374786377,
      "step": 10982
    },
    {
      "epoch": 1.0473251342773437e-05,
      "step": 10982,
      "training_step_time": 6.823528289794922
    },
    {
      "epoch": 1.0474205017089844e-05,
      "model_forward_time": 1.7075841426849365,
      "step": 10983
    },
    {
      "epoch": 1.0474205017089844e-05,
      "step": 10983,
      "training_step_time": 6.823440313339233
    },
    {
      "epoch": 1.047515869140625e-05,
      "model_forward_time": 1.705843210220337,
      "step": 10984
    },
    {
      "epoch": 1.047515869140625e-05,
      "step": 10984,
      "training_step_time": 6.814202070236206
    },
    {
      "epoch": 1.0476112365722656e-05,
      "model_forward_time": 1.7145941257476807,
      "step": 10985
    },
    {
      "epoch": 1.0476112365722656e-05,
      "step": 10985,
      "training_step_time": 6.823084592819214
    },
    {
      "epoch": 1.0477066040039063e-05,
      "model_forward_time": 1.7032196521759033,
      "step": 10986
    },
    {
      "epoch": 1.0477066040039063e-05,
      "step": 10986,
      "training_step_time": 6.8276848793029785
    },
    {
      "epoch": 1.0478019714355469e-05,
      "model_forward_time": 1.709388256072998,
      "step": 10987
    },
    {
      "epoch": 1.0478019714355469e-05,
      "step": 10987,
      "training_step_time": 6.834611177444458
    },
    {
      "epoch": 1.0478973388671875e-05,
      "model_forward_time": 1.707416296005249,
      "step": 10988
    },
    {
      "epoch": 1.0478973388671875e-05,
      "step": 10988,
      "training_step_time": 6.802311897277832
    },
    {
      "epoch": 1.0479927062988282e-05,
      "model_forward_time": 2.2586891651153564,
      "step": 10989
    },
    {
      "epoch": 1.0479927062988282e-05,
      "step": 10989,
      "training_step_time": 7.439797878265381
    },
    {
      "epoch": 1.0480880737304688e-05,
      "grad_norm": 0.25768551230430603,
      "learning_rate": 9.902258024808167e-06,
      "loss": 0.0703,
      "step": 10990
    },
    {
      "epoch": 1.0480880737304688e-05,
      "model_forward_time": 1.7105469703674316,
      "step": 10990
    },
    {
      "dynamics_loss_avg": 0.055633393302559855,
      "epoch": 1.0480880737304688e-05,
      "step": 10990
    },
    {
      "action_loss_avg": 0.003274485270958394,
      "epoch": 1.0480880737304688e-05,
      "step": 10990
    },
    {
      "epoch": 1.0480880737304688e-05,
      "step": 10990,
      "training_step_time": 7.968748569488525
    },
    {
      "epoch": 1.0481834411621093e-05,
      "model_forward_time": 1.7087066173553467,
      "step": 10991
    },
    {
      "epoch": 1.0481834411621093e-05,
      "step": 10991,
      "training_step_time": 6.813180446624756
    },
    {
      "epoch": 1.04827880859375e-05,
      "model_forward_time": 1.7083117961883545,
      "step": 10992
    },
    {
      "epoch": 1.04827880859375e-05,
      "step": 10992,
      "training_step_time": 6.804512977600098
    },
    {
      "epoch": 1.0483741760253907e-05,
      "model_forward_time": 1.7102272510528564,
      "step": 10993
    },
    {
      "epoch": 1.0483741760253907e-05,
      "step": 10993,
      "training_step_time": 6.846476078033447
    },
    {
      "epoch": 1.0484695434570312e-05,
      "model_forward_time": 1.7032549381256104,
      "step": 10994
    },
    {
      "epoch": 1.0484695434570312e-05,
      "step": 10994,
      "training_step_time": 6.812939643859863
    },
    {
      "epoch": 1.0485649108886718e-05,
      "model_forward_time": 1.7037677764892578,
      "step": 10995
    },
    {
      "epoch": 1.0485649108886718e-05,
      "step": 10995,
      "training_step_time": 6.78011679649353
    },
    {
      "epoch": 1.0486602783203125e-05,
      "model_forward_time": 1.7140462398529053,
      "step": 10996
    },
    {
      "epoch": 1.0486602783203125e-05,
      "step": 10996,
      "training_step_time": 6.812229156494141
    },
    {
      "epoch": 1.0487556457519531e-05,
      "model_forward_time": 1.7055602073669434,
      "step": 10997
    },
    {
      "epoch": 1.0487556457519531e-05,
      "step": 10997,
      "training_step_time": 6.815901279449463
    },
    {
      "epoch": 1.0488510131835937e-05,
      "model_forward_time": 2.3172996044158936,
      "step": 10998
    },
    {
      "epoch": 1.0488510131835937e-05,
      "step": 10998,
      "training_step_time": 7.772335052490234
    },
    {
      "epoch": 1.0489463806152344e-05,
      "model_forward_time": 1.7434544563293457,
      "step": 10999
    },
    {
      "epoch": 1.0489463806152344e-05,
      "step": 10999,
      "training_step_time": 7.870347023010254
    },
    {
      "epoch": 1.049041748046875e-05,
      "grad_norm": 0.28757092356681824,
      "learning_rate": 9.901932419399263e-06,
      "loss": 0.0689,
      "step": 11000
    },
    {
      "epoch": 1.049041748046875e-05,
      "model_forward_time": 1.7118585109710693,
      "step": 11000
    },
    {
      "dynamics_loss_avg": 0.05801947005093098,
      "epoch": 1.049041748046875e-05,
      "step": 11000
    },
    {
      "action_loss_avg": 0.003344252018723637,
      "epoch": 1.049041748046875e-05,
      "step": 11000
    },
    {
      "epoch": 1.049041748046875e-05,
      "step": 11000,
      "training_step_time": 6.939472913742065
    },
    {
      "epoch": 1.0491371154785156e-05,
      "model_forward_time": 1.7060141563415527,
      "step": 11001
    },
    {
      "epoch": 1.0491371154785156e-05,
      "step": 11001,
      "training_step_time": 6.808154106140137
    },
    {
      "epoch": 1.0492324829101563e-05,
      "model_forward_time": 1.7123608589172363,
      "step": 11002
    },
    {
      "epoch": 1.0492324829101563e-05,
      "step": 11002,
      "training_step_time": 6.786117315292358
    },
    {
      "epoch": 1.0493278503417969e-05,
      "model_forward_time": 1.7096045017242432,
      "step": 11003
    },
    {
      "epoch": 1.0493278503417969e-05,
      "step": 11003,
      "training_step_time": 6.813456773757935
    },
    {
      "epoch": 1.0494232177734375e-05,
      "model_forward_time": 1.7242708206176758,
      "step": 11004
    },
    {
      "epoch": 1.0494232177734375e-05,
      "step": 11004,
      "training_step_time": 6.7920098304748535
    },
    {
      "epoch": 1.0495185852050782e-05,
      "model_forward_time": 1.7163732051849365,
      "step": 11005
    },
    {
      "epoch": 1.0495185852050782e-05,
      "step": 11005,
      "training_step_time": 6.767018556594849
    },
    {
      "epoch": 1.0496139526367188e-05,
      "model_forward_time": 2.713772773742676,
      "step": 11006
    },
    {
      "epoch": 1.0496139526367188e-05,
      "step": 11006,
      "training_step_time": 8.33945894241333
    },
    {
      "epoch": 1.0497093200683593e-05,
      "model_forward_time": 1.7151031494140625,
      "step": 11007
    },
    {
      "epoch": 1.0497093200683593e-05,
      "step": 11007,
      "training_step_time": 7.076026201248169
    },
    {
      "epoch": 1.0498046875e-05,
      "model_forward_time": 1.7205445766448975,
      "step": 11008
    },
    {
      "epoch": 1.0498046875e-05,
      "step": 11008,
      "training_step_time": 9.24246859550476
    },
    {
      "epoch": 1.0499000549316407e-05,
      "model_forward_time": 1.7113852500915527,
      "step": 11009
    },
    {
      "epoch": 1.0499000549316407e-05,
      "step": 11009,
      "training_step_time": 7.524474143981934
    },
    {
      "epoch": 1.0499954223632812e-05,
      "grad_norm": 0.22205515205860138,
      "learning_rate": 9.9016062779224e-06,
      "loss": 0.07,
      "step": 11010
    },
    {
      "epoch": 1.0499954223632812e-05,
      "model_forward_time": 1.7108471393585205,
      "step": 11010
    },
    {
      "dynamics_loss_avg": 0.05994782876223326,
      "epoch": 1.0499954223632812e-05,
      "step": 11010
    },
    {
      "action_loss_avg": 0.004347759147640318,
      "epoch": 1.0499954223632812e-05,
      "step": 11010
    },
    {
      "epoch": 1.0499954223632812e-05,
      "step": 11010,
      "training_step_time": 6.789714813232422
    },
    {
      "epoch": 1.0500907897949218e-05,
      "model_forward_time": 1.7235291004180908,
      "step": 11011
    },
    {
      "epoch": 1.0500907897949218e-05,
      "step": 11011,
      "training_step_time": 6.817599773406982
    },
    {
      "epoch": 1.0501861572265625e-05,
      "model_forward_time": 1.6966803073883057,
      "step": 11012
    },
    {
      "epoch": 1.0501861572265625e-05,
      "step": 11012,
      "training_step_time": 6.717448949813843
    },
    {
      "epoch": 1.0502815246582031e-05,
      "model_forward_time": 1.712545394897461,
      "step": 11013
    },
    {
      "epoch": 1.0502815246582031e-05,
      "step": 11013,
      "training_step_time": 6.823317289352417
    },
    {
      "epoch": 1.0503768920898437e-05,
      "model_forward_time": 1.7093894481658936,
      "step": 11014
    },
    {
      "epoch": 1.0503768920898437e-05,
      "step": 11014,
      "training_step_time": 6.782240390777588
    },
    {
      "epoch": 1.0504722595214844e-05,
      "model_forward_time": 1.8833682537078857,
      "step": 11015
    },
    {
      "epoch": 1.0504722595214844e-05,
      "step": 11015,
      "training_step_time": 6.782124757766724
    },
    {
      "epoch": 1.050567626953125e-05,
      "model_forward_time": 1.7392864227294922,
      "step": 11016
    },
    {
      "epoch": 1.050567626953125e-05,
      "step": 11016,
      "training_step_time": 7.520828723907471
    },
    {
      "epoch": 1.0506629943847656e-05,
      "model_forward_time": 1.7111806869506836,
      "step": 11017
    },
    {
      "epoch": 1.0506629943847656e-05,
      "step": 11017,
      "training_step_time": 8.392745018005371
    },
    {
      "epoch": 1.0507583618164063e-05,
      "model_forward_time": 1.7093274593353271,
      "step": 11018
    },
    {
      "epoch": 1.0507583618164063e-05,
      "step": 11018,
      "training_step_time": 7.569567680358887
    },
    {
      "epoch": 1.0508537292480469e-05,
      "model_forward_time": 1.7158546447753906,
      "step": 11019
    },
    {
      "epoch": 1.0508537292480469e-05,
      "step": 11019,
      "training_step_time": 6.776170492172241
    },
    {
      "epoch": 1.0509490966796875e-05,
      "grad_norm": 0.2540247142314911,
      "learning_rate": 9.901279600413242e-06,
      "loss": 0.0669,
      "step": 11020
    },
    {
      "epoch": 1.0509490966796875e-05,
      "model_forward_time": 1.7151105403900146,
      "step": 11020
    },
    {
      "dynamics_loss_avg": 0.07870240435004235,
      "epoch": 1.0509490966796875e-05,
      "step": 11020
    },
    {
      "action_loss_avg": 0.00481695777270943,
      "epoch": 1.0509490966796875e-05,
      "step": 11020
    },
    {
      "epoch": 1.0509490966796875e-05,
      "step": 11020,
      "training_step_time": 6.78566312789917
    },
    {
      "epoch": 1.0510444641113282e-05,
      "model_forward_time": 1.7171080112457275,
      "step": 11021
    },
    {
      "epoch": 1.0510444641113282e-05,
      "step": 11021,
      "training_step_time": 6.7692036628723145
    },
    {
      "epoch": 1.0511398315429688e-05,
      "model_forward_time": 1.7275283336639404,
      "step": 11022
    },
    {
      "epoch": 1.0511398315429688e-05,
      "step": 11022,
      "training_step_time": 6.763510227203369
    },
    {
      "epoch": 1.0512351989746093e-05,
      "model_forward_time": 1.7396173477172852,
      "step": 11023
    },
    {
      "epoch": 1.0512351989746093e-05,
      "step": 11023,
      "training_step_time": 6.7736732959747314
    },
    {
      "epoch": 1.05133056640625e-05,
      "model_forward_time": 1.7378895282745361,
      "step": 11024
    },
    {
      "epoch": 1.05133056640625e-05,
      "step": 11024,
      "training_step_time": 6.759059906005859
    },
    {
      "epoch": 1.0514259338378907e-05,
      "model_forward_time": 2.9111809730529785,
      "step": 11025
    },
    {
      "epoch": 1.0514259338378907e-05,
      "step": 11025,
      "training_step_time": 7.787020921707153
    },
    {
      "epoch": 1.0515213012695312e-05,
      "model_forward_time": 1.758058786392212,
      "step": 11026
    },
    {
      "epoch": 1.0515213012695312e-05,
      "step": 11026,
      "training_step_time": 7.7281904220581055
    },
    {
      "epoch": 1.0516166687011718e-05,
      "model_forward_time": 1.737973928451538,
      "step": 11027
    },
    {
      "epoch": 1.0516166687011718e-05,
      "step": 11027,
      "training_step_time": 7.909331560134888
    },
    {
      "epoch": 1.0517120361328125e-05,
      "model_forward_time": 1.7414228916168213,
      "step": 11028
    },
    {
      "epoch": 1.0517120361328125e-05,
      "step": 11028,
      "training_step_time": 7.416449785232544
    },
    {
      "epoch": 1.0518074035644531e-05,
      "model_forward_time": 1.744307518005371,
      "step": 11029
    },
    {
      "epoch": 1.0518074035644531e-05,
      "step": 11029,
      "training_step_time": 6.814640760421753
    },
    {
      "epoch": 1.0519027709960937e-05,
      "grad_norm": 0.15399333834648132,
      "learning_rate": 9.90095238690752e-06,
      "loss": 0.0649,
      "step": 11030
    },
    {
      "epoch": 1.0519027709960937e-05,
      "model_forward_time": 1.7401833534240723,
      "step": 11030
    },
    {
      "dynamics_loss_avg": 0.05854339636862278,
      "epoch": 1.0519027709960937e-05,
      "step": 11030
    },
    {
      "action_loss_avg": 0.0032609960122499614,
      "epoch": 1.0519027709960937e-05,
      "step": 11030
    },
    {
      "epoch": 1.0519027709960937e-05,
      "step": 11030,
      "training_step_time": 6.8171679973602295
    },
    {
      "epoch": 1.0519981384277344e-05,
      "model_forward_time": 1.7396762371063232,
      "step": 11031
    },
    {
      "epoch": 1.0519981384277344e-05,
      "step": 11031,
      "training_step_time": 6.803387880325317
    },
    {
      "epoch": 1.052093505859375e-05,
      "model_forward_time": 1.7352473735809326,
      "step": 11032
    },
    {
      "epoch": 1.052093505859375e-05,
      "step": 11032,
      "training_step_time": 6.820497512817383
    },
    {
      "epoch": 1.0521888732910156e-05,
      "model_forward_time": 1.7305688858032227,
      "step": 11033
    },
    {
      "epoch": 1.0521888732910156e-05,
      "step": 11033,
      "training_step_time": 6.820597171783447
    },
    {
      "epoch": 1.0522842407226563e-05,
      "model_forward_time": 2.9254984855651855,
      "step": 11034
    },
    {
      "epoch": 1.0522842407226563e-05,
      "step": 11034,
      "training_step_time": 8.31958270072937
    },
    {
      "epoch": 1.0523796081542969e-05,
      "model_forward_time": 1.7536590099334717,
      "step": 11035
    },
    {
      "epoch": 1.0523796081542969e-05,
      "step": 11035,
      "training_step_time": 8.755192041397095
    },
    {
      "epoch": 1.0524749755859375e-05,
      "model_forward_time": 1.7355258464813232,
      "step": 11036
    },
    {
      "epoch": 1.0524749755859375e-05,
      "step": 11036,
      "training_step_time": 7.855620622634888
    },
    {
      "epoch": 1.0525703430175782e-05,
      "model_forward_time": 1.7363648414611816,
      "step": 11037
    },
    {
      "epoch": 1.0525703430175782e-05,
      "step": 11037,
      "training_step_time": 7.526151895523071
    },
    {
      "epoch": 1.0526657104492188e-05,
      "model_forward_time": 1.7421910762786865,
      "step": 11038
    },
    {
      "epoch": 1.0526657104492188e-05,
      "step": 11038,
      "training_step_time": 6.796708822250366
    },
    {
      "epoch": 1.0527610778808593e-05,
      "model_forward_time": 1.7369136810302734,
      "step": 11039
    },
    {
      "epoch": 1.0527610778808593e-05,
      "step": 11039,
      "training_step_time": 6.806784391403198
    },
    {
      "epoch": 1.0528564453125e-05,
      "grad_norm": 0.21928884088993073,
      "learning_rate": 9.90062463744101e-06,
      "loss": 0.0683,
      "step": 11040
    },
    {
      "epoch": 1.0528564453125e-05,
      "model_forward_time": 1.733565330505371,
      "step": 11040
    },
    {
      "dynamics_loss_avg": 0.06549575794488191,
      "epoch": 1.0528564453125e-05,
      "step": 11040
    },
    {
      "action_loss_avg": 0.006241940706968307,
      "epoch": 1.0528564453125e-05,
      "step": 11040
    },
    {
      "epoch": 1.0528564453125e-05,
      "step": 11040,
      "training_step_time": 6.790830612182617
    },
    {
      "epoch": 1.0529518127441407e-05,
      "model_forward_time": 1.7398130893707275,
      "step": 11041
    },
    {
      "epoch": 1.0529518127441407e-05,
      "step": 11041,
      "training_step_time": 6.7547767162323
    },
    {
      "epoch": 1.0530471801757812e-05,
      "model_forward_time": 1.7323038578033447,
      "step": 11042
    },
    {
      "epoch": 1.0530471801757812e-05,
      "step": 11042,
      "training_step_time": 6.784852743148804
    },
    {
      "epoch": 1.0531425476074218e-05,
      "model_forward_time": 2.9302613735198975,
      "step": 11043
    },
    {
      "epoch": 1.0531425476074218e-05,
      "step": 11043,
      "training_step_time": 7.732391357421875
    },
    {
      "epoch": 1.0532379150390625e-05,
      "model_forward_time": 1.7495057582855225,
      "step": 11044
    },
    {
      "epoch": 1.0532379150390625e-05,
      "step": 11044,
      "training_step_time": 7.661458492279053
    },
    {
      "epoch": 1.0533332824707031e-05,
      "model_forward_time": 1.7421257495880127,
      "step": 11045
    },
    {
      "epoch": 1.0533332824707031e-05,
      "step": 11045,
      "training_step_time": 8.641630411148071
    },
    {
      "epoch": 1.0534286499023437e-05,
      "model_forward_time": 1.737072467803955,
      "step": 11046
    },
    {
      "epoch": 1.0534286499023437e-05,
      "step": 11046,
      "training_step_time": 7.637026786804199
    },
    {
      "epoch": 1.0535240173339844e-05,
      "model_forward_time": 1.733915090560913,
      "step": 11047
    },
    {
      "epoch": 1.0535240173339844e-05,
      "step": 11047,
      "training_step_time": 6.787031173706055
    },
    {
      "epoch": 1.053619384765625e-05,
      "model_forward_time": 1.733548641204834,
      "step": 11048
    },
    {
      "epoch": 1.053619384765625e-05,
      "step": 11048,
      "training_step_time": 6.754526138305664
    },
    {
      "epoch": 1.0537147521972656e-05,
      "model_forward_time": 1.7388389110565186,
      "step": 11049
    },
    {
      "epoch": 1.0537147521972656e-05,
      "step": 11049,
      "training_step_time": 6.724754095077515
    },
    {
      "epoch": 1.0538101196289063e-05,
      "grad_norm": 0.3784559369087219,
      "learning_rate": 9.900296352049558e-06,
      "loss": 0.0698,
      "step": 11050
    },
    {
      "epoch": 1.0538101196289063e-05,
      "model_forward_time": 1.7346415519714355,
      "step": 11050
    },
    {
      "dynamics_loss_avg": 0.050596689246594904,
      "epoch": 1.0538101196289063e-05,
      "step": 11050
    },
    {
      "action_loss_avg": 0.00307698356336914,
      "epoch": 1.0538101196289063e-05,
      "step": 11050
    },
    {
      "epoch": 1.0538101196289063e-05,
      "step": 11050,
      "training_step_time": 6.773843050003052
    },
    {
      "epoch": 1.0539054870605469e-05,
      "model_forward_time": 1.7346603870391846,
      "step": 11051
    },
    {
      "epoch": 1.0539054870605469e-05,
      "step": 11051,
      "training_step_time": 6.756741523742676
    },
    {
      "epoch": 1.0540008544921875e-05,
      "model_forward_time": 2.9088776111602783,
      "step": 11052
    },
    {
      "epoch": 1.0540008544921875e-05,
      "step": 11052,
      "training_step_time": 7.724479913711548
    },
    {
      "epoch": 1.0540962219238282e-05,
      "model_forward_time": 1.7456245422363281,
      "step": 11053
    },
    {
      "epoch": 1.0540962219238282e-05,
      "step": 11053,
      "training_step_time": 7.613543748855591
    },
    {
      "epoch": 1.0541915893554688e-05,
      "model_forward_time": 1.7309768199920654,
      "step": 11054
    },
    {
      "epoch": 1.0541915893554688e-05,
      "step": 11054,
      "training_step_time": 8.591145515441895
    },
    {
      "epoch": 1.0542869567871093e-05,
      "model_forward_time": 1.0244181156158447,
      "step": 11055
    },
    {
      "epoch": 1.0542869567871093e-05,
      "step": 11055,
      "training_step_time": 7.85065221786499
    },
    {
      "epoch": 1.05438232421875e-05,
      "model_forward_time": 1.735138177871704,
      "step": 11056
    },
    {
      "epoch": 1.05438232421875e-05,
      "step": 11056,
      "training_step_time": 6.773886680603027
    },
    {
      "epoch": 1.0544776916503907e-05,
      "model_forward_time": 1.7337000370025635,
      "step": 11057
    },
    {
      "epoch": 1.0544776916503907e-05,
      "step": 11057,
      "training_step_time": 6.747854232788086
    },
    {
      "epoch": 1.0545730590820312e-05,
      "model_forward_time": 1.729583978652954,
      "step": 11058
    },
    {
      "epoch": 1.0545730590820312e-05,
      "step": 11058,
      "training_step_time": 6.7962610721588135
    },
    {
      "epoch": 1.0546684265136718e-05,
      "model_forward_time": 1.3625879287719727,
      "step": 11059
    },
    {
      "epoch": 1.0546684265136718e-05,
      "step": 11059,
      "training_step_time": 6.816732168197632
    },
    {
      "epoch": 1.0547637939453125e-05,
      "grad_norm": 0.29940617084503174,
      "learning_rate": 9.899967530769064e-06,
      "loss": 0.0692,
      "step": 11060
    },
    {
      "epoch": 1.0547637939453125e-05,
      "model_forward_time": 1.7391808032989502,
      "step": 11060
    },
    {
      "dynamics_loss_avg": 0.058738145418465136,
      "epoch": 1.0547637939453125e-05,
      "step": 11060
    },
    {
      "action_loss_avg": 0.0038795334228780122,
      "epoch": 1.0547637939453125e-05,
      "step": 11060
    },
    {
      "epoch": 1.0547637939453125e-05,
      "step": 11060,
      "training_step_time": 6.800013065338135
    },
    {
      "epoch": 1.0548591613769531e-05,
      "model_forward_time": 1.7338731288909912,
      "step": 11061
    },
    {
      "epoch": 1.0548591613769531e-05,
      "step": 11061,
      "training_step_time": 6.818722486495972
    },
    {
      "epoch": 1.0549545288085937e-05,
      "model_forward_time": 2.941701889038086,
      "step": 11062
    },
    {
      "epoch": 1.0549545288085937e-05,
      "step": 11062,
      "training_step_time": 7.805454969406128
    },
    {
      "epoch": 1.0550498962402344e-05,
      "model_forward_time": 1.736844539642334,
      "step": 11063
    },
    {
      "epoch": 1.0550498962402344e-05,
      "step": 11063,
      "training_step_time": 11.474274396896362
    },
    {
      "epoch": 1.055145263671875e-05,
      "model_forward_time": 1.7395634651184082,
      "step": 11064
    },
    {
      "epoch": 1.055145263671875e-05,
      "step": 11064,
      "training_step_time": 7.65076470375061
    },
    {
      "epoch": 1.0552406311035156e-05,
      "model_forward_time": 1.3616139888763428,
      "step": 11065
    },
    {
      "epoch": 1.0552406311035156e-05,
      "step": 11065,
      "training_step_time": 7.4546425342559814
    },
    {
      "epoch": 1.0553359985351563e-05,
      "model_forward_time": 1.7338721752166748,
      "step": 11066
    },
    {
      "epoch": 1.0553359985351563e-05,
      "step": 11066,
      "training_step_time": 6.800206184387207
    },
    {
      "epoch": 1.0554313659667969e-05,
      "model_forward_time": 1.7397818565368652,
      "step": 11067
    },
    {
      "epoch": 1.0554313659667969e-05,
      "step": 11067,
      "training_step_time": 6.782627582550049
    },
    {
      "epoch": 1.0555267333984375e-05,
      "model_forward_time": 1.7361605167388916,
      "step": 11068
    },
    {
      "epoch": 1.0555267333984375e-05,
      "step": 11068,
      "training_step_time": 6.749619245529175
    },
    {
      "epoch": 1.0556221008300782e-05,
      "model_forward_time": 1.7244925498962402,
      "step": 11069
    },
    {
      "epoch": 1.0556221008300782e-05,
      "step": 11069,
      "training_step_time": 6.793900728225708
    },
    {
      "epoch": 1.0557174682617188e-05,
      "grad_norm": 0.2484826296567917,
      "learning_rate": 9.899638173635488e-06,
      "loss": 0.0744,
      "step": 11070
    },
    {
      "epoch": 1.0557174682617188e-05,
      "model_forward_time": 1.7348883152008057,
      "step": 11070
    },
    {
      "dynamics_loss_avg": 0.07385894376784563,
      "epoch": 1.0557174682617188e-05,
      "step": 11070
    },
    {
      "action_loss_avg": 0.004989535408094525,
      "epoch": 1.0557174682617188e-05,
      "step": 11070
    },
    {
      "epoch": 1.0557174682617188e-05,
      "step": 11070,
      "training_step_time": 6.80026650428772
    },
    {
      "epoch": 1.0558128356933593e-05,
      "model_forward_time": 2.848076581954956,
      "step": 11071
    },
    {
      "epoch": 1.0558128356933593e-05,
      "step": 11071,
      "training_step_time": 8.194950819015503
    },
    {
      "epoch": 1.055908203125e-05,
      "model_forward_time": 1.737905502319336,
      "step": 11072
    },
    {
      "epoch": 1.055908203125e-05,
      "step": 11072,
      "training_step_time": 9.693296670913696
    },
    {
      "epoch": 1.0560035705566407e-05,
      "model_forward_time": 1.733515977859497,
      "step": 11073
    },
    {
      "epoch": 1.0560035705566407e-05,
      "step": 11073,
      "training_step_time": 8.67697548866272
    },
    {
      "epoch": 1.0560989379882812e-05,
      "model_forward_time": 1.7338173389434814,
      "step": 11074
    },
    {
      "epoch": 1.0560989379882812e-05,
      "step": 11074,
      "training_step_time": 7.579925298690796
    },
    {
      "epoch": 1.0561943054199218e-05,
      "model_forward_time": 1.7321443557739258,
      "step": 11075
    },
    {
      "epoch": 1.0561943054199218e-05,
      "step": 11075,
      "training_step_time": 6.7768638134002686
    },
    {
      "epoch": 1.0562896728515625e-05,
      "model_forward_time": 1.7320854663848877,
      "step": 11076
    },
    {
      "epoch": 1.0562896728515625e-05,
      "step": 11076,
      "training_step_time": 6.787862539291382
    },
    {
      "epoch": 1.0563850402832031e-05,
      "model_forward_time": 1.7355034351348877,
      "step": 11077
    },
    {
      "epoch": 1.0563850402832031e-05,
      "step": 11077,
      "training_step_time": 6.820365905761719
    },
    {
      "epoch": 1.0564804077148437e-05,
      "model_forward_time": 1.7351107597351074,
      "step": 11078
    },
    {
      "epoch": 1.0564804077148437e-05,
      "step": 11078,
      "training_step_time": 6.841609716415405
    },
    {
      "epoch": 1.0565757751464844e-05,
      "model_forward_time": 1.7336921691894531,
      "step": 11079
    },
    {
      "epoch": 1.0565757751464844e-05,
      "step": 11079,
      "training_step_time": 6.769016742706299
    },
    {
      "epoch": 1.056671142578125e-05,
      "grad_norm": 0.31965556740760803,
      "learning_rate": 9.899308280684849e-06,
      "loss": 0.0737,
      "step": 11080
    },
    {
      "epoch": 1.056671142578125e-05,
      "model_forward_time": 2.4706294536590576,
      "step": 11080
    },
    {
      "dynamics_loss_avg": 0.061291516199707984,
      "epoch": 1.056671142578125e-05,
      "step": 11080
    },
    {
      "action_loss_avg": 0.0035670068580657245,
      "epoch": 1.056671142578125e-05,
      "step": 11080
    },
    {
      "epoch": 1.056671142578125e-05,
      "step": 11080,
      "training_step_time": 7.523403167724609
    },
    {
      "epoch": 1.0567665100097656e-05,
      "model_forward_time": 1.7569396495819092,
      "step": 11081
    },
    {
      "epoch": 1.0567665100097656e-05,
      "step": 11081,
      "training_step_time": 9.93662977218628
    },
    {
      "epoch": 1.0568618774414063e-05,
      "model_forward_time": 1.7373878955841064,
      "step": 11082
    },
    {
      "epoch": 1.0568618774414063e-05,
      "step": 11082,
      "training_step_time": 8.424999475479126
    },
    {
      "epoch": 1.0569572448730469e-05,
      "model_forward_time": 1.7356479167938232,
      "step": 11083
    },
    {
      "epoch": 1.0569572448730469e-05,
      "step": 11083,
      "training_step_time": 7.482360601425171
    },
    {
      "epoch": 1.0570526123046875e-05,
      "model_forward_time": 1.730262041091919,
      "step": 11084
    },
    {
      "epoch": 1.0570526123046875e-05,
      "step": 11084,
      "training_step_time": 6.716581583023071
    },
    {
      "epoch": 1.0571479797363282e-05,
      "model_forward_time": 1.7352814674377441,
      "step": 11085
    },
    {
      "epoch": 1.0571479797363282e-05,
      "step": 11085,
      "training_step_time": 6.745784759521484
    },
    {
      "epoch": 1.0572433471679688e-05,
      "model_forward_time": 1.3635826110839844,
      "step": 11086
    },
    {
      "epoch": 1.0572433471679688e-05,
      "step": 11086,
      "training_step_time": 6.735594034194946
    },
    {
      "epoch": 1.0573387145996094e-05,
      "model_forward_time": 1.7345490455627441,
      "step": 11087
    },
    {
      "epoch": 1.0573387145996094e-05,
      "step": 11087,
      "training_step_time": 6.666661024093628
    },
    {
      "epoch": 1.05743408203125e-05,
      "model_forward_time": 1.7343621253967285,
      "step": 11088
    },
    {
      "epoch": 1.05743408203125e-05,
      "step": 11088,
      "training_step_time": 6.703164100646973
    },
    {
      "epoch": 1.0575294494628907e-05,
      "model_forward_time": 1.7313594818115234,
      "step": 11089
    },
    {
      "epoch": 1.0575294494628907e-05,
      "step": 11089,
      "training_step_time": 7.2578558921813965
    },
    {
      "epoch": 1.0576248168945312e-05,
      "grad_norm": 0.36133283376693726,
      "learning_rate": 9.898977851953223e-06,
      "loss": 0.0705,
      "step": 11090
    },
    {
      "epoch": 1.0576248168945312e-05,
      "model_forward_time": 2.8431520462036133,
      "step": 11090
    },
    {
      "dynamics_loss_avg": 0.057585947774350645,
      "epoch": 1.0576248168945312e-05,
      "step": 11090
    },
    {
      "action_loss_avg": 0.004738007043488324,
      "epoch": 1.0576248168945312e-05,
      "step": 11090
    },
    {
      "epoch": 1.0576248168945312e-05,
      "step": 11090,
      "training_step_time": 8.817427396774292
    },
    {
      "epoch": 1.0577201843261718e-05,
      "model_forward_time": 1.7361891269683838,
      "step": 11091
    },
    {
      "epoch": 1.0577201843261718e-05,
      "step": 11091,
      "training_step_time": 9.526771545410156
    },
    {
      "epoch": 1.0578155517578126e-05,
      "model_forward_time": 1.7320754528045654,
      "step": 11092
    },
    {
      "epoch": 1.0578155517578126e-05,
      "step": 11092,
      "training_step_time": 7.537776708602905
    },
    {
      "epoch": 1.0579109191894531e-05,
      "model_forward_time": 1.732909917831421,
      "step": 11093
    },
    {
      "epoch": 1.0579109191894531e-05,
      "step": 11093,
      "training_step_time": 6.962905645370483
    },
    {
      "epoch": 1.0580062866210937e-05,
      "model_forward_time": 1.7253878116607666,
      "step": 11094
    },
    {
      "epoch": 1.0580062866210937e-05,
      "step": 11094,
      "training_step_time": 6.702701568603516
    },
    {
      "epoch": 1.0581016540527344e-05,
      "model_forward_time": 1.7383944988250732,
      "step": 11095
    },
    {
      "epoch": 1.0581016540527344e-05,
      "step": 11095,
      "training_step_time": 6.722699403762817
    },
    {
      "epoch": 1.058197021484375e-05,
      "model_forward_time": 1.7384352684020996,
      "step": 11096
    },
    {
      "epoch": 1.058197021484375e-05,
      "step": 11096,
      "training_step_time": 6.718084812164307
    },
    {
      "epoch": 1.0582923889160156e-05,
      "model_forward_time": 1.7396726608276367,
      "step": 11097
    },
    {
      "epoch": 1.0582923889160156e-05,
      "step": 11097,
      "training_step_time": 6.7134788036346436
    },
    {
      "epoch": 1.0583877563476563e-05,
      "model_forward_time": 1.7364566326141357,
      "step": 11098
    },
    {
      "epoch": 1.0583877563476563e-05,
      "step": 11098,
      "training_step_time": 6.689911365509033
    },
    {
      "epoch": 1.0584831237792969e-05,
      "model_forward_time": 2.8509228229522705,
      "step": 11099
    },
    {
      "epoch": 1.0584831237792969e-05,
      "step": 11099,
      "training_step_time": 9.13526463508606
    },
    {
      "epoch": 1.0585784912109375e-05,
      "grad_norm": 0.29983341693878174,
      "learning_rate": 9.898646887476742e-06,
      "loss": 0.0706,
      "step": 11100
    },
    {
      "epoch": 1.0585784912109375e-05,
      "model_forward_time": 1.0638914108276367,
      "step": 11100
    },
    {
      "dynamics_loss_avg": 0.07598978597670794,
      "epoch": 1.0585784912109375e-05,
      "step": 11100
    },
    {
      "action_loss_avg": 0.004978012660285458,
      "epoch": 1.0585784912109375e-05,
      "step": 11100
    },
    {
      "epoch": 1.0585784912109375e-05,
      "step": 11100,
      "training_step_time": 8.944402933120728
    },
    {
      "epoch": 1.0586738586425782e-05,
      "model_forward_time": 1.759307622909546,
      "step": 11101
    },
    {
      "epoch": 1.0586738586425782e-05,
      "step": 11101,
      "training_step_time": 7.564809560775757
    },
    {
      "epoch": 1.0587692260742188e-05,
      "model_forward_time": 1.7646963596343994,
      "step": 11102
    },
    {
      "epoch": 1.0587692260742188e-05,
      "step": 11102,
      "training_step_time": 7.384628057479858
    },
    {
      "epoch": 1.0588645935058594e-05,
      "model_forward_time": 1.7556087970733643,
      "step": 11103
    },
    {
      "epoch": 1.0588645935058594e-05,
      "step": 11103,
      "training_step_time": 6.722188234329224
    },
    {
      "epoch": 1.0589599609375e-05,
      "model_forward_time": 1.757136344909668,
      "step": 11104
    },
    {
      "epoch": 1.0589599609375e-05,
      "step": 11104,
      "training_step_time": 6.721411943435669
    },
    {
      "epoch": 1.0590553283691407e-05,
      "model_forward_time": 1.7502684593200684,
      "step": 11105
    },
    {
      "epoch": 1.0590553283691407e-05,
      "step": 11105,
      "training_step_time": 6.688519239425659
    },
    {
      "epoch": 1.0591506958007812e-05,
      "model_forward_time": 1.7533411979675293,
      "step": 11106
    },
    {
      "epoch": 1.0591506958007812e-05,
      "step": 11106,
      "training_step_time": 6.713843822479248
    },
    {
      "epoch": 1.0592460632324218e-05,
      "model_forward_time": 1.7516446113586426,
      "step": 11107
    },
    {
      "epoch": 1.0592460632324218e-05,
      "step": 11107,
      "training_step_time": 6.7042810916900635
    },
    {
      "epoch": 1.0593414306640626e-05,
      "model_forward_time": 2.8553192615509033,
      "step": 11108
    },
    {
      "epoch": 1.0593414306640626e-05,
      "step": 11108,
      "training_step_time": 7.848539352416992
    },
    {
      "epoch": 1.0594367980957031e-05,
      "model_forward_time": 1.7661819458007812,
      "step": 11109
    },
    {
      "epoch": 1.0594367980957031e-05,
      "step": 11109,
      "training_step_time": 10.718789100646973
    },
    {
      "epoch": 1.0595321655273437e-05,
      "grad_norm": 0.14800183475017548,
      "learning_rate": 9.898315387291604e-06,
      "loss": 0.0645,
      "step": 11110
    },
    {
      "epoch": 1.0595321655273437e-05,
      "model_forward_time": 1.755692481994629,
      "step": 11110
    },
    {
      "dynamics_loss_avg": 0.05898916386067867,
      "epoch": 1.0595321655273437e-05,
      "step": 11110
    },
    {
      "action_loss_avg": 0.003192449233029038,
      "epoch": 1.0595321655273437e-05,
      "step": 11110
    },
    {
      "epoch": 1.0595321655273437e-05,
      "step": 11110,
      "training_step_time": 10.388952255249023
    },
    {
      "epoch": 1.0596275329589844e-05,
      "model_forward_time": 1.7629735469818115,
      "step": 11111
    },
    {
      "epoch": 1.0596275329589844e-05,
      "step": 11111,
      "training_step_time": 7.572404384613037
    },
    {
      "epoch": 1.059722900390625e-05,
      "model_forward_time": 1.7556095123291016,
      "step": 11112
    },
    {
      "epoch": 1.059722900390625e-05,
      "step": 11112,
      "training_step_time": 6.683335542678833
    },
    {
      "epoch": 1.0598182678222656e-05,
      "model_forward_time": 1.7443268299102783,
      "step": 11113
    },
    {
      "epoch": 1.0598182678222656e-05,
      "step": 11113,
      "training_step_time": 6.756237745285034
    },
    {
      "epoch": 1.0599136352539063e-05,
      "model_forward_time": 1.7500500679016113,
      "step": 11114
    },
    {
      "epoch": 1.0599136352539063e-05,
      "step": 11114,
      "training_step_time": 6.786255598068237
    },
    {
      "epoch": 1.0600090026855469e-05,
      "model_forward_time": 1.0657658576965332,
      "step": 11115
    },
    {
      "epoch": 1.0600090026855469e-05,
      "step": 11115,
      "training_step_time": 6.756334543228149
    },
    {
      "epoch": 1.0601043701171875e-05,
      "model_forward_time": 1.7567856311798096,
      "step": 11116
    },
    {
      "epoch": 1.0601043701171875e-05,
      "step": 11116,
      "training_step_time": 6.787181615829468
    },
    {
      "epoch": 1.0601997375488282e-05,
      "model_forward_time": 1.7589929103851318,
      "step": 11117
    },
    {
      "epoch": 1.0601997375488282e-05,
      "step": 11117,
      "training_step_time": 8.336606502532959
    },
    {
      "epoch": 1.0602951049804688e-05,
      "model_forward_time": 2.904233694076538,
      "step": 11118
    },
    {
      "epoch": 1.0602951049804688e-05,
      "step": 11118,
      "training_step_time": 9.226853370666504
    },
    {
      "epoch": 1.0603904724121094e-05,
      "model_forward_time": 1.752868890762329,
      "step": 11119
    },
    {
      "epoch": 1.0603904724121094e-05,
      "step": 11119,
      "training_step_time": 8.721685647964478
    },
    {
      "epoch": 1.06048583984375e-05,
      "grad_norm": 0.17420707643032074,
      "learning_rate": 9.897983351434059e-06,
      "loss": 0.0692,
      "step": 11120
    },
    {
      "epoch": 1.06048583984375e-05,
      "model_forward_time": 1.7573060989379883,
      "step": 11120
    },
    {
      "dynamics_loss_avg": 0.06627570316195489,
      "epoch": 1.06048583984375e-05,
      "step": 11120
    },
    {
      "action_loss_avg": 0.004294780490454286,
      "epoch": 1.06048583984375e-05,
      "step": 11120
    },
    {
      "epoch": 1.06048583984375e-05,
      "step": 11120,
      "training_step_time": 7.590075254440308
    },
    {
      "epoch": 1.0605812072753907e-05,
      "model_forward_time": 1.7573447227478027,
      "step": 11121
    },
    {
      "epoch": 1.0605812072753907e-05,
      "step": 11121,
      "training_step_time": 6.9156577587127686
    },
    {
      "epoch": 1.0606765747070312e-05,
      "model_forward_time": 1.7538139820098877,
      "step": 11122
    },
    {
      "epoch": 1.0606765747070312e-05,
      "step": 11122,
      "training_step_time": 6.661831855773926
    },
    {
      "epoch": 1.0607719421386718e-05,
      "model_forward_time": 1.7604820728302002,
      "step": 11123
    },
    {
      "epoch": 1.0607719421386718e-05,
      "step": 11123,
      "training_step_time": 6.716641426086426
    },
    {
      "epoch": 1.0608673095703126e-05,
      "model_forward_time": 1.7594268321990967,
      "step": 11124
    },
    {
      "epoch": 1.0608673095703126e-05,
      "step": 11124,
      "training_step_time": 6.701415777206421
    },
    {
      "epoch": 1.0609626770019531e-05,
      "model_forward_time": 1.764486312866211,
      "step": 11125
    },
    {
      "epoch": 1.0609626770019531e-05,
      "step": 11125,
      "training_step_time": 6.641835451126099
    },
    {
      "epoch": 1.0610580444335937e-05,
      "model_forward_time": 1.7546212673187256,
      "step": 11126
    },
    {
      "epoch": 1.0610580444335937e-05,
      "step": 11126,
      "training_step_time": 8.493263244628906
    },
    {
      "epoch": 1.0611534118652344e-05,
      "model_forward_time": 2.955202341079712,
      "step": 11127
    },
    {
      "epoch": 1.0611534118652344e-05,
      "step": 11127,
      "training_step_time": 9.741523504257202
    },
    {
      "epoch": 1.061248779296875e-05,
      "model_forward_time": 1.752826452255249,
      "step": 11128
    },
    {
      "epoch": 1.061248779296875e-05,
      "step": 11128,
      "training_step_time": 9.053292751312256
    },
    {
      "epoch": 1.0613441467285156e-05,
      "model_forward_time": 1.7663002014160156,
      "step": 11129
    },
    {
      "epoch": 1.0613441467285156e-05,
      "step": 11129,
      "training_step_time": 7.5922462940216064
    },
    {
      "epoch": 1.0614395141601563e-05,
      "grad_norm": 0.24693609774112701,
      "learning_rate": 9.89765077994042e-06,
      "loss": 0.0638,
      "step": 11130
    },
    {
      "epoch": 1.0614395141601563e-05,
      "model_forward_time": 1.7585315704345703,
      "step": 11130
    },
    {
      "dynamics_loss_avg": 0.05470382831990719,
      "epoch": 1.0614395141601563e-05,
      "step": 11130
    },
    {
      "action_loss_avg": 0.003934104926884174,
      "epoch": 1.0614395141601563e-05,
      "step": 11130
    },
    {
      "epoch": 1.0614395141601563e-05,
      "step": 11130,
      "training_step_time": 6.981109142303467
    },
    {
      "epoch": 1.0615348815917969e-05,
      "model_forward_time": 1.7527110576629639,
      "step": 11131
    },
    {
      "epoch": 1.0615348815917969e-05,
      "step": 11131,
      "training_step_time": 6.674618244171143
    },
    {
      "epoch": 1.0616302490234375e-05,
      "model_forward_time": 1.7560932636260986,
      "step": 11132
    },
    {
      "epoch": 1.0616302490234375e-05,
      "step": 11132,
      "training_step_time": 6.731953144073486
    },
    {
      "epoch": 1.061725616455078e-05,
      "model_forward_time": 1.7401719093322754,
      "step": 11133
    },
    {
      "epoch": 1.061725616455078e-05,
      "step": 11133,
      "training_step_time": 6.634763956069946
    },
    {
      "epoch": 1.0618209838867188e-05,
      "model_forward_time": 1.7523212432861328,
      "step": 11134
    },
    {
      "epoch": 1.0618209838867188e-05,
      "step": 11134,
      "training_step_time": 6.740992546081543
    },
    {
      "epoch": 1.0619163513183594e-05,
      "model_forward_time": 1.7516326904296875,
      "step": 11135
    },
    {
      "epoch": 1.0619163513183594e-05,
      "step": 11135,
      "training_step_time": 7.4949774742126465
    },
    {
      "epoch": 1.06201171875e-05,
      "model_forward_time": 2.8976027965545654,
      "step": 11136
    },
    {
      "epoch": 1.06201171875e-05,
      "step": 11136,
      "training_step_time": 9.397724866867065
    },
    {
      "epoch": 1.0621070861816407e-05,
      "model_forward_time": 1.7534451484680176,
      "step": 11137
    },
    {
      "epoch": 1.0621070861816407e-05,
      "step": 11137,
      "training_step_time": 8.944394826889038
    },
    {
      "epoch": 1.0622024536132812e-05,
      "model_forward_time": 1.744565725326538,
      "step": 11138
    },
    {
      "epoch": 1.0622024536132812e-05,
      "step": 11138,
      "training_step_time": 8.60869312286377
    },
    {
      "epoch": 1.0622978210449218e-05,
      "model_forward_time": 1.7589082717895508,
      "step": 11139
    },
    {
      "epoch": 1.0622978210449218e-05,
      "step": 11139,
      "training_step_time": 7.433002948760986
    },
    {
      "epoch": 1.0623931884765626e-05,
      "grad_norm": 0.3340379297733307,
      "learning_rate": 9.897317672847055e-06,
      "loss": 0.0714,
      "step": 11140
    },
    {
      "epoch": 1.0623931884765626e-05,
      "model_forward_time": 1.7579143047332764,
      "step": 11140
    },
    {
      "dynamics_loss_avg": 0.06943001393228769,
      "epoch": 1.0623931884765626e-05,
      "step": 11140
    },
    {
      "action_loss_avg": 0.00484192474395968,
      "epoch": 1.0623931884765626e-05,
      "step": 11140
    },
    {
      "epoch": 1.0623931884765626e-05,
      "step": 11140,
      "training_step_time": 6.701641321182251
    },
    {
      "epoch": 1.0624885559082031e-05,
      "model_forward_time": 1.7382476329803467,
      "step": 11141
    },
    {
      "epoch": 1.0624885559082031e-05,
      "step": 11141,
      "training_step_time": 6.6627538204193115
    },
    {
      "epoch": 1.0625839233398437e-05,
      "model_forward_time": 1.7616360187530518,
      "step": 11142
    },
    {
      "epoch": 1.0625839233398437e-05,
      "step": 11142,
      "training_step_time": 6.701960325241089
    },
    {
      "epoch": 1.0626792907714844e-05,
      "model_forward_time": 1.0668351650238037,
      "step": 11143
    },
    {
      "epoch": 1.0626792907714844e-05,
      "step": 11143,
      "training_step_time": 6.713270664215088
    },
    {
      "epoch": 1.062774658203125e-05,
      "model_forward_time": 1.7516446113586426,
      "step": 11144
    },
    {
      "epoch": 1.062774658203125e-05,
      "step": 11144,
      "training_step_time": 7.5373077392578125
    },
    {
      "epoch": 1.0628700256347656e-05,
      "model_forward_time": 1.780714988708496,
      "step": 11145
    },
    {
      "epoch": 1.0628700256347656e-05,
      "step": 11145,
      "training_step_time": 8.513235569000244
    },
    {
      "epoch": 1.0629653930664063e-05,
      "model_forward_time": 2.9570672512054443,
      "step": 11146
    },
    {
      "epoch": 1.0629653930664063e-05,
      "step": 11146,
      "training_step_time": 9.106236219406128
    },
    {
      "epoch": 1.0630607604980469e-05,
      "model_forward_time": 1.735889196395874,
      "step": 11147
    },
    {
      "epoch": 1.0630607604980469e-05,
      "step": 11147,
      "training_step_time": 8.192715167999268
    },
    {
      "epoch": 1.0631561279296875e-05,
      "model_forward_time": 1.7468578815460205,
      "step": 11148
    },
    {
      "epoch": 1.0631561279296875e-05,
      "step": 11148,
      "training_step_time": 7.5617241859436035
    },
    {
      "epoch": 1.063251495361328e-05,
      "model_forward_time": 1.7487773895263672,
      "step": 11149
    },
    {
      "epoch": 1.063251495361328e-05,
      "step": 11149,
      "training_step_time": 6.706001281738281
    },
    {
      "epoch": 1.0633468627929688e-05,
      "grad_norm": 0.24611607193946838,
      "learning_rate": 9.89698403019039e-06,
      "loss": 0.0689,
      "step": 11150
    },
    {
      "epoch": 1.0633468627929688e-05,
      "model_forward_time": 1.0633556842803955,
      "step": 11150
    },
    {
      "dynamics_loss_avg": 0.058374727889895436,
      "epoch": 1.0633468627929688e-05,
      "step": 11150
    },
    {
      "action_loss_avg": 0.004341324605047703,
      "epoch": 1.0633468627929688e-05,
      "step": 11150
    },
    {
      "epoch": 1.0633468627929688e-05,
      "step": 11150,
      "training_step_time": 6.695620536804199
    },
    {
      "epoch": 1.0634422302246094e-05,
      "model_forward_time": 1.7658965587615967,
      "step": 11151
    },
    {
      "epoch": 1.0634422302246094e-05,
      "step": 11151,
      "training_step_time": 6.768282413482666
    },
    {
      "epoch": 1.06353759765625e-05,
      "model_forward_time": 1.7532649040222168,
      "step": 11152
    },
    {
      "epoch": 1.06353759765625e-05,
      "step": 11152,
      "training_step_time": 6.6888206005096436
    },
    {
      "epoch": 1.0636329650878907e-05,
      "model_forward_time": 1.7398815155029297,
      "step": 11153
    },
    {
      "epoch": 1.0636329650878907e-05,
      "step": 11153,
      "training_step_time": 8.574221849441528
    },
    {
      "epoch": 1.0637283325195312e-05,
      "model_forward_time": 1.758716106414795,
      "step": 11154
    },
    {
      "epoch": 1.0637283325195312e-05,
      "step": 11154,
      "training_step_time": 8.552367687225342
    },
    {
      "epoch": 1.0638236999511718e-05,
      "model_forward_time": 2.910064458847046,
      "step": 11155
    },
    {
      "epoch": 1.0638236999511718e-05,
      "step": 11155,
      "training_step_time": 8.999993562698364
    },
    {
      "epoch": 1.0639190673828126e-05,
      "model_forward_time": 1.7550294399261475,
      "step": 11156
    },
    {
      "epoch": 1.0639190673828126e-05,
      "step": 11156,
      "training_step_time": 8.609312295913696
    },
    {
      "epoch": 1.0640144348144531e-05,
      "model_forward_time": 1.750211238861084,
      "step": 11157
    },
    {
      "epoch": 1.0640144348144531e-05,
      "step": 11157,
      "training_step_time": 7.524872064590454
    },
    {
      "epoch": 1.0641098022460937e-05,
      "model_forward_time": 1.3942995071411133,
      "step": 11158
    },
    {
      "epoch": 1.0641098022460937e-05,
      "step": 11158,
      "training_step_time": 6.714351177215576
    },
    {
      "epoch": 1.0642051696777344e-05,
      "model_forward_time": 1.753854513168335,
      "step": 11159
    },
    {
      "epoch": 1.0642051696777344e-05,
      "step": 11159,
      "training_step_time": 6.6692750453948975
    },
    {
      "epoch": 1.064300537109375e-05,
      "grad_norm": 0.19576486945152283,
      "learning_rate": 9.896649852006916e-06,
      "loss": 0.0723,
      "step": 11160
    },
    {
      "epoch": 1.064300537109375e-05,
      "model_forward_time": 1.755241870880127,
      "step": 11160
    },
    {
      "dynamics_loss_avg": 0.06966334357857704,
      "epoch": 1.064300537109375e-05,
      "step": 11160
    },
    {
      "action_loss_avg": 0.003109712339937687,
      "epoch": 1.064300537109375e-05,
      "step": 11160
    },
    {
      "epoch": 1.064300537109375e-05,
      "step": 11160,
      "training_step_time": 6.685292959213257
    },
    {
      "epoch": 1.0643959045410156e-05,
      "model_forward_time": 1.7553071975708008,
      "step": 11161
    },
    {
      "epoch": 1.0643959045410156e-05,
      "step": 11161,
      "training_step_time": 6.708275079727173
    },
    {
      "epoch": 1.0644912719726563e-05,
      "model_forward_time": 1.7537078857421875,
      "step": 11162
    },
    {
      "epoch": 1.0644912719726563e-05,
      "step": 11162,
      "training_step_time": 7.610752105712891
    },
    {
      "epoch": 1.0645866394042969e-05,
      "model_forward_time": 1.7925388813018799,
      "step": 11163
    },
    {
      "epoch": 1.0645866394042969e-05,
      "step": 11163,
      "training_step_time": 8.51746916770935
    },
    {
      "epoch": 1.0646820068359375e-05,
      "model_forward_time": 1.743920087814331,
      "step": 11164
    },
    {
      "epoch": 1.0646820068359375e-05,
      "step": 11164,
      "training_step_time": 8.881688117980957
    },
    {
      "epoch": 1.064777374267578e-05,
      "model_forward_time": 1.7582292556762695,
      "step": 11165
    },
    {
      "epoch": 1.064777374267578e-05,
      "step": 11165,
      "training_step_time": 7.610931873321533
    },
    {
      "epoch": 1.0648727416992188e-05,
      "model_forward_time": 1.761652946472168,
      "step": 11166
    },
    {
      "epoch": 1.0648727416992188e-05,
      "step": 11166,
      "training_step_time": 7.530852794647217
    },
    {
      "epoch": 1.0649681091308594e-05,
      "model_forward_time": 1.7642207145690918,
      "step": 11167
    },
    {
      "epoch": 1.0649681091308594e-05,
      "step": 11167,
      "training_step_time": 7.373858690261841
    },
    {
      "epoch": 1.0650634765625e-05,
      "model_forward_time": 1.7364578247070312,
      "step": 11168
    },
    {
      "epoch": 1.0650634765625e-05,
      "step": 11168,
      "training_step_time": 6.678009510040283
    },
    {
      "epoch": 1.0651588439941407e-05,
      "model_forward_time": 1.7418060302734375,
      "step": 11169
    },
    {
      "epoch": 1.0651588439941407e-05,
      "step": 11169,
      "training_step_time": 6.729712247848511
    },
    {
      "epoch": 1.0652542114257812e-05,
      "grad_norm": 0.24844397604465485,
      "learning_rate": 9.896315138333177e-06,
      "loss": 0.0722,
      "step": 11170
    },
    {
      "epoch": 1.0652542114257812e-05,
      "model_forward_time": 1.7409274578094482,
      "step": 11170
    },
    {
      "dynamics_loss_avg": 0.07553807608783245,
      "epoch": 1.0652542114257812e-05,
      "step": 11170
    },
    {
      "action_loss_avg": 0.0035839647985994815,
      "epoch": 1.0652542114257812e-05,
      "step": 11170
    },
    {
      "epoch": 1.0652542114257812e-05,
      "step": 11170,
      "training_step_time": 6.6655073165893555
    },
    {
      "epoch": 1.0653495788574218e-05,
      "model_forward_time": 1.7467436790466309,
      "step": 11171
    },
    {
      "epoch": 1.0653495788574218e-05,
      "step": 11171,
      "training_step_time": 8.774963855743408
    },
    {
      "epoch": 1.0654449462890626e-05,
      "model_forward_time": 1.7458631992340088,
      "step": 11172
    },
    {
      "epoch": 1.0654449462890626e-05,
      "step": 11172,
      "training_step_time": 8.209545612335205
    },
    {
      "epoch": 1.0655403137207031e-05,
      "model_forward_time": 1.741652488708496,
      "step": 11173
    },
    {
      "epoch": 1.0655403137207031e-05,
      "step": 11173,
      "training_step_time": 10.268105745315552
    },
    {
      "epoch": 1.0656356811523437e-05,
      "model_forward_time": 2.9165663719177246,
      "step": 11174
    },
    {
      "epoch": 1.0656356811523437e-05,
      "step": 11174,
      "training_step_time": 9.535561800003052
    },
    {
      "epoch": 1.0657310485839844e-05,
      "model_forward_time": 1.7384834289550781,
      "step": 11175
    },
    {
      "epoch": 1.0657310485839844e-05,
      "step": 11175,
      "training_step_time": 7.541378021240234
    },
    {
      "epoch": 1.065826416015625e-05,
      "model_forward_time": 1.7407793998718262,
      "step": 11176
    },
    {
      "epoch": 1.065826416015625e-05,
      "step": 11176,
      "training_step_time": 7.493344306945801
    },
    {
      "epoch": 1.0659217834472656e-05,
      "model_forward_time": 1.040308952331543,
      "step": 11177
    },
    {
      "epoch": 1.0659217834472656e-05,
      "step": 11177,
      "training_step_time": 7.055154800415039
    },
    {
      "epoch": 1.0660171508789063e-05,
      "model_forward_time": 1.7435154914855957,
      "step": 11178
    },
    {
      "epoch": 1.0660171508789063e-05,
      "step": 11178,
      "training_step_time": 6.682159900665283
    },
    {
      "epoch": 1.0661125183105469e-05,
      "model_forward_time": 1.7851297855377197,
      "step": 11179
    },
    {
      "epoch": 1.0661125183105469e-05,
      "step": 11179,
      "training_step_time": 6.6484129428863525
    },
    {
      "epoch": 1.0662078857421875e-05,
      "grad_norm": 0.21490058302879333,
      "learning_rate": 9.895979889205775e-06,
      "loss": 0.0658,
      "step": 11180
    },
    {
      "epoch": 1.0662078857421875e-05,
      "model_forward_time": 1.752305030822754,
      "step": 11180
    },
    {
      "dynamics_loss_avg": 0.062019108608365056,
      "epoch": 1.0662078857421875e-05,
      "step": 11180
    },
    {
      "action_loss_avg": 0.00399989727884531,
      "epoch": 1.0662078857421875e-05,
      "step": 11180
    },
    {
      "epoch": 1.0662078857421875e-05,
      "step": 11180,
      "training_step_time": 7.455069303512573
    },
    {
      "epoch": 1.066303253173828e-05,
      "model_forward_time": 1.751168966293335,
      "step": 11181
    },
    {
      "epoch": 1.066303253173828e-05,
      "step": 11181,
      "training_step_time": 7.427679061889648
    },
    {
      "epoch": 1.0663986206054688e-05,
      "model_forward_time": 1.8040554523468018,
      "step": 11182
    },
    {
      "epoch": 1.0663986206054688e-05,
      "step": 11182,
      "training_step_time": 7.62031364440918
    },
    {
      "epoch": 1.0664939880371094e-05,
      "model_forward_time": 2.890918254852295,
      "step": 11183
    },
    {
      "epoch": 1.0664939880371094e-05,
      "step": 11183,
      "training_step_time": 7.594463348388672
    },
    {
      "epoch": 1.06658935546875e-05,
      "model_forward_time": 1.7445759773254395,
      "step": 11184
    },
    {
      "epoch": 1.06658935546875e-05,
      "step": 11184,
      "training_step_time": 8.553625345230103
    },
    {
      "epoch": 1.0666847229003907e-05,
      "model_forward_time": 1.7507801055908203,
      "step": 11185
    },
    {
      "epoch": 1.0666847229003907e-05,
      "step": 11185,
      "training_step_time": 7.508323669433594
    },
    {
      "epoch": 1.0667800903320312e-05,
      "model_forward_time": 1.7576477527618408,
      "step": 11186
    },
    {
      "epoch": 1.0667800903320312e-05,
      "step": 11186,
      "training_step_time": 7.107278108596802
    },
    {
      "epoch": 1.0668754577636718e-05,
      "model_forward_time": 1.7487874031066895,
      "step": 11187
    },
    {
      "epoch": 1.0668754577636718e-05,
      "step": 11187,
      "training_step_time": 6.6216020584106445
    },
    {
      "epoch": 1.0669708251953126e-05,
      "model_forward_time": 1.736356258392334,
      "step": 11188
    },
    {
      "epoch": 1.0669708251953126e-05,
      "step": 11188,
      "training_step_time": 6.657487630844116
    },
    {
      "epoch": 1.0670661926269531e-05,
      "model_forward_time": 1.740642786026001,
      "step": 11189
    },
    {
      "epoch": 1.0670661926269531e-05,
      "step": 11189,
      "training_step_time": 8.171469449996948
    },
    {
      "epoch": 1.0671615600585937e-05,
      "grad_norm": 0.1759171038866043,
      "learning_rate": 9.895644104661372e-06,
      "loss": 0.0656,
      "step": 11190
    },
    {
      "epoch": 1.0671615600585937e-05,
      "model_forward_time": 1.7559211254119873,
      "step": 11190
    },
    {
      "dynamics_loss_avg": 0.07536689937114716,
      "epoch": 1.0671615600585937e-05,
      "step": 11190
    },
    {
      "action_loss_avg": 0.004607230285182595,
      "epoch": 1.0671615600585937e-05,
      "step": 11190
    },
    {
      "epoch": 1.0671615600585937e-05,
      "step": 11190,
      "training_step_time": 7.454058885574341
    },
    {
      "epoch": 1.0672569274902344e-05,
      "model_forward_time": 1.7951383590698242,
      "step": 11191
    },
    {
      "epoch": 1.0672569274902344e-05,
      "step": 11191,
      "training_step_time": 8.95160722732544
    },
    {
      "epoch": 1.067352294921875e-05,
      "model_forward_time": 2.9224424362182617,
      "step": 11192
    },
    {
      "epoch": 1.067352294921875e-05,
      "step": 11192,
      "training_step_time": 8.472120523452759
    },
    {
      "epoch": 1.0674476623535156e-05,
      "model_forward_time": 1.7610619068145752,
      "step": 11193
    },
    {
      "epoch": 1.0674476623535156e-05,
      "step": 11193,
      "training_step_time": 9.298903703689575
    },
    {
      "epoch": 1.0675430297851563e-05,
      "model_forward_time": 1.759065866470337,
      "step": 11194
    },
    {
      "epoch": 1.0675430297851563e-05,
      "step": 11194,
      "training_step_time": 7.559533357620239
    },
    {
      "epoch": 1.0676383972167969e-05,
      "model_forward_time": 1.7590889930725098,
      "step": 11195
    },
    {
      "epoch": 1.0676383972167969e-05,
      "step": 11195,
      "training_step_time": 7.195632696151733
    },
    {
      "epoch": 1.0677337646484375e-05,
      "model_forward_time": 1.7549211978912354,
      "step": 11196
    },
    {
      "epoch": 1.0677337646484375e-05,
      "step": 11196,
      "training_step_time": 6.710232257843018
    },
    {
      "epoch": 1.067829132080078e-05,
      "model_forward_time": 1.7435824871063232,
      "step": 11197
    },
    {
      "epoch": 1.067829132080078e-05,
      "step": 11197,
      "training_step_time": 6.6650230884552
    },
    {
      "epoch": 1.0679244995117188e-05,
      "model_forward_time": 1.7501788139343262,
      "step": 11198
    },
    {
      "epoch": 1.0679244995117188e-05,
      "step": 11198,
      "training_step_time": 8.495604276657104
    },
    {
      "epoch": 1.0680198669433594e-05,
      "model_forward_time": 1.7514772415161133,
      "step": 11199
    },
    {
      "epoch": 1.0680198669433594e-05,
      "step": 11199,
      "training_step_time": 7.594368934631348
    },
    {
      "epoch": 1.068115234375e-05,
      "grad_norm": 0.2445945292711258,
      "learning_rate": 9.895307784736692e-06,
      "loss": 0.066,
      "step": 11200
    },
    {
      "epoch": 1.068115234375e-05,
      "model_forward_time": 1.8169524669647217,
      "step": 11200
    },
    {
      "dynamics_loss_avg": 0.057954294607043266,
      "epoch": 1.068115234375e-05,
      "step": 11200
    },
    {
      "action_loss_avg": 0.0032992739113979043,
      "epoch": 1.068115234375e-05,
      "step": 11200
    },
    {
      "epoch": 1.068115234375e-05,
      "step": 11200,
      "training_step_time": 10.364710092544556
    },
    {
      "epoch": 1.0682106018066407e-05,
      "model_forward_time": 2.8409080505371094,
      "step": 11201
    },
    {
      "epoch": 1.0682106018066407e-05,
      "step": 11201,
      "training_step_time": 10.211164951324463
    },
    {
      "epoch": 1.0683059692382812e-05,
      "model_forward_time": 1.7607293128967285,
      "step": 11202
    },
    {
      "epoch": 1.0683059692382812e-05,
      "step": 11202,
      "training_step_time": 8.346887826919556
    },
    {
      "epoch": 1.0684013366699218e-05,
      "model_forward_time": 1.7459526062011719,
      "step": 11203
    },
    {
      "epoch": 1.0684013366699218e-05,
      "step": 11203,
      "training_step_time": 7.554315090179443
    },
    {
      "epoch": 1.0684967041015626e-05,
      "model_forward_time": 1.7404935359954834,
      "step": 11204
    },
    {
      "epoch": 1.0684967041015626e-05,
      "step": 11204,
      "training_step_time": 7.439671277999878
    },
    {
      "epoch": 1.0685920715332031e-05,
      "model_forward_time": 1.7541117668151855,
      "step": 11205
    },
    {
      "epoch": 1.0685920715332031e-05,
      "step": 11205,
      "training_step_time": 7.150043249130249
    },
    {
      "epoch": 1.0686874389648437e-05,
      "model_forward_time": 1.3775415420532227,
      "step": 11206
    },
    {
      "epoch": 1.0686874389648437e-05,
      "step": 11206,
      "training_step_time": 6.739082336425781
    },
    {
      "epoch": 1.0687828063964844e-05,
      "model_forward_time": 1.7300386428833008,
      "step": 11207
    },
    {
      "epoch": 1.0687828063964844e-05,
      "step": 11207,
      "training_step_time": 7.558451414108276
    },
    {
      "epoch": 1.068878173828125e-05,
      "model_forward_time": 1.7263410091400146,
      "step": 11208
    },
    {
      "epoch": 1.068878173828125e-05,
      "step": 11208,
      "training_step_time": 8.536556720733643
    },
    {
      "epoch": 1.0689735412597656e-05,
      "model_forward_time": 1.762718677520752,
      "step": 11209
    },
    {
      "epoch": 1.0689735412597656e-05,
      "step": 11209,
      "training_step_time": 8.49284553527832
    },
    {
      "epoch": 1.0690689086914063e-05,
      "grad_norm": 0.15374593436717987,
      "learning_rate": 9.894970929468512e-06,
      "loss": 0.0647,
      "step": 11210
    },
    {
      "epoch": 1.0690689086914063e-05,
      "model_forward_time": 1.7420132160186768,
      "step": 11210
    },
    {
      "dynamics_loss_avg": 0.05212215334177017,
      "epoch": 1.0690689086914063e-05,
      "step": 11210
    },
    {
      "action_loss_avg": 0.002290279953740537,
      "epoch": 1.0690689086914063e-05,
      "step": 11210
    },
    {
      "epoch": 1.0690689086914063e-05,
      "step": 11210,
      "training_step_time": 9.898247480392456
    },
    {
      "epoch": 1.0691642761230469e-05,
      "model_forward_time": 2.926037311553955,
      "step": 11211
    },
    {
      "epoch": 1.0691642761230469e-05,
      "step": 11211,
      "training_step_time": 9.907192707061768
    },
    {
      "epoch": 1.0692596435546875e-05,
      "model_forward_time": 1.7460765838623047,
      "step": 11212
    },
    {
      "epoch": 1.0692596435546875e-05,
      "step": 11212,
      "training_step_time": 8.070032119750977
    },
    {
      "epoch": 1.069355010986328e-05,
      "model_forward_time": 1.7325844764709473,
      "step": 11213
    },
    {
      "epoch": 1.069355010986328e-05,
      "step": 11213,
      "training_step_time": 7.330045223236084
    },
    {
      "epoch": 1.0694503784179688e-05,
      "model_forward_time": 1.7319180965423584,
      "step": 11214
    },
    {
      "epoch": 1.0694503784179688e-05,
      "step": 11214,
      "training_step_time": 7.13786506652832
    },
    {
      "epoch": 1.0695457458496094e-05,
      "model_forward_time": 1.7362194061279297,
      "step": 11215
    },
    {
      "epoch": 1.0695457458496094e-05,
      "step": 11215,
      "training_step_time": 6.682512044906616
    },
    {
      "epoch": 1.06964111328125e-05,
      "model_forward_time": 1.7431001663208008,
      "step": 11216
    },
    {
      "epoch": 1.06964111328125e-05,
      "step": 11216,
      "training_step_time": 7.457667112350464
    },
    {
      "epoch": 1.0697364807128907e-05,
      "model_forward_time": 1.7330832481384277,
      "step": 11217
    },
    {
      "epoch": 1.0697364807128907e-05,
      "step": 11217,
      "training_step_time": 8.436389207839966
    },
    {
      "epoch": 1.0698318481445312e-05,
      "model_forward_time": 1.7435462474822998,
      "step": 11218
    },
    {
      "epoch": 1.0698318481445312e-05,
      "step": 11218,
      "training_step_time": 9.88261866569519
    },
    {
      "epoch": 1.0699272155761718e-05,
      "model_forward_time": 1.7228960990905762,
      "step": 11219
    },
    {
      "epoch": 1.0699272155761718e-05,
      "step": 11219,
      "training_step_time": 8.61029577255249
    },
    {
      "epoch": 1.0700225830078126e-05,
      "grad_norm": 0.1990823745727539,
      "learning_rate": 9.894633538893671e-06,
      "loss": 0.0635,
      "step": 11220
    },
    {
      "epoch": 1.0700225830078126e-05,
      "model_forward_time": 2.916154384613037,
      "step": 11220
    },
    {
      "dynamics_loss_avg": 0.0627179317176342,
      "epoch": 1.0700225830078126e-05,
      "step": 11220
    },
    {
      "action_loss_avg": 0.002961631427751854,
      "epoch": 1.0700225830078126e-05,
      "step": 11220
    },
    {
      "epoch": 1.0700225830078126e-05,
      "step": 11220,
      "training_step_time": 9.37312912940979
    },
    {
      "epoch": 1.0701179504394531e-05,
      "model_forward_time": 1.7321434020996094,
      "step": 11221
    },
    {
      "epoch": 1.0701179504394531e-05,
      "step": 11221,
      "training_step_time": 7.592727422714233
    },
    {
      "epoch": 1.0702133178710937e-05,
      "model_forward_time": 1.766451358795166,
      "step": 11222
    },
    {
      "epoch": 1.0702133178710937e-05,
      "step": 11222,
      "training_step_time": 7.4399573802948
    },
    {
      "epoch": 1.0703086853027344e-05,
      "model_forward_time": 1.7447950839996338,
      "step": 11223
    },
    {
      "epoch": 1.0703086853027344e-05,
      "step": 11223,
      "training_step_time": 7.211319446563721
    },
    {
      "epoch": 1.070404052734375e-05,
      "model_forward_time": 1.7467646598815918,
      "step": 11224
    },
    {
      "epoch": 1.070404052734375e-05,
      "step": 11224,
      "training_step_time": 6.792142629623413
    },
    {
      "epoch": 1.0704994201660156e-05,
      "model_forward_time": 1.747063398361206,
      "step": 11225
    },
    {
      "epoch": 1.0704994201660156e-05,
      "step": 11225,
      "training_step_time": 7.661589860916138
    },
    {
      "epoch": 1.0705947875976563e-05,
      "model_forward_time": 1.7484238147735596,
      "step": 11226
    },
    {
      "epoch": 1.0705947875976563e-05,
      "step": 11226,
      "training_step_time": 8.622795104980469
    },
    {
      "epoch": 1.0706901550292969e-05,
      "model_forward_time": 1.7510206699371338,
      "step": 11227
    },
    {
      "epoch": 1.0706901550292969e-05,
      "step": 11227,
      "training_step_time": 7.642361640930176
    },
    {
      "epoch": 1.0707855224609375e-05,
      "model_forward_time": 1.8137085437774658,
      "step": 11228
    },
    {
      "epoch": 1.0707855224609375e-05,
      "step": 11228,
      "training_step_time": 8.836005687713623
    },
    {
      "epoch": 1.070880889892578e-05,
      "model_forward_time": 2.8565731048583984,
      "step": 11229
    },
    {
      "epoch": 1.070880889892578e-05,
      "step": 11229,
      "training_step_time": 7.761984348297119
    },
    {
      "epoch": 1.0709762573242188e-05,
      "grad_norm": 0.2021438628435135,
      "learning_rate": 9.894295613049065e-06,
      "loss": 0.0668,
      "step": 11230
    },
    {
      "epoch": 1.0709762573242188e-05,
      "model_forward_time": 1.7563164234161377,
      "step": 11230
    },
    {
      "dynamics_loss_avg": 0.08475847281515599,
      "epoch": 1.0709762573242188e-05,
      "step": 11230
    },
    {
      "action_loss_avg": 0.003948007896542549,
      "epoch": 1.0709762573242188e-05,
      "step": 11230
    },
    {
      "epoch": 1.0709762573242188e-05,
      "step": 11230,
      "training_step_time": 8.341679811477661
    },
    {
      "epoch": 1.0710716247558594e-05,
      "model_forward_time": 1.7556798458099365,
      "step": 11231
    },
    {
      "epoch": 1.0710716247558594e-05,
      "step": 11231,
      "training_step_time": 7.83252739906311
    },
    {
      "epoch": 1.0711669921875e-05,
      "model_forward_time": 1.7476329803466797,
      "step": 11232
    },
    {
      "epoch": 1.0711669921875e-05,
      "step": 11232,
      "training_step_time": 8.016205072402954
    },
    {
      "epoch": 1.0712623596191407e-05,
      "model_forward_time": 1.752450942993164,
      "step": 11233
    },
    {
      "epoch": 1.0712623596191407e-05,
      "step": 11233,
      "training_step_time": 6.8593385219573975
    },
    {
      "epoch": 1.0713577270507812e-05,
      "model_forward_time": 1.7413654327392578,
      "step": 11234
    },
    {
      "epoch": 1.0713577270507812e-05,
      "step": 11234,
      "training_step_time": 7.648731470108032
    },
    {
      "epoch": 1.0714530944824218e-05,
      "model_forward_time": 1.757167100906372,
      "step": 11235
    },
    {
      "epoch": 1.0714530944824218e-05,
      "step": 11235,
      "training_step_time": 9.560441732406616
    },
    {
      "epoch": 1.0715484619140626e-05,
      "model_forward_time": 1.7412846088409424,
      "step": 11236
    },
    {
      "epoch": 1.0715484619140626e-05,
      "step": 11236,
      "training_step_time": 7.706673622131348
    },
    {
      "epoch": 1.0716438293457031e-05,
      "model_forward_time": 1.8254141807556152,
      "step": 11237
    },
    {
      "epoch": 1.0716438293457031e-05,
      "step": 11237,
      "training_step_time": 8.74181079864502
    },
    {
      "epoch": 1.0717391967773437e-05,
      "model_forward_time": 2.870579957962036,
      "step": 11238
    },
    {
      "epoch": 1.0717391967773437e-05,
      "step": 11238,
      "training_step_time": 8.498252868652344
    },
    {
      "epoch": 1.0718345642089844e-05,
      "model_forward_time": 1.756883144378662,
      "step": 11239
    },
    {
      "epoch": 1.0718345642089844e-05,
      "step": 11239,
      "training_step_time": 7.777566432952881
    },
    {
      "epoch": 1.071929931640625e-05,
      "grad_norm": 0.2465008795261383,
      "learning_rate": 9.893957151971649e-06,
      "loss": 0.0676,
      "step": 11240
    },
    {
      "epoch": 1.071929931640625e-05,
      "model_forward_time": 1.7418413162231445,
      "step": 11240
    },
    {
      "dynamics_loss_avg": 0.04544129855930805,
      "epoch": 1.071929931640625e-05,
      "step": 11240
    },
    {
      "action_loss_avg": 0.0020112535101361574,
      "epoch": 1.071929931640625e-05,
      "step": 11240
    },
    {
      "epoch": 1.071929931640625e-05,
      "step": 11240,
      "training_step_time": 7.688373565673828
    },
    {
      "epoch": 1.0720252990722656e-05,
      "model_forward_time": 1.7524051666259766,
      "step": 11241
    },
    {
      "epoch": 1.0720252990722656e-05,
      "step": 11241,
      "training_step_time": 7.150415658950806
    },
    {
      "epoch": 1.0721206665039063e-05,
      "model_forward_time": 1.7638170719146729,
      "step": 11242
    },
    {
      "epoch": 1.0721206665039063e-05,
      "step": 11242,
      "training_step_time": 7.28580641746521
    },
    {
      "epoch": 1.0722160339355469e-05,
      "model_forward_time": 1.7572252750396729,
      "step": 11243
    },
    {
      "epoch": 1.0722160339355469e-05,
      "step": 11243,
      "training_step_time": 7.689598560333252
    },
    {
      "epoch": 1.0723114013671875e-05,
      "model_forward_time": 1.7534713745117188,
      "step": 11244
    },
    {
      "epoch": 1.0723114013671875e-05,
      "step": 11244,
      "training_step_time": 7.484462022781372
    },
    {
      "epoch": 1.072406768798828e-05,
      "model_forward_time": 1.7475764751434326,
      "step": 11245
    },
    {
      "epoch": 1.072406768798828e-05,
      "step": 11245,
      "training_step_time": 7.870415925979614
    },
    {
      "epoch": 1.0725021362304688e-05,
      "model_forward_time": 1.8121144771575928,
      "step": 11246
    },
    {
      "epoch": 1.0725021362304688e-05,
      "step": 11246,
      "training_step_time": 9.52622127532959
    },
    {
      "epoch": 1.0725975036621094e-05,
      "model_forward_time": 1.749448537826538,
      "step": 11247
    },
    {
      "epoch": 1.0725975036621094e-05,
      "step": 11247,
      "training_step_time": 8.62653923034668
    },
    {
      "epoch": 1.07269287109375e-05,
      "model_forward_time": 1.7534606456756592,
      "step": 11248
    },
    {
      "epoch": 1.07269287109375e-05,
      "step": 11248,
      "training_step_time": 8.359212398529053
    },
    {
      "epoch": 1.0727882385253907e-05,
      "model_forward_time": 1.7562882900238037,
      "step": 11249
    },
    {
      "epoch": 1.0727882385253907e-05,
      "step": 11249,
      "training_step_time": 7.894877195358276
    },
    {
      "epoch": 1.0728836059570312e-05,
      "grad_norm": 0.2337803840637207,
      "learning_rate": 9.893618155698437e-06,
      "loss": 0.069,
      "step": 11250
    },
    {
      "epoch": 1.0728836059570312e-05,
      "model_forward_time": 1.7592017650604248,
      "step": 11250
    },
    {
      "dynamics_loss_avg": 0.061041181348264216,
      "epoch": 1.0728836059570312e-05,
      "step": 11250
    },
    {
      "action_loss_avg": 0.003441183548420668,
      "epoch": 1.0728836059570312e-05,
      "step": 11250
    },
    {
      "epoch": 1.0728836059570312e-05,
      "step": 11250,
      "training_step_time": 7.12822413444519
    },
    {
      "epoch": 1.0729789733886718e-05,
      "model_forward_time": 1.7419660091400146,
      "step": 11251
    },
    {
      "epoch": 1.0729789733886718e-05,
      "step": 11251,
      "training_step_time": 7.2557053565979
    },
    {
      "epoch": 1.0730743408203126e-05,
      "model_forward_time": 1.745290756225586,
      "step": 11252
    },
    {
      "epoch": 1.0730743408203126e-05,
      "step": 11252,
      "training_step_time": 7.617736101150513
    },
    {
      "epoch": 1.0731697082519531e-05,
      "model_forward_time": 1.3656895160675049,
      "step": 11253
    },
    {
      "epoch": 1.0731697082519531e-05,
      "step": 11253,
      "training_step_time": 7.443091630935669
    },
    {
      "epoch": 1.0732650756835937e-05,
      "model_forward_time": 1.75514817237854,
      "step": 11254
    },
    {
      "epoch": 1.0732650756835937e-05,
      "step": 11254,
      "training_step_time": 7.7525012493133545
    },
    {
      "epoch": 1.0733604431152344e-05,
      "model_forward_time": 1.0470085144042969,
      "step": 11255
    },
    {
      "epoch": 1.0733604431152344e-05,
      "step": 11255,
      "training_step_time": 7.952251195907593
    },
    {
      "epoch": 1.073455810546875e-05,
      "model_forward_time": 1.7444305419921875,
      "step": 11256
    },
    {
      "epoch": 1.073455810546875e-05,
      "step": 11256,
      "training_step_time": 9.023021936416626
    },
    {
      "epoch": 1.0735511779785156e-05,
      "model_forward_time": 2.5377581119537354,
      "step": 11257
    },
    {
      "epoch": 1.0735511779785156e-05,
      "step": 11257,
      "training_step_time": 7.788303375244141
    },
    {
      "epoch": 1.0736465454101562e-05,
      "model_forward_time": 1.7489945888519287,
      "step": 11258
    },
    {
      "epoch": 1.0736465454101562e-05,
      "step": 11258,
      "training_step_time": 8.813786268234253
    },
    {
      "epoch": 1.0737419128417969e-05,
      "model_forward_time": 1.741774320602417,
      "step": 11259
    },
    {
      "epoch": 1.0737419128417969e-05,
      "step": 11259,
      "training_step_time": 7.642685651779175
    },
    {
      "epoch": 1.0738372802734375e-05,
      "grad_norm": 0.35249218344688416,
      "learning_rate": 9.8932786242665e-06,
      "loss": 0.0671,
      "step": 11260
    },
    {
      "epoch": 1.0738372802734375e-05,
      "model_forward_time": 1.723935842514038,
      "step": 11260
    },
    {
      "dynamics_loss_avg": 0.07509139012545347,
      "epoch": 1.0738372802734375e-05,
      "step": 11260
    },
    {
      "action_loss_avg": 0.005517546075861901,
      "epoch": 1.0738372802734375e-05,
      "step": 11260
    },
    {
      "epoch": 1.0738372802734375e-05,
      "step": 11260,
      "training_step_time": 7.164656400680542
    },
    {
      "epoch": 1.073932647705078e-05,
      "model_forward_time": 1.753255844116211,
      "step": 11261
    },
    {
      "epoch": 1.073932647705078e-05,
      "step": 11261,
      "training_step_time": 7.6297008991241455
    },
    {
      "epoch": 1.0740280151367188e-05,
      "model_forward_time": 1.7517929077148438,
      "step": 11262
    },
    {
      "epoch": 1.0740280151367188e-05,
      "step": 11262,
      "training_step_time": 7.340996026992798
    },
    {
      "epoch": 1.0741233825683594e-05,
      "model_forward_time": 1.740464448928833,
      "step": 11263
    },
    {
      "epoch": 1.0741233825683594e-05,
      "step": 11263,
      "training_step_time": 7.68449592590332
    },
    {
      "epoch": 1.07421875e-05,
      "model_forward_time": 1.7407515048980713,
      "step": 11264
    },
    {
      "epoch": 1.07421875e-05,
      "step": 11264,
      "training_step_time": 9.035410404205322
    },
    {
      "epoch": 1.0743141174316407e-05,
      "model_forward_time": 1.7335164546966553,
      "step": 11265
    },
    {
      "epoch": 1.0743141174316407e-05,
      "step": 11265,
      "training_step_time": 9.590596675872803
    },
    {
      "epoch": 1.0744094848632813e-05,
      "model_forward_time": 2.840792417526245,
      "step": 11266
    },
    {
      "epoch": 1.0744094848632813e-05,
      "step": 11266,
      "training_step_time": 8.27401614189148
    },
    {
      "epoch": 1.0745048522949218e-05,
      "model_forward_time": 1.7419986724853516,
      "step": 11267
    },
    {
      "epoch": 1.0745048522949218e-05,
      "step": 11267,
      "training_step_time": 8.701049566268921
    },
    {
      "epoch": 1.0746002197265626e-05,
      "model_forward_time": 1.7407491207122803,
      "step": 11268
    },
    {
      "epoch": 1.0746002197265626e-05,
      "step": 11268,
      "training_step_time": 8.407358884811401
    },
    {
      "epoch": 1.0746955871582031e-05,
      "model_forward_time": 1.38730788230896,
      "step": 11269
    },
    {
      "epoch": 1.0746955871582031e-05,
      "step": 11269,
      "training_step_time": 7.86919641494751
    },
    {
      "epoch": 1.0747909545898437e-05,
      "grad_norm": 0.26948821544647217,
      "learning_rate": 9.89293855771297e-06,
      "loss": 0.067,
      "step": 11270
    },
    {
      "epoch": 1.0747909545898437e-05,
      "model_forward_time": 1.7493352890014648,
      "step": 11270
    },
    {
      "dynamics_loss_avg": 0.061968004517257216,
      "epoch": 1.0747909545898437e-05,
      "step": 11270
    },
    {
      "action_loss_avg": 0.003508552734274417,
      "epoch": 1.0747909545898437e-05,
      "step": 11270
    },
    {
      "epoch": 1.0747909545898437e-05,
      "step": 11270,
      "training_step_time": 7.797592401504517
    },
    {
      "epoch": 1.0748863220214845e-05,
      "model_forward_time": 1.750657558441162,
      "step": 11271
    },
    {
      "epoch": 1.0748863220214845e-05,
      "step": 11271,
      "training_step_time": 6.788600206375122
    },
    {
      "epoch": 1.074981689453125e-05,
      "model_forward_time": 1.7565925121307373,
      "step": 11272
    },
    {
      "epoch": 1.074981689453125e-05,
      "step": 11272,
      "training_step_time": 7.684486389160156
    },
    {
      "epoch": 1.0750770568847656e-05,
      "model_forward_time": 1.7485628128051758,
      "step": 11273
    },
    {
      "epoch": 1.0750770568847656e-05,
      "step": 11273,
      "training_step_time": 7.5959460735321045
    },
    {
      "epoch": 1.0751724243164062e-05,
      "model_forward_time": 1.8145103454589844,
      "step": 11274
    },
    {
      "epoch": 1.0751724243164062e-05,
      "step": 11274,
      "training_step_time": 7.841591119766235
    },
    {
      "epoch": 1.0752677917480469e-05,
      "model_forward_time": 1.7479212284088135,
      "step": 11275
    },
    {
      "epoch": 1.0752677917480469e-05,
      "step": 11275,
      "training_step_time": 10.269845485687256
    },
    {
      "epoch": 1.0753631591796875e-05,
      "model_forward_time": 2.898521900177002,
      "step": 11276
    },
    {
      "epoch": 1.0753631591796875e-05,
      "step": 11276,
      "training_step_time": 8.261252880096436
    },
    {
      "epoch": 1.075458526611328e-05,
      "model_forward_time": 1.7299110889434814,
      "step": 11277
    },
    {
      "epoch": 1.075458526611328e-05,
      "step": 11277,
      "training_step_time": 7.72176718711853
    },
    {
      "epoch": 1.0755538940429688e-05,
      "model_forward_time": 1.7292938232421875,
      "step": 11278
    },
    {
      "epoch": 1.0755538940429688e-05,
      "step": 11278,
      "training_step_time": 7.223198652267456
    },
    {
      "epoch": 1.0756492614746094e-05,
      "model_forward_time": 1.7254817485809326,
      "step": 11279
    },
    {
      "epoch": 1.0756492614746094e-05,
      "step": 11279,
      "training_step_time": 7.7000274658203125
    },
    {
      "epoch": 1.07574462890625e-05,
      "grad_norm": 0.23887647688388824,
      "learning_rate": 9.892597956075037e-06,
      "loss": 0.0692,
      "step": 11280
    },
    {
      "epoch": 1.07574462890625e-05,
      "model_forward_time": 1.7304449081420898,
      "step": 11280
    },
    {
      "dynamics_loss_avg": 0.06739999670535327,
      "epoch": 1.07574462890625e-05,
      "step": 11280
    },
    {
      "action_loss_avg": 0.004496489380835555,
      "epoch": 1.07574462890625e-05,
      "step": 11280
    },
    {
      "epoch": 1.07574462890625e-05,
      "step": 11280,
      "training_step_time": 6.790083646774292
    },
    {
      "epoch": 1.0758399963378907e-05,
      "model_forward_time": 1.3651986122131348,
      "step": 11281
    },
    {
      "epoch": 1.0758399963378907e-05,
      "step": 11281,
      "training_step_time": 7.694901943206787
    },
    {
      "epoch": 1.0759353637695313e-05,
      "model_forward_time": 1.7353239059448242,
      "step": 11282
    },
    {
      "epoch": 1.0759353637695313e-05,
      "step": 11282,
      "training_step_time": 8.682575464248657
    },
    {
      "epoch": 1.0760307312011718e-05,
      "model_forward_time": 1.790043830871582,
      "step": 11283
    },
    {
      "epoch": 1.0760307312011718e-05,
      "step": 11283,
      "training_step_time": 8.32523250579834
    },
    {
      "epoch": 1.0761260986328126e-05,
      "model_forward_time": 1.7235920429229736,
      "step": 11284
    },
    {
      "epoch": 1.0761260986328126e-05,
      "step": 11284,
      "training_step_time": 8.700474262237549
    },
    {
      "epoch": 1.0762214660644531e-05,
      "model_forward_time": 2.8980863094329834,
      "step": 11285
    },
    {
      "epoch": 1.0762214660644531e-05,
      "step": 11285,
      "training_step_time": 7.770495176315308
    },
    {
      "epoch": 1.0763168334960937e-05,
      "model_forward_time": 1.716902494430542,
      "step": 11286
    },
    {
      "epoch": 1.0763168334960937e-05,
      "step": 11286,
      "training_step_time": 7.735804796218872
    },
    {
      "epoch": 1.0764122009277345e-05,
      "model_forward_time": 1.7261760234832764,
      "step": 11287
    },
    {
      "epoch": 1.0764122009277345e-05,
      "step": 11287,
      "training_step_time": 7.32732367515564
    },
    {
      "epoch": 1.076507568359375e-05,
      "model_forward_time": 1.727522373199463,
      "step": 11288
    },
    {
      "epoch": 1.076507568359375e-05,
      "step": 11288,
      "training_step_time": 7.690011501312256
    },
    {
      "epoch": 1.0766029357910156e-05,
      "model_forward_time": 1.706897258758545,
      "step": 11289
    },
    {
      "epoch": 1.0766029357910156e-05,
      "step": 11289,
      "training_step_time": 7.262383937835693
    },
    {
      "epoch": 1.0766983032226562e-05,
      "grad_norm": 0.23993122577667236,
      "learning_rate": 9.892256819389947e-06,
      "loss": 0.0697,
      "step": 11290
    },
    {
      "epoch": 1.0766983032226562e-05,
      "model_forward_time": 1.7151994705200195,
      "step": 11290
    },
    {
      "dynamics_loss_avg": 0.06745104100555181,
      "epoch": 1.0766983032226562e-05,
      "step": 11290
    },
    {
      "action_loss_avg": 0.0037137736333534123,
      "epoch": 1.0766983032226562e-05,
      "step": 11290
    },
    {
      "epoch": 1.0766983032226562e-05,
      "step": 11290,
      "training_step_time": 7.632480144500732
    },
    {
      "epoch": 1.0767936706542969e-05,
      "model_forward_time": 1.7275168895721436,
      "step": 11291
    },
    {
      "epoch": 1.0767936706542969e-05,
      "step": 11291,
      "training_step_time": 7.57263970375061
    },
    {
      "epoch": 1.0768890380859375e-05,
      "model_forward_time": 1.7351703643798828,
      "step": 11292
    },
    {
      "epoch": 1.0768890380859375e-05,
      "step": 11292,
      "training_step_time": 8.384830474853516
    },
    {
      "epoch": 1.076984405517578e-05,
      "model_forward_time": 1.7213518619537354,
      "step": 11293
    },
    {
      "epoch": 1.076984405517578e-05,
      "step": 11293,
      "training_step_time": 9.452521562576294
    },
    {
      "epoch": 1.0770797729492188e-05,
      "model_forward_time": 2.866534948348999,
      "step": 11294
    },
    {
      "epoch": 1.0770797729492188e-05,
      "step": 11294,
      "training_step_time": 7.643855810165405
    },
    {
      "epoch": 1.0771751403808594e-05,
      "model_forward_time": 1.716322660446167,
      "step": 11295
    },
    {
      "epoch": 1.0771751403808594e-05,
      "step": 11295,
      "training_step_time": 8.346231937408447
    },
    {
      "epoch": 1.0772705078125e-05,
      "model_forward_time": 1.7198200225830078,
      "step": 11296
    },
    {
      "epoch": 1.0772705078125e-05,
      "step": 11296,
      "training_step_time": 7.7821996212005615
    },
    {
      "epoch": 1.0773658752441407e-05,
      "model_forward_time": 1.717435598373413,
      "step": 11297
    },
    {
      "epoch": 1.0773658752441407e-05,
      "step": 11297,
      "training_step_time": 7.670727491378784
    },
    {
      "epoch": 1.0774612426757813e-05,
      "model_forward_time": 1.7158818244934082,
      "step": 11298
    },
    {
      "epoch": 1.0774612426757813e-05,
      "step": 11298,
      "training_step_time": 7.342521905899048
    },
    {
      "epoch": 1.0775566101074218e-05,
      "model_forward_time": 1.713773488998413,
      "step": 11299
    },
    {
      "epoch": 1.0775566101074218e-05,
      "step": 11299,
      "training_step_time": 8.900434970855713
    },
    {
      "epoch": 1.0776519775390626e-05,
      "grad_norm": 0.19062253832817078,
      "learning_rate": 9.891915147695006e-06,
      "loss": 0.0666,
      "step": 11300
    },
    {
      "epoch": 1.0776519775390626e-05,
      "model_forward_time": 1.7099015712738037,
      "step": 11300
    },
    {
      "dynamics_loss_avg": 0.058293068781495094,
      "epoch": 1.0776519775390626e-05,
      "step": 11300
    },
    {
      "action_loss_avg": 0.003183074726257473,
      "epoch": 1.0776519775390626e-05,
      "step": 11300
    },
    {
      "epoch": 1.0776519775390626e-05,
      "step": 11300,
      "training_step_time": 8.536005020141602
    },
    {
      "epoch": 1.0777473449707031e-05,
      "model_forward_time": 1.7369370460510254,
      "step": 11301
    },
    {
      "epoch": 1.0777473449707031e-05,
      "step": 11301,
      "training_step_time": 8.768355131149292
    },
    {
      "epoch": 1.0778427124023437e-05,
      "model_forward_time": 1.7185900211334229,
      "step": 11302
    },
    {
      "epoch": 1.0778427124023437e-05,
      "step": 11302,
      "training_step_time": 9.358915567398071
    },
    {
      "epoch": 1.0779380798339845e-05,
      "model_forward_time": 2.8306219577789307,
      "step": 11303
    },
    {
      "epoch": 1.0779380798339845e-05,
      "step": 11303,
      "training_step_time": 7.78530478477478
    },
    {
      "epoch": 1.078033447265625e-05,
      "model_forward_time": 1.7080798149108887,
      "step": 11304
    },
    {
      "epoch": 1.078033447265625e-05,
      "step": 11304,
      "training_step_time": 7.741431474685669
    },
    {
      "epoch": 1.0781288146972656e-05,
      "model_forward_time": 1.7161550521850586,
      "step": 11305
    },
    {
      "epoch": 1.0781288146972656e-05,
      "step": 11305,
      "training_step_time": 7.751510143280029
    },
    {
      "epoch": 1.0782241821289062e-05,
      "model_forward_time": 1.7156744003295898,
      "step": 11306
    },
    {
      "epoch": 1.0782241821289062e-05,
      "step": 11306,
      "training_step_time": 6.77799654006958
    },
    {
      "epoch": 1.0783195495605469e-05,
      "model_forward_time": 1.7385056018829346,
      "step": 11307
    },
    {
      "epoch": 1.0783195495605469e-05,
      "step": 11307,
      "training_step_time": 7.665992498397827
    },
    {
      "epoch": 1.0784149169921875e-05,
      "model_forward_time": 1.3570775985717773,
      "step": 11308
    },
    {
      "epoch": 1.0784149169921875e-05,
      "step": 11308,
      "training_step_time": 7.506100654602051
    },
    {
      "epoch": 1.078510284423828e-05,
      "model_forward_time": 1.7107596397399902,
      "step": 11309
    },
    {
      "epoch": 1.078510284423828e-05,
      "step": 11309,
      "training_step_time": 8.757493734359741
    },
    {
      "epoch": 1.0786056518554688e-05,
      "grad_norm": 0.2821926176548004,
      "learning_rate": 9.891572941027579e-06,
      "loss": 0.0629,
      "step": 11310
    },
    {
      "epoch": 1.0786056518554688e-05,
      "model_forward_time": 1.714794397354126,
      "step": 11310
    },
    {
      "dynamics_loss_avg": 0.06388344634324312,
      "epoch": 1.0786056518554688e-05,
      "step": 11310
    },
    {
      "action_loss_avg": 0.003670220618369058,
      "epoch": 1.0786056518554688e-05,
      "step": 11310
    },
    {
      "epoch": 1.0786056518554688e-05,
      "step": 11310,
      "training_step_time": 8.043055057525635
    },
    {
      "epoch": 1.0787010192871094e-05,
      "model_forward_time": 1.7284479141235352,
      "step": 11311
    },
    {
      "epoch": 1.0787010192871094e-05,
      "step": 11311,
      "training_step_time": 7.848689079284668
    },
    {
      "epoch": 1.07879638671875e-05,
      "model_forward_time": 1.7651522159576416,
      "step": 11312
    },
    {
      "epoch": 1.07879638671875e-05,
      "step": 11312,
      "training_step_time": 7.854810476303101
    },
    {
      "epoch": 1.0788917541503907e-05,
      "model_forward_time": 2.9394052028656006,
      "step": 11313
    },
    {
      "epoch": 1.0788917541503907e-05,
      "step": 11313,
      "training_step_time": 8.558271408081055
    },
    {
      "epoch": 1.0789871215820313e-05,
      "model_forward_time": 1.732351541519165,
      "step": 11314
    },
    {
      "epoch": 1.0789871215820313e-05,
      "step": 11314,
      "training_step_time": 7.8446009159088135
    },
    {
      "epoch": 1.0790824890136718e-05,
      "model_forward_time": 1.7288925647735596,
      "step": 11315
    },
    {
      "epoch": 1.0790824890136718e-05,
      "step": 11315,
      "training_step_time": 7.3549323081970215
    },
    {
      "epoch": 1.0791778564453126e-05,
      "model_forward_time": 1.7255115509033203,
      "step": 11316
    },
    {
      "epoch": 1.0791778564453126e-05,
      "step": 11316,
      "training_step_time": 7.807496547698975
    },
    {
      "epoch": 1.0792732238769531e-05,
      "model_forward_time": 1.7310194969177246,
      "step": 11317
    },
    {
      "epoch": 1.0792732238769531e-05,
      "step": 11317,
      "training_step_time": 7.517456293106079
    },
    {
      "epoch": 1.0793685913085937e-05,
      "model_forward_time": 1.7371835708618164,
      "step": 11318
    },
    {
      "epoch": 1.0793685913085937e-05,
      "step": 11318,
      "training_step_time": 7.738522529602051
    },
    {
      "epoch": 1.0794639587402345e-05,
      "model_forward_time": 1.7208576202392578,
      "step": 11319
    },
    {
      "epoch": 1.0794639587402345e-05,
      "step": 11319,
      "training_step_time": 7.612981081008911
    },
    {
      "epoch": 1.079559326171875e-05,
      "grad_norm": 0.20178565382957458,
      "learning_rate": 9.89123019942509e-06,
      "loss": 0.0632,
      "step": 11320
    },
    {
      "epoch": 1.079559326171875e-05,
      "model_forward_time": 1.7367615699768066,
      "step": 11320
    },
    {
      "dynamics_loss_avg": 0.07937699723988771,
      "epoch": 1.079559326171875e-05,
      "step": 11320
    },
    {
      "action_loss_avg": 0.003882763592991978,
      "epoch": 1.079559326171875e-05,
      "step": 11320
    },
    {
      "epoch": 1.079559326171875e-05,
      "step": 11320,
      "training_step_time": 7.870697736740112
    },
    {
      "epoch": 1.0796546936035156e-05,
      "model_forward_time": 1.7643406391143799,
      "step": 11321
    },
    {
      "epoch": 1.0796546936035156e-05,
      "step": 11321,
      "training_step_time": 9.611724615097046
    },
    {
      "epoch": 1.0797500610351562e-05,
      "model_forward_time": 2.928788185119629,
      "step": 11322
    },
    {
      "epoch": 1.0797500610351562e-05,
      "step": 11322,
      "training_step_time": 9.692176580429077
    },
    {
      "epoch": 1.0798454284667969e-05,
      "model_forward_time": 1.7372283935546875,
      "step": 11323
    },
    {
      "epoch": 1.0798454284667969e-05,
      "step": 11323,
      "training_step_time": 7.772345542907715
    },
    {
      "epoch": 1.0799407958984375e-05,
      "model_forward_time": 1.7282180786132812,
      "step": 11324
    },
    {
      "epoch": 1.0799407958984375e-05,
      "step": 11324,
      "training_step_time": 7.389580011367798
    },
    {
      "epoch": 1.080036163330078e-05,
      "model_forward_time": 1.7312090396881104,
      "step": 11325
    },
    {
      "epoch": 1.080036163330078e-05,
      "step": 11325,
      "training_step_time": 7.671800851821899
    },
    {
      "epoch": 1.0801315307617188e-05,
      "model_forward_time": 1.7293415069580078,
      "step": 11326
    },
    {
      "epoch": 1.0801315307617188e-05,
      "step": 11326,
      "training_step_time": 7.498328685760498
    },
    {
      "epoch": 1.0802268981933594e-05,
      "model_forward_time": 1.7295281887054443,
      "step": 11327
    },
    {
      "epoch": 1.0802268981933594e-05,
      "step": 11327,
      "training_step_time": 7.805526971817017
    },
    {
      "epoch": 1.080322265625e-05,
      "model_forward_time": 1.7300310134887695,
      "step": 11328
    },
    {
      "epoch": 1.080322265625e-05,
      "step": 11328,
      "training_step_time": 7.5011303424835205
    },
    {
      "epoch": 1.0804176330566407e-05,
      "model_forward_time": 1.755211591720581,
      "step": 11329
    },
    {
      "epoch": 1.0804176330566407e-05,
      "step": 11329,
      "training_step_time": 8.36920166015625
    },
    {
      "epoch": 1.0805130004882813e-05,
      "grad_norm": 0.24222026765346527,
      "learning_rate": 9.890886922925021e-06,
      "loss": 0.0631,
      "step": 11330
    },
    {
      "epoch": 1.0805130004882813e-05,
      "model_forward_time": 1.72438645362854,
      "step": 11330
    },
    {
      "dynamics_loss_avg": 0.05072124265134335,
      "epoch": 1.0805130004882813e-05,
      "step": 11330
    },
    {
      "action_loss_avg": 0.002420068741776049,
      "epoch": 1.0805130004882813e-05,
      "step": 11330
    },
    {
      "epoch": 1.0805130004882813e-05,
      "step": 11330,
      "training_step_time": 7.807913303375244
    },
    {
      "epoch": 1.0806083679199218e-05,
      "model_forward_time": 2.9514448642730713,
      "step": 11331
    },
    {
      "epoch": 1.0806083679199218e-05,
      "step": 11331,
      "training_step_time": 10.730114698410034
    },
    {
      "epoch": 1.0807037353515626e-05,
      "model_forward_time": 1.0130667686462402,
      "step": 11332
    },
    {
      "epoch": 1.0807037353515626e-05,
      "step": 11332,
      "training_step_time": 7.739462375640869
    },
    {
      "epoch": 1.0807991027832031e-05,
      "model_forward_time": 1.7286338806152344,
      "step": 11333
    },
    {
      "epoch": 1.0807991027832031e-05,
      "step": 11333,
      "training_step_time": 7.586413621902466
    },
    {
      "epoch": 1.0808944702148437e-05,
      "model_forward_time": 1.3635826110839844,
      "step": 11334
    },
    {
      "epoch": 1.0808944702148437e-05,
      "step": 11334,
      "training_step_time": 7.590179681777954
    },
    {
      "epoch": 1.0809898376464845e-05,
      "model_forward_time": 1.7230658531188965,
      "step": 11335
    },
    {
      "epoch": 1.0809898376464845e-05,
      "step": 11335,
      "training_step_time": 7.534518241882324
    },
    {
      "epoch": 1.081085205078125e-05,
      "model_forward_time": 1.7328171730041504,
      "step": 11336
    },
    {
      "epoch": 1.081085205078125e-05,
      "step": 11336,
      "training_step_time": 7.715505838394165
    },
    {
      "epoch": 1.0811805725097656e-05,
      "model_forward_time": 1.72808837890625,
      "step": 11337
    },
    {
      "epoch": 1.0811805725097656e-05,
      "step": 11337,
      "training_step_time": 7.5176825523376465
    },
    {
      "epoch": 1.0812759399414062e-05,
      "model_forward_time": 1.749889850616455,
      "step": 11338
    },
    {
      "epoch": 1.0812759399414062e-05,
      "step": 11338,
      "training_step_time": 8.455694198608398
    },
    {
      "epoch": 1.0813713073730469e-05,
      "model_forward_time": 1.7416675090789795,
      "step": 11339
    },
    {
      "epoch": 1.0813713073730469e-05,
      "step": 11339,
      "training_step_time": 9.432103395462036
    },
    {
      "epoch": 1.0814666748046875e-05,
      "grad_norm": 0.2123555988073349,
      "learning_rate": 9.890543111564909e-06,
      "loss": 0.0671,
      "step": 11340
    },
    {
      "epoch": 1.0814666748046875e-05,
      "model_forward_time": 1.7470176219940186,
      "step": 11340
    },
    {
      "dynamics_loss_avg": 0.05899274386465549,
      "epoch": 1.0814666748046875e-05,
      "step": 11340
    },
    {
      "action_loss_avg": 0.004054419044405222,
      "epoch": 1.0814666748046875e-05,
      "step": 11340
    },
    {
      "epoch": 1.0814666748046875e-05,
      "step": 11340,
      "training_step_time": 9.807605743408203
    },
    {
      "epoch": 1.081562042236328e-05,
      "model_forward_time": 2.9192893505096436,
      "step": 11341
    },
    {
      "epoch": 1.081562042236328e-05,
      "step": 11341,
      "training_step_time": 7.756804943084717
    },
    {
      "epoch": 1.0816574096679688e-05,
      "model_forward_time": 1.73317551612854,
      "step": 11342
    },
    {
      "epoch": 1.0816574096679688e-05,
      "step": 11342,
      "training_step_time": 7.599061489105225
    },
    {
      "epoch": 1.0817527770996094e-05,
      "model_forward_time": 1.7395806312561035,
      "step": 11343
    },
    {
      "epoch": 1.0817527770996094e-05,
      "step": 11343,
      "training_step_time": 7.408552169799805
    },
    {
      "epoch": 1.08184814453125e-05,
      "model_forward_time": 1.7199506759643555,
      "step": 11344
    },
    {
      "epoch": 1.08184814453125e-05,
      "step": 11344,
      "training_step_time": 7.41385555267334
    },
    {
      "epoch": 1.0819435119628907e-05,
      "model_forward_time": 1.730006456375122,
      "step": 11345
    },
    {
      "epoch": 1.0819435119628907e-05,
      "step": 11345,
      "training_step_time": 7.695577621459961
    },
    {
      "epoch": 1.0820388793945313e-05,
      "model_forward_time": 1.7287988662719727,
      "step": 11346
    },
    {
      "epoch": 1.0820388793945313e-05,
      "step": 11346,
      "training_step_time": 8.24867558479309
    },
    {
      "epoch": 1.0821342468261718e-05,
      "model_forward_time": 1.7275419235229492,
      "step": 11347
    },
    {
      "epoch": 1.0821342468261718e-05,
      "step": 11347,
      "training_step_time": 8.680674076080322
    },
    {
      "epoch": 1.0822296142578126e-05,
      "model_forward_time": 1.7241008281707764,
      "step": 11348
    },
    {
      "epoch": 1.0822296142578126e-05,
      "step": 11348,
      "training_step_time": 9.195436477661133
    },
    {
      "epoch": 1.0823249816894531e-05,
      "model_forward_time": 1.74599289894104,
      "step": 11349
    },
    {
      "epoch": 1.0823249816894531e-05,
      "step": 11349,
      "training_step_time": 9.07734203338623
    },
    {
      "epoch": 1.0824203491210937e-05,
      "grad_norm": 0.3143664300441742,
      "learning_rate": 9.890198765382358e-06,
      "loss": 0.0649,
      "step": 11350
    },
    {
      "epoch": 1.0824203491210937e-05,
      "model_forward_time": 2.953787088394165,
      "step": 11350
    },
    {
      "dynamics_loss_avg": 0.04552391413599253,
      "epoch": 1.0824203491210937e-05,
      "step": 11350
    },
    {
      "action_loss_avg": 0.003359198966063559,
      "epoch": 1.0824203491210937e-05,
      "step": 11350
    },
    {
      "epoch": 1.0824203491210937e-05,
      "step": 11350,
      "training_step_time": 9.263763189315796
    },
    {
      "epoch": 1.0825157165527345e-05,
      "model_forward_time": 1.7539572715759277,
      "step": 11351
    },
    {
      "epoch": 1.0825157165527345e-05,
      "step": 11351,
      "training_step_time": 8.645499229431152
    },
    {
      "epoch": 1.082611083984375e-05,
      "model_forward_time": 1.7394635677337646,
      "step": 11352
    },
    {
      "epoch": 1.082611083984375e-05,
      "step": 11352,
      "training_step_time": 6.869051456451416
    },
    {
      "epoch": 1.0827064514160156e-05,
      "model_forward_time": 1.7360098361968994,
      "step": 11353
    },
    {
      "epoch": 1.0827064514160156e-05,
      "step": 11353,
      "training_step_time": 7.566606521606445
    },
    {
      "epoch": 1.0828018188476562e-05,
      "model_forward_time": 1.7294318675994873,
      "step": 11354
    },
    {
      "epoch": 1.0828018188476562e-05,
      "step": 11354,
      "training_step_time": 8.54619812965393
    },
    {
      "epoch": 1.082897186279297e-05,
      "model_forward_time": 1.7405567169189453,
      "step": 11355
    },
    {
      "epoch": 1.082897186279297e-05,
      "step": 11355,
      "training_step_time": 7.657167911529541
    },
    {
      "epoch": 1.0829925537109375e-05,
      "model_forward_time": 1.8348884582519531,
      "step": 11356
    },
    {
      "epoch": 1.0829925537109375e-05,
      "step": 11356,
      "training_step_time": 7.667002201080322
    },
    {
      "epoch": 1.083087921142578e-05,
      "model_forward_time": 1.734525442123413,
      "step": 11357
    },
    {
      "epoch": 1.083087921142578e-05,
      "step": 11357,
      "training_step_time": 9.52759313583374
    },
    {
      "epoch": 1.0831832885742188e-05,
      "model_forward_time": 1.732008457183838,
      "step": 11358
    },
    {
      "epoch": 1.0831832885742188e-05,
      "step": 11358,
      "training_step_time": 8.305399656295776
    },
    {
      "epoch": 1.0832786560058594e-05,
      "model_forward_time": 2.8664615154266357,
      "step": 11359
    },
    {
      "epoch": 1.0832786560058594e-05,
      "step": 11359,
      "training_step_time": 8.730859756469727
    },
    {
      "epoch": 1.0833740234375e-05,
      "grad_norm": 0.1746089905500412,
      "learning_rate": 9.889853884415022e-06,
      "loss": 0.0678,
      "step": 11360
    },
    {
      "epoch": 1.0833740234375e-05,
      "model_forward_time": 1.3749849796295166,
      "step": 11360
    },
    {
      "dynamics_loss_avg": 0.06150259543210268,
      "epoch": 1.0833740234375e-05,
      "step": 11360
    },
    {
      "action_loss_avg": 0.0030211300356313585,
      "epoch": 1.0833740234375e-05,
      "step": 11360
    },
    {
      "epoch": 1.0833740234375e-05,
      "step": 11360,
      "training_step_time": 8.34290361404419
    },
    {
      "epoch": 1.0834693908691407e-05,
      "model_forward_time": 1.022599220275879,
      "step": 11361
    },
    {
      "epoch": 1.0834693908691407e-05,
      "step": 11361,
      "training_step_time": 7.6699934005737305
    },
    {
      "epoch": 1.0835647583007813e-05,
      "model_forward_time": 1.7393536567687988,
      "step": 11362
    },
    {
      "epoch": 1.0835647583007813e-05,
      "step": 11362,
      "training_step_time": 7.568361520767212
    },
    {
      "epoch": 1.0836601257324218e-05,
      "model_forward_time": 1.3752403259277344,
      "step": 11363
    },
    {
      "epoch": 1.0836601257324218e-05,
      "step": 11363,
      "training_step_time": 7.406900644302368
    },
    {
      "epoch": 1.0837554931640626e-05,
      "model_forward_time": 1.735879898071289,
      "step": 11364
    },
    {
      "epoch": 1.0837554931640626e-05,
      "step": 11364,
      "training_step_time": 7.686589002609253
    },
    {
      "epoch": 1.0838508605957031e-05,
      "model_forward_time": 1.8038859367370605,
      "step": 11365
    },
    {
      "epoch": 1.0838508605957031e-05,
      "step": 11365,
      "training_step_time": 7.624277114868164
    },
    {
      "epoch": 1.0839462280273437e-05,
      "model_forward_time": 1.746001958847046,
      "step": 11366
    },
    {
      "epoch": 1.0839462280273437e-05,
      "step": 11366,
      "training_step_time": 8.37608814239502
    },
    {
      "epoch": 1.0840415954589845e-05,
      "model_forward_time": 1.7369980812072754,
      "step": 11367
    },
    {
      "epoch": 1.0840415954589845e-05,
      "step": 11367,
      "training_step_time": 7.903587341308594
    },
    {
      "epoch": 1.084136962890625e-05,
      "model_forward_time": 2.8958041667938232,
      "step": 11368
    },
    {
      "epoch": 1.084136962890625e-05,
      "step": 11368,
      "training_step_time": 8.543993473052979
    },
    {
      "epoch": 1.0842323303222656e-05,
      "model_forward_time": 1.7472827434539795,
      "step": 11369
    },
    {
      "epoch": 1.0842323303222656e-05,
      "step": 11369,
      "training_step_time": 7.682008504867554
    },
    {
      "epoch": 1.0843276977539062e-05,
      "grad_norm": 0.14838585257530212,
      "learning_rate": 9.889508468700615e-06,
      "loss": 0.0679,
      "step": 11370
    },
    {
      "epoch": 1.0843276977539062e-05,
      "model_forward_time": 1.7409205436706543,
      "step": 11370
    },
    {
      "dynamics_loss_avg": 0.06680312231183053,
      "epoch": 1.0843276977539062e-05,
      "step": 11370
    },
    {
      "action_loss_avg": 0.003661000507418066,
      "epoch": 1.0843276977539062e-05,
      "step": 11370
    },
    {
      "epoch": 1.0843276977539062e-05,
      "step": 11370,
      "training_step_time": 7.6971657276153564
    },
    {
      "epoch": 1.084423065185547e-05,
      "model_forward_time": 1.737243413925171,
      "step": 11371
    },
    {
      "epoch": 1.084423065185547e-05,
      "step": 11371,
      "training_step_time": 7.696369409561157
    },
    {
      "epoch": 1.0845184326171875e-05,
      "model_forward_time": 1.7360820770263672,
      "step": 11372
    },
    {
      "epoch": 1.0845184326171875e-05,
      "step": 11372,
      "training_step_time": 7.506831884384155
    },
    {
      "epoch": 1.084613800048828e-05,
      "model_forward_time": 1.755117416381836,
      "step": 11373
    },
    {
      "epoch": 1.084613800048828e-05,
      "step": 11373,
      "training_step_time": 8.345064401626587
    },
    {
      "epoch": 1.0847091674804688e-05,
      "model_forward_time": 1.8097765445709229,
      "step": 11374
    },
    {
      "epoch": 1.0847091674804688e-05,
      "step": 11374,
      "training_step_time": 7.679932594299316
    },
    {
      "epoch": 1.0848045349121094e-05,
      "model_forward_time": 1.718496561050415,
      "step": 11375
    },
    {
      "epoch": 1.0848045349121094e-05,
      "step": 11375,
      "training_step_time": 7.85277795791626
    },
    {
      "epoch": 1.08489990234375e-05,
      "model_forward_time": 1.734884262084961,
      "step": 11376
    },
    {
      "epoch": 1.08489990234375e-05,
      "step": 11376,
      "training_step_time": 9.415449857711792
    },
    {
      "epoch": 1.0849952697753907e-05,
      "model_forward_time": 1.7294976711273193,
      "step": 11377
    },
    {
      "epoch": 1.0849952697753907e-05,
      "step": 11377,
      "training_step_time": 7.6926939487457275
    },
    {
      "epoch": 1.0850906372070313e-05,
      "model_forward_time": 2.854124069213867,
      "step": 11378
    },
    {
      "epoch": 1.0850906372070313e-05,
      "step": 11378,
      "training_step_time": 8.756826162338257
    },
    {
      "epoch": 1.0851860046386718e-05,
      "model_forward_time": 1.7409842014312744,
      "step": 11379
    },
    {
      "epoch": 1.0851860046386718e-05,
      "step": 11379,
      "training_step_time": 7.614500045776367
    },
    {
      "epoch": 1.0852813720703126e-05,
      "grad_norm": 0.18900060653686523,
      "learning_rate": 9.889162518276915e-06,
      "loss": 0.068,
      "step": 11380
    },
    {
      "epoch": 1.0852813720703126e-05,
      "model_forward_time": 1.7199368476867676,
      "step": 11380
    },
    {
      "dynamics_loss_avg": 0.04576627379283309,
      "epoch": 1.0852813720703126e-05,
      "step": 11380
    },
    {
      "action_loss_avg": 0.0037356949993409216,
      "epoch": 1.0852813720703126e-05,
      "step": 11380
    },
    {
      "epoch": 1.0852813720703126e-05,
      "step": 11380,
      "training_step_time": 7.624652862548828
    },
    {
      "epoch": 1.0853767395019531e-05,
      "model_forward_time": 1.7295210361480713,
      "step": 11381
    },
    {
      "epoch": 1.0853767395019531e-05,
      "step": 11381,
      "training_step_time": 7.511408567428589
    },
    {
      "epoch": 1.0854721069335937e-05,
      "model_forward_time": 1.7370364665985107,
      "step": 11382
    },
    {
      "epoch": 1.0854721069335937e-05,
      "step": 11382,
      "training_step_time": 7.803254842758179
    },
    {
      "epoch": 1.0855674743652343e-05,
      "model_forward_time": 1.8017606735229492,
      "step": 11383
    },
    {
      "epoch": 1.0855674743652343e-05,
      "step": 11383,
      "training_step_time": 8.634921312332153
    },
    {
      "epoch": 1.085662841796875e-05,
      "model_forward_time": 1.722193956375122,
      "step": 11384
    },
    {
      "epoch": 1.085662841796875e-05,
      "step": 11384,
      "training_step_time": 8.61335301399231
    },
    {
      "epoch": 1.0857582092285156e-05,
      "model_forward_time": 1.7331171035766602,
      "step": 11385
    },
    {
      "epoch": 1.0857582092285156e-05,
      "step": 11385,
      "training_step_time": 8.714343309402466
    },
    {
      "epoch": 1.0858535766601562e-05,
      "model_forward_time": 1.721616268157959,
      "step": 11386
    },
    {
      "epoch": 1.0858535766601562e-05,
      "step": 11386,
      "training_step_time": 7.847519159317017
    },
    {
      "epoch": 1.085948944091797e-05,
      "model_forward_time": 2.887167453765869,
      "step": 11387
    },
    {
      "epoch": 1.085948944091797e-05,
      "step": 11387,
      "training_step_time": 8.616204023361206
    },
    {
      "epoch": 1.0860443115234375e-05,
      "model_forward_time": 1.7275314331054688,
      "step": 11388
    },
    {
      "epoch": 1.0860443115234375e-05,
      "step": 11388,
      "training_step_time": 7.644893646240234
    },
    {
      "epoch": 1.086139678955078e-05,
      "model_forward_time": 1.7311339378356934,
      "step": 11389
    },
    {
      "epoch": 1.086139678955078e-05,
      "step": 11389,
      "training_step_time": 7.700602293014526
    },
    {
      "epoch": 1.0862350463867188e-05,
      "grad_norm": 0.15986913442611694,
      "learning_rate": 9.888816033181751e-06,
      "loss": 0.065,
      "step": 11390
    },
    {
      "epoch": 1.0862350463867188e-05,
      "model_forward_time": 1.7279458045959473,
      "step": 11390
    },
    {
      "dynamics_loss_avg": 0.05958389155566692,
      "epoch": 1.0862350463867188e-05,
      "step": 11390
    },
    {
      "action_loss_avg": 0.002578311605611816,
      "epoch": 1.0862350463867188e-05,
      "step": 11390
    },
    {
      "epoch": 1.0862350463867188e-05,
      "step": 11390,
      "training_step_time": 7.514099597930908
    },
    {
      "epoch": 1.0863304138183594e-05,
      "model_forward_time": 1.726318120956421,
      "step": 11391
    },
    {
      "epoch": 1.0863304138183594e-05,
      "step": 11391,
      "training_step_time": 8.28029203414917
    },
    {
      "epoch": 1.08642578125e-05,
      "model_forward_time": 1.770878791809082,
      "step": 11392
    },
    {
      "epoch": 1.08642578125e-05,
      "step": 11392,
      "training_step_time": 7.4765520095825195
    },
    {
      "epoch": 1.0865211486816407e-05,
      "model_forward_time": 1.7256724834442139,
      "step": 11393
    },
    {
      "epoch": 1.0865211486816407e-05,
      "step": 11393,
      "training_step_time": 8.296082973480225
    },
    {
      "epoch": 1.0866165161132813e-05,
      "model_forward_time": 1.7279231548309326,
      "step": 11394
    },
    {
      "epoch": 1.0866165161132813e-05,
      "step": 11394,
      "training_step_time": 9.326014995574951
    },
    {
      "epoch": 1.0867118835449218e-05,
      "model_forward_time": 1.7268028259277344,
      "step": 11395
    },
    {
      "epoch": 1.0867118835449218e-05,
      "step": 11395,
      "training_step_time": 8.619290828704834
    },
    {
      "epoch": 1.0868072509765626e-05,
      "model_forward_time": 2.885366678237915,
      "step": 11396
    },
    {
      "epoch": 1.0868072509765626e-05,
      "step": 11396,
      "training_step_time": 8.581820011138916
    },
    {
      "epoch": 1.0869026184082031e-05,
      "model_forward_time": 1.7316532135009766,
      "step": 11397
    },
    {
      "epoch": 1.0869026184082031e-05,
      "step": 11397,
      "training_step_time": 7.731265544891357
    },
    {
      "epoch": 1.0869979858398437e-05,
      "model_forward_time": 1.7285149097442627,
      "step": 11398
    },
    {
      "epoch": 1.0869979858398437e-05,
      "step": 11398,
      "training_step_time": 7.674553155899048
    },
    {
      "epoch": 1.0870933532714843e-05,
      "model_forward_time": 1.726109504699707,
      "step": 11399
    },
    {
      "epoch": 1.0870933532714843e-05,
      "step": 11399,
      "training_step_time": 7.2438249588012695
    },
    {
      "epoch": 1.087188720703125e-05,
      "grad_norm": 0.1780836135149002,
      "learning_rate": 9.888469013453018e-06,
      "loss": 0.0715,
      "step": 11400
    },
    {
      "epoch": 1.087188720703125e-05,
      "model_forward_time": 1.7350103855133057,
      "step": 11400
    },
    {
      "dynamics_loss_avg": 0.08801226690411568,
      "epoch": 1.087188720703125e-05,
      "step": 11400
    },
    {
      "action_loss_avg": 0.00550442905514501,
      "epoch": 1.087188720703125e-05,
      "step": 11400
    },
    {
      "epoch": 1.087188720703125e-05,
      "step": 11400,
      "training_step_time": 7.752515077590942
    },
    {
      "epoch": 1.0872840881347656e-05,
      "model_forward_time": 1.7577998638153076,
      "step": 11401
    },
    {
      "epoch": 1.0872840881347656e-05,
      "step": 11401,
      "training_step_time": 7.479130506515503
    },
    {
      "epoch": 1.0873794555664062e-05,
      "model_forward_time": 1.7280843257904053,
      "step": 11402
    },
    {
      "epoch": 1.0873794555664062e-05,
      "step": 11402,
      "training_step_time": 7.677284240722656
    },
    {
      "epoch": 1.087474822998047e-05,
      "model_forward_time": 1.730818748474121,
      "step": 11403
    },
    {
      "epoch": 1.087474822998047e-05,
      "step": 11403,
      "training_step_time": 9.065041542053223
    },
    {
      "epoch": 1.0875701904296875e-05,
      "model_forward_time": 1.7263169288635254,
      "step": 11404
    },
    {
      "epoch": 1.0875701904296875e-05,
      "step": 11404,
      "training_step_time": 8.374659538269043
    },
    {
      "epoch": 1.087665557861328e-05,
      "model_forward_time": 2.941649913787842,
      "step": 11405
    },
    {
      "epoch": 1.087665557861328e-05,
      "step": 11405,
      "training_step_time": 10.085217475891113
    },
    {
      "epoch": 1.0877609252929688e-05,
      "model_forward_time": 1.7235016822814941,
      "step": 11406
    },
    {
      "epoch": 1.0877609252929688e-05,
      "step": 11406,
      "training_step_time": 7.694140911102295
    },
    {
      "epoch": 1.0878562927246094e-05,
      "model_forward_time": 1.7359912395477295,
      "step": 11407
    },
    {
      "epoch": 1.0878562927246094e-05,
      "step": 11407,
      "training_step_time": 7.635472536087036
    },
    {
      "epoch": 1.08795166015625e-05,
      "model_forward_time": 1.7305102348327637,
      "step": 11408
    },
    {
      "epoch": 1.08795166015625e-05,
      "step": 11408,
      "training_step_time": 7.751367568969727
    },
    {
      "epoch": 1.0880470275878907e-05,
      "model_forward_time": 1.7308299541473389,
      "step": 11409
    },
    {
      "epoch": 1.0880470275878907e-05,
      "step": 11409,
      "training_step_time": 7.5718255043029785
    },
    {
      "epoch": 1.0881423950195313e-05,
      "grad_norm": 0.22016355395317078,
      "learning_rate": 9.888121459128663e-06,
      "loss": 0.0668,
      "step": 11410
    },
    {
      "epoch": 1.0881423950195313e-05,
      "model_forward_time": 1.3702187538146973,
      "step": 11410
    },
    {
      "dynamics_loss_avg": 0.07187417633831501,
      "epoch": 1.0881423950195313e-05,
      "step": 11410
    },
    {
      "action_loss_avg": 0.0037390208104625343,
      "epoch": 1.0881423950195313e-05,
      "step": 11410
    },
    {
      "epoch": 1.0881423950195313e-05,
      "step": 11410,
      "training_step_time": 7.739830017089844
    },
    {
      "epoch": 1.0882377624511718e-05,
      "model_forward_time": 1.7372100353240967,
      "step": 11411
    },
    {
      "epoch": 1.0882377624511718e-05,
      "step": 11411,
      "training_step_time": 8.65215802192688
    },
    {
      "epoch": 1.0883331298828126e-05,
      "model_forward_time": 1.7268486022949219,
      "step": 11412
    },
    {
      "epoch": 1.0883331298828126e-05,
      "step": 11412,
      "training_step_time": 7.806277513504028
    },
    {
      "epoch": 1.0884284973144531e-05,
      "model_forward_time": 1.7283976078033447,
      "step": 11413
    },
    {
      "epoch": 1.0884284973144531e-05,
      "step": 11413,
      "training_step_time": 9.443814516067505
    },
    {
      "epoch": 1.0885238647460937e-05,
      "model_forward_time": 2.8912160396575928,
      "step": 11414
    },
    {
      "epoch": 1.0885238647460937e-05,
      "step": 11414,
      "training_step_time": 7.7192583084106445
    },
    {
      "epoch": 1.0886192321777343e-05,
      "model_forward_time": 1.742159366607666,
      "step": 11415
    },
    {
      "epoch": 1.0886192321777343e-05,
      "step": 11415,
      "training_step_time": 8.548647165298462
    },
    {
      "epoch": 1.088714599609375e-05,
      "model_forward_time": 1.733881950378418,
      "step": 11416
    },
    {
      "epoch": 1.088714599609375e-05,
      "step": 11416,
      "training_step_time": 7.655678033828735
    },
    {
      "epoch": 1.0888099670410156e-05,
      "model_forward_time": 1.72548246383667,
      "step": 11417
    },
    {
      "epoch": 1.0888099670410156e-05,
      "step": 11417,
      "training_step_time": 7.740848541259766
    },
    {
      "epoch": 1.0889053344726562e-05,
      "model_forward_time": 1.7309281826019287,
      "step": 11418
    },
    {
      "epoch": 1.0889053344726562e-05,
      "step": 11418,
      "training_step_time": 7.753840684890747
    },
    {
      "epoch": 1.089000701904297e-05,
      "model_forward_time": 1.7265949249267578,
      "step": 11419
    },
    {
      "epoch": 1.089000701904297e-05,
      "step": 11419,
      "training_step_time": 7.728281736373901
    },
    {
      "epoch": 1.0890960693359375e-05,
      "grad_norm": 0.3007012903690338,
      "learning_rate": 9.887773370246693e-06,
      "loss": 0.0702,
      "step": 11420
    },
    {
      "epoch": 1.0890960693359375e-05,
      "model_forward_time": 1.8029975891113281,
      "step": 11420
    },
    {
      "dynamics_loss_avg": 0.0709763839840889,
      "epoch": 1.0890960693359375e-05,
      "step": 11420
    },
    {
      "action_loss_avg": 0.0034181974770035595,
      "epoch": 1.0890960693359375e-05,
      "step": 11420
    },
    {
      "epoch": 1.0890960693359375e-05,
      "step": 11420,
      "training_step_time": 8.839581489562988
    },
    {
      "epoch": 1.089191436767578e-05,
      "model_forward_time": 1.7298674583435059,
      "step": 11421
    },
    {
      "epoch": 1.089191436767578e-05,
      "step": 11421,
      "training_step_time": 7.8579161167144775
    },
    {
      "epoch": 1.0892868041992188e-05,
      "model_forward_time": 1.7308812141418457,
      "step": 11422
    },
    {
      "epoch": 1.0892868041992188e-05,
      "step": 11422,
      "training_step_time": 7.8290696144104
    },
    {
      "epoch": 1.0893821716308594e-05,
      "model_forward_time": 2.888533353805542,
      "step": 11423
    },
    {
      "epoch": 1.0893821716308594e-05,
      "step": 11423,
      "training_step_time": 8.741406440734863
    },
    {
      "epoch": 1.0894775390625e-05,
      "model_forward_time": 1.7307472229003906,
      "step": 11424
    },
    {
      "epoch": 1.0894775390625e-05,
      "step": 11424,
      "training_step_time": 8.788442611694336
    },
    {
      "epoch": 1.0895729064941407e-05,
      "model_forward_time": 1.7359211444854736,
      "step": 11425
    },
    {
      "epoch": 1.0895729064941407e-05,
      "step": 11425,
      "training_step_time": 8.636572122573853
    },
    {
      "epoch": 1.0896682739257813e-05,
      "model_forward_time": 1.7275869846343994,
      "step": 11426
    },
    {
      "epoch": 1.0896682739257813e-05,
      "step": 11426,
      "training_step_time": 7.684665679931641
    },
    {
      "epoch": 1.0897636413574218e-05,
      "model_forward_time": 1.7301051616668701,
      "step": 11427
    },
    {
      "epoch": 1.0897636413574218e-05,
      "step": 11427,
      "training_step_time": 8.58500862121582
    },
    {
      "epoch": 1.0898590087890626e-05,
      "model_forward_time": 1.7297897338867188,
      "step": 11428
    },
    {
      "epoch": 1.0898590087890626e-05,
      "step": 11428,
      "training_step_time": 8.352961778640747
    },
    {
      "epoch": 1.0899543762207032e-05,
      "model_forward_time": 1.8060202598571777,
      "step": 11429
    },
    {
      "epoch": 1.0899543762207032e-05,
      "step": 11429,
      "training_step_time": 7.724326133728027
    },
    {
      "epoch": 1.0900497436523437e-05,
      "grad_norm": 0.23589417338371277,
      "learning_rate": 9.887424746845177e-06,
      "loss": 0.0642,
      "step": 11430
    },
    {
      "epoch": 1.0900497436523437e-05,
      "model_forward_time": 1.737898588180542,
      "step": 11430
    },
    {
      "dynamics_loss_avg": 0.05102362707257271,
      "epoch": 1.0900497436523437e-05,
      "step": 11430
    },
    {
      "action_loss_avg": 0.0027321477187797426,
      "epoch": 1.0900497436523437e-05,
      "step": 11430
    },
    {
      "epoch": 1.0900497436523437e-05,
      "step": 11430,
      "training_step_time": 7.799867868423462
    },
    {
      "epoch": 1.0901451110839843e-05,
      "model_forward_time": 1.7340633869171143,
      "step": 11431
    },
    {
      "epoch": 1.0901451110839843e-05,
      "step": 11431,
      "training_step_time": 8.564069271087646
    },
    {
      "epoch": 1.090240478515625e-05,
      "model_forward_time": 2.935080051422119,
      "step": 11432
    },
    {
      "epoch": 1.090240478515625e-05,
      "step": 11432,
      "training_step_time": 8.38256287574768
    },
    {
      "epoch": 1.0903358459472656e-05,
      "model_forward_time": 1.728342056274414,
      "step": 11433
    },
    {
      "epoch": 1.0903358459472656e-05,
      "step": 11433,
      "training_step_time": 8.321256399154663
    },
    {
      "epoch": 1.0904312133789062e-05,
      "model_forward_time": 1.729757308959961,
      "step": 11434
    },
    {
      "epoch": 1.0904312133789062e-05,
      "step": 11434,
      "training_step_time": 8.5098717212677
    },
    {
      "epoch": 1.090526580810547e-05,
      "model_forward_time": 1.7318198680877686,
      "step": 11435
    },
    {
      "epoch": 1.090526580810547e-05,
      "step": 11435,
      "training_step_time": 7.554975271224976
    },
    {
      "epoch": 1.0906219482421875e-05,
      "model_forward_time": 1.7271764278411865,
      "step": 11436
    },
    {
      "epoch": 1.0906219482421875e-05,
      "step": 11436,
      "training_step_time": 8.823367595672607
    },
    {
      "epoch": 1.090717315673828e-05,
      "model_forward_time": 1.741096019744873,
      "step": 11437
    },
    {
      "epoch": 1.090717315673828e-05,
      "step": 11437,
      "training_step_time": 7.7232184410095215
    },
    {
      "epoch": 1.0908126831054688e-05,
      "model_forward_time": 1.8027753829956055,
      "step": 11438
    },
    {
      "epoch": 1.0908126831054688e-05,
      "step": 11438,
      "training_step_time": 7.793935775756836
    },
    {
      "epoch": 1.0909080505371094e-05,
      "model_forward_time": 1.7327210903167725,
      "step": 11439
    },
    {
      "epoch": 1.0909080505371094e-05,
      "step": 11439,
      "training_step_time": 8.571071147918701
    },
    {
      "epoch": 1.09100341796875e-05,
      "grad_norm": 0.1749619096517563,
      "learning_rate": 9.887075588962238e-06,
      "loss": 0.0714,
      "step": 11440
    },
    {
      "epoch": 1.09100341796875e-05,
      "model_forward_time": 1.7518000602722168,
      "step": 11440
    },
    {
      "dynamics_loss_avg": 0.0736074060201645,
      "epoch": 1.09100341796875e-05,
      "step": 11440
    },
    {
      "action_loss_avg": 0.0044697806471958755,
      "epoch": 1.09100341796875e-05,
      "step": 11440
    },
    {
      "epoch": 1.09100341796875e-05,
      "step": 11440,
      "training_step_time": 8.879154443740845
    },
    {
      "epoch": 1.0910987854003907e-05,
      "model_forward_time": 2.8467764854431152,
      "step": 11441
    },
    {
      "epoch": 1.0910987854003907e-05,
      "step": 11441,
      "training_step_time": 7.896313905715942
    },
    {
      "epoch": 1.0911941528320313e-05,
      "model_forward_time": 1.7317922115325928,
      "step": 11442
    },
    {
      "epoch": 1.0911941528320313e-05,
      "step": 11442,
      "training_step_time": 7.920673131942749
    },
    {
      "epoch": 1.0912895202636718e-05,
      "model_forward_time": 1.7331509590148926,
      "step": 11443
    },
    {
      "epoch": 1.0912895202636718e-05,
      "step": 11443,
      "training_step_time": 8.61339545249939
    },
    {
      "epoch": 1.0913848876953126e-05,
      "model_forward_time": 1.724125623703003,
      "step": 11444
    },
    {
      "epoch": 1.0913848876953126e-05,
      "step": 11444,
      "training_step_time": 7.290653467178345
    },
    {
      "epoch": 1.0914802551269532e-05,
      "model_forward_time": 1.7433483600616455,
      "step": 11445
    },
    {
      "epoch": 1.0914802551269532e-05,
      "step": 11445,
      "training_step_time": 7.701406002044678
    },
    {
      "epoch": 1.0915756225585937e-05,
      "model_forward_time": 1.73508620262146,
      "step": 11446
    },
    {
      "epoch": 1.0915756225585937e-05,
      "step": 11446,
      "training_step_time": 7.760529279708862
    },
    {
      "epoch": 1.0916709899902343e-05,
      "model_forward_time": 1.784048318862915,
      "step": 11447
    },
    {
      "epoch": 1.0916709899902343e-05,
      "step": 11447,
      "training_step_time": 9.437385082244873
    },
    {
      "epoch": 1.091766357421875e-05,
      "model_forward_time": 1.7352752685546875,
      "step": 11448
    },
    {
      "epoch": 1.091766357421875e-05,
      "step": 11448,
      "training_step_time": 8.303434371948242
    },
    {
      "epoch": 1.0918617248535156e-05,
      "model_forward_time": 1.7377169132232666,
      "step": 11449
    },
    {
      "epoch": 1.0918617248535156e-05,
      "step": 11449,
      "training_step_time": 7.783857583999634
    },
    {
      "epoch": 1.0919570922851562e-05,
      "grad_norm": 0.2017742395401001,
      "learning_rate": 9.88672589663606e-06,
      "loss": 0.0716,
      "step": 11450
    },
    {
      "epoch": 1.0919570922851562e-05,
      "model_forward_time": 1.7391314506530762,
      "step": 11450
    },
    {
      "dynamics_loss_avg": 0.0551283236593008,
      "epoch": 1.0919570922851562e-05,
      "step": 11450
    },
    {
      "action_loss_avg": 0.0028892529429867865,
      "epoch": 1.0919570922851562e-05,
      "step": 11450
    },
    {
      "epoch": 1.0919570922851562e-05,
      "step": 11450,
      "training_step_time": 8.881962299346924
    },
    {
      "epoch": 1.092052459716797e-05,
      "model_forward_time": 1.7318928241729736,
      "step": 11451
    },
    {
      "epoch": 1.092052459716797e-05,
      "step": 11451,
      "training_step_time": 7.796785831451416
    },
    {
      "epoch": 1.0921478271484375e-05,
      "model_forward_time": 1.7353696823120117,
      "step": 11452
    },
    {
      "epoch": 1.0921478271484375e-05,
      "step": 11452,
      "training_step_time": 7.840448379516602
    },
    {
      "epoch": 1.092243194580078e-05,
      "model_forward_time": 1.7678592205047607,
      "step": 11453
    },
    {
      "epoch": 1.092243194580078e-05,
      "step": 11453,
      "training_step_time": 7.534622430801392
    },
    {
      "epoch": 1.0923385620117188e-05,
      "model_forward_time": 1.730893611907959,
      "step": 11454
    },
    {
      "epoch": 1.0923385620117188e-05,
      "step": 11454,
      "training_step_time": 7.6794726848602295
    },
    {
      "epoch": 1.0924339294433594e-05,
      "model_forward_time": 1.7368652820587158,
      "step": 11455
    },
    {
      "epoch": 1.0924339294433594e-05,
      "step": 11455,
      "training_step_time": 7.717520713806152
    },
    {
      "epoch": 1.092529296875e-05,
      "model_forward_time": 1.7491145133972168,
      "step": 11456
    },
    {
      "epoch": 1.092529296875e-05,
      "step": 11456,
      "training_step_time": 7.666936874389648
    },
    {
      "epoch": 1.0926246643066407e-05,
      "model_forward_time": 1.7357697486877441,
      "step": 11457
    },
    {
      "epoch": 1.0926246643066407e-05,
      "step": 11457,
      "training_step_time": 8.359549522399902
    },
    {
      "epoch": 1.0927200317382813e-05,
      "model_forward_time": 1.7376675605773926,
      "step": 11458
    },
    {
      "epoch": 1.0927200317382813e-05,
      "step": 11458,
      "training_step_time": 7.866153717041016
    },
    {
      "epoch": 1.0928153991699218e-05,
      "model_forward_time": 1.7380919456481934,
      "step": 11459
    },
    {
      "epoch": 1.0928153991699218e-05,
      "step": 11459,
      "training_step_time": 8.881656169891357
    },
    {
      "epoch": 1.0929107666015626e-05,
      "grad_norm": 0.20859526097774506,
      "learning_rate": 9.886375669904886e-06,
      "loss": 0.0735,
      "step": 11460
    },
    {
      "epoch": 1.0929107666015626e-05,
      "model_forward_time": 2.886594533920288,
      "step": 11460
    },
    {
      "dynamics_loss_avg": 0.07601339556276798,
      "epoch": 1.0929107666015626e-05,
      "step": 11460
    },
    {
      "action_loss_avg": 0.004256119125057012,
      "epoch": 1.0929107666015626e-05,
      "step": 11460
    },
    {
      "epoch": 1.0929107666015626e-05,
      "step": 11460,
      "training_step_time": 7.807096719741821
    },
    {
      "epoch": 1.0930061340332032e-05,
      "model_forward_time": 1.7414851188659668,
      "step": 11461
    },
    {
      "epoch": 1.0930061340332032e-05,
      "step": 11461,
      "training_step_time": 8.790592432022095
    },
    {
      "epoch": 1.0931015014648437e-05,
      "model_forward_time": 1.7213833332061768,
      "step": 11462
    },
    {
      "epoch": 1.0931015014648437e-05,
      "step": 11462,
      "training_step_time": 7.512672185897827
    },
    {
      "epoch": 1.0931968688964843e-05,
      "model_forward_time": 1.7450764179229736,
      "step": 11463
    },
    {
      "epoch": 1.0931968688964843e-05,
      "step": 11463,
      "training_step_time": 7.34592604637146
    },
    {
      "epoch": 1.093292236328125e-05,
      "model_forward_time": 1.7469663619995117,
      "step": 11464
    },
    {
      "epoch": 1.093292236328125e-05,
      "step": 11464,
      "training_step_time": 7.72963809967041
    },
    {
      "epoch": 1.0933876037597656e-05,
      "model_forward_time": 1.732541561126709,
      "step": 11465
    },
    {
      "epoch": 1.0933876037597656e-05,
      "step": 11465,
      "training_step_time": 7.654659271240234
    },
    {
      "epoch": 1.0934829711914062e-05,
      "model_forward_time": 1.8154895305633545,
      "step": 11466
    },
    {
      "epoch": 1.0934829711914062e-05,
      "step": 11466,
      "training_step_time": 7.696408271789551
    },
    {
      "epoch": 1.093578338623047e-05,
      "model_forward_time": 1.7331206798553467,
      "step": 11467
    },
    {
      "epoch": 1.093578338623047e-05,
      "step": 11467,
      "training_step_time": 8.485667705535889
    },
    {
      "epoch": 1.0936737060546875e-05,
      "model_forward_time": 1.7316138744354248,
      "step": 11468
    },
    {
      "epoch": 1.0936737060546875e-05,
      "step": 11468,
      "training_step_time": 9.128544807434082
    },
    {
      "epoch": 1.093769073486328e-05,
      "model_forward_time": 2.941311836242676,
      "step": 11469
    },
    {
      "epoch": 1.093769073486328e-05,
      "step": 11469,
      "training_step_time": 8.79705286026001
    },
    {
      "epoch": 1.0938644409179688e-05,
      "grad_norm": 0.21878530085086823,
      "learning_rate": 9.886024908807015e-06,
      "loss": 0.0719,
      "step": 11470
    },
    {
      "epoch": 1.0938644409179688e-05,
      "model_forward_time": 1.7422764301300049,
      "step": 11470
    },
    {
      "dynamics_loss_avg": 0.073447185754776,
      "epoch": 1.0938644409179688e-05,
      "step": 11470
    },
    {
      "action_loss_avg": 0.003430133534129709,
      "epoch": 1.0938644409179688e-05,
      "step": 11470
    },
    {
      "epoch": 1.0938644409179688e-05,
      "step": 11470,
      "training_step_time": 8.65067195892334
    },
    {
      "epoch": 1.0939598083496094e-05,
      "model_forward_time": 1.7376718521118164,
      "step": 11471
    },
    {
      "epoch": 1.0939598083496094e-05,
      "step": 11471,
      "training_step_time": 8.691053867340088
    },
    {
      "epoch": 1.09405517578125e-05,
      "model_forward_time": 1.7302162647247314,
      "step": 11472
    },
    {
      "epoch": 1.09405517578125e-05,
      "step": 11472,
      "training_step_time": 7.103495121002197
    },
    {
      "epoch": 1.0941505432128907e-05,
      "model_forward_time": 1.7378668785095215,
      "step": 11473
    },
    {
      "epoch": 1.0941505432128907e-05,
      "step": 11473,
      "training_step_time": 7.706483840942383
    },
    {
      "epoch": 1.0942459106445313e-05,
      "model_forward_time": 1.7399258613586426,
      "step": 11474
    },
    {
      "epoch": 1.0942459106445313e-05,
      "step": 11474,
      "training_step_time": 7.581219434738159
    },
    {
      "epoch": 1.0943412780761718e-05,
      "model_forward_time": 1.7509739398956299,
      "step": 11475
    },
    {
      "epoch": 1.0943412780761718e-05,
      "step": 11475,
      "training_step_time": 7.6863932609558105
    },
    {
      "epoch": 1.0944366455078126e-05,
      "model_forward_time": 1.7248260974884033,
      "step": 11476
    },
    {
      "epoch": 1.0944366455078126e-05,
      "step": 11476,
      "training_step_time": 9.69743824005127
    },
    {
      "epoch": 1.0945320129394532e-05,
      "model_forward_time": 1.391261339187622,
      "step": 11477
    },
    {
      "epoch": 1.0945320129394532e-05,
      "step": 11477,
      "training_step_time": 7.621626615524292
    },
    {
      "epoch": 1.0946273803710937e-05,
      "model_forward_time": 2.8579559326171875,
      "step": 11478
    },
    {
      "epoch": 1.0946273803710937e-05,
      "step": 11478,
      "training_step_time": 8.967748165130615
    },
    {
      "epoch": 1.0947227478027343e-05,
      "model_forward_time": 1.7301528453826904,
      "step": 11479
    },
    {
      "epoch": 1.0947227478027343e-05,
      "step": 11479,
      "training_step_time": 9.473615646362305
    },
    {
      "epoch": 1.094818115234375e-05,
      "grad_norm": 0.18909265100955963,
      "learning_rate": 9.885673613380806e-06,
      "loss": 0.0662,
      "step": 11480
    },
    {
      "epoch": 1.094818115234375e-05,
      "model_forward_time": 1.7321512699127197,
      "step": 11480
    },
    {
      "dynamics_loss_avg": 0.06436597034335137,
      "epoch": 1.094818115234375e-05,
      "step": 11480
    },
    {
      "action_loss_avg": 0.0036169037106446923,
      "epoch": 1.094818115234375e-05,
      "step": 11480
    },
    {
      "epoch": 1.094818115234375e-05,
      "step": 11480,
      "training_step_time": 8.333082914352417
    },
    {
      "epoch": 1.0949134826660156e-05,
      "model_forward_time": 1.7333965301513672,
      "step": 11481
    },
    {
      "epoch": 1.0949134826660156e-05,
      "step": 11481,
      "training_step_time": 7.541736602783203
    },
    {
      "epoch": 1.0950088500976562e-05,
      "model_forward_time": 1.731457233428955,
      "step": 11482
    },
    {
      "epoch": 1.0950088500976562e-05,
      "step": 11482,
      "training_step_time": 8.619426488876343
    },
    {
      "epoch": 1.095104217529297e-05,
      "model_forward_time": 1.7310941219329834,
      "step": 11483
    },
    {
      "epoch": 1.095104217529297e-05,
      "step": 11483,
      "training_step_time": 7.631139516830444
    },
    {
      "epoch": 1.0951995849609375e-05,
      "model_forward_time": 1.7567470073699951,
      "step": 11484
    },
    {
      "epoch": 1.0951995849609375e-05,
      "step": 11484,
      "training_step_time": 7.554995059967041
    },
    {
      "epoch": 1.095294952392578e-05,
      "model_forward_time": 1.7306692600250244,
      "step": 11485
    },
    {
      "epoch": 1.095294952392578e-05,
      "step": 11485,
      "training_step_time": 8.43706464767456
    },
    {
      "epoch": 1.0953903198242188e-05,
      "model_forward_time": 1.7658207416534424,
      "step": 11486
    },
    {
      "epoch": 1.0953903198242188e-05,
      "step": 11486,
      "training_step_time": 8.627269983291626
    },
    {
      "epoch": 1.0954856872558594e-05,
      "model_forward_time": 2.8341076374053955,
      "step": 11487
    },
    {
      "epoch": 1.0954856872558594e-05,
      "step": 11487,
      "training_step_time": 9.739272594451904
    },
    {
      "epoch": 1.0955810546875e-05,
      "model_forward_time": 1.7195000648498535,
      "step": 11488
    },
    {
      "epoch": 1.0955810546875e-05,
      "step": 11488,
      "training_step_time": 8.701746225357056
    },
    {
      "epoch": 1.0956764221191407e-05,
      "model_forward_time": 1.7338407039642334,
      "step": 11489
    },
    {
      "epoch": 1.0956764221191407e-05,
      "step": 11489,
      "training_step_time": 7.773273944854736
    },
    {
      "epoch": 1.0957717895507813e-05,
      "grad_norm": 0.2087973654270172,
      "learning_rate": 9.885321783664677e-06,
      "loss": 0.0678,
      "step": 11490
    },
    {
      "epoch": 1.0957717895507813e-05,
      "model_forward_time": 1.7292017936706543,
      "step": 11490
    },
    {
      "dynamics_loss_avg": 0.08470441065728665,
      "epoch": 1.0957717895507813e-05,
      "step": 11490
    },
    {
      "action_loss_avg": 0.0047199820633977655,
      "epoch": 1.0957717895507813e-05,
      "step": 11490
    },
    {
      "epoch": 1.0957717895507813e-05,
      "step": 11490,
      "training_step_time": 8.027377367019653
    },
    {
      "epoch": 1.0958671569824218e-05,
      "model_forward_time": 1.7299294471740723,
      "step": 11491
    },
    {
      "epoch": 1.0958671569824218e-05,
      "step": 11491,
      "training_step_time": 7.640519618988037
    },
    {
      "epoch": 1.0959625244140626e-05,
      "model_forward_time": 1.7333025932312012,
      "step": 11492
    },
    {
      "epoch": 1.0959625244140626e-05,
      "step": 11492,
      "training_step_time": 7.673377752304077
    },
    {
      "epoch": 1.0960578918457032e-05,
      "model_forward_time": 1.7628333568572998,
      "step": 11493
    },
    {
      "epoch": 1.0960578918457032e-05,
      "step": 11493,
      "training_step_time": 8.503088474273682
    },
    {
      "epoch": 1.0961532592773437e-05,
      "model_forward_time": 1.7283668518066406,
      "step": 11494
    },
    {
      "epoch": 1.0961532592773437e-05,
      "step": 11494,
      "training_step_time": 7.691124677658081
    },
    {
      "epoch": 1.0962486267089843e-05,
      "model_forward_time": 1.7670178413391113,
      "step": 11495
    },
    {
      "epoch": 1.0962486267089843e-05,
      "step": 11495,
      "training_step_time": 8.597168922424316
    },
    {
      "epoch": 1.096343994140625e-05,
      "model_forward_time": 1.7347886562347412,
      "step": 11496
    },
    {
      "epoch": 1.096343994140625e-05,
      "step": 11496,
      "training_step_time": 8.386904954910278
    },
    {
      "epoch": 1.0964393615722656e-05,
      "model_forward_time": 2.894728422164917,
      "step": 11497
    },
    {
      "epoch": 1.0964393615722656e-05,
      "step": 11497,
      "training_step_time": 7.771996736526489
    },
    {
      "epoch": 1.0965347290039062e-05,
      "model_forward_time": 1.7359046936035156,
      "step": 11498
    },
    {
      "epoch": 1.0965347290039062e-05,
      "step": 11498,
      "training_step_time": 7.7595624923706055
    },
    {
      "epoch": 1.096630096435547e-05,
      "model_forward_time": 1.7351751327514648,
      "step": 11499
    },
    {
      "epoch": 1.096630096435547e-05,
      "step": 11499,
      "training_step_time": 8.043861865997314
    },
    {
      "epoch": 1.0967254638671875e-05,
      "grad_norm": 0.18772099912166595,
      "learning_rate": 9.884969419697102e-06,
      "loss": 0.0625,
      "step": 11500
    },
    {
      "epoch": 1.0967254638671875e-05,
      "model_forward_time": 1.7266757488250732,
      "step": 11500
    },
    {
      "dynamics_loss_avg": 0.05506750643253326,
      "epoch": 1.0967254638671875e-05,
      "step": 11500
    },
    {
      "action_loss_avg": 0.0029907147341873496,
      "epoch": 1.0967254638671875e-05,
      "step": 11500
    },
    {
      "epoch": 1.0967254638671875e-05,
      "step": 11500,
      "training_step_time": 7.7084455490112305
    },
    {
      "epoch": 1.096820831298828e-05,
      "model_forward_time": 1.7362239360809326,
      "step": 11501
    },
    {
      "epoch": 1.096820831298828e-05,
      "step": 11501,
      "training_step_time": 7.753389596939087
    },
    {
      "epoch": 1.0969161987304688e-05,
      "model_forward_time": 1.7446670532226562,
      "step": 11502
    },
    {
      "epoch": 1.0969161987304688e-05,
      "step": 11502,
      "training_step_time": 8.406363725662231
    },
    {
      "epoch": 1.0970115661621094e-05,
      "model_forward_time": 1.7431442737579346,
      "step": 11503
    },
    {
      "epoch": 1.0970115661621094e-05,
      "step": 11503,
      "training_step_time": 7.787217617034912
    },
    {
      "epoch": 1.09710693359375e-05,
      "model_forward_time": 1.7635798454284668,
      "step": 11504
    },
    {
      "epoch": 1.09710693359375e-05,
      "step": 11504,
      "training_step_time": 8.701231002807617
    },
    {
      "epoch": 1.0972023010253907e-05,
      "model_forward_time": 1.7311780452728271,
      "step": 11505
    },
    {
      "epoch": 1.0972023010253907e-05,
      "step": 11505,
      "training_step_time": 7.754831314086914
    },
    {
      "epoch": 1.0972976684570313e-05,
      "model_forward_time": 2.9177041053771973,
      "step": 11506
    },
    {
      "epoch": 1.0972976684570313e-05,
      "step": 11506,
      "training_step_time": 8.666951894760132
    },
    {
      "epoch": 1.0973930358886718e-05,
      "model_forward_time": 1.7334282398223877,
      "step": 11507
    },
    {
      "epoch": 1.0973930358886718e-05,
      "step": 11507,
      "training_step_time": 8.874688148498535
    },
    {
      "epoch": 1.0974884033203124e-05,
      "model_forward_time": 1.7304506301879883,
      "step": 11508
    },
    {
      "epoch": 1.0974884033203124e-05,
      "step": 11508,
      "training_step_time": 7.677005290985107
    },
    {
      "epoch": 1.0975837707519532e-05,
      "model_forward_time": 1.7333958148956299,
      "step": 11509
    },
    {
      "epoch": 1.0975837707519532e-05,
      "step": 11509,
      "training_step_time": 7.6917946338653564
    },
    {
      "epoch": 1.0976791381835937e-05,
      "grad_norm": 0.17474688589572906,
      "learning_rate": 9.884616521516614e-06,
      "loss": 0.0594,
      "step": 11510
    },
    {
      "epoch": 1.0976791381835937e-05,
      "model_forward_time": 1.7252802848815918,
      "step": 11510
    },
    {
      "dynamics_loss_avg": 0.0686943294480443,
      "epoch": 1.0976791381835937e-05,
      "step": 11510
    },
    {
      "action_loss_avg": 0.0037080289795994757,
      "epoch": 1.0976791381835937e-05,
      "step": 11510
    },
    {
      "epoch": 1.0976791381835937e-05,
      "step": 11510,
      "training_step_time": 8.588354587554932
    },
    {
      "epoch": 1.0977745056152343e-05,
      "model_forward_time": 1.7197041511535645,
      "step": 11511
    },
    {
      "epoch": 1.0977745056152343e-05,
      "step": 11511,
      "training_step_time": 8.18636679649353
    },
    {
      "epoch": 1.097869873046875e-05,
      "model_forward_time": 1.7214646339416504,
      "step": 11512
    },
    {
      "epoch": 1.097869873046875e-05,
      "step": 11512,
      "training_step_time": 7.759083271026611
    },
    {
      "epoch": 1.0979652404785156e-05,
      "model_forward_time": 1.7726101875305176,
      "step": 11513
    },
    {
      "epoch": 1.0979652404785156e-05,
      "step": 11513,
      "training_step_time": 7.801123857498169
    },
    {
      "epoch": 1.0980606079101562e-05,
      "model_forward_time": 1.7323734760284424,
      "step": 11514
    },
    {
      "epoch": 1.0980606079101562e-05,
      "step": 11514,
      "training_step_time": 8.323261499404907
    },
    {
      "epoch": 1.098155975341797e-05,
      "model_forward_time": 2.8835089206695557,
      "step": 11515
    },
    {
      "epoch": 1.098155975341797e-05,
      "step": 11515,
      "training_step_time": 9.295455694198608
    },
    {
      "epoch": 1.0982513427734375e-05,
      "model_forward_time": 1.7280066013336182,
      "step": 11516
    },
    {
      "epoch": 1.0982513427734375e-05,
      "step": 11516,
      "training_step_time": 7.739889144897461
    },
    {
      "epoch": 1.098346710205078e-05,
      "model_forward_time": 1.7318201065063477,
      "step": 11517
    },
    {
      "epoch": 1.098346710205078e-05,
      "step": 11517,
      "training_step_time": 8.709073066711426
    },
    {
      "epoch": 1.0984420776367188e-05,
      "model_forward_time": 1.3571205139160156,
      "step": 11518
    },
    {
      "epoch": 1.0984420776367188e-05,
      "step": 11518,
      "training_step_time": 8.46011757850647
    },
    {
      "epoch": 1.0985374450683594e-05,
      "model_forward_time": 1.7293288707733154,
      "step": 11519
    },
    {
      "epoch": 1.0985374450683594e-05,
      "step": 11519,
      "training_step_time": 7.692718267440796
    },
    {
      "epoch": 1.0986328125e-05,
      "grad_norm": 0.2172808051109314,
      "learning_rate": 9.88426308916181e-06,
      "loss": 0.0629,
      "step": 11520
    },
    {
      "epoch": 1.0986328125e-05,
      "model_forward_time": 1.737241506576538,
      "step": 11520
    },
    {
      "dynamics_loss_avg": 0.04854293055832386,
      "epoch": 1.0986328125e-05,
      "step": 11520
    },
    {
      "action_loss_avg": 0.0033994289289694278,
      "epoch": 1.0986328125e-05,
      "step": 11520
    },
    {
      "epoch": 1.0986328125e-05,
      "step": 11520,
      "training_step_time": 8.730420351028442
    },
    {
      "epoch": 1.0987281799316407e-05,
      "model_forward_time": 1.7425007820129395,
      "step": 11521
    },
    {
      "epoch": 1.0987281799316407e-05,
      "step": 11521,
      "training_step_time": 7.730506896972656
    },
    {
      "epoch": 1.0988235473632813e-05,
      "model_forward_time": 1.755525827407837,
      "step": 11522
    },
    {
      "epoch": 1.0988235473632813e-05,
      "step": 11522,
      "training_step_time": 7.724519968032837
    },
    {
      "epoch": 1.0989189147949218e-05,
      "model_forward_time": 1.7280199527740479,
      "step": 11523
    },
    {
      "epoch": 1.0989189147949218e-05,
      "step": 11523,
      "training_step_time": 8.507348537445068
    },
    {
      "epoch": 1.0990142822265624e-05,
      "model_forward_time": 2.856358051300049,
      "step": 11524
    },
    {
      "epoch": 1.0990142822265624e-05,
      "step": 11524,
      "training_step_time": 7.760169982910156
    },
    {
      "epoch": 1.0991096496582032e-05,
      "model_forward_time": 1.042532205581665,
      "step": 11525
    },
    {
      "epoch": 1.0991096496582032e-05,
      "step": 11525,
      "training_step_time": 8.723060131072998
    },
    {
      "epoch": 1.0992050170898437e-05,
      "model_forward_time": 1.739905595779419,
      "step": 11526
    },
    {
      "epoch": 1.0992050170898437e-05,
      "step": 11526,
      "training_step_time": 8.54257583618164
    },
    {
      "epoch": 1.0993003845214843e-05,
      "model_forward_time": 1.7374131679534912,
      "step": 11527
    },
    {
      "epoch": 1.0993003845214843e-05,
      "step": 11527,
      "training_step_time": 8.522906303405762
    },
    {
      "epoch": 1.099395751953125e-05,
      "model_forward_time": 1.7393031120300293,
      "step": 11528
    },
    {
      "epoch": 1.099395751953125e-05,
      "step": 11528,
      "training_step_time": 7.695197582244873
    },
    {
      "epoch": 1.0994911193847656e-05,
      "model_forward_time": 1.7333495616912842,
      "step": 11529
    },
    {
      "epoch": 1.0994911193847656e-05,
      "step": 11529,
      "training_step_time": 7.708284139633179
    },
    {
      "epoch": 1.0995864868164062e-05,
      "grad_norm": 0.16963058710098267,
      "learning_rate": 9.883909122671336e-06,
      "loss": 0.0663,
      "step": 11530
    },
    {
      "epoch": 1.0995864868164062e-05,
      "model_forward_time": 1.8141920566558838,
      "step": 11530
    },
    {
      "dynamics_loss_avg": 0.052553508430719376,
      "epoch": 1.0995864868164062e-05,
      "step": 11530
    },
    {
      "action_loss_avg": 0.002862481167539954,
      "epoch": 1.0995864868164062e-05,
      "step": 11530
    },
    {
      "epoch": 1.0995864868164062e-05,
      "step": 11530,
      "training_step_time": 7.802602052688599
    },
    {
      "epoch": 1.099681854248047e-05,
      "model_forward_time": 1.723581075668335,
      "step": 11531
    },
    {
      "epoch": 1.099681854248047e-05,
      "step": 11531,
      "training_step_time": 7.739233732223511
    },
    {
      "epoch": 1.0997772216796875e-05,
      "model_forward_time": 1.7315387725830078,
      "step": 11532
    },
    {
      "epoch": 1.0997772216796875e-05,
      "step": 11532,
      "training_step_time": 8.769465684890747
    },
    {
      "epoch": 1.099872589111328e-05,
      "model_forward_time": 1.3796887397766113,
      "step": 11533
    },
    {
      "epoch": 1.099872589111328e-05,
      "step": 11533,
      "training_step_time": 7.811047792434692
    },
    {
      "epoch": 1.0999679565429688e-05,
      "model_forward_time": 2.8743319511413574,
      "step": 11534
    },
    {
      "epoch": 1.0999679565429688e-05,
      "step": 11534,
      "training_step_time": 7.780959129333496
    },
    {
      "epoch": 1.1000633239746094e-05,
      "model_forward_time": 1.735032558441162,
      "step": 11535
    },
    {
      "epoch": 1.1000633239746094e-05,
      "step": 11535,
      "training_step_time": 7.807227849960327
    },
    {
      "epoch": 1.10015869140625e-05,
      "model_forward_time": 1.7429296970367432,
      "step": 11536
    },
    {
      "epoch": 1.10015869140625e-05,
      "step": 11536,
      "training_step_time": 7.552417993545532
    },
    {
      "epoch": 1.1002540588378907e-05,
      "model_forward_time": 1.7308766841888428,
      "step": 11537
    },
    {
      "epoch": 1.1002540588378907e-05,
      "step": 11537,
      "training_step_time": 8.606051445007324
    },
    {
      "epoch": 1.1003494262695313e-05,
      "model_forward_time": 1.7308804988861084,
      "step": 11538
    },
    {
      "epoch": 1.1003494262695313e-05,
      "step": 11538,
      "training_step_time": 8.288633346557617
    },
    {
      "epoch": 1.1004447937011718e-05,
      "model_forward_time": 1.7937650680541992,
      "step": 11539
    },
    {
      "epoch": 1.1004447937011718e-05,
      "step": 11539,
      "training_step_time": 8.87244701385498
    },
    {
      "epoch": 1.1005401611328124e-05,
      "grad_norm": 0.21540449559688568,
      "learning_rate": 9.883554622083904e-06,
      "loss": 0.068,
      "step": 11540
    },
    {
      "epoch": 1.1005401611328124e-05,
      "model_forward_time": 1.7432823181152344,
      "step": 11540
    },
    {
      "dynamics_loss_avg": 0.05793689377605915,
      "epoch": 1.1005401611328124e-05,
      "step": 11540
    },
    {
      "action_loss_avg": 0.002678797650150955,
      "epoch": 1.1005401611328124e-05,
      "step": 11540
    },
    {
      "epoch": 1.1005401611328124e-05,
      "step": 11540,
      "training_step_time": 8.443438291549683
    },
    {
      "epoch": 1.1006355285644532e-05,
      "model_forward_time": 1.7364997863769531,
      "step": 11541
    },
    {
      "epoch": 1.1006355285644532e-05,
      "step": 11541,
      "training_step_time": 8.700799226760864
    },
    {
      "epoch": 1.1007308959960937e-05,
      "model_forward_time": 1.7352795600891113,
      "step": 11542
    },
    {
      "epoch": 1.1007308959960937e-05,
      "step": 11542,
      "training_step_time": 7.901787042617798
    },
    {
      "epoch": 1.1008262634277343e-05,
      "model_forward_time": 2.856016159057617,
      "step": 11543
    },
    {
      "epoch": 1.1008262634277343e-05,
      "step": 11543,
      "training_step_time": 7.784001588821411
    },
    {
      "epoch": 1.100921630859375e-05,
      "model_forward_time": 1.734328269958496,
      "step": 11544
    },
    {
      "epoch": 1.100921630859375e-05,
      "step": 11544,
      "training_step_time": 8.566609859466553
    },
    {
      "epoch": 1.1010169982910156e-05,
      "model_forward_time": 1.7418663501739502,
      "step": 11545
    },
    {
      "epoch": 1.1010169982910156e-05,
      "step": 11545,
      "training_step_time": 7.817339897155762
    },
    {
      "epoch": 1.1011123657226562e-05,
      "model_forward_time": 1.741407871246338,
      "step": 11546
    },
    {
      "epoch": 1.1011123657226562e-05,
      "step": 11546,
      "training_step_time": 7.708856582641602
    },
    {
      "epoch": 1.101207733154297e-05,
      "model_forward_time": 1.7425107955932617,
      "step": 11547
    },
    {
      "epoch": 1.101207733154297e-05,
      "step": 11547,
      "training_step_time": 7.693031549453735
    },
    {
      "epoch": 1.1013031005859375e-05,
      "model_forward_time": 1.8291211128234863,
      "step": 11548
    },
    {
      "epoch": 1.1013031005859375e-05,
      "step": 11548,
      "training_step_time": 7.714807748794556
    },
    {
      "epoch": 1.101398468017578e-05,
      "model_forward_time": 1.7342493534088135,
      "step": 11549
    },
    {
      "epoch": 1.101398468017578e-05,
      "step": 11549,
      "training_step_time": 8.749353170394897
    },
    {
      "epoch": 1.1014938354492188e-05,
      "grad_norm": 0.1805555671453476,
      "learning_rate": 9.88319958743828e-06,
      "loss": 0.0695,
      "step": 11550
    },
    {
      "epoch": 1.1014938354492188e-05,
      "model_forward_time": 1.7464847564697266,
      "step": 11550
    },
    {
      "dynamics_loss_avg": 0.07527382746338844,
      "epoch": 1.1014938354492188e-05,
      "step": 11550
    },
    {
      "action_loss_avg": 0.003550682298373431,
      "epoch": 1.1014938354492188e-05,
      "step": 11550
    },
    {
      "epoch": 1.1014938354492188e-05,
      "step": 11550,
      "training_step_time": 8.649640560150146
    },
    {
      "epoch": 1.1015892028808594e-05,
      "model_forward_time": 1.7338054180145264,
      "step": 11551
    },
    {
      "epoch": 1.1015892028808594e-05,
      "step": 11551,
      "training_step_time": 8.630358219146729
    },
    {
      "epoch": 1.1016845703125e-05,
      "model_forward_time": 2.9170515537261963,
      "step": 11552
    },
    {
      "epoch": 1.1016845703125e-05,
      "step": 11552,
      "training_step_time": 8.350880861282349
    },
    {
      "epoch": 1.1017799377441407e-05,
      "model_forward_time": 1.7362701892852783,
      "step": 11553
    },
    {
      "epoch": 1.1017799377441407e-05,
      "step": 11553,
      "training_step_time": 8.705436706542969
    },
    {
      "epoch": 1.1018753051757813e-05,
      "model_forward_time": 1.7361176013946533,
      "step": 11554
    },
    {
      "epoch": 1.1018753051757813e-05,
      "step": 11554,
      "training_step_time": 7.691419363021851
    },
    {
      "epoch": 1.1019706726074218e-05,
      "model_forward_time": 1.744436264038086,
      "step": 11555
    },
    {
      "epoch": 1.1019706726074218e-05,
      "step": 11555,
      "training_step_time": 7.833126068115234
    },
    {
      "epoch": 1.1020660400390624e-05,
      "model_forward_time": 1.7446808815002441,
      "step": 11556
    },
    {
      "epoch": 1.1020660400390624e-05,
      "step": 11556,
      "training_step_time": 7.717850685119629
    },
    {
      "epoch": 1.1021614074707032e-05,
      "model_forward_time": 1.8058578968048096,
      "step": 11557
    },
    {
      "epoch": 1.1021614074707032e-05,
      "step": 11557,
      "training_step_time": 8.459571838378906
    },
    {
      "epoch": 1.1022567749023437e-05,
      "model_forward_time": 1.7289953231811523,
      "step": 11558
    },
    {
      "epoch": 1.1022567749023437e-05,
      "step": 11558,
      "training_step_time": 7.789526462554932
    },
    {
      "epoch": 1.1023521423339843e-05,
      "model_forward_time": 1.7351436614990234,
      "step": 11559
    },
    {
      "epoch": 1.1023521423339843e-05,
      "step": 11559,
      "training_step_time": 9.528550386428833
    },
    {
      "epoch": 1.102447509765625e-05,
      "grad_norm": 0.15852966904640198,
      "learning_rate": 9.882844018773293e-06,
      "loss": 0.0668,
      "step": 11560
    },
    {
      "epoch": 1.102447509765625e-05,
      "model_forward_time": 1.7503821849822998,
      "step": 11560
    },
    {
      "dynamics_loss_avg": 0.05762299224734306,
      "epoch": 1.102447509765625e-05,
      "step": 11560
    },
    {
      "action_loss_avg": 0.002558930998202413,
      "epoch": 1.102447509765625e-05,
      "step": 11560
    },
    {
      "epoch": 1.102447509765625e-05,
      "step": 11560,
      "training_step_time": 7.82711386680603
    },
    {
      "epoch": 1.1025428771972656e-05,
      "model_forward_time": 1.7278850078582764,
      "step": 11561
    },
    {
      "epoch": 1.1025428771972656e-05,
      "step": 11561,
      "training_step_time": 8.269646167755127
    },
    {
      "epoch": 1.1026382446289062e-05,
      "model_forward_time": 1.7303807735443115,
      "step": 11562
    },
    {
      "epoch": 1.1026382446289062e-05,
      "step": 11562,
      "training_step_time": 9.58964228630066
    },
    {
      "epoch": 1.102733612060547e-05,
      "model_forward_time": 1.733128309249878,
      "step": 11563
    },
    {
      "epoch": 1.102733612060547e-05,
      "step": 11563,
      "training_step_time": 7.828550815582275
    },
    {
      "epoch": 1.1028289794921875e-05,
      "model_forward_time": 1.7378246784210205,
      "step": 11564
    },
    {
      "epoch": 1.1028289794921875e-05,
      "step": 11564,
      "training_step_time": 7.795342206954956
    },
    {
      "epoch": 1.102924346923828e-05,
      "model_forward_time": 1.736753225326538,
      "step": 11565
    },
    {
      "epoch": 1.102924346923828e-05,
      "step": 11565,
      "training_step_time": 7.694853067398071
    },
    {
      "epoch": 1.1030197143554688e-05,
      "model_forward_time": 1.8135998249053955,
      "step": 11566
    },
    {
      "epoch": 1.1030197143554688e-05,
      "step": 11566,
      "training_step_time": 7.758738279342651
    },
    {
      "epoch": 1.1031150817871094e-05,
      "model_forward_time": 1.736293077468872,
      "step": 11567
    },
    {
      "epoch": 1.1031150817871094e-05,
      "step": 11567,
      "training_step_time": 7.746809959411621
    },
    {
      "epoch": 1.10321044921875e-05,
      "model_forward_time": 1.732370138168335,
      "step": 11568
    },
    {
      "epoch": 1.10321044921875e-05,
      "step": 11568,
      "training_step_time": 9.384968757629395
    },
    {
      "epoch": 1.1033058166503907e-05,
      "model_forward_time": 1.743107557296753,
      "step": 11569
    },
    {
      "epoch": 1.1033058166503907e-05,
      "step": 11569,
      "training_step_time": 8.818904638290405
    },
    {
      "epoch": 1.1034011840820313e-05,
      "grad_norm": 0.19422677159309387,
      "learning_rate": 9.882487916127824e-06,
      "loss": 0.0691,
      "step": 11570
    },
    {
      "epoch": 1.1034011840820313e-05,
      "model_forward_time": 1.737198829650879,
      "step": 11570
    },
    {
      "dynamics_loss_avg": 0.052584372274577615,
      "epoch": 1.1034011840820313e-05,
      "step": 11570
    },
    {
      "action_loss_avg": 0.003690728323999792,
      "epoch": 1.1034011840820313e-05,
      "step": 11570
    },
    {
      "epoch": 1.1034011840820313e-05,
      "step": 11570,
      "training_step_time": 9.37857699394226
    },
    {
      "epoch": 1.1034965515136718e-05,
      "model_forward_time": 2.986952304840088,
      "step": 11571
    },
    {
      "epoch": 1.1034965515136718e-05,
      "step": 11571,
      "training_step_time": 7.880511045455933
    },
    {
      "epoch": 1.1035919189453124e-05,
      "model_forward_time": 1.7405786514282227,
      "step": 11572
    },
    {
      "epoch": 1.1035919189453124e-05,
      "step": 11572,
      "training_step_time": 7.705248117446899
    },
    {
      "epoch": 1.1036872863769532e-05,
      "model_forward_time": 1.7389283180236816,
      "step": 11573
    },
    {
      "epoch": 1.1036872863769532e-05,
      "step": 11573,
      "training_step_time": 7.7271599769592285
    },
    {
      "epoch": 1.1037826538085937e-05,
      "model_forward_time": 1.7368032932281494,
      "step": 11574
    },
    {
      "epoch": 1.1037826538085937e-05,
      "step": 11574,
      "training_step_time": 7.6488025188446045
    },
    {
      "epoch": 1.1038780212402343e-05,
      "model_forward_time": 1.8040745258331299,
      "step": 11575
    },
    {
      "epoch": 1.1038780212402343e-05,
      "step": 11575,
      "training_step_time": 7.716517686843872
    },
    {
      "epoch": 1.103973388671875e-05,
      "model_forward_time": 1.7446792125701904,
      "step": 11576
    },
    {
      "epoch": 1.103973388671875e-05,
      "step": 11576,
      "training_step_time": 8.204718828201294
    },
    {
      "epoch": 1.1040687561035156e-05,
      "model_forward_time": 1.7571446895599365,
      "step": 11577
    },
    {
      "epoch": 1.1040687561035156e-05,
      "step": 11577,
      "training_step_time": 7.746825218200684
    },
    {
      "epoch": 1.1041641235351562e-05,
      "model_forward_time": 1.7500724792480469,
      "step": 11578
    },
    {
      "epoch": 1.1041641235351562e-05,
      "step": 11578,
      "training_step_time": 7.815676689147949
    },
    {
      "epoch": 1.104259490966797e-05,
      "model_forward_time": 1.746239423751831,
      "step": 11579
    },
    {
      "epoch": 1.104259490966797e-05,
      "step": 11579,
      "training_step_time": 8.878141641616821
    },
    {
      "epoch": 1.1043548583984375e-05,
      "grad_norm": 0.25740739703178406,
      "learning_rate": 9.882131279540816e-06,
      "loss": 0.0704,
      "step": 11580
    },
    {
      "epoch": 1.1043548583984375e-05,
      "model_forward_time": 2.916717052459717,
      "step": 11580
    },
    {
      "dynamics_loss_avg": 0.059228602796792984,
      "epoch": 1.1043548583984375e-05,
      "step": 11580
    },
    {
      "action_loss_avg": 0.0032488075201399626,
      "epoch": 1.1043548583984375e-05,
      "step": 11580
    },
    {
      "epoch": 1.1043548583984375e-05,
      "step": 11580,
      "training_step_time": 7.844804763793945
    },
    {
      "epoch": 1.104450225830078e-05,
      "model_forward_time": 1.747800588607788,
      "step": 11581
    },
    {
      "epoch": 1.104450225830078e-05,
      "step": 11581,
      "training_step_time": 7.817535400390625
    },
    {
      "epoch": 1.1045455932617188e-05,
      "model_forward_time": 1.7538888454437256,
      "step": 11582
    },
    {
      "epoch": 1.1045455932617188e-05,
      "step": 11582,
      "training_step_time": 7.771643161773682
    },
    {
      "epoch": 1.1046409606933594e-05,
      "model_forward_time": 1.7471232414245605,
      "step": 11583
    },
    {
      "epoch": 1.1046409606933594e-05,
      "step": 11583,
      "training_step_time": 7.664336919784546
    },
    {
      "epoch": 1.104736328125e-05,
      "model_forward_time": 1.816798448562622,
      "step": 11584
    },
    {
      "epoch": 1.104736328125e-05,
      "step": 11584,
      "training_step_time": 7.705711364746094
    },
    {
      "epoch": 1.1048316955566407e-05,
      "model_forward_time": 1.7517671585083008,
      "step": 11585
    },
    {
      "epoch": 1.1048316955566407e-05,
      "step": 11585,
      "training_step_time": 8.409893274307251
    },
    {
      "epoch": 1.1049270629882813e-05,
      "model_forward_time": 1.74534273147583,
      "step": 11586
    },
    {
      "epoch": 1.1049270629882813e-05,
      "step": 11586,
      "training_step_time": 7.722585201263428
    },
    {
      "epoch": 1.1050224304199219e-05,
      "model_forward_time": 1.7437787055969238,
      "step": 11587
    },
    {
      "epoch": 1.1050224304199219e-05,
      "step": 11587,
      "training_step_time": 8.411879539489746
    },
    {
      "epoch": 1.1051177978515624e-05,
      "model_forward_time": 1.7576122283935547,
      "step": 11588
    },
    {
      "epoch": 1.1051177978515624e-05,
      "step": 11588,
      "training_step_time": 7.651440382003784
    },
    {
      "epoch": 1.1052131652832032e-05,
      "model_forward_time": 2.889585494995117,
      "step": 11589
    },
    {
      "epoch": 1.1052131652832032e-05,
      "step": 11589,
      "training_step_time": 7.693713665008545
    },
    {
      "epoch": 1.1053085327148437e-05,
      "grad_norm": 0.20199047029018402,
      "learning_rate": 9.881774109051272e-06,
      "loss": 0.0717,
      "step": 11590
    },
    {
      "epoch": 1.1053085327148437e-05,
      "model_forward_time": 1.755624771118164,
      "step": 11590
    },
    {
      "dynamics_loss_avg": 0.062006843835115434,
      "epoch": 1.1053085327148437e-05,
      "step": 11590
    },
    {
      "action_loss_avg": 0.003256782388780266,
      "epoch": 1.1053085327148437e-05,
      "step": 11590
    },
    {
      "epoch": 1.1053085327148437e-05,
      "step": 11590,
      "training_step_time": 7.684035778045654
    },
    {
      "epoch": 1.1054039001464843e-05,
      "model_forward_time": 1.7534654140472412,
      "step": 11591
    },
    {
      "epoch": 1.1054039001464843e-05,
      "step": 11591,
      "training_step_time": 7.699143648147583
    },
    {
      "epoch": 1.105499267578125e-05,
      "model_forward_time": 1.7472329139709473,
      "step": 11592
    },
    {
      "epoch": 1.105499267578125e-05,
      "step": 11592,
      "training_step_time": 7.586445569992065
    },
    {
      "epoch": 1.1055946350097656e-05,
      "model_forward_time": 1.8114848136901855,
      "step": 11593
    },
    {
      "epoch": 1.1055946350097656e-05,
      "step": 11593,
      "training_step_time": 7.652555465698242
    },
    {
      "epoch": 1.1056900024414062e-05,
      "model_forward_time": 1.755746841430664,
      "step": 11594
    },
    {
      "epoch": 1.1056900024414062e-05,
      "step": 11594,
      "training_step_time": 8.443986654281616
    },
    {
      "epoch": 1.105785369873047e-05,
      "model_forward_time": 1.737563133239746,
      "step": 11595
    },
    {
      "epoch": 1.105785369873047e-05,
      "step": 11595,
      "training_step_time": 8.202341079711914
    },
    {
      "epoch": 1.1058807373046875e-05,
      "model_forward_time": 1.7281906604766846,
      "step": 11596
    },
    {
      "epoch": 1.1058807373046875e-05,
      "step": 11596,
      "training_step_time": 7.655399799346924
    },
    {
      "epoch": 1.105976104736328e-05,
      "model_forward_time": 1.7186222076416016,
      "step": 11597
    },
    {
      "epoch": 1.105976104736328e-05,
      "step": 11597,
      "training_step_time": 8.80757474899292
    },
    {
      "epoch": 1.1060714721679688e-05,
      "model_forward_time": 2.303534746170044,
      "step": 11598
    },
    {
      "epoch": 1.1060714721679688e-05,
      "step": 11598,
      "training_step_time": 8.134157419204712
    },
    {
      "epoch": 1.1061668395996094e-05,
      "model_forward_time": 1.7410428524017334,
      "step": 11599
    },
    {
      "epoch": 1.1061668395996094e-05,
      "step": 11599,
      "training_step_time": 7.683058977127075
    },
    {
      "epoch": 1.10626220703125e-05,
      "grad_norm": 0.20700117945671082,
      "learning_rate": 9.881416404698251e-06,
      "loss": 0.066,
      "step": 11600
    },
    {
      "epoch": 1.10626220703125e-05,
      "model_forward_time": 1.7126624584197998,
      "step": 11600
    },
    {
      "dynamics_loss_avg": 0.0605636365711689,
      "epoch": 1.10626220703125e-05,
      "step": 11600
    },
    {
      "action_loss_avg": 0.003226662229280919,
      "epoch": 1.10626220703125e-05,
      "step": 11600
    },
    {
      "epoch": 1.10626220703125e-05,
      "step": 11600,
      "training_step_time": 9.520434141159058
    },
    {
      "epoch": 1.1063575744628907e-05,
      "model_forward_time": 1.732090950012207,
      "step": 11601
    },
    {
      "epoch": 1.1063575744628907e-05,
      "step": 11601,
      "training_step_time": 7.766534805297852
    },
    {
      "epoch": 1.1064529418945313e-05,
      "model_forward_time": 1.717627763748169,
      "step": 11602
    },
    {
      "epoch": 1.1064529418945313e-05,
      "step": 11602,
      "training_step_time": 7.801877975463867
    },
    {
      "epoch": 1.1065483093261719e-05,
      "model_forward_time": 1.7194104194641113,
      "step": 11603
    },
    {
      "epoch": 1.1065483093261719e-05,
      "step": 11603,
      "training_step_time": 7.959367275238037
    },
    {
      "epoch": 1.1066436767578124e-05,
      "model_forward_time": 1.7393138408660889,
      "step": 11604
    },
    {
      "epoch": 1.1066436767578124e-05,
      "step": 11604,
      "training_step_time": 8.449774026870728
    },
    {
      "epoch": 1.1067390441894532e-05,
      "model_forward_time": 1.7186999320983887,
      "step": 11605
    },
    {
      "epoch": 1.1067390441894532e-05,
      "step": 11605,
      "training_step_time": 7.6986730098724365
    },
    {
      "epoch": 1.1068344116210937e-05,
      "model_forward_time": 1.720705509185791,
      "step": 11606
    },
    {
      "epoch": 1.1068344116210937e-05,
      "step": 11606,
      "training_step_time": 7.770740270614624
    },
    {
      "epoch": 1.1069297790527343e-05,
      "model_forward_time": 2.26493763923645,
      "step": 11607
    },
    {
      "epoch": 1.1069297790527343e-05,
      "step": 11607,
      "training_step_time": 7.731515884399414
    },
    {
      "epoch": 1.107025146484375e-05,
      "model_forward_time": 1.745469093322754,
      "step": 11608
    },
    {
      "epoch": 1.107025146484375e-05,
      "step": 11608,
      "training_step_time": 7.725078105926514
    },
    {
      "epoch": 1.1071205139160156e-05,
      "model_forward_time": 1.724010705947876,
      "step": 11609
    },
    {
      "epoch": 1.1071205139160156e-05,
      "step": 11609,
      "training_step_time": 8.82229495048523
    },
    {
      "epoch": 1.1072158813476562e-05,
      "grad_norm": 0.2651270031929016,
      "learning_rate": 9.881058166520873e-06,
      "loss": 0.067,
      "step": 11610
    },
    {
      "epoch": 1.1072158813476562e-05,
      "model_forward_time": 1.7143194675445557,
      "step": 11610
    },
    {
      "dynamics_loss_avg": 0.055197723954916,
      "epoch": 1.1072158813476562e-05,
      "step": 11610
    },
    {
      "action_loss_avg": 0.0024751055520027877,
      "epoch": 1.1072158813476562e-05,
      "step": 11610
    },
    {
      "epoch": 1.1072158813476562e-05,
      "step": 11610,
      "training_step_time": 8.08315396308899
    },
    {
      "epoch": 1.107311248779297e-05,
      "model_forward_time": 1.7207820415496826,
      "step": 11611
    },
    {
      "epoch": 1.107311248779297e-05,
      "step": 11611,
      "training_step_time": 8.49480414390564
    },
    {
      "epoch": 1.1074066162109375e-05,
      "model_forward_time": 1.3387298583984375,
      "step": 11612
    },
    {
      "epoch": 1.1074066162109375e-05,
      "step": 11612,
      "training_step_time": 7.736032485961914
    },
    {
      "epoch": 1.107501983642578e-05,
      "model_forward_time": 1.7442512512207031,
      "step": 11613
    },
    {
      "epoch": 1.107501983642578e-05,
      "step": 11613,
      "training_step_time": 7.821564435958862
    },
    {
      "epoch": 1.1075973510742188e-05,
      "model_forward_time": 1.7351462841033936,
      "step": 11614
    },
    {
      "epoch": 1.1075973510742188e-05,
      "step": 11614,
      "training_step_time": 7.7035300731658936
    },
    {
      "epoch": 1.1076927185058594e-05,
      "model_forward_time": 1.734067678451538,
      "step": 11615
    },
    {
      "epoch": 1.1076927185058594e-05,
      "step": 11615,
      "training_step_time": 7.994132041931152
    },
    {
      "epoch": 1.1077880859375e-05,
      "model_forward_time": 1.7270817756652832,
      "step": 11616
    },
    {
      "epoch": 1.1077880859375e-05,
      "step": 11616,
      "training_step_time": 10.274890899658203
    },
    {
      "epoch": 1.1078834533691407e-05,
      "model_forward_time": 2.2972521781921387,
      "step": 11617
    },
    {
      "epoch": 1.1078834533691407e-05,
      "step": 11617,
      "training_step_time": 8.691792011260986
    },
    {
      "epoch": 1.1079788208007813e-05,
      "model_forward_time": 1.7319450378417969,
      "step": 11618
    },
    {
      "epoch": 1.1079788208007813e-05,
      "step": 11618,
      "training_step_time": 7.749690294265747
    },
    {
      "epoch": 1.1080741882324219e-05,
      "model_forward_time": 1.7273719310760498,
      "step": 11619
    },
    {
      "epoch": 1.1080741882324219e-05,
      "step": 11619,
      "training_step_time": 7.588210582733154
    },
    {
      "epoch": 1.1081695556640624e-05,
      "grad_norm": 0.39016422629356384,
      "learning_rate": 9.88069939455831e-06,
      "loss": 0.0704,
      "step": 11620
    },
    {
      "epoch": 1.1081695556640624e-05,
      "model_forward_time": 1.7202305793762207,
      "step": 11620
    },
    {
      "dynamics_loss_avg": 0.05411405023187399,
      "epoch": 1.1081695556640624e-05,
      "step": 11620
    },
    {
      "action_loss_avg": 0.003850539104314521,
      "epoch": 1.1081695556640624e-05,
      "step": 11620
    },
    {
      "epoch": 1.1081695556640624e-05,
      "step": 11620,
      "training_step_time": 7.691556692123413
    },
    {
      "epoch": 1.1082649230957032e-05,
      "model_forward_time": 1.7284343242645264,
      "step": 11621
    },
    {
      "epoch": 1.1082649230957032e-05,
      "step": 11621,
      "training_step_time": 7.747733116149902
    },
    {
      "epoch": 1.1083602905273437e-05,
      "model_forward_time": 1.7249152660369873,
      "step": 11622
    },
    {
      "epoch": 1.1083602905273437e-05,
      "step": 11622,
      "training_step_time": 7.716999530792236
    },
    {
      "epoch": 1.1084556579589843e-05,
      "model_forward_time": 1.7302851676940918,
      "step": 11623
    },
    {
      "epoch": 1.1084556579589843e-05,
      "step": 11623,
      "training_step_time": 9.183090448379517
    },
    {
      "epoch": 1.108551025390625e-05,
      "model_forward_time": 1.727480173110962,
      "step": 11624
    },
    {
      "epoch": 1.108551025390625e-05,
      "step": 11624,
      "training_step_time": 8.80484914779663
    },
    {
      "epoch": 1.1086463928222656e-05,
      "model_forward_time": 1.7284841537475586,
      "step": 11625
    },
    {
      "epoch": 1.1086463928222656e-05,
      "step": 11625,
      "training_step_time": 8.428679466247559
    },
    {
      "epoch": 1.1087417602539062e-05,
      "model_forward_time": 2.2931041717529297,
      "step": 11626
    },
    {
      "epoch": 1.1087417602539062e-05,
      "step": 11626,
      "training_step_time": 7.68646764755249
    },
    {
      "epoch": 1.108837127685547e-05,
      "model_forward_time": 1.7481880187988281,
      "step": 11627
    },
    {
      "epoch": 1.108837127685547e-05,
      "step": 11627,
      "training_step_time": 8.758561849594116
    },
    {
      "epoch": 1.1089324951171875e-05,
      "model_forward_time": 1.7190840244293213,
      "step": 11628
    },
    {
      "epoch": 1.1089324951171875e-05,
      "step": 11628,
      "training_step_time": 7.510675430297852
    },
    {
      "epoch": 1.109027862548828e-05,
      "model_forward_time": 1.7199814319610596,
      "step": 11629
    },
    {
      "epoch": 1.109027862548828e-05,
      "step": 11629,
      "training_step_time": 7.745167016983032
    },
    {
      "epoch": 1.1091232299804688e-05,
      "grad_norm": 0.33553048968315125,
      "learning_rate": 9.880340088849801e-06,
      "loss": 0.0712,
      "step": 11630
    },
    {
      "epoch": 1.1091232299804688e-05,
      "model_forward_time": 1.7218377590179443,
      "step": 11630
    },
    {
      "dynamics_loss_avg": 0.04498415403068066,
      "epoch": 1.1091232299804688e-05,
      "step": 11630
    },
    {
      "action_loss_avg": 0.0020545253879390656,
      "epoch": 1.1091232299804688e-05,
      "step": 11630
    },
    {
      "epoch": 1.1091232299804688e-05,
      "step": 11630,
      "training_step_time": 9.467448711395264
    },
    {
      "epoch": 1.1092185974121094e-05,
      "model_forward_time": 1.733102560043335,
      "step": 11631
    },
    {
      "epoch": 1.1092185974121094e-05,
      "step": 11631,
      "training_step_time": 8.844307899475098
    },
    {
      "epoch": 1.10931396484375e-05,
      "model_forward_time": 1.7275683879852295,
      "step": 11632
    },
    {
      "epoch": 1.10931396484375e-05,
      "step": 11632,
      "training_step_time": 8.57686734199524
    },
    {
      "epoch": 1.1094093322753905e-05,
      "model_forward_time": 1.7259259223937988,
      "step": 11633
    },
    {
      "epoch": 1.1094093322753905e-05,
      "step": 11633,
      "training_step_time": 8.324704647064209
    },
    {
      "epoch": 1.1095046997070313e-05,
      "model_forward_time": 1.7247886657714844,
      "step": 11634
    },
    {
      "epoch": 1.1095046997070313e-05,
      "step": 11634,
      "training_step_time": 7.840136766433716
    },
    {
      "epoch": 1.1096000671386719e-05,
      "model_forward_time": 2.291799545288086,
      "step": 11635
    },
    {
      "epoch": 1.1096000671386719e-05,
      "step": 11635,
      "training_step_time": 8.377066612243652
    },
    {
      "epoch": 1.1096954345703124e-05,
      "model_forward_time": 1.750702142715454,
      "step": 11636
    },
    {
      "epoch": 1.1096954345703124e-05,
      "step": 11636,
      "training_step_time": 9.654178857803345
    },
    {
      "epoch": 1.1097908020019532e-05,
      "model_forward_time": 1.7281603813171387,
      "step": 11637
    },
    {
      "epoch": 1.1097908020019532e-05,
      "step": 11637,
      "training_step_time": 7.860305070877075
    },
    {
      "epoch": 1.1098861694335937e-05,
      "model_forward_time": 1.7171399593353271,
      "step": 11638
    },
    {
      "epoch": 1.1098861694335937e-05,
      "step": 11638,
      "training_step_time": 7.802525043487549
    },
    {
      "epoch": 1.1099815368652343e-05,
      "model_forward_time": 1.7256615161895752,
      "step": 11639
    },
    {
      "epoch": 1.1099815368652343e-05,
      "step": 11639,
      "training_step_time": 7.766348361968994
    },
    {
      "epoch": 1.110076904296875e-05,
      "grad_norm": 0.3120737075805664,
      "learning_rate": 9.879980249434637e-06,
      "loss": 0.0688,
      "step": 11640
    },
    {
      "epoch": 1.110076904296875e-05,
      "model_forward_time": 1.7204041481018066,
      "step": 11640
    },
    {
      "dynamics_loss_avg": 0.0632846923545003,
      "epoch": 1.110076904296875e-05,
      "step": 11640
    },
    {
      "action_loss_avg": 0.003690060612279922,
      "epoch": 1.110076904296875e-05,
      "step": 11640
    },
    {
      "epoch": 1.110076904296875e-05,
      "step": 11640,
      "training_step_time": 8.785666465759277
    },
    {
      "epoch": 1.1101722717285156e-05,
      "model_forward_time": 1.7345614433288574,
      "step": 11641
    },
    {
      "epoch": 1.1101722717285156e-05,
      "step": 11641,
      "training_step_time": 8.417495012283325
    },
    {
      "epoch": 1.1102676391601562e-05,
      "model_forward_time": 2.308805227279663,
      "step": 11642
    },
    {
      "epoch": 1.1102676391601562e-05,
      "step": 11642,
      "training_step_time": 8.407505989074707
    },
    {
      "epoch": 1.110363006591797e-05,
      "model_forward_time": 1.7275147438049316,
      "step": 11643
    },
    {
      "epoch": 1.110363006591797e-05,
      "step": 11643,
      "training_step_time": 7.8425984382629395
    },
    {
      "epoch": 1.1104583740234375e-05,
      "model_forward_time": 1.752293586730957,
      "step": 11644
    },
    {
      "epoch": 1.1104583740234375e-05,
      "step": 11644,
      "training_step_time": 6.952204942703247
    },
    {
      "epoch": 1.110553741455078e-05,
      "model_forward_time": 1.7458646297454834,
      "step": 11645
    },
    {
      "epoch": 1.110553741455078e-05,
      "step": 11645,
      "training_step_time": 6.852128267288208
    },
    {
      "epoch": 1.1106491088867188e-05,
      "model_forward_time": 1.7398111820220947,
      "step": 11646
    },
    {
      "epoch": 1.1106491088867188e-05,
      "step": 11646,
      "training_step_time": 6.84815526008606
    },
    {
      "epoch": 1.1107444763183594e-05,
      "model_forward_time": 1.7414047718048096,
      "step": 11647
    },
    {
      "epoch": 1.1107444763183594e-05,
      "step": 11647,
      "training_step_time": 6.855313777923584
    },
    {
      "epoch": 1.11083984375e-05,
      "model_forward_time": 1.736264944076538,
      "step": 11648
    },
    {
      "epoch": 1.11083984375e-05,
      "step": 11648,
      "training_step_time": 6.829070091247559
    },
    {
      "epoch": 1.1109352111816405e-05,
      "model_forward_time": 1.7467453479766846,
      "step": 11649
    },
    {
      "epoch": 1.1109352111816405e-05,
      "step": 11649,
      "training_step_time": 6.786190509796143
    },
    {
      "epoch": 1.1110305786132813e-05,
      "grad_norm": 0.2591173052787781,
      "learning_rate": 9.879619876352169e-06,
      "loss": 0.0733,
      "step": 11650
    },
    {
      "epoch": 1.1110305786132813e-05,
      "model_forward_time": 1.7408578395843506,
      "step": 11650
    },
    {
      "dynamics_loss_avg": 0.06593964286148549,
      "epoch": 1.1110305786132813e-05,
      "step": 11650
    },
    {
      "action_loss_avg": 0.00303823210997507,
      "epoch": 1.1110305786132813e-05,
      "step": 11650
    },
    {
      "epoch": 1.1110305786132813e-05,
      "step": 11650,
      "training_step_time": 6.808286428451538
    },
    {
      "epoch": 1.1111259460449219e-05,
      "model_forward_time": 2.307250738143921,
      "step": 11651
    },
    {
      "epoch": 1.1111259460449219e-05,
      "step": 11651,
      "training_step_time": 9.688498973846436
    },
    {
      "epoch": 1.1112213134765624e-05,
      "model_forward_time": 1.7488384246826172,
      "step": 11652
    },
    {
      "epoch": 1.1112213134765624e-05,
      "step": 11652,
      "training_step_time": 7.796345472335815
    },
    {
      "epoch": 1.1113166809082032e-05,
      "model_forward_time": 1.745027780532837,
      "step": 11653
    },
    {
      "epoch": 1.1113166809082032e-05,
      "step": 11653,
      "training_step_time": 6.791192293167114
    },
    {
      "epoch": 1.1114120483398437e-05,
      "model_forward_time": 1.7473912239074707,
      "step": 11654
    },
    {
      "epoch": 1.1114120483398437e-05,
      "step": 11654,
      "training_step_time": 6.8406572341918945
    },
    {
      "epoch": 1.1115074157714843e-05,
      "model_forward_time": 1.7498369216918945,
      "step": 11655
    },
    {
      "epoch": 1.1115074157714843e-05,
      "step": 11655,
      "training_step_time": 6.87049412727356
    },
    {
      "epoch": 1.111602783203125e-05,
      "model_forward_time": 1.7325375080108643,
      "step": 11656
    },
    {
      "epoch": 1.111602783203125e-05,
      "step": 11656,
      "training_step_time": 6.81555700302124
    },
    {
      "epoch": 1.1116981506347656e-05,
      "model_forward_time": 1.7495076656341553,
      "step": 11657
    },
    {
      "epoch": 1.1116981506347656e-05,
      "step": 11657,
      "training_step_time": 6.8313822746276855
    },
    {
      "epoch": 1.1117935180664062e-05,
      "model_forward_time": 1.7442545890808105,
      "step": 11658
    },
    {
      "epoch": 1.1117935180664062e-05,
      "step": 11658,
      "training_step_time": 6.847360849380493
    },
    {
      "epoch": 1.111888885498047e-05,
      "model_forward_time": 1.7422635555267334,
      "step": 11659
    },
    {
      "epoch": 1.111888885498047e-05,
      "step": 11659,
      "training_step_time": 6.800675392150879
    },
    {
      "epoch": 1.1119842529296875e-05,
      "grad_norm": 0.1957484781742096,
      "learning_rate": 9.879258969641809e-06,
      "loss": 0.0707,
      "step": 11660
    },
    {
      "epoch": 1.1119842529296875e-05,
      "model_forward_time": 2.3335983753204346,
      "step": 11660
    },
    {
      "dynamics_loss_avg": 0.06476664282381535,
      "epoch": 1.1119842529296875e-05,
      "step": 11660
    },
    {
      "action_loss_avg": 0.004264157637953759,
      "epoch": 1.1119842529296875e-05,
      "step": 11660
    },
    {
      "epoch": 1.1119842529296875e-05,
      "step": 11660,
      "training_step_time": 7.577644348144531
    },
    {
      "epoch": 1.1120796203613281e-05,
      "model_forward_time": 1.7557034492492676,
      "step": 11661
    },
    {
      "epoch": 1.1120796203613281e-05,
      "step": 11661,
      "training_step_time": 7.8531575202941895
    },
    {
      "epoch": 1.1121749877929688e-05,
      "model_forward_time": 1.774759292602539,
      "step": 11662
    },
    {
      "epoch": 1.1121749877929688e-05,
      "step": 11662,
      "training_step_time": 7.634408950805664
    },
    {
      "epoch": 1.1122703552246094e-05,
      "model_forward_time": 1.7501482963562012,
      "step": 11663
    },
    {
      "epoch": 1.1122703552246094e-05,
      "step": 11663,
      "training_step_time": 6.799132823944092
    },
    {
      "epoch": 1.11236572265625e-05,
      "model_forward_time": 1.7507281303405762,
      "step": 11664
    },
    {
      "epoch": 1.11236572265625e-05,
      "step": 11664,
      "training_step_time": 6.8149003982543945
    },
    {
      "epoch": 1.1124610900878905e-05,
      "model_forward_time": 1.7513656616210938,
      "step": 11665
    },
    {
      "epoch": 1.1124610900878905e-05,
      "step": 11665,
      "training_step_time": 6.800435781478882
    },
    {
      "epoch": 1.1125564575195313e-05,
      "model_forward_time": 1.7652802467346191,
      "step": 11666
    },
    {
      "epoch": 1.1125564575195313e-05,
      "step": 11666,
      "training_step_time": 6.814442157745361
    },
    {
      "epoch": 1.1126518249511719e-05,
      "model_forward_time": 1.3880822658538818,
      "step": 11667
    },
    {
      "epoch": 1.1126518249511719e-05,
      "step": 11667,
      "training_step_time": 6.817824125289917
    },
    {
      "epoch": 1.1127471923828124e-05,
      "model_forward_time": 1.7573270797729492,
      "step": 11668
    },
    {
      "epoch": 1.1127471923828124e-05,
      "step": 11668,
      "training_step_time": 6.814882040023804
    },
    {
      "epoch": 1.1128425598144532e-05,
      "model_forward_time": 1.7724511623382568,
      "step": 11669
    },
    {
      "epoch": 1.1128425598144532e-05,
      "step": 11669,
      "training_step_time": 7.337831735610962
    },
    {
      "epoch": 1.1129379272460937e-05,
      "grad_norm": 0.29255712032318115,
      "learning_rate": 9.878897529343023e-06,
      "loss": 0.0704,
      "step": 11670
    },
    {
      "epoch": 1.1129379272460937e-05,
      "model_forward_time": 2.3320014476776123,
      "step": 11670
    },
    {
      "dynamics_loss_avg": 0.06855512410402298,
      "epoch": 1.1129379272460937e-05,
      "step": 11670
    },
    {
      "action_loss_avg": 0.003377303248271346,
      "epoch": 1.1129379272460937e-05,
      "step": 11670
    },
    {
      "epoch": 1.1129379272460937e-05,
      "step": 11670,
      "training_step_time": 8.854514122009277
    },
    {
      "epoch": 1.1130332946777343e-05,
      "model_forward_time": 1.745546817779541,
      "step": 11671
    },
    {
      "epoch": 1.1130332946777343e-05,
      "step": 11671,
      "training_step_time": 7.630418300628662
    },
    {
      "epoch": 1.113128662109375e-05,
      "model_forward_time": 1.7376670837402344,
      "step": 11672
    },
    {
      "epoch": 1.113128662109375e-05,
      "step": 11672,
      "training_step_time": 6.816522598266602
    },
    {
      "epoch": 1.1132240295410156e-05,
      "model_forward_time": 1.7443692684173584,
      "step": 11673
    },
    {
      "epoch": 1.1132240295410156e-05,
      "step": 11673,
      "training_step_time": 6.809803247451782
    },
    {
      "epoch": 1.1133193969726562e-05,
      "model_forward_time": 1.7486581802368164,
      "step": 11674
    },
    {
      "epoch": 1.1133193969726562e-05,
      "step": 11674,
      "training_step_time": 6.857786178588867
    },
    {
      "epoch": 1.113414764404297e-05,
      "model_forward_time": 1.3732671737670898,
      "step": 11675
    },
    {
      "epoch": 1.113414764404297e-05,
      "step": 11675,
      "training_step_time": 6.846083164215088
    },
    {
      "epoch": 1.1135101318359375e-05,
      "model_forward_time": 1.747528076171875,
      "step": 11676
    },
    {
      "epoch": 1.1135101318359375e-05,
      "step": 11676,
      "training_step_time": 6.933377981185913
    },
    {
      "epoch": 1.1136054992675781e-05,
      "model_forward_time": 1.7469768524169922,
      "step": 11677
    },
    {
      "epoch": 1.1136054992675781e-05,
      "step": 11677,
      "training_step_time": 6.854628086090088
    },
    {
      "epoch": 1.1137008666992188e-05,
      "model_forward_time": 1.7661728858947754,
      "step": 11678
    },
    {
      "epoch": 1.1137008666992188e-05,
      "step": 11678,
      "training_step_time": 6.868671894073486
    },
    {
      "epoch": 1.1137962341308594e-05,
      "model_forward_time": 2.3410274982452393,
      "step": 11679
    },
    {
      "epoch": 1.1137962341308594e-05,
      "step": 11679,
      "training_step_time": 11.706878662109375
    },
    {
      "epoch": 1.1138916015625e-05,
      "grad_norm": 0.2554081678390503,
      "learning_rate": 9.87853555549534e-06,
      "loss": 0.0666,
      "step": 11680
    },
    {
      "epoch": 1.1138916015625e-05,
      "model_forward_time": 1.7423803806304932,
      "step": 11680
    },
    {
      "dynamics_loss_avg": 0.056180972047150134,
      "epoch": 1.1138916015625e-05,
      "step": 11680
    },
    {
      "action_loss_avg": 0.0027036476880311967,
      "epoch": 1.1138916015625e-05,
      "step": 11680
    },
    {
      "epoch": 1.1138916015625e-05,
      "step": 11680,
      "training_step_time": 7.754912614822388
    },
    {
      "epoch": 1.1139869689941405e-05,
      "model_forward_time": 1.7388031482696533,
      "step": 11681
    },
    {
      "epoch": 1.1139869689941405e-05,
      "step": 11681,
      "training_step_time": 7.651137828826904
    },
    {
      "epoch": 1.1140823364257813e-05,
      "model_forward_time": 1.747426986694336,
      "step": 11682
    },
    {
      "epoch": 1.1140823364257813e-05,
      "step": 11682,
      "training_step_time": 6.797415733337402
    },
    {
      "epoch": 1.1141777038574219e-05,
      "model_forward_time": 1.7444391250610352,
      "step": 11683
    },
    {
      "epoch": 1.1141777038574219e-05,
      "step": 11683,
      "training_step_time": 6.850879907608032
    },
    {
      "epoch": 1.1142730712890624e-05,
      "model_forward_time": 1.7458858489990234,
      "step": 11684
    },
    {
      "epoch": 1.1142730712890624e-05,
      "step": 11684,
      "training_step_time": 6.835462808609009
    },
    {
      "epoch": 1.1143684387207032e-05,
      "model_forward_time": 1.732055902481079,
      "step": 11685
    },
    {
      "epoch": 1.1143684387207032e-05,
      "step": 11685,
      "training_step_time": 6.714252471923828
    },
    {
      "epoch": 1.1144638061523437e-05,
      "model_forward_time": 1.7422206401824951,
      "step": 11686
    },
    {
      "epoch": 1.1144638061523437e-05,
      "step": 11686,
      "training_step_time": 6.840490818023682
    },
    {
      "epoch": 1.1145591735839843e-05,
      "model_forward_time": 1.0365688800811768,
      "step": 11687
    },
    {
      "epoch": 1.1145591735839843e-05,
      "step": 11687,
      "training_step_time": 6.829232692718506
    },
    {
      "epoch": 1.114654541015625e-05,
      "model_forward_time": 1.7510056495666504,
      "step": 11688
    },
    {
      "epoch": 1.114654541015625e-05,
      "step": 11688,
      "training_step_time": 10.533477544784546
    },
    {
      "epoch": 1.1147499084472656e-05,
      "model_forward_time": 1.7506468296051025,
      "step": 11689
    },
    {
      "epoch": 1.1147499084472656e-05,
      "step": 11689,
      "training_step_time": 7.839369535446167
    },
    {
      "epoch": 1.1148452758789062e-05,
      "grad_norm": 0.19376212358474731,
      "learning_rate": 9.87817304813834e-06,
      "loss": 0.0613,
      "step": 11690
    },
    {
      "epoch": 1.1148452758789062e-05,
      "model_forward_time": 1.7515063285827637,
      "step": 11690
    },
    {
      "dynamics_loss_avg": 0.052025698125362396,
      "epoch": 1.1148452758789062e-05,
      "step": 11690
    },
    {
      "action_loss_avg": 0.003399557957891375,
      "epoch": 1.1148452758789062e-05,
      "step": 11690
    },
    {
      "epoch": 1.1148452758789062e-05,
      "step": 11690,
      "training_step_time": 7.776224613189697
    },
    {
      "epoch": 1.114940643310547e-05,
      "model_forward_time": 1.74214768409729,
      "step": 11691
    },
    {
      "epoch": 1.114940643310547e-05,
      "step": 11691,
      "training_step_time": 6.847206115722656
    },
    {
      "epoch": 1.1150360107421875e-05,
      "model_forward_time": 1.7473163604736328,
      "step": 11692
    },
    {
      "epoch": 1.1150360107421875e-05,
      "step": 11692,
      "training_step_time": 6.851591110229492
    },
    {
      "epoch": 1.1151313781738281e-05,
      "model_forward_time": 1.7548284530639648,
      "step": 11693
    },
    {
      "epoch": 1.1151313781738281e-05,
      "step": 11693,
      "training_step_time": 6.8465330600738525
    },
    {
      "epoch": 1.1152267456054688e-05,
      "model_forward_time": 1.762716293334961,
      "step": 11694
    },
    {
      "epoch": 1.1152267456054688e-05,
      "step": 11694,
      "training_step_time": 6.810179710388184
    },
    {
      "epoch": 1.1153221130371094e-05,
      "model_forward_time": 1.7683262825012207,
      "step": 11695
    },
    {
      "epoch": 1.1153221130371094e-05,
      "step": 11695,
      "training_step_time": 6.870349407196045
    },
    {
      "epoch": 1.11541748046875e-05,
      "model_forward_time": 1.7547547817230225,
      "step": 11696
    },
    {
      "epoch": 1.11541748046875e-05,
      "step": 11696,
      "training_step_time": 6.865119695663452
    },
    {
      "epoch": 1.1155128479003905e-05,
      "model_forward_time": 1.7574405670166016,
      "step": 11697
    },
    {
      "epoch": 1.1155128479003905e-05,
      "step": 11697,
      "training_step_time": 11.697868347167969
    },
    {
      "epoch": 1.1156082153320313e-05,
      "model_forward_time": 2.328474283218384,
      "step": 11698
    },
    {
      "epoch": 1.1156082153320313e-05,
      "step": 11698,
      "training_step_time": 7.7971274852752686
    },
    {
      "epoch": 1.1157035827636719e-05,
      "model_forward_time": 1.7530360221862793,
      "step": 11699
    },
    {
      "epoch": 1.1157035827636719e-05,
      "step": 11699,
      "training_step_time": 7.72692084312439
    },
    {
      "epoch": 1.1157989501953124e-05,
      "grad_norm": 0.16891518235206604,
      "learning_rate": 9.877810007311673e-06,
      "loss": 0.063,
      "step": 11700
    },
    {
      "epoch": 1.1157989501953124e-05,
      "model_forward_time": 1.7668449878692627,
      "step": 11700
    },
    {
      "dynamics_loss_avg": 0.06648631580173969,
      "epoch": 1.1157989501953124e-05,
      "step": 11700
    },
    {
      "action_loss_avg": 0.0028016701806336643,
      "epoch": 1.1157989501953124e-05,
      "step": 11700
    },
    {
      "epoch": 1.1157989501953124e-05,
      "step": 11700,
      "training_step_time": 6.834900379180908
    },
    {
      "epoch": 1.1158943176269532e-05,
      "model_forward_time": 1.0585741996765137,
      "step": 11701
    },
    {
      "epoch": 1.1158943176269532e-05,
      "step": 11701,
      "training_step_time": 6.858693361282349
    },
    {
      "epoch": 1.1159896850585937e-05,
      "model_forward_time": 1.7622714042663574,
      "step": 11702
    },
    {
      "epoch": 1.1159896850585937e-05,
      "step": 11702,
      "training_step_time": 6.874696969985962
    },
    {
      "epoch": 1.1160850524902343e-05,
      "model_forward_time": 1.7563567161560059,
      "step": 11703
    },
    {
      "epoch": 1.1160850524902343e-05,
      "step": 11703,
      "training_step_time": 6.856722831726074
    },
    {
      "epoch": 1.116180419921875e-05,
      "model_forward_time": 1.7559218406677246,
      "step": 11704
    },
    {
      "epoch": 1.116180419921875e-05,
      "step": 11704,
      "training_step_time": 6.836469888687134
    },
    {
      "epoch": 1.1162757873535156e-05,
      "model_forward_time": 1.7593576908111572,
      "step": 11705
    },
    {
      "epoch": 1.1162757873535156e-05,
      "step": 11705,
      "training_step_time": 6.923250198364258
    },
    {
      "epoch": 1.1163711547851562e-05,
      "model_forward_time": 1.7449729442596436,
      "step": 11706
    },
    {
      "epoch": 1.1163711547851562e-05,
      "step": 11706,
      "training_step_time": 10.737302541732788
    },
    {
      "epoch": 1.116466522216797e-05,
      "model_forward_time": 2.3262972831726074,
      "step": 11707
    },
    {
      "epoch": 1.116466522216797e-05,
      "step": 11707,
      "training_step_time": 8.391165494918823
    },
    {
      "epoch": 1.1165618896484375e-05,
      "model_forward_time": 1.7778677940368652,
      "step": 11708
    },
    {
      "epoch": 1.1165618896484375e-05,
      "step": 11708,
      "training_step_time": 7.707507610321045
    },
    {
      "epoch": 1.1166572570800781e-05,
      "model_forward_time": 1.7793848514556885,
      "step": 11709
    },
    {
      "epoch": 1.1166572570800781e-05,
      "step": 11709,
      "training_step_time": 6.837011814117432
    },
    {
      "epoch": 1.1167526245117188e-05,
      "grad_norm": 0.21885347366333008,
      "learning_rate": 9.877446433055036e-06,
      "loss": 0.0678,
      "step": 11710
    },
    {
      "epoch": 1.1167526245117188e-05,
      "model_forward_time": 1.7644095420837402,
      "step": 11710
    },
    {
      "dynamics_loss_avg": 0.06991640571504831,
      "epoch": 1.1167526245117188e-05,
      "step": 11710
    },
    {
      "action_loss_avg": 0.0027779402676969766,
      "epoch": 1.1167526245117188e-05,
      "step": 11710
    },
    {
      "epoch": 1.1167526245117188e-05,
      "step": 11710,
      "training_step_time": 6.825543403625488
    },
    {
      "epoch": 1.1168479919433594e-05,
      "model_forward_time": 1.7552461624145508,
      "step": 11711
    },
    {
      "epoch": 1.1168479919433594e-05,
      "step": 11711,
      "training_step_time": 6.85491943359375
    },
    {
      "epoch": 1.116943359375e-05,
      "model_forward_time": 1.757624864578247,
      "step": 11712
    },
    {
      "epoch": 1.116943359375e-05,
      "step": 11712,
      "training_step_time": 6.8962624073028564
    },
    {
      "epoch": 1.1170387268066405e-05,
      "model_forward_time": 1.393068790435791,
      "step": 11713
    },
    {
      "epoch": 1.1170387268066405e-05,
      "step": 11713,
      "training_step_time": 6.858884334564209
    },
    {
      "epoch": 1.1171340942382813e-05,
      "model_forward_time": 1.7568869590759277,
      "step": 11714
    },
    {
      "epoch": 1.1171340942382813e-05,
      "step": 11714,
      "training_step_time": 6.820031404495239
    },
    {
      "epoch": 1.1172294616699219e-05,
      "model_forward_time": 1.7596263885498047,
      "step": 11715
    },
    {
      "epoch": 1.1172294616699219e-05,
      "step": 11715,
      "training_step_time": 8.45382833480835
    },
    {
      "epoch": 1.1173248291015624e-05,
      "model_forward_time": 2.3068010807037354,
      "step": 11716
    },
    {
      "epoch": 1.1173248291015624e-05,
      "step": 11716,
      "training_step_time": 8.381504535675049
    },
    {
      "epoch": 1.1174201965332032e-05,
      "model_forward_time": 1.7546486854553223,
      "step": 11717
    },
    {
      "epoch": 1.1174201965332032e-05,
      "step": 11717,
      "training_step_time": 7.895503044128418
    },
    {
      "epoch": 1.1175155639648437e-05,
      "model_forward_time": 1.7798125743865967,
      "step": 11718
    },
    {
      "epoch": 1.1175155639648437e-05,
      "step": 11718,
      "training_step_time": 7.7735912799835205
    },
    {
      "epoch": 1.1176109313964843e-05,
      "model_forward_time": 1.7389733791351318,
      "step": 11719
    },
    {
      "epoch": 1.1176109313964843e-05,
      "step": 11719,
      "training_step_time": 6.89316987991333
    },
    {
      "epoch": 1.117706298828125e-05,
      "grad_norm": 0.25203102827072144,
      "learning_rate": 9.877082325408191e-06,
      "loss": 0.0603,
      "step": 11720
    },
    {
      "epoch": 1.117706298828125e-05,
      "model_forward_time": 1.74015474319458,
      "step": 11720
    },
    {
      "dynamics_loss_avg": 0.035667916759848595,
      "epoch": 1.117706298828125e-05,
      "step": 11720
    },
    {
      "action_loss_avg": 0.00245721606479492,
      "epoch": 1.117706298828125e-05,
      "step": 11720
    },
    {
      "epoch": 1.117706298828125e-05,
      "step": 11720,
      "training_step_time": 6.965500831604004
    },
    {
      "epoch": 1.1178016662597656e-05,
      "model_forward_time": 1.7581844329833984,
      "step": 11721
    },
    {
      "epoch": 1.1178016662597656e-05,
      "step": 11721,
      "training_step_time": 6.939129590988159
    },
    {
      "epoch": 1.1178970336914062e-05,
      "model_forward_time": 1.748502492904663,
      "step": 11722
    },
    {
      "epoch": 1.1178970336914062e-05,
      "step": 11722,
      "training_step_time": 7.040335178375244
    },
    {
      "epoch": 1.117992401123047e-05,
      "model_forward_time": 1.7402141094207764,
      "step": 11723
    },
    {
      "epoch": 1.117992401123047e-05,
      "step": 11723,
      "training_step_time": 6.98288369178772
    },
    {
      "epoch": 1.1180877685546875e-05,
      "model_forward_time": 1.739835500717163,
      "step": 11724
    },
    {
      "epoch": 1.1180877685546875e-05,
      "step": 11724,
      "training_step_time": 8.602022886276245
    },
    {
      "epoch": 1.1181831359863281e-05,
      "model_forward_time": 1.7545571327209473,
      "step": 11725
    },
    {
      "epoch": 1.1181831359863281e-05,
      "step": 11725,
      "training_step_time": 11.379552841186523
    },
    {
      "epoch": 1.1182785034179688e-05,
      "model_forward_time": 2.300269365310669,
      "step": 11726
    },
    {
      "epoch": 1.1182785034179688e-05,
      "step": 11726,
      "training_step_time": 7.91693639755249
    },
    {
      "epoch": 1.1183738708496094e-05,
      "model_forward_time": 1.7823765277862549,
      "step": 11727
    },
    {
      "epoch": 1.1183738708496094e-05,
      "step": 11727,
      "training_step_time": 7.846893310546875
    },
    {
      "epoch": 1.11846923828125e-05,
      "model_forward_time": 1.7419652938842773,
      "step": 11728
    },
    {
      "epoch": 1.11846923828125e-05,
      "step": 11728,
      "training_step_time": 6.920532464981079
    },
    {
      "epoch": 1.1185646057128905e-05,
      "model_forward_time": 1.7539610862731934,
      "step": 11729
    },
    {
      "epoch": 1.1185646057128905e-05,
      "step": 11729,
      "training_step_time": 6.903272867202759
    },
    {
      "epoch": 1.1186599731445313e-05,
      "grad_norm": 0.22418327629566193,
      "learning_rate": 9.876717684410954e-06,
      "loss": 0.0675,
      "step": 11730
    },
    {
      "epoch": 1.1186599731445313e-05,
      "model_forward_time": 1.7498929500579834,
      "step": 11730
    },
    {
      "dynamics_loss_avg": 0.06403925772756339,
      "epoch": 1.1186599731445313e-05,
      "step": 11730
    },
    {
      "action_loss_avg": 0.0029262733296491206,
      "epoch": 1.1186599731445313e-05,
      "step": 11730
    },
    {
      "epoch": 1.1186599731445313e-05,
      "step": 11730,
      "training_step_time": 6.929746866226196
    },
    {
      "epoch": 1.1187553405761719e-05,
      "model_forward_time": 1.7536253929138184,
      "step": 11731
    },
    {
      "epoch": 1.1187553405761719e-05,
      "step": 11731,
      "training_step_time": 6.947705268859863
    },
    {
      "epoch": 1.1188507080078124e-05,
      "model_forward_time": 1.7590649127960205,
      "step": 11732
    },
    {
      "epoch": 1.1188507080078124e-05,
      "step": 11732,
      "training_step_time": 6.930096626281738
    },
    {
      "epoch": 1.1189460754394532e-05,
      "model_forward_time": 1.755066156387329,
      "step": 11733
    },
    {
      "epoch": 1.1189460754394532e-05,
      "step": 11733,
      "training_step_time": 7.926672697067261
    },
    {
      "epoch": 1.1190414428710937e-05,
      "model_forward_time": 1.7589397430419922,
      "step": 11734
    },
    {
      "epoch": 1.1190414428710937e-05,
      "step": 11734,
      "training_step_time": 10.277399778366089
    },
    {
      "epoch": 1.1191368103027343e-05,
      "model_forward_time": 2.3389453887939453,
      "step": 11735
    },
    {
      "epoch": 1.1191368103027343e-05,
      "step": 11735,
      "training_step_time": 8.681101560592651
    },
    {
      "epoch": 1.119232177734375e-05,
      "model_forward_time": 1.7513911724090576,
      "step": 11736
    },
    {
      "epoch": 1.119232177734375e-05,
      "step": 11736,
      "training_step_time": 7.8029632568359375
    },
    {
      "epoch": 1.1193275451660156e-05,
      "model_forward_time": 1.0462121963500977,
      "step": 11737
    },
    {
      "epoch": 1.1193275451660156e-05,
      "step": 11737,
      "training_step_time": 6.925302028656006
    },
    {
      "epoch": 1.1194229125976562e-05,
      "model_forward_time": 1.7683093547821045,
      "step": 11738
    },
    {
      "epoch": 1.1194229125976562e-05,
      "step": 11738,
      "training_step_time": 6.942389249801636
    },
    {
      "epoch": 1.119518280029297e-05,
      "model_forward_time": 1.7753190994262695,
      "step": 11739
    },
    {
      "epoch": 1.119518280029297e-05,
      "step": 11739,
      "training_step_time": 6.932759523391724
    },
    {
      "epoch": 1.1196136474609375e-05,
      "grad_norm": 0.2380160689353943,
      "learning_rate": 9.876352510103205e-06,
      "loss": 0.0642,
      "step": 11740
    },
    {
      "epoch": 1.1196136474609375e-05,
      "model_forward_time": 1.7576649188995361,
      "step": 11740
    },
    {
      "dynamics_loss_avg": 0.0491898350417614,
      "epoch": 1.1196136474609375e-05,
      "step": 11740
    },
    {
      "action_loss_avg": 0.0030726109631359576,
      "epoch": 1.1196136474609375e-05,
      "step": 11740
    },
    {
      "epoch": 1.1196136474609375e-05,
      "step": 11740,
      "training_step_time": 6.930463552474976
    },
    {
      "epoch": 1.1197090148925781e-05,
      "model_forward_time": 1.7477061748504639,
      "step": 11741
    },
    {
      "epoch": 1.1197090148925781e-05,
      "step": 11741,
      "training_step_time": 6.866687536239624
    },
    {
      "epoch": 1.1198043823242188e-05,
      "model_forward_time": 1.3953642845153809,
      "step": 11742
    },
    {
      "epoch": 1.1198043823242188e-05,
      "step": 11742,
      "training_step_time": 7.769047260284424
    },
    {
      "epoch": 1.1198997497558594e-05,
      "model_forward_time": 1.7645037174224854,
      "step": 11743
    },
    {
      "epoch": 1.1198997497558594e-05,
      "step": 11743,
      "training_step_time": 10.87703013420105
    },
    {
      "epoch": 1.1199951171875e-05,
      "model_forward_time": 2.3124189376831055,
      "step": 11744
    },
    {
      "epoch": 1.1199951171875e-05,
      "step": 11744,
      "training_step_time": 9.570145845413208
    },
    {
      "epoch": 1.1200904846191406e-05,
      "model_forward_time": 1.7799656391143799,
      "step": 11745
    },
    {
      "epoch": 1.1200904846191406e-05,
      "step": 11745,
      "training_step_time": 7.868956565856934
    },
    {
      "epoch": 1.1201858520507813e-05,
      "model_forward_time": 1.740410566329956,
      "step": 11746
    },
    {
      "epoch": 1.1201858520507813e-05,
      "step": 11746,
      "training_step_time": 7.736169099807739
    },
    {
      "epoch": 1.1202812194824219e-05,
      "model_forward_time": 1.7410597801208496,
      "step": 11747
    },
    {
      "epoch": 1.1202812194824219e-05,
      "step": 11747,
      "training_step_time": 6.885847330093384
    },
    {
      "epoch": 1.1203765869140624e-05,
      "model_forward_time": 1.7357144355773926,
      "step": 11748
    },
    {
      "epoch": 1.1203765869140624e-05,
      "step": 11748,
      "training_step_time": 6.7666239738464355
    },
    {
      "epoch": 1.1204719543457032e-05,
      "model_forward_time": 1.7354331016540527,
      "step": 11749
    },
    {
      "epoch": 1.1204719543457032e-05,
      "step": 11749,
      "training_step_time": 6.893269777297974
    },
    {
      "epoch": 1.1205673217773438e-05,
      "grad_norm": 0.19761115312576294,
      "learning_rate": 9.875986802524874e-06,
      "loss": 0.0669,
      "step": 11750
    },
    {
      "epoch": 1.1205673217773438e-05,
      "model_forward_time": 1.7387287616729736,
      "step": 11750
    },
    {
      "dynamics_loss_avg": 0.05322931996779516,
      "epoch": 1.1205673217773438e-05,
      "step": 11750
    },
    {
      "action_loss_avg": 0.0026165931223658843,
      "epoch": 1.1205673217773438e-05,
      "step": 11750
    },
    {
      "epoch": 1.1205673217773438e-05,
      "step": 11750,
      "training_step_time": 6.849825859069824
    },
    {
      "epoch": 1.1206626892089843e-05,
      "model_forward_time": 1.037684440612793,
      "step": 11751
    },
    {
      "epoch": 1.1206626892089843e-05,
      "step": 11751,
      "training_step_time": 7.907425880432129
    },
    {
      "epoch": 1.120758056640625e-05,
      "model_forward_time": 1.7368724346160889,
      "step": 11752
    },
    {
      "epoch": 1.120758056640625e-05,
      "step": 11752,
      "training_step_time": 8.525688409805298
    },
    {
      "epoch": 1.1208534240722656e-05,
      "model_forward_time": 1.7390966415405273,
      "step": 11753
    },
    {
      "epoch": 1.1208534240722656e-05,
      "step": 11753,
      "training_step_time": 8.846725940704346
    },
    {
      "epoch": 1.1209487915039062e-05,
      "model_forward_time": 2.2895917892456055,
      "step": 11754
    },
    {
      "epoch": 1.1209487915039062e-05,
      "step": 11754,
      "training_step_time": 7.7410502433776855
    },
    {
      "epoch": 1.121044158935547e-05,
      "model_forward_time": 1.7367351055145264,
      "step": 11755
    },
    {
      "epoch": 1.121044158935547e-05,
      "step": 11755,
      "training_step_time": 7.762184143066406
    },
    {
      "epoch": 1.1211395263671875e-05,
      "model_forward_time": 1.7498176097869873,
      "step": 11756
    },
    {
      "epoch": 1.1211395263671875e-05,
      "step": 11756,
      "training_step_time": 6.951783895492554
    },
    {
      "epoch": 1.1212348937988281e-05,
      "model_forward_time": 1.7527580261230469,
      "step": 11757
    },
    {
      "epoch": 1.1212348937988281e-05,
      "step": 11757,
      "training_step_time": 6.969943523406982
    },
    {
      "epoch": 1.1213302612304687e-05,
      "model_forward_time": 1.7569751739501953,
      "step": 11758
    },
    {
      "epoch": 1.1213302612304687e-05,
      "step": 11758,
      "training_step_time": 6.974876403808594
    },
    {
      "epoch": 1.1214256286621094e-05,
      "model_forward_time": 1.7576987743377686,
      "step": 11759
    },
    {
      "epoch": 1.1214256286621094e-05,
      "step": 11759,
      "training_step_time": 6.9710071086883545
    },
    {
      "epoch": 1.12152099609375e-05,
      "grad_norm": 0.23108148574829102,
      "learning_rate": 9.875620561715959e-06,
      "loss": 0.0684,
      "step": 11760
    },
    {
      "epoch": 1.12152099609375e-05,
      "model_forward_time": 1.7543795108795166,
      "step": 11760
    },
    {
      "dynamics_loss_avg": 0.059141626581549644,
      "epoch": 1.12152099609375e-05,
      "step": 11760
    },
    {
      "action_loss_avg": 0.003999186778673902,
      "epoch": 1.12152099609375e-05,
      "step": 11760
    },
    {
      "epoch": 1.12152099609375e-05,
      "step": 11760,
      "training_step_time": 7.8714282512664795
    },
    {
      "epoch": 1.1216163635253906e-05,
      "model_forward_time": 1.7651257514953613,
      "step": 11761
    },
    {
      "epoch": 1.1216163635253906e-05,
      "step": 11761,
      "training_step_time": 8.56797456741333
    },
    {
      "epoch": 1.1217117309570313e-05,
      "model_forward_time": 1.3882758617401123,
      "step": 11762
    },
    {
      "epoch": 1.1217117309570313e-05,
      "step": 11762,
      "training_step_time": 8.402172565460205
    },
    {
      "epoch": 1.1218070983886719e-05,
      "model_forward_time": 2.3350841999053955,
      "step": 11763
    },
    {
      "epoch": 1.1218070983886719e-05,
      "step": 11763,
      "training_step_time": 7.858465194702148
    },
    {
      "epoch": 1.1219024658203124e-05,
      "model_forward_time": 1.751188039779663,
      "step": 11764
    },
    {
      "epoch": 1.1219024658203124e-05,
      "step": 11764,
      "training_step_time": 7.748036861419678
    },
    {
      "epoch": 1.1219978332519532e-05,
      "model_forward_time": 1.7565937042236328,
      "step": 11765
    },
    {
      "epoch": 1.1219978332519532e-05,
      "step": 11765,
      "training_step_time": 6.912088632583618
    },
    {
      "epoch": 1.1220932006835938e-05,
      "model_forward_time": 1.7645044326782227,
      "step": 11766
    },
    {
      "epoch": 1.1220932006835938e-05,
      "step": 11766,
      "training_step_time": 6.8984644412994385
    },
    {
      "epoch": 1.1221885681152343e-05,
      "model_forward_time": 1.7475786209106445,
      "step": 11767
    },
    {
      "epoch": 1.1221885681152343e-05,
      "step": 11767,
      "training_step_time": 6.901671648025513
    },
    {
      "epoch": 1.122283935546875e-05,
      "model_forward_time": 1.7573351860046387,
      "step": 11768
    },
    {
      "epoch": 1.122283935546875e-05,
      "step": 11768,
      "training_step_time": 6.927744388580322
    },
    {
      "epoch": 1.1223793029785156e-05,
      "model_forward_time": 1.7519292831420898,
      "step": 11769
    },
    {
      "epoch": 1.1223793029785156e-05,
      "step": 11769,
      "training_step_time": 8.919316053390503
    },
    {
      "epoch": 1.1224746704101562e-05,
      "grad_norm": 0.23435431718826294,
      "learning_rate": 9.87525378771651e-06,
      "loss": 0.0691,
      "step": 11770
    },
    {
      "epoch": 1.1224746704101562e-05,
      "model_forward_time": 1.7561306953430176,
      "step": 11770
    },
    {
      "dynamics_loss_avg": 0.059291801787912844,
      "epoch": 1.1224746704101562e-05,
      "step": 11770
    },
    {
      "action_loss_avg": 0.003128190594725311,
      "epoch": 1.1224746704101562e-05,
      "step": 11770
    },
    {
      "epoch": 1.1224746704101562e-05,
      "step": 11770,
      "training_step_time": 8.681321620941162
    },
    {
      "epoch": 1.122570037841797e-05,
      "model_forward_time": 1.749767780303955,
      "step": 11771
    },
    {
      "epoch": 1.122570037841797e-05,
      "step": 11771,
      "training_step_time": 7.799994230270386
    },
    {
      "epoch": 1.1226654052734375e-05,
      "model_forward_time": 1.790501356124878,
      "step": 11772
    },
    {
      "epoch": 1.1226654052734375e-05,
      "step": 11772,
      "training_step_time": 8.266765832901001
    },
    {
      "epoch": 1.1227607727050781e-05,
      "model_forward_time": 1.752960205078125,
      "step": 11773
    },
    {
      "epoch": 1.1227607727050781e-05,
      "step": 11773,
      "training_step_time": 7.636300802230835
    },
    {
      "epoch": 1.1228561401367187e-05,
      "model_forward_time": 1.7549827098846436,
      "step": 11774
    },
    {
      "epoch": 1.1228561401367187e-05,
      "step": 11774,
      "training_step_time": 7.086043357849121
    },
    {
      "epoch": 1.1229515075683594e-05,
      "model_forward_time": 1.7588167190551758,
      "step": 11775
    },
    {
      "epoch": 1.1229515075683594e-05,
      "step": 11775,
      "training_step_time": 6.8032917976379395
    },
    {
      "epoch": 1.123046875e-05,
      "model_forward_time": 1.7643380165100098,
      "step": 11776
    },
    {
      "epoch": 1.123046875e-05,
      "step": 11776,
      "training_step_time": 6.757532596588135
    },
    {
      "epoch": 1.1231422424316406e-05,
      "model_forward_time": 1.7738380432128906,
      "step": 11777
    },
    {
      "epoch": 1.1231422424316406e-05,
      "step": 11777,
      "training_step_time": 6.778248310089111
    },
    {
      "epoch": 1.1232376098632813e-05,
      "model_forward_time": 1.7349097728729248,
      "step": 11778
    },
    {
      "epoch": 1.1232376098632813e-05,
      "step": 11778,
      "training_step_time": 7.689234018325806
    },
    {
      "epoch": 1.1233329772949219e-05,
      "model_forward_time": 1.7440073490142822,
      "step": 11779
    },
    {
      "epoch": 1.1233329772949219e-05,
      "step": 11779,
      "training_step_time": 8.707597970962524
    },
    {
      "epoch": 1.1234283447265624e-05,
      "grad_norm": 0.23664027452468872,
      "learning_rate": 9.874886480566637e-06,
      "loss": 0.0726,
      "step": 11780
    },
    {
      "epoch": 1.1234283447265624e-05,
      "model_forward_time": 1.72446870803833,
      "step": 11780
    },
    {
      "dynamics_loss_avg": 0.061426757276058196,
      "epoch": 1.1234283447265624e-05,
      "step": 11780
    },
    {
      "action_loss_avg": 0.00288850330398418,
      "epoch": 1.1234283447265624e-05,
      "step": 11780
    },
    {
      "epoch": 1.1234283447265624e-05,
      "step": 11780,
      "training_step_time": 8.59327507019043
    },
    {
      "epoch": 1.1235237121582032e-05,
      "model_forward_time": 1.7605071067810059,
      "step": 11781
    },
    {
      "epoch": 1.1235237121582032e-05,
      "step": 11781,
      "training_step_time": 8.75169324874878
    },
    {
      "epoch": 1.1236190795898438e-05,
      "model_forward_time": 2.2906501293182373,
      "step": 11782
    },
    {
      "epoch": 1.1236190795898438e-05,
      "step": 11782,
      "training_step_time": 7.721539735794067
    },
    {
      "epoch": 1.1237144470214843e-05,
      "model_forward_time": 1.7426223754882812,
      "step": 11783
    },
    {
      "epoch": 1.1237144470214843e-05,
      "step": 11783,
      "training_step_time": 7.670739650726318
    },
    {
      "epoch": 1.123809814453125e-05,
      "model_forward_time": 1.741706371307373,
      "step": 11784
    },
    {
      "epoch": 1.123809814453125e-05,
      "step": 11784,
      "training_step_time": 6.856709718704224
    },
    {
      "epoch": 1.1239051818847656e-05,
      "model_forward_time": 1.7448198795318604,
      "step": 11785
    },
    {
      "epoch": 1.1239051818847656e-05,
      "step": 11785,
      "training_step_time": 6.877439737319946
    },
    {
      "epoch": 1.1240005493164062e-05,
      "model_forward_time": 1.7190778255462646,
      "step": 11786
    },
    {
      "epoch": 1.1240005493164062e-05,
      "step": 11786,
      "training_step_time": 6.756385087966919
    },
    {
      "epoch": 1.124095916748047e-05,
      "model_forward_time": 1.7543911933898926,
      "step": 11787
    },
    {
      "epoch": 1.124095916748047e-05,
      "step": 11787,
      "training_step_time": 8.592465162277222
    },
    {
      "epoch": 1.1241912841796875e-05,
      "model_forward_time": 1.7627501487731934,
      "step": 11788
    },
    {
      "epoch": 1.1241912841796875e-05,
      "step": 11788,
      "training_step_time": 7.707038879394531
    },
    {
      "epoch": 1.1242866516113281e-05,
      "model_forward_time": 1.7728710174560547,
      "step": 11789
    },
    {
      "epoch": 1.1242866516113281e-05,
      "step": 11789,
      "training_step_time": 9.224689960479736
    },
    {
      "epoch": 1.1243820190429687e-05,
      "grad_norm": 0.24256175756454468,
      "learning_rate": 9.874518640306507e-06,
      "loss": 0.0686,
      "step": 11790
    },
    {
      "epoch": 1.1243820190429687e-05,
      "model_forward_time": 1.7517673969268799,
      "step": 11790
    },
    {
      "dynamics_loss_avg": 0.047971873357892034,
      "epoch": 1.1243820190429687e-05,
      "step": 11790
    },
    {
      "action_loss_avg": 0.0026462696259841323,
      "epoch": 1.1243820190429687e-05,
      "step": 11790
    },
    {
      "epoch": 1.1243820190429687e-05,
      "step": 11790,
      "training_step_time": 9.084957122802734
    },
    {
      "epoch": 1.1244773864746094e-05,
      "model_forward_time": 2.2820260524749756,
      "step": 11791
    },
    {
      "epoch": 1.1244773864746094e-05,
      "step": 11791,
      "training_step_time": 7.723190784454346
    },
    {
      "epoch": 1.12457275390625e-05,
      "model_forward_time": 1.743830919265747,
      "step": 11792
    },
    {
      "epoch": 1.12457275390625e-05,
      "step": 11792,
      "training_step_time": 7.3576271533966064
    },
    {
      "epoch": 1.1246681213378906e-05,
      "model_forward_time": 1.7376694679260254,
      "step": 11793
    },
    {
      "epoch": 1.1246681213378906e-05,
      "step": 11793,
      "training_step_time": 6.85883092880249
    },
    {
      "epoch": 1.1247634887695313e-05,
      "model_forward_time": 1.7487883567810059,
      "step": 11794
    },
    {
      "epoch": 1.1247634887695313e-05,
      "step": 11794,
      "training_step_time": 6.840589761734009
    },
    {
      "epoch": 1.1248588562011719e-05,
      "model_forward_time": 1.7400379180908203,
      "step": 11795
    },
    {
      "epoch": 1.1248588562011719e-05,
      "step": 11795,
      "training_step_time": 6.818489074707031
    },
    {
      "epoch": 1.1249542236328124e-05,
      "model_forward_time": 1.7393219470977783,
      "step": 11796
    },
    {
      "epoch": 1.1249542236328124e-05,
      "step": 11796,
      "training_step_time": 7.700407266616821
    },
    {
      "epoch": 1.1250495910644532e-05,
      "model_forward_time": 1.7309598922729492,
      "step": 11797
    },
    {
      "epoch": 1.1250495910644532e-05,
      "step": 11797,
      "training_step_time": 7.697323799133301
    },
    {
      "epoch": 1.1251449584960938e-05,
      "model_forward_time": 1.7394204139709473,
      "step": 11798
    },
    {
      "epoch": 1.1251449584960938e-05,
      "step": 11798,
      "training_step_time": 8.743489503860474
    },
    {
      "epoch": 1.1252403259277343e-05,
      "model_forward_time": 1.0328423976898193,
      "step": 11799
    },
    {
      "epoch": 1.1252403259277343e-05,
      "step": 11799,
      "training_step_time": 8.899267196655273
    },
    {
      "epoch": 1.125335693359375e-05,
      "grad_norm": 0.2941148281097412,
      "learning_rate": 9.874150266976348e-06,
      "loss": 0.0688,
      "step": 11800
    },
    {
      "epoch": 1.125335693359375e-05,
      "model_forward_time": 1.4370574951171875,
      "step": 11800
    },
    {
      "dynamics_loss_avg": 0.06644241884350777,
      "epoch": 1.125335693359375e-05,
      "step": 11800
    },
    {
      "action_loss_avg": 0.004192249127663672,
      "epoch": 1.125335693359375e-05,
      "step": 11800
    },
    {
      "epoch": 1.125335693359375e-05,
      "step": 11800,
      "training_step_time": 9.383206367492676
    },
    {
      "epoch": 1.1254310607910156e-05,
      "model_forward_time": 1.7285206317901611,
      "step": 11801
    },
    {
      "epoch": 1.1254310607910156e-05,
      "step": 11801,
      "training_step_time": 7.756370544433594
    },
    {
      "epoch": 1.1255264282226562e-05,
      "model_forward_time": 1.7403738498687744,
      "step": 11802
    },
    {
      "epoch": 1.1255264282226562e-05,
      "step": 11802,
      "training_step_time": 6.91978907585144
    },
    {
      "epoch": 1.125621795654297e-05,
      "model_forward_time": 1.7364156246185303,
      "step": 11803
    },
    {
      "epoch": 1.125621795654297e-05,
      "step": 11803,
      "training_step_time": 6.8507044315338135
    },
    {
      "epoch": 1.1257171630859375e-05,
      "model_forward_time": 1.7345378398895264,
      "step": 11804
    },
    {
      "epoch": 1.1257171630859375e-05,
      "step": 11804,
      "training_step_time": 6.876402854919434
    },
    {
      "epoch": 1.1258125305175781e-05,
      "model_forward_time": 1.7322468757629395,
      "step": 11805
    },
    {
      "epoch": 1.1258125305175781e-05,
      "step": 11805,
      "training_step_time": 7.752604722976685
    },
    {
      "epoch": 1.1259078979492187e-05,
      "model_forward_time": 1.7361395359039307,
      "step": 11806
    },
    {
      "epoch": 1.1259078979492187e-05,
      "step": 11806,
      "training_step_time": 7.522129058837891
    },
    {
      "epoch": 1.1260032653808594e-05,
      "model_forward_time": 1.7193853855133057,
      "step": 11807
    },
    {
      "epoch": 1.1260032653808594e-05,
      "step": 11807,
      "training_step_time": 9.118461608886719
    },
    {
      "epoch": 1.1260986328125e-05,
      "model_forward_time": 1.7287132740020752,
      "step": 11808
    },
    {
      "epoch": 1.1260986328125e-05,
      "step": 11808,
      "training_step_time": 9.57674765586853
    },
    {
      "epoch": 1.1261940002441406e-05,
      "model_forward_time": 1.7879552841186523,
      "step": 11809
    },
    {
      "epoch": 1.1261940002441406e-05,
      "step": 11809,
      "training_step_time": 7.823627233505249
    },
    {
      "epoch": 1.1262893676757813e-05,
      "grad_norm": 0.29083314538002014,
      "learning_rate": 9.873781360616444e-06,
      "loss": 0.0656,
      "step": 11810
    },
    {
      "epoch": 1.1262893676757813e-05,
      "model_forward_time": 2.272855520248413,
      "step": 11810
    },
    {
      "dynamics_loss_avg": 0.053114294074475765,
      "epoch": 1.1262893676757813e-05,
      "step": 11810
    },
    {
      "action_loss_avg": 0.002833868231391534,
      "epoch": 1.1262893676757813e-05,
      "step": 11810
    },
    {
      "epoch": 1.1262893676757813e-05,
      "step": 11810,
      "training_step_time": 7.699834108352661
    },
    {
      "epoch": 1.1263847351074219e-05,
      "model_forward_time": 1.7390003204345703,
      "step": 11811
    },
    {
      "epoch": 1.1263847351074219e-05,
      "step": 11811,
      "training_step_time": 7.6795055866241455
    },
    {
      "epoch": 1.1264801025390624e-05,
      "model_forward_time": 1.7286317348480225,
      "step": 11812
    },
    {
      "epoch": 1.1264801025390624e-05,
      "step": 11812,
      "training_step_time": 6.790701866149902
    },
    {
      "epoch": 1.1265754699707032e-05,
      "model_forward_time": 1.7346131801605225,
      "step": 11813
    },
    {
      "epoch": 1.1265754699707032e-05,
      "step": 11813,
      "training_step_time": 6.798498630523682
    },
    {
      "epoch": 1.1266708374023438e-05,
      "model_forward_time": 1.7428867816925049,
      "step": 11814
    },
    {
      "epoch": 1.1266708374023438e-05,
      "step": 11814,
      "training_step_time": 7.659264802932739
    },
    {
      "epoch": 1.1267662048339843e-05,
      "model_forward_time": 1.7404448986053467,
      "step": 11815
    },
    {
      "epoch": 1.1267662048339843e-05,
      "step": 11815,
      "training_step_time": 7.368807792663574
    },
    {
      "epoch": 1.126861572265625e-05,
      "model_forward_time": 1.7291603088378906,
      "step": 11816
    },
    {
      "epoch": 1.126861572265625e-05,
      "step": 11816,
      "training_step_time": 8.645785331726074
    },
    {
      "epoch": 1.1269569396972656e-05,
      "model_forward_time": 1.7328851222991943,
      "step": 11817
    },
    {
      "epoch": 1.1269569396972656e-05,
      "step": 11817,
      "training_step_time": 9.73371410369873
    },
    {
      "epoch": 1.1270523071289062e-05,
      "model_forward_time": 1.7695367336273193,
      "step": 11818
    },
    {
      "epoch": 1.1270523071289062e-05,
      "step": 11818,
      "training_step_time": 7.8412230014801025
    },
    {
      "epoch": 1.127147674560547e-05,
      "model_forward_time": 2.270819664001465,
      "step": 11819
    },
    {
      "epoch": 1.127147674560547e-05,
      "step": 11819,
      "training_step_time": 8.413191795349121
    },
    {
      "epoch": 1.1272430419921875e-05,
      "grad_norm": 0.1915724277496338,
      "learning_rate": 9.873411921267138e-06,
      "loss": 0.0709,
      "step": 11820
    },
    {
      "epoch": 1.1272430419921875e-05,
      "model_forward_time": 1.7327353954315186,
      "step": 11820
    },
    {
      "dynamics_loss_avg": 0.059134446550160646,
      "epoch": 1.1272430419921875e-05,
      "step": 11820
    },
    {
      "action_loss_avg": 0.002366857253946364,
      "epoch": 1.1272430419921875e-05,
      "step": 11820
    },
    {
      "epoch": 1.1272430419921875e-05,
      "step": 11820,
      "training_step_time": 7.779691934585571
    },
    {
      "epoch": 1.1273384094238281e-05,
      "model_forward_time": 1.7337779998779297,
      "step": 11821
    },
    {
      "epoch": 1.1273384094238281e-05,
      "step": 11821,
      "training_step_time": 6.925649404525757
    },
    {
      "epoch": 1.1274337768554687e-05,
      "model_forward_time": 1.7495551109313965,
      "step": 11822
    },
    {
      "epoch": 1.1274337768554687e-05,
      "step": 11822,
      "training_step_time": 6.835476636886597
    },
    {
      "epoch": 1.1275291442871094e-05,
      "model_forward_time": 1.7347376346588135,
      "step": 11823
    },
    {
      "epoch": 1.1275291442871094e-05,
      "step": 11823,
      "training_step_time": 7.684410333633423
    },
    {
      "epoch": 1.12762451171875e-05,
      "model_forward_time": 1.7298331260681152,
      "step": 11824
    },
    {
      "epoch": 1.12762451171875e-05,
      "step": 11824,
      "training_step_time": 7.477991342544556
    },
    {
      "epoch": 1.1277198791503906e-05,
      "model_forward_time": 1.7325479984283447,
      "step": 11825
    },
    {
      "epoch": 1.1277198791503906e-05,
      "step": 11825,
      "training_step_time": 8.35295557975769
    },
    {
      "epoch": 1.1278152465820313e-05,
      "model_forward_time": 1.7314364910125732,
      "step": 11826
    },
    {
      "epoch": 1.1278152465820313e-05,
      "step": 11826,
      "training_step_time": 10.443925619125366
    },
    {
      "epoch": 1.1279106140136719e-05,
      "model_forward_time": 1.7248783111572266,
      "step": 11827
    },
    {
      "epoch": 1.1279106140136719e-05,
      "step": 11827,
      "training_step_time": 8.31086277961731
    },
    {
      "epoch": 1.1280059814453124e-05,
      "model_forward_time": 2.2816476821899414,
      "step": 11828
    },
    {
      "epoch": 1.1280059814453124e-05,
      "step": 11828,
      "training_step_time": 7.705301761627197
    },
    {
      "epoch": 1.1281013488769532e-05,
      "model_forward_time": 1.7352838516235352,
      "step": 11829
    },
    {
      "epoch": 1.1281013488769532e-05,
      "step": 11829,
      "training_step_time": 7.77783989906311
    },
    {
      "epoch": 1.1281967163085938e-05,
      "grad_norm": 0.2606842815876007,
      "learning_rate": 9.87304194896883e-06,
      "loss": 0.0673,
      "step": 11830
    },
    {
      "epoch": 1.1281967163085938e-05,
      "model_forward_time": 1.7331383228302002,
      "step": 11830
    },
    {
      "dynamics_loss_avg": 0.04815030749887228,
      "epoch": 1.1281967163085938e-05,
      "step": 11830
    },
    {
      "action_loss_avg": 0.002343782567186281,
      "epoch": 1.1281967163085938e-05,
      "step": 11830
    },
    {
      "epoch": 1.1281967163085938e-05,
      "step": 11830,
      "training_step_time": 6.805403470993042
    },
    {
      "epoch": 1.1282920837402343e-05,
      "model_forward_time": 1.7383193969726562,
      "step": 11831
    },
    {
      "epoch": 1.1282920837402343e-05,
      "step": 11831,
      "training_step_time": 6.810042381286621
    },
    {
      "epoch": 1.128387451171875e-05,
      "model_forward_time": 1.740687608718872,
      "step": 11832
    },
    {
      "epoch": 1.128387451171875e-05,
      "step": 11832,
      "training_step_time": 7.67626428604126
    },
    {
      "epoch": 1.1284828186035156e-05,
      "model_forward_time": 1.741865634918213,
      "step": 11833
    },
    {
      "epoch": 1.1284828186035156e-05,
      "step": 11833,
      "training_step_time": 8.714740514755249
    },
    {
      "epoch": 1.1285781860351562e-05,
      "model_forward_time": 1.7418758869171143,
      "step": 11834
    },
    {
      "epoch": 1.1285781860351562e-05,
      "step": 11834,
      "training_step_time": 7.762963771820068
    },
    {
      "epoch": 1.128673553466797e-05,
      "model_forward_time": 1.7383699417114258,
      "step": 11835
    },
    {
      "epoch": 1.128673553466797e-05,
      "step": 11835,
      "training_step_time": 9.645056962966919
    },
    {
      "epoch": 1.1287689208984375e-05,
      "model_forward_time": 1.735255479812622,
      "step": 11836
    },
    {
      "epoch": 1.1287689208984375e-05,
      "step": 11836,
      "training_step_time": 8.551233530044556
    },
    {
      "epoch": 1.1288642883300781e-05,
      "model_forward_time": 2.310070276260376,
      "step": 11837
    },
    {
      "epoch": 1.1288642883300781e-05,
      "step": 11837,
      "training_step_time": 7.670409917831421
    },
    {
      "epoch": 1.1289596557617187e-05,
      "model_forward_time": 1.7391974925994873,
      "step": 11838
    },
    {
      "epoch": 1.1289596557617187e-05,
      "step": 11838,
      "training_step_time": 7.826065540313721
    },
    {
      "epoch": 1.1290550231933594e-05,
      "model_forward_time": 1.7417826652526855,
      "step": 11839
    },
    {
      "epoch": 1.1290550231933594e-05,
      "step": 11839,
      "training_step_time": 6.841069459915161
    },
    {
      "epoch": 1.129150390625e-05,
      "grad_norm": 0.18059273064136505,
      "learning_rate": 9.872671443761983e-06,
      "loss": 0.0704,
      "step": 11840
    },
    {
      "epoch": 1.129150390625e-05,
      "model_forward_time": 1.73799729347229,
      "step": 11840
    },
    {
      "dynamics_loss_avg": 0.05478053204715252,
      "epoch": 1.129150390625e-05,
      "step": 11840
    },
    {
      "action_loss_avg": 0.0034565816167742014,
      "epoch": 1.129150390625e-05,
      "step": 11840
    },
    {
      "epoch": 1.129150390625e-05,
      "step": 11840,
      "training_step_time": 6.911747455596924
    },
    {
      "epoch": 1.1292457580566406e-05,
      "model_forward_time": 1.743332862854004,
      "step": 11841
    },
    {
      "epoch": 1.1292457580566406e-05,
      "step": 11841,
      "training_step_time": 7.729647636413574
    },
    {
      "epoch": 1.1293411254882813e-05,
      "model_forward_time": 1.742701530456543,
      "step": 11842
    },
    {
      "epoch": 1.1293411254882813e-05,
      "step": 11842,
      "training_step_time": 7.153710126876831
    },
    {
      "epoch": 1.1294364929199219e-05,
      "model_forward_time": 1.7462823390960693,
      "step": 11843
    },
    {
      "epoch": 1.1294364929199219e-05,
      "step": 11843,
      "training_step_time": 7.792735815048218
    },
    {
      "epoch": 1.1295318603515624e-05,
      "model_forward_time": 1.7417786121368408,
      "step": 11844
    },
    {
      "epoch": 1.1295318603515624e-05,
      "step": 11844,
      "training_step_time": 9.740830421447754
    },
    {
      "epoch": 1.1296272277832032e-05,
      "model_forward_time": 1.7438676357269287,
      "step": 11845
    },
    {
      "epoch": 1.1296272277832032e-05,
      "step": 11845,
      "training_step_time": 7.834499835968018
    },
    {
      "epoch": 1.1297225952148438e-05,
      "model_forward_time": 2.3600668907165527,
      "step": 11846
    },
    {
      "epoch": 1.1297225952148438e-05,
      "step": 11846,
      "training_step_time": 8.73220157623291
    },
    {
      "epoch": 1.1298179626464843e-05,
      "model_forward_time": 1.7367656230926514,
      "step": 11847
    },
    {
      "epoch": 1.1298179626464843e-05,
      "step": 11847,
      "training_step_time": 8.118258237838745
    },
    {
      "epoch": 1.129913330078125e-05,
      "model_forward_time": 1.7394042015075684,
      "step": 11848
    },
    {
      "epoch": 1.129913330078125e-05,
      "step": 11848,
      "training_step_time": 7.784460544586182
    },
    {
      "epoch": 1.1300086975097656e-05,
      "model_forward_time": 1.7466707229614258,
      "step": 11849
    },
    {
      "epoch": 1.1300086975097656e-05,
      "step": 11849,
      "training_step_time": 6.803633213043213
    },
    {
      "epoch": 1.1301040649414062e-05,
      "grad_norm": 0.2681204676628113,
      "learning_rate": 9.87230040568711e-06,
      "loss": 0.0696,
      "step": 11850
    },
    {
      "epoch": 1.1301040649414062e-05,
      "model_forward_time": 1.7376244068145752,
      "step": 11850
    },
    {
      "dynamics_loss_avg": 0.06498421132564544,
      "epoch": 1.1301040649414062e-05,
      "step": 11850
    },
    {
      "action_loss_avg": 0.0033835111709777264,
      "epoch": 1.1301040649414062e-05,
      "step": 11850
    },
    {
      "epoch": 1.1301040649414062e-05,
      "step": 11850,
      "training_step_time": 7.632739305496216
    },
    {
      "epoch": 1.130199432373047e-05,
      "model_forward_time": 1.7441718578338623,
      "step": 11851
    },
    {
      "epoch": 1.130199432373047e-05,
      "step": 11851,
      "training_step_time": 6.794532299041748
    },
    {
      "epoch": 1.1302947998046875e-05,
      "model_forward_time": 1.7426350116729736,
      "step": 11852
    },
    {
      "epoch": 1.1302947998046875e-05,
      "step": 11852,
      "training_step_time": 7.696854114532471
    },
    {
      "epoch": 1.1303901672363281e-05,
      "model_forward_time": 1.7346405982971191,
      "step": 11853
    },
    {
      "epoch": 1.1303901672363281e-05,
      "step": 11853,
      "training_step_time": 9.7906174659729
    },
    {
      "epoch": 1.1304855346679687e-05,
      "model_forward_time": 1.7385532855987549,
      "step": 11854
    },
    {
      "epoch": 1.1304855346679687e-05,
      "step": 11854,
      "training_step_time": 7.795876502990723
    },
    {
      "epoch": 1.1305809020996094e-05,
      "model_forward_time": 1.7454755306243896,
      "step": 11855
    },
    {
      "epoch": 1.1305809020996094e-05,
      "step": 11855,
      "training_step_time": 8.897446393966675
    },
    {
      "epoch": 1.13067626953125e-05,
      "model_forward_time": 1.741696834564209,
      "step": 11856
    },
    {
      "epoch": 1.13067626953125e-05,
      "step": 11856,
      "training_step_time": 7.675570011138916
    },
    {
      "epoch": 1.1307716369628906e-05,
      "model_forward_time": 1.7410056591033936,
      "step": 11857
    },
    {
      "epoch": 1.1307716369628906e-05,
      "step": 11857,
      "training_step_time": 7.724413871765137
    },
    {
      "epoch": 1.1308670043945313e-05,
      "model_forward_time": 1.745600938796997,
      "step": 11858
    },
    {
      "epoch": 1.1308670043945313e-05,
      "step": 11858,
      "training_step_time": 6.819490671157837
    },
    {
      "epoch": 1.1309623718261719e-05,
      "model_forward_time": 1.7377276420593262,
      "step": 11859
    },
    {
      "epoch": 1.1309623718261719e-05,
      "step": 11859,
      "training_step_time": 7.677396059036255
    },
    {
      "epoch": 1.1310577392578124e-05,
      "grad_norm": 0.23794104158878326,
      "learning_rate": 9.871928834784793e-06,
      "loss": 0.0712,
      "step": 11860
    },
    {
      "epoch": 1.1310577392578124e-05,
      "model_forward_time": 1.7355759143829346,
      "step": 11860
    },
    {
      "dynamics_loss_avg": 0.06490019075572491,
      "epoch": 1.1310577392578124e-05,
      "step": 11860
    },
    {
      "action_loss_avg": 0.003268509707413614,
      "epoch": 1.1310577392578124e-05,
      "step": 11860
    },
    {
      "epoch": 1.1310577392578124e-05,
      "step": 11860,
      "training_step_time": 6.785619735717773
    },
    {
      "epoch": 1.1311531066894532e-05,
      "model_forward_time": 1.73518705368042,
      "step": 11861
    },
    {
      "epoch": 1.1311531066894532e-05,
      "step": 11861,
      "training_step_time": 7.696780443191528
    },
    {
      "epoch": 1.1312484741210938e-05,
      "model_forward_time": 1.7402856349945068,
      "step": 11862
    },
    {
      "epoch": 1.1312484741210938e-05,
      "step": 11862,
      "training_step_time": 8.6452476978302
    },
    {
      "epoch": 1.1313438415527343e-05,
      "model_forward_time": 1.7344045639038086,
      "step": 11863
    },
    {
      "epoch": 1.1313438415527343e-05,
      "step": 11863,
      "training_step_time": 8.391559839248657
    },
    {
      "epoch": 1.131439208984375e-05,
      "model_forward_time": 1.7398850917816162,
      "step": 11864
    },
    {
      "epoch": 1.131439208984375e-05,
      "step": 11864,
      "training_step_time": 8.297379493713379
    },
    {
      "epoch": 1.1315345764160156e-05,
      "model_forward_time": 2.308199167251587,
      "step": 11865
    },
    {
      "epoch": 1.1315345764160156e-05,
      "step": 11865,
      "training_step_time": 7.603976249694824
    },
    {
      "epoch": 1.1316299438476562e-05,
      "model_forward_time": 1.7408068180084229,
      "step": 11866
    },
    {
      "epoch": 1.1316299438476562e-05,
      "step": 11866,
      "training_step_time": 7.744582414627075
    },
    {
      "epoch": 1.131725311279297e-05,
      "model_forward_time": 1.7525277137756348,
      "step": 11867
    },
    {
      "epoch": 1.131725311279297e-05,
      "step": 11867,
      "training_step_time": 6.8901660442352295
    },
    {
      "epoch": 1.1318206787109375e-05,
      "model_forward_time": 1.7497665882110596,
      "step": 11868
    },
    {
      "epoch": 1.1318206787109375e-05,
      "step": 11868,
      "training_step_time": 6.788763046264648
    },
    {
      "epoch": 1.1319160461425781e-05,
      "model_forward_time": 1.7549705505371094,
      "step": 11869
    },
    {
      "epoch": 1.1319160461425781e-05,
      "step": 11869,
      "training_step_time": 7.746699810028076
    },
    {
      "epoch": 1.1320114135742187e-05,
      "grad_norm": 0.22005774080753326,
      "learning_rate": 9.871556731095662e-06,
      "loss": 0.0702,
      "step": 11870
    },
    {
      "epoch": 1.1320114135742187e-05,
      "model_forward_time": 1.7378273010253906,
      "step": 11870
    },
    {
      "dynamics_loss_avg": 0.07308344542980194,
      "epoch": 1.1320114135742187e-05,
      "step": 11870
    },
    {
      "action_loss_avg": 0.0041496064746752385,
      "epoch": 1.1320114135742187e-05,
      "step": 11870
    },
    {
      "epoch": 1.1320114135742187e-05,
      "step": 11870,
      "training_step_time": 7.7114996910095215
    },
    {
      "epoch": 1.1321067810058594e-05,
      "model_forward_time": 1.7531707286834717,
      "step": 11871
    },
    {
      "epoch": 1.1321067810058594e-05,
      "step": 11871,
      "training_step_time": 8.447827100753784
    },
    {
      "epoch": 1.1322021484375e-05,
      "model_forward_time": 1.0300376415252686,
      "step": 11872
    },
    {
      "epoch": 1.1322021484375e-05,
      "step": 11872,
      "training_step_time": 9.394680261611938
    },
    {
      "epoch": 1.1322975158691406e-05,
      "model_forward_time": 1.7467851638793945,
      "step": 11873
    },
    {
      "epoch": 1.1322975158691406e-05,
      "step": 11873,
      "training_step_time": 9.448283433914185
    },
    {
      "epoch": 1.1323928833007813e-05,
      "model_forward_time": 2.84928035736084,
      "step": 11874
    },
    {
      "epoch": 1.1323928833007813e-05,
      "step": 11874,
      "training_step_time": 7.85483193397522
    },
    {
      "epoch": 1.1324882507324219e-05,
      "model_forward_time": 1.7459065914154053,
      "step": 11875
    },
    {
      "epoch": 1.1324882507324219e-05,
      "step": 11875,
      "training_step_time": 7.7076499462127686
    },
    {
      "epoch": 1.1325836181640624e-05,
      "model_forward_time": 1.7583246231079102,
      "step": 11876
    },
    {
      "epoch": 1.1325836181640624e-05,
      "step": 11876,
      "training_step_time": 6.879824638366699
    },
    {
      "epoch": 1.1326789855957032e-05,
      "model_forward_time": 1.750145435333252,
      "step": 11877
    },
    {
      "epoch": 1.1326789855957032e-05,
      "step": 11877,
      "training_step_time": 6.858289480209351
    },
    {
      "epoch": 1.1327743530273438e-05,
      "model_forward_time": 1.7419872283935547,
      "step": 11878
    },
    {
      "epoch": 1.1327743530273438e-05,
      "step": 11878,
      "training_step_time": 7.635438680648804
    },
    {
      "epoch": 1.1328697204589843e-05,
      "model_forward_time": 1.740588665008545,
      "step": 11879
    },
    {
      "epoch": 1.1328697204589843e-05,
      "step": 11879,
      "training_step_time": 7.977967262268066
    },
    {
      "epoch": 1.132965087890625e-05,
      "grad_norm": 0.2895621955394745,
      "learning_rate": 9.87118409466041e-06,
      "loss": 0.0637,
      "step": 11880
    },
    {
      "epoch": 1.132965087890625e-05,
      "model_forward_time": 1.7465109825134277,
      "step": 11880
    },
    {
      "dynamics_loss_avg": 0.0702799690887332,
      "epoch": 1.132965087890625e-05,
      "step": 11880
    },
    {
      "action_loss_avg": 0.005958632822148502,
      "epoch": 1.132965087890625e-05,
      "step": 11880
    },
    {
      "epoch": 1.132965087890625e-05,
      "step": 11880,
      "training_step_time": 7.811984539031982
    },
    {
      "epoch": 1.1330604553222656e-05,
      "model_forward_time": 1.8082199096679688,
      "step": 11881
    },
    {
      "epoch": 1.1330604553222656e-05,
      "step": 11881,
      "training_step_time": 10.514464855194092
    },
    {
      "epoch": 1.1331558227539062e-05,
      "model_forward_time": 1.7474696636199951,
      "step": 11882
    },
    {
      "epoch": 1.1331558227539062e-05,
      "step": 11882,
      "training_step_time": 9.255650043487549
    },
    {
      "epoch": 1.1332511901855468e-05,
      "model_forward_time": 1.3802800178527832,
      "step": 11883
    },
    {
      "epoch": 1.1332511901855468e-05,
      "step": 11883,
      "training_step_time": 7.783375263214111
    },
    {
      "epoch": 1.1333465576171875e-05,
      "model_forward_time": 1.740703821182251,
      "step": 11884
    },
    {
      "epoch": 1.1333465576171875e-05,
      "step": 11884,
      "training_step_time": 7.710751295089722
    },
    {
      "epoch": 1.1334419250488281e-05,
      "model_forward_time": 1.7463340759277344,
      "step": 11885
    },
    {
      "epoch": 1.1334419250488281e-05,
      "step": 11885,
      "training_step_time": 6.828593969345093
    },
    {
      "epoch": 1.1335372924804687e-05,
      "model_forward_time": 1.743229627609253,
      "step": 11886
    },
    {
      "epoch": 1.1335372924804687e-05,
      "step": 11886,
      "training_step_time": 6.863621234893799
    },
    {
      "epoch": 1.1336326599121094e-05,
      "model_forward_time": 1.7551007270812988,
      "step": 11887
    },
    {
      "epoch": 1.1336326599121094e-05,
      "step": 11887,
      "training_step_time": 6.822169065475464
    },
    {
      "epoch": 1.13372802734375e-05,
      "model_forward_time": 1.7491750717163086,
      "step": 11888
    },
    {
      "epoch": 1.13372802734375e-05,
      "step": 11888,
      "training_step_time": 8.445533990859985
    },
    {
      "epoch": 1.1338233947753906e-05,
      "model_forward_time": 1.7444818019866943,
      "step": 11889
    },
    {
      "epoch": 1.1338233947753906e-05,
      "step": 11889,
      "training_step_time": 7.723320722579956
    },
    {
      "epoch": 1.1339187622070313e-05,
      "grad_norm": 0.29399681091308594,
      "learning_rate": 9.870810925519792e-06,
      "loss": 0.0656,
      "step": 11890
    },
    {
      "epoch": 1.1339187622070313e-05,
      "model_forward_time": 1.800964593887329,
      "step": 11890
    },
    {
      "dynamics_loss_avg": 0.07327999463304877,
      "epoch": 1.1339187622070313e-05,
      "step": 11890
    },
    {
      "action_loss_avg": 0.004229505965486169,
      "epoch": 1.1339187622070313e-05,
      "step": 11890
    },
    {
      "epoch": 1.1339187622070313e-05,
      "step": 11890,
      "training_step_time": 8.69015884399414
    },
    {
      "epoch": 1.1340141296386719e-05,
      "model_forward_time": 1.7503440380096436,
      "step": 11891
    },
    {
      "epoch": 1.1340141296386719e-05,
      "step": 11891,
      "training_step_time": 9.019335508346558
    },
    {
      "epoch": 1.1341094970703124e-05,
      "model_forward_time": 1.7577788829803467,
      "step": 11892
    },
    {
      "epoch": 1.1341094970703124e-05,
      "step": 11892,
      "training_step_time": 7.752162456512451
    },
    {
      "epoch": 1.1342048645019532e-05,
      "model_forward_time": 2.9083950519561768,
      "step": 11893
    },
    {
      "epoch": 1.1342048645019532e-05,
      "step": 11893,
      "training_step_time": 7.728155612945557
    },
    {
      "epoch": 1.1343002319335938e-05,
      "model_forward_time": 1.777984857559204,
      "step": 11894
    },
    {
      "epoch": 1.1343002319335938e-05,
      "step": 11894,
      "training_step_time": 7.65372633934021
    },
    {
      "epoch": 1.1343955993652343e-05,
      "model_forward_time": 1.756554126739502,
      "step": 11895
    },
    {
      "epoch": 1.1343955993652343e-05,
      "step": 11895,
      "training_step_time": 6.772880554199219
    },
    {
      "epoch": 1.134490966796875e-05,
      "model_forward_time": 1.7591114044189453,
      "step": 11896
    },
    {
      "epoch": 1.134490966796875e-05,
      "step": 11896,
      "training_step_time": 6.75627064704895
    },
    {
      "epoch": 1.1345863342285156e-05,
      "model_forward_time": 1.732027530670166,
      "step": 11897
    },
    {
      "epoch": 1.1345863342285156e-05,
      "step": 11897,
      "training_step_time": 7.582536458969116
    },
    {
      "epoch": 1.1346817016601562e-05,
      "model_forward_time": 1.7410423755645752,
      "step": 11898
    },
    {
      "epoch": 1.1346817016601562e-05,
      "step": 11898,
      "training_step_time": 7.7879109382629395
    },
    {
      "epoch": 1.1347770690917968e-05,
      "model_forward_time": 1.7455615997314453,
      "step": 11899
    },
    {
      "epoch": 1.1347770690917968e-05,
      "step": 11899,
      "training_step_time": 8.683523416519165
    },
    {
      "epoch": 1.1348724365234375e-05,
      "grad_norm": 0.3346741199493408,
      "learning_rate": 9.870437223714612e-06,
      "loss": 0.063,
      "step": 11900
    },
    {
      "epoch": 1.1348724365234375e-05,
      "model_forward_time": 1.746443271636963,
      "step": 11900
    },
    {
      "dynamics_loss_avg": 0.05516475960612297,
      "epoch": 1.1348724365234375e-05,
      "step": 11900
    },
    {
      "action_loss_avg": 0.0025409622525330634,
      "epoch": 1.1348724365234375e-05,
      "step": 11900
    },
    {
      "epoch": 1.1348724365234375e-05,
      "step": 11900,
      "training_step_time": 10.5558180809021
    },
    {
      "epoch": 1.1349678039550781e-05,
      "model_forward_time": 1.7336952686309814,
      "step": 11901
    },
    {
      "epoch": 1.1349678039550781e-05,
      "step": 11901,
      "training_step_time": 9.086392164230347
    },
    {
      "epoch": 1.1350631713867187e-05,
      "model_forward_time": 2.9014530181884766,
      "step": 11902
    },
    {
      "epoch": 1.1350631713867187e-05,
      "step": 11902,
      "training_step_time": 7.651634693145752
    },
    {
      "epoch": 1.1351585388183594e-05,
      "model_forward_time": 1.7386107444763184,
      "step": 11903
    },
    {
      "epoch": 1.1351585388183594e-05,
      "step": 11903,
      "training_step_time": 7.6070520877838135
    },
    {
      "epoch": 1.13525390625e-05,
      "model_forward_time": 1.7262630462646484,
      "step": 11904
    },
    {
      "epoch": 1.13525390625e-05,
      "step": 11904,
      "training_step_time": 6.7640063762664795
    },
    {
      "epoch": 1.1353492736816406e-05,
      "model_forward_time": 1.7249836921691895,
      "step": 11905
    },
    {
      "epoch": 1.1353492736816406e-05,
      "step": 11905,
      "training_step_time": 6.7471394538879395
    },
    {
      "epoch": 1.1354446411132813e-05,
      "model_forward_time": 1.7243945598602295,
      "step": 11906
    },
    {
      "epoch": 1.1354446411132813e-05,
      "step": 11906,
      "training_step_time": 7.59425687789917
    },
    {
      "epoch": 1.1355400085449219e-05,
      "model_forward_time": 1.7640049457550049,
      "step": 11907
    },
    {
      "epoch": 1.1355400085449219e-05,
      "step": 11907,
      "training_step_time": 7.684948444366455
    },
    {
      "epoch": 1.1356353759765624e-05,
      "model_forward_time": 1.3891186714172363,
      "step": 11908
    },
    {
      "epoch": 1.1356353759765624e-05,
      "step": 11908,
      "training_step_time": 7.727176904678345
    },
    {
      "epoch": 1.1357307434082032e-05,
      "model_forward_time": 1.737422227859497,
      "step": 11909
    },
    {
      "epoch": 1.1357307434082032e-05,
      "step": 11909,
      "training_step_time": 9.483473777770996
    },
    {
      "epoch": 1.1358261108398438e-05,
      "grad_norm": 0.2422727644443512,
      "learning_rate": 9.87006298928574e-06,
      "loss": 0.0665,
      "step": 11910
    },
    {
      "epoch": 1.1358261108398438e-05,
      "model_forward_time": 1.7506201267242432,
      "step": 11910
    },
    {
      "dynamics_loss_avg": 0.058761333115398884,
      "epoch": 1.1358261108398438e-05,
      "step": 11910
    },
    {
      "action_loss_avg": 0.0028426857374142856,
      "epoch": 1.1358261108398438e-05,
      "step": 11910
    },
    {
      "epoch": 1.1358261108398438e-05,
      "step": 11910,
      "training_step_time": 8.668671131134033
    },
    {
      "epoch": 1.1359214782714843e-05,
      "model_forward_time": 2.304734230041504,
      "step": 11911
    },
    {
      "epoch": 1.1359214782714843e-05,
      "step": 11911,
      "training_step_time": 8.707129955291748
    },
    {
      "epoch": 1.136016845703125e-05,
      "model_forward_time": 1.8073616027832031,
      "step": 11912
    },
    {
      "epoch": 1.136016845703125e-05,
      "step": 11912,
      "training_step_time": 7.5956199169158936
    },
    {
      "epoch": 1.1361122131347656e-05,
      "model_forward_time": 1.7530126571655273,
      "step": 11913
    },
    {
      "epoch": 1.1361122131347656e-05,
      "step": 11913,
      "training_step_time": 6.722419738769531
    },
    {
      "epoch": 1.1362075805664062e-05,
      "model_forward_time": 1.7297606468200684,
      "step": 11914
    },
    {
      "epoch": 1.1362075805664062e-05,
      "step": 11914,
      "training_step_time": 6.76986289024353
    },
    {
      "epoch": 1.1363029479980468e-05,
      "model_forward_time": 1.7278416156768799,
      "step": 11915
    },
    {
      "epoch": 1.1363029479980468e-05,
      "step": 11915,
      "training_step_time": 7.543920278549194
    },
    {
      "epoch": 1.1363983154296875e-05,
      "model_forward_time": 1.727529525756836,
      "step": 11916
    },
    {
      "epoch": 1.1363983154296875e-05,
      "step": 11916,
      "training_step_time": 7.606823921203613
    },
    {
      "epoch": 1.1364936828613281e-05,
      "model_forward_time": 1.7271404266357422,
      "step": 11917
    },
    {
      "epoch": 1.1364936828613281e-05,
      "step": 11917,
      "training_step_time": 8.392516613006592
    },
    {
      "epoch": 1.1365890502929687e-05,
      "model_forward_time": 1.7261252403259277,
      "step": 11918
    },
    {
      "epoch": 1.1365890502929687e-05,
      "step": 11918,
      "training_step_time": 9.856334686279297
    },
    {
      "epoch": 1.1366844177246094e-05,
      "model_forward_time": 1.7220587730407715,
      "step": 11919
    },
    {
      "epoch": 1.1366844177246094e-05,
      "step": 11919,
      "training_step_time": 7.645951509475708
    },
    {
      "epoch": 1.13677978515625e-05,
      "grad_norm": 0.14387141168117523,
      "learning_rate": 9.869688222274104e-06,
      "loss": 0.0689,
      "step": 11920
    },
    {
      "epoch": 1.13677978515625e-05,
      "model_forward_time": 2.265723466873169,
      "step": 11920
    },
    {
      "dynamics_loss_avg": 0.06267203651368618,
      "epoch": 1.13677978515625e-05,
      "step": 11920
    },
    {
      "action_loss_avg": 0.003497704508481547,
      "epoch": 1.13677978515625e-05,
      "step": 11920
    },
    {
      "epoch": 1.13677978515625e-05,
      "step": 11920,
      "training_step_time": 8.559363842010498
    },
    {
      "epoch": 1.1368751525878906e-05,
      "model_forward_time": 1.785109043121338,
      "step": 11921
    },
    {
      "epoch": 1.1368751525878906e-05,
      "step": 11921,
      "training_step_time": 8.534887313842773
    },
    {
      "epoch": 1.1369705200195313e-05,
      "model_forward_time": 1.730520248413086,
      "step": 11922
    },
    {
      "epoch": 1.1369705200195313e-05,
      "step": 11922,
      "training_step_time": 6.7161924839019775
    },
    {
      "epoch": 1.1370658874511719e-05,
      "model_forward_time": 1.7314598560333252,
      "step": 11923
    },
    {
      "epoch": 1.1370658874511719e-05,
      "step": 11923,
      "training_step_time": 6.766644477844238
    },
    {
      "epoch": 1.1371612548828125e-05,
      "model_forward_time": 1.7332406044006348,
      "step": 11924
    },
    {
      "epoch": 1.1371612548828125e-05,
      "step": 11924,
      "training_step_time": 7.38523268699646
    },
    {
      "epoch": 1.1372566223144532e-05,
      "model_forward_time": 1.7353034019470215,
      "step": 11925
    },
    {
      "epoch": 1.1372566223144532e-05,
      "step": 11925,
      "training_step_time": 8.63343095779419
    },
    {
      "epoch": 1.1373519897460938e-05,
      "model_forward_time": 1.727292537689209,
      "step": 11926
    },
    {
      "epoch": 1.1373519897460938e-05,
      "step": 11926,
      "training_step_time": 7.693896532058716
    },
    {
      "epoch": 1.1374473571777343e-05,
      "model_forward_time": 1.724482536315918,
      "step": 11927
    },
    {
      "epoch": 1.1374473571777343e-05,
      "step": 11927,
      "training_step_time": 9.391780376434326
    },
    {
      "epoch": 1.137542724609375e-05,
      "model_forward_time": 1.7238566875457764,
      "step": 11928
    },
    {
      "epoch": 1.137542724609375e-05,
      "step": 11928,
      "training_step_time": 8.153427600860596
    },
    {
      "epoch": 1.1376380920410157e-05,
      "model_forward_time": 2.2573165893554688,
      "step": 11929
    },
    {
      "epoch": 1.1376380920410157e-05,
      "step": 11929,
      "training_step_time": 7.741690635681152
    },
    {
      "epoch": 1.1377334594726562e-05,
      "grad_norm": 0.2662833034992218,
      "learning_rate": 9.869312922720682e-06,
      "loss": 0.0677,
      "step": 11930
    },
    {
      "epoch": 1.1377334594726562e-05,
      "model_forward_time": 1.7692158222198486,
      "step": 11930
    },
    {
      "dynamics_loss_avg": 0.057118627894669774,
      "epoch": 1.1377334594726562e-05,
      "step": 11930
    },
    {
      "action_loss_avg": 0.0037131249147932975,
      "epoch": 1.1377334594726562e-05,
      "step": 11930
    },
    {
      "epoch": 1.1377334594726562e-05,
      "step": 11930,
      "training_step_time": 7.712742328643799
    },
    {
      "epoch": 1.1378288269042968e-05,
      "model_forward_time": 1.7346248626708984,
      "step": 11931
    },
    {
      "epoch": 1.1378288269042968e-05,
      "step": 11931,
      "training_step_time": 7.610597372055054
    },
    {
      "epoch": 1.1379241943359375e-05,
      "model_forward_time": 1.7306387424468994,
      "step": 11932
    },
    {
      "epoch": 1.1379241943359375e-05,
      "step": 11932,
      "training_step_time": 6.700234651565552
    },
    {
      "epoch": 1.1380195617675781e-05,
      "model_forward_time": 1.7348308563232422,
      "step": 11933
    },
    {
      "epoch": 1.1380195617675781e-05,
      "step": 11933,
      "training_step_time": 7.34909462928772
    },
    {
      "epoch": 1.1381149291992187e-05,
      "model_forward_time": 1.7309763431549072,
      "step": 11934
    },
    {
      "epoch": 1.1381149291992187e-05,
      "step": 11934,
      "training_step_time": 7.762240409851074
    },
    {
      "epoch": 1.1382102966308594e-05,
      "model_forward_time": 1.7288594245910645,
      "step": 11935
    },
    {
      "epoch": 1.1382102966308594e-05,
      "step": 11935,
      "training_step_time": 7.689643859863281
    },
    {
      "epoch": 1.1383056640625e-05,
      "model_forward_time": 1.7323338985443115,
      "step": 11936
    },
    {
      "epoch": 1.1383056640625e-05,
      "step": 11936,
      "training_step_time": 7.81189227104187
    },
    {
      "epoch": 1.1384010314941406e-05,
      "model_forward_time": 1.7181129455566406,
      "step": 11937
    },
    {
      "epoch": 1.1384010314941406e-05,
      "step": 11937,
      "training_step_time": 8.385284185409546
    },
    {
      "epoch": 1.1384963989257813e-05,
      "model_forward_time": 2.2747271060943604,
      "step": 11938
    },
    {
      "epoch": 1.1384963989257813e-05,
      "step": 11938,
      "training_step_time": 7.73220419883728
    },
    {
      "epoch": 1.1385917663574219e-05,
      "model_forward_time": 1.757216215133667,
      "step": 11939
    },
    {
      "epoch": 1.1385917663574219e-05,
      "step": 11939,
      "training_step_time": 7.811872243881226
    },
    {
      "epoch": 1.1386871337890625e-05,
      "grad_norm": 0.23738057911396027,
      "learning_rate": 9.868937090666521e-06,
      "loss": 0.0632,
      "step": 11940
    },
    {
      "epoch": 1.1386871337890625e-05,
      "model_forward_time": 1.7197482585906982,
      "step": 11940
    },
    {
      "dynamics_loss_avg": 0.052218999899923804,
      "epoch": 1.1386871337890625e-05,
      "step": 11940
    },
    {
      "action_loss_avg": 0.0023915382684208454,
      "epoch": 1.1386871337890625e-05,
      "step": 11940
    },
    {
      "epoch": 1.1386871337890625e-05,
      "step": 11940,
      "training_step_time": 7.755892992019653
    },
    {
      "epoch": 1.1387825012207032e-05,
      "model_forward_time": 1.7250511646270752,
      "step": 11941
    },
    {
      "epoch": 1.1387825012207032e-05,
      "step": 11941,
      "training_step_time": 6.858654260635376
    },
    {
      "epoch": 1.1388778686523438e-05,
      "model_forward_time": 1.7259047031402588,
      "step": 11942
    },
    {
      "epoch": 1.1388778686523438e-05,
      "step": 11942,
      "training_step_time": 7.384131669998169
    },
    {
      "epoch": 1.1389732360839843e-05,
      "model_forward_time": 1.7285494804382324,
      "step": 11943
    },
    {
      "epoch": 1.1389732360839843e-05,
      "step": 11943,
      "training_step_time": 7.848966836929321
    },
    {
      "epoch": 1.139068603515625e-05,
      "model_forward_time": 1.717561960220337,
      "step": 11944
    },
    {
      "epoch": 1.139068603515625e-05,
      "step": 11944,
      "training_step_time": 7.881055116653442
    },
    {
      "epoch": 1.1391639709472657e-05,
      "model_forward_time": 1.7170863151550293,
      "step": 11945
    },
    {
      "epoch": 1.1391639709472657e-05,
      "step": 11945,
      "training_step_time": 8.321934461593628
    },
    {
      "epoch": 1.1392593383789062e-05,
      "model_forward_time": 1.7322802543640137,
      "step": 11946
    },
    {
      "epoch": 1.1392593383789062e-05,
      "step": 11946,
      "training_step_time": 7.655019998550415
    },
    {
      "epoch": 1.1393547058105468e-05,
      "model_forward_time": 2.264801502227783,
      "step": 11947
    },
    {
      "epoch": 1.1393547058105468e-05,
      "step": 11947,
      "training_step_time": 7.799283027648926
    },
    {
      "epoch": 1.1394500732421875e-05,
      "model_forward_time": 1.7464599609375,
      "step": 11948
    },
    {
      "epoch": 1.1394500732421875e-05,
      "step": 11948,
      "training_step_time": 8.501599788665771
    },
    {
      "epoch": 1.1395454406738281e-05,
      "model_forward_time": 1.7466721534729004,
      "step": 11949
    },
    {
      "epoch": 1.1395454406738281e-05,
      "step": 11949,
      "training_step_time": 7.747172117233276
    },
    {
      "epoch": 1.1396408081054687e-05,
      "grad_norm": 0.2890896201133728,
      "learning_rate": 9.86856072615272e-06,
      "loss": 0.0692,
      "step": 11950
    },
    {
      "epoch": 1.1396408081054687e-05,
      "model_forward_time": 1.7371885776519775,
      "step": 11950
    },
    {
      "dynamics_loss_avg": 0.05788545683026314,
      "epoch": 1.1396408081054687e-05,
      "step": 11950
    },
    {
      "action_loss_avg": 0.0037483041174709796,
      "epoch": 1.1396408081054687e-05,
      "step": 11950
    },
    {
      "epoch": 1.1396408081054687e-05,
      "step": 11950,
      "training_step_time": 7.711223840713501
    },
    {
      "epoch": 1.1397361755371094e-05,
      "model_forward_time": 1.7463479042053223,
      "step": 11951
    },
    {
      "epoch": 1.1397361755371094e-05,
      "step": 11951,
      "training_step_time": 7.320021867752075
    },
    {
      "epoch": 1.13983154296875e-05,
      "model_forward_time": 1.746697187423706,
      "step": 11952
    },
    {
      "epoch": 1.13983154296875e-05,
      "step": 11952,
      "training_step_time": 8.563754081726074
    },
    {
      "epoch": 1.1399269104003906e-05,
      "model_forward_time": 1.7455241680145264,
      "step": 11953
    },
    {
      "epoch": 1.1399269104003906e-05,
      "step": 11953,
      "training_step_time": 7.870050668716431
    },
    {
      "epoch": 1.1400222778320313e-05,
      "model_forward_time": 1.757084608078003,
      "step": 11954
    },
    {
      "epoch": 1.1400222778320313e-05,
      "step": 11954,
      "training_step_time": 8.411827325820923
    },
    {
      "epoch": 1.1401176452636719e-05,
      "model_forward_time": 1.3877942562103271,
      "step": 11955
    },
    {
      "epoch": 1.1401176452636719e-05,
      "step": 11955,
      "training_step_time": 9.415513515472412
    },
    {
      "epoch": 1.1402130126953125e-05,
      "model_forward_time": 1.7501356601715088,
      "step": 11956
    },
    {
      "epoch": 1.1402130126953125e-05,
      "step": 11956,
      "training_step_time": 7.604997396469116
    },
    {
      "epoch": 1.1403083801269532e-05,
      "model_forward_time": 2.3091537952423096,
      "step": 11957
    },
    {
      "epoch": 1.1403083801269532e-05,
      "step": 11957,
      "training_step_time": 7.832353591918945
    },
    {
      "epoch": 1.1404037475585938e-05,
      "model_forward_time": 1.7586307525634766,
      "step": 11958
    },
    {
      "epoch": 1.1404037475585938e-05,
      "step": 11958,
      "training_step_time": 7.68787407875061
    },
    {
      "epoch": 1.1404991149902343e-05,
      "model_forward_time": 1.7575080394744873,
      "step": 11959
    },
    {
      "epoch": 1.1404991149902343e-05,
      "step": 11959,
      "training_step_time": 7.766748428344727
    },
    {
      "epoch": 1.140594482421875e-05,
      "grad_norm": 0.1768205165863037,
      "learning_rate": 9.868183829220439e-06,
      "loss": 0.0706,
      "step": 11960
    },
    {
      "epoch": 1.140594482421875e-05,
      "model_forward_time": 1.3961875438690186,
      "step": 11960
    },
    {
      "dynamics_loss_avg": 0.07309606969356537,
      "epoch": 1.140594482421875e-05,
      "step": 11960
    },
    {
      "action_loss_avg": 0.003457227602484636,
      "epoch": 1.140594482421875e-05,
      "step": 11960
    },
    {
      "epoch": 1.140594482421875e-05,
      "step": 11960,
      "training_step_time": 7.150878190994263
    },
    {
      "epoch": 1.1406898498535157e-05,
      "model_forward_time": 1.7613847255706787,
      "step": 11961
    },
    {
      "epoch": 1.1406898498535157e-05,
      "step": 11961,
      "training_step_time": 7.766158580780029
    },
    {
      "epoch": 1.1407852172851562e-05,
      "model_forward_time": 1.7535653114318848,
      "step": 11962
    },
    {
      "epoch": 1.1407852172851562e-05,
      "step": 11962,
      "training_step_time": 7.683936834335327
    },
    {
      "epoch": 1.1408805847167968e-05,
      "model_forward_time": 1.7602388858795166,
      "step": 11963
    },
    {
      "epoch": 1.1408805847167968e-05,
      "step": 11963,
      "training_step_time": 9.755322933197021
    },
    {
      "epoch": 1.1409759521484375e-05,
      "model_forward_time": 1.7547438144683838,
      "step": 11964
    },
    {
      "epoch": 1.1409759521484375e-05,
      "step": 11964,
      "training_step_time": 8.530140161514282
    },
    {
      "epoch": 1.1410713195800781e-05,
      "model_forward_time": 1.7529692649841309,
      "step": 11965
    },
    {
      "epoch": 1.1410713195800781e-05,
      "step": 11965,
      "training_step_time": 7.680610418319702
    },
    {
      "epoch": 1.1411666870117187e-05,
      "model_forward_time": 2.314955234527588,
      "step": 11966
    },
    {
      "epoch": 1.1411666870117187e-05,
      "step": 11966,
      "training_step_time": 7.738006353378296
    },
    {
      "epoch": 1.1412620544433594e-05,
      "model_forward_time": 1.8261756896972656,
      "step": 11967
    },
    {
      "epoch": 1.1412620544433594e-05,
      "step": 11967,
      "training_step_time": 7.6097118854522705
    },
    {
      "epoch": 1.141357421875e-05,
      "model_forward_time": 1.774599313735962,
      "step": 11968
    },
    {
      "epoch": 1.141357421875e-05,
      "step": 11968,
      "training_step_time": 7.6251959800720215
    },
    {
      "epoch": 1.1414527893066406e-05,
      "model_forward_time": 1.7701406478881836,
      "step": 11969
    },
    {
      "epoch": 1.1414527893066406e-05,
      "step": 11969,
      "training_step_time": 7.022388458251953
    },
    {
      "epoch": 1.1415481567382813e-05,
      "grad_norm": 0.2424299269914627,
      "learning_rate": 9.867806399910892e-06,
      "loss": 0.0682,
      "step": 11970
    },
    {
      "epoch": 1.1415481567382813e-05,
      "model_forward_time": 1.7755889892578125,
      "step": 11970
    },
    {
      "dynamics_loss_avg": 0.06599068343639374,
      "epoch": 1.1415481567382813e-05,
      "step": 11970
    },
    {
      "action_loss_avg": 0.0025859809655230494,
      "epoch": 1.1415481567382813e-05,
      "step": 11970
    },
    {
      "epoch": 1.1415481567382813e-05,
      "step": 11970,
      "training_step_time": 7.632263422012329
    },
    {
      "epoch": 1.1416435241699219e-05,
      "model_forward_time": 1.7552783489227295,
      "step": 11971
    },
    {
      "epoch": 1.1416435241699219e-05,
      "step": 11971,
      "training_step_time": 7.65470552444458
    },
    {
      "epoch": 1.1417388916015625e-05,
      "model_forward_time": 1.7696540355682373,
      "step": 11972
    },
    {
      "epoch": 1.1417388916015625e-05,
      "step": 11972,
      "training_step_time": 7.745044946670532
    },
    {
      "epoch": 1.1418342590332032e-05,
      "model_forward_time": 1.758817434310913,
      "step": 11973
    },
    {
      "epoch": 1.1418342590332032e-05,
      "step": 11973,
      "training_step_time": 9.332380294799805
    },
    {
      "epoch": 1.1419296264648438e-05,
      "model_forward_time": 1.7683477401733398,
      "step": 11974
    },
    {
      "epoch": 1.1419296264648438e-05,
      "step": 11974,
      "training_step_time": 10.593851089477539
    },
    {
      "epoch": 1.1420249938964843e-05,
      "model_forward_time": 2.3292572498321533,
      "step": 11975
    },
    {
      "epoch": 1.1420249938964843e-05,
      "step": 11975,
      "training_step_time": 8.422679901123047
    },
    {
      "epoch": 1.142120361328125e-05,
      "model_forward_time": 1.8262319564819336,
      "step": 11976
    },
    {
      "epoch": 1.142120361328125e-05,
      "step": 11976,
      "training_step_time": 7.697609901428223
    },
    {
      "epoch": 1.1422157287597657e-05,
      "model_forward_time": 1.7773668766021729,
      "step": 11977
    },
    {
      "epoch": 1.1422157287597657e-05,
      "step": 11977,
      "training_step_time": 7.1975250244140625
    },
    {
      "epoch": 1.1423110961914062e-05,
      "model_forward_time": 1.7739136219024658,
      "step": 11978
    },
    {
      "epoch": 1.1423110961914062e-05,
      "step": 11978,
      "training_step_time": 7.0603272914886475
    },
    {
      "epoch": 1.1424064636230468e-05,
      "model_forward_time": 1.7676997184753418,
      "step": 11979
    },
    {
      "epoch": 1.1424064636230468e-05,
      "step": 11979,
      "training_step_time": 7.647014141082764
    },
    {
      "epoch": 1.1425018310546875e-05,
      "grad_norm": 0.2560140788555145,
      "learning_rate": 9.867428438265357e-06,
      "loss": 0.0708,
      "step": 11980
    },
    {
      "epoch": 1.1425018310546875e-05,
      "model_forward_time": 1.7643299102783203,
      "step": 11980
    },
    {
      "dynamics_loss_avg": 0.06666224505752325,
      "epoch": 1.1425018310546875e-05,
      "step": 11980
    },
    {
      "action_loss_avg": 0.0037749463459476828,
      "epoch": 1.1425018310546875e-05,
      "step": 11980
    },
    {
      "epoch": 1.1425018310546875e-05,
      "step": 11980,
      "training_step_time": 7.651516675949097
    },
    {
      "epoch": 1.1425971984863281e-05,
      "model_forward_time": 1.7723240852355957,
      "step": 11981
    },
    {
      "epoch": 1.1425971984863281e-05,
      "step": 11981,
      "training_step_time": 7.753182649612427
    },
    {
      "epoch": 1.1426925659179687e-05,
      "model_forward_time": 1.7692043781280518,
      "step": 11982
    },
    {
      "epoch": 1.1426925659179687e-05,
      "step": 11982,
      "training_step_time": 8.728299140930176
    },
    {
      "epoch": 1.1427879333496094e-05,
      "model_forward_time": 1.7746467590332031,
      "step": 11983
    },
    {
      "epoch": 1.1427879333496094e-05,
      "step": 11983,
      "training_step_time": 7.740351915359497
    },
    {
      "epoch": 1.14288330078125e-05,
      "model_forward_time": 1.7480754852294922,
      "step": 11984
    },
    {
      "epoch": 1.14288330078125e-05,
      "step": 11984,
      "training_step_time": 9.792404890060425
    },
    {
      "epoch": 1.1429786682128906e-05,
      "model_forward_time": 1.8021752834320068,
      "step": 11985
    },
    {
      "epoch": 1.1429786682128906e-05,
      "step": 11985,
      "training_step_time": 8.681989669799805
    },
    {
      "epoch": 1.1430740356445313e-05,
      "model_forward_time": 1.7551214694976807,
      "step": 11986
    },
    {
      "epoch": 1.1430740356445313e-05,
      "step": 11986,
      "training_step_time": 7.616046667098999
    },
    {
      "epoch": 1.1431694030761719e-05,
      "model_forward_time": 1.7365288734436035,
      "step": 11987
    },
    {
      "epoch": 1.1431694030761719e-05,
      "step": 11987,
      "training_step_time": 6.991641283035278
    },
    {
      "epoch": 1.1432647705078125e-05,
      "model_forward_time": 1.7672202587127686,
      "step": 11988
    },
    {
      "epoch": 1.1432647705078125e-05,
      "step": 11988,
      "training_step_time": 7.62192440032959
    },
    {
      "epoch": 1.1433601379394532e-05,
      "model_forward_time": 1.7577438354492188,
      "step": 11989
    },
    {
      "epoch": 1.1433601379394532e-05,
      "step": 11989,
      "training_step_time": 8.371583223342896
    },
    {
      "epoch": 1.1434555053710938e-05,
      "grad_norm": 0.31721192598342896,
      "learning_rate": 9.867049944325165e-06,
      "loss": 0.0723,
      "step": 11990
    },
    {
      "epoch": 1.1434555053710938e-05,
      "model_forward_time": 1.7566585540771484,
      "step": 11990
    },
    {
      "dynamics_loss_avg": 0.06465930379927158,
      "epoch": 1.1434555053710938e-05,
      "step": 11990
    },
    {
      "action_loss_avg": 0.0024623673234600574,
      "epoch": 1.1434555053710938e-05,
      "step": 11990
    },
    {
      "epoch": 1.1434555053710938e-05,
      "step": 11990,
      "training_step_time": 7.6791627407073975
    },
    {
      "epoch": 1.1435508728027343e-05,
      "model_forward_time": 1.769261121749878,
      "step": 11991
    },
    {
      "epoch": 1.1435508728027343e-05,
      "step": 11991,
      "training_step_time": 9.31423830986023
    },
    {
      "epoch": 1.143646240234375e-05,
      "model_forward_time": 1.763911485671997,
      "step": 11992
    },
    {
      "epoch": 1.143646240234375e-05,
      "step": 11992,
      "training_step_time": 7.760200262069702
    },
    {
      "epoch": 1.1437416076660157e-05,
      "model_forward_time": 1.7509527206420898,
      "step": 11993
    },
    {
      "epoch": 1.1437416076660157e-05,
      "step": 11993,
      "training_step_time": 7.659647464752197
    },
    {
      "epoch": 1.1438369750976562e-05,
      "model_forward_time": 1.5951709747314453,
      "step": 11994
    },
    {
      "epoch": 1.1438369750976562e-05,
      "step": 11994,
      "training_step_time": 8.746402025222778
    },
    {
      "epoch": 1.1439323425292968e-05,
      "model_forward_time": 1.7335078716278076,
      "step": 11995
    },
    {
      "epoch": 1.1439323425292968e-05,
      "step": 11995,
      "training_step_time": 7.600921392440796
    },
    {
      "epoch": 1.1440277099609375e-05,
      "model_forward_time": 1.751795768737793,
      "step": 11996
    },
    {
      "epoch": 1.1440277099609375e-05,
      "step": 11996,
      "training_step_time": 7.188296794891357
    },
    {
      "epoch": 1.1441230773925781e-05,
      "model_forward_time": 1.7673516273498535,
      "step": 11997
    },
    {
      "epoch": 1.1441230773925781e-05,
      "step": 11997,
      "training_step_time": 7.7043962478637695
    },
    {
      "epoch": 1.1442184448242187e-05,
      "model_forward_time": 1.7840018272399902,
      "step": 11998
    },
    {
      "epoch": 1.1442184448242187e-05,
      "step": 11998,
      "training_step_time": 8.147896528244019
    },
    {
      "epoch": 1.1443138122558594e-05,
      "model_forward_time": 1.0920183658599854,
      "step": 11999
    },
    {
      "epoch": 1.1443138122558594e-05,
      "step": 11999,
      "training_step_time": 7.6695404052734375
    },
    {
      "epoch": 1.1444091796875e-05,
      "grad_norm": 0.3800320029258728,
      "learning_rate": 9.86667091813171e-06,
      "loss": 0.0754,
      "step": 12000
    },
    {
      "epoch": 1.1444091796875e-05,
      "model_forward_time": 1.7223474979400635,
      "step": 12000
    },
    {
      "dynamics_loss_avg": 0.05769921187311411,
      "epoch": 1.1444091796875e-05,
      "step": 12000
    },
    {
      "action_loss_avg": 0.002751199377235025,
      "epoch": 1.1444091796875e-05,
      "step": 12000
    },
    {
      "epoch": 1.1444091796875e-05,
      "step": 12000,
      "training_step_time": 11.037520408630371
    },
    {
      "epoch": 1.1445045471191406e-05,
      "model_forward_time": 1.7494122982025146,
      "step": 12001
    },
    {
      "epoch": 1.1445045471191406e-05,
      "step": 12001,
      "training_step_time": 10.160300254821777
    },
    {
      "epoch": 1.1445999145507813e-05,
      "model_forward_time": 1.7252299785614014,
      "step": 12002
    },
    {
      "epoch": 1.1445999145507813e-05,
      "step": 12002,
      "training_step_time": 7.645371198654175
    },
    {
      "epoch": 1.1446952819824219e-05,
      "model_forward_time": 2.7540695667266846,
      "step": 12003
    },
    {
      "epoch": 1.1446952819824219e-05,
      "step": 12003,
      "training_step_time": 8.876598358154297
    },
    {
      "epoch": 1.1447906494140625e-05,
      "model_forward_time": 1.7363526821136475,
      "step": 12004
    },
    {
      "epoch": 1.1447906494140625e-05,
      "step": 12004,
      "training_step_time": 7.746646165847778
    },
    {
      "epoch": 1.1448860168457032e-05,
      "model_forward_time": 1.733243465423584,
      "step": 12005
    },
    {
      "epoch": 1.1448860168457032e-05,
      "step": 12005,
      "training_step_time": 7.3151774406433105
    },
    {
      "epoch": 1.1449813842773438e-05,
      "model_forward_time": 1.7433242797851562,
      "step": 12006
    },
    {
      "epoch": 1.1449813842773438e-05,
      "step": 12006,
      "training_step_time": 9.324512004852295
    },
    {
      "epoch": 1.1450767517089843e-05,
      "model_forward_time": 1.7399404048919678,
      "step": 12007
    },
    {
      "epoch": 1.1450767517089843e-05,
      "step": 12007,
      "training_step_time": 8.554366111755371
    },
    {
      "epoch": 1.145172119140625e-05,
      "model_forward_time": 1.7561063766479492,
      "step": 12008
    },
    {
      "epoch": 1.145172119140625e-05,
      "step": 12008,
      "training_step_time": 9.650519847869873
    },
    {
      "epoch": 1.1452674865722657e-05,
      "model_forward_time": 1.748918056488037,
      "step": 12009
    },
    {
      "epoch": 1.1452674865722657e-05,
      "step": 12009,
      "training_step_time": 8.37600827217102
    },
    {
      "epoch": 1.1453628540039062e-05,
      "grad_norm": 0.19824974238872528,
      "learning_rate": 9.866291359726439e-06,
      "loss": 0.0756,
      "step": 12010
    },
    {
      "epoch": 1.1453628540039062e-05,
      "model_forward_time": 1.7261896133422852,
      "step": 12010
    },
    {
      "dynamics_loss_avg": 0.05960044786334038,
      "epoch": 1.1453628540039062e-05,
      "step": 12010
    },
    {
      "action_loss_avg": 0.0033818782889284194,
      "epoch": 1.1453628540039062e-05,
      "step": 12010
    },
    {
      "epoch": 1.1453628540039062e-05,
      "step": 12010,
      "training_step_time": 9.506468057632446
    },
    {
      "epoch": 1.1454582214355468e-05,
      "model_forward_time": 1.749955654144287,
      "step": 12011
    },
    {
      "epoch": 1.1454582214355468e-05,
      "step": 12011,
      "training_step_time": 8.727687358856201
    },
    {
      "epoch": 1.1455535888671875e-05,
      "model_forward_time": 2.932788848876953,
      "step": 12012
    },
    {
      "epoch": 1.1455535888671875e-05,
      "step": 12012,
      "training_step_time": 7.79775071144104
    },
    {
      "epoch": 1.1456489562988281e-05,
      "model_forward_time": 1.725431203842163,
      "step": 12013
    },
    {
      "epoch": 1.1456489562988281e-05,
      "step": 12013,
      "training_step_time": 7.768364429473877
    },
    {
      "epoch": 1.1457443237304687e-05,
      "model_forward_time": 1.7306692600250244,
      "step": 12014
    },
    {
      "epoch": 1.1457443237304687e-05,
      "step": 12014,
      "training_step_time": 7.66302752494812
    },
    {
      "epoch": 1.1458396911621094e-05,
      "model_forward_time": 1.7314276695251465,
      "step": 12015
    },
    {
      "epoch": 1.1458396911621094e-05,
      "step": 12015,
      "training_step_time": 8.582457304000854
    },
    {
      "epoch": 1.14593505859375e-05,
      "model_forward_time": 1.7311220169067383,
      "step": 12016
    },
    {
      "epoch": 1.14593505859375e-05,
      "step": 12016,
      "training_step_time": 8.66988730430603
    },
    {
      "epoch": 1.1460304260253906e-05,
      "model_forward_time": 1.7371318340301514,
      "step": 12017
    },
    {
      "epoch": 1.1460304260253906e-05,
      "step": 12017,
      "training_step_time": 8.378460168838501
    },
    {
      "epoch": 1.1461257934570313e-05,
      "model_forward_time": 1.753814458847046,
      "step": 12018
    },
    {
      "epoch": 1.1461257934570313e-05,
      "step": 12018,
      "training_step_time": 7.795897483825684
    },
    {
      "epoch": 1.1462211608886719e-05,
      "model_forward_time": 1.7360928058624268,
      "step": 12019
    },
    {
      "epoch": 1.1462211608886719e-05,
      "step": 12019,
      "training_step_time": 8.489254236221313
    },
    {
      "epoch": 1.1463165283203125e-05,
      "grad_norm": 0.4175103008747101,
      "learning_rate": 9.865911269150862e-06,
      "loss": 0.0733,
      "step": 12020
    },
    {
      "epoch": 1.1463165283203125e-05,
      "model_forward_time": 1.7547895908355713,
      "step": 12020
    },
    {
      "dynamics_loss_avg": 0.08690661340951919,
      "epoch": 1.1463165283203125e-05,
      "step": 12020
    },
    {
      "action_loss_avg": 0.008196250966284424,
      "epoch": 1.1463165283203125e-05,
      "step": 12020
    },
    {
      "epoch": 1.1463165283203125e-05,
      "step": 12020,
      "training_step_time": 8.74058985710144
    },
    {
      "epoch": 1.1464118957519532e-05,
      "model_forward_time": 2.880182981491089,
      "step": 12021
    },
    {
      "epoch": 1.1464118957519532e-05,
      "step": 12021,
      "training_step_time": 7.774385452270508
    },
    {
      "epoch": 1.1465072631835938e-05,
      "model_forward_time": 1.7393951416015625,
      "step": 12022
    },
    {
      "epoch": 1.1465072631835938e-05,
      "step": 12022,
      "training_step_time": 7.7473859786987305
    },
    {
      "epoch": 1.1466026306152343e-05,
      "model_forward_time": 1.7415657043457031,
      "step": 12023
    },
    {
      "epoch": 1.1466026306152343e-05,
      "step": 12023,
      "training_step_time": 7.615276336669922
    },
    {
      "epoch": 1.146697998046875e-05,
      "model_forward_time": 1.7368357181549072,
      "step": 12024
    },
    {
      "epoch": 1.146697998046875e-05,
      "step": 12024,
      "training_step_time": 7.6289403438568115
    },
    {
      "epoch": 1.1467933654785157e-05,
      "model_forward_time": 1.7403266429901123,
      "step": 12025
    },
    {
      "epoch": 1.1467933654785157e-05,
      "step": 12025,
      "training_step_time": 7.690242767333984
    },
    {
      "epoch": 1.1468887329101562e-05,
      "model_forward_time": 1.733412265777588,
      "step": 12026
    },
    {
      "epoch": 1.1468887329101562e-05,
      "step": 12026,
      "training_step_time": 8.114845275878906
    },
    {
      "epoch": 1.1469841003417968e-05,
      "model_forward_time": 1.7494118213653564,
      "step": 12027
    },
    {
      "epoch": 1.1469841003417968e-05,
      "step": 12027,
      "training_step_time": 8.540672302246094
    },
    {
      "epoch": 1.1470794677734375e-05,
      "model_forward_time": 1.7382512092590332,
      "step": 12028
    },
    {
      "epoch": 1.1470794677734375e-05,
      "step": 12028,
      "training_step_time": 8.929092168807983
    },
    {
      "epoch": 1.1471748352050781e-05,
      "model_forward_time": 1.7398240566253662,
      "step": 12029
    },
    {
      "epoch": 1.1471748352050781e-05,
      "step": 12029,
      "training_step_time": 8.364307641983032
    },
    {
      "epoch": 1.1472702026367187e-05,
      "grad_norm": 0.2921599745750427,
      "learning_rate": 9.865530646446544e-06,
      "loss": 0.0692,
      "step": 12030
    },
    {
      "epoch": 1.1472702026367187e-05,
      "model_forward_time": 2.84831166267395,
      "step": 12030
    },
    {
      "dynamics_loss_avg": 0.055669380351901054,
      "epoch": 1.1472702026367187e-05,
      "step": 12030
    },
    {
      "action_loss_avg": 0.0038259682478383183,
      "epoch": 1.1472702026367187e-05,
      "step": 12030
    },
    {
      "epoch": 1.1472702026367187e-05,
      "step": 12030,
      "training_step_time": 7.72329568862915
    },
    {
      "epoch": 1.1473655700683594e-05,
      "model_forward_time": 1.7397022247314453,
      "step": 12031
    },
    {
      "epoch": 1.1473655700683594e-05,
      "step": 12031,
      "training_step_time": 8.787033081054688
    },
    {
      "epoch": 1.1474609375e-05,
      "model_forward_time": 1.744600772857666,
      "step": 12032
    },
    {
      "epoch": 1.1474609375e-05,
      "step": 12032,
      "training_step_time": 7.645291328430176
    },
    {
      "epoch": 1.1475563049316406e-05,
      "model_forward_time": 1.747704029083252,
      "step": 12033
    },
    {
      "epoch": 1.1475563049316406e-05,
      "step": 12033,
      "training_step_time": 7.077553987503052
    },
    {
      "epoch": 1.1476516723632813e-05,
      "model_forward_time": 1.731020212173462,
      "step": 12034
    },
    {
      "epoch": 1.1476516723632813e-05,
      "step": 12034,
      "training_step_time": 7.673398494720459
    },
    {
      "epoch": 1.1477470397949219e-05,
      "model_forward_time": 1.7288713455200195,
      "step": 12035
    },
    {
      "epoch": 1.1477470397949219e-05,
      "step": 12035,
      "training_step_time": 7.3747124671936035
    },
    {
      "epoch": 1.1478424072265625e-05,
      "model_forward_time": 1.7516560554504395,
      "step": 12036
    },
    {
      "epoch": 1.1478424072265625e-05,
      "step": 12036,
      "training_step_time": 7.832807540893555
    },
    {
      "epoch": 1.1479377746582032e-05,
      "model_forward_time": 1.7433269023895264,
      "step": 12037
    },
    {
      "epoch": 1.1479377746582032e-05,
      "step": 12037,
      "training_step_time": 8.702736854553223
    },
    {
      "epoch": 1.1480331420898438e-05,
      "model_forward_time": 1.7448358535766602,
      "step": 12038
    },
    {
      "epoch": 1.1480331420898438e-05,
      "step": 12038,
      "training_step_time": 8.785938501358032
    },
    {
      "epoch": 1.1481285095214843e-05,
      "model_forward_time": 1.7713367938995361,
      "step": 12039
    },
    {
      "epoch": 1.1481285095214843e-05,
      "step": 12039,
      "training_step_time": 7.71813440322876
    },
    {
      "epoch": 1.148223876953125e-05,
      "grad_norm": 0.23762166500091553,
      "learning_rate": 9.865149491655111e-06,
      "loss": 0.0713,
      "step": 12040
    },
    {
      "epoch": 1.148223876953125e-05,
      "model_forward_time": 3.004546880722046,
      "step": 12040
    },
    {
      "dynamics_loss_avg": 0.06926156468689441,
      "epoch": 1.148223876953125e-05,
      "step": 12040
    },
    {
      "action_loss_avg": 0.005056089395657182,
      "epoch": 1.148223876953125e-05,
      "step": 12040
    },
    {
      "epoch": 1.148223876953125e-05,
      "step": 12040,
      "training_step_time": 7.872582674026489
    },
    {
      "epoch": 1.1483192443847657e-05,
      "model_forward_time": 1.7384028434753418,
      "step": 12041
    },
    {
      "epoch": 1.1483192443847657e-05,
      "step": 12041,
      "training_step_time": 7.730968475341797
    },
    {
      "epoch": 1.1484146118164062e-05,
      "model_forward_time": 1.7286088466644287,
      "step": 12042
    },
    {
      "epoch": 1.1484146118164062e-05,
      "step": 12042,
      "training_step_time": 7.622172594070435
    },
    {
      "epoch": 1.1485099792480468e-05,
      "model_forward_time": 1.7351861000061035,
      "step": 12043
    },
    {
      "epoch": 1.1485099792480468e-05,
      "step": 12043,
      "training_step_time": 7.639674663543701
    },
    {
      "epoch": 1.1486053466796875e-05,
      "model_forward_time": 1.3610072135925293,
      "step": 12044
    },
    {
      "epoch": 1.1486053466796875e-05,
      "step": 12044,
      "training_step_time": 7.708390235900879
    },
    {
      "epoch": 1.1487007141113281e-05,
      "model_forward_time": 1.7292094230651855,
      "step": 12045
    },
    {
      "epoch": 1.1487007141113281e-05,
      "step": 12045,
      "training_step_time": 8.395787715911865
    },
    {
      "epoch": 1.1487960815429687e-05,
      "model_forward_time": 1.7578544616699219,
      "step": 12046
    },
    {
      "epoch": 1.1487960815429687e-05,
      "step": 12046,
      "training_step_time": 7.740222692489624
    },
    {
      "epoch": 1.1488914489746094e-05,
      "model_forward_time": 1.7433216571807861,
      "step": 12047
    },
    {
      "epoch": 1.1488914489746094e-05,
      "step": 12047,
      "training_step_time": 8.206767320632935
    },
    {
      "epoch": 1.14898681640625e-05,
      "model_forward_time": 1.7528278827667236,
      "step": 12048
    },
    {
      "epoch": 1.14898681640625e-05,
      "step": 12048,
      "training_step_time": 8.736685752868652
    },
    {
      "epoch": 1.1490821838378906e-05,
      "model_forward_time": 2.9659054279327393,
      "step": 12049
    },
    {
      "epoch": 1.1490821838378906e-05,
      "step": 12049,
      "training_step_time": 9.460261106491089
    },
    {
      "epoch": 1.1491775512695313e-05,
      "grad_norm": 0.20966199040412903,
      "learning_rate": 9.864767804818242e-06,
      "loss": 0.0684,
      "step": 12050
    },
    {
      "epoch": 1.1491775512695313e-05,
      "model_forward_time": 1.726700782775879,
      "step": 12050
    },
    {
      "dynamics_loss_avg": 0.05468099066056311,
      "epoch": 1.1491775512695313e-05,
      "step": 12050
    },
    {
      "action_loss_avg": 0.004141989082563668,
      "epoch": 1.1491775512695313e-05,
      "step": 12050
    },
    {
      "epoch": 1.1491775512695313e-05,
      "step": 12050,
      "training_step_time": 7.770161867141724
    },
    {
      "epoch": 1.1492729187011719e-05,
      "model_forward_time": 1.7479345798492432,
      "step": 12051
    },
    {
      "epoch": 1.1492729187011719e-05,
      "step": 12051,
      "training_step_time": 7.599135875701904
    },
    {
      "epoch": 1.1493682861328125e-05,
      "model_forward_time": 1.7402253150939941,
      "step": 12052
    },
    {
      "epoch": 1.1493682861328125e-05,
      "step": 12052,
      "training_step_time": 7.744274616241455
    },
    {
      "epoch": 1.1494636535644532e-05,
      "model_forward_time": 1.7311668395996094,
      "step": 12053
    },
    {
      "epoch": 1.1494636535644532e-05,
      "step": 12053,
      "training_step_time": 8.314934253692627
    },
    {
      "epoch": 1.1495590209960938e-05,
      "model_forward_time": 1.7445828914642334,
      "step": 12054
    },
    {
      "epoch": 1.1495590209960938e-05,
      "step": 12054,
      "training_step_time": 7.641808986663818
    },
    {
      "epoch": 1.1496543884277343e-05,
      "model_forward_time": 1.750781774520874,
      "step": 12055
    },
    {
      "epoch": 1.1496543884277343e-05,
      "step": 12055,
      "training_step_time": 8.438016414642334
    },
    {
      "epoch": 1.149749755859375e-05,
      "model_forward_time": 1.7355315685272217,
      "step": 12056
    },
    {
      "epoch": 1.149749755859375e-05,
      "step": 12056,
      "training_step_time": 8.63953161239624
    },
    {
      "epoch": 1.1498451232910157e-05,
      "model_forward_time": 1.7542922496795654,
      "step": 12057
    },
    {
      "epoch": 1.1498451232910157e-05,
      "step": 12057,
      "training_step_time": 8.419236183166504
    },
    {
      "epoch": 1.1499404907226562e-05,
      "model_forward_time": 2.939567804336548,
      "step": 12058
    },
    {
      "epoch": 1.1499404907226562e-05,
      "step": 12058,
      "training_step_time": 7.797187805175781
    },
    {
      "epoch": 1.1500358581542968e-05,
      "model_forward_time": 1.7479476928710938,
      "step": 12059
    },
    {
      "epoch": 1.1500358581542968e-05,
      "step": 12059,
      "training_step_time": 7.800812721252441
    },
    {
      "epoch": 1.1501312255859375e-05,
      "grad_norm": 0.273261159658432,
      "learning_rate": 9.86438558597768e-06,
      "loss": 0.0705,
      "step": 12060
    },
    {
      "epoch": 1.1501312255859375e-05,
      "model_forward_time": 1.738161563873291,
      "step": 12060
    },
    {
      "dynamics_loss_avg": 0.06609776969999075,
      "epoch": 1.1501312255859375e-05,
      "step": 12060
    },
    {
      "action_loss_avg": 0.0038536649604793637,
      "epoch": 1.1501312255859375e-05,
      "step": 12060
    },
    {
      "epoch": 1.1501312255859375e-05,
      "step": 12060,
      "training_step_time": 7.628174543380737
    },
    {
      "epoch": 1.1502265930175781e-05,
      "model_forward_time": 1.7377574443817139,
      "step": 12061
    },
    {
      "epoch": 1.1502265930175781e-05,
      "step": 12061,
      "training_step_time": 7.711039066314697
    },
    {
      "epoch": 1.1503219604492187e-05,
      "model_forward_time": 1.7518372535705566,
      "step": 12062
    },
    {
      "epoch": 1.1503219604492187e-05,
      "step": 12062,
      "training_step_time": 7.6784117221832275
    },
    {
      "epoch": 1.1504173278808594e-05,
      "model_forward_time": 1.7397527694702148,
      "step": 12063
    },
    {
      "epoch": 1.1504173278808594e-05,
      "step": 12063,
      "training_step_time": 7.613969087600708
    },
    {
      "epoch": 1.1505126953125e-05,
      "model_forward_time": 1.764939546585083,
      "step": 12064
    },
    {
      "epoch": 1.1505126953125e-05,
      "step": 12064,
      "training_step_time": 8.729861736297607
    },
    {
      "epoch": 1.1506080627441406e-05,
      "model_forward_time": 1.7412655353546143,
      "step": 12065
    },
    {
      "epoch": 1.1506080627441406e-05,
      "step": 12065,
      "training_step_time": 8.723073244094849
    },
    {
      "epoch": 1.1507034301757813e-05,
      "model_forward_time": 1.7364764213562012,
      "step": 12066
    },
    {
      "epoch": 1.1507034301757813e-05,
      "step": 12066,
      "training_step_time": 7.71894097328186
    },
    {
      "epoch": 1.1507987976074219e-05,
      "model_forward_time": 2.8455703258514404,
      "step": 12067
    },
    {
      "epoch": 1.1507987976074219e-05,
      "step": 12067,
      "training_step_time": 7.789958238601685
    },
    {
      "epoch": 1.1508941650390625e-05,
      "model_forward_time": 1.7370481491088867,
      "step": 12068
    },
    {
      "epoch": 1.1508941650390625e-05,
      "step": 12068,
      "training_step_time": 7.803866863250732
    },
    {
      "epoch": 1.1509895324707032e-05,
      "model_forward_time": 1.7271618843078613,
      "step": 12069
    },
    {
      "epoch": 1.1509895324707032e-05,
      "step": 12069,
      "training_step_time": 7.466080188751221
    },
    {
      "epoch": 1.1510848999023438e-05,
      "grad_norm": 0.19363956153392792,
      "learning_rate": 9.864002835175226e-06,
      "loss": 0.0654,
      "step": 12070
    },
    {
      "epoch": 1.1510848999023438e-05,
      "model_forward_time": 1.7451269626617432,
      "step": 12070
    },
    {
      "dynamics_loss_avg": 0.05673935599625111,
      "epoch": 1.1510848999023438e-05,
      "step": 12070
    },
    {
      "action_loss_avg": 0.0031152442446909846,
      "epoch": 1.1510848999023438e-05,
      "step": 12070
    },
    {
      "epoch": 1.1510848999023438e-05,
      "step": 12070,
      "training_step_time": 7.719700574874878
    },
    {
      "epoch": 1.1511802673339843e-05,
      "model_forward_time": 1.7370338439941406,
      "step": 12071
    },
    {
      "epoch": 1.1511802673339843e-05,
      "step": 12071,
      "training_step_time": 7.698080539703369
    },
    {
      "epoch": 1.151275634765625e-05,
      "model_forward_time": 1.7388010025024414,
      "step": 12072
    },
    {
      "epoch": 1.151275634765625e-05,
      "step": 12072,
      "training_step_time": 8.686988115310669
    },
    {
      "epoch": 1.1513710021972657e-05,
      "model_forward_time": 1.755958080291748,
      "step": 12073
    },
    {
      "epoch": 1.1513710021972657e-05,
      "step": 12073,
      "training_step_time": 7.727141618728638
    },
    {
      "epoch": 1.1514663696289062e-05,
      "model_forward_time": 1.7420473098754883,
      "step": 12074
    },
    {
      "epoch": 1.1514663696289062e-05,
      "step": 12074,
      "training_step_time": 7.893054723739624
    },
    {
      "epoch": 1.1515617370605468e-05,
      "model_forward_time": 1.7310268878936768,
      "step": 12075
    },
    {
      "epoch": 1.1515617370605468e-05,
      "step": 12075,
      "training_step_time": 9.466785907745361
    },
    {
      "epoch": 1.1516571044921875e-05,
      "model_forward_time": 2.264655590057373,
      "step": 12076
    },
    {
      "epoch": 1.1516571044921875e-05,
      "step": 12076,
      "training_step_time": 8.41526746749878
    },
    {
      "epoch": 1.1517524719238281e-05,
      "model_forward_time": 1.7392995357513428,
      "step": 12077
    },
    {
      "epoch": 1.1517524719238281e-05,
      "step": 12077,
      "training_step_time": 7.832067966461182
    },
    {
      "epoch": 1.1518478393554687e-05,
      "model_forward_time": 1.7967360019683838,
      "step": 12078
    },
    {
      "epoch": 1.1518478393554687e-05,
      "step": 12078,
      "training_step_time": 7.386361360549927
    },
    {
      "epoch": 1.1519432067871094e-05,
      "model_forward_time": 1.7369375228881836,
      "step": 12079
    },
    {
      "epoch": 1.1519432067871094e-05,
      "step": 12079,
      "training_step_time": 8.170557022094727
    },
    {
      "epoch": 1.15203857421875e-05,
      "grad_norm": 0.2704731225967407,
      "learning_rate": 9.863619552452734e-06,
      "loss": 0.0686,
      "step": 12080
    },
    {
      "epoch": 1.15203857421875e-05,
      "model_forward_time": 1.7352168560028076,
      "step": 12080
    },
    {
      "dynamics_loss_avg": 0.04886300889775157,
      "epoch": 1.15203857421875e-05,
      "step": 12080
    },
    {
      "action_loss_avg": 0.0041276122443377975,
      "epoch": 1.15203857421875e-05,
      "step": 12080
    },
    {
      "epoch": 1.15203857421875e-05,
      "step": 12080,
      "training_step_time": 7.604770660400391
    },
    {
      "epoch": 1.1521339416503906e-05,
      "model_forward_time": 1.7375860214233398,
      "step": 12081
    },
    {
      "epoch": 1.1521339416503906e-05,
      "step": 12081,
      "training_step_time": 7.802880525588989
    },
    {
      "epoch": 1.1522293090820313e-05,
      "model_forward_time": 1.7418932914733887,
      "step": 12082
    },
    {
      "epoch": 1.1522293090820313e-05,
      "step": 12082,
      "training_step_time": 8.71937894821167
    },
    {
      "epoch": 1.1523246765136719e-05,
      "model_forward_time": 1.7363865375518799,
      "step": 12083
    },
    {
      "epoch": 1.1523246765136719e-05,
      "step": 12083,
      "training_step_time": 8.367044687271118
    },
    {
      "epoch": 1.1524200439453125e-05,
      "model_forward_time": 1.7440886497497559,
      "step": 12084
    },
    {
      "epoch": 1.1524200439453125e-05,
      "step": 12084,
      "training_step_time": 8.518461227416992
    },
    {
      "epoch": 1.1525154113769532e-05,
      "model_forward_time": 1.731604814529419,
      "step": 12085
    },
    {
      "epoch": 1.1525154113769532e-05,
      "step": 12085,
      "training_step_time": 8.766762733459473
    },
    {
      "epoch": 1.1526107788085938e-05,
      "model_forward_time": 1.7412919998168945,
      "step": 12086
    },
    {
      "epoch": 1.1526107788085938e-05,
      "step": 12086,
      "training_step_time": 8.65720820426941
    },
    {
      "epoch": 1.1527061462402344e-05,
      "model_forward_time": 1.781963586807251,
      "step": 12087
    },
    {
      "epoch": 1.1527061462402344e-05,
      "step": 12087,
      "training_step_time": 7.699963331222534
    },
    {
      "epoch": 1.152801513671875e-05,
      "model_forward_time": 1.7131381034851074,
      "step": 12088
    },
    {
      "epoch": 1.152801513671875e-05,
      "step": 12088,
      "training_step_time": 8.567306280136108
    },
    {
      "epoch": 1.1528968811035157e-05,
      "model_forward_time": 1.734290599822998,
      "step": 12089
    },
    {
      "epoch": 1.1528968811035157e-05,
      "step": 12089,
      "training_step_time": 7.725133895874023
    },
    {
      "epoch": 1.1529922485351562e-05,
      "grad_norm": 0.22538819909095764,
      "learning_rate": 9.86323573785212e-06,
      "loss": 0.0707,
      "step": 12090
    },
    {
      "epoch": 1.1529922485351562e-05,
      "model_forward_time": 1.7379481792449951,
      "step": 12090
    },
    {
      "dynamics_loss_avg": 0.05727410167455673,
      "epoch": 1.1529922485351562e-05,
      "step": 12090
    },
    {
      "action_loss_avg": 0.0042576557432767006,
      "epoch": 1.1529922485351562e-05,
      "step": 12090
    },
    {
      "epoch": 1.1529922485351562e-05,
      "step": 12090,
      "training_step_time": 7.71726131439209
    },
    {
      "epoch": 1.1530876159667968e-05,
      "model_forward_time": 1.7351984977722168,
      "step": 12091
    },
    {
      "epoch": 1.1530876159667968e-05,
      "step": 12091,
      "training_step_time": 8.2721529006958
    },
    {
      "epoch": 1.1531829833984376e-05,
      "model_forward_time": 1.7372403144836426,
      "step": 12092
    },
    {
      "epoch": 1.1531829833984376e-05,
      "step": 12092,
      "training_step_time": 8.722050428390503
    },
    {
      "epoch": 1.1532783508300781e-05,
      "model_forward_time": 1.7370102405548096,
      "step": 12093
    },
    {
      "epoch": 1.1532783508300781e-05,
      "step": 12093,
      "training_step_time": 9.5536470413208
    },
    {
      "epoch": 1.1533737182617187e-05,
      "model_forward_time": 1.7355241775512695,
      "step": 12094
    },
    {
      "epoch": 1.1533737182617187e-05,
      "step": 12094,
      "training_step_time": 7.719210624694824
    },
    {
      "epoch": 1.1534690856933594e-05,
      "model_forward_time": 2.252253770828247,
      "step": 12095
    },
    {
      "epoch": 1.1534690856933594e-05,
      "step": 12095,
      "training_step_time": 7.636731863021851
    },
    {
      "epoch": 1.153564453125e-05,
      "model_forward_time": 1.7795979976654053,
      "step": 12096
    },
    {
      "epoch": 1.153564453125e-05,
      "step": 12096,
      "training_step_time": 7.749756813049316
    },
    {
      "epoch": 1.1536598205566406e-05,
      "model_forward_time": 1.7199301719665527,
      "step": 12097
    },
    {
      "epoch": 1.1536598205566406e-05,
      "step": 12097,
      "training_step_time": 7.625819444656372
    },
    {
      "epoch": 1.1537551879882813e-05,
      "model_forward_time": 1.7370083332061768,
      "step": 12098
    },
    {
      "epoch": 1.1537551879882813e-05,
      "step": 12098,
      "training_step_time": 7.629650354385376
    },
    {
      "epoch": 1.1538505554199219e-05,
      "model_forward_time": 1.7427475452423096,
      "step": 12099
    },
    {
      "epoch": 1.1538505554199219e-05,
      "step": 12099,
      "training_step_time": 7.674535274505615
    },
    {
      "epoch": 1.1539459228515625e-05,
      "grad_norm": 0.29804927110671997,
      "learning_rate": 9.862851391415357e-06,
      "loss": 0.0681,
      "step": 12100
    },
    {
      "epoch": 1.1539459228515625e-05,
      "model_forward_time": 1.7482750415802002,
      "step": 12100
    },
    {
      "dynamics_loss_avg": 0.05716045126318932,
      "epoch": 1.1539459228515625e-05,
      "step": 12100
    },
    {
      "action_loss_avg": 0.005645787442335859,
      "epoch": 1.1539459228515625e-05,
      "step": 12100
    },
    {
      "epoch": 1.1539459228515625e-05,
      "step": 12100,
      "training_step_time": 7.689235687255859
    },
    {
      "epoch": 1.1540412902832032e-05,
      "model_forward_time": 1.7379534244537354,
      "step": 12101
    },
    {
      "epoch": 1.1540412902832032e-05,
      "step": 12101,
      "training_step_time": 7.734261751174927
    },
    {
      "epoch": 1.1541366577148438e-05,
      "model_forward_time": 1.7421596050262451,
      "step": 12102
    },
    {
      "epoch": 1.1541366577148438e-05,
      "step": 12102,
      "training_step_time": 9.85814118385315
    },
    {
      "epoch": 1.1542320251464844e-05,
      "model_forward_time": 1.7474007606506348,
      "step": 12103
    },
    {
      "epoch": 1.1542320251464844e-05,
      "step": 12103,
      "training_step_time": 9.21021032333374
    },
    {
      "epoch": 1.154327392578125e-05,
      "model_forward_time": 2.278104782104492,
      "step": 12104
    },
    {
      "epoch": 1.154327392578125e-05,
      "step": 12104,
      "training_step_time": 7.7461042404174805
    },
    {
      "epoch": 1.1544227600097657e-05,
      "model_forward_time": 1.7611982822418213,
      "step": 12105
    },
    {
      "epoch": 1.1544227600097657e-05,
      "step": 12105,
      "training_step_time": 7.7616894245147705
    },
    {
      "epoch": 1.1545181274414062e-05,
      "model_forward_time": 1.7374663352966309,
      "step": 12106
    },
    {
      "epoch": 1.1545181274414062e-05,
      "step": 12106,
      "training_step_time": 7.579555034637451
    },
    {
      "epoch": 1.1546134948730468e-05,
      "model_forward_time": 1.7590477466583252,
      "step": 12107
    },
    {
      "epoch": 1.1546134948730468e-05,
      "step": 12107,
      "training_step_time": 7.687155723571777
    },
    {
      "epoch": 1.1547088623046876e-05,
      "model_forward_time": 1.7923243045806885,
      "step": 12108
    },
    {
      "epoch": 1.1547088623046876e-05,
      "step": 12108,
      "training_step_time": 7.650851011276245
    },
    {
      "epoch": 1.1548042297363281e-05,
      "model_forward_time": 1.742103099822998,
      "step": 12109
    },
    {
      "epoch": 1.1548042297363281e-05,
      "step": 12109,
      "training_step_time": 8.125556230545044
    },
    {
      "epoch": 1.1548995971679687e-05,
      "grad_norm": 0.18507981300354004,
      "learning_rate": 9.862466513184477e-06,
      "loss": 0.0679,
      "step": 12110
    },
    {
      "epoch": 1.1548995971679687e-05,
      "model_forward_time": 1.7489402294158936,
      "step": 12110
    },
    {
      "dynamics_loss_avg": 0.057493920996785165,
      "epoch": 1.1548995971679687e-05,
      "step": 12110
    },
    {
      "action_loss_avg": 0.003942183509934694,
      "epoch": 1.1548995971679687e-05,
      "step": 12110
    },
    {
      "epoch": 1.1548995971679687e-05,
      "step": 12110,
      "training_step_time": 9.581394910812378
    },
    {
      "epoch": 1.1549949645996094e-05,
      "model_forward_time": 1.7526295185089111,
      "step": 12111
    },
    {
      "epoch": 1.1549949645996094e-05,
      "step": 12111,
      "training_step_time": 8.520820617675781
    },
    {
      "epoch": 1.15509033203125e-05,
      "model_forward_time": 1.751194953918457,
      "step": 12112
    },
    {
      "epoch": 1.15509033203125e-05,
      "step": 12112,
      "training_step_time": 8.75337529182434
    },
    {
      "epoch": 1.1551856994628906e-05,
      "model_forward_time": 2.28639817237854,
      "step": 12113
    },
    {
      "epoch": 1.1551856994628906e-05,
      "step": 12113,
      "training_step_time": 7.648448705673218
    },
    {
      "epoch": 1.1552810668945313e-05,
      "model_forward_time": 1.7490170001983643,
      "step": 12114
    },
    {
      "epoch": 1.1552810668945313e-05,
      "step": 12114,
      "training_step_time": 7.747246503829956
    },
    {
      "epoch": 1.1553764343261719e-05,
      "model_forward_time": 1.7974762916564941,
      "step": 12115
    },
    {
      "epoch": 1.1553764343261719e-05,
      "step": 12115,
      "training_step_time": 7.444600820541382
    },
    {
      "epoch": 1.1554718017578125e-05,
      "model_forward_time": 1.037919044494629,
      "step": 12116
    },
    {
      "epoch": 1.1554718017578125e-05,
      "step": 12116,
      "training_step_time": 7.6108558177948
    },
    {
      "epoch": 1.1555671691894532e-05,
      "model_forward_time": 1.7461416721343994,
      "step": 12117
    },
    {
      "epoch": 1.1555671691894532e-05,
      "step": 12117,
      "training_step_time": 7.5919883251190186
    },
    {
      "epoch": 1.1556625366210938e-05,
      "model_forward_time": 1.75262451171875,
      "step": 12118
    },
    {
      "epoch": 1.1556625366210938e-05,
      "step": 12118,
      "training_step_time": 8.707237482070923
    },
    {
      "epoch": 1.1557579040527344e-05,
      "model_forward_time": 1.747502326965332,
      "step": 12119
    },
    {
      "epoch": 1.1557579040527344e-05,
      "step": 12119,
      "training_step_time": 7.708987474441528
    },
    {
      "epoch": 1.155853271484375e-05,
      "grad_norm": 0.24223215878009796,
      "learning_rate": 9.86208110320157e-06,
      "loss": 0.0667,
      "step": 12120
    },
    {
      "epoch": 1.155853271484375e-05,
      "model_forward_time": 1.7418944835662842,
      "step": 12120
    },
    {
      "dynamics_loss_avg": 0.07160760778933764,
      "epoch": 1.155853271484375e-05,
      "step": 12120
    },
    {
      "action_loss_avg": 0.007278932980261743,
      "epoch": 1.155853271484375e-05,
      "step": 12120
    },
    {
      "epoch": 1.155853271484375e-05,
      "step": 12120,
      "training_step_time": 9.727309942245483
    },
    {
      "epoch": 1.1559486389160157e-05,
      "model_forward_time": 1.7473368644714355,
      "step": 12121
    },
    {
      "epoch": 1.1559486389160157e-05,
      "step": 12121,
      "training_step_time": 8.79931354522705
    },
    {
      "epoch": 1.1560440063476562e-05,
      "model_forward_time": 1.7355420589447021,
      "step": 12122
    },
    {
      "epoch": 1.1560440063476562e-05,
      "step": 12122,
      "training_step_time": 8.7747802734375
    },
    {
      "epoch": 1.1561393737792968e-05,
      "model_forward_time": 1.3804590702056885,
      "step": 12123
    },
    {
      "epoch": 1.1561393737792968e-05,
      "step": 12123,
      "training_step_time": 7.725657224655151
    },
    {
      "epoch": 1.1562347412109376e-05,
      "model_forward_time": 1.781768798828125,
      "step": 12124
    },
    {
      "epoch": 1.1562347412109376e-05,
      "step": 12124,
      "training_step_time": 7.6314685344696045
    },
    {
      "epoch": 1.1563301086425781e-05,
      "model_forward_time": 1.3816289901733398,
      "step": 12125
    },
    {
      "epoch": 1.1563301086425781e-05,
      "step": 12125,
      "training_step_time": 8.611213207244873
    },
    {
      "epoch": 1.1564254760742187e-05,
      "model_forward_time": 1.7414216995239258,
      "step": 12126
    },
    {
      "epoch": 1.1564254760742187e-05,
      "step": 12126,
      "training_step_time": 7.635849952697754
    },
    {
      "epoch": 1.1565208435058594e-05,
      "model_forward_time": 1.7514753341674805,
      "step": 12127
    },
    {
      "epoch": 1.1565208435058594e-05,
      "step": 12127,
      "training_step_time": 7.738466501235962
    },
    {
      "epoch": 1.1566162109375e-05,
      "model_forward_time": 1.7469911575317383,
      "step": 12128
    },
    {
      "epoch": 1.1566162109375e-05,
      "step": 12128,
      "training_step_time": 8.314613342285156
    },
    {
      "epoch": 1.1567115783691406e-05,
      "model_forward_time": 1.7511377334594727,
      "step": 12129
    },
    {
      "epoch": 1.1567115783691406e-05,
      "step": 12129,
      "training_step_time": 9.415422439575195
    },
    {
      "epoch": 1.1568069458007813e-05,
      "grad_norm": 0.3063417673110962,
      "learning_rate": 9.861695161508784e-06,
      "loss": 0.0684,
      "step": 12130
    },
    {
      "epoch": 1.1568069458007813e-05,
      "model_forward_time": 1.744204044342041,
      "step": 12130
    },
    {
      "dynamics_loss_avg": 0.04895004685968161,
      "epoch": 1.1568069458007813e-05,
      "step": 12130
    },
    {
      "action_loss_avg": 0.0037263657432049513,
      "epoch": 1.1568069458007813e-05,
      "step": 12130
    },
    {
      "epoch": 1.1568069458007813e-05,
      "step": 12130,
      "training_step_time": 7.778042554855347
    },
    {
      "epoch": 1.1569023132324219e-05,
      "model_forward_time": 1.739394187927246,
      "step": 12131
    },
    {
      "epoch": 1.1569023132324219e-05,
      "step": 12131,
      "training_step_time": 7.6485676765441895
    },
    {
      "epoch": 1.1569976806640625e-05,
      "model_forward_time": 2.2747488021850586,
      "step": 12132
    },
    {
      "epoch": 1.1569976806640625e-05,
      "step": 12132,
      "training_step_time": 7.7479963302612305
    },
    {
      "epoch": 1.157093048095703e-05,
      "model_forward_time": 1.7615361213684082,
      "step": 12133
    },
    {
      "epoch": 1.157093048095703e-05,
      "step": 12133,
      "training_step_time": 8.271491289138794
    },
    {
      "epoch": 1.1571884155273438e-05,
      "model_forward_time": 1.7473270893096924,
      "step": 12134
    },
    {
      "epoch": 1.1571884155273438e-05,
      "step": 12134,
      "training_step_time": 7.626772165298462
    },
    {
      "epoch": 1.1572837829589844e-05,
      "model_forward_time": 1.7489678859710693,
      "step": 12135
    },
    {
      "epoch": 1.1572837829589844e-05,
      "step": 12135,
      "training_step_time": 7.6203413009643555
    },
    {
      "epoch": 1.157379150390625e-05,
      "model_forward_time": 1.7486517429351807,
      "step": 12136
    },
    {
      "epoch": 1.157379150390625e-05,
      "step": 12136,
      "training_step_time": 7.745171308517456
    },
    {
      "epoch": 1.1574745178222657e-05,
      "model_forward_time": 1.7552862167358398,
      "step": 12137
    },
    {
      "epoch": 1.1574745178222657e-05,
      "step": 12137,
      "training_step_time": 7.69780158996582
    },
    {
      "epoch": 1.1575698852539062e-05,
      "model_forward_time": 1.7466120719909668,
      "step": 12138
    },
    {
      "epoch": 1.1575698852539062e-05,
      "step": 12138,
      "training_step_time": 7.691786527633667
    },
    {
      "epoch": 1.1576652526855468e-05,
      "model_forward_time": 1.7456507682800293,
      "step": 12139
    },
    {
      "epoch": 1.1576652526855468e-05,
      "step": 12139,
      "training_step_time": 8.245221614837646
    },
    {
      "epoch": 1.1577606201171876e-05,
      "grad_norm": 0.2282259166240692,
      "learning_rate": 9.861308688148323e-06,
      "loss": 0.0692,
      "step": 12140
    },
    {
      "epoch": 1.1577606201171876e-05,
      "model_forward_time": 1.743152379989624,
      "step": 12140
    },
    {
      "dynamics_loss_avg": 0.06861629635095597,
      "epoch": 1.1577606201171876e-05,
      "step": 12140
    },
    {
      "action_loss_avg": 0.003775854635750875,
      "epoch": 1.1577606201171876e-05,
      "step": 12140
    },
    {
      "epoch": 1.1577606201171876e-05,
      "step": 12140,
      "training_step_time": 9.768370389938354
    },
    {
      "epoch": 1.1578559875488281e-05,
      "model_forward_time": 2.277740001678467,
      "step": 12141
    },
    {
      "epoch": 1.1578559875488281e-05,
      "step": 12141,
      "training_step_time": 7.6829729080200195
    },
    {
      "epoch": 1.1579513549804687e-05,
      "model_forward_time": 1.726435899734497,
      "step": 12142
    },
    {
      "epoch": 1.1579513549804687e-05,
      "step": 12142,
      "training_step_time": 8.41754150390625
    },
    {
      "epoch": 1.1580467224121094e-05,
      "model_forward_time": 1.7414345741271973,
      "step": 12143
    },
    {
      "epoch": 1.1580467224121094e-05,
      "step": 12143,
      "training_step_time": 7.691490173339844
    },
    {
      "epoch": 1.15814208984375e-05,
      "model_forward_time": 1.348583698272705,
      "step": 12144
    },
    {
      "epoch": 1.15814208984375e-05,
      "step": 12144,
      "training_step_time": 7.607989311218262
    },
    {
      "epoch": 1.1582374572753906e-05,
      "model_forward_time": 1.7385904788970947,
      "step": 12145
    },
    {
      "epoch": 1.1582374572753906e-05,
      "step": 12145,
      "training_step_time": 7.182485818862915
    },
    {
      "epoch": 1.1583328247070313e-05,
      "model_forward_time": 1.3503386974334717,
      "step": 12146
    },
    {
      "epoch": 1.1583328247070313e-05,
      "step": 12146,
      "training_step_time": 8.383702278137207
    },
    {
      "epoch": 1.1584281921386719e-05,
      "model_forward_time": 1.7289917469024658,
      "step": 12147
    },
    {
      "epoch": 1.1584281921386719e-05,
      "step": 12147,
      "training_step_time": 7.689441204071045
    },
    {
      "epoch": 1.1585235595703125e-05,
      "model_forward_time": 1.7421543598175049,
      "step": 12148
    },
    {
      "epoch": 1.1585235595703125e-05,
      "step": 12148,
      "training_step_time": 8.778918981552124
    },
    {
      "epoch": 1.158618927001953e-05,
      "model_forward_time": 1.7430384159088135,
      "step": 12149
    },
    {
      "epoch": 1.158618927001953e-05,
      "step": 12149,
      "training_step_time": 7.689183235168457
    },
    {
      "epoch": 1.1587142944335938e-05,
      "grad_norm": 0.226648211479187,
      "learning_rate": 9.860921683162455e-06,
      "loss": 0.0682,
      "step": 12150
    },
    {
      "epoch": 1.1587142944335938e-05,
      "model_forward_time": 1.7268717288970947,
      "step": 12150
    },
    {
      "dynamics_loss_avg": 0.07595718912780285,
      "epoch": 1.1587142944335938e-05,
      "step": 12150
    },
    {
      "action_loss_avg": 0.005784282216336578,
      "epoch": 1.1587142944335938e-05,
      "step": 12150
    },
    {
      "epoch": 1.1587142944335938e-05,
      "step": 12150,
      "training_step_time": 7.69983696937561
    },
    {
      "epoch": 1.1588096618652344e-05,
      "model_forward_time": 1.0191707611083984,
      "step": 12151
    },
    {
      "epoch": 1.1588096618652344e-05,
      "step": 12151,
      "training_step_time": 7.73943829536438
    },
    {
      "epoch": 1.158905029296875e-05,
      "model_forward_time": 1.731752872467041,
      "step": 12152
    },
    {
      "epoch": 1.158905029296875e-05,
      "step": 12152,
      "training_step_time": 7.5688560009002686
    },
    {
      "epoch": 1.1590003967285157e-05,
      "model_forward_time": 1.7339038848876953,
      "step": 12153
    },
    {
      "epoch": 1.1590003967285157e-05,
      "step": 12153,
      "training_step_time": 7.620947599411011
    },
    {
      "epoch": 1.1590957641601562e-05,
      "model_forward_time": 1.7342162132263184,
      "step": 12154
    },
    {
      "epoch": 1.1590957641601562e-05,
      "step": 12154,
      "training_step_time": 7.123896598815918
    },
    {
      "epoch": 1.1591911315917968e-05,
      "model_forward_time": 1.736396312713623,
      "step": 12155
    },
    {
      "epoch": 1.1591911315917968e-05,
      "step": 12155,
      "training_step_time": 8.160017251968384
    },
    {
      "epoch": 1.1592864990234376e-05,
      "model_forward_time": 1.7332170009613037,
      "step": 12156
    },
    {
      "epoch": 1.1592864990234376e-05,
      "step": 12156,
      "training_step_time": 8.857447624206543
    },
    {
      "epoch": 1.1593818664550781e-05,
      "model_forward_time": 1.7478423118591309,
      "step": 12157
    },
    {
      "epoch": 1.1593818664550781e-05,
      "step": 12157,
      "training_step_time": 9.399098873138428
    },
    {
      "epoch": 1.1594772338867187e-05,
      "model_forward_time": 1.7850096225738525,
      "step": 12158
    },
    {
      "epoch": 1.1594772338867187e-05,
      "step": 12158,
      "training_step_time": 7.656615257263184
    },
    {
      "epoch": 1.1595726013183594e-05,
      "model_forward_time": 1.7260799407958984,
      "step": 12159
    },
    {
      "epoch": 1.1595726013183594e-05,
      "step": 12159,
      "training_step_time": 8.288853645324707
    },
    {
      "epoch": 1.15966796875e-05,
      "grad_norm": 0.1907573789358139,
      "learning_rate": 9.8605341465935e-06,
      "loss": 0.0681,
      "step": 12160
    },
    {
      "epoch": 1.15966796875e-05,
      "model_forward_time": 2.9521148204803467,
      "step": 12160
    },
    {
      "dynamics_loss_avg": 0.06523709362372757,
      "epoch": 1.15966796875e-05,
      "step": 12160
    },
    {
      "action_loss_avg": 0.005030390957836061,
      "epoch": 1.15966796875e-05,
      "step": 12160
    },
    {
      "epoch": 1.15966796875e-05,
      "step": 12160,
      "training_step_time": 7.775848865509033
    },
    {
      "epoch": 1.1597633361816406e-05,
      "model_forward_time": 1.728356122970581,
      "step": 12161
    },
    {
      "epoch": 1.1597633361816406e-05,
      "step": 12161,
      "training_step_time": 8.395780086517334
    },
    {
      "epoch": 1.1598587036132813e-05,
      "model_forward_time": 1.7407593727111816,
      "step": 12162
    },
    {
      "epoch": 1.1598587036132813e-05,
      "step": 12162,
      "training_step_time": 7.695228099822998
    },
    {
      "epoch": 1.1599540710449219e-05,
      "model_forward_time": 1.736189603805542,
      "step": 12163
    },
    {
      "epoch": 1.1599540710449219e-05,
      "step": 12163,
      "training_step_time": 6.8304102420806885
    },
    {
      "epoch": 1.1600494384765625e-05,
      "model_forward_time": 1.7272701263427734,
      "step": 12164
    },
    {
      "epoch": 1.1600494384765625e-05,
      "step": 12164,
      "training_step_time": 7.680827379226685
    },
    {
      "epoch": 1.160144805908203e-05,
      "model_forward_time": 1.739903211593628,
      "step": 12165
    },
    {
      "epoch": 1.160144805908203e-05,
      "step": 12165,
      "training_step_time": 8.413852214813232
    },
    {
      "epoch": 1.1602401733398438e-05,
      "model_forward_time": 1.753556251525879,
      "step": 12166
    },
    {
      "epoch": 1.1602401733398438e-05,
      "step": 12166,
      "training_step_time": 9.434827327728271
    },
    {
      "epoch": 1.1603355407714844e-05,
      "model_forward_time": 1.7458274364471436,
      "step": 12167
    },
    {
      "epoch": 1.1603355407714844e-05,
      "step": 12167,
      "training_step_time": 9.6513991355896
    },
    {
      "epoch": 1.160430908203125e-05,
      "model_forward_time": 1.727616310119629,
      "step": 12168
    },
    {
      "epoch": 1.160430908203125e-05,
      "step": 12168,
      "training_step_time": 8.652833461761475
    },
    {
      "epoch": 1.1605262756347657e-05,
      "model_forward_time": 2.5212504863739014,
      "step": 12169
    },
    {
      "epoch": 1.1605262756347657e-05,
      "step": 12169,
      "training_step_time": 7.708737134933472
    },
    {
      "epoch": 1.1606216430664062e-05,
      "grad_norm": 0.2570585608482361,
      "learning_rate": 9.860146078483837e-06,
      "loss": 0.0706,
      "step": 12170
    },
    {
      "epoch": 1.1606216430664062e-05,
      "model_forward_time": 1.726215124130249,
      "step": 12170
    },
    {
      "dynamics_loss_avg": 0.05585229620337486,
      "epoch": 1.1606216430664062e-05,
      "step": 12170
    },
    {
      "action_loss_avg": 0.0036129293264821173,
      "epoch": 1.1606216430664062e-05,
      "step": 12170
    },
    {
      "epoch": 1.1606216430664062e-05,
      "step": 12170,
      "training_step_time": 7.49890661239624
    },
    {
      "epoch": 1.1607170104980468e-05,
      "model_forward_time": 1.7352707386016846,
      "step": 12171
    },
    {
      "epoch": 1.1607170104980468e-05,
      "step": 12171,
      "training_step_time": 7.695919513702393
    },
    {
      "epoch": 1.1608123779296876e-05,
      "model_forward_time": 1.7439792156219482,
      "step": 12172
    },
    {
      "epoch": 1.1608123779296876e-05,
      "step": 12172,
      "training_step_time": 7.579148530960083
    },
    {
      "epoch": 1.1609077453613281e-05,
      "model_forward_time": 1.7299377918243408,
      "step": 12173
    },
    {
      "epoch": 1.1609077453613281e-05,
      "step": 12173,
      "training_step_time": 8.283710718154907
    },
    {
      "epoch": 1.1610031127929687e-05,
      "model_forward_time": 1.7401647567749023,
      "step": 12174
    },
    {
      "epoch": 1.1610031127929687e-05,
      "step": 12174,
      "training_step_time": 7.7337164878845215
    },
    {
      "epoch": 1.1610984802246094e-05,
      "model_forward_time": 1.7537775039672852,
      "step": 12175
    },
    {
      "epoch": 1.1610984802246094e-05,
      "step": 12175,
      "training_step_time": 9.12231731414795
    },
    {
      "epoch": 1.16119384765625e-05,
      "model_forward_time": 1.751457691192627,
      "step": 12176
    },
    {
      "epoch": 1.16119384765625e-05,
      "step": 12176,
      "training_step_time": 8.638619184494019
    },
    {
      "epoch": 1.1612892150878906e-05,
      "model_forward_time": 1.7433643341064453,
      "step": 12177
    },
    {
      "epoch": 1.1612892150878906e-05,
      "step": 12177,
      "training_step_time": 7.676920175552368
    },
    {
      "epoch": 1.1613845825195313e-05,
      "model_forward_time": 2.828794002532959,
      "step": 12178
    },
    {
      "epoch": 1.1613845825195313e-05,
      "step": 12178,
      "training_step_time": 8.678564310073853
    },
    {
      "epoch": 1.1614799499511719e-05,
      "model_forward_time": 1.7230160236358643,
      "step": 12179
    },
    {
      "epoch": 1.1614799499511719e-05,
      "step": 12179,
      "training_step_time": 7.464504957199097
    },
    {
      "epoch": 1.1615753173828125e-05,
      "grad_norm": 0.23618412017822266,
      "learning_rate": 9.859757478875905e-06,
      "loss": 0.0675,
      "step": 12180
    },
    {
      "epoch": 1.1615753173828125e-05,
      "model_forward_time": 1.734740972518921,
      "step": 12180
    },
    {
      "dynamics_loss_avg": 0.045495749823749064,
      "epoch": 1.1615753173828125e-05,
      "step": 12180
    },
    {
      "action_loss_avg": 0.0033721898886142298,
      "epoch": 1.1615753173828125e-05,
      "step": 12180
    },
    {
      "epoch": 1.1615753173828125e-05,
      "step": 12180,
      "training_step_time": 7.738391160964966
    },
    {
      "epoch": 1.161670684814453e-05,
      "model_forward_time": 1.7373802661895752,
      "step": 12181
    },
    {
      "epoch": 1.161670684814453e-05,
      "step": 12181,
      "training_step_time": 7.232371807098389
    },
    {
      "epoch": 1.1617660522460938e-05,
      "model_forward_time": 1.732417345046997,
      "step": 12182
    },
    {
      "epoch": 1.1617660522460938e-05,
      "step": 12182,
      "training_step_time": 7.7861902713775635
    },
    {
      "epoch": 1.1618614196777344e-05,
      "model_forward_time": 1.7430675029754639,
      "step": 12183
    },
    {
      "epoch": 1.1618614196777344e-05,
      "step": 12183,
      "training_step_time": 8.292460680007935
    },
    {
      "epoch": 1.161956787109375e-05,
      "model_forward_time": 1.712907314300537,
      "step": 12184
    },
    {
      "epoch": 1.161956787109375e-05,
      "step": 12184,
      "training_step_time": 8.773334503173828
    },
    {
      "epoch": 1.1620521545410157e-05,
      "model_forward_time": 1.7555758953094482,
      "step": 12185
    },
    {
      "epoch": 1.1620521545410157e-05,
      "step": 12185,
      "training_step_time": 7.745259523391724
    },
    {
      "epoch": 1.1621475219726562e-05,
      "model_forward_time": 1.7283198833465576,
      "step": 12186
    },
    {
      "epoch": 1.1621475219726562e-05,
      "step": 12186,
      "training_step_time": 8.624921798706055
    },
    {
      "epoch": 1.1622428894042968e-05,
      "model_forward_time": 1.7299561500549316,
      "step": 12187
    },
    {
      "epoch": 1.1622428894042968e-05,
      "step": 12187,
      "training_step_time": 7.6719489097595215
    },
    {
      "epoch": 1.1623382568359376e-05,
      "model_forward_time": 2.9506020545959473,
      "step": 12188
    },
    {
      "epoch": 1.1623382568359376e-05,
      "step": 12188,
      "training_step_time": 7.811983108520508
    },
    {
      "epoch": 1.1624336242675781e-05,
      "model_forward_time": 1.7334752082824707,
      "step": 12189
    },
    {
      "epoch": 1.1624336242675781e-05,
      "step": 12189,
      "training_step_time": 7.672170639038086
    },
    {
      "epoch": 1.1625289916992187e-05,
      "grad_norm": 0.26098230481147766,
      "learning_rate": 9.859368347812204e-06,
      "loss": 0.0698,
      "step": 12190
    },
    {
      "epoch": 1.1625289916992187e-05,
      "model_forward_time": 1.7139692306518555,
      "step": 12190
    },
    {
      "dynamics_loss_avg": 0.050414837524294855,
      "epoch": 1.1625289916992187e-05,
      "step": 12190
    },
    {
      "action_loss_avg": 0.0023199651041068137,
      "epoch": 1.1625289916992187e-05,
      "step": 12190
    },
    {
      "epoch": 1.1625289916992187e-05,
      "step": 12190,
      "training_step_time": 7.0917134284973145
    },
    {
      "epoch": 1.1626243591308594e-05,
      "model_forward_time": 1.7407214641571045,
      "step": 12191
    },
    {
      "epoch": 1.1626243591308594e-05,
      "step": 12191,
      "training_step_time": 7.793044090270996
    },
    {
      "epoch": 1.1627197265625e-05,
      "model_forward_time": 1.72975754737854,
      "step": 12192
    },
    {
      "epoch": 1.1627197265625e-05,
      "step": 12192,
      "training_step_time": 7.733027219772339
    },
    {
      "epoch": 1.1628150939941406e-05,
      "model_forward_time": 1.7240939140319824,
      "step": 12193
    },
    {
      "epoch": 1.1628150939941406e-05,
      "step": 12193,
      "training_step_time": 7.756703853607178
    },
    {
      "epoch": 1.1629104614257813e-05,
      "model_forward_time": 1.8356173038482666,
      "step": 12194
    },
    {
      "epoch": 1.1629104614257813e-05,
      "step": 12194,
      "training_step_time": 8.437389612197876
    },
    {
      "epoch": 1.1630058288574219e-05,
      "model_forward_time": 1.76664137840271,
      "step": 12195
    },
    {
      "epoch": 1.1630058288574219e-05,
      "step": 12195,
      "training_step_time": 7.748883247375488
    },
    {
      "epoch": 1.1631011962890625e-05,
      "model_forward_time": 1.736307144165039,
      "step": 12196
    },
    {
      "epoch": 1.1631011962890625e-05,
      "step": 12196,
      "training_step_time": 8.431374311447144
    },
    {
      "epoch": 1.163196563720703e-05,
      "model_forward_time": 2.980074405670166,
      "step": 12197
    },
    {
      "epoch": 1.163196563720703e-05,
      "step": 12197,
      "training_step_time": 7.850295543670654
    },
    {
      "epoch": 1.1632919311523438e-05,
      "model_forward_time": 1.73014497756958,
      "step": 12198
    },
    {
      "epoch": 1.1632919311523438e-05,
      "step": 12198,
      "training_step_time": 8.542855024337769
    },
    {
      "epoch": 1.1633872985839844e-05,
      "model_forward_time": 1.7511394023895264,
      "step": 12199
    },
    {
      "epoch": 1.1633872985839844e-05,
      "step": 12199,
      "training_step_time": 7.352608680725098
    },
    {
      "epoch": 1.163482666015625e-05,
      "grad_norm": 0.3839133679866791,
      "learning_rate": 9.858978685335286e-06,
      "loss": 0.0675,
      "step": 12200
    },
    {
      "epoch": 1.163482666015625e-05,
      "model_forward_time": 1.742443561553955,
      "step": 12200
    },
    {
      "dynamics_loss_avg": 0.08295487724244595,
      "epoch": 1.163482666015625e-05,
      "step": 12200
    },
    {
      "action_loss_avg": 0.0068314192933030425,
      "epoch": 1.163482666015625e-05,
      "step": 12200
    },
    {
      "epoch": 1.163482666015625e-05,
      "step": 12200,
      "training_step_time": 8.893917798995972
    },
    {
      "epoch": 1.1635780334472657e-05,
      "model_forward_time": 1.7412350177764893,
      "step": 12201
    },
    {
      "epoch": 1.1635780334472657e-05,
      "step": 12201,
      "training_step_time": 8.506549835205078
    },
    {
      "epoch": 1.1636734008789062e-05,
      "model_forward_time": 1.734381914138794,
      "step": 12202
    },
    {
      "epoch": 1.1636734008789062e-05,
      "step": 12202,
      "training_step_time": 9.075531721115112
    },
    {
      "epoch": 1.1637687683105468e-05,
      "model_forward_time": 1.767237663269043,
      "step": 12203
    },
    {
      "epoch": 1.1637687683105468e-05,
      "step": 12203,
      "training_step_time": 10.776612997055054
    },
    {
      "epoch": 1.1638641357421876e-05,
      "model_forward_time": 1.7375097274780273,
      "step": 12204
    },
    {
      "epoch": 1.1638641357421876e-05,
      "step": 12204,
      "training_step_time": 7.798006057739258
    },
    {
      "epoch": 1.1639595031738281e-05,
      "model_forward_time": 1.7467308044433594,
      "step": 12205
    },
    {
      "epoch": 1.1639595031738281e-05,
      "step": 12205,
      "training_step_time": 8.370731592178345
    },
    {
      "epoch": 1.1640548706054687e-05,
      "model_forward_time": 2.9132721424102783,
      "step": 12206
    },
    {
      "epoch": 1.1640548706054687e-05,
      "step": 12206,
      "training_step_time": 7.780186414718628
    },
    {
      "epoch": 1.1641502380371094e-05,
      "model_forward_time": 1.7351267337799072,
      "step": 12207
    },
    {
      "epoch": 1.1641502380371094e-05,
      "step": 12207,
      "training_step_time": 7.650211572647095
    },
    {
      "epoch": 1.16424560546875e-05,
      "model_forward_time": 1.7546002864837646,
      "step": 12208
    },
    {
      "epoch": 1.16424560546875e-05,
      "step": 12208,
      "training_step_time": 7.519710063934326
    },
    {
      "epoch": 1.1643409729003906e-05,
      "model_forward_time": 1.7381985187530518,
      "step": 12209
    },
    {
      "epoch": 1.1643409729003906e-05,
      "step": 12209,
      "training_step_time": 7.809422492980957
    },
    {
      "epoch": 1.1644363403320313e-05,
      "grad_norm": 0.29986321926116943,
      "learning_rate": 9.858588491487762e-06,
      "loss": 0.0691,
      "step": 12210
    },
    {
      "epoch": 1.1644363403320313e-05,
      "model_forward_time": 1.7369539737701416,
      "step": 12210
    },
    {
      "dynamics_loss_avg": 0.0770425971597433,
      "epoch": 1.1644363403320313e-05,
      "step": 12210
    },
    {
      "action_loss_avg": 0.006149967201054096,
      "epoch": 1.1644363403320313e-05,
      "step": 12210
    },
    {
      "epoch": 1.1644363403320313e-05,
      "step": 12210,
      "training_step_time": 7.761579990386963
    },
    {
      "epoch": 1.1645317077636719e-05,
      "model_forward_time": 1.7569892406463623,
      "step": 12211
    },
    {
      "epoch": 1.1645317077636719e-05,
      "step": 12211,
      "training_step_time": 8.258769035339355
    },
    {
      "epoch": 1.1646270751953125e-05,
      "model_forward_time": 1.7370994091033936,
      "step": 12212
    },
    {
      "epoch": 1.1646270751953125e-05,
      "step": 12212,
      "training_step_time": 9.674962282180786
    },
    {
      "epoch": 1.164722442626953e-05,
      "model_forward_time": 1.7438125610351562,
      "step": 12213
    },
    {
      "epoch": 1.164722442626953e-05,
      "step": 12213,
      "training_step_time": 8.325860738754272
    },
    {
      "epoch": 1.1648178100585938e-05,
      "model_forward_time": 1.7379419803619385,
      "step": 12214
    },
    {
      "epoch": 1.1648178100585938e-05,
      "step": 12214,
      "training_step_time": 7.740734338760376
    },
    {
      "epoch": 1.1649131774902344e-05,
      "model_forward_time": 2.959851026535034,
      "step": 12215
    },
    {
      "epoch": 1.1649131774902344e-05,
      "step": 12215,
      "training_step_time": 7.841289281845093
    },
    {
      "epoch": 1.165008544921875e-05,
      "model_forward_time": 1.7400038242340088,
      "step": 12216
    },
    {
      "epoch": 1.165008544921875e-05,
      "step": 12216,
      "training_step_time": 8.506032228469849
    },
    {
      "epoch": 1.1651039123535157e-05,
      "model_forward_time": 1.7438099384307861,
      "step": 12217
    },
    {
      "epoch": 1.1651039123535157e-05,
      "step": 12217,
      "training_step_time": 7.670536518096924
    },
    {
      "epoch": 1.1651992797851562e-05,
      "model_forward_time": 1.7529704570770264,
      "step": 12218
    },
    {
      "epoch": 1.1651992797851562e-05,
      "step": 12218,
      "training_step_time": 7.617228031158447
    },
    {
      "epoch": 1.1652946472167968e-05,
      "model_forward_time": 1.7402117252349854,
      "step": 12219
    },
    {
      "epoch": 1.1652946472167968e-05,
      "step": 12219,
      "training_step_time": 8.761430025100708
    },
    {
      "epoch": 1.1653900146484376e-05,
      "grad_norm": 0.20051038265228271,
      "learning_rate": 9.858197766312307e-06,
      "loss": 0.0652,
      "step": 12220
    },
    {
      "epoch": 1.1653900146484376e-05,
      "model_forward_time": 1.7698965072631836,
      "step": 12220
    },
    {
      "dynamics_loss_avg": 0.05965167563408613,
      "epoch": 1.1653900146484376e-05,
      "step": 12220
    },
    {
      "action_loss_avg": 0.004115787311457097,
      "epoch": 1.1653900146484376e-05,
      "step": 12220
    },
    {
      "epoch": 1.1653900146484376e-05,
      "step": 12220,
      "training_step_time": 8.525889158248901
    },
    {
      "epoch": 1.1654853820800781e-05,
      "model_forward_time": 1.7417564392089844,
      "step": 12221
    },
    {
      "epoch": 1.1654853820800781e-05,
      "step": 12221,
      "training_step_time": 9.539606809616089
    },
    {
      "epoch": 1.1655807495117187e-05,
      "model_forward_time": 1.8717784881591797,
      "step": 12222
    },
    {
      "epoch": 1.1655807495117187e-05,
      "step": 12222,
      "training_step_time": 8.768242120742798
    },
    {
      "epoch": 1.1656761169433594e-05,
      "model_forward_time": 1.742673397064209,
      "step": 12223
    },
    {
      "epoch": 1.1656761169433594e-05,
      "step": 12223,
      "training_step_time": 8.793539762496948
    },
    {
      "epoch": 1.165771484375e-05,
      "model_forward_time": 2.886634588241577,
      "step": 12224
    },
    {
      "epoch": 1.165771484375e-05,
      "step": 12224,
      "training_step_time": 8.734912395477295
    },
    {
      "epoch": 1.1658668518066406e-05,
      "model_forward_time": 1.741610050201416,
      "step": 12225
    },
    {
      "epoch": 1.1658668518066406e-05,
      "step": 12225,
      "training_step_time": 7.716987371444702
    },
    {
      "epoch": 1.1659622192382813e-05,
      "model_forward_time": 1.7216784954071045,
      "step": 12226
    },
    {
      "epoch": 1.1659622192382813e-05,
      "step": 12226,
      "training_step_time": 7.661701440811157
    },
    {
      "epoch": 1.1660575866699219e-05,
      "model_forward_time": 1.7404918670654297,
      "step": 12227
    },
    {
      "epoch": 1.1660575866699219e-05,
      "step": 12227,
      "training_step_time": 7.600331544876099
    },
    {
      "epoch": 1.1661529541015625e-05,
      "model_forward_time": 1.7293891906738281,
      "step": 12228
    },
    {
      "epoch": 1.1661529541015625e-05,
      "step": 12228,
      "training_step_time": 7.689978837966919
    },
    {
      "epoch": 1.166248321533203e-05,
      "model_forward_time": 1.7285101413726807,
      "step": 12229
    },
    {
      "epoch": 1.166248321533203e-05,
      "step": 12229,
      "training_step_time": 7.890648365020752
    },
    {
      "epoch": 1.1663436889648438e-05,
      "grad_norm": 0.1969166398048401,
      "learning_rate": 9.857806509851649e-06,
      "loss": 0.0619,
      "step": 12230
    },
    {
      "epoch": 1.1663436889648438e-05,
      "model_forward_time": 1.732673168182373,
      "step": 12230
    },
    {
      "dynamics_loss_avg": 0.05403746590018273,
      "epoch": 1.1663436889648438e-05,
      "step": 12230
    },
    {
      "action_loss_avg": 0.0029462238831911235,
      "epoch": 1.1663436889648438e-05,
      "step": 12230
    },
    {
      "epoch": 1.1663436889648438e-05,
      "step": 12230,
      "training_step_time": 8.824418783187866
    },
    {
      "epoch": 1.1664390563964844e-05,
      "model_forward_time": 1.7891976833343506,
      "step": 12231
    },
    {
      "epoch": 1.1664390563964844e-05,
      "step": 12231,
      "training_step_time": 8.98994517326355
    },
    {
      "epoch": 1.166534423828125e-05,
      "model_forward_time": 1.7309885025024414,
      "step": 12232
    },
    {
      "epoch": 1.166534423828125e-05,
      "step": 12232,
      "training_step_time": 7.836172819137573
    },
    {
      "epoch": 1.1666297912597657e-05,
      "model_forward_time": 1.7320342063903809,
      "step": 12233
    },
    {
      "epoch": 1.1666297912597657e-05,
      "step": 12233,
      "training_step_time": 9.786755084991455
    },
    {
      "epoch": 1.1667251586914062e-05,
      "model_forward_time": 1.733064889907837,
      "step": 12234
    },
    {
      "epoch": 1.1667251586914062e-05,
      "step": 12234,
      "training_step_time": 8.573937892913818
    },
    {
      "epoch": 1.1668205261230468e-05,
      "model_forward_time": 1.7261006832122803,
      "step": 12235
    },
    {
      "epoch": 1.1668205261230468e-05,
      "step": 12235,
      "training_step_time": 7.766258955001831
    },
    {
      "epoch": 1.1669158935546876e-05,
      "model_forward_time": 1.7422385215759277,
      "step": 12236
    },
    {
      "epoch": 1.1669158935546876e-05,
      "step": 12236,
      "training_step_time": 7.739590167999268
    },
    {
      "epoch": 1.1670112609863281e-05,
      "model_forward_time": 1.7290327548980713,
      "step": 12237
    },
    {
      "epoch": 1.1670112609863281e-05,
      "step": 12237,
      "training_step_time": 7.797107219696045
    },
    {
      "epoch": 1.1671066284179687e-05,
      "model_forward_time": 1.724816083908081,
      "step": 12238
    },
    {
      "epoch": 1.1671066284179687e-05,
      "step": 12238,
      "training_step_time": 7.9431211948394775
    },
    {
      "epoch": 1.1672019958496094e-05,
      "model_forward_time": 1.7336242198944092,
      "step": 12239
    },
    {
      "epoch": 1.1672019958496094e-05,
      "step": 12239,
      "training_step_time": 7.818317651748657
    },
    {
      "epoch": 1.16729736328125e-05,
      "grad_norm": 0.2274186611175537,
      "learning_rate": 9.857414722148574e-06,
      "loss": 0.065,
      "step": 12240
    },
    {
      "epoch": 1.16729736328125e-05,
      "model_forward_time": 1.7504949569702148,
      "step": 12240
    },
    {
      "dynamics_loss_avg": 0.053299876302480696,
      "epoch": 1.16729736328125e-05,
      "step": 12240
    },
    {
      "action_loss_avg": 0.004618905333336443,
      "epoch": 1.16729736328125e-05,
      "step": 12240
    },
    {
      "epoch": 1.16729736328125e-05,
      "step": 12240,
      "training_step_time": 8.776097297668457
    },
    {
      "epoch": 1.1673927307128906e-05,
      "model_forward_time": 1.73158860206604,
      "step": 12241
    },
    {
      "epoch": 1.1673927307128906e-05,
      "step": 12241,
      "training_step_time": 8.633411884307861
    },
    {
      "epoch": 1.1674880981445313e-05,
      "model_forward_time": 1.7371349334716797,
      "step": 12242
    },
    {
      "epoch": 1.1674880981445313e-05,
      "step": 12242,
      "training_step_time": 8.35641860961914
    },
    {
      "epoch": 1.1675834655761719e-05,
      "model_forward_time": 2.877302885055542,
      "step": 12243
    },
    {
      "epoch": 1.1675834655761719e-05,
      "step": 12243,
      "training_step_time": 7.673110485076904
    },
    {
      "epoch": 1.1676788330078125e-05,
      "model_forward_time": 1.730665922164917,
      "step": 12244
    },
    {
      "epoch": 1.1676788330078125e-05,
      "step": 12244,
      "training_step_time": 7.660943269729614
    },
    {
      "epoch": 1.167774200439453e-05,
      "model_forward_time": 1.7363643646240234,
      "step": 12245
    },
    {
      "epoch": 1.167774200439453e-05,
      "step": 12245,
      "training_step_time": 7.669001340866089
    },
    {
      "epoch": 1.1678695678710938e-05,
      "model_forward_time": 1.7354512214660645,
      "step": 12246
    },
    {
      "epoch": 1.1678695678710938e-05,
      "step": 12246,
      "training_step_time": 7.599299192428589
    },
    {
      "epoch": 1.1679649353027344e-05,
      "model_forward_time": 1.7253432273864746,
      "step": 12247
    },
    {
      "epoch": 1.1679649353027344e-05,
      "step": 12247,
      "training_step_time": 7.822726011276245
    },
    {
      "epoch": 1.168060302734375e-05,
      "model_forward_time": 1.7306227684020996,
      "step": 12248
    },
    {
      "epoch": 1.168060302734375e-05,
      "step": 12248,
      "training_step_time": 8.757857084274292
    },
    {
      "epoch": 1.1681556701660157e-05,
      "model_forward_time": 1.7270479202270508,
      "step": 12249
    },
    {
      "epoch": 1.1681556701660157e-05,
      "step": 12249,
      "training_step_time": 10.169906854629517
    },
    {
      "epoch": 1.1682510375976562e-05,
      "grad_norm": 0.23098430037498474,
      "learning_rate": 9.857022403245929e-06,
      "loss": 0.0636,
      "step": 12250
    },
    {
      "epoch": 1.1682510375976562e-05,
      "model_forward_time": 1.8140935897827148,
      "step": 12250
    },
    {
      "dynamics_loss_avg": 0.06012695850804448,
      "epoch": 1.1682510375976562e-05,
      "step": 12250
    },
    {
      "action_loss_avg": 0.0041174014681018885,
      "epoch": 1.1682510375976562e-05,
      "step": 12250
    },
    {
      "epoch": 1.1682510375976562e-05,
      "step": 12250,
      "training_step_time": 8.577991008758545
    },
    {
      "epoch": 1.1683464050292968e-05,
      "model_forward_time": 1.7260925769805908,
      "step": 12251
    },
    {
      "epoch": 1.1683464050292968e-05,
      "step": 12251,
      "training_step_time": 8.310466051101685
    },
    {
      "epoch": 1.1684417724609376e-05,
      "model_forward_time": 2.8706321716308594,
      "step": 12252
    },
    {
      "epoch": 1.1684417724609376e-05,
      "step": 12252,
      "training_step_time": 7.683041334152222
    },
    {
      "epoch": 1.1685371398925781e-05,
      "model_forward_time": 1.7242934703826904,
      "step": 12253
    },
    {
      "epoch": 1.1685371398925781e-05,
      "step": 12253,
      "training_step_time": 7.627172470092773
    },
    {
      "epoch": 1.1686325073242187e-05,
      "model_forward_time": 1.3489482402801514,
      "step": 12254
    },
    {
      "epoch": 1.1686325073242187e-05,
      "step": 12254,
      "training_step_time": 7.577990293502808
    },
    {
      "epoch": 1.1687278747558594e-05,
      "model_forward_time": 1.7374193668365479,
      "step": 12255
    },
    {
      "epoch": 1.1687278747558594e-05,
      "step": 12255,
      "training_step_time": 7.429524183273315
    },
    {
      "epoch": 1.1688232421875e-05,
      "model_forward_time": 1.7210500240325928,
      "step": 12256
    },
    {
      "epoch": 1.1688232421875e-05,
      "step": 12256,
      "training_step_time": 7.763608455657959
    },
    {
      "epoch": 1.1689186096191406e-05,
      "model_forward_time": 1.7388930320739746,
      "step": 12257
    },
    {
      "epoch": 1.1689186096191406e-05,
      "step": 12257,
      "training_step_time": 8.34639859199524
    },
    {
      "epoch": 1.1690139770507812e-05,
      "model_forward_time": 1.7313191890716553,
      "step": 12258
    },
    {
      "epoch": 1.1690139770507812e-05,
      "step": 12258,
      "training_step_time": 8.285287857055664
    },
    {
      "epoch": 1.1691093444824219e-05,
      "model_forward_time": 1.7974488735198975,
      "step": 12259
    },
    {
      "epoch": 1.1691093444824219e-05,
      "step": 12259,
      "training_step_time": 8.737772941589355
    },
    {
      "epoch": 1.1692047119140625e-05,
      "grad_norm": 0.22448566555976868,
      "learning_rate": 9.856629553186615e-06,
      "loss": 0.0633,
      "step": 12260
    },
    {
      "epoch": 1.1692047119140625e-05,
      "model_forward_time": 1.7315700054168701,
      "step": 12260
    },
    {
      "dynamics_loss_avg": 0.054968990944325925,
      "epoch": 1.1692047119140625e-05,
      "step": 12260
    },
    {
      "action_loss_avg": 0.0037051981198601425,
      "epoch": 1.1692047119140625e-05,
      "step": 12260
    },
    {
      "epoch": 1.1692047119140625e-05,
      "step": 12260,
      "training_step_time": 9.632710933685303
    },
    {
      "epoch": 1.169300079345703e-05,
      "model_forward_time": 2.862283706665039,
      "step": 12261
    },
    {
      "epoch": 1.169300079345703e-05,
      "step": 12261,
      "training_step_time": 8.16991138458252
    },
    {
      "epoch": 1.1693954467773438e-05,
      "model_forward_time": 1.7322359085083008,
      "step": 12262
    },
    {
      "epoch": 1.1693954467773438e-05,
      "step": 12262,
      "training_step_time": 7.563887596130371
    },
    {
      "epoch": 1.1694908142089844e-05,
      "model_forward_time": 1.729149580001831,
      "step": 12263
    },
    {
      "epoch": 1.1694908142089844e-05,
      "step": 12263,
      "training_step_time": 7.5411059856414795
    },
    {
      "epoch": 1.169586181640625e-05,
      "model_forward_time": 1.7387006282806396,
      "step": 12264
    },
    {
      "epoch": 1.169586181640625e-05,
      "step": 12264,
      "training_step_time": 7.581967830657959
    },
    {
      "epoch": 1.1696815490722657e-05,
      "model_forward_time": 1.7355563640594482,
      "step": 12265
    },
    {
      "epoch": 1.1696815490722657e-05,
      "step": 12265,
      "training_step_time": 8.30329155921936
    },
    {
      "epoch": 1.1697769165039063e-05,
      "model_forward_time": 1.733159065246582,
      "step": 12266
    },
    {
      "epoch": 1.1697769165039063e-05,
      "step": 12266,
      "training_step_time": 7.801275014877319
    },
    {
      "epoch": 1.1698722839355468e-05,
      "model_forward_time": 1.725210428237915,
      "step": 12267
    },
    {
      "epoch": 1.1698722839355468e-05,
      "step": 12267,
      "training_step_time": 7.642451763153076
    },
    {
      "epoch": 1.1699676513671876e-05,
      "model_forward_time": 1.777273416519165,
      "step": 12268
    },
    {
      "epoch": 1.1699676513671876e-05,
      "step": 12268,
      "training_step_time": 7.737850666046143
    },
    {
      "epoch": 1.1700630187988281e-05,
      "model_forward_time": 1.72697114944458,
      "step": 12269
    },
    {
      "epoch": 1.1700630187988281e-05,
      "step": 12269,
      "training_step_time": 8.515286684036255
    },
    {
      "epoch": 1.1701583862304687e-05,
      "grad_norm": 0.16865268349647522,
      "learning_rate": 9.856236172013595e-06,
      "loss": 0.0643,
      "step": 12270
    },
    {
      "epoch": 1.1701583862304687e-05,
      "model_forward_time": 2.826338529586792,
      "step": 12270
    },
    {
      "dynamics_loss_avg": 0.0459062933921814,
      "epoch": 1.1701583862304687e-05,
      "step": 12270
    },
    {
      "action_loss_avg": 0.002394128078594804,
      "epoch": 1.1701583862304687e-05,
      "step": 12270
    },
    {
      "epoch": 1.1701583862304687e-05,
      "step": 12270,
      "training_step_time": 7.612717151641846
    },
    {
      "epoch": 1.1702537536621095e-05,
      "model_forward_time": 1.7251334190368652,
      "step": 12271
    },
    {
      "epoch": 1.1702537536621095e-05,
      "step": 12271,
      "training_step_time": 8.6767897605896
    },
    {
      "epoch": 1.17034912109375e-05,
      "model_forward_time": 1.720728874206543,
      "step": 12272
    },
    {
      "epoch": 1.17034912109375e-05,
      "step": 12272,
      "training_step_time": 7.654388666152954
    },
    {
      "epoch": 1.1704444885253906e-05,
      "model_forward_time": 1.724820852279663,
      "step": 12273
    },
    {
      "epoch": 1.1704444885253906e-05,
      "step": 12273,
      "training_step_time": 7.550223112106323
    },
    {
      "epoch": 1.1705398559570312e-05,
      "model_forward_time": 1.002547264099121,
      "step": 12274
    },
    {
      "epoch": 1.1705398559570312e-05,
      "step": 12274,
      "training_step_time": 8.629181146621704
    },
    {
      "epoch": 1.1706352233886719e-05,
      "model_forward_time": 1.727491855621338,
      "step": 12275
    },
    {
      "epoch": 1.1706352233886719e-05,
      "step": 12275,
      "training_step_time": 7.892366409301758
    },
    {
      "epoch": 1.1707305908203125e-05,
      "model_forward_time": 1.7335102558135986,
      "step": 12276
    },
    {
      "epoch": 1.1707305908203125e-05,
      "step": 12276,
      "training_step_time": 7.7352135181427
    },
    {
      "epoch": 1.170825958251953e-05,
      "model_forward_time": 1.7376787662506104,
      "step": 12277
    },
    {
      "epoch": 1.170825958251953e-05,
      "step": 12277,
      "training_step_time": 10.338876485824585
    },
    {
      "epoch": 1.1709213256835938e-05,
      "model_forward_time": 1.7233211994171143,
      "step": 12278
    },
    {
      "epoch": 1.1709213256835938e-05,
      "step": 12278,
      "training_step_time": 8.811686754226685
    },
    {
      "epoch": 1.1710166931152344e-05,
      "model_forward_time": 1.738036870956421,
      "step": 12279
    },
    {
      "epoch": 1.1710166931152344e-05,
      "step": 12279,
      "training_step_time": 7.574481010437012
    },
    {
      "epoch": 1.171112060546875e-05,
      "grad_norm": 0.22783862054347992,
      "learning_rate": 9.85584225976989e-06,
      "loss": 0.0717,
      "step": 12280
    },
    {
      "epoch": 1.171112060546875e-05,
      "model_forward_time": 2.9187467098236084,
      "step": 12280
    },
    {
      "dynamics_loss_avg": 0.04986190507188439,
      "epoch": 1.171112060546875e-05,
      "step": 12280
    },
    {
      "action_loss_avg": 0.002744963380973786,
      "epoch": 1.171112060546875e-05,
      "step": 12280
    },
    {
      "epoch": 1.171112060546875e-05,
      "step": 12280,
      "training_step_time": 7.768448829650879
    },
    {
      "epoch": 1.1712074279785157e-05,
      "model_forward_time": 1.7242774963378906,
      "step": 12281
    },
    {
      "epoch": 1.1712074279785157e-05,
      "step": 12281,
      "training_step_time": 7.633121490478516
    },
    {
      "epoch": 1.1713027954101563e-05,
      "model_forward_time": 1.7302823066711426,
      "step": 12282
    },
    {
      "epoch": 1.1713027954101563e-05,
      "step": 12282,
      "training_step_time": 7.44444727897644
    },
    {
      "epoch": 1.1713981628417968e-05,
      "model_forward_time": 1.7252357006072998,
      "step": 12283
    },
    {
      "epoch": 1.1713981628417968e-05,
      "step": 12283,
      "training_step_time": 7.643484354019165
    },
    {
      "epoch": 1.1714935302734376e-05,
      "model_forward_time": 1.7293751239776611,
      "step": 12284
    },
    {
      "epoch": 1.1714935302734376e-05,
      "step": 12284,
      "training_step_time": 7.881676435470581
    },
    {
      "epoch": 1.1715888977050781e-05,
      "model_forward_time": 1.7318079471588135,
      "step": 12285
    },
    {
      "epoch": 1.1715888977050781e-05,
      "step": 12285,
      "training_step_time": 7.738020658493042
    },
    {
      "epoch": 1.1716842651367187e-05,
      "model_forward_time": 1.7283124923706055,
      "step": 12286
    },
    {
      "epoch": 1.1716842651367187e-05,
      "step": 12286,
      "training_step_time": 8.883346796035767
    },
    {
      "epoch": 1.1717796325683595e-05,
      "model_forward_time": 1.8290598392486572,
      "step": 12287
    },
    {
      "epoch": 1.1717796325683595e-05,
      "step": 12287,
      "training_step_time": 7.816873550415039
    },
    {
      "epoch": 1.171875e-05,
      "model_forward_time": 1.7370710372924805,
      "step": 12288
    },
    {
      "epoch": 1.171875e-05,
      "step": 12288,
      "training_step_time": 7.684219837188721
    },
    {
      "epoch": 1.1719703674316406e-05,
      "model_forward_time": 2.9254536628723145,
      "step": 12289
    },
    {
      "epoch": 1.1719703674316406e-05,
      "step": 12289,
      "training_step_time": 7.756174564361572
    },
    {
      "epoch": 1.1720657348632812e-05,
      "grad_norm": 0.23132944107055664,
      "learning_rate": 9.855447816498576e-06,
      "loss": 0.0725,
      "step": 12290
    },
    {
      "epoch": 1.1720657348632812e-05,
      "model_forward_time": 1.7369780540466309,
      "step": 12290
    },
    {
      "dynamics_loss_avg": 0.05592430280521512,
      "epoch": 1.1720657348632812e-05,
      "step": 12290
    },
    {
      "action_loss_avg": 0.003304945433046669,
      "epoch": 1.1720657348632812e-05,
      "step": 12290
    },
    {
      "epoch": 1.1720657348632812e-05,
      "step": 12290,
      "training_step_time": 7.763057231903076
    },
    {
      "epoch": 1.1721611022949219e-05,
      "model_forward_time": 1.740149974822998,
      "step": 12291
    },
    {
      "epoch": 1.1721611022949219e-05,
      "step": 12291,
      "training_step_time": 7.527104139328003
    },
    {
      "epoch": 1.1722564697265625e-05,
      "model_forward_time": 1.744727373123169,
      "step": 12292
    },
    {
      "epoch": 1.1722564697265625e-05,
      "step": 12292,
      "training_step_time": 7.667285442352295
    },
    {
      "epoch": 1.172351837158203e-05,
      "model_forward_time": 1.7428538799285889,
      "step": 12293
    },
    {
      "epoch": 1.172351837158203e-05,
      "step": 12293,
      "training_step_time": 9.732635736465454
    },
    {
      "epoch": 1.1724472045898438e-05,
      "model_forward_time": 1.7406344413757324,
      "step": 12294
    },
    {
      "epoch": 1.1724472045898438e-05,
      "step": 12294,
      "training_step_time": 8.653769254684448
    },
    {
      "epoch": 1.1725425720214844e-05,
      "model_forward_time": 1.7195370197296143,
      "step": 12295
    },
    {
      "epoch": 1.1725425720214844e-05,
      "step": 12295,
      "training_step_time": 8.567713022232056
    },
    {
      "epoch": 1.172637939453125e-05,
      "model_forward_time": 1.7972519397735596,
      "step": 12296
    },
    {
      "epoch": 1.172637939453125e-05,
      "step": 12296,
      "training_step_time": 9.260572671890259
    },
    {
      "epoch": 1.1727333068847657e-05,
      "model_forward_time": 1.7299449443817139,
      "step": 12297
    },
    {
      "epoch": 1.1727333068847657e-05,
      "step": 12297,
      "training_step_time": 7.649720191955566
    },
    {
      "epoch": 1.1728286743164063e-05,
      "model_forward_time": 2.8569893836975098,
      "step": 12298
    },
    {
      "epoch": 1.1728286743164063e-05,
      "step": 12298,
      "training_step_time": 7.6826372146606445
    },
    {
      "epoch": 1.1729240417480468e-05,
      "model_forward_time": 1.7359886169433594,
      "step": 12299
    },
    {
      "epoch": 1.1729240417480468e-05,
      "step": 12299,
      "training_step_time": 7.636819124221802
    },
    {
      "epoch": 1.1730194091796876e-05,
      "grad_norm": 0.1760442703962326,
      "learning_rate": 9.855052842242786e-06,
      "loss": 0.0733,
      "step": 12300
    },
    {
      "epoch": 1.1730194091796876e-05,
      "model_forward_time": 1.7445130348205566,
      "step": 12300
    },
    {
      "dynamics_loss_avg": 0.05874124187976122,
      "epoch": 1.1730194091796876e-05,
      "step": 12300
    },
    {
      "action_loss_avg": 0.003455955279059708,
      "epoch": 1.1730194091796876e-05,
      "step": 12300
    },
    {
      "epoch": 1.1730194091796876e-05,
      "step": 12300,
      "training_step_time": 8.523664474487305
    },
    {
      "epoch": 1.1731147766113281e-05,
      "model_forward_time": 1.725954294204712,
      "step": 12301
    },
    {
      "epoch": 1.1731147766113281e-05,
      "step": 12301,
      "training_step_time": 8.922054052352905
    },
    {
      "epoch": 1.1732101440429687e-05,
      "model_forward_time": 1.8217668533325195,
      "step": 12302
    },
    {
      "epoch": 1.1732101440429687e-05,
      "step": 12302,
      "training_step_time": 6.780062198638916
    },
    {
      "epoch": 1.1733055114746095e-05,
      "model_forward_time": 1.7431130409240723,
      "step": 12303
    },
    {
      "epoch": 1.1733055114746095e-05,
      "step": 12303,
      "training_step_time": 6.686289310455322
    },
    {
      "epoch": 1.17340087890625e-05,
      "model_forward_time": 1.3457038402557373,
      "step": 12304
    },
    {
      "epoch": 1.17340087890625e-05,
      "step": 12304,
      "training_step_time": 6.730503082275391
    },
    {
      "epoch": 1.1734962463378906e-05,
      "model_forward_time": 1.7320277690887451,
      "step": 12305
    },
    {
      "epoch": 1.1734962463378906e-05,
      "step": 12305,
      "training_step_time": 6.833911895751953
    },
    {
      "epoch": 1.1735916137695312e-05,
      "model_forward_time": 1.7336828708648682,
      "step": 12306
    },
    {
      "epoch": 1.1735916137695312e-05,
      "step": 12306,
      "training_step_time": 6.739897012710571
    },
    {
      "epoch": 1.1736869812011719e-05,
      "model_forward_time": 1.740856409072876,
      "step": 12307
    },
    {
      "epoch": 1.1736869812011719e-05,
      "step": 12307,
      "training_step_time": 6.7940757274627686
    },
    {
      "epoch": 1.1737823486328125e-05,
      "model_forward_time": 1.7455687522888184,
      "step": 12308
    },
    {
      "epoch": 1.1737823486328125e-05,
      "step": 12308,
      "training_step_time": 6.748131036758423
    },
    {
      "epoch": 1.173877716064453e-05,
      "model_forward_time": 1.7386951446533203,
      "step": 12309
    },
    {
      "epoch": 1.173877716064453e-05,
      "step": 12309,
      "training_step_time": 6.761981248855591
    },
    {
      "epoch": 1.1739730834960938e-05,
      "grad_norm": 0.21286986768245697,
      "learning_rate": 9.85465733704572e-06,
      "loss": 0.0748,
      "step": 12310
    },
    {
      "epoch": 1.1739730834960938e-05,
      "model_forward_time": 2.9159090518951416,
      "step": 12310
    },
    {
      "dynamics_loss_avg": 0.08584557119756937,
      "epoch": 1.1739730834960938e-05,
      "step": 12310
    },
    {
      "action_loss_avg": 0.005323635134845972,
      "epoch": 1.1739730834960938e-05,
      "step": 12310
    },
    {
      "epoch": 1.1739730834960938e-05,
      "step": 12310,
      "training_step_time": 7.724647760391235
    },
    {
      "epoch": 1.1740684509277344e-05,
      "model_forward_time": 1.7344825267791748,
      "step": 12311
    },
    {
      "epoch": 1.1740684509277344e-05,
      "step": 12311,
      "training_step_time": 7.822236776351929
    },
    {
      "epoch": 1.174163818359375e-05,
      "model_forward_time": 1.74385666847229,
      "step": 12312
    },
    {
      "epoch": 1.174163818359375e-05,
      "step": 12312,
      "training_step_time": 6.788624286651611
    },
    {
      "epoch": 1.1742591857910157e-05,
      "model_forward_time": 1.7374744415283203,
      "step": 12313
    },
    {
      "epoch": 1.1742591857910157e-05,
      "step": 12313,
      "training_step_time": 6.8002095222473145
    },
    {
      "epoch": 1.1743545532226563e-05,
      "model_forward_time": 1.7302136421203613,
      "step": 12314
    },
    {
      "epoch": 1.1743545532226563e-05,
      "step": 12314,
      "training_step_time": 6.752985715866089
    },
    {
      "epoch": 1.1744499206542968e-05,
      "model_forward_time": 1.7404122352600098,
      "step": 12315
    },
    {
      "epoch": 1.1744499206542968e-05,
      "step": 12315,
      "training_step_time": 6.763322114944458
    },
    {
      "epoch": 1.1745452880859376e-05,
      "model_forward_time": 1.7400341033935547,
      "step": 12316
    },
    {
      "epoch": 1.1745452880859376e-05,
      "step": 12316,
      "training_step_time": 6.782778739929199
    },
    {
      "epoch": 1.1746406555175781e-05,
      "model_forward_time": 1.7187306880950928,
      "step": 12317
    },
    {
      "epoch": 1.1746406555175781e-05,
      "step": 12317,
      "training_step_time": 6.782152414321899
    },
    {
      "epoch": 1.1747360229492187e-05,
      "model_forward_time": 1.7450661659240723,
      "step": 12318
    },
    {
      "epoch": 1.1747360229492187e-05,
      "step": 12318,
      "training_step_time": 6.7912445068359375
    },
    {
      "epoch": 1.1748313903808595e-05,
      "model_forward_time": 2.9469332695007324,
      "step": 12319
    },
    {
      "epoch": 1.1748313903808595e-05,
      "step": 12319,
      "training_step_time": 7.755553722381592
    },
    {
      "epoch": 1.1749267578125e-05,
      "grad_norm": 0.18980473279953003,
      "learning_rate": 9.854261300950625e-06,
      "loss": 0.0691,
      "step": 12320
    },
    {
      "epoch": 1.1749267578125e-05,
      "model_forward_time": 1.7396714687347412,
      "step": 12320
    },
    {
      "dynamics_loss_avg": 0.0764796132221818,
      "epoch": 1.1749267578125e-05,
      "step": 12320
    },
    {
      "action_loss_avg": 0.004562771628843621,
      "epoch": 1.1749267578125e-05,
      "step": 12320
    },
    {
      "epoch": 1.1749267578125e-05,
      "step": 12320,
      "training_step_time": 7.888977289199829
    },
    {
      "epoch": 1.1750221252441406e-05,
      "model_forward_time": 1.7347931861877441,
      "step": 12321
    },
    {
      "epoch": 1.1750221252441406e-05,
      "step": 12321,
      "training_step_time": 7.6360180377960205
    },
    {
      "epoch": 1.1751174926757812e-05,
      "model_forward_time": 1.7329590320587158,
      "step": 12322
    },
    {
      "epoch": 1.1751174926757812e-05,
      "step": 12322,
      "training_step_time": 6.744338035583496
    },
    {
      "epoch": 1.1752128601074219e-05,
      "model_forward_time": 1.7327334880828857,
      "step": 12323
    },
    {
      "epoch": 1.1752128601074219e-05,
      "step": 12323,
      "training_step_time": 6.7237279415130615
    },
    {
      "epoch": 1.1753082275390625e-05,
      "model_forward_time": 1.7283856868743896,
      "step": 12324
    },
    {
      "epoch": 1.1753082275390625e-05,
      "step": 12324,
      "training_step_time": 6.802032709121704
    },
    {
      "epoch": 1.175403594970703e-05,
      "model_forward_time": 1.7365410327911377,
      "step": 12325
    },
    {
      "epoch": 1.175403594970703e-05,
      "step": 12325,
      "training_step_time": 6.748411655426025
    },
    {
      "epoch": 1.1754989624023438e-05,
      "model_forward_time": 1.360020399093628,
      "step": 12326
    },
    {
      "epoch": 1.1754989624023438e-05,
      "step": 12326,
      "training_step_time": 6.779306411743164
    },
    {
      "epoch": 1.1755943298339844e-05,
      "model_forward_time": 1.7451727390289307,
      "step": 12327
    },
    {
      "epoch": 1.1755943298339844e-05,
      "step": 12327,
      "training_step_time": 6.834653377532959
    },
    {
      "epoch": 1.175689697265625e-05,
      "model_forward_time": 2.869386672973633,
      "step": 12328
    },
    {
      "epoch": 1.175689697265625e-05,
      "step": 12328,
      "training_step_time": 8.72553014755249
    },
    {
      "epoch": 1.1757850646972657e-05,
      "model_forward_time": 1.7655668258666992,
      "step": 12329
    },
    {
      "epoch": 1.1757850646972657e-05,
      "step": 12329,
      "training_step_time": 8.611244440078735
    },
    {
      "epoch": 1.1758804321289063e-05,
      "grad_norm": 0.11918112635612488,
      "learning_rate": 9.853864734000812e-06,
      "loss": 0.0696,
      "step": 12330
    },
    {
      "epoch": 1.1758804321289063e-05,
      "model_forward_time": 1.3639006614685059,
      "step": 12330
    },
    {
      "dynamics_loss_avg": 0.052189233712852,
      "epoch": 1.1758804321289063e-05,
      "step": 12330
    },
    {
      "action_loss_avg": 0.003413202380761504,
      "epoch": 1.1758804321289063e-05,
      "step": 12330
    },
    {
      "epoch": 1.1758804321289063e-05,
      "step": 12330,
      "training_step_time": 7.684819459915161
    },
    {
      "epoch": 1.1759757995605468e-05,
      "model_forward_time": 1.7367792129516602,
      "step": 12331
    },
    {
      "epoch": 1.1759757995605468e-05,
      "step": 12331,
      "training_step_time": 6.728107929229736
    },
    {
      "epoch": 1.1760711669921876e-05,
      "model_forward_time": 1.7390034198760986,
      "step": 12332
    },
    {
      "epoch": 1.1760711669921876e-05,
      "step": 12332,
      "training_step_time": 6.780886650085449
    },
    {
      "epoch": 1.1761665344238281e-05,
      "model_forward_time": 1.7356271743774414,
      "step": 12333
    },
    {
      "epoch": 1.1761665344238281e-05,
      "step": 12333,
      "training_step_time": 6.720261812210083
    },
    {
      "epoch": 1.1762619018554687e-05,
      "model_forward_time": 1.7399492263793945,
      "step": 12334
    },
    {
      "epoch": 1.1762619018554687e-05,
      "step": 12334,
      "training_step_time": 6.70080304145813
    },
    {
      "epoch": 1.1763572692871095e-05,
      "model_forward_time": 1.7430646419525146,
      "step": 12335
    },
    {
      "epoch": 1.1763572692871095e-05,
      "step": 12335,
      "training_step_time": 6.738877058029175
    },
    {
      "epoch": 1.17645263671875e-05,
      "model_forward_time": 1.0221929550170898,
      "step": 12336
    },
    {
      "epoch": 1.17645263671875e-05,
      "step": 12336,
      "training_step_time": 6.705601453781128
    },
    {
      "epoch": 1.1765480041503906e-05,
      "model_forward_time": 1.7264204025268555,
      "step": 12337
    },
    {
      "epoch": 1.1765480041503906e-05,
      "step": 12337,
      "training_step_time": 7.299245357513428
    },
    {
      "epoch": 1.1766433715820312e-05,
      "model_forward_time": 2.917604923248291,
      "step": 12338
    },
    {
      "epoch": 1.1766433715820312e-05,
      "step": 12338,
      "training_step_time": 9.353787660598755
    },
    {
      "epoch": 1.1767387390136719e-05,
      "model_forward_time": 1.7392029762268066,
      "step": 12339
    },
    {
      "epoch": 1.1767387390136719e-05,
      "step": 12339,
      "training_step_time": 7.678644418716431
    },
    {
      "epoch": 1.1768341064453125e-05,
      "grad_norm": 0.1930042952299118,
      "learning_rate": 9.853467636239652e-06,
      "loss": 0.0693,
      "step": 12340
    },
    {
      "epoch": 1.1768341064453125e-05,
      "model_forward_time": 1.7321388721466064,
      "step": 12340
    },
    {
      "dynamics_loss_avg": 0.06946263778954745,
      "epoch": 1.1768341064453125e-05,
      "step": 12340
    },
    {
      "action_loss_avg": 0.004519623710075393,
      "epoch": 1.1768341064453125e-05,
      "step": 12340
    },
    {
      "epoch": 1.1768341064453125e-05,
      "step": 12340,
      "training_step_time": 6.754028558731079
    },
    {
      "epoch": 1.176929473876953e-05,
      "model_forward_time": 1.7329092025756836,
      "step": 12341
    },
    {
      "epoch": 1.176929473876953e-05,
      "step": 12341,
      "training_step_time": 6.762102365493774
    },
    {
      "epoch": 1.1770248413085938e-05,
      "model_forward_time": 1.7480385303497314,
      "step": 12342
    },
    {
      "epoch": 1.1770248413085938e-05,
      "step": 12342,
      "training_step_time": 6.774948596954346
    },
    {
      "epoch": 1.1771202087402344e-05,
      "model_forward_time": 1.7368767261505127,
      "step": 12343
    },
    {
      "epoch": 1.1771202087402344e-05,
      "step": 12343,
      "training_step_time": 6.704565763473511
    },
    {
      "epoch": 1.177215576171875e-05,
      "model_forward_time": 1.7337181568145752,
      "step": 12344
    },
    {
      "epoch": 1.177215576171875e-05,
      "step": 12344,
      "training_step_time": 6.714331865310669
    },
    {
      "epoch": 1.1773109436035157e-05,
      "model_forward_time": 1.7358741760253906,
      "step": 12345
    },
    {
      "epoch": 1.1773109436035157e-05,
      "step": 12345,
      "training_step_time": 6.805233955383301
    },
    {
      "epoch": 1.1774063110351563e-05,
      "model_forward_time": 1.7272124290466309,
      "step": 12346
    },
    {
      "epoch": 1.1774063110351563e-05,
      "step": 12346,
      "training_step_time": 7.286984443664551
    },
    {
      "epoch": 1.1775016784667968e-05,
      "model_forward_time": 2.995781183242798,
      "step": 12347
    },
    {
      "epoch": 1.1775016784667968e-05,
      "step": 12347,
      "training_step_time": 9.821658372879028
    },
    {
      "epoch": 1.1775970458984376e-05,
      "model_forward_time": 1.759505271911621,
      "step": 12348
    },
    {
      "epoch": 1.1775970458984376e-05,
      "step": 12348,
      "training_step_time": 7.682951211929321
    },
    {
      "epoch": 1.1776924133300781e-05,
      "model_forward_time": 1.7208893299102783,
      "step": 12349
    },
    {
      "epoch": 1.1776924133300781e-05,
      "step": 12349,
      "training_step_time": 6.748331546783447
    },
    {
      "epoch": 1.1777877807617187e-05,
      "grad_norm": 0.220355823636055,
      "learning_rate": 9.853070007710565e-06,
      "loss": 0.0674,
      "step": 12350
    },
    {
      "epoch": 1.1777877807617187e-05,
      "model_forward_time": 1.743476152420044,
      "step": 12350
    },
    {
      "dynamics_loss_avg": 0.05107854679226875,
      "epoch": 1.1777877807617187e-05,
      "step": 12350
    },
    {
      "action_loss_avg": 0.004291919490788132,
      "epoch": 1.1777877807617187e-05,
      "step": 12350
    },
    {
      "epoch": 1.1777877807617187e-05,
      "step": 12350,
      "training_step_time": 6.817122220993042
    },
    {
      "epoch": 1.1778831481933595e-05,
      "model_forward_time": 1.0249953269958496,
      "step": 12351
    },
    {
      "epoch": 1.1778831481933595e-05,
      "step": 12351,
      "training_step_time": 6.79155969619751
    },
    {
      "epoch": 1.177978515625e-05,
      "model_forward_time": 1.7311344146728516,
      "step": 12352
    },
    {
      "epoch": 1.177978515625e-05,
      "step": 12352,
      "training_step_time": 6.766517162322998
    },
    {
      "epoch": 1.1780738830566406e-05,
      "model_forward_time": 1.7336106300354004,
      "step": 12353
    },
    {
      "epoch": 1.1780738830566406e-05,
      "step": 12353,
      "training_step_time": 6.853067874908447
    },
    {
      "epoch": 1.1781692504882812e-05,
      "model_forward_time": 1.7319777011871338,
      "step": 12354
    },
    {
      "epoch": 1.1781692504882812e-05,
      "step": 12354,
      "training_step_time": 6.786941766738892
    },
    {
      "epoch": 1.178264617919922e-05,
      "model_forward_time": 1.7277369499206543,
      "step": 12355
    },
    {
      "epoch": 1.178264617919922e-05,
      "step": 12355,
      "training_step_time": 6.824451208114624
    },
    {
      "epoch": 1.1783599853515625e-05,
      "model_forward_time": 1.7591242790222168,
      "step": 12356
    },
    {
      "epoch": 1.1783599853515625e-05,
      "step": 12356,
      "training_step_time": 8.550414800643921
    },
    {
      "epoch": 1.178455352783203e-05,
      "model_forward_time": 2.9377222061157227,
      "step": 12357
    },
    {
      "epoch": 1.178455352783203e-05,
      "step": 12357,
      "training_step_time": 9.941848993301392
    },
    {
      "epoch": 1.1785507202148438e-05,
      "model_forward_time": 1.7393999099731445,
      "step": 12358
    },
    {
      "epoch": 1.1785507202148438e-05,
      "step": 12358,
      "training_step_time": 7.827500104904175
    },
    {
      "epoch": 1.1786460876464844e-05,
      "model_forward_time": 1.7345709800720215,
      "step": 12359
    },
    {
      "epoch": 1.1786460876464844e-05,
      "step": 12359,
      "training_step_time": 6.78445839881897
    },
    {
      "epoch": 1.178741455078125e-05,
      "grad_norm": 0.21049518883228302,
      "learning_rate": 9.852671848457039e-06,
      "loss": 0.0712,
      "step": 12360
    },
    {
      "epoch": 1.178741455078125e-05,
      "model_forward_time": 1.7290830612182617,
      "step": 12360
    },
    {
      "dynamics_loss_avg": 0.08455189615488053,
      "epoch": 1.178741455078125e-05,
      "step": 12360
    },
    {
      "action_loss_avg": 0.004681317642098293,
      "epoch": 1.178741455078125e-05,
      "step": 12360
    },
    {
      "epoch": 1.178741455078125e-05,
      "step": 12360,
      "training_step_time": 6.774085283279419
    },
    {
      "epoch": 1.1788368225097657e-05,
      "model_forward_time": 1.945131778717041,
      "step": 12361
    },
    {
      "epoch": 1.1788368225097657e-05,
      "step": 12361,
      "training_step_time": 6.7707436084747314
    },
    {
      "epoch": 1.1789321899414063e-05,
      "model_forward_time": 1.9479093551635742,
      "step": 12362
    },
    {
      "epoch": 1.1789321899414063e-05,
      "step": 12362,
      "training_step_time": 6.751008987426758
    },
    {
      "epoch": 1.1790275573730468e-05,
      "model_forward_time": 1.941307544708252,
      "step": 12363
    },
    {
      "epoch": 1.1790275573730468e-05,
      "step": 12363,
      "training_step_time": 6.755524396896362
    },
    {
      "epoch": 1.1791229248046876e-05,
      "model_forward_time": 1.1862900257110596,
      "step": 12364
    },
    {
      "epoch": 1.1791229248046876e-05,
      "step": 12364,
      "training_step_time": 6.820433855056763
    },
    {
      "epoch": 1.1792182922363281e-05,
      "model_forward_time": 1.9322340488433838,
      "step": 12365
    },
    {
      "epoch": 1.1792182922363281e-05,
      "step": 12365,
      "training_step_time": 7.8070220947265625
    },
    {
      "epoch": 1.1793136596679687e-05,
      "model_forward_time": 3.1128909587860107,
      "step": 12366
    },
    {
      "epoch": 1.1793136596679687e-05,
      "step": 12366,
      "training_step_time": 10.31849193572998
    },
    {
      "epoch": 1.1794090270996095e-05,
      "model_forward_time": 1.9506409168243408,
      "step": 12367
    },
    {
      "epoch": 1.1794090270996095e-05,
      "step": 12367,
      "training_step_time": 7.805225610733032
    },
    {
      "epoch": 1.17950439453125e-05,
      "model_forward_time": 1.924201250076294,
      "step": 12368
    },
    {
      "epoch": 1.17950439453125e-05,
      "step": 12368,
      "training_step_time": 6.8841259479522705
    },
    {
      "epoch": 1.1795997619628906e-05,
      "model_forward_time": 1.9612751007080078,
      "step": 12369
    },
    {
      "epoch": 1.1795997619628906e-05,
      "step": 12369,
      "training_step_time": 6.940903186798096
    },
    {
      "epoch": 1.1796951293945312e-05,
      "grad_norm": 0.1669686883687973,
      "learning_rate": 9.852273158522616e-06,
      "loss": 0.0701,
      "step": 12370
    },
    {
      "epoch": 1.1796951293945312e-05,
      "model_forward_time": 1.9464335441589355,
      "step": 12370
    },
    {
      "dynamics_loss_avg": 0.06079897545278072,
      "epoch": 1.1796951293945312e-05,
      "step": 12370
    },
    {
      "action_loss_avg": 0.004410360148176551,
      "epoch": 1.1796951293945312e-05,
      "step": 12370
    },
    {
      "epoch": 1.1796951293945312e-05,
      "step": 12370,
      "training_step_time": 6.833382844924927
    },
    {
      "epoch": 1.179790496826172e-05,
      "model_forward_time": 1.9297890663146973,
      "step": 12371
    },
    {
      "epoch": 1.179790496826172e-05,
      "step": 12371,
      "training_step_time": 6.846032619476318
    },
    {
      "epoch": 1.1798858642578125e-05,
      "model_forward_time": 1.94331955909729,
      "step": 12372
    },
    {
      "epoch": 1.1798858642578125e-05,
      "step": 12372,
      "training_step_time": 6.855298042297363
    },
    {
      "epoch": 1.179981231689453e-05,
      "model_forward_time": 1.9491307735443115,
      "step": 12373
    },
    {
      "epoch": 1.179981231689453e-05,
      "step": 12373,
      "training_step_time": 6.8517396450042725
    },
    {
      "epoch": 1.1800765991210938e-05,
      "model_forward_time": 1.9609503746032715,
      "step": 12374
    },
    {
      "epoch": 1.1800765991210938e-05,
      "step": 12374,
      "training_step_time": 8.866106033325195
    },
    {
      "epoch": 1.1801719665527344e-05,
      "model_forward_time": 3.0667524337768555,
      "step": 12375
    },
    {
      "epoch": 1.1801719665527344e-05,
      "step": 12375,
      "training_step_time": 8.508291721343994
    },
    {
      "epoch": 1.180267333984375e-05,
      "model_forward_time": 1.9493837356567383,
      "step": 12376
    },
    {
      "epoch": 1.180267333984375e-05,
      "step": 12376,
      "training_step_time": 7.7522337436676025
    },
    {
      "epoch": 1.1803627014160157e-05,
      "model_forward_time": 1.9562320709228516,
      "step": 12377
    },
    {
      "epoch": 1.1803627014160157e-05,
      "step": 12377,
      "training_step_time": 6.898374795913696
    },
    {
      "epoch": 1.1804580688476563e-05,
      "model_forward_time": 1.94380784034729,
      "step": 12378
    },
    {
      "epoch": 1.1804580688476563e-05,
      "step": 12378,
      "training_step_time": 6.861719846725464
    },
    {
      "epoch": 1.1805534362792968e-05,
      "model_forward_time": 1.9328136444091797,
      "step": 12379
    },
    {
      "epoch": 1.1805534362792968e-05,
      "step": 12379,
      "training_step_time": 6.87826943397522
    },
    {
      "epoch": 1.1806488037109376e-05,
      "grad_norm": 0.25213029980659485,
      "learning_rate": 9.851873937950896e-06,
      "loss": 0.0699,
      "step": 12380
    },
    {
      "epoch": 1.1806488037109376e-05,
      "model_forward_time": 1.947695255279541,
      "step": 12380
    },
    {
      "dynamics_loss_avg": 0.0675217367708683,
      "epoch": 1.1806488037109376e-05,
      "step": 12380
    },
    {
      "action_loss_avg": 0.0036945183179341256,
      "epoch": 1.1806488037109376e-05,
      "step": 12380
    },
    {
      "epoch": 1.1806488037109376e-05,
      "step": 12380,
      "training_step_time": 6.959591865539551
    },
    {
      "epoch": 1.1807441711425781e-05,
      "model_forward_time": 1.9467315673828125,
      "step": 12381
    },
    {
      "epoch": 1.1807441711425781e-05,
      "step": 12381,
      "training_step_time": 6.894712448120117
    },
    {
      "epoch": 1.1808395385742187e-05,
      "model_forward_time": 1.9556236267089844,
      "step": 12382
    },
    {
      "epoch": 1.1808395385742187e-05,
      "step": 12382,
      "training_step_time": 6.88261342048645
    },
    {
      "epoch": 1.1809349060058593e-05,
      "model_forward_time": 1.9518415927886963,
      "step": 12383
    },
    {
      "epoch": 1.1809349060058593e-05,
      "step": 12383,
      "training_step_time": 9.862669944763184
    },
    {
      "epoch": 1.1810302734375e-05,
      "model_forward_time": 3.0502474308013916,
      "step": 12384
    },
    {
      "epoch": 1.1810302734375e-05,
      "step": 12384,
      "training_step_time": 8.66462254524231
    },
    {
      "epoch": 1.1811256408691406e-05,
      "model_forward_time": 1.7852578163146973,
      "step": 12385
    },
    {
      "epoch": 1.1811256408691406e-05,
      "step": 12385,
      "training_step_time": 7.834791898727417
    },
    {
      "epoch": 1.1812210083007812e-05,
      "model_forward_time": 1.7579238414764404,
      "step": 12386
    },
    {
      "epoch": 1.1812210083007812e-05,
      "step": 12386,
      "training_step_time": 7.644813299179077
    },
    {
      "epoch": 1.181316375732422e-05,
      "model_forward_time": 1.7444956302642822,
      "step": 12387
    },
    {
      "epoch": 1.181316375732422e-05,
      "step": 12387,
      "training_step_time": 6.836166143417358
    },
    {
      "epoch": 1.1814117431640625e-05,
      "model_forward_time": 1.7355289459228516,
      "step": 12388
    },
    {
      "epoch": 1.1814117431640625e-05,
      "step": 12388,
      "training_step_time": 6.861291885375977
    },
    {
      "epoch": 1.181507110595703e-05,
      "model_forward_time": 1.746473789215088,
      "step": 12389
    },
    {
      "epoch": 1.181507110595703e-05,
      "step": 12389,
      "training_step_time": 6.951260805130005
    },
    {
      "epoch": 1.1816024780273438e-05,
      "grad_norm": 0.18556231260299683,
      "learning_rate": 9.851474186785536e-06,
      "loss": 0.0702,
      "step": 12390
    },
    {
      "epoch": 1.1816024780273438e-05,
      "model_forward_time": 1.7513079643249512,
      "step": 12390
    },
    {
      "dynamics_loss_avg": 0.06184350233525038,
      "epoch": 1.1816024780273438e-05,
      "step": 12390
    },
    {
      "action_loss_avg": 0.003769572312012315,
      "epoch": 1.1816024780273438e-05,
      "step": 12390
    },
    {
      "epoch": 1.1816024780273438e-05,
      "step": 12390,
      "training_step_time": 6.891961336135864
    },
    {
      "epoch": 1.1816978454589844e-05,
      "model_forward_time": 1.7548117637634277,
      "step": 12391
    },
    {
      "epoch": 1.1816978454589844e-05,
      "step": 12391,
      "training_step_time": 6.882264614105225
    },
    {
      "epoch": 1.181793212890625e-05,
      "model_forward_time": 1.73531174659729,
      "step": 12392
    },
    {
      "epoch": 1.181793212890625e-05,
      "step": 12392,
      "training_step_time": 8.973560094833374
    },
    {
      "epoch": 1.1818885803222657e-05,
      "model_forward_time": 1.7947168350219727,
      "step": 12393
    },
    {
      "epoch": 1.1818885803222657e-05,
      "step": 12393,
      "training_step_time": 11.236283540725708
    },
    {
      "epoch": 1.1819839477539063e-05,
      "model_forward_time": 2.9434094429016113,
      "step": 12394
    },
    {
      "epoch": 1.1819839477539063e-05,
      "step": 12394,
      "training_step_time": 7.8679420948028564
    },
    {
      "epoch": 1.1820793151855468e-05,
      "model_forward_time": 1.7589929103851318,
      "step": 12395
    },
    {
      "epoch": 1.1820793151855468e-05,
      "step": 12395,
      "training_step_time": 7.728376388549805
    },
    {
      "epoch": 1.1821746826171876e-05,
      "model_forward_time": 1.7482364177703857,
      "step": 12396
    },
    {
      "epoch": 1.1821746826171876e-05,
      "step": 12396,
      "training_step_time": 6.8457629680633545
    },
    {
      "epoch": 1.1822700500488281e-05,
      "model_forward_time": 1.7492659091949463,
      "step": 12397
    },
    {
      "epoch": 1.1822700500488281e-05,
      "step": 12397,
      "training_step_time": 6.870338201522827
    },
    {
      "epoch": 1.1823654174804687e-05,
      "model_forward_time": 1.7484498023986816,
      "step": 12398
    },
    {
      "epoch": 1.1823654174804687e-05,
      "step": 12398,
      "training_step_time": 6.916778087615967
    },
    {
      "epoch": 1.1824607849121093e-05,
      "model_forward_time": 1.7603342533111572,
      "step": 12399
    },
    {
      "epoch": 1.1824607849121093e-05,
      "step": 12399,
      "training_step_time": 6.862254619598389
    },
    {
      "epoch": 1.18255615234375e-05,
      "grad_norm": 0.30699479579925537,
      "learning_rate": 9.851073905070255e-06,
      "loss": 0.0695,
      "step": 12400
    },
    {
      "epoch": 1.18255615234375e-05,
      "model_forward_time": 1.7455627918243408,
      "step": 12400
    },
    {
      "dynamics_loss_avg": 0.05783942565321922,
      "epoch": 1.18255615234375e-05,
      "step": 12400
    },
    {
      "action_loss_avg": 0.0037638078443706035,
      "epoch": 1.18255615234375e-05,
      "step": 12400
    },
    {
      "epoch": 1.18255615234375e-05,
      "step": 12400,
      "training_step_time": 6.893918037414551
    },
    {
      "epoch": 1.1826515197753906e-05,
      "model_forward_time": 1.7187175750732422,
      "step": 12401
    },
    {
      "epoch": 1.1826515197753906e-05,
      "step": 12401,
      "training_step_time": 8.233258724212646
    },
    {
      "epoch": 1.1827468872070312e-05,
      "model_forward_time": 1.758744478225708,
      "step": 12402
    },
    {
      "epoch": 1.1827468872070312e-05,
      "step": 12402,
      "training_step_time": 9.066089391708374
    },
    {
      "epoch": 1.182842254638672e-05,
      "model_forward_time": 2.995333433151245,
      "step": 12403
    },
    {
      "epoch": 1.182842254638672e-05,
      "step": 12403,
      "training_step_time": 7.89463996887207
    },
    {
      "epoch": 1.1829376220703125e-05,
      "model_forward_time": 1.7407424449920654,
      "step": 12404
    },
    {
      "epoch": 1.1829376220703125e-05,
      "step": 12404,
      "training_step_time": 7.841344833374023
    },
    {
      "epoch": 1.183032989501953e-05,
      "model_forward_time": 1.736962080001831,
      "step": 12405
    },
    {
      "epoch": 1.183032989501953e-05,
      "step": 12405,
      "training_step_time": 6.864908456802368
    },
    {
      "epoch": 1.1831283569335938e-05,
      "model_forward_time": 1.7720942497253418,
      "step": 12406
    },
    {
      "epoch": 1.1831283569335938e-05,
      "step": 12406,
      "training_step_time": 6.883001804351807
    },
    {
      "epoch": 1.1832237243652344e-05,
      "model_forward_time": 1.7395820617675781,
      "step": 12407
    },
    {
      "epoch": 1.1832237243652344e-05,
      "step": 12407,
      "training_step_time": 6.887802600860596
    },
    {
      "epoch": 1.183319091796875e-05,
      "model_forward_time": 1.7704815864562988,
      "step": 12408
    },
    {
      "epoch": 1.183319091796875e-05,
      "step": 12408,
      "training_step_time": 6.850287914276123
    },
    {
      "epoch": 1.1834144592285157e-05,
      "model_forward_time": 1.7742924690246582,
      "step": 12409
    },
    {
      "epoch": 1.1834144592285157e-05,
      "step": 12409,
      "training_step_time": 6.85395622253418
    },
    {
      "epoch": 1.1835098266601563e-05,
      "grad_norm": 0.25273430347442627,
      "learning_rate": 9.850673092848824e-06,
      "loss": 0.0649,
      "step": 12410
    },
    {
      "epoch": 1.1835098266601563e-05,
      "model_forward_time": 1.755748987197876,
      "step": 12410
    },
    {
      "dynamics_loss_avg": 0.07774659227579832,
      "epoch": 1.1835098266601563e-05,
      "step": 12410
    },
    {
      "action_loss_avg": 0.005049753660568967,
      "epoch": 1.1835098266601563e-05,
      "step": 12410
    },
    {
      "epoch": 1.1835098266601563e-05,
      "step": 12410,
      "training_step_time": 7.594707727432251
    },
    {
      "epoch": 1.1836051940917968e-05,
      "model_forward_time": 1.7294766902923584,
      "step": 12411
    },
    {
      "epoch": 1.1836051940917968e-05,
      "step": 12411,
      "training_step_time": 8.962512731552124
    },
    {
      "epoch": 1.1837005615234376e-05,
      "model_forward_time": 2.940669059753418,
      "step": 12412
    },
    {
      "epoch": 1.1837005615234376e-05,
      "step": 12412,
      "training_step_time": 10.594286680221558
    },
    {
      "epoch": 1.1837959289550781e-05,
      "model_forward_time": 1.7370190620422363,
      "step": 12413
    },
    {
      "epoch": 1.1837959289550781e-05,
      "step": 12413,
      "training_step_time": 7.786458492279053
    },
    {
      "epoch": 1.1838912963867187e-05,
      "model_forward_time": 1.7485525608062744,
      "step": 12414
    },
    {
      "epoch": 1.1838912963867187e-05,
      "step": 12414,
      "training_step_time": 7.657683849334717
    },
    {
      "epoch": 1.1839866638183593e-05,
      "model_forward_time": 1.7387981414794922,
      "step": 12415
    },
    {
      "epoch": 1.1839866638183593e-05,
      "step": 12415,
      "training_step_time": 6.8041486740112305
    },
    {
      "epoch": 1.18408203125e-05,
      "model_forward_time": 1.736264944076538,
      "step": 12416
    },
    {
      "epoch": 1.18408203125e-05,
      "step": 12416,
      "training_step_time": 6.85705041885376
    },
    {
      "epoch": 1.1841773986816406e-05,
      "model_forward_time": 1.7453782558441162,
      "step": 12417
    },
    {
      "epoch": 1.1841773986816406e-05,
      "step": 12417,
      "training_step_time": 6.758885622024536
    },
    {
      "epoch": 1.1842727661132812e-05,
      "model_forward_time": 1.7355103492736816,
      "step": 12418
    },
    {
      "epoch": 1.1842727661132812e-05,
      "step": 12418,
      "training_step_time": 6.808748960494995
    },
    {
      "epoch": 1.184368133544922e-05,
      "model_forward_time": 1.7421324253082275,
      "step": 12419
    },
    {
      "epoch": 1.184368133544922e-05,
      "step": 12419,
      "training_step_time": 7.736051797866821
    },
    {
      "epoch": 1.1844635009765625e-05,
      "grad_norm": 0.29517969489097595,
      "learning_rate": 9.850271750165077e-06,
      "loss": 0.0683,
      "step": 12420
    },
    {
      "epoch": 1.1844635009765625e-05,
      "model_forward_time": 1.7448296546936035,
      "step": 12420
    },
    {
      "dynamics_loss_avg": 0.06315667256712913,
      "epoch": 1.1844635009765625e-05,
      "step": 12420
    },
    {
      "action_loss_avg": 0.003772403404582292,
      "epoch": 1.1844635009765625e-05,
      "step": 12420
    },
    {
      "epoch": 1.1844635009765625e-05,
      "step": 12420,
      "training_step_time": 9.010530471801758
    },
    {
      "epoch": 1.184558868408203e-05,
      "model_forward_time": 2.8758318424224854,
      "step": 12421
    },
    {
      "epoch": 1.184558868408203e-05,
      "step": 12421,
      "training_step_time": 9.892848253250122
    },
    {
      "epoch": 1.1846542358398438e-05,
      "model_forward_time": 1.7416863441467285,
      "step": 12422
    },
    {
      "epoch": 1.1846542358398438e-05,
      "step": 12422,
      "training_step_time": 9.521689891815186
    },
    {
      "epoch": 1.1847496032714844e-05,
      "model_forward_time": 1.7399165630340576,
      "step": 12423
    },
    {
      "epoch": 1.1847496032714844e-05,
      "step": 12423,
      "training_step_time": 7.427424907684326
    },
    {
      "epoch": 1.184844970703125e-05,
      "model_forward_time": 1.7466216087341309,
      "step": 12424
    },
    {
      "epoch": 1.184844970703125e-05,
      "step": 12424,
      "training_step_time": 6.799970865249634
    },
    {
      "epoch": 1.1849403381347657e-05,
      "model_forward_time": 1.7417218685150146,
      "step": 12425
    },
    {
      "epoch": 1.1849403381347657e-05,
      "step": 12425,
      "training_step_time": 6.897870302200317
    },
    {
      "epoch": 1.1850357055664063e-05,
      "model_forward_time": 1.7426323890686035,
      "step": 12426
    },
    {
      "epoch": 1.1850357055664063e-05,
      "step": 12426,
      "training_step_time": 6.834010362625122
    },
    {
      "epoch": 1.1851310729980468e-05,
      "model_forward_time": 1.7326805591583252,
      "step": 12427
    },
    {
      "epoch": 1.1851310729980468e-05,
      "step": 12427,
      "training_step_time": 6.858102083206177
    },
    {
      "epoch": 1.1852264404296876e-05,
      "model_forward_time": 1.7590749263763428,
      "step": 12428
    },
    {
      "epoch": 1.1852264404296876e-05,
      "step": 12428,
      "training_step_time": 7.76547646522522
    },
    {
      "epoch": 1.1853218078613282e-05,
      "model_forward_time": 1.7664098739624023,
      "step": 12429
    },
    {
      "epoch": 1.1853218078613282e-05,
      "step": 12429,
      "training_step_time": 8.743076086044312
    },
    {
      "epoch": 1.1854171752929687e-05,
      "grad_norm": 0.22063741087913513,
      "learning_rate": 9.849869877062903e-06,
      "loss": 0.0663,
      "step": 12430
    },
    {
      "epoch": 1.1854171752929687e-05,
      "model_forward_time": 2.9025206565856934,
      "step": 12430
    },
    {
      "dynamics_loss_avg": 0.069798156991601,
      "epoch": 1.1854171752929687e-05,
      "step": 12430
    },
    {
      "action_loss_avg": 0.004282224143389612,
      "epoch": 1.1854171752929687e-05,
      "step": 12430
    },
    {
      "epoch": 1.1854171752929687e-05,
      "step": 12430,
      "training_step_time": 9.825863361358643
    },
    {
      "epoch": 1.1855125427246093e-05,
      "model_forward_time": 1.7625761032104492,
      "step": 12431
    },
    {
      "epoch": 1.1855125427246093e-05,
      "step": 12431,
      "training_step_time": 7.664851427078247
    },
    {
      "epoch": 1.18560791015625e-05,
      "model_forward_time": 1.7567474842071533,
      "step": 12432
    },
    {
      "epoch": 1.18560791015625e-05,
      "step": 12432,
      "training_step_time": 7.568865060806274
    },
    {
      "epoch": 1.1857032775878906e-05,
      "model_forward_time": 1.734724998474121,
      "step": 12433
    },
    {
      "epoch": 1.1857032775878906e-05,
      "step": 12433,
      "training_step_time": 6.61956787109375
    },
    {
      "epoch": 1.1857986450195312e-05,
      "model_forward_time": 1.7509801387786865,
      "step": 12434
    },
    {
      "epoch": 1.1857986450195312e-05,
      "step": 12434,
      "training_step_time": 6.7244791984558105
    },
    {
      "epoch": 1.185894012451172e-05,
      "model_forward_time": 1.7580170631408691,
      "step": 12435
    },
    {
      "epoch": 1.185894012451172e-05,
      "step": 12435,
      "training_step_time": 6.675034046173096
    },
    {
      "epoch": 1.1859893798828125e-05,
      "model_forward_time": 1.7505178451538086,
      "step": 12436
    },
    {
      "epoch": 1.1859893798828125e-05,
      "step": 12436,
      "training_step_time": 6.702397346496582
    },
    {
      "epoch": 1.186084747314453e-05,
      "model_forward_time": 1.7516429424285889,
      "step": 12437
    },
    {
      "epoch": 1.186084747314453e-05,
      "step": 12437,
      "training_step_time": 7.557757139205933
    },
    {
      "epoch": 1.1861801147460938e-05,
      "model_forward_time": 1.7577965259552002,
      "step": 12438
    },
    {
      "epoch": 1.1861801147460938e-05,
      "step": 12438,
      "training_step_time": 8.562416791915894
    },
    {
      "epoch": 1.1862754821777344e-05,
      "model_forward_time": 1.7612395286560059,
      "step": 12439
    },
    {
      "epoch": 1.1862754821777344e-05,
      "step": 12439,
      "training_step_time": 9.675880432128906
    },
    {
      "epoch": 1.186370849609375e-05,
      "grad_norm": 0.16699004173278809,
      "learning_rate": 9.849467473586252e-06,
      "loss": 0.0597,
      "step": 12440
    },
    {
      "epoch": 1.186370849609375e-05,
      "model_forward_time": 1.747427225112915,
      "step": 12440
    },
    {
      "dynamics_loss_avg": 0.06082656625658274,
      "epoch": 1.186370849609375e-05,
      "step": 12440
    },
    {
      "action_loss_avg": 0.003751515719341114,
      "epoch": 1.186370849609375e-05,
      "step": 12440
    },
    {
      "epoch": 1.186370849609375e-05,
      "step": 12440,
      "training_step_time": 8.57283616065979
    },
    {
      "epoch": 1.1864662170410157e-05,
      "model_forward_time": 1.748530626296997,
      "step": 12441
    },
    {
      "epoch": 1.1864662170410157e-05,
      "step": 12441,
      "training_step_time": 7.629561901092529
    },
    {
      "epoch": 1.1865615844726563e-05,
      "model_forward_time": 1.7525355815887451,
      "step": 12442
    },
    {
      "epoch": 1.1865615844726563e-05,
      "step": 12442,
      "training_step_time": 6.604926586151123
    },
    {
      "epoch": 1.1866569519042968e-05,
      "model_forward_time": 1.751866102218628,
      "step": 12443
    },
    {
      "epoch": 1.1866569519042968e-05,
      "step": 12443,
      "training_step_time": 6.6894683837890625
    },
    {
      "epoch": 1.1867523193359376e-05,
      "model_forward_time": 1.753286600112915,
      "step": 12444
    },
    {
      "epoch": 1.1867523193359376e-05,
      "step": 12444,
      "training_step_time": 6.650468111038208
    },
    {
      "epoch": 1.1868476867675782e-05,
      "model_forward_time": 1.7471249103546143,
      "step": 12445
    },
    {
      "epoch": 1.1868476867675782e-05,
      "step": 12445,
      "training_step_time": 6.693827867507935
    },
    {
      "epoch": 1.1869430541992187e-05,
      "model_forward_time": 1.7455575466156006,
      "step": 12446
    },
    {
      "epoch": 1.1869430541992187e-05,
      "step": 12446,
      "training_step_time": 7.59608006477356
    },
    {
      "epoch": 1.1870384216308593e-05,
      "model_forward_time": 1.7572495937347412,
      "step": 12447
    },
    {
      "epoch": 1.1870384216308593e-05,
      "step": 12447,
      "training_step_time": 8.108359813690186
    },
    {
      "epoch": 1.1871337890625e-05,
      "model_forward_time": 1.7505261898040771,
      "step": 12448
    },
    {
      "epoch": 1.1871337890625e-05,
      "step": 12448,
      "training_step_time": 8.644691228866577
    },
    {
      "epoch": 1.1872291564941406e-05,
      "model_forward_time": 2.867518186569214,
      "step": 12449
    },
    {
      "epoch": 1.1872291564941406e-05,
      "step": 12449,
      "training_step_time": 7.613973140716553
    },
    {
      "epoch": 1.1873245239257812e-05,
      "grad_norm": 0.2345506101846695,
      "learning_rate": 9.849064539779127e-06,
      "loss": 0.0643,
      "step": 12450
    },
    {
      "epoch": 1.1873245239257812e-05,
      "model_forward_time": 1.745159387588501,
      "step": 12450
    },
    {
      "dynamics_loss_avg": 0.04264874029904604,
      "epoch": 1.1873245239257812e-05,
      "step": 12450
    },
    {
      "action_loss_avg": 0.0016884431417565793,
      "epoch": 1.1873245239257812e-05,
      "step": 12450
    },
    {
      "epoch": 1.1873245239257812e-05,
      "step": 12450,
      "training_step_time": 7.652790069580078
    },
    {
      "epoch": 1.187419891357422e-05,
      "model_forward_time": 1.7502870559692383,
      "step": 12451
    },
    {
      "epoch": 1.187419891357422e-05,
      "step": 12451,
      "training_step_time": 7.509547710418701
    },
    {
      "epoch": 1.1875152587890625e-05,
      "model_forward_time": 1.3943402767181396,
      "step": 12452
    },
    {
      "epoch": 1.1875152587890625e-05,
      "step": 12452,
      "training_step_time": 6.607797622680664
    },
    {
      "epoch": 1.187610626220703e-05,
      "model_forward_time": 1.7446563243865967,
      "step": 12453
    },
    {
      "epoch": 1.187610626220703e-05,
      "step": 12453,
      "training_step_time": 6.701686143875122
    },
    {
      "epoch": 1.1877059936523438e-05,
      "model_forward_time": 1.739166259765625,
      "step": 12454
    },
    {
      "epoch": 1.1877059936523438e-05,
      "step": 12454,
      "training_step_time": 6.6974265575408936
    },
    {
      "epoch": 1.1878013610839844e-05,
      "model_forward_time": 1.759772777557373,
      "step": 12455
    },
    {
      "epoch": 1.1878013610839844e-05,
      "step": 12455,
      "training_step_time": 7.460169076919556
    },
    {
      "epoch": 1.187896728515625e-05,
      "model_forward_time": 1.052351474761963,
      "step": 12456
    },
    {
      "epoch": 1.187896728515625e-05,
      "step": 12456,
      "training_step_time": 7.587414979934692
    },
    {
      "epoch": 1.1879920959472657e-05,
      "model_forward_time": 1.7510902881622314,
      "step": 12457
    },
    {
      "epoch": 1.1879920959472657e-05,
      "step": 12457,
      "training_step_time": 8.42423415184021
    },
    {
      "epoch": 1.1880874633789063e-05,
      "model_forward_time": 2.970817804336548,
      "step": 12458
    },
    {
      "epoch": 1.1880874633789063e-05,
      "step": 12458,
      "training_step_time": 7.728780508041382
    },
    {
      "epoch": 1.1881828308105468e-05,
      "model_forward_time": 1.7496864795684814,
      "step": 12459
    },
    {
      "epoch": 1.1881828308105468e-05,
      "step": 12459,
      "training_step_time": 7.615422487258911
    },
    {
      "epoch": 1.1882781982421876e-05,
      "grad_norm": 0.17356574535369873,
      "learning_rate": 9.848661075685595e-06,
      "loss": 0.0632,
      "step": 12460
    },
    {
      "epoch": 1.1882781982421876e-05,
      "model_forward_time": 1.7397551536560059,
      "step": 12460
    },
    {
      "dynamics_loss_avg": 0.05365514550358057,
      "epoch": 1.1882781982421876e-05,
      "step": 12460
    },
    {
      "action_loss_avg": 0.00347886768868193,
      "epoch": 1.1882781982421876e-05,
      "step": 12460
    },
    {
      "epoch": 1.1882781982421876e-05,
      "step": 12460,
      "training_step_time": 8.501047372817993
    },
    {
      "epoch": 1.1883735656738282e-05,
      "model_forward_time": 1.7576484680175781,
      "step": 12461
    },
    {
      "epoch": 1.1883735656738282e-05,
      "step": 12461,
      "training_step_time": 6.676599740982056
    },
    {
      "epoch": 1.1884689331054687e-05,
      "model_forward_time": 1.7163939476013184,
      "step": 12462
    },
    {
      "epoch": 1.1884689331054687e-05,
      "step": 12462,
      "training_step_time": 6.736708879470825
    },
    {
      "epoch": 1.1885643005371093e-05,
      "model_forward_time": 1.7382922172546387,
      "step": 12463
    },
    {
      "epoch": 1.1885643005371093e-05,
      "step": 12463,
      "training_step_time": 6.69716215133667
    },
    {
      "epoch": 1.18865966796875e-05,
      "model_forward_time": 1.741323471069336,
      "step": 12464
    },
    {
      "epoch": 1.18865966796875e-05,
      "step": 12464,
      "training_step_time": 7.513857126235962
    },
    {
      "epoch": 1.1887550354003906e-05,
      "model_forward_time": 1.7457566261291504,
      "step": 12465
    },
    {
      "epoch": 1.1887550354003906e-05,
      "step": 12465,
      "training_step_time": 7.599489688873291
    },
    {
      "epoch": 1.1888504028320312e-05,
      "model_forward_time": 1.7439050674438477,
      "step": 12466
    },
    {
      "epoch": 1.1888504028320312e-05,
      "step": 12466,
      "training_step_time": 9.27461576461792
    },
    {
      "epoch": 1.188945770263672e-05,
      "model_forward_time": 2.929276704788208,
      "step": 12467
    },
    {
      "epoch": 1.188945770263672e-05,
      "step": 12467,
      "training_step_time": 8.537070751190186
    },
    {
      "epoch": 1.1890411376953125e-05,
      "model_forward_time": 1.7492475509643555,
      "step": 12468
    },
    {
      "epoch": 1.1890411376953125e-05,
      "step": 12468,
      "training_step_time": 9.036933660507202
    },
    {
      "epoch": 1.189136505126953e-05,
      "model_forward_time": 1.7420551776885986,
      "step": 12469
    },
    {
      "epoch": 1.189136505126953e-05,
      "step": 12469,
      "training_step_time": 7.647913694381714
    },
    {
      "epoch": 1.1892318725585938e-05,
      "grad_norm": 0.22586442530155182,
      "learning_rate": 9.848257081349779e-06,
      "loss": 0.0629,
      "step": 12470
    },
    {
      "epoch": 1.1892318725585938e-05,
      "model_forward_time": 1.7517504692077637,
      "step": 12470
    },
    {
      "dynamics_loss_avg": 0.047227161657065156,
      "epoch": 1.1892318725585938e-05,
      "step": 12470
    },
    {
      "action_loss_avg": 0.001841299602529034,
      "epoch": 1.1892318725585938e-05,
      "step": 12470
    },
    {
      "epoch": 1.1892318725585938e-05,
      "step": 12470,
      "training_step_time": 7.048433542251587
    },
    {
      "epoch": 1.1893272399902344e-05,
      "model_forward_time": 1.7422568798065186,
      "step": 12471
    },
    {
      "epoch": 1.1893272399902344e-05,
      "step": 12471,
      "training_step_time": 6.713054895401001
    },
    {
      "epoch": 1.189422607421875e-05,
      "model_forward_time": 1.743750810623169,
      "step": 12472
    },
    {
      "epoch": 1.189422607421875e-05,
      "step": 12472,
      "training_step_time": 6.648846626281738
    },
    {
      "epoch": 1.1895179748535157e-05,
      "model_forward_time": 1.7508957386016846,
      "step": 12473
    },
    {
      "epoch": 1.1895179748535157e-05,
      "step": 12473,
      "training_step_time": 7.452101230621338
    },
    {
      "epoch": 1.1896133422851563e-05,
      "model_forward_time": 1.7575535774230957,
      "step": 12474
    },
    {
      "epoch": 1.1896133422851563e-05,
      "step": 12474,
      "training_step_time": 8.483346939086914
    },
    {
      "epoch": 1.1897087097167968e-05,
      "model_forward_time": 1.7574243545532227,
      "step": 12475
    },
    {
      "epoch": 1.1897087097167968e-05,
      "step": 12475,
      "training_step_time": 7.844284534454346
    },
    {
      "epoch": 1.1898040771484376e-05,
      "model_forward_time": 1.8074331283569336,
      "step": 12476
    },
    {
      "epoch": 1.1898040771484376e-05,
      "step": 12476,
      "training_step_time": 9.631492137908936
    },
    {
      "epoch": 1.1898994445800782e-05,
      "model_forward_time": 2.9271819591522217,
      "step": 12477
    },
    {
      "epoch": 1.1898994445800782e-05,
      "step": 12477,
      "training_step_time": 10.553947448730469
    },
    {
      "epoch": 1.1899948120117187e-05,
      "model_forward_time": 1.7601335048675537,
      "step": 12478
    },
    {
      "epoch": 1.1899948120117187e-05,
      "step": 12478,
      "training_step_time": 7.6365439891815186
    },
    {
      "epoch": 1.1900901794433593e-05,
      "model_forward_time": 1.7699143886566162,
      "step": 12479
    },
    {
      "epoch": 1.1900901794433593e-05,
      "step": 12479,
      "training_step_time": 7.4909987449646
    },
    {
      "epoch": 1.190185546875e-05,
      "grad_norm": 0.19224107265472412,
      "learning_rate": 9.847852556815858e-06,
      "loss": 0.0669,
      "step": 12480
    },
    {
      "epoch": 1.190185546875e-05,
      "model_forward_time": 1.760786533355713,
      "step": 12480
    },
    {
      "dynamics_loss_avg": 0.054086067155003546,
      "epoch": 1.190185546875e-05,
      "step": 12480
    },
    {
      "action_loss_avg": 0.0032821652945131063,
      "epoch": 1.190185546875e-05,
      "step": 12480
    },
    {
      "epoch": 1.190185546875e-05,
      "step": 12480,
      "training_step_time": 6.658538341522217
    },
    {
      "epoch": 1.1902809143066406e-05,
      "model_forward_time": 1.7716577053070068,
      "step": 12481
    },
    {
      "epoch": 1.1902809143066406e-05,
      "step": 12481,
      "training_step_time": 6.729683876037598
    },
    {
      "epoch": 1.1903762817382812e-05,
      "model_forward_time": 1.7517719268798828,
      "step": 12482
    },
    {
      "epoch": 1.1903762817382812e-05,
      "step": 12482,
      "training_step_time": 7.345214128494263
    },
    {
      "epoch": 1.190471649169922e-05,
      "model_forward_time": 1.7569003105163574,
      "step": 12483
    },
    {
      "epoch": 1.190471649169922e-05,
      "step": 12483,
      "training_step_time": 7.563585042953491
    },
    {
      "epoch": 1.1905670166015625e-05,
      "model_forward_time": 1.755150556564331,
      "step": 12484
    },
    {
      "epoch": 1.1905670166015625e-05,
      "step": 12484,
      "training_step_time": 7.737431287765503
    },
    {
      "epoch": 1.190662384033203e-05,
      "model_forward_time": 1.7777609825134277,
      "step": 12485
    },
    {
      "epoch": 1.190662384033203e-05,
      "step": 12485,
      "training_step_time": 8.614986181259155
    },
    {
      "epoch": 1.1907577514648438e-05,
      "model_forward_time": 2.9258172512054443,
      "step": 12486
    },
    {
      "epoch": 1.1907577514648438e-05,
      "step": 12486,
      "training_step_time": 8.195313692092896
    },
    {
      "epoch": 1.1908531188964844e-05,
      "model_forward_time": 1.756342887878418,
      "step": 12487
    },
    {
      "epoch": 1.1908531188964844e-05,
      "step": 12487,
      "training_step_time": 7.599100112915039
    },
    {
      "epoch": 1.190948486328125e-05,
      "model_forward_time": 1.760746955871582,
      "step": 12488
    },
    {
      "epoch": 1.190948486328125e-05,
      "step": 12488,
      "training_step_time": 7.530431747436523
    },
    {
      "epoch": 1.1910438537597657e-05,
      "model_forward_time": 1.7592475414276123,
      "step": 12489
    },
    {
      "epoch": 1.1910438537597657e-05,
      "step": 12489,
      "training_step_time": 6.647248268127441
    },
    {
      "epoch": 1.1911392211914063e-05,
      "grad_norm": 0.3001171052455902,
      "learning_rate": 9.847447502128069e-06,
      "loss": 0.0705,
      "step": 12490
    },
    {
      "epoch": 1.1911392211914063e-05,
      "model_forward_time": 1.7543928623199463,
      "step": 12490
    },
    {
      "dynamics_loss_avg": 0.06228565704077482,
      "epoch": 1.1911392211914063e-05,
      "step": 12490
    },
    {
      "action_loss_avg": 0.0029467043408658356,
      "epoch": 1.1911392211914063e-05,
      "step": 12490
    },
    {
      "epoch": 1.1911392211914063e-05,
      "step": 12490,
      "training_step_time": 6.666335344314575
    },
    {
      "epoch": 1.1912345886230468e-05,
      "model_forward_time": 1.7608866691589355,
      "step": 12491
    },
    {
      "epoch": 1.1912345886230468e-05,
      "step": 12491,
      "training_step_time": 7.3799169063568115
    },
    {
      "epoch": 1.1913299560546876e-05,
      "model_forward_time": 1.7598412036895752,
      "step": 12492
    },
    {
      "epoch": 1.1913299560546876e-05,
      "step": 12492,
      "training_step_time": 7.915571928024292
    },
    {
      "epoch": 1.1914253234863282e-05,
      "model_forward_time": 1.768357515335083,
      "step": 12493
    },
    {
      "epoch": 1.1914253234863282e-05,
      "step": 12493,
      "training_step_time": 8.631268978118896
    },
    {
      "epoch": 1.1915206909179687e-05,
      "model_forward_time": 1.76725435256958,
      "step": 12494
    },
    {
      "epoch": 1.1915206909179687e-05,
      "step": 12494,
      "training_step_time": 8.85489273071289
    },
    {
      "epoch": 1.1916160583496093e-05,
      "model_forward_time": 2.943995952606201,
      "step": 12495
    },
    {
      "epoch": 1.1916160583496093e-05,
      "step": 12495,
      "training_step_time": 8.759424209594727
    },
    {
      "epoch": 1.19171142578125e-05,
      "model_forward_time": 1.7691025733947754,
      "step": 12496
    },
    {
      "epoch": 1.19171142578125e-05,
      "step": 12496,
      "training_step_time": 7.630714416503906
    },
    {
      "epoch": 1.1918067932128906e-05,
      "model_forward_time": 1.7597761154174805,
      "step": 12497
    },
    {
      "epoch": 1.1918067932128906e-05,
      "step": 12497,
      "training_step_time": 7.555450916290283
    },
    {
      "epoch": 1.1919021606445312e-05,
      "model_forward_time": 1.7651424407958984,
      "step": 12498
    },
    {
      "epoch": 1.1919021606445312e-05,
      "step": 12498,
      "training_step_time": 7.041544675827026
    },
    {
      "epoch": 1.191997528076172e-05,
      "model_forward_time": 1.7624728679656982,
      "step": 12499
    },
    {
      "epoch": 1.191997528076172e-05,
      "step": 12499,
      "training_step_time": 6.784777879714966
    },
    {
      "epoch": 1.1920928955078125e-05,
      "grad_norm": 0.2089117467403412,
      "learning_rate": 9.847041917330708e-06,
      "loss": 0.0677,
      "step": 12500
    },
    {
      "epoch": 1.1920928955078125e-05,
      "model_forward_time": 1.7600970268249512,
      "step": 12500
    },
    {
      "dynamics_loss_avg": 0.041647890023887156,
      "epoch": 1.1920928955078125e-05,
      "step": 12500
    },
    {
      "action_loss_avg": 0.0028972180880373344,
      "epoch": 1.1920928955078125e-05,
      "step": 12500
    },
    {
      "epoch": 1.1920928955078125e-05,
      "step": 12500,
      "training_step_time": 7.46633243560791
    },
    {
      "epoch": 1.192188262939453e-05,
      "model_forward_time": 1.760831356048584,
      "step": 12501
    },
    {
      "epoch": 1.192188262939453e-05,
      "step": 12501,
      "training_step_time": 7.558732032775879
    },
    {
      "epoch": 1.1922836303710938e-05,
      "model_forward_time": 1.751889944076538,
      "step": 12502
    },
    {
      "epoch": 1.1922836303710938e-05,
      "step": 12502,
      "training_step_time": 7.558938264846802
    },
    {
      "epoch": 1.1923789978027344e-05,
      "model_forward_time": 1.7696263790130615,
      "step": 12503
    },
    {
      "epoch": 1.1923789978027344e-05,
      "step": 12503,
      "training_step_time": 9.38154911994934
    },
    {
      "epoch": 1.192474365234375e-05,
      "model_forward_time": 2.971971273422241,
      "step": 12504
    },
    {
      "epoch": 1.192474365234375e-05,
      "step": 12504,
      "training_step_time": 8.676959037780762
    },
    {
      "epoch": 1.1925697326660157e-05,
      "model_forward_time": 1.7584877014160156,
      "step": 12505
    },
    {
      "epoch": 1.1925697326660157e-05,
      "step": 12505,
      "training_step_time": 9.173526525497437
    },
    {
      "epoch": 1.1926651000976563e-05,
      "model_forward_time": 1.7566008567810059,
      "step": 12506
    },
    {
      "epoch": 1.1926651000976563e-05,
      "step": 12506,
      "training_step_time": 7.7854530811309814
    },
    {
      "epoch": 1.1927604675292968e-05,
      "model_forward_time": 1.764843225479126,
      "step": 12507
    },
    {
      "epoch": 1.1927604675292968e-05,
      "step": 12507,
      "training_step_time": 7.239984750747681
    },
    {
      "epoch": 1.1928558349609374e-05,
      "model_forward_time": 1.76149582862854,
      "step": 12508
    },
    {
      "epoch": 1.1928558349609374e-05,
      "step": 12508,
      "training_step_time": 6.816624879837036
    },
    {
      "epoch": 1.1929512023925782e-05,
      "model_forward_time": 1.7568609714508057,
      "step": 12509
    },
    {
      "epoch": 1.1929512023925782e-05,
      "step": 12509,
      "training_step_time": 7.540499925613403
    },
    {
      "epoch": 1.1930465698242187e-05,
      "grad_norm": 0.18509072065353394,
      "learning_rate": 9.846635802468132e-06,
      "loss": 0.0743,
      "step": 12510
    },
    {
      "epoch": 1.1930465698242187e-05,
      "model_forward_time": 1.7621450424194336,
      "step": 12510
    },
    {
      "dynamics_loss_avg": 0.07007400393486023,
      "epoch": 1.1930465698242187e-05,
      "step": 12510
    },
    {
      "action_loss_avg": 0.007014780538156629,
      "epoch": 1.1930465698242187e-05,
      "step": 12510
    },
    {
      "epoch": 1.1930465698242187e-05,
      "step": 12510,
      "training_step_time": 7.574886083602905
    },
    {
      "epoch": 1.1931419372558593e-05,
      "model_forward_time": 1.7612054347991943,
      "step": 12511
    },
    {
      "epoch": 1.1931419372558593e-05,
      "step": 12511,
      "training_step_time": 7.641158103942871
    },
    {
      "epoch": 1.1932373046875e-05,
      "model_forward_time": 1.7576615810394287,
      "step": 12512
    },
    {
      "epoch": 1.1932373046875e-05,
      "step": 12512,
      "training_step_time": 9.545780181884766
    },
    {
      "epoch": 1.1933326721191406e-05,
      "model_forward_time": 2.9355785846710205,
      "step": 12513
    },
    {
      "epoch": 1.1933326721191406e-05,
      "step": 12513,
      "training_step_time": 11.02315616607666
    },
    {
      "epoch": 1.1934280395507812e-05,
      "model_forward_time": 1.7613472938537598,
      "step": 12514
    },
    {
      "epoch": 1.1934280395507812e-05,
      "step": 12514,
      "training_step_time": 9.539917945861816
    },
    {
      "epoch": 1.193523406982422e-05,
      "model_forward_time": 1.7630562782287598,
      "step": 12515
    },
    {
      "epoch": 1.193523406982422e-05,
      "step": 12515,
      "training_step_time": 7.71612811088562
    },
    {
      "epoch": 1.1936187744140625e-05,
      "model_forward_time": 1.7592625617980957,
      "step": 12516
    },
    {
      "epoch": 1.1936187744140625e-05,
      "step": 12516,
      "training_step_time": 7.209450721740723
    },
    {
      "epoch": 1.193714141845703e-05,
      "model_forward_time": 1.7550246715545654,
      "step": 12517
    },
    {
      "epoch": 1.193714141845703e-05,
      "step": 12517,
      "training_step_time": 6.802699327468872
    },
    {
      "epoch": 1.1938095092773438e-05,
      "model_forward_time": 1.7403695583343506,
      "step": 12518
    },
    {
      "epoch": 1.1938095092773438e-05,
      "step": 12518,
      "training_step_time": 7.379180908203125
    },
    {
      "epoch": 1.1939048767089844e-05,
      "model_forward_time": 1.7584762573242188,
      "step": 12519
    },
    {
      "epoch": 1.1939048767089844e-05,
      "step": 12519,
      "training_step_time": 7.526155948638916
    },
    {
      "epoch": 1.194000244140625e-05,
      "grad_norm": 0.17690251767635345,
      "learning_rate": 9.84622915758475e-06,
      "loss": 0.0661,
      "step": 12520
    },
    {
      "epoch": 1.194000244140625e-05,
      "model_forward_time": 1.7536063194274902,
      "step": 12520
    },
    {
      "dynamics_loss_avg": 0.06120514404028654,
      "epoch": 1.194000244140625e-05,
      "step": 12520
    },
    {
      "action_loss_avg": 0.0036588217888493093,
      "epoch": 1.194000244140625e-05,
      "step": 12520
    },
    {
      "epoch": 1.194000244140625e-05,
      "step": 12520,
      "training_step_time": 7.588113307952881
    },
    {
      "epoch": 1.1940956115722657e-05,
      "model_forward_time": 1.7634727954864502,
      "step": 12521
    },
    {
      "epoch": 1.1940956115722657e-05,
      "step": 12521,
      "training_step_time": 9.171459197998047
    },
    {
      "epoch": 1.1941909790039063e-05,
      "model_forward_time": 1.7999460697174072,
      "step": 12522
    },
    {
      "epoch": 1.1941909790039063e-05,
      "step": 12522,
      "training_step_time": 8.742823839187622
    },
    {
      "epoch": 1.1942863464355468e-05,
      "model_forward_time": 1.7533178329467773,
      "step": 12523
    },
    {
      "epoch": 1.1942863464355468e-05,
      "step": 12523,
      "training_step_time": 9.704107284545898
    },
    {
      "epoch": 1.1943817138671874e-05,
      "model_forward_time": 1.7576861381530762,
      "step": 12524
    },
    {
      "epoch": 1.1943817138671874e-05,
      "step": 12524,
      "training_step_time": 8.419852018356323
    },
    {
      "epoch": 1.1944770812988282e-05,
      "model_forward_time": 1.75811767578125,
      "step": 12525
    },
    {
      "epoch": 1.1944770812988282e-05,
      "step": 12525,
      "training_step_time": 7.269927978515625
    },
    {
      "epoch": 1.1945724487304687e-05,
      "model_forward_time": 1.7719738483428955,
      "step": 12526
    },
    {
      "epoch": 1.1945724487304687e-05,
      "step": 12526,
      "training_step_time": 6.792827844619751
    },
    {
      "epoch": 1.1946678161621093e-05,
      "model_forward_time": 1.7471628189086914,
      "step": 12527
    },
    {
      "epoch": 1.1946678161621093e-05,
      "step": 12527,
      "training_step_time": 7.503998279571533
    },
    {
      "epoch": 1.19476318359375e-05,
      "model_forward_time": 1.7629344463348389,
      "step": 12528
    },
    {
      "epoch": 1.19476318359375e-05,
      "step": 12528,
      "training_step_time": 7.4866554737091064
    },
    {
      "epoch": 1.1948585510253906e-05,
      "model_forward_time": 1.7548091411590576,
      "step": 12529
    },
    {
      "epoch": 1.1948585510253906e-05,
      "step": 12529,
      "training_step_time": 7.593147277832031
    },
    {
      "epoch": 1.1949539184570312e-05,
      "grad_norm": 0.15337716042995453,
      "learning_rate": 9.845821982725034e-06,
      "loss": 0.0677,
      "step": 12530
    },
    {
      "epoch": 1.1949539184570312e-05,
      "model_forward_time": 1.7573254108428955,
      "step": 12530
    },
    {
      "dynamics_loss_avg": 0.05127346776425838,
      "epoch": 1.1949539184570312e-05,
      "step": 12530
    },
    {
      "action_loss_avg": 0.0031557968643028287,
      "epoch": 1.1949539184570312e-05,
      "step": 12530
    },
    {
      "epoch": 1.1949539184570312e-05,
      "step": 12530,
      "training_step_time": 9.249759674072266
    },
    {
      "epoch": 1.195049285888672e-05,
      "model_forward_time": 1.7264513969421387,
      "step": 12531
    },
    {
      "epoch": 1.195049285888672e-05,
      "step": 12531,
      "training_step_time": 8.67953372001648
    },
    {
      "epoch": 1.1951446533203125e-05,
      "model_forward_time": 2.909594774246216,
      "step": 12532
    },
    {
      "epoch": 1.1951446533203125e-05,
      "step": 12532,
      "training_step_time": 8.682321786880493
    },
    {
      "epoch": 1.195240020751953e-05,
      "model_forward_time": 1.7463769912719727,
      "step": 12533
    },
    {
      "epoch": 1.195240020751953e-05,
      "step": 12533,
      "training_step_time": 7.600522756576538
    },
    {
      "epoch": 1.1953353881835938e-05,
      "model_forward_time": 1.8298494815826416,
      "step": 12534
    },
    {
      "epoch": 1.1953353881835938e-05,
      "step": 12534,
      "training_step_time": 7.603062391281128
    },
    {
      "epoch": 1.1954307556152344e-05,
      "model_forward_time": 1.7489373683929443,
      "step": 12535
    },
    {
      "epoch": 1.1954307556152344e-05,
      "step": 12535,
      "training_step_time": 7.123299837112427
    },
    {
      "epoch": 1.195526123046875e-05,
      "model_forward_time": 1.7465364933013916,
      "step": 12536
    },
    {
      "epoch": 1.195526123046875e-05,
      "step": 12536,
      "training_step_time": 7.4730682373046875
    },
    {
      "epoch": 1.1956214904785157e-05,
      "model_forward_time": 1.7510054111480713,
      "step": 12537
    },
    {
      "epoch": 1.1956214904785157e-05,
      "step": 12537,
      "training_step_time": 7.177171945571899
    },
    {
      "epoch": 1.1957168579101563e-05,
      "model_forward_time": 1.7482950687408447,
      "step": 12538
    },
    {
      "epoch": 1.1957168579101563e-05,
      "step": 12538,
      "training_step_time": 7.499536514282227
    },
    {
      "epoch": 1.1958122253417968e-05,
      "model_forward_time": 1.746389389038086,
      "step": 12539
    },
    {
      "epoch": 1.1958122253417968e-05,
      "step": 12539,
      "training_step_time": 8.342808246612549
    },
    {
      "epoch": 1.1959075927734374e-05,
      "grad_norm": 0.2266114056110382,
      "learning_rate": 9.845414277933513e-06,
      "loss": 0.0692,
      "step": 12540
    },
    {
      "epoch": 1.1959075927734374e-05,
      "model_forward_time": 1.739171028137207,
      "step": 12540
    },
    {
      "dynamics_loss_avg": 0.06272514145821333,
      "epoch": 1.1959075927734374e-05,
      "step": 12540
    },
    {
      "action_loss_avg": 0.0028526386595331132,
      "epoch": 1.1959075927734374e-05,
      "step": 12540
    },
    {
      "epoch": 1.1959075927734374e-05,
      "step": 12540,
      "training_step_time": 9.142944812774658
    },
    {
      "epoch": 1.1960029602050782e-05,
      "model_forward_time": 2.8715319633483887,
      "step": 12541
    },
    {
      "epoch": 1.1960029602050782e-05,
      "step": 12541,
      "training_step_time": 8.222537994384766
    },
    {
      "epoch": 1.1960983276367187e-05,
      "model_forward_time": 1.751131296157837,
      "step": 12542
    },
    {
      "epoch": 1.1960983276367187e-05,
      "step": 12542,
      "training_step_time": 7.6983323097229
    },
    {
      "epoch": 1.1961936950683593e-05,
      "model_forward_time": 1.3859519958496094,
      "step": 12543
    },
    {
      "epoch": 1.1961936950683593e-05,
      "step": 12543,
      "training_step_time": 7.646489858627319
    },
    {
      "epoch": 1.1962890625e-05,
      "model_forward_time": 1.7514443397521973,
      "step": 12544
    },
    {
      "epoch": 1.1962890625e-05,
      "step": 12544,
      "training_step_time": 7.082740306854248
    },
    {
      "epoch": 1.1963844299316406e-05,
      "model_forward_time": 1.7468199729919434,
      "step": 12545
    },
    {
      "epoch": 1.1963844299316406e-05,
      "step": 12545,
      "training_step_time": 7.3082075119018555
    },
    {
      "epoch": 1.1964797973632812e-05,
      "model_forward_time": 1.7524523735046387,
      "step": 12546
    },
    {
      "epoch": 1.1964797973632812e-05,
      "step": 12546,
      "training_step_time": 6.744187116622925
    },
    {
      "epoch": 1.196575164794922e-05,
      "model_forward_time": 1.744666337966919,
      "step": 12547
    },
    {
      "epoch": 1.196575164794922e-05,
      "step": 12547,
      "training_step_time": 7.515242338180542
    },
    {
      "epoch": 1.1966705322265625e-05,
      "model_forward_time": 1.747861623764038,
      "step": 12548
    },
    {
      "epoch": 1.1966705322265625e-05,
      "step": 12548,
      "training_step_time": 7.683581590652466
    },
    {
      "epoch": 1.196765899658203e-05,
      "model_forward_time": 1.7772529125213623,
      "step": 12549
    },
    {
      "epoch": 1.196765899658203e-05,
      "step": 12549,
      "training_step_time": 8.738449335098267
    },
    {
      "epoch": 1.1968612670898438e-05,
      "grad_norm": 0.20190346240997314,
      "learning_rate": 9.84500604325477e-06,
      "loss": 0.0735,
      "step": 12550
    },
    {
      "epoch": 1.1968612670898438e-05,
      "model_forward_time": 2.9063403606414795,
      "step": 12550
    },
    {
      "dynamics_loss_avg": 0.06650072671473026,
      "epoch": 1.1968612670898438e-05,
      "step": 12550
    },
    {
      "action_loss_avg": 0.0036334024800453333,
      "epoch": 1.1968612670898438e-05,
      "step": 12550
    },
    {
      "epoch": 1.1968612670898438e-05,
      "step": 12550,
      "training_step_time": 7.6287713050842285
    },
    {
      "epoch": 1.1969566345214844e-05,
      "model_forward_time": 1.7551417350769043,
      "step": 12551
    },
    {
      "epoch": 1.1969566345214844e-05,
      "step": 12551,
      "training_step_time": 7.667937994003296
    },
    {
      "epoch": 1.197052001953125e-05,
      "model_forward_time": 1.7459659576416016,
      "step": 12552
    },
    {
      "epoch": 1.197052001953125e-05,
      "step": 12552,
      "training_step_time": 7.593358993530273
    },
    {
      "epoch": 1.1971473693847657e-05,
      "model_forward_time": 1.7462518215179443,
      "step": 12553
    },
    {
      "epoch": 1.1971473693847657e-05,
      "step": 12553,
      "training_step_time": 7.5145368576049805
    },
    {
      "epoch": 1.1972427368164063e-05,
      "model_forward_time": 1.7609870433807373,
      "step": 12554
    },
    {
      "epoch": 1.1972427368164063e-05,
      "step": 12554,
      "training_step_time": 7.03637170791626
    },
    {
      "epoch": 1.1973381042480468e-05,
      "model_forward_time": 1.7605080604553223,
      "step": 12555
    },
    {
      "epoch": 1.1973381042480468e-05,
      "step": 12555,
      "training_step_time": 7.349132061004639
    },
    {
      "epoch": 1.1974334716796874e-05,
      "model_forward_time": 1.756255865097046,
      "step": 12556
    },
    {
      "epoch": 1.1974334716796874e-05,
      "step": 12556,
      "training_step_time": 7.52793288230896
    },
    {
      "epoch": 1.1975288391113282e-05,
      "model_forward_time": 1.768254041671753,
      "step": 12557
    },
    {
      "epoch": 1.1975288391113282e-05,
      "step": 12557,
      "training_step_time": 8.89451551437378
    },
    {
      "epoch": 1.1976242065429687e-05,
      "model_forward_time": 1.7583374977111816,
      "step": 12558
    },
    {
      "epoch": 1.1976242065429687e-05,
      "step": 12558,
      "training_step_time": 8.784501314163208
    },
    {
      "epoch": 1.1977195739746093e-05,
      "model_forward_time": 2.8496358394622803,
      "step": 12559
    },
    {
      "epoch": 1.1977195739746093e-05,
      "step": 12559,
      "training_step_time": 8.595805168151855
    },
    {
      "epoch": 1.19781494140625e-05,
      "grad_norm": 0.15993139147758484,
      "learning_rate": 9.844597278733453e-06,
      "loss": 0.0725,
      "step": 12560
    },
    {
      "epoch": 1.19781494140625e-05,
      "model_forward_time": 1.8350043296813965,
      "step": 12560
    },
    {
      "dynamics_loss_avg": 0.04412644151598215,
      "epoch": 1.19781494140625e-05,
      "step": 12560
    },
    {
      "action_loss_avg": 0.0027827298152260484,
      "epoch": 1.19781494140625e-05,
      "step": 12560
    },
    {
      "epoch": 1.19781494140625e-05,
      "step": 12560,
      "training_step_time": 7.676222085952759
    },
    {
      "epoch": 1.1979103088378906e-05,
      "model_forward_time": 1.759803295135498,
      "step": 12561
    },
    {
      "epoch": 1.1979103088378906e-05,
      "step": 12561,
      "training_step_time": 7.675445556640625
    },
    {
      "epoch": 1.1980056762695312e-05,
      "model_forward_time": 1.751671314239502,
      "step": 12562
    },
    {
      "epoch": 1.1980056762695312e-05,
      "step": 12562,
      "training_step_time": 7.690720558166504
    },
    {
      "epoch": 1.198101043701172e-05,
      "model_forward_time": 1.7623224258422852,
      "step": 12563
    },
    {
      "epoch": 1.198101043701172e-05,
      "step": 12563,
      "training_step_time": 7.1457765102386475
    },
    {
      "epoch": 1.1981964111328125e-05,
      "model_forward_time": 1.398787498474121,
      "step": 12564
    },
    {
      "epoch": 1.1981964111328125e-05,
      "step": 12564,
      "training_step_time": 7.499901533126831
    },
    {
      "epoch": 1.198291778564453e-05,
      "model_forward_time": 1.755769968032837,
      "step": 12565
    },
    {
      "epoch": 1.198291778564453e-05,
      "step": 12565,
      "training_step_time": 7.71375298500061
    },
    {
      "epoch": 1.1983871459960938e-05,
      "model_forward_time": 1.7556219100952148,
      "step": 12566
    },
    {
      "epoch": 1.1983871459960938e-05,
      "step": 12566,
      "training_step_time": 7.658299684524536
    },
    {
      "epoch": 1.1984825134277344e-05,
      "model_forward_time": 1.76906156539917,
      "step": 12567
    },
    {
      "epoch": 1.1984825134277344e-05,
      "step": 12567,
      "training_step_time": 8.767088651657104
    },
    {
      "epoch": 1.198577880859375e-05,
      "model_forward_time": 1.761000394821167,
      "step": 12568
    },
    {
      "epoch": 1.198577880859375e-05,
      "step": 12568,
      "training_step_time": 8.732874870300293
    },
    {
      "epoch": 1.1986732482910157e-05,
      "model_forward_time": 2.9148426055908203,
      "step": 12569
    },
    {
      "epoch": 1.1986732482910157e-05,
      "step": 12569,
      "training_step_time": 8.611368417739868
    },
    {
      "epoch": 1.1987686157226563e-05,
      "grad_norm": 0.16058441996574402,
      "learning_rate": 9.844187984414259e-06,
      "loss": 0.072,
      "step": 12570
    },
    {
      "epoch": 1.1987686157226563e-05,
      "model_forward_time": 1.7636711597442627,
      "step": 12570
    },
    {
      "dynamics_loss_avg": 0.06204352490603924,
      "epoch": 1.1987686157226563e-05,
      "step": 12570
    },
    {
      "action_loss_avg": 0.003353638941189274,
      "epoch": 1.1987686157226563e-05,
      "step": 12570
    },
    {
      "epoch": 1.1987686157226563e-05,
      "step": 12570,
      "training_step_time": 7.635037422180176
    },
    {
      "epoch": 1.1988639831542968e-05,
      "model_forward_time": 1.756544589996338,
      "step": 12571
    },
    {
      "epoch": 1.1988639831542968e-05,
      "step": 12571,
      "training_step_time": 7.5887205600738525
    },
    {
      "epoch": 1.1989593505859374e-05,
      "model_forward_time": 1.7574756145477295,
      "step": 12572
    },
    {
      "epoch": 1.1989593505859374e-05,
      "step": 12572,
      "training_step_time": 7.038824558258057
    },
    {
      "epoch": 1.1990547180175782e-05,
      "model_forward_time": 1.7564139366149902,
      "step": 12573
    },
    {
      "epoch": 1.1990547180175782e-05,
      "step": 12573,
      "training_step_time": 7.411956071853638
    },
    {
      "epoch": 1.1991500854492187e-05,
      "model_forward_time": 1.7695586681365967,
      "step": 12574
    },
    {
      "epoch": 1.1991500854492187e-05,
      "step": 12574,
      "training_step_time": 8.347710609436035
    },
    {
      "epoch": 1.1992454528808593e-05,
      "model_forward_time": 1.7562518119812012,
      "step": 12575
    },
    {
      "epoch": 1.1992454528808593e-05,
      "step": 12575,
      "training_step_time": 7.946026086807251
    },
    {
      "epoch": 1.1993408203125e-05,
      "model_forward_time": 1.7560276985168457,
      "step": 12576
    },
    {
      "epoch": 1.1993408203125e-05,
      "step": 12576,
      "training_step_time": 8.903713703155518
    },
    {
      "epoch": 1.1994361877441406e-05,
      "model_forward_time": 1.7612240314483643,
      "step": 12577
    },
    {
      "epoch": 1.1994361877441406e-05,
      "step": 12577,
      "training_step_time": 8.492136001586914
    },
    {
      "epoch": 1.1995315551757812e-05,
      "model_forward_time": 2.924151659011841,
      "step": 12578
    },
    {
      "epoch": 1.1995315551757812e-05,
      "step": 12578,
      "training_step_time": 9.374135494232178
    },
    {
      "epoch": 1.199626922607422e-05,
      "model_forward_time": 1.755230188369751,
      "step": 12579
    },
    {
      "epoch": 1.199626922607422e-05,
      "step": 12579,
      "training_step_time": 7.554505825042725
    },
    {
      "epoch": 1.1997222900390625e-05,
      "grad_norm": 0.23808416724205017,
      "learning_rate": 9.84377816034195e-06,
      "loss": 0.0666,
      "step": 12580
    },
    {
      "epoch": 1.1997222900390625e-05,
      "model_forward_time": 1.7596814632415771,
      "step": 12580
    },
    {
      "dynamics_loss_avg": 0.05923077929764986,
      "epoch": 1.1997222900390625e-05,
      "step": 12580
    },
    {
      "action_loss_avg": 0.003770202212035656,
      "epoch": 1.1997222900390625e-05,
      "step": 12580
    },
    {
      "epoch": 1.1997222900390625e-05,
      "step": 12580,
      "training_step_time": 7.56850790977478
    },
    {
      "epoch": 1.199817657470703e-05,
      "model_forward_time": 1.4017915725708008,
      "step": 12581
    },
    {
      "epoch": 1.199817657470703e-05,
      "step": 12581,
      "training_step_time": 7.084092855453491
    },
    {
      "epoch": 1.1999130249023438e-05,
      "model_forward_time": 1.754624605178833,
      "step": 12582
    },
    {
      "epoch": 1.1999130249023438e-05,
      "step": 12582,
      "training_step_time": 7.3353166580200195
    },
    {
      "epoch": 1.2000083923339844e-05,
      "model_forward_time": 1.7617955207824707,
      "step": 12583
    },
    {
      "epoch": 1.2000083923339844e-05,
      "step": 12583,
      "training_step_time": 7.57220983505249
    },
    {
      "epoch": 1.200103759765625e-05,
      "model_forward_time": 1.7596371173858643,
      "step": 12584
    },
    {
      "epoch": 1.200103759765625e-05,
      "step": 12584,
      "training_step_time": 8.475210666656494
    },
    {
      "epoch": 1.2001991271972657e-05,
      "model_forward_time": 1.7585060596466064,
      "step": 12585
    },
    {
      "epoch": 1.2001991271972657e-05,
      "step": 12585,
      "training_step_time": 8.739428281784058
    },
    {
      "epoch": 1.2002944946289063e-05,
      "model_forward_time": 1.7555172443389893,
      "step": 12586
    },
    {
      "epoch": 1.2002944946289063e-05,
      "step": 12586,
      "training_step_time": 9.539110660552979
    },
    {
      "epoch": 1.2003898620605469e-05,
      "model_forward_time": 2.9552078247070312,
      "step": 12587
    },
    {
      "epoch": 1.2003898620605469e-05,
      "step": 12587,
      "training_step_time": 8.242121696472168
    },
    {
      "epoch": 1.2004852294921874e-05,
      "model_forward_time": 1.7629072666168213,
      "step": 12588
    },
    {
      "epoch": 1.2004852294921874e-05,
      "step": 12588,
      "training_step_time": 7.574910640716553
    },
    {
      "epoch": 1.2005805969238282e-05,
      "model_forward_time": 1.786447525024414,
      "step": 12589
    },
    {
      "epoch": 1.2005805969238282e-05,
      "step": 12589,
      "training_step_time": 8.669464588165283
    },
    {
      "epoch": 1.2006759643554687e-05,
      "grad_norm": 0.22435466945171356,
      "learning_rate": 9.843367806561345e-06,
      "loss": 0.0649,
      "step": 12590
    },
    {
      "epoch": 1.2006759643554687e-05,
      "model_forward_time": 1.7685339450836182,
      "step": 12590
    },
    {
      "dynamics_loss_avg": 0.06066578160971403,
      "epoch": 1.2006759643554687e-05,
      "step": 12590
    },
    {
      "action_loss_avg": 0.004661713831592351,
      "epoch": 1.2006759643554687e-05,
      "step": 12590
    },
    {
      "epoch": 1.2006759643554687e-05,
      "step": 12590,
      "training_step_time": 7.579314708709717
    },
    {
      "epoch": 1.2007713317871093e-05,
      "model_forward_time": 1.8348350524902344,
      "step": 12591
    },
    {
      "epoch": 1.2007713317871093e-05,
      "step": 12591,
      "training_step_time": 6.716973304748535
    },
    {
      "epoch": 1.20086669921875e-05,
      "model_forward_time": 1.7587013244628906,
      "step": 12592
    },
    {
      "epoch": 1.20086669921875e-05,
      "step": 12592,
      "training_step_time": 7.546257257461548
    },
    {
      "epoch": 1.2009620666503906e-05,
      "model_forward_time": 1.7579646110534668,
      "step": 12593
    },
    {
      "epoch": 1.2009620666503906e-05,
      "step": 12593,
      "training_step_time": 8.165300607681274
    },
    {
      "epoch": 1.2010574340820312e-05,
      "model_forward_time": 1.7648837566375732,
      "step": 12594
    },
    {
      "epoch": 1.2010574340820312e-05,
      "step": 12594,
      "training_step_time": 7.486977577209473
    },
    {
      "epoch": 1.201152801513672e-05,
      "model_forward_time": 1.7614295482635498,
      "step": 12595
    },
    {
      "epoch": 1.201152801513672e-05,
      "step": 12595,
      "training_step_time": 7.8933634757995605
    },
    {
      "epoch": 1.2012481689453125e-05,
      "model_forward_time": 2.844043731689453,
      "step": 12596
    },
    {
      "epoch": 1.2012481689453125e-05,
      "step": 12596,
      "training_step_time": 8.616358041763306
    },
    {
      "epoch": 1.201343536376953e-05,
      "model_forward_time": 1.815314531326294,
      "step": 12597
    },
    {
      "epoch": 1.201343536376953e-05,
      "step": 12597,
      "training_step_time": 8.332742691040039
    },
    {
      "epoch": 1.2014389038085938e-05,
      "model_forward_time": 1.7678680419921875,
      "step": 12598
    },
    {
      "epoch": 1.2014389038085938e-05,
      "step": 12598,
      "training_step_time": 7.782164573669434
    },
    {
      "epoch": 1.2015342712402344e-05,
      "model_forward_time": 1.7544498443603516,
      "step": 12599
    },
    {
      "epoch": 1.2015342712402344e-05,
      "step": 12599,
      "training_step_time": 7.568491220474243
    },
    {
      "epoch": 1.201629638671875e-05,
      "grad_norm": 0.21147672832012177,
      "learning_rate": 9.842956923117318e-06,
      "loss": 0.0684,
      "step": 12600
    },
    {
      "epoch": 1.201629638671875e-05,
      "model_forward_time": 1.764230489730835,
      "step": 12600
    },
    {
      "dynamics_loss_avg": 0.07031995318830013,
      "epoch": 1.201629638671875e-05,
      "step": 12600
    },
    {
      "action_loss_avg": 0.00454802846943494,
      "epoch": 1.201629638671875e-05,
      "step": 12600
    },
    {
      "epoch": 1.201629638671875e-05,
      "step": 12600,
      "training_step_time": 6.89384388923645
    },
    {
      "epoch": 1.2017250061035157e-05,
      "model_forward_time": 1.73846435546875,
      "step": 12601
    },
    {
      "epoch": 1.2017250061035157e-05,
      "step": 12601,
      "training_step_time": 7.671404838562012
    },
    {
      "epoch": 1.2018203735351563e-05,
      "model_forward_time": 1.7614705562591553,
      "step": 12602
    },
    {
      "epoch": 1.2018203735351563e-05,
      "step": 12602,
      "training_step_time": 9.33688473701477
    },
    {
      "epoch": 1.2019157409667969e-05,
      "model_forward_time": 1.7532191276550293,
      "step": 12603
    },
    {
      "epoch": 1.2019157409667969e-05,
      "step": 12603,
      "training_step_time": 10.04827070236206
    },
    {
      "epoch": 1.2020111083984374e-05,
      "model_forward_time": 1.7523667812347412,
      "step": 12604
    },
    {
      "epoch": 1.2020111083984374e-05,
      "step": 12604,
      "training_step_time": 10.674973487854004
    },
    {
      "epoch": 1.2021064758300782e-05,
      "model_forward_time": 1.765122890472412,
      "step": 12605
    },
    {
      "epoch": 1.2021064758300782e-05,
      "step": 12605,
      "training_step_time": 7.865415334701538
    },
    {
      "epoch": 1.2022018432617187e-05,
      "model_forward_time": 2.905371904373169,
      "step": 12606
    },
    {
      "epoch": 1.2022018432617187e-05,
      "step": 12606,
      "training_step_time": 8.862656831741333
    },
    {
      "epoch": 1.2022972106933593e-05,
      "model_forward_time": 1.7521357536315918,
      "step": 12607
    },
    {
      "epoch": 1.2022972106933593e-05,
      "step": 12607,
      "training_step_time": 7.725311994552612
    },
    {
      "epoch": 1.202392578125e-05,
      "model_forward_time": 1.737072229385376,
      "step": 12608
    },
    {
      "epoch": 1.202392578125e-05,
      "step": 12608,
      "training_step_time": 7.866476774215698
    },
    {
      "epoch": 1.2024879455566406e-05,
      "model_forward_time": 1.390995979309082,
      "step": 12609
    },
    {
      "epoch": 1.2024879455566406e-05,
      "step": 12609,
      "training_step_time": 6.868943452835083
    },
    {
      "epoch": 1.2025833129882812e-05,
      "grad_norm": 0.20776674151420593,
      "learning_rate": 9.842545510054803e-06,
      "loss": 0.0657,
      "step": 12610
    },
    {
      "epoch": 1.2025833129882812e-05,
      "model_forward_time": 1.767045259475708,
      "step": 12610
    },
    {
      "dynamics_loss_avg": 0.06312718763947486,
      "epoch": 1.2025833129882812e-05,
      "step": 12610
    },
    {
      "action_loss_avg": 0.0028253007563762366,
      "epoch": 1.2025833129882812e-05,
      "step": 12610
    },
    {
      "epoch": 1.2025833129882812e-05,
      "step": 12610,
      "training_step_time": 7.62598729133606
    },
    {
      "epoch": 1.202678680419922e-05,
      "model_forward_time": 1.75270676612854,
      "step": 12611
    },
    {
      "epoch": 1.202678680419922e-05,
      "step": 12611,
      "training_step_time": 7.709892749786377
    },
    {
      "epoch": 1.2027740478515625e-05,
      "model_forward_time": 1.7585930824279785,
      "step": 12612
    },
    {
      "epoch": 1.2027740478515625e-05,
      "step": 12612,
      "training_step_time": 9.289568424224854
    },
    {
      "epoch": 1.202869415283203e-05,
      "model_forward_time": 1.7458264827728271,
      "step": 12613
    },
    {
      "epoch": 1.202869415283203e-05,
      "step": 12613,
      "training_step_time": 8.046822309494019
    },
    {
      "epoch": 1.2029647827148438e-05,
      "model_forward_time": 1.7666997909545898,
      "step": 12614
    },
    {
      "epoch": 1.2029647827148438e-05,
      "step": 12614,
      "training_step_time": 8.695072412490845
    },
    {
      "epoch": 1.2030601501464844e-05,
      "model_forward_time": 2.5744175910949707,
      "step": 12615
    },
    {
      "epoch": 1.2030601501464844e-05,
      "step": 12615,
      "training_step_time": 9.66124677658081
    },
    {
      "epoch": 1.203155517578125e-05,
      "model_forward_time": 1.7553884983062744,
      "step": 12616
    },
    {
      "epoch": 1.203155517578125e-05,
      "step": 12616,
      "training_step_time": 9.111977577209473
    },
    {
      "epoch": 1.2032508850097657e-05,
      "model_forward_time": 1.7654192447662354,
      "step": 12617
    },
    {
      "epoch": 1.2032508850097657e-05,
      "step": 12617,
      "training_step_time": 7.839086294174194
    },
    {
      "epoch": 1.2033462524414063e-05,
      "model_forward_time": 1.7583096027374268,
      "step": 12618
    },
    {
      "epoch": 1.2033462524414063e-05,
      "step": 12618,
      "training_step_time": 6.870128154754639
    },
    {
      "epoch": 1.2034416198730469e-05,
      "model_forward_time": 1.7647178173065186,
      "step": 12619
    },
    {
      "epoch": 1.2034416198730469e-05,
      "step": 12619,
      "training_step_time": 7.544552803039551
    },
    {
      "epoch": 1.2035369873046874e-05,
      "grad_norm": 0.25780460238456726,
      "learning_rate": 9.842133567418793e-06,
      "loss": 0.0731,
      "step": 12620
    },
    {
      "epoch": 1.2035369873046874e-05,
      "model_forward_time": 1.752145767211914,
      "step": 12620
    },
    {
      "dynamics_loss_avg": 0.055722613073885444,
      "epoch": 1.2035369873046874e-05,
      "step": 12620
    },
    {
      "action_loss_avg": 0.0043331222317647185,
      "epoch": 1.2035369873046874e-05,
      "step": 12620
    },
    {
      "epoch": 1.2035369873046874e-05,
      "step": 12620,
      "training_step_time": 7.7752366065979
    },
    {
      "epoch": 1.2036323547363282e-05,
      "model_forward_time": 1.7641303539276123,
      "step": 12621
    },
    {
      "epoch": 1.2036323547363282e-05,
      "step": 12621,
      "training_step_time": 7.570092678070068
    },
    {
      "epoch": 1.2037277221679687e-05,
      "model_forward_time": 1.7561562061309814,
      "step": 12622
    },
    {
      "epoch": 1.2037277221679687e-05,
      "step": 12622,
      "training_step_time": 7.939333438873291
    },
    {
      "epoch": 1.2038230895996093e-05,
      "model_forward_time": 1.7469184398651123,
      "step": 12623
    },
    {
      "epoch": 1.2038230895996093e-05,
      "step": 12623,
      "training_step_time": 7.792659044265747
    },
    {
      "epoch": 1.20391845703125e-05,
      "model_forward_time": 2.98350191116333,
      "step": 12624
    },
    {
      "epoch": 1.20391845703125e-05,
      "step": 12624,
      "training_step_time": 9.976383686065674
    },
    {
      "epoch": 1.2040138244628906e-05,
      "model_forward_time": 1.8419160842895508,
      "step": 12625
    },
    {
      "epoch": 1.2040138244628906e-05,
      "step": 12625,
      "training_step_time": 8.902271032333374
    },
    {
      "epoch": 1.2041091918945312e-05,
      "model_forward_time": 1.7542710304260254,
      "step": 12626
    },
    {
      "epoch": 1.2041091918945312e-05,
      "step": 12626,
      "training_step_time": 7.711157321929932
    },
    {
      "epoch": 1.204204559326172e-05,
      "model_forward_time": 1.75471830368042,
      "step": 12627
    },
    {
      "epoch": 1.204204559326172e-05,
      "step": 12627,
      "training_step_time": 6.93941855430603
    },
    {
      "epoch": 1.2042999267578125e-05,
      "model_forward_time": 1.7541003227233887,
      "step": 12628
    },
    {
      "epoch": 1.2042999267578125e-05,
      "step": 12628,
      "training_step_time": 7.532668828964233
    },
    {
      "epoch": 1.204395294189453e-05,
      "model_forward_time": 1.7559328079223633,
      "step": 12629
    },
    {
      "epoch": 1.204395294189453e-05,
      "step": 12629,
      "training_step_time": 7.730123519897461
    },
    {
      "epoch": 1.2044906616210938e-05,
      "grad_norm": 0.33249613642692566,
      "learning_rate": 9.841721095254334e-06,
      "loss": 0.074,
      "step": 12630
    },
    {
      "epoch": 1.2044906616210938e-05,
      "model_forward_time": 1.7551729679107666,
      "step": 12630
    },
    {
      "dynamics_loss_avg": 0.07788787744939327,
      "epoch": 1.2044906616210938e-05,
      "step": 12630
    },
    {
      "action_loss_avg": 0.006845673033967614,
      "epoch": 1.2044906616210938e-05,
      "step": 12630
    },
    {
      "epoch": 1.2044906616210938e-05,
      "step": 12630,
      "training_step_time": 7.7045934200286865
    },
    {
      "epoch": 1.2045860290527344e-05,
      "model_forward_time": 1.7548267841339111,
      "step": 12631
    },
    {
      "epoch": 1.2045860290527344e-05,
      "step": 12631,
      "training_step_time": 9.374259948730469
    },
    {
      "epoch": 1.204681396484375e-05,
      "model_forward_time": 1.7630717754364014,
      "step": 12632
    },
    {
      "epoch": 1.204681396484375e-05,
      "step": 12632,
      "training_step_time": 9.313413381576538
    },
    {
      "epoch": 1.2047767639160155e-05,
      "model_forward_time": 1.8289635181427002,
      "step": 12633
    },
    {
      "epoch": 1.2047767639160155e-05,
      "step": 12633,
      "training_step_time": 9.668757200241089
    },
    {
      "epoch": 1.2048721313476563e-05,
      "model_forward_time": 1.73994779586792,
      "step": 12634
    },
    {
      "epoch": 1.2048721313476563e-05,
      "step": 12634,
      "training_step_time": 8.489381790161133
    },
    {
      "epoch": 1.2049674987792969e-05,
      "model_forward_time": 1.7531273365020752,
      "step": 12635
    },
    {
      "epoch": 1.2049674987792969e-05,
      "step": 12635,
      "training_step_time": 8.374481678009033
    },
    {
      "epoch": 1.2050628662109374e-05,
      "model_forward_time": 1.8303883075714111,
      "step": 12636
    },
    {
      "epoch": 1.2050628662109374e-05,
      "step": 12636,
      "training_step_time": 7.718987464904785
    },
    {
      "epoch": 1.2051582336425782e-05,
      "model_forward_time": 1.7589001655578613,
      "step": 12637
    },
    {
      "epoch": 1.2051582336425782e-05,
      "step": 12637,
      "training_step_time": 6.9781317710876465
    },
    {
      "epoch": 1.2052536010742187e-05,
      "model_forward_time": 1.7607038021087646,
      "step": 12638
    },
    {
      "epoch": 1.2052536010742187e-05,
      "step": 12638,
      "training_step_time": 8.685878992080688
    },
    {
      "epoch": 1.2053489685058593e-05,
      "model_forward_time": 1.7610273361206055,
      "step": 12639
    },
    {
      "epoch": 1.2053489685058593e-05,
      "step": 12639,
      "training_step_time": 7.698333263397217
    },
    {
      "epoch": 1.2054443359375e-05,
      "grad_norm": 0.18341882526874542,
      "learning_rate": 9.841308093606536e-06,
      "loss": 0.0705,
      "step": 12640
    },
    {
      "epoch": 1.2054443359375e-05,
      "model_forward_time": 1.7483916282653809,
      "step": 12640
    },
    {
      "dynamics_loss_avg": 0.05477095106616616,
      "epoch": 1.2054443359375e-05,
      "step": 12640
    },
    {
      "action_loss_avg": 0.003160621481947601,
      "epoch": 1.2054443359375e-05,
      "step": 12640
    },
    {
      "epoch": 1.2054443359375e-05,
      "step": 12640,
      "training_step_time": 7.713560104370117
    },
    {
      "epoch": 1.2055397033691406e-05,
      "model_forward_time": 1.7580885887145996,
      "step": 12641
    },
    {
      "epoch": 1.2055397033691406e-05,
      "step": 12641,
      "training_step_time": 7.789440393447876
    },
    {
      "epoch": 1.2056350708007812e-05,
      "model_forward_time": 1.7879931926727295,
      "step": 12642
    },
    {
      "epoch": 1.2056350708007812e-05,
      "step": 12642,
      "training_step_time": 8.391000986099243
    },
    {
      "epoch": 1.205730438232422e-05,
      "model_forward_time": 2.922788143157959,
      "step": 12643
    },
    {
      "epoch": 1.205730438232422e-05,
      "step": 12643,
      "training_step_time": 8.851848602294922
    },
    {
      "epoch": 1.2058258056640625e-05,
      "model_forward_time": 1.7604100704193115,
      "step": 12644
    },
    {
      "epoch": 1.2058258056640625e-05,
      "step": 12644,
      "training_step_time": 8.521587610244751
    },
    {
      "epoch": 1.205921173095703e-05,
      "model_forward_time": 1.7531883716583252,
      "step": 12645
    },
    {
      "epoch": 1.205921173095703e-05,
      "step": 12645,
      "training_step_time": 7.8409647941589355
    },
    {
      "epoch": 1.2060165405273438e-05,
      "model_forward_time": 1.7571995258331299,
      "step": 12646
    },
    {
      "epoch": 1.2060165405273438e-05,
      "step": 12646,
      "training_step_time": 6.877938508987427
    },
    {
      "epoch": 1.2061119079589844e-05,
      "model_forward_time": 1.7564127445220947,
      "step": 12647
    },
    {
      "epoch": 1.2061119079589844e-05,
      "step": 12647,
      "training_step_time": 7.567424058914185
    },
    {
      "epoch": 1.206207275390625e-05,
      "model_forward_time": 1.7564196586608887,
      "step": 12648
    },
    {
      "epoch": 1.206207275390625e-05,
      "step": 12648,
      "training_step_time": 7.7281413078308105
    },
    {
      "epoch": 1.2063026428222655e-05,
      "model_forward_time": 1.754979133605957,
      "step": 12649
    },
    {
      "epoch": 1.2063026428222655e-05,
      "step": 12649,
      "training_step_time": 8.40247893333435
    },
    {
      "epoch": 1.2063980102539063e-05,
      "grad_norm": 0.1726803034543991,
      "learning_rate": 9.840894562520565e-06,
      "loss": 0.0677,
      "step": 12650
    },
    {
      "epoch": 1.2063980102539063e-05,
      "model_forward_time": 1.7598152160644531,
      "step": 12650
    },
    {
      "dynamics_loss_avg": 0.055381718277931216,
      "epoch": 1.2063980102539063e-05,
      "step": 12650
    },
    {
      "action_loss_avg": 0.0033960821048822252,
      "epoch": 1.2063980102539063e-05,
      "step": 12650
    },
    {
      "epoch": 1.2063980102539063e-05,
      "step": 12650,
      "training_step_time": 7.8889055252075195
    },
    {
      "epoch": 1.2064933776855469e-05,
      "model_forward_time": 1.751173734664917,
      "step": 12651
    },
    {
      "epoch": 1.2064933776855469e-05,
      "step": 12651,
      "training_step_time": 9.721474647521973
    },
    {
      "epoch": 1.2065887451171874e-05,
      "model_forward_time": 2.9022440910339355,
      "step": 12652
    },
    {
      "epoch": 1.2065887451171874e-05,
      "step": 12652,
      "training_step_time": 9.838626384735107
    },
    {
      "epoch": 1.2066841125488282e-05,
      "model_forward_time": 1.7570452690124512,
      "step": 12653
    },
    {
      "epoch": 1.2066841125488282e-05,
      "step": 12653,
      "training_step_time": 7.756011247634888
    },
    {
      "epoch": 1.2067794799804687e-05,
      "model_forward_time": 1.7567224502563477,
      "step": 12654
    },
    {
      "epoch": 1.2067794799804687e-05,
      "step": 12654,
      "training_step_time": 8.760170698165894
    },
    {
      "epoch": 1.2068748474121093e-05,
      "model_forward_time": 1.7541511058807373,
      "step": 12655
    },
    {
      "epoch": 1.2068748474121093e-05,
      "step": 12655,
      "training_step_time": 6.8888099193573
    },
    {
      "epoch": 1.20697021484375e-05,
      "model_forward_time": 1.7375459671020508,
      "step": 12656
    },
    {
      "epoch": 1.20697021484375e-05,
      "step": 12656,
      "training_step_time": 7.573391437530518
    },
    {
      "epoch": 1.2070655822753906e-05,
      "model_forward_time": 1.756152629852295,
      "step": 12657
    },
    {
      "epoch": 1.2070655822753906e-05,
      "step": 12657,
      "training_step_time": 7.678002119064331
    },
    {
      "epoch": 1.2071609497070312e-05,
      "model_forward_time": 1.7458064556121826,
      "step": 12658
    },
    {
      "epoch": 1.2071609497070312e-05,
      "step": 12658,
      "training_step_time": 7.696817398071289
    },
    {
      "epoch": 1.207256317138672e-05,
      "model_forward_time": 1.7457754611968994,
      "step": 12659
    },
    {
      "epoch": 1.207256317138672e-05,
      "step": 12659,
      "training_step_time": 7.918968677520752
    },
    {
      "epoch": 1.2073516845703125e-05,
      "grad_norm": 0.23698750138282776,
      "learning_rate": 9.84048050204164e-06,
      "loss": 0.0699,
      "step": 12660
    },
    {
      "epoch": 1.2073516845703125e-05,
      "model_forward_time": 1.7342875003814697,
      "step": 12660
    },
    {
      "dynamics_loss_avg": 0.07406506165862084,
      "epoch": 1.2073516845703125e-05,
      "step": 12660
    },
    {
      "action_loss_avg": 0.0034055838827043773,
      "epoch": 1.2073516845703125e-05,
      "step": 12660
    },
    {
      "epoch": 1.2073516845703125e-05,
      "step": 12660,
      "training_step_time": 7.743808269500732
    },
    {
      "epoch": 1.2074470520019531e-05,
      "model_forward_time": 2.934389591217041,
      "step": 12661
    },
    {
      "epoch": 1.2074470520019531e-05,
      "step": 12661,
      "training_step_time": 9.365968227386475
    },
    {
      "epoch": 1.2075424194335938e-05,
      "model_forward_time": 1.7414584159851074,
      "step": 12662
    },
    {
      "epoch": 1.2075424194335938e-05,
      "step": 12662,
      "training_step_time": 8.527967929840088
    },
    {
      "epoch": 1.2076377868652344e-05,
      "model_forward_time": 1.7482528686523438,
      "step": 12663
    },
    {
      "epoch": 1.2076377868652344e-05,
      "step": 12663,
      "training_step_time": 7.690185785293579
    },
    {
      "epoch": 1.207733154296875e-05,
      "model_forward_time": 1.7424376010894775,
      "step": 12664
    },
    {
      "epoch": 1.207733154296875e-05,
      "step": 12664,
      "training_step_time": 7.778980731964111
    },
    {
      "epoch": 1.2078285217285155e-05,
      "model_forward_time": 1.7469594478607178,
      "step": 12665
    },
    {
      "epoch": 1.2078285217285155e-05,
      "step": 12665,
      "training_step_time": 7.600624322891235
    },
    {
      "epoch": 1.2079238891601563e-05,
      "model_forward_time": 1.7440192699432373,
      "step": 12666
    },
    {
      "epoch": 1.2079238891601563e-05,
      "step": 12666,
      "training_step_time": 7.678951025009155
    },
    {
      "epoch": 1.2080192565917969e-05,
      "model_forward_time": 1.7480778694152832,
      "step": 12667
    },
    {
      "epoch": 1.2080192565917969e-05,
      "step": 12667,
      "training_step_time": 7.536104202270508
    },
    {
      "epoch": 1.2081146240234374e-05,
      "model_forward_time": 1.744781494140625,
      "step": 12668
    },
    {
      "epoch": 1.2081146240234374e-05,
      "step": 12668,
      "training_step_time": 7.70068883895874
    },
    {
      "epoch": 1.2082099914550782e-05,
      "model_forward_time": 1.7503225803375244,
      "step": 12669
    },
    {
      "epoch": 1.2082099914550782e-05,
      "step": 12669,
      "training_step_time": 8.850844621658325
    },
    {
      "epoch": 1.2083053588867187e-05,
      "grad_norm": 0.23086288571357727,
      "learning_rate": 9.840065912215048e-06,
      "loss": 0.068,
      "step": 12670
    },
    {
      "epoch": 1.2083053588867187e-05,
      "model_forward_time": 2.9164483547210693,
      "step": 12670
    },
    {
      "dynamics_loss_avg": 0.06697294916957616,
      "epoch": 1.2083053588867187e-05,
      "step": 12670
    },
    {
      "action_loss_avg": 0.0032797432388179002,
      "epoch": 1.2083053588867187e-05,
      "step": 12670
    },
    {
      "epoch": 1.2083053588867187e-05,
      "step": 12670,
      "training_step_time": 8.584903001785278
    },
    {
      "epoch": 1.2084007263183593e-05,
      "model_forward_time": 1.7428059577941895,
      "step": 12671
    },
    {
      "epoch": 1.2084007263183593e-05,
      "step": 12671,
      "training_step_time": 7.771739482879639
    },
    {
      "epoch": 1.20849609375e-05,
      "model_forward_time": 1.3727126121520996,
      "step": 12672
    },
    {
      "epoch": 1.20849609375e-05,
      "step": 12672,
      "training_step_time": 8.730349063873291
    },
    {
      "epoch": 1.2085914611816406e-05,
      "model_forward_time": 1.7513208389282227,
      "step": 12673
    },
    {
      "epoch": 1.2085914611816406e-05,
      "step": 12673,
      "training_step_time": 7.726091623306274
    },
    {
      "epoch": 1.2086868286132812e-05,
      "model_forward_time": 1.7414915561676025,
      "step": 12674
    },
    {
      "epoch": 1.2086868286132812e-05,
      "step": 12674,
      "training_step_time": 7.585792303085327
    },
    {
      "epoch": 1.208782196044922e-05,
      "model_forward_time": 1.7509348392486572,
      "step": 12675
    },
    {
      "epoch": 1.208782196044922e-05,
      "step": 12675,
      "training_step_time": 7.794694423675537
    },
    {
      "epoch": 1.2088775634765625e-05,
      "model_forward_time": 1.7446033954620361,
      "step": 12676
    },
    {
      "epoch": 1.2088775634765625e-05,
      "step": 12676,
      "training_step_time": 8.35373044013977
    },
    {
      "epoch": 1.2089729309082031e-05,
      "model_forward_time": 1.7432913780212402,
      "step": 12677
    },
    {
      "epoch": 1.2089729309082031e-05,
      "step": 12677,
      "training_step_time": 8.213785648345947
    },
    {
      "epoch": 1.2090682983398438e-05,
      "model_forward_time": 1.748981237411499,
      "step": 12678
    },
    {
      "epoch": 1.2090682983398438e-05,
      "step": 12678,
      "training_step_time": 7.894502401351929
    },
    {
      "epoch": 1.2091636657714844e-05,
      "model_forward_time": 1.8143393993377686,
      "step": 12679
    },
    {
      "epoch": 1.2091636657714844e-05,
      "step": 12679,
      "training_step_time": 7.758064270019531
    },
    {
      "epoch": 1.209259033203125e-05,
      "grad_norm": 0.3035185933113098,
      "learning_rate": 9.839650793086123e-06,
      "loss": 0.0678,
      "step": 12680
    },
    {
      "epoch": 1.209259033203125e-05,
      "model_forward_time": 1.725128412246704,
      "step": 12680
    },
    {
      "dynamics_loss_avg": 0.04861239362508059,
      "epoch": 1.209259033203125e-05,
      "step": 12680
    },
    {
      "action_loss_avg": 0.003711037500761449,
      "epoch": 1.209259033203125e-05,
      "step": 12680
    },
    {
      "epoch": 1.209259033203125e-05,
      "step": 12680,
      "training_step_time": 7.718914985656738
    },
    {
      "epoch": 1.2093544006347655e-05,
      "model_forward_time": 1.7281208038330078,
      "step": 12681
    },
    {
      "epoch": 1.2093544006347655e-05,
      "step": 12681,
      "training_step_time": 8.415460109710693
    },
    {
      "epoch": 1.2094497680664063e-05,
      "model_forward_time": 1.7510528564453125,
      "step": 12682
    },
    {
      "epoch": 1.2094497680664063e-05,
      "step": 12682,
      "training_step_time": 7.838039875030518
    },
    {
      "epoch": 1.2095451354980469e-05,
      "model_forward_time": 1.7309021949768066,
      "step": 12683
    },
    {
      "epoch": 1.2095451354980469e-05,
      "step": 12683,
      "training_step_time": 7.690428256988525
    },
    {
      "epoch": 1.2096405029296874e-05,
      "model_forward_time": 1.7659239768981934,
      "step": 12684
    },
    {
      "epoch": 1.2096405029296874e-05,
      "step": 12684,
      "training_step_time": 7.76061224937439
    },
    {
      "epoch": 1.2097358703613282e-05,
      "model_forward_time": 1.7995479106903076,
      "step": 12685
    },
    {
      "epoch": 1.2097358703613282e-05,
      "step": 12685,
      "training_step_time": 8.69768738746643
    },
    {
      "epoch": 1.2098312377929687e-05,
      "model_forward_time": 1.7634310722351074,
      "step": 12686
    },
    {
      "epoch": 1.2098312377929687e-05,
      "step": 12686,
      "training_step_time": 7.698540925979614
    },
    {
      "epoch": 1.2099266052246093e-05,
      "model_forward_time": 1.7594571113586426,
      "step": 12687
    },
    {
      "epoch": 1.2099266052246093e-05,
      "step": 12687,
      "training_step_time": 8.930556297302246
    },
    {
      "epoch": 1.21002197265625e-05,
      "model_forward_time": 1.8330776691436768,
      "step": 12688
    },
    {
      "epoch": 1.21002197265625e-05,
      "step": 12688,
      "training_step_time": 7.837185382843018
    },
    {
      "epoch": 1.2101173400878906e-05,
      "model_forward_time": 2.951866626739502,
      "step": 12689
    },
    {
      "epoch": 1.2101173400878906e-05,
      "step": 12689,
      "training_step_time": 8.919744729995728
    },
    {
      "epoch": 1.2102127075195312e-05,
      "grad_norm": 0.29602405428886414,
      "learning_rate": 9.839235144700266e-06,
      "loss": 0.0698,
      "step": 12690
    },
    {
      "epoch": 1.2102127075195312e-05,
      "model_forward_time": 1.762223720550537,
      "step": 12690
    },
    {
      "dynamics_loss_avg": 0.05597331360913813,
      "epoch": 1.2102127075195312e-05,
      "step": 12690
    },
    {
      "action_loss_avg": 0.004141170909861103,
      "epoch": 1.2102127075195312e-05,
      "step": 12690
    },
    {
      "epoch": 1.2102127075195312e-05,
      "step": 12690,
      "training_step_time": 7.674032211303711
    },
    {
      "epoch": 1.210308074951172e-05,
      "model_forward_time": 1.7947611808776855,
      "step": 12691
    },
    {
      "epoch": 1.210308074951172e-05,
      "step": 12691,
      "training_step_time": 7.846680164337158
    },
    {
      "epoch": 1.2104034423828125e-05,
      "model_forward_time": 1.7474958896636963,
      "step": 12692
    },
    {
      "epoch": 1.2104034423828125e-05,
      "step": 12692,
      "training_step_time": 7.8188910484313965
    },
    {
      "epoch": 1.2104988098144531e-05,
      "model_forward_time": 1.7643063068389893,
      "step": 12693
    },
    {
      "epoch": 1.2104988098144531e-05,
      "step": 12693,
      "training_step_time": 7.689431667327881
    },
    {
      "epoch": 1.2105941772460938e-05,
      "model_forward_time": 1.7592580318450928,
      "step": 12694
    },
    {
      "epoch": 1.2105941772460938e-05,
      "step": 12694,
      "training_step_time": 7.79330039024353
    },
    {
      "epoch": 1.2106895446777344e-05,
      "model_forward_time": 1.754662036895752,
      "step": 12695
    },
    {
      "epoch": 1.2106895446777344e-05,
      "step": 12695,
      "training_step_time": 7.768558740615845
    },
    {
      "epoch": 1.210784912109375e-05,
      "model_forward_time": 1.7562875747680664,
      "step": 12696
    },
    {
      "epoch": 1.210784912109375e-05,
      "step": 12696,
      "training_step_time": 8.827493667602539
    },
    {
      "epoch": 1.2108802795410155e-05,
      "model_forward_time": 1.8192696571350098,
      "step": 12697
    },
    {
      "epoch": 1.2108802795410155e-05,
      "step": 12697,
      "training_step_time": 7.842772722244263
    },
    {
      "epoch": 1.2109756469726563e-05,
      "model_forward_time": 2.911595582962036,
      "step": 12698
    },
    {
      "epoch": 1.2109756469726563e-05,
      "step": 12698,
      "training_step_time": 8.863230228424072
    },
    {
      "epoch": 1.2110710144042969e-05,
      "model_forward_time": 1.8467023372650146,
      "step": 12699
    },
    {
      "epoch": 1.2110710144042969e-05,
      "step": 12699,
      "training_step_time": 9.270287275314331
    },
    {
      "epoch": 1.2111663818359374e-05,
      "grad_norm": 0.2167549729347229,
      "learning_rate": 9.838818967102926e-06,
      "loss": 0.0682,
      "step": 12700
    },
    {
      "epoch": 1.2111663818359374e-05,
      "model_forward_time": 1.7466726303100586,
      "step": 12700
    },
    {
      "dynamics_loss_avg": 0.05422339178621769,
      "epoch": 1.2111663818359374e-05,
      "step": 12700
    },
    {
      "action_loss_avg": 0.00330055242520757,
      "epoch": 1.2111663818359374e-05,
      "step": 12700
    },
    {
      "epoch": 1.2111663818359374e-05,
      "step": 12700,
      "training_step_time": 7.684501647949219
    },
    {
      "epoch": 1.2112617492675782e-05,
      "model_forward_time": 1.7403182983398438,
      "step": 12701
    },
    {
      "epoch": 1.2112617492675782e-05,
      "step": 12701,
      "training_step_time": 8.604175090789795
    },
    {
      "epoch": 1.2113571166992187e-05,
      "model_forward_time": 1.7442989349365234,
      "step": 12702
    },
    {
      "epoch": 1.2113571166992187e-05,
      "step": 12702,
      "training_step_time": 8.047414779663086
    },
    {
      "epoch": 1.2114524841308593e-05,
      "model_forward_time": 1.747356653213501,
      "step": 12703
    },
    {
      "epoch": 1.2114524841308593e-05,
      "step": 12703,
      "training_step_time": 7.728273153305054
    },
    {
      "epoch": 1.2115478515625e-05,
      "model_forward_time": 1.7282185554504395,
      "step": 12704
    },
    {
      "epoch": 1.2115478515625e-05,
      "step": 12704,
      "training_step_time": 7.683900833129883
    },
    {
      "epoch": 1.2116432189941406e-05,
      "model_forward_time": 1.7436511516571045,
      "step": 12705
    },
    {
      "epoch": 1.2116432189941406e-05,
      "step": 12705,
      "training_step_time": 7.762130975723267
    },
    {
      "epoch": 1.2117385864257812e-05,
      "model_forward_time": 1.7395789623260498,
      "step": 12706
    },
    {
      "epoch": 1.2117385864257812e-05,
      "step": 12706,
      "training_step_time": 7.772530794143677
    },
    {
      "epoch": 1.211833953857422e-05,
      "model_forward_time": 2.943488597869873,
      "step": 12707
    },
    {
      "epoch": 1.211833953857422e-05,
      "step": 12707,
      "training_step_time": 7.838685989379883
    },
    {
      "epoch": 1.2119293212890625e-05,
      "model_forward_time": 1.7438154220581055,
      "step": 12708
    },
    {
      "epoch": 1.2119293212890625e-05,
      "step": 12708,
      "training_step_time": 7.756899833679199
    },
    {
      "epoch": 1.2120246887207031e-05,
      "model_forward_time": 1.7687833309173584,
      "step": 12709
    },
    {
      "epoch": 1.2120246887207031e-05,
      "step": 12709,
      "training_step_time": 7.686120510101318
    },
    {
      "epoch": 1.2121200561523438e-05,
      "grad_norm": 0.23794054985046387,
      "learning_rate": 9.838402260339621e-06,
      "loss": 0.0684,
      "step": 12710
    },
    {
      "epoch": 1.2121200561523438e-05,
      "model_forward_time": 1.741560697555542,
      "step": 12710
    },
    {
      "dynamics_loss_avg": 0.062472637929022315,
      "epoch": 1.2121200561523438e-05,
      "step": 12710
    },
    {
      "action_loss_avg": 0.002940040541579947,
      "epoch": 1.2121200561523438e-05,
      "step": 12710
    },
    {
      "epoch": 1.2121200561523438e-05,
      "step": 12710,
      "training_step_time": 7.702653408050537
    },
    {
      "epoch": 1.2122154235839844e-05,
      "model_forward_time": 1.7353780269622803,
      "step": 12711
    },
    {
      "epoch": 1.2122154235839844e-05,
      "step": 12711,
      "training_step_time": 7.6520984172821045
    },
    {
      "epoch": 1.212310791015625e-05,
      "model_forward_time": 1.7433855533599854,
      "step": 12712
    },
    {
      "epoch": 1.212310791015625e-05,
      "step": 12712,
      "training_step_time": 7.649328231811523
    },
    {
      "epoch": 1.2124061584472655e-05,
      "model_forward_time": 1.740041732788086,
      "step": 12713
    },
    {
      "epoch": 1.2124061584472655e-05,
      "step": 12713,
      "training_step_time": 8.776320457458496
    },
    {
      "epoch": 1.2125015258789063e-05,
      "model_forward_time": 1.7357995510101318,
      "step": 12714
    },
    {
      "epoch": 1.2125015258789063e-05,
      "step": 12714,
      "training_step_time": 7.7062366008758545
    },
    {
      "epoch": 1.2125968933105469e-05,
      "model_forward_time": 1.7397255897521973,
      "step": 12715
    },
    {
      "epoch": 1.2125968933105469e-05,
      "step": 12715,
      "training_step_time": 7.806257963180542
    },
    {
      "epoch": 1.2126922607421874e-05,
      "model_forward_time": 2.880941390991211,
      "step": 12716
    },
    {
      "epoch": 1.2126922607421874e-05,
      "step": 12716,
      "training_step_time": 7.826988935470581
    },
    {
      "epoch": 1.2127876281738282e-05,
      "model_forward_time": 1.7484323978424072,
      "step": 12717
    },
    {
      "epoch": 1.2127876281738282e-05,
      "step": 12717,
      "training_step_time": 9.46738052368164
    },
    {
      "epoch": 1.2128829956054687e-05,
      "model_forward_time": 1.3795640468597412,
      "step": 12718
    },
    {
      "epoch": 1.2128829956054687e-05,
      "step": 12718,
      "training_step_time": 8.307555437088013
    },
    {
      "epoch": 1.2129783630371093e-05,
      "model_forward_time": 1.7439613342285156,
      "step": 12719
    },
    {
      "epoch": 1.2129783630371093e-05,
      "step": 12719,
      "training_step_time": 7.722703695297241
    },
    {
      "epoch": 1.21307373046875e-05,
      "grad_norm": 0.2303679883480072,
      "learning_rate": 9.837985024455918e-06,
      "loss": 0.0647,
      "step": 12720
    },
    {
      "epoch": 1.21307373046875e-05,
      "model_forward_time": 1.749366283416748,
      "step": 12720
    },
    {
      "dynamics_loss_avg": 0.07122944556176662,
      "epoch": 1.21307373046875e-05,
      "step": 12720
    },
    {
      "action_loss_avg": 0.004317776241805404,
      "epoch": 1.21307373046875e-05,
      "step": 12720
    },
    {
      "epoch": 1.21307373046875e-05,
      "step": 12720,
      "training_step_time": 7.581799268722534
    },
    {
      "epoch": 1.2131690979003906e-05,
      "model_forward_time": 1.738203763961792,
      "step": 12721
    },
    {
      "epoch": 1.2131690979003906e-05,
      "step": 12721,
      "training_step_time": 7.775331735610962
    },
    {
      "epoch": 1.2132644653320312e-05,
      "model_forward_time": 1.744337797164917,
      "step": 12722
    },
    {
      "epoch": 1.2132644653320312e-05,
      "step": 12722,
      "training_step_time": 7.7529542446136475
    },
    {
      "epoch": 1.213359832763672e-05,
      "model_forward_time": 1.3695964813232422,
      "step": 12723
    },
    {
      "epoch": 1.213359832763672e-05,
      "step": 12723,
      "training_step_time": 8.731620788574219
    },
    {
      "epoch": 1.2134552001953125e-05,
      "model_forward_time": 1.7433514595031738,
      "step": 12724
    },
    {
      "epoch": 1.2134552001953125e-05,
      "step": 12724,
      "training_step_time": 8.753951072692871
    },
    {
      "epoch": 1.2135505676269531e-05,
      "model_forward_time": 1.7592401504516602,
      "step": 12725
    },
    {
      "epoch": 1.2135505676269531e-05,
      "step": 12725,
      "training_step_time": 8.42986536026001
    },
    {
      "epoch": 1.2136459350585938e-05,
      "model_forward_time": 2.927405595779419,
      "step": 12726
    },
    {
      "epoch": 1.2136459350585938e-05,
      "step": 12726,
      "training_step_time": 8.702204942703247
    },
    {
      "epoch": 1.2137413024902344e-05,
      "model_forward_time": 1.7484281063079834,
      "step": 12727
    },
    {
      "epoch": 1.2137413024902344e-05,
      "step": 12727,
      "training_step_time": 8.485903024673462
    },
    {
      "epoch": 1.213836669921875e-05,
      "model_forward_time": 1.7447457313537598,
      "step": 12728
    },
    {
      "epoch": 1.213836669921875e-05,
      "step": 12728,
      "training_step_time": 8.82387399673462
    },
    {
      "epoch": 1.2139320373535155e-05,
      "model_forward_time": 1.7466952800750732,
      "step": 12729
    },
    {
      "epoch": 1.2139320373535155e-05,
      "step": 12729,
      "training_step_time": 7.6844704151153564
    },
    {
      "epoch": 1.2140274047851563e-05,
      "grad_norm": 0.22235777974128723,
      "learning_rate": 9.837567259497448e-06,
      "loss": 0.067,
      "step": 12730
    },
    {
      "epoch": 1.2140274047851563e-05,
      "model_forward_time": 1.751575231552124,
      "step": 12730
    },
    {
      "dynamics_loss_avg": 0.05923484545201063,
      "epoch": 1.2140274047851563e-05,
      "step": 12730
    },
    {
      "action_loss_avg": 0.0027722625876776875,
      "epoch": 1.2140274047851563e-05,
      "step": 12730
    },
    {
      "epoch": 1.2140274047851563e-05,
      "step": 12730,
      "training_step_time": 7.787015914916992
    },
    {
      "epoch": 1.2141227722167969e-05,
      "model_forward_time": 1.7388334274291992,
      "step": 12731
    },
    {
      "epoch": 1.2141227722167969e-05,
      "step": 12731,
      "training_step_time": 7.69469952583313
    },
    {
      "epoch": 1.2142181396484374e-05,
      "model_forward_time": 1.7308917045593262,
      "step": 12732
    },
    {
      "epoch": 1.2142181396484374e-05,
      "step": 12732,
      "training_step_time": 7.72161865234375
    },
    {
      "epoch": 1.2143135070800782e-05,
      "model_forward_time": 1.7505555152893066,
      "step": 12733
    },
    {
      "epoch": 1.2143135070800782e-05,
      "step": 12733,
      "training_step_time": 9.378741264343262
    },
    {
      "epoch": 1.2144088745117187e-05,
      "model_forward_time": 1.773636817932129,
      "step": 12734
    },
    {
      "epoch": 1.2144088745117187e-05,
      "step": 12734,
      "training_step_time": 8.846835136413574
    },
    {
      "epoch": 1.2145042419433593e-05,
      "model_forward_time": 2.6348447799682617,
      "step": 12735
    },
    {
      "epoch": 1.2145042419433593e-05,
      "step": 12735,
      "training_step_time": 7.72367525100708
    },
    {
      "epoch": 1.214599609375e-05,
      "model_forward_time": 1.7580692768096924,
      "step": 12736
    },
    {
      "epoch": 1.214599609375e-05,
      "step": 12736,
      "training_step_time": 7.663987874984741
    },
    {
      "epoch": 1.2146949768066406e-05,
      "model_forward_time": 1.7214012145996094,
      "step": 12737
    },
    {
      "epoch": 1.2146949768066406e-05,
      "step": 12737,
      "training_step_time": 7.761090040206909
    },
    {
      "epoch": 1.2147903442382812e-05,
      "model_forward_time": 1.7424960136413574,
      "step": 12738
    },
    {
      "epoch": 1.2147903442382812e-05,
      "step": 12738,
      "training_step_time": 7.757638216018677
    },
    {
      "epoch": 1.214885711669922e-05,
      "model_forward_time": 1.7424943447113037,
      "step": 12739
    },
    {
      "epoch": 1.214885711669922e-05,
      "step": 12739,
      "training_step_time": 7.699988126754761
    },
    {
      "epoch": 1.2149810791015625e-05,
      "grad_norm": 0.20802447199821472,
      "learning_rate": 9.837148965509894e-06,
      "loss": 0.0604,
      "step": 12740
    },
    {
      "epoch": 1.2149810791015625e-05,
      "model_forward_time": 1.7331397533416748,
      "step": 12740
    },
    {
      "dynamics_loss_avg": 0.05890838522464037,
      "epoch": 1.2149810791015625e-05,
      "step": 12740
    },
    {
      "action_loss_avg": 0.0051321328734047714,
      "epoch": 1.2149810791015625e-05,
      "step": 12740
    },
    {
      "epoch": 1.2149810791015625e-05,
      "step": 12740,
      "training_step_time": 7.628764629364014
    },
    {
      "epoch": 1.2150764465332031e-05,
      "model_forward_time": 1.7704193592071533,
      "step": 12741
    },
    {
      "epoch": 1.2150764465332031e-05,
      "step": 12741,
      "training_step_time": 7.7638936042785645
    },
    {
      "epoch": 1.2151718139648438e-05,
      "model_forward_time": 1.7445733547210693,
      "step": 12742
    },
    {
      "epoch": 1.2151718139648438e-05,
      "step": 12742,
      "training_step_time": 7.952384948730469
    },
    {
      "epoch": 1.2152671813964844e-05,
      "model_forward_time": 1.7505311965942383,
      "step": 12743
    },
    {
      "epoch": 1.2152671813964844e-05,
      "step": 12743,
      "training_step_time": 7.8417699337005615
    },
    {
      "epoch": 1.215362548828125e-05,
      "model_forward_time": 2.3236982822418213,
      "step": 12744
    },
    {
      "epoch": 1.215362548828125e-05,
      "step": 12744,
      "training_step_time": 8.51433801651001
    },
    {
      "epoch": 1.2154579162597656e-05,
      "model_forward_time": 1.7620728015899658,
      "step": 12745
    },
    {
      "epoch": 1.2154579162597656e-05,
      "step": 12745,
      "training_step_time": 8.355997085571289
    },
    {
      "epoch": 1.2155532836914063e-05,
      "model_forward_time": 1.7441105842590332,
      "step": 12746
    },
    {
      "epoch": 1.2155532836914063e-05,
      "step": 12746,
      "training_step_time": 7.718917608261108
    },
    {
      "epoch": 1.2156486511230469e-05,
      "model_forward_time": 1.7477378845214844,
      "step": 12747
    },
    {
      "epoch": 1.2156486511230469e-05,
      "step": 12747,
      "training_step_time": 7.845676422119141
    },
    {
      "epoch": 1.2157440185546874e-05,
      "model_forward_time": 1.7582967281341553,
      "step": 12748
    },
    {
      "epoch": 1.2157440185546874e-05,
      "step": 12748,
      "training_step_time": 8.00883936882019
    },
    {
      "epoch": 1.2158393859863282e-05,
      "model_forward_time": 1.3628134727478027,
      "step": 12749
    },
    {
      "epoch": 1.2158393859863282e-05,
      "step": 12749,
      "training_step_time": 8.315852642059326
    },
    {
      "epoch": 1.2159347534179688e-05,
      "grad_norm": 0.2533165514469147,
      "learning_rate": 9.836730142539001e-06,
      "loss": 0.0655,
      "step": 12750
    },
    {
      "epoch": 1.2159347534179688e-05,
      "model_forward_time": 1.7534384727478027,
      "step": 12750
    },
    {
      "dynamics_loss_avg": 0.05911646001040936,
      "epoch": 1.2159347534179688e-05,
      "step": 12750
    },
    {
      "action_loss_avg": 0.0033259291667491198,
      "epoch": 1.2159347534179688e-05,
      "step": 12750
    },
    {
      "epoch": 1.2159347534179688e-05,
      "step": 12750,
      "training_step_time": 8.58038854598999
    },
    {
      "epoch": 1.2160301208496093e-05,
      "model_forward_time": 1.743730068206787,
      "step": 12751
    },
    {
      "epoch": 1.2160301208496093e-05,
      "step": 12751,
      "training_step_time": 8.76658844947815
    },
    {
      "epoch": 1.21612548828125e-05,
      "model_forward_time": 1.7467827796936035,
      "step": 12752
    },
    {
      "epoch": 1.21612548828125e-05,
      "step": 12752,
      "training_step_time": 8.762502431869507
    },
    {
      "epoch": 1.2162208557128906e-05,
      "model_forward_time": 2.2660396099090576,
      "step": 12753
    },
    {
      "epoch": 1.2162208557128906e-05,
      "step": 12753,
      "training_step_time": 8.857154846191406
    },
    {
      "epoch": 1.2163162231445312e-05,
      "model_forward_time": 1.745391845703125,
      "step": 12754
    },
    {
      "epoch": 1.2163162231445312e-05,
      "step": 12754,
      "training_step_time": 7.828700542449951
    },
    {
      "epoch": 1.216411590576172e-05,
      "model_forward_time": 1.3630704879760742,
      "step": 12755
    },
    {
      "epoch": 1.216411590576172e-05,
      "step": 12755,
      "training_step_time": 7.75142240524292
    },
    {
      "epoch": 1.2165069580078125e-05,
      "model_forward_time": 1.742851734161377,
      "step": 12756
    },
    {
      "epoch": 1.2165069580078125e-05,
      "step": 12756,
      "training_step_time": 7.804004669189453
    },
    {
      "epoch": 1.2166023254394531e-05,
      "model_forward_time": 1.746009349822998,
      "step": 12757
    },
    {
      "epoch": 1.2166023254394531e-05,
      "step": 12757,
      "training_step_time": 8.623312711715698
    },
    {
      "epoch": 1.2166976928710937e-05,
      "model_forward_time": 1.749551773071289,
      "step": 12758
    },
    {
      "epoch": 1.2166976928710937e-05,
      "step": 12758,
      "training_step_time": 8.542086362838745
    },
    {
      "epoch": 1.2167930603027344e-05,
      "model_forward_time": 1.7438561916351318,
      "step": 12759
    },
    {
      "epoch": 1.2167930603027344e-05,
      "step": 12759,
      "training_step_time": 9.316406965255737
    },
    {
      "epoch": 1.216888427734375e-05,
      "grad_norm": 0.2806495726108551,
      "learning_rate": 9.836310790630575e-06,
      "loss": 0.0635,
      "step": 12760
    },
    {
      "epoch": 1.216888427734375e-05,
      "model_forward_time": 1.7958385944366455,
      "step": 12760
    },
    {
      "dynamics_loss_avg": 0.04192010760307312,
      "epoch": 1.216888427734375e-05,
      "step": 12760
    },
    {
      "action_loss_avg": 0.002192184835439548,
      "epoch": 1.216888427734375e-05,
      "step": 12760
    },
    {
      "epoch": 1.216888427734375e-05,
      "step": 12760,
      "training_step_time": 7.83585262298584
    },
    {
      "epoch": 1.2169837951660156e-05,
      "model_forward_time": 1.7353315353393555,
      "step": 12761
    },
    {
      "epoch": 1.2169837951660156e-05,
      "step": 12761,
      "training_step_time": 7.88293981552124
    },
    {
      "epoch": 1.2170791625976563e-05,
      "model_forward_time": 1.7431578636169434,
      "step": 12762
    },
    {
      "epoch": 1.2170791625976563e-05,
      "step": 12762,
      "training_step_time": 9.285693407058716
    },
    {
      "epoch": 1.2171745300292969e-05,
      "model_forward_time": 2.298543930053711,
      "step": 12763
    },
    {
      "epoch": 1.2171745300292969e-05,
      "step": 12763,
      "training_step_time": 8.463235139846802
    },
    {
      "epoch": 1.2172698974609374e-05,
      "model_forward_time": 1.737173318862915,
      "step": 12764
    },
    {
      "epoch": 1.2172698974609374e-05,
      "step": 12764,
      "training_step_time": 7.702537536621094
    },
    {
      "epoch": 1.2173652648925782e-05,
      "model_forward_time": 1.7280118465423584,
      "step": 12765
    },
    {
      "epoch": 1.2173652648925782e-05,
      "step": 12765,
      "training_step_time": 7.819694995880127
    },
    {
      "epoch": 1.2174606323242188e-05,
      "model_forward_time": 1.0480985641479492,
      "step": 12766
    },
    {
      "epoch": 1.2174606323242188e-05,
      "step": 12766,
      "training_step_time": 7.539752244949341
    },
    {
      "epoch": 1.2175559997558593e-05,
      "model_forward_time": 1.7635788917541504,
      "step": 12767
    },
    {
      "epoch": 1.2175559997558593e-05,
      "step": 12767,
      "training_step_time": 8.793466567993164
    },
    {
      "epoch": 1.2176513671875e-05,
      "model_forward_time": 1.7445874214172363,
      "step": 12768
    },
    {
      "epoch": 1.2176513671875e-05,
      "step": 12768,
      "training_step_time": 7.869046211242676
    },
    {
      "epoch": 1.2177467346191406e-05,
      "model_forward_time": 1.7876629829406738,
      "step": 12769
    },
    {
      "epoch": 1.2177467346191406e-05,
      "step": 12769,
      "training_step_time": 7.780738592147827
    },
    {
      "epoch": 1.2178421020507812e-05,
      "grad_norm": 0.2997543215751648,
      "learning_rate": 9.83589090983047e-06,
      "loss": 0.0668,
      "step": 12770
    },
    {
      "epoch": 1.2178421020507812e-05,
      "model_forward_time": 1.7457001209259033,
      "step": 12770
    },
    {
      "dynamics_loss_avg": 0.06595597732812167,
      "epoch": 1.2178421020507812e-05,
      "step": 12770
    },
    {
      "action_loss_avg": 0.003985804109834135,
      "epoch": 1.2178421020507812e-05,
      "step": 12770
    },
    {
      "epoch": 1.2178421020507812e-05,
      "step": 12770,
      "training_step_time": 7.898960590362549
    },
    {
      "epoch": 1.217937469482422e-05,
      "model_forward_time": 1.7283477783203125,
      "step": 12771
    },
    {
      "epoch": 1.217937469482422e-05,
      "step": 12771,
      "training_step_time": 9.353898286819458
    },
    {
      "epoch": 1.2180328369140625e-05,
      "model_forward_time": 2.2975049018859863,
      "step": 12772
    },
    {
      "epoch": 1.2180328369140625e-05,
      "step": 12772,
      "training_step_time": 7.811807632446289
    },
    {
      "epoch": 1.2181282043457031e-05,
      "model_forward_time": 1.7449042797088623,
      "step": 12773
    },
    {
      "epoch": 1.2181282043457031e-05,
      "step": 12773,
      "training_step_time": 7.738026857376099
    },
    {
      "epoch": 1.2182235717773437e-05,
      "model_forward_time": 1.7264652252197266,
      "step": 12774
    },
    {
      "epoch": 1.2182235717773437e-05,
      "step": 12774,
      "training_step_time": 8.822676181793213
    },
    {
      "epoch": 1.2183189392089844e-05,
      "model_forward_time": 1.7262215614318848,
      "step": 12775
    },
    {
      "epoch": 1.2183189392089844e-05,
      "step": 12775,
      "training_step_time": 7.7855284214019775
    },
    {
      "epoch": 1.218414306640625e-05,
      "model_forward_time": 1.7311408519744873,
      "step": 12776
    },
    {
      "epoch": 1.218414306640625e-05,
      "step": 12776,
      "training_step_time": 8.223080158233643
    },
    {
      "epoch": 1.2185096740722656e-05,
      "model_forward_time": 1.7271089553833008,
      "step": 12777
    },
    {
      "epoch": 1.2185096740722656e-05,
      "step": 12777,
      "training_step_time": 7.764156818389893
    },
    {
      "epoch": 1.2186050415039063e-05,
      "model_forward_time": 1.7751173973083496,
      "step": 12778
    },
    {
      "epoch": 1.2186050415039063e-05,
      "step": 12778,
      "training_step_time": 7.74242639541626
    },
    {
      "epoch": 1.2187004089355469e-05,
      "model_forward_time": 1.720841884613037,
      "step": 12779
    },
    {
      "epoch": 1.2187004089355469e-05,
      "step": 12779,
      "training_step_time": 10.524279356002808
    },
    {
      "epoch": 1.2187957763671874e-05,
      "grad_norm": 0.327163964509964,
      "learning_rate": 9.835470500184604e-06,
      "loss": 0.0712,
      "step": 12780
    },
    {
      "epoch": 1.2187957763671874e-05,
      "model_forward_time": 1.7280027866363525,
      "step": 12780
    },
    {
      "dynamics_loss_avg": 0.07658005207777023,
      "epoch": 1.2187957763671874e-05,
      "step": 12780
    },
    {
      "action_loss_avg": 0.005092255596537143,
      "epoch": 1.2187957763671874e-05,
      "step": 12780
    },
    {
      "epoch": 1.2187957763671874e-05,
      "step": 12780,
      "training_step_time": 7.81079363822937
    },
    {
      "epoch": 1.2188911437988282e-05,
      "model_forward_time": 2.2636258602142334,
      "step": 12781
    },
    {
      "epoch": 1.2188911437988282e-05,
      "step": 12781,
      "training_step_time": 10.76337218284607
    },
    {
      "epoch": 1.2189865112304688e-05,
      "model_forward_time": 1.7420554161071777,
      "step": 12782
    },
    {
      "epoch": 1.2189865112304688e-05,
      "step": 12782,
      "training_step_time": 7.825902700424194
    },
    {
      "epoch": 1.2190818786621093e-05,
      "model_forward_time": 1.7457101345062256,
      "step": 12783
    },
    {
      "epoch": 1.2190818786621093e-05,
      "step": 12783,
      "training_step_time": 7.814879655838013
    },
    {
      "epoch": 1.21917724609375e-05,
      "model_forward_time": 1.7551872730255127,
      "step": 12784
    },
    {
      "epoch": 1.21917724609375e-05,
      "step": 12784,
      "training_step_time": 9.128836154937744
    },
    {
      "epoch": 1.2192726135253906e-05,
      "model_forward_time": 1.743717908859253,
      "step": 12785
    },
    {
      "epoch": 1.2192726135253906e-05,
      "step": 12785,
      "training_step_time": 7.752765655517578
    },
    {
      "epoch": 1.2193679809570312e-05,
      "model_forward_time": 1.719268560409546,
      "step": 12786
    },
    {
      "epoch": 1.2193679809570312e-05,
      "step": 12786,
      "training_step_time": 8.732170581817627
    },
    {
      "epoch": 1.219463348388672e-05,
      "model_forward_time": 1.7526068687438965,
      "step": 12787
    },
    {
      "epoch": 1.219463348388672e-05,
      "step": 12787,
      "training_step_time": 7.691822052001953
    },
    {
      "epoch": 1.2195587158203125e-05,
      "model_forward_time": 1.373293399810791,
      "step": 12788
    },
    {
      "epoch": 1.2195587158203125e-05,
      "step": 12788,
      "training_step_time": 8.666913270950317
    },
    {
      "epoch": 1.2196540832519531e-05,
      "model_forward_time": 1.7456002235412598,
      "step": 12789
    },
    {
      "epoch": 1.2196540832519531e-05,
      "step": 12789,
      "training_step_time": 8.784284591674805
    },
    {
      "epoch": 1.2197494506835937e-05,
      "grad_norm": 0.26098865270614624,
      "learning_rate": 9.835049561738958e-06,
      "loss": 0.0663,
      "step": 12790
    },
    {
      "epoch": 1.2197494506835937e-05,
      "model_forward_time": 1.7476348876953125,
      "step": 12790
    },
    {
      "dynamics_loss_avg": 0.048227735608816144,
      "epoch": 1.2197494506835937e-05,
      "step": 12790
    },
    {
      "action_loss_avg": 0.003722030471544713,
      "epoch": 1.2197494506835937e-05,
      "step": 12790
    },
    {
      "epoch": 1.2197494506835937e-05,
      "step": 12790,
      "training_step_time": 7.79969048500061
    },
    {
      "epoch": 1.2198448181152344e-05,
      "model_forward_time": 2.2967381477355957,
      "step": 12791
    },
    {
      "epoch": 1.2198448181152344e-05,
      "step": 12791,
      "training_step_time": 7.80502724647522
    },
    {
      "epoch": 1.219940185546875e-05,
      "model_forward_time": 1.7492866516113281,
      "step": 12792
    },
    {
      "epoch": 1.219940185546875e-05,
      "step": 12792,
      "training_step_time": 7.776601552963257
    },
    {
      "epoch": 1.2200355529785156e-05,
      "model_forward_time": 1.7460837364196777,
      "step": 12793
    },
    {
      "epoch": 1.2200355529785156e-05,
      "step": 12793,
      "training_step_time": 8.735414743423462
    },
    {
      "epoch": 1.2201309204101563e-05,
      "model_forward_time": 1.74428129196167,
      "step": 12794
    },
    {
      "epoch": 1.2201309204101563e-05,
      "step": 12794,
      "training_step_time": 7.73090386390686
    },
    {
      "epoch": 1.2202262878417969e-05,
      "model_forward_time": 1.7447264194488525,
      "step": 12795
    },
    {
      "epoch": 1.2202262878417969e-05,
      "step": 12795,
      "training_step_time": 7.710987329483032
    },
    {
      "epoch": 1.2203216552734374e-05,
      "model_forward_time": 1.747514009475708,
      "step": 12796
    },
    {
      "epoch": 1.2203216552734374e-05,
      "step": 12796,
      "training_step_time": 7.744965553283691
    },
    {
      "epoch": 1.2204170227050782e-05,
      "model_forward_time": 1.7982101440429688,
      "step": 12797
    },
    {
      "epoch": 1.2204170227050782e-05,
      "step": 12797,
      "training_step_time": 8.50789499282837
    },
    {
      "epoch": 1.2205123901367188e-05,
      "model_forward_time": 1.7480478286743164,
      "step": 12798
    },
    {
      "epoch": 1.2205123901367188e-05,
      "step": 12798,
      "training_step_time": 7.690673589706421
    },
    {
      "epoch": 1.2206077575683593e-05,
      "model_forward_time": 1.7478580474853516,
      "step": 12799
    },
    {
      "epoch": 1.2206077575683593e-05,
      "step": 12799,
      "training_step_time": 8.604949712753296
    },
    {
      "epoch": 1.220703125e-05,
      "grad_norm": 0.23135533928871155,
      "learning_rate": 9.834628094539559e-06,
      "loss": 0.067,
      "step": 12800
    },
    {
      "epoch": 1.220703125e-05,
      "model_forward_time": 2.2888104915618896,
      "step": 12800
    },
    {
      "dynamics_loss_avg": 0.05699190865270794,
      "epoch": 1.220703125e-05,
      "step": 12800
    },
    {
      "action_loss_avg": 0.003613894328009337,
      "epoch": 1.220703125e-05,
      "step": 12800
    },
    {
      "epoch": 1.220703125e-05,
      "step": 12800,
      "training_step_time": 8.780484914779663
    },
    {
      "epoch": 1.2207984924316406e-05,
      "model_forward_time": 1.752948522567749,
      "step": 12801
    },
    {
      "epoch": 1.2207984924316406e-05,
      "step": 12801,
      "training_step_time": 8.720600605010986
    },
    {
      "epoch": 1.2208938598632812e-05,
      "model_forward_time": 1.7416493892669678,
      "step": 12802
    },
    {
      "epoch": 1.2208938598632812e-05,
      "step": 12802,
      "training_step_time": 7.6274707317352295
    },
    {
      "epoch": 1.220989227294922e-05,
      "model_forward_time": 1.749422550201416,
      "step": 12803
    },
    {
      "epoch": 1.220989227294922e-05,
      "step": 12803,
      "training_step_time": 7.703146934509277
    },
    {
      "epoch": 1.2210845947265625e-05,
      "model_forward_time": 1.7405869960784912,
      "step": 12804
    },
    {
      "epoch": 1.2210845947265625e-05,
      "step": 12804,
      "training_step_time": 7.743851184844971
    },
    {
      "epoch": 1.2211799621582031e-05,
      "model_forward_time": 1.749551773071289,
      "step": 12805
    },
    {
      "epoch": 1.2211799621582031e-05,
      "step": 12805,
      "training_step_time": 8.818596124649048
    },
    {
      "epoch": 1.2212753295898437e-05,
      "model_forward_time": 1.8027381896972656,
      "step": 12806
    },
    {
      "epoch": 1.2212753295898437e-05,
      "step": 12806,
      "training_step_time": 7.772159099578857
    },
    {
      "epoch": 1.2213706970214844e-05,
      "model_forward_time": 1.7389130592346191,
      "step": 12807
    },
    {
      "epoch": 1.2213706970214844e-05,
      "step": 12807,
      "training_step_time": 7.84351658821106
    },
    {
      "epoch": 1.221466064453125e-05,
      "model_forward_time": 1.742889642715454,
      "step": 12808
    },
    {
      "epoch": 1.221466064453125e-05,
      "step": 12808,
      "training_step_time": 7.829299449920654
    },
    {
      "epoch": 1.2215614318847656e-05,
      "model_forward_time": 2.322608470916748,
      "step": 12809
    },
    {
      "epoch": 1.2215614318847656e-05,
      "step": 12809,
      "training_step_time": 8.497605323791504
    },
    {
      "epoch": 1.2216567993164063e-05,
      "grad_norm": 0.22178982198238373,
      "learning_rate": 9.834206098632499e-06,
      "loss": 0.0654,
      "step": 12810
    },
    {
      "epoch": 1.2216567993164063e-05,
      "model_forward_time": 1.7438786029815674,
      "step": 12810
    },
    {
      "dynamics_loss_avg": 0.05805278830230236,
      "epoch": 1.2216567993164063e-05,
      "step": 12810
    },
    {
      "action_loss_avg": 0.0029331907513551413,
      "epoch": 1.2216567993164063e-05,
      "step": 12810
    },
    {
      "epoch": 1.2216567993164063e-05,
      "step": 12810,
      "training_step_time": 8.740847826004028
    },
    {
      "epoch": 1.2217521667480469e-05,
      "model_forward_time": 1.74288010597229,
      "step": 12811
    },
    {
      "epoch": 1.2217521667480469e-05,
      "step": 12811,
      "training_step_time": 7.695139408111572
    },
    {
      "epoch": 1.2218475341796874e-05,
      "model_forward_time": 1.7388911247253418,
      "step": 12812
    },
    {
      "epoch": 1.2218475341796874e-05,
      "step": 12812,
      "training_step_time": 8.930150985717773
    },
    {
      "epoch": 1.2219429016113282e-05,
      "model_forward_time": 1.7413902282714844,
      "step": 12813
    },
    {
      "epoch": 1.2219429016113282e-05,
      "step": 12813,
      "training_step_time": 8.442785501480103
    },
    {
      "epoch": 1.2220382690429688e-05,
      "model_forward_time": 1.761237382888794,
      "step": 12814
    },
    {
      "epoch": 1.2220382690429688e-05,
      "step": 12814,
      "training_step_time": 8.32319164276123
    },
    {
      "epoch": 1.2221336364746093e-05,
      "model_forward_time": 1.7631754875183105,
      "step": 12815
    },
    {
      "epoch": 1.2221336364746093e-05,
      "step": 12815,
      "training_step_time": 7.83025860786438
    },
    {
      "epoch": 1.22222900390625e-05,
      "model_forward_time": 1.7446813583374023,
      "step": 12816
    },
    {
      "epoch": 1.22222900390625e-05,
      "step": 12816,
      "training_step_time": 7.802382707595825
    },
    {
      "epoch": 1.2223243713378906e-05,
      "model_forward_time": 1.039186954498291,
      "step": 12817
    },
    {
      "epoch": 1.2223243713378906e-05,
      "step": 12817,
      "training_step_time": 8.347848892211914
    },
    {
      "epoch": 1.2224197387695312e-05,
      "model_forward_time": 1.742018222808838,
      "step": 12818
    },
    {
      "epoch": 1.2224197387695312e-05,
      "step": 12818,
      "training_step_time": 9.886922359466553
    },
    {
      "epoch": 1.222515106201172e-05,
      "model_forward_time": 2.2859036922454834,
      "step": 12819
    },
    {
      "epoch": 1.222515106201172e-05,
      "step": 12819,
      "training_step_time": 7.800546646118164
    },
    {
      "epoch": 1.2226104736328125e-05,
      "grad_norm": 0.22297632694244385,
      "learning_rate": 9.833783574063931e-06,
      "loss": 0.0692,
      "step": 12820
    },
    {
      "epoch": 1.2226104736328125e-05,
      "model_forward_time": 1.738483190536499,
      "step": 12820
    },
    {
      "dynamics_loss_avg": 0.06320742340758442,
      "epoch": 1.2226104736328125e-05,
      "step": 12820
    },
    {
      "action_loss_avg": 0.0033264999918174,
      "epoch": 1.2226104736328125e-05,
      "step": 12820
    },
    {
      "epoch": 1.2226104736328125e-05,
      "step": 12820,
      "training_step_time": 7.71382474899292
    },
    {
      "epoch": 1.2227058410644531e-05,
      "model_forward_time": 1.7447857856750488,
      "step": 12821
    },
    {
      "epoch": 1.2227058410644531e-05,
      "step": 12821,
      "training_step_time": 7.78886342048645
    },
    {
      "epoch": 1.2228012084960937e-05,
      "model_forward_time": 1.7525315284729004,
      "step": 12822
    },
    {
      "epoch": 1.2228012084960937e-05,
      "step": 12822,
      "training_step_time": 7.681960105895996
    },
    {
      "epoch": 1.2228965759277344e-05,
      "model_forward_time": 1.744133472442627,
      "step": 12823
    },
    {
      "epoch": 1.2228965759277344e-05,
      "step": 12823,
      "training_step_time": 7.69707989692688
    },
    {
      "epoch": 1.222991943359375e-05,
      "model_forward_time": 1.7523632049560547,
      "step": 12824
    },
    {
      "epoch": 1.222991943359375e-05,
      "step": 12824,
      "training_step_time": 7.713033199310303
    },
    {
      "epoch": 1.2230873107910156e-05,
      "model_forward_time": 1.7416870594024658,
      "step": 12825
    },
    {
      "epoch": 1.2230873107910156e-05,
      "step": 12825,
      "training_step_time": 9.90226936340332
    },
    {
      "epoch": 1.2231826782226563e-05,
      "model_forward_time": 1.7396962642669678,
      "step": 12826
    },
    {
      "epoch": 1.2231826782226563e-05,
      "step": 12826,
      "training_step_time": 7.756918668746948
    },
    {
      "epoch": 1.2232780456542969e-05,
      "model_forward_time": 1.7415032386779785,
      "step": 12827
    },
    {
      "epoch": 1.2232780456542969e-05,
      "step": 12827,
      "training_step_time": 7.917279005050659
    },
    {
      "epoch": 1.2233734130859374e-05,
      "model_forward_time": 2.302588701248169,
      "step": 12828
    },
    {
      "epoch": 1.2233734130859374e-05,
      "step": 12828,
      "training_step_time": 9.306820631027222
    },
    {
      "epoch": 1.2234687805175782e-05,
      "model_forward_time": 1.7347581386566162,
      "step": 12829
    },
    {
      "epoch": 1.2234687805175782e-05,
      "step": 12829,
      "training_step_time": 7.734321117401123
    },
    {
      "epoch": 1.2235641479492188e-05,
      "grad_norm": 0.19072477519512177,
      "learning_rate": 9.833360520880057e-06,
      "loss": 0.0692,
      "step": 12830
    },
    {
      "epoch": 1.2235641479492188e-05,
      "model_forward_time": 1.747520923614502,
      "step": 12830
    },
    {
      "dynamics_loss_avg": 0.06042760889977217,
      "epoch": 1.2235641479492188e-05,
      "step": 12830
    },
    {
      "action_loss_avg": 0.003739490651059896,
      "epoch": 1.2235641479492188e-05,
      "step": 12830
    },
    {
      "epoch": 1.2235641479492188e-05,
      "step": 12830,
      "training_step_time": 7.813063621520996
    },
    {
      "epoch": 1.2236595153808593e-05,
      "model_forward_time": 1.7466542720794678,
      "step": 12831
    },
    {
      "epoch": 1.2236595153808593e-05,
      "step": 12831,
      "training_step_time": 7.560732126235962
    },
    {
      "epoch": 1.2237548828125e-05,
      "model_forward_time": 1.7333974838256836,
      "step": 12832
    },
    {
      "epoch": 1.2237548828125e-05,
      "step": 12832,
      "training_step_time": 8.409563064575195
    },
    {
      "epoch": 1.2238502502441406e-05,
      "model_forward_time": 1.7449266910552979,
      "step": 12833
    },
    {
      "epoch": 1.2238502502441406e-05,
      "step": 12833,
      "training_step_time": 7.606900215148926
    },
    {
      "epoch": 1.2239456176757812e-05,
      "model_forward_time": 1.7939271926879883,
      "step": 12834
    },
    {
      "epoch": 1.2239456176757812e-05,
      "step": 12834,
      "training_step_time": 8.58152961730957
    },
    {
      "epoch": 1.224040985107422e-05,
      "model_forward_time": 1.7460949420928955,
      "step": 12835
    },
    {
      "epoch": 1.224040985107422e-05,
      "step": 12835,
      "training_step_time": 8.729329586029053
    },
    {
      "epoch": 1.2241363525390625e-05,
      "model_forward_time": 1.7350211143493652,
      "step": 12836
    },
    {
      "epoch": 1.2241363525390625e-05,
      "step": 12836,
      "training_step_time": 8.494834423065186
    },
    {
      "epoch": 1.2242317199707031e-05,
      "model_forward_time": 2.300100803375244,
      "step": 12837
    },
    {
      "epoch": 1.2242317199707031e-05,
      "step": 12837,
      "training_step_time": 9.637532949447632
    },
    {
      "epoch": 1.2243270874023437e-05,
      "model_forward_time": 1.7332396507263184,
      "step": 12838
    },
    {
      "epoch": 1.2243270874023437e-05,
      "step": 12838,
      "training_step_time": 8.55819320678711
    },
    {
      "epoch": 1.2244224548339844e-05,
      "model_forward_time": 1.73941969871521,
      "step": 12839
    },
    {
      "epoch": 1.2244224548339844e-05,
      "step": 12839,
      "training_step_time": 7.54863715171814
    },
    {
      "epoch": 1.224517822265625e-05,
      "grad_norm": 0.5855985283851624,
      "learning_rate": 9.832936939127145e-06,
      "loss": 0.0627,
      "step": 12840
    },
    {
      "epoch": 1.224517822265625e-05,
      "model_forward_time": 1.7382302284240723,
      "step": 12840
    },
    {
      "dynamics_loss_avg": 0.05472854282706976,
      "epoch": 1.224517822265625e-05,
      "step": 12840
    },
    {
      "action_loss_avg": 0.003164025707519613,
      "epoch": 1.224517822265625e-05,
      "step": 12840
    },
    {
      "epoch": 1.224517822265625e-05,
      "step": 12840,
      "training_step_time": 7.592289924621582
    },
    {
      "epoch": 1.2246131896972656e-05,
      "model_forward_time": 1.736203670501709,
      "step": 12841
    },
    {
      "epoch": 1.2246131896972656e-05,
      "step": 12841,
      "training_step_time": 8.684817790985107
    },
    {
      "epoch": 1.2247085571289063e-05,
      "model_forward_time": 1.7482664585113525,
      "step": 12842
    },
    {
      "epoch": 1.2247085571289063e-05,
      "step": 12842,
      "training_step_time": 7.5596442222595215
    },
    {
      "epoch": 1.2248039245605469e-05,
      "model_forward_time": 1.7667174339294434,
      "step": 12843
    },
    {
      "epoch": 1.2248039245605469e-05,
      "step": 12843,
      "training_step_time": 8.555846691131592
    },
    {
      "epoch": 1.2248992919921874e-05,
      "model_forward_time": 1.735304594039917,
      "step": 12844
    },
    {
      "epoch": 1.2248992919921874e-05,
      "step": 12844,
      "training_step_time": 7.862686634063721
    },
    {
      "epoch": 1.2249946594238282e-05,
      "model_forward_time": 1.731750249862671,
      "step": 12845
    },
    {
      "epoch": 1.2249946594238282e-05,
      "step": 12845,
      "training_step_time": 9.544312477111816
    },
    {
      "epoch": 1.2250900268554688e-05,
      "model_forward_time": 2.3008134365081787,
      "step": 12846
    },
    {
      "epoch": 1.2250900268554688e-05,
      "step": 12846,
      "training_step_time": 7.845730304718018
    },
    {
      "epoch": 1.2251853942871093e-05,
      "model_forward_time": 1.7345101833343506,
      "step": 12847
    },
    {
      "epoch": 1.2251853942871093e-05,
      "step": 12847,
      "training_step_time": 8.346982479095459
    },
    {
      "epoch": 1.22528076171875e-05,
      "model_forward_time": 1.727743148803711,
      "step": 12848
    },
    {
      "epoch": 1.22528076171875e-05,
      "step": 12848,
      "training_step_time": 7.645235776901245
    },
    {
      "epoch": 1.2253761291503906e-05,
      "model_forward_time": 1.7355570793151855,
      "step": 12849
    },
    {
      "epoch": 1.2253761291503906e-05,
      "step": 12849,
      "training_step_time": 7.801931142807007
    },
    {
      "epoch": 1.2254714965820312e-05,
      "grad_norm": 0.1735788732767105,
      "learning_rate": 9.832512828851514e-06,
      "loss": 0.0627,
      "step": 12850
    },
    {
      "epoch": 1.2254714965820312e-05,
      "model_forward_time": 1.726916790008545,
      "step": 12850
    },
    {
      "dynamics_loss_avg": 0.05993912424892187,
      "epoch": 1.2254714965820312e-05,
      "step": 12850
    },
    {
      "action_loss_avg": 0.0054666913580149416,
      "epoch": 1.2254714965820312e-05,
      "step": 12850
    },
    {
      "epoch": 1.2254714965820312e-05,
      "step": 12850,
      "training_step_time": 7.606889963150024
    },
    {
      "epoch": 1.225566864013672e-05,
      "model_forward_time": 1.7334237098693848,
      "step": 12851
    },
    {
      "epoch": 1.225566864013672e-05,
      "step": 12851,
      "training_step_time": 7.691309690475464
    },
    {
      "epoch": 1.2256622314453125e-05,
      "model_forward_time": 1.7575819492340088,
      "step": 12852
    },
    {
      "epoch": 1.2256622314453125e-05,
      "step": 12852,
      "training_step_time": 8.772187232971191
    },
    {
      "epoch": 1.2257575988769531e-05,
      "model_forward_time": 1.7369003295898438,
      "step": 12853
    },
    {
      "epoch": 1.2257575988769531e-05,
      "step": 12853,
      "training_step_time": 7.787278413772583
    },
    {
      "epoch": 1.2258529663085937e-05,
      "model_forward_time": 1.7405352592468262,
      "step": 12854
    },
    {
      "epoch": 1.2258529663085937e-05,
      "step": 12854,
      "training_step_time": 8.631923913955688
    },
    {
      "epoch": 1.2259483337402344e-05,
      "model_forward_time": 2.2686901092529297,
      "step": 12855
    },
    {
      "epoch": 1.2259483337402344e-05,
      "step": 12855,
      "training_step_time": 8.440468072891235
    },
    {
      "epoch": 1.226043701171875e-05,
      "model_forward_time": 1.7352840900421143,
      "step": 12856
    },
    {
      "epoch": 1.226043701171875e-05,
      "step": 12856,
      "training_step_time": 7.735278844833374
    },
    {
      "epoch": 1.2261390686035156e-05,
      "model_forward_time": 1.7284741401672363,
      "step": 12857
    },
    {
      "epoch": 1.2261390686035156e-05,
      "step": 12857,
      "training_step_time": 7.65039849281311
    },
    {
      "epoch": 1.2262344360351563e-05,
      "model_forward_time": 1.7405288219451904,
      "step": 12858
    },
    {
      "epoch": 1.2262344360351563e-05,
      "step": 12858,
      "training_step_time": 7.863668918609619
    },
    {
      "epoch": 1.2263298034667969e-05,
      "model_forward_time": 1.7453148365020752,
      "step": 12859
    },
    {
      "epoch": 1.2263298034667969e-05,
      "step": 12859,
      "training_step_time": 8.665692567825317
    },
    {
      "epoch": 1.2264251708984374e-05,
      "grad_norm": 0.1801510453224182,
      "learning_rate": 9.832088190099548e-06,
      "loss": 0.0653,
      "step": 12860
    },
    {
      "epoch": 1.2264251708984374e-05,
      "model_forward_time": 1.7300145626068115,
      "step": 12860
    },
    {
      "dynamics_loss_avg": 0.052442125603556634,
      "epoch": 1.2264251708984374e-05,
      "step": 12860
    },
    {
      "action_loss_avg": 0.002699610451236367,
      "epoch": 1.2264251708984374e-05,
      "step": 12860
    },
    {
      "epoch": 1.2264251708984374e-05,
      "step": 12860,
      "training_step_time": 8.522350788116455
    },
    {
      "epoch": 1.2265205383300782e-05,
      "model_forward_time": 1.7406868934631348,
      "step": 12861
    },
    {
      "epoch": 1.2265205383300782e-05,
      "step": 12861,
      "training_step_time": 8.40228819847107
    },
    {
      "epoch": 1.2266159057617188e-05,
      "model_forward_time": 1.7344865798950195,
      "step": 12862
    },
    {
      "epoch": 1.2266159057617188e-05,
      "step": 12862,
      "training_step_time": 7.699546813964844
    },
    {
      "epoch": 1.2267112731933593e-05,
      "model_forward_time": 1.7333097457885742,
      "step": 12863
    },
    {
      "epoch": 1.2267112731933593e-05,
      "step": 12863,
      "training_step_time": 8.320583820343018
    },
    {
      "epoch": 1.226806640625e-05,
      "model_forward_time": 1.7385892868041992,
      "step": 12864
    },
    {
      "epoch": 1.226806640625e-05,
      "step": 12864,
      "training_step_time": 9.578511476516724
    },
    {
      "epoch": 1.2269020080566406e-05,
      "model_forward_time": 2.2673819065093994,
      "step": 12865
    },
    {
      "epoch": 1.2269020080566406e-05,
      "step": 12865,
      "training_step_time": 7.777322769165039
    },
    {
      "epoch": 1.2269973754882812e-05,
      "model_forward_time": 1.732274055480957,
      "step": 12866
    },
    {
      "epoch": 1.2269973754882812e-05,
      "step": 12866,
      "training_step_time": 7.71905779838562
    },
    {
      "epoch": 1.227092742919922e-05,
      "model_forward_time": 1.7317228317260742,
      "step": 12867
    },
    {
      "epoch": 1.227092742919922e-05,
      "step": 12867,
      "training_step_time": 8.900297403335571
    },
    {
      "epoch": 1.2271881103515625e-05,
      "model_forward_time": 1.7373762130737305,
      "step": 12868
    },
    {
      "epoch": 1.2271881103515625e-05,
      "step": 12868,
      "training_step_time": 7.65262770652771
    },
    {
      "epoch": 1.2272834777832031e-05,
      "model_forward_time": 1.7373158931732178,
      "step": 12869
    },
    {
      "epoch": 1.2272834777832031e-05,
      "step": 12869,
      "training_step_time": 7.645941257476807
    },
    {
      "epoch": 1.2273788452148437e-05,
      "grad_norm": 0.18024680018424988,
      "learning_rate": 9.831663022917679e-06,
      "loss": 0.0616,
      "step": 12870
    },
    {
      "epoch": 1.2273788452148437e-05,
      "model_forward_time": 1.7293205261230469,
      "step": 12870
    },
    {
      "dynamics_loss_avg": 0.06188991889357567,
      "epoch": 1.2273788452148437e-05,
      "step": 12870
    },
    {
      "action_loss_avg": 0.004853886656928807,
      "epoch": 1.2273788452148437e-05,
      "step": 12870
    },
    {
      "epoch": 1.2273788452148437e-05,
      "step": 12870,
      "training_step_time": 7.629022121429443
    },
    {
      "epoch": 1.2274742126464844e-05,
      "model_forward_time": 1.7899973392486572,
      "step": 12871
    },
    {
      "epoch": 1.2274742126464844e-05,
      "step": 12871,
      "training_step_time": 7.697526693344116
    },
    {
      "epoch": 1.227569580078125e-05,
      "model_forward_time": 1.7364418506622314,
      "step": 12872
    },
    {
      "epoch": 1.227569580078125e-05,
      "step": 12872,
      "training_step_time": 8.344370126724243
    },
    {
      "epoch": 1.2276649475097656e-05,
      "model_forward_time": 1.7338128089904785,
      "step": 12873
    },
    {
      "epoch": 1.2276649475097656e-05,
      "step": 12873,
      "training_step_time": 7.825054168701172
    },
    {
      "epoch": 1.2277603149414063e-05,
      "model_forward_time": 2.276195764541626,
      "step": 12874
    },
    {
      "epoch": 1.2277603149414063e-05,
      "step": 12874,
      "training_step_time": 8.622720718383789
    },
    {
      "epoch": 1.2278556823730469e-05,
      "model_forward_time": 1.7252509593963623,
      "step": 12875
    },
    {
      "epoch": 1.2278556823730469e-05,
      "step": 12875,
      "training_step_time": 8.550808429718018
    },
    {
      "epoch": 1.2279510498046874e-05,
      "model_forward_time": 1.7432684898376465,
      "step": 12876
    },
    {
      "epoch": 1.2279510498046874e-05,
      "step": 12876,
      "training_step_time": 7.774702548980713
    },
    {
      "epoch": 1.2280464172363282e-05,
      "model_forward_time": 1.738293170928955,
      "step": 12877
    },
    {
      "epoch": 1.2280464172363282e-05,
      "step": 12877,
      "training_step_time": 8.199484586715698
    },
    {
      "epoch": 1.2281417846679688e-05,
      "model_forward_time": 1.739687204360962,
      "step": 12878
    },
    {
      "epoch": 1.2281417846679688e-05,
      "step": 12878,
      "training_step_time": 7.731898069381714
    },
    {
      "epoch": 1.2282371520996093e-05,
      "model_forward_time": 1.740391731262207,
      "step": 12879
    },
    {
      "epoch": 1.2282371520996093e-05,
      "step": 12879,
      "training_step_time": 8.728914022445679
    },
    {
      "epoch": 1.22833251953125e-05,
      "grad_norm": 0.18095792829990387,
      "learning_rate": 9.831237327352408e-06,
      "loss": 0.0672,
      "step": 12880
    },
    {
      "epoch": 1.22833251953125e-05,
      "model_forward_time": 1.7816712856292725,
      "step": 12880
    },
    {
      "dynamics_loss_avg": 0.05235541686415672,
      "epoch": 1.22833251953125e-05,
      "step": 12880
    },
    {
      "action_loss_avg": 0.003280822825036012,
      "epoch": 1.22833251953125e-05,
      "step": 12880
    },
    {
      "epoch": 1.22833251953125e-05,
      "step": 12880,
      "training_step_time": 8.794311046600342
    },
    {
      "epoch": 1.2284278869628906e-05,
      "model_forward_time": 1.7356045246124268,
      "step": 12881
    },
    {
      "epoch": 1.2284278869628906e-05,
      "step": 12881,
      "training_step_time": 7.729562520980835
    },
    {
      "epoch": 1.2285232543945312e-05,
      "model_forward_time": 1.7230579853057861,
      "step": 12882
    },
    {
      "epoch": 1.2285232543945312e-05,
      "step": 12882,
      "training_step_time": 8.832111120223999
    },
    {
      "epoch": 1.2286186218261718e-05,
      "model_forward_time": 2.2722666263580322,
      "step": 12883
    },
    {
      "epoch": 1.2286186218261718e-05,
      "step": 12883,
      "training_step_time": 7.810131072998047
    },
    {
      "epoch": 1.2287139892578125e-05,
      "model_forward_time": 1.7411201000213623,
      "step": 12884
    },
    {
      "epoch": 1.2287139892578125e-05,
      "step": 12884,
      "training_step_time": 7.731937646865845
    },
    {
      "epoch": 1.2288093566894531e-05,
      "model_forward_time": 1.7337324619293213,
      "step": 12885
    },
    {
      "epoch": 1.2288093566894531e-05,
      "step": 12885,
      "training_step_time": 7.684776067733765
    },
    {
      "epoch": 1.2289047241210937e-05,
      "model_forward_time": 1.745610237121582,
      "step": 12886
    },
    {
      "epoch": 1.2289047241210937e-05,
      "step": 12886,
      "training_step_time": 7.711786508560181
    },
    {
      "epoch": 1.2290000915527344e-05,
      "model_forward_time": 1.7405917644500732,
      "step": 12887
    },
    {
      "epoch": 1.2290000915527344e-05,
      "step": 12887,
      "training_step_time": 7.715953826904297
    },
    {
      "epoch": 1.229095458984375e-05,
      "model_forward_time": 1.7377371788024902,
      "step": 12888
    },
    {
      "epoch": 1.229095458984375e-05,
      "step": 12888,
      "training_step_time": 7.651460409164429
    },
    {
      "epoch": 1.2291908264160156e-05,
      "model_forward_time": 1.7926671504974365,
      "step": 12889
    },
    {
      "epoch": 1.2291908264160156e-05,
      "step": 12889,
      "training_step_time": 8.647195100784302
    },
    {
      "epoch": 1.2292861938476563e-05,
      "grad_norm": 0.2553100883960724,
      "learning_rate": 9.830811103450286e-06,
      "loss": 0.0697,
      "step": 12890
    },
    {
      "epoch": 1.2292861938476563e-05,
      "model_forward_time": 1.7391448020935059,
      "step": 12890
    },
    {
      "dynamics_loss_avg": 0.05611440576612949,
      "epoch": 1.2292861938476563e-05,
      "step": 12890
    },
    {
      "action_loss_avg": 0.0029380062012933196,
      "epoch": 1.2292861938476563e-05,
      "step": 12890
    },
    {
      "epoch": 1.2292861938476563e-05,
      "step": 12890,
      "training_step_time": 7.782445430755615
    },
    {
      "epoch": 1.2293815612792969e-05,
      "model_forward_time": 1.7266170978546143,
      "step": 12891
    },
    {
      "epoch": 1.2293815612792969e-05,
      "step": 12891,
      "training_step_time": 7.763537168502808
    },
    {
      "epoch": 1.2294769287109374e-05,
      "model_forward_time": 2.2892229557037354,
      "step": 12892
    },
    {
      "epoch": 1.2294769287109374e-05,
      "step": 12892,
      "training_step_time": 7.842477321624756
    },
    {
      "epoch": 1.2295722961425782e-05,
      "model_forward_time": 1.732668161392212,
      "step": 12893
    },
    {
      "epoch": 1.2295722961425782e-05,
      "step": 12893,
      "training_step_time": 8.245377540588379
    },
    {
      "epoch": 1.2296676635742188e-05,
      "model_forward_time": 1.7401783466339111,
      "step": 12894
    },
    {
      "epoch": 1.2296676635742188e-05,
      "step": 12894,
      "training_step_time": 7.6597723960876465
    },
    {
      "epoch": 1.2297630310058593e-05,
      "model_forward_time": 1.7358512878417969,
      "step": 12895
    },
    {
      "epoch": 1.2297630310058593e-05,
      "step": 12895,
      "training_step_time": 8.636849164962769
    },
    {
      "epoch": 1.2298583984375e-05,
      "model_forward_time": 1.7723138332366943,
      "step": 12896
    },
    {
      "epoch": 1.2298583984375e-05,
      "step": 12896,
      "training_step_time": 7.7083752155303955
    },
    {
      "epoch": 1.2299537658691406e-05,
      "model_forward_time": 1.7283029556274414,
      "step": 12897
    },
    {
      "epoch": 1.2299537658691406e-05,
      "step": 12897,
      "training_step_time": 8.65649127960205
    },
    {
      "epoch": 1.2300491333007812e-05,
      "model_forward_time": 1.7714593410491943,
      "step": 12898
    },
    {
      "epoch": 1.2300491333007812e-05,
      "step": 12898,
      "training_step_time": 7.725338697433472
    },
    {
      "epoch": 1.2301445007324218e-05,
      "model_forward_time": 1.7447004318237305,
      "step": 12899
    },
    {
      "epoch": 1.2301445007324218e-05,
      "step": 12899,
      "training_step_time": 8.721758365631104
    },
    {
      "epoch": 1.2302398681640625e-05,
      "grad_norm": 0.23689451813697815,
      "learning_rate": 9.830384351257924e-06,
      "loss": 0.0719,
      "step": 12900
    },
    {
      "epoch": 1.2302398681640625e-05,
      "model_forward_time": 1.7371270656585693,
      "step": 12900
    },
    {
      "dynamics_loss_avg": 0.05488335397094488,
      "epoch": 1.2302398681640625e-05,
      "step": 12900
    },
    {
      "action_loss_avg": 0.0020933788095135243,
      "epoch": 1.2302398681640625e-05,
      "step": 12900
    },
    {
      "epoch": 1.2302398681640625e-05,
      "step": 12900,
      "training_step_time": 7.73811411857605
    },
    {
      "epoch": 1.2303352355957031e-05,
      "model_forward_time": 2.2948646545410156,
      "step": 12901
    },
    {
      "epoch": 1.2303352355957031e-05,
      "step": 12901,
      "training_step_time": 8.824368953704834
    },
    {
      "epoch": 1.2304306030273437e-05,
      "model_forward_time": 1.7190368175506592,
      "step": 12902
    },
    {
      "epoch": 1.2304306030273437e-05,
      "step": 12902,
      "training_step_time": 8.458923578262329
    },
    {
      "epoch": 1.2305259704589844e-05,
      "model_forward_time": 1.7329308986663818,
      "step": 12903
    },
    {
      "epoch": 1.2305259704589844e-05,
      "step": 12903,
      "training_step_time": 8.550590753555298
    },
    {
      "epoch": 1.230621337890625e-05,
      "model_forward_time": 1.7334191799163818,
      "step": 12904
    },
    {
      "epoch": 1.230621337890625e-05,
      "step": 12904,
      "training_step_time": 7.865854024887085
    },
    {
      "epoch": 1.2307167053222656e-05,
      "model_forward_time": 1.7314724922180176,
      "step": 12905
    },
    {
      "epoch": 1.2307167053222656e-05,
      "step": 12905,
      "training_step_time": 7.675123453140259
    },
    {
      "epoch": 1.2308120727539063e-05,
      "model_forward_time": 1.7413554191589355,
      "step": 12906
    },
    {
      "epoch": 1.2308120727539063e-05,
      "step": 12906,
      "training_step_time": 8.824966430664062
    },
    {
      "epoch": 1.2309074401855469e-05,
      "model_forward_time": 1.7733039855957031,
      "step": 12907
    },
    {
      "epoch": 1.2309074401855469e-05,
      "step": 12907,
      "training_step_time": 7.886536598205566
    },
    {
      "epoch": 1.2310028076171874e-05,
      "model_forward_time": 1.7326760292053223,
      "step": 12908
    },
    {
      "epoch": 1.2310028076171874e-05,
      "step": 12908,
      "training_step_time": 7.695459604263306
    },
    {
      "epoch": 1.2310981750488282e-05,
      "model_forward_time": 1.732131004333496,
      "step": 12909
    },
    {
      "epoch": 1.2310981750488282e-05,
      "step": 12909,
      "training_step_time": 8.981655597686768
    },
    {
      "epoch": 1.2311935424804688e-05,
      "grad_norm": 0.24448879063129425,
      "learning_rate": 9.829957070821993e-06,
      "loss": 0.0703,
      "step": 12910
    },
    {
      "epoch": 1.2311935424804688e-05,
      "model_forward_time": 1.7287156581878662,
      "step": 12910
    },
    {
      "dynamics_loss_avg": 0.06539221368730068,
      "epoch": 1.2311935424804688e-05,
      "step": 12910
    },
    {
      "action_loss_avg": 0.0034987310296855866,
      "epoch": 1.2311935424804688e-05,
      "step": 12910
    },
    {
      "epoch": 1.2311935424804688e-05,
      "step": 12910,
      "training_step_time": 9.87915849685669
    },
    {
      "epoch": 1.2312889099121093e-05,
      "model_forward_time": 1.7354776859283447,
      "step": 12911
    },
    {
      "epoch": 1.2312889099121093e-05,
      "step": 12911,
      "training_step_time": 7.732433557510376
    },
    {
      "epoch": 1.23138427734375e-05,
      "model_forward_time": 1.7310354709625244,
      "step": 12912
    },
    {
      "epoch": 1.23138427734375e-05,
      "step": 12912,
      "training_step_time": 7.714235067367554
    },
    {
      "epoch": 1.2314796447753906e-05,
      "model_forward_time": 1.7437374591827393,
      "step": 12913
    },
    {
      "epoch": 1.2314796447753906e-05,
      "step": 12913,
      "training_step_time": 8.392263174057007
    },
    {
      "epoch": 1.2315750122070312e-05,
      "model_forward_time": 1.732483148574829,
      "step": 12914
    },
    {
      "epoch": 1.2315750122070312e-05,
      "step": 12914,
      "training_step_time": 8.485817670822144
    },
    {
      "epoch": 1.2316703796386718e-05,
      "model_forward_time": 1.7404825687408447,
      "step": 12915
    },
    {
      "epoch": 1.2316703796386718e-05,
      "step": 12915,
      "training_step_time": 7.820403099060059
    },
    {
      "epoch": 1.2317657470703125e-05,
      "model_forward_time": 1.7333190441131592,
      "step": 12916
    },
    {
      "epoch": 1.2317657470703125e-05,
      "step": 12916,
      "training_step_time": 7.995153188705444
    },
    {
      "epoch": 1.2318611145019531e-05,
      "model_forward_time": 1.7885901927947998,
      "step": 12917
    },
    {
      "epoch": 1.2318611145019531e-05,
      "step": 12917,
      "training_step_time": 7.757436513900757
    },
    {
      "epoch": 1.2319564819335937e-05,
      "model_forward_time": 1.3691043853759766,
      "step": 12918
    },
    {
      "epoch": 1.2319564819335937e-05,
      "step": 12918,
      "training_step_time": 8.214434146881104
    },
    {
      "epoch": 1.2320518493652344e-05,
      "model_forward_time": 1.7280552387237549,
      "step": 12919
    },
    {
      "epoch": 1.2320518493652344e-05,
      "step": 12919,
      "training_step_time": 7.783917427062988
    },
    {
      "epoch": 1.232147216796875e-05,
      "grad_norm": 0.27069762349128723,
      "learning_rate": 9.829529262189219e-06,
      "loss": 0.0637,
      "step": 12920
    },
    {
      "epoch": 1.232147216796875e-05,
      "model_forward_time": 2.2717981338500977,
      "step": 12920
    },
    {
      "dynamics_loss_avg": 0.05804200880229473,
      "epoch": 1.232147216796875e-05,
      "step": 12920
    },
    {
      "action_loss_avg": 0.0030722923635039478,
      "epoch": 1.232147216796875e-05,
      "step": 12920
    },
    {
      "epoch": 1.232147216796875e-05,
      "step": 12920,
      "training_step_time": 7.772367000579834
    },
    {
      "epoch": 1.2322425842285156e-05,
      "model_forward_time": 1.7388577461242676,
      "step": 12921
    },
    {
      "epoch": 1.2322425842285156e-05,
      "step": 12921,
      "training_step_time": 9.353761434555054
    },
    {
      "epoch": 1.2323379516601563e-05,
      "model_forward_time": 1.737396001815796,
      "step": 12922
    },
    {
      "epoch": 1.2323379516601563e-05,
      "step": 12922,
      "training_step_time": 7.7508790493011475
    },
    {
      "epoch": 1.2324333190917969e-05,
      "model_forward_time": 1.7403552532196045,
      "step": 12923
    },
    {
      "epoch": 1.2324333190917969e-05,
      "step": 12923,
      "training_step_time": 7.724093437194824
    },
    {
      "epoch": 1.2325286865234375e-05,
      "model_forward_time": 1.7358770370483398,
      "step": 12924
    },
    {
      "epoch": 1.2325286865234375e-05,
      "step": 12924,
      "training_step_time": 8.744895696640015
    },
    {
      "epoch": 1.2326240539550782e-05,
      "model_forward_time": 1.7203571796417236,
      "step": 12925
    },
    {
      "epoch": 1.2326240539550782e-05,
      "step": 12925,
      "training_step_time": 7.910768747329712
    },
    {
      "epoch": 1.2327194213867188e-05,
      "model_forward_time": 1.7818176746368408,
      "step": 12926
    },
    {
      "epoch": 1.2327194213867188e-05,
      "step": 12926,
      "training_step_time": 8.6797456741333
    },
    {
      "epoch": 1.2328147888183593e-05,
      "model_forward_time": 1.744659185409546,
      "step": 12927
    },
    {
      "epoch": 1.2328147888183593e-05,
      "step": 12927,
      "training_step_time": 7.562902212142944
    },
    {
      "epoch": 1.23291015625e-05,
      "model_forward_time": 1.7426321506500244,
      "step": 12928
    },
    {
      "epoch": 1.23291015625e-05,
      "step": 12928,
      "training_step_time": 8.701326370239258
    },
    {
      "epoch": 1.2330055236816407e-05,
      "model_forward_time": 2.2778215408325195,
      "step": 12929
    },
    {
      "epoch": 1.2330055236816407e-05,
      "step": 12929,
      "training_step_time": 7.580495834350586
    },
    {
      "epoch": 1.2331008911132812e-05,
      "grad_norm": 0.2059771865606308,
      "learning_rate": 9.829100925406386e-06,
      "loss": 0.0673,
      "step": 12930
    },
    {
      "epoch": 1.2331008911132812e-05,
      "model_forward_time": 1.7537705898284912,
      "step": 12930
    },
    {
      "dynamics_loss_avg": 0.06493509896099567,
      "epoch": 1.2331008911132812e-05,
      "step": 12930
    },
    {
      "action_loss_avg": 0.003753491910174489,
      "epoch": 1.2331008911132812e-05,
      "step": 12930
    },
    {
      "epoch": 1.2331008911132812e-05,
      "step": 12930,
      "training_step_time": 7.535414934158325
    },
    {
      "epoch": 1.2331962585449218e-05,
      "model_forward_time": 1.7448503971099854,
      "step": 12931
    },
    {
      "epoch": 1.2331962585449218e-05,
      "step": 12931,
      "training_step_time": 7.571784019470215
    },
    {
      "epoch": 1.2332916259765625e-05,
      "model_forward_time": 1.7442772388458252,
      "step": 12932
    },
    {
      "epoch": 1.2332916259765625e-05,
      "step": 12932,
      "training_step_time": 7.567888498306274
    },
    {
      "epoch": 1.2333869934082031e-05,
      "model_forward_time": 1.746755599975586,
      "step": 12933
    },
    {
      "epoch": 1.2333869934082031e-05,
      "step": 12933,
      "training_step_time": 7.553583383560181
    },
    {
      "epoch": 1.2334823608398437e-05,
      "model_forward_time": 1.7431540489196777,
      "step": 12934
    },
    {
      "epoch": 1.2334823608398437e-05,
      "step": 12934,
      "training_step_time": 7.66058349609375
    },
    {
      "epoch": 1.2335777282714844e-05,
      "model_forward_time": 1.7916855812072754,
      "step": 12935
    },
    {
      "epoch": 1.2335777282714844e-05,
      "step": 12935,
      "training_step_time": 7.791168689727783
    },
    {
      "epoch": 1.233673095703125e-05,
      "model_forward_time": 1.7381389141082764,
      "step": 12936
    },
    {
      "epoch": 1.233673095703125e-05,
      "step": 12936,
      "training_step_time": 8.561099529266357
    },
    {
      "epoch": 1.2337684631347656e-05,
      "model_forward_time": 1.3737163543701172,
      "step": 12937
    },
    {
      "epoch": 1.2337684631347656e-05,
      "step": 12937,
      "training_step_time": 9.652606725692749
    },
    {
      "epoch": 1.2338638305664063e-05,
      "model_forward_time": 2.281327486038208,
      "step": 12938
    },
    {
      "epoch": 1.2338638305664063e-05,
      "step": 12938,
      "training_step_time": 7.784246921539307
    },
    {
      "epoch": 1.2339591979980469e-05,
      "model_forward_time": 1.7428100109100342,
      "step": 12939
    },
    {
      "epoch": 1.2339591979980469e-05,
      "step": 12939,
      "training_step_time": 7.649008750915527
    },
    {
      "epoch": 1.2340545654296875e-05,
      "grad_norm": 0.19593489170074463,
      "learning_rate": 9.828672060520333e-06,
      "loss": 0.0692,
      "step": 12940
    },
    {
      "epoch": 1.2340545654296875e-05,
      "model_forward_time": 1.7290639877319336,
      "step": 12940
    },
    {
      "dynamics_loss_avg": 0.050882446207106115,
      "epoch": 1.2340545654296875e-05,
      "step": 12940
    },
    {
      "action_loss_avg": 0.0033496671239845455,
      "epoch": 1.2340545654296875e-05,
      "step": 12940
    },
    {
      "epoch": 1.2340545654296875e-05,
      "step": 12940,
      "training_step_time": 7.707889556884766
    },
    {
      "epoch": 1.2341499328613282e-05,
      "model_forward_time": 1.728755235671997,
      "step": 12941
    },
    {
      "epoch": 1.2341499328613282e-05,
      "step": 12941,
      "training_step_time": 7.812408447265625
    },
    {
      "epoch": 1.2342453002929688e-05,
      "model_forward_time": 1.7442984580993652,
      "step": 12942
    },
    {
      "epoch": 1.2342453002929688e-05,
      "step": 12942,
      "training_step_time": 7.683620452880859
    },
    {
      "epoch": 1.2343406677246093e-05,
      "model_forward_time": 1.7416768074035645,
      "step": 12943
    },
    {
      "epoch": 1.2343406677246093e-05,
      "step": 12943,
      "training_step_time": 7.605918645858765
    },
    {
      "epoch": 1.23443603515625e-05,
      "model_forward_time": 1.7861742973327637,
      "step": 12944
    },
    {
      "epoch": 1.23443603515625e-05,
      "step": 12944,
      "training_step_time": 8.458657264709473
    },
    {
      "epoch": 1.2345314025878907e-05,
      "model_forward_time": 1.74017333984375,
      "step": 12945
    },
    {
      "epoch": 1.2345314025878907e-05,
      "step": 12945,
      "training_step_time": 7.738832473754883
    },
    {
      "epoch": 1.2346267700195312e-05,
      "model_forward_time": 1.7433044910430908,
      "step": 12946
    },
    {
      "epoch": 1.2346267700195312e-05,
      "step": 12946,
      "training_step_time": 7.608671426773071
    },
    {
      "epoch": 1.2347221374511718e-05,
      "model_forward_time": 1.7455344200134277,
      "step": 12947
    },
    {
      "epoch": 1.2347221374511718e-05,
      "step": 12947,
      "training_step_time": 8.685399770736694
    },
    {
      "epoch": 1.2348175048828125e-05,
      "model_forward_time": 1.735428810119629,
      "step": 12948
    },
    {
      "epoch": 1.2348175048828125e-05,
      "step": 12948,
      "training_step_time": 7.600103855133057
    },
    {
      "epoch": 1.2349128723144531e-05,
      "model_forward_time": 1.7452828884124756,
      "step": 12949
    },
    {
      "epoch": 1.2349128723144531e-05,
      "step": 12949,
      "training_step_time": 8.661649703979492
    },
    {
      "epoch": 1.2350082397460937e-05,
      "grad_norm": 0.20327609777450562,
      "learning_rate": 9.828242667577967e-06,
      "loss": 0.0694,
      "step": 12950
    },
    {
      "epoch": 1.2350082397460937e-05,
      "model_forward_time": 1.760805368423462,
      "step": 12950
    },
    {
      "dynamics_loss_avg": 0.044631263520568606,
      "epoch": 1.2350082397460937e-05,
      "step": 12950
    },
    {
      "action_loss_avg": 0.003552272915840149,
      "epoch": 1.2350082397460937e-05,
      "step": 12950
    },
    {
      "epoch": 1.2350082397460937e-05,
      "step": 12950,
      "training_step_time": 7.706934690475464
    },
    {
      "epoch": 1.2351036071777344e-05,
      "model_forward_time": 1.7583651542663574,
      "step": 12951
    },
    {
      "epoch": 1.2351036071777344e-05,
      "step": 12951,
      "training_step_time": 7.457907199859619
    },
    {
      "epoch": 1.235198974609375e-05,
      "model_forward_time": 1.7680165767669678,
      "step": 12952
    },
    {
      "epoch": 1.235198974609375e-05,
      "step": 12952,
      "training_step_time": 7.52864670753479
    },
    {
      "epoch": 1.2352943420410156e-05,
      "model_forward_time": 1.8094825744628906,
      "step": 12953
    },
    {
      "epoch": 1.2352943420410156e-05,
      "step": 12953,
      "training_step_time": 7.541042327880859
    },
    {
      "epoch": 1.2353897094726563e-05,
      "model_forward_time": 1.755723237991333,
      "step": 12954
    },
    {
      "epoch": 1.2353897094726563e-05,
      "step": 12954,
      "training_step_time": 7.616865634918213
    },
    {
      "epoch": 1.2354850769042969e-05,
      "model_forward_time": 1.7630505561828613,
      "step": 12955
    },
    {
      "epoch": 1.2354850769042969e-05,
      "step": 12955,
      "training_step_time": 9.356622457504272
    },
    {
      "epoch": 1.2355804443359375e-05,
      "model_forward_time": 1.3852050304412842,
      "step": 12956
    },
    {
      "epoch": 1.2355804443359375e-05,
      "step": 12956,
      "training_step_time": 8.584994554519653
    },
    {
      "epoch": 1.2356758117675782e-05,
      "model_forward_time": 2.3790125846862793,
      "step": 12957
    },
    {
      "epoch": 1.2356758117675782e-05,
      "step": 12957,
      "training_step_time": 8.45928692817688
    },
    {
      "epoch": 1.2357711791992188e-05,
      "model_forward_time": 1.7509949207305908,
      "step": 12958
    },
    {
      "epoch": 1.2357711791992188e-05,
      "step": 12958,
      "training_step_time": 8.539976119995117
    },
    {
      "epoch": 1.2358665466308593e-05,
      "model_forward_time": 1.373507022857666,
      "step": 12959
    },
    {
      "epoch": 1.2358665466308593e-05,
      "step": 12959,
      "training_step_time": 7.710472345352173
    },
    {
      "epoch": 1.2359619140625e-05,
      "grad_norm": 0.21224918961524963,
      "learning_rate": 9.82781274662624e-06,
      "loss": 0.0656,
      "step": 12960
    },
    {
      "epoch": 1.2359619140625e-05,
      "model_forward_time": 1.7465903759002686,
      "step": 12960
    },
    {
      "dynamics_loss_avg": 0.07365418002009391,
      "epoch": 1.2359619140625e-05,
      "step": 12960
    },
    {
      "action_loss_avg": 0.00420302675338462,
      "epoch": 1.2359619140625e-05,
      "step": 12960
    },
    {
      "epoch": 1.2359619140625e-05,
      "step": 12960,
      "training_step_time": 7.494996070861816
    },
    {
      "epoch": 1.2360572814941407e-05,
      "model_forward_time": 1.7428441047668457,
      "step": 12961
    },
    {
      "epoch": 1.2360572814941407e-05,
      "step": 12961,
      "training_step_time": 7.568820238113403
    },
    {
      "epoch": 1.2361526489257812e-05,
      "model_forward_time": 1.7677381038665771,
      "step": 12962
    },
    {
      "epoch": 1.2361526489257812e-05,
      "step": 12962,
      "training_step_time": 7.603224992752075
    },
    {
      "epoch": 1.2362480163574218e-05,
      "model_forward_time": 1.7406291961669922,
      "step": 12963
    },
    {
      "epoch": 1.2362480163574218e-05,
      "step": 12963,
      "training_step_time": 7.67046332359314
    },
    {
      "epoch": 1.2363433837890625e-05,
      "model_forward_time": 1.7421321868896484,
      "step": 12964
    },
    {
      "epoch": 1.2363433837890625e-05,
      "step": 12964,
      "training_step_time": 7.6703267097473145
    },
    {
      "epoch": 1.2364387512207031e-05,
      "model_forward_time": 1.743018627166748,
      "step": 12965
    },
    {
      "epoch": 1.2364387512207031e-05,
      "step": 12965,
      "training_step_time": 7.66368293762207
    },
    {
      "epoch": 1.2365341186523437e-05,
      "model_forward_time": 2.27778959274292,
      "step": 12966
    },
    {
      "epoch": 1.2365341186523437e-05,
      "step": 12966,
      "training_step_time": 8.67665696144104
    },
    {
      "epoch": 1.2366294860839844e-05,
      "model_forward_time": 1.742220401763916,
      "step": 12967
    },
    {
      "epoch": 1.2366294860839844e-05,
      "step": 12967,
      "training_step_time": 8.615350246429443
    },
    {
      "epoch": 1.236724853515625e-05,
      "model_forward_time": 1.7574732303619385,
      "step": 12968
    },
    {
      "epoch": 1.236724853515625e-05,
      "step": 12968,
      "training_step_time": 8.539237260818481
    },
    {
      "epoch": 1.2368202209472656e-05,
      "model_forward_time": 1.7382192611694336,
      "step": 12969
    },
    {
      "epoch": 1.2368202209472656e-05,
      "step": 12969,
      "training_step_time": 7.476334810256958
    },
    {
      "epoch": 1.2369155883789063e-05,
      "grad_norm": 0.17663928866386414,
      "learning_rate": 9.82738229771217e-06,
      "loss": 0.0647,
      "step": 12970
    },
    {
      "epoch": 1.2369155883789063e-05,
      "model_forward_time": 1.7435111999511719,
      "step": 12970
    },
    {
      "dynamics_loss_avg": 0.07386095263063908,
      "epoch": 1.2369155883789063e-05,
      "step": 12970
    },
    {
      "action_loss_avg": 0.00552443228662014,
      "epoch": 1.2369155883789063e-05,
      "step": 12970
    },
    {
      "epoch": 1.2369155883789063e-05,
      "step": 12970,
      "training_step_time": 8.11132264137268
    },
    {
      "epoch": 1.2370109558105469e-05,
      "model_forward_time": 1.7464227676391602,
      "step": 12971
    },
    {
      "epoch": 1.2370109558105469e-05,
      "step": 12971,
      "training_step_time": 8.777141571044922
    },
    {
      "epoch": 1.2371063232421875e-05,
      "model_forward_time": 1.7447428703308105,
      "step": 12972
    },
    {
      "epoch": 1.2371063232421875e-05,
      "step": 12972,
      "training_step_time": 9.227802753448486
    },
    {
      "epoch": 1.2372016906738282e-05,
      "model_forward_time": 1.7440595626831055,
      "step": 12973
    },
    {
      "epoch": 1.2372016906738282e-05,
      "step": 12973,
      "training_step_time": 7.658113956451416
    },
    {
      "epoch": 1.2372970581054688e-05,
      "model_forward_time": 1.7525267601013184,
      "step": 12974
    },
    {
      "epoch": 1.2372970581054688e-05,
      "step": 12974,
      "training_step_time": 8.496635913848877
    },
    {
      "epoch": 1.2373924255371093e-05,
      "model_forward_time": 2.274348258972168,
      "step": 12975
    },
    {
      "epoch": 1.2373924255371093e-05,
      "step": 12975,
      "training_step_time": 7.6191935539245605
    },
    {
      "epoch": 1.23748779296875e-05,
      "model_forward_time": 1.7445292472839355,
      "step": 12976
    },
    {
      "epoch": 1.23748779296875e-05,
      "step": 12976,
      "training_step_time": 8.327540636062622
    },
    {
      "epoch": 1.2375831604003907e-05,
      "model_forward_time": 1.753986120223999,
      "step": 12977
    },
    {
      "epoch": 1.2375831604003907e-05,
      "step": 12977,
      "training_step_time": 7.667134761810303
    },
    {
      "epoch": 1.2376785278320312e-05,
      "model_forward_time": 1.75406813621521,
      "step": 12978
    },
    {
      "epoch": 1.2376785278320312e-05,
      "step": 12978,
      "training_step_time": 9.73072600364685
    },
    {
      "epoch": 1.2377738952636718e-05,
      "model_forward_time": 1.7574150562286377,
      "step": 12979
    },
    {
      "epoch": 1.2377738952636718e-05,
      "step": 12979,
      "training_step_time": 8.953465223312378
    },
    {
      "epoch": 1.2378692626953125e-05,
      "grad_norm": 0.1762295514345169,
      "learning_rate": 9.82695132088283e-06,
      "loss": 0.0656,
      "step": 12980
    },
    {
      "epoch": 1.2378692626953125e-05,
      "model_forward_time": 1.7557060718536377,
      "step": 12980
    },
    {
      "dynamics_loss_avg": 0.06489076968282462,
      "epoch": 1.2378692626953125e-05,
      "step": 12980
    },
    {
      "action_loss_avg": 0.0030614650197094305,
      "epoch": 1.2378692626953125e-05,
      "step": 12980
    },
    {
      "epoch": 1.2378692626953125e-05,
      "step": 12980,
      "training_step_time": 6.87769341468811
    },
    {
      "epoch": 1.2379646301269531e-05,
      "model_forward_time": 1.7530345916748047,
      "step": 12981
    },
    {
      "epoch": 1.2379646301269531e-05,
      "step": 12981,
      "training_step_time": 6.882426738739014
    },
    {
      "epoch": 1.2380599975585937e-05,
      "model_forward_time": 1.75547194480896,
      "step": 12982
    },
    {
      "epoch": 1.2380599975585937e-05,
      "step": 12982,
      "training_step_time": 6.90379786491394
    },
    {
      "epoch": 1.2381553649902344e-05,
      "model_forward_time": 1.7616398334503174,
      "step": 12983
    },
    {
      "epoch": 1.2381553649902344e-05,
      "step": 12983,
      "training_step_time": 6.83706521987915
    },
    {
      "epoch": 1.238250732421875e-05,
      "model_forward_time": 1.754990577697754,
      "step": 12984
    },
    {
      "epoch": 1.238250732421875e-05,
      "step": 12984,
      "training_step_time": 6.860594272613525
    },
    {
      "epoch": 1.2383460998535156e-05,
      "model_forward_time": 1.7521915435791016,
      "step": 12985
    },
    {
      "epoch": 1.2383460998535156e-05,
      "step": 12985,
      "training_step_time": 6.903060436248779
    },
    {
      "epoch": 1.2384414672851563e-05,
      "model_forward_time": 1.7909650802612305,
      "step": 12986
    },
    {
      "epoch": 1.2384414672851563e-05,
      "step": 12986,
      "training_step_time": 6.806349754333496
    },
    {
      "epoch": 1.2385368347167969e-05,
      "model_forward_time": 1.7495698928833008,
      "step": 12987
    },
    {
      "epoch": 1.2385368347167969e-05,
      "step": 12987,
      "training_step_time": 6.785003423690796
    },
    {
      "epoch": 1.2386322021484375e-05,
      "model_forward_time": 2.3500683307647705,
      "step": 12988
    },
    {
      "epoch": 1.2386322021484375e-05,
      "step": 12988,
      "training_step_time": 8.305782556533813
    },
    {
      "epoch": 1.2387275695800782e-05,
      "model_forward_time": 1.7475571632385254,
      "step": 12989
    },
    {
      "epoch": 1.2387275695800782e-05,
      "step": 12989,
      "training_step_time": 7.703920841217041
    },
    {
      "epoch": 1.2388229370117188e-05,
      "grad_norm": 0.2552522122859955,
      "learning_rate": 9.826519816185351e-06,
      "loss": 0.0698,
      "step": 12990
    },
    {
      "epoch": 1.2388229370117188e-05,
      "model_forward_time": 1.054208517074585,
      "step": 12990
    },
    {
      "dynamics_loss_avg": 0.057099363021552565,
      "epoch": 1.2388229370117188e-05,
      "step": 12990
    },
    {
      "action_loss_avg": 0.003340454481076449,
      "epoch": 1.2388229370117188e-05,
      "step": 12990
    },
    {
      "epoch": 1.2388229370117188e-05,
      "step": 12990,
      "training_step_time": 6.692433834075928
    },
    {
      "epoch": 1.2389183044433593e-05,
      "model_forward_time": 1.7510042190551758,
      "step": 12991
    },
    {
      "epoch": 1.2389183044433593e-05,
      "step": 12991,
      "training_step_time": 6.729898929595947
    },
    {
      "epoch": 1.239013671875e-05,
      "model_forward_time": 1.7533066272735596,
      "step": 12992
    },
    {
      "epoch": 1.239013671875e-05,
      "step": 12992,
      "training_step_time": 6.744138240814209
    },
    {
      "epoch": 1.2391090393066407e-05,
      "model_forward_time": 1.7518420219421387,
      "step": 12993
    },
    {
      "epoch": 1.2391090393066407e-05,
      "step": 12993,
      "training_step_time": 6.754079580307007
    },
    {
      "epoch": 1.2392044067382812e-05,
      "model_forward_time": 1.7490394115447998,
      "step": 12994
    },
    {
      "epoch": 1.2392044067382812e-05,
      "step": 12994,
      "training_step_time": 6.748420000076294
    },
    {
      "epoch": 1.2392997741699218e-05,
      "model_forward_time": 1.7543060779571533,
      "step": 12995
    },
    {
      "epoch": 1.2392997741699218e-05,
      "step": 12995,
      "training_step_time": 6.7558512687683105
    },
    {
      "epoch": 1.2393951416015625e-05,
      "model_forward_time": 1.758577823638916,
      "step": 12996
    },
    {
      "epoch": 1.2393951416015625e-05,
      "step": 12996,
      "training_step_time": 6.746473073959351
    },
    {
      "epoch": 1.2394905090332031e-05,
      "model_forward_time": 2.3529446125030518,
      "step": 12997
    },
    {
      "epoch": 1.2394905090332031e-05,
      "step": 12997,
      "training_step_time": 7.283588409423828
    },
    {
      "epoch": 1.2395858764648437e-05,
      "model_forward_time": 1.7953484058380127,
      "step": 12998
    },
    {
      "epoch": 1.2395858764648437e-05,
      "step": 12998,
      "training_step_time": 9.685677528381348
    },
    {
      "epoch": 1.2396812438964844e-05,
      "model_forward_time": 1.7381255626678467,
      "step": 12999
    },
    {
      "epoch": 1.2396812438964844e-05,
      "step": 12999,
      "training_step_time": 7.452011823654175
    },
    {
      "epoch": 1.239776611328125e-05,
      "grad_norm": 0.19097213447093964,
      "learning_rate": 9.82608778366692e-06,
      "loss": 0.0733,
      "step": 13000
    },
    {
      "epoch": 1.239776611328125e-05,
      "model_forward_time": 1.7389552593231201,
      "step": 13000
    },
    {
      "dynamics_loss_avg": 0.04786785440519452,
      "epoch": 1.239776611328125e-05,
      "step": 13000
    },
    {
      "action_loss_avg": 0.002788598963525146,
      "epoch": 1.239776611328125e-05,
      "step": 13000
    },
    {
      "epoch": 1.239776611328125e-05,
      "step": 13000,
      "training_step_time": 6.680156707763672
    },
    {
      "epoch": 1.2398719787597656e-05,
      "model_forward_time": 1.7527523040771484,
      "step": 13001
    },
    {
      "epoch": 1.2398719787597656e-05,
      "step": 13001,
      "training_step_time": 6.9077467918396
    },
    {
      "epoch": 1.2399673461914063e-05,
      "model_forward_time": 1.7662849426269531,
      "step": 13002
    },
    {
      "epoch": 1.2399673461914063e-05,
      "step": 13002,
      "training_step_time": 6.900512456893921
    },
    {
      "epoch": 1.2400627136230469e-05,
      "model_forward_time": 1.396759271621704,
      "step": 13003
    },
    {
      "epoch": 1.2400627136230469e-05,
      "step": 13003,
      "training_step_time": 6.898245573043823
    },
    {
      "epoch": 1.2401580810546875e-05,
      "model_forward_time": 1.7519261837005615,
      "step": 13004
    },
    {
      "epoch": 1.2401580810546875e-05,
      "step": 13004,
      "training_step_time": 6.887195110321045
    },
    {
      "epoch": 1.2402534484863282e-05,
      "model_forward_time": 1.7505135536193848,
      "step": 13005
    },
    {
      "epoch": 1.2402534484863282e-05,
      "step": 13005,
      "training_step_time": 6.8745458126068115
    },
    {
      "epoch": 1.2403488159179688e-05,
      "model_forward_time": 2.72941517829895,
      "step": 13006
    },
    {
      "epoch": 1.2403488159179688e-05,
      "step": 13006,
      "training_step_time": 8.727423191070557
    },
    {
      "epoch": 1.2404441833496093e-05,
      "model_forward_time": 1.7395238876342773,
      "step": 13007
    },
    {
      "epoch": 1.2404441833496093e-05,
      "step": 13007,
      "training_step_time": 11.537570476531982
    },
    {
      "epoch": 1.24053955078125e-05,
      "model_forward_time": 1.7517194747924805,
      "step": 13008
    },
    {
      "epoch": 1.24053955078125e-05,
      "step": 13008,
      "training_step_time": 7.9199607372283936
    },
    {
      "epoch": 1.2406349182128907e-05,
      "model_forward_time": 1.757702350616455,
      "step": 13009
    },
    {
      "epoch": 1.2406349182128907e-05,
      "step": 13009,
      "training_step_time": 6.906028747558594
    },
    {
      "epoch": 1.2407302856445312e-05,
      "grad_norm": 0.17559407651424408,
      "learning_rate": 9.825655223374787e-06,
      "loss": 0.0745,
      "step": 13010
    },
    {
      "epoch": 1.2407302856445312e-05,
      "model_forward_time": 1.7496707439422607,
      "step": 13010
    },
    {
      "dynamics_loss_avg": 0.060539484955370426,
      "epoch": 1.2407302856445312e-05,
      "step": 13010
    },
    {
      "action_loss_avg": 0.0020918927621096373,
      "epoch": 1.2407302856445312e-05,
      "step": 13010
    },
    {
      "epoch": 1.2407302856445312e-05,
      "step": 13010,
      "training_step_time": 6.86974024772644
    },
    {
      "epoch": 1.2408256530761718e-05,
      "model_forward_time": 1.7529098987579346,
      "step": 13011
    },
    {
      "epoch": 1.2408256530761718e-05,
      "step": 13011,
      "training_step_time": 6.879730939865112
    },
    {
      "epoch": 1.2409210205078125e-05,
      "model_forward_time": 1.7500700950622559,
      "step": 13012
    },
    {
      "epoch": 1.2409210205078125e-05,
      "step": 13012,
      "training_step_time": 6.851814270019531
    },
    {
      "epoch": 1.2410163879394531e-05,
      "model_forward_time": 1.7482950687408447,
      "step": 13013
    },
    {
      "epoch": 1.2410163879394531e-05,
      "step": 13013,
      "training_step_time": 6.8429155349731445
    },
    {
      "epoch": 1.2411117553710937e-05,
      "model_forward_time": 1.7498564720153809,
      "step": 13014
    },
    {
      "epoch": 1.2411117553710937e-05,
      "step": 13014,
      "training_step_time": 6.898086786270142
    },
    {
      "epoch": 1.2412071228027344e-05,
      "model_forward_time": 2.28407883644104,
      "step": 13015
    },
    {
      "epoch": 1.2412071228027344e-05,
      "step": 13015,
      "training_step_time": 7.235930681228638
    },
    {
      "epoch": 1.241302490234375e-05,
      "model_forward_time": 1.764108657836914,
      "step": 13016
    },
    {
      "epoch": 1.241302490234375e-05,
      "step": 13016,
      "training_step_time": 7.856938362121582
    },
    {
      "epoch": 1.2413978576660156e-05,
      "model_forward_time": 1.7437708377838135,
      "step": 13017
    },
    {
      "epoch": 1.2413978576660156e-05,
      "step": 13017,
      "training_step_time": 7.855745553970337
    },
    {
      "epoch": 1.2414932250976563e-05,
      "model_forward_time": 1.753891944885254,
      "step": 13018
    },
    {
      "epoch": 1.2414932250976563e-05,
      "step": 13018,
      "training_step_time": 6.889254570007324
    },
    {
      "epoch": 1.2415885925292969e-05,
      "model_forward_time": 1.741204023361206,
      "step": 13019
    },
    {
      "epoch": 1.2415885925292969e-05,
      "step": 13019,
      "training_step_time": 6.8199005126953125
    },
    {
      "epoch": 1.2416839599609375e-05,
      "grad_norm": 0.2364208847284317,
      "learning_rate": 9.825222135356253e-06,
      "loss": 0.0726,
      "step": 13020
    },
    {
      "epoch": 1.2416839599609375e-05,
      "model_forward_time": 1.7469358444213867,
      "step": 13020
    },
    {
      "dynamics_loss_avg": 0.07828148361295462,
      "epoch": 1.2416839599609375e-05,
      "step": 13020
    },
    {
      "action_loss_avg": 0.0032959082978777587,
      "epoch": 1.2416839599609375e-05,
      "step": 13020
    },
    {
      "epoch": 1.2416839599609375e-05,
      "step": 13020,
      "training_step_time": 6.8509204387664795
    },
    {
      "epoch": 1.2417793273925782e-05,
      "model_forward_time": 1.7486369609832764,
      "step": 13021
    },
    {
      "epoch": 1.2417793273925782e-05,
      "step": 13021,
      "training_step_time": 6.838465929031372
    },
    {
      "epoch": 1.2418746948242188e-05,
      "model_forward_time": 1.745347023010254,
      "step": 13022
    },
    {
      "epoch": 1.2418746948242188e-05,
      "step": 13022,
      "training_step_time": 6.881000518798828
    },
    {
      "epoch": 1.2419700622558593e-05,
      "model_forward_time": 1.7420504093170166,
      "step": 13023
    },
    {
      "epoch": 1.2419700622558593e-05,
      "step": 13023,
      "training_step_time": 6.874335289001465
    },
    {
      "epoch": 1.2420654296875e-05,
      "model_forward_time": 2.283726215362549,
      "step": 13024
    },
    {
      "epoch": 1.2420654296875e-05,
      "step": 13024,
      "training_step_time": 7.2248265743255615
    },
    {
      "epoch": 1.2421607971191407e-05,
      "model_forward_time": 1.7851512432098389,
      "step": 13025
    },
    {
      "epoch": 1.2421607971191407e-05,
      "step": 13025,
      "training_step_time": 7.806262254714966
    },
    {
      "epoch": 1.2422561645507812e-05,
      "model_forward_time": 1.7643458843231201,
      "step": 13026
    },
    {
      "epoch": 1.2422561645507812e-05,
      "step": 13026,
      "training_step_time": 7.869968891143799
    },
    {
      "epoch": 1.2423515319824218e-05,
      "model_forward_time": 1.753021478652954,
      "step": 13027
    },
    {
      "epoch": 1.2423515319824218e-05,
      "step": 13027,
      "training_step_time": 6.914521217346191
    },
    {
      "epoch": 1.2424468994140625e-05,
      "model_forward_time": 1.758281946182251,
      "step": 13028
    },
    {
      "epoch": 1.2424468994140625e-05,
      "step": 13028,
      "training_step_time": 6.882391691207886
    },
    {
      "epoch": 1.2425422668457031e-05,
      "model_forward_time": 1.7542099952697754,
      "step": 13029
    },
    {
      "epoch": 1.2425422668457031e-05,
      "step": 13029,
      "training_step_time": 6.879721403121948
    },
    {
      "epoch": 1.2426376342773437e-05,
      "grad_norm": 0.2467137575149536,
      "learning_rate": 9.82478851965868e-06,
      "loss": 0.0662,
      "step": 13030
    },
    {
      "epoch": 1.2426376342773437e-05,
      "model_forward_time": 1.7622973918914795,
      "step": 13030
    },
    {
      "dynamics_loss_avg": 0.05471900533884764,
      "epoch": 1.2426376342773437e-05,
      "step": 13030
    },
    {
      "action_loss_avg": 0.003366433782503009,
      "epoch": 1.2426376342773437e-05,
      "step": 13030
    },
    {
      "epoch": 1.2426376342773437e-05,
      "step": 13030,
      "training_step_time": 6.93236517906189
    },
    {
      "epoch": 1.2427330017089844e-05,
      "model_forward_time": 1.7537078857421875,
      "step": 13031
    },
    {
      "epoch": 1.2427330017089844e-05,
      "step": 13031,
      "training_step_time": 6.9319562911987305
    },
    {
      "epoch": 1.242828369140625e-05,
      "model_forward_time": 1.7247705459594727,
      "step": 13032
    },
    {
      "epoch": 1.242828369140625e-05,
      "step": 13032,
      "training_step_time": 6.885339975357056
    },
    {
      "epoch": 1.2429237365722656e-05,
      "model_forward_time": 2.310030221939087,
      "step": 13033
    },
    {
      "epoch": 1.2429237365722656e-05,
      "step": 13033,
      "training_step_time": 7.220189332962036
    },
    {
      "epoch": 1.2430191040039063e-05,
      "model_forward_time": 1.7784020900726318,
      "step": 13034
    },
    {
      "epoch": 1.2430191040039063e-05,
      "step": 13034,
      "training_step_time": 8.833611488342285
    },
    {
      "epoch": 1.2431144714355469e-05,
      "model_forward_time": 1.7389984130859375,
      "step": 13035
    },
    {
      "epoch": 1.2431144714355469e-05,
      "step": 13035,
      "training_step_time": 8.758597373962402
    },
    {
      "epoch": 1.2432098388671875e-05,
      "model_forward_time": 1.741769552230835,
      "step": 13036
    },
    {
      "epoch": 1.2432098388671875e-05,
      "step": 13036,
      "training_step_time": 7.924323797225952
    },
    {
      "epoch": 1.2433052062988282e-05,
      "model_forward_time": 1.7590250968933105,
      "step": 13037
    },
    {
      "epoch": 1.2433052062988282e-05,
      "step": 13037,
      "training_step_time": 6.87960147857666
    },
    {
      "epoch": 1.2434005737304688e-05,
      "model_forward_time": 1.7444918155670166,
      "step": 13038
    },
    {
      "epoch": 1.2434005737304688e-05,
      "step": 13038,
      "training_step_time": 6.838117837905884
    },
    {
      "epoch": 1.2434959411621093e-05,
      "model_forward_time": 1.7440273761749268,
      "step": 13039
    },
    {
      "epoch": 1.2434959411621093e-05,
      "step": 13039,
      "training_step_time": 6.852950572967529
    },
    {
      "epoch": 1.24359130859375e-05,
      "grad_norm": 0.25268277525901794,
      "learning_rate": 9.82435437632949e-06,
      "loss": 0.0692,
      "step": 13040
    },
    {
      "epoch": 1.24359130859375e-05,
      "model_forward_time": 1.770444631576538,
      "step": 13040
    },
    {
      "dynamics_loss_avg": 0.05602968353778124,
      "epoch": 1.24359130859375e-05,
      "step": 13040
    },
    {
      "action_loss_avg": 0.003846234525553882,
      "epoch": 1.24359130859375e-05,
      "step": 13040
    },
    {
      "epoch": 1.24359130859375e-05,
      "step": 13040,
      "training_step_time": 6.943770885467529
    },
    {
      "epoch": 1.2436866760253907e-05,
      "model_forward_time": 1.7468631267547607,
      "step": 13041
    },
    {
      "epoch": 1.2436866760253907e-05,
      "step": 13041,
      "training_step_time": 6.901597023010254
    },
    {
      "epoch": 1.2437820434570312e-05,
      "model_forward_time": 2.2692017555236816,
      "step": 13042
    },
    {
      "epoch": 1.2437820434570312e-05,
      "step": 13042,
      "training_step_time": 7.244280815124512
    },
    {
      "epoch": 1.2438774108886718e-05,
      "model_forward_time": 1.3642420768737793,
      "step": 13043
    },
    {
      "epoch": 1.2438774108886718e-05,
      "step": 13043,
      "training_step_time": 8.651300191879272
    },
    {
      "epoch": 1.2439727783203125e-05,
      "model_forward_time": 1.7188291549682617,
      "step": 13044
    },
    {
      "epoch": 1.2439727783203125e-05,
      "step": 13044,
      "training_step_time": 9.813958406448364
    },
    {
      "epoch": 1.2440681457519531e-05,
      "model_forward_time": 1.7136108875274658,
      "step": 13045
    },
    {
      "epoch": 1.2440681457519531e-05,
      "step": 13045,
      "training_step_time": 7.761316537857056
    },
    {
      "epoch": 1.2441635131835937e-05,
      "model_forward_time": 1.7244603633880615,
      "step": 13046
    },
    {
      "epoch": 1.2441635131835937e-05,
      "step": 13046,
      "training_step_time": 6.884340524673462
    },
    {
      "epoch": 1.2442588806152344e-05,
      "model_forward_time": 1.031442642211914,
      "step": 13047
    },
    {
      "epoch": 1.2442588806152344e-05,
      "step": 13047,
      "training_step_time": 6.884062051773071
    },
    {
      "epoch": 1.244354248046875e-05,
      "model_forward_time": 1.7212097644805908,
      "step": 13048
    },
    {
      "epoch": 1.244354248046875e-05,
      "step": 13048,
      "training_step_time": 6.889740705490112
    },
    {
      "epoch": 1.2444496154785156e-05,
      "model_forward_time": 1.7346632480621338,
      "step": 13049
    },
    {
      "epoch": 1.2444496154785156e-05,
      "step": 13049,
      "training_step_time": 6.882338047027588
    },
    {
      "epoch": 1.2445449829101563e-05,
      "grad_norm": 0.314688503742218,
      "learning_rate": 9.823919705416158e-06,
      "loss": 0.0723,
      "step": 13050
    },
    {
      "epoch": 1.2445449829101563e-05,
      "model_forward_time": 1.7363371849060059,
      "step": 13050
    },
    {
      "dynamics_loss_avg": 0.05655442196875811,
      "epoch": 1.2445449829101563e-05,
      "step": 13050
    },
    {
      "action_loss_avg": 0.002792009781114757,
      "epoch": 1.2445449829101563e-05,
      "step": 13050
    },
    {
      "epoch": 1.2445449829101563e-05,
      "step": 13050,
      "training_step_time": 6.889949321746826
    },
    {
      "epoch": 1.2446403503417969e-05,
      "model_forward_time": 1.7414889335632324,
      "step": 13051
    },
    {
      "epoch": 1.2446403503417969e-05,
      "step": 13051,
      "training_step_time": 6.883625745773315
    },
    {
      "epoch": 1.2447357177734375e-05,
      "model_forward_time": 2.286738634109497,
      "step": 13052
    },
    {
      "epoch": 1.2447357177734375e-05,
      "step": 13052,
      "training_step_time": 7.79689884185791
    },
    {
      "epoch": 1.2448310852050782e-05,
      "model_forward_time": 1.7803101539611816,
      "step": 13053
    },
    {
      "epoch": 1.2448310852050782e-05,
      "step": 13053,
      "training_step_time": 11.131358623504639
    },
    {
      "epoch": 1.2449264526367188e-05,
      "model_forward_time": 1.7307403087615967,
      "step": 13054
    },
    {
      "epoch": 1.2449264526367188e-05,
      "step": 13054,
      "training_step_time": 7.858654499053955
    },
    {
      "epoch": 1.2450218200683593e-05,
      "model_forward_time": 1.7303857803344727,
      "step": 13055
    },
    {
      "epoch": 1.2450218200683593e-05,
      "step": 13055,
      "training_step_time": 6.955467700958252
    },
    {
      "epoch": 1.2451171875e-05,
      "model_forward_time": 1.761021614074707,
      "step": 13056
    },
    {
      "epoch": 1.2451171875e-05,
      "step": 13056,
      "training_step_time": 6.974025726318359
    },
    {
      "epoch": 1.2452125549316407e-05,
      "model_forward_time": 1.7373261451721191,
      "step": 13057
    },
    {
      "epoch": 1.2452125549316407e-05,
      "step": 13057,
      "training_step_time": 6.934296131134033
    },
    {
      "epoch": 1.2453079223632812e-05,
      "model_forward_time": 1.7269525527954102,
      "step": 13058
    },
    {
      "epoch": 1.2453079223632812e-05,
      "step": 13058,
      "training_step_time": 6.902048826217651
    },
    {
      "epoch": 1.2454032897949218e-05,
      "model_forward_time": 1.7104511260986328,
      "step": 13059
    },
    {
      "epoch": 1.2454032897949218e-05,
      "step": 13059,
      "training_step_time": 6.911006212234497
    },
    {
      "epoch": 1.2454986572265625e-05,
      "grad_norm": 0.3247981369495392,
      "learning_rate": 9.82348450696622e-06,
      "loss": 0.072,
      "step": 13060
    },
    {
      "epoch": 1.2454986572265625e-05,
      "model_forward_time": 1.7168052196502686,
      "step": 13060
    },
    {
      "dynamics_loss_avg": 0.047336832713335754,
      "epoch": 1.2454986572265625e-05,
      "step": 13060
    },
    {
      "action_loss_avg": 0.0031908402510453016,
      "epoch": 1.2454986572265625e-05,
      "step": 13060
    },
    {
      "epoch": 1.2454986572265625e-05,
      "step": 13060,
      "training_step_time": 6.929562568664551
    },
    {
      "epoch": 1.2455940246582031e-05,
      "model_forward_time": 2.3390402793884277,
      "step": 13061
    },
    {
      "epoch": 1.2455940246582031e-05,
      "step": 13061,
      "training_step_time": 7.755859375
    },
    {
      "epoch": 1.2456893920898437e-05,
      "model_forward_time": 1.7470831871032715,
      "step": 13062
    },
    {
      "epoch": 1.2456893920898437e-05,
      "step": 13062,
      "training_step_time": 9.80626893043518
    },
    {
      "epoch": 1.2457847595214844e-05,
      "model_forward_time": 1.7319889068603516,
      "step": 13063
    },
    {
      "epoch": 1.2457847595214844e-05,
      "step": 13063,
      "training_step_time": 8.723582029342651
    },
    {
      "epoch": 1.245880126953125e-05,
      "model_forward_time": 1.7217698097229004,
      "step": 13064
    },
    {
      "epoch": 1.245880126953125e-05,
      "step": 13064,
      "training_step_time": 6.870547771453857
    },
    {
      "epoch": 1.2459754943847656e-05,
      "model_forward_time": 1.7248733043670654,
      "step": 13065
    },
    {
      "epoch": 1.2459754943847656e-05,
      "step": 13065,
      "training_step_time": 6.843905210494995
    },
    {
      "epoch": 1.2460708618164063e-05,
      "model_forward_time": 1.7163667678833008,
      "step": 13066
    },
    {
      "epoch": 1.2460708618164063e-05,
      "step": 13066,
      "training_step_time": 6.853878021240234
    },
    {
      "epoch": 1.2461662292480469e-05,
      "model_forward_time": 1.720686674118042,
      "step": 13067
    },
    {
      "epoch": 1.2461662292480469e-05,
      "step": 13067,
      "training_step_time": 6.882499933242798
    },
    {
      "epoch": 1.2462615966796875e-05,
      "model_forward_time": 1.7084295749664307,
      "step": 13068
    },
    {
      "epoch": 1.2462615966796875e-05,
      "step": 13068,
      "training_step_time": 6.848368883132935
    },
    {
      "epoch": 1.2463569641113282e-05,
      "model_forward_time": 1.719308853149414,
      "step": 13069
    },
    {
      "epoch": 1.2463569641113282e-05,
      "step": 13069,
      "training_step_time": 6.832648515701294
    },
    {
      "epoch": 1.2464523315429688e-05,
      "grad_norm": 0.3417036831378937,
      "learning_rate": 9.82304878102727e-06,
      "loss": 0.0711,
      "step": 13070
    },
    {
      "epoch": 1.2464523315429688e-05,
      "model_forward_time": 1.8704273700714111,
      "step": 13070
    },
    {
      "dynamics_loss_avg": 0.05888613723218441,
      "epoch": 1.2464523315429688e-05,
      "step": 13070
    },
    {
      "action_loss_avg": 0.004019230417907238,
      "epoch": 1.2464523315429688e-05,
      "step": 13070
    },
    {
      "epoch": 1.2464523315429688e-05,
      "step": 13070,
      "training_step_time": 7.727050542831421
    },
    {
      "epoch": 1.2465476989746093e-05,
      "model_forward_time": 1.7155861854553223,
      "step": 13071
    },
    {
      "epoch": 1.2465476989746093e-05,
      "step": 13071,
      "training_step_time": 11.288447618484497
    },
    {
      "epoch": 1.24664306640625e-05,
      "model_forward_time": 1.7779791355133057,
      "step": 13072
    },
    {
      "epoch": 1.24664306640625e-05,
      "step": 13072,
      "training_step_time": 7.902406692504883
    },
    {
      "epoch": 1.2467384338378907e-05,
      "model_forward_time": 1.7165021896362305,
      "step": 13073
    },
    {
      "epoch": 1.2467384338378907e-05,
      "step": 13073,
      "training_step_time": 7.668973445892334
    },
    {
      "epoch": 1.2468338012695312e-05,
      "model_forward_time": 1.714341640472412,
      "step": 13074
    },
    {
      "epoch": 1.2468338012695312e-05,
      "step": 13074,
      "training_step_time": 6.888629674911499
    },
    {
      "epoch": 1.2469291687011718e-05,
      "model_forward_time": 1.7283756732940674,
      "step": 13075
    },
    {
      "epoch": 1.2469291687011718e-05,
      "step": 13075,
      "training_step_time": 6.842031478881836
    },
    {
      "epoch": 1.2470245361328125e-05,
      "model_forward_time": 1.3614513874053955,
      "step": 13076
    },
    {
      "epoch": 1.2470245361328125e-05,
      "step": 13076,
      "training_step_time": 6.814090728759766
    },
    {
      "epoch": 1.2471199035644531e-05,
      "model_forward_time": 1.7176401615142822,
      "step": 13077
    },
    {
      "epoch": 1.2471199035644531e-05,
      "step": 13077,
      "training_step_time": 6.84231972694397
    },
    {
      "epoch": 1.2472152709960937e-05,
      "model_forward_time": 1.7255232334136963,
      "step": 13078
    },
    {
      "epoch": 1.2472152709960937e-05,
      "step": 13078,
      "training_step_time": 6.852729797363281
    },
    {
      "epoch": 1.2473106384277344e-05,
      "model_forward_time": 1.7225112915039062,
      "step": 13079
    },
    {
      "epoch": 1.2473106384277344e-05,
      "step": 13079,
      "training_step_time": 7.623308420181274
    },
    {
      "epoch": 1.247406005859375e-05,
      "grad_norm": 0.3245602548122406,
      "learning_rate": 9.822612527646953e-06,
      "loss": 0.0688,
      "step": 13080
    },
    {
      "epoch": 1.247406005859375e-05,
      "model_forward_time": 2.2527687549591064,
      "step": 13080
    },
    {
      "dynamics_loss_avg": 0.06927669253200293,
      "epoch": 1.247406005859375e-05,
      "step": 13080
    },
    {
      "action_loss_avg": 0.005527961277402937,
      "epoch": 1.247406005859375e-05,
      "step": 13080
    },
    {
      "epoch": 1.247406005859375e-05,
      "step": 13080,
      "training_step_time": 8.579269886016846
    },
    {
      "epoch": 1.2475013732910156e-05,
      "model_forward_time": 1.7575533390045166,
      "step": 13081
    },
    {
      "epoch": 1.2475013732910156e-05,
      "step": 13081,
      "training_step_time": 9.776479244232178
    },
    {
      "epoch": 1.2475967407226563e-05,
      "model_forward_time": 1.7373073101043701,
      "step": 13082
    },
    {
      "epoch": 1.2475967407226563e-05,
      "step": 13082,
      "training_step_time": 7.71793270111084
    },
    {
      "epoch": 1.2476921081542969e-05,
      "model_forward_time": 1.0295426845550537,
      "step": 13083
    },
    {
      "epoch": 1.2476921081542969e-05,
      "step": 13083,
      "training_step_time": 6.905382871627808
    },
    {
      "epoch": 1.2477874755859375e-05,
      "model_forward_time": 1.73150634765625,
      "step": 13084
    },
    {
      "epoch": 1.2477874755859375e-05,
      "step": 13084,
      "training_step_time": 6.883356809616089
    },
    {
      "epoch": 1.2478828430175782e-05,
      "model_forward_time": 1.7301599979400635,
      "step": 13085
    },
    {
      "epoch": 1.2478828430175782e-05,
      "step": 13085,
      "training_step_time": 6.75234317779541
    },
    {
      "epoch": 1.2479782104492188e-05,
      "model_forward_time": 1.7170829772949219,
      "step": 13086
    },
    {
      "epoch": 1.2479782104492188e-05,
      "step": 13086,
      "training_step_time": 6.911370754241943
    },
    {
      "epoch": 1.2480735778808594e-05,
      "model_forward_time": 1.723365306854248,
      "step": 13087
    },
    {
      "epoch": 1.2480735778808594e-05,
      "step": 13087,
      "training_step_time": 6.9148969650268555
    },
    {
      "epoch": 1.2481689453125e-05,
      "model_forward_time": 1.7174906730651855,
      "step": 13088
    },
    {
      "epoch": 1.2481689453125e-05,
      "step": 13088,
      "training_step_time": 7.187730550765991
    },
    {
      "epoch": 1.2482643127441407e-05,
      "model_forward_time": 2.261687755584717,
      "step": 13089
    },
    {
      "epoch": 1.2482643127441407e-05,
      "step": 13089,
      "training_step_time": 8.730082035064697
    },
    {
      "epoch": 1.2483596801757812e-05,
      "grad_norm": 0.21247701346874237,
      "learning_rate": 9.822175746872984e-06,
      "loss": 0.0702,
      "step": 13090
    },
    {
      "epoch": 1.2483596801757812e-05,
      "model_forward_time": 1.7770984172821045,
      "step": 13090
    },
    {
      "dynamics_loss_avg": 0.0694590762257576,
      "epoch": 1.2483596801757812e-05,
      "step": 13090
    },
    {
      "action_loss_avg": 0.0036421028547920288,
      "epoch": 1.2483596801757812e-05,
      "step": 13090
    },
    {
      "epoch": 1.2483596801757812e-05,
      "step": 13090,
      "training_step_time": 9.667036771774292
    },
    {
      "epoch": 1.2484550476074218e-05,
      "model_forward_time": 1.7385690212249756,
      "step": 13091
    },
    {
      "epoch": 1.2484550476074218e-05,
      "step": 13091,
      "training_step_time": 7.76347017288208
    },
    {
      "epoch": 1.2485504150390626e-05,
      "model_forward_time": 1.733628273010254,
      "step": 13092
    },
    {
      "epoch": 1.2485504150390626e-05,
      "step": 13092,
      "training_step_time": 6.7784106731414795
    },
    {
      "epoch": 1.2486457824707031e-05,
      "model_forward_time": 1.7256591320037842,
      "step": 13093
    },
    {
      "epoch": 1.2486457824707031e-05,
      "step": 13093,
      "training_step_time": 6.881478548049927
    },
    {
      "epoch": 1.2487411499023437e-05,
      "model_forward_time": 1.7040119171142578,
      "step": 13094
    },
    {
      "epoch": 1.2487411499023437e-05,
      "step": 13094,
      "training_step_time": 6.886915922164917
    },
    {
      "epoch": 1.2488365173339844e-05,
      "model_forward_time": 1.724958896636963,
      "step": 13095
    },
    {
      "epoch": 1.2488365173339844e-05,
      "step": 13095,
      "training_step_time": 6.844793319702148
    },
    {
      "epoch": 1.248931884765625e-05,
      "model_forward_time": 1.7255549430847168,
      "step": 13096
    },
    {
      "epoch": 1.248931884765625e-05,
      "step": 13096,
      "training_step_time": 6.857441663742065
    },
    {
      "epoch": 1.2490272521972656e-05,
      "model_forward_time": 1.7236766815185547,
      "step": 13097
    },
    {
      "epoch": 1.2490272521972656e-05,
      "step": 13097,
      "training_step_time": 7.174862384796143
    },
    {
      "epoch": 1.2491226196289063e-05,
      "model_forward_time": 1.7333104610443115,
      "step": 13098
    },
    {
      "epoch": 1.2491226196289063e-05,
      "step": 13098,
      "training_step_time": 8.70987582206726
    },
    {
      "epoch": 1.2492179870605469e-05,
      "model_forward_time": 1.7781102657318115,
      "step": 13099
    },
    {
      "epoch": 1.2492179870605469e-05,
      "step": 13099,
      "training_step_time": 10.407581567764282
    },
    {
      "epoch": 1.2493133544921875e-05,
      "grad_norm": 0.27057716250419617,
      "learning_rate": 9.821738438753123e-06,
      "loss": 0.0733,
      "step": 13100
    },
    {
      "epoch": 1.2493133544921875e-05,
      "model_forward_time": 1.7202253341674805,
      "step": 13100
    },
    {
      "dynamics_loss_avg": 0.07900344543159007,
      "epoch": 1.2493133544921875e-05,
      "step": 13100
    },
    {
      "action_loss_avg": 0.0035281444434076548,
      "epoch": 1.2493133544921875e-05,
      "step": 13100
    },
    {
      "epoch": 1.2493133544921875e-05,
      "step": 13100,
      "training_step_time": 7.82308292388916
    },
    {
      "epoch": 1.2494087219238282e-05,
      "model_forward_time": 1.7325899600982666,
      "step": 13101
    },
    {
      "epoch": 1.2494087219238282e-05,
      "step": 13101,
      "training_step_time": 6.813578844070435
    },
    {
      "epoch": 1.2495040893554688e-05,
      "model_forward_time": 1.7151024341583252,
      "step": 13102
    },
    {
      "epoch": 1.2495040893554688e-05,
      "step": 13102,
      "training_step_time": 6.825798749923706
    },
    {
      "epoch": 1.2495994567871094e-05,
      "model_forward_time": 1.7191832065582275,
      "step": 13103
    },
    {
      "epoch": 1.2495994567871094e-05,
      "step": 13103,
      "training_step_time": 6.832409381866455
    },
    {
      "epoch": 1.24969482421875e-05,
      "model_forward_time": 1.725623369216919,
      "step": 13104
    },
    {
      "epoch": 1.24969482421875e-05,
      "step": 13104,
      "training_step_time": 6.858199596405029
    },
    {
      "epoch": 1.2497901916503907e-05,
      "model_forward_time": 1.726489782333374,
      "step": 13105
    },
    {
      "epoch": 1.2497901916503907e-05,
      "step": 13105,
      "training_step_time": 6.81158971786499
    },
    {
      "epoch": 1.2498855590820312e-05,
      "model_forward_time": 1.7015023231506348,
      "step": 13106
    },
    {
      "epoch": 1.2498855590820312e-05,
      "step": 13106,
      "training_step_time": 6.839677810668945
    },
    {
      "epoch": 1.2499809265136718e-05,
      "model_forward_time": 1.7117087841033936,
      "step": 13107
    },
    {
      "epoch": 1.2499809265136718e-05,
      "step": 13107,
      "training_step_time": 7.738748550415039
    },
    {
      "epoch": 1.2500762939453126e-05,
      "model_forward_time": 2.314497709274292,
      "step": 13108
    },
    {
      "epoch": 1.2500762939453126e-05,
      "step": 13108,
      "training_step_time": 9.836271524429321
    },
    {
      "epoch": 1.2501716613769531e-05,
      "model_forward_time": 1.7215242385864258,
      "step": 13109
    },
    {
      "epoch": 1.2501716613769531e-05,
      "step": 13109,
      "training_step_time": 7.8973588943481445
    },
    {
      "epoch": 1.2502670288085937e-05,
      "grad_norm": 0.27827247977256775,
      "learning_rate": 9.821300603335197e-06,
      "loss": 0.0718,
      "step": 13110
    },
    {
      "epoch": 1.2502670288085937e-05,
      "model_forward_time": 1.7169911861419678,
      "step": 13110
    },
    {
      "dynamics_loss_avg": 0.06400795672088862,
      "epoch": 1.2502670288085937e-05,
      "step": 13110
    },
    {
      "action_loss_avg": 0.0035482995794154704,
      "epoch": 1.2502670288085937e-05,
      "step": 13110
    },
    {
      "epoch": 1.2502670288085937e-05,
      "step": 13110,
      "training_step_time": 7.50766134262085
    },
    {
      "epoch": 1.2503623962402344e-05,
      "model_forward_time": 1.7187442779541016,
      "step": 13111
    },
    {
      "epoch": 1.2503623962402344e-05,
      "step": 13111,
      "training_step_time": 6.885867118835449
    },
    {
      "epoch": 1.250457763671875e-05,
      "model_forward_time": 1.7211387157440186,
      "step": 13112
    },
    {
      "epoch": 1.250457763671875e-05,
      "step": 13112,
      "training_step_time": 6.883605718612671
    },
    {
      "epoch": 1.2505531311035156e-05,
      "model_forward_time": 1.7222936153411865,
      "step": 13113
    },
    {
      "epoch": 1.2505531311035156e-05,
      "step": 13113,
      "training_step_time": 6.903120040893555
    },
    {
      "epoch": 1.2506484985351563e-05,
      "model_forward_time": 1.7336175441741943,
      "step": 13114
    },
    {
      "epoch": 1.2506484985351563e-05,
      "step": 13114,
      "training_step_time": 6.889812707901001
    },
    {
      "epoch": 1.2507438659667969e-05,
      "model_forward_time": 1.7549643516540527,
      "step": 13115
    },
    {
      "epoch": 1.2507438659667969e-05,
      "step": 13115,
      "training_step_time": 6.901224851608276
    },
    {
      "epoch": 1.2508392333984375e-05,
      "model_forward_time": 1.7236332893371582,
      "step": 13116
    },
    {
      "epoch": 1.2508392333984375e-05,
      "step": 13116,
      "training_step_time": 7.681814432144165
    },
    {
      "epoch": 1.2509346008300782e-05,
      "model_forward_time": 2.2793009281158447,
      "step": 13117
    },
    {
      "epoch": 1.2509346008300782e-05,
      "step": 13117,
      "training_step_time": 11.345022916793823
    },
    {
      "epoch": 1.2510299682617188e-05,
      "model_forward_time": 1.7335200309753418,
      "step": 13118
    },
    {
      "epoch": 1.2510299682617188e-05,
      "step": 13118,
      "training_step_time": 8.344599485397339
    },
    {
      "epoch": 1.2511253356933594e-05,
      "model_forward_time": 1.711181402206421,
      "step": 13119
    },
    {
      "epoch": 1.2511253356933594e-05,
      "step": 13119,
      "training_step_time": 7.775264024734497
    },
    {
      "epoch": 1.251220703125e-05,
      "grad_norm": 0.1649027019739151,
      "learning_rate": 9.820862240667086e-06,
      "loss": 0.0703,
      "step": 13120
    },
    {
      "epoch": 1.251220703125e-05,
      "model_forward_time": 1.7231566905975342,
      "step": 13120
    },
    {
      "dynamics_loss_avg": 0.07127966340631246,
      "epoch": 1.251220703125e-05,
      "step": 13120
    },
    {
      "action_loss_avg": 0.0029229592182673514,
      "epoch": 1.251220703125e-05,
      "step": 13120
    },
    {
      "epoch": 1.251220703125e-05,
      "step": 13120,
      "training_step_time": 6.918044328689575
    },
    {
      "epoch": 1.2513160705566407e-05,
      "model_forward_time": 1.730682134628296,
      "step": 13121
    },
    {
      "epoch": 1.2513160705566407e-05,
      "step": 13121,
      "training_step_time": 6.909985065460205
    },
    {
      "epoch": 1.2514114379882812e-05,
      "model_forward_time": 1.7193725109100342,
      "step": 13122
    },
    {
      "epoch": 1.2514114379882812e-05,
      "step": 13122,
      "training_step_time": 6.912504434585571
    },
    {
      "epoch": 1.2515068054199218e-05,
      "model_forward_time": 1.716637134552002,
      "step": 13123
    },
    {
      "epoch": 1.2515068054199218e-05,
      "step": 13123,
      "training_step_time": 6.899212121963501
    },
    {
      "epoch": 1.2516021728515626e-05,
      "model_forward_time": 1.7246265411376953,
      "step": 13124
    },
    {
      "epoch": 1.2516021728515626e-05,
      "step": 13124,
      "training_step_time": 6.887598037719727
    },
    {
      "epoch": 1.2516975402832031e-05,
      "model_forward_time": 1.020186185836792,
      "step": 13125
    },
    {
      "epoch": 1.2516975402832031e-05,
      "step": 13125,
      "training_step_time": 7.420243263244629
    },
    {
      "epoch": 1.2517929077148437e-05,
      "model_forward_time": 2.2323858737945557,
      "step": 13126
    },
    {
      "epoch": 1.2517929077148437e-05,
      "step": 13126,
      "training_step_time": 11.15948486328125
    },
    {
      "epoch": 1.2518882751464844e-05,
      "model_forward_time": 1.7298448085784912,
      "step": 13127
    },
    {
      "epoch": 1.2518882751464844e-05,
      "step": 13127,
      "training_step_time": 11.3995521068573
    },
    {
      "epoch": 1.251983642578125e-05,
      "model_forward_time": 1.7078533172607422,
      "step": 13128
    },
    {
      "epoch": 1.251983642578125e-05,
      "step": 13128,
      "training_step_time": 7.641510963439941
    },
    {
      "epoch": 1.2520790100097656e-05,
      "model_forward_time": 1.7073533535003662,
      "step": 13129
    },
    {
      "epoch": 1.2520790100097656e-05,
      "step": 13129,
      "training_step_time": 6.687150001525879
    },
    {
      "epoch": 1.2521743774414063e-05,
      "grad_norm": 0.15921330451965332,
      "learning_rate": 9.820423350796726e-06,
      "loss": 0.0674,
      "step": 13130
    },
    {
      "epoch": 1.2521743774414063e-05,
      "model_forward_time": 1.7167935371398926,
      "step": 13130
    },
    {
      "dynamics_loss_avg": 0.04797297213226557,
      "epoch": 1.2521743774414063e-05,
      "step": 13130
    },
    {
      "action_loss_avg": 0.0036308955284766855,
      "epoch": 1.2521743774414063e-05,
      "step": 13130
    },
    {
      "epoch": 1.2521743774414063e-05,
      "step": 13130,
      "training_step_time": 6.777079343795776
    },
    {
      "epoch": 1.2522697448730469e-05,
      "model_forward_time": 1.7096703052520752,
      "step": 13131
    },
    {
      "epoch": 1.2522697448730469e-05,
      "step": 13131,
      "training_step_time": 6.782684564590454
    },
    {
      "epoch": 1.2523651123046875e-05,
      "model_forward_time": 1.7043046951293945,
      "step": 13132
    },
    {
      "epoch": 1.2523651123046875e-05,
      "step": 13132,
      "training_step_time": 6.695087432861328
    },
    {
      "epoch": 1.252460479736328e-05,
      "model_forward_time": 1.3424780368804932,
      "step": 13133
    },
    {
      "epoch": 1.252460479736328e-05,
      "step": 13133,
      "training_step_time": 6.755248546600342
    },
    {
      "epoch": 1.2525558471679688e-05,
      "model_forward_time": 1.721064805984497,
      "step": 13134
    },
    {
      "epoch": 1.2525558471679688e-05,
      "step": 13134,
      "training_step_time": 7.435335159301758
    },
    {
      "epoch": 1.2526512145996094e-05,
      "model_forward_time": 1.3620884418487549,
      "step": 13135
    },
    {
      "epoch": 1.2526512145996094e-05,
      "step": 13135,
      "training_step_time": 9.532204389572144
    },
    {
      "epoch": 1.25274658203125e-05,
      "model_forward_time": 1.7748541831970215,
      "step": 13136
    },
    {
      "epoch": 1.25274658203125e-05,
      "step": 13136,
      "training_step_time": 10.514468908309937
    },
    {
      "epoch": 1.2528419494628907e-05,
      "model_forward_time": 1.0247716903686523,
      "step": 13137
    },
    {
      "epoch": 1.2528419494628907e-05,
      "step": 13137,
      "training_step_time": 8.692144393920898
    },
    {
      "epoch": 1.2529373168945312e-05,
      "model_forward_time": 1.7253849506378174,
      "step": 13138
    },
    {
      "epoch": 1.2529373168945312e-05,
      "step": 13138,
      "training_step_time": 6.6919846534729
    },
    {
      "epoch": 1.2530326843261718e-05,
      "model_forward_time": 1.7292120456695557,
      "step": 13139
    },
    {
      "epoch": 1.2530326843261718e-05,
      "step": 13139,
      "training_step_time": 6.670398235321045
    },
    {
      "epoch": 1.2531280517578126e-05,
      "grad_norm": 0.18625155091285706,
      "learning_rate": 9.819983933772119e-06,
      "loss": 0.0679,
      "step": 13140
    },
    {
      "epoch": 1.2531280517578126e-05,
      "model_forward_time": 1.7229087352752686,
      "step": 13140
    },
    {
      "dynamics_loss_avg": 0.05406408281996846,
      "epoch": 1.2531280517578126e-05,
      "step": 13140
    },
    {
      "action_loss_avg": 0.0039037214941345156,
      "epoch": 1.2531280517578126e-05,
      "step": 13140
    },
    {
      "epoch": 1.2531280517578126e-05,
      "step": 13140,
      "training_step_time": 6.729666233062744
    },
    {
      "epoch": 1.2532234191894531e-05,
      "model_forward_time": 1.7177021503448486,
      "step": 13141
    },
    {
      "epoch": 1.2532234191894531e-05,
      "step": 13141,
      "training_step_time": 6.6849799156188965
    },
    {
      "epoch": 1.2533187866210937e-05,
      "model_forward_time": 1.7637982368469238,
      "step": 13142
    },
    {
      "epoch": 1.2533187866210937e-05,
      "step": 13142,
      "training_step_time": 6.732118129730225
    },
    {
      "epoch": 1.2534141540527344e-05,
      "model_forward_time": 1.722609519958496,
      "step": 13143
    },
    {
      "epoch": 1.2534141540527344e-05,
      "step": 13143,
      "training_step_time": 8.413297414779663
    },
    {
      "epoch": 1.253509521484375e-05,
      "model_forward_time": 1.7226734161376953,
      "step": 13144
    },
    {
      "epoch": 1.253509521484375e-05,
      "step": 13144,
      "training_step_time": 9.568818092346191
    },
    {
      "epoch": 1.2536048889160156e-05,
      "model_forward_time": 2.2905962467193604,
      "step": 13145
    },
    {
      "epoch": 1.2536048889160156e-05,
      "step": 13145,
      "training_step_time": 8.676065921783447
    },
    {
      "epoch": 1.2537002563476563e-05,
      "model_forward_time": 1.7234084606170654,
      "step": 13146
    },
    {
      "epoch": 1.2537002563476563e-05,
      "step": 13146,
      "training_step_time": 7.745991945266724
    },
    {
      "epoch": 1.2537956237792969e-05,
      "model_forward_time": 1.7205781936645508,
      "step": 13147
    },
    {
      "epoch": 1.2537956237792969e-05,
      "step": 13147,
      "training_step_time": 7.478656768798828
    },
    {
      "epoch": 1.2538909912109375e-05,
      "model_forward_time": 1.7137980461120605,
      "step": 13148
    },
    {
      "epoch": 1.2538909912109375e-05,
      "step": 13148,
      "training_step_time": 6.731137990951538
    },
    {
      "epoch": 1.253986358642578e-05,
      "model_forward_time": 1.7204837799072266,
      "step": 13149
    },
    {
      "epoch": 1.253986358642578e-05,
      "step": 13149,
      "training_step_time": 6.75226616859436
    },
    {
      "epoch": 1.2540817260742188e-05,
      "grad_norm": 0.20907364785671234,
      "learning_rate": 9.819543989641314e-06,
      "loss": 0.0724,
      "step": 13150
    },
    {
      "epoch": 1.2540817260742188e-05,
      "model_forward_time": 1.717923879623413,
      "step": 13150
    },
    {
      "dynamics_loss_avg": 0.09196320772171021,
      "epoch": 1.2540817260742188e-05,
      "step": 13150
    },
    {
      "action_loss_avg": 0.0059137674164958295,
      "epoch": 1.2540817260742188e-05,
      "step": 13150
    },
    {
      "epoch": 1.2540817260742188e-05,
      "step": 13150,
      "training_step_time": 6.741909742355347
    },
    {
      "epoch": 1.2541770935058594e-05,
      "model_forward_time": 1.7170109748840332,
      "step": 13151
    },
    {
      "epoch": 1.2541770935058594e-05,
      "step": 13151,
      "training_step_time": 6.733573913574219
    },
    {
      "epoch": 1.2542724609375e-05,
      "model_forward_time": 1.7175633907318115,
      "step": 13152
    },
    {
      "epoch": 1.2542724609375e-05,
      "step": 13152,
      "training_step_time": 7.442666292190552
    },
    {
      "epoch": 1.2543678283691407e-05,
      "model_forward_time": 1.7147314548492432,
      "step": 13153
    },
    {
      "epoch": 1.2543678283691407e-05,
      "step": 13153,
      "training_step_time": 7.590985059738159
    },
    {
      "epoch": 1.2544631958007812e-05,
      "model_forward_time": 2.280299663543701,
      "step": 13154
    },
    {
      "epoch": 1.2544631958007812e-05,
      "step": 13154,
      "training_step_time": 9.52293348312378
    },
    {
      "epoch": 1.2545585632324218e-05,
      "model_forward_time": 1.7127478122711182,
      "step": 13155
    },
    {
      "epoch": 1.2545585632324218e-05,
      "step": 13155,
      "training_step_time": 9.713438987731934
    },
    {
      "epoch": 1.2546539306640626e-05,
      "model_forward_time": 1.71478271484375,
      "step": 13156
    },
    {
      "epoch": 1.2546539306640626e-05,
      "step": 13156,
      "training_step_time": 7.598890781402588
    },
    {
      "epoch": 1.2547492980957031e-05,
      "model_forward_time": 1.7377305030822754,
      "step": 13157
    },
    {
      "epoch": 1.2547492980957031e-05,
      "step": 13157,
      "training_step_time": 6.699120044708252
    },
    {
      "epoch": 1.2548446655273437e-05,
      "model_forward_time": 1.7163753509521484,
      "step": 13158
    },
    {
      "epoch": 1.2548446655273437e-05,
      "step": 13158,
      "training_step_time": 6.697448968887329
    },
    {
      "epoch": 1.2549400329589844e-05,
      "model_forward_time": 1.7169218063354492,
      "step": 13159
    },
    {
      "epoch": 1.2549400329589844e-05,
      "step": 13159,
      "training_step_time": 6.710774183273315
    },
    {
      "epoch": 1.255035400390625e-05,
      "grad_norm": 0.23178207874298096,
      "learning_rate": 9.819103518452425e-06,
      "loss": 0.0692,
      "step": 13160
    },
    {
      "epoch": 1.255035400390625e-05,
      "model_forward_time": 1.7167928218841553,
      "step": 13160
    },
    {
      "dynamics_loss_avg": 0.05583011545240879,
      "epoch": 1.255035400390625e-05,
      "step": 13160
    },
    {
      "action_loss_avg": 0.002566992217907682,
      "epoch": 1.255035400390625e-05,
      "step": 13160
    },
    {
      "epoch": 1.255035400390625e-05,
      "step": 13160,
      "training_step_time": 6.707743406295776
    },
    {
      "epoch": 1.2551307678222656e-05,
      "model_forward_time": 1.7243969440460205,
      "step": 13161
    },
    {
      "epoch": 1.2551307678222656e-05,
      "step": 13161,
      "training_step_time": 7.385696887969971
    },
    {
      "epoch": 1.2552261352539063e-05,
      "model_forward_time": 1.7215001583099365,
      "step": 13162
    },
    {
      "epoch": 1.2552261352539063e-05,
      "step": 13162,
      "training_step_time": 7.567861557006836
    },
    {
      "epoch": 1.2553215026855469e-05,
      "model_forward_time": 2.254542112350464,
      "step": 13163
    },
    {
      "epoch": 1.2553215026855469e-05,
      "step": 13163,
      "training_step_time": 8.36013126373291
    },
    {
      "epoch": 1.2554168701171875e-05,
      "model_forward_time": 1.7793247699737549,
      "step": 13164
    },
    {
      "epoch": 1.2554168701171875e-05,
      "step": 13164,
      "training_step_time": 9.683765411376953
    },
    {
      "epoch": 1.255512237548828e-05,
      "model_forward_time": 1.7236993312835693,
      "step": 13165
    },
    {
      "epoch": 1.255512237548828e-05,
      "step": 13165,
      "training_step_time": 7.604136228561401
    },
    {
      "epoch": 1.2556076049804688e-05,
      "model_forward_time": 1.7241923809051514,
      "step": 13166
    },
    {
      "epoch": 1.2556076049804688e-05,
      "step": 13166,
      "training_step_time": 6.668426752090454
    },
    {
      "epoch": 1.2557029724121094e-05,
      "model_forward_time": 1.721346139907837,
      "step": 13167
    },
    {
      "epoch": 1.2557029724121094e-05,
      "step": 13167,
      "training_step_time": 6.674150705337524
    },
    {
      "epoch": 1.25579833984375e-05,
      "model_forward_time": 1.7123308181762695,
      "step": 13168
    },
    {
      "epoch": 1.25579833984375e-05,
      "step": 13168,
      "training_step_time": 6.646971225738525
    },
    {
      "epoch": 1.2558937072753907e-05,
      "model_forward_time": 1.7276616096496582,
      "step": 13169
    },
    {
      "epoch": 1.2558937072753907e-05,
      "step": 13169,
      "training_step_time": 6.659626483917236
    },
    {
      "epoch": 1.2559890747070312e-05,
      "grad_norm": 0.285563588142395,
      "learning_rate": 9.818662520253619e-06,
      "loss": 0.075,
      "step": 13170
    },
    {
      "epoch": 1.2559890747070312e-05,
      "model_forward_time": 1.7655067443847656,
      "step": 13170
    },
    {
      "dynamics_loss_avg": 0.06487200837582349,
      "epoch": 1.2559890747070312e-05,
      "step": 13170
    },
    {
      "action_loss_avg": 0.004655376041773707,
      "epoch": 1.2559890747070312e-05,
      "step": 13170
    },
    {
      "epoch": 1.2559890747070312e-05,
      "step": 13170,
      "training_step_time": 7.950691223144531
    },
    {
      "epoch": 1.2560844421386718e-05,
      "model_forward_time": 1.7111492156982422,
      "step": 13171
    },
    {
      "epoch": 1.2560844421386718e-05,
      "step": 13171,
      "training_step_time": 7.5400307178497314
    },
    {
      "epoch": 1.2561798095703126e-05,
      "model_forward_time": 1.7109689712524414,
      "step": 13172
    },
    {
      "epoch": 1.2561798095703126e-05,
      "step": 13172,
      "training_step_time": 8.634116411209106
    },
    {
      "epoch": 1.2562751770019531e-05,
      "model_forward_time": 2.270993709564209,
      "step": 13173
    },
    {
      "epoch": 1.2562751770019531e-05,
      "step": 13173,
      "training_step_time": 10.218538522720337
    },
    {
      "epoch": 1.2563705444335937e-05,
      "model_forward_time": 1.0256834030151367,
      "step": 13174
    },
    {
      "epoch": 1.2563705444335937e-05,
      "step": 13174,
      "training_step_time": 7.648715019226074
    },
    {
      "epoch": 1.2564659118652344e-05,
      "model_forward_time": 1.7128932476043701,
      "step": 13175
    },
    {
      "epoch": 1.2564659118652344e-05,
      "step": 13175,
      "training_step_time": 7.4021995067596436
    },
    {
      "epoch": 1.256561279296875e-05,
      "model_forward_time": 1.7158691883087158,
      "step": 13176
    },
    {
      "epoch": 1.256561279296875e-05,
      "step": 13176,
      "training_step_time": 6.678775787353516
    },
    {
      "epoch": 1.2566566467285156e-05,
      "model_forward_time": 1.7165729999542236,
      "step": 13177
    },
    {
      "epoch": 1.2566566467285156e-05,
      "step": 13177,
      "training_step_time": 6.68395471572876
    },
    {
      "epoch": 1.2567520141601563e-05,
      "model_forward_time": 1.7209584712982178,
      "step": 13178
    },
    {
      "epoch": 1.2567520141601563e-05,
      "step": 13178,
      "training_step_time": 6.667412281036377
    },
    {
      "epoch": 1.2568473815917969e-05,
      "model_forward_time": 1.7149302959442139,
      "step": 13179
    },
    {
      "epoch": 1.2568473815917969e-05,
      "step": 13179,
      "training_step_time": 7.384202718734741
    },
    {
      "epoch": 1.2569427490234375e-05,
      "grad_norm": 0.22087796032428741,
      "learning_rate": 9.818220995093126e-06,
      "loss": 0.0751,
      "step": 13180
    },
    {
      "epoch": 1.2569427490234375e-05,
      "model_forward_time": 1.7140276432037354,
      "step": 13180
    },
    {
      "dynamics_loss_avg": 0.05154400710016489,
      "epoch": 1.2569427490234375e-05,
      "step": 13180
    },
    {
      "action_loss_avg": 0.005094397463835776,
      "epoch": 1.2569427490234375e-05,
      "step": 13180
    },
    {
      "epoch": 1.2569427490234375e-05,
      "step": 13180,
      "training_step_time": 7.545590162277222
    },
    {
      "epoch": 1.257038116455078e-05,
      "model_forward_time": 1.7073352336883545,
      "step": 13181
    },
    {
      "epoch": 1.257038116455078e-05,
      "step": 13181,
      "training_step_time": 8.720238447189331
    },
    {
      "epoch": 1.2571334838867188e-05,
      "model_forward_time": 1.8710041046142578,
      "step": 13182
    },
    {
      "epoch": 1.2571334838867188e-05,
      "step": 13182,
      "training_step_time": 9.331831216812134
    },
    {
      "epoch": 1.2572288513183594e-05,
      "model_forward_time": 1.7096459865570068,
      "step": 13183
    },
    {
      "epoch": 1.2572288513183594e-05,
      "step": 13183,
      "training_step_time": 7.7694091796875
    },
    {
      "epoch": 1.25732421875e-05,
      "model_forward_time": 1.7169609069824219,
      "step": 13184
    },
    {
      "epoch": 1.25732421875e-05,
      "step": 13184,
      "training_step_time": 7.563442945480347
    },
    {
      "epoch": 1.2574195861816407e-05,
      "model_forward_time": 1.7335104942321777,
      "step": 13185
    },
    {
      "epoch": 1.2574195861816407e-05,
      "step": 13185,
      "training_step_time": 6.771477222442627
    },
    {
      "epoch": 1.2575149536132812e-05,
      "model_forward_time": 1.7198073863983154,
      "step": 13186
    },
    {
      "epoch": 1.2575149536132812e-05,
      "step": 13186,
      "training_step_time": 6.886645555496216
    },
    {
      "epoch": 1.2576103210449218e-05,
      "model_forward_time": 1.7296881675720215,
      "step": 13187
    },
    {
      "epoch": 1.2576103210449218e-05,
      "step": 13187,
      "training_step_time": 6.926135301589966
    },
    {
      "epoch": 1.2577056884765626e-05,
      "model_forward_time": 1.735687255859375,
      "step": 13188
    },
    {
      "epoch": 1.2577056884765626e-05,
      "step": 13188,
      "training_step_time": 8.007092714309692
    },
    {
      "epoch": 1.2578010559082031e-05,
      "model_forward_time": 1.714998483657837,
      "step": 13189
    },
    {
      "epoch": 1.2578010559082031e-05,
      "step": 13189,
      "training_step_time": 7.590547323226929
    },
    {
      "epoch": 1.2578964233398437e-05,
      "grad_norm": 0.24737583100795746,
      "learning_rate": 9.817778943019228e-06,
      "loss": 0.0706,
      "step": 13190
    },
    {
      "epoch": 1.2578964233398437e-05,
      "model_forward_time": 1.7100284099578857,
      "step": 13190
    },
    {
      "dynamics_loss_avg": 0.06180023197084665,
      "epoch": 1.2578964233398437e-05,
      "step": 13190
    },
    {
      "action_loss_avg": 0.0025045952119398863,
      "epoch": 1.2578964233398437e-05,
      "step": 13190
    },
    {
      "epoch": 1.2578964233398437e-05,
      "step": 13190,
      "training_step_time": 7.68815803527832
    },
    {
      "epoch": 1.2579917907714844e-05,
      "model_forward_time": 2.2434558868408203,
      "step": 13191
    },
    {
      "epoch": 1.2579917907714844e-05,
      "step": 13191,
      "training_step_time": 8.65234112739563
    },
    {
      "epoch": 1.258087158203125e-05,
      "model_forward_time": 1.7764806747436523,
      "step": 13192
    },
    {
      "epoch": 1.258087158203125e-05,
      "step": 13192,
      "training_step_time": 7.866679906845093
    },
    {
      "epoch": 1.2581825256347656e-05,
      "model_forward_time": 1.734407901763916,
      "step": 13193
    },
    {
      "epoch": 1.2581825256347656e-05,
      "step": 13193,
      "training_step_time": 7.677599191665649
    },
    {
      "epoch": 1.2582778930664063e-05,
      "model_forward_time": 1.732919692993164,
      "step": 13194
    },
    {
      "epoch": 1.2582778930664063e-05,
      "step": 13194,
      "training_step_time": 6.803715944290161
    },
    {
      "epoch": 1.2583732604980469e-05,
      "model_forward_time": 1.7277932167053223,
      "step": 13195
    },
    {
      "epoch": 1.2583732604980469e-05,
      "step": 13195,
      "training_step_time": 6.881620407104492
    },
    {
      "epoch": 1.2584686279296875e-05,
      "model_forward_time": 1.7147371768951416,
      "step": 13196
    },
    {
      "epoch": 1.2584686279296875e-05,
      "step": 13196,
      "training_step_time": 6.879112243652344
    },
    {
      "epoch": 1.258563995361328e-05,
      "model_forward_time": 1.7236683368682861,
      "step": 13197
    },
    {
      "epoch": 1.258563995361328e-05,
      "step": 13197,
      "training_step_time": 7.418664216995239
    },
    {
      "epoch": 1.2586593627929688e-05,
      "model_forward_time": 1.7345952987670898,
      "step": 13198
    },
    {
      "epoch": 1.2586593627929688e-05,
      "step": 13198,
      "training_step_time": 7.499094724655151
    },
    {
      "epoch": 1.2587547302246094e-05,
      "model_forward_time": 1.7235257625579834,
      "step": 13199
    },
    {
      "epoch": 1.2587547302246094e-05,
      "step": 13199,
      "training_step_time": 8.49626350402832
    },
    {
      "epoch": 1.25885009765625e-05,
      "grad_norm": 0.2104243040084839,
      "learning_rate": 9.81733636408027e-06,
      "loss": 0.0643,
      "step": 13200
    },
    {
      "epoch": 1.25885009765625e-05,
      "model_forward_time": 1.7050023078918457,
      "step": 13200
    },
    {
      "dynamics_loss_avg": 0.05362768555060029,
      "epoch": 1.25885009765625e-05,
      "step": 13200
    },
    {
      "action_loss_avg": 0.003706590028014034,
      "epoch": 1.25885009765625e-05,
      "step": 13200
    },
    {
      "epoch": 1.25885009765625e-05,
      "step": 13200,
      "training_step_time": 8.826108932495117
    },
    {
      "epoch": 1.2589454650878907e-05,
      "model_forward_time": 2.3027374744415283,
      "step": 13201
    },
    {
      "epoch": 1.2589454650878907e-05,
      "step": 13201,
      "training_step_time": 8.858530282974243
    },
    {
      "epoch": 1.2590408325195312e-05,
      "model_forward_time": 1.7285871505737305,
      "step": 13202
    },
    {
      "epoch": 1.2590408325195312e-05,
      "step": 13202,
      "training_step_time": 7.814346551895142
    },
    {
      "epoch": 1.2591361999511718e-05,
      "model_forward_time": 1.7857797145843506,
      "step": 13203
    },
    {
      "epoch": 1.2591361999511718e-05,
      "step": 13203,
      "training_step_time": 7.388203382492065
    },
    {
      "epoch": 1.2592315673828126e-05,
      "model_forward_time": 1.723844289779663,
      "step": 13204
    },
    {
      "epoch": 1.2592315673828126e-05,
      "step": 13204,
      "training_step_time": 6.931231498718262
    },
    {
      "epoch": 1.2593269348144531e-05,
      "model_forward_time": 1.720736026763916,
      "step": 13205
    },
    {
      "epoch": 1.2593269348144531e-05,
      "step": 13205,
      "training_step_time": 6.92810320854187
    },
    {
      "epoch": 1.2594223022460937e-05,
      "model_forward_time": 1.7298662662506104,
      "step": 13206
    },
    {
      "epoch": 1.2594223022460937e-05,
      "step": 13206,
      "training_step_time": 7.494964122772217
    },
    {
      "epoch": 1.2595176696777344e-05,
      "model_forward_time": 1.3583834171295166,
      "step": 13207
    },
    {
      "epoch": 1.2595176696777344e-05,
      "step": 13207,
      "training_step_time": 7.604834079742432
    },
    {
      "epoch": 1.259613037109375e-05,
      "model_forward_time": 1.7219464778900146,
      "step": 13208
    },
    {
      "epoch": 1.259613037109375e-05,
      "step": 13208,
      "training_step_time": 7.7746193408966064
    },
    {
      "epoch": 1.2597084045410156e-05,
      "model_forward_time": 1.7165319919586182,
      "step": 13209
    },
    {
      "epoch": 1.2597084045410156e-05,
      "step": 13209,
      "training_step_time": 10.368691444396973
    },
    {
      "epoch": 1.2598037719726563e-05,
      "grad_norm": 0.34839972853660583,
      "learning_rate": 9.816893258324649e-06,
      "loss": 0.0665,
      "step": 13210
    },
    {
      "epoch": 1.2598037719726563e-05,
      "model_forward_time": 2.2639482021331787,
      "step": 13210
    },
    {
      "dynamics_loss_avg": 0.06457997057586909,
      "epoch": 1.2598037719726563e-05,
      "step": 13210
    },
    {
      "action_loss_avg": 0.003264355787541717,
      "epoch": 1.2598037719726563e-05,
      "step": 13210
    },
    {
      "epoch": 1.2598037719726563e-05,
      "step": 13210,
      "training_step_time": 9.03455662727356
    },
    {
      "epoch": 1.2598991394042969e-05,
      "model_forward_time": 1.7702291011810303,
      "step": 13211
    },
    {
      "epoch": 1.2598991394042969e-05,
      "step": 13211,
      "training_step_time": 7.91038966178894
    },
    {
      "epoch": 1.2599945068359375e-05,
      "model_forward_time": 1.7205891609191895,
      "step": 13212
    },
    {
      "epoch": 1.2599945068359375e-05,
      "step": 13212,
      "training_step_time": 7.408420562744141
    },
    {
      "epoch": 1.260089874267578e-05,
      "model_forward_time": 1.7258403301239014,
      "step": 13213
    },
    {
      "epoch": 1.260089874267578e-05,
      "step": 13213,
      "training_step_time": 6.827659606933594
    },
    {
      "epoch": 1.2601852416992188e-05,
      "model_forward_time": 1.732046365737915,
      "step": 13214
    },
    {
      "epoch": 1.2601852416992188e-05,
      "step": 13214,
      "training_step_time": 6.930596113204956
    },
    {
      "epoch": 1.2602806091308594e-05,
      "model_forward_time": 1.7285521030426025,
      "step": 13215
    },
    {
      "epoch": 1.2602806091308594e-05,
      "step": 13215,
      "training_step_time": 6.921579837799072
    },
    {
      "epoch": 1.2603759765625e-05,
      "model_forward_time": 1.7240548133850098,
      "step": 13216
    },
    {
      "epoch": 1.2603759765625e-05,
      "step": 13216,
      "training_step_time": 7.598353147506714
    },
    {
      "epoch": 1.2604713439941407e-05,
      "model_forward_time": 1.721672773361206,
      "step": 13217
    },
    {
      "epoch": 1.2604713439941407e-05,
      "step": 13217,
      "training_step_time": 7.752177715301514
    },
    {
      "epoch": 1.2605667114257812e-05,
      "model_forward_time": 1.7296361923217773,
      "step": 13218
    },
    {
      "epoch": 1.2605667114257812e-05,
      "step": 13218,
      "training_step_time": 8.813086748123169
    },
    {
      "epoch": 1.2606620788574218e-05,
      "model_forward_time": 2.2826905250549316,
      "step": 13219
    },
    {
      "epoch": 1.2606620788574218e-05,
      "step": 13219,
      "training_step_time": 10.024007797241211
    },
    {
      "epoch": 1.2607574462890626e-05,
      "grad_norm": 0.1662725806236267,
      "learning_rate": 9.816449625800824e-06,
      "loss": 0.0606,
      "step": 13220
    },
    {
      "epoch": 1.2607574462890626e-05,
      "model_forward_time": 1.7540898323059082,
      "step": 13220
    },
    {
      "dynamics_loss_avg": 0.07234279699623584,
      "epoch": 1.2607574462890626e-05,
      "step": 13220
    },
    {
      "action_loss_avg": 0.006991335493512452,
      "epoch": 1.2607574462890626e-05,
      "step": 13220
    },
    {
      "epoch": 1.2607574462890626e-05,
      "step": 13220,
      "training_step_time": 7.812598466873169
    },
    {
      "epoch": 1.2608528137207031e-05,
      "model_forward_time": 1.724754810333252,
      "step": 13221
    },
    {
      "epoch": 1.2608528137207031e-05,
      "step": 13221,
      "training_step_time": 7.643428802490234
    },
    {
      "epoch": 1.2609481811523437e-05,
      "model_forward_time": 1.727240800857544,
      "step": 13222
    },
    {
      "epoch": 1.2609481811523437e-05,
      "step": 13222,
      "training_step_time": 6.9077208042144775
    },
    {
      "epoch": 1.2610435485839844e-05,
      "model_forward_time": 1.727388858795166,
      "step": 13223
    },
    {
      "epoch": 1.2610435485839844e-05,
      "step": 13223,
      "training_step_time": 6.817347764968872
    },
    {
      "epoch": 1.261138916015625e-05,
      "model_forward_time": 1.728344202041626,
      "step": 13224
    },
    {
      "epoch": 1.261138916015625e-05,
      "step": 13224,
      "training_step_time": 6.855372190475464
    },
    {
      "epoch": 1.2612342834472656e-05,
      "model_forward_time": 1.723616600036621,
      "step": 13225
    },
    {
      "epoch": 1.2612342834472656e-05,
      "step": 13225,
      "training_step_time": 7.394085645675659
    },
    {
      "epoch": 1.2613296508789063e-05,
      "model_forward_time": 1.7274293899536133,
      "step": 13226
    },
    {
      "epoch": 1.2613296508789063e-05,
      "step": 13226,
      "training_step_time": 8.454509258270264
    },
    {
      "epoch": 1.2614250183105469e-05,
      "model_forward_time": 1.706355094909668,
      "step": 13227
    },
    {
      "epoch": 1.2614250183105469e-05,
      "step": 13227,
      "training_step_time": 7.675098180770874
    },
    {
      "epoch": 1.2615203857421875e-05,
      "model_forward_time": 2.262608051300049,
      "step": 13228
    },
    {
      "epoch": 1.2615203857421875e-05,
      "step": 13228,
      "training_step_time": 7.86761736869812
    },
    {
      "epoch": 1.261615753173828e-05,
      "model_forward_time": 1.7460455894470215,
      "step": 13229
    },
    {
      "epoch": 1.261615753173828e-05,
      "step": 13229,
      "training_step_time": 7.744872093200684
    },
    {
      "epoch": 1.2617111206054688e-05,
      "grad_norm": 0.24300432205200195,
      "learning_rate": 9.816005466557309e-06,
      "loss": 0.064,
      "step": 13230
    },
    {
      "epoch": 1.2617111206054688e-05,
      "model_forward_time": 1.7185535430908203,
      "step": 13230
    },
    {
      "dynamics_loss_avg": 0.04800082985311747,
      "epoch": 1.2617111206054688e-05,
      "step": 13230
    },
    {
      "action_loss_avg": 0.002561630157288164,
      "epoch": 1.2617111206054688e-05,
      "step": 13230
    },
    {
      "epoch": 1.2617111206054688e-05,
      "step": 13230,
      "training_step_time": 7.595089912414551
    },
    {
      "epoch": 1.2618064880371094e-05,
      "model_forward_time": 1.7193043231964111,
      "step": 13231
    },
    {
      "epoch": 1.2618064880371094e-05,
      "step": 13231,
      "training_step_time": 6.855321407318115
    },
    {
      "epoch": 1.26190185546875e-05,
      "model_forward_time": 1.7135169506072998,
      "step": 13232
    },
    {
      "epoch": 1.26190185546875e-05,
      "step": 13232,
      "training_step_time": 6.880216598510742
    },
    {
      "epoch": 1.2619972229003907e-05,
      "model_forward_time": 1.7304465770721436,
      "step": 13233
    },
    {
      "epoch": 1.2619972229003907e-05,
      "step": 13233,
      "training_step_time": 6.860730886459351
    },
    {
      "epoch": 1.2620925903320312e-05,
      "model_forward_time": 1.7355320453643799,
      "step": 13234
    },
    {
      "epoch": 1.2620925903320312e-05,
      "step": 13234,
      "training_step_time": 7.529494285583496
    },
    {
      "epoch": 1.2621879577636718e-05,
      "model_forward_time": 1.337751865386963,
      "step": 13235
    },
    {
      "epoch": 1.2621879577636718e-05,
      "step": 13235,
      "training_step_time": 7.718047380447388
    },
    {
      "epoch": 1.2622833251953126e-05,
      "model_forward_time": 1.7197911739349365,
      "step": 13236
    },
    {
      "epoch": 1.2622833251953126e-05,
      "step": 13236,
      "training_step_time": 8.228017568588257
    },
    {
      "epoch": 1.2623786926269531e-05,
      "model_forward_time": 1.7353365421295166,
      "step": 13237
    },
    {
      "epoch": 1.2623786926269531e-05,
      "step": 13237,
      "training_step_time": 9.553242444992065
    },
    {
      "epoch": 1.2624740600585937e-05,
      "model_forward_time": 2.269327402114868,
      "step": 13238
    },
    {
      "epoch": 1.2624740600585937e-05,
      "step": 13238,
      "training_step_time": 8.679926633834839
    },
    {
      "epoch": 1.2625694274902344e-05,
      "model_forward_time": 1.7711231708526611,
      "step": 13239
    },
    {
      "epoch": 1.2625694274902344e-05,
      "step": 13239,
      "training_step_time": 9.08181881904602
    },
    {
      "epoch": 1.262664794921875e-05,
      "grad_norm": 0.13863317668437958,
      "learning_rate": 9.815560780642675e-06,
      "loss": 0.0697,
      "step": 13240
    },
    {
      "epoch": 1.262664794921875e-05,
      "model_forward_time": 1.7275922298431396,
      "step": 13240
    },
    {
      "dynamics_loss_avg": 0.06019252091646195,
      "epoch": 1.262664794921875e-05,
      "step": 13240
    },
    {
      "action_loss_avg": 0.0025464931910391896,
      "epoch": 1.262664794921875e-05,
      "step": 13240
    },
    {
      "epoch": 1.262664794921875e-05,
      "step": 13240,
      "training_step_time": 7.34543251991272
    },
    {
      "epoch": 1.2627601623535156e-05,
      "model_forward_time": 1.7216858863830566,
      "step": 13241
    },
    {
      "epoch": 1.2627601623535156e-05,
      "step": 13241,
      "training_step_time": 6.863266229629517
    },
    {
      "epoch": 1.2628555297851563e-05,
      "model_forward_time": 1.7232739925384521,
      "step": 13242
    },
    {
      "epoch": 1.2628555297851563e-05,
      "step": 13242,
      "training_step_time": 6.896786212921143
    },
    {
      "epoch": 1.2629508972167969e-05,
      "model_forward_time": 1.7208666801452637,
      "step": 13243
    },
    {
      "epoch": 1.2629508972167969e-05,
      "step": 13243,
      "training_step_time": 7.271152496337891
    },
    {
      "epoch": 1.2630462646484375e-05,
      "model_forward_time": 1.7204103469848633,
      "step": 13244
    },
    {
      "epoch": 1.2630462646484375e-05,
      "step": 13244,
      "training_step_time": 7.753978729248047
    },
    {
      "epoch": 1.263141632080078e-05,
      "model_forward_time": 1.7216517925262451,
      "step": 13245
    },
    {
      "epoch": 1.263141632080078e-05,
      "step": 13245,
      "training_step_time": 8.599440574645996
    },
    {
      "epoch": 1.2632369995117188e-05,
      "model_forward_time": 1.7240517139434814,
      "step": 13246
    },
    {
      "epoch": 1.2632369995117188e-05,
      "step": 13246,
      "training_step_time": 8.766116619110107
    },
    {
      "epoch": 1.2633323669433594e-05,
      "model_forward_time": 2.263927936553955,
      "step": 13247
    },
    {
      "epoch": 1.2633323669433594e-05,
      "step": 13247,
      "training_step_time": 9.752803325653076
    },
    {
      "epoch": 1.263427734375e-05,
      "model_forward_time": 1.7798244953155518,
      "step": 13248
    },
    {
      "epoch": 1.263427734375e-05,
      "step": 13248,
      "training_step_time": 7.670166969299316
    },
    {
      "epoch": 1.2635231018066407e-05,
      "model_forward_time": 1.7227294445037842,
      "step": 13249
    },
    {
      "epoch": 1.2635231018066407e-05,
      "step": 13249,
      "training_step_time": 7.519234657287598
    },
    {
      "epoch": 1.2636184692382812e-05,
      "grad_norm": 0.18049398064613342,
      "learning_rate": 9.815115568105555e-06,
      "loss": 0.0652,
      "step": 13250
    },
    {
      "epoch": 1.2636184692382812e-05,
      "model_forward_time": 1.7174208164215088,
      "step": 13250
    },
    {
      "dynamics_loss_avg": 0.06241090707480908,
      "epoch": 1.2636184692382812e-05,
      "step": 13250
    },
    {
      "action_loss_avg": 0.002525609254371375,
      "epoch": 1.2636184692382812e-05,
      "step": 13250
    },
    {
      "epoch": 1.2636184692382812e-05,
      "step": 13250,
      "training_step_time": 6.765917539596558
    },
    {
      "epoch": 1.2637138366699218e-05,
      "model_forward_time": 1.726379632949829,
      "step": 13251
    },
    {
      "epoch": 1.2637138366699218e-05,
      "step": 13251,
      "training_step_time": 6.792970895767212
    },
    {
      "epoch": 1.2638092041015626e-05,
      "model_forward_time": 1.717313289642334,
      "step": 13252
    },
    {
      "epoch": 1.2638092041015626e-05,
      "step": 13252,
      "training_step_time": 7.036729335784912
    },
    {
      "epoch": 1.2639045715332031e-05,
      "model_forward_time": 1.7184269428253174,
      "step": 13253
    },
    {
      "epoch": 1.2639045715332031e-05,
      "step": 13253,
      "training_step_time": 7.594296455383301
    },
    {
      "epoch": 1.2639999389648437e-05,
      "model_forward_time": 1.7185029983520508,
      "step": 13254
    },
    {
      "epoch": 1.2639999389648437e-05,
      "step": 13254,
      "training_step_time": 7.644791603088379
    },
    {
      "epoch": 1.2640953063964844e-05,
      "model_forward_time": 1.7225558757781982,
      "step": 13255
    },
    {
      "epoch": 1.2640953063964844e-05,
      "step": 13255,
      "training_step_time": 8.4666268825531
    },
    {
      "epoch": 1.264190673828125e-05,
      "model_forward_time": 2.285869598388672,
      "step": 13256
    },
    {
      "epoch": 1.264190673828125e-05,
      "step": 13256,
      "training_step_time": 8.323893070220947
    },
    {
      "epoch": 1.2642860412597656e-05,
      "model_forward_time": 1.7676045894622803,
      "step": 13257
    },
    {
      "epoch": 1.2642860412597656e-05,
      "step": 13257,
      "training_step_time": 8.583973169326782
    },
    {
      "epoch": 1.2643814086914062e-05,
      "model_forward_time": 1.7264301776885986,
      "step": 13258
    },
    {
      "epoch": 1.2643814086914062e-05,
      "step": 13258,
      "training_step_time": 7.45789647102356
    },
    {
      "epoch": 1.2644767761230469e-05,
      "model_forward_time": 1.0253689289093018,
      "step": 13259
    },
    {
      "epoch": 1.2644767761230469e-05,
      "step": 13259,
      "training_step_time": 6.745631217956543
    },
    {
      "epoch": 1.2645721435546875e-05,
      "grad_norm": 0.19850344955921173,
      "learning_rate": 9.814669828994638e-06,
      "loss": 0.067,
      "step": 13260
    },
    {
      "epoch": 1.2645721435546875e-05,
      "model_forward_time": 1.7148268222808838,
      "step": 13260
    },
    {
      "dynamics_loss_avg": 0.04672706667333841,
      "epoch": 1.2645721435546875e-05,
      "step": 13260
    },
    {
      "action_loss_avg": 0.0018160880310460926,
      "epoch": 1.2645721435546875e-05,
      "step": 13260
    },
    {
      "epoch": 1.2645721435546875e-05,
      "step": 13260,
      "training_step_time": 6.747209787368774
    },
    {
      "epoch": 1.264667510986328e-05,
      "model_forward_time": 1.7210569381713867,
      "step": 13261
    },
    {
      "epoch": 1.264667510986328e-05,
      "step": 13261,
      "training_step_time": 6.991123199462891
    },
    {
      "epoch": 1.2647628784179688e-05,
      "model_forward_time": 1.7286064624786377,
      "step": 13262
    },
    {
      "epoch": 1.2647628784179688e-05,
      "step": 13262,
      "training_step_time": 7.641639947891235
    },
    {
      "epoch": 1.2648582458496094e-05,
      "model_forward_time": 1.723588466644287,
      "step": 13263
    },
    {
      "epoch": 1.2648582458496094e-05,
      "step": 13263,
      "training_step_time": 7.58130145072937
    },
    {
      "epoch": 1.26495361328125e-05,
      "model_forward_time": 1.7285165786743164,
      "step": 13264
    },
    {
      "epoch": 1.26495361328125e-05,
      "step": 13264,
      "training_step_time": 8.738074779510498
    },
    {
      "epoch": 1.2650489807128907e-05,
      "model_forward_time": 1.7203311920166016,
      "step": 13265
    },
    {
      "epoch": 1.2650489807128907e-05,
      "step": 13265,
      "training_step_time": 9.299445390701294
    },
    {
      "epoch": 1.2651443481445313e-05,
      "model_forward_time": 1.754650354385376,
      "step": 13266
    },
    {
      "epoch": 1.2651443481445313e-05,
      "step": 13266,
      "training_step_time": 7.678348541259766
    },
    {
      "epoch": 1.2652397155761718e-05,
      "model_forward_time": 1.7258329391479492,
      "step": 13267
    },
    {
      "epoch": 1.2652397155761718e-05,
      "step": 13267,
      "training_step_time": 7.5878684520721436
    },
    {
      "epoch": 1.2653350830078126e-05,
      "model_forward_time": 1.7203314304351807,
      "step": 13268
    },
    {
      "epoch": 1.2653350830078126e-05,
      "step": 13268,
      "training_step_time": 7.440227508544922
    },
    {
      "epoch": 1.2654304504394531e-05,
      "model_forward_time": 1.7184340953826904,
      "step": 13269
    },
    {
      "epoch": 1.2654304504394531e-05,
      "step": 13269,
      "training_step_time": 6.782603740692139
    },
    {
      "epoch": 1.2655258178710937e-05,
      "grad_norm": 0.1899513453245163,
      "learning_rate": 9.814223563358665e-06,
      "loss": 0.0653,
      "step": 13270
    },
    {
      "epoch": 1.2655258178710937e-05,
      "model_forward_time": 1.3519506454467773,
      "step": 13270
    },
    {
      "dynamics_loss_avg": 0.057120096124708655,
      "epoch": 1.2655258178710937e-05,
      "step": 13270
    },
    {
      "action_loss_avg": 0.002432167180813849,
      "epoch": 1.2655258178710937e-05,
      "step": 13270
    },
    {
      "epoch": 1.2655258178710937e-05,
      "step": 13270,
      "training_step_time": 6.99866247177124
    },
    {
      "epoch": 1.2656211853027345e-05,
      "model_forward_time": 1.7226951122283936,
      "step": 13271
    },
    {
      "epoch": 1.2656211853027345e-05,
      "step": 13271,
      "training_step_time": 8.376227378845215
    },
    {
      "epoch": 1.265716552734375e-05,
      "model_forward_time": 1.7225942611694336,
      "step": 13272
    },
    {
      "epoch": 1.265716552734375e-05,
      "step": 13272,
      "training_step_time": 7.503716468811035
    },
    {
      "epoch": 1.2658119201660156e-05,
      "model_forward_time": 1.7243170738220215,
      "step": 13273
    },
    {
      "epoch": 1.2658119201660156e-05,
      "step": 13273,
      "training_step_time": 7.539430618286133
    },
    {
      "epoch": 1.2659072875976562e-05,
      "model_forward_time": 1.7186915874481201,
      "step": 13274
    },
    {
      "epoch": 1.2659072875976562e-05,
      "step": 13274,
      "training_step_time": 8.697415828704834
    },
    {
      "epoch": 1.2660026550292969e-05,
      "model_forward_time": 2.3119213581085205,
      "step": 13275
    },
    {
      "epoch": 1.2660026550292969e-05,
      "step": 13275,
      "training_step_time": 7.627811431884766
    },
    {
      "epoch": 1.2660980224609375e-05,
      "model_forward_time": 1.6983888149261475,
      "step": 13276
    },
    {
      "epoch": 1.2660980224609375e-05,
      "step": 13276,
      "training_step_time": 8.393897294998169
    },
    {
      "epoch": 1.266193389892578e-05,
      "model_forward_time": 1.7295360565185547,
      "step": 13277
    },
    {
      "epoch": 1.266193389892578e-05,
      "step": 13277,
      "training_step_time": 7.461416721343994
    },
    {
      "epoch": 1.2662887573242188e-05,
      "model_forward_time": 1.7147343158721924,
      "step": 13278
    },
    {
      "epoch": 1.2662887573242188e-05,
      "step": 13278,
      "training_step_time": 6.763874292373657
    },
    {
      "epoch": 1.2663841247558594e-05,
      "model_forward_time": 1.7131409645080566,
      "step": 13279
    },
    {
      "epoch": 1.2663841247558594e-05,
      "step": 13279,
      "training_step_time": 6.965082168579102
    },
    {
      "epoch": 1.2664794921875e-05,
      "grad_norm": 0.2733643352985382,
      "learning_rate": 9.813776771246443e-06,
      "loss": 0.0655,
      "step": 13280
    },
    {
      "epoch": 1.2664794921875e-05,
      "model_forward_time": 1.7276568412780762,
      "step": 13280
    },
    {
      "dynamics_loss_avg": 0.06295682713389397,
      "epoch": 1.2664794921875e-05,
      "step": 13280
    },
    {
      "action_loss_avg": 0.0029586889198981226,
      "epoch": 1.2664794921875e-05,
      "step": 13280
    },
    {
      "epoch": 1.2664794921875e-05,
      "step": 13280,
      "training_step_time": 7.515716552734375
    },
    {
      "epoch": 1.2665748596191407e-05,
      "model_forward_time": 1.3516767024993896,
      "step": 13281
    },
    {
      "epoch": 1.2665748596191407e-05,
      "step": 13281,
      "training_step_time": 8.400895357131958
    },
    {
      "epoch": 1.2666702270507813e-05,
      "model_forward_time": 1.7282164096832275,
      "step": 13282
    },
    {
      "epoch": 1.2666702270507813e-05,
      "step": 13282,
      "training_step_time": 8.079602718353271
    },
    {
      "epoch": 1.2667655944824218e-05,
      "model_forward_time": 1.7262940406799316,
      "step": 13283
    },
    {
      "epoch": 1.2667655944824218e-05,
      "step": 13283,
      "training_step_time": 8.600770235061646
    },
    {
      "epoch": 1.2668609619140626e-05,
      "model_forward_time": 2.3131215572357178,
      "step": 13284
    },
    {
      "epoch": 1.2668609619140626e-05,
      "step": 13284,
      "training_step_time": 7.888321161270142
    },
    {
      "epoch": 1.2669563293457031e-05,
      "model_forward_time": 1.724548101425171,
      "step": 13285
    },
    {
      "epoch": 1.2669563293457031e-05,
      "step": 13285,
      "training_step_time": 8.446780443191528
    },
    {
      "epoch": 1.2670516967773437e-05,
      "model_forward_time": 1.7204582691192627,
      "step": 13286
    },
    {
      "epoch": 1.2670516967773437e-05,
      "step": 13286,
      "training_step_time": 7.371018171310425
    },
    {
      "epoch": 1.2671470642089845e-05,
      "model_forward_time": 1.7240962982177734,
      "step": 13287
    },
    {
      "epoch": 1.2671470642089845e-05,
      "step": 13287,
      "training_step_time": 6.873166084289551
    },
    {
      "epoch": 1.267242431640625e-05,
      "model_forward_time": 1.7185115814208984,
      "step": 13288
    },
    {
      "epoch": 1.267242431640625e-05,
      "step": 13288,
      "training_step_time": 7.125525236129761
    },
    {
      "epoch": 1.2673377990722656e-05,
      "model_forward_time": 1.7278709411621094,
      "step": 13289
    },
    {
      "epoch": 1.2673377990722656e-05,
      "step": 13289,
      "training_step_time": 7.817406415939331
    },
    {
      "epoch": 1.2674331665039062e-05,
      "grad_norm": 0.2286214828491211,
      "learning_rate": 9.81332945270683e-06,
      "loss": 0.0631,
      "step": 13290
    },
    {
      "epoch": 1.2674331665039062e-05,
      "model_forward_time": 1.7110450267791748,
      "step": 13290
    },
    {
      "dynamics_loss_avg": 0.06612990163266659,
      "epoch": 1.2674331665039062e-05,
      "step": 13290
    },
    {
      "action_loss_avg": 0.0027832220832351593,
      "epoch": 1.2674331665039062e-05,
      "step": 13290
    },
    {
      "epoch": 1.2674331665039062e-05,
      "step": 13290,
      "training_step_time": 7.449767351150513
    },
    {
      "epoch": 1.2675285339355469e-05,
      "model_forward_time": 1.7227873802185059,
      "step": 13291
    },
    {
      "epoch": 1.2675285339355469e-05,
      "step": 13291,
      "training_step_time": 8.192600727081299
    },
    {
      "epoch": 1.2676239013671875e-05,
      "model_forward_time": 1.7225825786590576,
      "step": 13292
    },
    {
      "epoch": 1.2676239013671875e-05,
      "step": 13292,
      "training_step_time": 9.430596113204956
    },
    {
      "epoch": 1.267719268798828e-05,
      "model_forward_time": 2.26296329498291,
      "step": 13293
    },
    {
      "epoch": 1.267719268798828e-05,
      "step": 13293,
      "training_step_time": 9.357641696929932
    },
    {
      "epoch": 1.2678146362304688e-05,
      "model_forward_time": 1.77736234664917,
      "step": 13294
    },
    {
      "epoch": 1.2678146362304688e-05,
      "step": 13294,
      "training_step_time": 7.776470184326172
    },
    {
      "epoch": 1.2679100036621094e-05,
      "model_forward_time": 1.7236604690551758,
      "step": 13295
    },
    {
      "epoch": 1.2679100036621094e-05,
      "step": 13295,
      "training_step_time": 7.3288350105285645
    },
    {
      "epoch": 1.26800537109375e-05,
      "model_forward_time": 1.7220256328582764,
      "step": 13296
    },
    {
      "epoch": 1.26800537109375e-05,
      "step": 13296,
      "training_step_time": 7.575164079666138
    },
    {
      "epoch": 1.2681007385253907e-05,
      "model_forward_time": 1.7252662181854248,
      "step": 13297
    },
    {
      "epoch": 1.2681007385253907e-05,
      "step": 13297,
      "training_step_time": 7.066153049468994
    },
    {
      "epoch": 1.2681961059570313e-05,
      "model_forward_time": 1.744410514831543,
      "step": 13298
    },
    {
      "epoch": 1.2681961059570313e-05,
      "step": 13298,
      "training_step_time": 7.739730596542358
    },
    {
      "epoch": 1.2682914733886718e-05,
      "model_forward_time": 1.7235755920410156,
      "step": 13299
    },
    {
      "epoch": 1.2682914733886718e-05,
      "step": 13299,
      "training_step_time": 7.510358810424805
    },
    {
      "epoch": 1.2683868408203126e-05,
      "grad_norm": 0.23185604810714722,
      "learning_rate": 9.812881607788744e-06,
      "loss": 0.0631,
      "step": 13300
    },
    {
      "epoch": 1.2683868408203126e-05,
      "model_forward_time": 1.7315196990966797,
      "step": 13300
    },
    {
      "dynamics_loss_avg": 0.05786842126399279,
      "epoch": 1.2683868408203126e-05,
      "step": 13300
    },
    {
      "action_loss_avg": 0.0037478990911040456,
      "epoch": 1.2683868408203126e-05,
      "step": 13300
    },
    {
      "epoch": 1.2683868408203126e-05,
      "step": 13300,
      "training_step_time": 7.671401500701904
    },
    {
      "epoch": 1.2684822082519531e-05,
      "model_forward_time": 1.7186768054962158,
      "step": 13301
    },
    {
      "epoch": 1.2684822082519531e-05,
      "step": 13301,
      "training_step_time": 10.346510171890259
    },
    {
      "epoch": 1.2685775756835937e-05,
      "model_forward_time": 1.702437162399292,
      "step": 13302
    },
    {
      "epoch": 1.2685775756835937e-05,
      "step": 13302,
      "training_step_time": 8.580037593841553
    },
    {
      "epoch": 1.2686729431152345e-05,
      "model_forward_time": 2.308870315551758,
      "step": 13303
    },
    {
      "epoch": 1.2686729431152345e-05,
      "step": 13303,
      "training_step_time": 7.813814163208008
    },
    {
      "epoch": 1.268768310546875e-05,
      "model_forward_time": 1.7251524925231934,
      "step": 13304
    },
    {
      "epoch": 1.268768310546875e-05,
      "step": 13304,
      "training_step_time": 7.682104825973511
    },
    {
      "epoch": 1.2688636779785156e-05,
      "model_forward_time": 1.7197458744049072,
      "step": 13305
    },
    {
      "epoch": 1.2688636779785156e-05,
      "step": 13305,
      "training_step_time": 7.526183843612671
    },
    {
      "epoch": 1.2689590454101562e-05,
      "model_forward_time": 1.727393388748169,
      "step": 13306
    },
    {
      "epoch": 1.2689590454101562e-05,
      "step": 13306,
      "training_step_time": 7.385928153991699
    },
    {
      "epoch": 1.2690544128417969e-05,
      "model_forward_time": 1.7022068500518799,
      "step": 13307
    },
    {
      "epoch": 1.2690544128417969e-05,
      "step": 13307,
      "training_step_time": 7.688949346542358
    },
    {
      "epoch": 1.2691497802734375e-05,
      "model_forward_time": 1.7191903591156006,
      "step": 13308
    },
    {
      "epoch": 1.2691497802734375e-05,
      "step": 13308,
      "training_step_time": 7.484111547470093
    },
    {
      "epoch": 1.269245147705078e-05,
      "model_forward_time": 1.7244455814361572,
      "step": 13309
    },
    {
      "epoch": 1.269245147705078e-05,
      "step": 13309,
      "training_step_time": 7.652995586395264
    },
    {
      "epoch": 1.2693405151367188e-05,
      "grad_norm": 0.1363833099603653,
      "learning_rate": 9.812433236541162e-06,
      "loss": 0.0615,
      "step": 13310
    },
    {
      "epoch": 1.2693405151367188e-05,
      "model_forward_time": 1.7353219985961914,
      "step": 13310
    },
    {
      "dynamics_loss_avg": 0.059490554966032505,
      "epoch": 1.2693405151367188e-05,
      "step": 13310
    },
    {
      "action_loss_avg": 0.002685285930056125,
      "epoch": 1.2693405151367188e-05,
      "step": 13310
    },
    {
      "epoch": 1.2693405151367188e-05,
      "step": 13310,
      "training_step_time": 9.68311619758606
    },
    {
      "epoch": 1.2694358825683594e-05,
      "model_forward_time": 1.722306728363037,
      "step": 13311
    },
    {
      "epoch": 1.2694358825683594e-05,
      "step": 13311,
      "training_step_time": 10.473389387130737
    },
    {
      "epoch": 1.26953125e-05,
      "model_forward_time": 2.2898354530334473,
      "step": 13312
    },
    {
      "epoch": 1.26953125e-05,
      "step": 13312,
      "training_step_time": 8.403001546859741
    },
    {
      "epoch": 1.2696266174316407e-05,
      "model_forward_time": 1.7150743007659912,
      "step": 13313
    },
    {
      "epoch": 1.2696266174316407e-05,
      "step": 13313,
      "training_step_time": 7.6168835163116455
    },
    {
      "epoch": 1.2697219848632813e-05,
      "model_forward_time": 1.735887050628662,
      "step": 13314
    },
    {
      "epoch": 1.2697219848632813e-05,
      "step": 13314,
      "training_step_time": 6.863460063934326
    },
    {
      "epoch": 1.2698173522949218e-05,
      "model_forward_time": 1.7370383739471436,
      "step": 13315
    },
    {
      "epoch": 1.2698173522949218e-05,
      "step": 13315,
      "training_step_time": 7.440187931060791
    },
    {
      "epoch": 1.2699127197265626e-05,
      "model_forward_time": 1.7251017093658447,
      "step": 13316
    },
    {
      "epoch": 1.2699127197265626e-05,
      "step": 13316,
      "training_step_time": 7.551076173782349
    },
    {
      "epoch": 1.2700080871582031e-05,
      "model_forward_time": 1.7169287204742432,
      "step": 13317
    },
    {
      "epoch": 1.2700080871582031e-05,
      "step": 13317,
      "training_step_time": 8.077099323272705
    },
    {
      "epoch": 1.2701034545898437e-05,
      "model_forward_time": 1.7251629829406738,
      "step": 13318
    },
    {
      "epoch": 1.2701034545898437e-05,
      "step": 13318,
      "training_step_time": 8.115863561630249
    },
    {
      "epoch": 1.2701988220214845e-05,
      "model_forward_time": 1.725419044494629,
      "step": 13319
    },
    {
      "epoch": 1.2701988220214845e-05,
      "step": 13319,
      "training_step_time": 7.809039115905762
    },
    {
      "epoch": 1.270294189453125e-05,
      "grad_norm": 0.1517331600189209,
      "learning_rate": 9.811984339013116e-06,
      "loss": 0.0635,
      "step": 13320
    },
    {
      "epoch": 1.270294189453125e-05,
      "model_forward_time": 1.7202200889587402,
      "step": 13320
    },
    {
      "dynamics_loss_avg": 0.05630751363933086,
      "epoch": 1.270294189453125e-05,
      "step": 13320
    },
    {
      "action_loss_avg": 0.0032168915146030485,
      "epoch": 1.270294189453125e-05,
      "step": 13320
    },
    {
      "epoch": 1.270294189453125e-05,
      "step": 13320,
      "training_step_time": 11.129784345626831
    },
    {
      "epoch": 1.2703895568847656e-05,
      "model_forward_time": 2.2656373977661133,
      "step": 13321
    },
    {
      "epoch": 1.2703895568847656e-05,
      "step": 13321,
      "training_step_time": 9.132808446884155
    },
    {
      "epoch": 1.2704849243164062e-05,
      "model_forward_time": 1.7149336338043213,
      "step": 13322
    },
    {
      "epoch": 1.2704849243164062e-05,
      "step": 13322,
      "training_step_time": 7.790374994277954
    },
    {
      "epoch": 1.2705802917480469e-05,
      "model_forward_time": 1.7141046524047852,
      "step": 13323
    },
    {
      "epoch": 1.2705802917480469e-05,
      "step": 13323,
      "training_step_time": 7.458030939102173
    },
    {
      "epoch": 1.2706756591796875e-05,
      "model_forward_time": 1.734717607498169,
      "step": 13324
    },
    {
      "epoch": 1.2706756591796875e-05,
      "step": 13324,
      "training_step_time": 7.638648748397827
    },
    {
      "epoch": 1.270771026611328e-05,
      "model_forward_time": 1.7351198196411133,
      "step": 13325
    },
    {
      "epoch": 1.270771026611328e-05,
      "step": 13325,
      "training_step_time": 7.157451629638672
    },
    {
      "epoch": 1.2708663940429688e-05,
      "model_forward_time": 1.7126445770263672,
      "step": 13326
    },
    {
      "epoch": 1.2708663940429688e-05,
      "step": 13326,
      "training_step_time": 7.7552289962768555
    },
    {
      "epoch": 1.2709617614746094e-05,
      "model_forward_time": 1.7259562015533447,
      "step": 13327
    },
    {
      "epoch": 1.2709617614746094e-05,
      "step": 13327,
      "training_step_time": 8.236158847808838
    },
    {
      "epoch": 1.27105712890625e-05,
      "model_forward_time": 1.7201871871948242,
      "step": 13328
    },
    {
      "epoch": 1.27105712890625e-05,
      "step": 13328,
      "training_step_time": 8.426631927490234
    },
    {
      "epoch": 1.2711524963378907e-05,
      "model_forward_time": 1.71915864944458,
      "step": 13329
    },
    {
      "epoch": 1.2711524963378907e-05,
      "step": 13329,
      "training_step_time": 10.060804843902588
    },
    {
      "epoch": 1.2712478637695313e-05,
      "grad_norm": 0.23096370697021484,
      "learning_rate": 9.811534915253698e-06,
      "loss": 0.0661,
      "step": 13330
    },
    {
      "epoch": 1.2712478637695313e-05,
      "model_forward_time": 2.2607321739196777,
      "step": 13330
    },
    {
      "dynamics_loss_avg": 0.06526240706443787,
      "epoch": 1.2712478637695313e-05,
      "step": 13330
    },
    {
      "action_loss_avg": 0.003671816748101264,
      "epoch": 1.2712478637695313e-05,
      "step": 13330
    },
    {
      "epoch": 1.2712478637695313e-05,
      "step": 13330,
      "training_step_time": 8.340934991836548
    },
    {
      "epoch": 1.2713432312011718e-05,
      "model_forward_time": 1.7722597122192383,
      "step": 13331
    },
    {
      "epoch": 1.2713432312011718e-05,
      "step": 13331,
      "training_step_time": 7.677379608154297
    },
    {
      "epoch": 1.2714385986328126e-05,
      "model_forward_time": 1.7256231307983398,
      "step": 13332
    },
    {
      "epoch": 1.2714385986328126e-05,
      "step": 13332,
      "training_step_time": 7.494224309921265
    },
    {
      "epoch": 1.2715339660644531e-05,
      "model_forward_time": 1.723398208618164,
      "step": 13333
    },
    {
      "epoch": 1.2715339660644531e-05,
      "step": 13333,
      "training_step_time": 7.596982002258301
    },
    {
      "epoch": 1.2716293334960937e-05,
      "model_forward_time": 1.7030248641967773,
      "step": 13334
    },
    {
      "epoch": 1.2716293334960937e-05,
      "step": 13334,
      "training_step_time": 7.201197147369385
    },
    {
      "epoch": 1.2717247009277345e-05,
      "model_forward_time": 1.6972246170043945,
      "step": 13335
    },
    {
      "epoch": 1.2717247009277345e-05,
      "step": 13335,
      "training_step_time": 8.422010898590088
    },
    {
      "epoch": 1.271820068359375e-05,
      "model_forward_time": 1.717167854309082,
      "step": 13336
    },
    {
      "epoch": 1.271820068359375e-05,
      "step": 13336,
      "training_step_time": 7.4840874671936035
    },
    {
      "epoch": 1.2719154357910156e-05,
      "model_forward_time": 1.7216975688934326,
      "step": 13337
    },
    {
      "epoch": 1.2719154357910156e-05,
      "step": 13337,
      "training_step_time": 8.572172164916992
    },
    {
      "epoch": 1.2720108032226562e-05,
      "model_forward_time": 1.7119414806365967,
      "step": 13338
    },
    {
      "epoch": 1.2720108032226562e-05,
      "step": 13338,
      "training_step_time": 8.348291397094727
    },
    {
      "epoch": 1.2721061706542969e-05,
      "model_forward_time": 2.283342123031616,
      "step": 13339
    },
    {
      "epoch": 1.2721061706542969e-05,
      "step": 13339,
      "training_step_time": 10.133421897888184
    },
    {
      "epoch": 1.2722015380859375e-05,
      "grad_norm": 0.2880057394504547,
      "learning_rate": 9.811084965312057e-06,
      "loss": 0.0655,
      "step": 13340
    },
    {
      "epoch": 1.2722015380859375e-05,
      "model_forward_time": 1.013972282409668,
      "step": 13340
    },
    {
      "dynamics_loss_avg": 0.05425156373530626,
      "epoch": 1.2722015380859375e-05,
      "step": 13340
    },
    {
      "action_loss_avg": 0.0023065569344908,
      "epoch": 1.2722015380859375e-05,
      "step": 13340
    },
    {
      "epoch": 1.2722015380859375e-05,
      "step": 13340,
      "training_step_time": 7.579374313354492
    },
    {
      "epoch": 1.272296905517578e-05,
      "model_forward_time": 1.7277007102966309,
      "step": 13341
    },
    {
      "epoch": 1.272296905517578e-05,
      "step": 13341,
      "training_step_time": 7.296913146972656
    },
    {
      "epoch": 1.2723922729492188e-05,
      "model_forward_time": 1.7175602912902832,
      "step": 13342
    },
    {
      "epoch": 1.2723922729492188e-05,
      "step": 13342,
      "training_step_time": 7.508085250854492
    },
    {
      "epoch": 1.2724876403808594e-05,
      "model_forward_time": 1.7169322967529297,
      "step": 13343
    },
    {
      "epoch": 1.2724876403808594e-05,
      "step": 13343,
      "training_step_time": 6.998134136199951
    },
    {
      "epoch": 1.2725830078125e-05,
      "model_forward_time": 1.7187669277191162,
      "step": 13344
    },
    {
      "epoch": 1.2725830078125e-05,
      "step": 13344,
      "training_step_time": 7.591669797897339
    },
    {
      "epoch": 1.2726783752441407e-05,
      "model_forward_time": 1.724299669265747,
      "step": 13345
    },
    {
      "epoch": 1.2726783752441407e-05,
      "step": 13345,
      "training_step_time": 7.4566969871521
    },
    {
      "epoch": 1.2727737426757813e-05,
      "model_forward_time": 1.7144432067871094,
      "step": 13346
    },
    {
      "epoch": 1.2727737426757813e-05,
      "step": 13346,
      "training_step_time": 7.751374244689941
    },
    {
      "epoch": 1.2728691101074218e-05,
      "model_forward_time": 1.7207467555999756,
      "step": 13347
    },
    {
      "epoch": 1.2728691101074218e-05,
      "step": 13347,
      "training_step_time": 7.845702409744263
    },
    {
      "epoch": 1.2729644775390626e-05,
      "model_forward_time": 1.7261202335357666,
      "step": 13348
    },
    {
      "epoch": 1.2729644775390626e-05,
      "step": 13348,
      "training_step_time": 8.453322649002075
    },
    {
      "epoch": 1.2730598449707031e-05,
      "model_forward_time": 2.30598521232605,
      "step": 13349
    },
    {
      "epoch": 1.2730598449707031e-05,
      "step": 13349,
      "training_step_time": 7.840487957000732
    },
    {
      "epoch": 1.2731552124023437e-05,
      "grad_norm": 0.2386346459388733,
      "learning_rate": 9.810634489237396e-06,
      "loss": 0.0679,
      "step": 13350
    },
    {
      "epoch": 1.2731552124023437e-05,
      "model_forward_time": 1.7060275077819824,
      "step": 13350
    },
    {
      "dynamics_loss_avg": 0.056932912208139896,
      "epoch": 1.2731552124023437e-05,
      "step": 13350
    },
    {
      "action_loss_avg": 0.004473203304223716,
      "epoch": 1.2731552124023437e-05,
      "step": 13350
    },
    {
      "epoch": 1.2731552124023437e-05,
      "step": 13350,
      "training_step_time": 7.80385422706604
    },
    {
      "epoch": 1.2732505798339845e-05,
      "model_forward_time": 1.3588569164276123,
      "step": 13351
    },
    {
      "epoch": 1.2732505798339845e-05,
      "step": 13351,
      "training_step_time": 7.003583669662476
    },
    {
      "epoch": 1.273345947265625e-05,
      "model_forward_time": 1.7152178287506104,
      "step": 13352
    },
    {
      "epoch": 1.273345947265625e-05,
      "step": 13352,
      "training_step_time": 7.63451075553894
    },
    {
      "epoch": 1.2734413146972656e-05,
      "model_forward_time": 1.6924734115600586,
      "step": 13353
    },
    {
      "epoch": 1.2734413146972656e-05,
      "step": 13353,
      "training_step_time": 7.905495882034302
    },
    {
      "epoch": 1.2735366821289062e-05,
      "model_forward_time": 1.7217254638671875,
      "step": 13354
    },
    {
      "epoch": 1.2735366821289062e-05,
      "step": 13354,
      "training_step_time": 7.687520265579224
    },
    {
      "epoch": 1.273632049560547e-05,
      "model_forward_time": 1.7217750549316406,
      "step": 13355
    },
    {
      "epoch": 1.273632049560547e-05,
      "step": 13355,
      "training_step_time": 7.924489259719849
    },
    {
      "epoch": 1.2737274169921875e-05,
      "model_forward_time": 1.7202904224395752,
      "step": 13356
    },
    {
      "epoch": 1.2737274169921875e-05,
      "step": 13356,
      "training_step_time": 7.94568133354187
    },
    {
      "epoch": 1.273822784423828e-05,
      "model_forward_time": 1.718137264251709,
      "step": 13357
    },
    {
      "epoch": 1.273822784423828e-05,
      "step": 13357,
      "training_step_time": 10.770293712615967
    },
    {
      "epoch": 1.2739181518554688e-05,
      "model_forward_time": 2.3064963817596436,
      "step": 13358
    },
    {
      "epoch": 1.2739181518554688e-05,
      "step": 13358,
      "training_step_time": 8.419722557067871
    },
    {
      "epoch": 1.2740135192871094e-05,
      "model_forward_time": 1.7206366062164307,
      "step": 13359
    },
    {
      "epoch": 1.2740135192871094e-05,
      "step": 13359,
      "training_step_time": 7.838889122009277
    },
    {
      "epoch": 1.27410888671875e-05,
      "grad_norm": 0.30912116169929504,
      "learning_rate": 9.81018348707898e-06,
      "loss": 0.0645,
      "step": 13360
    },
    {
      "epoch": 1.27410888671875e-05,
      "model_forward_time": 1.69429612159729,
      "step": 13360
    },
    {
      "dynamics_loss_avg": 0.04662957899272442,
      "epoch": 1.27410888671875e-05,
      "step": 13360
    },
    {
      "action_loss_avg": 0.0028096935129724443,
      "epoch": 1.27410888671875e-05,
      "step": 13360
    },
    {
      "epoch": 1.27410888671875e-05,
      "step": 13360,
      "training_step_time": 7.330386638641357
    },
    {
      "epoch": 1.2742042541503907e-05,
      "model_forward_time": 1.720857858657837,
      "step": 13361
    },
    {
      "epoch": 1.2742042541503907e-05,
      "step": 13361,
      "training_step_time": 7.63314962387085
    },
    {
      "epoch": 1.2742996215820313e-05,
      "model_forward_time": 1.717320442199707,
      "step": 13362
    },
    {
      "epoch": 1.2742996215820313e-05,
      "step": 13362,
      "training_step_time": 7.786947727203369
    },
    {
      "epoch": 1.2743949890136718e-05,
      "model_forward_time": 1.718141794204712,
      "step": 13363
    },
    {
      "epoch": 1.2743949890136718e-05,
      "step": 13363,
      "training_step_time": 7.728049278259277
    },
    {
      "epoch": 1.2744903564453126e-05,
      "model_forward_time": 1.721888780593872,
      "step": 13364
    },
    {
      "epoch": 1.2744903564453126e-05,
      "step": 13364,
      "training_step_time": 7.9373779296875
    },
    {
      "epoch": 1.2745857238769531e-05,
      "model_forward_time": 1.7145311832427979,
      "step": 13365
    },
    {
      "epoch": 1.2745857238769531e-05,
      "step": 13365,
      "training_step_time": 8.979816436767578
    },
    {
      "epoch": 1.2746810913085937e-05,
      "model_forward_time": 1.7235348224639893,
      "step": 13366
    },
    {
      "epoch": 1.2746810913085937e-05,
      "step": 13366,
      "training_step_time": 8.90853476524353
    },
    {
      "epoch": 1.2747764587402345e-05,
      "model_forward_time": 2.295379400253296,
      "step": 13367
    },
    {
      "epoch": 1.2747764587402345e-05,
      "step": 13367,
      "training_step_time": 7.849188804626465
    },
    {
      "epoch": 1.274871826171875e-05,
      "model_forward_time": 1.7207744121551514,
      "step": 13368
    },
    {
      "epoch": 1.274871826171875e-05,
      "step": 13368,
      "training_step_time": 7.83988356590271
    },
    {
      "epoch": 1.2749671936035156e-05,
      "model_forward_time": 1.7217116355895996,
      "step": 13369
    },
    {
      "epoch": 1.2749671936035156e-05,
      "step": 13369,
      "training_step_time": 7.583423852920532
    },
    {
      "epoch": 1.2750625610351562e-05,
      "grad_norm": 0.28174614906311035,
      "learning_rate": 9.80973195888613e-06,
      "loss": 0.0672,
      "step": 13370
    },
    {
      "epoch": 1.2750625610351562e-05,
      "model_forward_time": 1.7147371768951416,
      "step": 13370
    },
    {
      "dynamics_loss_avg": 0.06248622741550207,
      "epoch": 1.2750625610351562e-05,
      "step": 13370
    },
    {
      "action_loss_avg": 0.003811340202810243,
      "epoch": 1.2750625610351562e-05,
      "step": 13370
    },
    {
      "epoch": 1.2750625610351562e-05,
      "step": 13370,
      "training_step_time": 6.989372968673706
    },
    {
      "epoch": 1.275157928466797e-05,
      "model_forward_time": 1.7228586673736572,
      "step": 13371
    },
    {
      "epoch": 1.275157928466797e-05,
      "step": 13371,
      "training_step_time": 8.155992984771729
    },
    {
      "epoch": 1.2752532958984375e-05,
      "model_forward_time": 1.7269902229309082,
      "step": 13372
    },
    {
      "epoch": 1.2752532958984375e-05,
      "step": 13372,
      "training_step_time": 7.50250506401062
    },
    {
      "epoch": 1.275348663330078e-05,
      "model_forward_time": 1.7261016368865967,
      "step": 13373
    },
    {
      "epoch": 1.275348663330078e-05,
      "step": 13373,
      "training_step_time": 7.915060997009277
    },
    {
      "epoch": 1.2754440307617188e-05,
      "model_forward_time": 1.7165288925170898,
      "step": 13374
    },
    {
      "epoch": 1.2754440307617188e-05,
      "step": 13374,
      "training_step_time": 7.839750528335571
    },
    {
      "epoch": 1.2755393981933594e-05,
      "model_forward_time": 1.722266435623169,
      "step": 13375
    },
    {
      "epoch": 1.2755393981933594e-05,
      "step": 13375,
      "training_step_time": 8.9044508934021
    },
    {
      "epoch": 1.275634765625e-05,
      "model_forward_time": 1.7273681163787842,
      "step": 13376
    },
    {
      "epoch": 1.275634765625e-05,
      "step": 13376,
      "training_step_time": 8.891341924667358
    },
    {
      "epoch": 1.2757301330566407e-05,
      "model_forward_time": 1.7740509510040283,
      "step": 13377
    },
    {
      "epoch": 1.2757301330566407e-05,
      "step": 13377,
      "training_step_time": 7.875314950942993
    },
    {
      "epoch": 1.2758255004882813e-05,
      "model_forward_time": 1.7256619930267334,
      "step": 13378
    },
    {
      "epoch": 1.2758255004882813e-05,
      "step": 13378,
      "training_step_time": 8.400175094604492
    },
    {
      "epoch": 1.2759208679199218e-05,
      "model_forward_time": 1.733137845993042,
      "step": 13379
    },
    {
      "epoch": 1.2759208679199218e-05,
      "step": 13379,
      "training_step_time": 6.941099405288696
    },
    {
      "epoch": 1.2760162353515626e-05,
      "grad_norm": 0.23825189471244812,
      "learning_rate": 9.809279904708224e-06,
      "loss": 0.0645,
      "step": 13380
    },
    {
      "epoch": 1.2760162353515626e-05,
      "model_forward_time": 1.7153122425079346,
      "step": 13380
    },
    {
      "dynamics_loss_avg": 0.07727172560989856,
      "epoch": 1.2760162353515626e-05,
      "step": 13380
    },
    {
      "action_loss_avg": 0.00500441063195467,
      "epoch": 1.2760162353515626e-05,
      "step": 13380
    },
    {
      "epoch": 1.2760162353515626e-05,
      "step": 13380,
      "training_step_time": 8.257462978363037
    },
    {
      "epoch": 1.2761116027832031e-05,
      "model_forward_time": 1.7273218631744385,
      "step": 13381
    },
    {
      "epoch": 1.2761116027832031e-05,
      "step": 13381,
      "training_step_time": 7.839856863021851
    },
    {
      "epoch": 1.2762069702148437e-05,
      "model_forward_time": 1.7296698093414307,
      "step": 13382
    },
    {
      "epoch": 1.2762069702148437e-05,
      "step": 13382,
      "training_step_time": 7.799825191497803
    },
    {
      "epoch": 1.2763023376464843e-05,
      "model_forward_time": 1.7249603271484375,
      "step": 13383
    },
    {
      "epoch": 1.2763023376464843e-05,
      "step": 13383,
      "training_step_time": 8.401894807815552
    },
    {
      "epoch": 1.276397705078125e-05,
      "model_forward_time": 1.7175912857055664,
      "step": 13384
    },
    {
      "epoch": 1.276397705078125e-05,
      "step": 13384,
      "training_step_time": 7.944607973098755
    },
    {
      "epoch": 1.2764930725097656e-05,
      "model_forward_time": 1.3519494533538818,
      "step": 13385
    },
    {
      "epoch": 1.2764930725097656e-05,
      "step": 13385,
      "training_step_time": 8.454422235488892
    },
    {
      "epoch": 1.2765884399414062e-05,
      "model_forward_time": 2.272496223449707,
      "step": 13386
    },
    {
      "epoch": 1.2765884399414062e-05,
      "step": 13386,
      "training_step_time": 8.823693990707397
    },
    {
      "epoch": 1.276683807373047e-05,
      "model_forward_time": 1.7122082710266113,
      "step": 13387
    },
    {
      "epoch": 1.276683807373047e-05,
      "step": 13387,
      "training_step_time": 7.5648815631866455
    },
    {
      "epoch": 1.2767791748046875e-05,
      "model_forward_time": 1.7188713550567627,
      "step": 13388
    },
    {
      "epoch": 1.2767791748046875e-05,
      "step": 13388,
      "training_step_time": 7.432963848114014
    },
    {
      "epoch": 1.276874542236328e-05,
      "model_forward_time": 1.72001314163208,
      "step": 13389
    },
    {
      "epoch": 1.276874542236328e-05,
      "step": 13389,
      "training_step_time": 7.6908509731292725
    },
    {
      "epoch": 1.2769699096679688e-05,
      "grad_norm": 0.20958931744098663,
      "learning_rate": 9.808827324594698e-06,
      "loss": 0.0657,
      "step": 13390
    },
    {
      "epoch": 1.2769699096679688e-05,
      "model_forward_time": 1.7219042778015137,
      "step": 13390
    },
    {
      "dynamics_loss_avg": 0.04985749991610646,
      "epoch": 1.2769699096679688e-05,
      "step": 13390
    },
    {
      "action_loss_avg": 0.002027918113162741,
      "epoch": 1.2769699096679688e-05,
      "step": 13390
    },
    {
      "epoch": 1.2769699096679688e-05,
      "step": 13390,
      "training_step_time": 7.617359399795532
    },
    {
      "epoch": 1.2770652770996094e-05,
      "model_forward_time": 1.7223186492919922,
      "step": 13391
    },
    {
      "epoch": 1.2770652770996094e-05,
      "step": 13391,
      "training_step_time": 8.339853286743164
    },
    {
      "epoch": 1.27716064453125e-05,
      "model_forward_time": 1.718961477279663,
      "step": 13392
    },
    {
      "epoch": 1.27716064453125e-05,
      "step": 13392,
      "training_step_time": 8.477265119552612
    },
    {
      "epoch": 1.2772560119628907e-05,
      "model_forward_time": 1.7225077152252197,
      "step": 13393
    },
    {
      "epoch": 1.2772560119628907e-05,
      "step": 13393,
      "training_step_time": 7.913844347000122
    },
    {
      "epoch": 1.2773513793945313e-05,
      "model_forward_time": 1.7190008163452148,
      "step": 13394
    },
    {
      "epoch": 1.2773513793945313e-05,
      "step": 13394,
      "training_step_time": 9.365647792816162
    },
    {
      "epoch": 1.2774467468261718e-05,
      "model_forward_time": 2.262307643890381,
      "step": 13395
    },
    {
      "epoch": 1.2774467468261718e-05,
      "step": 13395,
      "training_step_time": 8.455422401428223
    },
    {
      "epoch": 1.2775421142578126e-05,
      "model_forward_time": 1.719451665878296,
      "step": 13396
    },
    {
      "epoch": 1.2775421142578126e-05,
      "step": 13396,
      "training_step_time": 7.7475292682647705
    },
    {
      "epoch": 1.2776374816894531e-05,
      "model_forward_time": 1.7233474254608154,
      "step": 13397
    },
    {
      "epoch": 1.2776374816894531e-05,
      "step": 13397,
      "training_step_time": 7.401259899139404
    },
    {
      "epoch": 1.2777328491210937e-05,
      "model_forward_time": 1.721097469329834,
      "step": 13398
    },
    {
      "epoch": 1.2777328491210937e-05,
      "step": 13398,
      "training_step_time": 7.656294822692871
    },
    {
      "epoch": 1.2778282165527343e-05,
      "model_forward_time": 1.715808629989624,
      "step": 13399
    },
    {
      "epoch": 1.2778282165527343e-05,
      "step": 13399,
      "training_step_time": 7.493472099304199
    },
    {
      "epoch": 1.277923583984375e-05,
      "grad_norm": 0.19884461164474487,
      "learning_rate": 9.808374218595048e-06,
      "loss": 0.0629,
      "step": 13400
    },
    {
      "epoch": 1.277923583984375e-05,
      "model_forward_time": 1.7184984683990479,
      "step": 13400
    },
    {
      "dynamics_loss_avg": 0.052835429273545745,
      "epoch": 1.277923583984375e-05,
      "step": 13400
    },
    {
      "action_loss_avg": 0.002405749843455851,
      "epoch": 1.277923583984375e-05,
      "step": 13400
    },
    {
      "epoch": 1.277923583984375e-05,
      "step": 13400,
      "training_step_time": 7.6992456912994385
    },
    {
      "epoch": 1.2780189514160156e-05,
      "model_forward_time": 1.7123494148254395,
      "step": 13401
    },
    {
      "epoch": 1.2780189514160156e-05,
      "step": 13401,
      "training_step_time": 8.492398977279663
    },
    {
      "epoch": 1.2781143188476562e-05,
      "model_forward_time": 1.7303152084350586,
      "step": 13402
    },
    {
      "epoch": 1.2781143188476562e-05,
      "step": 13402,
      "training_step_time": 8.916686058044434
    },
    {
      "epoch": 1.278209686279297e-05,
      "model_forward_time": 1.716007947921753,
      "step": 13403
    },
    {
      "epoch": 1.278209686279297e-05,
      "step": 13403,
      "training_step_time": 7.864858865737915
    },
    {
      "epoch": 1.2783050537109375e-05,
      "model_forward_time": 1.9250125885009766,
      "step": 13404
    },
    {
      "epoch": 1.2783050537109375e-05,
      "step": 13404,
      "training_step_time": 7.904516696929932
    },
    {
      "epoch": 1.278400421142578e-05,
      "model_forward_time": 1.7257397174835205,
      "step": 13405
    },
    {
      "epoch": 1.278400421142578e-05,
      "step": 13405,
      "training_step_time": 8.060807704925537
    },
    {
      "epoch": 1.2784957885742188e-05,
      "model_forward_time": 1.7193152904510498,
      "step": 13406
    },
    {
      "epoch": 1.2784957885742188e-05,
      "step": 13406,
      "training_step_time": 7.58358097076416
    },
    {
      "epoch": 1.2785911560058594e-05,
      "model_forward_time": 1.7192089557647705,
      "step": 13407
    },
    {
      "epoch": 1.2785911560058594e-05,
      "step": 13407,
      "training_step_time": 7.017770051956177
    },
    {
      "epoch": 1.2786865234375e-05,
      "model_forward_time": 1.705920696258545,
      "step": 13408
    },
    {
      "epoch": 1.2786865234375e-05,
      "step": 13408,
      "training_step_time": 7.691534757614136
    },
    {
      "epoch": 1.2787818908691407e-05,
      "model_forward_time": 1.717475414276123,
      "step": 13409
    },
    {
      "epoch": 1.2787818908691407e-05,
      "step": 13409,
      "training_step_time": 7.70262885093689
    },
    {
      "epoch": 1.2788772583007813e-05,
      "grad_norm": 0.2539592385292053,
      "learning_rate": 9.807920586758821e-06,
      "loss": 0.068,
      "step": 13410
    },
    {
      "epoch": 1.2788772583007813e-05,
      "model_forward_time": 1.7274773120880127,
      "step": 13410
    },
    {
      "dynamics_loss_avg": 0.07131464499980211,
      "epoch": 1.2788772583007813e-05,
      "step": 13410
    },
    {
      "action_loss_avg": 0.0026983393297996374,
      "epoch": 1.2788772583007813e-05,
      "step": 13410
    },
    {
      "epoch": 1.2788772583007813e-05,
      "step": 13410,
      "training_step_time": 8.651004791259766
    },
    {
      "epoch": 1.2789726257324218e-05,
      "model_forward_time": 1.348891019821167,
      "step": 13411
    },
    {
      "epoch": 1.2789726257324218e-05,
      "step": 13411,
      "training_step_time": 8.814515829086304
    },
    {
      "epoch": 1.2790679931640626e-05,
      "model_forward_time": 1.7285938262939453,
      "step": 13412
    },
    {
      "epoch": 1.2790679931640626e-05,
      "step": 13412,
      "training_step_time": 8.898889064788818
    },
    {
      "epoch": 1.2791633605957031e-05,
      "model_forward_time": 2.249671697616577,
      "step": 13413
    },
    {
      "epoch": 1.2791633605957031e-05,
      "step": 13413,
      "training_step_time": 7.723093271255493
    },
    {
      "epoch": 1.2792587280273437e-05,
      "model_forward_time": 1.7168989181518555,
      "step": 13414
    },
    {
      "epoch": 1.2792587280273437e-05,
      "step": 13414,
      "training_step_time": 7.835291862487793
    },
    {
      "epoch": 1.2793540954589843e-05,
      "model_forward_time": 1.7170393466949463,
      "step": 13415
    },
    {
      "epoch": 1.2793540954589843e-05,
      "step": 13415,
      "training_step_time": 7.5713019371032715
    },
    {
      "epoch": 1.279449462890625e-05,
      "model_forward_time": 1.7526142597198486,
      "step": 13416
    },
    {
      "epoch": 1.279449462890625e-05,
      "step": 13416,
      "training_step_time": 6.986781120300293
    },
    {
      "epoch": 1.2795448303222656e-05,
      "model_forward_time": 1.7216567993164062,
      "step": 13417
    },
    {
      "epoch": 1.2795448303222656e-05,
      "step": 13417,
      "training_step_time": 7.67237401008606
    },
    {
      "epoch": 1.2796401977539062e-05,
      "model_forward_time": 1.3472213745117188,
      "step": 13418
    },
    {
      "epoch": 1.2796401977539062e-05,
      "step": 13418,
      "training_step_time": 8.549246072769165
    },
    {
      "epoch": 1.279735565185547e-05,
      "model_forward_time": 1.7184998989105225,
      "step": 13419
    },
    {
      "epoch": 1.279735565185547e-05,
      "step": 13419,
      "training_step_time": 7.930133581161499
    },
    {
      "epoch": 1.2798309326171875e-05,
      "grad_norm": 0.2587926387786865,
      "learning_rate": 9.807466429135627e-06,
      "loss": 0.0688,
      "step": 13420
    },
    {
      "epoch": 1.2798309326171875e-05,
      "model_forward_time": 1.725632905960083,
      "step": 13420
    },
    {
      "dynamics_loss_avg": 0.05478844605386257,
      "epoch": 1.2798309326171875e-05,
      "step": 13420
    },
    {
      "action_loss_avg": 0.0023631881398614497,
      "epoch": 1.2798309326171875e-05,
      "step": 13420
    },
    {
      "epoch": 1.2798309326171875e-05,
      "step": 13420,
      "training_step_time": 8.830513954162598
    },
    {
      "epoch": 1.279926300048828e-05,
      "model_forward_time": 1.7217907905578613,
      "step": 13421
    },
    {
      "epoch": 1.279926300048828e-05,
      "step": 13421,
      "training_step_time": 10.41998839378357
    },
    {
      "epoch": 1.2800216674804688e-05,
      "model_forward_time": 1.7226099967956543,
      "step": 13422
    },
    {
      "epoch": 1.2800216674804688e-05,
      "step": 13422,
      "training_step_time": 9.75501275062561
    },
    {
      "epoch": 1.2801170349121094e-05,
      "model_forward_time": 2.271528959274292,
      "step": 13423
    },
    {
      "epoch": 1.2801170349121094e-05,
      "step": 13423,
      "training_step_time": 8.449719190597534
    },
    {
      "epoch": 1.28021240234375e-05,
      "model_forward_time": 1.7245078086853027,
      "step": 13424
    },
    {
      "epoch": 1.28021240234375e-05,
      "step": 13424,
      "training_step_time": 7.772963523864746
    },
    {
      "epoch": 1.2803077697753907e-05,
      "model_forward_time": 1.6980576515197754,
      "step": 13425
    },
    {
      "epoch": 1.2803077697753907e-05,
      "step": 13425,
      "training_step_time": 7.259976148605347
    },
    {
      "epoch": 1.2804031372070313e-05,
      "model_forward_time": 1.7219641208648682,
      "step": 13426
    },
    {
      "epoch": 1.2804031372070313e-05,
      "step": 13426,
      "training_step_time": 7.57414984703064
    },
    {
      "epoch": 1.2804985046386718e-05,
      "model_forward_time": 1.728161334991455,
      "step": 13427
    },
    {
      "epoch": 1.2804985046386718e-05,
      "step": 13427,
      "training_step_time": 7.7334418296813965
    },
    {
      "epoch": 1.2805938720703126e-05,
      "model_forward_time": 1.7227308750152588,
      "step": 13428
    },
    {
      "epoch": 1.2805938720703126e-05,
      "step": 13428,
      "training_step_time": 8.578440427780151
    },
    {
      "epoch": 1.2806892395019532e-05,
      "model_forward_time": 1.7197012901306152,
      "step": 13429
    },
    {
      "epoch": 1.2806892395019532e-05,
      "step": 13429,
      "training_step_time": 8.340682744979858
    },
    {
      "epoch": 1.2807846069335937e-05,
      "grad_norm": 0.2775968313217163,
      "learning_rate": 9.807011745775132e-06,
      "loss": 0.0643,
      "step": 13430
    },
    {
      "epoch": 1.2807846069335937e-05,
      "model_forward_time": 1.6942176818847656,
      "step": 13430
    },
    {
      "dynamics_loss_avg": 0.054225437063723805,
      "epoch": 1.2807846069335937e-05,
      "step": 13430
    },
    {
      "action_loss_avg": 0.002887179306708276,
      "epoch": 1.2807846069335937e-05,
      "step": 13430
    },
    {
      "epoch": 1.2807846069335937e-05,
      "step": 13430,
      "training_step_time": 9.080770254135132
    },
    {
      "epoch": 1.2808799743652343e-05,
      "model_forward_time": 1.7253203392028809,
      "step": 13431
    },
    {
      "epoch": 1.2808799743652343e-05,
      "step": 13431,
      "training_step_time": 9.530720233917236
    },
    {
      "epoch": 1.280975341796875e-05,
      "model_forward_time": 2.2816107273101807,
      "step": 13432
    },
    {
      "epoch": 1.280975341796875e-05,
      "step": 13432,
      "training_step_time": 8.75313401222229
    },
    {
      "epoch": 1.2810707092285156e-05,
      "model_forward_time": 1.7145333290100098,
      "step": 13433
    },
    {
      "epoch": 1.2810707092285156e-05,
      "step": 13433,
      "training_step_time": 7.751447677612305
    },
    {
      "epoch": 1.2811660766601562e-05,
      "model_forward_time": 1.3402509689331055,
      "step": 13434
    },
    {
      "epoch": 1.2811660766601562e-05,
      "step": 13434,
      "training_step_time": 7.389063358306885
    },
    {
      "epoch": 1.281261444091797e-05,
      "model_forward_time": 1.6883962154388428,
      "step": 13435
    },
    {
      "epoch": 1.281261444091797e-05,
      "step": 13435,
      "training_step_time": 7.199438810348511
    },
    {
      "epoch": 1.2813568115234375e-05,
      "model_forward_time": 1.7162184715270996,
      "step": 13436
    },
    {
      "epoch": 1.2813568115234375e-05,
      "step": 13436,
      "training_step_time": 7.669585704803467
    },
    {
      "epoch": 1.281452178955078e-05,
      "model_forward_time": 1.7248706817626953,
      "step": 13437
    },
    {
      "epoch": 1.281452178955078e-05,
      "step": 13437,
      "training_step_time": 7.577665328979492
    },
    {
      "epoch": 1.2815475463867188e-05,
      "model_forward_time": 1.7111952304840088,
      "step": 13438
    },
    {
      "epoch": 1.2815475463867188e-05,
      "step": 13438,
      "training_step_time": 8.954943895339966
    },
    {
      "epoch": 1.2816429138183594e-05,
      "model_forward_time": 1.733370065689087,
      "step": 13439
    },
    {
      "epoch": 1.2816429138183594e-05,
      "step": 13439,
      "training_step_time": 8.753987073898315
    },
    {
      "epoch": 1.28173828125e-05,
      "grad_norm": 0.2700028717517853,
      "learning_rate": 9.80655653672706e-06,
      "loss": 0.0605,
      "step": 13440
    },
    {
      "epoch": 1.28173828125e-05,
      "model_forward_time": 1.7117812633514404,
      "step": 13440
    },
    {
      "dynamics_loss_avg": 0.05717929396778345,
      "epoch": 1.28173828125e-05,
      "step": 13440
    },
    {
      "action_loss_avg": 0.0023720033233985303,
      "epoch": 1.28173828125e-05,
      "step": 13440
    },
    {
      "epoch": 1.28173828125e-05,
      "step": 13440,
      "training_step_time": 10.189446687698364
    },
    {
      "epoch": 1.2818336486816407e-05,
      "model_forward_time": 2.27645206451416,
      "step": 13441
    },
    {
      "epoch": 1.2818336486816407e-05,
      "step": 13441,
      "training_step_time": 8.637542247772217
    },
    {
      "epoch": 1.2819290161132813e-05,
      "model_forward_time": 1.7403783798217773,
      "step": 13442
    },
    {
      "epoch": 1.2819290161132813e-05,
      "step": 13442,
      "training_step_time": 7.8393425941467285
    },
    {
      "epoch": 1.2820243835449218e-05,
      "model_forward_time": 1.72676682472229,
      "step": 13443
    },
    {
      "epoch": 1.2820243835449218e-05,
      "step": 13443,
      "training_step_time": 7.400495767593384
    },
    {
      "epoch": 1.2821197509765626e-05,
      "model_forward_time": 1.7238619327545166,
      "step": 13444
    },
    {
      "epoch": 1.2821197509765626e-05,
      "step": 13444,
      "training_step_time": 7.158445835113525
    },
    {
      "epoch": 1.2822151184082032e-05,
      "model_forward_time": 1.7302892208099365,
      "step": 13445
    },
    {
      "epoch": 1.2822151184082032e-05,
      "step": 13445,
      "training_step_time": 8.70796537399292
    },
    {
      "epoch": 1.2823104858398437e-05,
      "model_forward_time": 1.7327921390533447,
      "step": 13446
    },
    {
      "epoch": 1.2823104858398437e-05,
      "step": 13446,
      "training_step_time": 8.40840220451355
    },
    {
      "epoch": 1.2824058532714843e-05,
      "model_forward_time": 1.7290198802947998,
      "step": 13447
    },
    {
      "epoch": 1.2824058532714843e-05,
      "step": 13447,
      "training_step_time": 8.89961552619934
    },
    {
      "epoch": 1.282501220703125e-05,
      "model_forward_time": 1.7352981567382812,
      "step": 13448
    },
    {
      "epoch": 1.282501220703125e-05,
      "step": 13448,
      "training_step_time": 8.689125776290894
    },
    {
      "epoch": 1.2825965881347656e-05,
      "model_forward_time": 1.7274446487426758,
      "step": 13449
    },
    {
      "epoch": 1.2825965881347656e-05,
      "step": 13449,
      "training_step_time": 7.843280792236328
    },
    {
      "epoch": 1.2826919555664062e-05,
      "grad_norm": 0.20978930592536926,
      "learning_rate": 9.806100802041192e-06,
      "loss": 0.0585,
      "step": 13450
    },
    {
      "epoch": 1.2826919555664062e-05,
      "model_forward_time": 1.7824814319610596,
      "step": 13450
    },
    {
      "dynamics_loss_avg": 0.0610340716317296,
      "epoch": 1.2826919555664062e-05,
      "step": 13450
    },
    {
      "action_loss_avg": 0.004217695316765457,
      "epoch": 1.2826919555664062e-05,
      "step": 13450
    },
    {
      "epoch": 1.2826919555664062e-05,
      "step": 13450,
      "training_step_time": 8.377115488052368
    },
    {
      "epoch": 1.282787322998047e-05,
      "model_forward_time": 2.2588651180267334,
      "step": 13451
    },
    {
      "epoch": 1.282787322998047e-05,
      "step": 13451,
      "training_step_time": 7.876831769943237
    },
    {
      "epoch": 1.2828826904296875e-05,
      "model_forward_time": 1.725188970565796,
      "step": 13452
    },
    {
      "epoch": 1.2828826904296875e-05,
      "step": 13452,
      "training_step_time": 7.676831245422363
    },
    {
      "epoch": 1.282978057861328e-05,
      "model_forward_time": 1.7336559295654297,
      "step": 13453
    },
    {
      "epoch": 1.282978057861328e-05,
      "step": 13453,
      "training_step_time": 7.224876642227173
    },
    {
      "epoch": 1.2830734252929688e-05,
      "model_forward_time": 1.746079444885254,
      "step": 13454
    },
    {
      "epoch": 1.2830734252929688e-05,
      "step": 13454,
      "training_step_time": 7.701647520065308
    },
    {
      "epoch": 1.2831687927246094e-05,
      "model_forward_time": 1.7288336753845215,
      "step": 13455
    },
    {
      "epoch": 1.2831687927246094e-05,
      "step": 13455,
      "training_step_time": 7.600793361663818
    },
    {
      "epoch": 1.28326416015625e-05,
      "model_forward_time": 1.7384047508239746,
      "step": 13456
    },
    {
      "epoch": 1.28326416015625e-05,
      "step": 13456,
      "training_step_time": 7.898426294326782
    },
    {
      "epoch": 1.2833595275878907e-05,
      "model_forward_time": 1.7279391288757324,
      "step": 13457
    },
    {
      "epoch": 1.2833595275878907e-05,
      "step": 13457,
      "training_step_time": 8.184694051742554
    },
    {
      "epoch": 1.2834548950195313e-05,
      "model_forward_time": 1.7306678295135498,
      "step": 13458
    },
    {
      "epoch": 1.2834548950195313e-05,
      "step": 13458,
      "training_step_time": 8.58609390258789
    },
    {
      "epoch": 1.2835502624511718e-05,
      "model_forward_time": 1.7879853248596191,
      "step": 13459
    },
    {
      "epoch": 1.2835502624511718e-05,
      "step": 13459,
      "training_step_time": 9.444305181503296
    },
    {
      "epoch": 1.2836456298828126e-05,
      "grad_norm": 0.18687885999679565,
      "learning_rate": 9.805644541767367e-06,
      "loss": 0.0605,
      "step": 13460
    },
    {
      "epoch": 1.2836456298828126e-05,
      "model_forward_time": 2.2734477519989014,
      "step": 13460
    },
    {
      "dynamics_loss_avg": 0.04977595116943121,
      "epoch": 1.2836456298828126e-05,
      "step": 13460
    },
    {
      "action_loss_avg": 0.0022107875032816083,
      "epoch": 1.2836456298828126e-05,
      "step": 13460
    },
    {
      "epoch": 1.2836456298828126e-05,
      "step": 13460,
      "training_step_time": 8.437141180038452
    },
    {
      "epoch": 1.2837409973144532e-05,
      "model_forward_time": 1.7331249713897705,
      "step": 13461
    },
    {
      "epoch": 1.2837409973144532e-05,
      "step": 13461,
      "training_step_time": 7.55589747428894
    },
    {
      "epoch": 1.2838363647460937e-05,
      "model_forward_time": 1.727884292602539,
      "step": 13462
    },
    {
      "epoch": 1.2838363647460937e-05,
      "step": 13462,
      "training_step_time": 7.538750886917114
    },
    {
      "epoch": 1.2839317321777343e-05,
      "model_forward_time": 1.7339863777160645,
      "step": 13463
    },
    {
      "epoch": 1.2839317321777343e-05,
      "step": 13463,
      "training_step_time": 8.198389530181885
    },
    {
      "epoch": 1.284027099609375e-05,
      "model_forward_time": 1.7284204959869385,
      "step": 13464
    },
    {
      "epoch": 1.284027099609375e-05,
      "step": 13464,
      "training_step_time": 7.696391820907593
    },
    {
      "epoch": 1.2841224670410156e-05,
      "model_forward_time": 1.7290332317352295,
      "step": 13465
    },
    {
      "epoch": 1.2841224670410156e-05,
      "step": 13465,
      "training_step_time": 8.036391973495483
    },
    {
      "epoch": 1.2842178344726562e-05,
      "model_forward_time": 1.720588207244873,
      "step": 13466
    },
    {
      "epoch": 1.2842178344726562e-05,
      "step": 13466,
      "training_step_time": 8.77045226097107
    },
    {
      "epoch": 1.284313201904297e-05,
      "model_forward_time": 1.7353556156158447,
      "step": 13467
    },
    {
      "epoch": 1.284313201904297e-05,
      "step": 13467,
      "training_step_time": 9.872238874435425
    },
    {
      "epoch": 1.2844085693359375e-05,
      "model_forward_time": 1.7801499366760254,
      "step": 13468
    },
    {
      "epoch": 1.2844085693359375e-05,
      "step": 13468,
      "training_step_time": 8.436224460601807
    },
    {
      "epoch": 1.284503936767578e-05,
      "model_forward_time": 2.283299446105957,
      "step": 13469
    },
    {
      "epoch": 1.284503936767578e-05,
      "step": 13469,
      "training_step_time": 7.775615930557251
    },
    {
      "epoch": 1.2845993041992188e-05,
      "grad_norm": 0.21597027778625488,
      "learning_rate": 9.805187755955479e-06,
      "loss": 0.0618,
      "step": 13470
    },
    {
      "epoch": 1.2845993041992188e-05,
      "model_forward_time": 1.729123592376709,
      "step": 13470
    },
    {
      "dynamics_loss_avg": 0.049193134717643264,
      "epoch": 1.2845993041992188e-05,
      "step": 13470
    },
    {
      "action_loss_avg": 0.0026586622174363584,
      "epoch": 1.2845993041992188e-05,
      "step": 13470
    },
    {
      "epoch": 1.2845993041992188e-05,
      "step": 13470,
      "training_step_time": 7.871651649475098
    },
    {
      "epoch": 1.2846946716308594e-05,
      "model_forward_time": 1.7243642807006836,
      "step": 13471
    },
    {
      "epoch": 1.2846946716308594e-05,
      "step": 13471,
      "training_step_time": 7.659451007843018
    },
    {
      "epoch": 1.2847900390625e-05,
      "model_forward_time": 1.7166593074798584,
      "step": 13472
    },
    {
      "epoch": 1.2847900390625e-05,
      "step": 13472,
      "training_step_time": 6.950825929641724
    },
    {
      "epoch": 1.2848854064941407e-05,
      "model_forward_time": 1.7437419891357422,
      "step": 13473
    },
    {
      "epoch": 1.2848854064941407e-05,
      "step": 13473,
      "training_step_time": 7.807273626327515
    },
    {
      "epoch": 1.2849807739257813e-05,
      "model_forward_time": 1.730743169784546,
      "step": 13474
    },
    {
      "epoch": 1.2849807739257813e-05,
      "step": 13474,
      "training_step_time": 8.635315418243408
    },
    {
      "epoch": 1.2850761413574218e-05,
      "model_forward_time": 1.7424893379211426,
      "step": 13475
    },
    {
      "epoch": 1.2850761413574218e-05,
      "step": 13475,
      "training_step_time": 8.922142028808594
    },
    {
      "epoch": 1.2851715087890626e-05,
      "model_forward_time": 1.7353792190551758,
      "step": 13476
    },
    {
      "epoch": 1.2851715087890626e-05,
      "step": 13476,
      "training_step_time": 8.417486667633057
    },
    {
      "epoch": 1.2852668762207032e-05,
      "model_forward_time": 1.76774001121521,
      "step": 13477
    },
    {
      "epoch": 1.2852668762207032e-05,
      "step": 13477,
      "training_step_time": 8.755258798599243
    },
    {
      "epoch": 1.2853622436523437e-05,
      "model_forward_time": 2.27339768409729,
      "step": 13478
    },
    {
      "epoch": 1.2853622436523437e-05,
      "step": 13478,
      "training_step_time": 8.577196598052979
    },
    {
      "epoch": 1.2854576110839843e-05,
      "model_forward_time": 1.7343974113464355,
      "step": 13479
    },
    {
      "epoch": 1.2854576110839843e-05,
      "step": 13479,
      "training_step_time": 7.950117826461792
    },
    {
      "epoch": 1.285552978515625e-05,
      "grad_norm": 0.1635395884513855,
      "learning_rate": 9.804730444655483e-06,
      "loss": 0.0625,
      "step": 13480
    },
    {
      "epoch": 1.285552978515625e-05,
      "model_forward_time": 1.7314484119415283,
      "step": 13480
    },
    {
      "dynamics_loss_avg": 0.0424218064174056,
      "epoch": 1.285552978515625e-05,
      "step": 13480
    },
    {
      "action_loss_avg": 0.0023557318723760544,
      "epoch": 1.285552978515625e-05,
      "step": 13480
    },
    {
      "epoch": 1.285552978515625e-05,
      "step": 13480,
      "training_step_time": 7.680009841918945
    },
    {
      "epoch": 1.2856483459472656e-05,
      "model_forward_time": 1.729799509048462,
      "step": 13481
    },
    {
      "epoch": 1.2856483459472656e-05,
      "step": 13481,
      "training_step_time": 7.0245795249938965
    },
    {
      "epoch": 1.2857437133789062e-05,
      "model_forward_time": 1.7286839485168457,
      "step": 13482
    },
    {
      "epoch": 1.2857437133789062e-05,
      "step": 13482,
      "training_step_time": 9.028693675994873
    },
    {
      "epoch": 1.285839080810547e-05,
      "model_forward_time": 1.7510888576507568,
      "step": 13483
    },
    {
      "epoch": 1.285839080810547e-05,
      "step": 13483,
      "training_step_time": 7.642842531204224
    },
    {
      "epoch": 1.2859344482421875e-05,
      "model_forward_time": 1.73585844039917,
      "step": 13484
    },
    {
      "epoch": 1.2859344482421875e-05,
      "step": 13484,
      "training_step_time": 7.969502687454224
    },
    {
      "epoch": 1.286029815673828e-05,
      "model_forward_time": 1.7310116291046143,
      "step": 13485
    },
    {
      "epoch": 1.286029815673828e-05,
      "step": 13485,
      "training_step_time": 8.905669927597046
    },
    {
      "epoch": 1.2861251831054688e-05,
      "model_forward_time": 1.7316718101501465,
      "step": 13486
    },
    {
      "epoch": 1.2861251831054688e-05,
      "step": 13486,
      "training_step_time": 8.820282459259033
    },
    {
      "epoch": 1.2862205505371094e-05,
      "model_forward_time": 2.290886878967285,
      "step": 13487
    },
    {
      "epoch": 1.2862205505371094e-05,
      "step": 13487,
      "training_step_time": 7.830712080001831
    },
    {
      "epoch": 1.28631591796875e-05,
      "model_forward_time": 1.7334284782409668,
      "step": 13488
    },
    {
      "epoch": 1.28631591796875e-05,
      "step": 13488,
      "training_step_time": 8.694006443023682
    },
    {
      "epoch": 1.2864112854003907e-05,
      "model_forward_time": 1.7350578308105469,
      "step": 13489
    },
    {
      "epoch": 1.2864112854003907e-05,
      "step": 13489,
      "training_step_time": 7.624391794204712
    },
    {
      "epoch": 1.2865066528320313e-05,
      "grad_norm": 0.1163238063454628,
      "learning_rate": 9.804272607917388e-06,
      "loss": 0.0663,
      "step": 13490
    },
    {
      "epoch": 1.2865066528320313e-05,
      "model_forward_time": 1.7310926914215088,
      "step": 13490
    },
    {
      "dynamics_loss_avg": 0.060470477864146235,
      "epoch": 1.2865066528320313e-05,
      "step": 13490
    },
    {
      "action_loss_avg": 0.0034036892699077725,
      "epoch": 1.2865066528320313e-05,
      "step": 13490
    },
    {
      "epoch": 1.2865066528320313e-05,
      "step": 13490,
      "training_step_time": 7.043415069580078
    },
    {
      "epoch": 1.2866020202636718e-05,
      "model_forward_time": 1.735323429107666,
      "step": 13491
    },
    {
      "epoch": 1.2866020202636718e-05,
      "step": 13491,
      "training_step_time": 7.732118844985962
    },
    {
      "epoch": 1.2866973876953126e-05,
      "model_forward_time": 1.7353174686431885,
      "step": 13492
    },
    {
      "epoch": 1.2866973876953126e-05,
      "step": 13492,
      "training_step_time": 7.452264785766602
    },
    {
      "epoch": 1.2867927551269532e-05,
      "model_forward_time": 1.7114558219909668,
      "step": 13493
    },
    {
      "epoch": 1.2867927551269532e-05,
      "step": 13493,
      "training_step_time": 7.807467699050903
    },
    {
      "epoch": 1.2868881225585937e-05,
      "model_forward_time": 1.7295522689819336,
      "step": 13494
    },
    {
      "epoch": 1.2868881225585937e-05,
      "step": 13494,
      "training_step_time": 8.477036237716675
    },
    {
      "epoch": 1.2869834899902343e-05,
      "model_forward_time": 1.7563903331756592,
      "step": 13495
    },
    {
      "epoch": 1.2869834899902343e-05,
      "step": 13495,
      "training_step_time": 9.742135286331177
    },
    {
      "epoch": 1.287078857421875e-05,
      "model_forward_time": 1.7767021656036377,
      "step": 13496
    },
    {
      "epoch": 1.287078857421875e-05,
      "step": 13496,
      "training_step_time": 8.923834562301636
    },
    {
      "epoch": 1.2871742248535156e-05,
      "model_forward_time": 1.7150499820709229,
      "step": 13497
    },
    {
      "epoch": 1.2871742248535156e-05,
      "step": 13497,
      "training_step_time": 8.893643379211426
    },
    {
      "epoch": 1.2872695922851562e-05,
      "model_forward_time": 1.7328674793243408,
      "step": 13498
    },
    {
      "epoch": 1.2872695922851562e-05,
      "step": 13498,
      "training_step_time": 7.609323740005493
    },
    {
      "epoch": 1.287364959716797e-05,
      "model_forward_time": 1.7337236404418945,
      "step": 13499
    },
    {
      "epoch": 1.287364959716797e-05,
      "step": 13499,
      "training_step_time": 7.323877811431885
    },
    {
      "epoch": 1.2874603271484375e-05,
      "grad_norm": 0.207765132188797,
      "learning_rate": 9.803814245791266e-06,
      "loss": 0.0687,
      "step": 13500
    },
    {
      "epoch": 1.2874603271484375e-05,
      "model_forward_time": 1.7244267463684082,
      "step": 13500
    },
    {
      "dynamics_loss_avg": 0.06544125564396382,
      "epoch": 1.2874603271484375e-05,
      "step": 13500
    },
    {
      "action_loss_avg": 0.002633394824806601,
      "epoch": 1.2874603271484375e-05,
      "step": 13500
    },
    {
      "epoch": 1.2874603271484375e-05,
      "step": 13500,
      "training_step_time": 7.741065502166748
    },
    {
      "epoch": 1.287555694580078e-05,
      "model_forward_time": 1.7229747772216797,
      "step": 13501
    },
    {
      "epoch": 1.287555694580078e-05,
      "step": 13501,
      "training_step_time": 8.544272184371948
    },
    {
      "epoch": 1.2876510620117188e-05,
      "model_forward_time": 1.724168062210083,
      "step": 13502
    },
    {
      "epoch": 1.2876510620117188e-05,
      "step": 13502,
      "training_step_time": 7.908043622970581
    },
    {
      "epoch": 1.2877464294433594e-05,
      "model_forward_time": 1.718658447265625,
      "step": 13503
    },
    {
      "epoch": 1.2877464294433594e-05,
      "step": 13503,
      "training_step_time": 7.926514625549316
    },
    {
      "epoch": 1.287841796875e-05,
      "model_forward_time": 1.7199287414550781,
      "step": 13504
    },
    {
      "epoch": 1.287841796875e-05,
      "step": 13504,
      "training_step_time": 8.655104160308838
    },
    {
      "epoch": 1.2879371643066407e-05,
      "model_forward_time": 1.7435410022735596,
      "step": 13505
    },
    {
      "epoch": 1.2879371643066407e-05,
      "step": 13505,
      "training_step_time": 8.42021131515503
    },
    {
      "epoch": 1.2880325317382813e-05,
      "model_forward_time": 1.5310556888580322,
      "step": 13506
    },
    {
      "epoch": 1.2880325317382813e-05,
      "step": 13506,
      "training_step_time": 7.919213771820068
    },
    {
      "epoch": 1.2881278991699218e-05,
      "model_forward_time": 1.7177927494049072,
      "step": 13507
    },
    {
      "epoch": 1.2881278991699218e-05,
      "step": 13507,
      "training_step_time": 7.825238227844238
    },
    {
      "epoch": 1.2882232666015624e-05,
      "model_forward_time": 1.6972675323486328,
      "step": 13508
    },
    {
      "epoch": 1.2882232666015624e-05,
      "step": 13508,
      "training_step_time": 7.63478684425354
    },
    {
      "epoch": 1.2883186340332032e-05,
      "model_forward_time": 1.7249751091003418,
      "step": 13509
    },
    {
      "epoch": 1.2883186340332032e-05,
      "step": 13509,
      "training_step_time": 8.386699438095093
    },
    {
      "epoch": 1.2884140014648437e-05,
      "grad_norm": 0.28173333406448364,
      "learning_rate": 9.80335535832724e-06,
      "loss": 0.0691,
      "step": 13510
    },
    {
      "epoch": 1.2884140014648437e-05,
      "model_forward_time": 1.3541769981384277,
      "step": 13510
    },
    {
      "dynamics_loss_avg": 0.04840423781424761,
      "epoch": 1.2884140014648437e-05,
      "step": 13510
    },
    {
      "action_loss_avg": 0.00219540495891124,
      "epoch": 1.2884140014648437e-05,
      "step": 13510
    },
    {
      "epoch": 1.2884140014648437e-05,
      "step": 13510,
      "training_step_time": 7.630882978439331
    },
    {
      "epoch": 1.2885093688964843e-05,
      "model_forward_time": 1.7287991046905518,
      "step": 13511
    },
    {
      "epoch": 1.2885093688964843e-05,
      "step": 13511,
      "training_step_time": 7.79207181930542
    },
    {
      "epoch": 1.288604736328125e-05,
      "model_forward_time": 1.7290117740631104,
      "step": 13512
    },
    {
      "epoch": 1.288604736328125e-05,
      "step": 13512,
      "training_step_time": 8.496693849563599
    },
    {
      "epoch": 1.2887001037597656e-05,
      "model_forward_time": 1.7240681648254395,
      "step": 13513
    },
    {
      "epoch": 1.2887001037597656e-05,
      "step": 13513,
      "training_step_time": 9.499368667602539
    },
    {
      "epoch": 1.2887954711914062e-05,
      "model_forward_time": 1.7247633934020996,
      "step": 13514
    },
    {
      "epoch": 1.2887954711914062e-05,
      "step": 13514,
      "training_step_time": 8.762542009353638
    },
    {
      "epoch": 1.288890838623047e-05,
      "model_forward_time": 2.2603561878204346,
      "step": 13515
    },
    {
      "epoch": 1.288890838623047e-05,
      "step": 13515,
      "training_step_time": 7.882568597793579
    },
    {
      "epoch": 1.2889862060546875e-05,
      "model_forward_time": 1.0249392986297607,
      "step": 13516
    },
    {
      "epoch": 1.2889862060546875e-05,
      "step": 13516,
      "training_step_time": 7.7852089405059814
    },
    {
      "epoch": 1.289081573486328e-05,
      "model_forward_time": 1.7038366794586182,
      "step": 13517
    },
    {
      "epoch": 1.289081573486328e-05,
      "step": 13517,
      "training_step_time": 7.629326581954956
    },
    {
      "epoch": 1.2891769409179688e-05,
      "model_forward_time": 1.7055490016937256,
      "step": 13518
    },
    {
      "epoch": 1.2891769409179688e-05,
      "step": 13518,
      "training_step_time": 8.484561204910278
    },
    {
      "epoch": 1.2892723083496094e-05,
      "model_forward_time": 1.7107298374176025,
      "step": 13519
    },
    {
      "epoch": 1.2892723083496094e-05,
      "step": 13519,
      "training_step_time": 7.776025295257568
    },
    {
      "epoch": 1.28936767578125e-05,
      "grad_norm": 0.2814994752407074,
      "learning_rate": 9.802895945575492e-06,
      "loss": 0.0706,
      "step": 13520
    },
    {
      "epoch": 1.28936767578125e-05,
      "model_forward_time": 1.7107086181640625,
      "step": 13520
    },
    {
      "dynamics_loss_avg": 0.055790405906736854,
      "epoch": 1.28936767578125e-05,
      "step": 13520
    },
    {
      "action_loss_avg": 0.003023237001616508,
      "epoch": 1.28936767578125e-05,
      "step": 13520
    },
    {
      "epoch": 1.28936767578125e-05,
      "step": 13520,
      "training_step_time": 7.82983660697937
    },
    {
      "epoch": 1.2894630432128907e-05,
      "model_forward_time": 1.3310480117797852,
      "step": 13521
    },
    {
      "epoch": 1.2894630432128907e-05,
      "step": 13521,
      "training_step_time": 8.573022842407227
    },
    {
      "epoch": 1.2895584106445313e-05,
      "model_forward_time": 1.7011964321136475,
      "step": 13522
    },
    {
      "epoch": 1.2895584106445313e-05,
      "step": 13522,
      "training_step_time": 7.907106637954712
    },
    {
      "epoch": 1.2896537780761718e-05,
      "model_forward_time": 1.3348541259765625,
      "step": 13523
    },
    {
      "epoch": 1.2896537780761718e-05,
      "step": 13523,
      "training_step_time": 8.78694486618042
    },
    {
      "epoch": 1.2897491455078124e-05,
      "model_forward_time": 1.7314023971557617,
      "step": 13524
    },
    {
      "epoch": 1.2897491455078124e-05,
      "step": 13524,
      "training_step_time": 7.968157052993774
    },
    {
      "epoch": 1.2898445129394532e-05,
      "model_forward_time": 2.2473862171173096,
      "step": 13525
    },
    {
      "epoch": 1.2898445129394532e-05,
      "step": 13525,
      "training_step_time": 7.873598098754883
    },
    {
      "epoch": 1.2899398803710937e-05,
      "model_forward_time": 1.7306628227233887,
      "step": 13526
    },
    {
      "epoch": 1.2899398803710937e-05,
      "step": 13526,
      "training_step_time": 7.721208333969116
    },
    {
      "epoch": 1.2900352478027343e-05,
      "model_forward_time": 1.7251029014587402,
      "step": 13527
    },
    {
      "epoch": 1.2900352478027343e-05,
      "step": 13527,
      "training_step_time": 7.647622585296631
    },
    {
      "epoch": 1.290130615234375e-05,
      "model_forward_time": 1.7258937358856201,
      "step": 13528
    },
    {
      "epoch": 1.290130615234375e-05,
      "step": 13528,
      "training_step_time": 8.592090129852295
    },
    {
      "epoch": 1.2902259826660156e-05,
      "model_forward_time": 1.7963440418243408,
      "step": 13529
    },
    {
      "epoch": 1.2902259826660156e-05,
      "step": 13529,
      "training_step_time": 7.855764865875244
    },
    {
      "epoch": 1.2903213500976562e-05,
      "grad_norm": 0.2223832756280899,
      "learning_rate": 9.802436007586267e-06,
      "loss": 0.0693,
      "step": 13530
    },
    {
      "epoch": 1.2903213500976562e-05,
      "model_forward_time": 1.7302753925323486,
      "step": 13530
    },
    {
      "dynamics_loss_avg": 0.06034652702510357,
      "epoch": 1.2903213500976562e-05,
      "step": 13530
    },
    {
      "action_loss_avg": 0.0027160826488398016,
      "epoch": 1.2903213500976562e-05,
      "step": 13530
    },
    {
      "epoch": 1.2903213500976562e-05,
      "step": 13530,
      "training_step_time": 7.966366291046143
    },
    {
      "epoch": 1.290416717529297e-05,
      "model_forward_time": 1.720560073852539,
      "step": 13531
    },
    {
      "epoch": 1.290416717529297e-05,
      "step": 13531,
      "training_step_time": 8.925818681716919
    },
    {
      "epoch": 1.2905120849609375e-05,
      "model_forward_time": 1.7266571521759033,
      "step": 13532
    },
    {
      "epoch": 1.2905120849609375e-05,
      "step": 13532,
      "training_step_time": 7.9585206508636475
    },
    {
      "epoch": 1.290607452392578e-05,
      "model_forward_time": 1.34881591796875,
      "step": 13533
    },
    {
      "epoch": 1.290607452392578e-05,
      "step": 13533,
      "training_step_time": 8.494283437728882
    },
    {
      "epoch": 1.2907028198242188e-05,
      "model_forward_time": 2.2820844650268555,
      "step": 13534
    },
    {
      "epoch": 1.2907028198242188e-05,
      "step": 13534,
      "training_step_time": 8.516608238220215
    },
    {
      "epoch": 1.2907981872558594e-05,
      "model_forward_time": 1.7248497009277344,
      "step": 13535
    },
    {
      "epoch": 1.2907981872558594e-05,
      "step": 13535,
      "training_step_time": 7.836580991744995
    },
    {
      "epoch": 1.2908935546875e-05,
      "model_forward_time": 1.7241714000701904,
      "step": 13536
    },
    {
      "epoch": 1.2908935546875e-05,
      "step": 13536,
      "training_step_time": 7.725625276565552
    },
    {
      "epoch": 1.2909889221191407e-05,
      "model_forward_time": 1.7293691635131836,
      "step": 13537
    },
    {
      "epoch": 1.2909889221191407e-05,
      "step": 13537,
      "training_step_time": 7.730949878692627
    },
    {
      "epoch": 1.2910842895507813e-05,
      "model_forward_time": 1.7850170135498047,
      "step": 13538
    },
    {
      "epoch": 1.2910842895507813e-05,
      "step": 13538,
      "training_step_time": 7.863480567932129
    },
    {
      "epoch": 1.2911796569824218e-05,
      "model_forward_time": 1.7299880981445312,
      "step": 13539
    },
    {
      "epoch": 1.2911796569824218e-05,
      "step": 13539,
      "training_step_time": 7.905459642410278
    },
    {
      "epoch": 1.2912750244140624e-05,
      "grad_norm": 0.22663737833499908,
      "learning_rate": 9.801975544409858e-06,
      "loss": 0.0707,
      "step": 13540
    },
    {
      "epoch": 1.2912750244140624e-05,
      "model_forward_time": 1.7196991443634033,
      "step": 13540
    },
    {
      "dynamics_loss_avg": 0.052954509016126396,
      "epoch": 1.2912750244140624e-05,
      "step": 13540
    },
    {
      "action_loss_avg": 0.0024561520520364867,
      "epoch": 1.2912750244140624e-05,
      "step": 13540
    },
    {
      "epoch": 1.2912750244140624e-05,
      "step": 13540,
      "training_step_time": 9.018262147903442
    },
    {
      "epoch": 1.2913703918457032e-05,
      "model_forward_time": 1.7256388664245605,
      "step": 13541
    },
    {
      "epoch": 1.2913703918457032e-05,
      "step": 13541,
      "training_step_time": 8.764458894729614
    },
    {
      "epoch": 1.2914657592773437e-05,
      "model_forward_time": 1.7241230010986328,
      "step": 13542
    },
    {
      "epoch": 1.2914657592773437e-05,
      "step": 13542,
      "training_step_time": 7.808953046798706
    },
    {
      "epoch": 1.2915611267089843e-05,
      "model_forward_time": 1.7239112854003906,
      "step": 13543
    },
    {
      "epoch": 1.2915611267089843e-05,
      "step": 13543,
      "training_step_time": 8.833107948303223
    },
    {
      "epoch": 1.291656494140625e-05,
      "model_forward_time": 1.726811408996582,
      "step": 13544
    },
    {
      "epoch": 1.291656494140625e-05,
      "step": 13544,
      "training_step_time": 8.31402063369751
    },
    {
      "epoch": 1.2917518615722656e-05,
      "model_forward_time": 1.7236642837524414,
      "step": 13545
    },
    {
      "epoch": 1.2917518615722656e-05,
      "step": 13545,
      "training_step_time": 7.508051156997681
    },
    {
      "epoch": 1.2918472290039062e-05,
      "model_forward_time": 1.739955186843872,
      "step": 13546
    },
    {
      "epoch": 1.2918472290039062e-05,
      "step": 13546,
      "training_step_time": 7.608612298965454
    },
    {
      "epoch": 1.291942596435547e-05,
      "model_forward_time": 1.772038221359253,
      "step": 13547
    },
    {
      "epoch": 1.291942596435547e-05,
      "step": 13547,
      "training_step_time": 7.716075658798218
    },
    {
      "epoch": 1.2920379638671875e-05,
      "model_forward_time": 1.7269344329833984,
      "step": 13548
    },
    {
      "epoch": 1.2920379638671875e-05,
      "step": 13548,
      "training_step_time": 7.763657331466675
    },
    {
      "epoch": 1.292133331298828e-05,
      "model_forward_time": 1.7270228862762451,
      "step": 13549
    },
    {
      "epoch": 1.292133331298828e-05,
      "step": 13549,
      "training_step_time": 9.542328119277954
    },
    {
      "epoch": 1.2922286987304688e-05,
      "grad_norm": 0.1786627620458603,
      "learning_rate": 9.801514556096624e-06,
      "loss": 0.068,
      "step": 13550
    },
    {
      "epoch": 1.2922286987304688e-05,
      "model_forward_time": 1.7260193824768066,
      "step": 13550
    },
    {
      "dynamics_loss_avg": 0.06574930772185325,
      "epoch": 1.2922286987304688e-05,
      "step": 13550
    },
    {
      "action_loss_avg": 0.002590597124071792,
      "epoch": 1.2922286987304688e-05,
      "step": 13550
    },
    {
      "epoch": 1.2922286987304688e-05,
      "step": 13550,
      "training_step_time": 9.598928451538086
    },
    {
      "epoch": 1.2923240661621094e-05,
      "model_forward_time": 1.7263422012329102,
      "step": 13551
    },
    {
      "epoch": 1.2923240661621094e-05,
      "step": 13551,
      "training_step_time": 8.95020580291748
    },
    {
      "epoch": 1.29241943359375e-05,
      "model_forward_time": 1.7289247512817383,
      "step": 13552
    },
    {
      "epoch": 1.29241943359375e-05,
      "step": 13552,
      "training_step_time": 7.846558570861816
    },
    {
      "epoch": 1.2925148010253907e-05,
      "model_forward_time": 2.2779901027679443,
      "step": 13553
    },
    {
      "epoch": 1.2925148010253907e-05,
      "step": 13553,
      "training_step_time": 7.797163248062134
    },
    {
      "epoch": 1.2926101684570313e-05,
      "model_forward_time": 1.7237544059753418,
      "step": 13554
    },
    {
      "epoch": 1.2926101684570313e-05,
      "step": 13554,
      "training_step_time": 7.873032808303833
    },
    {
      "epoch": 1.2927055358886718e-05,
      "model_forward_time": 1.7212882041931152,
      "step": 13555
    },
    {
      "epoch": 1.2927055358886718e-05,
      "step": 13555,
      "training_step_time": 8.286202430725098
    },
    {
      "epoch": 1.2928009033203124e-05,
      "model_forward_time": 1.740462303161621,
      "step": 13556
    },
    {
      "epoch": 1.2928009033203124e-05,
      "step": 13556,
      "training_step_time": 7.821219205856323
    },
    {
      "epoch": 1.2928962707519532e-05,
      "model_forward_time": 1.728604793548584,
      "step": 13557
    },
    {
      "epoch": 1.2928962707519532e-05,
      "step": 13557,
      "training_step_time": 7.774372816085815
    },
    {
      "epoch": 1.2929916381835937e-05,
      "model_forward_time": 1.7353594303131104,
      "step": 13558
    },
    {
      "epoch": 1.2929916381835937e-05,
      "step": 13558,
      "training_step_time": 7.900202035903931
    },
    {
      "epoch": 1.2930870056152343e-05,
      "model_forward_time": 1.7245306968688965,
      "step": 13559
    },
    {
      "epoch": 1.2930870056152343e-05,
      "step": 13559,
      "training_step_time": 9.754438638687134
    },
    {
      "epoch": 1.293182373046875e-05,
      "grad_norm": 0.14770950376987457,
      "learning_rate": 9.801053042696977e-06,
      "loss": 0.0696,
      "step": 13560
    },
    {
      "epoch": 1.293182373046875e-05,
      "model_forward_time": 1.7292799949645996,
      "step": 13560
    },
    {
      "dynamics_loss_avg": 0.06751146726310253,
      "epoch": 1.293182373046875e-05,
      "step": 13560
    },
    {
      "action_loss_avg": 0.003304747160291299,
      "epoch": 1.293182373046875e-05,
      "step": 13560
    },
    {
      "epoch": 1.293182373046875e-05,
      "step": 13560,
      "training_step_time": 7.9628496170043945
    },
    {
      "epoch": 1.2932777404785156e-05,
      "model_forward_time": 1.7277288436889648,
      "step": 13561
    },
    {
      "epoch": 1.2932777404785156e-05,
      "step": 13561,
      "training_step_time": 9.451382637023926
    },
    {
      "epoch": 1.2933731079101562e-05,
      "model_forward_time": 2.274473190307617,
      "step": 13562
    },
    {
      "epoch": 1.2933731079101562e-05,
      "step": 13562,
      "training_step_time": 7.8932859897613525
    },
    {
      "epoch": 1.293468475341797e-05,
      "model_forward_time": 1.7308695316314697,
      "step": 13563
    },
    {
      "epoch": 1.293468475341797e-05,
      "step": 13563,
      "training_step_time": 7.831190824508667
    },
    {
      "epoch": 1.2935638427734375e-05,
      "model_forward_time": 1.729856252670288,
      "step": 13564
    },
    {
      "epoch": 1.2935638427734375e-05,
      "step": 13564,
      "training_step_time": 7.613830327987671
    },
    {
      "epoch": 1.293659210205078e-05,
      "model_forward_time": 1.7291004657745361,
      "step": 13565
    },
    {
      "epoch": 1.293659210205078e-05,
      "step": 13565,
      "training_step_time": 8.140655040740967
    },
    {
      "epoch": 1.2937545776367188e-05,
      "model_forward_time": 1.724517822265625,
      "step": 13566
    },
    {
      "epoch": 1.2937545776367188e-05,
      "step": 13566,
      "training_step_time": 9.90935492515564
    },
    {
      "epoch": 1.2938499450683594e-05,
      "model_forward_time": 1.7367022037506104,
      "step": 13567
    },
    {
      "epoch": 1.2938499450683594e-05,
      "step": 13567,
      "training_step_time": 8.749149084091187
    },
    {
      "epoch": 1.2939453125e-05,
      "model_forward_time": 1.7241275310516357,
      "step": 13568
    },
    {
      "epoch": 1.2939453125e-05,
      "step": 13568,
      "training_step_time": 9.890066146850586
    },
    {
      "epoch": 1.2940406799316407e-05,
      "model_forward_time": 1.7262165546417236,
      "step": 13569
    },
    {
      "epoch": 1.2940406799316407e-05,
      "step": 13569,
      "training_step_time": 7.832997798919678
    },
    {
      "epoch": 1.2941360473632813e-05,
      "grad_norm": 0.174541175365448,
      "learning_rate": 9.800591004261388e-06,
      "loss": 0.0678,
      "step": 13570
    },
    {
      "epoch": 1.2941360473632813e-05,
      "model_forward_time": 1.7278542518615723,
      "step": 13570
    },
    {
      "dynamics_loss_avg": 0.07004662323743105,
      "epoch": 1.2941360473632813e-05,
      "step": 13570
    },
    {
      "action_loss_avg": 0.0035666691313963384,
      "epoch": 1.2941360473632813e-05,
      "step": 13570
    },
    {
      "epoch": 1.2941360473632813e-05,
      "step": 13570,
      "training_step_time": 8.731466054916382
    },
    {
      "epoch": 1.2942314147949218e-05,
      "model_forward_time": 2.2421860694885254,
      "step": 13571
    },
    {
      "epoch": 1.2942314147949218e-05,
      "step": 13571,
      "training_step_time": 7.909715175628662
    },
    {
      "epoch": 1.2943267822265624e-05,
      "model_forward_time": 1.7284157276153564,
      "step": 13572
    },
    {
      "epoch": 1.2943267822265624e-05,
      "step": 13572,
      "training_step_time": 8.578115940093994
    },
    {
      "epoch": 1.2944221496582032e-05,
      "model_forward_time": 1.730907678604126,
      "step": 13573
    },
    {
      "epoch": 1.2944221496582032e-05,
      "step": 13573,
      "training_step_time": 7.560935020446777
    },
    {
      "epoch": 1.2945175170898437e-05,
      "model_forward_time": 1.726050615310669,
      "step": 13574
    },
    {
      "epoch": 1.2945175170898437e-05,
      "step": 13574,
      "training_step_time": 7.771897077560425
    },
    {
      "epoch": 1.2946128845214843e-05,
      "model_forward_time": 1.7962334156036377,
      "step": 13575
    },
    {
      "epoch": 1.2946128845214843e-05,
      "step": 13575,
      "training_step_time": 9.400172710418701
    },
    {
      "epoch": 1.294708251953125e-05,
      "model_forward_time": 1.7406954765319824,
      "step": 13576
    },
    {
      "epoch": 1.294708251953125e-05,
      "step": 13576,
      "training_step_time": 8.861571073532104
    },
    {
      "epoch": 1.2948036193847656e-05,
      "model_forward_time": 1.7653403282165527,
      "step": 13577
    },
    {
      "epoch": 1.2948036193847656e-05,
      "step": 13577,
      "training_step_time": 8.017943859100342
    },
    {
      "epoch": 1.2948989868164062e-05,
      "model_forward_time": 1.738781213760376,
      "step": 13578
    },
    {
      "epoch": 1.2948989868164062e-05,
      "step": 13578,
      "training_step_time": 8.644914150238037
    },
    {
      "epoch": 1.294994354248047e-05,
      "model_forward_time": 1.7357163429260254,
      "step": 13579
    },
    {
      "epoch": 1.294994354248047e-05,
      "step": 13579,
      "training_step_time": 7.821571111679077
    },
    {
      "epoch": 1.2950897216796875e-05,
      "grad_norm": 0.26633980870246887,
      "learning_rate": 9.800128440840386e-06,
      "loss": 0.0715,
      "step": 13580
    },
    {
      "epoch": 1.2950897216796875e-05,
      "model_forward_time": 2.276775598526001,
      "step": 13580
    },
    {
      "dynamics_loss_avg": 0.05521320290863514,
      "epoch": 1.2950897216796875e-05,
      "step": 13580
    },
    {
      "action_loss_avg": 0.002241389185655862,
      "epoch": 1.2950897216796875e-05,
      "step": 13580
    },
    {
      "epoch": 1.2950897216796875e-05,
      "step": 13580,
      "training_step_time": 8.901121377944946
    },
    {
      "epoch": 1.295185089111328e-05,
      "model_forward_time": 1.7484099864959717,
      "step": 13581
    },
    {
      "epoch": 1.295185089111328e-05,
      "step": 13581,
      "training_step_time": 8.644394874572754
    },
    {
      "epoch": 1.2952804565429688e-05,
      "model_forward_time": 1.7454602718353271,
      "step": 13582
    },
    {
      "epoch": 1.2952804565429688e-05,
      "step": 13582,
      "training_step_time": 7.627936840057373
    },
    {
      "epoch": 1.2953758239746094e-05,
      "model_forward_time": 1.733313798904419,
      "step": 13583
    },
    {
      "epoch": 1.2953758239746094e-05,
      "step": 13583,
      "training_step_time": 7.719382286071777
    },
    {
      "epoch": 1.29547119140625e-05,
      "model_forward_time": 1.781214714050293,
      "step": 13584
    },
    {
      "epoch": 1.29547119140625e-05,
      "step": 13584,
      "training_step_time": 8.545250654220581
    },
    {
      "epoch": 1.2955665588378907e-05,
      "model_forward_time": 1.7437443733215332,
      "step": 13585
    },
    {
      "epoch": 1.2955665588378907e-05,
      "step": 13585,
      "training_step_time": 7.80401873588562
    },
    {
      "epoch": 1.2956619262695313e-05,
      "model_forward_time": 1.741002082824707,
      "step": 13586
    },
    {
      "epoch": 1.2956619262695313e-05,
      "step": 13586,
      "training_step_time": 8.750691652297974
    },
    {
      "epoch": 1.2957572937011719e-05,
      "model_forward_time": 1.7426173686981201,
      "step": 13587
    },
    {
      "epoch": 1.2957572937011719e-05,
      "step": 13587,
      "training_step_time": 7.9766845703125
    },
    {
      "epoch": 1.2958526611328124e-05,
      "model_forward_time": 1.740696668624878,
      "step": 13588
    },
    {
      "epoch": 1.2958526611328124e-05,
      "step": 13588,
      "training_step_time": 7.966600179672241
    },
    {
      "epoch": 1.2959480285644532e-05,
      "model_forward_time": 2.2946462631225586,
      "step": 13589
    },
    {
      "epoch": 1.2959480285644532e-05,
      "step": 13589,
      "training_step_time": 7.881806373596191
    },
    {
      "epoch": 1.2960433959960937e-05,
      "grad_norm": 0.17607910931110382,
      "learning_rate": 9.799665352484551e-06,
      "loss": 0.0674,
      "step": 13590
    },
    {
      "epoch": 1.2960433959960937e-05,
      "model_forward_time": 1.7372992038726807,
      "step": 13590
    },
    {
      "dynamics_loss_avg": 0.054684427380561826,
      "epoch": 1.2960433959960937e-05,
      "step": 13590
    },
    {
      "action_loss_avg": 0.0026701205759309234,
      "epoch": 1.2960433959960937e-05,
      "step": 13590
    },
    {
      "epoch": 1.2960433959960937e-05,
      "step": 13590,
      "training_step_time": 7.919554948806763
    },
    {
      "epoch": 1.2961387634277343e-05,
      "model_forward_time": 1.7415401935577393,
      "step": 13591
    },
    {
      "epoch": 1.2961387634277343e-05,
      "step": 13591,
      "training_step_time": 7.789167404174805
    },
    {
      "epoch": 1.296234130859375e-05,
      "model_forward_time": 1.7574405670166016,
      "step": 13592
    },
    {
      "epoch": 1.296234130859375e-05,
      "step": 13592,
      "training_step_time": 8.461998224258423
    },
    {
      "epoch": 1.2963294982910156e-05,
      "model_forward_time": 1.7723405361175537,
      "step": 13593
    },
    {
      "epoch": 1.2963294982910156e-05,
      "step": 13593,
      "training_step_time": 8.726550579071045
    },
    {
      "epoch": 1.2964248657226562e-05,
      "model_forward_time": 1.741088628768921,
      "step": 13594
    },
    {
      "epoch": 1.2964248657226562e-05,
      "step": 13594,
      "training_step_time": 8.43527603149414
    },
    {
      "epoch": 1.296520233154297e-05,
      "model_forward_time": 1.742414951324463,
      "step": 13595
    },
    {
      "epoch": 1.296520233154297e-05,
      "step": 13595,
      "training_step_time": 8.870352745056152
    },
    {
      "epoch": 1.2966156005859375e-05,
      "model_forward_time": 1.7356696128845215,
      "step": 13596
    },
    {
      "epoch": 1.2966156005859375e-05,
      "step": 13596,
      "training_step_time": 8.783588886260986
    },
    {
      "epoch": 1.296710968017578e-05,
      "model_forward_time": 1.7306139469146729,
      "step": 13597
    },
    {
      "epoch": 1.296710968017578e-05,
      "step": 13597,
      "training_step_time": 9.816141128540039
    },
    {
      "epoch": 1.2968063354492188e-05,
      "model_forward_time": 1.7367770671844482,
      "step": 13598
    },
    {
      "epoch": 1.2968063354492188e-05,
      "step": 13598,
      "training_step_time": 8.891324043273926
    },
    {
      "epoch": 1.2969017028808594e-05,
      "model_forward_time": 2.2771215438842773,
      "step": 13599
    },
    {
      "epoch": 1.2969017028808594e-05,
      "step": 13599,
      "training_step_time": 8.761328220367432
    },
    {
      "epoch": 1.2969970703125e-05,
      "grad_norm": 0.23475107550621033,
      "learning_rate": 9.799201739244532e-06,
      "loss": 0.0653,
      "step": 13600
    },
    {
      "epoch": 1.2969970703125e-05,
      "model_forward_time": 1.7340424060821533,
      "step": 13600
    },
    {
      "dynamics_loss_avg": 0.06764742583036423,
      "epoch": 1.2969970703125e-05,
      "step": 13600
    },
    {
      "action_loss_avg": 0.003477733687032014,
      "epoch": 1.2969970703125e-05,
      "step": 13600
    },
    {
      "epoch": 1.2969970703125e-05,
      "step": 13600,
      "training_step_time": 7.79023814201355
    },
    {
      "epoch": 1.2970924377441407e-05,
      "model_forward_time": 1.7284026145935059,
      "step": 13601
    },
    {
      "epoch": 1.2970924377441407e-05,
      "step": 13601,
      "training_step_time": 7.6685402393341064
    },
    {
      "epoch": 1.2971878051757813e-05,
      "model_forward_time": 1.7408030033111572,
      "step": 13602
    },
    {
      "epoch": 1.2971878051757813e-05,
      "step": 13602,
      "training_step_time": 9.29223084449768
    },
    {
      "epoch": 1.2972831726074219e-05,
      "model_forward_time": 1.7382094860076904,
      "step": 13603
    },
    {
      "epoch": 1.2972831726074219e-05,
      "step": 13603,
      "training_step_time": 7.88185977935791
    },
    {
      "epoch": 1.2973785400390624e-05,
      "model_forward_time": 1.7352778911590576,
      "step": 13604
    },
    {
      "epoch": 1.2973785400390624e-05,
      "step": 13604,
      "training_step_time": 8.540562629699707
    },
    {
      "epoch": 1.2974739074707032e-05,
      "model_forward_time": 1.728273868560791,
      "step": 13605
    },
    {
      "epoch": 1.2974739074707032e-05,
      "step": 13605,
      "training_step_time": 8.636078834533691
    },
    {
      "epoch": 1.2975692749023437e-05,
      "model_forward_time": 1.7319977283477783,
      "step": 13606
    },
    {
      "epoch": 1.2975692749023437e-05,
      "step": 13606,
      "training_step_time": 7.848789215087891
    },
    {
      "epoch": 1.2976646423339843e-05,
      "model_forward_time": 1.7376823425292969,
      "step": 13607
    },
    {
      "epoch": 1.2976646423339843e-05,
      "step": 13607,
      "training_step_time": 7.899397850036621
    },
    {
      "epoch": 1.297760009765625e-05,
      "model_forward_time": 2.2728941440582275,
      "step": 13608
    },
    {
      "epoch": 1.297760009765625e-05,
      "step": 13608,
      "training_step_time": 8.445510149002075
    },
    {
      "epoch": 1.2978553771972656e-05,
      "model_forward_time": 1.7373416423797607,
      "step": 13609
    },
    {
      "epoch": 1.2978553771972656e-05,
      "step": 13609,
      "training_step_time": 7.842251539230347
    },
    {
      "epoch": 1.2979507446289062e-05,
      "grad_norm": 0.16088493168354034,
      "learning_rate": 9.798737601171025e-06,
      "loss": 0.0666,
      "step": 13610
    },
    {
      "epoch": 1.2979507446289062e-05,
      "model_forward_time": 1.7322018146514893,
      "step": 13610
    },
    {
      "dynamics_loss_avg": 0.05475547555834055,
      "epoch": 1.2979507446289062e-05,
      "step": 13610
    },
    {
      "action_loss_avg": 0.0020475843455642464,
      "epoch": 1.2979507446289062e-05,
      "step": 13610
    },
    {
      "epoch": 1.2979507446289062e-05,
      "step": 13610,
      "training_step_time": 7.510267734527588
    },
    {
      "epoch": 1.298046112060547e-05,
      "model_forward_time": 1.7324104309082031,
      "step": 13611
    },
    {
      "epoch": 1.298046112060547e-05,
      "step": 13611,
      "training_step_time": 7.699965000152588
    },
    {
      "epoch": 1.2981414794921875e-05,
      "model_forward_time": 1.8109393119812012,
      "step": 13612
    },
    {
      "epoch": 1.2981414794921875e-05,
      "step": 13612,
      "training_step_time": 7.848396062850952
    },
    {
      "epoch": 1.298236846923828e-05,
      "model_forward_time": 1.728389024734497,
      "step": 13613
    },
    {
      "epoch": 1.298236846923828e-05,
      "step": 13613,
      "training_step_time": 7.920235872268677
    },
    {
      "epoch": 1.2983322143554688e-05,
      "model_forward_time": 1.7295465469360352,
      "step": 13614
    },
    {
      "epoch": 1.2983322143554688e-05,
      "step": 13614,
      "training_step_time": 7.925402402877808
    },
    {
      "epoch": 1.2984275817871094e-05,
      "model_forward_time": 1.7416284084320068,
      "step": 13615
    },
    {
      "epoch": 1.2984275817871094e-05,
      "step": 13615,
      "training_step_time": 7.791385173797607
    },
    {
      "epoch": 1.29852294921875e-05,
      "model_forward_time": 1.7289984226226807,
      "step": 13616
    },
    {
      "epoch": 1.29852294921875e-05,
      "step": 13616,
      "training_step_time": 7.946898937225342
    },
    {
      "epoch": 1.2986183166503907e-05,
      "model_forward_time": 2.275358200073242,
      "step": 13617
    },
    {
      "epoch": 1.2986183166503907e-05,
      "step": 13617,
      "training_step_time": 8.71794605255127
    },
    {
      "epoch": 1.2987136840820313e-05,
      "model_forward_time": 1.7582578659057617,
      "step": 13618
    },
    {
      "epoch": 1.2987136840820313e-05,
      "step": 13618,
      "training_step_time": 7.824188232421875
    },
    {
      "epoch": 1.2988090515136719e-05,
      "model_forward_time": 1.033022165298462,
      "step": 13619
    },
    {
      "epoch": 1.2988090515136719e-05,
      "step": 13619,
      "training_step_time": 7.548720836639404
    },
    {
      "epoch": 1.2989044189453124e-05,
      "grad_norm": 0.17974436283111572,
      "learning_rate": 9.79827293831479e-06,
      "loss": 0.0623,
      "step": 13620
    },
    {
      "epoch": 1.2989044189453124e-05,
      "model_forward_time": 1.736363172531128,
      "step": 13620
    },
    {
      "dynamics_loss_avg": 0.0574656818062067,
      "epoch": 1.2989044189453124e-05,
      "step": 13620
    },
    {
      "action_loss_avg": 0.002804926235694438,
      "epoch": 1.2989044189453124e-05,
      "step": 13620
    },
    {
      "epoch": 1.2989044189453124e-05,
      "step": 13620,
      "training_step_time": 7.771697282791138
    },
    {
      "epoch": 1.2989997863769532e-05,
      "model_forward_time": 1.8064000606536865,
      "step": 13621
    },
    {
      "epoch": 1.2989997863769532e-05,
      "step": 13621,
      "training_step_time": 7.883642911911011
    },
    {
      "epoch": 1.2990951538085937e-05,
      "model_forward_time": 1.737123966217041,
      "step": 13622
    },
    {
      "epoch": 1.2990951538085937e-05,
      "step": 13622,
      "training_step_time": 8.859442949295044
    },
    {
      "epoch": 1.2991905212402343e-05,
      "model_forward_time": 1.747950792312622,
      "step": 13623
    },
    {
      "epoch": 1.2991905212402343e-05,
      "step": 13623,
      "training_step_time": 9.947113513946533
    },
    {
      "epoch": 1.299285888671875e-05,
      "model_forward_time": 1.745966911315918,
      "step": 13624
    },
    {
      "epoch": 1.299285888671875e-05,
      "step": 13624,
      "training_step_time": 7.842415809631348
    },
    {
      "epoch": 1.2993812561035156e-05,
      "model_forward_time": 1.741495132446289,
      "step": 13625
    },
    {
      "epoch": 1.2993812561035156e-05,
      "step": 13625,
      "training_step_time": 8.672499656677246
    },
    {
      "epoch": 1.2994766235351562e-05,
      "model_forward_time": 2.2588114738464355,
      "step": 13626
    },
    {
      "epoch": 1.2994766235351562e-05,
      "step": 13626,
      "training_step_time": 8.755630254745483
    },
    {
      "epoch": 1.299571990966797e-05,
      "model_forward_time": 1.7414164543151855,
      "step": 13627
    },
    {
      "epoch": 1.299571990966797e-05,
      "step": 13627,
      "training_step_time": 8.372183322906494
    },
    {
      "epoch": 1.2996673583984375e-05,
      "model_forward_time": 1.7459819316864014,
      "step": 13628
    },
    {
      "epoch": 1.2996673583984375e-05,
      "step": 13628,
      "training_step_time": 7.534530162811279
    },
    {
      "epoch": 1.299762725830078e-05,
      "model_forward_time": 1.741346836090088,
      "step": 13629
    },
    {
      "epoch": 1.299762725830078e-05,
      "step": 13629,
      "training_step_time": 7.79422664642334
    },
    {
      "epoch": 1.2998580932617188e-05,
      "grad_norm": 0.20409242808818817,
      "learning_rate": 9.797807750726638e-06,
      "loss": 0.065,
      "step": 13630
    },
    {
      "epoch": 1.2998580932617188e-05,
      "model_forward_time": 1.7806966304779053,
      "step": 13630
    },
    {
      "dynamics_loss_avg": 0.05867674667388201,
      "epoch": 1.2998580932617188e-05,
      "step": 13630
    },
    {
      "action_loss_avg": 0.004222866811323911,
      "epoch": 1.2998580932617188e-05,
      "step": 13630
    },
    {
      "epoch": 1.2998580932617188e-05,
      "step": 13630,
      "training_step_time": 7.8179919719696045
    },
    {
      "epoch": 1.2999534606933594e-05,
      "model_forward_time": 1.7417042255401611,
      "step": 13631
    },
    {
      "epoch": 1.2999534606933594e-05,
      "step": 13631,
      "training_step_time": 7.81821870803833
    },
    {
      "epoch": 1.300048828125e-05,
      "model_forward_time": 1.7431278228759766,
      "step": 13632
    },
    {
      "epoch": 1.300048828125e-05,
      "step": 13632,
      "training_step_time": 9.786428213119507
    },
    {
      "epoch": 1.3001441955566405e-05,
      "model_forward_time": 1.7421631813049316,
      "step": 13633
    },
    {
      "epoch": 1.3001441955566405e-05,
      "step": 13633,
      "training_step_time": 8.8690345287323
    },
    {
      "epoch": 1.3002395629882813e-05,
      "model_forward_time": 1.743196725845337,
      "step": 13634
    },
    {
      "epoch": 1.3002395629882813e-05,
      "step": 13634,
      "training_step_time": 8.835143804550171
    },
    {
      "epoch": 1.3003349304199219e-05,
      "model_forward_time": 1.7406272888183594,
      "step": 13635
    },
    {
      "epoch": 1.3003349304199219e-05,
      "step": 13635,
      "training_step_time": 7.971235275268555
    },
    {
      "epoch": 1.3004302978515624e-05,
      "model_forward_time": 2.2853293418884277,
      "step": 13636
    },
    {
      "epoch": 1.3004302978515624e-05,
      "step": 13636,
      "training_step_time": 8.409761190414429
    },
    {
      "epoch": 1.3005256652832032e-05,
      "model_forward_time": 1.7434515953063965,
      "step": 13637
    },
    {
      "epoch": 1.3005256652832032e-05,
      "step": 13637,
      "training_step_time": 8.278357982635498
    },
    {
      "epoch": 1.3006210327148437e-05,
      "model_forward_time": 1.7222495079040527,
      "step": 13638
    },
    {
      "epoch": 1.3006210327148437e-05,
      "step": 13638,
      "training_step_time": 7.792694807052612
    },
    {
      "epoch": 1.3007164001464843e-05,
      "model_forward_time": 1.7597198486328125,
      "step": 13639
    },
    {
      "epoch": 1.3007164001464843e-05,
      "step": 13639,
      "training_step_time": 7.868535757064819
    },
    {
      "epoch": 1.300811767578125e-05,
      "grad_norm": 0.23299795389175415,
      "learning_rate": 9.797342038457447e-06,
      "loss": 0.0615,
      "step": 13640
    },
    {
      "epoch": 1.300811767578125e-05,
      "model_forward_time": 1.724945306777954,
      "step": 13640
    },
    {
      "dynamics_loss_avg": 0.06293706465512514,
      "epoch": 1.300811767578125e-05,
      "step": 13640
    },
    {
      "action_loss_avg": 0.004727908753557131,
      "epoch": 1.300811767578125e-05,
      "step": 13640
    },
    {
      "epoch": 1.300811767578125e-05,
      "step": 13640,
      "training_step_time": 7.7999677658081055
    },
    {
      "epoch": 1.3009071350097656e-05,
      "model_forward_time": 1.7318122386932373,
      "step": 13641
    },
    {
      "epoch": 1.3009071350097656e-05,
      "step": 13641,
      "training_step_time": 10.048439741134644
    },
    {
      "epoch": 1.3010025024414062e-05,
      "model_forward_time": 1.7290003299713135,
      "step": 13642
    },
    {
      "epoch": 1.3010025024414062e-05,
      "step": 13642,
      "training_step_time": 8.797282934188843
    },
    {
      "epoch": 1.301097869873047e-05,
      "model_forward_time": 1.7368836402893066,
      "step": 13643
    },
    {
      "epoch": 1.301097869873047e-05,
      "step": 13643,
      "training_step_time": 6.941248178482056
    },
    {
      "epoch": 1.3011932373046875e-05,
      "model_forward_time": 1.7326529026031494,
      "step": 13644
    },
    {
      "epoch": 1.3011932373046875e-05,
      "step": 13644,
      "training_step_time": 6.873350143432617
    },
    {
      "epoch": 1.301288604736328e-05,
      "model_forward_time": 1.73274827003479,
      "step": 13645
    },
    {
      "epoch": 1.301288604736328e-05,
      "step": 13645,
      "training_step_time": 6.934541463851929
    },
    {
      "epoch": 1.3013839721679688e-05,
      "model_forward_time": 1.7293224334716797,
      "step": 13646
    },
    {
      "epoch": 1.3013839721679688e-05,
      "step": 13646,
      "training_step_time": 7.005128622055054
    },
    {
      "epoch": 1.3014793395996094e-05,
      "model_forward_time": 1.7351000308990479,
      "step": 13647
    },
    {
      "epoch": 1.3014793395996094e-05,
      "step": 13647,
      "training_step_time": 7.042162179946899
    },
    {
      "epoch": 1.30157470703125e-05,
      "model_forward_time": 1.3634603023529053,
      "step": 13648
    },
    {
      "epoch": 1.30157470703125e-05,
      "step": 13648,
      "training_step_time": 6.9376044273376465
    },
    {
      "epoch": 1.3016700744628905e-05,
      "model_forward_time": 1.7268245220184326,
      "step": 13649
    },
    {
      "epoch": 1.3016700744628905e-05,
      "step": 13649,
      "training_step_time": 6.999129772186279
    },
    {
      "epoch": 1.3017654418945313e-05,
      "grad_norm": 0.16882647573947906,
      "learning_rate": 9.796875801558141e-06,
      "loss": 0.0658,
      "step": 13650
    },
    {
      "epoch": 1.3017654418945313e-05,
      "model_forward_time": 1.7269055843353271,
      "step": 13650
    },
    {
      "dynamics_loss_avg": 0.04633232764899731,
      "epoch": 1.3017654418945313e-05,
      "step": 13650
    },
    {
      "action_loss_avg": 0.0029517475981265307,
      "epoch": 1.3017654418945313e-05,
      "step": 13650
    },
    {
      "epoch": 1.3017654418945313e-05,
      "step": 13650,
      "training_step_time": 6.934664487838745
    },
    {
      "epoch": 1.3018608093261719e-05,
      "model_forward_time": 2.2547216415405273,
      "step": 13651
    },
    {
      "epoch": 1.3018608093261719e-05,
      "step": 13651,
      "training_step_time": 8.464550733566284
    },
    {
      "epoch": 1.3019561767578124e-05,
      "model_forward_time": 1.7323381900787354,
      "step": 13652
    },
    {
      "epoch": 1.3019561767578124e-05,
      "step": 13652,
      "training_step_time": 7.8266355991363525
    },
    {
      "epoch": 1.3020515441894532e-05,
      "model_forward_time": 1.7409286499023438,
      "step": 13653
    },
    {
      "epoch": 1.3020515441894532e-05,
      "step": 13653,
      "training_step_time": 7.00060248374939
    },
    {
      "epoch": 1.3021469116210937e-05,
      "model_forward_time": 1.7429561614990234,
      "step": 13654
    },
    {
      "epoch": 1.3021469116210937e-05,
      "step": 13654,
      "training_step_time": 6.965322732925415
    },
    {
      "epoch": 1.3022422790527343e-05,
      "model_forward_time": 1.7414891719818115,
      "step": 13655
    },
    {
      "epoch": 1.3022422790527343e-05,
      "step": 13655,
      "training_step_time": 6.949540853500366
    },
    {
      "epoch": 1.302337646484375e-05,
      "model_forward_time": 1.7455482482910156,
      "step": 13656
    },
    {
      "epoch": 1.302337646484375e-05,
      "step": 13656,
      "training_step_time": 7.007162809371948
    },
    {
      "epoch": 1.3024330139160156e-05,
      "model_forward_time": 1.7287685871124268,
      "step": 13657
    },
    {
      "epoch": 1.3024330139160156e-05,
      "step": 13657,
      "training_step_time": 6.982960939407349
    },
    {
      "epoch": 1.3025283813476562e-05,
      "model_forward_time": 1.7424602508544922,
      "step": 13658
    },
    {
      "epoch": 1.3025283813476562e-05,
      "step": 13658,
      "training_step_time": 6.96142053604126
    },
    {
      "epoch": 1.302623748779297e-05,
      "model_forward_time": 1.740828514099121,
      "step": 13659
    },
    {
      "epoch": 1.302623748779297e-05,
      "step": 13659,
      "training_step_time": 6.972861051559448
    },
    {
      "epoch": 1.3027191162109375e-05,
      "grad_norm": 0.18642506003379822,
      "learning_rate": 9.796409040079711e-06,
      "loss": 0.0657,
      "step": 13660
    },
    {
      "epoch": 1.3027191162109375e-05,
      "model_forward_time": 2.2906088829040527,
      "step": 13660
    },
    {
      "dynamics_loss_avg": 0.04803057238459587,
      "epoch": 1.3027191162109375e-05,
      "step": 13660
    },
    {
      "action_loss_avg": 0.0032107761944644153,
      "epoch": 1.3027191162109375e-05,
      "step": 13660
    },
    {
      "epoch": 1.3027191162109375e-05,
      "step": 13660,
      "training_step_time": 10.696762800216675
    },
    {
      "epoch": 1.3028144836425781e-05,
      "model_forward_time": 1.7476506233215332,
      "step": 13661
    },
    {
      "epoch": 1.3028144836425781e-05,
      "step": 13661,
      "training_step_time": 7.849786281585693
    },
    {
      "epoch": 1.3029098510742188e-05,
      "model_forward_time": 1.7423524856567383,
      "step": 13662
    },
    {
      "epoch": 1.3029098510742188e-05,
      "step": 13662,
      "training_step_time": 7.101981163024902
    },
    {
      "epoch": 1.3030052185058594e-05,
      "model_forward_time": 1.7246346473693848,
      "step": 13663
    },
    {
      "epoch": 1.3030052185058594e-05,
      "step": 13663,
      "training_step_time": 6.918139219284058
    },
    {
      "epoch": 1.3031005859375e-05,
      "model_forward_time": 1.7337770462036133,
      "step": 13664
    },
    {
      "epoch": 1.3031005859375e-05,
      "step": 13664,
      "training_step_time": 7.024230718612671
    },
    {
      "epoch": 1.3031959533691405e-05,
      "model_forward_time": 1.7349977493286133,
      "step": 13665
    },
    {
      "epoch": 1.3031959533691405e-05,
      "step": 13665,
      "training_step_time": 7.030441761016846
    },
    {
      "epoch": 1.3032913208007813e-05,
      "model_forward_time": 1.7362439632415771,
      "step": 13666
    },
    {
      "epoch": 1.3032913208007813e-05,
      "step": 13666,
      "training_step_time": 6.954526901245117
    },
    {
      "epoch": 1.3033866882324219e-05,
      "model_forward_time": 1.7393126487731934,
      "step": 13667
    },
    {
      "epoch": 1.3033866882324219e-05,
      "step": 13667,
      "training_step_time": 6.959595441818237
    },
    {
      "epoch": 1.3034820556640624e-05,
      "model_forward_time": 1.7268381118774414,
      "step": 13668
    },
    {
      "epoch": 1.3034820556640624e-05,
      "step": 13668,
      "training_step_time": 7.0357506275177
    },
    {
      "epoch": 1.3035774230957032e-05,
      "model_forward_time": 2.253804922103882,
      "step": 13669
    },
    {
      "epoch": 1.3035774230957032e-05,
      "step": 13669,
      "training_step_time": 10.882115364074707
    },
    {
      "epoch": 1.3036727905273437e-05,
      "grad_norm": 0.16713352501392365,
      "learning_rate": 9.795941754073198e-06,
      "loss": 0.072,
      "step": 13670
    },
    {
      "epoch": 1.3036727905273437e-05,
      "model_forward_time": 1.7273120880126953,
      "step": 13670
    },
    {
      "dynamics_loss_avg": 0.047099494375288486,
      "epoch": 1.3036727905273437e-05,
      "step": 13670
    },
    {
      "action_loss_avg": 0.0019048866059165448,
      "epoch": 1.3036727905273437e-05,
      "step": 13670
    },
    {
      "epoch": 1.3036727905273437e-05,
      "step": 13670,
      "training_step_time": 7.901794910430908
    },
    {
      "epoch": 1.3037681579589843e-05,
      "model_forward_time": 1.7345895767211914,
      "step": 13671
    },
    {
      "epoch": 1.3037681579589843e-05,
      "step": 13671,
      "training_step_time": 6.855510234832764
    },
    {
      "epoch": 1.303863525390625e-05,
      "model_forward_time": 1.7296926975250244,
      "step": 13672
    },
    {
      "epoch": 1.303863525390625e-05,
      "step": 13672,
      "training_step_time": 6.9022300243377686
    },
    {
      "epoch": 1.3039588928222656e-05,
      "model_forward_time": 1.7464163303375244,
      "step": 13673
    },
    {
      "epoch": 1.3039588928222656e-05,
      "step": 13673,
      "training_step_time": 6.878873586654663
    },
    {
      "epoch": 1.3040542602539062e-05,
      "model_forward_time": 1.7234859466552734,
      "step": 13674
    },
    {
      "epoch": 1.3040542602539062e-05,
      "step": 13674,
      "training_step_time": 6.851815462112427
    },
    {
      "epoch": 1.304149627685547e-05,
      "model_forward_time": 1.732771396636963,
      "step": 13675
    },
    {
      "epoch": 1.304149627685547e-05,
      "step": 13675,
      "training_step_time": 6.86336874961853
    },
    {
      "epoch": 1.3042449951171875e-05,
      "model_forward_time": 1.3593053817749023,
      "step": 13676
    },
    {
      "epoch": 1.3042449951171875e-05,
      "step": 13676,
      "training_step_time": 6.906602621078491
    },
    {
      "epoch": 1.3043403625488281e-05,
      "model_forward_time": 1.7311246395111084,
      "step": 13677
    },
    {
      "epoch": 1.3043403625488281e-05,
      "step": 13677,
      "training_step_time": 6.937977075576782
    },
    {
      "epoch": 1.3044357299804688e-05,
      "model_forward_time": 2.27022385597229,
      "step": 13678
    },
    {
      "epoch": 1.3044357299804688e-05,
      "step": 13678,
      "training_step_time": 10.969446182250977
    },
    {
      "epoch": 1.3045310974121094e-05,
      "model_forward_time": 1.733626127243042,
      "step": 13679
    },
    {
      "epoch": 1.3045310974121094e-05,
      "step": 13679,
      "training_step_time": 8.97510051727295
    },
    {
      "epoch": 1.30462646484375e-05,
      "grad_norm": 0.20101094245910645,
      "learning_rate": 9.795473943589707e-06,
      "loss": 0.064,
      "step": 13680
    },
    {
      "epoch": 1.30462646484375e-05,
      "model_forward_time": 1.740752935409546,
      "step": 13680
    },
    {
      "dynamics_loss_avg": 0.05792554412037134,
      "epoch": 1.30462646484375e-05,
      "step": 13680
    },
    {
      "action_loss_avg": 0.003279030753765255,
      "epoch": 1.30462646484375e-05,
      "step": 13680
    },
    {
      "epoch": 1.30462646484375e-05,
      "step": 13680,
      "training_step_time": 7.461853504180908
    },
    {
      "epoch": 1.3047218322753905e-05,
      "model_forward_time": 1.7366888523101807,
      "step": 13681
    },
    {
      "epoch": 1.3047218322753905e-05,
      "step": 13681,
      "training_step_time": 6.8852221965789795
    },
    {
      "epoch": 1.3048171997070313e-05,
      "model_forward_time": 1.734647274017334,
      "step": 13682
    },
    {
      "epoch": 1.3048171997070313e-05,
      "step": 13682,
      "training_step_time": 6.843642711639404
    },
    {
      "epoch": 1.3049125671386719e-05,
      "model_forward_time": 1.7351360321044922,
      "step": 13683
    },
    {
      "epoch": 1.3049125671386719e-05,
      "step": 13683,
      "training_step_time": 6.843590021133423
    },
    {
      "epoch": 1.3050079345703124e-05,
      "model_forward_time": 1.7047953605651855,
      "step": 13684
    },
    {
      "epoch": 1.3050079345703124e-05,
      "step": 13684,
      "training_step_time": 6.760747671127319
    },
    {
      "epoch": 1.3051033020019532e-05,
      "model_forward_time": 1.7369983196258545,
      "step": 13685
    },
    {
      "epoch": 1.3051033020019532e-05,
      "step": 13685,
      "training_step_time": 6.882267475128174
    },
    {
      "epoch": 1.3051986694335937e-05,
      "model_forward_time": 1.7334485054016113,
      "step": 13686
    },
    {
      "epoch": 1.3051986694335937e-05,
      "step": 13686,
      "training_step_time": 6.914617300033569
    },
    {
      "epoch": 1.3052940368652343e-05,
      "model_forward_time": 1.731529712677002,
      "step": 13687
    },
    {
      "epoch": 1.3052940368652343e-05,
      "step": 13687,
      "training_step_time": 8.340584754943848
    },
    {
      "epoch": 1.305389404296875e-05,
      "model_forward_time": 2.3236308097839355,
      "step": 13688
    },
    {
      "epoch": 1.305389404296875e-05,
      "step": 13688,
      "training_step_time": 9.69789171218872
    },
    {
      "epoch": 1.3054847717285156e-05,
      "model_forward_time": 1.0316877365112305,
      "step": 13689
    },
    {
      "epoch": 1.3054847717285156e-05,
      "step": 13689,
      "training_step_time": 7.651479482650757
    },
    {
      "epoch": 1.3055801391601562e-05,
      "grad_norm": 0.1917700469493866,
      "learning_rate": 9.795005608680394e-06,
      "loss": 0.0655,
      "step": 13690
    },
    {
      "epoch": 1.3055801391601562e-05,
      "model_forward_time": 1.730773687362671,
      "step": 13690
    },
    {
      "dynamics_loss_avg": 0.04709868291392923,
      "epoch": 1.3055801391601562e-05,
      "step": 13690
    },
    {
      "action_loss_avg": 0.0030666681355796753,
      "epoch": 1.3055801391601562e-05,
      "step": 13690
    },
    {
      "epoch": 1.3055801391601562e-05,
      "step": 13690,
      "training_step_time": 6.89609169960022
    },
    {
      "epoch": 1.305675506591797e-05,
      "model_forward_time": 1.7304463386535645,
      "step": 13691
    },
    {
      "epoch": 1.305675506591797e-05,
      "step": 13691,
      "training_step_time": 6.912726402282715
    },
    {
      "epoch": 1.3057708740234375e-05,
      "model_forward_time": 1.3638725280761719,
      "step": 13692
    },
    {
      "epoch": 1.3057708740234375e-05,
      "step": 13692,
      "training_step_time": 6.8646767139434814
    },
    {
      "epoch": 1.3058662414550781e-05,
      "model_forward_time": 1.7328290939331055,
      "step": 13693
    },
    {
      "epoch": 1.3058662414550781e-05,
      "step": 13693,
      "training_step_time": 6.865701913833618
    },
    {
      "epoch": 1.3059616088867188e-05,
      "model_forward_time": 1.7375240325927734,
      "step": 13694
    },
    {
      "epoch": 1.3059616088867188e-05,
      "step": 13694,
      "training_step_time": 6.829639196395874
    },
    {
      "epoch": 1.3060569763183594e-05,
      "model_forward_time": 1.7425627708435059,
      "step": 13695
    },
    {
      "epoch": 1.3060569763183594e-05,
      "step": 13695,
      "training_step_time": 6.807063102722168
    },
    {
      "epoch": 1.30615234375e-05,
      "model_forward_time": 1.6956496238708496,
      "step": 13696
    },
    {
      "epoch": 1.30615234375e-05,
      "step": 13696,
      "training_step_time": 7.679513692855835
    },
    {
      "epoch": 1.3062477111816405e-05,
      "model_forward_time": 2.258869171142578,
      "step": 13697
    },
    {
      "epoch": 1.3062477111816405e-05,
      "step": 13697,
      "training_step_time": 11.208262205123901
    },
    {
      "epoch": 1.3063430786132813e-05,
      "model_forward_time": 1.73618483543396,
      "step": 13698
    },
    {
      "epoch": 1.3063430786132813e-05,
      "step": 13698,
      "training_step_time": 7.799781084060669
    },
    {
      "epoch": 1.3064384460449219e-05,
      "model_forward_time": 1.7329351902008057,
      "step": 13699
    },
    {
      "epoch": 1.3064384460449219e-05,
      "step": 13699,
      "training_step_time": 6.800918340682983
    },
    {
      "epoch": 1.3065338134765624e-05,
      "grad_norm": 0.2623950242996216,
      "learning_rate": 9.794536749396477e-06,
      "loss": 0.0664,
      "step": 13700
    },
    {
      "epoch": 1.3065338134765624e-05,
      "model_forward_time": 1.0219552516937256,
      "step": 13700
    },
    {
      "dynamics_loss_avg": 0.06474277805536985,
      "epoch": 1.3065338134765624e-05,
      "step": 13700
    },
    {
      "action_loss_avg": 0.005713885359000415,
      "epoch": 1.3065338134765624e-05,
      "step": 13700
    },
    {
      "epoch": 1.3065338134765624e-05,
      "step": 13700,
      "training_step_time": 6.805198907852173
    },
    {
      "epoch": 1.3066291809082032e-05,
      "model_forward_time": 1.7342684268951416,
      "step": 13701
    },
    {
      "epoch": 1.3066291809082032e-05,
      "step": 13701,
      "training_step_time": 6.8625078201293945
    },
    {
      "epoch": 1.3067245483398437e-05,
      "model_forward_time": 1.7316958904266357,
      "step": 13702
    },
    {
      "epoch": 1.3067245483398437e-05,
      "step": 13702,
      "training_step_time": 6.858152866363525
    },
    {
      "epoch": 1.3068199157714843e-05,
      "model_forward_time": 1.7309651374816895,
      "step": 13703
    },
    {
      "epoch": 1.3068199157714843e-05,
      "step": 13703,
      "training_step_time": 6.9016313552856445
    },
    {
      "epoch": 1.306915283203125e-05,
      "model_forward_time": 1.7294235229492188,
      "step": 13704
    },
    {
      "epoch": 1.306915283203125e-05,
      "step": 13704,
      "training_step_time": 6.889436960220337
    },
    {
      "epoch": 1.3070106506347656e-05,
      "model_forward_time": 1.7306737899780273,
      "step": 13705
    },
    {
      "epoch": 1.3070106506347656e-05,
      "step": 13705,
      "training_step_time": 7.655956029891968
    },
    {
      "epoch": 1.3071060180664062e-05,
      "model_forward_time": 1.7611429691314697,
      "step": 13706
    },
    {
      "epoch": 1.3071060180664062e-05,
      "step": 13706,
      "training_step_time": 11.354223012924194
    },
    {
      "epoch": 1.307201385498047e-05,
      "model_forward_time": 2.271310329437256,
      "step": 13707
    },
    {
      "epoch": 1.307201385498047e-05,
      "step": 13707,
      "training_step_time": 7.8359375
    },
    {
      "epoch": 1.3072967529296875e-05,
      "model_forward_time": 1.7337050437927246,
      "step": 13708
    },
    {
      "epoch": 1.3072967529296875e-05,
      "step": 13708,
      "training_step_time": 6.8538453578948975
    },
    {
      "epoch": 1.3073921203613281e-05,
      "model_forward_time": 1.7346177101135254,
      "step": 13709
    },
    {
      "epoch": 1.3073921203613281e-05,
      "step": 13709,
      "training_step_time": 6.886024713516235
    },
    {
      "epoch": 1.3074874877929688e-05,
      "grad_norm": 0.3965229094028473,
      "learning_rate": 9.79406736578923e-06,
      "loss": 0.0672,
      "step": 13710
    },
    {
      "epoch": 1.3074874877929688e-05,
      "model_forward_time": 1.738102674484253,
      "step": 13710
    },
    {
      "dynamics_loss_avg": 0.0564139224588871,
      "epoch": 1.3074874877929688e-05,
      "step": 13710
    },
    {
      "action_loss_avg": 0.002526576811214909,
      "epoch": 1.3074874877929688e-05,
      "step": 13710
    },
    {
      "epoch": 1.3074874877929688e-05,
      "step": 13710,
      "training_step_time": 6.853955984115601
    },
    {
      "epoch": 1.3075828552246094e-05,
      "model_forward_time": 1.7320373058319092,
      "step": 13711
    },
    {
      "epoch": 1.3075828552246094e-05,
      "step": 13711,
      "training_step_time": 6.847537040710449
    },
    {
      "epoch": 1.30767822265625e-05,
      "model_forward_time": 1.734067440032959,
      "step": 13712
    },
    {
      "epoch": 1.30767822265625e-05,
      "step": 13712,
      "training_step_time": 6.835596561431885
    },
    {
      "epoch": 1.3077735900878905e-05,
      "model_forward_time": 1.7332019805908203,
      "step": 13713
    },
    {
      "epoch": 1.3077735900878905e-05,
      "step": 13713,
      "training_step_time": 6.877429246902466
    },
    {
      "epoch": 1.3078689575195313e-05,
      "model_forward_time": 1.7320137023925781,
      "step": 13714
    },
    {
      "epoch": 1.3078689575195313e-05,
      "step": 13714,
      "training_step_time": 7.636337757110596
    },
    {
      "epoch": 1.3079643249511719e-05,
      "model_forward_time": 1.8044679164886475,
      "step": 13715
    },
    {
      "epoch": 1.3079643249511719e-05,
      "step": 13715,
      "training_step_time": 9.441415309906006
    },
    {
      "epoch": 1.3080596923828124e-05,
      "model_forward_time": 2.2712349891662598,
      "step": 13716
    },
    {
      "epoch": 1.3080596923828124e-05,
      "step": 13716,
      "training_step_time": 8.68470287322998
    },
    {
      "epoch": 1.3081550598144532e-05,
      "model_forward_time": 1.7420034408569336,
      "step": 13717
    },
    {
      "epoch": 1.3081550598144532e-05,
      "step": 13717,
      "training_step_time": 6.959894180297852
    },
    {
      "epoch": 1.3082504272460937e-05,
      "model_forward_time": 1.738154649734497,
      "step": 13718
    },
    {
      "epoch": 1.3082504272460937e-05,
      "step": 13718,
      "training_step_time": 6.996395111083984
    },
    {
      "epoch": 1.3083457946777343e-05,
      "model_forward_time": 1.7359349727630615,
      "step": 13719
    },
    {
      "epoch": 1.3083457946777343e-05,
      "step": 13719,
      "training_step_time": 6.953291416168213
    },
    {
      "epoch": 1.308441162109375e-05,
      "grad_norm": 0.21593910455703735,
      "learning_rate": 9.793597457909985e-06,
      "loss": 0.067,
      "step": 13720
    },
    {
      "epoch": 1.308441162109375e-05,
      "model_forward_time": 1.7253010272979736,
      "step": 13720
    },
    {
      "dynamics_loss_avg": 0.06365201398730277,
      "epoch": 1.308441162109375e-05,
      "step": 13720
    },
    {
      "action_loss_avg": 0.0026869252673350275,
      "epoch": 1.308441162109375e-05,
      "step": 13720
    },
    {
      "epoch": 1.308441162109375e-05,
      "step": 13720,
      "training_step_time": 6.944355249404907
    },
    {
      "epoch": 1.3085365295410156e-05,
      "model_forward_time": 1.7316484451293945,
      "step": 13721
    },
    {
      "epoch": 1.3085365295410156e-05,
      "step": 13721,
      "training_step_time": 6.9919960498809814
    },
    {
      "epoch": 1.3086318969726562e-05,
      "model_forward_time": 1.734499216079712,
      "step": 13722
    },
    {
      "epoch": 1.3086318969726562e-05,
      "step": 13722,
      "training_step_time": 6.918798208236694
    },
    {
      "epoch": 1.308727264404297e-05,
      "model_forward_time": 1.7254986763000488,
      "step": 13723
    },
    {
      "epoch": 1.308727264404297e-05,
      "step": 13723,
      "training_step_time": 7.638925075531006
    },
    {
      "epoch": 1.3088226318359375e-05,
      "model_forward_time": 1.803570032119751,
      "step": 13724
    },
    {
      "epoch": 1.3088226318359375e-05,
      "step": 13724,
      "training_step_time": 10.33867597579956
    },
    {
      "epoch": 1.3089179992675781e-05,
      "model_forward_time": 2.272883892059326,
      "step": 13725
    },
    {
      "epoch": 1.3089179992675781e-05,
      "step": 13725,
      "training_step_time": 9.346831560134888
    },
    {
      "epoch": 1.3090133666992188e-05,
      "model_forward_time": 1.7485215663909912,
      "step": 13726
    },
    {
      "epoch": 1.3090133666992188e-05,
      "step": 13726,
      "training_step_time": 7.810845851898193
    },
    {
      "epoch": 1.3091087341308594e-05,
      "model_forward_time": 1.7319231033325195,
      "step": 13727
    },
    {
      "epoch": 1.3091087341308594e-05,
      "step": 13727,
      "training_step_time": 6.940359115600586
    },
    {
      "epoch": 1.3092041015625e-05,
      "model_forward_time": 1.7328290939331055,
      "step": 13728
    },
    {
      "epoch": 1.3092041015625e-05,
      "step": 13728,
      "training_step_time": 6.913247585296631
    },
    {
      "epoch": 1.3092994689941405e-05,
      "model_forward_time": 1.7298576831817627,
      "step": 13729
    },
    {
      "epoch": 1.3092994689941405e-05,
      "step": 13729,
      "training_step_time": 6.904027462005615
    },
    {
      "epoch": 1.3093948364257813e-05,
      "grad_norm": 0.18354538083076477,
      "learning_rate": 9.793127025810127e-06,
      "loss": 0.0661,
      "step": 13730
    },
    {
      "epoch": 1.3093948364257813e-05,
      "model_forward_time": 1.7286200523376465,
      "step": 13730
    },
    {
      "dynamics_loss_avg": 0.07197571471333504,
      "epoch": 1.3093948364257813e-05,
      "step": 13730
    },
    {
      "action_loss_avg": 0.003732019988819957,
      "epoch": 1.3093948364257813e-05,
      "step": 13730
    },
    {
      "epoch": 1.3093948364257813e-05,
      "step": 13730,
      "training_step_time": 6.950763463973999
    },
    {
      "epoch": 1.3094902038574219e-05,
      "model_forward_time": 1.73494553565979,
      "step": 13731
    },
    {
      "epoch": 1.3094902038574219e-05,
      "step": 13731,
      "training_step_time": 6.924299716949463
    },
    {
      "epoch": 1.3095855712890624e-05,
      "model_forward_time": 1.7369098663330078,
      "step": 13732
    },
    {
      "epoch": 1.3095855712890624e-05,
      "step": 13732,
      "training_step_time": 7.536907196044922
    },
    {
      "epoch": 1.3096809387207032e-05,
      "model_forward_time": 1.8196120262145996,
      "step": 13733
    },
    {
      "epoch": 1.3096809387207032e-05,
      "step": 13733,
      "training_step_time": 8.40941596031189
    },
    {
      "epoch": 1.3097763061523437e-05,
      "model_forward_time": 2.275331735610962,
      "step": 13734
    },
    {
      "epoch": 1.3097763061523437e-05,
      "step": 13734,
      "training_step_time": 9.470642805099487
    },
    {
      "epoch": 1.3098716735839843e-05,
      "model_forward_time": 1.726698398590088,
      "step": 13735
    },
    {
      "epoch": 1.3098716735839843e-05,
      "step": 13735,
      "training_step_time": 7.797448635101318
    },
    {
      "epoch": 1.309967041015625e-05,
      "model_forward_time": 1.7304108142852783,
      "step": 13736
    },
    {
      "epoch": 1.309967041015625e-05,
      "step": 13736,
      "training_step_time": 6.924534559249878
    },
    {
      "epoch": 1.3100624084472656e-05,
      "model_forward_time": 1.735840082168579,
      "step": 13737
    },
    {
      "epoch": 1.3100624084472656e-05,
      "step": 13737,
      "training_step_time": 6.902345180511475
    },
    {
      "epoch": 1.3101577758789062e-05,
      "model_forward_time": 1.3423409461975098,
      "step": 13738
    },
    {
      "epoch": 1.3101577758789062e-05,
      "step": 13738,
      "training_step_time": 6.881688356399536
    },
    {
      "epoch": 1.310253143310547e-05,
      "model_forward_time": 1.732395887374878,
      "step": 13739
    },
    {
      "epoch": 1.310253143310547e-05,
      "step": 13739,
      "training_step_time": 6.911278009414673
    },
    {
      "epoch": 1.3103485107421875e-05,
      "grad_norm": 0.1908484399318695,
      "learning_rate": 9.792656069541105e-06,
      "loss": 0.0678,
      "step": 13740
    },
    {
      "epoch": 1.3103485107421875e-05,
      "model_forward_time": 1.7243690490722656,
      "step": 13740
    },
    {
      "dynamics_loss_avg": 0.05705257961526513,
      "epoch": 1.3103485107421875e-05,
      "step": 13740
    },
    {
      "action_loss_avg": 0.0034064229199429973,
      "epoch": 1.3103485107421875e-05,
      "step": 13740
    },
    {
      "epoch": 1.3103485107421875e-05,
      "step": 13740,
      "training_step_time": 6.903802156448364
    },
    {
      "epoch": 1.3104438781738281e-05,
      "model_forward_time": 1.7446873188018799,
      "step": 13741
    },
    {
      "epoch": 1.3104438781738281e-05,
      "step": 13741,
      "training_step_time": 7.607238054275513
    },
    {
      "epoch": 1.3105392456054688e-05,
      "model_forward_time": 1.8006906509399414,
      "step": 13742
    },
    {
      "epoch": 1.3105392456054688e-05,
      "step": 13742,
      "training_step_time": 8.373831748962402
    },
    {
      "epoch": 1.3106346130371094e-05,
      "model_forward_time": 1.7269301414489746,
      "step": 13743
    },
    {
      "epoch": 1.3106346130371094e-05,
      "step": 13743,
      "training_step_time": 10.18722653388977
    },
    {
      "epoch": 1.31072998046875e-05,
      "model_forward_time": 2.2723429203033447,
      "step": 13744
    },
    {
      "epoch": 1.31072998046875e-05,
      "step": 13744,
      "training_step_time": 7.864060401916504
    },
    {
      "epoch": 1.3108253479003906e-05,
      "model_forward_time": 1.7378590106964111,
      "step": 13745
    },
    {
      "epoch": 1.3108253479003906e-05,
      "step": 13745,
      "training_step_time": 7.513064861297607
    },
    {
      "epoch": 1.3109207153320313e-05,
      "model_forward_time": 1.726841688156128,
      "step": 13746
    },
    {
      "epoch": 1.3109207153320313e-05,
      "step": 13746,
      "training_step_time": 6.896680116653442
    },
    {
      "epoch": 1.3110160827636719e-05,
      "model_forward_time": 1.7298624515533447,
      "step": 13747
    },
    {
      "epoch": 1.3110160827636719e-05,
      "step": 13747,
      "training_step_time": 6.937066555023193
    },
    {
      "epoch": 1.3111114501953124e-05,
      "model_forward_time": 1.7367172241210938,
      "step": 13748
    },
    {
      "epoch": 1.3111114501953124e-05,
      "step": 13748,
      "training_step_time": 6.922933101654053
    },
    {
      "epoch": 1.3112068176269532e-05,
      "model_forward_time": 1.7203540802001953,
      "step": 13749
    },
    {
      "epoch": 1.3112068176269532e-05,
      "step": 13749,
      "training_step_time": 6.936827182769775
    },
    {
      "epoch": 1.3113021850585938e-05,
      "grad_norm": 0.20927223563194275,
      "learning_rate": 9.792184589154419e-06,
      "loss": 0.0691,
      "step": 13750
    },
    {
      "epoch": 1.3113021850585938e-05,
      "model_forward_time": 1.7295615673065186,
      "step": 13750
    },
    {
      "dynamics_loss_avg": 0.06604936495423316,
      "epoch": 1.3113021850585938e-05,
      "step": 13750
    },
    {
      "action_loss_avg": 0.002432667254470289,
      "epoch": 1.3113021850585938e-05,
      "step": 13750
    },
    {
      "epoch": 1.3113021850585938e-05,
      "step": 13750,
      "training_step_time": 7.577720880508423
    },
    {
      "epoch": 1.3113975524902343e-05,
      "model_forward_time": 1.800424337387085,
      "step": 13751
    },
    {
      "epoch": 1.3113975524902343e-05,
      "step": 13751,
      "training_step_time": 9.507917642593384
    },
    {
      "epoch": 1.311492919921875e-05,
      "model_forward_time": 1.7308683395385742,
      "step": 13752
    },
    {
      "epoch": 1.311492919921875e-05,
      "step": 13752,
      "training_step_time": 9.3792245388031
    },
    {
      "epoch": 1.3115882873535156e-05,
      "model_forward_time": 2.274296760559082,
      "step": 13753
    },
    {
      "epoch": 1.3115882873535156e-05,
      "step": 13753,
      "training_step_time": 7.906314134597778
    },
    {
      "epoch": 1.3116836547851562e-05,
      "model_forward_time": 1.7369616031646729,
      "step": 13754
    },
    {
      "epoch": 1.3116836547851562e-05,
      "step": 13754,
      "training_step_time": 7.752796411514282
    },
    {
      "epoch": 1.311779022216797e-05,
      "model_forward_time": 1.7298696041107178,
      "step": 13755
    },
    {
      "epoch": 1.311779022216797e-05,
      "step": 13755,
      "training_step_time": 6.913711309432983
    },
    {
      "epoch": 1.3118743896484375e-05,
      "model_forward_time": 1.7359790802001953,
      "step": 13756
    },
    {
      "epoch": 1.3118743896484375e-05,
      "step": 13756,
      "training_step_time": 6.9130213260650635
    },
    {
      "epoch": 1.3119697570800781e-05,
      "model_forward_time": 1.7301702499389648,
      "step": 13757
    },
    {
      "epoch": 1.3119697570800781e-05,
      "step": 13757,
      "training_step_time": 6.947442531585693
    },
    {
      "epoch": 1.3120651245117187e-05,
      "model_forward_time": 1.7334177494049072,
      "step": 13758
    },
    {
      "epoch": 1.3120651245117187e-05,
      "step": 13758,
      "training_step_time": 6.941384315490723
    },
    {
      "epoch": 1.3121604919433594e-05,
      "model_forward_time": 1.3657007217407227,
      "step": 13759
    },
    {
      "epoch": 1.3121604919433594e-05,
      "step": 13759,
      "training_step_time": 7.543579339981079
    },
    {
      "epoch": 1.312255859375e-05,
      "grad_norm": 0.18884621560573578,
      "learning_rate": 9.791712584701634e-06,
      "loss": 0.0705,
      "step": 13760
    },
    {
      "epoch": 1.312255859375e-05,
      "model_forward_time": 1.8078572750091553,
      "step": 13760
    },
    {
      "dynamics_loss_avg": 0.06301391497254372,
      "epoch": 1.312255859375e-05,
      "step": 13760
    },
    {
      "action_loss_avg": 0.0036551831057295204,
      "epoch": 1.312255859375e-05,
      "step": 13760
    },
    {
      "epoch": 1.312255859375e-05,
      "step": 13760,
      "training_step_time": 7.7998106479644775
    },
    {
      "epoch": 1.3123512268066406e-05,
      "model_forward_time": 1.729095458984375,
      "step": 13761
    },
    {
      "epoch": 1.3123512268066406e-05,
      "step": 13761,
      "training_step_time": 12.6638822555542
    },
    {
      "epoch": 1.3124465942382813e-05,
      "model_forward_time": 2.2692480087280273,
      "step": 13762
    },
    {
      "epoch": 1.3124465942382813e-05,
      "step": 13762,
      "training_step_time": 7.834110975265503
    },
    {
      "epoch": 1.3125419616699219e-05,
      "model_forward_time": 1.7070591449737549,
      "step": 13763
    },
    {
      "epoch": 1.3125419616699219e-05,
      "step": 13763,
      "training_step_time": 7.770458221435547
    },
    {
      "epoch": 1.3126373291015624e-05,
      "model_forward_time": 1.7311315536499023,
      "step": 13764
    },
    {
      "epoch": 1.3126373291015624e-05,
      "step": 13764,
      "training_step_time": 6.888703107833862
    },
    {
      "epoch": 1.3127326965332032e-05,
      "model_forward_time": 1.7498118877410889,
      "step": 13765
    },
    {
      "epoch": 1.3127326965332032e-05,
      "step": 13765,
      "training_step_time": 6.897096872329712
    },
    {
      "epoch": 1.3128280639648438e-05,
      "model_forward_time": 1.7461209297180176,
      "step": 13766
    },
    {
      "epoch": 1.3128280639648438e-05,
      "step": 13766,
      "training_step_time": 6.931766510009766
    },
    {
      "epoch": 1.3129234313964843e-05,
      "model_forward_time": 1.731607437133789,
      "step": 13767
    },
    {
      "epoch": 1.3129234313964843e-05,
      "step": 13767,
      "training_step_time": 6.842828989028931
    },
    {
      "epoch": 1.313018798828125e-05,
      "model_forward_time": 1.7233376502990723,
      "step": 13768
    },
    {
      "epoch": 1.313018798828125e-05,
      "step": 13768,
      "training_step_time": 7.537161588668823
    },
    {
      "epoch": 1.3131141662597656e-05,
      "model_forward_time": 1.7471551895141602,
      "step": 13769
    },
    {
      "epoch": 1.3131141662597656e-05,
      "step": 13769,
      "training_step_time": 8.552292823791504
    },
    {
      "epoch": 1.3132095336914062e-05,
      "grad_norm": 0.24179154634475708,
      "learning_rate": 9.791240056234363e-06,
      "loss": 0.0715,
      "step": 13770
    },
    {
      "epoch": 1.3132095336914062e-05,
      "model_forward_time": 1.7202198505401611,
      "step": 13770
    },
    {
      "dynamics_loss_avg": 0.06997332833707333,
      "epoch": 1.3132095336914062e-05,
      "step": 13770
    },
    {
      "action_loss_avg": 0.004074622795451432,
      "epoch": 1.3132095336914062e-05,
      "step": 13770
    },
    {
      "epoch": 1.3132095336914062e-05,
      "step": 13770,
      "training_step_time": 10.586857080459595
    },
    {
      "epoch": 1.313304901123047e-05,
      "model_forward_time": 2.248051643371582,
      "step": 13771
    },
    {
      "epoch": 1.313304901123047e-05,
      "step": 13771,
      "training_step_time": 9.025653600692749
    },
    {
      "epoch": 1.3134002685546875e-05,
      "model_forward_time": 1.7248263359069824,
      "step": 13772
    },
    {
      "epoch": 1.3134002685546875e-05,
      "step": 13772,
      "training_step_time": 7.912068128585815
    },
    {
      "epoch": 1.3134956359863281e-05,
      "model_forward_time": 1.725870132446289,
      "step": 13773
    },
    {
      "epoch": 1.3134956359863281e-05,
      "step": 13773,
      "training_step_time": 7.591768026351929
    },
    {
      "epoch": 1.3135910034179687e-05,
      "model_forward_time": 1.7274463176727295,
      "step": 13774
    },
    {
      "epoch": 1.3135910034179687e-05,
      "step": 13774,
      "training_step_time": 6.944505453109741
    },
    {
      "epoch": 1.3136863708496094e-05,
      "model_forward_time": 1.7241871356964111,
      "step": 13775
    },
    {
      "epoch": 1.3136863708496094e-05,
      "step": 13775,
      "training_step_time": 6.930262327194214
    },
    {
      "epoch": 1.31378173828125e-05,
      "model_forward_time": 1.7296111583709717,
      "step": 13776
    },
    {
      "epoch": 1.31378173828125e-05,
      "step": 13776,
      "training_step_time": 6.94016695022583
    },
    {
      "epoch": 1.3138771057128906e-05,
      "model_forward_time": 1.7292428016662598,
      "step": 13777
    },
    {
      "epoch": 1.3138771057128906e-05,
      "step": 13777,
      "training_step_time": 7.649310827255249
    },
    {
      "epoch": 1.3139724731445313e-05,
      "model_forward_time": 1.7279114723205566,
      "step": 13778
    },
    {
      "epoch": 1.3139724731445313e-05,
      "step": 13778,
      "training_step_time": 7.785980224609375
    },
    {
      "epoch": 1.3140678405761719e-05,
      "model_forward_time": 1.7199828624725342,
      "step": 13779
    },
    {
      "epoch": 1.3140678405761719e-05,
      "step": 13779,
      "training_step_time": 8.989128112792969
    },
    {
      "epoch": 1.3141632080078124e-05,
      "grad_norm": 0.21382135152816772,
      "learning_rate": 9.790767003804283e-06,
      "loss": 0.0701,
      "step": 13780
    },
    {
      "epoch": 1.3141632080078124e-05,
      "model_forward_time": 1.7173223495483398,
      "step": 13780
    },
    {
      "dynamics_loss_avg": 0.061738256737589835,
      "epoch": 1.3141632080078124e-05,
      "step": 13780
    },
    {
      "action_loss_avg": 0.003017466820892878,
      "epoch": 1.3141632080078124e-05,
      "step": 13780
    },
    {
      "epoch": 1.3141632080078124e-05,
      "step": 13780,
      "training_step_time": 9.96926498413086
    },
    {
      "epoch": 1.3142585754394532e-05,
      "model_forward_time": 1.7243800163269043,
      "step": 13781
    },
    {
      "epoch": 1.3142585754394532e-05,
      "step": 13781,
      "training_step_time": 8.619895219802856
    },
    {
      "epoch": 1.3143539428710938e-05,
      "model_forward_time": 1.736624002456665,
      "step": 13782
    },
    {
      "epoch": 1.3143539428710938e-05,
      "step": 13782,
      "training_step_time": 7.818892002105713
    },
    {
      "epoch": 1.3144493103027343e-05,
      "model_forward_time": 1.7344236373901367,
      "step": 13783
    },
    {
      "epoch": 1.3144493103027343e-05,
      "step": 13783,
      "training_step_time": 6.913881063461304
    },
    {
      "epoch": 1.314544677734375e-05,
      "model_forward_time": 1.7332756519317627,
      "step": 13784
    },
    {
      "epoch": 1.314544677734375e-05,
      "step": 13784,
      "training_step_time": 6.945227146148682
    },
    {
      "epoch": 1.3146400451660156e-05,
      "model_forward_time": 1.7387356758117676,
      "step": 13785
    },
    {
      "epoch": 1.3146400451660156e-05,
      "step": 13785,
      "training_step_time": 6.988373756408691
    },
    {
      "epoch": 1.3147354125976562e-05,
      "model_forward_time": 1.7398250102996826,
      "step": 13786
    },
    {
      "epoch": 1.3147354125976562e-05,
      "step": 13786,
      "training_step_time": 7.590633153915405
    },
    {
      "epoch": 1.314830780029297e-05,
      "model_forward_time": 1.7430016994476318,
      "step": 13787
    },
    {
      "epoch": 1.314830780029297e-05,
      "step": 13787,
      "training_step_time": 7.586280584335327
    },
    {
      "epoch": 1.3149261474609375e-05,
      "model_forward_time": 1.8046469688415527,
      "step": 13788
    },
    {
      "epoch": 1.3149261474609375e-05,
      "step": 13788,
      "training_step_time": 8.428964614868164
    },
    {
      "epoch": 1.3150215148925781e-05,
      "model_forward_time": 1.7368249893188477,
      "step": 13789
    },
    {
      "epoch": 1.3150215148925781e-05,
      "step": 13789,
      "training_step_time": 7.918739557266235
    },
    {
      "epoch": 1.3151168823242187e-05,
      "grad_norm": 0.30832982063293457,
      "learning_rate": 9.790293427463126e-06,
      "loss": 0.0688,
      "step": 13790
    },
    {
      "epoch": 1.3151168823242187e-05,
      "model_forward_time": 2.262983560562134,
      "step": 13790
    },
    {
      "dynamics_loss_avg": 0.0649664407595992,
      "epoch": 1.3151168823242187e-05,
      "step": 13790
    },
    {
      "action_loss_avg": 0.004687667824327946,
      "epoch": 1.3151168823242187e-05,
      "step": 13790
    },
    {
      "epoch": 1.3151168823242187e-05,
      "step": 13790,
      "training_step_time": 8.384625911712646
    },
    {
      "epoch": 1.3152122497558594e-05,
      "model_forward_time": 1.7409899234771729,
      "step": 13791
    },
    {
      "epoch": 1.3152122497558594e-05,
      "step": 13791,
      "training_step_time": 7.887512683868408
    },
    {
      "epoch": 1.3153076171875e-05,
      "model_forward_time": 1.7330193519592285,
      "step": 13792
    },
    {
      "epoch": 1.3153076171875e-05,
      "step": 13792,
      "training_step_time": 6.973869800567627
    },
    {
      "epoch": 1.3154029846191406e-05,
      "model_forward_time": 1.7349212169647217,
      "step": 13793
    },
    {
      "epoch": 1.3154029846191406e-05,
      "step": 13793,
      "training_step_time": 6.904052257537842
    },
    {
      "epoch": 1.3154983520507813e-05,
      "model_forward_time": 1.6950809955596924,
      "step": 13794
    },
    {
      "epoch": 1.3154983520507813e-05,
      "step": 13794,
      "training_step_time": 6.974184274673462
    },
    {
      "epoch": 1.3155937194824219e-05,
      "model_forward_time": 1.7069764137268066,
      "step": 13795
    },
    {
      "epoch": 1.3155937194824219e-05,
      "step": 13795,
      "training_step_time": 7.659294366836548
    },
    {
      "epoch": 1.3156890869140624e-05,
      "model_forward_time": 1.7140862941741943,
      "step": 13796
    },
    {
      "epoch": 1.3156890869140624e-05,
      "step": 13796,
      "training_step_time": 7.629727363586426
    },
    {
      "epoch": 1.3157844543457032e-05,
      "model_forward_time": 1.7372949123382568,
      "step": 13797
    },
    {
      "epoch": 1.3157844543457032e-05,
      "step": 13797,
      "training_step_time": 7.930245637893677
    },
    {
      "epoch": 1.3158798217773438e-05,
      "model_forward_time": 1.7075457572937012,
      "step": 13798
    },
    {
      "epoch": 1.3158798217773438e-05,
      "step": 13798,
      "training_step_time": 10.811985731124878
    },
    {
      "epoch": 1.3159751892089843e-05,
      "model_forward_time": 2.239222288131714,
      "step": 13799
    },
    {
      "epoch": 1.3159751892089843e-05,
      "step": 13799,
      "training_step_time": 10.336085319519043
    },
    {
      "epoch": 1.316070556640625e-05,
      "grad_norm": 0.17287728190422058,
      "learning_rate": 9.789819327262685e-06,
      "loss": 0.0657,
      "step": 13800
    },
    {
      "epoch": 1.316070556640625e-05,
      "model_forward_time": 1.7057080268859863,
      "step": 13800
    },
    {
      "dynamics_loss_avg": 0.06724849622696638,
      "epoch": 1.316070556640625e-05,
      "step": 13800
    },
    {
      "action_loss_avg": 0.0038466561876703055,
      "epoch": 1.316070556640625e-05,
      "step": 13800
    },
    {
      "epoch": 1.316070556640625e-05,
      "step": 13800,
      "training_step_time": 7.924060583114624
    },
    {
      "epoch": 1.3161659240722656e-05,
      "model_forward_time": 1.7058281898498535,
      "step": 13801
    },
    {
      "epoch": 1.3161659240722656e-05,
      "step": 13801,
      "training_step_time": 6.959333658218384
    },
    {
      "epoch": 1.3162612915039062e-05,
      "model_forward_time": 1.7077584266662598,
      "step": 13802
    },
    {
      "epoch": 1.3162612915039062e-05,
      "step": 13802,
      "training_step_time": 6.941744089126587
    },
    {
      "epoch": 1.316356658935547e-05,
      "model_forward_time": 1.7017769813537598,
      "step": 13803
    },
    {
      "epoch": 1.316356658935547e-05,
      "step": 13803,
      "training_step_time": 6.895313024520874
    },
    {
      "epoch": 1.3164520263671875e-05,
      "model_forward_time": 1.703049659729004,
      "step": 13804
    },
    {
      "epoch": 1.3164520263671875e-05,
      "step": 13804,
      "training_step_time": 6.934003114700317
    },
    {
      "epoch": 1.3165473937988281e-05,
      "model_forward_time": 1.7001829147338867,
      "step": 13805
    },
    {
      "epoch": 1.3165473937988281e-05,
      "step": 13805,
      "training_step_time": 7.699993848800659
    },
    {
      "epoch": 1.3166427612304687e-05,
      "model_forward_time": 1.733151912689209,
      "step": 13806
    },
    {
      "epoch": 1.3166427612304687e-05,
      "step": 13806,
      "training_step_time": 8.858247518539429
    },
    {
      "epoch": 1.3167381286621094e-05,
      "model_forward_time": 1.7130658626556396,
      "step": 13807
    },
    {
      "epoch": 1.3167381286621094e-05,
      "step": 13807,
      "training_step_time": 10.049967288970947
    },
    {
      "epoch": 1.31683349609375e-05,
      "model_forward_time": 2.227489948272705,
      "step": 13808
    },
    {
      "epoch": 1.31683349609375e-05,
      "step": 13808,
      "training_step_time": 8.921728610992432
    },
    {
      "epoch": 1.3169288635253906e-05,
      "model_forward_time": 1.7134599685668945,
      "step": 13809
    },
    {
      "epoch": 1.3169288635253906e-05,
      "step": 13809,
      "training_step_time": 8.55129861831665
    },
    {
      "epoch": 1.3170242309570313e-05,
      "grad_norm": 0.22786656022071838,
      "learning_rate": 9.7893447032548e-06,
      "loss": 0.0642,
      "step": 13810
    },
    {
      "epoch": 1.3170242309570313e-05,
      "model_forward_time": 1.694434404373169,
      "step": 13810
    },
    {
      "dynamics_loss_avg": 0.07612186819314956,
      "epoch": 1.3170242309570313e-05,
      "step": 13810
    },
    {
      "action_loss_avg": 0.003905856714118272,
      "epoch": 1.3170242309570313e-05,
      "step": 13810
    },
    {
      "epoch": 1.3170242309570313e-05,
      "step": 13810,
      "training_step_time": 7.94213342666626
    },
    {
      "epoch": 1.3171195983886719e-05,
      "model_forward_time": 1.722163438796997,
      "step": 13811
    },
    {
      "epoch": 1.3171195983886719e-05,
      "step": 13811,
      "training_step_time": 6.986120700836182
    },
    {
      "epoch": 1.3172149658203124e-05,
      "model_forward_time": 1.7166976928710938,
      "step": 13812
    },
    {
      "epoch": 1.3172149658203124e-05,
      "step": 13812,
      "training_step_time": 6.936628341674805
    },
    {
      "epoch": 1.3173103332519532e-05,
      "model_forward_time": 1.712630033493042,
      "step": 13813
    },
    {
      "epoch": 1.3173103332519532e-05,
      "step": 13813,
      "training_step_time": 6.9477949142456055
    },
    {
      "epoch": 1.3174057006835938e-05,
      "model_forward_time": 1.7093377113342285,
      "step": 13814
    },
    {
      "epoch": 1.3174057006835938e-05,
      "step": 13814,
      "training_step_time": 7.659180402755737
    },
    {
      "epoch": 1.3175010681152343e-05,
      "model_forward_time": 1.7215909957885742,
      "step": 13815
    },
    {
      "epoch": 1.3175010681152343e-05,
      "step": 13815,
      "training_step_time": 7.917433023452759
    },
    {
      "epoch": 1.317596435546875e-05,
      "model_forward_time": 1.7104003429412842,
      "step": 13816
    },
    {
      "epoch": 1.317596435546875e-05,
      "step": 13816,
      "training_step_time": 8.724194049835205
    },
    {
      "epoch": 1.3176918029785156e-05,
      "model_forward_time": 1.5024335384368896,
      "step": 13817
    },
    {
      "epoch": 1.3176918029785156e-05,
      "step": 13817,
      "training_step_time": 8.935421705245972
    },
    {
      "epoch": 1.3177871704101562e-05,
      "model_forward_time": 1.7071523666381836,
      "step": 13818
    },
    {
      "epoch": 1.3177871704101562e-05,
      "step": 13818,
      "training_step_time": 7.833481550216675
    },
    {
      "epoch": 1.317882537841797e-05,
      "model_forward_time": 1.7147629261016846,
      "step": 13819
    },
    {
      "epoch": 1.317882537841797e-05,
      "step": 13819,
      "training_step_time": 7.969600200653076
    },
    {
      "epoch": 1.3179779052734375e-05,
      "grad_norm": 0.2281138002872467,
      "learning_rate": 9.78886955549138e-06,
      "loss": 0.0615,
      "step": 13820
    },
    {
      "epoch": 1.3179779052734375e-05,
      "model_forward_time": 1.6981470584869385,
      "step": 13820
    },
    {
      "dynamics_loss_avg": 0.05679375594481826,
      "epoch": 1.3179779052734375e-05,
      "step": 13820
    },
    {
      "action_loss_avg": 0.0043652769993059335,
      "epoch": 1.3179779052734375e-05,
      "step": 13820
    },
    {
      "epoch": 1.3179779052734375e-05,
      "step": 13820,
      "training_step_time": 7.036667108535767
    },
    {
      "epoch": 1.3180732727050781e-05,
      "model_forward_time": 1.7090692520141602,
      "step": 13821
    },
    {
      "epoch": 1.3180732727050781e-05,
      "step": 13821,
      "training_step_time": 7.005671262741089
    },
    {
      "epoch": 1.3181686401367187e-05,
      "model_forward_time": 1.7165007591247559,
      "step": 13822
    },
    {
      "epoch": 1.3181686401367187e-05,
      "step": 13822,
      "training_step_time": 7.008117914199829
    },
    {
      "epoch": 1.3182640075683594e-05,
      "model_forward_time": 1.748610258102417,
      "step": 13823
    },
    {
      "epoch": 1.3182640075683594e-05,
      "step": 13823,
      "training_step_time": 7.710719108581543
    },
    {
      "epoch": 1.318359375e-05,
      "model_forward_time": 1.7337536811828613,
      "step": 13824
    },
    {
      "epoch": 1.318359375e-05,
      "step": 13824,
      "training_step_time": 9.44321084022522
    },
    {
      "epoch": 1.3184547424316406e-05,
      "model_forward_time": 1.7541577816009521,
      "step": 13825
    },
    {
      "epoch": 1.3184547424316406e-05,
      "step": 13825,
      "training_step_time": 8.165951013565063
    },
    {
      "epoch": 1.3185501098632813e-05,
      "model_forward_time": 1.7303402423858643,
      "step": 13826
    },
    {
      "epoch": 1.3185501098632813e-05,
      "step": 13826,
      "training_step_time": 8.518117427825928
    },
    {
      "epoch": 1.3186454772949219e-05,
      "model_forward_time": 2.963731527328491,
      "step": 13827
    },
    {
      "epoch": 1.3186454772949219e-05,
      "step": 13827,
      "training_step_time": 7.736631393432617
    },
    {
      "epoch": 1.3187408447265624e-05,
      "model_forward_time": 1.738905906677246,
      "step": 13828
    },
    {
      "epoch": 1.3187408447265624e-05,
      "step": 13828,
      "training_step_time": 7.734581470489502
    },
    {
      "epoch": 1.3188362121582032e-05,
      "model_forward_time": 1.7339622974395752,
      "step": 13829
    },
    {
      "epoch": 1.3188362121582032e-05,
      "step": 13829,
      "training_step_time": 6.746525287628174
    },
    {
      "epoch": 1.3189315795898438e-05,
      "grad_norm": 0.20497417449951172,
      "learning_rate": 9.788393884024387e-06,
      "loss": 0.065,
      "step": 13830
    },
    {
      "epoch": 1.3189315795898438e-05,
      "model_forward_time": 1.737513780593872,
      "step": 13830
    },
    {
      "dynamics_loss_avg": 0.06872403118759393,
      "epoch": 1.3189315795898438e-05,
      "step": 13830
    },
    {
      "action_loss_avg": 0.003528571955394,
      "epoch": 1.3189315795898438e-05,
      "step": 13830
    },
    {
      "epoch": 1.3189315795898438e-05,
      "step": 13830,
      "training_step_time": 6.758974313735962
    },
    {
      "epoch": 1.3190269470214843e-05,
      "model_forward_time": 1.739330768585205,
      "step": 13831
    },
    {
      "epoch": 1.3190269470214843e-05,
      "step": 13831,
      "training_step_time": 7.0050294399261475
    },
    {
      "epoch": 1.319122314453125e-05,
      "model_forward_time": 1.7421743869781494,
      "step": 13832
    },
    {
      "epoch": 1.319122314453125e-05,
      "step": 13832,
      "training_step_time": 7.680881023406982
    },
    {
      "epoch": 1.3192176818847656e-05,
      "model_forward_time": 1.733757734298706,
      "step": 13833
    },
    {
      "epoch": 1.3192176818847656e-05,
      "step": 13833,
      "training_step_time": 8.041618824005127
    },
    {
      "epoch": 1.3193130493164062e-05,
      "model_forward_time": 1.739861249923706,
      "step": 13834
    },
    {
      "epoch": 1.3193130493164062e-05,
      "step": 13834,
      "training_step_time": 8.04370379447937
    },
    {
      "epoch": 1.319408416748047e-05,
      "model_forward_time": 1.7404975891113281,
      "step": 13835
    },
    {
      "epoch": 1.319408416748047e-05,
      "step": 13835,
      "training_step_time": 8.003661632537842
    },
    {
      "epoch": 1.3195037841796875e-05,
      "model_forward_time": 2.930922746658325,
      "step": 13836
    },
    {
      "epoch": 1.3195037841796875e-05,
      "step": 13836,
      "training_step_time": 8.73807978630066
    },
    {
      "epoch": 1.3195991516113281e-05,
      "model_forward_time": 1.74760103225708,
      "step": 13837
    },
    {
      "epoch": 1.3195991516113281e-05,
      "step": 13837,
      "training_step_time": 8.019603490829468
    },
    {
      "epoch": 1.3196945190429687e-05,
      "model_forward_time": 1.7333459854125977,
      "step": 13838
    },
    {
      "epoch": 1.3196945190429687e-05,
      "step": 13838,
      "training_step_time": 7.075811862945557
    },
    {
      "epoch": 1.3197898864746094e-05,
      "model_forward_time": 1.381422519683838,
      "step": 13839
    },
    {
      "epoch": 1.3197898864746094e-05,
      "step": 13839,
      "training_step_time": 6.966086149215698
    },
    {
      "epoch": 1.31988525390625e-05,
      "grad_norm": 0.1972760260105133,
      "learning_rate": 9.787917688905836e-06,
      "loss": 0.0662,
      "step": 13840
    },
    {
      "epoch": 1.31988525390625e-05,
      "model_forward_time": 1.733459711074829,
      "step": 13840
    },
    {
      "dynamics_loss_avg": 0.06662317421287298,
      "epoch": 1.31988525390625e-05,
      "step": 13840
    },
    {
      "action_loss_avg": 0.004581534949829802,
      "epoch": 1.31988525390625e-05,
      "step": 13840
    },
    {
      "epoch": 1.31988525390625e-05,
      "step": 13840,
      "training_step_time": 7.021409034729004
    },
    {
      "epoch": 1.3199806213378906e-05,
      "model_forward_time": 1.7451987266540527,
      "step": 13841
    },
    {
      "epoch": 1.3199806213378906e-05,
      "step": 13841,
      "training_step_time": 7.67413592338562
    },
    {
      "epoch": 1.3200759887695313e-05,
      "model_forward_time": 1.382744550704956,
      "step": 13842
    },
    {
      "epoch": 1.3200759887695313e-05,
      "step": 13842,
      "training_step_time": 7.963812589645386
    },
    {
      "epoch": 1.3201713562011719e-05,
      "model_forward_time": 1.7406609058380127,
      "step": 13843
    },
    {
      "epoch": 1.3201713562011719e-05,
      "step": 13843,
      "training_step_time": 9.430222272872925
    },
    {
      "epoch": 1.3202667236328124e-05,
      "model_forward_time": 1.7398457527160645,
      "step": 13844
    },
    {
      "epoch": 1.3202667236328124e-05,
      "step": 13844,
      "training_step_time": 9.718219518661499
    },
    {
      "epoch": 1.3203620910644532e-05,
      "model_forward_time": 2.8670268058776855,
      "step": 13845
    },
    {
      "epoch": 1.3203620910644532e-05,
      "step": 13845,
      "training_step_time": 9.019868612289429
    },
    {
      "epoch": 1.3204574584960938e-05,
      "model_forward_time": 1.8523736000061035,
      "step": 13846
    },
    {
      "epoch": 1.3204574584960938e-05,
      "step": 13846,
      "training_step_time": 7.991605758666992
    },
    {
      "epoch": 1.3205528259277343e-05,
      "model_forward_time": 1.7452847957611084,
      "step": 13847
    },
    {
      "epoch": 1.3205528259277343e-05,
      "step": 13847,
      "training_step_time": 7.840043544769287
    },
    {
      "epoch": 1.320648193359375e-05,
      "model_forward_time": 1.7341821193695068,
      "step": 13848
    },
    {
      "epoch": 1.320648193359375e-05,
      "step": 13848,
      "training_step_time": 7.043994188308716
    },
    {
      "epoch": 1.3207435607910156e-05,
      "model_forward_time": 1.7354810237884521,
      "step": 13849
    },
    {
      "epoch": 1.3207435607910156e-05,
      "step": 13849,
      "training_step_time": 6.916479587554932
    },
    {
      "epoch": 1.3208389282226562e-05,
      "grad_norm": 0.20333750545978546,
      "learning_rate": 9.787440970187807e-06,
      "loss": 0.0682,
      "step": 13850
    },
    {
      "epoch": 1.3208389282226562e-05,
      "model_forward_time": 1.7329623699188232,
      "step": 13850
    },
    {
      "dynamics_loss_avg": 0.05779835917055607,
      "epoch": 1.3208389282226562e-05,
      "step": 13850
    },
    {
      "action_loss_avg": 0.002842248120578006,
      "epoch": 1.3208389282226562e-05,
      "step": 13850
    },
    {
      "epoch": 1.3208389282226562e-05,
      "step": 13850,
      "training_step_time": 7.700559139251709
    },
    {
      "epoch": 1.320934295654297e-05,
      "model_forward_time": 1.7428507804870605,
      "step": 13851
    },
    {
      "epoch": 1.320934295654297e-05,
      "step": 13851,
      "training_step_time": 7.933596849441528
    },
    {
      "epoch": 1.3210296630859375e-05,
      "model_forward_time": 1.7349793910980225,
      "step": 13852
    },
    {
      "epoch": 1.3210296630859375e-05,
      "step": 13852,
      "training_step_time": 7.97383451461792
    },
    {
      "epoch": 1.3211250305175781e-05,
      "model_forward_time": 1.045125961303711,
      "step": 13853
    },
    {
      "epoch": 1.3211250305175781e-05,
      "step": 13853,
      "training_step_time": 10.274394750595093
    },
    {
      "epoch": 1.3212203979492187e-05,
      "model_forward_time": 1.7496616840362549,
      "step": 13854
    },
    {
      "epoch": 1.3212203979492187e-05,
      "step": 13854,
      "training_step_time": 9.855797290802002
    },
    {
      "epoch": 1.3213157653808594e-05,
      "model_forward_time": 2.8756983280181885,
      "step": 13855
    },
    {
      "epoch": 1.3213157653808594e-05,
      "step": 13855,
      "training_step_time": 7.7517125606536865
    },
    {
      "epoch": 1.3214111328125e-05,
      "model_forward_time": 1.7364168167114258,
      "step": 13856
    },
    {
      "epoch": 1.3214111328125e-05,
      "step": 13856,
      "training_step_time": 7.991264581680298
    },
    {
      "epoch": 1.3215065002441406e-05,
      "model_forward_time": 1.747358798980713,
      "step": 13857
    },
    {
      "epoch": 1.3215065002441406e-05,
      "step": 13857,
      "training_step_time": 7.186470031738281
    },
    {
      "epoch": 1.3216018676757813e-05,
      "model_forward_time": 1.7400848865509033,
      "step": 13858
    },
    {
      "epoch": 1.3216018676757813e-05,
      "step": 13858,
      "training_step_time": 7.02953839302063
    },
    {
      "epoch": 1.3216972351074219e-05,
      "model_forward_time": 1.7333166599273682,
      "step": 13859
    },
    {
      "epoch": 1.3216972351074219e-05,
      "step": 13859,
      "training_step_time": 7.608170747756958
    },
    {
      "epoch": 1.3217926025390624e-05,
      "grad_norm": 0.21473407745361328,
      "learning_rate": 9.786963727922429e-06,
      "loss": 0.064,
      "step": 13860
    },
    {
      "epoch": 1.3217926025390624e-05,
      "model_forward_time": 1.7276155948638916,
      "step": 13860
    },
    {
      "dynamics_loss_avg": 0.05501326359808445,
      "epoch": 1.3217926025390624e-05,
      "step": 13860
    },
    {
      "action_loss_avg": 0.0029825397999957205,
      "epoch": 1.3217926025390624e-05,
      "step": 13860
    },
    {
      "epoch": 1.3217926025390624e-05,
      "step": 13860,
      "training_step_time": 7.9356091022491455
    },
    {
      "epoch": 1.3218879699707032e-05,
      "model_forward_time": 1.7505731582641602,
      "step": 13861
    },
    {
      "epoch": 1.3218879699707032e-05,
      "step": 13861,
      "training_step_time": 7.823389530181885
    },
    {
      "epoch": 1.3219833374023438e-05,
      "model_forward_time": 1.7406747341156006,
      "step": 13862
    },
    {
      "epoch": 1.3219833374023438e-05,
      "step": 13862,
      "training_step_time": 8.590790510177612
    },
    {
      "epoch": 1.3220787048339843e-05,
      "model_forward_time": 1.727802038192749,
      "step": 13863
    },
    {
      "epoch": 1.3220787048339843e-05,
      "step": 13863,
      "training_step_time": 8.755497694015503
    },
    {
      "epoch": 1.322174072265625e-05,
      "model_forward_time": 2.885162115097046,
      "step": 13864
    },
    {
      "epoch": 1.322174072265625e-05,
      "step": 13864,
      "training_step_time": 8.022051811218262
    },
    {
      "epoch": 1.3222694396972656e-05,
      "model_forward_time": 1.7365360260009766,
      "step": 13865
    },
    {
      "epoch": 1.3222694396972656e-05,
      "step": 13865,
      "training_step_time": 7.915915250778198
    },
    {
      "epoch": 1.3223648071289062e-05,
      "model_forward_time": 1.7401247024536133,
      "step": 13866
    },
    {
      "epoch": 1.3223648071289062e-05,
      "step": 13866,
      "training_step_time": 7.01488184928894
    },
    {
      "epoch": 1.322460174560547e-05,
      "model_forward_time": 1.7377965450286865,
      "step": 13867
    },
    {
      "epoch": 1.322460174560547e-05,
      "step": 13867,
      "training_step_time": 6.977047443389893
    },
    {
      "epoch": 1.3225555419921875e-05,
      "model_forward_time": 1.3749449253082275,
      "step": 13868
    },
    {
      "epoch": 1.3225555419921875e-05,
      "step": 13868,
      "training_step_time": 7.592301368713379
    },
    {
      "epoch": 1.3226509094238281e-05,
      "model_forward_time": 1.7389540672302246,
      "step": 13869
    },
    {
      "epoch": 1.3226509094238281e-05,
      "step": 13869,
      "training_step_time": 7.969338893890381
    },
    {
      "epoch": 1.3227462768554687e-05,
      "grad_norm": 0.12006863951683044,
      "learning_rate": 9.786485962161898e-06,
      "loss": 0.0613,
      "step": 13870
    },
    {
      "epoch": 1.3227462768554687e-05,
      "model_forward_time": 1.7393324375152588,
      "step": 13870
    },
    {
      "dynamics_loss_avg": 0.04411461316049099,
      "epoch": 1.3227462768554687e-05,
      "step": 13870
    },
    {
      "action_loss_avg": 0.002341366291511804,
      "epoch": 1.3227462768554687e-05,
      "step": 13870
    },
    {
      "epoch": 1.3227462768554687e-05,
      "step": 13870,
      "training_step_time": 8.741370439529419
    },
    {
      "epoch": 1.3228416442871094e-05,
      "model_forward_time": 1.742633581161499,
      "step": 13871
    },
    {
      "epoch": 1.3228416442871094e-05,
      "step": 13871,
      "training_step_time": 8.482021808624268
    },
    {
      "epoch": 1.32293701171875e-05,
      "model_forward_time": 1.7388741970062256,
      "step": 13872
    },
    {
      "epoch": 1.32293701171875e-05,
      "step": 13872,
      "training_step_time": 8.550580263137817
    },
    {
      "epoch": 1.3230323791503906e-05,
      "model_forward_time": 2.85152530670166,
      "step": 13873
    },
    {
      "epoch": 1.3230323791503906e-05,
      "step": 13873,
      "training_step_time": 9.043214559555054
    },
    {
      "epoch": 1.3231277465820313e-05,
      "model_forward_time": 1.7363049983978271,
      "step": 13874
    },
    {
      "epoch": 1.3231277465820313e-05,
      "step": 13874,
      "training_step_time": 7.96229887008667
    },
    {
      "epoch": 1.3232231140136719e-05,
      "model_forward_time": 1.746842384338379,
      "step": 13875
    },
    {
      "epoch": 1.3232231140136719e-05,
      "step": 13875,
      "training_step_time": 6.984999656677246
    },
    {
      "epoch": 1.3233184814453124e-05,
      "model_forward_time": 1.7349226474761963,
      "step": 13876
    },
    {
      "epoch": 1.3233184814453124e-05,
      "step": 13876,
      "training_step_time": 7.000719308853149
    },
    {
      "epoch": 1.3234138488769532e-05,
      "model_forward_time": 1.7336575984954834,
      "step": 13877
    },
    {
      "epoch": 1.3234138488769532e-05,
      "step": 13877,
      "training_step_time": 7.655533790588379
    },
    {
      "epoch": 1.3235092163085938e-05,
      "model_forward_time": 1.7548437118530273,
      "step": 13878
    },
    {
      "epoch": 1.3235092163085938e-05,
      "step": 13878,
      "training_step_time": 7.826003313064575
    },
    {
      "epoch": 1.3236045837402343e-05,
      "model_forward_time": 1.734800100326538,
      "step": 13879
    },
    {
      "epoch": 1.3236045837402343e-05,
      "step": 13879,
      "training_step_time": 8.787713050842285
    },
    {
      "epoch": 1.323699951171875e-05,
      "grad_norm": 0.10848554968833923,
      "learning_rate": 9.786007672958456e-06,
      "loss": 0.0638,
      "step": 13880
    },
    {
      "epoch": 1.323699951171875e-05,
      "model_forward_time": 1.734428882598877,
      "step": 13880
    },
    {
      "dynamics_loss_avg": 0.05211107172071934,
      "epoch": 1.323699951171875e-05,
      "step": 13880
    },
    {
      "action_loss_avg": 0.00231810491241049,
      "epoch": 1.323699951171875e-05,
      "step": 13880
    },
    {
      "epoch": 1.323699951171875e-05,
      "step": 13880,
      "training_step_time": 7.891100168228149
    },
    {
      "epoch": 1.3237953186035156e-05,
      "model_forward_time": 1.7348575592041016,
      "step": 13881
    },
    {
      "epoch": 1.3237953186035156e-05,
      "step": 13881,
      "training_step_time": 10.210236310958862
    },
    {
      "epoch": 1.3238906860351562e-05,
      "model_forward_time": 2.8479559421539307,
      "step": 13882
    },
    {
      "epoch": 1.3238906860351562e-05,
      "step": 13882,
      "training_step_time": 9.910881280899048
    },
    {
      "epoch": 1.3239860534667968e-05,
      "model_forward_time": 1.7348823547363281,
      "step": 13883
    },
    {
      "epoch": 1.3239860534667968e-05,
      "step": 13883,
      "training_step_time": 8.990230083465576
    },
    {
      "epoch": 1.3240814208984375e-05,
      "model_forward_time": 1.7428579330444336,
      "step": 13884
    },
    {
      "epoch": 1.3240814208984375e-05,
      "step": 13884,
      "training_step_time": 7.754358530044556
    },
    {
      "epoch": 1.3241767883300781e-05,
      "model_forward_time": 1.7424428462982178,
      "step": 13885
    },
    {
      "epoch": 1.3241767883300781e-05,
      "step": 13885,
      "training_step_time": 7.010821580886841
    },
    {
      "epoch": 1.3242721557617187e-05,
      "model_forward_time": 1.7408051490783691,
      "step": 13886
    },
    {
      "epoch": 1.3242721557617187e-05,
      "step": 13886,
      "training_step_time": 7.749846696853638
    },
    {
      "epoch": 1.3243675231933594e-05,
      "model_forward_time": 1.7343552112579346,
      "step": 13887
    },
    {
      "epoch": 1.3243675231933594e-05,
      "step": 13887,
      "training_step_time": 7.829487085342407
    },
    {
      "epoch": 1.324462890625e-05,
      "model_forward_time": 1.7424030303955078,
      "step": 13888
    },
    {
      "epoch": 1.324462890625e-05,
      "step": 13888,
      "training_step_time": 7.939428329467773
    },
    {
      "epoch": 1.3245582580566406e-05,
      "model_forward_time": 1.7465362548828125,
      "step": 13889
    },
    {
      "epoch": 1.3245582580566406e-05,
      "step": 13889,
      "training_step_time": 7.924698114395142
    },
    {
      "epoch": 1.3246536254882813e-05,
      "grad_norm": 0.1727735549211502,
      "learning_rate": 9.78552886036441e-06,
      "loss": 0.0634,
      "step": 13890
    },
    {
      "epoch": 1.3246536254882813e-05,
      "model_forward_time": 1.7313590049743652,
      "step": 13890
    },
    {
      "dynamics_loss_avg": 0.06932625882327556,
      "epoch": 1.3246536254882813e-05,
      "step": 13890
    },
    {
      "action_loss_avg": 0.0034324961598031223,
      "epoch": 1.3246536254882813e-05,
      "step": 13890
    },
    {
      "epoch": 1.3246536254882813e-05,
      "step": 13890,
      "training_step_time": 8.823834896087646
    },
    {
      "epoch": 1.3247489929199219e-05,
      "model_forward_time": 1.73679780960083,
      "step": 13891
    },
    {
      "epoch": 1.3247489929199219e-05,
      "step": 13891,
      "training_step_time": 9.030866146087646
    },
    {
      "epoch": 1.3248443603515624e-05,
      "model_forward_time": 2.8537228107452393,
      "step": 13892
    },
    {
      "epoch": 1.3248443603515624e-05,
      "step": 13892,
      "training_step_time": 7.957507610321045
    },
    {
      "epoch": 1.3249397277832032e-05,
      "model_forward_time": 1.7395648956298828,
      "step": 13893
    },
    {
      "epoch": 1.3249397277832032e-05,
      "step": 13893,
      "training_step_time": 7.925231456756592
    },
    {
      "epoch": 1.3250350952148438e-05,
      "model_forward_time": 1.746537446975708,
      "step": 13894
    },
    {
      "epoch": 1.3250350952148438e-05,
      "step": 13894,
      "training_step_time": 7.019992351531982
    },
    {
      "epoch": 1.3251304626464843e-05,
      "model_forward_time": 1.727384328842163,
      "step": 13895
    },
    {
      "epoch": 1.3251304626464843e-05,
      "step": 13895,
      "training_step_time": 7.634472131729126
    },
    {
      "epoch": 1.325225830078125e-05,
      "model_forward_time": 1.7428703308105469,
      "step": 13896
    },
    {
      "epoch": 1.325225830078125e-05,
      "step": 13896,
      "training_step_time": 7.623345851898193
    },
    {
      "epoch": 1.3253211975097656e-05,
      "model_forward_time": 1.7464079856872559,
      "step": 13897
    },
    {
      "epoch": 1.3253211975097656e-05,
      "step": 13897,
      "training_step_time": 8.025185585021973
    },
    {
      "epoch": 1.3254165649414062e-05,
      "model_forward_time": 1.7306058406829834,
      "step": 13898
    },
    {
      "epoch": 1.3254165649414062e-05,
      "step": 13898,
      "training_step_time": 7.845056056976318
    },
    {
      "epoch": 1.3255119323730468e-05,
      "model_forward_time": 1.735234260559082,
      "step": 13899
    },
    {
      "epoch": 1.3255119323730468e-05,
      "step": 13899,
      "training_step_time": 8.903329133987427
    },
    {
      "epoch": 1.3256072998046875e-05,
      "grad_norm": 0.17399011552333832,
      "learning_rate": 9.785049524432124e-06,
      "loss": 0.0664,
      "step": 13900
    },
    {
      "epoch": 1.3256072998046875e-05,
      "model_forward_time": 1.7335965633392334,
      "step": 13900
    },
    {
      "dynamics_loss_avg": 0.07740194723010063,
      "epoch": 1.3256072998046875e-05,
      "step": 13900
    },
    {
      "action_loss_avg": 0.0025075390527490526,
      "epoch": 1.3256072998046875e-05,
      "step": 13900
    },
    {
      "epoch": 1.3256072998046875e-05,
      "step": 13900,
      "training_step_time": 8.287333250045776
    },
    {
      "epoch": 1.3257026672363281e-05,
      "model_forward_time": 2.855008125305176,
      "step": 13901
    },
    {
      "epoch": 1.3257026672363281e-05,
      "step": 13901,
      "training_step_time": 7.808781862258911
    },
    {
      "epoch": 1.3257980346679687e-05,
      "model_forward_time": 1.748363971710205,
      "step": 13902
    },
    {
      "epoch": 1.3257980346679687e-05,
      "step": 13902,
      "training_step_time": 8.706554651260376
    },
    {
      "epoch": 1.3258934020996094e-05,
      "model_forward_time": 1.734086275100708,
      "step": 13903
    },
    {
      "epoch": 1.3258934020996094e-05,
      "step": 13903,
      "training_step_time": 6.837408065795898
    },
    {
      "epoch": 1.32598876953125e-05,
      "model_forward_time": 1.7332658767700195,
      "step": 13904
    },
    {
      "epoch": 1.32598876953125e-05,
      "step": 13904,
      "training_step_time": 7.484314918518066
    },
    {
      "epoch": 1.3260841369628906e-05,
      "model_forward_time": 1.7377941608428955,
      "step": 13905
    },
    {
      "epoch": 1.3260841369628906e-05,
      "step": 13905,
      "training_step_time": 6.882498741149902
    },
    {
      "epoch": 1.3261795043945313e-05,
      "model_forward_time": 1.735710620880127,
      "step": 13906
    },
    {
      "epoch": 1.3261795043945313e-05,
      "step": 13906,
      "training_step_time": 7.810145854949951
    },
    {
      "epoch": 1.3262748718261719e-05,
      "model_forward_time": 1.380305290222168,
      "step": 13907
    },
    {
      "epoch": 1.3262748718261719e-05,
      "step": 13907,
      "training_step_time": 7.68199348449707
    },
    {
      "epoch": 1.3263702392578124e-05,
      "model_forward_time": 1.737565040588379,
      "step": 13908
    },
    {
      "epoch": 1.3263702392578124e-05,
      "step": 13908,
      "training_step_time": 10.811434984207153
    },
    {
      "epoch": 1.3264656066894532e-05,
      "model_forward_time": 1.3844106197357178,
      "step": 13909
    },
    {
      "epoch": 1.3264656066894532e-05,
      "step": 13909,
      "training_step_time": 9.183979034423828
    },
    {
      "epoch": 1.3265609741210938e-05,
      "grad_norm": 0.17693300545215607,
      "learning_rate": 9.784569665214016e-06,
      "loss": 0.0661,
      "step": 13910
    },
    {
      "epoch": 1.3265609741210938e-05,
      "model_forward_time": 2.486311674118042,
      "step": 13910
    },
    {
      "dynamics_loss_avg": 0.05961842890828848,
      "epoch": 1.3265609741210938e-05,
      "step": 13910
    },
    {
      "action_loss_avg": 0.003912133537232876,
      "epoch": 1.3265609741210938e-05,
      "step": 13910
    },
    {
      "epoch": 1.3265609741210938e-05,
      "step": 13910,
      "training_step_time": 7.770565748214722
    },
    {
      "epoch": 1.3266563415527343e-05,
      "model_forward_time": 1.7368824481964111,
      "step": 13911
    },
    {
      "epoch": 1.3266563415527343e-05,
      "step": 13911,
      "training_step_time": 7.676980972290039
    },
    {
      "epoch": 1.326751708984375e-05,
      "model_forward_time": 1.7393245697021484,
      "step": 13912
    },
    {
      "epoch": 1.326751708984375e-05,
      "step": 13912,
      "training_step_time": 7.77760910987854
    },
    {
      "epoch": 1.3268470764160156e-05,
      "model_forward_time": 1.7535531520843506,
      "step": 13913
    },
    {
      "epoch": 1.3268470764160156e-05,
      "step": 13913,
      "training_step_time": 7.437223434448242
    },
    {
      "epoch": 1.3269424438476562e-05,
      "model_forward_time": 1.7324140071868896,
      "step": 13914
    },
    {
      "epoch": 1.3269424438476562e-05,
      "step": 13914,
      "training_step_time": 6.815333604812622
    },
    {
      "epoch": 1.3270378112792968e-05,
      "model_forward_time": 1.7319891452789307,
      "step": 13915
    },
    {
      "epoch": 1.3270378112792968e-05,
      "step": 13915,
      "training_step_time": 8.71303391456604
    },
    {
      "epoch": 1.3271331787109375e-05,
      "model_forward_time": 1.7473883628845215,
      "step": 13916
    },
    {
      "epoch": 1.3271331787109375e-05,
      "step": 13916,
      "training_step_time": 8.540395021438599
    },
    {
      "epoch": 1.3272285461425781e-05,
      "model_forward_time": 1.753479242324829,
      "step": 13917
    },
    {
      "epoch": 1.3272285461425781e-05,
      "step": 13917,
      "training_step_time": 8.708937406539917
    },
    {
      "epoch": 1.3273239135742187e-05,
      "model_forward_time": 1.7452661991119385,
      "step": 13918
    },
    {
      "epoch": 1.3273239135742187e-05,
      "step": 13918,
      "training_step_time": 8.939520120620728
    },
    {
      "epoch": 1.3274192810058594e-05,
      "model_forward_time": 1.7882182598114014,
      "step": 13919
    },
    {
      "epoch": 1.3274192810058594e-05,
      "step": 13919,
      "training_step_time": 7.689558982849121
    },
    {
      "epoch": 1.3275146484375e-05,
      "grad_norm": 0.22035810351371765,
      "learning_rate": 9.784089282762563e-06,
      "loss": 0.0657,
      "step": 13920
    },
    {
      "epoch": 1.3275146484375e-05,
      "model_forward_time": 2.976374387741089,
      "step": 13920
    },
    {
      "dynamics_loss_avg": 0.06559603661298752,
      "epoch": 1.3275146484375e-05,
      "step": 13920
    },
    {
      "action_loss_avg": 0.003387742640916258,
      "epoch": 1.3275146484375e-05,
      "step": 13920
    },
    {
      "epoch": 1.3275146484375e-05,
      "step": 13920,
      "training_step_time": 8.62813401222229
    },
    {
      "epoch": 1.3276100158691406e-05,
      "model_forward_time": 1.7417166233062744,
      "step": 13921
    },
    {
      "epoch": 1.3276100158691406e-05,
      "step": 13921,
      "training_step_time": 7.725083351135254
    },
    {
      "epoch": 1.3277053833007813e-05,
      "model_forward_time": 1.7309904098510742,
      "step": 13922
    },
    {
      "epoch": 1.3277053833007813e-05,
      "step": 13922,
      "training_step_time": 6.802687883377075
    },
    {
      "epoch": 1.3278007507324219e-05,
      "model_forward_time": 1.7449278831481934,
      "step": 13923
    },
    {
      "epoch": 1.3278007507324219e-05,
      "step": 13923,
      "training_step_time": 7.446753978729248
    },
    {
      "epoch": 1.3278961181640625e-05,
      "model_forward_time": 1.737931251525879,
      "step": 13924
    },
    {
      "epoch": 1.3278961181640625e-05,
      "step": 13924,
      "training_step_time": 7.638747453689575
    },
    {
      "epoch": 1.3279914855957032e-05,
      "model_forward_time": 1.7352111339569092,
      "step": 13925
    },
    {
      "epoch": 1.3279914855957032e-05,
      "step": 13925,
      "training_step_time": 7.7086498737335205
    },
    {
      "epoch": 1.3280868530273438e-05,
      "model_forward_time": 1.7385249137878418,
      "step": 13926
    },
    {
      "epoch": 1.3280868530273438e-05,
      "step": 13926,
      "training_step_time": 8.415201425552368
    },
    {
      "epoch": 1.3281822204589843e-05,
      "model_forward_time": 1.3863017559051514,
      "step": 13927
    },
    {
      "epoch": 1.3281822204589843e-05,
      "step": 13927,
      "training_step_time": 8.339121580123901
    },
    {
      "epoch": 1.328277587890625e-05,
      "model_forward_time": 1.7470014095306396,
      "step": 13928
    },
    {
      "epoch": 1.328277587890625e-05,
      "step": 13928,
      "training_step_time": 8.641830205917358
    },
    {
      "epoch": 1.3283729553222657e-05,
      "model_forward_time": 2.9344310760498047,
      "step": 13929
    },
    {
      "epoch": 1.3283729553222657e-05,
      "step": 13929,
      "training_step_time": 7.827993154525757
    },
    {
      "epoch": 1.3284683227539062e-05,
      "grad_norm": 0.15330128371715546,
      "learning_rate": 9.7836083771303e-06,
      "loss": 0.0669,
      "step": 13930
    },
    {
      "epoch": 1.3284683227539062e-05,
      "model_forward_time": 1.7311608791351318,
      "step": 13930
    },
    {
      "dynamics_loss_avg": 0.06433964893221855,
      "epoch": 1.3284683227539062e-05,
      "step": 13930
    },
    {
      "action_loss_avg": 0.002261439093854278,
      "epoch": 1.3284683227539062e-05,
      "step": 13930
    },
    {
      "epoch": 1.3284683227539062e-05,
      "step": 13930,
      "training_step_time": 7.52149510383606
    },
    {
      "epoch": 1.3285636901855468e-05,
      "model_forward_time": 1.733374834060669,
      "step": 13931
    },
    {
      "epoch": 1.3285636901855468e-05,
      "step": 13931,
      "training_step_time": 7.73079776763916
    },
    {
      "epoch": 1.3286590576171875e-05,
      "model_forward_time": 1.738102674484253,
      "step": 13932
    },
    {
      "epoch": 1.3286590576171875e-05,
      "step": 13932,
      "training_step_time": 7.4598846435546875
    },
    {
      "epoch": 1.3287544250488281e-05,
      "model_forward_time": 1.7335662841796875,
      "step": 13933
    },
    {
      "epoch": 1.3287544250488281e-05,
      "step": 13933,
      "training_step_time": 7.723756790161133
    },
    {
      "epoch": 1.3288497924804687e-05,
      "model_forward_time": 1.7320060729980469,
      "step": 13934
    },
    {
      "epoch": 1.3288497924804687e-05,
      "step": 13934,
      "training_step_time": 7.632117748260498
    },
    {
      "epoch": 1.3289451599121094e-05,
      "model_forward_time": 1.7440669536590576,
      "step": 13935
    },
    {
      "epoch": 1.3289451599121094e-05,
      "step": 13935,
      "training_step_time": 8.709645509719849
    },
    {
      "epoch": 1.32904052734375e-05,
      "model_forward_time": 1.7367849349975586,
      "step": 13936
    },
    {
      "epoch": 1.32904052734375e-05,
      "step": 13936,
      "training_step_time": 9.824527502059937
    },
    {
      "epoch": 1.3291358947753906e-05,
      "model_forward_time": 1.7443087100982666,
      "step": 13937
    },
    {
      "epoch": 1.3291358947753906e-05,
      "step": 13937,
      "training_step_time": 8.23158049583435
    },
    {
      "epoch": 1.3292312622070313e-05,
      "model_forward_time": 2.900534152984619,
      "step": 13938
    },
    {
      "epoch": 1.3292312622070313e-05,
      "step": 13938,
      "training_step_time": 7.722238779067993
    },
    {
      "epoch": 1.3293266296386719e-05,
      "model_forward_time": 1.7444489002227783,
      "step": 13939
    },
    {
      "epoch": 1.3293266296386719e-05,
      "step": 13939,
      "training_step_time": 7.435095310211182
    },
    {
      "epoch": 1.3294219970703125e-05,
      "grad_norm": 0.2532975375652313,
      "learning_rate": 9.783126948369817e-06,
      "loss": 0.0699,
      "step": 13940
    },
    {
      "epoch": 1.3294219970703125e-05,
      "model_forward_time": 1.7489891052246094,
      "step": 13940
    },
    {
      "dynamics_loss_avg": 0.052057243138551715,
      "epoch": 1.3294219970703125e-05,
      "step": 13940
    },
    {
      "action_loss_avg": 0.0020626042620278894,
      "epoch": 1.3294219970703125e-05,
      "step": 13940
    },
    {
      "epoch": 1.3294219970703125e-05,
      "step": 13940,
      "training_step_time": 7.7391579151153564
    },
    {
      "epoch": 1.3295173645019532e-05,
      "model_forward_time": 1.7385871410369873,
      "step": 13941
    },
    {
      "epoch": 1.3295173645019532e-05,
      "step": 13941,
      "training_step_time": 7.409404516220093
    },
    {
      "epoch": 1.3296127319335938e-05,
      "model_forward_time": 1.7682631015777588,
      "step": 13942
    },
    {
      "epoch": 1.3296127319335938e-05,
      "step": 13942,
      "training_step_time": 7.646759033203125
    },
    {
      "epoch": 1.3297080993652343e-05,
      "model_forward_time": 1.3717925548553467,
      "step": 13943
    },
    {
      "epoch": 1.3297080993652343e-05,
      "step": 13943,
      "training_step_time": 7.468233585357666
    },
    {
      "epoch": 1.329803466796875e-05,
      "model_forward_time": 1.7453889846801758,
      "step": 13944
    },
    {
      "epoch": 1.329803466796875e-05,
      "step": 13944,
      "training_step_time": 9.56293249130249
    },
    {
      "epoch": 1.3298988342285157e-05,
      "model_forward_time": 1.7517189979553223,
      "step": 13945
    },
    {
      "epoch": 1.3298988342285157e-05,
      "step": 13945,
      "training_step_time": 8.957651138305664
    },
    {
      "epoch": 1.3299942016601562e-05,
      "model_forward_time": 1.7788658142089844,
      "step": 13946
    },
    {
      "epoch": 1.3299942016601562e-05,
      "step": 13946,
      "training_step_time": 9.503699779510498
    },
    {
      "epoch": 1.3300895690917968e-05,
      "model_forward_time": 2.9232592582702637,
      "step": 13947
    },
    {
      "epoch": 1.3300895690917968e-05,
      "step": 13947,
      "training_step_time": 7.75467586517334
    },
    {
      "epoch": 1.3301849365234375e-05,
      "model_forward_time": 1.7425882816314697,
      "step": 13948
    },
    {
      "epoch": 1.3301849365234375e-05,
      "step": 13948,
      "training_step_time": 7.6664628982543945
    },
    {
      "epoch": 1.3302803039550781e-05,
      "model_forward_time": 1.7501468658447266,
      "step": 13949
    },
    {
      "epoch": 1.3302803039550781e-05,
      "step": 13949,
      "training_step_time": 7.739393472671509
    },
    {
      "epoch": 1.3303756713867187e-05,
      "grad_norm": 0.21506449580192566,
      "learning_rate": 9.78264499653376e-06,
      "loss": 0.0691,
      "step": 13950
    },
    {
      "epoch": 1.3303756713867187e-05,
      "model_forward_time": 1.7375297546386719,
      "step": 13950
    },
    {
      "dynamics_loss_avg": 0.05738499853760004,
      "epoch": 1.3303756713867187e-05,
      "step": 13950
    },
    {
      "action_loss_avg": 0.002124649682082236,
      "epoch": 1.3303756713867187e-05,
      "step": 13950
    },
    {
      "epoch": 1.3303756713867187e-05,
      "step": 13950,
      "training_step_time": 7.430292844772339
    },
    {
      "epoch": 1.3304710388183594e-05,
      "model_forward_time": 1.7384529113769531,
      "step": 13951
    },
    {
      "epoch": 1.3304710388183594e-05,
      "step": 13951,
      "training_step_time": 7.704780340194702
    },
    {
      "epoch": 1.33056640625e-05,
      "model_forward_time": 1.744025707244873,
      "step": 13952
    },
    {
      "epoch": 1.33056640625e-05,
      "step": 13952,
      "training_step_time": 8.511184930801392
    },
    {
      "epoch": 1.3306617736816406e-05,
      "model_forward_time": 1.7396159172058105,
      "step": 13953
    },
    {
      "epoch": 1.3306617736816406e-05,
      "step": 13953,
      "training_step_time": 7.683023691177368
    },
    {
      "epoch": 1.3307571411132813e-05,
      "model_forward_time": 1.7335481643676758,
      "step": 13954
    },
    {
      "epoch": 1.3307571411132813e-05,
      "step": 13954,
      "training_step_time": 8.601096153259277
    },
    {
      "epoch": 1.3308525085449219e-05,
      "model_forward_time": 1.7507679462432861,
      "step": 13955
    },
    {
      "epoch": 1.3308525085449219e-05,
      "step": 13955,
      "training_step_time": 8.539550542831421
    },
    {
      "epoch": 1.3309478759765625e-05,
      "model_forward_time": 1.790832757949829,
      "step": 13956
    },
    {
      "epoch": 1.3309478759765625e-05,
      "step": 13956,
      "training_step_time": 10.17573356628418
    },
    {
      "epoch": 1.3310432434082032e-05,
      "model_forward_time": 1.7375428676605225,
      "step": 13957
    },
    {
      "epoch": 1.3310432434082032e-05,
      "step": 13957,
      "training_step_time": 7.582826137542725
    },
    {
      "epoch": 1.3311386108398438e-05,
      "model_forward_time": 1.7530090808868408,
      "step": 13958
    },
    {
      "epoch": 1.3311386108398438e-05,
      "step": 13958,
      "training_step_time": 7.704012870788574
    },
    {
      "epoch": 1.3312339782714843e-05,
      "model_forward_time": 1.7519164085388184,
      "step": 13959
    },
    {
      "epoch": 1.3312339782714843e-05,
      "step": 13959,
      "training_step_time": 7.3909666538238525
    },
    {
      "epoch": 1.331329345703125e-05,
      "grad_norm": 0.1873590350151062,
      "learning_rate": 9.782162521674838e-06,
      "loss": 0.0659,
      "step": 13960
    },
    {
      "epoch": 1.331329345703125e-05,
      "model_forward_time": 1.7346482276916504,
      "step": 13960
    },
    {
      "dynamics_loss_avg": 0.0558457687497139,
      "epoch": 1.331329345703125e-05,
      "step": 13960
    },
    {
      "action_loss_avg": 0.0026022860198281705,
      "epoch": 1.331329345703125e-05,
      "step": 13960
    },
    {
      "epoch": 1.331329345703125e-05,
      "step": 13960,
      "training_step_time": 7.6736156940460205
    },
    {
      "epoch": 1.3314247131347657e-05,
      "model_forward_time": 1.7347841262817383,
      "step": 13961
    },
    {
      "epoch": 1.3314247131347657e-05,
      "step": 13961,
      "training_step_time": 8.501874685287476
    },
    {
      "epoch": 1.3315200805664062e-05,
      "model_forward_time": 1.7492282390594482,
      "step": 13962
    },
    {
      "epoch": 1.3315200805664062e-05,
      "step": 13962,
      "training_step_time": 7.708452463150024
    },
    {
      "epoch": 1.3316154479980468e-05,
      "model_forward_time": 1.7376248836517334,
      "step": 13963
    },
    {
      "epoch": 1.3316154479980468e-05,
      "step": 13963,
      "training_step_time": 8.699569463729858
    },
    {
      "epoch": 1.3317108154296875e-05,
      "model_forward_time": 1.7382495403289795,
      "step": 13964
    },
    {
      "epoch": 1.3317108154296875e-05,
      "step": 13964,
      "training_step_time": 7.715704917907715
    },
    {
      "epoch": 1.3318061828613281e-05,
      "model_forward_time": 1.7818732261657715,
      "step": 13965
    },
    {
      "epoch": 1.3318061828613281e-05,
      "step": 13965,
      "training_step_time": 7.674258708953857
    },
    {
      "epoch": 1.3319015502929687e-05,
      "model_forward_time": 2.960792064666748,
      "step": 13966
    },
    {
      "epoch": 1.3319015502929687e-05,
      "step": 13966,
      "training_step_time": 8.541591167449951
    },
    {
      "epoch": 1.3319969177246094e-05,
      "model_forward_time": 1.7408394813537598,
      "step": 13967
    },
    {
      "epoch": 1.3319969177246094e-05,
      "step": 13967,
      "training_step_time": 7.707862854003906
    },
    {
      "epoch": 1.33209228515625e-05,
      "model_forward_time": 1.7395367622375488,
      "step": 13968
    },
    {
      "epoch": 1.33209228515625e-05,
      "step": 13968,
      "training_step_time": 6.69909143447876
    },
    {
      "epoch": 1.3321876525878906e-05,
      "model_forward_time": 1.76334810256958,
      "step": 13969
    },
    {
      "epoch": 1.3321876525878906e-05,
      "step": 13969,
      "training_step_time": 7.631299257278442
    },
    {
      "epoch": 1.3322830200195313e-05,
      "grad_norm": 0.21103395521640778,
      "learning_rate": 9.781679523845813e-06,
      "loss": 0.0666,
      "step": 13970
    },
    {
      "epoch": 1.3322830200195313e-05,
      "model_forward_time": 1.7369961738586426,
      "step": 13970
    },
    {
      "dynamics_loss_avg": 0.06623852923512459,
      "epoch": 1.3322830200195313e-05,
      "step": 13970
    },
    {
      "action_loss_avg": 0.0028923685196787117,
      "epoch": 1.3322830200195313e-05,
      "step": 13970
    },
    {
      "epoch": 1.3322830200195313e-05,
      "step": 13970,
      "training_step_time": 7.675577640533447
    },
    {
      "epoch": 1.3323783874511719e-05,
      "model_forward_time": 1.7551143169403076,
      "step": 13971
    },
    {
      "epoch": 1.3323783874511719e-05,
      "step": 13971,
      "training_step_time": 8.498534440994263
    },
    {
      "epoch": 1.3324737548828125e-05,
      "model_forward_time": 1.7379682064056396,
      "step": 13972
    },
    {
      "epoch": 1.3324737548828125e-05,
      "step": 13972,
      "training_step_time": 8.18816351890564
    },
    {
      "epoch": 1.3325691223144532e-05,
      "model_forward_time": 1.3819756507873535,
      "step": 13973
    },
    {
      "epoch": 1.3325691223144532e-05,
      "step": 13973,
      "training_step_time": 8.483228921890259
    },
    {
      "epoch": 1.3326644897460938e-05,
      "model_forward_time": 1.7334554195404053,
      "step": 13974
    },
    {
      "epoch": 1.3326644897460938e-05,
      "step": 13974,
      "training_step_time": 9.886610507965088
    },
    {
      "epoch": 1.3327598571777343e-05,
      "model_forward_time": 2.892895460128784,
      "step": 13975
    },
    {
      "epoch": 1.3327598571777343e-05,
      "step": 13975,
      "training_step_time": 9.066910982131958
    },
    {
      "epoch": 1.332855224609375e-05,
      "model_forward_time": 1.738072156906128,
      "step": 13976
    },
    {
      "epoch": 1.332855224609375e-05,
      "step": 13976,
      "training_step_time": 7.57110333442688
    },
    {
      "epoch": 1.3329505920410157e-05,
      "model_forward_time": 1.0485785007476807,
      "step": 13977
    },
    {
      "epoch": 1.3329505920410157e-05,
      "step": 13977,
      "training_step_time": 7.729355573654175
    },
    {
      "epoch": 1.3330459594726562e-05,
      "model_forward_time": 1.728790283203125,
      "step": 13978
    },
    {
      "epoch": 1.3330459594726562e-05,
      "step": 13978,
      "training_step_time": 7.597796678543091
    },
    {
      "epoch": 1.3331413269042968e-05,
      "model_forward_time": 1.739546537399292,
      "step": 13979
    },
    {
      "epoch": 1.3331413269042968e-05,
      "step": 13979,
      "training_step_time": 8.516887426376343
    },
    {
      "epoch": 1.3332366943359375e-05,
      "grad_norm": 0.12969471514225006,
      "learning_rate": 9.781196003099503e-06,
      "loss": 0.0664,
      "step": 13980
    },
    {
      "epoch": 1.3332366943359375e-05,
      "model_forward_time": 1.7398324012756348,
      "step": 13980
    },
    {
      "dynamics_loss_avg": 0.05949776619672775,
      "epoch": 1.3332366943359375e-05,
      "step": 13980
    },
    {
      "action_loss_avg": 0.003093009698204696,
      "epoch": 1.3332366943359375e-05,
      "step": 13980
    },
    {
      "epoch": 1.3332366943359375e-05,
      "step": 13980,
      "training_step_time": 7.648946523666382
    },
    {
      "epoch": 1.3333320617675781e-05,
      "model_forward_time": 1.7535052299499512,
      "step": 13981
    },
    {
      "epoch": 1.3333320617675781e-05,
      "step": 13981,
      "training_step_time": 9.3764009475708
    },
    {
      "epoch": 1.3334274291992187e-05,
      "model_forward_time": 1.0505099296569824,
      "step": 13982
    },
    {
      "epoch": 1.3334274291992187e-05,
      "step": 13982,
      "training_step_time": 8.511836051940918
    },
    {
      "epoch": 1.3335227966308594e-05,
      "model_forward_time": 1.7423267364501953,
      "step": 13983
    },
    {
      "epoch": 1.3335227966308594e-05,
      "step": 13983,
      "training_step_time": 8.723307609558105
    },
    {
      "epoch": 1.3336181640625e-05,
      "model_forward_time": 1.870558500289917,
      "step": 13984
    },
    {
      "epoch": 1.3336181640625e-05,
      "step": 13984,
      "training_step_time": 7.674809455871582
    },
    {
      "epoch": 1.3337135314941406e-05,
      "model_forward_time": 2.9374871253967285,
      "step": 13985
    },
    {
      "epoch": 1.3337135314941406e-05,
      "step": 13985,
      "training_step_time": 7.770663738250732
    },
    {
      "epoch": 1.3338088989257813e-05,
      "model_forward_time": 1.7443337440490723,
      "step": 13986
    },
    {
      "epoch": 1.3338088989257813e-05,
      "step": 13986,
      "training_step_time": 7.839274644851685
    },
    {
      "epoch": 1.3339042663574219e-05,
      "model_forward_time": 1.767915964126587,
      "step": 13987
    },
    {
      "epoch": 1.3339042663574219e-05,
      "step": 13987,
      "training_step_time": 8.453436613082886
    },
    {
      "epoch": 1.3339996337890625e-05,
      "model_forward_time": 1.7392053604125977,
      "step": 13988
    },
    {
      "epoch": 1.3339996337890625e-05,
      "step": 13988,
      "training_step_time": 7.6749536991119385
    },
    {
      "epoch": 1.3340950012207032e-05,
      "model_forward_time": 1.7790396213531494,
      "step": 13989
    },
    {
      "epoch": 1.3340950012207032e-05,
      "step": 13989,
      "training_step_time": 7.6201043128967285
    },
    {
      "epoch": 1.3341903686523438e-05,
      "grad_norm": 0.1800549179315567,
      "learning_rate": 9.780711959488786e-06,
      "loss": 0.0636,
      "step": 13990
    },
    {
      "epoch": 1.3341903686523438e-05,
      "model_forward_time": 1.7672479152679443,
      "step": 13990
    },
    {
      "dynamics_loss_avg": 0.06000589206814766,
      "epoch": 1.3341903686523438e-05,
      "step": 13990
    },
    {
      "action_loss_avg": 0.002947287727147341,
      "epoch": 1.3341903686523438e-05,
      "step": 13990
    },
    {
      "epoch": 1.3341903686523438e-05,
      "step": 13990,
      "training_step_time": 8.757453680038452
    },
    {
      "epoch": 1.3342857360839843e-05,
      "model_forward_time": 1.754749059677124,
      "step": 13991
    },
    {
      "epoch": 1.3342857360839843e-05,
      "step": 13991,
      "training_step_time": 8.313992738723755
    },
    {
      "epoch": 1.334381103515625e-05,
      "model_forward_time": 1.7503697872161865,
      "step": 13992
    },
    {
      "epoch": 1.334381103515625e-05,
      "step": 13992,
      "training_step_time": 8.319495439529419
    },
    {
      "epoch": 1.3344764709472657e-05,
      "model_forward_time": 1.8406040668487549,
      "step": 13993
    },
    {
      "epoch": 1.3344764709472657e-05,
      "step": 13993,
      "training_step_time": 8.541966915130615
    },
    {
      "epoch": 1.3345718383789062e-05,
      "model_forward_time": 2.967329502105713,
      "step": 13994
    },
    {
      "epoch": 1.3345718383789062e-05,
      "step": 13994,
      "training_step_time": 7.746628284454346
    },
    {
      "epoch": 1.3346672058105468e-05,
      "model_forward_time": 1.7442803382873535,
      "step": 13995
    },
    {
      "epoch": 1.3346672058105468e-05,
      "step": 13995,
      "training_step_time": 7.696525812149048
    },
    {
      "epoch": 1.3347625732421875e-05,
      "model_forward_time": 1.7473530769348145,
      "step": 13996
    },
    {
      "epoch": 1.3347625732421875e-05,
      "step": 13996,
      "training_step_time": 7.404934883117676
    },
    {
      "epoch": 1.3348579406738281e-05,
      "model_forward_time": 1.7586054801940918,
      "step": 13997
    },
    {
      "epoch": 1.3348579406738281e-05,
      "step": 13997,
      "training_step_time": 8.369216203689575
    },
    {
      "epoch": 1.3349533081054687e-05,
      "model_forward_time": 1.827277660369873,
      "step": 13998
    },
    {
      "epoch": 1.3349533081054687e-05,
      "step": 13998,
      "training_step_time": 7.785789251327515
    },
    {
      "epoch": 1.3350486755371094e-05,
      "model_forward_time": 1.7545490264892578,
      "step": 13999
    },
    {
      "epoch": 1.3350486755371094e-05,
      "step": 13999,
      "training_step_time": 8.53801417350769
    },
    {
      "epoch": 1.33514404296875e-05,
      "grad_norm": 0.18745243549346924,
      "learning_rate": 9.780227393066599e-06,
      "loss": 0.067,
      "step": 14000
    },
    {
      "epoch": 1.33514404296875e-05,
      "model_forward_time": 1.3361971378326416,
      "step": 14000
    },
    {
      "dynamics_loss_avg": 0.0718828983604908,
      "epoch": 1.33514404296875e-05,
      "step": 14000
    },
    {
      "action_loss_avg": 0.0027656181424390524,
      "epoch": 1.33514404296875e-05,
      "step": 14000
    },
    {
      "epoch": 1.33514404296875e-05,
      "step": 14000,
      "training_step_time": 11.896597146987915
    },
    {
      "epoch": 1.3352394104003906e-05,
      "model_forward_time": 1.384657859802246,
      "step": 14001
    },
    {
      "epoch": 1.3352394104003906e-05,
      "step": 14001,
      "training_step_time": 8.755348205566406
    },
    {
      "epoch": 1.3353347778320313e-05,
      "model_forward_time": 1.8321447372436523,
      "step": 14002
    },
    {
      "epoch": 1.3353347778320313e-05,
      "step": 14002,
      "training_step_time": 9.614523887634277
    },
    {
      "epoch": 1.3354301452636719e-05,
      "model_forward_time": 3.3686177730560303,
      "step": 14003
    },
    {
      "epoch": 1.3354301452636719e-05,
      "step": 14003,
      "training_step_time": 9.124789953231812
    },
    {
      "epoch": 1.3355255126953125e-05,
      "model_forward_time": 1.7531630992889404,
      "step": 14004
    },
    {
      "epoch": 1.3355255126953125e-05,
      "step": 14004,
      "training_step_time": 6.832279205322266
    },
    {
      "epoch": 1.3356208801269532e-05,
      "model_forward_time": 1.750072717666626,
      "step": 14005
    },
    {
      "epoch": 1.3356208801269532e-05,
      "step": 14005,
      "training_step_time": 8.391894578933716
    },
    {
      "epoch": 1.3357162475585938e-05,
      "model_forward_time": 1.376678466796875,
      "step": 14006
    },
    {
      "epoch": 1.3357162475585938e-05,
      "step": 14006,
      "training_step_time": 7.815969705581665
    },
    {
      "epoch": 1.3358116149902343e-05,
      "model_forward_time": 1.7390563488006592,
      "step": 14007
    },
    {
      "epoch": 1.3358116149902343e-05,
      "step": 14007,
      "training_step_time": 7.680057525634766
    },
    {
      "epoch": 1.335906982421875e-05,
      "model_forward_time": 1.751708984375,
      "step": 14008
    },
    {
      "epoch": 1.335906982421875e-05,
      "step": 14008,
      "training_step_time": 8.195339918136597
    },
    {
      "epoch": 1.3360023498535157e-05,
      "model_forward_time": 1.7430033683776855,
      "step": 14009
    },
    {
      "epoch": 1.3360023498535157e-05,
      "step": 14009,
      "training_step_time": 7.683390855789185
    },
    {
      "epoch": 1.3360977172851562e-05,
      "grad_norm": 0.2818194627761841,
      "learning_rate": 9.77974230388593e-06,
      "loss": 0.0751,
      "step": 14010
    },
    {
      "epoch": 1.3360977172851562e-05,
      "model_forward_time": 1.7452738285064697,
      "step": 14010
    },
    {
      "dynamics_loss_avg": 0.05854304069653153,
      "epoch": 1.3360977172851562e-05,
      "step": 14010
    },
    {
      "action_loss_avg": 0.0024512178119039164,
      "epoch": 1.3360977172851562e-05,
      "step": 14010
    },
    {
      "epoch": 1.3360977172851562e-05,
      "step": 14010,
      "training_step_time": 9.657836437225342
    },
    {
      "epoch": 1.3361930847167968e-05,
      "model_forward_time": 1.8042855262756348,
      "step": 14011
    },
    {
      "epoch": 1.3361930847167968e-05,
      "step": 14011,
      "training_step_time": 8.149552822113037
    },
    {
      "epoch": 1.3362884521484375e-05,
      "model_forward_time": 2.476559638977051,
      "step": 14012
    },
    {
      "epoch": 1.3362884521484375e-05,
      "step": 14012,
      "training_step_time": 7.587118864059448
    },
    {
      "epoch": 1.3363838195800781e-05,
      "model_forward_time": 1.7448227405548096,
      "step": 14013
    },
    {
      "epoch": 1.3363838195800781e-05,
      "step": 14013,
      "training_step_time": 7.5007336139678955
    },
    {
      "epoch": 1.3364791870117187e-05,
      "model_forward_time": 1.745739459991455,
      "step": 14014
    },
    {
      "epoch": 1.3364791870117187e-05,
      "step": 14014,
      "training_step_time": 7.5959532260894775
    },
    {
      "epoch": 1.3365745544433594e-05,
      "model_forward_time": 1.7394740581512451,
      "step": 14015
    },
    {
      "epoch": 1.3365745544433594e-05,
      "step": 14015,
      "training_step_time": 7.63476037979126
    },
    {
      "epoch": 1.336669921875e-05,
      "model_forward_time": 1.7466185092926025,
      "step": 14016
    },
    {
      "epoch": 1.336669921875e-05,
      "step": 14016,
      "training_step_time": 8.531614065170288
    },
    {
      "epoch": 1.3367652893066406e-05,
      "model_forward_time": 1.738429069519043,
      "step": 14017
    },
    {
      "epoch": 1.3367652893066406e-05,
      "step": 14017,
      "training_step_time": 9.813204288482666
    },
    {
      "epoch": 1.3368606567382813e-05,
      "model_forward_time": 1.7314269542694092,
      "step": 14018
    },
    {
      "epoch": 1.3368606567382813e-05,
      "step": 14018,
      "training_step_time": 9.558721542358398
    },
    {
      "epoch": 1.3369560241699219e-05,
      "model_forward_time": 1.380769968032837,
      "step": 14019
    },
    {
      "epoch": 1.3369560241699219e-05,
      "step": 14019,
      "training_step_time": 8.70523977279663
    },
    {
      "epoch": 1.3370513916015625e-05,
      "grad_norm": 0.2411193996667862,
      "learning_rate": 9.77925669199983e-06,
      "loss": 0.0754,
      "step": 14020
    },
    {
      "epoch": 1.3370513916015625e-05,
      "model_forward_time": 1.773611068725586,
      "step": 14020
    },
    {
      "dynamics_loss_avg": 0.07327046729624272,
      "epoch": 1.3370513916015625e-05,
      "step": 14020
    },
    {
      "action_loss_avg": 0.003962778672575951,
      "epoch": 1.3370513916015625e-05,
      "step": 14020
    },
    {
      "epoch": 1.3370513916015625e-05,
      "step": 14020,
      "training_step_time": 7.739948034286499
    },
    {
      "epoch": 1.3371467590332032e-05,
      "model_forward_time": 1.7421526908874512,
      "step": 14021
    },
    {
      "epoch": 1.3371467590332032e-05,
      "step": 14021,
      "training_step_time": 8.256303071975708
    },
    {
      "epoch": 1.3372421264648438e-05,
      "model_forward_time": 2.856066942214966,
      "step": 14022
    },
    {
      "epoch": 1.3372421264648438e-05,
      "step": 14022,
      "training_step_time": 7.677896499633789
    },
    {
      "epoch": 1.3373374938964843e-05,
      "model_forward_time": 1.7433497905731201,
      "step": 14023
    },
    {
      "epoch": 1.3373374938964843e-05,
      "step": 14023,
      "training_step_time": 7.762526988983154
    },
    {
      "epoch": 1.337432861328125e-05,
      "model_forward_time": 1.750509262084961,
      "step": 14024
    },
    {
      "epoch": 1.337432861328125e-05,
      "step": 14024,
      "training_step_time": 7.648029327392578
    },
    {
      "epoch": 1.3375282287597657e-05,
      "model_forward_time": 1.7489030361175537,
      "step": 14025
    },
    {
      "epoch": 1.3375282287597657e-05,
      "step": 14025,
      "training_step_time": 7.697289943695068
    },
    {
      "epoch": 1.3376235961914062e-05,
      "model_forward_time": 1.7441401481628418,
      "step": 14026
    },
    {
      "epoch": 1.3376235961914062e-05,
      "step": 14026,
      "training_step_time": 7.583406686782837
    },
    {
      "epoch": 1.3377189636230468e-05,
      "model_forward_time": 1.7467706203460693,
      "step": 14027
    },
    {
      "epoch": 1.3377189636230468e-05,
      "step": 14027,
      "training_step_time": 7.666501045227051
    },
    {
      "epoch": 1.3378143310546875e-05,
      "model_forward_time": 1.7399625778198242,
      "step": 14028
    },
    {
      "epoch": 1.3378143310546875e-05,
      "step": 14028,
      "training_step_time": 9.58583688735962
    },
    {
      "epoch": 1.3379096984863281e-05,
      "model_forward_time": 1.7409579753875732,
      "step": 14029
    },
    {
      "epoch": 1.3379096984863281e-05,
      "step": 14029,
      "training_step_time": 9.950461387634277
    },
    {
      "epoch": 1.3380050659179687e-05,
      "grad_norm": 0.34988290071487427,
      "learning_rate": 9.778770557461404e-06,
      "loss": 0.0692,
      "step": 14030
    },
    {
      "epoch": 1.3380050659179687e-05,
      "model_forward_time": 1.7479307651519775,
      "step": 14030
    },
    {
      "dynamics_loss_avg": 0.059286443144083024,
      "epoch": 1.3380050659179687e-05,
      "step": 14030
    },
    {
      "action_loss_avg": 0.0033446596702560783,
      "epoch": 1.3380050659179687e-05,
      "step": 14030
    },
    {
      "epoch": 1.3380050659179687e-05,
      "step": 14030,
      "training_step_time": 7.724886655807495
    },
    {
      "epoch": 1.3381004333496094e-05,
      "model_forward_time": 2.8721115589141846,
      "step": 14031
    },
    {
      "epoch": 1.3381004333496094e-05,
      "step": 14031,
      "training_step_time": 7.652903079986572
    },
    {
      "epoch": 1.33819580078125e-05,
      "model_forward_time": 1.7505221366882324,
      "step": 14032
    },
    {
      "epoch": 1.33819580078125e-05,
      "step": 14032,
      "training_step_time": 7.7154481410980225
    },
    {
      "epoch": 1.3382911682128906e-05,
      "model_forward_time": 1.7447190284729004,
      "step": 14033
    },
    {
      "epoch": 1.3382911682128906e-05,
      "step": 14033,
      "training_step_time": 7.649852275848389
    },
    {
      "epoch": 1.3383865356445313e-05,
      "model_forward_time": 1.7377774715423584,
      "step": 14034
    },
    {
      "epoch": 1.3383865356445313e-05,
      "step": 14034,
      "training_step_time": 7.531841993331909
    },
    {
      "epoch": 1.3384819030761719e-05,
      "model_forward_time": 1.7495205402374268,
      "step": 14035
    },
    {
      "epoch": 1.3384819030761719e-05,
      "step": 14035,
      "training_step_time": 7.516197204589844
    },
    {
      "epoch": 1.3385772705078125e-05,
      "model_forward_time": 1.7498791217803955,
      "step": 14036
    },
    {
      "epoch": 1.3385772705078125e-05,
      "step": 14036,
      "training_step_time": 7.718895673751831
    },
    {
      "epoch": 1.3386726379394532e-05,
      "model_forward_time": 1.7480425834655762,
      "step": 14037
    },
    {
      "epoch": 1.3386726379394532e-05,
      "step": 14037,
      "training_step_time": 8.196991682052612
    },
    {
      "epoch": 1.3387680053710938e-05,
      "model_forward_time": 1.7475950717926025,
      "step": 14038
    },
    {
      "epoch": 1.3387680053710938e-05,
      "step": 14038,
      "training_step_time": 8.401659488677979
    },
    {
      "epoch": 1.3388633728027343e-05,
      "model_forward_time": 1.850386142730713,
      "step": 14039
    },
    {
      "epoch": 1.3388633728027343e-05,
      "step": 14039,
      "training_step_time": 8.435504674911499
    },
    {
      "epoch": 1.338958740234375e-05,
      "grad_norm": 0.13317401707172394,
      "learning_rate": 9.778283900323813e-06,
      "loss": 0.0674,
      "step": 14040
    },
    {
      "epoch": 1.338958740234375e-05,
      "model_forward_time": 2.868539571762085,
      "step": 14040
    },
    {
      "dynamics_loss_avg": 0.08016551062464713,
      "epoch": 1.338958740234375e-05,
      "step": 14040
    },
    {
      "action_loss_avg": 0.0050523353042080995,
      "epoch": 1.338958740234375e-05,
      "step": 14040
    },
    {
      "epoch": 1.338958740234375e-05,
      "step": 14040,
      "training_step_time": 7.693535327911377
    },
    {
      "epoch": 1.3390541076660157e-05,
      "model_forward_time": 1.7533080577850342,
      "step": 14041
    },
    {
      "epoch": 1.3390541076660157e-05,
      "step": 14041,
      "training_step_time": 7.613274812698364
    },
    {
      "epoch": 1.3391494750976562e-05,
      "model_forward_time": 1.745265007019043,
      "step": 14042
    },
    {
      "epoch": 1.3391494750976562e-05,
      "step": 14042,
      "training_step_time": 7.6035802364349365
    },
    {
      "epoch": 1.3392448425292968e-05,
      "model_forward_time": 1.7527785301208496,
      "step": 14043
    },
    {
      "epoch": 1.3392448425292968e-05,
      "step": 14043,
      "training_step_time": 8.457693338394165
    },
    {
      "epoch": 1.3393402099609375e-05,
      "model_forward_time": 1.7437989711761475,
      "step": 14044
    },
    {
      "epoch": 1.3393402099609375e-05,
      "step": 14044,
      "training_step_time": 7.501267433166504
    },
    {
      "epoch": 1.3394355773925781e-05,
      "model_forward_time": 1.740708589553833,
      "step": 14045
    },
    {
      "epoch": 1.3394355773925781e-05,
      "step": 14045,
      "training_step_time": 7.619930028915405
    },
    {
      "epoch": 1.3395309448242187e-05,
      "model_forward_time": 1.7499210834503174,
      "step": 14046
    },
    {
      "epoch": 1.3395309448242187e-05,
      "step": 14046,
      "training_step_time": 9.187750101089478
    },
    {
      "epoch": 1.3396263122558594e-05,
      "model_forward_time": 1.7432491779327393,
      "step": 14047
    },
    {
      "epoch": 1.3396263122558594e-05,
      "step": 14047,
      "training_step_time": 10.186333179473877
    },
    {
      "epoch": 1.3397216796875e-05,
      "model_forward_time": 1.8528976440429688,
      "step": 14048
    },
    {
      "epoch": 1.3397216796875e-05,
      "step": 14048,
      "training_step_time": 7.617067575454712
    },
    {
      "epoch": 1.3398170471191406e-05,
      "model_forward_time": 2.8666694164276123,
      "step": 14049
    },
    {
      "epoch": 1.3398170471191406e-05,
      "step": 14049,
      "training_step_time": 7.694855690002441
    },
    {
      "epoch": 1.3399124145507813e-05,
      "grad_norm": 0.3082301914691925,
      "learning_rate": 9.777796720640279e-06,
      "loss": 0.066,
      "step": 14050
    },
    {
      "epoch": 1.3399124145507813e-05,
      "model_forward_time": 1.7533464431762695,
      "step": 14050
    },
    {
      "dynamics_loss_avg": 0.0673204492777586,
      "epoch": 1.3399124145507813e-05,
      "step": 14050
    },
    {
      "action_loss_avg": 0.0039249963941983875,
      "epoch": 1.3399124145507813e-05,
      "step": 14050
    },
    {
      "epoch": 1.3399124145507813e-05,
      "step": 14050,
      "training_step_time": 7.64114236831665
    },
    {
      "epoch": 1.3400077819824219e-05,
      "model_forward_time": 1.7410833835601807,
      "step": 14051
    },
    {
      "epoch": 1.3400077819824219e-05,
      "step": 14051,
      "training_step_time": 7.348475694656372
    },
    {
      "epoch": 1.3401031494140625e-05,
      "model_forward_time": 1.7445571422576904,
      "step": 14052
    },
    {
      "epoch": 1.3401031494140625e-05,
      "step": 14052,
      "training_step_time": 7.659673452377319
    },
    {
      "epoch": 1.3401985168457032e-05,
      "model_forward_time": 1.745927333831787,
      "step": 14053
    },
    {
      "epoch": 1.3401985168457032e-05,
      "step": 14053,
      "training_step_time": 8.599730491638184
    },
    {
      "epoch": 1.3402938842773438e-05,
      "model_forward_time": 1.744534969329834,
      "step": 14054
    },
    {
      "epoch": 1.3402938842773438e-05,
      "step": 14054,
      "training_step_time": 7.613983154296875
    },
    {
      "epoch": 1.3403892517089843e-05,
      "model_forward_time": 1.7459080219268799,
      "step": 14055
    },
    {
      "epoch": 1.3403892517089843e-05,
      "step": 14055,
      "training_step_time": 7.747246980667114
    },
    {
      "epoch": 1.340484619140625e-05,
      "model_forward_time": 1.7480876445770264,
      "step": 14056
    },
    {
      "epoch": 1.340484619140625e-05,
      "step": 14056,
      "training_step_time": 7.6910481452941895
    },
    {
      "epoch": 1.3405799865722657e-05,
      "model_forward_time": 1.8498876094818115,
      "step": 14057
    },
    {
      "epoch": 1.3405799865722657e-05,
      "step": 14057,
      "training_step_time": 8.103596925735474
    },
    {
      "epoch": 1.3406753540039062e-05,
      "model_forward_time": 2.9086406230926514,
      "step": 14058
    },
    {
      "epoch": 1.3406753540039062e-05,
      "step": 14058,
      "training_step_time": 7.657936334609985
    },
    {
      "epoch": 1.3407707214355468e-05,
      "model_forward_time": 1.74318528175354,
      "step": 14059
    },
    {
      "epoch": 1.3407707214355468e-05,
      "step": 14059,
      "training_step_time": 7.582565546035767
    },
    {
      "epoch": 1.3408660888671875e-05,
      "grad_norm": 0.21549367904663086,
      "learning_rate": 9.777309018464078e-06,
      "loss": 0.0649,
      "step": 14060
    },
    {
      "epoch": 1.3408660888671875e-05,
      "model_forward_time": 1.7690255641937256,
      "step": 14060
    },
    {
      "dynamics_loss_avg": 0.06785088367760181,
      "epoch": 1.3408660888671875e-05,
      "step": 14060
    },
    {
      "action_loss_avg": 0.0030316054355353115,
      "epoch": 1.3408660888671875e-05,
      "step": 14060
    },
    {
      "epoch": 1.3408660888671875e-05,
      "step": 14060,
      "training_step_time": 7.236544609069824
    },
    {
      "epoch": 1.3409614562988281e-05,
      "model_forward_time": 1.7484550476074219,
      "step": 14061
    },
    {
      "epoch": 1.3409614562988281e-05,
      "step": 14061,
      "training_step_time": 7.633705377578735
    },
    {
      "epoch": 1.3410568237304687e-05,
      "model_forward_time": 1.7500088214874268,
      "step": 14062
    },
    {
      "epoch": 1.3410568237304687e-05,
      "step": 14062,
      "training_step_time": 7.583990573883057
    },
    {
      "epoch": 1.3411521911621094e-05,
      "model_forward_time": 1.7482953071594238,
      "step": 14063
    },
    {
      "epoch": 1.3411521911621094e-05,
      "step": 14063,
      "training_step_time": 8.072089910507202
    },
    {
      "epoch": 1.34124755859375e-05,
      "model_forward_time": 1.7565977573394775,
      "step": 14064
    },
    {
      "epoch": 1.34124755859375e-05,
      "step": 14064,
      "training_step_time": 8.504272937774658
    },
    {
      "epoch": 1.3413429260253906e-05,
      "model_forward_time": 1.7447590827941895,
      "step": 14065
    },
    {
      "epoch": 1.3413429260253906e-05,
      "step": 14065,
      "training_step_time": 7.686414957046509
    },
    {
      "epoch": 1.3414382934570313e-05,
      "model_forward_time": 1.8726727962493896,
      "step": 14066
    },
    {
      "epoch": 1.3414382934570313e-05,
      "step": 14066,
      "training_step_time": 7.604022026062012
    },
    {
      "epoch": 1.3415336608886719e-05,
      "model_forward_time": 2.9088900089263916,
      "step": 14067
    },
    {
      "epoch": 1.3415336608886719e-05,
      "step": 14067,
      "training_step_time": 8.496330261230469
    },
    {
      "epoch": 1.3416290283203125e-05,
      "model_forward_time": 1.7479627132415771,
      "step": 14068
    },
    {
      "epoch": 1.3416290283203125e-05,
      "step": 14068,
      "training_step_time": 7.547154903411865
    },
    {
      "epoch": 1.3417243957519532e-05,
      "model_forward_time": 1.7495718002319336,
      "step": 14069
    },
    {
      "epoch": 1.3417243957519532e-05,
      "step": 14069,
      "training_step_time": 7.611386299133301
    },
    {
      "epoch": 1.3418197631835938e-05,
      "grad_norm": 0.16367654502391815,
      "learning_rate": 9.776820793848547e-06,
      "loss": 0.0666,
      "step": 14070
    },
    {
      "epoch": 1.3418197631835938e-05,
      "model_forward_time": 1.7519285678863525,
      "step": 14070
    },
    {
      "dynamics_loss_avg": 0.054827280528843406,
      "epoch": 1.3418197631835938e-05,
      "step": 14070
    },
    {
      "action_loss_avg": 0.0037104172690305857,
      "epoch": 1.3418197631835938e-05,
      "step": 14070
    },
    {
      "epoch": 1.3418197631835938e-05,
      "step": 14070,
      "training_step_time": 7.634073257446289
    },
    {
      "epoch": 1.3419151306152343e-05,
      "model_forward_time": 1.7416422367095947,
      "step": 14071
    },
    {
      "epoch": 1.3419151306152343e-05,
      "step": 14071,
      "training_step_time": 7.61548638343811
    },
    {
      "epoch": 1.342010498046875e-05,
      "model_forward_time": 1.7522647380828857,
      "step": 14072
    },
    {
      "epoch": 1.342010498046875e-05,
      "step": 14072,
      "training_step_time": 7.5372583866119385
    },
    {
      "epoch": 1.3421058654785157e-05,
      "model_forward_time": 1.7489500045776367,
      "step": 14073
    },
    {
      "epoch": 1.3421058654785157e-05,
      "step": 14073,
      "training_step_time": 8.477643251419067
    },
    {
      "epoch": 1.3422012329101562e-05,
      "model_forward_time": 1.759401559829712,
      "step": 14074
    },
    {
      "epoch": 1.3422012329101562e-05,
      "step": 14074,
      "training_step_time": 8.236451387405396
    },
    {
      "epoch": 1.3422966003417968e-05,
      "model_forward_time": 1.851393699645996,
      "step": 14075
    },
    {
      "epoch": 1.3422966003417968e-05,
      "step": 14075,
      "training_step_time": 8.28938603401184
    },
    {
      "epoch": 1.3423919677734375e-05,
      "model_forward_time": 2.8879170417785645,
      "step": 14076
    },
    {
      "epoch": 1.3423919677734375e-05,
      "step": 14076,
      "training_step_time": 8.673058271408081
    },
    {
      "epoch": 1.3424873352050781e-05,
      "model_forward_time": 1.7616941928863525,
      "step": 14077
    },
    {
      "epoch": 1.3424873352050781e-05,
      "step": 14077,
      "training_step_time": 7.5503199100494385
    },
    {
      "epoch": 1.3425827026367187e-05,
      "model_forward_time": 1.7607450485229492,
      "step": 14078
    },
    {
      "epoch": 1.3425827026367187e-05,
      "step": 14078,
      "training_step_time": 7.6565587520599365
    },
    {
      "epoch": 1.3426780700683594e-05,
      "model_forward_time": 1.7582728862762451,
      "step": 14079
    },
    {
      "epoch": 1.3426780700683594e-05,
      "step": 14079,
      "training_step_time": 7.600708723068237
    },
    {
      "epoch": 1.3427734375e-05,
      "grad_norm": 0.16397395730018616,
      "learning_rate": 9.776332046847074e-06,
      "loss": 0.0666,
      "step": 14080
    },
    {
      "epoch": 1.3427734375e-05,
      "model_forward_time": 1.7664623260498047,
      "step": 14080
    },
    {
      "dynamics_loss_avg": 0.047759775444865225,
      "epoch": 1.3427734375e-05,
      "step": 14080
    },
    {
      "action_loss_avg": 0.0035890393424779176,
      "epoch": 1.3427734375e-05,
      "step": 14080
    },
    {
      "epoch": 1.3427734375e-05,
      "step": 14080,
      "training_step_time": 7.578597068786621
    },
    {
      "epoch": 1.3428688049316406e-05,
      "model_forward_time": 1.7573370933532715,
      "step": 14081
    },
    {
      "epoch": 1.3428688049316406e-05,
      "step": 14081,
      "training_step_time": 7.527696371078491
    },
    {
      "epoch": 1.3429641723632813e-05,
      "model_forward_time": 1.7545292377471924,
      "step": 14082
    },
    {
      "epoch": 1.3429641723632813e-05,
      "step": 14082,
      "training_step_time": 9.248712301254272
    },
    {
      "epoch": 1.3430595397949219e-05,
      "model_forward_time": 1.4090492725372314,
      "step": 14083
    },
    {
      "epoch": 1.3430595397949219e-05,
      "step": 14083,
      "training_step_time": 7.592775583267212
    },
    {
      "epoch": 1.3431549072265625e-05,
      "model_forward_time": 1.8234236240386963,
      "step": 14084
    },
    {
      "epoch": 1.3431549072265625e-05,
      "step": 14084,
      "training_step_time": 7.674432277679443
    },
    {
      "epoch": 1.3432502746582032e-05,
      "model_forward_time": 2.925290822982788,
      "step": 14085
    },
    {
      "epoch": 1.3432502746582032e-05,
      "step": 14085,
      "training_step_time": 7.645419120788574
    },
    {
      "epoch": 1.3433456420898438e-05,
      "model_forward_time": 1.7629168033599854,
      "step": 14086
    },
    {
      "epoch": 1.3433456420898438e-05,
      "step": 14086,
      "training_step_time": 7.604906320571899
    },
    {
      "epoch": 1.3434410095214844e-05,
      "model_forward_time": 1.7529957294464111,
      "step": 14087
    },
    {
      "epoch": 1.3434410095214844e-05,
      "step": 14087,
      "training_step_time": 7.600473642349243
    },
    {
      "epoch": 1.343536376953125e-05,
      "model_forward_time": 1.761052131652832,
      "step": 14088
    },
    {
      "epoch": 1.343536376953125e-05,
      "step": 14088,
      "training_step_time": 8.524056673049927
    },
    {
      "epoch": 1.3436317443847657e-05,
      "model_forward_time": 1.7642252445220947,
      "step": 14089
    },
    {
      "epoch": 1.3436317443847657e-05,
      "step": 14089,
      "training_step_time": 8.323206424713135
    },
    {
      "epoch": 1.3437271118164062e-05,
      "grad_norm": 0.19580945372581482,
      "learning_rate": 9.775842777513112e-06,
      "loss": 0.0626,
      "step": 14090
    },
    {
      "epoch": 1.3437271118164062e-05,
      "model_forward_time": 1.7771151065826416,
      "step": 14090
    },
    {
      "dynamics_loss_avg": 0.049001206271350385,
      "epoch": 1.3437271118164062e-05,
      "step": 14090
    },
    {
      "action_loss_avg": 0.0034175184206105768,
      "epoch": 1.3437271118164062e-05,
      "step": 14090
    },
    {
      "epoch": 1.3437271118164062e-05,
      "step": 14090,
      "training_step_time": 7.497774362564087
    },
    {
      "epoch": 1.3438224792480468e-05,
      "model_forward_time": 1.7620916366577148,
      "step": 14091
    },
    {
      "epoch": 1.3438224792480468e-05,
      "step": 14091,
      "training_step_time": 8.437650680541992
    },
    {
      "epoch": 1.3439178466796876e-05,
      "model_forward_time": 1.7645933628082275,
      "step": 14092
    },
    {
      "epoch": 1.3439178466796876e-05,
      "step": 14092,
      "training_step_time": 10.316339015960693
    },
    {
      "epoch": 1.3440132141113281e-05,
      "model_forward_time": 1.7595524787902832,
      "step": 14093
    },
    {
      "epoch": 1.3440132141113281e-05,
      "step": 14093,
      "training_step_time": 8.61491084098816
    },
    {
      "epoch": 1.3441085815429687e-05,
      "model_forward_time": 2.962716817855835,
      "step": 14094
    },
    {
      "epoch": 1.3441085815429687e-05,
      "step": 14094,
      "training_step_time": 7.706524848937988
    },
    {
      "epoch": 1.3442039489746094e-05,
      "model_forward_time": 1.732170820236206,
      "step": 14095
    },
    {
      "epoch": 1.3442039489746094e-05,
      "step": 14095,
      "training_step_time": 8.35267162322998
    },
    {
      "epoch": 1.34429931640625e-05,
      "model_forward_time": 1.7586734294891357,
      "step": 14096
    },
    {
      "epoch": 1.34429931640625e-05,
      "step": 14096,
      "training_step_time": 7.632106065750122
    },
    {
      "epoch": 1.3443946838378906e-05,
      "model_forward_time": 1.7522594928741455,
      "step": 14097
    },
    {
      "epoch": 1.3443946838378906e-05,
      "step": 14097,
      "training_step_time": 7.263035774230957
    },
    {
      "epoch": 1.3444900512695313e-05,
      "model_forward_time": 1.747725009918213,
      "step": 14098
    },
    {
      "epoch": 1.3444900512695313e-05,
      "step": 14098,
      "training_step_time": 7.600451469421387
    },
    {
      "epoch": 1.3445854187011719e-05,
      "model_forward_time": 1.756047248840332,
      "step": 14099
    },
    {
      "epoch": 1.3445854187011719e-05,
      "step": 14099,
      "training_step_time": 7.468363285064697
    },
    {
      "epoch": 1.3446807861328125e-05,
      "grad_norm": 0.2226284146308899,
      "learning_rate": 9.775352985900164e-06,
      "loss": 0.0672,
      "step": 14100
    },
    {
      "epoch": 1.3446807861328125e-05,
      "model_forward_time": 1.7638583183288574,
      "step": 14100
    },
    {
      "dynamics_loss_avg": 0.07281178422272205,
      "epoch": 1.3446807861328125e-05,
      "step": 14100
    },
    {
      "action_loss_avg": 0.004521274630678818,
      "epoch": 1.3446807861328125e-05,
      "step": 14100
    },
    {
      "epoch": 1.3446807861328125e-05,
      "step": 14100,
      "training_step_time": 7.6094582080841064
    },
    {
      "epoch": 1.3447761535644532e-05,
      "model_forward_time": 1.7636799812316895,
      "step": 14101
    },
    {
      "epoch": 1.3447761535644532e-05,
      "step": 14101,
      "training_step_time": 8.692783117294312
    },
    {
      "epoch": 1.3448715209960938e-05,
      "model_forward_time": 1.7593047618865967,
      "step": 14102
    },
    {
      "epoch": 1.3448715209960938e-05,
      "step": 14102,
      "training_step_time": 8.214414834976196
    },
    {
      "epoch": 1.3449668884277344e-05,
      "model_forward_time": 2.8919389247894287,
      "step": 14103
    },
    {
      "epoch": 1.3449668884277344e-05,
      "step": 14103,
      "training_step_time": 9.665640592575073
    },
    {
      "epoch": 1.345062255859375e-05,
      "model_forward_time": 1.4074954986572266,
      "step": 14104
    },
    {
      "epoch": 1.345062255859375e-05,
      "step": 14104,
      "training_step_time": 7.604585409164429
    },
    {
      "epoch": 1.3451576232910157e-05,
      "model_forward_time": 1.7676866054534912,
      "step": 14105
    },
    {
      "epoch": 1.3451576232910157e-05,
      "step": 14105,
      "training_step_time": 7.565523147583008
    },
    {
      "epoch": 1.3452529907226562e-05,
      "model_forward_time": 1.7671825885772705,
      "step": 14106
    },
    {
      "epoch": 1.3452529907226562e-05,
      "step": 14106,
      "training_step_time": 7.2742533683776855
    },
    {
      "epoch": 1.3453483581542968e-05,
      "model_forward_time": 1.7612552642822266,
      "step": 14107
    },
    {
      "epoch": 1.3453483581542968e-05,
      "step": 14107,
      "training_step_time": 7.568310976028442
    },
    {
      "epoch": 1.3454437255859376e-05,
      "model_forward_time": 1.7531383037567139,
      "step": 14108
    },
    {
      "epoch": 1.3454437255859376e-05,
      "step": 14108,
      "training_step_time": 7.401514053344727
    },
    {
      "epoch": 1.3455390930175781e-05,
      "model_forward_time": 1.3775489330291748,
      "step": 14109
    },
    {
      "epoch": 1.3455390930175781e-05,
      "step": 14109,
      "training_step_time": 7.526339054107666
    },
    {
      "epoch": 1.3456344604492187e-05,
      "grad_norm": 0.3030591607093811,
      "learning_rate": 9.774862672061791e-06,
      "loss": 0.0685,
      "step": 14110
    },
    {
      "epoch": 1.3456344604492187e-05,
      "model_forward_time": 1.7718760967254639,
      "step": 14110
    },
    {
      "dynamics_loss_avg": 0.06344649661332369,
      "epoch": 1.3456344604492187e-05,
      "step": 14110
    },
    {
      "action_loss_avg": 0.004015117057133466,
      "epoch": 1.3456344604492187e-05,
      "step": 14110
    },
    {
      "epoch": 1.3456344604492187e-05,
      "step": 14110,
      "training_step_time": 7.65716814994812
    },
    {
      "epoch": 1.3457298278808594e-05,
      "model_forward_time": 1.7911877632141113,
      "step": 14111
    },
    {
      "epoch": 1.3457298278808594e-05,
      "step": 14111,
      "training_step_time": 9.473000764846802
    },
    {
      "epoch": 1.3458251953125e-05,
      "model_forward_time": 2.9120490550994873,
      "step": 14112
    },
    {
      "epoch": 1.3458251953125e-05,
      "step": 14112,
      "training_step_time": 7.639249563217163
    },
    {
      "epoch": 1.3459205627441406e-05,
      "model_forward_time": 1.7559847831726074,
      "step": 14113
    },
    {
      "epoch": 1.3459205627441406e-05,
      "step": 14113,
      "training_step_time": 7.648804426193237
    },
    {
      "epoch": 1.3460159301757813e-05,
      "model_forward_time": 1.756669044494629,
      "step": 14114
    },
    {
      "epoch": 1.3460159301757813e-05,
      "step": 14114,
      "training_step_time": 7.549242734909058
    },
    {
      "epoch": 1.3461112976074219e-05,
      "model_forward_time": 1.7673897743225098,
      "step": 14115
    },
    {
      "epoch": 1.3461112976074219e-05,
      "step": 14115,
      "training_step_time": 7.571518182754517
    },
    {
      "epoch": 1.3462066650390625e-05,
      "model_forward_time": 1.7598531246185303,
      "step": 14116
    },
    {
      "epoch": 1.3462066650390625e-05,
      "step": 14116,
      "training_step_time": 7.649008512496948
    },
    {
      "epoch": 1.3463020324707032e-05,
      "model_forward_time": 1.7559666633605957,
      "step": 14117
    },
    {
      "epoch": 1.3463020324707032e-05,
      "step": 14117,
      "training_step_time": 7.405842304229736
    },
    {
      "epoch": 1.3463973999023438e-05,
      "model_forward_time": 1.7622616291046143,
      "step": 14118
    },
    {
      "epoch": 1.3463973999023438e-05,
      "step": 14118,
      "training_step_time": 7.5473668575286865
    },
    {
      "epoch": 1.3464927673339844e-05,
      "model_forward_time": 1.7843546867370605,
      "step": 14119
    },
    {
      "epoch": 1.3464927673339844e-05,
      "step": 14119,
      "training_step_time": 7.641820430755615
    },
    {
      "epoch": 1.346588134765625e-05,
      "grad_norm": 0.15615780651569366,
      "learning_rate": 9.774371836051616e-06,
      "loss": 0.0662,
      "step": 14120
    },
    {
      "epoch": 1.346588134765625e-05,
      "model_forward_time": 1.776317834854126,
      "step": 14120
    },
    {
      "dynamics_loss_avg": 0.05348484106361866,
      "epoch": 1.346588134765625e-05,
      "step": 14120
    },
    {
      "action_loss_avg": 0.0027577711036428807,
      "epoch": 1.346588134765625e-05,
      "step": 14120
    },
    {
      "epoch": 1.346588134765625e-05,
      "step": 14120,
      "training_step_time": 8.65193486213684
    },
    {
      "epoch": 1.3466835021972657e-05,
      "model_forward_time": 1.7663211822509766,
      "step": 14121
    },
    {
      "epoch": 1.3466835021972657e-05,
      "step": 14121,
      "training_step_time": 9.157593011856079
    },
    {
      "epoch": 1.3467788696289062e-05,
      "model_forward_time": 2.282536506652832,
      "step": 14122
    },
    {
      "epoch": 1.3467788696289062e-05,
      "step": 14122,
      "training_step_time": 7.636066675186157
    },
    {
      "epoch": 1.3468742370605468e-05,
      "model_forward_time": 1.7082812786102295,
      "step": 14123
    },
    {
      "epoch": 1.3468742370605468e-05,
      "step": 14123,
      "training_step_time": 8.409796714782715
    },
    {
      "epoch": 1.3469696044921876e-05,
      "model_forward_time": 1.7091984748840332,
      "step": 14124
    },
    {
      "epoch": 1.3469696044921876e-05,
      "step": 14124,
      "training_step_time": 7.6517229080200195
    },
    {
      "epoch": 1.3470649719238281e-05,
      "model_forward_time": 1.7250080108642578,
      "step": 14125
    },
    {
      "epoch": 1.3470649719238281e-05,
      "step": 14125,
      "training_step_time": 7.646903038024902
    },
    {
      "epoch": 1.3471603393554687e-05,
      "model_forward_time": 1.708165168762207,
      "step": 14126
    },
    {
      "epoch": 1.3471603393554687e-05,
      "step": 14126,
      "training_step_time": 7.351987838745117
    },
    {
      "epoch": 1.3472557067871094e-05,
      "model_forward_time": 1.7076585292816162,
      "step": 14127
    },
    {
      "epoch": 1.3472557067871094e-05,
      "step": 14127,
      "training_step_time": 7.577421426773071
    },
    {
      "epoch": 1.34735107421875e-05,
      "model_forward_time": 1.712740182876587,
      "step": 14128
    },
    {
      "epoch": 1.34735107421875e-05,
      "step": 14128,
      "training_step_time": 8.223570823669434
    },
    {
      "epoch": 1.3474464416503906e-05,
      "model_forward_time": 1.7074015140533447,
      "step": 14129
    },
    {
      "epoch": 1.3474464416503906e-05,
      "step": 14129,
      "training_step_time": 9.46851396560669
    },
    {
      "epoch": 1.3475418090820313e-05,
      "grad_norm": 0.26307111978530884,
      "learning_rate": 9.773880477923316e-06,
      "loss": 0.0641,
      "step": 14130
    },
    {
      "epoch": 1.3475418090820313e-05,
      "model_forward_time": 1.7169508934020996,
      "step": 14130
    },
    {
      "dynamics_loss_avg": 0.05976803908124566,
      "epoch": 1.3475418090820313e-05,
      "step": 14130
    },
    {
      "action_loss_avg": 0.004113131924532354,
      "epoch": 1.3475418090820313e-05,
      "step": 14130
    },
    {
      "epoch": 1.3475418090820313e-05,
      "step": 14130,
      "training_step_time": 7.68360447883606
    },
    {
      "epoch": 1.3476371765136719e-05,
      "model_forward_time": 2.2778875827789307,
      "step": 14131
    },
    {
      "epoch": 1.3476371765136719e-05,
      "step": 14131,
      "training_step_time": 7.6013970375061035
    },
    {
      "epoch": 1.3477325439453125e-05,
      "model_forward_time": 1.7007453441619873,
      "step": 14132
    },
    {
      "epoch": 1.3477325439453125e-05,
      "step": 14132,
      "training_step_time": 8.44737720489502
    },
    {
      "epoch": 1.347827911376953e-05,
      "model_forward_time": 1.6954569816589355,
      "step": 14133
    },
    {
      "epoch": 1.347827911376953e-05,
      "step": 14133,
      "training_step_time": 7.647511005401611
    },
    {
      "epoch": 1.3479232788085938e-05,
      "model_forward_time": 1.7025461196899414,
      "step": 14134
    },
    {
      "epoch": 1.3479232788085938e-05,
      "step": 14134,
      "training_step_time": 7.631436586380005
    },
    {
      "epoch": 1.3480186462402344e-05,
      "model_forward_time": 1.7076265811920166,
      "step": 14135
    },
    {
      "epoch": 1.3480186462402344e-05,
      "step": 14135,
      "training_step_time": 7.290499925613403
    },
    {
      "epoch": 1.348114013671875e-05,
      "model_forward_time": 1.7064533233642578,
      "step": 14136
    },
    {
      "epoch": 1.348114013671875e-05,
      "step": 14136,
      "training_step_time": 8.347970008850098
    },
    {
      "epoch": 1.3482093811035157e-05,
      "model_forward_time": 1.7050917148590088,
      "step": 14137
    },
    {
      "epoch": 1.3482093811035157e-05,
      "step": 14137,
      "training_step_time": 7.634597063064575
    },
    {
      "epoch": 1.3483047485351562e-05,
      "model_forward_time": 1.7059657573699951,
      "step": 14138
    },
    {
      "epoch": 1.3483047485351562e-05,
      "step": 14138,
      "training_step_time": 7.641224384307861
    },
    {
      "epoch": 1.3484001159667968e-05,
      "model_forward_time": 1.7109136581420898,
      "step": 14139
    },
    {
      "epoch": 1.3484001159667968e-05,
      "step": 14139,
      "training_step_time": 9.110458850860596
    },
    {
      "epoch": 1.3484954833984376e-05,
      "grad_norm": 0.2888999879360199,
      "learning_rate": 9.773388597730623e-06,
      "loss": 0.0668,
      "step": 14140
    },
    {
      "epoch": 1.3484954833984376e-05,
      "model_forward_time": 2.245669364929199,
      "step": 14140
    },
    {
      "dynamics_loss_avg": 0.07659456394612789,
      "epoch": 1.3484954833984376e-05,
      "step": 14140
    },
    {
      "action_loss_avg": 0.005290716385934502,
      "epoch": 1.3484954833984376e-05,
      "step": 14140
    },
    {
      "epoch": 1.3484954833984376e-05,
      "step": 14140,
      "training_step_time": 7.616411447525024
    },
    {
      "epoch": 1.3485908508300781e-05,
      "model_forward_time": 1.7086982727050781,
      "step": 14141
    },
    {
      "epoch": 1.3485908508300781e-05,
      "step": 14141,
      "training_step_time": 7.612812757492065
    },
    {
      "epoch": 1.3486862182617187e-05,
      "model_forward_time": 1.709446907043457,
      "step": 14142
    },
    {
      "epoch": 1.3486862182617187e-05,
      "step": 14142,
      "training_step_time": 7.691539287567139
    },
    {
      "epoch": 1.3487815856933594e-05,
      "model_forward_time": 1.7040379047393799,
      "step": 14143
    },
    {
      "epoch": 1.3487815856933594e-05,
      "step": 14143,
      "training_step_time": 8.618101119995117
    },
    {
      "epoch": 1.348876953125e-05,
      "model_forward_time": 1.7300693988800049,
      "step": 14144
    },
    {
      "epoch": 1.348876953125e-05,
      "step": 14144,
      "training_step_time": 7.6164140701293945
    },
    {
      "epoch": 1.3489723205566406e-05,
      "model_forward_time": 1.7088367938995361,
      "step": 14145
    },
    {
      "epoch": 1.3489723205566406e-05,
      "step": 14145,
      "training_step_time": 8.547343969345093
    },
    {
      "epoch": 1.3490676879882813e-05,
      "model_forward_time": 1.7015514373779297,
      "step": 14146
    },
    {
      "epoch": 1.3490676879882813e-05,
      "step": 14146,
      "training_step_time": 7.692253351211548
    },
    {
      "epoch": 1.3491630554199219e-05,
      "model_forward_time": 1.7061305046081543,
      "step": 14147
    },
    {
      "epoch": 1.3491630554199219e-05,
      "step": 14147,
      "training_step_time": 8.667028665542603
    },
    {
      "epoch": 1.3492584228515625e-05,
      "model_forward_time": 1.7040224075317383,
      "step": 14148
    },
    {
      "epoch": 1.3492584228515625e-05,
      "step": 14148,
      "training_step_time": 7.703048467636108
    },
    {
      "epoch": 1.349353790283203e-05,
      "model_forward_time": 1.8847770690917969,
      "step": 14149
    },
    {
      "epoch": 1.349353790283203e-05,
      "step": 14149,
      "training_step_time": 8.411672592163086
    },
    {
      "epoch": 1.3494491577148438e-05,
      "grad_norm": 0.22259478271007538,
      "learning_rate": 9.77289619552733e-06,
      "loss": 0.0674,
      "step": 14150
    },
    {
      "epoch": 1.3494491577148438e-05,
      "model_forward_time": 1.7000701427459717,
      "step": 14150
    },
    {
      "dynamics_loss_avg": 0.056986639462411406,
      "epoch": 1.3494491577148438e-05,
      "step": 14150
    },
    {
      "action_loss_avg": 0.003566542168846354,
      "epoch": 1.3494491577148438e-05,
      "step": 14150
    },
    {
      "epoch": 1.3494491577148438e-05,
      "step": 14150,
      "training_step_time": 7.565876483917236
    },
    {
      "epoch": 1.3495445251464844e-05,
      "model_forward_time": 1.689737319946289,
      "step": 14151
    },
    {
      "epoch": 1.3495445251464844e-05,
      "step": 14151,
      "training_step_time": 7.616793394088745
    },
    {
      "epoch": 1.349639892578125e-05,
      "model_forward_time": 1.7118186950683594,
      "step": 14152
    },
    {
      "epoch": 1.349639892578125e-05,
      "step": 14152,
      "training_step_time": 7.565250635147095
    },
    {
      "epoch": 1.3497352600097657e-05,
      "model_forward_time": 1.7309894561767578,
      "step": 14153
    },
    {
      "epoch": 1.3497352600097657e-05,
      "step": 14153,
      "training_step_time": 7.6313395500183105
    },
    {
      "epoch": 1.3498306274414062e-05,
      "model_forward_time": 1.7040317058563232,
      "step": 14154
    },
    {
      "epoch": 1.3498306274414062e-05,
      "step": 14154,
      "training_step_time": 7.373489141464233
    },
    {
      "epoch": 1.3499259948730468e-05,
      "model_forward_time": 1.706613302230835,
      "step": 14155
    },
    {
      "epoch": 1.3499259948730468e-05,
      "step": 14155,
      "training_step_time": 7.655534505844116
    },
    {
      "epoch": 1.3500213623046876e-05,
      "model_forward_time": 1.7074854373931885,
      "step": 14156
    },
    {
      "epoch": 1.3500213623046876e-05,
      "step": 14156,
      "training_step_time": 7.526920318603516
    },
    {
      "epoch": 1.3501167297363281e-05,
      "model_forward_time": 1.711543083190918,
      "step": 14157
    },
    {
      "epoch": 1.3501167297363281e-05,
      "step": 14157,
      "training_step_time": 8.616517066955566
    },
    {
      "epoch": 1.3502120971679687e-05,
      "model_forward_time": 2.2346956729888916,
      "step": 14158
    },
    {
      "epoch": 1.3502120971679687e-05,
      "step": 14158,
      "training_step_time": 7.629601001739502
    },
    {
      "epoch": 1.3503074645996094e-05,
      "model_forward_time": 1.7090530395507812,
      "step": 14159
    },
    {
      "epoch": 1.3503074645996094e-05,
      "step": 14159,
      "training_step_time": 8.391212224960327
    },
    {
      "epoch": 1.35040283203125e-05,
      "grad_norm": 0.22508111596107483,
      "learning_rate": 9.772403271367286e-06,
      "loss": 0.0679,
      "step": 14160
    },
    {
      "epoch": 1.35040283203125e-05,
      "model_forward_time": 1.7001595497131348,
      "step": 14160
    },
    {
      "dynamics_loss_avg": 0.06182884154841304,
      "epoch": 1.35040283203125e-05,
      "step": 14160
    },
    {
      "action_loss_avg": 0.0049153782427310945,
      "epoch": 1.35040283203125e-05,
      "step": 14160
    },
    {
      "epoch": 1.35040283203125e-05,
      "step": 14160,
      "training_step_time": 7.481928110122681
    },
    {
      "epoch": 1.3504981994628906e-05,
      "model_forward_time": 1.7095601558685303,
      "step": 14161
    },
    {
      "epoch": 1.3504981994628906e-05,
      "step": 14161,
      "training_step_time": 8.685611009597778
    },
    {
      "epoch": 1.3505935668945313e-05,
      "model_forward_time": 1.719062328338623,
      "step": 14162
    },
    {
      "epoch": 1.3505935668945313e-05,
      "step": 14162,
      "training_step_time": 7.637357473373413
    },
    {
      "epoch": 1.3506889343261719e-05,
      "model_forward_time": 1.7085974216461182,
      "step": 14163
    },
    {
      "epoch": 1.3506889343261719e-05,
      "step": 14163,
      "training_step_time": 8.210619926452637
    },
    {
      "epoch": 1.3507843017578125e-05,
      "model_forward_time": 1.7097280025482178,
      "step": 14164
    },
    {
      "epoch": 1.3507843017578125e-05,
      "step": 14164,
      "training_step_time": 7.671911001205444
    },
    {
      "epoch": 1.350879669189453e-05,
      "model_forward_time": 1.7032673358917236,
      "step": 14165
    },
    {
      "epoch": 1.350879669189453e-05,
      "step": 14165,
      "training_step_time": 8.478713989257812
    },
    {
      "epoch": 1.3509750366210938e-05,
      "model_forward_time": 1.7071926593780518,
      "step": 14166
    },
    {
      "epoch": 1.3509750366210938e-05,
      "step": 14166,
      "training_step_time": 7.66364312171936
    },
    {
      "epoch": 1.3510704040527344e-05,
      "model_forward_time": 1.7061693668365479,
      "step": 14167
    },
    {
      "epoch": 1.3510704040527344e-05,
      "step": 14167,
      "training_step_time": 8.223489046096802
    },
    {
      "epoch": 1.351165771484375e-05,
      "model_forward_time": 2.2588422298431396,
      "step": 14168
    },
    {
      "epoch": 1.351165771484375e-05,
      "step": 14168,
      "training_step_time": 7.529707431793213
    },
    {
      "epoch": 1.3512611389160157e-05,
      "model_forward_time": 1.7164342403411865,
      "step": 14169
    },
    {
      "epoch": 1.3512611389160157e-05,
      "step": 14169,
      "training_step_time": 7.486481189727783
    },
    {
      "epoch": 1.3513565063476562e-05,
      "grad_norm": 0.28114813566207886,
      "learning_rate": 9.771909825304396e-06,
      "loss": 0.0706,
      "step": 14170
    },
    {
      "epoch": 1.3513565063476562e-05,
      "model_forward_time": 1.318286657333374,
      "step": 14170
    },
    {
      "dynamics_loss_avg": 0.06475712768733502,
      "epoch": 1.3513565063476562e-05,
      "step": 14170
    },
    {
      "action_loss_avg": 0.003323164995526895,
      "epoch": 1.3513565063476562e-05,
      "step": 14170
    },
    {
      "epoch": 1.3513565063476562e-05,
      "step": 14170,
      "training_step_time": 7.6000120639801025
    },
    {
      "epoch": 1.3514518737792968e-05,
      "model_forward_time": 1.71944260597229,
      "step": 14171
    },
    {
      "epoch": 1.3514518737792968e-05,
      "step": 14171,
      "training_step_time": 7.634166479110718
    },
    {
      "epoch": 1.3515472412109376e-05,
      "model_forward_time": 1.7112579345703125,
      "step": 14172
    },
    {
      "epoch": 1.3515472412109376e-05,
      "step": 14172,
      "training_step_time": 7.360403776168823
    },
    {
      "epoch": 1.3516426086425781e-05,
      "model_forward_time": 1.7073466777801514,
      "step": 14173
    },
    {
      "epoch": 1.3516426086425781e-05,
      "step": 14173,
      "training_step_time": 8.625625371932983
    },
    {
      "epoch": 1.3517379760742187e-05,
      "model_forward_time": 1.7069497108459473,
      "step": 14174
    },
    {
      "epoch": 1.3517379760742187e-05,
      "step": 14174,
      "training_step_time": 8.194824457168579
    },
    {
      "epoch": 1.3518333435058594e-05,
      "model_forward_time": 1.7183403968811035,
      "step": 14175
    },
    {
      "epoch": 1.3518333435058594e-05,
      "step": 14175,
      "training_step_time": 8.573035955429077
    },
    {
      "epoch": 1.3519287109375e-05,
      "model_forward_time": 1.7038726806640625,
      "step": 14176
    },
    {
      "epoch": 1.3519287109375e-05,
      "step": 14176,
      "training_step_time": 9.06964659690857
    },
    {
      "epoch": 1.3520240783691406e-05,
      "model_forward_time": 2.2516837120056152,
      "step": 14177
    },
    {
      "epoch": 1.3520240783691406e-05,
      "step": 14177,
      "training_step_time": 8.084360122680664
    },
    {
      "epoch": 1.3521194458007813e-05,
      "model_forward_time": 1.7017951011657715,
      "step": 14178
    },
    {
      "epoch": 1.3521194458007813e-05,
      "step": 14178,
      "training_step_time": 7.530340909957886
    },
    {
      "epoch": 1.3522148132324219e-05,
      "model_forward_time": 1.7083473205566406,
      "step": 14179
    },
    {
      "epoch": 1.3522148132324219e-05,
      "step": 14179,
      "training_step_time": 7.59591007232666
    },
    {
      "epoch": 1.3523101806640625e-05,
      "grad_norm": 0.38714927434921265,
      "learning_rate": 9.771415857392618e-06,
      "loss": 0.0704,
      "step": 14180
    },
    {
      "epoch": 1.3523101806640625e-05,
      "model_forward_time": 1.7174720764160156,
      "step": 14180
    },
    {
      "dynamics_loss_avg": 0.07099501378834247,
      "epoch": 1.3523101806640625e-05,
      "step": 14180
    },
    {
      "action_loss_avg": 0.004479836183600127,
      "epoch": 1.3523101806640625e-05,
      "step": 14180
    },
    {
      "epoch": 1.3523101806640625e-05,
      "step": 14180,
      "training_step_time": 8.56625247001648
    },
    {
      "epoch": 1.352405548095703e-05,
      "model_forward_time": 1.6992368698120117,
      "step": 14181
    },
    {
      "epoch": 1.352405548095703e-05,
      "step": 14181,
      "training_step_time": 7.362239122390747
    },
    {
      "epoch": 1.3525009155273438e-05,
      "model_forward_time": 1.7086830139160156,
      "step": 14182
    },
    {
      "epoch": 1.3525009155273438e-05,
      "step": 14182,
      "training_step_time": 7.710617542266846
    },
    {
      "epoch": 1.3525962829589844e-05,
      "model_forward_time": 1.6982204914093018,
      "step": 14183
    },
    {
      "epoch": 1.3525962829589844e-05,
      "step": 14183,
      "training_step_time": 7.759371042251587
    },
    {
      "epoch": 1.352691650390625e-05,
      "model_forward_time": 1.7027645111083984,
      "step": 14184
    },
    {
      "epoch": 1.352691650390625e-05,
      "step": 14184,
      "training_step_time": 7.5598304271698
    },
    {
      "epoch": 1.3527870178222657e-05,
      "model_forward_time": 1.3254995346069336,
      "step": 14185
    },
    {
      "epoch": 1.3527870178222657e-05,
      "step": 14185,
      "training_step_time": 8.230985164642334
    },
    {
      "epoch": 1.3528823852539062e-05,
      "model_forward_time": 1.8394992351531982,
      "step": 14186
    },
    {
      "epoch": 1.3528823852539062e-05,
      "step": 14186,
      "training_step_time": 7.609236478805542
    },
    {
      "epoch": 1.3529777526855468e-05,
      "model_forward_time": 1.7009203433990479,
      "step": 14187
    },
    {
      "epoch": 1.3529777526855468e-05,
      "step": 14187,
      "training_step_time": 7.535161972045898
    },
    {
      "epoch": 1.3530731201171876e-05,
      "model_forward_time": 1.7104947566986084,
      "step": 14188
    },
    {
      "epoch": 1.3530731201171876e-05,
      "step": 14188,
      "training_step_time": 8.413794994354248
    },
    {
      "epoch": 1.3531684875488281e-05,
      "model_forward_time": 1.733363151550293,
      "step": 14189
    },
    {
      "epoch": 1.3531684875488281e-05,
      "step": 14189,
      "training_step_time": 7.592727899551392
    },
    {
      "epoch": 1.3532638549804687e-05,
      "grad_norm": 0.3457420766353607,
      "learning_rate": 9.77092136768598e-06,
      "loss": 0.0641,
      "step": 14190
    },
    {
      "epoch": 1.3532638549804687e-05,
      "model_forward_time": 1.6949503421783447,
      "step": 14190
    },
    {
      "dynamics_loss_avg": 0.06171528249979019,
      "epoch": 1.3532638549804687e-05,
      "step": 14190
    },
    {
      "action_loss_avg": 0.005314014514442533,
      "epoch": 1.3532638549804687e-05,
      "step": 14190
    },
    {
      "epoch": 1.3532638549804687e-05,
      "step": 14190,
      "training_step_time": 7.620248556137085
    },
    {
      "epoch": 1.3533592224121094e-05,
      "model_forward_time": 1.707880973815918,
      "step": 14191
    },
    {
      "epoch": 1.3533592224121094e-05,
      "step": 14191,
      "training_step_time": 9.449511051177979
    },
    {
      "epoch": 1.35345458984375e-05,
      "model_forward_time": 1.7027549743652344,
      "step": 14192
    },
    {
      "epoch": 1.35345458984375e-05,
      "step": 14192,
      "training_step_time": 7.939199447631836
    },
    {
      "epoch": 1.3535499572753906e-05,
      "model_forward_time": 1.711745262145996,
      "step": 14193
    },
    {
      "epoch": 1.3535499572753906e-05,
      "step": 14193,
      "training_step_time": 8.860301733016968
    },
    {
      "epoch": 1.3536453247070313e-05,
      "model_forward_time": 1.7029860019683838,
      "step": 14194
    },
    {
      "epoch": 1.3536453247070313e-05,
      "step": 14194,
      "training_step_time": 9.75276803970337
    },
    {
      "epoch": 1.3537406921386719e-05,
      "model_forward_time": 1.7141175270080566,
      "step": 14195
    },
    {
      "epoch": 1.3537406921386719e-05,
      "step": 14195,
      "training_step_time": 8.899773120880127
    },
    {
      "epoch": 1.3538360595703125e-05,
      "model_forward_time": 2.233941078186035,
      "step": 14196
    },
    {
      "epoch": 1.3538360595703125e-05,
      "step": 14196,
      "training_step_time": 7.738751649856567
    },
    {
      "epoch": 1.353931427001953e-05,
      "model_forward_time": 1.7650949954986572,
      "step": 14197
    },
    {
      "epoch": 1.353931427001953e-05,
      "step": 14197,
      "training_step_time": 7.740583419799805
    },
    {
      "epoch": 1.3540267944335938e-05,
      "model_forward_time": 1.395007610321045,
      "step": 14198
    },
    {
      "epoch": 1.3540267944335938e-05,
      "step": 14198,
      "training_step_time": 7.72853684425354
    },
    {
      "epoch": 1.3541221618652344e-05,
      "model_forward_time": 1.7595856189727783,
      "step": 14199
    },
    {
      "epoch": 1.3541221618652344e-05,
      "step": 14199,
      "training_step_time": 7.582469701766968
    },
    {
      "epoch": 1.354217529296875e-05,
      "grad_norm": 0.2895437479019165,
      "learning_rate": 9.77042635623855e-06,
      "loss": 0.0689,
      "step": 14200
    },
    {
      "epoch": 1.354217529296875e-05,
      "model_forward_time": 1.762328863143921,
      "step": 14200
    },
    {
      "dynamics_loss_avg": 0.08512815460562706,
      "epoch": 1.354217529296875e-05,
      "step": 14200
    },
    {
      "action_loss_avg": 0.005124594154767692,
      "epoch": 1.354217529296875e-05,
      "step": 14200
    },
    {
      "epoch": 1.354217529296875e-05,
      "step": 14200,
      "training_step_time": 8.362709999084473
    },
    {
      "epoch": 1.3543128967285157e-05,
      "model_forward_time": 1.7706341743469238,
      "step": 14201
    },
    {
      "epoch": 1.3543128967285157e-05,
      "step": 14201,
      "training_step_time": 7.7001283168792725
    },
    {
      "epoch": 1.3544082641601562e-05,
      "model_forward_time": 1.7538092136383057,
      "step": 14202
    },
    {
      "epoch": 1.3544082641601562e-05,
      "step": 14202,
      "training_step_time": 7.664015769958496
    },
    {
      "epoch": 1.3545036315917968e-05,
      "model_forward_time": 1.7676477432250977,
      "step": 14203
    },
    {
      "epoch": 1.3545036315917968e-05,
      "step": 14203,
      "training_step_time": 10.72499680519104
    },
    {
      "epoch": 1.3545989990234376e-05,
      "model_forward_time": 1.7834768295288086,
      "step": 14204
    },
    {
      "epoch": 1.3545989990234376e-05,
      "step": 14204,
      "training_step_time": 8.306552648544312
    },
    {
      "epoch": 1.3546943664550781e-05,
      "model_forward_time": 2.9016783237457275,
      "step": 14205
    },
    {
      "epoch": 1.3546943664550781e-05,
      "step": 14205,
      "training_step_time": 8.606292009353638
    },
    {
      "epoch": 1.3547897338867187e-05,
      "model_forward_time": 1.7617926597595215,
      "step": 14206
    },
    {
      "epoch": 1.3547897338867187e-05,
      "step": 14206,
      "training_step_time": 7.516538143157959
    },
    {
      "epoch": 1.3548851013183594e-05,
      "model_forward_time": 1.75677490234375,
      "step": 14207
    },
    {
      "epoch": 1.3548851013183594e-05,
      "step": 14207,
      "training_step_time": 7.605180025100708
    },
    {
      "epoch": 1.35498046875e-05,
      "model_forward_time": 1.7545170783996582,
      "step": 14208
    },
    {
      "epoch": 1.35498046875e-05,
      "step": 14208,
      "training_step_time": 7.307434558868408
    },
    {
      "epoch": 1.3550758361816406e-05,
      "model_forward_time": 1.7518315315246582,
      "step": 14209
    },
    {
      "epoch": 1.3550758361816406e-05,
      "step": 14209,
      "training_step_time": 7.529740571975708
    },
    {
      "epoch": 1.3551712036132813e-05,
      "grad_norm": 0.27550485730171204,
      "learning_rate": 9.76993082310447e-06,
      "loss": 0.0639,
      "step": 14210
    },
    {
      "epoch": 1.3551712036132813e-05,
      "model_forward_time": 1.7624998092651367,
      "step": 14210
    },
    {
      "dynamics_loss_avg": 0.07225389927625656,
      "epoch": 1.3551712036132813e-05,
      "step": 14210
    },
    {
      "action_loss_avg": 0.004748831497272477,
      "epoch": 1.3551712036132813e-05,
      "step": 14210
    },
    {
      "epoch": 1.3551712036132813e-05,
      "step": 14210,
      "training_step_time": 7.600810527801514
    },
    {
      "epoch": 1.3552665710449219e-05,
      "model_forward_time": 1.7599740028381348,
      "step": 14211
    },
    {
      "epoch": 1.3552665710449219e-05,
      "step": 14211,
      "training_step_time": 8.127911567687988
    },
    {
      "epoch": 1.3553619384765625e-05,
      "model_forward_time": 1.7631089687347412,
      "step": 14212
    },
    {
      "epoch": 1.3553619384765625e-05,
      "step": 14212,
      "training_step_time": 7.588452339172363
    },
    {
      "epoch": 1.355457305908203e-05,
      "model_forward_time": 1.7579808235168457,
      "step": 14213
    },
    {
      "epoch": 1.355457305908203e-05,
      "step": 14213,
      "training_step_time": 8.72076153755188
    },
    {
      "epoch": 1.3555526733398438e-05,
      "model_forward_time": 2.9460361003875732,
      "step": 14214
    },
    {
      "epoch": 1.3555526733398438e-05,
      "step": 14214,
      "training_step_time": 7.677417755126953
    },
    {
      "epoch": 1.3556480407714844e-05,
      "model_forward_time": 1.7215659618377686,
      "step": 14215
    },
    {
      "epoch": 1.3556480407714844e-05,
      "step": 14215,
      "training_step_time": 7.591572284698486
    },
    {
      "epoch": 1.355743408203125e-05,
      "model_forward_time": 1.7459437847137451,
      "step": 14216
    },
    {
      "epoch": 1.355743408203125e-05,
      "step": 14216,
      "training_step_time": 7.599803686141968
    },
    {
      "epoch": 1.3558387756347657e-05,
      "model_forward_time": 1.7529408931732178,
      "step": 14217
    },
    {
      "epoch": 1.3558387756347657e-05,
      "step": 14217,
      "training_step_time": 8.32957649230957
    },
    {
      "epoch": 1.3559341430664062e-05,
      "model_forward_time": 1.7436728477478027,
      "step": 14218
    },
    {
      "epoch": 1.3559341430664062e-05,
      "step": 14218,
      "training_step_time": 8.253723382949829
    },
    {
      "epoch": 1.3560295104980468e-05,
      "model_forward_time": 1.7515990734100342,
      "step": 14219
    },
    {
      "epoch": 1.3560295104980468e-05,
      "step": 14219,
      "training_step_time": 7.606678247451782
    },
    {
      "epoch": 1.3561248779296876e-05,
      "grad_norm": 0.2864522933959961,
      "learning_rate": 9.769434768337926e-06,
      "loss": 0.0664,
      "step": 14220
    },
    {
      "epoch": 1.3561248779296876e-05,
      "model_forward_time": 1.7510993480682373,
      "step": 14220
    },
    {
      "dynamics_loss_avg": 0.06182005014270544,
      "epoch": 1.3561248779296876e-05,
      "step": 14220
    },
    {
      "action_loss_avg": 0.0036421250086277725,
      "epoch": 1.3561248779296876e-05,
      "step": 14220
    },
    {
      "epoch": 1.3561248779296876e-05,
      "step": 14220,
      "training_step_time": 10.314823627471924
    },
    {
      "epoch": 1.3562202453613281e-05,
      "model_forward_time": 1.7541532516479492,
      "step": 14221
    },
    {
      "epoch": 1.3562202453613281e-05,
      "step": 14221,
      "training_step_time": 8.679678201675415
    },
    {
      "epoch": 1.3563156127929687e-05,
      "model_forward_time": 1.3839471340179443,
      "step": 14222
    },
    {
      "epoch": 1.3563156127929687e-05,
      "step": 14222,
      "training_step_time": 7.661996364593506
    },
    {
      "epoch": 1.3564109802246094e-05,
      "model_forward_time": 1.8553752899169922,
      "step": 14223
    },
    {
      "epoch": 1.3564109802246094e-05,
      "step": 14223,
      "training_step_time": 9.089306592941284
    },
    {
      "epoch": 1.35650634765625e-05,
      "model_forward_time": 1.7504308223724365,
      "step": 14224
    },
    {
      "epoch": 1.35650634765625e-05,
      "step": 14224,
      "training_step_time": 7.540095806121826
    },
    {
      "epoch": 1.3566017150878906e-05,
      "model_forward_time": 1.7507891654968262,
      "step": 14225
    },
    {
      "epoch": 1.3566017150878906e-05,
      "step": 14225,
      "training_step_time": 8.436917543411255
    },
    {
      "epoch": 1.3566970825195313e-05,
      "model_forward_time": 1.7554314136505127,
      "step": 14226
    },
    {
      "epoch": 1.3566970825195313e-05,
      "step": 14226,
      "training_step_time": 7.593494176864624
    },
    {
      "epoch": 1.3567924499511719e-05,
      "model_forward_time": 1.7547080516815186,
      "step": 14227
    },
    {
      "epoch": 1.3567924499511719e-05,
      "step": 14227,
      "training_step_time": 7.596397638320923
    },
    {
      "epoch": 1.3568878173828125e-05,
      "model_forward_time": 1.7532618045806885,
      "step": 14228
    },
    {
      "epoch": 1.3568878173828125e-05,
      "step": 14228,
      "training_step_time": 9.201272964477539
    },
    {
      "epoch": 1.356983184814453e-05,
      "model_forward_time": 1.7672953605651855,
      "step": 14229
    },
    {
      "epoch": 1.356983184814453e-05,
      "step": 14229,
      "training_step_time": 8.3723623752594
    },
    {
      "epoch": 1.3570785522460938e-05,
      "grad_norm": 0.21734800934791565,
      "learning_rate": 9.768938191993165e-06,
      "loss": 0.0644,
      "step": 14230
    },
    {
      "epoch": 1.3570785522460938e-05,
      "model_forward_time": 1.7666990756988525,
      "step": 14230
    },
    {
      "dynamics_loss_avg": 0.054852941259741786,
      "epoch": 1.3570785522460938e-05,
      "step": 14230
    },
    {
      "action_loss_avg": 0.0029748948640190066,
      "epoch": 1.3570785522460938e-05,
      "step": 14230
    },
    {
      "epoch": 1.3570785522460938e-05,
      "step": 14230,
      "training_step_time": 8.307590007781982
    },
    {
      "epoch": 1.3571739196777344e-05,
      "model_forward_time": 1.7683470249176025,
      "step": 14231
    },
    {
      "epoch": 1.3571739196777344e-05,
      "step": 14231,
      "training_step_time": 8.229437112808228
    },
    {
      "epoch": 1.357269287109375e-05,
      "model_forward_time": 1.8505809307098389,
      "step": 14232
    },
    {
      "epoch": 1.357269287109375e-05,
      "step": 14232,
      "training_step_time": 7.566519260406494
    },
    {
      "epoch": 1.3573646545410157e-05,
      "model_forward_time": 2.9249439239501953,
      "step": 14233
    },
    {
      "epoch": 1.3573646545410157e-05,
      "step": 14233,
      "training_step_time": 8.675173044204712
    },
    {
      "epoch": 1.3574600219726562e-05,
      "model_forward_time": 1.7624666690826416,
      "step": 14234
    },
    {
      "epoch": 1.3574600219726562e-05,
      "step": 14234,
      "training_step_time": 7.534230947494507
    },
    {
      "epoch": 1.3575553894042968e-05,
      "model_forward_time": 1.770956039428711,
      "step": 14235
    },
    {
      "epoch": 1.3575553894042968e-05,
      "step": 14235,
      "training_step_time": 7.61687707901001
    },
    {
      "epoch": 1.3576507568359376e-05,
      "model_forward_time": 1.761225938796997,
      "step": 14236
    },
    {
      "epoch": 1.3576507568359376e-05,
      "step": 14236,
      "training_step_time": 7.563703298568726
    },
    {
      "epoch": 1.3577461242675781e-05,
      "model_forward_time": 1.7575178146362305,
      "step": 14237
    },
    {
      "epoch": 1.3577461242675781e-05,
      "step": 14237,
      "training_step_time": 8.462256669998169
    },
    {
      "epoch": 1.3578414916992187e-05,
      "model_forward_time": 1.7711431980133057,
      "step": 14238
    },
    {
      "epoch": 1.3578414916992187e-05,
      "step": 14238,
      "training_step_time": 8.619415521621704
    },
    {
      "epoch": 1.3579368591308594e-05,
      "model_forward_time": 1.7554519176483154,
      "step": 14239
    },
    {
      "epoch": 1.3579368591308594e-05,
      "step": 14239,
      "training_step_time": 8.721348762512207
    },
    {
      "epoch": 1.3580322265625e-05,
      "grad_norm": 0.198981374502182,
      "learning_rate": 9.768441094124495e-06,
      "loss": 0.0656,
      "step": 14240
    },
    {
      "epoch": 1.3580322265625e-05,
      "model_forward_time": 1.758948564529419,
      "step": 14240
    },
    {
      "dynamics_loss_avg": 0.04868187112733722,
      "epoch": 1.3580322265625e-05,
      "step": 14240
    },
    {
      "action_loss_avg": 0.0022371887287590654,
      "epoch": 1.3580322265625e-05,
      "step": 14240
    },
    {
      "epoch": 1.3580322265625e-05,
      "step": 14240,
      "training_step_time": 8.529511451721191
    },
    {
      "epoch": 1.3581275939941406e-05,
      "model_forward_time": 1.816359281539917,
      "step": 14241
    },
    {
      "epoch": 1.3581275939941406e-05,
      "step": 14241,
      "training_step_time": 8.14348554611206
    },
    {
      "epoch": 1.3582229614257813e-05,
      "model_forward_time": 2.9486491680145264,
      "step": 14242
    },
    {
      "epoch": 1.3582229614257813e-05,
      "step": 14242,
      "training_step_time": 7.641542911529541
    },
    {
      "epoch": 1.3583183288574219e-05,
      "model_forward_time": 1.7666113376617432,
      "step": 14243
    },
    {
      "epoch": 1.3583183288574219e-05,
      "step": 14243,
      "training_step_time": 8.04194974899292
    },
    {
      "epoch": 1.3584136962890625e-05,
      "model_forward_time": 1.7587177753448486,
      "step": 14244
    },
    {
      "epoch": 1.3584136962890625e-05,
      "step": 14244,
      "training_step_time": 7.556252956390381
    },
    {
      "epoch": 1.358509063720703e-05,
      "model_forward_time": 1.760246753692627,
      "step": 14245
    },
    {
      "epoch": 1.358509063720703e-05,
      "step": 14245,
      "training_step_time": 7.491230249404907
    },
    {
      "epoch": 1.3586044311523438e-05,
      "model_forward_time": 1.7580935955047607,
      "step": 14246
    },
    {
      "epoch": 1.3586044311523438e-05,
      "step": 14246,
      "training_step_time": 7.555641174316406
    },
    {
      "epoch": 1.3586997985839844e-05,
      "model_forward_time": 1.7670907974243164,
      "step": 14247
    },
    {
      "epoch": 1.3586997985839844e-05,
      "step": 14247,
      "training_step_time": 7.609670400619507
    },
    {
      "epoch": 1.358795166015625e-05,
      "model_forward_time": 1.770815372467041,
      "step": 14248
    },
    {
      "epoch": 1.358795166015625e-05,
      "step": 14248,
      "training_step_time": 7.6333534717559814
    },
    {
      "epoch": 1.3588905334472657e-05,
      "model_forward_time": 1.7724745273590088,
      "step": 14249
    },
    {
      "epoch": 1.3588905334472657e-05,
      "step": 14249,
      "training_step_time": 8.175989866256714
    },
    {
      "epoch": 1.3589859008789062e-05,
      "grad_norm": 0.21036531031131744,
      "learning_rate": 9.767943474786274e-06,
      "loss": 0.0682,
      "step": 14250
    },
    {
      "epoch": 1.3589859008789062e-05,
      "model_forward_time": 1.805377721786499,
      "step": 14250
    },
    {
      "dynamics_loss_avg": 0.08063051924109459,
      "epoch": 1.3589859008789062e-05,
      "step": 14250
    },
    {
      "action_loss_avg": 0.005233414215035736,
      "epoch": 1.3589859008789062e-05,
      "step": 14250
    },
    {
      "epoch": 1.3589859008789062e-05,
      "step": 14250,
      "training_step_time": 7.5674824714660645
    },
    {
      "epoch": 1.3590812683105468e-05,
      "model_forward_time": 2.9028513431549072,
      "step": 14251
    },
    {
      "epoch": 1.3590812683105468e-05,
      "step": 14251,
      "training_step_time": 7.594965219497681
    },
    {
      "epoch": 1.3591766357421876e-05,
      "model_forward_time": 1.765742540359497,
      "step": 14252
    },
    {
      "epoch": 1.3591766357421876e-05,
      "step": 14252,
      "training_step_time": 7.524564981460571
    },
    {
      "epoch": 1.3592720031738281e-05,
      "model_forward_time": 1.7571525573730469,
      "step": 14253
    },
    {
      "epoch": 1.3592720031738281e-05,
      "step": 14253,
      "training_step_time": 7.616391658782959
    },
    {
      "epoch": 1.3593673706054687e-05,
      "model_forward_time": 1.7610070705413818,
      "step": 14254
    },
    {
      "epoch": 1.3593673706054687e-05,
      "step": 14254,
      "training_step_time": 8.26814079284668
    },
    {
      "epoch": 1.3594627380371094e-05,
      "model_forward_time": 1.764355182647705,
      "step": 14255
    },
    {
      "epoch": 1.3594627380371094e-05,
      "step": 14255,
      "training_step_time": 7.45302414894104
    },
    {
      "epoch": 1.35955810546875e-05,
      "model_forward_time": 1.7579045295715332,
      "step": 14256
    },
    {
      "epoch": 1.35955810546875e-05,
      "step": 14256,
      "training_step_time": 7.615783452987671
    },
    {
      "epoch": 1.3596534729003906e-05,
      "model_forward_time": 1.7668232917785645,
      "step": 14257
    },
    {
      "epoch": 1.3596534729003906e-05,
      "step": 14257,
      "training_step_time": 7.658470392227173
    },
    {
      "epoch": 1.3597488403320312e-05,
      "model_forward_time": 1.400193452835083,
      "step": 14258
    },
    {
      "epoch": 1.3597488403320312e-05,
      "step": 14258,
      "training_step_time": 9.748632431030273
    },
    {
      "epoch": 1.3598442077636719e-05,
      "model_forward_time": 1.769730806350708,
      "step": 14259
    },
    {
      "epoch": 1.3598442077636719e-05,
      "step": 14259,
      "training_step_time": 9.608731031417847
    },
    {
      "epoch": 1.3599395751953125e-05,
      "grad_norm": 0.19390691816806793,
      "learning_rate": 9.767445334032924e-06,
      "loss": 0.0676,
      "step": 14260
    },
    {
      "epoch": 1.3599395751953125e-05,
      "model_forward_time": 2.8979334831237793,
      "step": 14260
    },
    {
      "dynamics_loss_avg": 0.060758470371365544,
      "epoch": 1.3599395751953125e-05,
      "step": 14260
    },
    {
      "action_loss_avg": 0.004540905263274908,
      "epoch": 1.3599395751953125e-05,
      "step": 14260
    },
    {
      "epoch": 1.3599395751953125e-05,
      "step": 14260,
      "training_step_time": 7.651199102401733
    },
    {
      "epoch": 1.360034942626953e-05,
      "model_forward_time": 1.7572553157806396,
      "step": 14261
    },
    {
      "epoch": 1.360034942626953e-05,
      "step": 14261,
      "training_step_time": 7.562800645828247
    },
    {
      "epoch": 1.3601303100585938e-05,
      "model_forward_time": 1.7587831020355225,
      "step": 14262
    },
    {
      "epoch": 1.3601303100585938e-05,
      "step": 14262,
      "training_step_time": 8.67758846282959
    },
    {
      "epoch": 1.3602256774902344e-05,
      "model_forward_time": 1.7679150104522705,
      "step": 14263
    },
    {
      "epoch": 1.3602256774902344e-05,
      "step": 14263,
      "training_step_time": 7.314932823181152
    },
    {
      "epoch": 1.360321044921875e-05,
      "model_forward_time": 1.7675063610076904,
      "step": 14264
    },
    {
      "epoch": 1.360321044921875e-05,
      "step": 14264,
      "training_step_time": 7.547931432723999
    },
    {
      "epoch": 1.3604164123535157e-05,
      "model_forward_time": 1.7615408897399902,
      "step": 14265
    },
    {
      "epoch": 1.3604164123535157e-05,
      "step": 14265,
      "training_step_time": 8.179190635681152
    },
    {
      "epoch": 1.3605117797851563e-05,
      "model_forward_time": 1.7722156047821045,
      "step": 14266
    },
    {
      "epoch": 1.3605117797851563e-05,
      "step": 14266,
      "training_step_time": 8.440356492996216
    },
    {
      "epoch": 1.3606071472167968e-05,
      "model_forward_time": 1.767345666885376,
      "step": 14267
    },
    {
      "epoch": 1.3606071472167968e-05,
      "step": 14267,
      "training_step_time": 8.789397954940796
    },
    {
      "epoch": 1.3607025146484376e-05,
      "model_forward_time": 1.7627530097961426,
      "step": 14268
    },
    {
      "epoch": 1.3607025146484376e-05,
      "step": 14268,
      "training_step_time": 8.724768161773682
    },
    {
      "epoch": 1.3607978820800781e-05,
      "model_forward_time": 2.946467399597168,
      "step": 14269
    },
    {
      "epoch": 1.3607978820800781e-05,
      "step": 14269,
      "training_step_time": 7.787643909454346
    },
    {
      "epoch": 1.3608932495117187e-05,
      "grad_norm": 0.22659149765968323,
      "learning_rate": 9.76694667191892e-06,
      "loss": 0.0698,
      "step": 14270
    },
    {
      "epoch": 1.3608932495117187e-05,
      "model_forward_time": 1.7482781410217285,
      "step": 14270
    },
    {
      "dynamics_loss_avg": 0.0640125822275877,
      "epoch": 1.3608932495117187e-05,
      "step": 14270
    },
    {
      "action_loss_avg": 0.002979656250681728,
      "epoch": 1.3608932495117187e-05,
      "step": 14270
    },
    {
      "epoch": 1.3608932495117187e-05,
      "step": 14270,
      "training_step_time": 7.681328058242798
    },
    {
      "epoch": 1.3609886169433595e-05,
      "model_forward_time": 1.7818317413330078,
      "step": 14271
    },
    {
      "epoch": 1.3609886169433595e-05,
      "step": 14271,
      "training_step_time": 7.732112407684326
    },
    {
      "epoch": 1.361083984375e-05,
      "model_forward_time": 1.7631745338439941,
      "step": 14272
    },
    {
      "epoch": 1.361083984375e-05,
      "step": 14272,
      "training_step_time": 7.108340740203857
    },
    {
      "epoch": 1.3611793518066406e-05,
      "model_forward_time": 1.7612347602844238,
      "step": 14273
    },
    {
      "epoch": 1.3611793518066406e-05,
      "step": 14273,
      "training_step_time": 7.703262805938721
    },
    {
      "epoch": 1.3612747192382812e-05,
      "model_forward_time": 1.7594208717346191,
      "step": 14274
    },
    {
      "epoch": 1.3612747192382812e-05,
      "step": 14274,
      "training_step_time": 8.29325270652771
    },
    {
      "epoch": 1.3613700866699219e-05,
      "model_forward_time": 1.7627408504486084,
      "step": 14275
    },
    {
      "epoch": 1.3613700866699219e-05,
      "step": 14275,
      "training_step_time": 8.34045934677124
    },
    {
      "epoch": 1.3614654541015625e-05,
      "model_forward_time": 1.766890525817871,
      "step": 14276
    },
    {
      "epoch": 1.3614654541015625e-05,
      "step": 14276,
      "training_step_time": 8.81337022781372
    },
    {
      "epoch": 1.361560821533203e-05,
      "model_forward_time": 1.760249137878418,
      "step": 14277
    },
    {
      "epoch": 1.361560821533203e-05,
      "step": 14277,
      "training_step_time": 7.688415765762329
    },
    {
      "epoch": 1.3616561889648438e-05,
      "model_forward_time": 1.8371143341064453,
      "step": 14278
    },
    {
      "epoch": 1.3616561889648438e-05,
      "step": 14278,
      "training_step_time": 8.524576902389526
    },
    {
      "epoch": 1.3617515563964844e-05,
      "model_forward_time": 2.892943859100342,
      "step": 14279
    },
    {
      "epoch": 1.3617515563964844e-05,
      "step": 14279,
      "training_step_time": 7.736940860748291
    },
    {
      "epoch": 1.361846923828125e-05,
      "grad_norm": 0.1819038987159729,
      "learning_rate": 9.766447488498795e-06,
      "loss": 0.0689,
      "step": 14280
    },
    {
      "epoch": 1.361846923828125e-05,
      "model_forward_time": 1.7560474872589111,
      "step": 14280
    },
    {
      "dynamics_loss_avg": 0.06689221877604723,
      "epoch": 1.361846923828125e-05,
      "step": 14280
    },
    {
      "action_loss_avg": 0.004578385868808254,
      "epoch": 1.361846923828125e-05,
      "step": 14280
    },
    {
      "epoch": 1.361846923828125e-05,
      "step": 14280,
      "training_step_time": 7.69606614112854
    },
    {
      "epoch": 1.3619422912597657e-05,
      "model_forward_time": 1.3936219215393066,
      "step": 14281
    },
    {
      "epoch": 1.3619422912597657e-05,
      "step": 14281,
      "training_step_time": 7.231715679168701
    },
    {
      "epoch": 1.3620376586914063e-05,
      "model_forward_time": 1.7606008052825928,
      "step": 14282
    },
    {
      "epoch": 1.3620376586914063e-05,
      "step": 14282,
      "training_step_time": 7.650352716445923
    },
    {
      "epoch": 1.3621330261230468e-05,
      "model_forward_time": 1.7662806510925293,
      "step": 14283
    },
    {
      "epoch": 1.3621330261230468e-05,
      "step": 14283,
      "training_step_time": 7.721935510635376
    },
    {
      "epoch": 1.3622283935546876e-05,
      "model_forward_time": 1.4015116691589355,
      "step": 14284
    },
    {
      "epoch": 1.3622283935546876e-05,
      "step": 14284,
      "training_step_time": 7.7308878898620605
    },
    {
      "epoch": 1.3623237609863281e-05,
      "model_forward_time": 1.7775564193725586,
      "step": 14285
    },
    {
      "epoch": 1.3623237609863281e-05,
      "step": 14285,
      "training_step_time": 8.855817079544067
    },
    {
      "epoch": 1.3624191284179687e-05,
      "model_forward_time": 1.7639060020446777,
      "step": 14286
    },
    {
      "epoch": 1.3624191284179687e-05,
      "step": 14286,
      "training_step_time": 8.469287157058716
    },
    {
      "epoch": 1.3625144958496095e-05,
      "model_forward_time": 1.8087682723999023,
      "step": 14287
    },
    {
      "epoch": 1.3625144958496095e-05,
      "step": 14287,
      "training_step_time": 8.299304723739624
    },
    {
      "epoch": 1.36260986328125e-05,
      "model_forward_time": 2.919131278991699,
      "step": 14288
    },
    {
      "epoch": 1.36260986328125e-05,
      "step": 14288,
      "training_step_time": 8.727818012237549
    },
    {
      "epoch": 1.3627052307128906e-05,
      "model_forward_time": 1.761904239654541,
      "step": 14289
    },
    {
      "epoch": 1.3627052307128906e-05,
      "step": 14289,
      "training_step_time": 8.751526594161987
    },
    {
      "epoch": 1.3628005981445312e-05,
      "grad_norm": 0.20761552453041077,
      "learning_rate": 9.765947783827138e-06,
      "loss": 0.0625,
      "step": 14290
    },
    {
      "epoch": 1.3628005981445312e-05,
      "model_forward_time": 1.7561523914337158,
      "step": 14290
    },
    {
      "dynamics_loss_avg": 0.05500722527503967,
      "epoch": 1.3628005981445312e-05,
      "step": 14290
    },
    {
      "action_loss_avg": 0.0029373355908319356,
      "epoch": 1.3628005981445312e-05,
      "step": 14290
    },
    {
      "epoch": 1.3628005981445312e-05,
      "step": 14290,
      "training_step_time": 7.706599950790405
    },
    {
      "epoch": 1.3628959655761719e-05,
      "model_forward_time": 1.7633662223815918,
      "step": 14291
    },
    {
      "epoch": 1.3628959655761719e-05,
      "step": 14291,
      "training_step_time": 7.653259992599487
    },
    {
      "epoch": 1.3629913330078125e-05,
      "model_forward_time": 1.391261100769043,
      "step": 14292
    },
    {
      "epoch": 1.3629913330078125e-05,
      "step": 14292,
      "training_step_time": 7.735352993011475
    },
    {
      "epoch": 1.363086700439453e-05,
      "model_forward_time": 1.7400805950164795,
      "step": 14293
    },
    {
      "epoch": 1.363086700439453e-05,
      "step": 14293,
      "training_step_time": 7.786312818527222
    },
    {
      "epoch": 1.3631820678710938e-05,
      "model_forward_time": 1.763733148574829,
      "step": 14294
    },
    {
      "epoch": 1.3631820678710938e-05,
      "step": 14294,
      "training_step_time": 8.663995265960693
    },
    {
      "epoch": 1.3632774353027344e-05,
      "model_forward_time": 1.7659738063812256,
      "step": 14295
    },
    {
      "epoch": 1.3632774353027344e-05,
      "step": 14295,
      "training_step_time": 7.723574876785278
    },
    {
      "epoch": 1.363372802734375e-05,
      "model_forward_time": 1.7826554775238037,
      "step": 14296
    },
    {
      "epoch": 1.363372802734375e-05,
      "step": 14296,
      "training_step_time": 7.789067983627319
    },
    {
      "epoch": 1.3634681701660157e-05,
      "model_forward_time": 2.867427110671997,
      "step": 14297
    },
    {
      "epoch": 1.3634681701660157e-05,
      "step": 14297,
      "training_step_time": 8.733133554458618
    },
    {
      "epoch": 1.3635635375976563e-05,
      "model_forward_time": 1.758976697921753,
      "step": 14298
    },
    {
      "epoch": 1.3635635375976563e-05,
      "step": 14298,
      "training_step_time": 8.506655216217041
    },
    {
      "epoch": 1.3636589050292968e-05,
      "model_forward_time": 1.757875919342041,
      "step": 14299
    },
    {
      "epoch": 1.3636589050292968e-05,
      "step": 14299,
      "training_step_time": 8.908764123916626
    },
    {
      "epoch": 1.3637542724609376e-05,
      "grad_norm": 0.2046966850757599,
      "learning_rate": 9.7654475579586e-06,
      "loss": 0.0692,
      "step": 14300
    },
    {
      "epoch": 1.3637542724609376e-05,
      "model_forward_time": 1.7619433403015137,
      "step": 14300
    },
    {
      "dynamics_loss_avg": 0.06937025710940362,
      "epoch": 1.3637542724609376e-05,
      "step": 14300
    },
    {
      "action_loss_avg": 0.003482678788714111,
      "epoch": 1.3637542724609376e-05,
      "step": 14300
    },
    {
      "epoch": 1.3637542724609376e-05,
      "step": 14300,
      "training_step_time": 8.815020084381104
    },
    {
      "epoch": 1.3638496398925781e-05,
      "model_forward_time": 1.755112648010254,
      "step": 14301
    },
    {
      "epoch": 1.3638496398925781e-05,
      "step": 14301,
      "training_step_time": 6.887385606765747
    },
    {
      "epoch": 1.3639450073242187e-05,
      "model_forward_time": 1.7621338367462158,
      "step": 14302
    },
    {
      "epoch": 1.3639450073242187e-05,
      "step": 14302,
      "training_step_time": 6.882818937301636
    },
    {
      "epoch": 1.3640403747558595e-05,
      "model_forward_time": 1.7764830589294434,
      "step": 14303
    },
    {
      "epoch": 1.3640403747558595e-05,
      "step": 14303,
      "training_step_time": 6.865518569946289
    },
    {
      "epoch": 1.3641357421875e-05,
      "model_forward_time": 1.7606453895568848,
      "step": 14304
    },
    {
      "epoch": 1.3641357421875e-05,
      "step": 14304,
      "training_step_time": 6.862128019332886
    },
    {
      "epoch": 1.3642311096191406e-05,
      "model_forward_time": 1.7565202713012695,
      "step": 14305
    },
    {
      "epoch": 1.3642311096191406e-05,
      "step": 14305,
      "training_step_time": 6.91006326675415
    },
    {
      "epoch": 1.3643264770507812e-05,
      "model_forward_time": 1.7593324184417725,
      "step": 14306
    },
    {
      "epoch": 1.3643264770507812e-05,
      "step": 14306,
      "training_step_time": 6.856450796127319
    },
    {
      "epoch": 1.3644218444824219e-05,
      "model_forward_time": 1.7263553142547607,
      "step": 14307
    },
    {
      "epoch": 1.3644218444824219e-05,
      "step": 14307,
      "training_step_time": 6.834186792373657
    },
    {
      "epoch": 1.3645172119140625e-05,
      "model_forward_time": 1.7398669719696045,
      "step": 14308
    },
    {
      "epoch": 1.3645172119140625e-05,
      "step": 14308,
      "training_step_time": 7.371058702468872
    },
    {
      "epoch": 1.364612579345703e-05,
      "model_forward_time": 1.836503267288208,
      "step": 14309
    },
    {
      "epoch": 1.364612579345703e-05,
      "step": 14309,
      "training_step_time": 7.728486061096191
    },
    {
      "epoch": 1.3647079467773438e-05,
      "grad_norm": 0.25805914402008057,
      "learning_rate": 9.76494681094788e-06,
      "loss": 0.069,
      "step": 14310
    },
    {
      "epoch": 1.3647079467773438e-05,
      "model_forward_time": 1.7275049686431885,
      "step": 14310
    },
    {
      "dynamics_loss_avg": 0.07791887372732162,
      "epoch": 1.3647079467773438e-05,
      "step": 14310
    },
    {
      "action_loss_avg": 0.003035685943905264,
      "epoch": 1.3647079467773438e-05,
      "step": 14310
    },
    {
      "epoch": 1.3647079467773438e-05,
      "step": 14310,
      "training_step_time": 7.9051597118377686
    },
    {
      "epoch": 1.3648033142089844e-05,
      "model_forward_time": 1.7340333461761475,
      "step": 14311
    },
    {
      "epoch": 1.3648033142089844e-05,
      "step": 14311,
      "training_step_time": 6.8180670738220215
    },
    {
      "epoch": 1.364898681640625e-05,
      "model_forward_time": 1.7259588241577148,
      "step": 14312
    },
    {
      "epoch": 1.364898681640625e-05,
      "step": 14312,
      "training_step_time": 6.791696310043335
    },
    {
      "epoch": 1.3649940490722657e-05,
      "model_forward_time": 1.7474510669708252,
      "step": 14313
    },
    {
      "epoch": 1.3649940490722657e-05,
      "step": 14313,
      "training_step_time": 6.816972255706787
    },
    {
      "epoch": 1.3650894165039063e-05,
      "model_forward_time": 1.7276217937469482,
      "step": 14314
    },
    {
      "epoch": 1.3650894165039063e-05,
      "step": 14314,
      "training_step_time": 6.788207292556763
    },
    {
      "epoch": 1.3651847839355468e-05,
      "model_forward_time": 1.7452337741851807,
      "step": 14315
    },
    {
      "epoch": 1.3651847839355468e-05,
      "step": 14315,
      "training_step_time": 6.784687519073486
    },
    {
      "epoch": 1.3652801513671876e-05,
      "model_forward_time": 1.7348556518554688,
      "step": 14316
    },
    {
      "epoch": 1.3652801513671876e-05,
      "step": 14316,
      "training_step_time": 6.795552730560303
    },
    {
      "epoch": 1.3653755187988281e-05,
      "model_forward_time": 1.7276108264923096,
      "step": 14317
    },
    {
      "epoch": 1.3653755187988281e-05,
      "step": 14317,
      "training_step_time": 7.340121030807495
    },
    {
      "epoch": 1.3654708862304687e-05,
      "model_forward_time": 1.8289756774902344,
      "step": 14318
    },
    {
      "epoch": 1.3654708862304687e-05,
      "step": 14318,
      "training_step_time": 8.681204795837402
    },
    {
      "epoch": 1.3655662536621095e-05,
      "model_forward_time": 1.743095874786377,
      "step": 14319
    },
    {
      "epoch": 1.3655662536621095e-05,
      "step": 14319,
      "training_step_time": 7.794805288314819
    },
    {
      "epoch": 1.36566162109375e-05,
      "grad_norm": 0.1911533623933792,
      "learning_rate": 9.764445542849739e-06,
      "loss": 0.0674,
      "step": 14320
    },
    {
      "epoch": 1.36566162109375e-05,
      "model_forward_time": 1.7707750797271729,
      "step": 14320
    },
    {
      "dynamics_loss_avg": 0.06923936791718006,
      "epoch": 1.36566162109375e-05,
      "step": 14320
    },
    {
      "action_loss_avg": 0.003479854273609817,
      "epoch": 1.36566162109375e-05,
      "step": 14320
    },
    {
      "epoch": 1.36566162109375e-05,
      "step": 14320,
      "training_step_time": 7.6066248416900635
    },
    {
      "epoch": 1.3657569885253906e-05,
      "model_forward_time": 1.7567710876464844,
      "step": 14321
    },
    {
      "epoch": 1.3657569885253906e-05,
      "step": 14321,
      "training_step_time": 6.810348987579346
    },
    {
      "epoch": 1.3658523559570312e-05,
      "model_forward_time": 1.7626545429229736,
      "step": 14322
    },
    {
      "epoch": 1.3658523559570312e-05,
      "step": 14322,
      "training_step_time": 6.8013975620269775
    },
    {
      "epoch": 1.3659477233886719e-05,
      "model_forward_time": 1.7609508037567139,
      "step": 14323
    },
    {
      "epoch": 1.3659477233886719e-05,
      "step": 14323,
      "training_step_time": 6.791300535202026
    },
    {
      "epoch": 1.3660430908203125e-05,
      "model_forward_time": 1.7667205333709717,
      "step": 14324
    },
    {
      "epoch": 1.3660430908203125e-05,
      "step": 14324,
      "training_step_time": 6.79268479347229
    },
    {
      "epoch": 1.366138458251953e-05,
      "model_forward_time": 1.766193151473999,
      "step": 14325
    },
    {
      "epoch": 1.366138458251953e-05,
      "step": 14325,
      "training_step_time": 6.771006345748901
    },
    {
      "epoch": 1.3662338256835938e-05,
      "model_forward_time": 1.7464745044708252,
      "step": 14326
    },
    {
      "epoch": 1.3662338256835938e-05,
      "step": 14326,
      "training_step_time": 7.3398449420928955
    },
    {
      "epoch": 1.3663291931152344e-05,
      "model_forward_time": 1.8374311923980713,
      "step": 14327
    },
    {
      "epoch": 1.3663291931152344e-05,
      "step": 14327,
      "training_step_time": 7.48437237739563
    },
    {
      "epoch": 1.366424560546875e-05,
      "model_forward_time": 1.7562966346740723,
      "step": 14328
    },
    {
      "epoch": 1.366424560546875e-05,
      "step": 14328,
      "training_step_time": 7.7439820766448975
    },
    {
      "epoch": 1.3665199279785157e-05,
      "model_forward_time": 1.7663977146148682,
      "step": 14329
    },
    {
      "epoch": 1.3665199279785157e-05,
      "step": 14329,
      "training_step_time": 7.637675046920776
    },
    {
      "epoch": 1.3666152954101563e-05,
      "grad_norm": 0.1958189308643341,
      "learning_rate": 9.763943753719e-06,
      "loss": 0.068,
      "step": 14330
    },
    {
      "epoch": 1.3666152954101563e-05,
      "model_forward_time": 1.769735336303711,
      "step": 14330
    },
    {
      "dynamics_loss_avg": 0.06475352756679058,
      "epoch": 1.3666152954101563e-05,
      "step": 14330
    },
    {
      "action_loss_avg": 0.0037298416835255923,
      "epoch": 1.3666152954101563e-05,
      "step": 14330
    },
    {
      "epoch": 1.3666152954101563e-05,
      "step": 14330,
      "training_step_time": 6.776418685913086
    },
    {
      "epoch": 1.3667106628417968e-05,
      "model_forward_time": 1.7712945938110352,
      "step": 14331
    },
    {
      "epoch": 1.3667106628417968e-05,
      "step": 14331,
      "training_step_time": 6.792953252792358
    },
    {
      "epoch": 1.3668060302734376e-05,
      "model_forward_time": 1.7706122398376465,
      "step": 14332
    },
    {
      "epoch": 1.3668060302734376e-05,
      "step": 14332,
      "training_step_time": 6.76520562171936
    },
    {
      "epoch": 1.3669013977050781e-05,
      "model_forward_time": 1.7677357196807861,
      "step": 14333
    },
    {
      "epoch": 1.3669013977050781e-05,
      "step": 14333,
      "training_step_time": 6.7691991329193115
    },
    {
      "epoch": 1.3669967651367187e-05,
      "model_forward_time": 1.7613718509674072,
      "step": 14334
    },
    {
      "epoch": 1.3669967651367187e-05,
      "step": 14334,
      "training_step_time": 6.787512302398682
    },
    {
      "epoch": 1.3670921325683595e-05,
      "model_forward_time": 1.764707088470459,
      "step": 14335
    },
    {
      "epoch": 1.3670921325683595e-05,
      "step": 14335,
      "training_step_time": 7.3218348026275635
    },
    {
      "epoch": 1.3671875e-05,
      "model_forward_time": 1.8106236457824707,
      "step": 14336
    },
    {
      "epoch": 1.3671875e-05,
      "step": 14336,
      "training_step_time": 7.499902725219727
    },
    {
      "epoch": 1.3672828674316406e-05,
      "model_forward_time": 1.727024793624878,
      "step": 14337
    },
    {
      "epoch": 1.3672828674316406e-05,
      "step": 14337,
      "training_step_time": 7.740757942199707
    },
    {
      "epoch": 1.3673782348632812e-05,
      "model_forward_time": 1.7450897693634033,
      "step": 14338
    },
    {
      "epoch": 1.3673782348632812e-05,
      "step": 14338,
      "training_step_time": 7.8664796352386475
    },
    {
      "epoch": 1.3674736022949219e-05,
      "model_forward_time": 1.7684431076049805,
      "step": 14339
    },
    {
      "epoch": 1.3674736022949219e-05,
      "step": 14339,
      "training_step_time": 6.782301187515259
    },
    {
      "epoch": 1.3675689697265625e-05,
      "grad_norm": 0.2648404538631439,
      "learning_rate": 9.76344144361053e-06,
      "loss": 0.0677,
      "step": 14340
    },
    {
      "epoch": 1.3675689697265625e-05,
      "model_forward_time": 1.7254366874694824,
      "step": 14340
    },
    {
      "dynamics_loss_avg": 0.06273552482016384,
      "epoch": 1.3675689697265625e-05,
      "step": 14340
    },
    {
      "action_loss_avg": 0.003468193634762429,
      "epoch": 1.3675689697265625e-05,
      "step": 14340
    },
    {
      "epoch": 1.3675689697265625e-05,
      "step": 14340,
      "training_step_time": 6.7861011028289795
    },
    {
      "epoch": 1.367664337158203e-05,
      "model_forward_time": 1.7382128238677979,
      "step": 14341
    },
    {
      "epoch": 1.367664337158203e-05,
      "step": 14341,
      "training_step_time": 6.780698537826538
    },
    {
      "epoch": 1.3677597045898438e-05,
      "model_forward_time": 1.7592298984527588,
      "step": 14342
    },
    {
      "epoch": 1.3677597045898438e-05,
      "step": 14342,
      "training_step_time": 6.759829521179199
    },
    {
      "epoch": 1.3678550720214844e-05,
      "model_forward_time": 1.7642321586608887,
      "step": 14343
    },
    {
      "epoch": 1.3678550720214844e-05,
      "step": 14343,
      "training_step_time": 6.6662139892578125
    },
    {
      "epoch": 1.367950439453125e-05,
      "model_forward_time": 1.7752842903137207,
      "step": 14344
    },
    {
      "epoch": 1.367950439453125e-05,
      "step": 14344,
      "training_step_time": 8.3154616355896
    },
    {
      "epoch": 1.3680458068847657e-05,
      "model_forward_time": 1.7976257801055908,
      "step": 14345
    },
    {
      "epoch": 1.3680458068847657e-05,
      "step": 14345,
      "training_step_time": 6.6218414306640625
    },
    {
      "epoch": 1.3681411743164063e-05,
      "model_forward_time": 1.8650665283203125,
      "step": 14346
    },
    {
      "epoch": 1.3681411743164063e-05,
      "step": 14346,
      "training_step_time": 9.22560167312622
    },
    {
      "epoch": 1.3682365417480468e-05,
      "model_forward_time": 1.7681553363800049,
      "step": 14347
    },
    {
      "epoch": 1.3682365417480468e-05,
      "step": 14347,
      "training_step_time": 7.839615821838379
    },
    {
      "epoch": 1.3683319091796876e-05,
      "model_forward_time": 1.7639095783233643,
      "step": 14348
    },
    {
      "epoch": 1.3683319091796876e-05,
      "step": 14348,
      "training_step_time": 6.846917390823364
    },
    {
      "epoch": 1.3684272766113281e-05,
      "model_forward_time": 1.7656519412994385,
      "step": 14349
    },
    {
      "epoch": 1.3684272766113281e-05,
      "step": 14349,
      "training_step_time": 6.862091064453125
    },
    {
      "epoch": 1.3685226440429687e-05,
      "grad_norm": 0.17636248469352722,
      "learning_rate": 9.762938612579269e-06,
      "loss": 0.0717,
      "step": 14350
    },
    {
      "epoch": 1.3685226440429687e-05,
      "model_forward_time": 1.7562029361724854,
      "step": 14350
    },
    {
      "dynamics_loss_avg": 0.05860830862075091,
      "epoch": 1.3685226440429687e-05,
      "step": 14350
    },
    {
      "action_loss_avg": 0.0028985351498704405,
      "epoch": 1.3685226440429687e-05,
      "step": 14350
    },
    {
      "epoch": 1.3685226440429687e-05,
      "step": 14350,
      "training_step_time": 6.844865798950195
    },
    {
      "epoch": 1.3686180114746095e-05,
      "model_forward_time": 1.7603189945220947,
      "step": 14351
    },
    {
      "epoch": 1.3686180114746095e-05,
      "step": 14351,
      "training_step_time": 6.819387435913086
    },
    {
      "epoch": 1.36871337890625e-05,
      "model_forward_time": 1.7581586837768555,
      "step": 14352
    },
    {
      "epoch": 1.36871337890625e-05,
      "step": 14352,
      "training_step_time": 6.753018617630005
    },
    {
      "epoch": 1.3688087463378906e-05,
      "model_forward_time": 1.762498378753662,
      "step": 14353
    },
    {
      "epoch": 1.3688087463378906e-05,
      "step": 14353,
      "training_step_time": 7.356035470962524
    },
    {
      "epoch": 1.3689041137695312e-05,
      "model_forward_time": 1.7617716789245605,
      "step": 14354
    },
    {
      "epoch": 1.3689041137695312e-05,
      "step": 14354,
      "training_step_time": 7.190444707870483
    },
    {
      "epoch": 1.368999481201172e-05,
      "model_forward_time": 1.8670406341552734,
      "step": 14355
    },
    {
      "epoch": 1.368999481201172e-05,
      "step": 14355,
      "training_step_time": 8.620702028274536
    },
    {
      "epoch": 1.3690948486328125e-05,
      "model_forward_time": 1.7639033794403076,
      "step": 14356
    },
    {
      "epoch": 1.3690948486328125e-05,
      "step": 14356,
      "training_step_time": 7.770483493804932
    },
    {
      "epoch": 1.369190216064453e-05,
      "model_forward_time": 1.7576234340667725,
      "step": 14357
    },
    {
      "epoch": 1.369190216064453e-05,
      "step": 14357,
      "training_step_time": 7.650930881500244
    },
    {
      "epoch": 1.3692855834960938e-05,
      "model_forward_time": 1.7727673053741455,
      "step": 14358
    },
    {
      "epoch": 1.3692855834960938e-05,
      "step": 14358,
      "training_step_time": 6.793494939804077
    },
    {
      "epoch": 1.3693809509277344e-05,
      "model_forward_time": 1.7679381370544434,
      "step": 14359
    },
    {
      "epoch": 1.3693809509277344e-05,
      "step": 14359,
      "training_step_time": 6.780963182449341
    },
    {
      "epoch": 1.369476318359375e-05,
      "grad_norm": 0.2237207293510437,
      "learning_rate": 9.762435260680203e-06,
      "loss": 0.0723,
      "step": 14360
    },
    {
      "epoch": 1.369476318359375e-05,
      "model_forward_time": 1.773289680480957,
      "step": 14360
    },
    {
      "dynamics_loss_avg": 0.07272162176668644,
      "epoch": 1.369476318359375e-05,
      "step": 14360
    },
    {
      "action_loss_avg": 0.0027913038327824326,
      "epoch": 1.369476318359375e-05,
      "step": 14360
    },
    {
      "epoch": 1.369476318359375e-05,
      "step": 14360,
      "training_step_time": 6.819182872772217
    },
    {
      "epoch": 1.3695716857910157e-05,
      "model_forward_time": 1.7695798873901367,
      "step": 14361
    },
    {
      "epoch": 1.3695716857910157e-05,
      "step": 14361,
      "training_step_time": 6.80440616607666
    },
    {
      "epoch": 1.3696670532226563e-05,
      "model_forward_time": 1.7644612789154053,
      "step": 14362
    },
    {
      "epoch": 1.3696670532226563e-05,
      "step": 14362,
      "training_step_time": 6.8324878215789795
    },
    {
      "epoch": 1.3697624206542968e-05,
      "model_forward_time": 2.892195224761963,
      "step": 14363
    },
    {
      "epoch": 1.3697624206542968e-05,
      "step": 14363,
      "training_step_time": 7.746735334396362
    },
    {
      "epoch": 1.3698577880859376e-05,
      "model_forward_time": 1.8672621250152588,
      "step": 14364
    },
    {
      "epoch": 1.3698577880859376e-05,
      "step": 14364,
      "training_step_time": 7.830418348312378
    },
    {
      "epoch": 1.3699531555175781e-05,
      "model_forward_time": 1.7556977272033691,
      "step": 14365
    },
    {
      "epoch": 1.3699531555175781e-05,
      "step": 14365,
      "training_step_time": 8.30367136001587
    },
    {
      "epoch": 1.3700485229492187e-05,
      "model_forward_time": 1.7660744190216064,
      "step": 14366
    },
    {
      "epoch": 1.3700485229492187e-05,
      "step": 14366,
      "training_step_time": 7.778124570846558
    },
    {
      "epoch": 1.3701438903808595e-05,
      "model_forward_time": 1.767134428024292,
      "step": 14367
    },
    {
      "epoch": 1.3701438903808595e-05,
      "step": 14367,
      "training_step_time": 6.826161623001099
    },
    {
      "epoch": 1.3702392578125e-05,
      "model_forward_time": 1.759413719177246,
      "step": 14368
    },
    {
      "epoch": 1.3702392578125e-05,
      "step": 14368,
      "training_step_time": 6.783995151519775
    },
    {
      "epoch": 1.3703346252441406e-05,
      "model_forward_time": 1.7584006786346436,
      "step": 14369
    },
    {
      "epoch": 1.3703346252441406e-05,
      "step": 14369,
      "training_step_time": 6.8358941078186035
    },
    {
      "epoch": 1.3704299926757812e-05,
      "grad_norm": 0.21324844658374786,
      "learning_rate": 9.761931387968375e-06,
      "loss": 0.0659,
      "step": 14370
    },
    {
      "epoch": 1.3704299926757812e-05,
      "model_forward_time": 1.7619645595550537,
      "step": 14370
    },
    {
      "dynamics_loss_avg": 0.054246424324810506,
      "epoch": 1.3704299926757812e-05,
      "step": 14370
    },
    {
      "action_loss_avg": 0.0026026440667919815,
      "epoch": 1.3704299926757812e-05,
      "step": 14370
    },
    {
      "epoch": 1.3704299926757812e-05,
      "step": 14370,
      "training_step_time": 6.8403637409210205
    },
    {
      "epoch": 1.370525360107422e-05,
      "model_forward_time": 1.7603628635406494,
      "step": 14371
    },
    {
      "epoch": 1.370525360107422e-05,
      "step": 14371,
      "training_step_time": 6.878516435623169
    },
    {
      "epoch": 1.3706207275390625e-05,
      "model_forward_time": 2.9932963848114014,
      "step": 14372
    },
    {
      "epoch": 1.3706207275390625e-05,
      "step": 14372,
      "training_step_time": 7.849379539489746
    },
    {
      "epoch": 1.370716094970703e-05,
      "model_forward_time": 1.8702340126037598,
      "step": 14373
    },
    {
      "epoch": 1.370716094970703e-05,
      "step": 14373,
      "training_step_time": 9.619707107543945
    },
    {
      "epoch": 1.3708114624023438e-05,
      "model_forward_time": 1.7647597789764404,
      "step": 14374
    },
    {
      "epoch": 1.3708114624023438e-05,
      "step": 14374,
      "training_step_time": 10.248862743377686
    },
    {
      "epoch": 1.3709068298339844e-05,
      "model_forward_time": 1.7621080875396729,
      "step": 14375
    },
    {
      "epoch": 1.3709068298339844e-05,
      "step": 14375,
      "training_step_time": 7.728881597518921
    },
    {
      "epoch": 1.371002197265625e-05,
      "model_forward_time": 1.7693676948547363,
      "step": 14376
    },
    {
      "epoch": 1.371002197265625e-05,
      "step": 14376,
      "training_step_time": 7.489419937133789
    },
    {
      "epoch": 1.3710975646972657e-05,
      "model_forward_time": 1.7625820636749268,
      "step": 14377
    },
    {
      "epoch": 1.3710975646972657e-05,
      "step": 14377,
      "training_step_time": 6.811916351318359
    },
    {
      "epoch": 1.3711929321289063e-05,
      "model_forward_time": 1.7675976753234863,
      "step": 14378
    },
    {
      "epoch": 1.3711929321289063e-05,
      "step": 14378,
      "training_step_time": 6.799089193344116
    },
    {
      "epoch": 1.3712882995605468e-05,
      "model_forward_time": 1.761282205581665,
      "step": 14379
    },
    {
      "epoch": 1.3712882995605468e-05,
      "step": 14379,
      "training_step_time": 6.815468788146973
    },
    {
      "epoch": 1.3713836669921876e-05,
      "grad_norm": 0.21138682961463928,
      "learning_rate": 9.76142699449889e-06,
      "loss": 0.0673,
      "step": 14380
    },
    {
      "epoch": 1.3713836669921876e-05,
      "model_forward_time": 1.7719743251800537,
      "step": 14380
    },
    {
      "dynamics_loss_avg": 0.06243498809635639,
      "epoch": 1.3713836669921876e-05,
      "step": 14380
    },
    {
      "action_loss_avg": 0.0047142751165665684,
      "epoch": 1.3713836669921876e-05,
      "step": 14380
    },
    {
      "epoch": 1.3713836669921876e-05,
      "step": 14380,
      "training_step_time": 6.852089881896973
    },
    {
      "epoch": 1.3714790344238281e-05,
      "model_forward_time": 2.9221787452697754,
      "step": 14381
    },
    {
      "epoch": 1.3714790344238281e-05,
      "step": 14381,
      "training_step_time": 7.762923955917358
    },
    {
      "epoch": 1.3715744018554687e-05,
      "model_forward_time": 1.8737695217132568,
      "step": 14382
    },
    {
      "epoch": 1.3715744018554687e-05,
      "step": 14382,
      "training_step_time": 7.860148906707764
    },
    {
      "epoch": 1.3716697692871093e-05,
      "model_forward_time": 1.403465747833252,
      "step": 14383
    },
    {
      "epoch": 1.3716697692871093e-05,
      "step": 14383,
      "training_step_time": 11.159140825271606
    },
    {
      "epoch": 1.37176513671875e-05,
      "model_forward_time": 1.763667345046997,
      "step": 14384
    },
    {
      "epoch": 1.37176513671875e-05,
      "step": 14384,
      "training_step_time": 8.288663387298584
    },
    {
      "epoch": 1.3718605041503906e-05,
      "model_forward_time": 1.7667231559753418,
      "step": 14385
    },
    {
      "epoch": 1.3718605041503906e-05,
      "step": 14385,
      "training_step_time": 6.809174537658691
    },
    {
      "epoch": 1.3719558715820312e-05,
      "model_forward_time": 1.764908790588379,
      "step": 14386
    },
    {
      "epoch": 1.3719558715820312e-05,
      "step": 14386,
      "training_step_time": 6.836421012878418
    },
    {
      "epoch": 1.372051239013672e-05,
      "model_forward_time": 1.762840747833252,
      "step": 14387
    },
    {
      "epoch": 1.372051239013672e-05,
      "step": 14387,
      "training_step_time": 6.883028745651245
    },
    {
      "epoch": 1.3721466064453125e-05,
      "model_forward_time": 1.7681059837341309,
      "step": 14388
    },
    {
      "epoch": 1.3721466064453125e-05,
      "step": 14388,
      "training_step_time": 6.964540243148804
    },
    {
      "epoch": 1.372241973876953e-05,
      "model_forward_time": 1.7550182342529297,
      "step": 14389
    },
    {
      "epoch": 1.372241973876953e-05,
      "step": 14389,
      "training_step_time": 6.899290084838867
    },
    {
      "epoch": 1.3723373413085938e-05,
      "grad_norm": 0.21322718262672424,
      "learning_rate": 9.760922080326908e-06,
      "loss": 0.063,
      "step": 14390
    },
    {
      "epoch": 1.3723373413085938e-05,
      "model_forward_time": 2.8924171924591064,
      "step": 14390
    },
    {
      "dynamics_loss_avg": 0.0672576442360878,
      "epoch": 1.3723373413085938e-05,
      "step": 14390
    },
    {
      "action_loss_avg": 0.004063539713388309,
      "epoch": 1.3723373413085938e-05,
      "step": 14390
    },
    {
      "epoch": 1.3723373413085938e-05,
      "step": 14390,
      "training_step_time": 7.7900567054748535
    },
    {
      "epoch": 1.3724327087402344e-05,
      "model_forward_time": 1.85841965675354,
      "step": 14391
    },
    {
      "epoch": 1.3724327087402344e-05,
      "step": 14391,
      "training_step_time": 7.854176998138428
    },
    {
      "epoch": 1.372528076171875e-05,
      "model_forward_time": 1.7662672996520996,
      "step": 14392
    },
    {
      "epoch": 1.372528076171875e-05,
      "step": 14392,
      "training_step_time": 9.621284484863281
    },
    {
      "epoch": 1.3726234436035157e-05,
      "model_forward_time": 1.7847862243652344,
      "step": 14393
    },
    {
      "epoch": 1.3726234436035157e-05,
      "step": 14393,
      "training_step_time": 7.791344404220581
    },
    {
      "epoch": 1.3727188110351563e-05,
      "model_forward_time": 1.7690582275390625,
      "step": 14394
    },
    {
      "epoch": 1.3727188110351563e-05,
      "step": 14394,
      "training_step_time": 7.678331613540649
    },
    {
      "epoch": 1.3728141784667968e-05,
      "model_forward_time": 1.7651033401489258,
      "step": 14395
    },
    {
      "epoch": 1.3728141784667968e-05,
      "step": 14395,
      "training_step_time": 6.748775243759155
    },
    {
      "epoch": 1.3729095458984376e-05,
      "model_forward_time": 1.7595734596252441,
      "step": 14396
    },
    {
      "epoch": 1.3729095458984376e-05,
      "step": 14396,
      "training_step_time": 6.697202444076538
    },
    {
      "epoch": 1.3730049133300781e-05,
      "model_forward_time": 1.763838768005371,
      "step": 14397
    },
    {
      "epoch": 1.3730049133300781e-05,
      "step": 14397,
      "training_step_time": 6.800003528594971
    },
    {
      "epoch": 1.3731002807617187e-05,
      "model_forward_time": 1.7595877647399902,
      "step": 14398
    },
    {
      "epoch": 1.3731002807617187e-05,
      "step": 14398,
      "training_step_time": 6.759773015975952
    },
    {
      "epoch": 1.3731956481933593e-05,
      "model_forward_time": 2.858050584793091,
      "step": 14399
    },
    {
      "epoch": 1.3731956481933593e-05,
      "step": 14399,
      "training_step_time": 7.6877076625823975
    },
    {
      "epoch": 1.373291015625e-05,
      "grad_norm": 0.186324343085289,
      "learning_rate": 9.760416645507644e-06,
      "loss": 0.0616,
      "step": 14400
    },
    {
      "epoch": 1.373291015625e-05,
      "model_forward_time": 1.8678724765777588,
      "step": 14400
    },
    {
      "dynamics_loss_avg": 0.07573385890573263,
      "epoch": 1.373291015625e-05,
      "step": 14400
    },
    {
      "action_loss_avg": 0.004200229432899505,
      "epoch": 1.373291015625e-05,
      "step": 14400
    },
    {
      "epoch": 1.373291015625e-05,
      "step": 14400,
      "training_step_time": 7.773572683334351
    },
    {
      "epoch": 1.3733863830566406e-05,
      "model_forward_time": 1.7643065452575684,
      "step": 14401
    },
    {
      "epoch": 1.3733863830566406e-05,
      "step": 14401,
      "training_step_time": 7.658524513244629
    },
    {
      "epoch": 1.3734817504882812e-05,
      "model_forward_time": 1.7188622951507568,
      "step": 14402
    },
    {
      "epoch": 1.3734817504882812e-05,
      "step": 14402,
      "training_step_time": 8.749521255493164
    },
    {
      "epoch": 1.373577117919922e-05,
      "model_forward_time": 1.751941204071045,
      "step": 14403
    },
    {
      "epoch": 1.373577117919922e-05,
      "step": 14403,
      "training_step_time": 7.589344263076782
    },
    {
      "epoch": 1.3736724853515625e-05,
      "model_forward_time": 1.752697467803955,
      "step": 14404
    },
    {
      "epoch": 1.3736724853515625e-05,
      "step": 14404,
      "training_step_time": 6.700863361358643
    },
    {
      "epoch": 1.373767852783203e-05,
      "model_forward_time": 1.7518858909606934,
      "step": 14405
    },
    {
      "epoch": 1.373767852783203e-05,
      "step": 14405,
      "training_step_time": 6.750287294387817
    },
    {
      "epoch": 1.3738632202148438e-05,
      "model_forward_time": 1.7439069747924805,
      "step": 14406
    },
    {
      "epoch": 1.3738632202148438e-05,
      "step": 14406,
      "training_step_time": 6.7507383823394775
    },
    {
      "epoch": 1.3739585876464844e-05,
      "model_forward_time": 1.7499260902404785,
      "step": 14407
    },
    {
      "epoch": 1.3739585876464844e-05,
      "step": 14407,
      "training_step_time": 6.818233013153076
    },
    {
      "epoch": 1.374053955078125e-05,
      "model_forward_time": 1.3861353397369385,
      "step": 14408
    },
    {
      "epoch": 1.374053955078125e-05,
      "step": 14408,
      "training_step_time": 7.106173515319824
    },
    {
      "epoch": 1.3741493225097657e-05,
      "model_forward_time": 2.8982372283935547,
      "step": 14409
    },
    {
      "epoch": 1.3741493225097657e-05,
      "step": 14409,
      "training_step_time": 7.737942218780518
    },
    {
      "epoch": 1.3742446899414063e-05,
      "grad_norm": 0.1952248513698578,
      "learning_rate": 9.759910690096375e-06,
      "loss": 0.0641,
      "step": 14410
    },
    {
      "epoch": 1.3742446899414063e-05,
      "model_forward_time": 1.7290239334106445,
      "step": 14410
    },
    {
      "dynamics_loss_avg": 0.06119050290435553,
      "epoch": 1.3742446899414063e-05,
      "step": 14410
    },
    {
      "action_loss_avg": 0.003897237905766815,
      "epoch": 1.3742446899414063e-05,
      "step": 14410
    },
    {
      "epoch": 1.3742446899414063e-05,
      "step": 14410,
      "training_step_time": 9.640202522277832
    },
    {
      "epoch": 1.3743400573730468e-05,
      "model_forward_time": 1.3796296119689941,
      "step": 14411
    },
    {
      "epoch": 1.3743400573730468e-05,
      "step": 14411,
      "training_step_time": 9.513190507888794
    },
    {
      "epoch": 1.3744354248046876e-05,
      "model_forward_time": 1.7599432468414307,
      "step": 14412
    },
    {
      "epoch": 1.3744354248046876e-05,
      "step": 14412,
      "training_step_time": 8.119544744491577
    },
    {
      "epoch": 1.3745307922363281e-05,
      "model_forward_time": 1.7448277473449707,
      "step": 14413
    },
    {
      "epoch": 1.3745307922363281e-05,
      "step": 14413,
      "training_step_time": 6.7782087326049805
    },
    {
      "epoch": 1.3746261596679687e-05,
      "model_forward_time": 1.746913194656372,
      "step": 14414
    },
    {
      "epoch": 1.3746261596679687e-05,
      "step": 14414,
      "training_step_time": 6.803819894790649
    },
    {
      "epoch": 1.3747215270996093e-05,
      "model_forward_time": 1.7995660305023193,
      "step": 14415
    },
    {
      "epoch": 1.3747215270996093e-05,
      "step": 14415,
      "training_step_time": 6.7363715171813965
    },
    {
      "epoch": 1.37481689453125e-05,
      "model_forward_time": 1.7521781921386719,
      "step": 14416
    },
    {
      "epoch": 1.37481689453125e-05,
      "step": 14416,
      "training_step_time": 6.711643695831299
    },
    {
      "epoch": 1.3749122619628906e-05,
      "model_forward_time": 1.7388675212860107,
      "step": 14417
    },
    {
      "epoch": 1.3749122619628906e-05,
      "step": 14417,
      "training_step_time": 7.067455530166626
    },
    {
      "epoch": 1.3750076293945312e-05,
      "model_forward_time": 2.856605052947998,
      "step": 14418
    },
    {
      "epoch": 1.3750076293945312e-05,
      "step": 14418,
      "training_step_time": 8.679847240447998
    },
    {
      "epoch": 1.375102996826172e-05,
      "model_forward_time": 1.8693747520446777,
      "step": 14419
    },
    {
      "epoch": 1.375102996826172e-05,
      "step": 14419,
      "training_step_time": 9.56345248222351
    },
    {
      "epoch": 1.3751983642578125e-05,
      "grad_norm": 0.18585777282714844,
      "learning_rate": 9.75940421414843e-06,
      "loss": 0.0663,
      "step": 14420
    },
    {
      "epoch": 1.3751983642578125e-05,
      "model_forward_time": 1.7318179607391357,
      "step": 14420
    },
    {
      "dynamics_loss_avg": 0.06666877111420036,
      "epoch": 1.3751983642578125e-05,
      "step": 14420
    },
    {
      "action_loss_avg": 0.002865364367607981,
      "epoch": 1.3751983642578125e-05,
      "step": 14420
    },
    {
      "epoch": 1.3751983642578125e-05,
      "step": 14420,
      "training_step_time": 10.083993673324585
    },
    {
      "epoch": 1.375293731689453e-05,
      "model_forward_time": 1.7501335144042969,
      "step": 14421
    },
    {
      "epoch": 1.375293731689453e-05,
      "step": 14421,
      "training_step_time": 7.551010370254517
    },
    {
      "epoch": 1.3753890991210938e-05,
      "model_forward_time": 1.7362103462219238,
      "step": 14422
    },
    {
      "epoch": 1.3753890991210938e-05,
      "step": 14422,
      "training_step_time": 7.535138845443726
    },
    {
      "epoch": 1.3754844665527344e-05,
      "model_forward_time": 1.75864839553833,
      "step": 14423
    },
    {
      "epoch": 1.3754844665527344e-05,
      "step": 14423,
      "training_step_time": 6.705845594406128
    },
    {
      "epoch": 1.375579833984375e-05,
      "model_forward_time": 1.7427680492401123,
      "step": 14424
    },
    {
      "epoch": 1.375579833984375e-05,
      "step": 14424,
      "training_step_time": 6.735311508178711
    },
    {
      "epoch": 1.3756752014160157e-05,
      "model_forward_time": 1.74757719039917,
      "step": 14425
    },
    {
      "epoch": 1.3756752014160157e-05,
      "step": 14425,
      "training_step_time": 6.721925497055054
    },
    {
      "epoch": 1.3757705688476563e-05,
      "model_forward_time": 1.7339229583740234,
      "step": 14426
    },
    {
      "epoch": 1.3757705688476563e-05,
      "step": 14426,
      "training_step_time": 7.057795286178589
    },
    {
      "epoch": 1.3758659362792968e-05,
      "model_forward_time": 2.8405375480651855,
      "step": 14427
    },
    {
      "epoch": 1.3758659362792968e-05,
      "step": 14427,
      "training_step_time": 7.752575874328613
    },
    {
      "epoch": 1.3759613037109376e-05,
      "model_forward_time": 1.5014009475708008,
      "step": 14428
    },
    {
      "epoch": 1.3759613037109376e-05,
      "step": 14428,
      "training_step_time": 9.293043613433838
    },
    {
      "epoch": 1.3760566711425782e-05,
      "model_forward_time": 1.748687505722046,
      "step": 14429
    },
    {
      "epoch": 1.3760566711425782e-05,
      "step": 14429,
      "training_step_time": 7.72006893157959
    },
    {
      "epoch": 1.3761520385742187e-05,
      "grad_norm": 0.18632455170154572,
      "learning_rate": 9.758897217719191e-06,
      "loss": 0.0662,
      "step": 14430
    },
    {
      "epoch": 1.3761520385742187e-05,
      "model_forward_time": 1.7292051315307617,
      "step": 14430
    },
    {
      "dynamics_loss_avg": 0.05143632413819432,
      "epoch": 1.3761520385742187e-05,
      "step": 14430
    },
    {
      "action_loss_avg": 0.0038466316706035285,
      "epoch": 1.3761520385742187e-05,
      "step": 14430
    },
    {
      "epoch": 1.3761520385742187e-05,
      "step": 14430,
      "training_step_time": 7.6217896938323975
    },
    {
      "epoch": 1.3762474060058593e-05,
      "model_forward_time": 1.7527718544006348,
      "step": 14431
    },
    {
      "epoch": 1.3762474060058593e-05,
      "step": 14431,
      "training_step_time": 7.554143190383911
    },
    {
      "epoch": 1.3763427734375e-05,
      "model_forward_time": 1.7748041152954102,
      "step": 14432
    },
    {
      "epoch": 1.3763427734375e-05,
      "step": 14432,
      "training_step_time": 6.733926296234131
    },
    {
      "epoch": 1.3764381408691406e-05,
      "model_forward_time": 1.748199701309204,
      "step": 14433
    },
    {
      "epoch": 1.3764381408691406e-05,
      "step": 14433,
      "training_step_time": 6.682271718978882
    },
    {
      "epoch": 1.3765335083007812e-05,
      "model_forward_time": 1.7593817710876465,
      "step": 14434
    },
    {
      "epoch": 1.3765335083007812e-05,
      "step": 14434,
      "training_step_time": 6.673159122467041
    },
    {
      "epoch": 1.376628875732422e-05,
      "model_forward_time": 1.7520732879638672,
      "step": 14435
    },
    {
      "epoch": 1.376628875732422e-05,
      "step": 14435,
      "training_step_time": 7.224400520324707
    },
    {
      "epoch": 1.3767242431640625e-05,
      "model_forward_time": 1.7529947757720947,
      "step": 14436
    },
    {
      "epoch": 1.3767242431640625e-05,
      "step": 14436,
      "training_step_time": 7.704473495483398
    },
    {
      "epoch": 1.376819610595703e-05,
      "model_forward_time": 2.9335319995880127,
      "step": 14437
    },
    {
      "epoch": 1.376819610595703e-05,
      "step": 14437,
      "training_step_time": 8.710072040557861
    },
    {
      "epoch": 1.3769149780273438e-05,
      "model_forward_time": 1.74428129196167,
      "step": 14438
    },
    {
      "epoch": 1.3769149780273438e-05,
      "step": 14438,
      "training_step_time": 7.674846172332764
    },
    {
      "epoch": 1.3770103454589844e-05,
      "model_forward_time": 1.760075569152832,
      "step": 14439
    },
    {
      "epoch": 1.3770103454589844e-05,
      "step": 14439,
      "training_step_time": 8.635948896408081
    },
    {
      "epoch": 1.377105712890625e-05,
      "grad_norm": 0.30295199155807495,
      "learning_rate": 9.758389700864113e-06,
      "loss": 0.065,
      "step": 14440
    },
    {
      "epoch": 1.377105712890625e-05,
      "model_forward_time": 1.7453405857086182,
      "step": 14440
    },
    {
      "dynamics_loss_avg": 0.04981076177209616,
      "epoch": 1.377105712890625e-05,
      "step": 14440
    },
    {
      "action_loss_avg": 0.00256708741071634,
      "epoch": 1.377105712890625e-05,
      "step": 14440
    },
    {
      "epoch": 1.377105712890625e-05,
      "step": 14440,
      "training_step_time": 7.560211896896362
    },
    {
      "epoch": 1.3772010803222657e-05,
      "model_forward_time": 1.7487947940826416,
      "step": 14441
    },
    {
      "epoch": 1.3772010803222657e-05,
      "step": 14441,
      "training_step_time": 6.691237926483154
    },
    {
      "epoch": 1.3772964477539063e-05,
      "model_forward_time": 1.7603986263275146,
      "step": 14442
    },
    {
      "epoch": 1.3772964477539063e-05,
      "step": 14442,
      "training_step_time": 6.673135280609131
    },
    {
      "epoch": 1.3773918151855468e-05,
      "model_forward_time": 1.7758533954620361,
      "step": 14443
    },
    {
      "epoch": 1.3773918151855468e-05,
      "step": 14443,
      "training_step_time": 6.631453037261963
    },
    {
      "epoch": 1.3774871826171876e-05,
      "model_forward_time": 1.7566332817077637,
      "step": 14444
    },
    {
      "epoch": 1.3774871826171876e-05,
      "step": 14444,
      "training_step_time": 6.663760423660278
    },
    {
      "epoch": 1.3775825500488282e-05,
      "model_forward_time": 1.753551959991455,
      "step": 14445
    },
    {
      "epoch": 1.3775825500488282e-05,
      "step": 14445,
      "training_step_time": 7.775593280792236
    },
    {
      "epoch": 1.3776779174804687e-05,
      "model_forward_time": 2.9916679859161377,
      "step": 14446
    },
    {
      "epoch": 1.3776779174804687e-05,
      "step": 14446,
      "training_step_time": 8.264639377593994
    },
    {
      "epoch": 1.3777732849121093e-05,
      "model_forward_time": 1.4030125141143799,
      "step": 14447
    },
    {
      "epoch": 1.3777732849121093e-05,
      "step": 14447,
      "training_step_time": 9.453871726989746
    },
    {
      "epoch": 1.37786865234375e-05,
      "model_forward_time": 1.7749454975128174,
      "step": 14448
    },
    {
      "epoch": 1.37786865234375e-05,
      "step": 14448,
      "training_step_time": 9.38820767402649
    },
    {
      "epoch": 1.3779640197753906e-05,
      "model_forward_time": 1.7731413841247559,
      "step": 14449
    },
    {
      "epoch": 1.3779640197753906e-05,
      "step": 14449,
      "training_step_time": 7.618099212646484
    },
    {
      "epoch": 1.3780593872070312e-05,
      "grad_norm": 0.18530488014221191,
      "learning_rate": 9.75788166363869e-06,
      "loss": 0.0648,
      "step": 14450
    },
    {
      "epoch": 1.3780593872070312e-05,
      "model_forward_time": 1.77073335647583,
      "step": 14450
    },
    {
      "dynamics_loss_avg": 0.06380517799407244,
      "epoch": 1.3780593872070312e-05,
      "step": 14450
    },
    {
      "action_loss_avg": 0.004486372834071517,
      "epoch": 1.3780593872070312e-05,
      "step": 14450
    },
    {
      "epoch": 1.3780593872070312e-05,
      "step": 14450,
      "training_step_time": 6.6908087730407715
    },
    {
      "epoch": 1.378154754638672e-05,
      "model_forward_time": 1.7642991542816162,
      "step": 14451
    },
    {
      "epoch": 1.378154754638672e-05,
      "step": 14451,
      "training_step_time": 6.702229022979736
    },
    {
      "epoch": 1.3782501220703125e-05,
      "model_forward_time": 1.762751579284668,
      "step": 14452
    },
    {
      "epoch": 1.3782501220703125e-05,
      "step": 14452,
      "training_step_time": 6.684035539627075
    },
    {
      "epoch": 1.378345489501953e-05,
      "model_forward_time": 1.7682549953460693,
      "step": 14453
    },
    {
      "epoch": 1.378345489501953e-05,
      "step": 14453,
      "training_step_time": 6.680291414260864
    },
    {
      "epoch": 1.3784408569335938e-05,
      "model_forward_time": 1.7694995403289795,
      "step": 14454
    },
    {
      "epoch": 1.3784408569335938e-05,
      "step": 14454,
      "training_step_time": 7.766540288925171
    },
    {
      "epoch": 1.3785362243652344e-05,
      "model_forward_time": 2.926177501678467,
      "step": 14455
    },
    {
      "epoch": 1.3785362243652344e-05,
      "step": 14455,
      "training_step_time": 8.558248519897461
    },
    {
      "epoch": 1.378631591796875e-05,
      "model_forward_time": 1.6962156295776367,
      "step": 14456
    },
    {
      "epoch": 1.378631591796875e-05,
      "step": 14456,
      "training_step_time": 10.58803391456604
    },
    {
      "epoch": 1.3787269592285157e-05,
      "model_forward_time": 1.7114372253417969,
      "step": 14457
    },
    {
      "epoch": 1.3787269592285157e-05,
      "step": 14457,
      "training_step_time": 8.54952883720398
    },
    {
      "epoch": 1.3788223266601563e-05,
      "model_forward_time": 1.713416337966919,
      "step": 14458
    },
    {
      "epoch": 1.3788223266601563e-05,
      "step": 14458,
      "training_step_time": 7.592164993286133
    },
    {
      "epoch": 1.3789176940917968e-05,
      "model_forward_time": 1.71022629737854,
      "step": 14459
    },
    {
      "epoch": 1.3789176940917968e-05,
      "step": 14459,
      "training_step_time": 6.707443714141846
    },
    {
      "epoch": 1.3790130615234376e-05,
      "grad_norm": 0.18093283474445343,
      "learning_rate": 9.757373106098478e-06,
      "loss": 0.0681,
      "step": 14460
    },
    {
      "epoch": 1.3790130615234376e-05,
      "model_forward_time": 1.7023303508758545,
      "step": 14460
    },
    {
      "dynamics_loss_avg": 0.08689296692609787,
      "epoch": 1.3790130615234376e-05,
      "step": 14460
    },
    {
      "action_loss_avg": 0.006423177185934037,
      "epoch": 1.3790130615234376e-05,
      "step": 14460
    },
    {
      "epoch": 1.3790130615234376e-05,
      "step": 14460,
      "training_step_time": 6.677721738815308
    },
    {
      "epoch": 1.3791084289550782e-05,
      "model_forward_time": 1.7062199115753174,
      "step": 14461
    },
    {
      "epoch": 1.3791084289550782e-05,
      "step": 14461,
      "training_step_time": 6.704563140869141
    },
    {
      "epoch": 1.3792037963867187e-05,
      "model_forward_time": 1.7002532482147217,
      "step": 14462
    },
    {
      "epoch": 1.3792037963867187e-05,
      "step": 14462,
      "training_step_time": 6.695386171340942
    },
    {
      "epoch": 1.3792991638183593e-05,
      "model_forward_time": 1.7127301692962646,
      "step": 14463
    },
    {
      "epoch": 1.3792991638183593e-05,
      "step": 14463,
      "training_step_time": 7.740782737731934
    },
    {
      "epoch": 1.37939453125e-05,
      "model_forward_time": 1.3364338874816895,
      "step": 14464
    },
    {
      "epoch": 1.37939453125e-05,
      "step": 14464,
      "training_step_time": 7.654566526412964
    },
    {
      "epoch": 1.3794898986816406e-05,
      "model_forward_time": 1.7118422985076904,
      "step": 14465
    },
    {
      "epoch": 1.3794898986816406e-05,
      "step": 14465,
      "training_step_time": 8.402881145477295
    },
    {
      "epoch": 1.3795852661132812e-05,
      "model_forward_time": 1.7407286167144775,
      "step": 14466
    },
    {
      "epoch": 1.3795852661132812e-05,
      "step": 14466,
      "training_step_time": 8.288948059082031
    },
    {
      "epoch": 1.379680633544922e-05,
      "model_forward_time": 1.7086067199707031,
      "step": 14467
    },
    {
      "epoch": 1.379680633544922e-05,
      "step": 14467,
      "training_step_time": 7.572078704833984
    },
    {
      "epoch": 1.3797760009765625e-05,
      "model_forward_time": 1.711740255355835,
      "step": 14468
    },
    {
      "epoch": 1.3797760009765625e-05,
      "step": 14468,
      "training_step_time": 7.588700771331787
    },
    {
      "epoch": 1.379871368408203e-05,
      "model_forward_time": 1.712721824645996,
      "step": 14469
    },
    {
      "epoch": 1.379871368408203e-05,
      "step": 14469,
      "training_step_time": 6.662709712982178
    },
    {
      "epoch": 1.3799667358398438e-05,
      "grad_norm": 0.1779119074344635,
      "learning_rate": 9.756864028299098e-06,
      "loss": 0.0703,
      "step": 14470
    },
    {
      "epoch": 1.3799667358398438e-05,
      "model_forward_time": 1.709892749786377,
      "step": 14470
    },
    {
      "dynamics_loss_avg": 0.07048030979931355,
      "epoch": 1.3799667358398438e-05,
      "step": 14470
    },
    {
      "action_loss_avg": 0.0040539205423556265,
      "epoch": 1.3799667358398438e-05,
      "step": 14470
    },
    {
      "epoch": 1.3799667358398438e-05,
      "step": 14470,
      "training_step_time": 6.686916828155518
    },
    {
      "epoch": 1.3800621032714844e-05,
      "model_forward_time": 1.726513385772705,
      "step": 14471
    },
    {
      "epoch": 1.3800621032714844e-05,
      "step": 14471,
      "training_step_time": 6.690147876739502
    },
    {
      "epoch": 1.380157470703125e-05,
      "model_forward_time": 1.7328073978424072,
      "step": 14472
    },
    {
      "epoch": 1.380157470703125e-05,
      "step": 14472,
      "training_step_time": 8.78219747543335
    },
    {
      "epoch": 1.3802528381347657e-05,
      "model_forward_time": 1.7732772827148438,
      "step": 14473
    },
    {
      "epoch": 1.3802528381347657e-05,
      "step": 14473,
      "training_step_time": 7.668099403381348
    },
    {
      "epoch": 1.3803482055664063e-05,
      "model_forward_time": 2.2610974311828613,
      "step": 14474
    },
    {
      "epoch": 1.3803482055664063e-05,
      "step": 14474,
      "training_step_time": 9.245134353637695
    },
    {
      "epoch": 1.3804435729980468e-05,
      "model_forward_time": 1.7255587577819824,
      "step": 14475
    },
    {
      "epoch": 1.3804435729980468e-05,
      "step": 14475,
      "training_step_time": 9.999328374862671
    },
    {
      "epoch": 1.3805389404296876e-05,
      "model_forward_time": 1.7318992614746094,
      "step": 14476
    },
    {
      "epoch": 1.3805389404296876e-05,
      "step": 14476,
      "training_step_time": 8.331271886825562
    },
    {
      "epoch": 1.3806343078613282e-05,
      "model_forward_time": 1.7341344356536865,
      "step": 14477
    },
    {
      "epoch": 1.3806343078613282e-05,
      "step": 14477,
      "training_step_time": 7.613079786300659
    },
    {
      "epoch": 1.3807296752929687e-05,
      "model_forward_time": 1.729867696762085,
      "step": 14478
    },
    {
      "epoch": 1.3807296752929687e-05,
      "step": 14478,
      "training_step_time": 6.709433555603027
    },
    {
      "epoch": 1.3808250427246093e-05,
      "model_forward_time": 1.7332284450531006,
      "step": 14479
    },
    {
      "epoch": 1.3808250427246093e-05,
      "step": 14479,
      "training_step_time": 6.680649518966675
    },
    {
      "epoch": 1.38092041015625e-05,
      "grad_norm": 0.16525588929653168,
      "learning_rate": 9.75635443029622e-06,
      "loss": 0.0674,
      "step": 14480
    },
    {
      "epoch": 1.38092041015625e-05,
      "model_forward_time": 1.706935167312622,
      "step": 14480
    },
    {
      "dynamics_loss_avg": 0.05128881316632032,
      "epoch": 1.38092041015625e-05,
      "step": 14480
    },
    {
      "action_loss_avg": 0.0030098593211732804,
      "epoch": 1.38092041015625e-05,
      "step": 14480
    },
    {
      "epoch": 1.38092041015625e-05,
      "step": 14480,
      "training_step_time": 6.68871283531189
    },
    {
      "epoch": 1.3810157775878906e-05,
      "model_forward_time": 1.7283034324645996,
      "step": 14481
    },
    {
      "epoch": 1.3810157775878906e-05,
      "step": 14481,
      "training_step_time": 7.574885129928589
    },
    {
      "epoch": 1.3811111450195312e-05,
      "model_forward_time": 1.7629568576812744,
      "step": 14482
    },
    {
      "epoch": 1.3811111450195312e-05,
      "step": 14482,
      "training_step_time": 8.783701419830322
    },
    {
      "epoch": 1.381206512451172e-05,
      "model_forward_time": 2.2540197372436523,
      "step": 14483
    },
    {
      "epoch": 1.381206512451172e-05,
      "step": 14483,
      "training_step_time": 7.659618139266968
    },
    {
      "epoch": 1.3813018798828125e-05,
      "model_forward_time": 1.720447063446045,
      "step": 14484
    },
    {
      "epoch": 1.3813018798828125e-05,
      "step": 14484,
      "training_step_time": 11.489051342010498
    },
    {
      "epoch": 1.381397247314453e-05,
      "model_forward_time": 1.7233712673187256,
      "step": 14485
    },
    {
      "epoch": 1.381397247314453e-05,
      "step": 14485,
      "training_step_time": 7.5502519607543945
    },
    {
      "epoch": 1.3814926147460938e-05,
      "model_forward_time": 1.7230477333068848,
      "step": 14486
    },
    {
      "epoch": 1.3814926147460938e-05,
      "step": 14486,
      "training_step_time": 7.540942668914795
    },
    {
      "epoch": 1.3815879821777344e-05,
      "model_forward_time": 1.7228894233703613,
      "step": 14487
    },
    {
      "epoch": 1.3815879821777344e-05,
      "step": 14487,
      "training_step_time": 6.670521974563599
    },
    {
      "epoch": 1.381683349609375e-05,
      "model_forward_time": 1.727560043334961,
      "step": 14488
    },
    {
      "epoch": 1.381683349609375e-05,
      "step": 14488,
      "training_step_time": 6.699033498764038
    },
    {
      "epoch": 1.3817787170410157e-05,
      "model_forward_time": 1.7183818817138672,
      "step": 14489
    },
    {
      "epoch": 1.3817787170410157e-05,
      "step": 14489,
      "training_step_time": 6.65602445602417
    },
    {
      "epoch": 1.3818740844726563e-05,
      "grad_norm": 0.1242603063583374,
      "learning_rate": 9.755844312145572e-06,
      "loss": 0.068,
      "step": 14490
    },
    {
      "epoch": 1.3818740844726563e-05,
      "model_forward_time": 1.7153723239898682,
      "step": 14490
    },
    {
      "dynamics_loss_avg": 0.06993596535176039,
      "epoch": 1.3818740844726563e-05,
      "step": 14490
    },
    {
      "action_loss_avg": 0.003782351699192077,
      "epoch": 1.3818740844726563e-05,
      "step": 14490
    },
    {
      "epoch": 1.3818740844726563e-05,
      "step": 14490,
      "training_step_time": 7.477350473403931
    },
    {
      "epoch": 1.3819694519042968e-05,
      "model_forward_time": 1.4034433364868164,
      "step": 14491
    },
    {
      "epoch": 1.3819694519042968e-05,
      "step": 14491,
      "training_step_time": 7.583645582199097
    },
    {
      "epoch": 1.3820648193359376e-05,
      "model_forward_time": 2.2444827556610107,
      "step": 14492
    },
    {
      "epoch": 1.3820648193359376e-05,
      "step": 14492,
      "training_step_time": 9.445406436920166
    },
    {
      "epoch": 1.3821601867675782e-05,
      "model_forward_time": 1.7201862335205078,
      "step": 14493
    },
    {
      "epoch": 1.3821601867675782e-05,
      "step": 14493,
      "training_step_time": 7.609715700149536
    },
    {
      "epoch": 1.3822555541992187e-05,
      "model_forward_time": 1.7225451469421387,
      "step": 14494
    },
    {
      "epoch": 1.3822555541992187e-05,
      "step": 14494,
      "training_step_time": 7.628347158432007
    },
    {
      "epoch": 1.3823509216308593e-05,
      "model_forward_time": 1.7238667011260986,
      "step": 14495
    },
    {
      "epoch": 1.3823509216308593e-05,
      "step": 14495,
      "training_step_time": 7.602604866027832
    },
    {
      "epoch": 1.3824462890625e-05,
      "model_forward_time": 1.7178130149841309,
      "step": 14496
    },
    {
      "epoch": 1.3824462890625e-05,
      "step": 14496,
      "training_step_time": 7.600881338119507
    },
    {
      "epoch": 1.3825416564941406e-05,
      "model_forward_time": 1.7194240093231201,
      "step": 14497
    },
    {
      "epoch": 1.3825416564941406e-05,
      "step": 14497,
      "training_step_time": 6.731658935546875
    },
    {
      "epoch": 1.3826370239257812e-05,
      "model_forward_time": 1.719693899154663,
      "step": 14498
    },
    {
      "epoch": 1.3826370239257812e-05,
      "step": 14498,
      "training_step_time": 6.733385324478149
    },
    {
      "epoch": 1.382732391357422e-05,
      "model_forward_time": 1.3462944030761719,
      "step": 14499
    },
    {
      "epoch": 1.382732391357422e-05,
      "step": 14499,
      "training_step_time": 7.568347454071045
    },
    {
      "epoch": 1.3828277587890625e-05,
      "grad_norm": 0.21220603585243225,
      "learning_rate": 9.755333673902941e-06,
      "loss": 0.0692,
      "step": 14500
    },
    {
      "epoch": 1.3828277587890625e-05,
      "model_forward_time": 1.760956048965454,
      "step": 14500
    },
    {
      "dynamics_loss_avg": 0.05754489451646805,
      "epoch": 1.3828277587890625e-05,
      "step": 14500
    },
    {
      "action_loss_avg": 0.002470122999511659,
      "epoch": 1.3828277587890625e-05,
      "step": 14500
    },
    {
      "epoch": 1.3828277587890625e-05,
      "step": 14500,
      "training_step_time": 7.8081910610198975
    },
    {
      "epoch": 1.382923126220703e-05,
      "model_forward_time": 2.253767490386963,
      "step": 14501
    },
    {
      "epoch": 1.382923126220703e-05,
      "step": 14501,
      "training_step_time": 7.614823579788208
    },
    {
      "epoch": 1.3830184936523438e-05,
      "model_forward_time": 1.7187025547027588,
      "step": 14502
    },
    {
      "epoch": 1.3830184936523438e-05,
      "step": 14502,
      "training_step_time": 9.523597955703735
    },
    {
      "epoch": 1.3831138610839844e-05,
      "model_forward_time": 1.3469557762145996,
      "step": 14503
    },
    {
      "epoch": 1.3831138610839844e-05,
      "step": 14503,
      "training_step_time": 8.693421840667725
    },
    {
      "epoch": 1.383209228515625e-05,
      "model_forward_time": 1.7007403373718262,
      "step": 14504
    },
    {
      "epoch": 1.383209228515625e-05,
      "step": 14504,
      "training_step_time": 7.547626972198486
    },
    {
      "epoch": 1.3833045959472657e-05,
      "model_forward_time": 1.3269307613372803,
      "step": 14505
    },
    {
      "epoch": 1.3833045959472657e-05,
      "step": 14505,
      "training_step_time": 7.629995346069336
    },
    {
      "epoch": 1.3833999633789063e-05,
      "model_forward_time": 1.7144742012023926,
      "step": 14506
    },
    {
      "epoch": 1.3833999633789063e-05,
      "step": 14506,
      "training_step_time": 6.740377902984619
    },
    {
      "epoch": 1.3834953308105468e-05,
      "model_forward_time": 1.7129356861114502,
      "step": 14507
    },
    {
      "epoch": 1.3834953308105468e-05,
      "step": 14507,
      "training_step_time": 6.754999160766602
    },
    {
      "epoch": 1.3835906982421874e-05,
      "model_forward_time": 1.7162630558013916,
      "step": 14508
    },
    {
      "epoch": 1.3835906982421874e-05,
      "step": 14508,
      "training_step_time": 7.595701694488525
    },
    {
      "epoch": 1.3836860656738282e-05,
      "model_forward_time": 1.7229297161102295,
      "step": 14509
    },
    {
      "epoch": 1.3836860656738282e-05,
      "step": 14509,
      "training_step_time": 7.780411243438721
    },
    {
      "epoch": 1.3837814331054687e-05,
      "grad_norm": 0.23422838747501373,
      "learning_rate": 9.75482251562417e-06,
      "loss": 0.0682,
      "step": 14510
    },
    {
      "epoch": 1.3837814331054687e-05,
      "model_forward_time": 1.7103900909423828,
      "step": 14510
    },
    {
      "dynamics_loss_avg": 0.07039213925600052,
      "epoch": 1.3837814331054687e-05,
      "step": 14510
    },
    {
      "action_loss_avg": 0.004550805693725124,
      "epoch": 1.3837814331054687e-05,
      "step": 14510
    },
    {
      "epoch": 1.3837814331054687e-05,
      "step": 14510,
      "training_step_time": 7.6804540157318115
    },
    {
      "epoch": 1.3838768005371093e-05,
      "model_forward_time": 2.254120111465454,
      "step": 14511
    },
    {
      "epoch": 1.3838768005371093e-05,
      "step": 14511,
      "training_step_time": 9.200068950653076
    },
    {
      "epoch": 1.38397216796875e-05,
      "model_forward_time": 1.7116096019744873,
      "step": 14512
    },
    {
      "epoch": 1.38397216796875e-05,
      "step": 14512,
      "training_step_time": 7.6853272914886475
    },
    {
      "epoch": 1.3840675354003906e-05,
      "model_forward_time": 1.7087576389312744,
      "step": 14513
    },
    {
      "epoch": 1.3840675354003906e-05,
      "step": 14513,
      "training_step_time": 8.072734355926514
    },
    {
      "epoch": 1.3841629028320312e-05,
      "model_forward_time": 1.718705177307129,
      "step": 14514
    },
    {
      "epoch": 1.3841629028320312e-05,
      "step": 14514,
      "training_step_time": 7.54734468460083
    },
    {
      "epoch": 1.384258270263672e-05,
      "model_forward_time": 1.7119805812835693,
      "step": 14515
    },
    {
      "epoch": 1.384258270263672e-05,
      "step": 14515,
      "training_step_time": 6.7435996532440186
    },
    {
      "epoch": 1.3843536376953125e-05,
      "model_forward_time": 1.7074170112609863,
      "step": 14516
    },
    {
      "epoch": 1.3843536376953125e-05,
      "step": 14516,
      "training_step_time": 6.742341756820679
    },
    {
      "epoch": 1.384449005126953e-05,
      "model_forward_time": 1.7136173248291016,
      "step": 14517
    },
    {
      "epoch": 1.384449005126953e-05,
      "step": 14517,
      "training_step_time": 7.580124616622925
    },
    {
      "epoch": 1.3845443725585938e-05,
      "model_forward_time": 1.7201929092407227,
      "step": 14518
    },
    {
      "epoch": 1.3845443725585938e-05,
      "step": 14518,
      "training_step_time": 7.416675329208374
    },
    {
      "epoch": 1.3846397399902344e-05,
      "model_forward_time": 1.7107272148132324,
      "step": 14519
    },
    {
      "epoch": 1.3846397399902344e-05,
      "step": 14519,
      "training_step_time": 7.822713375091553
    },
    {
      "epoch": 1.384735107421875e-05,
      "grad_norm": 0.22681595385074615,
      "learning_rate": 9.754310837365155e-06,
      "loss": 0.0681,
      "step": 14520
    },
    {
      "epoch": 1.384735107421875e-05,
      "model_forward_time": 2.233442783355713,
      "step": 14520
    },
    {
      "dynamics_loss_avg": 0.08291733004152775,
      "epoch": 1.384735107421875e-05,
      "step": 14520
    },
    {
      "action_loss_avg": 0.006893484294414521,
      "epoch": 1.384735107421875e-05,
      "step": 14520
    },
    {
      "epoch": 1.384735107421875e-05,
      "step": 14520,
      "training_step_time": 7.703074216842651
    },
    {
      "epoch": 1.3848304748535157e-05,
      "model_forward_time": 1.711277961730957,
      "step": 14521
    },
    {
      "epoch": 1.3848304748535157e-05,
      "step": 14521,
      "training_step_time": 11.396965742111206
    },
    {
      "epoch": 1.3849258422851563e-05,
      "model_forward_time": 1.7136893272399902,
      "step": 14522
    },
    {
      "epoch": 1.3849258422851563e-05,
      "step": 14522,
      "training_step_time": 8.21903657913208
    },
    {
      "epoch": 1.3850212097167968e-05,
      "model_forward_time": 1.7078208923339844,
      "step": 14523
    },
    {
      "epoch": 1.3850212097167968e-05,
      "step": 14523,
      "training_step_time": 7.620361566543579
    },
    {
      "epoch": 1.3851165771484374e-05,
      "model_forward_time": 1.707921028137207,
      "step": 14524
    },
    {
      "epoch": 1.3851165771484374e-05,
      "step": 14524,
      "training_step_time": 7.556735992431641
    },
    {
      "epoch": 1.3852119445800782e-05,
      "model_forward_time": 1.7110049724578857,
      "step": 14525
    },
    {
      "epoch": 1.3852119445800782e-05,
      "step": 14525,
      "training_step_time": 6.782226800918579
    },
    {
      "epoch": 1.3853073120117187e-05,
      "model_forward_time": 1.71156907081604,
      "step": 14526
    },
    {
      "epoch": 1.3853073120117187e-05,
      "step": 14526,
      "training_step_time": 7.5817694664001465
    },
    {
      "epoch": 1.3854026794433593e-05,
      "model_forward_time": 1.7170400619506836,
      "step": 14527
    },
    {
      "epoch": 1.3854026794433593e-05,
      "step": 14527,
      "training_step_time": 7.117800235748291
    },
    {
      "epoch": 1.385498046875e-05,
      "model_forward_time": 1.7083542346954346,
      "step": 14528
    },
    {
      "epoch": 1.385498046875e-05,
      "step": 14528,
      "training_step_time": 8.549509763717651
    },
    {
      "epoch": 1.3855934143066406e-05,
      "model_forward_time": 2.287783622741699,
      "step": 14529
    },
    {
      "epoch": 1.3855934143066406e-05,
      "step": 14529,
      "training_step_time": 7.685024261474609
    },
    {
      "epoch": 1.3856887817382812e-05,
      "grad_norm": 0.256896048784256,
      "learning_rate": 9.753798639181856e-06,
      "loss": 0.0673,
      "step": 14530
    },
    {
      "epoch": 1.3856887817382812e-05,
      "model_forward_time": 1.7052662372589111,
      "step": 14530
    },
    {
      "dynamics_loss_avg": 0.0734811369329691,
      "epoch": 1.3856887817382812e-05,
      "step": 14530
    },
    {
      "action_loss_avg": 0.002984365832526237,
      "epoch": 1.3856887817382812e-05,
      "step": 14530
    },
    {
      "epoch": 1.3856887817382812e-05,
      "step": 14530,
      "training_step_time": 7.689738988876343
    },
    {
      "epoch": 1.385784149169922e-05,
      "model_forward_time": 1.7326650619506836,
      "step": 14531
    },
    {
      "epoch": 1.385784149169922e-05,
      "step": 14531,
      "training_step_time": 8.71140432357788
    },
    {
      "epoch": 1.3858795166015625e-05,
      "model_forward_time": 1.7125256061553955,
      "step": 14532
    },
    {
      "epoch": 1.3858795166015625e-05,
      "step": 14532,
      "training_step_time": 7.654297113418579
    },
    {
      "epoch": 1.385974884033203e-05,
      "model_forward_time": 1.7123997211456299,
      "step": 14533
    },
    {
      "epoch": 1.385974884033203e-05,
      "step": 14533,
      "training_step_time": 7.547744274139404
    },
    {
      "epoch": 1.3860702514648438e-05,
      "model_forward_time": 1.715104579925537,
      "step": 14534
    },
    {
      "epoch": 1.3860702514648438e-05,
      "step": 14534,
      "training_step_time": 6.7330145835876465
    },
    {
      "epoch": 1.3861656188964844e-05,
      "model_forward_time": 1.7444946765899658,
      "step": 14535
    },
    {
      "epoch": 1.3861656188964844e-05,
      "step": 14535,
      "training_step_time": 7.569155931472778
    },
    {
      "epoch": 1.386260986328125e-05,
      "model_forward_time": 1.7067294120788574,
      "step": 14536
    },
    {
      "epoch": 1.386260986328125e-05,
      "step": 14536,
      "training_step_time": 6.766096353530884
    },
    {
      "epoch": 1.3863563537597657e-05,
      "model_forward_time": 1.7166833877563477,
      "step": 14537
    },
    {
      "epoch": 1.3863563537597657e-05,
      "step": 14537,
      "training_step_time": 8.429442167282104
    },
    {
      "epoch": 1.3864517211914063e-05,
      "model_forward_time": 2.2818284034729004,
      "step": 14538
    },
    {
      "epoch": 1.3864517211914063e-05,
      "step": 14538,
      "training_step_time": 7.63883376121521
    },
    {
      "epoch": 1.3865470886230468e-05,
      "model_forward_time": 1.7188341617584229,
      "step": 14539
    },
    {
      "epoch": 1.3865470886230468e-05,
      "step": 14539,
      "training_step_time": 10.54718828201294
    },
    {
      "epoch": 1.3866424560546874e-05,
      "grad_norm": 0.19237887859344482,
      "learning_rate": 9.753285921130286e-06,
      "loss": 0.0645,
      "step": 14540
    },
    {
      "epoch": 1.3866424560546874e-05,
      "model_forward_time": 1.771486759185791,
      "step": 14540
    },
    {
      "dynamics_loss_avg": 0.04545336235314608,
      "epoch": 1.3866424560546874e-05,
      "step": 14540
    },
    {
      "action_loss_avg": 0.0029309854377061128,
      "epoch": 1.3866424560546874e-05,
      "step": 14540
    },
    {
      "epoch": 1.3866424560546874e-05,
      "step": 14540,
      "training_step_time": 10.081710815429688
    },
    {
      "epoch": 1.3867378234863282e-05,
      "model_forward_time": 1.735661268234253,
      "step": 14541
    },
    {
      "epoch": 1.3867378234863282e-05,
      "step": 14541,
      "training_step_time": 7.563640832901001
    },
    {
      "epoch": 1.3868331909179687e-05,
      "model_forward_time": 1.7236802577972412,
      "step": 14542
    },
    {
      "epoch": 1.3868331909179687e-05,
      "step": 14542,
      "training_step_time": 7.625497341156006
    },
    {
      "epoch": 1.3869285583496093e-05,
      "model_forward_time": 1.7187910079956055,
      "step": 14543
    },
    {
      "epoch": 1.3869285583496093e-05,
      "step": 14543,
      "training_step_time": 6.764894008636475
    },
    {
      "epoch": 1.38702392578125e-05,
      "model_forward_time": 1.722085952758789,
      "step": 14544
    },
    {
      "epoch": 1.38702392578125e-05,
      "step": 14544,
      "training_step_time": 7.599078178405762
    },
    {
      "epoch": 1.3871192932128906e-05,
      "model_forward_time": 1.7328691482543945,
      "step": 14545
    },
    {
      "epoch": 1.3871192932128906e-05,
      "step": 14545,
      "training_step_time": 7.290673732757568
    },
    {
      "epoch": 1.3872146606445312e-05,
      "model_forward_time": 1.726142406463623,
      "step": 14546
    },
    {
      "epoch": 1.3872146606445312e-05,
      "step": 14546,
      "training_step_time": 7.5938191413879395
    },
    {
      "epoch": 1.387310028076172e-05,
      "model_forward_time": 2.267002820968628,
      "step": 14547
    },
    {
      "epoch": 1.387310028076172e-05,
      "step": 14547,
      "training_step_time": 8.477903127670288
    },
    {
      "epoch": 1.3874053955078125e-05,
      "model_forward_time": 1.7263126373291016,
      "step": 14548
    },
    {
      "epoch": 1.3874053955078125e-05,
      "step": 14548,
      "training_step_time": 7.743194580078125
    },
    {
      "epoch": 1.387500762939453e-05,
      "model_forward_time": 1.7495379447937012,
      "step": 14549
    },
    {
      "epoch": 1.387500762939453e-05,
      "step": 14549,
      "training_step_time": 8.723652839660645
    },
    {
      "epoch": 1.3875961303710938e-05,
      "grad_norm": 0.18612942099571228,
      "learning_rate": 9.752772683266512e-06,
      "loss": 0.0666,
      "step": 14550
    },
    {
      "epoch": 1.3875961303710938e-05,
      "model_forward_time": 1.735044240951538,
      "step": 14550
    },
    {
      "dynamics_loss_avg": 0.07876596599817276,
      "epoch": 1.3875961303710938e-05,
      "step": 14550
    },
    {
      "action_loss_avg": 0.0032638007309287787,
      "epoch": 1.3875961303710938e-05,
      "step": 14550
    },
    {
      "epoch": 1.3875961303710938e-05,
      "step": 14550,
      "training_step_time": 9.120034217834473
    },
    {
      "epoch": 1.3876914978027344e-05,
      "model_forward_time": 1.731067180633545,
      "step": 14551
    },
    {
      "epoch": 1.3876914978027344e-05,
      "step": 14551,
      "training_step_time": 7.683551549911499
    },
    {
      "epoch": 1.387786865234375e-05,
      "model_forward_time": 1.705805778503418,
      "step": 14552
    },
    {
      "epoch": 1.387786865234375e-05,
      "step": 14552,
      "training_step_time": 6.774054765701294
    },
    {
      "epoch": 1.3878822326660157e-05,
      "model_forward_time": 1.7384448051452637,
      "step": 14553
    },
    {
      "epoch": 1.3878822326660157e-05,
      "step": 14553,
      "training_step_time": 8.22805905342102
    },
    {
      "epoch": 1.3879776000976563e-05,
      "model_forward_time": 1.7346398830413818,
      "step": 14554
    },
    {
      "epoch": 1.3879776000976563e-05,
      "step": 14554,
      "training_step_time": 7.315358638763428
    },
    {
      "epoch": 1.3880729675292968e-05,
      "model_forward_time": 1.7302238941192627,
      "step": 14555
    },
    {
      "epoch": 1.3880729675292968e-05,
      "step": 14555,
      "training_step_time": 8.617512226104736
    },
    {
      "epoch": 1.3881683349609374e-05,
      "model_forward_time": 1.7383100986480713,
      "step": 14556
    },
    {
      "epoch": 1.3881683349609374e-05,
      "step": 14556,
      "training_step_time": 7.584397077560425
    },
    {
      "epoch": 1.3882637023925782e-05,
      "model_forward_time": 2.2676658630371094,
      "step": 14557
    },
    {
      "epoch": 1.3882637023925782e-05,
      "step": 14557,
      "training_step_time": 7.766499280929565
    },
    {
      "epoch": 1.3883590698242187e-05,
      "model_forward_time": 1.7398278713226318,
      "step": 14558
    },
    {
      "epoch": 1.3883590698242187e-05,
      "step": 14558,
      "training_step_time": 7.80120849609375
    },
    {
      "epoch": 1.3884544372558593e-05,
      "model_forward_time": 1.7344529628753662,
      "step": 14559
    },
    {
      "epoch": 1.3884544372558593e-05,
      "step": 14559,
      "training_step_time": 8.236694097518921
    },
    {
      "epoch": 1.3885498046875e-05,
      "grad_norm": 0.24965469539165497,
      "learning_rate": 9.752258925646665e-06,
      "loss": 0.0636,
      "step": 14560
    },
    {
      "epoch": 1.3885498046875e-05,
      "model_forward_time": 1.7330946922302246,
      "step": 14560
    },
    {
      "dynamics_loss_avg": 0.06798291988670827,
      "epoch": 1.3885498046875e-05,
      "step": 14560
    },
    {
      "action_loss_avg": 0.004175334156025201,
      "epoch": 1.3885498046875e-05,
      "step": 14560
    },
    {
      "epoch": 1.3885498046875e-05,
      "step": 14560,
      "training_step_time": 7.672818660736084
    },
    {
      "epoch": 1.3886451721191406e-05,
      "model_forward_time": 1.7281863689422607,
      "step": 14561
    },
    {
      "epoch": 1.3886451721191406e-05,
      "step": 14561,
      "training_step_time": 6.753339052200317
    },
    {
      "epoch": 1.3887405395507812e-05,
      "model_forward_time": 1.7363004684448242,
      "step": 14562
    },
    {
      "epoch": 1.3887405395507812e-05,
      "step": 14562,
      "training_step_time": 7.64757513999939
    },
    {
      "epoch": 1.388835906982422e-05,
      "model_forward_time": 1.7299082279205322,
      "step": 14563
    },
    {
      "epoch": 1.388835906982422e-05,
      "step": 14563,
      "training_step_time": 7.543469667434692
    },
    {
      "epoch": 1.3889312744140625e-05,
      "model_forward_time": 1.7274200916290283,
      "step": 14564
    },
    {
      "epoch": 1.3889312744140625e-05,
      "step": 14564,
      "training_step_time": 7.573385715484619
    },
    {
      "epoch": 1.389026641845703e-05,
      "model_forward_time": 1.7456386089324951,
      "step": 14565
    },
    {
      "epoch": 1.389026641845703e-05,
      "step": 14565,
      "training_step_time": 7.677855730056763
    },
    {
      "epoch": 1.3891220092773438e-05,
      "model_forward_time": 2.264835834503174,
      "step": 14566
    },
    {
      "epoch": 1.3891220092773438e-05,
      "step": 14566,
      "training_step_time": 8.725899934768677
    },
    {
      "epoch": 1.3892173767089844e-05,
      "model_forward_time": 1.705176830291748,
      "step": 14567
    },
    {
      "epoch": 1.3892173767089844e-05,
      "step": 14567,
      "training_step_time": 8.465549945831299
    },
    {
      "epoch": 1.389312744140625e-05,
      "model_forward_time": 1.7243399620056152,
      "step": 14568
    },
    {
      "epoch": 1.389312744140625e-05,
      "step": 14568,
      "training_step_time": 8.270368337631226
    },
    {
      "epoch": 1.3894081115722657e-05,
      "model_forward_time": 1.7246358394622803,
      "step": 14569
    },
    {
      "epoch": 1.3894081115722657e-05,
      "step": 14569,
      "training_step_time": 7.6831443309783936
    },
    {
      "epoch": 1.3895034790039063e-05,
      "grad_norm": 0.1792455017566681,
      "learning_rate": 9.751744648326926e-06,
      "loss": 0.0657,
      "step": 14570
    },
    {
      "epoch": 1.3895034790039063e-05,
      "model_forward_time": 1.7264091968536377,
      "step": 14570
    },
    {
      "dynamics_loss_avg": 0.056394048035144806,
      "epoch": 1.3895034790039063e-05,
      "step": 14570
    },
    {
      "action_loss_avg": 0.003569836204405874,
      "epoch": 1.3895034790039063e-05,
      "step": 14570
    },
    {
      "epoch": 1.3895034790039063e-05,
      "step": 14570,
      "training_step_time": 6.7447896003723145
    },
    {
      "epoch": 1.3895988464355468e-05,
      "model_forward_time": 1.7113535404205322,
      "step": 14571
    },
    {
      "epoch": 1.3895988464355468e-05,
      "step": 14571,
      "training_step_time": 7.553070545196533
    },
    {
      "epoch": 1.3896942138671874e-05,
      "model_forward_time": 1.725273609161377,
      "step": 14572
    },
    {
      "epoch": 1.3896942138671874e-05,
      "step": 14572,
      "training_step_time": 7.5235209465026855
    },
    {
      "epoch": 1.3897895812988282e-05,
      "model_forward_time": 1.7263433933258057,
      "step": 14573
    },
    {
      "epoch": 1.3897895812988282e-05,
      "step": 14573,
      "training_step_time": 8.034022569656372
    },
    {
      "epoch": 1.3898849487304687e-05,
      "model_forward_time": 1.7307751178741455,
      "step": 14574
    },
    {
      "epoch": 1.3898849487304687e-05,
      "step": 14574,
      "training_step_time": 7.718766212463379
    },
    {
      "epoch": 1.3899803161621093e-05,
      "model_forward_time": 2.2542243003845215,
      "step": 14575
    },
    {
      "epoch": 1.3899803161621093e-05,
      "step": 14575,
      "training_step_time": 9.0484037399292
    },
    {
      "epoch": 1.39007568359375e-05,
      "model_forward_time": 1.730320930480957,
      "step": 14576
    },
    {
      "epoch": 1.39007568359375e-05,
      "step": 14576,
      "training_step_time": 9.652148962020874
    },
    {
      "epoch": 1.3901710510253906e-05,
      "model_forward_time": 1.8111817836761475,
      "step": 14577
    },
    {
      "epoch": 1.3901710510253906e-05,
      "step": 14577,
      "training_step_time": 8.510238647460938
    },
    {
      "epoch": 1.3902664184570312e-05,
      "model_forward_time": 1.7246301174163818,
      "step": 14578
    },
    {
      "epoch": 1.3902664184570312e-05,
      "step": 14578,
      "training_step_time": 7.492208003997803
    },
    {
      "epoch": 1.390361785888672e-05,
      "model_forward_time": 1.7299625873565674,
      "step": 14579
    },
    {
      "epoch": 1.390361785888672e-05,
      "step": 14579,
      "training_step_time": 7.693063497543335
    },
    {
      "epoch": 1.3904571533203125e-05,
      "grad_norm": 0.17882126569747925,
      "learning_rate": 9.751229851363535e-06,
      "loss": 0.0682,
      "step": 14580
    },
    {
      "epoch": 1.3904571533203125e-05,
      "model_forward_time": 1.72529935836792,
      "step": 14580
    },
    {
      "dynamics_loss_avg": 0.06542472597211599,
      "epoch": 1.3904571533203125e-05,
      "step": 14580
    },
    {
      "action_loss_avg": 0.0036627999623306097,
      "epoch": 1.3904571533203125e-05,
      "step": 14580
    },
    {
      "epoch": 1.3904571533203125e-05,
      "step": 14580,
      "training_step_time": 7.622801065444946
    },
    {
      "epoch": 1.390552520751953e-05,
      "model_forward_time": 1.731903314590454,
      "step": 14581
    },
    {
      "epoch": 1.390552520751953e-05,
      "step": 14581,
      "training_step_time": 7.497751235961914
    },
    {
      "epoch": 1.3906478881835938e-05,
      "model_forward_time": 1.730069637298584,
      "step": 14582
    },
    {
      "epoch": 1.3906478881835938e-05,
      "step": 14582,
      "training_step_time": 7.015249252319336
    },
    {
      "epoch": 1.3907432556152344e-05,
      "model_forward_time": 1.7283782958984375,
      "step": 14583
    },
    {
      "epoch": 1.3907432556152344e-05,
      "step": 14583,
      "training_step_time": 8.292649984359741
    },
    {
      "epoch": 1.390838623046875e-05,
      "model_forward_time": 2.254093647003174,
      "step": 14584
    },
    {
      "epoch": 1.390838623046875e-05,
      "step": 14584,
      "training_step_time": 8.371596336364746
    },
    {
      "epoch": 1.3909339904785157e-05,
      "model_forward_time": 1.7120821475982666,
      "step": 14585
    },
    {
      "epoch": 1.3909339904785157e-05,
      "step": 14585,
      "training_step_time": 9.709172248840332
    },
    {
      "epoch": 1.3910293579101563e-05,
      "model_forward_time": 1.7367668151855469,
      "step": 14586
    },
    {
      "epoch": 1.3910293579101563e-05,
      "step": 14586,
      "training_step_time": 9.049608707427979
    },
    {
      "epoch": 1.3911247253417969e-05,
      "model_forward_time": 1.7135674953460693,
      "step": 14587
    },
    {
      "epoch": 1.3911247253417969e-05,
      "step": 14587,
      "training_step_time": 7.598980188369751
    },
    {
      "epoch": 1.3912200927734374e-05,
      "model_forward_time": 1.7130835056304932,
      "step": 14588
    },
    {
      "epoch": 1.3912200927734374e-05,
      "step": 14588,
      "training_step_time": 7.677410125732422
    },
    {
      "epoch": 1.3913154602050782e-05,
      "model_forward_time": 1.7141621112823486,
      "step": 14589
    },
    {
      "epoch": 1.3913154602050782e-05,
      "step": 14589,
      "training_step_time": 7.642598628997803
    },
    {
      "epoch": 1.3914108276367187e-05,
      "grad_norm": 0.19644658267498016,
      "learning_rate": 9.750714534812793e-06,
      "loss": 0.0698,
      "step": 14590
    },
    {
      "epoch": 1.3914108276367187e-05,
      "model_forward_time": 1.708411693572998,
      "step": 14590
    },
    {
      "dynamics_loss_avg": 0.06727822981774807,
      "epoch": 1.3914108276367187e-05,
      "step": 14590
    },
    {
      "action_loss_avg": 0.003917702112812549,
      "epoch": 1.3914108276367187e-05,
      "step": 14590
    },
    {
      "epoch": 1.3914108276367187e-05,
      "step": 14590,
      "training_step_time": 7.48813271522522
    },
    {
      "epoch": 1.3915061950683593e-05,
      "model_forward_time": 1.7195472717285156,
      "step": 14591
    },
    {
      "epoch": 1.3915061950683593e-05,
      "step": 14591,
      "training_step_time": 7.044129133224487
    },
    {
      "epoch": 1.3916015625e-05,
      "model_forward_time": 1.7170584201812744,
      "step": 14592
    },
    {
      "epoch": 1.3916015625e-05,
      "step": 14592,
      "training_step_time": 8.581286191940308
    },
    {
      "epoch": 1.3916969299316406e-05,
      "model_forward_time": 2.274862766265869,
      "step": 14593
    },
    {
      "epoch": 1.3916969299316406e-05,
      "step": 14593,
      "training_step_time": 8.288546562194824
    },
    {
      "epoch": 1.3917922973632812e-05,
      "model_forward_time": 1.7078979015350342,
      "step": 14594
    },
    {
      "epoch": 1.3917922973632812e-05,
      "step": 14594,
      "training_step_time": 7.7141876220703125
    },
    {
      "epoch": 1.391887664794922e-05,
      "model_forward_time": 1.728872299194336,
      "step": 14595
    },
    {
      "epoch": 1.391887664794922e-05,
      "step": 14595,
      "training_step_time": 8.170722723007202
    },
    {
      "epoch": 1.3919830322265625e-05,
      "model_forward_time": 1.713979721069336,
      "step": 14596
    },
    {
      "epoch": 1.3919830322265625e-05,
      "step": 14596,
      "training_step_time": 7.6420934200286865
    },
    {
      "epoch": 1.392078399658203e-05,
      "model_forward_time": 1.7107937335968018,
      "step": 14597
    },
    {
      "epoch": 1.392078399658203e-05,
      "step": 14597,
      "training_step_time": 7.639556884765625
    },
    {
      "epoch": 1.3921737670898438e-05,
      "model_forward_time": 1.7107901573181152,
      "step": 14598
    },
    {
      "epoch": 1.3921737670898438e-05,
      "step": 14598,
      "training_step_time": 7.574082136154175
    },
    {
      "epoch": 1.3922691345214844e-05,
      "model_forward_time": 1.3316056728363037,
      "step": 14599
    },
    {
      "epoch": 1.3922691345214844e-05,
      "step": 14599,
      "training_step_time": 7.470268726348877
    },
    {
      "epoch": 1.392364501953125e-05,
      "grad_norm": 0.28613725304603577,
      "learning_rate": 9.750198698731053e-06,
      "loss": 0.073,
      "step": 14600
    },
    {
      "epoch": 1.392364501953125e-05,
      "model_forward_time": 1.7115778923034668,
      "step": 14600
    },
    {
      "dynamics_loss_avg": 0.06330064721405507,
      "epoch": 1.392364501953125e-05,
      "step": 14600
    },
    {
      "action_loss_avg": 0.00442670717020519,
      "epoch": 1.392364501953125e-05,
      "step": 14600
    },
    {
      "epoch": 1.392364501953125e-05,
      "step": 14600,
      "training_step_time": 7.038888931274414
    },
    {
      "epoch": 1.3924598693847657e-05,
      "model_forward_time": 1.7205884456634521,
      "step": 14601
    },
    {
      "epoch": 1.3924598693847657e-05,
      "step": 14601,
      "training_step_time": 7.583446264266968
    },
    {
      "epoch": 1.3925552368164063e-05,
      "model_forward_time": 1.710230827331543,
      "step": 14602
    },
    {
      "epoch": 1.3925552368164063e-05,
      "step": 14602,
      "training_step_time": 7.723859786987305
    },
    {
      "epoch": 1.3926506042480469e-05,
      "model_forward_time": 2.2357475757598877,
      "step": 14603
    },
    {
      "epoch": 1.3926506042480469e-05,
      "step": 14603,
      "training_step_time": 7.68359112739563
    },
    {
      "epoch": 1.3927459716796874e-05,
      "model_forward_time": 1.7108476161956787,
      "step": 14604
    },
    {
      "epoch": 1.3927459716796874e-05,
      "step": 14604,
      "training_step_time": 8.640807390213013
    },
    {
      "epoch": 1.3928413391113282e-05,
      "model_forward_time": 1.7035086154937744,
      "step": 14605
    },
    {
      "epoch": 1.3928413391113282e-05,
      "step": 14605,
      "training_step_time": 7.696409225463867
    },
    {
      "epoch": 1.3929367065429687e-05,
      "model_forward_time": 1.705888271331787,
      "step": 14606
    },
    {
      "epoch": 1.3929367065429687e-05,
      "step": 14606,
      "training_step_time": 7.620266437530518
    },
    {
      "epoch": 1.3930320739746093e-05,
      "model_forward_time": 1.7006630897521973,
      "step": 14607
    },
    {
      "epoch": 1.3930320739746093e-05,
      "step": 14607,
      "training_step_time": 7.812278985977173
    },
    {
      "epoch": 1.39312744140625e-05,
      "model_forward_time": 1.7031877040863037,
      "step": 14608
    },
    {
      "epoch": 1.39312744140625e-05,
      "step": 14608,
      "training_step_time": 7.6292054653167725
    },
    {
      "epoch": 1.3932228088378906e-05,
      "model_forward_time": 1.709273338317871,
      "step": 14609
    },
    {
      "epoch": 1.3932228088378906e-05,
      "step": 14609,
      "training_step_time": 7.580265522003174
    },
    {
      "epoch": 1.3933181762695312e-05,
      "grad_norm": 0.22070233523845673,
      "learning_rate": 9.749682343174723e-06,
      "loss": 0.0737,
      "step": 14610
    },
    {
      "epoch": 1.3933181762695312e-05,
      "model_forward_time": 1.6932275295257568,
      "step": 14610
    },
    {
      "dynamics_loss_avg": 0.049693088978528976,
      "epoch": 1.3933181762695312e-05,
      "step": 14610
    },
    {
      "action_loss_avg": 0.0020214766380377116,
      "epoch": 1.3933181762695312e-05,
      "step": 14610
    },
    {
      "epoch": 1.3933181762695312e-05,
      "step": 14610,
      "training_step_time": 7.597182273864746
    },
    {
      "epoch": 1.393413543701172e-05,
      "model_forward_time": 1.7204432487487793,
      "step": 14611
    },
    {
      "epoch": 1.393413543701172e-05,
      "step": 14611,
      "training_step_time": 9.403995275497437
    },
    {
      "epoch": 1.3935089111328125e-05,
      "model_forward_time": 2.2308027744293213,
      "step": 14612
    },
    {
      "epoch": 1.3935089111328125e-05,
      "step": 14612,
      "training_step_time": 7.813316106796265
    },
    {
      "epoch": 1.393604278564453e-05,
      "model_forward_time": 1.7015485763549805,
      "step": 14613
    },
    {
      "epoch": 1.393604278564453e-05,
      "step": 14613,
      "training_step_time": 12.337119102478027
    },
    {
      "epoch": 1.3936996459960938e-05,
      "model_forward_time": 1.7196691036224365,
      "step": 14614
    },
    {
      "epoch": 1.3936996459960938e-05,
      "step": 14614,
      "training_step_time": 8.675825834274292
    },
    {
      "epoch": 1.3937950134277344e-05,
      "model_forward_time": 1.712188720703125,
      "step": 14615
    },
    {
      "epoch": 1.3937950134277344e-05,
      "step": 14615,
      "training_step_time": 7.653421878814697
    },
    {
      "epoch": 1.393890380859375e-05,
      "model_forward_time": 1.7124314308166504,
      "step": 14616
    },
    {
      "epoch": 1.393890380859375e-05,
      "step": 14616,
      "training_step_time": 7.706839323043823
    },
    {
      "epoch": 1.3939857482910157e-05,
      "model_forward_time": 1.718515157699585,
      "step": 14617
    },
    {
      "epoch": 1.3939857482910157e-05,
      "step": 14617,
      "training_step_time": 7.7658491134643555
    },
    {
      "epoch": 1.3940811157226563e-05,
      "model_forward_time": 1.7080035209655762,
      "step": 14618
    },
    {
      "epoch": 1.3940811157226563e-05,
      "step": 14618,
      "training_step_time": 7.5706562995910645
    },
    {
      "epoch": 1.3941764831542969e-05,
      "model_forward_time": 1.7154004573822021,
      "step": 14619
    },
    {
      "epoch": 1.3941764831542969e-05,
      "step": 14619,
      "training_step_time": 7.463676691055298
    },
    {
      "epoch": 1.3942718505859374e-05,
      "grad_norm": 0.17085710167884827,
      "learning_rate": 9.749165468200273e-06,
      "loss": 0.0738,
      "step": 14620
    },
    {
      "epoch": 1.3942718505859374e-05,
      "model_forward_time": 1.7276790142059326,
      "step": 14620
    },
    {
      "dynamics_loss_avg": 0.07465472742915154,
      "epoch": 1.3942718505859374e-05,
      "step": 14620
    },
    {
      "action_loss_avg": 0.003716946131316945,
      "epoch": 1.3942718505859374e-05,
      "step": 14620
    },
    {
      "epoch": 1.3942718505859374e-05,
      "step": 14620,
      "training_step_time": 7.600340366363525
    },
    {
      "epoch": 1.3943672180175782e-05,
      "model_forward_time": 2.2236196994781494,
      "step": 14621
    },
    {
      "epoch": 1.3943672180175782e-05,
      "step": 14621,
      "training_step_time": 7.7943503856658936
    },
    {
      "epoch": 1.3944625854492187e-05,
      "model_forward_time": 1.712083101272583,
      "step": 14622
    },
    {
      "epoch": 1.3944625854492187e-05,
      "step": 14622,
      "training_step_time": 7.815791368484497
    },
    {
      "epoch": 1.3945579528808593e-05,
      "model_forward_time": 1.7285923957824707,
      "step": 14623
    },
    {
      "epoch": 1.3945579528808593e-05,
      "step": 14623,
      "training_step_time": 8.226333618164062
    },
    {
      "epoch": 1.3946533203125e-05,
      "model_forward_time": 1.708301067352295,
      "step": 14624
    },
    {
      "epoch": 1.3946533203125e-05,
      "step": 14624,
      "training_step_time": 7.787837266921997
    },
    {
      "epoch": 1.3947486877441406e-05,
      "model_forward_time": 1.708658218383789,
      "step": 14625
    },
    {
      "epoch": 1.3947486877441406e-05,
      "step": 14625,
      "training_step_time": 7.711561918258667
    },
    {
      "epoch": 1.3948440551757812e-05,
      "model_forward_time": 1.704925537109375,
      "step": 14626
    },
    {
      "epoch": 1.3948440551757812e-05,
      "step": 14626,
      "training_step_time": 7.548245429992676
    },
    {
      "epoch": 1.394939422607422e-05,
      "model_forward_time": 1.6972887516021729,
      "step": 14627
    },
    {
      "epoch": 1.394939422607422e-05,
      "step": 14627,
      "training_step_time": 7.442829608917236
    },
    {
      "epoch": 1.3950347900390625e-05,
      "model_forward_time": 1.7055010795593262,
      "step": 14628
    },
    {
      "epoch": 1.3950347900390625e-05,
      "step": 14628,
      "training_step_time": 7.414306879043579
    },
    {
      "epoch": 1.395130157470703e-05,
      "model_forward_time": 1.7192261219024658,
      "step": 14629
    },
    {
      "epoch": 1.395130157470703e-05,
      "step": 14629,
      "training_step_time": 7.650399208068848
    },
    {
      "epoch": 1.3952255249023438e-05,
      "grad_norm": 0.23838526010513306,
      "learning_rate": 9.748648073864229e-06,
      "loss": 0.0735,
      "step": 14630
    },
    {
      "epoch": 1.3952255249023438e-05,
      "model_forward_time": 2.243472099304199,
      "step": 14630
    },
    {
      "dynamics_loss_avg": 0.07342565134167671,
      "epoch": 1.3952255249023438e-05,
      "step": 14630
    },
    {
      "action_loss_avg": 0.0037719474988989533,
      "epoch": 1.3952255249023438e-05,
      "step": 14630
    },
    {
      "epoch": 1.3952255249023438e-05,
      "step": 14630,
      "training_step_time": 9.719268560409546
    },
    {
      "epoch": 1.3953208923339844e-05,
      "model_forward_time": 1.7054994106292725,
      "step": 14631
    },
    {
      "epoch": 1.3953208923339844e-05,
      "step": 14631,
      "training_step_time": 9.306374549865723
    },
    {
      "epoch": 1.395416259765625e-05,
      "model_forward_time": 1.726041316986084,
      "step": 14632
    },
    {
      "epoch": 1.395416259765625e-05,
      "step": 14632,
      "training_step_time": 8.480891466140747
    },
    {
      "epoch": 1.3955116271972655e-05,
      "model_forward_time": 1.6980934143066406,
      "step": 14633
    },
    {
      "epoch": 1.3955116271972655e-05,
      "step": 14633,
      "training_step_time": 8.78045105934143
    },
    {
      "epoch": 1.3956069946289063e-05,
      "model_forward_time": 1.7051777839660645,
      "step": 14634
    },
    {
      "epoch": 1.3956069946289063e-05,
      "step": 14634,
      "training_step_time": 7.628688812255859
    },
    {
      "epoch": 1.3957023620605469e-05,
      "model_forward_time": 1.709583044052124,
      "step": 14635
    },
    {
      "epoch": 1.3957023620605469e-05,
      "step": 14635,
      "training_step_time": 7.573807954788208
    },
    {
      "epoch": 1.3957977294921874e-05,
      "model_forward_time": 1.7062087059020996,
      "step": 14636
    },
    {
      "epoch": 1.3957977294921874e-05,
      "step": 14636,
      "training_step_time": 7.296147346496582
    },
    {
      "epoch": 1.3958930969238282e-05,
      "model_forward_time": 1.6919865608215332,
      "step": 14637
    },
    {
      "epoch": 1.3958930969238282e-05,
      "step": 14637,
      "training_step_time": 7.382061958312988
    },
    {
      "epoch": 1.3959884643554687e-05,
      "model_forward_time": 1.7084808349609375,
      "step": 14638
    },
    {
      "epoch": 1.3959884643554687e-05,
      "step": 14638,
      "training_step_time": 7.6042702198028564
    },
    {
      "epoch": 1.3960838317871093e-05,
      "model_forward_time": 2.233581304550171,
      "step": 14639
    },
    {
      "epoch": 1.3960838317871093e-05,
      "step": 14639,
      "training_step_time": 8.537562608718872
    },
    {
      "epoch": 1.39617919921875e-05,
      "grad_norm": 0.16645774245262146,
      "learning_rate": 9.74813016022317e-06,
      "loss": 0.072,
      "step": 14640
    },
    {
      "epoch": 1.39617919921875e-05,
      "model_forward_time": 1.7005302906036377,
      "step": 14640
    },
    {
      "dynamics_loss_avg": 0.06885079313069582,
      "epoch": 1.39617919921875e-05,
      "step": 14640
    },
    {
      "action_loss_avg": 0.0026794776553288103,
      "epoch": 1.39617919921875e-05,
      "step": 14640
    },
    {
      "epoch": 1.39617919921875e-05,
      "step": 14640,
      "training_step_time": 8.52707839012146
    },
    {
      "epoch": 1.3962745666503906e-05,
      "model_forward_time": 1.7144525051116943,
      "step": 14641
    },
    {
      "epoch": 1.3962745666503906e-05,
      "step": 14641,
      "training_step_time": 8.522421836853027
    },
    {
      "epoch": 1.3963699340820312e-05,
      "model_forward_time": 1.7033660411834717,
      "step": 14642
    },
    {
      "epoch": 1.3963699340820312e-05,
      "step": 14642,
      "training_step_time": 7.732022047042847
    },
    {
      "epoch": 1.396465301513672e-05,
      "model_forward_time": 1.7057850360870361,
      "step": 14643
    },
    {
      "epoch": 1.396465301513672e-05,
      "step": 14643,
      "training_step_time": 7.65128755569458
    },
    {
      "epoch": 1.3965606689453125e-05,
      "model_forward_time": 1.7080097198486328,
      "step": 14644
    },
    {
      "epoch": 1.3965606689453125e-05,
      "step": 14644,
      "training_step_time": 7.563692092895508
    },
    {
      "epoch": 1.396656036376953e-05,
      "model_forward_time": 1.7077546119689941,
      "step": 14645
    },
    {
      "epoch": 1.396656036376953e-05,
      "step": 14645,
      "training_step_time": 7.339442253112793
    },
    {
      "epoch": 1.3967514038085938e-05,
      "model_forward_time": 1.7022743225097656,
      "step": 14646
    },
    {
      "epoch": 1.3967514038085938e-05,
      "step": 14646,
      "training_step_time": 7.522414207458496
    },
    {
      "epoch": 1.3968467712402344e-05,
      "model_forward_time": 1.7050890922546387,
      "step": 14647
    },
    {
      "epoch": 1.3968467712402344e-05,
      "step": 14647,
      "training_step_time": 7.664241790771484
    },
    {
      "epoch": 1.396942138671875e-05,
      "model_forward_time": 1.3471598625183105,
      "step": 14648
    },
    {
      "epoch": 1.396942138671875e-05,
      "step": 14648,
      "training_step_time": 8.844170331954956
    },
    {
      "epoch": 1.3970375061035155e-05,
      "model_forward_time": 2.236304759979248,
      "step": 14649
    },
    {
      "epoch": 1.3970375061035155e-05,
      "step": 14649,
      "training_step_time": 7.774552583694458
    },
    {
      "epoch": 1.3971328735351563e-05,
      "grad_norm": 0.24939553439617157,
      "learning_rate": 9.747611727333734e-06,
      "loss": 0.0675,
      "step": 14650
    },
    {
      "epoch": 1.3971328735351563e-05,
      "model_forward_time": 1.7057182788848877,
      "step": 14650
    },
    {
      "dynamics_loss_avg": 0.06731648296117783,
      "epoch": 1.3971328735351563e-05,
      "step": 14650
    },
    {
      "action_loss_avg": 0.003803509077988565,
      "epoch": 1.3971328735351563e-05,
      "step": 14650
    },
    {
      "epoch": 1.3971328735351563e-05,
      "step": 14650,
      "training_step_time": 7.692328691482544
    },
    {
      "epoch": 1.3972282409667969e-05,
      "model_forward_time": 1.729355812072754,
      "step": 14651
    },
    {
      "epoch": 1.3972282409667969e-05,
      "step": 14651,
      "training_step_time": 7.724714756011963
    },
    {
      "epoch": 1.3973236083984374e-05,
      "model_forward_time": 1.707181692123413,
      "step": 14652
    },
    {
      "epoch": 1.3973236083984374e-05,
      "step": 14652,
      "training_step_time": 8.088807821273804
    },
    {
      "epoch": 1.3974189758300782e-05,
      "model_forward_time": 1.7054526805877686,
      "step": 14653
    },
    {
      "epoch": 1.3974189758300782e-05,
      "step": 14653,
      "training_step_time": 7.354033708572388
    },
    {
      "epoch": 1.3975143432617187e-05,
      "model_forward_time": 1.7094154357910156,
      "step": 14654
    },
    {
      "epoch": 1.3975143432617187e-05,
      "step": 14654,
      "training_step_time": 7.5809526443481445
    },
    {
      "epoch": 1.3976097106933593e-05,
      "model_forward_time": 1.709524393081665,
      "step": 14655
    },
    {
      "epoch": 1.3976097106933593e-05,
      "step": 14655,
      "training_step_time": 7.523497819900513
    },
    {
      "epoch": 1.397705078125e-05,
      "model_forward_time": 1.7125039100646973,
      "step": 14656
    },
    {
      "epoch": 1.397705078125e-05,
      "step": 14656,
      "training_step_time": 7.5509116649627686
    },
    {
      "epoch": 1.3978004455566406e-05,
      "model_forward_time": 1.7186808586120605,
      "step": 14657
    },
    {
      "epoch": 1.3978004455566406e-05,
      "step": 14657,
      "training_step_time": 7.669705629348755
    },
    {
      "epoch": 1.3978958129882812e-05,
      "model_forward_time": 2.2347869873046875,
      "step": 14658
    },
    {
      "epoch": 1.3978958129882812e-05,
      "step": 14658,
      "training_step_time": 9.840887784957886
    },
    {
      "epoch": 1.397991180419922e-05,
      "model_forward_time": 1.7129077911376953,
      "step": 14659
    },
    {
      "epoch": 1.397991180419922e-05,
      "step": 14659,
      "training_step_time": 8.692330598831177
    },
    {
      "epoch": 1.3980865478515625e-05,
      "grad_norm": 0.23739133775234222,
      "learning_rate": 9.74709277525262e-06,
      "loss": 0.0686,
      "step": 14660
    },
    {
      "epoch": 1.3980865478515625e-05,
      "model_forward_time": 1.718228816986084,
      "step": 14660
    },
    {
      "dynamics_loss_avg": 0.0711546067148447,
      "epoch": 1.3980865478515625e-05,
      "step": 14660
    },
    {
      "action_loss_avg": 0.003359978983644396,
      "epoch": 1.3980865478515625e-05,
      "step": 14660
    },
    {
      "epoch": 1.3980865478515625e-05,
      "step": 14660,
      "training_step_time": 8.20953917503357
    },
    {
      "epoch": 1.3981819152832031e-05,
      "model_forward_time": 1.7031903266906738,
      "step": 14661
    },
    {
      "epoch": 1.3981819152832031e-05,
      "step": 14661,
      "training_step_time": 8.723923206329346
    },
    {
      "epoch": 1.3982772827148438e-05,
      "model_forward_time": 1.6987907886505127,
      "step": 14662
    },
    {
      "epoch": 1.3982772827148438e-05,
      "step": 14662,
      "training_step_time": 7.563741683959961
    },
    {
      "epoch": 1.3983726501464844e-05,
      "model_forward_time": 1.7117083072662354,
      "step": 14663
    },
    {
      "epoch": 1.3983726501464844e-05,
      "step": 14663,
      "training_step_time": 7.609407663345337
    },
    {
      "epoch": 1.398468017578125e-05,
      "model_forward_time": 1.7081928253173828,
      "step": 14664
    },
    {
      "epoch": 1.398468017578125e-05,
      "step": 14664,
      "training_step_time": 7.504725694656372
    },
    {
      "epoch": 1.3985633850097655e-05,
      "model_forward_time": 1.7099437713623047,
      "step": 14665
    },
    {
      "epoch": 1.3985633850097655e-05,
      "step": 14665,
      "training_step_time": 7.5420427322387695
    },
    {
      "epoch": 1.3986587524414063e-05,
      "model_forward_time": 1.7146780490875244,
      "step": 14666
    },
    {
      "epoch": 1.3986587524414063e-05,
      "step": 14666,
      "training_step_time": 8.386561632156372
    },
    {
      "epoch": 1.3987541198730469e-05,
      "model_forward_time": 2.240617275238037,
      "step": 14667
    },
    {
      "epoch": 1.3987541198730469e-05,
      "step": 14667,
      "training_step_time": 7.914826154708862
    },
    {
      "epoch": 1.3988494873046874e-05,
      "model_forward_time": 1.3360285758972168,
      "step": 14668
    },
    {
      "epoch": 1.3988494873046874e-05,
      "step": 14668,
      "training_step_time": 8.520948648452759
    },
    {
      "epoch": 1.3989448547363282e-05,
      "model_forward_time": 1.730522871017456,
      "step": 14669
    },
    {
      "epoch": 1.3989448547363282e-05,
      "step": 14669,
      "training_step_time": 7.7080090045928955
    },
    {
      "epoch": 1.3990402221679687e-05,
      "grad_norm": 0.2590319514274597,
      "learning_rate": 9.746573304036576e-06,
      "loss": 0.0681,
      "step": 14670
    },
    {
      "epoch": 1.3990402221679687e-05,
      "model_forward_time": 1.708155632019043,
      "step": 14670
    },
    {
      "dynamics_loss_avg": 0.059647439233958724,
      "epoch": 1.3990402221679687e-05,
      "step": 14670
    },
    {
      "action_loss_avg": 0.0031718793790787457,
      "epoch": 1.3990402221679687e-05,
      "step": 14670
    },
    {
      "epoch": 1.3990402221679687e-05,
      "step": 14670,
      "training_step_time": 9.202035188674927
    },
    {
      "epoch": 1.3991355895996093e-05,
      "model_forward_time": 1.6987791061401367,
      "step": 14671
    },
    {
      "epoch": 1.3991355895996093e-05,
      "step": 14671,
      "training_step_time": 7.653801202774048
    },
    {
      "epoch": 1.39923095703125e-05,
      "model_forward_time": 1.706972599029541,
      "step": 14672
    },
    {
      "epoch": 1.39923095703125e-05,
      "step": 14672,
      "training_step_time": 7.603781700134277
    },
    {
      "epoch": 1.3993263244628906e-05,
      "model_forward_time": 1.3312695026397705,
      "step": 14673
    },
    {
      "epoch": 1.3993263244628906e-05,
      "step": 14673,
      "training_step_time": 7.550142049789429
    },
    {
      "epoch": 1.3994216918945312e-05,
      "model_forward_time": 1.7069993019104004,
      "step": 14674
    },
    {
      "epoch": 1.3994216918945312e-05,
      "step": 14674,
      "training_step_time": 7.634258508682251
    },
    {
      "epoch": 1.399517059326172e-05,
      "model_forward_time": 1.720696210861206,
      "step": 14675
    },
    {
      "epoch": 1.399517059326172e-05,
      "step": 14675,
      "training_step_time": 7.67814564704895
    },
    {
      "epoch": 1.3996124267578125e-05,
      "model_forward_time": 1.7117276191711426,
      "step": 14676
    },
    {
      "epoch": 1.3996124267578125e-05,
      "step": 14676,
      "training_step_time": 8.289588451385498
    },
    {
      "epoch": 1.3997077941894531e-05,
      "model_forward_time": 1.7028307914733887,
      "step": 14677
    },
    {
      "epoch": 1.3997077941894531e-05,
      "step": 14677,
      "training_step_time": 7.790101528167725
    },
    {
      "epoch": 1.3998031616210938e-05,
      "model_forward_time": 1.737619400024414,
      "step": 14678
    },
    {
      "epoch": 1.3998031616210938e-05,
      "step": 14678,
      "training_step_time": 8.43044662475586
    },
    {
      "epoch": 1.3998985290527344e-05,
      "model_forward_time": 1.7142252922058105,
      "step": 14679
    },
    {
      "epoch": 1.3998985290527344e-05,
      "step": 14679,
      "training_step_time": 8.80093765258789
    },
    {
      "epoch": 1.399993896484375e-05,
      "grad_norm": 0.20631295442581177,
      "learning_rate": 9.746053313742413e-06,
      "loss": 0.0702,
      "step": 14680
    },
    {
      "epoch": 1.399993896484375e-05,
      "model_forward_time": 1.713413953781128,
      "step": 14680
    },
    {
      "dynamics_loss_avg": 0.0651008978486061,
      "epoch": 1.399993896484375e-05,
      "step": 14680
    },
    {
      "action_loss_avg": 0.004678121121833101,
      "epoch": 1.399993896484375e-05,
      "step": 14680
    },
    {
      "epoch": 1.399993896484375e-05,
      "step": 14680,
      "training_step_time": 7.727639675140381
    },
    {
      "epoch": 1.4000892639160155e-05,
      "model_forward_time": 1.7204785346984863,
      "step": 14681
    },
    {
      "epoch": 1.4000892639160155e-05,
      "step": 14681,
      "training_step_time": 7.618136167526245
    },
    {
      "epoch": 1.4001846313476563e-05,
      "model_forward_time": 1.7191975116729736,
      "step": 14682
    },
    {
      "epoch": 1.4001846313476563e-05,
      "step": 14682,
      "training_step_time": 7.544863939285278
    },
    {
      "epoch": 1.4002799987792969e-05,
      "model_forward_time": 1.7231659889221191,
      "step": 14683
    },
    {
      "epoch": 1.4002799987792969e-05,
      "step": 14683,
      "training_step_time": 7.535504102706909
    },
    {
      "epoch": 1.4003753662109374e-05,
      "model_forward_time": 1.719670057296753,
      "step": 14684
    },
    {
      "epoch": 1.4003753662109374e-05,
      "step": 14684,
      "training_step_time": 7.707862138748169
    },
    {
      "epoch": 1.4004707336425782e-05,
      "model_forward_time": 1.721597671508789,
      "step": 14685
    },
    {
      "epoch": 1.4004707336425782e-05,
      "step": 14685,
      "training_step_time": 8.754908084869385
    },
    {
      "epoch": 1.4005661010742187e-05,
      "model_forward_time": 1.8483998775482178,
      "step": 14686
    },
    {
      "epoch": 1.4005661010742187e-05,
      "step": 14686,
      "training_step_time": 9.101254940032959
    },
    {
      "epoch": 1.4006614685058593e-05,
      "model_forward_time": 1.740241289138794,
      "step": 14687
    },
    {
      "epoch": 1.4006614685058593e-05,
      "step": 14687,
      "training_step_time": 8.253166913986206
    },
    {
      "epoch": 1.4007568359375e-05,
      "model_forward_time": 1.7244131565093994,
      "step": 14688
    },
    {
      "epoch": 1.4007568359375e-05,
      "step": 14688,
      "training_step_time": 7.814082622528076
    },
    {
      "epoch": 1.4008522033691406e-05,
      "model_forward_time": 1.7175908088684082,
      "step": 14689
    },
    {
      "epoch": 1.4008522033691406e-05,
      "step": 14689,
      "training_step_time": 8.431877613067627
    },
    {
      "epoch": 1.4009475708007812e-05,
      "grad_norm": 0.3221113085746765,
      "learning_rate": 9.745532804426994e-06,
      "loss": 0.0706,
      "step": 14690
    },
    {
      "epoch": 1.4009475708007812e-05,
      "model_forward_time": 1.7079224586486816,
      "step": 14690
    },
    {
      "dynamics_loss_avg": 0.06250621685758233,
      "epoch": 1.4009475708007812e-05,
      "step": 14690
    },
    {
      "action_loss_avg": 0.0034759262460283937,
      "epoch": 1.4009475708007812e-05,
      "step": 14690
    },
    {
      "epoch": 1.4009475708007812e-05,
      "step": 14690,
      "training_step_time": 7.593951463699341
    },
    {
      "epoch": 1.401042938232422e-05,
      "model_forward_time": 1.691650390625,
      "step": 14691
    },
    {
      "epoch": 1.401042938232422e-05,
      "step": 14691,
      "training_step_time": 7.261640548706055
    },
    {
      "epoch": 1.4011383056640625e-05,
      "model_forward_time": 1.7090673446655273,
      "step": 14692
    },
    {
      "epoch": 1.4011383056640625e-05,
      "step": 14692,
      "training_step_time": 7.513412714004517
    },
    {
      "epoch": 1.4012336730957031e-05,
      "model_forward_time": 1.7044939994812012,
      "step": 14693
    },
    {
      "epoch": 1.4012336730957031e-05,
      "step": 14693,
      "training_step_time": 7.6088337898254395
    },
    {
      "epoch": 1.4013290405273438e-05,
      "model_forward_time": 1.7079124450683594,
      "step": 14694
    },
    {
      "epoch": 1.4013290405273438e-05,
      "step": 14694,
      "training_step_time": 8.617207050323486
    },
    {
      "epoch": 1.4014244079589844e-05,
      "model_forward_time": 2.235266923904419,
      "step": 14695
    },
    {
      "epoch": 1.4014244079589844e-05,
      "step": 14695,
      "training_step_time": 10.618559122085571
    },
    {
      "epoch": 1.401519775390625e-05,
      "model_forward_time": 1.7269642353057861,
      "step": 14696
    },
    {
      "epoch": 1.401519775390625e-05,
      "step": 14696,
      "training_step_time": 7.760395050048828
    },
    {
      "epoch": 1.4016151428222655e-05,
      "model_forward_time": 1.7057456970214844,
      "step": 14697
    },
    {
      "epoch": 1.4016151428222655e-05,
      "step": 14697,
      "training_step_time": 7.676225662231445
    },
    {
      "epoch": 1.4017105102539063e-05,
      "model_forward_time": 1.7089810371398926,
      "step": 14698
    },
    {
      "epoch": 1.4017105102539063e-05,
      "step": 14698,
      "training_step_time": 7.6317832469940186
    },
    {
      "epoch": 1.4018058776855469e-05,
      "model_forward_time": 1.7091937065124512,
      "step": 14699
    },
    {
      "epoch": 1.4018058776855469e-05,
      "step": 14699,
      "training_step_time": 7.6964874267578125
    },
    {
      "epoch": 1.4019012451171874e-05,
      "grad_norm": 0.18430808186531067,
      "learning_rate": 9.745011776147243e-06,
      "loss": 0.0714,
      "step": 14700
    },
    {
      "epoch": 1.4019012451171874e-05,
      "model_forward_time": 1.701434850692749,
      "step": 14700
    },
    {
      "dynamics_loss_avg": 0.05854221880435943,
      "epoch": 1.4019012451171874e-05,
      "step": 14700
    },
    {
      "action_loss_avg": 0.0022044516168534755,
      "epoch": 1.4019012451171874e-05,
      "step": 14700
    },
    {
      "epoch": 1.4019012451171874e-05,
      "step": 14700,
      "training_step_time": 7.575026512145996
    },
    {
      "epoch": 1.4019966125488282e-05,
      "model_forward_time": 1.7060647010803223,
      "step": 14701
    },
    {
      "epoch": 1.4019966125488282e-05,
      "step": 14701,
      "training_step_time": 8.573726892471313
    },
    {
      "epoch": 1.4020919799804687e-05,
      "model_forward_time": 1.7025856971740723,
      "step": 14702
    },
    {
      "epoch": 1.4020919799804687e-05,
      "step": 14702,
      "training_step_time": 8.293651103973389
    },
    {
      "epoch": 1.4021873474121093e-05,
      "model_forward_time": 1.725471019744873,
      "step": 14703
    },
    {
      "epoch": 1.4021873474121093e-05,
      "step": 14703,
      "training_step_time": 8.617379903793335
    },
    {
      "epoch": 1.40228271484375e-05,
      "model_forward_time": 2.251476526260376,
      "step": 14704
    },
    {
      "epoch": 1.40228271484375e-05,
      "step": 14704,
      "training_step_time": 7.768877744674683
    },
    {
      "epoch": 1.4023780822753906e-05,
      "model_forward_time": 1.7163362503051758,
      "step": 14705
    },
    {
      "epoch": 1.4023780822753906e-05,
      "step": 14705,
      "training_step_time": 8.752650499343872
    },
    {
      "epoch": 1.4024734497070312e-05,
      "model_forward_time": 1.698976755142212,
      "step": 14706
    },
    {
      "epoch": 1.4024734497070312e-05,
      "step": 14706,
      "training_step_time": 7.665913343429565
    },
    {
      "epoch": 1.402568817138672e-05,
      "model_forward_time": 1.708824872970581,
      "step": 14707
    },
    {
      "epoch": 1.402568817138672e-05,
      "step": 14707,
      "training_step_time": 8.706754446029663
    },
    {
      "epoch": 1.4026641845703125e-05,
      "model_forward_time": 1.7089111804962158,
      "step": 14708
    },
    {
      "epoch": 1.4026641845703125e-05,
      "step": 14708,
      "training_step_time": 7.687795162200928
    },
    {
      "epoch": 1.4027595520019531e-05,
      "model_forward_time": 1.7015986442565918,
      "step": 14709
    },
    {
      "epoch": 1.4027595520019531e-05,
      "step": 14709,
      "training_step_time": 7.602429389953613
    },
    {
      "epoch": 1.4028549194335938e-05,
      "grad_norm": 0.28607943654060364,
      "learning_rate": 9.744490228960137e-06,
      "loss": 0.068,
      "step": 14710
    },
    {
      "epoch": 1.4028549194335938e-05,
      "model_forward_time": 1.706986904144287,
      "step": 14710
    },
    {
      "dynamics_loss_avg": 0.05053628738969564,
      "epoch": 1.4028549194335938e-05,
      "step": 14710
    },
    {
      "action_loss_avg": 0.002370150719070807,
      "epoch": 1.4028549194335938e-05,
      "step": 14710
    },
    {
      "epoch": 1.4028549194335938e-05,
      "step": 14710,
      "training_step_time": 7.550023078918457
    },
    {
      "epoch": 1.4029502868652344e-05,
      "model_forward_time": 1.709815263748169,
      "step": 14711
    },
    {
      "epoch": 1.4029502868652344e-05,
      "step": 14711,
      "training_step_time": 8.653163433074951
    },
    {
      "epoch": 1.403045654296875e-05,
      "model_forward_time": 1.7241618633270264,
      "step": 14712
    },
    {
      "epoch": 1.403045654296875e-05,
      "step": 14712,
      "training_step_time": 7.712382555007935
    },
    {
      "epoch": 1.4031410217285155e-05,
      "model_forward_time": 2.2670819759368896,
      "step": 14713
    },
    {
      "epoch": 1.4031410217285155e-05,
      "step": 14713,
      "training_step_time": 8.198078155517578
    },
    {
      "epoch": 1.4032363891601563e-05,
      "model_forward_time": 1.7059741020202637,
      "step": 14714
    },
    {
      "epoch": 1.4032363891601563e-05,
      "step": 14714,
      "training_step_time": 8.730612516403198
    },
    {
      "epoch": 1.4033317565917969e-05,
      "model_forward_time": 1.7024438381195068,
      "step": 14715
    },
    {
      "epoch": 1.4033317565917969e-05,
      "step": 14715,
      "training_step_time": 8.626406908035278
    },
    {
      "epoch": 1.4034271240234374e-05,
      "model_forward_time": 1.3310542106628418,
      "step": 14716
    },
    {
      "epoch": 1.4034271240234374e-05,
      "step": 14716,
      "training_step_time": 7.674680709838867
    },
    {
      "epoch": 1.4035224914550782e-05,
      "model_forward_time": 1.7032279968261719,
      "step": 14717
    },
    {
      "epoch": 1.4035224914550782e-05,
      "step": 14717,
      "training_step_time": 7.622815370559692
    },
    {
      "epoch": 1.4036178588867187e-05,
      "model_forward_time": 1.7089123725891113,
      "step": 14718
    },
    {
      "epoch": 1.4036178588867187e-05,
      "step": 14718,
      "training_step_time": 7.652757406234741
    },
    {
      "epoch": 1.4037132263183593e-05,
      "model_forward_time": 1.7214500904083252,
      "step": 14719
    },
    {
      "epoch": 1.4037132263183593e-05,
      "step": 14719,
      "training_step_time": 7.528942108154297
    },
    {
      "epoch": 1.40380859375e-05,
      "grad_norm": 0.24000044167041779,
      "learning_rate": 9.743968162922714e-06,
      "loss": 0.0726,
      "step": 14720
    },
    {
      "epoch": 1.40380859375e-05,
      "model_forward_time": 1.7024755477905273,
      "step": 14720
    },
    {
      "dynamics_loss_avg": 0.07382498048245907,
      "epoch": 1.40380859375e-05,
      "step": 14720
    },
    {
      "action_loss_avg": 0.0036780889320652934,
      "epoch": 1.40380859375e-05,
      "step": 14720
    },
    {
      "epoch": 1.40380859375e-05,
      "step": 14720,
      "training_step_time": 8.271380424499512
    },
    {
      "epoch": 1.4039039611816406e-05,
      "model_forward_time": 1.7167608737945557,
      "step": 14721
    },
    {
      "epoch": 1.4039039611816406e-05,
      "step": 14721,
      "training_step_time": 7.658283948898315
    },
    {
      "epoch": 1.4039993286132812e-05,
      "model_forward_time": 1.707129955291748,
      "step": 14722
    },
    {
      "epoch": 1.4039993286132812e-05,
      "step": 14722,
      "training_step_time": 8.429472923278809
    },
    {
      "epoch": 1.404094696044922e-05,
      "model_forward_time": 2.236452102661133,
      "step": 14723
    },
    {
      "epoch": 1.404094696044922e-05,
      "step": 14723,
      "training_step_time": 7.698307037353516
    },
    {
      "epoch": 1.4041900634765625e-05,
      "model_forward_time": 1.7313613891601562,
      "step": 14724
    },
    {
      "epoch": 1.4041900634765625e-05,
      "step": 14724,
      "training_step_time": 7.848697185516357
    },
    {
      "epoch": 1.4042854309082031e-05,
      "model_forward_time": 1.7068815231323242,
      "step": 14725
    },
    {
      "epoch": 1.4042854309082031e-05,
      "step": 14725,
      "training_step_time": 8.650348901748657
    },
    {
      "epoch": 1.4043807983398438e-05,
      "model_forward_time": 1.7075963020324707,
      "step": 14726
    },
    {
      "epoch": 1.4043807983398438e-05,
      "step": 14726,
      "training_step_time": 7.622157573699951
    },
    {
      "epoch": 1.4044761657714844e-05,
      "model_forward_time": 1.7067139148712158,
      "step": 14727
    },
    {
      "epoch": 1.4044761657714844e-05,
      "step": 14727,
      "training_step_time": 7.619457244873047
    },
    {
      "epoch": 1.404571533203125e-05,
      "model_forward_time": 1.7110316753387451,
      "step": 14728
    },
    {
      "epoch": 1.404571533203125e-05,
      "step": 14728,
      "training_step_time": 7.298780918121338
    },
    {
      "epoch": 1.4046669006347655e-05,
      "model_forward_time": 1.706479549407959,
      "step": 14729
    },
    {
      "epoch": 1.4046669006347655e-05,
      "step": 14729,
      "training_step_time": 8.251412391662598
    },
    {
      "epoch": 1.4047622680664063e-05,
      "grad_norm": 0.27334707975387573,
      "learning_rate": 9.743445578092064e-06,
      "loss": 0.0695,
      "step": 14730
    },
    {
      "epoch": 1.4047622680664063e-05,
      "model_forward_time": 1.70294189453125,
      "step": 14730
    },
    {
      "dynamics_loss_avg": 0.0505992135964334,
      "epoch": 1.4047622680664063e-05,
      "step": 14730
    },
    {
      "action_loss_avg": 0.0036626643384806814,
      "epoch": 1.4047622680664063e-05,
      "step": 14730
    },
    {
      "epoch": 1.4047622680664063e-05,
      "step": 14730,
      "training_step_time": 7.637827157974243
    },
    {
      "epoch": 1.4048576354980469e-05,
      "model_forward_time": 1.6965875625610352,
      "step": 14731
    },
    {
      "epoch": 1.4048576354980469e-05,
      "step": 14731,
      "training_step_time": 7.555387496948242
    },
    {
      "epoch": 1.4049530029296874e-05,
      "model_forward_time": 2.2582709789276123,
      "step": 14732
    },
    {
      "epoch": 1.4049530029296874e-05,
      "step": 14732,
      "training_step_time": 8.412631273269653
    },
    {
      "epoch": 1.4050483703613282e-05,
      "model_forward_time": 1.7372229099273682,
      "step": 14733
    },
    {
      "epoch": 1.4050483703613282e-05,
      "step": 14733,
      "training_step_time": 7.795636415481567
    },
    {
      "epoch": 1.4051437377929687e-05,
      "model_forward_time": 1.7161788940429688,
      "step": 14734
    },
    {
      "epoch": 1.4051437377929687e-05,
      "step": 14734,
      "training_step_time": 8.496256828308105
    },
    {
      "epoch": 1.4052391052246093e-05,
      "model_forward_time": 1.7192246913909912,
      "step": 14735
    },
    {
      "epoch": 1.4052391052246093e-05,
      "step": 14735,
      "training_step_time": 7.589773416519165
    },
    {
      "epoch": 1.40533447265625e-05,
      "model_forward_time": 1.7109997272491455,
      "step": 14736
    },
    {
      "epoch": 1.40533447265625e-05,
      "step": 14736,
      "training_step_time": 7.713292360305786
    },
    {
      "epoch": 1.4054298400878906e-05,
      "model_forward_time": 1.709733247756958,
      "step": 14737
    },
    {
      "epoch": 1.4054298400878906e-05,
      "step": 14737,
      "training_step_time": 7.302180051803589
    },
    {
      "epoch": 1.4055252075195312e-05,
      "model_forward_time": 1.7125725746154785,
      "step": 14738
    },
    {
      "epoch": 1.4055252075195312e-05,
      "step": 14738,
      "training_step_time": 8.530505180358887
    },
    {
      "epoch": 1.405620574951172e-05,
      "model_forward_time": 1.7089459896087646,
      "step": 14739
    },
    {
      "epoch": 1.405620574951172e-05,
      "step": 14739,
      "training_step_time": 7.625677824020386
    },
    {
      "epoch": 1.4057159423828125e-05,
      "grad_norm": 0.28473031520843506,
      "learning_rate": 9.742922474525338e-06,
      "loss": 0.0663,
      "step": 14740
    },
    {
      "epoch": 1.4057159423828125e-05,
      "model_forward_time": 1.718853235244751,
      "step": 14740
    },
    {
      "dynamics_loss_avg": 0.07140521854162216,
      "epoch": 1.4057159423828125e-05,
      "step": 14740
    },
    {
      "action_loss_avg": 0.0035452018026262523,
      "epoch": 1.4057159423828125e-05,
      "step": 14740
    },
    {
      "epoch": 1.4057159423828125e-05,
      "step": 14740,
      "training_step_time": 9.166532516479492
    },
    {
      "epoch": 1.4058113098144531e-05,
      "model_forward_time": 2.239994525909424,
      "step": 14741
    },
    {
      "epoch": 1.4058113098144531e-05,
      "step": 14741,
      "training_step_time": 10.999730348587036
    },
    {
      "epoch": 1.4059066772460938e-05,
      "model_forward_time": 1.7144513130187988,
      "step": 14742
    },
    {
      "epoch": 1.4059066772460938e-05,
      "step": 14742,
      "training_step_time": 7.919640302658081
    },
    {
      "epoch": 1.4060020446777344e-05,
      "model_forward_time": 1.6905527114868164,
      "step": 14743
    },
    {
      "epoch": 1.4060020446777344e-05,
      "step": 14743,
      "training_step_time": 7.7528395652771
    },
    {
      "epoch": 1.406097412109375e-05,
      "model_forward_time": 1.7100722789764404,
      "step": 14744
    },
    {
      "epoch": 1.406097412109375e-05,
      "step": 14744,
      "training_step_time": 7.747035980224609
    },
    {
      "epoch": 1.4061927795410156e-05,
      "model_forward_time": 1.7075421810150146,
      "step": 14745
    },
    {
      "epoch": 1.4061927795410156e-05,
      "step": 14745,
      "training_step_time": 7.810405254364014
    },
    {
      "epoch": 1.4062881469726563e-05,
      "model_forward_time": 1.6887242794036865,
      "step": 14746
    },
    {
      "epoch": 1.4062881469726563e-05,
      "step": 14746,
      "training_step_time": 7.445807695388794
    },
    {
      "epoch": 1.4063835144042969e-05,
      "model_forward_time": 1.7082691192626953,
      "step": 14747
    },
    {
      "epoch": 1.4063835144042969e-05,
      "step": 14747,
      "training_step_time": 7.71166729927063
    },
    {
      "epoch": 1.4064788818359374e-05,
      "model_forward_time": 1.7035229206085205,
      "step": 14748
    },
    {
      "epoch": 1.4064788818359374e-05,
      "step": 14748,
      "training_step_time": 8.669785499572754
    },
    {
      "epoch": 1.4065742492675782e-05,
      "model_forward_time": 1.7244093418121338,
      "step": 14749
    },
    {
      "epoch": 1.4065742492675782e-05,
      "step": 14749,
      "training_step_time": 8.694697380065918
    },
    {
      "epoch": 1.4066696166992188e-05,
      "grad_norm": 0.15201279520988464,
      "learning_rate": 9.742398852279742e-06,
      "loss": 0.0718,
      "step": 14750
    },
    {
      "epoch": 1.4066696166992188e-05,
      "model_forward_time": 2.2368710041046143,
      "step": 14750
    },
    {
      "dynamics_loss_avg": 0.06552947759628296,
      "epoch": 1.4066696166992188e-05,
      "step": 14750
    },
    {
      "action_loss_avg": 0.002129623794462532,
      "epoch": 1.4066696166992188e-05,
      "step": 14750
    },
    {
      "epoch": 1.4066696166992188e-05,
      "step": 14750,
      "training_step_time": 7.827255010604858
    },
    {
      "epoch": 1.4067649841308593e-05,
      "model_forward_time": 1.7083525657653809,
      "step": 14751
    },
    {
      "epoch": 1.4067649841308593e-05,
      "step": 14751,
      "training_step_time": 8.954447984695435
    },
    {
      "epoch": 1.4068603515625e-05,
      "model_forward_time": 1.7012062072753906,
      "step": 14752
    },
    {
      "epoch": 1.4068603515625e-05,
      "step": 14752,
      "training_step_time": 7.952116966247559
    },
    {
      "epoch": 1.4069557189941406e-05,
      "model_forward_time": 1.698711633682251,
      "step": 14753
    },
    {
      "epoch": 1.4069557189941406e-05,
      "step": 14753,
      "training_step_time": 8.68512511253357
    },
    {
      "epoch": 1.4070510864257812e-05,
      "model_forward_time": 1.7053542137145996,
      "step": 14754
    },
    {
      "epoch": 1.4070510864257812e-05,
      "step": 14754,
      "training_step_time": 7.783635139465332
    },
    {
      "epoch": 1.407146453857422e-05,
      "model_forward_time": 1.7054755687713623,
      "step": 14755
    },
    {
      "epoch": 1.407146453857422e-05,
      "step": 14755,
      "training_step_time": 7.776740074157715
    },
    {
      "epoch": 1.4072418212890625e-05,
      "model_forward_time": 1.7091257572174072,
      "step": 14756
    },
    {
      "epoch": 1.4072418212890625e-05,
      "step": 14756,
      "training_step_time": 7.707783937454224
    },
    {
      "epoch": 1.4073371887207031e-05,
      "model_forward_time": 1.7099003791809082,
      "step": 14757
    },
    {
      "epoch": 1.4073371887207031e-05,
      "step": 14757,
      "training_step_time": 8.656770706176758
    },
    {
      "epoch": 1.4074325561523437e-05,
      "model_forward_time": 1.7362699508666992,
      "step": 14758
    },
    {
      "epoch": 1.4074325561523437e-05,
      "step": 14758,
      "training_step_time": 8.62505555152893
    },
    {
      "epoch": 1.4075279235839844e-05,
      "model_forward_time": 2.248453140258789,
      "step": 14759
    },
    {
      "epoch": 1.4075279235839844e-05,
      "step": 14759,
      "training_step_time": 7.877396821975708
    },
    {
      "epoch": 1.407623291015625e-05,
      "grad_norm": 0.2875256836414337,
      "learning_rate": 9.741874711412535e-06,
      "loss": 0.0681,
      "step": 14760
    },
    {
      "epoch": 1.407623291015625e-05,
      "model_forward_time": 1.6994283199310303,
      "step": 14760
    },
    {
      "dynamics_loss_avg": 0.042949285730719565,
      "epoch": 1.407623291015625e-05,
      "step": 14760
    },
    {
      "action_loss_avg": 0.0022008540749084205,
      "epoch": 1.407623291015625e-05,
      "step": 14760
    },
    {
      "epoch": 1.407623291015625e-05,
      "step": 14760,
      "training_step_time": 9.3087637424469
    },
    {
      "epoch": 1.4077186584472656e-05,
      "model_forward_time": 1.7292146682739258,
      "step": 14761
    },
    {
      "epoch": 1.4077186584472656e-05,
      "step": 14761,
      "training_step_time": 7.850966691970825
    },
    {
      "epoch": 1.4078140258789063e-05,
      "model_forward_time": 1.70603609085083,
      "step": 14762
    },
    {
      "epoch": 1.4078140258789063e-05,
      "step": 14762,
      "training_step_time": 8.468397617340088
    },
    {
      "epoch": 1.4079093933105469e-05,
      "model_forward_time": 1.7118678092956543,
      "step": 14763
    },
    {
      "epoch": 1.4079093933105469e-05,
      "step": 14763,
      "training_step_time": 8.429399013519287
    },
    {
      "epoch": 1.4080047607421874e-05,
      "model_forward_time": 1.7116179466247559,
      "step": 14764
    },
    {
      "epoch": 1.4080047607421874e-05,
      "step": 14764,
      "training_step_time": 7.788551330566406
    },
    {
      "epoch": 1.4081001281738282e-05,
      "model_forward_time": 1.709010362625122,
      "step": 14765
    },
    {
      "epoch": 1.4081001281738282e-05,
      "step": 14765,
      "training_step_time": 7.632726669311523
    },
    {
      "epoch": 1.4081954956054688e-05,
      "model_forward_time": 1.7094542980194092,
      "step": 14766
    },
    {
      "epoch": 1.4081954956054688e-05,
      "step": 14766,
      "training_step_time": 8.653148889541626
    },
    {
      "epoch": 1.4082908630371093e-05,
      "model_forward_time": 1.7340455055236816,
      "step": 14767
    },
    {
      "epoch": 1.4082908630371093e-05,
      "step": 14767,
      "training_step_time": 7.756634712219238
    },
    {
      "epoch": 1.40838623046875e-05,
      "model_forward_time": 1.7115683555603027,
      "step": 14768
    },
    {
      "epoch": 1.40838623046875e-05,
      "step": 14768,
      "training_step_time": 8.986209869384766
    },
    {
      "epoch": 1.4084815979003906e-05,
      "model_forward_time": 1.7036633491516113,
      "step": 14769
    },
    {
      "epoch": 1.4084815979003906e-05,
      "step": 14769,
      "training_step_time": 7.869550704956055
    },
    {
      "epoch": 1.4085769653320312e-05,
      "grad_norm": 0.2622789144515991,
      "learning_rate": 9.741350051981043e-06,
      "loss": 0.0647,
      "step": 14770
    },
    {
      "epoch": 1.4085769653320312e-05,
      "model_forward_time": 1.7260346412658691,
      "step": 14770
    },
    {
      "dynamics_loss_avg": 0.06005593165755272,
      "epoch": 1.4085769653320312e-05,
      "step": 14770
    },
    {
      "action_loss_avg": 0.004349900136003271,
      "epoch": 1.4085769653320312e-05,
      "step": 14770
    },
    {
      "epoch": 1.4085769653320312e-05,
      "step": 14770,
      "training_step_time": 7.9798264503479
    },
    {
      "epoch": 1.408672332763672e-05,
      "model_forward_time": 1.7070856094360352,
      "step": 14771
    },
    {
      "epoch": 1.408672332763672e-05,
      "step": 14771,
      "training_step_time": 7.71520733833313
    },
    {
      "epoch": 1.4087677001953125e-05,
      "model_forward_time": 1.3363940715789795,
      "step": 14772
    },
    {
      "epoch": 1.4087677001953125e-05,
      "step": 14772,
      "training_step_time": 8.324280738830566
    },
    {
      "epoch": 1.4088630676269531e-05,
      "model_forward_time": 1.7113940715789795,
      "step": 14773
    },
    {
      "epoch": 1.4088630676269531e-05,
      "step": 14773,
      "training_step_time": 7.778934955596924
    },
    {
      "epoch": 1.4089584350585937e-05,
      "model_forward_time": 1.707383394241333,
      "step": 14774
    },
    {
      "epoch": 1.4089584350585937e-05,
      "step": 14774,
      "training_step_time": 8.688188314437866
    },
    {
      "epoch": 1.4090538024902344e-05,
      "model_forward_time": 1.707470178604126,
      "step": 14775
    },
    {
      "epoch": 1.4090538024902344e-05,
      "step": 14775,
      "training_step_time": 8.382525205612183
    },
    {
      "epoch": 1.409149169921875e-05,
      "model_forward_time": 1.6981761455535889,
      "step": 14776
    },
    {
      "epoch": 1.409149169921875e-05,
      "step": 14776,
      "training_step_time": 7.811569929122925
    },
    {
      "epoch": 1.4092445373535156e-05,
      "model_forward_time": 1.7057180404663086,
      "step": 14777
    },
    {
      "epoch": 1.4092445373535156e-05,
      "step": 14777,
      "training_step_time": 7.892484903335571
    },
    {
      "epoch": 1.4093399047851563e-05,
      "model_forward_time": 2.2346127033233643,
      "step": 14778
    },
    {
      "epoch": 1.4093399047851563e-05,
      "step": 14778,
      "training_step_time": 8.497483253479004
    },
    {
      "epoch": 1.4094352722167969e-05,
      "model_forward_time": 1.725292682647705,
      "step": 14779
    },
    {
      "epoch": 1.4094352722167969e-05,
      "step": 14779,
      "training_step_time": 7.861050844192505
    },
    {
      "epoch": 1.4095306396484374e-05,
      "grad_norm": 0.1380564570426941,
      "learning_rate": 9.740824874042633e-06,
      "loss": 0.0634,
      "step": 14780
    },
    {
      "epoch": 1.4095306396484374e-05,
      "model_forward_time": 1.700791597366333,
      "step": 14780
    },
    {
      "dynamics_loss_avg": 0.06152764754369855,
      "epoch": 1.4095306396484374e-05,
      "step": 14780
    },
    {
      "action_loss_avg": 0.003996393192210235,
      "epoch": 1.4095306396484374e-05,
      "step": 14780
    },
    {
      "epoch": 1.4095306396484374e-05,
      "step": 14780,
      "training_step_time": 7.831857681274414
    },
    {
      "epoch": 1.4096260070800782e-05,
      "model_forward_time": 1.7059290409088135,
      "step": 14781
    },
    {
      "epoch": 1.4096260070800782e-05,
      "step": 14781,
      "training_step_time": 8.350212097167969
    },
    {
      "epoch": 1.4097213745117188e-05,
      "model_forward_time": 1.7089571952819824,
      "step": 14782
    },
    {
      "epoch": 1.4097213745117188e-05,
      "step": 14782,
      "training_step_time": 7.819272994995117
    },
    {
      "epoch": 1.4098167419433593e-05,
      "model_forward_time": 1.7129521369934082,
      "step": 14783
    },
    {
      "epoch": 1.4098167419433593e-05,
      "step": 14783,
      "training_step_time": 7.524726152420044
    },
    {
      "epoch": 1.409912109375e-05,
      "model_forward_time": 1.704953670501709,
      "step": 14784
    },
    {
      "epoch": 1.409912109375e-05,
      "step": 14784,
      "training_step_time": 7.752455472946167
    },
    {
      "epoch": 1.4100074768066406e-05,
      "model_forward_time": 1.7093589305877686,
      "step": 14785
    },
    {
      "epoch": 1.4100074768066406e-05,
      "step": 14785,
      "training_step_time": 7.884886264801025
    },
    {
      "epoch": 1.4101028442382812e-05,
      "model_forward_time": 1.7289695739746094,
      "step": 14786
    },
    {
      "epoch": 1.4101028442382812e-05,
      "step": 14786,
      "training_step_time": 8.516251802444458
    },
    {
      "epoch": 1.410198211669922e-05,
      "model_forward_time": 2.233581304550171,
      "step": 14787
    },
    {
      "epoch": 1.410198211669922e-05,
      "step": 14787,
      "training_step_time": 9.395723581314087
    },
    {
      "epoch": 1.4102935791015625e-05,
      "model_forward_time": 1.7097668647766113,
      "step": 14788
    },
    {
      "epoch": 1.4102935791015625e-05,
      "step": 14788,
      "training_step_time": 7.973504543304443
    },
    {
      "epoch": 1.4103889465332031e-05,
      "model_forward_time": 1.7248599529266357,
      "step": 14789
    },
    {
      "epoch": 1.4103889465332031e-05,
      "step": 14789,
      "training_step_time": 7.918910264968872
    },
    {
      "epoch": 1.4104843139648437e-05,
      "grad_norm": 0.11834064871072769,
      "learning_rate": 9.740299177654747e-06,
      "loss": 0.0634,
      "step": 14790
    },
    {
      "epoch": 1.4104843139648437e-05,
      "model_forward_time": 1.6998789310455322,
      "step": 14790
    },
    {
      "dynamics_loss_avg": 0.08981158062815667,
      "epoch": 1.4104843139648437e-05,
      "step": 14790
    },
    {
      "action_loss_avg": 0.005546331952791661,
      "epoch": 1.4104843139648437e-05,
      "step": 14790
    },
    {
      "epoch": 1.4104843139648437e-05,
      "step": 14790,
      "training_step_time": 10.359929084777832
    },
    {
      "epoch": 1.4105796813964844e-05,
      "model_forward_time": 1.7181832790374756,
      "step": 14791
    },
    {
      "epoch": 1.4105796813964844e-05,
      "step": 14791,
      "training_step_time": 7.854090452194214
    },
    {
      "epoch": 1.410675048828125e-05,
      "model_forward_time": 1.7118644714355469,
      "step": 14792
    },
    {
      "epoch": 1.410675048828125e-05,
      "step": 14792,
      "training_step_time": 7.575685501098633
    },
    {
      "epoch": 1.4107704162597656e-05,
      "model_forward_time": 1.7176759243011475,
      "step": 14793
    },
    {
      "epoch": 1.4107704162597656e-05,
      "step": 14793,
      "training_step_time": 7.733175754547119
    },
    {
      "epoch": 1.4108657836914063e-05,
      "model_forward_time": 1.7134020328521729,
      "step": 14794
    },
    {
      "epoch": 1.4108657836914063e-05,
      "step": 14794,
      "training_step_time": 7.8331379890441895
    },
    {
      "epoch": 1.4109611511230469e-05,
      "model_forward_time": 1.717000961303711,
      "step": 14795
    },
    {
      "epoch": 1.4109611511230469e-05,
      "step": 14795,
      "training_step_time": 8.320175886154175
    },
    {
      "epoch": 1.4110565185546874e-05,
      "model_forward_time": 2.239844560623169,
      "step": 14796
    },
    {
      "epoch": 1.4110565185546874e-05,
      "step": 14796,
      "training_step_time": 7.85078239440918
    },
    {
      "epoch": 1.4111518859863282e-05,
      "model_forward_time": 1.703566074371338,
      "step": 14797
    },
    {
      "epoch": 1.4111518859863282e-05,
      "step": 14797,
      "training_step_time": 7.929532766342163
    },
    {
      "epoch": 1.4112472534179688e-05,
      "model_forward_time": 1.7271294593811035,
      "step": 14798
    },
    {
      "epoch": 1.4112472534179688e-05,
      "step": 14798,
      "training_step_time": 8.480246305465698
    },
    {
      "epoch": 1.4113426208496093e-05,
      "model_forward_time": 1.7111268043518066,
      "step": 14799
    },
    {
      "epoch": 1.4113426208496093e-05,
      "step": 14799,
      "training_step_time": 8.687377691268921
    },
    {
      "epoch": 1.41143798828125e-05,
      "grad_norm": 0.1590777188539505,
      "learning_rate": 9.739772962874867e-06,
      "loss": 0.0667,
      "step": 14800
    },
    {
      "epoch": 1.41143798828125e-05,
      "model_forward_time": 1.7070538997650146,
      "step": 14800
    },
    {
      "dynamics_loss_avg": 0.07662916332483291,
      "epoch": 1.41143798828125e-05,
      "step": 14800
    },
    {
      "action_loss_avg": 0.0042883915361016985,
      "epoch": 1.41143798828125e-05,
      "step": 14800
    },
    {
      "epoch": 1.41143798828125e-05,
      "step": 14800,
      "training_step_time": 8.429197549819946
    },
    {
      "epoch": 1.4115333557128906e-05,
      "model_forward_time": 1.7101056575775146,
      "step": 14801
    },
    {
      "epoch": 1.4115333557128906e-05,
      "step": 14801,
      "training_step_time": 7.526487350463867
    },
    {
      "epoch": 1.4116287231445312e-05,
      "model_forward_time": 1.7083659172058105,
      "step": 14802
    },
    {
      "epoch": 1.4116287231445312e-05,
      "step": 14802,
      "training_step_time": 7.611198902130127
    },
    {
      "epoch": 1.411724090576172e-05,
      "model_forward_time": 1.7093708515167236,
      "step": 14803
    },
    {
      "epoch": 1.411724090576172e-05,
      "step": 14803,
      "training_step_time": 7.838373899459839
    },
    {
      "epoch": 1.4118194580078125e-05,
      "model_forward_time": 1.7222232818603516,
      "step": 14804
    },
    {
      "epoch": 1.4118194580078125e-05,
      "step": 14804,
      "training_step_time": 7.896936416625977
    },
    {
      "epoch": 1.4119148254394531e-05,
      "model_forward_time": 2.2419724464416504,
      "step": 14805
    },
    {
      "epoch": 1.4119148254394531e-05,
      "step": 14805,
      "training_step_time": 8.440006494522095
    },
    {
      "epoch": 1.4120101928710937e-05,
      "model_forward_time": 1.6801583766937256,
      "step": 14806
    },
    {
      "epoch": 1.4120101928710937e-05,
      "step": 14806,
      "training_step_time": 8.784913063049316
    },
    {
      "epoch": 1.4121055603027344e-05,
      "model_forward_time": 1.7058820724487305,
      "step": 14807
    },
    {
      "epoch": 1.4121055603027344e-05,
      "step": 14807,
      "training_step_time": 8.715181350708008
    },
    {
      "epoch": 1.412200927734375e-05,
      "model_forward_time": 1.687612771987915,
      "step": 14808
    },
    {
      "epoch": 1.412200927734375e-05,
      "step": 14808,
      "training_step_time": 9.195691347122192
    },
    {
      "epoch": 1.4122962951660156e-05,
      "model_forward_time": 1.704500436782837,
      "step": 14809
    },
    {
      "epoch": 1.4122962951660156e-05,
      "step": 14809,
      "training_step_time": 7.808258056640625
    },
    {
      "epoch": 1.4123916625976563e-05,
      "grad_norm": 0.21016664803028107,
      "learning_rate": 9.739246229760542e-06,
      "loss": 0.0711,
      "step": 14810
    },
    {
      "epoch": 1.4123916625976563e-05,
      "model_forward_time": 1.6966280937194824,
      "step": 14810
    },
    {
      "dynamics_loss_avg": 0.06560493875294923,
      "epoch": 1.4123916625976563e-05,
      "step": 14810
    },
    {
      "action_loss_avg": 0.005187530617695302,
      "epoch": 1.4123916625976563e-05,
      "step": 14810
    },
    {
      "epoch": 1.4123916625976563e-05,
      "step": 14810,
      "training_step_time": 8.455654859542847
    },
    {
      "epoch": 1.4124870300292969e-05,
      "model_forward_time": 1.700124740600586,
      "step": 14811
    },
    {
      "epoch": 1.4124870300292969e-05,
      "step": 14811,
      "training_step_time": 7.623163461685181
    },
    {
      "epoch": 1.4125823974609374e-05,
      "model_forward_time": 1.7008588314056396,
      "step": 14812
    },
    {
      "epoch": 1.4125823974609374e-05,
      "step": 14812,
      "training_step_time": 8.715298414230347
    },
    {
      "epoch": 1.4126777648925782e-05,
      "model_forward_time": 1.707531452178955,
      "step": 14813
    },
    {
      "epoch": 1.4126777648925782e-05,
      "step": 14813,
      "training_step_time": 9.080724954605103
    },
    {
      "epoch": 1.4127731323242188e-05,
      "model_forward_time": 2.2398626804351807,
      "step": 14814
    },
    {
      "epoch": 1.4127731323242188e-05,
      "step": 14814,
      "training_step_time": 8.900483846664429
    },
    {
      "epoch": 1.4128684997558593e-05,
      "model_forward_time": 1.7007646560668945,
      "step": 14815
    },
    {
      "epoch": 1.4128684997558593e-05,
      "step": 14815,
      "training_step_time": 10.535475492477417
    },
    {
      "epoch": 1.4129638671875e-05,
      "model_forward_time": 1.7063031196594238,
      "step": 14816
    },
    {
      "epoch": 1.4129638671875e-05,
      "step": 14816,
      "training_step_time": 7.8627119064331055
    },
    {
      "epoch": 1.4130592346191406e-05,
      "model_forward_time": 1.6975879669189453,
      "step": 14817
    },
    {
      "epoch": 1.4130592346191406e-05,
      "step": 14817,
      "training_step_time": 7.702627897262573
    },
    {
      "epoch": 1.4131546020507812e-05,
      "model_forward_time": 1.7086360454559326,
      "step": 14818
    },
    {
      "epoch": 1.4131546020507812e-05,
      "step": 14818,
      "training_step_time": 7.755065202713013
    },
    {
      "epoch": 1.413249969482422e-05,
      "model_forward_time": 1.7023935317993164,
      "step": 14819
    },
    {
      "epoch": 1.413249969482422e-05,
      "step": 14819,
      "training_step_time": 7.805736780166626
    },
    {
      "epoch": 1.4133453369140625e-05,
      "grad_norm": 0.1956436038017273,
      "learning_rate": 9.738718978369376e-06,
      "loss": 0.0685,
      "step": 14820
    },
    {
      "epoch": 1.4133453369140625e-05,
      "model_forward_time": 1.6970882415771484,
      "step": 14820
    },
    {
      "dynamics_loss_avg": 0.06515893787145614,
      "epoch": 1.4133453369140625e-05,
      "step": 14820
    },
    {
      "action_loss_avg": 0.003655562875792384,
      "epoch": 1.4133453369140625e-05,
      "step": 14820
    },
    {
      "epoch": 1.4133453369140625e-05,
      "step": 14820,
      "training_step_time": 7.55915904045105
    },
    {
      "epoch": 1.4134407043457031e-05,
      "model_forward_time": 1.7005188465118408,
      "step": 14821
    },
    {
      "epoch": 1.4134407043457031e-05,
      "step": 14821,
      "training_step_time": 7.724620342254639
    },
    {
      "epoch": 1.4135360717773437e-05,
      "model_forward_time": 1.7004189491271973,
      "step": 14822
    },
    {
      "epoch": 1.4135360717773437e-05,
      "step": 14822,
      "training_step_time": 8.499266862869263
    },
    {
      "epoch": 1.4136314392089844e-05,
      "model_forward_time": 2.232409954071045,
      "step": 14823
    },
    {
      "epoch": 1.4136314392089844e-05,
      "step": 14823,
      "training_step_time": 8.47134780883789
    },
    {
      "epoch": 1.413726806640625e-05,
      "model_forward_time": 1.7036912441253662,
      "step": 14824
    },
    {
      "epoch": 1.413726806640625e-05,
      "step": 14824,
      "training_step_time": 8.745132446289062
    },
    {
      "epoch": 1.4138221740722656e-05,
      "model_forward_time": 1.3257765769958496,
      "step": 14825
    },
    {
      "epoch": 1.4138221740722656e-05,
      "step": 14825,
      "training_step_time": 9.70773458480835
    },
    {
      "epoch": 1.4139175415039063e-05,
      "model_forward_time": 1.7797887325286865,
      "step": 14826
    },
    {
      "epoch": 1.4139175415039063e-05,
      "step": 14826,
      "training_step_time": 7.808435440063477
    },
    {
      "epoch": 1.4140129089355469e-05,
      "model_forward_time": 1.7242639064788818,
      "step": 14827
    },
    {
      "epoch": 1.4140129089355469e-05,
      "step": 14827,
      "training_step_time": 7.601474046707153
    },
    {
      "epoch": 1.4141082763671874e-05,
      "model_forward_time": 1.761690378189087,
      "step": 14828
    },
    {
      "epoch": 1.4141082763671874e-05,
      "step": 14828,
      "training_step_time": 7.571985960006714
    },
    {
      "epoch": 1.4142036437988282e-05,
      "model_forward_time": 1.7118198871612549,
      "step": 14829
    },
    {
      "epoch": 1.4142036437988282e-05,
      "step": 14829,
      "training_step_time": 7.294415712356567
    },
    {
      "epoch": 1.4142990112304688e-05,
      "grad_norm": 0.2286529690027237,
      "learning_rate": 9.738191208759025e-06,
      "loss": 0.0678,
      "step": 14830
    },
    {
      "epoch": 1.4142990112304688e-05,
      "model_forward_time": 1.71815824508667,
      "step": 14830
    },
    {
      "dynamics_loss_avg": 0.057371632941067216,
      "epoch": 1.4142990112304688e-05,
      "step": 14830
    },
    {
      "action_loss_avg": 0.0026838750927709043,
      "epoch": 1.4142990112304688e-05,
      "step": 14830
    },
    {
      "epoch": 1.4142990112304688e-05,
      "step": 14830,
      "training_step_time": 7.5268683433532715
    },
    {
      "epoch": 1.4143943786621093e-05,
      "model_forward_time": 1.7280337810516357,
      "step": 14831
    },
    {
      "epoch": 1.4143943786621093e-05,
      "step": 14831,
      "training_step_time": 8.61728024482727
    },
    {
      "epoch": 1.41448974609375e-05,
      "model_forward_time": 2.265512704849243,
      "step": 14832
    },
    {
      "epoch": 1.41448974609375e-05,
      "step": 14832,
      "training_step_time": 7.668099641799927
    },
    {
      "epoch": 1.4145851135253906e-05,
      "model_forward_time": 1.7308392524719238,
      "step": 14833
    },
    {
      "epoch": 1.4145851135253906e-05,
      "step": 14833,
      "training_step_time": 8.56588339805603
    },
    {
      "epoch": 1.4146804809570312e-05,
      "model_forward_time": 1.737018346786499,
      "step": 14834
    },
    {
      "epoch": 1.4146804809570312e-05,
      "step": 14834,
      "training_step_time": 8.58453631401062
    },
    {
      "epoch": 1.414775848388672e-05,
      "model_forward_time": 1.7561116218566895,
      "step": 14835
    },
    {
      "epoch": 1.414775848388672e-05,
      "step": 14835,
      "training_step_time": 9.122932195663452
    },
    {
      "epoch": 1.4148712158203125e-05,
      "model_forward_time": 1.7334496974945068,
      "step": 14836
    },
    {
      "epoch": 1.4148712158203125e-05,
      "step": 14836,
      "training_step_time": 7.686539173126221
    },
    {
      "epoch": 1.4149665832519531e-05,
      "model_forward_time": 1.7246387004852295,
      "step": 14837
    },
    {
      "epoch": 1.4149665832519531e-05,
      "step": 14837,
      "training_step_time": 7.674708127975464
    },
    {
      "epoch": 1.4150619506835937e-05,
      "model_forward_time": 1.7257657051086426,
      "step": 14838
    },
    {
      "epoch": 1.4150619506835937e-05,
      "step": 14838,
      "training_step_time": 8.150484800338745
    },
    {
      "epoch": 1.4151573181152344e-05,
      "model_forward_time": 1.7262036800384521,
      "step": 14839
    },
    {
      "epoch": 1.4151573181152344e-05,
      "step": 14839,
      "training_step_time": 7.549267530441284
    },
    {
      "epoch": 1.415252685546875e-05,
      "grad_norm": 0.3534063994884491,
      "learning_rate": 9.73766292098721e-06,
      "loss": 0.0688,
      "step": 14840
    },
    {
      "epoch": 1.415252685546875e-05,
      "model_forward_time": 1.7241888046264648,
      "step": 14840
    },
    {
      "dynamics_loss_avg": 0.0797862395644188,
      "epoch": 1.415252685546875e-05,
      "step": 14840
    },
    {
      "action_loss_avg": 0.00513300159946084,
      "epoch": 1.415252685546875e-05,
      "step": 14840
    },
    {
      "epoch": 1.415252685546875e-05,
      "step": 14840,
      "training_step_time": 7.67536997795105
    },
    {
      "epoch": 1.4153480529785156e-05,
      "model_forward_time": 2.2830467224121094,
      "step": 14841
    },
    {
      "epoch": 1.4153480529785156e-05,
      "step": 14841,
      "training_step_time": 7.683387756347656
    },
    {
      "epoch": 1.4154434204101563e-05,
      "model_forward_time": 1.7346415519714355,
      "step": 14842
    },
    {
      "epoch": 1.4154434204101563e-05,
      "step": 14842,
      "training_step_time": 8.709511756896973
    },
    {
      "epoch": 1.4155387878417969e-05,
      "model_forward_time": 1.7084050178527832,
      "step": 14843
    },
    {
      "epoch": 1.4155387878417969e-05,
      "step": 14843,
      "training_step_time": 8.281598806381226
    },
    {
      "epoch": 1.4156341552734374e-05,
      "model_forward_time": 1.728724718093872,
      "step": 14844
    },
    {
      "epoch": 1.4156341552734374e-05,
      "step": 14844,
      "training_step_time": 9.57425308227539
    },
    {
      "epoch": 1.4157295227050782e-05,
      "model_forward_time": 1.715165138244629,
      "step": 14845
    },
    {
      "epoch": 1.4157295227050782e-05,
      "step": 14845,
      "training_step_time": 7.685636758804321
    },
    {
      "epoch": 1.4158248901367188e-05,
      "model_forward_time": 1.7149326801300049,
      "step": 14846
    },
    {
      "epoch": 1.4158248901367188e-05,
      "step": 14846,
      "training_step_time": 8.495110273361206
    },
    {
      "epoch": 1.4159202575683593e-05,
      "model_forward_time": 1.715404987335205,
      "step": 14847
    },
    {
      "epoch": 1.4159202575683593e-05,
      "step": 14847,
      "training_step_time": 7.373224258422852
    },
    {
      "epoch": 1.416015625e-05,
      "model_forward_time": 1.71372389793396,
      "step": 14848
    },
    {
      "epoch": 1.416015625e-05,
      "step": 14848,
      "training_step_time": 7.505938291549683
    },
    {
      "epoch": 1.4161109924316406e-05,
      "model_forward_time": 1.717454433441162,
      "step": 14849
    },
    {
      "epoch": 1.4161109924316406e-05,
      "step": 14849,
      "training_step_time": 7.687838315963745
    },
    {
      "epoch": 1.4162063598632812e-05,
      "grad_norm": 0.22770123183727264,
      "learning_rate": 9.7371341151117e-06,
      "loss": 0.0636,
      "step": 14850
    },
    {
      "epoch": 1.4162063598632812e-05,
      "model_forward_time": 2.255345582962036,
      "step": 14850
    },
    {
      "dynamics_loss_avg": 0.06697972305119038,
      "epoch": 1.4162063598632812e-05,
      "step": 14850
    },
    {
      "action_loss_avg": 0.0052172094816342,
      "epoch": 1.4162063598632812e-05,
      "step": 14850
    },
    {
      "epoch": 1.4162063598632812e-05,
      "step": 14850,
      "training_step_time": 7.657543182373047
    },
    {
      "epoch": 1.416301727294922e-05,
      "model_forward_time": 1.722421646118164,
      "step": 14851
    },
    {
      "epoch": 1.416301727294922e-05,
      "step": 14851,
      "training_step_time": 9.146681070327759
    },
    {
      "epoch": 1.4163970947265625e-05,
      "model_forward_time": 1.7121469974517822,
      "step": 14852
    },
    {
      "epoch": 1.4163970947265625e-05,
      "step": 14852,
      "training_step_time": 8.753745079040527
    },
    {
      "epoch": 1.4164924621582031e-05,
      "model_forward_time": 1.7113873958587646,
      "step": 14853
    },
    {
      "epoch": 1.4164924621582031e-05,
      "step": 14853,
      "training_step_time": 9.094717502593994
    },
    {
      "epoch": 1.4165878295898437e-05,
      "model_forward_time": 1.720165491104126,
      "step": 14854
    },
    {
      "epoch": 1.4165878295898437e-05,
      "step": 14854,
      "training_step_time": 8.23095440864563
    },
    {
      "epoch": 1.4166831970214844e-05,
      "model_forward_time": 1.7176578044891357,
      "step": 14855
    },
    {
      "epoch": 1.4166831970214844e-05,
      "step": 14855,
      "training_step_time": 9.16346001625061
    },
    {
      "epoch": 1.416778564453125e-05,
      "model_forward_time": 1.693422794342041,
      "step": 14856
    },
    {
      "epoch": 1.416778564453125e-05,
      "step": 14856,
      "training_step_time": 7.3697450160980225
    },
    {
      "epoch": 1.4168739318847656e-05,
      "model_forward_time": 1.708343744277954,
      "step": 14857
    },
    {
      "epoch": 1.4168739318847656e-05,
      "step": 14857,
      "training_step_time": 7.563321828842163
    },
    {
      "epoch": 1.4169692993164063e-05,
      "model_forward_time": 1.709411382675171,
      "step": 14858
    },
    {
      "epoch": 1.4169692993164063e-05,
      "step": 14858,
      "training_step_time": 8.489971399307251
    },
    {
      "epoch": 1.4170646667480469e-05,
      "model_forward_time": 2.231966972351074,
      "step": 14859
    },
    {
      "epoch": 1.4170646667480469e-05,
      "step": 14859,
      "training_step_time": 8.76030945777893
    },
    {
      "epoch": 1.4171600341796874e-05,
      "grad_norm": 0.3089791536331177,
      "learning_rate": 9.736604791190323e-06,
      "loss": 0.0629,
      "step": 14860
    },
    {
      "epoch": 1.4171600341796874e-05,
      "model_forward_time": 1.6990692615509033,
      "step": 14860
    },
    {
      "dynamics_loss_avg": 0.06294464021921158,
      "epoch": 1.4171600341796874e-05,
      "step": 14860
    },
    {
      "action_loss_avg": 0.004952848929679021,
      "epoch": 1.4171600341796874e-05,
      "step": 14860
    },
    {
      "epoch": 1.4171600341796874e-05,
      "step": 14860,
      "training_step_time": 9.099520921707153
    },
    {
      "epoch": 1.4172554016113282e-05,
      "model_forward_time": 1.699927806854248,
      "step": 14861
    },
    {
      "epoch": 1.4172554016113282e-05,
      "step": 14861,
      "training_step_time": 7.645004749298096
    },
    {
      "epoch": 1.4173507690429688e-05,
      "model_forward_time": 1.7135062217712402,
      "step": 14862
    },
    {
      "epoch": 1.4173507690429688e-05,
      "step": 14862,
      "training_step_time": 7.707464218139648
    },
    {
      "epoch": 1.4174461364746093e-05,
      "model_forward_time": 1.7839241027832031,
      "step": 14863
    },
    {
      "epoch": 1.4174461364746093e-05,
      "step": 14863,
      "training_step_time": 8.233306169509888
    },
    {
      "epoch": 1.41754150390625e-05,
      "model_forward_time": 1.6940903663635254,
      "step": 14864
    },
    {
      "epoch": 1.41754150390625e-05,
      "step": 14864,
      "training_step_time": 7.682610750198364
    },
    {
      "epoch": 1.4176368713378906e-05,
      "model_forward_time": 1.3305680751800537,
      "step": 14865
    },
    {
      "epoch": 1.4176368713378906e-05,
      "step": 14865,
      "training_step_time": 7.626796007156372
    },
    {
      "epoch": 1.4177322387695312e-05,
      "model_forward_time": 1.7155661582946777,
      "step": 14866
    },
    {
      "epoch": 1.4177322387695312e-05,
      "step": 14866,
      "training_step_time": 7.50649619102478
    },
    {
      "epoch": 1.417827606201172e-05,
      "model_forward_time": 1.7110939025878906,
      "step": 14867
    },
    {
      "epoch": 1.417827606201172e-05,
      "step": 14867,
      "training_step_time": 8.706647157669067
    },
    {
      "epoch": 1.4179229736328125e-05,
      "model_forward_time": 1.7132363319396973,
      "step": 14868
    },
    {
      "epoch": 1.4179229736328125e-05,
      "step": 14868,
      "training_step_time": 7.636525392532349
    },
    {
      "epoch": 1.4180183410644531e-05,
      "model_forward_time": 2.2512662410736084,
      "step": 14869
    },
    {
      "epoch": 1.4180183410644531e-05,
      "step": 14869,
      "training_step_time": 8.6327383518219
    },
    {
      "epoch": 1.4181137084960937e-05,
      "grad_norm": 0.13488814234733582,
      "learning_rate": 9.73607494928097e-06,
      "loss": 0.0648,
      "step": 14870
    },
    {
      "epoch": 1.4181137084960937e-05,
      "model_forward_time": 1.703890323638916,
      "step": 14870
    },
    {
      "dynamics_loss_avg": 0.06803300268948079,
      "epoch": 1.4181137084960937e-05,
      "step": 14870
    },
    {
      "action_loss_avg": 0.003968805889599025,
      "epoch": 1.4181137084960937e-05,
      "step": 14870
    },
    {
      "epoch": 1.4181137084960937e-05,
      "step": 14870,
      "training_step_time": 8.683948755264282
    },
    {
      "epoch": 1.4182090759277344e-05,
      "model_forward_time": 1.7137517929077148,
      "step": 14871
    },
    {
      "epoch": 1.4182090759277344e-05,
      "step": 14871,
      "training_step_time": 8.511311054229736
    },
    {
      "epoch": 1.418304443359375e-05,
      "model_forward_time": 1.764817714691162,
      "step": 14872
    },
    {
      "epoch": 1.418304443359375e-05,
      "step": 14872,
      "training_step_time": 9.179672002792358
    },
    {
      "epoch": 1.4183998107910156e-05,
      "model_forward_time": 1.715571403503418,
      "step": 14873
    },
    {
      "epoch": 1.4183998107910156e-05,
      "step": 14873,
      "training_step_time": 7.699429273605347
    },
    {
      "epoch": 1.4184951782226563e-05,
      "model_forward_time": 1.713174819946289,
      "step": 14874
    },
    {
      "epoch": 1.4184951782226563e-05,
      "step": 14874,
      "training_step_time": 8.665939092636108
    },
    {
      "epoch": 1.4185905456542969e-05,
      "model_forward_time": 1.7095725536346436,
      "step": 14875
    },
    {
      "epoch": 1.4185905456542969e-05,
      "step": 14875,
      "training_step_time": 7.518038988113403
    },
    {
      "epoch": 1.4186859130859374e-05,
      "model_forward_time": 1.3378818035125732,
      "step": 14876
    },
    {
      "epoch": 1.4186859130859374e-05,
      "step": 14876,
      "training_step_time": 8.587032556533813
    },
    {
      "epoch": 1.4187812805175782e-05,
      "model_forward_time": 1.7065098285675049,
      "step": 14877
    },
    {
      "epoch": 1.4187812805175782e-05,
      "step": 14877,
      "training_step_time": 7.59430456161499
    },
    {
      "epoch": 1.4188766479492188e-05,
      "model_forward_time": 2.2618281841278076,
      "step": 14878
    },
    {
      "epoch": 1.4188766479492188e-05,
      "step": 14878,
      "training_step_time": 7.61255145072937
    },
    {
      "epoch": 1.4189720153808593e-05,
      "model_forward_time": 1.7102653980255127,
      "step": 14879
    },
    {
      "epoch": 1.4189720153808593e-05,
      "step": 14879,
      "training_step_time": 7.799599885940552
    },
    {
      "epoch": 1.4190673828125e-05,
      "grad_norm": 0.12725815176963806,
      "learning_rate": 9.73554458944158e-06,
      "loss": 0.0614,
      "step": 14880
    },
    {
      "epoch": 1.4190673828125e-05,
      "model_forward_time": 1.7007224559783936,
      "step": 14880
    },
    {
      "dynamics_loss_avg": 0.06912178881466388,
      "epoch": 1.4190673828125e-05,
      "step": 14880
    },
    {
      "action_loss_avg": 0.0033570418716408314,
      "epoch": 1.4190673828125e-05,
      "step": 14880
    },
    {
      "epoch": 1.4190673828125e-05,
      "step": 14880,
      "training_step_time": 7.771160364151001
    },
    {
      "epoch": 1.4191627502441406e-05,
      "model_forward_time": 1.7349135875701904,
      "step": 14881
    },
    {
      "epoch": 1.4191627502441406e-05,
      "step": 14881,
      "training_step_time": 8.345879077911377
    },
    {
      "epoch": 1.4192581176757812e-05,
      "model_forward_time": 1.3310165405273438,
      "step": 14882
    },
    {
      "epoch": 1.4192581176757812e-05,
      "step": 14882,
      "training_step_time": 8.596128225326538
    },
    {
      "epoch": 1.4193534851074218e-05,
      "model_forward_time": 1.703927755355835,
      "step": 14883
    },
    {
      "epoch": 1.4193534851074218e-05,
      "step": 14883,
      "training_step_time": 7.54830527305603
    },
    {
      "epoch": 1.4194488525390625e-05,
      "model_forward_time": 1.7112054824829102,
      "step": 14884
    },
    {
      "epoch": 1.4194488525390625e-05,
      "step": 14884,
      "training_step_time": 7.634601354598999
    },
    {
      "epoch": 1.4195442199707031e-05,
      "model_forward_time": 1.7131133079528809,
      "step": 14885
    },
    {
      "epoch": 1.4195442199707031e-05,
      "step": 14885,
      "training_step_time": 8.006072282791138
    },
    {
      "epoch": 1.4196395874023437e-05,
      "model_forward_time": 1.7123663425445557,
      "step": 14886
    },
    {
      "epoch": 1.4196395874023437e-05,
      "step": 14886,
      "training_step_time": 7.674546957015991
    },
    {
      "epoch": 1.4197349548339844e-05,
      "model_forward_time": 2.240058422088623,
      "step": 14887
    },
    {
      "epoch": 1.4197349548339844e-05,
      "step": 14887,
      "training_step_time": 9.236632823944092
    },
    {
      "epoch": 1.419830322265625e-05,
      "model_forward_time": 1.7090020179748535,
      "step": 14888
    },
    {
      "epoch": 1.419830322265625e-05,
      "step": 14888,
      "training_step_time": 7.765956878662109
    },
    {
      "epoch": 1.4199256896972656e-05,
      "model_forward_time": 1.7141096591949463,
      "step": 14889
    },
    {
      "epoch": 1.4199256896972656e-05,
      "step": 14889,
      "training_step_time": 8.638207912445068
    },
    {
      "epoch": 1.4200210571289063e-05,
      "grad_norm": 0.24057058990001678,
      "learning_rate": 9.735013711730155e-06,
      "loss": 0.0648,
      "step": 14890
    },
    {
      "epoch": 1.4200210571289063e-05,
      "model_forward_time": 1.7123873233795166,
      "step": 14890
    },
    {
      "dynamics_loss_avg": 0.049262918531894684,
      "epoch": 1.4200210571289063e-05,
      "step": 14890
    },
    {
      "action_loss_avg": 0.0027316201652865858,
      "epoch": 1.4200210571289063e-05,
      "step": 14890
    },
    {
      "epoch": 1.4200210571289063e-05,
      "step": 14890,
      "training_step_time": 7.7960169315338135
    },
    {
      "epoch": 1.4201164245605469e-05,
      "model_forward_time": 1.7087714672088623,
      "step": 14891
    },
    {
      "epoch": 1.4201164245605469e-05,
      "step": 14891,
      "training_step_time": 7.684852123260498
    },
    {
      "epoch": 1.4202117919921874e-05,
      "model_forward_time": 1.7141320705413818,
      "step": 14892
    },
    {
      "epoch": 1.4202117919921874e-05,
      "step": 14892,
      "training_step_time": 7.8365724086761475
    },
    {
      "epoch": 1.4203071594238282e-05,
      "model_forward_time": 1.720418930053711,
      "step": 14893
    },
    {
      "epoch": 1.4203071594238282e-05,
      "step": 14893,
      "training_step_time": 7.719990253448486
    },
    {
      "epoch": 1.4204025268554688e-05,
      "model_forward_time": 1.705787181854248,
      "step": 14894
    },
    {
      "epoch": 1.4204025268554688e-05,
      "step": 14894,
      "training_step_time": 7.2788262367248535
    },
    {
      "epoch": 1.4204978942871093e-05,
      "model_forward_time": 1.7154862880706787,
      "step": 14895
    },
    {
      "epoch": 1.4204978942871093e-05,
      "step": 14895,
      "training_step_time": 8.666285514831543
    },
    {
      "epoch": 1.42059326171875e-05,
      "model_forward_time": 1.7063965797424316,
      "step": 14896
    },
    {
      "epoch": 1.42059326171875e-05,
      "step": 14896,
      "training_step_time": 8.233291387557983
    },
    {
      "epoch": 1.4206886291503906e-05,
      "model_forward_time": 2.246225118637085,
      "step": 14897
    },
    {
      "epoch": 1.4206886291503906e-05,
      "step": 14897,
      "training_step_time": 9.20500898361206
    },
    {
      "epoch": 1.4207839965820312e-05,
      "model_forward_time": 1.7018260955810547,
      "step": 14898
    },
    {
      "epoch": 1.4207839965820312e-05,
      "step": 14898,
      "training_step_time": 7.777052164077759
    },
    {
      "epoch": 1.4208793640136718e-05,
      "model_forward_time": 1.7106778621673584,
      "step": 14899
    },
    {
      "epoch": 1.4208793640136718e-05,
      "step": 14899,
      "training_step_time": 7.761211156845093
    },
    {
      "epoch": 1.4209747314453125e-05,
      "grad_norm": 0.1982765644788742,
      "learning_rate": 9.734482316204747e-06,
      "loss": 0.0603,
      "step": 14900
    },
    {
      "epoch": 1.4209747314453125e-05,
      "model_forward_time": 1.777956485748291,
      "step": 14900
    },
    {
      "dynamics_loss_avg": 0.06349623948335648,
      "epoch": 1.4209747314453125e-05,
      "step": 14900
    },
    {
      "action_loss_avg": 0.005295267177280039,
      "epoch": 1.4209747314453125e-05,
      "step": 14900
    },
    {
      "epoch": 1.4209747314453125e-05,
      "step": 14900,
      "training_step_time": 7.6392364501953125
    },
    {
      "epoch": 1.4210700988769531e-05,
      "model_forward_time": 1.7132625579833984,
      "step": 14901
    },
    {
      "epoch": 1.4210700988769531e-05,
      "step": 14901,
      "training_step_time": 7.726344347000122
    },
    {
      "epoch": 1.4211654663085937e-05,
      "model_forward_time": 1.708221673965454,
      "step": 14902
    },
    {
      "epoch": 1.4211654663085937e-05,
      "step": 14902,
      "training_step_time": 7.6659955978393555
    },
    {
      "epoch": 1.4212608337402344e-05,
      "model_forward_time": 1.708608627319336,
      "step": 14903
    },
    {
      "epoch": 1.4212608337402344e-05,
      "step": 14903,
      "training_step_time": 7.411700963973999
    },
    {
      "epoch": 1.421356201171875e-05,
      "model_forward_time": 1.7105684280395508,
      "step": 14904
    },
    {
      "epoch": 1.421356201171875e-05,
      "step": 14904,
      "training_step_time": 7.657828330993652
    },
    {
      "epoch": 1.4214515686035156e-05,
      "model_forward_time": 1.709763526916504,
      "step": 14905
    },
    {
      "epoch": 1.4214515686035156e-05,
      "step": 14905,
      "training_step_time": 8.558086395263672
    },
    {
      "epoch": 1.4215469360351563e-05,
      "model_forward_time": 2.2549407482147217,
      "step": 14906
    },
    {
      "epoch": 1.4215469360351563e-05,
      "step": 14906,
      "training_step_time": 9.592848062515259
    },
    {
      "epoch": 1.4216423034667969e-05,
      "model_forward_time": 1.6998591423034668,
      "step": 14907
    },
    {
      "epoch": 1.4216423034667969e-05,
      "step": 14907,
      "training_step_time": 8.73987865447998
    },
    {
      "epoch": 1.4217376708984374e-05,
      "model_forward_time": 1.704390525817871,
      "step": 14908
    },
    {
      "epoch": 1.4217376708984374e-05,
      "step": 14908,
      "training_step_time": 7.8246283531188965
    },
    {
      "epoch": 1.4218330383300782e-05,
      "model_forward_time": 1.7651503086090088,
      "step": 14909
    },
    {
      "epoch": 1.4218330383300782e-05,
      "step": 14909,
      "training_step_time": 9.633937358856201
    },
    {
      "epoch": 1.4219284057617188e-05,
      "grad_norm": 0.24176135659217834,
      "learning_rate": 9.733950402923473e-06,
      "loss": 0.0636,
      "step": 14910
    },
    {
      "epoch": 1.4219284057617188e-05,
      "model_forward_time": 1.7006816864013672,
      "step": 14910
    },
    {
      "dynamics_loss_avg": 0.06846053712069988,
      "epoch": 1.4219284057617188e-05,
      "step": 14910
    },
    {
      "action_loss_avg": 0.0031035578693263234,
      "epoch": 1.4219284057617188e-05,
      "step": 14910
    },
    {
      "epoch": 1.4219284057617188e-05,
      "step": 14910,
      "training_step_time": 7.705288887023926
    },
    {
      "epoch": 1.4220237731933593e-05,
      "model_forward_time": 1.7169668674468994,
      "step": 14911
    },
    {
      "epoch": 1.4220237731933593e-05,
      "step": 14911,
      "training_step_time": 7.7046637535095215
    },
    {
      "epoch": 1.422119140625e-05,
      "model_forward_time": 1.3336360454559326,
      "step": 14912
    },
    {
      "epoch": 1.422119140625e-05,
      "step": 14912,
      "training_step_time": 7.4582905769348145
    },
    {
      "epoch": 1.4222145080566406e-05,
      "model_forward_time": 1.7105932235717773,
      "step": 14913
    },
    {
      "epoch": 1.4222145080566406e-05,
      "step": 14913,
      "training_step_time": 7.689270257949829
    },
    {
      "epoch": 1.4223098754882812e-05,
      "model_forward_time": 1.7143075466156006,
      "step": 14914
    },
    {
      "epoch": 1.4223098754882812e-05,
      "step": 14914,
      "training_step_time": 8.47631287574768
    },
    {
      "epoch": 1.4224052429199218e-05,
      "model_forward_time": 2.2710931301116943,
      "step": 14915
    },
    {
      "epoch": 1.4224052429199218e-05,
      "step": 14915,
      "training_step_time": 8.495358228683472
    },
    {
      "epoch": 1.4225006103515625e-05,
      "model_forward_time": 1.7087175846099854,
      "step": 14916
    },
    {
      "epoch": 1.4225006103515625e-05,
      "step": 14916,
      "training_step_time": 8.599798440933228
    },
    {
      "epoch": 1.4225959777832031e-05,
      "model_forward_time": 1.715343952178955,
      "step": 14917
    },
    {
      "epoch": 1.4225959777832031e-05,
      "step": 14917,
      "training_step_time": 7.818423748016357
    },
    {
      "epoch": 1.4226913452148437e-05,
      "model_forward_time": 1.7439956665039062,
      "step": 14918
    },
    {
      "epoch": 1.4226913452148437e-05,
      "step": 14918,
      "training_step_time": 7.818997621536255
    },
    {
      "epoch": 1.4227867126464844e-05,
      "model_forward_time": 1.7161891460418701,
      "step": 14919
    },
    {
      "epoch": 1.4227867126464844e-05,
      "step": 14919,
      "training_step_time": 7.7223803997039795
    },
    {
      "epoch": 1.422882080078125e-05,
      "grad_norm": 0.2639586925506592,
      "learning_rate": 9.7334179719445e-06,
      "loss": 0.0646,
      "step": 14920
    },
    {
      "epoch": 1.422882080078125e-05,
      "model_forward_time": 1.703289270401001,
      "step": 14920
    },
    {
      "dynamics_loss_avg": 0.062255905196070674,
      "epoch": 1.422882080078125e-05,
      "step": 14920
    },
    {
      "action_loss_avg": 0.003912408818723634,
      "epoch": 1.422882080078125e-05,
      "step": 14920
    },
    {
      "epoch": 1.422882080078125e-05,
      "step": 14920,
      "training_step_time": 8.583659410476685
    },
    {
      "epoch": 1.4229774475097656e-05,
      "model_forward_time": 1.714510202407837,
      "step": 14921
    },
    {
      "epoch": 1.4229774475097656e-05,
      "step": 14921,
      "training_step_time": 7.493818998336792
    },
    {
      "epoch": 1.4230728149414063e-05,
      "model_forward_time": 1.7124199867248535,
      "step": 14922
    },
    {
      "epoch": 1.4230728149414063e-05,
      "step": 14922,
      "training_step_time": 7.61509370803833
    },
    {
      "epoch": 1.4231681823730469e-05,
      "model_forward_time": 1.7064025402069092,
      "step": 14923
    },
    {
      "epoch": 1.4231681823730469e-05,
      "step": 14923,
      "training_step_time": 7.701705455780029
    },
    {
      "epoch": 1.4232635498046875e-05,
      "model_forward_time": 2.2308404445648193,
      "step": 14924
    },
    {
      "epoch": 1.4232635498046875e-05,
      "step": 14924,
      "training_step_time": 7.690810918807983
    },
    {
      "epoch": 1.4233589172363282e-05,
      "model_forward_time": 1.7120048999786377,
      "step": 14925
    },
    {
      "epoch": 1.4233589172363282e-05,
      "step": 14925,
      "training_step_time": 9.311689853668213
    },
    {
      "epoch": 1.4234542846679688e-05,
      "model_forward_time": 1.6862716674804688,
      "step": 14926
    },
    {
      "epoch": 1.4234542846679688e-05,
      "step": 14926,
      "training_step_time": 7.690090656280518
    },
    {
      "epoch": 1.4235496520996093e-05,
      "model_forward_time": 1.697810173034668,
      "step": 14927
    },
    {
      "epoch": 1.4235496520996093e-05,
      "step": 14927,
      "training_step_time": 7.825420141220093
    },
    {
      "epoch": 1.42364501953125e-05,
      "model_forward_time": 1.6866674423217773,
      "step": 14928
    },
    {
      "epoch": 1.42364501953125e-05,
      "step": 14928,
      "training_step_time": 8.247638702392578
    },
    {
      "epoch": 1.4237403869628907e-05,
      "model_forward_time": 1.6862833499908447,
      "step": 14929
    },
    {
      "epoch": 1.4237403869628907e-05,
      "step": 14929,
      "training_step_time": 7.570454359054565
    },
    {
      "epoch": 1.4238357543945312e-05,
      "grad_norm": 0.27945709228515625,
      "learning_rate": 9.732885023326054e-06,
      "loss": 0.0662,
      "step": 14930
    },
    {
      "epoch": 1.4238357543945312e-05,
      "model_forward_time": 1.3126003742218018,
      "step": 14930
    },
    {
      "dynamics_loss_avg": 0.05252547524869442,
      "epoch": 1.4238357543945312e-05,
      "step": 14930
    },
    {
      "action_loss_avg": 0.002865101461065933,
      "epoch": 1.4238357543945312e-05,
      "step": 14930
    },
    {
      "epoch": 1.4238357543945312e-05,
      "step": 14930,
      "training_step_time": 7.717266082763672
    },
    {
      "epoch": 1.4239311218261718e-05,
      "model_forward_time": 1.6953277587890625,
      "step": 14931
    },
    {
      "epoch": 1.4239311218261718e-05,
      "step": 14931,
      "training_step_time": 7.507672548294067
    },
    {
      "epoch": 1.4240264892578125e-05,
      "model_forward_time": 1.685518503189087,
      "step": 14932
    },
    {
      "epoch": 1.4240264892578125e-05,
      "step": 14932,
      "training_step_time": 7.705477952957153
    },
    {
      "epoch": 1.4241218566894531e-05,
      "model_forward_time": 1.6872668266296387,
      "step": 14933
    },
    {
      "epoch": 1.4241218566894531e-05,
      "step": 14933,
      "training_step_time": 7.759801626205444
    },
    {
      "epoch": 1.4242172241210937e-05,
      "model_forward_time": 2.227766275405884,
      "step": 14934
    },
    {
      "epoch": 1.4242172241210937e-05,
      "step": 14934,
      "training_step_time": 7.719975471496582
    },
    {
      "epoch": 1.4243125915527344e-05,
      "model_forward_time": 1.691990613937378,
      "step": 14935
    },
    {
      "epoch": 1.4243125915527344e-05,
      "step": 14935,
      "training_step_time": 7.8339903354644775
    },
    {
      "epoch": 1.424407958984375e-05,
      "model_forward_time": 1.6923329830169678,
      "step": 14936
    },
    {
      "epoch": 1.424407958984375e-05,
      "step": 14936,
      "training_step_time": 8.35794973373413
    },
    {
      "epoch": 1.4245033264160156e-05,
      "model_forward_time": 1.6951954364776611,
      "step": 14937
    },
    {
      "epoch": 1.4245033264160156e-05,
      "step": 14937,
      "training_step_time": 8.64407229423523
    },
    {
      "epoch": 1.4245986938476563e-05,
      "model_forward_time": 1.6860787868499756,
      "step": 14938
    },
    {
      "epoch": 1.4245986938476563e-05,
      "step": 14938,
      "training_step_time": 8.59969449043274
    },
    {
      "epoch": 1.4246940612792969e-05,
      "model_forward_time": 1.6992218494415283,
      "step": 14939
    },
    {
      "epoch": 1.4246940612792969e-05,
      "step": 14939,
      "training_step_time": 7.712191104888916
    },
    {
      "epoch": 1.4247894287109375e-05,
      "grad_norm": 0.1908395141363144,
      "learning_rate": 9.732351557126419e-06,
      "loss": 0.0662,
      "step": 14940
    },
    {
      "epoch": 1.4247894287109375e-05,
      "model_forward_time": 1.6949083805084229,
      "step": 14940
    },
    {
      "dynamics_loss_avg": 0.06290573384612799,
      "epoch": 1.4247894287109375e-05,
      "step": 14940
    },
    {
      "action_loss_avg": 0.003571516729425639,
      "epoch": 1.4247894287109375e-05,
      "step": 14940
    },
    {
      "epoch": 1.4247894287109375e-05,
      "step": 14940,
      "training_step_time": 7.178269863128662
    },
    {
      "epoch": 1.4248847961425782e-05,
      "model_forward_time": 1.6994009017944336,
      "step": 14941
    },
    {
      "epoch": 1.4248847961425782e-05,
      "step": 14941,
      "training_step_time": 8.50045657157898
    },
    {
      "epoch": 1.4249801635742188e-05,
      "model_forward_time": 1.6985094547271729,
      "step": 14942
    },
    {
      "epoch": 1.4249801635742188e-05,
      "step": 14942,
      "training_step_time": 8.430105209350586
    },
    {
      "epoch": 1.4250755310058593e-05,
      "model_forward_time": 2.2434608936309814,
      "step": 14943
    },
    {
      "epoch": 1.4250755310058593e-05,
      "step": 14943,
      "training_step_time": 8.382813453674316
    },
    {
      "epoch": 1.4251708984375e-05,
      "model_forward_time": 1.6923091411590576,
      "step": 14944
    },
    {
      "epoch": 1.4251708984375e-05,
      "step": 14944,
      "training_step_time": 7.80972957611084
    },
    {
      "epoch": 1.4252662658691407e-05,
      "model_forward_time": 1.7019457817077637,
      "step": 14945
    },
    {
      "epoch": 1.4252662658691407e-05,
      "step": 14945,
      "training_step_time": 7.734663009643555
    },
    {
      "epoch": 1.4253616333007812e-05,
      "model_forward_time": 1.7292594909667969,
      "step": 14946
    },
    {
      "epoch": 1.4253616333007812e-05,
      "step": 14946,
      "training_step_time": 9.539308547973633
    },
    {
      "epoch": 1.4254570007324218e-05,
      "model_forward_time": 1.6974658966064453,
      "step": 14947
    },
    {
      "epoch": 1.4254570007324218e-05,
      "step": 14947,
      "training_step_time": 8.371747255325317
    },
    {
      "epoch": 1.4255523681640625e-05,
      "model_forward_time": 1.6997506618499756,
      "step": 14948
    },
    {
      "epoch": 1.4255523681640625e-05,
      "step": 14948,
      "training_step_time": 7.708362579345703
    },
    {
      "epoch": 1.4256477355957031e-05,
      "model_forward_time": 1.6923930644989014,
      "step": 14949
    },
    {
      "epoch": 1.4256477355957031e-05,
      "step": 14949,
      "training_step_time": 7.189691781997681
    },
    {
      "epoch": 1.4257431030273437e-05,
      "grad_norm": 0.24612607061862946,
      "learning_rate": 9.73181757340393e-06,
      "loss": 0.0684,
      "step": 14950
    },
    {
      "epoch": 1.4257431030273437e-05,
      "model_forward_time": 1.6924948692321777,
      "step": 14950
    },
    {
      "dynamics_loss_avg": 0.06534434985369444,
      "epoch": 1.4257431030273437e-05,
      "step": 14950
    },
    {
      "action_loss_avg": 0.0033518068899866193,
      "epoch": 1.4257431030273437e-05,
      "step": 14950
    },
    {
      "epoch": 1.4257431030273437e-05,
      "step": 14950,
      "training_step_time": 8.63258671760559
    },
    {
      "epoch": 1.4258384704589844e-05,
      "model_forward_time": 1.3115289211273193,
      "step": 14951
    },
    {
      "epoch": 1.4258384704589844e-05,
      "step": 14951,
      "training_step_time": 7.73045015335083
    },
    {
      "epoch": 1.425933837890625e-05,
      "model_forward_time": 2.2452828884124756,
      "step": 14952
    },
    {
      "epoch": 1.425933837890625e-05,
      "step": 14952,
      "training_step_time": 7.670713424682617
    },
    {
      "epoch": 1.4260292053222656e-05,
      "model_forward_time": 1.3223843574523926,
      "step": 14953
    },
    {
      "epoch": 1.4260292053222656e-05,
      "step": 14953,
      "training_step_time": 8.335397481918335
    },
    {
      "epoch": 1.4261245727539063e-05,
      "model_forward_time": 1.6911861896514893,
      "step": 14954
    },
    {
      "epoch": 1.4261245727539063e-05,
      "step": 14954,
      "training_step_time": 8.257546424865723
    },
    {
      "epoch": 1.4262199401855469e-05,
      "model_forward_time": 1.7231714725494385,
      "step": 14955
    },
    {
      "epoch": 1.4262199401855469e-05,
      "step": 14955,
      "training_step_time": 10.175343036651611
    },
    {
      "epoch": 1.4263153076171875e-05,
      "model_forward_time": 1.6970043182373047,
      "step": 14956
    },
    {
      "epoch": 1.4263153076171875e-05,
      "step": 14956,
      "training_step_time": 7.758082151412964
    },
    {
      "epoch": 1.4264106750488282e-05,
      "model_forward_time": 1.6964795589447021,
      "step": 14957
    },
    {
      "epoch": 1.4264106750488282e-05,
      "step": 14957,
      "training_step_time": 8.211560249328613
    },
    {
      "epoch": 1.4265060424804688e-05,
      "model_forward_time": 1.6982924938201904,
      "step": 14958
    },
    {
      "epoch": 1.4265060424804688e-05,
      "step": 14958,
      "training_step_time": 8.387265682220459
    },
    {
      "epoch": 1.4266014099121093e-05,
      "model_forward_time": 1.7061760425567627,
      "step": 14959
    },
    {
      "epoch": 1.4266014099121093e-05,
      "step": 14959,
      "training_step_time": 9.454872608184814
    },
    {
      "epoch": 1.42669677734375e-05,
      "grad_norm": 0.319496750831604,
      "learning_rate": 9.731283072216985e-06,
      "loss": 0.0672,
      "step": 14960
    },
    {
      "epoch": 1.42669677734375e-05,
      "model_forward_time": 1.698169469833374,
      "step": 14960
    },
    {
      "dynamics_loss_avg": 0.06808502078056336,
      "epoch": 1.42669677734375e-05,
      "step": 14960
    },
    {
      "action_loss_avg": 0.003987019963096828,
      "epoch": 1.42669677734375e-05,
      "step": 14960
    },
    {
      "epoch": 1.42669677734375e-05,
      "step": 14960,
      "training_step_time": 7.673701524734497
    },
    {
      "epoch": 1.4267921447753907e-05,
      "model_forward_time": 2.238478422164917,
      "step": 14961
    },
    {
      "epoch": 1.4267921447753907e-05,
      "step": 14961,
      "training_step_time": 8.32513427734375
    },
    {
      "epoch": 1.4268875122070312e-05,
      "model_forward_time": 1.6972684860229492,
      "step": 14962
    },
    {
      "epoch": 1.4268875122070312e-05,
      "step": 14962,
      "training_step_time": 8.491008281707764
    },
    {
      "epoch": 1.4269828796386718e-05,
      "model_forward_time": 1.6953909397125244,
      "step": 14963
    },
    {
      "epoch": 1.4269828796386718e-05,
      "step": 14963,
      "training_step_time": 8.436487674713135
    },
    {
      "epoch": 1.4270782470703125e-05,
      "model_forward_time": 1.7103729248046875,
      "step": 14964
    },
    {
      "epoch": 1.4270782470703125e-05,
      "step": 14964,
      "training_step_time": 9.331536293029785
    },
    {
      "epoch": 1.4271736145019531e-05,
      "model_forward_time": 1.3302886486053467,
      "step": 14965
    },
    {
      "epoch": 1.4271736145019531e-05,
      "step": 14965,
      "training_step_time": 7.635084629058838
    },
    {
      "epoch": 1.4272689819335937e-05,
      "model_forward_time": 1.7035331726074219,
      "step": 14966
    },
    {
      "epoch": 1.4272689819335937e-05,
      "step": 14966,
      "training_step_time": 7.570056438446045
    },
    {
      "epoch": 1.4273643493652344e-05,
      "model_forward_time": 1.7036669254302979,
      "step": 14967
    },
    {
      "epoch": 1.4273643493652344e-05,
      "step": 14967,
      "training_step_time": 7.467118740081787
    },
    {
      "epoch": 1.427459716796875e-05,
      "model_forward_time": 1.7022426128387451,
      "step": 14968
    },
    {
      "epoch": 1.427459716796875e-05,
      "step": 14968,
      "training_step_time": 7.487244606018066
    },
    {
      "epoch": 1.4275550842285156e-05,
      "model_forward_time": 1.684325933456421,
      "step": 14969
    },
    {
      "epoch": 1.4275550842285156e-05,
      "step": 14969,
      "training_step_time": 7.730852127075195
    },
    {
      "epoch": 1.4276504516601563e-05,
      "grad_norm": 0.2978423833847046,
      "learning_rate": 9.73074805362404e-06,
      "loss": 0.0673,
      "step": 14970
    },
    {
      "epoch": 1.4276504516601563e-05,
      "model_forward_time": 1.718355417251587,
      "step": 14970
    },
    {
      "dynamics_loss_avg": 0.07011909633874894,
      "epoch": 1.4276504516601563e-05,
      "step": 14970
    },
    {
      "action_loss_avg": 0.003683730540797114,
      "epoch": 1.4276504516601563e-05,
      "step": 14970
    },
    {
      "epoch": 1.4276504516601563e-05,
      "step": 14970,
      "training_step_time": 7.7054383754730225
    },
    {
      "epoch": 1.4277458190917969e-05,
      "model_forward_time": 2.2251229286193848,
      "step": 14971
    },
    {
      "epoch": 1.4277458190917969e-05,
      "step": 14971,
      "training_step_time": 7.747677803039551
    },
    {
      "epoch": 1.4278411865234375e-05,
      "model_forward_time": 1.6963996887207031,
      "step": 14972
    },
    {
      "epoch": 1.4278411865234375e-05,
      "step": 14972,
      "training_step_time": 8.334333181381226
    },
    {
      "epoch": 1.4279365539550782e-05,
      "model_forward_time": 1.698667049407959,
      "step": 14973
    },
    {
      "epoch": 1.4279365539550782e-05,
      "step": 14973,
      "training_step_time": 9.107789993286133
    },
    {
      "epoch": 1.4280319213867188e-05,
      "model_forward_time": 1.7276573181152344,
      "step": 14974
    },
    {
      "epoch": 1.4280319213867188e-05,
      "step": 14974,
      "training_step_time": 7.822123765945435
    },
    {
      "epoch": 1.4281272888183593e-05,
      "model_forward_time": 1.6987404823303223,
      "step": 14975
    },
    {
      "epoch": 1.4281272888183593e-05,
      "step": 14975,
      "training_step_time": 7.548744201660156
    },
    {
      "epoch": 1.42822265625e-05,
      "model_forward_time": 1.6960346698760986,
      "step": 14976
    },
    {
      "epoch": 1.42822265625e-05,
      "step": 14976,
      "training_step_time": 7.6532838344573975
    },
    {
      "epoch": 1.4283180236816407e-05,
      "model_forward_time": 1.699258804321289,
      "step": 14977
    },
    {
      "epoch": 1.4283180236816407e-05,
      "step": 14977,
      "training_step_time": 7.603886604309082
    },
    {
      "epoch": 1.4284133911132812e-05,
      "model_forward_time": 1.6991796493530273,
      "step": 14978
    },
    {
      "epoch": 1.4284133911132812e-05,
      "step": 14978,
      "training_step_time": 8.143066644668579
    },
    {
      "epoch": 1.4285087585449218e-05,
      "model_forward_time": 1.709984302520752,
      "step": 14979
    },
    {
      "epoch": 1.4285087585449218e-05,
      "step": 14979,
      "training_step_time": 9.277452945709229
    },
    {
      "epoch": 1.4286041259765625e-05,
      "grad_norm": 0.36466190218925476,
      "learning_rate": 9.730212517683598e-06,
      "loss": 0.0701,
      "step": 14980
    },
    {
      "epoch": 1.4286041259765625e-05,
      "model_forward_time": 2.2364110946655273,
      "step": 14980
    },
    {
      "dynamics_loss_avg": 0.058706112764775754,
      "epoch": 1.4286041259765625e-05,
      "step": 14980
    },
    {
      "action_loss_avg": 0.0027884568902663887,
      "epoch": 1.4286041259765625e-05,
      "step": 14980
    },
    {
      "epoch": 1.4286041259765625e-05,
      "step": 14980,
      "training_step_time": 8.157498836517334
    },
    {
      "epoch": 1.4286994934082031e-05,
      "model_forward_time": 2.2052462100982666,
      "step": 14981
    },
    {
      "epoch": 1.4286994934082031e-05,
      "step": 14981,
      "training_step_time": 8.408429861068726
    },
    {
      "epoch": 1.4287948608398437e-05,
      "model_forward_time": 1.6987404823303223,
      "step": 14982
    },
    {
      "epoch": 1.4287948608398437e-05,
      "step": 14982,
      "training_step_time": 7.781028747558594
    },
    {
      "epoch": 1.4288902282714844e-05,
      "model_forward_time": 1.6877591609954834,
      "step": 14983
    },
    {
      "epoch": 1.4288902282714844e-05,
      "step": 14983,
      "training_step_time": 6.835244178771973
    },
    {
      "epoch": 1.428985595703125e-05,
      "model_forward_time": 1.6884050369262695,
      "step": 14984
    },
    {
      "epoch": 1.428985595703125e-05,
      "step": 14984,
      "training_step_time": 6.820465803146362
    },
    {
      "epoch": 1.4290809631347656e-05,
      "model_forward_time": 1.3138666152954102,
      "step": 14985
    },
    {
      "epoch": 1.4290809631347656e-05,
      "step": 14985,
      "training_step_time": 6.819272041320801
    },
    {
      "epoch": 1.4291763305664063e-05,
      "model_forward_time": 1.6998014450073242,
      "step": 14986
    },
    {
      "epoch": 1.4291763305664063e-05,
      "step": 14986,
      "training_step_time": 6.869650602340698
    },
    {
      "epoch": 1.4292716979980469e-05,
      "model_forward_time": 1.3161530494689941,
      "step": 14987
    },
    {
      "epoch": 1.4292716979980469e-05,
      "step": 14987,
      "training_step_time": 6.714171409606934
    },
    {
      "epoch": 1.4293670654296875e-05,
      "model_forward_time": 1.7005107402801514,
      "step": 14988
    },
    {
      "epoch": 1.4293670654296875e-05,
      "step": 14988,
      "training_step_time": 6.812390327453613
    },
    {
      "epoch": 1.4294624328613282e-05,
      "model_forward_time": 1.690096139907837,
      "step": 14989
    },
    {
      "epoch": 1.4294624328613282e-05,
      "step": 14989,
      "training_step_time": 6.816197156906128
    },
    {
      "epoch": 1.4295578002929688e-05,
      "grad_norm": 0.3771693706512451,
      "learning_rate": 9.729676464454227e-06,
      "loss": 0.0728,
      "step": 14990
    },
    {
      "epoch": 1.4295578002929688e-05,
      "model_forward_time": 2.2384965419769287,
      "step": 14990
    },
    {
      "dynamics_loss_avg": 0.06110998000949621,
      "epoch": 1.4295578002929688e-05,
      "step": 14990
    },
    {
      "action_loss_avg": 0.0025671075098216535,
      "epoch": 1.4295578002929688e-05,
      "step": 14990
    },
    {
      "epoch": 1.4295578002929688e-05,
      "step": 14990,
      "training_step_time": 8.844781398773193
    },
    {
      "epoch": 1.4296531677246093e-05,
      "model_forward_time": 1.681394100189209,
      "step": 14991
    },
    {
      "epoch": 1.4296531677246093e-05,
      "step": 14991,
      "training_step_time": 7.721014976501465
    },
    {
      "epoch": 1.42974853515625e-05,
      "model_forward_time": 1.3158633708953857,
      "step": 14992
    },
    {
      "epoch": 1.42974853515625e-05,
      "step": 14992,
      "training_step_time": 6.865064859390259
    },
    {
      "epoch": 1.4298439025878907e-05,
      "model_forward_time": 1.6891040802001953,
      "step": 14993
    },
    {
      "epoch": 1.4298439025878907e-05,
      "step": 14993,
      "training_step_time": 6.8034257888793945
    },
    {
      "epoch": 1.4299392700195312e-05,
      "model_forward_time": 1.6901898384094238,
      "step": 14994
    },
    {
      "epoch": 1.4299392700195312e-05,
      "step": 14994,
      "training_step_time": 6.78572154045105
    },
    {
      "epoch": 1.4300346374511718e-05,
      "model_forward_time": 1.6817607879638672,
      "step": 14995
    },
    {
      "epoch": 1.4300346374511718e-05,
      "step": 14995,
      "training_step_time": 6.779623508453369
    },
    {
      "epoch": 1.4301300048828125e-05,
      "model_forward_time": 1.6922614574432373,
      "step": 14996
    },
    {
      "epoch": 1.4301300048828125e-05,
      "step": 14996,
      "training_step_time": 6.855621576309204
    },
    {
      "epoch": 1.4302253723144531e-05,
      "model_forward_time": 1.6922190189361572,
      "step": 14997
    },
    {
      "epoch": 1.4302253723144531e-05,
      "step": 14997,
      "training_step_time": 6.805062532424927
    },
    {
      "epoch": 1.4303207397460937e-05,
      "model_forward_time": 1.7248115539550781,
      "step": 14998
    },
    {
      "epoch": 1.4303207397460937e-05,
      "step": 14998,
      "training_step_time": 6.959415435791016
    },
    {
      "epoch": 1.4304161071777344e-05,
      "model_forward_time": 2.3116750717163086,
      "step": 14999
    },
    {
      "epoch": 1.4304161071777344e-05,
      "step": 14999,
      "training_step_time": 7.9460368156433105
    },
    {
      "epoch": 1.430511474609375e-05,
      "grad_norm": 0.2398129552602768,
      "learning_rate": 9.729139893994551e-06,
      "loss": 0.0667,
      "step": 15000
    },
    {
      "epoch": 1.430511474609375e-05,
      "model_forward_time": 1.80792236328125,
      "step": 15000
    },
    {
      "dynamics_loss_avg": 0.05301525089889765,
      "epoch": 1.430511474609375e-05,
      "step": 15000
    },
    {
      "action_loss_avg": 0.0024806568049825727,
      "epoch": 1.430511474609375e-05,
      "step": 15000
    },
    {
      "epoch": 1.430511474609375e-05,
      "step": 15000,
      "training_step_time": 7.17525839805603
    },
    {
      "epoch": 1.4306068420410156e-05,
      "model_forward_time": 1.7671079635620117,
      "step": 15001
    },
    {
      "epoch": 1.4306068420410156e-05,
      "step": 15001,
      "training_step_time": 6.664701223373413
    },
    {
      "epoch": 1.4307022094726563e-05,
      "model_forward_time": 1.7700221538543701,
      "step": 15002
    },
    {
      "epoch": 1.4307022094726563e-05,
      "step": 15002,
      "training_step_time": 6.6666953563690186
    },
    {
      "epoch": 1.4307975769042969e-05,
      "model_forward_time": 1.7643797397613525,
      "step": 15003
    },
    {
      "epoch": 1.4307975769042969e-05,
      "step": 15003,
      "training_step_time": 6.744411468505859
    },
    {
      "epoch": 1.4308929443359375e-05,
      "model_forward_time": 1.7676451206207275,
      "step": 15004
    },
    {
      "epoch": 1.4308929443359375e-05,
      "step": 15004,
      "training_step_time": 6.737069606781006
    },
    {
      "epoch": 1.4309883117675782e-05,
      "model_forward_time": 1.7671689987182617,
      "step": 15005
    },
    {
      "epoch": 1.4309883117675782e-05,
      "step": 15005,
      "training_step_time": 6.736807823181152
    },
    {
      "epoch": 1.4310836791992188e-05,
      "model_forward_time": 1.7736740112304688,
      "step": 15006
    },
    {
      "epoch": 1.4310836791992188e-05,
      "step": 15006,
      "training_step_time": 6.739495038986206
    },
    {
      "epoch": 1.4311790466308593e-05,
      "model_forward_time": 1.7753405570983887,
      "step": 15007
    },
    {
      "epoch": 1.4311790466308593e-05,
      "step": 15007,
      "training_step_time": 6.784960985183716
    },
    {
      "epoch": 1.4312744140625e-05,
      "model_forward_time": 3.3949882984161377,
      "step": 15008
    },
    {
      "epoch": 1.4312744140625e-05,
      "step": 15008,
      "training_step_time": 8.2421293258667
    },
    {
      "epoch": 1.4313697814941407e-05,
      "model_forward_time": 1.7599167823791504,
      "step": 15009
    },
    {
      "epoch": 1.4313697814941407e-05,
      "step": 15009,
      "training_step_time": 7.8140270709991455
    },
    {
      "epoch": 1.4314651489257812e-05,
      "grad_norm": 0.26306915283203125,
      "learning_rate": 9.728602806363244e-06,
      "loss": 0.0649,
      "step": 15010
    },
    {
      "epoch": 1.4314651489257812e-05,
      "model_forward_time": 1.7729952335357666,
      "step": 15010
    },
    {
      "dynamics_loss_avg": 0.06500234287232161,
      "epoch": 1.4314651489257812e-05,
      "step": 15010
    },
    {
      "action_loss_avg": 0.0032728405407397077,
      "epoch": 1.4314651489257812e-05,
      "step": 15010
    },
    {
      "epoch": 1.4314651489257812e-05,
      "step": 15010,
      "training_step_time": 7.1680662631988525
    },
    {
      "epoch": 1.4315605163574218e-05,
      "model_forward_time": 1.7629311084747314,
      "step": 15011
    },
    {
      "epoch": 1.4315605163574218e-05,
      "step": 15011,
      "training_step_time": 6.7288830280303955
    },
    {
      "epoch": 1.4316558837890625e-05,
      "model_forward_time": 1.7605092525482178,
      "step": 15012
    },
    {
      "epoch": 1.4316558837890625e-05,
      "step": 15012,
      "training_step_time": 6.742673873901367
    },
    {
      "epoch": 1.4317512512207031e-05,
      "model_forward_time": 1.7692224979400635,
      "step": 15013
    },
    {
      "epoch": 1.4317512512207031e-05,
      "step": 15013,
      "training_step_time": 6.74639630317688
    },
    {
      "epoch": 1.4318466186523437e-05,
      "model_forward_time": 1.7527990341186523,
      "step": 15014
    },
    {
      "epoch": 1.4318466186523437e-05,
      "step": 15014,
      "training_step_time": 6.809643507003784
    },
    {
      "epoch": 1.4319419860839844e-05,
      "model_forward_time": 1.7609913349151611,
      "step": 15015
    },
    {
      "epoch": 1.4319419860839844e-05,
      "step": 15015,
      "training_step_time": 6.785724878311157
    },
    {
      "epoch": 1.432037353515625e-05,
      "model_forward_time": 1.7770183086395264,
      "step": 15016
    },
    {
      "epoch": 1.432037353515625e-05,
      "step": 15016,
      "training_step_time": 6.804986953735352
    },
    {
      "epoch": 1.4321327209472656e-05,
      "model_forward_time": 2.910957098007202,
      "step": 15017
    },
    {
      "epoch": 1.4321327209472656e-05,
      "step": 15017,
      "training_step_time": 8.250302076339722
    },
    {
      "epoch": 1.4322280883789063e-05,
      "model_forward_time": 1.748091459274292,
      "step": 15018
    },
    {
      "epoch": 1.4322280883789063e-05,
      "step": 15018,
      "training_step_time": 9.318842887878418
    },
    {
      "epoch": 1.4323234558105469e-05,
      "model_forward_time": 1.766171932220459,
      "step": 15019
    },
    {
      "epoch": 1.4323234558105469e-05,
      "step": 15019,
      "training_step_time": 7.341197729110718
    },
    {
      "epoch": 1.4324188232421875e-05,
      "grad_norm": 0.2618502080440521,
      "learning_rate": 9.728065201619044e-06,
      "loss": 0.0731,
      "step": 15020
    },
    {
      "epoch": 1.4324188232421875e-05,
      "model_forward_time": 1.75807785987854,
      "step": 15020
    },
    {
      "dynamics_loss_avg": 0.07047356013208628,
      "epoch": 1.4324188232421875e-05,
      "step": 15020
    },
    {
      "action_loss_avg": 0.0036185371689498426,
      "epoch": 1.4324188232421875e-05,
      "step": 15020
    },
    {
      "epoch": 1.4324188232421875e-05,
      "step": 15020,
      "training_step_time": 6.792792797088623
    },
    {
      "epoch": 1.4325141906738282e-05,
      "model_forward_time": 1.7589707374572754,
      "step": 15021
    },
    {
      "epoch": 1.4325141906738282e-05,
      "step": 15021,
      "training_step_time": 6.817403316497803
    },
    {
      "epoch": 1.4326095581054688e-05,
      "model_forward_time": 1.7671301364898682,
      "step": 15022
    },
    {
      "epoch": 1.4326095581054688e-05,
      "step": 15022,
      "training_step_time": 6.824749708175659
    },
    {
      "epoch": 1.4327049255371093e-05,
      "model_forward_time": 1.7670865058898926,
      "step": 15023
    },
    {
      "epoch": 1.4327049255371093e-05,
      "step": 15023,
      "training_step_time": 6.800049543380737
    },
    {
      "epoch": 1.43280029296875e-05,
      "model_forward_time": 1.76129150390625,
      "step": 15024
    },
    {
      "epoch": 1.43280029296875e-05,
      "step": 15024,
      "training_step_time": 6.803779125213623
    },
    {
      "epoch": 1.4328956604003907e-05,
      "model_forward_time": 1.7564537525177002,
      "step": 15025
    },
    {
      "epoch": 1.4328956604003907e-05,
      "step": 15025,
      "training_step_time": 6.797858238220215
    },
    {
      "epoch": 1.4329910278320312e-05,
      "model_forward_time": 2.83927845954895,
      "step": 15026
    },
    {
      "epoch": 1.4329910278320312e-05,
      "step": 15026,
      "training_step_time": 7.679521322250366
    },
    {
      "epoch": 1.4330863952636718e-05,
      "model_forward_time": 1.86820650100708,
      "step": 15027
    },
    {
      "epoch": 1.4330863952636718e-05,
      "step": 15027,
      "training_step_time": 10.303759813308716
    },
    {
      "epoch": 1.4331817626953125e-05,
      "model_forward_time": 1.7505135536193848,
      "step": 15028
    },
    {
      "epoch": 1.4331817626953125e-05,
      "step": 15028,
      "training_step_time": 7.59584641456604
    },
    {
      "epoch": 1.4332771301269531e-05,
      "model_forward_time": 1.755483627319336,
      "step": 15029
    },
    {
      "epoch": 1.4332771301269531e-05,
      "step": 15029,
      "training_step_time": 6.790324687957764
    },
    {
      "epoch": 1.4333724975585937e-05,
      "grad_norm": 0.20510230958461761,
      "learning_rate": 9.727527079820743e-06,
      "loss": 0.0659,
      "step": 15030
    },
    {
      "epoch": 1.4333724975585937e-05,
      "model_forward_time": 1.7653086185455322,
      "step": 15030
    },
    {
      "dynamics_loss_avg": 0.052848283760249616,
      "epoch": 1.4333724975585937e-05,
      "step": 15030
    },
    {
      "action_loss_avg": 0.0024878808646462856,
      "epoch": 1.4333724975585937e-05,
      "step": 15030
    },
    {
      "epoch": 1.4333724975585937e-05,
      "step": 15030,
      "training_step_time": 6.8269124031066895
    },
    {
      "epoch": 1.4334678649902344e-05,
      "model_forward_time": 1.7635762691497803,
      "step": 15031
    },
    {
      "epoch": 1.4334678649902344e-05,
      "step": 15031,
      "training_step_time": 6.7737953662872314
    },
    {
      "epoch": 1.433563232421875e-05,
      "model_forward_time": 1.7573051452636719,
      "step": 15032
    },
    {
      "epoch": 1.433563232421875e-05,
      "step": 15032,
      "training_step_time": 6.768951892852783
    },
    {
      "epoch": 1.4336585998535156e-05,
      "model_forward_time": 1.753396987915039,
      "step": 15033
    },
    {
      "epoch": 1.4336585998535156e-05,
      "step": 15033,
      "training_step_time": 6.747652530670166
    },
    {
      "epoch": 1.4337539672851563e-05,
      "model_forward_time": 1.7569036483764648,
      "step": 15034
    },
    {
      "epoch": 1.4337539672851563e-05,
      "step": 15034,
      "training_step_time": 6.744985580444336
    },
    {
      "epoch": 1.4338493347167969e-05,
      "model_forward_time": 1.7641329765319824,
      "step": 15035
    },
    {
      "epoch": 1.4338493347167969e-05,
      "step": 15035,
      "training_step_time": 6.7499682903289795
    },
    {
      "epoch": 1.4339447021484375e-05,
      "model_forward_time": 2.9025027751922607,
      "step": 15036
    },
    {
      "epoch": 1.4339447021484375e-05,
      "step": 15036,
      "training_step_time": 9.282567501068115
    },
    {
      "epoch": 1.4340400695800782e-05,
      "model_forward_time": 1.757448434829712,
      "step": 15037
    },
    {
      "epoch": 1.4340400695800782e-05,
      "step": 15037,
      "training_step_time": 7.799176931381226
    },
    {
      "epoch": 1.4341354370117188e-05,
      "model_forward_time": 1.7710061073303223,
      "step": 15038
    },
    {
      "epoch": 1.4341354370117188e-05,
      "step": 15038,
      "training_step_time": 7.235591173171997
    },
    {
      "epoch": 1.4342308044433593e-05,
      "model_forward_time": 1.7744505405426025,
      "step": 15039
    },
    {
      "epoch": 1.4342308044433593e-05,
      "step": 15039,
      "training_step_time": 6.808733940124512
    },
    {
      "epoch": 1.434326171875e-05,
      "grad_norm": 0.30256640911102295,
      "learning_rate": 9.726988441027186e-06,
      "loss": 0.0703,
      "step": 15040
    },
    {
      "epoch": 1.434326171875e-05,
      "model_forward_time": 1.7553248405456543,
      "step": 15040
    },
    {
      "dynamics_loss_avg": 0.0753177149221301,
      "epoch": 1.434326171875e-05,
      "step": 15040
    },
    {
      "action_loss_avg": 0.004086480988189578,
      "epoch": 1.434326171875e-05,
      "step": 15040
    },
    {
      "epoch": 1.434326171875e-05,
      "step": 15040,
      "training_step_time": 6.977783679962158
    },
    {
      "epoch": 1.4344215393066407e-05,
      "model_forward_time": 1.7644317150115967,
      "step": 15041
    },
    {
      "epoch": 1.4344215393066407e-05,
      "step": 15041,
      "training_step_time": 6.838890075683594
    },
    {
      "epoch": 1.4345169067382812e-05,
      "model_forward_time": 1.767707347869873,
      "step": 15042
    },
    {
      "epoch": 1.4345169067382812e-05,
      "step": 15042,
      "training_step_time": 6.853707790374756
    },
    {
      "epoch": 1.4346122741699218e-05,
      "model_forward_time": 1.7904174327850342,
      "step": 15043
    },
    {
      "epoch": 1.4346122741699218e-05,
      "step": 15043,
      "training_step_time": 6.823422908782959
    },
    {
      "epoch": 1.4347076416015625e-05,
      "model_forward_time": 1.7714242935180664,
      "step": 15044
    },
    {
      "epoch": 1.4347076416015625e-05,
      "step": 15044,
      "training_step_time": 6.861694097518921
    },
    {
      "epoch": 1.4348030090332031e-05,
      "model_forward_time": 2.5512380599975586,
      "step": 15045
    },
    {
      "epoch": 1.4348030090332031e-05,
      "step": 15045,
      "training_step_time": 10.802143812179565
    },
    {
      "epoch": 1.4348983764648437e-05,
      "model_forward_time": 1.76629638671875,
      "step": 15046
    },
    {
      "epoch": 1.4348983764648437e-05,
      "step": 15046,
      "training_step_time": 7.879235744476318
    },
    {
      "epoch": 1.4349937438964844e-05,
      "model_forward_time": 1.748056411743164,
      "step": 15047
    },
    {
      "epoch": 1.4349937438964844e-05,
      "step": 15047,
      "training_step_time": 7.499328136444092
    },
    {
      "epoch": 1.435089111328125e-05,
      "model_forward_time": 1.7494590282440186,
      "step": 15048
    },
    {
      "epoch": 1.435089111328125e-05,
      "step": 15048,
      "training_step_time": 6.914703607559204
    },
    {
      "epoch": 1.4351844787597656e-05,
      "model_forward_time": 1.771376609802246,
      "step": 15049
    },
    {
      "epoch": 1.4351844787597656e-05,
      "step": 15049,
      "training_step_time": 6.910141944885254
    },
    {
      "epoch": 1.4352798461914063e-05,
      "grad_norm": 0.13028332591056824,
      "learning_rate": 9.726449285297281e-06,
      "loss": 0.0671,
      "step": 15050
    },
    {
      "epoch": 1.4352798461914063e-05,
      "model_forward_time": 1.7595171928405762,
      "step": 15050
    },
    {
      "dynamics_loss_avg": 0.07488665543496609,
      "epoch": 1.4352798461914063e-05,
      "step": 15050
    },
    {
      "action_loss_avg": 0.004086283873766661,
      "epoch": 1.4352798461914063e-05,
      "step": 15050
    },
    {
      "epoch": 1.4352798461914063e-05,
      "step": 15050,
      "training_step_time": 6.842855215072632
    },
    {
      "epoch": 1.4353752136230469e-05,
      "model_forward_time": 1.752002239227295,
      "step": 15051
    },
    {
      "epoch": 1.4353752136230469e-05,
      "step": 15051,
      "training_step_time": 6.793531656265259
    },
    {
      "epoch": 1.4354705810546875e-05,
      "model_forward_time": 1.7580904960632324,
      "step": 15052
    },
    {
      "epoch": 1.4354705810546875e-05,
      "step": 15052,
      "training_step_time": 6.80939245223999
    },
    {
      "epoch": 1.4355659484863282e-05,
      "model_forward_time": 1.752087116241455,
      "step": 15053
    },
    {
      "epoch": 1.4355659484863282e-05,
      "step": 15053,
      "training_step_time": 6.861838102340698
    },
    {
      "epoch": 1.4356613159179688e-05,
      "model_forward_time": 2.9389209747314453,
      "step": 15054
    },
    {
      "epoch": 1.4356613159179688e-05,
      "step": 15054,
      "training_step_time": 9.03319525718689
    },
    {
      "epoch": 1.4357566833496093e-05,
      "model_forward_time": 1.7648608684539795,
      "step": 15055
    },
    {
      "epoch": 1.4357566833496093e-05,
      "step": 15055,
      "training_step_time": 9.626015901565552
    },
    {
      "epoch": 1.43585205078125e-05,
      "model_forward_time": 1.7730703353881836,
      "step": 15056
    },
    {
      "epoch": 1.43585205078125e-05,
      "step": 15056,
      "training_step_time": 7.548914670944214
    },
    {
      "epoch": 1.4359474182128907e-05,
      "model_forward_time": 1.757810354232788,
      "step": 15057
    },
    {
      "epoch": 1.4359474182128907e-05,
      "step": 15057,
      "training_step_time": 6.929494380950928
    },
    {
      "epoch": 1.4360427856445312e-05,
      "model_forward_time": 1.7625350952148438,
      "step": 15058
    },
    {
      "epoch": 1.4360427856445312e-05,
      "step": 15058,
      "training_step_time": 6.779206037521362
    },
    {
      "epoch": 1.4361381530761718e-05,
      "model_forward_time": 1.7487549781799316,
      "step": 15059
    },
    {
      "epoch": 1.4361381530761718e-05,
      "step": 15059,
      "training_step_time": 6.796103477478027
    },
    {
      "epoch": 1.4362335205078125e-05,
      "grad_norm": 0.2625442445278168,
      "learning_rate": 9.72590961268999e-06,
      "loss": 0.0669,
      "step": 15060
    },
    {
      "epoch": 1.4362335205078125e-05,
      "model_forward_time": 1.7648239135742188,
      "step": 15060
    },
    {
      "dynamics_loss_avg": 0.049174247309565544,
      "epoch": 1.4362335205078125e-05,
      "step": 15060
    },
    {
      "action_loss_avg": 0.0025940777268260717,
      "epoch": 1.4362335205078125e-05,
      "step": 15060
    },
    {
      "epoch": 1.4362335205078125e-05,
      "step": 15060,
      "training_step_time": 6.822722911834717
    },
    {
      "epoch": 1.4363288879394531e-05,
      "model_forward_time": 1.458071231842041,
      "step": 15061
    },
    {
      "epoch": 1.4363288879394531e-05,
      "step": 15061,
      "training_step_time": 6.738364934921265
    },
    {
      "epoch": 1.4364242553710937e-05,
      "model_forward_time": 1.7761003971099854,
      "step": 15062
    },
    {
      "epoch": 1.4364242553710937e-05,
      "step": 15062,
      "training_step_time": 6.786458730697632
    },
    {
      "epoch": 1.4365196228027344e-05,
      "model_forward_time": 2.835134506225586,
      "step": 15063
    },
    {
      "epoch": 1.4365196228027344e-05,
      "step": 15063,
      "training_step_time": 8.913740873336792
    },
    {
      "epoch": 1.436614990234375e-05,
      "model_forward_time": 1.8695297241210938,
      "step": 15064
    },
    {
      "epoch": 1.436614990234375e-05,
      "step": 15064,
      "training_step_time": 8.431671857833862
    },
    {
      "epoch": 1.4367103576660156e-05,
      "model_forward_time": 1.7598686218261719,
      "step": 15065
    },
    {
      "epoch": 1.4367103576660156e-05,
      "step": 15065,
      "training_step_time": 8.824379205703735
    },
    {
      "epoch": 1.4368057250976563e-05,
      "model_forward_time": 1.7538087368011475,
      "step": 15066
    },
    {
      "epoch": 1.4368057250976563e-05,
      "step": 15066,
      "training_step_time": 6.787420988082886
    },
    {
      "epoch": 1.4369010925292969e-05,
      "model_forward_time": 1.7306859493255615,
      "step": 15067
    },
    {
      "epoch": 1.4369010925292969e-05,
      "step": 15067,
      "training_step_time": 6.804174900054932
    },
    {
      "epoch": 1.4369964599609375e-05,
      "model_forward_time": 1.7344138622283936,
      "step": 15068
    },
    {
      "epoch": 1.4369964599609375e-05,
      "step": 15068,
      "training_step_time": 6.820786952972412
    },
    {
      "epoch": 1.4370918273925782e-05,
      "model_forward_time": 1.735975742340088,
      "step": 15069
    },
    {
      "epoch": 1.4370918273925782e-05,
      "step": 15069,
      "training_step_time": 6.777230978012085
    },
    {
      "epoch": 1.4371871948242188e-05,
      "grad_norm": 0.13247016072273254,
      "learning_rate": 9.725369423264327e-06,
      "loss": 0.0697,
      "step": 15070
    },
    {
      "epoch": 1.4371871948242188e-05,
      "model_forward_time": 1.7411234378814697,
      "step": 15070
    },
    {
      "dynamics_loss_avg": 0.07028191965073347,
      "epoch": 1.4371871948242188e-05,
      "step": 15070
    },
    {
      "action_loss_avg": 0.0045629896922037005,
      "epoch": 1.4371871948242188e-05,
      "step": 15070
    },
    {
      "epoch": 1.4371871948242188e-05,
      "step": 15070,
      "training_step_time": 6.849745512008667
    },
    {
      "epoch": 1.4372825622558593e-05,
      "model_forward_time": 1.7488784790039062,
      "step": 15071
    },
    {
      "epoch": 1.4372825622558593e-05,
      "step": 15071,
      "training_step_time": 6.821364164352417
    },
    {
      "epoch": 1.4373779296875e-05,
      "model_forward_time": 1.7414426803588867,
      "step": 15072
    },
    {
      "epoch": 1.4373779296875e-05,
      "step": 15072,
      "training_step_time": 7.914412975311279
    },
    {
      "epoch": 1.4374732971191407e-05,
      "model_forward_time": 2.886687755584717,
      "step": 15073
    },
    {
      "epoch": 1.4374732971191407e-05,
      "step": 15073,
      "training_step_time": 8.968437671661377
    },
    {
      "epoch": 1.4375686645507812e-05,
      "model_forward_time": 1.7365384101867676,
      "step": 15074
    },
    {
      "epoch": 1.4375686645507812e-05,
      "step": 15074,
      "training_step_time": 7.861527442932129
    },
    {
      "epoch": 1.4376640319824218e-05,
      "model_forward_time": 1.742875337600708,
      "step": 15075
    },
    {
      "epoch": 1.4376640319824218e-05,
      "step": 15075,
      "training_step_time": 7.246063232421875
    },
    {
      "epoch": 1.4377593994140625e-05,
      "model_forward_time": 1.3959832191467285,
      "step": 15076
    },
    {
      "epoch": 1.4377593994140625e-05,
      "step": 15076,
      "training_step_time": 6.834796667098999
    },
    {
      "epoch": 1.4378547668457031e-05,
      "model_forward_time": 1.7398812770843506,
      "step": 15077
    },
    {
      "epoch": 1.4378547668457031e-05,
      "step": 15077,
      "training_step_time": 6.822444200515747
    },
    {
      "epoch": 1.4379501342773437e-05,
      "model_forward_time": 1.758479356765747,
      "step": 15078
    },
    {
      "epoch": 1.4379501342773437e-05,
      "step": 15078,
      "training_step_time": 6.80617880821228
    },
    {
      "epoch": 1.4380455017089844e-05,
      "model_forward_time": 1.729224443435669,
      "step": 15079
    },
    {
      "epoch": 1.4380455017089844e-05,
      "step": 15079,
      "training_step_time": 6.819750785827637
    },
    {
      "epoch": 1.438140869140625e-05,
      "grad_norm": 0.2520754039287567,
      "learning_rate": 9.724828717079369e-06,
      "loss": 0.0707,
      "step": 15080
    },
    {
      "epoch": 1.438140869140625e-05,
      "model_forward_time": 1.7408926486968994,
      "step": 15080
    },
    {
      "dynamics_loss_avg": 0.05330276899039745,
      "epoch": 1.438140869140625e-05,
      "step": 15080
    },
    {
      "action_loss_avg": 0.0034592175215948373,
      "epoch": 1.438140869140625e-05,
      "step": 15080
    },
    {
      "epoch": 1.438140869140625e-05,
      "step": 15080,
      "training_step_time": 6.8022620677948
    },
    {
      "epoch": 1.4382362365722656e-05,
      "model_forward_time": 1.055856704711914,
      "step": 15081
    },
    {
      "epoch": 1.4382362365722656e-05,
      "step": 15081,
      "training_step_time": 7.907421588897705
    },
    {
      "epoch": 1.4383316040039063e-05,
      "model_forward_time": 2.85428524017334,
      "step": 15082
    },
    {
      "epoch": 1.4383316040039063e-05,
      "step": 15082,
      "training_step_time": 7.924857139587402
    },
    {
      "epoch": 1.4384269714355469e-05,
      "model_forward_time": 1.7423112392425537,
      "step": 15083
    },
    {
      "epoch": 1.4384269714355469e-05,
      "step": 15083,
      "training_step_time": 7.888879776000977
    },
    {
      "epoch": 1.4385223388671875e-05,
      "model_forward_time": 1.73455810546875,
      "step": 15084
    },
    {
      "epoch": 1.4385223388671875e-05,
      "step": 15084,
      "training_step_time": 7.358192443847656
    },
    {
      "epoch": 1.4386177062988282e-05,
      "model_forward_time": 1.7441003322601318,
      "step": 15085
    },
    {
      "epoch": 1.4386177062988282e-05,
      "step": 15085,
      "training_step_time": 6.840404033660889
    },
    {
      "epoch": 1.4387130737304688e-05,
      "model_forward_time": 1.755005121231079,
      "step": 15086
    },
    {
      "epoch": 1.4387130737304688e-05,
      "step": 15086,
      "training_step_time": 6.8497819900512695
    },
    {
      "epoch": 1.4388084411621094e-05,
      "model_forward_time": 1.749044418334961,
      "step": 15087
    },
    {
      "epoch": 1.4388084411621094e-05,
      "step": 15087,
      "training_step_time": 6.80765700340271
    },
    {
      "epoch": 1.43890380859375e-05,
      "model_forward_time": 1.7568929195404053,
      "step": 15088
    },
    {
      "epoch": 1.43890380859375e-05,
      "step": 15088,
      "training_step_time": 6.835834980010986
    },
    {
      "epoch": 1.4389991760253907e-05,
      "model_forward_time": 1.7591578960418701,
      "step": 15089
    },
    {
      "epoch": 1.4389991760253907e-05,
      "step": 15089,
      "training_step_time": 6.892233848571777
    },
    {
      "epoch": 1.4390945434570312e-05,
      "grad_norm": 0.3350079655647278,
      "learning_rate": 9.724287494194247e-06,
      "loss": 0.0661,
      "step": 15090
    },
    {
      "epoch": 1.4390945434570312e-05,
      "model_forward_time": 1.725921869277954,
      "step": 15090
    },
    {
      "dynamics_loss_avg": 0.07161987889558077,
      "epoch": 1.4390945434570312e-05,
      "step": 15090
    },
    {
      "action_loss_avg": 0.00537364250048995,
      "epoch": 1.4390945434570312e-05,
      "step": 15090
    },
    {
      "epoch": 1.4390945434570312e-05,
      "step": 15090,
      "training_step_time": 8.61519193649292
    },
    {
      "epoch": 1.4391899108886718e-05,
      "model_forward_time": 2.883998155593872,
      "step": 15091
    },
    {
      "epoch": 1.4391899108886718e-05,
      "step": 15091,
      "training_step_time": 9.987357378005981
    },
    {
      "epoch": 1.4392852783203126e-05,
      "model_forward_time": 1.7563838958740234,
      "step": 15092
    },
    {
      "epoch": 1.4392852783203126e-05,
      "step": 15092,
      "training_step_time": 8.8567476272583
    },
    {
      "epoch": 1.4393806457519531e-05,
      "model_forward_time": 1.7559669017791748,
      "step": 15093
    },
    {
      "epoch": 1.4393806457519531e-05,
      "step": 15093,
      "training_step_time": 7.570315837860107
    },
    {
      "epoch": 1.4394760131835937e-05,
      "model_forward_time": 1.75270414352417,
      "step": 15094
    },
    {
      "epoch": 1.4394760131835937e-05,
      "step": 15094,
      "training_step_time": 6.910326719284058
    },
    {
      "epoch": 1.4395713806152344e-05,
      "model_forward_time": 1.3765392303466797,
      "step": 15095
    },
    {
      "epoch": 1.4395713806152344e-05,
      "step": 15095,
      "training_step_time": 6.876805067062378
    },
    {
      "epoch": 1.439666748046875e-05,
      "model_forward_time": 1.7375879287719727,
      "step": 15096
    },
    {
      "epoch": 1.439666748046875e-05,
      "step": 15096,
      "training_step_time": 6.80884575843811
    },
    {
      "epoch": 1.4397621154785156e-05,
      "model_forward_time": 1.7378020286560059,
      "step": 15097
    },
    {
      "epoch": 1.4397621154785156e-05,
      "step": 15097,
      "training_step_time": 6.847334623336792
    },
    {
      "epoch": 1.4398574829101563e-05,
      "model_forward_time": 1.7455427646636963,
      "step": 15098
    },
    {
      "epoch": 1.4398574829101563e-05,
      "step": 15098,
      "training_step_time": 6.87464714050293
    },
    {
      "epoch": 1.4399528503417969e-05,
      "model_forward_time": 1.7461769580841064,
      "step": 15099
    },
    {
      "epoch": 1.4399528503417969e-05,
      "step": 15099,
      "training_step_time": 7.665327548980713
    },
    {
      "epoch": 1.4400482177734375e-05,
      "grad_norm": 0.2769047021865845,
      "learning_rate": 9.723745754668148e-06,
      "loss": 0.0701,
      "step": 15100
    },
    {
      "epoch": 1.4400482177734375e-05,
      "model_forward_time": 1.7591865062713623,
      "step": 15100
    },
    {
      "dynamics_loss_avg": 0.08495590612292289,
      "epoch": 1.4400482177734375e-05,
      "step": 15100
    },
    {
      "action_loss_avg": 0.004125534137710929,
      "epoch": 1.4400482177734375e-05,
      "step": 15100
    },
    {
      "epoch": 1.4400482177734375e-05,
      "step": 15100,
      "training_step_time": 8.612353563308716
    },
    {
      "epoch": 1.4401435852050782e-05,
      "model_forward_time": 2.89101243019104,
      "step": 15101
    },
    {
      "epoch": 1.4401435852050782e-05,
      "step": 15101,
      "training_step_time": 9.843435049057007
    },
    {
      "epoch": 1.4402389526367188e-05,
      "model_forward_time": 1.7404685020446777,
      "step": 15102
    },
    {
      "epoch": 1.4402389526367188e-05,
      "step": 15102,
      "training_step_time": 7.723366975784302
    },
    {
      "epoch": 1.4403343200683594e-05,
      "model_forward_time": 1.7538721561431885,
      "step": 15103
    },
    {
      "epoch": 1.4403343200683594e-05,
      "step": 15103,
      "training_step_time": 7.18806266784668
    },
    {
      "epoch": 1.4404296875e-05,
      "model_forward_time": 1.7415471076965332,
      "step": 15104
    },
    {
      "epoch": 1.4404296875e-05,
      "step": 15104,
      "training_step_time": 6.829607725143433
    },
    {
      "epoch": 1.4405250549316407e-05,
      "model_forward_time": 1.7492594718933105,
      "step": 15105
    },
    {
      "epoch": 1.4405250549316407e-05,
      "step": 15105,
      "training_step_time": 6.826565980911255
    },
    {
      "epoch": 1.4406204223632812e-05,
      "model_forward_time": 1.7533481121063232,
      "step": 15106
    },
    {
      "epoch": 1.4406204223632812e-05,
      "step": 15106,
      "training_step_time": 6.839476108551025
    },
    {
      "epoch": 1.4407157897949218e-05,
      "model_forward_time": 1.7364280223846436,
      "step": 15107
    },
    {
      "epoch": 1.4407157897949218e-05,
      "step": 15107,
      "training_step_time": 6.815082788467407
    },
    {
      "epoch": 1.4408111572265626e-05,
      "model_forward_time": 1.7346389293670654,
      "step": 15108
    },
    {
      "epoch": 1.4408111572265626e-05,
      "step": 15108,
      "training_step_time": 7.6598405838012695
    },
    {
      "epoch": 1.4409065246582031e-05,
      "model_forward_time": 1.7400004863739014,
      "step": 15109
    },
    {
      "epoch": 1.4409065246582031e-05,
      "step": 15109,
      "training_step_time": 8.464416265487671
    },
    {
      "epoch": 1.4410018920898437e-05,
      "grad_norm": 0.21348701417446136,
      "learning_rate": 9.723203498560313e-06,
      "loss": 0.0703,
      "step": 15110
    },
    {
      "epoch": 1.4410018920898437e-05,
      "model_forward_time": 2.9405059814453125,
      "step": 15110
    },
    {
      "dynamics_loss_avg": 0.05066165430471301,
      "epoch": 1.4410018920898437e-05,
      "step": 15110
    },
    {
      "action_loss_avg": 0.002379800355993211,
      "epoch": 1.4410018920898437e-05,
      "step": 15110
    },
    {
      "epoch": 1.4410018920898437e-05,
      "step": 15110,
      "training_step_time": 9.625449180603027
    },
    {
      "epoch": 1.4410972595214844e-05,
      "model_forward_time": 1.7476232051849365,
      "step": 15111
    },
    {
      "epoch": 1.4410972595214844e-05,
      "step": 15111,
      "training_step_time": 8.818610429763794
    },
    {
      "epoch": 1.441192626953125e-05,
      "model_forward_time": 1.7359857559204102,
      "step": 15112
    },
    {
      "epoch": 1.441192626953125e-05,
      "step": 15112,
      "training_step_time": 7.225434064865112
    },
    {
      "epoch": 1.4412879943847656e-05,
      "model_forward_time": 1.3751721382141113,
      "step": 15113
    },
    {
      "epoch": 1.4412879943847656e-05,
      "step": 15113,
      "training_step_time": 6.859794855117798
    },
    {
      "epoch": 1.4413833618164063e-05,
      "model_forward_time": 1.7446115016937256,
      "step": 15114
    },
    {
      "epoch": 1.4413833618164063e-05,
      "step": 15114,
      "training_step_time": 6.8810343742370605
    },
    {
      "epoch": 1.4414787292480469e-05,
      "model_forward_time": 1.7513272762298584,
      "step": 15115
    },
    {
      "epoch": 1.4414787292480469e-05,
      "step": 15115,
      "training_step_time": 6.891422510147095
    },
    {
      "epoch": 1.4415740966796875e-05,
      "model_forward_time": 1.7560296058654785,
      "step": 15116
    },
    {
      "epoch": 1.4415740966796875e-05,
      "step": 15116,
      "training_step_time": 6.804319620132446
    },
    {
      "epoch": 1.4416694641113282e-05,
      "model_forward_time": 1.7535924911499023,
      "step": 15117
    },
    {
      "epoch": 1.4416694641113282e-05,
      "step": 15117,
      "training_step_time": 7.6249840259552
    },
    {
      "epoch": 1.4417648315429688e-05,
      "model_forward_time": 1.7504122257232666,
      "step": 15118
    },
    {
      "epoch": 1.4417648315429688e-05,
      "step": 15118,
      "training_step_time": 8.294246196746826
    },
    {
      "epoch": 1.4418601989746094e-05,
      "model_forward_time": 2.9272916316986084,
      "step": 15119
    },
    {
      "epoch": 1.4418601989746094e-05,
      "step": 15119,
      "training_step_time": 10.265300273895264
    },
    {
      "epoch": 1.44195556640625e-05,
      "grad_norm": 0.1994410902261734,
      "learning_rate": 9.722660725930046e-06,
      "loss": 0.0645,
      "step": 15120
    },
    {
      "epoch": 1.44195556640625e-05,
      "model_forward_time": 1.7328636646270752,
      "step": 15120
    },
    {
      "dynamics_loss_avg": 0.051632587984204295,
      "epoch": 1.44195556640625e-05,
      "step": 15120
    },
    {
      "action_loss_avg": 0.003044471435714513,
      "epoch": 1.44195556640625e-05,
      "step": 15120
    },
    {
      "epoch": 1.44195556640625e-05,
      "step": 15120,
      "training_step_time": 7.7567524909973145
    },
    {
      "epoch": 1.4420509338378907e-05,
      "model_forward_time": 1.7301061153411865,
      "step": 15121
    },
    {
      "epoch": 1.4420509338378907e-05,
      "step": 15121,
      "training_step_time": 7.706806659698486
    },
    {
      "epoch": 1.4421463012695312e-05,
      "model_forward_time": 1.7401108741760254,
      "step": 15122
    },
    {
      "epoch": 1.4421463012695312e-05,
      "step": 15122,
      "training_step_time": 6.87309455871582
    },
    {
      "epoch": 1.4422416687011718e-05,
      "model_forward_time": 1.7675244808197021,
      "step": 15123
    },
    {
      "epoch": 1.4422416687011718e-05,
      "step": 15123,
      "training_step_time": 6.86071515083313
    },
    {
      "epoch": 1.4423370361328126e-05,
      "model_forward_time": 1.7560672760009766,
      "step": 15124
    },
    {
      "epoch": 1.4423370361328126e-05,
      "step": 15124,
      "training_step_time": 6.858419895172119
    },
    {
      "epoch": 1.4424324035644531e-05,
      "model_forward_time": 1.7349958419799805,
      "step": 15125
    },
    {
      "epoch": 1.4424324035644531e-05,
      "step": 15125,
      "training_step_time": 6.896425247192383
    },
    {
      "epoch": 1.4425277709960937e-05,
      "model_forward_time": 1.7617926597595215,
      "step": 15126
    },
    {
      "epoch": 1.4425277709960937e-05,
      "step": 15126,
      "training_step_time": 7.561715126037598
    },
    {
      "epoch": 1.4426231384277344e-05,
      "model_forward_time": 1.7559950351715088,
      "step": 15127
    },
    {
      "epoch": 1.4426231384277344e-05,
      "step": 15127,
      "training_step_time": 7.823145866394043
    },
    {
      "epoch": 1.442718505859375e-05,
      "model_forward_time": 2.909640312194824,
      "step": 15128
    },
    {
      "epoch": 1.442718505859375e-05,
      "step": 15128,
      "training_step_time": 8.61447548866272
    },
    {
      "epoch": 1.4428138732910156e-05,
      "model_forward_time": 1.7498877048492432,
      "step": 15129
    },
    {
      "epoch": 1.4428138732910156e-05,
      "step": 15129,
      "training_step_time": 8.631556510925293
    },
    {
      "epoch": 1.4429092407226563e-05,
      "grad_norm": 0.23648449778556824,
      "learning_rate": 9.722117436836701e-06,
      "loss": 0.0675,
      "step": 15130
    },
    {
      "epoch": 1.4429092407226563e-05,
      "model_forward_time": 1.7294821739196777,
      "step": 15130
    },
    {
      "dynamics_loss_avg": 0.0719839408993721,
      "epoch": 1.4429092407226563e-05,
      "step": 15130
    },
    {
      "action_loss_avg": 0.003113646269775927,
      "epoch": 1.4429092407226563e-05,
      "step": 15130
    },
    {
      "epoch": 1.4429092407226563e-05,
      "step": 15130,
      "training_step_time": 7.6755359172821045
    },
    {
      "epoch": 1.4430046081542969e-05,
      "model_forward_time": 1.7630424499511719,
      "step": 15131
    },
    {
      "epoch": 1.4430046081542969e-05,
      "step": 15131,
      "training_step_time": 7.1405956745147705
    },
    {
      "epoch": 1.4430999755859375e-05,
      "model_forward_time": 1.7599835395812988,
      "step": 15132
    },
    {
      "epoch": 1.4430999755859375e-05,
      "step": 15132,
      "training_step_time": 6.7947375774383545
    },
    {
      "epoch": 1.443195343017578e-05,
      "model_forward_time": 1.7442400455474854,
      "step": 15133
    },
    {
      "epoch": 1.443195343017578e-05,
      "step": 15133,
      "training_step_time": 6.813894748687744
    },
    {
      "epoch": 1.4432907104492188e-05,
      "model_forward_time": 1.736220121383667,
      "step": 15134
    },
    {
      "epoch": 1.4432907104492188e-05,
      "step": 15134,
      "training_step_time": 6.722487211227417
    },
    {
      "epoch": 1.4433860778808594e-05,
      "model_forward_time": 1.7600243091583252,
      "step": 15135
    },
    {
      "epoch": 1.4433860778808594e-05,
      "step": 15135,
      "training_step_time": 8.26866865158081
    },
    {
      "epoch": 1.4434814453125e-05,
      "model_forward_time": 1.751979112625122,
      "step": 15136
    },
    {
      "epoch": 1.4434814453125e-05,
      "step": 15136,
      "training_step_time": 8.686070680618286
    },
    {
      "epoch": 1.4435768127441407e-05,
      "model_forward_time": 1.8111121654510498,
      "step": 15137
    },
    {
      "epoch": 1.4435768127441407e-05,
      "step": 15137,
      "training_step_time": 10.318281173706055
    },
    {
      "epoch": 1.4436721801757812e-05,
      "model_forward_time": 1.7568371295928955,
      "step": 15138
    },
    {
      "epoch": 1.4436721801757812e-05,
      "step": 15138,
      "training_step_time": 7.796957492828369
    },
    {
      "epoch": 1.4437675476074218e-05,
      "model_forward_time": 1.7255828380584717,
      "step": 15139
    },
    {
      "epoch": 1.4437675476074218e-05,
      "step": 15139,
      "training_step_time": 7.623887777328491
    },
    {
      "epoch": 1.4438629150390626e-05,
      "grad_norm": 0.19316045939922333,
      "learning_rate": 9.721573631339697e-06,
      "loss": 0.0705,
      "step": 15140
    },
    {
      "epoch": 1.4438629150390626e-05,
      "model_forward_time": 1.7411868572235107,
      "step": 15140
    },
    {
      "dynamics_loss_avg": 0.07030214369297028,
      "epoch": 1.4438629150390626e-05,
      "step": 15140
    },
    {
      "action_loss_avg": 0.004559251666069031,
      "epoch": 1.4438629150390626e-05,
      "step": 15140
    },
    {
      "epoch": 1.4438629150390626e-05,
      "step": 15140,
      "training_step_time": 7.064121246337891
    },
    {
      "epoch": 1.4439582824707031e-05,
      "model_forward_time": 1.7553069591522217,
      "step": 15141
    },
    {
      "epoch": 1.4439582824707031e-05,
      "step": 15141,
      "training_step_time": 6.700211763381958
    },
    {
      "epoch": 1.4440536499023437e-05,
      "model_forward_time": 1.761098861694336,
      "step": 15142
    },
    {
      "epoch": 1.4440536499023437e-05,
      "step": 15142,
      "training_step_time": 6.726662874221802
    },
    {
      "epoch": 1.4441490173339844e-05,
      "model_forward_time": 1.7364277839660645,
      "step": 15143
    },
    {
      "epoch": 1.4441490173339844e-05,
      "step": 15143,
      "training_step_time": 6.722782373428345
    },
    {
      "epoch": 1.444244384765625e-05,
      "model_forward_time": 1.7252130508422852,
      "step": 15144
    },
    {
      "epoch": 1.444244384765625e-05,
      "step": 15144,
      "training_step_time": 8.428980112075806
    },
    {
      "epoch": 1.4443397521972656e-05,
      "model_forward_time": 1.7379918098449707,
      "step": 15145
    },
    {
      "epoch": 1.4443397521972656e-05,
      "step": 15145,
      "training_step_time": 7.672896862030029
    },
    {
      "epoch": 1.4444351196289063e-05,
      "model_forward_time": 1.7787232398986816,
      "step": 15146
    },
    {
      "epoch": 1.4444351196289063e-05,
      "step": 15146,
      "training_step_time": 7.790400743484497
    },
    {
      "epoch": 1.4445304870605469e-05,
      "model_forward_time": 2.8810956478118896,
      "step": 15147
    },
    {
      "epoch": 1.4445304870605469e-05,
      "step": 15147,
      "training_step_time": 8.700237035751343
    },
    {
      "epoch": 1.4446258544921875e-05,
      "model_forward_time": 1.743560791015625,
      "step": 15148
    },
    {
      "epoch": 1.4446258544921875e-05,
      "step": 15148,
      "training_step_time": 7.850565433502197
    },
    {
      "epoch": 1.444721221923828e-05,
      "model_forward_time": 1.731893539428711,
      "step": 15149
    },
    {
      "epoch": 1.444721221923828e-05,
      "step": 15149,
      "training_step_time": 7.2256834506988525
    },
    {
      "epoch": 1.4448165893554688e-05,
      "grad_norm": 0.24064037203788757,
      "learning_rate": 9.721029309498493e-06,
      "loss": 0.0691,
      "step": 15150
    },
    {
      "epoch": 1.4448165893554688e-05,
      "model_forward_time": 1.7311937808990479,
      "step": 15150
    },
    {
      "dynamics_loss_avg": 0.06942337304353714,
      "epoch": 1.4448165893554688e-05,
      "step": 15150
    },
    {
      "action_loss_avg": 0.003931787621695548,
      "epoch": 1.4448165893554688e-05,
      "step": 15150
    },
    {
      "epoch": 1.4448165893554688e-05,
      "step": 15150,
      "training_step_time": 6.787233352661133
    },
    {
      "epoch": 1.4449119567871094e-05,
      "model_forward_time": 1.7533290386199951,
      "step": 15151
    },
    {
      "epoch": 1.4449119567871094e-05,
      "step": 15151,
      "training_step_time": 6.837723970413208
    },
    {
      "epoch": 1.44500732421875e-05,
      "model_forward_time": 1.758845329284668,
      "step": 15152
    },
    {
      "epoch": 1.44500732421875e-05,
      "step": 15152,
      "training_step_time": 6.883168697357178
    },
    {
      "epoch": 1.4451026916503907e-05,
      "model_forward_time": 1.7632944583892822,
      "step": 15153
    },
    {
      "epoch": 1.4451026916503907e-05,
      "step": 15153,
      "training_step_time": 7.549630403518677
    },
    {
      "epoch": 1.4451980590820312e-05,
      "model_forward_time": 1.7413687705993652,
      "step": 15154
    },
    {
      "epoch": 1.4451980590820312e-05,
      "step": 15154,
      "training_step_time": 7.787686109542847
    },
    {
      "epoch": 1.4452934265136718e-05,
      "model_forward_time": 1.7746498584747314,
      "step": 15155
    },
    {
      "epoch": 1.4452934265136718e-05,
      "step": 15155,
      "training_step_time": 7.861627578735352
    },
    {
      "epoch": 1.4453887939453126e-05,
      "model_forward_time": 2.919912099838257,
      "step": 15156
    },
    {
      "epoch": 1.4453887939453126e-05,
      "step": 15156,
      "training_step_time": 9.391024827957153
    },
    {
      "epoch": 1.4454841613769531e-05,
      "model_forward_time": 1.758207082748413,
      "step": 15157
    },
    {
      "epoch": 1.4454841613769531e-05,
      "step": 15157,
      "training_step_time": 7.826212644577026
    },
    {
      "epoch": 1.4455795288085937e-05,
      "model_forward_time": 1.7636566162109375,
      "step": 15158
    },
    {
      "epoch": 1.4455795288085937e-05,
      "step": 15158,
      "training_step_time": 7.714289665222168
    },
    {
      "epoch": 1.4456748962402344e-05,
      "model_forward_time": 1.7582736015319824,
      "step": 15159
    },
    {
      "epoch": 1.4456748962402344e-05,
      "step": 15159,
      "training_step_time": 6.828006744384766
    },
    {
      "epoch": 1.445770263671875e-05,
      "grad_norm": 0.2127859890460968,
      "learning_rate": 9.720484471372626e-06,
      "loss": 0.0669,
      "step": 15160
    },
    {
      "epoch": 1.445770263671875e-05,
      "model_forward_time": 1.7916254997253418,
      "step": 15160
    },
    {
      "dynamics_loss_avg": 0.07246486842632294,
      "epoch": 1.445770263671875e-05,
      "step": 15160
    },
    {
      "action_loss_avg": 0.004902102262713015,
      "epoch": 1.445770263671875e-05,
      "step": 15160
    },
    {
      "epoch": 1.445770263671875e-05,
      "step": 15160,
      "training_step_time": 6.818974733352661
    },
    {
      "epoch": 1.4458656311035156e-05,
      "model_forward_time": 1.7545981407165527,
      "step": 15161
    },
    {
      "epoch": 1.4458656311035156e-05,
      "step": 15161,
      "training_step_time": 6.8517773151397705
    },
    {
      "epoch": 1.4459609985351563e-05,
      "model_forward_time": 1.7871696949005127,
      "step": 15162
    },
    {
      "epoch": 1.4459609985351563e-05,
      "step": 15162,
      "training_step_time": 7.573443412780762
    },
    {
      "epoch": 1.4460563659667969e-05,
      "model_forward_time": 1.7565388679504395,
      "step": 15163
    },
    {
      "epoch": 1.4460563659667969e-05,
      "step": 15163,
      "training_step_time": 7.744463205337524
    },
    {
      "epoch": 1.4461517333984375e-05,
      "model_forward_time": 1.7512710094451904,
      "step": 15164
    },
    {
      "epoch": 1.4461517333984375e-05,
      "step": 15164,
      "training_step_time": 8.614519357681274
    },
    {
      "epoch": 1.446247100830078e-05,
      "model_forward_time": 2.866410255432129,
      "step": 15165
    },
    {
      "epoch": 1.446247100830078e-05,
      "step": 15165,
      "training_step_time": 9.497869729995728
    },
    {
      "epoch": 1.4463424682617188e-05,
      "model_forward_time": 1.759324550628662,
      "step": 15166
    },
    {
      "epoch": 1.4463424682617188e-05,
      "step": 15166,
      "training_step_time": 7.716526985168457
    },
    {
      "epoch": 1.4464378356933594e-05,
      "model_forward_time": 1.7670247554779053,
      "step": 15167
    },
    {
      "epoch": 1.4464378356933594e-05,
      "step": 15167,
      "training_step_time": 7.710035800933838
    },
    {
      "epoch": 1.446533203125e-05,
      "model_forward_time": 1.760085105895996,
      "step": 15168
    },
    {
      "epoch": 1.446533203125e-05,
      "step": 15168,
      "training_step_time": 7.193078994750977
    },
    {
      "epoch": 1.4466285705566407e-05,
      "model_forward_time": 1.7673959732055664,
      "step": 15169
    },
    {
      "epoch": 1.4466285705566407e-05,
      "step": 15169,
      "training_step_time": 6.86647891998291
    },
    {
      "epoch": 1.4467239379882812e-05,
      "grad_norm": 0.16134695708751678,
      "learning_rate": 9.719939117021674e-06,
      "loss": 0.0655,
      "step": 15170
    },
    {
      "epoch": 1.4467239379882812e-05,
      "model_forward_time": 1.7417469024658203,
      "step": 15170
    },
    {
      "dynamics_loss_avg": 0.06184842549264431,
      "epoch": 1.4467239379882812e-05,
      "step": 15170
    },
    {
      "action_loss_avg": 0.0036404160200618207,
      "epoch": 1.4467239379882812e-05,
      "step": 15170
    },
    {
      "epoch": 1.4467239379882812e-05,
      "step": 15170,
      "training_step_time": 6.756786823272705
    },
    {
      "epoch": 1.4468193054199218e-05,
      "model_forward_time": 1.7277090549468994,
      "step": 15171
    },
    {
      "epoch": 1.4468193054199218e-05,
      "step": 15171,
      "training_step_time": 8.314027070999146
    },
    {
      "epoch": 1.4469146728515626e-05,
      "model_forward_time": 1.7612168788909912,
      "step": 15172
    },
    {
      "epoch": 1.4469146728515626e-05,
      "step": 15172,
      "training_step_time": 7.663747072219849
    },
    {
      "epoch": 1.4470100402832031e-05,
      "model_forward_time": 1.7542016506195068,
      "step": 15173
    },
    {
      "epoch": 1.4470100402832031e-05,
      "step": 15173,
      "training_step_time": 7.750908851623535
    },
    {
      "epoch": 1.4471054077148437e-05,
      "model_forward_time": 2.9085874557495117,
      "step": 15174
    },
    {
      "epoch": 1.4471054077148437e-05,
      "step": 15174,
      "training_step_time": 9.541584968566895
    },
    {
      "epoch": 1.4472007751464844e-05,
      "model_forward_time": 1.7629985809326172,
      "step": 15175
    },
    {
      "epoch": 1.4472007751464844e-05,
      "step": 15175,
      "training_step_time": 8.372986316680908
    },
    {
      "epoch": 1.447296142578125e-05,
      "model_forward_time": 1.7681150436401367,
      "step": 15176
    },
    {
      "epoch": 1.447296142578125e-05,
      "step": 15176,
      "training_step_time": 7.8493382930755615
    },
    {
      "epoch": 1.4473915100097656e-05,
      "model_forward_time": 1.767348289489746,
      "step": 15177
    },
    {
      "epoch": 1.4473915100097656e-05,
      "step": 15177,
      "training_step_time": 7.263592481613159
    },
    {
      "epoch": 1.4474868774414063e-05,
      "model_forward_time": 1.7701306343078613,
      "step": 15178
    },
    {
      "epoch": 1.4474868774414063e-05,
      "step": 15178,
      "training_step_time": 6.926502704620361
    },
    {
      "epoch": 1.4475822448730469e-05,
      "model_forward_time": 1.7463951110839844,
      "step": 15179
    },
    {
      "epoch": 1.4475822448730469e-05,
      "step": 15179,
      "training_step_time": 6.952498435974121
    },
    {
      "epoch": 1.4476776123046875e-05,
      "grad_norm": 0.19269360601902008,
      "learning_rate": 9.719393246505275e-06,
      "loss": 0.0637,
      "step": 15180
    },
    {
      "epoch": 1.4476776123046875e-05,
      "model_forward_time": 1.7610983848571777,
      "step": 15180
    },
    {
      "dynamics_loss_avg": 0.07587442211806775,
      "epoch": 1.4476776123046875e-05,
      "step": 15180
    },
    {
      "action_loss_avg": 0.004206897970288992,
      "epoch": 1.4476776123046875e-05,
      "step": 15180
    },
    {
      "epoch": 1.4476776123046875e-05,
      "step": 15180,
      "training_step_time": 7.548375368118286
    },
    {
      "epoch": 1.447772979736328e-05,
      "model_forward_time": 1.7699899673461914,
      "step": 15181
    },
    {
      "epoch": 1.447772979736328e-05,
      "step": 15181,
      "training_step_time": 7.836309194564819
    },
    {
      "epoch": 1.4478683471679688e-05,
      "model_forward_time": 1.7623000144958496,
      "step": 15182
    },
    {
      "epoch": 1.4478683471679688e-05,
      "step": 15182,
      "training_step_time": 7.7633795738220215
    },
    {
      "epoch": 1.4479637145996094e-05,
      "model_forward_time": 2.8866848945617676,
      "step": 15183
    },
    {
      "epoch": 1.4479637145996094e-05,
      "step": 15183,
      "training_step_time": 8.901116609573364
    },
    {
      "epoch": 1.44805908203125e-05,
      "model_forward_time": 1.7444884777069092,
      "step": 15184
    },
    {
      "epoch": 1.44805908203125e-05,
      "step": 15184,
      "training_step_time": 9.264435052871704
    },
    {
      "epoch": 1.4481544494628907e-05,
      "model_forward_time": 1.731670618057251,
      "step": 15185
    },
    {
      "epoch": 1.4481544494628907e-05,
      "step": 15185,
      "training_step_time": 7.646947622299194
    },
    {
      "epoch": 1.4482498168945312e-05,
      "model_forward_time": 1.7650830745697021,
      "step": 15186
    },
    {
      "epoch": 1.4482498168945312e-05,
      "step": 15186,
      "training_step_time": 7.299960374832153
    },
    {
      "epoch": 1.4483451843261718e-05,
      "model_forward_time": 1.7371528148651123,
      "step": 15187
    },
    {
      "epoch": 1.4483451843261718e-05,
      "step": 15187,
      "training_step_time": 6.881530523300171
    },
    {
      "epoch": 1.4484405517578126e-05,
      "model_forward_time": 1.7371315956115723,
      "step": 15188
    },
    {
      "epoch": 1.4484405517578126e-05,
      "step": 15188,
      "training_step_time": 6.917252063751221
    },
    {
      "epoch": 1.4485359191894531e-05,
      "model_forward_time": 1.7357177734375,
      "step": 15189
    },
    {
      "epoch": 1.4485359191894531e-05,
      "step": 15189,
      "training_step_time": 7.363257169723511
    },
    {
      "epoch": 1.4486312866210937e-05,
      "grad_norm": 0.1651640385389328,
      "learning_rate": 9.718846859883128e-06,
      "loss": 0.0658,
      "step": 15190
    },
    {
      "epoch": 1.4486312866210937e-05,
      "model_forward_time": 1.7431137561798096,
      "step": 15190
    },
    {
      "dynamics_loss_avg": 0.06800514943897724,
      "epoch": 1.4486312866210937e-05,
      "step": 15190
    },
    {
      "action_loss_avg": 0.003112984582548961,
      "epoch": 1.4486312866210937e-05,
      "step": 15190
    },
    {
      "epoch": 1.4486312866210937e-05,
      "step": 15190,
      "training_step_time": 8.400764465332031
    },
    {
      "epoch": 1.4487266540527344e-05,
      "model_forward_time": 1.7477657794952393,
      "step": 15191
    },
    {
      "epoch": 1.4487266540527344e-05,
      "step": 15191,
      "training_step_time": 7.804736137390137
    },
    {
      "epoch": 1.448822021484375e-05,
      "model_forward_time": 1.769294261932373,
      "step": 15192
    },
    {
      "epoch": 1.448822021484375e-05,
      "step": 15192,
      "training_step_time": 8.699064254760742
    },
    {
      "epoch": 1.4489173889160156e-05,
      "model_forward_time": 2.940202474594116,
      "step": 15193
    },
    {
      "epoch": 1.4489173889160156e-05,
      "step": 15193,
      "training_step_time": 9.307044982910156
    },
    {
      "epoch": 1.4490127563476563e-05,
      "model_forward_time": 1.7637534141540527,
      "step": 15194
    },
    {
      "epoch": 1.4490127563476563e-05,
      "step": 15194,
      "training_step_time": 7.843760251998901
    },
    {
      "epoch": 1.4491081237792969e-05,
      "model_forward_time": 1.761244535446167,
      "step": 15195
    },
    {
      "epoch": 1.4491081237792969e-05,
      "step": 15195,
      "training_step_time": 7.79436993598938
    },
    {
      "epoch": 1.4492034912109375e-05,
      "model_forward_time": 1.767061710357666,
      "step": 15196
    },
    {
      "epoch": 1.4492034912109375e-05,
      "step": 15196,
      "training_step_time": 7.253835916519165
    },
    {
      "epoch": 1.449298858642578e-05,
      "model_forward_time": 1.7658097743988037,
      "step": 15197
    },
    {
      "epoch": 1.449298858642578e-05,
      "step": 15197,
      "training_step_time": 6.871295213699341
    },
    {
      "epoch": 1.4493942260742188e-05,
      "model_forward_time": 1.758059024810791,
      "step": 15198
    },
    {
      "epoch": 1.4493942260742188e-05,
      "step": 15198,
      "training_step_time": 8.325321674346924
    },
    {
      "epoch": 1.4494895935058594e-05,
      "model_forward_time": 1.7700285911560059,
      "step": 15199
    },
    {
      "epoch": 1.4494895935058594e-05,
      "step": 15199,
      "training_step_time": 8.417749643325806
    },
    {
      "epoch": 1.4495849609375e-05,
      "grad_norm": 0.1767643839120865,
      "learning_rate": 9.718299957214982e-06,
      "loss": 0.0635,
      "step": 15200
    },
    {
      "epoch": 1.4495849609375e-05,
      "model_forward_time": 1.7673485279083252,
      "step": 15200
    },
    {
      "dynamics_loss_avg": 0.06693398989737034,
      "epoch": 1.4495849609375e-05,
      "step": 15200
    },
    {
      "action_loss_avg": 0.005121855763718486,
      "epoch": 1.4495849609375e-05,
      "step": 15200
    },
    {
      "epoch": 1.4495849609375e-05,
      "step": 15200,
      "training_step_time": 7.838148832321167
    },
    {
      "epoch": 1.4496803283691407e-05,
      "model_forward_time": 1.8227815628051758,
      "step": 15201
    },
    {
      "epoch": 1.4496803283691407e-05,
      "step": 15201,
      "training_step_time": 7.863778591156006
    },
    {
      "epoch": 1.4497756958007812e-05,
      "model_forward_time": 2.978973627090454,
      "step": 15202
    },
    {
      "epoch": 1.4497756958007812e-05,
      "step": 15202,
      "training_step_time": 8.6701819896698
    },
    {
      "epoch": 1.4498710632324218e-05,
      "model_forward_time": 1.766460657119751,
      "step": 15203
    },
    {
      "epoch": 1.4498710632324218e-05,
      "step": 15203,
      "training_step_time": 7.932547330856323
    },
    {
      "epoch": 1.4499664306640626e-05,
      "model_forward_time": 1.7619092464447021,
      "step": 15204
    },
    {
      "epoch": 1.4499664306640626e-05,
      "step": 15204,
      "training_step_time": 7.824981689453125
    },
    {
      "epoch": 1.4500617980957031e-05,
      "model_forward_time": 1.9369285106658936,
      "step": 15205
    },
    {
      "epoch": 1.4500617980957031e-05,
      "step": 15205,
      "training_step_time": 7.26534366607666
    },
    {
      "epoch": 1.4501571655273437e-05,
      "model_forward_time": 1.925065040588379,
      "step": 15206
    },
    {
      "epoch": 1.4501571655273437e-05,
      "step": 15206,
      "training_step_time": 6.940349578857422
    },
    {
      "epoch": 1.4502525329589844e-05,
      "model_forward_time": 1.9212579727172852,
      "step": 15207
    },
    {
      "epoch": 1.4502525329589844e-05,
      "step": 15207,
      "training_step_time": 7.504320859909058
    },
    {
      "epoch": 1.450347900390625e-05,
      "model_forward_time": 1.9399652481079102,
      "step": 15208
    },
    {
      "epoch": 1.450347900390625e-05,
      "step": 15208,
      "training_step_time": 8.979235887527466
    },
    {
      "epoch": 1.4504432678222656e-05,
      "model_forward_time": 1.929307460784912,
      "step": 15209
    },
    {
      "epoch": 1.4504432678222656e-05,
      "step": 15209,
      "training_step_time": 7.880650281906128
    },
    {
      "epoch": 1.4505386352539063e-05,
      "grad_norm": 0.14074188470840454,
      "learning_rate": 9.717752538560647e-06,
      "loss": 0.0626,
      "step": 15210
    },
    {
      "epoch": 1.4505386352539063e-05,
      "model_forward_time": 1.94987154006958,
      "step": 15210
    },
    {
      "dynamics_loss_avg": 0.06646523978561163,
      "epoch": 1.4505386352539063e-05,
      "step": 15210
    },
    {
      "action_loss_avg": 0.0030546231660991906,
      "epoch": 1.4505386352539063e-05,
      "step": 15210
    },
    {
      "epoch": 1.4505386352539063e-05,
      "step": 15210,
      "training_step_time": 8.892378091812134
    },
    {
      "epoch": 1.4506340026855469e-05,
      "model_forward_time": 3.086519479751587,
      "step": 15211
    },
    {
      "epoch": 1.4506340026855469e-05,
      "step": 15211,
      "training_step_time": 8.596509218215942
    },
    {
      "epoch": 1.4507293701171875e-05,
      "model_forward_time": 1.9072473049163818,
      "step": 15212
    },
    {
      "epoch": 1.4507293701171875e-05,
      "step": 15212,
      "training_step_time": 7.934630393981934
    },
    {
      "epoch": 1.450824737548828e-05,
      "model_forward_time": 1.9224553108215332,
      "step": 15213
    },
    {
      "epoch": 1.450824737548828e-05,
      "step": 15213,
      "training_step_time": 7.882059812545776
    },
    {
      "epoch": 1.4509201049804688e-05,
      "model_forward_time": 1.9430017471313477,
      "step": 15214
    },
    {
      "epoch": 1.4509201049804688e-05,
      "step": 15214,
      "training_step_time": 7.324333429336548
    },
    {
      "epoch": 1.4510154724121094e-05,
      "model_forward_time": 1.9168925285339355,
      "step": 15215
    },
    {
      "epoch": 1.4510154724121094e-05,
      "step": 15215,
      "training_step_time": 6.93037486076355
    },
    {
      "epoch": 1.45111083984375e-05,
      "model_forward_time": 1.9145119190216064,
      "step": 15216
    },
    {
      "epoch": 1.45111083984375e-05,
      "step": 15216,
      "training_step_time": 7.508625030517578
    },
    {
      "epoch": 1.4512062072753907e-05,
      "model_forward_time": 1.9165105819702148,
      "step": 15217
    },
    {
      "epoch": 1.4512062072753907e-05,
      "step": 15217,
      "training_step_time": 7.92419695854187
    },
    {
      "epoch": 1.4513015747070312e-05,
      "model_forward_time": 1.2039093971252441,
      "step": 15218
    },
    {
      "epoch": 1.4513015747070312e-05,
      "step": 15218,
      "training_step_time": 8.384424209594727
    },
    {
      "epoch": 1.4513969421386718e-05,
      "model_forward_time": 1.5718433856964111,
      "step": 15219
    },
    {
      "epoch": 1.4513969421386718e-05,
      "step": 15219,
      "training_step_time": 8.92300033569336
    },
    {
      "epoch": 1.4514923095703126e-05,
      "grad_norm": 0.17658041417598724,
      "learning_rate": 9.717204603979986e-06,
      "loss": 0.0645,
      "step": 15220
    },
    {
      "epoch": 1.4514923095703126e-05,
      "model_forward_time": 1.9185853004455566,
      "step": 15220
    },
    {
      "dynamics_loss_avg": 0.05267006773501635,
      "epoch": 1.4514923095703126e-05,
      "step": 15220
    },
    {
      "action_loss_avg": 0.001801524794427678,
      "epoch": 1.4514923095703126e-05,
      "step": 15220
    },
    {
      "epoch": 1.4514923095703126e-05,
      "step": 15220,
      "training_step_time": 9.81615924835205
    },
    {
      "epoch": 1.4515876770019531e-05,
      "model_forward_time": 2.999793529510498,
      "step": 15221
    },
    {
      "epoch": 1.4515876770019531e-05,
      "step": 15221,
      "training_step_time": 10.262428522109985
    },
    {
      "epoch": 1.4516830444335937e-05,
      "model_forward_time": 1.9235644340515137,
      "step": 15222
    },
    {
      "epoch": 1.4516830444335937e-05,
      "step": 15222,
      "training_step_time": 7.853917360305786
    },
    {
      "epoch": 1.4517784118652344e-05,
      "model_forward_time": 1.9045827388763428,
      "step": 15223
    },
    {
      "epoch": 1.4517784118652344e-05,
      "step": 15223,
      "training_step_time": 7.641478061676025
    },
    {
      "epoch": 1.451873779296875e-05,
      "model_forward_time": 1.9104206562042236,
      "step": 15224
    },
    {
      "epoch": 1.451873779296875e-05,
      "step": 15224,
      "training_step_time": 6.993764162063599
    },
    {
      "epoch": 1.4519691467285156e-05,
      "model_forward_time": 1.9309945106506348,
      "step": 15225
    },
    {
      "epoch": 1.4519691467285156e-05,
      "step": 15225,
      "training_step_time": 7.51219367980957
    },
    {
      "epoch": 1.4520645141601563e-05,
      "model_forward_time": 1.914177656173706,
      "step": 15226
    },
    {
      "epoch": 1.4520645141601563e-05,
      "step": 15226,
      "training_step_time": 7.918872594833374
    },
    {
      "epoch": 1.4521598815917969e-05,
      "model_forward_time": 1.565887212753296,
      "step": 15227
    },
    {
      "epoch": 1.4521598815917969e-05,
      "step": 15227,
      "training_step_time": 7.832604885101318
    },
    {
      "epoch": 1.4522552490234375e-05,
      "model_forward_time": 1.9198598861694336,
      "step": 15228
    },
    {
      "epoch": 1.4522552490234375e-05,
      "step": 15228,
      "training_step_time": 8.716775894165039
    },
    {
      "epoch": 1.452350616455078e-05,
      "model_forward_time": 1.903106451034546,
      "step": 15229
    },
    {
      "epoch": 1.452350616455078e-05,
      "step": 15229,
      "training_step_time": 7.900352239608765
    },
    {
      "epoch": 1.4524459838867188e-05,
      "grad_norm": 0.13385717570781708,
      "learning_rate": 9.716656153532924e-06,
      "loss": 0.0617,
      "step": 15230
    },
    {
      "epoch": 1.4524459838867188e-05,
      "model_forward_time": 2.6110222339630127,
      "step": 15230
    },
    {
      "dynamics_loss_avg": 0.05691476780921221,
      "epoch": 1.4524459838867188e-05,
      "step": 15230
    },
    {
      "action_loss_avg": 0.0033380761276930572,
      "epoch": 1.4524459838867188e-05,
      "step": 15230
    },
    {
      "epoch": 1.4524459838867188e-05,
      "step": 15230,
      "training_step_time": 8.795830011367798
    },
    {
      "epoch": 1.4525413513183594e-05,
      "model_forward_time": 1.9108054637908936,
      "step": 15231
    },
    {
      "epoch": 1.4525413513183594e-05,
      "step": 15231,
      "training_step_time": 7.862483978271484
    },
    {
      "epoch": 1.45263671875e-05,
      "model_forward_time": 1.9178237915039062,
      "step": 15232
    },
    {
      "epoch": 1.45263671875e-05,
      "step": 15232,
      "training_step_time": 7.570560932159424
    },
    {
      "epoch": 1.4527320861816407e-05,
      "model_forward_time": 1.9343898296356201,
      "step": 15233
    },
    {
      "epoch": 1.4527320861816407e-05,
      "step": 15233,
      "training_step_time": 6.990411996841431
    },
    {
      "epoch": 1.4528274536132812e-05,
      "model_forward_time": 1.5614607334136963,
      "step": 15234
    },
    {
      "epoch": 1.4528274536132812e-05,
      "step": 15234,
      "training_step_time": 7.4539265632629395
    },
    {
      "epoch": 1.4529228210449218e-05,
      "model_forward_time": 1.9168791770935059,
      "step": 15235
    },
    {
      "epoch": 1.4529228210449218e-05,
      "step": 15235,
      "training_step_time": 7.848036289215088
    },
    {
      "epoch": 1.4530181884765626e-05,
      "model_forward_time": 1.939129114151001,
      "step": 15236
    },
    {
      "epoch": 1.4530181884765626e-05,
      "step": 15236,
      "training_step_time": 7.761464357376099
    },
    {
      "epoch": 1.4531135559082031e-05,
      "model_forward_time": 1.58573317527771,
      "step": 15237
    },
    {
      "epoch": 1.4531135559082031e-05,
      "step": 15237,
      "training_step_time": 8.772183895111084
    },
    {
      "epoch": 1.4532089233398437e-05,
      "model_forward_time": 1.5562522411346436,
      "step": 15238
    },
    {
      "epoch": 1.4532089233398437e-05,
      "step": 15238,
      "training_step_time": 9.457589864730835
    },
    {
      "epoch": 1.4533042907714844e-05,
      "model_forward_time": 3.0164594650268555,
      "step": 15239
    },
    {
      "epoch": 1.4533042907714844e-05,
      "step": 15239,
      "training_step_time": 8.518519639968872
    },
    {
      "epoch": 1.453399658203125e-05,
      "grad_norm": 0.14054636657238007,
      "learning_rate": 9.716107187279434e-06,
      "loss": 0.0648,
      "step": 15240
    },
    {
      "epoch": 1.453399658203125e-05,
      "model_forward_time": 1.9211511611938477,
      "step": 15240
    },
    {
      "dynamics_loss_avg": 0.06275290101766587,
      "epoch": 1.453399658203125e-05,
      "step": 15240
    },
    {
      "action_loss_avg": 0.0030657367897219956,
      "epoch": 1.453399658203125e-05,
      "step": 15240
    },
    {
      "epoch": 1.453399658203125e-05,
      "step": 15240,
      "training_step_time": 7.840150356292725
    },
    {
      "epoch": 1.4534950256347656e-05,
      "model_forward_time": 1.9150211811065674,
      "step": 15241
    },
    {
      "epoch": 1.4534950256347656e-05,
      "step": 15241,
      "training_step_time": 7.633586168289185
    },
    {
      "epoch": 1.4535903930664063e-05,
      "model_forward_time": 1.5839664936065674,
      "step": 15242
    },
    {
      "epoch": 1.4535903930664063e-05,
      "step": 15242,
      "training_step_time": 7.26111912727356
    },
    {
      "epoch": 1.4536857604980469e-05,
      "model_forward_time": 1.9456260204315186,
      "step": 15243
    },
    {
      "epoch": 1.4536857604980469e-05,
      "step": 15243,
      "training_step_time": 7.467260122299194
    },
    {
      "epoch": 1.4537811279296875e-05,
      "model_forward_time": 1.9367260932922363,
      "step": 15244
    },
    {
      "epoch": 1.4537811279296875e-05,
      "step": 15244,
      "training_step_time": 7.772033452987671
    },
    {
      "epoch": 1.453876495361328e-05,
      "model_forward_time": 1.9105892181396484,
      "step": 15245
    },
    {
      "epoch": 1.453876495361328e-05,
      "step": 15245,
      "training_step_time": 7.8035523891448975
    },
    {
      "epoch": 1.4539718627929688e-05,
      "model_forward_time": 1.9203159809112549,
      "step": 15246
    },
    {
      "epoch": 1.4539718627929688e-05,
      "step": 15246,
      "training_step_time": 8.692543983459473
    },
    {
      "epoch": 1.4540672302246094e-05,
      "model_forward_time": 1.9314193725585938,
      "step": 15247
    },
    {
      "epoch": 1.4540672302246094e-05,
      "step": 15247,
      "training_step_time": 7.829833030700684
    },
    {
      "epoch": 1.45416259765625e-05,
      "model_forward_time": 1.9136812686920166,
      "step": 15248
    },
    {
      "epoch": 1.45416259765625e-05,
      "step": 15248,
      "training_step_time": 8.439192533493042
    },
    {
      "epoch": 1.4542579650878907e-05,
      "model_forward_time": 3.0151968002319336,
      "step": 15249
    },
    {
      "epoch": 1.4542579650878907e-05,
      "step": 15249,
      "training_step_time": 8.47367548942566
    },
    {
      "epoch": 1.4543533325195312e-05,
      "grad_norm": 0.19213427603244781,
      "learning_rate": 9.715557705279554e-06,
      "loss": 0.0674,
      "step": 15250
    },
    {
      "epoch": 1.4543533325195312e-05,
      "model_forward_time": 1.5487253665924072,
      "step": 15250
    },
    {
      "dynamics_loss_avg": 0.05578317968174815,
      "epoch": 1.4543533325195312e-05,
      "step": 15250
    },
    {
      "action_loss_avg": 0.002450577507261187,
      "epoch": 1.4543533325195312e-05,
      "step": 15250
    },
    {
      "epoch": 1.4543533325195312e-05,
      "step": 15250,
      "training_step_time": 7.784239292144775
    },
    {
      "epoch": 1.4544486999511718e-05,
      "model_forward_time": 1.9289824962615967,
      "step": 15251
    },
    {
      "epoch": 1.4544486999511718e-05,
      "step": 15251,
      "training_step_time": 7.246607065200806
    },
    {
      "epoch": 1.4545440673828126e-05,
      "model_forward_time": 1.907888412475586,
      "step": 15252
    },
    {
      "epoch": 1.4545440673828126e-05,
      "step": 15252,
      "training_step_time": 7.443705320358276
    },
    {
      "epoch": 1.4546394348144531e-05,
      "model_forward_time": 1.9249019622802734,
      "step": 15253
    },
    {
      "epoch": 1.4546394348144531e-05,
      "step": 15253,
      "training_step_time": 7.801088333129883
    },
    {
      "epoch": 1.4547348022460937e-05,
      "model_forward_time": 1.8946740627288818,
      "step": 15254
    },
    {
      "epoch": 1.4547348022460937e-05,
      "step": 15254,
      "training_step_time": 8.657173871994019
    },
    {
      "epoch": 1.4548301696777344e-05,
      "model_forward_time": 1.9234609603881836,
      "step": 15255
    },
    {
      "epoch": 1.4548301696777344e-05,
      "step": 15255,
      "training_step_time": 8.84528398513794
    },
    {
      "epoch": 1.454925537109375e-05,
      "model_forward_time": 1.9368188381195068,
      "step": 15256
    },
    {
      "epoch": 1.454925537109375e-05,
      "step": 15256,
      "training_step_time": 7.889809608459473
    },
    {
      "epoch": 1.4550209045410156e-05,
      "model_forward_time": 1.9237868785858154,
      "step": 15257
    },
    {
      "epoch": 1.4550209045410156e-05,
      "step": 15257,
      "training_step_time": 9.196207046508789
    },
    {
      "epoch": 1.4551162719726562e-05,
      "model_forward_time": 3.038252115249634,
      "step": 15258
    },
    {
      "epoch": 1.4551162719726562e-05,
      "step": 15258,
      "training_step_time": 7.994111061096191
    },
    {
      "epoch": 1.4552116394042969e-05,
      "model_forward_time": 1.9252126216888428,
      "step": 15259
    },
    {
      "epoch": 1.4552116394042969e-05,
      "step": 15259,
      "training_step_time": 8.267778635025024
    },
    {
      "epoch": 1.4553070068359375e-05,
      "grad_norm": 0.15861709415912628,
      "learning_rate": 9.715007707593372e-06,
      "loss": 0.0714,
      "step": 15260
    },
    {
      "epoch": 1.4553070068359375e-05,
      "model_forward_time": 1.876664400100708,
      "step": 15260
    },
    {
      "dynamics_loss_avg": 0.062902501411736,
      "epoch": 1.4553070068359375e-05,
      "step": 15260
    },
    {
      "action_loss_avg": 0.0037268696760293096,
      "epoch": 1.4553070068359375e-05,
      "step": 15260
    },
    {
      "epoch": 1.4553070068359375e-05,
      "step": 15260,
      "training_step_time": 7.341437816619873
    },
    {
      "epoch": 1.455402374267578e-05,
      "model_forward_time": 1.9145922660827637,
      "step": 15261
    },
    {
      "epoch": 1.455402374267578e-05,
      "step": 15261,
      "training_step_time": 7.53725004196167
    },
    {
      "epoch": 1.4554977416992188e-05,
      "model_forward_time": 1.923724889755249,
      "step": 15262
    },
    {
      "epoch": 1.4554977416992188e-05,
      "step": 15262,
      "training_step_time": 7.8846399784088135
    },
    {
      "epoch": 1.4555931091308594e-05,
      "model_forward_time": 1.9152796268463135,
      "step": 15263
    },
    {
      "epoch": 1.4555931091308594e-05,
      "step": 15263,
      "training_step_time": 7.996111869812012
    },
    {
      "epoch": 1.4556884765625e-05,
      "model_forward_time": 1.9155402183532715,
      "step": 15264
    },
    {
      "epoch": 1.4556884765625e-05,
      "step": 15264,
      "training_step_time": 7.820793390274048
    },
    {
      "epoch": 1.4557838439941407e-05,
      "model_forward_time": 1.9006428718566895,
      "step": 15265
    },
    {
      "epoch": 1.4557838439941407e-05,
      "step": 15265,
      "training_step_time": 7.863144874572754
    },
    {
      "epoch": 1.4558792114257813e-05,
      "model_forward_time": 1.909867286682129,
      "step": 15266
    },
    {
      "epoch": 1.4558792114257813e-05,
      "step": 15266,
      "training_step_time": 8.39724087715149
    },
    {
      "epoch": 1.4559745788574218e-05,
      "model_forward_time": 3.0522310733795166,
      "step": 15267
    },
    {
      "epoch": 1.4559745788574218e-05,
      "step": 15267,
      "training_step_time": 10.096630811691284
    },
    {
      "epoch": 1.4560699462890626e-05,
      "model_forward_time": 1.9337308406829834,
      "step": 15268
    },
    {
      "epoch": 1.4560699462890626e-05,
      "step": 15268,
      "training_step_time": 8.017013549804688
    },
    {
      "epoch": 1.4561653137207031e-05,
      "model_forward_time": 1.5589869022369385,
      "step": 15269
    },
    {
      "epoch": 1.4561653137207031e-05,
      "step": 15269,
      "training_step_time": 7.647054672241211
    },
    {
      "epoch": 1.4562606811523437e-05,
      "grad_norm": 0.2026168406009674,
      "learning_rate": 9.714457194281037e-06,
      "loss": 0.0666,
      "step": 15270
    },
    {
      "epoch": 1.4562606811523437e-05,
      "model_forward_time": 1.9084186553955078,
      "step": 15270
    },
    {
      "dynamics_loss_avg": 0.06638391111046076,
      "epoch": 1.4562606811523437e-05,
      "step": 15270
    },
    {
      "action_loss_avg": 0.004310089908540249,
      "epoch": 1.4562606811523437e-05,
      "step": 15270
    },
    {
      "epoch": 1.4562606811523437e-05,
      "step": 15270,
      "training_step_time": 7.5414111614227295
    },
    {
      "epoch": 1.4563560485839845e-05,
      "model_forward_time": 1.908968448638916,
      "step": 15271
    },
    {
      "epoch": 1.4563560485839845e-05,
      "step": 15271,
      "training_step_time": 7.846814870834351
    },
    {
      "epoch": 1.456451416015625e-05,
      "model_forward_time": 1.9095933437347412,
      "step": 15272
    },
    {
      "epoch": 1.456451416015625e-05,
      "step": 15272,
      "training_step_time": 7.978066682815552
    },
    {
      "epoch": 1.4565467834472656e-05,
      "model_forward_time": 1.577296257019043,
      "step": 15273
    },
    {
      "epoch": 1.4565467834472656e-05,
      "step": 15273,
      "training_step_time": 7.822391986846924
    },
    {
      "epoch": 1.4566421508789062e-05,
      "model_forward_time": 1.913381814956665,
      "step": 15274
    },
    {
      "epoch": 1.4566421508789062e-05,
      "step": 15274,
      "training_step_time": 9.022801876068115
    },
    {
      "epoch": 1.4567375183105469e-05,
      "model_forward_time": 1.9058010578155518,
      "step": 15275
    },
    {
      "epoch": 1.4567375183105469e-05,
      "step": 15275,
      "training_step_time": 8.210354804992676
    },
    {
      "epoch": 1.4568328857421875e-05,
      "model_forward_time": 1.5398590564727783,
      "step": 15276
    },
    {
      "epoch": 1.4568328857421875e-05,
      "step": 15276,
      "training_step_time": 9.702727794647217
    },
    {
      "epoch": 1.456928253173828e-05,
      "model_forward_time": 2.672102212905884,
      "step": 15277
    },
    {
      "epoch": 1.456928253173828e-05,
      "step": 15277,
      "training_step_time": 7.877379894256592
    },
    {
      "epoch": 1.4570236206054688e-05,
      "model_forward_time": 1.9196794033050537,
      "step": 15278
    },
    {
      "epoch": 1.4570236206054688e-05,
      "step": 15278,
      "training_step_time": 7.859142780303955
    },
    {
      "epoch": 1.4571189880371094e-05,
      "model_forward_time": 1.9356348514556885,
      "step": 15279
    },
    {
      "epoch": 1.4571189880371094e-05,
      "step": 15279,
      "training_step_time": 7.477881669998169
    },
    {
      "epoch": 1.45721435546875e-05,
      "grad_norm": 0.16064822673797607,
      "learning_rate": 9.713906165402751e-06,
      "loss": 0.066,
      "step": 15280
    },
    {
      "epoch": 1.45721435546875e-05,
      "model_forward_time": 1.9308104515075684,
      "step": 15280
    },
    {
      "dynamics_loss_avg": 0.0794940335676074,
      "epoch": 1.45721435546875e-05,
      "step": 15280
    },
    {
      "action_loss_avg": 0.0064624572056345645,
      "epoch": 1.45721435546875e-05,
      "step": 15280
    },
    {
      "epoch": 1.45721435546875e-05,
      "step": 15280,
      "training_step_time": 7.725581645965576
    },
    {
      "epoch": 1.4573097229003907e-05,
      "model_forward_time": 1.9142005443572998,
      "step": 15281
    },
    {
      "epoch": 1.4573097229003907e-05,
      "step": 15281,
      "training_step_time": 7.80034065246582
    },
    {
      "epoch": 1.4574050903320313e-05,
      "model_forward_time": 1.9095609188079834,
      "step": 15282
    },
    {
      "epoch": 1.4574050903320313e-05,
      "step": 15282,
      "training_step_time": 8.381913185119629
    },
    {
      "epoch": 1.4575004577636718e-05,
      "model_forward_time": 1.9072084426879883,
      "step": 15283
    },
    {
      "epoch": 1.4575004577636718e-05,
      "step": 15283,
      "training_step_time": 7.786546230316162
    },
    {
      "epoch": 1.4575958251953126e-05,
      "model_forward_time": 1.9430899620056152,
      "step": 15284
    },
    {
      "epoch": 1.4575958251953126e-05,
      "step": 15284,
      "training_step_time": 8.422078132629395
    },
    {
      "epoch": 1.4576911926269531e-05,
      "model_forward_time": 1.853029727935791,
      "step": 15285
    },
    {
      "epoch": 1.4576911926269531e-05,
      "step": 15285,
      "training_step_time": 9.965111255645752
    },
    {
      "epoch": 1.4577865600585937e-05,
      "model_forward_time": 3.0289435386657715,
      "step": 15286
    },
    {
      "epoch": 1.4577865600585937e-05,
      "step": 15286,
      "training_step_time": 9.392938137054443
    },
    {
      "epoch": 1.4578819274902345e-05,
      "model_forward_time": 1.9201011657714844,
      "step": 15287
    },
    {
      "epoch": 1.4578819274902345e-05,
      "step": 15287,
      "training_step_time": 7.797397136688232
    },
    {
      "epoch": 1.457977294921875e-05,
      "model_forward_time": 1.8780722618103027,
      "step": 15288
    },
    {
      "epoch": 1.457977294921875e-05,
      "step": 15288,
      "training_step_time": 7.4696691036224365
    },
    {
      "epoch": 1.4580726623535156e-05,
      "model_forward_time": 1.9175703525543213,
      "step": 15289
    },
    {
      "epoch": 1.4580726623535156e-05,
      "step": 15289,
      "training_step_time": 7.7445292472839355
    },
    {
      "epoch": 1.4581680297851562e-05,
      "grad_norm": 0.19532158970832825,
      "learning_rate": 9.713354621018775e-06,
      "loss": 0.0651,
      "step": 15290
    },
    {
      "epoch": 1.4581680297851562e-05,
      "model_forward_time": 1.9145214557647705,
      "step": 15290
    },
    {
      "dynamics_loss_avg": 0.05110529977828264,
      "epoch": 1.4581680297851562e-05,
      "step": 15290
    },
    {
      "action_loss_avg": 0.0024221273488365114,
      "epoch": 1.4581680297851562e-05,
      "step": 15290
    },
    {
      "epoch": 1.4581680297851562e-05,
      "step": 15290,
      "training_step_time": 8.005753517150879
    },
    {
      "epoch": 1.4582633972167969e-05,
      "model_forward_time": 1.912602424621582,
      "step": 15291
    },
    {
      "epoch": 1.4582633972167969e-05,
      "step": 15291,
      "training_step_time": 8.805707454681396
    },
    {
      "epoch": 1.4583587646484375e-05,
      "model_forward_time": 1.936837911605835,
      "step": 15292
    },
    {
      "epoch": 1.4583587646484375e-05,
      "step": 15292,
      "training_step_time": 8.28607964515686
    },
    {
      "epoch": 1.458454132080078e-05,
      "model_forward_time": 1.9012880325317383,
      "step": 15293
    },
    {
      "epoch": 1.458454132080078e-05,
      "step": 15293,
      "training_step_time": 7.829219818115234
    },
    {
      "epoch": 1.4585494995117188e-05,
      "model_forward_time": 1.907428503036499,
      "step": 15294
    },
    {
      "epoch": 1.4585494995117188e-05,
      "step": 15294,
      "training_step_time": 8.581003904342651
    },
    {
      "epoch": 1.4586448669433594e-05,
      "model_forward_time": 3.04626727104187,
      "step": 15295
    },
    {
      "epoch": 1.4586448669433594e-05,
      "step": 15295,
      "training_step_time": 8.02856731414795
    },
    {
      "epoch": 1.458740234375e-05,
      "model_forward_time": 1.9416651725769043,
      "step": 15296
    },
    {
      "epoch": 1.458740234375e-05,
      "step": 15296,
      "training_step_time": 7.889010190963745
    },
    {
      "epoch": 1.4588356018066407e-05,
      "model_forward_time": 1.9276208877563477,
      "step": 15297
    },
    {
      "epoch": 1.4588356018066407e-05,
      "step": 15297,
      "training_step_time": 7.817040205001831
    },
    {
      "epoch": 1.4589309692382813e-05,
      "model_forward_time": 1.9328575134277344,
      "step": 15298
    },
    {
      "epoch": 1.4589309692382813e-05,
      "step": 15298,
      "training_step_time": 7.81969952583313
    },
    {
      "epoch": 1.4590263366699218e-05,
      "model_forward_time": 1.9202656745910645,
      "step": 15299
    },
    {
      "epoch": 1.4590263366699218e-05,
      "step": 15299,
      "training_step_time": 8.129031896591187
    },
    {
      "epoch": 1.4591217041015626e-05,
      "grad_norm": 0.16055242717266083,
      "learning_rate": 9.712802561189423e-06,
      "loss": 0.0694,
      "step": 15300
    },
    {
      "epoch": 1.4591217041015626e-05,
      "model_forward_time": 1.9205048084259033,
      "step": 15300
    },
    {
      "dynamics_loss_avg": 0.06489756368100644,
      "epoch": 1.4591217041015626e-05,
      "step": 15300
    },
    {
      "action_loss_avg": 0.005793403997085988,
      "epoch": 1.4591217041015626e-05,
      "step": 15300
    },
    {
      "epoch": 1.4591217041015626e-05,
      "step": 15300,
      "training_step_time": 7.800215721130371
    },
    {
      "epoch": 1.4592170715332031e-05,
      "model_forward_time": 1.9330127239227295,
      "step": 15301
    },
    {
      "epoch": 1.4592170715332031e-05,
      "step": 15301,
      "training_step_time": 9.702505350112915
    },
    {
      "epoch": 1.4593124389648437e-05,
      "model_forward_time": 1.9421768188476562,
      "step": 15302
    },
    {
      "epoch": 1.4593124389648437e-05,
      "step": 15302,
      "training_step_time": 9.219203233718872
    },
    {
      "epoch": 1.4594078063964845e-05,
      "model_forward_time": 1.9459359645843506,
      "step": 15303
    },
    {
      "epoch": 1.4594078063964845e-05,
      "step": 15303,
      "training_step_time": 7.608376979827881
    },
    {
      "epoch": 1.459503173828125e-05,
      "model_forward_time": 1.9104375839233398,
      "step": 15304
    },
    {
      "epoch": 1.459503173828125e-05,
      "step": 15304,
      "training_step_time": 9.699976444244385
    },
    {
      "epoch": 1.4595985412597656e-05,
      "model_forward_time": 1.9400382041931152,
      "step": 15305
    },
    {
      "epoch": 1.4595985412597656e-05,
      "step": 15305,
      "training_step_time": 7.712060451507568
    },
    {
      "epoch": 1.4596939086914062e-05,
      "model_forward_time": 1.9337706565856934,
      "step": 15306
    },
    {
      "epoch": 1.4596939086914062e-05,
      "step": 15306,
      "training_step_time": 7.568577766418457
    },
    {
      "epoch": 1.4597892761230469e-05,
      "model_forward_time": 1.9089090824127197,
      "step": 15307
    },
    {
      "epoch": 1.4597892761230469e-05,
      "step": 15307,
      "training_step_time": 7.587425947189331
    },
    {
      "epoch": 1.4598846435546875e-05,
      "model_forward_time": 1.9303627014160156,
      "step": 15308
    },
    {
      "epoch": 1.4598846435546875e-05,
      "step": 15308,
      "training_step_time": 7.355207920074463
    },
    {
      "epoch": 1.459980010986328e-05,
      "model_forward_time": 1.9257724285125732,
      "step": 15309
    },
    {
      "epoch": 1.459980010986328e-05,
      "step": 15309,
      "training_step_time": 7.742335557937622
    },
    {
      "epoch": 1.4600753784179688e-05,
      "grad_norm": 0.1928657442331314,
      "learning_rate": 9.712249985975069e-06,
      "loss": 0.0696,
      "step": 15310
    },
    {
      "epoch": 1.4600753784179688e-05,
      "model_forward_time": 1.8988614082336426,
      "step": 15310
    },
    {
      "dynamics_loss_avg": 0.08071309961378574,
      "epoch": 1.4600753784179688e-05,
      "step": 15310
    },
    {
      "action_loss_avg": 0.0029942448949441315,
      "epoch": 1.4600753784179688e-05,
      "step": 15310
    },
    {
      "epoch": 1.4600753784179688e-05,
      "step": 15310,
      "training_step_time": 7.71410870552063
    },
    {
      "epoch": 1.4601707458496094e-05,
      "model_forward_time": 1.9416897296905518,
      "step": 15311
    },
    {
      "epoch": 1.4601707458496094e-05,
      "step": 15311,
      "training_step_time": 9.66412353515625
    },
    {
      "epoch": 1.46026611328125e-05,
      "model_forward_time": 1.917118787765503,
      "step": 15312
    },
    {
      "epoch": 1.46026611328125e-05,
      "step": 15312,
      "training_step_time": 7.588577747344971
    },
    {
      "epoch": 1.4603614807128907e-05,
      "model_forward_time": 1.9181230068206787,
      "step": 15313
    },
    {
      "epoch": 1.4603614807128907e-05,
      "step": 15313,
      "training_step_time": 7.725096702575684
    },
    {
      "epoch": 1.4604568481445313e-05,
      "model_forward_time": 2.965050220489502,
      "step": 15314
    },
    {
      "epoch": 1.4604568481445313e-05,
      "step": 15314,
      "training_step_time": 7.743786573410034
    },
    {
      "epoch": 1.4605522155761718e-05,
      "model_forward_time": 1.8961849212646484,
      "step": 15315
    },
    {
      "epoch": 1.4605522155761718e-05,
      "step": 15315,
      "training_step_time": 7.6112401485443115
    },
    {
      "epoch": 1.4606475830078126e-05,
      "model_forward_time": 1.9204943180084229,
      "step": 15316
    },
    {
      "epoch": 1.4606475830078126e-05,
      "step": 15316,
      "training_step_time": 7.613159656524658
    },
    {
      "epoch": 1.4607429504394531e-05,
      "model_forward_time": 1.9346332550048828,
      "step": 15317
    },
    {
      "epoch": 1.4607429504394531e-05,
      "step": 15317,
      "training_step_time": 7.383713245391846
    },
    {
      "epoch": 1.4608383178710937e-05,
      "model_forward_time": 1.8916380405426025,
      "step": 15318
    },
    {
      "epoch": 1.4608383178710937e-05,
      "step": 15318,
      "training_step_time": 7.787308931350708
    },
    {
      "epoch": 1.4609336853027345e-05,
      "model_forward_time": 1.9307584762573242,
      "step": 15319
    },
    {
      "epoch": 1.4609336853027345e-05,
      "step": 15319,
      "training_step_time": 7.726927757263184
    },
    {
      "epoch": 1.461029052734375e-05,
      "grad_norm": 0.1373901069164276,
      "learning_rate": 9.71169689543614e-06,
      "loss": 0.0655,
      "step": 15320
    },
    {
      "epoch": 1.461029052734375e-05,
      "model_forward_time": 1.8855493068695068,
      "step": 15320
    },
    {
      "dynamics_loss_avg": 0.061028246954083445,
      "epoch": 1.461029052734375e-05,
      "step": 15320
    },
    {
      "action_loss_avg": 0.0032061382778920235,
      "epoch": 1.461029052734375e-05,
      "step": 15320
    },
    {
      "epoch": 1.461029052734375e-05,
      "step": 15320,
      "training_step_time": 9.884025573730469
    },
    {
      "epoch": 1.4611244201660156e-05,
      "model_forward_time": 1.9101760387420654,
      "step": 15321
    },
    {
      "epoch": 1.4611244201660156e-05,
      "step": 15321,
      "training_step_time": 9.071513414382935
    },
    {
      "epoch": 1.4612197875976562e-05,
      "model_forward_time": 1.916764736175537,
      "step": 15322
    },
    {
      "epoch": 1.4612197875976562e-05,
      "step": 15322,
      "training_step_time": 9.254253625869751
    },
    {
      "epoch": 1.4613151550292969e-05,
      "model_forward_time": 3.009687900543213,
      "step": 15323
    },
    {
      "epoch": 1.4613151550292969e-05,
      "step": 15323,
      "training_step_time": 9.29949951171875
    },
    {
      "epoch": 1.4614105224609375e-05,
      "model_forward_time": 1.9126677513122559,
      "step": 15324
    },
    {
      "epoch": 1.4614105224609375e-05,
      "step": 15324,
      "training_step_time": 7.699278116226196
    },
    {
      "epoch": 1.461505889892578e-05,
      "model_forward_time": 1.9254379272460938,
      "step": 15325
    },
    {
      "epoch": 1.461505889892578e-05,
      "step": 15325,
      "training_step_time": 7.621370315551758
    },
    {
      "epoch": 1.4616012573242188e-05,
      "model_forward_time": 1.9301402568817139,
      "step": 15326
    },
    {
      "epoch": 1.4616012573242188e-05,
      "step": 15326,
      "training_step_time": 7.413300514221191
    },
    {
      "epoch": 1.4616966247558594e-05,
      "model_forward_time": 1.9123411178588867,
      "step": 15327
    },
    {
      "epoch": 1.4616966247558594e-05,
      "step": 15327,
      "training_step_time": 7.771488428115845
    },
    {
      "epoch": 1.4617919921875e-05,
      "model_forward_time": 1.9069328308105469,
      "step": 15328
    },
    {
      "epoch": 1.4617919921875e-05,
      "step": 15328,
      "training_step_time": 7.642940282821655
    },
    {
      "epoch": 1.4618873596191407e-05,
      "model_forward_time": 1.922602891921997,
      "step": 15329
    },
    {
      "epoch": 1.4618873596191407e-05,
      "step": 15329,
      "training_step_time": 8.353436708450317
    },
    {
      "epoch": 1.4619827270507813e-05,
      "grad_norm": 0.17546536028385162,
      "learning_rate": 9.711143289633123e-06,
      "loss": 0.0638,
      "step": 15330
    },
    {
      "epoch": 1.4619827270507813e-05,
      "model_forward_time": 1.9167263507843018,
      "step": 15330
    },
    {
      "dynamics_loss_avg": 0.05848824623972178,
      "epoch": 1.4619827270507813e-05,
      "step": 15330
    },
    {
      "action_loss_avg": 0.003297130833379924,
      "epoch": 1.4619827270507813e-05,
      "step": 15330
    },
    {
      "epoch": 1.4619827270507813e-05,
      "step": 15330,
      "training_step_time": 8.735987663269043
    },
    {
      "epoch": 1.4620780944824218e-05,
      "model_forward_time": 1.9122982025146484,
      "step": 15331
    },
    {
      "epoch": 1.4620780944824218e-05,
      "step": 15331,
      "training_step_time": 8.404462099075317
    },
    {
      "epoch": 1.4621734619140626e-05,
      "model_forward_time": 3.041998863220215,
      "step": 15332
    },
    {
      "epoch": 1.4621734619140626e-05,
      "step": 15332,
      "training_step_time": 7.907840728759766
    },
    {
      "epoch": 1.4622688293457031e-05,
      "model_forward_time": 1.9268295764923096,
      "step": 15333
    },
    {
      "epoch": 1.4622688293457031e-05,
      "step": 15333,
      "training_step_time": 7.707913637161255
    },
    {
      "epoch": 1.4623641967773437e-05,
      "model_forward_time": 1.9357900619506836,
      "step": 15334
    },
    {
      "epoch": 1.4623641967773437e-05,
      "step": 15334,
      "training_step_time": 7.655363321304321
    },
    {
      "epoch": 1.4624595642089845e-05,
      "model_forward_time": 1.916271448135376,
      "step": 15335
    },
    {
      "epoch": 1.4624595642089845e-05,
      "step": 15335,
      "training_step_time": 7.430500030517578
    },
    {
      "epoch": 1.462554931640625e-05,
      "model_forward_time": 1.9272465705871582,
      "step": 15336
    },
    {
      "epoch": 1.462554931640625e-05,
      "step": 15336,
      "training_step_time": 7.777334928512573
    },
    {
      "epoch": 1.4626502990722656e-05,
      "model_forward_time": 1.894078016281128,
      "step": 15337
    },
    {
      "epoch": 1.4626502990722656e-05,
      "step": 15337,
      "training_step_time": 8.336468696594238
    },
    {
      "epoch": 1.4627456665039062e-05,
      "model_forward_time": 1.9262726306915283,
      "step": 15338
    },
    {
      "epoch": 1.4627456665039062e-05,
      "step": 15338,
      "training_step_time": 7.755838394165039
    },
    {
      "epoch": 1.4628410339355469e-05,
      "model_forward_time": 1.9431476593017578,
      "step": 15339
    },
    {
      "epoch": 1.4628410339355469e-05,
      "step": 15339,
      "training_step_time": 7.691147804260254
    },
    {
      "epoch": 1.4629364013671875e-05,
      "grad_norm": 0.15013544261455536,
      "learning_rate": 9.71058916862656e-06,
      "loss": 0.067,
      "step": 15340
    },
    {
      "epoch": 1.4629364013671875e-05,
      "model_forward_time": 1.8959484100341797,
      "step": 15340
    },
    {
      "dynamics_loss_avg": 0.0882404524832964,
      "epoch": 1.4629364013671875e-05,
      "step": 15340
    },
    {
      "action_loss_avg": 0.004379730043001473,
      "epoch": 1.4629364013671875e-05,
      "step": 15340
    },
    {
      "epoch": 1.4629364013671875e-05,
      "step": 15340,
      "training_step_time": 7.703964710235596
    },
    {
      "epoch": 1.463031768798828e-05,
      "model_forward_time": 3.0344903469085693,
      "step": 15341
    },
    {
      "epoch": 1.463031768798828e-05,
      "step": 15341,
      "training_step_time": 9.22922658920288
    },
    {
      "epoch": 1.4631271362304688e-05,
      "model_forward_time": 1.926208734512329,
      "step": 15342
    },
    {
      "epoch": 1.4631271362304688e-05,
      "step": 15342,
      "training_step_time": 8.436187267303467
    },
    {
      "epoch": 1.4632225036621094e-05,
      "model_forward_time": 1.9317505359649658,
      "step": 15343
    },
    {
      "epoch": 1.4632225036621094e-05,
      "step": 15343,
      "training_step_time": 7.5826027393341064
    },
    {
      "epoch": 1.46331787109375e-05,
      "model_forward_time": 1.919853925704956,
      "step": 15344
    },
    {
      "epoch": 1.46331787109375e-05,
      "step": 15344,
      "training_step_time": 7.629892587661743
    },
    {
      "epoch": 1.4634132385253907e-05,
      "model_forward_time": 1.9244296550750732,
      "step": 15345
    },
    {
      "epoch": 1.4634132385253907e-05,
      "step": 15345,
      "training_step_time": 7.725193738937378
    },
    {
      "epoch": 1.4635086059570313e-05,
      "model_forward_time": 1.9293813705444336,
      "step": 15346
    },
    {
      "epoch": 1.4635086059570313e-05,
      "step": 15346,
      "training_step_time": 8.296138525009155
    },
    {
      "epoch": 1.4636039733886718e-05,
      "model_forward_time": 1.919632911682129,
      "step": 15347
    },
    {
      "epoch": 1.4636039733886718e-05,
      "step": 15347,
      "training_step_time": 7.720250844955444
    },
    {
      "epoch": 1.4636993408203126e-05,
      "model_forward_time": 1.9221086502075195,
      "step": 15348
    },
    {
      "epoch": 1.4636993408203126e-05,
      "step": 15348,
      "training_step_time": 9.202495813369751
    },
    {
      "epoch": 1.4637947082519531e-05,
      "model_forward_time": 1.9030332565307617,
      "step": 15349
    },
    {
      "epoch": 1.4637947082519531e-05,
      "step": 15349,
      "training_step_time": 9.760854482650757
    },
    {
      "epoch": 1.4638900756835937e-05,
      "grad_norm": 0.15094482898712158,
      "learning_rate": 9.710034532477047e-06,
      "loss": 0.0623,
      "step": 15350
    },
    {
      "epoch": 1.4638900756835937e-05,
      "model_forward_time": 3.0148370265960693,
      "step": 15350
    },
    {
      "dynamics_loss_avg": 0.0593323515728116,
      "epoch": 1.4638900756835937e-05,
      "step": 15350
    },
    {
      "action_loss_avg": 0.0040529163205064835,
      "epoch": 1.4638900756835937e-05,
      "step": 15350
    },
    {
      "epoch": 1.4638900756835937e-05,
      "step": 15350,
      "training_step_time": 7.7820799350738525
    },
    {
      "epoch": 1.4639854431152345e-05,
      "model_forward_time": 1.927410364151001,
      "step": 15351
    },
    {
      "epoch": 1.4639854431152345e-05,
      "step": 15351,
      "training_step_time": 9.228415966033936
    },
    {
      "epoch": 1.464080810546875e-05,
      "model_forward_time": 1.8893811702728271,
      "step": 15352
    },
    {
      "epoch": 1.464080810546875e-05,
      "step": 15352,
      "training_step_time": 7.686081886291504
    },
    {
      "epoch": 1.4641761779785156e-05,
      "model_forward_time": 1.7692358493804932,
      "step": 15353
    },
    {
      "epoch": 1.4641761779785156e-05,
      "step": 15353,
      "training_step_time": 8.552204132080078
    },
    {
      "epoch": 1.4642715454101562e-05,
      "model_forward_time": 1.7681193351745605,
      "step": 15354
    },
    {
      "epoch": 1.4642715454101562e-05,
      "step": 15354,
      "training_step_time": 7.757564067840576
    },
    {
      "epoch": 1.464366912841797e-05,
      "model_forward_time": 1.7660722732543945,
      "step": 15355
    },
    {
      "epoch": 1.464366912841797e-05,
      "step": 15355,
      "training_step_time": 7.69397234916687
    },
    {
      "epoch": 1.4644622802734375e-05,
      "model_forward_time": 1.7671446800231934,
      "step": 15356
    },
    {
      "epoch": 1.4644622802734375e-05,
      "step": 15356,
      "training_step_time": 8.22860336303711
    },
    {
      "epoch": 1.464557647705078e-05,
      "model_forward_time": 1.7666187286376953,
      "step": 15357
    },
    {
      "epoch": 1.464557647705078e-05,
      "step": 15357,
      "training_step_time": 8.62609601020813
    },
    {
      "epoch": 1.4646530151367188e-05,
      "model_forward_time": 1.761754035949707,
      "step": 15358
    },
    {
      "epoch": 1.4646530151367188e-05,
      "step": 15358,
      "training_step_time": 7.686494588851929
    },
    {
      "epoch": 1.4647483825683594e-05,
      "model_forward_time": 2.893240451812744,
      "step": 15359
    },
    {
      "epoch": 1.4647483825683594e-05,
      "step": 15359,
      "training_step_time": 7.6631760597229
    },
    {
      "epoch": 1.46484375e-05,
      "grad_norm": 0.2338275909423828,
      "learning_rate": 9.709479381245239e-06,
      "loss": 0.0655,
      "step": 15360
    },
    {
      "epoch": 1.46484375e-05,
      "model_forward_time": 1.7670457363128662,
      "step": 15360
    },
    {
      "dynamics_loss_avg": 0.07092725783586502,
      "epoch": 1.46484375e-05,
      "step": 15360
    },
    {
      "action_loss_avg": 0.0030218847561627625,
      "epoch": 1.46484375e-05,
      "step": 15360
    },
    {
      "epoch": 1.46484375e-05,
      "step": 15360,
      "training_step_time": 8.66020917892456
    },
    {
      "epoch": 1.4649391174316407e-05,
      "model_forward_time": 1.7579572200775146,
      "step": 15361
    },
    {
      "epoch": 1.4649391174316407e-05,
      "step": 15361,
      "training_step_time": 7.636402130126953
    },
    {
      "epoch": 1.4650344848632813e-05,
      "model_forward_time": 1.7663171291351318,
      "step": 15362
    },
    {
      "epoch": 1.4650344848632813e-05,
      "step": 15362,
      "training_step_time": 7.579861879348755
    },
    {
      "epoch": 1.4651298522949218e-05,
      "model_forward_time": 1.770768642425537,
      "step": 15363
    },
    {
      "epoch": 1.4651298522949218e-05,
      "step": 15363,
      "training_step_time": 7.661575078964233
    },
    {
      "epoch": 1.4652252197265626e-05,
      "model_forward_time": 1.7657859325408936,
      "step": 15364
    },
    {
      "epoch": 1.4652252197265626e-05,
      "step": 15364,
      "training_step_time": 7.522676229476929
    },
    {
      "epoch": 1.4653205871582031e-05,
      "model_forward_time": 1.7659974098205566,
      "step": 15365
    },
    {
      "epoch": 1.4653205871582031e-05,
      "step": 15365,
      "training_step_time": 7.617183446884155
    },
    {
      "epoch": 1.4654159545898437e-05,
      "model_forward_time": 1.8448903560638428,
      "step": 15366
    },
    {
      "epoch": 1.4654159545898437e-05,
      "step": 15366,
      "training_step_time": 8.351113080978394
    },
    {
      "epoch": 1.4655113220214845e-05,
      "model_forward_time": 1.764664888381958,
      "step": 15367
    },
    {
      "epoch": 1.4655113220214845e-05,
      "step": 15367,
      "training_step_time": 9.070426940917969
    },
    {
      "epoch": 1.465606689453125e-05,
      "model_forward_time": 1.0755226612091064,
      "step": 15368
    },
    {
      "epoch": 1.465606689453125e-05,
      "step": 15368,
      "training_step_time": 8.254584312438965
    },
    {
      "epoch": 1.4657020568847656e-05,
      "model_forward_time": 2.8844590187072754,
      "step": 15369
    },
    {
      "epoch": 1.4657020568847656e-05,
      "step": 15369,
      "training_step_time": 7.6109089851379395
    },
    {
      "epoch": 1.4657974243164062e-05,
      "grad_norm": 0.17781521379947662,
      "learning_rate": 9.708923714991847e-06,
      "loss": 0.0671,
      "step": 15370
    },
    {
      "epoch": 1.4657974243164062e-05,
      "model_forward_time": 1.7650785446166992,
      "step": 15370
    },
    {
      "dynamics_loss_avg": 0.047310632281005384,
      "epoch": 1.4657974243164062e-05,
      "step": 15370
    },
    {
      "action_loss_avg": 0.0037350784754380585,
      "epoch": 1.4657974243164062e-05,
      "step": 15370
    },
    {
      "epoch": 1.4657974243164062e-05,
      "step": 15370,
      "training_step_time": 7.728006839752197
    },
    {
      "epoch": 1.465892791748047e-05,
      "model_forward_time": 1.7705318927764893,
      "step": 15371
    },
    {
      "epoch": 1.465892791748047e-05,
      "step": 15371,
      "training_step_time": 7.61530613899231
    },
    {
      "epoch": 1.4659881591796875e-05,
      "model_forward_time": 1.7688076496124268,
      "step": 15372
    },
    {
      "epoch": 1.4659881591796875e-05,
      "step": 15372,
      "training_step_time": 7.7385149002075195
    },
    {
      "epoch": 1.466083526611328e-05,
      "model_forward_time": 1.7757256031036377,
      "step": 15373
    },
    {
      "epoch": 1.466083526611328e-05,
      "step": 15373,
      "training_step_time": 7.548659563064575
    },
    {
      "epoch": 1.4661788940429688e-05,
      "model_forward_time": 1.7661454677581787,
      "step": 15374
    },
    {
      "epoch": 1.4661788940429688e-05,
      "step": 15374,
      "training_step_time": 7.648129940032959
    },
    {
      "epoch": 1.4662742614746094e-05,
      "model_forward_time": 1.7723779678344727,
      "step": 15375
    },
    {
      "epoch": 1.4662742614746094e-05,
      "step": 15375,
      "training_step_time": 9.381475687026978
    },
    {
      "epoch": 1.46636962890625e-05,
      "model_forward_time": 1.7709522247314453,
      "step": 15376
    },
    {
      "epoch": 1.46636962890625e-05,
      "step": 15376,
      "training_step_time": 7.678921222686768
    },
    {
      "epoch": 1.4664649963378907e-05,
      "model_forward_time": 1.7682089805603027,
      "step": 15377
    },
    {
      "epoch": 1.4664649963378907e-05,
      "step": 15377,
      "training_step_time": 10.51326584815979
    },
    {
      "epoch": 1.4665603637695313e-05,
      "model_forward_time": 2.914621353149414,
      "step": 15378
    },
    {
      "epoch": 1.4665603637695313e-05,
      "step": 15378,
      "training_step_time": 8.416007041931152
    },
    {
      "epoch": 1.4666557312011718e-05,
      "model_forward_time": 1.7589256763458252,
      "step": 15379
    },
    {
      "epoch": 1.4666557312011718e-05,
      "step": 15379,
      "training_step_time": 7.791776657104492
    },
    {
      "epoch": 1.4667510986328126e-05,
      "grad_norm": 0.1824580579996109,
      "learning_rate": 9.708367533777637e-06,
      "loss": 0.0707,
      "step": 15380
    },
    {
      "epoch": 1.4667510986328126e-05,
      "model_forward_time": 1.7550199031829834,
      "step": 15380
    },
    {
      "dynamics_loss_avg": 0.08037055470049381,
      "epoch": 1.4667510986328126e-05,
      "step": 15380
    },
    {
      "action_loss_avg": 0.003650710277725011,
      "epoch": 1.4667510986328126e-05,
      "step": 15380
    },
    {
      "epoch": 1.4667510986328126e-05,
      "step": 15380,
      "training_step_time": 7.637132406234741
    },
    {
      "epoch": 1.4668464660644531e-05,
      "model_forward_time": 1.76572585105896,
      "step": 15381
    },
    {
      "epoch": 1.4668464660644531e-05,
      "step": 15381,
      "training_step_time": 7.725491285324097
    },
    {
      "epoch": 1.4669418334960937e-05,
      "model_forward_time": 1.7511498928070068,
      "step": 15382
    },
    {
      "epoch": 1.4669418334960937e-05,
      "step": 15382,
      "training_step_time": 7.478494167327881
    },
    {
      "epoch": 1.4670372009277343e-05,
      "model_forward_time": 1.7497246265411377,
      "step": 15383
    },
    {
      "epoch": 1.4670372009277343e-05,
      "step": 15383,
      "training_step_time": 9.310691595077515
    },
    {
      "epoch": 1.467132568359375e-05,
      "model_forward_time": 1.7521417140960693,
      "step": 15384
    },
    {
      "epoch": 1.467132568359375e-05,
      "step": 15384,
      "training_step_time": 7.675624370574951
    },
    {
      "epoch": 1.4672279357910156e-05,
      "model_forward_time": 1.4008574485778809,
      "step": 15385
    },
    {
      "epoch": 1.4672279357910156e-05,
      "step": 15385,
      "training_step_time": 8.191187143325806
    },
    {
      "epoch": 1.4673233032226562e-05,
      "model_forward_time": 1.767634630203247,
      "step": 15386
    },
    {
      "epoch": 1.4673233032226562e-05,
      "step": 15386,
      "training_step_time": 8.676556587219238
    },
    {
      "epoch": 1.467418670654297e-05,
      "model_forward_time": 2.893038511276245,
      "step": 15387
    },
    {
      "epoch": 1.467418670654297e-05,
      "step": 15387,
      "training_step_time": 9.146756649017334
    },
    {
      "epoch": 1.4675140380859375e-05,
      "model_forward_time": 1.7700343132019043,
      "step": 15388
    },
    {
      "epoch": 1.4675140380859375e-05,
      "step": 15388,
      "training_step_time": 8.454911947250366
    },
    {
      "epoch": 1.467609405517578e-05,
      "model_forward_time": 1.7761168479919434,
      "step": 15389
    },
    {
      "epoch": 1.467609405517578e-05,
      "step": 15389,
      "training_step_time": 7.615596294403076
    },
    {
      "epoch": 1.4677047729492188e-05,
      "grad_norm": 0.19236204028129578,
      "learning_rate": 9.707810837663432e-06,
      "loss": 0.0674,
      "step": 15390
    },
    {
      "epoch": 1.4677047729492188e-05,
      "model_forward_time": 1.7603247165679932,
      "step": 15390
    },
    {
      "dynamics_loss_avg": 0.04891913644969463,
      "epoch": 1.4677047729492188e-05,
      "step": 15390
    },
    {
      "action_loss_avg": 0.0027293750550597905,
      "epoch": 1.4677047729492188e-05,
      "step": 15390
    },
    {
      "epoch": 1.4677047729492188e-05,
      "step": 15390,
      "training_step_time": 7.631878614425659
    },
    {
      "epoch": 1.4678001403808594e-05,
      "model_forward_time": 1.7541217803955078,
      "step": 15391
    },
    {
      "epoch": 1.4678001403808594e-05,
      "step": 15391,
      "training_step_time": 7.105777263641357
    },
    {
      "epoch": 1.4678955078125e-05,
      "model_forward_time": 1.7695269584655762,
      "step": 15392
    },
    {
      "epoch": 1.4678955078125e-05,
      "step": 15392,
      "training_step_time": 7.647499322891235
    },
    {
      "epoch": 1.4679908752441407e-05,
      "model_forward_time": 1.7682008743286133,
      "step": 15393
    },
    {
      "epoch": 1.4679908752441407e-05,
      "step": 15393,
      "training_step_time": 7.6497414112091064
    },
    {
      "epoch": 1.4680862426757813e-05,
      "model_forward_time": 1.759664535522461,
      "step": 15394
    },
    {
      "epoch": 1.4680862426757813e-05,
      "step": 15394,
      "training_step_time": 8.179705142974854
    },
    {
      "epoch": 1.4681816101074218e-05,
      "model_forward_time": 1.762627363204956,
      "step": 15395
    },
    {
      "epoch": 1.4681816101074218e-05,
      "step": 15395,
      "training_step_time": 8.9673171043396
    },
    {
      "epoch": 1.4682769775390626e-05,
      "model_forward_time": 1.3972694873809814,
      "step": 15396
    },
    {
      "epoch": 1.4682769775390626e-05,
      "step": 15396,
      "training_step_time": 8.493280172348022
    },
    {
      "epoch": 1.4683723449707031e-05,
      "model_forward_time": 2.906313180923462,
      "step": 15397
    },
    {
      "epoch": 1.4683723449707031e-05,
      "step": 15397,
      "training_step_time": 7.746818780899048
    },
    {
      "epoch": 1.4684677124023437e-05,
      "model_forward_time": 1.757448434829712,
      "step": 15398
    },
    {
      "epoch": 1.4684677124023437e-05,
      "step": 15398,
      "training_step_time": 7.491926431655884
    },
    {
      "epoch": 1.4685630798339843e-05,
      "model_forward_time": 1.7702622413635254,
      "step": 15399
    },
    {
      "epoch": 1.4685630798339843e-05,
      "step": 15399,
      "training_step_time": 7.588823556900024
    },
    {
      "epoch": 1.468658447265625e-05,
      "grad_norm": 0.11052180081605911,
      "learning_rate": 9.707253626710112e-06,
      "loss": 0.0687,
      "step": 15400
    },
    {
      "epoch": 1.468658447265625e-05,
      "model_forward_time": 1.7625527381896973,
      "step": 15400
    },
    {
      "dynamics_loss_avg": 0.05116649549454451,
      "epoch": 1.468658447265625e-05,
      "step": 15400
    },
    {
      "action_loss_avg": 0.003333978506270796,
      "epoch": 1.468658447265625e-05,
      "step": 15400
    },
    {
      "epoch": 1.468658447265625e-05,
      "step": 15400,
      "training_step_time": 7.076396703720093
    },
    {
      "epoch": 1.4687538146972656e-05,
      "model_forward_time": 1.7613332271575928,
      "step": 15401
    },
    {
      "epoch": 1.4687538146972656e-05,
      "step": 15401,
      "training_step_time": 8.297386407852173
    },
    {
      "epoch": 1.4688491821289062e-05,
      "model_forward_time": 1.765183687210083,
      "step": 15402
    },
    {
      "epoch": 1.4688491821289062e-05,
      "step": 15402,
      "training_step_time": 7.6479878425598145
    },
    {
      "epoch": 1.468944549560547e-05,
      "model_forward_time": 1.768179178237915,
      "step": 15403
    },
    {
      "epoch": 1.468944549560547e-05,
      "step": 15403,
      "training_step_time": 9.638489961624146
    },
    {
      "epoch": 1.4690399169921875e-05,
      "model_forward_time": 1.7716622352600098,
      "step": 15404
    },
    {
      "epoch": 1.4690399169921875e-05,
      "step": 15404,
      "training_step_time": 8.452409267425537
    },
    {
      "epoch": 1.469135284423828e-05,
      "model_forward_time": 1.7683701515197754,
      "step": 15405
    },
    {
      "epoch": 1.469135284423828e-05,
      "step": 15405,
      "training_step_time": 9.030472993850708
    },
    {
      "epoch": 1.4692306518554688e-05,
      "model_forward_time": 2.933229684829712,
      "step": 15406
    },
    {
      "epoch": 1.4692306518554688e-05,
      "step": 15406,
      "training_step_time": 7.678045988082886
    },
    {
      "epoch": 1.4693260192871094e-05,
      "model_forward_time": 1.7534480094909668,
      "step": 15407
    },
    {
      "epoch": 1.4693260192871094e-05,
      "step": 15407,
      "training_step_time": 7.678597450256348
    },
    {
      "epoch": 1.46942138671875e-05,
      "model_forward_time": 1.764709711074829,
      "step": 15408
    },
    {
      "epoch": 1.46942138671875e-05,
      "step": 15408,
      "training_step_time": 7.586990594863892
    },
    {
      "epoch": 1.4695167541503907e-05,
      "model_forward_time": 1.7598786354064941,
      "step": 15409
    },
    {
      "epoch": 1.4695167541503907e-05,
      "step": 15409,
      "training_step_time": 7.357700347900391
    },
    {
      "epoch": 1.4696121215820313e-05,
      "grad_norm": 0.18550479412078857,
      "learning_rate": 9.706695900978613e-06,
      "loss": 0.0681,
      "step": 15410
    },
    {
      "epoch": 1.4696121215820313e-05,
      "model_forward_time": 1.7613797187805176,
      "step": 15410
    },
    {
      "dynamics_loss_avg": 0.06137761417776346,
      "epoch": 1.4696121215820313e-05,
      "step": 15410
    },
    {
      "action_loss_avg": 0.0039001585100777447,
      "epoch": 1.4696121215820313e-05,
      "step": 15410
    },
    {
      "epoch": 1.4696121215820313e-05,
      "step": 15410,
      "training_step_time": 8.449925661087036
    },
    {
      "epoch": 1.4697074890136718e-05,
      "model_forward_time": 1.76353120803833,
      "step": 15411
    },
    {
      "epoch": 1.4697074890136718e-05,
      "step": 15411,
      "training_step_time": 7.579858779907227
    },
    {
      "epoch": 1.4698028564453126e-05,
      "model_forward_time": 1.7615749835968018,
      "step": 15412
    },
    {
      "epoch": 1.4698028564453126e-05,
      "step": 15412,
      "training_step_time": 7.616602420806885
    },
    {
      "epoch": 1.4698982238769531e-05,
      "model_forward_time": 1.7562839984893799,
      "step": 15413
    },
    {
      "epoch": 1.4698982238769531e-05,
      "step": 15413,
      "training_step_time": 8.471678733825684
    },
    {
      "epoch": 1.4699935913085937e-05,
      "model_forward_time": 1.7610013484954834,
      "step": 15414
    },
    {
      "epoch": 1.4699935913085937e-05,
      "step": 15414,
      "training_step_time": 7.704930305480957
    },
    {
      "epoch": 1.4700889587402343e-05,
      "model_forward_time": 2.9832379817962646,
      "step": 15415
    },
    {
      "epoch": 1.4700889587402343e-05,
      "step": 15415,
      "training_step_time": 8.2404043674469
    },
    {
      "epoch": 1.470184326171875e-05,
      "model_forward_time": 1.7892215251922607,
      "step": 15416
    },
    {
      "epoch": 1.470184326171875e-05,
      "step": 15416,
      "training_step_time": 9.373590469360352
    },
    {
      "epoch": 1.4702796936035156e-05,
      "model_forward_time": 1.762634038925171,
      "step": 15417
    },
    {
      "epoch": 1.4702796936035156e-05,
      "step": 15417,
      "training_step_time": 7.608509540557861
    },
    {
      "epoch": 1.4703750610351562e-05,
      "model_forward_time": 1.793177604675293,
      "step": 15418
    },
    {
      "epoch": 1.4703750610351562e-05,
      "step": 15418,
      "training_step_time": 7.338072061538696
    },
    {
      "epoch": 1.470470428466797e-05,
      "model_forward_time": 1.7676477432250977,
      "step": 15419
    },
    {
      "epoch": 1.470470428466797e-05,
      "step": 15419,
      "training_step_time": 7.5445685386657715
    },
    {
      "epoch": 1.4705657958984375e-05,
      "grad_norm": 0.2120824158191681,
      "learning_rate": 9.706137660529926e-06,
      "loss": 0.0672,
      "step": 15420
    },
    {
      "epoch": 1.4705657958984375e-05,
      "model_forward_time": 1.7577250003814697,
      "step": 15420
    },
    {
      "dynamics_loss_avg": 0.06200340297073126,
      "epoch": 1.4705657958984375e-05,
      "step": 15420
    },
    {
      "action_loss_avg": 0.0029885673895478247,
      "epoch": 1.4705657958984375e-05,
      "step": 15420
    },
    {
      "epoch": 1.4705657958984375e-05,
      "step": 15420,
      "training_step_time": 7.6538405418396
    },
    {
      "epoch": 1.470661163330078e-05,
      "model_forward_time": 1.7538104057312012,
      "step": 15421
    },
    {
      "epoch": 1.470661163330078e-05,
      "step": 15421,
      "training_step_time": 8.179135084152222
    },
    {
      "epoch": 1.4707565307617188e-05,
      "model_forward_time": 1.762587547302246,
      "step": 15422
    },
    {
      "epoch": 1.4707565307617188e-05,
      "step": 15422,
      "training_step_time": 8.23725175857544
    },
    {
      "epoch": 1.4708518981933594e-05,
      "model_forward_time": 1.757960557937622,
      "step": 15423
    },
    {
      "epoch": 1.4708518981933594e-05,
      "step": 15423,
      "training_step_time": 10.325364351272583
    },
    {
      "epoch": 1.470947265625e-05,
      "model_forward_time": 2.917214870452881,
      "step": 15424
    },
    {
      "epoch": 1.470947265625e-05,
      "step": 15424,
      "training_step_time": 8.722586870193481
    },
    {
      "epoch": 1.4710426330566407e-05,
      "model_forward_time": 1.760979175567627,
      "step": 15425
    },
    {
      "epoch": 1.4710426330566407e-05,
      "step": 15425,
      "training_step_time": 7.803602933883667
    },
    {
      "epoch": 1.4711380004882813e-05,
      "model_forward_time": 1.7612700462341309,
      "step": 15426
    },
    {
      "epoch": 1.4711380004882813e-05,
      "step": 15426,
      "training_step_time": 7.846823215484619
    },
    {
      "epoch": 1.4712333679199218e-05,
      "model_forward_time": 1.7575654983520508,
      "step": 15427
    },
    {
      "epoch": 1.4712333679199218e-05,
      "step": 15427,
      "training_step_time": 7.335351943969727
    },
    {
      "epoch": 1.4713287353515626e-05,
      "model_forward_time": 1.9062294960021973,
      "step": 15428
    },
    {
      "epoch": 1.4713287353515626e-05,
      "step": 15428,
      "training_step_time": 7.656571388244629
    },
    {
      "epoch": 1.4714241027832032e-05,
      "model_forward_time": 1.8896565437316895,
      "step": 15429
    },
    {
      "epoch": 1.4714241027832032e-05,
      "step": 15429,
      "training_step_time": 7.6660475730896
    },
    {
      "epoch": 1.4715194702148437e-05,
      "grad_norm": 0.19601093232631683,
      "learning_rate": 9.705578905425102e-06,
      "loss": 0.0723,
      "step": 15430
    },
    {
      "epoch": 1.4715194702148437e-05,
      "model_forward_time": 1.9324262142181396,
      "step": 15430
    },
    {
      "dynamics_loss_avg": 0.06306134536862373,
      "epoch": 1.4715194702148437e-05,
      "step": 15430
    },
    {
      "action_loss_avg": 0.003270677069667727,
      "epoch": 1.4715194702148437e-05,
      "step": 15430
    },
    {
      "epoch": 1.4715194702148437e-05,
      "step": 15430,
      "training_step_time": 7.752054452896118
    },
    {
      "epoch": 1.4716148376464843e-05,
      "model_forward_time": 1.9342591762542725,
      "step": 15431
    },
    {
      "epoch": 1.4716148376464843e-05,
      "step": 15431,
      "training_step_time": 8.282246351242065
    },
    {
      "epoch": 1.471710205078125e-05,
      "model_forward_time": 1.8903827667236328,
      "step": 15432
    },
    {
      "epoch": 1.471710205078125e-05,
      "step": 15432,
      "training_step_time": 7.810650825500488
    },
    {
      "epoch": 1.4718055725097656e-05,
      "model_forward_time": 1.8889641761779785,
      "step": 15433
    },
    {
      "epoch": 1.4718055725097656e-05,
      "step": 15433,
      "training_step_time": 8.719828605651855
    },
    {
      "epoch": 1.4719009399414062e-05,
      "model_forward_time": 2.999711275100708,
      "step": 15434
    },
    {
      "epoch": 1.4719009399414062e-05,
      "step": 15434,
      "training_step_time": 7.8066370487213135
    },
    {
      "epoch": 1.471996307373047e-05,
      "model_forward_time": 1.9320390224456787,
      "step": 15435
    },
    {
      "epoch": 1.471996307373047e-05,
      "step": 15435,
      "training_step_time": 7.732600450515747
    },
    {
      "epoch": 1.4720916748046875e-05,
      "model_forward_time": 1.9029650688171387,
      "step": 15436
    },
    {
      "epoch": 1.4720916748046875e-05,
      "step": 15436,
      "training_step_time": 7.684099435806274
    },
    {
      "epoch": 1.472187042236328e-05,
      "model_forward_time": 1.87760329246521,
      "step": 15437
    },
    {
      "epoch": 1.472187042236328e-05,
      "step": 15437,
      "training_step_time": 7.670801639556885
    },
    {
      "epoch": 1.4722824096679688e-05,
      "model_forward_time": 1.9369020462036133,
      "step": 15438
    },
    {
      "epoch": 1.4722824096679688e-05,
      "step": 15438,
      "training_step_time": 8.522994041442871
    },
    {
      "epoch": 1.4723777770996094e-05,
      "model_forward_time": 1.9128081798553467,
      "step": 15439
    },
    {
      "epoch": 1.4723777770996094e-05,
      "step": 15439,
      "training_step_time": 7.696322441101074
    },
    {
      "epoch": 1.47247314453125e-05,
      "grad_norm": 0.15615370869636536,
      "learning_rate": 9.705019635725242e-06,
      "loss": 0.0703,
      "step": 15440
    },
    {
      "epoch": 1.47247314453125e-05,
      "model_forward_time": 1.9543569087982178,
      "step": 15440
    },
    {
      "dynamics_loss_avg": 0.0694640887901187,
      "epoch": 1.47247314453125e-05,
      "step": 15440
    },
    {
      "action_loss_avg": 0.002973850799025968,
      "epoch": 1.47247314453125e-05,
      "step": 15440
    },
    {
      "epoch": 1.47247314453125e-05,
      "step": 15440,
      "training_step_time": 8.656789064407349
    },
    {
      "epoch": 1.4725685119628907e-05,
      "model_forward_time": 1.9233005046844482,
      "step": 15441
    },
    {
      "epoch": 1.4725685119628907e-05,
      "step": 15441,
      "training_step_time": 8.585160255432129
    },
    {
      "epoch": 1.4726638793945313e-05,
      "model_forward_time": 1.922132968902588,
      "step": 15442
    },
    {
      "epoch": 1.4726638793945313e-05,
      "step": 15442,
      "training_step_time": 9.155214786529541
    },
    {
      "epoch": 1.4727592468261718e-05,
      "model_forward_time": 3.0284292697906494,
      "step": 15443
    },
    {
      "epoch": 1.4727592468261718e-05,
      "step": 15443,
      "training_step_time": 7.842134714126587
    },
    {
      "epoch": 1.4728546142578126e-05,
      "model_forward_time": 1.8985588550567627,
      "step": 15444
    },
    {
      "epoch": 1.4728546142578126e-05,
      "step": 15444,
      "training_step_time": 8.507370471954346
    },
    {
      "epoch": 1.4729499816894532e-05,
      "model_forward_time": 1.9225564002990723,
      "step": 15445
    },
    {
      "epoch": 1.4729499816894532e-05,
      "step": 15445,
      "training_step_time": 7.624059677124023
    },
    {
      "epoch": 1.4730453491210937e-05,
      "model_forward_time": 1.937208890914917,
      "step": 15446
    },
    {
      "epoch": 1.4730453491210937e-05,
      "step": 15446,
      "training_step_time": 7.6096155643463135
    },
    {
      "epoch": 1.4731407165527343e-05,
      "model_forward_time": 1.9031598567962646,
      "step": 15447
    },
    {
      "epoch": 1.4731407165527343e-05,
      "step": 15447,
      "training_step_time": 7.4420037269592285
    },
    {
      "epoch": 1.473236083984375e-05,
      "model_forward_time": 1.9180409908294678,
      "step": 15448
    },
    {
      "epoch": 1.473236083984375e-05,
      "step": 15448,
      "training_step_time": 7.786122560501099
    },
    {
      "epoch": 1.4733314514160156e-05,
      "model_forward_time": 1.5645217895507812,
      "step": 15449
    },
    {
      "epoch": 1.4733314514160156e-05,
      "step": 15449,
      "training_step_time": 10.049871444702148
    },
    {
      "epoch": 1.4734268188476562e-05,
      "grad_norm": 0.12498180568218231,
      "learning_rate": 9.704459851491508e-06,
      "loss": 0.0677,
      "step": 15450
    },
    {
      "epoch": 1.4734268188476562e-05,
      "model_forward_time": 1.9067485332489014,
      "step": 15450
    },
    {
      "dynamics_loss_avg": 0.05269230008125305,
      "epoch": 1.4734268188476562e-05,
      "step": 15450
    },
    {
      "action_loss_avg": 0.002158539730589837,
      "epoch": 1.4734268188476562e-05,
      "step": 15450
    },
    {
      "epoch": 1.4734268188476562e-05,
      "step": 15450,
      "training_step_time": 8.378499031066895
    },
    {
      "epoch": 1.473522186279297e-05,
      "model_forward_time": 1.8927927017211914,
      "step": 15451
    },
    {
      "epoch": 1.473522186279297e-05,
      "step": 15451,
      "training_step_time": 8.479007482528687
    },
    {
      "epoch": 1.4736175537109375e-05,
      "model_forward_time": 3.0569772720336914,
      "step": 15452
    },
    {
      "epoch": 1.4736175537109375e-05,
      "step": 15452,
      "training_step_time": 7.861214876174927
    },
    {
      "epoch": 1.473712921142578e-05,
      "model_forward_time": 1.929891586303711,
      "step": 15453
    },
    {
      "epoch": 1.473712921142578e-05,
      "step": 15453,
      "training_step_time": 8.75887131690979
    },
    {
      "epoch": 1.4738082885742188e-05,
      "model_forward_time": 1.9208400249481201,
      "step": 15454
    },
    {
      "epoch": 1.4738082885742188e-05,
      "step": 15454,
      "training_step_time": 8.145126342773438
    },
    {
      "epoch": 1.4739036560058594e-05,
      "model_forward_time": 1.9097206592559814,
      "step": 15455
    },
    {
      "epoch": 1.4739036560058594e-05,
      "step": 15455,
      "training_step_time": 7.659731864929199
    },
    {
      "epoch": 1.4739990234375e-05,
      "model_forward_time": 1.9208033084869385,
      "step": 15456
    },
    {
      "epoch": 1.4739990234375e-05,
      "step": 15456,
      "training_step_time": 7.402239561080933
    },
    {
      "epoch": 1.4740943908691407e-05,
      "model_forward_time": 1.931255578994751,
      "step": 15457
    },
    {
      "epoch": 1.4740943908691407e-05,
      "step": 15457,
      "training_step_time": 7.652583599090576
    },
    {
      "epoch": 1.4741897583007813e-05,
      "model_forward_time": 1.9164257049560547,
      "step": 15458
    },
    {
      "epoch": 1.4741897583007813e-05,
      "step": 15458,
      "training_step_time": 8.302379846572876
    },
    {
      "epoch": 1.4742851257324218e-05,
      "model_forward_time": 1.9291763305664062,
      "step": 15459
    },
    {
      "epoch": 1.4742851257324218e-05,
      "step": 15459,
      "training_step_time": 8.203364372253418
    },
    {
      "epoch": 1.4743804931640626e-05,
      "grad_norm": 0.2750885784626007,
      "learning_rate": 9.703899552785119e-06,
      "loss": 0.0644,
      "step": 15460
    },
    {
      "epoch": 1.4743804931640626e-05,
      "model_forward_time": 1.894679069519043,
      "step": 15460
    },
    {
      "dynamics_loss_avg": 0.05320924520492554,
      "epoch": 1.4743804931640626e-05,
      "step": 15460
    },
    {
      "action_loss_avg": 0.0027711484231986105,
      "epoch": 1.4743804931640626e-05,
      "step": 15460
    },
    {
      "epoch": 1.4743804931640626e-05,
      "step": 15460,
      "training_step_time": 8.81530499458313
    },
    {
      "epoch": 1.4744758605957032e-05,
      "model_forward_time": 3.070547580718994,
      "step": 15461
    },
    {
      "epoch": 1.4744758605957032e-05,
      "step": 15461,
      "training_step_time": 8.800800800323486
    },
    {
      "epoch": 1.4745712280273437e-05,
      "model_forward_time": 1.905517816543579,
      "step": 15462
    },
    {
      "epoch": 1.4745712280273437e-05,
      "step": 15462,
      "training_step_time": 8.569527864456177
    },
    {
      "epoch": 1.4746665954589843e-05,
      "model_forward_time": 1.9349110126495361,
      "step": 15463
    },
    {
      "epoch": 1.4746665954589843e-05,
      "step": 15463,
      "training_step_time": 7.581094026565552
    },
    {
      "epoch": 1.474761962890625e-05,
      "model_forward_time": 1.9292733669281006,
      "step": 15464
    },
    {
      "epoch": 1.474761962890625e-05,
      "step": 15464,
      "training_step_time": 7.6089911460876465
    },
    {
      "epoch": 1.4748573303222656e-05,
      "model_forward_time": 1.9027092456817627,
      "step": 15465
    },
    {
      "epoch": 1.4748573303222656e-05,
      "step": 15465,
      "training_step_time": 7.404780864715576
    },
    {
      "epoch": 1.4749526977539062e-05,
      "model_forward_time": 1.952951192855835,
      "step": 15466
    },
    {
      "epoch": 1.4749526977539062e-05,
      "step": 15466,
      "training_step_time": 7.646048069000244
    },
    {
      "epoch": 1.475048065185547e-05,
      "model_forward_time": 1.9321026802062988,
      "step": 15467
    },
    {
      "epoch": 1.475048065185547e-05,
      "step": 15467,
      "training_step_time": 7.596642255783081
    },
    {
      "epoch": 1.4751434326171875e-05,
      "model_forward_time": 1.9236583709716797,
      "step": 15468
    },
    {
      "epoch": 1.4751434326171875e-05,
      "step": 15468,
      "training_step_time": 7.707457065582275
    },
    {
      "epoch": 1.475238800048828e-05,
      "model_forward_time": 1.934823989868164,
      "step": 15469
    },
    {
      "epoch": 1.475238800048828e-05,
      "step": 15469,
      "training_step_time": 8.315103769302368
    },
    {
      "epoch": 1.4753341674804688e-05,
      "grad_norm": 0.20258992910385132,
      "learning_rate": 9.703338739667347e-06,
      "loss": 0.0616,
      "step": 15470
    },
    {
      "epoch": 1.4753341674804688e-05,
      "model_forward_time": 1.9146504402160645,
      "step": 15470
    },
    {
      "dynamics_loss_avg": 0.0717851672321558,
      "epoch": 1.4753341674804688e-05,
      "step": 15470
    },
    {
      "action_loss_avg": 0.004332722746767104,
      "epoch": 1.4753341674804688e-05,
      "step": 15470
    },
    {
      "epoch": 1.4753341674804688e-05,
      "step": 15470,
      "training_step_time": 9.837545394897461
    },
    {
      "epoch": 1.4754295349121094e-05,
      "model_forward_time": 1.9252097606658936,
      "step": 15471
    },
    {
      "epoch": 1.4754295349121094e-05,
      "step": 15471,
      "training_step_time": 7.565847158432007
    },
    {
      "epoch": 1.47552490234375e-05,
      "model_forward_time": 1.9187681674957275,
      "step": 15472
    },
    {
      "epoch": 1.47552490234375e-05,
      "step": 15472,
      "training_step_time": 7.663204193115234
    },
    {
      "epoch": 1.4756202697753907e-05,
      "model_forward_time": 1.9347569942474365,
      "step": 15473
    },
    {
      "epoch": 1.4756202697753907e-05,
      "step": 15473,
      "training_step_time": 7.540854454040527
    },
    {
      "epoch": 1.4757156372070313e-05,
      "model_forward_time": 1.9173943996429443,
      "step": 15474
    },
    {
      "epoch": 1.4757156372070313e-05,
      "step": 15474,
      "training_step_time": 7.3622705936431885
    },
    {
      "epoch": 1.4758110046386718e-05,
      "model_forward_time": 1.8786792755126953,
      "step": 15475
    },
    {
      "epoch": 1.4758110046386718e-05,
      "step": 15475,
      "training_step_time": 7.542346954345703
    },
    {
      "epoch": 1.4759063720703126e-05,
      "model_forward_time": 1.927800178527832,
      "step": 15476
    },
    {
      "epoch": 1.4759063720703126e-05,
      "step": 15476,
      "training_step_time": 7.695852279663086
    },
    {
      "epoch": 1.4760017395019532e-05,
      "model_forward_time": 1.9482364654541016,
      "step": 15477
    },
    {
      "epoch": 1.4760017395019532e-05,
      "step": 15477,
      "training_step_time": 7.681381464004517
    },
    {
      "epoch": 1.4760971069335937e-05,
      "model_forward_time": 1.9263336658477783,
      "step": 15478
    },
    {
      "epoch": 1.4760971069335937e-05,
      "step": 15478,
      "training_step_time": 8.483772277832031
    },
    {
      "epoch": 1.4761924743652343e-05,
      "model_forward_time": 1.9091293811798096,
      "step": 15479
    },
    {
      "epoch": 1.4761924743652343e-05,
      "step": 15479,
      "training_step_time": 7.5804688930511475
    },
    {
      "epoch": 1.476287841796875e-05,
      "grad_norm": 0.14781443774700165,
      "learning_rate": 9.70277741219952e-06,
      "loss": 0.0632,
      "step": 15480
    },
    {
      "epoch": 1.476287841796875e-05,
      "model_forward_time": 3.079836368560791,
      "step": 15480
    },
    {
      "dynamics_loss_avg": 0.04988703466951847,
      "epoch": 1.476287841796875e-05,
      "step": 15480
    },
    {
      "action_loss_avg": 0.0029509769985452293,
      "epoch": 1.476287841796875e-05,
      "step": 15480
    },
    {
      "epoch": 1.476287841796875e-05,
      "step": 15480,
      "training_step_time": 7.894350528717041
    },
    {
      "epoch": 1.4763832092285156e-05,
      "model_forward_time": 1.9187893867492676,
      "step": 15481
    },
    {
      "epoch": 1.4763832092285156e-05,
      "step": 15481,
      "training_step_time": 8.774465322494507
    },
    {
      "epoch": 1.4764785766601562e-05,
      "model_forward_time": 1.9187242984771729,
      "step": 15482
    },
    {
      "epoch": 1.4764785766601562e-05,
      "step": 15482,
      "training_step_time": 8.60259222984314
    },
    {
      "epoch": 1.476573944091797e-05,
      "model_forward_time": 1.9154586791992188,
      "step": 15483
    },
    {
      "epoch": 1.476573944091797e-05,
      "step": 15483,
      "training_step_time": 8.421199798583984
    },
    {
      "epoch": 1.4766693115234375e-05,
      "model_forward_time": 1.9133338928222656,
      "step": 15484
    },
    {
      "epoch": 1.4766693115234375e-05,
      "step": 15484,
      "training_step_time": 7.628268241882324
    },
    {
      "epoch": 1.476764678955078e-05,
      "model_forward_time": 1.9409408569335938,
      "step": 15485
    },
    {
      "epoch": 1.476764678955078e-05,
      "step": 15485,
      "training_step_time": 7.716860771179199
    },
    {
      "epoch": 1.4768600463867188e-05,
      "model_forward_time": 1.9191210269927979,
      "step": 15486
    },
    {
      "epoch": 1.4768600463867188e-05,
      "step": 15486,
      "training_step_time": 8.529537439346313
    },
    {
      "epoch": 1.4769554138183594e-05,
      "model_forward_time": 1.9287519454956055,
      "step": 15487
    },
    {
      "epoch": 1.4769554138183594e-05,
      "step": 15487,
      "training_step_time": 8.491425275802612
    },
    {
      "epoch": 1.47705078125e-05,
      "model_forward_time": 1.9401030540466309,
      "step": 15488
    },
    {
      "epoch": 1.47705078125e-05,
      "step": 15488,
      "training_step_time": 8.195136547088623
    },
    {
      "epoch": 1.4771461486816407e-05,
      "model_forward_time": 3.0426135063171387,
      "step": 15489
    },
    {
      "epoch": 1.4771461486816407e-05,
      "step": 15489,
      "training_step_time": 7.797789812088013
    },
    {
      "epoch": 1.4772415161132813e-05,
      "grad_norm": 0.20965862274169922,
      "learning_rate": 9.702215570443028e-06,
      "loss": 0.0666,
      "step": 15490
    },
    {
      "epoch": 1.4772415161132813e-05,
      "model_forward_time": 1.9135239124298096,
      "step": 15490
    },
    {
      "dynamics_loss_avg": 0.056163839250803,
      "epoch": 1.4772415161132813e-05,
      "step": 15490
    },
    {
      "action_loss_avg": 0.0029447440523654223,
      "epoch": 1.4772415161132813e-05,
      "step": 15490
    },
    {
      "epoch": 1.4772415161132813e-05,
      "step": 15490,
      "training_step_time": 7.659142017364502
    },
    {
      "epoch": 1.4773368835449218e-05,
      "model_forward_time": 1.9284539222717285,
      "step": 15491
    },
    {
      "epoch": 1.4773368835449218e-05,
      "step": 15491,
      "training_step_time": 7.640874862670898
    },
    {
      "epoch": 1.4774322509765626e-05,
      "model_forward_time": 1.9117977619171143,
      "step": 15492
    },
    {
      "epoch": 1.4774322509765626e-05,
      "step": 15492,
      "training_step_time": 7.56714940071106
    },
    {
      "epoch": 1.4775276184082032e-05,
      "model_forward_time": 1.9423956871032715,
      "step": 15493
    },
    {
      "epoch": 1.4775276184082032e-05,
      "step": 15493,
      "training_step_time": 8.430910348892212
    },
    {
      "epoch": 1.4776229858398437e-05,
      "model_forward_time": 1.907623291015625,
      "step": 15494
    },
    {
      "epoch": 1.4776229858398437e-05,
      "step": 15494,
      "training_step_time": 9.233890533447266
    },
    {
      "epoch": 1.4777183532714843e-05,
      "model_forward_time": 1.5755815505981445,
      "step": 15495
    },
    {
      "epoch": 1.4777183532714843e-05,
      "step": 15495,
      "training_step_time": 9.378325462341309
    },
    {
      "epoch": 1.477813720703125e-05,
      "model_forward_time": 1.9229185581207275,
      "step": 15496
    },
    {
      "epoch": 1.477813720703125e-05,
      "step": 15496,
      "training_step_time": 8.593422412872314
    },
    {
      "epoch": 1.4779090881347656e-05,
      "model_forward_time": 1.5547449588775635,
      "step": 15497
    },
    {
      "epoch": 1.4779090881347656e-05,
      "step": 15497,
      "training_step_time": 8.566380500793457
    },
    {
      "epoch": 1.4780044555664062e-05,
      "model_forward_time": 3.0158209800720215,
      "step": 15498
    },
    {
      "epoch": 1.4780044555664062e-05,
      "step": 15498,
      "training_step_time": 7.801958322525024
    },
    {
      "epoch": 1.478099822998047e-05,
      "model_forward_time": 1.889779806137085,
      "step": 15499
    },
    {
      "epoch": 1.478099822998047e-05,
      "step": 15499,
      "training_step_time": 7.6179163455963135
    },
    {
      "epoch": 1.4781951904296875e-05,
      "grad_norm": 0.20947307348251343,
      "learning_rate": 9.701653214459308e-06,
      "loss": 0.0657,
      "step": 15500
    },
    {
      "epoch": 1.4781951904296875e-05,
      "model_forward_time": 1.9391634464263916,
      "step": 15500
    },
    {
      "dynamics_loss_avg": 0.049795282632112504,
      "epoch": 1.4781951904296875e-05,
      "step": 15500
    },
    {
      "action_loss_avg": 0.002428213629173115,
      "epoch": 1.4781951904296875e-05,
      "step": 15500
    },
    {
      "epoch": 1.4781951904296875e-05,
      "step": 15500,
      "training_step_time": 7.7897772789001465
    },
    {
      "epoch": 1.478290557861328e-05,
      "model_forward_time": 1.9387502670288086,
      "step": 15501
    },
    {
      "epoch": 1.478290557861328e-05,
      "step": 15501,
      "training_step_time": 7.627483367919922
    },
    {
      "epoch": 1.4783859252929688e-05,
      "model_forward_time": 1.9011425971984863,
      "step": 15502
    },
    {
      "epoch": 1.4783859252929688e-05,
      "step": 15502,
      "training_step_time": 7.66330099105835
    },
    {
      "epoch": 1.4784812927246094e-05,
      "model_forward_time": 1.930870771408081,
      "step": 15503
    },
    {
      "epoch": 1.4784812927246094e-05,
      "step": 15503,
      "training_step_time": 7.789461851119995
    },
    {
      "epoch": 1.47857666015625e-05,
      "model_forward_time": 1.9492270946502686,
      "step": 15504
    },
    {
      "epoch": 1.47857666015625e-05,
      "step": 15504,
      "training_step_time": 7.670875072479248
    },
    {
      "epoch": 1.4786720275878907e-05,
      "model_forward_time": 1.916006326675415,
      "step": 15505
    },
    {
      "epoch": 1.4786720275878907e-05,
      "step": 15505,
      "training_step_time": 7.929309844970703
    },
    {
      "epoch": 1.4787673950195313e-05,
      "model_forward_time": 1.9197156429290771,
      "step": 15506
    },
    {
      "epoch": 1.4787673950195313e-05,
      "step": 15506,
      "training_step_time": 8.283192157745361
    },
    {
      "epoch": 1.4788627624511718e-05,
      "model_forward_time": 1.9055273532867432,
      "step": 15507
    },
    {
      "epoch": 1.4788627624511718e-05,
      "step": 15507,
      "training_step_time": 10.460787296295166
    },
    {
      "epoch": 1.4789581298828124e-05,
      "model_forward_time": 1.9258010387420654,
      "step": 15508
    },
    {
      "epoch": 1.4789581298828124e-05,
      "step": 15508,
      "training_step_time": 8.225601196289062
    },
    {
      "epoch": 1.4790534973144532e-05,
      "model_forward_time": 1.9440586566925049,
      "step": 15509
    },
    {
      "epoch": 1.4790534973144532e-05,
      "step": 15509,
      "training_step_time": 7.752338886260986
    },
    {
      "epoch": 1.4791488647460937e-05,
      "grad_norm": 0.178152397274971,
      "learning_rate": 9.701090344309864e-06,
      "loss": 0.0628,
      "step": 15510
    },
    {
      "epoch": 1.4791488647460937e-05,
      "model_forward_time": 1.9125511646270752,
      "step": 15510
    },
    {
      "dynamics_loss_avg": 0.05230008102953434,
      "epoch": 1.4791488647460937e-05,
      "step": 15510
    },
    {
      "action_loss_avg": 0.0036387250467669217,
      "epoch": 1.4791488647460937e-05,
      "step": 15510
    },
    {
      "epoch": 1.4791488647460937e-05,
      "step": 15510,
      "training_step_time": 7.680013656616211
    },
    {
      "epoch": 1.4792442321777343e-05,
      "model_forward_time": 1.9341034889221191,
      "step": 15511
    },
    {
      "epoch": 1.4792442321777343e-05,
      "step": 15511,
      "training_step_time": 8.464746475219727
    },
    {
      "epoch": 1.479339599609375e-05,
      "model_forward_time": 1.9405808448791504,
      "step": 15512
    },
    {
      "epoch": 1.479339599609375e-05,
      "step": 15512,
      "training_step_time": 8.60072660446167
    },
    {
      "epoch": 1.4794349670410156e-05,
      "model_forward_time": 1.5817203521728516,
      "step": 15513
    },
    {
      "epoch": 1.4794349670410156e-05,
      "step": 15513,
      "training_step_time": 8.894911527633667
    },
    {
      "epoch": 1.4795303344726562e-05,
      "model_forward_time": 1.9234418869018555,
      "step": 15514
    },
    {
      "epoch": 1.4795303344726562e-05,
      "step": 15514,
      "training_step_time": 8.444202423095703
    },
    {
      "epoch": 1.479625701904297e-05,
      "model_forward_time": 1.932983636856079,
      "step": 15515
    },
    {
      "epoch": 1.479625701904297e-05,
      "step": 15515,
      "training_step_time": 7.71683931350708
    },
    {
      "epoch": 1.4797210693359375e-05,
      "model_forward_time": 1.8868446350097656,
      "step": 15516
    },
    {
      "epoch": 1.4797210693359375e-05,
      "step": 15516,
      "training_step_time": 7.652979612350464
    },
    {
      "epoch": 1.479816436767578e-05,
      "model_forward_time": 3.0478675365448,
      "step": 15517
    },
    {
      "epoch": 1.479816436767578e-05,
      "step": 15517,
      "training_step_time": 7.863966703414917
    },
    {
      "epoch": 1.4799118041992188e-05,
      "model_forward_time": 1.932758092880249,
      "step": 15518
    },
    {
      "epoch": 1.4799118041992188e-05,
      "step": 15518,
      "training_step_time": 8.602238178253174
    },
    {
      "epoch": 1.4800071716308594e-05,
      "model_forward_time": 1.9212915897369385,
      "step": 15519
    },
    {
      "epoch": 1.4800071716308594e-05,
      "step": 15519,
      "training_step_time": 7.6920976638793945
    },
    {
      "epoch": 1.4801025390625e-05,
      "grad_norm": 0.19516031444072723,
      "learning_rate": 9.700526960056247e-06,
      "loss": 0.0667,
      "step": 15520
    },
    {
      "epoch": 1.4801025390625e-05,
      "model_forward_time": 1.946258783340454,
      "step": 15520
    },
    {
      "dynamics_loss_avg": 0.06274668741971254,
      "epoch": 1.4801025390625e-05,
      "step": 15520
    },
    {
      "action_loss_avg": 0.0032787704141810535,
      "epoch": 1.4801025390625e-05,
      "step": 15520
    },
    {
      "epoch": 1.4801025390625e-05,
      "step": 15520,
      "training_step_time": 7.423641681671143
    },
    {
      "epoch": 1.4801979064941407e-05,
      "model_forward_time": 1.9295098781585693,
      "step": 15521
    },
    {
      "epoch": 1.4801979064941407e-05,
      "step": 15521,
      "training_step_time": 7.589703798294067
    },
    {
      "epoch": 1.4802932739257813e-05,
      "model_forward_time": 1.9216015338897705,
      "step": 15522
    },
    {
      "epoch": 1.4802932739257813e-05,
      "step": 15522,
      "training_step_time": 8.691872835159302
    },
    {
      "epoch": 1.4803886413574218e-05,
      "model_forward_time": 1.9371466636657715,
      "step": 15523
    },
    {
      "epoch": 1.4803886413574218e-05,
      "step": 15523,
      "training_step_time": 8.824183464050293
    },
    {
      "epoch": 1.4804840087890624e-05,
      "model_forward_time": 1.9239580631256104,
      "step": 15524
    },
    {
      "epoch": 1.4804840087890624e-05,
      "step": 15524,
      "training_step_time": 8.623016357421875
    },
    {
      "epoch": 1.4805793762207032e-05,
      "model_forward_time": 1.9217772483825684,
      "step": 15525
    },
    {
      "epoch": 1.4805793762207032e-05,
      "step": 15525,
      "training_step_time": 7.748138189315796
    },
    {
      "epoch": 1.4806747436523437e-05,
      "model_forward_time": 3.031827926635742,
      "step": 15526
    },
    {
      "epoch": 1.4806747436523437e-05,
      "step": 15526,
      "training_step_time": 7.792770862579346
    },
    {
      "epoch": 1.4807701110839843e-05,
      "model_forward_time": 1.9419023990631104,
      "step": 15527
    },
    {
      "epoch": 1.4807701110839843e-05,
      "step": 15527,
      "training_step_time": 7.751762628555298
    },
    {
      "epoch": 1.480865478515625e-05,
      "model_forward_time": 1.5840444564819336,
      "step": 15528
    },
    {
      "epoch": 1.480865478515625e-05,
      "step": 15528,
      "training_step_time": 7.682307958602905
    },
    {
      "epoch": 1.4809608459472656e-05,
      "model_forward_time": 1.9306411743164062,
      "step": 15529
    },
    {
      "epoch": 1.4809608459472656e-05,
      "step": 15529,
      "training_step_time": 7.439547538757324
    },
    {
      "epoch": 1.4810562133789062e-05,
      "grad_norm": 0.26247406005859375,
      "learning_rate": 9.69996306176007e-06,
      "loss": 0.0675,
      "step": 15530
    },
    {
      "epoch": 1.4810562133789062e-05,
      "model_forward_time": 1.941232681274414,
      "step": 15530
    },
    {
      "dynamics_loss_avg": 0.05343758147209883,
      "epoch": 1.4810562133789062e-05,
      "step": 15530
    },
    {
      "action_loss_avg": 0.003143403370631859,
      "epoch": 1.4810562133789062e-05,
      "step": 15530
    },
    {
      "epoch": 1.4810562133789062e-05,
      "step": 15530,
      "training_step_time": 8.66065764427185
    },
    {
      "epoch": 1.481151580810547e-05,
      "model_forward_time": 1.9373915195465088,
      "step": 15531
    },
    {
      "epoch": 1.481151580810547e-05,
      "step": 15531,
      "training_step_time": 8.37834644317627
    },
    {
      "epoch": 1.4812469482421875e-05,
      "model_forward_time": 1.9441473484039307,
      "step": 15532
    },
    {
      "epoch": 1.4812469482421875e-05,
      "step": 15532,
      "training_step_time": 8.432756423950195
    },
    {
      "epoch": 1.481342315673828e-05,
      "model_forward_time": 1.90775465965271,
      "step": 15533
    },
    {
      "epoch": 1.481342315673828e-05,
      "step": 15533,
      "training_step_time": 9.223456144332886
    },
    {
      "epoch": 1.4814376831054688e-05,
      "model_forward_time": 1.9226620197296143,
      "step": 15534
    },
    {
      "epoch": 1.4814376831054688e-05,
      "step": 15534,
      "training_step_time": 8.740131616592407
    },
    {
      "epoch": 1.4815330505371094e-05,
      "model_forward_time": 3.028208017349243,
      "step": 15535
    },
    {
      "epoch": 1.4815330505371094e-05,
      "step": 15535,
      "training_step_time": 8.85132908821106
    },
    {
      "epoch": 1.48162841796875e-05,
      "model_forward_time": 1.9327952861785889,
      "step": 15536
    },
    {
      "epoch": 1.48162841796875e-05,
      "step": 15536,
      "training_step_time": 8.735323667526245
    },
    {
      "epoch": 1.4817237854003907e-05,
      "model_forward_time": 1.9180858135223389,
      "step": 15537
    },
    {
      "epoch": 1.4817237854003907e-05,
      "step": 15537,
      "training_step_time": 7.682914972305298
    },
    {
      "epoch": 1.4818191528320313e-05,
      "model_forward_time": 1.9192464351654053,
      "step": 15538
    },
    {
      "epoch": 1.4818191528320313e-05,
      "step": 15538,
      "training_step_time": 7.4748454093933105
    },
    {
      "epoch": 1.4819145202636718e-05,
      "model_forward_time": 1.9198927879333496,
      "step": 15539
    },
    {
      "epoch": 1.4819145202636718e-05,
      "step": 15539,
      "training_step_time": 8.680824279785156
    },
    {
      "epoch": 1.4820098876953124e-05,
      "grad_norm": 0.2465505599975586,
      "learning_rate": 9.699398649482997e-06,
      "loss": 0.0652,
      "step": 15540
    },
    {
      "epoch": 1.4820098876953124e-05,
      "model_forward_time": 1.9141030311584473,
      "step": 15540
    },
    {
      "dynamics_loss_avg": 0.060173085890710354,
      "epoch": 1.4820098876953124e-05,
      "step": 15540
    },
    {
      "action_loss_avg": 0.0029943151981569826,
      "epoch": 1.4820098876953124e-05,
      "step": 15540
    },
    {
      "epoch": 1.4820098876953124e-05,
      "step": 15540,
      "training_step_time": 9.209099769592285
    },
    {
      "epoch": 1.4821052551269532e-05,
      "model_forward_time": 1.9224708080291748,
      "step": 15541
    },
    {
      "epoch": 1.4821052551269532e-05,
      "step": 15541,
      "training_step_time": 7.703878164291382
    },
    {
      "epoch": 1.4822006225585937e-05,
      "model_forward_time": 1.9197216033935547,
      "step": 15542
    },
    {
      "epoch": 1.4822006225585937e-05,
      "step": 15542,
      "training_step_time": 9.329567909240723
    },
    {
      "epoch": 1.4822959899902343e-05,
      "model_forward_time": 1.933279275894165,
      "step": 15543
    },
    {
      "epoch": 1.4822959899902343e-05,
      "step": 15543,
      "training_step_time": 8.316479444503784
    },
    {
      "epoch": 1.482391357421875e-05,
      "model_forward_time": 1.9226868152618408,
      "step": 15544
    },
    {
      "epoch": 1.482391357421875e-05,
      "step": 15544,
      "training_step_time": 8.411090612411499
    },
    {
      "epoch": 1.4824867248535156e-05,
      "model_forward_time": 2.9376115798950195,
      "step": 15545
    },
    {
      "epoch": 1.4824867248535156e-05,
      "step": 15545,
      "training_step_time": 7.748239278793335
    },
    {
      "epoch": 1.4825820922851562e-05,
      "model_forward_time": 1.9374604225158691,
      "step": 15546
    },
    {
      "epoch": 1.4825820922851562e-05,
      "step": 15546,
      "training_step_time": 7.747218608856201
    },
    {
      "epoch": 1.482677459716797e-05,
      "model_forward_time": 1.897939920425415,
      "step": 15547
    },
    {
      "epoch": 1.482677459716797e-05,
      "step": 15547,
      "training_step_time": 7.498666048049927
    },
    {
      "epoch": 1.4827728271484375e-05,
      "model_forward_time": 1.9295687675476074,
      "step": 15548
    },
    {
      "epoch": 1.4827728271484375e-05,
      "step": 15548,
      "training_step_time": 7.703478097915649
    },
    {
      "epoch": 1.482868194580078e-05,
      "model_forward_time": 1.927722692489624,
      "step": 15549
    },
    {
      "epoch": 1.482868194580078e-05,
      "step": 15549,
      "training_step_time": 8.760907173156738
    },
    {
      "epoch": 1.4829635620117188e-05,
      "grad_norm": 0.2865636944770813,
      "learning_rate": 9.698833723286754e-06,
      "loss": 0.0665,
      "step": 15550
    },
    {
      "epoch": 1.4829635620117188e-05,
      "model_forward_time": 1.90389084815979,
      "step": 15550
    },
    {
      "dynamics_loss_avg": 0.05054690465331078,
      "epoch": 1.4829635620117188e-05,
      "step": 15550
    },
    {
      "action_loss_avg": 0.002860327078087721,
      "epoch": 1.4829635620117188e-05,
      "step": 15550
    },
    {
      "epoch": 1.4829635620117188e-05,
      "step": 15550,
      "training_step_time": 8.41237735748291
    },
    {
      "epoch": 1.4830589294433594e-05,
      "model_forward_time": 1.9358336925506592,
      "step": 15551
    },
    {
      "epoch": 1.4830589294433594e-05,
      "step": 15551,
      "training_step_time": 8.709395408630371
    },
    {
      "epoch": 1.483154296875e-05,
      "model_forward_time": 1.921724796295166,
      "step": 15552
    },
    {
      "epoch": 1.483154296875e-05,
      "step": 15552,
      "training_step_time": 7.704318523406982
    },
    {
      "epoch": 1.4832496643066407e-05,
      "model_forward_time": 1.858933925628662,
      "step": 15553
    },
    {
      "epoch": 1.4832496643066407e-05,
      "step": 15553,
      "training_step_time": 8.634751796722412
    },
    {
      "epoch": 1.4833450317382813e-05,
      "model_forward_time": 2.921097993850708,
      "step": 15554
    },
    {
      "epoch": 1.4833450317382813e-05,
      "step": 15554,
      "training_step_time": 7.69512939453125
    },
    {
      "epoch": 1.4834403991699218e-05,
      "model_forward_time": 1.7516121864318848,
      "step": 15555
    },
    {
      "epoch": 1.4834403991699218e-05,
      "step": 15555,
      "training_step_time": 8.421526432037354
    },
    {
      "epoch": 1.4835357666015624e-05,
      "model_forward_time": 1.7523183822631836,
      "step": 15556
    },
    {
      "epoch": 1.4835357666015624e-05,
      "step": 15556,
      "training_step_time": 7.474354267120361
    },
    {
      "epoch": 1.4836311340332032e-05,
      "model_forward_time": 1.7636528015136719,
      "step": 15557
    },
    {
      "epoch": 1.4836311340332032e-05,
      "step": 15557,
      "training_step_time": 7.528818368911743
    },
    {
      "epoch": 1.4837265014648437e-05,
      "model_forward_time": 1.755732536315918,
      "step": 15558
    },
    {
      "epoch": 1.4837265014648437e-05,
      "step": 15558,
      "training_step_time": 8.186429738998413
    },
    {
      "epoch": 1.4838218688964843e-05,
      "model_forward_time": 1.7549517154693604,
      "step": 15559
    },
    {
      "epoch": 1.4838218688964843e-05,
      "step": 15559,
      "training_step_time": 8.407563924789429
    },
    {
      "epoch": 1.483917236328125e-05,
      "grad_norm": 0.30894961953163147,
      "learning_rate": 9.698268283233119e-06,
      "loss": 0.0662,
      "step": 15560
    },
    {
      "epoch": 1.483917236328125e-05,
      "model_forward_time": 1.799821138381958,
      "step": 15560
    },
    {
      "dynamics_loss_avg": 0.05519318636506796,
      "epoch": 1.483917236328125e-05,
      "step": 15560
    },
    {
      "action_loss_avg": 0.002986969507765025,
      "epoch": 1.483917236328125e-05,
      "step": 15560
    },
    {
      "epoch": 1.483917236328125e-05,
      "step": 15560,
      "training_step_time": 8.408942937850952
    },
    {
      "epoch": 1.4840126037597656e-05,
      "model_forward_time": 1.7485301494598389,
      "step": 15561
    },
    {
      "epoch": 1.4840126037597656e-05,
      "step": 15561,
      "training_step_time": 8.678805589675903
    },
    {
      "epoch": 1.4841079711914062e-05,
      "model_forward_time": 1.7514348030090332,
      "step": 15562
    },
    {
      "epoch": 1.4841079711914062e-05,
      "step": 15562,
      "training_step_time": 8.67842173576355
    },
    {
      "epoch": 1.484203338623047e-05,
      "model_forward_time": 2.913806915283203,
      "step": 15563
    },
    {
      "epoch": 1.484203338623047e-05,
      "step": 15563,
      "training_step_time": 7.735666036605835
    },
    {
      "epoch": 1.4842987060546875e-05,
      "model_forward_time": 1.7607100009918213,
      "step": 15564
    },
    {
      "epoch": 1.4842987060546875e-05,
      "step": 15564,
      "training_step_time": 7.716739177703857
    },
    {
      "epoch": 1.484394073486328e-05,
      "model_forward_time": 1.7526848316192627,
      "step": 15565
    },
    {
      "epoch": 1.484394073486328e-05,
      "step": 15565,
      "training_step_time": 8.85014271736145
    },
    {
      "epoch": 1.4844894409179688e-05,
      "model_forward_time": 1.7528493404388428,
      "step": 15566
    },
    {
      "epoch": 1.4844894409179688e-05,
      "step": 15566,
      "training_step_time": 8.85713815689087
    },
    {
      "epoch": 1.4845848083496094e-05,
      "model_forward_time": 1.7504894733428955,
      "step": 15567
    },
    {
      "epoch": 1.4845848083496094e-05,
      "step": 15567,
      "training_step_time": 8.803714275360107
    },
    {
      "epoch": 1.48468017578125e-05,
      "model_forward_time": 1.753316879272461,
      "step": 15568
    },
    {
      "epoch": 1.48468017578125e-05,
      "step": 15568,
      "training_step_time": 7.743471384048462
    },
    {
      "epoch": 1.4847755432128907e-05,
      "model_forward_time": 1.7888355255126953,
      "step": 15569
    },
    {
      "epoch": 1.4847755432128907e-05,
      "step": 15569,
      "training_step_time": 8.675990343093872
    },
    {
      "epoch": 1.4848709106445313e-05,
      "grad_norm": 0.2645444869995117,
      "learning_rate": 9.69770232938393e-06,
      "loss": 0.0682,
      "step": 15570
    },
    {
      "epoch": 1.4848709106445313e-05,
      "model_forward_time": 1.7785027027130127,
      "step": 15570
    },
    {
      "dynamics_loss_avg": 0.06015081778168678,
      "epoch": 1.4848709106445313e-05,
      "step": 15570
    },
    {
      "action_loss_avg": 0.003327664267271757,
      "epoch": 1.4848709106445313e-05,
      "step": 15570
    },
    {
      "epoch": 1.4848709106445313e-05,
      "step": 15570,
      "training_step_time": 8.463585376739502
    },
    {
      "epoch": 1.4849662780761718e-05,
      "model_forward_time": 1.7530920505523682,
      "step": 15571
    },
    {
      "epoch": 1.4849662780761718e-05,
      "step": 15571,
      "training_step_time": 8.801063060760498
    },
    {
      "epoch": 1.4850616455078124e-05,
      "model_forward_time": 2.897071599960327,
      "step": 15572
    },
    {
      "epoch": 1.4850616455078124e-05,
      "step": 15572,
      "training_step_time": 8.3468599319458
    },
    {
      "epoch": 1.4851570129394532e-05,
      "model_forward_time": 1.7491354942321777,
      "step": 15573
    },
    {
      "epoch": 1.4851570129394532e-05,
      "step": 15573,
      "training_step_time": 8.414138078689575
    },
    {
      "epoch": 1.4852523803710937e-05,
      "model_forward_time": 1.7511076927185059,
      "step": 15574
    },
    {
      "epoch": 1.4852523803710937e-05,
      "step": 15574,
      "training_step_time": 7.656754493713379
    },
    {
      "epoch": 1.4853477478027343e-05,
      "model_forward_time": 1.757394552230835,
      "step": 15575
    },
    {
      "epoch": 1.4853477478027343e-05,
      "step": 15575,
      "training_step_time": 8.278169393539429
    },
    {
      "epoch": 1.485443115234375e-05,
      "model_forward_time": 1.388756513595581,
      "step": 15576
    },
    {
      "epoch": 1.485443115234375e-05,
      "step": 15576,
      "training_step_time": 7.719484329223633
    },
    {
      "epoch": 1.4855384826660156e-05,
      "model_forward_time": 1.7599585056304932,
      "step": 15577
    },
    {
      "epoch": 1.4855384826660156e-05,
      "step": 15577,
      "training_step_time": 7.837171316146851
    },
    {
      "epoch": 1.4856338500976562e-05,
      "model_forward_time": 1.773603916168213,
      "step": 15578
    },
    {
      "epoch": 1.4856338500976562e-05,
      "step": 15578,
      "training_step_time": 7.604613542556763
    },
    {
      "epoch": 1.485729217529297e-05,
      "model_forward_time": 1.8293139934539795,
      "step": 15579
    },
    {
      "epoch": 1.485729217529297e-05,
      "step": 15579,
      "training_step_time": 9.268535614013672
    },
    {
      "epoch": 1.4858245849609375e-05,
      "grad_norm": 0.32241055369377136,
      "learning_rate": 9.697135861801075e-06,
      "loss": 0.0714,
      "step": 15580
    },
    {
      "epoch": 1.4858245849609375e-05,
      "model_forward_time": 1.7673773765563965,
      "step": 15580
    },
    {
      "dynamics_loss_avg": 0.07281357999891043,
      "epoch": 1.4858245849609375e-05,
      "step": 15580
    },
    {
      "action_loss_avg": 0.003304961498361081,
      "epoch": 1.4858245849609375e-05,
      "step": 15580
    },
    {
      "epoch": 1.4858245849609375e-05,
      "step": 15580,
      "training_step_time": 8.525132894515991
    },
    {
      "epoch": 1.485919952392578e-05,
      "model_forward_time": 2.8478949069976807,
      "step": 15581
    },
    {
      "epoch": 1.485919952392578e-05,
      "step": 15581,
      "training_step_time": 7.72810959815979
    },
    {
      "epoch": 1.4860153198242188e-05,
      "model_forward_time": 1.7591922283172607,
      "step": 15582
    },
    {
      "epoch": 1.4860153198242188e-05,
      "step": 15582,
      "training_step_time": 7.643029451370239
    },
    {
      "epoch": 1.4861106872558594e-05,
      "model_forward_time": 1.763960361480713,
      "step": 15583
    },
    {
      "epoch": 1.4861106872558594e-05,
      "step": 15583,
      "training_step_time": 8.629355192184448
    },
    {
      "epoch": 1.4862060546875e-05,
      "model_forward_time": 1.7637417316436768,
      "step": 15584
    },
    {
      "epoch": 1.4862060546875e-05,
      "step": 15584,
      "training_step_time": 8.190959453582764
    },
    {
      "epoch": 1.4863014221191407e-05,
      "model_forward_time": 1.7654411792755127,
      "step": 15585
    },
    {
      "epoch": 1.4863014221191407e-05,
      "step": 15585,
      "training_step_time": 7.7716710567474365
    },
    {
      "epoch": 1.4863967895507813e-05,
      "model_forward_time": 1.761810302734375,
      "step": 15586
    },
    {
      "epoch": 1.4863967895507813e-05,
      "step": 15586,
      "training_step_time": 8.519186019897461
    },
    {
      "epoch": 1.4864921569824219e-05,
      "model_forward_time": 1.7564566135406494,
      "step": 15587
    },
    {
      "epoch": 1.4864921569824219e-05,
      "step": 15587,
      "training_step_time": 8.27851128578186
    },
    {
      "epoch": 1.4865875244140624e-05,
      "model_forward_time": 1.83880615234375,
      "step": 15588
    },
    {
      "epoch": 1.4865875244140624e-05,
      "step": 15588,
      "training_step_time": 9.140785932540894
    },
    {
      "epoch": 1.4866828918457032e-05,
      "model_forward_time": 1.759873867034912,
      "step": 15589
    },
    {
      "epoch": 1.4866828918457032e-05,
      "step": 15589,
      "training_step_time": 10.272495746612549
    },
    {
      "epoch": 1.4867782592773437e-05,
      "grad_norm": 0.2208503633737564,
      "learning_rate": 9.696568880546506e-06,
      "loss": 0.0683,
      "step": 15590
    },
    {
      "epoch": 1.4867782592773437e-05,
      "model_forward_time": 1.7643306255340576,
      "step": 15590
    },
    {
      "dynamics_loss_avg": 0.05704167690128088,
      "epoch": 1.4867782592773437e-05,
      "step": 15590
    },
    {
      "action_loss_avg": 0.002618346473900601,
      "epoch": 1.4867782592773437e-05,
      "step": 15590
    },
    {
      "epoch": 1.4867782592773437e-05,
      "step": 15590,
      "training_step_time": 8.307660341262817
    },
    {
      "epoch": 1.4868736267089843e-05,
      "model_forward_time": 2.932996988296509,
      "step": 15591
    },
    {
      "epoch": 1.4868736267089843e-05,
      "step": 15591,
      "training_step_time": 7.754868268966675
    },
    {
      "epoch": 1.486968994140625e-05,
      "model_forward_time": 1.7670187950134277,
      "step": 15592
    },
    {
      "epoch": 1.486968994140625e-05,
      "step": 15592,
      "training_step_time": 7.629436016082764
    },
    {
      "epoch": 1.4870643615722656e-05,
      "model_forward_time": 1.7659103870391846,
      "step": 15593
    },
    {
      "epoch": 1.4870643615722656e-05,
      "step": 15593,
      "training_step_time": 7.594017744064331
    },
    {
      "epoch": 1.4871597290039062e-05,
      "model_forward_time": 1.762120246887207,
      "step": 15594
    },
    {
      "epoch": 1.4871597290039062e-05,
      "step": 15594,
      "training_step_time": 7.711815595626831
    },
    {
      "epoch": 1.487255096435547e-05,
      "model_forward_time": 1.7658758163452148,
      "step": 15595
    },
    {
      "epoch": 1.487255096435547e-05,
      "step": 15595,
      "training_step_time": 8.302301406860352
    },
    {
      "epoch": 1.4873504638671875e-05,
      "model_forward_time": 1.756197214126587,
      "step": 15596
    },
    {
      "epoch": 1.4873504638671875e-05,
      "step": 15596,
      "training_step_time": 10.167500257492065
    },
    {
      "epoch": 1.487445831298828e-05,
      "model_forward_time": 1.828387975692749,
      "step": 15597
    },
    {
      "epoch": 1.487445831298828e-05,
      "step": 15597,
      "training_step_time": 8.725358963012695
    },
    {
      "epoch": 1.4875411987304688e-05,
      "model_forward_time": 1.7837963104248047,
      "step": 15598
    },
    {
      "epoch": 1.4875411987304688e-05,
      "step": 15598,
      "training_step_time": 8.320267915725708
    },
    {
      "epoch": 1.4876365661621094e-05,
      "model_forward_time": 1.7666258811950684,
      "step": 15599
    },
    {
      "epoch": 1.4876365661621094e-05,
      "step": 15599,
      "training_step_time": 7.72160005569458
    },
    {
      "epoch": 1.48773193359375e-05,
      "grad_norm": 0.30550992488861084,
      "learning_rate": 9.696001385682223e-06,
      "loss": 0.067,
      "step": 15600
    },
    {
      "epoch": 1.48773193359375e-05,
      "model_forward_time": 2.951273202896118,
      "step": 15600
    },
    {
      "dynamics_loss_avg": 0.06348404623568057,
      "epoch": 1.48773193359375e-05,
      "step": 15600
    },
    {
      "action_loss_avg": 0.0032815069309435783,
      "epoch": 1.48773193359375e-05,
      "step": 15600
    },
    {
      "epoch": 1.48773193359375e-05,
      "step": 15600,
      "training_step_time": 7.7684736251831055
    },
    {
      "epoch": 1.4878273010253907e-05,
      "model_forward_time": 1.7623226642608643,
      "step": 15601
    },
    {
      "epoch": 1.4878273010253907e-05,
      "step": 15601,
      "training_step_time": 7.7398645877838135
    },
    {
      "epoch": 1.4879226684570313e-05,
      "model_forward_time": 1.7573533058166504,
      "step": 15602
    },
    {
      "epoch": 1.4879226684570313e-05,
      "step": 15602,
      "training_step_time": 7.688092231750488
    },
    {
      "epoch": 1.4880180358886719e-05,
      "model_forward_time": 1.801469087600708,
      "step": 15603
    },
    {
      "epoch": 1.4880180358886719e-05,
      "step": 15603,
      "training_step_time": 7.545274019241333
    },
    {
      "epoch": 1.4881134033203124e-05,
      "model_forward_time": 1.7859537601470947,
      "step": 15604
    },
    {
      "epoch": 1.4881134033203124e-05,
      "step": 15604,
      "training_step_time": 7.7883172035217285
    },
    {
      "epoch": 1.4882087707519532e-05,
      "model_forward_time": 1.761610746383667,
      "step": 15605
    },
    {
      "epoch": 1.4882087707519532e-05,
      "step": 15605,
      "training_step_time": 7.6536171436309814
    },
    {
      "epoch": 1.4883041381835937e-05,
      "model_forward_time": 1.7792201042175293,
      "step": 15606
    },
    {
      "epoch": 1.4883041381835937e-05,
      "step": 15606,
      "training_step_time": 7.666700839996338
    },
    {
      "epoch": 1.4883995056152343e-05,
      "model_forward_time": 1.7531092166900635,
      "step": 15607
    },
    {
      "epoch": 1.4883995056152343e-05,
      "step": 15607,
      "training_step_time": 7.883711338043213
    },
    {
      "epoch": 1.488494873046875e-05,
      "model_forward_time": 1.7617528438568115,
      "step": 15608
    },
    {
      "epoch": 1.488494873046875e-05,
      "step": 15608,
      "training_step_time": 8.974148273468018
    },
    {
      "epoch": 1.4885902404785156e-05,
      "model_forward_time": 2.9239680767059326,
      "step": 15609
    },
    {
      "epoch": 1.4885902404785156e-05,
      "step": 15609,
      "training_step_time": 7.7342753410339355
    },
    {
      "epoch": 1.4886856079101562e-05,
      "grad_norm": 0.21638448536396027,
      "learning_rate": 9.695433377270291e-06,
      "loss": 0.0691,
      "step": 15610
    },
    {
      "epoch": 1.4886856079101562e-05,
      "model_forward_time": 1.7542598247528076,
      "step": 15610
    },
    {
      "dynamics_loss_avg": 0.0580591581761837,
      "epoch": 1.4886856079101562e-05,
      "step": 15610
    },
    {
      "action_loss_avg": 0.003244296205230057,
      "epoch": 1.4886856079101562e-05,
      "step": 15610
    },
    {
      "epoch": 1.4886856079101562e-05,
      "step": 15610,
      "training_step_time": 8.528656005859375
    },
    {
      "epoch": 1.488780975341797e-05,
      "model_forward_time": 1.7734980583190918,
      "step": 15611
    },
    {
      "epoch": 1.488780975341797e-05,
      "step": 15611,
      "training_step_time": 8.218440055847168
    },
    {
      "epoch": 1.4888763427734375e-05,
      "model_forward_time": 1.070272445678711,
      "step": 15612
    },
    {
      "epoch": 1.4888763427734375e-05,
      "step": 15612,
      "training_step_time": 7.6629228591918945
    },
    {
      "epoch": 1.488971710205078e-05,
      "model_forward_time": 1.7644217014312744,
      "step": 15613
    },
    {
      "epoch": 1.488971710205078e-05,
      "step": 15613,
      "training_step_time": 7.8179662227630615
    },
    {
      "epoch": 1.4890670776367188e-05,
      "model_forward_time": 1.7599055767059326,
      "step": 15614
    },
    {
      "epoch": 1.4890670776367188e-05,
      "step": 15614,
      "training_step_time": 7.941479206085205
    },
    {
      "epoch": 1.4891624450683594e-05,
      "model_forward_time": 1.76851224899292,
      "step": 15615
    },
    {
      "epoch": 1.4891624450683594e-05,
      "step": 15615,
      "training_step_time": 8.276051044464111
    },
    {
      "epoch": 1.4892578125e-05,
      "model_forward_time": 1.8284938335418701,
      "step": 15616
    },
    {
      "epoch": 1.4892578125e-05,
      "step": 15616,
      "training_step_time": 8.43174123764038
    },
    {
      "epoch": 1.4893531799316407e-05,
      "model_forward_time": 1.7511537075042725,
      "step": 15617
    },
    {
      "epoch": 1.4893531799316407e-05,
      "step": 15617,
      "training_step_time": 9.173082828521729
    },
    {
      "epoch": 1.4894485473632813e-05,
      "model_forward_time": 1.7557494640350342,
      "step": 15618
    },
    {
      "epoch": 1.4894485473632813e-05,
      "step": 15618,
      "training_step_time": 8.765817880630493
    },
    {
      "epoch": 1.4895439147949219e-05,
      "model_forward_time": 1.757904291152954,
      "step": 15619
    },
    {
      "epoch": 1.4895439147949219e-05,
      "step": 15619,
      "training_step_time": 7.720712900161743
    },
    {
      "epoch": 1.4896392822265624e-05,
      "grad_norm": 0.29450127482414246,
      "learning_rate": 9.694864855372824e-06,
      "loss": 0.0684,
      "step": 15620
    },
    {
      "epoch": 1.4896392822265624e-05,
      "model_forward_time": 1.7618746757507324,
      "step": 15620
    },
    {
      "dynamics_loss_avg": 0.06663596043363214,
      "epoch": 1.4896392822265624e-05,
      "step": 15620
    },
    {
      "action_loss_avg": 0.00357374552404508,
      "epoch": 1.4896392822265624e-05,
      "step": 15620
    },
    {
      "epoch": 1.4896392822265624e-05,
      "step": 15620,
      "training_step_time": 7.696959733963013
    },
    {
      "epoch": 1.4897346496582032e-05,
      "model_forward_time": 1.749659538269043,
      "step": 15621
    },
    {
      "epoch": 1.4897346496582032e-05,
      "step": 15621,
      "training_step_time": 7.770627498626709
    },
    {
      "epoch": 1.4898300170898437e-05,
      "model_forward_time": 1.752309799194336,
      "step": 15622
    },
    {
      "epoch": 1.4898300170898437e-05,
      "step": 15622,
      "training_step_time": 8.358553647994995
    },
    {
      "epoch": 1.4899253845214843e-05,
      "model_forward_time": 1.7492733001708984,
      "step": 15623
    },
    {
      "epoch": 1.4899253845214843e-05,
      "step": 15623,
      "training_step_time": 8.805119276046753
    },
    {
      "epoch": 1.490020751953125e-05,
      "model_forward_time": 1.7577166557312012,
      "step": 15624
    },
    {
      "epoch": 1.490020751953125e-05,
      "step": 15624,
      "training_step_time": 8.644729137420654
    },
    {
      "epoch": 1.4901161193847656e-05,
      "model_forward_time": 1.8296434879302979,
      "step": 15625
    },
    {
      "epoch": 1.4901161193847656e-05,
      "step": 15625,
      "training_step_time": 7.751620531082153
    },
    {
      "epoch": 1.4902114868164062e-05,
      "model_forward_time": 1.7476000785827637,
      "step": 15626
    },
    {
      "epoch": 1.4902114868164062e-05,
      "step": 15626,
      "training_step_time": 8.923208475112915
    },
    {
      "epoch": 1.490306854248047e-05,
      "model_forward_time": 1.7532896995544434,
      "step": 15627
    },
    {
      "epoch": 1.490306854248047e-05,
      "step": 15627,
      "training_step_time": 7.733871698379517
    },
    {
      "epoch": 1.4904022216796875e-05,
      "model_forward_time": 2.527501106262207,
      "step": 15628
    },
    {
      "epoch": 1.4904022216796875e-05,
      "step": 15628,
      "training_step_time": 7.841732025146484
    },
    {
      "epoch": 1.490497589111328e-05,
      "model_forward_time": 1.7582077980041504,
      "step": 15629
    },
    {
      "epoch": 1.490497589111328e-05,
      "step": 15629,
      "training_step_time": 8.877824783325195
    },
    {
      "epoch": 1.4905929565429688e-05,
      "grad_norm": 0.2891431152820587,
      "learning_rate": 9.694295820051996e-06,
      "loss": 0.0652,
      "step": 15630
    },
    {
      "epoch": 1.4905929565429688e-05,
      "model_forward_time": 1.7425649166107178,
      "step": 15630
    },
    {
      "dynamics_loss_avg": 0.060854466445744035,
      "epoch": 1.4905929565429688e-05,
      "step": 15630
    },
    {
      "action_loss_avg": 0.004206248431000858,
      "epoch": 1.4905929565429688e-05,
      "step": 15630
    },
    {
      "epoch": 1.4905929565429688e-05,
      "step": 15630,
      "training_step_time": 7.7213380336761475
    },
    {
      "epoch": 1.4906883239746094e-05,
      "model_forward_time": 1.767357349395752,
      "step": 15631
    },
    {
      "epoch": 1.4906883239746094e-05,
      "step": 15631,
      "training_step_time": 7.7887187004089355
    },
    {
      "epoch": 1.49078369140625e-05,
      "model_forward_time": 1.7633605003356934,
      "step": 15632
    },
    {
      "epoch": 1.49078369140625e-05,
      "step": 15632,
      "training_step_time": 7.753705263137817
    },
    {
      "epoch": 1.4908790588378905e-05,
      "model_forward_time": 1.772209644317627,
      "step": 15633
    },
    {
      "epoch": 1.4908790588378905e-05,
      "step": 15633,
      "training_step_time": 7.623088836669922
    },
    {
      "epoch": 1.4909744262695313e-05,
      "model_forward_time": 1.8359498977661133,
      "step": 15634
    },
    {
      "epoch": 1.4909744262695313e-05,
      "step": 15634,
      "training_step_time": 9.606890439987183
    },
    {
      "epoch": 1.4910697937011719e-05,
      "model_forward_time": 1.76531982421875,
      "step": 15635
    },
    {
      "epoch": 1.4910697937011719e-05,
      "step": 15635,
      "training_step_time": 8.44216537475586
    },
    {
      "epoch": 1.4911651611328124e-05,
      "model_forward_time": 1.7668883800506592,
      "step": 15636
    },
    {
      "epoch": 1.4911651611328124e-05,
      "step": 15636,
      "training_step_time": 7.684642791748047
    },
    {
      "epoch": 1.4912605285644532e-05,
      "model_forward_time": 2.94527530670166,
      "step": 15637
    },
    {
      "epoch": 1.4912605285644532e-05,
      "step": 15637,
      "training_step_time": 7.901949405670166
    },
    {
      "epoch": 1.4913558959960937e-05,
      "model_forward_time": 1.7603609561920166,
      "step": 15638
    },
    {
      "epoch": 1.4913558959960937e-05,
      "step": 15638,
      "training_step_time": 7.736568450927734
    },
    {
      "epoch": 1.4914512634277343e-05,
      "model_forward_time": 1.7609455585479736,
      "step": 15639
    },
    {
      "epoch": 1.4914512634277343e-05,
      "step": 15639,
      "training_step_time": 7.778220176696777
    },
    {
      "epoch": 1.491546630859375e-05,
      "grad_norm": 0.3127107620239258,
      "learning_rate": 9.693726271370032e-06,
      "loss": 0.0663,
      "step": 15640
    },
    {
      "epoch": 1.491546630859375e-05,
      "model_forward_time": 1.3981971740722656,
      "step": 15640
    },
    {
      "dynamics_loss_avg": 0.051103474013507365,
      "epoch": 1.491546630859375e-05,
      "step": 15640
    },
    {
      "action_loss_avg": 0.0020484771695919333,
      "epoch": 1.491546630859375e-05,
      "step": 15640
    },
    {
      "epoch": 1.491546630859375e-05,
      "step": 15640,
      "training_step_time": 7.833590030670166
    },
    {
      "epoch": 1.4916419982910156e-05,
      "model_forward_time": 1.4147148132324219,
      "step": 15641
    },
    {
      "epoch": 1.4916419982910156e-05,
      "step": 15641,
      "training_step_time": 8.749315738677979
    },
    {
      "epoch": 1.4917373657226562e-05,
      "model_forward_time": 1.931262493133545,
      "step": 15642
    },
    {
      "epoch": 1.4917373657226562e-05,
      "step": 15642,
      "training_step_time": 7.777340412139893
    },
    {
      "epoch": 1.491832733154297e-05,
      "model_forward_time": 1.9251728057861328,
      "step": 15643
    },
    {
      "epoch": 1.491832733154297e-05,
      "step": 15643,
      "training_step_time": 6.833690643310547
    },
    {
      "epoch": 1.4919281005859375e-05,
      "model_forward_time": 1.9239883422851562,
      "step": 15644
    },
    {
      "epoch": 1.4919281005859375e-05,
      "step": 15644,
      "training_step_time": 6.927513360977173
    },
    {
      "epoch": 1.492023468017578e-05,
      "model_forward_time": 1.9279990196228027,
      "step": 15645
    },
    {
      "epoch": 1.492023468017578e-05,
      "step": 15645,
      "training_step_time": 6.734977722167969
    },
    {
      "epoch": 1.4921188354492188e-05,
      "model_forward_time": 1.8890748023986816,
      "step": 15646
    },
    {
      "epoch": 1.4921188354492188e-05,
      "step": 15646,
      "training_step_time": 6.81786584854126
    },
    {
      "epoch": 1.4922142028808594e-05,
      "model_forward_time": 1.544511079788208,
      "step": 15647
    },
    {
      "epoch": 1.4922142028808594e-05,
      "step": 15647,
      "training_step_time": 6.821579933166504
    },
    {
      "epoch": 1.4923095703125e-05,
      "model_forward_time": 1.8911848068237305,
      "step": 15648
    },
    {
      "epoch": 1.4923095703125e-05,
      "step": 15648,
      "training_step_time": 6.890044212341309
    },
    {
      "epoch": 1.4924049377441405e-05,
      "model_forward_time": 1.9218764305114746,
      "step": 15649
    },
    {
      "epoch": 1.4924049377441405e-05,
      "step": 15649,
      "training_step_time": 6.890604019165039
    },
    {
      "epoch": 1.4925003051757813e-05,
      "grad_norm": 0.26944872736930847,
      "learning_rate": 9.693156209389221e-06,
      "loss": 0.065,
      "step": 15650
    },
    {
      "epoch": 1.4925003051757813e-05,
      "model_forward_time": 1.912705421447754,
      "step": 15650
    },
    {
      "dynamics_loss_avg": 0.05438988842070103,
      "epoch": 1.4925003051757813e-05,
      "step": 15650
    },
    {
      "action_loss_avg": 0.004961517755873501,
      "epoch": 1.4925003051757813e-05,
      "step": 15650
    },
    {
      "epoch": 1.4925003051757813e-05,
      "step": 15650,
      "training_step_time": 11.08322286605835
    },
    {
      "epoch": 1.4925956726074219e-05,
      "model_forward_time": 2.9785125255584717,
      "step": 15651
    },
    {
      "epoch": 1.4925956726074219e-05,
      "step": 15651,
      "training_step_time": 7.967439889907837
    },
    {
      "epoch": 1.4926910400390624e-05,
      "model_forward_time": 1.9410347938537598,
      "step": 15652
    },
    {
      "epoch": 1.4926910400390624e-05,
      "step": 15652,
      "training_step_time": 7.314180135726929
    },
    {
      "epoch": 1.4927864074707032e-05,
      "model_forward_time": 1.8821728229522705,
      "step": 15653
    },
    {
      "epoch": 1.4927864074707032e-05,
      "step": 15653,
      "training_step_time": 6.827230930328369
    },
    {
      "epoch": 1.4928817749023437e-05,
      "model_forward_time": 1.9372212886810303,
      "step": 15654
    },
    {
      "epoch": 1.4928817749023437e-05,
      "step": 15654,
      "training_step_time": 6.839784145355225
    },
    {
      "epoch": 1.4929771423339843e-05,
      "model_forward_time": 1.9013183116912842,
      "step": 15655
    },
    {
      "epoch": 1.4929771423339843e-05,
      "step": 15655,
      "training_step_time": 6.8199121952056885
    },
    {
      "epoch": 1.493072509765625e-05,
      "model_forward_time": 1.9213454723358154,
      "step": 15656
    },
    {
      "epoch": 1.493072509765625e-05,
      "step": 15656,
      "training_step_time": 6.8121349811553955
    },
    {
      "epoch": 1.4931678771972656e-05,
      "model_forward_time": 1.9401483535766602,
      "step": 15657
    },
    {
      "epoch": 1.4931678771972656e-05,
      "step": 15657,
      "training_step_time": 6.7819366455078125
    },
    {
      "epoch": 1.4932632446289062e-05,
      "model_forward_time": 1.9022350311279297,
      "step": 15658
    },
    {
      "epoch": 1.4932632446289062e-05,
      "step": 15658,
      "training_step_time": 6.793363571166992
    },
    {
      "epoch": 1.493358612060547e-05,
      "model_forward_time": 1.925107479095459,
      "step": 15659
    },
    {
      "epoch": 1.493358612060547e-05,
      "step": 15659,
      "training_step_time": 6.782973527908325
    },
    {
      "epoch": 1.4934539794921875e-05,
      "grad_norm": 0.2901770770549774,
      "learning_rate": 9.692585634171906e-06,
      "loss": 0.0635,
      "step": 15660
    },
    {
      "epoch": 1.4934539794921875e-05,
      "model_forward_time": 3.0341615676879883,
      "step": 15660
    },
    {
      "dynamics_loss_avg": 0.07038047220557928,
      "epoch": 1.4934539794921875e-05,
      "step": 15660
    },
    {
      "action_loss_avg": 0.003953241533599794,
      "epoch": 1.4934539794921875e-05,
      "step": 15660
    },
    {
      "epoch": 1.4934539794921875e-05,
      "step": 15660,
      "training_step_time": 7.825354099273682
    },
    {
      "epoch": 1.4935493469238281e-05,
      "model_forward_time": 1.9343087673187256,
      "step": 15661
    },
    {
      "epoch": 1.4935493469238281e-05,
      "step": 15661,
      "training_step_time": 7.285309553146362
    },
    {
      "epoch": 1.4936447143554688e-05,
      "model_forward_time": 1.922234058380127,
      "step": 15662
    },
    {
      "epoch": 1.4936447143554688e-05,
      "step": 15662,
      "training_step_time": 6.853515625
    },
    {
      "epoch": 1.4937400817871094e-05,
      "model_forward_time": 1.908390760421753,
      "step": 15663
    },
    {
      "epoch": 1.4937400817871094e-05,
      "step": 15663,
      "training_step_time": 6.813729286193848
    },
    {
      "epoch": 1.49383544921875e-05,
      "model_forward_time": 1.9411664009094238,
      "step": 15664
    },
    {
      "epoch": 1.49383544921875e-05,
      "step": 15664,
      "training_step_time": 6.804579019546509
    },
    {
      "epoch": 1.4939308166503905e-05,
      "model_forward_time": 1.8943209648132324,
      "step": 15665
    },
    {
      "epoch": 1.4939308166503905e-05,
      "step": 15665,
      "training_step_time": 6.807766437530518
    },
    {
      "epoch": 1.4940261840820313e-05,
      "model_forward_time": 1.9282939434051514,
      "step": 15666
    },
    {
      "epoch": 1.4940261840820313e-05,
      "step": 15666,
      "training_step_time": 6.77191686630249
    },
    {
      "epoch": 1.4941215515136719e-05,
      "model_forward_time": 1.9335761070251465,
      "step": 15667
    },
    {
      "epoch": 1.4941215515136719e-05,
      "step": 15667,
      "training_step_time": 6.806737661361694
    },
    {
      "epoch": 1.4942169189453124e-05,
      "model_forward_time": 1.9200353622436523,
      "step": 15668
    },
    {
      "epoch": 1.4942169189453124e-05,
      "step": 15668,
      "training_step_time": 6.795175552368164
    },
    {
      "epoch": 1.4943122863769532e-05,
      "model_forward_time": 3.008908271789551,
      "step": 15669
    },
    {
      "epoch": 1.4943122863769532e-05,
      "step": 15669,
      "training_step_time": 9.365927934646606
    },
    {
      "epoch": 1.4944076538085937e-05,
      "grad_norm": 0.28276491165161133,
      "learning_rate": 9.692014545780477e-06,
      "loss": 0.0641,
      "step": 15670
    },
    {
      "epoch": 1.4944076538085937e-05,
      "model_forward_time": 1.7451815605163574,
      "step": 15670
    },
    {
      "dynamics_loss_avg": 0.05683670472353697,
      "epoch": 1.4944076538085937e-05,
      "step": 15670
    },
    {
      "action_loss_avg": 0.002850269706686959,
      "epoch": 1.4944076538085937e-05,
      "step": 15670
    },
    {
      "epoch": 1.4944076538085937e-05,
      "step": 15670,
      "training_step_time": 7.511934280395508
    },
    {
      "epoch": 1.4945030212402343e-05,
      "model_forward_time": 1.7514615058898926,
      "step": 15671
    },
    {
      "epoch": 1.4945030212402343e-05,
      "step": 15671,
      "training_step_time": 6.75562310218811
    },
    {
      "epoch": 1.494598388671875e-05,
      "model_forward_time": 1.752467393875122,
      "step": 15672
    },
    {
      "epoch": 1.494598388671875e-05,
      "step": 15672,
      "training_step_time": 6.80101203918457
    },
    {
      "epoch": 1.4946937561035156e-05,
      "model_forward_time": 1.758394718170166,
      "step": 15673
    },
    {
      "epoch": 1.4946937561035156e-05,
      "step": 15673,
      "training_step_time": 6.800964832305908
    },
    {
      "epoch": 1.4947891235351562e-05,
      "model_forward_time": 1.7592439651489258,
      "step": 15674
    },
    {
      "epoch": 1.4947891235351562e-05,
      "step": 15674,
      "training_step_time": 6.800725221633911
    },
    {
      "epoch": 1.494884490966797e-05,
      "model_forward_time": 1.7501134872436523,
      "step": 15675
    },
    {
      "epoch": 1.494884490966797e-05,
      "step": 15675,
      "training_step_time": 6.754127264022827
    },
    {
      "epoch": 1.4949798583984375e-05,
      "model_forward_time": 1.755127191543579,
      "step": 15676
    },
    {
      "epoch": 1.4949798583984375e-05,
      "step": 15676,
      "training_step_time": 6.859866142272949
    },
    {
      "epoch": 1.4950752258300781e-05,
      "model_forward_time": 1.7503490447998047,
      "step": 15677
    },
    {
      "epoch": 1.4950752258300781e-05,
      "step": 15677,
      "training_step_time": 6.792576551437378
    },
    {
      "epoch": 1.4951705932617188e-05,
      "model_forward_time": 2.9370434284210205,
      "step": 15678
    },
    {
      "epoch": 1.4951705932617188e-05,
      "step": 15678,
      "training_step_time": 11.020948648452759
    },
    {
      "epoch": 1.4952659606933594e-05,
      "model_forward_time": 1.7561941146850586,
      "step": 15679
    },
    {
      "epoch": 1.4952659606933594e-05,
      "step": 15679,
      "training_step_time": 7.7813401222229
    },
    {
      "epoch": 1.495361328125e-05,
      "grad_norm": 0.21242755651474,
      "learning_rate": 9.691442944277393e-06,
      "loss": 0.0682,
      "step": 15680
    },
    {
      "epoch": 1.495361328125e-05,
      "model_forward_time": 1.770869255065918,
      "step": 15680
    },
    {
      "dynamics_loss_avg": 0.056659283116459846,
      "epoch": 1.495361328125e-05,
      "step": 15680
    },
    {
      "action_loss_avg": 0.002872230822686106,
      "epoch": 1.495361328125e-05,
      "step": 15680
    },
    {
      "epoch": 1.495361328125e-05,
      "step": 15680,
      "training_step_time": 6.801748037338257
    },
    {
      "epoch": 1.4954566955566405e-05,
      "model_forward_time": 1.787651538848877,
      "step": 15681
    },
    {
      "epoch": 1.4954566955566405e-05,
      "step": 15681,
      "training_step_time": 6.794010162353516
    },
    {
      "epoch": 1.4955520629882813e-05,
      "model_forward_time": 1.7608883380889893,
      "step": 15682
    },
    {
      "epoch": 1.4955520629882813e-05,
      "step": 15682,
      "training_step_time": 6.768332481384277
    },
    {
      "epoch": 1.4956474304199219e-05,
      "model_forward_time": 1.7613861560821533,
      "step": 15683
    },
    {
      "epoch": 1.4956474304199219e-05,
      "step": 15683,
      "training_step_time": 6.841998338699341
    },
    {
      "epoch": 1.4957427978515624e-05,
      "model_forward_time": 1.763242483139038,
      "step": 15684
    },
    {
      "epoch": 1.4957427978515624e-05,
      "step": 15684,
      "training_step_time": 6.880250692367554
    },
    {
      "epoch": 1.4958381652832032e-05,
      "model_forward_time": 1.7623920440673828,
      "step": 15685
    },
    {
      "epoch": 1.4958381652832032e-05,
      "step": 15685,
      "training_step_time": 6.841594696044922
    },
    {
      "epoch": 1.4959335327148437e-05,
      "model_forward_time": 1.7666001319885254,
      "step": 15686
    },
    {
      "epoch": 1.4959335327148437e-05,
      "step": 15686,
      "training_step_time": 6.854531764984131
    },
    {
      "epoch": 1.4960289001464843e-05,
      "model_forward_time": 1.8030760288238525,
      "step": 15687
    },
    {
      "epoch": 1.4960289001464843e-05,
      "step": 15687,
      "training_step_time": 11.09847116470337
    },
    {
      "epoch": 1.496124267578125e-05,
      "model_forward_time": 2.9424099922180176,
      "step": 15688
    },
    {
      "epoch": 1.496124267578125e-05,
      "step": 15688,
      "training_step_time": 7.836615085601807
    },
    {
      "epoch": 1.4962196350097656e-05,
      "model_forward_time": 1.7644734382629395,
      "step": 15689
    },
    {
      "epoch": 1.4962196350097656e-05,
      "step": 15689,
      "training_step_time": 6.926624774932861
    },
    {
      "epoch": 1.4963150024414062e-05,
      "grad_norm": 0.2311132550239563,
      "learning_rate": 9.690870829725162e-06,
      "loss": 0.0696,
      "step": 15690
    },
    {
      "epoch": 1.4963150024414062e-05,
      "model_forward_time": 1.4041728973388672,
      "step": 15690
    },
    {
      "dynamics_loss_avg": 0.06575438883155585,
      "epoch": 1.4963150024414062e-05,
      "step": 15690
    },
    {
      "action_loss_avg": 0.0041342443786561486,
      "epoch": 1.4963150024414062e-05,
      "step": 15690
    },
    {
      "epoch": 1.4963150024414062e-05,
      "step": 15690,
      "training_step_time": 6.88383936882019
    },
    {
      "epoch": 1.496410369873047e-05,
      "model_forward_time": 1.764293909072876,
      "step": 15691
    },
    {
      "epoch": 1.496410369873047e-05,
      "step": 15691,
      "training_step_time": 6.869466066360474
    },
    {
      "epoch": 1.4965057373046875e-05,
      "model_forward_time": 1.765773057937622,
      "step": 15692
    },
    {
      "epoch": 1.4965057373046875e-05,
      "step": 15692,
      "training_step_time": 6.905208110809326
    },
    {
      "epoch": 1.4966011047363281e-05,
      "model_forward_time": 1.7606542110443115,
      "step": 15693
    },
    {
      "epoch": 1.4966011047363281e-05,
      "step": 15693,
      "training_step_time": 6.888665437698364
    },
    {
      "epoch": 1.4966964721679688e-05,
      "model_forward_time": 1.7569780349731445,
      "step": 15694
    },
    {
      "epoch": 1.4966964721679688e-05,
      "step": 15694,
      "training_step_time": 6.952918291091919
    },
    {
      "epoch": 1.4967918395996094e-05,
      "model_forward_time": 1.7501204013824463,
      "step": 15695
    },
    {
      "epoch": 1.4967918395996094e-05,
      "step": 15695,
      "training_step_time": 6.9179770946502686
    },
    {
      "epoch": 1.49688720703125e-05,
      "model_forward_time": 1.7701618671417236,
      "step": 15696
    },
    {
      "epoch": 1.49688720703125e-05,
      "step": 15696,
      "training_step_time": 9.792911052703857
    },
    {
      "epoch": 1.4969825744628905e-05,
      "model_forward_time": 2.943845510482788,
      "step": 15697
    },
    {
      "epoch": 1.4969825744628905e-05,
      "step": 15697,
      "training_step_time": 8.868494272232056
    },
    {
      "epoch": 1.4970779418945313e-05,
      "model_forward_time": 1.7354538440704346,
      "step": 15698
    },
    {
      "epoch": 1.4970779418945313e-05,
      "step": 15698,
      "training_step_time": 7.543095350265503
    },
    {
      "epoch": 1.4971733093261719e-05,
      "model_forward_time": 1.7545051574707031,
      "step": 15699
    },
    {
      "epoch": 1.4971733093261719e-05,
      "step": 15699,
      "training_step_time": 6.907113075256348
    },
    {
      "epoch": 1.4972686767578124e-05,
      "grad_norm": 0.2459271252155304,
      "learning_rate": 9.69029820218635e-06,
      "loss": 0.0656,
      "step": 15700
    },
    {
      "epoch": 1.4972686767578124e-05,
      "model_forward_time": 1.7619872093200684,
      "step": 15700
    },
    {
      "dynamics_loss_avg": 0.056728259474039075,
      "epoch": 1.4972686767578124e-05,
      "step": 15700
    },
    {
      "action_loss_avg": 0.0030571156472433358,
      "epoch": 1.4972686767578124e-05,
      "step": 15700
    },
    {
      "epoch": 1.4972686767578124e-05,
      "step": 15700,
      "training_step_time": 6.856763601303101
    },
    {
      "epoch": 1.4973640441894532e-05,
      "model_forward_time": 1.7559113502502441,
      "step": 15701
    },
    {
      "epoch": 1.4973640441894532e-05,
      "step": 15701,
      "training_step_time": 6.858243465423584
    },
    {
      "epoch": 1.4974594116210937e-05,
      "model_forward_time": 1.7513434886932373,
      "step": 15702
    },
    {
      "epoch": 1.4974594116210937e-05,
      "step": 15702,
      "training_step_time": 6.8782806396484375
    },
    {
      "epoch": 1.4975547790527343e-05,
      "model_forward_time": 1.7703142166137695,
      "step": 15703
    },
    {
      "epoch": 1.4975547790527343e-05,
      "step": 15703,
      "training_step_time": 6.899286270141602
    },
    {
      "epoch": 1.497650146484375e-05,
      "model_forward_time": 1.771836280822754,
      "step": 15704
    },
    {
      "epoch": 1.497650146484375e-05,
      "step": 15704,
      "training_step_time": 6.937360525131226
    },
    {
      "epoch": 1.4977455139160156e-05,
      "model_forward_time": 1.761073350906372,
      "step": 15705
    },
    {
      "epoch": 1.4977455139160156e-05,
      "step": 15705,
      "training_step_time": 9.456412076950073
    },
    {
      "epoch": 1.4978408813476562e-05,
      "model_forward_time": 2.9404611587524414,
      "step": 15706
    },
    {
      "epoch": 1.4978408813476562e-05,
      "step": 15706,
      "training_step_time": 8.941337823867798
    },
    {
      "epoch": 1.497936248779297e-05,
      "model_forward_time": 1.7618308067321777,
      "step": 15707
    },
    {
      "epoch": 1.497936248779297e-05,
      "step": 15707,
      "training_step_time": 7.821447372436523
    },
    {
      "epoch": 1.4980316162109375e-05,
      "model_forward_time": 1.4105267524719238,
      "step": 15708
    },
    {
      "epoch": 1.4980316162109375e-05,
      "step": 15708,
      "training_step_time": 6.963265657424927
    },
    {
      "epoch": 1.4981269836425781e-05,
      "model_forward_time": 1.756408929824829,
      "step": 15709
    },
    {
      "epoch": 1.4981269836425781e-05,
      "step": 15709,
      "training_step_time": 6.9033403396606445
    },
    {
      "epoch": 1.4982223510742188e-05,
      "grad_norm": 0.259378045797348,
      "learning_rate": 9.689725061723579e-06,
      "loss": 0.0642,
      "step": 15710
    },
    {
      "epoch": 1.4982223510742188e-05,
      "model_forward_time": 1.7453410625457764,
      "step": 15710
    },
    {
      "dynamics_loss_avg": 0.059521679393947126,
      "epoch": 1.4982223510742188e-05,
      "step": 15710
    },
    {
      "action_loss_avg": 0.0032238510204479097,
      "epoch": 1.4982223510742188e-05,
      "step": 15710
    },
    {
      "epoch": 1.4982223510742188e-05,
      "step": 15710,
      "training_step_time": 6.885617971420288
    },
    {
      "epoch": 1.4983177185058594e-05,
      "model_forward_time": 1.7562642097473145,
      "step": 15711
    },
    {
      "epoch": 1.4983177185058594e-05,
      "step": 15711,
      "training_step_time": 6.895285129547119
    },
    {
      "epoch": 1.4984130859375e-05,
      "model_forward_time": 1.7635254859924316,
      "step": 15712
    },
    {
      "epoch": 1.4984130859375e-05,
      "step": 15712,
      "training_step_time": 6.8563618659973145
    },
    {
      "epoch": 1.4985084533691405e-05,
      "model_forward_time": 1.7574949264526367,
      "step": 15713
    },
    {
      "epoch": 1.4985084533691405e-05,
      "step": 15713,
      "training_step_time": 6.849167108535767
    },
    {
      "epoch": 1.4986038208007813e-05,
      "model_forward_time": 1.7499165534973145,
      "step": 15714
    },
    {
      "epoch": 1.4986038208007813e-05,
      "step": 15714,
      "training_step_time": 7.854126453399658
    },
    {
      "epoch": 1.4986991882324219e-05,
      "model_forward_time": 2.9294631481170654,
      "step": 15715
    },
    {
      "epoch": 1.4986991882324219e-05,
      "step": 15715,
      "training_step_time": 9.598591327667236
    },
    {
      "epoch": 1.4987945556640624e-05,
      "model_forward_time": 1.7570526599884033,
      "step": 15716
    },
    {
      "epoch": 1.4987945556640624e-05,
      "step": 15716,
      "training_step_time": 7.7901389598846436
    },
    {
      "epoch": 1.4988899230957032e-05,
      "model_forward_time": 1.7615821361541748,
      "step": 15717
    },
    {
      "epoch": 1.4988899230957032e-05,
      "step": 15717,
      "training_step_time": 6.85289454460144
    },
    {
      "epoch": 1.4989852905273437e-05,
      "model_forward_time": 1.7600672245025635,
      "step": 15718
    },
    {
      "epoch": 1.4989852905273437e-05,
      "step": 15718,
      "training_step_time": 6.867060661315918
    },
    {
      "epoch": 1.4990806579589843e-05,
      "model_forward_time": 1.7617623805999756,
      "step": 15719
    },
    {
      "epoch": 1.4990806579589843e-05,
      "step": 15719,
      "training_step_time": 6.8855109214782715
    },
    {
      "epoch": 1.499176025390625e-05,
      "grad_norm": 0.29892972111701965,
      "learning_rate": 9.689151408399527e-06,
      "loss": 0.0633,
      "step": 15720
    },
    {
      "epoch": 1.499176025390625e-05,
      "model_forward_time": 1.7394108772277832,
      "step": 15720
    },
    {
      "dynamics_loss_avg": 0.06564296148717404,
      "epoch": 1.499176025390625e-05,
      "step": 15720
    },
    {
      "action_loss_avg": 0.0027139188488945363,
      "epoch": 1.499176025390625e-05,
      "step": 15720
    },
    {
      "epoch": 1.499176025390625e-05,
      "step": 15720,
      "training_step_time": 6.893768072128296
    },
    {
      "epoch": 1.4992713928222656e-05,
      "model_forward_time": 1.757141351699829,
      "step": 15721
    },
    {
      "epoch": 1.4992713928222656e-05,
      "step": 15721,
      "training_step_time": 6.8211047649383545
    },
    {
      "epoch": 1.4993667602539062e-05,
      "model_forward_time": 1.753615379333496,
      "step": 15722
    },
    {
      "epoch": 1.4993667602539062e-05,
      "step": 15722,
      "training_step_time": 6.820937156677246
    },
    {
      "epoch": 1.499462127685547e-05,
      "model_forward_time": 1.7558355331420898,
      "step": 15723
    },
    {
      "epoch": 1.499462127685547e-05,
      "step": 15723,
      "training_step_time": 8.338025331497192
    },
    {
      "epoch": 1.4995574951171875e-05,
      "model_forward_time": 2.8841381072998047,
      "step": 15724
    },
    {
      "epoch": 1.4995574951171875e-05,
      "step": 15724,
      "training_step_time": 10.322059869766235
    },
    {
      "epoch": 1.4996528625488281e-05,
      "model_forward_time": 1.7504820823669434,
      "step": 15725
    },
    {
      "epoch": 1.4996528625488281e-05,
      "step": 15725,
      "training_step_time": 8.373383283615112
    },
    {
      "epoch": 1.4997482299804688e-05,
      "model_forward_time": 1.7566263675689697,
      "step": 15726
    },
    {
      "epoch": 1.4997482299804688e-05,
      "step": 15726,
      "training_step_time": 7.600223779678345
    },
    {
      "epoch": 1.4998435974121094e-05,
      "model_forward_time": 1.7591784000396729,
      "step": 15727
    },
    {
      "epoch": 1.4998435974121094e-05,
      "step": 15727,
      "training_step_time": 6.845782518386841
    },
    {
      "epoch": 1.49993896484375e-05,
      "model_forward_time": 1.7555575370788574,
      "step": 15728
    },
    {
      "epoch": 1.49993896484375e-05,
      "step": 15728,
      "training_step_time": 6.8174028396606445
    },
    {
      "epoch": 1.5000343322753905e-05,
      "model_forward_time": 1.7579398155212402,
      "step": 15729
    },
    {
      "epoch": 1.5000343322753905e-05,
      "step": 15729,
      "training_step_time": 6.871615171432495
    },
    {
      "epoch": 1.5001296997070313e-05,
      "grad_norm": 0.2120227813720703,
      "learning_rate": 9.688577242276925e-06,
      "loss": 0.0671,
      "step": 15730
    },
    {
      "epoch": 1.5001296997070313e-05,
      "model_forward_time": 1.7039110660552979,
      "step": 15730
    },
    {
      "dynamics_loss_avg": 0.053535524569451806,
      "epoch": 1.5001296997070313e-05,
      "step": 15730
    },
    {
      "action_loss_avg": 0.003543731159879826,
      "epoch": 1.5001296997070313e-05,
      "step": 15730
    },
    {
      "epoch": 1.5001296997070313e-05,
      "step": 15730,
      "training_step_time": 6.865800380706787
    },
    {
      "epoch": 1.5002250671386719e-05,
      "model_forward_time": 1.7111084461212158,
      "step": 15731
    },
    {
      "epoch": 1.5002250671386719e-05,
      "step": 15731,
      "training_step_time": 6.839393854141235
    },
    {
      "epoch": 1.5003204345703124e-05,
      "model_forward_time": 1.7282495498657227,
      "step": 15732
    },
    {
      "epoch": 1.5003204345703124e-05,
      "step": 15732,
      "training_step_time": 7.753046989440918
    },
    {
      "epoch": 1.5004158020019532e-05,
      "model_forward_time": 1.718076467514038,
      "step": 15733
    },
    {
      "epoch": 1.5004158020019532e-05,
      "step": 15733,
      "training_step_time": 9.995561599731445
    },
    {
      "epoch": 1.5005111694335937e-05,
      "model_forward_time": 2.2765040397644043,
      "step": 15734
    },
    {
      "epoch": 1.5005111694335937e-05,
      "step": 15734,
      "training_step_time": 8.763980150222778
    },
    {
      "epoch": 1.5006065368652343e-05,
      "model_forward_time": 1.7230465412139893,
      "step": 15735
    },
    {
      "epoch": 1.5006065368652343e-05,
      "step": 15735,
      "training_step_time": 7.691844463348389
    },
    {
      "epoch": 1.500701904296875e-05,
      "model_forward_time": 1.722055196762085,
      "step": 15736
    },
    {
      "epoch": 1.500701904296875e-05,
      "step": 15736,
      "training_step_time": 6.822096824645996
    },
    {
      "epoch": 1.5007972717285156e-05,
      "model_forward_time": 1.7316474914550781,
      "step": 15737
    },
    {
      "epoch": 1.5007972717285156e-05,
      "step": 15737,
      "training_step_time": 6.844947576522827
    },
    {
      "epoch": 1.5008926391601562e-05,
      "model_forward_time": 1.7241976261138916,
      "step": 15738
    },
    {
      "epoch": 1.5008926391601562e-05,
      "step": 15738,
      "training_step_time": 6.8242573738098145
    },
    {
      "epoch": 1.500988006591797e-05,
      "model_forward_time": 1.7176852226257324,
      "step": 15739
    },
    {
      "epoch": 1.500988006591797e-05,
      "step": 15739,
      "training_step_time": 6.817023515701294
    },
    {
      "epoch": 1.5010833740234375e-05,
      "grad_norm": 0.26862603425979614,
      "learning_rate": 9.688002563418566e-06,
      "loss": 0.0641,
      "step": 15740
    },
    {
      "epoch": 1.5010833740234375e-05,
      "model_forward_time": 1.724721908569336,
      "step": 15740
    },
    {
      "dynamics_loss_avg": 0.07063991166651248,
      "epoch": 1.5010833740234375e-05,
      "step": 15740
    },
    {
      "action_loss_avg": 0.003226312668994069,
      "epoch": 1.5010833740234375e-05,
      "step": 15740
    },
    {
      "epoch": 1.5010833740234375e-05,
      "step": 15740,
      "training_step_time": 6.83993935585022
    },
    {
      "epoch": 1.5011787414550781e-05,
      "model_forward_time": 1.7113356590270996,
      "step": 15741
    },
    {
      "epoch": 1.5011787414550781e-05,
      "step": 15741,
      "training_step_time": 9.070383310317993
    },
    {
      "epoch": 1.5012741088867188e-05,
      "model_forward_time": 1.715317726135254,
      "step": 15742
    },
    {
      "epoch": 1.5012741088867188e-05,
      "step": 15742,
      "training_step_time": 8.91447138786316
    },
    {
      "epoch": 1.5013694763183594e-05,
      "model_forward_time": 2.278177499771118,
      "step": 15743
    },
    {
      "epoch": 1.5013694763183594e-05,
      "step": 15743,
      "training_step_time": 7.817577362060547
    },
    {
      "epoch": 1.50146484375e-05,
      "model_forward_time": 1.7183539867401123,
      "step": 15744
    },
    {
      "epoch": 1.50146484375e-05,
      "step": 15744,
      "training_step_time": 7.705135822296143
    },
    {
      "epoch": 1.5015602111816406e-05,
      "model_forward_time": 1.715682029724121,
      "step": 15745
    },
    {
      "epoch": 1.5015602111816406e-05,
      "step": 15745,
      "training_step_time": 6.861740589141846
    },
    {
      "epoch": 1.5016555786132813e-05,
      "model_forward_time": 1.7180993556976318,
      "step": 15746
    },
    {
      "epoch": 1.5016555786132813e-05,
      "step": 15746,
      "training_step_time": 6.860779523849487
    },
    {
      "epoch": 1.5017509460449219e-05,
      "model_forward_time": 1.721264123916626,
      "step": 15747
    },
    {
      "epoch": 1.5017509460449219e-05,
      "step": 15747,
      "training_step_time": 6.87175440788269
    },
    {
      "epoch": 1.5018463134765624e-05,
      "model_forward_time": 1.7237815856933594,
      "step": 15748
    },
    {
      "epoch": 1.5018463134765624e-05,
      "step": 15748,
      "training_step_time": 6.831539869308472
    },
    {
      "epoch": 1.5019416809082032e-05,
      "model_forward_time": 1.7203760147094727,
      "step": 15749
    },
    {
      "epoch": 1.5019416809082032e-05,
      "step": 15749,
      "training_step_time": 6.850008010864258
    },
    {
      "epoch": 1.5020370483398438e-05,
      "grad_norm": 0.2163112908601761,
      "learning_rate": 9.687427371887293e-06,
      "loss": 0.0645,
      "step": 15750
    },
    {
      "epoch": 1.5020370483398438e-05,
      "model_forward_time": 1.7127461433410645,
      "step": 15750
    },
    {
      "dynamics_loss_avg": 0.06292342115193605,
      "epoch": 1.5020370483398438e-05,
      "step": 15750
    },
    {
      "action_loss_avg": 0.0027419743477366866,
      "epoch": 1.5020370483398438e-05,
      "step": 15750
    },
    {
      "epoch": 1.5020370483398438e-05,
      "step": 15750,
      "training_step_time": 7.74384617805481
    },
    {
      "epoch": 1.5021324157714843e-05,
      "model_forward_time": 1.7241878509521484,
      "step": 15751
    },
    {
      "epoch": 1.5021324157714843e-05,
      "step": 15751,
      "training_step_time": 9.482797622680664
    },
    {
      "epoch": 1.502227783203125e-05,
      "model_forward_time": 2.2732021808624268,
      "step": 15752
    },
    {
      "epoch": 1.502227783203125e-05,
      "step": 15752,
      "training_step_time": 8.842513799667358
    },
    {
      "epoch": 1.5023231506347656e-05,
      "model_forward_time": 1.7207543849945068,
      "step": 15753
    },
    {
      "epoch": 1.5023231506347656e-05,
      "step": 15753,
      "training_step_time": 7.775502681732178
    },
    {
      "epoch": 1.5024185180664062e-05,
      "model_forward_time": 1.7230429649353027,
      "step": 15754
    },
    {
      "epoch": 1.5024185180664062e-05,
      "step": 15754,
      "training_step_time": 6.916037321090698
    },
    {
      "epoch": 1.502513885498047e-05,
      "model_forward_time": 1.7223241329193115,
      "step": 15755
    },
    {
      "epoch": 1.502513885498047e-05,
      "step": 15755,
      "training_step_time": 6.903168201446533
    },
    {
      "epoch": 1.5026092529296875e-05,
      "model_forward_time": 1.7175002098083496,
      "step": 15756
    },
    {
      "epoch": 1.5026092529296875e-05,
      "step": 15756,
      "training_step_time": 6.905169486999512
    },
    {
      "epoch": 1.5027046203613281e-05,
      "model_forward_time": 1.7232134342193604,
      "step": 15757
    },
    {
      "epoch": 1.5027046203613281e-05,
      "step": 15757,
      "training_step_time": 6.833827495574951
    },
    {
      "epoch": 1.5027999877929687e-05,
      "model_forward_time": 1.722970724105835,
      "step": 15758
    },
    {
      "epoch": 1.5027999877929687e-05,
      "step": 15758,
      "training_step_time": 6.80619740486145
    },
    {
      "epoch": 1.5028953552246094e-05,
      "model_forward_time": 1.7162959575653076,
      "step": 15759
    },
    {
      "epoch": 1.5028953552246094e-05,
      "step": 15759,
      "training_step_time": 7.706023216247559
    },
    {
      "epoch": 1.50299072265625e-05,
      "grad_norm": 0.18583180010318756,
      "learning_rate": 9.686851667746013e-06,
      "loss": 0.0655,
      "step": 15760
    },
    {
      "epoch": 1.50299072265625e-05,
      "model_forward_time": 1.7216577529907227,
      "step": 15760
    },
    {
      "dynamics_loss_avg": 0.07190257925540208,
      "epoch": 1.50299072265625e-05,
      "step": 15760
    },
    {
      "action_loss_avg": 0.004330225917510688,
      "epoch": 1.50299072265625e-05,
      "step": 15760
    },
    {
      "epoch": 1.50299072265625e-05,
      "step": 15760,
      "training_step_time": 7.767043352127075
    },
    {
      "epoch": 1.5030860900878906e-05,
      "model_forward_time": 2.2660484313964844,
      "step": 15761
    },
    {
      "epoch": 1.5030860900878906e-05,
      "step": 15761,
      "training_step_time": 9.386269569396973
    },
    {
      "epoch": 1.5031814575195313e-05,
      "model_forward_time": 1.721440076828003,
      "step": 15762
    },
    {
      "epoch": 1.5031814575195313e-05,
      "step": 15762,
      "training_step_time": 8.437937259674072
    },
    {
      "epoch": 1.5032768249511719e-05,
      "model_forward_time": 1.3725621700286865,
      "step": 15763
    },
    {
      "epoch": 1.5032768249511719e-05,
      "step": 15763,
      "training_step_time": 7.599552154541016
    },
    {
      "epoch": 1.5033721923828124e-05,
      "model_forward_time": 1.7314057350158691,
      "step": 15764
    },
    {
      "epoch": 1.5033721923828124e-05,
      "step": 15764,
      "training_step_time": 6.849687099456787
    },
    {
      "epoch": 1.5034675598144532e-05,
      "model_forward_time": 1.7368206977844238,
      "step": 15765
    },
    {
      "epoch": 1.5034675598144532e-05,
      "step": 15765,
      "training_step_time": 6.8907997608184814
    },
    {
      "epoch": 1.5035629272460938e-05,
      "model_forward_time": 1.7073619365692139,
      "step": 15766
    },
    {
      "epoch": 1.5035629272460938e-05,
      "step": 15766,
      "training_step_time": 6.873222589492798
    },
    {
      "epoch": 1.5036582946777343e-05,
      "model_forward_time": 1.722010850906372,
      "step": 15767
    },
    {
      "epoch": 1.5036582946777343e-05,
      "step": 15767,
      "training_step_time": 6.89225959777832
    },
    {
      "epoch": 1.503753662109375e-05,
      "model_forward_time": 1.3550353050231934,
      "step": 15768
    },
    {
      "epoch": 1.503753662109375e-05,
      "step": 15768,
      "training_step_time": 8.542737007141113
    },
    {
      "epoch": 1.5038490295410156e-05,
      "model_forward_time": 1.7260725498199463,
      "step": 15769
    },
    {
      "epoch": 1.5038490295410156e-05,
      "step": 15769,
      "training_step_time": 7.8479015827178955
    },
    {
      "epoch": 1.5039443969726562e-05,
      "grad_norm": 0.22569678723812103,
      "learning_rate": 9.686275451057678e-06,
      "loss": 0.0655,
      "step": 15770
    },
    {
      "epoch": 1.5039443969726562e-05,
      "model_forward_time": 1.7435917854309082,
      "step": 15770
    },
    {
      "dynamics_loss_avg": 0.06869396194815636,
      "epoch": 1.5039443969726562e-05,
      "step": 15770
    },
    {
      "action_loss_avg": 0.0037942908762488513,
      "epoch": 1.5039443969726562e-05,
      "step": 15770
    },
    {
      "epoch": 1.5039443969726562e-05,
      "step": 15770,
      "training_step_time": 10.857954263687134
    },
    {
      "epoch": 1.504039764404297e-05,
      "model_forward_time": 2.2599868774414062,
      "step": 15771
    },
    {
      "epoch": 1.504039764404297e-05,
      "step": 15771,
      "training_step_time": 9.616700410842896
    },
    {
      "epoch": 1.5041351318359375e-05,
      "model_forward_time": 1.720496654510498,
      "step": 15772
    },
    {
      "epoch": 1.5041351318359375e-05,
      "step": 15772,
      "training_step_time": 7.701414585113525
    },
    {
      "epoch": 1.5042304992675781e-05,
      "model_forward_time": 1.7180941104888916,
      "step": 15773
    },
    {
      "epoch": 1.5042304992675781e-05,
      "step": 15773,
      "training_step_time": 6.881843566894531
    },
    {
      "epoch": 1.5043258666992187e-05,
      "model_forward_time": 1.717750072479248,
      "step": 15774
    },
    {
      "epoch": 1.5043258666992187e-05,
      "step": 15774,
      "training_step_time": 6.803574085235596
    },
    {
      "epoch": 1.5044212341308594e-05,
      "model_forward_time": 1.7235612869262695,
      "step": 15775
    },
    {
      "epoch": 1.5044212341308594e-05,
      "step": 15775,
      "training_step_time": 6.743261098861694
    },
    {
      "epoch": 1.5045166015625e-05,
      "model_forward_time": 1.355811595916748,
      "step": 15776
    },
    {
      "epoch": 1.5045166015625e-05,
      "step": 15776,
      "training_step_time": 6.790168523788452
    },
    {
      "epoch": 1.5046119689941406e-05,
      "model_forward_time": 1.7290165424346924,
      "step": 15777
    },
    {
      "epoch": 1.5046119689941406e-05,
      "step": 15777,
      "training_step_time": 7.591019153594971
    },
    {
      "epoch": 1.5047073364257813e-05,
      "model_forward_time": 1.7027082443237305,
      "step": 15778
    },
    {
      "epoch": 1.5047073364257813e-05,
      "step": 15778,
      "training_step_time": 7.719763994216919
    },
    {
      "epoch": 1.5048027038574219e-05,
      "model_forward_time": 1.7212352752685547,
      "step": 15779
    },
    {
      "epoch": 1.5048027038574219e-05,
      "step": 15779,
      "training_step_time": 8.629186153411865
    },
    {
      "epoch": 1.5048980712890624e-05,
      "grad_norm": 0.13371552526950836,
      "learning_rate": 9.685698721885309e-06,
      "loss": 0.0653,
      "step": 15780
    },
    {
      "epoch": 1.5048980712890624e-05,
      "model_forward_time": 2.2623510360717773,
      "step": 15780
    },
    {
      "dynamics_loss_avg": 0.05480420310050249,
      "epoch": 1.5048980712890624e-05,
      "step": 15780
    },
    {
      "action_loss_avg": 0.002461938466876745,
      "epoch": 1.5048980712890624e-05,
      "step": 15780
    },
    {
      "epoch": 1.5048980712890624e-05,
      "step": 15780,
      "training_step_time": 7.936273813247681
    },
    {
      "epoch": 1.5049934387207032e-05,
      "model_forward_time": 1.7316839694976807,
      "step": 15781
    },
    {
      "epoch": 1.5049934387207032e-05,
      "step": 15781,
      "training_step_time": 8.66266417503357
    },
    {
      "epoch": 1.5050888061523438e-05,
      "model_forward_time": 1.7184474468231201,
      "step": 15782
    },
    {
      "epoch": 1.5050888061523438e-05,
      "step": 15782,
      "training_step_time": 7.773390293121338
    },
    {
      "epoch": 1.5051841735839843e-05,
      "model_forward_time": 1.7209293842315674,
      "step": 15783
    },
    {
      "epoch": 1.5051841735839843e-05,
      "step": 15783,
      "training_step_time": 6.900379657745361
    },
    {
      "epoch": 1.505279541015625e-05,
      "model_forward_time": 1.728053331375122,
      "step": 15784
    },
    {
      "epoch": 1.505279541015625e-05,
      "step": 15784,
      "training_step_time": 6.892351150512695
    },
    {
      "epoch": 1.5053749084472656e-05,
      "model_forward_time": 1.721928358078003,
      "step": 15785
    },
    {
      "epoch": 1.5053749084472656e-05,
      "step": 15785,
      "training_step_time": 6.756594181060791
    },
    {
      "epoch": 1.5054702758789062e-05,
      "model_forward_time": 1.723454236984253,
      "step": 15786
    },
    {
      "epoch": 1.5054702758789062e-05,
      "step": 15786,
      "training_step_time": 7.808950185775757
    },
    {
      "epoch": 1.505565643310547e-05,
      "model_forward_time": 1.7095959186553955,
      "step": 15787
    },
    {
      "epoch": 1.505565643310547e-05,
      "step": 15787,
      "training_step_time": 7.867351055145264
    },
    {
      "epoch": 1.5056610107421875e-05,
      "model_forward_time": 1.7249035835266113,
      "step": 15788
    },
    {
      "epoch": 1.5056610107421875e-05,
      "step": 15788,
      "training_step_time": 8.60347843170166
    },
    {
      "epoch": 1.5057563781738281e-05,
      "model_forward_time": 2.2943458557128906,
      "step": 15789
    },
    {
      "epoch": 1.5057563781738281e-05,
      "step": 15789,
      "training_step_time": 8.936006546020508
    },
    {
      "epoch": 1.5058517456054687e-05,
      "grad_norm": 0.14348915219306946,
      "learning_rate": 9.68512148029197e-06,
      "loss": 0.0614,
      "step": 15790
    },
    {
      "epoch": 1.5058517456054687e-05,
      "model_forward_time": 1.7241785526275635,
      "step": 15790
    },
    {
      "dynamics_loss_avg": 0.06141712944954634,
      "epoch": 1.5058517456054687e-05,
      "step": 15790
    },
    {
      "action_loss_avg": 0.002767395693808794,
      "epoch": 1.5058517456054687e-05,
      "step": 15790
    },
    {
      "epoch": 1.5058517456054687e-05,
      "step": 15790,
      "training_step_time": 7.935960054397583
    },
    {
      "epoch": 1.5059471130371094e-05,
      "model_forward_time": 1.7188234329223633,
      "step": 15791
    },
    {
      "epoch": 1.5059471130371094e-05,
      "step": 15791,
      "training_step_time": 7.660280466079712
    },
    {
      "epoch": 1.50604248046875e-05,
      "model_forward_time": 1.7209374904632568,
      "step": 15792
    },
    {
      "epoch": 1.50604248046875e-05,
      "step": 15792,
      "training_step_time": 6.844827890396118
    },
    {
      "epoch": 1.5061378479003906e-05,
      "model_forward_time": 1.721353530883789,
      "step": 15793
    },
    {
      "epoch": 1.5061378479003906e-05,
      "step": 15793,
      "training_step_time": 6.900996685028076
    },
    {
      "epoch": 1.5062332153320313e-05,
      "model_forward_time": 1.7348108291625977,
      "step": 15794
    },
    {
      "epoch": 1.5062332153320313e-05,
      "step": 15794,
      "training_step_time": 6.963430404663086
    },
    {
      "epoch": 1.5063285827636719e-05,
      "model_forward_time": 1.7207434177398682,
      "step": 15795
    },
    {
      "epoch": 1.5063285827636719e-05,
      "step": 15795,
      "training_step_time": 7.780411720275879
    },
    {
      "epoch": 1.5064239501953124e-05,
      "model_forward_time": 1.7393319606781006,
      "step": 15796
    },
    {
      "epoch": 1.5064239501953124e-05,
      "step": 15796,
      "training_step_time": 8.338709592819214
    },
    {
      "epoch": 1.5065193176269532e-05,
      "model_forward_time": 1.7225184440612793,
      "step": 15797
    },
    {
      "epoch": 1.5065193176269532e-05,
      "step": 15797,
      "training_step_time": 9.71500825881958
    },
    {
      "epoch": 1.5066146850585938e-05,
      "model_forward_time": 2.2916290760040283,
      "step": 15798
    },
    {
      "epoch": 1.5066146850585938e-05,
      "step": 15798,
      "training_step_time": 8.334795236587524
    },
    {
      "epoch": 1.5067100524902343e-05,
      "model_forward_time": 1.7344017028808594,
      "step": 15799
    },
    {
      "epoch": 1.5067100524902343e-05,
      "step": 15799,
      "training_step_time": 9.275134086608887
    },
    {
      "epoch": 1.506805419921875e-05,
      "grad_norm": 0.1793927550315857,
      "learning_rate": 9.68454372634079e-06,
      "loss": 0.0666,
      "step": 15800
    },
    {
      "epoch": 1.506805419921875e-05,
      "model_forward_time": 1.7418897151947021,
      "step": 15800
    },
    {
      "dynamics_loss_avg": 0.05865310709923506,
      "epoch": 1.506805419921875e-05,
      "step": 15800
    },
    {
      "action_loss_avg": 0.0031290118815377353,
      "epoch": 1.506805419921875e-05,
      "step": 15800
    },
    {
      "epoch": 1.506805419921875e-05,
      "step": 15800,
      "training_step_time": 8.31344723701477
    },
    {
      "epoch": 1.5069007873535156e-05,
      "model_forward_time": 1.7336981296539307,
      "step": 15801
    },
    {
      "epoch": 1.5069007873535156e-05,
      "step": 15801,
      "training_step_time": 6.803686618804932
    },
    {
      "epoch": 1.5069961547851562e-05,
      "model_forward_time": 1.7363803386688232,
      "step": 15802
    },
    {
      "epoch": 1.5069961547851562e-05,
      "step": 15802,
      "training_step_time": 6.807123422622681
    },
    {
      "epoch": 1.507091522216797e-05,
      "model_forward_time": 1.736185073852539,
      "step": 15803
    },
    {
      "epoch": 1.507091522216797e-05,
      "step": 15803,
      "training_step_time": 6.807324647903442
    },
    {
      "epoch": 1.5071868896484375e-05,
      "model_forward_time": 1.7457563877105713,
      "step": 15804
    },
    {
      "epoch": 1.5071868896484375e-05,
      "step": 15804,
      "training_step_time": 7.604822158813477
    },
    {
      "epoch": 1.5072822570800781e-05,
      "model_forward_time": 1.731731653213501,
      "step": 15805
    },
    {
      "epoch": 1.5072822570800781e-05,
      "step": 15805,
      "training_step_time": 8.456209659576416
    },
    {
      "epoch": 1.5073776245117187e-05,
      "model_forward_time": 1.7327699661254883,
      "step": 15806
    },
    {
      "epoch": 1.5073776245117187e-05,
      "step": 15806,
      "training_step_time": 8.93370532989502
    },
    {
      "epoch": 1.5074729919433594e-05,
      "model_forward_time": 2.283022880554199,
      "step": 15807
    },
    {
      "epoch": 1.5074729919433594e-05,
      "step": 15807,
      "training_step_time": 11.494255304336548
    },
    {
      "epoch": 1.507568359375e-05,
      "model_forward_time": 1.7443349361419678,
      "step": 15808
    },
    {
      "epoch": 1.507568359375e-05,
      "step": 15808,
      "training_step_time": 7.978607654571533
    },
    {
      "epoch": 1.5076637268066406e-05,
      "model_forward_time": 1.7325525283813477,
      "step": 15809
    },
    {
      "epoch": 1.5076637268066406e-05,
      "step": 15809,
      "training_step_time": 7.638327121734619
    },
    {
      "epoch": 1.5077590942382813e-05,
      "grad_norm": 0.20563584566116333,
      "learning_rate": 9.683965460094953e-06,
      "loss": 0.0659,
      "step": 15810
    },
    {
      "epoch": 1.5077590942382813e-05,
      "model_forward_time": 1.739217758178711,
      "step": 15810
    },
    {
      "dynamics_loss_avg": 0.06461559124290943,
      "epoch": 1.5077590942382813e-05,
      "step": 15810
    },
    {
      "action_loss_avg": 0.003197660151636228,
      "epoch": 1.5077590942382813e-05,
      "step": 15810
    },
    {
      "epoch": 1.5077590942382813e-05,
      "step": 15810,
      "training_step_time": 7.779385566711426
    },
    {
      "epoch": 1.5078544616699219e-05,
      "model_forward_time": 1.7333383560180664,
      "step": 15811
    },
    {
      "epoch": 1.5078544616699219e-05,
      "step": 15811,
      "training_step_time": 6.900332689285278
    },
    {
      "epoch": 1.5079498291015624e-05,
      "model_forward_time": 1.7319371700286865,
      "step": 15812
    },
    {
      "epoch": 1.5079498291015624e-05,
      "step": 15812,
      "training_step_time": 6.916954278945923
    },
    {
      "epoch": 1.5080451965332032e-05,
      "model_forward_time": 1.724501132965088,
      "step": 15813
    },
    {
      "epoch": 1.5080451965332032e-05,
      "step": 15813,
      "training_step_time": 7.743654727935791
    },
    {
      "epoch": 1.5081405639648438e-05,
      "model_forward_time": 1.724555492401123,
      "step": 15814
    },
    {
      "epoch": 1.5081405639648438e-05,
      "step": 15814,
      "training_step_time": 9.027715682983398
    },
    {
      "epoch": 1.5082359313964843e-05,
      "model_forward_time": 1.3493738174438477,
      "step": 15815
    },
    {
      "epoch": 1.5082359313964843e-05,
      "step": 15815,
      "training_step_time": 9.257202863693237
    },
    {
      "epoch": 1.508331298828125e-05,
      "model_forward_time": 1.743624210357666,
      "step": 15816
    },
    {
      "epoch": 1.508331298828125e-05,
      "step": 15816,
      "training_step_time": 8.583922386169434
    },
    {
      "epoch": 1.5084266662597656e-05,
      "model_forward_time": 2.263904333114624,
      "step": 15817
    },
    {
      "epoch": 1.5084266662597656e-05,
      "step": 15817,
      "training_step_time": 7.982684373855591
    },
    {
      "epoch": 1.5085220336914062e-05,
      "model_forward_time": 1.741614818572998,
      "step": 15818
    },
    {
      "epoch": 1.5085220336914062e-05,
      "step": 15818,
      "training_step_time": 7.594066143035889
    },
    {
      "epoch": 1.508617401123047e-05,
      "model_forward_time": 1.7304065227508545,
      "step": 15819
    },
    {
      "epoch": 1.508617401123047e-05,
      "step": 15819,
      "training_step_time": 7.731757640838623
    },
    {
      "epoch": 1.5087127685546875e-05,
      "grad_norm": 0.2946147322654724,
      "learning_rate": 9.683386681617695e-06,
      "loss": 0.0652,
      "step": 15820
    },
    {
      "epoch": 1.5087127685546875e-05,
      "model_forward_time": 1.7264244556427002,
      "step": 15820
    },
    {
      "dynamics_loss_avg": 0.05401646364480257,
      "epoch": 1.5087127685546875e-05,
      "step": 15820
    },
    {
      "action_loss_avg": 0.0033721481217071415,
      "epoch": 1.5087127685546875e-05,
      "step": 15820
    },
    {
      "epoch": 1.5087127685546875e-05,
      "step": 15820,
      "training_step_time": 6.943640947341919
    },
    {
      "epoch": 1.5088081359863281e-05,
      "model_forward_time": 1.7422709465026855,
      "step": 15821
    },
    {
      "epoch": 1.5088081359863281e-05,
      "step": 15821,
      "training_step_time": 6.947788238525391
    },
    {
      "epoch": 1.5089035034179687e-05,
      "model_forward_time": 1.7280473709106445,
      "step": 15822
    },
    {
      "epoch": 1.5089035034179687e-05,
      "step": 15822,
      "training_step_time": 7.450818061828613
    },
    {
      "epoch": 1.5089988708496094e-05,
      "model_forward_time": 1.7274534702301025,
      "step": 15823
    },
    {
      "epoch": 1.5089988708496094e-05,
      "step": 15823,
      "training_step_time": 8.342184066772461
    },
    {
      "epoch": 1.50909423828125e-05,
      "model_forward_time": 1.7294037342071533,
      "step": 15824
    },
    {
      "epoch": 1.50909423828125e-05,
      "step": 15824,
      "training_step_time": 7.898034334182739
    },
    {
      "epoch": 1.5091896057128906e-05,
      "model_forward_time": 1.7069668769836426,
      "step": 15825
    },
    {
      "epoch": 1.5091896057128906e-05,
      "step": 15825,
      "training_step_time": 8.413299560546875
    },
    {
      "epoch": 1.5092849731445313e-05,
      "model_forward_time": 2.2646586894989014,
      "step": 15826
    },
    {
      "epoch": 1.5092849731445313e-05,
      "step": 15826,
      "training_step_time": 8.431796312332153
    },
    {
      "epoch": 1.5093803405761719e-05,
      "model_forward_time": 1.734724998474121,
      "step": 15827
    },
    {
      "epoch": 1.5093803405761719e-05,
      "step": 15827,
      "training_step_time": 7.855274677276611
    },
    {
      "epoch": 1.5094757080078124e-05,
      "model_forward_time": 1.7273881435394287,
      "step": 15828
    },
    {
      "epoch": 1.5094757080078124e-05,
      "step": 15828,
      "training_step_time": 7.622748374938965
    },
    {
      "epoch": 1.5095710754394532e-05,
      "model_forward_time": 1.732264518737793,
      "step": 15829
    },
    {
      "epoch": 1.5095710754394532e-05,
      "step": 15829,
      "training_step_time": 6.915386915206909
    },
    {
      "epoch": 1.5096664428710938e-05,
      "grad_norm": 0.26055708527565,
      "learning_rate": 9.68280739097231e-06,
      "loss": 0.0668,
      "step": 15830
    },
    {
      "epoch": 1.5096664428710938e-05,
      "model_forward_time": 1.7422804832458496,
      "step": 15830
    },
    {
      "dynamics_loss_avg": 0.05194260068237781,
      "epoch": 1.5096664428710938e-05,
      "step": 15830
    },
    {
      "action_loss_avg": 0.003903439361602068,
      "epoch": 1.5096664428710938e-05,
      "step": 15830
    },
    {
      "epoch": 1.5096664428710938e-05,
      "step": 15830,
      "training_step_time": 6.920824289321899
    },
    {
      "epoch": 1.5097618103027343e-05,
      "model_forward_time": 1.7282490730285645,
      "step": 15831
    },
    {
      "epoch": 1.5097618103027343e-05,
      "step": 15831,
      "training_step_time": 6.898519992828369
    },
    {
      "epoch": 1.509857177734375e-05,
      "model_forward_time": 1.7246453762054443,
      "step": 15832
    },
    {
      "epoch": 1.509857177734375e-05,
      "step": 15832,
      "training_step_time": 7.8893444538116455
    },
    {
      "epoch": 1.5099525451660156e-05,
      "model_forward_time": 1.7139623165130615,
      "step": 15833
    },
    {
      "epoch": 1.5099525451660156e-05,
      "step": 15833,
      "training_step_time": 7.861334800720215
    },
    {
      "epoch": 1.5100479125976562e-05,
      "model_forward_time": 1.7252857685089111,
      "step": 15834
    },
    {
      "epoch": 1.5100479125976562e-05,
      "step": 15834,
      "training_step_time": 8.770493507385254
    },
    {
      "epoch": 1.510143280029297e-05,
      "model_forward_time": 2.285439968109131,
      "step": 15835
    },
    {
      "epoch": 1.510143280029297e-05,
      "step": 15835,
      "training_step_time": 8.868405818939209
    },
    {
      "epoch": 1.5102386474609375e-05,
      "model_forward_time": 1.7430686950683594,
      "step": 15836
    },
    {
      "epoch": 1.5102386474609375e-05,
      "step": 15836,
      "training_step_time": 7.879759073257446
    },
    {
      "epoch": 1.5103340148925781e-05,
      "model_forward_time": 1.7154066562652588,
      "step": 15837
    },
    {
      "epoch": 1.5103340148925781e-05,
      "step": 15837,
      "training_step_time": 8.334925174713135
    },
    {
      "epoch": 1.5104293823242187e-05,
      "model_forward_time": 1.7194032669067383,
      "step": 15838
    },
    {
      "epoch": 1.5104293823242187e-05,
      "step": 15838,
      "training_step_time": 7.473043441772461
    },
    {
      "epoch": 1.5105247497558594e-05,
      "model_forward_time": 1.7168233394622803,
      "step": 15839
    },
    {
      "epoch": 1.5105247497558594e-05,
      "step": 15839,
      "training_step_time": 6.9569618701934814
    },
    {
      "epoch": 1.5106201171875e-05,
      "grad_norm": 0.2065059244632721,
      "learning_rate": 9.682227588222148e-06,
      "loss": 0.0632,
      "step": 15840
    },
    {
      "epoch": 1.5106201171875e-05,
      "model_forward_time": 1.7151808738708496,
      "step": 15840
    },
    {
      "dynamics_loss_avg": 0.0607430474832654,
      "epoch": 1.5106201171875e-05,
      "step": 15840
    },
    {
      "action_loss_avg": 0.002811232378007844,
      "epoch": 1.5106201171875e-05,
      "step": 15840
    },
    {
      "epoch": 1.5106201171875e-05,
      "step": 15840,
      "training_step_time": 6.906906604766846
    },
    {
      "epoch": 1.5107154846191406e-05,
      "model_forward_time": 1.7105190753936768,
      "step": 15841
    },
    {
      "epoch": 1.5107154846191406e-05,
      "step": 15841,
      "training_step_time": 8.69058609008789
    },
    {
      "epoch": 1.5108108520507813e-05,
      "model_forward_time": 1.7099802494049072,
      "step": 15842
    },
    {
      "epoch": 1.5108108520507813e-05,
      "step": 15842,
      "training_step_time": 8.529973268508911
    },
    {
      "epoch": 1.5109062194824219e-05,
      "model_forward_time": 1.3530385494232178,
      "step": 15843
    },
    {
      "epoch": 1.5109062194824219e-05,
      "step": 15843,
      "training_step_time": 9.512600421905518
    },
    {
      "epoch": 1.5110015869140624e-05,
      "model_forward_time": 2.284775495529175,
      "step": 15844
    },
    {
      "epoch": 1.5110015869140624e-05,
      "step": 15844,
      "training_step_time": 9.34455418586731
    },
    {
      "epoch": 1.5110969543457032e-05,
      "model_forward_time": 1.718604564666748,
      "step": 15845
    },
    {
      "epoch": 1.5110969543457032e-05,
      "step": 15845,
      "training_step_time": 7.833244562149048
    },
    {
      "epoch": 1.5111923217773438e-05,
      "model_forward_time": 1.7233433723449707,
      "step": 15846
    },
    {
      "epoch": 1.5111923217773438e-05,
      "step": 15846,
      "training_step_time": 7.6928870677948
    },
    {
      "epoch": 1.5112876892089843e-05,
      "model_forward_time": 1.7173125743865967,
      "step": 15847
    },
    {
      "epoch": 1.5112876892089843e-05,
      "step": 15847,
      "training_step_time": 7.681154251098633
    },
    {
      "epoch": 1.511383056640625e-05,
      "model_forward_time": 1.7138574123382568,
      "step": 15848
    },
    {
      "epoch": 1.511383056640625e-05,
      "step": 15848,
      "training_step_time": 6.804565668106079
    },
    {
      "epoch": 1.5114784240722656e-05,
      "model_forward_time": 1.7258210182189941,
      "step": 15849
    },
    {
      "epoch": 1.5114784240722656e-05,
      "step": 15849,
      "training_step_time": 6.826331377029419
    },
    {
      "epoch": 1.5115737915039062e-05,
      "grad_norm": 0.14657969772815704,
      "learning_rate": 9.681647273430618e-06,
      "loss": 0.0638,
      "step": 15850
    },
    {
      "epoch": 1.5115737915039062e-05,
      "model_forward_time": 1.7390186786651611,
      "step": 15850
    },
    {
      "dynamics_loss_avg": 0.07253624647855758,
      "epoch": 1.5115737915039062e-05,
      "step": 15850
    },
    {
      "action_loss_avg": 0.003427206666674465,
      "epoch": 1.5115737915039062e-05,
      "step": 15850
    },
    {
      "epoch": 1.5115737915039062e-05,
      "step": 15850,
      "training_step_time": 7.692019462585449
    },
    {
      "epoch": 1.511669158935547e-05,
      "model_forward_time": 1.719512939453125,
      "step": 15851
    },
    {
      "epoch": 1.511669158935547e-05,
      "step": 15851,
      "training_step_time": 9.548198938369751
    },
    {
      "epoch": 1.5117645263671875e-05,
      "model_forward_time": 1.7210242748260498,
      "step": 15852
    },
    {
      "epoch": 1.5117645263671875e-05,
      "step": 15852,
      "training_step_time": 9.272965669631958
    },
    {
      "epoch": 1.5118598937988281e-05,
      "model_forward_time": 1.7424414157867432,
      "step": 15853
    },
    {
      "epoch": 1.5118598937988281e-05,
      "step": 15853,
      "training_step_time": 7.8587257862091064
    },
    {
      "epoch": 1.5119552612304687e-05,
      "model_forward_time": 2.269359588623047,
      "step": 15854
    },
    {
      "epoch": 1.5119552612304687e-05,
      "step": 15854,
      "training_step_time": 8.864882946014404
    },
    {
      "epoch": 1.5120506286621094e-05,
      "model_forward_time": 1.7194547653198242,
      "step": 15855
    },
    {
      "epoch": 1.5120506286621094e-05,
      "step": 15855,
      "training_step_time": 7.527696847915649
    },
    {
      "epoch": 1.51214599609375e-05,
      "model_forward_time": 1.7174270153045654,
      "step": 15856
    },
    {
      "epoch": 1.51214599609375e-05,
      "step": 15856,
      "training_step_time": 7.7097063064575195
    },
    {
      "epoch": 1.5122413635253906e-05,
      "model_forward_time": 1.716848611831665,
      "step": 15857
    },
    {
      "epoch": 1.5122413635253906e-05,
      "step": 15857,
      "training_step_time": 6.782062292098999
    },
    {
      "epoch": 1.5123367309570313e-05,
      "model_forward_time": 1.7114543914794922,
      "step": 15858
    },
    {
      "epoch": 1.5123367309570313e-05,
      "step": 15858,
      "training_step_time": 6.853872776031494
    },
    {
      "epoch": 1.5124320983886719e-05,
      "model_forward_time": 1.7145178318023682,
      "step": 15859
    },
    {
      "epoch": 1.5124320983886719e-05,
      "step": 15859,
      "training_step_time": 7.5937113761901855
    },
    {
      "epoch": 1.5125274658203124e-05,
      "grad_norm": 0.14564257860183716,
      "learning_rate": 9.681066446661183e-06,
      "loss": 0.0663,
      "step": 15860
    },
    {
      "epoch": 1.5125274658203124e-05,
      "model_forward_time": 1.717538595199585,
      "step": 15860
    },
    {
      "dynamics_loss_avg": 0.06225163750350475,
      "epoch": 1.5125274658203124e-05,
      "step": 15860
    },
    {
      "action_loss_avg": 0.0032672270550392567,
      "epoch": 1.5125274658203124e-05,
      "step": 15860
    },
    {
      "epoch": 1.5125274658203124e-05,
      "step": 15860,
      "training_step_time": 7.691158056259155
    },
    {
      "epoch": 1.5126228332519532e-05,
      "model_forward_time": 1.7218835353851318,
      "step": 15861
    },
    {
      "epoch": 1.5126228332519532e-05,
      "step": 15861,
      "training_step_time": 9.222586631774902
    },
    {
      "epoch": 1.5127182006835938e-05,
      "model_forward_time": 1.7215359210968018,
      "step": 15862
    },
    {
      "epoch": 1.5127182006835938e-05,
      "step": 15862,
      "training_step_time": 7.837321758270264
    },
    {
      "epoch": 1.5128135681152343e-05,
      "model_forward_time": 2.2364284992218018,
      "step": 15863
    },
    {
      "epoch": 1.5128135681152343e-05,
      "step": 15863,
      "training_step_time": 10.081551313400269
    },
    {
      "epoch": 1.512908935546875e-05,
      "model_forward_time": 1.7243645191192627,
      "step": 15864
    },
    {
      "epoch": 1.512908935546875e-05,
      "step": 15864,
      "training_step_time": 8.720261335372925
    },
    {
      "epoch": 1.5130043029785156e-05,
      "model_forward_time": 1.718576192855835,
      "step": 15865
    },
    {
      "epoch": 1.5130043029785156e-05,
      "step": 15865,
      "training_step_time": 7.654768705368042
    },
    {
      "epoch": 1.5130996704101562e-05,
      "model_forward_time": 1.716815710067749,
      "step": 15866
    },
    {
      "epoch": 1.5130996704101562e-05,
      "step": 15866,
      "training_step_time": 7.39885687828064
    },
    {
      "epoch": 1.513195037841797e-05,
      "model_forward_time": 1.7254304885864258,
      "step": 15867
    },
    {
      "epoch": 1.513195037841797e-05,
      "step": 15867,
      "training_step_time": 6.88881516456604
    },
    {
      "epoch": 1.5132904052734375e-05,
      "model_forward_time": 1.7537732124328613,
      "step": 15868
    },
    {
      "epoch": 1.5132904052734375e-05,
      "step": 15868,
      "training_step_time": 7.6060097217559814
    },
    {
      "epoch": 1.5133857727050781e-05,
      "model_forward_time": 1.7310431003570557,
      "step": 15869
    },
    {
      "epoch": 1.5133857727050781e-05,
      "step": 15869,
      "training_step_time": 7.6865315437316895
    },
    {
      "epoch": 1.5134811401367187e-05,
      "grad_norm": 0.15746907889842987,
      "learning_rate": 9.680485107977357e-06,
      "loss": 0.06,
      "step": 15870
    },
    {
      "epoch": 1.5134811401367187e-05,
      "model_forward_time": 1.7367689609527588,
      "step": 15870
    },
    {
      "dynamics_loss_avg": 0.05842580422759056,
      "epoch": 1.5134811401367187e-05,
      "step": 15870
    },
    {
      "action_loss_avg": 0.002882011129986495,
      "epoch": 1.5134811401367187e-05,
      "step": 15870
    },
    {
      "epoch": 1.5134811401367187e-05,
      "step": 15870,
      "training_step_time": 8.760007381439209
    },
    {
      "epoch": 1.5135765075683594e-05,
      "model_forward_time": 1.7311930656433105,
      "step": 15871
    },
    {
      "epoch": 1.5135765075683594e-05,
      "step": 15871,
      "training_step_time": 8.453687906265259
    },
    {
      "epoch": 1.513671875e-05,
      "model_forward_time": 2.294830083847046,
      "step": 15872
    },
    {
      "epoch": 1.513671875e-05,
      "step": 15872,
      "training_step_time": 9.689659357070923
    },
    {
      "epoch": 1.5137672424316406e-05,
      "model_forward_time": 1.7277295589447021,
      "step": 15873
    },
    {
      "epoch": 1.5137672424316406e-05,
      "step": 15873,
      "training_step_time": 8.517708539962769
    },
    {
      "epoch": 1.5138626098632813e-05,
      "model_forward_time": 1.7331552505493164,
      "step": 15874
    },
    {
      "epoch": 1.5138626098632813e-05,
      "step": 15874,
      "training_step_time": 7.756662130355835
    },
    {
      "epoch": 1.5139579772949219e-05,
      "model_forward_time": 1.7237789630889893,
      "step": 15875
    },
    {
      "epoch": 1.5139579772949219e-05,
      "step": 15875,
      "training_step_time": 7.696769952774048
    },
    {
      "epoch": 1.5140533447265624e-05,
      "model_forward_time": 1.7356653213500977,
      "step": 15876
    },
    {
      "epoch": 1.5140533447265624e-05,
      "step": 15876,
      "training_step_time": 6.9009175300598145
    },
    {
      "epoch": 1.5141487121582032e-05,
      "model_forward_time": 1.7304809093475342,
      "step": 15877
    },
    {
      "epoch": 1.5141487121582032e-05,
      "step": 15877,
      "training_step_time": 7.08647894859314
    },
    {
      "epoch": 1.5142440795898438e-05,
      "model_forward_time": 1.73403000831604,
      "step": 15878
    },
    {
      "epoch": 1.5142440795898438e-05,
      "step": 15878,
      "training_step_time": 7.847909927368164
    },
    {
      "epoch": 1.5143394470214843e-05,
      "model_forward_time": 1.7308650016784668,
      "step": 15879
    },
    {
      "epoch": 1.5143394470214843e-05,
      "step": 15879,
      "training_step_time": 8.780467510223389
    },
    {
      "epoch": 1.514434814453125e-05,
      "grad_norm": 0.1425141841173172,
      "learning_rate": 9.679903257442715e-06,
      "loss": 0.0626,
      "step": 15880
    },
    {
      "epoch": 1.514434814453125e-05,
      "model_forward_time": 1.0104475021362305,
      "step": 15880
    },
    {
      "dynamics_loss_avg": 0.07330487500876189,
      "epoch": 1.514434814453125e-05,
      "step": 15880
    },
    {
      "action_loss_avg": 0.0032179855625145137,
      "epoch": 1.514434814453125e-05,
      "step": 15880
    },
    {
      "epoch": 1.514434814453125e-05,
      "step": 15880,
      "training_step_time": 9.38927674293518
    },
    {
      "epoch": 1.5145301818847656e-05,
      "model_forward_time": 1.7788465023040771,
      "step": 15881
    },
    {
      "epoch": 1.5145301818847656e-05,
      "step": 15881,
      "training_step_time": 11.651844501495361
    },
    {
      "epoch": 1.5146255493164062e-05,
      "model_forward_time": 1.7349920272827148,
      "step": 15882
    },
    {
      "epoch": 1.5146255493164062e-05,
      "step": 15882,
      "training_step_time": 7.901263952255249
    },
    {
      "epoch": 1.5147209167480468e-05,
      "model_forward_time": 1.7273612022399902,
      "step": 15883
    },
    {
      "epoch": 1.5147209167480468e-05,
      "step": 15883,
      "training_step_time": 7.71861457824707
    },
    {
      "epoch": 1.5148162841796875e-05,
      "model_forward_time": 1.7398686408996582,
      "step": 15884
    },
    {
      "epoch": 1.5148162841796875e-05,
      "step": 15884,
      "training_step_time": 7.790538787841797
    },
    {
      "epoch": 1.5149116516113281e-05,
      "model_forward_time": 1.7347073554992676,
      "step": 15885
    },
    {
      "epoch": 1.5149116516113281e-05,
      "step": 15885,
      "training_step_time": 6.97136664390564
    },
    {
      "epoch": 1.5150070190429687e-05,
      "model_forward_time": 1.7304983139038086,
      "step": 15886
    },
    {
      "epoch": 1.5150070190429687e-05,
      "step": 15886,
      "training_step_time": 7.092066287994385
    },
    {
      "epoch": 1.5151023864746094e-05,
      "model_forward_time": 1.7384727001190186,
      "step": 15887
    },
    {
      "epoch": 1.5151023864746094e-05,
      "step": 15887,
      "training_step_time": 7.7307727336883545
    },
    {
      "epoch": 1.51519775390625e-05,
      "model_forward_time": 1.730781078338623,
      "step": 15888
    },
    {
      "epoch": 1.51519775390625e-05,
      "step": 15888,
      "training_step_time": 7.700098276138306
    },
    {
      "epoch": 1.5152931213378906e-05,
      "model_forward_time": 1.7341361045837402,
      "step": 15889
    },
    {
      "epoch": 1.5152931213378906e-05,
      "step": 15889,
      "training_step_time": 9.514178037643433
    },
    {
      "epoch": 1.5153884887695313e-05,
      "grad_norm": 0.15029087662696838,
      "learning_rate": 9.679320895120892e-06,
      "loss": 0.0614,
      "step": 15890
    },
    {
      "epoch": 1.5153884887695313e-05,
      "model_forward_time": 1.7757744789123535,
      "step": 15890
    },
    {
      "dynamics_loss_avg": 0.06151199322193861,
      "epoch": 1.5153884887695313e-05,
      "step": 15890
    },
    {
      "action_loss_avg": 0.002824180817697197,
      "epoch": 1.5153884887695313e-05,
      "step": 15890
    },
    {
      "epoch": 1.5153884887695313e-05,
      "step": 15890,
      "training_step_time": 9.204934120178223
    },
    {
      "epoch": 1.5154838562011719e-05,
      "model_forward_time": 2.31169056892395,
      "step": 15891
    },
    {
      "epoch": 1.5154838562011719e-05,
      "step": 15891,
      "training_step_time": 7.72100567817688
    },
    {
      "epoch": 1.5155792236328124e-05,
      "model_forward_time": 1.7399108409881592,
      "step": 15892
    },
    {
      "epoch": 1.5155792236328124e-05,
      "step": 15892,
      "training_step_time": 7.450499773025513
    },
    {
      "epoch": 1.5156745910644532e-05,
      "model_forward_time": 1.7368474006652832,
      "step": 15893
    },
    {
      "epoch": 1.5156745910644532e-05,
      "step": 15893,
      "training_step_time": 7.528377056121826
    },
    {
      "epoch": 1.5157699584960938e-05,
      "model_forward_time": 1.7325034141540527,
      "step": 15894
    },
    {
      "epoch": 1.5157699584960938e-05,
      "step": 15894,
      "training_step_time": 7.322396516799927
    },
    {
      "epoch": 1.5158653259277343e-05,
      "model_forward_time": 1.7267718315124512,
      "step": 15895
    },
    {
      "epoch": 1.5158653259277343e-05,
      "step": 15895,
      "training_step_time": 6.950769901275635
    },
    {
      "epoch": 1.515960693359375e-05,
      "model_forward_time": 1.7367057800292969,
      "step": 15896
    },
    {
      "epoch": 1.515960693359375e-05,
      "step": 15896,
      "training_step_time": 7.658702373504639
    },
    {
      "epoch": 1.5160560607910156e-05,
      "model_forward_time": 1.703564167022705,
      "step": 15897
    },
    {
      "epoch": 1.5160560607910156e-05,
      "step": 15897,
      "training_step_time": 7.606950044631958
    },
    {
      "epoch": 1.5161514282226562e-05,
      "model_forward_time": 1.716430902481079,
      "step": 15898
    },
    {
      "epoch": 1.5161514282226562e-05,
      "step": 15898,
      "training_step_time": 8.898283243179321
    },
    {
      "epoch": 1.5162467956542968e-05,
      "model_forward_time": 1.738128423690796,
      "step": 15899
    },
    {
      "epoch": 1.5162467956542968e-05,
      "step": 15899,
      "training_step_time": 8.461873292922974
    },
    {
      "epoch": 1.5163421630859375e-05,
      "grad_norm": 0.18558242917060852,
      "learning_rate": 9.67873802107557e-06,
      "loss": 0.0699,
      "step": 15900
    },
    {
      "epoch": 1.5163421630859375e-05,
      "model_forward_time": 2.259761333465576,
      "step": 15900
    },
    {
      "dynamics_loss_avg": 0.05584017285145819,
      "epoch": 1.5163421630859375e-05,
      "step": 15900
    },
    {
      "action_loss_avg": 0.0023426980420481413,
      "epoch": 1.5163421630859375e-05,
      "step": 15900
    },
    {
      "epoch": 1.5163421630859375e-05,
      "step": 15900,
      "training_step_time": 7.725917339324951
    },
    {
      "epoch": 1.5164375305175781e-05,
      "model_forward_time": 1.7285161018371582,
      "step": 15901
    },
    {
      "epoch": 1.5164375305175781e-05,
      "step": 15901,
      "training_step_time": 7.632179498672485
    },
    {
      "epoch": 1.5165328979492187e-05,
      "model_forward_time": 1.72745680809021,
      "step": 15902
    },
    {
      "epoch": 1.5165328979492187e-05,
      "step": 15902,
      "training_step_time": 7.706705808639526
    },
    {
      "epoch": 1.5166282653808594e-05,
      "model_forward_time": 1.7210543155670166,
      "step": 15903
    },
    {
      "epoch": 1.5166282653808594e-05,
      "step": 15903,
      "training_step_time": 7.6220033168792725
    },
    {
      "epoch": 1.5167236328125e-05,
      "model_forward_time": 1.7201569080352783,
      "step": 15904
    },
    {
      "epoch": 1.5167236328125e-05,
      "step": 15904,
      "training_step_time": 6.9978861808776855
    },
    {
      "epoch": 1.5168190002441406e-05,
      "model_forward_time": 1.7330334186553955,
      "step": 15905
    },
    {
      "epoch": 1.5168190002441406e-05,
      "step": 15905,
      "training_step_time": 7.681136131286621
    },
    {
      "epoch": 1.5169143676757813e-05,
      "model_forward_time": 1.3436951637268066,
      "step": 15906
    },
    {
      "epoch": 1.5169143676757813e-05,
      "step": 15906,
      "training_step_time": 7.652533054351807
    },
    {
      "epoch": 1.5170097351074219e-05,
      "model_forward_time": 1.732032299041748,
      "step": 15907
    },
    {
      "epoch": 1.5170097351074219e-05,
      "step": 15907,
      "training_step_time": 9.269941091537476
    },
    {
      "epoch": 1.5171051025390624e-05,
      "model_forward_time": 1.720930576324463,
      "step": 15908
    },
    {
      "epoch": 1.5171051025390624e-05,
      "step": 15908,
      "training_step_time": 7.681817054748535
    },
    {
      "epoch": 1.5172004699707032e-05,
      "model_forward_time": 2.274413585662842,
      "step": 15909
    },
    {
      "epoch": 1.5172004699707032e-05,
      "step": 15909,
      "training_step_time": 7.759614706039429
    },
    {
      "epoch": 1.5172958374023438e-05,
      "grad_norm": 0.16669069230556488,
      "learning_rate": 9.678154635370492e-06,
      "loss": 0.0688,
      "step": 15910
    },
    {
      "epoch": 1.5172958374023438e-05,
      "model_forward_time": 1.7251253128051758,
      "step": 15910
    },
    {
      "dynamics_loss_avg": 0.05744206327944994,
      "epoch": 1.5172958374023438e-05,
      "step": 15910
    },
    {
      "action_loss_avg": 0.0030153188388794662,
      "epoch": 1.5172958374023438e-05,
      "step": 15910
    },
    {
      "epoch": 1.5172958374023438e-05,
      "step": 15910,
      "training_step_time": 7.640549659729004
    },
    {
      "epoch": 1.5173912048339843e-05,
      "model_forward_time": 1.7272288799285889,
      "step": 15911
    },
    {
      "epoch": 1.5173912048339843e-05,
      "step": 15911,
      "training_step_time": 7.631181955337524
    },
    {
      "epoch": 1.517486572265625e-05,
      "model_forward_time": 1.7364542484283447,
      "step": 15912
    },
    {
      "epoch": 1.517486572265625e-05,
      "step": 15912,
      "training_step_time": 7.61666464805603
    },
    {
      "epoch": 1.5175819396972656e-05,
      "model_forward_time": 1.7329747676849365,
      "step": 15913
    },
    {
      "epoch": 1.5175819396972656e-05,
      "step": 15913,
      "training_step_time": 7.3674328327178955
    },
    {
      "epoch": 1.5176773071289062e-05,
      "model_forward_time": 1.736440896987915,
      "step": 15914
    },
    {
      "epoch": 1.5176773071289062e-05,
      "step": 15914,
      "training_step_time": 7.610407829284668
    },
    {
      "epoch": 1.5177726745605468e-05,
      "model_forward_time": 1.7294793128967285,
      "step": 15915
    },
    {
      "epoch": 1.5177726745605468e-05,
      "step": 15915,
      "training_step_time": 7.657465219497681
    },
    {
      "epoch": 1.5178680419921875e-05,
      "model_forward_time": 1.7406761646270752,
      "step": 15916
    },
    {
      "epoch": 1.5178680419921875e-05,
      "step": 15916,
      "training_step_time": 8.724935531616211
    },
    {
      "epoch": 1.5179634094238281e-05,
      "model_forward_time": 1.724888801574707,
      "step": 15917
    },
    {
      "epoch": 1.5179634094238281e-05,
      "step": 15917,
      "training_step_time": 7.7702412605285645
    },
    {
      "epoch": 1.5180587768554687e-05,
      "model_forward_time": 2.3015406131744385,
      "step": 15918
    },
    {
      "epoch": 1.5180587768554687e-05,
      "step": 15918,
      "training_step_time": 7.8421630859375
    },
    {
      "epoch": 1.5181541442871094e-05,
      "model_forward_time": 1.7300739288330078,
      "step": 15919
    },
    {
      "epoch": 1.5181541442871094e-05,
      "step": 15919,
      "training_step_time": 7.694593906402588
    },
    {
      "epoch": 1.51824951171875e-05,
      "grad_norm": 0.1897202730178833,
      "learning_rate": 9.677570738069457e-06,
      "loss": 0.0671,
      "step": 15920
    },
    {
      "epoch": 1.51824951171875e-05,
      "model_forward_time": 1.7307610511779785,
      "step": 15920
    },
    {
      "dynamics_loss_avg": 0.05173183511942625,
      "epoch": 1.51824951171875e-05,
      "step": 15920
    },
    {
      "action_loss_avg": 0.002226697816513479,
      "epoch": 1.51824951171875e-05,
      "step": 15920
    },
    {
      "epoch": 1.51824951171875e-05,
      "step": 15920,
      "training_step_time": 7.523668050765991
    },
    {
      "epoch": 1.5183448791503906e-05,
      "model_forward_time": 1.7249412536621094,
      "step": 15921
    },
    {
      "epoch": 1.5183448791503906e-05,
      "step": 15921,
      "training_step_time": 7.689400911331177
    },
    {
      "epoch": 1.5184402465820313e-05,
      "model_forward_time": 1.7374331951141357,
      "step": 15922
    },
    {
      "epoch": 1.5184402465820313e-05,
      "step": 15922,
      "training_step_time": 7.441832780838013
    },
    {
      "epoch": 1.5185356140136719e-05,
      "model_forward_time": 1.732893705368042,
      "step": 15923
    },
    {
      "epoch": 1.5185356140136719e-05,
      "step": 15923,
      "training_step_time": 7.4807305335998535
    },
    {
      "epoch": 1.5186309814453125e-05,
      "model_forward_time": 1.7333238124847412,
      "step": 15924
    },
    {
      "epoch": 1.5186309814453125e-05,
      "step": 15924,
      "training_step_time": 7.734658718109131
    },
    {
      "epoch": 1.5187263488769532e-05,
      "model_forward_time": 1.7279939651489258,
      "step": 15925
    },
    {
      "epoch": 1.5187263488769532e-05,
      "step": 15925,
      "training_step_time": 9.561091899871826
    },
    {
      "epoch": 1.5188217163085938e-05,
      "model_forward_time": 1.7387504577636719,
      "step": 15926
    },
    {
      "epoch": 1.5188217163085938e-05,
      "step": 15926,
      "training_step_time": 8.567067861557007
    },
    {
      "epoch": 1.5189170837402343e-05,
      "model_forward_time": 1.755079984664917,
      "step": 15927
    },
    {
      "epoch": 1.5189170837402343e-05,
      "step": 15927,
      "training_step_time": 9.417090654373169
    },
    {
      "epoch": 1.519012451171875e-05,
      "model_forward_time": 2.26511549949646,
      "step": 15928
    },
    {
      "epoch": 1.519012451171875e-05,
      "step": 15928,
      "training_step_time": 8.753728151321411
    },
    {
      "epoch": 1.5191078186035157e-05,
      "model_forward_time": 1.7423312664031982,
      "step": 15929
    },
    {
      "epoch": 1.5191078186035157e-05,
      "step": 15929,
      "training_step_time": 7.635244846343994
    },
    {
      "epoch": 1.5192031860351562e-05,
      "grad_norm": 0.2556036114692688,
      "learning_rate": 9.676986329236319e-06,
      "loss": 0.0712,
      "step": 15930
    },
    {
      "epoch": 1.5192031860351562e-05,
      "model_forward_time": 1.7199177742004395,
      "step": 15930
    },
    {
      "dynamics_loss_avg": 0.0528147354722023,
      "epoch": 1.5192031860351562e-05,
      "step": 15930
    },
    {
      "action_loss_avg": 0.004591162397991866,
      "epoch": 1.5192031860351562e-05,
      "step": 15930
    },
    {
      "epoch": 1.5192031860351562e-05,
      "step": 15930,
      "training_step_time": 7.59123158454895
    },
    {
      "epoch": 1.5192985534667968e-05,
      "model_forward_time": 1.7389929294586182,
      "step": 15931
    },
    {
      "epoch": 1.5192985534667968e-05,
      "step": 15931,
      "training_step_time": 7.365995407104492
    },
    {
      "epoch": 1.5193939208984375e-05,
      "model_forward_time": 1.7276887893676758,
      "step": 15932
    },
    {
      "epoch": 1.5193939208984375e-05,
      "step": 15932,
      "training_step_time": 7.399867534637451
    },
    {
      "epoch": 1.5194892883300781e-05,
      "model_forward_time": 1.7301859855651855,
      "step": 15933
    },
    {
      "epoch": 1.5194892883300781e-05,
      "step": 15933,
      "training_step_time": 9.498663902282715
    },
    {
      "epoch": 1.5195846557617187e-05,
      "model_forward_time": 1.7431106567382812,
      "step": 15934
    },
    {
      "epoch": 1.5195846557617187e-05,
      "step": 15934,
      "training_step_time": 8.496497631072998
    },
    {
      "epoch": 1.5196800231933594e-05,
      "model_forward_time": 1.3473398685455322,
      "step": 15935
    },
    {
      "epoch": 1.5196800231933594e-05,
      "step": 15935,
      "training_step_time": 9.121241331100464
    },
    {
      "epoch": 1.519775390625e-05,
      "model_forward_time": 1.7399444580078125,
      "step": 15936
    },
    {
      "epoch": 1.519775390625e-05,
      "step": 15936,
      "training_step_time": 8.427441358566284
    },
    {
      "epoch": 1.5198707580566406e-05,
      "model_forward_time": 2.2739803791046143,
      "step": 15937
    },
    {
      "epoch": 1.5198707580566406e-05,
      "step": 15937,
      "training_step_time": 8.499474287033081
    },
    {
      "epoch": 1.5199661254882813e-05,
      "model_forward_time": 1.7461562156677246,
      "step": 15938
    },
    {
      "epoch": 1.5199661254882813e-05,
      "step": 15938,
      "training_step_time": 7.902297735214233
    },
    {
      "epoch": 1.5200614929199219e-05,
      "model_forward_time": 1.7348124980926514,
      "step": 15939
    },
    {
      "epoch": 1.5200614929199219e-05,
      "step": 15939,
      "training_step_time": 8.340567588806152
    },
    {
      "epoch": 1.5201568603515625e-05,
      "grad_norm": 0.24747641384601593,
      "learning_rate": 9.676401408934988e-06,
      "loss": 0.0674,
      "step": 15940
    },
    {
      "epoch": 1.5201568603515625e-05,
      "model_forward_time": 1.7440736293792725,
      "step": 15940
    },
    {
      "dynamics_loss_avg": 0.07118982188403607,
      "epoch": 1.5201568603515625e-05,
      "step": 15940
    },
    {
      "action_loss_avg": 0.0033203113707713783,
      "epoch": 1.5201568603515625e-05,
      "step": 15940
    },
    {
      "epoch": 1.5201568603515625e-05,
      "step": 15940,
      "training_step_time": 7.688344955444336
    },
    {
      "epoch": 1.5202522277832032e-05,
      "model_forward_time": 1.740058183670044,
      "step": 15941
    },
    {
      "epoch": 1.5202522277832032e-05,
      "step": 15941,
      "training_step_time": 7.620106220245361
    },
    {
      "epoch": 1.5203475952148438e-05,
      "model_forward_time": 1.7457482814788818,
      "step": 15942
    },
    {
      "epoch": 1.5203475952148438e-05,
      "step": 15942,
      "training_step_time": 7.65604567527771
    },
    {
      "epoch": 1.5204429626464843e-05,
      "model_forward_time": 1.7360734939575195,
      "step": 15943
    },
    {
      "epoch": 1.5204429626464843e-05,
      "step": 15943,
      "training_step_time": 8.80479097366333
    },
    {
      "epoch": 1.520538330078125e-05,
      "model_forward_time": 1.7449088096618652,
      "step": 15944
    },
    {
      "epoch": 1.520538330078125e-05,
      "step": 15944,
      "training_step_time": 9.315274715423584
    },
    {
      "epoch": 1.5206336975097657e-05,
      "model_forward_time": 1.7458808422088623,
      "step": 15945
    },
    {
      "epoch": 1.5206336975097657e-05,
      "step": 15945,
      "training_step_time": 7.6602561473846436
    },
    {
      "epoch": 1.5207290649414062e-05,
      "model_forward_time": 2.2981667518615723,
      "step": 15946
    },
    {
      "epoch": 1.5207290649414062e-05,
      "step": 15946,
      "training_step_time": 7.759549379348755
    },
    {
      "epoch": 1.5208244323730468e-05,
      "model_forward_time": 1.747354507446289,
      "step": 15947
    },
    {
      "epoch": 1.5208244323730468e-05,
      "step": 15947,
      "training_step_time": 7.7523276805877686
    },
    {
      "epoch": 1.5209197998046875e-05,
      "model_forward_time": 1.7465438842773438,
      "step": 15948
    },
    {
      "epoch": 1.5209197998046875e-05,
      "step": 15948,
      "training_step_time": 8.353789806365967
    },
    {
      "epoch": 1.5210151672363281e-05,
      "model_forward_time": 1.7443370819091797,
      "step": 15949
    },
    {
      "epoch": 1.5210151672363281e-05,
      "step": 15949,
      "training_step_time": 7.579174041748047
    },
    {
      "epoch": 1.5211105346679687e-05,
      "grad_norm": 0.23281246423721313,
      "learning_rate": 9.675815977229429e-06,
      "loss": 0.0666,
      "step": 15950
    },
    {
      "epoch": 1.5211105346679687e-05,
      "model_forward_time": 1.742384433746338,
      "step": 15950
    },
    {
      "dynamics_loss_avg": 0.06751507725566626,
      "epoch": 1.5211105346679687e-05,
      "step": 15950
    },
    {
      "action_loss_avg": 0.0031508784391917287,
      "epoch": 1.5211105346679687e-05,
      "step": 15950
    },
    {
      "epoch": 1.5211105346679687e-05,
      "step": 15950,
      "training_step_time": 7.323284864425659
    },
    {
      "epoch": 1.5212059020996094e-05,
      "model_forward_time": 1.7416107654571533,
      "step": 15951
    },
    {
      "epoch": 1.5212059020996094e-05,
      "step": 15951,
      "training_step_time": 7.6281349658966064
    },
    {
      "epoch": 1.52130126953125e-05,
      "model_forward_time": 1.7295918464660645,
      "step": 15952
    },
    {
      "epoch": 1.52130126953125e-05,
      "step": 15952,
      "training_step_time": 7.624819755554199
    },
    {
      "epoch": 1.5213966369628906e-05,
      "model_forward_time": 1.747277021408081,
      "step": 15953
    },
    {
      "epoch": 1.5213966369628906e-05,
      "step": 15953,
      "training_step_time": 8.492693901062012
    },
    {
      "epoch": 1.5214920043945313e-05,
      "model_forward_time": 1.746412992477417,
      "step": 15954
    },
    {
      "epoch": 1.5214920043945313e-05,
      "step": 15954,
      "training_step_time": 8.508383750915527
    },
    {
      "epoch": 1.5215873718261719e-05,
      "model_forward_time": 2.3036580085754395,
      "step": 15955
    },
    {
      "epoch": 1.5215873718261719e-05,
      "step": 15955,
      "training_step_time": 7.757545709609985
    },
    {
      "epoch": 1.5216827392578125e-05,
      "model_forward_time": 1.750530481338501,
      "step": 15956
    },
    {
      "epoch": 1.5216827392578125e-05,
      "step": 15956,
      "training_step_time": 7.811101198196411
    },
    {
      "epoch": 1.5217781066894532e-05,
      "model_forward_time": 1.747854471206665,
      "step": 15957
    },
    {
      "epoch": 1.5217781066894532e-05,
      "step": 15957,
      "training_step_time": 7.821923732757568
    },
    {
      "epoch": 1.5218734741210938e-05,
      "model_forward_time": 1.7525157928466797,
      "step": 15958
    },
    {
      "epoch": 1.5218734741210938e-05,
      "step": 15958,
      "training_step_time": 7.636023998260498
    },
    {
      "epoch": 1.5219688415527343e-05,
      "model_forward_time": 1.7438170909881592,
      "step": 15959
    },
    {
      "epoch": 1.5219688415527343e-05,
      "step": 15959,
      "training_step_time": 7.3463335037231445
    },
    {
      "epoch": 1.522064208984375e-05,
      "grad_norm": 0.25738978385925293,
      "learning_rate": 9.675230034183665e-06,
      "loss": 0.0661,
      "step": 15960
    },
    {
      "epoch": 1.522064208984375e-05,
      "model_forward_time": 1.7421231269836426,
      "step": 15960
    },
    {
      "dynamics_loss_avg": 0.05894485656172037,
      "epoch": 1.522064208984375e-05,
      "step": 15960
    },
    {
      "action_loss_avg": 0.0027412494295276702,
      "epoch": 1.522064208984375e-05,
      "step": 15960
    },
    {
      "epoch": 1.522064208984375e-05,
      "step": 15960,
      "training_step_time": 7.6312267780303955
    },
    {
      "epoch": 1.5221595764160157e-05,
      "model_forward_time": 1.7492506504058838,
      "step": 15961
    },
    {
      "epoch": 1.5221595764160157e-05,
      "step": 15961,
      "training_step_time": 8.85866928100586
    },
    {
      "epoch": 1.5222549438476562e-05,
      "model_forward_time": 1.7452454566955566,
      "step": 15962
    },
    {
      "epoch": 1.5222549438476562e-05,
      "step": 15962,
      "training_step_time": 7.843261241912842
    },
    {
      "epoch": 1.5223503112792968e-05,
      "model_forward_time": 1.7508735656738281,
      "step": 15963
    },
    {
      "epoch": 1.5223503112792968e-05,
      "step": 15963,
      "training_step_time": 8.537914752960205
    },
    {
      "epoch": 1.5224456787109375e-05,
      "model_forward_time": 2.2828452587127686,
      "step": 15964
    },
    {
      "epoch": 1.5224456787109375e-05,
      "step": 15964,
      "training_step_time": 9.759737968444824
    },
    {
      "epoch": 1.5225410461425781e-05,
      "model_forward_time": 1.7419459819793701,
      "step": 15965
    },
    {
      "epoch": 1.5225410461425781e-05,
      "step": 15965,
      "training_step_time": 7.621073007583618
    },
    {
      "epoch": 1.5226364135742187e-05,
      "model_forward_time": 1.7447330951690674,
      "step": 15966
    },
    {
      "epoch": 1.5226364135742187e-05,
      "step": 15966,
      "training_step_time": 7.708611488342285
    },
    {
      "epoch": 1.5227317810058594e-05,
      "model_forward_time": 1.7380573749542236,
      "step": 15967
    },
    {
      "epoch": 1.5227317810058594e-05,
      "step": 15967,
      "training_step_time": 7.562839031219482
    },
    {
      "epoch": 1.5228271484375e-05,
      "model_forward_time": 1.7428953647613525,
      "step": 15968
    },
    {
      "epoch": 1.5228271484375e-05,
      "step": 15968,
      "training_step_time": 7.307825565338135
    },
    {
      "epoch": 1.5229225158691406e-05,
      "model_forward_time": 1.7382845878601074,
      "step": 15969
    },
    {
      "epoch": 1.5229225158691406e-05,
      "step": 15969,
      "training_step_time": 7.640724420547485
    },
    {
      "epoch": 1.5230178833007813e-05,
      "grad_norm": 0.14827503263950348,
      "learning_rate": 9.674643579861772e-06,
      "loss": 0.0696,
      "step": 15970
    },
    {
      "epoch": 1.5230178833007813e-05,
      "model_forward_time": 1.7369763851165771,
      "step": 15970
    },
    {
      "dynamics_loss_avg": 0.058027603290975095,
      "epoch": 1.5230178833007813e-05,
      "step": 15970
    },
    {
      "action_loss_avg": 0.0019674120587296782,
      "epoch": 1.5230178833007813e-05,
      "step": 15970
    },
    {
      "epoch": 1.5230178833007813e-05,
      "step": 15970,
      "training_step_time": 8.39495062828064
    },
    {
      "epoch": 1.5231132507324219e-05,
      "model_forward_time": 1.7083356380462646,
      "step": 15971
    },
    {
      "epoch": 1.5231132507324219e-05,
      "step": 15971,
      "training_step_time": 9.014092683792114
    },
    {
      "epoch": 1.5232086181640625e-05,
      "model_forward_time": 1.7386767864227295,
      "step": 15972
    },
    {
      "epoch": 1.5232086181640625e-05,
      "step": 15972,
      "training_step_time": 9.688508987426758
    },
    {
      "epoch": 1.5233039855957032e-05,
      "model_forward_time": 1.7742795944213867,
      "step": 15973
    },
    {
      "epoch": 1.5233039855957032e-05,
      "step": 15973,
      "training_step_time": 9.522272109985352
    },
    {
      "epoch": 1.5233993530273438e-05,
      "model_forward_time": 1.7794601917266846,
      "step": 15974
    },
    {
      "epoch": 1.5233993530273438e-05,
      "step": 15974,
      "training_step_time": 7.653367280960083
    },
    {
      "epoch": 1.5234947204589843e-05,
      "model_forward_time": 1.747663974761963,
      "step": 15975
    },
    {
      "epoch": 1.5234947204589843e-05,
      "step": 15975,
      "training_step_time": 7.70524263381958
    },
    {
      "epoch": 1.523590087890625e-05,
      "model_forward_time": 1.7402026653289795,
      "step": 15976
    },
    {
      "epoch": 1.523590087890625e-05,
      "step": 15976,
      "training_step_time": 7.697445392608643
    },
    {
      "epoch": 1.5236854553222657e-05,
      "model_forward_time": 1.7444238662719727,
      "step": 15977
    },
    {
      "epoch": 1.5236854553222657e-05,
      "step": 15977,
      "training_step_time": 7.460529804229736
    },
    {
      "epoch": 1.5237808227539062e-05,
      "model_forward_time": 1.750293254852295,
      "step": 15978
    },
    {
      "epoch": 1.5237808227539062e-05,
      "step": 15978,
      "training_step_time": 7.6223859786987305
    },
    {
      "epoch": 1.5238761901855468e-05,
      "model_forward_time": 1.7446236610412598,
      "step": 15979
    },
    {
      "epoch": 1.5238761901855468e-05,
      "step": 15979,
      "training_step_time": 7.592671155929565
    },
    {
      "epoch": 1.5239715576171875e-05,
      "grad_norm": 0.1858387589454651,
      "learning_rate": 9.674056614327886e-06,
      "loss": 0.0695,
      "step": 15980
    },
    {
      "epoch": 1.5239715576171875e-05,
      "model_forward_time": 1.733149528503418,
      "step": 15980
    },
    {
      "dynamics_loss_avg": 0.06383802806958556,
      "epoch": 1.5239715576171875e-05,
      "step": 15980
    },
    {
      "action_loss_avg": 0.0023855684587033464,
      "epoch": 1.5239715576171875e-05,
      "step": 15980
    },
    {
      "epoch": 1.5239715576171875e-05,
      "step": 15980,
      "training_step_time": 9.579952239990234
    },
    {
      "epoch": 1.5240669250488281e-05,
      "model_forward_time": 1.7353103160858154,
      "step": 15981
    },
    {
      "epoch": 1.5240669250488281e-05,
      "step": 15981,
      "training_step_time": 9.37875771522522
    },
    {
      "epoch": 1.5241622924804687e-05,
      "model_forward_time": 1.7311224937438965,
      "step": 15982
    },
    {
      "epoch": 1.5241622924804687e-05,
      "step": 15982,
      "training_step_time": 8.739426374435425
    },
    {
      "epoch": 1.5242576599121094e-05,
      "model_forward_time": 2.3321008682250977,
      "step": 15983
    },
    {
      "epoch": 1.5242576599121094e-05,
      "step": 15983,
      "training_step_time": 7.630982160568237
    },
    {
      "epoch": 1.52435302734375e-05,
      "model_forward_time": 1.7369410991668701,
      "step": 15984
    },
    {
      "epoch": 1.52435302734375e-05,
      "step": 15984,
      "training_step_time": 7.677489995956421
    },
    {
      "epoch": 1.5244483947753906e-05,
      "model_forward_time": 1.747788667678833,
      "step": 15985
    },
    {
      "epoch": 1.5244483947753906e-05,
      "step": 15985,
      "training_step_time": 7.7521984577178955
    },
    {
      "epoch": 1.5245437622070313e-05,
      "model_forward_time": 1.7421379089355469,
      "step": 15986
    },
    {
      "epoch": 1.5245437622070313e-05,
      "step": 15986,
      "training_step_time": 7.5831379890441895
    },
    {
      "epoch": 1.5246391296386719e-05,
      "model_forward_time": 1.7354636192321777,
      "step": 15987
    },
    {
      "epoch": 1.5246391296386719e-05,
      "step": 15987,
      "training_step_time": 7.622184991836548
    },
    {
      "epoch": 1.5247344970703125e-05,
      "model_forward_time": 1.718214750289917,
      "step": 15988
    },
    {
      "epoch": 1.5247344970703125e-05,
      "step": 15988,
      "training_step_time": 7.373614311218262
    },
    {
      "epoch": 1.5248298645019532e-05,
      "model_forward_time": 1.733128547668457,
      "step": 15989
    },
    {
      "epoch": 1.5248298645019532e-05,
      "step": 15989,
      "training_step_time": 7.64175820350647
    },
    {
      "epoch": 1.5249252319335938e-05,
      "grad_norm": 0.1783958077430725,
      "learning_rate": 9.673469137646198e-06,
      "loss": 0.0682,
      "step": 15990
    },
    {
      "epoch": 1.5249252319335938e-05,
      "model_forward_time": 1.7291054725646973,
      "step": 15990
    },
    {
      "dynamics_loss_avg": 0.08437666893005372,
      "epoch": 1.5249252319335938e-05,
      "step": 15990
    },
    {
      "action_loss_avg": 0.0031391900614835323,
      "epoch": 1.5249252319335938e-05,
      "step": 15990
    },
    {
      "epoch": 1.5249252319335938e-05,
      "step": 15990,
      "training_step_time": 8.795852899551392
    },
    {
      "epoch": 1.5250205993652343e-05,
      "model_forward_time": 1.3642027378082275,
      "step": 15991
    },
    {
      "epoch": 1.5250205993652343e-05,
      "step": 15991,
      "training_step_time": 8.216594934463501
    },
    {
      "epoch": 1.525115966796875e-05,
      "model_forward_time": 2.3072128295898438,
      "step": 15992
    },
    {
      "epoch": 1.525115966796875e-05,
      "step": 15992,
      "training_step_time": 7.738796949386597
    },
    {
      "epoch": 1.5252113342285157e-05,
      "model_forward_time": 1.7325026988983154,
      "step": 15993
    },
    {
      "epoch": 1.5252113342285157e-05,
      "step": 15993,
      "training_step_time": 7.899233102798462
    },
    {
      "epoch": 1.5253067016601562e-05,
      "model_forward_time": 1.7159123420715332,
      "step": 15994
    },
    {
      "epoch": 1.5253067016601562e-05,
      "step": 15994,
      "training_step_time": 7.7156548500061035
    },
    {
      "epoch": 1.5254020690917968e-05,
      "model_forward_time": 1.724005937576294,
      "step": 15995
    },
    {
      "epoch": 1.5254020690917968e-05,
      "step": 15995,
      "training_step_time": 7.604078769683838
    },
    {
      "epoch": 1.5254974365234375e-05,
      "model_forward_time": 1.7178568840026855,
      "step": 15996
    },
    {
      "epoch": 1.5254974365234375e-05,
      "step": 15996,
      "training_step_time": 7.690636157989502
    },
    {
      "epoch": 1.5255928039550781e-05,
      "model_forward_time": 1.7383534908294678,
      "step": 15997
    },
    {
      "epoch": 1.5255928039550781e-05,
      "step": 15997,
      "training_step_time": 7.4456610679626465
    },
    {
      "epoch": 1.5256881713867187e-05,
      "model_forward_time": 1.7669785022735596,
      "step": 15998
    },
    {
      "epoch": 1.5256881713867187e-05,
      "step": 15998,
      "training_step_time": 7.988638639450073
    },
    {
      "epoch": 1.5257835388183594e-05,
      "model_forward_time": 1.7749197483062744,
      "step": 15999
    },
    {
      "epoch": 1.5257835388183594e-05,
      "step": 15999,
      "training_step_time": 7.876724004745483
    },
    {
      "epoch": 1.52587890625e-05,
      "grad_norm": 0.19500839710235596,
      "learning_rate": 9.67288114988095e-06,
      "loss": 0.0669,
      "step": 16000
    },
    {
      "epoch": 1.52587890625e-05,
      "model_forward_time": 2.6912026405334473,
      "step": 16000
    },
    {
      "dynamics_loss_avg": 0.06401766054332256,
      "epoch": 1.52587890625e-05,
      "step": 16000
    },
    {
      "action_loss_avg": 0.002935985568910837,
      "epoch": 1.52587890625e-05,
      "step": 16000
    },
    {
      "epoch": 1.52587890625e-05,
      "step": 16000,
      "training_step_time": 10.536903381347656
    },
    {
      "epoch": 1.5259742736816407e-05,
      "model_forward_time": 1.735741138458252,
      "step": 16001
    },
    {
      "epoch": 1.5259742736816407e-05,
      "step": 16001,
      "training_step_time": 9.782276153564453
    },
    {
      "epoch": 1.526069641113281e-05,
      "model_forward_time": 1.7339060306549072,
      "step": 16002
    },
    {
      "epoch": 1.526069641113281e-05,
      "step": 16002,
      "training_step_time": 7.749139785766602
    },
    {
      "epoch": 1.526165008544922e-05,
      "model_forward_time": 1.7421844005584717,
      "step": 16003
    },
    {
      "epoch": 1.526165008544922e-05,
      "step": 16003,
      "training_step_time": 7.719929933547974
    },
    {
      "epoch": 1.5262603759765626e-05,
      "model_forward_time": 1.739018201828003,
      "step": 16004
    },
    {
      "epoch": 1.5262603759765626e-05,
      "step": 16004,
      "training_step_time": 7.084075689315796
    },
    {
      "epoch": 1.526355743408203e-05,
      "model_forward_time": 1.740405559539795,
      "step": 16005
    },
    {
      "epoch": 1.526355743408203e-05,
      "step": 16005,
      "training_step_time": 8.348846673965454
    },
    {
      "epoch": 1.5264511108398438e-05,
      "model_forward_time": 1.7412958145141602,
      "step": 16006
    },
    {
      "epoch": 1.5264511108398438e-05,
      "step": 16006,
      "training_step_time": 8.123678922653198
    },
    {
      "epoch": 1.5265464782714845e-05,
      "model_forward_time": 1.7383909225463867,
      "step": 16007
    },
    {
      "epoch": 1.5265464782714845e-05,
      "step": 16007,
      "training_step_time": 8.217257738113403
    },
    {
      "epoch": 1.526641845703125e-05,
      "model_forward_time": 1.7317278385162354,
      "step": 16008
    },
    {
      "epoch": 1.526641845703125e-05,
      "step": 16008,
      "training_step_time": 7.7000954151153564
    },
    {
      "epoch": 1.5267372131347657e-05,
      "model_forward_time": 1.7316961288452148,
      "step": 16009
    },
    {
      "epoch": 1.5267372131347657e-05,
      "step": 16009,
      "training_step_time": 9.11057710647583
    },
    {
      "epoch": 1.5268325805664064e-05,
      "grad_norm": 0.21804670989513397,
      "learning_rate": 9.672292651096446e-06,
      "loss": 0.0656,
      "step": 16010
    },
    {
      "epoch": 1.5268325805664064e-05,
      "model_forward_time": 2.2713475227355957,
      "step": 16010
    },
    {
      "dynamics_loss_avg": 0.059550581686198714,
      "epoch": 1.5268325805664064e-05,
      "step": 16010
    },
    {
      "action_loss_avg": 0.0032045860833022745,
      "epoch": 1.5268325805664064e-05,
      "step": 16010
    },
    {
      "epoch": 1.5268325805664064e-05,
      "step": 16010,
      "training_step_time": 8.318852424621582
    },
    {
      "epoch": 1.5269279479980468e-05,
      "model_forward_time": 1.7773244380950928,
      "step": 16011
    },
    {
      "epoch": 1.5269279479980468e-05,
      "step": 16011,
      "training_step_time": 7.6301093101501465
    },
    {
      "epoch": 1.5270233154296875e-05,
      "model_forward_time": 1.7338991165161133,
      "step": 16012
    },
    {
      "epoch": 1.5270233154296875e-05,
      "step": 16012,
      "training_step_time": 7.697308540344238
    },
    {
      "epoch": 1.5271186828613283e-05,
      "model_forward_time": 1.736525535583496,
      "step": 16013
    },
    {
      "epoch": 1.5271186828613283e-05,
      "step": 16013,
      "training_step_time": 7.683343172073364
    },
    {
      "epoch": 1.5272140502929687e-05,
      "model_forward_time": 1.7386932373046875,
      "step": 16014
    },
    {
      "epoch": 1.5272140502929687e-05,
      "step": 16014,
      "training_step_time": 7.677349805831909
    },
    {
      "epoch": 1.5273094177246094e-05,
      "model_forward_time": 1.7344067096710205,
      "step": 16015
    },
    {
      "epoch": 1.5273094177246094e-05,
      "step": 16015,
      "training_step_time": 7.439676284790039
    },
    {
      "epoch": 1.52740478515625e-05,
      "model_forward_time": 1.7418231964111328,
      "step": 16016
    },
    {
      "epoch": 1.52740478515625e-05,
      "step": 16016,
      "training_step_time": 7.651982069015503
    },
    {
      "epoch": 1.5275001525878906e-05,
      "model_forward_time": 1.7336955070495605,
      "step": 16017
    },
    {
      "epoch": 1.5275001525878906e-05,
      "step": 16017,
      "training_step_time": 7.777885913848877
    },
    {
      "epoch": 1.5275955200195313e-05,
      "model_forward_time": 1.7360494136810303,
      "step": 16018
    },
    {
      "epoch": 1.5275955200195313e-05,
      "step": 16018,
      "training_step_time": 8.244184017181396
    },
    {
      "epoch": 1.5276908874511717e-05,
      "model_forward_time": 2.2774548530578613,
      "step": 16019
    },
    {
      "epoch": 1.5276908874511717e-05,
      "step": 16019,
      "training_step_time": 7.757722616195679
    },
    {
      "epoch": 1.5277862548828125e-05,
      "grad_norm": 0.23527973890304565,
      "learning_rate": 9.671703641357041e-06,
      "loss": 0.0661,
      "step": 16020
    },
    {
      "epoch": 1.5277862548828125e-05,
      "model_forward_time": 1.775641679763794,
      "step": 16020
    },
    {
      "dynamics_loss_avg": 0.06445156121626497,
      "epoch": 1.5277862548828125e-05,
      "step": 16020
    },
    {
      "action_loss_avg": 0.002498945138358977,
      "epoch": 1.5277862548828125e-05,
      "step": 16020
    },
    {
      "epoch": 1.5277862548828125e-05,
      "step": 16020,
      "training_step_time": 8.666529178619385
    },
    {
      "epoch": 1.5278816223144532e-05,
      "model_forward_time": 1.7235932350158691,
      "step": 16021
    },
    {
      "epoch": 1.5278816223144532e-05,
      "step": 16021,
      "training_step_time": 7.874067544937134
    },
    {
      "epoch": 1.5279769897460936e-05,
      "model_forward_time": 1.7290425300598145,
      "step": 16022
    },
    {
      "epoch": 1.5279769897460936e-05,
      "step": 16022,
      "training_step_time": 7.698353052139282
    },
    {
      "epoch": 1.5280723571777343e-05,
      "model_forward_time": 1.733022689819336,
      "step": 16023
    },
    {
      "epoch": 1.5280723571777343e-05,
      "step": 16023,
      "training_step_time": 8.653562545776367
    },
    {
      "epoch": 1.528167724609375e-05,
      "model_forward_time": 1.7246198654174805,
      "step": 16024
    },
    {
      "epoch": 1.528167724609375e-05,
      "step": 16024,
      "training_step_time": 7.481288433074951
    },
    {
      "epoch": 1.5282630920410155e-05,
      "model_forward_time": 1.725006341934204,
      "step": 16025
    },
    {
      "epoch": 1.5282630920410155e-05,
      "step": 16025,
      "training_step_time": 7.531520128250122
    },
    {
      "epoch": 1.5283584594726562e-05,
      "model_forward_time": 1.7227447032928467,
      "step": 16026
    },
    {
      "epoch": 1.5283584594726562e-05,
      "step": 16026,
      "training_step_time": 8.592976808547974
    },
    {
      "epoch": 1.528453826904297e-05,
      "model_forward_time": 1.7338178157806396,
      "step": 16027
    },
    {
      "epoch": 1.528453826904297e-05,
      "step": 16027,
      "training_step_time": 7.7509589195251465
    },
    {
      "epoch": 1.5285491943359374e-05,
      "model_forward_time": 2.2692224979400635,
      "step": 16028
    },
    {
      "epoch": 1.5285491943359374e-05,
      "step": 16028,
      "training_step_time": 9.222204208374023
    },
    {
      "epoch": 1.528644561767578e-05,
      "model_forward_time": 1.7382831573486328,
      "step": 16029
    },
    {
      "epoch": 1.528644561767578e-05,
      "step": 16029,
      "training_step_time": 7.690261363983154
    },
    {
      "epoch": 1.528739929199219e-05,
      "grad_norm": 0.2412494271993637,
      "learning_rate": 9.671114120727152e-06,
      "loss": 0.0666,
      "step": 16030
    },
    {
      "epoch": 1.528739929199219e-05,
      "model_forward_time": 1.7365586757659912,
      "step": 16030
    },
    {
      "dynamics_loss_avg": 0.06691526807844639,
      "epoch": 1.528739929199219e-05,
      "step": 16030
    },
    {
      "action_loss_avg": 0.003463745571207255,
      "epoch": 1.528739929199219e-05,
      "step": 16030
    },
    {
      "epoch": 1.528739929199219e-05,
      "step": 16030,
      "training_step_time": 8.571066856384277
    },
    {
      "epoch": 1.5288352966308593e-05,
      "model_forward_time": 1.737281322479248,
      "step": 16031
    },
    {
      "epoch": 1.5288352966308593e-05,
      "step": 16031,
      "training_step_time": 7.64006781578064
    },
    {
      "epoch": 1.5289306640625e-05,
      "model_forward_time": 1.7379753589630127,
      "step": 16032
    },
    {
      "epoch": 1.5289306640625e-05,
      "step": 16032,
      "training_step_time": 7.631595134735107
    },
    {
      "epoch": 1.5290260314941407e-05,
      "model_forward_time": 1.7466180324554443,
      "step": 16033
    },
    {
      "epoch": 1.5290260314941407e-05,
      "step": 16033,
      "training_step_time": 7.825256109237671
    },
    {
      "epoch": 1.529121398925781e-05,
      "model_forward_time": 1.7437052726745605,
      "step": 16034
    },
    {
      "epoch": 1.529121398925781e-05,
      "step": 16034,
      "training_step_time": 8.033345937728882
    },
    {
      "epoch": 1.529216766357422e-05,
      "model_forward_time": 1.7379238605499268,
      "step": 16035
    },
    {
      "epoch": 1.529216766357422e-05,
      "step": 16035,
      "training_step_time": 8.736909866333008
    },
    {
      "epoch": 1.5293121337890626e-05,
      "model_forward_time": 1.7508080005645752,
      "step": 16036
    },
    {
      "epoch": 1.5293121337890626e-05,
      "step": 16036,
      "training_step_time": 10.134842157363892
    },
    {
      "epoch": 1.529407501220703e-05,
      "model_forward_time": 2.308661699295044,
      "step": 16037
    },
    {
      "epoch": 1.529407501220703e-05,
      "step": 16037,
      "training_step_time": 8.528345584869385
    },
    {
      "epoch": 1.5295028686523438e-05,
      "model_forward_time": 1.7288870811462402,
      "step": 16038
    },
    {
      "epoch": 1.5295028686523438e-05,
      "step": 16038,
      "training_step_time": 7.776223659515381
    },
    {
      "epoch": 1.5295982360839845e-05,
      "model_forward_time": 1.7304251194000244,
      "step": 16039
    },
    {
      "epoch": 1.5295982360839845e-05,
      "step": 16039,
      "training_step_time": 7.935187578201294
    },
    {
      "epoch": 1.529693603515625e-05,
      "grad_norm": 0.1475430130958557,
      "learning_rate": 9.670524089271242e-06,
      "loss": 0.0637,
      "step": 16040
    },
    {
      "epoch": 1.529693603515625e-05,
      "model_forward_time": 1.7325780391693115,
      "step": 16040
    },
    {
      "dynamics_loss_avg": 0.057153523061424495,
      "epoch": 1.529693603515625e-05,
      "step": 16040
    },
    {
      "action_loss_avg": 0.0031830441439524293,
      "epoch": 1.529693603515625e-05,
      "step": 16040
    },
    {
      "epoch": 1.529693603515625e-05,
      "step": 16040,
      "training_step_time": 7.725486516952515
    },
    {
      "epoch": 1.5297889709472657e-05,
      "model_forward_time": 1.3504409790039062,
      "step": 16041
    },
    {
      "epoch": 1.5297889709472657e-05,
      "step": 16041,
      "training_step_time": 6.813079357147217
    },
    {
      "epoch": 1.5298843383789064e-05,
      "model_forward_time": 1.7328946590423584,
      "step": 16042
    },
    {
      "epoch": 1.5298843383789064e-05,
      "step": 16042,
      "training_step_time": 7.679307460784912
    },
    {
      "epoch": 1.5299797058105468e-05,
      "model_forward_time": 1.719865083694458,
      "step": 16043
    },
    {
      "epoch": 1.5299797058105468e-05,
      "step": 16043,
      "training_step_time": 7.459152460098267
    },
    {
      "epoch": 1.5300750732421875e-05,
      "model_forward_time": 1.739647388458252,
      "step": 16044
    },
    {
      "epoch": 1.5300750732421875e-05,
      "step": 16044,
      "training_step_time": 8.367080688476562
    },
    {
      "epoch": 1.5301704406738283e-05,
      "model_forward_time": 1.7382254600524902,
      "step": 16045
    },
    {
      "epoch": 1.5301704406738283e-05,
      "step": 16045,
      "training_step_time": 8.439545154571533
    },
    {
      "epoch": 1.5302658081054687e-05,
      "model_forward_time": 1.7375314235687256,
      "step": 16046
    },
    {
      "epoch": 1.5302658081054687e-05,
      "step": 16046,
      "training_step_time": 8.414140701293945
    },
    {
      "epoch": 1.5303611755371094e-05,
      "model_forward_time": 2.2693779468536377,
      "step": 16047
    },
    {
      "epoch": 1.5303611755371094e-05,
      "step": 16047,
      "training_step_time": 7.669654607772827
    },
    {
      "epoch": 1.53045654296875e-05,
      "model_forward_time": 1.7397830486297607,
      "step": 16048
    },
    {
      "epoch": 1.53045654296875e-05,
      "step": 16048,
      "training_step_time": 7.575047016143799
    },
    {
      "epoch": 1.5305519104003906e-05,
      "model_forward_time": 1.7481026649475098,
      "step": 16049
    },
    {
      "epoch": 1.5305519104003906e-05,
      "step": 16049,
      "training_step_time": 7.623622179031372
    },
    {
      "epoch": 1.5306472778320313e-05,
      "grad_norm": 0.25021687150001526,
      "learning_rate": 9.669933547053843e-06,
      "loss": 0.0662,
      "step": 16050
    },
    {
      "epoch": 1.5306472778320313e-05,
      "model_forward_time": 1.731215238571167,
      "step": 16050
    },
    {
      "dynamics_loss_avg": 0.06068866606801748,
      "epoch": 1.5306472778320313e-05,
      "step": 16050
    },
    {
      "action_loss_avg": 0.004070689622312784,
      "epoch": 1.5306472778320313e-05,
      "step": 16050
    },
    {
      "epoch": 1.5306472778320313e-05,
      "step": 16050,
      "training_step_time": 7.656189441680908
    },
    {
      "epoch": 1.5307426452636717e-05,
      "model_forward_time": 1.7407104969024658,
      "step": 16051
    },
    {
      "epoch": 1.5307426452636717e-05,
      "step": 16051,
      "training_step_time": 7.559916734695435
    },
    {
      "epoch": 1.5308380126953125e-05,
      "model_forward_time": 1.7504093647003174,
      "step": 16052
    },
    {
      "epoch": 1.5308380126953125e-05,
      "step": 16052,
      "training_step_time": 7.651374816894531
    },
    {
      "epoch": 1.5309333801269532e-05,
      "model_forward_time": 1.7327873706817627,
      "step": 16053
    },
    {
      "epoch": 1.5309333801269532e-05,
      "step": 16053,
      "training_step_time": 7.6451239585876465
    },
    {
      "epoch": 1.5310287475585936e-05,
      "model_forward_time": 1.7462403774261475,
      "step": 16054
    },
    {
      "epoch": 1.5310287475585936e-05,
      "step": 16054,
      "training_step_time": 8.20542311668396
    },
    {
      "epoch": 1.5311241149902343e-05,
      "model_forward_time": 1.437455654144287,
      "step": 16055
    },
    {
      "epoch": 1.5311241149902343e-05,
      "step": 16055,
      "training_step_time": 8.749913454055786
    },
    {
      "epoch": 1.531219482421875e-05,
      "model_forward_time": 2.271331310272217,
      "step": 16056
    },
    {
      "epoch": 1.531219482421875e-05,
      "step": 16056,
      "training_step_time": 9.237163066864014
    },
    {
      "epoch": 1.5313148498535155e-05,
      "model_forward_time": 1.7332804203033447,
      "step": 16057
    },
    {
      "epoch": 1.5313148498535155e-05,
      "step": 16057,
      "training_step_time": 7.610148668289185
    },
    {
      "epoch": 1.5314102172851562e-05,
      "model_forward_time": 1.740131139755249,
      "step": 16058
    },
    {
      "epoch": 1.5314102172851562e-05,
      "step": 16058,
      "training_step_time": 7.766679048538208
    },
    {
      "epoch": 1.531505584716797e-05,
      "model_forward_time": 1.7467365264892578,
      "step": 16059
    },
    {
      "epoch": 1.531505584716797e-05,
      "step": 16059,
      "training_step_time": 7.653695344924927
    },
    {
      "epoch": 1.5316009521484374e-05,
      "grad_norm": 0.2221192717552185,
      "learning_rate": 9.66934249413953e-06,
      "loss": 0.0656,
      "step": 16060
    },
    {
      "epoch": 1.5316009521484374e-05,
      "model_forward_time": 1.7123355865478516,
      "step": 16060
    },
    {
      "dynamics_loss_avg": 0.06958990134298801,
      "epoch": 1.5316009521484374e-05,
      "step": 16060
    },
    {
      "action_loss_avg": 0.0030685113684739917,
      "epoch": 1.5316009521484374e-05,
      "step": 16060
    },
    {
      "epoch": 1.5316009521484374e-05,
      "step": 16060,
      "training_step_time": 7.541086435317993
    },
    {
      "epoch": 1.531696319580078e-05,
      "model_forward_time": 1.7373509407043457,
      "step": 16061
    },
    {
      "epoch": 1.531696319580078e-05,
      "step": 16061,
      "training_step_time": 7.658551454544067
    },
    {
      "epoch": 1.531791687011719e-05,
      "model_forward_time": 1.744612455368042,
      "step": 16062
    },
    {
      "epoch": 1.531791687011719e-05,
      "step": 16062,
      "training_step_time": 8.570671081542969
    },
    {
      "epoch": 1.5318870544433593e-05,
      "model_forward_time": 1.735889196395874,
      "step": 16063
    },
    {
      "epoch": 1.5318870544433593e-05,
      "step": 16063,
      "training_step_time": 7.675238847732544
    },
    {
      "epoch": 1.531982421875e-05,
      "model_forward_time": 1.8048148155212402,
      "step": 16064
    },
    {
      "epoch": 1.531982421875e-05,
      "step": 16064,
      "training_step_time": 8.384986877441406
    },
    {
      "epoch": 1.5320777893066407e-05,
      "model_forward_time": 2.3730764389038086,
      "step": 16065
    },
    {
      "epoch": 1.5320777893066407e-05,
      "step": 16065,
      "training_step_time": 7.6627490520477295
    },
    {
      "epoch": 1.532173156738281e-05,
      "model_forward_time": 1.818131923675537,
      "step": 16066
    },
    {
      "epoch": 1.532173156738281e-05,
      "step": 16066,
      "training_step_time": 8.345214128494263
    },
    {
      "epoch": 1.532268524169922e-05,
      "model_forward_time": 1.8448679447174072,
      "step": 16067
    },
    {
      "epoch": 1.532268524169922e-05,
      "step": 16067,
      "training_step_time": 7.704343795776367
    },
    {
      "epoch": 1.5323638916015626e-05,
      "model_forward_time": 1.8338344097137451,
      "step": 16068
    },
    {
      "epoch": 1.5323638916015626e-05,
      "step": 16068,
      "training_step_time": 7.768584728240967
    },
    {
      "epoch": 1.532459259033203e-05,
      "model_forward_time": 1.8189332485198975,
      "step": 16069
    },
    {
      "epoch": 1.532459259033203e-05,
      "step": 16069,
      "training_step_time": 7.632671117782593
    },
    {
      "epoch": 1.5325546264648438e-05,
      "grad_norm": 0.2713727355003357,
      "learning_rate": 9.668750930592944e-06,
      "loss": 0.0693,
      "step": 16070
    },
    {
      "epoch": 1.5325546264648438e-05,
      "model_forward_time": 1.8403127193450928,
      "step": 16070
    },
    {
      "dynamics_loss_avg": 0.0533231632784009,
      "epoch": 1.5325546264648438e-05,
      "step": 16070
    },
    {
      "action_loss_avg": 0.002481319016078487,
      "epoch": 1.5325546264648438e-05,
      "step": 16070
    },
    {
      "epoch": 1.5325546264648438e-05,
      "step": 16070,
      "training_step_time": 8.51210618019104
    },
    {
      "epoch": 1.5326499938964845e-05,
      "model_forward_time": 1.8408446311950684,
      "step": 16071
    },
    {
      "epoch": 1.5326499938964845e-05,
      "step": 16071,
      "training_step_time": 8.123429775238037
    },
    {
      "epoch": 1.532745361328125e-05,
      "model_forward_time": 1.8377220630645752,
      "step": 16072
    },
    {
      "epoch": 1.532745361328125e-05,
      "step": 16072,
      "training_step_time": 7.595238447189331
    },
    {
      "epoch": 1.5328407287597657e-05,
      "model_forward_time": 1.8369050025939941,
      "step": 16073
    },
    {
      "epoch": 1.5328407287597657e-05,
      "step": 16073,
      "training_step_time": 8.609567165374756
    },
    {
      "epoch": 1.5329360961914064e-05,
      "model_forward_time": 2.3547027111053467,
      "step": 16074
    },
    {
      "epoch": 1.5329360961914064e-05,
      "step": 16074,
      "training_step_time": 8.643792390823364
    },
    {
      "epoch": 1.5330314636230468e-05,
      "model_forward_time": 1.8363685607910156,
      "step": 16075
    },
    {
      "epoch": 1.5330314636230468e-05,
      "step": 16075,
      "training_step_time": 8.520581483840942
    },
    {
      "epoch": 1.5331268310546875e-05,
      "model_forward_time": 1.8352794647216797,
      "step": 16076
    },
    {
      "epoch": 1.5331268310546875e-05,
      "step": 16076,
      "training_step_time": 7.759483814239502
    },
    {
      "epoch": 1.5332221984863283e-05,
      "model_forward_time": 1.8346116542816162,
      "step": 16077
    },
    {
      "epoch": 1.5332221984863283e-05,
      "step": 16077,
      "training_step_time": 7.318127393722534
    },
    {
      "epoch": 1.5333175659179687e-05,
      "model_forward_time": 1.842771053314209,
      "step": 16078
    },
    {
      "epoch": 1.5333175659179687e-05,
      "step": 16078,
      "training_step_time": 7.6744325160980225
    },
    {
      "epoch": 1.5334129333496094e-05,
      "model_forward_time": 1.8392646312713623,
      "step": 16079
    },
    {
      "epoch": 1.5334129333496094e-05,
      "step": 16079,
      "training_step_time": 8.394313097000122
    },
    {
      "epoch": 1.53350830078125e-05,
      "grad_norm": 0.1636454463005066,
      "learning_rate": 9.668158856478774e-06,
      "loss": 0.064,
      "step": 16080
    },
    {
      "epoch": 1.53350830078125e-05,
      "model_forward_time": 1.8406507968902588,
      "step": 16080
    },
    {
      "dynamics_loss_avg": 0.05301428306847811,
      "epoch": 1.53350830078125e-05,
      "step": 16080
    },
    {
      "action_loss_avg": 0.004110094357747585,
      "epoch": 1.53350830078125e-05,
      "step": 16080
    },
    {
      "epoch": 1.53350830078125e-05,
      "step": 16080,
      "training_step_time": 7.515666961669922
    },
    {
      "epoch": 1.5336036682128906e-05,
      "model_forward_time": 1.8416306972503662,
      "step": 16081
    },
    {
      "epoch": 1.5336036682128906e-05,
      "step": 16081,
      "training_step_time": 7.612995862960815
    },
    {
      "epoch": 1.5336990356445313e-05,
      "model_forward_time": 1.827225923538208,
      "step": 16082
    },
    {
      "epoch": 1.5336990356445313e-05,
      "step": 16082,
      "training_step_time": 8.292345523834229
    },
    {
      "epoch": 1.5337944030761717e-05,
      "model_forward_time": 2.395214796066284,
      "step": 16083
    },
    {
      "epoch": 1.5337944030761717e-05,
      "step": 16083,
      "training_step_time": 8.747726440429688
    },
    {
      "epoch": 1.5338897705078125e-05,
      "model_forward_time": 1.842907190322876,
      "step": 16084
    },
    {
      "epoch": 1.5338897705078125e-05,
      "step": 16084,
      "training_step_time": 7.7295122146606445
    },
    {
      "epoch": 1.5339851379394532e-05,
      "model_forward_time": 1.8274197578430176,
      "step": 16085
    },
    {
      "epoch": 1.5339851379394532e-05,
      "step": 16085,
      "training_step_time": 7.602315664291382
    },
    {
      "epoch": 1.5340805053710936e-05,
      "model_forward_time": 1.8406059741973877,
      "step": 16086
    },
    {
      "epoch": 1.5340805053710936e-05,
      "step": 16086,
      "training_step_time": 7.654155969619751
    },
    {
      "epoch": 1.5341758728027344e-05,
      "model_forward_time": 1.8466997146606445,
      "step": 16087
    },
    {
      "epoch": 1.5341758728027344e-05,
      "step": 16087,
      "training_step_time": 7.777624607086182
    },
    {
      "epoch": 1.534271240234375e-05,
      "model_forward_time": 1.8378245830535889,
      "step": 16088
    },
    {
      "epoch": 1.534271240234375e-05,
      "step": 16088,
      "training_step_time": 7.578035831451416
    },
    {
      "epoch": 1.5343666076660155e-05,
      "model_forward_time": 1.8192825317382812,
      "step": 16089
    },
    {
      "epoch": 1.5343666076660155e-05,
      "step": 16089,
      "training_step_time": 9.097700357437134
    },
    {
      "epoch": 1.5344619750976562e-05,
      "grad_norm": 0.2312946617603302,
      "learning_rate": 9.667566271861771e-06,
      "loss": 0.0634,
      "step": 16090
    },
    {
      "epoch": 1.5344619750976562e-05,
      "model_forward_time": 1.843501091003418,
      "step": 16090
    },
    {
      "dynamics_loss_avg": 0.06795056853443385,
      "epoch": 1.5344619750976562e-05,
      "step": 16090
    },
    {
      "action_loss_avg": 0.005130504176486283,
      "epoch": 1.5344619750976562e-05,
      "step": 16090
    },
    {
      "epoch": 1.5344619750976562e-05,
      "step": 16090,
      "training_step_time": 7.519733190536499
    },
    {
      "epoch": 1.534557342529297e-05,
      "model_forward_time": 1.4727189540863037,
      "step": 16091
    },
    {
      "epoch": 1.534557342529297e-05,
      "step": 16091,
      "training_step_time": 7.6989147663116455
    },
    {
      "epoch": 1.5346527099609374e-05,
      "model_forward_time": 1.8355655670166016,
      "step": 16092
    },
    {
      "epoch": 1.5346527099609374e-05,
      "step": 16092,
      "training_step_time": 7.746393918991089
    },
    {
      "epoch": 1.534748077392578e-05,
      "model_forward_time": 2.3683197498321533,
      "step": 16093
    },
    {
      "epoch": 1.534748077392578e-05,
      "step": 16093,
      "training_step_time": 7.678478240966797
    },
    {
      "epoch": 1.534843444824219e-05,
      "model_forward_time": 1.8328444957733154,
      "step": 16094
    },
    {
      "epoch": 1.534843444824219e-05,
      "step": 16094,
      "training_step_time": 8.682230472564697
    },
    {
      "epoch": 1.5349388122558593e-05,
      "model_forward_time": 1.8419456481933594,
      "step": 16095
    },
    {
      "epoch": 1.5349388122558593e-05,
      "step": 16095,
      "training_step_time": 7.883848667144775
    },
    {
      "epoch": 1.5350341796875e-05,
      "model_forward_time": 1.4667186737060547,
      "step": 16096
    },
    {
      "epoch": 1.5350341796875e-05,
      "step": 16096,
      "training_step_time": 7.780439138412476
    },
    {
      "epoch": 1.5351295471191407e-05,
      "model_forward_time": 1.8202300071716309,
      "step": 16097
    },
    {
      "epoch": 1.5351295471191407e-05,
      "step": 16097,
      "training_step_time": 7.653506755828857
    },
    {
      "epoch": 1.535224914550781e-05,
      "model_forward_time": 1.8162078857421875,
      "step": 16098
    },
    {
      "epoch": 1.535224914550781e-05,
      "step": 16098,
      "training_step_time": 8.508326768875122
    },
    {
      "epoch": 1.535320281982422e-05,
      "model_forward_time": 1.845324993133545,
      "step": 16099
    },
    {
      "epoch": 1.535320281982422e-05,
      "step": 16099,
      "training_step_time": 7.769629955291748
    },
    {
      "epoch": 1.5354156494140626e-05,
      "grad_norm": 0.3100050687789917,
      "learning_rate": 9.666973176806737e-06,
      "loss": 0.0674,
      "step": 16100
    },
    {
      "epoch": 1.5354156494140626e-05,
      "model_forward_time": 1.836639642715454,
      "step": 16100
    },
    {
      "dynamics_loss_avg": 0.04812203384935856,
      "epoch": 1.5354156494140626e-05,
      "step": 16100
    },
    {
      "action_loss_avg": 0.0026284030405804514,
      "epoch": 1.5354156494140626e-05,
      "step": 16100
    },
    {
      "epoch": 1.5354156494140626e-05,
      "step": 16100,
      "training_step_time": 7.713189125061035
    },
    {
      "epoch": 1.535511016845703e-05,
      "model_forward_time": 1.843186855316162,
      "step": 16101
    },
    {
      "epoch": 1.535511016845703e-05,
      "step": 16101,
      "training_step_time": 8.738454103469849
    },
    {
      "epoch": 1.5356063842773438e-05,
      "model_forward_time": 2.352940320968628,
      "step": 16102
    },
    {
      "epoch": 1.5356063842773438e-05,
      "step": 16102,
      "training_step_time": 8.101531267166138
    },
    {
      "epoch": 1.5357017517089845e-05,
      "model_forward_time": 1.8355951309204102,
      "step": 16103
    },
    {
      "epoch": 1.5357017517089845e-05,
      "step": 16103,
      "training_step_time": 8.851928949356079
    },
    {
      "epoch": 1.535797119140625e-05,
      "model_forward_time": 1.8369123935699463,
      "step": 16104
    },
    {
      "epoch": 1.535797119140625e-05,
      "step": 16104,
      "training_step_time": 7.770218372344971
    },
    {
      "epoch": 1.5358924865722657e-05,
      "model_forward_time": 1.8339154720306396,
      "step": 16105
    },
    {
      "epoch": 1.5358924865722657e-05,
      "step": 16105,
      "training_step_time": 7.749330282211304
    },
    {
      "epoch": 1.5359878540039064e-05,
      "model_forward_time": 1.832418441772461,
      "step": 16106
    },
    {
      "epoch": 1.5359878540039064e-05,
      "step": 16106,
      "training_step_time": 7.558189392089844
    },
    {
      "epoch": 1.5360832214355468e-05,
      "model_forward_time": 1.8350090980529785,
      "step": 16107
    },
    {
      "epoch": 1.5360832214355468e-05,
      "step": 16107,
      "training_step_time": 7.612083673477173
    },
    {
      "epoch": 1.5361785888671876e-05,
      "model_forward_time": 1.836822271347046,
      "step": 16108
    },
    {
      "epoch": 1.5361785888671876e-05,
      "step": 16108,
      "training_step_time": 7.5783631801605225
    },
    {
      "epoch": 1.5362739562988283e-05,
      "model_forward_time": 1.8329615592956543,
      "step": 16109
    },
    {
      "epoch": 1.5362739562988283e-05,
      "step": 16109,
      "training_step_time": 9.642639398574829
    },
    {
      "epoch": 1.5363693237304687e-05,
      "grad_norm": 0.18794821202754974,
      "learning_rate": 9.666379571378535e-06,
      "loss": 0.0646,
      "step": 16110
    },
    {
      "epoch": 1.5363693237304687e-05,
      "model_forward_time": 1.8386611938476562,
      "step": 16110
    },
    {
      "dynamics_loss_avg": 0.07614349760115147,
      "epoch": 1.5363693237304687e-05,
      "step": 16110
    },
    {
      "action_loss_avg": 0.0033683467539958657,
      "epoch": 1.5363693237304687e-05,
      "step": 16110
    },
    {
      "epoch": 1.5363693237304687e-05,
      "step": 16110,
      "training_step_time": 8.507474660873413
    },
    {
      "epoch": 1.5364646911621094e-05,
      "model_forward_time": 2.386746644973755,
      "step": 16111
    },
    {
      "epoch": 1.5364646911621094e-05,
      "step": 16111,
      "training_step_time": 7.662929058074951
    },
    {
      "epoch": 1.53656005859375e-05,
      "model_forward_time": 1.838944673538208,
      "step": 16112
    },
    {
      "epoch": 1.53656005859375e-05,
      "step": 16112,
      "training_step_time": 7.823399066925049
    },
    {
      "epoch": 1.5366554260253906e-05,
      "model_forward_time": 1.8397986888885498,
      "step": 16113
    },
    {
      "epoch": 1.5366554260253906e-05,
      "step": 16113,
      "training_step_time": 7.757944822311401
    },
    {
      "epoch": 1.5367507934570313e-05,
      "model_forward_time": 1.8405635356903076,
      "step": 16114
    },
    {
      "epoch": 1.5367507934570313e-05,
      "step": 16114,
      "training_step_time": 7.745826959609985
    },
    {
      "epoch": 1.5368461608886717e-05,
      "model_forward_time": 1.8431751728057861,
      "step": 16115
    },
    {
      "epoch": 1.5368461608886717e-05,
      "step": 16115,
      "training_step_time": 7.741754531860352
    },
    {
      "epoch": 1.5369415283203125e-05,
      "model_forward_time": 1.8327970504760742,
      "step": 16116
    },
    {
      "epoch": 1.5369415283203125e-05,
      "step": 16116,
      "training_step_time": 7.685483932495117
    },
    {
      "epoch": 1.5370368957519532e-05,
      "model_forward_time": 1.8301072120666504,
      "step": 16117
    },
    {
      "epoch": 1.5370368957519532e-05,
      "step": 16117,
      "training_step_time": 8.334534168243408
    },
    {
      "epoch": 1.5371322631835936e-05,
      "model_forward_time": 1.8379673957824707,
      "step": 16118
    },
    {
      "epoch": 1.5371322631835936e-05,
      "step": 16118,
      "training_step_time": 8.687376737594604
    },
    {
      "epoch": 1.5372276306152344e-05,
      "model_forward_time": 1.7863006591796875,
      "step": 16119
    },
    {
      "epoch": 1.5372276306152344e-05,
      "step": 16119,
      "training_step_time": 8.688224077224731
    },
    {
      "epoch": 1.537322998046875e-05,
      "grad_norm": 0.1720438003540039,
      "learning_rate": 9.665785455642077e-06,
      "loss": 0.0663,
      "step": 16120
    },
    {
      "epoch": 1.537322998046875e-05,
      "model_forward_time": 2.2828915119171143,
      "step": 16120
    },
    {
      "dynamics_loss_avg": 0.08066053129732609,
      "epoch": 1.537322998046875e-05,
      "step": 16120
    },
    {
      "action_loss_avg": 0.004803641070611775,
      "epoch": 1.537322998046875e-05,
      "step": 16120
    },
    {
      "epoch": 1.537322998046875e-05,
      "step": 16120,
      "training_step_time": 7.765002489089966
    },
    {
      "epoch": 1.5374183654785155e-05,
      "model_forward_time": 1.7255353927612305,
      "step": 16121
    },
    {
      "epoch": 1.5374183654785155e-05,
      "step": 16121,
      "training_step_time": 7.804301500320435
    },
    {
      "epoch": 1.5375137329101562e-05,
      "model_forward_time": 1.714143991470337,
      "step": 16122
    },
    {
      "epoch": 1.5375137329101562e-05,
      "step": 16122,
      "training_step_time": 7.727102279663086
    },
    {
      "epoch": 1.537609100341797e-05,
      "model_forward_time": 1.7284514904022217,
      "step": 16123
    },
    {
      "epoch": 1.537609100341797e-05,
      "step": 16123,
      "training_step_time": 7.654563665390015
    },
    {
      "epoch": 1.5377044677734374e-05,
      "model_forward_time": 1.7234163284301758,
      "step": 16124
    },
    {
      "epoch": 1.5377044677734374e-05,
      "step": 16124,
      "training_step_time": 7.78540301322937
    },
    {
      "epoch": 1.537799835205078e-05,
      "model_forward_time": 1.7194337844848633,
      "step": 16125
    },
    {
      "epoch": 1.537799835205078e-05,
      "step": 16125,
      "training_step_time": 7.579836845397949
    },
    {
      "epoch": 1.537895202636719e-05,
      "model_forward_time": 1.742539405822754,
      "step": 16126
    },
    {
      "epoch": 1.537895202636719e-05,
      "step": 16126,
      "training_step_time": 7.662704944610596
    },
    {
      "epoch": 1.5379905700683593e-05,
      "model_forward_time": 1.7171213626861572,
      "step": 16127
    },
    {
      "epoch": 1.5379905700683593e-05,
      "step": 16127,
      "training_step_time": 7.7253499031066895
    },
    {
      "epoch": 1.5380859375e-05,
      "model_forward_time": 1.709223985671997,
      "step": 16128
    },
    {
      "epoch": 1.5380859375e-05,
      "step": 16128,
      "training_step_time": 8.739639282226562
    },
    {
      "epoch": 1.5381813049316408e-05,
      "model_forward_time": 1.7048161029815674,
      "step": 16129
    },
    {
      "epoch": 1.5381813049316408e-05,
      "step": 16129,
      "training_step_time": 9.21006155014038
    },
    {
      "epoch": 1.538276672363281e-05,
      "grad_norm": 0.3216361999511719,
      "learning_rate": 9.665190829662337e-06,
      "loss": 0.0683,
      "step": 16130
    },
    {
      "epoch": 1.538276672363281e-05,
      "model_forward_time": 1.3419291973114014,
      "step": 16130
    },
    {
      "dynamics_loss_avg": 0.05572177525609732,
      "epoch": 1.538276672363281e-05,
      "step": 16130
    },
    {
      "action_loss_avg": 0.00371224342379719,
      "epoch": 1.538276672363281e-05,
      "step": 16130
    },
    {
      "epoch": 1.538276672363281e-05,
      "step": 16130,
      "training_step_time": 8.368592739105225
    },
    {
      "epoch": 1.538372039794922e-05,
      "model_forward_time": 1.7218289375305176,
      "step": 16131
    },
    {
      "epoch": 1.538372039794922e-05,
      "step": 16131,
      "training_step_time": 8.629070520401001
    },
    {
      "epoch": 1.5384674072265626e-05,
      "model_forward_time": 1.7195565700531006,
      "step": 16132
    },
    {
      "epoch": 1.5384674072265626e-05,
      "step": 16132,
      "training_step_time": 7.688627481460571
    },
    {
      "epoch": 1.538562774658203e-05,
      "model_forward_time": 1.7118244171142578,
      "step": 16133
    },
    {
      "epoch": 1.538562774658203e-05,
      "step": 16133,
      "training_step_time": 7.595438480377197
    },
    {
      "epoch": 1.5386581420898438e-05,
      "model_forward_time": 1.726386547088623,
      "step": 16134
    },
    {
      "epoch": 1.5386581420898438e-05,
      "step": 16134,
      "training_step_time": 7.695264577865601
    },
    {
      "epoch": 1.5387535095214845e-05,
      "model_forward_time": 1.7186360359191895,
      "step": 16135
    },
    {
      "epoch": 1.5387535095214845e-05,
      "step": 16135,
      "training_step_time": 7.694742441177368
    },
    {
      "epoch": 1.538848876953125e-05,
      "model_forward_time": 1.7217340469360352,
      "step": 16136
    },
    {
      "epoch": 1.538848876953125e-05,
      "step": 16136,
      "training_step_time": 7.69917106628418
    },
    {
      "epoch": 1.5389442443847657e-05,
      "model_forward_time": 1.7170891761779785,
      "step": 16137
    },
    {
      "epoch": 1.5389442443847657e-05,
      "step": 16137,
      "training_step_time": 8.207242488861084
    },
    {
      "epoch": 1.5390396118164064e-05,
      "model_forward_time": 1.798666000366211,
      "step": 16138
    },
    {
      "epoch": 1.5390396118164064e-05,
      "step": 16138,
      "training_step_time": 8.433953046798706
    },
    {
      "epoch": 1.5391349792480468e-05,
      "model_forward_time": 2.2703890800476074,
      "step": 16139
    },
    {
      "epoch": 1.5391349792480468e-05,
      "step": 16139,
      "training_step_time": 8.331867933273315
    },
    {
      "epoch": 1.5392303466796876e-05,
      "grad_norm": 0.18448348343372345,
      "learning_rate": 9.664595693504341e-06,
      "loss": 0.065,
      "step": 16140
    },
    {
      "epoch": 1.5392303466796876e-05,
      "model_forward_time": 1.711085557937622,
      "step": 16140
    },
    {
      "dynamics_loss_avg": 0.04674232145771384,
      "epoch": 1.5392303466796876e-05,
      "step": 16140
    },
    {
      "action_loss_avg": 0.0025419113284442575,
      "epoch": 1.5392303466796876e-05,
      "step": 16140
    },
    {
      "epoch": 1.5392303466796876e-05,
      "step": 16140,
      "training_step_time": 7.705822706222534
    },
    {
      "epoch": 1.539325714111328e-05,
      "model_forward_time": 1.72330904006958,
      "step": 16141
    },
    {
      "epoch": 1.539325714111328e-05,
      "step": 16141,
      "training_step_time": 7.7938196659088135
    },
    {
      "epoch": 1.5394210815429687e-05,
      "model_forward_time": 1.71311616897583,
      "step": 16142
    },
    {
      "epoch": 1.5394210815429687e-05,
      "step": 16142,
      "training_step_time": 7.300216436386108
    },
    {
      "epoch": 1.5395164489746094e-05,
      "model_forward_time": 1.718590497970581,
      "step": 16143
    },
    {
      "epoch": 1.5395164489746094e-05,
      "step": 16143,
      "training_step_time": 7.674016714096069
    },
    {
      "epoch": 1.53961181640625e-05,
      "model_forward_time": 1.3342816829681396,
      "step": 16144
    },
    {
      "epoch": 1.53961181640625e-05,
      "step": 16144,
      "training_step_time": 8.451231479644775
    },
    {
      "epoch": 1.5397071838378906e-05,
      "model_forward_time": 1.717362880706787,
      "step": 16145
    },
    {
      "epoch": 1.5397071838378906e-05,
      "step": 16145,
      "training_step_time": 8.282121419906616
    },
    {
      "epoch": 1.5398025512695313e-05,
      "model_forward_time": 1.7142930030822754,
      "step": 16146
    },
    {
      "epoch": 1.5398025512695313e-05,
      "step": 16146,
      "training_step_time": 7.744966745376587
    },
    {
      "epoch": 1.5398979187011717e-05,
      "model_forward_time": 1.7919580936431885,
      "step": 16147
    },
    {
      "epoch": 1.5398979187011717e-05,
      "step": 16147,
      "training_step_time": 9.303077459335327
    },
    {
      "epoch": 1.5399932861328125e-05,
      "model_forward_time": 2.2831151485443115,
      "step": 16148
    },
    {
      "epoch": 1.5399932861328125e-05,
      "step": 16148,
      "training_step_time": 8.840205907821655
    },
    {
      "epoch": 1.5400886535644532e-05,
      "model_forward_time": 1.715381383895874,
      "step": 16149
    },
    {
      "epoch": 1.5400886535644532e-05,
      "step": 16149,
      "training_step_time": 7.753368616104126
    },
    {
      "epoch": 1.5401840209960936e-05,
      "grad_norm": 0.29979947209358215,
      "learning_rate": 9.664000047233175e-06,
      "loss": 0.0638,
      "step": 16150
    },
    {
      "epoch": 1.5401840209960936e-05,
      "model_forward_time": 1.713857650756836,
      "step": 16150
    },
    {
      "dynamics_loss_avg": 0.04347335565835238,
      "epoch": 1.5401840209960936e-05,
      "step": 16150
    },
    {
      "action_loss_avg": 0.0030754011240787804,
      "epoch": 1.5401840209960936e-05,
      "step": 16150
    },
    {
      "epoch": 1.5401840209960936e-05,
      "step": 16150,
      "training_step_time": 8.37633466720581
    },
    {
      "epoch": 1.5402793884277344e-05,
      "model_forward_time": 1.7261037826538086,
      "step": 16151
    },
    {
      "epoch": 1.5402793884277344e-05,
      "step": 16151,
      "training_step_time": 7.446824073791504
    },
    {
      "epoch": 1.540374755859375e-05,
      "model_forward_time": 1.7259888648986816,
      "step": 16152
    },
    {
      "epoch": 1.540374755859375e-05,
      "step": 16152,
      "training_step_time": 7.66887092590332
    },
    {
      "epoch": 1.5404701232910155e-05,
      "model_forward_time": 1.7169196605682373,
      "step": 16153
    },
    {
      "epoch": 1.5404701232910155e-05,
      "step": 16153,
      "training_step_time": 9.064200162887573
    },
    {
      "epoch": 1.5405654907226562e-05,
      "model_forward_time": 1.7309131622314453,
      "step": 16154
    },
    {
      "epoch": 1.5405654907226562e-05,
      "step": 16154,
      "training_step_time": 7.642043113708496
    },
    {
      "epoch": 1.540660858154297e-05,
      "model_forward_time": 1.7181522846221924,
      "step": 16155
    },
    {
      "epoch": 1.540660858154297e-05,
      "step": 16155,
      "training_step_time": 8.233023643493652
    },
    {
      "epoch": 1.5407562255859374e-05,
      "model_forward_time": 1.7795164585113525,
      "step": 16156
    },
    {
      "epoch": 1.5407562255859374e-05,
      "step": 16156,
      "training_step_time": 7.737040042877197
    },
    {
      "epoch": 1.540851593017578e-05,
      "model_forward_time": 2.2675423622131348,
      "step": 16157
    },
    {
      "epoch": 1.540851593017578e-05,
      "step": 16157,
      "training_step_time": 7.755496025085449
    },
    {
      "epoch": 1.540946960449219e-05,
      "model_forward_time": 1.71695876121521,
      "step": 16158
    },
    {
      "epoch": 1.540946960449219e-05,
      "step": 16158,
      "training_step_time": 7.8116559982299805
    },
    {
      "epoch": 1.5410423278808593e-05,
      "model_forward_time": 1.723902940750122,
      "step": 16159
    },
    {
      "epoch": 1.5410423278808593e-05,
      "step": 16159,
      "training_step_time": 7.723339796066284
    },
    {
      "epoch": 1.5411376953125e-05,
      "grad_norm": 0.15059229731559753,
      "learning_rate": 9.663403890913977e-06,
      "loss": 0.0666,
      "step": 16160
    },
    {
      "epoch": 1.5411376953125e-05,
      "model_forward_time": 1.7133808135986328,
      "step": 16160
    },
    {
      "dynamics_loss_avg": 0.053203458338975905,
      "epoch": 1.5411376953125e-05,
      "step": 16160
    },
    {
      "action_loss_avg": 0.0031513901543803514,
      "epoch": 1.5411376953125e-05,
      "step": 16160
    },
    {
      "epoch": 1.5411376953125e-05,
      "step": 16160,
      "training_step_time": 7.449733257293701
    },
    {
      "epoch": 1.5412330627441408e-05,
      "model_forward_time": 1.7215723991394043,
      "step": 16161
    },
    {
      "epoch": 1.5412330627441408e-05,
      "step": 16161,
      "training_step_time": 7.636028051376343
    },
    {
      "epoch": 1.541328430175781e-05,
      "model_forward_time": 1.7258195877075195,
      "step": 16162
    },
    {
      "epoch": 1.541328430175781e-05,
      "step": 16162,
      "training_step_time": 7.674652814865112
    },
    {
      "epoch": 1.541423797607422e-05,
      "model_forward_time": 1.717355489730835,
      "step": 16163
    },
    {
      "epoch": 1.541423797607422e-05,
      "step": 16163,
      "training_step_time": 8.023416996002197
    },
    {
      "epoch": 1.5415191650390626e-05,
      "model_forward_time": 1.7141263484954834,
      "step": 16164
    },
    {
      "epoch": 1.5415191650390626e-05,
      "step": 16164,
      "training_step_time": 7.853702068328857
    },
    {
      "epoch": 1.541614532470703e-05,
      "model_forward_time": 1.7687797546386719,
      "step": 16165
    },
    {
      "epoch": 1.541614532470703e-05,
      "step": 16165,
      "training_step_time": 8.25083065032959
    },
    {
      "epoch": 1.5417098999023438e-05,
      "model_forward_time": 1.7174646854400635,
      "step": 16166
    },
    {
      "epoch": 1.5417098999023438e-05,
      "step": 16166,
      "training_step_time": 8.208155155181885
    },
    {
      "epoch": 1.5418052673339845e-05,
      "model_forward_time": 1.7175910472869873,
      "step": 16167
    },
    {
      "epoch": 1.5418052673339845e-05,
      "step": 16167,
      "training_step_time": 7.682950496673584
    },
    {
      "epoch": 1.541900634765625e-05,
      "model_forward_time": 1.7203662395477295,
      "step": 16168
    },
    {
      "epoch": 1.541900634765625e-05,
      "step": 16168,
      "training_step_time": 9.76777720451355
    },
    {
      "epoch": 1.5419960021972657e-05,
      "model_forward_time": 1.717125654220581,
      "step": 16169
    },
    {
      "epoch": 1.5419960021972657e-05,
      "step": 16169,
      "training_step_time": 8.186363935470581
    },
    {
      "epoch": 1.5420913696289064e-05,
      "grad_norm": 0.16477249562740326,
      "learning_rate": 9.662807224611937e-06,
      "loss": 0.0684,
      "step": 16170
    },
    {
      "epoch": 1.5420913696289064e-05,
      "model_forward_time": 1.7167019844055176,
      "step": 16170
    },
    {
      "dynamics_loss_avg": 0.059671996347606185,
      "epoch": 1.5420913696289064e-05,
      "step": 16170
    },
    {
      "action_loss_avg": 0.003345840866677463,
      "epoch": 1.5420913696289064e-05,
      "step": 16170
    },
    {
      "epoch": 1.5420913696289064e-05,
      "step": 16170,
      "training_step_time": 7.583177804946899
    },
    {
      "epoch": 1.5421867370605468e-05,
      "model_forward_time": 1.7137939929962158,
      "step": 16171
    },
    {
      "epoch": 1.5421867370605468e-05,
      "step": 16171,
      "training_step_time": 7.652812957763672
    },
    {
      "epoch": 1.5422821044921876e-05,
      "model_forward_time": 1.7210659980773926,
      "step": 16172
    },
    {
      "epoch": 1.5422821044921876e-05,
      "step": 16172,
      "training_step_time": 8.117515325546265
    },
    {
      "epoch": 1.542377471923828e-05,
      "model_forward_time": 1.7204275131225586,
      "step": 16173
    },
    {
      "epoch": 1.542377471923828e-05,
      "step": 16173,
      "training_step_time": 7.666276216506958
    },
    {
      "epoch": 1.5424728393554687e-05,
      "model_forward_time": 1.7418529987335205,
      "step": 16174
    },
    {
      "epoch": 1.5424728393554687e-05,
      "step": 16174,
      "training_step_time": 7.798177719116211
    },
    {
      "epoch": 1.5425682067871094e-05,
      "model_forward_time": 1.7227799892425537,
      "step": 16175
    },
    {
      "epoch": 1.5425682067871094e-05,
      "step": 16175,
      "training_step_time": 8.752356052398682
    },
    {
      "epoch": 1.54266357421875e-05,
      "model_forward_time": 2.2626025676727295,
      "step": 16176
    },
    {
      "epoch": 1.54266357421875e-05,
      "step": 16176,
      "training_step_time": 8.300525665283203
    },
    {
      "epoch": 1.5427589416503906e-05,
      "model_forward_time": 1.7181496620178223,
      "step": 16177
    },
    {
      "epoch": 1.5427589416503906e-05,
      "step": 16177,
      "training_step_time": 8.822011232376099
    },
    {
      "epoch": 1.5428543090820313e-05,
      "model_forward_time": 1.7175989151000977,
      "step": 16178
    },
    {
      "epoch": 1.5428543090820313e-05,
      "step": 16178,
      "training_step_time": 7.780133485794067
    },
    {
      "epoch": 1.5429496765136717e-05,
      "model_forward_time": 1.7155330181121826,
      "step": 16179
    },
    {
      "epoch": 1.5429496765136717e-05,
      "step": 16179,
      "training_step_time": 7.6888768672943115
    },
    {
      "epoch": 1.5430450439453125e-05,
      "grad_norm": 0.27125245332717896,
      "learning_rate": 9.662210048392311e-06,
      "loss": 0.0654,
      "step": 16180
    },
    {
      "epoch": 1.5430450439453125e-05,
      "model_forward_time": 1.7210900783538818,
      "step": 16180
    },
    {
      "dynamics_loss_avg": 0.05251377262175083,
      "epoch": 1.5430450439453125e-05,
      "step": 16180
    },
    {
      "action_loss_avg": 0.0035308882215758786,
      "epoch": 1.5430450439453125e-05,
      "step": 16180
    },
    {
      "epoch": 1.5430450439453125e-05,
      "step": 16180,
      "training_step_time": 7.5865209102630615
    },
    {
      "epoch": 1.5431404113769532e-05,
      "model_forward_time": 1.712430477142334,
      "step": 16181
    },
    {
      "epoch": 1.5431404113769532e-05,
      "step": 16181,
      "training_step_time": 8.338603019714355
    },
    {
      "epoch": 1.5432357788085936e-05,
      "model_forward_time": 1.713881254196167,
      "step": 16182
    },
    {
      "epoch": 1.5432357788085936e-05,
      "step": 16182,
      "training_step_time": 7.978036165237427
    },
    {
      "epoch": 1.5433311462402344e-05,
      "model_forward_time": 1.7281343936920166,
      "step": 16183
    },
    {
      "epoch": 1.5433311462402344e-05,
      "step": 16183,
      "training_step_time": 8.831860303878784
    },
    {
      "epoch": 1.543426513671875e-05,
      "model_forward_time": 1.7125415802001953,
      "step": 16184
    },
    {
      "epoch": 1.543426513671875e-05,
      "step": 16184,
      "training_step_time": 7.691833019256592
    },
    {
      "epoch": 1.5435218811035155e-05,
      "model_forward_time": 2.2763612270355225,
      "step": 16185
    },
    {
      "epoch": 1.5435218811035155e-05,
      "step": 16185,
      "training_step_time": 8.850404977798462
    },
    {
      "epoch": 1.5436172485351562e-05,
      "model_forward_time": 1.7210068702697754,
      "step": 16186
    },
    {
      "epoch": 1.5436172485351562e-05,
      "step": 16186,
      "training_step_time": 9.806499242782593
    },
    {
      "epoch": 1.543712615966797e-05,
      "model_forward_time": 1.6899912357330322,
      "step": 16187
    },
    {
      "epoch": 1.543712615966797e-05,
      "step": 16187,
      "training_step_time": 9.22262454032898
    },
    {
      "epoch": 1.5438079833984374e-05,
      "model_forward_time": 1.6996276378631592,
      "step": 16188
    },
    {
      "epoch": 1.5438079833984374e-05,
      "step": 16188,
      "training_step_time": 7.625795602798462
    },
    {
      "epoch": 1.543903350830078e-05,
      "model_forward_time": 1.7074368000030518,
      "step": 16189
    },
    {
      "epoch": 1.543903350830078e-05,
      "step": 16189,
      "training_step_time": 7.571908473968506
    },
    {
      "epoch": 1.543998718261719e-05,
      "grad_norm": 0.27577996253967285,
      "learning_rate": 9.661612362320405e-06,
      "loss": 0.0715,
      "step": 16190
    },
    {
      "epoch": 1.543998718261719e-05,
      "model_forward_time": 1.6991496086120605,
      "step": 16190
    },
    {
      "dynamics_loss_avg": 0.057990416139364245,
      "epoch": 1.543998718261719e-05,
      "step": 16190
    },
    {
      "action_loss_avg": 0.0029681518208235502,
      "epoch": 1.543998718261719e-05,
      "step": 16190
    },
    {
      "epoch": 1.543998718261719e-05,
      "step": 16190,
      "training_step_time": 7.635122060775757
    },
    {
      "epoch": 1.5440940856933593e-05,
      "model_forward_time": 1.7117669582366943,
      "step": 16191
    },
    {
      "epoch": 1.5440940856933593e-05,
      "step": 16191,
      "training_step_time": 8.523732423782349
    },
    {
      "epoch": 1.544189453125e-05,
      "model_forward_time": 1.7046070098876953,
      "step": 16192
    },
    {
      "epoch": 1.544189453125e-05,
      "step": 16192,
      "training_step_time": 7.728323698043823
    },
    {
      "epoch": 1.5442848205566408e-05,
      "model_forward_time": 1.7206242084503174,
      "step": 16193
    },
    {
      "epoch": 1.5442848205566408e-05,
      "step": 16193,
      "training_step_time": 7.696485996246338
    },
    {
      "epoch": 1.544380187988281e-05,
      "model_forward_time": 2.2794508934020996,
      "step": 16194
    },
    {
      "epoch": 1.544380187988281e-05,
      "step": 16194,
      "training_step_time": 7.711424827575684
    },
    {
      "epoch": 1.544475555419922e-05,
      "model_forward_time": 1.7059788703918457,
      "step": 16195
    },
    {
      "epoch": 1.544475555419922e-05,
      "step": 16195,
      "training_step_time": 8.778337717056274
    },
    {
      "epoch": 1.5445709228515626e-05,
      "model_forward_time": 1.712648868560791,
      "step": 16196
    },
    {
      "epoch": 1.5445709228515626e-05,
      "step": 16196,
      "training_step_time": 8.554558038711548
    },
    {
      "epoch": 1.544666290283203e-05,
      "model_forward_time": 1.7084286212921143,
      "step": 16197
    },
    {
      "epoch": 1.544666290283203e-05,
      "step": 16197,
      "training_step_time": 7.603897571563721
    },
    {
      "epoch": 1.5447616577148438e-05,
      "model_forward_time": 1.6933016777038574,
      "step": 16198
    },
    {
      "epoch": 1.5447616577148438e-05,
      "step": 16198,
      "training_step_time": 8.1296067237854
    },
    {
      "epoch": 1.5448570251464845e-05,
      "model_forward_time": 1.7131023406982422,
      "step": 16199
    },
    {
      "epoch": 1.5448570251464845e-05,
      "step": 16199,
      "training_step_time": 8.099685907363892
    },
    {
      "epoch": 1.544952392578125e-05,
      "grad_norm": 0.18999776244163513,
      "learning_rate": 9.661014166461579e-06,
      "loss": 0.0635,
      "step": 16200
    },
    {
      "epoch": 1.544952392578125e-05,
      "model_forward_time": 1.7197420597076416,
      "step": 16200
    },
    {
      "dynamics_loss_avg": 0.06719450913369655,
      "epoch": 1.544952392578125e-05,
      "step": 16200
    },
    {
      "action_loss_avg": 0.005217427900061011,
      "epoch": 1.544952392578125e-05,
      "step": 16200
    },
    {
      "epoch": 1.544952392578125e-05,
      "step": 16200,
      "training_step_time": 8.086266040802002
    },
    {
      "epoch": 1.5450477600097657e-05,
      "model_forward_time": 1.7054119110107422,
      "step": 16201
    },
    {
      "epoch": 1.5450477600097657e-05,
      "step": 16201,
      "training_step_time": 7.681983470916748
    },
    {
      "epoch": 1.5451431274414064e-05,
      "model_forward_time": 1.7121391296386719,
      "step": 16202
    },
    {
      "epoch": 1.5451431274414064e-05,
      "step": 16202,
      "training_step_time": 7.731268882751465
    },
    {
      "epoch": 1.5452384948730468e-05,
      "model_forward_time": 2.276772975921631,
      "step": 16203
    },
    {
      "epoch": 1.5452384948730468e-05,
      "step": 16203,
      "training_step_time": 7.731604814529419
    },
    {
      "epoch": 1.5453338623046876e-05,
      "model_forward_time": 1.7248778343200684,
      "step": 16204
    },
    {
      "epoch": 1.5453338623046876e-05,
      "step": 16204,
      "training_step_time": 7.8156898021698
    },
    {
      "epoch": 1.545429229736328e-05,
      "model_forward_time": 1.7212584018707275,
      "step": 16205
    },
    {
      "epoch": 1.545429229736328e-05,
      "step": 16205,
      "training_step_time": 8.556999444961548
    },
    {
      "epoch": 1.5455245971679687e-05,
      "model_forward_time": 1.7184216976165771,
      "step": 16206
    },
    {
      "epoch": 1.5455245971679687e-05,
      "step": 16206,
      "training_step_time": 7.630048513412476
    },
    {
      "epoch": 1.5456199645996094e-05,
      "model_forward_time": 1.7250993251800537,
      "step": 16207
    },
    {
      "epoch": 1.5456199645996094e-05,
      "step": 16207,
      "training_step_time": 7.286463499069214
    },
    {
      "epoch": 1.54571533203125e-05,
      "model_forward_time": 1.7247743606567383,
      "step": 16208
    },
    {
      "epoch": 1.54571533203125e-05,
      "step": 16208,
      "training_step_time": 7.625706672668457
    },
    {
      "epoch": 1.5458106994628906e-05,
      "model_forward_time": 1.7623822689056396,
      "step": 16209
    },
    {
      "epoch": 1.5458106994628906e-05,
      "step": 16209,
      "training_step_time": 7.884851932525635
    },
    {
      "epoch": 1.5459060668945313e-05,
      "grad_norm": 0.1318093091249466,
      "learning_rate": 9.660415460881251e-06,
      "loss": 0.0624,
      "step": 16210
    },
    {
      "epoch": 1.5459060668945313e-05,
      "model_forward_time": 1.7204933166503906,
      "step": 16210
    },
    {
      "dynamics_loss_avg": 0.07207038588821887,
      "epoch": 1.5459060668945313e-05,
      "step": 16210
    },
    {
      "action_loss_avg": 0.0034565801382996143,
      "epoch": 1.5459060668945313e-05,
      "step": 16210
    },
    {
      "epoch": 1.5459060668945313e-05,
      "step": 16210,
      "training_step_time": 7.733835935592651
    },
    {
      "epoch": 1.5460014343261717e-05,
      "model_forward_time": 1.718531608581543,
      "step": 16211
    },
    {
      "epoch": 1.5460014343261717e-05,
      "step": 16211,
      "training_step_time": 8.30061388015747
    },
    {
      "epoch": 1.5460968017578125e-05,
      "model_forward_time": 2.2474730014801025,
      "step": 16212
    },
    {
      "epoch": 1.5460968017578125e-05,
      "step": 16212,
      "training_step_time": 8.556437969207764
    },
    {
      "epoch": 1.5461921691894532e-05,
      "model_forward_time": 1.7139966487884521,
      "step": 16213
    },
    {
      "epoch": 1.5461921691894532e-05,
      "step": 16213,
      "training_step_time": 8.574117422103882
    },
    {
      "epoch": 1.5462875366210936e-05,
      "model_forward_time": 1.7191162109375,
      "step": 16214
    },
    {
      "epoch": 1.5462875366210936e-05,
      "step": 16214,
      "training_step_time": 7.908030986785889
    },
    {
      "epoch": 1.5463829040527344e-05,
      "model_forward_time": 1.351257562637329,
      "step": 16215
    },
    {
      "epoch": 1.5463829040527344e-05,
      "step": 16215,
      "training_step_time": 7.637747049331665
    },
    {
      "epoch": 1.546478271484375e-05,
      "model_forward_time": 1.7249763011932373,
      "step": 16216
    },
    {
      "epoch": 1.546478271484375e-05,
      "step": 16216,
      "training_step_time": 7.268149137496948
    },
    {
      "epoch": 1.5465736389160155e-05,
      "model_forward_time": 1.7182588577270508,
      "step": 16217
    },
    {
      "epoch": 1.5465736389160155e-05,
      "step": 16217,
      "training_step_time": 8.569341659545898
    },
    {
      "epoch": 1.5466690063476562e-05,
      "model_forward_time": 1.7595665454864502,
      "step": 16218
    },
    {
      "epoch": 1.5466690063476562e-05,
      "step": 16218,
      "training_step_time": 8.704664945602417
    },
    {
      "epoch": 1.546764373779297e-05,
      "model_forward_time": 1.7197697162628174,
      "step": 16219
    },
    {
      "epoch": 1.546764373779297e-05,
      "step": 16219,
      "training_step_time": 8.890563726425171
    },
    {
      "epoch": 1.5468597412109374e-05,
      "grad_norm": 0.14913246035575867,
      "learning_rate": 9.659816245644895e-06,
      "loss": 0.065,
      "step": 16220
    },
    {
      "epoch": 1.5468597412109374e-05,
      "model_forward_time": 1.7134466171264648,
      "step": 16220
    },
    {
      "dynamics_loss_avg": 0.0680272962898016,
      "epoch": 1.5468597412109374e-05,
      "step": 16220
    },
    {
      "action_loss_avg": 0.002776452450780198,
      "epoch": 1.5468597412109374e-05,
      "step": 16220
    },
    {
      "epoch": 1.5468597412109374e-05,
      "step": 16220,
      "training_step_time": 8.775951623916626
    },
    {
      "epoch": 1.546955108642578e-05,
      "model_forward_time": 1.7170462608337402,
      "step": 16221
    },
    {
      "epoch": 1.546955108642578e-05,
      "step": 16221,
      "training_step_time": 7.698546409606934
    },
    {
      "epoch": 1.547050476074219e-05,
      "model_forward_time": 1.8918378353118896,
      "step": 16222
    },
    {
      "epoch": 1.547050476074219e-05,
      "step": 16222,
      "training_step_time": 8.378690004348755
    },
    {
      "epoch": 1.5471458435058593e-05,
      "model_forward_time": 1.7084553241729736,
      "step": 16223
    },
    {
      "epoch": 1.5471458435058593e-05,
      "step": 16223,
      "training_step_time": 8.574292421340942
    },
    {
      "epoch": 1.5472412109375e-05,
      "model_forward_time": 1.7250642776489258,
      "step": 16224
    },
    {
      "epoch": 1.5472412109375e-05,
      "step": 16224,
      "training_step_time": 7.762153387069702
    },
    {
      "epoch": 1.5473365783691408e-05,
      "model_forward_time": 1.722172737121582,
      "step": 16225
    },
    {
      "epoch": 1.5473365783691408e-05,
      "step": 16225,
      "training_step_time": 7.4381444454193115
    },
    {
      "epoch": 1.547431945800781e-05,
      "model_forward_time": 1.712951421737671,
      "step": 16226
    },
    {
      "epoch": 1.547431945800781e-05,
      "step": 16226,
      "training_step_time": 7.715868711471558
    },
    {
      "epoch": 1.547527313232422e-05,
      "model_forward_time": 1.7267651557922363,
      "step": 16227
    },
    {
      "epoch": 1.547527313232422e-05,
      "step": 16227,
      "training_step_time": 7.615833044052124
    },
    {
      "epoch": 1.5476226806640626e-05,
      "model_forward_time": 1.7345378398895264,
      "step": 16228
    },
    {
      "epoch": 1.5476226806640626e-05,
      "step": 16228,
      "training_step_time": 7.804334402084351
    },
    {
      "epoch": 1.547718048095703e-05,
      "model_forward_time": 1.7186930179595947,
      "step": 16229
    },
    {
      "epoch": 1.547718048095703e-05,
      "step": 16229,
      "training_step_time": 8.990164518356323
    },
    {
      "epoch": 1.5478134155273438e-05,
      "grad_norm": 0.23236188292503357,
      "learning_rate": 9.65921652081804e-06,
      "loss": 0.0587,
      "step": 16230
    },
    {
      "epoch": 1.5478134155273438e-05,
      "model_forward_time": 1.7264385223388672,
      "step": 16230
    },
    {
      "dynamics_loss_avg": 0.051023033633828166,
      "epoch": 1.5478134155273438e-05,
      "step": 16230
    },
    {
      "action_loss_avg": 0.003108822298236191,
      "epoch": 1.5478134155273438e-05,
      "step": 16230
    },
    {
      "epoch": 1.5478134155273438e-05,
      "step": 16230,
      "training_step_time": 8.435010194778442
    },
    {
      "epoch": 1.5479087829589845e-05,
      "model_forward_time": 2.2776269912719727,
      "step": 16231
    },
    {
      "epoch": 1.5479087829589845e-05,
      "step": 16231,
      "training_step_time": 7.591811656951904
    },
    {
      "epoch": 1.548004150390625e-05,
      "model_forward_time": 1.7020373344421387,
      "step": 16232
    },
    {
      "epoch": 1.548004150390625e-05,
      "step": 16232,
      "training_step_time": 7.751338005065918
    },
    {
      "epoch": 1.5480995178222657e-05,
      "model_forward_time": 1.7277238368988037,
      "step": 16233
    },
    {
      "epoch": 1.5480995178222657e-05,
      "step": 16233,
      "training_step_time": 7.680554389953613
    },
    {
      "epoch": 1.5481948852539064e-05,
      "model_forward_time": 1.7231669425964355,
      "step": 16234
    },
    {
      "epoch": 1.5481948852539064e-05,
      "step": 16234,
      "training_step_time": 7.484507322311401
    },
    {
      "epoch": 1.5482902526855468e-05,
      "model_forward_time": 1.7129664421081543,
      "step": 16235
    },
    {
      "epoch": 1.5482902526855468e-05,
      "step": 16235,
      "training_step_time": 7.592661142349243
    },
    {
      "epoch": 1.5483856201171876e-05,
      "model_forward_time": 1.723327875137329,
      "step": 16236
    },
    {
      "epoch": 1.5483856201171876e-05,
      "step": 16236,
      "training_step_time": 8.093235492706299
    },
    {
      "epoch": 1.548480987548828e-05,
      "model_forward_time": 1.7162208557128906,
      "step": 16237
    },
    {
      "epoch": 1.548480987548828e-05,
      "step": 16237,
      "training_step_time": 9.325555801391602
    },
    {
      "epoch": 1.5485763549804687e-05,
      "model_forward_time": 1.725464105606079,
      "step": 16238
    },
    {
      "epoch": 1.5485763549804687e-05,
      "step": 16238,
      "training_step_time": 7.616034984588623
    },
    {
      "epoch": 1.5486717224121094e-05,
      "model_forward_time": 1.734480857849121,
      "step": 16239
    },
    {
      "epoch": 1.5486717224121094e-05,
      "step": 16239,
      "training_step_time": 9.007423400878906
    },
    {
      "epoch": 1.54876708984375e-05,
      "grad_norm": 0.23610438406467438,
      "learning_rate": 9.658616286466271e-06,
      "loss": 0.0646,
      "step": 16240
    },
    {
      "epoch": 1.54876708984375e-05,
      "model_forward_time": 2.264449119567871,
      "step": 16240
    },
    {
      "dynamics_loss_avg": 0.06714060194790364,
      "epoch": 1.54876708984375e-05,
      "step": 16240
    },
    {
      "action_loss_avg": 0.004071615659631789,
      "epoch": 1.54876708984375e-05,
      "step": 16240
    },
    {
      "epoch": 1.54876708984375e-05,
      "step": 16240,
      "training_step_time": 7.59466552734375
    },
    {
      "epoch": 1.5488624572753906e-05,
      "model_forward_time": 1.713305950164795,
      "step": 16241
    },
    {
      "epoch": 1.5488624572753906e-05,
      "step": 16241,
      "training_step_time": 8.670165300369263
    },
    {
      "epoch": 1.5489578247070313e-05,
      "model_forward_time": 1.721207618713379,
      "step": 16242
    },
    {
      "epoch": 1.5489578247070313e-05,
      "step": 16242,
      "training_step_time": 7.660841941833496
    },
    {
      "epoch": 1.5490531921386717e-05,
      "model_forward_time": 1.726607084274292,
      "step": 16243
    },
    {
      "epoch": 1.5490531921386717e-05,
      "step": 16243,
      "training_step_time": 7.439682245254517
    },
    {
      "epoch": 1.5491485595703125e-05,
      "model_forward_time": 1.7196228504180908,
      "step": 16244
    },
    {
      "epoch": 1.5491485595703125e-05,
      "step": 16244,
      "training_step_time": 7.52575421333313
    },
    {
      "epoch": 1.5492439270019532e-05,
      "model_forward_time": 1.7154967784881592,
      "step": 16245
    },
    {
      "epoch": 1.5492439270019532e-05,
      "step": 16245,
      "training_step_time": 8.235745906829834
    },
    {
      "epoch": 1.5493392944335936e-05,
      "model_forward_time": 1.7665376663208008,
      "step": 16246
    },
    {
      "epoch": 1.5493392944335936e-05,
      "step": 16246,
      "training_step_time": 7.8770811557769775
    },
    {
      "epoch": 1.5494346618652344e-05,
      "model_forward_time": 1.7128193378448486,
      "step": 16247
    },
    {
      "epoch": 1.5494346618652344e-05,
      "step": 16247,
      "training_step_time": 7.661913156509399
    },
    {
      "epoch": 1.549530029296875e-05,
      "model_forward_time": 1.7177097797393799,
      "step": 16248
    },
    {
      "epoch": 1.549530029296875e-05,
      "step": 16248,
      "training_step_time": 8.171160459518433
    },
    {
      "epoch": 1.5496253967285155e-05,
      "model_forward_time": 1.7315950393676758,
      "step": 16249
    },
    {
      "epoch": 1.5496253967285155e-05,
      "step": 16249,
      "training_step_time": 8.102537393569946
    },
    {
      "epoch": 1.5497207641601562e-05,
      "grad_norm": 0.17693063616752625,
      "learning_rate": 9.65801554265523e-06,
      "loss": 0.0673,
      "step": 16250
    },
    {
      "epoch": 1.5497207641601562e-05,
      "model_forward_time": 1.7080471515655518,
      "step": 16250
    },
    {
      "dynamics_loss_avg": 0.05073977196589112,
      "epoch": 1.5497207641601562e-05,
      "step": 16250
    },
    {
      "action_loss_avg": 0.0024184532056096943,
      "epoch": 1.5497207641601562e-05,
      "step": 16250
    },
    {
      "epoch": 1.5497207641601562e-05,
      "step": 16250,
      "training_step_time": 8.239580392837524
    },
    {
      "epoch": 1.549816131591797e-05,
      "model_forward_time": 1.7166504859924316,
      "step": 16251
    },
    {
      "epoch": 1.549816131591797e-05,
      "step": 16251,
      "training_step_time": 8.786349534988403
    },
    {
      "epoch": 1.5499114990234374e-05,
      "model_forward_time": 1.7220947742462158,
      "step": 16252
    },
    {
      "epoch": 1.5499114990234374e-05,
      "step": 16252,
      "training_step_time": 7.5879621505737305
    },
    {
      "epoch": 1.550006866455078e-05,
      "model_forward_time": 1.7135136127471924,
      "step": 16253
    },
    {
      "epoch": 1.550006866455078e-05,
      "step": 16253,
      "training_step_time": 7.5268425941467285
    },
    {
      "epoch": 1.550102233886719e-05,
      "model_forward_time": 1.7305738925933838,
      "step": 16254
    },
    {
      "epoch": 1.550102233886719e-05,
      "step": 16254,
      "training_step_time": 8.439509630203247
    },
    {
      "epoch": 1.5501976013183593e-05,
      "model_forward_time": 1.760408639907837,
      "step": 16255
    },
    {
      "epoch": 1.5501976013183593e-05,
      "step": 16255,
      "training_step_time": 8.436643123626709
    },
    {
      "epoch": 1.55029296875e-05,
      "model_forward_time": 1.7159414291381836,
      "step": 16256
    },
    {
      "epoch": 1.55029296875e-05,
      "step": 16256,
      "training_step_time": 8.331744194030762
    },
    {
      "epoch": 1.5503883361816408e-05,
      "model_forward_time": 1.7176432609558105,
      "step": 16257
    },
    {
      "epoch": 1.5503883361816408e-05,
      "step": 16257,
      "training_step_time": 7.640024185180664
    },
    {
      "epoch": 1.550483703613281e-05,
      "model_forward_time": 1.7178068161010742,
      "step": 16258
    },
    {
      "epoch": 1.550483703613281e-05,
      "step": 16258,
      "training_step_time": 8.124438285827637
    },
    {
      "epoch": 1.550579071044922e-05,
      "model_forward_time": 2.276566982269287,
      "step": 16259
    },
    {
      "epoch": 1.550579071044922e-05,
      "step": 16259,
      "training_step_time": 7.7019202709198
    },
    {
      "epoch": 1.5506744384765626e-05,
      "grad_norm": 0.17208948731422424,
      "learning_rate": 9.657414289450613e-06,
      "loss": 0.0662,
      "step": 16260
    },
    {
      "epoch": 1.5506744384765626e-05,
      "model_forward_time": 1.715893268585205,
      "step": 16260
    },
    {
      "dynamics_loss_avg": 0.05053063789382577,
      "epoch": 1.5506744384765626e-05,
      "step": 16260
    },
    {
      "action_loss_avg": 0.0027328951458912344,
      "epoch": 1.5506744384765626e-05,
      "step": 16260
    },
    {
      "epoch": 1.5506744384765626e-05,
      "step": 16260,
      "training_step_time": 8.724256753921509
    },
    {
      "epoch": 1.550769805908203e-05,
      "model_forward_time": 1.7023754119873047,
      "step": 16261
    },
    {
      "epoch": 1.550769805908203e-05,
      "step": 16261,
      "training_step_time": 7.635948181152344
    },
    {
      "epoch": 1.5508651733398438e-05,
      "model_forward_time": 1.7214219570159912,
      "step": 16262
    },
    {
      "epoch": 1.5508651733398438e-05,
      "step": 16262,
      "training_step_time": 7.47580361366272
    },
    {
      "epoch": 1.5509605407714845e-05,
      "model_forward_time": 1.7186124324798584,
      "step": 16263
    },
    {
      "epoch": 1.5509605407714845e-05,
      "step": 16263,
      "training_step_time": 8.667253494262695
    },
    {
      "epoch": 1.551055908203125e-05,
      "model_forward_time": 1.7647054195404053,
      "step": 16264
    },
    {
      "epoch": 1.551055908203125e-05,
      "step": 16264,
      "training_step_time": 7.607619524002075
    },
    {
      "epoch": 1.5511512756347657e-05,
      "model_forward_time": 1.7238879203796387,
      "step": 16265
    },
    {
      "epoch": 1.5511512756347657e-05,
      "step": 16265,
      "training_step_time": 7.956176519393921
    },
    {
      "epoch": 1.551246643066406e-05,
      "model_forward_time": 1.7071115970611572,
      "step": 16266
    },
    {
      "epoch": 1.551246643066406e-05,
      "step": 16266,
      "training_step_time": 8.685565710067749
    },
    {
      "epoch": 1.5513420104980468e-05,
      "model_forward_time": 1.7238683700561523,
      "step": 16267
    },
    {
      "epoch": 1.5513420104980468e-05,
      "step": 16267,
      "training_step_time": 7.588772773742676
    },
    {
      "epoch": 1.5514373779296876e-05,
      "model_forward_time": 2.2634832859039307,
      "step": 16268
    },
    {
      "epoch": 1.5514373779296876e-05,
      "step": 16268,
      "training_step_time": 9.223185062408447
    },
    {
      "epoch": 1.551532745361328e-05,
      "model_forward_time": 1.734410285949707,
      "step": 16269
    },
    {
      "epoch": 1.551532745361328e-05,
      "step": 16269,
      "training_step_time": 8.286920547485352
    },
    {
      "epoch": 1.5516281127929687e-05,
      "grad_norm": 0.2155323326587677,
      "learning_rate": 9.656812526918171e-06,
      "loss": 0.0612,
      "step": 16270
    },
    {
      "epoch": 1.5516281127929687e-05,
      "model_forward_time": 1.7151384353637695,
      "step": 16270
    },
    {
      "dynamics_loss_avg": 0.06316950172185898,
      "epoch": 1.5516281127929687e-05,
      "step": 16270
    },
    {
      "action_loss_avg": 0.0035460135608445854,
      "epoch": 1.5516281127929687e-05,
      "step": 16270
    },
    {
      "epoch": 1.5516281127929687e-05,
      "step": 16270,
      "training_step_time": 7.642315626144409
    },
    {
      "epoch": 1.5517234802246095e-05,
      "model_forward_time": 1.7117702960968018,
      "step": 16271
    },
    {
      "epoch": 1.5517234802246095e-05,
      "step": 16271,
      "training_step_time": 7.521195411682129
    },
    {
      "epoch": 1.55181884765625e-05,
      "model_forward_time": 1.7265067100524902,
      "step": 16272
    },
    {
      "epoch": 1.55181884765625e-05,
      "step": 16272,
      "training_step_time": 7.652906179428101
    },
    {
      "epoch": 1.5519142150878906e-05,
      "model_forward_time": 1.3443474769592285,
      "step": 16273
    },
    {
      "epoch": 1.5519142150878906e-05,
      "step": 16273,
      "training_step_time": 8.443227052688599
    },
    {
      "epoch": 1.5520095825195313e-05,
      "model_forward_time": 1.7188715934753418,
      "step": 16274
    },
    {
      "epoch": 1.5520095825195313e-05,
      "step": 16274,
      "training_step_time": 8.44908881187439
    },
    {
      "epoch": 1.5521049499511717e-05,
      "model_forward_time": 1.7235198020935059,
      "step": 16275
    },
    {
      "epoch": 1.5521049499511717e-05,
      "step": 16275,
      "training_step_time": 8.666078567504883
    },
    {
      "epoch": 1.5522003173828125e-05,
      "model_forward_time": 1.7199244499206543,
      "step": 16276
    },
    {
      "epoch": 1.5522003173828125e-05,
      "step": 16276,
      "training_step_time": 7.600693702697754
    },
    {
      "epoch": 1.5522956848144532e-05,
      "model_forward_time": 2.2756783962249756,
      "step": 16277
    },
    {
      "epoch": 1.5522956848144532e-05,
      "step": 16277,
      "training_step_time": 8.219599723815918
    },
    {
      "epoch": 1.5523910522460936e-05,
      "model_forward_time": 1.720461368560791,
      "step": 16278
    },
    {
      "epoch": 1.5523910522460936e-05,
      "step": 16278,
      "training_step_time": 7.76394510269165
    },
    {
      "epoch": 1.5524864196777344e-05,
      "model_forward_time": 1.7248077392578125,
      "step": 16279
    },
    {
      "epoch": 1.5524864196777344e-05,
      "step": 16279,
      "training_step_time": 7.7617576122283936
    },
    {
      "epoch": 1.552581787109375e-05,
      "grad_norm": 0.1313798725605011,
      "learning_rate": 9.656210255123713e-06,
      "loss": 0.0635,
      "step": 16280
    },
    {
      "epoch": 1.552581787109375e-05,
      "model_forward_time": 1.7124934196472168,
      "step": 16280
    },
    {
      "dynamics_loss_avg": 0.061375385709106924,
      "epoch": 1.552581787109375e-05,
      "step": 16280
    },
    {
      "action_loss_avg": 0.0027229232946410775,
      "epoch": 1.552581787109375e-05,
      "step": 16280
    },
    {
      "epoch": 1.552581787109375e-05,
      "step": 16280,
      "training_step_time": 8.361268758773804
    },
    {
      "epoch": 1.5526771545410155e-05,
      "model_forward_time": 1.7200756072998047,
      "step": 16281
    },
    {
      "epoch": 1.5526771545410155e-05,
      "step": 16281,
      "training_step_time": 7.494186878204346
    },
    {
      "epoch": 1.5527725219726563e-05,
      "model_forward_time": 1.745408296585083,
      "step": 16282
    },
    {
      "epoch": 1.5527725219726563e-05,
      "step": 16282,
      "training_step_time": 7.643758773803711
    },
    {
      "epoch": 1.552867889404297e-05,
      "model_forward_time": 1.7165570259094238,
      "step": 16283
    },
    {
      "epoch": 1.552867889404297e-05,
      "step": 16283,
      "training_step_time": 7.88527512550354
    },
    {
      "epoch": 1.5529632568359374e-05,
      "model_forward_time": 1.7157137393951416,
      "step": 16284
    },
    {
      "epoch": 1.5529632568359374e-05,
      "step": 16284,
      "training_step_time": 8.33521842956543
    },
    {
      "epoch": 1.553058624267578e-05,
      "model_forward_time": 1.7158832550048828,
      "step": 16285
    },
    {
      "epoch": 1.553058624267578e-05,
      "step": 16285,
      "training_step_time": 8.249942779541016
    },
    {
      "epoch": 1.553153991699219e-05,
      "model_forward_time": 2.2899837493896484,
      "step": 16286
    },
    {
      "epoch": 1.553153991699219e-05,
      "step": 16286,
      "training_step_time": 8.12085247039795
    },
    {
      "epoch": 1.5532493591308593e-05,
      "model_forward_time": 1.7371461391448975,
      "step": 16287
    },
    {
      "epoch": 1.5532493591308593e-05,
      "step": 16287,
      "training_step_time": 8.586837768554688
    },
    {
      "epoch": 1.5533447265625e-05,
      "model_forward_time": 1.73146390914917,
      "step": 16288
    },
    {
      "epoch": 1.5533447265625e-05,
      "step": 16288,
      "training_step_time": 7.689876556396484
    },
    {
      "epoch": 1.5534400939941408e-05,
      "model_forward_time": 1.7285783290863037,
      "step": 16289
    },
    {
      "epoch": 1.5534400939941408e-05,
      "step": 16289,
      "training_step_time": 7.636514663696289
    },
    {
      "epoch": 1.553535461425781e-05,
      "grad_norm": 0.1493724137544632,
      "learning_rate": 9.655607474133101e-06,
      "loss": 0.063,
      "step": 16290
    },
    {
      "epoch": 1.553535461425781e-05,
      "model_forward_time": 1.7328157424926758,
      "step": 16290
    },
    {
      "dynamics_loss_avg": 0.0693532969802618,
      "epoch": 1.553535461425781e-05,
      "step": 16290
    },
    {
      "action_loss_avg": 0.0026611955894622953,
      "epoch": 1.553535461425781e-05,
      "step": 16290
    },
    {
      "epoch": 1.553535461425781e-05,
      "step": 16290,
      "training_step_time": 7.503523588180542
    },
    {
      "epoch": 1.553630828857422e-05,
      "model_forward_time": 1.7294011116027832,
      "step": 16291
    },
    {
      "epoch": 1.553630828857422e-05,
      "step": 16291,
      "training_step_time": 9.647176742553711
    },
    {
      "epoch": 1.5537261962890627e-05,
      "model_forward_time": 1.730154275894165,
      "step": 16292
    },
    {
      "epoch": 1.5537261962890627e-05,
      "step": 16292,
      "training_step_time": 9.530256509780884
    },
    {
      "epoch": 1.553821563720703e-05,
      "model_forward_time": 1.6990492343902588,
      "step": 16293
    },
    {
      "epoch": 1.553821563720703e-05,
      "step": 16293,
      "training_step_time": 7.679286003112793
    },
    {
      "epoch": 1.5539169311523438e-05,
      "model_forward_time": 1.716148853302002,
      "step": 16294
    },
    {
      "epoch": 1.5539169311523438e-05,
      "step": 16294,
      "training_step_time": 7.7273054122924805
    },
    {
      "epoch": 1.5540122985839845e-05,
      "model_forward_time": 1.716275691986084,
      "step": 16295
    },
    {
      "epoch": 1.5540122985839845e-05,
      "step": 16295,
      "training_step_time": 7.690005540847778
    },
    {
      "epoch": 1.554107666015625e-05,
      "model_forward_time": 2.2640979290008545,
      "step": 16296
    },
    {
      "epoch": 1.554107666015625e-05,
      "step": 16296,
      "training_step_time": 7.702738523483276
    },
    {
      "epoch": 1.5542030334472657e-05,
      "model_forward_time": 1.730799913406372,
      "step": 16297
    },
    {
      "epoch": 1.5542030334472657e-05,
      "step": 16297,
      "training_step_time": 8.380137920379639
    },
    {
      "epoch": 1.554298400878906e-05,
      "model_forward_time": 1.7186713218688965,
      "step": 16298
    },
    {
      "epoch": 1.554298400878906e-05,
      "step": 16298,
      "training_step_time": 7.713016748428345
    },
    {
      "epoch": 1.5543937683105468e-05,
      "model_forward_time": 1.7229301929473877,
      "step": 16299
    },
    {
      "epoch": 1.5543937683105468e-05,
      "step": 16299,
      "training_step_time": 7.509077787399292
    },
    {
      "epoch": 1.5544891357421876e-05,
      "grad_norm": 0.2147361785173416,
      "learning_rate": 9.655004184012256e-06,
      "loss": 0.0646,
      "step": 16300
    },
    {
      "epoch": 1.5544891357421876e-05,
      "model_forward_time": 1.7220425605773926,
      "step": 16300
    },
    {
      "dynamics_loss_avg": 0.06069435346871614,
      "epoch": 1.5544891357421876e-05,
      "step": 16300
    },
    {
      "action_loss_avg": 0.003079689567675814,
      "epoch": 1.5544891357421876e-05,
      "step": 16300
    },
    {
      "epoch": 1.5544891357421876e-05,
      "step": 16300,
      "training_step_time": 7.5584893226623535
    },
    {
      "epoch": 1.554584503173828e-05,
      "model_forward_time": 1.7200593948364258,
      "step": 16301
    },
    {
      "epoch": 1.554584503173828e-05,
      "step": 16301,
      "training_step_time": 7.620469570159912
    },
    {
      "epoch": 1.5546798706054687e-05,
      "model_forward_time": 1.735884428024292,
      "step": 16302
    },
    {
      "epoch": 1.5546798706054687e-05,
      "step": 16302,
      "training_step_time": 10.193567276000977
    },
    {
      "epoch": 1.5547752380371095e-05,
      "model_forward_time": 1.73716402053833,
      "step": 16303
    },
    {
      "epoch": 1.5547752380371095e-05,
      "step": 16303,
      "training_step_time": 8.671670913696289
    },
    {
      "epoch": 1.55487060546875e-05,
      "model_forward_time": 1.7344186305999756,
      "step": 16304
    },
    {
      "epoch": 1.55487060546875e-05,
      "step": 16304,
      "training_step_time": 6.716011047363281
    },
    {
      "epoch": 1.5549659729003906e-05,
      "model_forward_time": 1.7283844947814941,
      "step": 16305
    },
    {
      "epoch": 1.5549659729003906e-05,
      "step": 16305,
      "training_step_time": 6.715064525604248
    },
    {
      "epoch": 1.5550613403320313e-05,
      "model_forward_time": 1.7360949516296387,
      "step": 16306
    },
    {
      "epoch": 1.5550613403320313e-05,
      "step": 16306,
      "training_step_time": 6.759213447570801
    },
    {
      "epoch": 1.5551567077636717e-05,
      "model_forward_time": 1.7359423637390137,
      "step": 16307
    },
    {
      "epoch": 1.5551567077636717e-05,
      "step": 16307,
      "training_step_time": 6.735589504241943
    },
    {
      "epoch": 1.5552520751953125e-05,
      "model_forward_time": 1.7392728328704834,
      "step": 16308
    },
    {
      "epoch": 1.5552520751953125e-05,
      "step": 16308,
      "training_step_time": 6.722297668457031
    },
    {
      "epoch": 1.5553474426269532e-05,
      "model_forward_time": 1.7195367813110352,
      "step": 16309
    },
    {
      "epoch": 1.5553474426269532e-05,
      "step": 16309,
      "training_step_time": 6.725045680999756
    },
    {
      "epoch": 1.5554428100585936e-05,
      "grad_norm": 0.23134258389472961,
      "learning_rate": 9.654400384827151e-06,
      "loss": 0.0704,
      "step": 16310
    },
    {
      "epoch": 1.5554428100585936e-05,
      "model_forward_time": 1.7296419143676758,
      "step": 16310
    },
    {
      "dynamics_loss_avg": 0.056324197351932524,
      "epoch": 1.5554428100585936e-05,
      "step": 16310
    },
    {
      "action_loss_avg": 0.002593168360181153,
      "epoch": 1.5554428100585936e-05,
      "step": 16310
    },
    {
      "epoch": 1.5554428100585936e-05,
      "step": 16310,
      "training_step_time": 6.731969594955444
    },
    {
      "epoch": 1.5555381774902344e-05,
      "model_forward_time": 1.740868091583252,
      "step": 16311
    },
    {
      "epoch": 1.5555381774902344e-05,
      "step": 16311,
      "training_step_time": 6.721536159515381
    },
    {
      "epoch": 1.555633544921875e-05,
      "model_forward_time": 2.864902973175049,
      "step": 16312
    },
    {
      "epoch": 1.555633544921875e-05,
      "step": 16312,
      "training_step_time": 11.164999008178711
    },
    {
      "epoch": 1.5557289123535155e-05,
      "model_forward_time": 1.7442474365234375,
      "step": 16313
    },
    {
      "epoch": 1.5557289123535155e-05,
      "step": 16313,
      "training_step_time": 7.652478456497192
    },
    {
      "epoch": 1.5558242797851563e-05,
      "model_forward_time": 1.7418262958526611,
      "step": 16314
    },
    {
      "epoch": 1.5558242797851563e-05,
      "step": 16314,
      "training_step_time": 6.745031118392944
    },
    {
      "epoch": 1.555919647216797e-05,
      "model_forward_time": 1.740825891494751,
      "step": 16315
    },
    {
      "epoch": 1.555919647216797e-05,
      "step": 16315,
      "training_step_time": 6.816829442977905
    },
    {
      "epoch": 1.5560150146484374e-05,
      "model_forward_time": 1.7367103099822998,
      "step": 16316
    },
    {
      "epoch": 1.5560150146484374e-05,
      "step": 16316,
      "training_step_time": 6.7458837032318115
    },
    {
      "epoch": 1.556110382080078e-05,
      "model_forward_time": 1.7383615970611572,
      "step": 16317
    },
    {
      "epoch": 1.556110382080078e-05,
      "step": 16317,
      "training_step_time": 6.743522882461548
    },
    {
      "epoch": 1.556205749511719e-05,
      "model_forward_time": 1.74357271194458,
      "step": 16318
    },
    {
      "epoch": 1.556205749511719e-05,
      "step": 16318,
      "training_step_time": 6.796694040298462
    },
    {
      "epoch": 1.5563011169433593e-05,
      "model_forward_time": 1.7326538562774658,
      "step": 16319
    },
    {
      "epoch": 1.5563011169433593e-05,
      "step": 16319,
      "training_step_time": 6.746176719665527
    },
    {
      "epoch": 1.556396484375e-05,
      "grad_norm": 0.2198183536529541,
      "learning_rate": 9.653796076643818e-06,
      "loss": 0.0672,
      "step": 16320
    },
    {
      "epoch": 1.556396484375e-05,
      "model_forward_time": 1.7381668090820312,
      "step": 16320
    },
    {
      "dynamics_loss_avg": 0.06525720246136188,
      "epoch": 1.556396484375e-05,
      "step": 16320
    },
    {
      "action_loss_avg": 0.003742309962399304,
      "epoch": 1.556396484375e-05,
      "step": 16320
    },
    {
      "epoch": 1.556396484375e-05,
      "step": 16320,
      "training_step_time": 6.712504863739014
    },
    {
      "epoch": 1.5564918518066408e-05,
      "model_forward_time": 2.8664467334747314,
      "step": 16321
    },
    {
      "epoch": 1.5564918518066408e-05,
      "step": 16321,
      "training_step_time": 9.61829161643982
    },
    {
      "epoch": 1.5565872192382812e-05,
      "model_forward_time": 1.742969274520874,
      "step": 16322
    },
    {
      "epoch": 1.5565872192382812e-05,
      "step": 16322,
      "training_step_time": 7.752021312713623
    },
    {
      "epoch": 1.556682586669922e-05,
      "model_forward_time": 1.737377643585205,
      "step": 16323
    },
    {
      "epoch": 1.556682586669922e-05,
      "step": 16323,
      "training_step_time": 6.8095433712005615
    },
    {
      "epoch": 1.5567779541015627e-05,
      "model_forward_time": 1.735705852508545,
      "step": 16324
    },
    {
      "epoch": 1.5567779541015627e-05,
      "step": 16324,
      "training_step_time": 6.749985456466675
    },
    {
      "epoch": 1.556873321533203e-05,
      "model_forward_time": 1.7398474216461182,
      "step": 16325
    },
    {
      "epoch": 1.556873321533203e-05,
      "step": 16325,
      "training_step_time": 6.702077627182007
    },
    {
      "epoch": 1.5569686889648438e-05,
      "model_forward_time": 1.7412745952606201,
      "step": 16326
    },
    {
      "epoch": 1.5569686889648438e-05,
      "step": 16326,
      "training_step_time": 6.726666450500488
    },
    {
      "epoch": 1.5570640563964845e-05,
      "model_forward_time": 1.7340631484985352,
      "step": 16327
    },
    {
      "epoch": 1.5570640563964845e-05,
      "step": 16327,
      "training_step_time": 6.72081732749939
    },
    {
      "epoch": 1.557159423828125e-05,
      "model_forward_time": 1.7356317043304443,
      "step": 16328
    },
    {
      "epoch": 1.557159423828125e-05,
      "step": 16328,
      "training_step_time": 6.708130121231079
    },
    {
      "epoch": 1.5572547912597657e-05,
      "model_forward_time": 1.7453105449676514,
      "step": 16329
    },
    {
      "epoch": 1.5572547912597657e-05,
      "step": 16329,
      "training_step_time": 6.727116823196411
    },
    {
      "epoch": 1.557350158691406e-05,
      "grad_norm": 0.21691174805164337,
      "learning_rate": 9.653191259528345e-06,
      "loss": 0.0637,
      "step": 16330
    },
    {
      "epoch": 1.557350158691406e-05,
      "model_forward_time": 2.877694845199585,
      "step": 16330
    },
    {
      "dynamics_loss_avg": 0.0470423486083746,
      "epoch": 1.557350158691406e-05,
      "step": 16330
    },
    {
      "action_loss_avg": 0.0018819865770637989,
      "epoch": 1.557350158691406e-05,
      "step": 16330
    },
    {
      "epoch": 1.557350158691406e-05,
      "step": 16330,
      "training_step_time": 7.590263605117798
    },
    {
      "epoch": 1.5574455261230468e-05,
      "model_forward_time": 1.3632721900939941,
      "step": 16331
    },
    {
      "epoch": 1.5574455261230468e-05,
      "step": 16331,
      "training_step_time": 8.699000835418701
    },
    {
      "epoch": 1.5575408935546876e-05,
      "model_forward_time": 1.7451214790344238,
      "step": 16332
    },
    {
      "epoch": 1.5575408935546876e-05,
      "step": 16332,
      "training_step_time": 7.332570314407349
    },
    {
      "epoch": 1.557636260986328e-05,
      "model_forward_time": 1.7447559833526611,
      "step": 16333
    },
    {
      "epoch": 1.557636260986328e-05,
      "step": 16333,
      "training_step_time": 6.70245099067688
    },
    {
      "epoch": 1.5577316284179687e-05,
      "model_forward_time": 1.7505104541778564,
      "step": 16334
    },
    {
      "epoch": 1.5577316284179687e-05,
      "step": 16334,
      "training_step_time": 6.7098705768585205
    },
    {
      "epoch": 1.5578269958496095e-05,
      "model_forward_time": 1.7457644939422607,
      "step": 16335
    },
    {
      "epoch": 1.5578269958496095e-05,
      "step": 16335,
      "training_step_time": 6.6966094970703125
    },
    {
      "epoch": 1.55792236328125e-05,
      "model_forward_time": 1.7494375705718994,
      "step": 16336
    },
    {
      "epoch": 1.55792236328125e-05,
      "step": 16336,
      "training_step_time": 6.749176263809204
    },
    {
      "epoch": 1.5580177307128906e-05,
      "model_forward_time": 1.7419261932373047,
      "step": 16337
    },
    {
      "epoch": 1.5580177307128906e-05,
      "step": 16337,
      "training_step_time": 6.721613645553589
    },
    {
      "epoch": 1.5581130981445313e-05,
      "model_forward_time": 1.7471821308135986,
      "step": 16338
    },
    {
      "epoch": 1.5581130981445313e-05,
      "step": 16338,
      "training_step_time": 6.737049579620361
    },
    {
      "epoch": 1.5582084655761717e-05,
      "model_forward_time": 2.8638648986816406,
      "step": 16339
    },
    {
      "epoch": 1.5582084655761717e-05,
      "step": 16339,
      "training_step_time": 8.151498794555664
    },
    {
      "epoch": 1.5583038330078125e-05,
      "grad_norm": 0.24031537771224976,
      "learning_rate": 9.65258593354687e-06,
      "loss": 0.0704,
      "step": 16340
    },
    {
      "epoch": 1.5583038330078125e-05,
      "model_forward_time": 1.7461216449737549,
      "step": 16340
    },
    {
      "dynamics_loss_avg": 0.0698733726516366,
      "epoch": 1.5583038330078125e-05,
      "step": 16340
    },
    {
      "action_loss_avg": 0.0045000440499279645,
      "epoch": 1.5583038330078125e-05,
      "step": 16340
    },
    {
      "epoch": 1.5583038330078125e-05,
      "step": 16340,
      "training_step_time": 8.792548656463623
    },
    {
      "epoch": 1.5583992004394532e-05,
      "model_forward_time": 1.7462387084960938,
      "step": 16341
    },
    {
      "epoch": 1.5583992004394532e-05,
      "step": 16341,
      "training_step_time": 7.662472248077393
    },
    {
      "epoch": 1.5584945678710936e-05,
      "model_forward_time": 1.7440662384033203,
      "step": 16342
    },
    {
      "epoch": 1.5584945678710936e-05,
      "step": 16342,
      "training_step_time": 6.782662391662598
    },
    {
      "epoch": 1.5585899353027344e-05,
      "model_forward_time": 1.7404792308807373,
      "step": 16343
    },
    {
      "epoch": 1.5585899353027344e-05,
      "step": 16343,
      "training_step_time": 6.723195314407349
    },
    {
      "epoch": 1.558685302734375e-05,
      "model_forward_time": 1.7286620140075684,
      "step": 16344
    },
    {
      "epoch": 1.558685302734375e-05,
      "step": 16344,
      "training_step_time": 6.772265911102295
    },
    {
      "epoch": 1.5587806701660155e-05,
      "model_forward_time": 1.7437653541564941,
      "step": 16345
    },
    {
      "epoch": 1.5587806701660155e-05,
      "step": 16345,
      "training_step_time": 6.766634941101074
    },
    {
      "epoch": 1.5588760375976563e-05,
      "model_forward_time": 1.7840707302093506,
      "step": 16346
    },
    {
      "epoch": 1.5588760375976563e-05,
      "step": 16346,
      "training_step_time": 6.795435667037964
    },
    {
      "epoch": 1.558971405029297e-05,
      "model_forward_time": 1.7453510761260986,
      "step": 16347
    },
    {
      "epoch": 1.558971405029297e-05,
      "step": 16347,
      "training_step_time": 6.783262014389038
    },
    {
      "epoch": 1.5590667724609374e-05,
      "model_forward_time": 1.7502307891845703,
      "step": 16348
    },
    {
      "epoch": 1.5590667724609374e-05,
      "step": 16348,
      "training_step_time": 7.266803741455078
    },
    {
      "epoch": 1.559162139892578e-05,
      "model_forward_time": 2.9023244380950928,
      "step": 16349
    },
    {
      "epoch": 1.559162139892578e-05,
      "step": 16349,
      "training_step_time": 11.484302520751953
    },
    {
      "epoch": 1.559257507324219e-05,
      "grad_norm": 0.27388259768486023,
      "learning_rate": 9.651980098765592e-06,
      "loss": 0.0635,
      "step": 16350
    },
    {
      "epoch": 1.559257507324219e-05,
      "model_forward_time": 1.7476918697357178,
      "step": 16350
    },
    {
      "dynamics_loss_avg": 0.06019205078482628,
      "epoch": 1.559257507324219e-05,
      "step": 16350
    },
    {
      "action_loss_avg": 0.0029366585076786578,
      "epoch": 1.559257507324219e-05,
      "step": 16350
    },
    {
      "epoch": 1.559257507324219e-05,
      "step": 16350,
      "training_step_time": 7.784130811691284
    },
    {
      "epoch": 1.5593528747558593e-05,
      "model_forward_time": 1.7513148784637451,
      "step": 16351
    },
    {
      "epoch": 1.5593528747558593e-05,
      "step": 16351,
      "training_step_time": 6.732126712799072
    },
    {
      "epoch": 1.5594482421875e-05,
      "model_forward_time": 1.7527809143066406,
      "step": 16352
    },
    {
      "epoch": 1.5594482421875e-05,
      "step": 16352,
      "training_step_time": 6.768577337265015
    },
    {
      "epoch": 1.5595436096191408e-05,
      "model_forward_time": 1.7415871620178223,
      "step": 16353
    },
    {
      "epoch": 1.5595436096191408e-05,
      "step": 16353,
      "training_step_time": 6.6973748207092285
    },
    {
      "epoch": 1.5596389770507812e-05,
      "model_forward_time": 1.7496778964996338,
      "step": 16354
    },
    {
      "epoch": 1.5596389770507812e-05,
      "step": 16354,
      "training_step_time": 6.6947972774505615
    },
    {
      "epoch": 1.559734344482422e-05,
      "model_forward_time": 1.74290132522583,
      "step": 16355
    },
    {
      "epoch": 1.559734344482422e-05,
      "step": 16355,
      "training_step_time": 6.720875263214111
    },
    {
      "epoch": 1.5598297119140627e-05,
      "model_forward_time": 1.7477717399597168,
      "step": 16356
    },
    {
      "epoch": 1.5598297119140627e-05,
      "step": 16356,
      "training_step_time": 6.704038619995117
    },
    {
      "epoch": 1.559925079345703e-05,
      "model_forward_time": 1.74819016456604,
      "step": 16357
    },
    {
      "epoch": 1.559925079345703e-05,
      "step": 16357,
      "training_step_time": 7.2017128467559814
    },
    {
      "epoch": 1.5600204467773438e-05,
      "model_forward_time": 2.866697311401367,
      "step": 16358
    },
    {
      "epoch": 1.5600204467773438e-05,
      "step": 16358,
      "training_step_time": 11.20810842514038
    },
    {
      "epoch": 1.5601158142089845e-05,
      "model_forward_time": 1.7359228134155273,
      "step": 16359
    },
    {
      "epoch": 1.5601158142089845e-05,
      "step": 16359,
      "training_step_time": 7.746354103088379
    },
    {
      "epoch": 1.560211181640625e-05,
      "grad_norm": 0.2391211837530136,
      "learning_rate": 9.651373755250766e-06,
      "loss": 0.0668,
      "step": 16360
    },
    {
      "epoch": 1.560211181640625e-05,
      "model_forward_time": 1.7429907321929932,
      "step": 16360
    },
    {
      "dynamics_loss_avg": 0.05338132847100496,
      "epoch": 1.560211181640625e-05,
      "step": 16360
    },
    {
      "action_loss_avg": 0.002953244559466839,
      "epoch": 1.560211181640625e-05,
      "step": 16360
    },
    {
      "epoch": 1.560211181640625e-05,
      "step": 16360,
      "training_step_time": 7.2678446769714355
    },
    {
      "epoch": 1.5603065490722657e-05,
      "model_forward_time": 1.742586374282837,
      "step": 16361
    },
    {
      "epoch": 1.5603065490722657e-05,
      "step": 16361,
      "training_step_time": 6.687962770462036
    },
    {
      "epoch": 1.560401916503906e-05,
      "model_forward_time": 1.7370073795318604,
      "step": 16362
    },
    {
      "epoch": 1.560401916503906e-05,
      "step": 16362,
      "training_step_time": 6.686257362365723
    },
    {
      "epoch": 1.560497283935547e-05,
      "model_forward_time": 1.7449977397918701,
      "step": 16363
    },
    {
      "epoch": 1.560497283935547e-05,
      "step": 16363,
      "training_step_time": 6.74615216255188
    },
    {
      "epoch": 1.5605926513671876e-05,
      "model_forward_time": 1.7320384979248047,
      "step": 16364
    },
    {
      "epoch": 1.5605926513671876e-05,
      "step": 16364,
      "training_step_time": 6.715051174163818
    },
    {
      "epoch": 1.560688018798828e-05,
      "model_forward_time": 1.7468767166137695,
      "step": 16365
    },
    {
      "epoch": 1.560688018798828e-05,
      "step": 16365,
      "training_step_time": 6.710249185562134
    },
    {
      "epoch": 1.5607833862304687e-05,
      "model_forward_time": 1.7311351299285889,
      "step": 16366
    },
    {
      "epoch": 1.5607833862304687e-05,
      "step": 16366,
      "training_step_time": 7.216193437576294
    },
    {
      "epoch": 1.5608787536621095e-05,
      "model_forward_time": 2.9098377227783203,
      "step": 16367
    },
    {
      "epoch": 1.5608787536621095e-05,
      "step": 16367,
      "training_step_time": 9.555431365966797
    },
    {
      "epoch": 1.56097412109375e-05,
      "model_forward_time": 1.735459327697754,
      "step": 16368
    },
    {
      "epoch": 1.56097412109375e-05,
      "step": 16368,
      "training_step_time": 7.7091875076293945
    },
    {
      "epoch": 1.5610694885253906e-05,
      "model_forward_time": 1.7457733154296875,
      "step": 16369
    },
    {
      "epoch": 1.5610694885253906e-05,
      "step": 16369,
      "training_step_time": 7.6395509243011475
    },
    {
      "epoch": 1.5611648559570313e-05,
      "grad_norm": 0.12496937066316605,
      "learning_rate": 9.650766903068698e-06,
      "loss": 0.0686,
      "step": 16370
    },
    {
      "epoch": 1.5611648559570313e-05,
      "model_forward_time": 1.7374625205993652,
      "step": 16370
    },
    {
      "dynamics_loss_avg": 0.07207798976451159,
      "epoch": 1.5611648559570313e-05,
      "step": 16370
    },
    {
      "action_loss_avg": 0.0036598787526600065,
      "epoch": 1.5611648559570313e-05,
      "step": 16370
    },
    {
      "epoch": 1.5611648559570313e-05,
      "step": 16370,
      "training_step_time": 6.731396913528442
    },
    {
      "epoch": 1.5612602233886717e-05,
      "model_forward_time": 1.7437629699707031,
      "step": 16371
    },
    {
      "epoch": 1.5612602233886717e-05,
      "step": 16371,
      "training_step_time": 6.682501554489136
    },
    {
      "epoch": 1.5613555908203125e-05,
      "model_forward_time": 1.7470993995666504,
      "step": 16372
    },
    {
      "epoch": 1.5613555908203125e-05,
      "step": 16372,
      "training_step_time": 6.673792362213135
    },
    {
      "epoch": 1.5614509582519532e-05,
      "model_forward_time": 1.7376306056976318,
      "step": 16373
    },
    {
      "epoch": 1.5614509582519532e-05,
      "step": 16373,
      "training_step_time": 6.745848178863525
    },
    {
      "epoch": 1.5615463256835936e-05,
      "model_forward_time": 1.7370572090148926,
      "step": 16374
    },
    {
      "epoch": 1.5615463256835936e-05,
      "step": 16374,
      "training_step_time": 6.757154941558838
    },
    {
      "epoch": 1.5616416931152344e-05,
      "model_forward_time": 1.7372713088989258,
      "step": 16375
    },
    {
      "epoch": 1.5616416931152344e-05,
      "step": 16375,
      "training_step_time": 7.174933195114136
    },
    {
      "epoch": 1.561737060546875e-05,
      "model_forward_time": 2.879345417022705,
      "step": 16376
    },
    {
      "epoch": 1.561737060546875e-05,
      "step": 16376,
      "training_step_time": 8.144237756729126
    },
    {
      "epoch": 1.5618324279785155e-05,
      "model_forward_time": 1.7781765460968018,
      "step": 16377
    },
    {
      "epoch": 1.5618324279785155e-05,
      "step": 16377,
      "training_step_time": 10.303171396255493
    },
    {
      "epoch": 1.5619277954101563e-05,
      "model_forward_time": 1.7526206970214844,
      "step": 16378
    },
    {
      "epoch": 1.5619277954101563e-05,
      "step": 16378,
      "training_step_time": 7.634690046310425
    },
    {
      "epoch": 1.562023162841797e-05,
      "model_forward_time": 1.7587153911590576,
      "step": 16379
    },
    {
      "epoch": 1.562023162841797e-05,
      "step": 16379,
      "training_step_time": 6.692720890045166
    },
    {
      "epoch": 1.5621185302734374e-05,
      "grad_norm": 0.2006043642759323,
      "learning_rate": 9.650159542285754e-06,
      "loss": 0.067,
      "step": 16380
    },
    {
      "epoch": 1.5621185302734374e-05,
      "model_forward_time": 1.7551801204681396,
      "step": 16380
    },
    {
      "dynamics_loss_avg": 0.06781143173575402,
      "epoch": 1.5621185302734374e-05,
      "step": 16380
    },
    {
      "action_loss_avg": 0.003555422881618142,
      "epoch": 1.5621185302734374e-05,
      "step": 16380
    },
    {
      "epoch": 1.5621185302734374e-05,
      "step": 16380,
      "training_step_time": 6.677929401397705
    },
    {
      "epoch": 1.562213897705078e-05,
      "model_forward_time": 1.7531485557556152,
      "step": 16381
    },
    {
      "epoch": 1.562213897705078e-05,
      "step": 16381,
      "training_step_time": 6.730632066726685
    },
    {
      "epoch": 1.562309265136719e-05,
      "model_forward_time": 1.7562463283538818,
      "step": 16382
    },
    {
      "epoch": 1.562309265136719e-05,
      "step": 16382,
      "training_step_time": 6.715513706207275
    },
    {
      "epoch": 1.5624046325683593e-05,
      "model_forward_time": 1.7465147972106934,
      "step": 16383
    },
    {
      "epoch": 1.5624046325683593e-05,
      "step": 16383,
      "training_step_time": 6.69934344291687
    },
    {
      "epoch": 1.5625e-05,
      "model_forward_time": 1.7477500438690186,
      "step": 16384
    },
    {
      "epoch": 1.5625e-05,
      "step": 16384,
      "training_step_time": 7.20553994178772
    },
    {
      "epoch": 1.5625953674316408e-05,
      "model_forward_time": 2.8748421669006348,
      "step": 16385
    },
    {
      "epoch": 1.5625953674316408e-05,
      "step": 16385,
      "training_step_time": 9.187118768692017
    },
    {
      "epoch": 1.5626907348632812e-05,
      "model_forward_time": 1.764528512954712,
      "step": 16386
    },
    {
      "epoch": 1.5626907348632812e-05,
      "step": 16386,
      "training_step_time": 9.605880975723267
    },
    {
      "epoch": 1.562786102294922e-05,
      "model_forward_time": 1.7516920566558838,
      "step": 16387
    },
    {
      "epoch": 1.562786102294922e-05,
      "step": 16387,
      "training_step_time": 7.608829021453857
    },
    {
      "epoch": 1.5628814697265627e-05,
      "model_forward_time": 1.7564949989318848,
      "step": 16388
    },
    {
      "epoch": 1.5628814697265627e-05,
      "step": 16388,
      "training_step_time": 7.625371932983398
    },
    {
      "epoch": 1.562976837158203e-05,
      "model_forward_time": 1.7518179416656494,
      "step": 16389
    },
    {
      "epoch": 1.562976837158203e-05,
      "step": 16389,
      "training_step_time": 6.733712434768677
    },
    {
      "epoch": 1.5630722045898438e-05,
      "grad_norm": 0.2514289319515228,
      "learning_rate": 9.649551672968353e-06,
      "loss": 0.0675,
      "step": 16390
    },
    {
      "epoch": 1.5630722045898438e-05,
      "model_forward_time": 1.7615315914154053,
      "step": 16390
    },
    {
      "dynamics_loss_avg": 0.08312299735844135,
      "epoch": 1.5630722045898438e-05,
      "step": 16390
    },
    {
      "action_loss_avg": 0.004480442241765558,
      "epoch": 1.5630722045898438e-05,
      "step": 16390
    },
    {
      "epoch": 1.5630722045898438e-05,
      "step": 16390,
      "training_step_time": 6.744922161102295
    },
    {
      "epoch": 1.5631675720214842e-05,
      "model_forward_time": 1.7525651454925537,
      "step": 16391
    },
    {
      "epoch": 1.5631675720214842e-05,
      "step": 16391,
      "training_step_time": 6.7237043380737305
    },
    {
      "epoch": 1.563262939453125e-05,
      "model_forward_time": 1.7504818439483643,
      "step": 16392
    },
    {
      "epoch": 1.563262939453125e-05,
      "step": 16392,
      "training_step_time": 6.738388538360596
    },
    {
      "epoch": 1.5633583068847657e-05,
      "model_forward_time": 1.7553658485412598,
      "step": 16393
    },
    {
      "epoch": 1.5633583068847657e-05,
      "step": 16393,
      "training_step_time": 7.244281053543091
    },
    {
      "epoch": 1.563453674316406e-05,
      "model_forward_time": 1.7618072032928467,
      "step": 16394
    },
    {
      "epoch": 1.563453674316406e-05,
      "step": 16394,
      "training_step_time": 8.076359987258911
    },
    {
      "epoch": 1.563549041748047e-05,
      "model_forward_time": 1.8377833366394043,
      "step": 16395
    },
    {
      "epoch": 1.563549041748047e-05,
      "step": 16395,
      "training_step_time": 9.095510721206665
    },
    {
      "epoch": 1.5636444091796876e-05,
      "model_forward_time": 1.7484204769134521,
      "step": 16396
    },
    {
      "epoch": 1.5636444091796876e-05,
      "step": 16396,
      "training_step_time": 7.680384635925293
    },
    {
      "epoch": 1.563739776611328e-05,
      "model_forward_time": 1.7586538791656494,
      "step": 16397
    },
    {
      "epoch": 1.563739776611328e-05,
      "step": 16397,
      "training_step_time": 7.640112400054932
    },
    {
      "epoch": 1.5638351440429687e-05,
      "model_forward_time": 1.7545182704925537,
      "step": 16398
    },
    {
      "epoch": 1.5638351440429687e-05,
      "step": 16398,
      "training_step_time": 6.735416650772095
    },
    {
      "epoch": 1.5639305114746095e-05,
      "model_forward_time": 1.7540926933288574,
      "step": 16399
    },
    {
      "epoch": 1.5639305114746095e-05,
      "step": 16399,
      "training_step_time": 6.721999883651733
    },
    {
      "epoch": 1.56402587890625e-05,
      "grad_norm": 0.18091066181659698,
      "learning_rate": 9.648943295182974e-06,
      "loss": 0.0647,
      "step": 16400
    },
    {
      "epoch": 1.56402587890625e-05,
      "model_forward_time": 1.384751796722412,
      "step": 16400
    },
    {
      "dynamics_loss_avg": 0.0638182525523007,
      "epoch": 1.56402587890625e-05,
      "step": 16400
    },
    {
      "action_loss_avg": 0.002867493200028548,
      "epoch": 1.56402587890625e-05,
      "step": 16400
    },
    {
      "epoch": 1.56402587890625e-05,
      "step": 16400,
      "training_step_time": 6.81930947303772
    },
    {
      "epoch": 1.5641212463378906e-05,
      "model_forward_time": 1.7515404224395752,
      "step": 16401
    },
    {
      "epoch": 1.5641212463378906e-05,
      "step": 16401,
      "training_step_time": 6.798200845718384
    },
    {
      "epoch": 1.5642166137695313e-05,
      "model_forward_time": 1.7603471279144287,
      "step": 16402
    },
    {
      "epoch": 1.5642166137695313e-05,
      "step": 16402,
      "training_step_time": 7.212891578674316
    },
    {
      "epoch": 1.5643119812011718e-05,
      "model_forward_time": 1.7486178874969482,
      "step": 16403
    },
    {
      "epoch": 1.5643119812011718e-05,
      "step": 16403,
      "training_step_time": 8.307748317718506
    },
    {
      "epoch": 1.5644073486328125e-05,
      "model_forward_time": 2.8781702518463135,
      "step": 16404
    },
    {
      "epoch": 1.5644073486328125e-05,
      "step": 16404,
      "training_step_time": 10.750541925430298
    },
    {
      "epoch": 1.5645027160644532e-05,
      "model_forward_time": 1.7522318363189697,
      "step": 16405
    },
    {
      "epoch": 1.5645027160644532e-05,
      "step": 16405,
      "training_step_time": 7.673548698425293
    },
    {
      "epoch": 1.5645980834960936e-05,
      "model_forward_time": 1.7526252269744873,
      "step": 16406
    },
    {
      "epoch": 1.5645980834960936e-05,
      "step": 16406,
      "training_step_time": 7.658556222915649
    },
    {
      "epoch": 1.5646934509277344e-05,
      "model_forward_time": 1.7452952861785889,
      "step": 16407
    },
    {
      "epoch": 1.5646934509277344e-05,
      "step": 16407,
      "training_step_time": 6.78469705581665
    },
    {
      "epoch": 1.564788818359375e-05,
      "model_forward_time": 1.7450554370880127,
      "step": 16408
    },
    {
      "epoch": 1.564788818359375e-05,
      "step": 16408,
      "training_step_time": 6.719617605209351
    },
    {
      "epoch": 1.5648841857910155e-05,
      "model_forward_time": 1.7498443126678467,
      "step": 16409
    },
    {
      "epoch": 1.5648841857910155e-05,
      "step": 16409,
      "training_step_time": 6.739129066467285
    },
    {
      "epoch": 1.5649795532226563e-05,
      "grad_norm": 0.18466046452522278,
      "learning_rate": 9.648334408996145e-06,
      "loss": 0.0698,
      "step": 16410
    },
    {
      "epoch": 1.5649795532226563e-05,
      "model_forward_time": 1.7648663520812988,
      "step": 16410
    },
    {
      "dynamics_loss_avg": 0.06227926313877106,
      "epoch": 1.5649795532226563e-05,
      "step": 16410
    },
    {
      "action_loss_avg": 0.0026853569026570765,
      "epoch": 1.5649795532226563e-05,
      "step": 16410
    },
    {
      "epoch": 1.5649795532226563e-05,
      "step": 16410,
      "training_step_time": 6.7684714794158936
    },
    {
      "epoch": 1.565074920654297e-05,
      "model_forward_time": 1.7473936080932617,
      "step": 16411
    },
    {
      "epoch": 1.565074920654297e-05,
      "step": 16411,
      "training_step_time": 7.225938558578491
    },
    {
      "epoch": 1.5651702880859374e-05,
      "model_forward_time": 1.7551817893981934,
      "step": 16412
    },
    {
      "epoch": 1.5651702880859374e-05,
      "step": 16412,
      "training_step_time": 7.5816969871521
    },
    {
      "epoch": 1.565265655517578e-05,
      "model_forward_time": 2.9170358180999756,
      "step": 16413
    },
    {
      "epoch": 1.565265655517578e-05,
      "step": 16413,
      "training_step_time": 10.587659358978271
    },
    {
      "epoch": 1.565361022949219e-05,
      "model_forward_time": 1.74825119972229,
      "step": 16414
    },
    {
      "epoch": 1.565361022949219e-05,
      "step": 16414,
      "training_step_time": 10.821613073348999
    },
    {
      "epoch": 1.5654563903808593e-05,
      "model_forward_time": 1.7483763694763184,
      "step": 16415
    },
    {
      "epoch": 1.5654563903808593e-05,
      "step": 16415,
      "training_step_time": 7.655428886413574
    },
    {
      "epoch": 1.5655517578125e-05,
      "model_forward_time": 1.752993106842041,
      "step": 16416
    },
    {
      "epoch": 1.5655517578125e-05,
      "step": 16416,
      "training_step_time": 6.728983640670776
    },
    {
      "epoch": 1.5656471252441408e-05,
      "model_forward_time": 1.755225658416748,
      "step": 16417
    },
    {
      "epoch": 1.5656471252441408e-05,
      "step": 16417,
      "training_step_time": 6.722299575805664
    },
    {
      "epoch": 1.5657424926757812e-05,
      "model_forward_time": 1.7535607814788818,
      "step": 16418
    },
    {
      "epoch": 1.5657424926757812e-05,
      "step": 16418,
      "training_step_time": 6.73655366897583
    },
    {
      "epoch": 1.565837860107422e-05,
      "model_forward_time": 1.7589082717895508,
      "step": 16419
    },
    {
      "epoch": 1.565837860107422e-05,
      "step": 16419,
      "training_step_time": 6.762291431427002
    },
    {
      "epoch": 1.5659332275390627e-05,
      "grad_norm": 0.1546744406223297,
      "learning_rate": 9.647725014474452e-06,
      "loss": 0.0697,
      "step": 16420
    },
    {
      "epoch": 1.5659332275390627e-05,
      "model_forward_time": 1.7637238502502441,
      "step": 16420
    },
    {
      "dynamics_loss_avg": 0.059639340825378896,
      "epoch": 1.5659332275390627e-05,
      "step": 16420
    },
    {
      "action_loss_avg": 0.003391091019148007,
      "epoch": 1.5659332275390627e-05,
      "step": 16420
    },
    {
      "epoch": 1.5659332275390627e-05,
      "step": 16420,
      "training_step_time": 7.209136486053467
    },
    {
      "epoch": 1.566028594970703e-05,
      "model_forward_time": 1.7544903755187988,
      "step": 16421
    },
    {
      "epoch": 1.566028594970703e-05,
      "step": 16421,
      "training_step_time": 8.780556201934814
    },
    {
      "epoch": 1.5661239624023438e-05,
      "model_forward_time": 2.918175220489502,
      "step": 16422
    },
    {
      "epoch": 1.5661239624023438e-05,
      "step": 16422,
      "training_step_time": 8.13172698020935
    },
    {
      "epoch": 1.5662193298339842e-05,
      "model_forward_time": 1.7541084289550781,
      "step": 16423
    },
    {
      "epoch": 1.5662193298339842e-05,
      "step": 16423,
      "training_step_time": 8.686052799224854
    },
    {
      "epoch": 1.566314697265625e-05,
      "model_forward_time": 1.7613873481750488,
      "step": 16424
    },
    {
      "epoch": 1.566314697265625e-05,
      "step": 16424,
      "training_step_time": 7.615442276000977
    },
    {
      "epoch": 1.5664100646972657e-05,
      "model_forward_time": 1.7510986328125,
      "step": 16425
    },
    {
      "epoch": 1.5664100646972657e-05,
      "step": 16425,
      "training_step_time": 7.503222942352295
    },
    {
      "epoch": 1.566505432128906e-05,
      "model_forward_time": 1.7616674900054932,
      "step": 16426
    },
    {
      "epoch": 1.566505432128906e-05,
      "step": 16426,
      "training_step_time": 6.703986167907715
    },
    {
      "epoch": 1.566600799560547e-05,
      "model_forward_time": 1.7494916915893555,
      "step": 16427
    },
    {
      "epoch": 1.566600799560547e-05,
      "step": 16427,
      "training_step_time": 6.756159543991089
    },
    {
      "epoch": 1.5666961669921876e-05,
      "model_forward_time": 1.7480885982513428,
      "step": 16428
    },
    {
      "epoch": 1.5666961669921876e-05,
      "step": 16428,
      "training_step_time": 6.774677038192749
    },
    {
      "epoch": 1.566791534423828e-05,
      "model_forward_time": 1.756840467453003,
      "step": 16429
    },
    {
      "epoch": 1.566791534423828e-05,
      "step": 16429,
      "training_step_time": 7.17334771156311
    },
    {
      "epoch": 1.5668869018554687e-05,
      "grad_norm": 0.2157498598098755,
      "learning_rate": 9.64711511168454e-06,
      "loss": 0.0658,
      "step": 16430
    },
    {
      "epoch": 1.5668869018554687e-05,
      "model_forward_time": 1.7517492771148682,
      "step": 16430
    },
    {
      "dynamics_loss_avg": 0.07212560642510653,
      "epoch": 1.5668869018554687e-05,
      "step": 16430
    },
    {
      "action_loss_avg": 0.003785371739650145,
      "epoch": 1.5668869018554687e-05,
      "step": 16430
    },
    {
      "epoch": 1.5668869018554687e-05,
      "step": 16430,
      "training_step_time": 7.584644556045532
    },
    {
      "epoch": 1.5669822692871095e-05,
      "model_forward_time": 2.8367018699645996,
      "step": 16431
    },
    {
      "epoch": 1.5669822692871095e-05,
      "step": 16431,
      "training_step_time": 9.967248678207397
    },
    {
      "epoch": 1.56707763671875e-05,
      "model_forward_time": 1.8566677570343018,
      "step": 16432
    },
    {
      "epoch": 1.56707763671875e-05,
      "step": 16432,
      "training_step_time": 9.71863341331482
    },
    {
      "epoch": 1.5671730041503906e-05,
      "model_forward_time": 1.7586781978607178,
      "step": 16433
    },
    {
      "epoch": 1.5671730041503906e-05,
      "step": 16433,
      "training_step_time": 7.725221395492554
    },
    {
      "epoch": 1.5672683715820314e-05,
      "model_forward_time": 1.7439296245574951,
      "step": 16434
    },
    {
      "epoch": 1.5672683715820314e-05,
      "step": 16434,
      "training_step_time": 7.562941551208496
    },
    {
      "epoch": 1.5673637390136718e-05,
      "model_forward_time": 1.7484948635101318,
      "step": 16435
    },
    {
      "epoch": 1.5673637390136718e-05,
      "step": 16435,
      "training_step_time": 6.778462886810303
    },
    {
      "epoch": 1.5674591064453125e-05,
      "model_forward_time": 1.7550995349884033,
      "step": 16436
    },
    {
      "epoch": 1.5674591064453125e-05,
      "step": 16436,
      "training_step_time": 6.7787041664123535
    },
    {
      "epoch": 1.5675544738769532e-05,
      "model_forward_time": 1.750920295715332,
      "step": 16437
    },
    {
      "epoch": 1.5675544738769532e-05,
      "step": 16437,
      "training_step_time": 6.801609992980957
    },
    {
      "epoch": 1.5676498413085936e-05,
      "model_forward_time": 1.7547399997711182,
      "step": 16438
    },
    {
      "epoch": 1.5676498413085936e-05,
      "step": 16438,
      "training_step_time": 7.233116626739502
    },
    {
      "epoch": 1.5677452087402344e-05,
      "model_forward_time": 1.7567920684814453,
      "step": 16439
    },
    {
      "epoch": 1.5677452087402344e-05,
      "step": 16439,
      "training_step_time": 7.625469207763672
    },
    {
      "epoch": 1.567840576171875e-05,
      "grad_norm": 0.19723929464817047,
      "learning_rate": 9.646504700693107e-06,
      "loss": 0.0721,
      "step": 16440
    },
    {
      "epoch": 1.567840576171875e-05,
      "model_forward_time": 1.7327241897583008,
      "step": 16440
    },
    {
      "dynamics_loss_avg": 0.06533188223838807,
      "epoch": 1.567840576171875e-05,
      "step": 16440
    },
    {
      "action_loss_avg": 0.0029059268301352857,
      "epoch": 1.567840576171875e-05,
      "step": 16440
    },
    {
      "epoch": 1.567840576171875e-05,
      "step": 16440,
      "training_step_time": 7.549252033233643
    },
    {
      "epoch": 1.5679359436035155e-05,
      "model_forward_time": 2.903667688369751,
      "step": 16441
    },
    {
      "epoch": 1.5679359436035155e-05,
      "step": 16441,
      "training_step_time": 10.097302198410034
    },
    {
      "epoch": 1.5680313110351563e-05,
      "model_forward_time": 1.771714210510254,
      "step": 16442
    },
    {
      "epoch": 1.5680313110351563e-05,
      "step": 16442,
      "training_step_time": 7.672747850418091
    },
    {
      "epoch": 1.568126678466797e-05,
      "model_forward_time": 1.7523837089538574,
      "step": 16443
    },
    {
      "epoch": 1.568126678466797e-05,
      "step": 16443,
      "training_step_time": 7.625965356826782
    },
    {
      "epoch": 1.5682220458984374e-05,
      "model_forward_time": 1.7588260173797607,
      "step": 16444
    },
    {
      "epoch": 1.5682220458984374e-05,
      "step": 16444,
      "training_step_time": 6.717784881591797
    },
    {
      "epoch": 1.568317413330078e-05,
      "model_forward_time": 1.7528929710388184,
      "step": 16445
    },
    {
      "epoch": 1.568317413330078e-05,
      "step": 16445,
      "training_step_time": 6.721903085708618
    },
    {
      "epoch": 1.568412780761719e-05,
      "model_forward_time": 1.75758695602417,
      "step": 16446
    },
    {
      "epoch": 1.568412780761719e-05,
      "step": 16446,
      "training_step_time": 6.711204290390015
    },
    {
      "epoch": 1.5685081481933593e-05,
      "model_forward_time": 1.7573962211608887,
      "step": 16447
    },
    {
      "epoch": 1.5685081481933593e-05,
      "step": 16447,
      "training_step_time": 7.2468249797821045
    },
    {
      "epoch": 1.568603515625e-05,
      "model_forward_time": 1.7538070678710938,
      "step": 16448
    },
    {
      "epoch": 1.568603515625e-05,
      "step": 16448,
      "training_step_time": 7.580694675445557
    },
    {
      "epoch": 1.5686988830566408e-05,
      "model_forward_time": 1.760643482208252,
      "step": 16449
    },
    {
      "epoch": 1.5686988830566408e-05,
      "step": 16449,
      "training_step_time": 8.54780650138855
    },
    {
      "epoch": 1.5687942504882812e-05,
      "grad_norm": 0.25840264558792114,
      "learning_rate": 9.645893781566907e-06,
      "loss": 0.0683,
      "step": 16450
    },
    {
      "epoch": 1.5687942504882812e-05,
      "model_forward_time": 2.9733729362487793,
      "step": 16450
    },
    {
      "dynamics_loss_avg": 0.055664783157408235,
      "epoch": 1.5687942504882812e-05,
      "step": 16450
    },
    {
      "action_loss_avg": 0.00427470231661573,
      "epoch": 1.5687942504882812e-05,
      "step": 16450
    },
    {
      "epoch": 1.5687942504882812e-05,
      "step": 16450,
      "training_step_time": 10.18414306640625
    },
    {
      "epoch": 1.568889617919922e-05,
      "model_forward_time": 1.7527027130126953,
      "step": 16451
    },
    {
      "epoch": 1.568889617919922e-05,
      "step": 16451,
      "training_step_time": 7.694671630859375
    },
    {
      "epoch": 1.5689849853515627e-05,
      "model_forward_time": 1.7492127418518066,
      "step": 16452
    },
    {
      "epoch": 1.5689849853515627e-05,
      "step": 16452,
      "training_step_time": 7.623697519302368
    },
    {
      "epoch": 1.569080352783203e-05,
      "model_forward_time": 1.7556912899017334,
      "step": 16453
    },
    {
      "epoch": 1.569080352783203e-05,
      "step": 16453,
      "training_step_time": 7.584162712097168
    },
    {
      "epoch": 1.5691757202148438e-05,
      "model_forward_time": 1.7589690685272217,
      "step": 16454
    },
    {
      "epoch": 1.5691757202148438e-05,
      "step": 16454,
      "training_step_time": 6.745837450027466
    },
    {
      "epoch": 1.5692710876464842e-05,
      "model_forward_time": 1.7554118633270264,
      "step": 16455
    },
    {
      "epoch": 1.5692710876464842e-05,
      "step": 16455,
      "training_step_time": 6.725967884063721
    },
    {
      "epoch": 1.569366455078125e-05,
      "model_forward_time": 1.3938369750976562,
      "step": 16456
    },
    {
      "epoch": 1.569366455078125e-05,
      "step": 16456,
      "training_step_time": 7.168592929840088
    },
    {
      "epoch": 1.5694618225097657e-05,
      "model_forward_time": 1.7190768718719482,
      "step": 16457
    },
    {
      "epoch": 1.5694618225097657e-05,
      "step": 16457,
      "training_step_time": 7.556774139404297
    },
    {
      "epoch": 1.569557189941406e-05,
      "model_forward_time": 1.7439897060394287,
      "step": 16458
    },
    {
      "epoch": 1.569557189941406e-05,
      "step": 16458,
      "training_step_time": 8.168071746826172
    },
    {
      "epoch": 1.569652557373047e-05,
      "model_forward_time": 2.9084999561309814,
      "step": 16459
    },
    {
      "epoch": 1.569652557373047e-05,
      "step": 16459,
      "training_step_time": 8.384483575820923
    },
    {
      "epoch": 1.5697479248046876e-05,
      "grad_norm": 0.22323495149612427,
      "learning_rate": 9.645282354372745e-06,
      "loss": 0.0721,
      "step": 16460
    },
    {
      "epoch": 1.5697479248046876e-05,
      "model_forward_time": 1.7310824394226074,
      "step": 16460
    },
    {
      "dynamics_loss_avg": 0.06688208039849997,
      "epoch": 1.5697479248046876e-05,
      "step": 16460
    },
    {
      "action_loss_avg": 0.00519399103068281,
      "epoch": 1.5697479248046876e-05,
      "step": 16460
    },
    {
      "epoch": 1.5697479248046876e-05,
      "step": 16460,
      "training_step_time": 8.711860179901123
    },
    {
      "epoch": 1.569843292236328e-05,
      "model_forward_time": 1.7509384155273438,
      "step": 16461
    },
    {
      "epoch": 1.569843292236328e-05,
      "step": 16461,
      "training_step_time": 7.678695917129517
    },
    {
      "epoch": 1.5699386596679687e-05,
      "model_forward_time": 1.7494046688079834,
      "step": 16462
    },
    {
      "epoch": 1.5699386596679687e-05,
      "step": 16462,
      "training_step_time": 7.550786018371582
    },
    {
      "epoch": 1.5700340270996095e-05,
      "model_forward_time": 1.7496755123138428,
      "step": 16463
    },
    {
      "epoch": 1.5700340270996095e-05,
      "step": 16463,
      "training_step_time": 6.72959303855896
    },
    {
      "epoch": 1.57012939453125e-05,
      "model_forward_time": 1.7551119327545166,
      "step": 16464
    },
    {
      "epoch": 1.57012939453125e-05,
      "step": 16464,
      "training_step_time": 6.749065637588501
    },
    {
      "epoch": 1.5702247619628906e-05,
      "model_forward_time": 1.7455933094024658,
      "step": 16465
    },
    {
      "epoch": 1.5702247619628906e-05,
      "step": 16465,
      "training_step_time": 6.776171922683716
    },
    {
      "epoch": 1.5703201293945314e-05,
      "model_forward_time": 1.7456004619598389,
      "step": 16466
    },
    {
      "epoch": 1.5703201293945314e-05,
      "step": 16466,
      "training_step_time": 7.601590394973755
    },
    {
      "epoch": 1.5704154968261718e-05,
      "model_forward_time": 1.7233426570892334,
      "step": 16467
    },
    {
      "epoch": 1.5704154968261718e-05,
      "step": 16467,
      "training_step_time": 9.067440748214722
    },
    {
      "epoch": 1.5705108642578125e-05,
      "model_forward_time": 2.862273693084717,
      "step": 16468
    },
    {
      "epoch": 1.5705108642578125e-05,
      "step": 16468,
      "training_step_time": 10.985443830490112
    },
    {
      "epoch": 1.5706062316894532e-05,
      "model_forward_time": 1.7604119777679443,
      "step": 16469
    },
    {
      "epoch": 1.5706062316894532e-05,
      "step": 16469,
      "training_step_time": 8.476667881011963
    },
    {
      "epoch": 1.5707015991210936e-05,
      "grad_norm": 0.17665281891822815,
      "learning_rate": 9.64467041917749e-06,
      "loss": 0.0696,
      "step": 16470
    },
    {
      "epoch": 1.5707015991210936e-05,
      "model_forward_time": 1.748485803604126,
      "step": 16470
    },
    {
      "dynamics_loss_avg": 0.06869003176689148,
      "epoch": 1.5707015991210936e-05,
      "step": 16470
    },
    {
      "action_loss_avg": 0.002388570652692579,
      "epoch": 1.5707015991210936e-05,
      "step": 16470
    },
    {
      "epoch": 1.5707015991210936e-05,
      "step": 16470,
      "training_step_time": 7.6797895431518555
    },
    {
      "epoch": 1.5707969665527344e-05,
      "model_forward_time": 1.7422618865966797,
      "step": 16471
    },
    {
      "epoch": 1.5707969665527344e-05,
      "step": 16471,
      "training_step_time": 7.56064510345459
    },
    {
      "epoch": 1.570892333984375e-05,
      "model_forward_time": 1.7432689666748047,
      "step": 16472
    },
    {
      "epoch": 1.570892333984375e-05,
      "step": 16472,
      "training_step_time": 7.2311882972717285
    },
    {
      "epoch": 1.5709877014160155e-05,
      "model_forward_time": 1.7513015270233154,
      "step": 16473
    },
    {
      "epoch": 1.5709877014160155e-05,
      "step": 16473,
      "training_step_time": 6.767324686050415
    },
    {
      "epoch": 1.5710830688476563e-05,
      "model_forward_time": 1.7363719940185547,
      "step": 16474
    },
    {
      "epoch": 1.5710830688476563e-05,
      "step": 16474,
      "training_step_time": 6.7589781284332275
    },
    {
      "epoch": 1.571178436279297e-05,
      "model_forward_time": 1.7507107257843018,
      "step": 16475
    },
    {
      "epoch": 1.571178436279297e-05,
      "step": 16475,
      "training_step_time": 8.309873580932617
    },
    {
      "epoch": 1.5712738037109374e-05,
      "model_forward_time": 1.7419095039367676,
      "step": 16476
    },
    {
      "epoch": 1.5712738037109374e-05,
      "step": 16476,
      "training_step_time": 7.1998796463012695
    },
    {
      "epoch": 1.571369171142578e-05,
      "model_forward_time": 1.7428038120269775,
      "step": 16477
    },
    {
      "epoch": 1.571369171142578e-05,
      "step": 16477,
      "training_step_time": 10.138981819152832
    },
    {
      "epoch": 1.571464538574219e-05,
      "model_forward_time": 1.8443381786346436,
      "step": 16478
    },
    {
      "epoch": 1.571464538574219e-05,
      "step": 16478,
      "training_step_time": 8.342071056365967
    },
    {
      "epoch": 1.5715599060058593e-05,
      "model_forward_time": 1.745378017425537,
      "step": 16479
    },
    {
      "epoch": 1.5715599060058593e-05,
      "step": 16479,
      "training_step_time": 7.803865909576416
    },
    {
      "epoch": 1.5716552734375e-05,
      "grad_norm": 0.22962810099124908,
      "learning_rate": 9.644057976048063e-06,
      "loss": 0.0654,
      "step": 16480
    },
    {
      "epoch": 1.5716552734375e-05,
      "model_forward_time": 1.7366971969604492,
      "step": 16480
    },
    {
      "dynamics_loss_avg": 0.06349225156009197,
      "epoch": 1.5716552734375e-05,
      "step": 16480
    },
    {
      "action_loss_avg": 0.002950179803883657,
      "epoch": 1.5716552734375e-05,
      "step": 16480
    },
    {
      "epoch": 1.5716552734375e-05,
      "step": 16480,
      "training_step_time": 7.596168518066406
    },
    {
      "epoch": 1.5717506408691408e-05,
      "model_forward_time": 1.7452967166900635,
      "step": 16481
    },
    {
      "epoch": 1.5717506408691408e-05,
      "step": 16481,
      "training_step_time": 7.514359474182129
    },
    {
      "epoch": 1.5718460083007812e-05,
      "model_forward_time": 1.7437236309051514,
      "step": 16482
    },
    {
      "epoch": 1.5718460083007812e-05,
      "step": 16482,
      "training_step_time": 6.719614505767822
    },
    {
      "epoch": 1.571941375732422e-05,
      "model_forward_time": 1.7536203861236572,
      "step": 16483
    },
    {
      "epoch": 1.571941375732422e-05,
      "step": 16483,
      "training_step_time": 6.699227809906006
    },
    {
      "epoch": 1.5720367431640627e-05,
      "model_forward_time": 1.759599208831787,
      "step": 16484
    },
    {
      "epoch": 1.5720367431640627e-05,
      "step": 16484,
      "training_step_time": 7.308098793029785
    },
    {
      "epoch": 1.572132110595703e-05,
      "model_forward_time": 1.7484948635101318,
      "step": 16485
    },
    {
      "epoch": 1.572132110595703e-05,
      "step": 16485,
      "training_step_time": 8.933361530303955
    },
    {
      "epoch": 1.5722274780273438e-05,
      "model_forward_time": 1.7443721294403076,
      "step": 16486
    },
    {
      "epoch": 1.5722274780273438e-05,
      "step": 16486,
      "training_step_time": 8.91475796699524
    },
    {
      "epoch": 1.5723228454589842e-05,
      "model_forward_time": 2.8894293308258057,
      "step": 16487
    },
    {
      "epoch": 1.5723228454589842e-05,
      "step": 16487,
      "training_step_time": 9.862035274505615
    },
    {
      "epoch": 1.572418212890625e-05,
      "model_forward_time": 1.7350647449493408,
      "step": 16488
    },
    {
      "epoch": 1.572418212890625e-05,
      "step": 16488,
      "training_step_time": 7.85327935218811
    },
    {
      "epoch": 1.5725135803222657e-05,
      "model_forward_time": 1.7342596054077148,
      "step": 16489
    },
    {
      "epoch": 1.5725135803222657e-05,
      "step": 16489,
      "training_step_time": 8.08256483078003
    },
    {
      "epoch": 1.572608947753906e-05,
      "grad_norm": 0.242270827293396,
      "learning_rate": 9.643445025051436e-06,
      "loss": 0.068,
      "step": 16490
    },
    {
      "epoch": 1.572608947753906e-05,
      "model_forward_time": 1.7532594203948975,
      "step": 16490
    },
    {
      "dynamics_loss_avg": 0.08136615734547377,
      "epoch": 1.572608947753906e-05,
      "step": 16490
    },
    {
      "action_loss_avg": 0.004306551680201665,
      "epoch": 1.572608947753906e-05,
      "step": 16490
    },
    {
      "epoch": 1.572608947753906e-05,
      "step": 16490,
      "training_step_time": 7.530238389968872
    },
    {
      "epoch": 1.572704315185547e-05,
      "model_forward_time": 1.7604610919952393,
      "step": 16491
    },
    {
      "epoch": 1.572704315185547e-05,
      "step": 16491,
      "training_step_time": 6.844952583312988
    },
    {
      "epoch": 1.5727996826171876e-05,
      "model_forward_time": 1.7612791061401367,
      "step": 16492
    },
    {
      "epoch": 1.5727996826171876e-05,
      "step": 16492,
      "training_step_time": 6.864485025405884
    },
    {
      "epoch": 1.572895050048828e-05,
      "model_forward_time": 1.7631645202636719,
      "step": 16493
    },
    {
      "epoch": 1.572895050048828e-05,
      "step": 16493,
      "training_step_time": 7.36139702796936
    },
    {
      "epoch": 1.5729904174804687e-05,
      "model_forward_time": 1.7564473152160645,
      "step": 16494
    },
    {
      "epoch": 1.5729904174804687e-05,
      "step": 16494,
      "training_step_time": 7.727962017059326
    },
    {
      "epoch": 1.5730857849121095e-05,
      "model_forward_time": 1.7529499530792236,
      "step": 16495
    },
    {
      "epoch": 1.5730857849121095e-05,
      "step": 16495,
      "training_step_time": 8.390877962112427
    },
    {
      "epoch": 1.57318115234375e-05,
      "model_forward_time": 2.9476358890533447,
      "step": 16496
    },
    {
      "epoch": 1.57318115234375e-05,
      "step": 16496,
      "training_step_time": 8.366774082183838
    },
    {
      "epoch": 1.5732765197753906e-05,
      "model_forward_time": 1.756819486618042,
      "step": 16497
    },
    {
      "epoch": 1.5732765197753906e-05,
      "step": 16497,
      "training_step_time": 7.757201910018921
    },
    {
      "epoch": 1.5733718872070314e-05,
      "model_forward_time": 1.379211664199829,
      "step": 16498
    },
    {
      "epoch": 1.5733718872070314e-05,
      "step": 16498,
      "training_step_time": 7.6350977420806885
    },
    {
      "epoch": 1.5734672546386718e-05,
      "model_forward_time": 1.7566258907318115,
      "step": 16499
    },
    {
      "epoch": 1.5734672546386718e-05,
      "step": 16499,
      "training_step_time": 7.621475458145142
    },
    {
      "epoch": 1.5735626220703125e-05,
      "grad_norm": 0.31868594884872437,
      "learning_rate": 9.642831566254642e-06,
      "loss": 0.0654,
      "step": 16500
    },
    {
      "epoch": 1.5735626220703125e-05,
      "model_forward_time": 1.7330830097198486,
      "step": 16500
    },
    {
      "dynamics_loss_avg": 0.05398261174559593,
      "epoch": 1.5735626220703125e-05,
      "step": 16500
    },
    {
      "action_loss_avg": 0.0029531058971770108,
      "epoch": 1.5735626220703125e-05,
      "step": 16500
    },
    {
      "epoch": 1.5735626220703125e-05,
      "step": 16500,
      "training_step_time": 6.856571674346924
    },
    {
      "epoch": 1.5736579895019532e-05,
      "model_forward_time": 1.7496087551116943,
      "step": 16501
    },
    {
      "epoch": 1.5736579895019532e-05,
      "step": 16501,
      "training_step_time": 6.832050323486328
    },
    {
      "epoch": 1.5737533569335936e-05,
      "model_forward_time": 1.7410860061645508,
      "step": 16502
    },
    {
      "epoch": 1.5737533569335936e-05,
      "step": 16502,
      "training_step_time": 7.263538837432861
    },
    {
      "epoch": 1.5738487243652344e-05,
      "model_forward_time": 1.7514219284057617,
      "step": 16503
    },
    {
      "epoch": 1.5738487243652344e-05,
      "step": 16503,
      "training_step_time": 7.686552047729492
    },
    {
      "epoch": 1.573944091796875e-05,
      "model_forward_time": 1.751373291015625,
      "step": 16504
    },
    {
      "epoch": 1.573944091796875e-05,
      "step": 16504,
      "training_step_time": 7.583003282546997
    },
    {
      "epoch": 1.5740394592285155e-05,
      "model_forward_time": 2.857999324798584,
      "step": 16505
    },
    {
      "epoch": 1.5740394592285155e-05,
      "step": 16505,
      "training_step_time": 8.400861978530884
    },
    {
      "epoch": 1.5741348266601563e-05,
      "model_forward_time": 1.7513298988342285,
      "step": 16506
    },
    {
      "epoch": 1.5741348266601563e-05,
      "step": 16506,
      "training_step_time": 9.341347694396973
    },
    {
      "epoch": 1.574230194091797e-05,
      "model_forward_time": 1.7468857765197754,
      "step": 16507
    },
    {
      "epoch": 1.574230194091797e-05,
      "step": 16507,
      "training_step_time": 8.311374425888062
    },
    {
      "epoch": 1.5743255615234374e-05,
      "model_forward_time": 1.7478752136230469,
      "step": 16508
    },
    {
      "epoch": 1.5743255615234374e-05,
      "step": 16508,
      "training_step_time": 7.651979923248291
    },
    {
      "epoch": 1.574420928955078e-05,
      "model_forward_time": 1.7564263343811035,
      "step": 16509
    },
    {
      "epoch": 1.574420928955078e-05,
      "step": 16509,
      "training_step_time": 7.496291399002075
    },
    {
      "epoch": 1.574516296386719e-05,
      "grad_norm": 0.232106015086174,
      "learning_rate": 9.64221759972477e-06,
      "loss": 0.0635,
      "step": 16510
    },
    {
      "epoch": 1.574516296386719e-05,
      "model_forward_time": 1.7317094802856445,
      "step": 16510
    },
    {
      "dynamics_loss_avg": 0.06077071689069271,
      "epoch": 1.574516296386719e-05,
      "step": 16510
    },
    {
      "action_loss_avg": 0.002477124554570764,
      "epoch": 1.574516296386719e-05,
      "step": 16510
    },
    {
      "epoch": 1.574516296386719e-05,
      "step": 16510,
      "training_step_time": 6.788435697555542
    },
    {
      "epoch": 1.5746116638183593e-05,
      "model_forward_time": 1.7595293521881104,
      "step": 16511
    },
    {
      "epoch": 1.5746116638183593e-05,
      "step": 16511,
      "training_step_time": 6.780235528945923
    },
    {
      "epoch": 1.57470703125e-05,
      "model_forward_time": 1.7529776096343994,
      "step": 16512
    },
    {
      "epoch": 1.57470703125e-05,
      "step": 16512,
      "training_step_time": 7.601412534713745
    },
    {
      "epoch": 1.5748023986816408e-05,
      "model_forward_time": 1.747114896774292,
      "step": 16513
    },
    {
      "epoch": 1.5748023986816408e-05,
      "step": 16513,
      "training_step_time": 8.138586282730103
    },
    {
      "epoch": 1.5748977661132812e-05,
      "model_forward_time": 2.872692346572876,
      "step": 16514
    },
    {
      "epoch": 1.5748977661132812e-05,
      "step": 16514,
      "training_step_time": 8.481176614761353
    },
    {
      "epoch": 1.574993133544922e-05,
      "model_forward_time": 1.7532076835632324,
      "step": 16515
    },
    {
      "epoch": 1.574993133544922e-05,
      "step": 16515,
      "training_step_time": 8.800057888031006
    },
    {
      "epoch": 1.5750885009765623e-05,
      "model_forward_time": 1.7533597946166992,
      "step": 16516
    },
    {
      "epoch": 1.5750885009765623e-05,
      "step": 16516,
      "training_step_time": 7.784744501113892
    },
    {
      "epoch": 1.575183868408203e-05,
      "model_forward_time": 1.752810001373291,
      "step": 16517
    },
    {
      "epoch": 1.575183868408203e-05,
      "step": 16517,
      "training_step_time": 7.617372512817383
    },
    {
      "epoch": 1.5752792358398438e-05,
      "model_forward_time": 1.7490026950836182,
      "step": 16518
    },
    {
      "epoch": 1.5752792358398438e-05,
      "step": 16518,
      "training_step_time": 7.609159708023071
    },
    {
      "epoch": 1.5753746032714842e-05,
      "model_forward_time": 1.7573208808898926,
      "step": 16519
    },
    {
      "epoch": 1.5753746032714842e-05,
      "step": 16519,
      "training_step_time": 6.802907943725586
    },
    {
      "epoch": 1.575469970703125e-05,
      "grad_norm": 0.19508208334445953,
      "learning_rate": 9.64160312552896e-06,
      "loss": 0.0618,
      "step": 16520
    },
    {
      "epoch": 1.575469970703125e-05,
      "model_forward_time": 1.749058723449707,
      "step": 16520
    },
    {
      "dynamics_loss_avg": 0.06665347013622522,
      "epoch": 1.575469970703125e-05,
      "step": 16520
    },
    {
      "action_loss_avg": 0.0036926120112184433,
      "epoch": 1.575469970703125e-05,
      "step": 16520
    },
    {
      "epoch": 1.575469970703125e-05,
      "step": 16520,
      "training_step_time": 6.776875019073486
    },
    {
      "epoch": 1.5755653381347657e-05,
      "model_forward_time": 1.7169365882873535,
      "step": 16521
    },
    {
      "epoch": 1.5755653381347657e-05,
      "step": 16521,
      "training_step_time": 7.607003450393677
    },
    {
      "epoch": 1.575660705566406e-05,
      "model_forward_time": 1.7486004829406738,
      "step": 16522
    },
    {
      "epoch": 1.575660705566406e-05,
      "step": 16522,
      "training_step_time": 7.573587417602539
    },
    {
      "epoch": 1.575756072998047e-05,
      "model_forward_time": 1.7449619770050049,
      "step": 16523
    },
    {
      "epoch": 1.575756072998047e-05,
      "step": 16523,
      "training_step_time": 9.481962442398071
    },
    {
      "epoch": 1.5758514404296876e-05,
      "model_forward_time": 2.848972797393799,
      "step": 16524
    },
    {
      "epoch": 1.5758514404296876e-05,
      "step": 16524,
      "training_step_time": 10.214112281799316
    },
    {
      "epoch": 1.575946807861328e-05,
      "model_forward_time": 1.7508413791656494,
      "step": 16525
    },
    {
      "epoch": 1.575946807861328e-05,
      "step": 16525,
      "training_step_time": 8.294208765029907
    },
    {
      "epoch": 1.5760421752929687e-05,
      "model_forward_time": 1.7719659805297852,
      "step": 16526
    },
    {
      "epoch": 1.5760421752929687e-05,
      "step": 16526,
      "training_step_time": 9.08797311782837
    },
    {
      "epoch": 1.5761375427246095e-05,
      "model_forward_time": 1.7479472160339355,
      "step": 16527
    },
    {
      "epoch": 1.5761375427246095e-05,
      "step": 16527,
      "training_step_time": 7.507232666015625
    },
    {
      "epoch": 1.57623291015625e-05,
      "model_forward_time": 1.7522706985473633,
      "step": 16528
    },
    {
      "epoch": 1.57623291015625e-05,
      "step": 16528,
      "training_step_time": 6.749291181564331
    },
    {
      "epoch": 1.5763282775878906e-05,
      "model_forward_time": 1.740600824356079,
      "step": 16529
    },
    {
      "epoch": 1.5763282775878906e-05,
      "step": 16529,
      "training_step_time": 6.7127673625946045
    },
    {
      "epoch": 1.5764236450195314e-05,
      "grad_norm": 0.16270354390144348,
      "learning_rate": 9.64098814373441e-06,
      "loss": 0.0655,
      "step": 16530
    },
    {
      "epoch": 1.5764236450195314e-05,
      "model_forward_time": 1.7390267848968506,
      "step": 16530
    },
    {
      "dynamics_loss_avg": 0.05268226768821478,
      "epoch": 1.5764236450195314e-05,
      "step": 16530
    },
    {
      "action_loss_avg": 0.002681361074792221,
      "epoch": 1.5764236450195314e-05,
      "step": 16530
    },
    {
      "epoch": 1.5764236450195314e-05,
      "step": 16530,
      "training_step_time": 7.564165353775024
    },
    {
      "epoch": 1.5765190124511718e-05,
      "model_forward_time": 1.7193686962127686,
      "step": 16531
    },
    {
      "epoch": 1.5765190124511718e-05,
      "step": 16531,
      "training_step_time": 7.520211458206177
    },
    {
      "epoch": 1.5766143798828125e-05,
      "model_forward_time": 1.7440729141235352,
      "step": 16532
    },
    {
      "epoch": 1.5766143798828125e-05,
      "step": 16532,
      "training_step_time": 7.698300361633301
    },
    {
      "epoch": 1.5767097473144532e-05,
      "model_forward_time": 2.921173095703125,
      "step": 16533
    },
    {
      "epoch": 1.5767097473144532e-05,
      "step": 16533,
      "training_step_time": 10.29686450958252
    },
    {
      "epoch": 1.5768051147460936e-05,
      "model_forward_time": 1.7570199966430664,
      "step": 16534
    },
    {
      "epoch": 1.5768051147460936e-05,
      "step": 16534,
      "training_step_time": 8.68062162399292
    },
    {
      "epoch": 1.5769004821777344e-05,
      "model_forward_time": 1.7586634159088135,
      "step": 16535
    },
    {
      "epoch": 1.5769004821777344e-05,
      "step": 16535,
      "training_step_time": 7.638513803482056
    },
    {
      "epoch": 1.576995849609375e-05,
      "model_forward_time": 1.758725881576538,
      "step": 16536
    },
    {
      "epoch": 1.576995849609375e-05,
      "step": 16536,
      "training_step_time": 7.608093976974487
    },
    {
      "epoch": 1.5770912170410155e-05,
      "model_forward_time": 1.746649980545044,
      "step": 16537
    },
    {
      "epoch": 1.5770912170410155e-05,
      "step": 16537,
      "training_step_time": 6.773545503616333
    },
    {
      "epoch": 1.5771865844726563e-05,
      "model_forward_time": 1.7548375129699707,
      "step": 16538
    },
    {
      "epoch": 1.5771865844726563e-05,
      "step": 16538,
      "training_step_time": 6.89613938331604
    },
    {
      "epoch": 1.577281951904297e-05,
      "model_forward_time": 1.7343790531158447,
      "step": 16539
    },
    {
      "epoch": 1.577281951904297e-05,
      "step": 16539,
      "training_step_time": 7.442315101623535
    },
    {
      "epoch": 1.5773773193359374e-05,
      "grad_norm": 0.32982760667800903,
      "learning_rate": 9.640372654408373e-06,
      "loss": 0.0652,
      "step": 16540
    },
    {
      "epoch": 1.5773773193359374e-05,
      "model_forward_time": 1.7485687732696533,
      "step": 16540
    },
    {
      "dynamics_loss_avg": 0.05941269937902689,
      "epoch": 1.5773773193359374e-05,
      "step": 16540
    },
    {
      "action_loss_avg": 0.0030710521270520983,
      "epoch": 1.5773773193359374e-05,
      "step": 16540
    },
    {
      "epoch": 1.5773773193359374e-05,
      "step": 16540,
      "training_step_time": 7.8653411865234375
    },
    {
      "epoch": 1.577472686767578e-05,
      "model_forward_time": 1.7271356582641602,
      "step": 16541
    },
    {
      "epoch": 1.577472686767578e-05,
      "step": 16541,
      "training_step_time": 8.814166069030762
    },
    {
      "epoch": 1.577568054199219e-05,
      "model_forward_time": 2.918919086456299,
      "step": 16542
    },
    {
      "epoch": 1.577568054199219e-05,
      "step": 16542,
      "training_step_time": 8.931385517120361
    },
    {
      "epoch": 1.5776634216308593e-05,
      "model_forward_time": 1.7467265129089355,
      "step": 16543
    },
    {
      "epoch": 1.5776634216308593e-05,
      "step": 16543,
      "training_step_time": 8.57129693031311
    },
    {
      "epoch": 1.5777587890625e-05,
      "model_forward_time": 1.7500264644622803,
      "step": 16544
    },
    {
      "epoch": 1.5777587890625e-05,
      "step": 16544,
      "training_step_time": 7.903427839279175
    },
    {
      "epoch": 1.5778541564941408e-05,
      "model_forward_time": 1.7444243431091309,
      "step": 16545
    },
    {
      "epoch": 1.5778541564941408e-05,
      "step": 16545,
      "training_step_time": 7.795098781585693
    },
    {
      "epoch": 1.5779495239257812e-05,
      "model_forward_time": 1.75823974609375,
      "step": 16546
    },
    {
      "epoch": 1.5779495239257812e-05,
      "step": 16546,
      "training_step_time": 7.692535161972046
    },
    {
      "epoch": 1.578044891357422e-05,
      "model_forward_time": 1.7480177879333496,
      "step": 16547
    },
    {
      "epoch": 1.578044891357422e-05,
      "step": 16547,
      "training_step_time": 6.933388710021973
    },
    {
      "epoch": 1.5781402587890623e-05,
      "model_forward_time": 1.743804693222046,
      "step": 16548
    },
    {
      "epoch": 1.5781402587890623e-05,
      "step": 16548,
      "training_step_time": 7.397126197814941
    },
    {
      "epoch": 1.578235626220703e-05,
      "model_forward_time": 1.746795892715454,
      "step": 16549
    },
    {
      "epoch": 1.578235626220703e-05,
      "step": 16549,
      "training_step_time": 7.796173334121704
    },
    {
      "epoch": 1.5783309936523438e-05,
      "grad_norm": 0.28048649430274963,
      "learning_rate": 9.639756657618162e-06,
      "loss": 0.0654,
      "step": 16550
    },
    {
      "epoch": 1.5783309936523438e-05,
      "model_forward_time": 1.7267265319824219,
      "step": 16550
    },
    {
      "dynamics_loss_avg": 0.053722404129803183,
      "epoch": 1.5783309936523438e-05,
      "step": 16550
    },
    {
      "action_loss_avg": 0.004207446245709434,
      "epoch": 1.5783309936523438e-05,
      "step": 16550
    },
    {
      "epoch": 1.5783309936523438e-05,
      "step": 16550,
      "training_step_time": 8.486903190612793
    },
    {
      "epoch": 1.5784263610839842e-05,
      "model_forward_time": 2.908432960510254,
      "step": 16551
    },
    {
      "epoch": 1.5784263610839842e-05,
      "step": 16551,
      "training_step_time": 10.010519742965698
    },
    {
      "epoch": 1.578521728515625e-05,
      "model_forward_time": 1.7448716163635254,
      "step": 16552
    },
    {
      "epoch": 1.578521728515625e-05,
      "step": 16552,
      "training_step_time": 8.600399255752563
    },
    {
      "epoch": 1.5786170959472657e-05,
      "model_forward_time": 1.764991044998169,
      "step": 16553
    },
    {
      "epoch": 1.5786170959472657e-05,
      "step": 16553,
      "training_step_time": 7.922330856323242
    },
    {
      "epoch": 1.578712463378906e-05,
      "model_forward_time": 1.752464771270752,
      "step": 16554
    },
    {
      "epoch": 1.578712463378906e-05,
      "step": 16554,
      "training_step_time": 7.823768138885498
    },
    {
      "epoch": 1.578807830810547e-05,
      "model_forward_time": 1.7399938106536865,
      "step": 16555
    },
    {
      "epoch": 1.578807830810547e-05,
      "step": 16555,
      "training_step_time": 7.816381931304932
    },
    {
      "epoch": 1.5789031982421876e-05,
      "model_forward_time": 1.7481155395507812,
      "step": 16556
    },
    {
      "epoch": 1.5789031982421876e-05,
      "step": 16556,
      "training_step_time": 6.95188307762146
    },
    {
      "epoch": 1.578998565673828e-05,
      "model_forward_time": 1.7513766288757324,
      "step": 16557
    },
    {
      "epoch": 1.578998565673828e-05,
      "step": 16557,
      "training_step_time": 7.677660226821899
    },
    {
      "epoch": 1.5790939331054687e-05,
      "model_forward_time": 1.751478672027588,
      "step": 16558
    },
    {
      "epoch": 1.5790939331054687e-05,
      "step": 16558,
      "training_step_time": 7.911087512969971
    },
    {
      "epoch": 1.5791893005371095e-05,
      "model_forward_time": 1.7387235164642334,
      "step": 16559
    },
    {
      "epoch": 1.5791893005371095e-05,
      "step": 16559,
      "training_step_time": 8.589028120040894
    },
    {
      "epoch": 1.57928466796875e-05,
      "grad_norm": 0.2640913724899292,
      "learning_rate": 9.63914015343114e-06,
      "loss": 0.0675,
      "step": 16560
    },
    {
      "epoch": 1.57928466796875e-05,
      "model_forward_time": 2.8603103160858154,
      "step": 16560
    },
    {
      "dynamics_loss_avg": 0.060369175486266614,
      "epoch": 1.57928466796875e-05,
      "step": 16560
    },
    {
      "action_loss_avg": 0.0035011560656130316,
      "epoch": 1.57928466796875e-05,
      "step": 16560
    },
    {
      "epoch": 1.57928466796875e-05,
      "step": 16560,
      "training_step_time": 9.735810041427612
    },
    {
      "epoch": 1.5793800354003906e-05,
      "model_forward_time": 1.8459270000457764,
      "step": 16561
    },
    {
      "epoch": 1.5793800354003906e-05,
      "step": 16561,
      "training_step_time": 8.610053062438965
    },
    {
      "epoch": 1.5794754028320314e-05,
      "model_forward_time": 1.7507100105285645,
      "step": 16562
    },
    {
      "epoch": 1.5794754028320314e-05,
      "step": 16562,
      "training_step_time": 8.786730289459229
    },
    {
      "epoch": 1.5795707702636718e-05,
      "model_forward_time": 1.745136022567749,
      "step": 16563
    },
    {
      "epoch": 1.5795707702636718e-05,
      "step": 16563,
      "training_step_time": 7.831910133361816
    },
    {
      "epoch": 1.5796661376953125e-05,
      "model_forward_time": 1.7348127365112305,
      "step": 16564
    },
    {
      "epoch": 1.5796661376953125e-05,
      "step": 16564,
      "training_step_time": 7.72748064994812
    },
    {
      "epoch": 1.5797615051269532e-05,
      "model_forward_time": 1.7498266696929932,
      "step": 16565
    },
    {
      "epoch": 1.5797615051269532e-05,
      "step": 16565,
      "training_step_time": 6.946615695953369
    },
    {
      "epoch": 1.5798568725585936e-05,
      "model_forward_time": 1.7566916942596436,
      "step": 16566
    },
    {
      "epoch": 1.5798568725585936e-05,
      "step": 16566,
      "training_step_time": 6.940526723861694
    },
    {
      "epoch": 1.5799522399902344e-05,
      "model_forward_time": 1.7454965114593506,
      "step": 16567
    },
    {
      "epoch": 1.5799522399902344e-05,
      "step": 16567,
      "training_step_time": 7.854433059692383
    },
    {
      "epoch": 1.580047607421875e-05,
      "model_forward_time": 1.7475292682647705,
      "step": 16568
    },
    {
      "epoch": 1.580047607421875e-05,
      "step": 16568,
      "training_step_time": 7.791717529296875
    },
    {
      "epoch": 1.5801429748535155e-05,
      "model_forward_time": 1.7309908866882324,
      "step": 16569
    },
    {
      "epoch": 1.5801429748535155e-05,
      "step": 16569,
      "training_step_time": 8.37058401107788
    },
    {
      "epoch": 1.5802383422851563e-05,
      "grad_norm": 0.18113845586776733,
      "learning_rate": 9.638523141914721e-06,
      "loss": 0.0656,
      "step": 16570
    },
    {
      "epoch": 1.5802383422851563e-05,
      "model_forward_time": 1.8081371784210205,
      "step": 16570
    },
    {
      "dynamics_loss_avg": 0.06946121193468571,
      "epoch": 1.5802383422851563e-05,
      "step": 16570
    },
    {
      "action_loss_avg": 0.004075626842677593,
      "epoch": 1.5802383422851563e-05,
      "step": 16570
    },
    {
      "epoch": 1.5802383422851563e-05,
      "step": 16570,
      "training_step_time": 8.980666399002075
    },
    {
      "epoch": 1.580333709716797e-05,
      "model_forward_time": 1.7602977752685547,
      "step": 16571
    },
    {
      "epoch": 1.580333709716797e-05,
      "step": 16571,
      "training_step_time": 8.867363452911377
    },
    {
      "epoch": 1.5804290771484374e-05,
      "model_forward_time": 1.7553412914276123,
      "step": 16572
    },
    {
      "epoch": 1.5804290771484374e-05,
      "step": 16572,
      "training_step_time": 7.827969551086426
    },
    {
      "epoch": 1.580524444580078e-05,
      "model_forward_time": 1.750868797302246,
      "step": 16573
    },
    {
      "epoch": 1.580524444580078e-05,
      "step": 16573,
      "training_step_time": 7.858410358428955
    },
    {
      "epoch": 1.580619812011719e-05,
      "model_forward_time": 1.7469556331634521,
      "step": 16574
    },
    {
      "epoch": 1.580619812011719e-05,
      "step": 16574,
      "training_step_time": 7.807941913604736
    },
    {
      "epoch": 1.5807151794433593e-05,
      "model_forward_time": 1.7457916736602783,
      "step": 16575
    },
    {
      "epoch": 1.5807151794433593e-05,
      "step": 16575,
      "training_step_time": 6.979990005493164
    },
    {
      "epoch": 1.580810546875e-05,
      "model_forward_time": 1.7465229034423828,
      "step": 16576
    },
    {
      "epoch": 1.580810546875e-05,
      "step": 16576,
      "training_step_time": 7.839393377304077
    },
    {
      "epoch": 1.5809059143066408e-05,
      "model_forward_time": 1.7233517169952393,
      "step": 16577
    },
    {
      "epoch": 1.5809059143066408e-05,
      "step": 16577,
      "training_step_time": 7.814908504486084
    },
    {
      "epoch": 1.5810012817382812e-05,
      "model_forward_time": 1.733433723449707,
      "step": 16578
    },
    {
      "epoch": 1.5810012817382812e-05,
      "step": 16578,
      "training_step_time": 7.86908745765686
    },
    {
      "epoch": 1.581096649169922e-05,
      "model_forward_time": 3.011831283569336,
      "step": 16579
    },
    {
      "epoch": 1.581096649169922e-05,
      "step": 16579,
      "training_step_time": 9.781227588653564
    },
    {
      "epoch": 1.5811920166015623e-05,
      "grad_norm": 0.19302509725093842,
      "learning_rate": 9.637905623136388e-06,
      "loss": 0.0688,
      "step": 16580
    },
    {
      "epoch": 1.5811920166015623e-05,
      "model_forward_time": 1.7484118938446045,
      "step": 16580
    },
    {
      "dynamics_loss_avg": 0.07018122412264347,
      "epoch": 1.5811920166015623e-05,
      "step": 16580
    },
    {
      "action_loss_avg": 0.0038336234982125463,
      "epoch": 1.5811920166015623e-05,
      "step": 16580
    },
    {
      "epoch": 1.5811920166015623e-05,
      "step": 16580,
      "training_step_time": 8.841155767440796
    },
    {
      "epoch": 1.581287384033203e-05,
      "model_forward_time": 1.7494838237762451,
      "step": 16581
    },
    {
      "epoch": 1.581287384033203e-05,
      "step": 16581,
      "training_step_time": 7.939283609390259
    },
    {
      "epoch": 1.5813827514648438e-05,
      "model_forward_time": 1.7479465007781982,
      "step": 16582
    },
    {
      "epoch": 1.5813827514648438e-05,
      "step": 16582,
      "training_step_time": 7.880118131637573
    },
    {
      "epoch": 1.5814781188964842e-05,
      "model_forward_time": 1.7509584426879883,
      "step": 16583
    },
    {
      "epoch": 1.5814781188964842e-05,
      "step": 16583,
      "training_step_time": 7.777764797210693
    },
    {
      "epoch": 1.581573486328125e-05,
      "model_forward_time": 1.7434616088867188,
      "step": 16584
    },
    {
      "epoch": 1.581573486328125e-05,
      "step": 16584,
      "training_step_time": 6.999121427536011
    },
    {
      "epoch": 1.5816688537597657e-05,
      "model_forward_time": 1.7462551593780518,
      "step": 16585
    },
    {
      "epoch": 1.5816688537597657e-05,
      "step": 16585,
      "training_step_time": 7.730113983154297
    },
    {
      "epoch": 1.581764221191406e-05,
      "model_forward_time": 1.7477545738220215,
      "step": 16586
    },
    {
      "epoch": 1.581764221191406e-05,
      "step": 16586,
      "training_step_time": 7.904687404632568
    },
    {
      "epoch": 1.581859588623047e-05,
      "model_forward_time": 1.7412536144256592,
      "step": 16587
    },
    {
      "epoch": 1.581859588623047e-05,
      "step": 16587,
      "training_step_time": 8.329570293426514
    },
    {
      "epoch": 1.5819549560546876e-05,
      "model_forward_time": 2.8974645137786865,
      "step": 16588
    },
    {
      "epoch": 1.5819549560546876e-05,
      "step": 16588,
      "training_step_time": 9.462554931640625
    },
    {
      "epoch": 1.582050323486328e-05,
      "model_forward_time": 1.844012975692749,
      "step": 16589
    },
    {
      "epoch": 1.582050323486328e-05,
      "step": 16589,
      "training_step_time": 8.779649257659912
    },
    {
      "epoch": 1.5821456909179687e-05,
      "grad_norm": 0.2144690752029419,
      "learning_rate": 9.637287597163669e-06,
      "loss": 0.0637,
      "step": 16590
    },
    {
      "epoch": 1.5821456909179687e-05,
      "model_forward_time": 1.7506661415100098,
      "step": 16590
    },
    {
      "dynamics_loss_avg": 0.062027962692081925,
      "epoch": 1.5821456909179687e-05,
      "step": 16590
    },
    {
      "action_loss_avg": 0.0030968774924986066,
      "epoch": 1.5821456909179687e-05,
      "step": 16590
    },
    {
      "epoch": 1.5821456909179687e-05,
      "step": 16590,
      "training_step_time": 7.838930606842041
    },
    {
      "epoch": 1.5822410583496095e-05,
      "model_forward_time": 1.74190092086792,
      "step": 16591
    },
    {
      "epoch": 1.5822410583496095e-05,
      "step": 16591,
      "training_step_time": 7.799453973770142
    },
    {
      "epoch": 1.58233642578125e-05,
      "model_forward_time": 1.7456133365631104,
      "step": 16592
    },
    {
      "epoch": 1.58233642578125e-05,
      "step": 16592,
      "training_step_time": 7.805129528045654
    },
    {
      "epoch": 1.5824317932128906e-05,
      "model_forward_time": 1.7549200057983398,
      "step": 16593
    },
    {
      "epoch": 1.5824317932128906e-05,
      "step": 16593,
      "training_step_time": 6.962800979614258
    },
    {
      "epoch": 1.5825271606445314e-05,
      "model_forward_time": 1.7486560344696045,
      "step": 16594
    },
    {
      "epoch": 1.5825271606445314e-05,
      "step": 16594,
      "training_step_time": 7.670282602310181
    },
    {
      "epoch": 1.5826225280761718e-05,
      "model_forward_time": 1.7403547763824463,
      "step": 16595
    },
    {
      "epoch": 1.5826225280761718e-05,
      "step": 16595,
      "training_step_time": 7.909287452697754
    },
    {
      "epoch": 1.5827178955078125e-05,
      "model_forward_time": 1.7432794570922852,
      "step": 16596
    },
    {
      "epoch": 1.5827178955078125e-05,
      "step": 16596,
      "training_step_time": 7.748233795166016
    },
    {
      "epoch": 1.5828132629394532e-05,
      "model_forward_time": 2.862356662750244,
      "step": 16597
    },
    {
      "epoch": 1.5828132629394532e-05,
      "step": 16597,
      "training_step_time": 8.546279907226562
    },
    {
      "epoch": 1.5829086303710937e-05,
      "model_forward_time": 1.8747336864471436,
      "step": 16598
    },
    {
      "epoch": 1.5829086303710937e-05,
      "step": 16598,
      "training_step_time": 8.054725885391235
    },
    {
      "epoch": 1.5830039978027344e-05,
      "model_forward_time": 1.7464933395385742,
      "step": 16599
    },
    {
      "epoch": 1.5830039978027344e-05,
      "step": 16599,
      "training_step_time": 8.426370859146118
    },
    {
      "epoch": 1.583099365234375e-05,
      "grad_norm": 0.25477418303489685,
      "learning_rate": 9.63666906406415e-06,
      "loss": 0.0679,
      "step": 16600
    },
    {
      "epoch": 1.583099365234375e-05,
      "model_forward_time": 1.7563045024871826,
      "step": 16600
    },
    {
      "dynamics_loss_avg": 0.06315615214407444,
      "epoch": 1.583099365234375e-05,
      "step": 16600
    },
    {
      "action_loss_avg": 0.002606857847422361,
      "epoch": 1.583099365234375e-05,
      "step": 16600
    },
    {
      "epoch": 1.583099365234375e-05,
      "step": 16600,
      "training_step_time": 7.880609512329102
    },
    {
      "epoch": 1.5831947326660155e-05,
      "model_forward_time": 1.7405297756195068,
      "step": 16601
    },
    {
      "epoch": 1.5831947326660155e-05,
      "step": 16601,
      "training_step_time": 7.811033725738525
    },
    {
      "epoch": 1.5832901000976563e-05,
      "model_forward_time": 1.7501568794250488,
      "step": 16602
    },
    {
      "epoch": 1.5832901000976563e-05,
      "step": 16602,
      "training_step_time": 6.951286554336548
    },
    {
      "epoch": 1.583385467529297e-05,
      "model_forward_time": 1.738039493560791,
      "step": 16603
    },
    {
      "epoch": 1.583385467529297e-05,
      "step": 16603,
      "training_step_time": 7.684699773788452
    },
    {
      "epoch": 1.5834808349609374e-05,
      "model_forward_time": 1.744856357574463,
      "step": 16604
    },
    {
      "epoch": 1.5834808349609374e-05,
      "step": 16604,
      "training_step_time": 7.92116904258728
    },
    {
      "epoch": 1.583576202392578e-05,
      "model_forward_time": 1.741436243057251,
      "step": 16605
    },
    {
      "epoch": 1.583576202392578e-05,
      "step": 16605,
      "training_step_time": 7.7389442920684814
    },
    {
      "epoch": 1.583671569824219e-05,
      "model_forward_time": 2.827587842941284,
      "step": 16606
    },
    {
      "epoch": 1.583671569824219e-05,
      "step": 16606,
      "training_step_time": 9.099339246749878
    },
    {
      "epoch": 1.5837669372558593e-05,
      "model_forward_time": 1.8499302864074707,
      "step": 16607
    },
    {
      "epoch": 1.5837669372558593e-05,
      "step": 16607,
      "training_step_time": 10.453326225280762
    },
    {
      "epoch": 1.5838623046875e-05,
      "model_forward_time": 1.752359390258789,
      "step": 16608
    },
    {
      "epoch": 1.5838623046875e-05,
      "step": 16608,
      "training_step_time": 7.995476245880127
    },
    {
      "epoch": 1.5839576721191408e-05,
      "model_forward_time": 1.7562932968139648,
      "step": 16609
    },
    {
      "epoch": 1.5839576721191408e-05,
      "step": 16609,
      "training_step_time": 7.812937498092651
    },
    {
      "epoch": 1.5840530395507812e-05,
      "grad_norm": 0.18523256480693817,
      "learning_rate": 9.636050023905473e-06,
      "loss": 0.0663,
      "step": 16610
    },
    {
      "epoch": 1.5840530395507812e-05,
      "model_forward_time": 1.7562954425811768,
      "step": 16610
    },
    {
      "dynamics_loss_avg": 0.05628758668899536,
      "epoch": 1.5840530395507812e-05,
      "step": 16610
    },
    {
      "action_loss_avg": 0.002018473792122677,
      "epoch": 1.5840530395507812e-05,
      "step": 16610
    },
    {
      "epoch": 1.5840530395507812e-05,
      "step": 16610,
      "training_step_time": 7.906209707260132
    },
    {
      "epoch": 1.584148406982422e-05,
      "model_forward_time": 1.7543251514434814,
      "step": 16611
    },
    {
      "epoch": 1.584148406982422e-05,
      "step": 16611,
      "training_step_time": 7.400501728057861
    },
    {
      "epoch": 1.5842437744140623e-05,
      "model_forward_time": 1.745518445968628,
      "step": 16612
    },
    {
      "epoch": 1.5842437744140623e-05,
      "step": 16612,
      "training_step_time": 7.77862286567688
    },
    {
      "epoch": 1.584339141845703e-05,
      "model_forward_time": 1.7486820220947266,
      "step": 16613
    },
    {
      "epoch": 1.584339141845703e-05,
      "step": 16613,
      "training_step_time": 7.953855037689209
    },
    {
      "epoch": 1.5844345092773438e-05,
      "model_forward_time": 1.7510342597961426,
      "step": 16614
    },
    {
      "epoch": 1.5844345092773438e-05,
      "step": 16614,
      "training_step_time": 7.7357704639434814
    },
    {
      "epoch": 1.5845298767089842e-05,
      "model_forward_time": 2.825770854949951,
      "step": 16615
    },
    {
      "epoch": 1.5845298767089842e-05,
      "step": 16615,
      "training_step_time": 8.935678243637085
    },
    {
      "epoch": 1.584625244140625e-05,
      "model_forward_time": 1.8204147815704346,
      "step": 16616
    },
    {
      "epoch": 1.584625244140625e-05,
      "step": 16616,
      "training_step_time": 8.468832969665527
    },
    {
      "epoch": 1.5847206115722657e-05,
      "model_forward_time": 1.7554595470428467,
      "step": 16617
    },
    {
      "epoch": 1.5847206115722657e-05,
      "step": 16617,
      "training_step_time": 7.982182741165161
    },
    {
      "epoch": 1.584815979003906e-05,
      "model_forward_time": 1.758096694946289,
      "step": 16618
    },
    {
      "epoch": 1.584815979003906e-05,
      "step": 16618,
      "training_step_time": 7.894107341766357
    },
    {
      "epoch": 1.584911346435547e-05,
      "model_forward_time": 1.754058599472046,
      "step": 16619
    },
    {
      "epoch": 1.584911346435547e-05,
      "step": 16619,
      "training_step_time": 8.382941246032715
    },
    {
      "epoch": 1.5850067138671876e-05,
      "grad_norm": 0.20752355456352234,
      "learning_rate": 9.635430476755337e-06,
      "loss": 0.0702,
      "step": 16620
    },
    {
      "epoch": 1.5850067138671876e-05,
      "model_forward_time": 1.757279634475708,
      "step": 16620
    },
    {
      "dynamics_loss_avg": 0.07251825965940953,
      "epoch": 1.5850067138671876e-05,
      "step": 16620
    },
    {
      "action_loss_avg": 0.0032052404014393686,
      "epoch": 1.5850067138671876e-05,
      "step": 16620
    },
    {
      "epoch": 1.5850067138671876e-05,
      "step": 16620,
      "training_step_time": 7.90293288230896
    },
    {
      "epoch": 1.585102081298828e-05,
      "model_forward_time": 1.7471301555633545,
      "step": 16621
    },
    {
      "epoch": 1.585102081298828e-05,
      "step": 16621,
      "training_step_time": 7.71216082572937
    },
    {
      "epoch": 1.5851974487304687e-05,
      "model_forward_time": 1.7542622089385986,
      "step": 16622
    },
    {
      "epoch": 1.5851974487304687e-05,
      "step": 16622,
      "training_step_time": 7.868236064910889
    },
    {
      "epoch": 1.5852928161621095e-05,
      "model_forward_time": 1.7562544345855713,
      "step": 16623
    },
    {
      "epoch": 1.5852928161621095e-05,
      "step": 16623,
      "training_step_time": 8.228506326675415
    },
    {
      "epoch": 1.58538818359375e-05,
      "model_forward_time": 1.7522711753845215,
      "step": 16624
    },
    {
      "epoch": 1.58538818359375e-05,
      "step": 16624,
      "training_step_time": 7.907876014709473
    },
    {
      "epoch": 1.5854835510253906e-05,
      "model_forward_time": 2.63042950630188,
      "step": 16625
    },
    {
      "epoch": 1.5854835510253906e-05,
      "step": 16625,
      "training_step_time": 7.928127288818359
    },
    {
      "epoch": 1.5855789184570314e-05,
      "model_forward_time": 1.7646052837371826,
      "step": 16626
    },
    {
      "epoch": 1.5855789184570314e-05,
      "step": 16626,
      "training_step_time": 8.942476749420166
    },
    {
      "epoch": 1.5856742858886718e-05,
      "model_forward_time": 1.7574326992034912,
      "step": 16627
    },
    {
      "epoch": 1.5856742858886718e-05,
      "step": 16627,
      "training_step_time": 7.8283045291900635
    },
    {
      "epoch": 1.5857696533203125e-05,
      "model_forward_time": 1.7532892227172852,
      "step": 16628
    },
    {
      "epoch": 1.5857696533203125e-05,
      "step": 16628,
      "training_step_time": 7.796764850616455
    },
    {
      "epoch": 1.5858650207519533e-05,
      "model_forward_time": 1.7647373676300049,
      "step": 16629
    },
    {
      "epoch": 1.5858650207519533e-05,
      "step": 16629,
      "training_step_time": 8.390039920806885
    },
    {
      "epoch": 1.5859603881835937e-05,
      "grad_norm": 0.30708736181259155,
      "learning_rate": 9.634810422681491e-06,
      "loss": 0.0662,
      "step": 16630
    },
    {
      "epoch": 1.5859603881835937e-05,
      "model_forward_time": 1.7205562591552734,
      "step": 16630
    },
    {
      "dynamics_loss_avg": 0.0504416985437274,
      "epoch": 1.5859603881835937e-05,
      "step": 16630
    },
    {
      "action_loss_avg": 0.003307795891305432,
      "epoch": 1.5859603881835937e-05,
      "step": 16630
    },
    {
      "epoch": 1.5859603881835937e-05,
      "step": 16630,
      "training_step_time": 7.6916632652282715
    },
    {
      "epoch": 1.5860557556152344e-05,
      "model_forward_time": 1.7452952861785889,
      "step": 16631
    },
    {
      "epoch": 1.5860557556152344e-05,
      "step": 16631,
      "training_step_time": 8.780357837677002
    },
    {
      "epoch": 1.586151123046875e-05,
      "model_forward_time": 1.742464303970337,
      "step": 16632
    },
    {
      "epoch": 1.586151123046875e-05,
      "step": 16632,
      "training_step_time": 7.670256853103638
    },
    {
      "epoch": 1.5862464904785155e-05,
      "model_forward_time": 1.7230100631713867,
      "step": 16633
    },
    {
      "epoch": 1.5862464904785155e-05,
      "step": 16633,
      "training_step_time": 8.725728511810303
    },
    {
      "epoch": 1.5863418579101563e-05,
      "model_forward_time": 2.9507524967193604,
      "step": 16634
    },
    {
      "epoch": 1.5863418579101563e-05,
      "step": 16634,
      "training_step_time": 8.929360151290894
    },
    {
      "epoch": 1.586437225341797e-05,
      "model_forward_time": 1.7475972175598145,
      "step": 16635
    },
    {
      "epoch": 1.586437225341797e-05,
      "step": 16635,
      "training_step_time": 9.924937963485718
    },
    {
      "epoch": 1.5865325927734374e-05,
      "model_forward_time": 1.7465991973876953,
      "step": 16636
    },
    {
      "epoch": 1.5865325927734374e-05,
      "step": 16636,
      "training_step_time": 9.522820472717285
    },
    {
      "epoch": 1.586627960205078e-05,
      "model_forward_time": 1.7485060691833496,
      "step": 16637
    },
    {
      "epoch": 1.586627960205078e-05,
      "step": 16637,
      "training_step_time": 8.207660675048828
    },
    {
      "epoch": 1.586723327636719e-05,
      "model_forward_time": 1.7407946586608887,
      "step": 16638
    },
    {
      "epoch": 1.586723327636719e-05,
      "step": 16638,
      "training_step_time": 7.787256479263306
    },
    {
      "epoch": 1.5868186950683593e-05,
      "model_forward_time": 1.7323646545410156,
      "step": 16639
    },
    {
      "epoch": 1.5868186950683593e-05,
      "step": 16639,
      "training_step_time": 7.75396466255188
    },
    {
      "epoch": 1.5869140625e-05,
      "grad_norm": 0.24430856108665466,
      "learning_rate": 9.634189861751746e-06,
      "loss": 0.0639,
      "step": 16640
    },
    {
      "epoch": 1.5869140625e-05,
      "model_forward_time": 1.7195582389831543,
      "step": 16640
    },
    {
      "dynamics_loss_avg": 0.06472819549962879,
      "epoch": 1.5869140625e-05,
      "step": 16640
    },
    {
      "action_loss_avg": 0.0046724166022613645,
      "epoch": 1.5869140625e-05,
      "step": 16640
    },
    {
      "epoch": 1.5869140625e-05,
      "step": 16640,
      "training_step_time": 7.7556312084198
    },
    {
      "epoch": 1.5870094299316405e-05,
      "model_forward_time": 1.735440969467163,
      "step": 16641
    },
    {
      "epoch": 1.5870094299316405e-05,
      "step": 16641,
      "training_step_time": 7.861756801605225
    },
    {
      "epoch": 1.5871047973632812e-05,
      "model_forward_time": 1.7437472343444824,
      "step": 16642
    },
    {
      "epoch": 1.5871047973632812e-05,
      "step": 16642,
      "training_step_time": 7.870598077774048
    },
    {
      "epoch": 1.587200164794922e-05,
      "model_forward_time": 2.818687677383423,
      "step": 16643
    },
    {
      "epoch": 1.587200164794922e-05,
      "step": 16643,
      "training_step_time": 9.014986991882324
    },
    {
      "epoch": 1.5872955322265623e-05,
      "model_forward_time": 1.845494270324707,
      "step": 16644
    },
    {
      "epoch": 1.5872955322265623e-05,
      "step": 16644,
      "training_step_time": 7.858861207962036
    },
    {
      "epoch": 1.587390899658203e-05,
      "model_forward_time": 1.74259352684021,
      "step": 16645
    },
    {
      "epoch": 1.587390899658203e-05,
      "step": 16645,
      "training_step_time": 8.03404188156128
    },
    {
      "epoch": 1.5874862670898438e-05,
      "model_forward_time": 1.723235845565796,
      "step": 16646
    },
    {
      "epoch": 1.5874862670898438e-05,
      "step": 16646,
      "training_step_time": 7.820826530456543
    },
    {
      "epoch": 1.5875816345214842e-05,
      "model_forward_time": 1.7227911949157715,
      "step": 16647
    },
    {
      "epoch": 1.5875816345214842e-05,
      "step": 16647,
      "training_step_time": 7.853825092315674
    },
    {
      "epoch": 1.587677001953125e-05,
      "model_forward_time": 1.748556137084961,
      "step": 16648
    },
    {
      "epoch": 1.587677001953125e-05,
      "step": 16648,
      "training_step_time": 7.707654714584351
    },
    {
      "epoch": 1.5877723693847657e-05,
      "model_forward_time": 1.727461576461792,
      "step": 16649
    },
    {
      "epoch": 1.5877723693847657e-05,
      "step": 16649,
      "training_step_time": 7.7927939891815186
    },
    {
      "epoch": 1.587867736816406e-05,
      "grad_norm": 0.19561029970645905,
      "learning_rate": 9.633568794033967e-06,
      "loss": 0.0608,
      "step": 16650
    },
    {
      "epoch": 1.587867736816406e-05,
      "model_forward_time": 1.7177441120147705,
      "step": 16650
    },
    {
      "dynamics_loss_avg": 0.06999289020895957,
      "epoch": 1.587867736816406e-05,
      "step": 16650
    },
    {
      "action_loss_avg": 0.004729496361687779,
      "epoch": 1.587867736816406e-05,
      "step": 16650
    },
    {
      "epoch": 1.587867736816406e-05,
      "step": 16650,
      "training_step_time": 7.987392902374268
    },
    {
      "epoch": 1.587963104248047e-05,
      "model_forward_time": 1.743598461151123,
      "step": 16651
    },
    {
      "epoch": 1.587963104248047e-05,
      "step": 16651,
      "training_step_time": 8.735260009765625
    },
    {
      "epoch": 1.5880584716796876e-05,
      "model_forward_time": 2.8577356338500977,
      "step": 16652
    },
    {
      "epoch": 1.5880584716796876e-05,
      "step": 16652,
      "training_step_time": 9.672386169433594
    },
    {
      "epoch": 1.588153839111328e-05,
      "model_forward_time": 1.8327624797821045,
      "step": 16653
    },
    {
      "epoch": 1.588153839111328e-05,
      "step": 16653,
      "training_step_time": 8.982950448989868
    },
    {
      "epoch": 1.5882492065429687e-05,
      "model_forward_time": 1.7220091819763184,
      "step": 16654
    },
    {
      "epoch": 1.5882492065429687e-05,
      "step": 16654,
      "training_step_time": 9.190866708755493
    },
    {
      "epoch": 1.5883445739746095e-05,
      "model_forward_time": 1.7224533557891846,
      "step": 16655
    },
    {
      "epoch": 1.5883445739746095e-05,
      "step": 16655,
      "training_step_time": 7.7755818367004395
    },
    {
      "epoch": 1.58843994140625e-05,
      "model_forward_time": 1.730776071548462,
      "step": 16656
    },
    {
      "epoch": 1.58843994140625e-05,
      "step": 16656,
      "training_step_time": 7.791849136352539
    },
    {
      "epoch": 1.5885353088378906e-05,
      "model_forward_time": 1.743748664855957,
      "step": 16657
    },
    {
      "epoch": 1.5885353088378906e-05,
      "step": 16657,
      "training_step_time": 8.419184684753418
    },
    {
      "epoch": 1.5886306762695314e-05,
      "model_forward_time": 1.728257656097412,
      "step": 16658
    },
    {
      "epoch": 1.5886306762695314e-05,
      "step": 16658,
      "training_step_time": 7.7191689014434814
    },
    {
      "epoch": 1.5887260437011718e-05,
      "model_forward_time": 1.744664192199707,
      "step": 16659
    },
    {
      "epoch": 1.5887260437011718e-05,
      "step": 16659,
      "training_step_time": 7.87861704826355
    },
    {
      "epoch": 1.5888214111328125e-05,
      "grad_norm": 0.23880310356616974,
      "learning_rate": 9.632947219596069e-06,
      "loss": 0.0607,
      "step": 16660
    },
    {
      "epoch": 1.5888214111328125e-05,
      "model_forward_time": 1.716064691543579,
      "step": 16660
    },
    {
      "dynamics_loss_avg": 0.06437638271600007,
      "epoch": 1.5888214111328125e-05,
      "step": 16660
    },
    {
      "action_loss_avg": 0.0021324835484847426,
      "epoch": 1.5888214111328125e-05,
      "step": 16660
    },
    {
      "epoch": 1.5888214111328125e-05,
      "step": 16660,
      "training_step_time": 7.809701919555664
    },
    {
      "epoch": 1.5889167785644533e-05,
      "model_forward_time": 1.7400052547454834,
      "step": 16661
    },
    {
      "epoch": 1.5889167785644533e-05,
      "step": 16661,
      "training_step_time": 9.713977813720703
    },
    {
      "epoch": 1.5890121459960937e-05,
      "model_forward_time": 1.8107094764709473,
      "step": 16662
    },
    {
      "epoch": 1.5890121459960937e-05,
      "step": 16662,
      "training_step_time": 9.445241212844849
    },
    {
      "epoch": 1.5891075134277344e-05,
      "model_forward_time": 1.7520732879638672,
      "step": 16663
    },
    {
      "epoch": 1.5891075134277344e-05,
      "step": 16663,
      "training_step_time": 8.51168179512024
    },
    {
      "epoch": 1.589202880859375e-05,
      "model_forward_time": 1.767927646636963,
      "step": 16664
    },
    {
      "epoch": 1.589202880859375e-05,
      "step": 16664,
      "training_step_time": 8.520896673202515
    },
    {
      "epoch": 1.5892982482910155e-05,
      "model_forward_time": 1.7683584690093994,
      "step": 16665
    },
    {
      "epoch": 1.5892982482910155e-05,
      "step": 16665,
      "training_step_time": 7.576090335845947
    },
    {
      "epoch": 1.5893936157226563e-05,
      "model_forward_time": 1.768620491027832,
      "step": 16666
    },
    {
      "epoch": 1.5893936157226563e-05,
      "step": 16666,
      "training_step_time": 7.765323162078857
    },
    {
      "epoch": 1.589488983154297e-05,
      "model_forward_time": 1.761937141418457,
      "step": 16667
    },
    {
      "epoch": 1.589488983154297e-05,
      "step": 16667,
      "training_step_time": 7.7086169719696045
    },
    {
      "epoch": 1.5895843505859374e-05,
      "model_forward_time": 1.7550671100616455,
      "step": 16668
    },
    {
      "epoch": 1.5895843505859374e-05,
      "step": 16668,
      "training_step_time": 7.8686559200286865
    },
    {
      "epoch": 1.5896797180175782e-05,
      "model_forward_time": 1.759864091873169,
      "step": 16669
    },
    {
      "epoch": 1.5896797180175782e-05,
      "step": 16669,
      "training_step_time": 7.766178131103516
    },
    {
      "epoch": 1.589775085449219e-05,
      "grad_norm": 0.2405235767364502,
      "learning_rate": 9.63232513850603e-06,
      "loss": 0.0624,
      "step": 16670
    },
    {
      "epoch": 1.589775085449219e-05,
      "model_forward_time": 1.7399985790252686,
      "step": 16670
    },
    {
      "dynamics_loss_avg": 0.04838059907779098,
      "epoch": 1.589775085449219e-05,
      "step": 16670
    },
    {
      "action_loss_avg": 0.002483314857818186,
      "epoch": 1.589775085449219e-05,
      "step": 16670
    },
    {
      "epoch": 1.589775085449219e-05,
      "step": 16670,
      "training_step_time": 8.272180795669556
    },
    {
      "epoch": 1.5898704528808593e-05,
      "model_forward_time": 3.0299134254455566,
      "step": 16671
    },
    {
      "epoch": 1.5898704528808593e-05,
      "step": 16671,
      "training_step_time": 8.77608323097229
    },
    {
      "epoch": 1.5899658203125e-05,
      "model_forward_time": 1.7502539157867432,
      "step": 16672
    },
    {
      "epoch": 1.5899658203125e-05,
      "step": 16672,
      "training_step_time": 8.817151546478271
    },
    {
      "epoch": 1.5900611877441405e-05,
      "model_forward_time": 1.7706799507141113,
      "step": 16673
    },
    {
      "epoch": 1.5900611877441405e-05,
      "step": 16673,
      "training_step_time": 8.793053150177002
    },
    {
      "epoch": 1.5901565551757812e-05,
      "model_forward_time": 1.7682178020477295,
      "step": 16674
    },
    {
      "epoch": 1.5901565551757812e-05,
      "step": 16674,
      "training_step_time": 7.690638780593872
    },
    {
      "epoch": 1.590251922607422e-05,
      "model_forward_time": 1.402923345565796,
      "step": 16675
    },
    {
      "epoch": 1.590251922607422e-05,
      "step": 16675,
      "training_step_time": 7.758305311203003
    },
    {
      "epoch": 1.5903472900390623e-05,
      "model_forward_time": 1.7624897956848145,
      "step": 16676
    },
    {
      "epoch": 1.5903472900390623e-05,
      "step": 16676,
      "training_step_time": 7.721302270889282
    },
    {
      "epoch": 1.590442657470703e-05,
      "model_forward_time": 1.7484652996063232,
      "step": 16677
    },
    {
      "epoch": 1.590442657470703e-05,
      "step": 16677,
      "training_step_time": 7.798727512359619
    },
    {
      "epoch": 1.5905380249023438e-05,
      "model_forward_time": 1.76167631149292,
      "step": 16678
    },
    {
      "epoch": 1.5905380249023438e-05,
      "step": 16678,
      "training_step_time": 7.626194715499878
    },
    {
      "epoch": 1.5906333923339842e-05,
      "model_forward_time": 1.7616233825683594,
      "step": 16679
    },
    {
      "epoch": 1.5906333923339842e-05,
      "step": 16679,
      "training_step_time": 8.414292573928833
    },
    {
      "epoch": 1.590728759765625e-05,
      "grad_norm": 0.2251034528017044,
      "learning_rate": 9.631702550831878e-06,
      "loss": 0.0597,
      "step": 16680
    },
    {
      "epoch": 1.590728759765625e-05,
      "model_forward_time": 3.008291482925415,
      "step": 16680
    },
    {
      "dynamics_loss_avg": 0.05121974069625139,
      "epoch": 1.590728759765625e-05,
      "step": 16680
    },
    {
      "action_loss_avg": 0.002742098143789917,
      "epoch": 1.590728759765625e-05,
      "step": 16680
    },
    {
      "epoch": 1.590728759765625e-05,
      "step": 16680,
      "training_step_time": 8.484326362609863
    },
    {
      "epoch": 1.5908241271972657e-05,
      "model_forward_time": 1.767090082168579,
      "step": 16681
    },
    {
      "epoch": 1.5908241271972657e-05,
      "step": 16681,
      "training_step_time": 7.81603479385376
    },
    {
      "epoch": 1.590919494628906e-05,
      "model_forward_time": 1.7598183155059814,
      "step": 16682
    },
    {
      "epoch": 1.590919494628906e-05,
      "step": 16682,
      "training_step_time": 7.750996112823486
    },
    {
      "epoch": 1.591014862060547e-05,
      "model_forward_time": 1.7575621604919434,
      "step": 16683
    },
    {
      "epoch": 1.591014862060547e-05,
      "step": 16683,
      "training_step_time": 7.594470262527466
    },
    {
      "epoch": 1.5911102294921876e-05,
      "model_forward_time": 1.3832802772521973,
      "step": 16684
    },
    {
      "epoch": 1.5911102294921876e-05,
      "step": 16684,
      "training_step_time": 7.629333734512329
    },
    {
      "epoch": 1.591205596923828e-05,
      "model_forward_time": 1.7496261596679688,
      "step": 16685
    },
    {
      "epoch": 1.591205596923828e-05,
      "step": 16685,
      "training_step_time": 8.422399282455444
    },
    {
      "epoch": 1.5913009643554687e-05,
      "model_forward_time": 1.7744207382202148,
      "step": 16686
    },
    {
      "epoch": 1.5913009643554687e-05,
      "step": 16686,
      "training_step_time": 7.6748809814453125
    },
    {
      "epoch": 1.5913963317871095e-05,
      "model_forward_time": 1.7514135837554932,
      "step": 16687
    },
    {
      "epoch": 1.5913963317871095e-05,
      "step": 16687,
      "training_step_time": 7.839975595474243
    },
    {
      "epoch": 1.59149169921875e-05,
      "model_forward_time": 1.7617130279541016,
      "step": 16688
    },
    {
      "epoch": 1.59149169921875e-05,
      "step": 16688,
      "training_step_time": 8.080448865890503
    },
    {
      "epoch": 1.5915870666503906e-05,
      "model_forward_time": 2.877371311187744,
      "step": 16689
    },
    {
      "epoch": 1.5915870666503906e-05,
      "step": 16689,
      "training_step_time": 7.971220016479492
    },
    {
      "epoch": 1.5916824340820314e-05,
      "grad_norm": 0.13642549514770508,
      "learning_rate": 9.631079456641699e-06,
      "loss": 0.0575,
      "step": 16690
    },
    {
      "epoch": 1.5916824340820314e-05,
      "model_forward_time": 1.868781566619873,
      "step": 16690
    },
    {
      "dynamics_loss_avg": 0.05953192710876465,
      "epoch": 1.5916824340820314e-05,
      "step": 16690
    },
    {
      "action_loss_avg": 0.0024950766412075607,
      "epoch": 1.5916824340820314e-05,
      "step": 16690
    },
    {
      "epoch": 1.5916824340820314e-05,
      "step": 16690,
      "training_step_time": 8.712266206741333
    },
    {
      "epoch": 1.5917778015136718e-05,
      "model_forward_time": 1.770331621170044,
      "step": 16691
    },
    {
      "epoch": 1.5917778015136718e-05,
      "step": 16691,
      "training_step_time": 7.8013975620269775
    },
    {
      "epoch": 1.5918731689453125e-05,
      "model_forward_time": 1.7580525875091553,
      "step": 16692
    },
    {
      "epoch": 1.5918731689453125e-05,
      "step": 16692,
      "training_step_time": 7.660049676895142
    },
    {
      "epoch": 1.5919685363769533e-05,
      "model_forward_time": 1.7604641914367676,
      "step": 16693
    },
    {
      "epoch": 1.5919685363769533e-05,
      "step": 16693,
      "training_step_time": 7.586540699005127
    },
    {
      "epoch": 1.5920639038085937e-05,
      "model_forward_time": 1.7602427005767822,
      "step": 16694
    },
    {
      "epoch": 1.5920639038085937e-05,
      "step": 16694,
      "training_step_time": 7.48868465423584
    },
    {
      "epoch": 1.5921592712402344e-05,
      "model_forward_time": 1.7652561664581299,
      "step": 16695
    },
    {
      "epoch": 1.5921592712402344e-05,
      "step": 16695,
      "training_step_time": 7.549905300140381
    },
    {
      "epoch": 1.592254638671875e-05,
      "model_forward_time": 1.7599122524261475,
      "step": 16696
    },
    {
      "epoch": 1.592254638671875e-05,
      "step": 16696,
      "training_step_time": 7.756192922592163
    },
    {
      "epoch": 1.5923500061035155e-05,
      "model_forward_time": 1.7635114192962646,
      "step": 16697
    },
    {
      "epoch": 1.5923500061035155e-05,
      "step": 16697,
      "training_step_time": 7.669758081436157
    },
    {
      "epoch": 1.5924453735351563e-05,
      "model_forward_time": 1.7608346939086914,
      "step": 16698
    },
    {
      "epoch": 1.5924453735351563e-05,
      "step": 16698,
      "training_step_time": 9.833781480789185
    },
    {
      "epoch": 1.592540740966797e-05,
      "model_forward_time": 1.8458607196807861,
      "step": 16699
    },
    {
      "epoch": 1.592540740966797e-05,
      "step": 16699,
      "training_step_time": 8.544230461120605
    },
    {
      "epoch": 1.5926361083984374e-05,
      "grad_norm": 0.11008303612470627,
      "learning_rate": 9.630455856003633e-06,
      "loss": 0.0592,
      "step": 16700
    },
    {
      "epoch": 1.5926361083984374e-05,
      "model_forward_time": 1.7661256790161133,
      "step": 16700
    },
    {
      "dynamics_loss_avg": 0.057360834628343585,
      "epoch": 1.5926361083984374e-05,
      "step": 16700
    },
    {
      "action_loss_avg": 0.0023822194896638395,
      "epoch": 1.5926361083984374e-05,
      "step": 16700
    },
    {
      "epoch": 1.5926361083984374e-05,
      "step": 16700,
      "training_step_time": 7.871260643005371
    },
    {
      "epoch": 1.5927314758300782e-05,
      "model_forward_time": 1.7617242336273193,
      "step": 16701
    },
    {
      "epoch": 1.5927314758300782e-05,
      "step": 16701,
      "training_step_time": 7.621126890182495
    },
    {
      "epoch": 1.592826843261719e-05,
      "model_forward_time": 1.7560193538665771,
      "step": 16702
    },
    {
      "epoch": 1.592826843261719e-05,
      "step": 16702,
      "training_step_time": 7.603375196456909
    },
    {
      "epoch": 1.5929222106933593e-05,
      "model_forward_time": 1.7569241523742676,
      "step": 16703
    },
    {
      "epoch": 1.5929222106933593e-05,
      "step": 16703,
      "training_step_time": 7.675704717636108
    },
    {
      "epoch": 1.593017578125e-05,
      "model_forward_time": 1.7619915008544922,
      "step": 16704
    },
    {
      "epoch": 1.593017578125e-05,
      "step": 16704,
      "training_step_time": 7.56798791885376
    },
    {
      "epoch": 1.5931129455566405e-05,
      "model_forward_time": 1.7643473148345947,
      "step": 16705
    },
    {
      "epoch": 1.5931129455566405e-05,
      "step": 16705,
      "training_step_time": 8.602251529693604
    },
    {
      "epoch": 1.5932083129882812e-05,
      "model_forward_time": 1.7579371929168701,
      "step": 16706
    },
    {
      "epoch": 1.5932083129882812e-05,
      "step": 16706,
      "training_step_time": 8.559349536895752
    },
    {
      "epoch": 1.593303680419922e-05,
      "model_forward_time": 1.3887391090393066,
      "step": 16707
    },
    {
      "epoch": 1.593303680419922e-05,
      "step": 16707,
      "training_step_time": 7.779245853424072
    },
    {
      "epoch": 1.5933990478515623e-05,
      "model_forward_time": 2.9406542778015137,
      "step": 16708
    },
    {
      "epoch": 1.5933990478515623e-05,
      "step": 16708,
      "training_step_time": 7.885112524032593
    },
    {
      "epoch": 1.593494415283203e-05,
      "model_forward_time": 1.7574222087860107,
      "step": 16709
    },
    {
      "epoch": 1.593494415283203e-05,
      "step": 16709,
      "training_step_time": 9.70756196975708
    },
    {
      "epoch": 1.593589782714844e-05,
      "grad_norm": 0.28993621468544006,
      "learning_rate": 9.629831748985877e-06,
      "loss": 0.0634,
      "step": 16710
    },
    {
      "epoch": 1.593589782714844e-05,
      "model_forward_time": 1.7449588775634766,
      "step": 16710
    },
    {
      "dynamics_loss_avg": 0.06109078414738178,
      "epoch": 1.593589782714844e-05,
      "step": 16710
    },
    {
      "action_loss_avg": 0.00291886231279932,
      "epoch": 1.593589782714844e-05,
      "step": 16710
    },
    {
      "epoch": 1.593589782714844e-05,
      "step": 16710,
      "training_step_time": 8.409407138824463
    },
    {
      "epoch": 1.5936851501464842e-05,
      "model_forward_time": 1.7634763717651367,
      "step": 16711
    },
    {
      "epoch": 1.5936851501464842e-05,
      "step": 16711,
      "training_step_time": 8.833913564682007
    },
    {
      "epoch": 1.593780517578125e-05,
      "model_forward_time": 1.7616114616394043,
      "step": 16712
    },
    {
      "epoch": 1.593780517578125e-05,
      "step": 16712,
      "training_step_time": 7.820145845413208
    },
    {
      "epoch": 1.5938758850097657e-05,
      "model_forward_time": 1.7586112022399902,
      "step": 16713
    },
    {
      "epoch": 1.5938758850097657e-05,
      "step": 16713,
      "training_step_time": 7.741998195648193
    },
    {
      "epoch": 1.593971252441406e-05,
      "model_forward_time": 1.7543268203735352,
      "step": 16714
    },
    {
      "epoch": 1.593971252441406e-05,
      "step": 16714,
      "training_step_time": 7.79046106338501
    },
    {
      "epoch": 1.594066619873047e-05,
      "model_forward_time": 1.7690842151641846,
      "step": 16715
    },
    {
      "epoch": 1.594066619873047e-05,
      "step": 16715,
      "training_step_time": 7.759050607681274
    },
    {
      "epoch": 1.5941619873046876e-05,
      "model_forward_time": 1.7708678245544434,
      "step": 16716
    },
    {
      "epoch": 1.5941619873046876e-05,
      "step": 16716,
      "training_step_time": 8.391103029251099
    },
    {
      "epoch": 1.594257354736328e-05,
      "model_forward_time": 2.943657636642456,
      "step": 16717
    },
    {
      "epoch": 1.594257354736328e-05,
      "step": 16717,
      "training_step_time": 11.15703797340393
    },
    {
      "epoch": 1.5943527221679687e-05,
      "model_forward_time": 1.7611770629882812,
      "step": 16718
    },
    {
      "epoch": 1.5943527221679687e-05,
      "step": 16718,
      "training_step_time": 9.520665407180786
    },
    {
      "epoch": 1.5944480895996095e-05,
      "model_forward_time": 1.7633905410766602,
      "step": 16719
    },
    {
      "epoch": 1.5944480895996095e-05,
      "step": 16719,
      "training_step_time": 8.532384157180786
    },
    {
      "epoch": 1.59454345703125e-05,
      "grad_norm": 0.1886463165283203,
      "learning_rate": 9.62920713565668e-06,
      "loss": 0.0584,
      "step": 16720
    },
    {
      "epoch": 1.59454345703125e-05,
      "model_forward_time": 1.7713642120361328,
      "step": 16720
    },
    {
      "dynamics_loss_avg": 0.060284494608640674,
      "epoch": 1.59454345703125e-05,
      "step": 16720
    },
    {
      "action_loss_avg": 0.004882851958973334,
      "epoch": 1.59454345703125e-05,
      "step": 16720
    },
    {
      "epoch": 1.59454345703125e-05,
      "step": 16720,
      "training_step_time": 7.57432746887207
    },
    {
      "epoch": 1.5946388244628906e-05,
      "model_forward_time": 1.7597992420196533,
      "step": 16721
    },
    {
      "epoch": 1.5946388244628906e-05,
      "step": 16721,
      "training_step_time": 8.087891817092896
    },
    {
      "epoch": 1.5947341918945314e-05,
      "model_forward_time": 1.7575042247772217,
      "step": 16722
    },
    {
      "epoch": 1.5947341918945314e-05,
      "step": 16722,
      "training_step_time": 7.511100769042969
    },
    {
      "epoch": 1.5948295593261718e-05,
      "model_forward_time": 1.7571732997894287,
      "step": 16723
    },
    {
      "epoch": 1.5948295593261718e-05,
      "step": 16723,
      "training_step_time": 8.670705318450928
    },
    {
      "epoch": 1.5949249267578125e-05,
      "model_forward_time": 1.7508177757263184,
      "step": 16724
    },
    {
      "epoch": 1.5949249267578125e-05,
      "step": 16724,
      "training_step_time": 7.726803302764893
    },
    {
      "epoch": 1.5950202941894533e-05,
      "model_forward_time": 1.38919997215271,
      "step": 16725
    },
    {
      "epoch": 1.5950202941894533e-05,
      "step": 16725,
      "training_step_time": 7.791428804397583
    },
    {
      "epoch": 1.5951156616210937e-05,
      "model_forward_time": 2.8766722679138184,
      "step": 16726
    },
    {
      "epoch": 1.5951156616210937e-05,
      "step": 16726,
      "training_step_time": 8.627696752548218
    },
    {
      "epoch": 1.5952110290527344e-05,
      "model_forward_time": 1.8598783016204834,
      "step": 16727
    },
    {
      "epoch": 1.5952110290527344e-05,
      "step": 16727,
      "training_step_time": 7.96688985824585
    },
    {
      "epoch": 1.595306396484375e-05,
      "model_forward_time": 1.7659378051757812,
      "step": 16728
    },
    {
      "epoch": 1.595306396484375e-05,
      "step": 16728,
      "training_step_time": 8.572031021118164
    },
    {
      "epoch": 1.5954017639160155e-05,
      "model_forward_time": 1.7627794742584229,
      "step": 16729
    },
    {
      "epoch": 1.5954017639160155e-05,
      "step": 16729,
      "training_step_time": 7.791314601898193
    },
    {
      "epoch": 1.5954971313476563e-05,
      "grad_norm": 0.22341708838939667,
      "learning_rate": 9.628582016084353e-06,
      "loss": 0.0637,
      "step": 16730
    },
    {
      "epoch": 1.5954971313476563e-05,
      "model_forward_time": 1.7727651596069336,
      "step": 16730
    },
    {
      "dynamics_loss_avg": 0.06577474754303694,
      "epoch": 1.5954971313476563e-05,
      "step": 16730
    },
    {
      "action_loss_avg": 0.005515763652510941,
      "epoch": 1.5954971313476563e-05,
      "step": 16730
    },
    {
      "epoch": 1.5954971313476563e-05,
      "step": 16730,
      "training_step_time": 7.726050853729248
    },
    {
      "epoch": 1.595592498779297e-05,
      "model_forward_time": 1.7499761581420898,
      "step": 16731
    },
    {
      "epoch": 1.595592498779297e-05,
      "step": 16731,
      "training_step_time": 7.816748857498169
    },
    {
      "epoch": 1.5956878662109374e-05,
      "model_forward_time": 1.7603271007537842,
      "step": 16732
    },
    {
      "epoch": 1.5956878662109374e-05,
      "step": 16732,
      "training_step_time": 7.889243125915527
    },
    {
      "epoch": 1.5957832336425782e-05,
      "model_forward_time": 1.7829959392547607,
      "step": 16733
    },
    {
      "epoch": 1.5957832336425782e-05,
      "step": 16733,
      "training_step_time": 8.824885845184326
    },
    {
      "epoch": 1.595878601074219e-05,
      "model_forward_time": 1.7552258968353271,
      "step": 16734
    },
    {
      "epoch": 1.595878601074219e-05,
      "step": 16734,
      "training_step_time": 8.706316709518433
    },
    {
      "epoch": 1.5959739685058593e-05,
      "model_forward_time": 1.75376558303833,
      "step": 16735
    },
    {
      "epoch": 1.5959739685058593e-05,
      "step": 16735,
      "training_step_time": 10.037355899810791
    },
    {
      "epoch": 1.5960693359375e-05,
      "model_forward_time": 2.9757070541381836,
      "step": 16736
    },
    {
      "epoch": 1.5960693359375e-05,
      "step": 16736,
      "training_step_time": 8.992034196853638
    },
    {
      "epoch": 1.5961647033691405e-05,
      "model_forward_time": 1.7760035991668701,
      "step": 16737
    },
    {
      "epoch": 1.5961647033691405e-05,
      "step": 16737,
      "training_step_time": 7.8890700340271
    },
    {
      "epoch": 1.5962600708007812e-05,
      "model_forward_time": 1.7547097206115723,
      "step": 16738
    },
    {
      "epoch": 1.5962600708007812e-05,
      "step": 16738,
      "training_step_time": 9.134434700012207
    },
    {
      "epoch": 1.596355438232422e-05,
      "model_forward_time": 1.7709338665008545,
      "step": 16739
    },
    {
      "epoch": 1.596355438232422e-05,
      "step": 16739,
      "training_step_time": 7.746255397796631
    },
    {
      "epoch": 1.5964508056640624e-05,
      "grad_norm": 0.12765353918075562,
      "learning_rate": 9.627956390337254e-06,
      "loss": 0.0595,
      "step": 16740
    },
    {
      "epoch": 1.5964508056640624e-05,
      "model_forward_time": 1.7629709243774414,
      "step": 16740
    },
    {
      "dynamics_loss_avg": 0.04760458227247,
      "epoch": 1.5964508056640624e-05,
      "step": 16740
    },
    {
      "action_loss_avg": 0.0017908805806655438,
      "epoch": 1.5964508056640624e-05,
      "step": 16740
    },
    {
      "epoch": 1.5964508056640624e-05,
      "step": 16740,
      "training_step_time": 7.654834985733032
    },
    {
      "epoch": 1.596546173095703e-05,
      "model_forward_time": 1.7661192417144775,
      "step": 16741
    },
    {
      "epoch": 1.596546173095703e-05,
      "step": 16741,
      "training_step_time": 7.841681003570557
    },
    {
      "epoch": 1.596641540527344e-05,
      "model_forward_time": 1.761622428894043,
      "step": 16742
    },
    {
      "epoch": 1.596641540527344e-05,
      "step": 16742,
      "training_step_time": 7.741705656051636
    },
    {
      "epoch": 1.5967369079589842e-05,
      "model_forward_time": 1.7614686489105225,
      "step": 16743
    },
    {
      "epoch": 1.5967369079589842e-05,
      "step": 16743,
      "training_step_time": 9.110268592834473
    },
    {
      "epoch": 1.596832275390625e-05,
      "model_forward_time": 1.7599494457244873,
      "step": 16744
    },
    {
      "epoch": 1.596832275390625e-05,
      "step": 16744,
      "training_step_time": 8.657614946365356
    },
    {
      "epoch": 1.5969276428222657e-05,
      "model_forward_time": 2.992396354675293,
      "step": 16745
    },
    {
      "epoch": 1.5969276428222657e-05,
      "step": 16745,
      "training_step_time": 8.710631847381592
    },
    {
      "epoch": 1.597023010253906e-05,
      "model_forward_time": 1.745548963546753,
      "step": 16746
    },
    {
      "epoch": 1.597023010253906e-05,
      "step": 16746,
      "training_step_time": 7.6368794441223145
    },
    {
      "epoch": 1.597118377685547e-05,
      "model_forward_time": 1.7615208625793457,
      "step": 16747
    },
    {
      "epoch": 1.597118377685547e-05,
      "step": 16747,
      "training_step_time": 8.430156707763672
    },
    {
      "epoch": 1.5972137451171876e-05,
      "model_forward_time": 1.7638351917266846,
      "step": 16748
    },
    {
      "epoch": 1.5972137451171876e-05,
      "step": 16748,
      "training_step_time": 7.509735107421875
    },
    {
      "epoch": 1.597309112548828e-05,
      "model_forward_time": 1.7667005062103271,
      "step": 16749
    },
    {
      "epoch": 1.597309112548828e-05,
      "step": 16749,
      "training_step_time": 7.587995767593384
    },
    {
      "epoch": 1.5974044799804688e-05,
      "grad_norm": 0.22086599469184875,
      "learning_rate": 9.627330258483802e-06,
      "loss": 0.063,
      "step": 16750
    },
    {
      "epoch": 1.5974044799804688e-05,
      "model_forward_time": 1.7625298500061035,
      "step": 16750
    },
    {
      "dynamics_loss_avg": 0.038105175457894803,
      "epoch": 1.5974044799804688e-05,
      "step": 16750
    },
    {
      "action_loss_avg": 0.0024955722474260256,
      "epoch": 1.5974044799804688e-05,
      "step": 16750
    },
    {
      "epoch": 1.5974044799804688e-05,
      "step": 16750,
      "training_step_time": 7.5548083782196045
    },
    {
      "epoch": 1.5974998474121095e-05,
      "model_forward_time": 1.759737253189087,
      "step": 16751
    },
    {
      "epoch": 1.5974998474121095e-05,
      "step": 16751,
      "training_step_time": 7.628647565841675
    },
    {
      "epoch": 1.59759521484375e-05,
      "model_forward_time": 1.7613646984100342,
      "step": 16752
    },
    {
      "epoch": 1.59759521484375e-05,
      "step": 16752,
      "training_step_time": 7.62297248840332
    },
    {
      "epoch": 1.5976905822753906e-05,
      "model_forward_time": 1.767247200012207,
      "step": 16753
    },
    {
      "epoch": 1.5976905822753906e-05,
      "step": 16753,
      "training_step_time": 7.614946603775024
    },
    {
      "epoch": 1.5977859497070314e-05,
      "model_forward_time": 2.926276683807373,
      "step": 16754
    },
    {
      "epoch": 1.5977859497070314e-05,
      "step": 16754,
      "training_step_time": 7.746782541275024
    },
    {
      "epoch": 1.5978813171386718e-05,
      "model_forward_time": 1.8634498119354248,
      "step": 16755
    },
    {
      "epoch": 1.5978813171386718e-05,
      "step": 16755,
      "training_step_time": 8.637939929962158
    },
    {
      "epoch": 1.5979766845703125e-05,
      "model_forward_time": 1.7599012851715088,
      "step": 16756
    },
    {
      "epoch": 1.5979766845703125e-05,
      "step": 16756,
      "training_step_time": 7.73000168800354
    },
    {
      "epoch": 1.5980720520019533e-05,
      "model_forward_time": 1.7622621059417725,
      "step": 16757
    },
    {
      "epoch": 1.5980720520019533e-05,
      "step": 16757,
      "training_step_time": 7.595073223114014
    },
    {
      "epoch": 1.5981674194335937e-05,
      "model_forward_time": 1.7596204280853271,
      "step": 16758
    },
    {
      "epoch": 1.5981674194335937e-05,
      "step": 16758,
      "training_step_time": 7.65686297416687
    },
    {
      "epoch": 1.5982627868652344e-05,
      "model_forward_time": 1.3912153244018555,
      "step": 16759
    },
    {
      "epoch": 1.5982627868652344e-05,
      "step": 16759,
      "training_step_time": 7.509599924087524
    },
    {
      "epoch": 1.598358154296875e-05,
      "grad_norm": 0.1781841665506363,
      "learning_rate": 9.62670362059247e-06,
      "loss": 0.0621,
      "step": 16760
    },
    {
      "epoch": 1.598358154296875e-05,
      "model_forward_time": 1.7504138946533203,
      "step": 16760
    },
    {
      "dynamics_loss_avg": 0.054402418434619904,
      "epoch": 1.598358154296875e-05,
      "step": 16760
    },
    {
      "action_loss_avg": 0.0034907726338133218,
      "epoch": 1.598358154296875e-05,
      "step": 16760
    },
    {
      "epoch": 1.598358154296875e-05,
      "step": 16760,
      "training_step_time": 8.880329608917236
    },
    {
      "epoch": 1.5984535217285156e-05,
      "model_forward_time": 1.755387544631958,
      "step": 16761
    },
    {
      "epoch": 1.5984535217285156e-05,
      "step": 16761,
      "training_step_time": 7.7640321254730225
    },
    {
      "epoch": 1.5985488891601563e-05,
      "model_forward_time": 1.7602739334106445,
      "step": 16762
    },
    {
      "epoch": 1.5985488891601563e-05,
      "step": 16762,
      "training_step_time": 7.6879658699035645
    },
    {
      "epoch": 1.598644256591797e-05,
      "model_forward_time": 2.8730461597442627,
      "step": 16763
    },
    {
      "epoch": 1.598644256591797e-05,
      "step": 16763,
      "training_step_time": 8.59955883026123
    },
    {
      "epoch": 1.5987396240234374e-05,
      "model_forward_time": 1.862433671951294,
      "step": 16764
    },
    {
      "epoch": 1.5987396240234374e-05,
      "step": 16764,
      "training_step_time": 7.984224796295166
    },
    {
      "epoch": 1.5988349914550782e-05,
      "model_forward_time": 1.7651796340942383,
      "step": 16765
    },
    {
      "epoch": 1.5988349914550782e-05,
      "step": 16765,
      "training_step_time": 7.791995048522949
    },
    {
      "epoch": 1.5989303588867186e-05,
      "model_forward_time": 1.7643225193023682,
      "step": 16766
    },
    {
      "epoch": 1.5989303588867186e-05,
      "step": 16766,
      "training_step_time": 7.757797718048096
    },
    {
      "epoch": 1.5990257263183593e-05,
      "model_forward_time": 1.7582805156707764,
      "step": 16767
    },
    {
      "epoch": 1.5990257263183593e-05,
      "step": 16767,
      "training_step_time": 7.7927680015563965
    },
    {
      "epoch": 1.59912109375e-05,
      "model_forward_time": 1.7607293128967285,
      "step": 16768
    },
    {
      "epoch": 1.59912109375e-05,
      "step": 16768,
      "training_step_time": 7.662744045257568
    },
    {
      "epoch": 1.5992164611816405e-05,
      "model_forward_time": 1.7690677642822266,
      "step": 16769
    },
    {
      "epoch": 1.5992164611816405e-05,
      "step": 16769,
      "training_step_time": 7.875089645385742
    },
    {
      "epoch": 1.5993118286132812e-05,
      "grad_norm": 0.21391983330249786,
      "learning_rate": 9.626076476731786e-06,
      "loss": 0.0659,
      "step": 16770
    },
    {
      "epoch": 1.5993118286132812e-05,
      "model_forward_time": 1.7513558864593506,
      "step": 16770
    },
    {
      "dynamics_loss_avg": 0.06046200562268496,
      "epoch": 1.5993118286132812e-05,
      "step": 16770
    },
    {
      "action_loss_avg": 0.002292103960644454,
      "epoch": 1.5993118286132812e-05,
      "step": 16770
    },
    {
      "epoch": 1.5993118286132812e-05,
      "step": 16770,
      "training_step_time": 7.825659513473511
    },
    {
      "epoch": 1.599407196044922e-05,
      "model_forward_time": 1.7620062828063965,
      "step": 16771
    },
    {
      "epoch": 1.599407196044922e-05,
      "step": 16771,
      "training_step_time": 8.294959783554077
    },
    {
      "epoch": 1.5995025634765624e-05,
      "model_forward_time": 1.757157325744629,
      "step": 16772
    },
    {
      "epoch": 1.5995025634765624e-05,
      "step": 16772,
      "training_step_time": 9.226114273071289
    },
    {
      "epoch": 1.599597930908203e-05,
      "model_forward_time": 1.8317553997039795,
      "step": 16773
    },
    {
      "epoch": 1.599597930908203e-05,
      "step": 16773,
      "training_step_time": 8.159203290939331
    },
    {
      "epoch": 1.599693298339844e-05,
      "model_forward_time": 1.7643659114837646,
      "step": 16774
    },
    {
      "epoch": 1.599693298339844e-05,
      "step": 16774,
      "training_step_time": 8.412367820739746
    },
    {
      "epoch": 1.5997886657714842e-05,
      "model_forward_time": 1.7635576725006104,
      "step": 16775
    },
    {
      "epoch": 1.5997886657714842e-05,
      "step": 16775,
      "training_step_time": 7.681168556213379
    },
    {
      "epoch": 1.599884033203125e-05,
      "model_forward_time": 1.3975224494934082,
      "step": 16776
    },
    {
      "epoch": 1.599884033203125e-05,
      "step": 16776,
      "training_step_time": 7.552356481552124
    },
    {
      "epoch": 1.5999794006347657e-05,
      "model_forward_time": 1.7636768817901611,
      "step": 16777
    },
    {
      "epoch": 1.5999794006347657e-05,
      "step": 16777,
      "training_step_time": 8.376629829406738
    },
    {
      "epoch": 1.600074768066406e-05,
      "model_forward_time": 1.7503080368041992,
      "step": 16778
    },
    {
      "epoch": 1.600074768066406e-05,
      "step": 16778,
      "training_step_time": 8.19359827041626
    },
    {
      "epoch": 1.600170135498047e-05,
      "model_forward_time": 1.7532951831817627,
      "step": 16779
    },
    {
      "epoch": 1.600170135498047e-05,
      "step": 16779,
      "training_step_time": 7.638720273971558
    },
    {
      "epoch": 1.6002655029296876e-05,
      "grad_norm": 0.1794060468673706,
      "learning_rate": 9.625448826970336e-06,
      "loss": 0.063,
      "step": 16780
    },
    {
      "epoch": 1.6002655029296876e-05,
      "model_forward_time": 1.7886009216308594,
      "step": 16780
    },
    {
      "dynamics_loss_avg": 0.060721137933433054,
      "epoch": 1.6002655029296876e-05,
      "step": 16780
    },
    {
      "action_loss_avg": 0.0031579109898302706,
      "epoch": 1.6002655029296876e-05,
      "step": 16780
    },
    {
      "epoch": 1.6002655029296876e-05,
      "step": 16780,
      "training_step_time": 8.068046569824219
    },
    {
      "epoch": 1.600360870361328e-05,
      "model_forward_time": 1.7513065338134766,
      "step": 16781
    },
    {
      "epoch": 1.600360870361328e-05,
      "step": 16781,
      "training_step_time": 7.793276309967041
    },
    {
      "epoch": 1.6004562377929688e-05,
      "model_forward_time": 2.984800338745117,
      "step": 16782
    },
    {
      "epoch": 1.6004562377929688e-05,
      "step": 16782,
      "training_step_time": 8.776071071624756
    },
    {
      "epoch": 1.6005516052246095e-05,
      "model_forward_time": 1.7602736949920654,
      "step": 16783
    },
    {
      "epoch": 1.6005516052246095e-05,
      "step": 16783,
      "training_step_time": 9.024900436401367
    },
    {
      "epoch": 1.60064697265625e-05,
      "model_forward_time": 1.7704095840454102,
      "step": 16784
    },
    {
      "epoch": 1.60064697265625e-05,
      "step": 16784,
      "training_step_time": 7.615916013717651
    },
    {
      "epoch": 1.6007423400878906e-05,
      "model_forward_time": 1.7565293312072754,
      "step": 16785
    },
    {
      "epoch": 1.6007423400878906e-05,
      "step": 16785,
      "training_step_time": 7.434019088745117
    },
    {
      "epoch": 1.6008377075195314e-05,
      "model_forward_time": 1.7574107646942139,
      "step": 16786
    },
    {
      "epoch": 1.6008377075195314e-05,
      "step": 16786,
      "training_step_time": 7.640042304992676
    },
    {
      "epoch": 1.6009330749511718e-05,
      "model_forward_time": 1.7617101669311523,
      "step": 16787
    },
    {
      "epoch": 1.6009330749511718e-05,
      "step": 16787,
      "training_step_time": 8.781916856765747
    },
    {
      "epoch": 1.6010284423828125e-05,
      "model_forward_time": 1.7380955219268799,
      "step": 16788
    },
    {
      "epoch": 1.6010284423828125e-05,
      "step": 16788,
      "training_step_time": 8.791231393814087
    },
    {
      "epoch": 1.6011238098144533e-05,
      "model_forward_time": 1.7342383861541748,
      "step": 16789
    },
    {
      "epoch": 1.6011238098144533e-05,
      "step": 16789,
      "training_step_time": 7.6152801513671875
    },
    {
      "epoch": 1.6012191772460937e-05,
      "grad_norm": 0.17633867263793945,
      "learning_rate": 9.624820671376755e-06,
      "loss": 0.0649,
      "step": 16790
    },
    {
      "epoch": 1.6012191772460937e-05,
      "model_forward_time": 1.7261536121368408,
      "step": 16790
    },
    {
      "dynamics_loss_avg": 0.05416057063266635,
      "epoch": 1.6012191772460937e-05,
      "step": 16790
    },
    {
      "action_loss_avg": 0.0025398523721378297,
      "epoch": 1.6012191772460937e-05,
      "step": 16790
    },
    {
      "epoch": 1.6012191772460937e-05,
      "step": 16790,
      "training_step_time": 8.990226984024048
    },
    {
      "epoch": 1.6013145446777344e-05,
      "model_forward_time": 2.836620807647705,
      "step": 16791
    },
    {
      "epoch": 1.6013145446777344e-05,
      "step": 16791,
      "training_step_time": 9.316499948501587
    },
    {
      "epoch": 1.601409912109375e-05,
      "model_forward_time": 1.8440563678741455,
      "step": 16792
    },
    {
      "epoch": 1.601409912109375e-05,
      "step": 16792,
      "training_step_time": 8.623449563980103
    },
    {
      "epoch": 1.6015052795410156e-05,
      "model_forward_time": 1.7360551357269287,
      "step": 16793
    },
    {
      "epoch": 1.6015052795410156e-05,
      "step": 16793,
      "training_step_time": 7.663219690322876
    },
    {
      "epoch": 1.6016006469726563e-05,
      "model_forward_time": 1.7291226387023926,
      "step": 16794
    },
    {
      "epoch": 1.6016006469726563e-05,
      "step": 16794,
      "training_step_time": 7.607229948043823
    },
    {
      "epoch": 1.601696014404297e-05,
      "model_forward_time": 1.7483139038085938,
      "step": 16795
    },
    {
      "epoch": 1.601696014404297e-05,
      "step": 16795,
      "training_step_time": 7.671564340591431
    },
    {
      "epoch": 1.6017913818359374e-05,
      "model_forward_time": 1.7355477809906006,
      "step": 16796
    },
    {
      "epoch": 1.6017913818359374e-05,
      "step": 16796,
      "training_step_time": 8.208486557006836
    },
    {
      "epoch": 1.6018867492675782e-05,
      "model_forward_time": 1.7283778190612793,
      "step": 16797
    },
    {
      "epoch": 1.6018867492675782e-05,
      "step": 16797,
      "training_step_time": 7.727243900299072
    },
    {
      "epoch": 1.6019821166992186e-05,
      "model_forward_time": 1.7268743515014648,
      "step": 16798
    },
    {
      "epoch": 1.6019821166992186e-05,
      "step": 16798,
      "training_step_time": 8.393226146697998
    },
    {
      "epoch": 1.6020774841308593e-05,
      "model_forward_time": 1.733274221420288,
      "step": 16799
    },
    {
      "epoch": 1.6020774841308593e-05,
      "step": 16799,
      "training_step_time": 9.31731653213501
    },
    {
      "epoch": 1.6021728515625e-05,
      "grad_norm": 0.14712128043174744,
      "learning_rate": 9.624192010019739e-06,
      "loss": 0.066,
      "step": 16800
    },
    {
      "epoch": 1.6021728515625e-05,
      "model_forward_time": 2.4919140338897705,
      "step": 16800
    },
    {
      "dynamics_loss_avg": 0.05991463102400303,
      "epoch": 1.6021728515625e-05,
      "step": 16800
    },
    {
      "action_loss_avg": 0.0033550279680639506,
      "epoch": 1.6021728515625e-05,
      "step": 16800
    },
    {
      "epoch": 1.6021728515625e-05,
      "step": 16800,
      "training_step_time": 8.638777732849121
    },
    {
      "epoch": 1.6022682189941405e-05,
      "model_forward_time": 1.840360403060913,
      "step": 16801
    },
    {
      "epoch": 1.6022682189941405e-05,
      "step": 16801,
      "training_step_time": 7.881267309188843
    },
    {
      "epoch": 1.6023635864257812e-05,
      "model_forward_time": 1.716512680053711,
      "step": 16802
    },
    {
      "epoch": 1.6023635864257812e-05,
      "step": 16802,
      "training_step_time": 7.699222564697266
    },
    {
      "epoch": 1.602458953857422e-05,
      "model_forward_time": 1.726215124130249,
      "step": 16803
    },
    {
      "epoch": 1.602458953857422e-05,
      "step": 16803,
      "training_step_time": 8.61169171333313
    },
    {
      "epoch": 1.6025543212890624e-05,
      "model_forward_time": 1.7514371871948242,
      "step": 16804
    },
    {
      "epoch": 1.6025543212890624e-05,
      "step": 16804,
      "training_step_time": 7.665510654449463
    },
    {
      "epoch": 1.602649688720703e-05,
      "model_forward_time": 1.726616382598877,
      "step": 16805
    },
    {
      "epoch": 1.602649688720703e-05,
      "step": 16805,
      "training_step_time": 8.697116613388062
    },
    {
      "epoch": 1.602745056152344e-05,
      "model_forward_time": 1.7327280044555664,
      "step": 16806
    },
    {
      "epoch": 1.602745056152344e-05,
      "step": 16806,
      "training_step_time": 7.6935224533081055
    },
    {
      "epoch": 1.6028404235839842e-05,
      "model_forward_time": 1.7543659210205078,
      "step": 16807
    },
    {
      "epoch": 1.6028404235839842e-05,
      "step": 16807,
      "training_step_time": 7.622018098831177
    },
    {
      "epoch": 1.602935791015625e-05,
      "model_forward_time": 1.746203899383545,
      "step": 16808
    },
    {
      "epoch": 1.602935791015625e-05,
      "step": 16808,
      "training_step_time": 7.750058889389038
    },
    {
      "epoch": 1.6030311584472657e-05,
      "model_forward_time": 2.785517692565918,
      "step": 16809
    },
    {
      "epoch": 1.6030311584472657e-05,
      "step": 16809,
      "training_step_time": 7.920138359069824
    },
    {
      "epoch": 1.603126525878906e-05,
      "grad_norm": 0.16719436645507812,
      "learning_rate": 9.623562842968038e-06,
      "loss": 0.0641,
      "step": 16810
    },
    {
      "epoch": 1.603126525878906e-05,
      "model_forward_time": 1.8053274154663086,
      "step": 16810
    },
    {
      "dynamics_loss_avg": 0.04581058239564299,
      "epoch": 1.603126525878906e-05,
      "step": 16810
    },
    {
      "action_loss_avg": 0.0017483433475717902,
      "epoch": 1.603126525878906e-05,
      "step": 16810
    },
    {
      "epoch": 1.603126525878906e-05,
      "step": 16810,
      "training_step_time": 8.536564588546753
    },
    {
      "epoch": 1.603221893310547e-05,
      "model_forward_time": 1.7309880256652832,
      "step": 16811
    },
    {
      "epoch": 1.603221893310547e-05,
      "step": 16811,
      "training_step_time": 7.752079725265503
    },
    {
      "epoch": 1.6033172607421876e-05,
      "model_forward_time": 1.7487266063690186,
      "step": 16812
    },
    {
      "epoch": 1.6033172607421876e-05,
      "step": 16812,
      "training_step_time": 8.634098291397095
    },
    {
      "epoch": 1.603412628173828e-05,
      "model_forward_time": 1.7253775596618652,
      "step": 16813
    },
    {
      "epoch": 1.603412628173828e-05,
      "step": 16813,
      "training_step_time": 7.42504620552063
    },
    {
      "epoch": 1.6035079956054688e-05,
      "model_forward_time": 1.7533698081970215,
      "step": 16814
    },
    {
      "epoch": 1.6035079956054688e-05,
      "step": 16814,
      "training_step_time": 7.626779556274414
    },
    {
      "epoch": 1.6036033630371095e-05,
      "model_forward_time": 1.747582197189331,
      "step": 16815
    },
    {
      "epoch": 1.6036033630371095e-05,
      "step": 16815,
      "training_step_time": 7.800535440444946
    },
    {
      "epoch": 1.60369873046875e-05,
      "model_forward_time": 1.7552409172058105,
      "step": 16816
    },
    {
      "epoch": 1.60369873046875e-05,
      "step": 16816,
      "training_step_time": 7.654829263687134
    },
    {
      "epoch": 1.6037940979003906e-05,
      "model_forward_time": 1.757554292678833,
      "step": 16817
    },
    {
      "epoch": 1.6037940979003906e-05,
      "step": 16817,
      "training_step_time": 7.675822734832764
    },
    {
      "epoch": 1.6038894653320314e-05,
      "model_forward_time": 1.7584302425384521,
      "step": 16818
    },
    {
      "epoch": 1.6038894653320314e-05,
      "step": 16818,
      "training_step_time": 9.63161587715149
    },
    {
      "epoch": 1.6039848327636718e-05,
      "model_forward_time": 3.0213050842285156,
      "step": 16819
    },
    {
      "epoch": 1.6039848327636718e-05,
      "step": 16819,
      "training_step_time": 7.881489515304565
    },
    {
      "epoch": 1.6040802001953125e-05,
      "grad_norm": 0.12783904373645782,
      "learning_rate": 9.622933170290454e-06,
      "loss": 0.0655,
      "step": 16820
    },
    {
      "epoch": 1.6040802001953125e-05,
      "model_forward_time": 1.7384305000305176,
      "step": 16820
    },
    {
      "dynamics_loss_avg": 0.06221711933612824,
      "epoch": 1.6040802001953125e-05,
      "step": 16820
    },
    {
      "action_loss_avg": 0.0026753444806672634,
      "epoch": 1.6040802001953125e-05,
      "step": 16820
    },
    {
      "epoch": 1.6040802001953125e-05,
      "step": 16820,
      "training_step_time": 8.046854019165039
    },
    {
      "epoch": 1.6041755676269533e-05,
      "model_forward_time": 1.7603545188903809,
      "step": 16821
    },
    {
      "epoch": 1.6041755676269533e-05,
      "step": 16821,
      "training_step_time": 7.654669284820557
    },
    {
      "epoch": 1.6042709350585937e-05,
      "model_forward_time": 1.7563879489898682,
      "step": 16822
    },
    {
      "epoch": 1.6042709350585937e-05,
      "step": 16822,
      "training_step_time": 7.603071689605713
    },
    {
      "epoch": 1.6043663024902344e-05,
      "model_forward_time": 1.759972333908081,
      "step": 16823
    },
    {
      "epoch": 1.6043663024902344e-05,
      "step": 16823,
      "training_step_time": 7.657608985900879
    },
    {
      "epoch": 1.604461669921875e-05,
      "model_forward_time": 1.757418155670166,
      "step": 16824
    },
    {
      "epoch": 1.604461669921875e-05,
      "step": 16824,
      "training_step_time": 7.820559978485107
    },
    {
      "epoch": 1.6045570373535156e-05,
      "model_forward_time": 1.7351911067962646,
      "step": 16825
    },
    {
      "epoch": 1.6045570373535156e-05,
      "step": 16825,
      "training_step_time": 7.745889663696289
    },
    {
      "epoch": 1.6046524047851563e-05,
      "model_forward_time": 1.7514233589172363,
      "step": 16826
    },
    {
      "epoch": 1.6046524047851563e-05,
      "step": 16826,
      "training_step_time": 9.392677545547485
    },
    {
      "epoch": 1.604747772216797e-05,
      "model_forward_time": 1.7313313484191895,
      "step": 16827
    },
    {
      "epoch": 1.604747772216797e-05,
      "step": 16827,
      "training_step_time": 8.864713668823242
    },
    {
      "epoch": 1.6048431396484374e-05,
      "model_forward_time": 2.9138903617858887,
      "step": 16828
    },
    {
      "epoch": 1.6048431396484374e-05,
      "step": 16828,
      "training_step_time": 9.12608814239502
    },
    {
      "epoch": 1.6049385070800782e-05,
      "model_forward_time": 1.736501932144165,
      "step": 16829
    },
    {
      "epoch": 1.6049385070800782e-05,
      "step": 16829,
      "training_step_time": 8.707713603973389
    },
    {
      "epoch": 1.6050338745117186e-05,
      "grad_norm": 0.18344706296920776,
      "learning_rate": 9.62230299205585e-06,
      "loss": 0.0698,
      "step": 16830
    },
    {
      "epoch": 1.6050338745117186e-05,
      "model_forward_time": 1.7232317924499512,
      "step": 16830
    },
    {
      "dynamics_loss_avg": 0.07262523770332337,
      "epoch": 1.6050338745117186e-05,
      "step": 16830
    },
    {
      "action_loss_avg": 0.005150590883567929,
      "epoch": 1.6050338745117186e-05,
      "step": 16830
    },
    {
      "epoch": 1.6050338745117186e-05,
      "step": 16830,
      "training_step_time": 8.380719661712646
    },
    {
      "epoch": 1.6051292419433593e-05,
      "model_forward_time": 1.7281174659729004,
      "step": 16831
    },
    {
      "epoch": 1.6051292419433593e-05,
      "step": 16831,
      "training_step_time": 7.432543039321899
    },
    {
      "epoch": 1.605224609375e-05,
      "model_forward_time": 1.7278962135314941,
      "step": 16832
    },
    {
      "epoch": 1.605224609375e-05,
      "step": 16832,
      "training_step_time": 7.54334020614624
    },
    {
      "epoch": 1.6053199768066405e-05,
      "model_forward_time": 1.7303366661071777,
      "step": 16833
    },
    {
      "epoch": 1.6053199768066405e-05,
      "step": 16833,
      "training_step_time": 7.6344687938690186
    },
    {
      "epoch": 1.6054153442382812e-05,
      "model_forward_time": 1.7315173149108887,
      "step": 16834
    },
    {
      "epoch": 1.6054153442382812e-05,
      "step": 16834,
      "training_step_time": 7.60809063911438
    },
    {
      "epoch": 1.605510711669922e-05,
      "model_forward_time": 1.725454330444336,
      "step": 16835
    },
    {
      "epoch": 1.605510711669922e-05,
      "step": 16835,
      "training_step_time": 8.500153064727783
    },
    {
      "epoch": 1.6056060791015624e-05,
      "model_forward_time": 1.7271509170532227,
      "step": 16836
    },
    {
      "epoch": 1.6056060791015624e-05,
      "step": 16836,
      "training_step_time": 8.739672183990479
    },
    {
      "epoch": 1.605701446533203e-05,
      "model_forward_time": 2.864999532699585,
      "step": 16837
    },
    {
      "epoch": 1.605701446533203e-05,
      "step": 16837,
      "training_step_time": 7.751249551773071
    },
    {
      "epoch": 1.605796813964844e-05,
      "model_forward_time": 1.8339321613311768,
      "step": 16838
    },
    {
      "epoch": 1.605796813964844e-05,
      "step": 16838,
      "training_step_time": 8.572983741760254
    },
    {
      "epoch": 1.6058921813964842e-05,
      "model_forward_time": 1.7298312187194824,
      "step": 16839
    },
    {
      "epoch": 1.6058921813964842e-05,
      "step": 16839,
      "training_step_time": 8.16068696975708
    },
    {
      "epoch": 1.605987548828125e-05,
      "grad_norm": 0.19486668705940247,
      "learning_rate": 9.62167230833314e-06,
      "loss": 0.0609,
      "step": 16840
    },
    {
      "epoch": 1.605987548828125e-05,
      "model_forward_time": 1.7207069396972656,
      "step": 16840
    },
    {
      "dynamics_loss_avg": 0.05797269237227738,
      "epoch": 1.605987548828125e-05,
      "step": 16840
    },
    {
      "action_loss_avg": 0.0028075204943888822,
      "epoch": 1.605987548828125e-05,
      "step": 16840
    },
    {
      "epoch": 1.605987548828125e-05,
      "step": 16840,
      "training_step_time": 7.761756658554077
    },
    {
      "epoch": 1.6060829162597657e-05,
      "model_forward_time": 1.7335116863250732,
      "step": 16841
    },
    {
      "epoch": 1.6060829162597657e-05,
      "step": 16841,
      "training_step_time": 8.166454553604126
    },
    {
      "epoch": 1.606178283691406e-05,
      "model_forward_time": 1.7314913272857666,
      "step": 16842
    },
    {
      "epoch": 1.606178283691406e-05,
      "step": 16842,
      "training_step_time": 7.578983306884766
    },
    {
      "epoch": 1.606273651123047e-05,
      "model_forward_time": 1.7344512939453125,
      "step": 16843
    },
    {
      "epoch": 1.606273651123047e-05,
      "step": 16843,
      "training_step_time": 7.664439916610718
    },
    {
      "epoch": 1.6063690185546876e-05,
      "model_forward_time": 1.7288756370544434,
      "step": 16844
    },
    {
      "epoch": 1.6063690185546876e-05,
      "step": 16844,
      "training_step_time": 7.504385709762573
    },
    {
      "epoch": 1.606464385986328e-05,
      "model_forward_time": 1.7298362255096436,
      "step": 16845
    },
    {
      "epoch": 1.606464385986328e-05,
      "step": 16845,
      "training_step_time": 8.512823343276978
    },
    {
      "epoch": 1.6065597534179688e-05,
      "model_forward_time": 2.861863136291504,
      "step": 16846
    },
    {
      "epoch": 1.6065597534179688e-05,
      "step": 16846,
      "training_step_time": 8.614082098007202
    },
    {
      "epoch": 1.6066551208496095e-05,
      "model_forward_time": 1.8140771389007568,
      "step": 16847
    },
    {
      "epoch": 1.6066551208496095e-05,
      "step": 16847,
      "training_step_time": 7.595501184463501
    },
    {
      "epoch": 1.60675048828125e-05,
      "model_forward_time": 1.7340121269226074,
      "step": 16848
    },
    {
      "epoch": 1.60675048828125e-05,
      "step": 16848,
      "training_step_time": 7.761590480804443
    },
    {
      "epoch": 1.6068458557128906e-05,
      "model_forward_time": 1.7289338111877441,
      "step": 16849
    },
    {
      "epoch": 1.6068458557128906e-05,
      "step": 16849,
      "training_step_time": 7.716706037521362
    },
    {
      "epoch": 1.6069412231445314e-05,
      "grad_norm": 0.21894842386245728,
      "learning_rate": 9.621041119191295e-06,
      "loss": 0.0639,
      "step": 16850
    },
    {
      "epoch": 1.6069412231445314e-05,
      "model_forward_time": 1.7207577228546143,
      "step": 16850
    },
    {
      "dynamics_loss_avg": 0.06636103317141533,
      "epoch": 1.6069412231445314e-05,
      "step": 16850
    },
    {
      "action_loss_avg": 0.004792778857517987,
      "epoch": 1.6069412231445314e-05,
      "step": 16850
    },
    {
      "epoch": 1.6069412231445314e-05,
      "step": 16850,
      "training_step_time": 7.572015047073364
    },
    {
      "epoch": 1.6070365905761718e-05,
      "model_forward_time": 1.7293508052825928,
      "step": 16851
    },
    {
      "epoch": 1.6070365905761718e-05,
      "step": 16851,
      "training_step_time": 7.58725905418396
    },
    {
      "epoch": 1.6071319580078125e-05,
      "model_forward_time": 1.7303972244262695,
      "step": 16852
    },
    {
      "epoch": 1.6071319580078125e-05,
      "step": 16852,
      "training_step_time": 7.5923566818237305
    },
    {
      "epoch": 1.6072273254394533e-05,
      "model_forward_time": 1.7297394275665283,
      "step": 16853
    },
    {
      "epoch": 1.6072273254394533e-05,
      "step": 16853,
      "training_step_time": 7.551337718963623
    },
    {
      "epoch": 1.6073226928710937e-05,
      "model_forward_time": 1.7513787746429443,
      "step": 16854
    },
    {
      "epoch": 1.6073226928710937e-05,
      "step": 16854,
      "training_step_time": 7.688259601593018
    },
    {
      "epoch": 1.6074180603027344e-05,
      "model_forward_time": 2.819349765777588,
      "step": 16855
    },
    {
      "epoch": 1.6074180603027344e-05,
      "step": 16855,
      "training_step_time": 9.100485563278198
    },
    {
      "epoch": 1.607513427734375e-05,
      "model_forward_time": 1.813659906387329,
      "step": 16856
    },
    {
      "epoch": 1.607513427734375e-05,
      "step": 16856,
      "training_step_time": 8.748964548110962
    },
    {
      "epoch": 1.6076087951660156e-05,
      "model_forward_time": 1.7258415222167969,
      "step": 16857
    },
    {
      "epoch": 1.6076087951660156e-05,
      "step": 16857,
      "training_step_time": 7.722319841384888
    },
    {
      "epoch": 1.6077041625976563e-05,
      "model_forward_time": 1.755357265472412,
      "step": 16858
    },
    {
      "epoch": 1.6077041625976563e-05,
      "step": 16858,
      "training_step_time": 7.76018214225769
    },
    {
      "epoch": 1.607799530029297e-05,
      "model_forward_time": 1.7383499145507812,
      "step": 16859
    },
    {
      "epoch": 1.607799530029297e-05,
      "step": 16859,
      "training_step_time": 7.613402366638184
    },
    {
      "epoch": 1.6078948974609374e-05,
      "grad_norm": 0.18235717713832855,
      "learning_rate": 9.620409424699341e-06,
      "loss": 0.0639,
      "step": 16860
    },
    {
      "epoch": 1.6078948974609374e-05,
      "model_forward_time": 1.7298171520233154,
      "step": 16860
    },
    {
      "dynamics_loss_avg": 0.06875753328204155,
      "epoch": 1.6078948974609374e-05,
      "step": 16860
    },
    {
      "action_loss_avg": 0.00363629205385223,
      "epoch": 1.6078948974609374e-05,
      "step": 16860
    },
    {
      "epoch": 1.6078948974609374e-05,
      "step": 16860,
      "training_step_time": 7.582464218139648
    },
    {
      "epoch": 1.6079902648925782e-05,
      "model_forward_time": 1.7421848773956299,
      "step": 16861
    },
    {
      "epoch": 1.6079902648925782e-05,
      "step": 16861,
      "training_step_time": 7.622365713119507
    },
    {
      "epoch": 1.6080856323242186e-05,
      "model_forward_time": 1.7299365997314453,
      "step": 16862
    },
    {
      "epoch": 1.6080856323242186e-05,
      "step": 16862,
      "training_step_time": 8.216320753097534
    },
    {
      "epoch": 1.6081809997558593e-05,
      "model_forward_time": 1.7429962158203125,
      "step": 16863
    },
    {
      "epoch": 1.6081809997558593e-05,
      "step": 16863,
      "training_step_time": 8.095578670501709
    },
    {
      "epoch": 1.6082763671875e-05,
      "model_forward_time": 1.7458288669586182,
      "step": 16864
    },
    {
      "epoch": 1.6082763671875e-05,
      "step": 16864,
      "training_step_time": 8.45101261138916
    },
    {
      "epoch": 1.6083717346191405e-05,
      "model_forward_time": 2.926823139190674,
      "step": 16865
    },
    {
      "epoch": 1.6083717346191405e-05,
      "step": 16865,
      "training_step_time": 7.632220983505249
    },
    {
      "epoch": 1.6084671020507812e-05,
      "model_forward_time": 1.734581470489502,
      "step": 16866
    },
    {
      "epoch": 1.6084671020507812e-05,
      "step": 16866,
      "training_step_time": 9.21911358833313
    },
    {
      "epoch": 1.608562469482422e-05,
      "model_forward_time": 1.7382066249847412,
      "step": 16867
    },
    {
      "epoch": 1.608562469482422e-05,
      "step": 16867,
      "training_step_time": 8.229256629943848
    },
    {
      "epoch": 1.6086578369140624e-05,
      "model_forward_time": 1.7270221710205078,
      "step": 16868
    },
    {
      "epoch": 1.6086578369140624e-05,
      "step": 16868,
      "training_step_time": 7.915198564529419
    },
    {
      "epoch": 1.608753204345703e-05,
      "model_forward_time": 1.7281506061553955,
      "step": 16869
    },
    {
      "epoch": 1.608753204345703e-05,
      "step": 16869,
      "training_step_time": 7.550189256668091
    },
    {
      "epoch": 1.608848571777344e-05,
      "grad_norm": 0.21417874097824097,
      "learning_rate": 9.61977722492636e-06,
      "loss": 0.0663,
      "step": 16870
    },
    {
      "epoch": 1.608848571777344e-05,
      "model_forward_time": 1.7402641773223877,
      "step": 16870
    },
    {
      "dynamics_loss_avg": 0.07565823197364807,
      "epoch": 1.608848571777344e-05,
      "step": 16870
    },
    {
      "action_loss_avg": 0.002890960918739438,
      "epoch": 1.608848571777344e-05,
      "step": 16870
    },
    {
      "epoch": 1.608848571777344e-05,
      "step": 16870,
      "training_step_time": 8.775372505187988
    },
    {
      "epoch": 1.6089439392089842e-05,
      "model_forward_time": 1.7279255390167236,
      "step": 16871
    },
    {
      "epoch": 1.6089439392089842e-05,
      "step": 16871,
      "training_step_time": 7.608113050460815
    },
    {
      "epoch": 1.609039306640625e-05,
      "model_forward_time": 1.7447803020477295,
      "step": 16872
    },
    {
      "epoch": 1.609039306640625e-05,
      "step": 16872,
      "training_step_time": 9.309074640274048
    },
    {
      "epoch": 1.6091346740722657e-05,
      "model_forward_time": 1.7487967014312744,
      "step": 16873
    },
    {
      "epoch": 1.6091346740722657e-05,
      "step": 16873,
      "training_step_time": 7.80154013633728
    },
    {
      "epoch": 1.609230041503906e-05,
      "model_forward_time": 2.8880162239074707,
      "step": 16874
    },
    {
      "epoch": 1.609230041503906e-05,
      "step": 16874,
      "training_step_time": 9.115187883377075
    },
    {
      "epoch": 1.609325408935547e-05,
      "model_forward_time": 1.7590789794921875,
      "step": 16875
    },
    {
      "epoch": 1.609325408935547e-05,
      "step": 16875,
      "training_step_time": 7.875811576843262
    },
    {
      "epoch": 1.6094207763671876e-05,
      "model_forward_time": 1.706228494644165,
      "step": 16876
    },
    {
      "epoch": 1.6094207763671876e-05,
      "step": 16876,
      "training_step_time": 9.58847951889038
    },
    {
      "epoch": 1.609516143798828e-05,
      "model_forward_time": 1.748725175857544,
      "step": 16877
    },
    {
      "epoch": 1.609516143798828e-05,
      "step": 16877,
      "training_step_time": 7.899382591247559
    },
    {
      "epoch": 1.6096115112304688e-05,
      "model_forward_time": 1.7311503887176514,
      "step": 16878
    },
    {
      "epoch": 1.6096115112304688e-05,
      "step": 16878,
      "training_step_time": 7.692288160324097
    },
    {
      "epoch": 1.6097068786621095e-05,
      "model_forward_time": 1.7428903579711914,
      "step": 16879
    },
    {
      "epoch": 1.6097068786621095e-05,
      "step": 16879,
      "training_step_time": 7.74052357673645
    },
    {
      "epoch": 1.60980224609375e-05,
      "grad_norm": 0.18459416925907135,
      "learning_rate": 9.619144519941486e-06,
      "loss": 0.0647,
      "step": 16880
    },
    {
      "epoch": 1.60980224609375e-05,
      "model_forward_time": 1.7249891757965088,
      "step": 16880
    },
    {
      "dynamics_loss_avg": 0.045128504559397695,
      "epoch": 1.60980224609375e-05,
      "step": 16880
    },
    {
      "action_loss_avg": 0.002220730297267437,
      "epoch": 1.60980224609375e-05,
      "step": 16880
    },
    {
      "epoch": 1.60980224609375e-05,
      "step": 16880,
      "training_step_time": 8.405772686004639
    },
    {
      "epoch": 1.6098976135253906e-05,
      "model_forward_time": 1.7319717407226562,
      "step": 16881
    },
    {
      "epoch": 1.6098976135253906e-05,
      "step": 16881,
      "training_step_time": 7.724525451660156
    },
    {
      "epoch": 1.6099929809570314e-05,
      "model_forward_time": 1.7221894264221191,
      "step": 16882
    },
    {
      "epoch": 1.6099929809570314e-05,
      "step": 16882,
      "training_step_time": 7.824673175811768
    },
    {
      "epoch": 1.6100883483886718e-05,
      "model_forward_time": 2.4984047412872314,
      "step": 16883
    },
    {
      "epoch": 1.6100883483886718e-05,
      "step": 16883,
      "training_step_time": 9.257773160934448
    },
    {
      "epoch": 1.6101837158203125e-05,
      "model_forward_time": 1.8321418762207031,
      "step": 16884
    },
    {
      "epoch": 1.6101837158203125e-05,
      "step": 16884,
      "training_step_time": 7.80066442489624
    },
    {
      "epoch": 1.6102790832519533e-05,
      "model_forward_time": 1.7432444095611572,
      "step": 16885
    },
    {
      "epoch": 1.6102790832519533e-05,
      "step": 16885,
      "training_step_time": 7.86404275894165
    },
    {
      "epoch": 1.6103744506835937e-05,
      "model_forward_time": 1.7403199672698975,
      "step": 16886
    },
    {
      "epoch": 1.6103744506835937e-05,
      "step": 16886,
      "training_step_time": 7.982369899749756
    },
    {
      "epoch": 1.6104698181152344e-05,
      "model_forward_time": 1.7310898303985596,
      "step": 16887
    },
    {
      "epoch": 1.6104698181152344e-05,
      "step": 16887,
      "training_step_time": 7.721806049346924
    },
    {
      "epoch": 1.610565185546875e-05,
      "model_forward_time": 1.7184231281280518,
      "step": 16888
    },
    {
      "epoch": 1.610565185546875e-05,
      "step": 16888,
      "training_step_time": 7.73628044128418
    },
    {
      "epoch": 1.6106605529785156e-05,
      "model_forward_time": 1.7219264507293701,
      "step": 16889
    },
    {
      "epoch": 1.6106605529785156e-05,
      "step": 16889,
      "training_step_time": 7.757541179656982
    },
    {
      "epoch": 1.6107559204101563e-05,
      "grad_norm": 0.20299731194972992,
      "learning_rate": 9.618511309813911e-06,
      "loss": 0.0649,
      "step": 16890
    },
    {
      "epoch": 1.6107559204101563e-05,
      "model_forward_time": 1.7549643516540527,
      "step": 16890
    },
    {
      "dynamics_loss_avg": 0.06430334597826004,
      "epoch": 1.6107559204101563e-05,
      "step": 16890
    },
    {
      "action_loss_avg": 0.002118682151194662,
      "epoch": 1.6107559204101563e-05,
      "step": 16890
    },
    {
      "epoch": 1.6107559204101563e-05,
      "step": 16890,
      "training_step_time": 8.230377674102783
    },
    {
      "epoch": 1.6108512878417967e-05,
      "model_forward_time": 1.7450776100158691,
      "step": 16891
    },
    {
      "epoch": 1.6108512878417967e-05,
      "step": 16891,
      "training_step_time": 9.028522253036499
    },
    {
      "epoch": 1.6109466552734374e-05,
      "model_forward_time": 2.8333969116210938,
      "step": 16892
    },
    {
      "epoch": 1.6109466552734374e-05,
      "step": 16892,
      "training_step_time": 8.43074369430542
    },
    {
      "epoch": 1.6110420227050782e-05,
      "model_forward_time": 1.7833898067474365,
      "step": 16893
    },
    {
      "epoch": 1.6110420227050782e-05,
      "step": 16893,
      "training_step_time": 7.848961114883423
    },
    {
      "epoch": 1.6111373901367186e-05,
      "model_forward_time": 1.7293281555175781,
      "step": 16894
    },
    {
      "epoch": 1.6111373901367186e-05,
      "step": 16894,
      "training_step_time": 8.703361749649048
    },
    {
      "epoch": 1.6112327575683593e-05,
      "model_forward_time": 1.7266340255737305,
      "step": 16895
    },
    {
      "epoch": 1.6112327575683593e-05,
      "step": 16895,
      "training_step_time": 7.978802680969238
    },
    {
      "epoch": 1.611328125e-05,
      "model_forward_time": 1.7401130199432373,
      "step": 16896
    },
    {
      "epoch": 1.611328125e-05,
      "step": 16896,
      "training_step_time": 7.636220932006836
    },
    {
      "epoch": 1.6114234924316405e-05,
      "model_forward_time": 1.7202575206756592,
      "step": 16897
    },
    {
      "epoch": 1.6114234924316405e-05,
      "step": 16897,
      "training_step_time": 7.6963560581207275
    },
    {
      "epoch": 1.6115188598632812e-05,
      "model_forward_time": 1.7384390830993652,
      "step": 16898
    },
    {
      "epoch": 1.6115188598632812e-05,
      "step": 16898,
      "training_step_time": 8.78482460975647
    },
    {
      "epoch": 1.611614227294922e-05,
      "model_forward_time": 1.732814073562622,
      "step": 16899
    },
    {
      "epoch": 1.611614227294922e-05,
      "step": 16899,
      "training_step_time": 7.730254888534546
    },
    {
      "epoch": 1.6117095947265624e-05,
      "grad_norm": 0.13921420276165009,
      "learning_rate": 9.617877594612886e-06,
      "loss": 0.0641,
      "step": 16900
    },
    {
      "epoch": 1.6117095947265624e-05,
      "model_forward_time": 1.7206876277923584,
      "step": 16900
    },
    {
      "dynamics_loss_avg": 0.0720488827675581,
      "epoch": 1.6117095947265624e-05,
      "step": 16900
    },
    {
      "action_loss_avg": 0.004036090435693041,
      "epoch": 1.6117095947265624e-05,
      "step": 16900
    },
    {
      "epoch": 1.6117095947265624e-05,
      "step": 16900,
      "training_step_time": 8.766953468322754
    },
    {
      "epoch": 1.611804962158203e-05,
      "model_forward_time": 2.794567346572876,
      "step": 16901
    },
    {
      "epoch": 1.611804962158203e-05,
      "step": 16901,
      "training_step_time": 7.813899517059326
    },
    {
      "epoch": 1.611900329589844e-05,
      "model_forward_time": 1.74753999710083,
      "step": 16902
    },
    {
      "epoch": 1.611900329589844e-05,
      "step": 16902,
      "training_step_time": 8.326642990112305
    },
    {
      "epoch": 1.6119956970214843e-05,
      "model_forward_time": 1.7223591804504395,
      "step": 16903
    },
    {
      "epoch": 1.6119956970214843e-05,
      "step": 16903,
      "training_step_time": 8.94197392463684
    },
    {
      "epoch": 1.612091064453125e-05,
      "model_forward_time": 1.726754903793335,
      "step": 16904
    },
    {
      "epoch": 1.612091064453125e-05,
      "step": 16904,
      "training_step_time": 7.783631324768066
    },
    {
      "epoch": 1.6121864318847657e-05,
      "model_forward_time": 1.7338054180145264,
      "step": 16905
    },
    {
      "epoch": 1.6121864318847657e-05,
      "step": 16905,
      "training_step_time": 8.04510760307312
    },
    {
      "epoch": 1.612281799316406e-05,
      "model_forward_time": 1.7318451404571533,
      "step": 16906
    },
    {
      "epoch": 1.612281799316406e-05,
      "step": 16906,
      "training_step_time": 8.26455307006836
    },
    {
      "epoch": 1.612377166748047e-05,
      "model_forward_time": 1.7279486656188965,
      "step": 16907
    },
    {
      "epoch": 1.612377166748047e-05,
      "step": 16907,
      "training_step_time": 7.752631664276123
    },
    {
      "epoch": 1.6124725341796876e-05,
      "model_forward_time": 1.7288129329681396,
      "step": 16908
    },
    {
      "epoch": 1.6124725341796876e-05,
      "step": 16908,
      "training_step_time": 7.652763605117798
    },
    {
      "epoch": 1.612567901611328e-05,
      "model_forward_time": 1.7221786975860596,
      "step": 16909
    },
    {
      "epoch": 1.612567901611328e-05,
      "step": 16909,
      "training_step_time": 7.809999465942383
    },
    {
      "epoch": 1.6126632690429688e-05,
      "grad_norm": 0.12227106839418411,
      "learning_rate": 9.617243374407708e-06,
      "loss": 0.0632,
      "step": 16910
    },
    {
      "epoch": 1.6126632690429688e-05,
      "model_forward_time": 1.7305803298950195,
      "step": 16910
    },
    {
      "dynamics_loss_avg": 0.07300942428410054,
      "epoch": 1.6126632690429688e-05,
      "step": 16910
    },
    {
      "action_loss_avg": 0.003814273080206476,
      "epoch": 1.6126632690429688e-05,
      "step": 16910
    },
    {
      "epoch": 1.6126632690429688e-05,
      "step": 16910,
      "training_step_time": 9.326984167098999
    },
    {
      "epoch": 1.6127586364746095e-05,
      "model_forward_time": 2.88551664352417,
      "step": 16911
    },
    {
      "epoch": 1.6127586364746095e-05,
      "step": 16911,
      "training_step_time": 8.637386083602905
    },
    {
      "epoch": 1.61285400390625e-05,
      "model_forward_time": 1.8368487358093262,
      "step": 16912
    },
    {
      "epoch": 1.61285400390625e-05,
      "step": 16912,
      "training_step_time": 8.10959506034851
    },
    {
      "epoch": 1.6129493713378906e-05,
      "model_forward_time": 1.7430849075317383,
      "step": 16913
    },
    {
      "epoch": 1.6129493713378906e-05,
      "step": 16913,
      "training_step_time": 8.314007759094238
    },
    {
      "epoch": 1.6130447387695314e-05,
      "model_forward_time": 1.7409040927886963,
      "step": 16914
    },
    {
      "epoch": 1.6130447387695314e-05,
      "step": 16914,
      "training_step_time": 8.840503931045532
    },
    {
      "epoch": 1.6131401062011718e-05,
      "model_forward_time": 1.731513500213623,
      "step": 16915
    },
    {
      "epoch": 1.6131401062011718e-05,
      "step": 16915,
      "training_step_time": 7.706695795059204
    },
    {
      "epoch": 1.6132354736328125e-05,
      "model_forward_time": 1.7400565147399902,
      "step": 16916
    },
    {
      "epoch": 1.6132354736328125e-05,
      "step": 16916,
      "training_step_time": 8.052722692489624
    },
    {
      "epoch": 1.6133308410644533e-05,
      "model_forward_time": 1.7320177555084229,
      "step": 16917
    },
    {
      "epoch": 1.6133308410644533e-05,
      "step": 16917,
      "training_step_time": 7.572101593017578
    },
    {
      "epoch": 1.6134262084960937e-05,
      "model_forward_time": 1.737022876739502,
      "step": 16918
    },
    {
      "epoch": 1.6134262084960937e-05,
      "step": 16918,
      "training_step_time": 7.6838250160217285
    },
    {
      "epoch": 1.6135215759277344e-05,
      "model_forward_time": 1.7339558601379395,
      "step": 16919
    },
    {
      "epoch": 1.6135215759277344e-05,
      "step": 16919,
      "training_step_time": 9.03569483757019
    },
    {
      "epoch": 1.613616943359375e-05,
      "grad_norm": 0.16855591535568237,
      "learning_rate": 9.616608649267738e-06,
      "loss": 0.063,
      "step": 16920
    },
    {
      "epoch": 1.613616943359375e-05,
      "model_forward_time": 2.845736026763916,
      "step": 16920
    },
    {
      "dynamics_loss_avg": 0.04556858502328396,
      "epoch": 1.613616943359375e-05,
      "step": 16920
    },
    {
      "action_loss_avg": 0.0016419283871073275,
      "epoch": 1.613616943359375e-05,
      "step": 16920
    },
    {
      "epoch": 1.613616943359375e-05,
      "step": 16920,
      "training_step_time": 7.562063694000244
    },
    {
      "epoch": 1.6137123107910156e-05,
      "model_forward_time": 1.8504202365875244,
      "step": 16921
    },
    {
      "epoch": 1.6137123107910156e-05,
      "step": 16921,
      "training_step_time": 8.531369924545288
    },
    {
      "epoch": 1.6138076782226563e-05,
      "model_forward_time": 1.7293765544891357,
      "step": 16922
    },
    {
      "epoch": 1.6138076782226563e-05,
      "step": 16922,
      "training_step_time": 7.711498975753784
    },
    {
      "epoch": 1.6139030456542967e-05,
      "model_forward_time": 1.7274987697601318,
      "step": 16923
    },
    {
      "epoch": 1.6139030456542967e-05,
      "step": 16923,
      "training_step_time": 8.315176010131836
    },
    {
      "epoch": 1.6139984130859375e-05,
      "model_forward_time": 1.7310993671417236,
      "step": 16924
    },
    {
      "epoch": 1.6139984130859375e-05,
      "step": 16924,
      "training_step_time": 8.269980907440186
    },
    {
      "epoch": 1.6140937805175782e-05,
      "model_forward_time": 1.757849931716919,
      "step": 16925
    },
    {
      "epoch": 1.6140937805175782e-05,
      "step": 16925,
      "training_step_time": 7.597666501998901
    },
    {
      "epoch": 1.6141891479492186e-05,
      "model_forward_time": 1.7431647777557373,
      "step": 16926
    },
    {
      "epoch": 1.6141891479492186e-05,
      "step": 16926,
      "training_step_time": 7.609760761260986
    },
    {
      "epoch": 1.6142845153808593e-05,
      "model_forward_time": 1.7467553615570068,
      "step": 16927
    },
    {
      "epoch": 1.6142845153808593e-05,
      "step": 16927,
      "training_step_time": 7.67987060546875
    },
    {
      "epoch": 1.6143798828125e-05,
      "model_forward_time": 1.7769460678100586,
      "step": 16928
    },
    {
      "epoch": 1.6143798828125e-05,
      "step": 16928,
      "training_step_time": 7.639363050460815
    },
    {
      "epoch": 1.6144752502441405e-05,
      "model_forward_time": 2.9079370498657227,
      "step": 16929
    },
    {
      "epoch": 1.6144752502441405e-05,
      "step": 16929,
      "training_step_time": 8.697830200195312
    },
    {
      "epoch": 1.6145706176757812e-05,
      "grad_norm": 0.1686682105064392,
      "learning_rate": 9.615973419262385e-06,
      "loss": 0.0591,
      "step": 16930
    },
    {
      "epoch": 1.6145706176757812e-05,
      "model_forward_time": 1.8557350635528564,
      "step": 16930
    },
    {
      "dynamics_loss_avg": 0.06609607599675656,
      "epoch": 1.6145706176757812e-05,
      "step": 16930
    },
    {
      "action_loss_avg": 0.002850442158523947,
      "epoch": 1.6145706176757812e-05,
      "step": 16930
    },
    {
      "epoch": 1.6145706176757812e-05,
      "step": 16930,
      "training_step_time": 7.623669385910034
    },
    {
      "epoch": 1.614665985107422e-05,
      "model_forward_time": 1.7458109855651855,
      "step": 16931
    },
    {
      "epoch": 1.614665985107422e-05,
      "step": 16931,
      "training_step_time": 7.657046556472778
    },
    {
      "epoch": 1.6147613525390624e-05,
      "model_forward_time": 1.7585029602050781,
      "step": 16932
    },
    {
      "epoch": 1.6147613525390624e-05,
      "step": 16932,
      "training_step_time": 7.714393377304077
    },
    {
      "epoch": 1.614856719970703e-05,
      "model_forward_time": 1.7232820987701416,
      "step": 16933
    },
    {
      "epoch": 1.614856719970703e-05,
      "step": 16933,
      "training_step_time": 7.7322399616241455
    },
    {
      "epoch": 1.614952087402344e-05,
      "model_forward_time": 1.7383639812469482,
      "step": 16934
    },
    {
      "epoch": 1.614952087402344e-05,
      "step": 16934,
      "training_step_time": 7.58856725692749
    },
    {
      "epoch": 1.6150474548339843e-05,
      "model_forward_time": 1.7312147617340088,
      "step": 16935
    },
    {
      "epoch": 1.6150474548339843e-05,
      "step": 16935,
      "training_step_time": 7.630116939544678
    },
    {
      "epoch": 1.615142822265625e-05,
      "model_forward_time": 1.7451326847076416,
      "step": 16936
    },
    {
      "epoch": 1.615142822265625e-05,
      "step": 16936,
      "training_step_time": 7.731080055236816
    },
    {
      "epoch": 1.6152381896972657e-05,
      "model_forward_time": 1.736173152923584,
      "step": 16937
    },
    {
      "epoch": 1.6152381896972657e-05,
      "step": 16937,
      "training_step_time": 8.202716827392578
    },
    {
      "epoch": 1.615333557128906e-05,
      "model_forward_time": 2.861531972885132,
      "step": 16938
    },
    {
      "epoch": 1.615333557128906e-05,
      "step": 16938,
      "training_step_time": 8.213489532470703
    },
    {
      "epoch": 1.615428924560547e-05,
      "model_forward_time": 1.8294785022735596,
      "step": 16939
    },
    {
      "epoch": 1.615428924560547e-05,
      "step": 16939,
      "training_step_time": 10.090488195419312
    },
    {
      "epoch": 1.6155242919921876e-05,
      "grad_norm": 0.2335997223854065,
      "learning_rate": 9.61533768446112e-06,
      "loss": 0.0635,
      "step": 16940
    },
    {
      "epoch": 1.6155242919921876e-05,
      "model_forward_time": 1.7267744541168213,
      "step": 16940
    },
    {
      "dynamics_loss_avg": 0.07088412903249264,
      "epoch": 1.6155242919921876e-05,
      "step": 16940
    },
    {
      "action_loss_avg": 0.003188836958725005,
      "epoch": 1.6155242919921876e-05,
      "step": 16940
    },
    {
      "epoch": 1.6155242919921876e-05,
      "step": 16940,
      "training_step_time": 7.895127534866333
    },
    {
      "epoch": 1.615619659423828e-05,
      "model_forward_time": 1.7389507293701172,
      "step": 16941
    },
    {
      "epoch": 1.615619659423828e-05,
      "step": 16941,
      "training_step_time": 7.7130126953125
    },
    {
      "epoch": 1.6157150268554688e-05,
      "model_forward_time": 1.7178502082824707,
      "step": 16942
    },
    {
      "epoch": 1.6157150268554688e-05,
      "step": 16942,
      "training_step_time": 7.7901060581207275
    },
    {
      "epoch": 1.6158103942871095e-05,
      "model_forward_time": 1.7333667278289795,
      "step": 16943
    },
    {
      "epoch": 1.6158103942871095e-05,
      "step": 16943,
      "training_step_time": 7.56427264213562
    },
    {
      "epoch": 1.61590576171875e-05,
      "model_forward_time": 1.7348871231079102,
      "step": 16944
    },
    {
      "epoch": 1.61590576171875e-05,
      "step": 16944,
      "training_step_time": 8.531725645065308
    },
    {
      "epoch": 1.6160011291503907e-05,
      "model_forward_time": 1.7256903648376465,
      "step": 16945
    },
    {
      "epoch": 1.6160011291503907e-05,
      "step": 16945,
      "training_step_time": 8.156276226043701
    },
    {
      "epoch": 1.6160964965820314e-05,
      "model_forward_time": 1.739281415939331,
      "step": 16946
    },
    {
      "epoch": 1.6160964965820314e-05,
      "step": 16946,
      "training_step_time": 8.538760423660278
    },
    {
      "epoch": 1.6161918640136718e-05,
      "model_forward_time": 2.8019392490386963,
      "step": 16947
    },
    {
      "epoch": 1.6161918640136718e-05,
      "step": 16947,
      "training_step_time": 9.752237796783447
    },
    {
      "epoch": 1.6162872314453125e-05,
      "model_forward_time": 1.804410696029663,
      "step": 16948
    },
    {
      "epoch": 1.6162872314453125e-05,
      "step": 16948,
      "training_step_time": 10.302679777145386
    },
    {
      "epoch": 1.6163825988769533e-05,
      "model_forward_time": 1.728294849395752,
      "step": 16949
    },
    {
      "epoch": 1.6163825988769533e-05,
      "step": 16949,
      "training_step_time": 9.263010740280151
    },
    {
      "epoch": 1.6164779663085937e-05,
      "grad_norm": 0.13563154637813568,
      "learning_rate": 9.614701444933465e-06,
      "loss": 0.0632,
      "step": 16950
    },
    {
      "epoch": 1.6164779663085937e-05,
      "model_forward_time": 1.7183866500854492,
      "step": 16950
    },
    {
      "dynamics_loss_avg": 0.04875631798058748,
      "epoch": 1.6164779663085937e-05,
      "step": 16950
    },
    {
      "action_loss_avg": 0.0022004819300491365,
      "epoch": 1.6164779663085937e-05,
      "step": 16950
    },
    {
      "epoch": 1.6164779663085937e-05,
      "step": 16950,
      "training_step_time": 9.707855701446533
    },
    {
      "epoch": 1.6165733337402344e-05,
      "model_forward_time": 1.7385380268096924,
      "step": 16951
    },
    {
      "epoch": 1.6165733337402344e-05,
      "step": 16951,
      "training_step_time": 8.919542789459229
    },
    {
      "epoch": 1.616668701171875e-05,
      "model_forward_time": 1.751394510269165,
      "step": 16952
    },
    {
      "epoch": 1.616668701171875e-05,
      "step": 16952,
      "training_step_time": 7.69976282119751
    },
    {
      "epoch": 1.6167640686035156e-05,
      "model_forward_time": 1.7479028701782227,
      "step": 16953
    },
    {
      "epoch": 1.6167640686035156e-05,
      "step": 16953,
      "training_step_time": 8.62712287902832
    },
    {
      "epoch": 1.6168594360351563e-05,
      "model_forward_time": 1.7222435474395752,
      "step": 16954
    },
    {
      "epoch": 1.6168594360351563e-05,
      "step": 16954,
      "training_step_time": 7.676549673080444
    },
    {
      "epoch": 1.6169548034667967e-05,
      "model_forward_time": 1.7222933769226074,
      "step": 16955
    },
    {
      "epoch": 1.6169548034667967e-05,
      "step": 16955,
      "training_step_time": 7.84484601020813
    },
    {
      "epoch": 1.6170501708984375e-05,
      "model_forward_time": 1.7420876026153564,
      "step": 16956
    },
    {
      "epoch": 1.6170501708984375e-05,
      "step": 16956,
      "training_step_time": 8.872000217437744
    },
    {
      "epoch": 1.6171455383300782e-05,
      "model_forward_time": 2.8918232917785645,
      "step": 16957
    },
    {
      "epoch": 1.6171455383300782e-05,
      "step": 16957,
      "training_step_time": 8.955917119979858
    },
    {
      "epoch": 1.6172409057617186e-05,
      "model_forward_time": 1.7574729919433594,
      "step": 16958
    },
    {
      "epoch": 1.6172409057617186e-05,
      "step": 16958,
      "training_step_time": 7.713473081588745
    },
    {
      "epoch": 1.6173362731933593e-05,
      "model_forward_time": 1.7351772785186768,
      "step": 16959
    },
    {
      "epoch": 1.6173362731933593e-05,
      "step": 16959,
      "training_step_time": 7.51846718788147
    },
    {
      "epoch": 1.617431640625e-05,
      "grad_norm": 0.18725892901420593,
      "learning_rate": 9.614064700748997e-06,
      "loss": 0.0635,
      "step": 16960
    },
    {
      "epoch": 1.617431640625e-05,
      "model_forward_time": 1.723454475402832,
      "step": 16960
    },
    {
      "dynamics_loss_avg": 0.06488427557051182,
      "epoch": 1.617431640625e-05,
      "step": 16960
    },
    {
      "action_loss_avg": 0.002371424878947437,
      "epoch": 1.617431640625e-05,
      "step": 16960
    },
    {
      "epoch": 1.617431640625e-05,
      "step": 16960,
      "training_step_time": 7.731808423995972
    },
    {
      "epoch": 1.6175270080566405e-05,
      "model_forward_time": 1.7335104942321777,
      "step": 16961
    },
    {
      "epoch": 1.6175270080566405e-05,
      "step": 16961,
      "training_step_time": 7.754182815551758
    },
    {
      "epoch": 1.6176223754882812e-05,
      "model_forward_time": 1.7503697872161865,
      "step": 16962
    },
    {
      "epoch": 1.6176223754882812e-05,
      "step": 16962,
      "training_step_time": 8.163893699645996
    },
    {
      "epoch": 1.617717742919922e-05,
      "model_forward_time": 1.7364468574523926,
      "step": 16963
    },
    {
      "epoch": 1.617717742919922e-05,
      "step": 16963,
      "training_step_time": 7.590806484222412
    },
    {
      "epoch": 1.6178131103515624e-05,
      "model_forward_time": 1.7640938758850098,
      "step": 16964
    },
    {
      "epoch": 1.6178131103515624e-05,
      "step": 16964,
      "training_step_time": 9.217952489852905
    },
    {
      "epoch": 1.617908477783203e-05,
      "model_forward_time": 1.756758689880371,
      "step": 16965
    },
    {
      "epoch": 1.617908477783203e-05,
      "step": 16965,
      "training_step_time": 7.671624660491943
    },
    {
      "epoch": 1.618003845214844e-05,
      "model_forward_time": 2.9607691764831543,
      "step": 16966
    },
    {
      "epoch": 1.618003845214844e-05,
      "step": 16966,
      "training_step_time": 7.65723180770874
    },
    {
      "epoch": 1.6180992126464843e-05,
      "model_forward_time": 1.8336403369903564,
      "step": 16967
    },
    {
      "epoch": 1.6180992126464843e-05,
      "step": 16967,
      "training_step_time": 8.801166534423828
    },
    {
      "epoch": 1.618194580078125e-05,
      "model_forward_time": 1.7436916828155518,
      "step": 16968
    },
    {
      "epoch": 1.618194580078125e-05,
      "step": 16968,
      "training_step_time": 8.70635461807251
    },
    {
      "epoch": 1.6182899475097657e-05,
      "model_forward_time": 1.7372033596038818,
      "step": 16969
    },
    {
      "epoch": 1.6182899475097657e-05,
      "step": 16969,
      "training_step_time": 7.649282455444336
    },
    {
      "epoch": 1.618385314941406e-05,
      "grad_norm": 0.19616271555423737,
      "learning_rate": 9.613427451977352e-06,
      "loss": 0.0628,
      "step": 16970
    },
    {
      "epoch": 1.618385314941406e-05,
      "model_forward_time": 1.72843599319458,
      "step": 16970
    },
    {
      "dynamics_loss_avg": 0.06838252916932105,
      "epoch": 1.618385314941406e-05,
      "step": 16970
    },
    {
      "action_loss_avg": 0.0035770203801803293,
      "epoch": 1.618385314941406e-05,
      "step": 16970
    },
    {
      "epoch": 1.618385314941406e-05,
      "step": 16970,
      "training_step_time": 8.894141435623169
    },
    {
      "epoch": 1.618480682373047e-05,
      "model_forward_time": 1.7460556030273438,
      "step": 16971
    },
    {
      "epoch": 1.618480682373047e-05,
      "step": 16971,
      "training_step_time": 7.707804441452026
    },
    {
      "epoch": 1.6185760498046876e-05,
      "model_forward_time": 1.7433264255523682,
      "step": 16972
    },
    {
      "epoch": 1.6185760498046876e-05,
      "step": 16972,
      "training_step_time": 7.735536813735962
    },
    {
      "epoch": 1.618671417236328e-05,
      "model_forward_time": 1.733767032623291,
      "step": 16973
    },
    {
      "epoch": 1.618671417236328e-05,
      "step": 16973,
      "training_step_time": 9.61144733428955
    },
    {
      "epoch": 1.6187667846679688e-05,
      "model_forward_time": 1.748401403427124,
      "step": 16974
    },
    {
      "epoch": 1.6187667846679688e-05,
      "step": 16974,
      "training_step_time": 7.598086357116699
    },
    {
      "epoch": 1.6188621520996095e-05,
      "model_forward_time": 2.875194787979126,
      "step": 16975
    },
    {
      "epoch": 1.6188621520996095e-05,
      "step": 16975,
      "training_step_time": 7.708945274353027
    },
    {
      "epoch": 1.61895751953125e-05,
      "model_forward_time": 1.8478975296020508,
      "step": 16976
    },
    {
      "epoch": 1.61895751953125e-05,
      "step": 16976,
      "training_step_time": 7.648228406906128
    },
    {
      "epoch": 1.6190528869628907e-05,
      "model_forward_time": 1.7389681339263916,
      "step": 16977
    },
    {
      "epoch": 1.6190528869628907e-05,
      "step": 16977,
      "training_step_time": 7.622541904449463
    },
    {
      "epoch": 1.6191482543945314e-05,
      "model_forward_time": 1.7344560623168945,
      "step": 16978
    },
    {
      "epoch": 1.6191482543945314e-05,
      "step": 16978,
      "training_step_time": 7.782713174819946
    },
    {
      "epoch": 1.6192436218261718e-05,
      "model_forward_time": 2.301839590072632,
      "step": 16979
    },
    {
      "epoch": 1.6192436218261718e-05,
      "step": 16979,
      "training_step_time": 8.841373682022095
    },
    {
      "epoch": 1.6193389892578125e-05,
      "grad_norm": 0.16462500393390656,
      "learning_rate": 9.612789698688217e-06,
      "loss": 0.0674,
      "step": 16980
    },
    {
      "epoch": 1.6193389892578125e-05,
      "model_forward_time": 1.697941780090332,
      "step": 16980
    },
    {
      "dynamics_loss_avg": 0.07110226899385452,
      "epoch": 1.6193389892578125e-05,
      "step": 16980
    },
    {
      "action_loss_avg": 0.00337861679145135,
      "epoch": 1.6193389892578125e-05,
      "step": 16980
    },
    {
      "epoch": 1.6193389892578125e-05,
      "step": 16980,
      "training_step_time": 6.830595254898071
    },
    {
      "epoch": 1.6194343566894533e-05,
      "model_forward_time": 1.6997838020324707,
      "step": 16981
    },
    {
      "epoch": 1.6194343566894533e-05,
      "step": 16981,
      "training_step_time": 6.817167043685913
    },
    {
      "epoch": 1.6195297241210937e-05,
      "model_forward_time": 1.7079236507415771,
      "step": 16982
    },
    {
      "epoch": 1.6195297241210937e-05,
      "step": 16982,
      "training_step_time": 6.82732892036438
    },
    {
      "epoch": 1.6196250915527344e-05,
      "model_forward_time": 1.693343162536621,
      "step": 16983
    },
    {
      "epoch": 1.6196250915527344e-05,
      "step": 16983,
      "training_step_time": 6.771617889404297
    },
    {
      "epoch": 1.619720458984375e-05,
      "model_forward_time": 1.692018747329712,
      "step": 16984
    },
    {
      "epoch": 1.619720458984375e-05,
      "step": 16984,
      "training_step_time": 6.780934572219849
    },
    {
      "epoch": 1.6198158264160156e-05,
      "model_forward_time": 1.7033641338348389,
      "step": 16985
    },
    {
      "epoch": 1.6198158264160156e-05,
      "step": 16985,
      "training_step_time": 6.810847520828247
    },
    {
      "epoch": 1.6199111938476563e-05,
      "model_forward_time": 1.7081403732299805,
      "step": 16986
    },
    {
      "epoch": 1.6199111938476563e-05,
      "step": 16986,
      "training_step_time": 6.789888858795166
    },
    {
      "epoch": 1.6200065612792967e-05,
      "model_forward_time": 1.3174645900726318,
      "step": 16987
    },
    {
      "epoch": 1.6200065612792967e-05,
      "step": 16987,
      "training_step_time": 6.753685235977173
    },
    {
      "epoch": 1.6201019287109375e-05,
      "model_forward_time": 2.264047622680664,
      "step": 16988
    },
    {
      "epoch": 1.6201019287109375e-05,
      "step": 16988,
      "training_step_time": 10.790230512619019
    },
    {
      "epoch": 1.6201972961425782e-05,
      "model_forward_time": 1.7034852504730225,
      "step": 16989
    },
    {
      "epoch": 1.6201972961425782e-05,
      "step": 16989,
      "training_step_time": 7.633723735809326
    },
    {
      "epoch": 1.6202926635742186e-05,
      "grad_norm": 0.13979648053646088,
      "learning_rate": 9.612151440951334e-06,
      "loss": 0.0641,
      "step": 16990
    },
    {
      "epoch": 1.6202926635742186e-05,
      "model_forward_time": 1.697181224822998,
      "step": 16990
    },
    {
      "dynamics_loss_avg": 0.051331226155161856,
      "epoch": 1.6202926635742186e-05,
      "step": 16990
    },
    {
      "action_loss_avg": 0.0028582854953128844,
      "epoch": 1.6202926635742186e-05,
      "step": 16990
    },
    {
      "epoch": 1.6202926635742186e-05,
      "step": 16990,
      "training_step_time": 6.682183504104614
    },
    {
      "epoch": 1.6203880310058593e-05,
      "model_forward_time": 1.3259248733520508,
      "step": 16991
    },
    {
      "epoch": 1.6203880310058593e-05,
      "step": 16991,
      "training_step_time": 6.688076019287109
    },
    {
      "epoch": 1.6204833984375e-05,
      "model_forward_time": 1.70363187789917,
      "step": 16992
    },
    {
      "epoch": 1.6204833984375e-05,
      "step": 16992,
      "training_step_time": 6.716768264770508
    },
    {
      "epoch": 1.6205787658691405e-05,
      "model_forward_time": 1.7053439617156982,
      "step": 16993
    },
    {
      "epoch": 1.6205787658691405e-05,
      "step": 16993,
      "training_step_time": 6.689249277114868
    },
    {
      "epoch": 1.6206741333007812e-05,
      "model_forward_time": 1.7024552822113037,
      "step": 16994
    },
    {
      "epoch": 1.6206741333007812e-05,
      "step": 16994,
      "training_step_time": 6.748635530471802
    },
    {
      "epoch": 1.620769500732422e-05,
      "model_forward_time": 1.7077226638793945,
      "step": 16995
    },
    {
      "epoch": 1.620769500732422e-05,
      "step": 16995,
      "training_step_time": 6.720142841339111
    },
    {
      "epoch": 1.6208648681640624e-05,
      "model_forward_time": 1.7015938758850098,
      "step": 16996
    },
    {
      "epoch": 1.6208648681640624e-05,
      "step": 16996,
      "training_step_time": 6.742029190063477
    },
    {
      "epoch": 1.620960235595703e-05,
      "model_forward_time": 1.7050530910491943,
      "step": 16997
    },
    {
      "epoch": 1.620960235595703e-05,
      "step": 16997,
      "training_step_time": 7.2782697677612305
    },
    {
      "epoch": 1.621055603027344e-05,
      "model_forward_time": 2.3150315284729004,
      "step": 16998
    },
    {
      "epoch": 1.621055603027344e-05,
      "step": 16998,
      "training_step_time": 8.307950973510742
    },
    {
      "epoch": 1.6211509704589843e-05,
      "model_forward_time": 1.7347090244293213,
      "step": 16999
    },
    {
      "epoch": 1.6211509704589843e-05,
      "step": 16999,
      "training_step_time": 6.995372772216797
    },
    {
      "epoch": 1.621246337890625e-05,
      "grad_norm": 0.18036200106143951,
      "learning_rate": 9.611512678836506e-06,
      "loss": 0.0607,
      "step": 17000
    },
    {
      "epoch": 1.621246337890625e-05,
      "model_forward_time": 1.749340295791626,
      "step": 17000
    },
    {
      "dynamics_loss_avg": 0.056854887306690215,
      "epoch": 1.621246337890625e-05,
      "step": 17000
    },
    {
      "action_loss_avg": 0.002628292527515441,
      "epoch": 1.621246337890625e-05,
      "step": 17000
    },
    {
      "epoch": 1.621246337890625e-05,
      "step": 17000,
      "training_step_time": 6.744103908538818
    },
    {
      "epoch": 1.6213417053222657e-05,
      "model_forward_time": 1.7541875839233398,
      "step": 17001
    },
    {
      "epoch": 1.6213417053222657e-05,
      "step": 17001,
      "training_step_time": 7.011700391769409
    },
    {
      "epoch": 1.621437072753906e-05,
      "model_forward_time": 1.7314038276672363,
      "step": 17002
    },
    {
      "epoch": 1.621437072753906e-05,
      "step": 17002,
      "training_step_time": 6.80792498588562
    },
    {
      "epoch": 1.621532440185547e-05,
      "model_forward_time": 1.744260311126709,
      "step": 17003
    },
    {
      "epoch": 1.621532440185547e-05,
      "step": 17003,
      "training_step_time": 6.925345420837402
    },
    {
      "epoch": 1.6216278076171876e-05,
      "model_forward_time": 1.7532033920288086,
      "step": 17004
    },
    {
      "epoch": 1.6216278076171876e-05,
      "step": 17004,
      "training_step_time": 6.9768266677856445
    },
    {
      "epoch": 1.621723175048828e-05,
      "model_forward_time": 1.7451601028442383,
      "step": 17005
    },
    {
      "epoch": 1.621723175048828e-05,
      "step": 17005,
      "training_step_time": 6.9629082679748535
    },
    {
      "epoch": 1.6218185424804688e-05,
      "model_forward_time": 3.313964605331421,
      "step": 17006
    },
    {
      "epoch": 1.6218185424804688e-05,
      "step": 17006,
      "training_step_time": 8.821017265319824
    },
    {
      "epoch": 1.6219139099121095e-05,
      "model_forward_time": 1.847339153289795,
      "step": 17007
    },
    {
      "epoch": 1.6219139099121095e-05,
      "step": 17007,
      "training_step_time": 8.000820398330688
    },
    {
      "epoch": 1.62200927734375e-05,
      "model_forward_time": 1.765270709991455,
      "step": 17008
    },
    {
      "epoch": 1.62200927734375e-05,
      "step": 17008,
      "training_step_time": 7.5931267738342285
    },
    {
      "epoch": 1.6221046447753907e-05,
      "model_forward_time": 1.7412452697753906,
      "step": 17009
    },
    {
      "epoch": 1.6221046447753907e-05,
      "step": 17009,
      "training_step_time": 6.928335428237915
    },
    {
      "epoch": 1.6222000122070314e-05,
      "grad_norm": 0.24432168900966644,
      "learning_rate": 9.610873412413584e-06,
      "loss": 0.0652,
      "step": 17010
    },
    {
      "epoch": 1.6222000122070314e-05,
      "model_forward_time": 1.7526345252990723,
      "step": 17010
    },
    {
      "dynamics_loss_avg": 0.08957943581044674,
      "epoch": 1.6222000122070314e-05,
      "step": 17010
    },
    {
      "action_loss_avg": 0.003522429068107158,
      "epoch": 1.6222000122070314e-05,
      "step": 17010
    },
    {
      "epoch": 1.6222000122070314e-05,
      "step": 17010,
      "training_step_time": 6.947211265563965
    },
    {
      "epoch": 1.6222953796386718e-05,
      "model_forward_time": 1.7464873790740967,
      "step": 17011
    },
    {
      "epoch": 1.6222953796386718e-05,
      "step": 17011,
      "training_step_time": 6.890023708343506
    },
    {
      "epoch": 1.6223907470703125e-05,
      "model_forward_time": 1.7435226440429688,
      "step": 17012
    },
    {
      "epoch": 1.6223907470703125e-05,
      "step": 17012,
      "training_step_time": 6.933176279067993
    },
    {
      "epoch": 1.6224861145019533e-05,
      "model_forward_time": 1.7611231803894043,
      "step": 17013
    },
    {
      "epoch": 1.6224861145019533e-05,
      "step": 17013,
      "training_step_time": 6.9736549854278564
    },
    {
      "epoch": 1.6225814819335937e-05,
      "model_forward_time": 1.74717378616333,
      "step": 17014
    },
    {
      "epoch": 1.6225814819335937e-05,
      "step": 17014,
      "training_step_time": 6.979714632034302
    },
    {
      "epoch": 1.6226768493652344e-05,
      "model_forward_time": 1.3784799575805664,
      "step": 17015
    },
    {
      "epoch": 1.6226768493652344e-05,
      "step": 17015,
      "training_step_time": 6.946202278137207
    },
    {
      "epoch": 1.622772216796875e-05,
      "model_forward_time": 2.8667960166931152,
      "step": 17016
    },
    {
      "epoch": 1.622772216796875e-05,
      "step": 17016,
      "training_step_time": 9.071288824081421
    },
    {
      "epoch": 1.6228675842285156e-05,
      "model_forward_time": 1.729595422744751,
      "step": 17017
    },
    {
      "epoch": 1.6228675842285156e-05,
      "step": 17017,
      "training_step_time": 7.9300537109375
    },
    {
      "epoch": 1.6229629516601563e-05,
      "model_forward_time": 1.755216360092163,
      "step": 17018
    },
    {
      "epoch": 1.6229629516601563e-05,
      "step": 17018,
      "training_step_time": 6.951574802398682
    },
    {
      "epoch": 1.6230583190917967e-05,
      "model_forward_time": 1.733415126800537,
      "step": 17019
    },
    {
      "epoch": 1.6230583190917967e-05,
      "step": 17019,
      "training_step_time": 6.988290786743164
    },
    {
      "epoch": 1.6231536865234375e-05,
      "grad_norm": 0.23983646929264069,
      "learning_rate": 9.610233641752476e-06,
      "loss": 0.0634,
      "step": 17020
    },
    {
      "epoch": 1.6231536865234375e-05,
      "model_forward_time": 1.7337915897369385,
      "step": 17020
    },
    {
      "dynamics_loss_avg": 0.05701383426785469,
      "epoch": 1.6231536865234375e-05,
      "step": 17020
    },
    {
      "action_loss_avg": 0.0034032158204354347,
      "epoch": 1.6231536865234375e-05,
      "step": 17020
    },
    {
      "epoch": 1.6231536865234375e-05,
      "step": 17020,
      "training_step_time": 6.990903615951538
    },
    {
      "epoch": 1.6232490539550782e-05,
      "model_forward_time": 1.735539197921753,
      "step": 17021
    },
    {
      "epoch": 1.6232490539550782e-05,
      "step": 17021,
      "training_step_time": 6.925655364990234
    },
    {
      "epoch": 1.6233444213867186e-05,
      "model_forward_time": 1.7440972328186035,
      "step": 17022
    },
    {
      "epoch": 1.6233444213867186e-05,
      "step": 17022,
      "training_step_time": 6.9690163135528564
    },
    {
      "epoch": 1.6234397888183593e-05,
      "model_forward_time": 1.7379264831542969,
      "step": 17023
    },
    {
      "epoch": 1.6234397888183593e-05,
      "step": 17023,
      "training_step_time": 6.981982707977295
    },
    {
      "epoch": 1.62353515625e-05,
      "model_forward_time": 1.733705759048462,
      "step": 17024
    },
    {
      "epoch": 1.62353515625e-05,
      "step": 17024,
      "training_step_time": 6.924639463424683
    },
    {
      "epoch": 1.6236305236816405e-05,
      "model_forward_time": 2.8729052543640137,
      "step": 17025
    },
    {
      "epoch": 1.6236305236816405e-05,
      "step": 17025,
      "training_step_time": 9.419348001480103
    },
    {
      "epoch": 1.6237258911132812e-05,
      "model_forward_time": 1.7386393547058105,
      "step": 17026
    },
    {
      "epoch": 1.6237258911132812e-05,
      "step": 17026,
      "training_step_time": 7.884661436080933
    },
    {
      "epoch": 1.623821258544922e-05,
      "model_forward_time": 1.741645336151123,
      "step": 17027
    },
    {
      "epoch": 1.623821258544922e-05,
      "step": 17027,
      "training_step_time": 6.963525056838989
    },
    {
      "epoch": 1.6239166259765624e-05,
      "model_forward_time": 1.7549142837524414,
      "step": 17028
    },
    {
      "epoch": 1.6239166259765624e-05,
      "step": 17028,
      "training_step_time": 6.920888662338257
    },
    {
      "epoch": 1.624011993408203e-05,
      "model_forward_time": 1.7351186275482178,
      "step": 17029
    },
    {
      "epoch": 1.624011993408203e-05,
      "step": 17029,
      "training_step_time": 6.896754026412964
    },
    {
      "epoch": 1.624107360839844e-05,
      "grad_norm": 0.20571188628673553,
      "learning_rate": 9.609593366923151e-06,
      "loss": 0.0622,
      "step": 17030
    },
    {
      "epoch": 1.624107360839844e-05,
      "model_forward_time": 1.715944766998291,
      "step": 17030
    },
    {
      "dynamics_loss_avg": 0.08523619249463081,
      "epoch": 1.624107360839844e-05,
      "step": 17030
    },
    {
      "action_loss_avg": 0.006230976199731231,
      "epoch": 1.624107360839844e-05,
      "step": 17030
    },
    {
      "epoch": 1.624107360839844e-05,
      "step": 17030,
      "training_step_time": 6.984347343444824
    },
    {
      "epoch": 1.6242027282714843e-05,
      "model_forward_time": 1.7480967044830322,
      "step": 17031
    },
    {
      "epoch": 1.6242027282714843e-05,
      "step": 17031,
      "training_step_time": 6.949580192565918
    },
    {
      "epoch": 1.624298095703125e-05,
      "model_forward_time": 1.762592077255249,
      "step": 17032
    },
    {
      "epoch": 1.624298095703125e-05,
      "step": 17032,
      "training_step_time": 6.913994789123535
    },
    {
      "epoch": 1.6243934631347657e-05,
      "model_forward_time": 1.7585418224334717,
      "step": 17033
    },
    {
      "epoch": 1.6243934631347657e-05,
      "step": 17033,
      "training_step_time": 6.984723329544067
    },
    {
      "epoch": 1.624488830566406e-05,
      "model_forward_time": 2.8579273223876953,
      "step": 17034
    },
    {
      "epoch": 1.624488830566406e-05,
      "step": 17034,
      "training_step_time": 9.925605535507202
    },
    {
      "epoch": 1.624584197998047e-05,
      "model_forward_time": 1.7700295448303223,
      "step": 17035
    },
    {
      "epoch": 1.624584197998047e-05,
      "step": 17035,
      "training_step_time": 7.927306652069092
    },
    {
      "epoch": 1.6246795654296876e-05,
      "model_forward_time": 1.754563331604004,
      "step": 17036
    },
    {
      "epoch": 1.6246795654296876e-05,
      "step": 17036,
      "training_step_time": 6.948703050613403
    },
    {
      "epoch": 1.624774932861328e-05,
      "model_forward_time": 1.7427096366882324,
      "step": 17037
    },
    {
      "epoch": 1.624774932861328e-05,
      "step": 17037,
      "training_step_time": 6.946385860443115
    },
    {
      "epoch": 1.6248703002929688e-05,
      "model_forward_time": 1.740447998046875,
      "step": 17038
    },
    {
      "epoch": 1.6248703002929688e-05,
      "step": 17038,
      "training_step_time": 6.970535039901733
    },
    {
      "epoch": 1.6249656677246095e-05,
      "model_forward_time": 1.748152732849121,
      "step": 17039
    },
    {
      "epoch": 1.6249656677246095e-05,
      "step": 17039,
      "training_step_time": 6.9372570514678955
    },
    {
      "epoch": 1.62506103515625e-05,
      "grad_norm": 0.21804562211036682,
      "learning_rate": 9.608952587995624e-06,
      "loss": 0.0639,
      "step": 17040
    },
    {
      "epoch": 1.62506103515625e-05,
      "model_forward_time": 1.7246356010437012,
      "step": 17040
    },
    {
      "dynamics_loss_avg": 0.04662181995809078,
      "epoch": 1.62506103515625e-05,
      "step": 17040
    },
    {
      "action_loss_avg": 0.0028018130105920136,
      "epoch": 1.62506103515625e-05,
      "step": 17040
    },
    {
      "epoch": 1.62506103515625e-05,
      "step": 17040,
      "training_step_time": 6.930733680725098
    },
    {
      "epoch": 1.6251564025878907e-05,
      "model_forward_time": 1.751702070236206,
      "step": 17041
    },
    {
      "epoch": 1.6251564025878907e-05,
      "step": 17041,
      "training_step_time": 6.937747478485107
    },
    {
      "epoch": 1.6252517700195314e-05,
      "model_forward_time": 1.7351562976837158,
      "step": 17042
    },
    {
      "epoch": 1.6252517700195314e-05,
      "step": 17042,
      "training_step_time": 6.981617450714111
    },
    {
      "epoch": 1.6253471374511718e-05,
      "model_forward_time": 2.857621669769287,
      "step": 17043
    },
    {
      "epoch": 1.6253471374511718e-05,
      "step": 17043,
      "training_step_time": 12.539752006530762
    },
    {
      "epoch": 1.6254425048828125e-05,
      "model_forward_time": 1.8339946269989014,
      "step": 17044
    },
    {
      "epoch": 1.6254425048828125e-05,
      "step": 17044,
      "training_step_time": 10.369949579238892
    },
    {
      "epoch": 1.6255378723144533e-05,
      "model_forward_time": 1.7141616344451904,
      "step": 17045
    },
    {
      "epoch": 1.6255378723144533e-05,
      "step": 17045,
      "training_step_time": 7.832180500030518
    },
    {
      "epoch": 1.6256332397460937e-05,
      "model_forward_time": 1.766690731048584,
      "step": 17046
    },
    {
      "epoch": 1.6256332397460937e-05,
      "step": 17046,
      "training_step_time": 6.923819065093994
    },
    {
      "epoch": 1.6257286071777344e-05,
      "model_forward_time": 1.7669775485992432,
      "step": 17047
    },
    {
      "epoch": 1.6257286071777344e-05,
      "step": 17047,
      "training_step_time": 6.752304553985596
    },
    {
      "epoch": 1.625823974609375e-05,
      "model_forward_time": 1.7546806335449219,
      "step": 17048
    },
    {
      "epoch": 1.625823974609375e-05,
      "step": 17048,
      "training_step_time": 6.995806455612183
    },
    {
      "epoch": 1.6259193420410156e-05,
      "model_forward_time": 1.7509047985076904,
      "step": 17049
    },
    {
      "epoch": 1.6259193420410156e-05,
      "step": 17049,
      "training_step_time": 6.943886041641235
    },
    {
      "epoch": 1.6260147094726563e-05,
      "grad_norm": 0.18142759799957275,
      "learning_rate": 9.608311305039972e-06,
      "loss": 0.0656,
      "step": 17050
    },
    {
      "epoch": 1.6260147094726563e-05,
      "model_forward_time": 1.7459056377410889,
      "step": 17050
    },
    {
      "dynamics_loss_avg": 0.06971897054463624,
      "epoch": 1.6260147094726563e-05,
      "step": 17050
    },
    {
      "action_loss_avg": 0.0031361696776002644,
      "epoch": 1.6260147094726563e-05,
      "step": 17050
    },
    {
      "epoch": 1.6260147094726563e-05,
      "step": 17050,
      "training_step_time": 6.9227516651153564
    },
    {
      "epoch": 1.6261100769042967e-05,
      "model_forward_time": 1.75059175491333,
      "step": 17051
    },
    {
      "epoch": 1.6261100769042967e-05,
      "step": 17051,
      "training_step_time": 6.934858083724976
    },
    {
      "epoch": 1.6262054443359375e-05,
      "model_forward_time": 1.7526772022247314,
      "step": 17052
    },
    {
      "epoch": 1.6262054443359375e-05,
      "step": 17052,
      "training_step_time": 11.47023344039917
    },
    {
      "epoch": 1.6263008117675782e-05,
      "model_forward_time": 1.8358080387115479,
      "step": 17053
    },
    {
      "epoch": 1.6263008117675782e-05,
      "step": 17053,
      "training_step_time": 8.053437232971191
    },
    {
      "epoch": 1.6263961791992186e-05,
      "model_forward_time": 1.7508275508880615,
      "step": 17054
    },
    {
      "epoch": 1.6263961791992186e-05,
      "step": 17054,
      "training_step_time": 7.80612587928772
    },
    {
      "epoch": 1.6264915466308593e-05,
      "model_forward_time": 1.757765531539917,
      "step": 17055
    },
    {
      "epoch": 1.6264915466308593e-05,
      "step": 17055,
      "training_step_time": 6.891575813293457
    },
    {
      "epoch": 1.6265869140625e-05,
      "model_forward_time": 1.7756121158599854,
      "step": 17056
    },
    {
      "epoch": 1.6265869140625e-05,
      "step": 17056,
      "training_step_time": 6.923173427581787
    },
    {
      "epoch": 1.6266822814941405e-05,
      "model_forward_time": 1.7486650943756104,
      "step": 17057
    },
    {
      "epoch": 1.6266822814941405e-05,
      "step": 17057,
      "training_step_time": 6.933570623397827
    },
    {
      "epoch": 1.6267776489257812e-05,
      "model_forward_time": 1.74918532371521,
      "step": 17058
    },
    {
      "epoch": 1.6267776489257812e-05,
      "step": 17058,
      "training_step_time": 6.931317567825317
    },
    {
      "epoch": 1.626873016357422e-05,
      "model_forward_time": 1.7394835948944092,
      "step": 17059
    },
    {
      "epoch": 1.626873016357422e-05,
      "step": 17059,
      "training_step_time": 6.875430583953857
    },
    {
      "epoch": 1.6269683837890624e-05,
      "grad_norm": 0.1283750683069229,
      "learning_rate": 9.607669518126327e-06,
      "loss": 0.0656,
      "step": 17060
    },
    {
      "epoch": 1.6269683837890624e-05,
      "model_forward_time": 1.720367193222046,
      "step": 17060
    },
    {
      "dynamics_loss_avg": 0.06709273178130389,
      "epoch": 1.6269683837890624e-05,
      "step": 17060
    },
    {
      "action_loss_avg": 0.0036488506942987443,
      "epoch": 1.6269683837890624e-05,
      "step": 17060
    },
    {
      "epoch": 1.6269683837890624e-05,
      "step": 17060,
      "training_step_time": 6.760107517242432
    },
    {
      "epoch": 1.627063751220703e-05,
      "model_forward_time": 1.7476789951324463,
      "step": 17061
    },
    {
      "epoch": 1.627063751220703e-05,
      "step": 17061,
      "training_step_time": 9.26711130142212
    },
    {
      "epoch": 1.627159118652344e-05,
      "model_forward_time": 2.9590301513671875,
      "step": 17062
    },
    {
      "epoch": 1.627159118652344e-05,
      "step": 17062,
      "training_step_time": 9.371253728866577
    },
    {
      "epoch": 1.6272544860839843e-05,
      "model_forward_time": 1.7601542472839355,
      "step": 17063
    },
    {
      "epoch": 1.6272544860839843e-05,
      "step": 17063,
      "training_step_time": 7.733635187149048
    },
    {
      "epoch": 1.627349853515625e-05,
      "model_forward_time": 1.7600700855255127,
      "step": 17064
    },
    {
      "epoch": 1.627349853515625e-05,
      "step": 17064,
      "training_step_time": 7.579732894897461
    },
    {
      "epoch": 1.6274452209472657e-05,
      "model_forward_time": 1.752638339996338,
      "step": 17065
    },
    {
      "epoch": 1.6274452209472657e-05,
      "step": 17065,
      "training_step_time": 6.788787603378296
    },
    {
      "epoch": 1.627540588378906e-05,
      "model_forward_time": 1.3772120475769043,
      "step": 17066
    },
    {
      "epoch": 1.627540588378906e-05,
      "step": 17066,
      "training_step_time": 6.7465105056762695
    },
    {
      "epoch": 1.627635955810547e-05,
      "model_forward_time": 1.7352917194366455,
      "step": 17067
    },
    {
      "epoch": 1.627635955810547e-05,
      "step": 17067,
      "training_step_time": 6.753259181976318
    },
    {
      "epoch": 1.6277313232421876e-05,
      "model_forward_time": 1.734813928604126,
      "step": 17068
    },
    {
      "epoch": 1.6277313232421876e-05,
      "step": 17068,
      "training_step_time": 6.678696393966675
    },
    {
      "epoch": 1.627826690673828e-05,
      "model_forward_time": 1.7581427097320557,
      "step": 17069
    },
    {
      "epoch": 1.627826690673828e-05,
      "step": 17069,
      "training_step_time": 6.744180202484131
    },
    {
      "epoch": 1.6279220581054688e-05,
      "grad_norm": 0.1858735978603363,
      "learning_rate": 9.607027227324866e-06,
      "loss": 0.0593,
      "step": 17070
    },
    {
      "epoch": 1.6279220581054688e-05,
      "model_forward_time": 1.7311692237854004,
      "step": 17070
    },
    {
      "dynamics_loss_avg": 0.062318201549351214,
      "epoch": 1.6279220581054688e-05,
      "step": 17070
    },
    {
      "action_loss_avg": 0.0036107780470047146,
      "epoch": 1.6279220581054688e-05,
      "step": 17070
    },
    {
      "epoch": 1.6279220581054688e-05,
      "step": 17070,
      "training_step_time": 7.639398813247681
    },
    {
      "epoch": 1.6280174255371095e-05,
      "model_forward_time": 3.013432502746582,
      "step": 17071
    },
    {
      "epoch": 1.6280174255371095e-05,
      "step": 17071,
      "training_step_time": 8.631466388702393
    },
    {
      "epoch": 1.62811279296875e-05,
      "model_forward_time": 1.7497897148132324,
      "step": 17072
    },
    {
      "epoch": 1.62811279296875e-05,
      "step": 17072,
      "training_step_time": 7.834282636642456
    },
    {
      "epoch": 1.6282081604003907e-05,
      "model_forward_time": 1.738929271697998,
      "step": 17073
    },
    {
      "epoch": 1.6282081604003907e-05,
      "step": 17073,
      "training_step_time": 7.310779333114624
    },
    {
      "epoch": 1.6283035278320314e-05,
      "model_forward_time": 1.7484703063964844,
      "step": 17074
    },
    {
      "epoch": 1.6283035278320314e-05,
      "step": 17074,
      "training_step_time": 6.724185943603516
    },
    {
      "epoch": 1.6283988952636718e-05,
      "model_forward_time": 1.7475554943084717,
      "step": 17075
    },
    {
      "epoch": 1.6283988952636718e-05,
      "step": 17075,
      "training_step_time": 6.790367841720581
    },
    {
      "epoch": 1.6284942626953125e-05,
      "model_forward_time": 1.7425107955932617,
      "step": 17076
    },
    {
      "epoch": 1.6284942626953125e-05,
      "step": 17076,
      "training_step_time": 6.7597246170043945
    },
    {
      "epoch": 1.6285896301269533e-05,
      "model_forward_time": 1.7433440685272217,
      "step": 17077
    },
    {
      "epoch": 1.6285896301269533e-05,
      "step": 17077,
      "training_step_time": 6.7537078857421875
    },
    {
      "epoch": 1.6286849975585937e-05,
      "model_forward_time": 1.7489409446716309,
      "step": 17078
    },
    {
      "epoch": 1.6286849975585937e-05,
      "step": 17078,
      "training_step_time": 6.758459091186523
    },
    {
      "epoch": 1.6287803649902344e-05,
      "model_forward_time": 1.741194248199463,
      "step": 17079
    },
    {
      "epoch": 1.6287803649902344e-05,
      "step": 17079,
      "training_step_time": 7.664787769317627
    },
    {
      "epoch": 1.628875732421875e-05,
      "grad_norm": 0.15853071212768555,
      "learning_rate": 9.606384432705837e-06,
      "loss": 0.0671,
      "step": 17080
    },
    {
      "epoch": 1.628875732421875e-05,
      "model_forward_time": 2.92956805229187,
      "step": 17080
    },
    {
      "dynamics_loss_avg": 0.06354921702295542,
      "epoch": 1.628875732421875e-05,
      "step": 17080
    },
    {
      "action_loss_avg": 0.0029366204980760813,
      "epoch": 1.628875732421875e-05,
      "step": 17080
    },
    {
      "epoch": 1.628875732421875e-05,
      "step": 17080,
      "training_step_time": 8.288062810897827
    },
    {
      "epoch": 1.6289710998535156e-05,
      "model_forward_time": 1.8484416007995605,
      "step": 17081
    },
    {
      "epoch": 1.6289710998535156e-05,
      "step": 17081,
      "training_step_time": 7.917145252227783
    },
    {
      "epoch": 1.6290664672851563e-05,
      "model_forward_time": 1.7270801067352295,
      "step": 17082
    },
    {
      "epoch": 1.6290664672851563e-05,
      "step": 17082,
      "training_step_time": 8.062175989151001
    },
    {
      "epoch": 1.6291618347167967e-05,
      "model_forward_time": 1.67140531539917,
      "step": 17083
    },
    {
      "epoch": 1.6291618347167967e-05,
      "step": 17083,
      "training_step_time": 7.538910627365112
    },
    {
      "epoch": 1.6292572021484375e-05,
      "model_forward_time": 1.6870226860046387,
      "step": 17084
    },
    {
      "epoch": 1.6292572021484375e-05,
      "step": 17084,
      "training_step_time": 6.753578186035156
    },
    {
      "epoch": 1.6293525695800782e-05,
      "model_forward_time": 1.6854193210601807,
      "step": 17085
    },
    {
      "epoch": 1.6293525695800782e-05,
      "step": 17085,
      "training_step_time": 6.759178400039673
    },
    {
      "epoch": 1.6294479370117186e-05,
      "model_forward_time": 1.683703899383545,
      "step": 17086
    },
    {
      "epoch": 1.6294479370117186e-05,
      "step": 17086,
      "training_step_time": 6.792397737503052
    },
    {
      "epoch": 1.6295433044433594e-05,
      "model_forward_time": 1.6926665306091309,
      "step": 17087
    },
    {
      "epoch": 1.6295433044433594e-05,
      "step": 17087,
      "training_step_time": 6.771583795547485
    },
    {
      "epoch": 1.629638671875e-05,
      "model_forward_time": 1.686640739440918,
      "step": 17088
    },
    {
      "epoch": 1.629638671875e-05,
      "step": 17088,
      "training_step_time": 7.620638847351074
    },
    {
      "epoch": 1.6297340393066405e-05,
      "model_forward_time": 2.2502832412719727,
      "step": 17089
    },
    {
      "epoch": 1.6297340393066405e-05,
      "step": 17089,
      "training_step_time": 9.720766544342041
    },
    {
      "epoch": 1.6298294067382812e-05,
      "grad_norm": 0.20411404967308044,
      "learning_rate": 9.605741134339531e-06,
      "loss": 0.0638,
      "step": 17090
    },
    {
      "epoch": 1.6298294067382812e-05,
      "model_forward_time": 0.9746172428131104,
      "step": 17090
    },
    {
      "dynamics_loss_avg": 0.06139289569109678,
      "epoch": 1.6298294067382812e-05,
      "step": 17090
    },
    {
      "action_loss_avg": 0.0028182527457829565,
      "epoch": 1.6298294067382812e-05,
      "step": 17090
    },
    {
      "epoch": 1.6298294067382812e-05,
      "step": 17090,
      "training_step_time": 7.737074613571167
    },
    {
      "epoch": 1.629924774169922e-05,
      "model_forward_time": 1.688215732574463,
      "step": 17091
    },
    {
      "epoch": 1.629924774169922e-05,
      "step": 17091,
      "training_step_time": 7.866487741470337
    },
    {
      "epoch": 1.6300201416015624e-05,
      "model_forward_time": 1.6894068717956543,
      "step": 17092
    },
    {
      "epoch": 1.6300201416015624e-05,
      "step": 17092,
      "training_step_time": 7.583189964294434
    },
    {
      "epoch": 1.630115509033203e-05,
      "model_forward_time": 1.685103178024292,
      "step": 17093
    },
    {
      "epoch": 1.630115509033203e-05,
      "step": 17093,
      "training_step_time": 6.721364736557007
    },
    {
      "epoch": 1.630210876464844e-05,
      "model_forward_time": 1.6840229034423828,
      "step": 17094
    },
    {
      "epoch": 1.630210876464844e-05,
      "step": 17094,
      "training_step_time": 6.7599804401397705
    },
    {
      "epoch": 1.6303062438964843e-05,
      "model_forward_time": 1.689561128616333,
      "step": 17095
    },
    {
      "epoch": 1.6303062438964843e-05,
      "step": 17095,
      "training_step_time": 6.802520275115967
    },
    {
      "epoch": 1.630401611328125e-05,
      "model_forward_time": 1.6825754642486572,
      "step": 17096
    },
    {
      "epoch": 1.630401611328125e-05,
      "step": 17096,
      "training_step_time": 6.760114908218384
    },
    {
      "epoch": 1.6304969787597657e-05,
      "model_forward_time": 1.6825463771820068,
      "step": 17097
    },
    {
      "epoch": 1.6304969787597657e-05,
      "step": 17097,
      "training_step_time": 7.637526512145996
    },
    {
      "epoch": 1.630592346191406e-05,
      "model_forward_time": 1.7131054401397705,
      "step": 17098
    },
    {
      "epoch": 1.630592346191406e-05,
      "step": 17098,
      "training_step_time": 8.505924224853516
    },
    {
      "epoch": 1.630687713623047e-05,
      "model_forward_time": 2.215048313140869,
      "step": 17099
    },
    {
      "epoch": 1.630687713623047e-05,
      "step": 17099,
      "training_step_time": 8.730325698852539
    },
    {
      "epoch": 1.6307830810546876e-05,
      "grad_norm": 0.19230177998542786,
      "learning_rate": 9.6050973322963e-06,
      "loss": 0.0655,
      "step": 17100
    },
    {
      "epoch": 1.6307830810546876e-05,
      "model_forward_time": 1.689044713973999,
      "step": 17100
    },
    {
      "dynamics_loss_avg": 0.055765218660235406,
      "epoch": 1.6307830810546876e-05,
      "step": 17100
    },
    {
      "action_loss_avg": 0.0045262537198141215,
      "epoch": 1.6307830810546876e-05,
      "step": 17100
    },
    {
      "epoch": 1.6307830810546876e-05,
      "step": 17100,
      "training_step_time": 7.8922388553619385
    },
    {
      "epoch": 1.630878448486328e-05,
      "model_forward_time": 1.6860296726226807,
      "step": 17101
    },
    {
      "epoch": 1.630878448486328e-05,
      "step": 17101,
      "training_step_time": 7.56287407875061
    },
    {
      "epoch": 1.6309738159179688e-05,
      "model_forward_time": 1.6860125064849854,
      "step": 17102
    },
    {
      "epoch": 1.6309738159179688e-05,
      "step": 17102,
      "training_step_time": 6.741150379180908
    },
    {
      "epoch": 1.6310691833496095e-05,
      "model_forward_time": 1.6907634735107422,
      "step": 17103
    },
    {
      "epoch": 1.6310691833496095e-05,
      "step": 17103,
      "training_step_time": 6.713603258132935
    },
    {
      "epoch": 1.63116455078125e-05,
      "model_forward_time": 1.6919879913330078,
      "step": 17104
    },
    {
      "epoch": 1.63116455078125e-05,
      "step": 17104,
      "training_step_time": 6.735133647918701
    },
    {
      "epoch": 1.6312599182128907e-05,
      "model_forward_time": 1.687819242477417,
      "step": 17105
    },
    {
      "epoch": 1.6312599182128907e-05,
      "step": 17105,
      "training_step_time": 6.778009414672852
    },
    {
      "epoch": 1.6313552856445314e-05,
      "model_forward_time": 1.6897518634796143,
      "step": 17106
    },
    {
      "epoch": 1.6313552856445314e-05,
      "step": 17106,
      "training_step_time": 7.586706638336182
    },
    {
      "epoch": 1.6314506530761718e-05,
      "model_forward_time": 1.7038142681121826,
      "step": 17107
    },
    {
      "epoch": 1.6314506530761718e-05,
      "step": 17107,
      "training_step_time": 8.782565116882324
    },
    {
      "epoch": 1.6315460205078126e-05,
      "model_forward_time": 2.21044921875,
      "step": 17108
    },
    {
      "epoch": 1.6315460205078126e-05,
      "step": 17108,
      "training_step_time": 9.329413175582886
    },
    {
      "epoch": 1.6316413879394533e-05,
      "model_forward_time": 1.7078769207000732,
      "step": 17109
    },
    {
      "epoch": 1.6316413879394533e-05,
      "step": 17109,
      "training_step_time": 7.911712408065796
    },
    {
      "epoch": 1.6317367553710937e-05,
      "grad_norm": 0.23132914304733276,
      "learning_rate": 9.604453026646548e-06,
      "loss": 0.0643,
      "step": 17110
    },
    {
      "epoch": 1.6317367553710937e-05,
      "model_forward_time": 1.7089333534240723,
      "step": 17110
    },
    {
      "dynamics_loss_avg": 0.06167688425630331,
      "epoch": 1.6317367553710937e-05,
      "step": 17110
    },
    {
      "action_loss_avg": 0.0038641131657641383,
      "epoch": 1.6317367553710937e-05,
      "step": 17110
    },
    {
      "epoch": 1.6317367553710937e-05,
      "step": 17110,
      "training_step_time": 7.571145534515381
    },
    {
      "epoch": 1.6318321228027344e-05,
      "model_forward_time": 1.7068729400634766,
      "step": 17111
    },
    {
      "epoch": 1.6318321228027344e-05,
      "step": 17111,
      "training_step_time": 6.748937606811523
    },
    {
      "epoch": 1.631927490234375e-05,
      "model_forward_time": 1.7125797271728516,
      "step": 17112
    },
    {
      "epoch": 1.631927490234375e-05,
      "step": 17112,
      "training_step_time": 6.753169775009155
    },
    {
      "epoch": 1.6320228576660156e-05,
      "model_forward_time": 1.710179090499878,
      "step": 17113
    },
    {
      "epoch": 1.6320228576660156e-05,
      "step": 17113,
      "training_step_time": 6.790317058563232
    },
    {
      "epoch": 1.6321182250976563e-05,
      "model_forward_time": 1.7089545726776123,
      "step": 17114
    },
    {
      "epoch": 1.6321182250976563e-05,
      "step": 17114,
      "training_step_time": 6.771852254867554
    },
    {
      "epoch": 1.6322135925292967e-05,
      "model_forward_time": 1.7110095024108887,
      "step": 17115
    },
    {
      "epoch": 1.6322135925292967e-05,
      "step": 17115,
      "training_step_time": 6.809281826019287
    },
    {
      "epoch": 1.6323089599609375e-05,
      "model_forward_time": 1.7523460388183594,
      "step": 17116
    },
    {
      "epoch": 1.6323089599609375e-05,
      "step": 17116,
      "training_step_time": 8.37732720375061
    },
    {
      "epoch": 1.6324043273925782e-05,
      "model_forward_time": 2.2255711555480957,
      "step": 17117
    },
    {
      "epoch": 1.6324043273925782e-05,
      "step": 17117,
      "training_step_time": 8.585448026657104
    },
    {
      "epoch": 1.6324996948242186e-05,
      "model_forward_time": 1.709740161895752,
      "step": 17118
    },
    {
      "epoch": 1.6324996948242186e-05,
      "step": 17118,
      "training_step_time": 7.955202102661133
    },
    {
      "epoch": 1.6325950622558594e-05,
      "model_forward_time": 1.7067973613739014,
      "step": 17119
    },
    {
      "epoch": 1.6325950622558594e-05,
      "step": 17119,
      "training_step_time": 7.6438634395599365
    },
    {
      "epoch": 1.6326904296875e-05,
      "grad_norm": 0.19368772208690643,
      "learning_rate": 9.603808217460735e-06,
      "loss": 0.0667,
      "step": 17120
    },
    {
      "epoch": 1.6326904296875e-05,
      "model_forward_time": 1.7040834426879883,
      "step": 17120
    },
    {
      "dynamics_loss_avg": 0.060064639896154404,
      "epoch": 1.6326904296875e-05,
      "step": 17120
    },
    {
      "action_loss_avg": 0.00229509964119643,
      "epoch": 1.6326904296875e-05,
      "step": 17120
    },
    {
      "epoch": 1.6326904296875e-05,
      "step": 17120,
      "training_step_time": 6.806386947631836
    },
    {
      "epoch": 1.6327857971191405e-05,
      "model_forward_time": 1.7044589519500732,
      "step": 17121
    },
    {
      "epoch": 1.6327857971191405e-05,
      "step": 17121,
      "training_step_time": 6.823287487030029
    },
    {
      "epoch": 1.6328811645507812e-05,
      "model_forward_time": 1.7105660438537598,
      "step": 17122
    },
    {
      "epoch": 1.6328811645507812e-05,
      "step": 17122,
      "training_step_time": 6.779660224914551
    },
    {
      "epoch": 1.632976531982422e-05,
      "model_forward_time": 1.7118217945098877,
      "step": 17123
    },
    {
      "epoch": 1.632976531982422e-05,
      "step": 17123,
      "training_step_time": 6.816034555435181
    },
    {
      "epoch": 1.6330718994140624e-05,
      "model_forward_time": 1.7073452472686768,
      "step": 17124
    },
    {
      "epoch": 1.6330718994140624e-05,
      "step": 17124,
      "training_step_time": 6.772778749465942
    },
    {
      "epoch": 1.633167266845703e-05,
      "model_forward_time": 1.744800329208374,
      "step": 17125
    },
    {
      "epoch": 1.633167266845703e-05,
      "step": 17125,
      "training_step_time": 7.5940260887146
    },
    {
      "epoch": 1.633262634277344e-05,
      "model_forward_time": 2.2434520721435547,
      "step": 17126
    },
    {
      "epoch": 1.633262634277344e-05,
      "step": 17126,
      "training_step_time": 10.650132179260254
    },
    {
      "epoch": 1.6333580017089843e-05,
      "model_forward_time": 1.699927806854248,
      "step": 17127
    },
    {
      "epoch": 1.6333580017089843e-05,
      "step": 17127,
      "training_step_time": 10.593490600585938
    },
    {
      "epoch": 1.633453369140625e-05,
      "model_forward_time": 1.7121772766113281,
      "step": 17128
    },
    {
      "epoch": 1.633453369140625e-05,
      "step": 17128,
      "training_step_time": 7.436049461364746
    },
    {
      "epoch": 1.6335487365722658e-05,
      "model_forward_time": 1.3389642238616943,
      "step": 17129
    },
    {
      "epoch": 1.6335487365722658e-05,
      "step": 17129,
      "training_step_time": 7.57044529914856
    },
    {
      "epoch": 1.633644104003906e-05,
      "grad_norm": 0.3043598234653473,
      "learning_rate": 9.603162904809377e-06,
      "loss": 0.0633,
      "step": 17130
    },
    {
      "epoch": 1.633644104003906e-05,
      "model_forward_time": 1.7113983631134033,
      "step": 17130
    },
    {
      "dynamics_loss_avg": 0.07521739508956671,
      "epoch": 1.633644104003906e-05,
      "step": 17130
    },
    {
      "action_loss_avg": 0.005183685896918177,
      "epoch": 1.633644104003906e-05,
      "step": 17130
    },
    {
      "epoch": 1.633644104003906e-05,
      "step": 17130,
      "training_step_time": 6.738739490509033
    },
    {
      "epoch": 1.633739471435547e-05,
      "model_forward_time": 1.7155091762542725,
      "step": 17131
    },
    {
      "epoch": 1.633739471435547e-05,
      "step": 17131,
      "training_step_time": 6.728129148483276
    },
    {
      "epoch": 1.6338348388671876e-05,
      "model_forward_time": 1.715897798538208,
      "step": 17132
    },
    {
      "epoch": 1.6338348388671876e-05,
      "step": 17132,
      "training_step_time": 6.749090671539307
    },
    {
      "epoch": 1.633930206298828e-05,
      "model_forward_time": 1.7200572490692139,
      "step": 17133
    },
    {
      "epoch": 1.633930206298828e-05,
      "step": 17133,
      "training_step_time": 6.761388778686523
    },
    {
      "epoch": 1.6340255737304688e-05,
      "model_forward_time": 1.7278187274932861,
      "step": 17134
    },
    {
      "epoch": 1.6340255737304688e-05,
      "step": 17134,
      "training_step_time": 7.572207450866699
    },
    {
      "epoch": 1.6341209411621095e-05,
      "model_forward_time": 2.239391803741455,
      "step": 17135
    },
    {
      "epoch": 1.6341209411621095e-05,
      "step": 17135,
      "training_step_time": 8.578264474868774
    },
    {
      "epoch": 1.63421630859375e-05,
      "model_forward_time": 1.7152299880981445,
      "step": 17136
    },
    {
      "epoch": 1.63421630859375e-05,
      "step": 17136,
      "training_step_time": 8.893753290176392
    },
    {
      "epoch": 1.6343116760253907e-05,
      "model_forward_time": 1.7459535598754883,
      "step": 17137
    },
    {
      "epoch": 1.6343116760253907e-05,
      "step": 17137,
      "training_step_time": 7.700934410095215
    },
    {
      "epoch": 1.6344070434570314e-05,
      "model_forward_time": 1.715322494506836,
      "step": 17138
    },
    {
      "epoch": 1.6344070434570314e-05,
      "step": 17138,
      "training_step_time": 7.5946574211120605
    },
    {
      "epoch": 1.6345024108886718e-05,
      "model_forward_time": 1.7141149044036865,
      "step": 17139
    },
    {
      "epoch": 1.6345024108886718e-05,
      "step": 17139,
      "training_step_time": 6.7475786209106445
    },
    {
      "epoch": 1.6345977783203126e-05,
      "grad_norm": 0.28126367926597595,
      "learning_rate": 9.602517088763046e-06,
      "loss": 0.0662,
      "step": 17140
    },
    {
      "epoch": 1.6345977783203126e-05,
      "model_forward_time": 1.710526704788208,
      "step": 17140
    },
    {
      "dynamics_loss_avg": 0.08751910552382469,
      "epoch": 1.6345977783203126e-05,
      "step": 17140
    },
    {
      "action_loss_avg": 0.00445873171556741,
      "epoch": 1.6345977783203126e-05,
      "step": 17140
    },
    {
      "epoch": 1.6345977783203126e-05,
      "step": 17140,
      "training_step_time": 6.73838210105896
    },
    {
      "epoch": 1.634693145751953e-05,
      "model_forward_time": 1.7235314846038818,
      "step": 17141
    },
    {
      "epoch": 1.634693145751953e-05,
      "step": 17141,
      "training_step_time": 6.793986558914185
    },
    {
      "epoch": 1.6347885131835937e-05,
      "model_forward_time": 1.7191600799560547,
      "step": 17142
    },
    {
      "epoch": 1.6347885131835937e-05,
      "step": 17142,
      "training_step_time": 6.751239538192749
    },
    {
      "epoch": 1.6348838806152344e-05,
      "model_forward_time": 1.7208778858184814,
      "step": 17143
    },
    {
      "epoch": 1.6348838806152344e-05,
      "step": 17143,
      "training_step_time": 7.630876064300537
    },
    {
      "epoch": 1.634979248046875e-05,
      "model_forward_time": 1.7521419525146484,
      "step": 17144
    },
    {
      "epoch": 1.634979248046875e-05,
      "step": 17144,
      "training_step_time": 10.494771957397461
    },
    {
      "epoch": 1.6350746154785156e-05,
      "model_forward_time": 2.242140293121338,
      "step": 17145
    },
    {
      "epoch": 1.6350746154785156e-05,
      "step": 17145,
      "training_step_time": 12.078033685684204
    },
    {
      "epoch": 1.6351699829101563e-05,
      "model_forward_time": 1.7103784084320068,
      "step": 17146
    },
    {
      "epoch": 1.6351699829101563e-05,
      "step": 17146,
      "training_step_time": 7.948243618011475
    },
    {
      "epoch": 1.6352653503417967e-05,
      "model_forward_time": 1.708921194076538,
      "step": 17147
    },
    {
      "epoch": 1.6352653503417967e-05,
      "step": 17147,
      "training_step_time": 7.646619081497192
    },
    {
      "epoch": 1.6353607177734375e-05,
      "model_forward_time": 1.7106871604919434,
      "step": 17148
    },
    {
      "epoch": 1.6353607177734375e-05,
      "step": 17148,
      "training_step_time": 6.928187131881714
    },
    {
      "epoch": 1.6354560852050782e-05,
      "model_forward_time": 1.7173480987548828,
      "step": 17149
    },
    {
      "epoch": 1.6354560852050782e-05,
      "step": 17149,
      "training_step_time": 6.8561625480651855
    },
    {
      "epoch": 1.6355514526367186e-05,
      "grad_norm": 0.24565790593624115,
      "learning_rate": 9.601870769392366e-06,
      "loss": 0.0651,
      "step": 17150
    },
    {
      "epoch": 1.6355514526367186e-05,
      "model_forward_time": 1.7050726413726807,
      "step": 17150
    },
    {
      "dynamics_loss_avg": 0.08177626468241214,
      "epoch": 1.6355514526367186e-05,
      "step": 17150
    },
    {
      "action_loss_avg": 0.00515792021760717,
      "epoch": 1.6355514526367186e-05,
      "step": 17150
    },
    {
      "epoch": 1.6355514526367186e-05,
      "step": 17150,
      "training_step_time": 6.857015371322632
    },
    {
      "epoch": 1.6356468200683594e-05,
      "model_forward_time": 1.7144927978515625,
      "step": 17151
    },
    {
      "epoch": 1.6356468200683594e-05,
      "step": 17151,
      "training_step_time": 6.838552951812744
    },
    {
      "epoch": 1.6357421875e-05,
      "model_forward_time": 1.7080581188201904,
      "step": 17152
    },
    {
      "epoch": 1.6357421875e-05,
      "step": 17152,
      "training_step_time": 7.617340087890625
    },
    {
      "epoch": 1.6358375549316405e-05,
      "model_forward_time": 1.7408053874969482,
      "step": 17153
    },
    {
      "epoch": 1.6358375549316405e-05,
      "step": 17153,
      "training_step_time": 10.277465581893921
    },
    {
      "epoch": 1.6359329223632812e-05,
      "model_forward_time": 2.2431344985961914,
      "step": 17154
    },
    {
      "epoch": 1.6359329223632812e-05,
      "step": 17154,
      "training_step_time": 9.064838886260986
    },
    {
      "epoch": 1.636028289794922e-05,
      "model_forward_time": 1.7038850784301758,
      "step": 17155
    },
    {
      "epoch": 1.636028289794922e-05,
      "step": 17155,
      "training_step_time": 8.830116271972656
    },
    {
      "epoch": 1.6361236572265624e-05,
      "model_forward_time": 1.7076804637908936,
      "step": 17156
    },
    {
      "epoch": 1.6361236572265624e-05,
      "step": 17156,
      "training_step_time": 7.22751259803772
    },
    {
      "epoch": 1.636219024658203e-05,
      "model_forward_time": 1.7114856243133545,
      "step": 17157
    },
    {
      "epoch": 1.636219024658203e-05,
      "step": 17157,
      "training_step_time": 7.599799633026123
    },
    {
      "epoch": 1.636314392089844e-05,
      "model_forward_time": 1.7110121250152588,
      "step": 17158
    },
    {
      "epoch": 1.636314392089844e-05,
      "step": 17158,
      "training_step_time": 6.957058906555176
    },
    {
      "epoch": 1.6364097595214843e-05,
      "model_forward_time": 1.7118146419525146,
      "step": 17159
    },
    {
      "epoch": 1.6364097595214843e-05,
      "step": 17159,
      "training_step_time": 6.749469995498657
    },
    {
      "epoch": 1.636505126953125e-05,
      "grad_norm": 0.2376859039068222,
      "learning_rate": 9.601223946768017e-06,
      "loss": 0.0659,
      "step": 17160
    },
    {
      "epoch": 1.636505126953125e-05,
      "model_forward_time": 1.7119152545928955,
      "step": 17160
    },
    {
      "dynamics_loss_avg": 0.06059541068971157,
      "epoch": 1.636505126953125e-05,
      "step": 17160
    },
    {
      "action_loss_avg": 0.00252861165208742,
      "epoch": 1.636505126953125e-05,
      "step": 17160
    },
    {
      "epoch": 1.636505126953125e-05,
      "step": 17160,
      "training_step_time": 6.849487543106079
    },
    {
      "epoch": 1.6366004943847658e-05,
      "model_forward_time": 1.7170875072479248,
      "step": 17161
    },
    {
      "epoch": 1.6366004943847658e-05,
      "step": 17161,
      "training_step_time": 7.178780555725098
    },
    {
      "epoch": 1.636695861816406e-05,
      "model_forward_time": 1.7247531414031982,
      "step": 17162
    },
    {
      "epoch": 1.636695861816406e-05,
      "step": 17162,
      "training_step_time": 8.699983835220337
    },
    {
      "epoch": 1.636791229248047e-05,
      "model_forward_time": 2.2432096004486084,
      "step": 17163
    },
    {
      "epoch": 1.636791229248047e-05,
      "step": 17163,
      "training_step_time": 8.824023723602295
    },
    {
      "epoch": 1.6368865966796876e-05,
      "model_forward_time": 1.7117722034454346,
      "step": 17164
    },
    {
      "epoch": 1.6368865966796876e-05,
      "step": 17164,
      "training_step_time": 9.004979133605957
    },
    {
      "epoch": 1.636981964111328e-05,
      "model_forward_time": 1.7057042121887207,
      "step": 17165
    },
    {
      "epoch": 1.636981964111328e-05,
      "step": 17165,
      "training_step_time": 7.673722267150879
    },
    {
      "epoch": 1.6370773315429688e-05,
      "model_forward_time": 1.7112936973571777,
      "step": 17166
    },
    {
      "epoch": 1.6370773315429688e-05,
      "step": 17166,
      "training_step_time": 7.595280885696411
    },
    {
      "epoch": 1.6371726989746095e-05,
      "model_forward_time": 1.7153584957122803,
      "step": 17167
    },
    {
      "epoch": 1.6371726989746095e-05,
      "step": 17167,
      "training_step_time": 6.857855796813965
    },
    {
      "epoch": 1.63726806640625e-05,
      "model_forward_time": 1.7126507759094238,
      "step": 17168
    },
    {
      "epoch": 1.63726806640625e-05,
      "step": 17168,
      "training_step_time": 6.839285850524902
    },
    {
      "epoch": 1.6373634338378907e-05,
      "model_forward_time": 1.7164561748504639,
      "step": 17169
    },
    {
      "epoch": 1.6373634338378907e-05,
      "step": 17169,
      "training_step_time": 6.810171604156494
    },
    {
      "epoch": 1.6374588012695314e-05,
      "grad_norm": 0.22324569523334503,
      "learning_rate": 9.600576620960735e-06,
      "loss": 0.0676,
      "step": 17170
    },
    {
      "epoch": 1.6374588012695314e-05,
      "model_forward_time": 1.7123982906341553,
      "step": 17170
    },
    {
      "dynamics_loss_avg": 0.06793568395078182,
      "epoch": 1.6374588012695314e-05,
      "step": 17170
    },
    {
      "action_loss_avg": 0.0029866597964428366,
      "epoch": 1.6374588012695314e-05,
      "step": 17170
    },
    {
      "epoch": 1.6374588012695314e-05,
      "step": 17170,
      "training_step_time": 7.125925302505493
    },
    {
      "epoch": 1.6375541687011718e-05,
      "model_forward_time": 1.711393117904663,
      "step": 17171
    },
    {
      "epoch": 1.6375541687011718e-05,
      "step": 17171,
      "training_step_time": 8.217118978500366
    },
    {
      "epoch": 1.6376495361328126e-05,
      "model_forward_time": 2.226641893386841,
      "step": 17172
    },
    {
      "epoch": 1.6376495361328126e-05,
      "step": 17172,
      "training_step_time": 9.041776657104492
    },
    {
      "epoch": 1.637744903564453e-05,
      "model_forward_time": 1.6873939037322998,
      "step": 17173
    },
    {
      "epoch": 1.637744903564453e-05,
      "step": 17173,
      "training_step_time": 8.824636936187744
    },
    {
      "epoch": 1.6378402709960937e-05,
      "model_forward_time": 1.69620943069458,
      "step": 17174
    },
    {
      "epoch": 1.6378402709960937e-05,
      "step": 17174,
      "training_step_time": 7.693052291870117
    },
    {
      "epoch": 1.6379356384277344e-05,
      "model_forward_time": 1.7061965465545654,
      "step": 17175
    },
    {
      "epoch": 1.6379356384277344e-05,
      "step": 17175,
      "training_step_time": 7.610523223876953
    },
    {
      "epoch": 1.638031005859375e-05,
      "model_forward_time": 1.7022430896759033,
      "step": 17176
    },
    {
      "epoch": 1.638031005859375e-05,
      "step": 17176,
      "training_step_time": 6.826090335845947
    },
    {
      "epoch": 1.6381263732910156e-05,
      "model_forward_time": 1.702789545059204,
      "step": 17177
    },
    {
      "epoch": 1.6381263732910156e-05,
      "step": 17177,
      "training_step_time": 6.8250510692596436
    },
    {
      "epoch": 1.6382217407226563e-05,
      "model_forward_time": 1.7091505527496338,
      "step": 17178
    },
    {
      "epoch": 1.6382217407226563e-05,
      "step": 17178,
      "training_step_time": 6.8245320320129395
    },
    {
      "epoch": 1.6383171081542967e-05,
      "model_forward_time": 1.7011919021606445,
      "step": 17179
    },
    {
      "epoch": 1.6383171081542967e-05,
      "step": 17179,
      "training_step_time": 7.119071006774902
    },
    {
      "epoch": 1.6384124755859375e-05,
      "grad_norm": 0.2574726641178131,
      "learning_rate": 9.59992879204131e-06,
      "loss": 0.0668,
      "step": 17180
    },
    {
      "epoch": 1.6384124755859375e-05,
      "model_forward_time": 1.695202350616455,
      "step": 17180
    },
    {
      "dynamics_loss_avg": 0.059850841760635376,
      "epoch": 1.6384124755859375e-05,
      "step": 17180
    },
    {
      "action_loss_avg": 0.004127051751129329,
      "epoch": 1.6384124755859375e-05,
      "step": 17180
    },
    {
      "epoch": 1.6384124755859375e-05,
      "step": 17180,
      "training_step_time": 8.6236572265625
    },
    {
      "epoch": 1.6385078430175782e-05,
      "model_forward_time": 2.261201858520508,
      "step": 17181
    },
    {
      "epoch": 1.6385078430175782e-05,
      "step": 17181,
      "training_step_time": 9.586894750595093
    },
    {
      "epoch": 1.6386032104492186e-05,
      "model_forward_time": 1.710533857345581,
      "step": 17182
    },
    {
      "epoch": 1.6386032104492186e-05,
      "step": 17182,
      "training_step_time": 9.634811878204346
    },
    {
      "epoch": 1.6386985778808594e-05,
      "model_forward_time": 1.7036681175231934,
      "step": 17183
    },
    {
      "epoch": 1.6386985778808594e-05,
      "step": 17183,
      "training_step_time": 7.815629720687866
    },
    {
      "epoch": 1.6387939453125e-05,
      "model_forward_time": 1.7241461277008057,
      "step": 17184
    },
    {
      "epoch": 1.6387939453125e-05,
      "step": 17184,
      "training_step_time": 7.558249473571777
    },
    {
      "epoch": 1.6388893127441405e-05,
      "model_forward_time": 1.6865079402923584,
      "step": 17185
    },
    {
      "epoch": 1.6388893127441405e-05,
      "step": 17185,
      "training_step_time": 6.84899640083313
    },
    {
      "epoch": 1.6389846801757812e-05,
      "model_forward_time": 1.7012293338775635,
      "step": 17186
    },
    {
      "epoch": 1.6389846801757812e-05,
      "step": 17186,
      "training_step_time": 6.805068254470825
    },
    {
      "epoch": 1.639080047607422e-05,
      "model_forward_time": 1.7060110569000244,
      "step": 17187
    },
    {
      "epoch": 1.639080047607422e-05,
      "step": 17187,
      "training_step_time": 6.773604154586792
    },
    {
      "epoch": 1.6391754150390624e-05,
      "model_forward_time": 1.7056734561920166,
      "step": 17188
    },
    {
      "epoch": 1.6391754150390624e-05,
      "step": 17188,
      "training_step_time": 6.958532810211182
    },
    {
      "epoch": 1.639270782470703e-05,
      "model_forward_time": 1.6854472160339355,
      "step": 17189
    },
    {
      "epoch": 1.639270782470703e-05,
      "step": 17189,
      "training_step_time": 8.280525922775269
    },
    {
      "epoch": 1.639366149902344e-05,
      "grad_norm": 0.24782586097717285,
      "learning_rate": 9.599280460080587e-06,
      "loss": 0.0656,
      "step": 17190
    },
    {
      "epoch": 1.639366149902344e-05,
      "model_forward_time": 1.7419073581695557,
      "step": 17190
    },
    {
      "dynamics_loss_avg": 0.05509596932679415,
      "epoch": 1.639366149902344e-05,
      "step": 17190
    },
    {
      "action_loss_avg": 0.00337345558218658,
      "epoch": 1.639366149902344e-05,
      "step": 17190
    },
    {
      "epoch": 1.639366149902344e-05,
      "step": 17190,
      "training_step_time": 8.79117751121521
    },
    {
      "epoch": 1.6394615173339843e-05,
      "model_forward_time": 1.6954996585845947,
      "step": 17191
    },
    {
      "epoch": 1.6394615173339843e-05,
      "step": 17191,
      "training_step_time": 8.755505561828613
    },
    {
      "epoch": 1.639556884765625e-05,
      "model_forward_time": 1.7048039436340332,
      "step": 17192
    },
    {
      "epoch": 1.639556884765625e-05,
      "step": 17192,
      "training_step_time": 7.90989089012146
    },
    {
      "epoch": 1.6396522521972658e-05,
      "model_forward_time": 1.324556827545166,
      "step": 17193
    },
    {
      "epoch": 1.6396522521972658e-05,
      "step": 17193,
      "training_step_time": 7.4508373737335205
    },
    {
      "epoch": 1.639747619628906e-05,
      "model_forward_time": 1.6943473815917969,
      "step": 17194
    },
    {
      "epoch": 1.639747619628906e-05,
      "step": 17194,
      "training_step_time": 7.571955919265747
    },
    {
      "epoch": 1.639842987060547e-05,
      "model_forward_time": 1.7006144523620605,
      "step": 17195
    },
    {
      "epoch": 1.639842987060547e-05,
      "step": 17195,
      "training_step_time": 6.819153070449829
    },
    {
      "epoch": 1.6399383544921876e-05,
      "model_forward_time": 1.7097609043121338,
      "step": 17196
    },
    {
      "epoch": 1.6399383544921876e-05,
      "step": 17196,
      "training_step_time": 6.817973613739014
    },
    {
      "epoch": 1.640033721923828e-05,
      "model_forward_time": 1.7013988494873047,
      "step": 17197
    },
    {
      "epoch": 1.640033721923828e-05,
      "step": 17197,
      "training_step_time": 7.077689170837402
    },
    {
      "epoch": 1.6401290893554688e-05,
      "model_forward_time": 1.7005016803741455,
      "step": 17198
    },
    {
      "epoch": 1.6401290893554688e-05,
      "step": 17198,
      "training_step_time": 8.3038969039917
    },
    {
      "epoch": 1.6402244567871095e-05,
      "model_forward_time": 1.7075212001800537,
      "step": 17199
    },
    {
      "epoch": 1.6402244567871095e-05,
      "step": 17199,
      "training_step_time": 9.438045024871826
    },
    {
      "epoch": 1.64031982421875e-05,
      "grad_norm": 0.19245313107967377,
      "learning_rate": 9.59863162514947e-06,
      "loss": 0.0616,
      "step": 17200
    },
    {
      "epoch": 1.64031982421875e-05,
      "model_forward_time": 2.2252159118652344,
      "step": 17200
    },
    {
      "dynamics_loss_avg": 0.06621031370013952,
      "epoch": 1.64031982421875e-05,
      "step": 17200
    },
    {
      "action_loss_avg": 0.00353276381501928,
      "epoch": 1.64031982421875e-05,
      "step": 17200
    },
    {
      "epoch": 1.64031982421875e-05,
      "step": 17200,
      "training_step_time": 9.45758056640625
    },
    {
      "epoch": 1.6404151916503907e-05,
      "model_forward_time": 1.699958324432373,
      "step": 17201
    },
    {
      "epoch": 1.6404151916503907e-05,
      "step": 17201,
      "training_step_time": 7.96495509147644
    },
    {
      "epoch": 1.6405105590820314e-05,
      "model_forward_time": 1.7048413753509521,
      "step": 17202
    },
    {
      "epoch": 1.6405105590820314e-05,
      "step": 17202,
      "training_step_time": 7.754256963729858
    },
    {
      "epoch": 1.6406059265136718e-05,
      "model_forward_time": 1.7018187046051025,
      "step": 17203
    },
    {
      "epoch": 1.6406059265136718e-05,
      "step": 17203,
      "training_step_time": 7.711731672286987
    },
    {
      "epoch": 1.6407012939453126e-05,
      "model_forward_time": 1.6994197368621826,
      "step": 17204
    },
    {
      "epoch": 1.6407012939453126e-05,
      "step": 17204,
      "training_step_time": 6.875879526138306
    },
    {
      "epoch": 1.640796661376953e-05,
      "model_forward_time": 1.7075576782226562,
      "step": 17205
    },
    {
      "epoch": 1.640796661376953e-05,
      "step": 17205,
      "training_step_time": 6.85649561882019
    },
    {
      "epoch": 1.6408920288085937e-05,
      "model_forward_time": 1.705322027206421,
      "step": 17206
    },
    {
      "epoch": 1.6408920288085937e-05,
      "step": 17206,
      "training_step_time": 7.087897539138794
    },
    {
      "epoch": 1.6409873962402344e-05,
      "model_forward_time": 1.7007744312286377,
      "step": 17207
    },
    {
      "epoch": 1.6409873962402344e-05,
      "step": 17207,
      "training_step_time": 7.1459736824035645
    },
    {
      "epoch": 1.641082763671875e-05,
      "model_forward_time": 1.7027225494384766,
      "step": 17208
    },
    {
      "epoch": 1.641082763671875e-05,
      "step": 17208,
      "training_step_time": 8.472012519836426
    },
    {
      "epoch": 1.6411781311035156e-05,
      "model_forward_time": 2.261922836303711,
      "step": 17209
    },
    {
      "epoch": 1.6411781311035156e-05,
      "step": 17209,
      "training_step_time": 8.380638122558594
    },
    {
      "epoch": 1.6412734985351563e-05,
      "grad_norm": 0.1517893373966217,
      "learning_rate": 9.597982287318911e-06,
      "loss": 0.0636,
      "step": 17210
    },
    {
      "epoch": 1.6412734985351563e-05,
      "model_forward_time": 1.7016329765319824,
      "step": 17210
    },
    {
      "dynamics_loss_avg": 0.07708497978746891,
      "epoch": 1.6412734985351563e-05,
      "step": 17210
    },
    {
      "action_loss_avg": 0.003186270222067833,
      "epoch": 1.6412734985351563e-05,
      "step": 17210
    },
    {
      "epoch": 1.6412734985351563e-05,
      "step": 17210,
      "training_step_time": 8.01063060760498
    },
    {
      "epoch": 1.6413688659667967e-05,
      "model_forward_time": 1.698277235031128,
      "step": 17211
    },
    {
      "epoch": 1.6413688659667967e-05,
      "step": 17211,
      "training_step_time": 7.646050453186035
    },
    {
      "epoch": 1.6414642333984375e-05,
      "model_forward_time": 1.6928763389587402,
      "step": 17212
    },
    {
      "epoch": 1.6414642333984375e-05,
      "step": 17212,
      "training_step_time": 7.567226886749268
    },
    {
      "epoch": 1.6415596008300782e-05,
      "model_forward_time": 1.3169128894805908,
      "step": 17213
    },
    {
      "epoch": 1.6415596008300782e-05,
      "step": 17213,
      "training_step_time": 6.874109268188477
    },
    {
      "epoch": 1.6416549682617186e-05,
      "model_forward_time": 1.7027809619903564,
      "step": 17214
    },
    {
      "epoch": 1.6416549682617186e-05,
      "step": 17214,
      "training_step_time": 6.8306825160980225
    },
    {
      "epoch": 1.6417503356933594e-05,
      "model_forward_time": 0.9828033447265625,
      "step": 17215
    },
    {
      "epoch": 1.6417503356933594e-05,
      "step": 17215,
      "training_step_time": 7.15662956237793
    },
    {
      "epoch": 1.641845703125e-05,
      "model_forward_time": 1.7044599056243896,
      "step": 17216
    },
    {
      "epoch": 1.641845703125e-05,
      "step": 17216,
      "training_step_time": 7.130322456359863
    },
    {
      "epoch": 1.6419410705566405e-05,
      "model_forward_time": 1.7117729187011719,
      "step": 17217
    },
    {
      "epoch": 1.6419410705566405e-05,
      "step": 17217,
      "training_step_time": 9.25270414352417
    },
    {
      "epoch": 1.6420364379882812e-05,
      "model_forward_time": 1.7527284622192383,
      "step": 17218
    },
    {
      "epoch": 1.6420364379882812e-05,
      "step": 17218,
      "training_step_time": 9.530018091201782
    },
    {
      "epoch": 1.642131805419922e-05,
      "model_forward_time": 2.2379698753356934,
      "step": 17219
    },
    {
      "epoch": 1.642131805419922e-05,
      "step": 17219,
      "training_step_time": 8.975132703781128
    },
    {
      "epoch": 1.6422271728515624e-05,
      "grad_norm": 0.36497917771339417,
      "learning_rate": 9.597332446659924e-06,
      "loss": 0.0613,
      "step": 17220
    },
    {
      "epoch": 1.6422271728515624e-05,
      "model_forward_time": 1.7104449272155762,
      "step": 17220
    },
    {
      "dynamics_loss_avg": 0.07177469190210103,
      "epoch": 1.6422271728515624e-05,
      "step": 17220
    },
    {
      "action_loss_avg": 0.0034630770387593657,
      "epoch": 1.6422271728515624e-05,
      "step": 17220
    },
    {
      "epoch": 1.6422271728515624e-05,
      "step": 17220,
      "training_step_time": 8.14322018623352
    },
    {
      "epoch": 1.642322540283203e-05,
      "model_forward_time": 1.7065346240997314,
      "step": 17221
    },
    {
      "epoch": 1.642322540283203e-05,
      "step": 17221,
      "training_step_time": 7.742779731750488
    },
    {
      "epoch": 1.642417907714844e-05,
      "model_forward_time": 1.7037444114685059,
      "step": 17222
    },
    {
      "epoch": 1.642417907714844e-05,
      "step": 17222,
      "training_step_time": 6.848257064819336
    },
    {
      "epoch": 1.6425132751464843e-05,
      "model_forward_time": 1.7111783027648926,
      "step": 17223
    },
    {
      "epoch": 1.6425132751464843e-05,
      "step": 17223,
      "training_step_time": 6.821861028671265
    },
    {
      "epoch": 1.642608642578125e-05,
      "model_forward_time": 1.709240198135376,
      "step": 17224
    },
    {
      "epoch": 1.642608642578125e-05,
      "step": 17224,
      "training_step_time": 7.338024139404297
    },
    {
      "epoch": 1.6427040100097658e-05,
      "model_forward_time": 1.7059886455535889,
      "step": 17225
    },
    {
      "epoch": 1.6427040100097658e-05,
      "step": 17225,
      "training_step_time": 7.068591117858887
    },
    {
      "epoch": 1.642799377441406e-05,
      "model_forward_time": 1.710756778717041,
      "step": 17226
    },
    {
      "epoch": 1.642799377441406e-05,
      "step": 17226,
      "training_step_time": 8.48717999458313
    },
    {
      "epoch": 1.642894744873047e-05,
      "model_forward_time": 1.3340198993682861,
      "step": 17227
    },
    {
      "epoch": 1.642894744873047e-05,
      "step": 17227,
      "training_step_time": 9.724231481552124
    },
    {
      "epoch": 1.6429901123046876e-05,
      "model_forward_time": 2.2272210121154785,
      "step": 17228
    },
    {
      "epoch": 1.6429901123046876e-05,
      "step": 17228,
      "training_step_time": 9.662569761276245
    },
    {
      "epoch": 1.643085479736328e-05,
      "model_forward_time": 1.701801061630249,
      "step": 17229
    },
    {
      "epoch": 1.643085479736328e-05,
      "step": 17229,
      "training_step_time": 8.328093767166138
    },
    {
      "epoch": 1.6431808471679688e-05,
      "grad_norm": 0.1485070288181305,
      "learning_rate": 9.59668210324357e-06,
      "loss": 0.0623,
      "step": 17230
    },
    {
      "epoch": 1.6431808471679688e-05,
      "model_forward_time": 1.70749831199646,
      "step": 17230
    },
    {
      "dynamics_loss_avg": 0.07224368266761302,
      "epoch": 1.6431808471679688e-05,
      "step": 17230
    },
    {
      "action_loss_avg": 0.0042190875858068464,
      "epoch": 1.6431808471679688e-05,
      "step": 17230
    },
    {
      "epoch": 1.6431808471679688e-05,
      "step": 17230,
      "training_step_time": 7.73315691947937
    },
    {
      "epoch": 1.6432762145996095e-05,
      "model_forward_time": 1.70945143699646,
      "step": 17231
    },
    {
      "epoch": 1.6432762145996095e-05,
      "step": 17231,
      "training_step_time": 6.915773630142212
    },
    {
      "epoch": 1.64337158203125e-05,
      "model_forward_time": 1.3320376873016357,
      "step": 17232
    },
    {
      "epoch": 1.64337158203125e-05,
      "step": 17232,
      "training_step_time": 6.82124662399292
    },
    {
      "epoch": 1.6434669494628907e-05,
      "model_forward_time": 1.708531379699707,
      "step": 17233
    },
    {
      "epoch": 1.6434669494628907e-05,
      "step": 17233,
      "training_step_time": 6.877574682235718
    },
    {
      "epoch": 1.6435623168945314e-05,
      "model_forward_time": 1.7035057544708252,
      "step": 17234
    },
    {
      "epoch": 1.6435623168945314e-05,
      "step": 17234,
      "training_step_time": 7.05453085899353
    },
    {
      "epoch": 1.6436576843261718e-05,
      "model_forward_time": 1.7141458988189697,
      "step": 17235
    },
    {
      "epoch": 1.6436576843261718e-05,
      "step": 17235,
      "training_step_time": 8.546011686325073
    },
    {
      "epoch": 1.6437530517578126e-05,
      "model_forward_time": 1.708042860031128,
      "step": 17236
    },
    {
      "epoch": 1.6437530517578126e-05,
      "step": 17236,
      "training_step_time": 7.839627027511597
    },
    {
      "epoch": 1.643848419189453e-05,
      "model_forward_time": 2.2706263065338135,
      "step": 17237
    },
    {
      "epoch": 1.643848419189453e-05,
      "step": 17237,
      "training_step_time": 9.937493562698364
    },
    {
      "epoch": 1.6439437866210937e-05,
      "model_forward_time": 1.7053110599517822,
      "step": 17238
    },
    {
      "epoch": 1.6439437866210937e-05,
      "step": 17238,
      "training_step_time": 7.926626205444336
    },
    {
      "epoch": 1.6440391540527344e-05,
      "model_forward_time": 1.7238194942474365,
      "step": 17239
    },
    {
      "epoch": 1.6440391540527344e-05,
      "step": 17239,
      "training_step_time": 7.693300485610962
    },
    {
      "epoch": 1.644134521484375e-05,
      "grad_norm": 0.19547894597053528,
      "learning_rate": 9.596031257140974e-06,
      "loss": 0.0634,
      "step": 17240
    },
    {
      "epoch": 1.644134521484375e-05,
      "model_forward_time": 1.6978754997253418,
      "step": 17240
    },
    {
      "dynamics_loss_avg": 0.059610063582658766,
      "epoch": 1.644134521484375e-05,
      "step": 17240
    },
    {
      "action_loss_avg": 0.0035008281643968075,
      "epoch": 1.644134521484375e-05,
      "step": 17240
    },
    {
      "epoch": 1.644134521484375e-05,
      "step": 17240,
      "training_step_time": 7.630025625228882
    },
    {
      "epoch": 1.6442298889160156e-05,
      "model_forward_time": 1.7081801891326904,
      "step": 17241
    },
    {
      "epoch": 1.6442298889160156e-05,
      "step": 17241,
      "training_step_time": 6.8336029052734375
    },
    {
      "epoch": 1.6443252563476563e-05,
      "model_forward_time": 1.7047395706176758,
      "step": 17242
    },
    {
      "epoch": 1.6443252563476563e-05,
      "step": 17242,
      "training_step_time": 6.825031757354736
    },
    {
      "epoch": 1.6444206237792967e-05,
      "model_forward_time": 1.7208638191223145,
      "step": 17243
    },
    {
      "epoch": 1.6444206237792967e-05,
      "step": 17243,
      "training_step_time": 7.0816826820373535
    },
    {
      "epoch": 1.6445159912109375e-05,
      "model_forward_time": 1.70888352394104,
      "step": 17244
    },
    {
      "epoch": 1.6445159912109375e-05,
      "step": 17244,
      "training_step_time": 7.545669078826904
    },
    {
      "epoch": 1.6446113586425782e-05,
      "model_forward_time": 1.6901421546936035,
      "step": 17245
    },
    {
      "epoch": 1.6446113586425782e-05,
      "step": 17245,
      "training_step_time": 8.706680059432983
    },
    {
      "epoch": 1.6447067260742186e-05,
      "model_forward_time": 1.7526376247406006,
      "step": 17246
    },
    {
      "epoch": 1.6447067260742186e-05,
      "step": 17246,
      "training_step_time": 9.859916925430298
    },
    {
      "epoch": 1.6448020935058594e-05,
      "model_forward_time": 1.7053391933441162,
      "step": 17247
    },
    {
      "epoch": 1.6448020935058594e-05,
      "step": 17247,
      "training_step_time": 8.467895269393921
    },
    {
      "epoch": 1.6448974609375e-05,
      "model_forward_time": 1.718961477279663,
      "step": 17248
    },
    {
      "epoch": 1.6448974609375e-05,
      "step": 17248,
      "training_step_time": 7.714766025543213
    },
    {
      "epoch": 1.6449928283691405e-05,
      "model_forward_time": 1.7073922157287598,
      "step": 17249
    },
    {
      "epoch": 1.6449928283691405e-05,
      "step": 17249,
      "training_step_time": 7.612915515899658
    },
    {
      "epoch": 1.6450881958007812e-05,
      "grad_norm": 0.36679625511169434,
      "learning_rate": 9.59537990842331e-06,
      "loss": 0.0605,
      "step": 17250
    },
    {
      "epoch": 1.6450881958007812e-05,
      "model_forward_time": 1.7015600204467773,
      "step": 17250
    },
    {
      "dynamics_loss_avg": 0.05145498514175415,
      "epoch": 1.6450881958007812e-05,
      "step": 17250
    },
    {
      "action_loss_avg": 0.003101432402036153,
      "epoch": 1.6450881958007812e-05,
      "step": 17250
    },
    {
      "epoch": 1.6450881958007812e-05,
      "step": 17250,
      "training_step_time": 6.861562967300415
    },
    {
      "epoch": 1.645183563232422e-05,
      "model_forward_time": 1.7188737392425537,
      "step": 17251
    },
    {
      "epoch": 1.645183563232422e-05,
      "step": 17251,
      "training_step_time": 6.807052135467529
    },
    {
      "epoch": 1.6452789306640624e-05,
      "model_forward_time": 1.7118558883666992,
      "step": 17252
    },
    {
      "epoch": 1.6452789306640624e-05,
      "step": 17252,
      "training_step_time": 7.133419036865234
    },
    {
      "epoch": 1.645374298095703e-05,
      "model_forward_time": 1.7079224586486816,
      "step": 17253
    },
    {
      "epoch": 1.645374298095703e-05,
      "step": 17253,
      "training_step_time": 7.640093564987183
    },
    {
      "epoch": 1.645469665527344e-05,
      "model_forward_time": 1.704345464706421,
      "step": 17254
    },
    {
      "epoch": 1.645469665527344e-05,
      "step": 17254,
      "training_step_time": 8.806249141693115
    },
    {
      "epoch": 1.6455650329589843e-05,
      "model_forward_time": 1.7528955936431885,
      "step": 17255
    },
    {
      "epoch": 1.6455650329589843e-05,
      "step": 17255,
      "training_step_time": 8.59184980392456
    },
    {
      "epoch": 1.645660400390625e-05,
      "model_forward_time": 2.2101645469665527,
      "step": 17256
    },
    {
      "epoch": 1.645660400390625e-05,
      "step": 17256,
      "training_step_time": 7.942250967025757
    },
    {
      "epoch": 1.6457557678222658e-05,
      "model_forward_time": 1.729032039642334,
      "step": 17257
    },
    {
      "epoch": 1.6457557678222658e-05,
      "step": 17257,
      "training_step_time": 7.780736923217773
    },
    {
      "epoch": 1.645851135253906e-05,
      "model_forward_time": 1.3246734142303467,
      "step": 17258
    },
    {
      "epoch": 1.645851135253906e-05,
      "step": 17258,
      "training_step_time": 7.6641058921813965
    },
    {
      "epoch": 1.645946502685547e-05,
      "model_forward_time": 1.7042086124420166,
      "step": 17259
    },
    {
      "epoch": 1.645946502685547e-05,
      "step": 17259,
      "training_step_time": 6.888990163803101
    },
    {
      "epoch": 1.6460418701171876e-05,
      "grad_norm": 0.15220172703266144,
      "learning_rate": 9.594728057161806e-06,
      "loss": 0.0678,
      "step": 17260
    },
    {
      "epoch": 1.6460418701171876e-05,
      "model_forward_time": 1.7009572982788086,
      "step": 17260
    },
    {
      "dynamics_loss_avg": 0.08304677288979292,
      "epoch": 1.6460418701171876e-05,
      "step": 17260
    },
    {
      "action_loss_avg": 0.0032095737755298615,
      "epoch": 1.6460418701171876e-05,
      "step": 17260
    },
    {
      "epoch": 1.6460418701171876e-05,
      "step": 17260,
      "training_step_time": 6.881730318069458
    },
    {
      "epoch": 1.646137237548828e-05,
      "model_forward_time": 1.7196130752563477,
      "step": 17261
    },
    {
      "epoch": 1.646137237548828e-05,
      "step": 17261,
      "training_step_time": 7.118670225143433
    },
    {
      "epoch": 1.6462326049804688e-05,
      "model_forward_time": 1.7076334953308105,
      "step": 17262
    },
    {
      "epoch": 1.6462326049804688e-05,
      "step": 17262,
      "training_step_time": 8.625083446502686
    },
    {
      "epoch": 1.6463279724121095e-05,
      "model_forward_time": 1.707401990890503,
      "step": 17263
    },
    {
      "epoch": 1.6463279724121095e-05,
      "step": 17263,
      "training_step_time": 8.439656257629395
    },
    {
      "epoch": 1.64642333984375e-05,
      "model_forward_time": 1.7469983100891113,
      "step": 17264
    },
    {
      "epoch": 1.64642333984375e-05,
      "step": 17264,
      "training_step_time": 7.779442071914673
    },
    {
      "epoch": 1.6465187072753907e-05,
      "model_forward_time": 2.226766347885132,
      "step": 17265
    },
    {
      "epoch": 1.6465187072753907e-05,
      "step": 17265,
      "training_step_time": 7.955859184265137
    },
    {
      "epoch": 1.646614074707031e-05,
      "model_forward_time": 1.7130565643310547,
      "step": 17266
    },
    {
      "epoch": 1.646614074707031e-05,
      "step": 17266,
      "training_step_time": 7.759452819824219
    },
    {
      "epoch": 1.6467094421386718e-05,
      "model_forward_time": 1.7074596881866455,
      "step": 17267
    },
    {
      "epoch": 1.6467094421386718e-05,
      "step": 17267,
      "training_step_time": 7.6851654052734375
    },
    {
      "epoch": 1.6468048095703126e-05,
      "model_forward_time": 1.7018940448760986,
      "step": 17268
    },
    {
      "epoch": 1.6468048095703126e-05,
      "step": 17268,
      "training_step_time": 6.895583868026733
    },
    {
      "epoch": 1.646900177001953e-05,
      "model_forward_time": 1.706312894821167,
      "step": 17269
    },
    {
      "epoch": 1.646900177001953e-05,
      "step": 17269,
      "training_step_time": 6.8496928215026855
    },
    {
      "epoch": 1.6469955444335937e-05,
      "grad_norm": 0.1723267138004303,
      "learning_rate": 9.594075703427752e-06,
      "loss": 0.0662,
      "step": 17270
    },
    {
      "epoch": 1.6469955444335937e-05,
      "model_forward_time": 1.710127830505371,
      "step": 17270
    },
    {
      "dynamics_loss_avg": 0.06485535446554422,
      "epoch": 1.6469955444335937e-05,
      "step": 17270
    },
    {
      "action_loss_avg": 0.003953034162987023,
      "epoch": 1.6469955444335937e-05,
      "step": 17270
    },
    {
      "epoch": 1.6469955444335937e-05,
      "step": 17270,
      "training_step_time": 7.102518081665039
    },
    {
      "epoch": 1.6470909118652345e-05,
      "model_forward_time": 1.7087483406066895,
      "step": 17271
    },
    {
      "epoch": 1.6470909118652345e-05,
      "step": 17271,
      "training_step_time": 7.489243984222412
    },
    {
      "epoch": 1.647186279296875e-05,
      "model_forward_time": 1.7106726169586182,
      "step": 17272
    },
    {
      "epoch": 1.647186279296875e-05,
      "step": 17272,
      "training_step_time": 8.203619480133057
    },
    {
      "epoch": 1.6472816467285156e-05,
      "model_forward_time": 1.7336030006408691,
      "step": 17273
    },
    {
      "epoch": 1.6472816467285156e-05,
      "step": 17273,
      "training_step_time": 8.684291362762451
    },
    {
      "epoch": 1.6473770141601563e-05,
      "model_forward_time": 2.2517623901367188,
      "step": 17274
    },
    {
      "epoch": 1.6473770141601563e-05,
      "step": 17274,
      "training_step_time": 10.271838188171387
    },
    {
      "epoch": 1.6474723815917967e-05,
      "model_forward_time": 1.7224311828613281,
      "step": 17275
    },
    {
      "epoch": 1.6474723815917967e-05,
      "step": 17275,
      "training_step_time": 8.342644453048706
    },
    {
      "epoch": 1.6475677490234375e-05,
      "model_forward_time": 1.7117526531219482,
      "step": 17276
    },
    {
      "epoch": 1.6475677490234375e-05,
      "step": 17276,
      "training_step_time": 7.661616802215576
    },
    {
      "epoch": 1.6476631164550782e-05,
      "model_forward_time": 1.7193658351898193,
      "step": 17277
    },
    {
      "epoch": 1.6476631164550782e-05,
      "step": 17277,
      "training_step_time": 6.745286703109741
    },
    {
      "epoch": 1.6477584838867186e-05,
      "model_forward_time": 1.7204387187957764,
      "step": 17278
    },
    {
      "epoch": 1.6477584838867186e-05,
      "step": 17278,
      "training_step_time": 6.717840194702148
    },
    {
      "epoch": 1.6478538513183594e-05,
      "model_forward_time": 1.7190649509429932,
      "step": 17279
    },
    {
      "epoch": 1.6478538513183594e-05,
      "step": 17279,
      "training_step_time": 6.7233617305755615
    },
    {
      "epoch": 1.64794921875e-05,
      "grad_norm": 0.17840567231178284,
      "learning_rate": 9.593422847292485e-06,
      "loss": 0.0615,
      "step": 17280
    },
    {
      "epoch": 1.64794921875e-05,
      "model_forward_time": 1.704967975616455,
      "step": 17280
    },
    {
      "dynamics_loss_avg": 0.062418611533939836,
      "epoch": 1.64794921875e-05,
      "step": 17280
    },
    {
      "action_loss_avg": 0.003433084033895284,
      "epoch": 1.64794921875e-05,
      "step": 17280
    },
    {
      "epoch": 1.64794921875e-05,
      "step": 17280,
      "training_step_time": 7.157994747161865
    },
    {
      "epoch": 1.6480445861816405e-05,
      "model_forward_time": 1.7128548622131348,
      "step": 17281
    },
    {
      "epoch": 1.6480445861816405e-05,
      "step": 17281,
      "training_step_time": 8.241201877593994
    },
    {
      "epoch": 1.6481399536132813e-05,
      "model_forward_time": 1.7231662273406982,
      "step": 17282
    },
    {
      "epoch": 1.6481399536132813e-05,
      "step": 17282,
      "training_step_time": 8.266187906265259
    },
    {
      "epoch": 1.648235321044922e-05,
      "model_forward_time": 2.2898685932159424,
      "step": 17283
    },
    {
      "epoch": 1.648235321044922e-05,
      "step": 17283,
      "training_step_time": 8.910268306732178
    },
    {
      "epoch": 1.6483306884765624e-05,
      "model_forward_time": 1.721217393875122,
      "step": 17284
    },
    {
      "epoch": 1.6483306884765624e-05,
      "step": 17284,
      "training_step_time": 8.5064857006073
    },
    {
      "epoch": 1.648426055908203e-05,
      "model_forward_time": 1.7140588760375977,
      "step": 17285
    },
    {
      "epoch": 1.648426055908203e-05,
      "step": 17285,
      "training_step_time": 7.553354740142822
    },
    {
      "epoch": 1.648521423339844e-05,
      "model_forward_time": 1.716043472290039,
      "step": 17286
    },
    {
      "epoch": 1.648521423339844e-05,
      "step": 17286,
      "training_step_time": 6.766399145126343
    },
    {
      "epoch": 1.6486167907714843e-05,
      "model_forward_time": 1.7207460403442383,
      "step": 17287
    },
    {
      "epoch": 1.6486167907714843e-05,
      "step": 17287,
      "training_step_time": 6.811357736587524
    },
    {
      "epoch": 1.648712158203125e-05,
      "model_forward_time": 1.7168259620666504,
      "step": 17288
    },
    {
      "epoch": 1.648712158203125e-05,
      "step": 17288,
      "training_step_time": 6.783578872680664
    },
    {
      "epoch": 1.6488075256347658e-05,
      "model_forward_time": 1.7115788459777832,
      "step": 17289
    },
    {
      "epoch": 1.6488075256347658e-05,
      "step": 17289,
      "training_step_time": 7.200684547424316
    },
    {
      "epoch": 1.648902893066406e-05,
      "grad_norm": 0.14354532957077026,
      "learning_rate": 9.592769488827402e-06,
      "loss": 0.0626,
      "step": 17290
    },
    {
      "epoch": 1.648902893066406e-05,
      "model_forward_time": 1.7098369598388672,
      "step": 17290
    },
    {
      "dynamics_loss_avg": 0.06234574019908905,
      "epoch": 1.648902893066406e-05,
      "step": 17290
    },
    {
      "action_loss_avg": 0.002537709631724283,
      "epoch": 1.648902893066406e-05,
      "step": 17290
    },
    {
      "epoch": 1.648902893066406e-05,
      "step": 17290,
      "training_step_time": 7.543851852416992
    },
    {
      "epoch": 1.648998260498047e-05,
      "model_forward_time": 1.7152338027954102,
      "step": 17291
    },
    {
      "epoch": 1.648998260498047e-05,
      "step": 17291,
      "training_step_time": 7.567143201828003
    },
    {
      "epoch": 1.6490936279296877e-05,
      "model_forward_time": 2.2885329723358154,
      "step": 17292
    },
    {
      "epoch": 1.6490936279296877e-05,
      "step": 17292,
      "training_step_time": 8.901435613632202
    },
    {
      "epoch": 1.649188995361328e-05,
      "model_forward_time": 1.7105236053466797,
      "step": 17293
    },
    {
      "epoch": 1.649188995361328e-05,
      "step": 17293,
      "training_step_time": 10.660988092422485
    },
    {
      "epoch": 1.6492843627929688e-05,
      "model_forward_time": 1.7108793258666992,
      "step": 17294
    },
    {
      "epoch": 1.6492843627929688e-05,
      "step": 17294,
      "training_step_time": 7.567890405654907
    },
    {
      "epoch": 1.6493797302246095e-05,
      "model_forward_time": 1.7150077819824219,
      "step": 17295
    },
    {
      "epoch": 1.6493797302246095e-05,
      "step": 17295,
      "training_step_time": 7.551030874252319
    },
    {
      "epoch": 1.64947509765625e-05,
      "model_forward_time": 1.7210743427276611,
      "step": 17296
    },
    {
      "epoch": 1.64947509765625e-05,
      "step": 17296,
      "training_step_time": 6.734912872314453
    },
    {
      "epoch": 1.6495704650878907e-05,
      "model_forward_time": 1.720024824142456,
      "step": 17297
    },
    {
      "epoch": 1.6495704650878907e-05,
      "step": 17297,
      "training_step_time": 6.7188026905059814
    },
    {
      "epoch": 1.649665832519531e-05,
      "model_forward_time": 1.7135870456695557,
      "step": 17298
    },
    {
      "epoch": 1.649665832519531e-05,
      "step": 17298,
      "training_step_time": 7.9069132804870605
    },
    {
      "epoch": 1.6497611999511718e-05,
      "model_forward_time": 1.713141679763794,
      "step": 17299
    },
    {
      "epoch": 1.6497611999511718e-05,
      "step": 17299,
      "training_step_time": 7.526015520095825
    },
    {
      "epoch": 1.6498565673828126e-05,
      "grad_norm": 0.15334828197956085,
      "learning_rate": 9.592115628103953e-06,
      "loss": 0.0635,
      "step": 17300
    },
    {
      "epoch": 1.6498565673828126e-05,
      "model_forward_time": 1.71897554397583,
      "step": 17300
    },
    {
      "dynamics_loss_avg": 0.06517502348870038,
      "epoch": 1.6498565673828126e-05,
      "step": 17300
    },
    {
      "action_loss_avg": 0.003428985376376659,
      "epoch": 1.6498565673828126e-05,
      "step": 17300
    },
    {
      "epoch": 1.6498565673828126e-05,
      "step": 17300,
      "training_step_time": 7.572331190109253
    },
    {
      "epoch": 1.649951934814453e-05,
      "model_forward_time": 1.7356278896331787,
      "step": 17301
    },
    {
      "epoch": 1.649951934814453e-05,
      "step": 17301,
      "training_step_time": 8.559884309768677
    },
    {
      "epoch": 1.6500473022460937e-05,
      "model_forward_time": 2.2538869380950928,
      "step": 17302
    },
    {
      "epoch": 1.6500473022460937e-05,
      "step": 17302,
      "training_step_time": 8.482519388198853
    },
    {
      "epoch": 1.6501426696777345e-05,
      "model_forward_time": 1.7065045833587646,
      "step": 17303
    },
    {
      "epoch": 1.6501426696777345e-05,
      "step": 17303,
      "training_step_time": 7.712882041931152
    },
    {
      "epoch": 1.650238037109375e-05,
      "model_forward_time": 1.7131927013397217,
      "step": 17304
    },
    {
      "epoch": 1.650238037109375e-05,
      "step": 17304,
      "training_step_time": 7.630154609680176
    },
    {
      "epoch": 1.6503334045410156e-05,
      "model_forward_time": 1.3323853015899658,
      "step": 17305
    },
    {
      "epoch": 1.6503334045410156e-05,
      "step": 17305,
      "training_step_time": 6.73494553565979
    },
    {
      "epoch": 1.6504287719726563e-05,
      "model_forward_time": 1.7181265354156494,
      "step": 17306
    },
    {
      "epoch": 1.6504287719726563e-05,
      "step": 17306,
      "training_step_time": 6.75773549079895
    },
    {
      "epoch": 1.6505241394042967e-05,
      "model_forward_time": 1.7116084098815918,
      "step": 17307
    },
    {
      "epoch": 1.6505241394042967e-05,
      "step": 17307,
      "training_step_time": 7.084938287734985
    },
    {
      "epoch": 1.6506195068359375e-05,
      "model_forward_time": 1.7124130725860596,
      "step": 17308
    },
    {
      "epoch": 1.6506195068359375e-05,
      "step": 17308,
      "training_step_time": 8.386530876159668
    },
    {
      "epoch": 1.6507148742675782e-05,
      "model_forward_time": 1.713918685913086,
      "step": 17309
    },
    {
      "epoch": 1.6507148742675782e-05,
      "step": 17309,
      "training_step_time": 8.38584566116333
    },
    {
      "epoch": 1.6508102416992186e-05,
      "grad_norm": 0.19545793533325195,
      "learning_rate": 9.591461265193644e-06,
      "loss": 0.0611,
      "step": 17310
    },
    {
      "epoch": 1.6508102416992186e-05,
      "model_forward_time": 1.738037109375,
      "step": 17310
    },
    {
      "dynamics_loss_avg": 0.07007203213870525,
      "epoch": 1.6508102416992186e-05,
      "step": 17310
    },
    {
      "action_loss_avg": 0.0029337068408494815,
      "epoch": 1.6508102416992186e-05,
      "step": 17310
    },
    {
      "epoch": 1.6508102416992186e-05,
      "step": 17310,
      "training_step_time": 8.524025201797485
    },
    {
      "epoch": 1.6509056091308594e-05,
      "model_forward_time": 2.2595510482788086,
      "step": 17311
    },
    {
      "epoch": 1.6509056091308594e-05,
      "step": 17311,
      "training_step_time": 8.727954864501953
    },
    {
      "epoch": 1.6510009765625e-05,
      "model_forward_time": 1.7049822807312012,
      "step": 17312
    },
    {
      "epoch": 1.6510009765625e-05,
      "step": 17312,
      "training_step_time": 8.287027835845947
    },
    {
      "epoch": 1.6510963439941405e-05,
      "model_forward_time": 1.7081139087677002,
      "step": 17313
    },
    {
      "epoch": 1.6510963439941405e-05,
      "step": 17313,
      "training_step_time": 7.694410800933838
    },
    {
      "epoch": 1.6511917114257813e-05,
      "model_forward_time": 1.7074930667877197,
      "step": 17314
    },
    {
      "epoch": 1.6511917114257813e-05,
      "step": 17314,
      "training_step_time": 6.793163537979126
    },
    {
      "epoch": 1.651287078857422e-05,
      "model_forward_time": 1.714416265487671,
      "step": 17315
    },
    {
      "epoch": 1.651287078857422e-05,
      "step": 17315,
      "training_step_time": 6.812603950500488
    },
    {
      "epoch": 1.6513824462890624e-05,
      "model_forward_time": 1.7135009765625,
      "step": 17316
    },
    {
      "epoch": 1.6513824462890624e-05,
      "step": 17316,
      "training_step_time": 7.08711576461792
    },
    {
      "epoch": 1.651477813720703e-05,
      "model_forward_time": 1.7060651779174805,
      "step": 17317
    },
    {
      "epoch": 1.651477813720703e-05,
      "step": 17317,
      "training_step_time": 7.647562026977539
    },
    {
      "epoch": 1.651573181152344e-05,
      "model_forward_time": 1.7126564979553223,
      "step": 17318
    },
    {
      "epoch": 1.651573181152344e-05,
      "step": 17318,
      "training_step_time": 7.5613508224487305
    },
    {
      "epoch": 1.6516685485839843e-05,
      "model_forward_time": 1.7010459899902344,
      "step": 17319
    },
    {
      "epoch": 1.6516685485839843e-05,
      "step": 17319,
      "training_step_time": 7.657930850982666
    },
    {
      "epoch": 1.651763916015625e-05,
      "grad_norm": 0.21656771004199982,
      "learning_rate": 9.590806400168032e-06,
      "loss": 0.0579,
      "step": 17320
    },
    {
      "epoch": 1.651763916015625e-05,
      "model_forward_time": 2.2562639713287354,
      "step": 17320
    },
    {
      "dynamics_loss_avg": 0.056386591866612434,
      "epoch": 1.651763916015625e-05,
      "step": 17320
    },
    {
      "action_loss_avg": 0.003695118229370564,
      "epoch": 1.651763916015625e-05,
      "step": 17320
    },
    {
      "epoch": 1.651763916015625e-05,
      "step": 17320,
      "training_step_time": 8.367704391479492
    },
    {
      "epoch": 1.6518592834472658e-05,
      "model_forward_time": 1.7044315338134766,
      "step": 17321
    },
    {
      "epoch": 1.6518592834472658e-05,
      "step": 17321,
      "training_step_time": 8.372792720794678
    },
    {
      "epoch": 1.6519546508789062e-05,
      "model_forward_time": 1.709522008895874,
      "step": 17322
    },
    {
      "epoch": 1.6519546508789062e-05,
      "step": 17322,
      "training_step_time": 7.6111931800842285
    },
    {
      "epoch": 1.652050018310547e-05,
      "model_forward_time": 1.7097055912017822,
      "step": 17323
    },
    {
      "epoch": 1.652050018310547e-05,
      "step": 17323,
      "training_step_time": 7.527598142623901
    },
    {
      "epoch": 1.6521453857421877e-05,
      "model_forward_time": 1.7165718078613281,
      "step": 17324
    },
    {
      "epoch": 1.6521453857421877e-05,
      "step": 17324,
      "training_step_time": 6.727803945541382
    },
    {
      "epoch": 1.652240753173828e-05,
      "model_forward_time": 1.7116682529449463,
      "step": 17325
    },
    {
      "epoch": 1.652240753173828e-05,
      "step": 17325,
      "training_step_time": 6.678385019302368
    },
    {
      "epoch": 1.6523361206054688e-05,
      "model_forward_time": 1.7111914157867432,
      "step": 17326
    },
    {
      "epoch": 1.6523361206054688e-05,
      "step": 17326,
      "training_step_time": 7.511722564697266
    },
    {
      "epoch": 1.6524314880371095e-05,
      "model_forward_time": 1.7144935131072998,
      "step": 17327
    },
    {
      "epoch": 1.6524314880371095e-05,
      "step": 17327,
      "training_step_time": 7.4588847160339355
    },
    {
      "epoch": 1.65252685546875e-05,
      "model_forward_time": 1.717827558517456,
      "step": 17328
    },
    {
      "epoch": 1.65252685546875e-05,
      "step": 17328,
      "training_step_time": 9.323884963989258
    },
    {
      "epoch": 1.6526222229003907e-05,
      "model_forward_time": 2.276993751525879,
      "step": 17329
    },
    {
      "epoch": 1.6526222229003907e-05,
      "step": 17329,
      "training_step_time": 8.469470262527466
    },
    {
      "epoch": 1.652717590332031e-05,
      "grad_norm": 0.20280523598194122,
      "learning_rate": 9.590151033098735e-06,
      "loss": 0.0612,
      "step": 17330
    },
    {
      "epoch": 1.652717590332031e-05,
      "model_forward_time": 1.6997087001800537,
      "step": 17330
    },
    {
      "dynamics_loss_avg": 0.059575018100440504,
      "epoch": 1.652717590332031e-05,
      "step": 17330
    },
    {
      "action_loss_avg": 0.0031635956140235065,
      "epoch": 1.652717590332031e-05,
      "step": 17330
    },
    {
      "epoch": 1.652717590332031e-05,
      "step": 17330,
      "training_step_time": 7.664262294769287
    },
    {
      "epoch": 1.6528129577636718e-05,
      "model_forward_time": 1.7099308967590332,
      "step": 17331
    },
    {
      "epoch": 1.6528129577636718e-05,
      "step": 17331,
      "training_step_time": 7.606426239013672
    },
    {
      "epoch": 1.6529083251953126e-05,
      "model_forward_time": 1.7172389030456543,
      "step": 17332
    },
    {
      "epoch": 1.6529083251953126e-05,
      "step": 17332,
      "training_step_time": 7.547475814819336
    },
    {
      "epoch": 1.653003692626953e-05,
      "model_forward_time": 1.7192656993865967,
      "step": 17333
    },
    {
      "epoch": 1.653003692626953e-05,
      "step": 17333,
      "training_step_time": 6.732764005661011
    },
    {
      "epoch": 1.6530990600585937e-05,
      "model_forward_time": 1.715315818786621,
      "step": 17334
    },
    {
      "epoch": 1.6530990600585937e-05,
      "step": 17334,
      "training_step_time": 6.69491720199585
    },
    {
      "epoch": 1.6531944274902345e-05,
      "model_forward_time": 1.7135462760925293,
      "step": 17335
    },
    {
      "epoch": 1.6531944274902345e-05,
      "step": 17335,
      "training_step_time": 7.5561676025390625
    },
    {
      "epoch": 1.653289794921875e-05,
      "model_forward_time": 1.714195728302002,
      "step": 17336
    },
    {
      "epoch": 1.653289794921875e-05,
      "step": 17336,
      "training_step_time": 8.015002250671387
    },
    {
      "epoch": 1.6533851623535156e-05,
      "model_forward_time": 1.7137768268585205,
      "step": 17337
    },
    {
      "epoch": 1.6533851623535156e-05,
      "step": 17337,
      "training_step_time": 7.642192602157593
    },
    {
      "epoch": 1.6534805297851563e-05,
      "model_forward_time": 2.263282299041748,
      "step": 17338
    },
    {
      "epoch": 1.6534805297851563e-05,
      "step": 17338,
      "training_step_time": 8.751960754394531
    },
    {
      "epoch": 1.6535758972167967e-05,
      "model_forward_time": 1.7141246795654297,
      "step": 17339
    },
    {
      "epoch": 1.6535758972167967e-05,
      "step": 17339,
      "training_step_time": 7.690065622329712
    },
    {
      "epoch": 1.6536712646484375e-05,
      "grad_norm": 0.2509610652923584,
      "learning_rate": 9.589495164057422e-06,
      "loss": 0.0646,
      "step": 17340
    },
    {
      "epoch": 1.6536712646484375e-05,
      "model_forward_time": 1.707981824874878,
      "step": 17340
    },
    {
      "dynamics_loss_avg": 0.05257203970104456,
      "epoch": 1.6536712646484375e-05,
      "step": 17340
    },
    {
      "action_loss_avg": 0.0036700342316180468,
      "epoch": 1.6536712646484375e-05,
      "step": 17340
    },
    {
      "epoch": 1.6536712646484375e-05,
      "step": 17340,
      "training_step_time": 7.67732834815979
    },
    {
      "epoch": 1.6537666320800782e-05,
      "model_forward_time": 1.7153236865997314,
      "step": 17341
    },
    {
      "epoch": 1.6537666320800782e-05,
      "step": 17341,
      "training_step_time": 7.569783926010132
    },
    {
      "epoch": 1.6538619995117186e-05,
      "model_forward_time": 1.7107503414154053,
      "step": 17342
    },
    {
      "epoch": 1.6538619995117186e-05,
      "step": 17342,
      "training_step_time": 6.7121570110321045
    },
    {
      "epoch": 1.6539573669433594e-05,
      "model_forward_time": 1.7176845073699951,
      "step": 17343
    },
    {
      "epoch": 1.6539573669433594e-05,
      "step": 17343,
      "training_step_time": 6.700493812561035
    },
    {
      "epoch": 1.654052734375e-05,
      "model_forward_time": 1.69614577293396,
      "step": 17344
    },
    {
      "epoch": 1.654052734375e-05,
      "step": 17344,
      "training_step_time": 7.515902280807495
    },
    {
      "epoch": 1.6541481018066405e-05,
      "model_forward_time": 1.6934430599212646,
      "step": 17345
    },
    {
      "epoch": 1.6541481018066405e-05,
      "step": 17345,
      "training_step_time": 8.175864219665527
    },
    {
      "epoch": 1.6542434692382813e-05,
      "model_forward_time": 1.70747709274292,
      "step": 17346
    },
    {
      "epoch": 1.6542434692382813e-05,
      "step": 17346,
      "training_step_time": 9.142718076705933
    },
    {
      "epoch": 1.654338836669922e-05,
      "model_forward_time": 1.7216215133666992,
      "step": 17347
    },
    {
      "epoch": 1.654338836669922e-05,
      "step": 17347,
      "training_step_time": 7.734497547149658
    },
    {
      "epoch": 1.6544342041015624e-05,
      "model_forward_time": 2.2236292362213135,
      "step": 17348
    },
    {
      "epoch": 1.6544342041015624e-05,
      "step": 17348,
      "training_step_time": 7.835063457489014
    },
    {
      "epoch": 1.654529571533203e-05,
      "model_forward_time": 1.703505516052246,
      "step": 17349
    },
    {
      "epoch": 1.654529571533203e-05,
      "step": 17349,
      "training_step_time": 7.686046838760376
    },
    {
      "epoch": 1.654624938964844e-05,
      "grad_norm": 0.20525914430618286,
      "learning_rate": 9.58883879311582e-06,
      "loss": 0.0677,
      "step": 17350
    },
    {
      "epoch": 1.654624938964844e-05,
      "model_forward_time": 1.699235439300537,
      "step": 17350
    },
    {
      "dynamics_loss_avg": 0.05167837627232075,
      "epoch": 1.654624938964844e-05,
      "step": 17350
    },
    {
      "action_loss_avg": 0.0024982402625028043,
      "epoch": 1.654624938964844e-05,
      "step": 17350
    },
    {
      "epoch": 1.654624938964844e-05,
      "step": 17350,
      "training_step_time": 7.619154691696167
    },
    {
      "epoch": 1.6547203063964843e-05,
      "model_forward_time": 1.7048006057739258,
      "step": 17351
    },
    {
      "epoch": 1.6547203063964843e-05,
      "step": 17351,
      "training_step_time": 6.680999994277954
    },
    {
      "epoch": 1.654815673828125e-05,
      "model_forward_time": 1.7130706310272217,
      "step": 17352
    },
    {
      "epoch": 1.654815673828125e-05,
      "step": 17352,
      "training_step_time": 6.697343111038208
    },
    {
      "epoch": 1.6549110412597658e-05,
      "model_forward_time": 1.743962287902832,
      "step": 17353
    },
    {
      "epoch": 1.6549110412597658e-05,
      "step": 17353,
      "training_step_time": 8.412315607070923
    },
    {
      "epoch": 1.6550064086914062e-05,
      "model_forward_time": 0.98384690284729,
      "step": 17354
    },
    {
      "epoch": 1.6550064086914062e-05,
      "step": 17354,
      "training_step_time": 7.605898141860962
    },
    {
      "epoch": 1.655101776123047e-05,
      "model_forward_time": 1.7166633605957031,
      "step": 17355
    },
    {
      "epoch": 1.655101776123047e-05,
      "step": 17355,
      "training_step_time": 8.554267644882202
    },
    {
      "epoch": 1.6551971435546877e-05,
      "model_forward_time": 1.709913730621338,
      "step": 17356
    },
    {
      "epoch": 1.6551971435546877e-05,
      "step": 17356,
      "training_step_time": 7.730896949768066
    },
    {
      "epoch": 1.655292510986328e-05,
      "model_forward_time": 2.2713634967803955,
      "step": 17357
    },
    {
      "epoch": 1.655292510986328e-05,
      "step": 17357,
      "training_step_time": 8.838726043701172
    },
    {
      "epoch": 1.6553878784179688e-05,
      "model_forward_time": 1.7028515338897705,
      "step": 17358
    },
    {
      "epoch": 1.6553878784179688e-05,
      "step": 17358,
      "training_step_time": 7.6593828201293945
    },
    {
      "epoch": 1.6554832458496095e-05,
      "model_forward_time": 1.7130472660064697,
      "step": 17359
    },
    {
      "epoch": 1.6554832458496095e-05,
      "step": 17359,
      "training_step_time": 7.613988161087036
    },
    {
      "epoch": 1.65557861328125e-05,
      "grad_norm": 0.1444741189479828,
      "learning_rate": 9.588181920345704e-06,
      "loss": 0.0682,
      "step": 17360
    },
    {
      "epoch": 1.65557861328125e-05,
      "model_forward_time": 1.7047908306121826,
      "step": 17360
    },
    {
      "dynamics_loss_avg": 0.060641185380518436,
      "epoch": 1.65557861328125e-05,
      "step": 17360
    },
    {
      "action_loss_avg": 0.003404729481553659,
      "epoch": 1.65557861328125e-05,
      "step": 17360
    },
    {
      "epoch": 1.65557861328125e-05,
      "step": 17360,
      "training_step_time": 7.583129167556763
    },
    {
      "epoch": 1.6556739807128907e-05,
      "model_forward_time": 1.709223985671997,
      "step": 17361
    },
    {
      "epoch": 1.6556739807128907e-05,
      "step": 17361,
      "training_step_time": 6.72196364402771
    },
    {
      "epoch": 1.655769348144531e-05,
      "model_forward_time": 1.707932949066162,
      "step": 17362
    },
    {
      "epoch": 1.655769348144531e-05,
      "step": 17362,
      "training_step_time": 7.54053258895874
    },
    {
      "epoch": 1.655864715576172e-05,
      "model_forward_time": 1.698385238647461,
      "step": 17363
    },
    {
      "epoch": 1.655864715576172e-05,
      "step": 17363,
      "training_step_time": 8.131586074829102
    },
    {
      "epoch": 1.6559600830078126e-05,
      "model_forward_time": 1.7074096202850342,
      "step": 17364
    },
    {
      "epoch": 1.6559600830078126e-05,
      "step": 17364,
      "training_step_time": 8.594608783721924
    },
    {
      "epoch": 1.656055450439453e-05,
      "model_forward_time": 1.7121686935424805,
      "step": 17365
    },
    {
      "epoch": 1.656055450439453e-05,
      "step": 17365,
      "training_step_time": 9.1995849609375
    },
    {
      "epoch": 1.6561508178710937e-05,
      "model_forward_time": 2.2786052227020264,
      "step": 17366
    },
    {
      "epoch": 1.6561508178710937e-05,
      "step": 17366,
      "training_step_time": 7.863713026046753
    },
    {
      "epoch": 1.6562461853027345e-05,
      "model_forward_time": 1.705622911453247,
      "step": 17367
    },
    {
      "epoch": 1.6562461853027345e-05,
      "step": 17367,
      "training_step_time": 9.7713942527771
    },
    {
      "epoch": 1.656341552734375e-05,
      "model_forward_time": 1.7161195278167725,
      "step": 17368
    },
    {
      "epoch": 1.656341552734375e-05,
      "step": 17368,
      "training_step_time": 7.579140663146973
    },
    {
      "epoch": 1.6564369201660156e-05,
      "model_forward_time": 1.7145919799804688,
      "step": 17369
    },
    {
      "epoch": 1.6564369201660156e-05,
      "step": 17369,
      "training_step_time": 7.600529909133911
    },
    {
      "epoch": 1.6565322875976563e-05,
      "grad_norm": 0.175926074385643,
      "learning_rate": 9.587524545818914e-06,
      "loss": 0.0684,
      "step": 17370
    },
    {
      "epoch": 1.6565322875976563e-05,
      "model_forward_time": 1.7053046226501465,
      "step": 17370
    },
    {
      "dynamics_loss_avg": 0.06672650650143623,
      "epoch": 1.6565322875976563e-05,
      "step": 17370
    },
    {
      "action_loss_avg": 0.0023107179324142633,
      "epoch": 1.6565322875976563e-05,
      "step": 17370
    },
    {
      "epoch": 1.6565322875976563e-05,
      "step": 17370,
      "training_step_time": 6.7196080684661865
    },
    {
      "epoch": 1.6566276550292967e-05,
      "model_forward_time": 1.7056701183319092,
      "step": 17371
    },
    {
      "epoch": 1.6566276550292967e-05,
      "step": 17371,
      "training_step_time": 7.590734958648682
    },
    {
      "epoch": 1.6567230224609375e-05,
      "model_forward_time": 1.7135310173034668,
      "step": 17372
    },
    {
      "epoch": 1.6567230224609375e-05,
      "step": 17372,
      "training_step_time": 8.294893026351929
    },
    {
      "epoch": 1.6568183898925782e-05,
      "model_forward_time": 1.7102890014648438,
      "step": 17373
    },
    {
      "epoch": 1.6568183898925782e-05,
      "step": 17373,
      "training_step_time": 7.545836925506592
    },
    {
      "epoch": 1.6569137573242186e-05,
      "model_forward_time": 1.7064826488494873,
      "step": 17374
    },
    {
      "epoch": 1.6569137573242186e-05,
      "step": 17374,
      "training_step_time": 10.019358396530151
    },
    {
      "epoch": 1.6570091247558594e-05,
      "model_forward_time": 1.7514126300811768,
      "step": 17375
    },
    {
      "epoch": 1.6570091247558594e-05,
      "step": 17375,
      "training_step_time": 9.272905349731445
    },
    {
      "epoch": 1.6571044921875e-05,
      "model_forward_time": 1.6810493469238281,
      "step": 17376
    },
    {
      "epoch": 1.6571044921875e-05,
      "step": 17376,
      "training_step_time": 7.628864765167236
    },
    {
      "epoch": 1.6571998596191405e-05,
      "model_forward_time": 1.7058303356170654,
      "step": 17377
    },
    {
      "epoch": 1.6571998596191405e-05,
      "step": 17377,
      "training_step_time": 8.379083633422852
    },
    {
      "epoch": 1.6572952270507813e-05,
      "model_forward_time": 1.7051258087158203,
      "step": 17378
    },
    {
      "epoch": 1.6572952270507813e-05,
      "step": 17378,
      "training_step_time": 7.522619724273682
    },
    {
      "epoch": 1.657390594482422e-05,
      "model_forward_time": 1.7127001285552979,
      "step": 17379
    },
    {
      "epoch": 1.657390594482422e-05,
      "step": 17379,
      "training_step_time": 6.702199935913086
    },
    {
      "epoch": 1.6574859619140624e-05,
      "grad_norm": 0.1893681436777115,
      "learning_rate": 9.586866669607336e-06,
      "loss": 0.0701,
      "step": 17380
    },
    {
      "epoch": 1.6574859619140624e-05,
      "model_forward_time": 1.7004940509796143,
      "step": 17380
    },
    {
      "dynamics_loss_avg": 0.06521303039044142,
      "epoch": 1.6574859619140624e-05,
      "step": 17380
    },
    {
      "action_loss_avg": 0.003916836180724204,
      "epoch": 1.6574859619140624e-05,
      "step": 17380
    },
    {
      "epoch": 1.6574859619140624e-05,
      "step": 17380,
      "training_step_time": 7.539305686950684
    },
    {
      "epoch": 1.657581329345703e-05,
      "model_forward_time": 1.7131810188293457,
      "step": 17381
    },
    {
      "epoch": 1.657581329345703e-05,
      "step": 17381,
      "training_step_time": 7.622307300567627
    },
    {
      "epoch": 1.657676696777344e-05,
      "model_forward_time": 1.7311909198760986,
      "step": 17382
    },
    {
      "epoch": 1.657676696777344e-05,
      "step": 17382,
      "training_step_time": 8.555615186691284
    },
    {
      "epoch": 1.6577720642089843e-05,
      "model_forward_time": 1.7128574848175049,
      "step": 17383
    },
    {
      "epoch": 1.6577720642089843e-05,
      "step": 17383,
      "training_step_time": 9.211748361587524
    },
    {
      "epoch": 1.657867431640625e-05,
      "model_forward_time": 1.3263957500457764,
      "step": 17384
    },
    {
      "epoch": 1.657867431640625e-05,
      "step": 17384,
      "training_step_time": 8.874095678329468
    },
    {
      "epoch": 1.6579627990722658e-05,
      "model_forward_time": 2.250394105911255,
      "step": 17385
    },
    {
      "epoch": 1.6579627990722658e-05,
      "step": 17385,
      "training_step_time": 8.338399171829224
    },
    {
      "epoch": 1.6580581665039062e-05,
      "model_forward_time": 1.71162748336792,
      "step": 17386
    },
    {
      "epoch": 1.6580581665039062e-05,
      "step": 17386,
      "training_step_time": 8.638330459594727
    },
    {
      "epoch": 1.658153533935547e-05,
      "model_forward_time": 1.7108843326568604,
      "step": 17387
    },
    {
      "epoch": 1.658153533935547e-05,
      "step": 17387,
      "training_step_time": 7.610435724258423
    },
    {
      "epoch": 1.6582489013671877e-05,
      "model_forward_time": 1.7034196853637695,
      "step": 17388
    },
    {
      "epoch": 1.6582489013671877e-05,
      "step": 17388,
      "training_step_time": 7.558878421783447
    },
    {
      "epoch": 1.658344268798828e-05,
      "model_forward_time": 1.7139391899108887,
      "step": 17389
    },
    {
      "epoch": 1.658344268798828e-05,
      "step": 17389,
      "training_step_time": 7.562607765197754
    },
    {
      "epoch": 1.6584396362304688e-05,
      "grad_norm": 0.21754591166973114,
      "learning_rate": 9.586208291782915e-06,
      "loss": 0.0677,
      "step": 17390
    },
    {
      "epoch": 1.6584396362304688e-05,
      "model_forward_time": 1.7048344612121582,
      "step": 17390
    },
    {
      "dynamics_loss_avg": 0.052595583349466325,
      "epoch": 1.6584396362304688e-05,
      "step": 17390
    },
    {
      "action_loss_avg": 0.002281806734390557,
      "epoch": 1.6584396362304688e-05,
      "step": 17390
    },
    {
      "epoch": 1.6584396362304688e-05,
      "step": 17390,
      "training_step_time": 7.621261835098267
    },
    {
      "epoch": 1.6585350036621092e-05,
      "model_forward_time": 1.7124848365783691,
      "step": 17391
    },
    {
      "epoch": 1.6585350036621092e-05,
      "step": 17391,
      "training_step_time": 8.412344932556152
    },
    {
      "epoch": 1.65863037109375e-05,
      "model_forward_time": 1.7120857238769531,
      "step": 17392
    },
    {
      "epoch": 1.65863037109375e-05,
      "step": 17392,
      "training_step_time": 9.28596568107605
    },
    {
      "epoch": 1.6587257385253907e-05,
      "model_forward_time": 1.717003345489502,
      "step": 17393
    },
    {
      "epoch": 1.6587257385253907e-05,
      "step": 17393,
      "training_step_time": 8.63828158378601
    },
    {
      "epoch": 1.658821105957031e-05,
      "model_forward_time": 2.3017218112945557,
      "step": 17394
    },
    {
      "epoch": 1.658821105957031e-05,
      "step": 17394,
      "training_step_time": 7.922763347625732
    },
    {
      "epoch": 1.658916473388672e-05,
      "model_forward_time": 1.7098264694213867,
      "step": 17395
    },
    {
      "epoch": 1.658916473388672e-05,
      "step": 17395,
      "training_step_time": 7.736575365066528
    },
    {
      "epoch": 1.6590118408203126e-05,
      "model_forward_time": 1.7156236171722412,
      "step": 17396
    },
    {
      "epoch": 1.6590118408203126e-05,
      "step": 17396,
      "training_step_time": 7.731907844543457
    },
    {
      "epoch": 1.659107208251953e-05,
      "model_forward_time": 1.7125599384307861,
      "step": 17397
    },
    {
      "epoch": 1.659107208251953e-05,
      "step": 17397,
      "training_step_time": 7.647331953048706
    },
    {
      "epoch": 1.6592025756835937e-05,
      "model_forward_time": 1.7120654582977295,
      "step": 17398
    },
    {
      "epoch": 1.6592025756835937e-05,
      "step": 17398,
      "training_step_time": 7.542029619216919
    },
    {
      "epoch": 1.6592979431152345e-05,
      "model_forward_time": 1.7055120468139648,
      "step": 17399
    },
    {
      "epoch": 1.6592979431152345e-05,
      "step": 17399,
      "training_step_time": 7.540320158004761
    },
    {
      "epoch": 1.659393310546875e-05,
      "grad_norm": 0.37181147933006287,
      "learning_rate": 9.585549412417649e-06,
      "loss": 0.0664,
      "step": 17400
    },
    {
      "epoch": 1.659393310546875e-05,
      "model_forward_time": 1.3329355716705322,
      "step": 17400
    },
    {
      "dynamics_loss_avg": 0.08453040085732937,
      "epoch": 1.659393310546875e-05,
      "step": 17400
    },
    {
      "action_loss_avg": 0.003959002869669348,
      "epoch": 1.659393310546875e-05,
      "step": 17400
    },
    {
      "epoch": 1.659393310546875e-05,
      "step": 17400,
      "training_step_time": 9.560070037841797
    },
    {
      "epoch": 1.6594886779785156e-05,
      "model_forward_time": 0.9913380146026611,
      "step": 17401
    },
    {
      "epoch": 1.6594886779785156e-05,
      "step": 17401,
      "training_step_time": 8.406506538391113
    },
    {
      "epoch": 1.6595840454101563e-05,
      "model_forward_time": 1.7123026847839355,
      "step": 17402
    },
    {
      "epoch": 1.6595840454101563e-05,
      "step": 17402,
      "training_step_time": 7.735813617706299
    },
    {
      "epoch": 1.6596794128417968e-05,
      "model_forward_time": 2.2767128944396973,
      "step": 17403
    },
    {
      "epoch": 1.6596794128417968e-05,
      "step": 17403,
      "training_step_time": 9.433581590652466
    },
    {
      "epoch": 1.6597747802734375e-05,
      "model_forward_time": 1.707956314086914,
      "step": 17404
    },
    {
      "epoch": 1.6597747802734375e-05,
      "step": 17404,
      "training_step_time": 9.327204704284668
    },
    {
      "epoch": 1.6598701477050782e-05,
      "model_forward_time": 1.710916519165039,
      "step": 17405
    },
    {
      "epoch": 1.6598701477050782e-05,
      "step": 17405,
      "training_step_time": 7.814072132110596
    },
    {
      "epoch": 1.6599655151367186e-05,
      "model_forward_time": 1.7128145694732666,
      "step": 17406
    },
    {
      "epoch": 1.6599655151367186e-05,
      "step": 17406,
      "training_step_time": 7.617276668548584
    },
    {
      "epoch": 1.6600608825683594e-05,
      "model_forward_time": 1.7129759788513184,
      "step": 17407
    },
    {
      "epoch": 1.6600608825683594e-05,
      "step": 17407,
      "training_step_time": 7.580851078033447
    },
    {
      "epoch": 1.66015625e-05,
      "model_forward_time": 1.7088205814361572,
      "step": 17408
    },
    {
      "epoch": 1.66015625e-05,
      "step": 17408,
      "training_step_time": 7.472228288650513
    },
    {
      "epoch": 1.6602516174316405e-05,
      "model_forward_time": 1.7118244171142578,
      "step": 17409
    },
    {
      "epoch": 1.6602516174316405e-05,
      "step": 17409,
      "training_step_time": 7.655323505401611
    },
    {
      "epoch": 1.6603469848632813e-05,
      "grad_norm": 0.2813517451286316,
      "learning_rate": 9.584890031583595e-06,
      "loss": 0.0619,
      "step": 17410
    },
    {
      "epoch": 1.6603469848632813e-05,
      "model_forward_time": 1.6944382190704346,
      "step": 17410
    },
    {
      "dynamics_loss_avg": 0.058221946097910406,
      "epoch": 1.6603469848632813e-05,
      "step": 17410
    },
    {
      "action_loss_avg": 0.0031853433756623417,
      "epoch": 1.6603469848632813e-05,
      "step": 17410
    },
    {
      "epoch": 1.6603469848632813e-05,
      "step": 17410,
      "training_step_time": 9.151163101196289
    },
    {
      "epoch": 1.660442352294922e-05,
      "model_forward_time": 1.7069330215454102,
      "step": 17411
    },
    {
      "epoch": 1.660442352294922e-05,
      "step": 17411,
      "training_step_time": 8.727827548980713
    },
    {
      "epoch": 1.6605377197265624e-05,
      "model_forward_time": 2.2638332843780518,
      "step": 17412
    },
    {
      "epoch": 1.6605377197265624e-05,
      "step": 17412,
      "training_step_time": 8.476502656936646
    },
    {
      "epoch": 1.660633087158203e-05,
      "model_forward_time": 1.7132270336151123,
      "step": 17413
    },
    {
      "epoch": 1.660633087158203e-05,
      "step": 17413,
      "training_step_time": 8.668208360671997
    },
    {
      "epoch": 1.660728454589844e-05,
      "model_forward_time": 1.7093188762664795,
      "step": 17414
    },
    {
      "epoch": 1.660728454589844e-05,
      "step": 17414,
      "training_step_time": 7.725921630859375
    },
    {
      "epoch": 1.6608238220214843e-05,
      "model_forward_time": 1.7053894996643066,
      "step": 17415
    },
    {
      "epoch": 1.6608238220214843e-05,
      "step": 17415,
      "training_step_time": 7.23343825340271
    },
    {
      "epoch": 1.660919189453125e-05,
      "model_forward_time": 1.7100927829742432,
      "step": 17416
    },
    {
      "epoch": 1.660919189453125e-05,
      "step": 17416,
      "training_step_time": 7.583174228668213
    },
    {
      "epoch": 1.6610145568847658e-05,
      "model_forward_time": 1.7150952816009521,
      "step": 17417
    },
    {
      "epoch": 1.6610145568847658e-05,
      "step": 17417,
      "training_step_time": 6.994441270828247
    },
    {
      "epoch": 1.6611099243164062e-05,
      "model_forward_time": 1.7028019428253174,
      "step": 17418
    },
    {
      "epoch": 1.6611099243164062e-05,
      "step": 17418,
      "training_step_time": 8.148534536361694
    },
    {
      "epoch": 1.661205291748047e-05,
      "model_forward_time": 1.7051753997802734,
      "step": 17419
    },
    {
      "epoch": 1.661205291748047e-05,
      "step": 17419,
      "training_step_time": 8.09402871131897
    },
    {
      "epoch": 1.6613006591796877e-05,
      "grad_norm": 0.17857086658477783,
      "learning_rate": 9.584230149352862e-06,
      "loss": 0.0658,
      "step": 17420
    },
    {
      "epoch": 1.6613006591796877e-05,
      "model_forward_time": 1.700439214706421,
      "step": 17420
    },
    {
      "dynamics_loss_avg": 0.07451998218894004,
      "epoch": 1.6613006591796877e-05,
      "step": 17420
    },
    {
      "action_loss_avg": 0.006046248890925199,
      "epoch": 1.6613006591796877e-05,
      "step": 17420
    },
    {
      "epoch": 1.6613006591796877e-05,
      "step": 17420,
      "training_step_time": 8.737141370773315
    },
    {
      "epoch": 1.661396026611328e-05,
      "model_forward_time": 1.7451238632202148,
      "step": 17421
    },
    {
      "epoch": 1.661396026611328e-05,
      "step": 17421,
      "training_step_time": 8.67452883720398
    },
    {
      "epoch": 1.6614913940429688e-05,
      "model_forward_time": 2.252901077270508,
      "step": 17422
    },
    {
      "epoch": 1.6614913940429688e-05,
      "step": 17422,
      "training_step_time": 7.700654745101929
    },
    {
      "epoch": 1.6615867614746092e-05,
      "model_forward_time": 1.712374210357666,
      "step": 17423
    },
    {
      "epoch": 1.6615867614746092e-05,
      "step": 17423,
      "training_step_time": 7.698721170425415
    },
    {
      "epoch": 1.66168212890625e-05,
      "model_forward_time": 1.714240550994873,
      "step": 17424
    },
    {
      "epoch": 1.66168212890625e-05,
      "step": 17424,
      "training_step_time": 7.558054685592651
    },
    {
      "epoch": 1.6617774963378907e-05,
      "model_forward_time": 1.7160024642944336,
      "step": 17425
    },
    {
      "epoch": 1.6617774963378907e-05,
      "step": 17425,
      "training_step_time": 7.473092794418335
    },
    {
      "epoch": 1.661872863769531e-05,
      "model_forward_time": 1.7067511081695557,
      "step": 17426
    },
    {
      "epoch": 1.661872863769531e-05,
      "step": 17426,
      "training_step_time": 6.959455728530884
    },
    {
      "epoch": 1.661968231201172e-05,
      "model_forward_time": 1.7140405178070068,
      "step": 17427
    },
    {
      "epoch": 1.661968231201172e-05,
      "step": 17427,
      "training_step_time": 7.494840383529663
    },
    {
      "epoch": 1.6620635986328126e-05,
      "model_forward_time": 1.7093839645385742,
      "step": 17428
    },
    {
      "epoch": 1.6620635986328126e-05,
      "step": 17428,
      "training_step_time": 7.6244118213653564
    },
    {
      "epoch": 1.662158966064453e-05,
      "model_forward_time": 1.7121124267578125,
      "step": 17429
    },
    {
      "epoch": 1.662158966064453e-05,
      "step": 17429,
      "training_step_time": 8.907200574874878
    },
    {
      "epoch": 1.6622543334960937e-05,
      "grad_norm": 0.2525975704193115,
      "learning_rate": 9.58356976579761e-06,
      "loss": 0.0688,
      "step": 17430
    },
    {
      "epoch": 1.6622543334960937e-05,
      "model_forward_time": 1.7020330429077148,
      "step": 17430
    },
    {
      "dynamics_loss_avg": 0.0691611971706152,
      "epoch": 1.6622543334960937e-05,
      "step": 17430
    },
    {
      "action_loss_avg": 0.0037416192819364367,
      "epoch": 1.6622543334960937e-05,
      "step": 17430
    },
    {
      "epoch": 1.6622543334960937e-05,
      "step": 17430,
      "training_step_time": 7.679180860519409
    },
    {
      "epoch": 1.6623497009277345e-05,
      "model_forward_time": 2.2521848678588867,
      "step": 17431
    },
    {
      "epoch": 1.6623497009277345e-05,
      "step": 17431,
      "training_step_time": 9.364672899246216
    },
    {
      "epoch": 1.662445068359375e-05,
      "model_forward_time": 1.7117094993591309,
      "step": 17432
    },
    {
      "epoch": 1.662445068359375e-05,
      "step": 17432,
      "training_step_time": 7.672595500946045
    },
    {
      "epoch": 1.6625404357910156e-05,
      "model_forward_time": 1.7093262672424316,
      "step": 17433
    },
    {
      "epoch": 1.6625404357910156e-05,
      "step": 17433,
      "training_step_time": 7.508416414260864
    },
    {
      "epoch": 1.6626358032226564e-05,
      "model_forward_time": 1.707667350769043,
      "step": 17434
    },
    {
      "epoch": 1.6626358032226564e-05,
      "step": 17434,
      "training_step_time": 7.47487473487854
    },
    {
      "epoch": 1.6627311706542968e-05,
      "model_forward_time": 1.7118985652923584,
      "step": 17435
    },
    {
      "epoch": 1.6627311706542968e-05,
      "step": 17435,
      "training_step_time": 6.962723016738892
    },
    {
      "epoch": 1.6628265380859375e-05,
      "model_forward_time": 1.711669683456421,
      "step": 17436
    },
    {
      "epoch": 1.6628265380859375e-05,
      "step": 17436,
      "training_step_time": 7.482964038848877
    },
    {
      "epoch": 1.6629219055175782e-05,
      "model_forward_time": 1.7142767906188965,
      "step": 17437
    },
    {
      "epoch": 1.6629219055175782e-05,
      "step": 17437,
      "training_step_time": 8.457705974578857
    },
    {
      "epoch": 1.6630172729492186e-05,
      "model_forward_time": 1.3315930366516113,
      "step": 17438
    },
    {
      "epoch": 1.6630172729492186e-05,
      "step": 17438,
      "training_step_time": 8.56021237373352
    },
    {
      "epoch": 1.6631126403808594e-05,
      "model_forward_time": 1.7086913585662842,
      "step": 17439
    },
    {
      "epoch": 1.6631126403808594e-05,
      "step": 17439,
      "training_step_time": 9.727794408798218
    },
    {
      "epoch": 1.6632080078125e-05,
      "grad_norm": 0.2182130217552185,
      "learning_rate": 9.58290888099006e-06,
      "loss": 0.0692,
      "step": 17440
    },
    {
      "epoch": 1.6632080078125e-05,
      "model_forward_time": 2.2637526988983154,
      "step": 17440
    },
    {
      "dynamics_loss_avg": 0.0714585779234767,
      "epoch": 1.6632080078125e-05,
      "step": 17440
    },
    {
      "action_loss_avg": 0.002690970851108432,
      "epoch": 1.6632080078125e-05,
      "step": 17440
    },
    {
      "epoch": 1.6632080078125e-05,
      "step": 17440,
      "training_step_time": 7.787079334259033
    },
    {
      "epoch": 1.6633033752441405e-05,
      "model_forward_time": 1.7093334197998047,
      "step": 17441
    },
    {
      "epoch": 1.6633033752441405e-05,
      "step": 17441,
      "training_step_time": 7.736746788024902
    },
    {
      "epoch": 1.6633987426757813e-05,
      "model_forward_time": 1.7053182125091553,
      "step": 17442
    },
    {
      "epoch": 1.6633987426757813e-05,
      "step": 17442,
      "training_step_time": 7.556730031967163
    },
    {
      "epoch": 1.663494110107422e-05,
      "model_forward_time": 1.7099390029907227,
      "step": 17443
    },
    {
      "epoch": 1.663494110107422e-05,
      "step": 17443,
      "training_step_time": 7.57503867149353
    },
    {
      "epoch": 1.6635894775390624e-05,
      "model_forward_time": 1.7083585262298584,
      "step": 17444
    },
    {
      "epoch": 1.6635894775390624e-05,
      "step": 17444,
      "training_step_time": 7.046893358230591
    },
    {
      "epoch": 1.663684844970703e-05,
      "model_forward_time": 1.7130563259124756,
      "step": 17445
    },
    {
      "epoch": 1.663684844970703e-05,
      "step": 17445,
      "training_step_time": 7.480379104614258
    },
    {
      "epoch": 1.663780212402344e-05,
      "model_forward_time": 1.7137551307678223,
      "step": 17446
    },
    {
      "epoch": 1.663780212402344e-05,
      "step": 17446,
      "training_step_time": 7.7389891147613525
    },
    {
      "epoch": 1.6638755798339843e-05,
      "model_forward_time": 1.6986682415008545,
      "step": 17447
    },
    {
      "epoch": 1.6638755798339843e-05,
      "step": 17447,
      "training_step_time": 9.23323130607605
    },
    {
      "epoch": 1.663970947265625e-05,
      "model_forward_time": 1.7107880115509033,
      "step": 17448
    },
    {
      "epoch": 1.663970947265625e-05,
      "step": 17448,
      "training_step_time": 7.618995189666748
    },
    {
      "epoch": 1.6640663146972658e-05,
      "model_forward_time": 2.270847797393799,
      "step": 17449
    },
    {
      "epoch": 1.6640663146972658e-05,
      "step": 17449,
      "training_step_time": 8.982824087142944
    },
    {
      "epoch": 1.6641616821289062e-05,
      "grad_norm": 0.16917455196380615,
      "learning_rate": 9.582247495002486e-06,
      "loss": 0.0616,
      "step": 17450
    },
    {
      "epoch": 1.6641616821289062e-05,
      "model_forward_time": 1.702305555343628,
      "step": 17450
    },
    {
      "dynamics_loss_avg": 0.0708383807912469,
      "epoch": 1.6641616821289062e-05,
      "step": 17450
    },
    {
      "action_loss_avg": 0.0029656880768015982,
      "epoch": 1.6641616821289062e-05,
      "step": 17450
    },
    {
      "epoch": 1.6641616821289062e-05,
      "step": 17450,
      "training_step_time": 8.324825048446655
    },
    {
      "epoch": 1.664257049560547e-05,
      "model_forward_time": 1.7077977657318115,
      "step": 17451
    },
    {
      "epoch": 1.664257049560547e-05,
      "step": 17451,
      "training_step_time": 7.594189643859863
    },
    {
      "epoch": 1.6643524169921877e-05,
      "model_forward_time": 1.7076876163482666,
      "step": 17452
    },
    {
      "epoch": 1.6643524169921877e-05,
      "step": 17452,
      "training_step_time": 7.628793954849243
    },
    {
      "epoch": 1.664447784423828e-05,
      "model_forward_time": 1.7161047458648682,
      "step": 17453
    },
    {
      "epoch": 1.664447784423828e-05,
      "step": 17453,
      "training_step_time": 7.515987157821655
    },
    {
      "epoch": 1.6645431518554688e-05,
      "model_forward_time": 1.7097508907318115,
      "step": 17454
    },
    {
      "epoch": 1.6645431518554688e-05,
      "step": 17454,
      "training_step_time": 7.308961391448975
    },
    {
      "epoch": 1.6646385192871092e-05,
      "model_forward_time": 1.7085120677947998,
      "step": 17455
    },
    {
      "epoch": 1.6646385192871092e-05,
      "step": 17455,
      "training_step_time": 8.608468294143677
    },
    {
      "epoch": 1.66473388671875e-05,
      "model_forward_time": 1.715672492980957,
      "step": 17456
    },
    {
      "epoch": 1.66473388671875e-05,
      "step": 17456,
      "training_step_time": 7.63658881187439
    },
    {
      "epoch": 1.6648292541503907e-05,
      "model_forward_time": 1.7142133712768555,
      "step": 17457
    },
    {
      "epoch": 1.6648292541503907e-05,
      "step": 17457,
      "training_step_time": 8.341342210769653
    },
    {
      "epoch": 1.664924621582031e-05,
      "model_forward_time": 1.7469563484191895,
      "step": 17458
    },
    {
      "epoch": 1.664924621582031e-05,
      "step": 17458,
      "training_step_time": 8.805798292160034
    },
    {
      "epoch": 1.665019989013672e-05,
      "model_forward_time": 1.709275245666504,
      "step": 17459
    },
    {
      "epoch": 1.665019989013672e-05,
      "step": 17459,
      "training_step_time": 7.988657712936401
    },
    {
      "epoch": 1.6651153564453126e-05,
      "grad_norm": 0.176472470164299,
      "learning_rate": 9.581585607907214e-06,
      "loss": 0.0674,
      "step": 17460
    },
    {
      "epoch": 1.6651153564453126e-05,
      "model_forward_time": 1.7103588581085205,
      "step": 17460
    },
    {
      "dynamics_loss_avg": 0.06311026439070702,
      "epoch": 1.6651153564453126e-05,
      "step": 17460
    },
    {
      "action_loss_avg": 0.0033560730516910553,
      "epoch": 1.6651153564453126e-05,
      "step": 17460
    },
    {
      "epoch": 1.6651153564453126e-05,
      "step": 17460,
      "training_step_time": 7.679118633270264
    },
    {
      "epoch": 1.665210723876953e-05,
      "model_forward_time": 1.7192249298095703,
      "step": 17461
    },
    {
      "epoch": 1.665210723876953e-05,
      "step": 17461,
      "training_step_time": 7.690954923629761
    },
    {
      "epoch": 1.6653060913085937e-05,
      "model_forward_time": 1.7126002311706543,
      "step": 17462
    },
    {
      "epoch": 1.6653060913085937e-05,
      "step": 17462,
      "training_step_time": 7.655107021331787
    },
    {
      "epoch": 1.6654014587402345e-05,
      "model_forward_time": 1.7081201076507568,
      "step": 17463
    },
    {
      "epoch": 1.6654014587402345e-05,
      "step": 17463,
      "training_step_time": 7.4189066886901855
    },
    {
      "epoch": 1.665496826171875e-05,
      "model_forward_time": 1.7106544971466064,
      "step": 17464
    },
    {
      "epoch": 1.665496826171875e-05,
      "step": 17464,
      "training_step_time": 7.6450324058532715
    },
    {
      "epoch": 1.6655921936035156e-05,
      "model_forward_time": 1.714167594909668,
      "step": 17465
    },
    {
      "epoch": 1.6655921936035156e-05,
      "step": 17465,
      "training_step_time": 7.765205383300781
    },
    {
      "epoch": 1.6656875610351564e-05,
      "model_forward_time": 1.7148902416229248,
      "step": 17466
    },
    {
      "epoch": 1.6656875610351564e-05,
      "step": 17466,
      "training_step_time": 7.782894849777222
    },
    {
      "epoch": 1.6657829284667968e-05,
      "model_forward_time": 1.7317650318145752,
      "step": 17467
    },
    {
      "epoch": 1.6657829284667968e-05,
      "step": 17467,
      "training_step_time": 8.325001001358032
    },
    {
      "epoch": 1.6658782958984375e-05,
      "model_forward_time": 2.251492738723755,
      "step": 17468
    },
    {
      "epoch": 1.6658782958984375e-05,
      "step": 17468,
      "training_step_time": 8.623581647872925
    },
    {
      "epoch": 1.6659736633300782e-05,
      "model_forward_time": 1.7146587371826172,
      "step": 17469
    },
    {
      "epoch": 1.6659736633300782e-05,
      "step": 17469,
      "training_step_time": 8.127110481262207
    },
    {
      "epoch": 1.6660690307617186e-05,
      "grad_norm": 0.1968475729227066,
      "learning_rate": 9.580923219776628e-06,
      "loss": 0.0622,
      "step": 17470
    },
    {
      "epoch": 1.6660690307617186e-05,
      "model_forward_time": 1.7290115356445312,
      "step": 17470
    },
    {
      "dynamics_loss_avg": 0.07593068461865186,
      "epoch": 1.6660690307617186e-05,
      "step": 17470
    },
    {
      "action_loss_avg": 0.002690221497323364,
      "epoch": 1.6660690307617186e-05,
      "step": 17470
    },
    {
      "epoch": 1.6660690307617186e-05,
      "step": 17470,
      "training_step_time": 7.587842702865601
    },
    {
      "epoch": 1.6661643981933594e-05,
      "model_forward_time": 1.711543083190918,
      "step": 17471
    },
    {
      "epoch": 1.6661643981933594e-05,
      "step": 17471,
      "training_step_time": 8.08300256729126
    },
    {
      "epoch": 1.666259765625e-05,
      "model_forward_time": 1.7087244987487793,
      "step": 17472
    },
    {
      "epoch": 1.666259765625e-05,
      "step": 17472,
      "training_step_time": 7.239391803741455
    },
    {
      "epoch": 1.6663551330566405e-05,
      "model_forward_time": 1.7117340564727783,
      "step": 17473
    },
    {
      "epoch": 1.6663551330566405e-05,
      "step": 17473,
      "training_step_time": 7.471567869186401
    },
    {
      "epoch": 1.6664505004882813e-05,
      "model_forward_time": 1.7118191719055176,
      "step": 17474
    },
    {
      "epoch": 1.6664505004882813e-05,
      "step": 17474,
      "training_step_time": 8.41607117652893
    },
    {
      "epoch": 1.666545867919922e-05,
      "model_forward_time": 1.708873987197876,
      "step": 17475
    },
    {
      "epoch": 1.666545867919922e-05,
      "step": 17475,
      "training_step_time": 8.479416847229004
    },
    {
      "epoch": 1.6666412353515624e-05,
      "model_forward_time": 1.7146034240722656,
      "step": 17476
    },
    {
      "epoch": 1.6666412353515624e-05,
      "step": 17476,
      "training_step_time": 7.649187326431274
    },
    {
      "epoch": 1.666736602783203e-05,
      "model_forward_time": 2.2477645874023438,
      "step": 17477
    },
    {
      "epoch": 1.666736602783203e-05,
      "step": 17477,
      "training_step_time": 9.155681610107422
    },
    {
      "epoch": 1.666831970214844e-05,
      "model_forward_time": 1.708754301071167,
      "step": 17478
    },
    {
      "epoch": 1.666831970214844e-05,
      "step": 17478,
      "training_step_time": 7.574492454528809
    },
    {
      "epoch": 1.6669273376464843e-05,
      "model_forward_time": 1.711646318435669,
      "step": 17479
    },
    {
      "epoch": 1.6669273376464843e-05,
      "step": 17479,
      "training_step_time": 7.6013946533203125
    },
    {
      "epoch": 1.667022705078125e-05,
      "grad_norm": 0.2588205635547638,
      "learning_rate": 9.580260330683168e-06,
      "loss": 0.0709,
      "step": 17480
    },
    {
      "epoch": 1.667022705078125e-05,
      "model_forward_time": 1.7082033157348633,
      "step": 17480
    },
    {
      "dynamics_loss_avg": 0.073033332452178,
      "epoch": 1.667022705078125e-05,
      "step": 17480
    },
    {
      "action_loss_avg": 0.0036251476034522057,
      "epoch": 1.667022705078125e-05,
      "step": 17480
    },
    {
      "epoch": 1.667022705078125e-05,
      "step": 17480,
      "training_step_time": 7.586111068725586
    },
    {
      "epoch": 1.6671180725097658e-05,
      "model_forward_time": 1.7092556953430176,
      "step": 17481
    },
    {
      "epoch": 1.6671180725097658e-05,
      "step": 17481,
      "training_step_time": 7.09247350692749
    },
    {
      "epoch": 1.6672134399414062e-05,
      "model_forward_time": 1.7246243953704834,
      "step": 17482
    },
    {
      "epoch": 1.6672134399414062e-05,
      "step": 17482,
      "training_step_time": 7.542223215103149
    },
    {
      "epoch": 1.667308807373047e-05,
      "model_forward_time": 1.7191431522369385,
      "step": 17483
    },
    {
      "epoch": 1.667308807373047e-05,
      "step": 17483,
      "training_step_time": 7.6762449741363525
    },
    {
      "epoch": 1.6674041748046877e-05,
      "model_forward_time": 1.7209362983703613,
      "step": 17484
    },
    {
      "epoch": 1.6674041748046877e-05,
      "step": 17484,
      "training_step_time": 8.46414065361023
    },
    {
      "epoch": 1.667499542236328e-05,
      "model_forward_time": 1.7222685813903809,
      "step": 17485
    },
    {
      "epoch": 1.667499542236328e-05,
      "step": 17485,
      "training_step_time": 8.968029260635376
    },
    {
      "epoch": 1.6675949096679688e-05,
      "model_forward_time": 2.284252405166626,
      "step": 17486
    },
    {
      "epoch": 1.6675949096679688e-05,
      "step": 17486,
      "training_step_time": 8.856239557266235
    },
    {
      "epoch": 1.6676902770996092e-05,
      "model_forward_time": 1.7227635383605957,
      "step": 17487
    },
    {
      "epoch": 1.6676902770996092e-05,
      "step": 17487,
      "training_step_time": 9.587764501571655
    },
    {
      "epoch": 1.66778564453125e-05,
      "model_forward_time": 1.7251627445220947,
      "step": 17488
    },
    {
      "epoch": 1.66778564453125e-05,
      "step": 17488,
      "training_step_time": 7.59747314453125
    },
    {
      "epoch": 1.6678810119628907e-05,
      "model_forward_time": 1.7205026149749756,
      "step": 17489
    },
    {
      "epoch": 1.6678810119628907e-05,
      "step": 17489,
      "training_step_time": 7.652427434921265
    },
    {
      "epoch": 1.667976379394531e-05,
      "grad_norm": 0.16229867935180664,
      "learning_rate": 9.579596940699322e-06,
      "loss": 0.0663,
      "step": 17490
    },
    {
      "epoch": 1.667976379394531e-05,
      "model_forward_time": 1.7250354290008545,
      "step": 17490
    },
    {
      "dynamics_loss_avg": 0.061962774768471715,
      "epoch": 1.667976379394531e-05,
      "step": 17490
    },
    {
      "action_loss_avg": 0.0028776366903912275,
      "epoch": 1.667976379394531e-05,
      "step": 17490
    },
    {
      "epoch": 1.667976379394531e-05,
      "step": 17490,
      "training_step_time": 7.035895824432373
    },
    {
      "epoch": 1.668071746826172e-05,
      "model_forward_time": 1.7165582180023193,
      "step": 17491
    },
    {
      "epoch": 1.668071746826172e-05,
      "step": 17491,
      "training_step_time": 7.505589723587036
    },
    {
      "epoch": 1.6681671142578126e-05,
      "model_forward_time": 1.7068431377410889,
      "step": 17492
    },
    {
      "epoch": 1.6681671142578126e-05,
      "step": 17492,
      "training_step_time": 8.157056093215942
    },
    {
      "epoch": 1.668262481689453e-05,
      "model_forward_time": 1.7180562019348145,
      "step": 17493
    },
    {
      "epoch": 1.668262481689453e-05,
      "step": 17493,
      "training_step_time": 7.592170000076294
    },
    {
      "epoch": 1.6683578491210937e-05,
      "model_forward_time": 1.7155475616455078,
      "step": 17494
    },
    {
      "epoch": 1.6683578491210937e-05,
      "step": 17494,
      "training_step_time": 8.632001638412476
    },
    {
      "epoch": 1.6684532165527345e-05,
      "model_forward_time": 1.886718988418579,
      "step": 17495
    },
    {
      "epoch": 1.6684532165527345e-05,
      "step": 17495,
      "training_step_time": 7.781625986099243
    },
    {
      "epoch": 1.668548583984375e-05,
      "model_forward_time": 1.7184507846832275,
      "step": 17496
    },
    {
      "epoch": 1.668548583984375e-05,
      "step": 17496,
      "training_step_time": 7.845738410949707
    },
    {
      "epoch": 1.6686439514160156e-05,
      "model_forward_time": 1.7279391288757324,
      "step": 17497
    },
    {
      "epoch": 1.6686439514160156e-05,
      "step": 17497,
      "training_step_time": 7.746697187423706
    },
    {
      "epoch": 1.6687393188476564e-05,
      "model_forward_time": 1.715050220489502,
      "step": 17498
    },
    {
      "epoch": 1.6687393188476564e-05,
      "step": 17498,
      "training_step_time": 7.691478252410889
    },
    {
      "epoch": 1.6688346862792968e-05,
      "model_forward_time": 1.7170236110687256,
      "step": 17499
    },
    {
      "epoch": 1.6688346862792968e-05,
      "step": 17499,
      "training_step_time": 7.68862247467041
    },
    {
      "epoch": 1.6689300537109375e-05,
      "grad_norm": 0.20477190613746643,
      "learning_rate": 9.578933049897643e-06,
      "loss": 0.0672,
      "step": 17500
    },
    {
      "epoch": 1.6689300537109375e-05,
      "model_forward_time": 1.7065064907073975,
      "step": 17500
    },
    {
      "dynamics_loss_avg": 0.05485471338033676,
      "epoch": 1.6689300537109375e-05,
      "step": 17500
    },
    {
      "action_loss_avg": 0.002045708557125181,
      "epoch": 1.6689300537109375e-05,
      "step": 17500
    },
    {
      "epoch": 1.6689300537109375e-05,
      "step": 17500,
      "training_step_time": 8.514536619186401
    },
    {
      "epoch": 1.6690254211425782e-05,
      "model_forward_time": 1.7123019695281982,
      "step": 17501
    },
    {
      "epoch": 1.6690254211425782e-05,
      "step": 17501,
      "training_step_time": 8.676393032073975
    },
    {
      "epoch": 1.6691207885742186e-05,
      "model_forward_time": 1.7228038311004639,
      "step": 17502
    },
    {
      "epoch": 1.6691207885742186e-05,
      "step": 17502,
      "training_step_time": 8.752072095870972
    },
    {
      "epoch": 1.6692161560058594e-05,
      "model_forward_time": 1.7097902297973633,
      "step": 17503
    },
    {
      "epoch": 1.6692161560058594e-05,
      "step": 17503,
      "training_step_time": 8.799839496612549
    },
    {
      "epoch": 1.6693115234375e-05,
      "model_forward_time": 2.27260422706604,
      "step": 17504
    },
    {
      "epoch": 1.6693115234375e-05,
      "step": 17504,
      "training_step_time": 9.383408069610596
    },
    {
      "epoch": 1.6694068908691405e-05,
      "model_forward_time": 1.7117679119110107,
      "step": 17505
    },
    {
      "epoch": 1.6694068908691405e-05,
      "step": 17505,
      "training_step_time": 9.688007831573486
    },
    {
      "epoch": 1.6695022583007813e-05,
      "model_forward_time": 1.7144086360931396,
      "step": 17506
    },
    {
      "epoch": 1.6695022583007813e-05,
      "step": 17506,
      "training_step_time": 7.704354524612427
    },
    {
      "epoch": 1.669597625732422e-05,
      "model_forward_time": 1.7099852561950684,
      "step": 17507
    },
    {
      "epoch": 1.669597625732422e-05,
      "step": 17507,
      "training_step_time": 7.69471001625061
    },
    {
      "epoch": 1.6696929931640624e-05,
      "model_forward_time": 1.7284631729125977,
      "step": 17508
    },
    {
      "epoch": 1.6696929931640624e-05,
      "step": 17508,
      "training_step_time": 7.265830993652344
    },
    {
      "epoch": 1.669788360595703e-05,
      "model_forward_time": 1.7122654914855957,
      "step": 17509
    },
    {
      "epoch": 1.669788360595703e-05,
      "step": 17509,
      "training_step_time": 7.6120710372924805
    },
    {
      "epoch": 1.669883728027344e-05,
      "grad_norm": 0.15789343416690826,
      "learning_rate": 9.578268658350727e-06,
      "loss": 0.0671,
      "step": 17510
    },
    {
      "epoch": 1.669883728027344e-05,
      "model_forward_time": 1.7128658294677734,
      "step": 17510
    },
    {
      "dynamics_loss_avg": 0.06500053703784943,
      "epoch": 1.669883728027344e-05,
      "step": 17510
    },
    {
      "action_loss_avg": 0.002826114045456052,
      "epoch": 1.669883728027344e-05,
      "step": 17510
    },
    {
      "epoch": 1.669883728027344e-05,
      "step": 17510,
      "training_step_time": 9.254803895950317
    },
    {
      "epoch": 1.6699790954589843e-05,
      "model_forward_time": 1.7150022983551025,
      "step": 17511
    },
    {
      "epoch": 1.6699790954589843e-05,
      "step": 17511,
      "training_step_time": 8.41691517829895
    },
    {
      "epoch": 1.670074462890625e-05,
      "model_forward_time": 1.7149314880371094,
      "step": 17512
    },
    {
      "epoch": 1.670074462890625e-05,
      "step": 17512,
      "training_step_time": 8.32865595817566
    },
    {
      "epoch": 1.6701698303222658e-05,
      "model_forward_time": 1.733351230621338,
      "step": 17513
    },
    {
      "epoch": 1.6701698303222658e-05,
      "step": 17513,
      "training_step_time": 7.7434163093566895
    },
    {
      "epoch": 1.6702651977539062e-05,
      "model_forward_time": 2.259596109390259,
      "step": 17514
    },
    {
      "epoch": 1.6702651977539062e-05,
      "step": 17514,
      "training_step_time": 9.933632373809814
    },
    {
      "epoch": 1.670360565185547e-05,
      "model_forward_time": 1.3423547744750977,
      "step": 17515
    },
    {
      "epoch": 1.670360565185547e-05,
      "step": 17515,
      "training_step_time": 7.904978036880493
    },
    {
      "epoch": 1.6704559326171873e-05,
      "model_forward_time": 1.7180790901184082,
      "step": 17516
    },
    {
      "epoch": 1.6704559326171873e-05,
      "step": 17516,
      "training_step_time": 7.647148132324219
    },
    {
      "epoch": 1.670551300048828e-05,
      "model_forward_time": 1.7182631492614746,
      "step": 17517
    },
    {
      "epoch": 1.670551300048828e-05,
      "step": 17517,
      "training_step_time": 7.1498284339904785
    },
    {
      "epoch": 1.6706466674804688e-05,
      "model_forward_time": 1.717397928237915,
      "step": 17518
    },
    {
      "epoch": 1.6706466674804688e-05,
      "step": 17518,
      "training_step_time": 7.6019604206085205
    },
    {
      "epoch": 1.6707420349121092e-05,
      "model_forward_time": 1.6999051570892334,
      "step": 17519
    },
    {
      "epoch": 1.6707420349121092e-05,
      "step": 17519,
      "training_step_time": 7.527015209197998
    },
    {
      "epoch": 1.67083740234375e-05,
      "grad_norm": 0.23684823513031006,
      "learning_rate": 9.577603766131236e-06,
      "loss": 0.0668,
      "step": 17520
    },
    {
      "epoch": 1.67083740234375e-05,
      "model_forward_time": 1.7169222831726074,
      "step": 17520
    },
    {
      "dynamics_loss_avg": 0.07903936505317688,
      "epoch": 1.67083740234375e-05,
      "step": 17520
    },
    {
      "action_loss_avg": 0.004107061389368028,
      "epoch": 1.67083740234375e-05,
      "step": 17520
    },
    {
      "epoch": 1.67083740234375e-05,
      "step": 17520,
      "training_step_time": 8.301108121871948
    },
    {
      "epoch": 1.6709327697753907e-05,
      "model_forward_time": 1.716787338256836,
      "step": 17521
    },
    {
      "epoch": 1.6709327697753907e-05,
      "step": 17521,
      "training_step_time": 8.30697226524353
    },
    {
      "epoch": 1.671028137207031e-05,
      "model_forward_time": 1.712658166885376,
      "step": 17522
    },
    {
      "epoch": 1.671028137207031e-05,
      "step": 17522,
      "training_step_time": 9.541475057601929
    },
    {
      "epoch": 1.671123504638672e-05,
      "model_forward_time": 2.2674968242645264,
      "step": 17523
    },
    {
      "epoch": 1.671123504638672e-05,
      "step": 17523,
      "training_step_time": 8.863739967346191
    },
    {
      "epoch": 1.6712188720703126e-05,
      "model_forward_time": 1.7154889106750488,
      "step": 17524
    },
    {
      "epoch": 1.6712188720703126e-05,
      "step": 17524,
      "training_step_time": 8.987679481506348
    },
    {
      "epoch": 1.671314239501953e-05,
      "model_forward_time": 1.7176098823547363,
      "step": 17525
    },
    {
      "epoch": 1.671314239501953e-05,
      "step": 17525,
      "training_step_time": 7.380029201507568
    },
    {
      "epoch": 1.6714096069335937e-05,
      "model_forward_time": 1.7071912288665771,
      "step": 17526
    },
    {
      "epoch": 1.6714096069335937e-05,
      "step": 17526,
      "training_step_time": 7.585301160812378
    },
    {
      "epoch": 1.6715049743652345e-05,
      "model_forward_time": 1.7096753120422363,
      "step": 17527
    },
    {
      "epoch": 1.6715049743652345e-05,
      "step": 17527,
      "training_step_time": 7.497982740402222
    },
    {
      "epoch": 1.671600341796875e-05,
      "model_forward_time": 1.712245225906372,
      "step": 17528
    },
    {
      "epoch": 1.671600341796875e-05,
      "step": 17528,
      "training_step_time": 7.673161745071411
    },
    {
      "epoch": 1.6716957092285156e-05,
      "model_forward_time": 1.3361775875091553,
      "step": 17529
    },
    {
      "epoch": 1.6716957092285156e-05,
      "step": 17529,
      "training_step_time": 8.39961051940918
    },
    {
      "epoch": 1.6717910766601564e-05,
      "grad_norm": 0.16349002718925476,
      "learning_rate": 9.576938373311877e-06,
      "loss": 0.0658,
      "step": 17530
    },
    {
      "epoch": 1.6717910766601564e-05,
      "model_forward_time": 1.710892677307129,
      "step": 17530
    },
    {
      "dynamics_loss_avg": 0.05564800128340721,
      "epoch": 1.6717910766601564e-05,
      "step": 17530
    },
    {
      "action_loss_avg": 0.004227784485556186,
      "epoch": 1.6717910766601564e-05,
      "step": 17530
    },
    {
      "epoch": 1.6717910766601564e-05,
      "step": 17530,
      "training_step_time": 8.305823802947998
    },
    {
      "epoch": 1.6718864440917968e-05,
      "model_forward_time": 1.7201886177062988,
      "step": 17531
    },
    {
      "epoch": 1.6718864440917968e-05,
      "step": 17531,
      "training_step_time": 8.456943035125732
    },
    {
      "epoch": 1.6719818115234375e-05,
      "model_forward_time": 2.2880706787109375,
      "step": 17532
    },
    {
      "epoch": 1.6719818115234375e-05,
      "step": 17532,
      "training_step_time": 7.615063428878784
    },
    {
      "epoch": 1.6720771789550782e-05,
      "model_forward_time": 1.711688756942749,
      "step": 17533
    },
    {
      "epoch": 1.6720771789550782e-05,
      "step": 17533,
      "training_step_time": 7.879703760147095
    },
    {
      "epoch": 1.6721725463867186e-05,
      "model_forward_time": 1.7148833274841309,
      "step": 17534
    },
    {
      "epoch": 1.6721725463867186e-05,
      "step": 17534,
      "training_step_time": 7.494487524032593
    },
    {
      "epoch": 1.6722679138183594e-05,
      "model_forward_time": 1.7075974941253662,
      "step": 17535
    },
    {
      "epoch": 1.6722679138183594e-05,
      "step": 17535,
      "training_step_time": 7.521839141845703
    },
    {
      "epoch": 1.67236328125e-05,
      "model_forward_time": 1.7105860710144043,
      "step": 17536
    },
    {
      "epoch": 1.67236328125e-05,
      "step": 17536,
      "training_step_time": 7.268320798873901
    },
    {
      "epoch": 1.6724586486816405e-05,
      "model_forward_time": 1.718411922454834,
      "step": 17537
    },
    {
      "epoch": 1.6724586486816405e-05,
      "step": 17537,
      "training_step_time": 7.593043088912964
    },
    {
      "epoch": 1.6725540161132813e-05,
      "model_forward_time": 1.7097537517547607,
      "step": 17538
    },
    {
      "epoch": 1.6725540161132813e-05,
      "step": 17538,
      "training_step_time": 8.431122064590454
    },
    {
      "epoch": 1.672649383544922e-05,
      "model_forward_time": 1.7128987312316895,
      "step": 17539
    },
    {
      "epoch": 1.672649383544922e-05,
      "step": 17539,
      "training_step_time": 8.586125135421753
    },
    {
      "epoch": 1.6727447509765624e-05,
      "grad_norm": 0.3952004313468933,
      "learning_rate": 9.576272479965421e-06,
      "loss": 0.0645,
      "step": 17540
    },
    {
      "epoch": 1.6727447509765624e-05,
      "model_forward_time": 1.3370234966278076,
      "step": 17540
    },
    {
      "dynamics_loss_avg": 0.09787466526031494,
      "epoch": 1.6727447509765624e-05,
      "step": 17540
    },
    {
      "action_loss_avg": 0.004675640491768718,
      "epoch": 1.6727447509765624e-05,
      "step": 17540
    },
    {
      "epoch": 1.6727447509765624e-05,
      "step": 17540,
      "training_step_time": 8.393568277359009
    },
    {
      "epoch": 1.672840118408203e-05,
      "model_forward_time": 1.7624030113220215,
      "step": 17541
    },
    {
      "epoch": 1.672840118408203e-05,
      "step": 17541,
      "training_step_time": 7.612818241119385
    },
    {
      "epoch": 1.672935485839844e-05,
      "model_forward_time": 2.2820358276367188,
      "step": 17542
    },
    {
      "epoch": 1.672935485839844e-05,
      "step": 17542,
      "training_step_time": 7.906012058258057
    },
    {
      "epoch": 1.6730308532714843e-05,
      "model_forward_time": 1.7166202068328857,
      "step": 17543
    },
    {
      "epoch": 1.6730308532714843e-05,
      "step": 17543,
      "training_step_time": 7.518496036529541
    },
    {
      "epoch": 1.673126220703125e-05,
      "model_forward_time": 1.7154474258422852,
      "step": 17544
    },
    {
      "epoch": 1.673126220703125e-05,
      "step": 17544,
      "training_step_time": 7.559085369110107
    },
    {
      "epoch": 1.6732215881347658e-05,
      "model_forward_time": 1.7163810729980469,
      "step": 17545
    },
    {
      "epoch": 1.6732215881347658e-05,
      "step": 17545,
      "training_step_time": 7.250398397445679
    },
    {
      "epoch": 1.6733169555664062e-05,
      "model_forward_time": 1.712522029876709,
      "step": 17546
    },
    {
      "epoch": 1.6733169555664062e-05,
      "step": 17546,
      "training_step_time": 8.562243700027466
    },
    {
      "epoch": 1.673412322998047e-05,
      "model_forward_time": 1.710646152496338,
      "step": 17547
    },
    {
      "epoch": 1.673412322998047e-05,
      "step": 17547,
      "training_step_time": 8.102583885192871
    },
    {
      "epoch": 1.6735076904296873e-05,
      "model_forward_time": 1.7195584774017334,
      "step": 17548
    },
    {
      "epoch": 1.6735076904296873e-05,
      "step": 17548,
      "training_step_time": 7.637531280517578
    },
    {
      "epoch": 1.673603057861328e-05,
      "model_forward_time": 1.7128350734710693,
      "step": 17549
    },
    {
      "epoch": 1.673603057861328e-05,
      "step": 17549,
      "training_step_time": 8.336536407470703
    },
    {
      "epoch": 1.6736984252929688e-05,
      "grad_norm": 0.1374511569738388,
      "learning_rate": 9.575606086164687e-06,
      "loss": 0.0614,
      "step": 17550
    },
    {
      "epoch": 1.6736984252929688e-05,
      "model_forward_time": 1.7671921253204346,
      "step": 17550
    },
    {
      "dynamics_loss_avg": 0.059360187500715256,
      "epoch": 1.6736984252929688e-05,
      "step": 17550
    },
    {
      "action_loss_avg": 0.003328673297073692,
      "epoch": 1.6736984252929688e-05,
      "step": 17550
    },
    {
      "epoch": 1.6736984252929688e-05,
      "step": 17550,
      "training_step_time": 8.27909803390503
    },
    {
      "epoch": 1.6737937927246092e-05,
      "model_forward_time": 2.267718553543091,
      "step": 17551
    },
    {
      "epoch": 1.6737937927246092e-05,
      "step": 17551,
      "training_step_time": 9.261014699935913
    },
    {
      "epoch": 1.67388916015625e-05,
      "model_forward_time": 1.7177379131317139,
      "step": 17552
    },
    {
      "epoch": 1.67388916015625e-05,
      "step": 17552,
      "training_step_time": 7.5063395500183105
    },
    {
      "epoch": 1.6739845275878907e-05,
      "model_forward_time": 1.7157728672027588,
      "step": 17553
    },
    {
      "epoch": 1.6739845275878907e-05,
      "step": 17553,
      "training_step_time": 7.540521860122681
    },
    {
      "epoch": 1.674079895019531e-05,
      "model_forward_time": 1.7143046855926514,
      "step": 17554
    },
    {
      "epoch": 1.674079895019531e-05,
      "step": 17554,
      "training_step_time": 6.992488384246826
    },
    {
      "epoch": 1.674175262451172e-05,
      "model_forward_time": 1.7170631885528564,
      "step": 17555
    },
    {
      "epoch": 1.674175262451172e-05,
      "step": 17555,
      "training_step_time": 7.616162061691284
    },
    {
      "epoch": 1.6742706298828126e-05,
      "model_forward_time": 1.7143056392669678,
      "step": 17556
    },
    {
      "epoch": 1.6742706298828126e-05,
      "step": 17556,
      "training_step_time": 7.559874534606934
    },
    {
      "epoch": 1.674365997314453e-05,
      "model_forward_time": 1.7104454040527344,
      "step": 17557
    },
    {
      "epoch": 1.674365997314453e-05,
      "step": 17557,
      "training_step_time": 7.571028470993042
    },
    {
      "epoch": 1.6744613647460937e-05,
      "model_forward_time": 1.7045824527740479,
      "step": 17558
    },
    {
      "epoch": 1.6744613647460937e-05,
      "step": 17558,
      "training_step_time": 8.256656169891357
    },
    {
      "epoch": 1.6745567321777345e-05,
      "model_forward_time": 1.76316499710083,
      "step": 17559
    },
    {
      "epoch": 1.6745567321777345e-05,
      "step": 17559,
      "training_step_time": 7.756058216094971
    },
    {
      "epoch": 1.674652099609375e-05,
      "grad_norm": 0.17912927269935608,
      "learning_rate": 9.574939191982549e-06,
      "loss": 0.0613,
      "step": 17560
    },
    {
      "epoch": 1.674652099609375e-05,
      "model_forward_time": 2.262199878692627,
      "step": 17560
    },
    {
      "dynamics_loss_avg": 0.0682863837108016,
      "epoch": 1.674652099609375e-05,
      "step": 17560
    },
    {
      "action_loss_avg": 0.005162534792907536,
      "epoch": 1.674652099609375e-05,
      "step": 17560
    },
    {
      "epoch": 1.674652099609375e-05,
      "step": 17560,
      "training_step_time": 8.226228713989258
    },
    {
      "epoch": 1.6747474670410156e-05,
      "model_forward_time": 1.71254301071167,
      "step": 17561
    },
    {
      "epoch": 1.6747474670410156e-05,
      "step": 17561,
      "training_step_time": 7.773640155792236
    },
    {
      "epoch": 1.6748428344726564e-05,
      "model_forward_time": 1.7186176776885986,
      "step": 17562
    },
    {
      "epoch": 1.6748428344726564e-05,
      "step": 17562,
      "training_step_time": 8.058130025863647
    },
    {
      "epoch": 1.6749382019042968e-05,
      "model_forward_time": 1.7125957012176514,
      "step": 17563
    },
    {
      "epoch": 1.6749382019042968e-05,
      "step": 17563,
      "training_step_time": 6.661825895309448
    },
    {
      "epoch": 1.6750335693359375e-05,
      "model_forward_time": 1.7153613567352295,
      "step": 17564
    },
    {
      "epoch": 1.6750335693359375e-05,
      "step": 17564,
      "training_step_time": 7.6000683307647705
    },
    {
      "epoch": 1.6751289367675782e-05,
      "model_forward_time": 1.7019963264465332,
      "step": 17565
    },
    {
      "epoch": 1.6751289367675782e-05,
      "step": 17565,
      "training_step_time": 7.561109781265259
    },
    {
      "epoch": 1.6752243041992186e-05,
      "model_forward_time": 1.7146124839782715,
      "step": 17566
    },
    {
      "epoch": 1.6752243041992186e-05,
      "step": 17566,
      "training_step_time": 7.563458442687988
    },
    {
      "epoch": 1.6753196716308594e-05,
      "model_forward_time": 0.9913332462310791,
      "step": 17567
    },
    {
      "epoch": 1.6753196716308594e-05,
      "step": 17567,
      "training_step_time": 7.648786783218384
    },
    {
      "epoch": 1.6754150390625e-05,
      "model_forward_time": 1.7423210144042969,
      "step": 17568
    },
    {
      "epoch": 1.6754150390625e-05,
      "step": 17568,
      "training_step_time": 7.426758289337158
    },
    {
      "epoch": 1.6755104064941405e-05,
      "model_forward_time": 1.716249704360962,
      "step": 17569
    },
    {
      "epoch": 1.6755104064941405e-05,
      "step": 17569,
      "training_step_time": 7.695000171661377
    },
    {
      "epoch": 1.6756057739257813e-05,
      "grad_norm": 0.24565668404102325,
      "learning_rate": 9.574271797491942e-06,
      "loss": 0.0628,
      "step": 17570
    },
    {
      "epoch": 1.6756057739257813e-05,
      "model_forward_time": 2.2596254348754883,
      "step": 17570
    },
    {
      "dynamics_loss_avg": 0.06733405301347375,
      "epoch": 1.6756057739257813e-05,
      "step": 17570
    },
    {
      "action_loss_avg": 0.0023351001786068083,
      "epoch": 1.6756057739257813e-05,
      "step": 17570
    },
    {
      "epoch": 1.6756057739257813e-05,
      "step": 17570,
      "training_step_time": 8.3758225440979
    },
    {
      "epoch": 1.675701141357422e-05,
      "model_forward_time": 1.711446762084961,
      "step": 17571
    },
    {
      "epoch": 1.675701141357422e-05,
      "step": 17571,
      "training_step_time": 7.584360361099243
    },
    {
      "epoch": 1.6757965087890624e-05,
      "model_forward_time": 1.7163195610046387,
      "step": 17572
    },
    {
      "epoch": 1.6757965087890624e-05,
      "step": 17572,
      "training_step_time": 7.106686353683472
    },
    {
      "epoch": 1.675891876220703e-05,
      "model_forward_time": 1.7083067893981934,
      "step": 17573
    },
    {
      "epoch": 1.675891876220703e-05,
      "step": 17573,
      "training_step_time": 8.463177680969238
    },
    {
      "epoch": 1.675987243652344e-05,
      "model_forward_time": 1.714226245880127,
      "step": 17574
    },
    {
      "epoch": 1.675987243652344e-05,
      "step": 17574,
      "training_step_time": 8.122305154800415
    },
    {
      "epoch": 1.6760826110839843e-05,
      "model_forward_time": 1.7082974910736084,
      "step": 17575
    },
    {
      "epoch": 1.6760826110839843e-05,
      "step": 17575,
      "training_step_time": 9.042253494262695
    },
    {
      "epoch": 1.676177978515625e-05,
      "model_forward_time": 1.707705020904541,
      "step": 17576
    },
    {
      "epoch": 1.676177978515625e-05,
      "step": 17576,
      "training_step_time": 8.046439170837402
    },
    {
      "epoch": 1.6762733459472658e-05,
      "model_forward_time": 1.7256579399108887,
      "step": 17577
    },
    {
      "epoch": 1.6762733459472658e-05,
      "step": 17577,
      "training_step_time": 9.62231969833374
    },
    {
      "epoch": 1.6763687133789062e-05,
      "model_forward_time": 1.7127339839935303,
      "step": 17578
    },
    {
      "epoch": 1.6763687133789062e-05,
      "step": 17578,
      "training_step_time": 7.644733190536499
    },
    {
      "epoch": 1.676464080810547e-05,
      "model_forward_time": 2.255946636199951,
      "step": 17579
    },
    {
      "epoch": 1.676464080810547e-05,
      "step": 17579,
      "training_step_time": 10.761889219284058
    },
    {
      "epoch": 1.6765594482421873e-05,
      "grad_norm": 0.3053619861602783,
      "learning_rate": 9.573603902765847e-06,
      "loss": 0.0646,
      "step": 17580
    },
    {
      "epoch": 1.6765594482421873e-05,
      "model_forward_time": 1.708404541015625,
      "step": 17580
    },
    {
      "dynamics_loss_avg": 0.06307922713458539,
      "epoch": 1.6765594482421873e-05,
      "step": 17580
    },
    {
      "action_loss_avg": 0.0036658337397966535,
      "epoch": 1.6765594482421873e-05,
      "step": 17580
    },
    {
      "epoch": 1.6765594482421873e-05,
      "step": 17580,
      "training_step_time": 8.722319602966309
    },
    {
      "epoch": 1.676654815673828e-05,
      "model_forward_time": 1.7106740474700928,
      "step": 17581
    },
    {
      "epoch": 1.676654815673828e-05,
      "step": 17581,
      "training_step_time": 7.528919458389282
    },
    {
      "epoch": 1.6767501831054688e-05,
      "model_forward_time": 1.7073132991790771,
      "step": 17582
    },
    {
      "epoch": 1.6767501831054688e-05,
      "step": 17582,
      "training_step_time": 7.446678638458252
    },
    {
      "epoch": 1.6768455505371092e-05,
      "model_forward_time": 1.7132089138031006,
      "step": 17583
    },
    {
      "epoch": 1.6768455505371092e-05,
      "step": 17583,
      "training_step_time": 8.445162057876587
    },
    {
      "epoch": 1.67694091796875e-05,
      "model_forward_time": 1.7161009311676025,
      "step": 17584
    },
    {
      "epoch": 1.67694091796875e-05,
      "step": 17584,
      "training_step_time": 8.293827772140503
    },
    {
      "epoch": 1.6770362854003907e-05,
      "model_forward_time": 1.7093422412872314,
      "step": 17585
    },
    {
      "epoch": 1.6770362854003907e-05,
      "step": 17585,
      "training_step_time": 8.529421329498291
    },
    {
      "epoch": 1.677131652832031e-05,
      "model_forward_time": 1.7262816429138184,
      "step": 17586
    },
    {
      "epoch": 1.677131652832031e-05,
      "step": 17586,
      "training_step_time": 7.548128843307495
    },
    {
      "epoch": 1.677227020263672e-05,
      "model_forward_time": 1.7615036964416504,
      "step": 17587
    },
    {
      "epoch": 1.677227020263672e-05,
      "step": 17587,
      "training_step_time": 8.688044786453247
    },
    {
      "epoch": 1.6773223876953126e-05,
      "model_forward_time": 2.2529804706573486,
      "step": 17588
    },
    {
      "epoch": 1.6773223876953126e-05,
      "step": 17588,
      "training_step_time": 8.80124807357788
    },
    {
      "epoch": 1.677417755126953e-05,
      "model_forward_time": 1.7123844623565674,
      "step": 17589
    },
    {
      "epoch": 1.677417755126953e-05,
      "step": 17589,
      "training_step_time": 7.793508768081665
    },
    {
      "epoch": 1.6775131225585937e-05,
      "grad_norm": 0.3014695942401886,
      "learning_rate": 9.572935507877305e-06,
      "loss": 0.0598,
      "step": 17590
    },
    {
      "epoch": 1.6775131225585937e-05,
      "model_forward_time": 1.7144520282745361,
      "step": 17590
    },
    {
      "dynamics_loss_avg": 0.07114737927913665,
      "epoch": 1.6775131225585937e-05,
      "step": 17590
    },
    {
      "action_loss_avg": 0.0036649856716394424,
      "epoch": 1.6775131225585937e-05,
      "step": 17590
    },
    {
      "epoch": 1.6775131225585937e-05,
      "step": 17590,
      "training_step_time": 7.123136043548584
    },
    {
      "epoch": 1.6776084899902345e-05,
      "model_forward_time": 1.7099552154541016,
      "step": 17591
    },
    {
      "epoch": 1.6776084899902345e-05,
      "step": 17591,
      "training_step_time": 7.5588579177856445
    },
    {
      "epoch": 1.677703857421875e-05,
      "model_forward_time": 1.714094877243042,
      "step": 17592
    },
    {
      "epoch": 1.677703857421875e-05,
      "step": 17592,
      "training_step_time": 7.620424747467041
    },
    {
      "epoch": 1.6777992248535156e-05,
      "model_forward_time": 1.7090482711791992,
      "step": 17593
    },
    {
      "epoch": 1.6777992248535156e-05,
      "step": 17593,
      "training_step_time": 9.211585283279419
    },
    {
      "epoch": 1.6778945922851564e-05,
      "model_forward_time": 1.7028703689575195,
      "step": 17594
    },
    {
      "epoch": 1.6778945922851564e-05,
      "step": 17594,
      "training_step_time": 8.596194982528687
    },
    {
      "epoch": 1.6779899597167968e-05,
      "model_forward_time": 1.707327127456665,
      "step": 17595
    },
    {
      "epoch": 1.6779899597167968e-05,
      "step": 17595,
      "training_step_time": 7.648453712463379
    },
    {
      "epoch": 1.6780853271484375e-05,
      "model_forward_time": 1.7607035636901855,
      "step": 17596
    },
    {
      "epoch": 1.6780853271484375e-05,
      "step": 17596,
      "training_step_time": 7.526865720748901
    },
    {
      "epoch": 1.6781806945800782e-05,
      "model_forward_time": 2.245298385620117,
      "step": 17597
    },
    {
      "epoch": 1.6781806945800782e-05,
      "step": 17597,
      "training_step_time": 7.704084634780884
    },
    {
      "epoch": 1.6782760620117187e-05,
      "model_forward_time": 1.6955511569976807,
      "step": 17598
    },
    {
      "epoch": 1.6782760620117187e-05,
      "step": 17598,
      "training_step_time": 7.538347959518433
    },
    {
      "epoch": 1.6783714294433594e-05,
      "model_forward_time": 1.7046122550964355,
      "step": 17599
    },
    {
      "epoch": 1.6783714294433594e-05,
      "step": 17599,
      "training_step_time": 7.338366270065308
    },
    {
      "epoch": 1.678466796875e-05,
      "grad_norm": 0.18825338780879974,
      "learning_rate": 9.57226661289941e-06,
      "loss": 0.0625,
      "step": 17600
    },
    {
      "epoch": 1.678466796875e-05,
      "model_forward_time": 1.3294248580932617,
      "step": 17600
    },
    {
      "dynamics_loss_avg": 0.0645340757444501,
      "epoch": 1.678466796875e-05,
      "step": 17600
    },
    {
      "action_loss_avg": 0.0026743178605102003,
      "epoch": 1.678466796875e-05,
      "step": 17600
    },
    {
      "epoch": 1.678466796875e-05,
      "step": 17600,
      "training_step_time": 7.640970945358276
    },
    {
      "epoch": 1.6785621643066405e-05,
      "model_forward_time": 1.70074462890625,
      "step": 17601
    },
    {
      "epoch": 1.6785621643066405e-05,
      "step": 17601,
      "training_step_time": 7.5956196784973145
    },
    {
      "epoch": 1.6786575317382813e-05,
      "model_forward_time": 1.7054133415222168,
      "step": 17602
    },
    {
      "epoch": 1.6786575317382813e-05,
      "step": 17602,
      "training_step_time": 7.566730499267578
    },
    {
      "epoch": 1.678752899169922e-05,
      "model_forward_time": 1.6964352130889893,
      "step": 17603
    },
    {
      "epoch": 1.678752899169922e-05,
      "step": 17603,
      "training_step_time": 7.565946578979492
    },
    {
      "epoch": 1.6788482666015624e-05,
      "model_forward_time": 1.7039411067962646,
      "step": 17604
    },
    {
      "epoch": 1.6788482666015624e-05,
      "step": 17604,
      "training_step_time": 9.469207763671875
    },
    {
      "epoch": 1.678943634033203e-05,
      "model_forward_time": 1.7604548931121826,
      "step": 17605
    },
    {
      "epoch": 1.678943634033203e-05,
      "step": 17605,
      "training_step_time": 7.954253911972046
    },
    {
      "epoch": 1.679039001464844e-05,
      "model_forward_time": 1.3412995338439941,
      "step": 17606
    },
    {
      "epoch": 1.679039001464844e-05,
      "step": 17606,
      "training_step_time": 7.988995552062988
    },
    {
      "epoch": 1.6791343688964843e-05,
      "model_forward_time": 1.7199692726135254,
      "step": 17607
    },
    {
      "epoch": 1.6791343688964843e-05,
      "step": 17607,
      "training_step_time": 7.863001108169556
    },
    {
      "epoch": 1.679229736328125e-05,
      "model_forward_time": 1.6946964263916016,
      "step": 17608
    },
    {
      "epoch": 1.679229736328125e-05,
      "step": 17608,
      "training_step_time": 7.469449043273926
    },
    {
      "epoch": 1.6793251037597658e-05,
      "model_forward_time": 1.7064182758331299,
      "step": 17609
    },
    {
      "epoch": 1.6793251037597658e-05,
      "step": 17609,
      "training_step_time": 7.557430982589722
    },
    {
      "epoch": 1.6794204711914062e-05,
      "grad_norm": 0.16016897559165955,
      "learning_rate": 9.571597217905316e-06,
      "loss": 0.0667,
      "step": 17610
    },
    {
      "epoch": 1.6794204711914062e-05,
      "model_forward_time": 1.716432809829712,
      "step": 17610
    },
    {
      "dynamics_loss_avg": 0.06561856120824813,
      "epoch": 1.6794204711914062e-05,
      "step": 17610
    },
    {
      "action_loss_avg": 0.003423814394045621,
      "epoch": 1.6794204711914062e-05,
      "step": 17610
    },
    {
      "epoch": 1.6794204711914062e-05,
      "step": 17610,
      "training_step_time": 7.532026767730713
    },
    {
      "epoch": 1.679515838623047e-05,
      "model_forward_time": 1.7174649238586426,
      "step": 17611
    },
    {
      "epoch": 1.679515838623047e-05,
      "step": 17611,
      "training_step_time": 7.47542142868042
    },
    {
      "epoch": 1.6796112060546873e-05,
      "model_forward_time": 1.7123620510101318,
      "step": 17612
    },
    {
      "epoch": 1.6796112060546873e-05,
      "step": 17612,
      "training_step_time": 7.684263229370117
    },
    {
      "epoch": 1.679706573486328e-05,
      "model_forward_time": 1.7076501846313477,
      "step": 17613
    },
    {
      "epoch": 1.679706573486328e-05,
      "step": 17613,
      "training_step_time": 8.540607452392578
    },
    {
      "epoch": 1.6798019409179688e-05,
      "model_forward_time": 1.7653062343597412,
      "step": 17614
    },
    {
      "epoch": 1.6798019409179688e-05,
      "step": 17614,
      "training_step_time": 8.110866069793701
    },
    {
      "epoch": 1.6798973083496092e-05,
      "model_forward_time": 1.7117726802825928,
      "step": 17615
    },
    {
      "epoch": 1.6798973083496092e-05,
      "step": 17615,
      "training_step_time": 9.795518398284912
    },
    {
      "epoch": 1.67999267578125e-05,
      "model_forward_time": 1.7100982666015625,
      "step": 17616
    },
    {
      "epoch": 1.67999267578125e-05,
      "step": 17616,
      "training_step_time": 8.4944486618042
    },
    {
      "epoch": 1.6800880432128907e-05,
      "model_forward_time": 1.3380274772644043,
      "step": 17617
    },
    {
      "epoch": 1.6800880432128907e-05,
      "step": 17617,
      "training_step_time": 9.027239561080933
    },
    {
      "epoch": 1.680183410644531e-05,
      "model_forward_time": 1.7254269123077393,
      "step": 17618
    },
    {
      "epoch": 1.680183410644531e-05,
      "step": 17618,
      "training_step_time": 7.5932886600494385
    },
    {
      "epoch": 1.680278778076172e-05,
      "model_forward_time": 1.711287498474121,
      "step": 17619
    },
    {
      "epoch": 1.680278778076172e-05,
      "step": 17619,
      "training_step_time": 7.537277698516846
    },
    {
      "epoch": 1.6803741455078126e-05,
      "grad_norm": 0.14951759576797485,
      "learning_rate": 9.57092732296822e-06,
      "loss": 0.0687,
      "step": 17620
    },
    {
      "epoch": 1.6803741455078126e-05,
      "model_forward_time": 1.7121162414550781,
      "step": 17620
    },
    {
      "dynamics_loss_avg": 0.06272411718964577,
      "epoch": 1.6803741455078126e-05,
      "step": 17620
    },
    {
      "action_loss_avg": 0.0038977280026301743,
      "epoch": 1.6803741455078126e-05,
      "step": 17620
    },
    {
      "epoch": 1.6803741455078126e-05,
      "step": 17620,
      "training_step_time": 8.482811450958252
    },
    {
      "epoch": 1.680469512939453e-05,
      "model_forward_time": 1.7134475708007812,
      "step": 17621
    },
    {
      "epoch": 1.680469512939453e-05,
      "step": 17621,
      "training_step_time": 9.003848552703857
    },
    {
      "epoch": 1.6805648803710937e-05,
      "model_forward_time": 1.716454267501831,
      "step": 17622
    },
    {
      "epoch": 1.6805648803710937e-05,
      "step": 17622,
      "training_step_time": 8.169776916503906
    },
    {
      "epoch": 1.6806602478027345e-05,
      "model_forward_time": 1.766371250152588,
      "step": 17623
    },
    {
      "epoch": 1.6806602478027345e-05,
      "step": 17623,
      "training_step_time": 8.38678526878357
    },
    {
      "epoch": 1.680755615234375e-05,
      "model_forward_time": 1.71291184425354,
      "step": 17624
    },
    {
      "epoch": 1.680755615234375e-05,
      "step": 17624,
      "training_step_time": 7.6727588176727295
    },
    {
      "epoch": 1.6808509826660156e-05,
      "model_forward_time": 2.2592954635620117,
      "step": 17625
    },
    {
      "epoch": 1.6808509826660156e-05,
      "step": 17625,
      "training_step_time": 7.738026142120361
    },
    {
      "epoch": 1.6809463500976564e-05,
      "model_forward_time": 1.713327169418335,
      "step": 17626
    },
    {
      "epoch": 1.6809463500976564e-05,
      "step": 17626,
      "training_step_time": 9.03117322921753
    },
    {
      "epoch": 1.6810417175292968e-05,
      "model_forward_time": 1.7082176208496094,
      "step": 17627
    },
    {
      "epoch": 1.6810417175292968e-05,
      "step": 17627,
      "training_step_time": 7.658305883407593
    },
    {
      "epoch": 1.6811370849609375e-05,
      "model_forward_time": 1.711721658706665,
      "step": 17628
    },
    {
      "epoch": 1.6811370849609375e-05,
      "step": 17628,
      "training_step_time": 7.644663572311401
    },
    {
      "epoch": 1.6812324523925783e-05,
      "model_forward_time": 1.7095930576324463,
      "step": 17629
    },
    {
      "epoch": 1.6812324523925783e-05,
      "step": 17629,
      "training_step_time": 7.641740322113037
    },
    {
      "epoch": 1.6813278198242187e-05,
      "grad_norm": 0.18899279832839966,
      "learning_rate": 9.570256928161385e-06,
      "loss": 0.0677,
      "step": 17630
    },
    {
      "epoch": 1.6813278198242187e-05,
      "model_forward_time": 1.7126891613006592,
      "step": 17630
    },
    {
      "dynamics_loss_avg": 0.06502795517444611,
      "epoch": 1.6813278198242187e-05,
      "step": 17630
    },
    {
      "action_loss_avg": 0.0025017109990585597,
      "epoch": 1.6813278198242187e-05,
      "step": 17630
    },
    {
      "epoch": 1.6813278198242187e-05,
      "step": 17630,
      "training_step_time": 7.688150644302368
    },
    {
      "epoch": 1.6814231872558594e-05,
      "model_forward_time": 1.715198040008545,
      "step": 17631
    },
    {
      "epoch": 1.6814231872558594e-05,
      "step": 17631,
      "training_step_time": 8.478409051895142
    },
    {
      "epoch": 1.6815185546875e-05,
      "model_forward_time": 1.7627263069152832,
      "step": 17632
    },
    {
      "epoch": 1.6815185546875e-05,
      "step": 17632,
      "training_step_time": 7.712855815887451
    },
    {
      "epoch": 1.6816139221191405e-05,
      "model_forward_time": 1.7134292125701904,
      "step": 17633
    },
    {
      "epoch": 1.6816139221191405e-05,
      "step": 17633,
      "training_step_time": 8.247268676757812
    },
    {
      "epoch": 1.6817092895507813e-05,
      "model_forward_time": 2.285755157470703,
      "step": 17634
    },
    {
      "epoch": 1.6817092895507813e-05,
      "step": 17634,
      "training_step_time": 7.81312370300293
    },
    {
      "epoch": 1.681804656982422e-05,
      "model_forward_time": 1.7262730598449707,
      "step": 17635
    },
    {
      "epoch": 1.681804656982422e-05,
      "step": 17635,
      "training_step_time": 8.663318157196045
    },
    {
      "epoch": 1.6819000244140624e-05,
      "model_forward_time": 1.7017176151275635,
      "step": 17636
    },
    {
      "epoch": 1.6819000244140624e-05,
      "step": 17636,
      "training_step_time": 7.659220218658447
    },
    {
      "epoch": 1.681995391845703e-05,
      "model_forward_time": 1.714268684387207,
      "step": 17637
    },
    {
      "epoch": 1.681995391845703e-05,
      "step": 17637,
      "training_step_time": 7.517837047576904
    },
    {
      "epoch": 1.682090759277344e-05,
      "model_forward_time": 1.7108092308044434,
      "step": 17638
    },
    {
      "epoch": 1.682090759277344e-05,
      "step": 17638,
      "training_step_time": 7.749618291854858
    },
    {
      "epoch": 1.6821861267089843e-05,
      "model_forward_time": 1.721776008605957,
      "step": 17639
    },
    {
      "epoch": 1.6821861267089843e-05,
      "step": 17639,
      "training_step_time": 8.212597131729126
    },
    {
      "epoch": 1.682281494140625e-05,
      "grad_norm": 0.17309896647930145,
      "learning_rate": 9.569586033558126e-06,
      "loss": 0.0628,
      "step": 17640
    },
    {
      "epoch": 1.682281494140625e-05,
      "model_forward_time": 1.691622018814087,
      "step": 17640
    },
    {
      "dynamics_loss_avg": 0.06695756958797575,
      "epoch": 1.682281494140625e-05,
      "step": 17640
    },
    {
      "action_loss_avg": 0.002933017088798806,
      "epoch": 1.682281494140625e-05,
      "step": 17640
    },
    {
      "epoch": 1.682281494140625e-05,
      "step": 17640,
      "training_step_time": 10.618952751159668
    },
    {
      "epoch": 1.6823768615722655e-05,
      "model_forward_time": 1.7482762336730957,
      "step": 17641
    },
    {
      "epoch": 1.6823768615722655e-05,
      "step": 17641,
      "training_step_time": 7.694383859634399
    },
    {
      "epoch": 1.6824722290039062e-05,
      "model_forward_time": 1.7205851078033447,
      "step": 17642
    },
    {
      "epoch": 1.6824722290039062e-05,
      "step": 17642,
      "training_step_time": 8.38904070854187
    },
    {
      "epoch": 1.682567596435547e-05,
      "model_forward_time": 2.268474578857422,
      "step": 17643
    },
    {
      "epoch": 1.682567596435547e-05,
      "step": 17643,
      "training_step_time": 7.831909894943237
    },
    {
      "epoch": 1.6826629638671873e-05,
      "model_forward_time": 1.7088603973388672,
      "step": 17644
    },
    {
      "epoch": 1.6826629638671873e-05,
      "step": 17644,
      "training_step_time": 7.7044289112091064
    },
    {
      "epoch": 1.682758331298828e-05,
      "model_forward_time": 1.7045576572418213,
      "step": 17645
    },
    {
      "epoch": 1.682758331298828e-05,
      "step": 17645,
      "training_step_time": 7.627596855163574
    },
    {
      "epoch": 1.6828536987304688e-05,
      "model_forward_time": 1.7197175025939941,
      "step": 17646
    },
    {
      "epoch": 1.6828536987304688e-05,
      "step": 17646,
      "training_step_time": 7.7046332359313965
    },
    {
      "epoch": 1.6829490661621092e-05,
      "model_forward_time": 1.7087106704711914,
      "step": 17647
    },
    {
      "epoch": 1.6829490661621092e-05,
      "step": 17647,
      "training_step_time": 7.669826507568359
    },
    {
      "epoch": 1.68304443359375e-05,
      "model_forward_time": 1.7116408348083496,
      "step": 17648
    },
    {
      "epoch": 1.68304443359375e-05,
      "step": 17648,
      "training_step_time": 9.32073163986206
    },
    {
      "epoch": 1.6831398010253907e-05,
      "model_forward_time": 1.7069907188415527,
      "step": 17649
    },
    {
      "epoch": 1.6831398010253907e-05,
      "step": 17649,
      "training_step_time": 8.59714388847351
    },
    {
      "epoch": 1.683235168457031e-05,
      "grad_norm": 0.1485503762960434,
      "learning_rate": 9.568914639231807e-06,
      "loss": 0.0658,
      "step": 17650
    },
    {
      "epoch": 1.683235168457031e-05,
      "model_forward_time": 1.3419244289398193,
      "step": 17650
    },
    {
      "dynamics_loss_avg": 0.07167757842689752,
      "epoch": 1.683235168457031e-05,
      "step": 17650
    },
    {
      "action_loss_avg": 0.0033616570639424027,
      "epoch": 1.683235168457031e-05,
      "step": 17650
    },
    {
      "epoch": 1.683235168457031e-05,
      "step": 17650,
      "training_step_time": 6.7855072021484375
    },
    {
      "epoch": 1.683330535888672e-05,
      "model_forward_time": 1.7116410732269287,
      "step": 17651
    },
    {
      "epoch": 1.683330535888672e-05,
      "step": 17651,
      "training_step_time": 6.79130482673645
    },
    {
      "epoch": 1.6834259033203126e-05,
      "model_forward_time": 1.7138516902923584,
      "step": 17652
    },
    {
      "epoch": 1.6834259033203126e-05,
      "step": 17652,
      "training_step_time": 6.825434684753418
    },
    {
      "epoch": 1.683521270751953e-05,
      "model_forward_time": 1.7230420112609863,
      "step": 17653
    },
    {
      "epoch": 1.683521270751953e-05,
      "step": 17653,
      "training_step_time": 6.800329685211182
    },
    {
      "epoch": 1.6836166381835937e-05,
      "model_forward_time": 1.7347943782806396,
      "step": 17654
    },
    {
      "epoch": 1.6836166381835937e-05,
      "step": 17654,
      "training_step_time": 6.828943729400635
    },
    {
      "epoch": 1.6837120056152345e-05,
      "model_forward_time": 1.7292206287384033,
      "step": 17655
    },
    {
      "epoch": 1.6837120056152345e-05,
      "step": 17655,
      "training_step_time": 6.7066216468811035
    },
    {
      "epoch": 1.683807373046875e-05,
      "model_forward_time": 1.7315328121185303,
      "step": 17656
    },
    {
      "epoch": 1.683807373046875e-05,
      "step": 17656,
      "training_step_time": 6.706989765167236
    },
    {
      "epoch": 1.6839027404785156e-05,
      "model_forward_time": 1.731062412261963,
      "step": 17657
    },
    {
      "epoch": 1.6839027404785156e-05,
      "step": 17657,
      "training_step_time": 6.783883810043335
    },
    {
      "epoch": 1.6839981079101564e-05,
      "model_forward_time": 2.2594962120056152,
      "step": 17658
    },
    {
      "epoch": 1.6839981079101564e-05,
      "step": 17658,
      "training_step_time": 10.623308181762695
    },
    {
      "epoch": 1.6840934753417968e-05,
      "model_forward_time": 1.7251014709472656,
      "step": 17659
    },
    {
      "epoch": 1.6840934753417968e-05,
      "step": 17659,
      "training_step_time": 7.8275017738342285
    },
    {
      "epoch": 1.6841888427734375e-05,
      "grad_norm": 0.1766999363899231,
      "learning_rate": 9.568242745255853e-06,
      "loss": 0.0638,
      "step": 17660
    },
    {
      "epoch": 1.6841888427734375e-05,
      "model_forward_time": 1.7361171245574951,
      "step": 17660
    },
    {
      "dynamics_loss_avg": 0.06655368059873581,
      "epoch": 1.6841888427734375e-05,
      "step": 17660
    },
    {
      "action_loss_avg": 0.003873730404302478,
      "epoch": 1.6841888427734375e-05,
      "step": 17660
    },
    {
      "epoch": 1.6841888427734375e-05,
      "step": 17660,
      "training_step_time": 6.809506177902222
    },
    {
      "epoch": 1.6842842102050783e-05,
      "model_forward_time": 1.72841215133667,
      "step": 17661
    },
    {
      "epoch": 1.6842842102050783e-05,
      "step": 17661,
      "training_step_time": 6.812891483306885
    },
    {
      "epoch": 1.6843795776367187e-05,
      "model_forward_time": 1.7109835147857666,
      "step": 17662
    },
    {
      "epoch": 1.6843795776367187e-05,
      "step": 17662,
      "training_step_time": 6.834756851196289
    },
    {
      "epoch": 1.6844749450683594e-05,
      "model_forward_time": 1.7241652011871338,
      "step": 17663
    },
    {
      "epoch": 1.6844749450683594e-05,
      "step": 17663,
      "training_step_time": 6.850645303726196
    },
    {
      "epoch": 1.6845703125e-05,
      "model_forward_time": 1.7163383960723877,
      "step": 17664
    },
    {
      "epoch": 1.6845703125e-05,
      "step": 17664,
      "training_step_time": 6.793698072433472
    },
    {
      "epoch": 1.6846656799316405e-05,
      "model_forward_time": 1.722599744796753,
      "step": 17665
    },
    {
      "epoch": 1.6846656799316405e-05,
      "step": 17665,
      "training_step_time": 6.824324607849121
    },
    {
      "epoch": 1.6847610473632813e-05,
      "model_forward_time": 1.7337894439697266,
      "step": 17666
    },
    {
      "epoch": 1.6847610473632813e-05,
      "step": 17666,
      "training_step_time": 6.810532569885254
    },
    {
      "epoch": 1.684856414794922e-05,
      "model_forward_time": 2.2817413806915283,
      "step": 17667
    },
    {
      "epoch": 1.684856414794922e-05,
      "step": 17667,
      "training_step_time": 9.935943126678467
    },
    {
      "epoch": 1.6849517822265624e-05,
      "model_forward_time": 1.7191286087036133,
      "step": 17668
    },
    {
      "epoch": 1.6849517822265624e-05,
      "step": 17668,
      "training_step_time": 7.833108901977539
    },
    {
      "epoch": 1.6850471496582032e-05,
      "model_forward_time": 1.718592882156372,
      "step": 17669
    },
    {
      "epoch": 1.6850471496582032e-05,
      "step": 17669,
      "training_step_time": 6.726120233535767
    },
    {
      "epoch": 1.685142517089844e-05,
      "grad_norm": 0.27707621455192566,
      "learning_rate": 9.567570351703739e-06,
      "loss": 0.0657,
      "step": 17670
    },
    {
      "epoch": 1.685142517089844e-05,
      "model_forward_time": 1.7202799320220947,
      "step": 17670
    },
    {
      "dynamics_loss_avg": 0.0791453154757619,
      "epoch": 1.685142517089844e-05,
      "step": 17670
    },
    {
      "action_loss_avg": 0.0038570195669308306,
      "epoch": 1.685142517089844e-05,
      "step": 17670
    },
    {
      "epoch": 1.685142517089844e-05,
      "step": 17670,
      "training_step_time": 6.757763385772705
    },
    {
      "epoch": 1.6852378845214843e-05,
      "model_forward_time": 1.351264238357544,
      "step": 17671
    },
    {
      "epoch": 1.6852378845214843e-05,
      "step": 17671,
      "training_step_time": 6.795067548751831
    },
    {
      "epoch": 1.685333251953125e-05,
      "model_forward_time": 1.7226276397705078,
      "step": 17672
    },
    {
      "epoch": 1.685333251953125e-05,
      "step": 17672,
      "training_step_time": 6.75592565536499
    },
    {
      "epoch": 1.6854286193847655e-05,
      "model_forward_time": 1.7261970043182373,
      "step": 17673
    },
    {
      "epoch": 1.6854286193847655e-05,
      "step": 17673,
      "training_step_time": 6.8105645179748535
    },
    {
      "epoch": 1.6855239868164062e-05,
      "model_forward_time": 1.720867395401001,
      "step": 17674
    },
    {
      "epoch": 1.6855239868164062e-05,
      "step": 17674,
      "training_step_time": 6.814107179641724
    },
    {
      "epoch": 1.685619354248047e-05,
      "model_forward_time": 1.720552682876587,
      "step": 17675
    },
    {
      "epoch": 1.685619354248047e-05,
      "step": 17675,
      "training_step_time": 6.766865491867065
    },
    {
      "epoch": 1.6857147216796873e-05,
      "model_forward_time": 1.722703456878662,
      "step": 17676
    },
    {
      "epoch": 1.6857147216796873e-05,
      "step": 17676,
      "training_step_time": 9.106520652770996
    },
    {
      "epoch": 1.685810089111328e-05,
      "model_forward_time": 1.7235114574432373,
      "step": 17677
    },
    {
      "epoch": 1.685810089111328e-05,
      "step": 17677,
      "training_step_time": 7.922124624252319
    },
    {
      "epoch": 1.6859054565429688e-05,
      "model_forward_time": 1.71921706199646,
      "step": 17678
    },
    {
      "epoch": 1.6859054565429688e-05,
      "step": 17678,
      "training_step_time": 7.887172698974609
    },
    {
      "epoch": 1.6860008239746092e-05,
      "model_forward_time": 1.7204387187957764,
      "step": 17679
    },
    {
      "epoch": 1.6860008239746092e-05,
      "step": 17679,
      "training_step_time": 6.914222240447998
    },
    {
      "epoch": 1.68609619140625e-05,
      "grad_norm": 0.1980711817741394,
      "learning_rate": 9.566897458649e-06,
      "loss": 0.061,
      "step": 17680
    },
    {
      "epoch": 1.68609619140625e-05,
      "model_forward_time": 1.7206611633300781,
      "step": 17680
    },
    {
      "dynamics_loss_avg": 0.06276278086006641,
      "epoch": 1.68609619140625e-05,
      "step": 17680
    },
    {
      "action_loss_avg": 0.0028725103824399413,
      "epoch": 1.68609619140625e-05,
      "step": 17680
    },
    {
      "epoch": 1.68609619140625e-05,
      "step": 17680,
      "training_step_time": 6.859094619750977
    },
    {
      "epoch": 1.6861915588378907e-05,
      "model_forward_time": 1.7245635986328125,
      "step": 17681
    },
    {
      "epoch": 1.6861915588378907e-05,
      "step": 17681,
      "training_step_time": 6.846859693527222
    },
    {
      "epoch": 1.686286926269531e-05,
      "model_forward_time": 1.7235419750213623,
      "step": 17682
    },
    {
      "epoch": 1.686286926269531e-05,
      "step": 17682,
      "training_step_time": 6.8504111766815186
    },
    {
      "epoch": 1.686382293701172e-05,
      "model_forward_time": 1.7141611576080322,
      "step": 17683
    },
    {
      "epoch": 1.686382293701172e-05,
      "step": 17683,
      "training_step_time": 6.784628629684448
    },
    {
      "epoch": 1.6864776611328126e-05,
      "model_forward_time": 1.7276830673217773,
      "step": 17684
    },
    {
      "epoch": 1.6864776611328126e-05,
      "step": 17684,
      "training_step_time": 6.833019256591797
    },
    {
      "epoch": 1.686573028564453e-05,
      "model_forward_time": 1.7348086833953857,
      "step": 17685
    },
    {
      "epoch": 1.686573028564453e-05,
      "step": 17685,
      "training_step_time": 8.245330810546875
    },
    {
      "epoch": 1.6866683959960937e-05,
      "model_forward_time": 2.261747360229492,
      "step": 17686
    },
    {
      "epoch": 1.6866683959960937e-05,
      "step": 17686,
      "training_step_time": 10.256516933441162
    },
    {
      "epoch": 1.6867637634277345e-05,
      "model_forward_time": 1.7259650230407715,
      "step": 17687
    },
    {
      "epoch": 1.6867637634277345e-05,
      "step": 17687,
      "training_step_time": 7.790982484817505
    },
    {
      "epoch": 1.686859130859375e-05,
      "model_forward_time": 1.723623514175415,
      "step": 17688
    },
    {
      "epoch": 1.686859130859375e-05,
      "step": 17688,
      "training_step_time": 6.803821563720703
    },
    {
      "epoch": 1.6869544982910156e-05,
      "model_forward_time": 1.7216620445251465,
      "step": 17689
    },
    {
      "epoch": 1.6869544982910156e-05,
      "step": 17689,
      "training_step_time": 6.799190998077393
    },
    {
      "epoch": 1.6870498657226564e-05,
      "grad_norm": 0.18258138000965118,
      "learning_rate": 9.56622406616522e-06,
      "loss": 0.0713,
      "step": 17690
    },
    {
      "epoch": 1.6870498657226564e-05,
      "model_forward_time": 1.728532075881958,
      "step": 17690
    },
    {
      "dynamics_loss_avg": 0.06908545680344105,
      "epoch": 1.6870498657226564e-05,
      "step": 17690
    },
    {
      "action_loss_avg": 0.003962178330402822,
      "epoch": 1.6870498657226564e-05,
      "step": 17690
    },
    {
      "epoch": 1.6870498657226564e-05,
      "step": 17690,
      "training_step_time": 6.840734958648682
    },
    {
      "epoch": 1.6871452331542968e-05,
      "model_forward_time": 1.7288637161254883,
      "step": 17691
    },
    {
      "epoch": 1.6871452331542968e-05,
      "step": 17691,
      "training_step_time": 6.826002597808838
    },
    {
      "epoch": 1.6872406005859375e-05,
      "model_forward_time": 1.7169241905212402,
      "step": 17692
    },
    {
      "epoch": 1.6872406005859375e-05,
      "step": 17692,
      "training_step_time": 6.759007930755615
    },
    {
      "epoch": 1.6873359680175783e-05,
      "model_forward_time": 1.7255580425262451,
      "step": 17693
    },
    {
      "epoch": 1.6873359680175783e-05,
      "step": 17693,
      "training_step_time": 6.810397386550903
    },
    {
      "epoch": 1.6874313354492187e-05,
      "model_forward_time": 1.7291545867919922,
      "step": 17694
    },
    {
      "epoch": 1.6874313354492187e-05,
      "step": 17694,
      "training_step_time": 6.802103519439697
    },
    {
      "epoch": 1.6875267028808594e-05,
      "model_forward_time": 2.2707598209381104,
      "step": 17695
    },
    {
      "epoch": 1.6875267028808594e-05,
      "step": 17695,
      "training_step_time": 9.435341119766235
    },
    {
      "epoch": 1.6876220703125e-05,
      "model_forward_time": 1.7258174419403076,
      "step": 17696
    },
    {
      "epoch": 1.6876220703125e-05,
      "step": 17696,
      "training_step_time": 7.7279276847839355
    },
    {
      "epoch": 1.6877174377441405e-05,
      "model_forward_time": 1.732656478881836,
      "step": 17697
    },
    {
      "epoch": 1.6877174377441405e-05,
      "step": 17697,
      "training_step_time": 7.695465564727783
    },
    {
      "epoch": 1.6878128051757813e-05,
      "model_forward_time": 1.7259416580200195,
      "step": 17698
    },
    {
      "epoch": 1.6878128051757813e-05,
      "step": 17698,
      "training_step_time": 6.7454304695129395
    },
    {
      "epoch": 1.687908172607422e-05,
      "model_forward_time": 1.726750373840332,
      "step": 17699
    },
    {
      "epoch": 1.687908172607422e-05,
      "step": 17699,
      "training_step_time": 6.754656791687012
    },
    {
      "epoch": 1.6880035400390624e-05,
      "grad_norm": 0.1932796984910965,
      "learning_rate": 9.565550174326042e-06,
      "loss": 0.0698,
      "step": 17700
    },
    {
      "epoch": 1.6880035400390624e-05,
      "model_forward_time": 1.7297942638397217,
      "step": 17700
    },
    {
      "dynamics_loss_avg": 0.06226847656071186,
      "epoch": 1.6880035400390624e-05,
      "step": 17700
    },
    {
      "action_loss_avg": 0.0028847756038885564,
      "epoch": 1.6880035400390624e-05,
      "step": 17700
    },
    {
      "epoch": 1.6880035400390624e-05,
      "step": 17700,
      "training_step_time": 6.742624521255493
    },
    {
      "epoch": 1.6880989074707032e-05,
      "model_forward_time": 1.7295098304748535,
      "step": 17701
    },
    {
      "epoch": 1.6880989074707032e-05,
      "step": 17701,
      "training_step_time": 6.804265975952148
    },
    {
      "epoch": 1.688194274902344e-05,
      "model_forward_time": 1.720578670501709,
      "step": 17702
    },
    {
      "epoch": 1.688194274902344e-05,
      "step": 17702,
      "training_step_time": 6.774756669998169
    },
    {
      "epoch": 1.6882896423339843e-05,
      "model_forward_time": 1.7274634838104248,
      "step": 17703
    },
    {
      "epoch": 1.6882896423339843e-05,
      "step": 17703,
      "training_step_time": 6.764428377151489
    },
    {
      "epoch": 1.688385009765625e-05,
      "model_forward_time": 2.277832269668579,
      "step": 17704
    },
    {
      "epoch": 1.688385009765625e-05,
      "step": 17704,
      "training_step_time": 10.180086374282837
    },
    {
      "epoch": 1.6884803771972655e-05,
      "model_forward_time": 1.7287797927856445,
      "step": 17705
    },
    {
      "epoch": 1.6884803771972655e-05,
      "step": 17705,
      "training_step_time": 8.836429595947266
    },
    {
      "epoch": 1.6885757446289062e-05,
      "model_forward_time": 1.7269253730773926,
      "step": 17706
    },
    {
      "epoch": 1.6885757446289062e-05,
      "step": 17706,
      "training_step_time": 7.7281653881073
    },
    {
      "epoch": 1.688671112060547e-05,
      "model_forward_time": 1.7213895320892334,
      "step": 17707
    },
    {
      "epoch": 1.688671112060547e-05,
      "step": 17707,
      "training_step_time": 6.752637624740601
    },
    {
      "epoch": 1.6887664794921873e-05,
      "model_forward_time": 1.7254257202148438,
      "step": 17708
    },
    {
      "epoch": 1.6887664794921873e-05,
      "step": 17708,
      "training_step_time": 6.7391581535339355
    },
    {
      "epoch": 1.688861846923828e-05,
      "model_forward_time": 1.3565080165863037,
      "step": 17709
    },
    {
      "epoch": 1.688861846923828e-05,
      "step": 17709,
      "training_step_time": 6.789421081542969
    },
    {
      "epoch": 1.688957214355469e-05,
      "grad_norm": 0.23762232065200806,
      "learning_rate": 9.564875783205161e-06,
      "loss": 0.0638,
      "step": 17710
    },
    {
      "epoch": 1.688957214355469e-05,
      "model_forward_time": 1.7263591289520264,
      "step": 17710
    },
    {
      "dynamics_loss_avg": 0.055235986970365046,
      "epoch": 1.688957214355469e-05,
      "step": 17710
    },
    {
      "action_loss_avg": 0.0026017511496320366,
      "epoch": 1.688957214355469e-05,
      "step": 17710
    },
    {
      "epoch": 1.688957214355469e-05,
      "step": 17710,
      "training_step_time": 6.797502279281616
    },
    {
      "epoch": 1.6890525817871092e-05,
      "model_forward_time": 1.7359139919281006,
      "step": 17711
    },
    {
      "epoch": 1.6890525817871092e-05,
      "step": 17711,
      "training_step_time": 6.757957696914673
    },
    {
      "epoch": 1.68914794921875e-05,
      "model_forward_time": 1.726534366607666,
      "step": 17712
    },
    {
      "epoch": 1.68914794921875e-05,
      "step": 17712,
      "training_step_time": 6.803101301193237
    },
    {
      "epoch": 1.6892433166503907e-05,
      "model_forward_time": 2.276036024093628,
      "step": 17713
    },
    {
      "epoch": 1.6892433166503907e-05,
      "step": 17713,
      "training_step_time": 9.23752737045288
    },
    {
      "epoch": 1.689338684082031e-05,
      "model_forward_time": 1.7250726222991943,
      "step": 17714
    },
    {
      "epoch": 1.689338684082031e-05,
      "step": 17714,
      "training_step_time": 7.737310409545898
    },
    {
      "epoch": 1.689434051513672e-05,
      "model_forward_time": 1.7230422496795654,
      "step": 17715
    },
    {
      "epoch": 1.689434051513672e-05,
      "step": 17715,
      "training_step_time": 7.685186386108398
    },
    {
      "epoch": 1.6895294189453126e-05,
      "model_forward_time": 1.7251572608947754,
      "step": 17716
    },
    {
      "epoch": 1.6895294189453126e-05,
      "step": 17716,
      "training_step_time": 6.768568754196167
    },
    {
      "epoch": 1.689624786376953e-05,
      "model_forward_time": 1.7307608127593994,
      "step": 17717
    },
    {
      "epoch": 1.689624786376953e-05,
      "step": 17717,
      "training_step_time": 6.771799564361572
    },
    {
      "epoch": 1.6897201538085937e-05,
      "model_forward_time": 1.7242720127105713,
      "step": 17718
    },
    {
      "epoch": 1.6897201538085937e-05,
      "step": 17718,
      "training_step_time": 6.78102970123291
    },
    {
      "epoch": 1.6898155212402345e-05,
      "model_forward_time": 1.7283077239990234,
      "step": 17719
    },
    {
      "epoch": 1.6898155212402345e-05,
      "step": 17719,
      "training_step_time": 6.798825979232788
    },
    {
      "epoch": 1.689910888671875e-05,
      "grad_norm": 0.21478688716888428,
      "learning_rate": 9.56420089287633e-06,
      "loss": 0.0655,
      "step": 17720
    },
    {
      "epoch": 1.689910888671875e-05,
      "model_forward_time": 1.7225346565246582,
      "step": 17720
    },
    {
      "dynamics_loss_avg": 0.06468518376350403,
      "epoch": 1.689910888671875e-05,
      "step": 17720
    },
    {
      "action_loss_avg": 0.0032301908009685575,
      "epoch": 1.689910888671875e-05,
      "step": 17720
    },
    {
      "epoch": 1.689910888671875e-05,
      "step": 17720,
      "training_step_time": 6.859910011291504
    },
    {
      "epoch": 1.6900062561035156e-05,
      "model_forward_time": 1.7258434295654297,
      "step": 17721
    },
    {
      "epoch": 1.6900062561035156e-05,
      "step": 17721,
      "training_step_time": 6.805117845535278
    },
    {
      "epoch": 1.6901016235351564e-05,
      "model_forward_time": 1.7382144927978516,
      "step": 17722
    },
    {
      "epoch": 1.6901016235351564e-05,
      "step": 17722,
      "training_step_time": 9.673943996429443
    },
    {
      "epoch": 1.6901969909667968e-05,
      "model_forward_time": 2.26644229888916,
      "step": 17723
    },
    {
      "epoch": 1.6901969909667968e-05,
      "step": 17723,
      "training_step_time": 7.782266855239868
    },
    {
      "epoch": 1.6902923583984375e-05,
      "model_forward_time": 1.7249834537506104,
      "step": 17724
    },
    {
      "epoch": 1.6902923583984375e-05,
      "step": 17724,
      "training_step_time": 7.7908384799957275
    },
    {
      "epoch": 1.6903877258300783e-05,
      "model_forward_time": 1.7189993858337402,
      "step": 17725
    },
    {
      "epoch": 1.6903877258300783e-05,
      "step": 17725,
      "training_step_time": 6.782446622848511
    },
    {
      "epoch": 1.6904830932617187e-05,
      "model_forward_time": 1.729548692703247,
      "step": 17726
    },
    {
      "epoch": 1.6904830932617187e-05,
      "step": 17726,
      "training_step_time": 6.738138437271118
    },
    {
      "epoch": 1.6905784606933594e-05,
      "model_forward_time": 1.7292604446411133,
      "step": 17727
    },
    {
      "epoch": 1.6905784606933594e-05,
      "step": 17727,
      "training_step_time": 6.758438348770142
    },
    {
      "epoch": 1.690673828125e-05,
      "model_forward_time": 1.7236390113830566,
      "step": 17728
    },
    {
      "epoch": 1.690673828125e-05,
      "step": 17728,
      "training_step_time": 6.771507740020752
    },
    {
      "epoch": 1.6907691955566405e-05,
      "model_forward_time": 1.7226221561431885,
      "step": 17729
    },
    {
      "epoch": 1.6907691955566405e-05,
      "step": 17729,
      "training_step_time": 6.773881196975708
    },
    {
      "epoch": 1.6908645629882813e-05,
      "grad_norm": 0.18580330908298492,
      "learning_rate": 9.563525503413349e-06,
      "loss": 0.0656,
      "step": 17730
    },
    {
      "epoch": 1.6908645629882813e-05,
      "model_forward_time": 1.7291171550750732,
      "step": 17730
    },
    {
      "dynamics_loss_avg": 0.06798886954784393,
      "epoch": 1.6908645629882813e-05,
      "step": 17730
    },
    {
      "action_loss_avg": 0.002373912592884153,
      "epoch": 1.6908645629882813e-05,
      "step": 17730
    },
    {
      "epoch": 1.6908645629882813e-05,
      "step": 17730,
      "training_step_time": 6.728902578353882
    },
    {
      "epoch": 1.690959930419922e-05,
      "model_forward_time": 1.7209434509277344,
      "step": 17731
    },
    {
      "epoch": 1.690959930419922e-05,
      "step": 17731,
      "training_step_time": 8.621531248092651
    },
    {
      "epoch": 1.6910552978515624e-05,
      "model_forward_time": 2.2663540840148926,
      "step": 17732
    },
    {
      "epoch": 1.6910552978515624e-05,
      "step": 17732,
      "training_step_time": 9.839177131652832
    },
    {
      "epoch": 1.6911506652832032e-05,
      "model_forward_time": 1.71978759765625,
      "step": 17733
    },
    {
      "epoch": 1.6911506652832032e-05,
      "step": 17733,
      "training_step_time": 7.797767162322998
    },
    {
      "epoch": 1.691246032714844e-05,
      "model_forward_time": 1.7120966911315918,
      "step": 17734
    },
    {
      "epoch": 1.691246032714844e-05,
      "step": 17734,
      "training_step_time": 7.64493989944458
    },
    {
      "epoch": 1.6913414001464843e-05,
      "model_forward_time": 1.734945297241211,
      "step": 17735
    },
    {
      "epoch": 1.6913414001464843e-05,
      "step": 17735,
      "training_step_time": 6.8032872676849365
    },
    {
      "epoch": 1.691436767578125e-05,
      "model_forward_time": 1.7355058193206787,
      "step": 17736
    },
    {
      "epoch": 1.691436767578125e-05,
      "step": 17736,
      "training_step_time": 6.67513108253479
    },
    {
      "epoch": 1.6915321350097655e-05,
      "model_forward_time": 1.718052864074707,
      "step": 17737
    },
    {
      "epoch": 1.6915321350097655e-05,
      "step": 17737,
      "training_step_time": 6.772934913635254
    },
    {
      "epoch": 1.6916275024414062e-05,
      "model_forward_time": 1.7236535549163818,
      "step": 17738
    },
    {
      "epoch": 1.6916275024414062e-05,
      "step": 17738,
      "training_step_time": 6.755643367767334
    },
    {
      "epoch": 1.691722869873047e-05,
      "model_forward_time": 1.715698003768921,
      "step": 17739
    },
    {
      "epoch": 1.691722869873047e-05,
      "step": 17739,
      "training_step_time": 6.805055618286133
    },
    {
      "epoch": 1.6918182373046874e-05,
      "grad_norm": 0.14260874688625336,
      "learning_rate": 9.56284961489008e-06,
      "loss": 0.0614,
      "step": 17740
    },
    {
      "epoch": 1.6918182373046874e-05,
      "model_forward_time": 1.7287240028381348,
      "step": 17740
    },
    {
      "dynamics_loss_avg": 0.0817572295665741,
      "epoch": 1.6918182373046874e-05,
      "step": 17740
    },
    {
      "action_loss_avg": 0.003151120332768187,
      "epoch": 1.6918182373046874e-05,
      "step": 17740
    },
    {
      "epoch": 1.6918182373046874e-05,
      "step": 17740,
      "training_step_time": 7.718425035476685
    },
    {
      "epoch": 1.691913604736328e-05,
      "model_forward_time": 2.2770256996154785,
      "step": 17741
    },
    {
      "epoch": 1.691913604736328e-05,
      "step": 17741,
      "training_step_time": 8.76542067527771
    },
    {
      "epoch": 1.692008972167969e-05,
      "model_forward_time": 1.7191112041473389,
      "step": 17742
    },
    {
      "epoch": 1.692008972167969e-05,
      "step": 17742,
      "training_step_time": 7.846198320388794
    },
    {
      "epoch": 1.6921043395996092e-05,
      "model_forward_time": 1.7227425575256348,
      "step": 17743
    },
    {
      "epoch": 1.6921043395996092e-05,
      "step": 17743,
      "training_step_time": 7.676941633224487
    },
    {
      "epoch": 1.69219970703125e-05,
      "model_forward_time": 1.7265384197235107,
      "step": 17744
    },
    {
      "epoch": 1.69219970703125e-05,
      "step": 17744,
      "training_step_time": 6.877875566482544
    },
    {
      "epoch": 1.6922950744628907e-05,
      "model_forward_time": 1.724532127380371,
      "step": 17745
    },
    {
      "epoch": 1.6922950744628907e-05,
      "step": 17745,
      "training_step_time": 6.726508855819702
    },
    {
      "epoch": 1.692390441894531e-05,
      "model_forward_time": 1.7250690460205078,
      "step": 17746
    },
    {
      "epoch": 1.692390441894531e-05,
      "step": 17746,
      "training_step_time": 6.759379863739014
    },
    {
      "epoch": 1.692485809326172e-05,
      "model_forward_time": 1.7242372035980225,
      "step": 17747
    },
    {
      "epoch": 1.692485809326172e-05,
      "step": 17747,
      "training_step_time": 6.823462247848511
    },
    {
      "epoch": 1.6925811767578126e-05,
      "model_forward_time": 1.7248947620391846,
      "step": 17748
    },
    {
      "epoch": 1.6925811767578126e-05,
      "step": 17748,
      "training_step_time": 6.7496936321258545
    },
    {
      "epoch": 1.692676544189453e-05,
      "model_forward_time": 1.730290174484253,
      "step": 17749
    },
    {
      "epoch": 1.692676544189453e-05,
      "step": 17749,
      "training_step_time": 7.637686729431152
    },
    {
      "epoch": 1.6927719116210938e-05,
      "grad_norm": 0.26036134362220764,
      "learning_rate": 9.562173227380437e-06,
      "loss": 0.0627,
      "step": 17750
    },
    {
      "epoch": 1.6927719116210938e-05,
      "model_forward_time": 2.26566481590271,
      "step": 17750
    },
    {
      "dynamics_loss_avg": 0.08632818572223186,
      "epoch": 1.6927719116210938e-05,
      "step": 17750
    },
    {
      "action_loss_avg": 0.0047629204578697685,
      "epoch": 1.6927719116210938e-05,
      "step": 17750
    },
    {
      "epoch": 1.6927719116210938e-05,
      "step": 17750,
      "training_step_time": 9.687918901443481
    },
    {
      "epoch": 1.6928672790527345e-05,
      "model_forward_time": 1.7285351753234863,
      "step": 17751
    },
    {
      "epoch": 1.6928672790527345e-05,
      "step": 17751,
      "training_step_time": 8.765312194824219
    },
    {
      "epoch": 1.692962646484375e-05,
      "model_forward_time": 1.7259998321533203,
      "step": 17752
    },
    {
      "epoch": 1.692962646484375e-05,
      "step": 17752,
      "training_step_time": 7.745136260986328
    },
    {
      "epoch": 1.6930580139160156e-05,
      "model_forward_time": 1.7164463996887207,
      "step": 17753
    },
    {
      "epoch": 1.6930580139160156e-05,
      "step": 17753,
      "training_step_time": 6.753105401992798
    },
    {
      "epoch": 1.6931533813476564e-05,
      "model_forward_time": 1.7225706577301025,
      "step": 17754
    },
    {
      "epoch": 1.6931533813476564e-05,
      "step": 17754,
      "training_step_time": 6.82136344909668
    },
    {
      "epoch": 1.6932487487792968e-05,
      "model_forward_time": 1.729893445968628,
      "step": 17755
    },
    {
      "epoch": 1.6932487487792968e-05,
      "step": 17755,
      "training_step_time": 6.80136251449585
    },
    {
      "epoch": 1.6933441162109375e-05,
      "model_forward_time": 1.7245252132415771,
      "step": 17756
    },
    {
      "epoch": 1.6933441162109375e-05,
      "step": 17756,
      "training_step_time": 6.765313625335693
    },
    {
      "epoch": 1.6934394836425783e-05,
      "model_forward_time": 1.7244772911071777,
      "step": 17757
    },
    {
      "epoch": 1.6934394836425783e-05,
      "step": 17757,
      "training_step_time": 6.742062091827393
    },
    {
      "epoch": 1.6935348510742187e-05,
      "model_forward_time": 1.729456901550293,
      "step": 17758
    },
    {
      "epoch": 1.6935348510742187e-05,
      "step": 17758,
      "training_step_time": 7.563322067260742
    },
    {
      "epoch": 1.6936302185058594e-05,
      "model_forward_time": 2.276461124420166,
      "step": 17759
    },
    {
      "epoch": 1.6936302185058594e-05,
      "step": 17759,
      "training_step_time": 9.250328063964844
    },
    {
      "epoch": 1.6937255859375e-05,
      "grad_norm": 0.22443215548992157,
      "learning_rate": 9.561496340958389e-06,
      "loss": 0.062,
      "step": 17760
    },
    {
      "epoch": 1.6937255859375e-05,
      "model_forward_time": 1.3420977592468262,
      "step": 17760
    },
    {
      "dynamics_loss_avg": 0.06744038164615632,
      "epoch": 1.6937255859375e-05,
      "step": 17760
    },
    {
      "action_loss_avg": 0.003048273629974574,
      "epoch": 1.6937255859375e-05,
      "step": 17760
    },
    {
      "epoch": 1.6937255859375e-05,
      "step": 17760,
      "training_step_time": 8.46562647819519
    },
    {
      "epoch": 1.6938209533691406e-05,
      "model_forward_time": 1.3513686656951904,
      "step": 17761
    },
    {
      "epoch": 1.6938209533691406e-05,
      "step": 17761,
      "training_step_time": 7.643267631530762
    },
    {
      "epoch": 1.6939163208007813e-05,
      "model_forward_time": 1.7223281860351562,
      "step": 17762
    },
    {
      "epoch": 1.6939163208007813e-05,
      "step": 17762,
      "training_step_time": 6.811316251754761
    },
    {
      "epoch": 1.694011688232422e-05,
      "model_forward_time": 1.7276012897491455,
      "step": 17763
    },
    {
      "epoch": 1.694011688232422e-05,
      "step": 17763,
      "training_step_time": 6.779955863952637
    },
    {
      "epoch": 1.6941070556640624e-05,
      "model_forward_time": 1.7195656299591064,
      "step": 17764
    },
    {
      "epoch": 1.6941070556640624e-05,
      "step": 17764,
      "training_step_time": 6.799211263656616
    },
    {
      "epoch": 1.6942024230957032e-05,
      "model_forward_time": 1.723933458328247,
      "step": 17765
    },
    {
      "epoch": 1.6942024230957032e-05,
      "step": 17765,
      "training_step_time": 6.779079437255859
    },
    {
      "epoch": 1.6942977905273436e-05,
      "model_forward_time": 1.7271320819854736,
      "step": 17766
    },
    {
      "epoch": 1.6942977905273436e-05,
      "step": 17766,
      "training_step_time": 6.785527944564819
    },
    {
      "epoch": 1.6943931579589843e-05,
      "model_forward_time": 1.73283052444458,
      "step": 17767
    },
    {
      "epoch": 1.6943931579589843e-05,
      "step": 17767,
      "training_step_time": 7.635010480880737
    },
    {
      "epoch": 1.694488525390625e-05,
      "model_forward_time": 1.7256395816802979,
      "step": 17768
    },
    {
      "epoch": 1.694488525390625e-05,
      "step": 17768,
      "training_step_time": 8.812823057174683
    },
    {
      "epoch": 1.6945838928222655e-05,
      "model_forward_time": 2.275559186935425,
      "step": 17769
    },
    {
      "epoch": 1.6945838928222655e-05,
      "step": 17769,
      "training_step_time": 8.321897745132446
    },
    {
      "epoch": 1.6946792602539062e-05,
      "grad_norm": 0.20700998604297638,
      "learning_rate": 9.56081895569796e-06,
      "loss": 0.0611,
      "step": 17770
    },
    {
      "epoch": 1.6946792602539062e-05,
      "model_forward_time": 1.715245246887207,
      "step": 17770
    },
    {
      "dynamics_loss_avg": 0.0662972841411829,
      "epoch": 1.6946792602539062e-05,
      "step": 17770
    },
    {
      "action_loss_avg": 0.004498514311853797,
      "epoch": 1.6946792602539062e-05,
      "step": 17770
    },
    {
      "epoch": 1.6946792602539062e-05,
      "step": 17770,
      "training_step_time": 7.762613534927368
    },
    {
      "epoch": 1.694774627685547e-05,
      "model_forward_time": 1.7222130298614502,
      "step": 17771
    },
    {
      "epoch": 1.694774627685547e-05,
      "step": 17771,
      "training_step_time": 7.76904821395874
    },
    {
      "epoch": 1.6948699951171874e-05,
      "model_forward_time": 1.7255454063415527,
      "step": 17772
    },
    {
      "epoch": 1.6948699951171874e-05,
      "step": 17772,
      "training_step_time": 6.774649620056152
    },
    {
      "epoch": 1.694965362548828e-05,
      "model_forward_time": 1.7268402576446533,
      "step": 17773
    },
    {
      "epoch": 1.694965362548828e-05,
      "step": 17773,
      "training_step_time": 6.772961616516113
    },
    {
      "epoch": 1.695060729980469e-05,
      "model_forward_time": 1.724435567855835,
      "step": 17774
    },
    {
      "epoch": 1.695060729980469e-05,
      "step": 17774,
      "training_step_time": 6.766526699066162
    },
    {
      "epoch": 1.6951560974121092e-05,
      "model_forward_time": 1.7283432483673096,
      "step": 17775
    },
    {
      "epoch": 1.6951560974121092e-05,
      "step": 17775,
      "training_step_time": 6.8294782638549805
    },
    {
      "epoch": 1.69525146484375e-05,
      "model_forward_time": 1.725442886352539,
      "step": 17776
    },
    {
      "epoch": 1.69525146484375e-05,
      "step": 17776,
      "training_step_time": 7.553781032562256
    },
    {
      "epoch": 1.6953468322753907e-05,
      "model_forward_time": 1.7222340106964111,
      "step": 17777
    },
    {
      "epoch": 1.6953468322753907e-05,
      "step": 17777,
      "training_step_time": 8.667798280715942
    },
    {
      "epoch": 1.695442199707031e-05,
      "model_forward_time": 2.287670135498047,
      "step": 17778
    },
    {
      "epoch": 1.695442199707031e-05,
      "step": 17778,
      "training_step_time": 9.281610250473022
    },
    {
      "epoch": 1.695537567138672e-05,
      "model_forward_time": 1.7074999809265137,
      "step": 17779
    },
    {
      "epoch": 1.695537567138672e-05,
      "step": 17779,
      "training_step_time": 8.344980716705322
    },
    {
      "epoch": 1.6956329345703126e-05,
      "grad_norm": 0.211504727602005,
      "learning_rate": 9.560141071673228e-06,
      "loss": 0.0635,
      "step": 17780
    },
    {
      "epoch": 1.6956329345703126e-05,
      "model_forward_time": 1.3458991050720215,
      "step": 17780
    },
    {
      "dynamics_loss_avg": 0.060014546290040015,
      "epoch": 1.6956329345703126e-05,
      "step": 17780
    },
    {
      "action_loss_avg": 0.0023983173014130443,
      "epoch": 1.6956329345703126e-05,
      "step": 17780
    },
    {
      "epoch": 1.6956329345703126e-05,
      "step": 17780,
      "training_step_time": 7.8049938678741455
    },
    {
      "epoch": 1.695728302001953e-05,
      "model_forward_time": 1.725985050201416,
      "step": 17781
    },
    {
      "epoch": 1.695728302001953e-05,
      "step": 17781,
      "training_step_time": 6.729149103164673
    },
    {
      "epoch": 1.6958236694335938e-05,
      "model_forward_time": 1.723947525024414,
      "step": 17782
    },
    {
      "epoch": 1.6958236694335938e-05,
      "step": 17782,
      "training_step_time": 6.755584239959717
    },
    {
      "epoch": 1.6959190368652345e-05,
      "model_forward_time": 1.7230734825134277,
      "step": 17783
    },
    {
      "epoch": 1.6959190368652345e-05,
      "step": 17783,
      "training_step_time": 6.757394313812256
    },
    {
      "epoch": 1.696014404296875e-05,
      "model_forward_time": 1.7279777526855469,
      "step": 17784
    },
    {
      "epoch": 1.696014404296875e-05,
      "step": 17784,
      "training_step_time": 6.757488012313843
    },
    {
      "epoch": 1.6961097717285156e-05,
      "model_forward_time": 1.7336652278900146,
      "step": 17785
    },
    {
      "epoch": 1.6961097717285156e-05,
      "step": 17785,
      "training_step_time": 7.569961786270142
    },
    {
      "epoch": 1.6962051391601564e-05,
      "model_forward_time": 1.7232253551483154,
      "step": 17786
    },
    {
      "epoch": 1.6962051391601564e-05,
      "step": 17786,
      "training_step_time": 9.293780326843262
    },
    {
      "epoch": 1.6963005065917968e-05,
      "model_forward_time": 2.285679578781128,
      "step": 17787
    },
    {
      "epoch": 1.6963005065917968e-05,
      "step": 17787,
      "training_step_time": 10.220385789871216
    },
    {
      "epoch": 1.6963958740234375e-05,
      "model_forward_time": 1.7148497104644775,
      "step": 17788
    },
    {
      "epoch": 1.6963958740234375e-05,
      "step": 17788,
      "training_step_time": 7.722310543060303
    },
    {
      "epoch": 1.6964912414550783e-05,
      "model_forward_time": 1.718416452407837,
      "step": 17789
    },
    {
      "epoch": 1.6964912414550783e-05,
      "step": 17789,
      "training_step_time": 7.80419135093689
    },
    {
      "epoch": 1.6965866088867187e-05,
      "grad_norm": 0.18623360991477966,
      "learning_rate": 9.559462688958324e-06,
      "loss": 0.0634,
      "step": 17790
    },
    {
      "epoch": 1.6965866088867187e-05,
      "model_forward_time": 1.7244877815246582,
      "step": 17790
    },
    {
      "dynamics_loss_avg": 0.05734830480068922,
      "epoch": 1.6965866088867187e-05,
      "step": 17790
    },
    {
      "action_loss_avg": 0.0021289311698637903,
      "epoch": 1.6965866088867187e-05,
      "step": 17790
    },
    {
      "epoch": 1.6965866088867187e-05,
      "step": 17790,
      "training_step_time": 6.731938123703003
    },
    {
      "epoch": 1.6966819763183594e-05,
      "model_forward_time": 1.7219724655151367,
      "step": 17791
    },
    {
      "epoch": 1.6966819763183594e-05,
      "step": 17791,
      "training_step_time": 6.722381353378296
    },
    {
      "epoch": 1.69677734375e-05,
      "model_forward_time": 1.7182397842407227,
      "step": 17792
    },
    {
      "epoch": 1.69677734375e-05,
      "step": 17792,
      "training_step_time": 6.718120813369751
    },
    {
      "epoch": 1.6968727111816406e-05,
      "model_forward_time": 1.717043399810791,
      "step": 17793
    },
    {
      "epoch": 1.6968727111816406e-05,
      "step": 17793,
      "training_step_time": 6.735450506210327
    },
    {
      "epoch": 1.6969680786132813e-05,
      "model_forward_time": 1.718993902206421,
      "step": 17794
    },
    {
      "epoch": 1.6969680786132813e-05,
      "step": 17794,
      "training_step_time": 7.574782609939575
    },
    {
      "epoch": 1.697063446044922e-05,
      "model_forward_time": 1.7130556106567383,
      "step": 17795
    },
    {
      "epoch": 1.697063446044922e-05,
      "step": 17795,
      "training_step_time": 7.713732957839966
    },
    {
      "epoch": 1.6971588134765624e-05,
      "model_forward_time": 1.7542753219604492,
      "step": 17796
    },
    {
      "epoch": 1.6971588134765624e-05,
      "step": 17796,
      "training_step_time": 9.398719787597656
    },
    {
      "epoch": 1.6972541809082032e-05,
      "model_forward_time": 2.250866651535034,
      "step": 17797
    },
    {
      "epoch": 1.6972541809082032e-05,
      "step": 17797,
      "training_step_time": 8.750187873840332
    },
    {
      "epoch": 1.6973495483398436e-05,
      "model_forward_time": 1.720841884613037,
      "step": 17798
    },
    {
      "epoch": 1.6973495483398436e-05,
      "step": 17798,
      "training_step_time": 7.693675518035889
    },
    {
      "epoch": 1.6974449157714843e-05,
      "model_forward_time": 1.7172391414642334,
      "step": 17799
    },
    {
      "epoch": 1.6974449157714843e-05,
      "step": 17799,
      "training_step_time": 6.784290313720703
    },
    {
      "epoch": 1.697540283203125e-05,
      "grad_norm": 0.15885600447654724,
      "learning_rate": 9.558783807627434e-06,
      "loss": 0.0618,
      "step": 17800
    },
    {
      "epoch": 1.697540283203125e-05,
      "model_forward_time": 1.7132971286773682,
      "step": 17800
    },
    {
      "dynamics_loss_avg": 0.06468662358820439,
      "epoch": 1.697540283203125e-05,
      "step": 17800
    },
    {
      "action_loss_avg": 0.003991700033657253,
      "epoch": 1.697540283203125e-05,
      "step": 17800
    },
    {
      "epoch": 1.697540283203125e-05,
      "step": 17800,
      "training_step_time": 6.788992166519165
    },
    {
      "epoch": 1.6976356506347655e-05,
      "model_forward_time": 1.7258448600769043,
      "step": 17801
    },
    {
      "epoch": 1.6976356506347655e-05,
      "step": 17801,
      "training_step_time": 6.723592042922974
    },
    {
      "epoch": 1.6977310180664062e-05,
      "model_forward_time": 1.7193424701690674,
      "step": 17802
    },
    {
      "epoch": 1.6977310180664062e-05,
      "step": 17802,
      "training_step_time": 6.808779954910278
    },
    {
      "epoch": 1.697826385498047e-05,
      "model_forward_time": 1.7078566551208496,
      "step": 17803
    },
    {
      "epoch": 1.697826385498047e-05,
      "step": 17803,
      "training_step_time": 7.566848516464233
    },
    {
      "epoch": 1.6979217529296874e-05,
      "model_forward_time": 1.7162272930145264,
      "step": 17804
    },
    {
      "epoch": 1.6979217529296874e-05,
      "step": 17804,
      "training_step_time": 8.17743468284607
    },
    {
      "epoch": 1.698017120361328e-05,
      "model_forward_time": 1.3966052532196045,
      "step": 17805
    },
    {
      "epoch": 1.698017120361328e-05,
      "step": 17805,
      "training_step_time": 7.820020914077759
    },
    {
      "epoch": 1.698112487792969e-05,
      "model_forward_time": 2.2497382164001465,
      "step": 17806
    },
    {
      "epoch": 1.698112487792969e-05,
      "step": 17806,
      "training_step_time": 10.193581342697144
    },
    {
      "epoch": 1.6982078552246092e-05,
      "model_forward_time": 1.7195148468017578,
      "step": 17807
    },
    {
      "epoch": 1.6982078552246092e-05,
      "step": 17807,
      "training_step_time": 7.734166860580444
    },
    {
      "epoch": 1.69830322265625e-05,
      "model_forward_time": 1.716787338256836,
      "step": 17808
    },
    {
      "epoch": 1.69830322265625e-05,
      "step": 17808,
      "training_step_time": 6.722857236862183
    },
    {
      "epoch": 1.6983985900878907e-05,
      "model_forward_time": 1.7173411846160889,
      "step": 17809
    },
    {
      "epoch": 1.6983985900878907e-05,
      "step": 17809,
      "training_step_time": 6.721818685531616
    },
    {
      "epoch": 1.698493957519531e-05,
      "grad_norm": 0.20591498911380768,
      "learning_rate": 9.558104427754801e-06,
      "loss": 0.0648,
      "step": 17810
    },
    {
      "epoch": 1.698493957519531e-05,
      "model_forward_time": 1.7189974784851074,
      "step": 17810
    },
    {
      "dynamics_loss_avg": 0.07078647762537002,
      "epoch": 1.698493957519531e-05,
      "step": 17810
    },
    {
      "action_loss_avg": 0.004037537879776209,
      "epoch": 1.698493957519531e-05,
      "step": 17810
    },
    {
      "epoch": 1.698493957519531e-05,
      "step": 17810,
      "training_step_time": 6.7401885986328125
    },
    {
      "epoch": 1.698589324951172e-05,
      "model_forward_time": 1.710710048675537,
      "step": 17811
    },
    {
      "epoch": 1.698589324951172e-05,
      "step": 17811,
      "training_step_time": 6.722773790359497
    },
    {
      "epoch": 1.6986846923828126e-05,
      "model_forward_time": 1.731445074081421,
      "step": 17812
    },
    {
      "epoch": 1.6986846923828126e-05,
      "step": 17812,
      "training_step_time": 7.580144882202148
    },
    {
      "epoch": 1.698780059814453e-05,
      "model_forward_time": 1.713984489440918,
      "step": 17813
    },
    {
      "epoch": 1.698780059814453e-05,
      "step": 17813,
      "training_step_time": 7.717303514480591
    },
    {
      "epoch": 1.6988754272460938e-05,
      "model_forward_time": 1.7419848442077637,
      "step": 17814
    },
    {
      "epoch": 1.6988754272460938e-05,
      "step": 17814,
      "training_step_time": 9.284465551376343
    },
    {
      "epoch": 1.6989707946777345e-05,
      "model_forward_time": 1.5317835807800293,
      "step": 17815
    },
    {
      "epoch": 1.6989707946777345e-05,
      "step": 17815,
      "training_step_time": 9.808894872665405
    },
    {
      "epoch": 1.699066162109375e-05,
      "model_forward_time": 1.3360238075256348,
      "step": 17816
    },
    {
      "epoch": 1.699066162109375e-05,
      "step": 17816,
      "training_step_time": 7.6843581199646
    },
    {
      "epoch": 1.6991615295410156e-05,
      "model_forward_time": 1.3414967060089111,
      "step": 17817
    },
    {
      "epoch": 1.6991615295410156e-05,
      "step": 17817,
      "training_step_time": 7.6744866371154785
    },
    {
      "epoch": 1.6992568969726564e-05,
      "model_forward_time": 1.7130601406097412,
      "step": 17818
    },
    {
      "epoch": 1.6992568969726564e-05,
      "step": 17818,
      "training_step_time": 6.824645757675171
    },
    {
      "epoch": 1.6993522644042968e-05,
      "model_forward_time": 1.7168076038360596,
      "step": 17819
    },
    {
      "epoch": 1.6993522644042968e-05,
      "step": 17819,
      "training_step_time": 6.8363707065582275
    },
    {
      "epoch": 1.6994476318359375e-05,
      "grad_norm": 0.14876173436641693,
      "learning_rate": 9.557424549414723e-06,
      "loss": 0.0648,
      "step": 17820
    },
    {
      "epoch": 1.6994476318359375e-05,
      "model_forward_time": 1.7186121940612793,
      "step": 17820
    },
    {
      "dynamics_loss_avg": 0.053096186742186545,
      "epoch": 1.6994476318359375e-05,
      "step": 17820
    },
    {
      "action_loss_avg": 0.002791699080262333,
      "epoch": 1.6994476318359375e-05,
      "step": 17820
    },
    {
      "epoch": 1.6994476318359375e-05,
      "step": 17820,
      "training_step_time": 6.79436731338501
    },
    {
      "epoch": 1.6995429992675783e-05,
      "model_forward_time": 1.7269103527069092,
      "step": 17821
    },
    {
      "epoch": 1.6995429992675783e-05,
      "step": 17821,
      "training_step_time": 7.603042840957642
    },
    {
      "epoch": 1.6996383666992187e-05,
      "model_forward_time": 1.7158632278442383,
      "step": 17822
    },
    {
      "epoch": 1.6996383666992187e-05,
      "step": 17822,
      "training_step_time": 8.769432306289673
    },
    {
      "epoch": 1.6997337341308594e-05,
      "model_forward_time": 1.744990348815918,
      "step": 17823
    },
    {
      "epoch": 1.6997337341308594e-05,
      "step": 17823,
      "training_step_time": 8.754558563232422
    },
    {
      "epoch": 1.6998291015625e-05,
      "model_forward_time": 1.709446668624878,
      "step": 17824
    },
    {
      "epoch": 1.6998291015625e-05,
      "step": 17824,
      "training_step_time": 10.063979864120483
    },
    {
      "epoch": 1.6999244689941406e-05,
      "model_forward_time": 2.278041124343872,
      "step": 17825
    },
    {
      "epoch": 1.6999244689941406e-05,
      "step": 17825,
      "training_step_time": 8.617607116699219
    },
    {
      "epoch": 1.7000198364257813e-05,
      "model_forward_time": 1.7128355503082275,
      "step": 17826
    },
    {
      "epoch": 1.7000198364257813e-05,
      "step": 17826,
      "training_step_time": 7.75990891456604
    },
    {
      "epoch": 1.700115203857422e-05,
      "model_forward_time": 1.7168281078338623,
      "step": 17827
    },
    {
      "epoch": 1.700115203857422e-05,
      "step": 17827,
      "training_step_time": 7.829035520553589
    },
    {
      "epoch": 1.7002105712890624e-05,
      "model_forward_time": 1.7090795040130615,
      "step": 17828
    },
    {
      "epoch": 1.7002105712890624e-05,
      "step": 17828,
      "training_step_time": 6.856998920440674
    },
    {
      "epoch": 1.7003059387207032e-05,
      "model_forward_time": 1.7188217639923096,
      "step": 17829
    },
    {
      "epoch": 1.7003059387207032e-05,
      "step": 17829,
      "training_step_time": 6.850425720214844
    },
    {
      "epoch": 1.7004013061523436e-05,
      "grad_norm": 0.17509542405605316,
      "learning_rate": 9.556744172681546e-06,
      "loss": 0.0704,
      "step": 17830
    },
    {
      "epoch": 1.7004013061523436e-05,
      "model_forward_time": 1.7125370502471924,
      "step": 17830
    },
    {
      "dynamics_loss_avg": 0.04596878569573164,
      "epoch": 1.7004013061523436e-05,
      "step": 17830
    },
    {
      "action_loss_avg": 0.0017071845650207251,
      "epoch": 1.7004013061523436e-05,
      "step": 17830
    },
    {
      "epoch": 1.7004013061523436e-05,
      "step": 17830,
      "training_step_time": 7.552274465560913
    },
    {
      "epoch": 1.7004966735839843e-05,
      "model_forward_time": 1.724954605102539,
      "step": 17831
    },
    {
      "epoch": 1.7004966735839843e-05,
      "step": 17831,
      "training_step_time": 7.698355197906494
    },
    {
      "epoch": 1.700592041015625e-05,
      "model_forward_time": 1.724339246749878,
      "step": 17832
    },
    {
      "epoch": 1.700592041015625e-05,
      "step": 17832,
      "training_step_time": 9.177181959152222
    },
    {
      "epoch": 1.7006874084472655e-05,
      "model_forward_time": 1.7616219520568848,
      "step": 17833
    },
    {
      "epoch": 1.7006874084472655e-05,
      "step": 17833,
      "training_step_time": 8.281182527542114
    },
    {
      "epoch": 1.7007827758789062e-05,
      "model_forward_time": 2.2652721405029297,
      "step": 17834
    },
    {
      "epoch": 1.7007827758789062e-05,
      "step": 17834,
      "training_step_time": 9.477473020553589
    },
    {
      "epoch": 1.700878143310547e-05,
      "model_forward_time": 1.7206614017486572,
      "step": 17835
    },
    {
      "epoch": 1.700878143310547e-05,
      "step": 17835,
      "training_step_time": 7.622613430023193
    },
    {
      "epoch": 1.7009735107421874e-05,
      "model_forward_time": 1.700103759765625,
      "step": 17836
    },
    {
      "epoch": 1.7009735107421874e-05,
      "step": 17836,
      "training_step_time": 7.695466041564941
    },
    {
      "epoch": 1.701068878173828e-05,
      "model_forward_time": 1.7129557132720947,
      "step": 17837
    },
    {
      "epoch": 1.701068878173828e-05,
      "step": 17837,
      "training_step_time": 6.7562572956085205
    },
    {
      "epoch": 1.701164245605469e-05,
      "model_forward_time": 1.718235969543457,
      "step": 17838
    },
    {
      "epoch": 1.701164245605469e-05,
      "step": 17838,
      "training_step_time": 6.749070167541504
    },
    {
      "epoch": 1.7012596130371092e-05,
      "model_forward_time": 1.7093932628631592,
      "step": 17839
    },
    {
      "epoch": 1.7012596130371092e-05,
      "step": 17839,
      "training_step_time": 7.517564535140991
    },
    {
      "epoch": 1.70135498046875e-05,
      "grad_norm": 0.17878223955631256,
      "learning_rate": 9.556063297629677e-06,
      "loss": 0.0704,
      "step": 17840
    },
    {
      "epoch": 1.70135498046875e-05,
      "model_forward_time": 1.7215056419372559,
      "step": 17840
    },
    {
      "dynamics_loss_avg": 0.05566948615014553,
      "epoch": 1.70135498046875e-05,
      "step": 17840
    },
    {
      "action_loss_avg": 0.003353943140245974,
      "epoch": 1.70135498046875e-05,
      "step": 17840
    },
    {
      "epoch": 1.70135498046875e-05,
      "step": 17840,
      "training_step_time": 7.7128143310546875
    },
    {
      "epoch": 1.7014503479003907e-05,
      "model_forward_time": 1.716421127319336,
      "step": 17841
    },
    {
      "epoch": 1.7014503479003907e-05,
      "step": 17841,
      "training_step_time": 9.08785605430603
    },
    {
      "epoch": 1.701545715332031e-05,
      "model_forward_time": 1.767134428024292,
      "step": 17842
    },
    {
      "epoch": 1.701545715332031e-05,
      "step": 17842,
      "training_step_time": 8.369176626205444
    },
    {
      "epoch": 1.701641082763672e-05,
      "model_forward_time": 2.2767488956451416,
      "step": 17843
    },
    {
      "epoch": 1.701641082763672e-05,
      "step": 17843,
      "training_step_time": 8.577795267105103
    },
    {
      "epoch": 1.7017364501953126e-05,
      "model_forward_time": 1.7142236232757568,
      "step": 17844
    },
    {
      "epoch": 1.7017364501953126e-05,
      "step": 17844,
      "training_step_time": 7.671097278594971
    },
    {
      "epoch": 1.701831817626953e-05,
      "model_forward_time": 1.7109403610229492,
      "step": 17845
    },
    {
      "epoch": 1.701831817626953e-05,
      "step": 17845,
      "training_step_time": 7.7125937938690186
    },
    {
      "epoch": 1.7019271850585938e-05,
      "model_forward_time": 1.7187824249267578,
      "step": 17846
    },
    {
      "epoch": 1.7019271850585938e-05,
      "step": 17846,
      "training_step_time": 6.764298439025879
    },
    {
      "epoch": 1.7020225524902345e-05,
      "model_forward_time": 1.7144687175750732,
      "step": 17847
    },
    {
      "epoch": 1.7020225524902345e-05,
      "step": 17847,
      "training_step_time": 6.729106903076172
    },
    {
      "epoch": 1.702117919921875e-05,
      "model_forward_time": 1.7113673686981201,
      "step": 17848
    },
    {
      "epoch": 1.702117919921875e-05,
      "step": 17848,
      "training_step_time": 7.4997875690460205
    },
    {
      "epoch": 1.7022132873535156e-05,
      "model_forward_time": 1.6971101760864258,
      "step": 17849
    },
    {
      "epoch": 1.7022132873535156e-05,
      "step": 17849,
      "training_step_time": 7.577766180038452
    },
    {
      "epoch": 1.7023086547851564e-05,
      "grad_norm": 0.1527760773897171,
      "learning_rate": 9.555381924333577e-06,
      "loss": 0.0631,
      "step": 17850
    },
    {
      "epoch": 1.7023086547851564e-05,
      "model_forward_time": 1.7134532928466797,
      "step": 17850
    },
    {
      "dynamics_loss_avg": 0.06855419278144836,
      "epoch": 1.7023086547851564e-05,
      "step": 17850
    },
    {
      "action_loss_avg": 0.002840292442124337,
      "epoch": 1.7023086547851564e-05,
      "step": 17850
    },
    {
      "epoch": 1.7023086547851564e-05,
      "step": 17850,
      "training_step_time": 8.4130699634552
    },
    {
      "epoch": 1.7024040222167968e-05,
      "model_forward_time": 1.7625391483306885,
      "step": 17851
    },
    {
      "epoch": 1.7024040222167968e-05,
      "step": 17851,
      "training_step_time": 7.771855115890503
    },
    {
      "epoch": 1.7024993896484375e-05,
      "model_forward_time": 2.265289783477783,
      "step": 17852
    },
    {
      "epoch": 1.7024993896484375e-05,
      "step": 17852,
      "training_step_time": 8.468173265457153
    },
    {
      "epoch": 1.7025947570800783e-05,
      "model_forward_time": 1.7168495655059814,
      "step": 17853
    },
    {
      "epoch": 1.7025947570800783e-05,
      "step": 17853,
      "training_step_time": 7.788666725158691
    },
    {
      "epoch": 1.7026901245117187e-05,
      "model_forward_time": 1.7244107723236084,
      "step": 17854
    },
    {
      "epoch": 1.7026901245117187e-05,
      "step": 17854,
      "training_step_time": 7.690290689468384
    },
    {
      "epoch": 1.7027854919433594e-05,
      "model_forward_time": 1.7157111167907715,
      "step": 17855
    },
    {
      "epoch": 1.7027854919433594e-05,
      "step": 17855,
      "training_step_time": 6.802438497543335
    },
    {
      "epoch": 1.702880859375e-05,
      "model_forward_time": 1.7123973369598389,
      "step": 17856
    },
    {
      "epoch": 1.702880859375e-05,
      "step": 17856,
      "training_step_time": 6.881426572799683
    },
    {
      "epoch": 1.7029762268066406e-05,
      "model_forward_time": 1.7083699703216553,
      "step": 17857
    },
    {
      "epoch": 1.7029762268066406e-05,
      "step": 17857,
      "training_step_time": 6.850667476654053
    },
    {
      "epoch": 1.7030715942382813e-05,
      "model_forward_time": 1.705399990081787,
      "step": 17858
    },
    {
      "epoch": 1.7030715942382813e-05,
      "step": 17858,
      "training_step_time": 7.652347087860107
    },
    {
      "epoch": 1.703166961669922e-05,
      "model_forward_time": 1.7228035926818848,
      "step": 17859
    },
    {
      "epoch": 1.703166961669922e-05,
      "step": 17859,
      "training_step_time": 9.434453248977661
    },
    {
      "epoch": 1.7032623291015624e-05,
      "grad_norm": 0.16323301196098328,
      "learning_rate": 9.554700052867758e-06,
      "loss": 0.0623,
      "step": 17860
    },
    {
      "epoch": 1.7032623291015624e-05,
      "model_forward_time": 1.3382186889648438,
      "step": 17860
    },
    {
      "dynamics_loss_avg": 0.07095806412398815,
      "epoch": 1.7032623291015624e-05,
      "step": 17860
    },
    {
      "action_loss_avg": 0.002369475818704814,
      "epoch": 1.7032623291015624e-05,
      "step": 17860
    },
    {
      "epoch": 1.7032623291015624e-05,
      "step": 17860,
      "training_step_time": 8.642696619033813
    },
    {
      "epoch": 1.7033576965332032e-05,
      "model_forward_time": 1.709449052810669,
      "step": 17861
    },
    {
      "epoch": 1.7033576965332032e-05,
      "step": 17861,
      "training_step_time": 8.585219144821167
    },
    {
      "epoch": 1.7034530639648436e-05,
      "model_forward_time": 2.2363831996917725,
      "step": 17862
    },
    {
      "epoch": 1.7034530639648436e-05,
      "step": 17862,
      "training_step_time": 7.810595273971558
    },
    {
      "epoch": 1.7035484313964843e-05,
      "model_forward_time": 1.7088367938995361,
      "step": 17863
    },
    {
      "epoch": 1.7035484313964843e-05,
      "step": 17863,
      "training_step_time": 8.7748703956604
    },
    {
      "epoch": 1.703643798828125e-05,
      "model_forward_time": 1.7113614082336426,
      "step": 17864
    },
    {
      "epoch": 1.703643798828125e-05,
      "step": 17864,
      "training_step_time": 6.792129278182983
    },
    {
      "epoch": 1.7037391662597655e-05,
      "model_forward_time": 1.3439440727233887,
      "step": 17865
    },
    {
      "epoch": 1.7037391662597655e-05,
      "step": 17865,
      "training_step_time": 6.822040557861328
    },
    {
      "epoch": 1.7038345336914062e-05,
      "model_forward_time": 1.717151165008545,
      "step": 17866
    },
    {
      "epoch": 1.7038345336914062e-05,
      "step": 17866,
      "training_step_time": 6.796844005584717
    },
    {
      "epoch": 1.703929901123047e-05,
      "model_forward_time": 1.7066493034362793,
      "step": 17867
    },
    {
      "epoch": 1.703929901123047e-05,
      "step": 17867,
      "training_step_time": 7.677499771118164
    },
    {
      "epoch": 1.7040252685546874e-05,
      "model_forward_time": 1.7114732265472412,
      "step": 17868
    },
    {
      "epoch": 1.7040252685546874e-05,
      "step": 17868,
      "training_step_time": 7.788209676742554
    },
    {
      "epoch": 1.704120635986328e-05,
      "model_forward_time": 1.7137854099273682,
      "step": 17869
    },
    {
      "epoch": 1.704120635986328e-05,
      "step": 17869,
      "training_step_time": 10.126466274261475
    },
    {
      "epoch": 1.704216003417969e-05,
      "grad_norm": 0.1616949886083603,
      "learning_rate": 9.55401768330679e-06,
      "loss": 0.0663,
      "step": 17870
    },
    {
      "epoch": 1.704216003417969e-05,
      "model_forward_time": 1.338785171508789,
      "step": 17870
    },
    {
      "dynamics_loss_avg": 0.07908807434141636,
      "epoch": 1.704216003417969e-05,
      "step": 17870
    },
    {
      "action_loss_avg": 0.005390864354558289,
      "epoch": 1.704216003417969e-05,
      "step": 17870
    },
    {
      "epoch": 1.704216003417969e-05,
      "step": 17870,
      "training_step_time": 8.786572694778442
    },
    {
      "epoch": 1.7043113708496092e-05,
      "model_forward_time": 2.2596867084503174,
      "step": 17871
    },
    {
      "epoch": 1.7043113708496092e-05,
      "step": 17871,
      "training_step_time": 8.729037046432495
    },
    {
      "epoch": 1.70440673828125e-05,
      "model_forward_time": 1.7104077339172363,
      "step": 17872
    },
    {
      "epoch": 1.70440673828125e-05,
      "step": 17872,
      "training_step_time": 7.677995443344116
    },
    {
      "epoch": 1.7045021057128907e-05,
      "model_forward_time": 1.71260404586792,
      "step": 17873
    },
    {
      "epoch": 1.7045021057128907e-05,
      "step": 17873,
      "training_step_time": 7.684079885482788
    },
    {
      "epoch": 1.704597473144531e-05,
      "model_forward_time": 1.7064220905303955,
      "step": 17874
    },
    {
      "epoch": 1.704597473144531e-05,
      "step": 17874,
      "training_step_time": 6.673701047897339
    },
    {
      "epoch": 1.704692840576172e-05,
      "model_forward_time": 1.7135798931121826,
      "step": 17875
    },
    {
      "epoch": 1.704692840576172e-05,
      "step": 17875,
      "training_step_time": 6.731506109237671
    },
    {
      "epoch": 1.7047882080078126e-05,
      "model_forward_time": 1.3214354515075684,
      "step": 17876
    },
    {
      "epoch": 1.7047882080078126e-05,
      "step": 17876,
      "training_step_time": 7.7951648235321045
    },
    {
      "epoch": 1.704883575439453e-05,
      "model_forward_time": 1.716238260269165,
      "step": 17877
    },
    {
      "epoch": 1.704883575439453e-05,
      "step": 17877,
      "training_step_time": 7.6645188331604
    },
    {
      "epoch": 1.7049789428710938e-05,
      "model_forward_time": 1.7040209770202637,
      "step": 17878
    },
    {
      "epoch": 1.7049789428710938e-05,
      "step": 17878,
      "training_step_time": 8.629730224609375
    },
    {
      "epoch": 1.7050743103027345e-05,
      "model_forward_time": 1.752742052078247,
      "step": 17879
    },
    {
      "epoch": 1.7050743103027345e-05,
      "step": 17879,
      "training_step_time": 9.26291561126709
    },
    {
      "epoch": 1.705169677734375e-05,
      "grad_norm": 0.14467793703079224,
      "learning_rate": 9.553334815725295e-06,
      "loss": 0.0645,
      "step": 17880
    },
    {
      "epoch": 1.705169677734375e-05,
      "model_forward_time": 2.23445200920105,
      "step": 17880
    },
    {
      "dynamics_loss_avg": 0.0837618850171566,
      "epoch": 1.705169677734375e-05,
      "step": 17880
    },
    {
      "action_loss_avg": 0.0038320215186104178,
      "epoch": 1.705169677734375e-05,
      "step": 17880
    },
    {
      "epoch": 1.705169677734375e-05,
      "step": 17880,
      "training_step_time": 7.667501449584961
    },
    {
      "epoch": 1.7052650451660156e-05,
      "model_forward_time": 1.711554765701294,
      "step": 17881
    },
    {
      "epoch": 1.7052650451660156e-05,
      "step": 17881,
      "training_step_time": 7.685641050338745
    },
    {
      "epoch": 1.7053604125976564e-05,
      "model_forward_time": 1.7091214656829834,
      "step": 17882
    },
    {
      "epoch": 1.7053604125976564e-05,
      "step": 17882,
      "training_step_time": 7.723693132400513
    },
    {
      "epoch": 1.7054557800292968e-05,
      "model_forward_time": 1.7127647399902344,
      "step": 17883
    },
    {
      "epoch": 1.7054557800292968e-05,
      "step": 17883,
      "training_step_time": 6.733735799789429
    },
    {
      "epoch": 1.7055511474609375e-05,
      "model_forward_time": 1.70957612991333,
      "step": 17884
    },
    {
      "epoch": 1.7055511474609375e-05,
      "step": 17884,
      "training_step_time": 6.739011287689209
    },
    {
      "epoch": 1.7056465148925783e-05,
      "model_forward_time": 1.7114646434783936,
      "step": 17885
    },
    {
      "epoch": 1.7056465148925783e-05,
      "step": 17885,
      "training_step_time": 8.152783870697021
    },
    {
      "epoch": 1.7057418823242187e-05,
      "model_forward_time": 1.7051329612731934,
      "step": 17886
    },
    {
      "epoch": 1.7057418823242187e-05,
      "step": 17886,
      "training_step_time": 7.638876438140869
    },
    {
      "epoch": 1.7058372497558594e-05,
      "model_forward_time": 1.6905303001403809,
      "step": 17887
    },
    {
      "epoch": 1.7058372497558594e-05,
      "step": 17887,
      "training_step_time": 7.637845516204834
    },
    {
      "epoch": 1.7059326171875e-05,
      "model_forward_time": 1.3453290462493896,
      "step": 17888
    },
    {
      "epoch": 1.7059326171875e-05,
      "step": 17888,
      "training_step_time": 7.80780291557312
    },
    {
      "epoch": 1.7060279846191406e-05,
      "model_forward_time": 1.6913156509399414,
      "step": 17889
    },
    {
      "epoch": 1.7060279846191406e-05,
      "step": 17889,
      "training_step_time": 9.238548994064331
    },
    {
      "epoch": 1.7061233520507813e-05,
      "grad_norm": 0.1893203854560852,
      "learning_rate": 9.552651450197949e-06,
      "loss": 0.064,
      "step": 17890
    },
    {
      "epoch": 1.7061233520507813e-05,
      "model_forward_time": 2.2084474563598633,
      "step": 17890
    },
    {
      "dynamics_loss_avg": 0.07060711309313775,
      "epoch": 1.7061233520507813e-05,
      "step": 17890
    },
    {
      "action_loss_avg": 0.003155688988044858,
      "epoch": 1.7061233520507813e-05,
      "step": 17890
    },
    {
      "epoch": 1.7061233520507813e-05,
      "step": 17890,
      "training_step_time": 7.702630281448364
    },
    {
      "epoch": 1.7062187194824217e-05,
      "model_forward_time": 1.69785475730896,
      "step": 17891
    },
    {
      "epoch": 1.7062187194824217e-05,
      "step": 17891,
      "training_step_time": 7.807169198989868
    },
    {
      "epoch": 1.7063140869140624e-05,
      "model_forward_time": 1.7008070945739746,
      "step": 17892
    },
    {
      "epoch": 1.7063140869140624e-05,
      "step": 17892,
      "training_step_time": 6.773144245147705
    },
    {
      "epoch": 1.7064094543457032e-05,
      "model_forward_time": 1.699920415878296,
      "step": 17893
    },
    {
      "epoch": 1.7064094543457032e-05,
      "step": 17893,
      "training_step_time": 6.795519828796387
    },
    {
      "epoch": 1.7065048217773436e-05,
      "model_forward_time": 1.7007207870483398,
      "step": 17894
    },
    {
      "epoch": 1.7065048217773436e-05,
      "step": 17894,
      "training_step_time": 7.745434999465942
    },
    {
      "epoch": 1.7066001892089843e-05,
      "model_forward_time": 1.6983530521392822,
      "step": 17895
    },
    {
      "epoch": 1.7066001892089843e-05,
      "step": 17895,
      "training_step_time": 8.630894184112549
    },
    {
      "epoch": 1.706695556640625e-05,
      "model_forward_time": 1.695986270904541,
      "step": 17896
    },
    {
      "epoch": 1.706695556640625e-05,
      "step": 17896,
      "training_step_time": 8.692084550857544
    },
    {
      "epoch": 1.7067909240722655e-05,
      "model_forward_time": 1.7172579765319824,
      "step": 17897
    },
    {
      "epoch": 1.7067909240722655e-05,
      "step": 17897,
      "training_step_time": 7.817125082015991
    },
    {
      "epoch": 1.7068862915039062e-05,
      "model_forward_time": 1.7014143466949463,
      "step": 17898
    },
    {
      "epoch": 1.7068862915039062e-05,
      "step": 17898,
      "training_step_time": 8.950738906860352
    },
    {
      "epoch": 1.706981658935547e-05,
      "model_forward_time": 2.2186038494110107,
      "step": 17899
    },
    {
      "epoch": 1.706981658935547e-05,
      "step": 17899,
      "training_step_time": 8.238061904907227
    },
    {
      "epoch": 1.7070770263671874e-05,
      "grad_norm": 0.16806407272815704,
      "learning_rate": 9.551967586799487e-06,
      "loss": 0.0642,
      "step": 17900
    },
    {
      "epoch": 1.7070770263671874e-05,
      "model_forward_time": 1.6979897022247314,
      "step": 17900
    },
    {
      "dynamics_loss_avg": 0.04673758577555418,
      "epoch": 1.7070770263671874e-05,
      "step": 17900
    },
    {
      "action_loss_avg": 0.0032743917545303702,
      "epoch": 1.7070770263671874e-05,
      "step": 17900
    },
    {
      "epoch": 1.7070770263671874e-05,
      "step": 17900,
      "training_step_time": 7.698257684707642
    },
    {
      "epoch": 1.707172393798828e-05,
      "model_forward_time": 1.6945216655731201,
      "step": 17901
    },
    {
      "epoch": 1.707172393798828e-05,
      "step": 17901,
      "training_step_time": 6.758902311325073
    },
    {
      "epoch": 1.707267761230469e-05,
      "model_forward_time": 1.7017807960510254,
      "step": 17902
    },
    {
      "epoch": 1.707267761230469e-05,
      "step": 17902,
      "training_step_time": 6.754537582397461
    },
    {
      "epoch": 1.7073631286621093e-05,
      "model_forward_time": 1.3266525268554688,
      "step": 17903
    },
    {
      "epoch": 1.7073631286621093e-05,
      "step": 17903,
      "training_step_time": 7.716451644897461
    },
    {
      "epoch": 1.70745849609375e-05,
      "model_forward_time": 1.7051892280578613,
      "step": 17904
    },
    {
      "epoch": 1.70745849609375e-05,
      "step": 17904,
      "training_step_time": 7.541209697723389
    },
    {
      "epoch": 1.7075538635253907e-05,
      "model_forward_time": 1.699333906173706,
      "step": 17905
    },
    {
      "epoch": 1.7075538635253907e-05,
      "step": 17905,
      "training_step_time": 9.479218244552612
    },
    {
      "epoch": 1.707649230957031e-05,
      "model_forward_time": 1.7027997970581055,
      "step": 17906
    },
    {
      "epoch": 1.707649230957031e-05,
      "step": 17906,
      "training_step_time": 9.50768232345581
    },
    {
      "epoch": 1.707744598388672e-05,
      "model_forward_time": 1.7388520240783691,
      "step": 17907
    },
    {
      "epoch": 1.707744598388672e-05,
      "step": 17907,
      "training_step_time": 8.479101419448853
    },
    {
      "epoch": 1.7078399658203126e-05,
      "model_forward_time": 2.2503342628479004,
      "step": 17908
    },
    {
      "epoch": 1.7078399658203126e-05,
      "step": 17908,
      "training_step_time": 7.663243055343628
    },
    {
      "epoch": 1.707935333251953e-05,
      "model_forward_time": 1.7005548477172852,
      "step": 17909
    },
    {
      "epoch": 1.707935333251953e-05,
      "step": 17909,
      "training_step_time": 7.594712495803833
    },
    {
      "epoch": 1.7080307006835938e-05,
      "grad_norm": 0.13359008729457855,
      "learning_rate": 9.551283225604691e-06,
      "loss": 0.0653,
      "step": 17910
    },
    {
      "epoch": 1.7080307006835938e-05,
      "model_forward_time": 1.6955716609954834,
      "step": 17910
    },
    {
      "dynamics_loss_avg": 0.09520899653434753,
      "epoch": 1.7080307006835938e-05,
      "step": 17910
    },
    {
      "action_loss_avg": 0.004885026509873569,
      "epoch": 1.7080307006835938e-05,
      "step": 17910
    },
    {
      "epoch": 1.7080307006835938e-05,
      "step": 17910,
      "training_step_time": 6.67858362197876
    },
    {
      "epoch": 1.7081260681152345e-05,
      "model_forward_time": 1.702272653579712,
      "step": 17911
    },
    {
      "epoch": 1.7081260681152345e-05,
      "step": 17911,
      "training_step_time": 6.694787979125977
    },
    {
      "epoch": 1.708221435546875e-05,
      "model_forward_time": 1.696519136428833,
      "step": 17912
    },
    {
      "epoch": 1.708221435546875e-05,
      "step": 17912,
      "training_step_time": 6.647134780883789
    },
    {
      "epoch": 1.7083168029785156e-05,
      "model_forward_time": 1.699782371520996,
      "step": 17913
    },
    {
      "epoch": 1.7083168029785156e-05,
      "step": 17913,
      "training_step_time": 8.772775888442993
    },
    {
      "epoch": 1.7084121704101564e-05,
      "model_forward_time": 1.702082872390747,
      "step": 17914
    },
    {
      "epoch": 1.7084121704101564e-05,
      "step": 17914,
      "training_step_time": 8.468415021896362
    },
    {
      "epoch": 1.7085075378417968e-05,
      "model_forward_time": 1.702859878540039,
      "step": 17915
    },
    {
      "epoch": 1.7085075378417968e-05,
      "step": 17915,
      "training_step_time": 8.64674186706543
    },
    {
      "epoch": 1.7086029052734375e-05,
      "model_forward_time": 1.735887050628662,
      "step": 17916
    },
    {
      "epoch": 1.7086029052734375e-05,
      "step": 17916,
      "training_step_time": 8.257535934448242
    },
    {
      "epoch": 1.7086982727050783e-05,
      "model_forward_time": 2.218001365661621,
      "step": 17917
    },
    {
      "epoch": 1.7086982727050783e-05,
      "step": 17917,
      "training_step_time": 8.306592226028442
    },
    {
      "epoch": 1.7087936401367187e-05,
      "model_forward_time": 1.6971149444580078,
      "step": 17918
    },
    {
      "epoch": 1.7087936401367187e-05,
      "step": 17918,
      "training_step_time": 7.619722127914429
    },
    {
      "epoch": 1.7088890075683594e-05,
      "model_forward_time": 1.7019941806793213,
      "step": 17919
    },
    {
      "epoch": 1.7088890075683594e-05,
      "step": 17919,
      "training_step_time": 7.691909551620483
    },
    {
      "epoch": 1.708984375e-05,
      "grad_norm": 0.14201031625270844,
      "learning_rate": 9.550598366688406e-06,
      "loss": 0.0612,
      "step": 17920
    },
    {
      "epoch": 1.708984375e-05,
      "model_forward_time": 1.701345443725586,
      "step": 17920
    },
    {
      "dynamics_loss_avg": 0.06565524321049451,
      "epoch": 1.708984375e-05,
      "step": 17920
    },
    {
      "action_loss_avg": 0.003835929196793586,
      "epoch": 1.708984375e-05,
      "step": 17920
    },
    {
      "epoch": 1.708984375e-05,
      "step": 17920,
      "training_step_time": 6.699141263961792
    },
    {
      "epoch": 1.7090797424316406e-05,
      "model_forward_time": 1.7036831378936768,
      "step": 17921
    },
    {
      "epoch": 1.7090797424316406e-05,
      "step": 17921,
      "training_step_time": 6.707808256149292
    },
    {
      "epoch": 1.7091751098632813e-05,
      "model_forward_time": 1.7038965225219727,
      "step": 17922
    },
    {
      "epoch": 1.7091751098632813e-05,
      "step": 17922,
      "training_step_time": 7.593702554702759
    },
    {
      "epoch": 1.7092704772949217e-05,
      "model_forward_time": 1.700463056564331,
      "step": 17923
    },
    {
      "epoch": 1.7092704772949217e-05,
      "step": 17923,
      "training_step_time": 7.587526798248291
    },
    {
      "epoch": 1.7093658447265625e-05,
      "model_forward_time": 1.6987781524658203,
      "step": 17924
    },
    {
      "epoch": 1.7093658447265625e-05,
      "step": 17924,
      "training_step_time": 7.645919561386108
    },
    {
      "epoch": 1.7094612121582032e-05,
      "model_forward_time": 1.7438082695007324,
      "step": 17925
    },
    {
      "epoch": 1.7094612121582032e-05,
      "step": 17925,
      "training_step_time": 8.761362552642822
    },
    {
      "epoch": 1.7095565795898436e-05,
      "model_forward_time": 1.6999483108520508,
      "step": 17926
    },
    {
      "epoch": 1.7095565795898436e-05,
      "step": 17926,
      "training_step_time": 7.683558940887451
    },
    {
      "epoch": 1.7096519470214843e-05,
      "model_forward_time": 2.225886821746826,
      "step": 17927
    },
    {
      "epoch": 1.7096519470214843e-05,
      "step": 17927,
      "training_step_time": 9.714820623397827
    },
    {
      "epoch": 1.709747314453125e-05,
      "model_forward_time": 1.7125999927520752,
      "step": 17928
    },
    {
      "epoch": 1.709747314453125e-05,
      "step": 17928,
      "training_step_time": 7.694127321243286
    },
    {
      "epoch": 1.7098426818847655e-05,
      "model_forward_time": 1.7318503856658936,
      "step": 17929
    },
    {
      "epoch": 1.7098426818847655e-05,
      "step": 17929,
      "training_step_time": 6.688426971435547
    },
    {
      "epoch": 1.7099380493164062e-05,
      "grad_norm": 0.21045009791851044,
      "learning_rate": 9.549913010125527e-06,
      "loss": 0.0638,
      "step": 17930
    },
    {
      "epoch": 1.7099380493164062e-05,
      "model_forward_time": 1.7310771942138672,
      "step": 17930
    },
    {
      "dynamics_loss_avg": 0.06428535114973784,
      "epoch": 1.7099380493164062e-05,
      "step": 17930
    },
    {
      "action_loss_avg": 0.0033357912499923257,
      "epoch": 1.7099380493164062e-05,
      "step": 17930
    },
    {
      "epoch": 1.7099380493164062e-05,
      "step": 17930,
      "training_step_time": 6.70086407661438
    },
    {
      "epoch": 1.710033416748047e-05,
      "model_forward_time": 1.735522747039795,
      "step": 17931
    },
    {
      "epoch": 1.710033416748047e-05,
      "step": 17931,
      "training_step_time": 7.5577569007873535
    },
    {
      "epoch": 1.7101287841796874e-05,
      "model_forward_time": 1.7292900085449219,
      "step": 17932
    },
    {
      "epoch": 1.7101287841796874e-05,
      "step": 17932,
      "training_step_time": 7.54949688911438
    },
    {
      "epoch": 1.710224151611328e-05,
      "model_forward_time": 1.7264068126678467,
      "step": 17933
    },
    {
      "epoch": 1.710224151611328e-05,
      "step": 17933,
      "training_step_time": 7.65902304649353
    },
    {
      "epoch": 1.710319519042969e-05,
      "model_forward_time": 1.7210872173309326,
      "step": 17934
    },
    {
      "epoch": 1.710319519042969e-05,
      "step": 17934,
      "training_step_time": 9.381309032440186
    },
    {
      "epoch": 1.7104148864746093e-05,
      "model_forward_time": 1.7006580829620361,
      "step": 17935
    },
    {
      "epoch": 1.7104148864746093e-05,
      "step": 17935,
      "training_step_time": 7.720249176025391
    },
    {
      "epoch": 1.71051025390625e-05,
      "model_forward_time": 2.249654531478882,
      "step": 17936
    },
    {
      "epoch": 1.71051025390625e-05,
      "step": 17936,
      "training_step_time": 7.588943004608154
    },
    {
      "epoch": 1.7106056213378907e-05,
      "model_forward_time": 1.7125577926635742,
      "step": 17937
    },
    {
      "epoch": 1.7106056213378907e-05,
      "step": 17937,
      "training_step_time": 8.560234069824219
    },
    {
      "epoch": 1.710700988769531e-05,
      "model_forward_time": 1.7131755352020264,
      "step": 17938
    },
    {
      "epoch": 1.710700988769531e-05,
      "step": 17938,
      "training_step_time": 6.69533109664917
    },
    {
      "epoch": 1.710796356201172e-05,
      "model_forward_time": 1.7247912883758545,
      "step": 17939
    },
    {
      "epoch": 1.710796356201172e-05,
      "step": 17939,
      "training_step_time": 6.740017414093018
    },
    {
      "epoch": 1.7108917236328126e-05,
      "grad_norm": 0.1788124442100525,
      "learning_rate": 9.549227155991e-06,
      "loss": 0.0679,
      "step": 17940
    },
    {
      "epoch": 1.7108917236328126e-05,
      "model_forward_time": 1.7290637493133545,
      "step": 17940
    },
    {
      "dynamics_loss_avg": 0.07628296390175819,
      "epoch": 1.7108917236328126e-05,
      "step": 17940
    },
    {
      "action_loss_avg": 0.0032427199301309885,
      "epoch": 1.7108917236328126e-05,
      "step": 17940
    },
    {
      "epoch": 1.7108917236328126e-05,
      "step": 17940,
      "training_step_time": 7.384361743927002
    },
    {
      "epoch": 1.710987091064453e-05,
      "model_forward_time": 1.7106916904449463,
      "step": 17941
    },
    {
      "epoch": 1.710987091064453e-05,
      "step": 17941,
      "training_step_time": 7.709460973739624
    },
    {
      "epoch": 1.7110824584960938e-05,
      "model_forward_time": 1.7189671993255615,
      "step": 17942
    },
    {
      "epoch": 1.7110824584960938e-05,
      "step": 17942,
      "training_step_time": 8.387906789779663
    },
    {
      "epoch": 1.7111778259277345e-05,
      "model_forward_time": 1.7655284404754639,
      "step": 17943
    },
    {
      "epoch": 1.7111778259277345e-05,
      "step": 17943,
      "training_step_time": 8.721308708190918
    },
    {
      "epoch": 1.711273193359375e-05,
      "model_forward_time": 1.7151134014129639,
      "step": 17944
    },
    {
      "epoch": 1.711273193359375e-05,
      "step": 17944,
      "training_step_time": 8.724326372146606
    },
    {
      "epoch": 1.7113685607910157e-05,
      "model_forward_time": 2.2548301219940186,
      "step": 17945
    },
    {
      "epoch": 1.7113685607910157e-05,
      "step": 17945,
      "training_step_time": 7.639782428741455
    },
    {
      "epoch": 1.7114639282226564e-05,
      "model_forward_time": 1.7202751636505127,
      "step": 17946
    },
    {
      "epoch": 1.7114639282226564e-05,
      "step": 17946,
      "training_step_time": 7.512603521347046
    },
    {
      "epoch": 1.7115592956542968e-05,
      "model_forward_time": 1.7143349647521973,
      "step": 17947
    },
    {
      "epoch": 1.7115592956542968e-05,
      "step": 17947,
      "training_step_time": 7.647272825241089
    },
    {
      "epoch": 1.7116546630859375e-05,
      "model_forward_time": 1.712890386581421,
      "step": 17948
    },
    {
      "epoch": 1.7116546630859375e-05,
      "step": 17948,
      "training_step_time": 6.712659120559692
    },
    {
      "epoch": 1.7117500305175783e-05,
      "model_forward_time": 1.7158279418945312,
      "step": 17949
    },
    {
      "epoch": 1.7117500305175783e-05,
      "step": 17949,
      "training_step_time": 7.011633396148682
    },
    {
      "epoch": 1.7118453979492187e-05,
      "grad_norm": 0.25857141613960266,
      "learning_rate": 9.54854080435983e-06,
      "loss": 0.0689,
      "step": 17950
    },
    {
      "epoch": 1.7118453979492187e-05,
      "model_forward_time": 1.7155742645263672,
      "step": 17950
    },
    {
      "dynamics_loss_avg": 0.0505747364833951,
      "epoch": 1.7118453979492187e-05,
      "step": 17950
    },
    {
      "action_loss_avg": 0.002552618581103161,
      "epoch": 1.7118453979492187e-05,
      "step": 17950
    },
    {
      "epoch": 1.7118453979492187e-05,
      "step": 17950,
      "training_step_time": 7.612175226211548
    },
    {
      "epoch": 1.7119407653808594e-05,
      "model_forward_time": 1.714834213256836,
      "step": 17951
    },
    {
      "epoch": 1.7119407653808594e-05,
      "step": 17951,
      "training_step_time": 7.608625411987305
    },
    {
      "epoch": 1.7120361328125e-05,
      "model_forward_time": 1.7596678733825684,
      "step": 17952
    },
    {
      "epoch": 1.7120361328125e-05,
      "step": 17952,
      "training_step_time": 8.589361190795898
    },
    {
      "epoch": 1.7121315002441406e-05,
      "model_forward_time": 1.7135124206542969,
      "step": 17953
    },
    {
      "epoch": 1.7121315002441406e-05,
      "step": 17953,
      "training_step_time": 9.06264615058899
    },
    {
      "epoch": 1.7122268676757813e-05,
      "model_forward_time": 2.2658026218414307,
      "step": 17954
    },
    {
      "epoch": 1.7122268676757813e-05,
      "step": 17954,
      "training_step_time": 7.551851749420166
    },
    {
      "epoch": 1.7123222351074217e-05,
      "model_forward_time": 1.3390395641326904,
      "step": 17955
    },
    {
      "epoch": 1.7123222351074217e-05,
      "step": 17955,
      "training_step_time": 8.339785099029541
    },
    {
      "epoch": 1.7124176025390625e-05,
      "model_forward_time": 1.7204256057739258,
      "step": 17956
    },
    {
      "epoch": 1.7124176025390625e-05,
      "step": 17956,
      "training_step_time": 7.63967490196228
    },
    {
      "epoch": 1.7125129699707032e-05,
      "model_forward_time": 1.8216924667358398,
      "step": 17957
    },
    {
      "epoch": 1.7125129699707032e-05,
      "step": 17957,
      "training_step_time": 6.677657127380371
    },
    {
      "epoch": 1.7126083374023436e-05,
      "model_forward_time": 1.817817211151123,
      "step": 17958
    },
    {
      "epoch": 1.7126083374023436e-05,
      "step": 17958,
      "training_step_time": 6.695771932601929
    },
    {
      "epoch": 1.7127037048339843e-05,
      "model_forward_time": 1.8139846324920654,
      "step": 17959
    },
    {
      "epoch": 1.7127037048339843e-05,
      "step": 17959,
      "training_step_time": 7.580899953842163
    },
    {
      "epoch": 1.712799072265625e-05,
      "grad_norm": 0.27181798219680786,
      "learning_rate": 9.547853955307077e-06,
      "loss": 0.0685,
      "step": 17960
    },
    {
      "epoch": 1.712799072265625e-05,
      "model_forward_time": 1.8174731731414795,
      "step": 17960
    },
    {
      "dynamics_loss_avg": 0.053941452875733376,
      "epoch": 1.712799072265625e-05,
      "step": 17960
    },
    {
      "action_loss_avg": 0.003287030360661447,
      "epoch": 1.712799072265625e-05,
      "step": 17960
    },
    {
      "epoch": 1.712799072265625e-05,
      "step": 17960,
      "training_step_time": 9.121222496032715
    },
    {
      "epoch": 1.7128944396972655e-05,
      "model_forward_time": 1.8183469772338867,
      "step": 17961
    },
    {
      "epoch": 1.7128944396972655e-05,
      "step": 17961,
      "training_step_time": 7.755862712860107
    },
    {
      "epoch": 1.7129898071289062e-05,
      "model_forward_time": 1.8155148029327393,
      "step": 17962
    },
    {
      "epoch": 1.7129898071289062e-05,
      "step": 17962,
      "training_step_time": 9.208916664123535
    },
    {
      "epoch": 1.713085174560547e-05,
      "model_forward_time": 2.329559564590454,
      "step": 17963
    },
    {
      "epoch": 1.713085174560547e-05,
      "step": 17963,
      "training_step_time": 8.502500534057617
    },
    {
      "epoch": 1.7131805419921874e-05,
      "model_forward_time": 1.8132712841033936,
      "step": 17964
    },
    {
      "epoch": 1.7131805419921874e-05,
      "step": 17964,
      "training_step_time": 7.682045936584473
    },
    {
      "epoch": 1.713275909423828e-05,
      "model_forward_time": 1.444528579711914,
      "step": 17965
    },
    {
      "epoch": 1.713275909423828e-05,
      "step": 17965,
      "training_step_time": 7.593142747879028
    },
    {
      "epoch": 1.713371276855469e-05,
      "model_forward_time": 1.811370849609375,
      "step": 17966
    },
    {
      "epoch": 1.713371276855469e-05,
      "step": 17966,
      "training_step_time": 6.690020322799683
    },
    {
      "epoch": 1.7134666442871093e-05,
      "model_forward_time": 1.792088508605957,
      "step": 17967
    },
    {
      "epoch": 1.7134666442871093e-05,
      "step": 17967,
      "training_step_time": 6.653759717941284
    },
    {
      "epoch": 1.71356201171875e-05,
      "model_forward_time": 1.438981056213379,
      "step": 17968
    },
    {
      "epoch": 1.71356201171875e-05,
      "step": 17968,
      "training_step_time": 7.5343005657196045
    },
    {
      "epoch": 1.7136573791503907e-05,
      "model_forward_time": 1.8145396709442139,
      "step": 17969
    },
    {
      "epoch": 1.7136573791503907e-05,
      "step": 17969,
      "training_step_time": 8.46645450592041
    },
    {
      "epoch": 1.713752746582031e-05,
      "grad_norm": 0.1945594996213913,
      "learning_rate": 9.547166608907853e-06,
      "loss": 0.0637,
      "step": 17970
    },
    {
      "epoch": 1.713752746582031e-05,
      "model_forward_time": 1.819237470626831,
      "step": 17970
    },
    {
      "dynamics_loss_avg": 0.059223266877233983,
      "epoch": 1.713752746582031e-05,
      "step": 17970
    },
    {
      "action_loss_avg": 0.0035107461386360227,
      "epoch": 1.713752746582031e-05,
      "step": 17970
    },
    {
      "epoch": 1.713752746582031e-05,
      "step": 17970,
      "training_step_time": 8.518218278884888
    },
    {
      "epoch": 1.713848114013672e-05,
      "model_forward_time": 1.816159725189209,
      "step": 17971
    },
    {
      "epoch": 1.713848114013672e-05,
      "step": 17971,
      "training_step_time": 7.7818028926849365
    },
    {
      "epoch": 1.7139434814453126e-05,
      "model_forward_time": 1.4472529888153076,
      "step": 17972
    },
    {
      "epoch": 1.7139434814453126e-05,
      "step": 17972,
      "training_step_time": 8.274773836135864
    },
    {
      "epoch": 1.714038848876953e-05,
      "model_forward_time": 2.3236680030822754,
      "step": 17973
    },
    {
      "epoch": 1.714038848876953e-05,
      "step": 17973,
      "training_step_time": 8.490222454071045
    },
    {
      "epoch": 1.7141342163085938e-05,
      "model_forward_time": 1.7859630584716797,
      "step": 17974
    },
    {
      "epoch": 1.7141342163085938e-05,
      "step": 17974,
      "training_step_time": 8.606952428817749
    },
    {
      "epoch": 1.7142295837402345e-05,
      "model_forward_time": 1.7184784412384033,
      "step": 17975
    },
    {
      "epoch": 1.7142295837402345e-05,
      "step": 17975,
      "training_step_time": 7.622342824935913
    },
    {
      "epoch": 1.714324951171875e-05,
      "model_forward_time": 1.7220485210418701,
      "step": 17976
    },
    {
      "epoch": 1.714324951171875e-05,
      "step": 17976,
      "training_step_time": 6.717828273773193
    },
    {
      "epoch": 1.7144203186035157e-05,
      "model_forward_time": 1.7268261909484863,
      "step": 17977
    },
    {
      "epoch": 1.7144203186035157e-05,
      "step": 17977,
      "training_step_time": 7.498568058013916
    },
    {
      "epoch": 1.7145156860351564e-05,
      "model_forward_time": 1.7194092273712158,
      "step": 17978
    },
    {
      "epoch": 1.7145156860351564e-05,
      "step": 17978,
      "training_step_time": 8.54903531074524
    },
    {
      "epoch": 1.7146110534667968e-05,
      "model_forward_time": 1.6970272064208984,
      "step": 17979
    },
    {
      "epoch": 1.7146110534667968e-05,
      "step": 17979,
      "training_step_time": 8.214137554168701
    },
    {
      "epoch": 1.7147064208984375e-05,
      "grad_norm": 0.13594678044319153,
      "learning_rate": 9.546478765237326e-06,
      "loss": 0.0674,
      "step": 17980
    },
    {
      "epoch": 1.7147064208984375e-05,
      "model_forward_time": 1.7072570323944092,
      "step": 17980
    },
    {
      "dynamics_loss_avg": 0.0607509458437562,
      "epoch": 1.7147064208984375e-05,
      "step": 17980
    },
    {
      "action_loss_avg": 0.0038385795662179588,
      "epoch": 1.7147064208984375e-05,
      "step": 17980
    },
    {
      "epoch": 1.7147064208984375e-05,
      "step": 17980,
      "training_step_time": 9.195953607559204
    },
    {
      "epoch": 1.7148017883300783e-05,
      "model_forward_time": 1.703597068786621,
      "step": 17981
    },
    {
      "epoch": 1.7148017883300783e-05,
      "step": 17981,
      "training_step_time": 8.656072854995728
    },
    {
      "epoch": 1.7148971557617187e-05,
      "model_forward_time": 2.268965482711792,
      "step": 17982
    },
    {
      "epoch": 1.7148971557617187e-05,
      "step": 17982,
      "training_step_time": 8.56428599357605
    },
    {
      "epoch": 1.7149925231933594e-05,
      "model_forward_time": 1.7066717147827148,
      "step": 17983
    },
    {
      "epoch": 1.7149925231933594e-05,
      "step": 17983,
      "training_step_time": 7.526754856109619
    },
    {
      "epoch": 1.715087890625e-05,
      "model_forward_time": 1.7088210582733154,
      "step": 17984
    },
    {
      "epoch": 1.715087890625e-05,
      "step": 17984,
      "training_step_time": 7.537903070449829
    },
    {
      "epoch": 1.7151832580566406e-05,
      "model_forward_time": 1.7125647068023682,
      "step": 17985
    },
    {
      "epoch": 1.7151832580566406e-05,
      "step": 17985,
      "training_step_time": 6.728541374206543
    },
    {
      "epoch": 1.7152786254882813e-05,
      "model_forward_time": 1.707169532775879,
      "step": 17986
    },
    {
      "epoch": 1.7152786254882813e-05,
      "step": 17986,
      "training_step_time": 8.301295042037964
    },
    {
      "epoch": 1.7153739929199217e-05,
      "model_forward_time": 1.7039759159088135,
      "step": 17987
    },
    {
      "epoch": 1.7153739929199217e-05,
      "step": 17987,
      "training_step_time": 7.578866958618164
    },
    {
      "epoch": 1.7154693603515625e-05,
      "model_forward_time": 1.702524185180664,
      "step": 17988
    },
    {
      "epoch": 1.7154693603515625e-05,
      "step": 17988,
      "training_step_time": 7.659175872802734
    },
    {
      "epoch": 1.7155647277832032e-05,
      "model_forward_time": 1.7018277645111084,
      "step": 17989
    },
    {
      "epoch": 1.7155647277832032e-05,
      "step": 17989,
      "training_step_time": 8.639500856399536
    },
    {
      "epoch": 1.7156600952148436e-05,
      "grad_norm": 0.18705050647258759,
      "learning_rate": 9.545790424370715e-06,
      "loss": 0.0714,
      "step": 17990
    },
    {
      "epoch": 1.7156600952148436e-05,
      "model_forward_time": 1.7238352298736572,
      "step": 17990
    },
    {
      "dynamics_loss_avg": 0.08760465867817402,
      "epoch": 1.7156600952148436e-05,
      "step": 17990
    },
    {
      "action_loss_avg": 0.0032371832989156247,
      "epoch": 1.7156600952148436e-05,
      "step": 17990
    },
    {
      "epoch": 1.7156600952148436e-05,
      "step": 17990,
      "training_step_time": 7.606165170669556
    },
    {
      "epoch": 1.7157554626464843e-05,
      "model_forward_time": 2.274736166000366,
      "step": 17991
    },
    {
      "epoch": 1.7157554626464843e-05,
      "step": 17991,
      "training_step_time": 7.550741672515869
    },
    {
      "epoch": 1.715850830078125e-05,
      "model_forward_time": 1.7491216659545898,
      "step": 17992
    },
    {
      "epoch": 1.715850830078125e-05,
      "step": 17992,
      "training_step_time": 7.569873332977295
    },
    {
      "epoch": 1.7159461975097655e-05,
      "model_forward_time": 1.7076144218444824,
      "step": 17993
    },
    {
      "epoch": 1.7159461975097655e-05,
      "step": 17993,
      "training_step_time": 7.6350531578063965
    },
    {
      "epoch": 1.7160415649414062e-05,
      "model_forward_time": 1.682701826095581,
      "step": 17994
    },
    {
      "epoch": 1.7160415649414062e-05,
      "step": 17994,
      "training_step_time": 6.710089445114136
    },
    {
      "epoch": 1.716136932373047e-05,
      "model_forward_time": 1.6990008354187012,
      "step": 17995
    },
    {
      "epoch": 1.716136932373047e-05,
      "step": 17995,
      "training_step_time": 8.146583318710327
    },
    {
      "epoch": 1.7162322998046874e-05,
      "model_forward_time": 1.7060861587524414,
      "step": 17996
    },
    {
      "epoch": 1.7162322998046874e-05,
      "step": 17996,
      "training_step_time": 7.546733140945435
    },
    {
      "epoch": 1.716327667236328e-05,
      "model_forward_time": 1.3356857299804688,
      "step": 17997
    },
    {
      "epoch": 1.716327667236328e-05,
      "step": 17997,
      "training_step_time": 7.631382942199707
    },
    {
      "epoch": 1.716423034667969e-05,
      "model_forward_time": 1.7300305366516113,
      "step": 17998
    },
    {
      "epoch": 1.716423034667969e-05,
      "step": 17998,
      "training_step_time": 8.12496829032898
    },
    {
      "epoch": 1.7165184020996093e-05,
      "model_forward_time": 1.767730951309204,
      "step": 17999
    },
    {
      "epoch": 1.7165184020996093e-05,
      "step": 17999,
      "training_step_time": 10.079097986221313
    },
    {
      "epoch": 1.71661376953125e-05,
      "grad_norm": 0.317254900932312,
      "learning_rate": 9.5451015863833e-06,
      "loss": 0.0692,
      "step": 18000
    },
    {
      "epoch": 1.71661376953125e-05,
      "model_forward_time": 2.6897568702697754,
      "step": 18000
    },
    {
      "dynamics_loss_avg": 0.07465873956680298,
      "epoch": 1.71661376953125e-05,
      "step": 18000
    },
    {
      "action_loss_avg": 0.0037079099303809924,
      "epoch": 1.71661376953125e-05,
      "step": 18000
    },
    {
      "epoch": 1.71661376953125e-05,
      "step": 18000,
      "training_step_time": 8.473427534103394
    },
    {
      "epoch": 1.7167091369628907e-05,
      "model_forward_time": 1.7258780002593994,
      "step": 18001
    },
    {
      "epoch": 1.7167091369628907e-05,
      "step": 18001,
      "training_step_time": 10.579770803451538
    },
    {
      "epoch": 1.716804504394531e-05,
      "model_forward_time": 1.7136037349700928,
      "step": 18002
    },
    {
      "epoch": 1.716804504394531e-05,
      "step": 18002,
      "training_step_time": 7.483565807342529
    },
    {
      "epoch": 1.716899871826172e-05,
      "model_forward_time": 1.7000782489776611,
      "step": 18003
    },
    {
      "epoch": 1.716899871826172e-05,
      "step": 18003,
      "training_step_time": 8.494098424911499
    },
    {
      "epoch": 1.7169952392578126e-05,
      "model_forward_time": 1.7162790298461914,
      "step": 18004
    },
    {
      "epoch": 1.7169952392578126e-05,
      "step": 18004,
      "training_step_time": 8.90278935432434
    },
    {
      "epoch": 1.717090606689453e-05,
      "model_forward_time": 1.7072069644927979,
      "step": 18005
    },
    {
      "epoch": 1.717090606689453e-05,
      "step": 18005,
      "training_step_time": 8.649878740310669
    },
    {
      "epoch": 1.7171859741210938e-05,
      "model_forward_time": 1.7053911685943604,
      "step": 18006
    },
    {
      "epoch": 1.7171859741210938e-05,
      "step": 18006,
      "training_step_time": 8.480666399002075
    },
    {
      "epoch": 1.7172813415527345e-05,
      "model_forward_time": 1.712724208831787,
      "step": 18007
    },
    {
      "epoch": 1.7172813415527345e-05,
      "step": 18007,
      "training_step_time": 11.418012857437134
    },
    {
      "epoch": 1.717376708984375e-05,
      "model_forward_time": 1.7073640823364258,
      "step": 18008
    },
    {
      "epoch": 1.717376708984375e-05,
      "step": 18008,
      "training_step_time": 9.746301174163818
    },
    {
      "epoch": 1.7174720764160157e-05,
      "model_forward_time": 2.263087272644043,
      "step": 18009
    },
    {
      "epoch": 1.7174720764160157e-05,
      "step": 18009,
      "training_step_time": 8.621352434158325
    },
    {
      "epoch": 1.7175674438476564e-05,
      "grad_norm": 0.25876158475875854,
      "learning_rate": 9.544412251350407e-06,
      "loss": 0.0707,
      "step": 18010
    },
    {
      "epoch": 1.7175674438476564e-05,
      "model_forward_time": 1.7100656032562256,
      "step": 18010
    },
    {
      "dynamics_loss_avg": 0.05760728381574154,
      "epoch": 1.7175674438476564e-05,
      "step": 18010
    },
    {
      "action_loss_avg": 0.0025228470272850244,
      "epoch": 1.7175674438476564e-05,
      "step": 18010
    },
    {
      "epoch": 1.7175674438476564e-05,
      "step": 18010,
      "training_step_time": 8.617546558380127
    },
    {
      "epoch": 1.7176628112792968e-05,
      "model_forward_time": 1.7654774188995361,
      "step": 18011
    },
    {
      "epoch": 1.7176628112792968e-05,
      "step": 18011,
      "training_step_time": 7.684401750564575
    },
    {
      "epoch": 1.7177581787109375e-05,
      "model_forward_time": 1.7100160121917725,
      "step": 18012
    },
    {
      "epoch": 1.7177581787109375e-05,
      "step": 18012,
      "training_step_time": 7.270350933074951
    },
    {
      "epoch": 1.7178535461425783e-05,
      "model_forward_time": 1.7151367664337158,
      "step": 18013
    },
    {
      "epoch": 1.7178535461425783e-05,
      "step": 18013,
      "training_step_time": 7.537135362625122
    },
    {
      "epoch": 1.7179489135742187e-05,
      "model_forward_time": 1.7145931720733643,
      "step": 18014
    },
    {
      "epoch": 1.7179489135742187e-05,
      "step": 18014,
      "training_step_time": 7.6861395835876465
    },
    {
      "epoch": 1.7180442810058594e-05,
      "model_forward_time": 1.7187893390655518,
      "step": 18015
    },
    {
      "epoch": 1.7180442810058594e-05,
      "step": 18015,
      "training_step_time": 8.509115219116211
    },
    {
      "epoch": 1.7181396484375e-05,
      "model_forward_time": 1.710233449935913,
      "step": 18016
    },
    {
      "epoch": 1.7181396484375e-05,
      "step": 18016,
      "training_step_time": 9.370814561843872
    },
    {
      "epoch": 1.7182350158691406e-05,
      "model_forward_time": 1.713794231414795,
      "step": 18017
    },
    {
      "epoch": 1.7182350158691406e-05,
      "step": 18017,
      "training_step_time": 8.623449325561523
    },
    {
      "epoch": 1.7183303833007813e-05,
      "model_forward_time": 1.3454315662384033,
      "step": 18018
    },
    {
      "epoch": 1.7183303833007813e-05,
      "step": 18018,
      "training_step_time": 8.523434400558472
    },
    {
      "epoch": 1.7184257507324217e-05,
      "model_forward_time": 2.2617313861846924,
      "step": 18019
    },
    {
      "epoch": 1.7184257507324217e-05,
      "step": 18019,
      "training_step_time": 8.501481294631958
    },
    {
      "epoch": 1.7185211181640625e-05,
      "grad_norm": 0.2467217892408371,
      "learning_rate": 9.543722419347423e-06,
      "loss": 0.0662,
      "step": 18020
    },
    {
      "epoch": 1.7185211181640625e-05,
      "model_forward_time": 1.7687811851501465,
      "step": 18020
    },
    {
      "dynamics_loss_avg": 0.056982658430933954,
      "epoch": 1.7185211181640625e-05,
      "step": 18020
    },
    {
      "action_loss_avg": 0.0025477275368757544,
      "epoch": 1.7185211181640625e-05,
      "step": 18020
    },
    {
      "epoch": 1.7185211181640625e-05,
      "step": 18020,
      "training_step_time": 7.646589517593384
    },
    {
      "epoch": 1.7186164855957032e-05,
      "model_forward_time": 1.7156543731689453,
      "step": 18021
    },
    {
      "epoch": 1.7186164855957032e-05,
      "step": 18021,
      "training_step_time": 7.479483366012573
    },
    {
      "epoch": 1.7187118530273436e-05,
      "model_forward_time": 1.713127613067627,
      "step": 18022
    },
    {
      "epoch": 1.7187118530273436e-05,
      "step": 18022,
      "training_step_time": 7.42074990272522
    },
    {
      "epoch": 1.7188072204589843e-05,
      "model_forward_time": 1.7160966396331787,
      "step": 18023
    },
    {
      "epoch": 1.7188072204589843e-05,
      "step": 18023,
      "training_step_time": 7.466643810272217
    },
    {
      "epoch": 1.718902587890625e-05,
      "model_forward_time": 1.7043352127075195,
      "step": 18024
    },
    {
      "epoch": 1.718902587890625e-05,
      "step": 18024,
      "training_step_time": 8.897807359695435
    },
    {
      "epoch": 1.7189979553222655e-05,
      "model_forward_time": 1.7295396327972412,
      "step": 18025
    },
    {
      "epoch": 1.7189979553222655e-05,
      "step": 18025,
      "training_step_time": 9.56615161895752
    },
    {
      "epoch": 1.7190933227539062e-05,
      "model_forward_time": 1.7298784255981445,
      "step": 18026
    },
    {
      "epoch": 1.7190933227539062e-05,
      "step": 18026,
      "training_step_time": 7.614692687988281
    },
    {
      "epoch": 1.719188690185547e-05,
      "model_forward_time": 1.72483229637146,
      "step": 18027
    },
    {
      "epoch": 1.719188690185547e-05,
      "step": 18027,
      "training_step_time": 8.445903062820435
    },
    {
      "epoch": 1.7192840576171874e-05,
      "model_forward_time": 2.2900731563568115,
      "step": 18028
    },
    {
      "epoch": 1.7192840576171874e-05,
      "step": 18028,
      "training_step_time": 8.695868253707886
    },
    {
      "epoch": 1.719379425048828e-05,
      "model_forward_time": 1.7707080841064453,
      "step": 18029
    },
    {
      "epoch": 1.719379425048828e-05,
      "step": 18029,
      "training_step_time": 7.665858745574951
    },
    {
      "epoch": 1.719474792480469e-05,
      "grad_norm": 0.10954165458679199,
      "learning_rate": 9.543032090449789e-06,
      "loss": 0.0658,
      "step": 18030
    },
    {
      "epoch": 1.719474792480469e-05,
      "model_forward_time": 1.7687900066375732,
      "step": 18030
    },
    {
      "dynamics_loss_avg": 0.07429373860359192,
      "epoch": 1.719474792480469e-05,
      "step": 18030
    },
    {
      "action_loss_avg": 0.004277135385200382,
      "epoch": 1.719474792480469e-05,
      "step": 18030
    },
    {
      "epoch": 1.719474792480469e-05,
      "step": 18030,
      "training_step_time": 7.031421422958374
    },
    {
      "epoch": 1.7195701599121093e-05,
      "model_forward_time": 1.6924972534179688,
      "step": 18031
    },
    {
      "epoch": 1.7195701599121093e-05,
      "step": 18031,
      "training_step_time": 7.492431879043579
    },
    {
      "epoch": 1.71966552734375e-05,
      "model_forward_time": 1.7088947296142578,
      "step": 18032
    },
    {
      "epoch": 1.71966552734375e-05,
      "step": 18032,
      "training_step_time": 7.497226715087891
    },
    {
      "epoch": 1.7197608947753907e-05,
      "model_forward_time": 1.7131693363189697,
      "step": 18033
    },
    {
      "epoch": 1.7197608947753907e-05,
      "step": 18033,
      "training_step_time": 8.42553186416626
    },
    {
      "epoch": 1.719856262207031e-05,
      "model_forward_time": 1.7003874778747559,
      "step": 18034
    },
    {
      "epoch": 1.719856262207031e-05,
      "step": 18034,
      "training_step_time": 8.724666357040405
    },
    {
      "epoch": 1.719951629638672e-05,
      "model_forward_time": 1.7085456848144531,
      "step": 18035
    },
    {
      "epoch": 1.719951629638672e-05,
      "step": 18035,
      "training_step_time": 8.374258756637573
    },
    {
      "epoch": 1.7200469970703126e-05,
      "model_forward_time": 1.3430395126342773,
      "step": 18036
    },
    {
      "epoch": 1.7200469970703126e-05,
      "step": 18036,
      "training_step_time": 8.5971360206604
    },
    {
      "epoch": 1.720142364501953e-05,
      "model_forward_time": 2.2417352199554443,
      "step": 18037
    },
    {
      "epoch": 1.720142364501953e-05,
      "step": 18037,
      "training_step_time": 7.7393505573272705
    },
    {
      "epoch": 1.7202377319335938e-05,
      "model_forward_time": 1.7442355155944824,
      "step": 18038
    },
    {
      "epoch": 1.7202377319335938e-05,
      "step": 18038,
      "training_step_time": 7.782207727432251
    },
    {
      "epoch": 1.7203330993652345e-05,
      "model_forward_time": 1.7534501552581787,
      "step": 18039
    },
    {
      "epoch": 1.7203330993652345e-05,
      "step": 18039,
      "training_step_time": 7.268229007720947
    },
    {
      "epoch": 1.720428466796875e-05,
      "grad_norm": 0.17928336560726166,
      "learning_rate": 9.542341264732993e-06,
      "loss": 0.0648,
      "step": 18040
    },
    {
      "epoch": 1.720428466796875e-05,
      "model_forward_time": 1.7102084159851074,
      "step": 18040
    },
    {
      "dynamics_loss_avg": 0.06637705825269222,
      "epoch": 1.720428466796875e-05,
      "step": 18040
    },
    {
      "action_loss_avg": 0.002764610736630857,
      "epoch": 1.720428466796875e-05,
      "step": 18040
    },
    {
      "epoch": 1.720428466796875e-05,
      "step": 18040,
      "training_step_time": 7.639533281326294
    },
    {
      "epoch": 1.7205238342285157e-05,
      "model_forward_time": 1.6944539546966553,
      "step": 18041
    },
    {
      "epoch": 1.7205238342285157e-05,
      "step": 18041,
      "training_step_time": 7.298041105270386
    },
    {
      "epoch": 1.7206192016601564e-05,
      "model_forward_time": 1.7003579139709473,
      "step": 18042
    },
    {
      "epoch": 1.7206192016601564e-05,
      "step": 18042,
      "training_step_time": 7.65061354637146
    },
    {
      "epoch": 1.7207145690917968e-05,
      "model_forward_time": 1.700206995010376,
      "step": 18043
    },
    {
      "epoch": 1.7207145690917968e-05,
      "step": 18043,
      "training_step_time": 7.809366464614868
    },
    {
      "epoch": 1.7208099365234375e-05,
      "model_forward_time": 1.6975271701812744,
      "step": 18044
    },
    {
      "epoch": 1.7208099365234375e-05,
      "step": 18044,
      "training_step_time": 8.575483560562134
    },
    {
      "epoch": 1.7209053039550783e-05,
      "model_forward_time": 1.701094150543213,
      "step": 18045
    },
    {
      "epoch": 1.7209053039550783e-05,
      "step": 18045,
      "training_step_time": 8.279966354370117
    },
    {
      "epoch": 1.7210006713867187e-05,
      "model_forward_time": 2.250225782394409,
      "step": 18046
    },
    {
      "epoch": 1.7210006713867187e-05,
      "step": 18046,
      "training_step_time": 7.739374876022339
    },
    {
      "epoch": 1.7210960388183594e-05,
      "model_forward_time": 1.698106288909912,
      "step": 18047
    },
    {
      "epoch": 1.7210960388183594e-05,
      "step": 18047,
      "training_step_time": 7.682455539703369
    },
    {
      "epoch": 1.72119140625e-05,
      "model_forward_time": 1.7409100532531738,
      "step": 18048
    },
    {
      "epoch": 1.72119140625e-05,
      "step": 18048,
      "training_step_time": 7.49326229095459
    },
    {
      "epoch": 1.7212867736816406e-05,
      "model_forward_time": 1.7030937671661377,
      "step": 18049
    },
    {
      "epoch": 1.7212867736816406e-05,
      "step": 18049,
      "training_step_time": 7.585745573043823
    },
    {
      "epoch": 1.7213821411132813e-05,
      "grad_norm": 0.15214753150939941,
      "learning_rate": 9.541649942272586e-06,
      "loss": 0.0632,
      "step": 18050
    },
    {
      "epoch": 1.7213821411132813e-05,
      "model_forward_time": 1.7019553184509277,
      "step": 18050
    },
    {
      "dynamics_loss_avg": 0.06564473640173674,
      "epoch": 1.7213821411132813e-05,
      "step": 18050
    },
    {
      "action_loss_avg": 0.0028368980274535716,
      "epoch": 1.7213821411132813e-05,
      "step": 18050
    },
    {
      "epoch": 1.7213821411132813e-05,
      "step": 18050,
      "training_step_time": 7.8492655754089355
    },
    {
      "epoch": 1.7214775085449217e-05,
      "model_forward_time": 1.7017159461975098,
      "step": 18051
    },
    {
      "epoch": 1.7214775085449217e-05,
      "step": 18051,
      "training_step_time": 7.681467771530151
    },
    {
      "epoch": 1.7215728759765625e-05,
      "model_forward_time": 1.7021918296813965,
      "step": 18052
    },
    {
      "epoch": 1.7215728759765625e-05,
      "step": 18052,
      "training_step_time": 8.320549964904785
    },
    {
      "epoch": 1.7216682434082032e-05,
      "model_forward_time": 1.7017614841461182,
      "step": 18053
    },
    {
      "epoch": 1.7216682434082032e-05,
      "step": 18053,
      "training_step_time": 8.558329105377197
    },
    {
      "epoch": 1.7217636108398436e-05,
      "model_forward_time": 1.7150087356567383,
      "step": 18054
    },
    {
      "epoch": 1.7217636108398436e-05,
      "step": 18054,
      "training_step_time": 7.744051456451416
    },
    {
      "epoch": 1.7218589782714843e-05,
      "model_forward_time": 1.7125210762023926,
      "step": 18055
    },
    {
      "epoch": 1.7218589782714843e-05,
      "step": 18055,
      "training_step_time": 8.937958478927612
    },
    {
      "epoch": 1.721954345703125e-05,
      "model_forward_time": 1.7100443840026855,
      "step": 18056
    },
    {
      "epoch": 1.721954345703125e-05,
      "step": 18056,
      "training_step_time": 7.698585748672485
    },
    {
      "epoch": 1.7220497131347655e-05,
      "model_forward_time": 1.7328972816467285,
      "step": 18057
    },
    {
      "epoch": 1.7220497131347655e-05,
      "step": 18057,
      "training_step_time": 7.573185205459595
    },
    {
      "epoch": 1.7221450805664062e-05,
      "model_forward_time": 1.7061243057250977,
      "step": 18058
    },
    {
      "epoch": 1.7221450805664062e-05,
      "step": 18058,
      "training_step_time": 7.541936874389648
    },
    {
      "epoch": 1.722240447998047e-05,
      "model_forward_time": 1.7076225280761719,
      "step": 18059
    },
    {
      "epoch": 1.722240447998047e-05,
      "step": 18059,
      "training_step_time": 8.071438074111938
    },
    {
      "epoch": 1.7223358154296874e-05,
      "grad_norm": 0.16222098469734192,
      "learning_rate": 9.540958123144171e-06,
      "loss": 0.0638,
      "step": 18060
    },
    {
      "epoch": 1.7223358154296874e-05,
      "model_forward_time": 1.7126884460449219,
      "step": 18060
    },
    {
      "dynamics_loss_avg": 0.06151061095297337,
      "epoch": 1.7223358154296874e-05,
      "step": 18060
    },
    {
      "action_loss_avg": 0.0034007958369329573,
      "epoch": 1.7223358154296874e-05,
      "step": 18060
    },
    {
      "epoch": 1.7223358154296874e-05,
      "step": 18060,
      "training_step_time": 7.588583469390869
    },
    {
      "epoch": 1.722431182861328e-05,
      "model_forward_time": 1.715038537979126,
      "step": 18061
    },
    {
      "epoch": 1.722431182861328e-05,
      "step": 18061,
      "training_step_time": 8.448911905288696
    },
    {
      "epoch": 1.722526550292969e-05,
      "model_forward_time": 1.7049126625061035,
      "step": 18062
    },
    {
      "epoch": 1.722526550292969e-05,
      "step": 18062,
      "training_step_time": 8.491252422332764
    },
    {
      "epoch": 1.7226219177246093e-05,
      "model_forward_time": 1.7174055576324463,
      "step": 18063
    },
    {
      "epoch": 1.7226219177246093e-05,
      "step": 18063,
      "training_step_time": 8.654358863830566
    },
    {
      "epoch": 1.72271728515625e-05,
      "model_forward_time": 1.7142438888549805,
      "step": 18064
    },
    {
      "epoch": 1.72271728515625e-05,
      "step": 18064,
      "training_step_time": 7.647996664047241
    },
    {
      "epoch": 1.7228126525878907e-05,
      "model_forward_time": 2.2458605766296387,
      "step": 18065
    },
    {
      "epoch": 1.7228126525878907e-05,
      "step": 18065,
      "training_step_time": 7.759521245956421
    },
    {
      "epoch": 1.722908020019531e-05,
      "model_forward_time": 1.7244420051574707,
      "step": 18066
    },
    {
      "epoch": 1.722908020019531e-05,
      "step": 18066,
      "training_step_time": 7.608126640319824
    },
    {
      "epoch": 1.723003387451172e-05,
      "model_forward_time": 1.8115737438201904,
      "step": 18067
    },
    {
      "epoch": 1.723003387451172e-05,
      "step": 18067,
      "training_step_time": 7.012395620346069
    },
    {
      "epoch": 1.7230987548828126e-05,
      "model_forward_time": 1.8050379753112793,
      "step": 18068
    },
    {
      "epoch": 1.7230987548828126e-05,
      "step": 18068,
      "training_step_time": 7.565385103225708
    },
    {
      "epoch": 1.723194122314453e-05,
      "model_forward_time": 1.808119535446167,
      "step": 18069
    },
    {
      "epoch": 1.723194122314453e-05,
      "step": 18069,
      "training_step_time": 8.305805683135986
    },
    {
      "epoch": 1.7232894897460938e-05,
      "grad_norm": 0.14562581479549408,
      "learning_rate": 9.540265807423401e-06,
      "loss": 0.0633,
      "step": 18070
    },
    {
      "epoch": 1.7232894897460938e-05,
      "model_forward_time": 1.8062400817871094,
      "step": 18070
    },
    {
      "dynamics_loss_avg": 0.07243938185274601,
      "epoch": 1.7232894897460938e-05,
      "step": 18070
    },
    {
      "action_loss_avg": 0.0037363341252785176,
      "epoch": 1.7232894897460938e-05,
      "step": 18070
    },
    {
      "epoch": 1.7232894897460938e-05,
      "step": 18070,
      "training_step_time": 8.265245914459229
    },
    {
      "epoch": 1.7233848571777345e-05,
      "model_forward_time": 1.8079957962036133,
      "step": 18071
    },
    {
      "epoch": 1.7233848571777345e-05,
      "step": 18071,
      "training_step_time": 8.297555685043335
    },
    {
      "epoch": 1.723480224609375e-05,
      "model_forward_time": 1.8074259757995605,
      "step": 18072
    },
    {
      "epoch": 1.723480224609375e-05,
      "step": 18072,
      "training_step_time": 7.849349737167358
    },
    {
      "epoch": 1.7235755920410157e-05,
      "model_forward_time": 1.8061141967773438,
      "step": 18073
    },
    {
      "epoch": 1.7235755920410157e-05,
      "step": 18073,
      "training_step_time": 9.681071281433105
    },
    {
      "epoch": 1.7236709594726564e-05,
      "model_forward_time": 2.333428382873535,
      "step": 18074
    },
    {
      "epoch": 1.7236709594726564e-05,
      "step": 18074,
      "training_step_time": 8.94910216331482
    },
    {
      "epoch": 1.7237663269042968e-05,
      "model_forward_time": 1.8062324523925781,
      "step": 18075
    },
    {
      "epoch": 1.7237663269042968e-05,
      "step": 18075,
      "training_step_time": 8.719270706176758
    },
    {
      "epoch": 1.7238616943359375e-05,
      "model_forward_time": 1.811070203781128,
      "step": 18076
    },
    {
      "epoch": 1.7238616943359375e-05,
      "step": 18076,
      "training_step_time": 7.615966320037842
    },
    {
      "epoch": 1.7239570617675783e-05,
      "model_forward_time": 1.803508996963501,
      "step": 18077
    },
    {
      "epoch": 1.7239570617675783e-05,
      "step": 18077,
      "training_step_time": 7.7508766651153564
    },
    {
      "epoch": 1.7240524291992187e-05,
      "model_forward_time": 1.802098035812378,
      "step": 18078
    },
    {
      "epoch": 1.7240524291992187e-05,
      "step": 18078,
      "training_step_time": 8.444926738739014
    },
    {
      "epoch": 1.7241477966308594e-05,
      "model_forward_time": 1.4297354221343994,
      "step": 18079
    },
    {
      "epoch": 1.7241477966308594e-05,
      "step": 18079,
      "training_step_time": 9.043655633926392
    },
    {
      "epoch": 1.7242431640625e-05,
      "grad_norm": 0.18688072264194489,
      "learning_rate": 9.53957299518599e-06,
      "loss": 0.0693,
      "step": 18080
    },
    {
      "epoch": 1.7242431640625e-05,
      "model_forward_time": 1.8011915683746338,
      "step": 18080
    },
    {
      "dynamics_loss_avg": 0.05905848015099764,
      "epoch": 1.7242431640625e-05,
      "step": 18080
    },
    {
      "action_loss_avg": 0.002091550477780402,
      "epoch": 1.7242431640625e-05,
      "step": 18080
    },
    {
      "epoch": 1.7242431640625e-05,
      "step": 18080,
      "training_step_time": 7.721914529800415
    },
    {
      "epoch": 1.7243385314941406e-05,
      "model_forward_time": 1.8161401748657227,
      "step": 18081
    },
    {
      "epoch": 1.7243385314941406e-05,
      "step": 18081,
      "training_step_time": 8.941250324249268
    },
    {
      "epoch": 1.7244338989257813e-05,
      "model_forward_time": 1.428778886795044,
      "step": 18082
    },
    {
      "epoch": 1.7244338989257813e-05,
      "step": 18082,
      "training_step_time": 7.90002703666687
    },
    {
      "epoch": 1.7245292663574217e-05,
      "model_forward_time": 2.3410933017730713,
      "step": 18083
    },
    {
      "epoch": 1.7245292663574217e-05,
      "step": 18083,
      "training_step_time": 7.939971446990967
    },
    {
      "epoch": 1.7246246337890625e-05,
      "model_forward_time": 1.8017354011535645,
      "step": 18084
    },
    {
      "epoch": 1.7246246337890625e-05,
      "step": 18084,
      "training_step_time": 7.938429355621338
    },
    {
      "epoch": 1.7247200012207032e-05,
      "model_forward_time": 1.8092870712280273,
      "step": 18085
    },
    {
      "epoch": 1.7247200012207032e-05,
      "step": 18085,
      "training_step_time": 7.781047821044922
    },
    {
      "epoch": 1.7248153686523436e-05,
      "model_forward_time": 1.8098421096801758,
      "step": 18086
    },
    {
      "epoch": 1.7248153686523436e-05,
      "step": 18086,
      "training_step_time": 7.776438474655151
    },
    {
      "epoch": 1.7249107360839844e-05,
      "model_forward_time": 1.807931661605835,
      "step": 18087
    },
    {
      "epoch": 1.7249107360839844e-05,
      "step": 18087,
      "training_step_time": 7.466513395309448
    },
    {
      "epoch": 1.725006103515625e-05,
      "model_forward_time": 1.8093798160552979,
      "step": 18088
    },
    {
      "epoch": 1.725006103515625e-05,
      "step": 18088,
      "training_step_time": 7.625463008880615
    },
    {
      "epoch": 1.7251014709472655e-05,
      "model_forward_time": 1.8124265670776367,
      "step": 18089
    },
    {
      "epoch": 1.7251014709472655e-05,
      "step": 18089,
      "training_step_time": 8.533899545669556
    },
    {
      "epoch": 1.7251968383789062e-05,
      "grad_norm": 0.16989271342754364,
      "learning_rate": 9.5388796865077e-06,
      "loss": 0.0645,
      "step": 18090
    },
    {
      "epoch": 1.7251968383789062e-05,
      "model_forward_time": 1.8109965324401855,
      "step": 18090
    },
    {
      "dynamics_loss_avg": 0.06843584198504686,
      "epoch": 1.7251968383789062e-05,
      "step": 18090
    },
    {
      "action_loss_avg": 0.0034201615955680607,
      "epoch": 1.7251968383789062e-05,
      "step": 18090
    },
    {
      "epoch": 1.7251968383789062e-05,
      "step": 18090,
      "training_step_time": 7.67603874206543
    },
    {
      "epoch": 1.725292205810547e-05,
      "model_forward_time": 1.8087027072906494,
      "step": 18091
    },
    {
      "epoch": 1.725292205810547e-05,
      "step": 18091,
      "training_step_time": 9.5537269115448
    },
    {
      "epoch": 1.7253875732421874e-05,
      "model_forward_time": 1.8057878017425537,
      "step": 18092
    },
    {
      "epoch": 1.7253875732421874e-05,
      "step": 18092,
      "training_step_time": 8.67484998703003
    },
    {
      "epoch": 1.725482940673828e-05,
      "model_forward_time": 1.8041002750396729,
      "step": 18093
    },
    {
      "epoch": 1.725482940673828e-05,
      "step": 18093,
      "training_step_time": 8.245596647262573
    },
    {
      "epoch": 1.725578308105469e-05,
      "model_forward_time": 1.8015894889831543,
      "step": 18094
    },
    {
      "epoch": 1.725578308105469e-05,
      "step": 18094,
      "training_step_time": 7.46365213394165
    },
    {
      "epoch": 1.7256736755371093e-05,
      "model_forward_time": 1.8049452304840088,
      "step": 18095
    },
    {
      "epoch": 1.7256736755371093e-05,
      "step": 18095,
      "training_step_time": 8.414608240127563
    },
    {
      "epoch": 1.72576904296875e-05,
      "model_forward_time": 1.791715145111084,
      "step": 18096
    },
    {
      "epoch": 1.72576904296875e-05,
      "step": 18096,
      "training_step_time": 7.204261064529419
    },
    {
      "epoch": 1.7258644104003907e-05,
      "model_forward_time": 1.8045811653137207,
      "step": 18097
    },
    {
      "epoch": 1.7258644104003907e-05,
      "step": 18097,
      "training_step_time": 7.586186408996582
    },
    {
      "epoch": 1.725959777832031e-05,
      "model_forward_time": 1.8072984218597412,
      "step": 18098
    },
    {
      "epoch": 1.725959777832031e-05,
      "step": 18098,
      "training_step_time": 8.399355173110962
    },
    {
      "epoch": 1.726055145263672e-05,
      "model_forward_time": 1.813857078552246,
      "step": 18099
    },
    {
      "epoch": 1.726055145263672e-05,
      "step": 18099,
      "training_step_time": 8.177039623260498
    },
    {
      "epoch": 1.7261505126953126e-05,
      "grad_norm": 0.14458328485488892,
      "learning_rate": 9.538185881464354e-06,
      "loss": 0.0676,
      "step": 18100
    },
    {
      "epoch": 1.7261505126953126e-05,
      "model_forward_time": 1.4199638366699219,
      "step": 18100
    },
    {
      "dynamics_loss_avg": 0.05750978980213404,
      "epoch": 1.7261505126953126e-05,
      "step": 18100
    },
    {
      "action_loss_avg": 0.0024345130485016854,
      "epoch": 1.7261505126953126e-05,
      "step": 18100
    },
    {
      "epoch": 1.7261505126953126e-05,
      "step": 18100,
      "training_step_time": 7.659156560897827
    },
    {
      "epoch": 1.726245880126953e-05,
      "model_forward_time": 1.8074893951416016,
      "step": 18101
    },
    {
      "epoch": 1.726245880126953e-05,
      "step": 18101,
      "training_step_time": 8.155048608779907
    },
    {
      "epoch": 1.7263412475585938e-05,
      "model_forward_time": 1.9732413291931152,
      "step": 18102
    },
    {
      "epoch": 1.7263412475585938e-05,
      "step": 18102,
      "training_step_time": 7.644571304321289
    },
    {
      "epoch": 1.7264366149902345e-05,
      "model_forward_time": 1.805156946182251,
      "step": 18103
    },
    {
      "epoch": 1.7264366149902345e-05,
      "step": 18103,
      "training_step_time": 7.422536611557007
    },
    {
      "epoch": 1.726531982421875e-05,
      "model_forward_time": 1.811018705368042,
      "step": 18104
    },
    {
      "epoch": 1.726531982421875e-05,
      "step": 18104,
      "training_step_time": 7.5213234424591064
    },
    {
      "epoch": 1.7266273498535157e-05,
      "model_forward_time": 1.812741756439209,
      "step": 18105
    },
    {
      "epoch": 1.7266273498535157e-05,
      "step": 18105,
      "training_step_time": 7.398516893386841
    },
    {
      "epoch": 1.7267227172851564e-05,
      "model_forward_time": 1.807175874710083,
      "step": 18106
    },
    {
      "epoch": 1.7267227172851564e-05,
      "step": 18106,
      "training_step_time": 7.553725004196167
    },
    {
      "epoch": 1.7268180847167968e-05,
      "model_forward_time": 1.8065364360809326,
      "step": 18107
    },
    {
      "epoch": 1.7268180847167968e-05,
      "step": 18107,
      "training_step_time": 8.493046522140503
    },
    {
      "epoch": 1.7269134521484376e-05,
      "model_forward_time": 1.8167307376861572,
      "step": 18108
    },
    {
      "epoch": 1.7269134521484376e-05,
      "step": 18108,
      "training_step_time": 7.5999915599823
    },
    {
      "epoch": 1.7270088195800783e-05,
      "model_forward_time": 1.806861400604248,
      "step": 18109
    },
    {
      "epoch": 1.7270088195800783e-05,
      "step": 18109,
      "training_step_time": 7.69210958480835
    },
    {
      "epoch": 1.7271041870117187e-05,
      "grad_norm": 0.2669121026992798,
      "learning_rate": 9.53749158013182e-06,
      "loss": 0.0612,
      "step": 18110
    },
    {
      "epoch": 1.7271041870117187e-05,
      "model_forward_time": 1.802962303161621,
      "step": 18110
    },
    {
      "dynamics_loss_avg": 0.053896679542958736,
      "epoch": 1.7271041870117187e-05,
      "step": 18110
    },
    {
      "action_loss_avg": 0.0023923026164993644,
      "epoch": 1.7271041870117187e-05,
      "step": 18110
    },
    {
      "epoch": 1.7271041870117187e-05,
      "step": 18110,
      "training_step_time": 8.348275423049927
    },
    {
      "epoch": 1.7271995544433594e-05,
      "model_forward_time": 2.3419241905212402,
      "step": 18111
    },
    {
      "epoch": 1.7271995544433594e-05,
      "step": 18111,
      "training_step_time": 8.512181997299194
    },
    {
      "epoch": 1.727294921875e-05,
      "model_forward_time": 1.8051221370697021,
      "step": 18112
    },
    {
      "epoch": 1.727294921875e-05,
      "step": 18112,
      "training_step_time": 7.640536069869995
    },
    {
      "epoch": 1.7273902893066406e-05,
      "model_forward_time": 1.8073346614837646,
      "step": 18113
    },
    {
      "epoch": 1.7273902893066406e-05,
      "step": 18113,
      "training_step_time": 7.461989164352417
    },
    {
      "epoch": 1.7274856567382813e-05,
      "model_forward_time": 1.8035321235656738,
      "step": 18114
    },
    {
      "epoch": 1.7274856567382813e-05,
      "step": 18114,
      "training_step_time": 7.376375436782837
    },
    {
      "epoch": 1.7275810241699217e-05,
      "model_forward_time": 1.4316940307617188,
      "step": 18115
    },
    {
      "epoch": 1.7275810241699217e-05,
      "step": 18115,
      "training_step_time": 8.016348600387573
    },
    {
      "epoch": 1.7276763916015625e-05,
      "model_forward_time": 1.800269365310669,
      "step": 18116
    },
    {
      "epoch": 1.7276763916015625e-05,
      "step": 18116,
      "training_step_time": 7.673347234725952
    },
    {
      "epoch": 1.7277717590332032e-05,
      "model_forward_time": 1.813185691833496,
      "step": 18117
    },
    {
      "epoch": 1.7277717590332032e-05,
      "step": 18117,
      "training_step_time": 8.523137092590332
    },
    {
      "epoch": 1.7278671264648436e-05,
      "model_forward_time": 1.806555986404419,
      "step": 18118
    },
    {
      "epoch": 1.7278671264648436e-05,
      "step": 18118,
      "training_step_time": 7.700360536575317
    },
    {
      "epoch": 1.7279624938964844e-05,
      "model_forward_time": 1.8072121143341064,
      "step": 18119
    },
    {
      "epoch": 1.7279624938964844e-05,
      "step": 18119,
      "training_step_time": 8.59874415397644
    },
    {
      "epoch": 1.728057861328125e-05,
      "grad_norm": 0.2196635752916336,
      "learning_rate": 9.536796782586031e-06,
      "loss": 0.0623,
      "step": 18120
    },
    {
      "epoch": 1.728057861328125e-05,
      "model_forward_time": 1.8011605739593506,
      "step": 18120
    },
    {
      "dynamics_loss_avg": 0.06135166324675083,
      "epoch": 1.728057861328125e-05,
      "step": 18120
    },
    {
      "action_loss_avg": 0.0024613072542706504,
      "epoch": 1.728057861328125e-05,
      "step": 18120
    },
    {
      "epoch": 1.728057861328125e-05,
      "step": 18120,
      "training_step_time": 8.749666213989258
    },
    {
      "epoch": 1.7281532287597655e-05,
      "model_forward_time": 1.7995109558105469,
      "step": 18121
    },
    {
      "epoch": 1.7281532287597655e-05,
      "step": 18121,
      "training_step_time": 7.781176328659058
    },
    {
      "epoch": 1.7282485961914062e-05,
      "model_forward_time": 1.8051092624664307,
      "step": 18122
    },
    {
      "epoch": 1.7282485961914062e-05,
      "step": 18122,
      "training_step_time": 8.084405183792114
    },
    {
      "epoch": 1.728343963623047e-05,
      "model_forward_time": 1.8090498447418213,
      "step": 18123
    },
    {
      "epoch": 1.728343963623047e-05,
      "step": 18123,
      "training_step_time": 7.682779550552368
    },
    {
      "epoch": 1.7284393310546874e-05,
      "model_forward_time": 1.8044853210449219,
      "step": 18124
    },
    {
      "epoch": 1.7284393310546874e-05,
      "step": 18124,
      "training_step_time": 8.14086103439331
    },
    {
      "epoch": 1.728534698486328e-05,
      "model_forward_time": 1.8033256530761719,
      "step": 18125
    },
    {
      "epoch": 1.728534698486328e-05,
      "step": 18125,
      "training_step_time": 7.690948247909546
    },
    {
      "epoch": 1.728630065917969e-05,
      "model_forward_time": 1.8053538799285889,
      "step": 18126
    },
    {
      "epoch": 1.728630065917969e-05,
      "step": 18126,
      "training_step_time": 7.667041063308716
    },
    {
      "epoch": 1.7287254333496093e-05,
      "model_forward_time": 1.8056588172912598,
      "step": 18127
    },
    {
      "epoch": 1.7287254333496093e-05,
      "step": 18127,
      "training_step_time": 8.31780195236206
    },
    {
      "epoch": 1.72882080078125e-05,
      "model_forward_time": 1.8076896667480469,
      "step": 18128
    },
    {
      "epoch": 1.72882080078125e-05,
      "step": 18128,
      "training_step_time": 7.691833972930908
    },
    {
      "epoch": 1.7289161682128908e-05,
      "model_forward_time": 1.7933893203735352,
      "step": 18129
    },
    {
      "epoch": 1.7289161682128908e-05,
      "step": 18129,
      "training_step_time": 8.264562845230103
    },
    {
      "epoch": 1.729011535644531e-05,
      "grad_norm": 0.22286981344223022,
      "learning_rate": 9.536101488902966e-06,
      "loss": 0.0619,
      "step": 18130
    },
    {
      "epoch": 1.729011535644531e-05,
      "model_forward_time": 2.345360517501831,
      "step": 18130
    },
    {
      "dynamics_loss_avg": 0.0610165610909462,
      "epoch": 1.729011535644531e-05,
      "step": 18130
    },
    {
      "action_loss_avg": 0.0028124147618655117,
      "epoch": 1.729011535644531e-05,
      "step": 18130
    },
    {
      "epoch": 1.729011535644531e-05,
      "step": 18130,
      "training_step_time": 8.63746190071106
    },
    {
      "epoch": 1.729106903076172e-05,
      "model_forward_time": 1.433896541595459,
      "step": 18131
    },
    {
      "epoch": 1.729106903076172e-05,
      "step": 18131,
      "training_step_time": 7.4356114864349365
    },
    {
      "epoch": 1.7292022705078126e-05,
      "model_forward_time": 1.8151631355285645,
      "step": 18132
    },
    {
      "epoch": 1.7292022705078126e-05,
      "step": 18132,
      "training_step_time": 7.61283016204834
    },
    {
      "epoch": 1.729297637939453e-05,
      "model_forward_time": 1.0934607982635498,
      "step": 18133
    },
    {
      "epoch": 1.729297637939453e-05,
      "step": 18133,
      "training_step_time": 8.261530637741089
    },
    {
      "epoch": 1.7293930053710938e-05,
      "model_forward_time": 1.8146114349365234,
      "step": 18134
    },
    {
      "epoch": 1.7293930053710938e-05,
      "step": 18134,
      "training_step_time": 7.59799599647522
    },
    {
      "epoch": 1.7294883728027345e-05,
      "model_forward_time": 1.8222439289093018,
      "step": 18135
    },
    {
      "epoch": 1.7294883728027345e-05,
      "step": 18135,
      "training_step_time": 8.457359790802002
    },
    {
      "epoch": 1.729583740234375e-05,
      "model_forward_time": 1.8201124668121338,
      "step": 18136
    },
    {
      "epoch": 1.729583740234375e-05,
      "step": 18136,
      "training_step_time": 8.479514360427856
    },
    {
      "epoch": 1.7296791076660157e-05,
      "model_forward_time": 1.8198304176330566,
      "step": 18137
    },
    {
      "epoch": 1.7296791076660157e-05,
      "step": 18137,
      "training_step_time": 7.680249452590942
    },
    {
      "epoch": 1.7297744750976564e-05,
      "model_forward_time": 1.822758674621582,
      "step": 18138
    },
    {
      "epoch": 1.7297744750976564e-05,
      "step": 18138,
      "training_step_time": 8.280220985412598
    },
    {
      "epoch": 1.7298698425292968e-05,
      "model_forward_time": 2.359973430633545,
      "step": 18139
    },
    {
      "epoch": 1.7298698425292968e-05,
      "step": 18139,
      "training_step_time": 7.718869924545288
    },
    {
      "epoch": 1.7299652099609376e-05,
      "grad_norm": 0.21315062046051025,
      "learning_rate": 9.535405699158663e-06,
      "loss": 0.0631,
      "step": 18140
    },
    {
      "epoch": 1.7299652099609376e-05,
      "model_forward_time": 1.8195950984954834,
      "step": 18140
    },
    {
      "dynamics_loss_avg": 0.05728435106575489,
      "epoch": 1.7299652099609376e-05,
      "step": 18140
    },
    {
      "action_loss_avg": 0.0031168618093943222,
      "epoch": 1.7299652099609376e-05,
      "step": 18140
    },
    {
      "epoch": 1.7299652099609376e-05,
      "step": 18140,
      "training_step_time": 7.595305919647217
    },
    {
      "epoch": 1.730060577392578e-05,
      "model_forward_time": 1.8281371593475342,
      "step": 18141
    },
    {
      "epoch": 1.730060577392578e-05,
      "step": 18141,
      "training_step_time": 7.569093465805054
    },
    {
      "epoch": 1.7301559448242187e-05,
      "model_forward_time": 1.8219761848449707,
      "step": 18142
    },
    {
      "epoch": 1.7301559448242187e-05,
      "step": 18142,
      "training_step_time": 7.990516901016235
    },
    {
      "epoch": 1.7302513122558594e-05,
      "model_forward_time": 1.8243060111999512,
      "step": 18143
    },
    {
      "epoch": 1.7302513122558594e-05,
      "step": 18143,
      "training_step_time": 7.662920951843262
    },
    {
      "epoch": 1.7303466796875e-05,
      "model_forward_time": 1.8202581405639648,
      "step": 18144
    },
    {
      "epoch": 1.7303466796875e-05,
      "step": 18144,
      "training_step_time": 8.601135969161987
    },
    {
      "epoch": 1.7304420471191406e-05,
      "model_forward_time": 1.450505256652832,
      "step": 18145
    },
    {
      "epoch": 1.7304420471191406e-05,
      "step": 18145,
      "training_step_time": 9.669105768203735
    },
    {
      "epoch": 1.7305374145507813e-05,
      "model_forward_time": 1.796497106552124,
      "step": 18146
    },
    {
      "epoch": 1.7305374145507813e-05,
      "step": 18146,
      "training_step_time": 7.718581676483154
    },
    {
      "epoch": 1.7306327819824217e-05,
      "model_forward_time": 1.826265811920166,
      "step": 18147
    },
    {
      "epoch": 1.7306327819824217e-05,
      "step": 18147,
      "training_step_time": 8.370179176330566
    },
    {
      "epoch": 1.7307281494140625e-05,
      "model_forward_time": 1.8207402229309082,
      "step": 18148
    },
    {
      "epoch": 1.7307281494140625e-05,
      "step": 18148,
      "training_step_time": 8.399415254592896
    },
    {
      "epoch": 1.7308235168457032e-05,
      "model_forward_time": 2.3593766689300537,
      "step": 18149
    },
    {
      "epoch": 1.7308235168457032e-05,
      "step": 18149,
      "training_step_time": 8.479971408843994
    },
    {
      "epoch": 1.7309188842773436e-05,
      "grad_norm": 0.2398429960012436,
      "learning_rate": 9.53470941342921e-06,
      "loss": 0.063,
      "step": 18150
    },
    {
      "epoch": 1.7309188842773436e-05,
      "model_forward_time": 1.822530746459961,
      "step": 18150
    },
    {
      "dynamics_loss_avg": 0.050827374681830406,
      "epoch": 1.7309188842773436e-05,
      "step": 18150
    },
    {
      "action_loss_avg": 0.0023244121111929417,
      "epoch": 1.7309188842773436e-05,
      "step": 18150
    },
    {
      "epoch": 1.7309188842773436e-05,
      "step": 18150,
      "training_step_time": 7.5711493492126465
    },
    {
      "epoch": 1.7310142517089844e-05,
      "model_forward_time": 1.4471564292907715,
      "step": 18151
    },
    {
      "epoch": 1.7310142517089844e-05,
      "step": 18151,
      "training_step_time": 8.188808917999268
    },
    {
      "epoch": 1.731109619140625e-05,
      "model_forward_time": 1.8058533668518066,
      "step": 18152
    },
    {
      "epoch": 1.731109619140625e-05,
      "step": 18152,
      "training_step_time": 7.723018407821655
    },
    {
      "epoch": 1.7312049865722655e-05,
      "model_forward_time": 1.810422420501709,
      "step": 18153
    },
    {
      "epoch": 1.7312049865722655e-05,
      "step": 18153,
      "training_step_time": 7.491875171661377
    },
    {
      "epoch": 1.7313003540039062e-05,
      "model_forward_time": 1.828725814819336,
      "step": 18154
    },
    {
      "epoch": 1.7313003540039062e-05,
      "step": 18154,
      "training_step_time": 7.6892499923706055
    },
    {
      "epoch": 1.731395721435547e-05,
      "model_forward_time": 1.8179090023040771,
      "step": 18155
    },
    {
      "epoch": 1.731395721435547e-05,
      "step": 18155,
      "training_step_time": 8.78697919845581
    },
    {
      "epoch": 1.7314910888671874e-05,
      "model_forward_time": 1.8198401927947998,
      "step": 18156
    },
    {
      "epoch": 1.7314910888671874e-05,
      "step": 18156,
      "training_step_time": 9.154152631759644
    },
    {
      "epoch": 1.731586456298828e-05,
      "model_forward_time": 1.8107059001922607,
      "step": 18157
    },
    {
      "epoch": 1.731586456298828e-05,
      "step": 18157,
      "training_step_time": 8.811614036560059
    },
    {
      "epoch": 1.731681823730469e-05,
      "model_forward_time": 2.3369250297546387,
      "step": 18158
    },
    {
      "epoch": 1.731681823730469e-05,
      "step": 18158,
      "training_step_time": 8.623398780822754
    },
    {
      "epoch": 1.7317771911621093e-05,
      "model_forward_time": 1.8072807788848877,
      "step": 18159
    },
    {
      "epoch": 1.7317771911621093e-05,
      "step": 18159,
      "training_step_time": 7.4071691036224365
    },
    {
      "epoch": 1.73187255859375e-05,
      "grad_norm": 0.22224579751491547,
      "learning_rate": 9.534012631790756e-06,
      "loss": 0.0697,
      "step": 18160
    },
    {
      "epoch": 1.73187255859375e-05,
      "model_forward_time": 1.8136119842529297,
      "step": 18160
    },
    {
      "dynamics_loss_avg": 0.05384072195738554,
      "epoch": 1.73187255859375e-05,
      "step": 18160
    },
    {
      "action_loss_avg": 0.0016143087894306518,
      "epoch": 1.73187255859375e-05,
      "step": 18160
    },
    {
      "epoch": 1.73187255859375e-05,
      "step": 18160,
      "training_step_time": 7.550313472747803
    },
    {
      "epoch": 1.7319679260253908e-05,
      "model_forward_time": 1.8118271827697754,
      "step": 18161
    },
    {
      "epoch": 1.7319679260253908e-05,
      "step": 18161,
      "training_step_time": 8.62292766571045
    },
    {
      "epoch": 1.732063293457031e-05,
      "model_forward_time": 1.8088409900665283,
      "step": 18162
    },
    {
      "epoch": 1.732063293457031e-05,
      "step": 18162,
      "training_step_time": 8.281320810317993
    },
    {
      "epoch": 1.732158660888672e-05,
      "model_forward_time": 1.8209235668182373,
      "step": 18163
    },
    {
      "epoch": 1.732158660888672e-05,
      "step": 18163,
      "training_step_time": 7.702799558639526
    },
    {
      "epoch": 1.7322540283203126e-05,
      "model_forward_time": 1.816011905670166,
      "step": 18164
    },
    {
      "epoch": 1.7322540283203126e-05,
      "step": 18164,
      "training_step_time": 8.396985054016113
    },
    {
      "epoch": 1.732349395751953e-05,
      "model_forward_time": 1.8149633407592773,
      "step": 18165
    },
    {
      "epoch": 1.732349395751953e-05,
      "step": 18165,
      "training_step_time": 8.650059223175049
    },
    {
      "epoch": 1.7324447631835938e-05,
      "model_forward_time": 1.8180930614471436,
      "step": 18166
    },
    {
      "epoch": 1.7324447631835938e-05,
      "step": 18166,
      "training_step_time": 9.686941862106323
    },
    {
      "epoch": 1.7325401306152345e-05,
      "model_forward_time": 2.3480095863342285,
      "step": 18167
    },
    {
      "epoch": 1.7325401306152345e-05,
      "step": 18167,
      "training_step_time": 7.723077058792114
    },
    {
      "epoch": 1.732635498046875e-05,
      "model_forward_time": 1.806135654449463,
      "step": 18168
    },
    {
      "epoch": 1.732635498046875e-05,
      "step": 18168,
      "training_step_time": 7.5593812465667725
    },
    {
      "epoch": 1.7327308654785157e-05,
      "model_forward_time": 1.8114464282989502,
      "step": 18169
    },
    {
      "epoch": 1.7327308654785157e-05,
      "step": 18169,
      "training_step_time": 7.589538097381592
    },
    {
      "epoch": 1.7328262329101564e-05,
      "grad_norm": 0.22531549632549286,
      "learning_rate": 9.533315354319494e-06,
      "loss": 0.0652,
      "step": 18170
    },
    {
      "epoch": 1.7328262329101564e-05,
      "model_forward_time": 1.8075122833251953,
      "step": 18170
    },
    {
      "dynamics_loss_avg": 0.06793488543480634,
      "epoch": 1.7328262329101564e-05,
      "step": 18170
    },
    {
      "action_loss_avg": 0.003598677326226607,
      "epoch": 1.7328262329101564e-05,
      "step": 18170
    },
    {
      "epoch": 1.7328262329101564e-05,
      "step": 18170,
      "training_step_time": 7.604942083358765
    },
    {
      "epoch": 1.7329216003417968e-05,
      "model_forward_time": 1.8257296085357666,
      "step": 18171
    },
    {
      "epoch": 1.7329216003417968e-05,
      "step": 18171,
      "training_step_time": 8.205990552902222
    },
    {
      "epoch": 1.7330169677734376e-05,
      "model_forward_time": 1.8166849613189697,
      "step": 18172
    },
    {
      "epoch": 1.7330169677734376e-05,
      "step": 18172,
      "training_step_time": 7.709729909896851
    },
    {
      "epoch": 1.733112335205078e-05,
      "model_forward_time": 1.785649061203003,
      "step": 18173
    },
    {
      "epoch": 1.733112335205078e-05,
      "step": 18173,
      "training_step_time": 7.714383840560913
    },
    {
      "epoch": 1.7332077026367187e-05,
      "model_forward_time": 1.8024711608886719,
      "step": 18174
    },
    {
      "epoch": 1.7332077026367187e-05,
      "step": 18174,
      "training_step_time": 8.554608821868896
    },
    {
      "epoch": 1.7333030700683594e-05,
      "model_forward_time": 1.7942616939544678,
      "step": 18175
    },
    {
      "epoch": 1.7333030700683594e-05,
      "step": 18175,
      "training_step_time": 8.305602312088013
    },
    {
      "epoch": 1.7333984375e-05,
      "model_forward_time": 2.3286967277526855,
      "step": 18176
    },
    {
      "epoch": 1.7333984375e-05,
      "step": 18176,
      "training_step_time": 7.6954052448272705
    },
    {
      "epoch": 1.7334938049316406e-05,
      "model_forward_time": 1.7978196144104004,
      "step": 18177
    },
    {
      "epoch": 1.7334938049316406e-05,
      "step": 18177,
      "training_step_time": 7.53114652633667
    },
    {
      "epoch": 1.7335891723632813e-05,
      "model_forward_time": 1.7987475395202637,
      "step": 18178
    },
    {
      "epoch": 1.7335891723632813e-05,
      "step": 18178,
      "training_step_time": 7.6252501010894775
    },
    {
      "epoch": 1.7336845397949217e-05,
      "model_forward_time": 1.8079743385314941,
      "step": 18179
    },
    {
      "epoch": 1.7336845397949217e-05,
      "step": 18179,
      "training_step_time": 7.604313850402832
    },
    {
      "epoch": 1.7337799072265625e-05,
      "grad_norm": 0.23006436228752136,
      "learning_rate": 9.532617581091684e-06,
      "loss": 0.0697,
      "step": 18180
    },
    {
      "epoch": 1.7337799072265625e-05,
      "model_forward_time": 1.7975773811340332,
      "step": 18180
    },
    {
      "dynamics_loss_avg": 0.06288057491183281,
      "epoch": 1.7337799072265625e-05,
      "step": 18180
    },
    {
      "action_loss_avg": 0.0037270385422743858,
      "epoch": 1.7337799072265625e-05,
      "step": 18180
    },
    {
      "epoch": 1.7337799072265625e-05,
      "step": 18180,
      "training_step_time": 7.612668514251709
    },
    {
      "epoch": 1.7338752746582032e-05,
      "model_forward_time": 1.786484718322754,
      "step": 18181
    },
    {
      "epoch": 1.7338752746582032e-05,
      "step": 18181,
      "training_step_time": 7.689122676849365
    },
    {
      "epoch": 1.7339706420898436e-05,
      "model_forward_time": 1.8015990257263184,
      "step": 18182
    },
    {
      "epoch": 1.7339706420898436e-05,
      "step": 18182,
      "training_step_time": 9.446087837219238
    },
    {
      "epoch": 1.7340660095214844e-05,
      "model_forward_time": 1.8068885803222656,
      "step": 18183
    },
    {
      "epoch": 1.7340660095214844e-05,
      "step": 18183,
      "training_step_time": 10.553187370300293
    },
    {
      "epoch": 1.734161376953125e-05,
      "model_forward_time": 1.8030507564544678,
      "step": 18184
    },
    {
      "epoch": 1.734161376953125e-05,
      "step": 18184,
      "training_step_time": 7.786151170730591
    },
    {
      "epoch": 1.7342567443847655e-05,
      "model_forward_time": 2.3344521522521973,
      "step": 18185
    },
    {
      "epoch": 1.7342567443847655e-05,
      "step": 18185,
      "training_step_time": 7.681999683380127
    },
    {
      "epoch": 1.7343521118164062e-05,
      "model_forward_time": 1.8043673038482666,
      "step": 18186
    },
    {
      "epoch": 1.7343521118164062e-05,
      "step": 18186,
      "training_step_time": 7.630903482437134
    },
    {
      "epoch": 1.734447479248047e-05,
      "model_forward_time": 1.8074369430541992,
      "step": 18187
    },
    {
      "epoch": 1.734447479248047e-05,
      "step": 18187,
      "training_step_time": 7.706873893737793
    },
    {
      "epoch": 1.7345428466796874e-05,
      "model_forward_time": 1.7822582721710205,
      "step": 18188
    },
    {
      "epoch": 1.7345428466796874e-05,
      "step": 18188,
      "training_step_time": 7.6978230476379395
    },
    {
      "epoch": 1.734638214111328e-05,
      "model_forward_time": 1.43878173828125,
      "step": 18189
    },
    {
      "epoch": 1.734638214111328e-05,
      "step": 18189,
      "training_step_time": 7.635134696960449
    },
    {
      "epoch": 1.734733581542969e-05,
      "grad_norm": 0.22888655960559845,
      "learning_rate": 9.531919312183628e-06,
      "loss": 0.0711,
      "step": 18190
    },
    {
      "epoch": 1.734733581542969e-05,
      "model_forward_time": 1.079080581665039,
      "step": 18190
    },
    {
      "dynamics_loss_avg": 0.0530081132426858,
      "epoch": 1.734733581542969e-05,
      "step": 18190
    },
    {
      "action_loss_avg": 0.0025524013501126317,
      "epoch": 1.734733581542969e-05,
      "step": 18190
    },
    {
      "epoch": 1.734733581542969e-05,
      "step": 18190,
      "training_step_time": 8.596635103225708
    },
    {
      "epoch": 1.7348289489746093e-05,
      "model_forward_time": 1.8087186813354492,
      "step": 18191
    },
    {
      "epoch": 1.7348289489746093e-05,
      "step": 18191,
      "training_step_time": 8.812660694122314
    },
    {
      "epoch": 1.73492431640625e-05,
      "model_forward_time": 1.8090629577636719,
      "step": 18192
    },
    {
      "epoch": 1.73492431640625e-05,
      "step": 18192,
      "training_step_time": 8.720993995666504
    },
    {
      "epoch": 1.7350196838378908e-05,
      "model_forward_time": 1.8077623844146729,
      "step": 18193
    },
    {
      "epoch": 1.7350196838378908e-05,
      "step": 18193,
      "training_step_time": 8.543741226196289
    },
    {
      "epoch": 1.735115051269531e-05,
      "model_forward_time": 1.7954463958740234,
      "step": 18194
    },
    {
      "epoch": 1.735115051269531e-05,
      "step": 18194,
      "training_step_time": 7.8644325733184814
    },
    {
      "epoch": 1.735210418701172e-05,
      "model_forward_time": 2.3089122772216797,
      "step": 18195
    },
    {
      "epoch": 1.735210418701172e-05,
      "step": 18195,
      "training_step_time": 7.87240743637085
    },
    {
      "epoch": 1.7353057861328126e-05,
      "model_forward_time": 1.8078258037567139,
      "step": 18196
    },
    {
      "epoch": 1.7353057861328126e-05,
      "step": 18196,
      "training_step_time": 7.6173927783966064
    },
    {
      "epoch": 1.735401153564453e-05,
      "model_forward_time": 1.808410406112671,
      "step": 18197
    },
    {
      "epoch": 1.735401153564453e-05,
      "step": 18197,
      "training_step_time": 7.698162317276001
    },
    {
      "epoch": 1.7354965209960938e-05,
      "model_forward_time": 1.8072924613952637,
      "step": 18198
    },
    {
      "epoch": 1.7354965209960938e-05,
      "step": 18198,
      "training_step_time": 8.419151782989502
    },
    {
      "epoch": 1.7355918884277345e-05,
      "model_forward_time": 1.8082852363586426,
      "step": 18199
    },
    {
      "epoch": 1.7355918884277345e-05,
      "step": 18199,
      "training_step_time": 7.541210174560547
    },
    {
      "epoch": 1.735687255859375e-05,
      "grad_norm": 0.28129374980926514,
      "learning_rate": 9.531220547671689e-06,
      "loss": 0.0662,
      "step": 18200
    },
    {
      "epoch": 1.735687255859375e-05,
      "model_forward_time": 1.8169496059417725,
      "step": 18200
    },
    {
      "dynamics_loss_avg": 0.06334655284881592,
      "epoch": 1.735687255859375e-05,
      "step": 18200
    },
    {
      "action_loss_avg": 0.0030952234170399605,
      "epoch": 1.735687255859375e-05,
      "step": 18200
    },
    {
      "epoch": 1.735687255859375e-05,
      "step": 18200,
      "training_step_time": 8.418100118637085
    },
    {
      "epoch": 1.7357826232910157e-05,
      "model_forward_time": 1.8131656646728516,
      "step": 18201
    },
    {
      "epoch": 1.7357826232910157e-05,
      "step": 18201,
      "training_step_time": 8.617875337600708
    },
    {
      "epoch": 1.7358779907226564e-05,
      "model_forward_time": 1.7875542640686035,
      "step": 18202
    },
    {
      "epoch": 1.7358779907226564e-05,
      "step": 18202,
      "training_step_time": 7.758123874664307
    },
    {
      "epoch": 1.7359733581542968e-05,
      "model_forward_time": 1.8118550777435303,
      "step": 18203
    },
    {
      "epoch": 1.7359733581542968e-05,
      "step": 18203,
      "training_step_time": 8.78601622581482
    },
    {
      "epoch": 1.7360687255859376e-05,
      "model_forward_time": 2.344036102294922,
      "step": 18204
    },
    {
      "epoch": 1.7360687255859376e-05,
      "step": 18204,
      "training_step_time": 7.74715518951416
    },
    {
      "epoch": 1.736164093017578e-05,
      "model_forward_time": 1.8143339157104492,
      "step": 18205
    },
    {
      "epoch": 1.736164093017578e-05,
      "step": 18205,
      "training_step_time": 7.563577651977539
    },
    {
      "epoch": 1.7362594604492187e-05,
      "model_forward_time": 1.8103318214416504,
      "step": 18206
    },
    {
      "epoch": 1.7362594604492187e-05,
      "step": 18206,
      "training_step_time": 7.750911235809326
    },
    {
      "epoch": 1.7363548278808594e-05,
      "model_forward_time": 1.810699224472046,
      "step": 18207
    },
    {
      "epoch": 1.7363548278808594e-05,
      "step": 18207,
      "training_step_time": 7.673730373382568
    },
    {
      "epoch": 1.7364501953125e-05,
      "model_forward_time": 1.8108394145965576,
      "step": 18208
    },
    {
      "epoch": 1.7364501953125e-05,
      "step": 18208,
      "training_step_time": 7.547244071960449
    },
    {
      "epoch": 1.7365455627441406e-05,
      "model_forward_time": 1.818755865097046,
      "step": 18209
    },
    {
      "epoch": 1.7365455627441406e-05,
      "step": 18209,
      "training_step_time": 9.837570667266846
    },
    {
      "epoch": 1.7366409301757813e-05,
      "grad_norm": 0.2865695655345917,
      "learning_rate": 9.530521287632286e-06,
      "loss": 0.0645,
      "step": 18210
    },
    {
      "epoch": 1.7366409301757813e-05,
      "model_forward_time": 1.7983951568603516,
      "step": 18210
    },
    {
      "dynamics_loss_avg": 0.05264731030911207,
      "epoch": 1.7366409301757813e-05,
      "step": 18210
    },
    {
      "action_loss_avg": 0.0019360551319550722,
      "epoch": 1.7366409301757813e-05,
      "step": 18210
    },
    {
      "epoch": 1.7366409301757813e-05,
      "step": 18210,
      "training_step_time": 7.753636598587036
    },
    {
      "epoch": 1.7367362976074217e-05,
      "model_forward_time": 1.8040027618408203,
      "step": 18211
    },
    {
      "epoch": 1.7367362976074217e-05,
      "step": 18211,
      "training_step_time": 8.25367546081543
    },
    {
      "epoch": 1.7368316650390625e-05,
      "model_forward_time": 1.7974233627319336,
      "step": 18212
    },
    {
      "epoch": 1.7368316650390625e-05,
      "step": 18212,
      "training_step_time": 7.70458459854126
    },
    {
      "epoch": 1.7369270324707032e-05,
      "model_forward_time": 2.3273444175720215,
      "step": 18213
    },
    {
      "epoch": 1.7369270324707032e-05,
      "step": 18213,
      "training_step_time": 8.662703514099121
    },
    {
      "epoch": 1.7370223999023436e-05,
      "model_forward_time": 1.793463945388794,
      "step": 18214
    },
    {
      "epoch": 1.7370223999023436e-05,
      "step": 18214,
      "training_step_time": 7.563744306564331
    },
    {
      "epoch": 1.7371177673339844e-05,
      "model_forward_time": 1.8048725128173828,
      "step": 18215
    },
    {
      "epoch": 1.7371177673339844e-05,
      "step": 18215,
      "training_step_time": 7.673126935958862
    },
    {
      "epoch": 1.737213134765625e-05,
      "model_forward_time": 1.7988972663879395,
      "step": 18216
    },
    {
      "epoch": 1.737213134765625e-05,
      "step": 18216,
      "training_step_time": 8.325121641159058
    },
    {
      "epoch": 1.7373085021972655e-05,
      "model_forward_time": 1.8045802116394043,
      "step": 18217
    },
    {
      "epoch": 1.7373085021972655e-05,
      "step": 18217,
      "training_step_time": 8.324993371963501
    },
    {
      "epoch": 1.7374038696289062e-05,
      "model_forward_time": 1.7982330322265625,
      "step": 18218
    },
    {
      "epoch": 1.7374038696289062e-05,
      "step": 18218,
      "training_step_time": 9.555686473846436
    },
    {
      "epoch": 1.737499237060547e-05,
      "model_forward_time": 1.8039720058441162,
      "step": 18219
    },
    {
      "epoch": 1.737499237060547e-05,
      "step": 18219,
      "training_step_time": 7.739032745361328
    },
    {
      "epoch": 1.7375946044921874e-05,
      "grad_norm": 0.20960822701454163,
      "learning_rate": 9.529821532141884e-06,
      "loss": 0.0637,
      "step": 18220
    },
    {
      "epoch": 1.7375946044921874e-05,
      "model_forward_time": 1.7979857921600342,
      "step": 18220
    },
    {
      "dynamics_loss_avg": 0.06566872000694275,
      "epoch": 1.7375946044921874e-05,
      "step": 18220
    },
    {
      "action_loss_avg": 0.0033141590771265327,
      "epoch": 1.7375946044921874e-05,
      "step": 18220
    },
    {
      "epoch": 1.7375946044921874e-05,
      "step": 18220,
      "training_step_time": 8.405727624893188
    },
    {
      "epoch": 1.737689971923828e-05,
      "model_forward_time": 1.8026504516601562,
      "step": 18221
    },
    {
      "epoch": 1.737689971923828e-05,
      "step": 18221,
      "training_step_time": 7.721220970153809
    },
    {
      "epoch": 1.737785339355469e-05,
      "model_forward_time": 1.8006024360656738,
      "step": 18222
    },
    {
      "epoch": 1.737785339355469e-05,
      "step": 18222,
      "training_step_time": 9.243455648422241
    },
    {
      "epoch": 1.7378807067871093e-05,
      "model_forward_time": 2.3150246143341064,
      "step": 18223
    },
    {
      "epoch": 1.7378807067871093e-05,
      "step": 18223,
      "training_step_time": 7.557476997375488
    },
    {
      "epoch": 1.73797607421875e-05,
      "model_forward_time": 1.795987844467163,
      "step": 18224
    },
    {
      "epoch": 1.73797607421875e-05,
      "step": 18224,
      "training_step_time": 7.603567123413086
    },
    {
      "epoch": 1.7380714416503908e-05,
      "model_forward_time": 1.8073077201843262,
      "step": 18225
    },
    {
      "epoch": 1.7380714416503908e-05,
      "step": 18225,
      "training_step_time": 7.653814077377319
    },
    {
      "epoch": 1.738166809082031e-05,
      "model_forward_time": 1.7978878021240234,
      "step": 18226
    },
    {
      "epoch": 1.738166809082031e-05,
      "step": 18226,
      "training_step_time": 7.77475118637085
    },
    {
      "epoch": 1.738262176513672e-05,
      "model_forward_time": 1.7994530200958252,
      "step": 18227
    },
    {
      "epoch": 1.738262176513672e-05,
      "step": 18227,
      "training_step_time": 8.625560998916626
    },
    {
      "epoch": 1.7383575439453126e-05,
      "model_forward_time": 1.4297072887420654,
      "step": 18228
    },
    {
      "epoch": 1.7383575439453126e-05,
      "step": 18228,
      "training_step_time": 10.480459690093994
    },
    {
      "epoch": 1.738452911376953e-05,
      "model_forward_time": 1.7945280075073242,
      "step": 18229
    },
    {
      "epoch": 1.738452911376953e-05,
      "step": 18229,
      "training_step_time": 7.757248640060425
    },
    {
      "epoch": 1.7385482788085938e-05,
      "grad_norm": 0.2099003940820694,
      "learning_rate": 9.52912128127701e-06,
      "loss": 0.062,
      "step": 18230
    },
    {
      "epoch": 1.7385482788085938e-05,
      "model_forward_time": 1.7892310619354248,
      "step": 18230
    },
    {
      "dynamics_loss_avg": 0.06021132189780474,
      "epoch": 1.7385482788085938e-05,
      "step": 18230
    },
    {
      "action_loss_avg": 0.002750228485092521,
      "epoch": 1.7385482788085938e-05,
      "step": 18230
    },
    {
      "epoch": 1.7385482788085938e-05,
      "step": 18230,
      "training_step_time": 8.055800437927246
    },
    {
      "epoch": 1.7386436462402345e-05,
      "model_forward_time": 1.8067166805267334,
      "step": 18231
    },
    {
      "epoch": 1.7386436462402345e-05,
      "step": 18231,
      "training_step_time": 7.65527081489563
    },
    {
      "epoch": 1.738739013671875e-05,
      "model_forward_time": 2.304921865463257,
      "step": 18232
    },
    {
      "epoch": 1.738739013671875e-05,
      "step": 18232,
      "training_step_time": 7.748347759246826
    },
    {
      "epoch": 1.7388343811035157e-05,
      "model_forward_time": 1.796668291091919,
      "step": 18233
    },
    {
      "epoch": 1.7388343811035157e-05,
      "step": 18233,
      "training_step_time": 7.619342565536499
    },
    {
      "epoch": 1.7389297485351564e-05,
      "model_forward_time": 1.8007316589355469,
      "step": 18234
    },
    {
      "epoch": 1.7389297485351564e-05,
      "step": 18234,
      "training_step_time": 8.11617374420166
    },
    {
      "epoch": 1.7390251159667968e-05,
      "model_forward_time": 1.8006658554077148,
      "step": 18235
    },
    {
      "epoch": 1.7390251159667968e-05,
      "step": 18235,
      "training_step_time": 7.6375412940979
    },
    {
      "epoch": 1.7391204833984376e-05,
      "model_forward_time": 1.7991929054260254,
      "step": 18236
    },
    {
      "epoch": 1.7391204833984376e-05,
      "step": 18236,
      "training_step_time": 9.712640523910522
    },
    {
      "epoch": 1.739215850830078e-05,
      "model_forward_time": 1.8059062957763672,
      "step": 18237
    },
    {
      "epoch": 1.739215850830078e-05,
      "step": 18237,
      "training_step_time": 9.357301473617554
    },
    {
      "epoch": 1.7393112182617187e-05,
      "model_forward_time": 1.7979297637939453,
      "step": 18238
    },
    {
      "epoch": 1.7393112182617187e-05,
      "step": 18238,
      "training_step_time": 7.6613709926605225
    },
    {
      "epoch": 1.7394065856933594e-05,
      "model_forward_time": 1.8021860122680664,
      "step": 18239
    },
    {
      "epoch": 1.7394065856933594e-05,
      "step": 18239,
      "training_step_time": 7.621696472167969
    },
    {
      "epoch": 1.739501953125e-05,
      "grad_norm": 0.17749135196208954,
      "learning_rate": 9.528420535114245e-06,
      "loss": 0.0616,
      "step": 18240
    },
    {
      "epoch": 1.739501953125e-05,
      "model_forward_time": 1.799776315689087,
      "step": 18240
    },
    {
      "dynamics_loss_avg": 0.07021316476166248,
      "epoch": 1.739501953125e-05,
      "step": 18240
    },
    {
      "action_loss_avg": 0.0035955726343672723,
      "epoch": 1.739501953125e-05,
      "step": 18240
    },
    {
      "epoch": 1.739501953125e-05,
      "step": 18240,
      "training_step_time": 9.49809980392456
    },
    {
      "epoch": 1.7395973205566406e-05,
      "model_forward_time": 2.332815647125244,
      "step": 18241
    },
    {
      "epoch": 1.7395973205566406e-05,
      "step": 18241,
      "training_step_time": 7.711259603500366
    },
    {
      "epoch": 1.7396926879882813e-05,
      "model_forward_time": 1.8001608848571777,
      "step": 18242
    },
    {
      "epoch": 1.7396926879882813e-05,
      "step": 18242,
      "training_step_time": 7.65189003944397
    },
    {
      "epoch": 1.7397880554199217e-05,
      "model_forward_time": 1.8019981384277344,
      "step": 18243
    },
    {
      "epoch": 1.7397880554199217e-05,
      "step": 18243,
      "training_step_time": 7.596682786941528
    },
    {
      "epoch": 1.7398834228515625e-05,
      "model_forward_time": 1.793062448501587,
      "step": 18244
    },
    {
      "epoch": 1.7398834228515625e-05,
      "step": 18244,
      "training_step_time": 7.650027275085449
    },
    {
      "epoch": 1.7399787902832032e-05,
      "model_forward_time": 1.7954902648925781,
      "step": 18245
    },
    {
      "epoch": 1.7399787902832032e-05,
      "step": 18245,
      "training_step_time": 8.342259883880615
    },
    {
      "epoch": 1.7400741577148436e-05,
      "model_forward_time": 1.7966713905334473,
      "step": 18246
    },
    {
      "epoch": 1.7400741577148436e-05,
      "step": 18246,
      "training_step_time": 8.867330312728882
    },
    {
      "epoch": 1.7401695251464844e-05,
      "model_forward_time": 1.8123600482940674,
      "step": 18247
    },
    {
      "epoch": 1.7401695251464844e-05,
      "step": 18247,
      "training_step_time": 8.699606895446777
    },
    {
      "epoch": 1.740264892578125e-05,
      "model_forward_time": 1.80357027053833,
      "step": 18248
    },
    {
      "epoch": 1.740264892578125e-05,
      "step": 18248,
      "training_step_time": 8.654473543167114
    },
    {
      "epoch": 1.7403602600097655e-05,
      "model_forward_time": 1.8054594993591309,
      "step": 18249
    },
    {
      "epoch": 1.7403602600097655e-05,
      "step": 18249,
      "training_step_time": 7.693936109542847
    },
    {
      "epoch": 1.7404556274414062e-05,
      "grad_norm": 0.16574551165103912,
      "learning_rate": 9.527719293730215e-06,
      "loss": 0.0673,
      "step": 18250
    },
    {
      "epoch": 1.7404556274414062e-05,
      "model_forward_time": 2.3045432567596436,
      "step": 18250
    },
    {
      "dynamics_loss_avg": 0.04711284432560205,
      "epoch": 1.7404556274414062e-05,
      "step": 18250
    },
    {
      "action_loss_avg": 0.002526167460018769,
      "epoch": 1.7404556274414062e-05,
      "step": 18250
    },
    {
      "epoch": 1.7404556274414062e-05,
      "step": 18250,
      "training_step_time": 7.758068561553955
    },
    {
      "epoch": 1.740550994873047e-05,
      "model_forward_time": 1.7983453273773193,
      "step": 18251
    },
    {
      "epoch": 1.740550994873047e-05,
      "step": 18251,
      "training_step_time": 7.626789331436157
    },
    {
      "epoch": 1.7406463623046874e-05,
      "model_forward_time": 1.7978956699371338,
      "step": 18252
    },
    {
      "epoch": 1.7406463623046874e-05,
      "step": 18252,
      "training_step_time": 8.237685203552246
    },
    {
      "epoch": 1.740741729736328e-05,
      "model_forward_time": 1.802711009979248,
      "step": 18253
    },
    {
      "epoch": 1.740741729736328e-05,
      "step": 18253,
      "training_step_time": 8.533318519592285
    },
    {
      "epoch": 1.740837097167969e-05,
      "model_forward_time": 1.7956743240356445,
      "step": 18254
    },
    {
      "epoch": 1.740837097167969e-05,
      "step": 18254,
      "training_step_time": 7.804218530654907
    },
    {
      "epoch": 1.7409324645996093e-05,
      "model_forward_time": 1.7945797443389893,
      "step": 18255
    },
    {
      "epoch": 1.7409324645996093e-05,
      "step": 18255,
      "training_step_time": 8.284388780593872
    },
    {
      "epoch": 1.74102783203125e-05,
      "model_forward_time": 1.8018560409545898,
      "step": 18256
    },
    {
      "epoch": 1.74102783203125e-05,
      "step": 18256,
      "training_step_time": 7.847036838531494
    },
    {
      "epoch": 1.7411231994628908e-05,
      "model_forward_time": 1.7870190143585205,
      "step": 18257
    },
    {
      "epoch": 1.7411231994628908e-05,
      "step": 18257,
      "training_step_time": 7.6818687915802
    },
    {
      "epoch": 1.741218566894531e-05,
      "model_forward_time": 1.803865671157837,
      "step": 18258
    },
    {
      "epoch": 1.741218566894531e-05,
      "step": 18258,
      "training_step_time": 7.72710394859314
    },
    {
      "epoch": 1.741313934326172e-05,
      "model_forward_time": 1.804048776626587,
      "step": 18259
    },
    {
      "epoch": 1.741313934326172e-05,
      "step": 18259,
      "training_step_time": 8.298364877700806
    },
    {
      "epoch": 1.7414093017578126e-05,
      "grad_norm": 0.2549363970756531,
      "learning_rate": 9.527017557201612e-06,
      "loss": 0.0676,
      "step": 18260
    },
    {
      "epoch": 1.7414093017578126e-05,
      "model_forward_time": 1.8011348247528076,
      "step": 18260
    },
    {
      "dynamics_loss_avg": 0.05281651113182306,
      "epoch": 1.7414093017578126e-05,
      "step": 18260
    },
    {
      "action_loss_avg": 0.0024859341443516313,
      "epoch": 1.7414093017578126e-05,
      "step": 18260
    },
    {
      "epoch": 1.7414093017578126e-05,
      "step": 18260,
      "training_step_time": 9.070229768753052
    },
    {
      "epoch": 1.741504669189453e-05,
      "model_forward_time": 1.7959997653961182,
      "step": 18261
    },
    {
      "epoch": 1.741504669189453e-05,
      "step": 18261,
      "training_step_time": 7.5224997997283936
    },
    {
      "epoch": 1.7416000366210938e-05,
      "model_forward_time": 1.801671028137207,
      "step": 18262
    },
    {
      "epoch": 1.7416000366210938e-05,
      "step": 18262,
      "training_step_time": 8.896703481674194
    },
    {
      "epoch": 1.7416954040527345e-05,
      "model_forward_time": 1.801361322402954,
      "step": 18263
    },
    {
      "epoch": 1.7416954040527345e-05,
      "step": 18263,
      "training_step_time": 8.709381818771362
    },
    {
      "epoch": 1.741790771484375e-05,
      "model_forward_time": 1.8043370246887207,
      "step": 18264
    },
    {
      "epoch": 1.741790771484375e-05,
      "step": 18264,
      "training_step_time": 8.468930959701538
    },
    {
      "epoch": 1.7418861389160157e-05,
      "model_forward_time": 1.809279441833496,
      "step": 18265
    },
    {
      "epoch": 1.7418861389160157e-05,
      "step": 18265,
      "training_step_time": 9.372331857681274
    },
    {
      "epoch": 1.741981506347656e-05,
      "model_forward_time": 1.8023617267608643,
      "step": 18266
    },
    {
      "epoch": 1.741981506347656e-05,
      "step": 18266,
      "training_step_time": 8.424654245376587
    },
    {
      "epoch": 1.7420768737792968e-05,
      "model_forward_time": 1.8143401145935059,
      "step": 18267
    },
    {
      "epoch": 1.7420768737792968e-05,
      "step": 18267,
      "training_step_time": 7.637782335281372
    },
    {
      "epoch": 1.7421722412109376e-05,
      "model_forward_time": 1.7990834712982178,
      "step": 18268
    },
    {
      "epoch": 1.7421722412109376e-05,
      "step": 18268,
      "training_step_time": 7.784757852554321
    },
    {
      "epoch": 1.742267608642578e-05,
      "model_forward_time": 2.3232903480529785,
      "step": 18269
    },
    {
      "epoch": 1.742267608642578e-05,
      "step": 18269,
      "training_step_time": 7.637004852294922
    },
    {
      "epoch": 1.7423629760742187e-05,
      "grad_norm": 0.1679934412240982,
      "learning_rate": 9.526315325605176e-06,
      "loss": 0.0685,
      "step": 18270
    },
    {
      "epoch": 1.7423629760742187e-05,
      "model_forward_time": 1.8050975799560547,
      "step": 18270
    },
    {
      "dynamics_loss_avg": 0.05218036435544491,
      "epoch": 1.7423629760742187e-05,
      "step": 18270
    },
    {
      "action_loss_avg": 0.002548807713901624,
      "epoch": 1.7423629760742187e-05,
      "step": 18270
    },
    {
      "epoch": 1.7423629760742187e-05,
      "step": 18270,
      "training_step_time": 7.539350986480713
    },
    {
      "epoch": 1.7424583435058595e-05,
      "model_forward_time": 1.800079107284546,
      "step": 18271
    },
    {
      "epoch": 1.7424583435058595e-05,
      "step": 18271,
      "training_step_time": 7.574481248855591
    },
    {
      "epoch": 1.7425537109375e-05,
      "model_forward_time": 1.42698335647583,
      "step": 18272
    },
    {
      "epoch": 1.7425537109375e-05,
      "step": 18272,
      "training_step_time": 8.534592866897583
    },
    {
      "epoch": 1.7426490783691406e-05,
      "model_forward_time": 1.4182565212249756,
      "step": 18273
    },
    {
      "epoch": 1.7426490783691406e-05,
      "step": 18273,
      "training_step_time": 8.320385217666626
    },
    {
      "epoch": 1.7427444458007813e-05,
      "model_forward_time": 1.794959306716919,
      "step": 18274
    },
    {
      "epoch": 1.7427444458007813e-05,
      "step": 18274,
      "training_step_time": 8.26214599609375
    },
    {
      "epoch": 1.7428398132324217e-05,
      "model_forward_time": 1.804720401763916,
      "step": 18275
    },
    {
      "epoch": 1.7428398132324217e-05,
      "step": 18275,
      "training_step_time": 9.462464570999146
    },
    {
      "epoch": 1.7429351806640625e-05,
      "model_forward_time": 1.803213357925415,
      "step": 18276
    },
    {
      "epoch": 1.7429351806640625e-05,
      "step": 18276,
      "training_step_time": 8.238518714904785
    },
    {
      "epoch": 1.7430305480957032e-05,
      "model_forward_time": 1.7860221862792969,
      "step": 18277
    },
    {
      "epoch": 1.7430305480957032e-05,
      "step": 18277,
      "training_step_time": 7.651630163192749
    },
    {
      "epoch": 1.7431259155273436e-05,
      "model_forward_time": 2.299290895462036,
      "step": 18278
    },
    {
      "epoch": 1.7431259155273436e-05,
      "step": 18278,
      "training_step_time": 7.645554304122925
    },
    {
      "epoch": 1.7432212829589844e-05,
      "model_forward_time": 1.4291443824768066,
      "step": 18279
    },
    {
      "epoch": 1.7432212829589844e-05,
      "step": 18279,
      "training_step_time": 7.550443172454834
    },
    {
      "epoch": 1.743316650390625e-05,
      "grad_norm": 0.11045342683792114,
      "learning_rate": 9.525612599017699e-06,
      "loss": 0.0667,
      "step": 18280
    },
    {
      "epoch": 1.743316650390625e-05,
      "model_forward_time": 1.7866089344024658,
      "step": 18280
    },
    {
      "dynamics_loss_avg": 0.05267328396439552,
      "epoch": 1.743316650390625e-05,
      "step": 18280
    },
    {
      "action_loss_avg": 0.0021699825243558734,
      "epoch": 1.743316650390625e-05,
      "step": 18280
    },
    {
      "epoch": 1.743316650390625e-05,
      "step": 18280,
      "training_step_time": 7.482540130615234
    },
    {
      "epoch": 1.7434120178222655e-05,
      "model_forward_time": 1.7945277690887451,
      "step": 18281
    },
    {
      "epoch": 1.7434120178222655e-05,
      "step": 18281,
      "training_step_time": 8.587889432907104
    },
    {
      "epoch": 1.7435073852539063e-05,
      "model_forward_time": 1.7995615005493164,
      "step": 18282
    },
    {
      "epoch": 1.7435073852539063e-05,
      "step": 18282,
      "training_step_time": 8.227582693099976
    },
    {
      "epoch": 1.743602752685547e-05,
      "model_forward_time": 1.7964351177215576,
      "step": 18283
    },
    {
      "epoch": 1.743602752685547e-05,
      "step": 18283,
      "training_step_time": 8.684378862380981
    },
    {
      "epoch": 1.7436981201171874e-05,
      "model_forward_time": 1.802225112915039,
      "step": 18284
    },
    {
      "epoch": 1.7436981201171874e-05,
      "step": 18284,
      "training_step_time": 9.225623607635498
    },
    {
      "epoch": 1.743793487548828e-05,
      "model_forward_time": 1.801011562347412,
      "step": 18285
    },
    {
      "epoch": 1.743793487548828e-05,
      "step": 18285,
      "training_step_time": 8.09146499633789
    },
    {
      "epoch": 1.743888854980469e-05,
      "model_forward_time": 1.8001232147216797,
      "step": 18286
    },
    {
      "epoch": 1.743888854980469e-05,
      "step": 18286,
      "training_step_time": 8.09011197090149
    },
    {
      "epoch": 1.7439842224121093e-05,
      "model_forward_time": 2.3172645568847656,
      "step": 18287
    },
    {
      "epoch": 1.7439842224121093e-05,
      "step": 18287,
      "training_step_time": 8.203393697738647
    },
    {
      "epoch": 1.74407958984375e-05,
      "model_forward_time": 1.7951173782348633,
      "step": 18288
    },
    {
      "epoch": 1.74407958984375e-05,
      "step": 18288,
      "training_step_time": 8.678703546524048
    },
    {
      "epoch": 1.7441749572753908e-05,
      "model_forward_time": 1.7949986457824707,
      "step": 18289
    },
    {
      "epoch": 1.7441749572753908e-05,
      "step": 18289,
      "training_step_time": 7.473272085189819
    },
    {
      "epoch": 1.744270324707031e-05,
      "grad_norm": 0.18460214138031006,
      "learning_rate": 9.524909377516034e-06,
      "loss": 0.0618,
      "step": 18290
    },
    {
      "epoch": 1.744270324707031e-05,
      "model_forward_time": 1.7954812049865723,
      "step": 18290
    },
    {
      "dynamics_loss_avg": 0.06975427381694317,
      "epoch": 1.744270324707031e-05,
      "step": 18290
    },
    {
      "action_loss_avg": 0.0028516574820969254,
      "epoch": 1.744270324707031e-05,
      "step": 18290
    },
    {
      "epoch": 1.744270324707031e-05,
      "step": 18290,
      "training_step_time": 8.628477096557617
    },
    {
      "epoch": 1.744365692138672e-05,
      "model_forward_time": 1.8029866218566895,
      "step": 18291
    },
    {
      "epoch": 1.744365692138672e-05,
      "step": 18291,
      "training_step_time": 9.573403596878052
    },
    {
      "epoch": 1.7444610595703127e-05,
      "model_forward_time": 1.7996611595153809,
      "step": 18292
    },
    {
      "epoch": 1.7444610595703127e-05,
      "step": 18292,
      "training_step_time": 7.684280633926392
    },
    {
      "epoch": 1.744556427001953e-05,
      "model_forward_time": 1.8072326183319092,
      "step": 18293
    },
    {
      "epoch": 1.744556427001953e-05,
      "step": 18293,
      "training_step_time": 7.676117658615112
    },
    {
      "epoch": 1.7446517944335938e-05,
      "model_forward_time": 1.7972474098205566,
      "step": 18294
    },
    {
      "epoch": 1.7446517944335938e-05,
      "step": 18294,
      "training_step_time": 8.715336561203003
    },
    {
      "epoch": 1.7447471618652345e-05,
      "model_forward_time": 1.8044641017913818,
      "step": 18295
    },
    {
      "epoch": 1.7447471618652345e-05,
      "step": 18295,
      "training_step_time": 7.518631935119629
    },
    {
      "epoch": 1.744842529296875e-05,
      "model_forward_time": 1.793104887008667,
      "step": 18296
    },
    {
      "epoch": 1.744842529296875e-05,
      "step": 18296,
      "training_step_time": 7.664251089096069
    },
    {
      "epoch": 1.7449378967285157e-05,
      "model_forward_time": 2.3092870712280273,
      "step": 18297
    },
    {
      "epoch": 1.7449378967285157e-05,
      "step": 18297,
      "training_step_time": 7.589945554733276
    },
    {
      "epoch": 1.745033264160156e-05,
      "model_forward_time": 1.791372537612915,
      "step": 18298
    },
    {
      "epoch": 1.745033264160156e-05,
      "step": 18298,
      "training_step_time": 7.539676666259766
    },
    {
      "epoch": 1.7451286315917968e-05,
      "model_forward_time": 1.7978312969207764,
      "step": 18299
    },
    {
      "epoch": 1.7451286315917968e-05,
      "step": 18299,
      "training_step_time": 7.6075263023376465
    },
    {
      "epoch": 1.7452239990234376e-05,
      "grad_norm": 0.19070321321487427,
      "learning_rate": 9.524205661177081e-06,
      "loss": 0.0646,
      "step": 18300
    },
    {
      "epoch": 1.7452239990234376e-05,
      "model_forward_time": 1.7962703704833984,
      "step": 18300
    },
    {
      "dynamics_loss_avg": 0.055978947505354884,
      "epoch": 1.7452239990234376e-05,
      "step": 18300
    },
    {
      "action_loss_avg": 0.0019730376137886197,
      "epoch": 1.7452239990234376e-05,
      "step": 18300
    },
    {
      "epoch": 1.7452239990234376e-05,
      "step": 18300,
      "training_step_time": 8.787694931030273
    },
    {
      "epoch": 1.745319366455078e-05,
      "model_forward_time": 1.8017809391021729,
      "step": 18301
    },
    {
      "epoch": 1.745319366455078e-05,
      "step": 18301,
      "training_step_time": 9.184513807296753
    },
    {
      "epoch": 1.7454147338867187e-05,
      "model_forward_time": 1.8018293380737305,
      "step": 18302
    },
    {
      "epoch": 1.7454147338867187e-05,
      "step": 18302,
      "training_step_time": 8.437265396118164
    },
    {
      "epoch": 1.7455101013183595e-05,
      "model_forward_time": 1.800001859664917,
      "step": 18303
    },
    {
      "epoch": 1.7455101013183595e-05,
      "step": 18303,
      "training_step_time": 8.643790006637573
    },
    {
      "epoch": 1.74560546875e-05,
      "model_forward_time": 1.7976915836334229,
      "step": 18304
    },
    {
      "epoch": 1.74560546875e-05,
      "step": 18304,
      "training_step_time": 7.6896843910217285
    },
    {
      "epoch": 1.7457008361816406e-05,
      "model_forward_time": 1.417534351348877,
      "step": 18305
    },
    {
      "epoch": 1.7457008361816406e-05,
      "step": 18305,
      "training_step_time": 7.708552122116089
    },
    {
      "epoch": 1.7457962036132813e-05,
      "model_forward_time": 2.3164687156677246,
      "step": 18306
    },
    {
      "epoch": 1.7457962036132813e-05,
      "step": 18306,
      "training_step_time": 7.772454500198364
    },
    {
      "epoch": 1.7458915710449217e-05,
      "model_forward_time": 1.8021275997161865,
      "step": 18307
    },
    {
      "epoch": 1.7458915710449217e-05,
      "step": 18307,
      "training_step_time": 7.589679002761841
    },
    {
      "epoch": 1.7459869384765625e-05,
      "model_forward_time": 1.8017635345458984,
      "step": 18308
    },
    {
      "epoch": 1.7459869384765625e-05,
      "step": 18308,
      "training_step_time": 7.60499382019043
    },
    {
      "epoch": 1.7460823059082032e-05,
      "model_forward_time": 1.7978856563568115,
      "step": 18309
    },
    {
      "epoch": 1.7460823059082032e-05,
      "step": 18309,
      "training_step_time": 7.728469371795654
    },
    {
      "epoch": 1.7461776733398436e-05,
      "grad_norm": 0.2697910964488983,
      "learning_rate": 9.523501450077801e-06,
      "loss": 0.0624,
      "step": 18310
    },
    {
      "epoch": 1.7461776733398436e-05,
      "model_forward_time": 1.7961218357086182,
      "step": 18310
    },
    {
      "dynamics_loss_avg": 0.05792289283126593,
      "epoch": 1.7461776733398436e-05,
      "step": 18310
    },
    {
      "action_loss_avg": 0.00291623508092016,
      "epoch": 1.7461776733398436e-05,
      "step": 18310
    },
    {
      "epoch": 1.7461776733398436e-05,
      "step": 18310,
      "training_step_time": 7.680558919906616
    },
    {
      "epoch": 1.7462730407714844e-05,
      "model_forward_time": 1.7958979606628418,
      "step": 18311
    },
    {
      "epoch": 1.7462730407714844e-05,
      "step": 18311,
      "training_step_time": 8.679301738739014
    },
    {
      "epoch": 1.746368408203125e-05,
      "model_forward_time": 1.7950663566589355,
      "step": 18312
    },
    {
      "epoch": 1.746368408203125e-05,
      "step": 18312,
      "training_step_time": 7.741898775100708
    },
    {
      "epoch": 1.7464637756347655e-05,
      "model_forward_time": 1.8045985698699951,
      "step": 18313
    },
    {
      "epoch": 1.7464637756347655e-05,
      "step": 18313,
      "training_step_time": 7.8588011264801025
    },
    {
      "epoch": 1.7465591430664063e-05,
      "model_forward_time": 1.8035972118377686,
      "step": 18314
    },
    {
      "epoch": 1.7465591430664063e-05,
      "step": 18314,
      "training_step_time": 8.726294040679932
    },
    {
      "epoch": 1.746654510498047e-05,
      "model_forward_time": 1.7960598468780518,
      "step": 18315
    },
    {
      "epoch": 1.746654510498047e-05,
      "step": 18315,
      "training_step_time": 6.821092367172241
    },
    {
      "epoch": 1.7467498779296874e-05,
      "model_forward_time": 1.795029640197754,
      "step": 18316
    },
    {
      "epoch": 1.7467498779296874e-05,
      "step": 18316,
      "training_step_time": 6.854912519454956
    },
    {
      "epoch": 1.746845245361328e-05,
      "model_forward_time": 1.804051399230957,
      "step": 18317
    },
    {
      "epoch": 1.746845245361328e-05,
      "step": 18317,
      "training_step_time": 6.8252599239349365
    },
    {
      "epoch": 1.746940612792969e-05,
      "model_forward_time": 1.8062317371368408,
      "step": 18318
    },
    {
      "epoch": 1.746940612792969e-05,
      "step": 18318,
      "training_step_time": 6.874047756195068
    },
    {
      "epoch": 1.7470359802246093e-05,
      "model_forward_time": 1.8055353164672852,
      "step": 18319
    },
    {
      "epoch": 1.7470359802246093e-05,
      "step": 18319,
      "training_step_time": 6.812554359436035
    },
    {
      "epoch": 1.74713134765625e-05,
      "grad_norm": 0.2836366891860962,
      "learning_rate": 9.522796744295202e-06,
      "loss": 0.0654,
      "step": 18320
    },
    {
      "epoch": 1.74713134765625e-05,
      "model_forward_time": 1.0799856185913086,
      "step": 18320
    },
    {
      "dynamics_loss_avg": 0.07434389293193817,
      "epoch": 1.74713134765625e-05,
      "step": 18320
    },
    {
      "action_loss_avg": 0.0038909658323973417,
      "epoch": 1.74713134765625e-05,
      "step": 18320
    },
    {
      "epoch": 1.74713134765625e-05,
      "step": 18320,
      "training_step_time": 6.823761701583862
    },
    {
      "epoch": 1.7472267150878908e-05,
      "model_forward_time": 1.77781343460083,
      "step": 18321
    },
    {
      "epoch": 1.7472267150878908e-05,
      "step": 18321,
      "training_step_time": 6.876911640167236
    },
    {
      "epoch": 1.7473220825195312e-05,
      "model_forward_time": 1.8051426410675049,
      "step": 18322
    },
    {
      "epoch": 1.7473220825195312e-05,
      "step": 18322,
      "training_step_time": 6.820619821548462
    },
    {
      "epoch": 1.747417449951172e-05,
      "model_forward_time": 2.338287830352783,
      "step": 18323
    },
    {
      "epoch": 1.747417449951172e-05,
      "step": 18323,
      "training_step_time": 8.405903339385986
    },
    {
      "epoch": 1.7475128173828127e-05,
      "model_forward_time": 1.7935822010040283,
      "step": 18324
    },
    {
      "epoch": 1.7475128173828127e-05,
      "step": 18324,
      "training_step_time": 7.777911424636841
    },
    {
      "epoch": 1.747608184814453e-05,
      "model_forward_time": 1.8119308948516846,
      "step": 18325
    },
    {
      "epoch": 1.747608184814453e-05,
      "step": 18325,
      "training_step_time": 6.8506152629852295
    },
    {
      "epoch": 1.7477035522460938e-05,
      "model_forward_time": 1.8171567916870117,
      "step": 18326
    },
    {
      "epoch": 1.7477035522460938e-05,
      "step": 18326,
      "training_step_time": 6.795103549957275
    },
    {
      "epoch": 1.7477989196777345e-05,
      "model_forward_time": 1.8038151264190674,
      "step": 18327
    },
    {
      "epoch": 1.7477989196777345e-05,
      "step": 18327,
      "training_step_time": 6.742617607116699
    },
    {
      "epoch": 1.747894287109375e-05,
      "model_forward_time": 1.816061019897461,
      "step": 18328
    },
    {
      "epoch": 1.747894287109375e-05,
      "step": 18328,
      "training_step_time": 6.755702018737793
    },
    {
      "epoch": 1.7479896545410157e-05,
      "model_forward_time": 1.8150627613067627,
      "step": 18329
    },
    {
      "epoch": 1.7479896545410157e-05,
      "step": 18329,
      "training_step_time": 6.806847333908081
    },
    {
      "epoch": 1.748085021972656e-05,
      "grad_norm": 0.16918475925922394,
      "learning_rate": 9.522091543906353e-06,
      "loss": 0.0633,
      "step": 18330
    },
    {
      "epoch": 1.748085021972656e-05,
      "model_forward_time": 1.8149135112762451,
      "step": 18330
    },
    {
      "dynamics_loss_avg": 0.060923285968601706,
      "epoch": 1.748085021972656e-05,
      "step": 18330
    },
    {
      "action_loss_avg": 0.0021737017726991326,
      "epoch": 1.748085021972656e-05,
      "step": 18330
    },
    {
      "epoch": 1.748085021972656e-05,
      "step": 18330,
      "training_step_time": 6.736967325210571
    },
    {
      "epoch": 1.7481803894042968e-05,
      "model_forward_time": 1.8207063674926758,
      "step": 18331
    },
    {
      "epoch": 1.7481803894042968e-05,
      "step": 18331,
      "training_step_time": 6.855618000030518
    },
    {
      "epoch": 1.7482757568359376e-05,
      "model_forward_time": 2.3704137802124023,
      "step": 18332
    },
    {
      "epoch": 1.7482757568359376e-05,
      "step": 18332,
      "training_step_time": 8.668184995651245
    },
    {
      "epoch": 1.748371124267578e-05,
      "model_forward_time": 1.8185195922851562,
      "step": 18333
    },
    {
      "epoch": 1.748371124267578e-05,
      "step": 18333,
      "training_step_time": 9.214394092559814
    },
    {
      "epoch": 1.7484664916992187e-05,
      "model_forward_time": 1.8151168823242188,
      "step": 18334
    },
    {
      "epoch": 1.7484664916992187e-05,
      "step": 18334,
      "training_step_time": 6.804492235183716
    },
    {
      "epoch": 1.7485618591308595e-05,
      "model_forward_time": 1.805666446685791,
      "step": 18335
    },
    {
      "epoch": 1.7485618591308595e-05,
      "step": 18335,
      "training_step_time": 6.812643766403198
    },
    {
      "epoch": 1.7486572265625e-05,
      "model_forward_time": 1.8125994205474854,
      "step": 18336
    },
    {
      "epoch": 1.7486572265625e-05,
      "step": 18336,
      "training_step_time": 6.870702743530273
    },
    {
      "epoch": 1.7487525939941406e-05,
      "model_forward_time": 1.8215465545654297,
      "step": 18337
    },
    {
      "epoch": 1.7487525939941406e-05,
      "step": 18337,
      "training_step_time": 6.818319320678711
    },
    {
      "epoch": 1.7488479614257813e-05,
      "model_forward_time": 1.8132643699645996,
      "step": 18338
    },
    {
      "epoch": 1.7488479614257813e-05,
      "step": 18338,
      "training_step_time": 6.790724754333496
    },
    {
      "epoch": 1.7489433288574217e-05,
      "model_forward_time": 1.8189411163330078,
      "step": 18339
    },
    {
      "epoch": 1.7489433288574217e-05,
      "step": 18339,
      "training_step_time": 6.846457004547119
    },
    {
      "epoch": 1.7490386962890625e-05,
      "grad_norm": 0.3708915412425995,
      "learning_rate": 9.52138584898837e-06,
      "loss": 0.0644,
      "step": 18340
    },
    {
      "epoch": 1.7490386962890625e-05,
      "model_forward_time": 1.8091771602630615,
      "step": 18340
    },
    {
      "dynamics_loss_avg": 0.052977198362350465,
      "epoch": 1.7490386962890625e-05,
      "step": 18340
    },
    {
      "action_loss_avg": 0.0023499094997532664,
      "epoch": 1.7490386962890625e-05,
      "step": 18340
    },
    {
      "epoch": 1.7490386962890625e-05,
      "step": 18340,
      "training_step_time": 6.823902368545532
    },
    {
      "epoch": 1.7491340637207032e-05,
      "model_forward_time": 2.3453969955444336,
      "step": 18341
    },
    {
      "epoch": 1.7491340637207032e-05,
      "step": 18341,
      "training_step_time": 7.6115498542785645
    },
    {
      "epoch": 1.7492294311523436e-05,
      "model_forward_time": 1.8227880001068115,
      "step": 18342
    },
    {
      "epoch": 1.7492294311523436e-05,
      "step": 18342,
      "training_step_time": 9.806350708007812
    },
    {
      "epoch": 1.7493247985839844e-05,
      "model_forward_time": 1.8156402111053467,
      "step": 18343
    },
    {
      "epoch": 1.7493247985839844e-05,
      "step": 18343,
      "training_step_time": 7.70922064781189
    },
    {
      "epoch": 1.749420166015625e-05,
      "model_forward_time": 1.8201656341552734,
      "step": 18344
    },
    {
      "epoch": 1.749420166015625e-05,
      "step": 18344,
      "training_step_time": 6.856296539306641
    },
    {
      "epoch": 1.7495155334472655e-05,
      "model_forward_time": 1.816598892211914,
      "step": 18345
    },
    {
      "epoch": 1.7495155334472655e-05,
      "step": 18345,
      "training_step_time": 6.810879707336426
    },
    {
      "epoch": 1.7496109008789063e-05,
      "model_forward_time": 1.8172097206115723,
      "step": 18346
    },
    {
      "epoch": 1.7496109008789063e-05,
      "step": 18346,
      "training_step_time": 6.803028106689453
    },
    {
      "epoch": 1.749706268310547e-05,
      "model_forward_time": 1.8198719024658203,
      "step": 18347
    },
    {
      "epoch": 1.749706268310547e-05,
      "step": 18347,
      "training_step_time": 6.859411001205444
    },
    {
      "epoch": 1.7498016357421874e-05,
      "model_forward_time": 1.8140897750854492,
      "step": 18348
    },
    {
      "epoch": 1.7498016357421874e-05,
      "step": 18348,
      "training_step_time": 6.815558671951294
    },
    {
      "epoch": 1.749897003173828e-05,
      "model_forward_time": 1.8222765922546387,
      "step": 18349
    },
    {
      "epoch": 1.749897003173828e-05,
      "step": 18349,
      "training_step_time": 6.811016082763672
    },
    {
      "epoch": 1.749992370605469e-05,
      "grad_norm": 0.19588015973567963,
      "learning_rate": 9.520679659618428e-06,
      "loss": 0.065,
      "step": 18350
    },
    {
      "epoch": 1.749992370605469e-05,
      "model_forward_time": 1.807258129119873,
      "step": 18350
    },
    {
      "dynamics_loss_avg": 0.06512508997693658,
      "epoch": 1.749992370605469e-05,
      "step": 18350
    },
    {
      "action_loss_avg": 0.0026859317382331937,
      "epoch": 1.749992370605469e-05,
      "step": 18350
    },
    {
      "epoch": 1.749992370605469e-05,
      "step": 18350,
      "training_step_time": 7.96809983253479
    },
    {
      "epoch": 1.7500877380371093e-05,
      "model_forward_time": 2.344602346420288,
      "step": 18351
    },
    {
      "epoch": 1.7500877380371093e-05,
      "step": 18351,
      "training_step_time": 11.123865842819214
    },
    {
      "epoch": 1.75018310546875e-05,
      "model_forward_time": 1.8155500888824463,
      "step": 18352
    },
    {
      "epoch": 1.75018310546875e-05,
      "step": 18352,
      "training_step_time": 7.780027151107788
    },
    {
      "epoch": 1.7502784729003908e-05,
      "model_forward_time": 1.8093953132629395,
      "step": 18353
    },
    {
      "epoch": 1.7502784729003908e-05,
      "step": 18353,
      "training_step_time": 6.808843374252319
    },
    {
      "epoch": 1.7503738403320312e-05,
      "model_forward_time": 1.8153698444366455,
      "step": 18354
    },
    {
      "epoch": 1.7503738403320312e-05,
      "step": 18354,
      "training_step_time": 6.796517372131348
    },
    {
      "epoch": 1.750469207763672e-05,
      "model_forward_time": 1.8083100318908691,
      "step": 18355
    },
    {
      "epoch": 1.750469207763672e-05,
      "step": 18355,
      "training_step_time": 6.703351736068726
    },
    {
      "epoch": 1.7505645751953127e-05,
      "model_forward_time": 1.815638542175293,
      "step": 18356
    },
    {
      "epoch": 1.7505645751953127e-05,
      "step": 18356,
      "training_step_time": 6.820958852767944
    },
    {
      "epoch": 1.750659942626953e-05,
      "model_forward_time": 1.8149476051330566,
      "step": 18357
    },
    {
      "epoch": 1.750659942626953e-05,
      "step": 18357,
      "training_step_time": 6.799889326095581
    },
    {
      "epoch": 1.7507553100585938e-05,
      "model_forward_time": 1.817751407623291,
      "step": 18358
    },
    {
      "epoch": 1.7507553100585938e-05,
      "step": 18358,
      "training_step_time": 6.827681303024292
    },
    {
      "epoch": 1.7508506774902345e-05,
      "model_forward_time": 1.8089118003845215,
      "step": 18359
    },
    {
      "epoch": 1.7508506774902345e-05,
      "step": 18359,
      "training_step_time": 7.443398952484131
    },
    {
      "epoch": 1.750946044921875e-05,
      "grad_norm": 0.19587667286396027,
      "learning_rate": 9.519972975873755e-06,
      "loss": 0.0642,
      "step": 18360
    },
    {
      "epoch": 1.750946044921875e-05,
      "model_forward_time": 2.3555057048797607,
      "step": 18360
    },
    {
      "dynamics_loss_avg": 0.0512632954865694,
      "epoch": 1.750946044921875e-05,
      "step": 18360
    },
    {
      "action_loss_avg": 0.0020639860769733785,
      "epoch": 1.750946044921875e-05,
      "step": 18360
    },
    {
      "epoch": 1.750946044921875e-05,
      "step": 18360,
      "training_step_time": 8.473722219467163
    },
    {
      "epoch": 1.7510414123535157e-05,
      "model_forward_time": 1.4456629753112793,
      "step": 18361
    },
    {
      "epoch": 1.7510414123535157e-05,
      "step": 18361,
      "training_step_time": 7.812702417373657
    },
    {
      "epoch": 1.751136779785156e-05,
      "model_forward_time": 1.8192625045776367,
      "step": 18362
    },
    {
      "epoch": 1.751136779785156e-05,
      "step": 18362,
      "training_step_time": 7.634671449661255
    },
    {
      "epoch": 1.751232147216797e-05,
      "model_forward_time": 1.8282740116119385,
      "step": 18363
    },
    {
      "epoch": 1.751232147216797e-05,
      "step": 18363,
      "training_step_time": 6.860351085662842
    },
    {
      "epoch": 1.7513275146484376e-05,
      "model_forward_time": 1.8228509426116943,
      "step": 18364
    },
    {
      "epoch": 1.7513275146484376e-05,
      "step": 18364,
      "training_step_time": 6.814800500869751
    },
    {
      "epoch": 1.751422882080078e-05,
      "model_forward_time": 1.8332715034484863,
      "step": 18365
    },
    {
      "epoch": 1.751422882080078e-05,
      "step": 18365,
      "training_step_time": 6.836479902267456
    },
    {
      "epoch": 1.7515182495117187e-05,
      "model_forward_time": 1.8167171478271484,
      "step": 18366
    },
    {
      "epoch": 1.7515182495117187e-05,
      "step": 18366,
      "training_step_time": 6.850348949432373
    },
    {
      "epoch": 1.7516136169433595e-05,
      "model_forward_time": 1.8104643821716309,
      "step": 18367
    },
    {
      "epoch": 1.7516136169433595e-05,
      "step": 18367,
      "training_step_time": 6.829320192337036
    },
    {
      "epoch": 1.751708984375e-05,
      "model_forward_time": 1.8097834587097168,
      "step": 18368
    },
    {
      "epoch": 1.751708984375e-05,
      "step": 18368,
      "training_step_time": 6.775042295455933
    },
    {
      "epoch": 1.7518043518066406e-05,
      "model_forward_time": 1.6429409980773926,
      "step": 18369
    },
    {
      "epoch": 1.7518043518066406e-05,
      "step": 18369,
      "training_step_time": 9.577521562576294
    },
    {
      "epoch": 1.7518997192382813e-05,
      "grad_norm": 0.16368354856967926,
      "learning_rate": 9.519265797831634e-06,
      "loss": 0.0607,
      "step": 18370
    },
    {
      "epoch": 1.7518997192382813e-05,
      "model_forward_time": 1.8020203113555908,
      "step": 18370
    },
    {
      "dynamics_loss_avg": 0.043415462225675584,
      "epoch": 1.7518997192382813e-05,
      "step": 18370
    },
    {
      "action_loss_avg": 0.0011423710093367845,
      "epoch": 1.7518997192382813e-05,
      "step": 18370
    },
    {
      "epoch": 1.7518997192382813e-05,
      "step": 18370,
      "training_step_time": 8.776826620101929
    },
    {
      "epoch": 1.7519950866699217e-05,
      "model_forward_time": 1.8147623538970947,
      "step": 18371
    },
    {
      "epoch": 1.7519950866699217e-05,
      "step": 18371,
      "training_step_time": 7.755819797515869
    },
    {
      "epoch": 1.7520904541015625e-05,
      "model_forward_time": 1.8146772384643555,
      "step": 18372
    },
    {
      "epoch": 1.7520904541015625e-05,
      "step": 18372,
      "training_step_time": 6.79728102684021
    },
    {
      "epoch": 1.7521858215332032e-05,
      "model_forward_time": 1.8192219734191895,
      "step": 18373
    },
    {
      "epoch": 1.7521858215332032e-05,
      "step": 18373,
      "training_step_time": 6.832937717437744
    },
    {
      "epoch": 1.7522811889648436e-05,
      "model_forward_time": 1.8208844661712646,
      "step": 18374
    },
    {
      "epoch": 1.7522811889648436e-05,
      "step": 18374,
      "training_step_time": 6.808841705322266
    },
    {
      "epoch": 1.7523765563964844e-05,
      "model_forward_time": 1.8214576244354248,
      "step": 18375
    },
    {
      "epoch": 1.7523765563964844e-05,
      "step": 18375,
      "training_step_time": 6.7369139194488525
    },
    {
      "epoch": 1.752471923828125e-05,
      "model_forward_time": 1.8147907257080078,
      "step": 18376
    },
    {
      "epoch": 1.752471923828125e-05,
      "step": 18376,
      "training_step_time": 6.816904544830322
    },
    {
      "epoch": 1.7525672912597655e-05,
      "model_forward_time": 1.8224818706512451,
      "step": 18377
    },
    {
      "epoch": 1.7525672912597655e-05,
      "step": 18377,
      "training_step_time": 6.8332884311676025
    },
    {
      "epoch": 1.7526626586914063e-05,
      "model_forward_time": 1.8157389163970947,
      "step": 18378
    },
    {
      "epoch": 1.7526626586914063e-05,
      "step": 18378,
      "training_step_time": 8.40308928489685
    },
    {
      "epoch": 1.752758026123047e-05,
      "model_forward_time": 2.369385004043579,
      "step": 18379
    },
    {
      "epoch": 1.752758026123047e-05,
      "step": 18379,
      "training_step_time": 9.846537113189697
    },
    {
      "epoch": 1.7528533935546874e-05,
      "grad_norm": 0.28341683745384216,
      "learning_rate": 9.5185581255694e-06,
      "loss": 0.0628,
      "step": 18380
    },
    {
      "epoch": 1.7528533935546874e-05,
      "model_forward_time": 1.811640977859497,
      "step": 18380
    },
    {
      "dynamics_loss_avg": 0.056148833222687246,
      "epoch": 1.7528533935546874e-05,
      "step": 18380
    },
    {
      "action_loss_avg": 0.0029948934447020293,
      "epoch": 1.7528533935546874e-05,
      "step": 18380
    },
    {
      "epoch": 1.7528533935546874e-05,
      "step": 18380,
      "training_step_time": 7.94152569770813
    },
    {
      "epoch": 1.752948760986328e-05,
      "model_forward_time": 1.8220477104187012,
      "step": 18381
    },
    {
      "epoch": 1.752948760986328e-05,
      "step": 18381,
      "training_step_time": 6.82670259475708
    },
    {
      "epoch": 1.753044128417969e-05,
      "model_forward_time": 1.819648027420044,
      "step": 18382
    },
    {
      "epoch": 1.753044128417969e-05,
      "step": 18382,
      "training_step_time": 6.7851035594940186
    },
    {
      "epoch": 1.7531394958496093e-05,
      "model_forward_time": 1.8177456855773926,
      "step": 18383
    },
    {
      "epoch": 1.7531394958496093e-05,
      "step": 18383,
      "training_step_time": 6.803845167160034
    },
    {
      "epoch": 1.75323486328125e-05,
      "model_forward_time": 1.8136577606201172,
      "step": 18384
    },
    {
      "epoch": 1.75323486328125e-05,
      "step": 18384,
      "training_step_time": 6.8051323890686035
    },
    {
      "epoch": 1.7533302307128908e-05,
      "model_forward_time": 1.814110279083252,
      "step": 18385
    },
    {
      "epoch": 1.7533302307128908e-05,
      "step": 18385,
      "training_step_time": 6.804346561431885
    },
    {
      "epoch": 1.7534255981445312e-05,
      "model_forward_time": 1.8207812309265137,
      "step": 18386
    },
    {
      "epoch": 1.7534255981445312e-05,
      "step": 18386,
      "training_step_time": 6.843219518661499
    },
    {
      "epoch": 1.753520965576172e-05,
      "model_forward_time": 1.820383071899414,
      "step": 18387
    },
    {
      "epoch": 1.753520965576172e-05,
      "step": 18387,
      "training_step_time": 8.150103569030762
    },
    {
      "epoch": 1.7536163330078127e-05,
      "model_forward_time": 2.372211217880249,
      "step": 18388
    },
    {
      "epoch": 1.7536163330078127e-05,
      "step": 18388,
      "training_step_time": 8.515738010406494
    },
    {
      "epoch": 1.753711700439453e-05,
      "model_forward_time": 1.8228673934936523,
      "step": 18389
    },
    {
      "epoch": 1.753711700439453e-05,
      "step": 18389,
      "training_step_time": 7.9061009883880615
    },
    {
      "epoch": 1.7538070678710938e-05,
      "grad_norm": 0.18927180767059326,
      "learning_rate": 9.517849959164442e-06,
      "loss": 0.0582,
      "step": 18390
    },
    {
      "epoch": 1.7538070678710938e-05,
      "model_forward_time": 1.8129124641418457,
      "step": 18390
    },
    {
      "dynamics_loss_avg": 0.059675409272313115,
      "epoch": 1.7538070678710938e-05,
      "step": 18390
    },
    {
      "action_loss_avg": 0.002776300231926143,
      "epoch": 1.7538070678710938e-05,
      "step": 18390
    },
    {
      "epoch": 1.7538070678710938e-05,
      "step": 18390,
      "training_step_time": 6.826211929321289
    },
    {
      "epoch": 1.7539024353027342e-05,
      "model_forward_time": 1.8171913623809814,
      "step": 18391
    },
    {
      "epoch": 1.7539024353027342e-05,
      "step": 18391,
      "training_step_time": 6.768533229827881
    },
    {
      "epoch": 1.753997802734375e-05,
      "model_forward_time": 1.822678565979004,
      "step": 18392
    },
    {
      "epoch": 1.753997802734375e-05,
      "step": 18392,
      "training_step_time": 6.7472593784332275
    },
    {
      "epoch": 1.7540931701660157e-05,
      "model_forward_time": 1.822108507156372,
      "step": 18393
    },
    {
      "epoch": 1.7540931701660157e-05,
      "step": 18393,
      "training_step_time": 6.760113000869751
    },
    {
      "epoch": 1.754188537597656e-05,
      "model_forward_time": 1.8197133541107178,
      "step": 18394
    },
    {
      "epoch": 1.754188537597656e-05,
      "step": 18394,
      "training_step_time": 6.821507930755615
    },
    {
      "epoch": 1.754283905029297e-05,
      "model_forward_time": 1.8074135780334473,
      "step": 18395
    },
    {
      "epoch": 1.754283905029297e-05,
      "step": 18395,
      "training_step_time": 6.835997104644775
    },
    {
      "epoch": 1.7543792724609376e-05,
      "model_forward_time": 1.8138105869293213,
      "step": 18396
    },
    {
      "epoch": 1.7543792724609376e-05,
      "step": 18396,
      "training_step_time": 7.878643751144409
    },
    {
      "epoch": 1.754474639892578e-05,
      "model_forward_time": 2.3415350914001465,
      "step": 18397
    },
    {
      "epoch": 1.754474639892578e-05,
      "step": 18397,
      "training_step_time": 10.382917404174805
    },
    {
      "epoch": 1.7545700073242187e-05,
      "model_forward_time": 1.8354737758636475,
      "step": 18398
    },
    {
      "epoch": 1.7545700073242187e-05,
      "step": 18398,
      "training_step_time": 8.507246732711792
    },
    {
      "epoch": 1.7546653747558595e-05,
      "model_forward_time": 1.832047462463379,
      "step": 18399
    },
    {
      "epoch": 1.7546653747558595e-05,
      "step": 18399,
      "training_step_time": 7.652786016464233
    },
    {
      "epoch": 1.7547607421875e-05,
      "grad_norm": 0.1784534603357315,
      "learning_rate": 9.517141298694207e-06,
      "loss": 0.0625,
      "step": 18400
    },
    {
      "epoch": 1.7547607421875e-05,
      "model_forward_time": 1.83223557472229,
      "step": 18400
    },
    {
      "dynamics_loss_avg": 0.05426889564841986,
      "epoch": 1.7547607421875e-05,
      "step": 18400
    },
    {
      "action_loss_avg": 0.0029675769328605384,
      "epoch": 1.7547607421875e-05,
      "step": 18400
    },
    {
      "epoch": 1.7547607421875e-05,
      "step": 18400,
      "training_step_time": 6.8162500858306885
    },
    {
      "epoch": 1.7548561096191406e-05,
      "model_forward_time": 1.844099521636963,
      "step": 18401
    },
    {
      "epoch": 1.7548561096191406e-05,
      "step": 18401,
      "training_step_time": 6.816005229949951
    },
    {
      "epoch": 1.7549514770507813e-05,
      "model_forward_time": 1.8261206150054932,
      "step": 18402
    },
    {
      "epoch": 1.7549514770507813e-05,
      "step": 18402,
      "training_step_time": 6.907199144363403
    },
    {
      "epoch": 1.7550468444824218e-05,
      "model_forward_time": 1.829124927520752,
      "step": 18403
    },
    {
      "epoch": 1.7550468444824218e-05,
      "step": 18403,
      "training_step_time": 6.884186267852783
    },
    {
      "epoch": 1.7551422119140625e-05,
      "model_forward_time": 1.838348388671875,
      "step": 18404
    },
    {
      "epoch": 1.7551422119140625e-05,
      "step": 18404,
      "training_step_time": 6.814492225646973
    },
    {
      "epoch": 1.7552375793457032e-05,
      "model_forward_time": 1.8002254962921143,
      "step": 18405
    },
    {
      "epoch": 1.7552375793457032e-05,
      "step": 18405,
      "training_step_time": 7.46588659286499
    },
    {
      "epoch": 1.7553329467773436e-05,
      "model_forward_time": 2.3199801445007324,
      "step": 18406
    },
    {
      "epoch": 1.7553329467773436e-05,
      "step": 18406,
      "training_step_time": 9.926216840744019
    },
    {
      "epoch": 1.7554283142089844e-05,
      "model_forward_time": 1.811185359954834,
      "step": 18407
    },
    {
      "epoch": 1.7554283142089844e-05,
      "step": 18407,
      "training_step_time": 10.478990077972412
    },
    {
      "epoch": 1.755523681640625e-05,
      "model_forward_time": 1.8166368007659912,
      "step": 18408
    },
    {
      "epoch": 1.755523681640625e-05,
      "step": 18408,
      "training_step_time": 7.727585315704346
    },
    {
      "epoch": 1.7556190490722655e-05,
      "model_forward_time": 1.810811996459961,
      "step": 18409
    },
    {
      "epoch": 1.7556190490722655e-05,
      "step": 18409,
      "training_step_time": 6.855026483535767
    },
    {
      "epoch": 1.7557144165039063e-05,
      "grad_norm": 0.20331405103206635,
      "learning_rate": 9.516432144236187e-06,
      "loss": 0.0638,
      "step": 18410
    },
    {
      "epoch": 1.7557144165039063e-05,
      "model_forward_time": 1.8120839595794678,
      "step": 18410
    },
    {
      "dynamics_loss_avg": 0.06901779826730489,
      "epoch": 1.7557144165039063e-05,
      "step": 18410
    },
    {
      "action_loss_avg": 0.0023492949781939386,
      "epoch": 1.7557144165039063e-05,
      "step": 18410
    },
    {
      "epoch": 1.7557144165039063e-05,
      "step": 18410,
      "training_step_time": 6.881594657897949
    },
    {
      "epoch": 1.755809783935547e-05,
      "model_forward_time": 1.8201651573181152,
      "step": 18411
    },
    {
      "epoch": 1.755809783935547e-05,
      "step": 18411,
      "training_step_time": 6.893401861190796
    },
    {
      "epoch": 1.7559051513671874e-05,
      "model_forward_time": 1.8050425052642822,
      "step": 18412
    },
    {
      "epoch": 1.7559051513671874e-05,
      "step": 18412,
      "training_step_time": 6.821649074554443
    },
    {
      "epoch": 1.756000518798828e-05,
      "model_forward_time": 1.814148187637329,
      "step": 18413
    },
    {
      "epoch": 1.756000518798828e-05,
      "step": 18413,
      "training_step_time": 6.885363340377808
    },
    {
      "epoch": 1.756095886230469e-05,
      "model_forward_time": 1.0896351337432861,
      "step": 18414
    },
    {
      "epoch": 1.756095886230469e-05,
      "step": 18414,
      "training_step_time": 7.541778326034546
    },
    {
      "epoch": 1.7561912536621093e-05,
      "model_forward_time": 1.810910701751709,
      "step": 18415
    },
    {
      "epoch": 1.7561912536621093e-05,
      "step": 18415,
      "training_step_time": 10.292977571487427
    },
    {
      "epoch": 1.75628662109375e-05,
      "model_forward_time": 2.3624746799468994,
      "step": 18416
    },
    {
      "epoch": 1.75628662109375e-05,
      "step": 18416,
      "training_step_time": 9.194010019302368
    },
    {
      "epoch": 1.7563819885253908e-05,
      "model_forward_time": 1.818718671798706,
      "step": 18417
    },
    {
      "epoch": 1.7563819885253908e-05,
      "step": 18417,
      "training_step_time": 7.777939558029175
    },
    {
      "epoch": 1.7564773559570312e-05,
      "model_forward_time": 1.8106613159179688,
      "step": 18418
    },
    {
      "epoch": 1.7564773559570312e-05,
      "step": 18418,
      "training_step_time": 6.847349405288696
    },
    {
      "epoch": 1.756572723388672e-05,
      "model_forward_time": 1.8136258125305176,
      "step": 18419
    },
    {
      "epoch": 1.756572723388672e-05,
      "step": 18419,
      "training_step_time": 6.799143552780151
    },
    {
      "epoch": 1.7566680908203127e-05,
      "grad_norm": 0.20205585658550262,
      "learning_rate": 9.515722495867942e-06,
      "loss": 0.0666,
      "step": 18420
    },
    {
      "epoch": 1.7566680908203127e-05,
      "model_forward_time": 1.8095505237579346,
      "step": 18420
    },
    {
      "dynamics_loss_avg": 0.040763729996979234,
      "epoch": 1.7566680908203127e-05,
      "step": 18420
    },
    {
      "action_loss_avg": 0.0017065485328203068,
      "epoch": 1.7566680908203127e-05,
      "step": 18420
    },
    {
      "epoch": 1.7566680908203127e-05,
      "step": 18420,
      "training_step_time": 6.94746208190918
    },
    {
      "epoch": 1.756763458251953e-05,
      "model_forward_time": 1.8122737407684326,
      "step": 18421
    },
    {
      "epoch": 1.756763458251953e-05,
      "step": 18421,
      "training_step_time": 6.978913068771362
    },
    {
      "epoch": 1.7568588256835938e-05,
      "model_forward_time": 1.815232515335083,
      "step": 18422
    },
    {
      "epoch": 1.7568588256835938e-05,
      "step": 18422,
      "training_step_time": 6.937438726425171
    },
    {
      "epoch": 1.7569541931152342e-05,
      "model_forward_time": 1.8009955883026123,
      "step": 18423
    },
    {
      "epoch": 1.7569541931152342e-05,
      "step": 18423,
      "training_step_time": 7.595628976821899
    },
    {
      "epoch": 1.757049560546875e-05,
      "model_forward_time": 1.8093292713165283,
      "step": 18424
    },
    {
      "epoch": 1.757049560546875e-05,
      "step": 18424,
      "training_step_time": 7.820465087890625
    },
    {
      "epoch": 1.7571449279785157e-05,
      "model_forward_time": 2.3843512535095215,
      "step": 18425
    },
    {
      "epoch": 1.7571449279785157e-05,
      "step": 18425,
      "training_step_time": 11.364457845687866
    },
    {
      "epoch": 1.757240295410156e-05,
      "model_forward_time": 1.8149850368499756,
      "step": 18426
    },
    {
      "epoch": 1.757240295410156e-05,
      "step": 18426,
      "training_step_time": 7.9245288372039795
    },
    {
      "epoch": 1.757335662841797e-05,
      "model_forward_time": 1.802541732788086,
      "step": 18427
    },
    {
      "epoch": 1.757335662841797e-05,
      "step": 18427,
      "training_step_time": 7.809833765029907
    },
    {
      "epoch": 1.7574310302734376e-05,
      "model_forward_time": 1.8180930614471436,
      "step": 18428
    },
    {
      "epoch": 1.7574310302734376e-05,
      "step": 18428,
      "training_step_time": 6.977544069290161
    },
    {
      "epoch": 1.757526397705078e-05,
      "model_forward_time": 1.8191430568695068,
      "step": 18429
    },
    {
      "epoch": 1.757526397705078e-05,
      "step": 18429,
      "training_step_time": 6.99860405921936
    },
    {
      "epoch": 1.7576217651367187e-05,
      "grad_norm": 0.18980921804904938,
      "learning_rate": 9.515012353667073e-06,
      "loss": 0.066,
      "step": 18430
    },
    {
      "epoch": 1.7576217651367187e-05,
      "model_forward_time": 1.810462474822998,
      "step": 18430
    },
    {
      "dynamics_loss_avg": 0.05528910979628563,
      "epoch": 1.7576217651367187e-05,
      "step": 18430
    },
    {
      "action_loss_avg": 0.002425713703269139,
      "epoch": 1.7576217651367187e-05,
      "step": 18430
    },
    {
      "epoch": 1.7576217651367187e-05,
      "step": 18430,
      "training_step_time": 6.977316379547119
    },
    {
      "epoch": 1.7577171325683595e-05,
      "model_forward_time": 1.8098604679107666,
      "step": 18431
    },
    {
      "epoch": 1.7577171325683595e-05,
      "step": 18431,
      "training_step_time": 6.934155225753784
    },
    {
      "epoch": 1.7578125e-05,
      "model_forward_time": 1.815669298171997,
      "step": 18432
    },
    {
      "epoch": 1.7578125e-05,
      "step": 18432,
      "training_step_time": 8.116744041442871
    },
    {
      "epoch": 1.7579078674316406e-05,
      "model_forward_time": 1.810436725616455,
      "step": 18433
    },
    {
      "epoch": 1.7579078674316406e-05,
      "step": 18433,
      "training_step_time": 9.22338581085205
    },
    {
      "epoch": 1.7580032348632814e-05,
      "model_forward_time": 2.3425567150115967,
      "step": 18434
    },
    {
      "epoch": 1.7580032348632814e-05,
      "step": 18434,
      "training_step_time": 11.101656675338745
    },
    {
      "epoch": 1.7580986022949218e-05,
      "model_forward_time": 1.8095386028289795,
      "step": 18435
    },
    {
      "epoch": 1.7580986022949218e-05,
      "step": 18435,
      "training_step_time": 8.720694303512573
    },
    {
      "epoch": 1.7581939697265625e-05,
      "model_forward_time": 1.8116965293884277,
      "step": 18436
    },
    {
      "epoch": 1.7581939697265625e-05,
      "step": 18436,
      "training_step_time": 7.733839988708496
    },
    {
      "epoch": 1.7582893371582032e-05,
      "model_forward_time": 1.8276112079620361,
      "step": 18437
    },
    {
      "epoch": 1.7582893371582032e-05,
      "step": 18437,
      "training_step_time": 6.932133197784424
    },
    {
      "epoch": 1.7583847045898436e-05,
      "model_forward_time": 1.8016982078552246,
      "step": 18438
    },
    {
      "epoch": 1.7583847045898436e-05,
      "step": 18438,
      "training_step_time": 6.916430234909058
    },
    {
      "epoch": 1.7584800720214844e-05,
      "model_forward_time": 1.8173892498016357,
      "step": 18439
    },
    {
      "epoch": 1.7584800720214844e-05,
      "step": 18439,
      "training_step_time": 7.03632116317749
    },
    {
      "epoch": 1.758575439453125e-05,
      "grad_norm": 0.1177150085568428,
      "learning_rate": 9.51430171771124e-06,
      "loss": 0.0663,
      "step": 18440
    },
    {
      "epoch": 1.758575439453125e-05,
      "model_forward_time": 1.8165900707244873,
      "step": 18440
    },
    {
      "dynamics_loss_avg": 0.08985452018678189,
      "epoch": 1.758575439453125e-05,
      "step": 18440
    },
    {
      "action_loss_avg": 0.003024658316280693,
      "epoch": 1.758575439453125e-05,
      "step": 18440
    },
    {
      "epoch": 1.758575439453125e-05,
      "step": 18440,
      "training_step_time": 6.9356935024261475
    },
    {
      "epoch": 1.7586708068847655e-05,
      "model_forward_time": 1.8285760879516602,
      "step": 18441
    },
    {
      "epoch": 1.7586708068847655e-05,
      "step": 18441,
      "training_step_time": 6.910893678665161
    },
    {
      "epoch": 1.7587661743164063e-05,
      "model_forward_time": 1.81829833984375,
      "step": 18442
    },
    {
      "epoch": 1.7587661743164063e-05,
      "step": 18442,
      "training_step_time": 7.81286883354187
    },
    {
      "epoch": 1.758861541748047e-05,
      "model_forward_time": 2.3327040672302246,
      "step": 18443
    },
    {
      "epoch": 1.758861541748047e-05,
      "step": 18443,
      "training_step_time": 9.152085781097412
    },
    {
      "epoch": 1.7589569091796874e-05,
      "model_forward_time": 1.81111478805542,
      "step": 18444
    },
    {
      "epoch": 1.7589569091796874e-05,
      "step": 18444,
      "training_step_time": 8.025387048721313
    },
    {
      "epoch": 1.759052276611328e-05,
      "model_forward_time": 1.8166627883911133,
      "step": 18445
    },
    {
      "epoch": 1.759052276611328e-05,
      "step": 18445,
      "training_step_time": 7.803705930709839
    },
    {
      "epoch": 1.759147644042969e-05,
      "model_forward_time": 1.8118767738342285,
      "step": 18446
    },
    {
      "epoch": 1.759147644042969e-05,
      "step": 18446,
      "training_step_time": 6.9108006954193115
    },
    {
      "epoch": 1.7592430114746093e-05,
      "model_forward_time": 1.8168931007385254,
      "step": 18447
    },
    {
      "epoch": 1.7592430114746093e-05,
      "step": 18447,
      "training_step_time": 6.892449855804443
    },
    {
      "epoch": 1.75933837890625e-05,
      "model_forward_time": 1.8244123458862305,
      "step": 18448
    },
    {
      "epoch": 1.75933837890625e-05,
      "step": 18448,
      "training_step_time": 6.963365793228149
    },
    {
      "epoch": 1.7594337463378908e-05,
      "model_forward_time": 1.7964658737182617,
      "step": 18449
    },
    {
      "epoch": 1.7594337463378908e-05,
      "step": 18449,
      "training_step_time": 6.978269100189209
    },
    {
      "epoch": 1.7595291137695312e-05,
      "grad_norm": 0.19016796350479126,
      "learning_rate": 9.513590588078159e-06,
      "loss": 0.0707,
      "step": 18450
    },
    {
      "epoch": 1.7595291137695312e-05,
      "model_forward_time": 1.8189401626586914,
      "step": 18450
    },
    {
      "dynamics_loss_avg": 0.05974878631532192,
      "epoch": 1.7595291137695312e-05,
      "step": 18450
    },
    {
      "action_loss_avg": 0.0019533279701136053,
      "epoch": 1.7595291137695312e-05,
      "step": 18450
    },
    {
      "epoch": 1.7595291137695312e-05,
      "step": 18450,
      "training_step_time": 6.927855968475342
    },
    {
      "epoch": 1.759624481201172e-05,
      "model_forward_time": 1.8119540214538574,
      "step": 18451
    },
    {
      "epoch": 1.759624481201172e-05,
      "step": 18451,
      "training_step_time": 7.642342567443848
    },
    {
      "epoch": 1.7597198486328127e-05,
      "model_forward_time": 1.4413485527038574,
      "step": 18452
    },
    {
      "epoch": 1.7597198486328127e-05,
      "step": 18452,
      "training_step_time": 9.6819486618042
    },
    {
      "epoch": 1.759815216064453e-05,
      "model_forward_time": 1.8201320171356201,
      "step": 18453
    },
    {
      "epoch": 1.759815216064453e-05,
      "step": 18453,
      "training_step_time": 9.589032649993896
    },
    {
      "epoch": 1.7599105834960938e-05,
      "model_forward_time": 1.8324599266052246,
      "step": 18454
    },
    {
      "epoch": 1.7599105834960938e-05,
      "step": 18454,
      "training_step_time": 8.340163707733154
    },
    {
      "epoch": 1.7600059509277342e-05,
      "model_forward_time": 1.8177008628845215,
      "step": 18455
    },
    {
      "epoch": 1.7600059509277342e-05,
      "step": 18455,
      "training_step_time": 6.968095541000366
    },
    {
      "epoch": 1.760101318359375e-05,
      "model_forward_time": 1.818208932876587,
      "step": 18456
    },
    {
      "epoch": 1.760101318359375e-05,
      "step": 18456,
      "training_step_time": 6.950400114059448
    },
    {
      "epoch": 1.7601966857910157e-05,
      "model_forward_time": 1.8212573528289795,
      "step": 18457
    },
    {
      "epoch": 1.7601966857910157e-05,
      "step": 18457,
      "training_step_time": 6.973421096801758
    },
    {
      "epoch": 1.760292053222656e-05,
      "model_forward_time": 1.8156957626342773,
      "step": 18458
    },
    {
      "epoch": 1.760292053222656e-05,
      "step": 18458,
      "training_step_time": 7.006932735443115
    },
    {
      "epoch": 1.760387420654297e-05,
      "model_forward_time": 1.8161695003509521,
      "step": 18459
    },
    {
      "epoch": 1.760387420654297e-05,
      "step": 18459,
      "training_step_time": 6.9926135540008545
    },
    {
      "epoch": 1.7604827880859376e-05,
      "grad_norm": 0.14642038941383362,
      "learning_rate": 9.512878964845598e-06,
      "loss": 0.0676,
      "step": 18460
    },
    {
      "epoch": 1.7604827880859376e-05,
      "model_forward_time": 1.441053867340088,
      "step": 18460
    },
    {
      "dynamics_loss_avg": 0.05782592818140984,
      "epoch": 1.7604827880859376e-05,
      "step": 18460
    },
    {
      "action_loss_avg": 0.0025970577786210924,
      "epoch": 1.7604827880859376e-05,
      "step": 18460
    },
    {
      "epoch": 1.7604827880859376e-05,
      "step": 18460,
      "training_step_time": 8.11923599243164
    },
    {
      "epoch": 1.760578155517578e-05,
      "model_forward_time": 1.8187255859375,
      "step": 18461
    },
    {
      "epoch": 1.760578155517578e-05,
      "step": 18461,
      "training_step_time": 9.301515340805054
    },
    {
      "epoch": 1.7606735229492187e-05,
      "model_forward_time": 2.3870913982391357,
      "step": 18462
    },
    {
      "epoch": 1.7606735229492187e-05,
      "step": 18462,
      "training_step_time": 11.323308944702148
    },
    {
      "epoch": 1.7607688903808595e-05,
      "model_forward_time": 1.819467544555664,
      "step": 18463
    },
    {
      "epoch": 1.7607688903808595e-05,
      "step": 18463,
      "training_step_time": 7.666443586349487
    },
    {
      "epoch": 1.7608642578125e-05,
      "model_forward_time": 1.8087055683135986,
      "step": 18464
    },
    {
      "epoch": 1.7608642578125e-05,
      "step": 18464,
      "training_step_time": 7.612056255340576
    },
    {
      "epoch": 1.7609596252441406e-05,
      "model_forward_time": 1.8315119743347168,
      "step": 18465
    },
    {
      "epoch": 1.7609596252441406e-05,
      "step": 18465,
      "training_step_time": 6.856813430786133
    },
    {
      "epoch": 1.7610549926757814e-05,
      "model_forward_time": 1.8242101669311523,
      "step": 18466
    },
    {
      "epoch": 1.7610549926757814e-05,
      "step": 18466,
      "training_step_time": 6.824667453765869
    },
    {
      "epoch": 1.7611503601074218e-05,
      "model_forward_time": 1.827805757522583,
      "step": 18467
    },
    {
      "epoch": 1.7611503601074218e-05,
      "step": 18467,
      "training_step_time": 6.775562047958374
    },
    {
      "epoch": 1.7612457275390625e-05,
      "model_forward_time": 1.7488055229187012,
      "step": 18468
    },
    {
      "epoch": 1.7612457275390625e-05,
      "step": 18468,
      "training_step_time": 6.798896551132202
    },
    {
      "epoch": 1.7613410949707032e-05,
      "model_forward_time": 1.3107681274414062,
      "step": 18469
    },
    {
      "epoch": 1.7613410949707032e-05,
      "step": 18469,
      "training_step_time": 7.379925966262817
    },
    {
      "epoch": 1.7614364624023436e-05,
      "grad_norm": 0.21383020281791687,
      "learning_rate": 9.512166848091378e-06,
      "loss": 0.0648,
      "step": 18470
    },
    {
      "epoch": 1.7614364624023436e-05,
      "model_forward_time": 1.7053461074829102,
      "step": 18470
    },
    {
      "dynamics_loss_avg": 0.05637982469052076,
      "epoch": 1.7614364624023436e-05,
      "step": 18470
    },
    {
      "action_loss_avg": 0.003100442816503346,
      "epoch": 1.7614364624023436e-05,
      "step": 18470
    },
    {
      "epoch": 1.7614364624023436e-05,
      "step": 18470,
      "training_step_time": 10.147531986236572
    },
    {
      "epoch": 1.7615318298339844e-05,
      "model_forward_time": 1.9233038425445557,
      "step": 18471
    },
    {
      "epoch": 1.7615318298339844e-05,
      "step": 18471,
      "training_step_time": 10.662718534469604
    },
    {
      "epoch": 1.761627197265625e-05,
      "model_forward_time": 1.7041170597076416,
      "step": 18472
    },
    {
      "epoch": 1.761627197265625e-05,
      "step": 18472,
      "training_step_time": 7.783939599990845
    },
    {
      "epoch": 1.7617225646972655e-05,
      "model_forward_time": 1.7063002586364746,
      "step": 18473
    },
    {
      "epoch": 1.7617225646972655e-05,
      "step": 18473,
      "training_step_time": 7.599553108215332
    },
    {
      "epoch": 1.7618179321289063e-05,
      "model_forward_time": 1.7084414958953857,
      "step": 18474
    },
    {
      "epoch": 1.7618179321289063e-05,
      "step": 18474,
      "training_step_time": 6.812177658081055
    },
    {
      "epoch": 1.761913299560547e-05,
      "model_forward_time": 1.329697608947754,
      "step": 18475
    },
    {
      "epoch": 1.761913299560547e-05,
      "step": 18475,
      "training_step_time": 6.762895107269287
    },
    {
      "epoch": 1.7620086669921874e-05,
      "model_forward_time": 1.7001159191131592,
      "step": 18476
    },
    {
      "epoch": 1.7620086669921874e-05,
      "step": 18476,
      "training_step_time": 6.786084413528442
    },
    {
      "epoch": 1.762104034423828e-05,
      "model_forward_time": 1.7049548625946045,
      "step": 18477
    },
    {
      "epoch": 1.762104034423828e-05,
      "step": 18477,
      "training_step_time": 6.811599254608154
    },
    {
      "epoch": 1.762199401855469e-05,
      "model_forward_time": 1.7089121341705322,
      "step": 18478
    },
    {
      "epoch": 1.762199401855469e-05,
      "step": 18478,
      "training_step_time": 7.440291881561279
    },
    {
      "epoch": 1.7622947692871093e-05,
      "model_forward_time": 1.7096431255340576,
      "step": 18479
    },
    {
      "epoch": 1.7622947692871093e-05,
      "step": 18479,
      "training_step_time": 8.17713189125061
    },
    {
      "epoch": 1.76239013671875e-05,
      "grad_norm": 0.2556150257587433,
      "learning_rate": 9.511454237893377e-06,
      "loss": 0.0655,
      "step": 18480
    },
    {
      "epoch": 1.76239013671875e-05,
      "model_forward_time": 1.7427780628204346,
      "step": 18480
    },
    {
      "dynamics_loss_avg": 0.07148109301924706,
      "epoch": 1.76239013671875e-05,
      "step": 18480
    },
    {
      "action_loss_avg": 0.002866622747387737,
      "epoch": 1.76239013671875e-05,
      "step": 18480
    },
    {
      "epoch": 1.76239013671875e-05,
      "step": 18480,
      "training_step_time": 10.164393186569214
    },
    {
      "epoch": 1.7624855041503908e-05,
      "model_forward_time": 1.7041356563568115,
      "step": 18481
    },
    {
      "epoch": 1.7624855041503908e-05,
      "step": 18481,
      "training_step_time": 7.83014440536499
    },
    {
      "epoch": 1.7625808715820312e-05,
      "model_forward_time": 1.6880030632019043,
      "step": 18482
    },
    {
      "epoch": 1.7625808715820312e-05,
      "step": 18482,
      "training_step_time": 7.755019903182983
    },
    {
      "epoch": 1.762676239013672e-05,
      "model_forward_time": 1.6969232559204102,
      "step": 18483
    },
    {
      "epoch": 1.762676239013672e-05,
      "step": 18483,
      "training_step_time": 6.8285181522369385
    },
    {
      "epoch": 1.7627716064453127e-05,
      "model_forward_time": 1.7143876552581787,
      "step": 18484
    },
    {
      "epoch": 1.7627716064453127e-05,
      "step": 18484,
      "training_step_time": 6.842993974685669
    },
    {
      "epoch": 1.762866973876953e-05,
      "model_forward_time": 1.710362195968628,
      "step": 18485
    },
    {
      "epoch": 1.762866973876953e-05,
      "step": 18485,
      "training_step_time": 6.820713996887207
    },
    {
      "epoch": 1.7629623413085938e-05,
      "model_forward_time": 1.7348523139953613,
      "step": 18486
    },
    {
      "epoch": 1.7629623413085938e-05,
      "step": 18486,
      "training_step_time": 6.8194661140441895
    },
    {
      "epoch": 1.7630577087402342e-05,
      "model_forward_time": 1.7153899669647217,
      "step": 18487
    },
    {
      "epoch": 1.7630577087402342e-05,
      "step": 18487,
      "training_step_time": 7.932919979095459
    },
    {
      "epoch": 1.763153076171875e-05,
      "model_forward_time": 1.7228097915649414,
      "step": 18488
    },
    {
      "epoch": 1.763153076171875e-05,
      "step": 18488,
      "training_step_time": 8.21275782585144
    },
    {
      "epoch": 1.7632484436035157e-05,
      "model_forward_time": 1.742983341217041,
      "step": 18489
    },
    {
      "epoch": 1.7632484436035157e-05,
      "step": 18489,
      "training_step_time": 9.2886803150177
    },
    {
      "epoch": 1.763343811035156e-05,
      "grad_norm": 0.24166344106197357,
      "learning_rate": 9.51074113432952e-06,
      "loss": 0.0675,
      "step": 18490
    },
    {
      "epoch": 1.763343811035156e-05,
      "model_forward_time": 2.278343677520752,
      "step": 18490
    },
    {
      "dynamics_loss_avg": 0.06506971009075642,
      "epoch": 1.763343811035156e-05,
      "step": 18490
    },
    {
      "action_loss_avg": 0.0036662217869888993,
      "epoch": 1.763343811035156e-05,
      "step": 18490
    },
    {
      "epoch": 1.763343811035156e-05,
      "step": 18490,
      "training_step_time": 8.988332033157349
    },
    {
      "epoch": 1.763439178466797e-05,
      "model_forward_time": 1.6940324306488037,
      "step": 18491
    },
    {
      "epoch": 1.763439178466797e-05,
      "step": 18491,
      "training_step_time": 7.879733562469482
    },
    {
      "epoch": 1.7635345458984376e-05,
      "model_forward_time": 1.7104253768920898,
      "step": 18492
    },
    {
      "epoch": 1.7635345458984376e-05,
      "step": 18492,
      "training_step_time": 7.560782432556152
    },
    {
      "epoch": 1.763629913330078e-05,
      "model_forward_time": 1.7166707515716553,
      "step": 18493
    },
    {
      "epoch": 1.763629913330078e-05,
      "step": 18493,
      "training_step_time": 6.829618453979492
    },
    {
      "epoch": 1.7637252807617187e-05,
      "model_forward_time": 1.7247710227966309,
      "step": 18494
    },
    {
      "epoch": 1.7637252807617187e-05,
      "step": 18494,
      "training_step_time": 6.849308013916016
    },
    {
      "epoch": 1.7638206481933595e-05,
      "model_forward_time": 1.7195806503295898,
      "step": 18495
    },
    {
      "epoch": 1.7638206481933595e-05,
      "step": 18495,
      "training_step_time": 6.822708368301392
    },
    {
      "epoch": 1.763916015625e-05,
      "model_forward_time": 1.7171337604522705,
      "step": 18496
    },
    {
      "epoch": 1.763916015625e-05,
      "step": 18496,
      "training_step_time": 7.2519447803497314
    },
    {
      "epoch": 1.7640113830566406e-05,
      "model_forward_time": 1.7147986888885498,
      "step": 18497
    },
    {
      "epoch": 1.7640113830566406e-05,
      "step": 18497,
      "training_step_time": 8.256904602050781
    },
    {
      "epoch": 1.7641067504882814e-05,
      "model_forward_time": 1.7250175476074219,
      "step": 18498
    },
    {
      "epoch": 1.7641067504882814e-05,
      "step": 18498,
      "training_step_time": 9.490206480026245
    },
    {
      "epoch": 1.7642021179199218e-05,
      "model_forward_time": 2.2649765014648438,
      "step": 18499
    },
    {
      "epoch": 1.7642021179199218e-05,
      "step": 18499,
      "training_step_time": 9.25610089302063
    },
    {
      "epoch": 1.7642974853515625e-05,
      "grad_norm": 0.2542673945426941,
      "learning_rate": 9.510027537477797e-06,
      "loss": 0.0643,
      "step": 18500
    },
    {
      "epoch": 1.7642974853515625e-05,
      "model_forward_time": 1.6870465278625488,
      "step": 18500
    },
    {
      "dynamics_loss_avg": 0.07141775414347648,
      "epoch": 1.7642974853515625e-05,
      "step": 18500
    },
    {
      "action_loss_avg": 0.0022101514565292745,
      "epoch": 1.7642974853515625e-05,
      "step": 18500
    },
    {
      "epoch": 1.7642974853515625e-05,
      "step": 18500,
      "training_step_time": 7.929370164871216
    },
    {
      "epoch": 1.7643928527832032e-05,
      "model_forward_time": 1.719604253768921,
      "step": 18501
    },
    {
      "epoch": 1.7643928527832032e-05,
      "step": 18501,
      "training_step_time": 7.593342304229736
    },
    {
      "epoch": 1.7644882202148436e-05,
      "model_forward_time": 1.728926658630371,
      "step": 18502
    },
    {
      "epoch": 1.7644882202148436e-05,
      "step": 18502,
      "training_step_time": 6.828347444534302
    },
    {
      "epoch": 1.7645835876464844e-05,
      "model_forward_time": 1.7155861854553223,
      "step": 18503
    },
    {
      "epoch": 1.7645835876464844e-05,
      "step": 18503,
      "training_step_time": 6.750843286514282
    },
    {
      "epoch": 1.764678955078125e-05,
      "model_forward_time": 1.7185287475585938,
      "step": 18504
    },
    {
      "epoch": 1.764678955078125e-05,
      "step": 18504,
      "training_step_time": 6.852155685424805
    },
    {
      "epoch": 1.7647743225097655e-05,
      "model_forward_time": 1.7714815139770508,
      "step": 18505
    },
    {
      "epoch": 1.7647743225097655e-05,
      "step": 18505,
      "training_step_time": 7.210353851318359
    },
    {
      "epoch": 1.7648696899414063e-05,
      "model_forward_time": 1.7234036922454834,
      "step": 18506
    },
    {
      "epoch": 1.7648696899414063e-05,
      "step": 18506,
      "training_step_time": 7.610056400299072
    },
    {
      "epoch": 1.764965057373047e-05,
      "model_forward_time": 1.720099687576294,
      "step": 18507
    },
    {
      "epoch": 1.764965057373047e-05,
      "step": 18507,
      "training_step_time": 8.67270803451538
    },
    {
      "epoch": 1.7650604248046874e-05,
      "model_forward_time": 2.293802261352539,
      "step": 18508
    },
    {
      "epoch": 1.7650604248046874e-05,
      "step": 18508,
      "training_step_time": 9.811112642288208
    },
    {
      "epoch": 1.765155792236328e-05,
      "model_forward_time": 1.6971502304077148,
      "step": 18509
    },
    {
      "epoch": 1.765155792236328e-05,
      "step": 18509,
      "training_step_time": 8.673617124557495
    },
    {
      "epoch": 1.765251159667969e-05,
      "grad_norm": 0.16293546557426453,
      "learning_rate": 9.509313447416241e-06,
      "loss": 0.0611,
      "step": 18510
    },
    {
      "epoch": 1.765251159667969e-05,
      "model_forward_time": 1.6958012580871582,
      "step": 18510
    },
    {
      "dynamics_loss_avg": 0.05434726420789957,
      "epoch": 1.765251159667969e-05,
      "step": 18510
    },
    {
      "action_loss_avg": 0.0026785538415424527,
      "epoch": 1.765251159667969e-05,
      "step": 18510
    },
    {
      "epoch": 1.765251159667969e-05,
      "step": 18510,
      "training_step_time": 7.693048715591431
    },
    {
      "epoch": 1.7653465270996093e-05,
      "model_forward_time": 1.710914134979248,
      "step": 18511
    },
    {
      "epoch": 1.7653465270996093e-05,
      "step": 18511,
      "training_step_time": 6.892145156860352
    },
    {
      "epoch": 1.76544189453125e-05,
      "model_forward_time": 1.7012479305267334,
      "step": 18512
    },
    {
      "epoch": 1.76544189453125e-05,
      "step": 18512,
      "training_step_time": 6.871338367462158
    },
    {
      "epoch": 1.7655372619628908e-05,
      "model_forward_time": 1.7045459747314453,
      "step": 18513
    },
    {
      "epoch": 1.7655372619628908e-05,
      "step": 18513,
      "training_step_time": 6.914524078369141
    },
    {
      "epoch": 1.7656326293945312e-05,
      "model_forward_time": 1.6813931465148926,
      "step": 18514
    },
    {
      "epoch": 1.7656326293945312e-05,
      "step": 18514,
      "training_step_time": 7.326829433441162
    },
    {
      "epoch": 1.765727996826172e-05,
      "model_forward_time": 1.7004384994506836,
      "step": 18515
    },
    {
      "epoch": 1.765727996826172e-05,
      "step": 18515,
      "training_step_time": 9.061862230300903
    },
    {
      "epoch": 1.7658233642578123e-05,
      "model_forward_time": 1.6903183460235596,
      "step": 18516
    },
    {
      "epoch": 1.7658233642578123e-05,
      "step": 18516,
      "training_step_time": 8.65266752243042
    },
    {
      "epoch": 1.765918731689453e-05,
      "model_forward_time": 2.2797935009002686,
      "step": 18517
    },
    {
      "epoch": 1.765918731689453e-05,
      "step": 18517,
      "training_step_time": 9.572429418563843
    },
    {
      "epoch": 1.7660140991210938e-05,
      "model_forward_time": 1.6921114921569824,
      "step": 18518
    },
    {
      "epoch": 1.7660140991210938e-05,
      "step": 18518,
      "training_step_time": 7.751439332962036
    },
    {
      "epoch": 1.7661094665527342e-05,
      "model_forward_time": 1.7030308246612549,
      "step": 18519
    },
    {
      "epoch": 1.7661094665527342e-05,
      "step": 18519,
      "training_step_time": 7.858251571655273
    },
    {
      "epoch": 1.766204833984375e-05,
      "grad_norm": 0.24238216876983643,
      "learning_rate": 9.50859886422295e-06,
      "loss": 0.0668,
      "step": 18520
    },
    {
      "epoch": 1.766204833984375e-05,
      "model_forward_time": 1.6895077228546143,
      "step": 18520
    },
    {
      "dynamics_loss_avg": 0.07040526643395424,
      "epoch": 1.766204833984375e-05,
      "step": 18520
    },
    {
      "action_loss_avg": 0.0033243392186705022,
      "epoch": 1.766204833984375e-05,
      "step": 18520
    },
    {
      "epoch": 1.766204833984375e-05,
      "step": 18520,
      "training_step_time": 7.139555931091309
    },
    {
      "epoch": 1.7663002014160157e-05,
      "model_forward_time": 1.687493085861206,
      "step": 18521
    },
    {
      "epoch": 1.7663002014160157e-05,
      "step": 18521,
      "training_step_time": 6.833624839782715
    },
    {
      "epoch": 1.766395568847656e-05,
      "model_forward_time": 1.69334077835083,
      "step": 18522
    },
    {
      "epoch": 1.766395568847656e-05,
      "step": 18522,
      "training_step_time": 6.869975566864014
    },
    {
      "epoch": 1.766490936279297e-05,
      "model_forward_time": 1.701582908630371,
      "step": 18523
    },
    {
      "epoch": 1.766490936279297e-05,
      "step": 18523,
      "training_step_time": 7.2188804149627686
    },
    {
      "epoch": 1.7665863037109376e-05,
      "model_forward_time": 1.6896319389343262,
      "step": 18524
    },
    {
      "epoch": 1.7665863037109376e-05,
      "step": 18524,
      "training_step_time": 7.37861967086792
    },
    {
      "epoch": 1.766681671142578e-05,
      "model_forward_time": 1.6922831535339355,
      "step": 18525
    },
    {
      "epoch": 1.766681671142578e-05,
      "step": 18525,
      "training_step_time": 7.817112445831299
    },
    {
      "epoch": 1.7667770385742187e-05,
      "model_forward_time": 2.225714921951294,
      "step": 18526
    },
    {
      "epoch": 1.7667770385742187e-05,
      "step": 18526,
      "training_step_time": 8.602438688278198
    },
    {
      "epoch": 1.7668724060058595e-05,
      "model_forward_time": 1.797088384628296,
      "step": 18527
    },
    {
      "epoch": 1.7668724060058595e-05,
      "step": 18527,
      "training_step_time": 7.815538644790649
    },
    {
      "epoch": 1.7669677734375e-05,
      "model_forward_time": 1.7948925495147705,
      "step": 18528
    },
    {
      "epoch": 1.7669677734375e-05,
      "step": 18528,
      "training_step_time": 7.870437383651733
    },
    {
      "epoch": 1.7670631408691406e-05,
      "model_forward_time": 1.8039517402648926,
      "step": 18529
    },
    {
      "epoch": 1.7670631408691406e-05,
      "step": 18529,
      "training_step_time": 7.124932050704956
    },
    {
      "epoch": 1.7671585083007814e-05,
      "grad_norm": 0.2090841382741928,
      "learning_rate": 9.507883787976063e-06,
      "loss": 0.0636,
      "step": 18530
    },
    {
      "epoch": 1.7671585083007814e-05,
      "model_forward_time": 1.79060959815979,
      "step": 18530
    },
    {
      "dynamics_loss_avg": 0.05262753032147884,
      "epoch": 1.7671585083007814e-05,
      "step": 18530
    },
    {
      "action_loss_avg": 0.0020910875988192855,
      "epoch": 1.7671585083007814e-05,
      "step": 18530
    },
    {
      "epoch": 1.7671585083007814e-05,
      "step": 18530,
      "training_step_time": 6.833735466003418
    },
    {
      "epoch": 1.7672538757324218e-05,
      "model_forward_time": 1.796058177947998,
      "step": 18531
    },
    {
      "epoch": 1.7672538757324218e-05,
      "step": 18531,
      "training_step_time": 6.847511291503906
    },
    {
      "epoch": 1.7673492431640625e-05,
      "model_forward_time": 1.4225232601165771,
      "step": 18532
    },
    {
      "epoch": 1.7673492431640625e-05,
      "step": 18532,
      "training_step_time": 6.827693939208984
    },
    {
      "epoch": 1.7674446105957032e-05,
      "model_forward_time": 1.795546054840088,
      "step": 18533
    },
    {
      "epoch": 1.7674446105957032e-05,
      "step": 18533,
      "training_step_time": 8.223048448562622
    },
    {
      "epoch": 1.7675399780273436e-05,
      "model_forward_time": 1.7949550151824951,
      "step": 18534
    },
    {
      "epoch": 1.7675399780273436e-05,
      "step": 18534,
      "training_step_time": 8.723426818847656
    },
    {
      "epoch": 1.7676353454589844e-05,
      "model_forward_time": 1.7924888134002686,
      "step": 18535
    },
    {
      "epoch": 1.7676353454589844e-05,
      "step": 18535,
      "training_step_time": 9.53610348701477
    },
    {
      "epoch": 1.767730712890625e-05,
      "model_forward_time": 2.31329345703125,
      "step": 18536
    },
    {
      "epoch": 1.767730712890625e-05,
      "step": 18536,
      "training_step_time": 8.93652606010437
    },
    {
      "epoch": 1.7678260803222655e-05,
      "model_forward_time": 1.795396089553833,
      "step": 18537
    },
    {
      "epoch": 1.7678260803222655e-05,
      "step": 18537,
      "training_step_time": 8.896400213241577
    },
    {
      "epoch": 1.7679214477539063e-05,
      "model_forward_time": 1.7952759265899658,
      "step": 18538
    },
    {
      "epoch": 1.7679214477539063e-05,
      "step": 18538,
      "training_step_time": 7.126707315444946
    },
    {
      "epoch": 1.768016815185547e-05,
      "model_forward_time": 1.797799825668335,
      "step": 18539
    },
    {
      "epoch": 1.768016815185547e-05,
      "step": 18539,
      "training_step_time": 6.798198461532593
    },
    {
      "epoch": 1.7681121826171874e-05,
      "grad_norm": 0.14521467685699463,
      "learning_rate": 9.507168218753782e-06,
      "loss": 0.0663,
      "step": 18540
    },
    {
      "epoch": 1.7681121826171874e-05,
      "model_forward_time": 1.7964146137237549,
      "step": 18540
    },
    {
      "dynamics_loss_avg": 0.06918681040406227,
      "epoch": 1.7681121826171874e-05,
      "step": 18540
    },
    {
      "action_loss_avg": 0.0025841100024990736,
      "epoch": 1.7681121826171874e-05,
      "step": 18540
    },
    {
      "epoch": 1.7681121826171874e-05,
      "step": 18540,
      "training_step_time": 6.751704692840576
    },
    {
      "epoch": 1.768207550048828e-05,
      "model_forward_time": 1.8029022216796875,
      "step": 18541
    },
    {
      "epoch": 1.768207550048828e-05,
      "step": 18541,
      "training_step_time": 6.7689995765686035
    },
    {
      "epoch": 1.768302917480469e-05,
      "model_forward_time": 1.7945544719696045,
      "step": 18542
    },
    {
      "epoch": 1.768302917480469e-05,
      "step": 18542,
      "training_step_time": 7.378057479858398
    },
    {
      "epoch": 1.7683982849121093e-05,
      "model_forward_time": 1.7921092510223389,
      "step": 18543
    },
    {
      "epoch": 1.7683982849121093e-05,
      "step": 18543,
      "training_step_time": 8.144196271896362
    },
    {
      "epoch": 1.76849365234375e-05,
      "model_forward_time": 1.7988498210906982,
      "step": 18544
    },
    {
      "epoch": 1.76849365234375e-05,
      "step": 18544,
      "training_step_time": 8.428705930709839
    },
    {
      "epoch": 1.7685890197753908e-05,
      "model_forward_time": 1.931473970413208,
      "step": 18545
    },
    {
      "epoch": 1.7685890197753908e-05,
      "step": 18545,
      "training_step_time": 7.712183475494385
    },
    {
      "epoch": 1.7686843872070312e-05,
      "model_forward_time": 1.730647087097168,
      "step": 18546
    },
    {
      "epoch": 1.7686843872070312e-05,
      "step": 18546,
      "training_step_time": 7.7628350257873535
    },
    {
      "epoch": 1.768779754638672e-05,
      "model_forward_time": 1.694166660308838,
      "step": 18547
    },
    {
      "epoch": 1.768779754638672e-05,
      "step": 18547,
      "training_step_time": 7.453760385513306
    },
    {
      "epoch": 1.7688751220703123e-05,
      "model_forward_time": 1.6888577938079834,
      "step": 18548
    },
    {
      "epoch": 1.7688751220703123e-05,
      "step": 18548,
      "training_step_time": 6.76444149017334
    },
    {
      "epoch": 1.768970489501953e-05,
      "model_forward_time": 1.6867377758026123,
      "step": 18549
    },
    {
      "epoch": 1.768970489501953e-05,
      "step": 18549,
      "training_step_time": 6.773097515106201
    },
    {
      "epoch": 1.7690658569335938e-05,
      "grad_norm": 0.1465384066104889,
      "learning_rate": 9.50645215663436e-06,
      "loss": 0.0629,
      "step": 18550
    },
    {
      "epoch": 1.7690658569335938e-05,
      "model_forward_time": 1.68017578125,
      "step": 18550
    },
    {
      "dynamics_loss_avg": 0.08696875125169753,
      "epoch": 1.7690658569335938e-05,
      "step": 18550
    },
    {
      "action_loss_avg": 0.003839111630804837,
      "epoch": 1.7690658569335938e-05,
      "step": 18550
    },
    {
      "epoch": 1.7690658569335938e-05,
      "step": 18550,
      "training_step_time": 6.787164211273193
    },
    {
      "epoch": 1.7691612243652342e-05,
      "model_forward_time": 1.6960124969482422,
      "step": 18551
    },
    {
      "epoch": 1.7691612243652342e-05,
      "step": 18551,
      "training_step_time": 7.3645851612091064
    },
    {
      "epoch": 1.769256591796875e-05,
      "model_forward_time": 1.7043919563293457,
      "step": 18552
    },
    {
      "epoch": 1.769256591796875e-05,
      "step": 18552,
      "training_step_time": 7.613542556762695
    },
    {
      "epoch": 1.7693519592285157e-05,
      "model_forward_time": 1.7064647674560547,
      "step": 18553
    },
    {
      "epoch": 1.7693519592285157e-05,
      "step": 18553,
      "training_step_time": 10.1448655128479
    },
    {
      "epoch": 1.769447326660156e-05,
      "model_forward_time": 0.9964637756347656,
      "step": 18554
    },
    {
      "epoch": 1.769447326660156e-05,
      "step": 18554,
      "training_step_time": 8.59551191329956
    },
    {
      "epoch": 1.769542694091797e-05,
      "model_forward_time": 2.3126587867736816,
      "step": 18555
    },
    {
      "epoch": 1.769542694091797e-05,
      "step": 18555,
      "training_step_time": 9.265254974365234
    },
    {
      "epoch": 1.7696380615234376e-05,
      "model_forward_time": 1.7128710746765137,
      "step": 18556
    },
    {
      "epoch": 1.7696380615234376e-05,
      "step": 18556,
      "training_step_time": 7.5340492725372314
    },
    {
      "epoch": 1.769733428955078e-05,
      "model_forward_time": 1.7077515125274658,
      "step": 18557
    },
    {
      "epoch": 1.769733428955078e-05,
      "step": 18557,
      "training_step_time": 7.062666893005371
    },
    {
      "epoch": 1.7698287963867187e-05,
      "model_forward_time": 1.7038216590881348,
      "step": 18558
    },
    {
      "epoch": 1.7698287963867187e-05,
      "step": 18558,
      "training_step_time": 6.76341438293457
    },
    {
      "epoch": 1.7699241638183595e-05,
      "model_forward_time": 1.7109949588775635,
      "step": 18559
    },
    {
      "epoch": 1.7699241638183595e-05,
      "step": 18559,
      "training_step_time": 6.850313663482666
    },
    {
      "epoch": 1.77001953125e-05,
      "grad_norm": 0.18708732724189758,
      "learning_rate": 9.505735601696109e-06,
      "loss": 0.0632,
      "step": 18560
    },
    {
      "epoch": 1.77001953125e-05,
      "model_forward_time": 1.7060656547546387,
      "step": 18560
    },
    {
      "dynamics_loss_avg": 0.05879662688821554,
      "epoch": 1.77001953125e-05,
      "step": 18560
    },
    {
      "action_loss_avg": 0.0026694570668041708,
      "epoch": 1.77001953125e-05,
      "step": 18560
    },
    {
      "epoch": 1.77001953125e-05,
      "step": 18560,
      "training_step_time": 7.265471935272217
    },
    {
      "epoch": 1.7701148986816406e-05,
      "model_forward_time": 1.7053377628326416,
      "step": 18561
    },
    {
      "epoch": 1.7701148986816406e-05,
      "step": 18561,
      "training_step_time": 9.282712459564209
    },
    {
      "epoch": 1.7702102661132814e-05,
      "model_forward_time": 1.7033629417419434,
      "step": 18562
    },
    {
      "epoch": 1.7702102661132814e-05,
      "step": 18562,
      "training_step_time": 8.393871784210205
    },
    {
      "epoch": 1.7703056335449218e-05,
      "model_forward_time": 1.7066195011138916,
      "step": 18563
    },
    {
      "epoch": 1.7703056335449218e-05,
      "step": 18563,
      "training_step_time": 9.664358377456665
    },
    {
      "epoch": 1.7704010009765625e-05,
      "model_forward_time": 1.943572998046875,
      "step": 18564
    },
    {
      "epoch": 1.7704010009765625e-05,
      "step": 18564,
      "training_step_time": 7.803836107254028
    },
    {
      "epoch": 1.7704963684082032e-05,
      "model_forward_time": 1.7078578472137451,
      "step": 18565
    },
    {
      "epoch": 1.7704963684082032e-05,
      "step": 18565,
      "training_step_time": 7.6054089069366455
    },
    {
      "epoch": 1.7705917358398436e-05,
      "model_forward_time": 1.702120304107666,
      "step": 18566
    },
    {
      "epoch": 1.7705917358398436e-05,
      "step": 18566,
      "training_step_time": 7.124240875244141
    },
    {
      "epoch": 1.7706871032714844e-05,
      "model_forward_time": 1.7116508483886719,
      "step": 18567
    },
    {
      "epoch": 1.7706871032714844e-05,
      "step": 18567,
      "training_step_time": 6.768206834793091
    },
    {
      "epoch": 1.770782470703125e-05,
      "model_forward_time": 1.7068474292755127,
      "step": 18568
    },
    {
      "epoch": 1.770782470703125e-05,
      "step": 18568,
      "training_step_time": 6.799948215484619
    },
    {
      "epoch": 1.7708778381347655e-05,
      "model_forward_time": 1.7048907279968262,
      "step": 18569
    },
    {
      "epoch": 1.7708778381347655e-05,
      "step": 18569,
      "training_step_time": 7.935554027557373
    },
    {
      "epoch": 1.7709732055664063e-05,
      "grad_norm": 0.23108336329460144,
      "learning_rate": 9.505018554017385e-06,
      "loss": 0.0601,
      "step": 18570
    },
    {
      "epoch": 1.7709732055664063e-05,
      "model_forward_time": 1.6962530612945557,
      "step": 18570
    },
    {
      "dynamics_loss_avg": 0.07030994165688753,
      "epoch": 1.7709732055664063e-05,
      "step": 18570
    },
    {
      "action_loss_avg": 0.004790082154795527,
      "epoch": 1.7709732055664063e-05,
      "step": 18570
    },
    {
      "epoch": 1.7709732055664063e-05,
      "step": 18570,
      "training_step_time": 8.106906414031982
    },
    {
      "epoch": 1.771068572998047e-05,
      "model_forward_time": 1.715975284576416,
      "step": 18571
    },
    {
      "epoch": 1.771068572998047e-05,
      "step": 18571,
      "training_step_time": 7.792623281478882
    },
    {
      "epoch": 1.7711639404296874e-05,
      "model_forward_time": 1.701869010925293,
      "step": 18572
    },
    {
      "epoch": 1.7711639404296874e-05,
      "step": 18572,
      "training_step_time": 9.704437732696533
    },
    {
      "epoch": 1.771259307861328e-05,
      "model_forward_time": 2.2906229496002197,
      "step": 18573
    },
    {
      "epoch": 1.771259307861328e-05,
      "step": 18573,
      "training_step_time": 7.81357479095459
    },
    {
      "epoch": 1.771354675292969e-05,
      "model_forward_time": 1.7508845329284668,
      "step": 18574
    },
    {
      "epoch": 1.771354675292969e-05,
      "step": 18574,
      "training_step_time": 7.569977521896362
    },
    {
      "epoch": 1.7714500427246093e-05,
      "model_forward_time": 1.7405593395233154,
      "step": 18575
    },
    {
      "epoch": 1.7714500427246093e-05,
      "step": 18575,
      "training_step_time": 7.089585304260254
    },
    {
      "epoch": 1.77154541015625e-05,
      "model_forward_time": 1.7026128768920898,
      "step": 18576
    },
    {
      "epoch": 1.77154541015625e-05,
      "step": 18576,
      "training_step_time": 6.776130676269531
    },
    {
      "epoch": 1.7716407775878908e-05,
      "model_forward_time": 1.7146961688995361,
      "step": 18577
    },
    {
      "epoch": 1.7716407775878908e-05,
      "step": 18577,
      "training_step_time": 6.773756265640259
    },
    {
      "epoch": 1.7717361450195312e-05,
      "model_forward_time": 1.7102463245391846,
      "step": 18578
    },
    {
      "epoch": 1.7717361450195312e-05,
      "step": 18578,
      "training_step_time": 7.2004005908966064
    },
    {
      "epoch": 1.771831512451172e-05,
      "model_forward_time": 1.695863962173462,
      "step": 18579
    },
    {
      "epoch": 1.771831512451172e-05,
      "step": 18579,
      "training_step_time": 8.105478286743164
    },
    {
      "epoch": 1.7719268798828123e-05,
      "grad_norm": 0.24420705437660217,
      "learning_rate": 9.504301013676605e-06,
      "loss": 0.0616,
      "step": 18580
    },
    {
      "epoch": 1.7719268798828123e-05,
      "model_forward_time": 1.7058136463165283,
      "step": 18580
    },
    {
      "dynamics_loss_avg": 0.05583376809954643,
      "epoch": 1.7719268798828123e-05,
      "step": 18580
    },
    {
      "action_loss_avg": 0.003829309542197734,
      "epoch": 1.7719268798828123e-05,
      "step": 18580
    },
    {
      "epoch": 1.7719268798828123e-05,
      "step": 18580,
      "training_step_time": 7.661221981048584
    },
    {
      "epoch": 1.772022247314453e-05,
      "model_forward_time": 0.9956073760986328,
      "step": 18581
    },
    {
      "epoch": 1.772022247314453e-05,
      "step": 18581,
      "training_step_time": 8.378034353256226
    },
    {
      "epoch": 1.7721176147460938e-05,
      "model_forward_time": 1.705507755279541,
      "step": 18582
    },
    {
      "epoch": 1.7721176147460938e-05,
      "step": 18582,
      "training_step_time": 9.31566596031189
    },
    {
      "epoch": 1.7722129821777342e-05,
      "model_forward_time": 2.2909953594207764,
      "step": 18583
    },
    {
      "epoch": 1.7722129821777342e-05,
      "step": 18583,
      "training_step_time": 7.79777979850769
    },
    {
      "epoch": 1.772308349609375e-05,
      "model_forward_time": 1.7155463695526123,
      "step": 18584
    },
    {
      "epoch": 1.772308349609375e-05,
      "step": 18584,
      "training_step_time": 7.627183675765991
    },
    {
      "epoch": 1.7724037170410157e-05,
      "model_forward_time": 1.7162566184997559,
      "step": 18585
    },
    {
      "epoch": 1.7724037170410157e-05,
      "step": 18585,
      "training_step_time": 7.35246729850769
    },
    {
      "epoch": 1.772499084472656e-05,
      "model_forward_time": 1.7074313163757324,
      "step": 18586
    },
    {
      "epoch": 1.772499084472656e-05,
      "step": 18586,
      "training_step_time": 6.739022731781006
    },
    {
      "epoch": 1.772594451904297e-05,
      "model_forward_time": 1.712672472000122,
      "step": 18587
    },
    {
      "epoch": 1.772594451904297e-05,
      "step": 18587,
      "training_step_time": 7.167115211486816
    },
    {
      "epoch": 1.7726898193359376e-05,
      "model_forward_time": 1.3311488628387451,
      "step": 18588
    },
    {
      "epoch": 1.7726898193359376e-05,
      "step": 18588,
      "training_step_time": 7.3508594036102295
    },
    {
      "epoch": 1.772785186767578e-05,
      "model_forward_time": 1.7108004093170166,
      "step": 18589
    },
    {
      "epoch": 1.772785186767578e-05,
      "step": 18589,
      "training_step_time": 8.671293497085571
    },
    {
      "epoch": 1.7728805541992187e-05,
      "grad_norm": 0.21409521996974945,
      "learning_rate": 9.503582980752239e-06,
      "loss": 0.0642,
      "step": 18590
    },
    {
      "epoch": 1.7728805541992187e-05,
      "model_forward_time": 1.70912766456604,
      "step": 18590
    },
    {
      "dynamics_loss_avg": 0.05205578189343214,
      "epoch": 1.7728805541992187e-05,
      "step": 18590
    },
    {
      "action_loss_avg": 0.00200683472212404,
      "epoch": 1.7728805541992187e-05,
      "step": 18590
    },
    {
      "epoch": 1.7728805541992187e-05,
      "step": 18590,
      "training_step_time": 8.122317552566528
    },
    {
      "epoch": 1.7729759216308595e-05,
      "model_forward_time": 1.7129476070404053,
      "step": 18591
    },
    {
      "epoch": 1.7729759216308595e-05,
      "step": 18591,
      "training_step_time": 8.329269886016846
    },
    {
      "epoch": 1.7730712890625e-05,
      "model_forward_time": 2.31970477104187,
      "step": 18592
    },
    {
      "epoch": 1.7730712890625e-05,
      "step": 18592,
      "training_step_time": 7.716809988021851
    },
    {
      "epoch": 1.7731666564941406e-05,
      "model_forward_time": 1.7131505012512207,
      "step": 18593
    },
    {
      "epoch": 1.7731666564941406e-05,
      "step": 18593,
      "training_step_time": 7.565934181213379
    },
    {
      "epoch": 1.7732620239257814e-05,
      "model_forward_time": 1.710930585861206,
      "step": 18594
    },
    {
      "epoch": 1.7732620239257814e-05,
      "step": 18594,
      "training_step_time": 6.734136343002319
    },
    {
      "epoch": 1.7733573913574218e-05,
      "model_forward_time": 1.7195727825164795,
      "step": 18595
    },
    {
      "epoch": 1.7733573913574218e-05,
      "step": 18595,
      "training_step_time": 7.083170175552368
    },
    {
      "epoch": 1.7734527587890625e-05,
      "model_forward_time": 1.3300976753234863,
      "step": 18596
    },
    {
      "epoch": 1.7734527587890625e-05,
      "step": 18596,
      "training_step_time": 7.162712335586548
    },
    {
      "epoch": 1.7735481262207032e-05,
      "model_forward_time": 1.7112751007080078,
      "step": 18597
    },
    {
      "epoch": 1.7735481262207032e-05,
      "step": 18597,
      "training_step_time": 7.2411110401153564
    },
    {
      "epoch": 1.7736434936523437e-05,
      "model_forward_time": 1.716292381286621,
      "step": 18598
    },
    {
      "epoch": 1.7736434936523437e-05,
      "step": 18598,
      "training_step_time": 7.6171042919158936
    },
    {
      "epoch": 1.7737388610839844e-05,
      "model_forward_time": 1.7221839427947998,
      "step": 18599
    },
    {
      "epoch": 1.7737388610839844e-05,
      "step": 18599,
      "training_step_time": 9.70120096206665
    },
    {
      "epoch": 1.773834228515625e-05,
      "grad_norm": 0.34757640957832336,
      "learning_rate": 9.502864455322808e-06,
      "loss": 0.061,
      "step": 18600
    },
    {
      "epoch": 1.773834228515625e-05,
      "model_forward_time": 1.7065417766571045,
      "step": 18600
    },
    {
      "dynamics_loss_avg": 0.052259168773889544,
      "epoch": 1.773834228515625e-05,
      "step": 18600
    },
    {
      "action_loss_avg": 0.002080157119780779,
      "epoch": 1.773834228515625e-05,
      "step": 18600
    },
    {
      "epoch": 1.773834228515625e-05,
      "step": 18600,
      "training_step_time": 9.397302150726318
    },
    {
      "epoch": 1.7739295959472655e-05,
      "model_forward_time": 2.264221429824829,
      "step": 18601
    },
    {
      "epoch": 1.7739295959472655e-05,
      "step": 18601,
      "training_step_time": 8.813257217407227
    },
    {
      "epoch": 1.7740249633789063e-05,
      "model_forward_time": 1.7080271244049072,
      "step": 18602
    },
    {
      "epoch": 1.7740249633789063e-05,
      "step": 18602,
      "training_step_time": 7.619413614273071
    },
    {
      "epoch": 1.774120330810547e-05,
      "model_forward_time": 1.7175309658050537,
      "step": 18603
    },
    {
      "epoch": 1.774120330810547e-05,
      "step": 18603,
      "training_step_time": 7.071049213409424
    },
    {
      "epoch": 1.7742156982421874e-05,
      "model_forward_time": 1.7087182998657227,
      "step": 18604
    },
    {
      "epoch": 1.7742156982421874e-05,
      "step": 18604,
      "training_step_time": 7.0865089893341064
    },
    {
      "epoch": 1.774311065673828e-05,
      "model_forward_time": 1.6827082633972168,
      "step": 18605
    },
    {
      "epoch": 1.774311065673828e-05,
      "step": 18605,
      "training_step_time": 7.176055669784546
    },
    {
      "epoch": 1.774406433105469e-05,
      "model_forward_time": 1.696302890777588,
      "step": 18606
    },
    {
      "epoch": 1.774406433105469e-05,
      "step": 18606,
      "training_step_time": 7.268900394439697
    },
    {
      "epoch": 1.7745018005371093e-05,
      "model_forward_time": 1.6992626190185547,
      "step": 18607
    },
    {
      "epoch": 1.7745018005371093e-05,
      "step": 18607,
      "training_step_time": 7.570716142654419
    },
    {
      "epoch": 1.77459716796875e-05,
      "model_forward_time": 1.7060730457305908,
      "step": 18608
    },
    {
      "epoch": 1.77459716796875e-05,
      "step": 18608,
      "training_step_time": 8.235894680023193
    },
    {
      "epoch": 1.7746925354003908e-05,
      "model_forward_time": 1.7109296321868896,
      "step": 18609
    },
    {
      "epoch": 1.7746925354003908e-05,
      "step": 18609,
      "training_step_time": 9.355463981628418
    },
    {
      "epoch": 1.7747879028320312e-05,
      "grad_norm": 0.18282711505889893,
      "learning_rate": 9.502145437466891e-06,
      "loss": 0.0674,
      "step": 18610
    },
    {
      "epoch": 1.7747879028320312e-05,
      "model_forward_time": 1.700446605682373,
      "step": 18610
    },
    {
      "dynamics_loss_avg": 0.06549483872950077,
      "epoch": 1.7747879028320312e-05,
      "step": 18610
    },
    {
      "action_loss_avg": 0.0026921503827907144,
      "epoch": 1.7747879028320312e-05,
      "step": 18610
    },
    {
      "epoch": 1.7747879028320312e-05,
      "step": 18610,
      "training_step_time": 8.841575622558594
    },
    {
      "epoch": 1.774883270263672e-05,
      "model_forward_time": 1.755739450454712,
      "step": 18611
    },
    {
      "epoch": 1.774883270263672e-05,
      "step": 18611,
      "training_step_time": 8.760392189025879
    },
    {
      "epoch": 1.7749786376953123e-05,
      "model_forward_time": 1.7081761360168457,
      "step": 18612
    },
    {
      "epoch": 1.7749786376953123e-05,
      "step": 18612,
      "training_step_time": 7.1550211906433105
    },
    {
      "epoch": 1.775074005126953e-05,
      "model_forward_time": 1.7073924541473389,
      "step": 18613
    },
    {
      "epoch": 1.775074005126953e-05,
      "step": 18613,
      "training_step_time": 7.094838380813599
    },
    {
      "epoch": 1.7751693725585938e-05,
      "model_forward_time": 1.710850715637207,
      "step": 18614
    },
    {
      "epoch": 1.7751693725585938e-05,
      "step": 18614,
      "training_step_time": 7.21243953704834
    },
    {
      "epoch": 1.7752647399902342e-05,
      "model_forward_time": 1.7108805179595947,
      "step": 18615
    },
    {
      "epoch": 1.7752647399902342e-05,
      "step": 18615,
      "training_step_time": 7.292057991027832
    },
    {
      "epoch": 1.775360107421875e-05,
      "model_forward_time": 1.701068639755249,
      "step": 18616
    },
    {
      "epoch": 1.775360107421875e-05,
      "step": 18616,
      "training_step_time": 8.127306938171387
    },
    {
      "epoch": 1.7754554748535157e-05,
      "model_forward_time": 1.7066526412963867,
      "step": 18617
    },
    {
      "epoch": 1.7754554748535157e-05,
      "step": 18617,
      "training_step_time": 7.6585540771484375
    },
    {
      "epoch": 1.775550842285156e-05,
      "model_forward_time": 1.7056961059570312,
      "step": 18618
    },
    {
      "epoch": 1.775550842285156e-05,
      "step": 18618,
      "training_step_time": 9.229761123657227
    },
    {
      "epoch": 1.775646209716797e-05,
      "model_forward_time": 1.707878828048706,
      "step": 18619
    },
    {
      "epoch": 1.775646209716797e-05,
      "step": 18619,
      "training_step_time": 8.302318811416626
    },
    {
      "epoch": 1.7757415771484376e-05,
      "grad_norm": 0.2539824843406677,
      "learning_rate": 9.501425927263117e-06,
      "loss": 0.0658,
      "step": 18620
    },
    {
      "epoch": 1.7757415771484376e-05,
      "model_forward_time": 2.25361967086792,
      "step": 18620
    },
    {
      "dynamics_loss_avg": 0.049812767468392846,
      "epoch": 1.7757415771484376e-05,
      "step": 18620
    },
    {
      "action_loss_avg": 0.0031532879977021365,
      "epoch": 1.7757415771484376e-05,
      "step": 18620
    },
    {
      "epoch": 1.7757415771484376e-05,
      "step": 18620,
      "training_step_time": 7.741681337356567
    },
    {
      "epoch": 1.775836944580078e-05,
      "model_forward_time": 1.706300973892212,
      "step": 18621
    },
    {
      "epoch": 1.775836944580078e-05,
      "step": 18621,
      "training_step_time": 7.5696470737457275
    },
    {
      "epoch": 1.7759323120117187e-05,
      "model_forward_time": 1.7075207233428955,
      "step": 18622
    },
    {
      "epoch": 1.7759323120117187e-05,
      "step": 18622,
      "training_step_time": 6.766037940979004
    },
    {
      "epoch": 1.7760276794433595e-05,
      "model_forward_time": 1.712571144104004,
      "step": 18623
    },
    {
      "epoch": 1.7760276794433595e-05,
      "step": 18623,
      "training_step_time": 7.1782546043396
    },
    {
      "epoch": 1.776123046875e-05,
      "model_forward_time": 1.7122743129730225,
      "step": 18624
    },
    {
      "epoch": 1.776123046875e-05,
      "step": 18624,
      "training_step_time": 7.354227781295776
    },
    {
      "epoch": 1.7762184143066406e-05,
      "model_forward_time": 1.7019708156585693,
      "step": 18625
    },
    {
      "epoch": 1.7762184143066406e-05,
      "step": 18625,
      "training_step_time": 7.664084196090698
    },
    {
      "epoch": 1.7763137817382814e-05,
      "model_forward_time": 1.702909231185913,
      "step": 18626
    },
    {
      "epoch": 1.7763137817382814e-05,
      "step": 18626,
      "training_step_time": 7.580060958862305
    },
    {
      "epoch": 1.7764091491699218e-05,
      "model_forward_time": 1.706845998764038,
      "step": 18627
    },
    {
      "epoch": 1.7764091491699218e-05,
      "step": 18627,
      "training_step_time": 8.175907373428345
    },
    {
      "epoch": 1.7765045166015625e-05,
      "model_forward_time": 1.7025701999664307,
      "step": 18628
    },
    {
      "epoch": 1.7765045166015625e-05,
      "step": 18628,
      "training_step_time": 9.501031398773193
    },
    {
      "epoch": 1.7765998840332033e-05,
      "model_forward_time": 2.2585082054138184,
      "step": 18629
    },
    {
      "epoch": 1.7765998840332033e-05,
      "step": 18629,
      "training_step_time": 7.742863178253174
    },
    {
      "epoch": 1.7766952514648437e-05,
      "grad_norm": 0.19704534113407135,
      "learning_rate": 9.500705924790173e-06,
      "loss": 0.0629,
      "step": 18630
    },
    {
      "epoch": 1.7766952514648437e-05,
      "model_forward_time": 1.7161333560943604,
      "step": 18630
    },
    {
      "dynamics_loss_avg": 0.055016358569264415,
      "epoch": 1.7766952514648437e-05,
      "step": 18630
    },
    {
      "action_loss_avg": 0.0030248699127696454,
      "epoch": 1.7766952514648437e-05,
      "step": 18630
    },
    {
      "epoch": 1.7766952514648437e-05,
      "step": 18630,
      "training_step_time": 7.578213691711426
    },
    {
      "epoch": 1.7767906188964844e-05,
      "model_forward_time": 1.7165844440460205,
      "step": 18631
    },
    {
      "epoch": 1.7767906188964844e-05,
      "step": 18631,
      "training_step_time": 6.7644195556640625
    },
    {
      "epoch": 1.776885986328125e-05,
      "model_forward_time": 1.7247180938720703,
      "step": 18632
    },
    {
      "epoch": 1.776885986328125e-05,
      "step": 18632,
      "training_step_time": 7.726835489273071
    },
    {
      "epoch": 1.7769813537597655e-05,
      "model_forward_time": 1.7207927703857422,
      "step": 18633
    },
    {
      "epoch": 1.7769813537597655e-05,
      "step": 18633,
      "training_step_time": 6.827526569366455
    },
    {
      "epoch": 1.7770767211914063e-05,
      "model_forward_time": 1.7105178833007812,
      "step": 18634
    },
    {
      "epoch": 1.7770767211914063e-05,
      "step": 18634,
      "training_step_time": 7.639592170715332
    },
    {
      "epoch": 1.777172088623047e-05,
      "model_forward_time": 1.7102267742156982,
      "step": 18635
    },
    {
      "epoch": 1.777172088623047e-05,
      "step": 18635,
      "training_step_time": 9.006993293762207
    },
    {
      "epoch": 1.7772674560546874e-05,
      "model_forward_time": 1.701106309890747,
      "step": 18636
    },
    {
      "epoch": 1.7772674560546874e-05,
      "step": 18636,
      "training_step_time": 7.750825881958008
    },
    {
      "epoch": 1.777362823486328e-05,
      "model_forward_time": 1.7000038623809814,
      "step": 18637
    },
    {
      "epoch": 1.777362823486328e-05,
      "step": 18637,
      "training_step_time": 8.829684495925903
    },
    {
      "epoch": 1.777458190917969e-05,
      "model_forward_time": 2.25700044631958,
      "step": 18638
    },
    {
      "epoch": 1.777458190917969e-05,
      "step": 18638,
      "training_step_time": 9.871946811676025
    },
    {
      "epoch": 1.7775535583496093e-05,
      "model_forward_time": 1.703378438949585,
      "step": 18639
    },
    {
      "epoch": 1.7775535583496093e-05,
      "step": 18639,
      "training_step_time": 7.665902853012085
    },
    {
      "epoch": 1.77764892578125e-05,
      "grad_norm": 0.1732533872127533,
      "learning_rate": 9.499985430126794e-06,
      "loss": 0.0652,
      "step": 18640
    },
    {
      "epoch": 1.77764892578125e-05,
      "model_forward_time": 1.6915779113769531,
      "step": 18640
    },
    {
      "dynamics_loss_avg": 0.05255517070181668,
      "epoch": 1.77764892578125e-05,
      "step": 18640
    },
    {
      "action_loss_avg": 0.002250105468556285,
      "epoch": 1.77764892578125e-05,
      "step": 18640
    },
    {
      "epoch": 1.77764892578125e-05,
      "step": 18640,
      "training_step_time": 7.574302673339844
    },
    {
      "epoch": 1.7777442932128905e-05,
      "model_forward_time": 1.6993887424468994,
      "step": 18641
    },
    {
      "epoch": 1.7777442932128905e-05,
      "step": 18641,
      "training_step_time": 7.203269720077515
    },
    {
      "epoch": 1.7778396606445312e-05,
      "model_forward_time": 1.7035751342773438,
      "step": 18642
    },
    {
      "epoch": 1.7778396606445312e-05,
      "step": 18642,
      "training_step_time": 7.150946617126465
    },
    {
      "epoch": 1.777935028076172e-05,
      "model_forward_time": 1.7065212726593018,
      "step": 18643
    },
    {
      "epoch": 1.777935028076172e-05,
      "step": 18643,
      "training_step_time": 7.716809988021851
    },
    {
      "epoch": 1.7780303955078123e-05,
      "model_forward_time": 1.6965739727020264,
      "step": 18644
    },
    {
      "epoch": 1.7780303955078123e-05,
      "step": 18644,
      "training_step_time": 7.647802114486694
    },
    {
      "epoch": 1.778125762939453e-05,
      "model_forward_time": 1.6985504627227783,
      "step": 18645
    },
    {
      "epoch": 1.778125762939453e-05,
      "step": 18645,
      "training_step_time": 9.224913597106934
    },
    {
      "epoch": 1.7782211303710938e-05,
      "model_forward_time": 1.6919283866882324,
      "step": 18646
    },
    {
      "epoch": 1.7782211303710938e-05,
      "step": 18646,
      "training_step_time": 9.247610807418823
    },
    {
      "epoch": 1.7783164978027342e-05,
      "model_forward_time": 2.238783836364746,
      "step": 18647
    },
    {
      "epoch": 1.7783164978027342e-05,
      "step": 18647,
      "training_step_time": 9.241032361984253
    },
    {
      "epoch": 1.778411865234375e-05,
      "model_forward_time": 1.6903274059295654,
      "step": 18648
    },
    {
      "epoch": 1.778411865234375e-05,
      "step": 18648,
      "training_step_time": 7.647484540939331
    },
    {
      "epoch": 1.7785072326660157e-05,
      "model_forward_time": 1.6965253353118896,
      "step": 18649
    },
    {
      "epoch": 1.7785072326660157e-05,
      "step": 18649,
      "training_step_time": 7.579521894454956
    },
    {
      "epoch": 1.778602600097656e-05,
      "grad_norm": 0.12863768637180328,
      "learning_rate": 9.499264443351775e-06,
      "loss": 0.0668,
      "step": 18650
    },
    {
      "epoch": 1.778602600097656e-05,
      "model_forward_time": 1.6976699829101562,
      "step": 18650
    },
    {
      "dynamics_loss_avg": 0.06404740996658802,
      "epoch": 1.778602600097656e-05,
      "step": 18650
    },
    {
      "action_loss_avg": 0.0022172531054820864,
      "epoch": 1.778602600097656e-05,
      "step": 18650
    },
    {
      "epoch": 1.778602600097656e-05,
      "step": 18650,
      "training_step_time": 7.2368128299713135
    },
    {
      "epoch": 1.778697967529297e-05,
      "model_forward_time": 1.6973729133605957,
      "step": 18651
    },
    {
      "epoch": 1.778697967529297e-05,
      "step": 18651,
      "training_step_time": 7.14043402671814
    },
    {
      "epoch": 1.7787933349609376e-05,
      "model_forward_time": 1.6949255466461182,
      "step": 18652
    },
    {
      "epoch": 1.7787933349609376e-05,
      "step": 18652,
      "training_step_time": 7.693524599075317
    },
    {
      "epoch": 1.778888702392578e-05,
      "model_forward_time": 1.6944482326507568,
      "step": 18653
    },
    {
      "epoch": 1.778888702392578e-05,
      "step": 18653,
      "training_step_time": 8.367277145385742
    },
    {
      "epoch": 1.7789840698242187e-05,
      "model_forward_time": 1.7157649993896484,
      "step": 18654
    },
    {
      "epoch": 1.7789840698242187e-05,
      "step": 18654,
      "training_step_time": 9.778905391693115
    },
    {
      "epoch": 1.7790794372558595e-05,
      "model_forward_time": 1.7110722064971924,
      "step": 18655
    },
    {
      "epoch": 1.7790794372558595e-05,
      "step": 18655,
      "training_step_time": 9.698280811309814
    },
    {
      "epoch": 1.7791748046875e-05,
      "model_forward_time": 2.245152235031128,
      "step": 18656
    },
    {
      "epoch": 1.7791748046875e-05,
      "step": 18656,
      "training_step_time": 8.871209621429443
    },
    {
      "epoch": 1.7792701721191406e-05,
      "model_forward_time": 1.7083265781402588,
      "step": 18657
    },
    {
      "epoch": 1.7792701721191406e-05,
      "step": 18657,
      "training_step_time": 9.008727073669434
    },
    {
      "epoch": 1.7793655395507814e-05,
      "model_forward_time": 1.698638916015625,
      "step": 18658
    },
    {
      "epoch": 1.7793655395507814e-05,
      "step": 18658,
      "training_step_time": 7.483210563659668
    },
    {
      "epoch": 1.7794609069824218e-05,
      "model_forward_time": 1.7191593647003174,
      "step": 18659
    },
    {
      "epoch": 1.7794609069824218e-05,
      "step": 18659,
      "training_step_time": 7.210601806640625
    },
    {
      "epoch": 1.7795562744140625e-05,
      "grad_norm": 0.15291225910186768,
      "learning_rate": 9.498542964543961e-06,
      "loss": 0.0674,
      "step": 18660
    },
    {
      "epoch": 1.7795562744140625e-05,
      "model_forward_time": 1.323547601699829,
      "step": 18660
    },
    {
      "dynamics_loss_avg": 0.055593525804579255,
      "epoch": 1.7795562744140625e-05,
      "step": 18660
    },
    {
      "action_loss_avg": 0.002072835754370317,
      "epoch": 1.7795562744140625e-05,
      "step": 18660
    },
    {
      "epoch": 1.7795562744140625e-05,
      "step": 18660,
      "training_step_time": 6.739828109741211
    },
    {
      "epoch": 1.7796516418457033e-05,
      "model_forward_time": 1.711548089981079,
      "step": 18661
    },
    {
      "epoch": 1.7796516418457033e-05,
      "step": 18661,
      "training_step_time": 7.632261276245117
    },
    {
      "epoch": 1.7797470092773437e-05,
      "model_forward_time": 1.7014591693878174,
      "step": 18662
    },
    {
      "epoch": 1.7797470092773437e-05,
      "step": 18662,
      "training_step_time": 7.617681980133057
    },
    {
      "epoch": 1.7798423767089844e-05,
      "model_forward_time": 1.7161808013916016,
      "step": 18663
    },
    {
      "epoch": 1.7798423767089844e-05,
      "step": 18663,
      "training_step_time": 9.03194785118103
    },
    {
      "epoch": 1.779937744140625e-05,
      "model_forward_time": 1.7039940357208252,
      "step": 18664
    },
    {
      "epoch": 1.779937744140625e-05,
      "step": 18664,
      "training_step_time": 7.694017648696899
    },
    {
      "epoch": 1.7800331115722655e-05,
      "model_forward_time": 1.715013027191162,
      "step": 18665
    },
    {
      "epoch": 1.7800331115722655e-05,
      "step": 18665,
      "training_step_time": 8.601126432418823
    },
    {
      "epoch": 1.7801284790039063e-05,
      "model_forward_time": 2.2437305450439453,
      "step": 18666
    },
    {
      "epoch": 1.7801284790039063e-05,
      "step": 18666,
      "training_step_time": 7.675569772720337
    },
    {
      "epoch": 1.780223846435547e-05,
      "model_forward_time": 1.7063298225402832,
      "step": 18667
    },
    {
      "epoch": 1.780223846435547e-05,
      "step": 18667,
      "training_step_time": 7.54742693901062
    },
    {
      "epoch": 1.7803192138671874e-05,
      "model_forward_time": 1.709367036819458,
      "step": 18668
    },
    {
      "epoch": 1.7803192138671874e-05,
      "step": 18668,
      "training_step_time": 7.5529234409332275
    },
    {
      "epoch": 1.7804145812988282e-05,
      "model_forward_time": 1.7014782428741455,
      "step": 18669
    },
    {
      "epoch": 1.7804145812988282e-05,
      "step": 18669,
      "training_step_time": 6.730867385864258
    },
    {
      "epoch": 1.780509948730469e-05,
      "grad_norm": 0.1419086456298828,
      "learning_rate": 9.497820993782251e-06,
      "loss": 0.0668,
      "step": 18670
    },
    {
      "epoch": 1.780509948730469e-05,
      "model_forward_time": 1.695011854171753,
      "step": 18670
    },
    {
      "dynamics_loss_avg": 0.05081502143293619,
      "epoch": 1.780509948730469e-05,
      "step": 18670
    },
    {
      "action_loss_avg": 0.0031641123117879034,
      "epoch": 1.780509948730469e-05,
      "step": 18670
    },
    {
      "epoch": 1.780509948730469e-05,
      "step": 18670,
      "training_step_time": 7.581419944763184
    },
    {
      "epoch": 1.7806053161621093e-05,
      "model_forward_time": 1.7003703117370605,
      "step": 18671
    },
    {
      "epoch": 1.7806053161621093e-05,
      "step": 18671,
      "training_step_time": 7.577372312545776
    },
    {
      "epoch": 1.78070068359375e-05,
      "model_forward_time": 1.701754093170166,
      "step": 18672
    },
    {
      "epoch": 1.78070068359375e-05,
      "step": 18672,
      "training_step_time": 7.527498960494995
    },
    {
      "epoch": 1.7807960510253905e-05,
      "model_forward_time": 1.698174238204956,
      "step": 18673
    },
    {
      "epoch": 1.7807960510253905e-05,
      "step": 18673,
      "training_step_time": 8.21636152267456
    },
    {
      "epoch": 1.7808914184570312e-05,
      "model_forward_time": 1.7097525596618652,
      "step": 18674
    },
    {
      "epoch": 1.7808914184570312e-05,
      "step": 18674,
      "training_step_time": 7.74483323097229
    },
    {
      "epoch": 1.780986785888672e-05,
      "model_forward_time": 2.252988576889038,
      "step": 18675
    },
    {
      "epoch": 1.780986785888672e-05,
      "step": 18675,
      "training_step_time": 7.762084245681763
    },
    {
      "epoch": 1.7810821533203123e-05,
      "model_forward_time": 1.7049400806427002,
      "step": 18676
    },
    {
      "epoch": 1.7810821533203123e-05,
      "step": 18676,
      "training_step_time": 7.607606887817383
    },
    {
      "epoch": 1.781177520751953e-05,
      "model_forward_time": 1.7061474323272705,
      "step": 18677
    },
    {
      "epoch": 1.781177520751953e-05,
      "step": 18677,
      "training_step_time": 7.584115743637085
    },
    {
      "epoch": 1.7812728881835938e-05,
      "model_forward_time": 1.7070214748382568,
      "step": 18678
    },
    {
      "epoch": 1.7812728881835938e-05,
      "step": 18678,
      "training_step_time": 7.251542806625366
    },
    {
      "epoch": 1.7813682556152342e-05,
      "model_forward_time": 1.7076563835144043,
      "step": 18679
    },
    {
      "epoch": 1.7813682556152342e-05,
      "step": 18679,
      "training_step_time": 7.6388328075408936
    },
    {
      "epoch": 1.781463623046875e-05,
      "grad_norm": 0.23595860600471497,
      "learning_rate": 9.497098531145602e-06,
      "loss": 0.0639,
      "step": 18680
    },
    {
      "epoch": 1.781463623046875e-05,
      "model_forward_time": 1.6894352436065674,
      "step": 18680
    },
    {
      "dynamics_loss_avg": 0.061032692156732085,
      "epoch": 1.781463623046875e-05,
      "step": 18680
    },
    {
      "action_loss_avg": 0.003812992200255394,
      "epoch": 1.781463623046875e-05,
      "step": 18680
    },
    {
      "epoch": 1.781463623046875e-05,
      "step": 18680,
      "training_step_time": 7.619582653045654
    },
    {
      "epoch": 1.7815589904785157e-05,
      "model_forward_time": 1.70951247215271,
      "step": 18681
    },
    {
      "epoch": 1.7815589904785157e-05,
      "step": 18681,
      "training_step_time": 9.294795751571655
    },
    {
      "epoch": 1.781654357910156e-05,
      "model_forward_time": 1.323195457458496,
      "step": 18682
    },
    {
      "epoch": 1.781654357910156e-05,
      "step": 18682,
      "training_step_time": 7.630741834640503
    },
    {
      "epoch": 1.781749725341797e-05,
      "model_forward_time": 1.7126121520996094,
      "step": 18683
    },
    {
      "epoch": 1.781749725341797e-05,
      "step": 18683,
      "training_step_time": 8.615435123443604
    },
    {
      "epoch": 1.7818450927734376e-05,
      "model_forward_time": 2.2593843936920166,
      "step": 18684
    },
    {
      "epoch": 1.7818450927734376e-05,
      "step": 18684,
      "training_step_time": 8.190076351165771
    },
    {
      "epoch": 1.781940460205078e-05,
      "model_forward_time": 1.7050001621246338,
      "step": 18685
    },
    {
      "epoch": 1.781940460205078e-05,
      "step": 18685,
      "training_step_time": 7.585183620452881
    },
    {
      "epoch": 1.7820358276367187e-05,
      "model_forward_time": 1.7044832706451416,
      "step": 18686
    },
    {
      "epoch": 1.7820358276367187e-05,
      "step": 18686,
      "training_step_time": 7.507335662841797
    },
    {
      "epoch": 1.7821311950683595e-05,
      "model_forward_time": 1.703023910522461,
      "step": 18687
    },
    {
      "epoch": 1.7821311950683595e-05,
      "step": 18687,
      "training_step_time": 7.183046340942383
    },
    {
      "epoch": 1.7822265625e-05,
      "model_forward_time": 1.702411413192749,
      "step": 18688
    },
    {
      "epoch": 1.7822265625e-05,
      "step": 18688,
      "training_step_time": 7.613136053085327
    },
    {
      "epoch": 1.7823219299316406e-05,
      "model_forward_time": 1.7090179920196533,
      "step": 18689
    },
    {
      "epoch": 1.7823219299316406e-05,
      "step": 18689,
      "training_step_time": 7.672645330429077
    },
    {
      "epoch": 1.7824172973632814e-05,
      "grad_norm": 0.19217441976070404,
      "learning_rate": 9.496375576713017e-06,
      "loss": 0.0632,
      "step": 18690
    },
    {
      "epoch": 1.7824172973632814e-05,
      "model_forward_time": 1.6913444995880127,
      "step": 18690
    },
    {
      "dynamics_loss_avg": 0.07866807170212269,
      "epoch": 1.7824172973632814e-05,
      "step": 18690
    },
    {
      "action_loss_avg": 0.0050971296266652645,
      "epoch": 1.7824172973632814e-05,
      "step": 18690
    },
    {
      "epoch": 1.7824172973632814e-05,
      "step": 18690,
      "training_step_time": 8.621010303497314
    },
    {
      "epoch": 1.7825126647949218e-05,
      "model_forward_time": 1.7067866325378418,
      "step": 18691
    },
    {
      "epoch": 1.7825126647949218e-05,
      "step": 18691,
      "training_step_time": 9.193583011627197
    },
    {
      "epoch": 1.7826080322265625e-05,
      "model_forward_time": 1.7021334171295166,
      "step": 18692
    },
    {
      "epoch": 1.7826080322265625e-05,
      "step": 18692,
      "training_step_time": 7.747828722000122
    },
    {
      "epoch": 1.7827033996582033e-05,
      "model_forward_time": 2.2703042030334473,
      "step": 18693
    },
    {
      "epoch": 1.7827033996582033e-05,
      "step": 18693,
      "training_step_time": 7.81025505065918
    },
    {
      "epoch": 1.7827987670898437e-05,
      "model_forward_time": 1.7182934284210205,
      "step": 18694
    },
    {
      "epoch": 1.7827987670898437e-05,
      "step": 18694,
      "training_step_time": 8.662968873977661
    },
    {
      "epoch": 1.7828941345214844e-05,
      "model_forward_time": 1.706585168838501,
      "step": 18695
    },
    {
      "epoch": 1.7828941345214844e-05,
      "step": 18695,
      "training_step_time": 7.440768003463745
    },
    {
      "epoch": 1.782989501953125e-05,
      "model_forward_time": 1.712169885635376,
      "step": 18696
    },
    {
      "epoch": 1.782989501953125e-05,
      "step": 18696,
      "training_step_time": 7.4195380210876465
    },
    {
      "epoch": 1.7830848693847655e-05,
      "model_forward_time": 1.7091984748840332,
      "step": 18697
    },
    {
      "epoch": 1.7830848693847655e-05,
      "step": 18697,
      "training_step_time": 7.623173475265503
    },
    {
      "epoch": 1.7831802368164063e-05,
      "model_forward_time": 1.706578254699707,
      "step": 18698
    },
    {
      "epoch": 1.7831802368164063e-05,
      "step": 18698,
      "training_step_time": 7.66510796546936
    },
    {
      "epoch": 1.783275604248047e-05,
      "model_forward_time": 1.709153175354004,
      "step": 18699
    },
    {
      "epoch": 1.783275604248047e-05,
      "step": 18699,
      "training_step_time": 8.441563367843628
    },
    {
      "epoch": 1.7833709716796874e-05,
      "grad_norm": 0.21119463443756104,
      "learning_rate": 9.49565213056356e-06,
      "loss": 0.0661,
      "step": 18700
    },
    {
      "epoch": 1.7833709716796874e-05,
      "model_forward_time": 1.7013680934906006,
      "step": 18700
    },
    {
      "dynamics_loss_avg": 0.04401804823428392,
      "epoch": 1.7833709716796874e-05,
      "step": 18700
    },
    {
      "action_loss_avg": 0.0024227048474131153,
      "epoch": 1.7833709716796874e-05,
      "step": 18700
    },
    {
      "epoch": 1.7833709716796874e-05,
      "step": 18700,
      "training_step_time": 7.608083248138428
    },
    {
      "epoch": 1.7834663391113282e-05,
      "model_forward_time": 1.711585283279419,
      "step": 18701
    },
    {
      "epoch": 1.7834663391113282e-05,
      "step": 18701,
      "training_step_time": 7.692379951477051
    },
    {
      "epoch": 1.783561706542969e-05,
      "model_forward_time": 1.7481913566589355,
      "step": 18702
    },
    {
      "epoch": 1.783561706542969e-05,
      "step": 18702,
      "training_step_time": 8.842835426330566
    },
    {
      "epoch": 1.7836570739746093e-05,
      "model_forward_time": 1.7130622863769531,
      "step": 18703
    },
    {
      "epoch": 1.7836570739746093e-05,
      "step": 18703,
      "training_step_time": 7.707529067993164
    },
    {
      "epoch": 1.78375244140625e-05,
      "model_forward_time": 1.7093117237091064,
      "step": 18704
    },
    {
      "epoch": 1.78375244140625e-05,
      "step": 18704,
      "training_step_time": 7.574029922485352
    },
    {
      "epoch": 1.7838478088378905e-05,
      "model_forward_time": 1.715205430984497,
      "step": 18705
    },
    {
      "epoch": 1.7838478088378905e-05,
      "step": 18705,
      "training_step_time": 7.2763965129852295
    },
    {
      "epoch": 1.7839431762695312e-05,
      "model_forward_time": 1.7088799476623535,
      "step": 18706
    },
    {
      "epoch": 1.7839431762695312e-05,
      "step": 18706,
      "training_step_time": 7.575624942779541
    },
    {
      "epoch": 1.784038543701172e-05,
      "model_forward_time": 1.7076349258422852,
      "step": 18707
    },
    {
      "epoch": 1.784038543701172e-05,
      "step": 18707,
      "training_step_time": 7.614684581756592
    },
    {
      "epoch": 1.7841339111328123e-05,
      "model_forward_time": 1.7130324840545654,
      "step": 18708
    },
    {
      "epoch": 1.7841339111328123e-05,
      "step": 18708,
      "training_step_time": 7.579428434371948
    },
    {
      "epoch": 1.784229278564453e-05,
      "model_forward_time": 1.7151870727539062,
      "step": 18709
    },
    {
      "epoch": 1.784229278564453e-05,
      "step": 18709,
      "training_step_time": 7.577274322509766
    },
    {
      "epoch": 1.784324645996094e-05,
      "grad_norm": 0.1489320695400238,
      "learning_rate": 9.494928192776343e-06,
      "loss": 0.0666,
      "step": 18710
    },
    {
      "epoch": 1.784324645996094e-05,
      "model_forward_time": 1.7039554119110107,
      "step": 18710
    },
    {
      "dynamics_loss_avg": 0.057983482629060744,
      "epoch": 1.784324645996094e-05,
      "step": 18710
    },
    {
      "action_loss_avg": 0.001984544878359884,
      "epoch": 1.784324645996094e-05,
      "step": 18710
    },
    {
      "epoch": 1.784324645996094e-05,
      "step": 18710,
      "training_step_time": 7.676431894302368
    },
    {
      "epoch": 1.7844200134277342e-05,
      "model_forward_time": 1.7235298156738281,
      "step": 18711
    },
    {
      "epoch": 1.7844200134277342e-05,
      "step": 18711,
      "training_step_time": 7.701549530029297
    },
    {
      "epoch": 1.784515380859375e-05,
      "model_forward_time": 2.2465622425079346,
      "step": 18712
    },
    {
      "epoch": 1.784515380859375e-05,
      "step": 18712,
      "training_step_time": 7.733535528182983
    },
    {
      "epoch": 1.7846107482910157e-05,
      "model_forward_time": 1.7101969718933105,
      "step": 18713
    },
    {
      "epoch": 1.7846107482910157e-05,
      "step": 18713,
      "training_step_time": 7.573759078979492
    },
    {
      "epoch": 1.784706115722656e-05,
      "model_forward_time": 1.708503246307373,
      "step": 18714
    },
    {
      "epoch": 1.784706115722656e-05,
      "step": 18714,
      "training_step_time": 7.491104602813721
    },
    {
      "epoch": 1.784801483154297e-05,
      "model_forward_time": 1.715346097946167,
      "step": 18715
    },
    {
      "epoch": 1.784801483154297e-05,
      "step": 18715,
      "training_step_time": 7.628260850906372
    },
    {
      "epoch": 1.7848968505859376e-05,
      "model_forward_time": 1.7083654403686523,
      "step": 18716
    },
    {
      "epoch": 1.7848968505859376e-05,
      "step": 18716,
      "training_step_time": 7.576564311981201
    },
    {
      "epoch": 1.784992218017578e-05,
      "model_forward_time": 1.7103331089019775,
      "step": 18717
    },
    {
      "epoch": 1.784992218017578e-05,
      "step": 18717,
      "training_step_time": 7.944471836090088
    },
    {
      "epoch": 1.7850875854492187e-05,
      "model_forward_time": 1.714829921722412,
      "step": 18718
    },
    {
      "epoch": 1.7850875854492187e-05,
      "step": 18718,
      "training_step_time": 7.691357374191284
    },
    {
      "epoch": 1.7851829528808595e-05,
      "model_forward_time": 1.709679126739502,
      "step": 18719
    },
    {
      "epoch": 1.7851829528808595e-05,
      "step": 18719,
      "training_step_time": 8.521249771118164
    },
    {
      "epoch": 1.7852783203125e-05,
      "grad_norm": 0.19723844528198242,
      "learning_rate": 9.494203763430539e-06,
      "loss": 0.0644,
      "step": 18720
    },
    {
      "epoch": 1.7852783203125e-05,
      "model_forward_time": 1.707418441772461,
      "step": 18720
    },
    {
      "dynamics_loss_avg": 0.06853252593427897,
      "epoch": 1.7852783203125e-05,
      "step": 18720
    },
    {
      "action_loss_avg": 0.0030290653347037734,
      "epoch": 1.7852783203125e-05,
      "step": 18720
    },
    {
      "epoch": 1.7852783203125e-05,
      "step": 18720,
      "training_step_time": 7.659433841705322
    },
    {
      "epoch": 1.7853736877441406e-05,
      "model_forward_time": 2.230299234390259,
      "step": 18721
    },
    {
      "epoch": 1.7853736877441406e-05,
      "step": 18721,
      "training_step_time": 10.050377368927002
    },
    {
      "epoch": 1.7854690551757814e-05,
      "model_forward_time": 1.7175683975219727,
      "step": 18722
    },
    {
      "epoch": 1.7854690551757814e-05,
      "step": 18722,
      "training_step_time": 8.595785140991211
    },
    {
      "epoch": 1.7855644226074218e-05,
      "model_forward_time": 1.7103805541992188,
      "step": 18723
    },
    {
      "epoch": 1.7855644226074218e-05,
      "step": 18723,
      "training_step_time": 7.483330488204956
    },
    {
      "epoch": 1.7856597900390625e-05,
      "model_forward_time": 1.6924633979797363,
      "step": 18724
    },
    {
      "epoch": 1.7856597900390625e-05,
      "step": 18724,
      "training_step_time": 7.571691036224365
    },
    {
      "epoch": 1.7857551574707033e-05,
      "model_forward_time": 1.705702781677246,
      "step": 18725
    },
    {
      "epoch": 1.7857551574707033e-05,
      "step": 18725,
      "training_step_time": 7.531047344207764
    },
    {
      "epoch": 1.7858505249023437e-05,
      "model_forward_time": 1.7032384872436523,
      "step": 18726
    },
    {
      "epoch": 1.7858505249023437e-05,
      "step": 18726,
      "training_step_time": 7.410979509353638
    },
    {
      "epoch": 1.7859458923339844e-05,
      "model_forward_time": 1.7047624588012695,
      "step": 18727
    },
    {
      "epoch": 1.7859458923339844e-05,
      "step": 18727,
      "training_step_time": 8.648343086242676
    },
    {
      "epoch": 1.786041259765625e-05,
      "model_forward_time": 1.7092647552490234,
      "step": 18728
    },
    {
      "epoch": 1.786041259765625e-05,
      "step": 18728,
      "training_step_time": 8.480888605117798
    },
    {
      "epoch": 1.7861366271972655e-05,
      "model_forward_time": 1.7064309120178223,
      "step": 18729
    },
    {
      "epoch": 1.7861366271972655e-05,
      "step": 18729,
      "training_step_time": 8.684409618377686
    },
    {
      "epoch": 1.7862319946289063e-05,
      "grad_norm": 0.1441134661436081,
      "learning_rate": 9.493478842605367e-06,
      "loss": 0.059,
      "step": 18730
    },
    {
      "epoch": 1.7862319946289063e-05,
      "model_forward_time": 2.2655601501464844,
      "step": 18730
    },
    {
      "dynamics_loss_avg": 0.04190825615078211,
      "epoch": 1.7862319946289063e-05,
      "step": 18730
    },
    {
      "action_loss_avg": 0.0016971345874480903,
      "epoch": 1.7862319946289063e-05,
      "step": 18730
    },
    {
      "epoch": 1.7862319946289063e-05,
      "step": 18730,
      "training_step_time": 8.650091648101807
    },
    {
      "epoch": 1.786327362060547e-05,
      "model_forward_time": 1.6996612548828125,
      "step": 18731
    },
    {
      "epoch": 1.786327362060547e-05,
      "step": 18731,
      "training_step_time": 7.6015801429748535
    },
    {
      "epoch": 1.7864227294921874e-05,
      "model_forward_time": 1.7127623558044434,
      "step": 18732
    },
    {
      "epoch": 1.7864227294921874e-05,
      "step": 18732,
      "training_step_time": 7.538919687271118
    },
    {
      "epoch": 1.7865180969238282e-05,
      "model_forward_time": 1.6885793209075928,
      "step": 18733
    },
    {
      "epoch": 1.7865180969238282e-05,
      "step": 18733,
      "training_step_time": 7.622313976287842
    },
    {
      "epoch": 1.786613464355469e-05,
      "model_forward_time": 1.7218470573425293,
      "step": 18734
    },
    {
      "epoch": 1.786613464355469e-05,
      "step": 18734,
      "training_step_time": 8.441969394683838
    },
    {
      "epoch": 1.7867088317871093e-05,
      "model_forward_time": 1.7152411937713623,
      "step": 18735
    },
    {
      "epoch": 1.7867088317871093e-05,
      "step": 18735,
      "training_step_time": 8.188624382019043
    },
    {
      "epoch": 1.78680419921875e-05,
      "model_forward_time": 1.7038545608520508,
      "step": 18736
    },
    {
      "epoch": 1.78680419921875e-05,
      "step": 18736,
      "training_step_time": 7.574227571487427
    },
    {
      "epoch": 1.7868995666503905e-05,
      "model_forward_time": 1.7028591632843018,
      "step": 18737
    },
    {
      "epoch": 1.7868995666503905e-05,
      "step": 18737,
      "training_step_time": 9.781444311141968
    },
    {
      "epoch": 1.7869949340820312e-05,
      "model_forward_time": 1.7033138275146484,
      "step": 18738
    },
    {
      "epoch": 1.7869949340820312e-05,
      "step": 18738,
      "training_step_time": 10.220994710922241
    },
    {
      "epoch": 1.787090301513672e-05,
      "model_forward_time": 2.279554843902588,
      "step": 18739
    },
    {
      "epoch": 1.787090301513672e-05,
      "step": 18739,
      "training_step_time": 8.287992477416992
    },
    {
      "epoch": 1.7871856689453124e-05,
      "grad_norm": 0.2228485345840454,
      "learning_rate": 9.492753430380107e-06,
      "loss": 0.0641,
      "step": 18740
    },
    {
      "epoch": 1.7871856689453124e-05,
      "model_forward_time": 1.7015280723571777,
      "step": 18740
    },
    {
      "dynamics_loss_avg": 0.07322164252400398,
      "epoch": 1.7871856689453124e-05,
      "step": 18740
    },
    {
      "action_loss_avg": 0.0022119305620435627,
      "epoch": 1.7871856689453124e-05,
      "step": 18740
    },
    {
      "epoch": 1.7871856689453124e-05,
      "step": 18740,
      "training_step_time": 7.7583818435668945
    },
    {
      "epoch": 1.787281036376953e-05,
      "model_forward_time": 1.7132561206817627,
      "step": 18741
    },
    {
      "epoch": 1.787281036376953e-05,
      "step": 18741,
      "training_step_time": 7.526430130004883
    },
    {
      "epoch": 1.787376403808594e-05,
      "model_forward_time": 1.716289758682251,
      "step": 18742
    },
    {
      "epoch": 1.787376403808594e-05,
      "step": 18742,
      "training_step_time": 7.634474039077759
    },
    {
      "epoch": 1.7874717712402342e-05,
      "model_forward_time": 1.7108302116394043,
      "step": 18743
    },
    {
      "epoch": 1.7874717712402342e-05,
      "step": 18743,
      "training_step_time": 6.7711968421936035
    },
    {
      "epoch": 1.787567138671875e-05,
      "model_forward_time": 1.7091050148010254,
      "step": 18744
    },
    {
      "epoch": 1.787567138671875e-05,
      "step": 18744,
      "training_step_time": 7.547090291976929
    },
    {
      "epoch": 1.7876625061035157e-05,
      "model_forward_time": 1.718214511871338,
      "step": 18745
    },
    {
      "epoch": 1.7876625061035157e-05,
      "step": 18745,
      "training_step_time": 7.569571018218994
    },
    {
      "epoch": 1.787757873535156e-05,
      "model_forward_time": 1.7172493934631348,
      "step": 18746
    },
    {
      "epoch": 1.787757873535156e-05,
      "step": 18746,
      "training_step_time": 8.298972845077515
    },
    {
      "epoch": 1.787853240966797e-05,
      "model_forward_time": 1.7133948802947998,
      "step": 18747
    },
    {
      "epoch": 1.787853240966797e-05,
      "step": 18747,
      "training_step_time": 8.119213342666626
    },
    {
      "epoch": 1.7879486083984376e-05,
      "model_forward_time": 2.2705466747283936,
      "step": 18748
    },
    {
      "epoch": 1.7879486083984376e-05,
      "step": 18748,
      "training_step_time": 7.689871788024902
    },
    {
      "epoch": 1.788043975830078e-05,
      "model_forward_time": 1.7118792533874512,
      "step": 18749
    },
    {
      "epoch": 1.788043975830078e-05,
      "step": 18749,
      "training_step_time": 7.736669540405273
    },
    {
      "epoch": 1.7881393432617188e-05,
      "grad_norm": 0.30394500494003296,
      "learning_rate": 9.492027526834084e-06,
      "loss": 0.0604,
      "step": 18750
    },
    {
      "epoch": 1.7881393432617188e-05,
      "model_forward_time": 1.7060928344726562,
      "step": 18750
    },
    {
      "dynamics_loss_avg": 0.05673100538551808,
      "epoch": 1.7881393432617188e-05,
      "step": 18750
    },
    {
      "action_loss_avg": 0.0034605361695867034,
      "epoch": 1.7881393432617188e-05,
      "step": 18750
    },
    {
      "epoch": 1.7881393432617188e-05,
      "step": 18750,
      "training_step_time": 8.59286880493164
    },
    {
      "epoch": 1.7882347106933595e-05,
      "model_forward_time": 1.7374098300933838,
      "step": 18751
    },
    {
      "epoch": 1.7882347106933595e-05,
      "step": 18751,
      "training_step_time": 7.692205905914307
    },
    {
      "epoch": 1.788330078125e-05,
      "model_forward_time": 1.71053147315979,
      "step": 18752
    },
    {
      "epoch": 1.788330078125e-05,
      "step": 18752,
      "training_step_time": 6.742354869842529
    },
    {
      "epoch": 1.7884254455566406e-05,
      "model_forward_time": 1.7142252922058105,
      "step": 18753
    },
    {
      "epoch": 1.7884254455566406e-05,
      "step": 18753,
      "training_step_time": 7.669499158859253
    },
    {
      "epoch": 1.7885208129882814e-05,
      "model_forward_time": 1.7160143852233887,
      "step": 18754
    },
    {
      "epoch": 1.7885208129882814e-05,
      "step": 18754,
      "training_step_time": 7.40930700302124
    },
    {
      "epoch": 1.7886161804199218e-05,
      "model_forward_time": 1.7089426517486572,
      "step": 18755
    },
    {
      "epoch": 1.7886161804199218e-05,
      "step": 18755,
      "training_step_time": 7.67779541015625
    },
    {
      "epoch": 1.7887115478515625e-05,
      "model_forward_time": 1.7108290195465088,
      "step": 18756
    },
    {
      "epoch": 1.7887115478515625e-05,
      "step": 18756,
      "training_step_time": 7.592497825622559
    },
    {
      "epoch": 1.7888069152832033e-05,
      "model_forward_time": 1.7267045974731445,
      "step": 18757
    },
    {
      "epoch": 1.7888069152832033e-05,
      "step": 18757,
      "training_step_time": 8.463854551315308
    },
    {
      "epoch": 1.7889022827148437e-05,
      "model_forward_time": 2.2465310096740723,
      "step": 18758
    },
    {
      "epoch": 1.7889022827148437e-05,
      "step": 18758,
      "training_step_time": 8.563371896743774
    },
    {
      "epoch": 1.7889976501464844e-05,
      "model_forward_time": 1.710855484008789,
      "step": 18759
    },
    {
      "epoch": 1.7889976501464844e-05,
      "step": 18759,
      "training_step_time": 7.544825315475464
    },
    {
      "epoch": 1.789093017578125e-05,
      "grad_norm": 0.17642782628536224,
      "learning_rate": 9.491301132046685e-06,
      "loss": 0.061,
      "step": 18760
    },
    {
      "epoch": 1.789093017578125e-05,
      "model_forward_time": 1.7136893272399902,
      "step": 18760
    },
    {
      "dynamics_loss_avg": 0.062318320758640765,
      "epoch": 1.789093017578125e-05,
      "step": 18760
    },
    {
      "action_loss_avg": 0.003061255151988007,
      "epoch": 1.789093017578125e-05,
      "step": 18760
    },
    {
      "epoch": 1.789093017578125e-05,
      "step": 18760,
      "training_step_time": 7.6046905517578125
    },
    {
      "epoch": 1.7891883850097656e-05,
      "model_forward_time": 1.713782548904419,
      "step": 18761
    },
    {
      "epoch": 1.7891883850097656e-05,
      "step": 18761,
      "training_step_time": 6.740299701690674
    },
    {
      "epoch": 1.7892837524414063e-05,
      "model_forward_time": 1.7102768421173096,
      "step": 18762
    },
    {
      "epoch": 1.7892837524414063e-05,
      "step": 18762,
      "training_step_time": 8.11975383758545
    },
    {
      "epoch": 1.789379119873047e-05,
      "model_forward_time": 1.7179372310638428,
      "step": 18763
    },
    {
      "epoch": 1.789379119873047e-05,
      "step": 18763,
      "training_step_time": 7.941380739212036
    },
    {
      "epoch": 1.7894744873046874e-05,
      "model_forward_time": 1.716996431350708,
      "step": 18764
    },
    {
      "epoch": 1.7894744873046874e-05,
      "step": 18764,
      "training_step_time": 8.946698188781738
    },
    {
      "epoch": 1.7895698547363282e-05,
      "model_forward_time": 1.7181391716003418,
      "step": 18765
    },
    {
      "epoch": 1.7895698547363282e-05,
      "step": 18765,
      "training_step_time": 9.468831539154053
    },
    {
      "epoch": 1.7896652221679686e-05,
      "model_forward_time": 1.7102997303009033,
      "step": 18766
    },
    {
      "epoch": 1.7896652221679686e-05,
      "step": 18766,
      "training_step_time": 8.115352869033813
    },
    {
      "epoch": 1.7897605895996093e-05,
      "model_forward_time": 2.2859973907470703,
      "step": 18767
    },
    {
      "epoch": 1.7897605895996093e-05,
      "step": 18767,
      "training_step_time": 7.603400468826294
    },
    {
      "epoch": 1.78985595703125e-05,
      "model_forward_time": 1.7093322277069092,
      "step": 18768
    },
    {
      "epoch": 1.78985595703125e-05,
      "step": 18768,
      "training_step_time": 7.5479466915130615
    },
    {
      "epoch": 1.7899513244628905e-05,
      "model_forward_time": 1.7144551277160645,
      "step": 18769
    },
    {
      "epoch": 1.7899513244628905e-05,
      "step": 18769,
      "training_step_time": 8.423638582229614
    },
    {
      "epoch": 1.7900466918945312e-05,
      "grad_norm": 0.16883115470409393,
      "learning_rate": 9.490574246097346e-06,
      "loss": 0.0619,
      "step": 18770
    },
    {
      "epoch": 1.7900466918945312e-05,
      "model_forward_time": 1.7126522064208984,
      "step": 18770
    },
    {
      "dynamics_loss_avg": 0.051259015873074534,
      "epoch": 1.7900466918945312e-05,
      "step": 18770
    },
    {
      "action_loss_avg": 0.003734505723696202,
      "epoch": 1.7900466918945312e-05,
      "step": 18770
    },
    {
      "epoch": 1.7900466918945312e-05,
      "step": 18770,
      "training_step_time": 6.716200113296509
    },
    {
      "epoch": 1.790142059326172e-05,
      "model_forward_time": 1.7112796306610107,
      "step": 18771
    },
    {
      "epoch": 1.790142059326172e-05,
      "step": 18771,
      "training_step_time": 7.51384711265564
    },
    {
      "epoch": 1.7902374267578124e-05,
      "model_forward_time": 1.7080497741699219,
      "step": 18772
    },
    {
      "epoch": 1.7902374267578124e-05,
      "step": 18772,
      "training_step_time": 7.398345947265625
    },
    {
      "epoch": 1.790332794189453e-05,
      "model_forward_time": 1.7081222534179688,
      "step": 18773
    },
    {
      "epoch": 1.790332794189453e-05,
      "step": 18773,
      "training_step_time": 8.321967840194702
    },
    {
      "epoch": 1.790428161621094e-05,
      "model_forward_time": 1.7107045650482178,
      "step": 18774
    },
    {
      "epoch": 1.790428161621094e-05,
      "step": 18774,
      "training_step_time": 7.555135488510132
    },
    {
      "epoch": 1.7905235290527342e-05,
      "model_forward_time": 1.7134582996368408,
      "step": 18775
    },
    {
      "epoch": 1.7905235290527342e-05,
      "step": 18775,
      "training_step_time": 8.145503997802734
    },
    {
      "epoch": 1.790618896484375e-05,
      "model_forward_time": 2.285057783126831,
      "step": 18776
    },
    {
      "epoch": 1.790618896484375e-05,
      "step": 18776,
      "training_step_time": 8.163820028305054
    },
    {
      "epoch": 1.7907142639160157e-05,
      "model_forward_time": 1.7131974697113037,
      "step": 18777
    },
    {
      "epoch": 1.7907142639160157e-05,
      "step": 18777,
      "training_step_time": 8.132355451583862
    },
    {
      "epoch": 1.790809631347656e-05,
      "model_forward_time": 1.7114715576171875,
      "step": 18778
    },
    {
      "epoch": 1.790809631347656e-05,
      "step": 18778,
      "training_step_time": 7.661338567733765
    },
    {
      "epoch": 1.790904998779297e-05,
      "model_forward_time": 1.7201130390167236,
      "step": 18779
    },
    {
      "epoch": 1.790904998779297e-05,
      "step": 18779,
      "training_step_time": 7.1195313930511475
    },
    {
      "epoch": 1.7910003662109376e-05,
      "grad_norm": 0.1802886575460434,
      "learning_rate": 9.48984686906556e-06,
      "loss": 0.0624,
      "step": 18780
    },
    {
      "epoch": 1.7910003662109376e-05,
      "model_forward_time": 1.7090880870819092,
      "step": 18780
    },
    {
      "dynamics_loss_avg": 0.05982688087970019,
      "epoch": 1.7910003662109376e-05,
      "step": 18780
    },
    {
      "action_loss_avg": 0.003332862607203424,
      "epoch": 1.7910003662109376e-05,
      "step": 18780
    },
    {
      "epoch": 1.7910003662109376e-05,
      "step": 18780,
      "training_step_time": 7.323989391326904
    },
    {
      "epoch": 1.791095733642578e-05,
      "model_forward_time": 1.711676836013794,
      "step": 18781
    },
    {
      "epoch": 1.791095733642578e-05,
      "step": 18781,
      "training_step_time": 7.586116313934326
    },
    {
      "epoch": 1.7911911010742188e-05,
      "model_forward_time": 1.706031084060669,
      "step": 18782
    },
    {
      "epoch": 1.7911911010742188e-05,
      "step": 18782,
      "training_step_time": 9.485668182373047
    },
    {
      "epoch": 1.7912864685058595e-05,
      "model_forward_time": 1.7110090255737305,
      "step": 18783
    },
    {
      "epoch": 1.7912864685058595e-05,
      "step": 18783,
      "training_step_time": 9.057704210281372
    },
    {
      "epoch": 1.7913818359375e-05,
      "model_forward_time": 1.7099666595458984,
      "step": 18784
    },
    {
      "epoch": 1.7913818359375e-05,
      "step": 18784,
      "training_step_time": 8.456807136535645
    },
    {
      "epoch": 1.7914772033691406e-05,
      "model_forward_time": 2.246879816055298,
      "step": 18785
    },
    {
      "epoch": 1.7914772033691406e-05,
      "step": 18785,
      "training_step_time": 8.170338153839111
    },
    {
      "epoch": 1.7915725708007814e-05,
      "model_forward_time": 1.7113144397735596,
      "step": 18786
    },
    {
      "epoch": 1.7915725708007814e-05,
      "step": 18786,
      "training_step_time": 7.608175039291382
    },
    {
      "epoch": 1.7916679382324218e-05,
      "model_forward_time": 1.70365309715271,
      "step": 18787
    },
    {
      "epoch": 1.7916679382324218e-05,
      "step": 18787,
      "training_step_time": 7.591441869735718
    },
    {
      "epoch": 1.7917633056640625e-05,
      "model_forward_time": 1.7047934532165527,
      "step": 18788
    },
    {
      "epoch": 1.7917633056640625e-05,
      "step": 18788,
      "training_step_time": 7.564424991607666
    },
    {
      "epoch": 1.7918586730957033e-05,
      "model_forward_time": 1.707683801651001,
      "step": 18789
    },
    {
      "epoch": 1.7918586730957033e-05,
      "step": 18789,
      "training_step_time": 7.2539167404174805
    },
    {
      "epoch": 1.7919540405273437e-05,
      "grad_norm": 0.21057197451591492,
      "learning_rate": 9.48911900103087e-06,
      "loss": 0.0662,
      "step": 18790
    },
    {
      "epoch": 1.7919540405273437e-05,
      "model_forward_time": 1.7191340923309326,
      "step": 18790
    },
    {
      "dynamics_loss_avg": 0.05378443393856287,
      "epoch": 1.7919540405273437e-05,
      "step": 18790
    },
    {
      "action_loss_avg": 0.0023138330318033693,
      "epoch": 1.7919540405273437e-05,
      "step": 18790
    },
    {
      "epoch": 1.7919540405273437e-05,
      "step": 18790,
      "training_step_time": 7.5738136768341064
    },
    {
      "epoch": 1.7920494079589844e-05,
      "model_forward_time": 1.7146368026733398,
      "step": 18791
    },
    {
      "epoch": 1.7920494079589844e-05,
      "step": 18791,
      "training_step_time": 8.327407836914062
    },
    {
      "epoch": 1.792144775390625e-05,
      "model_forward_time": 1.7125587463378906,
      "step": 18792
    },
    {
      "epoch": 1.792144775390625e-05,
      "step": 18792,
      "training_step_time": 8.065993309020996
    },
    {
      "epoch": 1.7922401428222656e-05,
      "model_forward_time": 1.7180254459381104,
      "step": 18793
    },
    {
      "epoch": 1.7922401428222656e-05,
      "step": 18793,
      "training_step_time": 8.399552822113037
    },
    {
      "epoch": 1.7923355102539063e-05,
      "model_forward_time": 2.2594549655914307,
      "step": 18794
    },
    {
      "epoch": 1.7923355102539063e-05,
      "step": 18794,
      "training_step_time": 7.700246572494507
    },
    {
      "epoch": 1.792430877685547e-05,
      "model_forward_time": 1.7263429164886475,
      "step": 18795
    },
    {
      "epoch": 1.792430877685547e-05,
      "step": 18795,
      "training_step_time": 7.6211488246917725
    },
    {
      "epoch": 1.7925262451171874e-05,
      "model_forward_time": 1.719071388244629,
      "step": 18796
    },
    {
      "epoch": 1.7925262451171874e-05,
      "step": 18796,
      "training_step_time": 7.591488838195801
    },
    {
      "epoch": 1.7926216125488282e-05,
      "model_forward_time": 1.7121765613555908,
      "step": 18797
    },
    {
      "epoch": 1.7926216125488282e-05,
      "step": 18797,
      "training_step_time": 7.588874340057373
    },
    {
      "epoch": 1.7927169799804686e-05,
      "model_forward_time": 1.7167015075683594,
      "step": 18798
    },
    {
      "epoch": 1.7927169799804686e-05,
      "step": 18798,
      "training_step_time": 7.325861692428589
    },
    {
      "epoch": 1.7928123474121093e-05,
      "model_forward_time": 1.714357614517212,
      "step": 18799
    },
    {
      "epoch": 1.7928123474121093e-05,
      "step": 18799,
      "training_step_time": 7.639495372772217
    },
    {
      "epoch": 1.79290771484375e-05,
      "grad_norm": 0.24128006398677826,
      "learning_rate": 9.488390642072878e-06,
      "loss": 0.0643,
      "step": 18800
    },
    {
      "epoch": 1.79290771484375e-05,
      "model_forward_time": 1.719085454940796,
      "step": 18800
    },
    {
      "dynamics_loss_avg": 0.057338199205696584,
      "epoch": 1.79290771484375e-05,
      "step": 18800
    },
    {
      "action_loss_avg": 0.0027237353497184813,
      "epoch": 1.79290771484375e-05,
      "step": 18800
    },
    {
      "epoch": 1.79290771484375e-05,
      "step": 18800,
      "training_step_time": 8.97705864906311
    },
    {
      "epoch": 1.7930030822753905e-05,
      "model_forward_time": 1.7182772159576416,
      "step": 18801
    },
    {
      "epoch": 1.7930030822753905e-05,
      "step": 18801,
      "training_step_time": 7.663908004760742
    },
    {
      "epoch": 1.7930984497070312e-05,
      "model_forward_time": 1.720353364944458,
      "step": 18802
    },
    {
      "epoch": 1.7930984497070312e-05,
      "step": 18802,
      "training_step_time": 8.365917444229126
    },
    {
      "epoch": 1.793193817138672e-05,
      "model_forward_time": 1.7097814083099365,
      "step": 18803
    },
    {
      "epoch": 1.793193817138672e-05,
      "step": 18803,
      "training_step_time": 9.47829556465149
    },
    {
      "epoch": 1.7932891845703124e-05,
      "model_forward_time": 1.7144465446472168,
      "step": 18804
    },
    {
      "epoch": 1.7932891845703124e-05,
      "step": 18804,
      "training_step_time": 8.16348123550415
    },
    {
      "epoch": 1.793384552001953e-05,
      "model_forward_time": 1.7126495838165283,
      "step": 18805
    },
    {
      "epoch": 1.793384552001953e-05,
      "step": 18805,
      "training_step_time": 7.603906869888306
    },
    {
      "epoch": 1.793479919433594e-05,
      "model_forward_time": 1.7190053462982178,
      "step": 18806
    },
    {
      "epoch": 1.793479919433594e-05,
      "step": 18806,
      "training_step_time": 7.7315943241119385
    },
    {
      "epoch": 1.7935752868652342e-05,
      "model_forward_time": 1.7140929698944092,
      "step": 18807
    },
    {
      "epoch": 1.7935752868652342e-05,
      "step": 18807,
      "training_step_time": 7.33453369140625
    },
    {
      "epoch": 1.793670654296875e-05,
      "model_forward_time": 1.7124791145324707,
      "step": 18808
    },
    {
      "epoch": 1.793670654296875e-05,
      "step": 18808,
      "training_step_time": 7.633537769317627
    },
    {
      "epoch": 1.7937660217285157e-05,
      "model_forward_time": 1.7223281860351562,
      "step": 18809
    },
    {
      "epoch": 1.7937660217285157e-05,
      "step": 18809,
      "training_step_time": 7.458014726638794
    },
    {
      "epoch": 1.793861389160156e-05,
      "grad_norm": 0.2515372037887573,
      "learning_rate": 9.48766179227123e-06,
      "loss": 0.0635,
      "step": 18810
    },
    {
      "epoch": 1.793861389160156e-05,
      "model_forward_time": 1.7184271812438965,
      "step": 18810
    },
    {
      "dynamics_loss_avg": 0.056584162265062334,
      "epoch": 1.793861389160156e-05,
      "step": 18810
    },
    {
      "action_loss_avg": 0.0019179461989551783,
      "epoch": 1.793861389160156e-05,
      "step": 18810
    },
    {
      "epoch": 1.793861389160156e-05,
      "step": 18810,
      "training_step_time": 8.72799277305603
    },
    {
      "epoch": 1.793956756591797e-05,
      "model_forward_time": 1.7096097469329834,
      "step": 18811
    },
    {
      "epoch": 1.793956756591797e-05,
      "step": 18811,
      "training_step_time": 7.611922740936279
    },
    {
      "epoch": 1.7940521240234376e-05,
      "model_forward_time": 1.7175164222717285,
      "step": 18812
    },
    {
      "epoch": 1.7940521240234376e-05,
      "step": 18812,
      "training_step_time": 9.765841484069824
    },
    {
      "epoch": 1.794147491455078e-05,
      "model_forward_time": 1.7555768489837646,
      "step": 18813
    },
    {
      "epoch": 1.794147491455078e-05,
      "step": 18813,
      "training_step_time": 7.608657598495483
    },
    {
      "epoch": 1.7942428588867188e-05,
      "model_forward_time": 1.7148115634918213,
      "step": 18814
    },
    {
      "epoch": 1.7942428588867188e-05,
      "step": 18814,
      "training_step_time": 7.5985870361328125
    },
    {
      "epoch": 1.7943382263183595e-05,
      "model_forward_time": 1.7145054340362549,
      "step": 18815
    },
    {
      "epoch": 1.7943382263183595e-05,
      "step": 18815,
      "training_step_time": 7.698945045471191
    },
    {
      "epoch": 1.79443359375e-05,
      "model_forward_time": 1.3340237140655518,
      "step": 18816
    },
    {
      "epoch": 1.79443359375e-05,
      "step": 18816,
      "training_step_time": 7.353699207305908
    },
    {
      "epoch": 1.7945289611816406e-05,
      "model_forward_time": 1.7205328941345215,
      "step": 18817
    },
    {
      "epoch": 1.7945289611816406e-05,
      "step": 18817,
      "training_step_time": 7.468512773513794
    },
    {
      "epoch": 1.7946243286132814e-05,
      "model_forward_time": 1.6929690837860107,
      "step": 18818
    },
    {
      "epoch": 1.7946243286132814e-05,
      "step": 18818,
      "training_step_time": 7.570899724960327
    },
    {
      "epoch": 1.7947196960449218e-05,
      "model_forward_time": 1.7071678638458252,
      "step": 18819
    },
    {
      "epoch": 1.7947196960449218e-05,
      "step": 18819,
      "training_step_time": 9.39549469947815
    },
    {
      "epoch": 1.7948150634765625e-05,
      "grad_norm": 0.21774332225322723,
      "learning_rate": 9.486932451705637e-06,
      "loss": 0.0654,
      "step": 18820
    },
    {
      "epoch": 1.7948150634765625e-05,
      "model_forward_time": 1.7065956592559814,
      "step": 18820
    },
    {
      "dynamics_loss_avg": 0.05517588490620255,
      "epoch": 1.7948150634765625e-05,
      "step": 18820
    },
    {
      "action_loss_avg": 0.0017430006759241224,
      "epoch": 1.7948150634765625e-05,
      "step": 18820
    },
    {
      "epoch": 1.7948150634765625e-05,
      "step": 18820,
      "training_step_time": 8.680826663970947
    },
    {
      "epoch": 1.7949104309082033e-05,
      "model_forward_time": 1.7089920043945312,
      "step": 18821
    },
    {
      "epoch": 1.7949104309082033e-05,
      "step": 18821,
      "training_step_time": 7.657363653182983
    },
    {
      "epoch": 1.7950057983398437e-05,
      "model_forward_time": 2.2854321002960205,
      "step": 18822
    },
    {
      "epoch": 1.7950057983398437e-05,
      "step": 18822,
      "training_step_time": 9.245198965072632
    },
    {
      "epoch": 1.7951011657714844e-05,
      "model_forward_time": 1.7055761814117432,
      "step": 18823
    },
    {
      "epoch": 1.7951011657714844e-05,
      "step": 18823,
      "training_step_time": 7.61353325843811
    },
    {
      "epoch": 1.795196533203125e-05,
      "model_forward_time": 1.3324451446533203,
      "step": 18824
    },
    {
      "epoch": 1.795196533203125e-05,
      "step": 18824,
      "training_step_time": 7.567636728286743
    },
    {
      "epoch": 1.7952919006347656e-05,
      "model_forward_time": 1.7038068771362305,
      "step": 18825
    },
    {
      "epoch": 1.7952919006347656e-05,
      "step": 18825,
      "training_step_time": 7.438827276229858
    },
    {
      "epoch": 1.7953872680664063e-05,
      "model_forward_time": 1.6986298561096191,
      "step": 18826
    },
    {
      "epoch": 1.7953872680664063e-05,
      "step": 18826,
      "training_step_time": 7.332131385803223
    },
    {
      "epoch": 1.795482635498047e-05,
      "model_forward_time": 1.7069649696350098,
      "step": 18827
    },
    {
      "epoch": 1.795482635498047e-05,
      "step": 18827,
      "training_step_time": 8.386001110076904
    },
    {
      "epoch": 1.7955780029296874e-05,
      "model_forward_time": 1.6997113227844238,
      "step": 18828
    },
    {
      "epoch": 1.7955780029296874e-05,
      "step": 18828,
      "training_step_time": 8.482962846755981
    },
    {
      "epoch": 1.7956733703613282e-05,
      "model_forward_time": 1.7088589668273926,
      "step": 18829
    },
    {
      "epoch": 1.7956733703613282e-05,
      "step": 18829,
      "training_step_time": 9.659950256347656
    },
    {
      "epoch": 1.7957687377929686e-05,
      "grad_norm": 0.2174108624458313,
      "learning_rate": 9.486202620455858e-06,
      "loss": 0.0623,
      "step": 18830
    },
    {
      "epoch": 1.7957687377929686e-05,
      "model_forward_time": 1.6977202892303467,
      "step": 18830
    },
    {
      "dynamics_loss_avg": 0.04808501023799181,
      "epoch": 1.7957687377929686e-05,
      "step": 18830
    },
    {
      "action_loss_avg": 0.0020256748422980307,
      "epoch": 1.7957687377929686e-05,
      "step": 18830
    },
    {
      "epoch": 1.7957687377929686e-05,
      "step": 18830,
      "training_step_time": 8.146466732025146
    },
    {
      "epoch": 1.7958641052246093e-05,
      "model_forward_time": 2.278372049331665,
      "step": 18831
    },
    {
      "epoch": 1.7958641052246093e-05,
      "step": 18831,
      "training_step_time": 8.410959005355835
    },
    {
      "epoch": 1.79595947265625e-05,
      "model_forward_time": 1.702404260635376,
      "step": 18832
    },
    {
      "epoch": 1.79595947265625e-05,
      "step": 18832,
      "training_step_time": 7.674810886383057
    },
    {
      "epoch": 1.7960548400878905e-05,
      "model_forward_time": 1.7004554271697998,
      "step": 18833
    },
    {
      "epoch": 1.7960548400878905e-05,
      "step": 18833,
      "training_step_time": 7.630004167556763
    },
    {
      "epoch": 1.7961502075195312e-05,
      "model_forward_time": 1.6981666088104248,
      "step": 18834
    },
    {
      "epoch": 1.7961502075195312e-05,
      "step": 18834,
      "training_step_time": 7.684124231338501
    },
    {
      "epoch": 1.796245574951172e-05,
      "model_forward_time": 1.7010810375213623,
      "step": 18835
    },
    {
      "epoch": 1.796245574951172e-05,
      "step": 18835,
      "training_step_time": 7.004846572875977
    },
    {
      "epoch": 1.7963409423828124e-05,
      "model_forward_time": 1.7006418704986572,
      "step": 18836
    },
    {
      "epoch": 1.7963409423828124e-05,
      "step": 18836,
      "training_step_time": 7.611258029937744
    },
    {
      "epoch": 1.796436309814453e-05,
      "model_forward_time": 1.7025551795959473,
      "step": 18837
    },
    {
      "epoch": 1.796436309814453e-05,
      "step": 18837,
      "training_step_time": 7.5272016525268555
    },
    {
      "epoch": 1.796531677246094e-05,
      "model_forward_time": 1.7024831771850586,
      "step": 18838
    },
    {
      "epoch": 1.796531677246094e-05,
      "step": 18838,
      "training_step_time": 8.624082565307617
    },
    {
      "epoch": 1.7966270446777342e-05,
      "model_forward_time": 1.6997694969177246,
      "step": 18839
    },
    {
      "epoch": 1.7966270446777342e-05,
      "step": 18839,
      "training_step_time": 7.598345994949341
    },
    {
      "epoch": 1.796722412109375e-05,
      "grad_norm": 0.20340323448181152,
      "learning_rate": 9.485472298601705e-06,
      "loss": 0.0655,
      "step": 18840
    },
    {
      "epoch": 1.796722412109375e-05,
      "model_forward_time": 2.2835114002227783,
      "step": 18840
    },
    {
      "dynamics_loss_avg": 0.05772176254540682,
      "epoch": 1.796722412109375e-05,
      "step": 18840
    },
    {
      "action_loss_avg": 0.0019619517493993043,
      "epoch": 1.796722412109375e-05,
      "step": 18840
    },
    {
      "epoch": 1.796722412109375e-05,
      "step": 18840,
      "training_step_time": 8.706658601760864
    },
    {
      "epoch": 1.7968177795410157e-05,
      "model_forward_time": 1.7023851871490479,
      "step": 18841
    },
    {
      "epoch": 1.7968177795410157e-05,
      "step": 18841,
      "training_step_time": 8.55756402015686
    },
    {
      "epoch": 1.796913146972656e-05,
      "model_forward_time": 1.7042224407196045,
      "step": 18842
    },
    {
      "epoch": 1.796913146972656e-05,
      "step": 18842,
      "training_step_time": 7.712698221206665
    },
    {
      "epoch": 1.797008514404297e-05,
      "model_forward_time": 1.6985325813293457,
      "step": 18843
    },
    {
      "epoch": 1.797008514404297e-05,
      "step": 18843,
      "training_step_time": 7.621298313140869
    },
    {
      "epoch": 1.7971038818359376e-05,
      "model_forward_time": 1.698561429977417,
      "step": 18844
    },
    {
      "epoch": 1.7971038818359376e-05,
      "step": 18844,
      "training_step_time": 6.719571113586426
    },
    {
      "epoch": 1.797199249267578e-05,
      "model_forward_time": 1.7020900249481201,
      "step": 18845
    },
    {
      "epoch": 1.797199249267578e-05,
      "step": 18845,
      "training_step_time": 8.057259798049927
    },
    {
      "epoch": 1.7972946166992188e-05,
      "model_forward_time": 1.7039260864257812,
      "step": 18846
    },
    {
      "epoch": 1.7972946166992188e-05,
      "step": 18846,
      "training_step_time": 7.6448352336883545
    },
    {
      "epoch": 1.7973899841308595e-05,
      "model_forward_time": 1.700242519378662,
      "step": 18847
    },
    {
      "epoch": 1.7973899841308595e-05,
      "step": 18847,
      "training_step_time": 9.759907484054565
    },
    {
      "epoch": 1.7974853515625e-05,
      "model_forward_time": 1.7043914794921875,
      "step": 18848
    },
    {
      "epoch": 1.7974853515625e-05,
      "step": 18848,
      "training_step_time": 8.506789922714233
    },
    {
      "epoch": 1.7975807189941406e-05,
      "model_forward_time": 2.229938507080078,
      "step": 18849
    },
    {
      "epoch": 1.7975807189941406e-05,
      "step": 18849,
      "training_step_time": 9.274096727371216
    },
    {
      "epoch": 1.7976760864257814e-05,
      "grad_norm": 0.14787019789218903,
      "learning_rate": 9.484741486223044e-06,
      "loss": 0.0646,
      "step": 18850
    },
    {
      "epoch": 1.7976760864257814e-05,
      "model_forward_time": 1.383331298828125,
      "step": 18850
    },
    {
      "dynamics_loss_avg": 0.05618899231776595,
      "epoch": 1.7976760864257814e-05,
      "step": 18850
    },
    {
      "action_loss_avg": 0.0031963687855750323,
      "epoch": 1.7976760864257814e-05,
      "step": 18850
    },
    {
      "epoch": 1.7976760864257814e-05,
      "step": 18850,
      "training_step_time": 8.253722190856934
    },
    {
      "epoch": 1.7977714538574218e-05,
      "model_forward_time": 1.700486660003662,
      "step": 18851
    },
    {
      "epoch": 1.7977714538574218e-05,
      "step": 18851,
      "training_step_time": 7.626089096069336
    },
    {
      "epoch": 1.7978668212890625e-05,
      "model_forward_time": 1.7022180557250977,
      "step": 18852
    },
    {
      "epoch": 1.7978668212890625e-05,
      "step": 18852,
      "training_step_time": 7.664488792419434
    },
    {
      "epoch": 1.7979621887207033e-05,
      "model_forward_time": 1.700791835784912,
      "step": 18853
    },
    {
      "epoch": 1.7979621887207033e-05,
      "step": 18853,
      "training_step_time": 6.737457752227783
    },
    {
      "epoch": 1.7980575561523437e-05,
      "model_forward_time": 1.708798885345459,
      "step": 18854
    },
    {
      "epoch": 1.7980575561523437e-05,
      "step": 18854,
      "training_step_time": 7.724892854690552
    },
    {
      "epoch": 1.7981529235839844e-05,
      "model_forward_time": 1.7060904502868652,
      "step": 18855
    },
    {
      "epoch": 1.7981529235839844e-05,
      "step": 18855,
      "training_step_time": 8.3021559715271
    },
    {
      "epoch": 1.798248291015625e-05,
      "model_forward_time": 1.7399909496307373,
      "step": 18856
    },
    {
      "epoch": 1.798248291015625e-05,
      "step": 18856,
      "training_step_time": 7.640185594558716
    },
    {
      "epoch": 1.7983436584472656e-05,
      "model_forward_time": 1.734473466873169,
      "step": 18857
    },
    {
      "epoch": 1.7983436584472656e-05,
      "step": 18857,
      "training_step_time": 8.135896921157837
    },
    {
      "epoch": 1.7984390258789063e-05,
      "model_forward_time": 1.7334198951721191,
      "step": 18858
    },
    {
      "epoch": 1.7984390258789063e-05,
      "step": 18858,
      "training_step_time": 8.023233652114868
    },
    {
      "epoch": 1.798534393310547e-05,
      "model_forward_time": 1.7895267009735107,
      "step": 18859
    },
    {
      "epoch": 1.798534393310547e-05,
      "step": 18859,
      "training_step_time": 7.721343040466309
    },
    {
      "epoch": 1.7986297607421874e-05,
      "grad_norm": 0.1802242249250412,
      "learning_rate": 9.484010183399797e-06,
      "loss": 0.0644,
      "step": 18860
    },
    {
      "epoch": 1.7986297607421874e-05,
      "model_forward_time": 1.726517677307129,
      "step": 18860
    },
    {
      "dynamics_loss_avg": 0.0541105286218226,
      "epoch": 1.7986297607421874e-05,
      "step": 18860
    },
    {
      "action_loss_avg": 0.0019419699558056891,
      "epoch": 1.7986297607421874e-05,
      "step": 18860
    },
    {
      "epoch": 1.7986297607421874e-05,
      "step": 18860,
      "training_step_time": 7.53535270690918
    },
    {
      "epoch": 1.7987251281738282e-05,
      "model_forward_time": 1.7403497695922852,
      "step": 18861
    },
    {
      "epoch": 1.7987251281738282e-05,
      "step": 18861,
      "training_step_time": 7.677854299545288
    },
    {
      "epoch": 1.7988204956054686e-05,
      "model_forward_time": 1.737276315689087,
      "step": 18862
    },
    {
      "epoch": 1.7988204956054686e-05,
      "step": 18862,
      "training_step_time": 7.386388540267944
    },
    {
      "epoch": 1.7989158630371093e-05,
      "model_forward_time": 1.72990083694458,
      "step": 18863
    },
    {
      "epoch": 1.7989158630371093e-05,
      "step": 18863,
      "training_step_time": 7.598302602767944
    },
    {
      "epoch": 1.79901123046875e-05,
      "model_forward_time": 1.7307794094085693,
      "step": 18864
    },
    {
      "epoch": 1.79901123046875e-05,
      "step": 18864,
      "training_step_time": 8.111195802688599
    },
    {
      "epoch": 1.7991065979003905e-05,
      "model_forward_time": 1.740361213684082,
      "step": 18865
    },
    {
      "epoch": 1.7991065979003905e-05,
      "step": 18865,
      "training_step_time": 8.870994091033936
    },
    {
      "epoch": 1.7992019653320312e-05,
      "model_forward_time": 1.7843763828277588,
      "step": 18866
    },
    {
      "epoch": 1.7992019653320312e-05,
      "step": 18866,
      "training_step_time": 8.455520153045654
    },
    {
      "epoch": 1.799297332763672e-05,
      "model_forward_time": 1.7346775531768799,
      "step": 18867
    },
    {
      "epoch": 1.799297332763672e-05,
      "step": 18867,
      "training_step_time": 8.420868635177612
    },
    {
      "epoch": 1.7993927001953124e-05,
      "model_forward_time": 2.3635616302490234,
      "step": 18868
    },
    {
      "epoch": 1.7993927001953124e-05,
      "step": 18868,
      "training_step_time": 7.696974754333496
    },
    {
      "epoch": 1.799488067626953e-05,
      "model_forward_time": 1.7328386306762695,
      "step": 18869
    },
    {
      "epoch": 1.799488067626953e-05,
      "step": 18869,
      "training_step_time": 7.530853271484375
    },
    {
      "epoch": 1.799583435058594e-05,
      "grad_norm": 0.1401561051607132,
      "learning_rate": 9.483278390211939e-06,
      "loss": 0.0649,
      "step": 18870
    },
    {
      "epoch": 1.799583435058594e-05,
      "model_forward_time": 1.730210781097412,
      "step": 18870
    },
    {
      "dynamics_loss_avg": 0.05211715139448643,
      "epoch": 1.799583435058594e-05,
      "step": 18870
    },
    {
      "action_loss_avg": 0.00258385181077756,
      "epoch": 1.799583435058594e-05,
      "step": 18870
    },
    {
      "epoch": 1.799583435058594e-05,
      "step": 18870,
      "training_step_time": 8.688068389892578
    },
    {
      "epoch": 1.7996788024902342e-05,
      "model_forward_time": 1.7352283000946045,
      "step": 18871
    },
    {
      "epoch": 1.7996788024902342e-05,
      "step": 18871,
      "training_step_time": 7.378528594970703
    },
    {
      "epoch": 1.799774169921875e-05,
      "model_forward_time": 1.7437081336975098,
      "step": 18872
    },
    {
      "epoch": 1.799774169921875e-05,
      "step": 18872,
      "training_step_time": 7.717096567153931
    },
    {
      "epoch": 1.7998695373535157e-05,
      "model_forward_time": 1.7407126426696777,
      "step": 18873
    },
    {
      "epoch": 1.7998695373535157e-05,
      "step": 18873,
      "training_step_time": 7.53396463394165
    },
    {
      "epoch": 1.799964904785156e-05,
      "model_forward_time": 1.7365851402282715,
      "step": 18874
    },
    {
      "epoch": 1.799964904785156e-05,
      "step": 18874,
      "training_step_time": 7.614989995956421
    },
    {
      "epoch": 1.800060272216797e-05,
      "model_forward_time": 1.7427678108215332,
      "step": 18875
    },
    {
      "epoch": 1.800060272216797e-05,
      "step": 18875,
      "training_step_time": 8.258059740066528
    },
    {
      "epoch": 1.8001556396484376e-05,
      "model_forward_time": 1.7403881549835205,
      "step": 18876
    },
    {
      "epoch": 1.8001556396484376e-05,
      "step": 18876,
      "training_step_time": 8.447583436965942
    },
    {
      "epoch": 1.800251007080078e-05,
      "model_forward_time": 2.34810209274292,
      "step": 18877
    },
    {
      "epoch": 1.800251007080078e-05,
      "step": 18877,
      "training_step_time": 7.696261882781982
    },
    {
      "epoch": 1.8003463745117188e-05,
      "model_forward_time": 1.7413628101348877,
      "step": 18878
    },
    {
      "epoch": 1.8003463745117188e-05,
      "step": 18878,
      "training_step_time": 8.35484004020691
    },
    {
      "epoch": 1.8004417419433595e-05,
      "model_forward_time": 1.7372636795043945,
      "step": 18879
    },
    {
      "epoch": 1.8004417419433595e-05,
      "step": 18879,
      "training_step_time": 7.732603549957275
    },
    {
      "epoch": 1.800537109375e-05,
      "grad_norm": 0.17358578741550446,
      "learning_rate": 9.482546106739496e-06,
      "loss": 0.067,
      "step": 18880
    },
    {
      "epoch": 1.800537109375e-05,
      "model_forward_time": 1.7362451553344727,
      "step": 18880
    },
    {
      "dynamics_loss_avg": 0.06076495200395584,
      "epoch": 1.800537109375e-05,
      "step": 18880
    },
    {
      "action_loss_avg": 0.0025854778214124963,
      "epoch": 1.800537109375e-05,
      "step": 18880
    },
    {
      "epoch": 1.800537109375e-05,
      "step": 18880,
      "training_step_time": 7.7320497035980225
    },
    {
      "epoch": 1.8006324768066406e-05,
      "model_forward_time": 1.742199182510376,
      "step": 18881
    },
    {
      "epoch": 1.8006324768066406e-05,
      "step": 18881,
      "training_step_time": 7.766144752502441
    },
    {
      "epoch": 1.8007278442382814e-05,
      "model_forward_time": 1.739574909210205,
      "step": 18882
    },
    {
      "epoch": 1.8007278442382814e-05,
      "step": 18882,
      "training_step_time": 7.5583226680755615
    },
    {
      "epoch": 1.8008232116699218e-05,
      "model_forward_time": 1.7316529750823975,
      "step": 18883
    },
    {
      "epoch": 1.8008232116699218e-05,
      "step": 18883,
      "training_step_time": 9.27310562133789
    },
    {
      "epoch": 1.8009185791015625e-05,
      "model_forward_time": 1.7350716590881348,
      "step": 18884
    },
    {
      "epoch": 1.8009185791015625e-05,
      "step": 18884,
      "training_step_time": 9.138105392456055
    },
    {
      "epoch": 1.8010139465332033e-05,
      "model_forward_time": 1.747607946395874,
      "step": 18885
    },
    {
      "epoch": 1.8010139465332033e-05,
      "step": 18885,
      "training_step_time": 10.439360857009888
    },
    {
      "epoch": 1.8011093139648437e-05,
      "model_forward_time": 2.339430093765259,
      "step": 18886
    },
    {
      "epoch": 1.8011093139648437e-05,
      "step": 18886,
      "training_step_time": 8.675240516662598
    },
    {
      "epoch": 1.8012046813964844e-05,
      "model_forward_time": 1.7885370254516602,
      "step": 18887
    },
    {
      "epoch": 1.8012046813964844e-05,
      "step": 18887,
      "training_step_time": 7.63743782043457
    },
    {
      "epoch": 1.801300048828125e-05,
      "model_forward_time": 1.7010693550109863,
      "step": 18888
    },
    {
      "epoch": 1.801300048828125e-05,
      "step": 18888,
      "training_step_time": 8.112282991409302
    },
    {
      "epoch": 1.8013954162597656e-05,
      "model_forward_time": 1.7245614528656006,
      "step": 18889
    },
    {
      "epoch": 1.8013954162597656e-05,
      "step": 18889,
      "training_step_time": 7.616180419921875
    },
    {
      "epoch": 1.8014907836914063e-05,
      "grad_norm": 0.11404866725206375,
      "learning_rate": 9.48181333306255e-06,
      "loss": 0.0652,
      "step": 18890
    },
    {
      "epoch": 1.8014907836914063e-05,
      "model_forward_time": 1.7315073013305664,
      "step": 18890
    },
    {
      "dynamics_loss_avg": 0.060029417462646964,
      "epoch": 1.8014907836914063e-05,
      "step": 18890
    },
    {
      "action_loss_avg": 0.0020289614534704016,
      "epoch": 1.8014907836914063e-05,
      "step": 18890
    },
    {
      "epoch": 1.8014907836914063e-05,
      "step": 18890,
      "training_step_time": 7.455628871917725
    },
    {
      "epoch": 1.8015861511230467e-05,
      "model_forward_time": 1.7303247451782227,
      "step": 18891
    },
    {
      "epoch": 1.8015861511230467e-05,
      "step": 18891,
      "training_step_time": 7.575141429901123
    },
    {
      "epoch": 1.8016815185546874e-05,
      "model_forward_time": 1.7422215938568115,
      "step": 18892
    },
    {
      "epoch": 1.8016815185546874e-05,
      "step": 18892,
      "training_step_time": 7.627203702926636
    },
    {
      "epoch": 1.8017768859863282e-05,
      "model_forward_time": 1.7421269416809082,
      "step": 18893
    },
    {
      "epoch": 1.8017768859863282e-05,
      "step": 18893,
      "training_step_time": 9.618310451507568
    },
    {
      "epoch": 1.8018722534179686e-05,
      "model_forward_time": 1.7358689308166504,
      "step": 18894
    },
    {
      "epoch": 1.8018722534179686e-05,
      "step": 18894,
      "training_step_time": 7.627748489379883
    },
    {
      "epoch": 1.8019676208496093e-05,
      "model_forward_time": 2.318843126296997,
      "step": 18895
    },
    {
      "epoch": 1.8019676208496093e-05,
      "step": 18895,
      "training_step_time": 7.542151927947998
    },
    {
      "epoch": 1.80206298828125e-05,
      "model_forward_time": 1.3837864398956299,
      "step": 18896
    },
    {
      "epoch": 1.80206298828125e-05,
      "step": 18896,
      "training_step_time": 7.7282960414886475
    },
    {
      "epoch": 1.8021583557128905e-05,
      "model_forward_time": 1.7429614067077637,
      "step": 18897
    },
    {
      "epoch": 1.8021583557128905e-05,
      "step": 18897,
      "training_step_time": 7.674762725830078
    },
    {
      "epoch": 1.8022537231445312e-05,
      "model_forward_time": 1.7420668601989746,
      "step": 18898
    },
    {
      "epoch": 1.8022537231445312e-05,
      "step": 18898,
      "training_step_time": 7.6367433071136475
    },
    {
      "epoch": 1.802349090576172e-05,
      "model_forward_time": 1.7398655414581299,
      "step": 18899
    },
    {
      "epoch": 1.802349090576172e-05,
      "step": 18899,
      "training_step_time": 7.399214744567871
    },
    {
      "epoch": 1.8024444580078124e-05,
      "grad_norm": 0.16318151354789734,
      "learning_rate": 9.481080069261237e-06,
      "loss": 0.0683,
      "step": 18900
    },
    {
      "epoch": 1.8024444580078124e-05,
      "model_forward_time": 1.7356541156768799,
      "step": 18900
    },
    {
      "dynamics_loss_avg": 0.05422659041360021,
      "epoch": 1.8024444580078124e-05,
      "step": 18900
    },
    {
      "action_loss_avg": 0.0017595007957424968,
      "epoch": 1.8024444580078124e-05,
      "step": 18900
    },
    {
      "epoch": 1.8024444580078124e-05,
      "step": 18900,
      "training_step_time": 7.537187099456787
    },
    {
      "epoch": 1.802539825439453e-05,
      "model_forward_time": 1.7462742328643799,
      "step": 18901
    },
    {
      "epoch": 1.802539825439453e-05,
      "step": 18901,
      "training_step_time": 7.58194899559021
    },
    {
      "epoch": 1.802635192871094e-05,
      "model_forward_time": 1.7428052425384521,
      "step": 18902
    },
    {
      "epoch": 1.802635192871094e-05,
      "step": 18902,
      "training_step_time": 8.125786066055298
    },
    {
      "epoch": 1.8027305603027343e-05,
      "model_forward_time": 1.7339503765106201,
      "step": 18903
    },
    {
      "epoch": 1.8027305603027343e-05,
      "step": 18903,
      "training_step_time": 8.55408501625061
    },
    {
      "epoch": 1.802825927734375e-05,
      "model_forward_time": 1.7358787059783936,
      "step": 18904
    },
    {
      "epoch": 1.802825927734375e-05,
      "step": 18904,
      "training_step_time": 7.548686265945435
    },
    {
      "epoch": 1.8029212951660157e-05,
      "model_forward_time": 2.354694128036499,
      "step": 18905
    },
    {
      "epoch": 1.8029212951660157e-05,
      "step": 18905,
      "training_step_time": 8.315519571304321
    },
    {
      "epoch": 1.803016662597656e-05,
      "model_forward_time": 1.7419004440307617,
      "step": 18906
    },
    {
      "epoch": 1.803016662597656e-05,
      "step": 18906,
      "training_step_time": 7.707242488861084
    },
    {
      "epoch": 1.803112030029297e-05,
      "model_forward_time": 1.7422833442687988,
      "step": 18907
    },
    {
      "epoch": 1.803112030029297e-05,
      "step": 18907,
      "training_step_time": 7.765852689743042
    },
    {
      "epoch": 1.8032073974609376e-05,
      "model_forward_time": 1.7373480796813965,
      "step": 18908
    },
    {
      "epoch": 1.8032073974609376e-05,
      "step": 18908,
      "training_step_time": 7.56986403465271
    },
    {
      "epoch": 1.803302764892578e-05,
      "model_forward_time": 1.7464592456817627,
      "step": 18909
    },
    {
      "epoch": 1.803302764892578e-05,
      "step": 18909,
      "training_step_time": 7.647955417633057
    },
    {
      "epoch": 1.8033981323242188e-05,
      "grad_norm": 0.2151656150817871,
      "learning_rate": 9.480346315415744e-06,
      "loss": 0.0667,
      "step": 18910
    },
    {
      "epoch": 1.8033981323242188e-05,
      "model_forward_time": 1.7374980449676514,
      "step": 18910
    },
    {
      "dynamics_loss_avg": 0.06727296859025955,
      "epoch": 1.8033981323242188e-05,
      "step": 18910
    },
    {
      "action_loss_avg": 0.002240031724795699,
      "epoch": 1.8033981323242188e-05,
      "step": 18910
    },
    {
      "epoch": 1.8033981323242188e-05,
      "step": 18910,
      "training_step_time": 7.57805871963501
    },
    {
      "epoch": 1.8034934997558595e-05,
      "model_forward_time": 1.7366306781768799,
      "step": 18911
    },
    {
      "epoch": 1.8034934997558595e-05,
      "step": 18911,
      "training_step_time": 8.215029954910278
    },
    {
      "epoch": 1.8035888671875e-05,
      "model_forward_time": 1.7458386421203613,
      "step": 18912
    },
    {
      "epoch": 1.8035888671875e-05,
      "step": 18912,
      "training_step_time": 9.558241844177246
    },
    {
      "epoch": 1.8036842346191406e-05,
      "model_forward_time": 1.7340993881225586,
      "step": 18913
    },
    {
      "epoch": 1.8036842346191406e-05,
      "step": 18913,
      "training_step_time": 8.117501974105835
    },
    {
      "epoch": 1.8037796020507814e-05,
      "model_forward_time": 2.350674629211426,
      "step": 18914
    },
    {
      "epoch": 1.8037796020507814e-05,
      "step": 18914,
      "training_step_time": 7.790560960769653
    },
    {
      "epoch": 1.8038749694824218e-05,
      "model_forward_time": 1.740596055984497,
      "step": 18915
    },
    {
      "epoch": 1.8038749694824218e-05,
      "step": 18915,
      "training_step_time": 8.273658514022827
    },
    {
      "epoch": 1.8039703369140625e-05,
      "model_forward_time": 1.7135992050170898,
      "step": 18916
    },
    {
      "epoch": 1.8039703369140625e-05,
      "step": 18916,
      "training_step_time": 7.71659517288208
    },
    {
      "epoch": 1.8040657043457033e-05,
      "model_forward_time": 1.7438626289367676,
      "step": 18917
    },
    {
      "epoch": 1.8040657043457033e-05,
      "step": 18917,
      "training_step_time": 7.621699571609497
    },
    {
      "epoch": 1.8041610717773437e-05,
      "model_forward_time": 1.735302209854126,
      "step": 18918
    },
    {
      "epoch": 1.8041610717773437e-05,
      "step": 18918,
      "training_step_time": 7.454044818878174
    },
    {
      "epoch": 1.8042564392089844e-05,
      "model_forward_time": 1.7408912181854248,
      "step": 18919
    },
    {
      "epoch": 1.8042564392089844e-05,
      "step": 18919,
      "training_step_time": 7.567455768585205
    },
    {
      "epoch": 1.804351806640625e-05,
      "grad_norm": 0.2941973805427551,
      "learning_rate": 9.479612071606315e-06,
      "loss": 0.0681,
      "step": 18920
    },
    {
      "epoch": 1.804351806640625e-05,
      "model_forward_time": 1.7301557064056396,
      "step": 18920
    },
    {
      "dynamics_loss_avg": 0.08035441339015961,
      "epoch": 1.804351806640625e-05,
      "step": 18920
    },
    {
      "action_loss_avg": 0.003234553535003215,
      "epoch": 1.804351806640625e-05,
      "step": 18920
    },
    {
      "epoch": 1.804351806640625e-05,
      "step": 18920,
      "training_step_time": 9.56059193611145
    },
    {
      "epoch": 1.8044471740722656e-05,
      "model_forward_time": 1.732969045639038,
      "step": 18921
    },
    {
      "epoch": 1.8044471740722656e-05,
      "step": 18921,
      "training_step_time": 8.72823166847229
    },
    {
      "epoch": 1.8045425415039063e-05,
      "model_forward_time": 1.7311370372772217,
      "step": 18922
    },
    {
      "epoch": 1.8045425415039063e-05,
      "step": 18922,
      "training_step_time": 8.32867431640625
    },
    {
      "epoch": 1.8046379089355467e-05,
      "model_forward_time": 2.350221872329712,
      "step": 18923
    },
    {
      "epoch": 1.8046379089355467e-05,
      "step": 18923,
      "training_step_time": 8.515138626098633
    },
    {
      "epoch": 1.8047332763671875e-05,
      "model_forward_time": 1.736297607421875,
      "step": 18924
    },
    {
      "epoch": 1.8047332763671875e-05,
      "step": 18924,
      "training_step_time": 7.637509107589722
    },
    {
      "epoch": 1.8048286437988282e-05,
      "model_forward_time": 1.7431387901306152,
      "step": 18925
    },
    {
      "epoch": 1.8048286437988282e-05,
      "step": 18925,
      "training_step_time": 8.522886276245117
    },
    {
      "epoch": 1.8049240112304686e-05,
      "model_forward_time": 1.7452235221862793,
      "step": 18926
    },
    {
      "epoch": 1.8049240112304686e-05,
      "step": 18926,
      "training_step_time": 7.584212779998779
    },
    {
      "epoch": 1.8050193786621093e-05,
      "model_forward_time": 1.7381680011749268,
      "step": 18927
    },
    {
      "epoch": 1.8050193786621093e-05,
      "step": 18927,
      "training_step_time": 7.608245134353638
    },
    {
      "epoch": 1.80511474609375e-05,
      "model_forward_time": 1.7333130836486816,
      "step": 18928
    },
    {
      "epoch": 1.80511474609375e-05,
      "step": 18928,
      "training_step_time": 7.488742828369141
    },
    {
      "epoch": 1.8052101135253905e-05,
      "model_forward_time": 1.7541234493255615,
      "step": 18929
    },
    {
      "epoch": 1.8052101135253905e-05,
      "step": 18929,
      "training_step_time": 9.071171998977661
    },
    {
      "epoch": 1.8053054809570312e-05,
      "grad_norm": 0.42790788412094116,
      "learning_rate": 9.478877337913244e-06,
      "loss": 0.0679,
      "step": 18930
    },
    {
      "epoch": 1.8053054809570312e-05,
      "model_forward_time": 1.729738712310791,
      "step": 18930
    },
    {
      "dynamics_loss_avg": 0.06963106580078601,
      "epoch": 1.8053054809570312e-05,
      "step": 18930
    },
    {
      "action_loss_avg": 0.0029268235492054375,
      "epoch": 1.8053054809570312e-05,
      "step": 18930
    },
    {
      "epoch": 1.8053054809570312e-05,
      "step": 18930,
      "training_step_time": 8.204700231552124
    },
    {
      "epoch": 1.805400848388672e-05,
      "model_forward_time": 1.7470433712005615,
      "step": 18931
    },
    {
      "epoch": 1.805400848388672e-05,
      "step": 18931,
      "training_step_time": 8.375317335128784
    },
    {
      "epoch": 1.8054962158203124e-05,
      "model_forward_time": 2.3348000049591064,
      "step": 18932
    },
    {
      "epoch": 1.8054962158203124e-05,
      "step": 18932,
      "training_step_time": 10.135022401809692
    },
    {
      "epoch": 1.805591583251953e-05,
      "model_forward_time": 1.7923178672790527,
      "step": 18933
    },
    {
      "epoch": 1.805591583251953e-05,
      "step": 18933,
      "training_step_time": 8.541591882705688
    },
    {
      "epoch": 1.805686950683594e-05,
      "model_forward_time": 1.7384459972381592,
      "step": 18934
    },
    {
      "epoch": 1.805686950683594e-05,
      "step": 18934,
      "training_step_time": 7.630284070968628
    },
    {
      "epoch": 1.8057823181152343e-05,
      "model_forward_time": 1.7460906505584717,
      "step": 18935
    },
    {
      "epoch": 1.8057823181152343e-05,
      "step": 18935,
      "training_step_time": 7.7680885791778564
    },
    {
      "epoch": 1.805877685546875e-05,
      "model_forward_time": 1.7354907989501953,
      "step": 18936
    },
    {
      "epoch": 1.805877685546875e-05,
      "step": 18936,
      "training_step_time": 7.618671894073486
    },
    {
      "epoch": 1.8059730529785157e-05,
      "model_forward_time": 1.7391226291656494,
      "step": 18937
    },
    {
      "epoch": 1.8059730529785157e-05,
      "step": 18937,
      "training_step_time": 7.447601795196533
    },
    {
      "epoch": 1.806068420410156e-05,
      "model_forward_time": 1.7461740970611572,
      "step": 18938
    },
    {
      "epoch": 1.806068420410156e-05,
      "step": 18938,
      "training_step_time": 8.498592615127563
    },
    {
      "epoch": 1.806163787841797e-05,
      "model_forward_time": 1.7391915321350098,
      "step": 18939
    },
    {
      "epoch": 1.806163787841797e-05,
      "step": 18939,
      "training_step_time": 8.63838791847229
    },
    {
      "epoch": 1.8062591552734376e-05,
      "grad_norm": 0.32495200634002686,
      "learning_rate": 9.478142114416882e-06,
      "loss": 0.0647,
      "step": 18940
    },
    {
      "epoch": 1.8062591552734376e-05,
      "model_forward_time": 1.7379930019378662,
      "step": 18940
    },
    {
      "dynamics_loss_avg": 0.042333597876131536,
      "epoch": 1.8062591552734376e-05,
      "step": 18940
    },
    {
      "action_loss_avg": 0.0025462309073191135,
      "epoch": 1.8062591552734376e-05,
      "step": 18940
    },
    {
      "epoch": 1.8062591552734376e-05,
      "step": 18940,
      "training_step_time": 8.650564432144165
    },
    {
      "epoch": 1.806354522705078e-05,
      "model_forward_time": 2.304225206375122,
      "step": 18941
    },
    {
      "epoch": 1.806354522705078e-05,
      "step": 18941,
      "training_step_time": 8.525962352752686
    },
    {
      "epoch": 1.8064498901367188e-05,
      "model_forward_time": 1.7955210208892822,
      "step": 18942
    },
    {
      "epoch": 1.8064498901367188e-05,
      "step": 18942,
      "training_step_time": 7.660509347915649
    },
    {
      "epoch": 1.8065452575683595e-05,
      "model_forward_time": 1.7407784461975098,
      "step": 18943
    },
    {
      "epoch": 1.8065452575683595e-05,
      "step": 18943,
      "training_step_time": 7.699819564819336
    },
    {
      "epoch": 1.806640625e-05,
      "model_forward_time": 1.7415153980255127,
      "step": 18944
    },
    {
      "epoch": 1.806640625e-05,
      "step": 18944,
      "training_step_time": 7.849440097808838
    },
    {
      "epoch": 1.8067359924316407e-05,
      "model_forward_time": 1.7464239597320557,
      "step": 18945
    },
    {
      "epoch": 1.8067359924316407e-05,
      "step": 18945,
      "training_step_time": 7.583114147186279
    },
    {
      "epoch": 1.8068313598632814e-05,
      "model_forward_time": 1.746485710144043,
      "step": 18946
    },
    {
      "epoch": 1.8068313598632814e-05,
      "step": 18946,
      "training_step_time": 8.144600629806519
    },
    {
      "epoch": 1.8069267272949218e-05,
      "model_forward_time": 1.7443079948425293,
      "step": 18947
    },
    {
      "epoch": 1.8069267272949218e-05,
      "step": 18947,
      "training_step_time": 7.590605735778809
    },
    {
      "epoch": 1.8070220947265625e-05,
      "model_forward_time": 1.7446956634521484,
      "step": 18948
    },
    {
      "epoch": 1.8070220947265625e-05,
      "step": 18948,
      "training_step_time": 7.717058420181274
    },
    {
      "epoch": 1.8071174621582033e-05,
      "model_forward_time": 1.6970818042755127,
      "step": 18949
    },
    {
      "epoch": 1.8071174621582033e-05,
      "step": 18949,
      "training_step_time": 9.504411458969116
    },
    {
      "epoch": 1.8072128295898437e-05,
      "grad_norm": 0.36237573623657227,
      "learning_rate": 9.477406401197631e-06,
      "loss": 0.0612,
      "step": 18950
    },
    {
      "epoch": 1.8072128295898437e-05,
      "model_forward_time": 1.7108702659606934,
      "step": 18950
    },
    {
      "dynamics_loss_avg": 0.06510227341204881,
      "epoch": 1.8072128295898437e-05,
      "step": 18950
    },
    {
      "action_loss_avg": 0.003452240733895451,
      "epoch": 1.8072128295898437e-05,
      "step": 18950
    },
    {
      "epoch": 1.8072128295898437e-05,
      "step": 18950,
      "training_step_time": 8.714807748794556
    },
    {
      "epoch": 1.8073081970214844e-05,
      "model_forward_time": 1.7600772380828857,
      "step": 18951
    },
    {
      "epoch": 1.8073081970214844e-05,
      "step": 18951,
      "training_step_time": 8.246997356414795
    },
    {
      "epoch": 1.807403564453125e-05,
      "model_forward_time": 1.7176411151885986,
      "step": 18952
    },
    {
      "epoch": 1.807403564453125e-05,
      "step": 18952,
      "training_step_time": 7.816267251968384
    },
    {
      "epoch": 1.8074989318847656e-05,
      "model_forward_time": 1.716749906539917,
      "step": 18953
    },
    {
      "epoch": 1.8074989318847656e-05,
      "step": 18953,
      "training_step_time": 7.729812383651733
    },
    {
      "epoch": 1.8075942993164063e-05,
      "model_forward_time": 1.7120823860168457,
      "step": 18954
    },
    {
      "epoch": 1.8075942993164063e-05,
      "step": 18954,
      "training_step_time": 7.605034112930298
    },
    {
      "epoch": 1.8076896667480467e-05,
      "model_forward_time": 1.7126014232635498,
      "step": 18955
    },
    {
      "epoch": 1.8076896667480467e-05,
      "step": 18955,
      "training_step_time": 7.6262288093566895
    },
    {
      "epoch": 1.8077850341796875e-05,
      "model_forward_time": 1.7182302474975586,
      "step": 18956
    },
    {
      "epoch": 1.8077850341796875e-05,
      "step": 18956,
      "training_step_time": 7.557547092437744
    },
    {
      "epoch": 1.8078804016113282e-05,
      "model_forward_time": 1.7306251525878906,
      "step": 18957
    },
    {
      "epoch": 1.8078804016113282e-05,
      "step": 18957,
      "training_step_time": 8.28575325012207
    },
    {
      "epoch": 1.8079757690429686e-05,
      "model_forward_time": 1.7452147006988525,
      "step": 18958
    },
    {
      "epoch": 1.8079757690429686e-05,
      "step": 18958,
      "training_step_time": 7.661999940872192
    },
    {
      "epoch": 1.8080711364746093e-05,
      "model_forward_time": 1.7549371719360352,
      "step": 18959
    },
    {
      "epoch": 1.8080711364746093e-05,
      "step": 18959,
      "training_step_time": 8.139995336532593
    },
    {
      "epoch": 1.80816650390625e-05,
      "grad_norm": 0.18774767220020294,
      "learning_rate": 9.476670198335946e-06,
      "loss": 0.0646,
      "step": 18960
    },
    {
      "epoch": 1.80816650390625e-05,
      "model_forward_time": 1.7908895015716553,
      "step": 18960
    },
    {
      "dynamics_loss_avg": 0.05569920539855957,
      "epoch": 1.80816650390625e-05,
      "step": 18960
    },
    {
      "action_loss_avg": 0.002061655788565986,
      "epoch": 1.80816650390625e-05,
      "step": 18960
    },
    {
      "epoch": 1.80816650390625e-05,
      "step": 18960,
      "training_step_time": 7.561476945877075
    },
    {
      "epoch": 1.8082618713378905e-05,
      "model_forward_time": 1.7453229427337646,
      "step": 18961
    },
    {
      "epoch": 1.8082618713378905e-05,
      "step": 18961,
      "training_step_time": 7.782982349395752
    },
    {
      "epoch": 1.8083572387695312e-05,
      "model_forward_time": 1.7405054569244385,
      "step": 18962
    },
    {
      "epoch": 1.8083572387695312e-05,
      "step": 18962,
      "training_step_time": 7.983234167098999
    },
    {
      "epoch": 1.808452606201172e-05,
      "model_forward_time": 1.7427246570587158,
      "step": 18963
    },
    {
      "epoch": 1.808452606201172e-05,
      "step": 18963,
      "training_step_time": 7.511683940887451
    },
    {
      "epoch": 1.8085479736328124e-05,
      "model_forward_time": 1.7465975284576416,
      "step": 18964
    },
    {
      "epoch": 1.8085479736328124e-05,
      "step": 18964,
      "training_step_time": 7.5287926197052
    },
    {
      "epoch": 1.808643341064453e-05,
      "model_forward_time": 1.725865125656128,
      "step": 18965
    },
    {
      "epoch": 1.808643341064453e-05,
      "step": 18965,
      "training_step_time": 8.466098546981812
    },
    {
      "epoch": 1.808738708496094e-05,
      "model_forward_time": 1.7414438724517822,
      "step": 18966
    },
    {
      "epoch": 1.808738708496094e-05,
      "step": 18966,
      "training_step_time": 8.302817106246948
    },
    {
      "epoch": 1.8088340759277343e-05,
      "model_forward_time": 1.755925178527832,
      "step": 18967
    },
    {
      "epoch": 1.8088340759277343e-05,
      "step": 18967,
      "training_step_time": 9.328711986541748
    },
    {
      "epoch": 1.808929443359375e-05,
      "model_forward_time": 1.745518445968628,
      "step": 18968
    },
    {
      "epoch": 1.808929443359375e-05,
      "step": 18968,
      "training_step_time": 8.180103302001953
    },
    {
      "epoch": 1.8090248107910157e-05,
      "model_forward_time": 1.8002593517303467,
      "step": 18969
    },
    {
      "epoch": 1.8090248107910157e-05,
      "step": 18969,
      "training_step_time": 7.588479280471802
    },
    {
      "epoch": 1.809120178222656e-05,
      "grad_norm": 0.17849408090114594,
      "learning_rate": 9.475933505912341e-06,
      "loss": 0.0622,
      "step": 18970
    },
    {
      "epoch": 1.809120178222656e-05,
      "model_forward_time": 1.7287840843200684,
      "step": 18970
    },
    {
      "dynamics_loss_avg": 0.06345624197274446,
      "epoch": 1.809120178222656e-05,
      "step": 18970
    },
    {
      "action_loss_avg": 0.0021818333538249136,
      "epoch": 1.809120178222656e-05,
      "step": 18970
    },
    {
      "epoch": 1.809120178222656e-05,
      "step": 18970,
      "training_step_time": 8.232295751571655
    },
    {
      "epoch": 1.809215545654297e-05,
      "model_forward_time": 1.7555365562438965,
      "step": 18971
    },
    {
      "epoch": 1.809215545654297e-05,
      "step": 18971,
      "training_step_time": 8.597636222839355
    },
    {
      "epoch": 1.8093109130859376e-05,
      "model_forward_time": 1.7311153411865234,
      "step": 18972
    },
    {
      "epoch": 1.8093109130859376e-05,
      "step": 18972,
      "training_step_time": 7.626638412475586
    },
    {
      "epoch": 1.809406280517578e-05,
      "model_forward_time": 1.7366969585418701,
      "step": 18973
    },
    {
      "epoch": 1.809406280517578e-05,
      "step": 18973,
      "training_step_time": 7.570208787918091
    },
    {
      "epoch": 1.8095016479492188e-05,
      "model_forward_time": 1.7487003803253174,
      "step": 18974
    },
    {
      "epoch": 1.8095016479492188e-05,
      "step": 18974,
      "training_step_time": 8.13076376914978
    },
    {
      "epoch": 1.8095970153808595e-05,
      "model_forward_time": 1.7402048110961914,
      "step": 18975
    },
    {
      "epoch": 1.8095970153808595e-05,
      "step": 18975,
      "training_step_time": 9.241736650466919
    },
    {
      "epoch": 1.8096923828125e-05,
      "model_forward_time": 1.7463321685791016,
      "step": 18976
    },
    {
      "epoch": 1.8096923828125e-05,
      "step": 18976,
      "training_step_time": 8.592046976089478
    },
    {
      "epoch": 1.8097877502441407e-05,
      "model_forward_time": 1.354677677154541,
      "step": 18977
    },
    {
      "epoch": 1.8097877502441407e-05,
      "step": 18977,
      "training_step_time": 7.78983211517334
    },
    {
      "epoch": 1.8098831176757814e-05,
      "model_forward_time": 2.3734939098358154,
      "step": 18978
    },
    {
      "epoch": 1.8098831176757814e-05,
      "step": 18978,
      "training_step_time": 8.368743419647217
    },
    {
      "epoch": 1.8099784851074218e-05,
      "model_forward_time": 1.7390737533569336,
      "step": 18979
    },
    {
      "epoch": 1.8099784851074218e-05,
      "step": 18979,
      "training_step_time": 7.657251358032227
    },
    {
      "epoch": 1.8100738525390625e-05,
      "grad_norm": 0.22366639971733093,
      "learning_rate": 9.475196324007376e-06,
      "loss": 0.0637,
      "step": 18980
    },
    {
      "epoch": 1.8100738525390625e-05,
      "model_forward_time": 1.7348999977111816,
      "step": 18980
    },
    {
      "dynamics_loss_avg": 0.05769199077039957,
      "epoch": 1.8100738525390625e-05,
      "step": 18980
    },
    {
      "action_loss_avg": 0.0020928769023157655,
      "epoch": 1.8100738525390625e-05,
      "step": 18980
    },
    {
      "epoch": 1.8100738525390625e-05,
      "step": 18980,
      "training_step_time": 8.656795978546143
    },
    {
      "epoch": 1.8101692199707033e-05,
      "model_forward_time": 1.3618605136871338,
      "step": 18981
    },
    {
      "epoch": 1.8101692199707033e-05,
      "step": 18981,
      "training_step_time": 7.583504676818848
    },
    {
      "epoch": 1.8102645874023437e-05,
      "model_forward_time": 1.7437732219696045,
      "step": 18982
    },
    {
      "epoch": 1.8102645874023437e-05,
      "step": 18982,
      "training_step_time": 7.636041641235352
    },
    {
      "epoch": 1.8103599548339844e-05,
      "model_forward_time": 1.7411365509033203,
      "step": 18983
    },
    {
      "epoch": 1.8103599548339844e-05,
      "step": 18983,
      "training_step_time": 8.574809312820435
    },
    {
      "epoch": 1.810455322265625e-05,
      "model_forward_time": 1.7431631088256836,
      "step": 18984
    },
    {
      "epoch": 1.810455322265625e-05,
      "step": 18984,
      "training_step_time": 7.610776424407959
    },
    {
      "epoch": 1.8105506896972656e-05,
      "model_forward_time": 1.737708330154419,
      "step": 18985
    },
    {
      "epoch": 1.8105506896972656e-05,
      "step": 18985,
      "training_step_time": 7.558221340179443
    },
    {
      "epoch": 1.8106460571289063e-05,
      "model_forward_time": 1.700171709060669,
      "step": 18986
    },
    {
      "epoch": 1.8106460571289063e-05,
      "step": 18986,
      "training_step_time": 8.963744401931763
    },
    {
      "epoch": 1.8107414245605467e-05,
      "model_forward_time": 2.295814275741577,
      "step": 18987
    },
    {
      "epoch": 1.8107414245605467e-05,
      "step": 18987,
      "training_step_time": 8.487022161483765
    },
    {
      "epoch": 1.8108367919921875e-05,
      "model_forward_time": 1.7155709266662598,
      "step": 18988
    },
    {
      "epoch": 1.8108367919921875e-05,
      "step": 18988,
      "training_step_time": 7.659177303314209
    },
    {
      "epoch": 1.8109321594238282e-05,
      "model_forward_time": 1.7185125350952148,
      "step": 18989
    },
    {
      "epoch": 1.8109321594238282e-05,
      "step": 18989,
      "training_step_time": 7.644588232040405
    },
    {
      "epoch": 1.8110275268554686e-05,
      "grad_norm": 0.18496640026569366,
      "learning_rate": 9.47445865270167e-06,
      "loss": 0.0654,
      "step": 18990
    },
    {
      "epoch": 1.8110275268554686e-05,
      "model_forward_time": 1.3356986045837402,
      "step": 18990
    },
    {
      "dynamics_loss_avg": 0.05481184162199497,
      "epoch": 1.8110275268554686e-05,
      "step": 18990
    },
    {
      "action_loss_avg": 0.0023672256502322853,
      "epoch": 1.8110275268554686e-05,
      "step": 18990
    },
    {
      "epoch": 1.8110275268554686e-05,
      "step": 18990,
      "training_step_time": 7.762583494186401
    },
    {
      "epoch": 1.8111228942871093e-05,
      "model_forward_time": 1.715369462966919,
      "step": 18991
    },
    {
      "epoch": 1.8111228942871093e-05,
      "step": 18991,
      "training_step_time": 7.612763166427612
    },
    {
      "epoch": 1.81121826171875e-05,
      "model_forward_time": 1.7206082344055176,
      "step": 18992
    },
    {
      "epoch": 1.81121826171875e-05,
      "step": 18992,
      "training_step_time": 7.615922927856445
    },
    {
      "epoch": 1.8113136291503905e-05,
      "model_forward_time": 1.715221881866455,
      "step": 18993
    },
    {
      "epoch": 1.8113136291503905e-05,
      "step": 18993,
      "training_step_time": 7.666409492492676
    },
    {
      "epoch": 1.8114089965820312e-05,
      "model_forward_time": 1.7062129974365234,
      "step": 18994
    },
    {
      "epoch": 1.8114089965820312e-05,
      "step": 18994,
      "training_step_time": 7.646928548812866
    },
    {
      "epoch": 1.811504364013672e-05,
      "model_forward_time": 1.7180383205413818,
      "step": 18995
    },
    {
      "epoch": 1.811504364013672e-05,
      "step": 18995,
      "training_step_time": 8.740377187728882
    },
    {
      "epoch": 1.8115997314453124e-05,
      "model_forward_time": 2.3186068534851074,
      "step": 18996
    },
    {
      "epoch": 1.8115997314453124e-05,
      "step": 18996,
      "training_step_time": 7.711382150650024
    },
    {
      "epoch": 1.811695098876953e-05,
      "model_forward_time": 1.7167973518371582,
      "step": 18997
    },
    {
      "epoch": 1.811695098876953e-05,
      "step": 18997,
      "training_step_time": 8.569553136825562
    },
    {
      "epoch": 1.811790466308594e-05,
      "model_forward_time": 1.7431416511535645,
      "step": 18998
    },
    {
      "epoch": 1.811790466308594e-05,
      "step": 18998,
      "training_step_time": 8.046767950057983
    },
    {
      "epoch": 1.8118858337402343e-05,
      "model_forward_time": 1.7534716129302979,
      "step": 18999
    },
    {
      "epoch": 1.8118858337402343e-05,
      "step": 18999,
      "training_step_time": 8.622597932815552
    },
    {
      "epoch": 1.811981201171875e-05,
      "grad_norm": 0.19556504487991333,
      "learning_rate": 9.473720492075893e-06,
      "loss": 0.064,
      "step": 19000
    },
    {
      "epoch": 1.811981201171875e-05,
      "model_forward_time": 1.7271606922149658,
      "step": 19000
    },
    {
      "dynamics_loss_avg": 0.04896355196833611,
      "epoch": 1.811981201171875e-05,
      "step": 19000
    },
    {
      "action_loss_avg": 0.0018697433522902429,
      "epoch": 1.811981201171875e-05,
      "step": 19000
    },
    {
      "epoch": 1.811981201171875e-05,
      "step": 19000,
      "training_step_time": 6.618285179138184
    },
    {
      "epoch": 1.8120765686035157e-05,
      "model_forward_time": 1.7257421016693115,
      "step": 19001
    },
    {
      "epoch": 1.8120765686035157e-05,
      "step": 19001,
      "training_step_time": 6.840557098388672
    },
    {
      "epoch": 1.812171936035156e-05,
      "model_forward_time": 1.7084722518920898,
      "step": 19002
    },
    {
      "epoch": 1.812171936035156e-05,
      "step": 19002,
      "training_step_time": 6.850064754486084
    },
    {
      "epoch": 1.812267303466797e-05,
      "model_forward_time": 1.7118597030639648,
      "step": 19003
    },
    {
      "epoch": 1.812267303466797e-05,
      "step": 19003,
      "training_step_time": 6.87155556678772
    },
    {
      "epoch": 1.8123626708984376e-05,
      "model_forward_time": 1.7198584079742432,
      "step": 19004
    },
    {
      "epoch": 1.8123626708984376e-05,
      "step": 19004,
      "training_step_time": 6.882253170013428
    },
    {
      "epoch": 1.812458038330078e-05,
      "model_forward_time": 1.715320110321045,
      "step": 19005
    },
    {
      "epoch": 1.812458038330078e-05,
      "step": 19005,
      "training_step_time": 6.900744915008545
    },
    {
      "epoch": 1.8125534057617188e-05,
      "model_forward_time": 1.7163255214691162,
      "step": 19006
    },
    {
      "epoch": 1.8125534057617188e-05,
      "step": 19006,
      "training_step_time": 6.879770278930664
    },
    {
      "epoch": 1.8126487731933595e-05,
      "model_forward_time": 1.7160413265228271,
      "step": 19007
    },
    {
      "epoch": 1.8126487731933595e-05,
      "step": 19007,
      "training_step_time": 6.897573471069336
    },
    {
      "epoch": 1.812744140625e-05,
      "model_forward_time": 2.7320022583007812,
      "step": 19008
    },
    {
      "epoch": 1.812744140625e-05,
      "step": 19008,
      "training_step_time": 13.569424629211426
    },
    {
      "epoch": 1.8128395080566407e-05,
      "model_forward_time": 1.7571885585784912,
      "step": 19009
    },
    {
      "epoch": 1.8128395080566407e-05,
      "step": 19009,
      "training_step_time": 7.815112590789795
    },
    {
      "epoch": 1.8129348754882814e-05,
      "grad_norm": 0.2033265233039856,
      "learning_rate": 9.472981842210769e-06,
      "loss": 0.0686,
      "step": 19010
    },
    {
      "epoch": 1.8129348754882814e-05,
      "model_forward_time": 1.708695888519287,
      "step": 19010
    },
    {
      "dynamics_loss_avg": 0.06263716667890548,
      "epoch": 1.8129348754882814e-05,
      "step": 19010
    },
    {
      "action_loss_avg": 0.0037446336704306305,
      "epoch": 1.8129348754882814e-05,
      "step": 19010
    },
    {
      "epoch": 1.8129348754882814e-05,
      "step": 19010,
      "training_step_time": 6.792800664901733
    },
    {
      "epoch": 1.8130302429199218e-05,
      "model_forward_time": 1.7116985321044922,
      "step": 19011
    },
    {
      "epoch": 1.8130302429199218e-05,
      "step": 19011,
      "training_step_time": 6.744925498962402
    },
    {
      "epoch": 1.8131256103515625e-05,
      "model_forward_time": 1.7152738571166992,
      "step": 19012
    },
    {
      "epoch": 1.8131256103515625e-05,
      "step": 19012,
      "training_step_time": 6.732851982116699
    },
    {
      "epoch": 1.8132209777832033e-05,
      "model_forward_time": 1.7164781093597412,
      "step": 19013
    },
    {
      "epoch": 1.8132209777832033e-05,
      "step": 19013,
      "training_step_time": 6.743302345275879
    },
    {
      "epoch": 1.8133163452148437e-05,
      "model_forward_time": 1.7125813961029053,
      "step": 19014
    },
    {
      "epoch": 1.8133163452148437e-05,
      "step": 19014,
      "training_step_time": 6.737420558929443
    },
    {
      "epoch": 1.8134117126464844e-05,
      "model_forward_time": 1.7224481105804443,
      "step": 19015
    },
    {
      "epoch": 1.8134117126464844e-05,
      "step": 19015,
      "training_step_time": 6.743995428085327
    },
    {
      "epoch": 1.813507080078125e-05,
      "model_forward_time": 1.7164125442504883,
      "step": 19016
    },
    {
      "epoch": 1.813507080078125e-05,
      "step": 19016,
      "training_step_time": 6.770723819732666
    },
    {
      "epoch": 1.8136024475097656e-05,
      "model_forward_time": 2.2757904529571533,
      "step": 19017
    },
    {
      "epoch": 1.8136024475097656e-05,
      "step": 19017,
      "training_step_time": 8.912376165390015
    },
    {
      "epoch": 1.8136978149414063e-05,
      "model_forward_time": 1.763185739517212,
      "step": 19018
    },
    {
      "epoch": 1.8136978149414063e-05,
      "step": 19018,
      "training_step_time": 7.7839672565460205
    },
    {
      "epoch": 1.8137931823730467e-05,
      "model_forward_time": 1.7129082679748535,
      "step": 19019
    },
    {
      "epoch": 1.8137931823730467e-05,
      "step": 19019,
      "training_step_time": 6.74042010307312
    },
    {
      "epoch": 1.8138885498046875e-05,
      "grad_norm": 0.22595681250095367,
      "learning_rate": 9.472242703187075e-06,
      "loss": 0.0676,
      "step": 19020
    },
    {
      "epoch": 1.8138885498046875e-05,
      "model_forward_time": 1.706688642501831,
      "step": 19020
    },
    {
      "dynamics_loss_avg": 0.06351588480174541,
      "epoch": 1.8138885498046875e-05,
      "step": 19020
    },
    {
      "action_loss_avg": 0.0025419259909540415,
      "epoch": 1.8138885498046875e-05,
      "step": 19020
    },
    {
      "epoch": 1.8138885498046875e-05,
      "step": 19020,
      "training_step_time": 6.721144914627075
    },
    {
      "epoch": 1.8139839172363282e-05,
      "model_forward_time": 1.7148308753967285,
      "step": 19021
    },
    {
      "epoch": 1.8139839172363282e-05,
      "step": 19021,
      "training_step_time": 6.724092960357666
    },
    {
      "epoch": 1.8140792846679686e-05,
      "model_forward_time": 1.7070839405059814,
      "step": 19022
    },
    {
      "epoch": 1.8140792846679686e-05,
      "step": 19022,
      "training_step_time": 6.7606587409973145
    },
    {
      "epoch": 1.8141746520996093e-05,
      "model_forward_time": 1.7165942192077637,
      "step": 19023
    },
    {
      "epoch": 1.8141746520996093e-05,
      "step": 19023,
      "training_step_time": 6.744377851486206
    },
    {
      "epoch": 1.81427001953125e-05,
      "model_forward_time": 1.7136964797973633,
      "step": 19024
    },
    {
      "epoch": 1.81427001953125e-05,
      "step": 19024,
      "training_step_time": 6.750035047531128
    },
    {
      "epoch": 1.8143653869628905e-05,
      "model_forward_time": 1.7057132720947266,
      "step": 19025
    },
    {
      "epoch": 1.8143653869628905e-05,
      "step": 19025,
      "training_step_time": 6.777402877807617
    },
    {
      "epoch": 1.8144607543945312e-05,
      "model_forward_time": 2.2530453205108643,
      "step": 19026
    },
    {
      "epoch": 1.8144607543945312e-05,
      "step": 19026,
      "training_step_time": 7.046358108520508
    },
    {
      "epoch": 1.814556121826172e-05,
      "model_forward_time": 1.7496063709259033,
      "step": 19027
    },
    {
      "epoch": 1.814556121826172e-05,
      "step": 19027,
      "training_step_time": 7.8651344776153564
    },
    {
      "epoch": 1.8146514892578124e-05,
      "model_forward_time": 1.7101316452026367,
      "step": 19028
    },
    {
      "epoch": 1.8146514892578124e-05,
      "step": 19028,
      "training_step_time": 7.246050119400024
    },
    {
      "epoch": 1.814746856689453e-05,
      "model_forward_time": 1.707061529159546,
      "step": 19029
    },
    {
      "epoch": 1.814746856689453e-05,
      "step": 19029,
      "training_step_time": 6.816366195678711
    },
    {
      "epoch": 1.814842224121094e-05,
      "grad_norm": 0.15756350755691528,
      "learning_rate": 9.471503075085643e-06,
      "loss": 0.0651,
      "step": 19030
    },
    {
      "epoch": 1.814842224121094e-05,
      "model_forward_time": 1.6896517276763916,
      "step": 19030
    },
    {
      "dynamics_loss_avg": 0.05452358275651932,
      "epoch": 1.814842224121094e-05,
      "step": 19030
    },
    {
      "action_loss_avg": 0.003910379827721045,
      "epoch": 1.814842224121094e-05,
      "step": 19030
    },
    {
      "epoch": 1.814842224121094e-05,
      "step": 19030,
      "training_step_time": 6.81428861618042
    },
    {
      "epoch": 1.8149375915527343e-05,
      "model_forward_time": 1.7083210945129395,
      "step": 19031
    },
    {
      "epoch": 1.8149375915527343e-05,
      "step": 19031,
      "training_step_time": 6.843206167221069
    },
    {
      "epoch": 1.815032958984375e-05,
      "model_forward_time": 1.3301067352294922,
      "step": 19032
    },
    {
      "epoch": 1.815032958984375e-05,
      "step": 19032,
      "training_step_time": 6.8235273361206055
    },
    {
      "epoch": 1.8151283264160157e-05,
      "model_forward_time": 1.7118475437164307,
      "step": 19033
    },
    {
      "epoch": 1.8151283264160157e-05,
      "step": 19033,
      "training_step_time": 6.847599506378174
    },
    {
      "epoch": 1.815223693847656e-05,
      "model_forward_time": 1.7127478122711182,
      "step": 19034
    },
    {
      "epoch": 1.815223693847656e-05,
      "step": 19034,
      "training_step_time": 6.848714828491211
    },
    {
      "epoch": 1.815319061279297e-05,
      "model_forward_time": 1.701155662536621,
      "step": 19035
    },
    {
      "epoch": 1.815319061279297e-05,
      "step": 19035,
      "training_step_time": 7.784610271453857
    },
    {
      "epoch": 1.8154144287109376e-05,
      "model_forward_time": 2.2751736640930176,
      "step": 19036
    },
    {
      "epoch": 1.8154144287109376e-05,
      "step": 19036,
      "training_step_time": 7.910867691040039
    },
    {
      "epoch": 1.815509796142578e-05,
      "model_forward_time": 1.7189455032348633,
      "step": 19037
    },
    {
      "epoch": 1.815509796142578e-05,
      "step": 19037,
      "training_step_time": 7.739755153656006
    },
    {
      "epoch": 1.8156051635742188e-05,
      "model_forward_time": 1.7041945457458496,
      "step": 19038
    },
    {
      "epoch": 1.8156051635742188e-05,
      "step": 19038,
      "training_step_time": 6.8352885246276855
    },
    {
      "epoch": 1.8157005310058595e-05,
      "model_forward_time": 1.7169513702392578,
      "step": 19039
    },
    {
      "epoch": 1.8157005310058595e-05,
      "step": 19039,
      "training_step_time": 6.854088544845581
    },
    {
      "epoch": 1.8157958984375e-05,
      "grad_norm": 0.21737655997276306,
      "learning_rate": 9.470762957987359e-06,
      "loss": 0.0627,
      "step": 19040
    },
    {
      "epoch": 1.8157958984375e-05,
      "model_forward_time": 1.3149750232696533,
      "step": 19040
    },
    {
      "dynamics_loss_avg": 0.06243456918746233,
      "epoch": 1.8157958984375e-05,
      "step": 19040
    },
    {
      "action_loss_avg": 0.0038646286178845913,
      "epoch": 1.8157958984375e-05,
      "step": 19040
    },
    {
      "epoch": 1.8157958984375e-05,
      "step": 19040,
      "training_step_time": 6.821209192276001
    },
    {
      "epoch": 1.8158912658691407e-05,
      "model_forward_time": 1.7061817646026611,
      "step": 19041
    },
    {
      "epoch": 1.8158912658691407e-05,
      "step": 19041,
      "training_step_time": 6.800995826721191
    },
    {
      "epoch": 1.8159866333007814e-05,
      "model_forward_time": 1.7179934978485107,
      "step": 19042
    },
    {
      "epoch": 1.8159866333007814e-05,
      "step": 19042,
      "training_step_time": 6.79244327545166
    },
    {
      "epoch": 1.8160820007324218e-05,
      "model_forward_time": 1.709801435470581,
      "step": 19043
    },
    {
      "epoch": 1.8160820007324218e-05,
      "step": 19043,
      "training_step_time": 6.861819505691528
    },
    {
      "epoch": 1.8161773681640625e-05,
      "model_forward_time": 1.7079501152038574,
      "step": 19044
    },
    {
      "epoch": 1.8161773681640625e-05,
      "step": 19044,
      "training_step_time": 6.866763591766357
    },
    {
      "epoch": 1.8162727355957033e-05,
      "model_forward_time": 2.2792131900787354,
      "step": 19045
    },
    {
      "epoch": 1.8162727355957033e-05,
      "step": 19045,
      "training_step_time": 10.674851655960083
    },
    {
      "epoch": 1.8163681030273437e-05,
      "model_forward_time": 1.70259690284729,
      "step": 19046
    },
    {
      "epoch": 1.8163681030273437e-05,
      "step": 19046,
      "training_step_time": 7.685245990753174
    },
    {
      "epoch": 1.8164634704589844e-05,
      "model_forward_time": 1.7101194858551025,
      "step": 19047
    },
    {
      "epoch": 1.8164634704589844e-05,
      "step": 19047,
      "training_step_time": 6.835144758224487
    },
    {
      "epoch": 1.816558837890625e-05,
      "model_forward_time": 1.7103779315948486,
      "step": 19048
    },
    {
      "epoch": 1.816558837890625e-05,
      "step": 19048,
      "training_step_time": 6.740922212600708
    },
    {
      "epoch": 1.8166542053222656e-05,
      "model_forward_time": 1.712576150894165,
      "step": 19049
    },
    {
      "epoch": 1.8166542053222656e-05,
      "step": 19049,
      "training_step_time": 6.877128601074219
    },
    {
      "epoch": 1.8167495727539063e-05,
      "grad_norm": 0.1823749989271164,
      "learning_rate": 9.470022351973158e-06,
      "loss": 0.062,
      "step": 19050
    },
    {
      "epoch": 1.8167495727539063e-05,
      "model_forward_time": 1.7375965118408203,
      "step": 19050
    },
    {
      "dynamics_loss_avg": 0.06950685605406762,
      "epoch": 1.8167495727539063e-05,
      "step": 19050
    },
    {
      "action_loss_avg": 0.0029799018520861864,
      "epoch": 1.8167495727539063e-05,
      "step": 19050
    },
    {
      "epoch": 1.8167495727539063e-05,
      "step": 19050,
      "training_step_time": 6.894779920578003
    },
    {
      "epoch": 1.8168449401855467e-05,
      "model_forward_time": 1.7034103870391846,
      "step": 19051
    },
    {
      "epoch": 1.8168449401855467e-05,
      "step": 19051,
      "training_step_time": 6.830897092819214
    },
    {
      "epoch": 1.8169403076171875e-05,
      "model_forward_time": 1.7198448181152344,
      "step": 19052
    },
    {
      "epoch": 1.8169403076171875e-05,
      "step": 19052,
      "training_step_time": 6.904867887496948
    },
    {
      "epoch": 1.8170356750488282e-05,
      "model_forward_time": 1.722529411315918,
      "step": 19053
    },
    {
      "epoch": 1.8170356750488282e-05,
      "step": 19053,
      "training_step_time": 6.877671957015991
    },
    {
      "epoch": 1.8171310424804686e-05,
      "model_forward_time": 2.270360231399536,
      "step": 19054
    },
    {
      "epoch": 1.8171310424804686e-05,
      "step": 19054,
      "training_step_time": 9.950448036193848
    },
    {
      "epoch": 1.8172264099121093e-05,
      "model_forward_time": 1.7639918327331543,
      "step": 19055
    },
    {
      "epoch": 1.8172264099121093e-05,
      "step": 19055,
      "training_step_time": 7.946908712387085
    },
    {
      "epoch": 1.81732177734375e-05,
      "model_forward_time": 1.7284297943115234,
      "step": 19056
    },
    {
      "epoch": 1.81732177734375e-05,
      "step": 19056,
      "training_step_time": 6.864761114120483
    },
    {
      "epoch": 1.8174171447753905e-05,
      "model_forward_time": 1.7262263298034668,
      "step": 19057
    },
    {
      "epoch": 1.8174171447753905e-05,
      "step": 19057,
      "training_step_time": 6.862550735473633
    },
    {
      "epoch": 1.8175125122070312e-05,
      "model_forward_time": 1.7223021984100342,
      "step": 19058
    },
    {
      "epoch": 1.8175125122070312e-05,
      "step": 19058,
      "training_step_time": 6.891628265380859
    },
    {
      "epoch": 1.817607879638672e-05,
      "model_forward_time": 1.7317779064178467,
      "step": 19059
    },
    {
      "epoch": 1.817607879638672e-05,
      "step": 19059,
      "training_step_time": 6.896804094314575
    },
    {
      "epoch": 1.8177032470703124e-05,
      "grad_norm": 0.1754380315542221,
      "learning_rate": 9.469281257124033e-06,
      "loss": 0.0654,
      "step": 19060
    },
    {
      "epoch": 1.8177032470703124e-05,
      "model_forward_time": 1.7208058834075928,
      "step": 19060
    },
    {
      "dynamics_loss_avg": 0.07361808661371469,
      "epoch": 1.8177032470703124e-05,
      "step": 19060
    },
    {
      "action_loss_avg": 0.0038755839341320097,
      "epoch": 1.8177032470703124e-05,
      "step": 19060
    },
    {
      "epoch": 1.8177032470703124e-05,
      "step": 19060,
      "training_step_time": 6.8823511600494385
    },
    {
      "epoch": 1.817798614501953e-05,
      "model_forward_time": 1.7348544597625732,
      "step": 19061
    },
    {
      "epoch": 1.817798614501953e-05,
      "step": 19061,
      "training_step_time": 6.897114038467407
    },
    {
      "epoch": 1.817893981933594e-05,
      "model_forward_time": 1.7302641868591309,
      "step": 19062
    },
    {
      "epoch": 1.817893981933594e-05,
      "step": 19062,
      "training_step_time": 6.910421371459961
    },
    {
      "epoch": 1.8179893493652343e-05,
      "model_forward_time": 2.2753751277923584,
      "step": 19063
    },
    {
      "epoch": 1.8179893493652343e-05,
      "step": 19063,
      "training_step_time": 8.857421159744263
    },
    {
      "epoch": 1.818084716796875e-05,
      "model_forward_time": 1.7425227165222168,
      "step": 19064
    },
    {
      "epoch": 1.818084716796875e-05,
      "step": 19064,
      "training_step_time": 7.9751996994018555
    },
    {
      "epoch": 1.8181800842285157e-05,
      "model_forward_time": 1.737368106842041,
      "step": 19065
    },
    {
      "epoch": 1.8181800842285157e-05,
      "step": 19065,
      "training_step_time": 7.572913646697998
    },
    {
      "epoch": 1.818275451660156e-05,
      "model_forward_time": 1.736114740371704,
      "step": 19066
    },
    {
      "epoch": 1.818275451660156e-05,
      "step": 19066,
      "training_step_time": 6.888041257858276
    },
    {
      "epoch": 1.818370819091797e-05,
      "model_forward_time": 1.7412047386169434,
      "step": 19067
    },
    {
      "epoch": 1.818370819091797e-05,
      "step": 19067,
      "training_step_time": 6.918797731399536
    },
    {
      "epoch": 1.8184661865234376e-05,
      "model_forward_time": 1.7342755794525146,
      "step": 19068
    },
    {
      "epoch": 1.8184661865234376e-05,
      "step": 19068,
      "training_step_time": 6.9632887840271
    },
    {
      "epoch": 1.818561553955078e-05,
      "model_forward_time": 1.7353715896606445,
      "step": 19069
    },
    {
      "epoch": 1.818561553955078e-05,
      "step": 19069,
      "training_step_time": 6.892104625701904
    },
    {
      "epoch": 1.8186569213867188e-05,
      "grad_norm": 0.19171500205993652,
      "learning_rate": 9.46853967352103e-06,
      "loss": 0.0613,
      "step": 19070
    },
    {
      "epoch": 1.8186569213867188e-05,
      "model_forward_time": 1.7245686054229736,
      "step": 19070
    },
    {
      "dynamics_loss_avg": 0.05906922612339258,
      "epoch": 1.8186569213867188e-05,
      "step": 19070
    },
    {
      "action_loss_avg": 0.0032153025968000293,
      "epoch": 1.8186569213867188e-05,
      "step": 19070
    },
    {
      "epoch": 1.8186569213867188e-05,
      "step": 19070,
      "training_step_time": 6.900160074234009
    },
    {
      "epoch": 1.8187522888183595e-05,
      "model_forward_time": 1.7307357788085938,
      "step": 19071
    },
    {
      "epoch": 1.8187522888183595e-05,
      "step": 19071,
      "training_step_time": 6.907562732696533
    },
    {
      "epoch": 1.81884765625e-05,
      "model_forward_time": 1.7285590171813965,
      "step": 19072
    },
    {
      "epoch": 1.81884765625e-05,
      "step": 19072,
      "training_step_time": 8.363860368728638
    },
    {
      "epoch": 1.8189430236816407e-05,
      "model_forward_time": 2.307199239730835,
      "step": 19073
    },
    {
      "epoch": 1.8189430236816407e-05,
      "step": 19073,
      "training_step_time": 10.794287204742432
    },
    {
      "epoch": 1.8190383911132814e-05,
      "model_forward_time": 1.7264597415924072,
      "step": 19074
    },
    {
      "epoch": 1.8190383911132814e-05,
      "step": 19074,
      "training_step_time": 7.601800441741943
    },
    {
      "epoch": 1.8191337585449218e-05,
      "model_forward_time": 1.7239902019500732,
      "step": 19075
    },
    {
      "epoch": 1.8191337585449218e-05,
      "step": 19075,
      "training_step_time": 6.921738386154175
    },
    {
      "epoch": 1.8192291259765625e-05,
      "model_forward_time": 1.734375,
      "step": 19076
    },
    {
      "epoch": 1.8192291259765625e-05,
      "step": 19076,
      "training_step_time": 6.89453387260437
    },
    {
      "epoch": 1.8193244934082033e-05,
      "model_forward_time": 1.7307937145233154,
      "step": 19077
    },
    {
      "epoch": 1.8193244934082033e-05,
      "step": 19077,
      "training_step_time": 6.8506810665130615
    },
    {
      "epoch": 1.8194198608398437e-05,
      "model_forward_time": 1.7278449535369873,
      "step": 19078
    },
    {
      "epoch": 1.8194198608398437e-05,
      "step": 19078,
      "training_step_time": 6.908115863800049
    },
    {
      "epoch": 1.8195152282714844e-05,
      "model_forward_time": 1.7336766719818115,
      "step": 19079
    },
    {
      "epoch": 1.8195152282714844e-05,
      "step": 19079,
      "training_step_time": 6.886256694793701
    },
    {
      "epoch": 1.819610595703125e-05,
      "grad_norm": 0.177351713180542,
      "learning_rate": 9.467797601245246e-06,
      "loss": 0.0586,
      "step": 19080
    },
    {
      "epoch": 1.819610595703125e-05,
      "model_forward_time": 1.8205146789550781,
      "step": 19080
    },
    {
      "dynamics_loss_avg": 0.05761176683008671,
      "epoch": 1.819610595703125e-05,
      "step": 19080
    },
    {
      "action_loss_avg": 0.002161172340856865,
      "epoch": 1.819610595703125e-05,
      "step": 19080
    },
    {
      "epoch": 1.819610595703125e-05,
      "step": 19080,
      "training_step_time": 6.8577117919921875
    },
    {
      "epoch": 1.8197059631347656e-05,
      "model_forward_time": 1.8296129703521729,
      "step": 19081
    },
    {
      "epoch": 1.8197059631347656e-05,
      "step": 19081,
      "training_step_time": 7.434266567230225
    },
    {
      "epoch": 1.8198013305664063e-05,
      "model_forward_time": 2.4097933769226074,
      "step": 19082
    },
    {
      "epoch": 1.8198013305664063e-05,
      "step": 19082,
      "training_step_time": 8.657662153244019
    },
    {
      "epoch": 1.8198966979980467e-05,
      "model_forward_time": 1.8331530094146729,
      "step": 19083
    },
    {
      "epoch": 1.8198966979980467e-05,
      "step": 19083,
      "training_step_time": 7.613522529602051
    },
    {
      "epoch": 1.8199920654296875e-05,
      "model_forward_time": 1.8364636898040771,
      "step": 19084
    },
    {
      "epoch": 1.8199920654296875e-05,
      "step": 19084,
      "training_step_time": 7.300620794296265
    },
    {
      "epoch": 1.8200874328613282e-05,
      "model_forward_time": 1.8252859115600586,
      "step": 19085
    },
    {
      "epoch": 1.8200874328613282e-05,
      "step": 19085,
      "training_step_time": 6.733698129653931
    },
    {
      "epoch": 1.8201828002929686e-05,
      "model_forward_time": 1.4575631618499756,
      "step": 19086
    },
    {
      "epoch": 1.8201828002929686e-05,
      "step": 19086,
      "training_step_time": 6.747581481933594
    },
    {
      "epoch": 1.8202781677246094e-05,
      "model_forward_time": 1.4539167881011963,
      "step": 19087
    },
    {
      "epoch": 1.8202781677246094e-05,
      "step": 19087,
      "training_step_time": 6.768096208572388
    },
    {
      "epoch": 1.82037353515625e-05,
      "model_forward_time": 1.839689016342163,
      "step": 19088
    },
    {
      "epoch": 1.82037353515625e-05,
      "step": 19088,
      "training_step_time": 6.760805130004883
    },
    {
      "epoch": 1.8204689025878905e-05,
      "model_forward_time": 1.8394265174865723,
      "step": 19089
    },
    {
      "epoch": 1.8204689025878905e-05,
      "step": 19089,
      "training_step_time": 6.769423246383667
    },
    {
      "epoch": 1.8205642700195312e-05,
      "grad_norm": 0.14678792655467987,
      "learning_rate": 9.467055040377835e-06,
      "loss": 0.0606,
      "step": 19090
    },
    {
      "epoch": 1.8205642700195312e-05,
      "model_forward_time": 1.8447585105895996,
      "step": 19090
    },
    {
      "dynamics_loss_avg": 0.04938450194895268,
      "epoch": 1.8205642700195312e-05,
      "step": 19090
    },
    {
      "action_loss_avg": 0.006875952379778028,
      "epoch": 1.8205642700195312e-05,
      "step": 19090
    },
    {
      "epoch": 1.8205642700195312e-05,
      "step": 19090,
      "training_step_time": 7.313180208206177
    },
    {
      "epoch": 1.820659637451172e-05,
      "model_forward_time": 2.3910555839538574,
      "step": 19091
    },
    {
      "epoch": 1.820659637451172e-05,
      "step": 19091,
      "training_step_time": 12.530187606811523
    },
    {
      "epoch": 1.8207550048828124e-05,
      "model_forward_time": 1.838482141494751,
      "step": 19092
    },
    {
      "epoch": 1.8207550048828124e-05,
      "step": 19092,
      "training_step_time": 7.57494592666626
    },
    {
      "epoch": 1.820850372314453e-05,
      "model_forward_time": 1.8298885822296143,
      "step": 19093
    },
    {
      "epoch": 1.820850372314453e-05,
      "step": 19093,
      "training_step_time": 7.358221054077148
    },
    {
      "epoch": 1.820945739746094e-05,
      "model_forward_time": 1.8244907855987549,
      "step": 19094
    },
    {
      "epoch": 1.820945739746094e-05,
      "step": 19094,
      "training_step_time": 6.762706995010376
    },
    {
      "epoch": 1.8210411071777343e-05,
      "model_forward_time": 1.8303978443145752,
      "step": 19095
    },
    {
      "epoch": 1.8210411071777343e-05,
      "step": 19095,
      "training_step_time": 6.718636512756348
    },
    {
      "epoch": 1.821136474609375e-05,
      "model_forward_time": 1.8262321949005127,
      "step": 19096
    },
    {
      "epoch": 1.821136474609375e-05,
      "step": 19096,
      "training_step_time": 6.71906304359436
    },
    {
      "epoch": 1.8212318420410157e-05,
      "model_forward_time": 1.828965425491333,
      "step": 19097
    },
    {
      "epoch": 1.8212318420410157e-05,
      "step": 19097,
      "training_step_time": 6.766406297683716
    },
    {
      "epoch": 1.821327209472656e-05,
      "model_forward_time": 1.830596923828125,
      "step": 19098
    },
    {
      "epoch": 1.821327209472656e-05,
      "step": 19098,
      "training_step_time": 6.764925003051758
    },
    {
      "epoch": 1.821422576904297e-05,
      "model_forward_time": 1.8322367668151855,
      "step": 19099
    },
    {
      "epoch": 1.821422576904297e-05,
      "step": 19099,
      "training_step_time": 7.352971076965332
    },
    {
      "epoch": 1.8215179443359376e-05,
      "grad_norm": 0.1891254335641861,
      "learning_rate": 9.466311991e-06,
      "loss": 0.0605,
      "step": 19100
    },
    {
      "epoch": 1.8215179443359376e-05,
      "model_forward_time": 2.3693511486053467,
      "step": 19100
    },
    {
      "dynamics_loss_avg": 0.048340775165706874,
      "epoch": 1.8215179443359376e-05,
      "step": 19100
    },
    {
      "action_loss_avg": 0.0013567650050390512,
      "epoch": 1.8215179443359376e-05,
      "step": 19100
    },
    {
      "epoch": 1.8215179443359376e-05,
      "step": 19100,
      "training_step_time": 9.408050775527954
    },
    {
      "epoch": 1.821613311767578e-05,
      "model_forward_time": 1.8267543315887451,
      "step": 19101
    },
    {
      "epoch": 1.821613311767578e-05,
      "step": 19101,
      "training_step_time": 7.778344631195068
    },
    {
      "epoch": 1.8217086791992188e-05,
      "model_forward_time": 1.8305492401123047,
      "step": 19102
    },
    {
      "epoch": 1.8217086791992188e-05,
      "step": 19102,
      "training_step_time": 7.407078266143799
    },
    {
      "epoch": 1.8218040466308595e-05,
      "model_forward_time": 1.8269267082214355,
      "step": 19103
    },
    {
      "epoch": 1.8218040466308595e-05,
      "step": 19103,
      "training_step_time": 6.7256669998168945
    },
    {
      "epoch": 1.8218994140625e-05,
      "model_forward_time": 1.4626264572143555,
      "step": 19104
    },
    {
      "epoch": 1.8218994140625e-05,
      "step": 19104,
      "training_step_time": 6.760777473449707
    },
    {
      "epoch": 1.8219947814941407e-05,
      "model_forward_time": 1.8314266204833984,
      "step": 19105
    },
    {
      "epoch": 1.8219947814941407e-05,
      "step": 19105,
      "training_step_time": 6.690565824508667
    },
    {
      "epoch": 1.8220901489257814e-05,
      "model_forward_time": 1.8334050178527832,
      "step": 19106
    },
    {
      "epoch": 1.8220901489257814e-05,
      "step": 19106,
      "training_step_time": 6.758682012557983
    },
    {
      "epoch": 1.8221855163574218e-05,
      "model_forward_time": 1.8320729732513428,
      "step": 19107
    },
    {
      "epoch": 1.8221855163574218e-05,
      "step": 19107,
      "training_step_time": 6.714583158493042
    },
    {
      "epoch": 1.8222808837890626e-05,
      "model_forward_time": 1.8259551525115967,
      "step": 19108
    },
    {
      "epoch": 1.8222808837890626e-05,
      "step": 19108,
      "training_step_time": 7.395608186721802
    },
    {
      "epoch": 1.8223762512207033e-05,
      "model_forward_time": 1.825317621231079,
      "step": 19109
    },
    {
      "epoch": 1.8223762512207033e-05,
      "step": 19109,
      "training_step_time": 9.285314559936523
    },
    {
      "epoch": 1.8224716186523437e-05,
      "grad_norm": 0.22280225157737732,
      "learning_rate": 9.465568453193e-06,
      "loss": 0.0613,
      "step": 19110
    },
    {
      "epoch": 1.8224716186523437e-05,
      "model_forward_time": 2.3833391666412354,
      "step": 19110
    },
    {
      "dynamics_loss_avg": 0.054022437706589696,
      "epoch": 1.8224716186523437e-05,
      "step": 19110
    },
    {
      "action_loss_avg": 0.003234517341479659,
      "epoch": 1.8224716186523437e-05,
      "step": 19110
    },
    {
      "epoch": 1.8224716186523437e-05,
      "step": 19110,
      "training_step_time": 9.00626254081726
    },
    {
      "epoch": 1.8225669860839844e-05,
      "model_forward_time": 1.8326098918914795,
      "step": 19111
    },
    {
      "epoch": 1.8225669860839844e-05,
      "step": 19111,
      "training_step_time": 7.520040035247803
    },
    {
      "epoch": 1.822662353515625e-05,
      "model_forward_time": 1.8237614631652832,
      "step": 19112
    },
    {
      "epoch": 1.822662353515625e-05,
      "step": 19112,
      "training_step_time": 7.18155574798584
    },
    {
      "epoch": 1.8227577209472656e-05,
      "model_forward_time": 1.8348019123077393,
      "step": 19113
    },
    {
      "epoch": 1.8227577209472656e-05,
      "step": 19113,
      "training_step_time": 6.7987000942230225
    },
    {
      "epoch": 1.8228530883789063e-05,
      "model_forward_time": 1.8279051780700684,
      "step": 19114
    },
    {
      "epoch": 1.8228530883789063e-05,
      "step": 19114,
      "training_step_time": 6.780013799667358
    },
    {
      "epoch": 1.8229484558105467e-05,
      "model_forward_time": 1.8356764316558838,
      "step": 19115
    },
    {
      "epoch": 1.8229484558105467e-05,
      "step": 19115,
      "training_step_time": 6.809006214141846
    },
    {
      "epoch": 1.8230438232421875e-05,
      "model_forward_time": 1.830754041671753,
      "step": 19116
    },
    {
      "epoch": 1.8230438232421875e-05,
      "step": 19116,
      "training_step_time": 6.814417600631714
    },
    {
      "epoch": 1.8231391906738282e-05,
      "model_forward_time": 1.828852653503418,
      "step": 19117
    },
    {
      "epoch": 1.8231391906738282e-05,
      "step": 19117,
      "training_step_time": 7.356402158737183
    },
    {
      "epoch": 1.8232345581054686e-05,
      "model_forward_time": 1.8341379165649414,
      "step": 19118
    },
    {
      "epoch": 1.8232345581054686e-05,
      "step": 19118,
      "training_step_time": 9.686382293701172
    },
    {
      "epoch": 1.8233299255371094e-05,
      "model_forward_time": 2.3923988342285156,
      "step": 19119
    },
    {
      "epoch": 1.8233299255371094e-05,
      "step": 19119,
      "training_step_time": 7.78922700881958
    },
    {
      "epoch": 1.82342529296875e-05,
      "grad_norm": 0.20718103647232056,
      "learning_rate": 9.464824427038149e-06,
      "loss": 0.0633,
      "step": 19120
    },
    {
      "epoch": 1.82342529296875e-05,
      "model_forward_time": 1.8348846435546875,
      "step": 19120
    },
    {
      "dynamics_loss_avg": 0.04846965353935957,
      "epoch": 1.82342529296875e-05,
      "step": 19120
    },
    {
      "action_loss_avg": 0.0026334302907343953,
      "epoch": 1.82342529296875e-05,
      "step": 19120
    },
    {
      "epoch": 1.82342529296875e-05,
      "step": 19120,
      "training_step_time": 7.601787805557251
    },
    {
      "epoch": 1.8235206604003905e-05,
      "model_forward_time": 1.8286635875701904,
      "step": 19121
    },
    {
      "epoch": 1.8235206604003905e-05,
      "step": 19121,
      "training_step_time": 7.404496669769287
    },
    {
      "epoch": 1.8236160278320312e-05,
      "model_forward_time": 1.829723596572876,
      "step": 19122
    },
    {
      "epoch": 1.8236160278320312e-05,
      "step": 19122,
      "training_step_time": 6.811180591583252
    },
    {
      "epoch": 1.823711395263672e-05,
      "model_forward_time": 1.8285391330718994,
      "step": 19123
    },
    {
      "epoch": 1.823711395263672e-05,
      "step": 19123,
      "training_step_time": 6.782011270523071
    },
    {
      "epoch": 1.8238067626953124e-05,
      "model_forward_time": 1.828855276107788,
      "step": 19124
    },
    {
      "epoch": 1.8238067626953124e-05,
      "step": 19124,
      "training_step_time": 6.771181344985962
    },
    {
      "epoch": 1.823902130126953e-05,
      "model_forward_time": 1.8084406852722168,
      "step": 19125
    },
    {
      "epoch": 1.823902130126953e-05,
      "step": 19125,
      "training_step_time": 6.762543678283691
    },
    {
      "epoch": 1.823997497558594e-05,
      "model_forward_time": 1.8239970207214355,
      "step": 19126
    },
    {
      "epoch": 1.823997497558594e-05,
      "step": 19126,
      "training_step_time": 7.352827548980713
    },
    {
      "epoch": 1.8240928649902343e-05,
      "model_forward_time": 1.8246955871582031,
      "step": 19127
    },
    {
      "epoch": 1.8240928649902343e-05,
      "step": 19127,
      "training_step_time": 8.27831745147705
    },
    {
      "epoch": 1.824188232421875e-05,
      "model_forward_time": 2.379596471786499,
      "step": 19128
    },
    {
      "epoch": 1.824188232421875e-05,
      "step": 19128,
      "training_step_time": 8.699295282363892
    },
    {
      "epoch": 1.8242835998535158e-05,
      "model_forward_time": 1.8286323547363281,
      "step": 19129
    },
    {
      "epoch": 1.8242835998535158e-05,
      "step": 19129,
      "training_step_time": 7.812087774276733
    },
    {
      "epoch": 1.824378967285156e-05,
      "grad_norm": 0.25522851943969727,
      "learning_rate": 9.46407991261681e-06,
      "loss": 0.0665,
      "step": 19130
    },
    {
      "epoch": 1.824378967285156e-05,
      "model_forward_time": 1.837470531463623,
      "step": 19130
    },
    {
      "dynamics_loss_avg": 0.06774097625166178,
      "epoch": 1.824378967285156e-05,
      "step": 19130
    },
    {
      "action_loss_avg": 0.005117026309017092,
      "epoch": 1.824378967285156e-05,
      "step": 19130
    },
    {
      "epoch": 1.824378967285156e-05,
      "step": 19130,
      "training_step_time": 7.379584312438965
    },
    {
      "epoch": 1.824474334716797e-05,
      "model_forward_time": 1.828744649887085,
      "step": 19131
    },
    {
      "epoch": 1.824474334716797e-05,
      "step": 19131,
      "training_step_time": 7.226234436035156
    },
    {
      "epoch": 1.8245697021484376e-05,
      "model_forward_time": 1.832197666168213,
      "step": 19132
    },
    {
      "epoch": 1.8245697021484376e-05,
      "step": 19132,
      "training_step_time": 6.729037523269653
    },
    {
      "epoch": 1.824665069580078e-05,
      "model_forward_time": 1.8338079452514648,
      "step": 19133
    },
    {
      "epoch": 1.824665069580078e-05,
      "step": 19133,
      "training_step_time": 6.7558159828186035
    },
    {
      "epoch": 1.8247604370117188e-05,
      "model_forward_time": 1.8364965915679932,
      "step": 19134
    },
    {
      "epoch": 1.8247604370117188e-05,
      "step": 19134,
      "training_step_time": 6.743889093399048
    },
    {
      "epoch": 1.8248558044433595e-05,
      "model_forward_time": 1.8383824825286865,
      "step": 19135
    },
    {
      "epoch": 1.8248558044433595e-05,
      "step": 19135,
      "training_step_time": 7.270936489105225
    },
    {
      "epoch": 1.824951171875e-05,
      "model_forward_time": 1.454249382019043,
      "step": 19136
    },
    {
      "epoch": 1.824951171875e-05,
      "step": 19136,
      "training_step_time": 9.260517597198486
    },
    {
      "epoch": 1.8250465393066407e-05,
      "model_forward_time": 2.344862222671509,
      "step": 19137
    },
    {
      "epoch": 1.8250465393066407e-05,
      "step": 19137,
      "training_step_time": 11.341400861740112
    },
    {
      "epoch": 1.8251419067382814e-05,
      "model_forward_time": 1.8225319385528564,
      "step": 19138
    },
    {
      "epoch": 1.8251419067382814e-05,
      "step": 19138,
      "training_step_time": 8.689380407333374
    },
    {
      "epoch": 1.8252372741699218e-05,
      "model_forward_time": 1.8298454284667969,
      "step": 19139
    },
    {
      "epoch": 1.8252372741699218e-05,
      "step": 19139,
      "training_step_time": 7.359089136123657
    },
    {
      "epoch": 1.8253326416015626e-05,
      "grad_norm": 0.19173628091812134,
      "learning_rate": 9.463334910010404e-06,
      "loss": 0.064,
      "step": 19140
    },
    {
      "epoch": 1.8253326416015626e-05,
      "model_forward_time": 1.829014539718628,
      "step": 19140
    },
    {
      "dynamics_loss_avg": 0.04797523748129606,
      "epoch": 1.8253326416015626e-05,
      "step": 19140
    },
    {
      "action_loss_avg": 0.0028005207946989684,
      "epoch": 1.8253326416015626e-05,
      "step": 19140
    },
    {
      "epoch": 1.8253326416015626e-05,
      "step": 19140,
      "training_step_time": 7.20099401473999
    },
    {
      "epoch": 1.825428009033203e-05,
      "model_forward_time": 1.8225982189178467,
      "step": 19141
    },
    {
      "epoch": 1.825428009033203e-05,
      "step": 19141,
      "training_step_time": 6.7547385692596436
    },
    {
      "epoch": 1.8255233764648437e-05,
      "model_forward_time": 1.8238763809204102,
      "step": 19142
    },
    {
      "epoch": 1.8255233764648437e-05,
      "step": 19142,
      "training_step_time": 6.767072677612305
    },
    {
      "epoch": 1.8256187438964844e-05,
      "model_forward_time": 1.8208911418914795,
      "step": 19143
    },
    {
      "epoch": 1.8256187438964844e-05,
      "step": 19143,
      "training_step_time": 6.7440149784088135
    },
    {
      "epoch": 1.825714111328125e-05,
      "model_forward_time": 1.8169586658477783,
      "step": 19144
    },
    {
      "epoch": 1.825714111328125e-05,
      "step": 19144,
      "training_step_time": 6.878437280654907
    },
    {
      "epoch": 1.8258094787597656e-05,
      "model_forward_time": 1.8161942958831787,
      "step": 19145
    },
    {
      "epoch": 1.8258094787597656e-05,
      "step": 19145,
      "training_step_time": 8.377910375595093
    },
    {
      "epoch": 1.8259048461914063e-05,
      "model_forward_time": 1.8202502727508545,
      "step": 19146
    },
    {
      "epoch": 1.8259048461914063e-05,
      "step": 19146,
      "training_step_time": 10.138841390609741
    },
    {
      "epoch": 1.8260002136230467e-05,
      "model_forward_time": 1.8029396533966064,
      "step": 19147
    },
    {
      "epoch": 1.8260002136230467e-05,
      "step": 19147,
      "training_step_time": 7.84550404548645
    },
    {
      "epoch": 1.8260955810546875e-05,
      "model_forward_time": 1.831207036972046,
      "step": 19148
    },
    {
      "epoch": 1.8260955810546875e-05,
      "step": 19148,
      "training_step_time": 7.394439697265625
    },
    {
      "epoch": 1.8261909484863282e-05,
      "model_forward_time": 1.8172342777252197,
      "step": 19149
    },
    {
      "epoch": 1.8261909484863282e-05,
      "step": 19149,
      "training_step_time": 7.278384685516357
    },
    {
      "epoch": 1.8262863159179686e-05,
      "grad_norm": 0.18059104681015015,
      "learning_rate": 9.462589419300404e-06,
      "loss": 0.0685,
      "step": 19150
    },
    {
      "epoch": 1.8262863159179686e-05,
      "model_forward_time": 1.8239891529083252,
      "step": 19150
    },
    {
      "dynamics_loss_avg": 0.046236013807356355,
      "epoch": 1.8262863159179686e-05,
      "step": 19150
    },
    {
      "action_loss_avg": 0.0033550789405126123,
      "epoch": 1.8262863159179686e-05,
      "step": 19150
    },
    {
      "epoch": 1.8262863159179686e-05,
      "step": 19150,
      "training_step_time": 6.76799464225769
    },
    {
      "epoch": 1.8263816833496094e-05,
      "model_forward_time": 1.8234517574310303,
      "step": 19151
    },
    {
      "epoch": 1.8263816833496094e-05,
      "step": 19151,
      "training_step_time": 6.750580787658691
    },
    {
      "epoch": 1.82647705078125e-05,
      "model_forward_time": 1.8276982307434082,
      "step": 19152
    },
    {
      "epoch": 1.82647705078125e-05,
      "step": 19152,
      "training_step_time": 6.779659032821655
    },
    {
      "epoch": 1.8265724182128905e-05,
      "model_forward_time": 1.8339500427246094,
      "step": 19153
    },
    {
      "epoch": 1.8265724182128905e-05,
      "step": 19153,
      "training_step_time": 6.888602018356323
    },
    {
      "epoch": 1.8266677856445312e-05,
      "model_forward_time": 1.8161828517913818,
      "step": 19154
    },
    {
      "epoch": 1.8266677856445312e-05,
      "step": 19154,
      "training_step_time": 8.13392448425293
    },
    {
      "epoch": 1.826763153076172e-05,
      "model_forward_time": 1.820202350616455,
      "step": 19155
    },
    {
      "epoch": 1.826763153076172e-05,
      "step": 19155,
      "training_step_time": 8.960598230361938
    },
    {
      "epoch": 1.8268585205078124e-05,
      "model_forward_time": 2.39906644821167,
      "step": 19156
    },
    {
      "epoch": 1.8268585205078124e-05,
      "step": 19156,
      "training_step_time": 9.126402854919434
    },
    {
      "epoch": 1.826953887939453e-05,
      "model_forward_time": 1.8295249938964844,
      "step": 19157
    },
    {
      "epoch": 1.826953887939453e-05,
      "step": 19157,
      "training_step_time": 7.514235973358154
    },
    {
      "epoch": 1.827049255371094e-05,
      "model_forward_time": 1.826878309249878,
      "step": 19158
    },
    {
      "epoch": 1.827049255371094e-05,
      "step": 19158,
      "training_step_time": 7.214137077331543
    },
    {
      "epoch": 1.8271446228027343e-05,
      "model_forward_time": 1.8249003887176514,
      "step": 19159
    },
    {
      "epoch": 1.8271446228027343e-05,
      "step": 19159,
      "training_step_time": 6.747186183929443
    },
    {
      "epoch": 1.827239990234375e-05,
      "grad_norm": 0.29984962940216064,
      "learning_rate": 9.461843440568333e-06,
      "loss": 0.0637,
      "step": 19160
    },
    {
      "epoch": 1.827239990234375e-05,
      "model_forward_time": 1.8224799633026123,
      "step": 19160
    },
    {
      "dynamics_loss_avg": 0.06017179489135742,
      "epoch": 1.827239990234375e-05,
      "step": 19160
    },
    {
      "action_loss_avg": 0.002226777677424252,
      "epoch": 1.827239990234375e-05,
      "step": 19160
    },
    {
      "epoch": 1.827239990234375e-05,
      "step": 19160,
      "training_step_time": 6.739112615585327
    },
    {
      "epoch": 1.8273353576660158e-05,
      "model_forward_time": 1.824474811553955,
      "step": 19161
    },
    {
      "epoch": 1.8273353576660158e-05,
      "step": 19161,
      "training_step_time": 6.787208557128906
    },
    {
      "epoch": 1.827430725097656e-05,
      "model_forward_time": 1.8238272666931152,
      "step": 19162
    },
    {
      "epoch": 1.827430725097656e-05,
      "step": 19162,
      "training_step_time": 6.89978814125061
    },
    {
      "epoch": 1.827526092529297e-05,
      "model_forward_time": 1.8193552494049072,
      "step": 19163
    },
    {
      "epoch": 1.827526092529297e-05,
      "step": 19163,
      "training_step_time": 7.536432266235352
    },
    {
      "epoch": 1.8276214599609376e-05,
      "model_forward_time": 1.8241069316864014,
      "step": 19164
    },
    {
      "epoch": 1.8276214599609376e-05,
      "step": 19164,
      "training_step_time": 8.564954996109009
    },
    {
      "epoch": 1.827716827392578e-05,
      "model_forward_time": 2.364070415496826,
      "step": 19165
    },
    {
      "epoch": 1.827716827392578e-05,
      "step": 19165,
      "training_step_time": 8.882025241851807
    },
    {
      "epoch": 1.8278121948242188e-05,
      "model_forward_time": 1.8247385025024414,
      "step": 19166
    },
    {
      "epoch": 1.8278121948242188e-05,
      "step": 19166,
      "training_step_time": 7.702867746353149
    },
    {
      "epoch": 1.8279075622558595e-05,
      "model_forward_time": 1.8210253715515137,
      "step": 19167
    },
    {
      "epoch": 1.8279075622558595e-05,
      "step": 19167,
      "training_step_time": 7.064038276672363
    },
    {
      "epoch": 1.8280029296875e-05,
      "model_forward_time": 1.8183190822601318,
      "step": 19168
    },
    {
      "epoch": 1.8280029296875e-05,
      "step": 19168,
      "training_step_time": 7.17140793800354
    },
    {
      "epoch": 1.8280982971191407e-05,
      "model_forward_time": 1.8249821662902832,
      "step": 19169
    },
    {
      "epoch": 1.8280982971191407e-05,
      "step": 19169,
      "training_step_time": 6.769455671310425
    },
    {
      "epoch": 1.8281936645507814e-05,
      "grad_norm": 0.1938118189573288,
      "learning_rate": 9.461096973895774e-06,
      "loss": 0.065,
      "step": 19170
    },
    {
      "epoch": 1.8281936645507814e-05,
      "model_forward_time": 1.823859691619873,
      "step": 19170
    },
    {
      "dynamics_loss_avg": 0.06350937969982624,
      "epoch": 1.8281936645507814e-05,
      "step": 19170
    },
    {
      "action_loss_avg": 0.0032763949828222396,
      "epoch": 1.8281936645507814e-05,
      "step": 19170
    },
    {
      "epoch": 1.8281936645507814e-05,
      "step": 19170,
      "training_step_time": 6.730961561203003
    },
    {
      "epoch": 1.8282890319824218e-05,
      "model_forward_time": 1.8206593990325928,
      "step": 19171
    },
    {
      "epoch": 1.8282890319824218e-05,
      "step": 19171,
      "training_step_time": 6.908647060394287
    },
    {
      "epoch": 1.8283843994140626e-05,
      "model_forward_time": 1.8274929523468018,
      "step": 19172
    },
    {
      "epoch": 1.8283843994140626e-05,
      "step": 19172,
      "training_step_time": 7.51286506652832
    },
    {
      "epoch": 1.828479766845703e-05,
      "model_forward_time": 1.8247952461242676,
      "step": 19173
    },
    {
      "epoch": 1.828479766845703e-05,
      "step": 19173,
      "training_step_time": 8.635684251785278
    },
    {
      "epoch": 1.8285751342773437e-05,
      "model_forward_time": 2.3593387603759766,
      "step": 19174
    },
    {
      "epoch": 1.8285751342773437e-05,
      "step": 19174,
      "training_step_time": 7.738038063049316
    },
    {
      "epoch": 1.8286705017089844e-05,
      "model_forward_time": 1.8327255249023438,
      "step": 19175
    },
    {
      "epoch": 1.8286705017089844e-05,
      "step": 19175,
      "training_step_time": 7.844794034957886
    },
    {
      "epoch": 1.828765869140625e-05,
      "model_forward_time": 1.4450747966766357,
      "step": 19176
    },
    {
      "epoch": 1.828765869140625e-05,
      "step": 19176,
      "training_step_time": 7.423426151275635
    },
    {
      "epoch": 1.8288612365722656e-05,
      "model_forward_time": 1.824622631072998,
      "step": 19177
    },
    {
      "epoch": 1.8288612365722656e-05,
      "step": 19177,
      "training_step_time": 7.293412685394287
    },
    {
      "epoch": 1.8289566040039063e-05,
      "model_forward_time": 1.8180603981018066,
      "step": 19178
    },
    {
      "epoch": 1.8289566040039063e-05,
      "step": 19178,
      "training_step_time": 6.801126480102539
    },
    {
      "epoch": 1.8290519714355467e-05,
      "model_forward_time": 1.8193602561950684,
      "step": 19179
    },
    {
      "epoch": 1.8290519714355467e-05,
      "step": 19179,
      "training_step_time": 6.750118017196655
    },
    {
      "epoch": 1.8291473388671875e-05,
      "grad_norm": 0.21799077093601227,
      "learning_rate": 9.460350019364356e-06,
      "loss": 0.0641,
      "step": 19180
    },
    {
      "epoch": 1.8291473388671875e-05,
      "model_forward_time": 1.8234858512878418,
      "step": 19180
    },
    {
      "dynamics_loss_avg": 0.05566071206703782,
      "epoch": 1.8291473388671875e-05,
      "step": 19180
    },
    {
      "action_loss_avg": 0.003880278248107061,
      "epoch": 1.8291473388671875e-05,
      "step": 19180
    },
    {
      "epoch": 1.8291473388671875e-05,
      "step": 19180,
      "training_step_time": 6.924530029296875
    },
    {
      "epoch": 1.8292427062988282e-05,
      "model_forward_time": 1.8278436660766602,
      "step": 19181
    },
    {
      "epoch": 1.8292427062988282e-05,
      "step": 19181,
      "training_step_time": 7.566744327545166
    },
    {
      "epoch": 1.8293380737304686e-05,
      "model_forward_time": 1.8188934326171875,
      "step": 19182
    },
    {
      "epoch": 1.8293380737304686e-05,
      "step": 19182,
      "training_step_time": 9.827667713165283
    },
    {
      "epoch": 1.8294334411621094e-05,
      "model_forward_time": 2.3280344009399414,
      "step": 19183
    },
    {
      "epoch": 1.8294334411621094e-05,
      "step": 19183,
      "training_step_time": 8.898140668869019
    },
    {
      "epoch": 1.82952880859375e-05,
      "model_forward_time": 1.8155856132507324,
      "step": 19184
    },
    {
      "epoch": 1.82952880859375e-05,
      "step": 19184,
      "training_step_time": 7.788159608840942
    },
    {
      "epoch": 1.8296241760253905e-05,
      "model_forward_time": 1.8167476654052734,
      "step": 19185
    },
    {
      "epoch": 1.8296241760253905e-05,
      "step": 19185,
      "training_step_time": 7.556250333786011
    },
    {
      "epoch": 1.8297195434570312e-05,
      "model_forward_time": 1.8174080848693848,
      "step": 19186
    },
    {
      "epoch": 1.8297195434570312e-05,
      "step": 19186,
      "training_step_time": 7.064099311828613
    },
    {
      "epoch": 1.829814910888672e-05,
      "model_forward_time": 1.814772605895996,
      "step": 19187
    },
    {
      "epoch": 1.829814910888672e-05,
      "step": 19187,
      "training_step_time": 6.804480314254761
    },
    {
      "epoch": 1.8299102783203124e-05,
      "model_forward_time": 1.8179771900177002,
      "step": 19188
    },
    {
      "epoch": 1.8299102783203124e-05,
      "step": 19188,
      "training_step_time": 6.791879177093506
    },
    {
      "epoch": 1.830005645751953e-05,
      "model_forward_time": 1.8240466117858887,
      "step": 19189
    },
    {
      "epoch": 1.830005645751953e-05,
      "step": 19189,
      "training_step_time": 6.953889608383179
    },
    {
      "epoch": 1.830101013183594e-05,
      "grad_norm": 0.1899726539850235,
      "learning_rate": 9.459602577055765e-06,
      "loss": 0.0665,
      "step": 19190
    },
    {
      "epoch": 1.830101013183594e-05,
      "model_forward_time": 1.8191466331481934,
      "step": 19190
    },
    {
      "dynamics_loss_avg": 0.07201620936393738,
      "epoch": 1.830101013183594e-05,
      "step": 19190
    },
    {
      "action_loss_avg": 0.0034312176750972865,
      "epoch": 1.830101013183594e-05,
      "step": 19190
    },
    {
      "epoch": 1.830101013183594e-05,
      "step": 19190,
      "training_step_time": 7.651820659637451
    },
    {
      "epoch": 1.8301963806152343e-05,
      "model_forward_time": 1.8143110275268555,
      "step": 19191
    },
    {
      "epoch": 1.8301963806152343e-05,
      "step": 19191,
      "training_step_time": 8.590210199356079
    },
    {
      "epoch": 1.830291748046875e-05,
      "model_forward_time": 1.8226733207702637,
      "step": 19192
    },
    {
      "epoch": 1.830291748046875e-05,
      "step": 19192,
      "training_step_time": 9.2943594455719
    },
    {
      "epoch": 1.8303871154785158e-05,
      "model_forward_time": 1.9875905513763428,
      "step": 19193
    },
    {
      "epoch": 1.8303871154785158e-05,
      "step": 19193,
      "training_step_time": 7.783012866973877
    },
    {
      "epoch": 1.830482482910156e-05,
      "model_forward_time": 1.8204972743988037,
      "step": 19194
    },
    {
      "epoch": 1.830482482910156e-05,
      "step": 19194,
      "training_step_time": 7.598408460617065
    },
    {
      "epoch": 1.830577850341797e-05,
      "model_forward_time": 1.8251781463623047,
      "step": 19195
    },
    {
      "epoch": 1.830577850341797e-05,
      "step": 19195,
      "training_step_time": 7.207568168640137
    },
    {
      "epoch": 1.8306732177734376e-05,
      "model_forward_time": 1.8193039894104004,
      "step": 19196
    },
    {
      "epoch": 1.8306732177734376e-05,
      "step": 19196,
      "training_step_time": 6.794679880142212
    },
    {
      "epoch": 1.830768585205078e-05,
      "model_forward_time": 1.8310801982879639,
      "step": 19197
    },
    {
      "epoch": 1.830768585205078e-05,
      "step": 19197,
      "training_step_time": 6.775467872619629
    },
    {
      "epoch": 1.8308639526367188e-05,
      "model_forward_time": 1.8147063255310059,
      "step": 19198
    },
    {
      "epoch": 1.8308639526367188e-05,
      "step": 19198,
      "training_step_time": 6.77245569229126
    },
    {
      "epoch": 1.8309593200683595e-05,
      "model_forward_time": 1.8136937618255615,
      "step": 19199
    },
    {
      "epoch": 1.8309593200683595e-05,
      "step": 19199,
      "training_step_time": 7.589970350265503
    },
    {
      "epoch": 1.8310546875e-05,
      "grad_norm": 0.16746993362903595,
      "learning_rate": 9.458854647051739e-06,
      "loss": 0.0656,
      "step": 19200
    },
    {
      "epoch": 1.8310546875e-05,
      "model_forward_time": 1.8255794048309326,
      "step": 19200
    },
    {
      "dynamics_loss_avg": 0.050637788511812686,
      "epoch": 1.8310546875e-05,
      "step": 19200
    },
    {
      "action_loss_avg": 0.002527937287231907,
      "epoch": 1.8310546875e-05,
      "step": 19200
    },
    {
      "epoch": 1.8310546875e-05,
      "step": 19200,
      "training_step_time": 9.135765552520752
    },
    {
      "epoch": 1.8311500549316407e-05,
      "model_forward_time": 1.82344651222229,
      "step": 19201
    },
    {
      "epoch": 1.8311500549316407e-05,
      "step": 19201,
      "training_step_time": 9.155486345291138
    },
    {
      "epoch": 1.8312454223632814e-05,
      "model_forward_time": 2.347611427307129,
      "step": 19202
    },
    {
      "epoch": 1.8312454223632814e-05,
      "step": 19202,
      "training_step_time": 9.147679805755615
    },
    {
      "epoch": 1.8313407897949218e-05,
      "model_forward_time": 1.8190186023712158,
      "step": 19203
    },
    {
      "epoch": 1.8313407897949218e-05,
      "step": 19203,
      "training_step_time": 7.569977283477783
    },
    {
      "epoch": 1.8314361572265626e-05,
      "model_forward_time": 1.8128743171691895,
      "step": 19204
    },
    {
      "epoch": 1.8314361572265626e-05,
      "step": 19204,
      "training_step_time": 7.4666595458984375
    },
    {
      "epoch": 1.831531524658203e-05,
      "model_forward_time": 1.823979377746582,
      "step": 19205
    },
    {
      "epoch": 1.831531524658203e-05,
      "step": 19205,
      "training_step_time": 7.299454212188721
    },
    {
      "epoch": 1.8316268920898437e-05,
      "model_forward_time": 1.4481468200683594,
      "step": 19206
    },
    {
      "epoch": 1.8316268920898437e-05,
      "step": 19206,
      "training_step_time": 6.837203502655029
    },
    {
      "epoch": 1.8317222595214844e-05,
      "model_forward_time": 1.8258006572723389,
      "step": 19207
    },
    {
      "epoch": 1.8317222595214844e-05,
      "step": 19207,
      "training_step_time": 6.842167615890503
    },
    {
      "epoch": 1.831817626953125e-05,
      "model_forward_time": 1.8275461196899414,
      "step": 19208
    },
    {
      "epoch": 1.831817626953125e-05,
      "step": 19208,
      "training_step_time": 7.6338911056518555
    },
    {
      "epoch": 1.8319129943847656e-05,
      "model_forward_time": 1.8199763298034668,
      "step": 19209
    },
    {
      "epoch": 1.8319129943847656e-05,
      "step": 19209,
      "training_step_time": 8.485809326171875
    },
    {
      "epoch": 1.8320083618164063e-05,
      "grad_norm": 0.20061123371124268,
      "learning_rate": 9.458106229434076e-06,
      "loss": 0.061,
      "step": 19210
    },
    {
      "epoch": 1.8320083618164063e-05,
      "model_forward_time": 1.8179705142974854,
      "step": 19210
    },
    {
      "dynamics_loss_avg": 0.0631896011531353,
      "epoch": 1.8320083618164063e-05,
      "step": 19210
    },
    {
      "action_loss_avg": 0.0023079175502061845,
      "epoch": 1.8320083618164063e-05,
      "step": 19210
    },
    {
      "epoch": 1.8320083618164063e-05,
      "step": 19210,
      "training_step_time": 8.500269651412964
    },
    {
      "epoch": 1.8321037292480467e-05,
      "model_forward_time": 2.364055871963501,
      "step": 19211
    },
    {
      "epoch": 1.8321037292480467e-05,
      "step": 19211,
      "training_step_time": 9.247544527053833
    },
    {
      "epoch": 1.8321990966796875e-05,
      "model_forward_time": 1.8276920318603516,
      "step": 19212
    },
    {
      "epoch": 1.8321990966796875e-05,
      "step": 19212,
      "training_step_time": 8.63222861289978
    },
    {
      "epoch": 1.8322944641113282e-05,
      "model_forward_time": 1.8253915309906006,
      "step": 19213
    },
    {
      "epoch": 1.8322944641113282e-05,
      "step": 19213,
      "training_step_time": 7.445894002914429
    },
    {
      "epoch": 1.8323898315429686e-05,
      "model_forward_time": 1.8234097957611084,
      "step": 19214
    },
    {
      "epoch": 1.8323898315429686e-05,
      "step": 19214,
      "training_step_time": 7.29397177696228
    },
    {
      "epoch": 1.8324851989746094e-05,
      "model_forward_time": 1.8298985958099365,
      "step": 19215
    },
    {
      "epoch": 1.8324851989746094e-05,
      "step": 19215,
      "training_step_time": 6.7664618492126465
    },
    {
      "epoch": 1.83258056640625e-05,
      "model_forward_time": 1.7775309085845947,
      "step": 19216
    },
    {
      "epoch": 1.83258056640625e-05,
      "step": 19216,
      "training_step_time": 6.7688305377960205
    },
    {
      "epoch": 1.8326759338378905e-05,
      "model_forward_time": 1.8028564453125,
      "step": 19217
    },
    {
      "epoch": 1.8326759338378905e-05,
      "step": 19217,
      "training_step_time": 7.616772413253784
    },
    {
      "epoch": 1.8327713012695312e-05,
      "model_forward_time": 1.8079946041107178,
      "step": 19218
    },
    {
      "epoch": 1.8327713012695312e-05,
      "step": 19218,
      "training_step_time": 7.570791244506836
    },
    {
      "epoch": 1.832866668701172e-05,
      "model_forward_time": 1.8055248260498047,
      "step": 19219
    },
    {
      "epoch": 1.832866668701172e-05,
      "step": 19219,
      "training_step_time": 9.335877418518066
    },
    {
      "epoch": 1.8329620361328124e-05,
      "grad_norm": 0.25062277913093567,
      "learning_rate": 9.457357324284617e-06,
      "loss": 0.0641,
      "step": 19220
    },
    {
      "epoch": 1.8329620361328124e-05,
      "model_forward_time": 1.7888281345367432,
      "step": 19220
    },
    {
      "dynamics_loss_avg": 0.05621460285037756,
      "epoch": 1.8329620361328124e-05,
      "step": 19220
    },
    {
      "action_loss_avg": 0.0026840950711630287,
      "epoch": 1.8329620361328124e-05,
      "step": 19220
    },
    {
      "epoch": 1.8329620361328124e-05,
      "step": 19220,
      "training_step_time": 9.280001640319824
    },
    {
      "epoch": 1.833057403564453e-05,
      "model_forward_time": 1.80228590965271,
      "step": 19221
    },
    {
      "epoch": 1.833057403564453e-05,
      "step": 19221,
      "training_step_time": 8.153242349624634
    },
    {
      "epoch": 1.833152770996094e-05,
      "model_forward_time": 1.809027910232544,
      "step": 19222
    },
    {
      "epoch": 1.833152770996094e-05,
      "step": 19222,
      "training_step_time": 7.550332546234131
    },
    {
      "epoch": 1.8332481384277343e-05,
      "model_forward_time": 1.4350171089172363,
      "step": 19223
    },
    {
      "epoch": 1.8332481384277343e-05,
      "step": 19223,
      "training_step_time": 7.3882482051849365
    },
    {
      "epoch": 1.833343505859375e-05,
      "model_forward_time": 1.8005576133728027,
      "step": 19224
    },
    {
      "epoch": 1.833343505859375e-05,
      "step": 19224,
      "training_step_time": 6.75131368637085
    },
    {
      "epoch": 1.8334388732910158e-05,
      "model_forward_time": 1.804413080215454,
      "step": 19225
    },
    {
      "epoch": 1.8334388732910158e-05,
      "step": 19225,
      "training_step_time": 6.763746023178101
    },
    {
      "epoch": 1.833534240722656e-05,
      "model_forward_time": 1.8089900016784668,
      "step": 19226
    },
    {
      "epoch": 1.833534240722656e-05,
      "step": 19226,
      "training_step_time": 7.537102937698364
    },
    {
      "epoch": 1.833629608154297e-05,
      "model_forward_time": 1.8043322563171387,
      "step": 19227
    },
    {
      "epoch": 1.833629608154297e-05,
      "step": 19227,
      "training_step_time": 7.527641534805298
    },
    {
      "epoch": 1.8337249755859376e-05,
      "model_forward_time": 1.8050765991210938,
      "step": 19228
    },
    {
      "epoch": 1.8337249755859376e-05,
      "step": 19228,
      "training_step_time": 8.351974725723267
    },
    {
      "epoch": 1.833820343017578e-05,
      "model_forward_time": 1.8029305934906006,
      "step": 19229
    },
    {
      "epoch": 1.833820343017578e-05,
      "step": 19229,
      "training_step_time": 8.447577476501465
    },
    {
      "epoch": 1.8339157104492188e-05,
      "grad_norm": 0.2853924632072449,
      "learning_rate": 9.456607931685262e-06,
      "loss": 0.0654,
      "step": 19230
    },
    {
      "epoch": 1.8339157104492188e-05,
      "model_forward_time": 1.958092451095581,
      "step": 19230
    },
    {
      "dynamics_loss_avg": 0.06793066691607237,
      "epoch": 1.8339157104492188e-05,
      "step": 19230
    },
    {
      "action_loss_avg": 0.00371606313274242,
      "epoch": 1.8339157104492188e-05,
      "step": 19230
    },
    {
      "epoch": 1.8339157104492188e-05,
      "step": 19230,
      "training_step_time": 8.64247989654541
    },
    {
      "epoch": 1.8340110778808595e-05,
      "model_forward_time": 1.8088159561157227,
      "step": 19231
    },
    {
      "epoch": 1.8340110778808595e-05,
      "step": 19231,
      "training_step_time": 7.583407402038574
    },
    {
      "epoch": 1.8341064453125e-05,
      "model_forward_time": 1.8021337985992432,
      "step": 19232
    },
    {
      "epoch": 1.8341064453125e-05,
      "step": 19232,
      "training_step_time": 7.3667333126068115
    },
    {
      "epoch": 1.8342018127441407e-05,
      "model_forward_time": 1.8072729110717773,
      "step": 19233
    },
    {
      "epoch": 1.8342018127441407e-05,
      "step": 19233,
      "training_step_time": 6.74392294883728
    },
    {
      "epoch": 1.8342971801757814e-05,
      "model_forward_time": 1.8034405708312988,
      "step": 19234
    },
    {
      "epoch": 1.8342971801757814e-05,
      "step": 19234,
      "training_step_time": 6.8170812129974365
    },
    {
      "epoch": 1.8343925476074218e-05,
      "model_forward_time": 1.7979819774627686,
      "step": 19235
    },
    {
      "epoch": 1.8343925476074218e-05,
      "step": 19235,
      "training_step_time": 7.608632564544678
    },
    {
      "epoch": 1.8344879150390626e-05,
      "model_forward_time": 1.8053779602050781,
      "step": 19236
    },
    {
      "epoch": 1.8344879150390626e-05,
      "step": 19236,
      "training_step_time": 8.389402151107788
    },
    {
      "epoch": 1.834583282470703e-05,
      "model_forward_time": 1.8014748096466064,
      "step": 19237
    },
    {
      "epoch": 1.834583282470703e-05,
      "step": 19237,
      "training_step_time": 8.63527536392212
    },
    {
      "epoch": 1.8346786499023437e-05,
      "model_forward_time": 1.815171718597412,
      "step": 19238
    },
    {
      "epoch": 1.8346786499023437e-05,
      "step": 19238,
      "training_step_time": 10.704012155532837
    },
    {
      "epoch": 1.8347740173339844e-05,
      "model_forward_time": 2.327833652496338,
      "step": 19239
    },
    {
      "epoch": 1.8347740173339844e-05,
      "step": 19239,
      "training_step_time": 7.7496116161346436
    },
    {
      "epoch": 1.834869384765625e-05,
      "grad_norm": 0.3123108744621277,
      "learning_rate": 9.455858051717965e-06,
      "loss": 0.0642,
      "step": 19240
    },
    {
      "epoch": 1.834869384765625e-05,
      "model_forward_time": 1.7859492301940918,
      "step": 19240
    },
    {
      "dynamics_loss_avg": 0.07679358832538127,
      "epoch": 1.834869384765625e-05,
      "step": 19240
    },
    {
      "action_loss_avg": 0.003651411016471684,
      "epoch": 1.834869384765625e-05,
      "step": 19240
    },
    {
      "epoch": 1.834869384765625e-05,
      "step": 19240,
      "training_step_time": 7.58722448348999
    },
    {
      "epoch": 1.8349647521972656e-05,
      "model_forward_time": 1.805330514907837,
      "step": 19241
    },
    {
      "epoch": 1.8349647521972656e-05,
      "step": 19241,
      "training_step_time": 7.454430818557739
    },
    {
      "epoch": 1.8350601196289063e-05,
      "model_forward_time": 1.8090128898620605,
      "step": 19242
    },
    {
      "epoch": 1.8350601196289063e-05,
      "step": 19242,
      "training_step_time": 7.219801902770996
    },
    {
      "epoch": 1.8351554870605467e-05,
      "model_forward_time": 1.8110523223876953,
      "step": 19243
    },
    {
      "epoch": 1.8351554870605467e-05,
      "step": 19243,
      "training_step_time": 6.757288932800293
    },
    {
      "epoch": 1.8352508544921875e-05,
      "model_forward_time": 1.7630009651184082,
      "step": 19244
    },
    {
      "epoch": 1.8352508544921875e-05,
      "step": 19244,
      "training_step_time": 7.533735513687134
    },
    {
      "epoch": 1.8353462219238282e-05,
      "model_forward_time": 1.7139475345611572,
      "step": 19245
    },
    {
      "epoch": 1.8353462219238282e-05,
      "step": 19245,
      "training_step_time": 7.5580220222473145
    },
    {
      "epoch": 1.8354415893554686e-05,
      "model_forward_time": 1.7038867473602295,
      "step": 19246
    },
    {
      "epoch": 1.8354415893554686e-05,
      "step": 19246,
      "training_step_time": 7.631535530090332
    },
    {
      "epoch": 1.8355369567871094e-05,
      "model_forward_time": 1.706571102142334,
      "step": 19247
    },
    {
      "epoch": 1.8355369567871094e-05,
      "step": 19247,
      "training_step_time": 8.849065780639648
    },
    {
      "epoch": 1.83563232421875e-05,
      "model_forward_time": 2.23209285736084,
      "step": 19248
    },
    {
      "epoch": 1.83563232421875e-05,
      "step": 19248,
      "training_step_time": 8.50852918624878
    },
    {
      "epoch": 1.8357276916503905e-05,
      "model_forward_time": 1.711808204650879,
      "step": 19249
    },
    {
      "epoch": 1.8357276916503905e-05,
      "step": 19249,
      "training_step_time": 7.668841361999512
    },
    {
      "epoch": 1.8358230590820312e-05,
      "grad_norm": 0.22309042513370514,
      "learning_rate": 9.45510768446473e-06,
      "loss": 0.065,
      "step": 19250
    },
    {
      "epoch": 1.8358230590820312e-05,
      "model_forward_time": 1.7185866832733154,
      "step": 19250
    },
    {
      "dynamics_loss_avg": 0.07196092735975981,
      "epoch": 1.8358230590820312e-05,
      "step": 19250
    },
    {
      "action_loss_avg": 0.002413988026091829,
      "epoch": 1.8358230590820312e-05,
      "step": 19250
    },
    {
      "epoch": 1.8358230590820312e-05,
      "step": 19250,
      "training_step_time": 8.352383613586426
    },
    {
      "epoch": 1.835918426513672e-05,
      "model_forward_time": 1.7069721221923828,
      "step": 19251
    },
    {
      "epoch": 1.835918426513672e-05,
      "step": 19251,
      "training_step_time": 7.30640983581543
    },
    {
      "epoch": 1.8360137939453124e-05,
      "model_forward_time": 1.7125396728515625,
      "step": 19252
    },
    {
      "epoch": 1.8360137939453124e-05,
      "step": 19252,
      "training_step_time": 6.752079725265503
    },
    {
      "epoch": 1.836109161376953e-05,
      "model_forward_time": 1.7071142196655273,
      "step": 19253
    },
    {
      "epoch": 1.836109161376953e-05,
      "step": 19253,
      "training_step_time": 7.437576055526733
    },
    {
      "epoch": 1.836204528808594e-05,
      "model_forward_time": 1.7046830654144287,
      "step": 19254
    },
    {
      "epoch": 1.836204528808594e-05,
      "step": 19254,
      "training_step_time": 7.5177977085113525
    },
    {
      "epoch": 1.8362998962402343e-05,
      "model_forward_time": 1.7061679363250732,
      "step": 19255
    },
    {
      "epoch": 1.8362998962402343e-05,
      "step": 19255,
      "training_step_time": 7.666100740432739
    },
    {
      "epoch": 1.836395263671875e-05,
      "model_forward_time": 1.725449800491333,
      "step": 19256
    },
    {
      "epoch": 1.836395263671875e-05,
      "step": 19256,
      "training_step_time": 10.104601621627808
    },
    {
      "epoch": 1.8364906311035158e-05,
      "model_forward_time": 1.7052252292633057,
      "step": 19257
    },
    {
      "epoch": 1.8364906311035158e-05,
      "step": 19257,
      "training_step_time": 10.6497642993927
    },
    {
      "epoch": 1.836585998535156e-05,
      "model_forward_time": 1.7039194107055664,
      "step": 19258
    },
    {
      "epoch": 1.836585998535156e-05,
      "step": 19258,
      "training_step_time": 7.7710347175598145
    },
    {
      "epoch": 1.836681365966797e-05,
      "model_forward_time": 1.713350534439087,
      "step": 19259
    },
    {
      "epoch": 1.836681365966797e-05,
      "step": 19259,
      "training_step_time": 7.574558734893799
    },
    {
      "epoch": 1.8367767333984376e-05,
      "grad_norm": 0.23812055587768555,
      "learning_rate": 9.454356830007619e-06,
      "loss": 0.0622,
      "step": 19260
    },
    {
      "epoch": 1.8367767333984376e-05,
      "model_forward_time": 1.703843116760254,
      "step": 19260
    },
    {
      "dynamics_loss_avg": 0.0555045822635293,
      "epoch": 1.8367767333984376e-05,
      "step": 19260
    },
    {
      "action_loss_avg": 0.0023953782045282425,
      "epoch": 1.8367767333984376e-05,
      "step": 19260
    },
    {
      "epoch": 1.8367767333984376e-05,
      "step": 19260,
      "training_step_time": 7.393452167510986
    },
    {
      "epoch": 1.836872100830078e-05,
      "model_forward_time": 1.7016592025756836,
      "step": 19261
    },
    {
      "epoch": 1.836872100830078e-05,
      "step": 19261,
      "training_step_time": 6.828521013259888
    },
    {
      "epoch": 1.8369674682617188e-05,
      "model_forward_time": 1.7056725025177002,
      "step": 19262
    },
    {
      "epoch": 1.8369674682617188e-05,
      "step": 19262,
      "training_step_time": 7.645190954208374
    },
    {
      "epoch": 1.8370628356933595e-05,
      "model_forward_time": 1.7056059837341309,
      "step": 19263
    },
    {
      "epoch": 1.8370628356933595e-05,
      "step": 19263,
      "training_step_time": 7.642953395843506
    },
    {
      "epoch": 1.837158203125e-05,
      "model_forward_time": 1.7076351642608643,
      "step": 19264
    },
    {
      "epoch": 1.837158203125e-05,
      "step": 19264,
      "training_step_time": 7.767495393753052
    },
    {
      "epoch": 1.8372535705566407e-05,
      "model_forward_time": 1.7039790153503418,
      "step": 19265
    },
    {
      "epoch": 1.8372535705566407e-05,
      "step": 19265,
      "training_step_time": 8.456033945083618
    },
    {
      "epoch": 1.837348937988281e-05,
      "model_forward_time": 1.7090213298797607,
      "step": 19266
    },
    {
      "epoch": 1.837348937988281e-05,
      "step": 19266,
      "training_step_time": 8.70793628692627
    },
    {
      "epoch": 1.8374443054199218e-05,
      "model_forward_time": 2.2408926486968994,
      "step": 19267
    },
    {
      "epoch": 1.8374443054199218e-05,
      "step": 19267,
      "training_step_time": 7.683543920516968
    },
    {
      "epoch": 1.8375396728515626e-05,
      "model_forward_time": 1.7129096984863281,
      "step": 19268
    },
    {
      "epoch": 1.8375396728515626e-05,
      "step": 19268,
      "training_step_time": 7.573551654815674
    },
    {
      "epoch": 1.837635040283203e-05,
      "model_forward_time": 1.702434778213501,
      "step": 19269
    },
    {
      "epoch": 1.837635040283203e-05,
      "step": 19269,
      "training_step_time": 7.425205230712891
    },
    {
      "epoch": 1.8377304077148437e-05,
      "grad_norm": 0.18189460039138794,
      "learning_rate": 9.453605488428741e-06,
      "loss": 0.065,
      "step": 19270
    },
    {
      "epoch": 1.8377304077148437e-05,
      "model_forward_time": 1.7009799480438232,
      "step": 19270
    },
    {
      "dynamics_loss_avg": 0.05730114839971066,
      "epoch": 1.8377304077148437e-05,
      "step": 19270
    },
    {
      "action_loss_avg": 0.0024117818276863543,
      "epoch": 1.8377304077148437e-05,
      "step": 19270
    },
    {
      "epoch": 1.8377304077148437e-05,
      "step": 19270,
      "training_step_time": 7.280435800552368
    },
    {
      "epoch": 1.8378257751464845e-05,
      "model_forward_time": 1.7017505168914795,
      "step": 19271
    },
    {
      "epoch": 1.8378257751464845e-05,
      "step": 19271,
      "training_step_time": 7.6647679805755615
    },
    {
      "epoch": 1.837921142578125e-05,
      "model_forward_time": 1.7145967483520508,
      "step": 19272
    },
    {
      "epoch": 1.837921142578125e-05,
      "step": 19272,
      "training_step_time": 8.080500602722168
    },
    {
      "epoch": 1.8380165100097656e-05,
      "model_forward_time": 1.6963121891021729,
      "step": 19273
    },
    {
      "epoch": 1.8380165100097656e-05,
      "step": 19273,
      "training_step_time": 7.546926975250244
    },
    {
      "epoch": 1.8381118774414063e-05,
      "model_forward_time": 1.698951244354248,
      "step": 19274
    },
    {
      "epoch": 1.8381118774414063e-05,
      "step": 19274,
      "training_step_time": 8.623666524887085
    },
    {
      "epoch": 1.8382072448730467e-05,
      "model_forward_time": 0.9902558326721191,
      "step": 19275
    },
    {
      "epoch": 1.8382072448730467e-05,
      "step": 19275,
      "training_step_time": 12.307296752929688
    },
    {
      "epoch": 1.8383026123046875e-05,
      "model_forward_time": 2.252382516860962,
      "step": 19276
    },
    {
      "epoch": 1.8383026123046875e-05,
      "step": 19276,
      "training_step_time": 8.584145545959473
    },
    {
      "epoch": 1.8383979797363282e-05,
      "model_forward_time": 1.6943659782409668,
      "step": 19277
    },
    {
      "epoch": 1.8383979797363282e-05,
      "step": 19277,
      "training_step_time": 7.817120790481567
    },
    {
      "epoch": 1.8384933471679686e-05,
      "model_forward_time": 1.7086760997772217,
      "step": 19278
    },
    {
      "epoch": 1.8384933471679686e-05,
      "step": 19278,
      "training_step_time": 8.107281923294067
    },
    {
      "epoch": 1.8385887145996094e-05,
      "model_forward_time": 1.7009191513061523,
      "step": 19279
    },
    {
      "epoch": 1.8385887145996094e-05,
      "step": 19279,
      "training_step_time": 7.320131063461304
    },
    {
      "epoch": 1.83868408203125e-05,
      "grad_norm": 0.16991671919822693,
      "learning_rate": 9.452853659810262e-06,
      "loss": 0.064,
      "step": 19280
    },
    {
      "epoch": 1.83868408203125e-05,
      "model_forward_time": 1.7057127952575684,
      "step": 19280
    },
    {
      "dynamics_loss_avg": 0.06355944527313113,
      "epoch": 1.83868408203125e-05,
      "step": 19280
    },
    {
      "action_loss_avg": 0.0033107298309914767,
      "epoch": 1.83868408203125e-05,
      "step": 19280
    },
    {
      "epoch": 1.83868408203125e-05,
      "step": 19280,
      "training_step_time": 7.705003023147583
    },
    {
      "epoch": 1.8387794494628905e-05,
      "model_forward_time": 1.7029151916503906,
      "step": 19281
    },
    {
      "epoch": 1.8387794494628905e-05,
      "step": 19281,
      "training_step_time": 7.645448207855225
    },
    {
      "epoch": 1.8388748168945313e-05,
      "model_forward_time": 1.701866626739502,
      "step": 19282
    },
    {
      "epoch": 1.8388748168945313e-05,
      "step": 19282,
      "training_step_time": 7.647739887237549
    },
    {
      "epoch": 1.838970184326172e-05,
      "model_forward_time": 1.6997363567352295,
      "step": 19283
    },
    {
      "epoch": 1.838970184326172e-05,
      "step": 19283,
      "training_step_time": 7.700870752334595
    },
    {
      "epoch": 1.8390655517578124e-05,
      "model_forward_time": 1.7056283950805664,
      "step": 19284
    },
    {
      "epoch": 1.8390655517578124e-05,
      "step": 19284,
      "training_step_time": 8.685150384902954
    },
    {
      "epoch": 1.839160919189453e-05,
      "model_forward_time": 2.2350142002105713,
      "step": 19285
    },
    {
      "epoch": 1.839160919189453e-05,
      "step": 19285,
      "training_step_time": 9.192002534866333
    },
    {
      "epoch": 1.839256286621094e-05,
      "model_forward_time": 1.70174241065979,
      "step": 19286
    },
    {
      "epoch": 1.839256286621094e-05,
      "step": 19286,
      "training_step_time": 7.769864320755005
    },
    {
      "epoch": 1.8393516540527343e-05,
      "model_forward_time": 1.6993038654327393,
      "step": 19287
    },
    {
      "epoch": 1.8393516540527343e-05,
      "step": 19287,
      "training_step_time": 8.215709447860718
    },
    {
      "epoch": 1.839447021484375e-05,
      "model_forward_time": 1.697031021118164,
      "step": 19288
    },
    {
      "epoch": 1.839447021484375e-05,
      "step": 19288,
      "training_step_time": 7.265625476837158
    },
    {
      "epoch": 1.8395423889160158e-05,
      "model_forward_time": 1.7152585983276367,
      "step": 19289
    },
    {
      "epoch": 1.8395423889160158e-05,
      "step": 19289,
      "training_step_time": 7.614839792251587
    },
    {
      "epoch": 1.839637756347656e-05,
      "grad_norm": 0.15494991838932037,
      "learning_rate": 9.452101344234401e-06,
      "loss": 0.0636,
      "step": 19290
    },
    {
      "epoch": 1.839637756347656e-05,
      "model_forward_time": 1.7095842361450195,
      "step": 19290
    },
    {
      "dynamics_loss_avg": 0.043378423899412155,
      "epoch": 1.839637756347656e-05,
      "step": 19290
    },
    {
      "action_loss_avg": 0.0015904108178801835,
      "epoch": 1.839637756347656e-05,
      "step": 19290
    },
    {
      "epoch": 1.839637756347656e-05,
      "step": 19290,
      "training_step_time": 7.660896062850952
    },
    {
      "epoch": 1.839733123779297e-05,
      "model_forward_time": 1.7042789459228516,
      "step": 19291
    },
    {
      "epoch": 1.839733123779297e-05,
      "step": 19291,
      "training_step_time": 7.613436698913574
    },
    {
      "epoch": 1.8398284912109377e-05,
      "model_forward_time": 1.7445933818817139,
      "step": 19292
    },
    {
      "epoch": 1.8398284912109377e-05,
      "step": 19292,
      "training_step_time": 7.611798286437988
    },
    {
      "epoch": 1.839923858642578e-05,
      "model_forward_time": 1.7055680751800537,
      "step": 19293
    },
    {
      "epoch": 1.839923858642578e-05,
      "step": 19293,
      "training_step_time": 8.913756608963013
    },
    {
      "epoch": 1.8400192260742188e-05,
      "model_forward_time": 1.701458215713501,
      "step": 19294
    },
    {
      "epoch": 1.8400192260742188e-05,
      "step": 19294,
      "training_step_time": 8.436360359191895
    },
    {
      "epoch": 1.8401145935058595e-05,
      "model_forward_time": 2.2440319061279297,
      "step": 19295
    },
    {
      "epoch": 1.8401145935058595e-05,
      "step": 19295,
      "training_step_time": 8.255350351333618
    },
    {
      "epoch": 1.8402099609375e-05,
      "model_forward_time": 1.7288479804992676,
      "step": 19296
    },
    {
      "epoch": 1.8402099609375e-05,
      "step": 19296,
      "training_step_time": 7.426239490509033
    },
    {
      "epoch": 1.8403053283691407e-05,
      "model_forward_time": 1.713904619216919,
      "step": 19297
    },
    {
      "epoch": 1.8403053283691407e-05,
      "step": 19297,
      "training_step_time": 7.078458547592163
    },
    {
      "epoch": 1.840400695800781e-05,
      "model_forward_time": 1.710040807723999,
      "step": 19298
    },
    {
      "epoch": 1.840400695800781e-05,
      "step": 19298,
      "training_step_time": 7.299899578094482
    },
    {
      "epoch": 1.8404960632324218e-05,
      "model_forward_time": 1.7162041664123535,
      "step": 19299
    },
    {
      "epoch": 1.8404960632324218e-05,
      "step": 19299,
      "training_step_time": 7.667902708053589
    },
    {
      "epoch": 1.8405914306640626e-05,
      "grad_norm": 0.20551379024982452,
      "learning_rate": 9.451348541783432e-06,
      "loss": 0.0607,
      "step": 19300
    },
    {
      "epoch": 1.8405914306640626e-05,
      "model_forward_time": 1.714979648590088,
      "step": 19300
    },
    {
      "dynamics_loss_avg": 0.06491387318819761,
      "epoch": 1.8405914306640626e-05,
      "step": 19300
    },
    {
      "action_loss_avg": 0.0024768859846517445,
      "epoch": 1.8405914306640626e-05,
      "step": 19300
    },
    {
      "epoch": 1.8405914306640626e-05,
      "step": 19300,
      "training_step_time": 7.655543565750122
    },
    {
      "epoch": 1.840686798095703e-05,
      "model_forward_time": 1.7072997093200684,
      "step": 19301
    },
    {
      "epoch": 1.840686798095703e-05,
      "step": 19301,
      "training_step_time": 9.135578155517578
    },
    {
      "epoch": 1.8407821655273437e-05,
      "model_forward_time": 1.7081658840179443,
      "step": 19302
    },
    {
      "epoch": 1.8407821655273437e-05,
      "step": 19302,
      "training_step_time": 9.264030694961548
    },
    {
      "epoch": 1.8408775329589845e-05,
      "model_forward_time": 1.7273197174072266,
      "step": 19303
    },
    {
      "epoch": 1.8408775329589845e-05,
      "step": 19303,
      "training_step_time": 9.095987558364868
    },
    {
      "epoch": 1.840972900390625e-05,
      "model_forward_time": 2.2963719367980957,
      "step": 19304
    },
    {
      "epoch": 1.840972900390625e-05,
      "step": 19304,
      "training_step_time": 8.931192636489868
    },
    {
      "epoch": 1.8410682678222656e-05,
      "model_forward_time": 1.7159209251403809,
      "step": 19305
    },
    {
      "epoch": 1.8410682678222656e-05,
      "step": 19305,
      "training_step_time": 7.5746307373046875
    },
    {
      "epoch": 1.8411636352539063e-05,
      "model_forward_time": 1.7099285125732422,
      "step": 19306
    },
    {
      "epoch": 1.8411636352539063e-05,
      "step": 19306,
      "training_step_time": 7.37501859664917
    },
    {
      "epoch": 1.8412590026855467e-05,
      "model_forward_time": 1.7174737453460693,
      "step": 19307
    },
    {
      "epoch": 1.8412590026855467e-05,
      "step": 19307,
      "training_step_time": 7.2792723178863525
    },
    {
      "epoch": 1.8413543701171875e-05,
      "model_forward_time": 1.7045323848724365,
      "step": 19308
    },
    {
      "epoch": 1.8413543701171875e-05,
      "step": 19308,
      "training_step_time": 7.5027501583099365
    },
    {
      "epoch": 1.8414497375488282e-05,
      "model_forward_time": 1.7130348682403564,
      "step": 19309
    },
    {
      "epoch": 1.8414497375488282e-05,
      "step": 19309,
      "training_step_time": 7.236029863357544
    },
    {
      "epoch": 1.8415451049804686e-05,
      "grad_norm": 0.11381298303604126,
      "learning_rate": 9.450595252539678e-06,
      "loss": 0.0662,
      "step": 19310
    },
    {
      "epoch": 1.8415451049804686e-05,
      "model_forward_time": 1.7148566246032715,
      "step": 19310
    },
    {
      "dynamics_loss_avg": 0.06613438297063112,
      "epoch": 1.8415451049804686e-05,
      "step": 19310
    },
    {
      "action_loss_avg": 0.0024328280938789248,
      "epoch": 1.8415451049804686e-05,
      "step": 19310
    },
    {
      "epoch": 1.8415451049804686e-05,
      "step": 19310,
      "training_step_time": 8.376183032989502
    },
    {
      "epoch": 1.8416404724121094e-05,
      "model_forward_time": 1.725154161453247,
      "step": 19311
    },
    {
      "epoch": 1.8416404724121094e-05,
      "step": 19311,
      "training_step_time": 7.651173114776611
    },
    {
      "epoch": 1.84173583984375e-05,
      "model_forward_time": 1.7088801860809326,
      "step": 19312
    },
    {
      "epoch": 1.84173583984375e-05,
      "step": 19312,
      "training_step_time": 9.589309453964233
    },
    {
      "epoch": 1.8418312072753905e-05,
      "model_forward_time": 2.321479082107544,
      "step": 19313
    },
    {
      "epoch": 1.8418312072753905e-05,
      "step": 19313,
      "training_step_time": 8.227687358856201
    },
    {
      "epoch": 1.8419265747070313e-05,
      "model_forward_time": 1.7162189483642578,
      "step": 19314
    },
    {
      "epoch": 1.8419265747070313e-05,
      "step": 19314,
      "training_step_time": 8.211533069610596
    },
    {
      "epoch": 1.842021942138672e-05,
      "model_forward_time": 1.712005853652954,
      "step": 19315
    },
    {
      "epoch": 1.842021942138672e-05,
      "step": 19315,
      "training_step_time": 7.58068060874939
    },
    {
      "epoch": 1.8421173095703124e-05,
      "model_forward_time": 1.7121996879577637,
      "step": 19316
    },
    {
      "epoch": 1.8421173095703124e-05,
      "step": 19316,
      "training_step_time": 7.239426612854004
    },
    {
      "epoch": 1.842212677001953e-05,
      "model_forward_time": 1.7100753784179688,
      "step": 19317
    },
    {
      "epoch": 1.842212677001953e-05,
      "step": 19317,
      "training_step_time": 7.696709394454956
    },
    {
      "epoch": 1.842308044433594e-05,
      "model_forward_time": 1.7107605934143066,
      "step": 19318
    },
    {
      "epoch": 1.842308044433594e-05,
      "step": 19318,
      "training_step_time": 7.236368656158447
    },
    {
      "epoch": 1.8424034118652343e-05,
      "model_forward_time": 1.717695951461792,
      "step": 19319
    },
    {
      "epoch": 1.8424034118652343e-05,
      "step": 19319,
      "training_step_time": 7.683631420135498
    },
    {
      "epoch": 1.842498779296875e-05,
      "grad_norm": 0.18006782233715057,
      "learning_rate": 9.449841476585518e-06,
      "loss": 0.0639,
      "step": 19320
    },
    {
      "epoch": 1.842498779296875e-05,
      "model_forward_time": 1.7094590663909912,
      "step": 19320
    },
    {
      "dynamics_loss_avg": 0.07351978868246078,
      "epoch": 1.842498779296875e-05,
      "step": 19320
    },
    {
      "action_loss_avg": 0.004192762763705104,
      "epoch": 1.842498779296875e-05,
      "step": 19320
    },
    {
      "epoch": 1.842498779296875e-05,
      "step": 19320,
      "training_step_time": 8.320635795593262
    },
    {
      "epoch": 1.8425941467285158e-05,
      "model_forward_time": 1.7082998752593994,
      "step": 19321
    },
    {
      "epoch": 1.8425941467285158e-05,
      "step": 19321,
      "training_step_time": 8.77780270576477
    },
    {
      "epoch": 1.8426895141601562e-05,
      "model_forward_time": 2.30022931098938,
      "step": 19322
    },
    {
      "epoch": 1.8426895141601562e-05,
      "step": 19322,
      "training_step_time": 7.7610533237457275
    },
    {
      "epoch": 1.842784881591797e-05,
      "model_forward_time": 1.0047917366027832,
      "step": 19323
    },
    {
      "epoch": 1.842784881591797e-05,
      "step": 19323,
      "training_step_time": 7.741070508956909
    },
    {
      "epoch": 1.8428802490234377e-05,
      "model_forward_time": 1.7169549465179443,
      "step": 19324
    },
    {
      "epoch": 1.8428802490234377e-05,
      "step": 19324,
      "training_step_time": 7.571832180023193
    },
    {
      "epoch": 1.842975616455078e-05,
      "model_forward_time": 1.707777976989746,
      "step": 19325
    },
    {
      "epoch": 1.842975616455078e-05,
      "step": 19325,
      "training_step_time": 7.214468717575073
    },
    {
      "epoch": 1.8430709838867188e-05,
      "model_forward_time": 1.7155210971832275,
      "step": 19326
    },
    {
      "epoch": 1.8430709838867188e-05,
      "step": 19326,
      "training_step_time": 7.688147068023682
    },
    {
      "epoch": 1.8431663513183595e-05,
      "model_forward_time": 1.7140405178070068,
      "step": 19327
    },
    {
      "epoch": 1.8431663513183595e-05,
      "step": 19327,
      "training_step_time": 7.201133489608765
    },
    {
      "epoch": 1.84326171875e-05,
      "model_forward_time": 1.7221686840057373,
      "step": 19328
    },
    {
      "epoch": 1.84326171875e-05,
      "step": 19328,
      "training_step_time": 8.49903392791748
    },
    {
      "epoch": 1.8433570861816407e-05,
      "model_forward_time": 1.7502145767211914,
      "step": 19329
    },
    {
      "epoch": 1.8433570861816407e-05,
      "step": 19329,
      "training_step_time": 7.74015736579895
    },
    {
      "epoch": 1.843452453613281e-05,
      "grad_norm": 0.17813140153884888,
      "learning_rate": 9.449087214003386e-06,
      "loss": 0.0681,
      "step": 19330
    },
    {
      "epoch": 1.843452453613281e-05,
      "model_forward_time": 1.6964046955108643,
      "step": 19330
    },
    {
      "dynamics_loss_avg": 0.07262791264802218,
      "epoch": 1.843452453613281e-05,
      "step": 19330
    },
    {
      "action_loss_avg": 0.0024733595782890916,
      "epoch": 1.843452453613281e-05,
      "step": 19330
    },
    {
      "epoch": 1.843452453613281e-05,
      "step": 19330,
      "training_step_time": 10.456114053726196
    },
    {
      "epoch": 1.8435478210449218e-05,
      "model_forward_time": 1.7021639347076416,
      "step": 19331
    },
    {
      "epoch": 1.8435478210449218e-05,
      "step": 19331,
      "training_step_time": 8.653616905212402
    },
    {
      "epoch": 1.8436431884765626e-05,
      "model_forward_time": 2.342668056488037,
      "step": 19332
    },
    {
      "epoch": 1.8436431884765626e-05,
      "step": 19332,
      "training_step_time": 7.762832880020142
    },
    {
      "epoch": 1.843738555908203e-05,
      "model_forward_time": 1.7001299858093262,
      "step": 19333
    },
    {
      "epoch": 1.843738555908203e-05,
      "step": 19333,
      "training_step_time": 7.5579094886779785
    },
    {
      "epoch": 1.8438339233398437e-05,
      "model_forward_time": 1.7023544311523438,
      "step": 19334
    },
    {
      "epoch": 1.8438339233398437e-05,
      "step": 19334,
      "training_step_time": 7.6083550453186035
    },
    {
      "epoch": 1.8439292907714845e-05,
      "model_forward_time": 1.7149503231048584,
      "step": 19335
    },
    {
      "epoch": 1.8439292907714845e-05,
      "step": 19335,
      "training_step_time": 7.591562986373901
    },
    {
      "epoch": 1.844024658203125e-05,
      "model_forward_time": 1.7121906280517578,
      "step": 19336
    },
    {
      "epoch": 1.844024658203125e-05,
      "step": 19336,
      "training_step_time": 7.247325658798218
    },
    {
      "epoch": 1.8441200256347656e-05,
      "model_forward_time": 1.7122743129730225,
      "step": 19337
    },
    {
      "epoch": 1.8441200256347656e-05,
      "step": 19337,
      "training_step_time": 7.381428956985474
    },
    {
      "epoch": 1.8442153930664063e-05,
      "model_forward_time": 1.708270788192749,
      "step": 19338
    },
    {
      "epoch": 1.8442153930664063e-05,
      "step": 19338,
      "training_step_time": 8.673927783966064
    },
    {
      "epoch": 1.8443107604980467e-05,
      "model_forward_time": 1.7123064994812012,
      "step": 19339
    },
    {
      "epoch": 1.8443107604980467e-05,
      "step": 19339,
      "training_step_time": 10.652948379516602
    },
    {
      "epoch": 1.8444061279296875e-05,
      "grad_norm": 0.20585636794567108,
      "learning_rate": 9.448332464875764e-06,
      "loss": 0.0645,
      "step": 19340
    },
    {
      "epoch": 1.8444061279296875e-05,
      "model_forward_time": 1.7114284038543701,
      "step": 19340
    },
    {
      "dynamics_loss_avg": 0.054466002993285655,
      "epoch": 1.8444061279296875e-05,
      "step": 19340
    },
    {
      "action_loss_avg": 0.0035769784211879597,
      "epoch": 1.8444061279296875e-05,
      "step": 19340
    },
    {
      "epoch": 1.8444061279296875e-05,
      "step": 19340,
      "training_step_time": 8.36883020401001
    },
    {
      "epoch": 1.8445014953613282e-05,
      "model_forward_time": 2.3103432655334473,
      "step": 19341
    },
    {
      "epoch": 1.8445014953613282e-05,
      "step": 19341,
      "training_step_time": 8.57334303855896
    },
    {
      "epoch": 1.8445968627929686e-05,
      "model_forward_time": 1.7123606204986572,
      "step": 19342
    },
    {
      "epoch": 1.8445968627929686e-05,
      "step": 19342,
      "training_step_time": 7.578437566757202
    },
    {
      "epoch": 1.8446922302246094e-05,
      "model_forward_time": 1.7090446949005127,
      "step": 19343
    },
    {
      "epoch": 1.8446922302246094e-05,
      "step": 19343,
      "training_step_time": 7.670177698135376
    },
    {
      "epoch": 1.84478759765625e-05,
      "model_forward_time": 1.7070505619049072,
      "step": 19344
    },
    {
      "epoch": 1.84478759765625e-05,
      "step": 19344,
      "training_step_time": 7.612128019332886
    },
    {
      "epoch": 1.8448829650878905e-05,
      "model_forward_time": 1.7132799625396729,
      "step": 19345
    },
    {
      "epoch": 1.8448829650878905e-05,
      "step": 19345,
      "training_step_time": 7.240232706069946
    },
    {
      "epoch": 1.8449783325195313e-05,
      "model_forward_time": 1.712968349456787,
      "step": 19346
    },
    {
      "epoch": 1.8449783325195313e-05,
      "step": 19346,
      "training_step_time": 7.397339105606079
    },
    {
      "epoch": 1.845073699951172e-05,
      "model_forward_time": 1.7111306190490723,
      "step": 19347
    },
    {
      "epoch": 1.845073699951172e-05,
      "step": 19347,
      "training_step_time": 8.768537998199463
    },
    {
      "epoch": 1.8451690673828124e-05,
      "model_forward_time": 1.3556909561157227,
      "step": 19348
    },
    {
      "epoch": 1.8451690673828124e-05,
      "step": 19348,
      "training_step_time": 10.515816926956177
    },
    {
      "epoch": 1.845264434814453e-05,
      "model_forward_time": 1.726928949356079,
      "step": 19349
    },
    {
      "epoch": 1.845264434814453e-05,
      "step": 19349,
      "training_step_time": 7.724474191665649
    },
    {
      "epoch": 1.845359802246094e-05,
      "grad_norm": 0.19386260211467743,
      "learning_rate": 9.447577229285191e-06,
      "loss": 0.0686,
      "step": 19350
    },
    {
      "epoch": 1.845359802246094e-05,
      "model_forward_time": 2.310323715209961,
      "step": 19350
    },
    {
      "dynamics_loss_avg": 0.049741584621369836,
      "epoch": 1.845359802246094e-05,
      "step": 19350
    },
    {
      "action_loss_avg": 0.0030866243119817226,
      "epoch": 1.845359802246094e-05,
      "step": 19350
    },
    {
      "epoch": 1.845359802246094e-05,
      "step": 19350,
      "training_step_time": 7.606357097625732
    },
    {
      "epoch": 1.8454551696777343e-05,
      "model_forward_time": 1.7301280498504639,
      "step": 19351
    },
    {
      "epoch": 1.8454551696777343e-05,
      "step": 19351,
      "training_step_time": 7.571310520172119
    },
    {
      "epoch": 1.845550537109375e-05,
      "model_forward_time": 1.7174201011657715,
      "step": 19352
    },
    {
      "epoch": 1.845550537109375e-05,
      "step": 19352,
      "training_step_time": 7.643553256988525
    },
    {
      "epoch": 1.8456459045410158e-05,
      "model_forward_time": 1.7252130508422852,
      "step": 19353
    },
    {
      "epoch": 1.8456459045410158e-05,
      "step": 19353,
      "training_step_time": 7.552520036697388
    },
    {
      "epoch": 1.8457412719726562e-05,
      "model_forward_time": 1.723806381225586,
      "step": 19354
    },
    {
      "epoch": 1.8457412719726562e-05,
      "step": 19354,
      "training_step_time": 7.542384624481201
    },
    {
      "epoch": 1.845836639404297e-05,
      "model_forward_time": 1.7087547779083252,
      "step": 19355
    },
    {
      "epoch": 1.845836639404297e-05,
      "step": 19355,
      "training_step_time": 7.325653076171875
    },
    {
      "epoch": 1.8459320068359377e-05,
      "model_forward_time": 1.3546488285064697,
      "step": 19356
    },
    {
      "epoch": 1.8459320068359377e-05,
      "step": 19356,
      "training_step_time": 7.749614715576172
    },
    {
      "epoch": 1.846027374267578e-05,
      "model_forward_time": 1.7210323810577393,
      "step": 19357
    },
    {
      "epoch": 1.846027374267578e-05,
      "step": 19357,
      "training_step_time": 9.334034442901611
    },
    {
      "epoch": 1.8461227416992188e-05,
      "model_forward_time": 1.7228295803070068,
      "step": 19358
    },
    {
      "epoch": 1.8461227416992188e-05,
      "step": 19358,
      "training_step_time": 9.795027256011963
    },
    {
      "epoch": 1.8462181091308595e-05,
      "model_forward_time": 2.2627346515655518,
      "step": 19359
    },
    {
      "epoch": 1.8462181091308595e-05,
      "step": 19359,
      "training_step_time": 8.356226682662964
    },
    {
      "epoch": 1.8463134765625e-05,
      "grad_norm": 0.2687196135520935,
      "learning_rate": 9.446821507314261e-06,
      "loss": 0.0618,
      "step": 19360
    },
    {
      "epoch": 1.8463134765625e-05,
      "model_forward_time": 1.718367576599121,
      "step": 19360
    },
    {
      "dynamics_loss_avg": 0.06297448361292482,
      "epoch": 1.8463134765625e-05,
      "step": 19360
    },
    {
      "action_loss_avg": 0.002902374300174415,
      "epoch": 1.8463134765625e-05,
      "step": 19360
    },
    {
      "epoch": 1.8463134765625e-05,
      "step": 19360,
      "training_step_time": 7.603664398193359
    },
    {
      "epoch": 1.8464088439941407e-05,
      "model_forward_time": 1.7190039157867432,
      "step": 19361
    },
    {
      "epoch": 1.8464088439941407e-05,
      "step": 19361,
      "training_step_time": 7.397754430770874
    },
    {
      "epoch": 1.846504211425781e-05,
      "model_forward_time": 1.715684175491333,
      "step": 19362
    },
    {
      "epoch": 1.846504211425781e-05,
      "step": 19362,
      "training_step_time": 7.597952127456665
    },
    {
      "epoch": 1.846599578857422e-05,
      "model_forward_time": 1.7173421382904053,
      "step": 19363
    },
    {
      "epoch": 1.846599578857422e-05,
      "step": 19363,
      "training_step_time": 7.600959777832031
    },
    {
      "epoch": 1.8466949462890626e-05,
      "model_forward_time": 1.7060163021087646,
      "step": 19364
    },
    {
      "epoch": 1.8466949462890626e-05,
      "step": 19364,
      "training_step_time": 7.884484767913818
    },
    {
      "epoch": 1.846790313720703e-05,
      "model_forward_time": 1.716649055480957,
      "step": 19365
    },
    {
      "epoch": 1.846790313720703e-05,
      "step": 19365,
      "training_step_time": 7.576274156570435
    },
    {
      "epoch": 1.8468856811523437e-05,
      "model_forward_time": 1.6942882537841797,
      "step": 19366
    },
    {
      "epoch": 1.8468856811523437e-05,
      "step": 19366,
      "training_step_time": 8.619570016860962
    },
    {
      "epoch": 1.8469810485839845e-05,
      "model_forward_time": 1.7093589305877686,
      "step": 19367
    },
    {
      "epoch": 1.8469810485839845e-05,
      "step": 19367,
      "training_step_time": 8.683536529541016
    },
    {
      "epoch": 1.847076416015625e-05,
      "model_forward_time": 1.7123069763183594,
      "step": 19368
    },
    {
      "epoch": 1.847076416015625e-05,
      "step": 19368,
      "training_step_time": 8.66655158996582
    },
    {
      "epoch": 1.8471717834472656e-05,
      "model_forward_time": 2.277216911315918,
      "step": 19369
    },
    {
      "epoch": 1.8471717834472656e-05,
      "step": 19369,
      "training_step_time": 7.500166654586792
    },
    {
      "epoch": 1.8472671508789063e-05,
      "grad_norm": 0.23216475546360016,
      "learning_rate": 9.446065299045617e-06,
      "loss": 0.0631,
      "step": 19370
    },
    {
      "epoch": 1.8472671508789063e-05,
      "model_forward_time": 1.007655382156372,
      "step": 19370
    },
    {
      "dynamics_loss_avg": 0.08847763836383819,
      "epoch": 1.8472671508789063e-05,
      "step": 19370
    },
    {
      "action_loss_avg": 0.003884897811803967,
      "epoch": 1.8472671508789063e-05,
      "step": 19370
    },
    {
      "epoch": 1.8472671508789063e-05,
      "step": 19370,
      "training_step_time": 7.3769917488098145
    },
    {
      "epoch": 1.8473625183105467e-05,
      "model_forward_time": 1.725834846496582,
      "step": 19371
    },
    {
      "epoch": 1.8473625183105467e-05,
      "step": 19371,
      "training_step_time": 7.617686748504639
    },
    {
      "epoch": 1.8474578857421875e-05,
      "model_forward_time": 1.7097032070159912,
      "step": 19372
    },
    {
      "epoch": 1.8474578857421875e-05,
      "step": 19372,
      "training_step_time": 7.577111721038818
    },
    {
      "epoch": 1.8475532531738282e-05,
      "model_forward_time": 1.713683843612671,
      "step": 19373
    },
    {
      "epoch": 1.8475532531738282e-05,
      "step": 19373,
      "training_step_time": 8.238571405410767
    },
    {
      "epoch": 1.8476486206054686e-05,
      "model_forward_time": 1.725123405456543,
      "step": 19374
    },
    {
      "epoch": 1.8476486206054686e-05,
      "step": 19374,
      "training_step_time": 7.589423656463623
    },
    {
      "epoch": 1.8477439880371094e-05,
      "model_forward_time": 1.353461503982544,
      "step": 19375
    },
    {
      "epoch": 1.8477439880371094e-05,
      "step": 19375,
      "training_step_time": 7.745837926864624
    },
    {
      "epoch": 1.84783935546875e-05,
      "model_forward_time": 1.7358591556549072,
      "step": 19376
    },
    {
      "epoch": 1.84783935546875e-05,
      "step": 19376,
      "training_step_time": 9.140252590179443
    },
    {
      "epoch": 1.8479347229003905e-05,
      "model_forward_time": 1.6969997882843018,
      "step": 19377
    },
    {
      "epoch": 1.8479347229003905e-05,
      "step": 19377,
      "training_step_time": 9.744011163711548
    },
    {
      "epoch": 1.8480300903320313e-05,
      "model_forward_time": 2.2815418243408203,
      "step": 19378
    },
    {
      "epoch": 1.8480300903320313e-05,
      "step": 19378,
      "training_step_time": 9.155004501342773
    },
    {
      "epoch": 1.848125457763672e-05,
      "model_forward_time": 1.7150936126708984,
      "step": 19379
    },
    {
      "epoch": 1.848125457763672e-05,
      "step": 19379,
      "training_step_time": 7.422140598297119
    },
    {
      "epoch": 1.8482208251953124e-05,
      "grad_norm": 0.18295681476593018,
      "learning_rate": 9.445308604561956e-06,
      "loss": 0.0631,
      "step": 19380
    },
    {
      "epoch": 1.8482208251953124e-05,
      "model_forward_time": 1.7148823738098145,
      "step": 19380
    },
    {
      "dynamics_loss_avg": 0.06433262974023819,
      "epoch": 1.8482208251953124e-05,
      "step": 19380
    },
    {
      "action_loss_avg": 0.0026555551448836923,
      "epoch": 1.8482208251953124e-05,
      "step": 19380
    },
    {
      "epoch": 1.8482208251953124e-05,
      "step": 19380,
      "training_step_time": 7.645161867141724
    },
    {
      "epoch": 1.848316192626953e-05,
      "model_forward_time": 1.7336907386779785,
      "step": 19381
    },
    {
      "epoch": 1.848316192626953e-05,
      "step": 19381,
      "training_step_time": 7.490807056427002
    },
    {
      "epoch": 1.848411560058594e-05,
      "model_forward_time": 1.7299389839172363,
      "step": 19382
    },
    {
      "epoch": 1.848411560058594e-05,
      "step": 19382,
      "training_step_time": 7.387688159942627
    },
    {
      "epoch": 1.8485069274902343e-05,
      "model_forward_time": 1.7283854484558105,
      "step": 19383
    },
    {
      "epoch": 1.8485069274902343e-05,
      "step": 19383,
      "training_step_time": 7.666668891906738
    },
    {
      "epoch": 1.848602294921875e-05,
      "model_forward_time": 1.7224767208099365,
      "step": 19384
    },
    {
      "epoch": 1.848602294921875e-05,
      "step": 19384,
      "training_step_time": 8.67667555809021
    },
    {
      "epoch": 1.8486976623535158e-05,
      "model_forward_time": 1.7246832847595215,
      "step": 19385
    },
    {
      "epoch": 1.8486976623535158e-05,
      "step": 19385,
      "training_step_time": 8.576860189437866
    },
    {
      "epoch": 1.8487930297851562e-05,
      "model_forward_time": 1.725158452987671,
      "step": 19386
    },
    {
      "epoch": 1.8487930297851562e-05,
      "step": 19386,
      "training_step_time": 7.723085403442383
    },
    {
      "epoch": 1.848888397216797e-05,
      "model_forward_time": 2.297433853149414,
      "step": 19387
    },
    {
      "epoch": 1.848888397216797e-05,
      "step": 19387,
      "training_step_time": 9.379661321640015
    },
    {
      "epoch": 1.8489837646484377e-05,
      "model_forward_time": 1.7199761867523193,
      "step": 19388
    },
    {
      "epoch": 1.8489837646484377e-05,
      "step": 19388,
      "training_step_time": 7.561007499694824
    },
    {
      "epoch": 1.849079132080078e-05,
      "model_forward_time": 1.7185213565826416,
      "step": 19389
    },
    {
      "epoch": 1.849079132080078e-05,
      "step": 19389,
      "training_step_time": 7.626237630844116
    },
    {
      "epoch": 1.8491744995117188e-05,
      "grad_norm": 0.1632729470729828,
      "learning_rate": 9.444551423946028e-06,
      "loss": 0.0615,
      "step": 19390
    },
    {
      "epoch": 1.8491744995117188e-05,
      "model_forward_time": 1.7170765399932861,
      "step": 19390
    },
    {
      "dynamics_loss_avg": 0.05879569686949253,
      "epoch": 1.8491744995117188e-05,
      "step": 19390
    },
    {
      "action_loss_avg": 0.0019352595030795784,
      "epoch": 1.8491744995117188e-05,
      "step": 19390
    },
    {
      "epoch": 1.8491744995117188e-05,
      "step": 19390,
      "training_step_time": 7.282455205917358
    },
    {
      "epoch": 1.8492698669433592e-05,
      "model_forward_time": 1.7175259590148926,
      "step": 19391
    },
    {
      "epoch": 1.8492698669433592e-05,
      "step": 19391,
      "training_step_time": 7.696528196334839
    },
    {
      "epoch": 1.849365234375e-05,
      "model_forward_time": 1.7230160236358643,
      "step": 19392
    },
    {
      "epoch": 1.849365234375e-05,
      "step": 19392,
      "training_step_time": 8.653259754180908
    },
    {
      "epoch": 1.8494606018066407e-05,
      "model_forward_time": 1.7227578163146973,
      "step": 19393
    },
    {
      "epoch": 1.8494606018066407e-05,
      "step": 19393,
      "training_step_time": 9.458248376846313
    },
    {
      "epoch": 1.849555969238281e-05,
      "model_forward_time": 1.7214834690093994,
      "step": 19394
    },
    {
      "epoch": 1.849555969238281e-05,
      "step": 19394,
      "training_step_time": 7.941529035568237
    },
    {
      "epoch": 1.849651336669922e-05,
      "model_forward_time": 1.722501277923584,
      "step": 19395
    },
    {
      "epoch": 1.849651336669922e-05,
      "step": 19395,
      "training_step_time": 8.841932773590088
    },
    {
      "epoch": 1.8497467041015626e-05,
      "model_forward_time": 1.7152042388916016,
      "step": 19396
    },
    {
      "epoch": 1.8497467041015626e-05,
      "step": 19396,
      "training_step_time": 9.594090223312378
    },
    {
      "epoch": 1.849842071533203e-05,
      "model_forward_time": 1.7137401103973389,
      "step": 19397
    },
    {
      "epoch": 1.849842071533203e-05,
      "step": 19397,
      "training_step_time": 8.462955951690674
    },
    {
      "epoch": 1.8499374389648437e-05,
      "model_forward_time": 1.722012996673584,
      "step": 19398
    },
    {
      "epoch": 1.8499374389648437e-05,
      "step": 19398,
      "training_step_time": 7.677375555038452
    },
    {
      "epoch": 1.8500328063964845e-05,
      "model_forward_time": 1.7138686180114746,
      "step": 19399
    },
    {
      "epoch": 1.8500328063964845e-05,
      "step": 19399,
      "training_step_time": 7.2833216190338135
    },
    {
      "epoch": 1.850128173828125e-05,
      "grad_norm": 0.28367793560028076,
      "learning_rate": 9.44379375728064e-06,
      "loss": 0.0612,
      "step": 19400
    },
    {
      "epoch": 1.850128173828125e-05,
      "model_forward_time": 1.7148997783660889,
      "step": 19400
    },
    {
      "dynamics_loss_avg": 0.06919531095772982,
      "epoch": 1.850128173828125e-05,
      "step": 19400
    },
    {
      "action_loss_avg": 0.0027480059943627566,
      "epoch": 1.850128173828125e-05,
      "step": 19400
    },
    {
      "epoch": 1.850128173828125e-05,
      "step": 19400,
      "training_step_time": 7.653067111968994
    },
    {
      "epoch": 1.8502235412597656e-05,
      "model_forward_time": 1.7150323390960693,
      "step": 19401
    },
    {
      "epoch": 1.8502235412597656e-05,
      "step": 19401,
      "training_step_time": 8.298535108566284
    },
    {
      "epoch": 1.8503189086914063e-05,
      "model_forward_time": 1.710676670074463,
      "step": 19402
    },
    {
      "epoch": 1.8503189086914063e-05,
      "step": 19402,
      "training_step_time": 7.718825101852417
    },
    {
      "epoch": 1.8504142761230468e-05,
      "model_forward_time": 1.7267634868621826,
      "step": 19403
    },
    {
      "epoch": 1.8504142761230468e-05,
      "step": 19403,
      "training_step_time": 8.506057977676392
    },
    {
      "epoch": 1.8505096435546875e-05,
      "model_forward_time": 1.7124645709991455,
      "step": 19404
    },
    {
      "epoch": 1.8505096435546875e-05,
      "step": 19404,
      "training_step_time": 9.440504789352417
    },
    {
      "epoch": 1.8506050109863282e-05,
      "model_forward_time": 1.7269184589385986,
      "step": 19405
    },
    {
      "epoch": 1.8506050109863282e-05,
      "step": 19405,
      "training_step_time": 9.0154869556427
    },
    {
      "epoch": 1.8507003784179686e-05,
      "model_forward_time": 2.300058603286743,
      "step": 19406
    },
    {
      "epoch": 1.8507003784179686e-05,
      "step": 19406,
      "training_step_time": 7.741087913513184
    },
    {
      "epoch": 1.8507957458496094e-05,
      "model_forward_time": 1.7259297370910645,
      "step": 19407
    },
    {
      "epoch": 1.8507957458496094e-05,
      "step": 19407,
      "training_step_time": 7.590418338775635
    },
    {
      "epoch": 1.85089111328125e-05,
      "model_forward_time": 1.7145814895629883,
      "step": 19408
    },
    {
      "epoch": 1.85089111328125e-05,
      "step": 19408,
      "training_step_time": 7.647084474563599
    },
    {
      "epoch": 1.8509864807128905e-05,
      "model_forward_time": 1.7174310684204102,
      "step": 19409
    },
    {
      "epoch": 1.8509864807128905e-05,
      "step": 19409,
      "training_step_time": 7.661917686462402
    },
    {
      "epoch": 1.8510818481445313e-05,
      "grad_norm": 0.292914479970932,
      "learning_rate": 9.443035604648645e-06,
      "loss": 0.0652,
      "step": 19410
    },
    {
      "epoch": 1.8510818481445313e-05,
      "model_forward_time": 1.7165014743804932,
      "step": 19410
    },
    {
      "dynamics_loss_avg": 0.053996685706079,
      "epoch": 1.8510818481445313e-05,
      "step": 19410
    },
    {
      "action_loss_avg": 0.0023787378158885986,
      "epoch": 1.8510818481445313e-05,
      "step": 19410
    },
    {
      "epoch": 1.8510818481445313e-05,
      "step": 19410,
      "training_step_time": 7.379922151565552
    },
    {
      "epoch": 1.851177215576172e-05,
      "model_forward_time": 1.7229716777801514,
      "step": 19411
    },
    {
      "epoch": 1.851177215576172e-05,
      "step": 19411,
      "training_step_time": 7.644453048706055
    },
    {
      "epoch": 1.8512725830078124e-05,
      "model_forward_time": 1.706984519958496,
      "step": 19412
    },
    {
      "epoch": 1.8512725830078124e-05,
      "step": 19412,
      "training_step_time": 7.650787115097046
    },
    {
      "epoch": 1.851367950439453e-05,
      "model_forward_time": 1.7168018817901611,
      "step": 19413
    },
    {
      "epoch": 1.851367950439453e-05,
      "step": 19413,
      "training_step_time": 7.736579418182373
    },
    {
      "epoch": 1.851463317871094e-05,
      "model_forward_time": 1.7205383777618408,
      "step": 19414
    },
    {
      "epoch": 1.851463317871094e-05,
      "step": 19414,
      "training_step_time": 7.63077449798584
    },
    {
      "epoch": 1.8515586853027343e-05,
      "model_forward_time": 1.9792578220367432,
      "step": 19415
    },
    {
      "epoch": 1.8515586853027343e-05,
      "step": 19415,
      "training_step_time": 8.591440200805664
    },
    {
      "epoch": 1.851654052734375e-05,
      "model_forward_time": 1.7339484691619873,
      "step": 19416
    },
    {
      "epoch": 1.851654052734375e-05,
      "step": 19416,
      "training_step_time": 7.5915772914886475
    },
    {
      "epoch": 1.8517494201660158e-05,
      "model_forward_time": 1.7134902477264404,
      "step": 19417
    },
    {
      "epoch": 1.8517494201660158e-05,
      "step": 19417,
      "training_step_time": 7.650312185287476
    },
    {
      "epoch": 1.8518447875976562e-05,
      "model_forward_time": 1.7169616222381592,
      "step": 19418
    },
    {
      "epoch": 1.8518447875976562e-05,
      "step": 19418,
      "training_step_time": 7.627516031265259
    },
    {
      "epoch": 1.851940155029297e-05,
      "model_forward_time": 1.713470697402954,
      "step": 19419
    },
    {
      "epoch": 1.851940155029297e-05,
      "step": 19419,
      "training_step_time": 7.367394685745239
    },
    {
      "epoch": 1.8520355224609377e-05,
      "grad_norm": 0.29480621218681335,
      "learning_rate": 9.44227696613296e-06,
      "loss": 0.0656,
      "step": 19420
    },
    {
      "epoch": 1.8520355224609377e-05,
      "model_forward_time": 1.7245771884918213,
      "step": 19420
    },
    {
      "dynamics_loss_avg": 0.050875529553741215,
      "epoch": 1.8520355224609377e-05,
      "step": 19420
    },
    {
      "action_loss_avg": 0.0028425209573470058,
      "epoch": 1.8520355224609377e-05,
      "step": 19420
    },
    {
      "epoch": 1.8520355224609377e-05,
      "step": 19420,
      "training_step_time": 7.637723922729492
    },
    {
      "epoch": 1.852130889892578e-05,
      "model_forward_time": 1.7279319763183594,
      "step": 19421
    },
    {
      "epoch": 1.852130889892578e-05,
      "step": 19421,
      "training_step_time": 9.711527585983276
    },
    {
      "epoch": 1.8522262573242188e-05,
      "model_forward_time": 1.7157394886016846,
      "step": 19422
    },
    {
      "epoch": 1.8522262573242188e-05,
      "step": 19422,
      "training_step_time": 8.4591805934906
    },
    {
      "epoch": 1.8523216247558592e-05,
      "model_forward_time": 1.716953992843628,
      "step": 19423
    },
    {
      "epoch": 1.8523216247558592e-05,
      "step": 19423,
      "training_step_time": 8.379475355148315
    },
    {
      "epoch": 1.8524169921875e-05,
      "model_forward_time": 1.9878814220428467,
      "step": 19424
    },
    {
      "epoch": 1.8524169921875e-05,
      "step": 19424,
      "training_step_time": 7.766252517700195
    },
    {
      "epoch": 1.8525123596191407e-05,
      "model_forward_time": 1.7250890731811523,
      "step": 19425
    },
    {
      "epoch": 1.8525123596191407e-05,
      "step": 19425,
      "training_step_time": 7.620295286178589
    },
    {
      "epoch": 1.852607727050781e-05,
      "model_forward_time": 1.7222826480865479,
      "step": 19426
    },
    {
      "epoch": 1.852607727050781e-05,
      "step": 19426,
      "training_step_time": 7.6320812702178955
    },
    {
      "epoch": 1.852703094482422e-05,
      "model_forward_time": 1.7197322845458984,
      "step": 19427
    },
    {
      "epoch": 1.852703094482422e-05,
      "step": 19427,
      "training_step_time": 7.598897695541382
    },
    {
      "epoch": 1.8527984619140626e-05,
      "model_forward_time": 1.724858045578003,
      "step": 19428
    },
    {
      "epoch": 1.8527984619140626e-05,
      "step": 19428,
      "training_step_time": 7.449166774749756
    },
    {
      "epoch": 1.852893829345703e-05,
      "model_forward_time": 1.72853684425354,
      "step": 19429
    },
    {
      "epoch": 1.852893829345703e-05,
      "step": 19429,
      "training_step_time": 7.718897342681885
    },
    {
      "epoch": 1.8529891967773437e-05,
      "grad_norm": 0.33832278847694397,
      "learning_rate": 9.441517841816543e-06,
      "loss": 0.0679,
      "step": 19430
    },
    {
      "epoch": 1.8529891967773437e-05,
      "model_forward_time": 1.7218573093414307,
      "step": 19430
    },
    {
      "dynamics_loss_avg": 0.04244506387040019,
      "epoch": 1.8529891967773437e-05,
      "step": 19430
    },
    {
      "action_loss_avg": 0.0022019168798578905,
      "epoch": 1.8529891967773437e-05,
      "step": 19430
    },
    {
      "epoch": 1.8529891967773437e-05,
      "step": 19430,
      "training_step_time": 7.68524169921875
    },
    {
      "epoch": 1.8530845642089845e-05,
      "model_forward_time": 1.7245423793792725,
      "step": 19431
    },
    {
      "epoch": 1.8530845642089845e-05,
      "step": 19431,
      "training_step_time": 8.363046884536743
    },
    {
      "epoch": 1.853179931640625e-05,
      "model_forward_time": 1.7190415859222412,
      "step": 19432
    },
    {
      "epoch": 1.853179931640625e-05,
      "step": 19432,
      "training_step_time": 7.733758211135864
    },
    {
      "epoch": 1.8532752990722656e-05,
      "model_forward_time": 2.2880115509033203,
      "step": 19433
    },
    {
      "epoch": 1.8532752990722656e-05,
      "step": 19433,
      "training_step_time": 7.822190523147583
    },
    {
      "epoch": 1.8533706665039064e-05,
      "model_forward_time": 1.718878984451294,
      "step": 19434
    },
    {
      "epoch": 1.8533706665039064e-05,
      "step": 19434,
      "training_step_time": 7.709266424179077
    },
    {
      "epoch": 1.8534660339355468e-05,
      "model_forward_time": 1.720191240310669,
      "step": 19435
    },
    {
      "epoch": 1.8534660339355468e-05,
      "step": 19435,
      "training_step_time": 7.88890528678894
    },
    {
      "epoch": 1.8535614013671875e-05,
      "model_forward_time": 1.354858636856079,
      "step": 19436
    },
    {
      "epoch": 1.8535614013671875e-05,
      "step": 19436,
      "training_step_time": 8.346805334091187
    },
    {
      "epoch": 1.8536567687988282e-05,
      "model_forward_time": 1.7150819301605225,
      "step": 19437
    },
    {
      "epoch": 1.8536567687988282e-05,
      "step": 19437,
      "training_step_time": 7.386128902435303
    },
    {
      "epoch": 1.8537521362304686e-05,
      "model_forward_time": 1.719642162322998,
      "step": 19438
    },
    {
      "epoch": 1.8537521362304686e-05,
      "step": 19438,
      "training_step_time": 7.665909290313721
    },
    {
      "epoch": 1.8538475036621094e-05,
      "model_forward_time": 1.7175593376159668,
      "step": 19439
    },
    {
      "epoch": 1.8538475036621094e-05,
      "step": 19439,
      "training_step_time": 7.679527044296265
    },
    {
      "epoch": 1.85394287109375e-05,
      "grad_norm": 0.26318222284317017,
      "learning_rate": 9.440758231782414e-06,
      "loss": 0.0618,
      "step": 19440
    },
    {
      "epoch": 1.85394287109375e-05,
      "model_forward_time": 1.7201576232910156,
      "step": 19440
    },
    {
      "dynamics_loss_avg": 0.06623926665633917,
      "epoch": 1.85394287109375e-05,
      "step": 19440
    },
    {
      "action_loss_avg": 0.0032316686934791504,
      "epoch": 1.85394287109375e-05,
      "step": 19440
    },
    {
      "epoch": 1.85394287109375e-05,
      "step": 19440,
      "training_step_time": 8.514014959335327
    },
    {
      "epoch": 1.8540382385253905e-05,
      "model_forward_time": 1.7249128818511963,
      "step": 19441
    },
    {
      "epoch": 1.8540382385253905e-05,
      "step": 19441,
      "training_step_time": 8.828636169433594
    },
    {
      "epoch": 1.8541336059570313e-05,
      "model_forward_time": 2.270029067993164,
      "step": 19442
    },
    {
      "epoch": 1.8541336059570313e-05,
      "step": 19442,
      "training_step_time": 7.854958534240723
    },
    {
      "epoch": 1.854228973388672e-05,
      "model_forward_time": 1.7219700813293457,
      "step": 19443
    },
    {
      "epoch": 1.854228973388672e-05,
      "step": 19443,
      "training_step_time": 7.680042028427124
    },
    {
      "epoch": 1.8543243408203124e-05,
      "model_forward_time": 1.7202198505401611,
      "step": 19444
    },
    {
      "epoch": 1.8543243408203124e-05,
      "step": 19444,
      "training_step_time": 7.565744638442993
    },
    {
      "epoch": 1.854419708251953e-05,
      "model_forward_time": 1.7215118408203125,
      "step": 19445
    },
    {
      "epoch": 1.854419708251953e-05,
      "step": 19445,
      "training_step_time": 7.640468120574951
    },
    {
      "epoch": 1.854515075683594e-05,
      "model_forward_time": 1.7186832427978516,
      "step": 19446
    },
    {
      "epoch": 1.854515075683594e-05,
      "step": 19446,
      "training_step_time": 7.619233846664429
    },
    {
      "epoch": 1.8546104431152343e-05,
      "model_forward_time": 1.7200498580932617,
      "step": 19447
    },
    {
      "epoch": 1.8546104431152343e-05,
      "step": 19447,
      "training_step_time": 7.648353576660156
    },
    {
      "epoch": 1.854705810546875e-05,
      "model_forward_time": 1.7391376495361328,
      "step": 19448
    },
    {
      "epoch": 1.854705810546875e-05,
      "step": 19448,
      "training_step_time": 7.7493250370025635
    },
    {
      "epoch": 1.8548011779785158e-05,
      "model_forward_time": 1.7407238483428955,
      "step": 19449
    },
    {
      "epoch": 1.8548011779785158e-05,
      "step": 19449,
      "training_step_time": 8.309224367141724
    },
    {
      "epoch": 1.8548965454101562e-05,
      "grad_norm": 0.2221105843782425,
      "learning_rate": 9.43999813611364e-06,
      "loss": 0.068,
      "step": 19450
    },
    {
      "epoch": 1.8548965454101562e-05,
      "model_forward_time": 1.7177302837371826,
      "step": 19450
    },
    {
      "dynamics_loss_avg": 0.06059242654591799,
      "epoch": 1.8548965454101562e-05,
      "step": 19450
    },
    {
      "action_loss_avg": 0.0021303621120750903,
      "epoch": 1.8548965454101562e-05,
      "step": 19450
    },
    {
      "epoch": 1.8548965454101562e-05,
      "step": 19450,
      "training_step_time": 8.23521113395691
    },
    {
      "epoch": 1.854991912841797e-05,
      "model_forward_time": 1.3676095008850098,
      "step": 19451
    },
    {
      "epoch": 1.854991912841797e-05,
      "step": 19451,
      "training_step_time": 8.300642013549805
    },
    {
      "epoch": 1.8550872802734377e-05,
      "model_forward_time": 2.3012025356292725,
      "step": 19452
    },
    {
      "epoch": 1.8550872802734377e-05,
      "step": 19452,
      "training_step_time": 7.699198246002197
    },
    {
      "epoch": 1.855182647705078e-05,
      "model_forward_time": 1.7242035865783691,
      "step": 19453
    },
    {
      "epoch": 1.855182647705078e-05,
      "step": 19453,
      "training_step_time": 8.362920999526978
    },
    {
      "epoch": 1.8552780151367188e-05,
      "model_forward_time": 1.7201128005981445,
      "step": 19454
    },
    {
      "epoch": 1.8552780151367188e-05,
      "step": 19454,
      "training_step_time": 7.604551076889038
    },
    {
      "epoch": 1.8553733825683592e-05,
      "model_forward_time": 1.7251474857330322,
      "step": 19455
    },
    {
      "epoch": 1.8553733825683592e-05,
      "step": 19455,
      "training_step_time": 7.586895704269409
    },
    {
      "epoch": 1.85546875e-05,
      "model_forward_time": 1.7223966121673584,
      "step": 19456
    },
    {
      "epoch": 1.85546875e-05,
      "step": 19456,
      "training_step_time": 7.6073853969573975
    },
    {
      "epoch": 1.8555641174316407e-05,
      "model_forward_time": 1.721421241760254,
      "step": 19457
    },
    {
      "epoch": 1.8555641174316407e-05,
      "step": 19457,
      "training_step_time": 8.764830589294434
    },
    {
      "epoch": 1.855659484863281e-05,
      "model_forward_time": 1.7410039901733398,
      "step": 19458
    },
    {
      "epoch": 1.855659484863281e-05,
      "step": 19458,
      "training_step_time": 7.713376522064209
    },
    {
      "epoch": 1.855754852294922e-05,
      "model_forward_time": 1.7358894348144531,
      "step": 19459
    },
    {
      "epoch": 1.855754852294922e-05,
      "step": 19459,
      "training_step_time": 8.852822542190552
    },
    {
      "epoch": 1.8558502197265626e-05,
      "grad_norm": 0.2639317810535431,
      "learning_rate": 9.439237554893344e-06,
      "loss": 0.0636,
      "step": 19460
    },
    {
      "epoch": 1.8558502197265626e-05,
      "model_forward_time": 1.7202870845794678,
      "step": 19460
    },
    {
      "dynamics_loss_avg": 0.07019985988736152,
      "epoch": 1.8558502197265626e-05,
      "step": 19460
    },
    {
      "action_loss_avg": 0.004983877867925912,
      "epoch": 1.8558502197265626e-05,
      "step": 19460
    },
    {
      "epoch": 1.8558502197265626e-05,
      "step": 19460,
      "training_step_time": 8.115227699279785
    },
    {
      "epoch": 1.855945587158203e-05,
      "model_forward_time": 2.2884652614593506,
      "step": 19461
    },
    {
      "epoch": 1.855945587158203e-05,
      "step": 19461,
      "training_step_time": 10.384546995162964
    },
    {
      "epoch": 1.8560409545898437e-05,
      "model_forward_time": 1.7201220989227295,
      "step": 19462
    },
    {
      "epoch": 1.8560409545898437e-05,
      "step": 19462,
      "training_step_time": 7.536987781524658
    },
    {
      "epoch": 1.8561363220214845e-05,
      "model_forward_time": 1.721940040588379,
      "step": 19463
    },
    {
      "epoch": 1.8561363220214845e-05,
      "step": 19463,
      "training_step_time": 7.676657676696777
    },
    {
      "epoch": 1.856231689453125e-05,
      "model_forward_time": 1.7241320610046387,
      "step": 19464
    },
    {
      "epoch": 1.856231689453125e-05,
      "step": 19464,
      "training_step_time": 7.617686748504639
    },
    {
      "epoch": 1.8563270568847656e-05,
      "model_forward_time": 1.7217845916748047,
      "step": 19465
    },
    {
      "epoch": 1.8563270568847656e-05,
      "step": 19465,
      "training_step_time": 7.658116102218628
    },
    {
      "epoch": 1.8564224243164064e-05,
      "model_forward_time": 1.7259609699249268,
      "step": 19466
    },
    {
      "epoch": 1.8564224243164064e-05,
      "step": 19466,
      "training_step_time": 8.552548170089722
    },
    {
      "epoch": 1.8565177917480468e-05,
      "model_forward_time": 1.7189905643463135,
      "step": 19467
    },
    {
      "epoch": 1.8565177917480468e-05,
      "step": 19467,
      "training_step_time": 7.685438632965088
    },
    {
      "epoch": 1.8566131591796875e-05,
      "model_forward_time": 1.7166519165039062,
      "step": 19468
    },
    {
      "epoch": 1.8566131591796875e-05,
      "step": 19468,
      "training_step_time": 7.7285003662109375
    },
    {
      "epoch": 1.8567085266113282e-05,
      "model_forward_time": 1.7366578578948975,
      "step": 19469
    },
    {
      "epoch": 1.8567085266113282e-05,
      "step": 19469,
      "training_step_time": 7.790869951248169
    },
    {
      "epoch": 1.8568038940429686e-05,
      "grad_norm": 0.20026162266731262,
      "learning_rate": 9.438476488204706e-06,
      "loss": 0.0643,
      "step": 19470
    },
    {
      "epoch": 1.8568038940429686e-05,
      "model_forward_time": 2.2866621017456055,
      "step": 19470
    },
    {
      "dynamics_loss_avg": 0.07681434005498886,
      "epoch": 1.8568038940429686e-05,
      "step": 19470
    },
    {
      "action_loss_avg": 0.0031116977566853165,
      "epoch": 1.8568038940429686e-05,
      "step": 19470
    },
    {
      "epoch": 1.8568038940429686e-05,
      "step": 19470,
      "training_step_time": 7.682872772216797
    },
    {
      "epoch": 1.8568992614746094e-05,
      "model_forward_time": 1.717839241027832,
      "step": 19471
    },
    {
      "epoch": 1.8568992614746094e-05,
      "step": 19471,
      "training_step_time": 8.411378383636475
    },
    {
      "epoch": 1.85699462890625e-05,
      "model_forward_time": 1.7195308208465576,
      "step": 19472
    },
    {
      "epoch": 1.85699462890625e-05,
      "step": 19472,
      "training_step_time": 7.68328332901001
    },
    {
      "epoch": 1.8570899963378905e-05,
      "model_forward_time": 1.7223939895629883,
      "step": 19473
    },
    {
      "epoch": 1.8570899963378905e-05,
      "step": 19473,
      "training_step_time": 7.665262460708618
    },
    {
      "epoch": 1.8571853637695313e-05,
      "model_forward_time": 1.7195725440979004,
      "step": 19474
    },
    {
      "epoch": 1.8571853637695313e-05,
      "step": 19474,
      "training_step_time": 7.681994915008545
    },
    {
      "epoch": 1.857280731201172e-05,
      "model_forward_time": 1.725484848022461,
      "step": 19475
    },
    {
      "epoch": 1.857280731201172e-05,
      "step": 19475,
      "training_step_time": 8.274165868759155
    },
    {
      "epoch": 1.8573760986328124e-05,
      "model_forward_time": 1.722036361694336,
      "step": 19476
    },
    {
      "epoch": 1.8573760986328124e-05,
      "step": 19476,
      "training_step_time": 8.408328294754028
    },
    {
      "epoch": 1.857471466064453e-05,
      "model_forward_time": 1.7170133590698242,
      "step": 19477
    },
    {
      "epoch": 1.857471466064453e-05,
      "step": 19477,
      "training_step_time": 7.8020031452178955
    },
    {
      "epoch": 1.857566833496094e-05,
      "model_forward_time": 1.712129831314087,
      "step": 19478
    },
    {
      "epoch": 1.857566833496094e-05,
      "step": 19478,
      "training_step_time": 7.725893259048462
    },
    {
      "epoch": 1.8576622009277343e-05,
      "model_forward_time": 2.2748830318450928,
      "step": 19479
    },
    {
      "epoch": 1.8576622009277343e-05,
      "step": 19479,
      "training_step_time": 9.510581731796265
    },
    {
      "epoch": 1.857757568359375e-05,
      "grad_norm": 0.22535665333271027,
      "learning_rate": 9.43771493613095e-06,
      "loss": 0.0648,
      "step": 19480
    },
    {
      "epoch": 1.857757568359375e-05,
      "model_forward_time": 1.3578555583953857,
      "step": 19480
    },
    {
      "dynamics_loss_avg": 0.0664334911853075,
      "epoch": 1.857757568359375e-05,
      "step": 19480
    },
    {
      "action_loss_avg": 0.0019964523729868234,
      "epoch": 1.857757568359375e-05,
      "step": 19480
    },
    {
      "epoch": 1.857757568359375e-05,
      "step": 19480,
      "training_step_time": 8.373996257781982
    },
    {
      "epoch": 1.8578529357910158e-05,
      "model_forward_time": 1.7196295261383057,
      "step": 19481
    },
    {
      "epoch": 1.8578529357910158e-05,
      "step": 19481,
      "training_step_time": 7.593335390090942
    },
    {
      "epoch": 1.8579483032226562e-05,
      "model_forward_time": 1.363555669784546,
      "step": 19482
    },
    {
      "epoch": 1.8579483032226562e-05,
      "step": 19482,
      "training_step_time": 8.391264915466309
    },
    {
      "epoch": 1.858043670654297e-05,
      "model_forward_time": 1.7159981727600098,
      "step": 19483
    },
    {
      "epoch": 1.858043670654297e-05,
      "step": 19483,
      "training_step_time": 7.697114706039429
    },
    {
      "epoch": 1.8581390380859377e-05,
      "model_forward_time": 1.715489387512207,
      "step": 19484
    },
    {
      "epoch": 1.8581390380859377e-05,
      "step": 19484,
      "training_step_time": 8.576315879821777
    },
    {
      "epoch": 1.858234405517578e-05,
      "model_forward_time": 1.7143511772155762,
      "step": 19485
    },
    {
      "epoch": 1.858234405517578e-05,
      "step": 19485,
      "training_step_time": 8.686954259872437
    },
    {
      "epoch": 1.8583297729492188e-05,
      "model_forward_time": 1.7109489440917969,
      "step": 19486
    },
    {
      "epoch": 1.8583297729492188e-05,
      "step": 19486,
      "training_step_time": 7.779571294784546
    },
    {
      "epoch": 1.8584251403808592e-05,
      "model_forward_time": 1.721376895904541,
      "step": 19487
    },
    {
      "epoch": 1.8584251403808592e-05,
      "step": 19487,
      "training_step_time": 7.666178464889526
    },
    {
      "epoch": 1.8585205078125e-05,
      "model_forward_time": 1.359590768814087,
      "step": 19488
    },
    {
      "epoch": 1.8585205078125e-05,
      "step": 19488,
      "training_step_time": 7.675204038619995
    },
    {
      "epoch": 1.8586158752441407e-05,
      "model_forward_time": 2.2589335441589355,
      "step": 19489
    },
    {
      "epoch": 1.8586158752441407e-05,
      "step": 19489,
      "training_step_time": 7.702294111251831
    },
    {
      "epoch": 1.858711242675781e-05,
      "grad_norm": 0.17721134424209595,
      "learning_rate": 9.436952898755362e-06,
      "loss": 0.062,
      "step": 19490
    },
    {
      "epoch": 1.858711242675781e-05,
      "model_forward_time": 1.712395191192627,
      "step": 19490
    },
    {
      "dynamics_loss_avg": 0.06451003141701221,
      "epoch": 1.858711242675781e-05,
      "step": 19490
    },
    {
      "action_loss_avg": 0.002108455856796354,
      "epoch": 1.858711242675781e-05,
      "step": 19490
    },
    {
      "epoch": 1.858711242675781e-05,
      "step": 19490,
      "training_step_time": 7.406728982925415
    },
    {
      "epoch": 1.858806610107422e-05,
      "model_forward_time": 1.7185544967651367,
      "step": 19491
    },
    {
      "epoch": 1.858806610107422e-05,
      "step": 19491,
      "training_step_time": 7.620789051055908
    },
    {
      "epoch": 1.8589019775390626e-05,
      "model_forward_time": 1.7176597118377686,
      "step": 19492
    },
    {
      "epoch": 1.8589019775390626e-05,
      "step": 19492,
      "training_step_time": 7.65117073059082
    },
    {
      "epoch": 1.858997344970703e-05,
      "model_forward_time": 1.7235605716705322,
      "step": 19493
    },
    {
      "epoch": 1.858997344970703e-05,
      "step": 19493,
      "training_step_time": 7.834426641464233
    },
    {
      "epoch": 1.8590927124023437e-05,
      "model_forward_time": 1.7161543369293213,
      "step": 19494
    },
    {
      "epoch": 1.8590927124023437e-05,
      "step": 19494,
      "training_step_time": 7.607555150985718
    },
    {
      "epoch": 1.8591880798339845e-05,
      "model_forward_time": 1.7162277698516846,
      "step": 19495
    },
    {
      "epoch": 1.8591880798339845e-05,
      "step": 19495,
      "training_step_time": 8.210977554321289
    },
    {
      "epoch": 1.859283447265625e-05,
      "model_forward_time": 1.710695505142212,
      "step": 19496
    },
    {
      "epoch": 1.859283447265625e-05,
      "step": 19496,
      "training_step_time": 9.309701442718506
    },
    {
      "epoch": 1.8593788146972656e-05,
      "model_forward_time": 1.7120707035064697,
      "step": 19497
    },
    {
      "epoch": 1.8593788146972656e-05,
      "step": 19497,
      "training_step_time": 8.216446876525879
    },
    {
      "epoch": 1.8594741821289064e-05,
      "model_forward_time": 1.9212977886199951,
      "step": 19498
    },
    {
      "epoch": 1.8594741821289064e-05,
      "step": 19498,
      "training_step_time": 8.781265020370483
    },
    {
      "epoch": 1.8595695495605468e-05,
      "model_forward_time": 1.7213363647460938,
      "step": 19499
    },
    {
      "epoch": 1.8595695495605468e-05,
      "step": 19499,
      "training_step_time": 8.225280046463013
    },
    {
      "epoch": 1.8596649169921875e-05,
      "grad_norm": 0.19789233803749084,
      "learning_rate": 9.436190376161276e-06,
      "loss": 0.063,
      "step": 19500
    },
    {
      "epoch": 1.8596649169921875e-05,
      "model_forward_time": 1.7180633544921875,
      "step": 19500
    },
    {
      "dynamics_loss_avg": 0.06558305881917477,
      "epoch": 1.8596649169921875e-05,
      "step": 19500
    },
    {
      "action_loss_avg": 0.0030534745426848533,
      "epoch": 1.8596649169921875e-05,
      "step": 19500
    },
    {
      "epoch": 1.8596649169921875e-05,
      "step": 19500,
      "training_step_time": 7.627429962158203
    },
    {
      "epoch": 1.8597602844238282e-05,
      "model_forward_time": 1.7318065166473389,
      "step": 19501
    },
    {
      "epoch": 1.8597602844238282e-05,
      "step": 19501,
      "training_step_time": 7.722553014755249
    },
    {
      "epoch": 1.8598556518554686e-05,
      "model_forward_time": 1.7175912857055664,
      "step": 19502
    },
    {
      "epoch": 1.8598556518554686e-05,
      "step": 19502,
      "training_step_time": 7.8963682651519775
    },
    {
      "epoch": 1.8599510192871094e-05,
      "model_forward_time": 1.7185966968536377,
      "step": 19503
    },
    {
      "epoch": 1.8599510192871094e-05,
      "step": 19503,
      "training_step_time": 7.642667293548584
    },
    {
      "epoch": 1.86004638671875e-05,
      "model_forward_time": 1.727738618850708,
      "step": 19504
    },
    {
      "epoch": 1.86004638671875e-05,
      "step": 19504,
      "training_step_time": 7.70635461807251
    },
    {
      "epoch": 1.8601417541503905e-05,
      "model_forward_time": 1.7148451805114746,
      "step": 19505
    },
    {
      "epoch": 1.8601417541503905e-05,
      "step": 19505,
      "training_step_time": 7.7222559452056885
    },
    {
      "epoch": 1.8602371215820313e-05,
      "model_forward_time": 1.722081184387207,
      "step": 19506
    },
    {
      "epoch": 1.8602371215820313e-05,
      "step": 19506,
      "training_step_time": 8.555928230285645
    },
    {
      "epoch": 1.860332489013672e-05,
      "model_forward_time": 1.418691873550415,
      "step": 19507
    },
    {
      "epoch": 1.860332489013672e-05,
      "step": 19507,
      "training_step_time": 7.682631492614746
    },
    {
      "epoch": 1.8604278564453124e-05,
      "model_forward_time": 2.2453548908233643,
      "step": 19508
    },
    {
      "epoch": 1.8604278564453124e-05,
      "step": 19508,
      "training_step_time": 7.676198720932007
    },
    {
      "epoch": 1.860523223876953e-05,
      "model_forward_time": 1.7169981002807617,
      "step": 19509
    },
    {
      "epoch": 1.860523223876953e-05,
      "step": 19509,
      "training_step_time": 7.632193088531494
    },
    {
      "epoch": 1.860618591308594e-05,
      "grad_norm": 0.15957042574882507,
      "learning_rate": 9.43542736843208e-06,
      "loss": 0.0636,
      "step": 19510
    },
    {
      "epoch": 1.860618591308594e-05,
      "model_forward_time": 1.6957964897155762,
      "step": 19510
    },
    {
      "dynamics_loss_avg": 0.039590153470635416,
      "epoch": 1.860618591308594e-05,
      "step": 19510
    },
    {
      "action_loss_avg": 0.0018880371702834965,
      "epoch": 1.860618591308594e-05,
      "step": 19510
    },
    {
      "epoch": 1.860618591308594e-05,
      "step": 19510,
      "training_step_time": 7.700849294662476
    },
    {
      "epoch": 1.8607139587402343e-05,
      "model_forward_time": 1.7039616107940674,
      "step": 19511
    },
    {
      "epoch": 1.8607139587402343e-05,
      "step": 19511,
      "training_step_time": 7.725140810012817
    },
    {
      "epoch": 1.860809326171875e-05,
      "model_forward_time": 1.707167148590088,
      "step": 19512
    },
    {
      "epoch": 1.860809326171875e-05,
      "step": 19512,
      "training_step_time": 8.60637092590332
    },
    {
      "epoch": 1.8609046936035158e-05,
      "model_forward_time": 1.3412220478057861,
      "step": 19513
    },
    {
      "epoch": 1.8609046936035158e-05,
      "step": 19513,
      "training_step_time": 8.672454595565796
    },
    {
      "epoch": 1.8610000610351562e-05,
      "model_forward_time": 1.7181081771850586,
      "step": 19514
    },
    {
      "epoch": 1.8610000610351562e-05,
      "step": 19514,
      "training_step_time": 7.745340824127197
    },
    {
      "epoch": 1.861095428466797e-05,
      "model_forward_time": 1.7092750072479248,
      "step": 19515
    },
    {
      "epoch": 1.861095428466797e-05,
      "step": 19515,
      "training_step_time": 7.789785146713257
    },
    {
      "epoch": 1.8611907958984373e-05,
      "model_forward_time": 1.7604224681854248,
      "step": 19516
    },
    {
      "epoch": 1.8611907958984373e-05,
      "step": 19516,
      "training_step_time": 9.269691705703735
    },
    {
      "epoch": 1.861286163330078e-05,
      "model_forward_time": 2.243360757827759,
      "step": 19517
    },
    {
      "epoch": 1.861286163330078e-05,
      "step": 19517,
      "training_step_time": 7.7633843421936035
    },
    {
      "epoch": 1.8613815307617188e-05,
      "model_forward_time": 1.7130491733551025,
      "step": 19518
    },
    {
      "epoch": 1.8613815307617188e-05,
      "step": 19518,
      "training_step_time": 7.62441873550415
    },
    {
      "epoch": 1.8614768981933592e-05,
      "model_forward_time": 1.710603952407837,
      "step": 19519
    },
    {
      "epoch": 1.8614768981933592e-05,
      "step": 19519,
      "training_step_time": 7.707059860229492
    },
    {
      "epoch": 1.861572265625e-05,
      "grad_norm": 0.1363631933927536,
      "learning_rate": 9.434663875651216e-06,
      "loss": 0.0613,
      "step": 19520
    },
    {
      "epoch": 1.861572265625e-05,
      "model_forward_time": 1.7138698101043701,
      "step": 19520
    },
    {
      "dynamics_loss_avg": 0.05996199361979961,
      "epoch": 1.861572265625e-05,
      "step": 19520
    },
    {
      "action_loss_avg": 0.002920358086703345,
      "epoch": 1.861572265625e-05,
      "step": 19520
    },
    {
      "epoch": 1.861572265625e-05,
      "step": 19520,
      "training_step_time": 7.773001194000244
    },
    {
      "epoch": 1.8616676330566407e-05,
      "model_forward_time": 1.7138495445251465,
      "step": 19521
    },
    {
      "epoch": 1.8616676330566407e-05,
      "step": 19521,
      "training_step_time": 7.625594854354858
    },
    {
      "epoch": 1.861763000488281e-05,
      "model_forward_time": 1.7014799118041992,
      "step": 19522
    },
    {
      "epoch": 1.861763000488281e-05,
      "step": 19522,
      "training_step_time": 7.669483184814453
    },
    {
      "epoch": 1.861858367919922e-05,
      "model_forward_time": 1.7112226486206055,
      "step": 19523
    },
    {
      "epoch": 1.861858367919922e-05,
      "step": 19523,
      "training_step_time": 7.686071157455444
    },
    {
      "epoch": 1.8619537353515626e-05,
      "model_forward_time": 1.7470927238464355,
      "step": 19524
    },
    {
      "epoch": 1.8619537353515626e-05,
      "step": 19524,
      "training_step_time": 8.275449514389038
    },
    {
      "epoch": 1.862049102783203e-05,
      "model_forward_time": 1.7096598148345947,
      "step": 19525
    },
    {
      "epoch": 1.862049102783203e-05,
      "step": 19525,
      "training_step_time": 8.16693663597107
    },
    {
      "epoch": 1.8621444702148437e-05,
      "model_forward_time": 2.234098434448242,
      "step": 19526
    },
    {
      "epoch": 1.8621444702148437e-05,
      "step": 19526,
      "training_step_time": 7.665427207946777
    },
    {
      "epoch": 1.8622398376464845e-05,
      "model_forward_time": 1.7180900573730469,
      "step": 19527
    },
    {
      "epoch": 1.8622398376464845e-05,
      "step": 19527,
      "training_step_time": 8.281423330307007
    },
    {
      "epoch": 1.862335205078125e-05,
      "model_forward_time": 1.7057273387908936,
      "step": 19528
    },
    {
      "epoch": 1.862335205078125e-05,
      "step": 19528,
      "training_step_time": 7.594143390655518
    },
    {
      "epoch": 1.8624305725097656e-05,
      "model_forward_time": 1.7042276859283447,
      "step": 19529
    },
    {
      "epoch": 1.8624305725097656e-05,
      "step": 19529,
      "training_step_time": 7.873240947723389
    },
    {
      "epoch": 1.8625259399414064e-05,
      "grad_norm": 0.1537979543209076,
      "learning_rate": 9.433899897902177e-06,
      "loss": 0.0609,
      "step": 19530
    },
    {
      "epoch": 1.8625259399414064e-05,
      "model_forward_time": 1.7042427062988281,
      "step": 19530
    },
    {
      "dynamics_loss_avg": 0.08015436194837093,
      "epoch": 1.8625259399414064e-05,
      "step": 19530
    },
    {
      "action_loss_avg": 0.00324068155605346,
      "epoch": 1.8625259399414064e-05,
      "step": 19530
    },
    {
      "epoch": 1.8625259399414064e-05,
      "step": 19530,
      "training_step_time": 8.422852754592896
    },
    {
      "epoch": 1.8626213073730468e-05,
      "model_forward_time": 1.7064166069030762,
      "step": 19531
    },
    {
      "epoch": 1.8626213073730468e-05,
      "step": 19531,
      "training_step_time": 8.179414749145508
    },
    {
      "epoch": 1.8627166748046875e-05,
      "model_forward_time": 1.7006115913391113,
      "step": 19532
    },
    {
      "epoch": 1.8627166748046875e-05,
      "step": 19532,
      "training_step_time": 9.176309585571289
    },
    {
      "epoch": 1.8628120422363282e-05,
      "model_forward_time": 1.7032787799835205,
      "step": 19533
    },
    {
      "epoch": 1.8628120422363282e-05,
      "step": 19533,
      "training_step_time": 8.329061031341553
    },
    {
      "epoch": 1.8629074096679686e-05,
      "model_forward_time": 1.700223684310913,
      "step": 19534
    },
    {
      "epoch": 1.8629074096679686e-05,
      "step": 19534,
      "training_step_time": 8.48965048789978
    },
    {
      "epoch": 1.8630027770996094e-05,
      "model_forward_time": 2.274350166320801,
      "step": 19535
    },
    {
      "epoch": 1.8630027770996094e-05,
      "step": 19535,
      "training_step_time": 8.139081954956055
    },
    {
      "epoch": 1.86309814453125e-05,
      "model_forward_time": 1.3379840850830078,
      "step": 19536
    },
    {
      "epoch": 1.86309814453125e-05,
      "step": 19536,
      "training_step_time": 7.538535118103027
    },
    {
      "epoch": 1.8631935119628905e-05,
      "model_forward_time": 1.7063984870910645,
      "step": 19537
    },
    {
      "epoch": 1.8631935119628905e-05,
      "step": 19537,
      "training_step_time": 7.547256708145142
    },
    {
      "epoch": 1.8632888793945313e-05,
      "model_forward_time": 1.7054007053375244,
      "step": 19538
    },
    {
      "epoch": 1.8632888793945313e-05,
      "step": 19538,
      "training_step_time": 7.885187864303589
    },
    {
      "epoch": 1.863384246826172e-05,
      "model_forward_time": 1.7048883438110352,
      "step": 19539
    },
    {
      "epoch": 1.863384246826172e-05,
      "step": 19539,
      "training_step_time": 7.778993844985962
    },
    {
      "epoch": 1.8634796142578124e-05,
      "grad_norm": 0.17524410784244537,
      "learning_rate": 9.43313543526851e-06,
      "loss": 0.0588,
      "step": 19540
    },
    {
      "epoch": 1.8634796142578124e-05,
      "model_forward_time": 1.6983838081359863,
      "step": 19540
    },
    {
      "dynamics_loss_avg": 0.04781040009111166,
      "epoch": 1.8634796142578124e-05,
      "step": 19540
    },
    {
      "action_loss_avg": 0.0013462934468407184,
      "epoch": 1.8634796142578124e-05,
      "step": 19540
    },
    {
      "epoch": 1.8634796142578124e-05,
      "step": 19540,
      "training_step_time": 7.614238023757935
    },
    {
      "epoch": 1.863574981689453e-05,
      "model_forward_time": 1.7038230895996094,
      "step": 19541
    },
    {
      "epoch": 1.863574981689453e-05,
      "step": 19541,
      "training_step_time": 7.760476112365723
    },
    {
      "epoch": 1.863670349121094e-05,
      "model_forward_time": 1.7081413269042969,
      "step": 19542
    },
    {
      "epoch": 1.863670349121094e-05,
      "step": 19542,
      "training_step_time": 7.790020227432251
    },
    {
      "epoch": 1.8637657165527343e-05,
      "model_forward_time": 1.7011051177978516,
      "step": 19543
    },
    {
      "epoch": 1.8637657165527343e-05,
      "step": 19543,
      "training_step_time": 7.7264697551727295
    },
    {
      "epoch": 1.863861083984375e-05,
      "model_forward_time": 1.7488594055175781,
      "step": 19544
    },
    {
      "epoch": 1.863861083984375e-05,
      "step": 19544,
      "training_step_time": 7.610801935195923
    },
    {
      "epoch": 1.8639564514160158e-05,
      "model_forward_time": 1.8709208965301514,
      "step": 19545
    },
    {
      "epoch": 1.8639564514160158e-05,
      "step": 19545,
      "training_step_time": 8.346306085586548
    },
    {
      "epoch": 1.8640518188476562e-05,
      "model_forward_time": 1.7044575214385986,
      "step": 19546
    },
    {
      "epoch": 1.8640518188476562e-05,
      "step": 19546,
      "training_step_time": 7.710346221923828
    },
    {
      "epoch": 1.864147186279297e-05,
      "model_forward_time": 1.6963698863983154,
      "step": 19547
    },
    {
      "epoch": 1.864147186279297e-05,
      "step": 19547,
      "training_step_time": 7.791689157485962
    },
    {
      "epoch": 1.8642425537109373e-05,
      "model_forward_time": 1.7056381702423096,
      "step": 19548
    },
    {
      "epoch": 1.8642425537109373e-05,
      "step": 19548,
      "training_step_time": 7.839015007019043
    },
    {
      "epoch": 1.864337921142578e-05,
      "model_forward_time": 1.701735258102417,
      "step": 19549
    },
    {
      "epoch": 1.864337921142578e-05,
      "step": 19549,
      "training_step_time": 8.499787092208862
    },
    {
      "epoch": 1.8644332885742188e-05,
      "grad_norm": 0.14504462480545044,
      "learning_rate": 9.432370487833819e-06,
      "loss": 0.0599,
      "step": 19550
    },
    {
      "epoch": 1.8644332885742188e-05,
      "model_forward_time": 1.7020542621612549,
      "step": 19550
    },
    {
      "dynamics_loss_avg": 0.07134620174765587,
      "epoch": 1.8644332885742188e-05,
      "step": 19550
    },
    {
      "action_loss_avg": 0.003425530414097011,
      "epoch": 1.8644332885742188e-05,
      "step": 19550
    },
    {
      "epoch": 1.8644332885742188e-05,
      "step": 19550,
      "training_step_time": 7.785719633102417
    },
    {
      "epoch": 1.8645286560058592e-05,
      "model_forward_time": 1.3478169441223145,
      "step": 19551
    },
    {
      "epoch": 1.8645286560058592e-05,
      "step": 19551,
      "training_step_time": 10.375258922576904
    },
    {
      "epoch": 1.8646240234375e-05,
      "model_forward_time": 1.705244541168213,
      "step": 19552
    },
    {
      "epoch": 1.8646240234375e-05,
      "step": 19552,
      "training_step_time": 8.318046808242798
    },
    {
      "epoch": 1.8647193908691407e-05,
      "model_forward_time": 1.7242848873138428,
      "step": 19553
    },
    {
      "epoch": 1.8647193908691407e-05,
      "step": 19553,
      "training_step_time": 7.582254409790039
    },
    {
      "epoch": 1.864814758300781e-05,
      "model_forward_time": 2.254566192626953,
      "step": 19554
    },
    {
      "epoch": 1.864814758300781e-05,
      "step": 19554,
      "training_step_time": 8.497201681137085
    },
    {
      "epoch": 1.864910125732422e-05,
      "model_forward_time": 1.713580846786499,
      "step": 19555
    },
    {
      "epoch": 1.864910125732422e-05,
      "step": 19555,
      "training_step_time": 7.615289211273193
    },
    {
      "epoch": 1.8650054931640626e-05,
      "model_forward_time": 1.7080271244049072,
      "step": 19556
    },
    {
      "epoch": 1.8650054931640626e-05,
      "step": 19556,
      "training_step_time": 7.8717200756073
    },
    {
      "epoch": 1.865100860595703e-05,
      "model_forward_time": 1.7119812965393066,
      "step": 19557
    },
    {
      "epoch": 1.865100860595703e-05,
      "step": 19557,
      "training_step_time": 7.813067436218262
    },
    {
      "epoch": 1.8651962280273437e-05,
      "model_forward_time": 1.7131257057189941,
      "step": 19558
    },
    {
      "epoch": 1.8651962280273437e-05,
      "step": 19558,
      "training_step_time": 7.726963758468628
    },
    {
      "epoch": 1.8652915954589845e-05,
      "model_forward_time": 1.7160980701446533,
      "step": 19559
    },
    {
      "epoch": 1.8652915954589845e-05,
      "step": 19559,
      "training_step_time": 8.80948519706726
    },
    {
      "epoch": 1.865386962890625e-05,
      "grad_norm": 0.12256109714508057,
      "learning_rate": 9.431605055681755e-06,
      "loss": 0.0583,
      "step": 19560
    },
    {
      "epoch": 1.865386962890625e-05,
      "model_forward_time": 1.7091279029846191,
      "step": 19560
    },
    {
      "dynamics_loss_avg": 0.06146584637463093,
      "epoch": 1.865386962890625e-05,
      "step": 19560
    },
    {
      "action_loss_avg": 0.002111372916260734,
      "epoch": 1.865386962890625e-05,
      "step": 19560
    },
    {
      "epoch": 1.865386962890625e-05,
      "step": 19560,
      "training_step_time": 8.471677780151367
    },
    {
      "epoch": 1.8654823303222656e-05,
      "model_forward_time": 1.714641809463501,
      "step": 19561
    },
    {
      "epoch": 1.8654823303222656e-05,
      "step": 19561,
      "training_step_time": 7.8463826179504395
    },
    {
      "epoch": 1.8655776977539064e-05,
      "model_forward_time": 1.712479829788208,
      "step": 19562
    },
    {
      "epoch": 1.8655776977539064e-05,
      "step": 19562,
      "training_step_time": 7.73529052734375
    },
    {
      "epoch": 1.8656730651855468e-05,
      "model_forward_time": 2.2709531784057617,
      "step": 19563
    },
    {
      "epoch": 1.8656730651855468e-05,
      "step": 19563,
      "training_step_time": 7.721991300582886
    },
    {
      "epoch": 1.8657684326171875e-05,
      "model_forward_time": 1.7149126529693604,
      "step": 19564
    },
    {
      "epoch": 1.8657684326171875e-05,
      "step": 19564,
      "training_step_time": 7.714605808258057
    },
    {
      "epoch": 1.8658638000488282e-05,
      "model_forward_time": 1.7103042602539062,
      "step": 19565
    },
    {
      "epoch": 1.8658638000488282e-05,
      "step": 19565,
      "training_step_time": 7.7373435497283936
    },
    {
      "epoch": 1.8659591674804686e-05,
      "model_forward_time": 1.7041785717010498,
      "step": 19566
    },
    {
      "epoch": 1.8659591674804686e-05,
      "step": 19566,
      "training_step_time": 7.840991497039795
    },
    {
      "epoch": 1.8660545349121094e-05,
      "model_forward_time": 1.7144572734832764,
      "step": 19567
    },
    {
      "epoch": 1.8660545349121094e-05,
      "step": 19567,
      "training_step_time": 7.704108715057373
    },
    {
      "epoch": 1.86614990234375e-05,
      "model_forward_time": 1.7156085968017578,
      "step": 19568
    },
    {
      "epoch": 1.86614990234375e-05,
      "step": 19568,
      "training_step_time": 8.490975856781006
    },
    {
      "epoch": 1.8662452697753905e-05,
      "model_forward_time": 1.7035667896270752,
      "step": 19569
    },
    {
      "epoch": 1.8662452697753905e-05,
      "step": 19569,
      "training_step_time": 8.50074028968811
    },
    {
      "epoch": 1.8663406372070313e-05,
      "grad_norm": 0.10736264288425446,
      "learning_rate": 9.430839138896027e-06,
      "loss": 0.0588,
      "step": 19570
    },
    {
      "epoch": 1.8663406372070313e-05,
      "model_forward_time": 1.710310459136963,
      "step": 19570
    },
    {
      "dynamics_loss_avg": 0.05734887346625328,
      "epoch": 1.8663406372070313e-05,
      "step": 19570
    },
    {
      "action_loss_avg": 0.002532705638441257,
      "epoch": 1.8663406372070313e-05,
      "step": 19570
    },
    {
      "epoch": 1.8663406372070313e-05,
      "step": 19570,
      "training_step_time": 7.699530601501465
    },
    {
      "epoch": 1.866436004638672e-05,
      "model_forward_time": 1.7117455005645752,
      "step": 19571
    },
    {
      "epoch": 1.866436004638672e-05,
      "step": 19571,
      "training_step_time": 7.8118696212768555
    },
    {
      "epoch": 1.8665313720703124e-05,
      "model_forward_time": 2.2940940856933594,
      "step": 19572
    },
    {
      "epoch": 1.8665313720703124e-05,
      "step": 19572,
      "training_step_time": 8.20782208442688
    },
    {
      "epoch": 1.866626739501953e-05,
      "model_forward_time": 1.7086338996887207,
      "step": 19573
    },
    {
      "epoch": 1.866626739501953e-05,
      "step": 19573,
      "training_step_time": 7.687883377075195
    },
    {
      "epoch": 1.866722106933594e-05,
      "model_forward_time": 1.7023203372955322,
      "step": 19574
    },
    {
      "epoch": 1.866722106933594e-05,
      "step": 19574,
      "training_step_time": 8.657262563705444
    },
    {
      "epoch": 1.8668174743652343e-05,
      "model_forward_time": 1.715036153793335,
      "step": 19575
    },
    {
      "epoch": 1.8668174743652343e-05,
      "step": 19575,
      "training_step_time": 7.958648681640625
    },
    {
      "epoch": 1.866912841796875e-05,
      "model_forward_time": 1.7193293571472168,
      "step": 19576
    },
    {
      "epoch": 1.866912841796875e-05,
      "step": 19576,
      "training_step_time": 7.636808633804321
    },
    {
      "epoch": 1.8670082092285158e-05,
      "model_forward_time": 1.704190731048584,
      "step": 19577
    },
    {
      "epoch": 1.8670082092285158e-05,
      "step": 19577,
      "training_step_time": 7.701948881149292
    },
    {
      "epoch": 1.8671035766601562e-05,
      "model_forward_time": 1.7107768058776855,
      "step": 19578
    },
    {
      "epoch": 1.8671035766601562e-05,
      "step": 19578,
      "training_step_time": 8.573781251907349
    },
    {
      "epoch": 1.867198944091797e-05,
      "model_forward_time": 1.7127673625946045,
      "step": 19579
    },
    {
      "epoch": 1.867198944091797e-05,
      "step": 19579,
      "training_step_time": 7.777423143386841
    },
    {
      "epoch": 1.8672943115234373e-05,
      "grad_norm": 0.1630089432001114,
      "learning_rate": 9.43007273756039e-06,
      "loss": 0.0551,
      "step": 19580
    },
    {
      "epoch": 1.8672943115234373e-05,
      "model_forward_time": 1.71293044090271,
      "step": 19580
    },
    {
      "dynamics_loss_avg": 0.057723676040768626,
      "epoch": 1.8672943115234373e-05,
      "step": 19580
    },
    {
      "action_loss_avg": 0.0025437041418626904,
      "epoch": 1.8672943115234373e-05,
      "step": 19580
    },
    {
      "epoch": 1.8672943115234373e-05,
      "step": 19580,
      "training_step_time": 7.808218955993652
    },
    {
      "epoch": 1.867389678955078e-05,
      "model_forward_time": 2.287937641143799,
      "step": 19581
    },
    {
      "epoch": 1.867389678955078e-05,
      "step": 19581,
      "training_step_time": 8.604968786239624
    },
    {
      "epoch": 1.8674850463867188e-05,
      "model_forward_time": 1.7055118083953857,
      "step": 19582
    },
    {
      "epoch": 1.8674850463867188e-05,
      "step": 19582,
      "training_step_time": 7.586534261703491
    },
    {
      "epoch": 1.8675804138183592e-05,
      "model_forward_time": 1.7192060947418213,
      "step": 19583
    },
    {
      "epoch": 1.8675804138183592e-05,
      "step": 19583,
      "training_step_time": 7.603435754776001
    },
    {
      "epoch": 1.86767578125e-05,
      "model_forward_time": 1.711578369140625,
      "step": 19584
    },
    {
      "epoch": 1.86767578125e-05,
      "step": 19584,
      "training_step_time": 8.649549722671509
    },
    {
      "epoch": 1.8677711486816407e-05,
      "model_forward_time": 1.7074241638183594,
      "step": 19585
    },
    {
      "epoch": 1.8677711486816407e-05,
      "step": 19585,
      "training_step_time": 7.718285083770752
    },
    {
      "epoch": 1.867866516113281e-05,
      "model_forward_time": 1.7100927829742432,
      "step": 19586
    },
    {
      "epoch": 1.867866516113281e-05,
      "step": 19586,
      "training_step_time": 7.688367128372192
    },
    {
      "epoch": 1.867961883544922e-05,
      "model_forward_time": 1.7737579345703125,
      "step": 19587
    },
    {
      "epoch": 1.867961883544922e-05,
      "step": 19587,
      "training_step_time": 7.694500923156738
    },
    {
      "epoch": 1.8680572509765626e-05,
      "model_forward_time": 1.707838535308838,
      "step": 19588
    },
    {
      "epoch": 1.8680572509765626e-05,
      "step": 19588,
      "training_step_time": 7.8152501583099365
    },
    {
      "epoch": 1.868152618408203e-05,
      "model_forward_time": 1.7097327709197998,
      "step": 19589
    },
    {
      "epoch": 1.868152618408203e-05,
      "step": 19589,
      "training_step_time": 8.605428457260132
    },
    {
      "epoch": 1.8682479858398437e-05,
      "grad_norm": 0.2552591860294342,
      "learning_rate": 9.42930585175866e-06,
      "loss": 0.0607,
      "step": 19590
    },
    {
      "epoch": 1.8682479858398437e-05,
      "model_forward_time": 1.731097936630249,
      "step": 19590
    },
    {
      "dynamics_loss_avg": 0.06588077824562788,
      "epoch": 1.8682479858398437e-05,
      "step": 19590
    },
    {
      "action_loss_avg": 0.005038754374254495,
      "epoch": 1.8682479858398437e-05,
      "step": 19590
    },
    {
      "epoch": 1.8682479858398437e-05,
      "step": 19590,
      "training_step_time": 7.594513654708862
    },
    {
      "epoch": 1.8683433532714845e-05,
      "model_forward_time": 1.5521364212036133,
      "step": 19591
    },
    {
      "epoch": 1.8683433532714845e-05,
      "step": 19591,
      "training_step_time": 7.625880002975464
    },
    {
      "epoch": 1.868438720703125e-05,
      "model_forward_time": 1.699202537536621,
      "step": 19592
    },
    {
      "epoch": 1.868438720703125e-05,
      "step": 19592,
      "training_step_time": 8.624763250350952
    },
    {
      "epoch": 1.8685340881347656e-05,
      "model_forward_time": 1.7033770084381104,
      "step": 19593
    },
    {
      "epoch": 1.8685340881347656e-05,
      "step": 19593,
      "training_step_time": 7.64454984664917
    },
    {
      "epoch": 1.8686294555664064e-05,
      "model_forward_time": 1.7152698040008545,
      "step": 19594
    },
    {
      "epoch": 1.8686294555664064e-05,
      "step": 19594,
      "training_step_time": 8.0900297164917
    },
    {
      "epoch": 1.8687248229980468e-05,
      "model_forward_time": 1.707970380783081,
      "step": 19595
    },
    {
      "epoch": 1.8687248229980468e-05,
      "step": 19595,
      "training_step_time": 8.484190225601196
    },
    {
      "epoch": 1.8688201904296875e-05,
      "model_forward_time": 1.709805965423584,
      "step": 19596
    },
    {
      "epoch": 1.8688201904296875e-05,
      "step": 19596,
      "training_step_time": 7.692374229431152
    },
    {
      "epoch": 1.8689155578613282e-05,
      "model_forward_time": 1.711484432220459,
      "step": 19597
    },
    {
      "epoch": 1.8689155578613282e-05,
      "step": 19597,
      "training_step_time": 7.81670880317688
    },
    {
      "epoch": 1.8690109252929687e-05,
      "model_forward_time": 1.7083675861358643,
      "step": 19598
    },
    {
      "epoch": 1.8690109252929687e-05,
      "step": 19598,
      "training_step_time": 7.841169595718384
    },
    {
      "epoch": 1.8691062927246094e-05,
      "model_forward_time": 1.7185885906219482,
      "step": 19599
    },
    {
      "epoch": 1.8691062927246094e-05,
      "step": 19599,
      "training_step_time": 9.033649206161499
    },
    {
      "epoch": 1.86920166015625e-05,
      "grad_norm": 0.1345006227493286,
      "learning_rate": 9.428538481574699e-06,
      "loss": 0.0635,
      "step": 19600
    },
    {
      "epoch": 1.86920166015625e-05,
      "model_forward_time": 2.2565438747406006,
      "step": 19600
    },
    {
      "dynamics_loss_avg": 0.046285292413085696,
      "epoch": 1.86920166015625e-05,
      "step": 19600
    },
    {
      "action_loss_avg": 0.0028664644109085203,
      "epoch": 1.86920166015625e-05,
      "step": 19600
    },
    {
      "epoch": 1.86920166015625e-05,
      "step": 19600,
      "training_step_time": 7.622568130493164
    },
    {
      "epoch": 1.8692970275878905e-05,
      "model_forward_time": 1.7124478816986084,
      "step": 19601
    },
    {
      "epoch": 1.8692970275878905e-05,
      "step": 19601,
      "training_step_time": 7.6866466999053955
    },
    {
      "epoch": 1.8693923950195313e-05,
      "model_forward_time": 1.725801706314087,
      "step": 19602
    },
    {
      "epoch": 1.8693923950195313e-05,
      "step": 19602,
      "training_step_time": 7.669586896896362
    },
    {
      "epoch": 1.869487762451172e-05,
      "model_forward_time": 1.7080974578857422,
      "step": 19603
    },
    {
      "epoch": 1.869487762451172e-05,
      "step": 19603,
      "training_step_time": 8.12569522857666
    },
    {
      "epoch": 1.8695831298828124e-05,
      "model_forward_time": 1.7311694622039795,
      "step": 19604
    },
    {
      "epoch": 1.8695831298828124e-05,
      "step": 19604,
      "training_step_time": 9.100008964538574
    },
    {
      "epoch": 1.869678497314453e-05,
      "model_forward_time": 1.7349607944488525,
      "step": 19605
    },
    {
      "epoch": 1.869678497314453e-05,
      "step": 19605,
      "training_step_time": 8.247450113296509
    },
    {
      "epoch": 1.869773864746094e-05,
      "model_forward_time": 1.7131335735321045,
      "step": 19606
    },
    {
      "epoch": 1.869773864746094e-05,
      "step": 19606,
      "training_step_time": 7.789104461669922
    },
    {
      "epoch": 1.8698692321777343e-05,
      "model_forward_time": 1.7161016464233398,
      "step": 19607
    },
    {
      "epoch": 1.8698692321777343e-05,
      "step": 19607,
      "training_step_time": 7.74803900718689
    },
    {
      "epoch": 1.869964599609375e-05,
      "model_forward_time": 1.7068629264831543,
      "step": 19608
    },
    {
      "epoch": 1.869964599609375e-05,
      "step": 19608,
      "training_step_time": 7.639141798019409
    },
    {
      "epoch": 1.8700599670410158e-05,
      "model_forward_time": 2.2895004749298096,
      "step": 19609
    },
    {
      "epoch": 1.8700599670410158e-05,
      "step": 19609,
      "training_step_time": 7.61626935005188
    },
    {
      "epoch": 1.8701553344726562e-05,
      "grad_norm": 0.1475033164024353,
      "learning_rate": 9.42777062709243e-06,
      "loss": 0.0622,
      "step": 19610
    },
    {
      "epoch": 1.8701553344726562e-05,
      "model_forward_time": 1.71394944190979,
      "step": 19610
    },
    {
      "dynamics_loss_avg": 0.06303028501570225,
      "epoch": 1.8701553344726562e-05,
      "step": 19610
    },
    {
      "action_loss_avg": 0.002371195034356788,
      "epoch": 1.8701553344726562e-05,
      "step": 19610
    },
    {
      "epoch": 1.8701553344726562e-05,
      "step": 19610,
      "training_step_time": 7.7140655517578125
    },
    {
      "epoch": 1.870250701904297e-05,
      "model_forward_time": 1.705481767654419,
      "step": 19611
    },
    {
      "epoch": 1.870250701904297e-05,
      "step": 19611,
      "training_step_time": 7.696971416473389
    },
    {
      "epoch": 1.8703460693359373e-05,
      "model_forward_time": 1.714245319366455,
      "step": 19612
    },
    {
      "epoch": 1.8703460693359373e-05,
      "step": 19612,
      "training_step_time": 7.7408366203308105
    },
    {
      "epoch": 1.870441436767578e-05,
      "model_forward_time": 1.7117927074432373,
      "step": 19613
    },
    {
      "epoch": 1.870441436767578e-05,
      "step": 19613,
      "training_step_time": 8.73895788192749
    },
    {
      "epoch": 1.8705368041992188e-05,
      "model_forward_time": 1.71645188331604,
      "step": 19614
    },
    {
      "epoch": 1.8705368041992188e-05,
      "step": 19614,
      "training_step_time": 7.678270578384399
    },
    {
      "epoch": 1.8706321716308592e-05,
      "model_forward_time": 1.706045389175415,
      "step": 19615
    },
    {
      "epoch": 1.8706321716308592e-05,
      "step": 19615,
      "training_step_time": 7.796334981918335
    },
    {
      "epoch": 1.8707275390625e-05,
      "model_forward_time": 1.7075657844543457,
      "step": 19616
    },
    {
      "epoch": 1.8707275390625e-05,
      "step": 19616,
      "training_step_time": 9.722523927688599
    },
    {
      "epoch": 1.8708229064941407e-05,
      "model_forward_time": 1.7096097469329834,
      "step": 19617
    },
    {
      "epoch": 1.8708229064941407e-05,
      "step": 19617,
      "training_step_time": 9.434768676757812
    },
    {
      "epoch": 1.870918273925781e-05,
      "model_forward_time": 1.393587350845337,
      "step": 19618
    },
    {
      "epoch": 1.870918273925781e-05,
      "step": 19618,
      "training_step_time": 7.65815281867981
    },
    {
      "epoch": 1.871013641357422e-05,
      "model_forward_time": 2.2646219730377197,
      "step": 19619
    },
    {
      "epoch": 1.871013641357422e-05,
      "step": 19619,
      "training_step_time": 7.674414873123169
    },
    {
      "epoch": 1.8711090087890626e-05,
      "grad_norm": 0.17786404490470886,
      "learning_rate": 9.427002288395823e-06,
      "loss": 0.0614,
      "step": 19620
    },
    {
      "epoch": 1.8711090087890626e-05,
      "model_forward_time": 1.7134757041931152,
      "step": 19620
    },
    {
      "dynamics_loss_avg": 0.05476964898407459,
      "epoch": 1.8711090087890626e-05,
      "step": 19620
    },
    {
      "action_loss_avg": 0.0021286891016643493,
      "epoch": 1.8711090087890626e-05,
      "step": 19620
    },
    {
      "epoch": 1.8711090087890626e-05,
      "step": 19620,
      "training_step_time": 7.760204553604126
    },
    {
      "epoch": 1.871204376220703e-05,
      "model_forward_time": 1.7161805629730225,
      "step": 19621
    },
    {
      "epoch": 1.871204376220703e-05,
      "step": 19621,
      "training_step_time": 7.696040153503418
    },
    {
      "epoch": 1.8712997436523437e-05,
      "model_forward_time": 1.7082901000976562,
      "step": 19622
    },
    {
      "epoch": 1.8712997436523437e-05,
      "step": 19622,
      "training_step_time": 7.703678607940674
    },
    {
      "epoch": 1.8713951110839845e-05,
      "model_forward_time": 1.7166800498962402,
      "step": 19623
    },
    {
      "epoch": 1.8713951110839845e-05,
      "step": 19623,
      "training_step_time": 7.664073944091797
    },
    {
      "epoch": 1.871490478515625e-05,
      "model_forward_time": 1.713050127029419,
      "step": 19624
    },
    {
      "epoch": 1.871490478515625e-05,
      "step": 19624,
      "training_step_time": 7.788771152496338
    },
    {
      "epoch": 1.8715858459472656e-05,
      "model_forward_time": 1.709632396697998,
      "step": 19625
    },
    {
      "epoch": 1.8715858459472656e-05,
      "step": 19625,
      "training_step_time": 7.703884124755859
    },
    {
      "epoch": 1.8716812133789064e-05,
      "model_forward_time": 1.725365161895752,
      "step": 19626
    },
    {
      "epoch": 1.8716812133789064e-05,
      "step": 19626,
      "training_step_time": 7.663494348526001
    },
    {
      "epoch": 1.8717765808105468e-05,
      "model_forward_time": 1.3669848442077637,
      "step": 19627
    },
    {
      "epoch": 1.8717765808105468e-05,
      "step": 19627,
      "training_step_time": 8.504953622817993
    },
    {
      "epoch": 1.8718719482421875e-05,
      "model_forward_time": 2.2447564601898193,
      "step": 19628
    },
    {
      "epoch": 1.8718719482421875e-05,
      "step": 19628,
      "training_step_time": 8.559958219528198
    },
    {
      "epoch": 1.8719673156738283e-05,
      "model_forward_time": 1.7012887001037598,
      "step": 19629
    },
    {
      "epoch": 1.8719673156738283e-05,
      "step": 19629,
      "training_step_time": 7.6512908935546875
    },
    {
      "epoch": 1.8720626831054687e-05,
      "grad_norm": 0.2886386811733246,
      "learning_rate": 9.4262334655689e-06,
      "loss": 0.0658,
      "step": 19630
    },
    {
      "epoch": 1.8720626831054687e-05,
      "model_forward_time": 1.7090072631835938,
      "step": 19630
    },
    {
      "dynamics_loss_avg": 0.04693173887208104,
      "epoch": 1.8720626831054687e-05,
      "step": 19630
    },
    {
      "action_loss_avg": 0.0027883029077202083,
      "epoch": 1.8720626831054687e-05,
      "step": 19630
    },
    {
      "epoch": 1.8720626831054687e-05,
      "step": 19630,
      "training_step_time": 7.743552207946777
    },
    {
      "epoch": 1.8721580505371094e-05,
      "model_forward_time": 1.025221586227417,
      "step": 19631
    },
    {
      "epoch": 1.8721580505371094e-05,
      "step": 19631,
      "training_step_time": 8.743236780166626
    },
    {
      "epoch": 1.87225341796875e-05,
      "model_forward_time": 1.7169387340545654,
      "step": 19632
    },
    {
      "epoch": 1.87225341796875e-05,
      "step": 19632,
      "training_step_time": 7.868822813034058
    },
    {
      "epoch": 1.8723487854003905e-05,
      "model_forward_time": 1.7264480590820312,
      "step": 19633
    },
    {
      "epoch": 1.8723487854003905e-05,
      "step": 19633,
      "training_step_time": 9.794676303863525
    },
    {
      "epoch": 1.8724441528320313e-05,
      "model_forward_time": 1.7221083641052246,
      "step": 19634
    },
    {
      "epoch": 1.8724441528320313e-05,
      "step": 19634,
      "training_step_time": 9.280731916427612
    },
    {
      "epoch": 1.872539520263672e-05,
      "model_forward_time": 1.7346718311309814,
      "step": 19635
    },
    {
      "epoch": 1.872539520263672e-05,
      "step": 19635,
      "training_step_time": 7.738375425338745
    },
    {
      "epoch": 1.8726348876953124e-05,
      "model_forward_time": 1.7373771667480469,
      "step": 19636
    },
    {
      "epoch": 1.8726348876953124e-05,
      "step": 19636,
      "training_step_time": 7.717389345169067
    },
    {
      "epoch": 1.872730255126953e-05,
      "model_forward_time": 2.241079807281494,
      "step": 19637
    },
    {
      "epoch": 1.872730255126953e-05,
      "step": 19637,
      "training_step_time": 9.084050416946411
    },
    {
      "epoch": 1.872825622558594e-05,
      "model_forward_time": 1.7272067070007324,
      "step": 19638
    },
    {
      "epoch": 1.872825622558594e-05,
      "step": 19638,
      "training_step_time": 7.6867005825042725
    },
    {
      "epoch": 1.8729209899902343e-05,
      "model_forward_time": 1.7254352569580078,
      "step": 19639
    },
    {
      "epoch": 1.8729209899902343e-05,
      "step": 19639,
      "training_step_time": 7.606922149658203
    },
    {
      "epoch": 1.873016357421875e-05,
      "grad_norm": 0.30092543363571167,
      "learning_rate": 9.42546415869574e-06,
      "loss": 0.0665,
      "step": 19640
    },
    {
      "epoch": 1.873016357421875e-05,
      "model_forward_time": 1.7176692485809326,
      "step": 19640
    },
    {
      "dynamics_loss_avg": 0.05788344256579876,
      "epoch": 1.873016357421875e-05,
      "step": 19640
    },
    {
      "action_loss_avg": 0.002318754931911826,
      "epoch": 1.873016357421875e-05,
      "step": 19640
    },
    {
      "epoch": 1.873016357421875e-05,
      "step": 19640,
      "training_step_time": 7.6784281730651855
    },
    {
      "epoch": 1.8731117248535155e-05,
      "model_forward_time": 1.7258026599884033,
      "step": 19641
    },
    {
      "epoch": 1.8731117248535155e-05,
      "step": 19641,
      "training_step_time": 7.667009115219116
    },
    {
      "epoch": 1.8732070922851562e-05,
      "model_forward_time": 1.7190120220184326,
      "step": 19642
    },
    {
      "epoch": 1.8732070922851562e-05,
      "step": 19642,
      "training_step_time": 8.262158155441284
    },
    {
      "epoch": 1.873302459716797e-05,
      "model_forward_time": 1.713876485824585,
      "step": 19643
    },
    {
      "epoch": 1.873302459716797e-05,
      "step": 19643,
      "training_step_time": 7.739122152328491
    },
    {
      "epoch": 1.8733978271484373e-05,
      "model_forward_time": 1.7184100151062012,
      "step": 19644
    },
    {
      "epoch": 1.8733978271484373e-05,
      "step": 19644,
      "training_step_time": 7.772216796875
    },
    {
      "epoch": 1.873493194580078e-05,
      "model_forward_time": 1.7107892036437988,
      "step": 19645
    },
    {
      "epoch": 1.873493194580078e-05,
      "step": 19645,
      "training_step_time": 8.327749967575073
    },
    {
      "epoch": 1.8735885620117188e-05,
      "model_forward_time": 1.728797435760498,
      "step": 19646
    },
    {
      "epoch": 1.8735885620117188e-05,
      "step": 19646,
      "training_step_time": 7.706783056259155
    },
    {
      "epoch": 1.8736839294433592e-05,
      "model_forward_time": 2.261293888092041,
      "step": 19647
    },
    {
      "epoch": 1.8736839294433592e-05,
      "step": 19647,
      "training_step_time": 7.683981657028198
    },
    {
      "epoch": 1.873779296875e-05,
      "model_forward_time": 1.719801664352417,
      "step": 19648
    },
    {
      "epoch": 1.873779296875e-05,
      "step": 19648,
      "training_step_time": 7.63972020149231
    },
    {
      "epoch": 1.8738746643066407e-05,
      "model_forward_time": 1.730262041091919,
      "step": 19649
    },
    {
      "epoch": 1.8738746643066407e-05,
      "step": 19649,
      "training_step_time": 7.739141464233398
    },
    {
      "epoch": 1.873970031738281e-05,
      "grad_norm": 0.21412383019924164,
      "learning_rate": 9.424694367860475e-06,
      "loss": 0.0693,
      "step": 19650
    },
    {
      "epoch": 1.873970031738281e-05,
      "model_forward_time": 1.7262911796569824,
      "step": 19650
    },
    {
      "dynamics_loss_avg": 0.0617419496178627,
      "epoch": 1.873970031738281e-05,
      "step": 19650
    },
    {
      "action_loss_avg": 0.0021388366410974415,
      "epoch": 1.873970031738281e-05,
      "step": 19650
    },
    {
      "epoch": 1.873970031738281e-05,
      "step": 19650,
      "training_step_time": 9.262664794921875
    },
    {
      "epoch": 1.874065399169922e-05,
      "model_forward_time": 1.7306203842163086,
      "step": 19651
    },
    {
      "epoch": 1.874065399169922e-05,
      "step": 19651,
      "training_step_time": 8.358901023864746
    },
    {
      "epoch": 1.8741607666015626e-05,
      "model_forward_time": 1.7233269214630127,
      "step": 19652
    },
    {
      "epoch": 1.8741607666015626e-05,
      "step": 19652,
      "training_step_time": 8.543614864349365
    },
    {
      "epoch": 1.874256134033203e-05,
      "model_forward_time": 1.7333180904388428,
      "step": 19653
    },
    {
      "epoch": 1.874256134033203e-05,
      "step": 19653,
      "training_step_time": 7.6697564125061035
    },
    {
      "epoch": 1.8743515014648437e-05,
      "model_forward_time": 1.7295808792114258,
      "step": 19654
    },
    {
      "epoch": 1.8743515014648437e-05,
      "step": 19654,
      "training_step_time": 8.269477367401123
    },
    {
      "epoch": 1.8744468688964845e-05,
      "model_forward_time": 1.7790992259979248,
      "step": 19655
    },
    {
      "epoch": 1.8744468688964845e-05,
      "step": 19655,
      "training_step_time": 7.710139274597168
    },
    {
      "epoch": 1.874542236328125e-05,
      "model_forward_time": 2.258517265319824,
      "step": 19656
    },
    {
      "epoch": 1.874542236328125e-05,
      "step": 19656,
      "training_step_time": 7.733771324157715
    },
    {
      "epoch": 1.8746376037597656e-05,
      "model_forward_time": 1.7320239543914795,
      "step": 19657
    },
    {
      "epoch": 1.8746376037597656e-05,
      "step": 19657,
      "training_step_time": 7.826270341873169
    },
    {
      "epoch": 1.8747329711914064e-05,
      "model_forward_time": 1.7221770286560059,
      "step": 19658
    },
    {
      "epoch": 1.8747329711914064e-05,
      "step": 19658,
      "training_step_time": 8.607596635818481
    },
    {
      "epoch": 1.8748283386230468e-05,
      "model_forward_time": 1.7244911193847656,
      "step": 19659
    },
    {
      "epoch": 1.8748283386230468e-05,
      "step": 19659,
      "training_step_time": 8.226594924926758
    },
    {
      "epoch": 1.8749237060546875e-05,
      "grad_norm": 0.460385262966156,
      "learning_rate": 9.423924093147284e-06,
      "loss": 0.0716,
      "step": 19660
    },
    {
      "epoch": 1.8749237060546875e-05,
      "model_forward_time": 1.7245540618896484,
      "step": 19660
    },
    {
      "dynamics_loss_avg": 0.05452894121408462,
      "epoch": 1.8749237060546875e-05,
      "step": 19660
    },
    {
      "action_loss_avg": 0.002792343811597675,
      "epoch": 1.8749237060546875e-05,
      "step": 19660
    },
    {
      "epoch": 1.8749237060546875e-05,
      "step": 19660,
      "training_step_time": 7.756972789764404
    },
    {
      "epoch": 1.8750190734863283e-05,
      "model_forward_time": 1.7203164100646973,
      "step": 19661
    },
    {
      "epoch": 1.8750190734863283e-05,
      "step": 19661,
      "training_step_time": 8.77979588508606
    },
    {
      "epoch": 1.8751144409179687e-05,
      "model_forward_time": 1.7305200099945068,
      "step": 19662
    },
    {
      "epoch": 1.8751144409179687e-05,
      "step": 19662,
      "training_step_time": 7.598683595657349
    },
    {
      "epoch": 1.8752098083496094e-05,
      "model_forward_time": 1.7231616973876953,
      "step": 19663
    },
    {
      "epoch": 1.8752098083496094e-05,
      "step": 19663,
      "training_step_time": 8.467876434326172
    },
    {
      "epoch": 1.87530517578125e-05,
      "model_forward_time": 1.7809624671936035,
      "step": 19664
    },
    {
      "epoch": 1.87530517578125e-05,
      "step": 19664,
      "training_step_time": 7.6799516677856445
    },
    {
      "epoch": 1.8754005432128905e-05,
      "model_forward_time": 2.269158124923706,
      "step": 19665
    },
    {
      "epoch": 1.8754005432128905e-05,
      "step": 19665,
      "training_step_time": 8.531775712966919
    },
    {
      "epoch": 1.8754959106445313e-05,
      "model_forward_time": 1.722423791885376,
      "step": 19666
    },
    {
      "epoch": 1.8754959106445313e-05,
      "step": 19666,
      "training_step_time": 7.704526662826538
    },
    {
      "epoch": 1.875591278076172e-05,
      "model_forward_time": 1.7188024520874023,
      "step": 19667
    },
    {
      "epoch": 1.875591278076172e-05,
      "step": 19667,
      "training_step_time": 8.305143356323242
    },
    {
      "epoch": 1.8756866455078124e-05,
      "model_forward_time": 1.718078851699829,
      "step": 19668
    },
    {
      "epoch": 1.8756866455078124e-05,
      "step": 19668,
      "training_step_time": 9.255871295928955
    },
    {
      "epoch": 1.8757820129394532e-05,
      "model_forward_time": 1.7343566417694092,
      "step": 19669
    },
    {
      "epoch": 1.8757820129394532e-05,
      "step": 19669,
      "training_step_time": 6.787912607192993
    },
    {
      "epoch": 1.875877380371094e-05,
      "grad_norm": 0.1693067103624344,
      "learning_rate": 9.423153334640406e-06,
      "loss": 0.0629,
      "step": 19670
    },
    {
      "epoch": 1.875877380371094e-05,
      "model_forward_time": 1.7260253429412842,
      "step": 19670
    },
    {
      "dynamics_loss_avg": 0.0483149740844965,
      "epoch": 1.875877380371094e-05,
      "step": 19670
    },
    {
      "action_loss_avg": 0.002776238223304972,
      "epoch": 1.875877380371094e-05,
      "step": 19670
    },
    {
      "epoch": 1.875877380371094e-05,
      "step": 19670,
      "training_step_time": 6.768282175064087
    },
    {
      "epoch": 1.8759727478027343e-05,
      "model_forward_time": 1.7293555736541748,
      "step": 19671
    },
    {
      "epoch": 1.8759727478027343e-05,
      "step": 19671,
      "training_step_time": 6.76477575302124
    },
    {
      "epoch": 1.876068115234375e-05,
      "model_forward_time": 1.7152574062347412,
      "step": 19672
    },
    {
      "epoch": 1.876068115234375e-05,
      "step": 19672,
      "training_step_time": 6.783048868179321
    },
    {
      "epoch": 1.8761634826660155e-05,
      "model_forward_time": 1.7244164943695068,
      "step": 19673
    },
    {
      "epoch": 1.8761634826660155e-05,
      "step": 19673,
      "training_step_time": 6.7567079067230225
    },
    {
      "epoch": 1.8762588500976562e-05,
      "model_forward_time": 1.7310738563537598,
      "step": 19674
    },
    {
      "epoch": 1.8762588500976562e-05,
      "step": 19674,
      "training_step_time": 6.781305313110352
    },
    {
      "epoch": 1.876354217529297e-05,
      "model_forward_time": 1.7316246032714844,
      "step": 19675
    },
    {
      "epoch": 1.876354217529297e-05,
      "step": 19675,
      "training_step_time": 6.810176372528076
    },
    {
      "epoch": 1.8764495849609373e-05,
      "model_forward_time": 1.720308542251587,
      "step": 19676
    },
    {
      "epoch": 1.8764495849609373e-05,
      "step": 19676,
      "training_step_time": 6.719331502914429
    },
    {
      "epoch": 1.876544952392578e-05,
      "model_forward_time": 2.3050098419189453,
      "step": 19677
    },
    {
      "epoch": 1.876544952392578e-05,
      "step": 19677,
      "training_step_time": 7.526139974594116
    },
    {
      "epoch": 1.8766403198242188e-05,
      "model_forward_time": 1.7432053089141846,
      "step": 19678
    },
    {
      "epoch": 1.8766403198242188e-05,
      "step": 19678,
      "training_step_time": 7.763960361480713
    },
    {
      "epoch": 1.8767356872558592e-05,
      "model_forward_time": 1.7457420825958252,
      "step": 19679
    },
    {
      "epoch": 1.8767356872558592e-05,
      "step": 19679,
      "training_step_time": 6.7050628662109375
    },
    {
      "epoch": 1.8768310546875e-05,
      "grad_norm": 0.21524251997470856,
      "learning_rate": 9.422382092424131e-06,
      "loss": 0.0626,
      "step": 19680
    },
    {
      "epoch": 1.8768310546875e-05,
      "model_forward_time": 1.7274982929229736,
      "step": 19680
    },
    {
      "dynamics_loss_avg": 0.06025832574814558,
      "epoch": 1.8768310546875e-05,
      "step": 19680
    },
    {
      "action_loss_avg": 0.0031149580579949545,
      "epoch": 1.8768310546875e-05,
      "step": 19680
    },
    {
      "epoch": 1.8768310546875e-05,
      "step": 19680,
      "training_step_time": 6.728510856628418
    },
    {
      "epoch": 1.8769264221191407e-05,
      "model_forward_time": 1.7410714626312256,
      "step": 19681
    },
    {
      "epoch": 1.8769264221191407e-05,
      "step": 19681,
      "training_step_time": 6.717327356338501
    },
    {
      "epoch": 1.877021789550781e-05,
      "model_forward_time": 1.735957384109497,
      "step": 19682
    },
    {
      "epoch": 1.877021789550781e-05,
      "step": 19682,
      "training_step_time": 6.742460489273071
    },
    {
      "epoch": 1.877117156982422e-05,
      "model_forward_time": 1.7351396083831787,
      "step": 19683
    },
    {
      "epoch": 1.877117156982422e-05,
      "step": 19683,
      "training_step_time": 6.688199758529663
    },
    {
      "epoch": 1.8772125244140626e-05,
      "model_forward_time": 1.7309141159057617,
      "step": 19684
    },
    {
      "epoch": 1.8772125244140626e-05,
      "step": 19684,
      "training_step_time": 6.652941703796387
    },
    {
      "epoch": 1.877307891845703e-05,
      "model_forward_time": 1.7337448596954346,
      "step": 19685
    },
    {
      "epoch": 1.877307891845703e-05,
      "step": 19685,
      "training_step_time": 6.66235613822937
    },
    {
      "epoch": 1.8774032592773437e-05,
      "model_forward_time": 2.296083927154541,
      "step": 19686
    },
    {
      "epoch": 1.8774032592773437e-05,
      "step": 19686,
      "training_step_time": 7.4778969287872314
    },
    {
      "epoch": 1.8774986267089845e-05,
      "model_forward_time": 1.7308955192565918,
      "step": 19687
    },
    {
      "epoch": 1.8774986267089845e-05,
      "step": 19687,
      "training_step_time": 7.7327492237091064
    },
    {
      "epoch": 1.877593994140625e-05,
      "model_forward_time": 1.7331907749176025,
      "step": 19688
    },
    {
      "epoch": 1.877593994140625e-05,
      "step": 19688,
      "training_step_time": 6.668291330337524
    },
    {
      "epoch": 1.8776893615722656e-05,
      "model_forward_time": 1.746396780014038,
      "step": 19689
    },
    {
      "epoch": 1.8776893615722656e-05,
      "step": 19689,
      "training_step_time": 6.6891188621521
    },
    {
      "epoch": 1.8777847290039064e-05,
      "grad_norm": 0.23324552178382874,
      "learning_rate": 9.421610366582799e-06,
      "loss": 0.0615,
      "step": 19690
    },
    {
      "epoch": 1.8777847290039064e-05,
      "model_forward_time": 1.7361912727355957,
      "step": 19690
    },
    {
      "dynamics_loss_avg": 0.07265537120401859,
      "epoch": 1.8777847290039064e-05,
      "step": 19690
    },
    {
      "action_loss_avg": 0.003940613439772278,
      "epoch": 1.8777847290039064e-05,
      "step": 19690
    },
    {
      "epoch": 1.8777847290039064e-05,
      "step": 19690,
      "training_step_time": 6.695830821990967
    },
    {
      "epoch": 1.8778800964355468e-05,
      "model_forward_time": 1.7293815612792969,
      "step": 19691
    },
    {
      "epoch": 1.8778800964355468e-05,
      "step": 19691,
      "training_step_time": 6.6679017543792725
    },
    {
      "epoch": 1.8779754638671875e-05,
      "model_forward_time": 1.732501745223999,
      "step": 19692
    },
    {
      "epoch": 1.8779754638671875e-05,
      "step": 19692,
      "training_step_time": 6.698592185974121
    },
    {
      "epoch": 1.8780708312988283e-05,
      "model_forward_time": 1.7344825267791748,
      "step": 19693
    },
    {
      "epoch": 1.8780708312988283e-05,
      "step": 19693,
      "training_step_time": 6.689738035202026
    },
    {
      "epoch": 1.8781661987304687e-05,
      "model_forward_time": 1.7523515224456787,
      "step": 19694
    },
    {
      "epoch": 1.8781661987304687e-05,
      "step": 19694,
      "training_step_time": 6.740253210067749
    },
    {
      "epoch": 1.8782615661621094e-05,
      "model_forward_time": 2.343353748321533,
      "step": 19695
    },
    {
      "epoch": 1.8782615661621094e-05,
      "step": 19695,
      "training_step_time": 7.553161859512329
    },
    {
      "epoch": 1.87835693359375e-05,
      "model_forward_time": 1.7420682907104492,
      "step": 19696
    },
    {
      "epoch": 1.87835693359375e-05,
      "step": 19696,
      "training_step_time": 8.702891826629639
    },
    {
      "epoch": 1.8784523010253905e-05,
      "model_forward_time": 1.7254745960235596,
      "step": 19697
    },
    {
      "epoch": 1.8784523010253905e-05,
      "step": 19697,
      "training_step_time": 7.6103363037109375
    },
    {
      "epoch": 1.8785476684570313e-05,
      "model_forward_time": 1.7339000701904297,
      "step": 19698
    },
    {
      "epoch": 1.8785476684570313e-05,
      "step": 19698,
      "training_step_time": 6.752902507781982
    },
    {
      "epoch": 1.878643035888672e-05,
      "model_forward_time": 1.7398579120635986,
      "step": 19699
    },
    {
      "epoch": 1.878643035888672e-05,
      "step": 19699,
      "training_step_time": 6.71262001991272
    },
    {
      "epoch": 1.8787384033203124e-05,
      "grad_norm": 0.17606133222579956,
      "learning_rate": 9.420838157200803e-06,
      "loss": 0.0672,
      "step": 19700
    },
    {
      "epoch": 1.8787384033203124e-05,
      "model_forward_time": 1.7163958549499512,
      "step": 19700
    },
    {
      "dynamics_loss_avg": 0.05155448541045189,
      "epoch": 1.8787384033203124e-05,
      "step": 19700
    },
    {
      "action_loss_avg": 0.0015803524758666754,
      "epoch": 1.8787384033203124e-05,
      "step": 19700
    },
    {
      "epoch": 1.8787384033203124e-05,
      "step": 19700,
      "training_step_time": 6.694960594177246
    },
    {
      "epoch": 1.8788337707519532e-05,
      "model_forward_time": 1.7306876182556152,
      "step": 19701
    },
    {
      "epoch": 1.8788337707519532e-05,
      "step": 19701,
      "training_step_time": 6.698570966720581
    },
    {
      "epoch": 1.878929138183594e-05,
      "model_forward_time": 1.7423779964447021,
      "step": 19702
    },
    {
      "epoch": 1.878929138183594e-05,
      "step": 19702,
      "training_step_time": 6.719062805175781
    },
    {
      "epoch": 1.8790245056152343e-05,
      "model_forward_time": 1.7337150573730469,
      "step": 19703
    },
    {
      "epoch": 1.8790245056152343e-05,
      "step": 19703,
      "training_step_time": 6.72314977645874
    },
    {
      "epoch": 1.879119873046875e-05,
      "model_forward_time": 2.317490577697754,
      "step": 19704
    },
    {
      "epoch": 1.879119873046875e-05,
      "step": 19704,
      "training_step_time": 7.219921827316284
    },
    {
      "epoch": 1.8792152404785155e-05,
      "model_forward_time": 1.731813669204712,
      "step": 19705
    },
    {
      "epoch": 1.8792152404785155e-05,
      "step": 19705,
      "training_step_time": 9.19108510017395
    },
    {
      "epoch": 1.8793106079101562e-05,
      "model_forward_time": 1.7315294742584229,
      "step": 19706
    },
    {
      "epoch": 1.8793106079101562e-05,
      "step": 19706,
      "training_step_time": 8.315683603286743
    },
    {
      "epoch": 1.879405975341797e-05,
      "model_forward_time": 1.73734450340271,
      "step": 19707
    },
    {
      "epoch": 1.879405975341797e-05,
      "step": 19707,
      "training_step_time": 6.699991226196289
    },
    {
      "epoch": 1.8795013427734373e-05,
      "model_forward_time": 1.7451274394989014,
      "step": 19708
    },
    {
      "epoch": 1.8795013427734373e-05,
      "step": 19708,
      "training_step_time": 6.714309215545654
    },
    {
      "epoch": 1.879596710205078e-05,
      "model_forward_time": 1.7047593593597412,
      "step": 19709
    },
    {
      "epoch": 1.879596710205078e-05,
      "step": 19709,
      "training_step_time": 6.741862058639526
    },
    {
      "epoch": 1.879692077636719e-05,
      "grad_norm": 0.22795580327510834,
      "learning_rate": 9.420065464362595e-06,
      "loss": 0.0674,
      "step": 19710
    },
    {
      "epoch": 1.879692077636719e-05,
      "model_forward_time": 1.7363481521606445,
      "step": 19710
    },
    {
      "dynamics_loss_avg": 0.08230598121881486,
      "epoch": 1.879692077636719e-05,
      "step": 19710
    },
    {
      "action_loss_avg": 0.0037414388731122016,
      "epoch": 1.879692077636719e-05,
      "step": 19710
    },
    {
      "epoch": 1.879692077636719e-05,
      "step": 19710,
      "training_step_time": 6.697275876998901
    },
    {
      "epoch": 1.8797874450683592e-05,
      "model_forward_time": 1.7072093486785889,
      "step": 19711
    },
    {
      "epoch": 1.8797874450683592e-05,
      "step": 19711,
      "training_step_time": 6.665051221847534
    },
    {
      "epoch": 1.8798828125e-05,
      "model_forward_time": 1.729074239730835,
      "step": 19712
    },
    {
      "epoch": 1.8798828125e-05,
      "step": 19712,
      "training_step_time": 6.700728178024292
    },
    {
      "epoch": 1.8799781799316407e-05,
      "model_forward_time": 1.8798980712890625,
      "step": 19713
    },
    {
      "epoch": 1.8799781799316407e-05,
      "step": 19713,
      "training_step_time": 7.019291877746582
    },
    {
      "epoch": 1.880073547363281e-05,
      "model_forward_time": 1.7905762195587158,
      "step": 19714
    },
    {
      "epoch": 1.880073547363281e-05,
      "step": 19714,
      "training_step_time": 10.183513402938843
    },
    {
      "epoch": 1.880168914794922e-05,
      "model_forward_time": 1.7288150787353516,
      "step": 19715
    },
    {
      "epoch": 1.880168914794922e-05,
      "step": 19715,
      "training_step_time": 7.58119797706604
    },
    {
      "epoch": 1.8802642822265626e-05,
      "model_forward_time": 1.7434165477752686,
      "step": 19716
    },
    {
      "epoch": 1.8802642822265626e-05,
      "step": 19716,
      "training_step_time": 7.190717935562134
    },
    {
      "epoch": 1.880359649658203e-05,
      "model_forward_time": 1.7370035648345947,
      "step": 19717
    },
    {
      "epoch": 1.880359649658203e-05,
      "step": 19717,
      "training_step_time": 6.724168539047241
    },
    {
      "epoch": 1.8804550170898437e-05,
      "model_forward_time": 1.3727407455444336,
      "step": 19718
    },
    {
      "epoch": 1.8804550170898437e-05,
      "step": 19718,
      "training_step_time": 6.740564346313477
    },
    {
      "epoch": 1.8805503845214845e-05,
      "model_forward_time": 1.7343530654907227,
      "step": 19719
    },
    {
      "epoch": 1.8805503845214845e-05,
      "step": 19719,
      "training_step_time": 6.785220384597778
    },
    {
      "epoch": 1.880645751953125e-05,
      "grad_norm": 0.16779427230358124,
      "learning_rate": 9.419292288152673e-06,
      "loss": 0.0632,
      "step": 19720
    },
    {
      "epoch": 1.880645751953125e-05,
      "model_forward_time": 1.7336018085479736,
      "step": 19720
    },
    {
      "dynamics_loss_avg": 0.06052697412669659,
      "epoch": 1.880645751953125e-05,
      "step": 19720
    },
    {
      "action_loss_avg": 0.0024936393601819873,
      "epoch": 1.880645751953125e-05,
      "step": 19720
    },
    {
      "epoch": 1.880645751953125e-05,
      "step": 19720,
      "training_step_time": 6.783016920089722
    },
    {
      "epoch": 1.8807411193847656e-05,
      "model_forward_time": 1.7382795810699463,
      "step": 19721
    },
    {
      "epoch": 1.8807411193847656e-05,
      "step": 19721,
      "training_step_time": 6.726926326751709
    },
    {
      "epoch": 1.8808364868164064e-05,
      "model_forward_time": 1.7179203033447266,
      "step": 19722
    },
    {
      "epoch": 1.8808364868164064e-05,
      "step": 19722,
      "training_step_time": 7.031071662902832
    },
    {
      "epoch": 1.8809318542480468e-05,
      "model_forward_time": 2.3244597911834717,
      "step": 19723
    },
    {
      "epoch": 1.8809318542480468e-05,
      "step": 19723,
      "training_step_time": 9.542242527008057
    },
    {
      "epoch": 1.8810272216796875e-05,
      "model_forward_time": 1.726489782333374,
      "step": 19724
    },
    {
      "epoch": 1.8810272216796875e-05,
      "step": 19724,
      "training_step_time": 7.786550998687744
    },
    {
      "epoch": 1.8811225891113283e-05,
      "model_forward_time": 1.7323391437530518,
      "step": 19725
    },
    {
      "epoch": 1.8811225891113283e-05,
      "step": 19725,
      "training_step_time": 7.308184623718262
    },
    {
      "epoch": 1.8812179565429687e-05,
      "model_forward_time": 1.7359418869018555,
      "step": 19726
    },
    {
      "epoch": 1.8812179565429687e-05,
      "step": 19726,
      "training_step_time": 6.708919286727905
    },
    {
      "epoch": 1.8813133239746094e-05,
      "model_forward_time": 1.7329332828521729,
      "step": 19727
    },
    {
      "epoch": 1.8813133239746094e-05,
      "step": 19727,
      "training_step_time": 6.751316547393799
    },
    {
      "epoch": 1.88140869140625e-05,
      "model_forward_time": 1.736666202545166,
      "step": 19728
    },
    {
      "epoch": 1.88140869140625e-05,
      "step": 19728,
      "training_step_time": 6.711255073547363
    },
    {
      "epoch": 1.8815040588378905e-05,
      "model_forward_time": 1.7345504760742188,
      "step": 19729
    },
    {
      "epoch": 1.8815040588378905e-05,
      "step": 19729,
      "training_step_time": 6.701316833496094
    },
    {
      "epoch": 1.8815994262695313e-05,
      "grad_norm": 0.1231662705540657,
      "learning_rate": 9.418518628655589e-06,
      "loss": 0.0662,
      "step": 19730
    },
    {
      "epoch": 1.8815994262695313e-05,
      "model_forward_time": 1.7116127014160156,
      "step": 19730
    },
    {
      "dynamics_loss_avg": 0.06678121406584978,
      "epoch": 1.8815994262695313e-05,
      "step": 19730
    },
    {
      "action_loss_avg": 0.0025607275019865483,
      "epoch": 1.8815994262695313e-05,
      "step": 19730
    },
    {
      "epoch": 1.8815994262695313e-05,
      "step": 19730,
      "training_step_time": 6.734999418258667
    },
    {
      "epoch": 1.881694793701172e-05,
      "model_forward_time": 1.7318625450134277,
      "step": 19731
    },
    {
      "epoch": 1.881694793701172e-05,
      "step": 19731,
      "training_step_time": 6.750128746032715
    },
    {
      "epoch": 1.8817901611328124e-05,
      "model_forward_time": 2.3084208965301514,
      "step": 19732
    },
    {
      "epoch": 1.8817901611328124e-05,
      "step": 19732,
      "training_step_time": 8.449825048446655
    },
    {
      "epoch": 1.8818855285644532e-05,
      "model_forward_time": 1.7302639484405518,
      "step": 19733
    },
    {
      "epoch": 1.8818855285644532e-05,
      "step": 19733,
      "training_step_time": 8.699167728424072
    },
    {
      "epoch": 1.881980895996094e-05,
      "model_forward_time": 1.7334001064300537,
      "step": 19734
    },
    {
      "epoch": 1.881980895996094e-05,
      "step": 19734,
      "training_step_time": 7.49506139755249
    },
    {
      "epoch": 1.8820762634277343e-05,
      "model_forward_time": 1.7312567234039307,
      "step": 19735
    },
    {
      "epoch": 1.8820762634277343e-05,
      "step": 19735,
      "training_step_time": 6.751863479614258
    },
    {
      "epoch": 1.882171630859375e-05,
      "model_forward_time": 1.7373194694519043,
      "step": 19736
    },
    {
      "epoch": 1.882171630859375e-05,
      "step": 19736,
      "training_step_time": 6.701392650604248
    },
    {
      "epoch": 1.8822669982910155e-05,
      "model_forward_time": 1.7145824432373047,
      "step": 19737
    },
    {
      "epoch": 1.8822669982910155e-05,
      "step": 19737,
      "training_step_time": 6.7174973487854
    },
    {
      "epoch": 1.8823623657226562e-05,
      "model_forward_time": 1.718348741531372,
      "step": 19738
    },
    {
      "epoch": 1.8823623657226562e-05,
      "step": 19738,
      "training_step_time": 6.751085996627808
    },
    {
      "epoch": 1.882457733154297e-05,
      "model_forward_time": 1.7276885509490967,
      "step": 19739
    },
    {
      "epoch": 1.882457733154297e-05,
      "step": 19739,
      "training_step_time": 6.72350549697876
    },
    {
      "epoch": 1.8825531005859374e-05,
      "grad_norm": 0.28125864267349243,
      "learning_rate": 9.417744485955951e-06,
      "loss": 0.0668,
      "step": 19740
    },
    {
      "epoch": 1.8825531005859374e-05,
      "model_forward_time": 1.7310755252838135,
      "step": 19740
    },
    {
      "dynamics_loss_avg": 0.058184122666716576,
      "epoch": 1.8825531005859374e-05,
      "step": 19740
    },
    {
      "action_loss_avg": 0.0031113557051867247,
      "epoch": 1.8825531005859374e-05,
      "step": 19740
    },
    {
      "epoch": 1.8825531005859374e-05,
      "step": 19740,
      "training_step_time": 6.727247476577759
    },
    {
      "epoch": 1.882648468017578e-05,
      "model_forward_time": 2.3017306327819824,
      "step": 19741
    },
    {
      "epoch": 1.882648468017578e-05,
      "step": 19741,
      "training_step_time": 7.634383201599121
    },
    {
      "epoch": 1.882743835449219e-05,
      "model_forward_time": 1.7306952476501465,
      "step": 19742
    },
    {
      "epoch": 1.882743835449219e-05,
      "step": 19742,
      "training_step_time": 12.070539236068726
    },
    {
      "epoch": 1.8828392028808592e-05,
      "model_forward_time": 1.724787950515747,
      "step": 19743
    },
    {
      "epoch": 1.8828392028808592e-05,
      "step": 19743,
      "training_step_time": 7.420822858810425
    },
    {
      "epoch": 1.8829345703125e-05,
      "model_forward_time": 1.3587589263916016,
      "step": 19744
    },
    {
      "epoch": 1.8829345703125e-05,
      "step": 19744,
      "training_step_time": 6.687967300415039
    },
    {
      "epoch": 1.8830299377441407e-05,
      "model_forward_time": 1.3720710277557373,
      "step": 19745
    },
    {
      "epoch": 1.8830299377441407e-05,
      "step": 19745,
      "training_step_time": 6.681715250015259
    },
    {
      "epoch": 1.883125305175781e-05,
      "model_forward_time": 1.7366857528686523,
      "step": 19746
    },
    {
      "epoch": 1.883125305175781e-05,
      "step": 19746,
      "training_step_time": 6.705610036849976
    },
    {
      "epoch": 1.883220672607422e-05,
      "model_forward_time": 1.718689203262329,
      "step": 19747
    },
    {
      "epoch": 1.883220672607422e-05,
      "step": 19747,
      "training_step_time": 6.669487237930298
    },
    {
      "epoch": 1.8833160400390626e-05,
      "model_forward_time": 1.7378406524658203,
      "step": 19748
    },
    {
      "epoch": 1.8833160400390626e-05,
      "step": 19748,
      "training_step_time": 6.679142713546753
    },
    {
      "epoch": 1.883411407470703e-05,
      "model_forward_time": 1.7229225635528564,
      "step": 19749
    },
    {
      "epoch": 1.883411407470703e-05,
      "step": 19749,
      "training_step_time": 6.726733207702637
    },
    {
      "epoch": 1.8835067749023438e-05,
      "grad_norm": 0.17692697048187256,
      "learning_rate": 9.41696986013842e-06,
      "loss": 0.061,
      "step": 19750
    },
    {
      "epoch": 1.8835067749023438e-05,
      "model_forward_time": 2.2760887145996094,
      "step": 19750
    },
    {
      "dynamics_loss_avg": 0.0561694061383605,
      "epoch": 1.8835067749023438e-05,
      "step": 19750
    },
    {
      "action_loss_avg": 0.002431392256403342,
      "epoch": 1.8835067749023438e-05,
      "step": 19750
    },
    {
      "epoch": 1.8835067749023438e-05,
      "step": 19750,
      "training_step_time": 7.882519960403442
    },
    {
      "epoch": 1.8836021423339845e-05,
      "model_forward_time": 1.7784996032714844,
      "step": 19751
    },
    {
      "epoch": 1.8836021423339845e-05,
      "step": 19751,
      "training_step_time": 9.778017282485962
    },
    {
      "epoch": 1.883697509765625e-05,
      "model_forward_time": 1.73443603515625,
      "step": 19752
    },
    {
      "epoch": 1.883697509765625e-05,
      "step": 19752,
      "training_step_time": 7.5691869258880615
    },
    {
      "epoch": 1.8837928771972656e-05,
      "model_forward_time": 1.7165236473083496,
      "step": 19753
    },
    {
      "epoch": 1.8837928771972656e-05,
      "step": 19753,
      "training_step_time": 7.253137111663818
    },
    {
      "epoch": 1.8838882446289064e-05,
      "model_forward_time": 1.7324209213256836,
      "step": 19754
    },
    {
      "epoch": 1.8838882446289064e-05,
      "step": 19754,
      "training_step_time": 6.717540264129639
    },
    {
      "epoch": 1.8839836120605468e-05,
      "model_forward_time": 1.7334070205688477,
      "step": 19755
    },
    {
      "epoch": 1.8839836120605468e-05,
      "step": 19755,
      "training_step_time": 6.715853929519653
    },
    {
      "epoch": 1.8840789794921875e-05,
      "model_forward_time": 1.721125841140747,
      "step": 19756
    },
    {
      "epoch": 1.8840789794921875e-05,
      "step": 19756,
      "training_step_time": 6.70837664604187
    },
    {
      "epoch": 1.8841743469238283e-05,
      "model_forward_time": 1.732409954071045,
      "step": 19757
    },
    {
      "epoch": 1.8841743469238283e-05,
      "step": 19757,
      "training_step_time": 6.7142767906188965
    },
    {
      "epoch": 1.8842697143554687e-05,
      "model_forward_time": 1.7360868453979492,
      "step": 19758
    },
    {
      "epoch": 1.8842697143554687e-05,
      "step": 19758,
      "training_step_time": 6.71852970123291
    },
    {
      "epoch": 1.8843650817871094e-05,
      "model_forward_time": 2.2913546562194824,
      "step": 19759
    },
    {
      "epoch": 1.8843650817871094e-05,
      "step": 19759,
      "training_step_time": 7.107140302658081
    },
    {
      "epoch": 1.88446044921875e-05,
      "grad_norm": 0.1796373873949051,
      "learning_rate": 9.416194751287706e-06,
      "loss": 0.0637,
      "step": 19760
    },
    {
      "epoch": 1.88446044921875e-05,
      "model_forward_time": 1.7791852951049805,
      "step": 19760
    },
    {
      "dynamics_loss_avg": 0.0732062978670001,
      "epoch": 1.88446044921875e-05,
      "step": 19760
    },
    {
      "action_loss_avg": 0.0027204545505810527,
      "epoch": 1.88446044921875e-05,
      "step": 19760
    },
    {
      "epoch": 1.88446044921875e-05,
      "step": 19760,
      "training_step_time": 9.221101522445679
    },
    {
      "epoch": 1.8845558166503906e-05,
      "model_forward_time": 1.73020339012146,
      "step": 19761
    },
    {
      "epoch": 1.8845558166503906e-05,
      "step": 19761,
      "training_step_time": 7.649737596511841
    },
    {
      "epoch": 1.8846511840820313e-05,
      "model_forward_time": 1.734964370727539,
      "step": 19762
    },
    {
      "epoch": 1.8846511840820313e-05,
      "step": 19762,
      "training_step_time": 7.379844903945923
    },
    {
      "epoch": 1.884746551513672e-05,
      "model_forward_time": 1.7237439155578613,
      "step": 19763
    },
    {
      "epoch": 1.884746551513672e-05,
      "step": 19763,
      "training_step_time": 6.699084281921387
    },
    {
      "epoch": 1.8848419189453124e-05,
      "model_forward_time": 1.7156140804290771,
      "step": 19764
    },
    {
      "epoch": 1.8848419189453124e-05,
      "step": 19764,
      "training_step_time": 6.695909023284912
    },
    {
      "epoch": 1.8849372863769532e-05,
      "model_forward_time": 1.730982780456543,
      "step": 19765
    },
    {
      "epoch": 1.8849372863769532e-05,
      "step": 19765,
      "training_step_time": 6.72266411781311
    },
    {
      "epoch": 1.8850326538085936e-05,
      "model_forward_time": 1.7330992221832275,
      "step": 19766
    },
    {
      "epoch": 1.8850326538085936e-05,
      "step": 19766,
      "training_step_time": 6.668794870376587
    },
    {
      "epoch": 1.8851280212402343e-05,
      "model_forward_time": 1.7381389141082764,
      "step": 19767
    },
    {
      "epoch": 1.8851280212402343e-05,
      "step": 19767,
      "training_step_time": 6.680171012878418
    },
    {
      "epoch": 1.885223388671875e-05,
      "model_forward_time": 2.267728328704834,
      "step": 19768
    },
    {
      "epoch": 1.885223388671875e-05,
      "step": 19768,
      "training_step_time": 7.102234363555908
    },
    {
      "epoch": 1.8853187561035155e-05,
      "model_forward_time": 1.794917345046997,
      "step": 19769
    },
    {
      "epoch": 1.8853187561035155e-05,
      "step": 19769,
      "training_step_time": 9.652461528778076
    },
    {
      "epoch": 1.8854141235351562e-05,
      "grad_norm": 0.14113469421863556,
      "learning_rate": 9.415419159488572e-06,
      "loss": 0.0604,
      "step": 19770
    },
    {
      "epoch": 1.8854141235351562e-05,
      "model_forward_time": 1.730151891708374,
      "step": 19770
    },
    {
      "dynamics_loss_avg": 0.06710452046245337,
      "epoch": 1.8854141235351562e-05,
      "step": 19770
    },
    {
      "action_loss_avg": 0.0028878140030428765,
      "epoch": 1.8854141235351562e-05,
      "step": 19770
    },
    {
      "epoch": 1.8854141235351562e-05,
      "step": 19770,
      "training_step_time": 8.645849227905273
    },
    {
      "epoch": 1.885509490966797e-05,
      "model_forward_time": 1.7043046951293945,
      "step": 19771
    },
    {
      "epoch": 1.885509490966797e-05,
      "step": 19771,
      "training_step_time": 7.382350206375122
    },
    {
      "epoch": 1.8856048583984374e-05,
      "model_forward_time": 1.7379777431488037,
      "step": 19772
    },
    {
      "epoch": 1.8856048583984374e-05,
      "step": 19772,
      "training_step_time": 6.681640625
    },
    {
      "epoch": 1.885700225830078e-05,
      "model_forward_time": 1.7364122867584229,
      "step": 19773
    },
    {
      "epoch": 1.885700225830078e-05,
      "step": 19773,
      "training_step_time": 6.675961494445801
    },
    {
      "epoch": 1.885795593261719e-05,
      "model_forward_time": 1.7366504669189453,
      "step": 19774
    },
    {
      "epoch": 1.885795593261719e-05,
      "step": 19774,
      "training_step_time": 6.696038007736206
    },
    {
      "epoch": 1.8858909606933592e-05,
      "model_forward_time": 1.740795612335205,
      "step": 19775
    },
    {
      "epoch": 1.8858909606933592e-05,
      "step": 19775,
      "training_step_time": 6.67506217956543
    },
    {
      "epoch": 1.885986328125e-05,
      "model_forward_time": 1.7333722114562988,
      "step": 19776
    },
    {
      "epoch": 1.885986328125e-05,
      "step": 19776,
      "training_step_time": 6.661154270172119
    },
    {
      "epoch": 1.8860816955566407e-05,
      "model_forward_time": 1.7128968238830566,
      "step": 19777
    },
    {
      "epoch": 1.8860816955566407e-05,
      "step": 19777,
      "training_step_time": 7.047938823699951
    },
    {
      "epoch": 1.886177062988281e-05,
      "model_forward_time": 2.911675453186035,
      "step": 19778
    },
    {
      "epoch": 1.886177062988281e-05,
      "step": 19778,
      "training_step_time": 9.4443359375
    },
    {
      "epoch": 1.886272430419922e-05,
      "model_forward_time": 1.7266886234283447,
      "step": 19779
    },
    {
      "epoch": 1.886272430419922e-05,
      "step": 19779,
      "training_step_time": 10.510398387908936
    },
    {
      "epoch": 1.8863677978515626e-05,
      "grad_norm": 0.1484106183052063,
      "learning_rate": 9.414643084825837e-06,
      "loss": 0.0607,
      "step": 19780
    },
    {
      "epoch": 1.8863677978515626e-05,
      "model_forward_time": 1.727881908416748,
      "step": 19780
    },
    {
      "dynamics_loss_avg": 0.05403104443103075,
      "epoch": 1.8863677978515626e-05,
      "step": 19780
    },
    {
      "action_loss_avg": 0.002329789855866693,
      "epoch": 1.8863677978515626e-05,
      "step": 19780
    },
    {
      "epoch": 1.8863677978515626e-05,
      "step": 19780,
      "training_step_time": 7.463349342346191
    },
    {
      "epoch": 1.886463165283203e-05,
      "model_forward_time": 1.7226324081420898,
      "step": 19781
    },
    {
      "epoch": 1.886463165283203e-05,
      "step": 19781,
      "training_step_time": 6.64583420753479
    },
    {
      "epoch": 1.8865585327148438e-05,
      "model_forward_time": 1.7222745418548584,
      "step": 19782
    },
    {
      "epoch": 1.8865585327148438e-05,
      "step": 19782,
      "training_step_time": 6.679931640625
    },
    {
      "epoch": 1.8866539001464845e-05,
      "model_forward_time": 1.725684404373169,
      "step": 19783
    },
    {
      "epoch": 1.8866539001464845e-05,
      "step": 19783,
      "training_step_time": 6.647199630737305
    },
    {
      "epoch": 1.886749267578125e-05,
      "model_forward_time": 1.7181174755096436,
      "step": 19784
    },
    {
      "epoch": 1.886749267578125e-05,
      "step": 19784,
      "training_step_time": 6.690864086151123
    },
    {
      "epoch": 1.8868446350097656e-05,
      "model_forward_time": 1.724195957183838,
      "step": 19785
    },
    {
      "epoch": 1.8868446350097656e-05,
      "step": 19785,
      "training_step_time": 6.708890676498413
    },
    {
      "epoch": 1.8869400024414064e-05,
      "model_forward_time": 1.731086254119873,
      "step": 19786
    },
    {
      "epoch": 1.8869400024414064e-05,
      "step": 19786,
      "training_step_time": 6.674346685409546
    },
    {
      "epoch": 1.8870353698730468e-05,
      "model_forward_time": 2.9464845657348633,
      "step": 19787
    },
    {
      "epoch": 1.8870353698730468e-05,
      "step": 19787,
      "training_step_time": 8.539016485214233
    },
    {
      "epoch": 1.8871307373046875e-05,
      "model_forward_time": 1.7194204330444336,
      "step": 19788
    },
    {
      "epoch": 1.8871307373046875e-05,
      "step": 19788,
      "training_step_time": 9.943921566009521
    },
    {
      "epoch": 1.8872261047363283e-05,
      "model_forward_time": 1.0186126232147217,
      "step": 19789
    },
    {
      "epoch": 1.8872261047363283e-05,
      "step": 19789,
      "training_step_time": 8.314415454864502
    },
    {
      "epoch": 1.8873214721679687e-05,
      "grad_norm": 0.16645073890686035,
      "learning_rate": 9.413866527384372e-06,
      "loss": 0.0578,
      "step": 19790
    },
    {
      "epoch": 1.8873214721679687e-05,
      "model_forward_time": 1.7206847667694092,
      "step": 19790
    },
    {
      "dynamics_loss_avg": 0.06763915438205004,
      "epoch": 1.8873214721679687e-05,
      "step": 19790
    },
    {
      "action_loss_avg": 0.0026800807216204705,
      "epoch": 1.8873214721679687e-05,
      "step": 19790
    },
    {
      "epoch": 1.8873214721679687e-05,
      "step": 19790,
      "training_step_time": 6.702188491821289
    },
    {
      "epoch": 1.8874168395996094e-05,
      "model_forward_time": 1.727936029434204,
      "step": 19791
    },
    {
      "epoch": 1.8874168395996094e-05,
      "step": 19791,
      "training_step_time": 6.6949303150177
    },
    {
      "epoch": 1.88751220703125e-05,
      "model_forward_time": 1.7320356369018555,
      "step": 19792
    },
    {
      "epoch": 1.88751220703125e-05,
      "step": 19792,
      "training_step_time": 6.672683238983154
    },
    {
      "epoch": 1.8876075744628906e-05,
      "model_forward_time": 1.723402738571167,
      "step": 19793
    },
    {
      "epoch": 1.8876075744628906e-05,
      "step": 19793,
      "training_step_time": 6.670179605484009
    },
    {
      "epoch": 1.8877029418945313e-05,
      "model_forward_time": 1.723783254623413,
      "step": 19794
    },
    {
      "epoch": 1.8877029418945313e-05,
      "step": 19794,
      "training_step_time": 6.665922403335571
    },
    {
      "epoch": 1.887798309326172e-05,
      "model_forward_time": 1.7247958183288574,
      "step": 19795
    },
    {
      "epoch": 1.887798309326172e-05,
      "step": 19795,
      "training_step_time": 6.689294099807739
    },
    {
      "epoch": 1.8878936767578124e-05,
      "model_forward_time": 2.883847951889038,
      "step": 19796
    },
    {
      "epoch": 1.8878936767578124e-05,
      "step": 19796,
      "training_step_time": 8.419719457626343
    },
    {
      "epoch": 1.8879890441894532e-05,
      "model_forward_time": 1.7600769996643066,
      "step": 19797
    },
    {
      "epoch": 1.8879890441894532e-05,
      "step": 19797,
      "training_step_time": 9.016915321350098
    },
    {
      "epoch": 1.8880844116210936e-05,
      "model_forward_time": 1.7228679656982422,
      "step": 19798
    },
    {
      "epoch": 1.8880844116210936e-05,
      "step": 19798,
      "training_step_time": 9.154037714004517
    },
    {
      "epoch": 1.8881797790527343e-05,
      "model_forward_time": 1.3598699569702148,
      "step": 19799
    },
    {
      "epoch": 1.8881797790527343e-05,
      "step": 19799,
      "training_step_time": 7.210415840148926
    },
    {
      "epoch": 1.888275146484375e-05,
      "grad_norm": 0.17947641015052795,
      "learning_rate": 9.4130894872491e-06,
      "loss": 0.0568,
      "step": 19800
    },
    {
      "epoch": 1.888275146484375e-05,
      "model_forward_time": 1.7268009185791016,
      "step": 19800
    },
    {
      "dynamics_loss_avg": 0.05641137808561325,
      "epoch": 1.888275146484375e-05,
      "step": 19800
    },
    {
      "action_loss_avg": 0.0024984640651382504,
      "epoch": 1.888275146484375e-05,
      "step": 19800
    },
    {
      "epoch": 1.888275146484375e-05,
      "step": 19800,
      "training_step_time": 6.68157434463501
    },
    {
      "epoch": 1.8883705139160155e-05,
      "model_forward_time": 1.7216296195983887,
      "step": 19801
    },
    {
      "epoch": 1.8883705139160155e-05,
      "step": 19801,
      "training_step_time": 6.678982257843018
    },
    {
      "epoch": 1.8884658813476562e-05,
      "model_forward_time": 1.712583065032959,
      "step": 19802
    },
    {
      "epoch": 1.8884658813476562e-05,
      "step": 19802,
      "training_step_time": 6.667559862136841
    },
    {
      "epoch": 1.888561248779297e-05,
      "model_forward_time": 1.724015474319458,
      "step": 19803
    },
    {
      "epoch": 1.888561248779297e-05,
      "step": 19803,
      "training_step_time": 6.659175157546997
    },
    {
      "epoch": 1.8886566162109374e-05,
      "model_forward_time": 1.738168716430664,
      "step": 19804
    },
    {
      "epoch": 1.8886566162109374e-05,
      "step": 19804,
      "training_step_time": 6.66672945022583
    },
    {
      "epoch": 1.888751983642578e-05,
      "model_forward_time": 1.7382354736328125,
      "step": 19805
    },
    {
      "epoch": 1.888751983642578e-05,
      "step": 19805,
      "training_step_time": 8.303691625595093
    },
    {
      "epoch": 1.888847351074219e-05,
      "model_forward_time": 2.9150004386901855,
      "step": 19806
    },
    {
      "epoch": 1.888847351074219e-05,
      "step": 19806,
      "training_step_time": 11.496810674667358
    },
    {
      "epoch": 1.8889427185058592e-05,
      "model_forward_time": 1.719560146331787,
      "step": 19807
    },
    {
      "epoch": 1.8889427185058592e-05,
      "step": 19807,
      "training_step_time": 7.596561908721924
    },
    {
      "epoch": 1.8890380859375e-05,
      "model_forward_time": 1.722945213317871,
      "step": 19808
    },
    {
      "epoch": 1.8890380859375e-05,
      "step": 19808,
      "training_step_time": 7.34576416015625
    },
    {
      "epoch": 1.8891334533691407e-05,
      "model_forward_time": 1.7295746803283691,
      "step": 19809
    },
    {
      "epoch": 1.8891334533691407e-05,
      "step": 19809,
      "training_step_time": 6.673448085784912
    },
    {
      "epoch": 1.889228820800781e-05,
      "grad_norm": 0.15803837776184082,
      "learning_rate": 9.412311964504997e-06,
      "loss": 0.0577,
      "step": 19810
    },
    {
      "epoch": 1.889228820800781e-05,
      "model_forward_time": 1.726391315460205,
      "step": 19810
    },
    {
      "dynamics_loss_avg": 0.06765844076871871,
      "epoch": 1.889228820800781e-05,
      "step": 19810
    },
    {
      "action_loss_avg": 0.0034336571756284685,
      "epoch": 1.889228820800781e-05,
      "step": 19810
    },
    {
      "epoch": 1.889228820800781e-05,
      "step": 19810,
      "training_step_time": 6.733607769012451
    },
    {
      "epoch": 1.889324188232422e-05,
      "model_forward_time": 1.721024513244629,
      "step": 19811
    },
    {
      "epoch": 1.889324188232422e-05,
      "step": 19811,
      "training_step_time": 6.6733903884887695
    },
    {
      "epoch": 1.8894195556640626e-05,
      "model_forward_time": 1.7204034328460693,
      "step": 19812
    },
    {
      "epoch": 1.8894195556640626e-05,
      "step": 19812,
      "training_step_time": 6.656816244125366
    },
    {
      "epoch": 1.889514923095703e-05,
      "model_forward_time": 1.722520112991333,
      "step": 19813
    },
    {
      "epoch": 1.889514923095703e-05,
      "step": 19813,
      "training_step_time": 6.678629398345947
    },
    {
      "epoch": 1.8896102905273438e-05,
      "model_forward_time": 1.7337350845336914,
      "step": 19814
    },
    {
      "epoch": 1.8896102905273438e-05,
      "step": 19814,
      "training_step_time": 8.346399307250977
    },
    {
      "epoch": 1.8897056579589845e-05,
      "model_forward_time": 2.8965673446655273,
      "step": 19815
    },
    {
      "epoch": 1.8897056579589845e-05,
      "step": 19815,
      "training_step_time": 9.193157434463501
    },
    {
      "epoch": 1.889801025390625e-05,
      "model_forward_time": 1.7203090190887451,
      "step": 19816
    },
    {
      "epoch": 1.889801025390625e-05,
      "step": 19816,
      "training_step_time": 7.62789511680603
    },
    {
      "epoch": 1.8898963928222656e-05,
      "model_forward_time": 1.7250096797943115,
      "step": 19817
    },
    {
      "epoch": 1.8898963928222656e-05,
      "step": 19817,
      "training_step_time": 7.950793027877808
    },
    {
      "epoch": 1.8899917602539064e-05,
      "model_forward_time": 1.7271850109100342,
      "step": 19818
    },
    {
      "epoch": 1.8899917602539064e-05,
      "step": 19818,
      "training_step_time": 6.680149793624878
    },
    {
      "epoch": 1.8900871276855468e-05,
      "model_forward_time": 1.725043773651123,
      "step": 19819
    },
    {
      "epoch": 1.8900871276855468e-05,
      "step": 19819,
      "training_step_time": 6.692078113555908
    },
    {
      "epoch": 1.8901824951171875e-05,
      "grad_norm": 0.19199925661087036,
      "learning_rate": 9.411533959237091e-06,
      "loss": 0.0588,
      "step": 19820
    },
    {
      "epoch": 1.8901824951171875e-05,
      "model_forward_time": 1.7208895683288574,
      "step": 19820
    },
    {
      "dynamics_loss_avg": 0.06226347479969263,
      "epoch": 1.8901824951171875e-05,
      "step": 19820
    },
    {
      "action_loss_avg": 0.0020464842004003004,
      "epoch": 1.8901824951171875e-05,
      "step": 19820
    },
    {
      "epoch": 1.8901824951171875e-05,
      "step": 19820,
      "training_step_time": 6.711350202560425
    },
    {
      "epoch": 1.8902778625488283e-05,
      "model_forward_time": 1.730903148651123,
      "step": 19821
    },
    {
      "epoch": 1.8902778625488283e-05,
      "step": 19821,
      "training_step_time": 6.698032379150391
    },
    {
      "epoch": 1.8903732299804687e-05,
      "model_forward_time": 1.7245914936065674,
      "step": 19822
    },
    {
      "epoch": 1.8903732299804687e-05,
      "step": 19822,
      "training_step_time": 6.642060041427612
    },
    {
      "epoch": 1.8904685974121094e-05,
      "model_forward_time": 1.7243287563323975,
      "step": 19823
    },
    {
      "epoch": 1.8904685974121094e-05,
      "step": 19823,
      "training_step_time": 7.576497316360474
    },
    {
      "epoch": 1.89056396484375e-05,
      "model_forward_time": 2.9001717567443848,
      "step": 19824
    },
    {
      "epoch": 1.89056396484375e-05,
      "step": 19824,
      "training_step_time": 9.7280592918396
    },
    {
      "epoch": 1.8906593322753906e-05,
      "model_forward_time": 1.7125885486602783,
      "step": 19825
    },
    {
      "epoch": 1.8906593322753906e-05,
      "step": 19825,
      "training_step_time": 9.842909574508667
    },
    {
      "epoch": 1.8907546997070313e-05,
      "model_forward_time": 1.7197344303131104,
      "step": 19826
    },
    {
      "epoch": 1.8907546997070313e-05,
      "step": 19826,
      "training_step_time": 7.971287488937378
    },
    {
      "epoch": 1.890850067138672e-05,
      "model_forward_time": 1.7142698764801025,
      "step": 19827
    },
    {
      "epoch": 1.890850067138672e-05,
      "step": 19827,
      "training_step_time": 7.416108131408691
    },
    {
      "epoch": 1.8909454345703124e-05,
      "model_forward_time": 1.3490879535675049,
      "step": 19828
    },
    {
      "epoch": 1.8909454345703124e-05,
      "step": 19828,
      "training_step_time": 6.62009334564209
    },
    {
      "epoch": 1.8910408020019532e-05,
      "model_forward_time": 1.3439741134643555,
      "step": 19829
    },
    {
      "epoch": 1.8910408020019532e-05,
      "step": 19829,
      "training_step_time": 6.617467164993286
    },
    {
      "epoch": 1.8911361694335936e-05,
      "grad_norm": 0.23035095632076263,
      "learning_rate": 9.410755471530465e-06,
      "loss": 0.0585,
      "step": 19830
    },
    {
      "epoch": 1.8911361694335936e-05,
      "model_forward_time": 1.7162480354309082,
      "step": 19830
    },
    {
      "dynamics_loss_avg": 0.07184989675879479,
      "epoch": 1.8911361694335936e-05,
      "step": 19830
    },
    {
      "action_loss_avg": 0.0033235751383472235,
      "epoch": 1.8911361694335936e-05,
      "step": 19830
    },
    {
      "epoch": 1.8911361694335936e-05,
      "step": 19830,
      "training_step_time": 6.666466474533081
    },
    {
      "epoch": 1.8912315368652343e-05,
      "model_forward_time": 1.7158865928649902,
      "step": 19831
    },
    {
      "epoch": 1.8912315368652343e-05,
      "step": 19831,
      "training_step_time": 6.641151428222656
    },
    {
      "epoch": 1.891326904296875e-05,
      "model_forward_time": 1.7139465808868408,
      "step": 19832
    },
    {
      "epoch": 1.891326904296875e-05,
      "step": 19832,
      "training_step_time": 8.562885999679565
    },
    {
      "epoch": 1.8914222717285155e-05,
      "model_forward_time": 2.848280429840088,
      "step": 19833
    },
    {
      "epoch": 1.8914222717285155e-05,
      "step": 19833,
      "training_step_time": 8.436051845550537
    },
    {
      "epoch": 1.8915176391601562e-05,
      "model_forward_time": 1.719630241394043,
      "step": 19834
    },
    {
      "epoch": 1.8915176391601562e-05,
      "step": 19834,
      "training_step_time": 7.750200271606445
    },
    {
      "epoch": 1.891613006591797e-05,
      "model_forward_time": 1.7102515697479248,
      "step": 19835
    },
    {
      "epoch": 1.891613006591797e-05,
      "step": 19835,
      "training_step_time": 7.6983489990234375
    },
    {
      "epoch": 1.8917083740234374e-05,
      "model_forward_time": 1.7116293907165527,
      "step": 19836
    },
    {
      "epoch": 1.8917083740234374e-05,
      "step": 19836,
      "training_step_time": 7.416602849960327
    },
    {
      "epoch": 1.891803741455078e-05,
      "model_forward_time": 1.3578619956970215,
      "step": 19837
    },
    {
      "epoch": 1.891803741455078e-05,
      "step": 19837,
      "training_step_time": 6.807040691375732
    },
    {
      "epoch": 1.891899108886719e-05,
      "model_forward_time": 1.7147164344787598,
      "step": 19838
    },
    {
      "epoch": 1.891899108886719e-05,
      "step": 19838,
      "training_step_time": 6.746608018875122
    },
    {
      "epoch": 1.8919944763183592e-05,
      "model_forward_time": 1.715430736541748,
      "step": 19839
    },
    {
      "epoch": 1.8919944763183592e-05,
      "step": 19839,
      "training_step_time": 6.744980335235596
    },
    {
      "epoch": 1.89208984375e-05,
      "grad_norm": 0.18706375360488892,
      "learning_rate": 9.40997650147025e-06,
      "loss": 0.0609,
      "step": 19840
    },
    {
      "epoch": 1.89208984375e-05,
      "model_forward_time": 1.7181041240692139,
      "step": 19840
    },
    {
      "dynamics_loss_avg": 0.05590988732874393,
      "epoch": 1.89208984375e-05,
      "step": 19840
    },
    {
      "action_loss_avg": 0.0024600141448900105,
      "epoch": 1.89208984375e-05,
      "step": 19840
    },
    {
      "epoch": 1.89208984375e-05,
      "step": 19840,
      "training_step_time": 6.790482997894287
    },
    {
      "epoch": 1.8921852111816407e-05,
      "model_forward_time": 1.718780279159546,
      "step": 19841
    },
    {
      "epoch": 1.8921852111816407e-05,
      "step": 19841,
      "training_step_time": 8.32331895828247
    },
    {
      "epoch": 1.892280578613281e-05,
      "model_forward_time": 1.724574327468872,
      "step": 19842
    },
    {
      "epoch": 1.892280578613281e-05,
      "step": 19842,
      "training_step_time": 10.03282117843628
    },
    {
      "epoch": 1.892375946044922e-05,
      "model_forward_time": 2.2744667530059814,
      "step": 19843
    },
    {
      "epoch": 1.892375946044922e-05,
      "step": 19843,
      "training_step_time": 11.212905406951904
    },
    {
      "epoch": 1.8924713134765626e-05,
      "model_forward_time": 1.3656961917877197,
      "step": 19844
    },
    {
      "epoch": 1.8924713134765626e-05,
      "step": 19844,
      "training_step_time": 9.50461483001709
    },
    {
      "epoch": 1.892566680908203e-05,
      "model_forward_time": 1.7329134941101074,
      "step": 19845
    },
    {
      "epoch": 1.892566680908203e-05,
      "step": 19845,
      "training_step_time": 7.396130561828613
    },
    {
      "epoch": 1.8926620483398438e-05,
      "model_forward_time": 1.7369074821472168,
      "step": 19846
    },
    {
      "epoch": 1.8926620483398438e-05,
      "step": 19846,
      "training_step_time": 6.667097568511963
    },
    {
      "epoch": 1.8927574157714845e-05,
      "model_forward_time": 1.7300667762756348,
      "step": 19847
    },
    {
      "epoch": 1.8927574157714845e-05,
      "step": 19847,
      "training_step_time": 6.701599597930908
    },
    {
      "epoch": 1.892852783203125e-05,
      "model_forward_time": 1.7335376739501953,
      "step": 19848
    },
    {
      "epoch": 1.892852783203125e-05,
      "step": 19848,
      "training_step_time": 6.690396785736084
    },
    {
      "epoch": 1.8929481506347656e-05,
      "model_forward_time": 1.7310712337493896,
      "step": 19849
    },
    {
      "epoch": 1.8929481506347656e-05,
      "step": 19849,
      "training_step_time": 6.665754795074463
    },
    {
      "epoch": 1.8930435180664064e-05,
      "grad_norm": 0.14374980330467224,
      "learning_rate": 9.409197049141638e-06,
      "loss": 0.0628,
      "step": 19850
    },
    {
      "epoch": 1.8930435180664064e-05,
      "model_forward_time": 1.7266781330108643,
      "step": 19850
    },
    {
      "dynamics_loss_avg": 0.059314587898552415,
      "epoch": 1.8930435180664064e-05,
      "step": 19850
    },
    {
      "action_loss_avg": 0.0033991155622061344,
      "epoch": 1.8930435180664064e-05,
      "step": 19850
    },
    {
      "epoch": 1.8930435180664064e-05,
      "step": 19850,
      "training_step_time": 7.837329149246216
    },
    {
      "epoch": 1.8931388854980468e-05,
      "model_forward_time": 1.7303285598754883,
      "step": 19851
    },
    {
      "epoch": 1.8931388854980468e-05,
      "step": 19851,
      "training_step_time": 9.139144897460938
    },
    {
      "epoch": 1.8932342529296875e-05,
      "model_forward_time": 1.8936278820037842,
      "step": 19852
    },
    {
      "epoch": 1.8932342529296875e-05,
      "step": 19852,
      "training_step_time": 9.770053386688232
    },
    {
      "epoch": 1.8933296203613283e-05,
      "model_forward_time": 1.787611961364746,
      "step": 19853
    },
    {
      "epoch": 1.8933296203613283e-05,
      "step": 19853,
      "training_step_time": 8.726213932037354
    },
    {
      "epoch": 1.8934249877929687e-05,
      "model_forward_time": 1.733844518661499,
      "step": 19854
    },
    {
      "epoch": 1.8934249877929687e-05,
      "step": 19854,
      "training_step_time": 8.17471957206726
    },
    {
      "epoch": 1.8935203552246094e-05,
      "model_forward_time": 1.7140324115753174,
      "step": 19855
    },
    {
      "epoch": 1.8935203552246094e-05,
      "step": 19855,
      "training_step_time": 7.35715913772583
    },
    {
      "epoch": 1.89361572265625e-05,
      "model_forward_time": 1.729978084564209,
      "step": 19856
    },
    {
      "epoch": 1.89361572265625e-05,
      "step": 19856,
      "training_step_time": 6.662092208862305
    },
    {
      "epoch": 1.8937110900878906e-05,
      "model_forward_time": 1.7001357078552246,
      "step": 19857
    },
    {
      "epoch": 1.8937110900878906e-05,
      "step": 19857,
      "training_step_time": 6.647827863693237
    },
    {
      "epoch": 1.8938064575195313e-05,
      "model_forward_time": 1.7236120700836182,
      "step": 19858
    },
    {
      "epoch": 1.8938064575195313e-05,
      "step": 19858,
      "training_step_time": 6.695647716522217
    },
    {
      "epoch": 1.893901824951172e-05,
      "model_forward_time": 1.7218034267425537,
      "step": 19859
    },
    {
      "epoch": 1.893901824951172e-05,
      "step": 19859,
      "training_step_time": 7.147369146347046
    },
    {
      "epoch": 1.8939971923828124e-05,
      "grad_norm": 0.14860478043556213,
      "learning_rate": 9.408417114629863e-06,
      "loss": 0.0656,
      "step": 19860
    },
    {
      "epoch": 1.8939971923828124e-05,
      "model_forward_time": 1.358891248703003,
      "step": 19860
    },
    {
      "dynamics_loss_avg": 0.04922406803816557,
      "epoch": 1.8939971923828124e-05,
      "step": 19860
    },
    {
      "action_loss_avg": 0.001730100845452398,
      "epoch": 1.8939971923828124e-05,
      "step": 19860
    },
    {
      "epoch": 1.8939971923828124e-05,
      "step": 19860,
      "training_step_time": 8.119945049285889
    },
    {
      "epoch": 1.8940925598144532e-05,
      "model_forward_time": 1.7237026691436768,
      "step": 19861
    },
    {
      "epoch": 1.8940925598144532e-05,
      "step": 19861,
      "training_step_time": 8.899002313613892
    },
    {
      "epoch": 1.8941879272460936e-05,
      "model_forward_time": 1.7621746063232422,
      "step": 19862
    },
    {
      "epoch": 1.8941879272460936e-05,
      "step": 19862,
      "training_step_time": 7.8983986377716064
    },
    {
      "epoch": 1.8942832946777343e-05,
      "model_forward_time": 1.8169012069702148,
      "step": 19863
    },
    {
      "epoch": 1.8942832946777343e-05,
      "step": 19863,
      "training_step_time": 8.2099130153656
    },
    {
      "epoch": 1.894378662109375e-05,
      "model_forward_time": 1.8059377670288086,
      "step": 19864
    },
    {
      "epoch": 1.894378662109375e-05,
      "step": 19864,
      "training_step_time": 8.103302955627441
    },
    {
      "epoch": 1.8944740295410155e-05,
      "model_forward_time": 1.7847633361816406,
      "step": 19865
    },
    {
      "epoch": 1.8944740295410155e-05,
      "step": 19865,
      "training_step_time": 6.726892709732056
    },
    {
      "epoch": 1.8945693969726562e-05,
      "model_forward_time": 1.8049585819244385,
      "step": 19866
    },
    {
      "epoch": 1.8945693969726562e-05,
      "step": 19866,
      "training_step_time": 6.7158286571502686
    },
    {
      "epoch": 1.894664764404297e-05,
      "model_forward_time": 1.807828426361084,
      "step": 19867
    },
    {
      "epoch": 1.894664764404297e-05,
      "step": 19867,
      "training_step_time": 6.661566495895386
    },
    {
      "epoch": 1.8947601318359374e-05,
      "model_forward_time": 1.8081879615783691,
      "step": 19868
    },
    {
      "epoch": 1.8947601318359374e-05,
      "step": 19868,
      "training_step_time": 7.1837780475616455
    },
    {
      "epoch": 1.894855499267578e-05,
      "model_forward_time": 1.8057146072387695,
      "step": 19869
    },
    {
      "epoch": 1.894855499267578e-05,
      "step": 19869,
      "training_step_time": 8.634038925170898
    },
    {
      "epoch": 1.894950866699219e-05,
      "grad_norm": 0.19496485590934753,
      "learning_rate": 9.407636698020222e-06,
      "loss": 0.0645,
      "step": 19870
    },
    {
      "epoch": 1.894950866699219e-05,
      "model_forward_time": 1.813215732574463,
      "step": 19870
    },
    {
      "dynamics_loss_avg": 0.05623037703335285,
      "epoch": 1.894950866699219e-05,
      "step": 19870
    },
    {
      "action_loss_avg": 0.0018568748899269849,
      "epoch": 1.894950866699219e-05,
      "step": 19870
    },
    {
      "epoch": 1.894950866699219e-05,
      "step": 19870,
      "training_step_time": 8.217175006866455
    },
    {
      "epoch": 1.8950462341308592e-05,
      "model_forward_time": 1.8079383373260498,
      "step": 19871
    },
    {
      "epoch": 1.8950462341308592e-05,
      "step": 19871,
      "training_step_time": 8.391199111938477
    },
    {
      "epoch": 1.8951416015625e-05,
      "model_forward_time": 1.8031470775604248,
      "step": 19872
    },
    {
      "epoch": 1.8951416015625e-05,
      "step": 19872,
      "training_step_time": 7.791034936904907
    },
    {
      "epoch": 1.8952369689941407e-05,
      "model_forward_time": 1.810314416885376,
      "step": 19873
    },
    {
      "epoch": 1.8952369689941407e-05,
      "step": 19873,
      "training_step_time": 7.802572727203369
    },
    {
      "epoch": 1.895332336425781e-05,
      "model_forward_time": 1.8117353916168213,
      "step": 19874
    },
    {
      "epoch": 1.895332336425781e-05,
      "step": 19874,
      "training_step_time": 7.361596584320068
    },
    {
      "epoch": 1.895427703857422e-05,
      "model_forward_time": 1.8112711906433105,
      "step": 19875
    },
    {
      "epoch": 1.895427703857422e-05,
      "step": 19875,
      "training_step_time": 6.699048042297363
    },
    {
      "epoch": 1.8955230712890626e-05,
      "model_forward_time": 1.807755470275879,
      "step": 19876
    },
    {
      "epoch": 1.8955230712890626e-05,
      "step": 19876,
      "training_step_time": 6.719680547714233
    },
    {
      "epoch": 1.895618438720703e-05,
      "model_forward_time": 1.795889139175415,
      "step": 19877
    },
    {
      "epoch": 1.895618438720703e-05,
      "step": 19877,
      "training_step_time": 8.743306398391724
    },
    {
      "epoch": 1.8957138061523438e-05,
      "model_forward_time": 1.8026869297027588,
      "step": 19878
    },
    {
      "epoch": 1.8957138061523438e-05,
      "step": 19878,
      "training_step_time": 8.307580709457397
    },
    {
      "epoch": 1.8958091735839845e-05,
      "model_forward_time": 1.8077218532562256,
      "step": 19879
    },
    {
      "epoch": 1.8958091735839845e-05,
      "step": 19879,
      "training_step_time": 8.707838296890259
    },
    {
      "epoch": 1.895904541015625e-05,
      "grad_norm": 0.13807272911071777,
      "learning_rate": 9.406855799398057e-06,
      "loss": 0.0645,
      "step": 19880
    },
    {
      "epoch": 1.895904541015625e-05,
      "model_forward_time": 2.326439142227173,
      "step": 19880
    },
    {
      "dynamics_loss_avg": 0.053890852443873885,
      "epoch": 1.895904541015625e-05,
      "step": 19880
    },
    {
      "action_loss_avg": 0.0016728702350519598,
      "epoch": 1.895904541015625e-05,
      "step": 19880
    },
    {
      "epoch": 1.895904541015625e-05,
      "step": 19880,
      "training_step_time": 9.660309553146362
    },
    {
      "epoch": 1.8959999084472656e-05,
      "model_forward_time": 1.8056097030639648,
      "step": 19881
    },
    {
      "epoch": 1.8959999084472656e-05,
      "step": 19881,
      "training_step_time": 7.6444666385650635
    },
    {
      "epoch": 1.8960952758789064e-05,
      "model_forward_time": 1.4413409233093262,
      "step": 19882
    },
    {
      "epoch": 1.8960952758789064e-05,
      "step": 19882,
      "training_step_time": 7.411844730377197
    },
    {
      "epoch": 1.8961906433105468e-05,
      "model_forward_time": 1.8017117977142334,
      "step": 19883
    },
    {
      "epoch": 1.8961906433105468e-05,
      "step": 19883,
      "training_step_time": 7.439388275146484
    },
    {
      "epoch": 1.8962860107421875e-05,
      "model_forward_time": 1.8080174922943115,
      "step": 19884
    },
    {
      "epoch": 1.8962860107421875e-05,
      "step": 19884,
      "training_step_time": 6.720710039138794
    },
    {
      "epoch": 1.8963813781738283e-05,
      "model_forward_time": 1.8081402778625488,
      "step": 19885
    },
    {
      "epoch": 1.8963813781738283e-05,
      "step": 19885,
      "training_step_time": 6.744251489639282
    },
    {
      "epoch": 1.8964767456054687e-05,
      "model_forward_time": 1.810122013092041,
      "step": 19886
    },
    {
      "epoch": 1.8964767456054687e-05,
      "step": 19886,
      "training_step_time": 7.216622591018677
    },
    {
      "epoch": 1.8965721130371094e-05,
      "model_forward_time": 1.8142030239105225,
      "step": 19887
    },
    {
      "epoch": 1.8965721130371094e-05,
      "step": 19887,
      "training_step_time": 7.6243577003479
    },
    {
      "epoch": 1.89666748046875e-05,
      "model_forward_time": 1.8089370727539062,
      "step": 19888
    },
    {
      "epoch": 1.89666748046875e-05,
      "step": 19888,
      "training_step_time": 7.7790820598602295
    },
    {
      "epoch": 1.8967628479003906e-05,
      "model_forward_time": 2.327155828475952,
      "step": 19889
    },
    {
      "epoch": 1.8967628479003906e-05,
      "step": 19889,
      "training_step_time": 7.709475040435791
    },
    {
      "epoch": 1.8968582153320313e-05,
      "grad_norm": 0.17289970815181732,
      "learning_rate": 9.406074418848767e-06,
      "loss": 0.0658,
      "step": 19890
    },
    {
      "epoch": 1.8968582153320313e-05,
      "model_forward_time": 1.8060061931610107,
      "step": 19890
    },
    {
      "dynamics_loss_avg": 0.053689437732100484,
      "epoch": 1.8968582153320313e-05,
      "step": 19890
    },
    {
      "action_loss_avg": 0.0028185083123389632,
      "epoch": 1.8968582153320313e-05,
      "step": 19890
    },
    {
      "epoch": 1.8968582153320313e-05,
      "step": 19890,
      "training_step_time": 8.640872955322266
    },
    {
      "epoch": 1.8969535827636717e-05,
      "model_forward_time": 1.7992501258850098,
      "step": 19891
    },
    {
      "epoch": 1.8969535827636717e-05,
      "step": 19891,
      "training_step_time": 7.638681411743164
    },
    {
      "epoch": 1.8970489501953124e-05,
      "model_forward_time": 1.805706262588501,
      "step": 19892
    },
    {
      "epoch": 1.8970489501953124e-05,
      "step": 19892,
      "training_step_time": 7.343315362930298
    },
    {
      "epoch": 1.8971443176269532e-05,
      "model_forward_time": 1.8052546977996826,
      "step": 19893
    },
    {
      "epoch": 1.8971443176269532e-05,
      "step": 19893,
      "training_step_time": 6.694107532501221
    },
    {
      "epoch": 1.8972396850585936e-05,
      "model_forward_time": 1.8096325397491455,
      "step": 19894
    },
    {
      "epoch": 1.8972396850585936e-05,
      "step": 19894,
      "training_step_time": 6.660516977310181
    },
    {
      "epoch": 1.8973350524902343e-05,
      "model_forward_time": 1.8029978275299072,
      "step": 19895
    },
    {
      "epoch": 1.8973350524902343e-05,
      "step": 19895,
      "training_step_time": 7.143524408340454
    },
    {
      "epoch": 1.897430419921875e-05,
      "model_forward_time": 1.7944886684417725,
      "step": 19896
    },
    {
      "epoch": 1.897430419921875e-05,
      "step": 19896,
      "training_step_time": 8.422959327697754
    },
    {
      "epoch": 1.8975257873535155e-05,
      "model_forward_time": 1.8094089031219482,
      "step": 19897
    },
    {
      "epoch": 1.8975257873535155e-05,
      "step": 19897,
      "training_step_time": 8.978528022766113
    },
    {
      "epoch": 1.8976211547851562e-05,
      "model_forward_time": 2.32792592048645,
      "step": 19898
    },
    {
      "epoch": 1.8976211547851562e-05,
      "step": 19898,
      "training_step_time": 9.373604774475098
    },
    {
      "epoch": 1.897716522216797e-05,
      "model_forward_time": 1.8058576583862305,
      "step": 19899
    },
    {
      "epoch": 1.897716522216797e-05,
      "step": 19899,
      "training_step_time": 8.50883173942566
    },
    {
      "epoch": 1.8978118896484374e-05,
      "grad_norm": 0.12079418450593948,
      "learning_rate": 9.405292556457806e-06,
      "loss": 0.0615,
      "step": 19900
    },
    {
      "epoch": 1.8978118896484374e-05,
      "model_forward_time": 1.811065912246704,
      "step": 19900
    },
    {
      "dynamics_loss_avg": 0.054851115122437476,
      "epoch": 1.8978118896484374e-05,
      "step": 19900
    },
    {
      "action_loss_avg": 0.002927461074432358,
      "epoch": 1.8978118896484374e-05,
      "step": 19900
    },
    {
      "epoch": 1.8978118896484374e-05,
      "step": 19900,
      "training_step_time": 7.613064527511597
    },
    {
      "epoch": 1.897907257080078e-05,
      "model_forward_time": 1.4398915767669678,
      "step": 19901
    },
    {
      "epoch": 1.897907257080078e-05,
      "step": 19901,
      "training_step_time": 7.055482864379883
    },
    {
      "epoch": 1.898002624511719e-05,
      "model_forward_time": 1.8036525249481201,
      "step": 19902
    },
    {
      "epoch": 1.898002624511719e-05,
      "step": 19902,
      "training_step_time": 7.3944196701049805
    },
    {
      "epoch": 1.8980979919433593e-05,
      "model_forward_time": 1.4403138160705566,
      "step": 19903
    },
    {
      "epoch": 1.8980979919433593e-05,
      "step": 19903,
      "training_step_time": 6.677639961242676
    },
    {
      "epoch": 1.898193359375e-05,
      "model_forward_time": 1.8085110187530518,
      "step": 19904
    },
    {
      "epoch": 1.898193359375e-05,
      "step": 19904,
      "training_step_time": 7.186170339584351
    },
    {
      "epoch": 1.8982887268066407e-05,
      "model_forward_time": 1.8061203956604004,
      "step": 19905
    },
    {
      "epoch": 1.8982887268066407e-05,
      "step": 19905,
      "training_step_time": 7.447190046310425
    },
    {
      "epoch": 1.898384094238281e-05,
      "model_forward_time": 1.8034744262695312,
      "step": 19906
    },
    {
      "epoch": 1.898384094238281e-05,
      "step": 19906,
      "training_step_time": 7.729510307312012
    },
    {
      "epoch": 1.898479461669922e-05,
      "model_forward_time": 2.317556619644165,
      "step": 19907
    },
    {
      "epoch": 1.898479461669922e-05,
      "step": 19907,
      "training_step_time": 8.751230478286743
    },
    {
      "epoch": 1.8985748291015626e-05,
      "model_forward_time": 1.7223029136657715,
      "step": 19908
    },
    {
      "epoch": 1.8985748291015626e-05,
      "step": 19908,
      "training_step_time": 8.645971059799194
    },
    {
      "epoch": 1.898670196533203e-05,
      "model_forward_time": 1.721407175064087,
      "step": 19909
    },
    {
      "epoch": 1.898670196533203e-05,
      "step": 19909,
      "training_step_time": 8.35213041305542
    },
    {
      "epoch": 1.8987655639648438e-05,
      "grad_norm": 0.206633061170578,
      "learning_rate": 9.404510212310672e-06,
      "loss": 0.0631,
      "step": 19910
    },
    {
      "epoch": 1.8987655639648438e-05,
      "model_forward_time": 1.7105262279510498,
      "step": 19910
    },
    {
      "dynamics_loss_avg": 0.06371135339140892,
      "epoch": 1.8987655639648438e-05,
      "step": 19910
    },
    {
      "action_loss_avg": 0.002636233757948503,
      "epoch": 1.8987655639648438e-05,
      "step": 19910
    },
    {
      "epoch": 1.8987655639648438e-05,
      "step": 19910,
      "training_step_time": 7.297150611877441
    },
    {
      "epoch": 1.8988609313964845e-05,
      "model_forward_time": 1.7190375328063965,
      "step": 19911
    },
    {
      "epoch": 1.8988609313964845e-05,
      "step": 19911,
      "training_step_time": 7.572605133056641
    },
    {
      "epoch": 1.898956298828125e-05,
      "model_forward_time": 1.7208926677703857,
      "step": 19912
    },
    {
      "epoch": 1.898956298828125e-05,
      "step": 19912,
      "training_step_time": 6.705862522125244
    },
    {
      "epoch": 1.8990516662597656e-05,
      "model_forward_time": 1.716773271560669,
      "step": 19913
    },
    {
      "epoch": 1.8990516662597656e-05,
      "step": 19913,
      "training_step_time": 7.1787004470825195
    },
    {
      "epoch": 1.8991470336914064e-05,
      "model_forward_time": 1.7171895503997803,
      "step": 19914
    },
    {
      "epoch": 1.8991470336914064e-05,
      "step": 19914,
      "training_step_time": 7.378541707992554
    },
    {
      "epoch": 1.8992424011230468e-05,
      "model_forward_time": 1.7187049388885498,
      "step": 19915
    },
    {
      "epoch": 1.8992424011230468e-05,
      "step": 19915,
      "training_step_time": 7.720452308654785
    },
    {
      "epoch": 1.8993377685546875e-05,
      "model_forward_time": 1.7565958499908447,
      "step": 19916
    },
    {
      "epoch": 1.8993377685546875e-05,
      "step": 19916,
      "training_step_time": 9.664582967758179
    },
    {
      "epoch": 1.8994331359863283e-05,
      "model_forward_time": 2.232832193374634,
      "step": 19917
    },
    {
      "epoch": 1.8994331359863283e-05,
      "step": 19917,
      "training_step_time": 8.325592279434204
    },
    {
      "epoch": 1.8995285034179687e-05,
      "model_forward_time": 1.719958782196045,
      "step": 19918
    },
    {
      "epoch": 1.8995285034179687e-05,
      "step": 19918,
      "training_step_time": 7.648944854736328
    },
    {
      "epoch": 1.8996238708496094e-05,
      "model_forward_time": 1.715771198272705,
      "step": 19919
    },
    {
      "epoch": 1.8996238708496094e-05,
      "step": 19919,
      "training_step_time": 8.170339345932007
    },
    {
      "epoch": 1.89971923828125e-05,
      "grad_norm": 0.14656434953212738,
      "learning_rate": 9.403727386492923e-06,
      "loss": 0.0583,
      "step": 19920
    },
    {
      "epoch": 1.89971923828125e-05,
      "model_forward_time": 1.7141098976135254,
      "step": 19920
    },
    {
      "dynamics_loss_avg": 0.07308703288435936,
      "epoch": 1.89971923828125e-05,
      "step": 19920
    },
    {
      "action_loss_avg": 0.0026426555356010794,
      "epoch": 1.89971923828125e-05,
      "step": 19920
    },
    {
      "epoch": 1.89971923828125e-05,
      "step": 19920,
      "training_step_time": 6.723533868789673
    },
    {
      "epoch": 1.8998146057128906e-05,
      "model_forward_time": 1.7123076915740967,
      "step": 19921
    },
    {
      "epoch": 1.8998146057128906e-05,
      "step": 19921,
      "training_step_time": 7.402761459350586
    },
    {
      "epoch": 1.8999099731445313e-05,
      "model_forward_time": 1.7121713161468506,
      "step": 19922
    },
    {
      "epoch": 1.8999099731445313e-05,
      "step": 19922,
      "training_step_time": 7.254647493362427
    },
    {
      "epoch": 1.9000053405761717e-05,
      "model_forward_time": 1.7138261795043945,
      "step": 19923
    },
    {
      "epoch": 1.9000053405761717e-05,
      "step": 19923,
      "training_step_time": 7.415997266769409
    },
    {
      "epoch": 1.9001007080078125e-05,
      "model_forward_time": 1.7136921882629395,
      "step": 19924
    },
    {
      "epoch": 1.9001007080078125e-05,
      "step": 19924,
      "training_step_time": 7.706037521362305
    },
    {
      "epoch": 1.9001960754394532e-05,
      "model_forward_time": 1.7548363208770752,
      "step": 19925
    },
    {
      "epoch": 1.9001960754394532e-05,
      "step": 19925,
      "training_step_time": 8.722232580184937
    },
    {
      "epoch": 1.9002914428710936e-05,
      "model_forward_time": 1.8680520057678223,
      "step": 19926
    },
    {
      "epoch": 1.9002914428710936e-05,
      "step": 19926,
      "training_step_time": 9.347195386886597
    },
    {
      "epoch": 1.9003868103027343e-05,
      "model_forward_time": 1.7188642024993896,
      "step": 19927
    },
    {
      "epoch": 1.9003868103027343e-05,
      "step": 19927,
      "training_step_time": 7.732628583908081
    },
    {
      "epoch": 1.900482177734375e-05,
      "model_forward_time": 1.7198925018310547,
      "step": 19928
    },
    {
      "epoch": 1.900482177734375e-05,
      "step": 19928,
      "training_step_time": 7.637976408004761
    },
    {
      "epoch": 1.9005775451660155e-05,
      "model_forward_time": 1.7223563194274902,
      "step": 19929
    },
    {
      "epoch": 1.9005775451660155e-05,
      "step": 19929,
      "training_step_time": 7.31419563293457
    },
    {
      "epoch": 1.9006729125976562e-05,
      "grad_norm": 0.14010322093963623,
      "learning_rate": 9.40294407909017e-06,
      "loss": 0.0558,
      "step": 19930
    },
    {
      "epoch": 1.9006729125976562e-05,
      "model_forward_time": 1.7155561447143555,
      "step": 19930
    },
    {
      "dynamics_loss_avg": 0.06765767931938171,
      "epoch": 1.9006729125976562e-05,
      "step": 19930
    },
    {
      "action_loss_avg": 0.00415209275088273,
      "epoch": 1.9006729125976562e-05,
      "step": 19930
    },
    {
      "epoch": 1.9006729125976562e-05,
      "step": 19930,
      "training_step_time": 7.399324655532837
    },
    {
      "epoch": 1.900768280029297e-05,
      "model_forward_time": 1.7213075160980225,
      "step": 19931
    },
    {
      "epoch": 1.900768280029297e-05,
      "step": 19931,
      "training_step_time": 7.221646308898926
    },
    {
      "epoch": 1.9008636474609374e-05,
      "model_forward_time": 1.7203702926635742,
      "step": 19932
    },
    {
      "epoch": 1.9008636474609374e-05,
      "step": 19932,
      "training_step_time": 7.519124507904053
    },
    {
      "epoch": 1.900959014892578e-05,
      "model_forward_time": 1.7153286933898926,
      "step": 19933
    },
    {
      "epoch": 1.900959014892578e-05,
      "step": 19933,
      "training_step_time": 8.340821266174316
    },
    {
      "epoch": 1.901054382324219e-05,
      "model_forward_time": 1.7633929252624512,
      "step": 19934
    },
    {
      "epoch": 1.901054382324219e-05,
      "step": 19934,
      "training_step_time": 9.87565565109253
    },
    {
      "epoch": 1.9011497497558593e-05,
      "model_forward_time": 2.246856927871704,
      "step": 19935
    },
    {
      "epoch": 1.9011497497558593e-05,
      "step": 19935,
      "training_step_time": 10.130363702774048
    },
    {
      "epoch": 1.9012451171875e-05,
      "model_forward_time": 1.7167537212371826,
      "step": 19936
    },
    {
      "epoch": 1.9012451171875e-05,
      "step": 19936,
      "training_step_time": 9.642257452011108
    },
    {
      "epoch": 1.9013404846191407e-05,
      "model_forward_time": 1.3530354499816895,
      "step": 19937
    },
    {
      "epoch": 1.9013404846191407e-05,
      "step": 19937,
      "training_step_time": 8.419527769088745
    },
    {
      "epoch": 1.901435852050781e-05,
      "model_forward_time": 1.727602481842041,
      "step": 19938
    },
    {
      "epoch": 1.901435852050781e-05,
      "step": 19938,
      "training_step_time": 7.3295252323150635
    },
    {
      "epoch": 1.901531219482422e-05,
      "model_forward_time": 1.7339050769805908,
      "step": 19939
    },
    {
      "epoch": 1.901531219482422e-05,
      "step": 19939,
      "training_step_time": 7.387231349945068
    },
    {
      "epoch": 1.9016265869140626e-05,
      "grad_norm": 0.15155577659606934,
      "learning_rate": 9.402160290188069e-06,
      "loss": 0.06,
      "step": 19940
    },
    {
      "epoch": 1.9016265869140626e-05,
      "model_forward_time": 1.72135329246521,
      "step": 19940
    },
    {
      "dynamics_loss_avg": 0.06041617626324296,
      "epoch": 1.9016265869140626e-05,
      "step": 19940
    },
    {
      "action_loss_avg": 0.002299803995992988,
      "epoch": 1.9016265869140626e-05,
      "step": 19940
    },
    {
      "epoch": 1.9016265869140626e-05,
      "step": 19940,
      "training_step_time": 7.217947959899902
    },
    {
      "epoch": 1.901721954345703e-05,
      "model_forward_time": 1.729421615600586,
      "step": 19941
    },
    {
      "epoch": 1.901721954345703e-05,
      "step": 19941,
      "training_step_time": 7.399511337280273
    },
    {
      "epoch": 1.9018173217773438e-05,
      "model_forward_time": 1.3553612232208252,
      "step": 19942
    },
    {
      "epoch": 1.9018173217773438e-05,
      "step": 19942,
      "training_step_time": 7.532453298568726
    },
    {
      "epoch": 1.9019126892089845e-05,
      "model_forward_time": 1.77186918258667,
      "step": 19943
    },
    {
      "epoch": 1.9019126892089845e-05,
      "step": 19943,
      "training_step_time": 9.063901424407959
    },
    {
      "epoch": 1.902008056640625e-05,
      "model_forward_time": 1.7264330387115479,
      "step": 19944
    },
    {
      "epoch": 1.902008056640625e-05,
      "step": 19944,
      "training_step_time": 8.872259616851807
    },
    {
      "epoch": 1.9021034240722657e-05,
      "model_forward_time": 1.7354331016540527,
      "step": 19945
    },
    {
      "epoch": 1.9021034240722657e-05,
      "step": 19945,
      "training_step_time": 9.297334432601929
    },
    {
      "epoch": 1.9021987915039064e-05,
      "model_forward_time": 1.7261147499084473,
      "step": 19946
    },
    {
      "epoch": 1.9021987915039064e-05,
      "step": 19946,
      "training_step_time": 7.695099592208862
    },
    {
      "epoch": 1.9022941589355468e-05,
      "model_forward_time": 1.7231707572937012,
      "step": 19947
    },
    {
      "epoch": 1.9022941589355468e-05,
      "step": 19947,
      "training_step_time": 7.311903476715088
    },
    {
      "epoch": 1.9023895263671875e-05,
      "model_forward_time": 1.7272093296051025,
      "step": 19948
    },
    {
      "epoch": 1.9023895263671875e-05,
      "step": 19948,
      "training_step_time": 7.374654054641724
    },
    {
      "epoch": 1.9024848937988283e-05,
      "model_forward_time": 1.729454755783081,
      "step": 19949
    },
    {
      "epoch": 1.9024848937988283e-05,
      "step": 19949,
      "training_step_time": 7.2291247844696045
    },
    {
      "epoch": 1.9025802612304687e-05,
      "grad_norm": 0.21065078675746918,
      "learning_rate": 9.401376019872339e-06,
      "loss": 0.0607,
      "step": 19950
    },
    {
      "epoch": 1.9025802612304687e-05,
      "model_forward_time": 1.7196788787841797,
      "step": 19950
    },
    {
      "dynamics_loss_avg": 0.04950255043804645,
      "epoch": 1.9025802612304687e-05,
      "step": 19950
    },
    {
      "action_loss_avg": 0.0023236241599079222,
      "epoch": 1.9025802612304687e-05,
      "step": 19950
    },
    {
      "epoch": 1.9025802612304687e-05,
      "step": 19950,
      "training_step_time": 7.39407205581665
    },
    {
      "epoch": 1.9026756286621094e-05,
      "model_forward_time": 1.7361185550689697,
      "step": 19951
    },
    {
      "epoch": 1.9026756286621094e-05,
      "step": 19951,
      "training_step_time": 8.202820301055908
    },
    {
      "epoch": 1.90277099609375e-05,
      "model_forward_time": 1.8051438331604004,
      "step": 19952
    },
    {
      "epoch": 1.90277099609375e-05,
      "step": 19952,
      "training_step_time": 7.742684841156006
    },
    {
      "epoch": 1.9028663635253906e-05,
      "model_forward_time": 1.8228793144226074,
      "step": 19953
    },
    {
      "epoch": 1.9028663635253906e-05,
      "step": 19953,
      "training_step_time": 9.298909664154053
    },
    {
      "epoch": 1.9029617309570313e-05,
      "model_forward_time": 2.3547096252441406,
      "step": 19954
    },
    {
      "epoch": 1.9029617309570313e-05,
      "step": 19954,
      "training_step_time": 8.43307089805603
    },
    {
      "epoch": 1.9030570983886717e-05,
      "model_forward_time": 1.8098187446594238,
      "step": 19955
    },
    {
      "epoch": 1.9030570983886717e-05,
      "step": 19955,
      "training_step_time": 7.7333009243011475
    },
    {
      "epoch": 1.9031524658203125e-05,
      "model_forward_time": 1.7254877090454102,
      "step": 19956
    },
    {
      "epoch": 1.9031524658203125e-05,
      "step": 19956,
      "training_step_time": 7.3309314250946045
    },
    {
      "epoch": 1.9032478332519532e-05,
      "model_forward_time": 1.7297039031982422,
      "step": 19957
    },
    {
      "epoch": 1.9032478332519532e-05,
      "step": 19957,
      "training_step_time": 7.28495454788208
    },
    {
      "epoch": 1.9033432006835936e-05,
      "model_forward_time": 1.7263126373291016,
      "step": 19958
    },
    {
      "epoch": 1.9033432006835936e-05,
      "step": 19958,
      "training_step_time": 7.40169620513916
    },
    {
      "epoch": 1.9034385681152343e-05,
      "model_forward_time": 1.730571985244751,
      "step": 19959
    },
    {
      "epoch": 1.9034385681152343e-05,
      "step": 19959,
      "training_step_time": 7.968586206436157
    },
    {
      "epoch": 1.903533935546875e-05,
      "grad_norm": 0.1324402093887329,
      "learning_rate": 9.400591268228746e-06,
      "loss": 0.0652,
      "step": 19960
    },
    {
      "epoch": 1.903533935546875e-05,
      "model_forward_time": 1.7246654033660889,
      "step": 19960
    },
    {
      "dynamics_loss_avg": 0.050673889741301534,
      "epoch": 1.903533935546875e-05,
      "step": 19960
    },
    {
      "action_loss_avg": 0.0019106337480479852,
      "epoch": 1.903533935546875e-05,
      "step": 19960
    },
    {
      "epoch": 1.903533935546875e-05,
      "step": 19960,
      "training_step_time": 7.629114389419556
    },
    {
      "epoch": 1.9036293029785155e-05,
      "model_forward_time": 1.7271661758422852,
      "step": 19961
    },
    {
      "epoch": 1.9036293029785155e-05,
      "step": 19961,
      "training_step_time": 8.807256698608398
    },
    {
      "epoch": 1.9037246704101562e-05,
      "model_forward_time": 1.7261254787445068,
      "step": 19962
    },
    {
      "epoch": 1.9037246704101562e-05,
      "step": 19962,
      "training_step_time": 8.547677516937256
    },
    {
      "epoch": 1.903820037841797e-05,
      "model_forward_time": 2.2658629417419434,
      "step": 19963
    },
    {
      "epoch": 1.903820037841797e-05,
      "step": 19963,
      "training_step_time": 7.714771270751953
    },
    {
      "epoch": 1.9039154052734374e-05,
      "model_forward_time": 1.7733356952667236,
      "step": 19964
    },
    {
      "epoch": 1.9039154052734374e-05,
      "step": 19964,
      "training_step_time": 7.718096494674683
    },
    {
      "epoch": 1.904010772705078e-05,
      "model_forward_time": 1.730590581893921,
      "step": 19965
    },
    {
      "epoch": 1.904010772705078e-05,
      "step": 19965,
      "training_step_time": 7.727206707000732
    },
    {
      "epoch": 1.904106140136719e-05,
      "model_forward_time": 1.7184453010559082,
      "step": 19966
    },
    {
      "epoch": 1.904106140136719e-05,
      "step": 19966,
      "training_step_time": 7.311938524246216
    },
    {
      "epoch": 1.9042015075683593e-05,
      "model_forward_time": 1.7197377681732178,
      "step": 19967
    },
    {
      "epoch": 1.9042015075683593e-05,
      "step": 19967,
      "training_step_time": 7.4144370555877686
    },
    {
      "epoch": 1.904296875e-05,
      "model_forward_time": 1.7094862461090088,
      "step": 19968
    },
    {
      "epoch": 1.904296875e-05,
      "step": 19968,
      "training_step_time": 7.377087354660034
    },
    {
      "epoch": 1.9043922424316407e-05,
      "model_forward_time": 1.7147178649902344,
      "step": 19969
    },
    {
      "epoch": 1.9043922424316407e-05,
      "step": 19969,
      "training_step_time": 8.153326749801636
    },
    {
      "epoch": 1.904487609863281e-05,
      "grad_norm": 0.19896113872528076,
      "learning_rate": 9.399806035343106e-06,
      "loss": 0.0598,
      "step": 19970
    },
    {
      "epoch": 1.904487609863281e-05,
      "model_forward_time": 1.7113709449768066,
      "step": 19970
    },
    {
      "dynamics_loss_avg": 0.057639028131961825,
      "epoch": 1.904487609863281e-05,
      "step": 19970
    },
    {
      "action_loss_avg": 0.002589676692150533,
      "epoch": 1.904487609863281e-05,
      "step": 19970
    },
    {
      "epoch": 1.904487609863281e-05,
      "step": 19970,
      "training_step_time": 7.679728031158447
    },
    {
      "epoch": 1.904582977294922e-05,
      "model_forward_time": 1.7141034603118896,
      "step": 19971
    },
    {
      "epoch": 1.904582977294922e-05,
      "step": 19971,
      "training_step_time": 9.65703296661377
    },
    {
      "epoch": 1.9046783447265626e-05,
      "model_forward_time": 2.273481607437134,
      "step": 19972
    },
    {
      "epoch": 1.9046783447265626e-05,
      "step": 19972,
      "training_step_time": 9.54662823677063
    },
    {
      "epoch": 1.904773712158203e-05,
      "model_forward_time": 1.707679271697998,
      "step": 19973
    },
    {
      "epoch": 1.904773712158203e-05,
      "step": 19973,
      "training_step_time": 10.051645040512085
    },
    {
      "epoch": 1.9048690795898438e-05,
      "model_forward_time": 1.726766586303711,
      "step": 19974
    },
    {
      "epoch": 1.9048690795898438e-05,
      "step": 19974,
      "training_step_time": 7.717550754547119
    },
    {
      "epoch": 1.9049644470214845e-05,
      "model_forward_time": 1.7279999256134033,
      "step": 19975
    },
    {
      "epoch": 1.9049644470214845e-05,
      "step": 19975,
      "training_step_time": 7.355967283248901
    },
    {
      "epoch": 1.905059814453125e-05,
      "model_forward_time": 1.7323381900787354,
      "step": 19976
    },
    {
      "epoch": 1.905059814453125e-05,
      "step": 19976,
      "training_step_time": 7.37469220161438
    },
    {
      "epoch": 1.9051551818847657e-05,
      "model_forward_time": 1.7364144325256348,
      "step": 19977
    },
    {
      "epoch": 1.9051551818847657e-05,
      "step": 19977,
      "training_step_time": 7.495477199554443
    },
    {
      "epoch": 1.9052505493164064e-05,
      "model_forward_time": 1.7308886051177979,
      "step": 19978
    },
    {
      "epoch": 1.9052505493164064e-05,
      "step": 19978,
      "training_step_time": 7.528273582458496
    },
    {
      "epoch": 1.9053459167480468e-05,
      "model_forward_time": 1.7362029552459717,
      "step": 19979
    },
    {
      "epoch": 1.9053459167480468e-05,
      "step": 19979,
      "training_step_time": 8.484419345855713
    },
    {
      "epoch": 1.9054412841796875e-05,
      "grad_norm": 0.18693412840366364,
      "learning_rate": 9.399020321301295e-06,
      "loss": 0.0654,
      "step": 19980
    },
    {
      "epoch": 1.9054412841796875e-05,
      "model_forward_time": 1.742495059967041,
      "step": 19980
    },
    {
      "dynamics_loss_avg": 0.046429593116045,
      "epoch": 1.9054412841796875e-05,
      "step": 19980
    },
    {
      "action_loss_avg": 0.0017694456153549254,
      "epoch": 1.9054412841796875e-05,
      "step": 19980
    },
    {
      "epoch": 1.9054412841796875e-05,
      "step": 19980,
      "training_step_time": 7.851714134216309
    },
    {
      "epoch": 1.9055366516113283e-05,
      "model_forward_time": 2.9002132415771484,
      "step": 19981
    },
    {
      "epoch": 1.9055366516113283e-05,
      "step": 19981,
      "training_step_time": 9.622743368148804
    },
    {
      "epoch": 1.9056320190429687e-05,
      "model_forward_time": 1.7261261940002441,
      "step": 19982
    },
    {
      "epoch": 1.9056320190429687e-05,
      "step": 19982,
      "training_step_time": 9.609440565109253
    },
    {
      "epoch": 1.9057273864746094e-05,
      "model_forward_time": 1.7396810054779053,
      "step": 19983
    },
    {
      "epoch": 1.9057273864746094e-05,
      "step": 19983,
      "training_step_time": 7.693575859069824
    },
    {
      "epoch": 1.90582275390625e-05,
      "model_forward_time": 1.7288167476654053,
      "step": 19984
    },
    {
      "epoch": 1.90582275390625e-05,
      "step": 19984,
      "training_step_time": 7.776886940002441
    },
    {
      "epoch": 1.9059181213378906e-05,
      "model_forward_time": 1.730738878250122,
      "step": 19985
    },
    {
      "epoch": 1.9059181213378906e-05,
      "step": 19985,
      "training_step_time": 7.204765796661377
    },
    {
      "epoch": 1.9060134887695313e-05,
      "model_forward_time": 1.728553056716919,
      "step": 19986
    },
    {
      "epoch": 1.9060134887695313e-05,
      "step": 19986,
      "training_step_time": 7.476697683334351
    },
    {
      "epoch": 1.9061088562011717e-05,
      "model_forward_time": 1.7373950481414795,
      "step": 19987
    },
    {
      "epoch": 1.9061088562011717e-05,
      "step": 19987,
      "training_step_time": 7.1287243366241455
    },
    {
      "epoch": 1.9062042236328125e-05,
      "model_forward_time": 1.729626178741455,
      "step": 19988
    },
    {
      "epoch": 1.9062042236328125e-05,
      "step": 19988,
      "training_step_time": 8.37912917137146
    },
    {
      "epoch": 1.9062995910644532e-05,
      "model_forward_time": 1.7493484020233154,
      "step": 19989
    },
    {
      "epoch": 1.9062995910644532e-05,
      "step": 19989,
      "training_step_time": 7.808988332748413
    },
    {
      "epoch": 1.9063949584960936e-05,
      "grad_norm": 0.12221518158912659,
      "learning_rate": 9.398234126189234e-06,
      "loss": 0.0641,
      "step": 19990
    },
    {
      "epoch": 1.9063949584960936e-05,
      "model_forward_time": 2.8712706565856934,
      "step": 19990
    },
    {
      "dynamics_loss_avg": 0.059532472491264345,
      "epoch": 1.9063949584960936e-05,
      "step": 19990
    },
    {
      "action_loss_avg": 0.0034087148145772518,
      "epoch": 1.9063949584960936e-05,
      "step": 19990
    },
    {
      "epoch": 1.9063949584960936e-05,
      "step": 19990,
      "training_step_time": 8.241595983505249
    },
    {
      "epoch": 1.9064903259277343e-05,
      "model_forward_time": 1.721740484237671,
      "step": 19991
    },
    {
      "epoch": 1.9064903259277343e-05,
      "step": 19991,
      "training_step_time": 7.651907682418823
    },
    {
      "epoch": 1.906585693359375e-05,
      "model_forward_time": 1.7334370613098145,
      "step": 19992
    },
    {
      "epoch": 1.906585693359375e-05,
      "step": 19992,
      "training_step_time": 8.239988803863525
    },
    {
      "epoch": 1.9066810607910155e-05,
      "model_forward_time": 1.7374980449676514,
      "step": 19993
    },
    {
      "epoch": 1.9066810607910155e-05,
      "step": 19993,
      "training_step_time": 7.754972219467163
    },
    {
      "epoch": 1.9067764282226562e-05,
      "model_forward_time": 1.729363203048706,
      "step": 19994
    },
    {
      "epoch": 1.9067764282226562e-05,
      "step": 19994,
      "training_step_time": 7.155263662338257
    },
    {
      "epoch": 1.906871795654297e-05,
      "model_forward_time": 1.72735595703125,
      "step": 19995
    },
    {
      "epoch": 1.906871795654297e-05,
      "step": 19995,
      "training_step_time": 8.7982759475708
    },
    {
      "epoch": 1.9069671630859374e-05,
      "model_forward_time": 1.722625494003296,
      "step": 19996
    },
    {
      "epoch": 1.9069671630859374e-05,
      "step": 19996,
      "training_step_time": 7.131258487701416
    },
    {
      "epoch": 1.907062530517578e-05,
      "model_forward_time": 1.7300550937652588,
      "step": 19997
    },
    {
      "epoch": 1.907062530517578e-05,
      "step": 19997,
      "training_step_time": 8.175215005874634
    },
    {
      "epoch": 1.907157897949219e-05,
      "model_forward_time": 1.7519161701202393,
      "step": 19998
    },
    {
      "epoch": 1.907157897949219e-05,
      "step": 19998,
      "training_step_time": 8.07729721069336
    },
    {
      "epoch": 1.9072532653808593e-05,
      "model_forward_time": 1.4240751266479492,
      "step": 19999
    },
    {
      "epoch": 1.9072532653808593e-05,
      "step": 19999,
      "training_step_time": 8.577067613601685
    },
    {
      "epoch": 1.9073486328125e-05,
      "grad_norm": 0.19105027616024017,
      "learning_rate": 9.397447450092903e-06,
      "loss": 0.0607,
      "step": 20000
    },
    {
      "epoch": 1.9073486328125e-05,
      "model_forward_time": 1.6975629329681396,
      "step": 20000
    },
    {
      "dynamics_loss_avg": 0.046721866726875304,
      "epoch": 1.9073486328125e-05,
      "step": 20000
    },
    {
      "action_loss_avg": 0.003294336359249428,
      "epoch": 1.9073486328125e-05,
      "step": 20000
    },
    {
      "epoch": 1.9073486328125e-05,
      "step": 20000,
      "training_step_time": 8.934095859527588
    },
    {
      "epoch": 1.9074440002441407e-05,
      "model_forward_time": 1.70664644241333,
      "step": 20001
    },
    {
      "epoch": 1.9074440002441407e-05,
      "step": 20001,
      "training_step_time": 9.085370063781738
    },
    {
      "epoch": 1.907539367675781e-05,
      "model_forward_time": 1.7119309902191162,
      "step": 20002
    },
    {
      "epoch": 1.907539367675781e-05,
      "step": 20002,
      "training_step_time": 8.706406116485596
    },
    {
      "epoch": 1.907634735107422e-05,
      "model_forward_time": 1.7101848125457764,
      "step": 20003
    },
    {
      "epoch": 1.907634735107422e-05,
      "step": 20003,
      "training_step_time": 7.55666184425354
    },
    {
      "epoch": 1.9077301025390626e-05,
      "model_forward_time": 1.7055439949035645,
      "step": 20004
    },
    {
      "epoch": 1.9077301025390626e-05,
      "step": 20004,
      "training_step_time": 7.97736668586731
    },
    {
      "epoch": 1.907825469970703e-05,
      "model_forward_time": 1.7098865509033203,
      "step": 20005
    },
    {
      "epoch": 1.907825469970703e-05,
      "step": 20005,
      "training_step_time": 8.122051000595093
    },
    {
      "epoch": 1.9079208374023438e-05,
      "model_forward_time": 1.7068543434143066,
      "step": 20006
    },
    {
      "epoch": 1.9079208374023438e-05,
      "step": 20006,
      "training_step_time": 7.583092212677002
    },
    {
      "epoch": 1.9080162048339845e-05,
      "model_forward_time": 1.714402437210083,
      "step": 20007
    },
    {
      "epoch": 1.9080162048339845e-05,
      "step": 20007,
      "training_step_time": 9.343111991882324
    },
    {
      "epoch": 1.908111572265625e-05,
      "model_forward_time": 2.7716782093048096,
      "step": 20008
    },
    {
      "epoch": 1.908111572265625e-05,
      "step": 20008,
      "training_step_time": 10.389485836029053
    },
    {
      "epoch": 1.9082069396972657e-05,
      "model_forward_time": 1.7216365337371826,
      "step": 20009
    },
    {
      "epoch": 1.9082069396972657e-05,
      "step": 20009,
      "training_step_time": 8.320723056793213
    },
    {
      "epoch": 1.9083023071289064e-05,
      "grad_norm": 0.2741590440273285,
      "learning_rate": 9.39666029309833e-06,
      "loss": 0.0634,
      "step": 20010
    },
    {
      "epoch": 1.9083023071289064e-05,
      "model_forward_time": 1.7239329814910889,
      "step": 20010
    },
    {
      "dynamics_loss_avg": 0.06859682742506265,
      "epoch": 1.9083023071289064e-05,
      "step": 20010
    },
    {
      "action_loss_avg": 0.0025675607437733563,
      "epoch": 1.9083023071289064e-05,
      "step": 20010
    },
    {
      "epoch": 1.9083023071289064e-05,
      "step": 20010,
      "training_step_time": 7.7344324588775635
    },
    {
      "epoch": 1.9083976745605468e-05,
      "model_forward_time": 1.724632978439331,
      "step": 20011
    },
    {
      "epoch": 1.9083976745605468e-05,
      "step": 20011,
      "training_step_time": 7.60720682144165
    },
    {
      "epoch": 1.9084930419921875e-05,
      "model_forward_time": 1.728715181350708,
      "step": 20012
    },
    {
      "epoch": 1.9084930419921875e-05,
      "step": 20012,
      "training_step_time": 7.579553604125977
    },
    {
      "epoch": 1.9085884094238283e-05,
      "model_forward_time": 1.7294034957885742,
      "step": 20013
    },
    {
      "epoch": 1.9085884094238283e-05,
      "step": 20013,
      "training_step_time": 7.434936046600342
    },
    {
      "epoch": 1.9086837768554687e-05,
      "model_forward_time": 1.7273039817810059,
      "step": 20014
    },
    {
      "epoch": 1.9086837768554687e-05,
      "step": 20014,
      "training_step_time": 7.364276170730591
    },
    {
      "epoch": 1.9087791442871094e-05,
      "model_forward_time": 1.3629167079925537,
      "step": 20015
    },
    {
      "epoch": 1.9087791442871094e-05,
      "step": 20015,
      "training_step_time": 8.591150999069214
    },
    {
      "epoch": 1.90887451171875e-05,
      "model_forward_time": 1.7503559589385986,
      "step": 20016
    },
    {
      "epoch": 1.90887451171875e-05,
      "step": 20016,
      "training_step_time": 8.140850305557251
    },
    {
      "epoch": 1.9089698791503906e-05,
      "model_forward_time": 1.7265620231628418,
      "step": 20017
    },
    {
      "epoch": 1.9089698791503906e-05,
      "step": 20017,
      "training_step_time": 9.174577474594116
    },
    {
      "epoch": 1.9090652465820313e-05,
      "model_forward_time": 1.343113899230957,
      "step": 20018
    },
    {
      "epoch": 1.9090652465820313e-05,
      "step": 20018,
      "training_step_time": 7.582426071166992
    },
    {
      "epoch": 1.9091606140136717e-05,
      "model_forward_time": 1.728785753250122,
      "step": 20019
    },
    {
      "epoch": 1.9091606140136717e-05,
      "step": 20019,
      "training_step_time": 7.621914625167847
    },
    {
      "epoch": 1.9092559814453125e-05,
      "grad_norm": 0.1576111614704132,
      "learning_rate": 9.395872655291596e-06,
      "loss": 0.0656,
      "step": 20020
    },
    {
      "epoch": 1.9092559814453125e-05,
      "model_forward_time": 1.7192668914794922,
      "step": 20020
    },
    {
      "dynamics_loss_avg": 0.052776505891233685,
      "epoch": 1.9092559814453125e-05,
      "step": 20020
    },
    {
      "action_loss_avg": 0.0024366232682950795,
      "epoch": 1.9092559814453125e-05,
      "step": 20020
    },
    {
      "epoch": 1.9092559814453125e-05,
      "step": 20020,
      "training_step_time": 7.609771966934204
    },
    {
      "epoch": 1.9093513488769532e-05,
      "model_forward_time": 1.727004051208496,
      "step": 20021
    },
    {
      "epoch": 1.9093513488769532e-05,
      "step": 20021,
      "training_step_time": 7.613198280334473
    },
    {
      "epoch": 1.9094467163085936e-05,
      "model_forward_time": 1.7370874881744385,
      "step": 20022
    },
    {
      "epoch": 1.9094467163085936e-05,
      "step": 20022,
      "training_step_time": 7.427313327789307
    },
    {
      "epoch": 1.9095420837402343e-05,
      "model_forward_time": 1.7252907752990723,
      "step": 20023
    },
    {
      "epoch": 1.9095420837402343e-05,
      "step": 20023,
      "training_step_time": 7.256376028060913
    },
    {
      "epoch": 1.909637451171875e-05,
      "model_forward_time": 1.7263405323028564,
      "step": 20024
    },
    {
      "epoch": 1.909637451171875e-05,
      "step": 20024,
      "training_step_time": 7.704808473587036
    },
    {
      "epoch": 1.9097328186035155e-05,
      "model_forward_time": 1.7267396450042725,
      "step": 20025
    },
    {
      "epoch": 1.9097328186035155e-05,
      "step": 20025,
      "training_step_time": 8.269256830215454
    },
    {
      "epoch": 1.9098281860351562e-05,
      "model_forward_time": 1.72996187210083,
      "step": 20026
    },
    {
      "epoch": 1.9098281860351562e-05,
      "step": 20026,
      "training_step_time": 7.751067399978638
    },
    {
      "epoch": 1.909923553466797e-05,
      "model_forward_time": 2.260256767272949,
      "step": 20027
    },
    {
      "epoch": 1.909923553466797e-05,
      "step": 20027,
      "training_step_time": 8.136678695678711
    },
    {
      "epoch": 1.9100189208984374e-05,
      "model_forward_time": 1.3553321361541748,
      "step": 20028
    },
    {
      "epoch": 1.9100189208984374e-05,
      "step": 20028,
      "training_step_time": 8.41134524345398
    },
    {
      "epoch": 1.910114288330078e-05,
      "model_forward_time": 1.724924087524414,
      "step": 20029
    },
    {
      "epoch": 1.910114288330078e-05,
      "step": 20029,
      "training_step_time": 9.728418111801147
    },
    {
      "epoch": 1.910209655761719e-05,
      "grad_norm": 0.17707616090774536,
      "learning_rate": 9.395084536758838e-06,
      "loss": 0.064,
      "step": 20030
    },
    {
      "epoch": 1.910209655761719e-05,
      "model_forward_time": 1.7221882343292236,
      "step": 20030
    },
    {
      "dynamics_loss_avg": 0.06190889216959476,
      "epoch": 1.910209655761719e-05,
      "step": 20030
    },
    {
      "action_loss_avg": 0.002451074542477727,
      "epoch": 1.910209655761719e-05,
      "step": 20030
    },
    {
      "epoch": 1.910209655761719e-05,
      "step": 20030,
      "training_step_time": 7.631049633026123
    },
    {
      "epoch": 1.9103050231933593e-05,
      "model_forward_time": 1.725468635559082,
      "step": 20031
    },
    {
      "epoch": 1.9103050231933593e-05,
      "step": 20031,
      "training_step_time": 8.207290887832642
    },
    {
      "epoch": 1.910400390625e-05,
      "model_forward_time": 1.728508710861206,
      "step": 20032
    },
    {
      "epoch": 1.910400390625e-05,
      "step": 20032,
      "training_step_time": 7.337879657745361
    },
    {
      "epoch": 1.9104957580566407e-05,
      "model_forward_time": 1.726146936416626,
      "step": 20033
    },
    {
      "epoch": 1.9104957580566407e-05,
      "step": 20033,
      "training_step_time": 7.69591212272644
    },
    {
      "epoch": 1.910591125488281e-05,
      "model_forward_time": 1.7306647300720215,
      "step": 20034
    },
    {
      "epoch": 1.910591125488281e-05,
      "step": 20034,
      "training_step_time": 7.567823171615601
    },
    {
      "epoch": 1.910686492919922e-05,
      "model_forward_time": 1.7688589096069336,
      "step": 20035
    },
    {
      "epoch": 1.910686492919922e-05,
      "step": 20035,
      "training_step_time": 7.782325744628906
    },
    {
      "epoch": 1.9107818603515626e-05,
      "model_forward_time": 2.2586283683776855,
      "step": 20036
    },
    {
      "epoch": 1.9107818603515626e-05,
      "step": 20036,
      "training_step_time": 8.640903949737549
    },
    {
      "epoch": 1.910877227783203e-05,
      "model_forward_time": 1.7279887199401855,
      "step": 20037
    },
    {
      "epoch": 1.910877227783203e-05,
      "step": 20037,
      "training_step_time": 9.172591924667358
    },
    {
      "epoch": 1.9109725952148438e-05,
      "model_forward_time": 1.7270886898040771,
      "step": 20038
    },
    {
      "epoch": 1.9109725952148438e-05,
      "step": 20038,
      "training_step_time": 7.647520303726196
    },
    {
      "epoch": 1.9110679626464845e-05,
      "model_forward_time": 1.7257044315338135,
      "step": 20039
    },
    {
      "epoch": 1.9110679626464845e-05,
      "step": 20039,
      "training_step_time": 7.656277179718018
    },
    {
      "epoch": 1.911163330078125e-05,
      "grad_norm": 0.19720110297203064,
      "learning_rate": 9.394295937586243e-06,
      "loss": 0.0627,
      "step": 20040
    },
    {
      "epoch": 1.911163330078125e-05,
      "model_forward_time": 1.7247123718261719,
      "step": 20040
    },
    {
      "dynamics_loss_avg": 0.06227097120136023,
      "epoch": 1.911163330078125e-05,
      "step": 20040
    },
    {
      "action_loss_avg": 0.008298144722357393,
      "epoch": 1.911163330078125e-05,
      "step": 20040
    },
    {
      "epoch": 1.911163330078125e-05,
      "step": 20040,
      "training_step_time": 7.484795093536377
    },
    {
      "epoch": 1.9112586975097657e-05,
      "model_forward_time": 1.72487211227417,
      "step": 20041
    },
    {
      "epoch": 1.9112586975097657e-05,
      "step": 20041,
      "training_step_time": 7.384583473205566
    },
    {
      "epoch": 1.9113540649414064e-05,
      "model_forward_time": 1.7347064018249512,
      "step": 20042
    },
    {
      "epoch": 1.9113540649414064e-05,
      "step": 20042,
      "training_step_time": 8.100130796432495
    },
    {
      "epoch": 1.9114494323730468e-05,
      "model_forward_time": 1.729877233505249,
      "step": 20043
    },
    {
      "epoch": 1.9114494323730468e-05,
      "step": 20043,
      "training_step_time": 7.655952453613281
    },
    {
      "epoch": 1.9115447998046875e-05,
      "model_forward_time": 1.7648346424102783,
      "step": 20044
    },
    {
      "epoch": 1.9115447998046875e-05,
      "step": 20044,
      "training_step_time": 9.70075511932373
    },
    {
      "epoch": 1.9116401672363283e-05,
      "model_forward_time": 2.2672934532165527,
      "step": 20045
    },
    {
      "epoch": 1.9116401672363283e-05,
      "step": 20045,
      "training_step_time": 9.249889135360718
    },
    {
      "epoch": 1.9117355346679687e-05,
      "model_forward_time": 1.3578236103057861,
      "step": 20046
    },
    {
      "epoch": 1.9117355346679687e-05,
      "step": 20046,
      "training_step_time": 9.162243604660034
    },
    {
      "epoch": 1.9118309020996094e-05,
      "model_forward_time": 1.7286741733551025,
      "step": 20047
    },
    {
      "epoch": 1.9118309020996094e-05,
      "step": 20047,
      "training_step_time": 8.073991298675537
    },
    {
      "epoch": 1.91192626953125e-05,
      "model_forward_time": 1.7284467220306396,
      "step": 20048
    },
    {
      "epoch": 1.91192626953125e-05,
      "step": 20048,
      "training_step_time": 7.565100908279419
    },
    {
      "epoch": 1.9120216369628906e-05,
      "model_forward_time": 1.7300362586975098,
      "step": 20049
    },
    {
      "epoch": 1.9120216369628906e-05,
      "step": 20049,
      "training_step_time": 7.503929138183594
    },
    {
      "epoch": 1.9121170043945313e-05,
      "grad_norm": 0.1453435868024826,
      "learning_rate": 9.393506857860053e-06,
      "loss": 0.0674,
      "step": 20050
    },
    {
      "epoch": 1.9121170043945313e-05,
      "model_forward_time": 1.7279999256134033,
      "step": 20050
    },
    {
      "dynamics_loss_avg": 0.06529593952000141,
      "epoch": 1.9121170043945313e-05,
      "step": 20050
    },
    {
      "action_loss_avg": 0.002361201774328947,
      "epoch": 1.9121170043945313e-05,
      "step": 20050
    },
    {
      "epoch": 1.9121170043945313e-05,
      "step": 20050,
      "training_step_time": 7.356368541717529
    },
    {
      "epoch": 1.9122123718261717e-05,
      "model_forward_time": 1.7271606922149658,
      "step": 20051
    },
    {
      "epoch": 1.9122123718261717e-05,
      "step": 20051,
      "training_step_time": 7.496679306030273
    },
    {
      "epoch": 1.9123077392578125e-05,
      "model_forward_time": 1.73284912109375,
      "step": 20052
    },
    {
      "epoch": 1.9123077392578125e-05,
      "step": 20052,
      "training_step_time": 7.5383241176605225
    },
    {
      "epoch": 1.9124031066894532e-05,
      "model_forward_time": 1.7657554149627686,
      "step": 20053
    },
    {
      "epoch": 1.9124031066894532e-05,
      "step": 20053,
      "training_step_time": 7.757995843887329
    },
    {
      "epoch": 1.9124984741210936e-05,
      "model_forward_time": 2.2671966552734375,
      "step": 20054
    },
    {
      "epoch": 1.9124984741210936e-05,
      "step": 20054,
      "training_step_time": 9.357614278793335
    },
    {
      "epoch": 1.9125938415527343e-05,
      "model_forward_time": 1.7258284091949463,
      "step": 20055
    },
    {
      "epoch": 1.9125938415527343e-05,
      "step": 20055,
      "training_step_time": 7.5785973072052
    },
    {
      "epoch": 1.912689208984375e-05,
      "model_forward_time": 1.7225735187530518,
      "step": 20056
    },
    {
      "epoch": 1.912689208984375e-05,
      "step": 20056,
      "training_step_time": 8.359286069869995
    },
    {
      "epoch": 1.9127845764160155e-05,
      "model_forward_time": 1.7268903255462646,
      "step": 20057
    },
    {
      "epoch": 1.9127845764160155e-05,
      "step": 20057,
      "training_step_time": 7.5746400356292725
    },
    {
      "epoch": 1.9128799438476562e-05,
      "model_forward_time": 1.728379726409912,
      "step": 20058
    },
    {
      "epoch": 1.9128799438476562e-05,
      "step": 20058,
      "training_step_time": 7.552689075469971
    },
    {
      "epoch": 1.912975311279297e-05,
      "model_forward_time": 1.7248873710632324,
      "step": 20059
    },
    {
      "epoch": 1.912975311279297e-05,
      "step": 20059,
      "training_step_time": 7.358166694641113
    },
    {
      "epoch": 1.9130706787109374e-05,
      "grad_norm": 0.20365208387374878,
      "learning_rate": 9.392717297666557e-06,
      "loss": 0.0627,
      "step": 20060
    },
    {
      "epoch": 1.9130706787109374e-05,
      "model_forward_time": 1.72066330909729,
      "step": 20060
    },
    {
      "dynamics_loss_avg": 0.056655926443636415,
      "epoch": 1.9130706787109374e-05,
      "step": 20060
    },
    {
      "action_loss_avg": 0.0031494525144807996,
      "epoch": 1.9130706787109374e-05,
      "step": 20060
    },
    {
      "epoch": 1.9130706787109374e-05,
      "step": 20060,
      "training_step_time": 7.334601163864136
    },
    {
      "epoch": 1.913166046142578e-05,
      "model_forward_time": 1.7365670204162598,
      "step": 20061
    },
    {
      "epoch": 1.913166046142578e-05,
      "step": 20061,
      "training_step_time": 7.553377866744995
    },
    {
      "epoch": 1.913261413574219e-05,
      "model_forward_time": 1.3549275398254395,
      "step": 20062
    },
    {
      "epoch": 1.913261413574219e-05,
      "step": 20062,
      "training_step_time": 9.217876195907593
    },
    {
      "epoch": 1.9133567810058593e-05,
      "model_forward_time": 1.7268552780151367,
      "step": 20063
    },
    {
      "epoch": 1.9133567810058593e-05,
      "step": 20063,
      "training_step_time": 8.664525032043457
    },
    {
      "epoch": 1.9134521484375e-05,
      "model_forward_time": 2.2596142292022705,
      "step": 20064
    },
    {
      "epoch": 1.9134521484375e-05,
      "step": 20064,
      "training_step_time": 7.554752349853516
    },
    {
      "epoch": 1.9135475158691407e-05,
      "model_forward_time": 1.7338786125183105,
      "step": 20065
    },
    {
      "epoch": 1.9135475158691407e-05,
      "step": 20065,
      "training_step_time": 7.63503360748291
    },
    {
      "epoch": 1.913642883300781e-05,
      "model_forward_time": 1.3632214069366455,
      "step": 20066
    },
    {
      "epoch": 1.913642883300781e-05,
      "step": 20066,
      "training_step_time": 7.614352464675903
    },
    {
      "epoch": 1.913738250732422e-05,
      "model_forward_time": 1.7273740768432617,
      "step": 20067
    },
    {
      "epoch": 1.913738250732422e-05,
      "step": 20067,
      "training_step_time": 7.525931358337402
    },
    {
      "epoch": 1.9138336181640626e-05,
      "model_forward_time": 1.797879695892334,
      "step": 20068
    },
    {
      "epoch": 1.9138336181640626e-05,
      "step": 20068,
      "training_step_time": 7.394002437591553
    },
    {
      "epoch": 1.913928985595703e-05,
      "model_forward_time": 1.8165502548217773,
      "step": 20069
    },
    {
      "epoch": 1.913928985595703e-05,
      "step": 20069,
      "training_step_time": 7.376101970672607
    },
    {
      "epoch": 1.9140243530273438e-05,
      "grad_norm": 0.21605370938777924,
      "learning_rate": 9.391927257092102e-06,
      "loss": 0.0618,
      "step": 20070
    },
    {
      "epoch": 1.9140243530273438e-05,
      "model_forward_time": 1.8278470039367676,
      "step": 20070
    },
    {
      "dynamics_loss_avg": 0.05470913648605347,
      "epoch": 1.9140243530273438e-05,
      "step": 20070
    },
    {
      "action_loss_avg": 0.002834716648794711,
      "epoch": 1.9140243530273438e-05,
      "step": 20070
    },
    {
      "epoch": 1.9140243530273438e-05,
      "step": 20070,
      "training_step_time": 7.550338506698608
    },
    {
      "epoch": 1.9141197204589845e-05,
      "model_forward_time": 1.807830810546875,
      "step": 20071
    },
    {
      "epoch": 1.9141197204589845e-05,
      "step": 20071,
      "training_step_time": 8.22982144355774
    },
    {
      "epoch": 1.914215087890625e-05,
      "model_forward_time": 1.8289668560028076,
      "step": 20072
    },
    {
      "epoch": 1.914215087890625e-05,
      "step": 20072,
      "training_step_time": 8.150631189346313
    },
    {
      "epoch": 1.9143104553222657e-05,
      "model_forward_time": 2.3371574878692627,
      "step": 20073
    },
    {
      "epoch": 1.9143104553222657e-05,
      "step": 20073,
      "training_step_time": 8.646113157272339
    },
    {
      "epoch": 1.9144058227539064e-05,
      "model_forward_time": 1.8181869983673096,
      "step": 20074
    },
    {
      "epoch": 1.9144058227539064e-05,
      "step": 20074,
      "training_step_time": 7.662865877151489
    },
    {
      "epoch": 1.9145011901855468e-05,
      "model_forward_time": 1.828331470489502,
      "step": 20075
    },
    {
      "epoch": 1.9145011901855468e-05,
      "step": 20075,
      "training_step_time": 7.609793424606323
    },
    {
      "epoch": 1.9145965576171875e-05,
      "model_forward_time": 1.8271074295043945,
      "step": 20076
    },
    {
      "epoch": 1.9145965576171875e-05,
      "step": 20076,
      "training_step_time": 7.725565195083618
    },
    {
      "epoch": 1.9146919250488283e-05,
      "model_forward_time": 1.4589686393737793,
      "step": 20077
    },
    {
      "epoch": 1.9146919250488283e-05,
      "step": 20077,
      "training_step_time": 7.3235557079315186
    },
    {
      "epoch": 1.9147872924804687e-05,
      "model_forward_time": 1.824601650238037,
      "step": 20078
    },
    {
      "epoch": 1.9147872924804687e-05,
      "step": 20078,
      "training_step_time": 7.371671438217163
    },
    {
      "epoch": 1.9148826599121094e-05,
      "model_forward_time": 1.824108600616455,
      "step": 20079
    },
    {
      "epoch": 1.9148826599121094e-05,
      "step": 20079,
      "training_step_time": 7.565427541732788
    },
    {
      "epoch": 1.91497802734375e-05,
      "grad_norm": 0.20815186202526093,
      "learning_rate": 9.391136736223085e-06,
      "loss": 0.064,
      "step": 20080
    },
    {
      "epoch": 1.91497802734375e-05,
      "model_forward_time": 1.81746506690979,
      "step": 20080
    },
    {
      "dynamics_loss_avg": 0.05821414105594158,
      "epoch": 1.91497802734375e-05,
      "step": 20080
    },
    {
      "action_loss_avg": 0.0027090895979199558,
      "epoch": 1.91497802734375e-05,
      "step": 20080
    },
    {
      "epoch": 1.91497802734375e-05,
      "step": 20080,
      "training_step_time": 7.6164422035217285
    },
    {
      "epoch": 1.9150733947753906e-05,
      "model_forward_time": 1.8243286609649658,
      "step": 20081
    },
    {
      "epoch": 1.9150733947753906e-05,
      "step": 20081,
      "training_step_time": 8.597508907318115
    },
    {
      "epoch": 1.9151687622070313e-05,
      "model_forward_time": 2.328274726867676,
      "step": 20082
    },
    {
      "epoch": 1.9151687622070313e-05,
      "step": 20082,
      "training_step_time": 8.360184907913208
    },
    {
      "epoch": 1.9152641296386717e-05,
      "model_forward_time": 1.8220264911651611,
      "step": 20083
    },
    {
      "epoch": 1.9152641296386717e-05,
      "step": 20083,
      "training_step_time": 8.22046446800232
    },
    {
      "epoch": 1.9153594970703125e-05,
      "model_forward_time": 1.8176062107086182,
      "step": 20084
    },
    {
      "epoch": 1.9153594970703125e-05,
      "step": 20084,
      "training_step_time": 8.313808679580688
    },
    {
      "epoch": 1.9154548645019532e-05,
      "model_forward_time": 1.8153293132781982,
      "step": 20085
    },
    {
      "epoch": 1.9154548645019532e-05,
      "step": 20085,
      "training_step_time": 8.476355075836182
    },
    {
      "epoch": 1.9155502319335936e-05,
      "model_forward_time": 1.8298962116241455,
      "step": 20086
    },
    {
      "epoch": 1.9155502319335936e-05,
      "step": 20086,
      "training_step_time": 7.656248569488525
    },
    {
      "epoch": 1.9156455993652344e-05,
      "model_forward_time": 1.8230178356170654,
      "step": 20087
    },
    {
      "epoch": 1.9156455993652344e-05,
      "step": 20087,
      "training_step_time": 7.413518905639648
    },
    {
      "epoch": 1.915740966796875e-05,
      "model_forward_time": 1.8217720985412598,
      "step": 20088
    },
    {
      "epoch": 1.915740966796875e-05,
      "step": 20088,
      "training_step_time": 8.1155264377594
    },
    {
      "epoch": 1.9158363342285155e-05,
      "model_forward_time": 1.8172094821929932,
      "step": 20089
    },
    {
      "epoch": 1.9158363342285155e-05,
      "step": 20089,
      "training_step_time": 7.739267110824585
    },
    {
      "epoch": 1.9159317016601562e-05,
      "grad_norm": 0.1865670084953308,
      "learning_rate": 9.390345735145957e-06,
      "loss": 0.0662,
      "step": 20090
    },
    {
      "epoch": 1.9159317016601562e-05,
      "model_forward_time": 1.823199987411499,
      "step": 20090
    },
    {
      "dynamics_loss_avg": 0.06343797668814659,
      "epoch": 1.9159317016601562e-05,
      "step": 20090
    },
    {
      "action_loss_avg": 0.0040595699334517125,
      "epoch": 1.9159317016601562e-05,
      "step": 20090
    },
    {
      "epoch": 1.9159317016601562e-05,
      "step": 20090,
      "training_step_time": 8.650952816009521
    },
    {
      "epoch": 1.916027069091797e-05,
      "model_forward_time": 1.8201584815979004,
      "step": 20091
    },
    {
      "epoch": 1.916027069091797e-05,
      "step": 20091,
      "training_step_time": 8.504653453826904
    },
    {
      "epoch": 1.9161224365234374e-05,
      "model_forward_time": 2.3617498874664307,
      "step": 20092
    },
    {
      "epoch": 1.9161224365234374e-05,
      "step": 20092,
      "training_step_time": 8.203248262405396
    },
    {
      "epoch": 1.916217803955078e-05,
      "model_forward_time": 1.8165881633758545,
      "step": 20093
    },
    {
      "epoch": 1.916217803955078e-05,
      "step": 20093,
      "training_step_time": 7.719446897506714
    },
    {
      "epoch": 1.916313171386719e-05,
      "model_forward_time": 1.8331842422485352,
      "step": 20094
    },
    {
      "epoch": 1.916313171386719e-05,
      "step": 20094,
      "training_step_time": 7.688982009887695
    },
    {
      "epoch": 1.9164085388183593e-05,
      "model_forward_time": 1.4468533992767334,
      "step": 20095
    },
    {
      "epoch": 1.9164085388183593e-05,
      "step": 20095,
      "training_step_time": 7.390599012374878
    },
    {
      "epoch": 1.91650390625e-05,
      "model_forward_time": 1.8125665187835693,
      "step": 20096
    },
    {
      "epoch": 1.91650390625e-05,
      "step": 20096,
      "training_step_time": 7.501583576202393
    },
    {
      "epoch": 1.9165992736816407e-05,
      "model_forward_time": 1.8189961910247803,
      "step": 20097
    },
    {
      "epoch": 1.9165992736816407e-05,
      "step": 20097,
      "training_step_time": 7.410446405410767
    },
    {
      "epoch": 1.916694641113281e-05,
      "model_forward_time": 1.820741891860962,
      "step": 20098
    },
    {
      "epoch": 1.916694641113281e-05,
      "step": 20098,
      "training_step_time": 7.792389631271362
    },
    {
      "epoch": 1.916790008544922e-05,
      "model_forward_time": 1.8116929531097412,
      "step": 20099
    },
    {
      "epoch": 1.916790008544922e-05,
      "step": 20099,
      "training_step_time": 7.74864649772644
    },
    {
      "epoch": 1.9168853759765626e-05,
      "grad_norm": 0.22307060658931732,
      "learning_rate": 9.38955425394722e-06,
      "loss": 0.0663,
      "step": 20100
    },
    {
      "epoch": 1.9168853759765626e-05,
      "model_forward_time": 1.8159399032592773,
      "step": 20100
    },
    {
      "dynamics_loss_avg": 0.05690539181232453,
      "epoch": 1.9168853759765626e-05,
      "step": 20100
    },
    {
      "action_loss_avg": 0.002472813124768436,
      "epoch": 1.9168853759765626e-05,
      "step": 20100
    },
    {
      "epoch": 1.9168853759765626e-05,
      "step": 20100,
      "training_step_time": 8.696717977523804
    },
    {
      "epoch": 1.916980743408203e-05,
      "model_forward_time": 2.321460485458374,
      "step": 20101
    },
    {
      "epoch": 1.916980743408203e-05,
      "step": 20101,
      "training_step_time": 8.623738765716553
    },
    {
      "epoch": 1.9170761108398438e-05,
      "model_forward_time": 1.8181328773498535,
      "step": 20102
    },
    {
      "epoch": 1.9170761108398438e-05,
      "step": 20102,
      "training_step_time": 9.819942474365234
    },
    {
      "epoch": 1.9171714782714845e-05,
      "model_forward_time": 1.8192009925842285,
      "step": 20103
    },
    {
      "epoch": 1.9171714782714845e-05,
      "step": 20103,
      "training_step_time": 9.144973516464233
    },
    {
      "epoch": 1.917266845703125e-05,
      "model_forward_time": 1.821424961090088,
      "step": 20104
    },
    {
      "epoch": 1.917266845703125e-05,
      "step": 20104,
      "training_step_time": 7.383897066116333
    },
    {
      "epoch": 1.9173622131347657e-05,
      "model_forward_time": 1.8157191276550293,
      "step": 20105
    },
    {
      "epoch": 1.9173622131347657e-05,
      "step": 20105,
      "training_step_time": 7.534081935882568
    },
    {
      "epoch": 1.9174575805664064e-05,
      "model_forward_time": 1.8097269535064697,
      "step": 20106
    },
    {
      "epoch": 1.9174575805664064e-05,
      "step": 20106,
      "training_step_time": 7.447410583496094
    },
    {
      "epoch": 1.9175529479980468e-05,
      "model_forward_time": 1.8138694763183594,
      "step": 20107
    },
    {
      "epoch": 1.9175529479980468e-05,
      "step": 20107,
      "training_step_time": 8.764162540435791
    },
    {
      "epoch": 1.9176483154296876e-05,
      "model_forward_time": 1.8165054321289062,
      "step": 20108
    },
    {
      "epoch": 1.9176483154296876e-05,
      "step": 20108,
      "training_step_time": 8.239116668701172
    },
    {
      "epoch": 1.9177436828613283e-05,
      "model_forward_time": 1.8015332221984863,
      "step": 20109
    },
    {
      "epoch": 1.9177436828613283e-05,
      "step": 20109,
      "training_step_time": 9.287344694137573
    },
    {
      "epoch": 1.9178390502929687e-05,
      "grad_norm": 0.19654424488544464,
      "learning_rate": 9.388762292713428e-06,
      "loss": 0.0654,
      "step": 20110
    },
    {
      "epoch": 1.9178390502929687e-05,
      "model_forward_time": 2.318446636199951,
      "step": 20110
    },
    {
      "dynamics_loss_avg": 0.0655632432550192,
      "epoch": 1.9178390502929687e-05,
      "step": 20110
    },
    {
      "action_loss_avg": 0.002674625988584012,
      "epoch": 1.9178390502929687e-05,
      "step": 20110
    },
    {
      "epoch": 1.9178390502929687e-05,
      "step": 20110,
      "training_step_time": 8.325981378555298
    },
    {
      "epoch": 1.9179344177246094e-05,
      "model_forward_time": 1.8113064765930176,
      "step": 20111
    },
    {
      "epoch": 1.9179344177246094e-05,
      "step": 20111,
      "training_step_time": 7.687256574630737
    },
    {
      "epoch": 1.91802978515625e-05,
      "model_forward_time": 1.8125038146972656,
      "step": 20112
    },
    {
      "epoch": 1.91802978515625e-05,
      "step": 20112,
      "training_step_time": 7.6726884841918945
    },
    {
      "epoch": 1.9181251525878906e-05,
      "model_forward_time": 1.8132765293121338,
      "step": 20113
    },
    {
      "epoch": 1.9181251525878906e-05,
      "step": 20113,
      "training_step_time": 7.721585035324097
    },
    {
      "epoch": 1.9182205200195313e-05,
      "model_forward_time": 1.82012939453125,
      "step": 20114
    },
    {
      "epoch": 1.9182205200195313e-05,
      "step": 20114,
      "training_step_time": 7.501389265060425
    },
    {
      "epoch": 1.9183158874511717e-05,
      "model_forward_time": 1.8145256042480469,
      "step": 20115
    },
    {
      "epoch": 1.9183158874511717e-05,
      "step": 20115,
      "training_step_time": 8.150413036346436
    },
    {
      "epoch": 1.9184112548828125e-05,
      "model_forward_time": 1.817169189453125,
      "step": 20116
    },
    {
      "epoch": 1.9184112548828125e-05,
      "step": 20116,
      "training_step_time": 8.677863836288452
    },
    {
      "epoch": 1.9185066223144532e-05,
      "model_forward_time": 1.8031179904937744,
      "step": 20117
    },
    {
      "epoch": 1.9185066223144532e-05,
      "step": 20117,
      "training_step_time": 7.7235026359558105
    },
    {
      "epoch": 1.9186019897460936e-05,
      "model_forward_time": 1.8150219917297363,
      "step": 20118
    },
    {
      "epoch": 1.9186019897460936e-05,
      "step": 20118,
      "training_step_time": 8.716622591018677
    },
    {
      "epoch": 1.9186973571777344e-05,
      "model_forward_time": 2.329481363296509,
      "step": 20119
    },
    {
      "epoch": 1.9186973571777344e-05,
      "step": 20119,
      "training_step_time": 9.214211702346802
    },
    {
      "epoch": 1.918792724609375e-05,
      "grad_norm": 0.1380593329668045,
      "learning_rate": 9.387969851531191e-06,
      "loss": 0.0682,
      "step": 20120
    },
    {
      "epoch": 1.918792724609375e-05,
      "model_forward_time": 1.800335168838501,
      "step": 20120
    },
    {
      "dynamics_loss_avg": 0.0524034284055233,
      "epoch": 1.918792724609375e-05,
      "step": 20120
    },
    {
      "action_loss_avg": 0.0029447266133502126,
      "epoch": 1.918792724609375e-05,
      "step": 20120
    },
    {
      "epoch": 1.918792724609375e-05,
      "step": 20120,
      "training_step_time": 7.756521224975586
    },
    {
      "epoch": 1.9188880920410155e-05,
      "model_forward_time": 1.8149232864379883,
      "step": 20121
    },
    {
      "epoch": 1.9188880920410155e-05,
      "step": 20121,
      "training_step_time": 7.695685625076294
    },
    {
      "epoch": 1.9189834594726562e-05,
      "model_forward_time": 1.8182337284088135,
      "step": 20122
    },
    {
      "epoch": 1.9189834594726562e-05,
      "step": 20122,
      "training_step_time": 7.68737006187439
    },
    {
      "epoch": 1.919078826904297e-05,
      "model_forward_time": 1.8066763877868652,
      "step": 20123
    },
    {
      "epoch": 1.919078826904297e-05,
      "step": 20123,
      "training_step_time": 7.528422117233276
    },
    {
      "epoch": 1.9191741943359374e-05,
      "model_forward_time": 1.8040893077850342,
      "step": 20124
    },
    {
      "epoch": 1.9191741943359374e-05,
      "step": 20124,
      "training_step_time": 7.420467853546143
    },
    {
      "epoch": 1.919269561767578e-05,
      "model_forward_time": 1.809729814529419,
      "step": 20125
    },
    {
      "epoch": 1.919269561767578e-05,
      "step": 20125,
      "training_step_time": 8.017333269119263
    },
    {
      "epoch": 1.919364929199219e-05,
      "model_forward_time": 1.8130433559417725,
      "step": 20126
    },
    {
      "epoch": 1.919364929199219e-05,
      "step": 20126,
      "training_step_time": 8.697645425796509
    },
    {
      "epoch": 1.9194602966308593e-05,
      "model_forward_time": 1.8057434558868408,
      "step": 20127
    },
    {
      "epoch": 1.9194602966308593e-05,
      "step": 20127,
      "training_step_time": 7.682617664337158
    },
    {
      "epoch": 1.9195556640625e-05,
      "model_forward_time": 2.3563973903656006,
      "step": 20128
    },
    {
      "epoch": 1.9195556640625e-05,
      "step": 20128,
      "training_step_time": 8.416953802108765
    },
    {
      "epoch": 1.9196510314941408e-05,
      "model_forward_time": 1.8138422966003418,
      "step": 20129
    },
    {
      "epoch": 1.9196510314941408e-05,
      "step": 20129,
      "training_step_time": 8.480969190597534
    },
    {
      "epoch": 1.919746398925781e-05,
      "grad_norm": 0.15245573222637177,
      "learning_rate": 9.38717693048717e-06,
      "loss": 0.0686,
      "step": 20130
    },
    {
      "epoch": 1.919746398925781e-05,
      "model_forward_time": 1.8104112148284912,
      "step": 20130
    },
    {
      "dynamics_loss_avg": 0.05726869776844978,
      "epoch": 1.919746398925781e-05,
      "step": 20130
    },
    {
      "action_loss_avg": 0.0034489025361835956,
      "epoch": 1.919746398925781e-05,
      "step": 20130
    },
    {
      "epoch": 1.919746398925781e-05,
      "step": 20130,
      "training_step_time": 7.6546549797058105
    },
    {
      "epoch": 1.919841766357422e-05,
      "model_forward_time": 1.8097009658813477,
      "step": 20131
    },
    {
      "epoch": 1.919841766357422e-05,
      "step": 20131,
      "training_step_time": 7.667215347290039
    },
    {
      "epoch": 1.9199371337890626e-05,
      "model_forward_time": 1.802454948425293,
      "step": 20132
    },
    {
      "epoch": 1.9199371337890626e-05,
      "step": 20132,
      "training_step_time": 7.392826080322266
    },
    {
      "epoch": 1.920032501220703e-05,
      "model_forward_time": 1.8178198337554932,
      "step": 20133
    },
    {
      "epoch": 1.920032501220703e-05,
      "step": 20133,
      "training_step_time": 7.808480501174927
    },
    {
      "epoch": 1.9201278686523438e-05,
      "model_forward_time": 1.8127219676971436,
      "step": 20134
    },
    {
      "epoch": 1.9201278686523438e-05,
      "step": 20134,
      "training_step_time": 7.447860956192017
    },
    {
      "epoch": 1.9202232360839845e-05,
      "model_forward_time": 1.8112156391143799,
      "step": 20135
    },
    {
      "epoch": 1.9202232360839845e-05,
      "step": 20135,
      "training_step_time": 7.714745998382568
    },
    {
      "epoch": 1.920318603515625e-05,
      "model_forward_time": 1.8131523132324219,
      "step": 20136
    },
    {
      "epoch": 1.920318603515625e-05,
      "step": 20136,
      "training_step_time": 9.623063325881958
    },
    {
      "epoch": 1.9204139709472657e-05,
      "model_forward_time": 1.7907848358154297,
      "step": 20137
    },
    {
      "epoch": 1.9204139709472657e-05,
      "step": 20137,
      "training_step_time": 8.948633909225464
    },
    {
      "epoch": 1.9205093383789064e-05,
      "model_forward_time": 1.8120238780975342,
      "step": 20138
    },
    {
      "epoch": 1.9205093383789064e-05,
      "step": 20138,
      "training_step_time": 8.610124826431274
    },
    {
      "epoch": 1.9206047058105468e-05,
      "model_forward_time": 1.8084135055541992,
      "step": 20139
    },
    {
      "epoch": 1.9206047058105468e-05,
      "step": 20139,
      "training_step_time": 7.676935195922852
    },
    {
      "epoch": 1.9207000732421876e-05,
      "grad_norm": 0.1272566020488739,
      "learning_rate": 9.386383529668073e-06,
      "loss": 0.0685,
      "step": 20140
    },
    {
      "epoch": 1.9207000732421876e-05,
      "model_forward_time": 1.8066401481628418,
      "step": 20140
    },
    {
      "dynamics_loss_avg": 0.05729886181652546,
      "epoch": 1.9207000732421876e-05,
      "step": 20140
    },
    {
      "action_loss_avg": 0.002906227682251483,
      "epoch": 1.9207000732421876e-05,
      "step": 20140
    },
    {
      "epoch": 1.9207000732421876e-05,
      "step": 20140,
      "training_step_time": 7.718736171722412
    },
    {
      "epoch": 1.920795440673828e-05,
      "model_forward_time": 1.820319414138794,
      "step": 20141
    },
    {
      "epoch": 1.920795440673828e-05,
      "step": 20141,
      "training_step_time": 7.357038736343384
    },
    {
      "epoch": 1.9208908081054687e-05,
      "model_forward_time": 1.8190388679504395,
      "step": 20142
    },
    {
      "epoch": 1.9208908081054687e-05,
      "step": 20142,
      "training_step_time": 7.605791330337524
    },
    {
      "epoch": 1.9209861755371094e-05,
      "model_forward_time": 1.808262586593628,
      "step": 20143
    },
    {
      "epoch": 1.9209861755371094e-05,
      "step": 20143,
      "training_step_time": 7.432174921035767
    },
    {
      "epoch": 1.92108154296875e-05,
      "model_forward_time": 1.814528465270996,
      "step": 20144
    },
    {
      "epoch": 1.92108154296875e-05,
      "step": 20144,
      "training_step_time": 7.716472148895264
    },
    {
      "epoch": 1.9211769104003906e-05,
      "model_forward_time": 1.8193399906158447,
      "step": 20145
    },
    {
      "epoch": 1.9211769104003906e-05,
      "step": 20145,
      "training_step_time": 9.094106912612915
    },
    {
      "epoch": 1.9212722778320313e-05,
      "model_forward_time": 1.8133938312530518,
      "step": 20146
    },
    {
      "epoch": 1.9212722778320313e-05,
      "step": 20146,
      "training_step_time": 8.592708349227905
    },
    {
      "epoch": 1.9213676452636717e-05,
      "model_forward_time": 2.321568250656128,
      "step": 20147
    },
    {
      "epoch": 1.9213676452636717e-05,
      "step": 20147,
      "training_step_time": 8.204381942749023
    },
    {
      "epoch": 1.9214630126953125e-05,
      "model_forward_time": 1.810760259628296,
      "step": 20148
    },
    {
      "epoch": 1.9214630126953125e-05,
      "step": 20148,
      "training_step_time": 8.707784652709961
    },
    {
      "epoch": 1.9215583801269532e-05,
      "model_forward_time": 1.811814546585083,
      "step": 20149
    },
    {
      "epoch": 1.9215583801269532e-05,
      "step": 20149,
      "training_step_time": 8.595843315124512
    },
    {
      "epoch": 1.9216537475585936e-05,
      "grad_norm": 0.19226332008838654,
      "learning_rate": 9.385589649160669e-06,
      "loss": 0.0687,
      "step": 20150
    },
    {
      "epoch": 1.9216537475585936e-05,
      "model_forward_time": 1.8111381530761719,
      "step": 20150
    },
    {
      "dynamics_loss_avg": 0.07053101416677236,
      "epoch": 1.9216537475585936e-05,
      "step": 20150
    },
    {
      "action_loss_avg": 0.004860471596475691,
      "epoch": 1.9216537475585936e-05,
      "step": 20150
    },
    {
      "epoch": 1.9216537475585936e-05,
      "step": 20150,
      "training_step_time": 7.342800140380859
    },
    {
      "epoch": 1.9217491149902344e-05,
      "model_forward_time": 1.8037927150726318,
      "step": 20151
    },
    {
      "epoch": 1.9217491149902344e-05,
      "step": 20151,
      "training_step_time": 7.875735759735107
    },
    {
      "epoch": 1.921844482421875e-05,
      "model_forward_time": 1.8043937683105469,
      "step": 20152
    },
    {
      "epoch": 1.921844482421875e-05,
      "step": 20152,
      "training_step_time": 8.215694904327393
    },
    {
      "epoch": 1.9219398498535155e-05,
      "model_forward_time": 1.8151991367340088,
      "step": 20153
    },
    {
      "epoch": 1.9219398498535155e-05,
      "step": 20153,
      "training_step_time": 7.661419153213501
    },
    {
      "epoch": 1.9220352172851562e-05,
      "model_forward_time": 1.8097374439239502,
      "step": 20154
    },
    {
      "epoch": 1.9220352172851562e-05,
      "step": 20154,
      "training_step_time": 8.216481447219849
    },
    {
      "epoch": 1.922130584716797e-05,
      "model_forward_time": 1.813262939453125,
      "step": 20155
    },
    {
      "epoch": 1.922130584716797e-05,
      "step": 20155,
      "training_step_time": 8.772143602371216
    },
    {
      "epoch": 1.9222259521484374e-05,
      "model_forward_time": 2.305075168609619,
      "step": 20156
    },
    {
      "epoch": 1.9222259521484374e-05,
      "step": 20156,
      "training_step_time": 7.6909894943237305
    },
    {
      "epoch": 1.922321319580078e-05,
      "model_forward_time": 1.7969050407409668,
      "step": 20157
    },
    {
      "epoch": 1.922321319580078e-05,
      "step": 20157,
      "training_step_time": 8.177060842514038
    },
    {
      "epoch": 1.922416687011719e-05,
      "model_forward_time": 1.8064086437225342,
      "step": 20158
    },
    {
      "epoch": 1.922416687011719e-05,
      "step": 20158,
      "training_step_time": 8.707650899887085
    },
    {
      "epoch": 1.9225120544433593e-05,
      "model_forward_time": 1.8193860054016113,
      "step": 20159
    },
    {
      "epoch": 1.9225120544433593e-05,
      "step": 20159,
      "training_step_time": 7.497384071350098
    },
    {
      "epoch": 1.922607421875e-05,
      "grad_norm": 0.15923325717449188,
      "learning_rate": 9.384795289051775e-06,
      "loss": 0.0639,
      "step": 20160
    },
    {
      "epoch": 1.922607421875e-05,
      "model_forward_time": 1.8140060901641846,
      "step": 20160
    },
    {
      "dynamics_loss_avg": 0.05541094010695815,
      "epoch": 1.922607421875e-05,
      "step": 20160
    },
    {
      "action_loss_avg": 0.002783465926768258,
      "epoch": 1.922607421875e-05,
      "step": 20160
    },
    {
      "epoch": 1.922607421875e-05,
      "step": 20160,
      "training_step_time": 7.4907567501068115
    },
    {
      "epoch": 1.9227027893066408e-05,
      "model_forward_time": 1.8115603923797607,
      "step": 20161
    },
    {
      "epoch": 1.9227027893066408e-05,
      "step": 20161,
      "training_step_time": 7.417285442352295
    },
    {
      "epoch": 1.922798156738281e-05,
      "model_forward_time": 1.8127520084381104,
      "step": 20162
    },
    {
      "epoch": 1.922798156738281e-05,
      "step": 20162,
      "training_step_time": 7.629238128662109
    },
    {
      "epoch": 1.922893524169922e-05,
      "model_forward_time": 1.8212776184082031,
      "step": 20163
    },
    {
      "epoch": 1.922893524169922e-05,
      "step": 20163,
      "training_step_time": 7.688946008682251
    },
    {
      "epoch": 1.9229888916015626e-05,
      "model_forward_time": 1.817314863204956,
      "step": 20164
    },
    {
      "epoch": 1.9229888916015626e-05,
      "step": 20164,
      "training_step_time": 7.817089557647705
    },
    {
      "epoch": 1.923084259033203e-05,
      "model_forward_time": 2.2928824424743652,
      "step": 20165
    },
    {
      "epoch": 1.923084259033203e-05,
      "step": 20165,
      "training_step_time": 7.757220029830933
    },
    {
      "epoch": 1.9231796264648438e-05,
      "model_forward_time": 1.8087730407714844,
      "step": 20166
    },
    {
      "epoch": 1.9231796264648438e-05,
      "step": 20166,
      "training_step_time": 8.243843793869019
    },
    {
      "epoch": 1.9232749938964845e-05,
      "model_forward_time": 1.7992286682128906,
      "step": 20167
    },
    {
      "epoch": 1.9232749938964845e-05,
      "step": 20167,
      "training_step_time": 7.836280107498169
    },
    {
      "epoch": 1.923370361328125e-05,
      "model_forward_time": 1.8068211078643799,
      "step": 20168
    },
    {
      "epoch": 1.923370361328125e-05,
      "step": 20168,
      "training_step_time": 7.6993567943573
    },
    {
      "epoch": 1.9234657287597657e-05,
      "model_forward_time": 1.8130052089691162,
      "step": 20169
    },
    {
      "epoch": 1.9234657287597657e-05,
      "step": 20169,
      "training_step_time": 7.488966941833496
    },
    {
      "epoch": 1.9235610961914064e-05,
      "grad_norm": 0.17594683170318604,
      "learning_rate": 9.38400044942826e-06,
      "loss": 0.0658,
      "step": 20170
    },
    {
      "epoch": 1.9235610961914064e-05,
      "model_forward_time": 1.807934045791626,
      "step": 20170
    },
    {
      "dynamics_loss_avg": 0.05469124410301447,
      "epoch": 1.9235610961914064e-05,
      "step": 20170
    },
    {
      "action_loss_avg": 0.0030886022956110536,
      "epoch": 1.9235610961914064e-05,
      "step": 20170
    },
    {
      "epoch": 1.9235610961914064e-05,
      "step": 20170,
      "training_step_time": 7.420328378677368
    },
    {
      "epoch": 1.9236564636230468e-05,
      "model_forward_time": 1.8105533123016357,
      "step": 20171
    },
    {
      "epoch": 1.9236564636230468e-05,
      "step": 20171,
      "training_step_time": 8.323362350463867
    },
    {
      "epoch": 1.9237518310546876e-05,
      "model_forward_time": 1.815641164779663,
      "step": 20172
    },
    {
      "epoch": 1.9237518310546876e-05,
      "step": 20172,
      "training_step_time": 8.264708995819092
    },
    {
      "epoch": 1.923847198486328e-05,
      "model_forward_time": 1.817887544631958,
      "step": 20173
    },
    {
      "epoch": 1.923847198486328e-05,
      "step": 20173,
      "training_step_time": 8.347378969192505
    },
    {
      "epoch": 1.9239425659179687e-05,
      "model_forward_time": 2.3480663299560547,
      "step": 20174
    },
    {
      "epoch": 1.9239425659179687e-05,
      "step": 20174,
      "training_step_time": 7.617844820022583
    },
    {
      "epoch": 1.9240379333496094e-05,
      "model_forward_time": 1.8043155670166016,
      "step": 20175
    },
    {
      "epoch": 1.9240379333496094e-05,
      "step": 20175,
      "training_step_time": 7.623072862625122
    },
    {
      "epoch": 1.92413330078125e-05,
      "model_forward_time": 1.806122064590454,
      "step": 20176
    },
    {
      "epoch": 1.92413330078125e-05,
      "step": 20176,
      "training_step_time": 7.697665691375732
    },
    {
      "epoch": 1.9242286682128906e-05,
      "model_forward_time": 1.8147473335266113,
      "step": 20177
    },
    {
      "epoch": 1.9242286682128906e-05,
      "step": 20177,
      "training_step_time": 7.602645635604858
    },
    {
      "epoch": 1.9243240356445313e-05,
      "model_forward_time": 1.8117566108703613,
      "step": 20178
    },
    {
      "epoch": 1.9243240356445313e-05,
      "step": 20178,
      "training_step_time": 7.331636667251587
    },
    {
      "epoch": 1.9244194030761717e-05,
      "model_forward_time": 1.8059258460998535,
      "step": 20179
    },
    {
      "epoch": 1.9244194030761717e-05,
      "step": 20179,
      "training_step_time": 7.381097316741943
    },
    {
      "epoch": 1.9245147705078125e-05,
      "grad_norm": 0.12172972410917282,
      "learning_rate": 9.383205130377048e-06,
      "loss": 0.0715,
      "step": 20180
    },
    {
      "epoch": 1.9245147705078125e-05,
      "model_forward_time": 1.809197187423706,
      "step": 20180
    },
    {
      "dynamics_loss_avg": 0.07231575921177864,
      "epoch": 1.9245147705078125e-05,
      "step": 20180
    },
    {
      "action_loss_avg": 0.003146096353884786,
      "epoch": 1.9245147705078125e-05,
      "step": 20180
    },
    {
      "epoch": 1.9245147705078125e-05,
      "step": 20180,
      "training_step_time": 8.432360649108887
    },
    {
      "epoch": 1.9246101379394532e-05,
      "model_forward_time": 1.8082585334777832,
      "step": 20181
    },
    {
      "epoch": 1.9246101379394532e-05,
      "step": 20181,
      "training_step_time": 7.708517789840698
    },
    {
      "epoch": 1.9247055053710936e-05,
      "model_forward_time": 1.8226401805877686,
      "step": 20182
    },
    {
      "epoch": 1.9247055053710936e-05,
      "step": 20182,
      "training_step_time": 8.366181373596191
    },
    {
      "epoch": 1.9248008728027344e-05,
      "model_forward_time": 2.3583290576934814,
      "step": 20183
    },
    {
      "epoch": 1.9248008728027344e-05,
      "step": 20183,
      "training_step_time": 9.695664644241333
    },
    {
      "epoch": 1.924896240234375e-05,
      "model_forward_time": 1.8093376159667969,
      "step": 20184
    },
    {
      "epoch": 1.924896240234375e-05,
      "step": 20184,
      "training_step_time": 7.510290861129761
    },
    {
      "epoch": 1.9249916076660155e-05,
      "model_forward_time": 1.8192698955535889,
      "step": 20185
    },
    {
      "epoch": 1.9249916076660155e-05,
      "step": 20185,
      "training_step_time": 8.711785078048706
    },
    {
      "epoch": 1.9250869750976562e-05,
      "model_forward_time": 1.8145320415496826,
      "step": 20186
    },
    {
      "epoch": 1.9250869750976562e-05,
      "step": 20186,
      "training_step_time": 7.6629862785339355
    },
    {
      "epoch": 1.925182342529297e-05,
      "model_forward_time": 1.7953178882598877,
      "step": 20187
    },
    {
      "epoch": 1.925182342529297e-05,
      "step": 20187,
      "training_step_time": 7.346063852310181
    },
    {
      "epoch": 1.9252777099609374e-05,
      "model_forward_time": 1.812018871307373,
      "step": 20188
    },
    {
      "epoch": 1.9252777099609374e-05,
      "step": 20188,
      "training_step_time": 7.429597854614258
    },
    {
      "epoch": 1.925373077392578e-05,
      "model_forward_time": 1.8087518215179443,
      "step": 20189
    },
    {
      "epoch": 1.925373077392578e-05,
      "step": 20189,
      "training_step_time": 7.582554817199707
    },
    {
      "epoch": 1.925468444824219e-05,
      "grad_norm": 0.143982395529747,
      "learning_rate": 9.382409331985114e-06,
      "loss": 0.0639,
      "step": 20190
    },
    {
      "epoch": 1.925468444824219e-05,
      "model_forward_time": 1.4501490592956543,
      "step": 20190
    },
    {
      "dynamics_loss_avg": 0.06606847420334816,
      "epoch": 1.925468444824219e-05,
      "step": 20190
    },
    {
      "action_loss_avg": 0.003222521778661758,
      "epoch": 1.925468444824219e-05,
      "step": 20190
    },
    {
      "epoch": 1.925468444824219e-05,
      "step": 20190,
      "training_step_time": 7.604239225387573
    },
    {
      "epoch": 1.9255638122558593e-05,
      "model_forward_time": 1.811349868774414,
      "step": 20191
    },
    {
      "epoch": 1.9255638122558593e-05,
      "step": 20191,
      "training_step_time": 8.953051328659058
    },
    {
      "epoch": 1.9256591796875e-05,
      "model_forward_time": 1.8096153736114502,
      "step": 20192
    },
    {
      "epoch": 1.9256591796875e-05,
      "step": 20192,
      "training_step_time": 7.700627326965332
    },
    {
      "epoch": 1.9257545471191408e-05,
      "model_forward_time": 2.3362162113189697,
      "step": 20193
    },
    {
      "epoch": 1.9257545471191408e-05,
      "step": 20193,
      "training_step_time": 8.61967420578003
    },
    {
      "epoch": 1.925849914550781e-05,
      "model_forward_time": 1.8204174041748047,
      "step": 20194
    },
    {
      "epoch": 1.925849914550781e-05,
      "step": 20194,
      "training_step_time": 7.684577226638794
    },
    {
      "epoch": 1.925945281982422e-05,
      "model_forward_time": 1.8110454082489014,
      "step": 20195
    },
    {
      "epoch": 1.925945281982422e-05,
      "step": 20195,
      "training_step_time": 8.195083856582642
    },
    {
      "epoch": 1.9260406494140626e-05,
      "model_forward_time": 1.8164334297180176,
      "step": 20196
    },
    {
      "epoch": 1.9260406494140626e-05,
      "step": 20196,
      "training_step_time": 7.599783420562744
    },
    {
      "epoch": 1.926136016845703e-05,
      "model_forward_time": 1.8214867115020752,
      "step": 20197
    },
    {
      "epoch": 1.926136016845703e-05,
      "step": 20197,
      "training_step_time": 7.3698084354400635
    },
    {
      "epoch": 1.9262313842773438e-05,
      "model_forward_time": 1.6941072940826416,
      "step": 20198
    },
    {
      "epoch": 1.9262313842773438e-05,
      "step": 20198,
      "training_step_time": 8.415890216827393
    },
    {
      "epoch": 1.9263267517089845e-05,
      "model_forward_time": 1.712930679321289,
      "step": 20199
    },
    {
      "epoch": 1.9263267517089845e-05,
      "step": 20199,
      "training_step_time": 7.614877700805664
    },
    {
      "epoch": 1.926422119140625e-05,
      "grad_norm": 0.25495657324790955,
      "learning_rate": 9.381613054339482e-06,
      "loss": 0.0632,
      "step": 20200
    },
    {
      "epoch": 1.926422119140625e-05,
      "model_forward_time": 1.7056934833526611,
      "step": 20200
    },
    {
      "dynamics_loss_avg": 0.06776084788143635,
      "epoch": 1.926422119140625e-05,
      "step": 20200
    },
    {
      "action_loss_avg": 0.003382625541416928,
      "epoch": 1.926422119140625e-05,
      "step": 20200
    },
    {
      "epoch": 1.926422119140625e-05,
      "step": 20200,
      "training_step_time": 8.713738441467285
    },
    {
      "epoch": 1.9265174865722657e-05,
      "model_forward_time": 1.7500052452087402,
      "step": 20201
    },
    {
      "epoch": 1.9265174865722657e-05,
      "step": 20201,
      "training_step_time": 8.48293399810791
    },
    {
      "epoch": 1.9266128540039064e-05,
      "model_forward_time": 2.230933666229248,
      "step": 20202
    },
    {
      "epoch": 1.9266128540039064e-05,
      "step": 20202,
      "training_step_time": 7.740163087844849
    },
    {
      "epoch": 1.9267082214355468e-05,
      "model_forward_time": 1.7071654796600342,
      "step": 20203
    },
    {
      "epoch": 1.9267082214355468e-05,
      "step": 20203,
      "training_step_time": 8.398987054824829
    },
    {
      "epoch": 1.9268035888671876e-05,
      "model_forward_time": 1.7172651290893555,
      "step": 20204
    },
    {
      "epoch": 1.9268035888671876e-05,
      "step": 20204,
      "training_step_time": 7.793431520462036
    },
    {
      "epoch": 1.926898956298828e-05,
      "model_forward_time": 1.7021229267120361,
      "step": 20205
    },
    {
      "epoch": 1.926898956298828e-05,
      "step": 20205,
      "training_step_time": 7.6770665645599365
    },
    {
      "epoch": 1.9269943237304687e-05,
      "model_forward_time": 1.7065808773040771,
      "step": 20206
    },
    {
      "epoch": 1.9269943237304687e-05,
      "step": 20206,
      "training_step_time": 7.451349496841431
    },
    {
      "epoch": 1.9270896911621094e-05,
      "model_forward_time": 1.7098636627197266,
      "step": 20207
    },
    {
      "epoch": 1.9270896911621094e-05,
      "step": 20207,
      "training_step_time": 7.640925407409668
    },
    {
      "epoch": 1.92718505859375e-05,
      "model_forward_time": 1.7056233882904053,
      "step": 20208
    },
    {
      "epoch": 1.92718505859375e-05,
      "step": 20208,
      "training_step_time": 8.64425253868103
    },
    {
      "epoch": 1.9272804260253906e-05,
      "model_forward_time": 1.708723545074463,
      "step": 20209
    },
    {
      "epoch": 1.9272804260253906e-05,
      "step": 20209,
      "training_step_time": 8.485817432403564
    },
    {
      "epoch": 1.9273757934570313e-05,
      "grad_norm": 0.16734518110752106,
      "learning_rate": 9.380816297527235e-06,
      "loss": 0.0645,
      "step": 20210
    },
    {
      "epoch": 1.9273757934570313e-05,
      "model_forward_time": 1.7457053661346436,
      "step": 20210
    },
    {
      "dynamics_loss_avg": 0.06186047876253724,
      "epoch": 1.9273757934570313e-05,
      "step": 20210
    },
    {
      "action_loss_avg": 0.0033900294947670773,
      "epoch": 1.9273757934570313e-05,
      "step": 20210
    },
    {
      "epoch": 1.9273757934570313e-05,
      "step": 20210,
      "training_step_time": 9.608648777008057
    },
    {
      "epoch": 1.9274711608886717e-05,
      "model_forward_time": 2.2982423305511475,
      "step": 20211
    },
    {
      "epoch": 1.9274711608886717e-05,
      "step": 20211,
      "training_step_time": 9.27705430984497
    },
    {
      "epoch": 1.9275665283203125e-05,
      "model_forward_time": 1.7990727424621582,
      "step": 20212
    },
    {
      "epoch": 1.9275665283203125e-05,
      "step": 20212,
      "training_step_time": 7.711335897445679
    },
    {
      "epoch": 1.9276618957519532e-05,
      "model_forward_time": 1.8063478469848633,
      "step": 20213
    },
    {
      "epoch": 1.9276618957519532e-05,
      "step": 20213,
      "training_step_time": 7.7129294872283936
    },
    {
      "epoch": 1.9277572631835936e-05,
      "model_forward_time": 1.4361774921417236,
      "step": 20214
    },
    {
      "epoch": 1.9277572631835936e-05,
      "step": 20214,
      "training_step_time": 7.679042100906372
    },
    {
      "epoch": 1.9278526306152344e-05,
      "model_forward_time": 1.7893459796905518,
      "step": 20215
    },
    {
      "epoch": 1.9278526306152344e-05,
      "step": 20215,
      "training_step_time": 7.420756578445435
    },
    {
      "epoch": 1.927947998046875e-05,
      "model_forward_time": 1.801354169845581,
      "step": 20216
    },
    {
      "epoch": 1.927947998046875e-05,
      "step": 20216,
      "training_step_time": 7.6660802364349365
    },
    {
      "epoch": 1.9280433654785155e-05,
      "model_forward_time": 1.7833280563354492,
      "step": 20217
    },
    {
      "epoch": 1.9280433654785155e-05,
      "step": 20217,
      "training_step_time": 7.710031509399414
    },
    {
      "epoch": 1.9281387329101562e-05,
      "model_forward_time": 1.340059518814087,
      "step": 20218
    },
    {
      "epoch": 1.9281387329101562e-05,
      "step": 20218,
      "training_step_time": 8.720702648162842
    },
    {
      "epoch": 1.928234100341797e-05,
      "model_forward_time": 1.707362174987793,
      "step": 20219
    },
    {
      "epoch": 1.928234100341797e-05,
      "step": 20219,
      "training_step_time": 7.84324049949646
    },
    {
      "epoch": 1.9283294677734374e-05,
      "grad_norm": 0.2078913003206253,
      "learning_rate": 9.380019061635506e-06,
      "loss": 0.0597,
      "step": 20220
    },
    {
      "epoch": 1.9283294677734374e-05,
      "model_forward_time": 1.7047724723815918,
      "step": 20220
    },
    {
      "dynamics_loss_avg": 0.0607567485421896,
      "epoch": 1.9283294677734374e-05,
      "step": 20220
    },
    {
      "action_loss_avg": 0.003705971408635378,
      "epoch": 1.9283294677734374e-05,
      "step": 20220
    },
    {
      "epoch": 1.9283294677734374e-05,
      "step": 20220,
      "training_step_time": 7.79451322555542
    },
    {
      "epoch": 1.928424835205078e-05,
      "model_forward_time": 1.7605514526367188,
      "step": 20221
    },
    {
      "epoch": 1.928424835205078e-05,
      "step": 20221,
      "training_step_time": 7.828589916229248
    },
    {
      "epoch": 1.928520202636719e-05,
      "model_forward_time": 1.7014703750610352,
      "step": 20222
    },
    {
      "epoch": 1.928520202636719e-05,
      "step": 20222,
      "training_step_time": 7.705695390701294
    },
    {
      "epoch": 1.9286155700683593e-05,
      "model_forward_time": 1.7114391326904297,
      "step": 20223
    },
    {
      "epoch": 1.9286155700683593e-05,
      "step": 20223,
      "training_step_time": 7.815487384796143
    },
    {
      "epoch": 1.9287109375e-05,
      "model_forward_time": 1.70322847366333,
      "step": 20224
    },
    {
      "epoch": 1.9287109375e-05,
      "step": 20224,
      "training_step_time": 7.495265483856201
    },
    {
      "epoch": 1.9288063049316408e-05,
      "model_forward_time": 1.7094643115997314,
      "step": 20225
    },
    {
      "epoch": 1.9288063049316408e-05,
      "step": 20225,
      "training_step_time": 7.29835844039917
    },
    {
      "epoch": 1.928901672363281e-05,
      "model_forward_time": 1.7131259441375732,
      "step": 20226
    },
    {
      "epoch": 1.928901672363281e-05,
      "step": 20226,
      "training_step_time": 7.674044609069824
    },
    {
      "epoch": 1.928997039794922e-05,
      "model_forward_time": 1.344522476196289,
      "step": 20227
    },
    {
      "epoch": 1.928997039794922e-05,
      "step": 20227,
      "training_step_time": 7.705934047698975
    },
    {
      "epoch": 1.9290924072265626e-05,
      "model_forward_time": 1.7003049850463867,
      "step": 20228
    },
    {
      "epoch": 1.9290924072265626e-05,
      "step": 20228,
      "training_step_time": 9.261983871459961
    },
    {
      "epoch": 1.929187774658203e-05,
      "model_forward_time": 1.699242353439331,
      "step": 20229
    },
    {
      "epoch": 1.929187774658203e-05,
      "step": 20229,
      "training_step_time": 8.265878200531006
    },
    {
      "epoch": 1.9292831420898438e-05,
      "grad_norm": 0.20703205466270447,
      "learning_rate": 9.379221346751474e-06,
      "loss": 0.0632,
      "step": 20230
    },
    {
      "epoch": 1.9292831420898438e-05,
      "model_forward_time": 2.2868778705596924,
      "step": 20230
    },
    {
      "dynamics_loss_avg": 0.0665909718722105,
      "epoch": 1.9292831420898438e-05,
      "step": 20230
    },
    {
      "action_loss_avg": 0.009323715989012271,
      "epoch": 1.9292831420898438e-05,
      "step": 20230
    },
    {
      "epoch": 1.9292831420898438e-05,
      "step": 20230,
      "training_step_time": 8.525115013122559
    },
    {
      "epoch": 1.9293785095214845e-05,
      "model_forward_time": 1.7149813175201416,
      "step": 20231
    },
    {
      "epoch": 1.9293785095214845e-05,
      "step": 20231,
      "training_step_time": 8.779669761657715
    },
    {
      "epoch": 1.929473876953125e-05,
      "model_forward_time": 1.717543363571167,
      "step": 20232
    },
    {
      "epoch": 1.929473876953125e-05,
      "step": 20232,
      "training_step_time": 8.631103038787842
    },
    {
      "epoch": 1.9295692443847657e-05,
      "model_forward_time": 1.7142515182495117,
      "step": 20233
    },
    {
      "epoch": 1.9295692443847657e-05,
      "step": 20233,
      "training_step_time": 7.385090589523315
    },
    {
      "epoch": 1.9296646118164064e-05,
      "model_forward_time": 1.7137274742126465,
      "step": 20234
    },
    {
      "epoch": 1.9296646118164064e-05,
      "step": 20234,
      "training_step_time": 7.265782117843628
    },
    {
      "epoch": 1.9297599792480468e-05,
      "model_forward_time": 1.354870319366455,
      "step": 20235
    },
    {
      "epoch": 1.9297599792480468e-05,
      "step": 20235,
      "training_step_time": 7.611647605895996
    },
    {
      "epoch": 1.9298553466796876e-05,
      "model_forward_time": 1.33762788772583,
      "step": 20236
    },
    {
      "epoch": 1.9298553466796876e-05,
      "step": 20236,
      "training_step_time": 8.639693021774292
    },
    {
      "epoch": 1.929950714111328e-05,
      "model_forward_time": 1.7179982662200928,
      "step": 20237
    },
    {
      "epoch": 1.929950714111328e-05,
      "step": 20237,
      "training_step_time": 7.778337478637695
    },
    {
      "epoch": 1.9300460815429687e-05,
      "model_forward_time": 1.7095201015472412,
      "step": 20238
    },
    {
      "epoch": 1.9300460815429687e-05,
      "step": 20238,
      "training_step_time": 7.69133186340332
    },
    {
      "epoch": 1.9301414489746094e-05,
      "model_forward_time": 2.288088083267212,
      "step": 20239
    },
    {
      "epoch": 1.9301414489746094e-05,
      "step": 20239,
      "training_step_time": 7.781514883041382
    },
    {
      "epoch": 1.93023681640625e-05,
      "grad_norm": 0.25335410237312317,
      "learning_rate": 9.378423152962383e-06,
      "loss": 0.0636,
      "step": 20240
    },
    {
      "epoch": 1.93023681640625e-05,
      "model_forward_time": 1.715449333190918,
      "step": 20240
    },
    {
      "dynamics_loss_avg": 0.04227971825748682,
      "epoch": 1.93023681640625e-05,
      "step": 20240
    },
    {
      "action_loss_avg": 0.0035239261720562353,
      "epoch": 1.93023681640625e-05,
      "step": 20240
    },
    {
      "epoch": 1.93023681640625e-05,
      "step": 20240,
      "training_step_time": 7.7148473262786865
    },
    {
      "epoch": 1.9303321838378906e-05,
      "model_forward_time": 1.714611291885376,
      "step": 20241
    },
    {
      "epoch": 1.9303321838378906e-05,
      "step": 20241,
      "training_step_time": 7.682626962661743
    },
    {
      "epoch": 1.9304275512695313e-05,
      "model_forward_time": 1.7241599559783936,
      "step": 20242
    },
    {
      "epoch": 1.9304275512695313e-05,
      "step": 20242,
      "training_step_time": 7.582172632217407
    },
    {
      "epoch": 1.9305229187011717e-05,
      "model_forward_time": 1.7229411602020264,
      "step": 20243
    },
    {
      "epoch": 1.9305229187011717e-05,
      "step": 20243,
      "training_step_time": 6.734739303588867
    },
    {
      "epoch": 1.9306182861328125e-05,
      "model_forward_time": 1.711874008178711,
      "step": 20244
    },
    {
      "epoch": 1.9306182861328125e-05,
      "step": 20244,
      "training_step_time": 7.624317646026611
    },
    {
      "epoch": 1.9307136535644532e-05,
      "model_forward_time": 1.7073845863342285,
      "step": 20245
    },
    {
      "epoch": 1.9307136535644532e-05,
      "step": 20245,
      "training_step_time": 7.642021894454956
    },
    {
      "epoch": 1.9308090209960936e-05,
      "model_forward_time": 1.7173960208892822,
      "step": 20246
    },
    {
      "epoch": 1.9308090209960936e-05,
      "step": 20246,
      "training_step_time": 8.538599967956543
    },
    {
      "epoch": 1.9309043884277344e-05,
      "model_forward_time": 1.3264944553375244,
      "step": 20247
    },
    {
      "epoch": 1.9309043884277344e-05,
      "step": 20247,
      "training_step_time": 7.6477649211883545
    },
    {
      "epoch": 1.930999755859375e-05,
      "model_forward_time": 1.7468576431274414,
      "step": 20248
    },
    {
      "epoch": 1.930999755859375e-05,
      "step": 20248,
      "training_step_time": 9.716180086135864
    },
    {
      "epoch": 1.9310951232910155e-05,
      "model_forward_time": 2.2680137157440186,
      "step": 20249
    },
    {
      "epoch": 1.9310951232910155e-05,
      "step": 20249,
      "training_step_time": 7.666856527328491
    },
    {
      "epoch": 1.9311904907226562e-05,
      "grad_norm": 0.23560354113578796,
      "learning_rate": 9.377624480355517e-06,
      "loss": 0.0658,
      "step": 20250
    },
    {
      "epoch": 1.9311904907226562e-05,
      "model_forward_time": 1.7106561660766602,
      "step": 20250
    },
    {
      "dynamics_loss_avg": 0.05187956299632788,
      "epoch": 1.9311904907226562e-05,
      "step": 20250
    },
    {
      "action_loss_avg": 0.002762017271015793,
      "epoch": 1.9311904907226562e-05,
      "step": 20250
    },
    {
      "epoch": 1.9311904907226562e-05,
      "step": 20250,
      "training_step_time": 7.6730005741119385
    },
    {
      "epoch": 1.931285858154297e-05,
      "model_forward_time": 1.7345197200775146,
      "step": 20251
    },
    {
      "epoch": 1.931285858154297e-05,
      "step": 20251,
      "training_step_time": 8.259942770004272
    },
    {
      "epoch": 1.9313812255859374e-05,
      "model_forward_time": 1.349172592163086,
      "step": 20252
    },
    {
      "epoch": 1.9313812255859374e-05,
      "step": 20252,
      "training_step_time": 6.743311166763306
    },
    {
      "epoch": 1.931476593017578e-05,
      "model_forward_time": 1.7172014713287354,
      "step": 20253
    },
    {
      "epoch": 1.931476593017578e-05,
      "step": 20253,
      "training_step_time": 7.594541788101196
    },
    {
      "epoch": 1.931571960449219e-05,
      "model_forward_time": 1.7303271293640137,
      "step": 20254
    },
    {
      "epoch": 1.931571960449219e-05,
      "step": 20254,
      "training_step_time": 7.684649229049683
    },
    {
      "epoch": 1.9316673278808593e-05,
      "model_forward_time": 1.716172218322754,
      "step": 20255
    },
    {
      "epoch": 1.9316673278808593e-05,
      "step": 20255,
      "training_step_time": 8.728807210922241
    },
    {
      "epoch": 1.9317626953125e-05,
      "model_forward_time": 1.7108631134033203,
      "step": 20256
    },
    {
      "epoch": 1.9317626953125e-05,
      "step": 20256,
      "training_step_time": 8.50917673110962
    },
    {
      "epoch": 1.9318580627441408e-05,
      "model_forward_time": 1.760871410369873,
      "step": 20257
    },
    {
      "epoch": 1.9318580627441408e-05,
      "step": 20257,
      "training_step_time": 9.318479776382446
    },
    {
      "epoch": 1.931953430175781e-05,
      "model_forward_time": 2.2810847759246826,
      "step": 20258
    },
    {
      "epoch": 1.931953430175781e-05,
      "step": 20258,
      "training_step_time": 7.683748483657837
    },
    {
      "epoch": 1.932048797607422e-05,
      "model_forward_time": 1.7230932712554932,
      "step": 20259
    },
    {
      "epoch": 1.932048797607422e-05,
      "step": 20259,
      "training_step_time": 7.632810354232788
    },
    {
      "epoch": 1.9321441650390626e-05,
      "grad_norm": 0.19369536638259888,
      "learning_rate": 9.37682532901822e-06,
      "loss": 0.0623,
      "step": 20260
    },
    {
      "epoch": 1.9321441650390626e-05,
      "model_forward_time": 1.7125439643859863,
      "step": 20260
    },
    {
      "dynamics_loss_avg": 0.05042058937251568,
      "epoch": 1.9321441650390626e-05,
      "step": 20260
    },
    {
      "action_loss_avg": 0.0020824083941988645,
      "epoch": 1.9321441650390626e-05,
      "step": 20260
    },
    {
      "epoch": 1.9321441650390626e-05,
      "step": 20260,
      "training_step_time": 7.716568946838379
    },
    {
      "epoch": 1.932239532470703e-05,
      "model_forward_time": 1.7064363956451416,
      "step": 20261
    },
    {
      "epoch": 1.932239532470703e-05,
      "step": 20261,
      "training_step_time": 7.386132717132568
    },
    {
      "epoch": 1.9323348999023438e-05,
      "model_forward_time": 1.70432710647583,
      "step": 20262
    },
    {
      "epoch": 1.9323348999023438e-05,
      "step": 20262,
      "training_step_time": 7.6006739139556885
    },
    {
      "epoch": 1.9324302673339845e-05,
      "model_forward_time": 1.7045369148254395,
      "step": 20263
    },
    {
      "epoch": 1.9324302673339845e-05,
      "step": 20263,
      "training_step_time": 7.62325644493103
    },
    {
      "epoch": 1.932525634765625e-05,
      "model_forward_time": 1.6980564594268799,
      "step": 20264
    },
    {
      "epoch": 1.932525634765625e-05,
      "step": 20264,
      "training_step_time": 7.709779739379883
    },
    {
      "epoch": 1.9326210021972657e-05,
      "model_forward_time": 1.7059295177459717,
      "step": 20265
    },
    {
      "epoch": 1.9326210021972657e-05,
      "step": 20265,
      "training_step_time": 8.833359479904175
    },
    {
      "epoch": 1.932716369628906e-05,
      "model_forward_time": 1.7061669826507568,
      "step": 20266
    },
    {
      "epoch": 1.932716369628906e-05,
      "step": 20266,
      "training_step_time": 8.530368566513062
    },
    {
      "epoch": 1.9328117370605468e-05,
      "model_forward_time": 2.255648136138916,
      "step": 20267
    },
    {
      "epoch": 1.9328117370605468e-05,
      "step": 20267,
      "training_step_time": 7.668362379074097
    },
    {
      "epoch": 1.9329071044921876e-05,
      "model_forward_time": 1.7047297954559326,
      "step": 20268
    },
    {
      "epoch": 1.9329071044921876e-05,
      "step": 20268,
      "training_step_time": 7.677001237869263
    },
    {
      "epoch": 1.933002471923828e-05,
      "model_forward_time": 1.7118942737579346,
      "step": 20269
    },
    {
      "epoch": 1.933002471923828e-05,
      "step": 20269,
      "training_step_time": 7.722332954406738
    },
    {
      "epoch": 1.9330978393554687e-05,
      "grad_norm": 0.14271564781665802,
      "learning_rate": 9.376025699037885e-06,
      "loss": 0.065,
      "step": 20270
    },
    {
      "epoch": 1.9330978393554687e-05,
      "model_forward_time": 1.7061583995819092,
      "step": 20270
    },
    {
      "dynamics_loss_avg": 0.05667847134172917,
      "epoch": 1.9330978393554687e-05,
      "step": 20270
    },
    {
      "action_loss_avg": 0.002543967930250801,
      "epoch": 1.9330978393554687e-05,
      "step": 20270
    },
    {
      "epoch": 1.9330978393554687e-05,
      "step": 20270,
      "training_step_time": 7.613495826721191
    },
    {
      "epoch": 1.9331932067871095e-05,
      "model_forward_time": 1.7056174278259277,
      "step": 20271
    },
    {
      "epoch": 1.9331932067871095e-05,
      "step": 20271,
      "training_step_time": 7.273879051208496
    },
    {
      "epoch": 1.93328857421875e-05,
      "model_forward_time": 1.7050895690917969,
      "step": 20272
    },
    {
      "epoch": 1.93328857421875e-05,
      "step": 20272,
      "training_step_time": 8.326982259750366
    },
    {
      "epoch": 1.9333839416503906e-05,
      "model_forward_time": 1.7037134170532227,
      "step": 20273
    },
    {
      "epoch": 1.9333839416503906e-05,
      "step": 20273,
      "training_step_time": 9.262266397476196
    },
    {
      "epoch": 1.9334793090820313e-05,
      "model_forward_time": 1.703939437866211,
      "step": 20274
    },
    {
      "epoch": 1.9334793090820313e-05,
      "step": 20274,
      "training_step_time": 8.815024375915527
    },
    {
      "epoch": 1.9335746765136717e-05,
      "model_forward_time": 1.7174499034881592,
      "step": 20275
    },
    {
      "epoch": 1.9335746765136717e-05,
      "step": 20275,
      "training_step_time": 7.800841808319092
    },
    {
      "epoch": 1.9336700439453125e-05,
      "model_forward_time": 2.278653621673584,
      "step": 20276
    },
    {
      "epoch": 1.9336700439453125e-05,
      "step": 20276,
      "training_step_time": 8.85631513595581
    },
    {
      "epoch": 1.9337654113769532e-05,
      "model_forward_time": 1.7073540687561035,
      "step": 20277
    },
    {
      "epoch": 1.9337654113769532e-05,
      "step": 20277,
      "training_step_time": 9.2647385597229
    },
    {
      "epoch": 1.9338607788085936e-05,
      "model_forward_time": 1.6997003555297852,
      "step": 20278
    },
    {
      "epoch": 1.9338607788085936e-05,
      "step": 20278,
      "training_step_time": 7.681831359863281
    },
    {
      "epoch": 1.9339561462402344e-05,
      "model_forward_time": 1.7049577236175537,
      "step": 20279
    },
    {
      "epoch": 1.9339561462402344e-05,
      "step": 20279,
      "training_step_time": 7.649159908294678
    },
    {
      "epoch": 1.934051513671875e-05,
      "grad_norm": 0.17985127866268158,
      "learning_rate": 9.37522559050196e-06,
      "loss": 0.0658,
      "step": 20280
    },
    {
      "epoch": 1.934051513671875e-05,
      "model_forward_time": 1.7025012969970703,
      "step": 20280
    },
    {
      "dynamics_loss_avg": 0.06562234628945589,
      "epoch": 1.934051513671875e-05,
      "step": 20280
    },
    {
      "action_loss_avg": 0.0037431966135045513,
      "epoch": 1.934051513671875e-05,
      "step": 20280
    },
    {
      "epoch": 1.934051513671875e-05,
      "step": 20280,
      "training_step_time": 7.256247282028198
    },
    {
      "epoch": 1.9341468811035155e-05,
      "model_forward_time": 1.7048308849334717,
      "step": 20281
    },
    {
      "epoch": 1.9341468811035155e-05,
      "step": 20281,
      "training_step_time": 8.332947492599487
    },
    {
      "epoch": 1.9342422485351563e-05,
      "model_forward_time": 1.7018887996673584,
      "step": 20282
    },
    {
      "epoch": 1.9342422485351563e-05,
      "step": 20282,
      "training_step_time": 7.684529066085815
    },
    {
      "epoch": 1.934337615966797e-05,
      "model_forward_time": 1.7053849697113037,
      "step": 20283
    },
    {
      "epoch": 1.934337615966797e-05,
      "step": 20283,
      "training_step_time": 10.016647338867188
    },
    {
      "epoch": 1.9344329833984374e-05,
      "model_forward_time": 1.7118005752563477,
      "step": 20284
    },
    {
      "epoch": 1.9344329833984374e-05,
      "step": 20284,
      "training_step_time": 8.407192468643188
    },
    {
      "epoch": 1.934528350830078e-05,
      "model_forward_time": 1.7062830924987793,
      "step": 20285
    },
    {
      "epoch": 1.934528350830078e-05,
      "step": 20285,
      "training_step_time": 9.094972848892212
    },
    {
      "epoch": 1.934623718261719e-05,
      "model_forward_time": 1.3454689979553223,
      "step": 20286
    },
    {
      "epoch": 1.934623718261719e-05,
      "step": 20286,
      "training_step_time": 7.751052618026733
    },
    {
      "epoch": 1.9347190856933593e-05,
      "model_forward_time": 1.3358814716339111,
      "step": 20287
    },
    {
      "epoch": 1.9347190856933593e-05,
      "step": 20287,
      "training_step_time": 7.708521842956543
    },
    {
      "epoch": 1.934814453125e-05,
      "model_forward_time": 1.702089786529541,
      "step": 20288
    },
    {
      "epoch": 1.934814453125e-05,
      "step": 20288,
      "training_step_time": 7.748577356338501
    },
    {
      "epoch": 1.9349098205566408e-05,
      "model_forward_time": 1.701700210571289,
      "step": 20289
    },
    {
      "epoch": 1.9349098205566408e-05,
      "step": 20289,
      "training_step_time": 7.4071033000946045
    },
    {
      "epoch": 1.935005187988281e-05,
      "grad_norm": 0.2116568237543106,
      "learning_rate": 9.37442500349794e-06,
      "loss": 0.0654,
      "step": 20290
    },
    {
      "epoch": 1.935005187988281e-05,
      "model_forward_time": 1.3352723121643066,
      "step": 20290
    },
    {
      "dynamics_loss_avg": 0.06896969769150019,
      "epoch": 1.935005187988281e-05,
      "step": 20290
    },
    {
      "action_loss_avg": 0.0027570052596274765,
      "epoch": 1.935005187988281e-05,
      "step": 20290
    },
    {
      "epoch": 1.935005187988281e-05,
      "step": 20290,
      "training_step_time": 7.662355899810791
    },
    {
      "epoch": 1.935100555419922e-05,
      "model_forward_time": 1.7059667110443115,
      "step": 20291
    },
    {
      "epoch": 1.935100555419922e-05,
      "step": 20291,
      "training_step_time": 7.619308948516846
    },
    {
      "epoch": 1.9351959228515627e-05,
      "model_forward_time": 1.7070400714874268,
      "step": 20292
    },
    {
      "epoch": 1.9351959228515627e-05,
      "step": 20292,
      "training_step_time": 8.836750268936157
    },
    {
      "epoch": 1.935291290283203e-05,
      "model_forward_time": 1.7369189262390137,
      "step": 20293
    },
    {
      "epoch": 1.935291290283203e-05,
      "step": 20293,
      "training_step_time": 8.610033750534058
    },
    {
      "epoch": 1.9353866577148438e-05,
      "model_forward_time": 1.7043962478637695,
      "step": 20294
    },
    {
      "epoch": 1.9353866577148438e-05,
      "step": 20294,
      "training_step_time": 8.822240114212036
    },
    {
      "epoch": 1.9354820251464845e-05,
      "model_forward_time": 2.250990152359009,
      "step": 20295
    },
    {
      "epoch": 1.9354820251464845e-05,
      "step": 20295,
      "training_step_time": 7.715776205062866
    },
    {
      "epoch": 1.935577392578125e-05,
      "model_forward_time": 1.3341238498687744,
      "step": 20296
    },
    {
      "epoch": 1.935577392578125e-05,
      "step": 20296,
      "training_step_time": 7.660533428192139
    },
    {
      "epoch": 1.9356727600097657e-05,
      "model_forward_time": 1.7098150253295898,
      "step": 20297
    },
    {
      "epoch": 1.9356727600097657e-05,
      "step": 20297,
      "training_step_time": 7.7584710121154785
    },
    {
      "epoch": 1.935768127441406e-05,
      "model_forward_time": 1.7083971500396729,
      "step": 20298
    },
    {
      "epoch": 1.935768127441406e-05,
      "step": 20298,
      "training_step_time": 7.353702545166016
    },
    {
      "epoch": 1.9358634948730468e-05,
      "model_forward_time": 1.7077207565307617,
      "step": 20299
    },
    {
      "epoch": 1.9358634948730468e-05,
      "step": 20299,
      "training_step_time": 7.583384275436401
    },
    {
      "epoch": 1.9359588623046876e-05,
      "grad_norm": 0.1564246416091919,
      "learning_rate": 9.373623938113381e-06,
      "loss": 0.064,
      "step": 20300
    },
    {
      "epoch": 1.9359588623046876e-05,
      "model_forward_time": 1.703296184539795,
      "step": 20300
    },
    {
      "dynamics_loss_avg": 0.041695518977940084,
      "epoch": 1.9359588623046876e-05,
      "step": 20300
    },
    {
      "action_loss_avg": 0.0016812121612019837,
      "epoch": 1.9359588623046876e-05,
      "step": 20300
    },
    {
      "epoch": 1.9359588623046876e-05,
      "step": 20300,
      "training_step_time": 7.559267282485962
    },
    {
      "epoch": 1.936054229736328e-05,
      "model_forward_time": 1.716407299041748,
      "step": 20301
    },
    {
      "epoch": 1.936054229736328e-05,
      "step": 20301,
      "training_step_time": 8.209835529327393
    },
    {
      "epoch": 1.9361495971679687e-05,
      "model_forward_time": 1.7098708152770996,
      "step": 20302
    },
    {
      "epoch": 1.9361495971679687e-05,
      "step": 20302,
      "training_step_time": 8.595560073852539
    },
    {
      "epoch": 1.9362449645996095e-05,
      "model_forward_time": 1.7080776691436768,
      "step": 20303
    },
    {
      "epoch": 1.9362449645996095e-05,
      "step": 20303,
      "training_step_time": 8.37151575088501
    },
    {
      "epoch": 1.93634033203125e-05,
      "model_forward_time": 2.259155035018921,
      "step": 20304
    },
    {
      "epoch": 1.93634033203125e-05,
      "step": 20304,
      "training_step_time": 8.523446798324585
    },
    {
      "epoch": 1.9364356994628906e-05,
      "model_forward_time": 1.7070767879486084,
      "step": 20305
    },
    {
      "epoch": 1.9364356994628906e-05,
      "step": 20305,
      "training_step_time": 7.603435516357422
    },
    {
      "epoch": 1.9365310668945313e-05,
      "model_forward_time": 1.7127461433410645,
      "step": 20306
    },
    {
      "epoch": 1.9365310668945313e-05,
      "step": 20306,
      "training_step_time": 7.65160608291626
    },
    {
      "epoch": 1.9366264343261717e-05,
      "model_forward_time": 1.6982407569885254,
      "step": 20307
    },
    {
      "epoch": 1.9366264343261717e-05,
      "step": 20307,
      "training_step_time": 7.362863063812256
    },
    {
      "epoch": 1.9367218017578125e-05,
      "model_forward_time": 1.7106668949127197,
      "step": 20308
    },
    {
      "epoch": 1.9367218017578125e-05,
      "step": 20308,
      "training_step_time": 7.220386266708374
    },
    {
      "epoch": 1.9368171691894532e-05,
      "model_forward_time": 1.709815263748169,
      "step": 20309
    },
    {
      "epoch": 1.9368171691894532e-05,
      "step": 20309,
      "training_step_time": 7.637782096862793
    },
    {
      "epoch": 1.9369125366210936e-05,
      "grad_norm": 0.1916215419769287,
      "learning_rate": 9.372822394435884e-06,
      "loss": 0.0625,
      "step": 20310
    },
    {
      "epoch": 1.9369125366210936e-05,
      "model_forward_time": 1.69637131690979,
      "step": 20310
    },
    {
      "dynamics_loss_avg": 0.05418521203100681,
      "epoch": 1.9369125366210936e-05,
      "step": 20310
    },
    {
      "action_loss_avg": 0.0022576196701265872,
      "epoch": 1.9369125366210936e-05,
      "step": 20310
    },
    {
      "epoch": 1.9369125366210936e-05,
      "step": 20310,
      "training_step_time": 9.499958992004395
    },
    {
      "epoch": 1.9370079040527344e-05,
      "model_forward_time": 1.7117321491241455,
      "step": 20311
    },
    {
      "epoch": 1.9370079040527344e-05,
      "step": 20311,
      "training_step_time": 7.752580642700195
    },
    {
      "epoch": 1.937103271484375e-05,
      "model_forward_time": 1.7102575302124023,
      "step": 20312
    },
    {
      "epoch": 1.937103271484375e-05,
      "step": 20312,
      "training_step_time": 8.528372049331665
    },
    {
      "epoch": 1.9371986389160155e-05,
      "model_forward_time": 1.7094502449035645,
      "step": 20313
    },
    {
      "epoch": 1.9371986389160155e-05,
      "step": 20313,
      "training_step_time": 8.281198263168335
    },
    {
      "epoch": 1.9372940063476563e-05,
      "model_forward_time": 1.7224481105804443,
      "step": 20314
    },
    {
      "epoch": 1.9372940063476563e-05,
      "step": 20314,
      "training_step_time": 7.661984443664551
    },
    {
      "epoch": 1.937389373779297e-05,
      "model_forward_time": 1.7043139934539795,
      "step": 20315
    },
    {
      "epoch": 1.937389373779297e-05,
      "step": 20315,
      "training_step_time": 8.397874116897583
    },
    {
      "epoch": 1.9374847412109374e-05,
      "model_forward_time": 1.693755865097046,
      "step": 20316
    },
    {
      "epoch": 1.9374847412109374e-05,
      "step": 20316,
      "training_step_time": 7.490790128707886
    },
    {
      "epoch": 1.937580108642578e-05,
      "model_forward_time": 1.6844215393066406,
      "step": 20317
    },
    {
      "epoch": 1.937580108642578e-05,
      "step": 20317,
      "training_step_time": 7.032246112823486
    },
    {
      "epoch": 1.937675476074219e-05,
      "model_forward_time": 1.7048425674438477,
      "step": 20318
    },
    {
      "epoch": 1.937675476074219e-05,
      "step": 20318,
      "training_step_time": 8.267541408538818
    },
    {
      "epoch": 1.9377708435058593e-05,
      "model_forward_time": 1.6997036933898926,
      "step": 20319
    },
    {
      "epoch": 1.9377708435058593e-05,
      "step": 20319,
      "training_step_time": 7.692500591278076
    },
    {
      "epoch": 1.9378662109375e-05,
      "grad_norm": 0.1734965592622757,
      "learning_rate": 9.372020372553102e-06,
      "loss": 0.0644,
      "step": 20320
    },
    {
      "epoch": 1.9378662109375e-05,
      "model_forward_time": 1.7017698287963867,
      "step": 20320
    },
    {
      "dynamics_loss_avg": 0.06057583447545767,
      "epoch": 1.9378662109375e-05,
      "step": 20320
    },
    {
      "action_loss_avg": 0.005560180055908859,
      "epoch": 1.9378662109375e-05,
      "step": 20320
    },
    {
      "epoch": 1.9378662109375e-05,
      "step": 20320,
      "training_step_time": 7.667575359344482
    },
    {
      "epoch": 1.9379615783691408e-05,
      "model_forward_time": 1.7053754329681396,
      "step": 20321
    },
    {
      "epoch": 1.9379615783691408e-05,
      "step": 20321,
      "training_step_time": 9.22783899307251
    },
    {
      "epoch": 1.9380569458007812e-05,
      "model_forward_time": 1.7041592597961426,
      "step": 20322
    },
    {
      "epoch": 1.9380569458007812e-05,
      "step": 20322,
      "training_step_time": 8.684515237808228
    },
    {
      "epoch": 1.938152313232422e-05,
      "model_forward_time": 2.2557711601257324,
      "step": 20323
    },
    {
      "epoch": 1.938152313232422e-05,
      "step": 20323,
      "training_step_time": 10.782978534698486
    },
    {
      "epoch": 1.9382476806640627e-05,
      "model_forward_time": 1.7001566886901855,
      "step": 20324
    },
    {
      "epoch": 1.9382476806640627e-05,
      "step": 20324,
      "training_step_time": 7.612824201583862
    },
    {
      "epoch": 1.938343048095703e-05,
      "model_forward_time": 1.7013847827911377,
      "step": 20325
    },
    {
      "epoch": 1.938343048095703e-05,
      "step": 20325,
      "training_step_time": 7.6926000118255615
    },
    {
      "epoch": 1.9384384155273438e-05,
      "model_forward_time": 1.7033076286315918,
      "step": 20326
    },
    {
      "epoch": 1.9384384155273438e-05,
      "step": 20326,
      "training_step_time": 7.117109298706055
    },
    {
      "epoch": 1.9385337829589845e-05,
      "model_forward_time": 1.6950111389160156,
      "step": 20327
    },
    {
      "epoch": 1.9385337829589845e-05,
      "step": 20327,
      "training_step_time": 8.596585988998413
    },
    {
      "epoch": 1.938629150390625e-05,
      "model_forward_time": 1.704833745956421,
      "step": 20328
    },
    {
      "epoch": 1.938629150390625e-05,
      "step": 20328,
      "training_step_time": 7.641262531280518
    },
    {
      "epoch": 1.9387245178222657e-05,
      "model_forward_time": 1.7021660804748535,
      "step": 20329
    },
    {
      "epoch": 1.9387245178222657e-05,
      "step": 20329,
      "training_step_time": 7.705719709396362
    },
    {
      "epoch": 1.938819885253906e-05,
      "grad_norm": 0.17591318488121033,
      "learning_rate": 9.371217872552746e-06,
      "loss": 0.0594,
      "step": 20330
    },
    {
      "epoch": 1.938819885253906e-05,
      "model_forward_time": 1.6956040859222412,
      "step": 20330
    },
    {
      "dynamics_loss_avg": 0.06190832946449518,
      "epoch": 1.938819885253906e-05,
      "step": 20330
    },
    {
      "action_loss_avg": 0.004069574025925249,
      "epoch": 1.938819885253906e-05,
      "step": 20330
    },
    {
      "epoch": 1.938819885253906e-05,
      "step": 20330,
      "training_step_time": 8.690638303756714
    },
    {
      "epoch": 1.9389152526855468e-05,
      "model_forward_time": 1.696563720703125,
      "step": 20331
    },
    {
      "epoch": 1.9389152526855468e-05,
      "step": 20331,
      "training_step_time": 7.843195676803589
    },
    {
      "epoch": 1.9390106201171876e-05,
      "model_forward_time": 2.259570360183716,
      "step": 20332
    },
    {
      "epoch": 1.9390106201171876e-05,
      "step": 20332,
      "training_step_time": 9.506263494491577
    },
    {
      "epoch": 1.939105987548828e-05,
      "model_forward_time": 1.7057771682739258,
      "step": 20333
    },
    {
      "epoch": 1.939105987548828e-05,
      "step": 20333,
      "training_step_time": 7.65706729888916
    },
    {
      "epoch": 1.9392013549804687e-05,
      "model_forward_time": 1.7050127983093262,
      "step": 20334
    },
    {
      "epoch": 1.9392013549804687e-05,
      "step": 20334,
      "training_step_time": 7.678199768066406
    },
    {
      "epoch": 1.9392967224121095e-05,
      "model_forward_time": 1.7058007717132568,
      "step": 20335
    },
    {
      "epoch": 1.9392967224121095e-05,
      "step": 20335,
      "training_step_time": 7.336198568344116
    },
    {
      "epoch": 1.93939208984375e-05,
      "model_forward_time": 1.7013802528381348,
      "step": 20336
    },
    {
      "epoch": 1.93939208984375e-05,
      "step": 20336,
      "training_step_time": 7.588237285614014
    },
    {
      "epoch": 1.9394874572753906e-05,
      "model_forward_time": 2.848438262939453,
      "step": 20337
    },
    {
      "epoch": 1.9394874572753906e-05,
      "step": 20337,
      "training_step_time": 7.663765907287598
    },
    {
      "epoch": 1.9395828247070313e-05,
      "model_forward_time": 1.3179075717926025,
      "step": 20338
    },
    {
      "epoch": 1.9395828247070313e-05,
      "step": 20338,
      "training_step_time": 8.732270240783691
    },
    {
      "epoch": 1.9396781921386717e-05,
      "model_forward_time": 1.6900930404663086,
      "step": 20339
    },
    {
      "epoch": 1.9396781921386717e-05,
      "step": 20339,
      "training_step_time": 6.690209627151489
    },
    {
      "epoch": 1.9397735595703125e-05,
      "grad_norm": 0.2296830266714096,
      "learning_rate": 9.370414894522577e-06,
      "loss": 0.0606,
      "step": 20340
    },
    {
      "epoch": 1.9397735595703125e-05,
      "model_forward_time": 1.6929008960723877,
      "step": 20340
    },
    {
      "dynamics_loss_avg": 0.046603734605014324,
      "epoch": 1.9397735595703125e-05,
      "step": 20340
    },
    {
      "action_loss_avg": 0.0028421125607565045,
      "epoch": 1.9397735595703125e-05,
      "step": 20340
    },
    {
      "epoch": 1.9397735595703125e-05,
      "step": 20340,
      "training_step_time": 6.726271152496338
    },
    {
      "epoch": 1.9398689270019532e-05,
      "model_forward_time": 1.6950013637542725,
      "step": 20341
    },
    {
      "epoch": 1.9398689270019532e-05,
      "step": 20341,
      "training_step_time": 6.719966411590576
    },
    {
      "epoch": 1.9399642944335936e-05,
      "model_forward_time": 1.689528465270996,
      "step": 20342
    },
    {
      "epoch": 1.9399642944335936e-05,
      "step": 20342,
      "training_step_time": 6.764739990234375
    },
    {
      "epoch": 1.9400596618652344e-05,
      "model_forward_time": 1.6916134357452393,
      "step": 20343
    },
    {
      "epoch": 1.9400596618652344e-05,
      "step": 20343,
      "training_step_time": 6.744994640350342
    },
    {
      "epoch": 1.940155029296875e-05,
      "model_forward_time": 1.6898319721221924,
      "step": 20344
    },
    {
      "epoch": 1.940155029296875e-05,
      "step": 20344,
      "training_step_time": 6.718563556671143
    },
    {
      "epoch": 1.9402503967285155e-05,
      "model_forward_time": 1.687551736831665,
      "step": 20345
    },
    {
      "epoch": 1.9402503967285155e-05,
      "step": 20345,
      "training_step_time": 6.709662914276123
    },
    {
      "epoch": 1.9403457641601563e-05,
      "model_forward_time": 2.244992733001709,
      "step": 20346
    },
    {
      "epoch": 1.9403457641601563e-05,
      "step": 20346,
      "training_step_time": 8.703717470169067
    },
    {
      "epoch": 1.940441131591797e-05,
      "model_forward_time": 1.6918261051177979,
      "step": 20347
    },
    {
      "epoch": 1.940441131591797e-05,
      "step": 20347,
      "training_step_time": 12.597923040390015
    },
    {
      "epoch": 1.9405364990234374e-05,
      "model_forward_time": 1.6884009838104248,
      "step": 20348
    },
    {
      "epoch": 1.9405364990234374e-05,
      "step": 20348,
      "training_step_time": 7.746995449066162
    },
    {
      "epoch": 1.940631866455078e-05,
      "model_forward_time": 1.6859350204467773,
      "step": 20349
    },
    {
      "epoch": 1.940631866455078e-05,
      "step": 20349,
      "training_step_time": 6.736756324768066
    },
    {
      "epoch": 1.940727233886719e-05,
      "grad_norm": 0.13804465532302856,
      "learning_rate": 9.369611438550405e-06,
      "loss": 0.0608,
      "step": 20350
    },
    {
      "epoch": 1.940727233886719e-05,
      "model_forward_time": 1.689049243927002,
      "step": 20350
    },
    {
      "dynamics_loss_avg": 0.06006599832326174,
      "epoch": 1.940727233886719e-05,
      "step": 20350
    },
    {
      "action_loss_avg": 0.0028212792763952165,
      "epoch": 1.940727233886719e-05,
      "step": 20350
    },
    {
      "epoch": 1.940727233886719e-05,
      "step": 20350,
      "training_step_time": 6.75042724609375
    },
    {
      "epoch": 1.9408226013183593e-05,
      "model_forward_time": 1.6888339519500732,
      "step": 20351
    },
    {
      "epoch": 1.9408226013183593e-05,
      "step": 20351,
      "training_step_time": 6.716406583786011
    },
    {
      "epoch": 1.94091796875e-05,
      "model_forward_time": 1.3155910968780518,
      "step": 20352
    },
    {
      "epoch": 1.94091796875e-05,
      "step": 20352,
      "training_step_time": 6.730597972869873
    },
    {
      "epoch": 1.9410133361816408e-05,
      "model_forward_time": 1.6876311302185059,
      "step": 20353
    },
    {
      "epoch": 1.9410133361816408e-05,
      "step": 20353,
      "training_step_time": 6.731038570404053
    },
    {
      "epoch": 1.9411087036132812e-05,
      "model_forward_time": 1.6912152767181396,
      "step": 20354
    },
    {
      "epoch": 1.9411087036132812e-05,
      "step": 20354,
      "training_step_time": 6.72608494758606
    },
    {
      "epoch": 1.941204071044922e-05,
      "model_forward_time": 1.6908338069915771,
      "step": 20355
    },
    {
      "epoch": 1.941204071044922e-05,
      "step": 20355,
      "training_step_time": 8.144570350646973
    },
    {
      "epoch": 1.9412994384765627e-05,
      "model_forward_time": 2.228060722351074,
      "step": 20356
    },
    {
      "epoch": 1.9412994384765627e-05,
      "step": 20356,
      "training_step_time": 10.466126203536987
    },
    {
      "epoch": 1.941394805908203e-05,
      "model_forward_time": 1.7000582218170166,
      "step": 20357
    },
    {
      "epoch": 1.941394805908203e-05,
      "step": 20357,
      "training_step_time": 7.9686174392700195
    },
    {
      "epoch": 1.9414901733398438e-05,
      "model_forward_time": 1.689969539642334,
      "step": 20358
    },
    {
      "epoch": 1.9414901733398438e-05,
      "step": 20358,
      "training_step_time": 6.858670711517334
    },
    {
      "epoch": 1.9415855407714845e-05,
      "model_forward_time": 1.6944515705108643,
      "step": 20359
    },
    {
      "epoch": 1.9415855407714845e-05,
      "step": 20359,
      "training_step_time": 6.8577094078063965
    },
    {
      "epoch": 1.941680908203125e-05,
      "grad_norm": 0.17120693624019623,
      "learning_rate": 9.368807504724096e-06,
      "loss": 0.0636,
      "step": 20360
    },
    {
      "epoch": 1.941680908203125e-05,
      "model_forward_time": 1.6875569820404053,
      "step": 20360
    },
    {
      "dynamics_loss_avg": 0.0474535508081317,
      "epoch": 1.941680908203125e-05,
      "step": 20360
    },
    {
      "action_loss_avg": 0.0015448178921360523,
      "epoch": 1.941680908203125e-05,
      "step": 20360
    },
    {
      "epoch": 1.941680908203125e-05,
      "step": 20360,
      "training_step_time": 6.8712732791900635
    },
    {
      "epoch": 1.9417762756347657e-05,
      "model_forward_time": 1.6919548511505127,
      "step": 20361
    },
    {
      "epoch": 1.9417762756347657e-05,
      "step": 20361,
      "training_step_time": 6.864614248275757
    },
    {
      "epoch": 1.941871643066406e-05,
      "model_forward_time": 1.694244623184204,
      "step": 20362
    },
    {
      "epoch": 1.941871643066406e-05,
      "step": 20362,
      "training_step_time": 6.875605344772339
    },
    {
      "epoch": 1.941967010498047e-05,
      "model_forward_time": 1.6885499954223633,
      "step": 20363
    },
    {
      "epoch": 1.941967010498047e-05,
      "step": 20363,
      "training_step_time": 6.852485418319702
    },
    {
      "epoch": 1.9420623779296876e-05,
      "model_forward_time": 1.3151473999023438,
      "step": 20364
    },
    {
      "epoch": 1.9420623779296876e-05,
      "step": 20364,
      "training_step_time": 6.757652997970581
    },
    {
      "epoch": 1.942157745361328e-05,
      "model_forward_time": 2.269026756286621,
      "step": 20365
    },
    {
      "epoch": 1.942157745361328e-05,
      "step": 20365,
      "training_step_time": 10.465150117874146
    },
    {
      "epoch": 1.9422531127929687e-05,
      "model_forward_time": 1.6883509159088135,
      "step": 20366
    },
    {
      "epoch": 1.9422531127929687e-05,
      "step": 20366,
      "training_step_time": 7.9096808433532715
    },
    {
      "epoch": 1.9423484802246095e-05,
      "model_forward_time": 1.6883456707000732,
      "step": 20367
    },
    {
      "epoch": 1.9423484802246095e-05,
      "step": 20367,
      "training_step_time": 6.835608243942261
    },
    {
      "epoch": 1.94244384765625e-05,
      "model_forward_time": 1.6943809986114502,
      "step": 20368
    },
    {
      "epoch": 1.94244384765625e-05,
      "step": 20368,
      "training_step_time": 6.836789846420288
    },
    {
      "epoch": 1.9425392150878906e-05,
      "model_forward_time": 1.690396785736084,
      "step": 20369
    },
    {
      "epoch": 1.9425392150878906e-05,
      "step": 20369,
      "training_step_time": 6.789155721664429
    },
    {
      "epoch": 1.9426345825195313e-05,
      "grad_norm": 0.12316124141216278,
      "learning_rate": 9.368003093131566e-06,
      "loss": 0.0598,
      "step": 20370
    },
    {
      "epoch": 1.9426345825195313e-05,
      "model_forward_time": 1.6875526905059814,
      "step": 20370
    },
    {
      "dynamics_loss_avg": 0.04844050128012896,
      "epoch": 1.9426345825195313e-05,
      "step": 20370
    },
    {
      "action_loss_avg": 0.0023906799557153135,
      "epoch": 1.9426345825195313e-05,
      "step": 20370
    },
    {
      "epoch": 1.9426345825195313e-05,
      "step": 20370,
      "training_step_time": 6.851133346557617
    },
    {
      "epoch": 1.9427299499511717e-05,
      "model_forward_time": 1.680997610092163,
      "step": 20371
    },
    {
      "epoch": 1.9427299499511717e-05,
      "step": 20371,
      "training_step_time": 6.8529932498931885
    },
    {
      "epoch": 1.9428253173828125e-05,
      "model_forward_time": 1.6921281814575195,
      "step": 20372
    },
    {
      "epoch": 1.9428253173828125e-05,
      "step": 20372,
      "training_step_time": 6.799596071243286
    },
    {
      "epoch": 1.9429206848144532e-05,
      "model_forward_time": 1.6972017288208008,
      "step": 20373
    },
    {
      "epoch": 1.9429206848144532e-05,
      "step": 20373,
      "training_step_time": 6.777020454406738
    },
    {
      "epoch": 1.9430160522460936e-05,
      "model_forward_time": 2.2789549827575684,
      "step": 20374
    },
    {
      "epoch": 1.9430160522460936e-05,
      "step": 20374,
      "training_step_time": 8.316535234451294
    },
    {
      "epoch": 1.9431114196777344e-05,
      "model_forward_time": 1.708101749420166,
      "step": 20375
    },
    {
      "epoch": 1.9431114196777344e-05,
      "step": 20375,
      "training_step_time": 9.744979619979858
    },
    {
      "epoch": 1.943206787109375e-05,
      "model_forward_time": 1.6983933448791504,
      "step": 20376
    },
    {
      "epoch": 1.943206787109375e-05,
      "step": 20376,
      "training_step_time": 7.271791458129883
    },
    {
      "epoch": 1.9433021545410155e-05,
      "model_forward_time": 1.692143440246582,
      "step": 20377
    },
    {
      "epoch": 1.9433021545410155e-05,
      "step": 20377,
      "training_step_time": 6.816013336181641
    },
    {
      "epoch": 1.9433975219726563e-05,
      "model_forward_time": 1.6932909488677979,
      "step": 20378
    },
    {
      "epoch": 1.9433975219726563e-05,
      "step": 20378,
      "training_step_time": 6.849112510681152
    },
    {
      "epoch": 1.943492889404297e-05,
      "model_forward_time": 1.6885416507720947,
      "step": 20379
    },
    {
      "epoch": 1.943492889404297e-05,
      "step": 20379,
      "training_step_time": 6.8175013065338135
    },
    {
      "epoch": 1.9435882568359374e-05,
      "grad_norm": 0.17851899564266205,
      "learning_rate": 9.367198203860785e-06,
      "loss": 0.0659,
      "step": 20380
    },
    {
      "epoch": 1.9435882568359374e-05,
      "model_forward_time": 1.690542459487915,
      "step": 20380
    },
    {
      "dynamics_loss_avg": 0.05845503378659487,
      "epoch": 1.9435882568359374e-05,
      "step": 20380
    },
    {
      "action_loss_avg": 0.0021088944340590388,
      "epoch": 1.9435882568359374e-05,
      "step": 20380
    },
    {
      "epoch": 1.9435882568359374e-05,
      "step": 20380,
      "training_step_time": 6.777862310409546
    },
    {
      "epoch": 1.943683624267578e-05,
      "model_forward_time": 1.694624662399292,
      "step": 20381
    },
    {
      "epoch": 1.943683624267578e-05,
      "step": 20381,
      "training_step_time": 6.805800914764404
    },
    {
      "epoch": 1.943778991699219e-05,
      "model_forward_time": 1.6920840740203857,
      "step": 20382
    },
    {
      "epoch": 1.943778991699219e-05,
      "step": 20382,
      "training_step_time": 6.847142457962036
    },
    {
      "epoch": 1.9438743591308593e-05,
      "model_forward_time": 2.2603254318237305,
      "step": 20383
    },
    {
      "epoch": 1.9438743591308593e-05,
      "step": 20383,
      "training_step_time": 7.771666049957275
    },
    {
      "epoch": 1.9439697265625e-05,
      "model_forward_time": 1.7048029899597168,
      "step": 20384
    },
    {
      "epoch": 1.9439697265625e-05,
      "step": 20384,
      "training_step_time": 11.909502744674683
    },
    {
      "epoch": 1.9440650939941408e-05,
      "model_forward_time": 1.6876893043518066,
      "step": 20385
    },
    {
      "epoch": 1.9440650939941408e-05,
      "step": 20385,
      "training_step_time": 7.696065902709961
    },
    {
      "epoch": 1.9441604614257812e-05,
      "model_forward_time": 1.6922533512115479,
      "step": 20386
    },
    {
      "epoch": 1.9441604614257812e-05,
      "step": 20386,
      "training_step_time": 6.722109794616699
    },
    {
      "epoch": 1.944255828857422e-05,
      "model_forward_time": 1.6850895881652832,
      "step": 20387
    },
    {
      "epoch": 1.944255828857422e-05,
      "step": 20387,
      "training_step_time": 6.718013763427734
    },
    {
      "epoch": 1.9443511962890627e-05,
      "model_forward_time": 1.6922426223754883,
      "step": 20388
    },
    {
      "epoch": 1.9443511962890627e-05,
      "step": 20388,
      "training_step_time": 6.774514436721802
    },
    {
      "epoch": 1.944446563720703e-05,
      "model_forward_time": 1.3202474117279053,
      "step": 20389
    },
    {
      "epoch": 1.944446563720703e-05,
      "step": 20389,
      "training_step_time": 6.822141408920288
    },
    {
      "epoch": 1.9445419311523438e-05,
      "grad_norm": 0.14199039340019226,
      "learning_rate": 9.366392836999775e-06,
      "loss": 0.0639,
      "step": 20390
    },
    {
      "epoch": 1.9445419311523438e-05,
      "model_forward_time": 1.6902832984924316,
      "step": 20390
    },
    {
      "dynamics_loss_avg": 0.05872091595083475,
      "epoch": 1.9445419311523438e-05,
      "step": 20390
    },
    {
      "action_loss_avg": 0.0028632331217522734,
      "epoch": 1.9445419311523438e-05,
      "step": 20390
    },
    {
      "epoch": 1.9445419311523438e-05,
      "step": 20390,
      "training_step_time": 6.732887029647827
    },
    {
      "epoch": 1.9446372985839842e-05,
      "model_forward_time": 1.6882624626159668,
      "step": 20391
    },
    {
      "epoch": 1.9446372985839842e-05,
      "step": 20391,
      "training_step_time": 6.746241807937622
    },
    {
      "epoch": 1.944732666015625e-05,
      "model_forward_time": 1.729212999343872,
      "step": 20392
    },
    {
      "epoch": 1.944732666015625e-05,
      "step": 20392,
      "training_step_time": 7.725822448730469
    },
    {
      "epoch": 1.9448280334472657e-05,
      "model_forward_time": 2.2817654609680176,
      "step": 20393
    },
    {
      "epoch": 1.9448280334472657e-05,
      "step": 20393,
      "training_step_time": 11.987291097640991
    },
    {
      "epoch": 1.944923400878906e-05,
      "model_forward_time": 1.7172715663909912,
      "step": 20394
    },
    {
      "epoch": 1.944923400878906e-05,
      "step": 20394,
      "training_step_time": 7.680396795272827
    },
    {
      "epoch": 1.945018768310547e-05,
      "model_forward_time": 1.7244741916656494,
      "step": 20395
    },
    {
      "epoch": 1.945018768310547e-05,
      "step": 20395,
      "training_step_time": 6.889932632446289
    },
    {
      "epoch": 1.9451141357421876e-05,
      "model_forward_time": 1.7265348434448242,
      "step": 20396
    },
    {
      "epoch": 1.9451141357421876e-05,
      "step": 20396,
      "training_step_time": 6.757551908493042
    },
    {
      "epoch": 1.945209503173828e-05,
      "model_forward_time": 1.7293012142181396,
      "step": 20397
    },
    {
      "epoch": 1.945209503173828e-05,
      "step": 20397,
      "training_step_time": 6.746903896331787
    },
    {
      "epoch": 1.9453048706054687e-05,
      "model_forward_time": 1.7253358364105225,
      "step": 20398
    },
    {
      "epoch": 1.9453048706054687e-05,
      "step": 20398,
      "training_step_time": 6.802930116653442
    },
    {
      "epoch": 1.9454002380371095e-05,
      "model_forward_time": 1.7229278087615967,
      "step": 20399
    },
    {
      "epoch": 1.9454002380371095e-05,
      "step": 20399,
      "training_step_time": 6.764256954193115
    },
    {
      "epoch": 1.94549560546875e-05,
      "grad_norm": 0.15973196923732758,
      "learning_rate": 9.365586992636609e-06,
      "loss": 0.0671,
      "step": 20400
    },
    {
      "epoch": 1.94549560546875e-05,
      "model_forward_time": 1.719193458557129,
      "step": 20400
    },
    {
      "dynamics_loss_avg": 0.0483015825971961,
      "epoch": 1.94549560546875e-05,
      "step": 20400
    },
    {
      "action_loss_avg": 0.0019061995786614715,
      "epoch": 1.94549560546875e-05,
      "step": 20400
    },
    {
      "epoch": 1.94549560546875e-05,
      "step": 20400,
      "training_step_time": 6.790953874588013
    },
    {
      "epoch": 1.9455909729003906e-05,
      "model_forward_time": 1.7259149551391602,
      "step": 20401
    },
    {
      "epoch": 1.9455909729003906e-05,
      "step": 20401,
      "training_step_time": 8.793289422988892
    },
    {
      "epoch": 1.9456863403320313e-05,
      "model_forward_time": 2.289764165878296,
      "step": 20402
    },
    {
      "epoch": 1.9456863403320313e-05,
      "step": 20402,
      "training_step_time": 11.123982906341553
    },
    {
      "epoch": 1.9457817077636718e-05,
      "model_forward_time": 1.7225818634033203,
      "step": 20403
    },
    {
      "epoch": 1.9457817077636718e-05,
      "step": 20403,
      "training_step_time": 8.797071933746338
    },
    {
      "epoch": 1.9458770751953125e-05,
      "model_forward_time": 1.7249743938446045,
      "step": 20404
    },
    {
      "epoch": 1.9458770751953125e-05,
      "step": 20404,
      "training_step_time": 7.211205959320068
    },
    {
      "epoch": 1.9459724426269532e-05,
      "model_forward_time": 1.7228083610534668,
      "step": 20405
    },
    {
      "epoch": 1.9459724426269532e-05,
      "step": 20405,
      "training_step_time": 6.773069620132446
    },
    {
      "epoch": 1.9460678100585936e-05,
      "model_forward_time": 1.7298915386199951,
      "step": 20406
    },
    {
      "epoch": 1.9460678100585936e-05,
      "step": 20406,
      "training_step_time": 6.682842969894409
    },
    {
      "epoch": 1.9461631774902344e-05,
      "model_forward_time": 1.719621181488037,
      "step": 20407
    },
    {
      "epoch": 1.9461631774902344e-05,
      "step": 20407,
      "training_step_time": 6.7593464851379395
    },
    {
      "epoch": 1.946258544921875e-05,
      "model_forward_time": 1.7171759605407715,
      "step": 20408
    },
    {
      "epoch": 1.946258544921875e-05,
      "step": 20408,
      "training_step_time": 6.741591691970825
    },
    {
      "epoch": 1.9463539123535155e-05,
      "model_forward_time": 1.710033655166626,
      "step": 20409
    },
    {
      "epoch": 1.9463539123535155e-05,
      "step": 20409,
      "training_step_time": 6.720456600189209
    },
    {
      "epoch": 1.9464492797851563e-05,
      "grad_norm": 0.10680270195007324,
      "learning_rate": 9.364780670859412e-06,
      "loss": 0.0607,
      "step": 20410
    },
    {
      "epoch": 1.9464492797851563e-05,
      "model_forward_time": 1.706789255142212,
      "step": 20410
    },
    {
      "dynamics_loss_avg": 0.055050557106733324,
      "epoch": 1.9464492797851563e-05,
      "step": 20410
    },
    {
      "action_loss_avg": 0.003583533351775259,
      "epoch": 1.9464492797851563e-05,
      "step": 20410
    },
    {
      "epoch": 1.9464492797851563e-05,
      "step": 20410,
      "training_step_time": 7.596629858016968
    },
    {
      "epoch": 1.946544647216797e-05,
      "model_forward_time": 2.282927989959717,
      "step": 20411
    },
    {
      "epoch": 1.946544647216797e-05,
      "step": 20411,
      "training_step_time": 9.738134145736694
    },
    {
      "epoch": 1.9466400146484374e-05,
      "model_forward_time": 1.7326288223266602,
      "step": 20412
    },
    {
      "epoch": 1.9466400146484374e-05,
      "step": 20412,
      "training_step_time": 8.918787717819214
    },
    {
      "epoch": 1.946735382080078e-05,
      "model_forward_time": 1.7141454219818115,
      "step": 20413
    },
    {
      "epoch": 1.946735382080078e-05,
      "step": 20413,
      "training_step_time": 7.508171319961548
    },
    {
      "epoch": 1.946830749511719e-05,
      "model_forward_time": 1.7116775512695312,
      "step": 20414
    },
    {
      "epoch": 1.946830749511719e-05,
      "step": 20414,
      "training_step_time": 6.7788636684417725
    },
    {
      "epoch": 1.9469261169433593e-05,
      "model_forward_time": 1.7115423679351807,
      "step": 20415
    },
    {
      "epoch": 1.9469261169433593e-05,
      "step": 20415,
      "training_step_time": 6.808350086212158
    },
    {
      "epoch": 1.947021484375e-05,
      "model_forward_time": 1.7134499549865723,
      "step": 20416
    },
    {
      "epoch": 1.947021484375e-05,
      "step": 20416,
      "training_step_time": 6.774113655090332
    },
    {
      "epoch": 1.9471168518066408e-05,
      "model_forward_time": 1.7198858261108398,
      "step": 20417
    },
    {
      "epoch": 1.9471168518066408e-05,
      "step": 20417,
      "training_step_time": 6.763444900512695
    },
    {
      "epoch": 1.9472122192382812e-05,
      "model_forward_time": 1.7238919734954834,
      "step": 20418
    },
    {
      "epoch": 1.9472122192382812e-05,
      "step": 20418,
      "training_step_time": 6.780453443527222
    },
    {
      "epoch": 1.947307586669922e-05,
      "model_forward_time": 1.7127163410186768,
      "step": 20419
    },
    {
      "epoch": 1.947307586669922e-05,
      "step": 20419,
      "training_step_time": 7.589471340179443
    },
    {
      "epoch": 1.9474029541015627e-05,
      "grad_norm": 0.43835020065307617,
      "learning_rate": 9.363973871756365e-06,
      "loss": 0.0617,
      "step": 20420
    },
    {
      "epoch": 1.9474029541015627e-05,
      "model_forward_time": 2.2645339965820312,
      "step": 20420
    },
    {
      "dynamics_loss_avg": 0.04440104514360428,
      "epoch": 1.9474029541015627e-05,
      "step": 20420
    },
    {
      "action_loss_avg": 0.0028578507131896912,
      "epoch": 1.9474029541015627e-05,
      "step": 20420
    },
    {
      "epoch": 1.9474029541015627e-05,
      "step": 20420,
      "training_step_time": 7.77468204498291
    },
    {
      "epoch": 1.947498321533203e-05,
      "model_forward_time": 1.7362146377563477,
      "step": 20421
    },
    {
      "epoch": 1.947498321533203e-05,
      "step": 20421,
      "training_step_time": 10.870016813278198
    },
    {
      "epoch": 1.9475936889648438e-05,
      "model_forward_time": 1.7144782543182373,
      "step": 20422
    },
    {
      "epoch": 1.9475936889648438e-05,
      "step": 20422,
      "training_step_time": 7.8394615650177
    },
    {
      "epoch": 1.9476890563964842e-05,
      "model_forward_time": 1.7124860286712646,
      "step": 20423
    },
    {
      "epoch": 1.9476890563964842e-05,
      "step": 20423,
      "training_step_time": 6.681446075439453
    },
    {
      "epoch": 1.947784423828125e-05,
      "model_forward_time": 1.707693099975586,
      "step": 20424
    },
    {
      "epoch": 1.947784423828125e-05,
      "step": 20424,
      "training_step_time": 6.753617286682129
    },
    {
      "epoch": 1.9478797912597657e-05,
      "model_forward_time": 1.7007830142974854,
      "step": 20425
    },
    {
      "epoch": 1.9478797912597657e-05,
      "step": 20425,
      "training_step_time": 6.742845058441162
    },
    {
      "epoch": 1.947975158691406e-05,
      "model_forward_time": 1.304736614227295,
      "step": 20426
    },
    {
      "epoch": 1.947975158691406e-05,
      "step": 20426,
      "training_step_time": 6.645732164382935
    },
    {
      "epoch": 1.948070526123047e-05,
      "model_forward_time": 1.682539701461792,
      "step": 20427
    },
    {
      "epoch": 1.948070526123047e-05,
      "step": 20427,
      "training_step_time": 6.784087896347046
    },
    {
      "epoch": 1.9481658935546876e-05,
      "model_forward_time": 1.686873435974121,
      "step": 20428
    },
    {
      "epoch": 1.9481658935546876e-05,
      "step": 20428,
      "training_step_time": 7.261543035507202
    },
    {
      "epoch": 1.948261260986328e-05,
      "model_forward_time": 2.200169801712036,
      "step": 20429
    },
    {
      "epoch": 1.948261260986328e-05,
      "step": 20429,
      "training_step_time": 9.00481629371643
    },
    {
      "epoch": 1.9483566284179687e-05,
      "grad_norm": 0.17845870554447174,
      "learning_rate": 9.363166595415695e-06,
      "loss": 0.0625,
      "step": 20430
    },
    {
      "epoch": 1.9483566284179687e-05,
      "model_forward_time": 1.68699312210083,
      "step": 20430
    },
    {
      "dynamics_loss_avg": 0.07012042216956615,
      "epoch": 1.9483566284179687e-05,
      "step": 20430
    },
    {
      "action_loss_avg": 0.002995078853564337,
      "epoch": 1.9483566284179687e-05,
      "step": 20430
    },
    {
      "epoch": 1.9483566284179687e-05,
      "step": 20430,
      "training_step_time": 10.23334002494812
    },
    {
      "epoch": 1.9484519958496095e-05,
      "model_forward_time": 1.681678056716919,
      "step": 20431
    },
    {
      "epoch": 1.9484519958496095e-05,
      "step": 20431,
      "training_step_time": 7.70808482170105
    },
    {
      "epoch": 1.94854736328125e-05,
      "model_forward_time": 1.684743881225586,
      "step": 20432
    },
    {
      "epoch": 1.94854736328125e-05,
      "step": 20432,
      "training_step_time": 6.76020073890686
    },
    {
      "epoch": 1.9486427307128906e-05,
      "model_forward_time": 1.6888878345489502,
      "step": 20433
    },
    {
      "epoch": 1.9486427307128906e-05,
      "step": 20433,
      "training_step_time": 6.7427978515625
    },
    {
      "epoch": 1.9487380981445314e-05,
      "model_forward_time": 1.6831440925598145,
      "step": 20434
    },
    {
      "epoch": 1.9487380981445314e-05,
      "step": 20434,
      "training_step_time": 6.743125677108765
    },
    {
      "epoch": 1.9488334655761718e-05,
      "model_forward_time": 1.6851527690887451,
      "step": 20435
    },
    {
      "epoch": 1.9488334655761718e-05,
      "step": 20435,
      "training_step_time": 6.7661497592926025
    },
    {
      "epoch": 1.9489288330078125e-05,
      "model_forward_time": 1.680619716644287,
      "step": 20436
    },
    {
      "epoch": 1.9489288330078125e-05,
      "step": 20436,
      "training_step_time": 6.731971025466919
    },
    {
      "epoch": 1.9490242004394532e-05,
      "model_forward_time": 1.685105323791504,
      "step": 20437
    },
    {
      "epoch": 1.9490242004394532e-05,
      "step": 20437,
      "training_step_time": 7.2753307819366455
    },
    {
      "epoch": 1.9491195678710936e-05,
      "model_forward_time": 1.323305606842041,
      "step": 20438
    },
    {
      "epoch": 1.9491195678710936e-05,
      "step": 20438,
      "training_step_time": 7.760093450546265
    },
    {
      "epoch": 1.9492149353027344e-05,
      "model_forward_time": 2.2371668815612793,
      "step": 20439
    },
    {
      "epoch": 1.9492149353027344e-05,
      "step": 20439,
      "training_step_time": 10.483877420425415
    },
    {
      "epoch": 1.949310302734375e-05,
      "grad_norm": 0.143259659409523,
      "learning_rate": 9.362358841925688e-06,
      "loss": 0.0638,
      "step": 20440
    },
    {
      "epoch": 1.949310302734375e-05,
      "model_forward_time": 1.688239574432373,
      "step": 20440
    },
    {
      "dynamics_loss_avg": 0.055254006013274196,
      "epoch": 1.949310302734375e-05,
      "step": 20440
    },
    {
      "action_loss_avg": 0.002034278807695955,
      "epoch": 1.949310302734375e-05,
      "step": 20440
    },
    {
      "epoch": 1.949310302734375e-05,
      "step": 20440,
      "training_step_time": 8.394747257232666
    },
    {
      "epoch": 1.9494056701660155e-05,
      "model_forward_time": 1.6859371662139893,
      "step": 20441
    },
    {
      "epoch": 1.9494056701660155e-05,
      "step": 20441,
      "training_step_time": 7.530980348587036
    },
    {
      "epoch": 1.9495010375976563e-05,
      "model_forward_time": 1.6897914409637451,
      "step": 20442
    },
    {
      "epoch": 1.9495010375976563e-05,
      "step": 20442,
      "training_step_time": 6.744039058685303
    },
    {
      "epoch": 1.949596405029297e-05,
      "model_forward_time": 1.6867637634277344,
      "step": 20443
    },
    {
      "epoch": 1.949596405029297e-05,
      "step": 20443,
      "training_step_time": 6.761474609375
    },
    {
      "epoch": 1.9496917724609374e-05,
      "model_forward_time": 1.6949634552001953,
      "step": 20444
    },
    {
      "epoch": 1.9496917724609374e-05,
      "step": 20444,
      "training_step_time": 6.786246061325073
    },
    {
      "epoch": 1.949787139892578e-05,
      "model_forward_time": 1.6776611804962158,
      "step": 20445
    },
    {
      "epoch": 1.949787139892578e-05,
      "step": 20445,
      "training_step_time": 6.775854587554932
    },
    {
      "epoch": 1.949882507324219e-05,
      "model_forward_time": 1.6883819103240967,
      "step": 20446
    },
    {
      "epoch": 1.949882507324219e-05,
      "step": 20446,
      "training_step_time": 7.442271709442139
    },
    {
      "epoch": 1.9499778747558593e-05,
      "model_forward_time": 1.7225642204284668,
      "step": 20447
    },
    {
      "epoch": 1.9499778747558593e-05,
      "step": 20447,
      "training_step_time": 8.763985633850098
    },
    {
      "epoch": 1.9500732421875e-05,
      "model_forward_time": 2.2343225479125977,
      "step": 20448
    },
    {
      "epoch": 1.9500732421875e-05,
      "step": 20448,
      "training_step_time": 8.916720628738403
    },
    {
      "epoch": 1.9501686096191408e-05,
      "model_forward_time": 1.7034003734588623,
      "step": 20449
    },
    {
      "epoch": 1.9501686096191408e-05,
      "step": 20449,
      "training_step_time": 8.734511375427246
    },
    {
      "epoch": 1.9502639770507812e-05,
      "grad_norm": 0.13694380223751068,
      "learning_rate": 9.361550611374675e-06,
      "loss": 0.0687,
      "step": 20450
    },
    {
      "epoch": 1.9502639770507812e-05,
      "model_forward_time": 1.6883540153503418,
      "step": 20450
    },
    {
      "dynamics_loss_avg": 0.07069059796631336,
      "epoch": 1.9502639770507812e-05,
      "step": 20450
    },
    {
      "action_loss_avg": 0.0030608009546995163,
      "epoch": 1.9502639770507812e-05,
      "step": 20450
    },
    {
      "epoch": 1.9502639770507812e-05,
      "step": 20450,
      "training_step_time": 7.808253288269043
    },
    {
      "epoch": 1.950359344482422e-05,
      "model_forward_time": 1.6834335327148438,
      "step": 20451
    },
    {
      "epoch": 1.950359344482422e-05,
      "step": 20451,
      "training_step_time": 6.7822585105896
    },
    {
      "epoch": 1.9504547119140627e-05,
      "model_forward_time": 1.6857903003692627,
      "step": 20452
    },
    {
      "epoch": 1.9504547119140627e-05,
      "step": 20452,
      "training_step_time": 6.781742811203003
    },
    {
      "epoch": 1.950550079345703e-05,
      "model_forward_time": 1.6874661445617676,
      "step": 20453
    },
    {
      "epoch": 1.950550079345703e-05,
      "step": 20453,
      "training_step_time": 6.792375326156616
    },
    {
      "epoch": 1.9506454467773438e-05,
      "model_forward_time": 1.6885404586791992,
      "step": 20454
    },
    {
      "epoch": 1.9506454467773438e-05,
      "step": 20454,
      "training_step_time": 6.781313896179199
    },
    {
      "epoch": 1.9507408142089842e-05,
      "model_forward_time": 1.6852848529815674,
      "step": 20455
    },
    {
      "epoch": 1.9507408142089842e-05,
      "step": 20455,
      "training_step_time": 7.422973871231079
    },
    {
      "epoch": 1.950836181640625e-05,
      "model_forward_time": 1.6962919235229492,
      "step": 20456
    },
    {
      "epoch": 1.950836181640625e-05,
      "step": 20456,
      "training_step_time": 7.688480377197266
    },
    {
      "epoch": 1.9509315490722657e-05,
      "model_forward_time": 2.23014497756958,
      "step": 20457
    },
    {
      "epoch": 1.9509315490722657e-05,
      "step": 20457,
      "training_step_time": 10.652372360229492
    },
    {
      "epoch": 1.951026916503906e-05,
      "model_forward_time": 1.696108341217041,
      "step": 20458
    },
    {
      "epoch": 1.951026916503906e-05,
      "step": 20458,
      "training_step_time": 9.657994747161865
    },
    {
      "epoch": 1.951122283935547e-05,
      "model_forward_time": 1.6817233562469482,
      "step": 20459
    },
    {
      "epoch": 1.951122283935547e-05,
      "step": 20459,
      "training_step_time": 7.871240854263306
    },
    {
      "epoch": 1.9512176513671876e-05,
      "grad_norm": 0.24769559502601624,
      "learning_rate": 9.360741903851044e-06,
      "loss": 0.0659,
      "step": 20460
    },
    {
      "epoch": 1.9512176513671876e-05,
      "model_forward_time": 1.6879780292510986,
      "step": 20460
    },
    {
      "dynamics_loss_avg": 0.07075344286859035,
      "epoch": 1.9512176513671876e-05,
      "step": 20460
    },
    {
      "action_loss_avg": 0.003555637830868363,
      "epoch": 1.9512176513671876e-05,
      "step": 20460
    },
    {
      "epoch": 1.9512176513671876e-05,
      "step": 20460,
      "training_step_time": 6.725587606430054
    },
    {
      "epoch": 1.951313018798828e-05,
      "model_forward_time": 1.3123447895050049,
      "step": 20461
    },
    {
      "epoch": 1.951313018798828e-05,
      "step": 20461,
      "training_step_time": 6.764477729797363
    },
    {
      "epoch": 1.9514083862304687e-05,
      "model_forward_time": 1.687288761138916,
      "step": 20462
    },
    {
      "epoch": 1.9514083862304687e-05,
      "step": 20462,
      "training_step_time": 6.807647228240967
    },
    {
      "epoch": 1.9515037536621095e-05,
      "model_forward_time": 1.6784932613372803,
      "step": 20463
    },
    {
      "epoch": 1.9515037536621095e-05,
      "step": 20463,
      "training_step_time": 6.783388614654541
    },
    {
      "epoch": 1.95159912109375e-05,
      "model_forward_time": 1.6819767951965332,
      "step": 20464
    },
    {
      "epoch": 1.95159912109375e-05,
      "step": 20464,
      "training_step_time": 7.400208234786987
    },
    {
      "epoch": 1.9516944885253906e-05,
      "model_forward_time": 1.696214199066162,
      "step": 20465
    },
    {
      "epoch": 1.9516944885253906e-05,
      "step": 20465,
      "training_step_time": 7.694027900695801
    },
    {
      "epoch": 1.9517898559570314e-05,
      "model_forward_time": 1.6855909824371338,
      "step": 20466
    },
    {
      "epoch": 1.9517898559570314e-05,
      "step": 20466,
      "training_step_time": 9.005757331848145
    },
    {
      "epoch": 1.9518852233886718e-05,
      "model_forward_time": 1.6991400718688965,
      "step": 20467
    },
    {
      "epoch": 1.9518852233886718e-05,
      "step": 20467,
      "training_step_time": 7.897669076919556
    },
    {
      "epoch": 1.9519805908203125e-05,
      "model_forward_time": 1.684084415435791,
      "step": 20468
    },
    {
      "epoch": 1.9519805908203125e-05,
      "step": 20468,
      "training_step_time": 7.888126373291016
    },
    {
      "epoch": 1.9520759582519532e-05,
      "model_forward_time": 1.6871819496154785,
      "step": 20469
    },
    {
      "epoch": 1.9520759582519532e-05,
      "step": 20469,
      "training_step_time": 6.801863670349121
    },
    {
      "epoch": 1.9521713256835936e-05,
      "grad_norm": 0.1829753816127777,
      "learning_rate": 9.359932719443237e-06,
      "loss": 0.0669,
      "step": 20470
    },
    {
      "epoch": 1.9521713256835936e-05,
      "model_forward_time": 1.682593822479248,
      "step": 20470
    },
    {
      "dynamics_loss_avg": 0.04817053466103971,
      "epoch": 1.9521713256835936e-05,
      "step": 20470
    },
    {
      "action_loss_avg": 0.003775730775669217,
      "epoch": 1.9521713256835936e-05,
      "step": 20470
    },
    {
      "epoch": 1.9521713256835936e-05,
      "step": 20470,
      "training_step_time": 6.75534725189209
    },
    {
      "epoch": 1.9522666931152344e-05,
      "model_forward_time": 1.6889317035675049,
      "step": 20471
    },
    {
      "epoch": 1.9522666931152344e-05,
      "step": 20471,
      "training_step_time": 6.780540227890015
    },
    {
      "epoch": 1.952362060546875e-05,
      "model_forward_time": 1.6820125579833984,
      "step": 20472
    },
    {
      "epoch": 1.952362060546875e-05,
      "step": 20472,
      "training_step_time": 6.756983518600464
    },
    {
      "epoch": 1.9524574279785155e-05,
      "model_forward_time": 1.6887938976287842,
      "step": 20473
    },
    {
      "epoch": 1.9524574279785155e-05,
      "step": 20473,
      "training_step_time": 7.480013132095337
    },
    {
      "epoch": 1.9525527954101563e-05,
      "model_forward_time": 1.696096658706665,
      "step": 20474
    },
    {
      "epoch": 1.9525527954101563e-05,
      "step": 20474,
      "training_step_time": 7.7378082275390625
    },
    {
      "epoch": 1.952648162841797e-05,
      "model_forward_time": 1.7012484073638916,
      "step": 20475
    },
    {
      "epoch": 1.952648162841797e-05,
      "step": 20475,
      "training_step_time": 7.706573963165283
    },
    {
      "epoch": 1.9527435302734374e-05,
      "model_forward_time": 2.2140536308288574,
      "step": 20476
    },
    {
      "epoch": 1.9527435302734374e-05,
      "step": 20476,
      "training_step_time": 9.412038564682007
    },
    {
      "epoch": 1.952838897705078e-05,
      "model_forward_time": 1.6849207878112793,
      "step": 20477
    },
    {
      "epoch": 1.952838897705078e-05,
      "step": 20477,
      "training_step_time": 8.837140798568726
    },
    {
      "epoch": 1.952934265136719e-05,
      "model_forward_time": 1.7044262886047363,
      "step": 20478
    },
    {
      "epoch": 1.952934265136719e-05,
      "step": 20478,
      "training_step_time": 7.73174786567688
    },
    {
      "epoch": 1.9530296325683593e-05,
      "model_forward_time": 1.7042462825775146,
      "step": 20479
    },
    {
      "epoch": 1.9530296325683593e-05,
      "step": 20479,
      "training_step_time": 6.774999141693115
    },
    {
      "epoch": 1.953125e-05,
      "grad_norm": 0.2517111599445343,
      "learning_rate": 9.35912305823974e-06,
      "loss": 0.0677,
      "step": 20480
    },
    {
      "epoch": 1.953125e-05,
      "model_forward_time": 1.7159442901611328,
      "step": 20480
    },
    {
      "dynamics_loss_avg": 0.05849465411156416,
      "epoch": 1.953125e-05,
      "step": 20480
    },
    {
      "action_loss_avg": 0.0029020964866504073,
      "epoch": 1.953125e-05,
      "step": 20480
    },
    {
      "epoch": 1.953125e-05,
      "step": 20480,
      "training_step_time": 6.796564102172852
    },
    {
      "epoch": 1.9532203674316408e-05,
      "model_forward_time": 1.7226321697235107,
      "step": 20481
    },
    {
      "epoch": 1.9532203674316408e-05,
      "step": 20481,
      "training_step_time": 6.774943113327026
    },
    {
      "epoch": 1.9533157348632812e-05,
      "model_forward_time": 1.7276990413665771,
      "step": 20482
    },
    {
      "epoch": 1.9533157348632812e-05,
      "step": 20482,
      "training_step_time": 7.3867433071136475
    },
    {
      "epoch": 1.953411102294922e-05,
      "model_forward_time": 1.730234146118164,
      "step": 20483
    },
    {
      "epoch": 1.953411102294922e-05,
      "step": 20483,
      "training_step_time": 7.746645450592041
    },
    {
      "epoch": 1.9535064697265627e-05,
      "model_forward_time": 1.3945565223693848,
      "step": 20484
    },
    {
      "epoch": 1.9535064697265627e-05,
      "step": 20484,
      "training_step_time": 8.60353970527649
    },
    {
      "epoch": 1.953601837158203e-05,
      "model_forward_time": 2.2800331115722656,
      "step": 20485
    },
    {
      "epoch": 1.953601837158203e-05,
      "step": 20485,
      "training_step_time": 10.113964557647705
    },
    {
      "epoch": 1.9536972045898438e-05,
      "model_forward_time": 1.7270228862762451,
      "step": 20486
    },
    {
      "epoch": 1.9536972045898438e-05,
      "step": 20486,
      "training_step_time": 8.805714130401611
    },
    {
      "epoch": 1.9537925720214842e-05,
      "model_forward_time": 1.7194859981536865,
      "step": 20487
    },
    {
      "epoch": 1.9537925720214842e-05,
      "step": 20487,
      "training_step_time": 7.711465120315552
    },
    {
      "epoch": 1.953887939453125e-05,
      "model_forward_time": 1.7272398471832275,
      "step": 20488
    },
    {
      "epoch": 1.953887939453125e-05,
      "step": 20488,
      "training_step_time": 6.7826197147369385
    },
    {
      "epoch": 1.9539833068847657e-05,
      "model_forward_time": 1.7244644165039062,
      "step": 20489
    },
    {
      "epoch": 1.9539833068847657e-05,
      "step": 20489,
      "training_step_time": 6.76320481300354
    },
    {
      "epoch": 1.954078674316406e-05,
      "grad_norm": 0.2144152969121933,
      "learning_rate": 9.358312920329101e-06,
      "loss": 0.0667,
      "step": 20490
    },
    {
      "epoch": 1.954078674316406e-05,
      "model_forward_time": 1.714536428451538,
      "step": 20490
    },
    {
      "dynamics_loss_avg": 0.060839705541729926,
      "epoch": 1.954078674316406e-05,
      "step": 20490
    },
    {
      "action_loss_avg": 0.0052832312183454634,
      "epoch": 1.954078674316406e-05,
      "step": 20490
    },
    {
      "epoch": 1.954078674316406e-05,
      "step": 20490,
      "training_step_time": 6.774414539337158
    },
    {
      "epoch": 1.954174041748047e-05,
      "model_forward_time": 1.697218418121338,
      "step": 20491
    },
    {
      "epoch": 1.954174041748047e-05,
      "step": 20491,
      "training_step_time": 7.436349153518677
    },
    {
      "epoch": 1.9542694091796876e-05,
      "model_forward_time": 1.7111756801605225,
      "step": 20492
    },
    {
      "epoch": 1.9542694091796876e-05,
      "step": 20492,
      "training_step_time": 7.720489740371704
    },
    {
      "epoch": 1.954364776611328e-05,
      "model_forward_time": 1.7564740180969238,
      "step": 20493
    },
    {
      "epoch": 1.954364776611328e-05,
      "step": 20493,
      "training_step_time": 9.283133745193481
    },
    {
      "epoch": 1.9544601440429687e-05,
      "model_forward_time": 2.2618649005889893,
      "step": 20494
    },
    {
      "epoch": 1.9544601440429687e-05,
      "step": 20494,
      "training_step_time": 9.501784563064575
    },
    {
      "epoch": 1.9545555114746095e-05,
      "model_forward_time": 1.7090163230895996,
      "step": 20495
    },
    {
      "epoch": 1.9545555114746095e-05,
      "step": 20495,
      "training_step_time": 8.698147773742676
    },
    {
      "epoch": 1.95465087890625e-05,
      "model_forward_time": 1.7194926738739014,
      "step": 20496
    },
    {
      "epoch": 1.95465087890625e-05,
      "step": 20496,
      "training_step_time": 7.725382328033447
    },
    {
      "epoch": 1.9547462463378906e-05,
      "model_forward_time": 1.710371971130371,
      "step": 20497
    },
    {
      "epoch": 1.9547462463378906e-05,
      "step": 20497,
      "training_step_time": 6.752687454223633
    },
    {
      "epoch": 1.9548416137695314e-05,
      "model_forward_time": 1.7066552639007568,
      "step": 20498
    },
    {
      "epoch": 1.9548416137695314e-05,
      "step": 20498,
      "training_step_time": 6.785000562667847
    },
    {
      "epoch": 1.9549369812011718e-05,
      "model_forward_time": 1.7081706523895264,
      "step": 20499
    },
    {
      "epoch": 1.9549369812011718e-05,
      "step": 20499,
      "training_step_time": 6.753312587738037
    },
    {
      "epoch": 1.9550323486328125e-05,
      "grad_norm": 0.27383702993392944,
      "learning_rate": 9.357502305799914e-06,
      "loss": 0.0681,
      "step": 20500
    },
    {
      "epoch": 1.9550323486328125e-05,
      "model_forward_time": 1.7126352787017822,
      "step": 20500
    },
    {
      "dynamics_loss_avg": 0.05996289439499378,
      "epoch": 1.9550323486328125e-05,
      "step": 20500
    },
    {
      "action_loss_avg": 0.00397207717760466,
      "epoch": 1.9550323486328125e-05,
      "step": 20500
    },
    {
      "epoch": 1.9550323486328125e-05,
      "step": 20500,
      "training_step_time": 7.427480459213257
    },
    {
      "epoch": 1.9551277160644532e-05,
      "model_forward_time": 1.7124149799346924,
      "step": 20501
    },
    {
      "epoch": 1.9551277160644532e-05,
      "step": 20501,
      "training_step_time": 7.637516736984253
    },
    {
      "epoch": 1.9552230834960936e-05,
      "model_forward_time": 1.7228591442108154,
      "step": 20502
    },
    {
      "epoch": 1.9552230834960936e-05,
      "step": 20502,
      "training_step_time": 7.643687963485718
    },
    {
      "epoch": 1.9553184509277344e-05,
      "model_forward_time": 2.2420639991760254,
      "step": 20503
    },
    {
      "epoch": 1.9553184509277344e-05,
      "step": 20503,
      "training_step_time": 8.222827672958374
    },
    {
      "epoch": 1.955413818359375e-05,
      "model_forward_time": 1.7064261436462402,
      "step": 20504
    },
    {
      "epoch": 1.955413818359375e-05,
      "step": 20504,
      "training_step_time": 8.560307502746582
    },
    {
      "epoch": 1.9555091857910155e-05,
      "model_forward_time": 1.7122700214385986,
      "step": 20505
    },
    {
      "epoch": 1.9555091857910155e-05,
      "step": 20505,
      "training_step_time": 7.8863914012908936
    },
    {
      "epoch": 1.9556045532226563e-05,
      "model_forward_time": 1.7138125896453857,
      "step": 20506
    },
    {
      "epoch": 1.9556045532226563e-05,
      "step": 20506,
      "training_step_time": 7.29104471206665
    },
    {
      "epoch": 1.955699920654297e-05,
      "model_forward_time": 1.713881254196167,
      "step": 20507
    },
    {
      "epoch": 1.955699920654297e-05,
      "step": 20507,
      "training_step_time": 6.770637512207031
    },
    {
      "epoch": 1.9557952880859374e-05,
      "model_forward_time": 1.7269725799560547,
      "step": 20508
    },
    {
      "epoch": 1.9557952880859374e-05,
      "step": 20508,
      "training_step_time": 6.7608723640441895
    },
    {
      "epoch": 1.955890655517578e-05,
      "model_forward_time": 1.7297148704528809,
      "step": 20509
    },
    {
      "epoch": 1.955890655517578e-05,
      "step": 20509,
      "training_step_time": 7.452019691467285
    },
    {
      "epoch": 1.955986022949219e-05,
      "grad_norm": 0.219765305519104,
      "learning_rate": 9.356691214740825e-06,
      "loss": 0.0661,
      "step": 20510
    },
    {
      "epoch": 1.955986022949219e-05,
      "model_forward_time": 1.719651699066162,
      "step": 20510
    },
    {
      "dynamics_loss_avg": 0.08141022473573685,
      "epoch": 1.955986022949219e-05,
      "step": 20510
    },
    {
      "action_loss_avg": 0.0035296650312375277,
      "epoch": 1.955986022949219e-05,
      "step": 20510
    },
    {
      "epoch": 1.955986022949219e-05,
      "step": 20510,
      "training_step_time": 7.58887243270874
    },
    {
      "epoch": 1.9560813903808593e-05,
      "model_forward_time": 1.7284412384033203,
      "step": 20511
    },
    {
      "epoch": 1.9560813903808593e-05,
      "step": 20511,
      "training_step_time": 7.358359098434448
    },
    {
      "epoch": 1.9561767578125e-05,
      "model_forward_time": 1.8039803504943848,
      "step": 20512
    },
    {
      "epoch": 1.9561767578125e-05,
      "step": 20512,
      "training_step_time": 8.570990800857544
    },
    {
      "epoch": 1.9562721252441408e-05,
      "model_forward_time": 2.256950616836548,
      "step": 20513
    },
    {
      "epoch": 1.9562721252441408e-05,
      "step": 20513,
      "training_step_time": 8.901004314422607
    },
    {
      "epoch": 1.9563674926757812e-05,
      "model_forward_time": 1.7284901142120361,
      "step": 20514
    },
    {
      "epoch": 1.9563674926757812e-05,
      "step": 20514,
      "training_step_time": 8.974322080612183
    },
    {
      "epoch": 1.956462860107422e-05,
      "model_forward_time": 1.7204201221466064,
      "step": 20515
    },
    {
      "epoch": 1.956462860107422e-05,
      "step": 20515,
      "training_step_time": 7.323221445083618
    },
    {
      "epoch": 1.9565582275390623e-05,
      "model_forward_time": 1.731760025024414,
      "step": 20516
    },
    {
      "epoch": 1.9565582275390623e-05,
      "step": 20516,
      "training_step_time": 6.764910936355591
    },
    {
      "epoch": 1.956653594970703e-05,
      "model_forward_time": 1.7128334045410156,
      "step": 20517
    },
    {
      "epoch": 1.956653594970703e-05,
      "step": 20517,
      "training_step_time": 6.738339424133301
    },
    {
      "epoch": 1.9567489624023438e-05,
      "model_forward_time": 1.7194571495056152,
      "step": 20518
    },
    {
      "epoch": 1.9567489624023438e-05,
      "step": 20518,
      "training_step_time": 7.413154363632202
    },
    {
      "epoch": 1.9568443298339842e-05,
      "model_forward_time": 1.7170329093933105,
      "step": 20519
    },
    {
      "epoch": 1.9568443298339842e-05,
      "step": 20519,
      "training_step_time": 8.168370723724365
    },
    {
      "epoch": 1.956939697265625e-05,
      "grad_norm": 0.1890760064125061,
      "learning_rate": 9.355879647240535e-06,
      "loss": 0.0653,
      "step": 20520
    },
    {
      "epoch": 1.956939697265625e-05,
      "model_forward_time": 1.6984236240386963,
      "step": 20520
    },
    {
      "dynamics_loss_avg": 0.0513730838894844,
      "epoch": 1.956939697265625e-05,
      "step": 20520
    },
    {
      "action_loss_avg": 0.0024150453566107897,
      "epoch": 1.956939697265625e-05,
      "step": 20520
    },
    {
      "epoch": 1.956939697265625e-05,
      "step": 20520,
      "training_step_time": 8.291277647018433
    },
    {
      "epoch": 1.9570350646972657e-05,
      "model_forward_time": 1.7892916202545166,
      "step": 20521
    },
    {
      "epoch": 1.9570350646972657e-05,
      "step": 20521,
      "training_step_time": 9.752336025238037
    },
    {
      "epoch": 1.957130432128906e-05,
      "model_forward_time": 2.264655351638794,
      "step": 20522
    },
    {
      "epoch": 1.957130432128906e-05,
      "step": 20522,
      "training_step_time": 8.420644760131836
    },
    {
      "epoch": 1.957225799560547e-05,
      "model_forward_time": 1.7174224853515625,
      "step": 20523
    },
    {
      "epoch": 1.957225799560547e-05,
      "step": 20523,
      "training_step_time": 8.755258321762085
    },
    {
      "epoch": 1.9573211669921876e-05,
      "model_forward_time": 1.6974222660064697,
      "step": 20524
    },
    {
      "epoch": 1.9573211669921876e-05,
      "step": 20524,
      "training_step_time": 7.577682256698608
    },
    {
      "epoch": 1.957416534423828e-05,
      "model_forward_time": 1.723297357559204,
      "step": 20525
    },
    {
      "epoch": 1.957416534423828e-05,
      "step": 20525,
      "training_step_time": 6.772398948669434
    },
    {
      "epoch": 1.9575119018554687e-05,
      "model_forward_time": 1.3513858318328857,
      "step": 20526
    },
    {
      "epoch": 1.9575119018554687e-05,
      "step": 20526,
      "training_step_time": 6.793884038925171
    },
    {
      "epoch": 1.9576072692871095e-05,
      "model_forward_time": 1.722442865371704,
      "step": 20527
    },
    {
      "epoch": 1.9576072692871095e-05,
      "step": 20527,
      "training_step_time": 6.79296875
    },
    {
      "epoch": 1.95770263671875e-05,
      "model_forward_time": 1.754086971282959,
      "step": 20528
    },
    {
      "epoch": 1.95770263671875e-05,
      "step": 20528,
      "training_step_time": 7.633162498474121
    },
    {
      "epoch": 1.9577980041503906e-05,
      "model_forward_time": 1.733877420425415,
      "step": 20529
    },
    {
      "epoch": 1.9577980041503906e-05,
      "step": 20529,
      "training_step_time": 7.682938575744629
    },
    {
      "epoch": 1.9578933715820314e-05,
      "grad_norm": 0.16811072826385498,
      "learning_rate": 9.355067603387797e-06,
      "loss": 0.0683,
      "step": 20530
    },
    {
      "epoch": 1.9578933715820314e-05,
      "model_forward_time": 1.786527395248413,
      "step": 20530
    },
    {
      "dynamics_loss_avg": 0.06566468831151724,
      "epoch": 1.9578933715820314e-05,
      "step": 20530
    },
    {
      "action_loss_avg": 0.003184376144781709,
      "epoch": 1.9578933715820314e-05,
      "step": 20530
    },
    {
      "epoch": 1.9578933715820314e-05,
      "step": 20530,
      "training_step_time": 7.687174558639526
    },
    {
      "epoch": 1.9579887390136718e-05,
      "model_forward_time": 2.288949966430664,
      "step": 20531
    },
    {
      "epoch": 1.9579887390136718e-05,
      "step": 20531,
      "training_step_time": 10.235742330551147
    },
    {
      "epoch": 1.9580841064453125e-05,
      "model_forward_time": 1.723393440246582,
      "step": 20532
    },
    {
      "epoch": 1.9580841064453125e-05,
      "step": 20532,
      "training_step_time": 7.821623802185059
    },
    {
      "epoch": 1.9581794738769532e-05,
      "model_forward_time": 1.354628086090088,
      "step": 20533
    },
    {
      "epoch": 1.9581794738769532e-05,
      "step": 20533,
      "training_step_time": 7.9573469161987305
    },
    {
      "epoch": 1.9582748413085936e-05,
      "model_forward_time": 1.7233960628509521,
      "step": 20534
    },
    {
      "epoch": 1.9582748413085936e-05,
      "step": 20534,
      "training_step_time": 6.684743404388428
    },
    {
      "epoch": 1.9583702087402344e-05,
      "model_forward_time": 1.7255196571350098,
      "step": 20535
    },
    {
      "epoch": 1.9583702087402344e-05,
      "step": 20535,
      "training_step_time": 6.825352191925049
    },
    {
      "epoch": 1.958465576171875e-05,
      "model_forward_time": 1.7265949249267578,
      "step": 20536
    },
    {
      "epoch": 1.958465576171875e-05,
      "step": 20536,
      "training_step_time": 6.801437616348267
    },
    {
      "epoch": 1.9585609436035155e-05,
      "model_forward_time": 1.7272162437438965,
      "step": 20537
    },
    {
      "epoch": 1.9585609436035155e-05,
      "step": 20537,
      "training_step_time": 7.610432863235474
    },
    {
      "epoch": 1.9586563110351563e-05,
      "model_forward_time": 1.7290434837341309,
      "step": 20538
    },
    {
      "epoch": 1.9586563110351563e-05,
      "step": 20538,
      "training_step_time": 7.683305501937866
    },
    {
      "epoch": 1.958751678466797e-05,
      "model_forward_time": 1.7530066967010498,
      "step": 20539
    },
    {
      "epoch": 1.958751678466797e-05,
      "step": 20539,
      "training_step_time": 9.271553039550781
    },
    {
      "epoch": 1.9588470458984374e-05,
      "grad_norm": 0.1694677770137787,
      "learning_rate": 9.354255083271412e-06,
      "loss": 0.0667,
      "step": 20540
    },
    {
      "epoch": 1.9588470458984374e-05,
      "model_forward_time": 2.262105703353882,
      "step": 20540
    },
    {
      "dynamics_loss_avg": 0.06679800897836685,
      "epoch": 1.9588470458984374e-05,
      "step": 20540
    },
    {
      "action_loss_avg": 0.0040392215247266,
      "epoch": 1.9588470458984374e-05,
      "step": 20540
    },
    {
      "epoch": 1.9588470458984374e-05,
      "step": 20540,
      "training_step_time": 9.34158205986023
    },
    {
      "epoch": 1.958942413330078e-05,
      "model_forward_time": 1.7013986110687256,
      "step": 20541
    },
    {
      "epoch": 1.958942413330078e-05,
      "step": 20541,
      "training_step_time": 10.705537796020508
    },
    {
      "epoch": 1.959037780761719e-05,
      "model_forward_time": 1.7197363376617432,
      "step": 20542
    },
    {
      "epoch": 1.959037780761719e-05,
      "step": 20542,
      "training_step_time": 7.921809196472168
    },
    {
      "epoch": 1.9591331481933593e-05,
      "model_forward_time": 1.7193799018859863,
      "step": 20543
    },
    {
      "epoch": 1.9591331481933593e-05,
      "step": 20543,
      "training_step_time": 7.250053644180298
    },
    {
      "epoch": 1.959228515625e-05,
      "model_forward_time": 1.3369855880737305,
      "step": 20544
    },
    {
      "epoch": 1.959228515625e-05,
      "step": 20544,
      "training_step_time": 6.78259801864624
    },
    {
      "epoch": 1.9593238830566408e-05,
      "model_forward_time": 1.7170040607452393,
      "step": 20545
    },
    {
      "epoch": 1.9593238830566408e-05,
      "step": 20545,
      "training_step_time": 6.754150629043579
    },
    {
      "epoch": 1.9594192504882812e-05,
      "model_forward_time": 1.7177011966705322,
      "step": 20546
    },
    {
      "epoch": 1.9594192504882812e-05,
      "step": 20546,
      "training_step_time": 7.659705400466919
    },
    {
      "epoch": 1.959514617919922e-05,
      "model_forward_time": 1.3435206413269043,
      "step": 20547
    },
    {
      "epoch": 1.959514617919922e-05,
      "step": 20547,
      "training_step_time": 7.685807228088379
    },
    {
      "epoch": 1.9596099853515623e-05,
      "model_forward_time": 1.7358744144439697,
      "step": 20548
    },
    {
      "epoch": 1.9596099853515623e-05,
      "step": 20548,
      "training_step_time": 7.713615894317627
    },
    {
      "epoch": 1.959705352783203e-05,
      "model_forward_time": 1.7193055152893066,
      "step": 20549
    },
    {
      "epoch": 1.959705352783203e-05,
      "step": 20549,
      "training_step_time": 9.088088989257812
    },
    {
      "epoch": 1.9598007202148438e-05,
      "grad_norm": 0.22527596354484558,
      "learning_rate": 9.35344208698024e-06,
      "loss": 0.0669,
      "step": 20550
    },
    {
      "epoch": 1.9598007202148438e-05,
      "model_forward_time": 2.2415194511413574,
      "step": 20550
    },
    {
      "dynamics_loss_avg": 0.06558574233204126,
      "epoch": 1.9598007202148438e-05,
      "step": 20550
    },
    {
      "action_loss_avg": 0.003618518076837063,
      "epoch": 1.9598007202148438e-05,
      "step": 20550
    },
    {
      "epoch": 1.9598007202148438e-05,
      "step": 20550,
      "training_step_time": 8.633629322052002
    },
    {
      "epoch": 1.9598960876464842e-05,
      "model_forward_time": 1.7169651985168457,
      "step": 20551
    },
    {
      "epoch": 1.9598960876464842e-05,
      "step": 20551,
      "training_step_time": 7.797986745834351
    },
    {
      "epoch": 1.959991455078125e-05,
      "model_forward_time": 1.719043254852295,
      "step": 20552
    },
    {
      "epoch": 1.959991455078125e-05,
      "step": 20552,
      "training_step_time": 7.564922571182251
    },
    {
      "epoch": 1.9600868225097657e-05,
      "model_forward_time": 1.722273349761963,
      "step": 20553
    },
    {
      "epoch": 1.9600868225097657e-05,
      "step": 20553,
      "training_step_time": 7.2249133586883545
    },
    {
      "epoch": 1.960182189941406e-05,
      "model_forward_time": 1.7195687294006348,
      "step": 20554
    },
    {
      "epoch": 1.960182189941406e-05,
      "step": 20554,
      "training_step_time": 6.73933219909668
    },
    {
      "epoch": 1.960277557373047e-05,
      "model_forward_time": 1.7159538269042969,
      "step": 20555
    },
    {
      "epoch": 1.960277557373047e-05,
      "step": 20555,
      "training_step_time": 7.588072776794434
    },
    {
      "epoch": 1.9603729248046876e-05,
      "model_forward_time": 1.7233805656433105,
      "step": 20556
    },
    {
      "epoch": 1.9603729248046876e-05,
      "step": 20556,
      "training_step_time": 7.649188995361328
    },
    {
      "epoch": 1.960468292236328e-05,
      "model_forward_time": 1.7230682373046875,
      "step": 20557
    },
    {
      "epoch": 1.960468292236328e-05,
      "step": 20557,
      "training_step_time": 7.984545469284058
    },
    {
      "epoch": 1.9605636596679687e-05,
      "model_forward_time": 1.716426134109497,
      "step": 20558
    },
    {
      "epoch": 1.9605636596679687e-05,
      "step": 20558,
      "training_step_time": 8.609201431274414
    },
    {
      "epoch": 1.9606590270996095e-05,
      "model_forward_time": 2.2758171558380127,
      "step": 20559
    },
    {
      "epoch": 1.9606590270996095e-05,
      "step": 20559,
      "training_step_time": 7.686615228652954
    },
    {
      "epoch": 1.96075439453125e-05,
      "grad_norm": 0.20488551259040833,
      "learning_rate": 9.352628614603186e-06,
      "loss": 0.0656,
      "step": 20560
    },
    {
      "epoch": 1.96075439453125e-05,
      "model_forward_time": 1.7117831707000732,
      "step": 20560
    },
    {
      "dynamics_loss_avg": 0.0673245944082737,
      "epoch": 1.96075439453125e-05,
      "step": 20560
    },
    {
      "action_loss_avg": 0.002567218168405816,
      "epoch": 1.96075439453125e-05,
      "step": 20560
    },
    {
      "epoch": 1.96075439453125e-05,
      "step": 20560,
      "training_step_time": 7.8345677852630615
    },
    {
      "epoch": 1.9608497619628906e-05,
      "model_forward_time": 1.7226173877716064,
      "step": 20561
    },
    {
      "epoch": 1.9608497619628906e-05,
      "step": 20561,
      "training_step_time": 7.678932189941406
    },
    {
      "epoch": 1.9609451293945314e-05,
      "model_forward_time": 1.7181663513183594,
      "step": 20562
    },
    {
      "epoch": 1.9609451293945314e-05,
      "step": 20562,
      "training_step_time": 7.250255346298218
    },
    {
      "epoch": 1.9610404968261718e-05,
      "model_forward_time": 1.7263972759246826,
      "step": 20563
    },
    {
      "epoch": 1.9610404968261718e-05,
      "step": 20563,
      "training_step_time": 6.744728326797485
    },
    {
      "epoch": 1.9611358642578125e-05,
      "model_forward_time": 1.7121403217315674,
      "step": 20564
    },
    {
      "epoch": 1.9611358642578125e-05,
      "step": 20564,
      "training_step_time": 7.616052865982056
    },
    {
      "epoch": 1.9612312316894532e-05,
      "model_forward_time": 1.7185642719268799,
      "step": 20565
    },
    {
      "epoch": 1.9612312316894532e-05,
      "step": 20565,
      "training_step_time": 7.6177215576171875
    },
    {
      "epoch": 1.9613265991210936e-05,
      "model_forward_time": 1.7162294387817383,
      "step": 20566
    },
    {
      "epoch": 1.9613265991210936e-05,
      "step": 20566,
      "training_step_time": 8.434844255447388
    },
    {
      "epoch": 1.9614219665527344e-05,
      "model_forward_time": 1.7753682136535645,
      "step": 20567
    },
    {
      "epoch": 1.9614219665527344e-05,
      "step": 20567,
      "training_step_time": 7.7743141651153564
    },
    {
      "epoch": 1.961517333984375e-05,
      "model_forward_time": 2.2659502029418945,
      "step": 20568
    },
    {
      "epoch": 1.961517333984375e-05,
      "step": 20568,
      "training_step_time": 9.614564657211304
    },
    {
      "epoch": 1.9616127014160155e-05,
      "model_forward_time": 1.716062307357788,
      "step": 20569
    },
    {
      "epoch": 1.9616127014160155e-05,
      "step": 20569,
      "training_step_time": 8.912625074386597
    },
    {
      "epoch": 1.9617080688476563e-05,
      "grad_norm": 0.145818293094635,
      "learning_rate": 9.351814666229208e-06,
      "loss": 0.0601,
      "step": 20570
    },
    {
      "epoch": 1.9617080688476563e-05,
      "model_forward_time": 1.7144124507904053,
      "step": 20570
    },
    {
      "dynamics_loss_avg": 0.060033072531223294,
      "epoch": 1.9617080688476563e-05,
      "step": 20570
    },
    {
      "action_loss_avg": 0.0023098438512533905,
      "epoch": 1.9617080688476563e-05,
      "step": 20570
    },
    {
      "epoch": 1.9617080688476563e-05,
      "step": 20570,
      "training_step_time": 7.815501928329468
    },
    {
      "epoch": 1.961803436279297e-05,
      "model_forward_time": 1.717804193496704,
      "step": 20571
    },
    {
      "epoch": 1.961803436279297e-05,
      "step": 20571,
      "training_step_time": 7.277881860733032
    },
    {
      "epoch": 1.9618988037109374e-05,
      "model_forward_time": 1.7151134014129639,
      "step": 20572
    },
    {
      "epoch": 1.9618988037109374e-05,
      "step": 20572,
      "training_step_time": 6.766616582870483
    },
    {
      "epoch": 1.961994171142578e-05,
      "model_forward_time": 1.7186179161071777,
      "step": 20573
    },
    {
      "epoch": 1.961994171142578e-05,
      "step": 20573,
      "training_step_time": 7.597115993499756
    },
    {
      "epoch": 1.962089538574219e-05,
      "model_forward_time": 1.726255178451538,
      "step": 20574
    },
    {
      "epoch": 1.962089538574219e-05,
      "step": 20574,
      "training_step_time": 7.542071104049683
    },
    {
      "epoch": 1.9621849060058593e-05,
      "model_forward_time": 1.7171077728271484,
      "step": 20575
    },
    {
      "epoch": 1.9621849060058593e-05,
      "step": 20575,
      "training_step_time": 8.120072841644287
    },
    {
      "epoch": 1.9622802734375e-05,
      "model_forward_time": 1.7615962028503418,
      "step": 20576
    },
    {
      "epoch": 1.9622802734375e-05,
      "step": 20576,
      "training_step_time": 8.722549438476562
    },
    {
      "epoch": 1.9623756408691408e-05,
      "model_forward_time": 1.7374796867370605,
      "step": 20577
    },
    {
      "epoch": 1.9623756408691408e-05,
      "step": 20577,
      "training_step_time": 8.220990419387817
    },
    {
      "epoch": 1.9624710083007812e-05,
      "model_forward_time": 1.729588508605957,
      "step": 20578
    },
    {
      "epoch": 1.9624710083007812e-05,
      "step": 20578,
      "training_step_time": 8.606820106506348
    },
    {
      "epoch": 1.962566375732422e-05,
      "model_forward_time": 1.731299638748169,
      "step": 20579
    },
    {
      "epoch": 1.962566375732422e-05,
      "step": 20579,
      "training_step_time": 7.813318252563477
    },
    {
      "epoch": 1.9626617431640623e-05,
      "grad_norm": 0.146469384431839,
      "learning_rate": 9.351000241947324e-06,
      "loss": 0.0658,
      "step": 20580
    },
    {
      "epoch": 1.9626617431640623e-05,
      "model_forward_time": 1.8061044216156006,
      "step": 20580
    },
    {
      "dynamics_loss_avg": 0.0625115854665637,
      "epoch": 1.9626617431640623e-05,
      "step": 20580
    },
    {
      "action_loss_avg": 0.002932318695820868,
      "epoch": 1.9626617431640623e-05,
      "step": 20580
    },
    {
      "epoch": 1.9626617431640623e-05,
      "step": 20580,
      "training_step_time": 7.385570526123047
    },
    {
      "epoch": 1.962757110595703e-05,
      "model_forward_time": 1.8123395442962646,
      "step": 20581
    },
    {
      "epoch": 1.962757110595703e-05,
      "step": 20581,
      "training_step_time": 6.811086177825928
    },
    {
      "epoch": 1.9628524780273438e-05,
      "model_forward_time": 1.8194210529327393,
      "step": 20582
    },
    {
      "epoch": 1.9628524780273438e-05,
      "step": 20582,
      "training_step_time": 7.599138021469116
    },
    {
      "epoch": 1.9629478454589842e-05,
      "model_forward_time": 1.8149926662445068,
      "step": 20583
    },
    {
      "epoch": 1.9629478454589842e-05,
      "step": 20583,
      "training_step_time": 7.4525227546691895
    },
    {
      "epoch": 1.963043212890625e-05,
      "model_forward_time": 1.8243114948272705,
      "step": 20584
    },
    {
      "epoch": 1.963043212890625e-05,
      "step": 20584,
      "training_step_time": 7.66161322593689
    },
    {
      "epoch": 1.9631385803222657e-05,
      "model_forward_time": 1.8266029357910156,
      "step": 20585
    },
    {
      "epoch": 1.9631385803222657e-05,
      "step": 20585,
      "training_step_time": 9.2389235496521
    },
    {
      "epoch": 1.963233947753906e-05,
      "model_forward_time": 1.8299839496612549,
      "step": 20586
    },
    {
      "epoch": 1.963233947753906e-05,
      "step": 20586,
      "training_step_time": 9.704151391983032
    },
    {
      "epoch": 1.963329315185547e-05,
      "model_forward_time": 2.336545705795288,
      "step": 20587
    },
    {
      "epoch": 1.963329315185547e-05,
      "step": 20587,
      "training_step_time": 8.639671802520752
    },
    {
      "epoch": 1.9634246826171876e-05,
      "model_forward_time": 1.8295557498931885,
      "step": 20588
    },
    {
      "epoch": 1.9634246826171876e-05,
      "step": 20588,
      "training_step_time": 7.91969108581543
    },
    {
      "epoch": 1.963520050048828e-05,
      "model_forward_time": 1.8310229778289795,
      "step": 20589
    },
    {
      "epoch": 1.963520050048828e-05,
      "step": 20589,
      "training_step_time": 7.6649463176727295
    },
    {
      "epoch": 1.9636154174804687e-05,
      "grad_norm": 0.14832191169261932,
      "learning_rate": 9.350185341846593e-06,
      "loss": 0.0599,
      "step": 20590
    },
    {
      "epoch": 1.9636154174804687e-05,
      "model_forward_time": 1.832444429397583,
      "step": 20590
    },
    {
      "dynamics_loss_avg": 0.06889883205294609,
      "epoch": 1.9636154174804687e-05,
      "step": 20590
    },
    {
      "action_loss_avg": 0.0038786296965554355,
      "epoch": 1.9636154174804687e-05,
      "step": 20590
    },
    {
      "epoch": 1.9636154174804687e-05,
      "step": 20590,
      "training_step_time": 6.774212121963501
    },
    {
      "epoch": 1.9637107849121095e-05,
      "model_forward_time": 1.8258559703826904,
      "step": 20591
    },
    {
      "epoch": 1.9637107849121095e-05,
      "step": 20591,
      "training_step_time": 7.584144353866577
    },
    {
      "epoch": 1.96380615234375e-05,
      "model_forward_time": 1.8265552520751953,
      "step": 20592
    },
    {
      "epoch": 1.96380615234375e-05,
      "step": 20592,
      "training_step_time": 7.42235541343689
    },
    {
      "epoch": 1.9639015197753906e-05,
      "model_forward_time": 1.8203215599060059,
      "step": 20593
    },
    {
      "epoch": 1.9639015197753906e-05,
      "step": 20593,
      "training_step_time": 8.332134008407593
    },
    {
      "epoch": 1.9639968872070314e-05,
      "model_forward_time": 1.8176970481872559,
      "step": 20594
    },
    {
      "epoch": 1.9639968872070314e-05,
      "step": 20594,
      "training_step_time": 7.680231094360352
    },
    {
      "epoch": 1.9640922546386718e-05,
      "model_forward_time": 1.8196871280670166,
      "step": 20595
    },
    {
      "epoch": 1.9640922546386718e-05,
      "step": 20595,
      "training_step_time": 8.455061197280884
    },
    {
      "epoch": 1.9641876220703125e-05,
      "model_forward_time": 2.365082263946533,
      "step": 20596
    },
    {
      "epoch": 1.9641876220703125e-05,
      "step": 20596,
      "training_step_time": 9.602265119552612
    },
    {
      "epoch": 1.9642829895019532e-05,
      "model_forward_time": 1.820343017578125,
      "step": 20597
    },
    {
      "epoch": 1.9642829895019532e-05,
      "step": 20597,
      "training_step_time": 7.844465017318726
    },
    {
      "epoch": 1.9643783569335937e-05,
      "model_forward_time": 1.820258617401123,
      "step": 20598
    },
    {
      "epoch": 1.9643783569335937e-05,
      "step": 20598,
      "training_step_time": 7.723573923110962
    },
    {
      "epoch": 1.9644737243652344e-05,
      "model_forward_time": 1.8202755451202393,
      "step": 20599
    },
    {
      "epoch": 1.9644737243652344e-05,
      "step": 20599,
      "training_step_time": 6.7578442096710205
    },
    {
      "epoch": 1.964569091796875e-05,
      "grad_norm": 0.18113040924072266,
      "learning_rate": 9.349369966016135e-06,
      "loss": 0.0598,
      "step": 20600
    },
    {
      "epoch": 1.964569091796875e-05,
      "model_forward_time": 1.4627580642700195,
      "step": 20600
    },
    {
      "dynamics_loss_avg": 0.04628091538324952,
      "epoch": 1.964569091796875e-05,
      "step": 20600
    },
    {
      "action_loss_avg": 0.0032209060445893556,
      "epoch": 1.964569091796875e-05,
      "step": 20600
    },
    {
      "epoch": 1.964569091796875e-05,
      "step": 20600,
      "training_step_time": 7.07008957862854
    },
    {
      "epoch": 1.9646644592285155e-05,
      "model_forward_time": 1.8166370391845703,
      "step": 20601
    },
    {
      "epoch": 1.9646644592285155e-05,
      "step": 20601,
      "training_step_time": 7.605445861816406
    },
    {
      "epoch": 1.9647598266601563e-05,
      "model_forward_time": 1.82356595993042,
      "step": 20602
    },
    {
      "epoch": 1.9647598266601563e-05,
      "step": 20602,
      "training_step_time": 7.5891876220703125
    },
    {
      "epoch": 1.964855194091797e-05,
      "model_forward_time": 1.815307855606079,
      "step": 20603
    },
    {
      "epoch": 1.964855194091797e-05,
      "step": 20603,
      "training_step_time": 7.6310789585113525
    },
    {
      "epoch": 1.9649505615234374e-05,
      "model_forward_time": 1.8224282264709473,
      "step": 20604
    },
    {
      "epoch": 1.9649505615234374e-05,
      "step": 20604,
      "training_step_time": 9.549395561218262
    },
    {
      "epoch": 1.965045928955078e-05,
      "model_forward_time": 2.371946334838867,
      "step": 20605
    },
    {
      "epoch": 1.965045928955078e-05,
      "step": 20605,
      "training_step_time": 8.656411170959473
    },
    {
      "epoch": 1.965141296386719e-05,
      "model_forward_time": 1.8272488117218018,
      "step": 20606
    },
    {
      "epoch": 1.965141296386719e-05,
      "step": 20606,
      "training_step_time": 8.875083208084106
    },
    {
      "epoch": 1.9652366638183593e-05,
      "model_forward_time": 1.819371223449707,
      "step": 20607
    },
    {
      "epoch": 1.9652366638183593e-05,
      "step": 20607,
      "training_step_time": 7.612935781478882
    },
    {
      "epoch": 1.96533203125e-05,
      "model_forward_time": 1.8231985569000244,
      "step": 20608
    },
    {
      "epoch": 1.96533203125e-05,
      "step": 20608,
      "training_step_time": 6.725060224533081
    },
    {
      "epoch": 1.9654273986816408e-05,
      "model_forward_time": 1.8257691860198975,
      "step": 20609
    },
    {
      "epoch": 1.9654273986816408e-05,
      "step": 20609,
      "training_step_time": 7.125290870666504
    },
    {
      "epoch": 1.9655227661132812e-05,
      "grad_norm": 0.12422170490026474,
      "learning_rate": 9.348554114545116e-06,
      "loss": 0.0636,
      "step": 20610
    },
    {
      "epoch": 1.9655227661132812e-05,
      "model_forward_time": 1.8004117012023926,
      "step": 20610
    },
    {
      "dynamics_loss_avg": 0.06935592629015445,
      "epoch": 1.9655227661132812e-05,
      "step": 20610
    },
    {
      "action_loss_avg": 0.0035316161112859843,
      "epoch": 1.9655227661132812e-05,
      "step": 20610
    },
    {
      "epoch": 1.9655227661132812e-05,
      "step": 20610,
      "training_step_time": 7.627321481704712
    },
    {
      "epoch": 1.965618133544922e-05,
      "model_forward_time": 1.8226287364959717,
      "step": 20611
    },
    {
      "epoch": 1.965618133544922e-05,
      "step": 20611,
      "training_step_time": 8.15499997138977
    },
    {
      "epoch": 1.9657135009765623e-05,
      "model_forward_time": 1.8201100826263428,
      "step": 20612
    },
    {
      "epoch": 1.9657135009765623e-05,
      "step": 20612,
      "training_step_time": 7.6034095287323
    },
    {
      "epoch": 1.965808868408203e-05,
      "model_forward_time": 1.829294204711914,
      "step": 20613
    },
    {
      "epoch": 1.965808868408203e-05,
      "step": 20613,
      "training_step_time": 7.699400424957275
    },
    {
      "epoch": 1.9659042358398438e-05,
      "model_forward_time": 2.356471300125122,
      "step": 20614
    },
    {
      "epoch": 1.9659042358398438e-05,
      "step": 20614,
      "training_step_time": 8.562778949737549
    },
    {
      "epoch": 1.9659996032714842e-05,
      "model_forward_time": 1.8272500038146973,
      "step": 20615
    },
    {
      "epoch": 1.9659996032714842e-05,
      "step": 20615,
      "training_step_time": 8.658094644546509
    },
    {
      "epoch": 1.966094970703125e-05,
      "model_forward_time": 1.8254516124725342,
      "step": 20616
    },
    {
      "epoch": 1.966094970703125e-05,
      "step": 20616,
      "training_step_time": 8.393566608428955
    },
    {
      "epoch": 1.9661903381347657e-05,
      "model_forward_time": 1.821873426437378,
      "step": 20617
    },
    {
      "epoch": 1.9661903381347657e-05,
      "step": 20617,
      "training_step_time": 7.352113723754883
    },
    {
      "epoch": 1.966285705566406e-05,
      "model_forward_time": 1.8210644721984863,
      "step": 20618
    },
    {
      "epoch": 1.966285705566406e-05,
      "step": 20618,
      "training_step_time": 7.113375425338745
    },
    {
      "epoch": 1.966381072998047e-05,
      "model_forward_time": 1.8216373920440674,
      "step": 20619
    },
    {
      "epoch": 1.966381072998047e-05,
      "step": 20619,
      "training_step_time": 7.672089576721191
    },
    {
      "epoch": 1.9664764404296876e-05,
      "grad_norm": 0.19564500451087952,
      "learning_rate": 9.347737787522759e-06,
      "loss": 0.0662,
      "step": 20620
    },
    {
      "epoch": 1.9664764404296876e-05,
      "model_forward_time": 1.8179852962493896,
      "step": 20620
    },
    {
      "dynamics_loss_avg": 0.06468548253178596,
      "epoch": 1.9664764404296876e-05,
      "step": 20620
    },
    {
      "action_loss_avg": 0.0023331248783506455,
      "epoch": 1.9664764404296876e-05,
      "step": 20620
    },
    {
      "epoch": 1.9664764404296876e-05,
      "step": 20620,
      "training_step_time": 7.5002281665802
    },
    {
      "epoch": 1.966571807861328e-05,
      "model_forward_time": 1.819352626800537,
      "step": 20621
    },
    {
      "epoch": 1.966571807861328e-05,
      "step": 20621,
      "training_step_time": 7.632277011871338
    },
    {
      "epoch": 1.9666671752929687e-05,
      "model_forward_time": 1.8074679374694824,
      "step": 20622
    },
    {
      "epoch": 1.9666671752929687e-05,
      "step": 20622,
      "training_step_time": 7.6633195877075195
    },
    {
      "epoch": 1.9667625427246095e-05,
      "model_forward_time": 1.818856954574585,
      "step": 20623
    },
    {
      "epoch": 1.9667625427246095e-05,
      "step": 20623,
      "training_step_time": 8.444674253463745
    },
    {
      "epoch": 1.96685791015625e-05,
      "model_forward_time": 2.3501980304718018,
      "step": 20624
    },
    {
      "epoch": 1.96685791015625e-05,
      "step": 20624,
      "training_step_time": 9.525614738464355
    },
    {
      "epoch": 1.9669532775878906e-05,
      "model_forward_time": 1.8250062465667725,
      "step": 20625
    },
    {
      "epoch": 1.9669532775878906e-05,
      "step": 20625,
      "training_step_time": 8.448484420776367
    },
    {
      "epoch": 1.9670486450195314e-05,
      "model_forward_time": 1.8188273906707764,
      "step": 20626
    },
    {
      "epoch": 1.9670486450195314e-05,
      "step": 20626,
      "training_step_time": 7.366865634918213
    },
    {
      "epoch": 1.9671440124511718e-05,
      "model_forward_time": 1.4577791690826416,
      "step": 20627
    },
    {
      "epoch": 1.9671440124511718e-05,
      "step": 20627,
      "training_step_time": 7.110698461532593
    },
    {
      "epoch": 1.9672393798828125e-05,
      "model_forward_time": 1.8235645294189453,
      "step": 20628
    },
    {
      "epoch": 1.9672393798828125e-05,
      "step": 20628,
      "training_step_time": 7.594627618789673
    },
    {
      "epoch": 1.9673347473144533e-05,
      "model_forward_time": 1.8295059204101562,
      "step": 20629
    },
    {
      "epoch": 1.9673347473144533e-05,
      "step": 20629,
      "training_step_time": 7.440212249755859
    },
    {
      "epoch": 1.9674301147460937e-05,
      "grad_norm": 0.14162515103816986,
      "learning_rate": 9.346920985038333e-06,
      "loss": 0.0654,
      "step": 20630
    },
    {
      "epoch": 1.9674301147460937e-05,
      "model_forward_time": 1.8278729915618896,
      "step": 20630
    },
    {
      "dynamics_loss_avg": 0.05347215868532658,
      "epoch": 1.9674301147460937e-05,
      "step": 20630
    },
    {
      "action_loss_avg": 0.0032953088113572447,
      "epoch": 1.9674301147460937e-05,
      "step": 20630
    },
    {
      "epoch": 1.9674301147460937e-05,
      "step": 20630,
      "training_step_time": 8.50230598449707
    },
    {
      "epoch": 1.9675254821777344e-05,
      "model_forward_time": 1.8336513042449951,
      "step": 20631
    },
    {
      "epoch": 1.9675254821777344e-05,
      "step": 20631,
      "training_step_time": 8.250357866287231
    },
    {
      "epoch": 1.967620849609375e-05,
      "model_forward_time": 1.8196678161621094,
      "step": 20632
    },
    {
      "epoch": 1.967620849609375e-05,
      "step": 20632,
      "training_step_time": 10.267935752868652
    },
    {
      "epoch": 1.9677162170410155e-05,
      "model_forward_time": 2.351422071456909,
      "step": 20633
    },
    {
      "epoch": 1.9677162170410155e-05,
      "step": 20633,
      "training_step_time": 7.643366813659668
    },
    {
      "epoch": 1.9678115844726563e-05,
      "model_forward_time": 1.8257997035980225,
      "step": 20634
    },
    {
      "epoch": 1.9678115844726563e-05,
      "step": 20634,
      "training_step_time": 7.874395132064819
    },
    {
      "epoch": 1.967906951904297e-05,
      "model_forward_time": 1.830195665359497,
      "step": 20635
    },
    {
      "epoch": 1.967906951904297e-05,
      "step": 20635,
      "training_step_time": 7.597418546676636
    },
    {
      "epoch": 1.9680023193359374e-05,
      "model_forward_time": 1.8277430534362793,
      "step": 20636
    },
    {
      "epoch": 1.9680023193359374e-05,
      "step": 20636,
      "training_step_time": 6.779052019119263
    },
    {
      "epoch": 1.968097686767578e-05,
      "model_forward_time": 1.827876329421997,
      "step": 20637
    },
    {
      "epoch": 1.968097686767578e-05,
      "step": 20637,
      "training_step_time": 7.552083969116211
    },
    {
      "epoch": 1.968193054199219e-05,
      "model_forward_time": 1.8274545669555664,
      "step": 20638
    },
    {
      "epoch": 1.968193054199219e-05,
      "step": 20638,
      "training_step_time": 7.127542495727539
    },
    {
      "epoch": 1.9682884216308593e-05,
      "model_forward_time": 1.8227856159210205,
      "step": 20639
    },
    {
      "epoch": 1.9682884216308593e-05,
      "step": 20639,
      "training_step_time": 7.562329530715942
    },
    {
      "epoch": 1.9683837890625e-05,
      "grad_norm": 0.15625417232513428,
      "learning_rate": 9.346103707181162e-06,
      "loss": 0.061,
      "step": 20640
    },
    {
      "epoch": 1.9683837890625e-05,
      "model_forward_time": 1.8273334503173828,
      "step": 20640
    },
    {
      "dynamics_loss_avg": 0.051150822825729844,
      "epoch": 1.9683837890625e-05,
      "step": 20640
    },
    {
      "action_loss_avg": 0.00243650795891881,
      "epoch": 1.9683837890625e-05,
      "step": 20640
    },
    {
      "epoch": 1.9683837890625e-05,
      "step": 20640,
      "training_step_time": 8.63913083076477
    },
    {
      "epoch": 1.9684791564941405e-05,
      "model_forward_time": 1.8231263160705566,
      "step": 20641
    },
    {
      "epoch": 1.9684791564941405e-05,
      "step": 20641,
      "training_step_time": 10.27881932258606
    },
    {
      "epoch": 1.9685745239257812e-05,
      "model_forward_time": 2.3384323120117188,
      "step": 20642
    },
    {
      "epoch": 1.9685745239257812e-05,
      "step": 20642,
      "training_step_time": 8.367884397506714
    },
    {
      "epoch": 1.968669891357422e-05,
      "model_forward_time": 1.8279731273651123,
      "step": 20643
    },
    {
      "epoch": 1.968669891357422e-05,
      "step": 20643,
      "training_step_time": 7.819153070449829
    },
    {
      "epoch": 1.9687652587890623e-05,
      "model_forward_time": 1.825915813446045,
      "step": 20644
    },
    {
      "epoch": 1.9687652587890623e-05,
      "step": 20644,
      "training_step_time": 7.673283576965332
    },
    {
      "epoch": 1.968860626220703e-05,
      "model_forward_time": 1.8208410739898682,
      "step": 20645
    },
    {
      "epoch": 1.968860626220703e-05,
      "step": 20645,
      "training_step_time": 7.085143804550171
    },
    {
      "epoch": 1.9689559936523438e-05,
      "model_forward_time": 1.8262426853179932,
      "step": 20646
    },
    {
      "epoch": 1.9689559936523438e-05,
      "step": 20646,
      "training_step_time": 8.370112895965576
    },
    {
      "epoch": 1.9690513610839842e-05,
      "model_forward_time": 1.8300318717956543,
      "step": 20647
    },
    {
      "epoch": 1.9690513610839842e-05,
      "step": 20647,
      "training_step_time": 7.071563243865967
    },
    {
      "epoch": 1.969146728515625e-05,
      "model_forward_time": 1.4555282592773438,
      "step": 20648
    },
    {
      "epoch": 1.969146728515625e-05,
      "step": 20648,
      "training_step_time": 7.4859418869018555
    },
    {
      "epoch": 1.9692420959472657e-05,
      "model_forward_time": 1.8304109573364258,
      "step": 20649
    },
    {
      "epoch": 1.9692420959472657e-05,
      "step": 20649,
      "training_step_time": 7.620701551437378
    },
    {
      "epoch": 1.969337463378906e-05,
      "grad_norm": 0.33547836542129517,
      "learning_rate": 9.345285954040626e-06,
      "loss": 0.0636,
      "step": 20650
    },
    {
      "epoch": 1.969337463378906e-05,
      "model_forward_time": 1.8333945274353027,
      "step": 20650
    },
    {
      "dynamics_loss_avg": 0.06069829165935516,
      "epoch": 1.969337463378906e-05,
      "step": 20650
    },
    {
      "action_loss_avg": 0.0032565688190516083,
      "epoch": 1.969337463378906e-05,
      "step": 20650
    },
    {
      "epoch": 1.969337463378906e-05,
      "step": 20650,
      "training_step_time": 8.589841842651367
    },
    {
      "epoch": 1.969432830810547e-05,
      "model_forward_time": 2.359898090362549,
      "step": 20651
    },
    {
      "epoch": 1.969432830810547e-05,
      "step": 20651,
      "training_step_time": 8.5878005027771
    },
    {
      "epoch": 1.9695281982421876e-05,
      "model_forward_time": 1.81839919090271,
      "step": 20652
    },
    {
      "epoch": 1.9695281982421876e-05,
      "step": 20652,
      "training_step_time": 9.798410654067993
    },
    {
      "epoch": 1.969623565673828e-05,
      "model_forward_time": 1.8332970142364502,
      "step": 20653
    },
    {
      "epoch": 1.969623565673828e-05,
      "step": 20653,
      "training_step_time": 7.698411226272583
    },
    {
      "epoch": 1.9697189331054687e-05,
      "model_forward_time": 1.823493480682373,
      "step": 20654
    },
    {
      "epoch": 1.9697189331054687e-05,
      "step": 20654,
      "training_step_time": 7.390174150466919
    },
    {
      "epoch": 1.9698143005371095e-05,
      "model_forward_time": 1.8253960609436035,
      "step": 20655
    },
    {
      "epoch": 1.9698143005371095e-05,
      "step": 20655,
      "training_step_time": 7.4430623054504395
    },
    {
      "epoch": 1.96990966796875e-05,
      "model_forward_time": 1.8195562362670898,
      "step": 20656
    },
    {
      "epoch": 1.96990966796875e-05,
      "step": 20656,
      "training_step_time": 7.646016359329224
    },
    {
      "epoch": 1.9700050354003906e-05,
      "model_forward_time": 1.8072376251220703,
      "step": 20657
    },
    {
      "epoch": 1.9700050354003906e-05,
      "step": 20657,
      "training_step_time": 8.447473526000977
    },
    {
      "epoch": 1.9701004028320314e-05,
      "model_forward_time": 1.8298990726470947,
      "step": 20658
    },
    {
      "epoch": 1.9701004028320314e-05,
      "step": 20658,
      "training_step_time": 8.202964067459106
    },
    {
      "epoch": 1.9701957702636718e-05,
      "model_forward_time": 1.816509485244751,
      "step": 20659
    },
    {
      "epoch": 1.9701957702636718e-05,
      "step": 20659,
      "training_step_time": 7.7604289054870605
    },
    {
      "epoch": 1.9702911376953125e-05,
      "grad_norm": 0.24810266494750977,
      "learning_rate": 9.34446772570615e-06,
      "loss": 0.0681,
      "step": 20660
    },
    {
      "epoch": 1.9702911376953125e-05,
      "model_forward_time": 2.364206314086914,
      "step": 20660
    },
    {
      "dynamics_loss_avg": 0.05536162164062262,
      "epoch": 1.9702911376953125e-05,
      "step": 20660
    },
    {
      "action_loss_avg": 0.0022933204425498845,
      "epoch": 1.9702911376953125e-05,
      "step": 20660
    },
    {
      "epoch": 1.9702911376953125e-05,
      "step": 20660,
      "training_step_time": 8.518685102462769
    },
    {
      "epoch": 1.9703865051269533e-05,
      "model_forward_time": 1.820824146270752,
      "step": 20661
    },
    {
      "epoch": 1.9703865051269533e-05,
      "step": 20661,
      "training_step_time": 10.805502653121948
    },
    {
      "epoch": 1.9704818725585937e-05,
      "model_forward_time": 1.8284056186676025,
      "step": 20662
    },
    {
      "epoch": 1.9704818725585937e-05,
      "step": 20662,
      "training_step_time": 8.867799282073975
    },
    {
      "epoch": 1.9705772399902344e-05,
      "model_forward_time": 1.8052465915679932,
      "step": 20663
    },
    {
      "epoch": 1.9705772399902344e-05,
      "step": 20663,
      "training_step_time": 7.385281085968018
    },
    {
      "epoch": 1.970672607421875e-05,
      "model_forward_time": 1.8286488056182861,
      "step": 20664
    },
    {
      "epoch": 1.970672607421875e-05,
      "step": 20664,
      "training_step_time": 7.431399345397949
    },
    {
      "epoch": 1.9707679748535155e-05,
      "model_forward_time": 1.8253579139709473,
      "step": 20665
    },
    {
      "epoch": 1.9707679748535155e-05,
      "step": 20665,
      "training_step_time": 7.606868743896484
    },
    {
      "epoch": 1.9708633422851563e-05,
      "model_forward_time": 1.8344616889953613,
      "step": 20666
    },
    {
      "epoch": 1.9708633422851563e-05,
      "step": 20666,
      "training_step_time": 7.505207777023315
    },
    {
      "epoch": 1.970958709716797e-05,
      "model_forward_time": 1.8090486526489258,
      "step": 20667
    },
    {
      "epoch": 1.970958709716797e-05,
      "step": 20667,
      "training_step_time": 8.533297538757324
    },
    {
      "epoch": 1.9710540771484374e-05,
      "model_forward_time": 1.815328598022461,
      "step": 20668
    },
    {
      "epoch": 1.9710540771484374e-05,
      "step": 20668,
      "training_step_time": 7.78039288520813
    },
    {
      "epoch": 1.9711494445800782e-05,
      "model_forward_time": 1.4321107864379883,
      "step": 20669
    },
    {
      "epoch": 1.9711494445800782e-05,
      "step": 20669,
      "training_step_time": 8.538403749465942
    },
    {
      "epoch": 1.971244812011719e-05,
      "grad_norm": 0.1759319305419922,
      "learning_rate": 9.343649022267214e-06,
      "loss": 0.0619,
      "step": 20670
    },
    {
      "epoch": 1.971244812011719e-05,
      "model_forward_time": 2.363969087600708,
      "step": 20670
    },
    {
      "dynamics_loss_avg": 0.05869124792516232,
      "epoch": 1.971244812011719e-05,
      "step": 20670
    },
    {
      "action_loss_avg": 0.0030906304775271563,
      "epoch": 1.971244812011719e-05,
      "step": 20670
    },
    {
      "epoch": 1.971244812011719e-05,
      "step": 20670,
      "training_step_time": 7.811336040496826
    },
    {
      "epoch": 1.9713401794433593e-05,
      "model_forward_time": 1.8303313255310059,
      "step": 20671
    },
    {
      "epoch": 1.9713401794433593e-05,
      "step": 20671,
      "training_step_time": 8.968444585800171
    },
    {
      "epoch": 1.971435546875e-05,
      "model_forward_time": 1.8289177417755127,
      "step": 20672
    },
    {
      "epoch": 1.971435546875e-05,
      "step": 20672,
      "training_step_time": 8.433783054351807
    },
    {
      "epoch": 1.9715309143066405e-05,
      "model_forward_time": 1.8287687301635742,
      "step": 20673
    },
    {
      "epoch": 1.9715309143066405e-05,
      "step": 20673,
      "training_step_time": 7.553556442260742
    },
    {
      "epoch": 1.9716262817382812e-05,
      "model_forward_time": 1.8231737613677979,
      "step": 20674
    },
    {
      "epoch": 1.9716262817382812e-05,
      "step": 20674,
      "training_step_time": 7.6668946743011475
    },
    {
      "epoch": 1.971721649169922e-05,
      "model_forward_time": 1.8276793956756592,
      "step": 20675
    },
    {
      "epoch": 1.971721649169922e-05,
      "step": 20675,
      "training_step_time": 7.488934516906738
    },
    {
      "epoch": 1.9718170166015623e-05,
      "model_forward_time": 1.839301586151123,
      "step": 20676
    },
    {
      "epoch": 1.9718170166015623e-05,
      "step": 20676,
      "training_step_time": 7.69477391242981
    },
    {
      "epoch": 1.971912384033203e-05,
      "model_forward_time": 1.8262286186218262,
      "step": 20677
    },
    {
      "epoch": 1.971912384033203e-05,
      "step": 20677,
      "training_step_time": 9.378406047821045
    },
    {
      "epoch": 1.9720077514648438e-05,
      "model_forward_time": 1.834096908569336,
      "step": 20678
    },
    {
      "epoch": 1.9720077514648438e-05,
      "step": 20678,
      "training_step_time": 10.295866966247559
    },
    {
      "epoch": 1.9721031188964842e-05,
      "model_forward_time": 2.3642752170562744,
      "step": 20679
    },
    {
      "epoch": 1.9721031188964842e-05,
      "step": 20679,
      "training_step_time": 7.71114182472229
    },
    {
      "epoch": 1.972198486328125e-05,
      "grad_norm": 0.23605406284332275,
      "learning_rate": 9.342829843813354e-06,
      "loss": 0.0643,
      "step": 20680
    },
    {
      "epoch": 1.972198486328125e-05,
      "model_forward_time": 1.8249659538269043,
      "step": 20680
    },
    {
      "dynamics_loss_avg": 0.06357527021318674,
      "epoch": 1.972198486328125e-05,
      "step": 20680
    },
    {
      "action_loss_avg": 0.0026898019132204354,
      "epoch": 1.972198486328125e-05,
      "step": 20680
    },
    {
      "epoch": 1.972198486328125e-05,
      "step": 20680,
      "training_step_time": 8.944147109985352
    },
    {
      "epoch": 1.9722938537597657e-05,
      "model_forward_time": 1.8246870040893555,
      "step": 20681
    },
    {
      "epoch": 1.9722938537597657e-05,
      "step": 20681,
      "training_step_time": 7.708242893218994
    },
    {
      "epoch": 1.972389221191406e-05,
      "model_forward_time": 1.83491849899292,
      "step": 20682
    },
    {
      "epoch": 1.972389221191406e-05,
      "step": 20682,
      "training_step_time": 7.489352703094482
    },
    {
      "epoch": 1.972484588623047e-05,
      "model_forward_time": 1.4768681526184082,
      "step": 20683
    },
    {
      "epoch": 1.972484588623047e-05,
      "step": 20683,
      "training_step_time": 7.626901865005493
    },
    {
      "epoch": 1.9725799560546876e-05,
      "model_forward_time": 1.828226089477539,
      "step": 20684
    },
    {
      "epoch": 1.9725799560546876e-05,
      "step": 20684,
      "training_step_time": 7.329512119293213
    },
    {
      "epoch": 1.972675323486328e-05,
      "model_forward_time": 1.8291614055633545,
      "step": 20685
    },
    {
      "epoch": 1.972675323486328e-05,
      "step": 20685,
      "training_step_time": 7.572654724121094
    },
    {
      "epoch": 1.9727706909179687e-05,
      "model_forward_time": 1.8324241638183594,
      "step": 20686
    },
    {
      "epoch": 1.9727706909179687e-05,
      "step": 20686,
      "training_step_time": 7.726680040359497
    },
    {
      "epoch": 1.9728660583496095e-05,
      "model_forward_time": 1.8302316665649414,
      "step": 20687
    },
    {
      "epoch": 1.9728660583496095e-05,
      "step": 20687,
      "training_step_time": 9.643014192581177
    },
    {
      "epoch": 1.97296142578125e-05,
      "model_forward_time": 2.388471841812134,
      "step": 20688
    },
    {
      "epoch": 1.97296142578125e-05,
      "step": 20688,
      "training_step_time": 8.489776134490967
    },
    {
      "epoch": 1.9730567932128906e-05,
      "model_forward_time": 1.8253648281097412,
      "step": 20689
    },
    {
      "epoch": 1.9730567932128906e-05,
      "step": 20689,
      "training_step_time": 9.451884508132935
    },
    {
      "epoch": 1.9731521606445314e-05,
      "grad_norm": 0.13568846881389618,
      "learning_rate": 9.34201019043415e-06,
      "loss": 0.0665,
      "step": 20690
    },
    {
      "epoch": 1.9731521606445314e-05,
      "model_forward_time": 1.8265697956085205,
      "step": 20690
    },
    {
      "dynamics_loss_avg": 0.07049985285848379,
      "epoch": 1.9731521606445314e-05,
      "step": 20690
    },
    {
      "action_loss_avg": 0.0036136083595920353,
      "epoch": 1.9731521606445314e-05,
      "step": 20690
    },
    {
      "epoch": 1.9731521606445314e-05,
      "step": 20690,
      "training_step_time": 7.6122636795043945
    },
    {
      "epoch": 1.9732475280761718e-05,
      "model_forward_time": 1.8283002376556396,
      "step": 20691
    },
    {
      "epoch": 1.9732475280761718e-05,
      "step": 20691,
      "training_step_time": 7.3927977085113525
    },
    {
      "epoch": 1.9733428955078125e-05,
      "model_forward_time": 1.832514762878418,
      "step": 20692
    },
    {
      "epoch": 1.9733428955078125e-05,
      "step": 20692,
      "training_step_time": 7.480085134506226
    },
    {
      "epoch": 1.9734382629394533e-05,
      "model_forward_time": 1.832529067993164,
      "step": 20693
    },
    {
      "epoch": 1.9734382629394533e-05,
      "step": 20693,
      "training_step_time": 7.289911270141602
    },
    {
      "epoch": 1.9735336303710937e-05,
      "model_forward_time": 1.820544719696045,
      "step": 20694
    },
    {
      "epoch": 1.9735336303710937e-05,
      "step": 20694,
      "training_step_time": 7.560077667236328
    },
    {
      "epoch": 1.9736289978027344e-05,
      "model_forward_time": 1.8289766311645508,
      "step": 20695
    },
    {
      "epoch": 1.9736289978027344e-05,
      "step": 20695,
      "training_step_time": 7.590928077697754
    },
    {
      "epoch": 1.973724365234375e-05,
      "model_forward_time": 1.8396880626678467,
      "step": 20696
    },
    {
      "epoch": 1.973724365234375e-05,
      "step": 20696,
      "training_step_time": 8.395612478256226
    },
    {
      "epoch": 1.9738197326660155e-05,
      "model_forward_time": 1.8349685668945312,
      "step": 20697
    },
    {
      "epoch": 1.9738197326660155e-05,
      "step": 20697,
      "training_step_time": 10.970583200454712
    },
    {
      "epoch": 1.9739151000976563e-05,
      "model_forward_time": 1.825070858001709,
      "step": 20698
    },
    {
      "epoch": 1.9739151000976563e-05,
      "step": 20698,
      "training_step_time": 8.654085874557495
    },
    {
      "epoch": 1.974010467529297e-05,
      "model_forward_time": 1.7402446269989014,
      "step": 20699
    },
    {
      "epoch": 1.974010467529297e-05,
      "step": 20699,
      "training_step_time": 7.735942840576172
    },
    {
      "epoch": 1.9741058349609374e-05,
      "grad_norm": 0.1760181188583374,
      "learning_rate": 9.341190062219241e-06,
      "loss": 0.0635,
      "step": 20700
    },
    {
      "epoch": 1.9741058349609374e-05,
      "model_forward_time": 1.7304480075836182,
      "step": 20700
    },
    {
      "dynamics_loss_avg": 0.05957929231226444,
      "epoch": 1.9741058349609374e-05,
      "step": 20700
    },
    {
      "action_loss_avg": 0.0027548546058824286,
      "epoch": 1.9741058349609374e-05,
      "step": 20700
    },
    {
      "epoch": 1.9741058349609374e-05,
      "step": 20700,
      "training_step_time": 7.884942054748535
    },
    {
      "epoch": 1.9742012023925782e-05,
      "model_forward_time": 1.7495009899139404,
      "step": 20701
    },
    {
      "epoch": 1.9742012023925782e-05,
      "step": 20701,
      "training_step_time": 8.155261516571045
    },
    {
      "epoch": 1.974296569824219e-05,
      "model_forward_time": 1.7448511123657227,
      "step": 20702
    },
    {
      "epoch": 1.974296569824219e-05,
      "step": 20702,
      "training_step_time": 6.805861711502075
    },
    {
      "epoch": 1.9743919372558593e-05,
      "model_forward_time": 1.7448279857635498,
      "step": 20703
    },
    {
      "epoch": 1.9743919372558593e-05,
      "step": 20703,
      "training_step_time": 7.642132997512817
    },
    {
      "epoch": 1.9744873046875e-05,
      "model_forward_time": 1.744063377380371,
      "step": 20704
    },
    {
      "epoch": 1.9744873046875e-05,
      "step": 20704,
      "training_step_time": 7.727734804153442
    },
    {
      "epoch": 1.9745826721191405e-05,
      "model_forward_time": 1.734694480895996,
      "step": 20705
    },
    {
      "epoch": 1.9745826721191405e-05,
      "step": 20705,
      "training_step_time": 10.083165168762207
    },
    {
      "epoch": 1.9746780395507812e-05,
      "model_forward_time": 1.7496838569641113,
      "step": 20706
    },
    {
      "epoch": 1.9746780395507812e-05,
      "step": 20706,
      "training_step_time": 9.770784854888916
    },
    {
      "epoch": 1.974773406982422e-05,
      "model_forward_time": 2.8994390964508057,
      "step": 20707
    },
    {
      "epoch": 1.974773406982422e-05,
      "step": 20707,
      "training_step_time": 7.691608667373657
    },
    {
      "epoch": 1.9748687744140623e-05,
      "model_forward_time": 1.8479371070861816,
      "step": 20708
    },
    {
      "epoch": 1.9748687744140623e-05,
      "step": 20708,
      "training_step_time": 7.806999206542969
    },
    {
      "epoch": 1.974964141845703e-05,
      "model_forward_time": 1.7443575859069824,
      "step": 20709
    },
    {
      "epoch": 1.974964141845703e-05,
      "step": 20709,
      "training_step_time": 7.6032164096832275
    },
    {
      "epoch": 1.975059509277344e-05,
      "grad_norm": 0.1871628612279892,
      "learning_rate": 9.340369459258313e-06,
      "loss": 0.0613,
      "step": 20710
    },
    {
      "epoch": 1.975059509277344e-05,
      "model_forward_time": 1.7398004531860352,
      "step": 20710
    },
    {
      "dynamics_loss_avg": 0.04932465422898531,
      "epoch": 1.975059509277344e-05,
      "step": 20710
    },
    {
      "action_loss_avg": 0.002082714065909386,
      "epoch": 1.975059509277344e-05,
      "step": 20710
    },
    {
      "epoch": 1.975059509277344e-05,
      "step": 20710,
      "training_step_time": 7.454651355743408
    },
    {
      "epoch": 1.9751548767089842e-05,
      "model_forward_time": 1.7471294403076172,
      "step": 20711
    },
    {
      "epoch": 1.9751548767089842e-05,
      "step": 20711,
      "training_step_time": 7.5549399852752686
    },
    {
      "epoch": 1.975250244140625e-05,
      "model_forward_time": 1.739567756652832,
      "step": 20712
    },
    {
      "epoch": 1.975250244140625e-05,
      "step": 20712,
      "training_step_time": 7.556643486022949
    },
    {
      "epoch": 1.9753456115722657e-05,
      "model_forward_time": 1.7497763633728027,
      "step": 20713
    },
    {
      "epoch": 1.9753456115722657e-05,
      "step": 20713,
      "training_step_time": 7.653361797332764
    },
    {
      "epoch": 1.975440979003906e-05,
      "model_forward_time": 1.7380869388580322,
      "step": 20714
    },
    {
      "epoch": 1.975440979003906e-05,
      "step": 20714,
      "training_step_time": 9.193316459655762
    },
    {
      "epoch": 1.975536346435547e-05,
      "model_forward_time": 1.7425920963287354,
      "step": 20715
    },
    {
      "epoch": 1.975536346435547e-05,
      "step": 20715,
      "training_step_time": 7.675481796264648
    },
    {
      "epoch": 1.9756317138671876e-05,
      "model_forward_time": 2.919996738433838,
      "step": 20716
    },
    {
      "epoch": 1.9756317138671876e-05,
      "step": 20716,
      "training_step_time": 7.637978553771973
    },
    {
      "epoch": 1.975727081298828e-05,
      "model_forward_time": 1.8401198387145996,
      "step": 20717
    },
    {
      "epoch": 1.975727081298828e-05,
      "step": 20717,
      "training_step_time": 8.364537000656128
    },
    {
      "epoch": 1.9758224487304687e-05,
      "model_forward_time": 1.7622263431549072,
      "step": 20718
    },
    {
      "epoch": 1.9758224487304687e-05,
      "step": 20718,
      "training_step_time": 7.669695615768433
    },
    {
      "epoch": 1.9759178161621095e-05,
      "model_forward_time": 1.7444918155670166,
      "step": 20719
    },
    {
      "epoch": 1.9759178161621095e-05,
      "step": 20719,
      "training_step_time": 7.407167434692383
    },
    {
      "epoch": 1.97601318359375e-05,
      "grad_norm": 0.16154296696186066,
      "learning_rate": 9.339548381641107e-06,
      "loss": 0.0583,
      "step": 20720
    },
    {
      "epoch": 1.97601318359375e-05,
      "model_forward_time": 1.715512752532959,
      "step": 20720
    },
    {
      "dynamics_loss_avg": 0.0575107766315341,
      "epoch": 1.97601318359375e-05,
      "step": 20720
    },
    {
      "action_loss_avg": 0.0027250606101006268,
      "epoch": 1.97601318359375e-05,
      "step": 20720
    },
    {
      "epoch": 1.97601318359375e-05,
      "step": 20720,
      "training_step_time": 7.570656061172485
    },
    {
      "epoch": 1.9761085510253906e-05,
      "model_forward_time": 1.744102954864502,
      "step": 20721
    },
    {
      "epoch": 1.9761085510253906e-05,
      "step": 20721,
      "training_step_time": 7.606914758682251
    },
    {
      "epoch": 1.9762039184570314e-05,
      "model_forward_time": 1.7309339046478271,
      "step": 20722
    },
    {
      "epoch": 1.9762039184570314e-05,
      "step": 20722,
      "training_step_time": 8.006793737411499
    },
    {
      "epoch": 1.9762992858886718e-05,
      "model_forward_time": 1.7429883480072021,
      "step": 20723
    },
    {
      "epoch": 1.9762992858886718e-05,
      "step": 20723,
      "training_step_time": 8.892695426940918
    },
    {
      "epoch": 1.9763946533203125e-05,
      "model_forward_time": 1.7356171607971191,
      "step": 20724
    },
    {
      "epoch": 1.9763946533203125e-05,
      "step": 20724,
      "training_step_time": 11.114985466003418
    },
    {
      "epoch": 1.9764900207519533e-05,
      "model_forward_time": 2.8701183795928955,
      "step": 20725
    },
    {
      "epoch": 1.9764900207519533e-05,
      "step": 20725,
      "training_step_time": 7.7267560958862305
    },
    {
      "epoch": 1.9765853881835937e-05,
      "model_forward_time": 1.784193992614746,
      "step": 20726
    },
    {
      "epoch": 1.9765853881835937e-05,
      "step": 20726,
      "training_step_time": 8.62810730934143
    },
    {
      "epoch": 1.9766807556152344e-05,
      "model_forward_time": 1.8248260021209717,
      "step": 20727
    },
    {
      "epoch": 1.9766807556152344e-05,
      "step": 20727,
      "training_step_time": 7.665329694747925
    },
    {
      "epoch": 1.976776123046875e-05,
      "model_forward_time": 1.7238562107086182,
      "step": 20728
    },
    {
      "epoch": 1.976776123046875e-05,
      "step": 20728,
      "training_step_time": 7.466237545013428
    },
    {
      "epoch": 1.9768714904785155e-05,
      "model_forward_time": 1.72935152053833,
      "step": 20729
    },
    {
      "epoch": 1.9768714904785155e-05,
      "step": 20729,
      "training_step_time": 7.597118139266968
    },
    {
      "epoch": 1.9769668579101563e-05,
      "grad_norm": 0.1681002974510193,
      "learning_rate": 9.338726829457414e-06,
      "loss": 0.0619,
      "step": 20730
    },
    {
      "epoch": 1.9769668579101563e-05,
      "model_forward_time": 1.7179701328277588,
      "step": 20730
    },
    {
      "dynamics_loss_avg": 0.06356351897120475,
      "epoch": 1.9769668579101563e-05,
      "step": 20730
    },
    {
      "action_loss_avg": 0.0030272368108853697,
      "epoch": 1.9769668579101563e-05,
      "step": 20730
    },
    {
      "epoch": 1.9769668579101563e-05,
      "step": 20730,
      "training_step_time": 7.524810314178467
    },
    {
      "epoch": 1.977062225341797e-05,
      "model_forward_time": 1.7217235565185547,
      "step": 20731
    },
    {
      "epoch": 1.977062225341797e-05,
      "step": 20731,
      "training_step_time": 7.4638566970825195
    },
    {
      "epoch": 1.9771575927734374e-05,
      "model_forward_time": 1.7243800163269043,
      "step": 20732
    },
    {
      "epoch": 1.9771575927734374e-05,
      "step": 20732,
      "training_step_time": 8.267929792404175
    },
    {
      "epoch": 1.9772529602050782e-05,
      "model_forward_time": 1.7313897609710693,
      "step": 20733
    },
    {
      "epoch": 1.9772529602050782e-05,
      "step": 20733,
      "training_step_time": 8.305372953414917
    },
    {
      "epoch": 1.977348327636719e-05,
      "model_forward_time": 2.8362793922424316,
      "step": 20734
    },
    {
      "epoch": 1.977348327636719e-05,
      "step": 20734,
      "training_step_time": 7.714362859725952
    },
    {
      "epoch": 1.9774436950683593e-05,
      "model_forward_time": 1.727879524230957,
      "step": 20735
    },
    {
      "epoch": 1.9774436950683593e-05,
      "step": 20735,
      "training_step_time": 7.846521377563477
    },
    {
      "epoch": 1.9775390625e-05,
      "model_forward_time": 1.8174753189086914,
      "step": 20736
    },
    {
      "epoch": 1.9775390625e-05,
      "step": 20736,
      "training_step_time": 7.685244560241699
    },
    {
      "epoch": 1.9776344299316405e-05,
      "model_forward_time": 1.727954626083374,
      "step": 20737
    },
    {
      "epoch": 1.9776344299316405e-05,
      "step": 20737,
      "training_step_time": 7.391545534133911
    },
    {
      "epoch": 1.9777297973632812e-05,
      "model_forward_time": 1.7312839031219482,
      "step": 20738
    },
    {
      "epoch": 1.9777297973632812e-05,
      "step": 20738,
      "training_step_time": 7.594359397888184
    },
    {
      "epoch": 1.977825164794922e-05,
      "model_forward_time": 1.7305443286895752,
      "step": 20739
    },
    {
      "epoch": 1.977825164794922e-05,
      "step": 20739,
      "training_step_time": 7.560636043548584
    },
    {
      "epoch": 1.9779205322265624e-05,
      "grad_norm": 0.19069360196590424,
      "learning_rate": 9.337904802797077e-06,
      "loss": 0.0597,
      "step": 20740
    },
    {
      "epoch": 1.9779205322265624e-05,
      "model_forward_time": 1.7131502628326416,
      "step": 20740
    },
    {
      "dynamics_loss_avg": 0.056881540641188624,
      "epoch": 1.9779205322265624e-05,
      "step": 20740
    },
    {
      "action_loss_avg": 0.001986134197795764,
      "epoch": 1.9779205322265624e-05,
      "step": 20740
    },
    {
      "epoch": 1.9779205322265624e-05,
      "step": 20740,
      "training_step_time": 7.329643726348877
    },
    {
      "epoch": 1.978015899658203e-05,
      "model_forward_time": 1.7262802124023438,
      "step": 20741
    },
    {
      "epoch": 1.978015899658203e-05,
      "step": 20741,
      "training_step_time": 7.780615568161011
    },
    {
      "epoch": 1.978111267089844e-05,
      "model_forward_time": 1.7143175601959229,
      "step": 20742
    },
    {
      "epoch": 1.978111267089844e-05,
      "step": 20742,
      "training_step_time": 8.918077230453491
    },
    {
      "epoch": 1.9782066345214842e-05,
      "model_forward_time": 2.8482444286346436,
      "step": 20743
    },
    {
      "epoch": 1.9782066345214842e-05,
      "step": 20743,
      "training_step_time": 7.696665525436401
    },
    {
      "epoch": 1.978302001953125e-05,
      "model_forward_time": 1.7409532070159912,
      "step": 20744
    },
    {
      "epoch": 1.978302001953125e-05,
      "step": 20744,
      "training_step_time": 7.706172943115234
    },
    {
      "epoch": 1.9783973693847657e-05,
      "model_forward_time": 1.8004894256591797,
      "step": 20745
    },
    {
      "epoch": 1.9783973693847657e-05,
      "step": 20745,
      "training_step_time": 7.708441734313965
    },
    {
      "epoch": 1.978492736816406e-05,
      "model_forward_time": 1.7436103820800781,
      "step": 20746
    },
    {
      "epoch": 1.978492736816406e-05,
      "step": 20746,
      "training_step_time": 7.3173041343688965
    },
    {
      "epoch": 1.978588104248047e-05,
      "model_forward_time": 1.7352397441864014,
      "step": 20747
    },
    {
      "epoch": 1.978588104248047e-05,
      "step": 20747,
      "training_step_time": 7.356277942657471
    },
    {
      "epoch": 1.9786834716796876e-05,
      "model_forward_time": 1.7401671409606934,
      "step": 20748
    },
    {
      "epoch": 1.9786834716796876e-05,
      "step": 20748,
      "training_step_time": 7.585635423660278
    },
    {
      "epoch": 1.978778839111328e-05,
      "model_forward_time": 1.7279791831970215,
      "step": 20749
    },
    {
      "epoch": 1.978778839111328e-05,
      "step": 20749,
      "training_step_time": 7.444674968719482
    },
    {
      "epoch": 1.9788742065429688e-05,
      "grad_norm": 0.2272936999797821,
      "learning_rate": 9.337082301749993e-06,
      "loss": 0.0628,
      "step": 20750
    },
    {
      "epoch": 1.9788742065429688e-05,
      "model_forward_time": 1.7598567008972168,
      "step": 20750
    },
    {
      "dynamics_loss_avg": 0.043706048745661974,
      "epoch": 1.9788742065429688e-05,
      "step": 20750
    },
    {
      "action_loss_avg": 0.0012054087070282548,
      "epoch": 1.9788742065429688e-05,
      "step": 20750
    },
    {
      "epoch": 1.9788742065429688e-05,
      "step": 20750,
      "training_step_time": 7.721811532974243
    },
    {
      "epoch": 1.9789695739746095e-05,
      "model_forward_time": 1.726952075958252,
      "step": 20751
    },
    {
      "epoch": 1.9789695739746095e-05,
      "step": 20751,
      "training_step_time": 8.406044960021973
    },
    {
      "epoch": 1.97906494140625e-05,
      "model_forward_time": 1.726470708847046,
      "step": 20752
    },
    {
      "epoch": 1.97906494140625e-05,
      "step": 20752,
      "training_step_time": 9.119692325592041
    },
    {
      "epoch": 1.9791603088378906e-05,
      "model_forward_time": 2.851895332336426,
      "step": 20753
    },
    {
      "epoch": 1.9791603088378906e-05,
      "step": 20753,
      "training_step_time": 7.743664503097534
    },
    {
      "epoch": 1.9792556762695314e-05,
      "model_forward_time": 1.754645586013794,
      "step": 20754
    },
    {
      "epoch": 1.9792556762695314e-05,
      "step": 20754,
      "training_step_time": 9.888635158538818
    },
    {
      "epoch": 1.9793510437011718e-05,
      "model_forward_time": 1.726438045501709,
      "step": 20755
    },
    {
      "epoch": 1.9793510437011718e-05,
      "step": 20755,
      "training_step_time": 8.364853858947754
    },
    {
      "epoch": 1.9794464111328125e-05,
      "model_forward_time": 1.7268247604370117,
      "step": 20756
    },
    {
      "epoch": 1.9794464111328125e-05,
      "step": 20756,
      "training_step_time": 7.572770833969116
    },
    {
      "epoch": 1.9795417785644533e-05,
      "model_forward_time": 1.733996868133545,
      "step": 20757
    },
    {
      "epoch": 1.9795417785644533e-05,
      "step": 20757,
      "training_step_time": 7.561574459075928
    },
    {
      "epoch": 1.9796371459960937e-05,
      "model_forward_time": 1.7265114784240723,
      "step": 20758
    },
    {
      "epoch": 1.9796371459960937e-05,
      "step": 20758,
      "training_step_time": 7.372248411178589
    },
    {
      "epoch": 1.9797325134277344e-05,
      "model_forward_time": 1.7597999572753906,
      "step": 20759
    },
    {
      "epoch": 1.9797325134277344e-05,
      "step": 20759,
      "training_step_time": 7.7009971141815186
    },
    {
      "epoch": 1.979827880859375e-05,
      "grad_norm": 0.13768510520458221,
      "learning_rate": 9.33625932640611e-06,
      "loss": 0.0609,
      "step": 20760
    },
    {
      "epoch": 1.979827880859375e-05,
      "model_forward_time": 1.7212684154510498,
      "step": 20760
    },
    {
      "dynamics_loss_avg": 0.0511702761054039,
      "epoch": 1.979827880859375e-05,
      "step": 20760
    },
    {
      "action_loss_avg": 0.0018820866680471227,
      "epoch": 1.979827880859375e-05,
      "step": 20760
    },
    {
      "epoch": 1.979827880859375e-05,
      "step": 20760,
      "training_step_time": 7.925839900970459
    },
    {
      "epoch": 1.9799232482910156e-05,
      "model_forward_time": 1.709392786026001,
      "step": 20761
    },
    {
      "epoch": 1.9799232482910156e-05,
      "step": 20761,
      "training_step_time": 8.975737810134888
    },
    {
      "epoch": 1.9800186157226563e-05,
      "model_forward_time": 2.9070611000061035,
      "step": 20762
    },
    {
      "epoch": 1.9800186157226563e-05,
      "step": 20762,
      "training_step_time": 8.457334518432617
    },
    {
      "epoch": 1.980113983154297e-05,
      "model_forward_time": 1.7220332622528076,
      "step": 20763
    },
    {
      "epoch": 1.980113983154297e-05,
      "step": 20763,
      "training_step_time": 8.88460636138916
    },
    {
      "epoch": 1.9802093505859374e-05,
      "model_forward_time": 1.7476015090942383,
      "step": 20764
    },
    {
      "epoch": 1.9802093505859374e-05,
      "step": 20764,
      "training_step_time": 7.709425449371338
    },
    {
      "epoch": 1.9803047180175782e-05,
      "model_forward_time": 1.721001386642456,
      "step": 20765
    },
    {
      "epoch": 1.9803047180175782e-05,
      "step": 20765,
      "training_step_time": 7.607501983642578
    },
    {
      "epoch": 1.9804000854492186e-05,
      "model_forward_time": 1.733447551727295,
      "step": 20766
    },
    {
      "epoch": 1.9804000854492186e-05,
      "step": 20766,
      "training_step_time": 7.66110634803772
    },
    {
      "epoch": 1.9804954528808593e-05,
      "model_forward_time": 1.7306747436523438,
      "step": 20767
    },
    {
      "epoch": 1.9804954528808593e-05,
      "step": 20767,
      "training_step_time": 7.399905681610107
    },
    {
      "epoch": 1.9805908203125e-05,
      "model_forward_time": 1.7441024780273438,
      "step": 20768
    },
    {
      "epoch": 1.9805908203125e-05,
      "step": 20768,
      "training_step_time": 8.630844593048096
    },
    {
      "epoch": 1.9806861877441405e-05,
      "model_forward_time": 1.7288181781768799,
      "step": 20769
    },
    {
      "epoch": 1.9806861877441405e-05,
      "step": 20769,
      "training_step_time": 8.329524993896484
    },
    {
      "epoch": 1.9807815551757812e-05,
      "grad_norm": 0.12536880373954773,
      "learning_rate": 9.335435876855428e-06,
      "loss": 0.064,
      "step": 20770
    },
    {
      "epoch": 1.9807815551757812e-05,
      "model_forward_time": 1.7320494651794434,
      "step": 20770
    },
    {
      "dynamics_loss_avg": 0.05493124611675739,
      "epoch": 1.9807815551757812e-05,
      "step": 20770
    },
    {
      "action_loss_avg": 0.00314684979384765,
      "epoch": 1.9807815551757812e-05,
      "step": 20770
    },
    {
      "epoch": 1.9807815551757812e-05,
      "step": 20770,
      "training_step_time": 7.85645055770874
    },
    {
      "epoch": 1.980876922607422e-05,
      "model_forward_time": 2.894629955291748,
      "step": 20771
    },
    {
      "epoch": 1.980876922607422e-05,
      "step": 20771,
      "training_step_time": 7.701022386550903
    },
    {
      "epoch": 1.9809722900390624e-05,
      "model_forward_time": 1.7325537204742432,
      "step": 20772
    },
    {
      "epoch": 1.9809722900390624e-05,
      "step": 20772,
      "training_step_time": 8.705062866210938
    },
    {
      "epoch": 1.981067657470703e-05,
      "model_forward_time": 1.7509799003601074,
      "step": 20773
    },
    {
      "epoch": 1.981067657470703e-05,
      "step": 20773,
      "training_step_time": 8.69895076751709
    },
    {
      "epoch": 1.981163024902344e-05,
      "model_forward_time": 1.7260417938232422,
      "step": 20774
    },
    {
      "epoch": 1.981163024902344e-05,
      "step": 20774,
      "training_step_time": 7.5612263679504395
    },
    {
      "epoch": 1.9812583923339842e-05,
      "model_forward_time": 1.7263877391815186,
      "step": 20775
    },
    {
      "epoch": 1.9812583923339842e-05,
      "step": 20775,
      "training_step_time": 7.652297496795654
    },
    {
      "epoch": 1.981353759765625e-05,
      "model_forward_time": 1.728926658630371,
      "step": 20776
    },
    {
      "epoch": 1.981353759765625e-05,
      "step": 20776,
      "training_step_time": 7.279295206069946
    },
    {
      "epoch": 1.9814491271972657e-05,
      "model_forward_time": 1.7378613948822021,
      "step": 20777
    },
    {
      "epoch": 1.9814491271972657e-05,
      "step": 20777,
      "training_step_time": 7.694538116455078
    },
    {
      "epoch": 1.981544494628906e-05,
      "model_forward_time": 1.72727370262146,
      "step": 20778
    },
    {
      "epoch": 1.981544494628906e-05,
      "step": 20778,
      "training_step_time": 7.610930442810059
    },
    {
      "epoch": 1.981639862060547e-05,
      "model_forward_time": 1.7231087684631348,
      "step": 20779
    },
    {
      "epoch": 1.981639862060547e-05,
      "step": 20779,
      "training_step_time": 8.4037446975708
    },
    {
      "epoch": 1.9817352294921876e-05,
      "grad_norm": 0.17400337755680084,
      "learning_rate": 9.334611953187994e-06,
      "loss": 0.0623,
      "step": 20780
    },
    {
      "epoch": 1.9817352294921876e-05,
      "model_forward_time": 2.8887577056884766,
      "step": 20780
    },
    {
      "dynamics_loss_avg": 0.0537471916526556,
      "epoch": 1.9817352294921876e-05,
      "step": 20780
    },
    {
      "action_loss_avg": 0.002368814795045182,
      "epoch": 1.9817352294921876e-05,
      "step": 20780
    },
    {
      "epoch": 1.9817352294921876e-05,
      "step": 20780,
      "training_step_time": 7.72012186050415
    },
    {
      "epoch": 1.981830596923828e-05,
      "model_forward_time": 1.727910041809082,
      "step": 20781
    },
    {
      "epoch": 1.981830596923828e-05,
      "step": 20781,
      "training_step_time": 9.585200309753418
    },
    {
      "epoch": 1.9819259643554688e-05,
      "model_forward_time": 1.7595815658569336,
      "step": 20782
    },
    {
      "epoch": 1.9819259643554688e-05,
      "step": 20782,
      "training_step_time": 8.56347942352295
    },
    {
      "epoch": 1.9820213317871095e-05,
      "model_forward_time": 1.7325644493103027,
      "step": 20783
    },
    {
      "epoch": 1.9820213317871095e-05,
      "step": 20783,
      "training_step_time": 7.421064138412476
    },
    {
      "epoch": 1.98211669921875e-05,
      "model_forward_time": 1.7256748676300049,
      "step": 20784
    },
    {
      "epoch": 1.98211669921875e-05,
      "step": 20784,
      "training_step_time": 7.644029378890991
    },
    {
      "epoch": 1.9822120666503906e-05,
      "model_forward_time": 1.7294185161590576,
      "step": 20785
    },
    {
      "epoch": 1.9822120666503906e-05,
      "step": 20785,
      "training_step_time": 7.657029628753662
    },
    {
      "epoch": 1.9823074340820314e-05,
      "model_forward_time": 1.7273292541503906,
      "step": 20786
    },
    {
      "epoch": 1.9823074340820314e-05,
      "step": 20786,
      "training_step_time": 8.030480861663818
    },
    {
      "epoch": 1.9824028015136718e-05,
      "model_forward_time": 1.35097074508667,
      "step": 20787
    },
    {
      "epoch": 1.9824028015136718e-05,
      "step": 20787,
      "training_step_time": 7.782277822494507
    },
    {
      "epoch": 1.9824981689453125e-05,
      "model_forward_time": 1.729050874710083,
      "step": 20788
    },
    {
      "epoch": 1.9824981689453125e-05,
      "step": 20788,
      "training_step_time": 8.993790864944458
    },
    {
      "epoch": 1.9825935363769533e-05,
      "model_forward_time": 2.8150923252105713,
      "step": 20789
    },
    {
      "epoch": 1.9825935363769533e-05,
      "step": 20789,
      "training_step_time": 8.512215614318848
    },
    {
      "epoch": 1.9826889038085937e-05,
      "grad_norm": 0.20807042717933655,
      "learning_rate": 9.333787555493915e-06,
      "loss": 0.0657,
      "step": 20790
    },
    {
      "epoch": 1.9826889038085937e-05,
      "model_forward_time": 1.733025312423706,
      "step": 20790
    },
    {
      "dynamics_loss_avg": 0.050177342258393764,
      "epoch": 1.9826889038085937e-05,
      "step": 20790
    },
    {
      "action_loss_avg": 0.002088976139202714,
      "epoch": 1.9826889038085937e-05,
      "step": 20790
    },
    {
      "epoch": 1.9826889038085937e-05,
      "step": 20790,
      "training_step_time": 7.7772603034973145
    },
    {
      "epoch": 1.9827842712402344e-05,
      "model_forward_time": 1.7534523010253906,
      "step": 20791
    },
    {
      "epoch": 1.9827842712402344e-05,
      "step": 20791,
      "training_step_time": 7.829924821853638
    },
    {
      "epoch": 1.982879638671875e-05,
      "model_forward_time": 1.7290279865264893,
      "step": 20792
    },
    {
      "epoch": 1.982879638671875e-05,
      "step": 20792,
      "training_step_time": 8.837427616119385
    },
    {
      "epoch": 1.9829750061035156e-05,
      "model_forward_time": 1.7323362827301025,
      "step": 20793
    },
    {
      "epoch": 1.9829750061035156e-05,
      "step": 20793,
      "training_step_time": 7.6956446170806885
    },
    {
      "epoch": 1.9830703735351563e-05,
      "model_forward_time": 1.7234163284301758,
      "step": 20794
    },
    {
      "epoch": 1.9830703735351563e-05,
      "step": 20794,
      "training_step_time": 8.179373025894165
    },
    {
      "epoch": 1.983165740966797e-05,
      "model_forward_time": 1.7291178703308105,
      "step": 20795
    },
    {
      "epoch": 1.983165740966797e-05,
      "step": 20795,
      "training_step_time": 8.206287145614624
    },
    {
      "epoch": 1.9832611083984374e-05,
      "model_forward_time": 1.7601008415222168,
      "step": 20796
    },
    {
      "epoch": 1.9832611083984374e-05,
      "step": 20796,
      "training_step_time": 7.71459436416626
    },
    {
      "epoch": 1.9833564758300782e-05,
      "model_forward_time": 1.7283618450164795,
      "step": 20797
    },
    {
      "epoch": 1.9833564758300782e-05,
      "step": 20797,
      "training_step_time": 9.963363885879517
    },
    {
      "epoch": 1.9834518432617186e-05,
      "model_forward_time": 1.7243835926055908,
      "step": 20798
    },
    {
      "epoch": 1.9834518432617186e-05,
      "step": 20798,
      "training_step_time": 8.399812459945679
    },
    {
      "epoch": 1.9835472106933593e-05,
      "model_forward_time": 2.930413007736206,
      "step": 20799
    },
    {
      "epoch": 1.9835472106933593e-05,
      "step": 20799,
      "training_step_time": 9.775429487228394
    },
    {
      "epoch": 1.983642578125e-05,
      "grad_norm": 0.17749594151973724,
      "learning_rate": 9.332962683863344e-06,
      "loss": 0.0628,
      "step": 20800
    },
    {
      "epoch": 1.983642578125e-05,
      "model_forward_time": 1.7341244220733643,
      "step": 20800
    },
    {
      "dynamics_loss_avg": 0.06119481287896633,
      "epoch": 1.983642578125e-05,
      "step": 20800
    },
    {
      "action_loss_avg": 0.0029021395894233137,
      "epoch": 1.983642578125e-05,
      "step": 20800
    },
    {
      "epoch": 1.983642578125e-05,
      "step": 20800,
      "training_step_time": 8.904968500137329
    },
    {
      "epoch": 1.9837379455566405e-05,
      "model_forward_time": 1.732863426208496,
      "step": 20801
    },
    {
      "epoch": 1.9837379455566405e-05,
      "step": 20801,
      "training_step_time": 7.714095592498779
    },
    {
      "epoch": 1.9838333129882812e-05,
      "model_forward_time": 1.6989612579345703,
      "step": 20802
    },
    {
      "epoch": 1.9838333129882812e-05,
      "step": 20802,
      "training_step_time": 7.628634691238403
    },
    {
      "epoch": 1.983928680419922e-05,
      "model_forward_time": 1.7189970016479492,
      "step": 20803
    },
    {
      "epoch": 1.983928680419922e-05,
      "step": 20803,
      "training_step_time": 7.473981142044067
    },
    {
      "epoch": 1.9840240478515624e-05,
      "model_forward_time": 1.7135138511657715,
      "step": 20804
    },
    {
      "epoch": 1.9840240478515624e-05,
      "step": 20804,
      "training_step_time": 7.45435643196106
    },
    {
      "epoch": 1.984119415283203e-05,
      "model_forward_time": 1.7447311878204346,
      "step": 20805
    },
    {
      "epoch": 1.984119415283203e-05,
      "step": 20805,
      "training_step_time": 7.635368824005127
    },
    {
      "epoch": 1.984214782714844e-05,
      "model_forward_time": 0.6893398761749268,
      "step": 20806
    },
    {
      "epoch": 1.984214782714844e-05,
      "step": 20806,
      "training_step_time": 7.663330554962158
    },
    {
      "epoch": 1.9843101501464842e-05,
      "model_forward_time": 1.706653356552124,
      "step": 20807
    },
    {
      "epoch": 1.9843101501464842e-05,
      "step": 20807,
      "training_step_time": 9.343622922897339
    },
    {
      "epoch": 1.984405517578125e-05,
      "model_forward_time": 2.8240249156951904,
      "step": 20808
    },
    {
      "epoch": 1.984405517578125e-05,
      "step": 20808,
      "training_step_time": 7.604190349578857
    },
    {
      "epoch": 1.9845008850097657e-05,
      "model_forward_time": 1.712200403213501,
      "step": 20809
    },
    {
      "epoch": 1.9845008850097657e-05,
      "step": 20809,
      "training_step_time": 7.822965145111084
    },
    {
      "epoch": 1.984596252441406e-05,
      "grad_norm": 0.20940054953098297,
      "learning_rate": 9.33213733838649e-06,
      "loss": 0.0632,
      "step": 20810
    },
    {
      "epoch": 1.984596252441406e-05,
      "model_forward_time": 1.7525858879089355,
      "step": 20810
    },
    {
      "dynamics_loss_avg": 0.0495843892917037,
      "epoch": 1.984596252441406e-05,
      "step": 20810
    },
    {
      "action_loss_avg": 0.002044304966693744,
      "epoch": 1.984596252441406e-05,
      "step": 20810
    },
    {
      "epoch": 1.984596252441406e-05,
      "step": 20810,
      "training_step_time": 7.644094228744507
    },
    {
      "epoch": 1.984691619873047e-05,
      "model_forward_time": 1.7148213386535645,
      "step": 20811
    },
    {
      "epoch": 1.984691619873047e-05,
      "step": 20811,
      "training_step_time": 7.56807804107666
    },
    {
      "epoch": 1.9847869873046876e-05,
      "model_forward_time": 1.7137482166290283,
      "step": 20812
    },
    {
      "epoch": 1.9847869873046876e-05,
      "step": 20812,
      "training_step_time": 7.41865873336792
    },
    {
      "epoch": 1.984882354736328e-05,
      "model_forward_time": 1.7181532382965088,
      "step": 20813
    },
    {
      "epoch": 1.984882354736328e-05,
      "step": 20813,
      "training_step_time": 7.381038427352905
    },
    {
      "epoch": 1.9849777221679688e-05,
      "model_forward_time": 1.745206356048584,
      "step": 20814
    },
    {
      "epoch": 1.9849777221679688e-05,
      "step": 20814,
      "training_step_time": 7.558012008666992
    },
    {
      "epoch": 1.9850730895996095e-05,
      "model_forward_time": 0.6821627616882324,
      "step": 20815
    },
    {
      "epoch": 1.9850730895996095e-05,
      "step": 20815,
      "training_step_time": 9.182178735733032
    },
    {
      "epoch": 1.98516845703125e-05,
      "model_forward_time": 1.713728427886963,
      "step": 20816
    },
    {
      "epoch": 1.98516845703125e-05,
      "step": 20816,
      "training_step_time": 7.774205207824707
    },
    {
      "epoch": 1.9852638244628906e-05,
      "model_forward_time": 1.7127656936645508,
      "step": 20817
    },
    {
      "epoch": 1.9852638244628906e-05,
      "step": 20817,
      "training_step_time": 9.906063795089722
    },
    {
      "epoch": 1.9853591918945314e-05,
      "model_forward_time": 2.8637776374816895,
      "step": 20818
    },
    {
      "epoch": 1.9853591918945314e-05,
      "step": 20818,
      "training_step_time": 7.625964879989624
    },
    {
      "epoch": 1.9854545593261718e-05,
      "model_forward_time": 1.738548755645752,
      "step": 20819
    },
    {
      "epoch": 1.9854545593261718e-05,
      "step": 20819,
      "training_step_time": 7.748490571975708
    },
    {
      "epoch": 1.9855499267578125e-05,
      "grad_norm": 0.1831023395061493,
      "learning_rate": 9.33131151915361e-06,
      "loss": 0.0645,
      "step": 20820
    },
    {
      "epoch": 1.9855499267578125e-05,
      "model_forward_time": 1.7229242324829102,
      "step": 20820
    },
    {
      "dynamics_loss_avg": 0.057254348509013654,
      "epoch": 1.9855499267578125e-05,
      "step": 20820
    },
    {
      "action_loss_avg": 0.0016858446382684634,
      "epoch": 1.9855499267578125e-05,
      "step": 20820
    },
    {
      "epoch": 1.9855499267578125e-05,
      "step": 20820,
      "training_step_time": 7.681723117828369
    },
    {
      "epoch": 1.9856452941894533e-05,
      "model_forward_time": 1.71927809715271,
      "step": 20821
    },
    {
      "epoch": 1.9856452941894533e-05,
      "step": 20821,
      "training_step_time": 8.244863986968994
    },
    {
      "epoch": 1.9857406616210937e-05,
      "model_forward_time": 1.7211918830871582,
      "step": 20822
    },
    {
      "epoch": 1.9857406616210937e-05,
      "step": 20822,
      "training_step_time": 7.317147731781006
    },
    {
      "epoch": 1.9858360290527344e-05,
      "model_forward_time": 1.7427945137023926,
      "step": 20823
    },
    {
      "epoch": 1.9858360290527344e-05,
      "step": 20823,
      "training_step_time": 8.06831169128418
    },
    {
      "epoch": 1.985931396484375e-05,
      "model_forward_time": 1.7177202701568604,
      "step": 20824
    },
    {
      "epoch": 1.985931396484375e-05,
      "step": 20824,
      "training_step_time": 7.733674764633179
    },
    {
      "epoch": 1.9860267639160156e-05,
      "model_forward_time": 1.72224760055542,
      "step": 20825
    },
    {
      "epoch": 1.9860267639160156e-05,
      "step": 20825,
      "training_step_time": 9.374653100967407
    },
    {
      "epoch": 1.9861221313476563e-05,
      "model_forward_time": 1.7193875312805176,
      "step": 20826
    },
    {
      "epoch": 1.9861221313476563e-05,
      "step": 20826,
      "training_step_time": 9.376867294311523
    },
    {
      "epoch": 1.986217498779297e-05,
      "model_forward_time": 2.868591547012329,
      "step": 20827
    },
    {
      "epoch": 1.986217498779297e-05,
      "step": 20827,
      "training_step_time": 8.346688747406006
    },
    {
      "epoch": 1.9863128662109374e-05,
      "model_forward_time": 1.7376279830932617,
      "step": 20828
    },
    {
      "epoch": 1.9863128662109374e-05,
      "step": 20828,
      "training_step_time": 7.833686113357544
    },
    {
      "epoch": 1.9864082336425782e-05,
      "model_forward_time": 1.720961570739746,
      "step": 20829
    },
    {
      "epoch": 1.9864082336425782e-05,
      "step": 20829,
      "training_step_time": 7.6906819343566895
    },
    {
      "epoch": 1.9865036010742186e-05,
      "grad_norm": 0.169107586145401,
      "learning_rate": 9.330485226255013e-06,
      "loss": 0.0628,
      "step": 20830
    },
    {
      "epoch": 1.9865036010742186e-05,
      "model_forward_time": 1.7152059078216553,
      "step": 20830
    },
    {
      "dynamics_loss_avg": 0.06073840726166964,
      "epoch": 1.9865036010742186e-05,
      "step": 20830
    },
    {
      "action_loss_avg": 0.0037783040083013476,
      "epoch": 1.9865036010742186e-05,
      "step": 20830
    },
    {
      "epoch": 1.9865036010742186e-05,
      "step": 20830,
      "training_step_time": 7.536483526229858
    },
    {
      "epoch": 1.9865989685058593e-05,
      "model_forward_time": 1.7252135276794434,
      "step": 20831
    },
    {
      "epoch": 1.9865989685058593e-05,
      "step": 20831,
      "training_step_time": 8.107271194458008
    },
    {
      "epoch": 1.9866943359375e-05,
      "model_forward_time": 1.7488329410552979,
      "step": 20832
    },
    {
      "epoch": 1.9866943359375e-05,
      "step": 20832,
      "training_step_time": 8.497115850448608
    },
    {
      "epoch": 1.9867897033691405e-05,
      "model_forward_time": 1.7183828353881836,
      "step": 20833
    },
    {
      "epoch": 1.9867897033691405e-05,
      "step": 20833,
      "training_step_time": 7.7560999393463135
    },
    {
      "epoch": 1.9868850708007812e-05,
      "model_forward_time": 1.3457543849945068,
      "step": 20834
    },
    {
      "epoch": 1.9868850708007812e-05,
      "step": 20834,
      "training_step_time": 8.53812551498413
    },
    {
      "epoch": 1.986980438232422e-05,
      "model_forward_time": 1.7239902019500732,
      "step": 20835
    },
    {
      "epoch": 1.986980438232422e-05,
      "step": 20835,
      "training_step_time": 9.079591274261475
    },
    {
      "epoch": 1.9870758056640624e-05,
      "model_forward_time": 2.794935703277588,
      "step": 20836
    },
    {
      "epoch": 1.9870758056640624e-05,
      "step": 20836,
      "training_step_time": 7.618831396102905
    },
    {
      "epoch": 1.987171173095703e-05,
      "model_forward_time": 1.7382235527038574,
      "step": 20837
    },
    {
      "epoch": 1.987171173095703e-05,
      "step": 20837,
      "training_step_time": 7.86373233795166
    },
    {
      "epoch": 1.987266540527344e-05,
      "model_forward_time": 1.7211439609527588,
      "step": 20838
    },
    {
      "epoch": 1.987266540527344e-05,
      "step": 20838,
      "training_step_time": 7.6856770515441895
    },
    {
      "epoch": 1.9873619079589842e-05,
      "model_forward_time": 1.7116973400115967,
      "step": 20839
    },
    {
      "epoch": 1.9873619079589842e-05,
      "step": 20839,
      "training_step_time": 7.67509388923645
    },
    {
      "epoch": 1.987457275390625e-05,
      "grad_norm": 0.15803903341293335,
      "learning_rate": 9.329658459781061e-06,
      "loss": 0.0626,
      "step": 20840
    },
    {
      "epoch": 1.987457275390625e-05,
      "model_forward_time": 1.711878776550293,
      "step": 20840
    },
    {
      "dynamics_loss_avg": 0.05484442748129368,
      "epoch": 1.987457275390625e-05,
      "step": 20840
    },
    {
      "action_loss_avg": 0.0022083634801674636,
      "epoch": 1.987457275390625e-05,
      "step": 20840
    },
    {
      "epoch": 1.987457275390625e-05,
      "step": 20840,
      "training_step_time": 8.405842542648315
    },
    {
      "epoch": 1.9875526428222657e-05,
      "model_forward_time": 1.7441394329071045,
      "step": 20841
    },
    {
      "epoch": 1.9875526428222657e-05,
      "step": 20841,
      "training_step_time": 7.61860728263855
    },
    {
      "epoch": 1.987648010253906e-05,
      "model_forward_time": 1.7107841968536377,
      "step": 20842
    },
    {
      "epoch": 1.987648010253906e-05,
      "step": 20842,
      "training_step_time": 7.698138952255249
    },
    {
      "epoch": 1.987743377685547e-05,
      "model_forward_time": 1.7248601913452148,
      "step": 20843
    },
    {
      "epoch": 1.987743377685547e-05,
      "step": 20843,
      "training_step_time": 7.94301700592041
    },
    {
      "epoch": 1.9878387451171876e-05,
      "model_forward_time": 1.7235636711120605,
      "step": 20844
    },
    {
      "epoch": 1.9878387451171876e-05,
      "step": 20844,
      "training_step_time": 8.467915296554565
    },
    {
      "epoch": 1.987934112548828e-05,
      "model_forward_time": 1.7238538265228271,
      "step": 20845
    },
    {
      "epoch": 1.987934112548828e-05,
      "step": 20845,
      "training_step_time": 8.341958284378052
    },
    {
      "epoch": 1.9880294799804688e-05,
      "model_forward_time": 2.9659719467163086,
      "step": 20846
    },
    {
      "epoch": 1.9880294799804688e-05,
      "step": 20846,
      "training_step_time": 7.8873231410980225
    },
    {
      "epoch": 1.9881248474121095e-05,
      "model_forward_time": 1.7247405052185059,
      "step": 20847
    },
    {
      "epoch": 1.9881248474121095e-05,
      "step": 20847,
      "training_step_time": 8.189552783966064
    },
    {
      "epoch": 1.98822021484375e-05,
      "model_forward_time": 1.724135398864746,
      "step": 20848
    },
    {
      "epoch": 1.98822021484375e-05,
      "step": 20848,
      "training_step_time": 7.60170316696167
    },
    {
      "epoch": 1.9883155822753906e-05,
      "model_forward_time": 1.7250120639801025,
      "step": 20849
    },
    {
      "epoch": 1.9883155822753906e-05,
      "step": 20849,
      "training_step_time": 7.47325873374939
    },
    {
      "epoch": 1.9884109497070314e-05,
      "grad_norm": 0.18790526688098907,
      "learning_rate": 9.328831219822173e-06,
      "loss": 0.0661,
      "step": 20850
    },
    {
      "epoch": 1.9884109497070314e-05,
      "model_forward_time": 1.793393850326538,
      "step": 20850
    },
    {
      "dynamics_loss_avg": 0.0595699455589056,
      "epoch": 1.9884109497070314e-05,
      "step": 20850
    },
    {
      "action_loss_avg": 0.003431351168546826,
      "epoch": 1.9884109497070314e-05,
      "step": 20850
    },
    {
      "epoch": 1.9884109497070314e-05,
      "step": 20850,
      "training_step_time": 7.536867618560791
    },
    {
      "epoch": 1.9885063171386718e-05,
      "model_forward_time": 1.7317204475402832,
      "step": 20851
    },
    {
      "epoch": 1.9885063171386718e-05,
      "step": 20851,
      "training_step_time": 8.304991245269775
    },
    {
      "epoch": 1.9886016845703125e-05,
      "model_forward_time": 1.7239012718200684,
      "step": 20852
    },
    {
      "epoch": 1.9886016845703125e-05,
      "step": 20852,
      "training_step_time": 8.386780977249146
    },
    {
      "epoch": 1.9886970520019533e-05,
      "model_forward_time": 1.7120108604431152,
      "step": 20853
    },
    {
      "epoch": 1.9886970520019533e-05,
      "step": 20853,
      "training_step_time": 8.99721884727478
    },
    {
      "epoch": 1.9887924194335937e-05,
      "model_forward_time": 1.716135025024414,
      "step": 20854
    },
    {
      "epoch": 1.9887924194335937e-05,
      "step": 20854,
      "training_step_time": 8.462956190109253
    },
    {
      "epoch": 1.9888877868652344e-05,
      "model_forward_time": 2.916536331176758,
      "step": 20855
    },
    {
      "epoch": 1.9888877868652344e-05,
      "step": 20855,
      "training_step_time": 7.8496174812316895
    },
    {
      "epoch": 1.988983154296875e-05,
      "model_forward_time": 1.7337019443511963,
      "step": 20856
    },
    {
      "epoch": 1.988983154296875e-05,
      "step": 20856,
      "training_step_time": 8.276869058609009
    },
    {
      "epoch": 1.9890785217285156e-05,
      "model_forward_time": 1.7171127796173096,
      "step": 20857
    },
    {
      "epoch": 1.9890785217285156e-05,
      "step": 20857,
      "training_step_time": 7.563760757446289
    },
    {
      "epoch": 1.9891738891601563e-05,
      "model_forward_time": 1.7176470756530762,
      "step": 20858
    },
    {
      "epoch": 1.9891738891601563e-05,
      "step": 20858,
      "training_step_time": 7.4746253490448
    },
    {
      "epoch": 1.989269256591797e-05,
      "model_forward_time": 1.7439255714416504,
      "step": 20859
    },
    {
      "epoch": 1.989269256591797e-05,
      "step": 20859,
      "training_step_time": 7.590851068496704
    },
    {
      "epoch": 1.9893646240234374e-05,
      "grad_norm": 0.22366741299629211,
      "learning_rate": 9.328003506468808e-06,
      "loss": 0.0624,
      "step": 20860
    },
    {
      "epoch": 1.9893646240234374e-05,
      "model_forward_time": 1.3367884159088135,
      "step": 20860
    },
    {
      "dynamics_loss_avg": 0.07338135149329901,
      "epoch": 1.9893646240234374e-05,
      "step": 20860
    },
    {
      "action_loss_avg": 0.004415563540533185,
      "epoch": 1.9893646240234374e-05,
      "step": 20860
    },
    {
      "epoch": 1.9893646240234374e-05,
      "step": 20860,
      "training_step_time": 7.5228431224823
    },
    {
      "epoch": 1.9894599914550782e-05,
      "model_forward_time": 1.709282398223877,
      "step": 20861
    },
    {
      "epoch": 1.9894599914550782e-05,
      "step": 20861,
      "training_step_time": 8.563999652862549
    },
    {
      "epoch": 1.9895553588867186e-05,
      "model_forward_time": 1.7245490550994873,
      "step": 20862
    },
    {
      "epoch": 1.9895553588867186e-05,
      "step": 20862,
      "training_step_time": 9.528453588485718
    },
    {
      "epoch": 1.9896507263183593e-05,
      "model_forward_time": 1.717986822128296,
      "step": 20863
    },
    {
      "epoch": 1.9896507263183593e-05,
      "step": 20863,
      "training_step_time": 7.749124526977539
    },
    {
      "epoch": 1.98974609375e-05,
      "model_forward_time": 2.8330907821655273,
      "step": 20864
    },
    {
      "epoch": 1.98974609375e-05,
      "step": 20864,
      "training_step_time": 7.973459482192993
    },
    {
      "epoch": 1.9898414611816405e-05,
      "model_forward_time": 1.7365450859069824,
      "step": 20865
    },
    {
      "epoch": 1.9898414611816405e-05,
      "step": 20865,
      "training_step_time": 7.758305311203003
    },
    {
      "epoch": 1.9899368286132812e-05,
      "model_forward_time": 1.7184655666351318,
      "step": 20866
    },
    {
      "epoch": 1.9899368286132812e-05,
      "step": 20866,
      "training_step_time": 7.809046268463135
    },
    {
      "epoch": 1.990032196044922e-05,
      "model_forward_time": 1.7156226634979248,
      "step": 20867
    },
    {
      "epoch": 1.990032196044922e-05,
      "step": 20867,
      "training_step_time": 7.601394891738892
    },
    {
      "epoch": 1.9901275634765624e-05,
      "model_forward_time": 1.747344970703125,
      "step": 20868
    },
    {
      "epoch": 1.9901275634765624e-05,
      "step": 20868,
      "training_step_time": 7.72339940071106
    },
    {
      "epoch": 1.990222930908203e-05,
      "model_forward_time": 1.7157275676727295,
      "step": 20869
    },
    {
      "epoch": 1.990222930908203e-05,
      "step": 20869,
      "training_step_time": 7.472060918807983
    },
    {
      "epoch": 1.990318298339844e-05,
      "grad_norm": 0.14682866632938385,
      "learning_rate": 9.327175319811488e-06,
      "loss": 0.0647,
      "step": 20870
    },
    {
      "epoch": 1.990318298339844e-05,
      "model_forward_time": 1.7118494510650635,
      "step": 20870
    },
    {
      "dynamics_loss_avg": 0.05994761232286692,
      "epoch": 1.990318298339844e-05,
      "step": 20870
    },
    {
      "action_loss_avg": 0.004173008410725742,
      "epoch": 1.990318298339844e-05,
      "step": 20870
    },
    {
      "epoch": 1.990318298339844e-05,
      "step": 20870,
      "training_step_time": 8.390614032745361
    },
    {
      "epoch": 1.9904136657714842e-05,
      "model_forward_time": 1.721264362335205,
      "step": 20871
    },
    {
      "epoch": 1.9904136657714842e-05,
      "step": 20871,
      "training_step_time": 10.57557988166809
    },
    {
      "epoch": 1.990509033203125e-05,
      "model_forward_time": 1.71006441116333,
      "step": 20872
    },
    {
      "epoch": 1.990509033203125e-05,
      "step": 20872,
      "training_step_time": 8.381143808364868
    },
    {
      "epoch": 1.9906044006347657e-05,
      "model_forward_time": 2.8156003952026367,
      "step": 20873
    },
    {
      "epoch": 1.9906044006347657e-05,
      "step": 20873,
      "training_step_time": 7.914826154708862
    },
    {
      "epoch": 1.990699768066406e-05,
      "model_forward_time": 1.738050937652588,
      "step": 20874
    },
    {
      "epoch": 1.990699768066406e-05,
      "step": 20874,
      "training_step_time": 8.761875629425049
    },
    {
      "epoch": 1.990795135498047e-05,
      "model_forward_time": 1.717682123184204,
      "step": 20875
    },
    {
      "epoch": 1.990795135498047e-05,
      "step": 20875,
      "training_step_time": 7.753595590591431
    },
    {
      "epoch": 1.9908905029296876e-05,
      "model_forward_time": 1.7133262157440186,
      "step": 20876
    },
    {
      "epoch": 1.9908905029296876e-05,
      "step": 20876,
      "training_step_time": 7.645514488220215
    },
    {
      "epoch": 1.990985870361328e-05,
      "model_forward_time": 1.732741355895996,
      "step": 20877
    },
    {
      "epoch": 1.990985870361328e-05,
      "step": 20877,
      "training_step_time": 7.649455547332764
    },
    {
      "epoch": 1.9910812377929688e-05,
      "model_forward_time": 1.7156982421875,
      "step": 20878
    },
    {
      "epoch": 1.9910812377929688e-05,
      "step": 20878,
      "training_step_time": 7.567711353302002
    },
    {
      "epoch": 1.9911766052246095e-05,
      "model_forward_time": 1.715381145477295,
      "step": 20879
    },
    {
      "epoch": 1.9911766052246095e-05,
      "step": 20879,
      "training_step_time": 7.77695107460022
    },
    {
      "epoch": 1.99127197265625e-05,
      "grad_norm": 0.22049343585968018,
      "learning_rate": 9.32634665994078e-06,
      "loss": 0.0683,
      "step": 20880
    },
    {
      "epoch": 1.99127197265625e-05,
      "model_forward_time": 1.7119598388671875,
      "step": 20880
    },
    {
      "dynamics_loss_avg": 0.06717740707099437,
      "epoch": 1.99127197265625e-05,
      "step": 20880
    },
    {
      "action_loss_avg": 0.0019724794779904186,
      "epoch": 1.99127197265625e-05,
      "step": 20880
    },
    {
      "epoch": 1.99127197265625e-05,
      "step": 20880,
      "training_step_time": 8.895591497421265
    },
    {
      "epoch": 1.9913673400878906e-05,
      "model_forward_time": 1.7131552696228027,
      "step": 20881
    },
    {
      "epoch": 1.9913673400878906e-05,
      "step": 20881,
      "training_step_time": 9.530481815338135
    },
    {
      "epoch": 1.9914627075195314e-05,
      "model_forward_time": 1.718876838684082,
      "step": 20882
    },
    {
      "epoch": 1.9914627075195314e-05,
      "step": 20882,
      "training_step_time": 8.993814468383789
    },
    {
      "epoch": 1.9915580749511718e-05,
      "model_forward_time": 2.8863065242767334,
      "step": 20883
    },
    {
      "epoch": 1.9915580749511718e-05,
      "step": 20883,
      "training_step_time": 8.787877798080444
    },
    {
      "epoch": 1.9916534423828125e-05,
      "model_forward_time": 1.7388203144073486,
      "step": 20884
    },
    {
      "epoch": 1.9916534423828125e-05,
      "step": 20884,
      "training_step_time": 8.336905479431152
    },
    {
      "epoch": 1.9917488098144533e-05,
      "model_forward_time": 1.7168450355529785,
      "step": 20885
    },
    {
      "epoch": 1.9917488098144533e-05,
      "step": 20885,
      "training_step_time": 7.774392127990723
    },
    {
      "epoch": 1.9918441772460937e-05,
      "model_forward_time": 1.7431702613830566,
      "step": 20886
    },
    {
      "epoch": 1.9918441772460937e-05,
      "step": 20886,
      "training_step_time": 7.650415897369385
    },
    {
      "epoch": 1.9919395446777344e-05,
      "model_forward_time": 1.7180390357971191,
      "step": 20887
    },
    {
      "epoch": 1.9919395446777344e-05,
      "step": 20887,
      "training_step_time": 7.711270809173584
    },
    {
      "epoch": 1.992034912109375e-05,
      "model_forward_time": 1.7241640090942383,
      "step": 20888
    },
    {
      "epoch": 1.992034912109375e-05,
      "step": 20888,
      "training_step_time": 8.410813808441162
    },
    {
      "epoch": 1.9921302795410156e-05,
      "model_forward_time": 1.716238021850586,
      "step": 20889
    },
    {
      "epoch": 1.9921302795410156e-05,
      "step": 20889,
      "training_step_time": 9.965585231781006
    },
    {
      "epoch": 1.9922256469726563e-05,
      "grad_norm": 0.24594472348690033,
      "learning_rate": 9.325517526947308e-06,
      "loss": 0.0685,
      "step": 20890
    },
    {
      "epoch": 1.9922256469726563e-05,
      "model_forward_time": 1.710129737854004,
      "step": 20890
    },
    {
      "dynamics_loss_avg": 0.07214876282960177,
      "epoch": 1.9922256469726563e-05,
      "step": 20890
    },
    {
      "action_loss_avg": 0.0027508768835105,
      "epoch": 1.9922256469726563e-05,
      "step": 20890
    },
    {
      "epoch": 1.9922256469726563e-05,
      "step": 20890,
      "training_step_time": 7.8787901401519775
    },
    {
      "epoch": 1.9923210144042967e-05,
      "model_forward_time": 1.7137279510498047,
      "step": 20891
    },
    {
      "epoch": 1.9923210144042967e-05,
      "step": 20891,
      "training_step_time": 7.817652702331543
    },
    {
      "epoch": 1.9924163818359374e-05,
      "model_forward_time": 2.8908238410949707,
      "step": 20892
    },
    {
      "epoch": 1.9924163818359374e-05,
      "step": 20892,
      "training_step_time": 8.604491233825684
    },
    {
      "epoch": 1.9925117492675782e-05,
      "model_forward_time": 1.7341506481170654,
      "step": 20893
    },
    {
      "epoch": 1.9925117492675782e-05,
      "step": 20893,
      "training_step_time": 8.228832483291626
    },
    {
      "epoch": 1.9926071166992186e-05,
      "model_forward_time": 1.7142491340637207,
      "step": 20894
    },
    {
      "epoch": 1.9926071166992186e-05,
      "step": 20894,
      "training_step_time": 7.741391897201538
    },
    {
      "epoch": 1.9927024841308593e-05,
      "model_forward_time": 1.745295524597168,
      "step": 20895
    },
    {
      "epoch": 1.9927024841308593e-05,
      "step": 20895,
      "training_step_time": 7.616755247116089
    },
    {
      "epoch": 1.9927978515625e-05,
      "model_forward_time": 1.713813304901123,
      "step": 20896
    },
    {
      "epoch": 1.9927978515625e-05,
      "step": 20896,
      "training_step_time": 7.692469358444214
    },
    {
      "epoch": 1.9928932189941405e-05,
      "model_forward_time": 1.7165069580078125,
      "step": 20897
    },
    {
      "epoch": 1.9928932189941405e-05,
      "step": 20897,
      "training_step_time": 8.50154447555542
    },
    {
      "epoch": 1.9929885864257812e-05,
      "model_forward_time": 1.7171037197113037,
      "step": 20898
    },
    {
      "epoch": 1.9929885864257812e-05,
      "step": 20898,
      "training_step_time": 8.255216598510742
    },
    {
      "epoch": 1.993083953857422e-05,
      "model_forward_time": 1.7153849601745605,
      "step": 20899
    },
    {
      "epoch": 1.993083953857422e-05,
      "step": 20899,
      "training_step_time": 7.757991552352905
    },
    {
      "epoch": 1.9931793212890624e-05,
      "grad_norm": 0.1786651909351349,
      "learning_rate": 9.32468792092174e-06,
      "loss": 0.0644,
      "step": 20900
    },
    {
      "epoch": 1.9931793212890624e-05,
      "model_forward_time": 1.7171168327331543,
      "step": 20900
    },
    {
      "dynamics_loss_avg": 0.05688747055828571,
      "epoch": 1.9931793212890624e-05,
      "step": 20900
    },
    {
      "action_loss_avg": 0.0032336710137315094,
      "epoch": 1.9931793212890624e-05,
      "step": 20900
    },
    {
      "epoch": 1.9931793212890624e-05,
      "step": 20900,
      "training_step_time": 7.717545986175537
    },
    {
      "epoch": 1.993274688720703e-05,
      "model_forward_time": 2.8319709300994873,
      "step": 20901
    },
    {
      "epoch": 1.993274688720703e-05,
      "step": 20901,
      "training_step_time": 7.846375226974487
    },
    {
      "epoch": 1.993370056152344e-05,
      "model_forward_time": 1.7360482215881348,
      "step": 20902
    },
    {
      "epoch": 1.993370056152344e-05,
      "step": 20902,
      "training_step_time": 8.743340253829956
    },
    {
      "epoch": 1.9934654235839843e-05,
      "model_forward_time": 1.7157971858978271,
      "step": 20903
    },
    {
      "epoch": 1.9934654235839843e-05,
      "step": 20903,
      "training_step_time": 7.701217889785767
    },
    {
      "epoch": 1.993560791015625e-05,
      "model_forward_time": 1.7320644855499268,
      "step": 20904
    },
    {
      "epoch": 1.993560791015625e-05,
      "step": 20904,
      "training_step_time": 7.585095167160034
    },
    {
      "epoch": 1.9936561584472657e-05,
      "model_forward_time": 1.7204134464263916,
      "step": 20905
    },
    {
      "epoch": 1.9936561584472657e-05,
      "step": 20905,
      "training_step_time": 7.743253469467163
    },
    {
      "epoch": 1.993751525878906e-05,
      "model_forward_time": 1.7172093391418457,
      "step": 20906
    },
    {
      "epoch": 1.993751525878906e-05,
      "step": 20906,
      "training_step_time": 7.631258964538574
    },
    {
      "epoch": 1.993846893310547e-05,
      "model_forward_time": 1.716559886932373,
      "step": 20907
    },
    {
      "epoch": 1.993846893310547e-05,
      "step": 20907,
      "training_step_time": 8.655611753463745
    },
    {
      "epoch": 1.9939422607421876e-05,
      "model_forward_time": 1.7159900665283203,
      "step": 20908
    },
    {
      "epoch": 1.9939422607421876e-05,
      "step": 20908,
      "training_step_time": 9.518401622772217
    },
    {
      "epoch": 1.994037628173828e-05,
      "model_forward_time": 1.7077467441558838,
      "step": 20909
    },
    {
      "epoch": 1.994037628173828e-05,
      "step": 20909,
      "training_step_time": 8.70480465888977
    },
    {
      "epoch": 1.9941329956054688e-05,
      "grad_norm": 0.23029853403568268,
      "learning_rate": 9.323857841954805e-06,
      "loss": 0.0641,
      "step": 20910
    },
    {
      "epoch": 1.9941329956054688e-05,
      "model_forward_time": 2.829501152038574,
      "step": 20910
    },
    {
      "dynamics_loss_avg": 0.04839324597269297,
      "epoch": 1.9941329956054688e-05,
      "step": 20910
    },
    {
      "action_loss_avg": 0.0014531703665852547,
      "epoch": 1.9941329956054688e-05,
      "step": 20910
    },
    {
      "epoch": 1.9941329956054688e-05,
      "step": 20910,
      "training_step_time": 7.948425531387329
    },
    {
      "epoch": 1.9942283630371095e-05,
      "model_forward_time": 1.72330641746521,
      "step": 20911
    },
    {
      "epoch": 1.9942283630371095e-05,
      "step": 20911,
      "training_step_time": 9.203347444534302
    },
    {
      "epoch": 1.99432373046875e-05,
      "model_forward_time": 1.7156932353973389,
      "step": 20912
    },
    {
      "epoch": 1.99432373046875e-05,
      "step": 20912,
      "training_step_time": 7.731845378875732
    },
    {
      "epoch": 1.9944190979003906e-05,
      "model_forward_time": 1.72316575050354,
      "step": 20913
    },
    {
      "epoch": 1.9944190979003906e-05,
      "step": 20913,
      "training_step_time": 7.578475713729858
    },
    {
      "epoch": 1.9945144653320314e-05,
      "model_forward_time": 1.7122211456298828,
      "step": 20914
    },
    {
      "epoch": 1.9945144653320314e-05,
      "step": 20914,
      "training_step_time": 7.661623954772949
    },
    {
      "epoch": 1.9946098327636718e-05,
      "model_forward_time": 1.7135913372039795,
      "step": 20915
    },
    {
      "epoch": 1.9946098327636718e-05,
      "step": 20915,
      "training_step_time": 7.423473596572876
    },
    {
      "epoch": 1.9947052001953125e-05,
      "model_forward_time": 1.7151119709014893,
      "step": 20916
    },
    {
      "epoch": 1.9947052001953125e-05,
      "step": 20916,
      "training_step_time": 7.699979782104492
    },
    {
      "epoch": 1.9948005676269533e-05,
      "model_forward_time": 1.7135975360870361,
      "step": 20917
    },
    {
      "epoch": 1.9948005676269533e-05,
      "step": 20917,
      "training_step_time": 7.985110282897949
    },
    {
      "epoch": 1.9948959350585937e-05,
      "model_forward_time": 1.711838722229004,
      "step": 20918
    },
    {
      "epoch": 1.9948959350585937e-05,
      "step": 20918,
      "training_step_time": 7.794415712356567
    },
    {
      "epoch": 1.9949913024902344e-05,
      "model_forward_time": 2.8202834129333496,
      "step": 20919
    },
    {
      "epoch": 1.9949913024902344e-05,
      "step": 20919,
      "training_step_time": 7.906118631362915
    },
    {
      "epoch": 1.995086669921875e-05,
      "grad_norm": 0.2048514485359192,
      "learning_rate": 9.323027290137276e-06,
      "loss": 0.0599,
      "step": 20920
    },
    {
      "epoch": 1.995086669921875e-05,
      "model_forward_time": 1.7131235599517822,
      "step": 20920
    },
    {
      "dynamics_loss_avg": 0.06398451030254364,
      "epoch": 1.995086669921875e-05,
      "step": 20920
    },
    {
      "action_loss_avg": 0.0038180362549610435,
      "epoch": 1.995086669921875e-05,
      "step": 20920
    },
    {
      "epoch": 1.995086669921875e-05,
      "step": 20920,
      "training_step_time": 7.629064321517944
    },
    {
      "epoch": 1.9951820373535156e-05,
      "model_forward_time": 1.7417924404144287,
      "step": 20921
    },
    {
      "epoch": 1.9951820373535156e-05,
      "step": 20921,
      "training_step_time": 9.148619413375854
    },
    {
      "epoch": 1.9952774047851563e-05,
      "model_forward_time": 1.733304738998413,
      "step": 20922
    },
    {
      "epoch": 1.9952774047851563e-05,
      "step": 20922,
      "training_step_time": 7.4389543533325195
    },
    {
      "epoch": 1.9953727722167967e-05,
      "model_forward_time": 1.7188615798950195,
      "step": 20923
    },
    {
      "epoch": 1.9953727722167967e-05,
      "step": 20923,
      "training_step_time": 7.590177059173584
    },
    {
      "epoch": 1.9954681396484375e-05,
      "model_forward_time": 1.7141711711883545,
      "step": 20924
    },
    {
      "epoch": 1.9954681396484375e-05,
      "step": 20924,
      "training_step_time": 7.311223030090332
    },
    {
      "epoch": 1.9955635070800782e-05,
      "model_forward_time": 1.7199842929840088,
      "step": 20925
    },
    {
      "epoch": 1.9955635070800782e-05,
      "step": 20925,
      "training_step_time": 7.658700466156006
    },
    {
      "epoch": 1.9956588745117186e-05,
      "model_forward_time": 1.7144839763641357,
      "step": 20926
    },
    {
      "epoch": 1.9956588745117186e-05,
      "step": 20926,
      "training_step_time": 8.803163766860962
    },
    {
      "epoch": 1.9957542419433593e-05,
      "model_forward_time": 1.337425947189331,
      "step": 20927
    },
    {
      "epoch": 1.9957542419433593e-05,
      "step": 20927,
      "training_step_time": 8.327046394348145
    },
    {
      "epoch": 1.995849609375e-05,
      "model_forward_time": 1.714268445968628,
      "step": 20928
    },
    {
      "epoch": 1.995849609375e-05,
      "step": 20928,
      "training_step_time": 10.356395244598389
    },
    {
      "epoch": 1.9959449768066405e-05,
      "model_forward_time": 2.884681224822998,
      "step": 20929
    },
    {
      "epoch": 1.9959449768066405e-05,
      "step": 20929,
      "training_step_time": 8.495185613632202
    },
    {
      "epoch": 1.9960403442382812e-05,
      "grad_norm": 0.16647809743881226,
      "learning_rate": 9.322196265559982e-06,
      "loss": 0.0626,
      "step": 20930
    },
    {
      "epoch": 1.9960403442382812e-05,
      "model_forward_time": 1.7363615036010742,
      "step": 20930
    },
    {
      "dynamics_loss_avg": 0.07495390325784683,
      "epoch": 1.9960403442382812e-05,
      "step": 20930
    },
    {
      "action_loss_avg": 0.003307195450179279,
      "epoch": 1.9960403442382812e-05,
      "step": 20930
    },
    {
      "epoch": 1.9960403442382812e-05,
      "step": 20930,
      "training_step_time": 7.649789333343506
    },
    {
      "epoch": 1.996135711669922e-05,
      "model_forward_time": 1.714343786239624,
      "step": 20931
    },
    {
      "epoch": 1.996135711669922e-05,
      "step": 20931,
      "training_step_time": 7.665510654449463
    },
    {
      "epoch": 1.9962310791015624e-05,
      "model_forward_time": 1.7487139701843262,
      "step": 20932
    },
    {
      "epoch": 1.9962310791015624e-05,
      "step": 20932,
      "training_step_time": 7.584468603134155
    },
    {
      "epoch": 1.996326446533203e-05,
      "model_forward_time": 1.7152225971221924,
      "step": 20933
    },
    {
      "epoch": 1.996326446533203e-05,
      "step": 20933,
      "training_step_time": 7.811916828155518
    },
    {
      "epoch": 1.996421813964844e-05,
      "model_forward_time": 1.714259386062622,
      "step": 20934
    },
    {
      "epoch": 1.996421813964844e-05,
      "step": 20934,
      "training_step_time": 7.597570419311523
    },
    {
      "epoch": 1.9965171813964843e-05,
      "model_forward_time": 1.7137951850891113,
      "step": 20935
    },
    {
      "epoch": 1.9965171813964843e-05,
      "step": 20935,
      "training_step_time": 8.808462381362915
    },
    {
      "epoch": 1.996612548828125e-05,
      "model_forward_time": 1.7191944122314453,
      "step": 20936
    },
    {
      "epoch": 1.996612548828125e-05,
      "step": 20936,
      "training_step_time": 9.460747480392456
    },
    {
      "epoch": 1.9967079162597657e-05,
      "model_forward_time": 1.7073211669921875,
      "step": 20937
    },
    {
      "epoch": 1.9967079162597657e-05,
      "step": 20937,
      "training_step_time": 9.021069526672363
    },
    {
      "epoch": 1.996803283691406e-05,
      "model_forward_time": 2.8812763690948486,
      "step": 20938
    },
    {
      "epoch": 1.996803283691406e-05,
      "step": 20938,
      "training_step_time": 8.375620126724243
    },
    {
      "epoch": 1.996898651123047e-05,
      "model_forward_time": 1.7312958240509033,
      "step": 20939
    },
    {
      "epoch": 1.996898651123047e-05,
      "step": 20939,
      "training_step_time": 7.6926186084747314
    },
    {
      "epoch": 1.9969940185546876e-05,
      "grad_norm": 0.09752108156681061,
      "learning_rate": 9.321364768313803e-06,
      "loss": 0.0598,
      "step": 20940
    },
    {
      "epoch": 1.9969940185546876e-05,
      "model_forward_time": 1.7138936519622803,
      "step": 20940
    },
    {
      "dynamics_loss_avg": 0.054510955885052684,
      "epoch": 1.9969940185546876e-05,
      "step": 20940
    },
    {
      "action_loss_avg": 0.0018674967112019658,
      "epoch": 1.9969940185546876e-05,
      "step": 20940
    },
    {
      "epoch": 1.9969940185546876e-05,
      "step": 20940,
      "training_step_time": 7.606337785720825
    },
    {
      "epoch": 1.997089385986328e-05,
      "model_forward_time": 1.7433865070343018,
      "step": 20941
    },
    {
      "epoch": 1.997089385986328e-05,
      "step": 20941,
      "training_step_time": 7.538499355316162
    },
    {
      "epoch": 1.9971847534179688e-05,
      "model_forward_time": 1.7128016948699951,
      "step": 20942
    },
    {
      "epoch": 1.9971847534179688e-05,
      "step": 20942,
      "training_step_time": 7.341308355331421
    },
    {
      "epoch": 1.9972801208496095e-05,
      "model_forward_time": 1.720609188079834,
      "step": 20943
    },
    {
      "epoch": 1.9972801208496095e-05,
      "step": 20943,
      "training_step_time": 7.62049126625061
    },
    {
      "epoch": 1.99737548828125e-05,
      "model_forward_time": 1.712949275970459,
      "step": 20944
    },
    {
      "epoch": 1.99737548828125e-05,
      "step": 20944,
      "training_step_time": 9.586661577224731
    },
    {
      "epoch": 1.9974708557128907e-05,
      "model_forward_time": 1.7180399894714355,
      "step": 20945
    },
    {
      "epoch": 1.9974708557128907e-05,
      "step": 20945,
      "training_step_time": 7.820481777191162
    },
    {
      "epoch": 1.9975662231445314e-05,
      "model_forward_time": 1.71156907081604,
      "step": 20946
    },
    {
      "epoch": 1.9975662231445314e-05,
      "step": 20946,
      "training_step_time": 7.66006326675415
    },
    {
      "epoch": 1.9976615905761718e-05,
      "model_forward_time": 2.896435260772705,
      "step": 20947
    },
    {
      "epoch": 1.9976615905761718e-05,
      "step": 20947,
      "training_step_time": 7.728930950164795
    },
    {
      "epoch": 1.9977569580078125e-05,
      "model_forward_time": 1.7308692932128906,
      "step": 20948
    },
    {
      "epoch": 1.9977569580078125e-05,
      "step": 20948,
      "training_step_time": 8.60374903678894
    },
    {
      "epoch": 1.9978523254394533e-05,
      "model_forward_time": 1.72444748878479,
      "step": 20949
    },
    {
      "epoch": 1.9978523254394533e-05,
      "step": 20949,
      "training_step_time": 7.6172404289245605
    },
    {
      "epoch": 1.9979476928710937e-05,
      "grad_norm": 0.14206236600875854,
      "learning_rate": 9.32053279848967e-06,
      "loss": 0.0575,
      "step": 20950
    },
    {
      "epoch": 1.9979476928710937e-05,
      "model_forward_time": 1.746361494064331,
      "step": 20950
    },
    {
      "dynamics_loss_avg": 0.05921516437083483,
      "epoch": 1.9979476928710937e-05,
      "step": 20950
    },
    {
      "action_loss_avg": 0.0026318344287574293,
      "epoch": 1.9979476928710937e-05,
      "step": 20950
    },
    {
      "epoch": 1.9979476928710937e-05,
      "step": 20950,
      "training_step_time": 7.645585060119629
    },
    {
      "epoch": 1.9980430603027344e-05,
      "model_forward_time": 1.7183141708374023,
      "step": 20951
    },
    {
      "epoch": 1.9980430603027344e-05,
      "step": 20951,
      "training_step_time": 7.445440053939819
    },
    {
      "epoch": 1.998138427734375e-05,
      "model_forward_time": 1.7199933528900146,
      "step": 20952
    },
    {
      "epoch": 1.998138427734375e-05,
      "step": 20952,
      "training_step_time": 7.519489526748657
    },
    {
      "epoch": 1.9982337951660156e-05,
      "model_forward_time": 1.7211582660675049,
      "step": 20953
    },
    {
      "epoch": 1.9982337951660156e-05,
      "step": 20953,
      "training_step_time": 8.187686443328857
    },
    {
      "epoch": 1.9983291625976563e-05,
      "model_forward_time": 1.723006010055542,
      "step": 20954
    },
    {
      "epoch": 1.9983291625976563e-05,
      "step": 20954,
      "training_step_time": 8.737345457077026
    },
    {
      "epoch": 1.9984245300292967e-05,
      "model_forward_time": 1.724668264389038,
      "step": 20955
    },
    {
      "epoch": 1.9984245300292967e-05,
      "step": 20955,
      "training_step_time": 8.389504671096802
    },
    {
      "epoch": 1.9985198974609375e-05,
      "model_forward_time": 2.852152109146118,
      "step": 20956
    },
    {
      "epoch": 1.9985198974609375e-05,
      "step": 20956,
      "training_step_time": 7.841748237609863
    },
    {
      "epoch": 1.9986152648925782e-05,
      "model_forward_time": 1.7188377380371094,
      "step": 20957
    },
    {
      "epoch": 1.9986152648925782e-05,
      "step": 20957,
      "training_step_time": 7.587409973144531
    },
    {
      "epoch": 1.9987106323242186e-05,
      "model_forward_time": 1.747279167175293,
      "step": 20958
    },
    {
      "epoch": 1.9987106323242186e-05,
      "step": 20958,
      "training_step_time": 7.530590295791626
    },
    {
      "epoch": 1.9988059997558593e-05,
      "model_forward_time": 1.7336392402648926,
      "step": 20959
    },
    {
      "epoch": 1.9988059997558593e-05,
      "step": 20959,
      "training_step_time": 7.659173488616943
    },
    {
      "epoch": 1.9989013671875e-05,
      "grad_norm": 0.15188486874103546,
      "learning_rate": 9.319700356178567e-06,
      "loss": 0.0604,
      "step": 20960
    },
    {
      "epoch": 1.9989013671875e-05,
      "model_forward_time": 1.3439364433288574,
      "step": 20960
    },
    {
      "dynamics_loss_avg": 0.05736216083168984,
      "epoch": 1.9989013671875e-05,
      "step": 20960
    },
    {
      "action_loss_avg": 0.003972517093643546,
      "epoch": 1.9989013671875e-05,
      "step": 20960
    },
    {
      "epoch": 1.9989013671875e-05,
      "step": 20960,
      "training_step_time": 7.5458502769470215
    },
    {
      "epoch": 1.9989967346191405e-05,
      "model_forward_time": 1.7251439094543457,
      "step": 20961
    },
    {
      "epoch": 1.9989967346191405e-05,
      "step": 20961,
      "training_step_time": 7.526564359664917
    },
    {
      "epoch": 1.9990921020507812e-05,
      "model_forward_time": 1.723074197769165,
      "step": 20962
    },
    {
      "epoch": 1.9990921020507812e-05,
      "step": 20962,
      "training_step_time": 7.65606427192688
    },
    {
      "epoch": 1.999187469482422e-05,
      "model_forward_time": 1.7218570709228516,
      "step": 20963
    },
    {
      "epoch": 1.999187469482422e-05,
      "step": 20963,
      "training_step_time": 7.81467866897583
    },
    {
      "epoch": 1.9992828369140624e-05,
      "model_forward_time": 1.705996036529541,
      "step": 20964
    },
    {
      "epoch": 1.9992828369140624e-05,
      "step": 20964,
      "training_step_time": 7.676761150360107
    },
    {
      "epoch": 1.999378204345703e-05,
      "model_forward_time": 1.716869592666626,
      "step": 20965
    },
    {
      "epoch": 1.999378204345703e-05,
      "step": 20965,
      "training_step_time": 7.755229711532593
    },
    {
      "epoch": 1.999473571777344e-05,
      "model_forward_time": 1.710092544555664,
      "step": 20966
    },
    {
      "epoch": 1.999473571777344e-05,
      "step": 20966,
      "training_step_time": 8.73507022857666
    },
    {
      "epoch": 1.9995689392089843e-05,
      "model_forward_time": 1.7437756061553955,
      "step": 20967
    },
    {
      "epoch": 1.9995689392089843e-05,
      "step": 20967,
      "training_step_time": 7.547658443450928
    },
    {
      "epoch": 1.999664306640625e-05,
      "model_forward_time": 1.7283875942230225,
      "step": 20968
    },
    {
      "epoch": 1.999664306640625e-05,
      "step": 20968,
      "training_step_time": 7.559247016906738
    },
    {
      "epoch": 1.9997596740722657e-05,
      "model_forward_time": 1.722151756286621,
      "step": 20969
    },
    {
      "epoch": 1.9997596740722657e-05,
      "step": 20969,
      "training_step_time": 7.586975812911987
    },
    {
      "epoch": 1.999855041503906e-05,
      "grad_norm": 0.17928916215896606,
      "learning_rate": 9.318867441471527e-06,
      "loss": 0.0603,
      "step": 20970
    },
    {
      "epoch": 1.999855041503906e-05,
      "model_forward_time": 1.72025465965271,
      "step": 20970
    },
    {
      "dynamics_loss_avg": 0.05292091965675354,
      "epoch": 1.999855041503906e-05,
      "step": 20970
    },
    {
      "action_loss_avg": 0.0031368054158519955,
      "epoch": 1.999855041503906e-05,
      "step": 20970
    },
    {
      "epoch": 1.999855041503906e-05,
      "step": 20970,
      "training_step_time": 7.540378570556641
    },
    {
      "epoch": 1.999950408935547e-05,
      "model_forward_time": 1.7229337692260742,
      "step": 20971
    },
    {
      "epoch": 1.999950408935547e-05,
      "step": 20971,
      "training_step_time": 8.194509267807007
    },
    {
      "epoch": 2.0000457763671876e-05,
      "model_forward_time": 1.7242181301116943,
      "step": 20972
    },
    {
      "epoch": 2.0000457763671876e-05,
      "step": 20972,
      "training_step_time": 7.841653823852539
    },
    {
      "epoch": 2.000141143798828e-05,
      "model_forward_time": 1.7277352809906006,
      "step": 20973
    },
    {
      "epoch": 2.000141143798828e-05,
      "step": 20973,
      "training_step_time": 8.087789058685303
    },
    {
      "epoch": 2.0002365112304688e-05,
      "model_forward_time": 1.70994234085083,
      "step": 20974
    },
    {
      "epoch": 2.0002365112304688e-05,
      "step": 20974,
      "training_step_time": 7.758120775222778
    },
    {
      "epoch": 2.0003318786621095e-05,
      "model_forward_time": 2.86979341506958,
      "step": 20975
    },
    {
      "epoch": 2.0003318786621095e-05,
      "step": 20975,
      "training_step_time": 8.43908953666687
    },
    {
      "epoch": 2.00042724609375e-05,
      "model_forward_time": 1.7216355800628662,
      "step": 20976
    },
    {
      "epoch": 2.00042724609375e-05,
      "step": 20976,
      "training_step_time": 7.627809524536133
    },
    {
      "epoch": 2.0005226135253907e-05,
      "model_forward_time": 1.7076466083526611,
      "step": 20977
    },
    {
      "epoch": 2.0005226135253907e-05,
      "step": 20977,
      "training_step_time": 7.975904703140259
    },
    {
      "epoch": 2.0006179809570314e-05,
      "model_forward_time": 1.7051770687103271,
      "step": 20978
    },
    {
      "epoch": 2.0006179809570314e-05,
      "step": 20978,
      "training_step_time": 7.6192827224731445
    },
    {
      "epoch": 2.0007133483886718e-05,
      "model_forward_time": 1.710538387298584,
      "step": 20979
    },
    {
      "epoch": 2.0007133483886718e-05,
      "step": 20979,
      "training_step_time": 7.588969945907593
    },
    {
      "epoch": 2.0008087158203125e-05,
      "grad_norm": 0.17175808548927307,
      "learning_rate": 9.318034054459637e-06,
      "loss": 0.0585,
      "step": 20980
    },
    {
      "epoch": 2.0008087158203125e-05,
      "model_forward_time": 1.7113735675811768,
      "step": 20980
    },
    {
      "dynamics_loss_avg": 0.03929395172744989,
      "epoch": 2.0008087158203125e-05,
      "step": 20980
    },
    {
      "action_loss_avg": 0.002017315593548119,
      "epoch": 2.0008087158203125e-05,
      "step": 20980
    },
    {
      "epoch": 2.0008087158203125e-05,
      "step": 20980,
      "training_step_time": 8.190462827682495
    },
    {
      "epoch": 2.0009040832519533e-05,
      "model_forward_time": 1.721076488494873,
      "step": 20981
    },
    {
      "epoch": 2.0009040832519533e-05,
      "step": 20981,
      "training_step_time": 7.8307945728302
    },
    {
      "epoch": 2.0009994506835937e-05,
      "model_forward_time": 1.7236151695251465,
      "step": 20982
    },
    {
      "epoch": 2.0009994506835937e-05,
      "step": 20982,
      "training_step_time": 8.281593561172485
    },
    {
      "epoch": 2.0010948181152344e-05,
      "model_forward_time": 1.7165436744689941,
      "step": 20983
    },
    {
      "epoch": 2.0010948181152344e-05,
      "step": 20983,
      "training_step_time": 7.7513041496276855
    },
    {
      "epoch": 2.001190185546875e-05,
      "model_forward_time": 2.916355609893799,
      "step": 20984
    },
    {
      "epoch": 2.001190185546875e-05,
      "step": 20984,
      "training_step_time": 8.487703800201416
    },
    {
      "epoch": 2.0012855529785156e-05,
      "model_forward_time": 1.7374775409698486,
      "step": 20985
    },
    {
      "epoch": 2.0012855529785156e-05,
      "step": 20985,
      "training_step_time": 7.700181007385254
    },
    {
      "epoch": 2.0013809204101563e-05,
      "model_forward_time": 1.7226393222808838,
      "step": 20986
    },
    {
      "epoch": 2.0013809204101563e-05,
      "step": 20986,
      "training_step_time": 7.6660237312316895
    },
    {
      "epoch": 2.0014762878417967e-05,
      "model_forward_time": 1.7376203536987305,
      "step": 20987
    },
    {
      "epoch": 2.0014762878417967e-05,
      "step": 20987,
      "training_step_time": 7.710216760635376
    },
    {
      "epoch": 2.0015716552734375e-05,
      "model_forward_time": 1.7242457866668701,
      "step": 20988
    },
    {
      "epoch": 2.0015716552734375e-05,
      "step": 20988,
      "training_step_time": 8.634648323059082
    },
    {
      "epoch": 2.0016670227050782e-05,
      "model_forward_time": 1.7201082706451416,
      "step": 20989
    },
    {
      "epoch": 2.0016670227050782e-05,
      "step": 20989,
      "training_step_time": 7.70764684677124
    },
    {
      "epoch": 2.0017623901367186e-05,
      "grad_norm": 0.11783065646886826,
      "learning_rate": 9.317200195234035e-06,
      "loss": 0.0657,
      "step": 20990
    },
    {
      "epoch": 2.0017623901367186e-05,
      "model_forward_time": 1.7246580123901367,
      "step": 20990
    },
    {
      "dynamics_loss_avg": 0.04895306844264269,
      "epoch": 2.0017623901367186e-05,
      "step": 20990
    },
    {
      "action_loss_avg": 0.001781279081478715,
      "epoch": 2.0017623901367186e-05,
      "step": 20990
    },
    {
      "epoch": 2.0017623901367186e-05,
      "step": 20990,
      "training_step_time": 8.355748653411865
    },
    {
      "epoch": 2.0018577575683593e-05,
      "model_forward_time": 1.7221615314483643,
      "step": 20991
    },
    {
      "epoch": 2.0018577575683593e-05,
      "step": 20991,
      "training_step_time": 7.795430898666382
    },
    {
      "epoch": 2.001953125e-05,
      "model_forward_time": 1.722033977508545,
      "step": 20992
    },
    {
      "epoch": 2.001953125e-05,
      "step": 20992,
      "training_step_time": 7.760715484619141
    },
    {
      "epoch": 2.0020484924316405e-05,
      "model_forward_time": 2.8172225952148438,
      "step": 20993
    },
    {
      "epoch": 2.0020484924316405e-05,
      "step": 20993,
      "training_step_time": 8.553167581558228
    },
    {
      "epoch": 2.0021438598632812e-05,
      "model_forward_time": 1.7214319705963135,
      "step": 20994
    },
    {
      "epoch": 2.0021438598632812e-05,
      "step": 20994,
      "training_step_time": 8.521663427352905
    },
    {
      "epoch": 2.002239227294922e-05,
      "model_forward_time": 1.7233195304870605,
      "step": 20995
    },
    {
      "epoch": 2.002239227294922e-05,
      "step": 20995,
      "training_step_time": 8.224836587905884
    },
    {
      "epoch": 2.0023345947265624e-05,
      "model_forward_time": 1.7364616394042969,
      "step": 20996
    },
    {
      "epoch": 2.0023345947265624e-05,
      "step": 20996,
      "training_step_time": 7.548079013824463
    },
    {
      "epoch": 2.002429962158203e-05,
      "model_forward_time": 1.7411813735961914,
      "step": 20997
    },
    {
      "epoch": 2.002429962158203e-05,
      "step": 20997,
      "training_step_time": 7.665644645690918
    },
    {
      "epoch": 2.002525329589844e-05,
      "model_forward_time": 1.7986607551574707,
      "step": 20998
    },
    {
      "epoch": 2.002525329589844e-05,
      "step": 20998,
      "training_step_time": 7.84868311882019
    },
    {
      "epoch": 2.0026206970214843e-05,
      "model_forward_time": 1.7184326648712158,
      "step": 20999
    },
    {
      "epoch": 2.0026206970214843e-05,
      "step": 20999,
      "training_step_time": 7.942731618881226
    },
    {
      "epoch": 2.002716064453125e-05,
      "grad_norm": 0.18087974190711975,
      "learning_rate": 9.316365863885908e-06,
      "loss": 0.062,
      "step": 21000
    },
    {
      "epoch": 2.002716064453125e-05,
      "model_forward_time": 1.7416644096374512,
      "step": 21000
    },
    {
      "dynamics_loss_avg": 0.05695441886782646,
      "epoch": 2.002716064453125e-05,
      "step": 21000
    },
    {
      "action_loss_avg": 0.0023268520511919633,
      "epoch": 2.002716064453125e-05,
      "step": 21000
    },
    {
      "epoch": 2.002716064453125e-05,
      "step": 21000,
      "training_step_time": 6.561886310577393
    },
    {
      "epoch": 2.0028114318847657e-05,
      "model_forward_time": 1.746387004852295,
      "step": 21001
    },
    {
      "epoch": 2.0028114318847657e-05,
      "step": 21001,
      "training_step_time": 6.8095386028289795
    },
    {
      "epoch": 2.002906799316406e-05,
      "model_forward_time": 1.7445471286773682,
      "step": 21002
    },
    {
      "epoch": 2.002906799316406e-05,
      "step": 21002,
      "training_step_time": 6.820247650146484
    },
    {
      "epoch": 2.003002166748047e-05,
      "model_forward_time": 1.763382911682129,
      "step": 21003
    },
    {
      "epoch": 2.003002166748047e-05,
      "step": 21003,
      "training_step_time": 6.880294322967529
    },
    {
      "epoch": 2.0030975341796876e-05,
      "model_forward_time": 1.7524871826171875,
      "step": 21004
    },
    {
      "epoch": 2.0030975341796876e-05,
      "step": 21004,
      "training_step_time": 6.842691659927368
    },
    {
      "epoch": 2.003192901611328e-05,
      "model_forward_time": 1.7453887462615967,
      "step": 21005
    },
    {
      "epoch": 2.003192901611328e-05,
      "step": 21005,
      "training_step_time": 6.7914698123931885
    },
    {
      "epoch": 2.0032882690429688e-05,
      "model_forward_time": 1.7244787216186523,
      "step": 21006
    },
    {
      "epoch": 2.0032882690429688e-05,
      "step": 21006,
      "training_step_time": 6.817922830581665
    },
    {
      "epoch": 2.0033836364746095e-05,
      "model_forward_time": 1.7295424938201904,
      "step": 21007
    },
    {
      "epoch": 2.0033836364746095e-05,
      "step": 21007,
      "training_step_time": 6.841162443161011
    },
    {
      "epoch": 2.00347900390625e-05,
      "model_forward_time": 1.7420766353607178,
      "step": 21008
    },
    {
      "epoch": 2.00347900390625e-05,
      "step": 21008,
      "training_step_time": 8.921702861785889
    },
    {
      "epoch": 2.0035743713378907e-05,
      "model_forward_time": 2.7619423866271973,
      "step": 21009
    },
    {
      "epoch": 2.0035743713378907e-05,
      "step": 21009,
      "training_step_time": 7.660773038864136
    },
    {
      "epoch": 2.0036697387695314e-05,
      "grad_norm": 0.2485833764076233,
      "learning_rate": 9.315531060506503e-06,
      "loss": 0.0657,
      "step": 21010
    },
    {
      "epoch": 2.0036697387695314e-05,
      "model_forward_time": 1.7286038398742676,
      "step": 21010
    },
    {
      "dynamics_loss_avg": 0.0626552365720272,
      "epoch": 2.0036697387695314e-05,
      "step": 21010
    },
    {
      "action_loss_avg": 0.002891765086678788,
      "epoch": 2.0036697387695314e-05,
      "step": 21010
    },
    {
      "epoch": 2.0036697387695314e-05,
      "step": 21010,
      "training_step_time": 6.747655630111694
    },
    {
      "epoch": 2.0037651062011718e-05,
      "model_forward_time": 1.7446703910827637,
      "step": 21011
    },
    {
      "epoch": 2.0037651062011718e-05,
      "step": 21011,
      "training_step_time": 6.726689577102661
    },
    {
      "epoch": 2.0038604736328125e-05,
      "model_forward_time": 1.7414171695709229,
      "step": 21012
    },
    {
      "epoch": 2.0038604736328125e-05,
      "step": 21012,
      "training_step_time": 6.746260404586792
    },
    {
      "epoch": 2.0039558410644533e-05,
      "model_forward_time": 1.7229835987091064,
      "step": 21013
    },
    {
      "epoch": 2.0039558410644533e-05,
      "step": 21013,
      "training_step_time": 6.688876628875732
    },
    {
      "epoch": 2.0040512084960937e-05,
      "model_forward_time": 1.7305502891540527,
      "step": 21014
    },
    {
      "epoch": 2.0040512084960937e-05,
      "step": 21014,
      "training_step_time": 6.74861741065979
    },
    {
      "epoch": 2.0041465759277344e-05,
      "model_forward_time": 1.7330691814422607,
      "step": 21015
    },
    {
      "epoch": 2.0041465759277344e-05,
      "step": 21015,
      "training_step_time": 6.744199991226196
    },
    {
      "epoch": 2.004241943359375e-05,
      "model_forward_time": 1.7330100536346436,
      "step": 21016
    },
    {
      "epoch": 2.004241943359375e-05,
      "step": 21016,
      "training_step_time": 6.742493391036987
    },
    {
      "epoch": 2.0043373107910156e-05,
      "model_forward_time": 1.7402746677398682,
      "step": 21017
    },
    {
      "epoch": 2.0043373107910156e-05,
      "step": 21017,
      "training_step_time": 6.754080295562744
    },
    {
      "epoch": 2.0044326782226563e-05,
      "model_forward_time": 2.3112287521362305,
      "step": 21018
    },
    {
      "epoch": 2.0044326782226563e-05,
      "step": 21018,
      "training_step_time": 7.761642694473267
    },
    {
      "epoch": 2.0045280456542967e-05,
      "model_forward_time": 1.7424724102020264,
      "step": 21019
    },
    {
      "epoch": 2.0045280456542967e-05,
      "step": 21019,
      "training_step_time": 6.778658151626587
    },
    {
      "epoch": 2.0046234130859375e-05,
      "grad_norm": 0.18808908760547638,
      "learning_rate": 9.314695785187108e-06,
      "loss": 0.0662,
      "step": 21020
    },
    {
      "epoch": 2.0046234130859375e-05,
      "model_forward_time": 1.7295145988464355,
      "step": 21020
    },
    {
      "dynamics_loss_avg": 0.07558484729379415,
      "epoch": 2.0046234130859375e-05,
      "step": 21020
    },
    {
      "action_loss_avg": 0.005129491246771067,
      "epoch": 2.0046234130859375e-05,
      "step": 21020
    },
    {
      "epoch": 2.0046234130859375e-05,
      "step": 21020,
      "training_step_time": 6.771723031997681
    },
    {
      "epoch": 2.0047187805175782e-05,
      "model_forward_time": 1.7411413192749023,
      "step": 21021
    },
    {
      "epoch": 2.0047187805175782e-05,
      "step": 21021,
      "training_step_time": 6.776118278503418
    },
    {
      "epoch": 2.0048141479492186e-05,
      "model_forward_time": 1.761220932006836,
      "step": 21022
    },
    {
      "epoch": 2.0048141479492186e-05,
      "step": 21022,
      "training_step_time": 6.7838592529296875
    },
    {
      "epoch": 2.0049095153808593e-05,
      "model_forward_time": 1.745384931564331,
      "step": 21023
    },
    {
      "epoch": 2.0049095153808593e-05,
      "step": 21023,
      "training_step_time": 6.8787009716033936
    },
    {
      "epoch": 2.0050048828125e-05,
      "model_forward_time": 1.3572523593902588,
      "step": 21024
    },
    {
      "epoch": 2.0050048828125e-05,
      "step": 21024,
      "training_step_time": 6.801952362060547
    },
    {
      "epoch": 2.0051002502441405e-05,
      "model_forward_time": 1.736504316329956,
      "step": 21025
    },
    {
      "epoch": 2.0051002502441405e-05,
      "step": 21025,
      "training_step_time": 6.804419994354248
    },
    {
      "epoch": 2.0051956176757812e-05,
      "model_forward_time": 1.746610403060913,
      "step": 21026
    },
    {
      "epoch": 2.0051956176757812e-05,
      "step": 21026,
      "training_step_time": 6.806450128555298
    },
    {
      "epoch": 2.005290985107422e-05,
      "model_forward_time": 2.3542494773864746,
      "step": 21027
    },
    {
      "epoch": 2.005290985107422e-05,
      "step": 21027,
      "training_step_time": 7.897956371307373
    },
    {
      "epoch": 2.0053863525390624e-05,
      "model_forward_time": 1.7292168140411377,
      "step": 21028
    },
    {
      "epoch": 2.0053863525390624e-05,
      "step": 21028,
      "training_step_time": 6.84170389175415
    },
    {
      "epoch": 2.005481719970703e-05,
      "model_forward_time": 1.7472751140594482,
      "step": 21029
    },
    {
      "epoch": 2.005481719970703e-05,
      "step": 21029,
      "training_step_time": 6.824994802474976
    },
    {
      "epoch": 2.005577087402344e-05,
      "grad_norm": 0.32325148582458496,
      "learning_rate": 9.313860038019068e-06,
      "loss": 0.0645,
      "step": 21030
    },
    {
      "epoch": 2.005577087402344e-05,
      "model_forward_time": 1.743032455444336,
      "step": 21030
    },
    {
      "dynamics_loss_avg": 0.06856739167124033,
      "epoch": 2.005577087402344e-05,
      "step": 21030
    },
    {
      "action_loss_avg": 0.003605840867385268,
      "epoch": 2.005577087402344e-05,
      "step": 21030
    },
    {
      "epoch": 2.005577087402344e-05,
      "step": 21030,
      "training_step_time": 6.795904874801636
    },
    {
      "epoch": 2.0056724548339843e-05,
      "model_forward_time": 1.7483305931091309,
      "step": 21031
    },
    {
      "epoch": 2.0056724548339843e-05,
      "step": 21031,
      "training_step_time": 6.800708293914795
    },
    {
      "epoch": 2.005767822265625e-05,
      "model_forward_time": 1.7472658157348633,
      "step": 21032
    },
    {
      "epoch": 2.005767822265625e-05,
      "step": 21032,
      "training_step_time": 6.757776498794556
    },
    {
      "epoch": 2.0058631896972657e-05,
      "model_forward_time": 1.7507455348968506,
      "step": 21033
    },
    {
      "epoch": 2.0058631896972657e-05,
      "step": 21033,
      "training_step_time": 6.781372785568237
    },
    {
      "epoch": 2.005958557128906e-05,
      "model_forward_time": 1.814340353012085,
      "step": 21034
    },
    {
      "epoch": 2.005958557128906e-05,
      "step": 21034,
      "training_step_time": 6.777847528457642
    },
    {
      "epoch": 2.006053924560547e-05,
      "model_forward_time": 1.8398480415344238,
      "step": 21035
    },
    {
      "epoch": 2.006053924560547e-05,
      "step": 21035,
      "training_step_time": 6.822598457336426
    },
    {
      "epoch": 2.0061492919921876e-05,
      "model_forward_time": 2.356600046157837,
      "step": 21036
    },
    {
      "epoch": 2.0061492919921876e-05,
      "step": 21036,
      "training_step_time": 10.425239562988281
    },
    {
      "epoch": 2.006244659423828e-05,
      "model_forward_time": 1.8363902568817139,
      "step": 21037
    },
    {
      "epoch": 2.006244659423828e-05,
      "step": 21037,
      "training_step_time": 7.505563735961914
    },
    {
      "epoch": 2.0063400268554688e-05,
      "model_forward_time": 1.8296384811401367,
      "step": 21038
    },
    {
      "epoch": 2.0063400268554688e-05,
      "step": 21038,
      "training_step_time": 6.755780935287476
    },
    {
      "epoch": 2.0064353942871095e-05,
      "model_forward_time": 1.8363702297210693,
      "step": 21039
    },
    {
      "epoch": 2.0064353942871095e-05,
      "step": 21039,
      "training_step_time": 6.7477476596832275
    },
    {
      "epoch": 2.00653076171875e-05,
      "grad_norm": 0.16524256765842438,
      "learning_rate": 9.313023819093782e-06,
      "loss": 0.0653,
      "step": 21040
    },
    {
      "epoch": 2.00653076171875e-05,
      "model_forward_time": 1.83432936668396,
      "step": 21040
    },
    {
      "dynamics_loss_avg": 0.05120820049196482,
      "epoch": 2.00653076171875e-05,
      "step": 21040
    },
    {
      "action_loss_avg": 0.002494263753760606,
      "epoch": 2.00653076171875e-05,
      "step": 21040
    },
    {
      "epoch": 2.00653076171875e-05,
      "step": 21040,
      "training_step_time": 6.7539238929748535
    },
    {
      "epoch": 2.0066261291503907e-05,
      "model_forward_time": 1.112947702407837,
      "step": 21041
    },
    {
      "epoch": 2.0066261291503907e-05,
      "step": 21041,
      "training_step_time": 6.731760740280151
    },
    {
      "epoch": 2.0067214965820314e-05,
      "model_forward_time": 1.8318848609924316,
      "step": 21042
    },
    {
      "epoch": 2.0067214965820314e-05,
      "step": 21042,
      "training_step_time": 6.776730537414551
    },
    {
      "epoch": 2.0068168640136718e-05,
      "model_forward_time": 1.8328874111175537,
      "step": 21043
    },
    {
      "epoch": 2.0068168640136718e-05,
      "step": 21043,
      "training_step_time": 6.742621183395386
    },
    {
      "epoch": 2.0069122314453125e-05,
      "model_forward_time": 1.831329584121704,
      "step": 21044
    },
    {
      "epoch": 2.0069122314453125e-05,
      "step": 21044,
      "training_step_time": 6.7840821743011475
    },
    {
      "epoch": 2.0070075988769533e-05,
      "model_forward_time": 1.827657699584961,
      "step": 21045
    },
    {
      "epoch": 2.0070075988769533e-05,
      "step": 21045,
      "training_step_time": 8.390362024307251
    },
    {
      "epoch": 2.0071029663085937e-05,
      "model_forward_time": 2.3983640670776367,
      "step": 21046
    },
    {
      "epoch": 2.0071029663085937e-05,
      "step": 21046,
      "training_step_time": 7.7329816818237305
    },
    {
      "epoch": 2.0071983337402344e-05,
      "model_forward_time": 1.8419489860534668,
      "step": 21047
    },
    {
      "epoch": 2.0071983337402344e-05,
      "step": 21047,
      "training_step_time": 6.7410359382629395
    },
    {
      "epoch": 2.007293701171875e-05,
      "model_forward_time": 1.8283777236938477,
      "step": 21048
    },
    {
      "epoch": 2.007293701171875e-05,
      "step": 21048,
      "training_step_time": 6.709581136703491
    },
    {
      "epoch": 2.0073890686035156e-05,
      "model_forward_time": 1.8404173851013184,
      "step": 21049
    },
    {
      "epoch": 2.0073890686035156e-05,
      "step": 21049,
      "training_step_time": 6.740919351577759
    },
    {
      "epoch": 2.0074844360351563e-05,
      "grad_norm": 0.21034027636051178,
      "learning_rate": 9.312187128502696e-06,
      "loss": 0.0644,
      "step": 21050
    },
    {
      "epoch": 2.0074844360351563e-05,
      "model_forward_time": 1.843327283859253,
      "step": 21050
    },
    {
      "dynamics_loss_avg": 0.07619227245450019,
      "epoch": 2.0074844360351563e-05,
      "step": 21050
    },
    {
      "action_loss_avg": 0.0030861703446134923,
      "epoch": 2.0074844360351563e-05,
      "step": 21050
    },
    {
      "epoch": 2.0074844360351563e-05,
      "step": 21050,
      "training_step_time": 6.734541654586792
    },
    {
      "epoch": 2.0075798034667967e-05,
      "model_forward_time": 1.8298828601837158,
      "step": 21051
    },
    {
      "epoch": 2.0075798034667967e-05,
      "step": 21051,
      "training_step_time": 6.735570430755615
    },
    {
      "epoch": 2.0076751708984375e-05,
      "model_forward_time": 1.837311029434204,
      "step": 21052
    },
    {
      "epoch": 2.0076751708984375e-05,
      "step": 21052,
      "training_step_time": 6.71892523765564
    },
    {
      "epoch": 2.0077705383300782e-05,
      "model_forward_time": 1.8359456062316895,
      "step": 21053
    },
    {
      "epoch": 2.0077705383300782e-05,
      "step": 21053,
      "training_step_time": 6.760592699050903
    },
    {
      "epoch": 2.0078659057617186e-05,
      "model_forward_time": 1.8333208560943604,
      "step": 21054
    },
    {
      "epoch": 2.0078659057617186e-05,
      "step": 21054,
      "training_step_time": 9.322205781936646
    },
    {
      "epoch": 2.0079612731933593e-05,
      "model_forward_time": 2.3959431648254395,
      "step": 21055
    },
    {
      "epoch": 2.0079612731933593e-05,
      "step": 21055,
      "training_step_time": 7.620295286178589
    },
    {
      "epoch": 2.008056640625e-05,
      "model_forward_time": 1.8268647193908691,
      "step": 21056
    },
    {
      "epoch": 2.008056640625e-05,
      "step": 21056,
      "training_step_time": 7.571991920471191
    },
    {
      "epoch": 2.0081520080566405e-05,
      "model_forward_time": 1.8400969505310059,
      "step": 21057
    },
    {
      "epoch": 2.0081520080566405e-05,
      "step": 21057,
      "training_step_time": 6.717036247253418
    },
    {
      "epoch": 2.0082473754882812e-05,
      "model_forward_time": 1.839916706085205,
      "step": 21058
    },
    {
      "epoch": 2.0082473754882812e-05,
      "step": 21058,
      "training_step_time": 6.798689842224121
    },
    {
      "epoch": 2.008342742919922e-05,
      "model_forward_time": 1.836601972579956,
      "step": 21059
    },
    {
      "epoch": 2.008342742919922e-05,
      "step": 21059,
      "training_step_time": 6.790952444076538
    },
    {
      "epoch": 2.0084381103515624e-05,
      "grad_norm": 0.16318689286708832,
      "learning_rate": 9.311349966337308e-06,
      "loss": 0.0672,
      "step": 21060
    },
    {
      "epoch": 2.0084381103515624e-05,
      "model_forward_time": 1.8356339931488037,
      "step": 21060
    },
    {
      "dynamics_loss_avg": 0.05583439338952303,
      "epoch": 2.0084381103515624e-05,
      "step": 21060
    },
    {
      "action_loss_avg": 0.004665350634604692,
      "epoch": 2.0084381103515624e-05,
      "step": 21060
    },
    {
      "epoch": 2.0084381103515624e-05,
      "step": 21060,
      "training_step_time": 6.8017120361328125
    },
    {
      "epoch": 2.008533477783203e-05,
      "model_forward_time": 1.8324472904205322,
      "step": 21061
    },
    {
      "epoch": 2.008533477783203e-05,
      "step": 21061,
      "training_step_time": 6.800354242324829
    },
    {
      "epoch": 2.008628845214844e-05,
      "model_forward_time": 1.843144416809082,
      "step": 21062
    },
    {
      "epoch": 2.008628845214844e-05,
      "step": 21062,
      "training_step_time": 6.787163257598877
    },
    {
      "epoch": 2.0087242126464843e-05,
      "model_forward_time": 1.8291876316070557,
      "step": 21063
    },
    {
      "epoch": 2.0087242126464843e-05,
      "step": 21063,
      "training_step_time": 8.776564359664917
    },
    {
      "epoch": 2.008819580078125e-05,
      "model_forward_time": 2.3814966678619385,
      "step": 21064
    },
    {
      "epoch": 2.008819580078125e-05,
      "step": 21064,
      "training_step_time": 7.696913480758667
    },
    {
      "epoch": 2.0089149475097657e-05,
      "model_forward_time": 1.8346714973449707,
      "step": 21065
    },
    {
      "epoch": 2.0089149475097657e-05,
      "step": 21065,
      "training_step_time": 7.536823749542236
    },
    {
      "epoch": 2.009010314941406e-05,
      "model_forward_time": 1.8421261310577393,
      "step": 21066
    },
    {
      "epoch": 2.009010314941406e-05,
      "step": 21066,
      "training_step_time": 6.75926947593689
    },
    {
      "epoch": 2.009105682373047e-05,
      "model_forward_time": 1.8369371891021729,
      "step": 21067
    },
    {
      "epoch": 2.009105682373047e-05,
      "step": 21067,
      "training_step_time": 6.7682435512542725
    },
    {
      "epoch": 2.0092010498046876e-05,
      "model_forward_time": 1.8332910537719727,
      "step": 21068
    },
    {
      "epoch": 2.0092010498046876e-05,
      "step": 21068,
      "training_step_time": 6.769197225570679
    },
    {
      "epoch": 2.009296417236328e-05,
      "model_forward_time": 1.8366551399230957,
      "step": 21069
    },
    {
      "epoch": 2.009296417236328e-05,
      "step": 21069,
      "training_step_time": 6.792752265930176
    },
    {
      "epoch": 2.0093917846679688e-05,
      "grad_norm": 0.14227928221225739,
      "learning_rate": 9.31051233268917e-06,
      "loss": 0.0672,
      "step": 21070
    },
    {
      "epoch": 2.0093917846679688e-05,
      "model_forward_time": 1.8452889919281006,
      "step": 21070
    },
    {
      "dynamics_loss_avg": 0.05604371298104525,
      "epoch": 2.0093917846679688e-05,
      "step": 21070
    },
    {
      "action_loss_avg": 0.002733363083098084,
      "epoch": 2.0093917846679688e-05,
      "step": 21070
    },
    {
      "epoch": 2.0093917846679688e-05,
      "step": 21070,
      "training_step_time": 6.7491724491119385
    },
    {
      "epoch": 2.0094871520996095e-05,
      "model_forward_time": 1.8447484970092773,
      "step": 21071
    },
    {
      "epoch": 2.0094871520996095e-05,
      "step": 21071,
      "training_step_time": 6.801500082015991
    },
    {
      "epoch": 2.00958251953125e-05,
      "model_forward_time": 1.8405945301055908,
      "step": 21072
    },
    {
      "epoch": 2.00958251953125e-05,
      "step": 21072,
      "training_step_time": 8.479568243026733
    },
    {
      "epoch": 2.0096778869628907e-05,
      "model_forward_time": 2.3705732822418213,
      "step": 21073
    },
    {
      "epoch": 2.0096778869628907e-05,
      "step": 21073,
      "training_step_time": 9.919443130493164
    },
    {
      "epoch": 2.0097732543945314e-05,
      "model_forward_time": 1.8259944915771484,
      "step": 21074
    },
    {
      "epoch": 2.0097732543945314e-05,
      "step": 21074,
      "training_step_time": 7.533283948898315
    },
    {
      "epoch": 2.0098686218261718e-05,
      "model_forward_time": 1.8425686359405518,
      "step": 21075
    },
    {
      "epoch": 2.0098686218261718e-05,
      "step": 21075,
      "training_step_time": 7.531168460845947
    },
    {
      "epoch": 2.0099639892578125e-05,
      "model_forward_time": 1.8384864330291748,
      "step": 21076
    },
    {
      "epoch": 2.0099639892578125e-05,
      "step": 21076,
      "training_step_time": 6.755119800567627
    },
    {
      "epoch": 2.0100593566894533e-05,
      "model_forward_time": 1.8469648361206055,
      "step": 21077
    },
    {
      "epoch": 2.0100593566894533e-05,
      "step": 21077,
      "training_step_time": 6.814568281173706
    },
    {
      "epoch": 2.0101547241210937e-05,
      "model_forward_time": 1.8368330001831055,
      "step": 21078
    },
    {
      "epoch": 2.0101547241210937e-05,
      "step": 21078,
      "training_step_time": 6.773266792297363
    },
    {
      "epoch": 2.0102500915527344e-05,
      "model_forward_time": 1.834813117980957,
      "step": 21079
    },
    {
      "epoch": 2.0102500915527344e-05,
      "step": 21079,
      "training_step_time": 6.742067575454712
    },
    {
      "epoch": 2.010345458984375e-05,
      "grad_norm": 0.11769933253526688,
      "learning_rate": 9.309674227649883e-06,
      "loss": 0.0658,
      "step": 21080
    },
    {
      "epoch": 2.010345458984375e-05,
      "model_forward_time": 1.8326282501220703,
      "step": 21080
    },
    {
      "dynamics_loss_avg": 0.06368506252765656,
      "epoch": 2.010345458984375e-05,
      "step": 21080
    },
    {
      "action_loss_avg": 0.0036743406380992383,
      "epoch": 2.010345458984375e-05,
      "step": 21080
    },
    {
      "epoch": 2.010345458984375e-05,
      "step": 21080,
      "training_step_time": 6.7901611328125
    },
    {
      "epoch": 2.0104408264160156e-05,
      "model_forward_time": 1.822204828262329,
      "step": 21081
    },
    {
      "epoch": 2.0104408264160156e-05,
      "step": 21081,
      "training_step_time": 8.79625916481018
    },
    {
      "epoch": 2.0105361938476563e-05,
      "model_forward_time": 1.7996253967285156,
      "step": 21082
    },
    {
      "epoch": 2.0105361938476563e-05,
      "step": 21082,
      "training_step_time": 9.569672346115112
    },
    {
      "epoch": 2.0106315612792967e-05,
      "model_forward_time": 2.3288586139678955,
      "step": 21083
    },
    {
      "epoch": 2.0106315612792967e-05,
      "step": 21083,
      "training_step_time": 7.643654108047485
    },
    {
      "epoch": 2.0107269287109375e-05,
      "model_forward_time": 1.816075086593628,
      "step": 21084
    },
    {
      "epoch": 2.0107269287109375e-05,
      "step": 21084,
      "training_step_time": 7.616407632827759
    },
    {
      "epoch": 2.0108222961425782e-05,
      "model_forward_time": 1.8378722667694092,
      "step": 21085
    },
    {
      "epoch": 2.0108222961425782e-05,
      "step": 21085,
      "training_step_time": 6.89162540435791
    },
    {
      "epoch": 2.0109176635742186e-05,
      "model_forward_time": 1.8369500637054443,
      "step": 21086
    },
    {
      "epoch": 2.0109176635742186e-05,
      "step": 21086,
      "training_step_time": 6.8156938552856445
    },
    {
      "epoch": 2.0110130310058594e-05,
      "model_forward_time": 1.8265860080718994,
      "step": 21087
    },
    {
      "epoch": 2.0110130310058594e-05,
      "step": 21087,
      "training_step_time": 6.827737331390381
    },
    {
      "epoch": 2.0111083984375e-05,
      "model_forward_time": 1.8392274379730225,
      "step": 21088
    },
    {
      "epoch": 2.0111083984375e-05,
      "step": 21088,
      "training_step_time": 6.772378921508789
    },
    {
      "epoch": 2.0112037658691405e-05,
      "model_forward_time": 1.832437515258789,
      "step": 21089
    },
    {
      "epoch": 2.0112037658691405e-05,
      "step": 21089,
      "training_step_time": 6.789374589920044
    },
    {
      "epoch": 2.0112991333007812e-05,
      "grad_norm": 0.1264493763446808,
      "learning_rate": 9.308835651311103e-06,
      "loss": 0.067,
      "step": 21090
    },
    {
      "epoch": 2.0112991333007812e-05,
      "model_forward_time": 1.8316094875335693,
      "step": 21090
    },
    {
      "dynamics_loss_avg": 0.07263598181307315,
      "epoch": 2.0112991333007812e-05,
      "step": 21090
    },
    {
      "action_loss_avg": 0.00451752960216254,
      "epoch": 2.0112991333007812e-05,
      "step": 21090
    },
    {
      "epoch": 2.0112991333007812e-05,
      "step": 21090,
      "training_step_time": 8.68490982055664
    },
    {
      "epoch": 2.011394500732422e-05,
      "model_forward_time": 1.7990155220031738,
      "step": 21091
    },
    {
      "epoch": 2.011394500732422e-05,
      "step": 21091,
      "training_step_time": 8.723960399627686
    },
    {
      "epoch": 2.0114898681640624e-05,
      "model_forward_time": 2.368534564971924,
      "step": 21092
    },
    {
      "epoch": 2.0114898681640624e-05,
      "step": 21092,
      "training_step_time": 7.752615928649902
    },
    {
      "epoch": 2.011585235595703e-05,
      "model_forward_time": 1.8129899501800537,
      "step": 21093
    },
    {
      "epoch": 2.011585235595703e-05,
      "step": 21093,
      "training_step_time": 7.620660066604614
    },
    {
      "epoch": 2.011680603027344e-05,
      "model_forward_time": 1.8358254432678223,
      "step": 21094
    },
    {
      "epoch": 2.011680603027344e-05,
      "step": 21094,
      "training_step_time": 6.809132099151611
    },
    {
      "epoch": 2.0117759704589843e-05,
      "model_forward_time": 1.8383457660675049,
      "step": 21095
    },
    {
      "epoch": 2.0117759704589843e-05,
      "step": 21095,
      "training_step_time": 6.7921178340911865
    },
    {
      "epoch": 2.011871337890625e-05,
      "model_forward_time": 1.8247342109680176,
      "step": 21096
    },
    {
      "epoch": 2.011871337890625e-05,
      "step": 21096,
      "training_step_time": 6.829582214355469
    },
    {
      "epoch": 2.0119667053222657e-05,
      "model_forward_time": 1.8255553245544434,
      "step": 21097
    },
    {
      "epoch": 2.0119667053222657e-05,
      "step": 21097,
      "training_step_time": 6.8551037311553955
    },
    {
      "epoch": 2.012062072753906e-05,
      "model_forward_time": 1.8356752395629883,
      "step": 21098
    },
    {
      "epoch": 2.012062072753906e-05,
      "step": 21098,
      "training_step_time": 6.839468717575073
    },
    {
      "epoch": 2.012157440185547e-05,
      "model_forward_time": 1.8220410346984863,
      "step": 21099
    },
    {
      "epoch": 2.012157440185547e-05,
      "step": 21099,
      "training_step_time": 8.437268257141113
    },
    {
      "epoch": 2.0122528076171876e-05,
      "grad_norm": 0.14183109998703003,
      "learning_rate": 9.307996603764533e-06,
      "loss": 0.0661,
      "step": 21100
    },
    {
      "epoch": 2.0122528076171876e-05,
      "model_forward_time": 1.8325026035308838,
      "step": 21100
    },
    {
      "dynamics_loss_avg": 0.05777420978993177,
      "epoch": 2.0122528076171876e-05,
      "step": 21100
    },
    {
      "action_loss_avg": 0.0021734399429988115,
      "epoch": 2.0122528076171876e-05,
      "step": 21100
    },
    {
      "epoch": 2.0122528076171876e-05,
      "step": 21100,
      "training_step_time": 10.922239542007446
    },
    {
      "epoch": 2.012348175048828e-05,
      "model_forward_time": 2.3803718090057373,
      "step": 21101
    },
    {
      "epoch": 2.012348175048828e-05,
      "step": 21101,
      "training_step_time": 9.65635371208191
    },
    {
      "epoch": 2.0124435424804688e-05,
      "model_forward_time": 1.8389396667480469,
      "step": 21102
    },
    {
      "epoch": 2.0124435424804688e-05,
      "step": 21102,
      "training_step_time": 6.812180757522583
    },
    {
      "epoch": 2.0125389099121095e-05,
      "model_forward_time": 1.842677116394043,
      "step": 21103
    },
    {
      "epoch": 2.0125389099121095e-05,
      "step": 21103,
      "training_step_time": 7.621192932128906
    },
    {
      "epoch": 2.01263427734375e-05,
      "model_forward_time": 1.8422198295593262,
      "step": 21104
    },
    {
      "epoch": 2.01263427734375e-05,
      "step": 21104,
      "training_step_time": 6.789725303649902
    },
    {
      "epoch": 2.0127296447753907e-05,
      "model_forward_time": 1.8355326652526855,
      "step": 21105
    },
    {
      "epoch": 2.0127296447753907e-05,
      "step": 21105,
      "training_step_time": 6.750753164291382
    },
    {
      "epoch": 2.0128250122070314e-05,
      "model_forward_time": 1.8394591808319092,
      "step": 21106
    },
    {
      "epoch": 2.0128250122070314e-05,
      "step": 21106,
      "training_step_time": 6.796396970748901
    },
    {
      "epoch": 2.0129203796386718e-05,
      "model_forward_time": 1.833737850189209,
      "step": 21107
    },
    {
      "epoch": 2.0129203796386718e-05,
      "step": 21107,
      "training_step_time": 6.746914863586426
    },
    {
      "epoch": 2.0130157470703126e-05,
      "model_forward_time": 1.8369650840759277,
      "step": 21108
    },
    {
      "epoch": 2.0130157470703126e-05,
      "step": 21108,
      "training_step_time": 7.649302244186401
    },
    {
      "epoch": 2.0131111145019533e-05,
      "model_forward_time": 1.8346614837646484,
      "step": 21109
    },
    {
      "epoch": 2.0131111145019533e-05,
      "step": 21109,
      "training_step_time": 11.249539852142334
    },
    {
      "epoch": 2.0132064819335937e-05,
      "grad_norm": 0.19889074563980103,
      "learning_rate": 9.307157085101932e-06,
      "loss": 0.0666,
      "step": 21110
    },
    {
      "epoch": 2.0132064819335937e-05,
      "model_forward_time": 2.3717598915100098,
      "step": 21110
    },
    {
      "dynamics_loss_avg": 0.06290560886263848,
      "epoch": 2.0132064819335937e-05,
      "step": 21110
    },
    {
      "action_loss_avg": 0.003919813834363595,
      "epoch": 2.0132064819335937e-05,
      "step": 21110
    },
    {
      "epoch": 2.0132064819335937e-05,
      "step": 21110,
      "training_step_time": 8.794872999191284
    },
    {
      "epoch": 2.0133018493652344e-05,
      "model_forward_time": 1.8365850448608398,
      "step": 21111
    },
    {
      "epoch": 2.0133018493652344e-05,
      "step": 21111,
      "training_step_time": 7.731275796890259
    },
    {
      "epoch": 2.013397216796875e-05,
      "model_forward_time": 1.8356904983520508,
      "step": 21112
    },
    {
      "epoch": 2.013397216796875e-05,
      "step": 21112,
      "training_step_time": 7.616704940795898
    },
    {
      "epoch": 2.0134925842285156e-05,
      "model_forward_time": 1.8323235511779785,
      "step": 21113
    },
    {
      "epoch": 2.0134925842285156e-05,
      "step": 21113,
      "training_step_time": 6.711325407028198
    },
    {
      "epoch": 2.0135879516601563e-05,
      "model_forward_time": 1.832486867904663,
      "step": 21114
    },
    {
      "epoch": 2.0135879516601563e-05,
      "step": 21114,
      "training_step_time": 6.735070466995239
    },
    {
      "epoch": 2.0136833190917967e-05,
      "model_forward_time": 1.8376531600952148,
      "step": 21115
    },
    {
      "epoch": 2.0136833190917967e-05,
      "step": 21115,
      "training_step_time": 6.746490716934204
    },
    {
      "epoch": 2.0137786865234375e-05,
      "model_forward_time": 1.8342170715332031,
      "step": 21116
    },
    {
      "epoch": 2.0137786865234375e-05,
      "step": 21116,
      "training_step_time": 6.714797019958496
    },
    {
      "epoch": 2.0138740539550782e-05,
      "model_forward_time": 1.839165210723877,
      "step": 21117
    },
    {
      "epoch": 2.0138740539550782e-05,
      "step": 21117,
      "training_step_time": 7.609190464019775
    },
    {
      "epoch": 2.0139694213867186e-05,
      "model_forward_time": 1.8336410522460938,
      "step": 21118
    },
    {
      "epoch": 2.0139694213867186e-05,
      "step": 21118,
      "training_step_time": 10.597766160964966
    },
    {
      "epoch": 2.0140647888183594e-05,
      "model_forward_time": 2.37490177154541,
      "step": 21119
    },
    {
      "epoch": 2.0140647888183594e-05,
      "step": 21119,
      "training_step_time": 8.500976800918579
    },
    {
      "epoch": 2.01416015625e-05,
      "grad_norm": 0.14957284927368164,
      "learning_rate": 9.30631709541511e-06,
      "loss": 0.0644,
      "step": 21120
    },
    {
      "epoch": 2.01416015625e-05,
      "model_forward_time": 1.827937126159668,
      "step": 21120
    },
    {
      "dynamics_loss_avg": 0.056919447146356106,
      "epoch": 2.01416015625e-05,
      "step": 21120
    },
    {
      "action_loss_avg": 0.0050855646142736076,
      "epoch": 2.01416015625e-05,
      "step": 21120
    },
    {
      "epoch": 2.01416015625e-05,
      "step": 21120,
      "training_step_time": 8.285064458847046
    },
    {
      "epoch": 2.0142555236816405e-05,
      "model_forward_time": 1.8351597785949707,
      "step": 21121
    },
    {
      "epoch": 2.0142555236816405e-05,
      "step": 21121,
      "training_step_time": 7.656621217727661
    },
    {
      "epoch": 2.0143508911132812e-05,
      "model_forward_time": 1.835463523864746,
      "step": 21122
    },
    {
      "epoch": 2.0143508911132812e-05,
      "step": 21122,
      "training_step_time": 6.812456369400024
    },
    {
      "epoch": 2.014446258544922e-05,
      "model_forward_time": 1.8289353847503662,
      "step": 21123
    },
    {
      "epoch": 2.014446258544922e-05,
      "step": 21123,
      "training_step_time": 6.809999704360962
    },
    {
      "epoch": 2.0145416259765624e-05,
      "model_forward_time": 1.8299884796142578,
      "step": 21124
    },
    {
      "epoch": 2.0145416259765624e-05,
      "step": 21124,
      "training_step_time": 6.798553466796875
    },
    {
      "epoch": 2.014636993408203e-05,
      "model_forward_time": 1.8344013690948486,
      "step": 21125
    },
    {
      "epoch": 2.014636993408203e-05,
      "step": 21125,
      "training_step_time": 6.8136796951293945
    },
    {
      "epoch": 2.014732360839844e-05,
      "model_forward_time": 1.842581033706665,
      "step": 21126
    },
    {
      "epoch": 2.014732360839844e-05,
      "step": 21126,
      "training_step_time": 7.661118745803833
    },
    {
      "epoch": 2.0148277282714843e-05,
      "model_forward_time": 1.8269500732421875,
      "step": 21127
    },
    {
      "epoch": 2.0148277282714843e-05,
      "step": 21127,
      "training_step_time": 9.599124908447266
    },
    {
      "epoch": 2.014923095703125e-05,
      "model_forward_time": 2.3903615474700928,
      "step": 21128
    },
    {
      "epoch": 2.014923095703125e-05,
      "step": 21128,
      "training_step_time": 8.745648622512817
    },
    {
      "epoch": 2.0150184631347658e-05,
      "model_forward_time": 1.8369805812835693,
      "step": 21129
    },
    {
      "epoch": 2.0150184631347658e-05,
      "step": 21129,
      "training_step_time": 7.769526481628418
    },
    {
      "epoch": 2.015113830566406e-05,
      "grad_norm": 0.2138761729001999,
      "learning_rate": 9.305476634795922e-06,
      "loss": 0.0617,
      "step": 21130
    },
    {
      "epoch": 2.015113830566406e-05,
      "model_forward_time": 1.8302009105682373,
      "step": 21130
    },
    {
      "dynamics_loss_avg": 0.0532351978123188,
      "epoch": 2.015113830566406e-05,
      "step": 21130
    },
    {
      "action_loss_avg": 0.003258761530742049,
      "epoch": 2.015113830566406e-05,
      "step": 21130
    },
    {
      "epoch": 2.015113830566406e-05,
      "step": 21130,
      "training_step_time": 7.658448219299316
    },
    {
      "epoch": 2.015209197998047e-05,
      "model_forward_time": 1.8406050205230713,
      "step": 21131
    },
    {
      "epoch": 2.015209197998047e-05,
      "step": 21131,
      "training_step_time": 6.875951766967773
    },
    {
      "epoch": 2.0153045654296876e-05,
      "model_forward_time": 1.8310050964355469,
      "step": 21132
    },
    {
      "epoch": 2.0153045654296876e-05,
      "step": 21132,
      "training_step_time": 6.862682104110718
    },
    {
      "epoch": 2.015399932861328e-05,
      "model_forward_time": 1.83575439453125,
      "step": 21133
    },
    {
      "epoch": 2.015399932861328e-05,
      "step": 21133,
      "training_step_time": 6.8770363330841064
    },
    {
      "epoch": 2.0154953002929688e-05,
      "model_forward_time": 1.8373870849609375,
      "step": 21134
    },
    {
      "epoch": 2.0154953002929688e-05,
      "step": 21134,
      "training_step_time": 6.852270841598511
    },
    {
      "epoch": 2.0155906677246095e-05,
      "model_forward_time": 1.8261425495147705,
      "step": 21135
    },
    {
      "epoch": 2.0155906677246095e-05,
      "step": 21135,
      "training_step_time": 7.600102186203003
    },
    {
      "epoch": 2.01568603515625e-05,
      "model_forward_time": 1.8330161571502686,
      "step": 21136
    },
    {
      "epoch": 2.01568603515625e-05,
      "step": 21136,
      "training_step_time": 9.377660512924194
    },
    {
      "epoch": 2.0157814025878907e-05,
      "model_forward_time": 2.385617971420288,
      "step": 21137
    },
    {
      "epoch": 2.0157814025878907e-05,
      "step": 21137,
      "training_step_time": 8.954227447509766
    },
    {
      "epoch": 2.0158767700195314e-05,
      "model_forward_time": 1.7378020286560059,
      "step": 21138
    },
    {
      "epoch": 2.0158767700195314e-05,
      "step": 21138,
      "training_step_time": 7.7724080085754395
    },
    {
      "epoch": 2.0159721374511718e-05,
      "model_forward_time": 1.740523099899292,
      "step": 21139
    },
    {
      "epoch": 2.0159721374511718e-05,
      "step": 21139,
      "training_step_time": 7.691652536392212
    },
    {
      "epoch": 2.0160675048828126e-05,
      "grad_norm": 0.21735745668411255,
      "learning_rate": 9.304635703336284e-06,
      "loss": 0.0625,
      "step": 21140
    },
    {
      "epoch": 2.0160675048828126e-05,
      "model_forward_time": 1.7364397048950195,
      "step": 21140
    },
    {
      "dynamics_loss_avg": 0.07745757307857275,
      "epoch": 2.0160675048828126e-05,
      "step": 21140
    },
    {
      "action_loss_avg": 0.003564605786232278,
      "epoch": 2.0160675048828126e-05,
      "step": 21140
    },
    {
      "epoch": 2.0160675048828126e-05,
      "step": 21140,
      "training_step_time": 7.625826358795166
    },
    {
      "epoch": 2.016162872314453e-05,
      "model_forward_time": 1.7381055355072021,
      "step": 21141
    },
    {
      "epoch": 2.016162872314453e-05,
      "step": 21141,
      "training_step_time": 6.782957077026367
    },
    {
      "epoch": 2.0162582397460937e-05,
      "model_forward_time": 1.745610237121582,
      "step": 21142
    },
    {
      "epoch": 2.0162582397460937e-05,
      "step": 21142,
      "training_step_time": 6.797340631484985
    },
    {
      "epoch": 2.0163536071777344e-05,
      "model_forward_time": 1.7446997165679932,
      "step": 21143
    },
    {
      "epoch": 2.0163536071777344e-05,
      "step": 21143,
      "training_step_time": 6.783123016357422
    },
    {
      "epoch": 2.016448974609375e-05,
      "model_forward_time": 1.7450709342956543,
      "step": 21144
    },
    {
      "epoch": 2.016448974609375e-05,
      "step": 21144,
      "training_step_time": 8.430053949356079
    },
    {
      "epoch": 2.0165443420410156e-05,
      "model_forward_time": 1.8074681758880615,
      "step": 21145
    },
    {
      "epoch": 2.0165443420410156e-05,
      "step": 21145,
      "training_step_time": 8.169252395629883
    },
    {
      "epoch": 2.0166397094726563e-05,
      "model_forward_time": 2.9327826499938965,
      "step": 21146
    },
    {
      "epoch": 2.0166397094726563e-05,
      "step": 21146,
      "training_step_time": 9.020636320114136
    },
    {
      "epoch": 2.0167350769042967e-05,
      "model_forward_time": 1.753671407699585,
      "step": 21147
    },
    {
      "epoch": 2.0167350769042967e-05,
      "step": 21147,
      "training_step_time": 8.632018566131592
    },
    {
      "epoch": 2.0168304443359375e-05,
      "model_forward_time": 1.737908124923706,
      "step": 21148
    },
    {
      "epoch": 2.0168304443359375e-05,
      "step": 21148,
      "training_step_time": 7.799846649169922
    },
    {
      "epoch": 2.0169258117675782e-05,
      "model_forward_time": 1.7460055351257324,
      "step": 21149
    },
    {
      "epoch": 2.0169258117675782e-05,
      "step": 21149,
      "training_step_time": 7.84137749671936
    },
    {
      "epoch": 2.0170211791992186e-05,
      "grad_norm": 0.19422610104084015,
      "learning_rate": 9.303794301128157e-06,
      "loss": 0.0648,
      "step": 21150
    },
    {
      "epoch": 2.0170211791992186e-05,
      "model_forward_time": 1.7418694496154785,
      "step": 21150
    },
    {
      "dynamics_loss_avg": 0.07356637977063656,
      "epoch": 2.0170211791992186e-05,
      "step": 21150
    },
    {
      "action_loss_avg": 0.003913850133540109,
      "epoch": 2.0170211791992186e-05,
      "step": 21150
    },
    {
      "epoch": 2.0170211791992186e-05,
      "step": 21150,
      "training_step_time": 6.828515529632568
    },
    {
      "epoch": 2.0171165466308594e-05,
      "model_forward_time": 1.745610237121582,
      "step": 21151
    },
    {
      "epoch": 2.0171165466308594e-05,
      "step": 21151,
      "training_step_time": 6.7927162647247314
    },
    {
      "epoch": 2.0172119140625e-05,
      "model_forward_time": 1.7439839839935303,
      "step": 21152
    },
    {
      "epoch": 2.0172119140625e-05,
      "step": 21152,
      "training_step_time": 6.845548868179321
    },
    {
      "epoch": 2.0173072814941405e-05,
      "model_forward_time": 1.7183239459991455,
      "step": 21153
    },
    {
      "epoch": 2.0173072814941405e-05,
      "step": 21153,
      "training_step_time": 7.615903854370117
    },
    {
      "epoch": 2.0174026489257812e-05,
      "model_forward_time": 1.7609567642211914,
      "step": 21154
    },
    {
      "epoch": 2.0174026489257812e-05,
      "step": 21154,
      "training_step_time": 7.45468544960022
    },
    {
      "epoch": 2.017498016357422e-05,
      "model_forward_time": 2.8575687408447266,
      "step": 21155
    },
    {
      "epoch": 2.017498016357422e-05,
      "step": 21155,
      "training_step_time": 7.724011421203613
    },
    {
      "epoch": 2.0175933837890624e-05,
      "model_forward_time": 1.6983201503753662,
      "step": 21156
    },
    {
      "epoch": 2.0175933837890624e-05,
      "step": 21156,
      "training_step_time": 10.128318786621094
    },
    {
      "epoch": 2.017688751220703e-05,
      "model_forward_time": 1.7562685012817383,
      "step": 21157
    },
    {
      "epoch": 2.017688751220703e-05,
      "step": 21157,
      "training_step_time": 7.7941694259643555
    },
    {
      "epoch": 2.017784118652344e-05,
      "model_forward_time": 1.725234031677246,
      "step": 21158
    },
    {
      "epoch": 2.017784118652344e-05,
      "step": 21158,
      "training_step_time": 7.611975908279419
    },
    {
      "epoch": 2.0178794860839843e-05,
      "model_forward_time": 1.7254009246826172,
      "step": 21159
    },
    {
      "epoch": 2.0178794860839843e-05,
      "step": 21159,
      "training_step_time": 6.8098227977752686
    },
    {
      "epoch": 2.017974853515625e-05,
      "grad_norm": 0.14501841366291046,
      "learning_rate": 9.302952428263555e-06,
      "loss": 0.0605,
      "step": 21160
    },
    {
      "epoch": 2.017974853515625e-05,
      "model_forward_time": 1.7110557556152344,
      "step": 21160
    },
    {
      "dynamics_loss_avg": 0.06087448056787252,
      "epoch": 2.017974853515625e-05,
      "step": 21160
    },
    {
      "action_loss_avg": 0.0030666063947137447,
      "epoch": 2.017974853515625e-05,
      "step": 21160
    },
    {
      "epoch": 2.017974853515625e-05,
      "step": 21160,
      "training_step_time": 6.815241575241089
    },
    {
      "epoch": 2.0180702209472658e-05,
      "model_forward_time": 1.7184152603149414,
      "step": 21161
    },
    {
      "epoch": 2.0180702209472658e-05,
      "step": 21161,
      "training_step_time": 6.8455235958099365
    },
    {
      "epoch": 2.018165588378906e-05,
      "model_forward_time": 1.719858169555664,
      "step": 21162
    },
    {
      "epoch": 2.018165588378906e-05,
      "step": 21162,
      "training_step_time": 7.669340372085571
    },
    {
      "epoch": 2.018260955810547e-05,
      "model_forward_time": 1.7252120971679688,
      "step": 21163
    },
    {
      "epoch": 2.018260955810547e-05,
      "step": 21163,
      "training_step_time": 8.36160945892334
    },
    {
      "epoch": 2.0183563232421876e-05,
      "model_forward_time": 2.264242172241211,
      "step": 21164
    },
    {
      "epoch": 2.0183563232421876e-05,
      "step": 21164,
      "training_step_time": 7.692384719848633
    },
    {
      "epoch": 2.018451690673828e-05,
      "model_forward_time": 1.7206931114196777,
      "step": 21165
    },
    {
      "epoch": 2.018451690673828e-05,
      "step": 21165,
      "training_step_time": 8.524976015090942
    },
    {
      "epoch": 2.0185470581054688e-05,
      "model_forward_time": 1.7341885566711426,
      "step": 21166
    },
    {
      "epoch": 2.0185470581054688e-05,
      "step": 21166,
      "training_step_time": 7.797055244445801
    },
    {
      "epoch": 2.0186424255371095e-05,
      "model_forward_time": 1.7261738777160645,
      "step": 21167
    },
    {
      "epoch": 2.0186424255371095e-05,
      "step": 21167,
      "training_step_time": 7.7630720138549805
    },
    {
      "epoch": 2.01873779296875e-05,
      "model_forward_time": 1.7403068542480469,
      "step": 21168
    },
    {
      "epoch": 2.01873779296875e-05,
      "step": 21168,
      "training_step_time": 6.836222887039185
    },
    {
      "epoch": 2.0188331604003907e-05,
      "model_forward_time": 1.7152478694915771,
      "step": 21169
    },
    {
      "epoch": 2.0188331604003907e-05,
      "step": 21169,
      "training_step_time": 6.83025860786438
    },
    {
      "epoch": 2.0189285278320314e-05,
      "grad_norm": 0.1470821350812912,
      "learning_rate": 9.302110084834545e-06,
      "loss": 0.0624,
      "step": 21170
    },
    {
      "epoch": 2.0189285278320314e-05,
      "model_forward_time": 1.7125930786132812,
      "step": 21170
    },
    {
      "dynamics_loss_avg": 0.051227652467787266,
      "epoch": 2.0189285278320314e-05,
      "step": 21170
    },
    {
      "action_loss_avg": 0.0021186710451729595,
      "epoch": 2.0189285278320314e-05,
      "step": 21170
    },
    {
      "epoch": 2.0189285278320314e-05,
      "step": 21170,
      "training_step_time": 6.824868679046631
    },
    {
      "epoch": 2.0190238952636718e-05,
      "model_forward_time": 1.7340340614318848,
      "step": 21171
    },
    {
      "epoch": 2.0190238952636718e-05,
      "step": 21171,
      "training_step_time": 7.7389750480651855
    },
    {
      "epoch": 2.0191192626953126e-05,
      "model_forward_time": 1.7201123237609863,
      "step": 21172
    },
    {
      "epoch": 2.0191192626953126e-05,
      "step": 21172,
      "training_step_time": 8.64999532699585
    },
    {
      "epoch": 2.019214630126953e-05,
      "model_forward_time": 2.2525722980499268,
      "step": 21173
    },
    {
      "epoch": 2.019214630126953e-05,
      "step": 21173,
      "training_step_time": 8.286156415939331
    },
    {
      "epoch": 2.0193099975585937e-05,
      "model_forward_time": 1.7269065380096436,
      "step": 21174
    },
    {
      "epoch": 2.0193099975585937e-05,
      "step": 21174,
      "training_step_time": 9.091807126998901
    },
    {
      "epoch": 2.0194053649902344e-05,
      "model_forward_time": 1.7217390537261963,
      "step": 21175
    },
    {
      "epoch": 2.0194053649902344e-05,
      "step": 21175,
      "training_step_time": 7.786143779754639
    },
    {
      "epoch": 2.019500732421875e-05,
      "model_forward_time": 1.7915163040161133,
      "step": 21176
    },
    {
      "epoch": 2.019500732421875e-05,
      "step": 21176,
      "training_step_time": 7.847218990325928
    },
    {
      "epoch": 2.0195960998535156e-05,
      "model_forward_time": 1.7307276725769043,
      "step": 21177
    },
    {
      "epoch": 2.0195960998535156e-05,
      "step": 21177,
      "training_step_time": 7.7521812915802
    },
    {
      "epoch": 2.0196914672851563e-05,
      "model_forward_time": 1.7215805053710938,
      "step": 21178
    },
    {
      "epoch": 2.0196914672851563e-05,
      "step": 21178,
      "training_step_time": 6.885813474655151
    },
    {
      "epoch": 2.0197868347167967e-05,
      "model_forward_time": 1.7242646217346191,
      "step": 21179
    },
    {
      "epoch": 2.0197868347167967e-05,
      "step": 21179,
      "training_step_time": 6.8522868156433105
    },
    {
      "epoch": 2.0198822021484375e-05,
      "grad_norm": 0.17022769153118134,
      "learning_rate": 9.301267270933246e-06,
      "loss": 0.062,
      "step": 21180
    },
    {
      "epoch": 2.0198822021484375e-05,
      "model_forward_time": 1.7146382331848145,
      "step": 21180
    },
    {
      "dynamics_loss_avg": 0.05772015703842044,
      "epoch": 2.0198822021484375e-05,
      "step": 21180
    },
    {
      "action_loss_avg": 0.0029494687099941076,
      "epoch": 2.0198822021484375e-05,
      "step": 21180
    },
    {
      "epoch": 2.0198822021484375e-05,
      "step": 21180,
      "training_step_time": 7.62639594078064
    },
    {
      "epoch": 2.0199775695800782e-05,
      "model_forward_time": 1.7361552715301514,
      "step": 21181
    },
    {
      "epoch": 2.0199775695800782e-05,
      "step": 21181,
      "training_step_time": 7.541455268859863
    },
    {
      "epoch": 2.0200729370117186e-05,
      "model_forward_time": 2.280271053314209,
      "step": 21182
    },
    {
      "epoch": 2.0200729370117186e-05,
      "step": 21182,
      "training_step_time": 8.215879440307617
    },
    {
      "epoch": 2.0201683044433594e-05,
      "model_forward_time": 1.725954532623291,
      "step": 21183
    },
    {
      "epoch": 2.0201683044433594e-05,
      "step": 21183,
      "training_step_time": 9.465240240097046
    },
    {
      "epoch": 2.020263671875e-05,
      "model_forward_time": 1.72359299659729,
      "step": 21184
    },
    {
      "epoch": 2.020263671875e-05,
      "step": 21184,
      "training_step_time": 7.6962504386901855
    },
    {
      "epoch": 2.0203590393066405e-05,
      "model_forward_time": 1.7740566730499268,
      "step": 21185
    },
    {
      "epoch": 2.0203590393066405e-05,
      "step": 21185,
      "training_step_time": 7.802031993865967
    },
    {
      "epoch": 2.0204544067382812e-05,
      "model_forward_time": 1.7335684299468994,
      "step": 21186
    },
    {
      "epoch": 2.0204544067382812e-05,
      "step": 21186,
      "training_step_time": 7.690754652023315
    },
    {
      "epoch": 2.020549774169922e-05,
      "model_forward_time": 1.7220563888549805,
      "step": 21187
    },
    {
      "epoch": 2.020549774169922e-05,
      "step": 21187,
      "training_step_time": 6.795865535736084
    },
    {
      "epoch": 2.0206451416015624e-05,
      "model_forward_time": 1.7177643775939941,
      "step": 21188
    },
    {
      "epoch": 2.0206451416015624e-05,
      "step": 21188,
      "training_step_time": 6.804618835449219
    },
    {
      "epoch": 2.020740509033203e-05,
      "model_forward_time": 1.7264432907104492,
      "step": 21189
    },
    {
      "epoch": 2.020740509033203e-05,
      "step": 21189,
      "training_step_time": 7.617559432983398
    },
    {
      "epoch": 2.020835876464844e-05,
      "grad_norm": 0.1781187653541565,
      "learning_rate": 9.300423986651824e-06,
      "loss": 0.0612,
      "step": 21190
    },
    {
      "epoch": 2.020835876464844e-05,
      "model_forward_time": 1.7297828197479248,
      "step": 21190
    },
    {
      "dynamics_loss_avg": 0.05454638972878456,
      "epoch": 2.020835876464844e-05,
      "step": 21190
    },
    {
      "action_loss_avg": 0.0027904964168556034,
      "epoch": 2.020835876464844e-05,
      "step": 21190
    },
    {
      "epoch": 2.020835876464844e-05,
      "step": 21190,
      "training_step_time": 7.62438702583313
    },
    {
      "epoch": 2.0209312438964843e-05,
      "model_forward_time": 2.24601411819458,
      "step": 21191
    },
    {
      "epoch": 2.0209312438964843e-05,
      "step": 21191,
      "training_step_time": 8.545828342437744
    },
    {
      "epoch": 2.021026611328125e-05,
      "model_forward_time": 1.72487211227417,
      "step": 21192
    },
    {
      "epoch": 2.021026611328125e-05,
      "step": 21192,
      "training_step_time": 9.39687442779541
    },
    {
      "epoch": 2.0211219787597658e-05,
      "model_forward_time": 1.7186355590820312,
      "step": 21193
    },
    {
      "epoch": 2.0211219787597658e-05,
      "step": 21193,
      "training_step_time": 9.93433666229248
    },
    {
      "epoch": 2.021217346191406e-05,
      "model_forward_time": 1.7603726387023926,
      "step": 21194
    },
    {
      "epoch": 2.021217346191406e-05,
      "step": 21194,
      "training_step_time": 8.474323749542236
    },
    {
      "epoch": 2.021312713623047e-05,
      "model_forward_time": 1.7235846519470215,
      "step": 21195
    },
    {
      "epoch": 2.021312713623047e-05,
      "step": 21195,
      "training_step_time": 7.7620251178741455
    },
    {
      "epoch": 2.0214080810546876e-05,
      "model_forward_time": 1.7307507991790771,
      "step": 21196
    },
    {
      "epoch": 2.0214080810546876e-05,
      "step": 21196,
      "training_step_time": 6.835867643356323
    },
    {
      "epoch": 2.021503448486328e-05,
      "model_forward_time": 1.7175383567810059,
      "step": 21197
    },
    {
      "epoch": 2.021503448486328e-05,
      "step": 21197,
      "training_step_time": 6.835850715637207
    },
    {
      "epoch": 2.0215988159179688e-05,
      "model_forward_time": 1.7249608039855957,
      "step": 21198
    },
    {
      "epoch": 2.0215988159179688e-05,
      "step": 21198,
      "training_step_time": 7.639340162277222
    },
    {
      "epoch": 2.0216941833496095e-05,
      "model_forward_time": 1.7199046611785889,
      "step": 21199
    },
    {
      "epoch": 2.0216941833496095e-05,
      "step": 21199,
      "training_step_time": 7.6238908767700195
    },
    {
      "epoch": 2.02178955078125e-05,
      "grad_norm": 0.2123870700597763,
      "learning_rate": 9.299580232082501e-06,
      "loss": 0.0622,
      "step": 21200
    },
    {
      "epoch": 2.02178955078125e-05,
      "model_forward_time": 2.2511863708496094,
      "step": 21200
    },
    {
      "dynamics_loss_avg": 0.06010462325066328,
      "epoch": 2.02178955078125e-05,
      "step": 21200
    },
    {
      "action_loss_avg": 0.002741320780478418,
      "epoch": 2.02178955078125e-05,
      "step": 21200
    },
    {
      "epoch": 2.02178955078125e-05,
      "step": 21200,
      "training_step_time": 8.063870429992676
    },
    {
      "epoch": 2.0218849182128907e-05,
      "model_forward_time": 1.7279331684112549,
      "step": 21201
    },
    {
      "epoch": 2.0218849182128907e-05,
      "step": 21201,
      "training_step_time": 11.557387590408325
    },
    {
      "epoch": 2.0219802856445314e-05,
      "model_forward_time": 1.728065013885498,
      "step": 21202
    },
    {
      "epoch": 2.0219802856445314e-05,
      "step": 21202,
      "training_step_time": 7.876013517379761
    },
    {
      "epoch": 2.0220756530761718e-05,
      "model_forward_time": 1.7316479682922363,
      "step": 21203
    },
    {
      "epoch": 2.0220756530761718e-05,
      "step": 21203,
      "training_step_time": 7.802448511123657
    },
    {
      "epoch": 2.0221710205078126e-05,
      "model_forward_time": 1.7319114208221436,
      "step": 21204
    },
    {
      "epoch": 2.0221710205078126e-05,
      "step": 21204,
      "training_step_time": 7.757662296295166
    },
    {
      "epoch": 2.022266387939453e-05,
      "model_forward_time": 1.727647304534912,
      "step": 21205
    },
    {
      "epoch": 2.022266387939453e-05,
      "step": 21205,
      "training_step_time": 6.856087684631348
    },
    {
      "epoch": 2.0223617553710937e-05,
      "model_forward_time": 1.7203986644744873,
      "step": 21206
    },
    {
      "epoch": 2.0223617553710937e-05,
      "step": 21206,
      "training_step_time": 6.868997573852539
    },
    {
      "epoch": 2.0224571228027344e-05,
      "model_forward_time": 1.7202963829040527,
      "step": 21207
    },
    {
      "epoch": 2.0224571228027344e-05,
      "step": 21207,
      "training_step_time": 7.585195541381836
    },
    {
      "epoch": 2.022552490234375e-05,
      "model_forward_time": 1.7300341129302979,
      "step": 21208
    },
    {
      "epoch": 2.022552490234375e-05,
      "step": 21208,
      "training_step_time": 7.615656614303589
    },
    {
      "epoch": 2.0226478576660156e-05,
      "model_forward_time": 2.2526590824127197,
      "step": 21209
    },
    {
      "epoch": 2.0226478576660156e-05,
      "step": 21209,
      "training_step_time": 7.621769905090332
    },
    {
      "epoch": 2.0227432250976563e-05,
      "grad_norm": 0.19898144900798798,
      "learning_rate": 9.298736007317546e-06,
      "loss": 0.0634,
      "step": 21210
    },
    {
      "epoch": 2.0227432250976563e-05,
      "model_forward_time": 1.7094831466674805,
      "step": 21210
    },
    {
      "dynamics_loss_avg": 0.05396222211420536,
      "epoch": 2.0227432250976563e-05,
      "step": 21210
    },
    {
      "action_loss_avg": 0.0022330796346068382,
      "epoch": 2.0227432250976563e-05,
      "step": 21210
    },
    {
      "epoch": 2.0227432250976563e-05,
      "step": 21210,
      "training_step_time": 9.423383235931396
    },
    {
      "epoch": 2.0228385925292967e-05,
      "model_forward_time": 1.348820447921753,
      "step": 21211
    },
    {
      "epoch": 2.0228385925292967e-05,
      "step": 21211,
      "training_step_time": 7.834358215332031
    },
    {
      "epoch": 2.0229339599609375e-05,
      "model_forward_time": 1.7174344062805176,
      "step": 21212
    },
    {
      "epoch": 2.0229339599609375e-05,
      "step": 21212,
      "training_step_time": 7.70114278793335
    },
    {
      "epoch": 2.0230293273925782e-05,
      "model_forward_time": 1.7149486541748047,
      "step": 21213
    },
    {
      "epoch": 2.0230293273925782e-05,
      "step": 21213,
      "training_step_time": 7.706990003585815
    },
    {
      "epoch": 2.0231246948242186e-05,
      "model_forward_time": 1.71988844871521,
      "step": 21214
    },
    {
      "epoch": 2.0231246948242186e-05,
      "step": 21214,
      "training_step_time": 6.775381803512573
    },
    {
      "epoch": 2.0232200622558594e-05,
      "model_forward_time": 1.681431531906128,
      "step": 21215
    },
    {
      "epoch": 2.0232200622558594e-05,
      "step": 21215,
      "training_step_time": 6.776507377624512
    },
    {
      "epoch": 2.0233154296875e-05,
      "model_forward_time": 1.696427822113037,
      "step": 21216
    },
    {
      "epoch": 2.0233154296875e-05,
      "step": 21216,
      "training_step_time": 7.4262003898620605
    },
    {
      "epoch": 2.0234107971191405e-05,
      "model_forward_time": 1.6961669921875,
      "step": 21217
    },
    {
      "epoch": 2.0234107971191405e-05,
      "step": 21217,
      "training_step_time": 7.608915328979492
    },
    {
      "epoch": 2.0235061645507812e-05,
      "model_forward_time": 2.2376561164855957,
      "step": 21218
    },
    {
      "epoch": 2.0235061645507812e-05,
      "step": 21218,
      "training_step_time": 7.551004886627197
    },
    {
      "epoch": 2.023601531982422e-05,
      "model_forward_time": 1.6999101638793945,
      "step": 21219
    },
    {
      "epoch": 2.023601531982422e-05,
      "step": 21219,
      "training_step_time": 9.875795125961304
    },
    {
      "epoch": 2.0236968994140624e-05,
      "grad_norm": 0.1714766025543213,
      "learning_rate": 9.297891312449287e-06,
      "loss": 0.0652,
      "step": 21220
    },
    {
      "epoch": 2.0236968994140624e-05,
      "model_forward_time": 1.6955609321594238,
      "step": 21220
    },
    {
      "dynamics_loss_avg": 0.05904183015227318,
      "epoch": 2.0236968994140624e-05,
      "step": 21220
    },
    {
      "action_loss_avg": 0.0041177991952281445,
      "epoch": 2.0236968994140624e-05,
      "step": 21220
    },
    {
      "epoch": 2.0236968994140624e-05,
      "step": 21220,
      "training_step_time": 7.752598285675049
    },
    {
      "epoch": 2.023792266845703e-05,
      "model_forward_time": 1.6973390579223633,
      "step": 21221
    },
    {
      "epoch": 2.023792266845703e-05,
      "step": 21221,
      "training_step_time": 7.553428411483765
    },
    {
      "epoch": 2.023887634277344e-05,
      "model_forward_time": 1.7406296730041504,
      "step": 21222
    },
    {
      "epoch": 2.023887634277344e-05,
      "step": 21222,
      "training_step_time": 7.744579076766968
    },
    {
      "epoch": 2.0239830017089843e-05,
      "model_forward_time": 1.6997227668762207,
      "step": 21223
    },
    {
      "epoch": 2.0239830017089843e-05,
      "step": 21223,
      "training_step_time": 7.596790313720703
    },
    {
      "epoch": 2.024078369140625e-05,
      "model_forward_time": 1.6980462074279785,
      "step": 21224
    },
    {
      "epoch": 2.024078369140625e-05,
      "step": 21224,
      "training_step_time": 6.707327842712402
    },
    {
      "epoch": 2.0241737365722658e-05,
      "model_forward_time": 1.6980082988739014,
      "step": 21225
    },
    {
      "epoch": 2.0241737365722658e-05,
      "step": 21225,
      "training_step_time": 7.529767274856567
    },
    {
      "epoch": 2.024269104003906e-05,
      "model_forward_time": 1.6995034217834473,
      "step": 21226
    },
    {
      "epoch": 2.024269104003906e-05,
      "step": 21226,
      "training_step_time": 7.615320205688477
    },
    {
      "epoch": 2.024364471435547e-05,
      "model_forward_time": 2.2493317127227783,
      "step": 21227
    },
    {
      "epoch": 2.024364471435547e-05,
      "step": 21227,
      "training_step_time": 7.604163885116577
    },
    {
      "epoch": 2.0244598388671876e-05,
      "model_forward_time": 1.6997759342193604,
      "step": 21228
    },
    {
      "epoch": 2.0244598388671876e-05,
      "step": 21228,
      "training_step_time": 9.472644329071045
    },
    {
      "epoch": 2.024555206298828e-05,
      "model_forward_time": 1.7036056518554688,
      "step": 21229
    },
    {
      "epoch": 2.024555206298828e-05,
      "step": 21229,
      "training_step_time": 8.691034078598022
    },
    {
      "epoch": 2.0246505737304688e-05,
      "grad_norm": 0.13941854238510132,
      "learning_rate": 9.297046147570095e-06,
      "loss": 0.0658,
      "step": 21230
    },
    {
      "epoch": 2.0246505737304688e-05,
      "model_forward_time": 1.6936712265014648,
      "step": 21230
    },
    {
      "dynamics_loss_avg": 0.05037828423082828,
      "epoch": 2.0246505737304688e-05,
      "step": 21230
    },
    {
      "action_loss_avg": 0.003431786585133523,
      "epoch": 2.0246505737304688e-05,
      "step": 21230
    },
    {
      "epoch": 2.0246505737304688e-05,
      "step": 21230,
      "training_step_time": 8.395207166671753
    },
    {
      "epoch": 2.0247459411621095e-05,
      "model_forward_time": 1.7374775409698486,
      "step": 21231
    },
    {
      "epoch": 2.0247459411621095e-05,
      "step": 21231,
      "training_step_time": 7.724733352661133
    },
    {
      "epoch": 2.02484130859375e-05,
      "model_forward_time": 1.6989095211029053,
      "step": 21232
    },
    {
      "epoch": 2.02484130859375e-05,
      "step": 21232,
      "training_step_time": 7.655872821807861
    },
    {
      "epoch": 2.0249366760253907e-05,
      "model_forward_time": 1.705073595046997,
      "step": 21233
    },
    {
      "epoch": 2.0249366760253907e-05,
      "step": 21233,
      "training_step_time": 6.722476243972778
    },
    {
      "epoch": 2.0250320434570314e-05,
      "model_forward_time": 1.7065174579620361,
      "step": 21234
    },
    {
      "epoch": 2.0250320434570314e-05,
      "step": 21234,
      "training_step_time": 7.525010824203491
    },
    {
      "epoch": 2.0251274108886718e-05,
      "model_forward_time": 1.7010433673858643,
      "step": 21235
    },
    {
      "epoch": 2.0251274108886718e-05,
      "step": 21235,
      "training_step_time": 8.07860517501831
    },
    {
      "epoch": 2.0252227783203126e-05,
      "model_forward_time": 2.253305196762085,
      "step": 21236
    },
    {
      "epoch": 2.0252227783203126e-05,
      "step": 21236,
      "training_step_time": 7.664858818054199
    },
    {
      "epoch": 2.025318145751953e-05,
      "model_forward_time": 1.7086575031280518,
      "step": 21237
    },
    {
      "epoch": 2.025318145751953e-05,
      "step": 21237,
      "training_step_time": 9.178261995315552
    },
    {
      "epoch": 2.0254135131835937e-05,
      "model_forward_time": 1.7007648944854736,
      "step": 21238
    },
    {
      "epoch": 2.0254135131835937e-05,
      "step": 21238,
      "training_step_time": 8.654473781585693
    },
    {
      "epoch": 2.0255088806152344e-05,
      "model_forward_time": 1.7037463188171387,
      "step": 21239
    },
    {
      "epoch": 2.0255088806152344e-05,
      "step": 21239,
      "training_step_time": 9.127678632736206
    },
    {
      "epoch": 2.025604248046875e-05,
      "grad_norm": 0.13927733898162842,
      "learning_rate": 9.296200512772397e-06,
      "loss": 0.0636,
      "step": 21240
    },
    {
      "epoch": 2.025604248046875e-05,
      "model_forward_time": 1.7175142765045166,
      "step": 21240
    },
    {
      "dynamics_loss_avg": 0.054362977482378484,
      "epoch": 2.025604248046875e-05,
      "step": 21240
    },
    {
      "action_loss_avg": 0.002406797424191609,
      "epoch": 2.025604248046875e-05,
      "step": 21240
    },
    {
      "epoch": 2.025604248046875e-05,
      "step": 21240,
      "training_step_time": 8.223591327667236
    },
    {
      "epoch": 2.0256996154785156e-05,
      "model_forward_time": 1.7040379047393799,
      "step": 21241
    },
    {
      "epoch": 2.0256996154785156e-05,
      "step": 21241,
      "training_step_time": 7.60924220085144
    },
    {
      "epoch": 2.0257949829101563e-05,
      "model_forward_time": 1.7028207778930664,
      "step": 21242
    },
    {
      "epoch": 2.0257949829101563e-05,
      "step": 21242,
      "training_step_time": 6.712377548217773
    },
    {
      "epoch": 2.0258903503417967e-05,
      "model_forward_time": 1.697880744934082,
      "step": 21243
    },
    {
      "epoch": 2.0258903503417967e-05,
      "step": 21243,
      "training_step_time": 7.178422927856445
    },
    {
      "epoch": 2.0259857177734375e-05,
      "model_forward_time": 1.704756736755371,
      "step": 21244
    },
    {
      "epoch": 2.0259857177734375e-05,
      "step": 21244,
      "training_step_time": 7.51805567741394
    },
    {
      "epoch": 2.0260810852050782e-05,
      "model_forward_time": 2.252753496170044,
      "step": 21245
    },
    {
      "epoch": 2.0260810852050782e-05,
      "step": 21245,
      "training_step_time": 7.607249736785889
    },
    {
      "epoch": 2.0261764526367186e-05,
      "model_forward_time": 1.6976284980773926,
      "step": 21246
    },
    {
      "epoch": 2.0261764526367186e-05,
      "step": 21246,
      "training_step_time": 7.5662713050842285
    },
    {
      "epoch": 2.0262718200683594e-05,
      "model_forward_time": 1.6966679096221924,
      "step": 21247
    },
    {
      "epoch": 2.0262718200683594e-05,
      "step": 21247,
      "training_step_time": 8.082661867141724
    },
    {
      "epoch": 2.0263671875e-05,
      "model_forward_time": 1.704820156097412,
      "step": 21248
    },
    {
      "epoch": 2.0263671875e-05,
      "step": 21248,
      "training_step_time": 8.728269577026367
    },
    {
      "epoch": 2.0264625549316405e-05,
      "model_forward_time": 1.7036323547363281,
      "step": 21249
    },
    {
      "epoch": 2.0264625549316405e-05,
      "step": 21249,
      "training_step_time": 7.592272520065308
    },
    {
      "epoch": 2.0265579223632812e-05,
      "grad_norm": 0.22845031321048737,
      "learning_rate": 9.295354408148668e-06,
      "loss": 0.065,
      "step": 21250
    },
    {
      "epoch": 2.0265579223632812e-05,
      "model_forward_time": 1.7112805843353271,
      "step": 21250
    },
    {
      "dynamics_loss_avg": 0.05990013852715492,
      "epoch": 2.0265579223632812e-05,
      "step": 21250
    },
    {
      "action_loss_avg": 0.0027429512003436683,
      "epoch": 2.0265579223632812e-05,
      "step": 21250
    },
    {
      "epoch": 2.0265579223632812e-05,
      "step": 21250,
      "training_step_time": 8.277419090270996
    },
    {
      "epoch": 2.026653289794922e-05,
      "model_forward_time": 1.6997249126434326,
      "step": 21251
    },
    {
      "epoch": 2.026653289794922e-05,
      "step": 21251,
      "training_step_time": 7.616773843765259
    },
    {
      "epoch": 2.0267486572265624e-05,
      "model_forward_time": 1.7012219429016113,
      "step": 21252
    },
    {
      "epoch": 2.0267486572265624e-05,
      "step": 21252,
      "training_step_time": 7.229725360870361
    },
    {
      "epoch": 2.026844024658203e-05,
      "model_forward_time": 1.707564353942871,
      "step": 21253
    },
    {
      "epoch": 2.026844024658203e-05,
      "step": 21253,
      "training_step_time": 7.573262691497803
    },
    {
      "epoch": 2.026939392089844e-05,
      "model_forward_time": 2.232069969177246,
      "step": 21254
    },
    {
      "epoch": 2.026939392089844e-05,
      "step": 21254,
      "training_step_time": 7.601043701171875
    },
    {
      "epoch": 2.0270347595214843e-05,
      "model_forward_time": 1.7029662132263184,
      "step": 21255
    },
    {
      "epoch": 2.0270347595214843e-05,
      "step": 21255,
      "training_step_time": 8.524741888046265
    },
    {
      "epoch": 2.027130126953125e-05,
      "model_forward_time": 1.7023751735687256,
      "step": 21256
    },
    {
      "epoch": 2.027130126953125e-05,
      "step": 21256,
      "training_step_time": 7.56663966178894
    },
    {
      "epoch": 2.0272254943847658e-05,
      "model_forward_time": 1.7081775665283203,
      "step": 21257
    },
    {
      "epoch": 2.0272254943847658e-05,
      "step": 21257,
      "training_step_time": 8.31135606765747
    },
    {
      "epoch": 2.027320861816406e-05,
      "model_forward_time": 1.7074832916259766,
      "step": 21258
    },
    {
      "epoch": 2.027320861816406e-05,
      "step": 21258,
      "training_step_time": 7.515596866607666
    },
    {
      "epoch": 2.027416229248047e-05,
      "model_forward_time": 1.7675864696502686,
      "step": 21259
    },
    {
      "epoch": 2.027416229248047e-05,
      "step": 21259,
      "training_step_time": 7.626166820526123
    },
    {
      "epoch": 2.0275115966796876e-05,
      "grad_norm": 0.13507609069347382,
      "learning_rate": 9.294507833791442e-06,
      "loss": 0.0631,
      "step": 21260
    },
    {
      "epoch": 2.0275115966796876e-05,
      "model_forward_time": 1.6968638896942139,
      "step": 21260
    },
    {
      "dynamics_loss_avg": 0.06898710485547781,
      "epoch": 2.0275115966796876e-05,
      "step": 21260
    },
    {
      "action_loss_avg": 0.002532388805411756,
      "epoch": 2.0275115966796876e-05,
      "step": 21260
    },
    {
      "epoch": 2.0275115966796876e-05,
      "step": 21260,
      "training_step_time": 7.582510709762573
    },
    {
      "epoch": 2.027606964111328e-05,
      "model_forward_time": 1.7210140228271484,
      "step": 21261
    },
    {
      "epoch": 2.027606964111328e-05,
      "step": 21261,
      "training_step_time": 6.716442823410034
    },
    {
      "epoch": 2.0277023315429688e-05,
      "model_forward_time": 1.7180109024047852,
      "step": 21262
    },
    {
      "epoch": 2.0277023315429688e-05,
      "step": 21262,
      "training_step_time": 7.406677484512329
    },
    {
      "epoch": 2.0277976989746095e-05,
      "model_forward_time": 1.7204337120056152,
      "step": 21263
    },
    {
      "epoch": 2.0277976989746095e-05,
      "step": 21263,
      "training_step_time": 7.595020055770874
    },
    {
      "epoch": 2.02789306640625e-05,
      "model_forward_time": 2.2695910930633545,
      "step": 21264
    },
    {
      "epoch": 2.02789306640625e-05,
      "step": 21264,
      "training_step_time": 7.558800935745239
    },
    {
      "epoch": 2.0279884338378907e-05,
      "model_forward_time": 1.711979627609253,
      "step": 21265
    },
    {
      "epoch": 2.0279884338378907e-05,
      "step": 21265,
      "training_step_time": 8.489181280136108
    },
    {
      "epoch": 2.028083801269531e-05,
      "model_forward_time": 1.7223389148712158,
      "step": 21266
    },
    {
      "epoch": 2.028083801269531e-05,
      "step": 21266,
      "training_step_time": 10.312723159790039
    },
    {
      "epoch": 2.0281791687011718e-05,
      "model_forward_time": 1.7155685424804688,
      "step": 21267
    },
    {
      "epoch": 2.0281791687011718e-05,
      "step": 21267,
      "training_step_time": 9.484765768051147
    },
    {
      "epoch": 2.0282745361328126e-05,
      "model_forward_time": 1.7669270038604736,
      "step": 21268
    },
    {
      "epoch": 2.0282745361328126e-05,
      "step": 21268,
      "training_step_time": 7.6871113777160645
    },
    {
      "epoch": 2.028369903564453e-05,
      "model_forward_time": 1.7030062675476074,
      "step": 21269
    },
    {
      "epoch": 2.028369903564453e-05,
      "step": 21269,
      "training_step_time": 8.345571994781494
    },
    {
      "epoch": 2.0284652709960937e-05,
      "grad_norm": 0.18586136400699615,
      "learning_rate": 9.293660789793296e-06,
      "loss": 0.0639,
      "step": 21270
    },
    {
      "epoch": 2.0284652709960937e-05,
      "model_forward_time": 1.6984353065490723,
      "step": 21270
    },
    {
      "dynamics_loss_avg": 0.05212419293820858,
      "epoch": 2.0284652709960937e-05,
      "step": 21270
    },
    {
      "action_loss_avg": 0.0026781498512718827,
      "epoch": 2.0284652709960937e-05,
      "step": 21270
    },
    {
      "epoch": 2.0284652709960937e-05,
      "step": 21270,
      "training_step_time": 6.702965021133423
    },
    {
      "epoch": 2.0285606384277345e-05,
      "model_forward_time": 1.704793930053711,
      "step": 21271
    },
    {
      "epoch": 2.0285606384277345e-05,
      "step": 21271,
      "training_step_time": 8.257640838623047
    },
    {
      "epoch": 2.028656005859375e-05,
      "model_forward_time": 1.7003211975097656,
      "step": 21272
    },
    {
      "epoch": 2.028656005859375e-05,
      "step": 21272,
      "training_step_time": 7.667713403701782
    },
    {
      "epoch": 2.0287513732910156e-05,
      "model_forward_time": 2.259418249130249,
      "step": 21273
    },
    {
      "epoch": 2.0287513732910156e-05,
      "step": 21273,
      "training_step_time": 7.646195411682129
    },
    {
      "epoch": 2.0288467407226563e-05,
      "model_forward_time": 0.9883174896240234,
      "step": 21274
    },
    {
      "epoch": 2.0288467407226563e-05,
      "step": 21274,
      "training_step_time": 8.712282657623291
    },
    {
      "epoch": 2.0289421081542967e-05,
      "model_forward_time": 1.710202932357788,
      "step": 21275
    },
    {
      "epoch": 2.0289421081542967e-05,
      "step": 21275,
      "training_step_time": 9.416524648666382
    },
    {
      "epoch": 2.0290374755859375e-05,
      "model_forward_time": 1.7025699615478516,
      "step": 21276
    },
    {
      "epoch": 2.0290374755859375e-05,
      "step": 21276,
      "training_step_time": 8.320260524749756
    },
    {
      "epoch": 2.0291328430175782e-05,
      "model_forward_time": 1.7603788375854492,
      "step": 21277
    },
    {
      "epoch": 2.0291328430175782e-05,
      "step": 21277,
      "training_step_time": 8.373636722564697
    },
    {
      "epoch": 2.0292282104492186e-05,
      "model_forward_time": 1.7050979137420654,
      "step": 21278
    },
    {
      "epoch": 2.0292282104492186e-05,
      "step": 21278,
      "training_step_time": 7.49332594871521
    },
    {
      "epoch": 2.0293235778808594e-05,
      "model_forward_time": 1.7066762447357178,
      "step": 21279
    },
    {
      "epoch": 2.0293235778808594e-05,
      "step": 21279,
      "training_step_time": 7.266557693481445
    },
    {
      "epoch": 2.0294189453125e-05,
      "grad_norm": 0.16173315048217773,
      "learning_rate": 9.292813276246859e-06,
      "loss": 0.0629,
      "step": 21280
    },
    {
      "epoch": 2.0294189453125e-05,
      "model_forward_time": 1.6992883682250977,
      "step": 21280
    },
    {
      "dynamics_loss_avg": 0.07568233422935008,
      "epoch": 2.0294189453125e-05,
      "step": 21280
    },
    {
      "action_loss_avg": 0.0036730894935317338,
      "epoch": 2.0294189453125e-05,
      "step": 21280
    },
    {
      "epoch": 2.0294189453125e-05,
      "step": 21280,
      "training_step_time": 7.233016014099121
    },
    {
      "epoch": 2.0295143127441405e-05,
      "model_forward_time": 1.706284999847412,
      "step": 21281
    },
    {
      "epoch": 2.0295143127441405e-05,
      "step": 21281,
      "training_step_time": 7.64177131652832
    },
    {
      "epoch": 2.0296096801757813e-05,
      "model_forward_time": 2.2681515216827393,
      "step": 21282
    },
    {
      "epoch": 2.0296096801757813e-05,
      "step": 21282,
      "training_step_time": 7.593532562255859
    },
    {
      "epoch": 2.029705047607422e-05,
      "model_forward_time": 1.7060072422027588,
      "step": 21283
    },
    {
      "epoch": 2.029705047607422e-05,
      "step": 21283,
      "training_step_time": 7.637125492095947
    },
    {
      "epoch": 2.0298004150390624e-05,
      "model_forward_time": 1.7327096462249756,
      "step": 21284
    },
    {
      "epoch": 2.0298004150390624e-05,
      "step": 21284,
      "training_step_time": 7.602604150772095
    },
    {
      "epoch": 2.029895782470703e-05,
      "model_forward_time": 1.751596212387085,
      "step": 21285
    },
    {
      "epoch": 2.029895782470703e-05,
      "step": 21285,
      "training_step_time": 8.300227880477905
    },
    {
      "epoch": 2.029991149902344e-05,
      "model_forward_time": 1.746196985244751,
      "step": 21286
    },
    {
      "epoch": 2.029991149902344e-05,
      "step": 21286,
      "training_step_time": 7.752364158630371
    },
    {
      "epoch": 2.0300865173339843e-05,
      "model_forward_time": 1.7531378269195557,
      "step": 21287
    },
    {
      "epoch": 2.0300865173339843e-05,
      "step": 21287,
      "training_step_time": 7.5963990688323975
    },
    {
      "epoch": 2.030181884765625e-05,
      "model_forward_time": 1.750412940979004,
      "step": 21288
    },
    {
      "epoch": 2.030181884765625e-05,
      "step": 21288,
      "training_step_time": 8.008997917175293
    },
    {
      "epoch": 2.0302772521972658e-05,
      "model_forward_time": 1.7536065578460693,
      "step": 21289
    },
    {
      "epoch": 2.0302772521972658e-05,
      "step": 21289,
      "training_step_time": 7.2581446170806885
    },
    {
      "epoch": 2.030372619628906e-05,
      "grad_norm": 0.1366848349571228,
      "learning_rate": 9.291965293244817e-06,
      "loss": 0.0643,
      "step": 21290
    },
    {
      "epoch": 2.030372619628906e-05,
      "model_forward_time": 1.7457985877990723,
      "step": 21290
    },
    {
      "dynamics_loss_avg": 0.06568966349586844,
      "epoch": 2.030372619628906e-05,
      "step": 21290
    },
    {
      "action_loss_avg": 0.003103766101412475,
      "epoch": 2.030372619628906e-05,
      "step": 21290
    },
    {
      "epoch": 2.030372619628906e-05,
      "step": 21290,
      "training_step_time": 8.730477809906006
    },
    {
      "epoch": 2.030467987060547e-05,
      "model_forward_time": 1.7606964111328125,
      "step": 21291
    },
    {
      "epoch": 2.030467987060547e-05,
      "step": 21291,
      "training_step_time": 9.60788345336914
    },
    {
      "epoch": 2.0305633544921877e-05,
      "model_forward_time": 1.7766506671905518,
      "step": 21292
    },
    {
      "epoch": 2.0305633544921877e-05,
      "step": 21292,
      "training_step_time": 8.070928812026978
    },
    {
      "epoch": 2.030658721923828e-05,
      "model_forward_time": 1.7654426097869873,
      "step": 21293
    },
    {
      "epoch": 2.030658721923828e-05,
      "step": 21293,
      "training_step_time": 9.654574155807495
    },
    {
      "epoch": 2.0307540893554688e-05,
      "model_forward_time": 1.7600500583648682,
      "step": 21294
    },
    {
      "epoch": 2.0307540893554688e-05,
      "step": 21294,
      "training_step_time": 8.464879512786865
    },
    {
      "epoch": 2.0308494567871095e-05,
      "model_forward_time": 1.7341501712799072,
      "step": 21295
    },
    {
      "epoch": 2.0308494567871095e-05,
      "step": 21295,
      "training_step_time": 8.299982786178589
    },
    {
      "epoch": 2.03094482421875e-05,
      "model_forward_time": 1.7590751647949219,
      "step": 21296
    },
    {
      "epoch": 2.03094482421875e-05,
      "step": 21296,
      "training_step_time": 7.594390630722046
    },
    {
      "epoch": 2.0310401916503907e-05,
      "model_forward_time": 1.7417194843292236,
      "step": 21297
    },
    {
      "epoch": 2.0310401916503907e-05,
      "step": 21297,
      "training_step_time": 7.524389028549194
    },
    {
      "epoch": 2.031135559082031e-05,
      "model_forward_time": 1.7569479942321777,
      "step": 21298
    },
    {
      "epoch": 2.031135559082031e-05,
      "step": 21298,
      "training_step_time": 7.290328741073608
    },
    {
      "epoch": 2.0312309265136718e-05,
      "model_forward_time": 1.7559807300567627,
      "step": 21299
    },
    {
      "epoch": 2.0312309265136718e-05,
      "step": 21299,
      "training_step_time": 7.686052083969116
    },
    {
      "epoch": 2.0313262939453126e-05,
      "grad_norm": 0.2242438644170761,
      "learning_rate": 9.291116840879904e-06,
      "loss": 0.0618,
      "step": 21300
    },
    {
      "epoch": 2.0313262939453126e-05,
      "model_forward_time": 1.7425336837768555,
      "step": 21300
    },
    {
      "dynamics_loss_avg": 0.05127043463289738,
      "epoch": 2.0313262939453126e-05,
      "step": 21300
    },
    {
      "action_loss_avg": 0.005028166284319013,
      "epoch": 2.0313262939453126e-05,
      "step": 21300
    },
    {
      "epoch": 2.0313262939453126e-05,
      "step": 21300,
      "training_step_time": 7.687425136566162
    },
    {
      "epoch": 2.031421661376953e-05,
      "model_forward_time": 2.901013135910034,
      "step": 21301
    },
    {
      "epoch": 2.031421661376953e-05,
      "step": 21301,
      "training_step_time": 8.586911916732788
    },
    {
      "epoch": 2.0315170288085937e-05,
      "model_forward_time": 1.7517871856689453,
      "step": 21302
    },
    {
      "epoch": 2.0315170288085937e-05,
      "step": 21302,
      "training_step_time": 9.624722480773926
    },
    {
      "epoch": 2.0316123962402345e-05,
      "model_forward_time": 1.752213954925537,
      "step": 21303
    },
    {
      "epoch": 2.0316123962402345e-05,
      "step": 21303,
      "training_step_time": 7.968438625335693
    },
    {
      "epoch": 2.031707763671875e-05,
      "model_forward_time": 1.7359068393707275,
      "step": 21304
    },
    {
      "epoch": 2.031707763671875e-05,
      "step": 21304,
      "training_step_time": 10.022542715072632
    },
    {
      "epoch": 2.0318031311035156e-05,
      "model_forward_time": 1.7666676044464111,
      "step": 21305
    },
    {
      "epoch": 2.0318031311035156e-05,
      "step": 21305,
      "training_step_time": 7.64872145652771
    },
    {
      "epoch": 2.0318984985351563e-05,
      "model_forward_time": 1.746368646621704,
      "step": 21306
    },
    {
      "epoch": 2.0318984985351563e-05,
      "step": 21306,
      "training_step_time": 7.572958469390869
    },
    {
      "epoch": 2.0319938659667967e-05,
      "model_forward_time": 1.7556133270263672,
      "step": 21307
    },
    {
      "epoch": 2.0319938659667967e-05,
      "step": 21307,
      "training_step_time": 7.314172029495239
    },
    {
      "epoch": 2.0320892333984375e-05,
      "model_forward_time": 1.7483673095703125,
      "step": 21308
    },
    {
      "epoch": 2.0320892333984375e-05,
      "step": 21308,
      "training_step_time": 7.686123609542847
    },
    {
      "epoch": 2.0321846008300782e-05,
      "model_forward_time": 1.7516989707946777,
      "step": 21309
    },
    {
      "epoch": 2.0321846008300782e-05,
      "step": 21309,
      "training_step_time": 7.636786937713623
    },
    {
      "epoch": 2.0322799682617186e-05,
      "grad_norm": 0.22883178293704987,
      "learning_rate": 9.290267919244904e-06,
      "loss": 0.063,
      "step": 21310
    },
    {
      "epoch": 2.0322799682617186e-05,
      "model_forward_time": 2.9536781311035156,
      "step": 21310
    },
    {
      "dynamics_loss_avg": 0.054988445900380614,
      "epoch": 2.0322799682617186e-05,
      "step": 21310
    },
    {
      "action_loss_avg": 0.002688872240833007,
      "epoch": 2.0322799682617186e-05,
      "step": 21310
    },
    {
      "epoch": 2.0322799682617186e-05,
      "step": 21310,
      "training_step_time": 9.341655731201172
    },
    {
      "epoch": 2.0323753356933594e-05,
      "model_forward_time": 1.853165864944458,
      "step": 21311
    },
    {
      "epoch": 2.0323753356933594e-05,
      "step": 21311,
      "training_step_time": 8.226316452026367
    },
    {
      "epoch": 2.032470703125e-05,
      "model_forward_time": 1.0507349967956543,
      "step": 21312
    },
    {
      "epoch": 2.032470703125e-05,
      "step": 21312,
      "training_step_time": 8.85604476928711
    },
    {
      "epoch": 2.0325660705566405e-05,
      "model_forward_time": 1.74672269821167,
      "step": 21313
    },
    {
      "epoch": 2.0325660705566405e-05,
      "step": 21313,
      "training_step_time": 7.777669668197632
    },
    {
      "epoch": 2.0326614379882813e-05,
      "model_forward_time": 1.758655071258545,
      "step": 21314
    },
    {
      "epoch": 2.0326614379882813e-05,
      "step": 21314,
      "training_step_time": 8.289481163024902
    },
    {
      "epoch": 2.032756805419922e-05,
      "model_forward_time": 1.755018949508667,
      "step": 21315
    },
    {
      "epoch": 2.032756805419922e-05,
      "step": 21315,
      "training_step_time": 8.305511951446533
    },
    {
      "epoch": 2.0328521728515624e-05,
      "model_forward_time": 1.7645411491394043,
      "step": 21316
    },
    {
      "epoch": 2.0328521728515624e-05,
      "step": 21316,
      "training_step_time": 7.322391510009766
    },
    {
      "epoch": 2.032947540283203e-05,
      "model_forward_time": 1.7446317672729492,
      "step": 21317
    },
    {
      "epoch": 2.032947540283203e-05,
      "step": 21317,
      "training_step_time": 7.670205116271973
    },
    {
      "epoch": 2.033042907714844e-05,
      "model_forward_time": 1.7576913833618164,
      "step": 21318
    },
    {
      "epoch": 2.033042907714844e-05,
      "step": 21318,
      "training_step_time": 9.125454187393188
    },
    {
      "epoch": 2.0331382751464843e-05,
      "model_forward_time": 2.879209041595459,
      "step": 21319
    },
    {
      "epoch": 2.0331382751464843e-05,
      "step": 21319,
      "training_step_time": 7.8219592571258545
    },
    {
      "epoch": 2.033233642578125e-05,
      "grad_norm": 0.20107154548168182,
      "learning_rate": 9.289418528432655e-06,
      "loss": 0.0631,
      "step": 21320
    },
    {
      "epoch": 2.033233642578125e-05,
      "model_forward_time": 1.8186395168304443,
      "step": 21320
    },
    {
      "dynamics_loss_avg": 0.06000785641372204,
      "epoch": 2.033233642578125e-05,
      "step": 21320
    },
    {
      "action_loss_avg": 0.0040679262339835985,
      "epoch": 2.033233642578125e-05,
      "step": 21320
    },
    {
      "epoch": 2.033233642578125e-05,
      "step": 21320,
      "training_step_time": 9.11489224433899
    },
    {
      "epoch": 2.0333290100097658e-05,
      "model_forward_time": 1.755079746246338,
      "step": 21321
    },
    {
      "epoch": 2.0333290100097658e-05,
      "step": 21321,
      "training_step_time": 10.203739404678345
    },
    {
      "epoch": 2.0334243774414062e-05,
      "model_forward_time": 1.7454111576080322,
      "step": 21322
    },
    {
      "epoch": 2.0334243774414062e-05,
      "step": 21322,
      "training_step_time": 7.760979890823364
    },
    {
      "epoch": 2.033519744873047e-05,
      "model_forward_time": 1.7583873271942139,
      "step": 21323
    },
    {
      "epoch": 2.033519744873047e-05,
      "step": 21323,
      "training_step_time": 7.681582927703857
    },
    {
      "epoch": 2.0336151123046877e-05,
      "model_forward_time": 1.752962350845337,
      "step": 21324
    },
    {
      "epoch": 2.0336151123046877e-05,
      "step": 21324,
      "training_step_time": 7.6519834995269775
    },
    {
      "epoch": 2.033710479736328e-05,
      "model_forward_time": 1.754465103149414,
      "step": 21325
    },
    {
      "epoch": 2.033710479736328e-05,
      "step": 21325,
      "training_step_time": 7.551992893218994
    },
    {
      "epoch": 2.0338058471679688e-05,
      "model_forward_time": 1.755990743637085,
      "step": 21326
    },
    {
      "epoch": 2.0338058471679688e-05,
      "step": 21326,
      "training_step_time": 7.738994121551514
    },
    {
      "epoch": 2.0339012145996095e-05,
      "model_forward_time": 1.763106346130371,
      "step": 21327
    },
    {
      "epoch": 2.0339012145996095e-05,
      "step": 21327,
      "training_step_time": 7.672667503356934
    },
    {
      "epoch": 2.03399658203125e-05,
      "model_forward_time": 2.8280231952667236,
      "step": 21328
    },
    {
      "epoch": 2.03399658203125e-05,
      "step": 21328,
      "training_step_time": 7.776425123214722
    },
    {
      "epoch": 2.0340919494628907e-05,
      "model_forward_time": 1.7997708320617676,
      "step": 21329
    },
    {
      "epoch": 2.0340919494628907e-05,
      "step": 21329,
      "training_step_time": 8.635112524032593
    },
    {
      "epoch": 2.034187316894531e-05,
      "grad_norm": 0.19291868805885315,
      "learning_rate": 9.288568668536046e-06,
      "loss": 0.0626,
      "step": 21330
    },
    {
      "epoch": 2.034187316894531e-05,
      "model_forward_time": 1.7679057121276855,
      "step": 21330
    },
    {
      "dynamics_loss_avg": 0.06537888515740634,
      "epoch": 2.034187316894531e-05,
      "step": 21330
    },
    {
      "action_loss_avg": 0.002754887111950666,
      "epoch": 2.034187316894531e-05,
      "step": 21330
    },
    {
      "epoch": 2.034187316894531e-05,
      "step": 21330,
      "training_step_time": 8.501718282699585
    },
    {
      "epoch": 2.0342826843261718e-05,
      "model_forward_time": 1.75480318069458,
      "step": 21331
    },
    {
      "epoch": 2.0342826843261718e-05,
      "step": 21331,
      "training_step_time": 8.656180381774902
    },
    {
      "epoch": 2.0343780517578126e-05,
      "model_forward_time": 1.7563631534576416,
      "step": 21332
    },
    {
      "epoch": 2.0343780517578126e-05,
      "step": 21332,
      "training_step_time": 7.674721002578735
    },
    {
      "epoch": 2.034473419189453e-05,
      "model_forward_time": 1.7543904781341553,
      "step": 21333
    },
    {
      "epoch": 2.034473419189453e-05,
      "step": 21333,
      "training_step_time": 7.53235387802124
    },
    {
      "epoch": 2.0345687866210937e-05,
      "model_forward_time": 1.75126314163208,
      "step": 21334
    },
    {
      "epoch": 2.0345687866210937e-05,
      "step": 21334,
      "training_step_time": 7.514837026596069
    },
    {
      "epoch": 2.0346641540527345e-05,
      "model_forward_time": 1.755645751953125,
      "step": 21335
    },
    {
      "epoch": 2.0346641540527345e-05,
      "step": 21335,
      "training_step_time": 7.626311779022217
    },
    {
      "epoch": 2.034759521484375e-05,
      "model_forward_time": 1.760392427444458,
      "step": 21336
    },
    {
      "epoch": 2.034759521484375e-05,
      "step": 21336,
      "training_step_time": 7.757492303848267
    },
    {
      "epoch": 2.0348548889160156e-05,
      "model_forward_time": 1.7669999599456787,
      "step": 21337
    },
    {
      "epoch": 2.0348548889160156e-05,
      "step": 21337,
      "training_step_time": 8.800455570220947
    },
    {
      "epoch": 2.0349502563476563e-05,
      "model_forward_time": 2.8734824657440186,
      "step": 21338
    },
    {
      "epoch": 2.0349502563476563e-05,
      "step": 21338,
      "training_step_time": 9.419915437698364
    },
    {
      "epoch": 2.0350456237792967e-05,
      "model_forward_time": 1.7476115226745605,
      "step": 21339
    },
    {
      "epoch": 2.0350456237792967e-05,
      "step": 21339,
      "training_step_time": 8.951766729354858
    },
    {
      "epoch": 2.0351409912109375e-05,
      "grad_norm": 0.13356059789657593,
      "learning_rate": 9.287718339648013e-06,
      "loss": 0.0659,
      "step": 21340
    },
    {
      "epoch": 2.0351409912109375e-05,
      "model_forward_time": 1.7516582012176514,
      "step": 21340
    },
    {
      "dynamics_loss_avg": 0.06211394295096397,
      "epoch": 2.0351409912109375e-05,
      "step": 21340
    },
    {
      "action_loss_avg": 0.004163723648525774,
      "epoch": 2.0351409912109375e-05,
      "step": 21340
    },
    {
      "epoch": 2.0351409912109375e-05,
      "step": 21340,
      "training_step_time": 8.896311521530151
    },
    {
      "epoch": 2.0352363586425782e-05,
      "model_forward_time": 1.7488503456115723,
      "step": 21341
    },
    {
      "epoch": 2.0352363586425782e-05,
      "step": 21341,
      "training_step_time": 8.843252658843994
    },
    {
      "epoch": 2.0353317260742186e-05,
      "model_forward_time": 1.746483564376831,
      "step": 21342
    },
    {
      "epoch": 2.0353317260742186e-05,
      "step": 21342,
      "training_step_time": 8.097418069839478
    },
    {
      "epoch": 2.0354270935058594e-05,
      "model_forward_time": 1.7534244060516357,
      "step": 21343
    },
    {
      "epoch": 2.0354270935058594e-05,
      "step": 21343,
      "training_step_time": 7.5960822105407715
    },
    {
      "epoch": 2.0355224609375e-05,
      "model_forward_time": 1.752504825592041,
      "step": 21344
    },
    {
      "epoch": 2.0355224609375e-05,
      "step": 21344,
      "training_step_time": 7.69709849357605
    },
    {
      "epoch": 2.0356178283691405e-05,
      "model_forward_time": 1.7481889724731445,
      "step": 21345
    },
    {
      "epoch": 2.0356178283691405e-05,
      "step": 21345,
      "training_step_time": 7.683248281478882
    },
    {
      "epoch": 2.0357131958007813e-05,
      "model_forward_time": 1.7515342235565186,
      "step": 21346
    },
    {
      "epoch": 2.0357131958007813e-05,
      "step": 21346,
      "training_step_time": 7.619159936904907
    },
    {
      "epoch": 2.035808563232422e-05,
      "model_forward_time": 2.9090306758880615,
      "step": 21347
    },
    {
      "epoch": 2.035808563232422e-05,
      "step": 21347,
      "training_step_time": 8.484655141830444
    },
    {
      "epoch": 2.0359039306640624e-05,
      "model_forward_time": 1.8549766540527344,
      "step": 21348
    },
    {
      "epoch": 2.0359039306640624e-05,
      "step": 21348,
      "training_step_time": 7.704611539840698
    },
    {
      "epoch": 2.035999298095703e-05,
      "model_forward_time": 1.381937026977539,
      "step": 21349
    },
    {
      "epoch": 2.035999298095703e-05,
      "step": 21349,
      "training_step_time": 7.950929403305054
    },
    {
      "epoch": 2.036094665527344e-05,
      "grad_norm": 0.1654588133096695,
      "learning_rate": 9.286867541861551e-06,
      "loss": 0.0615,
      "step": 21350
    },
    {
      "epoch": 2.036094665527344e-05,
      "model_forward_time": 1.7380621433258057,
      "step": 21350
    },
    {
      "dynamics_loss_avg": 0.06786300502717495,
      "epoch": 2.036094665527344e-05,
      "step": 21350
    },
    {
      "action_loss_avg": 0.003749787650303915,
      "epoch": 2.036094665527344e-05,
      "step": 21350
    },
    {
      "epoch": 2.036094665527344e-05,
      "step": 21350,
      "training_step_time": 9.168054819107056
    },
    {
      "epoch": 2.0361900329589843e-05,
      "model_forward_time": 1.7462859153747559,
      "step": 21351
    },
    {
      "epoch": 2.0361900329589843e-05,
      "step": 21351,
      "training_step_time": 7.598911762237549
    },
    {
      "epoch": 2.036285400390625e-05,
      "model_forward_time": 1.7405869960784912,
      "step": 21352
    },
    {
      "epoch": 2.036285400390625e-05,
      "step": 21352,
      "training_step_time": 7.601412534713745
    },
    {
      "epoch": 2.0363807678222658e-05,
      "model_forward_time": 1.7504973411560059,
      "step": 21353
    },
    {
      "epoch": 2.0363807678222658e-05,
      "step": 21353,
      "training_step_time": 8.032024145126343
    },
    {
      "epoch": 2.0364761352539062e-05,
      "model_forward_time": 1.745818853378296,
      "step": 21354
    },
    {
      "epoch": 2.0364761352539062e-05,
      "step": 21354,
      "training_step_time": 7.6846232414245605
    },
    {
      "epoch": 2.036571502685547e-05,
      "model_forward_time": 1.749155044555664,
      "step": 21355
    },
    {
      "epoch": 2.036571502685547e-05,
      "step": 21355,
      "training_step_time": 7.578886985778809
    },
    {
      "epoch": 2.0366668701171877e-05,
      "model_forward_time": 2.884492874145508,
      "step": 21356
    },
    {
      "epoch": 2.0366668701171877e-05,
      "step": 21356,
      "training_step_time": 7.888009071350098
    },
    {
      "epoch": 2.036762237548828e-05,
      "model_forward_time": 1.827465295791626,
      "step": 21357
    },
    {
      "epoch": 2.036762237548828e-05,
      "step": 21357,
      "training_step_time": 9.56432056427002
    },
    {
      "epoch": 2.0368576049804688e-05,
      "model_forward_time": 1.7491035461425781,
      "step": 21358
    },
    {
      "epoch": 2.0368576049804688e-05,
      "step": 21358,
      "training_step_time": 7.94058632850647
    },
    {
      "epoch": 2.0369529724121095e-05,
      "model_forward_time": 1.744272232055664,
      "step": 21359
    },
    {
      "epoch": 2.0369529724121095e-05,
      "step": 21359,
      "training_step_time": 7.708823919296265
    },
    {
      "epoch": 2.03704833984375e-05,
      "grad_norm": 0.1668122559785843,
      "learning_rate": 9.2860162752697e-06,
      "loss": 0.067,
      "step": 21360
    },
    {
      "epoch": 2.03704833984375e-05,
      "model_forward_time": 1.7437443733215332,
      "step": 21360
    },
    {
      "dynamics_loss_avg": 0.06652859933674335,
      "epoch": 2.03704833984375e-05,
      "step": 21360
    },
    {
      "action_loss_avg": 0.0028470143442973495,
      "epoch": 2.03704833984375e-05,
      "step": 21360
    },
    {
      "epoch": 2.03704833984375e-05,
      "step": 21360,
      "training_step_time": 7.712871789932251
    },
    {
      "epoch": 2.0371437072753907e-05,
      "model_forward_time": 1.7244138717651367,
      "step": 21361
    },
    {
      "epoch": 2.0371437072753907e-05,
      "step": 21361,
      "training_step_time": 7.543256759643555
    },
    {
      "epoch": 2.037239074707031e-05,
      "model_forward_time": 1.7515032291412354,
      "step": 21362
    },
    {
      "epoch": 2.037239074707031e-05,
      "step": 21362,
      "training_step_time": 7.625729322433472
    },
    {
      "epoch": 2.037334442138672e-05,
      "model_forward_time": 1.7469325065612793,
      "step": 21363
    },
    {
      "epoch": 2.037334442138672e-05,
      "step": 21363,
      "training_step_time": 7.759798049926758
    },
    {
      "epoch": 2.0374298095703126e-05,
      "model_forward_time": 1.743760585784912,
      "step": 21364
    },
    {
      "epoch": 2.0374298095703126e-05,
      "step": 21364,
      "training_step_time": 7.671022176742554
    },
    {
      "epoch": 2.037525177001953e-05,
      "model_forward_time": 2.8674561977386475,
      "step": 21365
    },
    {
      "epoch": 2.037525177001953e-05,
      "step": 21365,
      "training_step_time": 7.943089246749878
    },
    {
      "epoch": 2.0376205444335937e-05,
      "model_forward_time": 1.820838451385498,
      "step": 21366
    },
    {
      "epoch": 2.0376205444335937e-05,
      "step": 21366,
      "training_step_time": 8.4944908618927
    },
    {
      "epoch": 2.0377159118652345e-05,
      "model_forward_time": 1.7418932914733887,
      "step": 21367
    },
    {
      "epoch": 2.0377159118652345e-05,
      "step": 21367,
      "training_step_time": 9.455662250518799
    },
    {
      "epoch": 2.037811279296875e-05,
      "model_forward_time": 1.7406291961669922,
      "step": 21368
    },
    {
      "epoch": 2.037811279296875e-05,
      "step": 21368,
      "training_step_time": 7.791769742965698
    },
    {
      "epoch": 2.0379066467285156e-05,
      "model_forward_time": 1.7418413162231445,
      "step": 21369
    },
    {
      "epoch": 2.0379066467285156e-05,
      "step": 21369,
      "training_step_time": 8.425830841064453
    },
    {
      "epoch": 2.0380020141601563e-05,
      "grad_norm": 0.16013985872268677,
      "learning_rate": 9.28516453996555e-06,
      "loss": 0.0671,
      "step": 21370
    },
    {
      "epoch": 2.0380020141601563e-05,
      "model_forward_time": 1.7321193218231201,
      "step": 21370
    },
    {
      "dynamics_loss_avg": 0.06430839709937572,
      "epoch": 2.0380020141601563e-05,
      "step": 21370
    },
    {
      "action_loss_avg": 0.002384701429400593,
      "epoch": 2.0380020141601563e-05,
      "step": 21370
    },
    {
      "epoch": 2.0380020141601563e-05,
      "step": 21370,
      "training_step_time": 8.326624393463135
    },
    {
      "epoch": 2.0380973815917967e-05,
      "model_forward_time": 1.7484970092773438,
      "step": 21371
    },
    {
      "epoch": 2.0380973815917967e-05,
      "step": 21371,
      "training_step_time": 7.585368394851685
    },
    {
      "epoch": 2.0381927490234375e-05,
      "model_forward_time": 1.7455976009368896,
      "step": 21372
    },
    {
      "epoch": 2.0381927490234375e-05,
      "step": 21372,
      "training_step_time": 8.72285771369934
    },
    {
      "epoch": 2.0382881164550782e-05,
      "model_forward_time": 1.7526743412017822,
      "step": 21373
    },
    {
      "epoch": 2.0382881164550782e-05,
      "step": 21373,
      "training_step_time": 7.565084218978882
    },
    {
      "epoch": 2.0383834838867186e-05,
      "model_forward_time": 1.7487130165100098,
      "step": 21374
    },
    {
      "epoch": 2.0383834838867186e-05,
      "step": 21374,
      "training_step_time": 9.565607070922852
    },
    {
      "epoch": 2.0384788513183594e-05,
      "model_forward_time": 1.751192569732666,
      "step": 21375
    },
    {
      "epoch": 2.0384788513183594e-05,
      "step": 21375,
      "training_step_time": 9.12814998626709
    },
    {
      "epoch": 2.03857421875e-05,
      "model_forward_time": 1.7424499988555908,
      "step": 21376
    },
    {
      "epoch": 2.03857421875e-05,
      "step": 21376,
      "training_step_time": 8.428610563278198
    },
    {
      "epoch": 2.0386695861816405e-05,
      "model_forward_time": 1.7518806457519531,
      "step": 21377
    },
    {
      "epoch": 2.0386695861816405e-05,
      "step": 21377,
      "training_step_time": 9.124462366104126
    },
    {
      "epoch": 2.0387649536132813e-05,
      "model_forward_time": 1.7373924255371094,
      "step": 21378
    },
    {
      "epoch": 2.0387649536132813e-05,
      "step": 21378,
      "training_step_time": 7.709691047668457
    },
    {
      "epoch": 2.038860321044922e-05,
      "model_forward_time": 1.7423880100250244,
      "step": 21379
    },
    {
      "epoch": 2.038860321044922e-05,
      "step": 21379,
      "training_step_time": 7.514070987701416
    },
    {
      "epoch": 2.0389556884765624e-05,
      "grad_norm": 0.20139500498771667,
      "learning_rate": 9.284312336042251e-06,
      "loss": 0.0634,
      "step": 21380
    },
    {
      "epoch": 2.0389556884765624e-05,
      "model_forward_time": 1.7484421730041504,
      "step": 21380
    },
    {
      "dynamics_loss_avg": 0.056768077239394185,
      "epoch": 2.0389556884765624e-05,
      "step": 21380
    },
    {
      "action_loss_avg": 0.0022232109331525862,
      "epoch": 2.0389556884765624e-05,
      "step": 21380
    },
    {
      "epoch": 2.0389556884765624e-05,
      "step": 21380,
      "training_step_time": 7.566098928451538
    },
    {
      "epoch": 2.039051055908203e-05,
      "model_forward_time": 1.7452383041381836,
      "step": 21381
    },
    {
      "epoch": 2.039051055908203e-05,
      "step": 21381,
      "training_step_time": 7.699277639389038
    },
    {
      "epoch": 2.039146423339844e-05,
      "model_forward_time": 1.7468533515930176,
      "step": 21382
    },
    {
      "epoch": 2.039146423339844e-05,
      "step": 21382,
      "training_step_time": 8.648533821105957
    },
    {
      "epoch": 2.0392417907714843e-05,
      "model_forward_time": 1.7366175651550293,
      "step": 21383
    },
    {
      "epoch": 2.0392417907714843e-05,
      "step": 21383,
      "training_step_time": 7.609086751937866
    },
    {
      "epoch": 2.039337158203125e-05,
      "model_forward_time": 2.8781793117523193,
      "step": 21384
    },
    {
      "epoch": 2.039337158203125e-05,
      "step": 21384,
      "training_step_time": 7.811579465866089
    },
    {
      "epoch": 2.0394325256347658e-05,
      "model_forward_time": 1.8265490531921387,
      "step": 21385
    },
    {
      "epoch": 2.0394325256347658e-05,
      "step": 21385,
      "training_step_time": 9.693722248077393
    },
    {
      "epoch": 2.0395278930664062e-05,
      "model_forward_time": 1.7377090454101562,
      "step": 21386
    },
    {
      "epoch": 2.0395278930664062e-05,
      "step": 21386,
      "training_step_time": 9.155572891235352
    },
    {
      "epoch": 2.039623260498047e-05,
      "model_forward_time": 1.741034746170044,
      "step": 21387
    },
    {
      "epoch": 2.039623260498047e-05,
      "step": 21387,
      "training_step_time": 8.70875883102417
    },
    {
      "epoch": 2.0397186279296877e-05,
      "model_forward_time": 1.7428841590881348,
      "step": 21388
    },
    {
      "epoch": 2.0397186279296877e-05,
      "step": 21388,
      "training_step_time": 7.3133544921875
    },
    {
      "epoch": 2.039813995361328e-05,
      "model_forward_time": 1.7483856678009033,
      "step": 21389
    },
    {
      "epoch": 2.039813995361328e-05,
      "step": 21389,
      "training_step_time": 8.152358531951904
    },
    {
      "epoch": 2.0399093627929688e-05,
      "grad_norm": 0.2070942223072052,
      "learning_rate": 9.283459663592997e-06,
      "loss": 0.0642,
      "step": 21390
    },
    {
      "epoch": 2.0399093627929688e-05,
      "model_forward_time": 1.7498698234558105,
      "step": 21390
    },
    {
      "dynamics_loss_avg": 0.06052822284400463,
      "epoch": 2.0399093627929688e-05,
      "step": 21390
    },
    {
      "action_loss_avg": 0.00321557346615009,
      "epoch": 2.0399093627929688e-05,
      "step": 21390
    },
    {
      "epoch": 2.0399093627929688e-05,
      "step": 21390,
      "training_step_time": 8.153788328170776
    },
    {
      "epoch": 2.0400047302246092e-05,
      "model_forward_time": 1.7424185276031494,
      "step": 21391
    },
    {
      "epoch": 2.0400047302246092e-05,
      "step": 21391,
      "training_step_time": 7.613089561462402
    },
    {
      "epoch": 2.04010009765625e-05,
      "model_forward_time": 1.7459533214569092,
      "step": 21392
    },
    {
      "epoch": 2.04010009765625e-05,
      "step": 21392,
      "training_step_time": 7.681341171264648
    },
    {
      "epoch": 2.0401954650878907e-05,
      "model_forward_time": 2.901535749435425,
      "step": 21393
    },
    {
      "epoch": 2.0401954650878907e-05,
      "step": 21393,
      "training_step_time": 8.614417314529419
    },
    {
      "epoch": 2.040290832519531e-05,
      "model_forward_time": 1.8416674137115479,
      "step": 21394
    },
    {
      "epoch": 2.040290832519531e-05,
      "step": 21394,
      "training_step_time": 8.368570566177368
    },
    {
      "epoch": 2.040386199951172e-05,
      "model_forward_time": 1.7436187267303467,
      "step": 21395
    },
    {
      "epoch": 2.040386199951172e-05,
      "step": 21395,
      "training_step_time": 8.830455541610718
    },
    {
      "epoch": 2.0404815673828126e-05,
      "model_forward_time": 1.7198116779327393,
      "step": 21396
    },
    {
      "epoch": 2.0404815673828126e-05,
      "step": 21396,
      "training_step_time": 9.438678979873657
    },
    {
      "epoch": 2.040576934814453e-05,
      "model_forward_time": 1.7347543239593506,
      "step": 21397
    },
    {
      "epoch": 2.040576934814453e-05,
      "step": 21397,
      "training_step_time": 7.638193845748901
    },
    {
      "epoch": 2.0406723022460937e-05,
      "model_forward_time": 1.7263460159301758,
      "step": 21398
    },
    {
      "epoch": 2.0406723022460937e-05,
      "step": 21398,
      "training_step_time": 7.602828502655029
    },
    {
      "epoch": 2.0407676696777345e-05,
      "model_forward_time": 1.7428929805755615,
      "step": 21399
    },
    {
      "epoch": 2.0407676696777345e-05,
      "step": 21399,
      "training_step_time": 7.615398645401001
    },
    {
      "epoch": 2.040863037109375e-05,
      "grad_norm": 0.186178058385849,
      "learning_rate": 9.282606522711033e-06,
      "loss": 0.0683,
      "step": 21400
    },
    {
      "epoch": 2.040863037109375e-05,
      "model_forward_time": 1.7269620895385742,
      "step": 21400
    },
    {
      "dynamics_loss_avg": 0.08186643831431865,
      "epoch": 2.040863037109375e-05,
      "step": 21400
    },
    {
      "action_loss_avg": 0.004573154263198376,
      "epoch": 2.040863037109375e-05,
      "step": 21400
    },
    {
      "epoch": 2.040863037109375e-05,
      "step": 21400,
      "training_step_time": 7.68349552154541
    },
    {
      "epoch": 2.0409584045410156e-05,
      "model_forward_time": 1.7425925731658936,
      "step": 21401
    },
    {
      "epoch": 2.0409584045410156e-05,
      "step": 21401,
      "training_step_time": 7.701425790786743
    },
    {
      "epoch": 2.0410537719726563e-05,
      "model_forward_time": 2.8865814208984375,
      "step": 21402
    },
    {
      "epoch": 2.0410537719726563e-05,
      "step": 21402,
      "training_step_time": 8.43549370765686
    },
    {
      "epoch": 2.0411491394042968e-05,
      "model_forward_time": 1.8154668807983398,
      "step": 21403
    },
    {
      "epoch": 2.0411491394042968e-05,
      "step": 21403,
      "training_step_time": 7.774026870727539
    },
    {
      "epoch": 2.0412445068359375e-05,
      "model_forward_time": 1.7485365867614746,
      "step": 21404
    },
    {
      "epoch": 2.0412445068359375e-05,
      "step": 21404,
      "training_step_time": 8.57439923286438
    },
    {
      "epoch": 2.0413398742675782e-05,
      "model_forward_time": 1.7316370010375977,
      "step": 21405
    },
    {
      "epoch": 2.0413398742675782e-05,
      "step": 21405,
      "training_step_time": 7.7668256759643555
    },
    {
      "epoch": 2.0414352416992186e-05,
      "model_forward_time": 1.740940809249878,
      "step": 21406
    },
    {
      "epoch": 2.0414352416992186e-05,
      "step": 21406,
      "training_step_time": 7.662129878997803
    },
    {
      "epoch": 2.0415306091308594e-05,
      "model_forward_time": 1.7300124168395996,
      "step": 21407
    },
    {
      "epoch": 2.0415306091308594e-05,
      "step": 21407,
      "training_step_time": 7.54496693611145
    },
    {
      "epoch": 2.0416259765625e-05,
      "model_forward_time": 1.7471537590026855,
      "step": 21408
    },
    {
      "epoch": 2.0416259765625e-05,
      "step": 21408,
      "training_step_time": 7.578978776931763
    },
    {
      "epoch": 2.0417213439941405e-05,
      "model_forward_time": 1.754852056503296,
      "step": 21409
    },
    {
      "epoch": 2.0417213439941405e-05,
      "step": 21409,
      "training_step_time": 7.757838010787964
    },
    {
      "epoch": 2.0418167114257813e-05,
      "grad_norm": 0.22360318899154663,
      "learning_rate": 9.281752913489658e-06,
      "loss": 0.0638,
      "step": 21410
    },
    {
      "epoch": 2.0418167114257813e-05,
      "model_forward_time": 1.7403624057769775,
      "step": 21410
    },
    {
      "dynamics_loss_avg": 0.06414887253195048,
      "epoch": 2.0418167114257813e-05,
      "step": 21410
    },
    {
      "action_loss_avg": 0.0032800106797367333,
      "epoch": 2.0418167114257813e-05,
      "step": 21410
    },
    {
      "epoch": 2.0418167114257813e-05,
      "step": 21410,
      "training_step_time": 7.669283390045166
    },
    {
      "epoch": 2.041912078857422e-05,
      "model_forward_time": 2.841998815536499,
      "step": 21411
    },
    {
      "epoch": 2.041912078857422e-05,
      "step": 21411,
      "training_step_time": 7.854298114776611
    },
    {
      "epoch": 2.0420074462890624e-05,
      "model_forward_time": 1.7923619747161865,
      "step": 21412
    },
    {
      "epoch": 2.0420074462890624e-05,
      "step": 21412,
      "training_step_time": 9.715853214263916
    },
    {
      "epoch": 2.042102813720703e-05,
      "model_forward_time": 1.7394423484802246,
      "step": 21413
    },
    {
      "epoch": 2.042102813720703e-05,
      "step": 21413,
      "training_step_time": 9.512950897216797
    },
    {
      "epoch": 2.042198181152344e-05,
      "model_forward_time": 1.7427833080291748,
      "step": 21414
    },
    {
      "epoch": 2.042198181152344e-05,
      "step": 21414,
      "training_step_time": 9.97637939453125
    },
    {
      "epoch": 2.0422935485839843e-05,
      "model_forward_time": 1.7441813945770264,
      "step": 21415
    },
    {
      "epoch": 2.0422935485839843e-05,
      "step": 21415,
      "training_step_time": 7.674783229827881
    },
    {
      "epoch": 2.042388916015625e-05,
      "model_forward_time": 1.770219326019287,
      "step": 21416
    },
    {
      "epoch": 2.042388916015625e-05,
      "step": 21416,
      "training_step_time": 7.584740400314331
    },
    {
      "epoch": 2.0424842834472658e-05,
      "model_forward_time": 1.3667595386505127,
      "step": 21417
    },
    {
      "epoch": 2.0424842834472658e-05,
      "step": 21417,
      "training_step_time": 7.592731952667236
    },
    {
      "epoch": 2.0425796508789062e-05,
      "model_forward_time": 1.752781629562378,
      "step": 21418
    },
    {
      "epoch": 2.0425796508789062e-05,
      "step": 21418,
      "training_step_time": 7.722764253616333
    },
    {
      "epoch": 2.042675018310547e-05,
      "model_forward_time": 1.7477781772613525,
      "step": 21419
    },
    {
      "epoch": 2.042675018310547e-05,
      "step": 21419,
      "training_step_time": 7.73075008392334
    },
    {
      "epoch": 2.0427703857421877e-05,
      "grad_norm": 0.22367793321609497,
      "learning_rate": 9.280898836022223e-06,
      "loss": 0.0653,
      "step": 21420
    },
    {
      "epoch": 2.0427703857421877e-05,
      "model_forward_time": 1.7328693866729736,
      "step": 21420
    },
    {
      "dynamics_loss_avg": 0.06634674221277237,
      "epoch": 2.0427703857421877e-05,
      "step": 21420
    },
    {
      "action_loss_avg": 0.0032681658980436624,
      "epoch": 2.0427703857421877e-05,
      "step": 21420
    },
    {
      "epoch": 2.0427703857421877e-05,
      "step": 21420,
      "training_step_time": 8.440732479095459
    },
    {
      "epoch": 2.042865753173828e-05,
      "model_forward_time": 1.7663507461547852,
      "step": 21421
    },
    {
      "epoch": 2.042865753173828e-05,
      "step": 21421,
      "training_step_time": 7.801758289337158
    },
    {
      "epoch": 2.0429611206054688e-05,
      "model_forward_time": 1.74149489402771,
      "step": 21422
    },
    {
      "epoch": 2.0429611206054688e-05,
      "step": 21422,
      "training_step_time": 8.472646236419678
    },
    {
      "epoch": 2.0430564880371092e-05,
      "model_forward_time": 1.7457492351531982,
      "step": 21423
    },
    {
      "epoch": 2.0430564880371092e-05,
      "step": 21423,
      "training_step_time": 8.914292812347412
    },
    {
      "epoch": 2.04315185546875e-05,
      "model_forward_time": 1.74349045753479,
      "step": 21424
    },
    {
      "epoch": 2.04315185546875e-05,
      "step": 21424,
      "training_step_time": 7.816608905792236
    },
    {
      "epoch": 2.0432472229003907e-05,
      "model_forward_time": 1.7450473308563232,
      "step": 21425
    },
    {
      "epoch": 2.0432472229003907e-05,
      "step": 21425,
      "training_step_time": 7.5683887004852295
    },
    {
      "epoch": 2.043342590332031e-05,
      "model_forward_time": 1.7533423900604248,
      "step": 21426
    },
    {
      "epoch": 2.043342590332031e-05,
      "step": 21426,
      "training_step_time": 8.14791226387024
    },
    {
      "epoch": 2.043437957763672e-05,
      "model_forward_time": 1.7493505477905273,
      "step": 21427
    },
    {
      "epoch": 2.043437957763672e-05,
      "step": 21427,
      "training_step_time": 7.729220628738403
    },
    {
      "epoch": 2.0435333251953126e-05,
      "model_forward_time": 1.7455685138702393,
      "step": 21428
    },
    {
      "epoch": 2.0435333251953126e-05,
      "step": 21428,
      "training_step_time": 7.7272186279296875
    },
    {
      "epoch": 2.043628692626953e-05,
      "model_forward_time": 1.7436211109161377,
      "step": 21429
    },
    {
      "epoch": 2.043628692626953e-05,
      "step": 21429,
      "training_step_time": 8.075801849365234
    },
    {
      "epoch": 2.0437240600585937e-05,
      "grad_norm": 0.20613166689872742,
      "learning_rate": 9.280044290402127e-06,
      "loss": 0.0649,
      "step": 21430
    },
    {
      "epoch": 2.0437240600585937e-05,
      "model_forward_time": 2.888197422027588,
      "step": 21430
    },
    {
      "dynamics_loss_avg": 0.07437129858881235,
      "epoch": 2.0437240600585937e-05,
      "step": 21430
    },
    {
      "action_loss_avg": 0.002372324460884556,
      "epoch": 2.0437240600585937e-05,
      "step": 21430
    },
    {
      "epoch": 2.0437240600585937e-05,
      "step": 21430,
      "training_step_time": 8.531863927841187
    },
    {
      "epoch": 2.0438194274902345e-05,
      "model_forward_time": 1.8419408798217773,
      "step": 21431
    },
    {
      "epoch": 2.0438194274902345e-05,
      "step": 21431,
      "training_step_time": 7.670700550079346
    },
    {
      "epoch": 2.043914794921875e-05,
      "model_forward_time": 1.7378594875335693,
      "step": 21432
    },
    {
      "epoch": 2.043914794921875e-05,
      "step": 21432,
      "training_step_time": 9.618237018585205
    },
    {
      "epoch": 2.0440101623535156e-05,
      "model_forward_time": 1.7476661205291748,
      "step": 21433
    },
    {
      "epoch": 2.0440101623535156e-05,
      "step": 21433,
      "training_step_time": 7.674832344055176
    },
    {
      "epoch": 2.0441055297851564e-05,
      "model_forward_time": 1.7184994220733643,
      "step": 21434
    },
    {
      "epoch": 2.0441055297851564e-05,
      "step": 21434,
      "training_step_time": 7.508009433746338
    },
    {
      "epoch": 2.0442008972167968e-05,
      "model_forward_time": 1.7499253749847412,
      "step": 21435
    },
    {
      "epoch": 2.0442008972167968e-05,
      "step": 21435,
      "training_step_time": 7.587148666381836
    },
    {
      "epoch": 2.0442962646484375e-05,
      "model_forward_time": 1.7487678527832031,
      "step": 21436
    },
    {
      "epoch": 2.0442962646484375e-05,
      "step": 21436,
      "training_step_time": 7.631991863250732
    },
    {
      "epoch": 2.0443916320800782e-05,
      "model_forward_time": 1.7400343418121338,
      "step": 21437
    },
    {
      "epoch": 2.0443916320800782e-05,
      "step": 21437,
      "training_step_time": 7.723517179489136
    },
    {
      "epoch": 2.0444869995117186e-05,
      "model_forward_time": 1.7504782676696777,
      "step": 21438
    },
    {
      "epoch": 2.0444869995117186e-05,
      "step": 21438,
      "training_step_time": 8.68053126335144
    },
    {
      "epoch": 2.0445823669433594e-05,
      "model_forward_time": 2.9154021739959717,
      "step": 21439
    },
    {
      "epoch": 2.0445823669433594e-05,
      "step": 21439,
      "training_step_time": 8.26322078704834
    },
    {
      "epoch": 2.044677734375e-05,
      "grad_norm": 0.1445409059524536,
      "learning_rate": 9.279189276722822e-06,
      "loss": 0.0645,
      "step": 21440
    },
    {
      "epoch": 2.044677734375e-05,
      "model_forward_time": 1.850036382675171,
      "step": 21440
    },
    {
      "dynamics_loss_avg": 0.08615107722580433,
      "epoch": 2.044677734375e-05,
      "step": 21440
    },
    {
      "action_loss_avg": 0.005640563229098916,
      "epoch": 2.044677734375e-05,
      "step": 21440
    },
    {
      "epoch": 2.044677734375e-05,
      "step": 21440,
      "training_step_time": 7.715424060821533
    },
    {
      "epoch": 2.0447731018066405e-05,
      "model_forward_time": 1.743464708328247,
      "step": 21441
    },
    {
      "epoch": 2.0447731018066405e-05,
      "step": 21441,
      "training_step_time": 7.837364435195923
    },
    {
      "epoch": 2.0448684692382813e-05,
      "model_forward_time": 1.7462148666381836,
      "step": 21442
    },
    {
      "epoch": 2.0448684692382813e-05,
      "step": 21442,
      "training_step_time": 7.6918416023254395
    },
    {
      "epoch": 2.044963836669922e-05,
      "model_forward_time": 1.7577388286590576,
      "step": 21443
    },
    {
      "epoch": 2.044963836669922e-05,
      "step": 21443,
      "training_step_time": 7.6501851081848145
    },
    {
      "epoch": 2.0450592041015624e-05,
      "model_forward_time": 1.7439212799072266,
      "step": 21444
    },
    {
      "epoch": 2.0450592041015624e-05,
      "step": 21444,
      "training_step_time": 7.635102033615112
    },
    {
      "epoch": 2.045154571533203e-05,
      "model_forward_time": 1.7439820766448975,
      "step": 21445
    },
    {
      "epoch": 2.045154571533203e-05,
      "step": 21445,
      "training_step_time": 7.784850597381592
    },
    {
      "epoch": 2.045249938964844e-05,
      "model_forward_time": 1.7453334331512451,
      "step": 21446
    },
    {
      "epoch": 2.045249938964844e-05,
      "step": 21446,
      "training_step_time": 7.77769660949707
    },
    {
      "epoch": 2.0453453063964843e-05,
      "model_forward_time": 1.7422356605529785,
      "step": 21447
    },
    {
      "epoch": 2.0453453063964843e-05,
      "step": 21447,
      "training_step_time": 7.769906997680664
    },
    {
      "epoch": 2.045440673828125e-05,
      "model_forward_time": 2.903693914413452,
      "step": 21448
    },
    {
      "epoch": 2.045440673828125e-05,
      "step": 21448,
      "training_step_time": 7.838505029678345
    },
    {
      "epoch": 2.0455360412597658e-05,
      "model_forward_time": 1.81388258934021,
      "step": 21449
    },
    {
      "epoch": 2.0455360412597658e-05,
      "step": 21449,
      "training_step_time": 7.719665765762329
    },
    {
      "epoch": 2.0456314086914062e-05,
      "grad_norm": 0.18770354986190796,
      "learning_rate": 9.278333795077812e-06,
      "loss": 0.0641,
      "step": 21450
    },
    {
      "epoch": 2.0456314086914062e-05,
      "model_forward_time": 1.7531700134277344,
      "step": 21450
    },
    {
      "dynamics_loss_avg": 0.05734096895903349,
      "epoch": 2.0456314086914062e-05,
      "step": 21450
    },
    {
      "action_loss_avg": 0.0028706446464639157,
      "epoch": 2.0456314086914062e-05,
      "step": 21450
    },
    {
      "epoch": 2.0456314086914062e-05,
      "step": 21450,
      "training_step_time": 8.972686767578125
    },
    {
      "epoch": 2.045726776123047e-05,
      "model_forward_time": 1.7500262260437012,
      "step": 21451
    },
    {
      "epoch": 2.045726776123047e-05,
      "step": 21451,
      "training_step_time": 7.791776657104492
    },
    {
      "epoch": 2.0458221435546877e-05,
      "model_forward_time": 1.7464869022369385,
      "step": 21452
    },
    {
      "epoch": 2.0458221435546877e-05,
      "step": 21452,
      "training_step_time": 7.823873043060303
    },
    {
      "epoch": 2.045917510986328e-05,
      "model_forward_time": 1.7410473823547363,
      "step": 21453
    },
    {
      "epoch": 2.045917510986328e-05,
      "step": 21453,
      "training_step_time": 8.327385425567627
    },
    {
      "epoch": 2.0460128784179688e-05,
      "model_forward_time": 1.7465341091156006,
      "step": 21454
    },
    {
      "epoch": 2.0460128784179688e-05,
      "step": 21454,
      "training_step_time": 7.379554986953735
    },
    {
      "epoch": 2.0461082458496092e-05,
      "model_forward_time": 1.7518610954284668,
      "step": 21455
    },
    {
      "epoch": 2.0461082458496092e-05,
      "step": 21455,
      "training_step_time": 8.636451244354248
    },
    {
      "epoch": 2.04620361328125e-05,
      "model_forward_time": 1.7486205101013184,
      "step": 21456
    },
    {
      "epoch": 2.04620361328125e-05,
      "step": 21456,
      "training_step_time": 7.7934253215789795
    },
    {
      "epoch": 2.0462989807128907e-05,
      "model_forward_time": 2.883629322052002,
      "step": 21457
    },
    {
      "epoch": 2.0462989807128907e-05,
      "step": 21457,
      "training_step_time": 7.965548276901245
    },
    {
      "epoch": 2.046394348144531e-05,
      "model_forward_time": 1.7827892303466797,
      "step": 21458
    },
    {
      "epoch": 2.046394348144531e-05,
      "step": 21458,
      "training_step_time": 7.823001861572266
    },
    {
      "epoch": 2.046489715576172e-05,
      "model_forward_time": 1.7404656410217285,
      "step": 21459
    },
    {
      "epoch": 2.046489715576172e-05,
      "step": 21459,
      "training_step_time": 7.799976825714111
    },
    {
      "epoch": 2.0465850830078126e-05,
      "grad_norm": 0.26314786076545715,
      "learning_rate": 9.277477845560649e-06,
      "loss": 0.0659,
      "step": 21460
    },
    {
      "epoch": 2.0465850830078126e-05,
      "model_forward_time": 1.7522683143615723,
      "step": 21460
    },
    {
      "dynamics_loss_avg": 0.05613132007420063,
      "epoch": 2.0465850830078126e-05,
      "step": 21460
    },
    {
      "action_loss_avg": 0.002563637204002589,
      "epoch": 2.0465850830078126e-05,
      "step": 21460
    },
    {
      "epoch": 2.0465850830078126e-05,
      "step": 21460,
      "training_step_time": 8.582682609558105
    },
    {
      "epoch": 2.046680450439453e-05,
      "model_forward_time": 1.743797779083252,
      "step": 21461
    },
    {
      "epoch": 2.046680450439453e-05,
      "step": 21461,
      "training_step_time": 7.7911765575408936
    },
    {
      "epoch": 2.0467758178710937e-05,
      "model_forward_time": 1.748194932937622,
      "step": 21462
    },
    {
      "epoch": 2.0467758178710937e-05,
      "step": 21462,
      "training_step_time": 7.618537187576294
    },
    {
      "epoch": 2.0468711853027345e-05,
      "model_forward_time": 1.7437965869903564,
      "step": 21463
    },
    {
      "epoch": 2.0468711853027345e-05,
      "step": 21463,
      "training_step_time": 7.7551186084747314
    },
    {
      "epoch": 2.046966552734375e-05,
      "model_forward_time": 1.7479548454284668,
      "step": 21464
    },
    {
      "epoch": 2.046966552734375e-05,
      "step": 21464,
      "training_step_time": 7.792635440826416
    },
    {
      "epoch": 2.0470619201660156e-05,
      "model_forward_time": 1.7415168285369873,
      "step": 21465
    },
    {
      "epoch": 2.0470619201660156e-05,
      "step": 21465,
      "training_step_time": 7.718261480331421
    },
    {
      "epoch": 2.0471572875976564e-05,
      "model_forward_time": 2.832782506942749,
      "step": 21466
    },
    {
      "epoch": 2.0471572875976564e-05,
      "step": 21466,
      "training_step_time": 7.675153017044067
    },
    {
      "epoch": 2.0472526550292968e-05,
      "model_forward_time": 1.746070384979248,
      "step": 21467
    },
    {
      "epoch": 2.0472526550292968e-05,
      "step": 21467,
      "training_step_time": 8.72217082977295
    },
    {
      "epoch": 2.0473480224609375e-05,
      "model_forward_time": 1.7390499114990234,
      "step": 21468
    },
    {
      "epoch": 2.0473480224609375e-05,
      "step": 21468,
      "training_step_time": 7.733883619308472
    },
    {
      "epoch": 2.0474433898925782e-05,
      "model_forward_time": 1.7467241287231445,
      "step": 21469
    },
    {
      "epoch": 2.0474433898925782e-05,
      "step": 21469,
      "training_step_time": 7.8806867599487305
    },
    {
      "epoch": 2.0475387573242186e-05,
      "grad_norm": 0.18182526528835297,
      "learning_rate": 9.276621428264943e-06,
      "loss": 0.0598,
      "step": 21470
    },
    {
      "epoch": 2.0475387573242186e-05,
      "model_forward_time": 1.740941047668457,
      "step": 21470
    },
    {
      "dynamics_loss_avg": 0.04778836006298661,
      "epoch": 2.0475387573242186e-05,
      "step": 21470
    },
    {
      "action_loss_avg": 0.0024692885810509322,
      "epoch": 2.0475387573242186e-05,
      "step": 21470
    },
    {
      "epoch": 2.0475387573242186e-05,
      "step": 21470,
      "training_step_time": 7.752845764160156
    },
    {
      "epoch": 2.0476341247558594e-05,
      "model_forward_time": 1.7432763576507568,
      "step": 21471
    },
    {
      "epoch": 2.0476341247558594e-05,
      "step": 21471,
      "training_step_time": 7.653584957122803
    },
    {
      "epoch": 2.0477294921875e-05,
      "model_forward_time": 1.0409750938415527,
      "step": 21472
    },
    {
      "epoch": 2.0477294921875e-05,
      "step": 21472,
      "training_step_time": 7.907013893127441
    },
    {
      "epoch": 2.0478248596191405e-05,
      "model_forward_time": 1.743847370147705,
      "step": 21473
    },
    {
      "epoch": 2.0478248596191405e-05,
      "step": 21473,
      "training_step_time": 8.846951246261597
    },
    {
      "epoch": 2.0479202270507813e-05,
      "model_forward_time": 1.739408254623413,
      "step": 21474
    },
    {
      "epoch": 2.0479202270507813e-05,
      "step": 21474,
      "training_step_time": 9.057362079620361
    },
    {
      "epoch": 2.048015594482422e-05,
      "model_forward_time": 1.7421481609344482,
      "step": 21475
    },
    {
      "epoch": 2.048015594482422e-05,
      "step": 21475,
      "training_step_time": 8.001399993896484
    },
    {
      "epoch": 2.0481109619140624e-05,
      "model_forward_time": 2.915118455886841,
      "step": 21476
    },
    {
      "epoch": 2.0481109619140624e-05,
      "step": 21476,
      "training_step_time": 8.476901292800903
    },
    {
      "epoch": 2.048206329345703e-05,
      "model_forward_time": 1.8609106540679932,
      "step": 21477
    },
    {
      "epoch": 2.048206329345703e-05,
      "step": 21477,
      "training_step_time": 7.693496942520142
    },
    {
      "epoch": 2.048301696777344e-05,
      "model_forward_time": 1.742175579071045,
      "step": 21478
    },
    {
      "epoch": 2.048301696777344e-05,
      "step": 21478,
      "training_step_time": 8.332797288894653
    },
    {
      "epoch": 2.0483970642089843e-05,
      "model_forward_time": 1.737936019897461,
      "step": 21479
    },
    {
      "epoch": 2.0483970642089843e-05,
      "step": 21479,
      "training_step_time": 9.287604570388794
    },
    {
      "epoch": 2.048492431640625e-05,
      "grad_norm": 0.19218070805072784,
      "learning_rate": 9.275764543284344e-06,
      "loss": 0.064,
      "step": 21480
    },
    {
      "epoch": 2.048492431640625e-05,
      "model_forward_time": 1.7445952892303467,
      "step": 21480
    },
    {
      "dynamics_loss_avg": 0.06705465465784073,
      "epoch": 2.048492431640625e-05,
      "step": 21480
    },
    {
      "action_loss_avg": 0.0031643738242564724,
      "epoch": 2.048492431640625e-05,
      "step": 21480
    },
    {
      "epoch": 2.048492431640625e-05,
      "step": 21480,
      "training_step_time": 7.735710382461548
    },
    {
      "epoch": 2.0485877990722658e-05,
      "model_forward_time": 1.7432928085327148,
      "step": 21481
    },
    {
      "epoch": 2.0485877990722658e-05,
      "step": 21481,
      "training_step_time": 7.7353246212005615
    },
    {
      "epoch": 2.0486831665039062e-05,
      "model_forward_time": 1.742438554763794,
      "step": 21482
    },
    {
      "epoch": 2.0486831665039062e-05,
      "step": 21482,
      "training_step_time": 7.752239227294922
    },
    {
      "epoch": 2.048778533935547e-05,
      "model_forward_time": 1.7439048290252686,
      "step": 21483
    },
    {
      "epoch": 2.048778533935547e-05,
      "step": 21483,
      "training_step_time": 7.631864309310913
    },
    {
      "epoch": 2.0488739013671877e-05,
      "model_forward_time": 1.7471048831939697,
      "step": 21484
    },
    {
      "epoch": 2.0488739013671877e-05,
      "step": 21484,
      "training_step_time": 7.820334434509277
    },
    {
      "epoch": 2.048969268798828e-05,
      "model_forward_time": 2.8669850826263428,
      "step": 21485
    },
    {
      "epoch": 2.048969268798828e-05,
      "step": 21485,
      "training_step_time": 7.6512534618377686
    },
    {
      "epoch": 2.0490646362304688e-05,
      "model_forward_time": 1.8246724605560303,
      "step": 21486
    },
    {
      "epoch": 2.0490646362304688e-05,
      "step": 21486,
      "training_step_time": 8.375256299972534
    },
    {
      "epoch": 2.0491600036621092e-05,
      "model_forward_time": 1.7481601238250732,
      "step": 21487
    },
    {
      "epoch": 2.0491600036621092e-05,
      "step": 21487,
      "training_step_time": 7.706496477127075
    },
    {
      "epoch": 2.04925537109375e-05,
      "model_forward_time": 1.7404084205627441,
      "step": 21488
    },
    {
      "epoch": 2.04925537109375e-05,
      "step": 21488,
      "training_step_time": 7.770488500595093
    },
    {
      "epoch": 2.0493507385253907e-05,
      "model_forward_time": 1.7302942276000977,
      "step": 21489
    },
    {
      "epoch": 2.0493507385253907e-05,
      "step": 21489,
      "training_step_time": 8.306692123413086
    },
    {
      "epoch": 2.049446105957031e-05,
      "grad_norm": 0.1893167942762375,
      "learning_rate": 9.274907190712566e-06,
      "loss": 0.0635,
      "step": 21490
    },
    {
      "epoch": 2.049446105957031e-05,
      "model_forward_time": 1.7197070121765137,
      "step": 21490
    },
    {
      "dynamics_loss_avg": 0.06111699696630239,
      "epoch": 2.049446105957031e-05,
      "step": 21490
    },
    {
      "action_loss_avg": 0.0023478947463445366,
      "epoch": 2.049446105957031e-05,
      "step": 21490
    },
    {
      "epoch": 2.049446105957031e-05,
      "step": 21490,
      "training_step_time": 7.625282526016235
    },
    {
      "epoch": 2.049541473388672e-05,
      "model_forward_time": 1.738910436630249,
      "step": 21491
    },
    {
      "epoch": 2.049541473388672e-05,
      "step": 21491,
      "training_step_time": 8.847235441207886
    },
    {
      "epoch": 2.0496368408203126e-05,
      "model_forward_time": 1.7960858345031738,
      "step": 21492
    },
    {
      "epoch": 2.0496368408203126e-05,
      "step": 21492,
      "training_step_time": 7.641102313995361
    },
    {
      "epoch": 2.049732208251953e-05,
      "model_forward_time": 1.733341932296753,
      "step": 21493
    },
    {
      "epoch": 2.049732208251953e-05,
      "step": 21493,
      "training_step_time": 8.732259511947632
    },
    {
      "epoch": 2.0498275756835937e-05,
      "model_forward_time": 2.8498072624206543,
      "step": 21494
    },
    {
      "epoch": 2.0498275756835937e-05,
      "step": 21494,
      "training_step_time": 9.335891723632812
    },
    {
      "epoch": 2.0499229431152345e-05,
      "model_forward_time": 1.7682316303253174,
      "step": 21495
    },
    {
      "epoch": 2.0499229431152345e-05,
      "step": 21495,
      "training_step_time": 9.299850702285767
    },
    {
      "epoch": 2.050018310546875e-05,
      "model_forward_time": 1.7407128810882568,
      "step": 21496
    },
    {
      "epoch": 2.050018310546875e-05,
      "step": 21496,
      "training_step_time": 8.410860776901245
    },
    {
      "epoch": 2.0501136779785156e-05,
      "model_forward_time": 1.7417235374450684,
      "step": 21497
    },
    {
      "epoch": 2.0501136779785156e-05,
      "step": 21497,
      "training_step_time": 8.85772705078125
    },
    {
      "epoch": 2.0502090454101564e-05,
      "model_forward_time": 1.7387421131134033,
      "step": 21498
    },
    {
      "epoch": 2.0502090454101564e-05,
      "step": 21498,
      "training_step_time": 7.777425050735474
    },
    {
      "epoch": 2.0503044128417968e-05,
      "model_forward_time": 1.746159553527832,
      "step": 21499
    },
    {
      "epoch": 2.0503044128417968e-05,
      "step": 21499,
      "training_step_time": 7.592316627502441
    },
    {
      "epoch": 2.0503997802734375e-05,
      "grad_norm": 0.21971391141414642,
      "learning_rate": 9.274049370643363e-06,
      "loss": 0.067,
      "step": 21500
    },
    {
      "epoch": 2.0503997802734375e-05,
      "model_forward_time": 1.7320969104766846,
      "step": 21500
    },
    {
      "dynamics_loss_avg": 0.05933834165334702,
      "epoch": 2.0503997802734375e-05,
      "step": 21500
    },
    {
      "action_loss_avg": 0.002895510697271675,
      "epoch": 2.0503997802734375e-05,
      "step": 21500
    },
    {
      "epoch": 2.0503997802734375e-05,
      "step": 21500,
      "training_step_time": 8.669143676757812
    },
    {
      "epoch": 2.0504951477050782e-05,
      "model_forward_time": 1.7360529899597168,
      "step": 21501
    },
    {
      "epoch": 2.0504951477050782e-05,
      "step": 21501,
      "training_step_time": 7.6625816822052
    },
    {
      "epoch": 2.0505905151367186e-05,
      "model_forward_time": 1.7451915740966797,
      "step": 21502
    },
    {
      "epoch": 2.0505905151367186e-05,
      "step": 21502,
      "training_step_time": 7.680708646774292
    },
    {
      "epoch": 2.0506858825683594e-05,
      "model_forward_time": 2.8159728050231934,
      "step": 21503
    },
    {
      "epoch": 2.0506858825683594e-05,
      "step": 21503,
      "training_step_time": 7.804843902587891
    },
    {
      "epoch": 2.05078125e-05,
      "model_forward_time": 1.7457830905914307,
      "step": 21504
    },
    {
      "epoch": 2.05078125e-05,
      "step": 21504,
      "training_step_time": 8.386217594146729
    },
    {
      "epoch": 2.0508766174316405e-05,
      "model_forward_time": 1.8603358268737793,
      "step": 21505
    },
    {
      "epoch": 2.0508766174316405e-05,
      "step": 21505,
      "training_step_time": 8.391831874847412
    },
    {
      "epoch": 2.0509719848632813e-05,
      "model_forward_time": 1.7419400215148926,
      "step": 21506
    },
    {
      "epoch": 2.0509719848632813e-05,
      "step": 21506,
      "training_step_time": 7.792677640914917
    },
    {
      "epoch": 2.051067352294922e-05,
      "model_forward_time": 1.7383534908294678,
      "step": 21507
    },
    {
      "epoch": 2.051067352294922e-05,
      "step": 21507,
      "training_step_time": 7.881423473358154
    },
    {
      "epoch": 2.0511627197265624e-05,
      "model_forward_time": 1.7442162036895752,
      "step": 21508
    },
    {
      "epoch": 2.0511627197265624e-05,
      "step": 21508,
      "training_step_time": 8.111191749572754
    },
    {
      "epoch": 2.051258087158203e-05,
      "model_forward_time": 1.7317569255828857,
      "step": 21509
    },
    {
      "epoch": 2.051258087158203e-05,
      "step": 21509,
      "training_step_time": 8.70908260345459
    },
    {
      "epoch": 2.051353454589844e-05,
      "grad_norm": 0.21437384188175201,
      "learning_rate": 9.273191083170548e-06,
      "loss": 0.0669,
      "step": 21510
    },
    {
      "epoch": 2.051353454589844e-05,
      "model_forward_time": 1.7440452575683594,
      "step": 21510
    },
    {
      "dynamics_loss_avg": 0.053339113295078275,
      "epoch": 2.051353454589844e-05,
      "step": 21510
    },
    {
      "action_loss_avg": 0.0018947836477309465,
      "epoch": 2.051353454589844e-05,
      "step": 21510
    },
    {
      "epoch": 2.051353454589844e-05,
      "step": 21510,
      "training_step_time": 7.738218784332275
    },
    {
      "epoch": 2.0514488220214843e-05,
      "model_forward_time": 1.7294425964355469,
      "step": 21511
    },
    {
      "epoch": 2.0514488220214843e-05,
      "step": 21511,
      "training_step_time": 8.36847972869873
    },
    {
      "epoch": 2.051544189453125e-05,
      "model_forward_time": 1.7400257587432861,
      "step": 21512
    },
    {
      "epoch": 2.051544189453125e-05,
      "step": 21512,
      "training_step_time": 7.928889036178589
    },
    {
      "epoch": 2.0516395568847658e-05,
      "model_forward_time": 2.8417556285858154,
      "step": 21513
    },
    {
      "epoch": 2.0516395568847658e-05,
      "step": 21513,
      "training_step_time": 8.505873918533325
    },
    {
      "epoch": 2.0517349243164062e-05,
      "model_forward_time": 1.8247170448303223,
      "step": 21514
    },
    {
      "epoch": 2.0517349243164062e-05,
      "step": 21514,
      "training_step_time": 8.207120418548584
    },
    {
      "epoch": 2.051830291748047e-05,
      "model_forward_time": 1.7464754581451416,
      "step": 21515
    },
    {
      "epoch": 2.051830291748047e-05,
      "step": 21515,
      "training_step_time": 7.739025354385376
    },
    {
      "epoch": 2.0519256591796873e-05,
      "model_forward_time": 1.735564947128296,
      "step": 21516
    },
    {
      "epoch": 2.0519256591796873e-05,
      "step": 21516,
      "training_step_time": 9.169056177139282
    },
    {
      "epoch": 2.052021026611328e-05,
      "model_forward_time": 1.7416372299194336,
      "step": 21517
    },
    {
      "epoch": 2.052021026611328e-05,
      "step": 21517,
      "training_step_time": 8.339840650558472
    },
    {
      "epoch": 2.0521163940429688e-05,
      "model_forward_time": 1.7269034385681152,
      "step": 21518
    },
    {
      "epoch": 2.0521163940429688e-05,
      "step": 21518,
      "training_step_time": 8.70749568939209
    },
    {
      "epoch": 2.0522117614746092e-05,
      "model_forward_time": 1.7507107257843018,
      "step": 21519
    },
    {
      "epoch": 2.0522117614746092e-05,
      "step": 21519,
      "training_step_time": 8.235631227493286
    },
    {
      "epoch": 2.05230712890625e-05,
      "grad_norm": 0.1701660305261612,
      "learning_rate": 9.272332328387979e-06,
      "loss": 0.0628,
      "step": 21520
    },
    {
      "epoch": 2.05230712890625e-05,
      "model_forward_time": 1.74709153175354,
      "step": 21520
    },
    {
      "dynamics_loss_avg": 0.04936903361231089,
      "epoch": 2.05230712890625e-05,
      "step": 21520
    },
    {
      "action_loss_avg": 0.0025269298814237116,
      "epoch": 2.05230712890625e-05,
      "step": 21520
    },
    {
      "epoch": 2.05230712890625e-05,
      "step": 21520,
      "training_step_time": 7.6365673542022705
    },
    {
      "epoch": 2.0524024963378907e-05,
      "model_forward_time": 1.7485668659210205,
      "step": 21521
    },
    {
      "epoch": 2.0524024963378907e-05,
      "step": 21521,
      "training_step_time": 7.918315172195435
    },
    {
      "epoch": 2.052497863769531e-05,
      "model_forward_time": 2.9495415687561035,
      "step": 21522
    },
    {
      "epoch": 2.052497863769531e-05,
      "step": 21522,
      "training_step_time": 9.63658618927002
    },
    {
      "epoch": 2.052593231201172e-05,
      "model_forward_time": 1.8054096698760986,
      "step": 21523
    },
    {
      "epoch": 2.052593231201172e-05,
      "step": 21523,
      "training_step_time": 8.342255353927612
    },
    {
      "epoch": 2.0526885986328126e-05,
      "model_forward_time": 1.7485036849975586,
      "step": 21524
    },
    {
      "epoch": 2.0526885986328126e-05,
      "step": 21524,
      "training_step_time": 8.842377424240112
    },
    {
      "epoch": 2.052783966064453e-05,
      "model_forward_time": 1.7632205486297607,
      "step": 21525
    },
    {
      "epoch": 2.052783966064453e-05,
      "step": 21525,
      "training_step_time": 9.54885745048523
    },
    {
      "epoch": 2.0528793334960937e-05,
      "model_forward_time": 1.7371301651000977,
      "step": 21526
    },
    {
      "epoch": 2.0528793334960937e-05,
      "step": 21526,
      "training_step_time": 7.696260929107666
    },
    {
      "epoch": 2.0529747009277345e-05,
      "model_forward_time": 1.751533031463623,
      "step": 21527
    },
    {
      "epoch": 2.0529747009277345e-05,
      "step": 21527,
      "training_step_time": 7.764842748641968
    },
    {
      "epoch": 2.053070068359375e-05,
      "model_forward_time": 1.745661735534668,
      "step": 21528
    },
    {
      "epoch": 2.053070068359375e-05,
      "step": 21528,
      "training_step_time": 7.714838027954102
    },
    {
      "epoch": 2.0531654357910156e-05,
      "model_forward_time": 1.74522066116333,
      "step": 21529
    },
    {
      "epoch": 2.0531654357910156e-05,
      "step": 21529,
      "training_step_time": 7.735987424850464
    },
    {
      "epoch": 2.0532608032226564e-05,
      "grad_norm": 0.1979120522737503,
      "learning_rate": 9.271473106389571e-06,
      "loss": 0.067,
      "step": 21530
    },
    {
      "epoch": 2.0532608032226564e-05,
      "model_forward_time": 1.751025676727295,
      "step": 21530
    },
    {
      "dynamics_loss_avg": 0.07570848166942597,
      "epoch": 2.0532608032226564e-05,
      "step": 21530
    },
    {
      "action_loss_avg": 0.002197338273981586,
      "epoch": 2.0532608032226564e-05,
      "step": 21530
    },
    {
      "epoch": 2.0532608032226564e-05,
      "step": 21530,
      "training_step_time": 8.726613283157349
    },
    {
      "epoch": 2.0533561706542968e-05,
      "model_forward_time": 2.86167573928833,
      "step": 21531
    },
    {
      "epoch": 2.0533561706542968e-05,
      "step": 21531,
      "training_step_time": 9.22578763961792
    },
    {
      "epoch": 2.0534515380859375e-05,
      "model_forward_time": 1.7573826313018799,
      "step": 21532
    },
    {
      "epoch": 2.0534515380859375e-05,
      "step": 21532,
      "training_step_time": 7.794963836669922
    },
    {
      "epoch": 2.0535469055175782e-05,
      "model_forward_time": 1.7506022453308105,
      "step": 21533
    },
    {
      "epoch": 2.0535469055175782e-05,
      "step": 21533,
      "training_step_time": 8.492914915084839
    },
    {
      "epoch": 2.0536422729492186e-05,
      "model_forward_time": 1.749645471572876,
      "step": 21534
    },
    {
      "epoch": 2.0536422729492186e-05,
      "step": 21534,
      "training_step_time": 10.064566612243652
    },
    {
      "epoch": 2.0537376403808594e-05,
      "model_forward_time": 1.7396633625030518,
      "step": 21535
    },
    {
      "epoch": 2.0537376403808594e-05,
      "step": 21535,
      "training_step_time": 7.7232654094696045
    },
    {
      "epoch": 2.0538330078125e-05,
      "model_forward_time": 1.7497565746307373,
      "step": 21536
    },
    {
      "epoch": 2.0538330078125e-05,
      "step": 21536,
      "training_step_time": 7.7158825397491455
    },
    {
      "epoch": 2.0539283752441405e-05,
      "model_forward_time": 1.747147560119629,
      "step": 21537
    },
    {
      "epoch": 2.0539283752441405e-05,
      "step": 21537,
      "training_step_time": 7.790959596633911
    },
    {
      "epoch": 2.0540237426757813e-05,
      "model_forward_time": 1.740060806274414,
      "step": 21538
    },
    {
      "epoch": 2.0540237426757813e-05,
      "step": 21538,
      "training_step_time": 8.539845705032349
    },
    {
      "epoch": 2.054119110107422e-05,
      "model_forward_time": 1.7451510429382324,
      "step": 21539
    },
    {
      "epoch": 2.054119110107422e-05,
      "step": 21539,
      "training_step_time": 7.935770034790039
    },
    {
      "epoch": 2.0542144775390624e-05,
      "grad_norm": 0.21952134370803833,
      "learning_rate": 9.270613417269287e-06,
      "loss": 0.0643,
      "step": 21540
    },
    {
      "epoch": 2.0542144775390624e-05,
      "model_forward_time": 2.841498851776123,
      "step": 21540
    },
    {
      "dynamics_loss_avg": 0.06016920953989029,
      "epoch": 2.0542144775390624e-05,
      "step": 21540
    },
    {
      "action_loss_avg": 0.002416699915193021,
      "epoch": 2.0542144775390624e-05,
      "step": 21540
    },
    {
      "epoch": 2.0542144775390624e-05,
      "step": 21540,
      "training_step_time": 8.574347257614136
    },
    {
      "epoch": 2.054309844970703e-05,
      "model_forward_time": 1.7483882904052734,
      "step": 21541
    },
    {
      "epoch": 2.054309844970703e-05,
      "step": 21541,
      "training_step_time": 8.279627561569214
    },
    {
      "epoch": 2.054405212402344e-05,
      "model_forward_time": 1.8360576629638672,
      "step": 21542
    },
    {
      "epoch": 2.054405212402344e-05,
      "step": 21542,
      "training_step_time": 9.394001007080078
    },
    {
      "epoch": 2.0545005798339843e-05,
      "model_forward_time": 1.7456817626953125,
      "step": 21543
    },
    {
      "epoch": 2.0545005798339843e-05,
      "step": 21543,
      "training_step_time": 9.648276090621948
    },
    {
      "epoch": 2.054595947265625e-05,
      "model_forward_time": 1.7333827018737793,
      "step": 21544
    },
    {
      "epoch": 2.054595947265625e-05,
      "step": 21544,
      "training_step_time": 7.715589284896851
    },
    {
      "epoch": 2.0546913146972658e-05,
      "model_forward_time": 1.7436115741729736,
      "step": 21545
    },
    {
      "epoch": 2.0546913146972658e-05,
      "step": 21545,
      "training_step_time": 7.670178413391113
    },
    {
      "epoch": 2.0547866821289062e-05,
      "model_forward_time": 1.7479057312011719,
      "step": 21546
    },
    {
      "epoch": 2.0547866821289062e-05,
      "step": 21546,
      "training_step_time": 7.698436498641968
    },
    {
      "epoch": 2.054882049560547e-05,
      "model_forward_time": 1.7396657466888428,
      "step": 21547
    },
    {
      "epoch": 2.054882049560547e-05,
      "step": 21547,
      "training_step_time": 7.678283214569092
    },
    {
      "epoch": 2.0549774169921873e-05,
      "model_forward_time": 1.7402827739715576,
      "step": 21548
    },
    {
      "epoch": 2.0549774169921873e-05,
      "step": 21548,
      "training_step_time": 8.242159366607666
    },
    {
      "epoch": 2.055072784423828e-05,
      "model_forward_time": 2.8577351570129395,
      "step": 21549
    },
    {
      "epoch": 2.055072784423828e-05,
      "step": 21549,
      "training_step_time": 8.289649724960327
    },
    {
      "epoch": 2.0551681518554688e-05,
      "grad_norm": 0.1740582138299942,
      "learning_rate": 9.269753261121139e-06,
      "loss": 0.063,
      "step": 21550
    },
    {
      "epoch": 2.0551681518554688e-05,
      "model_forward_time": 1.733698844909668,
      "step": 21550
    },
    {
      "dynamics_loss_avg": 0.06331703290343285,
      "epoch": 2.0551681518554688e-05,
      "step": 21550
    },
    {
      "action_loss_avg": 0.002107118279673159,
      "epoch": 2.0551681518554688e-05,
      "step": 21550
    },
    {
      "epoch": 2.0551681518554688e-05,
      "step": 21550,
      "training_step_time": 7.828344345092773
    },
    {
      "epoch": 2.0552635192871092e-05,
      "model_forward_time": 1.8533077239990234,
      "step": 21551
    },
    {
      "epoch": 2.0552635192871092e-05,
      "step": 21551,
      "training_step_time": 8.292771816253662
    },
    {
      "epoch": 2.05535888671875e-05,
      "model_forward_time": 1.7450811862945557,
      "step": 21552
    },
    {
      "epoch": 2.05535888671875e-05,
      "step": 21552,
      "training_step_time": 8.61473274230957
    },
    {
      "epoch": 2.0554542541503907e-05,
      "model_forward_time": 1.744180679321289,
      "step": 21553
    },
    {
      "epoch": 2.0554542541503907e-05,
      "step": 21553,
      "training_step_time": 8.342565298080444
    },
    {
      "epoch": 2.055549621582031e-05,
      "model_forward_time": 1.742539644241333,
      "step": 21554
    },
    {
      "epoch": 2.055549621582031e-05,
      "step": 21554,
      "training_step_time": 8.171686887741089
    },
    {
      "epoch": 2.055644989013672e-05,
      "model_forward_time": 1.7348804473876953,
      "step": 21555
    },
    {
      "epoch": 2.055644989013672e-05,
      "step": 21555,
      "training_step_time": 7.768672704696655
    },
    {
      "epoch": 2.0557403564453126e-05,
      "model_forward_time": 1.7499516010284424,
      "step": 21556
    },
    {
      "epoch": 2.0557403564453126e-05,
      "step": 21556,
      "training_step_time": 8.282665491104126
    },
    {
      "epoch": 2.055835723876953e-05,
      "model_forward_time": 1.7388319969177246,
      "step": 21557
    },
    {
      "epoch": 2.055835723876953e-05,
      "step": 21557,
      "training_step_time": 7.6064372062683105
    },
    {
      "epoch": 2.0559310913085937e-05,
      "model_forward_time": 1.7512626647949219,
      "step": 21558
    },
    {
      "epoch": 2.0559310913085937e-05,
      "step": 21558,
      "training_step_time": 8.153950452804565
    },
    {
      "epoch": 2.0560264587402345e-05,
      "model_forward_time": 1.7445104122161865,
      "step": 21559
    },
    {
      "epoch": 2.0560264587402345e-05,
      "step": 21559,
      "training_step_time": 7.594897270202637
    },
    {
      "epoch": 2.056121826171875e-05,
      "grad_norm": 0.19353823363780975,
      "learning_rate": 9.268892638039196e-06,
      "loss": 0.0657,
      "step": 21560
    },
    {
      "epoch": 2.056121826171875e-05,
      "model_forward_time": 1.8130114078521729,
      "step": 21560
    },
    {
      "dynamics_loss_avg": 0.04973573051393032,
      "epoch": 2.056121826171875e-05,
      "step": 21560
    },
    {
      "action_loss_avg": 0.002169448387576267,
      "epoch": 2.056121826171875e-05,
      "step": 21560
    },
    {
      "epoch": 2.056121826171875e-05,
      "step": 21560,
      "training_step_time": 9.49321985244751
    },
    {
      "epoch": 2.0562171936035156e-05,
      "model_forward_time": 1.7402946949005127,
      "step": 21561
    },
    {
      "epoch": 2.0562171936035156e-05,
      "step": 21561,
      "training_step_time": 7.929046154022217
    },
    {
      "epoch": 2.0563125610351564e-05,
      "model_forward_time": 1.7406954765319824,
      "step": 21562
    },
    {
      "epoch": 2.0563125610351564e-05,
      "step": 21562,
      "training_step_time": 8.678123950958252
    },
    {
      "epoch": 2.0564079284667968e-05,
      "model_forward_time": 1.7358083724975586,
      "step": 21563
    },
    {
      "epoch": 2.0564079284667968e-05,
      "step": 21563,
      "training_step_time": 7.639558553695679
    },
    {
      "epoch": 2.0565032958984375e-05,
      "model_forward_time": 1.7491569519042969,
      "step": 21564
    },
    {
      "epoch": 2.0565032958984375e-05,
      "step": 21564,
      "training_step_time": 7.832395076751709
    },
    {
      "epoch": 2.0565986633300782e-05,
      "model_forward_time": 1.7422089576721191,
      "step": 21565
    },
    {
      "epoch": 2.0565986633300782e-05,
      "step": 21565,
      "training_step_time": 7.693511724472046
    },
    {
      "epoch": 2.0566940307617186e-05,
      "model_forward_time": 1.7472529411315918,
      "step": 21566
    },
    {
      "epoch": 2.0566940307617186e-05,
      "step": 21566,
      "training_step_time": 7.592057704925537
    },
    {
      "epoch": 2.0567893981933594e-05,
      "model_forward_time": 1.7374813556671143,
      "step": 21567
    },
    {
      "epoch": 2.0567893981933594e-05,
      "step": 21567,
      "training_step_time": 7.63382625579834
    },
    {
      "epoch": 2.056884765625e-05,
      "model_forward_time": 2.891934871673584,
      "step": 21568
    },
    {
      "epoch": 2.056884765625e-05,
      "step": 21568,
      "training_step_time": 7.731056213378906
    },
    {
      "epoch": 2.0569801330566405e-05,
      "model_forward_time": 1.7987134456634521,
      "step": 21569
    },
    {
      "epoch": 2.0569801330566405e-05,
      "step": 21569,
      "training_step_time": 8.56530213356018
    },
    {
      "epoch": 2.0570755004882813e-05,
      "grad_norm": 0.14467641711235046,
      "learning_rate": 9.26803154811757e-06,
      "loss": 0.0623,
      "step": 21570
    },
    {
      "epoch": 2.0570755004882813e-05,
      "model_forward_time": 1.7392492294311523,
      "step": 21570
    },
    {
      "dynamics_loss_avg": 0.07095653750002384,
      "epoch": 2.0570755004882813e-05,
      "step": 21570
    },
    {
      "action_loss_avg": 0.0034978450858034194,
      "epoch": 2.0570755004882813e-05,
      "step": 21570
    },
    {
      "epoch": 2.0570755004882813e-05,
      "step": 21570,
      "training_step_time": 7.666534662246704
    },
    {
      "epoch": 2.057170867919922e-05,
      "model_forward_time": 1.7304377555847168,
      "step": 21571
    },
    {
      "epoch": 2.057170867919922e-05,
      "step": 21571,
      "training_step_time": 7.8856236934661865
    },
    {
      "epoch": 2.0572662353515624e-05,
      "model_forward_time": 1.7643449306488037,
      "step": 21572
    },
    {
      "epoch": 2.0572662353515624e-05,
      "step": 21572,
      "training_step_time": 8.371593713760376
    },
    {
      "epoch": 2.057361602783203e-05,
      "model_forward_time": 1.797421932220459,
      "step": 21573
    },
    {
      "epoch": 2.057361602783203e-05,
      "step": 21573,
      "training_step_time": 7.890200138092041
    },
    {
      "epoch": 2.057456970214844e-05,
      "model_forward_time": 1.732426404953003,
      "step": 21574
    },
    {
      "epoch": 2.057456970214844e-05,
      "step": 21574,
      "training_step_time": 7.705317735671997
    },
    {
      "epoch": 2.0575523376464843e-05,
      "model_forward_time": 1.7417793273925781,
      "step": 21575
    },
    {
      "epoch": 2.0575523376464843e-05,
      "step": 21575,
      "training_step_time": 7.672860860824585
    },
    {
      "epoch": 2.057647705078125e-05,
      "model_forward_time": 1.7990434169769287,
      "step": 21576
    },
    {
      "epoch": 2.057647705078125e-05,
      "step": 21576,
      "training_step_time": 7.755092620849609
    },
    {
      "epoch": 2.0577430725097658e-05,
      "model_forward_time": 2.901454448699951,
      "step": 21577
    },
    {
      "epoch": 2.0577430725097658e-05,
      "step": 21577,
      "training_step_time": 9.342841625213623
    },
    {
      "epoch": 2.0578384399414062e-05,
      "model_forward_time": 1.7407655715942383,
      "step": 21578
    },
    {
      "epoch": 2.0578384399414062e-05,
      "step": 21578,
      "training_step_time": 7.6985743045806885
    },
    {
      "epoch": 2.057933807373047e-05,
      "model_forward_time": 1.8205757141113281,
      "step": 21579
    },
    {
      "epoch": 2.057933807373047e-05,
      "step": 21579,
      "training_step_time": 7.676440477371216
    },
    {
      "epoch": 2.0580291748046873e-05,
      "grad_norm": 0.1218620017170906,
      "learning_rate": 9.26716999145043e-06,
      "loss": 0.0624,
      "step": 21580
    },
    {
      "epoch": 2.0580291748046873e-05,
      "model_forward_time": 1.7326319217681885,
      "step": 21580
    },
    {
      "dynamics_loss_avg": 0.061845078878104684,
      "epoch": 2.0580291748046873e-05,
      "step": 21580
    },
    {
      "action_loss_avg": 0.0043133387574926015,
      "epoch": 2.0580291748046873e-05,
      "step": 21580
    },
    {
      "epoch": 2.0580291748046873e-05,
      "step": 21580,
      "training_step_time": 8.452130556106567
    },
    {
      "epoch": 2.058124542236328e-05,
      "model_forward_time": 1.7359559535980225,
      "step": 21581
    },
    {
      "epoch": 2.058124542236328e-05,
      "step": 21581,
      "training_step_time": 7.791678190231323
    },
    {
      "epoch": 2.0582199096679688e-05,
      "model_forward_time": 1.7711286544799805,
      "step": 21582
    },
    {
      "epoch": 2.0582199096679688e-05,
      "step": 21582,
      "training_step_time": 8.405805110931396
    },
    {
      "epoch": 2.0583152770996092e-05,
      "model_forward_time": 1.7443804740905762,
      "step": 21583
    },
    {
      "epoch": 2.0583152770996092e-05,
      "step": 21583,
      "training_step_time": 7.630310773849487
    },
    {
      "epoch": 2.05841064453125e-05,
      "model_forward_time": 1.745558261871338,
      "step": 21584
    },
    {
      "epoch": 2.05841064453125e-05,
      "step": 21584,
      "training_step_time": 8.58367919921875
    },
    {
      "epoch": 2.0585060119628907e-05,
      "model_forward_time": 1.7444727420806885,
      "step": 21585
    },
    {
      "epoch": 2.0585060119628907e-05,
      "step": 21585,
      "training_step_time": 8.301652908325195
    },
    {
      "epoch": 2.058601379394531e-05,
      "model_forward_time": 2.8844094276428223,
      "step": 21586
    },
    {
      "epoch": 2.058601379394531e-05,
      "step": 21586,
      "training_step_time": 8.352901220321655
    },
    {
      "epoch": 2.058696746826172e-05,
      "model_forward_time": 1.7481269836425781,
      "step": 21587
    },
    {
      "epoch": 2.058696746826172e-05,
      "step": 21587,
      "training_step_time": 8.735052824020386
    },
    {
      "epoch": 2.0587921142578126e-05,
      "model_forward_time": 1.775730848312378,
      "step": 21588
    },
    {
      "epoch": 2.0587921142578126e-05,
      "step": 21588,
      "training_step_time": 7.657484531402588
    },
    {
      "epoch": 2.058887481689453e-05,
      "model_forward_time": 1.7488822937011719,
      "step": 21589
    },
    {
      "epoch": 2.058887481689453e-05,
      "step": 21589,
      "training_step_time": 8.69373345375061
    },
    {
      "epoch": 2.0589828491210937e-05,
      "grad_norm": 0.20666833221912384,
      "learning_rate": 9.266307968131998e-06,
      "loss": 0.0657,
      "step": 21590
    },
    {
      "epoch": 2.0589828491210937e-05,
      "model_forward_time": 1.7485675811767578,
      "step": 21590
    },
    {
      "dynamics_loss_avg": 0.06262334864586591,
      "epoch": 2.0589828491210937e-05,
      "step": 21590
    },
    {
      "action_loss_avg": 0.0021672270027920603,
      "epoch": 2.0589828491210937e-05,
      "step": 21590
    },
    {
      "epoch": 2.0589828491210937e-05,
      "step": 21590,
      "training_step_time": 7.640916109085083
    },
    {
      "epoch": 2.0590782165527345e-05,
      "model_forward_time": 1.7355427742004395,
      "step": 21591
    },
    {
      "epoch": 2.0590782165527345e-05,
      "step": 21591,
      "training_step_time": 7.668102264404297
    },
    {
      "epoch": 2.059173583984375e-05,
      "model_forward_time": 1.7461659908294678,
      "step": 21592
    },
    {
      "epoch": 2.059173583984375e-05,
      "step": 21592,
      "training_step_time": 8.101746082305908
    },
    {
      "epoch": 2.0592689514160156e-05,
      "model_forward_time": 1.742347002029419,
      "step": 21593
    },
    {
      "epoch": 2.0592689514160156e-05,
      "step": 21593,
      "training_step_time": 7.549170970916748
    },
    {
      "epoch": 2.0593643188476564e-05,
      "model_forward_time": 1.7492504119873047,
      "step": 21594
    },
    {
      "epoch": 2.0593643188476564e-05,
      "step": 21594,
      "training_step_time": 8.294963121414185
    },
    {
      "epoch": 2.0594596862792968e-05,
      "model_forward_time": 2.8628833293914795,
      "step": 21595
    },
    {
      "epoch": 2.0594596862792968e-05,
      "step": 21595,
      "training_step_time": 8.256969690322876
    },
    {
      "epoch": 2.0595550537109375e-05,
      "model_forward_time": 1.7432875633239746,
      "step": 21596
    },
    {
      "epoch": 2.0595550537109375e-05,
      "step": 21596,
      "training_step_time": 8.329080581665039
    },
    {
      "epoch": 2.0596504211425782e-05,
      "model_forward_time": 1.7446825504302979,
      "step": 21597
    },
    {
      "epoch": 2.0596504211425782e-05,
      "step": 21597,
      "training_step_time": 7.557864189147949
    },
    {
      "epoch": 2.0597457885742187e-05,
      "model_forward_time": 1.839139461517334,
      "step": 21598
    },
    {
      "epoch": 2.0597457885742187e-05,
      "step": 21598,
      "training_step_time": 7.690781831741333
    },
    {
      "epoch": 2.0598411560058594e-05,
      "model_forward_time": 1.7636418342590332,
      "step": 21599
    },
    {
      "epoch": 2.0598411560058594e-05,
      "step": 21599,
      "training_step_time": 8.627701759338379
    },
    {
      "epoch": 2.0599365234375e-05,
      "grad_norm": 0.10335170477628708,
      "learning_rate": 9.265445478256541e-06,
      "loss": 0.0592,
      "step": 21600
    },
    {
      "epoch": 2.0599365234375e-05,
      "model_forward_time": 1.7272038459777832,
      "step": 21600
    },
    {
      "dynamics_loss_avg": 0.051347232423722744,
      "epoch": 2.0599365234375e-05,
      "step": 21600
    },
    {
      "action_loss_avg": 0.0020624561002478004,
      "epoch": 2.0599365234375e-05,
      "step": 21600
    },
    {
      "epoch": 2.0599365234375e-05,
      "step": 21600,
      "training_step_time": 7.78813624382019
    },
    {
      "epoch": 2.0600318908691405e-05,
      "model_forward_time": 1.7401580810546875,
      "step": 21601
    },
    {
      "epoch": 2.0600318908691405e-05,
      "step": 21601,
      "training_step_time": 7.566838026046753
    },
    {
      "epoch": 2.0601272583007813e-05,
      "model_forward_time": 1.3665258884429932,
      "step": 21602
    },
    {
      "epoch": 2.0601272583007813e-05,
      "step": 21602,
      "training_step_time": 7.702680349349976
    },
    {
      "epoch": 2.060222625732422e-05,
      "model_forward_time": 1.7327499389648438,
      "step": 21603
    },
    {
      "epoch": 2.060222625732422e-05,
      "step": 21603,
      "training_step_time": 7.573169231414795
    },
    {
      "epoch": 2.0603179931640624e-05,
      "model_forward_time": 1.7416636943817139,
      "step": 21604
    },
    {
      "epoch": 2.0603179931640624e-05,
      "step": 21604,
      "training_step_time": 7.596369981765747
    },
    {
      "epoch": 2.060413360595703e-05,
      "model_forward_time": 2.8832998275756836,
      "step": 21605
    },
    {
      "epoch": 2.060413360595703e-05,
      "step": 21605,
      "training_step_time": 7.69951868057251
    },
    {
      "epoch": 2.060508728027344e-05,
      "model_forward_time": 1.7366557121276855,
      "step": 21606
    },
    {
      "epoch": 2.060508728027344e-05,
      "step": 21606,
      "training_step_time": 8.22850751876831
    },
    {
      "epoch": 2.0606040954589843e-05,
      "model_forward_time": 1.8701322078704834,
      "step": 21607
    },
    {
      "epoch": 2.0606040954589843e-05,
      "step": 21607,
      "training_step_time": 7.654280662536621
    },
    {
      "epoch": 2.060699462890625e-05,
      "model_forward_time": 1.7534198760986328,
      "step": 21608
    },
    {
      "epoch": 2.060699462890625e-05,
      "step": 21608,
      "training_step_time": 8.187266111373901
    },
    {
      "epoch": 2.0607948303222658e-05,
      "model_forward_time": 1.7540547847747803,
      "step": 21609
    },
    {
      "epoch": 2.0607948303222658e-05,
      "step": 21609,
      "training_step_time": 7.745725393295288
    },
    {
      "epoch": 2.0608901977539062e-05,
      "grad_norm": 0.23521658778190613,
      "learning_rate": 9.264582521918377e-06,
      "loss": 0.0624,
      "step": 21610
    },
    {
      "epoch": 2.0608901977539062e-05,
      "model_forward_time": 1.7544341087341309,
      "step": 21610
    },
    {
      "dynamics_loss_avg": 0.05039238166064024,
      "epoch": 2.0608901977539062e-05,
      "step": 21610
    },
    {
      "action_loss_avg": 0.002602732297964394,
      "epoch": 2.0608901977539062e-05,
      "step": 21610
    },
    {
      "epoch": 2.0608901977539062e-05,
      "step": 21610,
      "training_step_time": 7.529590606689453
    },
    {
      "epoch": 2.060985565185547e-05,
      "model_forward_time": 1.7523655891418457,
      "step": 21611
    },
    {
      "epoch": 2.060985565185547e-05,
      "step": 21611,
      "training_step_time": 7.609580993652344
    },
    {
      "epoch": 2.0610809326171873e-05,
      "model_forward_time": 1.7617347240447998,
      "step": 21612
    },
    {
      "epoch": 2.0610809326171873e-05,
      "step": 21612,
      "training_step_time": 8.032147645950317
    },
    {
      "epoch": 2.061176300048828e-05,
      "model_forward_time": 1.7524895668029785,
      "step": 21613
    },
    {
      "epoch": 2.061176300048828e-05,
      "step": 21613,
      "training_step_time": 7.568554401397705
    },
    {
      "epoch": 2.0612716674804688e-05,
      "model_forward_time": 2.9481804370880127,
      "step": 21614
    },
    {
      "epoch": 2.0612716674804688e-05,
      "step": 21614,
      "training_step_time": 8.490505456924438
    },
    {
      "epoch": 2.0613670349121092e-05,
      "model_forward_time": 1.7576112747192383,
      "step": 21615
    },
    {
      "epoch": 2.0613670349121092e-05,
      "step": 21615,
      "training_step_time": 7.6724138259887695
    },
    {
      "epoch": 2.06146240234375e-05,
      "model_forward_time": 1.8421554565429688,
      "step": 21616
    },
    {
      "epoch": 2.06146240234375e-05,
      "step": 21616,
      "training_step_time": 7.690709829330444
    },
    {
      "epoch": 2.0615577697753907e-05,
      "model_forward_time": 1.7494237422943115,
      "step": 21617
    },
    {
      "epoch": 2.0615577697753907e-05,
      "step": 21617,
      "training_step_time": 7.654742956161499
    },
    {
      "epoch": 2.061653137207031e-05,
      "model_forward_time": 1.7409470081329346,
      "step": 21618
    },
    {
      "epoch": 2.061653137207031e-05,
      "step": 21618,
      "training_step_time": 7.563918352127075
    },
    {
      "epoch": 2.061748504638672e-05,
      "model_forward_time": 1.7560927867889404,
      "step": 21619
    },
    {
      "epoch": 2.061748504638672e-05,
      "step": 21619,
      "training_step_time": 7.8928046226501465
    },
    {
      "epoch": 2.0618438720703126e-05,
      "grad_norm": 0.21451888978481293,
      "learning_rate": 9.263719099211881e-06,
      "loss": 0.0604,
      "step": 21620
    },
    {
      "epoch": 2.0618438720703126e-05,
      "model_forward_time": 1.7750425338745117,
      "step": 21620
    },
    {
      "dynamics_loss_avg": 0.06400119420140982,
      "epoch": 2.0618438720703126e-05,
      "step": 21620
    },
    {
      "action_loss_avg": 0.002727327769389376,
      "epoch": 2.0618438720703126e-05,
      "step": 21620
    },
    {
      "epoch": 2.0618438720703126e-05,
      "step": 21620,
      "training_step_time": 7.64450216293335
    },
    {
      "epoch": 2.061939239501953e-05,
      "model_forward_time": 1.7569239139556885,
      "step": 21621
    },
    {
      "epoch": 2.061939239501953e-05,
      "step": 21621,
      "training_step_time": 7.538045406341553
    },
    {
      "epoch": 2.0620346069335937e-05,
      "model_forward_time": 1.7567062377929688,
      "step": 21622
    },
    {
      "epoch": 2.0620346069335937e-05,
      "step": 21622,
      "training_step_time": 7.5776145458221436
    },
    {
      "epoch": 2.0621299743652345e-05,
      "model_forward_time": 2.9329593181610107,
      "step": 21623
    },
    {
      "epoch": 2.0621299743652345e-05,
      "step": 21623,
      "training_step_time": 8.249567985534668
    },
    {
      "epoch": 2.062225341796875e-05,
      "model_forward_time": 1.7548823356628418,
      "step": 21624
    },
    {
      "epoch": 2.062225341796875e-05,
      "step": 21624,
      "training_step_time": 8.518908023834229
    },
    {
      "epoch": 2.0623207092285156e-05,
      "model_forward_time": 1.7879881858825684,
      "step": 21625
    },
    {
      "epoch": 2.0623207092285156e-05,
      "step": 21625,
      "training_step_time": 8.470043897628784
    },
    {
      "epoch": 2.0624160766601564e-05,
      "model_forward_time": 1.7452304363250732,
      "step": 21626
    },
    {
      "epoch": 2.0624160766601564e-05,
      "step": 21626,
      "training_step_time": 8.252564430236816
    },
    {
      "epoch": 2.0625114440917968e-05,
      "model_forward_time": 1.7486369609832764,
      "step": 21627
    },
    {
      "epoch": 2.0625114440917968e-05,
      "step": 21627,
      "training_step_time": 7.765326261520386
    },
    {
      "epoch": 2.0626068115234375e-05,
      "model_forward_time": 1.7513489723205566,
      "step": 21628
    },
    {
      "epoch": 2.0626068115234375e-05,
      "step": 21628,
      "training_step_time": 7.7682788372039795
    },
    {
      "epoch": 2.0627021789550783e-05,
      "model_forward_time": 1.7481746673583984,
      "step": 21629
    },
    {
      "epoch": 2.0627021789550783e-05,
      "step": 21629,
      "training_step_time": 7.599821329116821
    },
    {
      "epoch": 2.0627975463867187e-05,
      "grad_norm": 0.21462266147136688,
      "learning_rate": 9.262855210231476e-06,
      "loss": 0.0653,
      "step": 21630
    },
    {
      "epoch": 2.0627975463867187e-05,
      "model_forward_time": 1.7595665454864502,
      "step": 21630
    },
    {
      "dynamics_loss_avg": 0.06402587182819844,
      "epoch": 2.0627975463867187e-05,
      "step": 21630
    },
    {
      "action_loss_avg": 0.0032986870501190424,
      "epoch": 2.0627975463867187e-05,
      "step": 21630
    },
    {
      "epoch": 2.0627975463867187e-05,
      "step": 21630,
      "training_step_time": 7.539246082305908
    },
    {
      "epoch": 2.0628929138183594e-05,
      "model_forward_time": 1.7398488521575928,
      "step": 21631
    },
    {
      "epoch": 2.0628929138183594e-05,
      "step": 21631,
      "training_step_time": 7.563941240310669
    },
    {
      "epoch": 2.06298828125e-05,
      "model_forward_time": 2.891256093978882,
      "step": 21632
    },
    {
      "epoch": 2.06298828125e-05,
      "step": 21632,
      "training_step_time": 8.624109983444214
    },
    {
      "epoch": 2.0630836486816405e-05,
      "model_forward_time": 1.7534122467041016,
      "step": 21633
    },
    {
      "epoch": 2.0630836486816405e-05,
      "step": 21633,
      "training_step_time": 9.40676736831665
    },
    {
      "epoch": 2.0631790161132813e-05,
      "model_forward_time": 1.7438709735870361,
      "step": 21634
    },
    {
      "epoch": 2.0631790161132813e-05,
      "step": 21634,
      "training_step_time": 8.842173337936401
    },
    {
      "epoch": 2.063274383544922e-05,
      "model_forward_time": 1.7463653087615967,
      "step": 21635
    },
    {
      "epoch": 2.063274383544922e-05,
      "step": 21635,
      "training_step_time": 8.31466031074524
    },
    {
      "epoch": 2.0633697509765624e-05,
      "model_forward_time": 1.7434637546539307,
      "step": 21636
    },
    {
      "epoch": 2.0633697509765624e-05,
      "step": 21636,
      "training_step_time": 7.647798299789429
    },
    {
      "epoch": 2.063465118408203e-05,
      "model_forward_time": 1.7498748302459717,
      "step": 21637
    },
    {
      "epoch": 2.063465118408203e-05,
      "step": 21637,
      "training_step_time": 7.760151386260986
    },
    {
      "epoch": 2.063560485839844e-05,
      "model_forward_time": 1.7457215785980225,
      "step": 21638
    },
    {
      "epoch": 2.063560485839844e-05,
      "step": 21638,
      "training_step_time": 8.144462823867798
    },
    {
      "epoch": 2.0636558532714843e-05,
      "model_forward_time": 1.7524495124816895,
      "step": 21639
    },
    {
      "epoch": 2.0636558532714843e-05,
      "step": 21639,
      "training_step_time": 7.5203845500946045
    },
    {
      "epoch": 2.063751220703125e-05,
      "grad_norm": 0.23502953350543976,
      "learning_rate": 9.261990855071634e-06,
      "loss": 0.0642,
      "step": 21640
    },
    {
      "epoch": 2.063751220703125e-05,
      "model_forward_time": 1.7463335990905762,
      "step": 21640
    },
    {
      "dynamics_loss_avg": 0.059643729217350484,
      "epoch": 2.063751220703125e-05,
      "step": 21640
    },
    {
      "action_loss_avg": 0.004254434973699972,
      "epoch": 2.063751220703125e-05,
      "step": 21640
    },
    {
      "epoch": 2.063751220703125e-05,
      "step": 21640,
      "training_step_time": 7.533517599105835
    },
    {
      "epoch": 2.0638465881347655e-05,
      "model_forward_time": 2.8326759338378906,
      "step": 21641
    },
    {
      "epoch": 2.0638465881347655e-05,
      "step": 21641,
      "training_step_time": 8.971654653549194
    },
    {
      "epoch": 2.0639419555664062e-05,
      "model_forward_time": 1.7400853633880615,
      "step": 21642
    },
    {
      "epoch": 2.0639419555664062e-05,
      "step": 21642,
      "training_step_time": 7.702265739440918
    },
    {
      "epoch": 2.064037322998047e-05,
      "model_forward_time": 1.742459774017334,
      "step": 21643
    },
    {
      "epoch": 2.064037322998047e-05,
      "step": 21643,
      "training_step_time": 7.851283073425293
    },
    {
      "epoch": 2.0641326904296873e-05,
      "model_forward_time": 1.863908290863037,
      "step": 21644
    },
    {
      "epoch": 2.0641326904296873e-05,
      "step": 21644,
      "training_step_time": 8.62007451057434
    },
    {
      "epoch": 2.064228057861328e-05,
      "model_forward_time": 1.7521469593048096,
      "step": 21645
    },
    {
      "epoch": 2.064228057861328e-05,
      "step": 21645,
      "training_step_time": 7.661986589431763
    },
    {
      "epoch": 2.0643234252929688e-05,
      "model_forward_time": 1.7444334030151367,
      "step": 21646
    },
    {
      "epoch": 2.0643234252929688e-05,
      "step": 21646,
      "training_step_time": 8.153892278671265
    },
    {
      "epoch": 2.0644187927246092e-05,
      "model_forward_time": 1.7505004405975342,
      "step": 21647
    },
    {
      "epoch": 2.0644187927246092e-05,
      "step": 21647,
      "training_step_time": 7.710005044937134
    },
    {
      "epoch": 2.06451416015625e-05,
      "model_forward_time": 1.7460484504699707,
      "step": 21648
    },
    {
      "epoch": 2.06451416015625e-05,
      "step": 21648,
      "training_step_time": 7.71742057800293
    },
    {
      "epoch": 2.0646095275878907e-05,
      "model_forward_time": 1.75075364112854,
      "step": 21649
    },
    {
      "epoch": 2.0646095275878907e-05,
      "step": 21649,
      "training_step_time": 8.510002613067627
    },
    {
      "epoch": 2.064704895019531e-05,
      "grad_norm": 0.18480589985847473,
      "learning_rate": 9.261126033826879e-06,
      "loss": 0.0671,
      "step": 21650
    },
    {
      "epoch": 2.064704895019531e-05,
      "model_forward_time": 2.8207435607910156,
      "step": 21650
    },
    {
      "dynamics_loss_avg": 0.07091140951961279,
      "epoch": 2.064704895019531e-05,
      "step": 21650
    },
    {
      "action_loss_avg": 0.002971943700686097,
      "epoch": 2.064704895019531e-05,
      "step": 21650
    },
    {
      "epoch": 2.064704895019531e-05,
      "step": 21650,
      "training_step_time": 7.600012302398682
    },
    {
      "epoch": 2.064800262451172e-05,
      "model_forward_time": 1.7512030601501465,
      "step": 21651
    },
    {
      "epoch": 2.064800262451172e-05,
      "step": 21651,
      "training_step_time": 8.4087655544281
    },
    {
      "epoch": 2.0648956298828126e-05,
      "model_forward_time": 1.7506213188171387,
      "step": 21652
    },
    {
      "epoch": 2.0648956298828126e-05,
      "step": 21652,
      "training_step_time": 7.821521759033203
    },
    {
      "epoch": 2.064990997314453e-05,
      "model_forward_time": 1.1052494049072266,
      "step": 21653
    },
    {
      "epoch": 2.064990997314453e-05,
      "step": 21653,
      "training_step_time": 9.190904140472412
    },
    {
      "epoch": 2.0650863647460937e-05,
      "model_forward_time": 1.7582731246948242,
      "step": 21654
    },
    {
      "epoch": 2.0650863647460937e-05,
      "step": 21654,
      "training_step_time": 7.676452159881592
    },
    {
      "epoch": 2.0651817321777345e-05,
      "model_forward_time": 1.7434720993041992,
      "step": 21655
    },
    {
      "epoch": 2.0651817321777345e-05,
      "step": 21655,
      "training_step_time": 7.611067056655884
    },
    {
      "epoch": 2.065277099609375e-05,
      "model_forward_time": 1.743945598602295,
      "step": 21656
    },
    {
      "epoch": 2.065277099609375e-05,
      "step": 21656,
      "training_step_time": 7.712586402893066
    },
    {
      "epoch": 2.0653724670410156e-05,
      "model_forward_time": 1.3881347179412842,
      "step": 21657
    },
    {
      "epoch": 2.0653724670410156e-05,
      "step": 21657,
      "training_step_time": 7.625014543533325
    },
    {
      "epoch": 2.0654678344726564e-05,
      "model_forward_time": 1.739135503768921,
      "step": 21658
    },
    {
      "epoch": 2.0654678344726564e-05,
      "step": 21658,
      "training_step_time": 8.23742961883545
    },
    {
      "epoch": 2.0655632019042968e-05,
      "model_forward_time": 1.7528679370880127,
      "step": 21659
    },
    {
      "epoch": 2.0655632019042968e-05,
      "step": 21659,
      "training_step_time": 7.5564844608306885
    },
    {
      "epoch": 2.0656585693359375e-05,
      "grad_norm": 0.24686241149902344,
      "learning_rate": 9.260260746591788e-06,
      "loss": 0.0671,
      "step": 21660
    },
    {
      "epoch": 2.0656585693359375e-05,
      "model_forward_time": 2.857130289077759,
      "step": 21660
    },
    {
      "dynamics_loss_avg": 0.06542429127730429,
      "epoch": 2.0656585693359375e-05,
      "step": 21660
    },
    {
      "action_loss_avg": 0.0022995990177150814,
      "epoch": 2.0656585693359375e-05,
      "step": 21660
    },
    {
      "epoch": 2.0656585693359375e-05,
      "step": 21660,
      "training_step_time": 10.43058180809021
    },
    {
      "epoch": 2.0657539367675783e-05,
      "model_forward_time": 1.7581827640533447,
      "step": 21661
    },
    {
      "epoch": 2.0657539367675783e-05,
      "step": 21661,
      "training_step_time": 9.284689664840698
    },
    {
      "epoch": 2.0658493041992187e-05,
      "model_forward_time": 1.853332281112671,
      "step": 21662
    },
    {
      "epoch": 2.0658493041992187e-05,
      "step": 21662,
      "training_step_time": 8.574213981628418
    },
    {
      "epoch": 2.0659446716308594e-05,
      "model_forward_time": 1.7423286437988281,
      "step": 21663
    },
    {
      "epoch": 2.0659446716308594e-05,
      "step": 21663,
      "training_step_time": 7.61812424659729
    },
    {
      "epoch": 2.0660400390625e-05,
      "model_forward_time": 1.7438297271728516,
      "step": 21664
    },
    {
      "epoch": 2.0660400390625e-05,
      "step": 21664,
      "training_step_time": 8.142332077026367
    },
    {
      "epoch": 2.0661354064941405e-05,
      "model_forward_time": 1.7463123798370361,
      "step": 21665
    },
    {
      "epoch": 2.0661354064941405e-05,
      "step": 21665,
      "training_step_time": 7.789766788482666
    },
    {
      "epoch": 2.0662307739257813e-05,
      "model_forward_time": 1.7526345252990723,
      "step": 21666
    },
    {
      "epoch": 2.0662307739257813e-05,
      "step": 21666,
      "training_step_time": 7.615602970123291
    },
    {
      "epoch": 2.066326141357422e-05,
      "model_forward_time": 1.7525098323822021,
      "step": 21667
    },
    {
      "epoch": 2.066326141357422e-05,
      "step": 21667,
      "training_step_time": 7.557516813278198
    },
    {
      "epoch": 2.0664215087890624e-05,
      "model_forward_time": 1.7534496784210205,
      "step": 21668
    },
    {
      "epoch": 2.0664215087890624e-05,
      "step": 21668,
      "training_step_time": 7.6177263259887695
    },
    {
      "epoch": 2.0665168762207032e-05,
      "model_forward_time": 1.7517309188842773,
      "step": 21669
    },
    {
      "epoch": 2.0665168762207032e-05,
      "step": 21669,
      "training_step_time": 8.903884887695312
    },
    {
      "epoch": 2.066612243652344e-05,
      "grad_norm": 0.1378113478422165,
      "learning_rate": 9.259394993460986e-06,
      "loss": 0.0646,
      "step": 21670
    },
    {
      "epoch": 2.066612243652344e-05,
      "model_forward_time": 1.7625701427459717,
      "step": 21670
    },
    {
      "dynamics_loss_avg": 0.0550568244419992,
      "epoch": 2.066612243652344e-05,
      "step": 21670
    },
    {
      "action_loss_avg": 0.002639452970470302,
      "epoch": 2.066612243652344e-05,
      "step": 21670
    },
    {
      "epoch": 2.066612243652344e-05,
      "step": 21670,
      "training_step_time": 8.402965068817139
    },
    {
      "epoch": 2.0667076110839843e-05,
      "model_forward_time": 1.8511817455291748,
      "step": 21671
    },
    {
      "epoch": 2.0667076110839843e-05,
      "step": 21671,
      "training_step_time": 7.719548463821411
    },
    {
      "epoch": 2.066802978515625e-05,
      "model_forward_time": 1.7520496845245361,
      "step": 21672
    },
    {
      "epoch": 2.066802978515625e-05,
      "step": 21672,
      "training_step_time": 7.664482831954956
    },
    {
      "epoch": 2.0668983459472655e-05,
      "model_forward_time": 1.7546653747558594,
      "step": 21673
    },
    {
      "epoch": 2.0668983459472655e-05,
      "step": 21673,
      "training_step_time": 8.40467119216919
    },
    {
      "epoch": 2.0669937133789062e-05,
      "model_forward_time": 1.7481145858764648,
      "step": 21674
    },
    {
      "epoch": 2.0669937133789062e-05,
      "step": 21674,
      "training_step_time": 7.806879281997681
    },
    {
      "epoch": 2.067089080810547e-05,
      "model_forward_time": 1.7522592544555664,
      "step": 21675
    },
    {
      "epoch": 2.067089080810547e-05,
      "step": 21675,
      "training_step_time": 7.563143968582153
    },
    {
      "epoch": 2.0671844482421873e-05,
      "model_forward_time": 1.7634432315826416,
      "step": 21676
    },
    {
      "epoch": 2.0671844482421873e-05,
      "step": 21676,
      "training_step_time": 8.575941562652588
    },
    {
      "epoch": 2.067279815673828e-05,
      "model_forward_time": 1.755598783493042,
      "step": 21677
    },
    {
      "epoch": 2.067279815673828e-05,
      "step": 21677,
      "training_step_time": 7.605813980102539
    },
    {
      "epoch": 2.0673751831054688e-05,
      "model_forward_time": 1.7530829906463623,
      "step": 21678
    },
    {
      "epoch": 2.0673751831054688e-05,
      "step": 21678,
      "training_step_time": 10.551816940307617
    },
    {
      "epoch": 2.0674705505371092e-05,
      "model_forward_time": 1.7570486068725586,
      "step": 21679
    },
    {
      "epoch": 2.0674705505371092e-05,
      "step": 21679,
      "training_step_time": 8.800095796585083
    },
    {
      "epoch": 2.06756591796875e-05,
      "grad_norm": 0.2144889086484909,
      "learning_rate": 9.258528774529152e-06,
      "loss": 0.0672,
      "step": 21680
    },
    {
      "epoch": 2.06756591796875e-05,
      "model_forward_time": 1.7621331214904785,
      "step": 21680
    },
    {
      "dynamics_loss_avg": 0.06896258816123009,
      "epoch": 2.06756591796875e-05,
      "step": 21680
    },
    {
      "action_loss_avg": 0.00363301751203835,
      "epoch": 2.06756591796875e-05,
      "step": 21680
    },
    {
      "epoch": 2.06756591796875e-05,
      "step": 21680,
      "training_step_time": 6.85392165184021
    },
    {
      "epoch": 2.0676612854003907e-05,
      "model_forward_time": 1.756460189819336,
      "step": 21681
    },
    {
      "epoch": 2.0676612854003907e-05,
      "step": 21681,
      "training_step_time": 6.842233419418335
    },
    {
      "epoch": 2.067756652832031e-05,
      "model_forward_time": 1.7610070705413818,
      "step": 21682
    },
    {
      "epoch": 2.067756652832031e-05,
      "step": 21682,
      "training_step_time": 6.842225790023804
    },
    {
      "epoch": 2.067852020263672e-05,
      "model_forward_time": 1.751657247543335,
      "step": 21683
    },
    {
      "epoch": 2.067852020263672e-05,
      "step": 21683,
      "training_step_time": 6.799111604690552
    },
    {
      "epoch": 2.0679473876953126e-05,
      "model_forward_time": 1.7573349475860596,
      "step": 21684
    },
    {
      "epoch": 2.0679473876953126e-05,
      "step": 21684,
      "training_step_time": 6.824029922485352
    },
    {
      "epoch": 2.068042755126953e-05,
      "model_forward_time": 1.7515661716461182,
      "step": 21685
    },
    {
      "epoch": 2.068042755126953e-05,
      "step": 21685,
      "training_step_time": 6.828683614730835
    },
    {
      "epoch": 2.0681381225585937e-05,
      "model_forward_time": 1.3848001956939697,
      "step": 21686
    },
    {
      "epoch": 2.0681381225585937e-05,
      "step": 21686,
      "training_step_time": 6.802812099456787
    },
    {
      "epoch": 2.0682334899902345e-05,
      "model_forward_time": 1.7483692169189453,
      "step": 21687
    },
    {
      "epoch": 2.0682334899902345e-05,
      "step": 21687,
      "training_step_time": 6.776065349578857
    },
    {
      "epoch": 2.068328857421875e-05,
      "model_forward_time": 2.8632099628448486,
      "step": 21688
    },
    {
      "epoch": 2.068328857421875e-05,
      "step": 21688,
      "training_step_time": 7.708328485488892
    },
    {
      "epoch": 2.0684242248535156e-05,
      "model_forward_time": 1.753607988357544,
      "step": 21689
    },
    {
      "epoch": 2.0684242248535156e-05,
      "step": 21689,
      "training_step_time": 7.864784240722656
    },
    {
      "epoch": 2.0685195922851564e-05,
      "grad_norm": 0.2081804722547531,
      "learning_rate": 9.257662089891013e-06,
      "loss": 0.0642,
      "step": 21690
    },
    {
      "epoch": 2.0685195922851564e-05,
      "model_forward_time": 1.7410225868225098,
      "step": 21690
    },
    {
      "dynamics_loss_avg": 0.07179012093693019,
      "epoch": 2.0685195922851564e-05,
      "step": 21690
    },
    {
      "action_loss_avg": 0.0028537825273815542,
      "epoch": 2.0685195922851564e-05,
      "step": 21690
    },
    {
      "epoch": 2.0685195922851564e-05,
      "step": 21690,
      "training_step_time": 6.783605098724365
    },
    {
      "epoch": 2.0686149597167968e-05,
      "model_forward_time": 1.7542574405670166,
      "step": 21691
    },
    {
      "epoch": 2.0686149597167968e-05,
      "step": 21691,
      "training_step_time": 6.780726432800293
    },
    {
      "epoch": 2.0687103271484375e-05,
      "model_forward_time": 1.7484498023986816,
      "step": 21692
    },
    {
      "epoch": 2.0687103271484375e-05,
      "step": 21692,
      "training_step_time": 6.783074617385864
    },
    {
      "epoch": 2.0688056945800783e-05,
      "model_forward_time": 1.7566437721252441,
      "step": 21693
    },
    {
      "epoch": 2.0688056945800783e-05,
      "step": 21693,
      "training_step_time": 6.780993461608887
    },
    {
      "epoch": 2.0689010620117187e-05,
      "model_forward_time": 1.752870798110962,
      "step": 21694
    },
    {
      "epoch": 2.0689010620117187e-05,
      "step": 21694,
      "training_step_time": 6.675191640853882
    },
    {
      "epoch": 2.0689964294433594e-05,
      "model_forward_time": 1.753157615661621,
      "step": 21695
    },
    {
      "epoch": 2.0689964294433594e-05,
      "step": 21695,
      "training_step_time": 6.793348550796509
    },
    {
      "epoch": 2.069091796875e-05,
      "model_forward_time": 1.7505710124969482,
      "step": 21696
    },
    {
      "epoch": 2.069091796875e-05,
      "step": 21696,
      "training_step_time": 6.751346588134766
    },
    {
      "epoch": 2.0691871643066405e-05,
      "model_forward_time": 2.883197069168091,
      "step": 21697
    },
    {
      "epoch": 2.0691871643066405e-05,
      "step": 21697,
      "training_step_time": 7.658849477767944
    },
    {
      "epoch": 2.0692825317382813e-05,
      "model_forward_time": 1.7534592151641846,
      "step": 21698
    },
    {
      "epoch": 2.0692825317382813e-05,
      "step": 21698,
      "training_step_time": 7.787838935852051
    },
    {
      "epoch": 2.069377899169922e-05,
      "model_forward_time": 1.7505319118499756,
      "step": 21699
    },
    {
      "epoch": 2.069377899169922e-05,
      "step": 21699,
      "training_step_time": 6.786926746368408
    },
    {
      "epoch": 2.0694732666015624e-05,
      "grad_norm": 0.23248477280139923,
      "learning_rate": 9.256794939641351e-06,
      "loss": 0.0622,
      "step": 21700
    },
    {
      "epoch": 2.0694732666015624e-05,
      "model_forward_time": 1.3879916667938232,
      "step": 21700
    },
    {
      "dynamics_loss_avg": 0.07425237763673068,
      "epoch": 2.0694732666015624e-05,
      "step": 21700
    },
    {
      "action_loss_avg": 0.0031766533909831198,
      "epoch": 2.0694732666015624e-05,
      "step": 21700
    },
    {
      "epoch": 2.0694732666015624e-05,
      "step": 21700,
      "training_step_time": 6.76289963722229
    },
    {
      "epoch": 2.0695686340332032e-05,
      "model_forward_time": 1.760188341140747,
      "step": 21701
    },
    {
      "epoch": 2.0695686340332032e-05,
      "step": 21701,
      "training_step_time": 6.792387008666992
    },
    {
      "epoch": 2.069664001464844e-05,
      "model_forward_time": 1.7544562816619873,
      "step": 21702
    },
    {
      "epoch": 2.069664001464844e-05,
      "step": 21702,
      "training_step_time": 6.780404567718506
    },
    {
      "epoch": 2.0697593688964843e-05,
      "model_forward_time": 1.7451379299163818,
      "step": 21703
    },
    {
      "epoch": 2.0697593688964843e-05,
      "step": 21703,
      "training_step_time": 6.762699365615845
    },
    {
      "epoch": 2.069854736328125e-05,
      "model_forward_time": 1.7555780410766602,
      "step": 21704
    },
    {
      "epoch": 2.069854736328125e-05,
      "step": 21704,
      "training_step_time": 6.770538568496704
    },
    {
      "epoch": 2.0699501037597655e-05,
      "model_forward_time": 1.756061315536499,
      "step": 21705
    },
    {
      "epoch": 2.0699501037597655e-05,
      "step": 21705,
      "training_step_time": 6.786391258239746
    },
    {
      "epoch": 2.0700454711914062e-05,
      "model_forward_time": 2.898331642150879,
      "step": 21706
    },
    {
      "epoch": 2.0700454711914062e-05,
      "step": 21706,
      "training_step_time": 7.62640905380249
    },
    {
      "epoch": 2.070140838623047e-05,
      "model_forward_time": 1.7526757717132568,
      "step": 21707
    },
    {
      "epoch": 2.070140838623047e-05,
      "step": 21707,
      "training_step_time": 8.38954472541809
    },
    {
      "epoch": 2.0702362060546873e-05,
      "model_forward_time": 1.7524447441101074,
      "step": 21708
    },
    {
      "epoch": 2.0702362060546873e-05,
      "step": 21708,
      "training_step_time": 7.504265069961548
    },
    {
      "epoch": 2.070331573486328e-05,
      "model_forward_time": 1.7502131462097168,
      "step": 21709
    },
    {
      "epoch": 2.070331573486328e-05,
      "step": 21709,
      "training_step_time": 6.829066514968872
    },
    {
      "epoch": 2.070426940917969e-05,
      "grad_norm": 0.1941414177417755,
      "learning_rate": 9.255927323874993e-06,
      "loss": 0.0679,
      "step": 21710
    },
    {
      "epoch": 2.070426940917969e-05,
      "model_forward_time": 1.7547240257263184,
      "step": 21710
    },
    {
      "dynamics_loss_avg": 0.08138748165220022,
      "epoch": 2.070426940917969e-05,
      "step": 21710
    },
    {
      "action_loss_avg": 0.0035520356614142656,
      "epoch": 2.070426940917969e-05,
      "step": 21710
    },
    {
      "epoch": 2.070426940917969e-05,
      "step": 21710,
      "training_step_time": 6.810565948486328
    },
    {
      "epoch": 2.0705223083496092e-05,
      "model_forward_time": 1.7503175735473633,
      "step": 21711
    },
    {
      "epoch": 2.0705223083496092e-05,
      "step": 21711,
      "training_step_time": 6.838082313537598
    },
    {
      "epoch": 2.07061767578125e-05,
      "model_forward_time": 1.7430675029754639,
      "step": 21712
    },
    {
      "epoch": 2.07061767578125e-05,
      "step": 21712,
      "training_step_time": 6.834228515625
    },
    {
      "epoch": 2.0707130432128907e-05,
      "model_forward_time": 1.7624247074127197,
      "step": 21713
    },
    {
      "epoch": 2.0707130432128907e-05,
      "step": 21713,
      "training_step_time": 6.8442652225494385
    },
    {
      "epoch": 2.070808410644531e-05,
      "model_forward_time": 1.7661609649658203,
      "step": 21714
    },
    {
      "epoch": 2.070808410644531e-05,
      "step": 21714,
      "training_step_time": 6.799569606781006
    },
    {
      "epoch": 2.070903778076172e-05,
      "model_forward_time": 2.8757588863372803,
      "step": 21715
    },
    {
      "epoch": 2.070903778076172e-05,
      "step": 21715,
      "training_step_time": 7.742897272109985
    },
    {
      "epoch": 2.0709991455078126e-05,
      "model_forward_time": 1.7578191757202148,
      "step": 21716
    },
    {
      "epoch": 2.0709991455078126e-05,
      "step": 21716,
      "training_step_time": 10.774155378341675
    },
    {
      "epoch": 2.071094512939453e-05,
      "model_forward_time": 1.75675630569458,
      "step": 21717
    },
    {
      "epoch": 2.071094512939453e-05,
      "step": 21717,
      "training_step_time": 7.803507328033447
    },
    {
      "epoch": 2.0711898803710937e-05,
      "model_forward_time": 1.7613129615783691,
      "step": 21718
    },
    {
      "epoch": 2.0711898803710937e-05,
      "step": 21718,
      "training_step_time": 6.788302898406982
    },
    {
      "epoch": 2.0712852478027345e-05,
      "model_forward_time": 1.7586281299591064,
      "step": 21719
    },
    {
      "epoch": 2.0712852478027345e-05,
      "step": 21719,
      "training_step_time": 6.8021745681762695
    },
    {
      "epoch": 2.071380615234375e-05,
      "grad_norm": 0.1569657027721405,
      "learning_rate": 9.255059242686822e-06,
      "loss": 0.0659,
      "step": 21720
    },
    {
      "epoch": 2.071380615234375e-05,
      "model_forward_time": 1.7448983192443848,
      "step": 21720
    },
    {
      "dynamics_loss_avg": 0.07919812351465225,
      "epoch": 2.071380615234375e-05,
      "step": 21720
    },
    {
      "action_loss_avg": 0.0032601371844066308,
      "epoch": 2.071380615234375e-05,
      "step": 21720
    },
    {
      "epoch": 2.071380615234375e-05,
      "step": 21720,
      "training_step_time": 6.791479110717773
    },
    {
      "epoch": 2.0714759826660156e-05,
      "model_forward_time": 1.753676414489746,
      "step": 21721
    },
    {
      "epoch": 2.0714759826660156e-05,
      "step": 21721,
      "training_step_time": 6.820789575576782
    },
    {
      "epoch": 2.0715713500976564e-05,
      "model_forward_time": 1.7597367763519287,
      "step": 21722
    },
    {
      "epoch": 2.0715713500976564e-05,
      "step": 21722,
      "training_step_time": 6.693701982498169
    },
    {
      "epoch": 2.0716667175292968e-05,
      "model_forward_time": 1.7593820095062256,
      "step": 21723
    },
    {
      "epoch": 2.0716667175292968e-05,
      "step": 21723,
      "training_step_time": 6.7747321128845215
    },
    {
      "epoch": 2.0717620849609375e-05,
      "model_forward_time": 1.7831377983093262,
      "step": 21724
    },
    {
      "epoch": 2.0717620849609375e-05,
      "step": 21724,
      "training_step_time": 7.679919242858887
    },
    {
      "epoch": 2.0718574523925783e-05,
      "model_forward_time": 2.9092981815338135,
      "step": 21725
    },
    {
      "epoch": 2.0718574523925783e-05,
      "step": 21725,
      "training_step_time": 9.271935224533081
    },
    {
      "epoch": 2.0719528198242187e-05,
      "model_forward_time": 1.7513189315795898,
      "step": 21726
    },
    {
      "epoch": 2.0719528198242187e-05,
      "step": 21726,
      "training_step_time": 7.71357536315918
    },
    {
      "epoch": 2.0720481872558594e-05,
      "model_forward_time": 1.7581634521484375,
      "step": 21727
    },
    {
      "epoch": 2.0720481872558594e-05,
      "step": 21727,
      "training_step_time": 6.835948944091797
    },
    {
      "epoch": 2.0721435546875e-05,
      "model_forward_time": 1.758286714553833,
      "step": 21728
    },
    {
      "epoch": 2.0721435546875e-05,
      "step": 21728,
      "training_step_time": 6.827674150466919
    },
    {
      "epoch": 2.0722389221191405e-05,
      "model_forward_time": 1.7556049823760986,
      "step": 21729
    },
    {
      "epoch": 2.0722389221191405e-05,
      "step": 21729,
      "training_step_time": 6.809926986694336
    },
    {
      "epoch": 2.0723342895507813e-05,
      "grad_norm": 0.1973605751991272,
      "learning_rate": 9.25419069617177e-06,
      "loss": 0.0638,
      "step": 21730
    },
    {
      "epoch": 2.0723342895507813e-05,
      "model_forward_time": 1.7377188205718994,
      "step": 21730
    },
    {
      "dynamics_loss_avg": 0.07105026226490736,
      "epoch": 2.0723342895507813e-05,
      "step": 21730
    },
    {
      "action_loss_avg": 0.002467770205112174,
      "epoch": 2.0723342895507813e-05,
      "step": 21730
    },
    {
      "epoch": 2.0723342895507813e-05,
      "step": 21730,
      "training_step_time": 6.851214170455933
    },
    {
      "epoch": 2.072429656982422e-05,
      "model_forward_time": 1.75571870803833,
      "step": 21731
    },
    {
      "epoch": 2.072429656982422e-05,
      "step": 21731,
      "training_step_time": 6.869975566864014
    },
    {
      "epoch": 2.0725250244140624e-05,
      "model_forward_time": 1.8140745162963867,
      "step": 21732
    },
    {
      "epoch": 2.0725250244140624e-05,
      "step": 21732,
      "training_step_time": 6.842255592346191
    },
    {
      "epoch": 2.0726203918457032e-05,
      "model_forward_time": 1.7510778903961182,
      "step": 21733
    },
    {
      "epoch": 2.0726203918457032e-05,
      "step": 21733,
      "training_step_time": 7.407420873641968
    },
    {
      "epoch": 2.072715759277344e-05,
      "model_forward_time": 2.8749027252197266,
      "step": 21734
    },
    {
      "epoch": 2.072715759277344e-05,
      "step": 21734,
      "training_step_time": 7.6628923416137695
    },
    {
      "epoch": 2.0728111267089843e-05,
      "model_forward_time": 1.3871359825134277,
      "step": 21735
    },
    {
      "epoch": 2.0728111267089843e-05,
      "step": 21735,
      "training_step_time": 7.813737630844116
    },
    {
      "epoch": 2.072906494140625e-05,
      "model_forward_time": 1.7470340728759766,
      "step": 21736
    },
    {
      "epoch": 2.072906494140625e-05,
      "step": 21736,
      "training_step_time": 7.1280012130737305
    },
    {
      "epoch": 2.0730018615722655e-05,
      "model_forward_time": 1.7584788799285889,
      "step": 21737
    },
    {
      "epoch": 2.0730018615722655e-05,
      "step": 21737,
      "training_step_time": 6.885855197906494
    },
    {
      "epoch": 2.0730972290039062e-05,
      "model_forward_time": 1.7582859992980957,
      "step": 21738
    },
    {
      "epoch": 2.0730972290039062e-05,
      "step": 21738,
      "training_step_time": 6.860465049743652
    },
    {
      "epoch": 2.073192596435547e-05,
      "model_forward_time": 1.7574543952941895,
      "step": 21739
    },
    {
      "epoch": 2.073192596435547e-05,
      "step": 21739,
      "training_step_time": 6.857872009277344
    },
    {
      "epoch": 2.0732879638671874e-05,
      "grad_norm": 0.15687605738639832,
      "learning_rate": 9.25332168442482e-06,
      "loss": 0.0611,
      "step": 21740
    },
    {
      "epoch": 2.0732879638671874e-05,
      "model_forward_time": 1.7461216449737549,
      "step": 21740
    },
    {
      "dynamics_loss_avg": 0.06047827992588282,
      "epoch": 2.0732879638671874e-05,
      "step": 21740
    },
    {
      "action_loss_avg": 0.0028947544167749585,
      "epoch": 2.0732879638671874e-05,
      "step": 21740
    },
    {
      "epoch": 2.0732879638671874e-05,
      "step": 21740,
      "training_step_time": 6.885833978652954
    },
    {
      "epoch": 2.073383331298828e-05,
      "model_forward_time": 1.7479760646820068,
      "step": 21741
    },
    {
      "epoch": 2.073383331298828e-05,
      "step": 21741,
      "training_step_time": 6.840747594833374
    },
    {
      "epoch": 2.073478698730469e-05,
      "model_forward_time": 1.756221055984497,
      "step": 21742
    },
    {
      "epoch": 2.073478698730469e-05,
      "step": 21742,
      "training_step_time": 6.859444856643677
    },
    {
      "epoch": 2.0735740661621092e-05,
      "model_forward_time": 2.8976633548736572,
      "step": 21743
    },
    {
      "epoch": 2.0735740661621092e-05,
      "step": 21743,
      "training_step_time": 7.709484338760376
    },
    {
      "epoch": 2.07366943359375e-05,
      "model_forward_time": 1.7471961975097656,
      "step": 21744
    },
    {
      "epoch": 2.07366943359375e-05,
      "step": 21744,
      "training_step_time": 8.388104677200317
    },
    {
      "epoch": 2.0737648010253907e-05,
      "model_forward_time": 1.7541489601135254,
      "step": 21745
    },
    {
      "epoch": 2.0737648010253907e-05,
      "step": 21745,
      "training_step_time": 7.67175030708313
    },
    {
      "epoch": 2.073860168457031e-05,
      "model_forward_time": 1.7561869621276855,
      "step": 21746
    },
    {
      "epoch": 2.073860168457031e-05,
      "step": 21746,
      "training_step_time": 6.800253868103027
    },
    {
      "epoch": 2.073955535888672e-05,
      "model_forward_time": 1.7543601989746094,
      "step": 21747
    },
    {
      "epoch": 2.073955535888672e-05,
      "step": 21747,
      "training_step_time": 6.819961071014404
    },
    {
      "epoch": 2.0740509033203126e-05,
      "model_forward_time": 1.750352144241333,
      "step": 21748
    },
    {
      "epoch": 2.0740509033203126e-05,
      "step": 21748,
      "training_step_time": 6.816704988479614
    },
    {
      "epoch": 2.074146270751953e-05,
      "model_forward_time": 1.7566614151000977,
      "step": 21749
    },
    {
      "epoch": 2.074146270751953e-05,
      "step": 21749,
      "training_step_time": 6.810978651046753
    },
    {
      "epoch": 2.0742416381835938e-05,
      "grad_norm": 0.15470391511917114,
      "learning_rate": 9.252452207541004e-06,
      "loss": 0.0619,
      "step": 21750
    },
    {
      "epoch": 2.0742416381835938e-05,
      "model_forward_time": 1.7539138793945312,
      "step": 21750
    },
    {
      "dynamics_loss_avg": 0.05566665846854448,
      "epoch": 2.0742416381835938e-05,
      "step": 21750
    },
    {
      "action_loss_avg": 0.0020516979901003653,
      "epoch": 2.0742416381835938e-05,
      "step": 21750
    },
    {
      "epoch": 2.0742416381835938e-05,
      "step": 21750,
      "training_step_time": 6.834635257720947
    },
    {
      "epoch": 2.0743370056152345e-05,
      "model_forward_time": 1.7538509368896484,
      "step": 21751
    },
    {
      "epoch": 2.0743370056152345e-05,
      "step": 21751,
      "training_step_time": 6.820458173751831
    },
    {
      "epoch": 2.074432373046875e-05,
      "model_forward_time": 2.893615961074829,
      "step": 21752
    },
    {
      "epoch": 2.074432373046875e-05,
      "step": 21752,
      "training_step_time": 7.657253980636597
    },
    {
      "epoch": 2.0745277404785156e-05,
      "model_forward_time": 1.7488155364990234,
      "step": 21753
    },
    {
      "epoch": 2.0745277404785156e-05,
      "step": 21753,
      "training_step_time": 9.047043323516846
    },
    {
      "epoch": 2.0746231079101564e-05,
      "model_forward_time": 1.7539176940917969,
      "step": 21754
    },
    {
      "epoch": 2.0746231079101564e-05,
      "step": 21754,
      "training_step_time": 7.675410270690918
    },
    {
      "epoch": 2.0747184753417968e-05,
      "model_forward_time": 1.7509665489196777,
      "step": 21755
    },
    {
      "epoch": 2.0747184753417968e-05,
      "step": 21755,
      "training_step_time": 6.814600467681885
    },
    {
      "epoch": 2.0748138427734375e-05,
      "model_forward_time": 1.752079963684082,
      "step": 21756
    },
    {
      "epoch": 2.0748138427734375e-05,
      "step": 21756,
      "training_step_time": 6.789246082305908
    },
    {
      "epoch": 2.0749092102050783e-05,
      "model_forward_time": 1.74943208694458,
      "step": 21757
    },
    {
      "epoch": 2.0749092102050783e-05,
      "step": 21757,
      "training_step_time": 6.836314678192139
    },
    {
      "epoch": 2.0750045776367187e-05,
      "model_forward_time": 1.7518529891967773,
      "step": 21758
    },
    {
      "epoch": 2.0750045776367187e-05,
      "step": 21758,
      "training_step_time": 6.85709285736084
    },
    {
      "epoch": 2.0750999450683594e-05,
      "model_forward_time": 1.750546932220459,
      "step": 21759
    },
    {
      "epoch": 2.0750999450683594e-05,
      "step": 21759,
      "training_step_time": 6.855593681335449
    },
    {
      "epoch": 2.0751953125e-05,
      "grad_norm": 0.1294207125902176,
      "learning_rate": 9.25158226561541e-06,
      "loss": 0.0601,
      "step": 21760
    },
    {
      "epoch": 2.0751953125e-05,
      "model_forward_time": 1.7616040706634521,
      "step": 21760
    },
    {
      "dynamics_loss_avg": 0.04766614958643913,
      "epoch": 2.0751953125e-05,
      "step": 21760
    },
    {
      "action_loss_avg": 0.00231121442920994,
      "epoch": 2.0751953125e-05,
      "step": 21760
    },
    {
      "epoch": 2.0751953125e-05,
      "step": 21760,
      "training_step_time": 6.835254907608032
    },
    {
      "epoch": 2.0752906799316406e-05,
      "model_forward_time": 1.7846853733062744,
      "step": 21761
    },
    {
      "epoch": 2.0752906799316406e-05,
      "step": 21761,
      "training_step_time": 8.218934535980225
    },
    {
      "epoch": 2.0753860473632813e-05,
      "model_forward_time": 1.7528667449951172,
      "step": 21762
    },
    {
      "epoch": 2.0753860473632813e-05,
      "step": 21762,
      "training_step_time": 9.518069744110107
    },
    {
      "epoch": 2.075481414794922e-05,
      "model_forward_time": 1.749873399734497,
      "step": 21763
    },
    {
      "epoch": 2.075481414794922e-05,
      "step": 21763,
      "training_step_time": 7.661219835281372
    },
    {
      "epoch": 2.0755767822265624e-05,
      "model_forward_time": 1.7479894161224365,
      "step": 21764
    },
    {
      "epoch": 2.0755767822265624e-05,
      "step": 21764,
      "training_step_time": 7.590858697891235
    },
    {
      "epoch": 2.0756721496582032e-05,
      "model_forward_time": 1.7545151710510254,
      "step": 21765
    },
    {
      "epoch": 2.0756721496582032e-05,
      "step": 21765,
      "training_step_time": 6.841552019119263
    },
    {
      "epoch": 2.0757675170898436e-05,
      "model_forward_time": 1.78248929977417,
      "step": 21766
    },
    {
      "epoch": 2.0757675170898436e-05,
      "step": 21766,
      "training_step_time": 6.838874816894531
    },
    {
      "epoch": 2.0758628845214843e-05,
      "model_forward_time": 1.7452170848846436,
      "step": 21767
    },
    {
      "epoch": 2.0758628845214843e-05,
      "step": 21767,
      "training_step_time": 6.699042320251465
    },
    {
      "epoch": 2.075958251953125e-05,
      "model_forward_time": 1.7464015483856201,
      "step": 21768
    },
    {
      "epoch": 2.075958251953125e-05,
      "step": 21768,
      "training_step_time": 6.804666042327881
    },
    {
      "epoch": 2.0760536193847655e-05,
      "model_forward_time": 1.7485733032226562,
      "step": 21769
    },
    {
      "epoch": 2.0760536193847655e-05,
      "step": 21769,
      "training_step_time": 6.862654447555542
    },
    {
      "epoch": 2.0761489868164062e-05,
      "grad_norm": 0.15416108071804047,
      "learning_rate": 9.250711858743169e-06,
      "loss": 0.0651,
      "step": 21770
    },
    {
      "epoch": 2.0761489868164062e-05,
      "model_forward_time": 1.752133846282959,
      "step": 21770
    },
    {
      "dynamics_loss_avg": 0.08104790821671486,
      "epoch": 2.0761489868164062e-05,
      "step": 21770
    },
    {
      "action_loss_avg": 0.003785904694814235,
      "epoch": 2.0761489868164062e-05,
      "step": 21770
    },
    {
      "epoch": 2.0761489868164062e-05,
      "step": 21770,
      "training_step_time": 8.506380319595337
    },
    {
      "epoch": 2.076244354248047e-05,
      "model_forward_time": 2.5070013999938965,
      "step": 21771
    },
    {
      "epoch": 2.076244354248047e-05,
      "step": 21771,
      "training_step_time": 9.666815042495728
    },
    {
      "epoch": 2.0763397216796874e-05,
      "model_forward_time": 1.7542448043823242,
      "step": 21772
    },
    {
      "epoch": 2.0763397216796874e-05,
      "step": 21772,
      "training_step_time": 7.77748703956604
    },
    {
      "epoch": 2.076435089111328e-05,
      "model_forward_time": 1.7429859638214111,
      "step": 21773
    },
    {
      "epoch": 2.076435089111328e-05,
      "step": 21773,
      "training_step_time": 7.576707124710083
    },
    {
      "epoch": 2.076530456542969e-05,
      "model_forward_time": 1.7582318782806396,
      "step": 21774
    },
    {
      "epoch": 2.076530456542969e-05,
      "step": 21774,
      "training_step_time": 6.721934080123901
    },
    {
      "epoch": 2.0766258239746092e-05,
      "model_forward_time": 1.7663888931274414,
      "step": 21775
    },
    {
      "epoch": 2.0766258239746092e-05,
      "step": 21775,
      "training_step_time": 6.712433338165283
    },
    {
      "epoch": 2.07672119140625e-05,
      "model_forward_time": 1.7508978843688965,
      "step": 21776
    },
    {
      "epoch": 2.07672119140625e-05,
      "step": 21776,
      "training_step_time": 6.707945346832275
    },
    {
      "epoch": 2.0768165588378907e-05,
      "model_forward_time": 1.7567880153656006,
      "step": 21777
    },
    {
      "epoch": 2.0768165588378907e-05,
      "step": 21777,
      "training_step_time": 6.710653066635132
    },
    {
      "epoch": 2.076911926269531e-05,
      "model_forward_time": 1.7464094161987305,
      "step": 21778
    },
    {
      "epoch": 2.076911926269531e-05,
      "step": 21778,
      "training_step_time": 6.705570220947266
    },
    {
      "epoch": 2.077007293701172e-05,
      "model_forward_time": 1.7513759136199951,
      "step": 21779
    },
    {
      "epoch": 2.077007293701172e-05,
      "step": 21779,
      "training_step_time": 7.110444068908691
    },
    {
      "epoch": 2.0771026611328126e-05,
      "grad_norm": 0.17772096395492554,
      "learning_rate": 9.24984098701947e-06,
      "loss": 0.0644,
      "step": 21780
    },
    {
      "epoch": 2.0771026611328126e-05,
      "model_forward_time": 2.8605289459228516,
      "step": 21780
    },
    {
      "dynamics_loss_avg": 0.06071455255150795,
      "epoch": 2.0771026611328126e-05,
      "step": 21780
    },
    {
      "action_loss_avg": 0.002694708545459434,
      "epoch": 2.0771026611328126e-05,
      "step": 21780
    },
    {
      "epoch": 2.0771026611328126e-05,
      "step": 21780,
      "training_step_time": 9.144028663635254
    },
    {
      "epoch": 2.077198028564453e-05,
      "model_forward_time": 1.7158050537109375,
      "step": 21781
    },
    {
      "epoch": 2.077198028564453e-05,
      "step": 21781,
      "training_step_time": 8.335985898971558
    },
    {
      "epoch": 2.0772933959960938e-05,
      "model_forward_time": 1.7509942054748535,
      "step": 21782
    },
    {
      "epoch": 2.0772933959960938e-05,
      "step": 21782,
      "training_step_time": 7.671613931655884
    },
    {
      "epoch": 2.0773887634277345e-05,
      "model_forward_time": 1.7498695850372314,
      "step": 21783
    },
    {
      "epoch": 2.0773887634277345e-05,
      "step": 21783,
      "training_step_time": 6.70428991317749
    },
    {
      "epoch": 2.077484130859375e-05,
      "model_forward_time": 1.7475647926330566,
      "step": 21784
    },
    {
      "epoch": 2.077484130859375e-05,
      "step": 21784,
      "training_step_time": 6.732642412185669
    },
    {
      "epoch": 2.0775794982910156e-05,
      "model_forward_time": 1.747593641281128,
      "step": 21785
    },
    {
      "epoch": 2.0775794982910156e-05,
      "step": 21785,
      "training_step_time": 6.674874305725098
    },
    {
      "epoch": 2.0776748657226564e-05,
      "model_forward_time": 1.7517821788787842,
      "step": 21786
    },
    {
      "epoch": 2.0776748657226564e-05,
      "step": 21786,
      "training_step_time": 6.73480486869812
    },
    {
      "epoch": 2.0777702331542968e-05,
      "model_forward_time": 1.7390854358673096,
      "step": 21787
    },
    {
      "epoch": 2.0777702331542968e-05,
      "step": 21787,
      "training_step_time": 6.732571601867676
    },
    {
      "epoch": 2.0778656005859375e-05,
      "model_forward_time": 1.7373316287994385,
      "step": 21788
    },
    {
      "epoch": 2.0778656005859375e-05,
      "step": 21788,
      "training_step_time": 7.1392717361450195
    },
    {
      "epoch": 2.0779609680175783e-05,
      "model_forward_time": 2.8540656566619873,
      "step": 21789
    },
    {
      "epoch": 2.0779609680175783e-05,
      "step": 21789,
      "training_step_time": 7.649252653121948
    },
    {
      "epoch": 2.0780563354492187e-05,
      "grad_norm": 0.17248789966106415,
      "learning_rate": 9.248969650539553e-06,
      "loss": 0.066,
      "step": 21790
    },
    {
      "epoch": 2.0780563354492187e-05,
      "model_forward_time": 1.727771520614624,
      "step": 21790
    },
    {
      "dynamics_loss_avg": 0.06542411148548126,
      "epoch": 2.0780563354492187e-05,
      "step": 21790
    },
    {
      "action_loss_avg": 0.0027532253647223116,
      "epoch": 2.0780563354492187e-05,
      "step": 21790
    },
    {
      "epoch": 2.0780563354492187e-05,
      "step": 21790,
      "training_step_time": 11.706995725631714
    },
    {
      "epoch": 2.0781517028808594e-05,
      "model_forward_time": 1.7351438999176025,
      "step": 21791
    },
    {
      "epoch": 2.0781517028808594e-05,
      "step": 21791,
      "training_step_time": 8.7137770652771
    },
    {
      "epoch": 2.0782470703125e-05,
      "model_forward_time": 1.7516613006591797,
      "step": 21792
    },
    {
      "epoch": 2.0782470703125e-05,
      "step": 21792,
      "training_step_time": 7.0852134227752686
    },
    {
      "epoch": 2.0783424377441406e-05,
      "model_forward_time": 1.770942211151123,
      "step": 21793
    },
    {
      "epoch": 2.0783424377441406e-05,
      "step": 21793,
      "training_step_time": 6.7183356285095215
    },
    {
      "epoch": 2.0784378051757813e-05,
      "model_forward_time": 1.74898099899292,
      "step": 21794
    },
    {
      "epoch": 2.0784378051757813e-05,
      "step": 21794,
      "training_step_time": 6.7039875984191895
    },
    {
      "epoch": 2.078533172607422e-05,
      "model_forward_time": 1.749626874923706,
      "step": 21795
    },
    {
      "epoch": 2.078533172607422e-05,
      "step": 21795,
      "training_step_time": 6.733988523483276
    },
    {
      "epoch": 2.0786285400390624e-05,
      "model_forward_time": 1.7584106922149658,
      "step": 21796
    },
    {
      "epoch": 2.0786285400390624e-05,
      "step": 21796,
      "training_step_time": 6.7122156620025635
    },
    {
      "epoch": 2.0787239074707032e-05,
      "model_forward_time": 1.7532215118408203,
      "step": 21797
    },
    {
      "epoch": 2.0787239074707032e-05,
      "step": 21797,
      "training_step_time": 6.88937520980835
    },
    {
      "epoch": 2.0788192749023436e-05,
      "model_forward_time": 2.8670601844787598,
      "step": 21798
    },
    {
      "epoch": 2.0788192749023436e-05,
      "step": 21798,
      "training_step_time": 9.071349382400513
    },
    {
      "epoch": 2.0789146423339843e-05,
      "model_forward_time": 1.7445240020751953,
      "step": 21799
    },
    {
      "epoch": 2.0789146423339843e-05,
      "step": 21799,
      "training_step_time": 11.343764305114746
    },
    {
      "epoch": 2.079010009765625e-05,
      "grad_norm": 0.16193833947181702,
      "learning_rate": 9.2480978493987e-06,
      "loss": 0.0668,
      "step": 21800
    },
    {
      "epoch": 2.079010009765625e-05,
      "model_forward_time": 1.7381885051727295,
      "step": 21800
    },
    {
      "dynamics_loss_avg": 0.060502082109451294,
      "epoch": 2.079010009765625e-05,
      "step": 21800
    },
    {
      "action_loss_avg": 0.0026504064851906152,
      "epoch": 2.079010009765625e-05,
      "step": 21800
    },
    {
      "epoch": 2.079010009765625e-05,
      "step": 21800,
      "training_step_time": 8.129053831100464
    },
    {
      "epoch": 2.0791053771972655e-05,
      "model_forward_time": 1.756120204925537,
      "step": 21801
    },
    {
      "epoch": 2.0791053771972655e-05,
      "step": 21801,
      "training_step_time": 7.788359880447388
    },
    {
      "epoch": 2.0792007446289062e-05,
      "model_forward_time": 1.7609937191009521,
      "step": 21802
    },
    {
      "epoch": 2.0792007446289062e-05,
      "step": 21802,
      "training_step_time": 6.86269211769104
    },
    {
      "epoch": 2.079296112060547e-05,
      "model_forward_time": 1.7365813255310059,
      "step": 21803
    },
    {
      "epoch": 2.079296112060547e-05,
      "step": 21803,
      "training_step_time": 6.8623833656311035
    },
    {
      "epoch": 2.0793914794921874e-05,
      "model_forward_time": 1.7577438354492188,
      "step": 21804
    },
    {
      "epoch": 2.0793914794921874e-05,
      "step": 21804,
      "training_step_time": 6.894134521484375
    },
    {
      "epoch": 2.079486846923828e-05,
      "model_forward_time": 1.7411689758300781,
      "step": 21805
    },
    {
      "epoch": 2.079486846923828e-05,
      "step": 21805,
      "training_step_time": 6.859105348587036
    },
    {
      "epoch": 2.079582214355469e-05,
      "model_forward_time": 1.7269177436828613,
      "step": 21806
    },
    {
      "epoch": 2.079582214355469e-05,
      "step": 21806,
      "training_step_time": 6.90369725227356
    },
    {
      "epoch": 2.0796775817871092e-05,
      "model_forward_time": 1.761542558670044,
      "step": 21807
    },
    {
      "epoch": 2.0796775817871092e-05,
      "step": 21807,
      "training_step_time": 7.97373366355896
    },
    {
      "epoch": 2.07977294921875e-05,
      "model_forward_time": 2.8539247512817383,
      "step": 21808
    },
    {
      "epoch": 2.07977294921875e-05,
      "step": 21808,
      "training_step_time": 8.780221700668335
    },
    {
      "epoch": 2.0798683166503907e-05,
      "model_forward_time": 1.728818416595459,
      "step": 21809
    },
    {
      "epoch": 2.0798683166503907e-05,
      "step": 21809,
      "training_step_time": 7.847299814224243
    },
    {
      "epoch": 2.079963684082031e-05,
      "grad_norm": 0.17621679604053497,
      "learning_rate": 9.247225583692256e-06,
      "loss": 0.0629,
      "step": 21810
    },
    {
      "epoch": 2.079963684082031e-05,
      "model_forward_time": 1.720118522644043,
      "step": 21810
    },
    {
      "dynamics_loss_avg": 0.06393108898773789,
      "epoch": 2.079963684082031e-05,
      "step": 21810
    },
    {
      "action_loss_avg": 0.00391406276030466,
      "epoch": 2.079963684082031e-05,
      "step": 21810
    },
    {
      "epoch": 2.079963684082031e-05,
      "step": 21810,
      "training_step_time": 7.617263555526733
    },
    {
      "epoch": 2.080059051513672e-05,
      "model_forward_time": 1.7310137748718262,
      "step": 21811
    },
    {
      "epoch": 2.080059051513672e-05,
      "step": 21811,
      "training_step_time": 6.663984298706055
    },
    {
      "epoch": 2.0801544189453126e-05,
      "model_forward_time": 1.7345991134643555,
      "step": 21812
    },
    {
      "epoch": 2.0801544189453126e-05,
      "step": 21812,
      "training_step_time": 6.686574220657349
    },
    {
      "epoch": 2.080249786376953e-05,
      "model_forward_time": 1.7328295707702637,
      "step": 21813
    },
    {
      "epoch": 2.080249786376953e-05,
      "step": 21813,
      "training_step_time": 6.686352014541626
    },
    {
      "epoch": 2.0803451538085938e-05,
      "model_forward_time": 1.7293260097503662,
      "step": 21814
    },
    {
      "epoch": 2.0803451538085938e-05,
      "step": 21814,
      "training_step_time": 6.717658042907715
    },
    {
      "epoch": 2.0804405212402345e-05,
      "model_forward_time": 1.7272515296936035,
      "step": 21815
    },
    {
      "epoch": 2.0804405212402345e-05,
      "step": 21815,
      "training_step_time": 6.747727394104004
    },
    {
      "epoch": 2.080535888671875e-05,
      "model_forward_time": 1.7465264797210693,
      "step": 21816
    },
    {
      "epoch": 2.080535888671875e-05,
      "step": 21816,
      "training_step_time": 8.685261487960815
    },
    {
      "epoch": 2.0806312561035156e-05,
      "model_forward_time": 2.8484573364257812,
      "step": 21817
    },
    {
      "epoch": 2.0806312561035156e-05,
      "step": 21817,
      "training_step_time": 9.565040349960327
    },
    {
      "epoch": 2.0807266235351564e-05,
      "model_forward_time": 1.7270786762237549,
      "step": 21818
    },
    {
      "epoch": 2.0807266235351564e-05,
      "step": 21818,
      "training_step_time": 8.879584550857544
    },
    {
      "epoch": 2.0808219909667968e-05,
      "model_forward_time": 1.7258059978485107,
      "step": 21819
    },
    {
      "epoch": 2.0808219909667968e-05,
      "step": 21819,
      "training_step_time": 7.685106515884399
    },
    {
      "epoch": 2.0809173583984375e-05,
      "grad_norm": 0.164768248796463,
      "learning_rate": 9.246352853515606e-06,
      "loss": 0.0638,
      "step": 21820
    },
    {
      "epoch": 2.0809173583984375e-05,
      "model_forward_time": 1.7363476753234863,
      "step": 21820
    },
    {
      "dynamics_loss_avg": 0.07599030286073685,
      "epoch": 2.0809173583984375e-05,
      "step": 21820
    },
    {
      "action_loss_avg": 0.0030934392765630037,
      "epoch": 2.0809173583984375e-05,
      "step": 21820
    },
    {
      "epoch": 2.0809173583984375e-05,
      "step": 21820,
      "training_step_time": 7.174494743347168
    },
    {
      "epoch": 2.0810127258300783e-05,
      "model_forward_time": 1.7270734310150146,
      "step": 21821
    },
    {
      "epoch": 2.0810127258300783e-05,
      "step": 21821,
      "training_step_time": 6.8023903369903564
    },
    {
      "epoch": 2.0811080932617187e-05,
      "model_forward_time": 1.7328588962554932,
      "step": 21822
    },
    {
      "epoch": 2.0811080932617187e-05,
      "step": 21822,
      "training_step_time": 6.796718120574951
    },
    {
      "epoch": 2.0812034606933594e-05,
      "model_forward_time": 1.731895923614502,
      "step": 21823
    },
    {
      "epoch": 2.0812034606933594e-05,
      "step": 21823,
      "training_step_time": 6.7953081130981445
    },
    {
      "epoch": 2.081298828125e-05,
      "model_forward_time": 1.7250947952270508,
      "step": 21824
    },
    {
      "epoch": 2.081298828125e-05,
      "step": 21824,
      "training_step_time": 6.824502944946289
    },
    {
      "epoch": 2.0813941955566406e-05,
      "model_forward_time": 1.7392854690551758,
      "step": 21825
    },
    {
      "epoch": 2.0813941955566406e-05,
      "step": 21825,
      "training_step_time": 7.6158037185668945
    },
    {
      "epoch": 2.0814895629882813e-05,
      "model_forward_time": 2.829620838165283,
      "step": 21826
    },
    {
      "epoch": 2.0814895629882813e-05,
      "step": 21826,
      "training_step_time": 7.756869316101074
    },
    {
      "epoch": 2.081584930419922e-05,
      "model_forward_time": 1.7267308235168457,
      "step": 21827
    },
    {
      "epoch": 2.081584930419922e-05,
      "step": 21827,
      "training_step_time": 11.114673852920532
    },
    {
      "epoch": 2.0816802978515624e-05,
      "model_forward_time": 1.7292039394378662,
      "step": 21828
    },
    {
      "epoch": 2.0816802978515624e-05,
      "step": 21828,
      "training_step_time": 7.707585334777832
    },
    {
      "epoch": 2.0817756652832032e-05,
      "model_forward_time": 1.7260897159576416,
      "step": 21829
    },
    {
      "epoch": 2.0817756652832032e-05,
      "step": 21829,
      "training_step_time": 7.138484954833984
    },
    {
      "epoch": 2.0818710327148436e-05,
      "grad_norm": 0.20420940220355988,
      "learning_rate": 9.245479658964193e-06,
      "loss": 0.0646,
      "step": 21830
    },
    {
      "epoch": 2.0818710327148436e-05,
      "model_forward_time": 1.7213659286499023,
      "step": 21830
    },
    {
      "dynamics_loss_avg": 0.03873906321823597,
      "epoch": 2.0818710327148436e-05,
      "step": 21830
    },
    {
      "action_loss_avg": 0.0018916900648036973,
      "epoch": 2.0818710327148436e-05,
      "step": 21830
    },
    {
      "epoch": 2.0818710327148436e-05,
      "step": 21830,
      "training_step_time": 6.721158504486084
    },
    {
      "epoch": 2.0819664001464843e-05,
      "model_forward_time": 1.7263131141662598,
      "step": 21831
    },
    {
      "epoch": 2.0819664001464843e-05,
      "step": 21831,
      "training_step_time": 6.750952243804932
    },
    {
      "epoch": 2.082061767578125e-05,
      "model_forward_time": 1.7247121334075928,
      "step": 21832
    },
    {
      "epoch": 2.082061767578125e-05,
      "step": 21832,
      "training_step_time": 6.733019590377808
    },
    {
      "epoch": 2.0821571350097655e-05,
      "model_forward_time": 1.7221076488494873,
      "step": 21833
    },
    {
      "epoch": 2.0821571350097655e-05,
      "step": 21833,
      "training_step_time": 6.742915630340576
    },
    {
      "epoch": 2.0822525024414062e-05,
      "model_forward_time": 1.730590581893921,
      "step": 21834
    },
    {
      "epoch": 2.0822525024414062e-05,
      "step": 21834,
      "training_step_time": 7.730613470077515
    },
    {
      "epoch": 2.082347869873047e-05,
      "model_forward_time": 2.8526837825775146,
      "step": 21835
    },
    {
      "epoch": 2.082347869873047e-05,
      "step": 21835,
      "training_step_time": 8.463975429534912
    },
    {
      "epoch": 2.0824432373046874e-05,
      "model_forward_time": 1.7282001972198486,
      "step": 21836
    },
    {
      "epoch": 2.0824432373046874e-05,
      "step": 21836,
      "training_step_time": 7.775207281112671
    },
    {
      "epoch": 2.082538604736328e-05,
      "model_forward_time": 1.7284092903137207,
      "step": 21837
    },
    {
      "epoch": 2.082538604736328e-05,
      "step": 21837,
      "training_step_time": 9.281598329544067
    },
    {
      "epoch": 2.082633972167969e-05,
      "model_forward_time": 1.7300827503204346,
      "step": 21838
    },
    {
      "epoch": 2.082633972167969e-05,
      "step": 21838,
      "training_step_time": 7.617804527282715
    },
    {
      "epoch": 2.0827293395996092e-05,
      "model_forward_time": 1.7266299724578857,
      "step": 21839
    },
    {
      "epoch": 2.0827293395996092e-05,
      "step": 21839,
      "training_step_time": 6.72968316078186
    },
    {
      "epoch": 2.08282470703125e-05,
      "grad_norm": 0.18977327644824982,
      "learning_rate": 9.244606000133507e-06,
      "loss": 0.0659,
      "step": 21840
    },
    {
      "epoch": 2.08282470703125e-05,
      "model_forward_time": 1.7330842018127441,
      "step": 21840
    },
    {
      "dynamics_loss_avg": 0.0638159267604351,
      "epoch": 2.08282470703125e-05,
      "step": 21840
    },
    {
      "action_loss_avg": 0.003396884771063924,
      "epoch": 2.08282470703125e-05,
      "step": 21840
    },
    {
      "epoch": 2.08282470703125e-05,
      "step": 21840,
      "training_step_time": 6.7501959800720215
    },
    {
      "epoch": 2.0829200744628907e-05,
      "model_forward_time": 1.730452060699463,
      "step": 21841
    },
    {
      "epoch": 2.0829200744628907e-05,
      "step": 21841,
      "training_step_time": 6.714400768280029
    },
    {
      "epoch": 2.083015441894531e-05,
      "model_forward_time": 1.3651702404022217,
      "step": 21842
    },
    {
      "epoch": 2.083015441894531e-05,
      "step": 21842,
      "training_step_time": 6.6835105419158936
    },
    {
      "epoch": 2.083110809326172e-05,
      "model_forward_time": 1.7321395874023438,
      "step": 21843
    },
    {
      "epoch": 2.083110809326172e-05,
      "step": 21843,
      "training_step_time": 6.706310510635376
    },
    {
      "epoch": 2.0832061767578126e-05,
      "model_forward_time": 2.849425792694092,
      "step": 21844
    },
    {
      "epoch": 2.0832061767578126e-05,
      "step": 21844,
      "training_step_time": 8.415538311004639
    },
    {
      "epoch": 2.083301544189453e-05,
      "model_forward_time": 1.7272260189056396,
      "step": 21845
    },
    {
      "epoch": 2.083301544189453e-05,
      "step": 21845,
      "training_step_time": 8.854834794998169
    },
    {
      "epoch": 2.0833969116210938e-05,
      "model_forward_time": 1.7299182415008545,
      "step": 21846
    },
    {
      "epoch": 2.0833969116210938e-05,
      "step": 21846,
      "training_step_time": 7.772681713104248
    },
    {
      "epoch": 2.0834922790527345e-05,
      "model_forward_time": 1.7302334308624268,
      "step": 21847
    },
    {
      "epoch": 2.0834922790527345e-05,
      "step": 21847,
      "training_step_time": 7.643613338470459
    },
    {
      "epoch": 2.083587646484375e-05,
      "model_forward_time": 1.7290668487548828,
      "step": 21848
    },
    {
      "epoch": 2.083587646484375e-05,
      "step": 21848,
      "training_step_time": 6.743204832077026
    },
    {
      "epoch": 2.0836830139160156e-05,
      "model_forward_time": 1.729766845703125,
      "step": 21849
    },
    {
      "epoch": 2.0836830139160156e-05,
      "step": 21849,
      "training_step_time": 6.715531826019287
    },
    {
      "epoch": 2.0837783813476564e-05,
      "grad_norm": 0.1486254185438156,
      "learning_rate": 9.24373187711909e-06,
      "loss": 0.0647,
      "step": 21850
    },
    {
      "epoch": 2.0837783813476564e-05,
      "model_forward_time": 1.7362444400787354,
      "step": 21850
    },
    {
      "dynamics_loss_avg": 0.07861883919686079,
      "epoch": 2.0837783813476564e-05,
      "step": 21850
    },
    {
      "action_loss_avg": 0.00392492082901299,
      "epoch": 2.0837783813476564e-05,
      "step": 21850
    },
    {
      "epoch": 2.0837783813476564e-05,
      "step": 21850,
      "training_step_time": 6.714353799819946
    },
    {
      "epoch": 2.0838737487792968e-05,
      "model_forward_time": 1.7292335033416748,
      "step": 21851
    },
    {
      "epoch": 2.0838737487792968e-05,
      "step": 21851,
      "training_step_time": 6.730480432510376
    },
    {
      "epoch": 2.0839691162109375e-05,
      "model_forward_time": 1.726210355758667,
      "step": 21852
    },
    {
      "epoch": 2.0839691162109375e-05,
      "step": 21852,
      "training_step_time": 6.724085807800293
    },
    {
      "epoch": 2.0840644836425783e-05,
      "model_forward_time": 1.7527296543121338,
      "step": 21853
    },
    {
      "epoch": 2.0840644836425783e-05,
      "step": 21853,
      "training_step_time": 8.555929183959961
    },
    {
      "epoch": 2.0841598510742187e-05,
      "model_forward_time": 2.8890531063079834,
      "step": 21854
    },
    {
      "epoch": 2.0841598510742187e-05,
      "step": 21854,
      "training_step_time": 9.622379779815674
    },
    {
      "epoch": 2.0842552185058594e-05,
      "model_forward_time": 1.728708028793335,
      "step": 21855
    },
    {
      "epoch": 2.0842552185058594e-05,
      "step": 21855,
      "training_step_time": 7.912067651748657
    },
    {
      "epoch": 2.0843505859375e-05,
      "model_forward_time": 1.726853609085083,
      "step": 21856
    },
    {
      "epoch": 2.0843505859375e-05,
      "step": 21856,
      "training_step_time": 7.68602180480957
    },
    {
      "epoch": 2.0844459533691406e-05,
      "model_forward_time": 1.7269506454467773,
      "step": 21857
    },
    {
      "epoch": 2.0844459533691406e-05,
      "step": 21857,
      "training_step_time": 6.764908075332642
    },
    {
      "epoch": 2.0845413208007813e-05,
      "model_forward_time": 1.7224972248077393,
      "step": 21858
    },
    {
      "epoch": 2.0845413208007813e-05,
      "step": 21858,
      "training_step_time": 6.734483480453491
    },
    {
      "epoch": 2.084636688232422e-05,
      "model_forward_time": 1.7265195846557617,
      "step": 21859
    },
    {
      "epoch": 2.084636688232422e-05,
      "step": 21859,
      "training_step_time": 6.753648519515991
    },
    {
      "epoch": 2.0847320556640624e-05,
      "grad_norm": 0.18510212004184723,
      "learning_rate": 9.242857290016537e-06,
      "loss": 0.0629,
      "step": 21860
    },
    {
      "epoch": 2.0847320556640624e-05,
      "model_forward_time": 1.7313523292541504,
      "step": 21860
    },
    {
      "dynamics_loss_avg": 0.0603840590454638,
      "epoch": 2.0847320556640624e-05,
      "step": 21860
    },
    {
      "action_loss_avg": 0.002859852829715237,
      "epoch": 2.0847320556640624e-05,
      "step": 21860
    },
    {
      "epoch": 2.0847320556640624e-05,
      "step": 21860,
      "training_step_time": 6.790549993515015
    },
    {
      "epoch": 2.0848274230957032e-05,
      "model_forward_time": 1.7204318046569824,
      "step": 21861
    },
    {
      "epoch": 2.0848274230957032e-05,
      "step": 21861,
      "training_step_time": 6.733350515365601
    },
    {
      "epoch": 2.0849227905273436e-05,
      "model_forward_time": 1.739271879196167,
      "step": 21862
    },
    {
      "epoch": 2.0849227905273436e-05,
      "step": 21862,
      "training_step_time": 7.772233247756958
    },
    {
      "epoch": 2.0850181579589843e-05,
      "model_forward_time": 2.9252328872680664,
      "step": 21863
    },
    {
      "epoch": 2.0850181579589843e-05,
      "step": 21863,
      "training_step_time": 9.815717935562134
    },
    {
      "epoch": 2.085113525390625e-05,
      "model_forward_time": 1.7284643650054932,
      "step": 21864
    },
    {
      "epoch": 2.085113525390625e-05,
      "step": 21864,
      "training_step_time": 8.38161325454712
    },
    {
      "epoch": 2.0852088928222655e-05,
      "model_forward_time": 1.7372148036956787,
      "step": 21865
    },
    {
      "epoch": 2.0852088928222655e-05,
      "step": 21865,
      "training_step_time": 7.618774652481079
    },
    {
      "epoch": 2.0853042602539062e-05,
      "model_forward_time": 1.7373247146606445,
      "step": 21866
    },
    {
      "epoch": 2.0853042602539062e-05,
      "step": 21866,
      "training_step_time": 7.143183708190918
    },
    {
      "epoch": 2.085399627685547e-05,
      "model_forward_time": 1.734689474105835,
      "step": 21867
    },
    {
      "epoch": 2.085399627685547e-05,
      "step": 21867,
      "training_step_time": 6.753675937652588
    },
    {
      "epoch": 2.0854949951171874e-05,
      "model_forward_time": 1.7407989501953125,
      "step": 21868
    },
    {
      "epoch": 2.0854949951171874e-05,
      "step": 21868,
      "training_step_time": 6.820055723190308
    },
    {
      "epoch": 2.085590362548828e-05,
      "model_forward_time": 1.7346909046173096,
      "step": 21869
    },
    {
      "epoch": 2.085590362548828e-05,
      "step": 21869,
      "training_step_time": 6.755089521408081
    },
    {
      "epoch": 2.085685729980469e-05,
      "grad_norm": 0.13251835107803345,
      "learning_rate": 9.241982238921489e-06,
      "loss": 0.0639,
      "step": 21870
    },
    {
      "epoch": 2.085685729980469e-05,
      "model_forward_time": 1.7397541999816895,
      "step": 21870
    },
    {
      "dynamics_loss_avg": 0.06455859597772359,
      "epoch": 2.085685729980469e-05,
      "step": 21870
    },
    {
      "action_loss_avg": 0.003123479842906818,
      "epoch": 2.085685729980469e-05,
      "step": 21870
    },
    {
      "epoch": 2.085685729980469e-05,
      "step": 21870,
      "training_step_time": 6.770426511764526
    },
    {
      "epoch": 2.0857810974121092e-05,
      "model_forward_time": 1.7303814888000488,
      "step": 21871
    },
    {
      "epoch": 2.0857810974121092e-05,
      "step": 21871,
      "training_step_time": 7.70154881477356
    },
    {
      "epoch": 2.08587646484375e-05,
      "model_forward_time": 2.8385636806488037,
      "step": 21872
    },
    {
      "epoch": 2.08587646484375e-05,
      "step": 21872,
      "training_step_time": 8.847009420394897
    },
    {
      "epoch": 2.0859718322753907e-05,
      "model_forward_time": 1.7714073657989502,
      "step": 21873
    },
    {
      "epoch": 2.0859718322753907e-05,
      "step": 21873,
      "training_step_time": 9.256604194641113
    },
    {
      "epoch": 2.086067199707031e-05,
      "model_forward_time": 1.7323570251464844,
      "step": 21874
    },
    {
      "epoch": 2.086067199707031e-05,
      "step": 21874,
      "training_step_time": 11.279623746871948
    },
    {
      "epoch": 2.086162567138672e-05,
      "model_forward_time": 1.7361507415771484,
      "step": 21875
    },
    {
      "epoch": 2.086162567138672e-05,
      "step": 21875,
      "training_step_time": 7.567506790161133
    },
    {
      "epoch": 2.0862579345703126e-05,
      "model_forward_time": 1.7327487468719482,
      "step": 21876
    },
    {
      "epoch": 2.0862579345703126e-05,
      "step": 21876,
      "training_step_time": 6.76901912689209
    },
    {
      "epoch": 2.086353302001953e-05,
      "model_forward_time": 1.7324624061584473,
      "step": 21877
    },
    {
      "epoch": 2.086353302001953e-05,
      "step": 21877,
      "training_step_time": 6.784769058227539
    },
    {
      "epoch": 2.0864486694335938e-05,
      "model_forward_time": 1.734849214553833,
      "step": 21878
    },
    {
      "epoch": 2.0864486694335938e-05,
      "step": 21878,
      "training_step_time": 6.735782146453857
    },
    {
      "epoch": 2.0865440368652345e-05,
      "model_forward_time": 1.7313191890716553,
      "step": 21879
    },
    {
      "epoch": 2.0865440368652345e-05,
      "step": 21879,
      "training_step_time": 6.737616777420044
    },
    {
      "epoch": 2.086639404296875e-05,
      "grad_norm": 0.19098755717277527,
      "learning_rate": 9.241106723929641e-06,
      "loss": 0.0614,
      "step": 21880
    },
    {
      "epoch": 2.086639404296875e-05,
      "model_forward_time": 1.7346255779266357,
      "step": 21880
    },
    {
      "dynamics_loss_avg": 0.0523084744811058,
      "epoch": 2.086639404296875e-05,
      "step": 21880
    },
    {
      "action_loss_avg": 0.004166107409400865,
      "epoch": 2.086639404296875e-05,
      "step": 21880
    },
    {
      "epoch": 2.086639404296875e-05,
      "step": 21880,
      "training_step_time": 7.65974760055542
    },
    {
      "epoch": 2.0867347717285156e-05,
      "model_forward_time": 2.5409767627716064,
      "step": 21881
    },
    {
      "epoch": 2.0867347717285156e-05,
      "step": 21881,
      "training_step_time": 8.587276697158813
    },
    {
      "epoch": 2.0868301391601564e-05,
      "model_forward_time": 1.7415144443511963,
      "step": 21882
    },
    {
      "epoch": 2.0868301391601564e-05,
      "step": 21882,
      "training_step_time": 8.346391201019287
    },
    {
      "epoch": 2.0869255065917968e-05,
      "model_forward_time": 1.734328031539917,
      "step": 21883
    },
    {
      "epoch": 2.0869255065917968e-05,
      "step": 21883,
      "training_step_time": 8.677443265914917
    },
    {
      "epoch": 2.0870208740234375e-05,
      "model_forward_time": 1.7326924800872803,
      "step": 21884
    },
    {
      "epoch": 2.0870208740234375e-05,
      "step": 21884,
      "training_step_time": 7.574078321456909
    },
    {
      "epoch": 2.0871162414550783e-05,
      "model_forward_time": 1.7289040088653564,
      "step": 21885
    },
    {
      "epoch": 2.0871162414550783e-05,
      "step": 21885,
      "training_step_time": 6.67521071434021
    },
    {
      "epoch": 2.0872116088867187e-05,
      "model_forward_time": 1.724670648574829,
      "step": 21886
    },
    {
      "epoch": 2.0872116088867187e-05,
      "step": 21886,
      "training_step_time": 6.642784357070923
    },
    {
      "epoch": 2.0873069763183594e-05,
      "model_forward_time": 1.737945318222046,
      "step": 21887
    },
    {
      "epoch": 2.0873069763183594e-05,
      "step": 21887,
      "training_step_time": 6.742257833480835
    },
    {
      "epoch": 2.08740234375e-05,
      "model_forward_time": 1.7372703552246094,
      "step": 21888
    },
    {
      "epoch": 2.08740234375e-05,
      "step": 21888,
      "training_step_time": 6.710388422012329
    },
    {
      "epoch": 2.0874977111816406e-05,
      "model_forward_time": 1.7333612442016602,
      "step": 21889
    },
    {
      "epoch": 2.0874977111816406e-05,
      "step": 21889,
      "training_step_time": 7.6022422313690186
    },
    {
      "epoch": 2.0875930786132813e-05,
      "grad_norm": 0.24528199434280396,
      "learning_rate": 9.240230745136737e-06,
      "loss": 0.0655,
      "step": 21890
    },
    {
      "epoch": 2.0875930786132813e-05,
      "model_forward_time": 1.7526226043701172,
      "step": 21890
    },
    {
      "dynamics_loss_avg": 0.08389564305543899,
      "epoch": 2.0875930786132813e-05,
      "step": 21890
    },
    {
      "action_loss_avg": 0.003814438183326274,
      "epoch": 2.0875930786132813e-05,
      "step": 21890
    },
    {
      "epoch": 2.0875930786132813e-05,
      "step": 21890,
      "training_step_time": 9.072744131088257
    },
    {
      "epoch": 2.0876884460449217e-05,
      "model_forward_time": 1.7511787414550781,
      "step": 21891
    },
    {
      "epoch": 2.0876884460449217e-05,
      "step": 21891,
      "training_step_time": 7.750824689865112
    },
    {
      "epoch": 2.0877838134765624e-05,
      "model_forward_time": 1.7162704467773438,
      "step": 21892
    },
    {
      "epoch": 2.0877838134765624e-05,
      "step": 21892,
      "training_step_time": 8.837207555770874
    },
    {
      "epoch": 2.0878791809082032e-05,
      "model_forward_time": 1.7348439693450928,
      "step": 21893
    },
    {
      "epoch": 2.0878791809082032e-05,
      "step": 21893,
      "training_step_time": 7.6542370319366455
    },
    {
      "epoch": 2.0879745483398436e-05,
      "model_forward_time": 1.7281146049499512,
      "step": 21894
    },
    {
      "epoch": 2.0879745483398436e-05,
      "step": 21894,
      "training_step_time": 6.704376220703125
    },
    {
      "epoch": 2.0880699157714843e-05,
      "model_forward_time": 1.7171897888183594,
      "step": 21895
    },
    {
      "epoch": 2.0880699157714843e-05,
      "step": 21895,
      "training_step_time": 6.702022314071655
    },
    {
      "epoch": 2.088165283203125e-05,
      "model_forward_time": 1.7351164817810059,
      "step": 21896
    },
    {
      "epoch": 2.088165283203125e-05,
      "step": 21896,
      "training_step_time": 6.742520809173584
    },
    {
      "epoch": 2.0882606506347655e-05,
      "model_forward_time": 1.7158949375152588,
      "step": 21897
    },
    {
      "epoch": 2.0882606506347655e-05,
      "step": 21897,
      "training_step_time": 6.731720685958862
    },
    {
      "epoch": 2.0883560180664062e-05,
      "model_forward_time": 1.7334120273590088,
      "step": 21898
    },
    {
      "epoch": 2.0883560180664062e-05,
      "step": 21898,
      "training_step_time": 7.597885847091675
    },
    {
      "epoch": 2.088451385498047e-05,
      "model_forward_time": 1.7383339405059814,
      "step": 21899
    },
    {
      "epoch": 2.088451385498047e-05,
      "step": 21899,
      "training_step_time": 8.287816762924194
    },
    {
      "epoch": 2.0885467529296874e-05,
      "grad_norm": 0.28338509798049927,
      "learning_rate": 9.239354302638576e-06,
      "loss": 0.0626,
      "step": 21900
    },
    {
      "epoch": 2.0885467529296874e-05,
      "model_forward_time": 2.9017295837402344,
      "step": 21900
    },
    {
      "dynamics_loss_avg": 0.05943007431924343,
      "epoch": 2.0885467529296874e-05,
      "step": 21900
    },
    {
      "action_loss_avg": 0.00468659196048975,
      "epoch": 2.0885467529296874e-05,
      "step": 21900
    },
    {
      "epoch": 2.0885467529296874e-05,
      "step": 21900,
      "training_step_time": 7.759960889816284
    },
    {
      "epoch": 2.088642120361328e-05,
      "model_forward_time": 1.7294502258300781,
      "step": 21901
    },
    {
      "epoch": 2.088642120361328e-05,
      "step": 21901,
      "training_step_time": 8.601263761520386
    },
    {
      "epoch": 2.088737487792969e-05,
      "model_forward_time": 1.73081374168396,
      "step": 21902
    },
    {
      "epoch": 2.088737487792969e-05,
      "step": 21902,
      "training_step_time": 7.758899450302124
    },
    {
      "epoch": 2.0888328552246093e-05,
      "model_forward_time": 1.7278785705566406,
      "step": 21903
    },
    {
      "epoch": 2.0888328552246093e-05,
      "step": 21903,
      "training_step_time": 6.723507881164551
    },
    {
      "epoch": 2.08892822265625e-05,
      "model_forward_time": 1.7266154289245605,
      "step": 21904
    },
    {
      "epoch": 2.08892822265625e-05,
      "step": 21904,
      "training_step_time": 6.715574502944946
    },
    {
      "epoch": 2.0890235900878907e-05,
      "model_forward_time": 1.7287554740905762,
      "step": 21905
    },
    {
      "epoch": 2.0890235900878907e-05,
      "step": 21905,
      "training_step_time": 6.741023302078247
    },
    {
      "epoch": 2.089118957519531e-05,
      "model_forward_time": 1.7294657230377197,
      "step": 21906
    },
    {
      "epoch": 2.089118957519531e-05,
      "step": 21906,
      "training_step_time": 6.725175142288208
    },
    {
      "epoch": 2.089214324951172e-05,
      "model_forward_time": 1.7214808464050293,
      "step": 21907
    },
    {
      "epoch": 2.089214324951172e-05,
      "step": 21907,
      "training_step_time": 7.6014790534973145
    },
    {
      "epoch": 2.0893096923828126e-05,
      "model_forward_time": 1.7313048839569092,
      "step": 21908
    },
    {
      "epoch": 2.0893096923828126e-05,
      "step": 21908,
      "training_step_time": 7.785505533218384
    },
    {
      "epoch": 2.089405059814453e-05,
      "model_forward_time": 2.9136457443237305,
      "step": 21909
    },
    {
      "epoch": 2.089405059814453e-05,
      "step": 21909,
      "training_step_time": 10.897159099578857
    },
    {
      "epoch": 2.0895004272460938e-05,
      "grad_norm": 0.23393002152442932,
      "learning_rate": 9.238477396531e-06,
      "loss": 0.0679,
      "step": 21910
    },
    {
      "epoch": 2.0895004272460938e-05,
      "model_forward_time": 1.726210594177246,
      "step": 21910
    },
    {
      "dynamics_loss_avg": 0.057669727504253386,
      "epoch": 2.0895004272460938e-05,
      "step": 21910
    },
    {
      "action_loss_avg": 0.002328389947069809,
      "epoch": 2.0895004272460938e-05,
      "step": 21910
    },
    {
      "epoch": 2.0895004272460938e-05,
      "step": 21910,
      "training_step_time": 7.782448053359985
    },
    {
      "epoch": 2.0895957946777345e-05,
      "model_forward_time": 1.726032018661499,
      "step": 21911
    },
    {
      "epoch": 2.0895957946777345e-05,
      "step": 21911,
      "training_step_time": 8.731051683425903
    },
    {
      "epoch": 2.089691162109375e-05,
      "model_forward_time": 1.7249441146850586,
      "step": 21912
    },
    {
      "epoch": 2.089691162109375e-05,
      "step": 21912,
      "training_step_time": 6.694854021072388
    },
    {
      "epoch": 2.0897865295410156e-05,
      "model_forward_time": 1.724888563156128,
      "step": 21913
    },
    {
      "epoch": 2.0897865295410156e-05,
      "step": 21913,
      "training_step_time": 6.697934627532959
    },
    {
      "epoch": 2.0898818969726564e-05,
      "model_forward_time": 1.7243077754974365,
      "step": 21914
    },
    {
      "epoch": 2.0898818969726564e-05,
      "step": 21914,
      "training_step_time": 6.707164287567139
    },
    {
      "epoch": 2.0899772644042968e-05,
      "model_forward_time": 1.724869728088379,
      "step": 21915
    },
    {
      "epoch": 2.0899772644042968e-05,
      "step": 21915,
      "training_step_time": 6.677671194076538
    },
    {
      "epoch": 2.0900726318359375e-05,
      "model_forward_time": 1.723412275314331,
      "step": 21916
    },
    {
      "epoch": 2.0900726318359375e-05,
      "step": 21916,
      "training_step_time": 7.609777927398682
    },
    {
      "epoch": 2.0901679992675783e-05,
      "model_forward_time": 1.7287778854370117,
      "step": 21917
    },
    {
      "epoch": 2.0901679992675783e-05,
      "step": 21917,
      "training_step_time": 7.721418142318726
    },
    {
      "epoch": 2.0902633666992187e-05,
      "model_forward_time": 2.908344030380249,
      "step": 21918
    },
    {
      "epoch": 2.0902633666992187e-05,
      "step": 21918,
      "training_step_time": 8.511046648025513
    },
    {
      "epoch": 2.0903587341308594e-05,
      "model_forward_time": 1.7121288776397705,
      "step": 21919
    },
    {
      "epoch": 2.0903587341308594e-05,
      "step": 21919,
      "training_step_time": 7.720388650894165
    },
    {
      "epoch": 2.0904541015625e-05,
      "grad_norm": 0.19856694340705872,
      "learning_rate": 9.23760002690991e-06,
      "loss": 0.0673,
      "step": 21920
    },
    {
      "epoch": 2.0904541015625e-05,
      "model_forward_time": 1.7082479000091553,
      "step": 21920
    },
    {
      "dynamics_loss_avg": 0.05266171097755432,
      "epoch": 2.0904541015625e-05,
      "step": 21920
    },
    {
      "action_loss_avg": 0.0018025170778855682,
      "epoch": 2.0904541015625e-05,
      "step": 21920
    },
    {
      "epoch": 2.0904541015625e-05,
      "step": 21920,
      "training_step_time": 8.494790077209473
    },
    {
      "epoch": 2.0905494689941406e-05,
      "model_forward_time": 1.717841386795044,
      "step": 21921
    },
    {
      "epoch": 2.0905494689941406e-05,
      "step": 21921,
      "training_step_time": 7.5092995166778564
    },
    {
      "epoch": 2.0906448364257813e-05,
      "model_forward_time": 1.7256464958190918,
      "step": 21922
    },
    {
      "epoch": 2.0906448364257813e-05,
      "step": 21922,
      "training_step_time": 6.6615400314331055
    },
    {
      "epoch": 2.0907402038574217e-05,
      "model_forward_time": 1.7177734375,
      "step": 21923
    },
    {
      "epoch": 2.0907402038574217e-05,
      "step": 21923,
      "training_step_time": 6.719001770019531
    },
    {
      "epoch": 2.0908355712890625e-05,
      "model_forward_time": 1.7262253761291504,
      "step": 21924
    },
    {
      "epoch": 2.0908355712890625e-05,
      "step": 21924,
      "training_step_time": 6.721499919891357
    },
    {
      "epoch": 2.0909309387207032e-05,
      "model_forward_time": 1.7320764064788818,
      "step": 21925
    },
    {
      "epoch": 2.0909309387207032e-05,
      "step": 21925,
      "training_step_time": 7.610923528671265
    },
    {
      "epoch": 2.0910263061523436e-05,
      "model_forward_time": 1.7287170886993408,
      "step": 21926
    },
    {
      "epoch": 2.0910263061523436e-05,
      "step": 21926,
      "training_step_time": 7.653997182846069
    },
    {
      "epoch": 2.0911216735839843e-05,
      "model_forward_time": 2.8963701725006104,
      "step": 21927
    },
    {
      "epoch": 2.0911216735839843e-05,
      "step": 21927,
      "training_step_time": 7.702206611633301
    },
    {
      "epoch": 2.091217041015625e-05,
      "model_forward_time": 1.7263851165771484,
      "step": 21928
    },
    {
      "epoch": 2.091217041015625e-05,
      "step": 21928,
      "training_step_time": 8.76323914527893
    },
    {
      "epoch": 2.0913124084472655e-05,
      "model_forward_time": 1.7211735248565674,
      "step": 21929
    },
    {
      "epoch": 2.0913124084472655e-05,
      "step": 21929,
      "training_step_time": 7.69133448600769
    },
    {
      "epoch": 2.0914077758789062e-05,
      "grad_norm": 0.16270627081394196,
      "learning_rate": 9.236722193871252e-06,
      "loss": 0.0648,
      "step": 21930
    },
    {
      "epoch": 2.0914077758789062e-05,
      "model_forward_time": 1.7168314456939697,
      "step": 21930
    },
    {
      "dynamics_loss_avg": 0.07010355573147535,
      "epoch": 2.0914077758789062e-05,
      "step": 21930
    },
    {
      "action_loss_avg": 0.00312411361373961,
      "epoch": 2.0914077758789062e-05,
      "step": 21930
    },
    {
      "epoch": 2.0914077758789062e-05,
      "step": 21930,
      "training_step_time": 7.631651163101196
    },
    {
      "epoch": 2.091503143310547e-05,
      "model_forward_time": 1.728652000427246,
      "step": 21931
    },
    {
      "epoch": 2.091503143310547e-05,
      "step": 21931,
      "training_step_time": 6.705959320068359
    },
    {
      "epoch": 2.0915985107421874e-05,
      "model_forward_time": 1.7241301536560059,
      "step": 21932
    },
    {
      "epoch": 2.0915985107421874e-05,
      "step": 21932,
      "training_step_time": 6.681926012039185
    },
    {
      "epoch": 2.091693878173828e-05,
      "model_forward_time": 1.7278048992156982,
      "step": 21933
    },
    {
      "epoch": 2.091693878173828e-05,
      "step": 21933,
      "training_step_time": 6.708579778671265
    },
    {
      "epoch": 2.091789245605469e-05,
      "model_forward_time": 1.3581666946411133,
      "step": 21934
    },
    {
      "epoch": 2.091789245605469e-05,
      "step": 21934,
      "training_step_time": 7.572242736816406
    },
    {
      "epoch": 2.0918846130371093e-05,
      "model_forward_time": 1.7191553115844727,
      "step": 21935
    },
    {
      "epoch": 2.0918846130371093e-05,
      "step": 21935,
      "training_step_time": 8.160456657409668
    },
    {
      "epoch": 2.09197998046875e-05,
      "model_forward_time": 2.8067879676818848,
      "step": 21936
    },
    {
      "epoch": 2.09197998046875e-05,
      "step": 21936,
      "training_step_time": 8.252042293548584
    },
    {
      "epoch": 2.0920753479003907e-05,
      "model_forward_time": 1.746528148651123,
      "step": 21937
    },
    {
      "epoch": 2.0920753479003907e-05,
      "step": 21937,
      "training_step_time": 9.794903755187988
    },
    {
      "epoch": 2.092170715332031e-05,
      "model_forward_time": 1.7245891094207764,
      "step": 21938
    },
    {
      "epoch": 2.092170715332031e-05,
      "step": 21938,
      "training_step_time": 8.517534971237183
    },
    {
      "epoch": 2.092266082763672e-05,
      "model_forward_time": 1.7282543182373047,
      "step": 21939
    },
    {
      "epoch": 2.092266082763672e-05,
      "step": 21939,
      "training_step_time": 7.794842481613159
    },
    {
      "epoch": 2.0923614501953126e-05,
      "grad_norm": 0.2240191400051117,
      "learning_rate": 9.235843897511023e-06,
      "loss": 0.058,
      "step": 21940
    },
    {
      "epoch": 2.0923614501953126e-05,
      "model_forward_time": 1.7258810997009277,
      "step": 21940
    },
    {
      "dynamics_loss_avg": 0.07002397775650024,
      "epoch": 2.0923614501953126e-05,
      "step": 21940
    },
    {
      "action_loss_avg": 0.0030677157337777317,
      "epoch": 2.0923614501953126e-05,
      "step": 21940
    },
    {
      "epoch": 2.0923614501953126e-05,
      "step": 21940,
      "training_step_time": 6.758620977401733
    },
    {
      "epoch": 2.092456817626953e-05,
      "model_forward_time": 1.7221736907958984,
      "step": 21941
    },
    {
      "epoch": 2.092456817626953e-05,
      "step": 21941,
      "training_step_time": 6.7358479499816895
    },
    {
      "epoch": 2.0925521850585938e-05,
      "model_forward_time": 1.7223036289215088,
      "step": 21942
    },
    {
      "epoch": 2.0925521850585938e-05,
      "step": 21942,
      "training_step_time": 6.72143292427063
    },
    {
      "epoch": 2.0926475524902345e-05,
      "model_forward_time": 1.71934175491333,
      "step": 21943
    },
    {
      "epoch": 2.0926475524902345e-05,
      "step": 21943,
      "training_step_time": 7.6021223068237305
    },
    {
      "epoch": 2.092742919921875e-05,
      "model_forward_time": 1.718458652496338,
      "step": 21944
    },
    {
      "epoch": 2.092742919921875e-05,
      "step": 21944,
      "training_step_time": 8.218846559524536
    },
    {
      "epoch": 2.0928382873535157e-05,
      "model_forward_time": 1.7056233882904053,
      "step": 21945
    },
    {
      "epoch": 2.0928382873535157e-05,
      "step": 21945,
      "training_step_time": 7.558730840682983
    },
    {
      "epoch": 2.0929336547851564e-05,
      "model_forward_time": 2.868990659713745,
      "step": 21946
    },
    {
      "epoch": 2.0929336547851564e-05,
      "step": 21946,
      "training_step_time": 8.501059293746948
    },
    {
      "epoch": 2.0930290222167968e-05,
      "model_forward_time": 1.7178630828857422,
      "step": 21947
    },
    {
      "epoch": 2.0930290222167968e-05,
      "step": 21947,
      "training_step_time": 10.777167320251465
    },
    {
      "epoch": 2.0931243896484375e-05,
      "model_forward_time": 1.7058439254760742,
      "step": 21948
    },
    {
      "epoch": 2.0931243896484375e-05,
      "step": 21948,
      "training_step_time": 7.738834857940674
    },
    {
      "epoch": 2.0932197570800783e-05,
      "model_forward_time": 1.7114489078521729,
      "step": 21949
    },
    {
      "epoch": 2.0932197570800783e-05,
      "step": 21949,
      "training_step_time": 7.430409669876099
    },
    {
      "epoch": 2.0933151245117187e-05,
      "grad_norm": 0.17494991421699524,
      "learning_rate": 9.234965137925277e-06,
      "loss": 0.0634,
      "step": 21950
    },
    {
      "epoch": 2.0933151245117187e-05,
      "model_forward_time": 1.7139508724212646,
      "step": 21950
    },
    {
      "dynamics_loss_avg": 0.05258917063474655,
      "epoch": 2.0933151245117187e-05,
      "step": 21950
    },
    {
      "action_loss_avg": 0.0017867486167233437,
      "epoch": 2.0933151245117187e-05,
      "step": 21950
    },
    {
      "epoch": 2.0933151245117187e-05,
      "step": 21950,
      "training_step_time": 6.717090606689453
    },
    {
      "epoch": 2.0934104919433594e-05,
      "model_forward_time": 1.7235565185546875,
      "step": 21951
    },
    {
      "epoch": 2.0934104919433594e-05,
      "step": 21951,
      "training_step_time": 6.729650497436523
    },
    {
      "epoch": 2.093505859375e-05,
      "model_forward_time": 1.7265417575836182,
      "step": 21952
    },
    {
      "epoch": 2.093505859375e-05,
      "step": 21952,
      "training_step_time": 7.5781025886535645
    },
    {
      "epoch": 2.0936012268066406e-05,
      "model_forward_time": 1.721820592880249,
      "step": 21953
    },
    {
      "epoch": 2.0936012268066406e-05,
      "step": 21953,
      "training_step_time": 7.516904592514038
    },
    {
      "epoch": 2.0936965942382813e-05,
      "model_forward_time": 1.7233490943908691,
      "step": 21954
    },
    {
      "epoch": 2.0936965942382813e-05,
      "step": 21954,
      "training_step_time": 8.333223342895508
    },
    {
      "epoch": 2.0937919616699217e-05,
      "model_forward_time": 2.8931431770324707,
      "step": 21955
    },
    {
      "epoch": 2.0937919616699217e-05,
      "step": 21955,
      "training_step_time": 8.850852966308594
    },
    {
      "epoch": 2.0938873291015625e-05,
      "model_forward_time": 1.7203378677368164,
      "step": 21956
    },
    {
      "epoch": 2.0938873291015625e-05,
      "step": 21956,
      "training_step_time": 8.423908710479736
    },
    {
      "epoch": 2.0939826965332032e-05,
      "model_forward_time": 1.7240242958068848,
      "step": 21957
    },
    {
      "epoch": 2.0939826965332032e-05,
      "step": 21957,
      "training_step_time": 8.251652002334595
    },
    {
      "epoch": 2.0940780639648436e-05,
      "model_forward_time": 1.7242066860198975,
      "step": 21958
    },
    {
      "epoch": 2.0940780639648436e-05,
      "step": 21958,
      "training_step_time": 7.476142168045044
    },
    {
      "epoch": 2.0941734313964843e-05,
      "model_forward_time": 1.7251441478729248,
      "step": 21959
    },
    {
      "epoch": 2.0941734313964843e-05,
      "step": 21959,
      "training_step_time": 7.469478607177734
    },
    {
      "epoch": 2.094268798828125e-05,
      "grad_norm": 0.20221538841724396,
      "learning_rate": 9.234085915210108e-06,
      "loss": 0.0617,
      "step": 21960
    },
    {
      "epoch": 2.094268798828125e-05,
      "model_forward_time": 1.713078260421753,
      "step": 21960
    },
    {
      "dynamics_loss_avg": 0.05702372118830681,
      "epoch": 2.094268798828125e-05,
      "step": 21960
    },
    {
      "action_loss_avg": 0.003276975074550137,
      "epoch": 2.094268798828125e-05,
      "step": 21960
    },
    {
      "epoch": 2.094268798828125e-05,
      "step": 21960,
      "training_step_time": 6.741980314254761
    },
    {
      "epoch": 2.0943641662597655e-05,
      "model_forward_time": 1.7212128639221191,
      "step": 21961
    },
    {
      "epoch": 2.0943641662597655e-05,
      "step": 21961,
      "training_step_time": 7.551286220550537
    },
    {
      "epoch": 2.0944595336914062e-05,
      "model_forward_time": 1.015718698501587,
      "step": 21962
    },
    {
      "epoch": 2.0944595336914062e-05,
      "step": 21962,
      "training_step_time": 7.678894996643066
    },
    {
      "epoch": 2.094554901123047e-05,
      "model_forward_time": 1.7130866050720215,
      "step": 21963
    },
    {
      "epoch": 2.094554901123047e-05,
      "step": 21963,
      "training_step_time": 8.241101503372192
    },
    {
      "epoch": 2.0946502685546874e-05,
      "model_forward_time": 2.8941452503204346,
      "step": 21964
    },
    {
      "epoch": 2.0946502685546874e-05,
      "step": 21964,
      "training_step_time": 8.836307287216187
    },
    {
      "epoch": 2.094745635986328e-05,
      "model_forward_time": 1.7211642265319824,
      "step": 21965
    },
    {
      "epoch": 2.094745635986328e-05,
      "step": 21965,
      "training_step_time": 8.695559740066528
    },
    {
      "epoch": 2.094841003417969e-05,
      "model_forward_time": 1.723722219467163,
      "step": 21966
    },
    {
      "epoch": 2.094841003417969e-05,
      "step": 21966,
      "training_step_time": 8.73454999923706
    },
    {
      "epoch": 2.0949363708496093e-05,
      "model_forward_time": 1.7164297103881836,
      "step": 21967
    },
    {
      "epoch": 2.0949363708496093e-05,
      "step": 21967,
      "training_step_time": 7.613651752471924
    },
    {
      "epoch": 2.09503173828125e-05,
      "model_forward_time": 1.7185003757476807,
      "step": 21968
    },
    {
      "epoch": 2.09503173828125e-05,
      "step": 21968,
      "training_step_time": 7.491254806518555
    },
    {
      "epoch": 2.0951271057128907e-05,
      "model_forward_time": 1.7116811275482178,
      "step": 21969
    },
    {
      "epoch": 2.0951271057128907e-05,
      "step": 21969,
      "training_step_time": 6.695911407470703
    },
    {
      "epoch": 2.095222473144531e-05,
      "grad_norm": 0.1289885938167572,
      "learning_rate": 9.233206229461669e-06,
      "loss": 0.0628,
      "step": 21970
    },
    {
      "epoch": 2.095222473144531e-05,
      "model_forward_time": 1.7108407020568848,
      "step": 21970
    },
    {
      "dynamics_loss_avg": 0.05870303940027952,
      "epoch": 2.095222473144531e-05,
      "step": 21970
    },
    {
      "action_loss_avg": 0.0036608212161809207,
      "epoch": 2.095222473144531e-05,
      "step": 21970
    },
    {
      "epoch": 2.095222473144531e-05,
      "step": 21970,
      "training_step_time": 7.5789995193481445
    },
    {
      "epoch": 2.095317840576172e-05,
      "model_forward_time": 1.7239155769348145,
      "step": 21971
    },
    {
      "epoch": 2.095317840576172e-05,
      "step": 21971,
      "training_step_time": 7.611993312835693
    },
    {
      "epoch": 2.0954132080078126e-05,
      "model_forward_time": 1.7087352275848389,
      "step": 21972
    },
    {
      "epoch": 2.0954132080078126e-05,
      "step": 21972,
      "training_step_time": 7.628010034561157
    },
    {
      "epoch": 2.095508575439453e-05,
      "model_forward_time": 1.7245759963989258,
      "step": 21973
    },
    {
      "epoch": 2.095508575439453e-05,
      "step": 21973,
      "training_step_time": 8.16623044013977
    },
    {
      "epoch": 2.0956039428710938e-05,
      "model_forward_time": 1.7366507053375244,
      "step": 21974
    },
    {
      "epoch": 2.0956039428710938e-05,
      "step": 21974,
      "training_step_time": 7.763923645019531
    },
    {
      "epoch": 2.0956993103027345e-05,
      "model_forward_time": 1.7219951152801514,
      "step": 21975
    },
    {
      "epoch": 2.0956993103027345e-05,
      "step": 21975,
      "training_step_time": 8.491996765136719
    },
    {
      "epoch": 2.095794677734375e-05,
      "model_forward_time": 1.7294607162475586,
      "step": 21976
    },
    {
      "epoch": 2.095794677734375e-05,
      "step": 21976,
      "training_step_time": 7.749248266220093
    },
    {
      "epoch": 2.0958900451660157e-05,
      "model_forward_time": 1.7160403728485107,
      "step": 21977
    },
    {
      "epoch": 2.0958900451660157e-05,
      "step": 21977,
      "training_step_time": 7.471948862075806
    },
    {
      "epoch": 2.0959854125976564e-05,
      "model_forward_time": 1.7250535488128662,
      "step": 21978
    },
    {
      "epoch": 2.0959854125976564e-05,
      "step": 21978,
      "training_step_time": 6.700603723526001
    },
    {
      "epoch": 2.0960807800292968e-05,
      "model_forward_time": 1.7284386157989502,
      "step": 21979
    },
    {
      "epoch": 2.0960807800292968e-05,
      "step": 21979,
      "training_step_time": 7.583039999008179
    },
    {
      "epoch": 2.0961761474609375e-05,
      "grad_norm": 0.1937609314918518,
      "learning_rate": 9.232326080776165e-06,
      "loss": 0.0594,
      "step": 21980
    },
    {
      "epoch": 2.0961761474609375e-05,
      "model_forward_time": 1.7268202304840088,
      "step": 21980
    },
    {
      "dynamics_loss_avg": 0.06439529992640018,
      "epoch": 2.0961761474609375e-05,
      "step": 21980
    },
    {
      "action_loss_avg": 0.002692747407127172,
      "epoch": 2.0961761474609375e-05,
      "step": 21980
    },
    {
      "epoch": 2.0961761474609375e-05,
      "step": 21980,
      "training_step_time": 7.568640947341919
    },
    {
      "epoch": 2.0962715148925783e-05,
      "model_forward_time": 1.7280912399291992,
      "step": 21981
    },
    {
      "epoch": 2.0962715148925783e-05,
      "step": 21981,
      "training_step_time": 7.670966863632202
    },
    {
      "epoch": 2.0963668823242187e-05,
      "model_forward_time": 1.7318391799926758,
      "step": 21982
    },
    {
      "epoch": 2.0963668823242187e-05,
      "step": 21982,
      "training_step_time": 7.692165374755859
    },
    {
      "epoch": 2.0964622497558594e-05,
      "model_forward_time": 2.9593591690063477,
      "step": 21983
    },
    {
      "epoch": 2.0964622497558594e-05,
      "step": 21983,
      "training_step_time": 7.698303461074829
    },
    {
      "epoch": 2.0965576171875e-05,
      "model_forward_time": 1.7310359477996826,
      "step": 21984
    },
    {
      "epoch": 2.0965576171875e-05,
      "step": 21984,
      "training_step_time": 8.393352508544922
    },
    {
      "epoch": 2.0966529846191406e-05,
      "model_forward_time": 1.7469501495361328,
      "step": 21985
    },
    {
      "epoch": 2.0966529846191406e-05,
      "step": 21985,
      "training_step_time": 9.496191263198853
    },
    {
      "epoch": 2.0967483520507813e-05,
      "model_forward_time": 1.721634864807129,
      "step": 21986
    },
    {
      "epoch": 2.0967483520507813e-05,
      "step": 21986,
      "training_step_time": 7.443856716156006
    },
    {
      "epoch": 2.0968437194824217e-05,
      "model_forward_time": 1.7291665077209473,
      "step": 21987
    },
    {
      "epoch": 2.0968437194824217e-05,
      "step": 21987,
      "training_step_time": 7.435397624969482
    },
    {
      "epoch": 2.0969390869140625e-05,
      "model_forward_time": 1.7345366477966309,
      "step": 21988
    },
    {
      "epoch": 2.0969390869140625e-05,
      "step": 21988,
      "training_step_time": 7.041171312332153
    },
    {
      "epoch": 2.0970344543457032e-05,
      "model_forward_time": 1.7391819953918457,
      "step": 21989
    },
    {
      "epoch": 2.0970344543457032e-05,
      "step": 21989,
      "training_step_time": 7.5954060554504395
    },
    {
      "epoch": 2.0971298217773436e-05,
      "grad_norm": 0.11471705883741379,
      "learning_rate": 9.23144546924984e-06,
      "loss": 0.0608,
      "step": 21990
    },
    {
      "epoch": 2.0971298217773436e-05,
      "model_forward_time": 1.7305171489715576,
      "step": 21990
    },
    {
      "dynamics_loss_avg": 0.050553133524954316,
      "epoch": 2.0971298217773436e-05,
      "step": 21990
    },
    {
      "action_loss_avg": 0.0022239373240154235,
      "epoch": 2.0971298217773436e-05,
      "step": 21990
    },
    {
      "epoch": 2.0971298217773436e-05,
      "step": 21990,
      "training_step_time": 7.682009696960449
    },
    {
      "epoch": 2.0972251892089843e-05,
      "model_forward_time": 1.7409160137176514,
      "step": 21991
    },
    {
      "epoch": 2.0972251892089843e-05,
      "step": 21991,
      "training_step_time": 7.774671316146851
    },
    {
      "epoch": 2.097320556640625e-05,
      "model_forward_time": 2.921388626098633,
      "step": 21992
    },
    {
      "epoch": 2.097320556640625e-05,
      "step": 21992,
      "training_step_time": 8.178258895874023
    },
    {
      "epoch": 2.0974159240722655e-05,
      "model_forward_time": 1.7358596324920654,
      "step": 21993
    },
    {
      "epoch": 2.0974159240722655e-05,
      "step": 21993,
      "training_step_time": 9.755085945129395
    },
    {
      "epoch": 2.0975112915039062e-05,
      "model_forward_time": 1.7429335117340088,
      "step": 21994
    },
    {
      "epoch": 2.0975112915039062e-05,
      "step": 21994,
      "training_step_time": 7.638427734375
    },
    {
      "epoch": 2.097606658935547e-05,
      "model_forward_time": 1.7181305885314941,
      "step": 21995
    },
    {
      "epoch": 2.097606658935547e-05,
      "step": 21995,
      "training_step_time": 7.642485618591309
    },
    {
      "epoch": 2.0977020263671874e-05,
      "model_forward_time": 1.7358031272888184,
      "step": 21996
    },
    {
      "epoch": 2.0977020263671874e-05,
      "step": 21996,
      "training_step_time": 7.542538404464722
    },
    {
      "epoch": 2.097797393798828e-05,
      "model_forward_time": 1.062911033630371,
      "step": 21997
    },
    {
      "epoch": 2.097797393798828e-05,
      "step": 21997,
      "training_step_time": 7.098416566848755
    },
    {
      "epoch": 2.097892761230469e-05,
      "model_forward_time": 1.7659449577331543,
      "step": 21998
    },
    {
      "epoch": 2.097892761230469e-05,
      "step": 21998,
      "training_step_time": 7.900249481201172
    },
    {
      "epoch": 2.0979881286621093e-05,
      "model_forward_time": 1.7325549125671387,
      "step": 21999
    },
    {
      "epoch": 2.0979881286621093e-05,
      "step": 21999,
      "training_step_time": 7.792229890823364
    },
    {
      "epoch": 2.09808349609375e-05,
      "grad_norm": 0.17838390171527863,
      "learning_rate": 9.230564394979001e-06,
      "loss": 0.0662,
      "step": 22000
    },
    {
      "epoch": 2.09808349609375e-05,
      "model_forward_time": 1.3853132724761963,
      "step": 22000
    },
    {
      "dynamics_loss_avg": 0.07253473959863185,
      "epoch": 2.09808349609375e-05,
      "step": 22000
    },
    {
      "action_loss_avg": 0.003900536516448483,
      "epoch": 2.09808349609375e-05,
      "step": 22000
    },
    {
      "epoch": 2.09808349609375e-05,
      "step": 22000,
      "training_step_time": 8.437721490859985
    },
    {
      "epoch": 2.0981788635253907e-05,
      "model_forward_time": 3.379223585128784,
      "step": 22001
    },
    {
      "epoch": 2.0981788635253907e-05,
      "step": 22001,
      "training_step_time": 12.344842195510864
    },
    {
      "epoch": 2.098274230957031e-05,
      "model_forward_time": 1.8634047508239746,
      "step": 22002
    },
    {
      "epoch": 2.098274230957031e-05,
      "step": 22002,
      "training_step_time": 10.375491380691528
    },
    {
      "epoch": 2.098369598388672e-05,
      "model_forward_time": 1.7670791149139404,
      "step": 22003
    },
    {
      "epoch": 2.098369598388672e-05,
      "step": 22003,
      "training_step_time": 9.697852611541748
    },
    {
      "epoch": 2.0984649658203126e-05,
      "model_forward_time": 1.7635173797607422,
      "step": 22004
    },
    {
      "epoch": 2.0984649658203126e-05,
      "step": 22004,
      "training_step_time": 7.829902410507202
    },
    {
      "epoch": 2.098560333251953e-05,
      "model_forward_time": 1.7729904651641846,
      "step": 22005
    },
    {
      "epoch": 2.098560333251953e-05,
      "step": 22005,
      "training_step_time": 7.5764923095703125
    },
    {
      "epoch": 2.0986557006835938e-05,
      "model_forward_time": 1.764685869216919,
      "step": 22006
    },
    {
      "epoch": 2.0986557006835938e-05,
      "step": 22006,
      "training_step_time": 7.106850862503052
    },
    {
      "epoch": 2.0987510681152345e-05,
      "model_forward_time": 1.7642455101013184,
      "step": 22007
    },
    {
      "epoch": 2.0987510681152345e-05,
      "step": 22007,
      "training_step_time": 8.56898283958435
    },
    {
      "epoch": 2.098846435546875e-05,
      "model_forward_time": 1.770716667175293,
      "step": 22008
    },
    {
      "epoch": 2.098846435546875e-05,
      "step": 22008,
      "training_step_time": 9.553025960922241
    },
    {
      "epoch": 2.0989418029785157e-05,
      "model_forward_time": 1.767939567565918,
      "step": 22009
    },
    {
      "epoch": 2.0989418029785157e-05,
      "step": 22009,
      "training_step_time": 8.73820948600769
    },
    {
      "epoch": 2.0990371704101564e-05,
      "grad_norm": 0.1887679100036621,
      "learning_rate": 9.229682858060001e-06,
      "loss": 0.0604,
      "step": 22010
    },
    {
      "epoch": 2.0990371704101564e-05,
      "model_forward_time": 2.9314169883728027,
      "step": 22010
    },
    {
      "dynamics_loss_avg": 0.053699250891804697,
      "epoch": 2.0990371704101564e-05,
      "step": 22010
    },
    {
      "action_loss_avg": 0.003240650426596403,
      "epoch": 2.0990371704101564e-05,
      "step": 22010
    },
    {
      "epoch": 2.0990371704101564e-05,
      "step": 22010,
      "training_step_time": 7.826223611831665
    },
    {
      "epoch": 2.0991325378417968e-05,
      "model_forward_time": 1.8606703281402588,
      "step": 22011
    },
    {
      "epoch": 2.0991325378417968e-05,
      "step": 22011,
      "training_step_time": 10.150866508483887
    },
    {
      "epoch": 2.0992279052734375e-05,
      "model_forward_time": 1.765202283859253,
      "step": 22012
    },
    {
      "epoch": 2.0992279052734375e-05,
      "step": 22012,
      "training_step_time": 7.707089424133301
    },
    {
      "epoch": 2.0993232727050783e-05,
      "model_forward_time": 1.0711820125579834,
      "step": 22013
    },
    {
      "epoch": 2.0993232727050783e-05,
      "step": 22013,
      "training_step_time": 8.37089991569519
    },
    {
      "epoch": 2.0994186401367187e-05,
      "model_forward_time": 1.7715401649475098,
      "step": 22014
    },
    {
      "epoch": 2.0994186401367187e-05,
      "step": 22014,
      "training_step_time": 7.448721647262573
    },
    {
      "epoch": 2.0995140075683594e-05,
      "model_forward_time": 1.7635159492492676,
      "step": 22015
    },
    {
      "epoch": 2.0995140075683594e-05,
      "step": 22015,
      "training_step_time": 7.041247844696045
    },
    {
      "epoch": 2.099609375e-05,
      "model_forward_time": 1.7621994018554688,
      "step": 22016
    },
    {
      "epoch": 2.099609375e-05,
      "step": 22016,
      "training_step_time": 7.602805852890015
    },
    {
      "epoch": 2.0997047424316406e-05,
      "model_forward_time": 1.7798547744750977,
      "step": 22017
    },
    {
      "epoch": 2.0997047424316406e-05,
      "step": 22017,
      "training_step_time": 7.995763301849365
    },
    {
      "epoch": 2.0998001098632813e-05,
      "model_forward_time": 1.7641019821166992,
      "step": 22018
    },
    {
      "epoch": 2.0998001098632813e-05,
      "step": 22018,
      "training_step_time": 7.579211235046387
    },
    {
      "epoch": 2.0998954772949217e-05,
      "model_forward_time": 1.7666974067687988,
      "step": 22019
    },
    {
      "epoch": 2.0998954772949217e-05,
      "step": 22019,
      "training_step_time": 8.895706176757812
    },
    {
      "epoch": 2.0999908447265625e-05,
      "grad_norm": 0.14869649708271027,
      "learning_rate": 9.228800858589242e-06,
      "loss": 0.0645,
      "step": 22020
    },
    {
      "epoch": 2.0999908447265625e-05,
      "model_forward_time": 2.890840768814087,
      "step": 22020
    },
    {
      "dynamics_loss_avg": 0.061653312481939795,
      "epoch": 2.0999908447265625e-05,
      "step": 22020
    },
    {
      "action_loss_avg": 0.003042091196402907,
      "epoch": 2.0999908447265625e-05,
      "step": 22020
    },
    {
      "epoch": 2.0999908447265625e-05,
      "step": 22020,
      "training_step_time": 7.8156514167785645
    },
    {
      "epoch": 2.1000862121582032e-05,
      "model_forward_time": 1.7628047466278076,
      "step": 22021
    },
    {
      "epoch": 2.1000862121582032e-05,
      "step": 22021,
      "training_step_time": 7.743762254714966
    },
    {
      "epoch": 2.1001815795898436e-05,
      "model_forward_time": 1.7531144618988037,
      "step": 22022
    },
    {
      "epoch": 2.1001815795898436e-05,
      "step": 22022,
      "training_step_time": 9.4714195728302
    },
    {
      "epoch": 2.1002769470214843e-05,
      "model_forward_time": 1.75325608253479,
      "step": 22023
    },
    {
      "epoch": 2.1002769470214843e-05,
      "step": 22023,
      "training_step_time": 7.79940128326416
    },
    {
      "epoch": 2.100372314453125e-05,
      "model_forward_time": 1.7477104663848877,
      "step": 22024
    },
    {
      "epoch": 2.100372314453125e-05,
      "step": 22024,
      "training_step_time": 6.878866195678711
    },
    {
      "epoch": 2.1004676818847655e-05,
      "model_forward_time": 1.7311663627624512,
      "step": 22025
    },
    {
      "epoch": 2.1004676818847655e-05,
      "step": 22025,
      "training_step_time": 7.600260496139526
    },
    {
      "epoch": 2.1005630493164062e-05,
      "model_forward_time": 1.3754236698150635,
      "step": 22026
    },
    {
      "epoch": 2.1005630493164062e-05,
      "step": 22026,
      "training_step_time": 7.497949600219727
    },
    {
      "epoch": 2.100658416748047e-05,
      "model_forward_time": 1.7560369968414307,
      "step": 22027
    },
    {
      "epoch": 2.100658416748047e-05,
      "step": 22027,
      "training_step_time": 8.692659616470337
    },
    {
      "epoch": 2.1007537841796874e-05,
      "model_forward_time": 0.7392430305480957,
      "step": 22028
    },
    {
      "epoch": 2.1007537841796874e-05,
      "step": 22028,
      "training_step_time": 8.65384554862976
    },
    {
      "epoch": 2.100849151611328e-05,
      "model_forward_time": 2.918795585632324,
      "step": 22029
    },
    {
      "epoch": 2.100849151611328e-05,
      "step": 22029,
      "training_step_time": 9.677129745483398
    },
    {
      "epoch": 2.100944519042969e-05,
      "grad_norm": 0.20506037771701813,
      "learning_rate": 9.22791839666318e-06,
      "loss": 0.0658,
      "step": 22030
    },
    {
      "epoch": 2.100944519042969e-05,
      "model_forward_time": 1.7525665760040283,
      "step": 22030
    },
    {
      "dynamics_loss_avg": 0.07394315302371979,
      "epoch": 2.100944519042969e-05,
      "step": 22030
    },
    {
      "action_loss_avg": 0.0043757286970503625,
      "epoch": 2.100944519042969e-05,
      "step": 22030
    },
    {
      "epoch": 2.100944519042969e-05,
      "step": 22030,
      "training_step_time": 8.767018556594849
    },
    {
      "epoch": 2.1010398864746093e-05,
      "model_forward_time": 1.7460711002349854,
      "step": 22031
    },
    {
      "epoch": 2.1010398864746093e-05,
      "step": 22031,
      "training_step_time": 8.677180767059326
    },
    {
      "epoch": 2.10113525390625e-05,
      "model_forward_time": 1.753568172454834,
      "step": 22032
    },
    {
      "epoch": 2.10113525390625e-05,
      "step": 22032,
      "training_step_time": 7.768754959106445
    },
    {
      "epoch": 2.1012306213378907e-05,
      "model_forward_time": 1.7625887393951416,
      "step": 22033
    },
    {
      "epoch": 2.1012306213378907e-05,
      "step": 22033,
      "training_step_time": 6.822161436080933
    },
    {
      "epoch": 2.101325988769531e-05,
      "model_forward_time": 1.7352488040924072,
      "step": 22034
    },
    {
      "epoch": 2.101325988769531e-05,
      "step": 22034,
      "training_step_time": 7.046985149383545
    },
    {
      "epoch": 2.101421356201172e-05,
      "model_forward_time": 1.7534945011138916,
      "step": 22035
    },
    {
      "epoch": 2.101421356201172e-05,
      "step": 22035,
      "training_step_time": 7.610386610031128
    },
    {
      "epoch": 2.1015167236328126e-05,
      "model_forward_time": 1.0669162273406982,
      "step": 22036
    },
    {
      "epoch": 2.1015167236328126e-05,
      "step": 22036,
      "training_step_time": 8.668907403945923
    },
    {
      "epoch": 2.101612091064453e-05,
      "model_forward_time": 1.7440738677978516,
      "step": 22037
    },
    {
      "epoch": 2.101612091064453e-05,
      "step": 22037,
      "training_step_time": 8.337724924087524
    },
    {
      "epoch": 2.1017074584960938e-05,
      "model_forward_time": 1.8481724262237549,
      "step": 22038
    },
    {
      "epoch": 2.1017074584960938e-05,
      "step": 22038,
      "training_step_time": 8.931586980819702
    },
    {
      "epoch": 2.1018028259277345e-05,
      "model_forward_time": 2.872263193130493,
      "step": 22039
    },
    {
      "epoch": 2.1018028259277345e-05,
      "step": 22039,
      "training_step_time": 8.663796663284302
    },
    {
      "epoch": 2.101898193359375e-05,
      "grad_norm": 0.17671272158622742,
      "learning_rate": 9.22703547237832e-06,
      "loss": 0.0645,
      "step": 22040
    },
    {
      "epoch": 2.101898193359375e-05,
      "model_forward_time": 1.7494254112243652,
      "step": 22040
    },
    {
      "dynamics_loss_avg": 0.05003057755529881,
      "epoch": 2.101898193359375e-05,
      "step": 22040
    },
    {
      "action_loss_avg": 0.0030545395915396512,
      "epoch": 2.101898193359375e-05,
      "step": 22040
    },
    {
      "epoch": 2.101898193359375e-05,
      "step": 22040,
      "training_step_time": 7.621726989746094
    },
    {
      "epoch": 2.1019935607910157e-05,
      "model_forward_time": 1.7538220882415771,
      "step": 22041
    },
    {
      "epoch": 2.1019935607910157e-05,
      "step": 22041,
      "training_step_time": 7.737325668334961
    },
    {
      "epoch": 2.1020889282226564e-05,
      "model_forward_time": 1.760871410369873,
      "step": 22042
    },
    {
      "epoch": 2.1020889282226564e-05,
      "step": 22042,
      "training_step_time": 7.469531297683716
    },
    {
      "epoch": 2.1021842956542968e-05,
      "model_forward_time": 1.7595932483673096,
      "step": 22043
    },
    {
      "epoch": 2.1021842956542968e-05,
      "step": 22043,
      "training_step_time": 7.050126791000366
    },
    {
      "epoch": 2.1022796630859375e-05,
      "model_forward_time": 1.7540888786315918,
      "step": 22044
    },
    {
      "epoch": 2.1022796630859375e-05,
      "step": 22044,
      "training_step_time": 7.542729616165161
    },
    {
      "epoch": 2.1023750305175783e-05,
      "model_forward_time": 1.7488243579864502,
      "step": 22045
    },
    {
      "epoch": 2.1023750305175783e-05,
      "step": 22045,
      "training_step_time": 8.44900393486023
    },
    {
      "epoch": 2.1024703979492187e-05,
      "model_forward_time": 1.747849702835083,
      "step": 22046
    },
    {
      "epoch": 2.1024703979492187e-05,
      "step": 22046,
      "training_step_time": 8.236088752746582
    },
    {
      "epoch": 2.1025657653808594e-05,
      "model_forward_time": 1.834596872329712,
      "step": 22047
    },
    {
      "epoch": 2.1025657653808594e-05,
      "step": 22047,
      "training_step_time": 8.842830419540405
    },
    {
      "epoch": 2.1026611328125e-05,
      "model_forward_time": 2.849087953567505,
      "step": 22048
    },
    {
      "epoch": 2.1026611328125e-05,
      "step": 22048,
      "training_step_time": 8.727001667022705
    },
    {
      "epoch": 2.1027565002441406e-05,
      "model_forward_time": 1.7492876052856445,
      "step": 22049
    },
    {
      "epoch": 2.1027565002441406e-05,
      "step": 22049,
      "training_step_time": 8.20438289642334
    },
    {
      "epoch": 2.1028518676757813e-05,
      "grad_norm": 0.1714780181646347,
      "learning_rate": 9.226152085831214e-06,
      "loss": 0.066,
      "step": 22050
    },
    {
      "epoch": 2.1028518676757813e-05,
      "model_forward_time": 1.7288541793823242,
      "step": 22050
    },
    {
      "dynamics_loss_avg": 0.05866081118583679,
      "epoch": 2.1028518676757813e-05,
      "step": 22050
    },
    {
      "action_loss_avg": 0.0025658322963863613,
      "epoch": 2.1028518676757813e-05,
      "step": 22050
    },
    {
      "epoch": 2.1028518676757813e-05,
      "step": 22050,
      "training_step_time": 7.664750576019287
    },
    {
      "epoch": 2.1029472351074217e-05,
      "model_forward_time": 1.7510912418365479,
      "step": 22051
    },
    {
      "epoch": 2.1029472351074217e-05,
      "step": 22051,
      "training_step_time": 7.4655680656433105
    },
    {
      "epoch": 2.1030426025390625e-05,
      "model_forward_time": 1.7518343925476074,
      "step": 22052
    },
    {
      "epoch": 2.1030426025390625e-05,
      "step": 22052,
      "training_step_time": 7.3362133502960205
    },
    {
      "epoch": 2.1031379699707032e-05,
      "model_forward_time": 1.7599096298217773,
      "step": 22053
    },
    {
      "epoch": 2.1031379699707032e-05,
      "step": 22053,
      "training_step_time": 8.40731430053711
    },
    {
      "epoch": 2.1032333374023436e-05,
      "model_forward_time": 1.745128870010376,
      "step": 22054
    },
    {
      "epoch": 2.1032333374023436e-05,
      "step": 22054,
      "training_step_time": 7.719568252563477
    },
    {
      "epoch": 2.1033287048339843e-05,
      "model_forward_time": 1.7549147605895996,
      "step": 22055
    },
    {
      "epoch": 2.1033287048339843e-05,
      "step": 22055,
      "training_step_time": 7.640794277191162
    },
    {
      "epoch": 2.103424072265625e-05,
      "model_forward_time": 1.4753642082214355,
      "step": 22056
    },
    {
      "epoch": 2.103424072265625e-05,
      "step": 22056,
      "training_step_time": 8.998499393463135
    },
    {
      "epoch": 2.1035194396972655e-05,
      "model_forward_time": 1.7575223445892334,
      "step": 22057
    },
    {
      "epoch": 2.1035194396972655e-05,
      "step": 22057,
      "training_step_time": 9.47394609451294
    },
    {
      "epoch": 2.1036148071289062e-05,
      "model_forward_time": 2.8836259841918945,
      "step": 22058
    },
    {
      "epoch": 2.1036148071289062e-05,
      "step": 22058,
      "training_step_time": 8.645287275314331
    },
    {
      "epoch": 2.103710174560547e-05,
      "model_forward_time": 1.7552356719970703,
      "step": 22059
    },
    {
      "epoch": 2.103710174560547e-05,
      "step": 22059,
      "training_step_time": 7.701295614242554
    },
    {
      "epoch": 2.1038055419921874e-05,
      "grad_norm": 0.30136892199516296,
      "learning_rate": 9.22526823711847e-06,
      "loss": 0.0667,
      "step": 22060
    },
    {
      "epoch": 2.1038055419921874e-05,
      "model_forward_time": 1.7694289684295654,
      "step": 22060
    },
    {
      "dynamics_loss_avg": 0.06403773073107004,
      "epoch": 2.1038055419921874e-05,
      "step": 22060
    },
    {
      "action_loss_avg": 0.003245239899843,
      "epoch": 2.1038055419921874e-05,
      "step": 22060
    },
    {
      "epoch": 2.1038055419921874e-05,
      "step": 22060,
      "training_step_time": 7.748966217041016
    },
    {
      "epoch": 2.103900909423828e-05,
      "model_forward_time": 1.7284855842590332,
      "step": 22061
    },
    {
      "epoch": 2.103900909423828e-05,
      "step": 22061,
      "training_step_time": 6.79450798034668
    },
    {
      "epoch": 2.103996276855469e-05,
      "model_forward_time": 1.7554199695587158,
      "step": 22062
    },
    {
      "epoch": 2.103996276855469e-05,
      "step": 22062,
      "training_step_time": 7.577237606048584
    },
    {
      "epoch": 2.1040916442871093e-05,
      "model_forward_time": 1.7415070533752441,
      "step": 22063
    },
    {
      "epoch": 2.1040916442871093e-05,
      "step": 22063,
      "training_step_time": 7.6738996505737305
    },
    {
      "epoch": 2.10418701171875e-05,
      "model_forward_time": 1.7439298629760742,
      "step": 22064
    },
    {
      "epoch": 2.10418701171875e-05,
      "step": 22064,
      "training_step_time": 10.662570476531982
    },
    {
      "epoch": 2.1042823791503907e-05,
      "model_forward_time": 1.8499112129211426,
      "step": 22065
    },
    {
      "epoch": 2.1042823791503907e-05,
      "step": 22065,
      "training_step_time": 8.976394414901733
    },
    {
      "epoch": 2.104377746582031e-05,
      "model_forward_time": 1.7482948303222656,
      "step": 22066
    },
    {
      "epoch": 2.104377746582031e-05,
      "step": 22066,
      "training_step_time": 7.758514881134033
    },
    {
      "epoch": 2.104473114013672e-05,
      "model_forward_time": 2.92875599861145,
      "step": 22067
    },
    {
      "epoch": 2.104473114013672e-05,
      "step": 22067,
      "training_step_time": 8.468390226364136
    },
    {
      "epoch": 2.1045684814453126e-05,
      "model_forward_time": 1.7500510215759277,
      "step": 22068
    },
    {
      "epoch": 2.1045684814453126e-05,
      "step": 22068,
      "training_step_time": 8.478100061416626
    },
    {
      "epoch": 2.104663848876953e-05,
      "model_forward_time": 1.3822922706604004,
      "step": 22069
    },
    {
      "epoch": 2.104663848876953e-05,
      "step": 22069,
      "training_step_time": 7.717727184295654
    },
    {
      "epoch": 2.1047592163085938e-05,
      "grad_norm": 0.2224372923374176,
      "learning_rate": 9.224383926336745e-06,
      "loss": 0.0641,
      "step": 22070
    },
    {
      "epoch": 2.1047592163085938e-05,
      "model_forward_time": 1.729285478591919,
      "step": 22070
    },
    {
      "dynamics_loss_avg": 0.06162677016109228,
      "epoch": 2.1047592163085938e-05,
      "step": 22070
    },
    {
      "action_loss_avg": 0.00288727858569473,
      "epoch": 2.1047592163085938e-05,
      "step": 22070
    },
    {
      "epoch": 2.1047592163085938e-05,
      "step": 22070,
      "training_step_time": 6.837147951126099
    },
    {
      "epoch": 2.1048545837402345e-05,
      "model_forward_time": 1.3838472366333008,
      "step": 22071
    },
    {
      "epoch": 2.1048545837402345e-05,
      "step": 22071,
      "training_step_time": 8.126560688018799
    },
    {
      "epoch": 2.104949951171875e-05,
      "model_forward_time": 1.3841125965118408,
      "step": 22072
    },
    {
      "epoch": 2.104949951171875e-05,
      "step": 22072,
      "training_step_time": 7.6852638721466064
    },
    {
      "epoch": 2.1050453186035157e-05,
      "model_forward_time": 1.7275888919830322,
      "step": 22073
    },
    {
      "epoch": 2.1050453186035157e-05,
      "step": 22073,
      "training_step_time": 7.547857046127319
    },
    {
      "epoch": 2.1051406860351564e-05,
      "model_forward_time": 1.8382618427276611,
      "step": 22074
    },
    {
      "epoch": 2.1051406860351564e-05,
      "step": 22074,
      "training_step_time": 9.123067617416382
    },
    {
      "epoch": 2.1052360534667968e-05,
      "model_forward_time": 1.751051425933838,
      "step": 22075
    },
    {
      "epoch": 2.1052360534667968e-05,
      "step": 22075,
      "training_step_time": 9.046847820281982
    },
    {
      "epoch": 2.1053314208984375e-05,
      "model_forward_time": 2.857816219329834,
      "step": 22076
    },
    {
      "epoch": 2.1053314208984375e-05,
      "step": 22076,
      "training_step_time": 8.593964099884033
    },
    {
      "epoch": 2.1054267883300783e-05,
      "model_forward_time": 1.7482037544250488,
      "step": 22077
    },
    {
      "epoch": 2.1054267883300783e-05,
      "step": 22077,
      "training_step_time": 7.659813404083252
    },
    {
      "epoch": 2.1055221557617187e-05,
      "model_forward_time": 1.752685785293579,
      "step": 22078
    },
    {
      "epoch": 2.1055221557617187e-05,
      "step": 22078,
      "training_step_time": 8.78174901008606
    },
    {
      "epoch": 2.1056175231933594e-05,
      "model_forward_time": 1.7498664855957031,
      "step": 22079
    },
    {
      "epoch": 2.1056175231933594e-05,
      "step": 22079,
      "training_step_time": 6.779951572418213
    },
    {
      "epoch": 2.105712890625e-05,
      "grad_norm": 0.19534339010715485,
      "learning_rate": 9.223499153582745e-06,
      "loss": 0.067,
      "step": 22080
    },
    {
      "epoch": 2.105712890625e-05,
      "model_forward_time": 1.7563560009002686,
      "step": 22080
    },
    {
      "dynamics_loss_avg": 0.0586935929954052,
      "epoch": 2.105712890625e-05,
      "step": 22080
    },
    {
      "action_loss_avg": 0.002717108599608764,
      "epoch": 2.105712890625e-05,
      "step": 22080
    },
    {
      "epoch": 2.105712890625e-05,
      "step": 22080,
      "training_step_time": 7.06628942489624
    },
    {
      "epoch": 2.1058082580566406e-05,
      "model_forward_time": 1.757633924484253,
      "step": 22081
    },
    {
      "epoch": 2.1058082580566406e-05,
      "step": 22081,
      "training_step_time": 7.590944766998291
    },
    {
      "epoch": 2.1059036254882813e-05,
      "model_forward_time": 1.748852014541626,
      "step": 22082
    },
    {
      "epoch": 2.1059036254882813e-05,
      "step": 22082,
      "training_step_time": 8.545645952224731
    },
    {
      "epoch": 2.1059989929199217e-05,
      "model_forward_time": 1.8595247268676758,
      "step": 22083
    },
    {
      "epoch": 2.1059989929199217e-05,
      "step": 22083,
      "training_step_time": 8.008018493652344
    },
    {
      "epoch": 2.1060943603515625e-05,
      "model_forward_time": 1.7418439388275146,
      "step": 22084
    },
    {
      "epoch": 2.1060943603515625e-05,
      "step": 22084,
      "training_step_time": 7.851625442504883
    },
    {
      "epoch": 2.1061897277832032e-05,
      "model_forward_time": 2.820223569869995,
      "step": 22085
    },
    {
      "epoch": 2.1061897277832032e-05,
      "step": 22085,
      "training_step_time": 9.174444675445557
    },
    {
      "epoch": 2.1062850952148436e-05,
      "model_forward_time": 1.390080213546753,
      "step": 22086
    },
    {
      "epoch": 2.1062850952148436e-05,
      "step": 22086,
      "training_step_time": 7.753525733947754
    },
    {
      "epoch": 2.1063804626464844e-05,
      "model_forward_time": 1.750805139541626,
      "step": 22087
    },
    {
      "epoch": 2.1063804626464844e-05,
      "step": 22087,
      "training_step_time": 7.7860143184661865
    },
    {
      "epoch": 2.106475830078125e-05,
      "model_forward_time": 1.7529723644256592,
      "step": 22088
    },
    {
      "epoch": 2.106475830078125e-05,
      "step": 22088,
      "training_step_time": 7.512330532073975
    },
    {
      "epoch": 2.1065711975097655e-05,
      "model_forward_time": 1.7545607089996338,
      "step": 22089
    },
    {
      "epoch": 2.1065711975097655e-05,
      "step": 22089,
      "training_step_time": 7.062060832977295
    },
    {
      "epoch": 2.1066665649414062e-05,
      "grad_norm": 0.16993074119091034,
      "learning_rate": 9.222613918953227e-06,
      "loss": 0.0657,
      "step": 22090
    },
    {
      "epoch": 2.1066665649414062e-05,
      "model_forward_time": 1.756267786026001,
      "step": 22090
    },
    {
      "dynamics_loss_avg": 0.057719160430133344,
      "epoch": 2.1066665649414062e-05,
      "step": 22090
    },
    {
      "action_loss_avg": 0.0022857573523651807,
      "epoch": 2.1066665649414062e-05,
      "step": 22090
    },
    {
      "epoch": 2.1066665649414062e-05,
      "step": 22090,
      "training_step_time": 7.574299097061157
    },
    {
      "epoch": 2.106761932373047e-05,
      "model_forward_time": 1.7491321563720703,
      "step": 22091
    },
    {
      "epoch": 2.106761932373047e-05,
      "step": 22091,
      "training_step_time": 7.669585943222046
    },
    {
      "epoch": 2.1068572998046874e-05,
      "model_forward_time": 1.863102674484253,
      "step": 22092
    },
    {
      "epoch": 2.1068572998046874e-05,
      "step": 22092,
      "training_step_time": 8.013028383255005
    },
    {
      "epoch": 2.106952667236328e-05,
      "model_forward_time": 1.749328851699829,
      "step": 22093
    },
    {
      "epoch": 2.106952667236328e-05,
      "step": 22093,
      "training_step_time": 9.84699034690857
    },
    {
      "epoch": 2.107048034667969e-05,
      "model_forward_time": 1.749706506729126,
      "step": 22094
    },
    {
      "epoch": 2.107048034667969e-05,
      "step": 22094,
      "training_step_time": 9.74174952507019
    },
    {
      "epoch": 2.1071434020996093e-05,
      "model_forward_time": 2.938490390777588,
      "step": 22095
    },
    {
      "epoch": 2.1071434020996093e-05,
      "step": 22095,
      "training_step_time": 7.747012138366699
    },
    {
      "epoch": 2.10723876953125e-05,
      "model_forward_time": 1.7606515884399414,
      "step": 22096
    },
    {
      "epoch": 2.10723876953125e-05,
      "step": 22096,
      "training_step_time": 7.735960483551025
    },
    {
      "epoch": 2.1073341369628907e-05,
      "model_forward_time": 1.7599458694458008,
      "step": 22097
    },
    {
      "epoch": 2.1073341369628907e-05,
      "step": 22097,
      "training_step_time": 7.441069602966309
    },
    {
      "epoch": 2.107429504394531e-05,
      "model_forward_time": 1.7634494304656982,
      "step": 22098
    },
    {
      "epoch": 2.107429504394531e-05,
      "step": 22098,
      "training_step_time": 6.802068471908569
    },
    {
      "epoch": 2.107524871826172e-05,
      "model_forward_time": 1.7637121677398682,
      "step": 22099
    },
    {
      "epoch": 2.107524871826172e-05,
      "step": 22099,
      "training_step_time": 8.328686714172363
    },
    {
      "epoch": 2.1076202392578126e-05,
      "grad_norm": 0.22768980264663696,
      "learning_rate": 9.221728222544998e-06,
      "loss": 0.0685,
      "step": 22100
    },
    {
      "epoch": 2.1076202392578126e-05,
      "model_forward_time": 1.7448723316192627,
      "step": 22100
    },
    {
      "dynamics_loss_avg": 0.04724523369222879,
      "epoch": 2.1076202392578126e-05,
      "step": 22100
    },
    {
      "action_loss_avg": 0.002364348969422281,
      "epoch": 2.1076202392578126e-05,
      "step": 22100
    },
    {
      "epoch": 2.1076202392578126e-05,
      "step": 22100,
      "training_step_time": 9.454311847686768
    },
    {
      "epoch": 2.107715606689453e-05,
      "model_forward_time": 1.8525307178497314,
      "step": 22101
    },
    {
      "epoch": 2.107715606689453e-05,
      "step": 22101,
      "training_step_time": 8.550376415252686
    },
    {
      "epoch": 2.1078109741210938e-05,
      "model_forward_time": 1.4008684158325195,
      "step": 22102
    },
    {
      "epoch": 2.1078109741210938e-05,
      "step": 22102,
      "training_step_time": 8.68122124671936
    },
    {
      "epoch": 2.1079063415527345e-05,
      "model_forward_time": 1.7360541820526123,
      "step": 22103
    },
    {
      "epoch": 2.1079063415527345e-05,
      "step": 22103,
      "training_step_time": 7.617425441741943
    },
    {
      "epoch": 2.108001708984375e-05,
      "model_forward_time": 2.876779794692993,
      "step": 22104
    },
    {
      "epoch": 2.108001708984375e-05,
      "step": 22104,
      "training_step_time": 8.628485441207886
    },
    {
      "epoch": 2.1080970764160157e-05,
      "model_forward_time": 1.7595527172088623,
      "step": 22105
    },
    {
      "epoch": 2.1080970764160157e-05,
      "step": 22105,
      "training_step_time": 7.547604084014893
    },
    {
      "epoch": 2.1081924438476564e-05,
      "model_forward_time": 1.769794225692749,
      "step": 22106
    },
    {
      "epoch": 2.1081924438476564e-05,
      "step": 22106,
      "training_step_time": 7.415683746337891
    },
    {
      "epoch": 2.1082878112792968e-05,
      "model_forward_time": 1.758103609085083,
      "step": 22107
    },
    {
      "epoch": 2.1082878112792968e-05,
      "step": 22107,
      "training_step_time": 6.870550155639648
    },
    {
      "epoch": 2.1083831787109376e-05,
      "model_forward_time": 1.7646985054016113,
      "step": 22108
    },
    {
      "epoch": 2.1083831787109376e-05,
      "step": 22108,
      "training_step_time": 7.6157166957855225
    },
    {
      "epoch": 2.1084785461425783e-05,
      "model_forward_time": 1.7556862831115723,
      "step": 22109
    },
    {
      "epoch": 2.1084785461425783e-05,
      "step": 22109,
      "training_step_time": 7.673668622970581
    },
    {
      "epoch": 2.1085739135742187e-05,
      "grad_norm": 0.13809402287006378,
      "learning_rate": 9.22084206445492e-06,
      "loss": 0.0672,
      "step": 22110
    },
    {
      "epoch": 2.1085739135742187e-05,
      "model_forward_time": 1.865842342376709,
      "step": 22110
    },
    {
      "dynamics_loss_avg": 0.04800620637834072,
      "epoch": 2.1085739135742187e-05,
      "step": 22110
    },
    {
      "action_loss_avg": 0.001577650592662394,
      "epoch": 2.1085739135742187e-05,
      "step": 22110
    },
    {
      "epoch": 2.1085739135742187e-05,
      "step": 22110,
      "training_step_time": 8.027323961257935
    },
    {
      "epoch": 2.1086692810058594e-05,
      "model_forward_time": 1.7646331787109375,
      "step": 22111
    },
    {
      "epoch": 2.1086692810058594e-05,
      "step": 22111,
      "training_step_time": 7.7075488567352295
    },
    {
      "epoch": 2.1087646484375e-05,
      "model_forward_time": 1.757197380065918,
      "step": 22112
    },
    {
      "epoch": 2.1087646484375e-05,
      "step": 22112,
      "training_step_time": 8.272240161895752
    },
    {
      "epoch": 2.1088600158691406e-05,
      "model_forward_time": 2.8512516021728516,
      "step": 22113
    },
    {
      "epoch": 2.1088600158691406e-05,
      "step": 22113,
      "training_step_time": 8.777865886688232
    },
    {
      "epoch": 2.1089553833007813e-05,
      "model_forward_time": 1.7500910758972168,
      "step": 22114
    },
    {
      "epoch": 2.1089553833007813e-05,
      "step": 22114,
      "training_step_time": 8.783154964447021
    },
    {
      "epoch": 2.1090507507324217e-05,
      "model_forward_time": 1.757375955581665,
      "step": 22115
    },
    {
      "epoch": 2.1090507507324217e-05,
      "step": 22115,
      "training_step_time": 7.755990028381348
    },
    {
      "epoch": 2.1091461181640625e-05,
      "model_forward_time": 1.7601399421691895,
      "step": 22116
    },
    {
      "epoch": 2.1091461181640625e-05,
      "step": 22116,
      "training_step_time": 6.840926647186279
    },
    {
      "epoch": 2.1092414855957032e-05,
      "model_forward_time": 1.7658922672271729,
      "step": 22117
    },
    {
      "epoch": 2.1092414855957032e-05,
      "step": 22117,
      "training_step_time": 7.611289978027344
    },
    {
      "epoch": 2.1093368530273436e-05,
      "model_forward_time": 1.741166591644287,
      "step": 22118
    },
    {
      "epoch": 2.1093368530273436e-05,
      "step": 22118,
      "training_step_time": 8.284246444702148
    },
    {
      "epoch": 2.1094322204589844e-05,
      "model_forward_time": 1.8352019786834717,
      "step": 22119
    },
    {
      "epoch": 2.1094322204589844e-05,
      "step": 22119,
      "training_step_time": 8.835299491882324
    },
    {
      "epoch": 2.109527587890625e-05,
      "grad_norm": 0.21803097426891327,
      "learning_rate": 9.2199554447799e-06,
      "loss": 0.0666,
      "step": 22120
    },
    {
      "epoch": 2.109527587890625e-05,
      "model_forward_time": 1.748455286026001,
      "step": 22120
    },
    {
      "dynamics_loss_avg": 0.05980552658438683,
      "epoch": 2.109527587890625e-05,
      "step": 22120
    },
    {
      "action_loss_avg": 0.002980127278715372,
      "epoch": 2.109527587890625e-05,
      "step": 22120
    },
    {
      "epoch": 2.109527587890625e-05,
      "step": 22120,
      "training_step_time": 7.812406301498413
    },
    {
      "epoch": 2.1096229553222655e-05,
      "model_forward_time": 1.7415056228637695,
      "step": 22121
    },
    {
      "epoch": 2.1096229553222655e-05,
      "step": 22121,
      "training_step_time": 7.770495891571045
    },
    {
      "epoch": 2.1097183227539062e-05,
      "model_forward_time": 1.7488794326782227,
      "step": 22122
    },
    {
      "epoch": 2.1097183227539062e-05,
      "step": 22122,
      "training_step_time": 9.322432279586792
    },
    {
      "epoch": 2.109813690185547e-05,
      "model_forward_time": 1.7523210048675537,
      "step": 22123
    },
    {
      "epoch": 2.109813690185547e-05,
      "step": 22123,
      "training_step_time": 7.775550365447998
    },
    {
      "epoch": 2.1099090576171874e-05,
      "model_forward_time": 1.752246379852295,
      "step": 22124
    },
    {
      "epoch": 2.1099090576171874e-05,
      "step": 22124,
      "training_step_time": 7.743447780609131
    },
    {
      "epoch": 2.110004425048828e-05,
      "model_forward_time": 1.7443835735321045,
      "step": 22125
    },
    {
      "epoch": 2.110004425048828e-05,
      "step": 22125,
      "training_step_time": 7.550597429275513
    },
    {
      "epoch": 2.110099792480469e-05,
      "model_forward_time": 1.756411075592041,
      "step": 22126
    },
    {
      "epoch": 2.110099792480469e-05,
      "step": 22126,
      "training_step_time": 7.620333433151245
    },
    {
      "epoch": 2.1101951599121093e-05,
      "model_forward_time": 1.7548375129699707,
      "step": 22127
    },
    {
      "epoch": 2.1101951599121093e-05,
      "step": 22127,
      "training_step_time": 8.443071603775024
    },
    {
      "epoch": 2.11029052734375e-05,
      "model_forward_time": 1.8479201793670654,
      "step": 22128
    },
    {
      "epoch": 2.11029052734375e-05,
      "step": 22128,
      "training_step_time": 8.850090026855469
    },
    {
      "epoch": 2.1103858947753908e-05,
      "model_forward_time": 1.750704288482666,
      "step": 22129
    },
    {
      "epoch": 2.1103858947753908e-05,
      "step": 22129,
      "training_step_time": 8.466691732406616
    },
    {
      "epoch": 2.110481262207031e-05,
      "grad_norm": 0.27008894085884094,
      "learning_rate": 9.219068363616896e-06,
      "loss": 0.0648,
      "step": 22130
    },
    {
      "epoch": 2.110481262207031e-05,
      "model_forward_time": 1.755613088607788,
      "step": 22130
    },
    {
      "dynamics_loss_avg": 0.05494933314621449,
      "epoch": 2.110481262207031e-05,
      "step": 22130
    },
    {
      "action_loss_avg": 0.002601326594594866,
      "epoch": 2.110481262207031e-05,
      "step": 22130
    },
    {
      "epoch": 2.110481262207031e-05,
      "step": 22130,
      "training_step_time": 8.59156584739685
    },
    {
      "epoch": 2.110576629638672e-05,
      "model_forward_time": 1.7452449798583984,
      "step": 22131
    },
    {
      "epoch": 2.110576629638672e-05,
      "step": 22131,
      "training_step_time": 9.194686889648438
    },
    {
      "epoch": 2.1106719970703126e-05,
      "model_forward_time": 2.894380569458008,
      "step": 22132
    },
    {
      "epoch": 2.1106719970703126e-05,
      "step": 22132,
      "training_step_time": 7.769439220428467
    },
    {
      "epoch": 2.110767364501953e-05,
      "model_forward_time": 1.3890984058380127,
      "step": 22133
    },
    {
      "epoch": 2.110767364501953e-05,
      "step": 22133,
      "training_step_time": 7.724308490753174
    },
    {
      "epoch": 2.1108627319335938e-05,
      "model_forward_time": 1.7485201358795166,
      "step": 22134
    },
    {
      "epoch": 2.1108627319335938e-05,
      "step": 22134,
      "training_step_time": 7.565831661224365
    },
    {
      "epoch": 2.1109580993652345e-05,
      "model_forward_time": 1.7407238483428955,
      "step": 22135
    },
    {
      "epoch": 2.1109580993652345e-05,
      "step": 22135,
      "training_step_time": 7.350141525268555
    },
    {
      "epoch": 2.111053466796875e-05,
      "model_forward_time": 1.7422966957092285,
      "step": 22136
    },
    {
      "epoch": 2.111053466796875e-05,
      "step": 22136,
      "training_step_time": 7.656850099563599
    },
    {
      "epoch": 2.1111488342285157e-05,
      "model_forward_time": 1.8179588317871094,
      "step": 22137
    },
    {
      "epoch": 2.1111488342285157e-05,
      "step": 22137,
      "training_step_time": 7.81345272064209
    },
    {
      "epoch": 2.1112442016601564e-05,
      "model_forward_time": 1.7539668083190918,
      "step": 22138
    },
    {
      "epoch": 2.1112442016601564e-05,
      "step": 22138,
      "training_step_time": 7.945829629898071
    },
    {
      "epoch": 2.1113395690917968e-05,
      "model_forward_time": 1.7443883419036865,
      "step": 22139
    },
    {
      "epoch": 2.1113395690917968e-05,
      "step": 22139,
      "training_step_time": 8.407669067382812
    },
    {
      "epoch": 2.1114349365234376e-05,
      "grad_norm": 0.2856235206127167,
      "learning_rate": 9.21818082106292e-06,
      "loss": 0.0655,
      "step": 22140
    },
    {
      "epoch": 2.1114349365234376e-05,
      "model_forward_time": 1.7348759174346924,
      "step": 22140
    },
    {
      "dynamics_loss_avg": 0.0542715622112155,
      "epoch": 2.1114349365234376e-05,
      "step": 22140
    },
    {
      "action_loss_avg": 0.0021890719508519396,
      "epoch": 2.1114349365234376e-05,
      "step": 22140
    },
    {
      "epoch": 2.1114349365234376e-05,
      "step": 22140,
      "training_step_time": 8.791116714477539
    },
    {
      "epoch": 2.111530303955078e-05,
      "model_forward_time": 2.838003635406494,
      "step": 22141
    },
    {
      "epoch": 2.111530303955078e-05,
      "step": 22141,
      "training_step_time": 8.336638450622559
    },
    {
      "epoch": 2.1116256713867187e-05,
      "model_forward_time": 1.746109962463379,
      "step": 22142
    },
    {
      "epoch": 2.1116256713867187e-05,
      "step": 22142,
      "training_step_time": 7.692399740219116
    },
    {
      "epoch": 2.1117210388183594e-05,
      "model_forward_time": 1.7499618530273438,
      "step": 22143
    },
    {
      "epoch": 2.1117210388183594e-05,
      "step": 22143,
      "training_step_time": 7.701690673828125
    },
    {
      "epoch": 2.11181640625e-05,
      "model_forward_time": 1.760087251663208,
      "step": 22144
    },
    {
      "epoch": 2.11181640625e-05,
      "step": 22144,
      "training_step_time": 6.9173102378845215
    },
    {
      "epoch": 2.1119117736816406e-05,
      "model_forward_time": 1.744248628616333,
      "step": 22145
    },
    {
      "epoch": 2.1119117736816406e-05,
      "step": 22145,
      "training_step_time": 8.403305530548096
    },
    {
      "epoch": 2.1120071411132813e-05,
      "model_forward_time": 1.850585699081421,
      "step": 22146
    },
    {
      "epoch": 2.1120071411132813e-05,
      "step": 22146,
      "training_step_time": 7.723341226577759
    },
    {
      "epoch": 2.1121025085449217e-05,
      "model_forward_time": 1.7494943141937256,
      "step": 22147
    },
    {
      "epoch": 2.1121025085449217e-05,
      "step": 22147,
      "training_step_time": 7.948923110961914
    },
    {
      "epoch": 2.1121978759765625e-05,
      "model_forward_time": 1.7287933826446533,
      "step": 22148
    },
    {
      "epoch": 2.1121978759765625e-05,
      "step": 22148,
      "training_step_time": 9.772275686264038
    },
    {
      "epoch": 2.1122932434082032e-05,
      "model_forward_time": 1.7317781448364258,
      "step": 22149
    },
    {
      "epoch": 2.1122932434082032e-05,
      "step": 22149,
      "training_step_time": 7.795582056045532
    },
    {
      "epoch": 2.1123886108398436e-05,
      "grad_norm": 0.16410820186138153,
      "learning_rate": 9.217292817215032e-06,
      "loss": 0.0687,
      "step": 22150
    },
    {
      "epoch": 2.1123886108398436e-05,
      "model_forward_time": 2.8365862369537354,
      "step": 22150
    },
    {
      "dynamics_loss_avg": 0.054290269128978254,
      "epoch": 2.1123886108398436e-05,
      "step": 22150
    },
    {
      "action_loss_avg": 0.0030886544380337,
      "epoch": 2.1123886108398436e-05,
      "step": 22150
    },
    {
      "epoch": 2.1123886108398436e-05,
      "step": 22150,
      "training_step_time": 8.448352098464966
    },
    {
      "epoch": 2.1124839782714844e-05,
      "model_forward_time": 1.7444572448730469,
      "step": 22151
    },
    {
      "epoch": 2.1124839782714844e-05,
      "step": 22151,
      "training_step_time": 8.886737823486328
    },
    {
      "epoch": 2.112579345703125e-05,
      "model_forward_time": 1.7463304996490479,
      "step": 22152
    },
    {
      "epoch": 2.112579345703125e-05,
      "step": 22152,
      "training_step_time": 7.690920829772949
    },
    {
      "epoch": 2.1126747131347655e-05,
      "model_forward_time": 1.746108055114746,
      "step": 22153
    },
    {
      "epoch": 2.1126747131347655e-05,
      "step": 22153,
      "training_step_time": 6.815431118011475
    },
    {
      "epoch": 2.1127700805664062e-05,
      "model_forward_time": 1.7385389804840088,
      "step": 22154
    },
    {
      "epoch": 2.1127700805664062e-05,
      "step": 22154,
      "training_step_time": 7.639420509338379
    },
    {
      "epoch": 2.112865447998047e-05,
      "model_forward_time": 1.8239305019378662,
      "step": 22155
    },
    {
      "epoch": 2.112865447998047e-05,
      "step": 22155,
      "training_step_time": 7.638044357299805
    },
    {
      "epoch": 2.1129608154296874e-05,
      "model_forward_time": 1.7467877864837646,
      "step": 22156
    },
    {
      "epoch": 2.1129608154296874e-05,
      "step": 22156,
      "training_step_time": 9.065975427627563
    },
    {
      "epoch": 2.113056182861328e-05,
      "model_forward_time": 1.736825704574585,
      "step": 22157
    },
    {
      "epoch": 2.113056182861328e-05,
      "step": 22157,
      "training_step_time": 8.622532606124878
    },
    {
      "epoch": 2.113151550292969e-05,
      "model_forward_time": 1.7376644611358643,
      "step": 22158
    },
    {
      "epoch": 2.113151550292969e-05,
      "step": 22158,
      "training_step_time": 7.855335235595703
    },
    {
      "epoch": 2.1132469177246093e-05,
      "model_forward_time": 2.7897956371307373,
      "step": 22159
    },
    {
      "epoch": 2.1132469177246093e-05,
      "step": 22159,
      "training_step_time": 9.069329261779785
    },
    {
      "epoch": 2.11334228515625e-05,
      "grad_norm": 0.18707075715065002,
      "learning_rate": 9.21640435217034e-06,
      "loss": 0.0682,
      "step": 22160
    },
    {
      "epoch": 2.11334228515625e-05,
      "model_forward_time": 1.7599363327026367,
      "step": 22160
    },
    {
      "dynamics_loss_avg": 0.06203647144138813,
      "epoch": 2.11334228515625e-05,
      "step": 22160
    },
    {
      "action_loss_avg": 0.0027611315017566085,
      "epoch": 2.11334228515625e-05,
      "step": 22160
    },
    {
      "epoch": 2.11334228515625e-05,
      "step": 22160,
      "training_step_time": 7.610848426818848
    },
    {
      "epoch": 2.1134376525878908e-05,
      "model_forward_time": 1.756056547164917,
      "step": 22161
    },
    {
      "epoch": 2.1134376525878908e-05,
      "step": 22161,
      "training_step_time": 7.748706817626953
    },
    {
      "epoch": 2.113533020019531e-05,
      "model_forward_time": 1.7508554458618164,
      "step": 22162
    },
    {
      "epoch": 2.113533020019531e-05,
      "step": 22162,
      "training_step_time": 6.798042297363281
    },
    {
      "epoch": 2.113628387451172e-05,
      "model_forward_time": 1.7548413276672363,
      "step": 22163
    },
    {
      "epoch": 2.113628387451172e-05,
      "step": 22163,
      "training_step_time": 7.598934173583984
    },
    {
      "epoch": 2.1137237548828126e-05,
      "model_forward_time": 1.8022840023040771,
      "step": 22164
    },
    {
      "epoch": 2.1137237548828126e-05,
      "step": 22164,
      "training_step_time": 7.665890693664551
    },
    {
      "epoch": 2.113819122314453e-05,
      "model_forward_time": 1.7531421184539795,
      "step": 22165
    },
    {
      "epoch": 2.113819122314453e-05,
      "step": 22165,
      "training_step_time": 8.978838920593262
    },
    {
      "epoch": 2.1139144897460938e-05,
      "model_forward_time": 1.7516982555389404,
      "step": 22166
    },
    {
      "epoch": 2.1139144897460938e-05,
      "step": 22166,
      "training_step_time": 7.6823201179504395
    },
    {
      "epoch": 2.1140098571777345e-05,
      "model_forward_time": 1.754028081893921,
      "step": 22167
    },
    {
      "epoch": 2.1140098571777345e-05,
      "step": 22167,
      "training_step_time": 9.423061609268188
    },
    {
      "epoch": 2.114105224609375e-05,
      "model_forward_time": 1.7533369064331055,
      "step": 22168
    },
    {
      "epoch": 2.114105224609375e-05,
      "step": 22168,
      "training_step_time": 8.08407187461853
    },
    {
      "epoch": 2.1142005920410157e-05,
      "model_forward_time": 2.8746731281280518,
      "step": 22169
    },
    {
      "epoch": 2.1142005920410157e-05,
      "step": 22169,
      "training_step_time": 7.624379396438599
    },
    {
      "epoch": 2.1142959594726564e-05,
      "grad_norm": 0.16745752096176147,
      "learning_rate": 9.215515426026006e-06,
      "loss": 0.0668,
      "step": 22170
    },
    {
      "epoch": 2.1142959594726564e-05,
      "model_forward_time": 1.7589213848114014,
      "step": 22170
    },
    {
      "dynamics_loss_avg": 0.052170244976878166,
      "epoch": 2.1142959594726564e-05,
      "step": 22170
    },
    {
      "action_loss_avg": 0.002789504604879767,
      "epoch": 2.1142959594726564e-05,
      "step": 22170
    },
    {
      "epoch": 2.1142959594726564e-05,
      "step": 22170,
      "training_step_time": 8.66348147392273
    },
    {
      "epoch": 2.1143913269042968e-05,
      "model_forward_time": 1.754638433456421,
      "step": 22171
    },
    {
      "epoch": 2.1143913269042968e-05,
      "step": 22171,
      "training_step_time": 6.782870054244995
    },
    {
      "epoch": 2.1144866943359376e-05,
      "model_forward_time": 1.7507894039154053,
      "step": 22172
    },
    {
      "epoch": 2.1144866943359376e-05,
      "step": 22172,
      "training_step_time": 7.59502100944519
    },
    {
      "epoch": 2.114582061767578e-05,
      "model_forward_time": 1.7795910835266113,
      "step": 22173
    },
    {
      "epoch": 2.114582061767578e-05,
      "step": 22173,
      "training_step_time": 7.633645296096802
    },
    {
      "epoch": 2.1146774291992187e-05,
      "model_forward_time": 1.7558164596557617,
      "step": 22174
    },
    {
      "epoch": 2.1146774291992187e-05,
      "step": 22174,
      "training_step_time": 7.651873350143433
    },
    {
      "epoch": 2.1147727966308594e-05,
      "model_forward_time": 1.7522683143615723,
      "step": 22175
    },
    {
      "epoch": 2.1147727966308594e-05,
      "step": 22175,
      "training_step_time": 7.90315842628479
    },
    {
      "epoch": 2.1148681640625e-05,
      "model_forward_time": 1.7515969276428223,
      "step": 22176
    },
    {
      "epoch": 2.1148681640625e-05,
      "step": 22176,
      "training_step_time": 7.76717209815979
    },
    {
      "epoch": 2.1149635314941406e-05,
      "model_forward_time": 1.7401657104492188,
      "step": 22177
    },
    {
      "epoch": 2.1149635314941406e-05,
      "step": 22177,
      "training_step_time": 10.565117597579956
    },
    {
      "epoch": 2.1150588989257813e-05,
      "model_forward_time": 2.887086868286133,
      "step": 22178
    },
    {
      "epoch": 2.1150588989257813e-05,
      "step": 22178,
      "training_step_time": 7.681478977203369
    },
    {
      "epoch": 2.1151542663574217e-05,
      "model_forward_time": 1.757577896118164,
      "step": 22179
    },
    {
      "epoch": 2.1151542663574217e-05,
      "step": 22179,
      "training_step_time": 7.596110582351685
    },
    {
      "epoch": 2.1152496337890625e-05,
      "grad_norm": 0.25017693638801575,
      "learning_rate": 9.214626038879247e-06,
      "loss": 0.0663,
      "step": 22180
    },
    {
      "epoch": 2.1152496337890625e-05,
      "model_forward_time": 1.7426700592041016,
      "step": 22180
    },
    {
      "dynamics_loss_avg": 0.06771661099046469,
      "epoch": 2.1152496337890625e-05,
      "step": 22180
    },
    {
      "action_loss_avg": 0.003824295927188359,
      "epoch": 2.1152496337890625e-05,
      "step": 22180
    },
    {
      "epoch": 2.1152496337890625e-05,
      "step": 22180,
      "training_step_time": 7.4859254360198975
    },
    {
      "epoch": 2.1153450012207032e-05,
      "model_forward_time": 1.7682769298553467,
      "step": 22181
    },
    {
      "epoch": 2.1153450012207032e-05,
      "step": 22181,
      "training_step_time": 7.622379541397095
    },
    {
      "epoch": 2.1154403686523436e-05,
      "model_forward_time": 1.7603988647460938,
      "step": 22182
    },
    {
      "epoch": 2.1154403686523436e-05,
      "step": 22182,
      "training_step_time": 9.141072273254395
    },
    {
      "epoch": 2.1155357360839844e-05,
      "model_forward_time": 1.7567472457885742,
      "step": 22183
    },
    {
      "epoch": 2.1155357360839844e-05,
      "step": 22183,
      "training_step_time": 7.642465829849243
    },
    {
      "epoch": 2.115631103515625e-05,
      "model_forward_time": 1.7582781314849854,
      "step": 22184
    },
    {
      "epoch": 2.115631103515625e-05,
      "step": 22184,
      "training_step_time": 8.532690286636353
    },
    {
      "epoch": 2.1157264709472655e-05,
      "model_forward_time": 1.7512366771697998,
      "step": 22185
    },
    {
      "epoch": 2.1157264709472655e-05,
      "step": 22185,
      "training_step_time": 8.849167585372925
    },
    {
      "epoch": 2.1158218383789062e-05,
      "model_forward_time": 1.7557661533355713,
      "step": 22186
    },
    {
      "epoch": 2.1158218383789062e-05,
      "step": 22186,
      "training_step_time": 8.199768781661987
    },
    {
      "epoch": 2.115917205810547e-05,
      "model_forward_time": 2.883608341217041,
      "step": 22187
    },
    {
      "epoch": 2.115917205810547e-05,
      "step": 22187,
      "training_step_time": 7.627310752868652
    },
    {
      "epoch": 2.1160125732421874e-05,
      "model_forward_time": 1.7508292198181152,
      "step": 22188
    },
    {
      "epoch": 2.1160125732421874e-05,
      "step": 22188,
      "training_step_time": 8.288054704666138
    },
    {
      "epoch": 2.116107940673828e-05,
      "model_forward_time": 1.753502607345581,
      "step": 22189
    },
    {
      "epoch": 2.116107940673828e-05,
      "step": 22189,
      "training_step_time": 7.598375558853149
    },
    {
      "epoch": 2.116203308105469e-05,
      "grad_norm": 0.17758916318416595,
      "learning_rate": 9.213736190827319e-06,
      "loss": 0.0657,
      "step": 22190
    },
    {
      "epoch": 2.116203308105469e-05,
      "model_forward_time": 1.7466826438903809,
      "step": 22190
    },
    {
      "dynamics_loss_avg": 0.048808991070836784,
      "epoch": 2.116203308105469e-05,
      "step": 22190
    },
    {
      "action_loss_avg": 0.0024007848929613828,
      "epoch": 2.116203308105469e-05,
      "step": 22190
    },
    {
      "epoch": 2.116203308105469e-05,
      "step": 22190,
      "training_step_time": 7.601718187332153
    },
    {
      "epoch": 2.1162986755371093e-05,
      "model_forward_time": 1.7484641075134277,
      "step": 22191
    },
    {
      "epoch": 2.1162986755371093e-05,
      "step": 22191,
      "training_step_time": 7.558818578720093
    },
    {
      "epoch": 2.11639404296875e-05,
      "model_forward_time": 1.8583359718322754,
      "step": 22192
    },
    {
      "epoch": 2.11639404296875e-05,
      "step": 22192,
      "training_step_time": 7.6662561893463135
    },
    {
      "epoch": 2.1164894104003908e-05,
      "model_forward_time": 1.7501862049102783,
      "step": 22193
    },
    {
      "epoch": 2.1164894104003908e-05,
      "step": 22193,
      "training_step_time": 8.837688446044922
    },
    {
      "epoch": 2.116584777832031e-05,
      "model_forward_time": 1.7449278831481934,
      "step": 22194
    },
    {
      "epoch": 2.116584777832031e-05,
      "step": 22194,
      "training_step_time": 9.177334308624268
    },
    {
      "epoch": 2.116680145263672e-05,
      "model_forward_time": 1.7476866245269775,
      "step": 22195
    },
    {
      "epoch": 2.116680145263672e-05,
      "step": 22195,
      "training_step_time": 7.697525501251221
    },
    {
      "epoch": 2.1167755126953126e-05,
      "model_forward_time": 2.8563072681427,
      "step": 22196
    },
    {
      "epoch": 2.1167755126953126e-05,
      "step": 22196,
      "training_step_time": 8.845624208450317
    },
    {
      "epoch": 2.116870880126953e-05,
      "model_forward_time": 1.7575857639312744,
      "step": 22197
    },
    {
      "epoch": 2.116870880126953e-05,
      "step": 22197,
      "training_step_time": 7.599032878875732
    },
    {
      "epoch": 2.1169662475585938e-05,
      "model_forward_time": 1.7521891593933105,
      "step": 22198
    },
    {
      "epoch": 2.1169662475585938e-05,
      "step": 22198,
      "training_step_time": 7.685520172119141
    },
    {
      "epoch": 2.1170616149902345e-05,
      "model_forward_time": 1.7526075839996338,
      "step": 22199
    },
    {
      "epoch": 2.1170616149902345e-05,
      "step": 22199,
      "training_step_time": 7.566028356552124
    },
    {
      "epoch": 2.117156982421875e-05,
      "grad_norm": 0.1451418399810791,
      "learning_rate": 9.212845881967536e-06,
      "loss": 0.0643,
      "step": 22200
    },
    {
      "epoch": 2.117156982421875e-05,
      "model_forward_time": 1.7227451801300049,
      "step": 22200
    },
    {
      "dynamics_loss_avg": 0.07511524111032486,
      "epoch": 2.117156982421875e-05,
      "step": 22200
    },
    {
      "action_loss_avg": 0.003550635080318898,
      "epoch": 2.117156982421875e-05,
      "step": 22200
    },
    {
      "epoch": 2.117156982421875e-05,
      "step": 22200,
      "training_step_time": 7.561878442764282
    },
    {
      "epoch": 2.1172523498535157e-05,
      "model_forward_time": 1.846285104751587,
      "step": 22201
    },
    {
      "epoch": 2.1172523498535157e-05,
      "step": 22201,
      "training_step_time": 9.627290964126587
    },
    {
      "epoch": 2.1173477172851564e-05,
      "model_forward_time": 1.7530484199523926,
      "step": 22202
    },
    {
      "epoch": 2.1173477172851564e-05,
      "step": 22202,
      "training_step_time": 8.749590873718262
    },
    {
      "epoch": 2.1174430847167968e-05,
      "model_forward_time": 1.738677740097046,
      "step": 22203
    },
    {
      "epoch": 2.1174430847167968e-05,
      "step": 22203,
      "training_step_time": 7.719729900360107
    },
    {
      "epoch": 2.1175384521484376e-05,
      "model_forward_time": 1.7437636852264404,
      "step": 22204
    },
    {
      "epoch": 2.1175384521484376e-05,
      "step": 22204,
      "training_step_time": 8.245907306671143
    },
    {
      "epoch": 2.117633819580078e-05,
      "model_forward_time": 2.88128924369812,
      "step": 22205
    },
    {
      "epoch": 2.117633819580078e-05,
      "step": 22205,
      "training_step_time": 8.468570470809937
    },
    {
      "epoch": 2.1177291870117187e-05,
      "model_forward_time": 1.7488837242126465,
      "step": 22206
    },
    {
      "epoch": 2.1177291870117187e-05,
      "step": 22206,
      "training_step_time": 7.6457436084747314
    },
    {
      "epoch": 2.1178245544433594e-05,
      "model_forward_time": 1.7518885135650635,
      "step": 22207
    },
    {
      "epoch": 2.1178245544433594e-05,
      "step": 22207,
      "training_step_time": 7.656895160675049
    },
    {
      "epoch": 2.117919921875e-05,
      "model_forward_time": 1.743375301361084,
      "step": 22208
    },
    {
      "epoch": 2.117919921875e-05,
      "step": 22208,
      "training_step_time": 7.001380205154419
    },
    {
      "epoch": 2.1180152893066406e-05,
      "model_forward_time": 1.7443208694458008,
      "step": 22209
    },
    {
      "epoch": 2.1180152893066406e-05,
      "step": 22209,
      "training_step_time": 7.484170913696289
    },
    {
      "epoch": 2.1181106567382813e-05,
      "grad_norm": 0.16866272687911987,
      "learning_rate": 9.211955112397262e-06,
      "loss": 0.0652,
      "step": 22210
    },
    {
      "epoch": 2.1181106567382813e-05,
      "model_forward_time": 1.8148071765899658,
      "step": 22210
    },
    {
      "dynamics_loss_avg": 0.07250876314938068,
      "epoch": 2.1181106567382813e-05,
      "step": 22210
    },
    {
      "action_loss_avg": 0.0031640034343581648,
      "epoch": 2.1181106567382813e-05,
      "step": 22210
    },
    {
      "epoch": 2.1181106567382813e-05,
      "step": 22210,
      "training_step_time": 7.609461784362793
    },
    {
      "epoch": 2.1182060241699217e-05,
      "model_forward_time": 1.7270760536193848,
      "step": 22211
    },
    {
      "epoch": 2.1182060241699217e-05,
      "step": 22211,
      "training_step_time": 7.812287330627441
    },
    {
      "epoch": 2.1183013916015625e-05,
      "model_forward_time": 1.7436916828155518,
      "step": 22212
    },
    {
      "epoch": 2.1183013916015625e-05,
      "step": 22212,
      "training_step_time": 7.6611998081207275
    },
    {
      "epoch": 2.1183967590332032e-05,
      "model_forward_time": 1.753035545349121,
      "step": 22213
    },
    {
      "epoch": 2.1183967590332032e-05,
      "step": 22213,
      "training_step_time": 8.255326747894287
    },
    {
      "epoch": 2.1184921264648436e-05,
      "model_forward_time": 2.8997457027435303,
      "step": 22214
    },
    {
      "epoch": 2.1184921264648436e-05,
      "step": 22214,
      "training_step_time": 9.011032581329346
    },
    {
      "epoch": 2.1185874938964844e-05,
      "model_forward_time": 1.7524211406707764,
      "step": 22215
    },
    {
      "epoch": 2.1185874938964844e-05,
      "step": 22215,
      "training_step_time": 7.581247329711914
    },
    {
      "epoch": 2.118682861328125e-05,
      "model_forward_time": 1.7602338790893555,
      "step": 22216
    },
    {
      "epoch": 2.118682861328125e-05,
      "step": 22216,
      "training_step_time": 7.59690260887146
    },
    {
      "epoch": 2.1187782287597655e-05,
      "model_forward_time": 1.7538843154907227,
      "step": 22217
    },
    {
      "epoch": 2.1187782287597655e-05,
      "step": 22217,
      "training_step_time": 7.30823540687561
    },
    {
      "epoch": 2.1188735961914062e-05,
      "model_forward_time": 1.7514286041259766,
      "step": 22218
    },
    {
      "epoch": 2.1188735961914062e-05,
      "step": 22218,
      "training_step_time": 7.146830797195435
    },
    {
      "epoch": 2.118968963623047e-05,
      "model_forward_time": 1.7841107845306396,
      "step": 22219
    },
    {
      "epoch": 2.118968963623047e-05,
      "step": 22219,
      "training_step_time": 7.662234544754028
    },
    {
      "epoch": 2.1190643310546874e-05,
      "grad_norm": 0.2279876470565796,
      "learning_rate": 9.21106388221391e-06,
      "loss": 0.0662,
      "step": 22220
    },
    {
      "epoch": 2.1190643310546874e-05,
      "model_forward_time": 1.7417488098144531,
      "step": 22220
    },
    {
      "dynamics_loss_avg": 0.04732249267399311,
      "epoch": 2.1190643310546874e-05,
      "step": 22220
    },
    {
      "action_loss_avg": 0.002284945355495438,
      "epoch": 2.1190643310546874e-05,
      "step": 22220
    },
    {
      "epoch": 2.1190643310546874e-05,
      "step": 22220,
      "training_step_time": 7.710380792617798
    },
    {
      "epoch": 2.119159698486328e-05,
      "model_forward_time": 1.7493479251861572,
      "step": 22221
    },
    {
      "epoch": 2.119159698486328e-05,
      "step": 22221,
      "training_step_time": 8.531006574630737
    },
    {
      "epoch": 2.119255065917969e-05,
      "model_forward_time": 1.751338243484497,
      "step": 22222
    },
    {
      "epoch": 2.119255065917969e-05,
      "step": 22222,
      "training_step_time": 8.604997634887695
    },
    {
      "epoch": 2.1193504333496093e-05,
      "model_forward_time": 2.925088882446289,
      "step": 22223
    },
    {
      "epoch": 2.1193504333496093e-05,
      "step": 22223,
      "training_step_time": 8.515290975570679
    },
    {
      "epoch": 2.11944580078125e-05,
      "model_forward_time": 1.752213716506958,
      "step": 22224
    },
    {
      "epoch": 2.11944580078125e-05,
      "step": 22224,
      "training_step_time": 8.33528447151184
    },
    {
      "epoch": 2.1195411682128908e-05,
      "model_forward_time": 1.7483677864074707,
      "step": 22225
    },
    {
      "epoch": 2.1195411682128908e-05,
      "step": 22225,
      "training_step_time": 7.6069114208221436
    },
    {
      "epoch": 2.119636535644531e-05,
      "model_forward_time": 1.754868984222412,
      "step": 22226
    },
    {
      "epoch": 2.119636535644531e-05,
      "step": 22226,
      "training_step_time": 7.656703233718872
    },
    {
      "epoch": 2.119731903076172e-05,
      "model_forward_time": 1.7481250762939453,
      "step": 22227
    },
    {
      "epoch": 2.119731903076172e-05,
      "step": 22227,
      "training_step_time": 6.97704005241394
    },
    {
      "epoch": 2.1198272705078126e-05,
      "model_forward_time": 1.7564141750335693,
      "step": 22228
    },
    {
      "epoch": 2.1198272705078126e-05,
      "step": 22228,
      "training_step_time": 8.452558040618896
    },
    {
      "epoch": 2.119922637939453e-05,
      "model_forward_time": 1.759495496749878,
      "step": 22229
    },
    {
      "epoch": 2.119922637939453e-05,
      "step": 22229,
      "training_step_time": 7.689221382141113
    },
    {
      "epoch": 2.1200180053710938e-05,
      "grad_norm": 0.1682819128036499,
      "learning_rate": 9.210172191514943e-06,
      "loss": 0.0641,
      "step": 22230
    },
    {
      "epoch": 2.1200180053710938e-05,
      "model_forward_time": 1.3970003128051758,
      "step": 22230
    },
    {
      "dynamics_loss_avg": 0.07598476745188236,
      "epoch": 2.1200180053710938e-05,
      "step": 22230
    },
    {
      "action_loss_avg": 0.003864484722726047,
      "epoch": 2.1200180053710938e-05,
      "step": 22230
    },
    {
      "epoch": 2.1200180053710938e-05,
      "step": 22230,
      "training_step_time": 8.504895210266113
    },
    {
      "epoch": 2.1201133728027345e-05,
      "model_forward_time": 1.7546145915985107,
      "step": 22231
    },
    {
      "epoch": 2.1201133728027345e-05,
      "step": 22231,
      "training_step_time": 7.680983066558838
    },
    {
      "epoch": 2.120208740234375e-05,
      "model_forward_time": 2.93009352684021,
      "step": 22232
    },
    {
      "epoch": 2.120208740234375e-05,
      "step": 22232,
      "training_step_time": 7.690340042114258
    },
    {
      "epoch": 2.1203041076660157e-05,
      "model_forward_time": 1.7495017051696777,
      "step": 22233
    },
    {
      "epoch": 2.1203041076660157e-05,
      "step": 22233,
      "training_step_time": 8.328193426132202
    },
    {
      "epoch": 2.1203994750976564e-05,
      "model_forward_time": 1.7548081874847412,
      "step": 22234
    },
    {
      "epoch": 2.1203994750976564e-05,
      "step": 22234,
      "training_step_time": 8.607151985168457
    },
    {
      "epoch": 2.1204948425292968e-05,
      "model_forward_time": 1.7496635913848877,
      "step": 22235
    },
    {
      "epoch": 2.1204948425292968e-05,
      "step": 22235,
      "training_step_time": 7.595435857772827
    },
    {
      "epoch": 2.1205902099609376e-05,
      "model_forward_time": 1.7519662380218506,
      "step": 22236
    },
    {
      "epoch": 2.1205902099609376e-05,
      "step": 22236,
      "training_step_time": 6.956142425537109
    },
    {
      "epoch": 2.120685577392578e-05,
      "model_forward_time": 1.7531757354736328,
      "step": 22237
    },
    {
      "epoch": 2.120685577392578e-05,
      "step": 22237,
      "training_step_time": 8.356043100357056
    },
    {
      "epoch": 2.1207809448242187e-05,
      "model_forward_time": 1.843963384628296,
      "step": 22238
    },
    {
      "epoch": 2.1207809448242187e-05,
      "step": 22238,
      "training_step_time": 7.787682771682739
    },
    {
      "epoch": 2.1208763122558594e-05,
      "model_forward_time": 1.7529056072235107,
      "step": 22239
    },
    {
      "epoch": 2.1208763122558594e-05,
      "step": 22239,
      "training_step_time": 7.495345115661621
    },
    {
      "epoch": 2.1209716796875e-05,
      "grad_norm": 0.12005316466093063,
      "learning_rate": 9.209280040397875e-06,
      "loss": 0.0652,
      "step": 22240
    },
    {
      "epoch": 2.1209716796875e-05,
      "model_forward_time": 1.7361774444580078,
      "step": 22240
    },
    {
      "dynamics_loss_avg": 0.058598111011087896,
      "epoch": 2.1209716796875e-05,
      "step": 22240
    },
    {
      "action_loss_avg": 0.0026520831044763325,
      "epoch": 2.1209716796875e-05,
      "step": 22240
    },
    {
      "epoch": 2.1209716796875e-05,
      "step": 22240,
      "training_step_time": 9.22760558128357
    },
    {
      "epoch": 2.1210670471191406e-05,
      "model_forward_time": 2.9064900875091553,
      "step": 22241
    },
    {
      "epoch": 2.1210670471191406e-05,
      "step": 22241,
      "training_step_time": 8.21774411201477
    },
    {
      "epoch": 2.1211624145507813e-05,
      "model_forward_time": 1.7587084770202637,
      "step": 22242
    },
    {
      "epoch": 2.1211624145507813e-05,
      "step": 22242,
      "training_step_time": 7.4974119663238525
    },
    {
      "epoch": 2.1212577819824217e-05,
      "model_forward_time": 1.754728078842163,
      "step": 22243
    },
    {
      "epoch": 2.1212577819824217e-05,
      "step": 22243,
      "training_step_time": 8.292556047439575
    },
    {
      "epoch": 2.1213531494140625e-05,
      "model_forward_time": 1.753873586654663,
      "step": 22244
    },
    {
      "epoch": 2.1213531494140625e-05,
      "step": 22244,
      "training_step_time": 7.600070953369141
    },
    {
      "epoch": 2.1214485168457032e-05,
      "model_forward_time": 1.7586777210235596,
      "step": 22245
    },
    {
      "epoch": 2.1214485168457032e-05,
      "step": 22245,
      "training_step_time": 6.74827766418457
    },
    {
      "epoch": 2.1215438842773436e-05,
      "model_forward_time": 1.750760793685913,
      "step": 22246
    },
    {
      "epoch": 2.1215438842773436e-05,
      "step": 22246,
      "training_step_time": 7.696091651916504
    },
    {
      "epoch": 2.1216392517089844e-05,
      "model_forward_time": 1.8563895225524902,
      "step": 22247
    },
    {
      "epoch": 2.1216392517089844e-05,
      "step": 22247,
      "training_step_time": 8.609342813491821
    },
    {
      "epoch": 2.121734619140625e-05,
      "model_forward_time": 1.752516269683838,
      "step": 22248
    },
    {
      "epoch": 2.121734619140625e-05,
      "step": 22248,
      "training_step_time": 8.144813060760498
    },
    {
      "epoch": 2.1218299865722655e-05,
      "model_forward_time": 1.7525420188903809,
      "step": 22249
    },
    {
      "epoch": 2.1218299865722655e-05,
      "step": 22249,
      "training_step_time": 8.725230932235718
    },
    {
      "epoch": 2.1219253540039062e-05,
      "grad_norm": 0.19228185713291168,
      "learning_rate": 9.208387428960269e-06,
      "loss": 0.0632,
      "step": 22250
    },
    {
      "epoch": 2.1219253540039062e-05,
      "model_forward_time": 2.914217948913574,
      "step": 22250
    },
    {
      "dynamics_loss_avg": 0.0716366708278656,
      "epoch": 2.1219253540039062e-05,
      "step": 22250
    },
    {
      "action_loss_avg": 0.0033427810107241386,
      "epoch": 2.1219253540039062e-05,
      "step": 22250
    },
    {
      "epoch": 2.1219253540039062e-05,
      "step": 22250,
      "training_step_time": 9.723636150360107
    },
    {
      "epoch": 2.122020721435547e-05,
      "model_forward_time": 1.7501907348632812,
      "step": 22251
    },
    {
      "epoch": 2.122020721435547e-05,
      "step": 22251,
      "training_step_time": 7.511878728866577
    },
    {
      "epoch": 2.1221160888671874e-05,
      "model_forward_time": 1.748037338256836,
      "step": 22252
    },
    {
      "epoch": 2.1221160888671874e-05,
      "step": 22252,
      "training_step_time": 7.614977121353149
    },
    {
      "epoch": 2.122211456298828e-05,
      "model_forward_time": 1.7358896732330322,
      "step": 22253
    },
    {
      "epoch": 2.122211456298828e-05,
      "step": 22253,
      "training_step_time": 7.617368459701538
    },
    {
      "epoch": 2.122306823730469e-05,
      "model_forward_time": 1.746424913406372,
      "step": 22254
    },
    {
      "epoch": 2.122306823730469e-05,
      "step": 22254,
      "training_step_time": 6.779693365097046
    },
    {
      "epoch": 2.1224021911621093e-05,
      "model_forward_time": 1.7570021152496338,
      "step": 22255
    },
    {
      "epoch": 2.1224021911621093e-05,
      "step": 22255,
      "training_step_time": 7.634019136428833
    },
    {
      "epoch": 2.12249755859375e-05,
      "model_forward_time": 1.8466064929962158,
      "step": 22256
    },
    {
      "epoch": 2.12249755859375e-05,
      "step": 22256,
      "training_step_time": 7.739791393280029
    },
    {
      "epoch": 2.1225929260253908e-05,
      "model_forward_time": 1.7564854621887207,
      "step": 22257
    },
    {
      "epoch": 2.1225929260253908e-05,
      "step": 22257,
      "training_step_time": 7.406954526901245
    },
    {
      "epoch": 2.122688293457031e-05,
      "model_forward_time": 1.755230188369751,
      "step": 22258
    },
    {
      "epoch": 2.122688293457031e-05,
      "step": 22258,
      "training_step_time": 7.658277273178101
    },
    {
      "epoch": 2.122783660888672e-05,
      "model_forward_time": 2.8953323364257812,
      "step": 22259
    },
    {
      "epoch": 2.122783660888672e-05,
      "step": 22259,
      "training_step_time": 9.403789520263672
    },
    {
      "epoch": 2.1228790283203126e-05,
      "grad_norm": 0.13848401606082916,
      "learning_rate": 9.20749435729974e-06,
      "loss": 0.0629,
      "step": 22260
    },
    {
      "epoch": 2.1228790283203126e-05,
      "model_forward_time": 1.7506098747253418,
      "step": 22260
    },
    {
      "dynamics_loss_avg": 0.05641629230231047,
      "epoch": 2.1228790283203126e-05,
      "step": 22260
    },
    {
      "action_loss_avg": 0.004051860515028238,
      "epoch": 2.1228790283203126e-05,
      "step": 22260
    },
    {
      "epoch": 2.1228790283203126e-05,
      "step": 22260,
      "training_step_time": 7.563938856124878
    },
    {
      "epoch": 2.122974395751953e-05,
      "model_forward_time": 1.7559564113616943,
      "step": 22261
    },
    {
      "epoch": 2.122974395751953e-05,
      "step": 22261,
      "training_step_time": 7.5428783893585205
    },
    {
      "epoch": 2.1230697631835938e-05,
      "model_forward_time": 1.753432273864746,
      "step": 22262
    },
    {
      "epoch": 2.1230697631835938e-05,
      "step": 22262,
      "training_step_time": 7.615782737731934
    },
    {
      "epoch": 2.1231651306152345e-05,
      "model_forward_time": 1.7629244327545166,
      "step": 22263
    },
    {
      "epoch": 2.1231651306152345e-05,
      "step": 22263,
      "training_step_time": 6.979490041732788
    },
    {
      "epoch": 2.123260498046875e-05,
      "model_forward_time": 1.7435245513916016,
      "step": 22264
    },
    {
      "epoch": 2.123260498046875e-05,
      "step": 22264,
      "training_step_time": 7.601433753967285
    },
    {
      "epoch": 2.1233558654785157e-05,
      "model_forward_time": 1.8455493450164795,
      "step": 22265
    },
    {
      "epoch": 2.1233558654785157e-05,
      "step": 22265,
      "training_step_time": 7.735551595687866
    },
    {
      "epoch": 2.123451232910156e-05,
      "model_forward_time": 1.071485996246338,
      "step": 22266
    },
    {
      "epoch": 2.123451232910156e-05,
      "step": 22266,
      "training_step_time": 7.485806941986084
    },
    {
      "epoch": 2.1235466003417968e-05,
      "model_forward_time": 1.7584023475646973,
      "step": 22267
    },
    {
      "epoch": 2.1235466003417968e-05,
      "step": 22267,
      "training_step_time": 7.575467586517334
    },
    {
      "epoch": 2.1236419677734376e-05,
      "model_forward_time": 1.7576406002044678,
      "step": 22268
    },
    {
      "epoch": 2.1236419677734376e-05,
      "step": 22268,
      "training_step_time": 8.73093295097351
    },
    {
      "epoch": 2.123737335205078e-05,
      "model_forward_time": 2.92692232131958,
      "step": 22269
    },
    {
      "epoch": 2.123737335205078e-05,
      "step": 22269,
      "training_step_time": 9.478664636611938
    },
    {
      "epoch": 2.1238327026367187e-05,
      "grad_norm": 0.14101816713809967,
      "learning_rate": 9.206600825513958e-06,
      "loss": 0.0642,
      "step": 22270
    },
    {
      "epoch": 2.1238327026367187e-05,
      "model_forward_time": 1.74934720993042,
      "step": 22270
    },
    {
      "dynamics_loss_avg": 0.07648515095934272,
      "epoch": 2.1238327026367187e-05,
      "step": 22270
    },
    {
      "action_loss_avg": 0.005291697004577145,
      "epoch": 2.1238327026367187e-05,
      "step": 22270
    },
    {
      "epoch": 2.1238327026367187e-05,
      "step": 22270,
      "training_step_time": 7.654768705368042
    },
    {
      "epoch": 2.1239280700683595e-05,
      "model_forward_time": 1.756617546081543,
      "step": 22271
    },
    {
      "epoch": 2.1239280700683595e-05,
      "step": 22271,
      "training_step_time": 9.169919490814209
    },
    {
      "epoch": 2.1240234375e-05,
      "model_forward_time": 1.7564148902893066,
      "step": 22272
    },
    {
      "epoch": 2.1240234375e-05,
      "step": 22272,
      "training_step_time": 7.563600540161133
    },
    {
      "epoch": 2.1241188049316406e-05,
      "model_forward_time": 1.7497546672821045,
      "step": 22273
    },
    {
      "epoch": 2.1241188049316406e-05,
      "step": 22273,
      "training_step_time": 7.579541921615601
    },
    {
      "epoch": 2.1242141723632813e-05,
      "model_forward_time": 1.8413801193237305,
      "step": 22274
    },
    {
      "epoch": 2.1242141723632813e-05,
      "step": 22274,
      "training_step_time": 7.780714273452759
    },
    {
      "epoch": 2.1243095397949217e-05,
      "model_forward_time": 1.7545390129089355,
      "step": 22275
    },
    {
      "epoch": 2.1243095397949217e-05,
      "step": 22275,
      "training_step_time": 8.088258743286133
    },
    {
      "epoch": 2.1244049072265625e-05,
      "model_forward_time": 1.7520720958709717,
      "step": 22276
    },
    {
      "epoch": 2.1244049072265625e-05,
      "step": 22276,
      "training_step_time": 8.579520225524902
    },
    {
      "epoch": 2.1245002746582032e-05,
      "model_forward_time": 1.752136468887329,
      "step": 22277
    },
    {
      "epoch": 2.1245002746582032e-05,
      "step": 22277,
      "training_step_time": 7.633859872817993
    },
    {
      "epoch": 2.1245956420898436e-05,
      "model_forward_time": 2.885093927383423,
      "step": 22278
    },
    {
      "epoch": 2.1245956420898436e-05,
      "step": 22278,
      "training_step_time": 8.189322710037231
    },
    {
      "epoch": 2.1246910095214844e-05,
      "model_forward_time": 1.7530598640441895,
      "step": 22279
    },
    {
      "epoch": 2.1246910095214844e-05,
      "step": 22279,
      "training_step_time": 8.58089828491211
    },
    {
      "epoch": 2.124786376953125e-05,
      "grad_norm": 0.18320363759994507,
      "learning_rate": 9.20570683370063e-06,
      "loss": 0.0626,
      "step": 22280
    },
    {
      "epoch": 2.124786376953125e-05,
      "model_forward_time": 1.7434282302856445,
      "step": 22280
    },
    {
      "dynamics_loss_avg": 0.06607024632394314,
      "epoch": 2.124786376953125e-05,
      "step": 22280
    },
    {
      "action_loss_avg": 0.0029794224188663065,
      "epoch": 2.124786376953125e-05,
      "step": 22280
    },
    {
      "epoch": 2.124786376953125e-05,
      "step": 22280,
      "training_step_time": 7.677435398101807
    },
    {
      "epoch": 2.1248817443847655e-05,
      "model_forward_time": 1.3903367519378662,
      "step": 22281
    },
    {
      "epoch": 2.1248817443847655e-05,
      "step": 22281,
      "training_step_time": 7.593526124954224
    },
    {
      "epoch": 2.1249771118164063e-05,
      "model_forward_time": 1.7473011016845703,
      "step": 22282
    },
    {
      "epoch": 2.1249771118164063e-05,
      "step": 22282,
      "training_step_time": 7.538009405136108
    },
    {
      "epoch": 2.125072479248047e-05,
      "model_forward_time": 1.844618320465088,
      "step": 22283
    },
    {
      "epoch": 2.125072479248047e-05,
      "step": 22283,
      "training_step_time": 7.765093803405762
    },
    {
      "epoch": 2.1251678466796874e-05,
      "model_forward_time": 1.7552769184112549,
      "step": 22284
    },
    {
      "epoch": 2.1251678466796874e-05,
      "step": 22284,
      "training_step_time": 7.598731517791748
    },
    {
      "epoch": 2.125263214111328e-05,
      "model_forward_time": 1.7480580806732178,
      "step": 22285
    },
    {
      "epoch": 2.125263214111328e-05,
      "step": 22285,
      "training_step_time": 8.226910829544067
    },
    {
      "epoch": 2.125358581542969e-05,
      "model_forward_time": 1.7574207782745361,
      "step": 22286
    },
    {
      "epoch": 2.125358581542969e-05,
      "step": 22286,
      "training_step_time": 7.612254619598389
    },
    {
      "epoch": 2.1254539489746093e-05,
      "model_forward_time": 2.8679392337799072,
      "step": 22287
    },
    {
      "epoch": 2.1254539489746093e-05,
      "step": 22287,
      "training_step_time": 7.668687582015991
    },
    {
      "epoch": 2.12554931640625e-05,
      "model_forward_time": 1.748044490814209,
      "step": 22288
    },
    {
      "epoch": 2.12554931640625e-05,
      "step": 22288,
      "training_step_time": 7.596572160720825
    },
    {
      "epoch": 2.1256446838378908e-05,
      "model_forward_time": 1.7579379081726074,
      "step": 22289
    },
    {
      "epoch": 2.1256446838378908e-05,
      "step": 22289,
      "training_step_time": 7.616121530532837
    },
    {
      "epoch": 2.125740051269531e-05,
      "grad_norm": 0.16388210654258728,
      "learning_rate": 9.204812381957529e-06,
      "loss": 0.0618,
      "step": 22290
    },
    {
      "epoch": 2.125740051269531e-05,
      "model_forward_time": 1.7550444602966309,
      "step": 22290
    },
    {
      "dynamics_loss_avg": 0.07023531757295132,
      "epoch": 2.125740051269531e-05,
      "step": 22290
    },
    {
      "action_loss_avg": 0.004447525058640167,
      "epoch": 2.125740051269531e-05,
      "step": 22290
    },
    {
      "epoch": 2.125740051269531e-05,
      "step": 22290,
      "training_step_time": 7.613956928253174
    },
    {
      "epoch": 2.125835418701172e-05,
      "model_forward_time": 1.742767572402954,
      "step": 22291
    },
    {
      "epoch": 2.125835418701172e-05,
      "step": 22291,
      "training_step_time": 6.7946484088897705
    },
    {
      "epoch": 2.1259307861328127e-05,
      "model_forward_time": 1.8604767322540283,
      "step": 22292
    },
    {
      "epoch": 2.1259307861328127e-05,
      "step": 22292,
      "training_step_time": 7.833497047424316
    },
    {
      "epoch": 2.126026153564453e-05,
      "model_forward_time": 1.742591142654419,
      "step": 22293
    },
    {
      "epoch": 2.126026153564453e-05,
      "step": 22293,
      "training_step_time": 7.560164928436279
    },
    {
      "epoch": 2.1261215209960938e-05,
      "model_forward_time": 1.7560434341430664,
      "step": 22294
    },
    {
      "epoch": 2.1261215209960938e-05,
      "step": 22294,
      "training_step_time": 7.416804313659668
    },
    {
      "epoch": 2.1262168884277345e-05,
      "model_forward_time": 0.739232063293457,
      "step": 22295
    },
    {
      "epoch": 2.1262168884277345e-05,
      "step": 22295,
      "training_step_time": 7.645094633102417
    },
    {
      "epoch": 2.126312255859375e-05,
      "model_forward_time": 1.7421238422393799,
      "step": 22296
    },
    {
      "epoch": 2.126312255859375e-05,
      "step": 22296,
      "training_step_time": 9.724574089050293
    },
    {
      "epoch": 2.1264076232910157e-05,
      "model_forward_time": 2.843020439147949,
      "step": 22297
    },
    {
      "epoch": 2.1264076232910157e-05,
      "step": 22297,
      "training_step_time": 8.473716497421265
    },
    {
      "epoch": 2.126502990722656e-05,
      "model_forward_time": 1.7507328987121582,
      "step": 22298
    },
    {
      "epoch": 2.126502990722656e-05,
      "step": 22298,
      "training_step_time": 8.195504188537598
    },
    {
      "epoch": 2.1265983581542968e-05,
      "model_forward_time": 1.7544174194335938,
      "step": 22299
    },
    {
      "epoch": 2.1265983581542968e-05,
      "step": 22299,
      "training_step_time": 7.571518898010254
    },
    {
      "epoch": 2.1266937255859376e-05,
      "grad_norm": 0.1818844974040985,
      "learning_rate": 9.203917470382466e-06,
      "loss": 0.0641,
      "step": 22300
    },
    {
      "epoch": 2.1266937255859376e-05,
      "model_forward_time": 1.75541090965271,
      "step": 22300
    },
    {
      "dynamics_loss_avg": 0.053083505295217034,
      "epoch": 2.1266937255859376e-05,
      "step": 22300
    },
    {
      "action_loss_avg": 0.0034446616045897827,
      "epoch": 2.1266937255859376e-05,
      "step": 22300
    },
    {
      "epoch": 2.1266937255859376e-05,
      "step": 22300,
      "training_step_time": 6.7546374797821045
    },
    {
      "epoch": 2.126789093017578e-05,
      "model_forward_time": 1.855830430984497,
      "step": 22301
    },
    {
      "epoch": 2.126789093017578e-05,
      "step": 22301,
      "training_step_time": 8.4414644241333
    },
    {
      "epoch": 2.1268844604492187e-05,
      "model_forward_time": 1.7590878009796143,
      "step": 22302
    },
    {
      "epoch": 2.1268844604492187e-05,
      "step": 22302,
      "training_step_time": 7.541685104370117
    },
    {
      "epoch": 2.1269798278808595e-05,
      "model_forward_time": 1.7322871685028076,
      "step": 22303
    },
    {
      "epoch": 2.1269798278808595e-05,
      "step": 22303,
      "training_step_time": 7.568683862686157
    },
    {
      "epoch": 2.1270751953125e-05,
      "model_forward_time": 1.7564740180969238,
      "step": 22304
    },
    {
      "epoch": 2.1270751953125e-05,
      "step": 22304,
      "training_step_time": 8.403531074523926
    },
    {
      "epoch": 2.1271705627441406e-05,
      "model_forward_time": 1.7535674571990967,
      "step": 22305
    },
    {
      "epoch": 2.1271705627441406e-05,
      "step": 22305,
      "training_step_time": 7.698349237442017
    },
    {
      "epoch": 2.1272659301757813e-05,
      "model_forward_time": 2.8515872955322266,
      "step": 22306
    },
    {
      "epoch": 2.1272659301757813e-05,
      "step": 22306,
      "training_step_time": 7.632418870925903
    },
    {
      "epoch": 2.1273612976074217e-05,
      "model_forward_time": 1.755159854888916,
      "step": 22307
    },
    {
      "epoch": 2.1273612976074217e-05,
      "step": 22307,
      "training_step_time": 8.553246974945068
    },
    {
      "epoch": 2.1274566650390625e-05,
      "model_forward_time": 1.0625617504119873,
      "step": 22308
    },
    {
      "epoch": 2.1274566650390625e-05,
      "step": 22308,
      "training_step_time": 8.574480772018433
    },
    {
      "epoch": 2.1275520324707032e-05,
      "model_forward_time": 1.7450437545776367,
      "step": 22309
    },
    {
      "epoch": 2.1275520324707032e-05,
      "step": 22309,
      "training_step_time": 7.1444079875946045
    },
    {
      "epoch": 2.1276473999023436e-05,
      "grad_norm": 0.23226283490657806,
      "learning_rate": 9.20302209907331e-06,
      "loss": 0.0615,
      "step": 22310
    },
    {
      "epoch": 2.1276473999023436e-05,
      "model_forward_time": 1.8584198951721191,
      "step": 22310
    },
    {
      "dynamics_loss_avg": 0.04371064491569996,
      "epoch": 2.1276473999023436e-05,
      "step": 22310
    },
    {
      "action_loss_avg": 0.0026666574471164496,
      "epoch": 2.1276473999023436e-05,
      "step": 22310
    },
    {
      "epoch": 2.1276473999023436e-05,
      "step": 22310,
      "training_step_time": 7.815388917922974
    },
    {
      "epoch": 2.1277427673339844e-05,
      "model_forward_time": 1.751189947128296,
      "step": 22311
    },
    {
      "epoch": 2.1277427673339844e-05,
      "step": 22311,
      "training_step_time": 7.557775259017944
    },
    {
      "epoch": 2.127838134765625e-05,
      "model_forward_time": 1.7613658905029297,
      "step": 22312
    },
    {
      "epoch": 2.127838134765625e-05,
      "step": 22312,
      "training_step_time": 7.553096771240234
    },
    {
      "epoch": 2.1279335021972655e-05,
      "model_forward_time": 1.7214083671569824,
      "step": 22313
    },
    {
      "epoch": 2.1279335021972655e-05,
      "step": 22313,
      "training_step_time": 8.333682537078857
    },
    {
      "epoch": 2.1280288696289063e-05,
      "model_forward_time": 1.7427363395690918,
      "step": 22314
    },
    {
      "epoch": 2.1280288696289063e-05,
      "step": 22314,
      "training_step_time": 9.219623327255249
    },
    {
      "epoch": 2.128124237060547e-05,
      "model_forward_time": 1.740600347518921,
      "step": 22315
    },
    {
      "epoch": 2.128124237060547e-05,
      "step": 22315,
      "training_step_time": 8.220154523849487
    },
    {
      "epoch": 2.1282196044921874e-05,
      "model_forward_time": 2.863924741744995,
      "step": 22316
    },
    {
      "epoch": 2.1282196044921874e-05,
      "step": 22316,
      "training_step_time": 7.5998616218566895
    },
    {
      "epoch": 2.128314971923828e-05,
      "model_forward_time": 1.7539098262786865,
      "step": 22317
    },
    {
      "epoch": 2.128314971923828e-05,
      "step": 22317,
      "training_step_time": 7.553939580917358
    },
    {
      "epoch": 2.128410339355469e-05,
      "model_forward_time": 1.7589852809906006,
      "step": 22318
    },
    {
      "epoch": 2.128410339355469e-05,
      "step": 22318,
      "training_step_time": 7.620011568069458
    },
    {
      "epoch": 2.1285057067871093e-05,
      "model_forward_time": 1.8364691734313965,
      "step": 22319
    },
    {
      "epoch": 2.1285057067871093e-05,
      "step": 22319,
      "training_step_time": 7.782854795455933
    },
    {
      "epoch": 2.12860107421875e-05,
      "grad_norm": 0.16438283026218414,
      "learning_rate": 9.202126268127975e-06,
      "loss": 0.0627,
      "step": 22320
    },
    {
      "epoch": 2.12860107421875e-05,
      "model_forward_time": 1.7439885139465332,
      "step": 22320
    },
    {
      "dynamics_loss_avg": 0.05544254705309868,
      "epoch": 2.12860107421875e-05,
      "step": 22320
    },
    {
      "action_loss_avg": 0.0032795542501844466,
      "epoch": 2.12860107421875e-05,
      "step": 22320
    },
    {
      "epoch": 2.12860107421875e-05,
      "step": 22320,
      "training_step_time": 8.287540912628174
    },
    {
      "epoch": 2.1286964416503908e-05,
      "model_forward_time": 1.756462574005127,
      "step": 22321
    },
    {
      "epoch": 2.1286964416503908e-05,
      "step": 22321,
      "training_step_time": 7.62389349937439
    },
    {
      "epoch": 2.1287918090820312e-05,
      "model_forward_time": 1.751081943511963,
      "step": 22322
    },
    {
      "epoch": 2.1287918090820312e-05,
      "step": 22322,
      "training_step_time": 7.560510873794556
    },
    {
      "epoch": 2.128887176513672e-05,
      "model_forward_time": 1.3611984252929688,
      "step": 22323
    },
    {
      "epoch": 2.128887176513672e-05,
      "step": 22323,
      "training_step_time": 8.594455480575562
    },
    {
      "epoch": 2.1289825439453127e-05,
      "model_forward_time": 1.75687575340271,
      "step": 22324
    },
    {
      "epoch": 2.1289825439453127e-05,
      "step": 22324,
      "training_step_time": 7.613911151885986
    },
    {
      "epoch": 2.129077911376953e-05,
      "model_forward_time": 2.9155519008636475,
      "step": 22325
    },
    {
      "epoch": 2.129077911376953e-05,
      "step": 22325,
      "training_step_time": 7.671465158462524
    },
    {
      "epoch": 2.1291732788085938e-05,
      "model_forward_time": 1.7537510395050049,
      "step": 22326
    },
    {
      "epoch": 2.1291732788085938e-05,
      "step": 22326,
      "training_step_time": 7.6079981327056885
    },
    {
      "epoch": 2.1292686462402345e-05,
      "model_forward_time": 1.7477073669433594,
      "step": 22327
    },
    {
      "epoch": 2.1292686462402345e-05,
      "step": 22327,
      "training_step_time": 7.588086366653442
    },
    {
      "epoch": 2.129364013671875e-05,
      "model_forward_time": 1.814011812210083,
      "step": 22328
    },
    {
      "epoch": 2.129364013671875e-05,
      "step": 22328,
      "training_step_time": 7.815733432769775
    },
    {
      "epoch": 2.1294593811035157e-05,
      "model_forward_time": 1.7672677040100098,
      "step": 22329
    },
    {
      "epoch": 2.1294593811035157e-05,
      "step": 22329,
      "training_step_time": 7.571296691894531
    },
    {
      "epoch": 2.129554748535156e-05,
      "grad_norm": 0.1653403639793396,
      "learning_rate": 9.20122997764443e-06,
      "loss": 0.0646,
      "step": 22330
    },
    {
      "epoch": 2.129554748535156e-05,
      "model_forward_time": 1.7380869388580322,
      "step": 22330
    },
    {
      "dynamics_loss_avg": 0.05477829687297344,
      "epoch": 2.129554748535156e-05,
      "step": 22330
    },
    {
      "action_loss_avg": 0.0025226277823094277,
      "epoch": 2.129554748535156e-05,
      "step": 22330
    },
    {
      "epoch": 2.129554748535156e-05,
      "step": 22330,
      "training_step_time": 7.4933836460113525
    },
    {
      "epoch": 2.1296501159667968e-05,
      "model_forward_time": 1.396714210510254,
      "step": 22331
    },
    {
      "epoch": 2.1296501159667968e-05,
      "step": 22331,
      "training_step_time": 8.120449542999268
    },
    {
      "epoch": 2.1297454833984376e-05,
      "model_forward_time": 1.3887736797332764,
      "step": 22332
    },
    {
      "epoch": 2.1297454833984376e-05,
      "step": 22332,
      "training_step_time": 7.698924779891968
    },
    {
      "epoch": 2.129840850830078e-05,
      "model_forward_time": 1.7483940124511719,
      "step": 22333
    },
    {
      "epoch": 2.129840850830078e-05,
      "step": 22333,
      "training_step_time": 8.598105192184448
    },
    {
      "epoch": 2.1299362182617187e-05,
      "model_forward_time": 2.8458645343780518,
      "step": 22334
    },
    {
      "epoch": 2.1299362182617187e-05,
      "step": 22334,
      "training_step_time": 8.327199697494507
    },
    {
      "epoch": 2.1300315856933595e-05,
      "model_forward_time": 1.746598243713379,
      "step": 22335
    },
    {
      "epoch": 2.1300315856933595e-05,
      "step": 22335,
      "training_step_time": 7.556476354598999
    },
    {
      "epoch": 2.130126953125e-05,
      "model_forward_time": 1.7552664279937744,
      "step": 22336
    },
    {
      "epoch": 2.130126953125e-05,
      "step": 22336,
      "training_step_time": 8.10541033744812
    },
    {
      "epoch": 2.1302223205566406e-05,
      "model_forward_time": 1.7286911010742188,
      "step": 22337
    },
    {
      "epoch": 2.1302223205566406e-05,
      "step": 22337,
      "training_step_time": 7.686477422714233
    },
    {
      "epoch": 2.1303176879882813e-05,
      "model_forward_time": 1.7539496421813965,
      "step": 22338
    },
    {
      "epoch": 2.1303176879882813e-05,
      "step": 22338,
      "training_step_time": 7.880642890930176
    },
    {
      "epoch": 2.1304130554199217e-05,
      "model_forward_time": 1.7602760791778564,
      "step": 22339
    },
    {
      "epoch": 2.1304130554199217e-05,
      "step": 22339,
      "training_step_time": 7.53657341003418
    },
    {
      "epoch": 2.1305084228515625e-05,
      "grad_norm": 0.12750090658664703,
      "learning_rate": 9.200333227720692e-06,
      "loss": 0.061,
      "step": 22340
    },
    {
      "epoch": 2.1305084228515625e-05,
      "model_forward_time": 1.7392044067382812,
      "step": 22340
    },
    {
      "dynamics_loss_avg": 0.06112288720905781,
      "epoch": 2.1305084228515625e-05,
      "step": 22340
    },
    {
      "action_loss_avg": 0.002914266288280487,
      "epoch": 2.1305084228515625e-05,
      "step": 22340
    },
    {
      "epoch": 2.1305084228515625e-05,
      "step": 22340,
      "training_step_time": 8.343139886856079
    },
    {
      "epoch": 2.1306037902832032e-05,
      "model_forward_time": 1.7644197940826416,
      "step": 22341
    },
    {
      "epoch": 2.1306037902832032e-05,
      "step": 22341,
      "training_step_time": 7.710008382797241
    },
    {
      "epoch": 2.1306991577148436e-05,
      "model_forward_time": 1.7596604824066162,
      "step": 22342
    },
    {
      "epoch": 2.1306991577148436e-05,
      "step": 22342,
      "training_step_time": 7.646936416625977
    },
    {
      "epoch": 2.1307945251464844e-05,
      "model_forward_time": 1.755824327468872,
      "step": 22343
    },
    {
      "epoch": 2.1307945251464844e-05,
      "step": 22343,
      "training_step_time": 7.698059320449829
    },
    {
      "epoch": 2.130889892578125e-05,
      "model_forward_time": 2.921109437942505,
      "step": 22344
    },
    {
      "epoch": 2.130889892578125e-05,
      "step": 22344,
      "training_step_time": 7.660082101821899
    },
    {
      "epoch": 2.1309852600097655e-05,
      "model_forward_time": 1.7607049942016602,
      "step": 22345
    },
    {
      "epoch": 2.1309852600097655e-05,
      "step": 22345,
      "training_step_time": 7.575884103775024
    },
    {
      "epoch": 2.1310806274414063e-05,
      "model_forward_time": 1.7569797039031982,
      "step": 22346
    },
    {
      "epoch": 2.1310806274414063e-05,
      "step": 22346,
      "training_step_time": 7.653281927108765
    },
    {
      "epoch": 2.131175994873047e-05,
      "model_forward_time": 1.8435707092285156,
      "step": 22347
    },
    {
      "epoch": 2.131175994873047e-05,
      "step": 22347,
      "training_step_time": 7.781803369522095
    },
    {
      "epoch": 2.1312713623046874e-05,
      "model_forward_time": 1.7467041015625,
      "step": 22348
    },
    {
      "epoch": 2.1312713623046874e-05,
      "step": 22348,
      "training_step_time": 8.284710168838501
    },
    {
      "epoch": 2.131366729736328e-05,
      "model_forward_time": 1.7548646926879883,
      "step": 22349
    },
    {
      "epoch": 2.131366729736328e-05,
      "step": 22349,
      "training_step_time": 7.551692962646484
    },
    {
      "epoch": 2.131462097167969e-05,
      "grad_norm": 0.18131613731384277,
      "learning_rate": 9.199436018454826e-06,
      "loss": 0.0636,
      "step": 22350
    },
    {
      "epoch": 2.131462097167969e-05,
      "model_forward_time": 1.7493987083435059,
      "step": 22350
    },
    {
      "dynamics_loss_avg": 0.054098059237003324,
      "epoch": 2.131462097167969e-05,
      "step": 22350
    },
    {
      "action_loss_avg": 0.003329483192646876,
      "epoch": 2.131462097167969e-05,
      "step": 22350
    },
    {
      "epoch": 2.131462097167969e-05,
      "step": 22350,
      "training_step_time": 8.085761308670044
    },
    {
      "epoch": 2.1315574645996093e-05,
      "model_forward_time": 1.41363525390625,
      "step": 22351
    },
    {
      "epoch": 2.1315574645996093e-05,
      "step": 22351,
      "training_step_time": 8.705613136291504
    },
    {
      "epoch": 2.13165283203125e-05,
      "model_forward_time": 1.7581615447998047,
      "step": 22352
    },
    {
      "epoch": 2.13165283203125e-05,
      "step": 22352,
      "training_step_time": 6.7510693073272705
    },
    {
      "epoch": 2.1317481994628908e-05,
      "model_forward_time": 1.748520851135254,
      "step": 22353
    },
    {
      "epoch": 2.1317481994628908e-05,
      "step": 22353,
      "training_step_time": 6.729989051818848
    },
    {
      "epoch": 2.1318435668945312e-05,
      "model_forward_time": 1.7568578720092773,
      "step": 22354
    },
    {
      "epoch": 2.1318435668945312e-05,
      "step": 22354,
      "training_step_time": 6.771593809127808
    },
    {
      "epoch": 2.131938934326172e-05,
      "model_forward_time": 1.0631463527679443,
      "step": 22355
    },
    {
      "epoch": 2.131938934326172e-05,
      "step": 22355,
      "training_step_time": 6.677462100982666
    },
    {
      "epoch": 2.1320343017578127e-05,
      "model_forward_time": 1.7482085227966309,
      "step": 22356
    },
    {
      "epoch": 2.1320343017578127e-05,
      "step": 22356,
      "training_step_time": 6.764280796051025
    },
    {
      "epoch": 2.132129669189453e-05,
      "model_forward_time": 1.7520744800567627,
      "step": 22357
    },
    {
      "epoch": 2.132129669189453e-05,
      "step": 22357,
      "training_step_time": 6.756634473800659
    },
    {
      "epoch": 2.1322250366210938e-05,
      "model_forward_time": 1.7573442459106445,
      "step": 22358
    },
    {
      "epoch": 2.1322250366210938e-05,
      "step": 22358,
      "training_step_time": 6.747483968734741
    },
    {
      "epoch": 2.1323204040527345e-05,
      "model_forward_time": 1.7535293102264404,
      "step": 22359
    },
    {
      "epoch": 2.1323204040527345e-05,
      "step": 22359,
      "training_step_time": 7.567386627197266
    },
    {
      "epoch": 2.132415771484375e-05,
      "grad_norm": 0.18192274868488312,
      "learning_rate": 9.198538349944953e-06,
      "loss": 0.0614,
      "step": 22360
    },
    {
      "epoch": 2.132415771484375e-05,
      "model_forward_time": 2.8900575637817383,
      "step": 22360
    },
    {
      "dynamics_loss_avg": 0.0558012917637825,
      "epoch": 2.132415771484375e-05,
      "step": 22360
    },
    {
      "action_loss_avg": 0.0021762155869510026,
      "epoch": 2.132415771484375e-05,
      "step": 22360
    },
    {
      "epoch": 2.132415771484375e-05,
      "step": 22360,
      "training_step_time": 9.757317543029785
    },
    {
      "epoch": 2.1325111389160157e-05,
      "model_forward_time": 1.7618260383605957,
      "step": 22361
    },
    {
      "epoch": 2.1325111389160157e-05,
      "step": 22361,
      "training_step_time": 7.627065896987915
    },
    {
      "epoch": 2.132606506347656e-05,
      "model_forward_time": 1.7539427280426025,
      "step": 22362
    },
    {
      "epoch": 2.132606506347656e-05,
      "step": 22362,
      "training_step_time": 6.7457263469696045
    },
    {
      "epoch": 2.132701873779297e-05,
      "model_forward_time": 1.755659818649292,
      "step": 22363
    },
    {
      "epoch": 2.132701873779297e-05,
      "step": 22363,
      "training_step_time": 6.742152452468872
    },
    {
      "epoch": 2.1327972412109376e-05,
      "model_forward_time": 1.759495496749878,
      "step": 22364
    },
    {
      "epoch": 2.1327972412109376e-05,
      "step": 22364,
      "training_step_time": 6.738032102584839
    },
    {
      "epoch": 2.132892608642578e-05,
      "model_forward_time": 1.7493774890899658,
      "step": 22365
    },
    {
      "epoch": 2.132892608642578e-05,
      "step": 22365,
      "training_step_time": 6.68857216835022
    },
    {
      "epoch": 2.1329879760742187e-05,
      "model_forward_time": 1.7533447742462158,
      "step": 22366
    },
    {
      "epoch": 2.1329879760742187e-05,
      "step": 22366,
      "training_step_time": 6.747068166732788
    },
    {
      "epoch": 2.1330833435058595e-05,
      "model_forward_time": 1.3797526359558105,
      "step": 22367
    },
    {
      "epoch": 2.1330833435058595e-05,
      "step": 22367,
      "training_step_time": 6.722110033035278
    },
    {
      "epoch": 2.1331787109375e-05,
      "model_forward_time": 1.7629451751708984,
      "step": 22368
    },
    {
      "epoch": 2.1331787109375e-05,
      "step": 22368,
      "training_step_time": 6.772500991821289
    },
    {
      "epoch": 2.1332740783691406e-05,
      "model_forward_time": 2.8840765953063965,
      "step": 22369
    },
    {
      "epoch": 2.1332740783691406e-05,
      "step": 22369,
      "training_step_time": 7.644831418991089
    },
    {
      "epoch": 2.1333694458007813e-05,
      "grad_norm": 0.13987639546394348,
      "learning_rate": 9.197640222289234e-06,
      "loss": 0.0611,
      "step": 22370
    },
    {
      "epoch": 2.1333694458007813e-05,
      "model_forward_time": 1.74379563331604,
      "step": 22370
    },
    {
      "dynamics_loss_avg": 0.0808318043127656,
      "epoch": 2.1333694458007813e-05,
      "step": 22370
    },
    {
      "action_loss_avg": 0.00351469109300524,
      "epoch": 2.1333694458007813e-05,
      "step": 22370
    },
    {
      "epoch": 2.1333694458007813e-05,
      "step": 22370,
      "training_step_time": 7.872033596038818
    },
    {
      "epoch": 2.1334648132324217e-05,
      "model_forward_time": 1.7550325393676758,
      "step": 22371
    },
    {
      "epoch": 2.1334648132324217e-05,
      "step": 22371,
      "training_step_time": 7.058784484863281
    },
    {
      "epoch": 2.1335601806640625e-05,
      "model_forward_time": 1.7520415782928467,
      "step": 22372
    },
    {
      "epoch": 2.1335601806640625e-05,
      "step": 22372,
      "training_step_time": 6.728243112564087
    },
    {
      "epoch": 2.1336555480957032e-05,
      "model_forward_time": 1.7506320476531982,
      "step": 22373
    },
    {
      "epoch": 2.1336555480957032e-05,
      "step": 22373,
      "training_step_time": 6.7614829540252686
    },
    {
      "epoch": 2.1337509155273436e-05,
      "model_forward_time": 1.753408432006836,
      "step": 22374
    },
    {
      "epoch": 2.1337509155273436e-05,
      "step": 22374,
      "training_step_time": 6.7679526805877686
    },
    {
      "epoch": 2.1338462829589844e-05,
      "model_forward_time": 1.3933775424957275,
      "step": 22375
    },
    {
      "epoch": 2.1338462829589844e-05,
      "step": 22375,
      "training_step_time": 6.736882209777832
    },
    {
      "epoch": 2.133941650390625e-05,
      "model_forward_time": 1.7569804191589355,
      "step": 22376
    },
    {
      "epoch": 2.133941650390625e-05,
      "step": 22376,
      "training_step_time": 6.796928882598877
    },
    {
      "epoch": 2.1340370178222655e-05,
      "model_forward_time": 1.7551162242889404,
      "step": 22377
    },
    {
      "epoch": 2.1340370178222655e-05,
      "step": 22377,
      "training_step_time": 6.79833984375
    },
    {
      "epoch": 2.1341323852539063e-05,
      "model_forward_time": 1.8071250915527344,
      "step": 22378
    },
    {
      "epoch": 2.1341323852539063e-05,
      "step": 22378,
      "training_step_time": 8.747899293899536
    },
    {
      "epoch": 2.134227752685547e-05,
      "model_forward_time": 2.8482038974761963,
      "step": 22379
    },
    {
      "epoch": 2.134227752685547e-05,
      "step": 22379,
      "training_step_time": 7.701249361038208
    },
    {
      "epoch": 2.1343231201171874e-05,
      "grad_norm": 0.1928032636642456,
      "learning_rate": 9.196741635585895e-06,
      "loss": 0.0591,
      "step": 22380
    },
    {
      "epoch": 2.1343231201171874e-05,
      "model_forward_time": 1.7441747188568115,
      "step": 22380
    },
    {
      "dynamics_loss_avg": 0.06700303163379431,
      "epoch": 2.1343231201171874e-05,
      "step": 22380
    },
    {
      "action_loss_avg": 0.00283008418045938,
      "epoch": 2.1343231201171874e-05,
      "step": 22380
    },
    {
      "epoch": 2.1343231201171874e-05,
      "step": 22380,
      "training_step_time": 7.390114784240723
    },
    {
      "epoch": 2.134418487548828e-05,
      "model_forward_time": 1.758678913116455,
      "step": 22381
    },
    {
      "epoch": 2.134418487548828e-05,
      "step": 22381,
      "training_step_time": 6.847318649291992
    },
    {
      "epoch": 2.134513854980469e-05,
      "model_forward_time": 1.3907628059387207,
      "step": 22382
    },
    {
      "epoch": 2.134513854980469e-05,
      "step": 22382,
      "training_step_time": 6.829794406890869
    },
    {
      "epoch": 2.1346092224121093e-05,
      "model_forward_time": 1.7291100025177002,
      "step": 22383
    },
    {
      "epoch": 2.1346092224121093e-05,
      "step": 22383,
      "training_step_time": 6.856304407119751
    },
    {
      "epoch": 2.13470458984375e-05,
      "model_forward_time": 1.7440762519836426,
      "step": 22384
    },
    {
      "epoch": 2.13470458984375e-05,
      "step": 22384,
      "training_step_time": 6.826565265655518
    },
    {
      "epoch": 2.1347999572753908e-05,
      "model_forward_time": 1.0559930801391602,
      "step": 22385
    },
    {
      "epoch": 2.1347999572753908e-05,
      "step": 22385,
      "training_step_time": 6.847188472747803
    },
    {
      "epoch": 2.1348953247070312e-05,
      "model_forward_time": 1.7353417873382568,
      "step": 22386
    },
    {
      "epoch": 2.1348953247070312e-05,
      "step": 22386,
      "training_step_time": 6.824663877487183
    },
    {
      "epoch": 2.134990692138672e-05,
      "model_forward_time": 1.8079309463500977,
      "step": 22387
    },
    {
      "epoch": 2.134990692138672e-05,
      "step": 22387,
      "training_step_time": 7.665741920471191
    },
    {
      "epoch": 2.1350860595703127e-05,
      "model_forward_time": 2.8561594486236572,
      "step": 22388
    },
    {
      "epoch": 2.1350860595703127e-05,
      "step": 22388,
      "training_step_time": 9.02753734588623
    },
    {
      "epoch": 2.135181427001953e-05,
      "model_forward_time": 1.758845329284668,
      "step": 22389
    },
    {
      "epoch": 2.135181427001953e-05,
      "step": 22389,
      "training_step_time": 7.618454217910767
    },
    {
      "epoch": 2.1352767944335938e-05,
      "grad_norm": 0.13360004127025604,
      "learning_rate": 9.195842589933199e-06,
      "loss": 0.0584,
      "step": 22390
    },
    {
      "epoch": 2.1352767944335938e-05,
      "model_forward_time": 1.7530477046966553,
      "step": 22390
    },
    {
      "dynamics_loss_avg": 0.059792684856802225,
      "epoch": 2.1352767944335938e-05,
      "step": 22390
    },
    {
      "action_loss_avg": 0.003626872310996987,
      "epoch": 2.1352767944335938e-05,
      "step": 22390
    },
    {
      "epoch": 2.1352767944335938e-05,
      "step": 22390,
      "training_step_time": 6.81743049621582
    },
    {
      "epoch": 2.1353721618652342e-05,
      "model_forward_time": 1.7654104232788086,
      "step": 22391
    },
    {
      "epoch": 2.1353721618652342e-05,
      "step": 22391,
      "training_step_time": 6.785989999771118
    },
    {
      "epoch": 2.135467529296875e-05,
      "model_forward_time": 1.7596435546875,
      "step": 22392
    },
    {
      "epoch": 2.135467529296875e-05,
      "step": 22392,
      "training_step_time": 6.736263275146484
    },
    {
      "epoch": 2.1355628967285157e-05,
      "model_forward_time": 1.76412034034729,
      "step": 22393
    },
    {
      "epoch": 2.1355628967285157e-05,
      "step": 22393,
      "training_step_time": 6.81058406829834
    },
    {
      "epoch": 2.135658264160156e-05,
      "model_forward_time": 1.7697882652282715,
      "step": 22394
    },
    {
      "epoch": 2.135658264160156e-05,
      "step": 22394,
      "training_step_time": 6.787619352340698
    },
    {
      "epoch": 2.135753631591797e-05,
      "model_forward_time": 1.7499074935913086,
      "step": 22395
    },
    {
      "epoch": 2.135753631591797e-05,
      "step": 22395,
      "training_step_time": 6.764750719070435
    },
    {
      "epoch": 2.1358489990234376e-05,
      "model_forward_time": 1.8204405307769775,
      "step": 22396
    },
    {
      "epoch": 2.1358489990234376e-05,
      "step": 22396,
      "training_step_time": 8.205765724182129
    },
    {
      "epoch": 2.135944366455078e-05,
      "model_forward_time": 2.836064100265503,
      "step": 22397
    },
    {
      "epoch": 2.135944366455078e-05,
      "step": 22397,
      "training_step_time": 9.669965267181396
    },
    {
      "epoch": 2.1360397338867187e-05,
      "model_forward_time": 1.7661118507385254,
      "step": 22398
    },
    {
      "epoch": 2.1360397338867187e-05,
      "step": 22398,
      "training_step_time": 7.566809892654419
    },
    {
      "epoch": 2.1361351013183595e-05,
      "model_forward_time": 1.7403230667114258,
      "step": 22399
    },
    {
      "epoch": 2.1361351013183595e-05,
      "step": 22399,
      "training_step_time": 6.99607515335083
    },
    {
      "epoch": 2.13623046875e-05,
      "grad_norm": 0.22250793874263763,
      "learning_rate": 9.194943085429466e-06,
      "loss": 0.0626,
      "step": 22400
    },
    {
      "epoch": 2.13623046875e-05,
      "model_forward_time": 1.7615749835968018,
      "step": 22400
    },
    {
      "dynamics_loss_avg": 0.05228454694151878,
      "epoch": 2.13623046875e-05,
      "step": 22400
    },
    {
      "action_loss_avg": 0.00320209683268331,
      "epoch": 2.13623046875e-05,
      "step": 22400
    },
    {
      "epoch": 2.13623046875e-05,
      "step": 22400,
      "training_step_time": 6.737473964691162
    },
    {
      "epoch": 2.1363258361816406e-05,
      "model_forward_time": 1.7649261951446533,
      "step": 22401
    },
    {
      "epoch": 2.1363258361816406e-05,
      "step": 22401,
      "training_step_time": 6.748480319976807
    },
    {
      "epoch": 2.1364212036132813e-05,
      "model_forward_time": 1.747678279876709,
      "step": 22402
    },
    {
      "epoch": 2.1364212036132813e-05,
      "step": 22402,
      "training_step_time": 6.765664100646973
    },
    {
      "epoch": 2.1365165710449218e-05,
      "model_forward_time": 1.7589004039764404,
      "step": 22403
    },
    {
      "epoch": 2.1365165710449218e-05,
      "step": 22403,
      "training_step_time": 6.771740436553955
    },
    {
      "epoch": 2.1366119384765625e-05,
      "model_forward_time": 1.7659904956817627,
      "step": 22404
    },
    {
      "epoch": 2.1366119384765625e-05,
      "step": 22404,
      "training_step_time": 6.782799959182739
    },
    {
      "epoch": 2.1367073059082032e-05,
      "model_forward_time": 1.7914535999298096,
      "step": 22405
    },
    {
      "epoch": 2.1367073059082032e-05,
      "step": 22405,
      "training_step_time": 7.375768423080444
    },
    {
      "epoch": 2.1368026733398436e-05,
      "model_forward_time": 1.757404088973999,
      "step": 22406
    },
    {
      "epoch": 2.1368026733398436e-05,
      "step": 22406,
      "training_step_time": 8.351906776428223
    },
    {
      "epoch": 2.1368980407714844e-05,
      "model_forward_time": 2.92590069770813,
      "step": 22407
    },
    {
      "epoch": 2.1368980407714844e-05,
      "step": 22407,
      "training_step_time": 8.195020198822021
    },
    {
      "epoch": 2.136993408203125e-05,
      "model_forward_time": 1.7664616107940674,
      "step": 22408
    },
    {
      "epoch": 2.136993408203125e-05,
      "step": 22408,
      "training_step_time": 7.059847354888916
    },
    {
      "epoch": 2.1370887756347655e-05,
      "model_forward_time": 1.7648429870605469,
      "step": 22409
    },
    {
      "epoch": 2.1370887756347655e-05,
      "step": 22409,
      "training_step_time": 6.80385947227478
    },
    {
      "epoch": 2.1371841430664063e-05,
      "grad_norm": 0.16956016421318054,
      "learning_rate": 9.194043122173066e-06,
      "loss": 0.0628,
      "step": 22410
    },
    {
      "epoch": 2.1371841430664063e-05,
      "model_forward_time": 1.772899866104126,
      "step": 22410
    },
    {
      "dynamics_loss_avg": 0.06735136974602937,
      "epoch": 2.1371841430664063e-05,
      "step": 22410
    },
    {
      "action_loss_avg": 0.0028869159577880055,
      "epoch": 2.1371841430664063e-05,
      "step": 22410
    },
    {
      "epoch": 2.1371841430664063e-05,
      "step": 22410,
      "training_step_time": 6.754350423812866
    },
    {
      "epoch": 2.137279510498047e-05,
      "model_forward_time": 1.7554385662078857,
      "step": 22411
    },
    {
      "epoch": 2.137279510498047e-05,
      "step": 22411,
      "training_step_time": 6.782952308654785
    },
    {
      "epoch": 2.1373748779296874e-05,
      "model_forward_time": 1.7603778839111328,
      "step": 22412
    },
    {
      "epoch": 2.1373748779296874e-05,
      "step": 22412,
      "training_step_time": 6.754694700241089
    },
    {
      "epoch": 2.137470245361328e-05,
      "model_forward_time": 1.768089771270752,
      "step": 22413
    },
    {
      "epoch": 2.137470245361328e-05,
      "step": 22413,
      "training_step_time": 6.771665334701538
    },
    {
      "epoch": 2.137565612792969e-05,
      "model_forward_time": 1.7756929397583008,
      "step": 22414
    },
    {
      "epoch": 2.137565612792969e-05,
      "step": 22414,
      "training_step_time": 7.633816957473755
    },
    {
      "epoch": 2.1376609802246093e-05,
      "model_forward_time": 1.7580139636993408,
      "step": 22415
    },
    {
      "epoch": 2.1376609802246093e-05,
      "step": 22415,
      "training_step_time": 11.495771169662476
    },
    {
      "epoch": 2.13775634765625e-05,
      "model_forward_time": 2.8754310607910156,
      "step": 22416
    },
    {
      "epoch": 2.13775634765625e-05,
      "step": 22416,
      "training_step_time": 7.672041893005371
    },
    {
      "epoch": 2.1378517150878908e-05,
      "model_forward_time": 1.7682545185089111,
      "step": 22417
    },
    {
      "epoch": 2.1378517150878908e-05,
      "step": 22417,
      "training_step_time": 7.390024423599243
    },
    {
      "epoch": 2.1379470825195312e-05,
      "model_forward_time": 1.7529575824737549,
      "step": 22418
    },
    {
      "epoch": 2.1379470825195312e-05,
      "step": 22418,
      "training_step_time": 6.757790803909302
    },
    {
      "epoch": 2.138042449951172e-05,
      "model_forward_time": 1.765573501586914,
      "step": 22419
    },
    {
      "epoch": 2.138042449951172e-05,
      "step": 22419,
      "training_step_time": 6.777655124664307
    },
    {
      "epoch": 2.1381378173828127e-05,
      "grad_norm": 0.16948923468589783,
      "learning_rate": 9.193142700262414e-06,
      "loss": 0.0591,
      "step": 22420
    },
    {
      "epoch": 2.1381378173828127e-05,
      "model_forward_time": 1.7721168994903564,
      "step": 22420
    },
    {
      "dynamics_loss_avg": 0.053171693533658984,
      "epoch": 2.1381378173828127e-05,
      "step": 22420
    },
    {
      "action_loss_avg": 0.00170735577121377,
      "epoch": 2.1381378173828127e-05,
      "step": 22420
    },
    {
      "epoch": 2.1381378173828127e-05,
      "step": 22420,
      "training_step_time": 6.783766269683838
    },
    {
      "epoch": 2.138233184814453e-05,
      "model_forward_time": 1.7619853019714355,
      "step": 22421
    },
    {
      "epoch": 2.138233184814453e-05,
      "step": 22421,
      "training_step_time": 6.783396005630493
    },
    {
      "epoch": 2.1383285522460938e-05,
      "model_forward_time": 1.7686882019042969,
      "step": 22422
    },
    {
      "epoch": 2.1383285522460938e-05,
      "step": 22422,
      "training_step_time": 6.776524782180786
    },
    {
      "epoch": 2.1384239196777342e-05,
      "model_forward_time": 1.7717459201812744,
      "step": 22423
    },
    {
      "epoch": 2.1384239196777342e-05,
      "step": 22423,
      "training_step_time": 6.769155263900757
    },
    {
      "epoch": 2.138519287109375e-05,
      "model_forward_time": 1.8597283363342285,
      "step": 22424
    },
    {
      "epoch": 2.138519287109375e-05,
      "step": 22424,
      "training_step_time": 8.792259931564331
    },
    {
      "epoch": 2.1386146545410157e-05,
      "model_forward_time": 2.8805346488952637,
      "step": 22425
    },
    {
      "epoch": 2.1386146545410157e-05,
      "step": 22425,
      "training_step_time": 9.858571767807007
    },
    {
      "epoch": 2.138710021972656e-05,
      "model_forward_time": 1.7615742683410645,
      "step": 22426
    },
    {
      "epoch": 2.138710021972656e-05,
      "step": 22426,
      "training_step_time": 7.375757694244385
    },
    {
      "epoch": 2.138805389404297e-05,
      "model_forward_time": 1.7573301792144775,
      "step": 22427
    },
    {
      "epoch": 2.138805389404297e-05,
      "step": 22427,
      "training_step_time": 7.069587469100952
    },
    {
      "epoch": 2.1389007568359376e-05,
      "model_forward_time": 1.7621841430664062,
      "step": 22428
    },
    {
      "epoch": 2.1389007568359376e-05,
      "step": 22428,
      "training_step_time": 6.784198999404907
    },
    {
      "epoch": 2.138996124267578e-05,
      "model_forward_time": 1.7634568214416504,
      "step": 22429
    },
    {
      "epoch": 2.138996124267578e-05,
      "step": 22429,
      "training_step_time": 6.779823541641235
    },
    {
      "epoch": 2.1390914916992187e-05,
      "grad_norm": 0.18642844259738922,
      "learning_rate": 9.19224181979598e-06,
      "loss": 0.0586,
      "step": 22430
    },
    {
      "epoch": 2.1390914916992187e-05,
      "model_forward_time": 1.7577311992645264,
      "step": 22430
    },
    {
      "dynamics_loss_avg": 0.06760255917906761,
      "epoch": 2.1390914916992187e-05,
      "step": 22430
    },
    {
      "action_loss_avg": 0.004423518071416765,
      "epoch": 2.1390914916992187e-05,
      "step": 22430
    },
    {
      "epoch": 2.1390914916992187e-05,
      "step": 22430,
      "training_step_time": 6.744711875915527
    },
    {
      "epoch": 2.1391868591308595e-05,
      "model_forward_time": 1.7613952159881592,
      "step": 22431
    },
    {
      "epoch": 2.1391868591308595e-05,
      "step": 22431,
      "training_step_time": 6.7650675773620605
    },
    {
      "epoch": 2.1392822265625e-05,
      "model_forward_time": 1.7610995769500732,
      "step": 22432
    },
    {
      "epoch": 2.1392822265625e-05,
      "step": 22432,
      "training_step_time": 6.780510425567627
    },
    {
      "epoch": 2.1393775939941406e-05,
      "model_forward_time": 1.8633615970611572,
      "step": 22433
    },
    {
      "epoch": 2.1393775939941406e-05,
      "step": 22433,
      "training_step_time": 8.613190650939941
    },
    {
      "epoch": 2.1394729614257814e-05,
      "model_forward_time": 2.914975166320801,
      "step": 22434
    },
    {
      "epoch": 2.1394729614257814e-05,
      "step": 22434,
      "training_step_time": 9.553264617919922
    },
    {
      "epoch": 2.1395683288574218e-05,
      "model_forward_time": 1.4040679931640625,
      "step": 22435
    },
    {
      "epoch": 2.1395683288574218e-05,
      "step": 22435,
      "training_step_time": 7.586731433868408
    },
    {
      "epoch": 2.1396636962890625e-05,
      "model_forward_time": 1.7653019428253174,
      "step": 22436
    },
    {
      "epoch": 2.1396636962890625e-05,
      "step": 22436,
      "training_step_time": 7.044791221618652
    },
    {
      "epoch": 2.1397590637207032e-05,
      "model_forward_time": 1.7573308944702148,
      "step": 22437
    },
    {
      "epoch": 2.1397590637207032e-05,
      "step": 22437,
      "training_step_time": 6.748585939407349
    },
    {
      "epoch": 2.1398544311523436e-05,
      "model_forward_time": 1.7646684646606445,
      "step": 22438
    },
    {
      "epoch": 2.1398544311523436e-05,
      "step": 22438,
      "training_step_time": 6.771836757659912
    },
    {
      "epoch": 2.1399497985839844e-05,
      "model_forward_time": 1.4033949375152588,
      "step": 22439
    },
    {
      "epoch": 2.1399497985839844e-05,
      "step": 22439,
      "training_step_time": 6.745514154434204
    },
    {
      "epoch": 2.140045166015625e-05,
      "grad_norm": 0.12568995356559753,
      "learning_rate": 9.191340480872284e-06,
      "loss": 0.0619,
      "step": 22440
    },
    {
      "epoch": 2.140045166015625e-05,
      "model_forward_time": 1.74448561668396,
      "step": 22440
    },
    {
      "dynamics_loss_avg": 0.06362467668950558,
      "epoch": 2.140045166015625e-05,
      "step": 22440
    },
    {
      "action_loss_avg": 0.0024176515929866583,
      "epoch": 2.140045166015625e-05,
      "step": 22440
    },
    {
      "epoch": 2.140045166015625e-05,
      "step": 22440,
      "training_step_time": 6.659396648406982
    },
    {
      "epoch": 2.1401405334472655e-05,
      "model_forward_time": 1.7550787925720215,
      "step": 22441
    },
    {
      "epoch": 2.1401405334472655e-05,
      "step": 22441,
      "training_step_time": 6.741616249084473
    },
    {
      "epoch": 2.1402359008789063e-05,
      "model_forward_time": 1.839322566986084,
      "step": 22442
    },
    {
      "epoch": 2.1402359008789063e-05,
      "step": 22442,
      "training_step_time": 7.614782333374023
    },
    {
      "epoch": 2.140331268310547e-05,
      "model_forward_time": 1.3984084129333496,
      "step": 22443
    },
    {
      "epoch": 2.140331268310547e-05,
      "step": 22443,
      "training_step_time": 12.258703470230103
    },
    {
      "epoch": 2.1404266357421874e-05,
      "model_forward_time": 2.8955976963043213,
      "step": 22444
    },
    {
      "epoch": 2.1404266357421874e-05,
      "step": 22444,
      "training_step_time": 7.674408674240112
    },
    {
      "epoch": 2.140522003173828e-05,
      "model_forward_time": 1.755615234375,
      "step": 22445
    },
    {
      "epoch": 2.140522003173828e-05,
      "step": 22445,
      "training_step_time": 7.085978984832764
    },
    {
      "epoch": 2.140617370605469e-05,
      "model_forward_time": 1.7458643913269043,
      "step": 22446
    },
    {
      "epoch": 2.140617370605469e-05,
      "step": 22446,
      "training_step_time": 6.803043365478516
    },
    {
      "epoch": 2.1407127380371093e-05,
      "model_forward_time": 1.7614269256591797,
      "step": 22447
    },
    {
      "epoch": 2.1407127380371093e-05,
      "step": 22447,
      "training_step_time": 6.794293165206909
    },
    {
      "epoch": 2.14080810546875e-05,
      "model_forward_time": 1.401085615158081,
      "step": 22448
    },
    {
      "epoch": 2.14080810546875e-05,
      "step": 22448,
      "training_step_time": 6.800134181976318
    },
    {
      "epoch": 2.1409034729003908e-05,
      "model_forward_time": 1.758385181427002,
      "step": 22449
    },
    {
      "epoch": 2.1409034729003908e-05,
      "step": 22449,
      "training_step_time": 6.815019607543945
    },
    {
      "epoch": 2.1409988403320312e-05,
      "grad_norm": 0.17267943918704987,
      "learning_rate": 9.190438683589895e-06,
      "loss": 0.0635,
      "step": 22450
    },
    {
      "epoch": 2.1409988403320312e-05,
      "model_forward_time": 1.759061574935913,
      "step": 22450
    },
    {
      "dynamics_loss_avg": 0.052522504981607196,
      "epoch": 2.1409988403320312e-05,
      "step": 22450
    },
    {
      "action_loss_avg": 0.001807784559787251,
      "epoch": 2.1409988403320312e-05,
      "step": 22450
    },
    {
      "epoch": 2.1409988403320312e-05,
      "step": 22450,
      "training_step_time": 6.818973779678345
    },
    {
      "epoch": 2.141094207763672e-05,
      "model_forward_time": 1.8262617588043213,
      "step": 22451
    },
    {
      "epoch": 2.141094207763672e-05,
      "step": 22451,
      "training_step_time": 7.603065013885498
    },
    {
      "epoch": 2.1411895751953127e-05,
      "model_forward_time": 1.7517681121826172,
      "step": 22452
    },
    {
      "epoch": 2.1411895751953127e-05,
      "step": 22452,
      "training_step_time": 10.779744148254395
    },
    {
      "epoch": 2.141284942626953e-05,
      "model_forward_time": 2.952759027481079,
      "step": 22453
    },
    {
      "epoch": 2.141284942626953e-05,
      "step": 22453,
      "training_step_time": 8.960243701934814
    },
    {
      "epoch": 2.1413803100585938e-05,
      "model_forward_time": 1.7623319625854492,
      "step": 22454
    },
    {
      "epoch": 2.1413803100585938e-05,
      "step": 22454,
      "training_step_time": 7.083571195602417
    },
    {
      "epoch": 2.1414756774902342e-05,
      "model_forward_time": 1.7634143829345703,
      "step": 22455
    },
    {
      "epoch": 2.1414756774902342e-05,
      "step": 22455,
      "training_step_time": 6.7603394985198975
    },
    {
      "epoch": 2.141571044921875e-05,
      "model_forward_time": 1.7654235363006592,
      "step": 22456
    },
    {
      "epoch": 2.141571044921875e-05,
      "step": 22456,
      "training_step_time": 6.74757981300354
    },
    {
      "epoch": 2.1416664123535157e-05,
      "model_forward_time": 1.7675747871398926,
      "step": 22457
    },
    {
      "epoch": 2.1416664123535157e-05,
      "step": 22457,
      "training_step_time": 6.748921155929565
    },
    {
      "epoch": 2.141761779785156e-05,
      "model_forward_time": 1.7580718994140625,
      "step": 22458
    },
    {
      "epoch": 2.141761779785156e-05,
      "step": 22458,
      "training_step_time": 6.7645368576049805
    },
    {
      "epoch": 2.141857147216797e-05,
      "model_forward_time": 1.7632520198822021,
      "step": 22459
    },
    {
      "epoch": 2.141857147216797e-05,
      "step": 22459,
      "training_step_time": 6.780572175979614
    },
    {
      "epoch": 2.1419525146484376e-05,
      "grad_norm": 0.12391158193349838,
      "learning_rate": 9.189536428047432e-06,
      "loss": 0.0619,
      "step": 22460
    },
    {
      "epoch": 2.1419525146484376e-05,
      "model_forward_time": 1.7776505947113037,
      "step": 22460
    },
    {
      "dynamics_loss_avg": 0.04387344159185887,
      "epoch": 2.1419525146484376e-05,
      "step": 22460
    },
    {
      "action_loss_avg": 0.0026301400270313025,
      "epoch": 2.1419525146484376e-05,
      "step": 22460
    },
    {
      "epoch": 2.1419525146484376e-05,
      "step": 22460,
      "training_step_time": 7.553976058959961
    },
    {
      "epoch": 2.142047882080078e-05,
      "model_forward_time": 1.752967357635498,
      "step": 22461
    },
    {
      "epoch": 2.142047882080078e-05,
      "step": 22461,
      "training_step_time": 9.668887615203857
    },
    {
      "epoch": 2.1421432495117187e-05,
      "model_forward_time": 2.9004292488098145,
      "step": 22462
    },
    {
      "epoch": 2.1421432495117187e-05,
      "step": 22462,
      "training_step_time": 7.665343284606934
    },
    {
      "epoch": 2.1422386169433595e-05,
      "model_forward_time": 1.7604889869689941,
      "step": 22463
    },
    {
      "epoch": 2.1422386169433595e-05,
      "step": 22463,
      "training_step_time": 7.558515548706055
    },
    {
      "epoch": 2.142333984375e-05,
      "model_forward_time": 1.754260540008545,
      "step": 22464
    },
    {
      "epoch": 2.142333984375e-05,
      "step": 22464,
      "training_step_time": 6.771814346313477
    },
    {
      "epoch": 2.1424293518066406e-05,
      "model_forward_time": 1.7570350170135498,
      "step": 22465
    },
    {
      "epoch": 2.1424293518066406e-05,
      "step": 22465,
      "training_step_time": 6.776322841644287
    },
    {
      "epoch": 2.1425247192382814e-05,
      "model_forward_time": 1.7686164379119873,
      "step": 22466
    },
    {
      "epoch": 2.1425247192382814e-05,
      "step": 22466,
      "training_step_time": 6.753827095031738
    },
    {
      "epoch": 2.1426200866699218e-05,
      "model_forward_time": 1.7396705150604248,
      "step": 22467
    },
    {
      "epoch": 2.1426200866699218e-05,
      "step": 22467,
      "training_step_time": 6.813018321990967
    },
    {
      "epoch": 2.1427154541015625e-05,
      "model_forward_time": 1.746396780014038,
      "step": 22468
    },
    {
      "epoch": 2.1427154541015625e-05,
      "step": 22468,
      "training_step_time": 6.81467604637146
    },
    {
      "epoch": 2.1428108215332032e-05,
      "model_forward_time": 1.7583022117614746,
      "step": 22469
    },
    {
      "epoch": 2.1428108215332032e-05,
      "step": 22469,
      "training_step_time": 7.636902093887329
    },
    {
      "epoch": 2.1429061889648436e-05,
      "grad_norm": 0.14552439749240875,
      "learning_rate": 9.188633714343565e-06,
      "loss": 0.064,
      "step": 22470
    },
    {
      "epoch": 2.1429061889648436e-05,
      "model_forward_time": 1.7633280754089355,
      "step": 22470
    },
    {
      "dynamics_loss_avg": 0.06838760375976563,
      "epoch": 2.1429061889648436e-05,
      "step": 22470
    },
    {
      "action_loss_avg": 0.003142284951172769,
      "epoch": 2.1429061889648436e-05,
      "step": 22470
    },
    {
      "epoch": 2.1429061889648436e-05,
      "step": 22470,
      "training_step_time": 8.387801885604858
    },
    {
      "epoch": 2.1430015563964844e-05,
      "model_forward_time": 2.89613676071167,
      "step": 22471
    },
    {
      "epoch": 2.1430015563964844e-05,
      "step": 22471,
      "training_step_time": 9.522635221481323
    },
    {
      "epoch": 2.143096923828125e-05,
      "model_forward_time": 1.755796194076538,
      "step": 22472
    },
    {
      "epoch": 2.143096923828125e-05,
      "step": 22472,
      "training_step_time": 7.694256544113159
    },
    {
      "epoch": 2.1431922912597655e-05,
      "model_forward_time": 1.747574806213379,
      "step": 22473
    },
    {
      "epoch": 2.1431922912597655e-05,
      "step": 22473,
      "training_step_time": 7.137299060821533
    },
    {
      "epoch": 2.1432876586914063e-05,
      "model_forward_time": 1.748811960220337,
      "step": 22474
    },
    {
      "epoch": 2.1432876586914063e-05,
      "step": 22474,
      "training_step_time": 6.775575637817383
    },
    {
      "epoch": 2.143383026123047e-05,
      "model_forward_time": 1.7572157382965088,
      "step": 22475
    },
    {
      "epoch": 2.143383026123047e-05,
      "step": 22475,
      "training_step_time": 6.774897813796997
    },
    {
      "epoch": 2.1434783935546874e-05,
      "model_forward_time": 1.7475292682647705,
      "step": 22476
    },
    {
      "epoch": 2.1434783935546874e-05,
      "step": 22476,
      "training_step_time": 6.77470064163208
    },
    {
      "epoch": 2.143573760986328e-05,
      "model_forward_time": 1.740760087966919,
      "step": 22477
    },
    {
      "epoch": 2.143573760986328e-05,
      "step": 22477,
      "training_step_time": 6.848909139633179
    },
    {
      "epoch": 2.143669128417969e-05,
      "model_forward_time": 1.7533767223358154,
      "step": 22478
    },
    {
      "epoch": 2.143669128417969e-05,
      "step": 22478,
      "training_step_time": 7.631917715072632
    },
    {
      "epoch": 2.1437644958496093e-05,
      "model_forward_time": 1.497089147567749,
      "step": 22479
    },
    {
      "epoch": 2.1437644958496093e-05,
      "step": 22479,
      "training_step_time": 8.8195481300354
    },
    {
      "epoch": 2.14385986328125e-05,
      "grad_norm": 0.1618189811706543,
      "learning_rate": 9.187730542577011e-06,
      "loss": 0.0663,
      "step": 22480
    },
    {
      "epoch": 2.14385986328125e-05,
      "model_forward_time": 2.8215906620025635,
      "step": 22480
    },
    {
      "dynamics_loss_avg": 0.056673999316990374,
      "epoch": 2.14385986328125e-05,
      "step": 22480
    },
    {
      "action_loss_avg": 0.002375936898170039,
      "epoch": 2.14385986328125e-05,
      "step": 22480
    },
    {
      "epoch": 2.14385986328125e-05,
      "step": 22480,
      "training_step_time": 9.172098875045776
    },
    {
      "epoch": 2.1439552307128908e-05,
      "model_forward_time": 1.7549026012420654,
      "step": 22481
    },
    {
      "epoch": 2.1439552307128908e-05,
      "step": 22481,
      "training_step_time": 7.6825268268585205
    },
    {
      "epoch": 2.1440505981445312e-05,
      "model_forward_time": 1.751941442489624,
      "step": 22482
    },
    {
      "epoch": 2.1440505981445312e-05,
      "step": 22482,
      "training_step_time": 7.101879596710205
    },
    {
      "epoch": 2.144145965576172e-05,
      "model_forward_time": 1.7534856796264648,
      "step": 22483
    },
    {
      "epoch": 2.144145965576172e-05,
      "step": 22483,
      "training_step_time": 6.754570007324219
    },
    {
      "epoch": 2.1442413330078127e-05,
      "model_forward_time": 1.7492461204528809,
      "step": 22484
    },
    {
      "epoch": 2.1442413330078127e-05,
      "step": 22484,
      "training_step_time": 6.747678995132446
    },
    {
      "epoch": 2.144336700439453e-05,
      "model_forward_time": 1.750960350036621,
      "step": 22485
    },
    {
      "epoch": 2.144336700439453e-05,
      "step": 22485,
      "training_step_time": 6.764807462692261
    },
    {
      "epoch": 2.1444320678710938e-05,
      "model_forward_time": 1.7600164413452148,
      "step": 22486
    },
    {
      "epoch": 2.1444320678710938e-05,
      "step": 22486,
      "training_step_time": 6.730092525482178
    },
    {
      "epoch": 2.1445274353027342e-05,
      "model_forward_time": 1.749129295349121,
      "step": 22487
    },
    {
      "epoch": 2.1445274353027342e-05,
      "step": 22487,
      "training_step_time": 7.518536329269409
    },
    {
      "epoch": 2.144622802734375e-05,
      "model_forward_time": 1.8351376056671143,
      "step": 22488
    },
    {
      "epoch": 2.144622802734375e-05,
      "step": 22488,
      "training_step_time": 7.658252239227295
    },
    {
      "epoch": 2.1447181701660157e-05,
      "model_forward_time": 1.7427899837493896,
      "step": 22489
    },
    {
      "epoch": 2.1447181701660157e-05,
      "step": 22489,
      "training_step_time": 7.719372987747192
    },
    {
      "epoch": 2.144813537597656e-05,
      "grad_norm": 0.12658129632472992,
      "learning_rate": 9.186826912846541e-06,
      "loss": 0.0663,
      "step": 22490
    },
    {
      "epoch": 2.144813537597656e-05,
      "model_forward_time": 2.889441967010498,
      "step": 22490
    },
    {
      "dynamics_loss_avg": 0.053995889611542225,
      "epoch": 2.144813537597656e-05,
      "step": 22490
    },
    {
      "action_loss_avg": 0.0027728807297535242,
      "epoch": 2.144813537597656e-05,
      "step": 22490
    },
    {
      "epoch": 2.144813537597656e-05,
      "step": 22490,
      "training_step_time": 7.626349925994873
    },
    {
      "epoch": 2.144908905029297e-05,
      "model_forward_time": 1.7503349781036377,
      "step": 22491
    },
    {
      "epoch": 2.144908905029297e-05,
      "step": 22491,
      "training_step_time": 7.035499095916748
    },
    {
      "epoch": 2.1450042724609376e-05,
      "model_forward_time": 1.746877908706665,
      "step": 22492
    },
    {
      "epoch": 2.1450042724609376e-05,
      "step": 22492,
      "training_step_time": 6.773837566375732
    },
    {
      "epoch": 2.145099639892578e-05,
      "model_forward_time": 1.7487013339996338,
      "step": 22493
    },
    {
      "epoch": 2.145099639892578e-05,
      "step": 22493,
      "training_step_time": 6.657495498657227
    },
    {
      "epoch": 2.1451950073242187e-05,
      "model_forward_time": 1.7569167613983154,
      "step": 22494
    },
    {
      "epoch": 2.1451950073242187e-05,
      "step": 22494,
      "training_step_time": 6.742671251296997
    },
    {
      "epoch": 2.1452903747558595e-05,
      "model_forward_time": 1.750889539718628,
      "step": 22495
    },
    {
      "epoch": 2.1452903747558595e-05,
      "step": 22495,
      "training_step_time": 6.741500616073608
    },
    {
      "epoch": 2.1453857421875e-05,
      "model_forward_time": 1.7611207962036133,
      "step": 22496
    },
    {
      "epoch": 2.1453857421875e-05,
      "step": 22496,
      "training_step_time": 7.41351842880249
    },
    {
      "epoch": 2.1454811096191406e-05,
      "model_forward_time": 1.8238115310668945,
      "step": 22497
    },
    {
      "epoch": 2.1454811096191406e-05,
      "step": 22497,
      "training_step_time": 8.165512323379517
    },
    {
      "epoch": 2.1455764770507814e-05,
      "model_forward_time": 1.7479910850524902,
      "step": 22498
    },
    {
      "epoch": 2.1455764770507814e-05,
      "step": 22498,
      "training_step_time": 10.329273700714111
    },
    {
      "epoch": 2.1456718444824218e-05,
      "model_forward_time": 2.8818676471710205,
      "step": 22499
    },
    {
      "epoch": 2.1456718444824218e-05,
      "step": 22499,
      "training_step_time": 8.43463397026062
    },
    {
      "epoch": 2.1457672119140625e-05,
      "grad_norm": 0.17023378610610962,
      "learning_rate": 9.185922825250975e-06,
      "loss": 0.06,
      "step": 22500
    },
    {
      "epoch": 2.1457672119140625e-05,
      "model_forward_time": 1.7630345821380615,
      "step": 22500
    },
    {
      "dynamics_loss_avg": 0.06343123354017735,
      "epoch": 2.1457672119140625e-05,
      "step": 22500
    },
    {
      "action_loss_avg": 0.002278469392331317,
      "epoch": 2.1457672119140625e-05,
      "step": 22500
    },
    {
      "epoch": 2.1457672119140625e-05,
      "step": 22500,
      "training_step_time": 7.612014055252075
    },
    {
      "epoch": 2.1458625793457032e-05,
      "model_forward_time": 1.7585904598236084,
      "step": 22501
    },
    {
      "epoch": 2.1458625793457032e-05,
      "step": 22501,
      "training_step_time": 6.734600067138672
    },
    {
      "epoch": 2.1459579467773436e-05,
      "model_forward_time": 1.7623186111450195,
      "step": 22502
    },
    {
      "epoch": 2.1459579467773436e-05,
      "step": 22502,
      "training_step_time": 6.741543531417847
    },
    {
      "epoch": 2.1460533142089844e-05,
      "model_forward_time": 1.757194995880127,
      "step": 22503
    },
    {
      "epoch": 2.1460533142089844e-05,
      "step": 22503,
      "training_step_time": 6.700850963592529
    },
    {
      "epoch": 2.146148681640625e-05,
      "model_forward_time": 1.7613577842712402,
      "step": 22504
    },
    {
      "epoch": 2.146148681640625e-05,
      "step": 22504,
      "training_step_time": 6.6692705154418945
    },
    {
      "epoch": 2.1462440490722655e-05,
      "model_forward_time": 1.7686073780059814,
      "step": 22505
    },
    {
      "epoch": 2.1462440490722655e-05,
      "step": 22505,
      "training_step_time": 6.680947780609131
    },
    {
      "epoch": 2.1463394165039063e-05,
      "model_forward_time": 1.7873568534851074,
      "step": 22506
    },
    {
      "epoch": 2.1463394165039063e-05,
      "step": 22506,
      "training_step_time": 8.408891677856445
    },
    {
      "epoch": 2.146434783935547e-05,
      "model_forward_time": 1.7546923160552979,
      "step": 22507
    },
    {
      "epoch": 2.146434783935547e-05,
      "step": 22507,
      "training_step_time": 8.601211309432983
    },
    {
      "epoch": 2.1465301513671874e-05,
      "model_forward_time": 2.9272189140319824,
      "step": 22508
    },
    {
      "epoch": 2.1465301513671874e-05,
      "step": 22508,
      "training_step_time": 8.679777383804321
    },
    {
      "epoch": 2.146625518798828e-05,
      "model_forward_time": 1.7604949474334717,
      "step": 22509
    },
    {
      "epoch": 2.146625518798828e-05,
      "step": 22509,
      "training_step_time": 8.272233247756958
    },
    {
      "epoch": 2.146720886230469e-05,
      "grad_norm": 0.2008395791053772,
      "learning_rate": 9.185018279889181e-06,
      "loss": 0.0617,
      "step": 22510
    },
    {
      "epoch": 2.146720886230469e-05,
      "model_forward_time": 1.754932165145874,
      "step": 22510
    },
    {
      "dynamics_loss_avg": 0.05482799969613552,
      "epoch": 2.146720886230469e-05,
      "step": 22510
    },
    {
      "action_loss_avg": 0.0017991023720242082,
      "epoch": 2.146720886230469e-05,
      "step": 22510
    },
    {
      "epoch": 2.146720886230469e-05,
      "step": 22510,
      "training_step_time": 7.587874412536621
    },
    {
      "epoch": 2.1468162536621093e-05,
      "model_forward_time": 1.762023687362671,
      "step": 22511
    },
    {
      "epoch": 2.1468162536621093e-05,
      "step": 22511,
      "training_step_time": 6.667144536972046
    },
    {
      "epoch": 2.14691162109375e-05,
      "model_forward_time": 1.765871524810791,
      "step": 22512
    },
    {
      "epoch": 2.14691162109375e-05,
      "step": 22512,
      "training_step_time": 6.67347526550293
    },
    {
      "epoch": 2.1470069885253908e-05,
      "model_forward_time": 1.762535810470581,
      "step": 22513
    },
    {
      "epoch": 2.1470069885253908e-05,
      "step": 22513,
      "training_step_time": 6.696048021316528
    },
    {
      "epoch": 2.1471023559570312e-05,
      "model_forward_time": 1.760202407836914,
      "step": 22514
    },
    {
      "epoch": 2.1471023559570312e-05,
      "step": 22514,
      "training_step_time": 6.6519482135772705
    },
    {
      "epoch": 2.147197723388672e-05,
      "model_forward_time": 1.768509864807129,
      "step": 22515
    },
    {
      "epoch": 2.147197723388672e-05,
      "step": 22515,
      "training_step_time": 8.184569120407104
    },
    {
      "epoch": 2.1472930908203123e-05,
      "model_forward_time": 1.7555289268493652,
      "step": 22516
    },
    {
      "epoch": 2.1472930908203123e-05,
      "step": 22516,
      "training_step_time": 8.393311738967896
    },
    {
      "epoch": 2.147388458251953e-05,
      "model_forward_time": 2.844968795776367,
      "step": 22517
    },
    {
      "epoch": 2.147388458251953e-05,
      "step": 22517,
      "training_step_time": 9.032351732254028
    },
    {
      "epoch": 2.1474838256835938e-05,
      "model_forward_time": 1.7653677463531494,
      "step": 22518
    },
    {
      "epoch": 2.1474838256835938e-05,
      "step": 22518,
      "training_step_time": 7.585348606109619
    },
    {
      "epoch": 2.1475791931152342e-05,
      "model_forward_time": 1.7637207508087158,
      "step": 22519
    },
    {
      "epoch": 2.1475791931152342e-05,
      "step": 22519,
      "training_step_time": 7.704244375228882
    },
    {
      "epoch": 2.147674560546875e-05,
      "grad_norm": 0.15658199787139893,
      "learning_rate": 9.184113276860083e-06,
      "loss": 0.0617,
      "step": 22520
    },
    {
      "epoch": 2.147674560546875e-05,
      "model_forward_time": 1.7587089538574219,
      "step": 22520
    },
    {
      "dynamics_loss_avg": 0.06722082290798426,
      "epoch": 2.147674560546875e-05,
      "step": 22520
    },
    {
      "action_loss_avg": 0.002992784173693508,
      "epoch": 2.147674560546875e-05,
      "step": 22520
    },
    {
      "epoch": 2.147674560546875e-05,
      "step": 22520,
      "training_step_time": 6.648753881454468
    },
    {
      "epoch": 2.1477699279785157e-05,
      "model_forward_time": 1.754776954650879,
      "step": 22521
    },
    {
      "epoch": 2.1477699279785157e-05,
      "step": 22521,
      "training_step_time": 6.659160852432251
    },
    {
      "epoch": 2.147865295410156e-05,
      "model_forward_time": 1.7674341201782227,
      "step": 22522
    },
    {
      "epoch": 2.147865295410156e-05,
      "step": 22522,
      "training_step_time": 6.658326864242554
    },
    {
      "epoch": 2.147960662841797e-05,
      "model_forward_time": 1.7795982360839844,
      "step": 22523
    },
    {
      "epoch": 2.147960662841797e-05,
      "step": 22523,
      "training_step_time": 6.658766508102417
    },
    {
      "epoch": 2.1480560302734376e-05,
      "model_forward_time": 1.7551805973052979,
      "step": 22524
    },
    {
      "epoch": 2.1480560302734376e-05,
      "step": 22524,
      "training_step_time": 7.584077596664429
    },
    {
      "epoch": 2.148151397705078e-05,
      "model_forward_time": 1.8471345901489258,
      "step": 22525
    },
    {
      "epoch": 2.148151397705078e-05,
      "step": 22525,
      "training_step_time": 9.152368545532227
    },
    {
      "epoch": 2.1482467651367187e-05,
      "model_forward_time": 2.875471591949463,
      "step": 22526
    },
    {
      "epoch": 2.1482467651367187e-05,
      "step": 22526,
      "training_step_time": 9.152772188186646
    },
    {
      "epoch": 2.1483421325683595e-05,
      "model_forward_time": 1.7572319507598877,
      "step": 22527
    },
    {
      "epoch": 2.1483421325683595e-05,
      "step": 22527,
      "training_step_time": 8.263179302215576
    },
    {
      "epoch": 2.1484375e-05,
      "model_forward_time": 1.765462875366211,
      "step": 22528
    },
    {
      "epoch": 2.1484375e-05,
      "step": 22528,
      "training_step_time": 7.550172567367554
    },
    {
      "epoch": 2.1485328674316406e-05,
      "model_forward_time": 1.760716199874878,
      "step": 22529
    },
    {
      "epoch": 2.1485328674316406e-05,
      "step": 22529,
      "training_step_time": 7.021192789077759
    },
    {
      "epoch": 2.1486282348632814e-05,
      "grad_norm": 0.165781632065773,
      "learning_rate": 9.183207816262645e-06,
      "loss": 0.0593,
      "step": 22530
    },
    {
      "epoch": 2.1486282348632814e-05,
      "model_forward_time": 1.765319585800171,
      "step": 22530
    },
    {
      "dynamics_loss_avg": 0.05872817672789097,
      "epoch": 2.1486282348632814e-05,
      "step": 22530
    },
    {
      "action_loss_avg": 0.002135982667095959,
      "epoch": 2.1486282348632814e-05,
      "step": 22530
    },
    {
      "epoch": 2.1486282348632814e-05,
      "step": 22530,
      "training_step_time": 6.720447063446045
    },
    {
      "epoch": 2.1487236022949218e-05,
      "model_forward_time": 1.920983076095581,
      "step": 22531
    },
    {
      "epoch": 2.1487236022949218e-05,
      "step": 22531,
      "training_step_time": 6.727844476699829
    },
    {
      "epoch": 2.1488189697265625e-05,
      "model_forward_time": 1.9269611835479736,
      "step": 22532
    },
    {
      "epoch": 2.1488189697265625e-05,
      "step": 22532,
      "training_step_time": 6.70498514175415
    },
    {
      "epoch": 2.1489143371582032e-05,
      "model_forward_time": 1.919299840927124,
      "step": 22533
    },
    {
      "epoch": 2.1489143371582032e-05,
      "step": 22533,
      "training_step_time": 7.886735200881958
    },
    {
      "epoch": 2.1490097045898436e-05,
      "model_forward_time": 1.929067611694336,
      "step": 22534
    },
    {
      "epoch": 2.1490097045898436e-05,
      "step": 22534,
      "training_step_time": 8.75429630279541
    },
    {
      "epoch": 2.1491050720214844e-05,
      "model_forward_time": 1.9236936569213867,
      "step": 22535
    },
    {
      "epoch": 2.1491050720214844e-05,
      "step": 22535,
      "training_step_time": 8.594244718551636
    },
    {
      "epoch": 2.149200439453125e-05,
      "model_forward_time": 2.583211660385132,
      "step": 22536
    },
    {
      "epoch": 2.149200439453125e-05,
      "step": 22536,
      "training_step_time": 8.93988037109375
    },
    {
      "epoch": 2.1492958068847655e-05,
      "model_forward_time": 1.8838598728179932,
      "step": 22537
    },
    {
      "epoch": 2.1492958068847655e-05,
      "step": 22537,
      "training_step_time": 8.098581552505493
    },
    {
      "epoch": 2.1493911743164063e-05,
      "model_forward_time": 1.5558335781097412,
      "step": 22538
    },
    {
      "epoch": 2.1493911743164063e-05,
      "step": 22538,
      "training_step_time": 7.166515350341797
    },
    {
      "epoch": 2.149486541748047e-05,
      "model_forward_time": 1.915947675704956,
      "step": 22539
    },
    {
      "epoch": 2.149486541748047e-05,
      "step": 22539,
      "training_step_time": 6.766243934631348
    },
    {
      "epoch": 2.1495819091796874e-05,
      "grad_norm": 0.14511793851852417,
      "learning_rate": 9.182301898195892e-06,
      "loss": 0.0626,
      "step": 22540
    },
    {
      "epoch": 2.1495819091796874e-05,
      "model_forward_time": 1.923973798751831,
      "step": 22540
    },
    {
      "dynamics_loss_avg": 0.0536678796634078,
      "epoch": 2.1495819091796874e-05,
      "step": 22540
    },
    {
      "action_loss_avg": 0.0027045999944675715,
      "epoch": 2.1495819091796874e-05,
      "step": 22540
    },
    {
      "epoch": 2.1495819091796874e-05,
      "step": 22540,
      "training_step_time": 6.732165813446045
    },
    {
      "epoch": 2.149677276611328e-05,
      "model_forward_time": 1.8976759910583496,
      "step": 22541
    },
    {
      "epoch": 2.149677276611328e-05,
      "step": 22541,
      "training_step_time": 6.757565021514893
    },
    {
      "epoch": 2.149772644042969e-05,
      "model_forward_time": 1.8837525844573975,
      "step": 22542
    },
    {
      "epoch": 2.149772644042969e-05,
      "step": 22542,
      "training_step_time": 6.700176000595093
    },
    {
      "epoch": 2.1498680114746093e-05,
      "model_forward_time": 1.9190673828125,
      "step": 22543
    },
    {
      "epoch": 2.1498680114746093e-05,
      "step": 22543,
      "training_step_time": 7.715533494949341
    },
    {
      "epoch": 2.14996337890625e-05,
      "model_forward_time": 1.8931925296783447,
      "step": 22544
    },
    {
      "epoch": 2.14996337890625e-05,
      "step": 22544,
      "training_step_time": 7.753398895263672
    },
    {
      "epoch": 2.1500587463378908e-05,
      "model_forward_time": 3.0137360095977783,
      "step": 22545
    },
    {
      "epoch": 2.1500587463378908e-05,
      "step": 22545,
      "training_step_time": 8.573887825012207
    },
    {
      "epoch": 2.1501541137695312e-05,
      "model_forward_time": 1.8899836540222168,
      "step": 22546
    },
    {
      "epoch": 2.1501541137695312e-05,
      "step": 22546,
      "training_step_time": 7.464742660522461
    },
    {
      "epoch": 2.150249481201172e-05,
      "model_forward_time": 1.90409517288208,
      "step": 22547
    },
    {
      "epoch": 2.150249481201172e-05,
      "step": 22547,
      "training_step_time": 6.993812084197998
    },
    {
      "epoch": 2.1503448486328123e-05,
      "model_forward_time": 1.9176599979400635,
      "step": 22548
    },
    {
      "epoch": 2.1503448486328123e-05,
      "step": 22548,
      "training_step_time": 7.012282371520996
    },
    {
      "epoch": 2.150440216064453e-05,
      "model_forward_time": 1.914738416671753,
      "step": 22549
    },
    {
      "epoch": 2.150440216064453e-05,
      "step": 22549,
      "training_step_time": 6.706358909606934
    },
    {
      "epoch": 2.1505355834960938e-05,
      "grad_norm": 0.18991468846797943,
      "learning_rate": 9.181395522758889e-06,
      "loss": 0.0585,
      "step": 22550
    },
    {
      "epoch": 2.1505355834960938e-05,
      "model_forward_time": 1.9196856021881104,
      "step": 22550
    },
    {
      "dynamics_loss_avg": 0.06887010429054499,
      "epoch": 2.1505355834960938e-05,
      "step": 22550
    },
    {
      "action_loss_avg": 0.0023121130230720154,
      "epoch": 2.1505355834960938e-05,
      "step": 22550
    },
    {
      "epoch": 2.1505355834960938e-05,
      "step": 22550,
      "training_step_time": 6.666630506515503
    },
    {
      "epoch": 2.1506309509277342e-05,
      "model_forward_time": 1.9267151355743408,
      "step": 22551
    },
    {
      "epoch": 2.1506309509277342e-05,
      "step": 22551,
      "training_step_time": 6.656707525253296
    },
    {
      "epoch": 2.150726318359375e-05,
      "model_forward_time": 1.9248089790344238,
      "step": 22552
    },
    {
      "epoch": 2.150726318359375e-05,
      "step": 22552,
      "training_step_time": 7.713014602661133
    },
    {
      "epoch": 2.1508216857910157e-05,
      "model_forward_time": 1.89339017868042,
      "step": 22553
    },
    {
      "epoch": 2.1508216857910157e-05,
      "step": 22553,
      "training_step_time": 9.26628828048706
    },
    {
      "epoch": 2.150917053222656e-05,
      "model_forward_time": 2.953968048095703,
      "step": 22554
    },
    {
      "epoch": 2.150917053222656e-05,
      "step": 22554,
      "training_step_time": 7.730469703674316
    },
    {
      "epoch": 2.151012420654297e-05,
      "model_forward_time": 1.9326133728027344,
      "step": 22555
    },
    {
      "epoch": 2.151012420654297e-05,
      "step": 22555,
      "training_step_time": 7.723595142364502
    },
    {
      "epoch": 2.1511077880859376e-05,
      "model_forward_time": 1.919555425643921,
      "step": 22556
    },
    {
      "epoch": 2.1511077880859376e-05,
      "step": 22556,
      "training_step_time": 7.05412745475769
    },
    {
      "epoch": 2.151203155517578e-05,
      "model_forward_time": 1.8994402885437012,
      "step": 22557
    },
    {
      "epoch": 2.151203155517578e-05,
      "step": 22557,
      "training_step_time": 7.0731401443481445
    },
    {
      "epoch": 2.1512985229492187e-05,
      "model_forward_time": 1.9086823463439941,
      "step": 22558
    },
    {
      "epoch": 2.1512985229492187e-05,
      "step": 22558,
      "training_step_time": 6.757527828216553
    },
    {
      "epoch": 2.1513938903808595e-05,
      "model_forward_time": 1.9082837104797363,
      "step": 22559
    },
    {
      "epoch": 2.1513938903808595e-05,
      "step": 22559,
      "training_step_time": 6.751754522323608
    },
    {
      "epoch": 2.1514892578125e-05,
      "grad_norm": 0.13963794708251953,
      "learning_rate": 9.18048869005076e-06,
      "loss": 0.0599,
      "step": 22560
    },
    {
      "epoch": 2.1514892578125e-05,
      "model_forward_time": 0.8615036010742188,
      "step": 22560
    },
    {
      "dynamics_loss_avg": 0.07870641630142927,
      "epoch": 2.1514892578125e-05,
      "step": 22560
    },
    {
      "action_loss_avg": 0.004443553200690076,
      "epoch": 2.1514892578125e-05,
      "step": 22560
    },
    {
      "epoch": 2.1514892578125e-05,
      "step": 22560,
      "training_step_time": 6.78521990776062
    },
    {
      "epoch": 2.1515846252441406e-05,
      "model_forward_time": 1.7539989948272705,
      "step": 22561
    },
    {
      "epoch": 2.1515846252441406e-05,
      "step": 22561,
      "training_step_time": 9.651732444763184
    },
    {
      "epoch": 2.1516799926757814e-05,
      "model_forward_time": 1.8166899681091309,
      "step": 22562
    },
    {
      "epoch": 2.1516799926757814e-05,
      "step": 22562,
      "training_step_time": 8.139589309692383
    },
    {
      "epoch": 2.1517753601074218e-05,
      "model_forward_time": 1.7570030689239502,
      "step": 22563
    },
    {
      "epoch": 2.1517753601074218e-05,
      "step": 22563,
      "training_step_time": 8.65354871749878
    },
    {
      "epoch": 2.1518707275390625e-05,
      "model_forward_time": 2.843125343322754,
      "step": 22564
    },
    {
      "epoch": 2.1518707275390625e-05,
      "step": 22564,
      "training_step_time": 7.640963315963745
    },
    {
      "epoch": 2.1519660949707032e-05,
      "model_forward_time": 1.7450790405273438,
      "step": 22565
    },
    {
      "epoch": 2.1519660949707032e-05,
      "step": 22565,
      "training_step_time": 7.345350742340088
    },
    {
      "epoch": 2.1520614624023436e-05,
      "model_forward_time": 1.7447521686553955,
      "step": 22566
    },
    {
      "epoch": 2.1520614624023436e-05,
      "step": 22566,
      "training_step_time": 7.031528472900391
    },
    {
      "epoch": 2.1521568298339844e-05,
      "model_forward_time": 1.7443056106567383,
      "step": 22567
    },
    {
      "epoch": 2.1521568298339844e-05,
      "step": 22567,
      "training_step_time": 6.718295574188232
    },
    {
      "epoch": 2.152252197265625e-05,
      "model_forward_time": 1.75520920753479,
      "step": 22568
    },
    {
      "epoch": 2.152252197265625e-05,
      "step": 22568,
      "training_step_time": 6.687195777893066
    },
    {
      "epoch": 2.1523475646972655e-05,
      "model_forward_time": 1.759446144104004,
      "step": 22569
    },
    {
      "epoch": 2.1523475646972655e-05,
      "step": 22569,
      "training_step_time": 6.808675289154053
    },
    {
      "epoch": 2.1524429321289063e-05,
      "grad_norm": 0.192087784409523,
      "learning_rate": 9.179581400170673e-06,
      "loss": 0.0581,
      "step": 22570
    },
    {
      "epoch": 2.1524429321289063e-05,
      "model_forward_time": 1.773831844329834,
      "step": 22570
    },
    {
      "dynamics_loss_avg": 0.07462734133005142,
      "epoch": 2.1524429321289063e-05,
      "step": 22570
    },
    {
      "action_loss_avg": 0.0035917436005547644,
      "epoch": 2.1524429321289063e-05,
      "step": 22570
    },
    {
      "epoch": 2.1524429321289063e-05,
      "step": 22570,
      "training_step_time": 7.740997552871704
    },
    {
      "epoch": 2.152538299560547e-05,
      "model_forward_time": 1.820352554321289,
      "step": 22571
    },
    {
      "epoch": 2.152538299560547e-05,
      "step": 22571,
      "training_step_time": 8.585002183914185
    },
    {
      "epoch": 2.1526336669921874e-05,
      "model_forward_time": 1.7571566104888916,
      "step": 22572
    },
    {
      "epoch": 2.1526336669921874e-05,
      "step": 22572,
      "training_step_time": 10.441316604614258
    },
    {
      "epoch": 2.152729034423828e-05,
      "model_forward_time": 2.860944986343384,
      "step": 22573
    },
    {
      "epoch": 2.152729034423828e-05,
      "step": 22573,
      "training_step_time": 7.600548028945923
    },
    {
      "epoch": 2.152824401855469e-05,
      "model_forward_time": 1.7543308734893799,
      "step": 22574
    },
    {
      "epoch": 2.152824401855469e-05,
      "step": 22574,
      "training_step_time": 7.483649015426636
    },
    {
      "epoch": 2.1529197692871093e-05,
      "model_forward_time": 1.7581467628479004,
      "step": 22575
    },
    {
      "epoch": 2.1529197692871093e-05,
      "step": 22575,
      "training_step_time": 7.030710697174072
    },
    {
      "epoch": 2.15301513671875e-05,
      "model_forward_time": 1.7607800960540771,
      "step": 22576
    },
    {
      "epoch": 2.15301513671875e-05,
      "step": 22576,
      "training_step_time": 6.6984734535217285
    },
    {
      "epoch": 2.1531105041503908e-05,
      "model_forward_time": 1.7648744583129883,
      "step": 22577
    },
    {
      "epoch": 2.1531105041503908e-05,
      "step": 22577,
      "training_step_time": 6.813830375671387
    },
    {
      "epoch": 2.1532058715820312e-05,
      "model_forward_time": 1.75494384765625,
      "step": 22578
    },
    {
      "epoch": 2.1532058715820312e-05,
      "step": 22578,
      "training_step_time": 6.762423992156982
    },
    {
      "epoch": 2.153301239013672e-05,
      "model_forward_time": 1.7531583309173584,
      "step": 22579
    },
    {
      "epoch": 2.153301239013672e-05,
      "step": 22579,
      "training_step_time": 7.892312526702881
    },
    {
      "epoch": 2.1533966064453123e-05,
      "grad_norm": 0.17211560904979706,
      "learning_rate": 9.178673653217845e-06,
      "loss": 0.0599,
      "step": 22580
    },
    {
      "epoch": 2.1533966064453123e-05,
      "model_forward_time": 1.7814388275146484,
      "step": 22580
    },
    {
      "dynamics_loss_avg": 0.05808357167989016,
      "epoch": 2.1533966064453123e-05,
      "step": 22580
    },
    {
      "action_loss_avg": 0.002278866624692455,
      "epoch": 2.1533966064453123e-05,
      "step": 22580
    },
    {
      "epoch": 2.1533966064453123e-05,
      "step": 22580,
      "training_step_time": 8.764304399490356
    },
    {
      "epoch": 2.153491973876953e-05,
      "model_forward_time": 1.7561495304107666,
      "step": 22581
    },
    {
      "epoch": 2.153491973876953e-05,
      "step": 22581,
      "training_step_time": 11.121575593948364
    },
    {
      "epoch": 2.1535873413085938e-05,
      "model_forward_time": 2.8631820678710938,
      "step": 22582
    },
    {
      "epoch": 2.1535873413085938e-05,
      "step": 22582,
      "training_step_time": 7.660545825958252
    },
    {
      "epoch": 2.1536827087402342e-05,
      "model_forward_time": 1.7535717487335205,
      "step": 22583
    },
    {
      "epoch": 2.1536827087402342e-05,
      "step": 22583,
      "training_step_time": 7.583494186401367
    },
    {
      "epoch": 2.153778076171875e-05,
      "model_forward_time": 1.7554898262023926,
      "step": 22584
    },
    {
      "epoch": 2.153778076171875e-05,
      "step": 22584,
      "training_step_time": 7.691065311431885
    },
    {
      "epoch": 2.1538734436035157e-05,
      "model_forward_time": 1.776550054550171,
      "step": 22585
    },
    {
      "epoch": 2.1538734436035157e-05,
      "step": 22585,
      "training_step_time": 7.066265344619751
    },
    {
      "epoch": 2.153968811035156e-05,
      "model_forward_time": 1.6988613605499268,
      "step": 22586
    },
    {
      "epoch": 2.153968811035156e-05,
      "step": 22586,
      "training_step_time": 6.753217458724976
    },
    {
      "epoch": 2.154064178466797e-05,
      "model_forward_time": 1.7056622505187988,
      "step": 22587
    },
    {
      "epoch": 2.154064178466797e-05,
      "step": 22587,
      "training_step_time": 6.800196170806885
    },
    {
      "epoch": 2.1541595458984376e-05,
      "model_forward_time": 1.7033166885375977,
      "step": 22588
    },
    {
      "epoch": 2.1541595458984376e-05,
      "step": 22588,
      "training_step_time": 7.842268943786621
    },
    {
      "epoch": 2.154254913330078e-05,
      "model_forward_time": 1.7104213237762451,
      "step": 22589
    },
    {
      "epoch": 2.154254913330078e-05,
      "step": 22589,
      "training_step_time": 8.653914451599121
    },
    {
      "epoch": 2.1543502807617187e-05,
      "grad_norm": 0.2227751761674881,
      "learning_rate": 9.177765449291552e-06,
      "loss": 0.0609,
      "step": 22590
    },
    {
      "epoch": 2.1543502807617187e-05,
      "model_forward_time": 1.69511079788208,
      "step": 22590
    },
    {
      "dynamics_loss_avg": 0.05107657331973314,
      "epoch": 2.1543502807617187e-05,
      "step": 22590
    },
    {
      "action_loss_avg": 0.002503305149730295,
      "epoch": 2.1543502807617187e-05,
      "step": 22590
    },
    {
      "epoch": 2.1543502807617187e-05,
      "step": 22590,
      "training_step_time": 9.295655012130737
    },
    {
      "epoch": 2.1544456481933595e-05,
      "model_forward_time": 2.2471330165863037,
      "step": 22591
    },
    {
      "epoch": 2.1544456481933595e-05,
      "step": 22591,
      "training_step_time": 9.15868592262268
    },
    {
      "epoch": 2.154541015625e-05,
      "model_forward_time": 1.7022881507873535,
      "step": 22592
    },
    {
      "epoch": 2.154541015625e-05,
      "step": 22592,
      "training_step_time": 7.6894824504852295
    },
    {
      "epoch": 2.1546363830566406e-05,
      "model_forward_time": 1.7046754360198975,
      "step": 22593
    },
    {
      "epoch": 2.1546363830566406e-05,
      "step": 22593,
      "training_step_time": 7.731343984603882
    },
    {
      "epoch": 2.1547317504882814e-05,
      "model_forward_time": 1.7107882499694824,
      "step": 22594
    },
    {
      "epoch": 2.1547317504882814e-05,
      "step": 22594,
      "training_step_time": 7.079663515090942
    },
    {
      "epoch": 2.1548271179199218e-05,
      "model_forward_time": 1.703298568725586,
      "step": 22595
    },
    {
      "epoch": 2.1548271179199218e-05,
      "step": 22595,
      "training_step_time": 6.8044633865356445
    },
    {
      "epoch": 2.1549224853515625e-05,
      "model_forward_time": 1.7043814659118652,
      "step": 22596
    },
    {
      "epoch": 2.1549224853515625e-05,
      "step": 22596,
      "training_step_time": 6.763298511505127
    },
    {
      "epoch": 2.1550178527832032e-05,
      "model_forward_time": 1.7001352310180664,
      "step": 22597
    },
    {
      "epoch": 2.1550178527832032e-05,
      "step": 22597,
      "training_step_time": 7.768127918243408
    },
    {
      "epoch": 2.1551132202148437e-05,
      "model_forward_time": 1.7086048126220703,
      "step": 22598
    },
    {
      "epoch": 2.1551132202148437e-05,
      "step": 22598,
      "training_step_time": 7.740772247314453
    },
    {
      "epoch": 2.1552085876464844e-05,
      "model_forward_time": 1.3192481994628906,
      "step": 22599
    },
    {
      "epoch": 2.1552085876464844e-05,
      "step": 22599,
      "training_step_time": 8.584386110305786
    },
    {
      "epoch": 2.155303955078125e-05,
      "grad_norm": 0.19482719898223877,
      "learning_rate": 9.176856788491109e-06,
      "loss": 0.0596,
      "step": 22600
    },
    {
      "epoch": 2.155303955078125e-05,
      "model_forward_time": 1.3334102630615234,
      "step": 22600
    },
    {
      "dynamics_loss_avg": 0.050250757671892646,
      "epoch": 2.155303955078125e-05,
      "step": 22600
    },
    {
      "action_loss_avg": 0.0031326699478086083,
      "epoch": 2.155303955078125e-05,
      "step": 22600
    },
    {
      "epoch": 2.155303955078125e-05,
      "step": 22600,
      "training_step_time": 8.526748895645142
    },
    {
      "epoch": 2.1553993225097655e-05,
      "model_forward_time": 2.257874011993408,
      "step": 22601
    },
    {
      "epoch": 2.1553993225097655e-05,
      "step": 22601,
      "training_step_time": 7.57087779045105
    },
    {
      "epoch": 2.1554946899414063e-05,
      "model_forward_time": 1.7038345336914062,
      "step": 22602
    },
    {
      "epoch": 2.1554946899414063e-05,
      "step": 22602,
      "training_step_time": 7.702175140380859
    },
    {
      "epoch": 2.155590057373047e-05,
      "model_forward_time": 1.7055137157440186,
      "step": 22603
    },
    {
      "epoch": 2.155590057373047e-05,
      "step": 22603,
      "training_step_time": 7.092136859893799
    },
    {
      "epoch": 2.1556854248046874e-05,
      "model_forward_time": 1.7085154056549072,
      "step": 22604
    },
    {
      "epoch": 2.1556854248046874e-05,
      "step": 22604,
      "training_step_time": 6.773390054702759
    },
    {
      "epoch": 2.155780792236328e-05,
      "model_forward_time": 1.694861650466919,
      "step": 22605
    },
    {
      "epoch": 2.155780792236328e-05,
      "step": 22605,
      "training_step_time": 6.799545764923096
    },
    {
      "epoch": 2.155876159667969e-05,
      "model_forward_time": 1.7056488990783691,
      "step": 22606
    },
    {
      "epoch": 2.155876159667969e-05,
      "step": 22606,
      "training_step_time": 7.682356834411621
    },
    {
      "epoch": 2.1559715270996093e-05,
      "model_forward_time": 1.7036266326904297,
      "step": 22607
    },
    {
      "epoch": 2.1559715270996093e-05,
      "step": 22607,
      "training_step_time": 7.76636528968811
    },
    {
      "epoch": 2.15606689453125e-05,
      "model_forward_time": 1.7088291645050049,
      "step": 22608
    },
    {
      "epoch": 2.15606689453125e-05,
      "step": 22608,
      "training_step_time": 9.25922155380249
    },
    {
      "epoch": 2.1561622619628908e-05,
      "model_forward_time": 1.7117877006530762,
      "step": 22609
    },
    {
      "epoch": 2.1561622619628908e-05,
      "step": 22609,
      "training_step_time": 9.09334945678711
    },
    {
      "epoch": 2.1562576293945312e-05,
      "grad_norm": 0.21228045225143433,
      "learning_rate": 9.175947670915888e-06,
      "loss": 0.0618,
      "step": 22610
    },
    {
      "epoch": 2.1562576293945312e-05,
      "model_forward_time": 2.251551628112793,
      "step": 22610
    },
    {
      "dynamics_loss_avg": 0.06519989557564258,
      "epoch": 2.1562576293945312e-05,
      "step": 22610
    },
    {
      "action_loss_avg": 0.003175139130325988,
      "epoch": 2.1562576293945312e-05,
      "step": 22610
    },
    {
      "epoch": 2.1562576293945312e-05,
      "step": 22610,
      "training_step_time": 7.57296895980835
    },
    {
      "epoch": 2.156352996826172e-05,
      "model_forward_time": 1.6940033435821533,
      "step": 22611
    },
    {
      "epoch": 2.156352996826172e-05,
      "step": 22611,
      "training_step_time": 8.241278171539307
    },
    {
      "epoch": 2.1564483642578123e-05,
      "model_forward_time": 1.7082529067993164,
      "step": 22612
    },
    {
      "epoch": 2.1564483642578123e-05,
      "step": 22612,
      "training_step_time": 7.364800930023193
    },
    {
      "epoch": 2.156543731689453e-05,
      "model_forward_time": 1.3205008506774902,
      "step": 22613
    },
    {
      "epoch": 2.156543731689453e-05,
      "step": 22613,
      "training_step_time": 6.800130128860474
    },
    {
      "epoch": 2.1566390991210938e-05,
      "model_forward_time": 1.7000889778137207,
      "step": 22614
    },
    {
      "epoch": 2.1566390991210938e-05,
      "step": 22614,
      "training_step_time": 6.8295578956604
    },
    {
      "epoch": 2.1567344665527342e-05,
      "model_forward_time": 1.699479579925537,
      "step": 22615
    },
    {
      "epoch": 2.1567344665527342e-05,
      "step": 22615,
      "training_step_time": 7.759822845458984
    },
    {
      "epoch": 2.156829833984375e-05,
      "model_forward_time": 1.6995184421539307,
      "step": 22616
    },
    {
      "epoch": 2.156829833984375e-05,
      "step": 22616,
      "training_step_time": 7.729826927185059
    },
    {
      "epoch": 2.1569252014160157e-05,
      "model_forward_time": 1.7097694873809814,
      "step": 22617
    },
    {
      "epoch": 2.1569252014160157e-05,
      "step": 22617,
      "training_step_time": 8.723217725753784
    },
    {
      "epoch": 2.157020568847656e-05,
      "model_forward_time": 1.6972055435180664,
      "step": 22618
    },
    {
      "epoch": 2.157020568847656e-05,
      "step": 22618,
      "training_step_time": 8.736511707305908
    },
    {
      "epoch": 2.157115936279297e-05,
      "model_forward_time": 2.2184290885925293,
      "step": 22619
    },
    {
      "epoch": 2.157115936279297e-05,
      "step": 22619,
      "training_step_time": 8.24268388748169
    },
    {
      "epoch": 2.1572113037109376e-05,
      "grad_norm": 0.22035863995552063,
      "learning_rate": 9.17503809666531e-06,
      "loss": 0.064,
      "step": 22620
    },
    {
      "epoch": 2.1572113037109376e-05,
      "model_forward_time": 1.6883342266082764,
      "step": 22620
    },
    {
      "dynamics_loss_avg": 0.07416630312800407,
      "epoch": 2.1572113037109376e-05,
      "step": 22620
    },
    {
      "action_loss_avg": 0.0038554724014829844,
      "epoch": 2.1572113037109376e-05,
      "step": 22620
    },
    {
      "epoch": 2.1572113037109376e-05,
      "step": 22620,
      "training_step_time": 7.66968846321106
    },
    {
      "epoch": 2.157306671142578e-05,
      "model_forward_time": 1.7028348445892334,
      "step": 22621
    },
    {
      "epoch": 2.157306671142578e-05,
      "step": 22621,
      "training_step_time": 7.723567724227905
    },
    {
      "epoch": 2.1574020385742187e-05,
      "model_forward_time": 1.700819492340088,
      "step": 22622
    },
    {
      "epoch": 2.1574020385742187e-05,
      "step": 22622,
      "training_step_time": 7.127335548400879
    },
    {
      "epoch": 2.1574974060058595e-05,
      "model_forward_time": 1.6954760551452637,
      "step": 22623
    },
    {
      "epoch": 2.1574974060058595e-05,
      "step": 22623,
      "training_step_time": 6.81288743019104
    },
    {
      "epoch": 2.1575927734375e-05,
      "model_forward_time": 1.6981017589569092,
      "step": 22624
    },
    {
      "epoch": 2.1575927734375e-05,
      "step": 22624,
      "training_step_time": 7.8746867179870605
    },
    {
      "epoch": 2.1576881408691406e-05,
      "model_forward_time": 1.7018139362335205,
      "step": 22625
    },
    {
      "epoch": 2.1576881408691406e-05,
      "step": 22625,
      "training_step_time": 9.26583743095398
    },
    {
      "epoch": 2.1577835083007814e-05,
      "model_forward_time": 1.7055609226226807,
      "step": 22626
    },
    {
      "epoch": 2.1577835083007814e-05,
      "step": 22626,
      "training_step_time": 8.749662399291992
    },
    {
      "epoch": 2.1578788757324218e-05,
      "model_forward_time": 1.6913387775421143,
      "step": 22627
    },
    {
      "epoch": 2.1578788757324218e-05,
      "step": 22627,
      "training_step_time": 8.164900302886963
    },
    {
      "epoch": 2.1579742431640625e-05,
      "model_forward_time": 1.711454153060913,
      "step": 22628
    },
    {
      "epoch": 2.1579742431640625e-05,
      "step": 22628,
      "training_step_time": 7.773496150970459
    },
    {
      "epoch": 2.1580696105957033e-05,
      "model_forward_time": 2.250101327896118,
      "step": 22629
    },
    {
      "epoch": 2.1580696105957033e-05,
      "step": 22629,
      "training_step_time": 7.791134595870972
    },
    {
      "epoch": 2.1581649780273437e-05,
      "grad_norm": 0.09078997373580933,
      "learning_rate": 9.174128065838842e-06,
      "loss": 0.0598,
      "step": 22630
    },
    {
      "epoch": 2.1581649780273437e-05,
      "model_forward_time": 1.7033534049987793,
      "step": 22630
    },
    {
      "dynamics_loss_avg": 0.05069017969071865,
      "epoch": 2.1581649780273437e-05,
      "step": 22630
    },
    {
      "action_loss_avg": 0.002736087533412501,
      "epoch": 2.1581649780273437e-05,
      "step": 22630
    },
    {
      "epoch": 2.1581649780273437e-05,
      "step": 22630,
      "training_step_time": 7.776881217956543
    },
    {
      "epoch": 2.1582603454589844e-05,
      "model_forward_time": 1.711998462677002,
      "step": 22631
    },
    {
      "epoch": 2.1582603454589844e-05,
      "step": 22631,
      "training_step_time": 7.640794038772583
    },
    {
      "epoch": 2.158355712890625e-05,
      "model_forward_time": 1.7188565731048584,
      "step": 22632
    },
    {
      "epoch": 2.158355712890625e-05,
      "step": 22632,
      "training_step_time": 6.840337514877319
    },
    {
      "epoch": 2.1584510803222655e-05,
      "model_forward_time": 1.7144830226898193,
      "step": 22633
    },
    {
      "epoch": 2.1584510803222655e-05,
      "step": 22633,
      "training_step_time": 7.895394325256348
    },
    {
      "epoch": 2.1585464477539063e-05,
      "model_forward_time": 1.712883472442627,
      "step": 22634
    },
    {
      "epoch": 2.1585464477539063e-05,
      "step": 22634,
      "training_step_time": 7.867051839828491
    },
    {
      "epoch": 2.158641815185547e-05,
      "model_forward_time": 1.7126381397247314,
      "step": 22635
    },
    {
      "epoch": 2.158641815185547e-05,
      "step": 22635,
      "training_step_time": 9.430789470672607
    },
    {
      "epoch": 2.1587371826171874e-05,
      "model_forward_time": 1.7080299854278564,
      "step": 22636
    },
    {
      "epoch": 2.1587371826171874e-05,
      "step": 22636,
      "training_step_time": 7.830672264099121
    },
    {
      "epoch": 2.158832550048828e-05,
      "model_forward_time": 1.7330973148345947,
      "step": 22637
    },
    {
      "epoch": 2.158832550048828e-05,
      "step": 22637,
      "training_step_time": 8.52716612815857
    },
    {
      "epoch": 2.158927917480469e-05,
      "model_forward_time": 2.2532496452331543,
      "step": 22638
    },
    {
      "epoch": 2.158927917480469e-05,
      "step": 22638,
      "training_step_time": 8.386008024215698
    },
    {
      "epoch": 2.1590232849121093e-05,
      "model_forward_time": 1.7141156196594238,
      "step": 22639
    },
    {
      "epoch": 2.1590232849121093e-05,
      "step": 22639,
      "training_step_time": 7.773466110229492
    },
    {
      "epoch": 2.15911865234375e-05,
      "grad_norm": 0.15999524295330048,
      "learning_rate": 9.173217578536002e-06,
      "loss": 0.061,
      "step": 22640
    },
    {
      "epoch": 2.15911865234375e-05,
      "model_forward_time": 1.7043273448944092,
      "step": 22640
    },
    {
      "dynamics_loss_avg": 0.06550819659605622,
      "epoch": 2.15911865234375e-05,
      "step": 22640
    },
    {
      "action_loss_avg": 0.003572447900660336,
      "epoch": 2.15911865234375e-05,
      "step": 22640
    },
    {
      "epoch": 2.15911865234375e-05,
      "step": 22640,
      "training_step_time": 7.680983781814575
    },
    {
      "epoch": 2.1592140197753905e-05,
      "model_forward_time": 1.7134678363800049,
      "step": 22641
    },
    {
      "epoch": 2.1592140197753905e-05,
      "step": 22641,
      "training_step_time": 6.8718342781066895
    },
    {
      "epoch": 2.1593093872070312e-05,
      "model_forward_time": 1.7127361297607422,
      "step": 22642
    },
    {
      "epoch": 2.1593093872070312e-05,
      "step": 22642,
      "training_step_time": 7.609049320220947
    },
    {
      "epoch": 2.159404754638672e-05,
      "model_forward_time": 1.7327768802642822,
      "step": 22643
    },
    {
      "epoch": 2.159404754638672e-05,
      "step": 22643,
      "training_step_time": 7.899291038513184
    },
    {
      "epoch": 2.1595001220703123e-05,
      "model_forward_time": 1.7021331787109375,
      "step": 22644
    },
    {
      "epoch": 2.1595001220703123e-05,
      "step": 22644,
      "training_step_time": 8.515901327133179
    },
    {
      "epoch": 2.159595489501953e-05,
      "model_forward_time": 1.7073795795440674,
      "step": 22645
    },
    {
      "epoch": 2.159595489501953e-05,
      "step": 22645,
      "training_step_time": 10.700847148895264
    },
    {
      "epoch": 2.1596908569335938e-05,
      "model_forward_time": 1.3173949718475342,
      "step": 22646
    },
    {
      "epoch": 2.1596908569335938e-05,
      "step": 22646,
      "training_step_time": 7.810020685195923
    },
    {
      "epoch": 2.1597862243652342e-05,
      "model_forward_time": 2.2728493213653564,
      "step": 22647
    },
    {
      "epoch": 2.1597862243652342e-05,
      "step": 22647,
      "training_step_time": 7.808952808380127
    },
    {
      "epoch": 2.159881591796875e-05,
      "model_forward_time": 0.9921267032623291,
      "step": 22648
    },
    {
      "epoch": 2.159881591796875e-05,
      "step": 22648,
      "training_step_time": 7.621707439422607
    },
    {
      "epoch": 2.1599769592285157e-05,
      "model_forward_time": 1.711594820022583,
      "step": 22649
    },
    {
      "epoch": 2.1599769592285157e-05,
      "step": 22649,
      "training_step_time": 7.740561008453369
    },
    {
      "epoch": 2.160072326660156e-05,
      "grad_norm": 0.13526120781898499,
      "learning_rate": 9.172306634856363e-06,
      "loss": 0.0602,
      "step": 22650
    },
    {
      "epoch": 2.160072326660156e-05,
      "model_forward_time": 1.7074484825134277,
      "step": 22650
    },
    {
      "dynamics_loss_avg": 0.06356810433790087,
      "epoch": 2.160072326660156e-05,
      "step": 22650
    },
    {
      "action_loss_avg": 0.0025906660477630792,
      "epoch": 2.160072326660156e-05,
      "step": 22650
    },
    {
      "epoch": 2.160072326660156e-05,
      "step": 22650,
      "training_step_time": 6.901638507843018
    },
    {
      "epoch": 2.160167694091797e-05,
      "model_forward_time": 1.7078087329864502,
      "step": 22651
    },
    {
      "epoch": 2.160167694091797e-05,
      "step": 22651,
      "training_step_time": 6.74241304397583
    },
    {
      "epoch": 2.1602630615234376e-05,
      "model_forward_time": 1.7108311653137207,
      "step": 22652
    },
    {
      "epoch": 2.1602630615234376e-05,
      "step": 22652,
      "training_step_time": 7.834441423416138
    },
    {
      "epoch": 2.160358428955078e-05,
      "model_forward_time": 1.7107694149017334,
      "step": 22653
    },
    {
      "epoch": 2.160358428955078e-05,
      "step": 22653,
      "training_step_time": 7.8203840255737305
    },
    {
      "epoch": 2.1604537963867187e-05,
      "model_forward_time": 1.7137410640716553,
      "step": 22654
    },
    {
      "epoch": 2.1604537963867187e-05,
      "step": 22654,
      "training_step_time": 8.342987775802612
    },
    {
      "epoch": 2.1605491638183595e-05,
      "model_forward_time": 1.7085700035095215,
      "step": 22655
    },
    {
      "epoch": 2.1605491638183595e-05,
      "step": 22655,
      "training_step_time": 7.793716669082642
    },
    {
      "epoch": 2.16064453125e-05,
      "model_forward_time": 1.7597520351409912,
      "step": 22656
    },
    {
      "epoch": 2.16064453125e-05,
      "step": 22656,
      "training_step_time": 8.568078756332397
    },
    {
      "epoch": 2.1607398986816406e-05,
      "model_forward_time": 2.236961841583252,
      "step": 22657
    },
    {
      "epoch": 2.1607398986816406e-05,
      "step": 22657,
      "training_step_time": 8.164656400680542
    },
    {
      "epoch": 2.1608352661132814e-05,
      "model_forward_time": 1.7065954208374023,
      "step": 22658
    },
    {
      "epoch": 2.1608352661132814e-05,
      "step": 22658,
      "training_step_time": 7.753899574279785
    },
    {
      "epoch": 2.1609306335449218e-05,
      "model_forward_time": 1.7155308723449707,
      "step": 22659
    },
    {
      "epoch": 2.1609306335449218e-05,
      "step": 22659,
      "training_step_time": 6.937267780303955
    },
    {
      "epoch": 2.1610260009765625e-05,
      "grad_norm": 0.1379246860742569,
      "learning_rate": 9.171395234899544e-06,
      "loss": 0.0635,
      "step": 22660
    },
    {
      "epoch": 2.1610260009765625e-05,
      "model_forward_time": 1.704411506652832,
      "step": 22660
    },
    {
      "dynamics_loss_avg": 0.061009074188768865,
      "epoch": 2.1610260009765625e-05,
      "step": 22660
    },
    {
      "action_loss_avg": 0.00435267086722888,
      "epoch": 2.1610260009765625e-05,
      "step": 22660
    },
    {
      "epoch": 2.1610260009765625e-05,
      "step": 22660,
      "training_step_time": 6.85505485534668
    },
    {
      "epoch": 2.1611213684082033e-05,
      "model_forward_time": 1.705894947052002,
      "step": 22661
    },
    {
      "epoch": 2.1611213684082033e-05,
      "step": 22661,
      "training_step_time": 8.027512788772583
    },
    {
      "epoch": 2.1612167358398437e-05,
      "model_forward_time": 1.701429843902588,
      "step": 22662
    },
    {
      "epoch": 2.1612167358398437e-05,
      "step": 22662,
      "training_step_time": 8.453503847122192
    },
    {
      "epoch": 2.1613121032714844e-05,
      "model_forward_time": 1.7035179138183594,
      "step": 22663
    },
    {
      "epoch": 2.1613121032714844e-05,
      "step": 22663,
      "training_step_time": 8.993938684463501
    },
    {
      "epoch": 2.161407470703125e-05,
      "model_forward_time": 1.7178006172180176,
      "step": 22664
    },
    {
      "epoch": 2.161407470703125e-05,
      "step": 22664,
      "training_step_time": 7.82059907913208
    },
    {
      "epoch": 2.1615028381347655e-05,
      "model_forward_time": 1.7547552585601807,
      "step": 22665
    },
    {
      "epoch": 2.1615028381347655e-05,
      "step": 22665,
      "training_step_time": 8.30260944366455
    },
    {
      "epoch": 2.1615982055664063e-05,
      "model_forward_time": 2.243427038192749,
      "step": 22666
    },
    {
      "epoch": 2.1615982055664063e-05,
      "step": 22666,
      "training_step_time": 7.660285234451294
    },
    {
      "epoch": 2.161693572998047e-05,
      "model_forward_time": 1.712292194366455,
      "step": 22667
    },
    {
      "epoch": 2.161693572998047e-05,
      "step": 22667,
      "training_step_time": 7.710275888442993
    },
    {
      "epoch": 2.1617889404296874e-05,
      "model_forward_time": 1.7071259021759033,
      "step": 22668
    },
    {
      "epoch": 2.1617889404296874e-05,
      "step": 22668,
      "training_step_time": 7.6180806159973145
    },
    {
      "epoch": 2.1618843078613282e-05,
      "model_forward_time": 1.7128124237060547,
      "step": 22669
    },
    {
      "epoch": 2.1618843078613282e-05,
      "step": 22669,
      "training_step_time": 6.868072032928467
    },
    {
      "epoch": 2.161979675292969e-05,
      "grad_norm": 0.12432613223791122,
      "learning_rate": 9.170483378765214e-06,
      "loss": 0.0608,
      "step": 22670
    },
    {
      "epoch": 2.161979675292969e-05,
      "model_forward_time": 1.7049052715301514,
      "step": 22670
    },
    {
      "dynamics_loss_avg": 0.04706790540367365,
      "epoch": 2.161979675292969e-05,
      "step": 22670
    },
    {
      "action_loss_avg": 0.0031020303838886322,
      "epoch": 2.161979675292969e-05,
      "step": 22670
    },
    {
      "epoch": 2.161979675292969e-05,
      "step": 22670,
      "training_step_time": 7.8624961376190186
    },
    {
      "epoch": 2.1620750427246093e-05,
      "model_forward_time": 1.7144148349761963,
      "step": 22671
    },
    {
      "epoch": 2.1620750427246093e-05,
      "step": 22671,
      "training_step_time": 8.54346513748169
    },
    {
      "epoch": 2.16217041015625e-05,
      "model_forward_time": 1.7060587406158447,
      "step": 22672
    },
    {
      "epoch": 2.16217041015625e-05,
      "step": 22672,
      "training_step_time": 7.627544403076172
    },
    {
      "epoch": 2.1622657775878905e-05,
      "model_forward_time": 1.7152655124664307,
      "step": 22673
    },
    {
      "epoch": 2.1622657775878905e-05,
      "step": 22673,
      "training_step_time": 7.811938762664795
    },
    {
      "epoch": 2.1623611450195312e-05,
      "model_forward_time": 1.758929967880249,
      "step": 22674
    },
    {
      "epoch": 2.1623611450195312e-05,
      "step": 22674,
      "training_step_time": 7.747904539108276
    },
    {
      "epoch": 2.162456512451172e-05,
      "model_forward_time": 2.2411611080169678,
      "step": 22675
    },
    {
      "epoch": 2.162456512451172e-05,
      "step": 22675,
      "training_step_time": 7.815714597702026
    },
    {
      "epoch": 2.1625518798828123e-05,
      "model_forward_time": 1.7149431705474854,
      "step": 22676
    },
    {
      "epoch": 2.1625518798828123e-05,
      "step": 22676,
      "training_step_time": 7.586685419082642
    },
    {
      "epoch": 2.162647247314453e-05,
      "model_forward_time": 1.6937639713287354,
      "step": 22677
    },
    {
      "epoch": 2.162647247314453e-05,
      "step": 22677,
      "training_step_time": 7.632311820983887
    },
    {
      "epoch": 2.1627426147460938e-05,
      "model_forward_time": 1.7105591297149658,
      "step": 22678
    },
    {
      "epoch": 2.1627426147460938e-05,
      "step": 22678,
      "training_step_time": 6.934616804122925
    },
    {
      "epoch": 2.1628379821777342e-05,
      "model_forward_time": 1.710913896560669,
      "step": 22679
    },
    {
      "epoch": 2.1628379821777342e-05,
      "step": 22679,
      "training_step_time": 7.927721261978149
    },
    {
      "epoch": 2.162933349609375e-05,
      "grad_norm": 0.17139670252799988,
      "learning_rate": 9.169571066553091e-06,
      "loss": 0.0634,
      "step": 22680
    },
    {
      "epoch": 2.162933349609375e-05,
      "model_forward_time": 1.714036464691162,
      "step": 22680
    },
    {
      "dynamics_loss_avg": 0.059601444937288764,
      "epoch": 2.162933349609375e-05,
      "step": 22680
    },
    {
      "action_loss_avg": 0.0025998306227847934,
      "epoch": 2.162933349609375e-05,
      "step": 22680
    },
    {
      "epoch": 2.162933349609375e-05,
      "step": 22680,
      "training_step_time": 7.846884489059448
    },
    {
      "epoch": 2.1630287170410157e-05,
      "model_forward_time": 1.7112081050872803,
      "step": 22681
    },
    {
      "epoch": 2.1630287170410157e-05,
      "step": 22681,
      "training_step_time": 8.291812419891357
    },
    {
      "epoch": 2.163124084472656e-05,
      "model_forward_time": 1.7038671970367432,
      "step": 22682
    },
    {
      "epoch": 2.163124084472656e-05,
      "step": 22682,
      "training_step_time": 7.81035041809082
    },
    {
      "epoch": 2.163219451904297e-05,
      "model_forward_time": 1.733881950378418,
      "step": 22683
    },
    {
      "epoch": 2.163219451904297e-05,
      "step": 22683,
      "training_step_time": 8.69086217880249
    },
    {
      "epoch": 2.1633148193359376e-05,
      "model_forward_time": 2.252758264541626,
      "step": 22684
    },
    {
      "epoch": 2.1633148193359376e-05,
      "step": 22684,
      "training_step_time": 8.471677780151367
    },
    {
      "epoch": 2.163410186767578e-05,
      "model_forward_time": 1.7130224704742432,
      "step": 22685
    },
    {
      "epoch": 2.163410186767578e-05,
      "step": 22685,
      "training_step_time": 8.179203748703003
    },
    {
      "epoch": 2.1635055541992187e-05,
      "model_forward_time": 1.7096107006072998,
      "step": 22686
    },
    {
      "epoch": 2.1635055541992187e-05,
      "step": 22686,
      "training_step_time": 7.638908624649048
    },
    {
      "epoch": 2.1636009216308595e-05,
      "model_forward_time": 1.7100498676300049,
      "step": 22687
    },
    {
      "epoch": 2.1636009216308595e-05,
      "step": 22687,
      "training_step_time": 6.898552179336548
    },
    {
      "epoch": 2.1636962890625e-05,
      "model_forward_time": 1.7201058864593506,
      "step": 22688
    },
    {
      "epoch": 2.1636962890625e-05,
      "step": 22688,
      "training_step_time": 6.885148048400879
    },
    {
      "epoch": 2.1637916564941406e-05,
      "model_forward_time": 1.7122199535369873,
      "step": 22689
    },
    {
      "epoch": 2.1637916564941406e-05,
      "step": 22689,
      "training_step_time": 7.946835279464722
    },
    {
      "epoch": 2.1638870239257814e-05,
      "grad_norm": 0.17536474764347076,
      "learning_rate": 9.168658298362947e-06,
      "loss": 0.0653,
      "step": 22690
    },
    {
      "epoch": 2.1638870239257814e-05,
      "model_forward_time": 1.7011744976043701,
      "step": 22690
    },
    {
      "dynamics_loss_avg": 0.05637634629383683,
      "epoch": 2.1638870239257814e-05,
      "step": 22690
    },
    {
      "action_loss_avg": 0.002169640373904258,
      "epoch": 2.1638870239257814e-05,
      "step": 22690
    },
    {
      "epoch": 2.1638870239257814e-05,
      "step": 22690,
      "training_step_time": 7.549423933029175
    },
    {
      "epoch": 2.1639823913574218e-05,
      "model_forward_time": 1.7163424491882324,
      "step": 22691
    },
    {
      "epoch": 2.1639823913574218e-05,
      "step": 22691,
      "training_step_time": 8.563069105148315
    },
    {
      "epoch": 2.1640777587890625e-05,
      "model_forward_time": 1.715606689453125,
      "step": 22692
    },
    {
      "epoch": 2.1640777587890625e-05,
      "step": 22692,
      "training_step_time": 10.165400743484497
    },
    {
      "epoch": 2.1641731262207033e-05,
      "model_forward_time": 2.2691524028778076,
      "step": 22693
    },
    {
      "epoch": 2.1641731262207033e-05,
      "step": 22693,
      "training_step_time": 7.706074476242065
    },
    {
      "epoch": 2.1642684936523437e-05,
      "model_forward_time": 1.7068088054656982,
      "step": 22694
    },
    {
      "epoch": 2.1642684936523437e-05,
      "step": 22694,
      "training_step_time": 7.587648391723633
    },
    {
      "epoch": 2.1643638610839844e-05,
      "model_forward_time": 1.706010341644287,
      "step": 22695
    },
    {
      "epoch": 2.1643638610839844e-05,
      "step": 22695,
      "training_step_time": 7.532463550567627
    },
    {
      "epoch": 2.164459228515625e-05,
      "model_forward_time": 1.7159416675567627,
      "step": 22696
    },
    {
      "epoch": 2.164459228515625e-05,
      "step": 22696,
      "training_step_time": 6.731720209121704
    },
    {
      "epoch": 2.1645545959472655e-05,
      "model_forward_time": 1.7144618034362793,
      "step": 22697
    },
    {
      "epoch": 2.1645545959472655e-05,
      "step": 22697,
      "training_step_time": 6.819234371185303
    },
    {
      "epoch": 2.1646499633789063e-05,
      "model_forward_time": 1.7082772254943848,
      "step": 22698
    },
    {
      "epoch": 2.1646499633789063e-05,
      "step": 22698,
      "training_step_time": 7.788922071456909
    },
    {
      "epoch": 2.164745330810547e-05,
      "model_forward_time": 1.7131662368774414,
      "step": 22699
    },
    {
      "epoch": 2.164745330810547e-05,
      "step": 22699,
      "training_step_time": 7.664356470108032
    },
    {
      "epoch": 2.1648406982421874e-05,
      "grad_norm": 0.09723064303398132,
      "learning_rate": 9.167745074294597e-06,
      "loss": 0.061,
      "step": 22700
    },
    {
      "epoch": 2.1648406982421874e-05,
      "model_forward_time": 1.7073862552642822,
      "step": 22700
    },
    {
      "dynamics_loss_avg": 0.06053965855389833,
      "epoch": 2.1648406982421874e-05,
      "step": 22700
    },
    {
      "action_loss_avg": 0.0016551447450183332,
      "epoch": 2.1648406982421874e-05,
      "step": 22700
    },
    {
      "epoch": 2.1648406982421874e-05,
      "step": 22700,
      "training_step_time": 9.232025384902954
    },
    {
      "epoch": 2.1649360656738282e-05,
      "model_forward_time": 1.7073051929473877,
      "step": 22701
    },
    {
      "epoch": 2.1649360656738282e-05,
      "step": 22701,
      "training_step_time": 9.591272354125977
    },
    {
      "epoch": 2.165031433105469e-05,
      "model_forward_time": 1.7569384574890137,
      "step": 22702
    },
    {
      "epoch": 2.165031433105469e-05,
      "step": 22702,
      "training_step_time": 8.230468273162842
    },
    {
      "epoch": 2.1651268005371093e-05,
      "model_forward_time": 1.7133288383483887,
      "step": 22703
    },
    {
      "epoch": 2.1651268005371093e-05,
      "step": 22703,
      "training_step_time": 7.6482508182525635
    },
    {
      "epoch": 2.16522216796875e-05,
      "model_forward_time": 1.7185213565826416,
      "step": 22704
    },
    {
      "epoch": 2.16522216796875e-05,
      "step": 22704,
      "training_step_time": 8.071726560592651
    },
    {
      "epoch": 2.1653175354003905e-05,
      "model_forward_time": 1.7108018398284912,
      "step": 22705
    },
    {
      "epoch": 2.1653175354003905e-05,
      "step": 22705,
      "training_step_time": 7.405707359313965
    },
    {
      "epoch": 2.1654129028320312e-05,
      "model_forward_time": 1.7057507038116455,
      "step": 22706
    },
    {
      "epoch": 2.1654129028320312e-05,
      "step": 22706,
      "training_step_time": 6.791090488433838
    },
    {
      "epoch": 2.165508270263672e-05,
      "model_forward_time": 1.711496114730835,
      "step": 22707
    },
    {
      "epoch": 2.165508270263672e-05,
      "step": 22707,
      "training_step_time": 8.751371383666992
    },
    {
      "epoch": 2.1656036376953123e-05,
      "model_forward_time": 1.3298206329345703,
      "step": 22708
    },
    {
      "epoch": 2.1656036376953123e-05,
      "step": 22708,
      "training_step_time": 7.651976108551025
    },
    {
      "epoch": 2.165699005126953e-05,
      "model_forward_time": 1.7090327739715576,
      "step": 22709
    },
    {
      "epoch": 2.165699005126953e-05,
      "step": 22709,
      "training_step_time": 8.437319993972778
    },
    {
      "epoch": 2.165794372558594e-05,
      "grad_norm": 0.10308724641799927,
      "learning_rate": 9.166831394447912e-06,
      "loss": 0.063,
      "step": 22710
    },
    {
      "epoch": 2.165794372558594e-05,
      "model_forward_time": 1.7024335861206055,
      "step": 22710
    },
    {
      "dynamics_loss_avg": 0.05713648926466704,
      "epoch": 2.165794372558594e-05,
      "step": 22710
    },
    {
      "action_loss_avg": 0.002277296391548589,
      "epoch": 2.165794372558594e-05,
      "step": 22710
    },
    {
      "epoch": 2.165794372558594e-05,
      "step": 22710,
      "training_step_time": 7.639875650405884
    },
    {
      "epoch": 2.1658897399902342e-05,
      "model_forward_time": 1.759075403213501,
      "step": 22711
    },
    {
      "epoch": 2.1658897399902342e-05,
      "step": 22711,
      "training_step_time": 7.720557451248169
    },
    {
      "epoch": 2.165985107421875e-05,
      "model_forward_time": 1.5098092555999756,
      "step": 22712
    },
    {
      "epoch": 2.165985107421875e-05,
      "step": 22712,
      "training_step_time": 7.64266037940979
    },
    {
      "epoch": 2.1660804748535157e-05,
      "model_forward_time": 1.7180118560791016,
      "step": 22713
    },
    {
      "epoch": 2.1660804748535157e-05,
      "step": 22713,
      "training_step_time": 8.237990617752075
    },
    {
      "epoch": 2.166175842285156e-05,
      "model_forward_time": 1.7039287090301514,
      "step": 22714
    },
    {
      "epoch": 2.166175842285156e-05,
      "step": 22714,
      "training_step_time": 7.597259283065796
    },
    {
      "epoch": 2.166271209716797e-05,
      "model_forward_time": 1.705942153930664,
      "step": 22715
    },
    {
      "epoch": 2.166271209716797e-05,
      "step": 22715,
      "training_step_time": 6.777018308639526
    },
    {
      "epoch": 2.1663665771484376e-05,
      "model_forward_time": 1.3366506099700928,
      "step": 22716
    },
    {
      "epoch": 2.1663665771484376e-05,
      "step": 22716,
      "training_step_time": 8.762901544570923
    },
    {
      "epoch": 2.166461944580078e-05,
      "model_forward_time": 1.7103774547576904,
      "step": 22717
    },
    {
      "epoch": 2.166461944580078e-05,
      "step": 22717,
      "training_step_time": 7.702473402023315
    },
    {
      "epoch": 2.1665573120117187e-05,
      "model_forward_time": 1.7084310054779053,
      "step": 22718
    },
    {
      "epoch": 2.1665573120117187e-05,
      "step": 22718,
      "training_step_time": 8.652035236358643
    },
    {
      "epoch": 2.1666526794433595e-05,
      "model_forward_time": 1.7092432975769043,
      "step": 22719
    },
    {
      "epoch": 2.1666526794433595e-05,
      "step": 22719,
      "training_step_time": 8.262865781784058
    },
    {
      "epoch": 2.166748046875e-05,
      "grad_norm": 0.30511194467544556,
      "learning_rate": 9.165917258922811e-06,
      "loss": 0.0608,
      "step": 22720
    },
    {
      "epoch": 2.166748046875e-05,
      "model_forward_time": 1.7528424263000488,
      "step": 22720
    },
    {
      "dynamics_loss_avg": 0.06996611468493938,
      "epoch": 2.166748046875e-05,
      "step": 22720
    },
    {
      "action_loss_avg": 0.0036839505133684724,
      "epoch": 2.166748046875e-05,
      "step": 22720
    },
    {
      "epoch": 2.166748046875e-05,
      "step": 22720,
      "training_step_time": 7.643941640853882
    },
    {
      "epoch": 2.1668434143066406e-05,
      "model_forward_time": 2.2509665489196777,
      "step": 22721
    },
    {
      "epoch": 2.1668434143066406e-05,
      "step": 22721,
      "training_step_time": 7.644012689590454
    },
    {
      "epoch": 2.1669387817382814e-05,
      "model_forward_time": 1.7094192504882812,
      "step": 22722
    },
    {
      "epoch": 2.1669387817382814e-05,
      "step": 22722,
      "training_step_time": 8.681452512741089
    },
    {
      "epoch": 2.1670341491699218e-05,
      "model_forward_time": 1.7028636932373047,
      "step": 22723
    },
    {
      "epoch": 2.1670341491699218e-05,
      "step": 22723,
      "training_step_time": 7.551420211791992
    },
    {
      "epoch": 2.1671295166015625e-05,
      "model_forward_time": 1.7133798599243164,
      "step": 22724
    },
    {
      "epoch": 2.1671295166015625e-05,
      "step": 22724,
      "training_step_time": 6.730351686477661
    },
    {
      "epoch": 2.1672248840332033e-05,
      "model_forward_time": 0.6735079288482666,
      "step": 22725
    },
    {
      "epoch": 2.1672248840332033e-05,
      "step": 22725,
      "training_step_time": 7.906142950057983
    },
    {
      "epoch": 2.1673202514648437e-05,
      "model_forward_time": 1.7134897708892822,
      "step": 22726
    },
    {
      "epoch": 2.1673202514648437e-05,
      "step": 22726,
      "training_step_time": 7.705960512161255
    },
    {
      "epoch": 2.1674156188964844e-05,
      "model_forward_time": 1.7133736610412598,
      "step": 22727
    },
    {
      "epoch": 2.1674156188964844e-05,
      "step": 22727,
      "training_step_time": 8.35205864906311
    },
    {
      "epoch": 2.167510986328125e-05,
      "model_forward_time": 1.7064728736877441,
      "step": 22728
    },
    {
      "epoch": 2.167510986328125e-05,
      "step": 22728,
      "training_step_time": 10.480151414871216
    },
    {
      "epoch": 2.1676063537597655e-05,
      "model_forward_time": 1.7439136505126953,
      "step": 22729
    },
    {
      "epoch": 2.1676063537597655e-05,
      "step": 22729,
      "training_step_time": 8.435606718063354
    },
    {
      "epoch": 2.1677017211914063e-05,
      "grad_norm": 0.2013174444437027,
      "learning_rate": 9.165002667819263e-06,
      "loss": 0.066,
      "step": 22730
    },
    {
      "epoch": 2.1677017211914063e-05,
      "model_forward_time": 1.715576171875,
      "step": 22730
    },
    {
      "dynamics_loss_avg": 0.08038307260721922,
      "epoch": 2.1677017211914063e-05,
      "step": 22730
    },
    {
      "action_loss_avg": 0.002981227863347158,
      "epoch": 2.1677017211914063e-05,
      "step": 22730
    },
    {
      "epoch": 2.1677017211914063e-05,
      "step": 22730,
      "training_step_time": 9.20539903640747
    },
    {
      "epoch": 2.167797088623047e-05,
      "model_forward_time": 2.2561399936676025,
      "step": 22731
    },
    {
      "epoch": 2.167797088623047e-05,
      "step": 22731,
      "training_step_time": 7.767091274261475
    },
    {
      "epoch": 2.1678924560546874e-05,
      "model_forward_time": 1.710376501083374,
      "step": 22732
    },
    {
      "epoch": 2.1678924560546874e-05,
      "step": 22732,
      "training_step_time": 7.5951151847839355
    },
    {
      "epoch": 2.1679878234863282e-05,
      "model_forward_time": 1.71793794631958,
      "step": 22733
    },
    {
      "epoch": 2.1679878234863282e-05,
      "step": 22733,
      "training_step_time": 6.845762252807617
    },
    {
      "epoch": 2.168083190917969e-05,
      "model_forward_time": 1.7292044162750244,
      "step": 22734
    },
    {
      "epoch": 2.168083190917969e-05,
      "step": 22734,
      "training_step_time": 8.647777080535889
    },
    {
      "epoch": 2.1681785583496093e-05,
      "model_forward_time": 1.6932036876678467,
      "step": 22735
    },
    {
      "epoch": 2.1681785583496093e-05,
      "step": 22735,
      "training_step_time": 7.780112981796265
    },
    {
      "epoch": 2.16827392578125e-05,
      "model_forward_time": 1.6969153881072998,
      "step": 22736
    },
    {
      "epoch": 2.16827392578125e-05,
      "step": 22736,
      "training_step_time": 8.48065972328186
    },
    {
      "epoch": 2.1683692932128905e-05,
      "model_forward_time": 1.6931660175323486,
      "step": 22737
    },
    {
      "epoch": 2.1683692932128905e-05,
      "step": 22737,
      "training_step_time": 8.710397958755493
    },
    {
      "epoch": 2.1684646606445312e-05,
      "model_forward_time": 1.6987802982330322,
      "step": 22738
    },
    {
      "epoch": 2.1684646606445312e-05,
      "step": 22738,
      "training_step_time": 9.26583456993103
    },
    {
      "epoch": 2.168560028076172e-05,
      "model_forward_time": 1.6868727207183838,
      "step": 22739
    },
    {
      "epoch": 2.168560028076172e-05,
      "step": 22739,
      "training_step_time": 7.802050828933716
    },
    {
      "epoch": 2.1686553955078124e-05,
      "grad_norm": 0.14148548245429993,
      "learning_rate": 9.164087621237283e-06,
      "loss": 0.0609,
      "step": 22740
    },
    {
      "epoch": 2.1686553955078124e-05,
      "model_forward_time": 1.3094236850738525,
      "step": 22740
    },
    {
      "dynamics_loss_avg": 0.05329139549285174,
      "epoch": 2.1686553955078124e-05,
      "step": 22740
    },
    {
      "action_loss_avg": 0.0029092524346197024,
      "epoch": 2.1686553955078124e-05,
      "step": 22740
    },
    {
      "epoch": 2.1686553955078124e-05,
      "step": 22740,
      "training_step_time": 7.579521179199219
    },
    {
      "epoch": 2.168750762939453e-05,
      "model_forward_time": 2.23463773727417,
      "step": 22741
    },
    {
      "epoch": 2.168750762939453e-05,
      "step": 22741,
      "training_step_time": 7.773716926574707
    },
    {
      "epoch": 2.168846130371094e-05,
      "model_forward_time": 1.690591812133789,
      "step": 22742
    },
    {
      "epoch": 2.168846130371094e-05,
      "step": 22742,
      "training_step_time": 6.81853461265564
    },
    {
      "epoch": 2.1689414978027342e-05,
      "model_forward_time": 1.6899223327636719,
      "step": 22743
    },
    {
      "epoch": 2.1689414978027342e-05,
      "step": 22743,
      "training_step_time": 7.11452317237854
    },
    {
      "epoch": 2.169036865234375e-05,
      "model_forward_time": 1.7048635482788086,
      "step": 22744
    },
    {
      "epoch": 2.169036865234375e-05,
      "step": 22744,
      "training_step_time": 7.921727895736694
    },
    {
      "epoch": 2.1691322326660157e-05,
      "model_forward_time": 1.6995718479156494,
      "step": 22745
    },
    {
      "epoch": 2.1691322326660157e-05,
      "step": 22745,
      "training_step_time": 7.853301525115967
    },
    {
      "epoch": 2.169227600097656e-05,
      "model_forward_time": 1.697366714477539,
      "step": 22746
    },
    {
      "epoch": 2.169227600097656e-05,
      "step": 22746,
      "training_step_time": 8.572861671447754
    },
    {
      "epoch": 2.169322967529297e-05,
      "model_forward_time": 1.698249101638794,
      "step": 22747
    },
    {
      "epoch": 2.169322967529297e-05,
      "step": 22747,
      "training_step_time": 8.378440141677856
    },
    {
      "epoch": 2.1694183349609376e-05,
      "model_forward_time": 1.7297141551971436,
      "step": 22748
    },
    {
      "epoch": 2.1694183349609376e-05,
      "step": 22748,
      "training_step_time": 8.739245653152466
    },
    {
      "epoch": 2.169513702392578e-05,
      "model_forward_time": 1.6996254920959473,
      "step": 22749
    },
    {
      "epoch": 2.169513702392578e-05,
      "step": 22749,
      "training_step_time": 7.807443141937256
    },
    {
      "epoch": 2.1696090698242188e-05,
      "grad_norm": 0.16414858400821686,
      "learning_rate": 9.163172119276942e-06,
      "loss": 0.0623,
      "step": 22750
    },
    {
      "epoch": 2.1696090698242188e-05,
      "model_forward_time": 2.2218220233917236,
      "step": 22750
    },
    {
      "dynamics_loss_avg": 0.06439176071435213,
      "epoch": 2.1696090698242188e-05,
      "step": 22750
    },
    {
      "action_loss_avg": 0.0020154503028606994,
      "epoch": 2.1696090698242188e-05,
      "step": 22750
    },
    {
      "epoch": 2.1696090698242188e-05,
      "step": 22750,
      "training_step_time": 8.772238731384277
    },
    {
      "epoch": 2.1697044372558595e-05,
      "model_forward_time": 1.7058584690093994,
      "step": 22751
    },
    {
      "epoch": 2.1697044372558595e-05,
      "step": 22751,
      "training_step_time": 7.75446629524231
    },
    {
      "epoch": 2.1697998046875e-05,
      "model_forward_time": 1.6938560009002686,
      "step": 22752
    },
    {
      "epoch": 2.1697998046875e-05,
      "step": 22752,
      "training_step_time": 7.248464345932007
    },
    {
      "epoch": 2.1698951721191406e-05,
      "model_forward_time": 1.6845636367797852,
      "step": 22753
    },
    {
      "epoch": 2.1698951721191406e-05,
      "step": 22753,
      "training_step_time": 8.810489177703857
    },
    {
      "epoch": 2.1699905395507814e-05,
      "model_forward_time": 1.6935646533966064,
      "step": 22754
    },
    {
      "epoch": 2.1699905395507814e-05,
      "step": 22754,
      "training_step_time": 8.709258317947388
    },
    {
      "epoch": 2.1700859069824218e-05,
      "model_forward_time": 1.6983842849731445,
      "step": 22755
    },
    {
      "epoch": 2.1700859069824218e-05,
      "step": 22755,
      "training_step_time": 8.997742652893066
    },
    {
      "epoch": 2.1701812744140625e-05,
      "model_forward_time": 1.7021539211273193,
      "step": 22756
    },
    {
      "epoch": 2.1701812744140625e-05,
      "step": 22756,
      "training_step_time": 8.567098140716553
    },
    {
      "epoch": 2.1702766418457033e-05,
      "model_forward_time": 1.7381234169006348,
      "step": 22757
    },
    {
      "epoch": 2.1702766418457033e-05,
      "step": 22757,
      "training_step_time": 7.8729777336120605
    },
    {
      "epoch": 2.1703720092773437e-05,
      "model_forward_time": 1.7037816047668457,
      "step": 22758
    },
    {
      "epoch": 2.1703720092773437e-05,
      "step": 22758,
      "training_step_time": 7.783118009567261
    },
    {
      "epoch": 2.1704673767089844e-05,
      "model_forward_time": 2.220459222793579,
      "step": 22759
    },
    {
      "epoch": 2.1704673767089844e-05,
      "step": 22759,
      "training_step_time": 7.94966459274292
    },
    {
      "epoch": 2.170562744140625e-05,
      "grad_norm": 0.1932062953710556,
      "learning_rate": 9.162256162038358e-06,
      "loss": 0.0621,
      "step": 22760
    },
    {
      "epoch": 2.170562744140625e-05,
      "model_forward_time": 1.682889461517334,
      "step": 22760
    },
    {
      "dynamics_loss_avg": 0.0560540284961462,
      "epoch": 2.170562744140625e-05,
      "step": 22760
    },
    {
      "action_loss_avg": 0.0026775305334012954,
      "epoch": 2.170562744140625e-05,
      "step": 22760
    },
    {
      "epoch": 2.170562744140625e-05,
      "step": 22760,
      "training_step_time": 7.7742815017700195
    },
    {
      "epoch": 2.1706581115722656e-05,
      "model_forward_time": 1.691399097442627,
      "step": 22761
    },
    {
      "epoch": 2.1706581115722656e-05,
      "step": 22761,
      "training_step_time": 7.270367860794067
    },
    {
      "epoch": 2.1707534790039063e-05,
      "model_forward_time": 1.7056834697723389,
      "step": 22762
    },
    {
      "epoch": 2.1707534790039063e-05,
      "step": 22762,
      "training_step_time": 8.117977857589722
    },
    {
      "epoch": 2.170848846435547e-05,
      "model_forward_time": 1.3156147003173828,
      "step": 22763
    },
    {
      "epoch": 2.170848846435547e-05,
      "step": 22763,
      "training_step_time": 7.971968412399292
    },
    {
      "epoch": 2.1709442138671874e-05,
      "model_forward_time": 1.7038214206695557,
      "step": 22764
    },
    {
      "epoch": 2.1709442138671874e-05,
      "step": 22764,
      "training_step_time": 7.77812385559082
    },
    {
      "epoch": 2.1710395812988282e-05,
      "model_forward_time": 1.7004454135894775,
      "step": 22765
    },
    {
      "epoch": 2.1710395812988282e-05,
      "step": 22765,
      "training_step_time": 9.360793590545654
    },
    {
      "epoch": 2.1711349487304686e-05,
      "model_forward_time": 1.7499146461486816,
      "step": 22766
    },
    {
      "epoch": 2.1711349487304686e-05,
      "step": 22766,
      "training_step_time": 7.903299570083618
    },
    {
      "epoch": 2.1712303161621093e-05,
      "model_forward_time": 1.69803786277771,
      "step": 22767
    },
    {
      "epoch": 2.1712303161621093e-05,
      "step": 22767,
      "training_step_time": 8.569620609283447
    },
    {
      "epoch": 2.17132568359375e-05,
      "model_forward_time": 2.2672059535980225,
      "step": 22768
    },
    {
      "epoch": 2.17132568359375e-05,
      "step": 22768,
      "training_step_time": 8.059727430343628
    },
    {
      "epoch": 2.1714210510253905e-05,
      "model_forward_time": 1.7028770446777344,
      "step": 22769
    },
    {
      "epoch": 2.1714210510253905e-05,
      "step": 22769,
      "training_step_time": 7.849677801132202
    },
    {
      "epoch": 2.1715164184570312e-05,
      "grad_norm": 0.21421538293361664,
      "learning_rate": 9.161339749621698e-06,
      "loss": 0.0659,
      "step": 22770
    },
    {
      "epoch": 2.1715164184570312e-05,
      "model_forward_time": 1.7010080814361572,
      "step": 22770
    },
    {
      "dynamics_loss_avg": 0.04850731659680605,
      "epoch": 2.1715164184570312e-05,
      "step": 22770
    },
    {
      "action_loss_avg": 0.001757205932517536,
      "epoch": 2.1715164184570312e-05,
      "step": 22770
    },
    {
      "epoch": 2.1715164184570312e-05,
      "step": 22770,
      "training_step_time": 7.314244031906128
    },
    {
      "epoch": 2.171611785888672e-05,
      "model_forward_time": 1.3274142742156982,
      "step": 22771
    },
    {
      "epoch": 2.171611785888672e-05,
      "step": 22771,
      "training_step_time": 9.544360876083374
    },
    {
      "epoch": 2.1717071533203124e-05,
      "model_forward_time": 1.7121293544769287,
      "step": 22772
    },
    {
      "epoch": 2.1717071533203124e-05,
      "step": 22772,
      "training_step_time": 7.784054279327393
    },
    {
      "epoch": 2.171802520751953e-05,
      "model_forward_time": 1.7113728523254395,
      "step": 22773
    },
    {
      "epoch": 2.171802520751953e-05,
      "step": 22773,
      "training_step_time": 8.265218496322632
    },
    {
      "epoch": 2.171897888183594e-05,
      "model_forward_time": 1.733574390411377,
      "step": 22774
    },
    {
      "epoch": 2.171897888183594e-05,
      "step": 22774,
      "training_step_time": 7.878320693969727
    },
    {
      "epoch": 2.1719932556152342e-05,
      "model_forward_time": 1.711961030960083,
      "step": 22775
    },
    {
      "epoch": 2.1719932556152342e-05,
      "step": 22775,
      "training_step_time": 10.708810567855835
    },
    {
      "epoch": 2.172088623046875e-05,
      "model_forward_time": 1.7127466201782227,
      "step": 22776
    },
    {
      "epoch": 2.172088623046875e-05,
      "step": 22776,
      "training_step_time": 7.892250299453735
    },
    {
      "epoch": 2.1721839904785157e-05,
      "model_forward_time": 1.704763650894165,
      "step": 22777
    },
    {
      "epoch": 2.1721839904785157e-05,
      "step": 22777,
      "training_step_time": 7.819667100906372
    },
    {
      "epoch": 2.172279357910156e-05,
      "model_forward_time": 2.256110668182373,
      "step": 22778
    },
    {
      "epoch": 2.172279357910156e-05,
      "step": 22778,
      "training_step_time": 7.815686464309692
    },
    {
      "epoch": 2.172374725341797e-05,
      "model_forward_time": 1.7105987071990967,
      "step": 22779
    },
    {
      "epoch": 2.172374725341797e-05,
      "step": 22779,
      "training_step_time": 7.796382904052734
    },
    {
      "epoch": 2.1724700927734376e-05,
      "grad_norm": 0.21182236075401306,
      "learning_rate": 9.160422882127177e-06,
      "loss": 0.0636,
      "step": 22780
    },
    {
      "epoch": 2.1724700927734376e-05,
      "model_forward_time": 1.7154035568237305,
      "step": 22780
    },
    {
      "dynamics_loss_avg": 0.054969477280974385,
      "epoch": 2.1724700927734376e-05,
      "step": 22780
    },
    {
      "action_loss_avg": 0.002969211235176772,
      "epoch": 2.1724700927734376e-05,
      "step": 22780
    },
    {
      "epoch": 2.1724700927734376e-05,
      "step": 22780,
      "training_step_time": 8.06562066078186
    },
    {
      "epoch": 2.172565460205078e-05,
      "model_forward_time": 1.71323561668396,
      "step": 22781
    },
    {
      "epoch": 2.172565460205078e-05,
      "step": 22781,
      "training_step_time": 9.509333610534668
    },
    {
      "epoch": 2.1726608276367188e-05,
      "model_forward_time": 1.723555088043213,
      "step": 22782
    },
    {
      "epoch": 2.1726608276367188e-05,
      "step": 22782,
      "training_step_time": 8.973817586898804
    },
    {
      "epoch": 2.1727561950683595e-05,
      "model_forward_time": 1.7105903625488281,
      "step": 22783
    },
    {
      "epoch": 2.1727561950683595e-05,
      "step": 22783,
      "training_step_time": 8.887994527816772
    },
    {
      "epoch": 2.1728515625e-05,
      "model_forward_time": 1.7155811786651611,
      "step": 22784
    },
    {
      "epoch": 2.1728515625e-05,
      "step": 22784,
      "training_step_time": 9.620691299438477
    },
    {
      "epoch": 2.1729469299316406e-05,
      "model_forward_time": 1.7618248462677002,
      "step": 22785
    },
    {
      "epoch": 2.1729469299316406e-05,
      "step": 22785,
      "training_step_time": 7.935634136199951
    },
    {
      "epoch": 2.1730422973632814e-05,
      "model_forward_time": 1.709979772567749,
      "step": 22786
    },
    {
      "epoch": 2.1730422973632814e-05,
      "step": 22786,
      "training_step_time": 7.9020490646362305
    },
    {
      "epoch": 2.1731376647949218e-05,
      "model_forward_time": 2.2432761192321777,
      "step": 22787
    },
    {
      "epoch": 2.1731376647949218e-05,
      "step": 22787,
      "training_step_time": 7.991589307785034
    },
    {
      "epoch": 2.1732330322265625e-05,
      "model_forward_time": 1.7047114372253418,
      "step": 22788
    },
    {
      "epoch": 2.1732330322265625e-05,
      "step": 22788,
      "training_step_time": 7.8598902225494385
    },
    {
      "epoch": 2.1733283996582033e-05,
      "model_forward_time": 1.707287073135376,
      "step": 22789
    },
    {
      "epoch": 2.1733283996582033e-05,
      "step": 22789,
      "training_step_time": 9.557729244232178
    },
    {
      "epoch": 2.1734237670898437e-05,
      "grad_norm": 0.22912073135375977,
      "learning_rate": 9.159505559655068e-06,
      "loss": 0.0615,
      "step": 22790
    },
    {
      "epoch": 2.1734237670898437e-05,
      "model_forward_time": 1.7008538246154785,
      "step": 22790
    },
    {
      "dynamics_loss_avg": 0.03984181880950928,
      "epoch": 2.1734237670898437e-05,
      "step": 22790
    },
    {
      "action_loss_avg": 0.0014517974166665226,
      "epoch": 2.1734237670898437e-05,
      "step": 22790
    },
    {
      "epoch": 2.1734237670898437e-05,
      "step": 22790,
      "training_step_time": 8.108048677444458
    },
    {
      "epoch": 2.1735191345214844e-05,
      "model_forward_time": 1.701756477355957,
      "step": 22791
    },
    {
      "epoch": 2.1735191345214844e-05,
      "step": 22791,
      "training_step_time": 8.743252277374268
    },
    {
      "epoch": 2.173614501953125e-05,
      "model_forward_time": 1.7062532901763916,
      "step": 22792
    },
    {
      "epoch": 2.173614501953125e-05,
      "step": 22792,
      "training_step_time": 8.041072607040405
    },
    {
      "epoch": 2.1737098693847656e-05,
      "model_forward_time": 1.7042672634124756,
      "step": 22793
    },
    {
      "epoch": 2.1737098693847656e-05,
      "step": 22793,
      "training_step_time": 8.834095478057861
    },
    {
      "epoch": 2.1738052368164063e-05,
      "model_forward_time": 1.7494962215423584,
      "step": 22794
    },
    {
      "epoch": 2.1738052368164063e-05,
      "step": 22794,
      "training_step_time": 8.869510889053345
    },
    {
      "epoch": 2.173900604248047e-05,
      "model_forward_time": 1.7023799419403076,
      "step": 22795
    },
    {
      "epoch": 2.173900604248047e-05,
      "step": 22795,
      "training_step_time": 8.218034029006958
    },
    {
      "epoch": 2.1739959716796874e-05,
      "model_forward_time": 2.242806911468506,
      "step": 22796
    },
    {
      "epoch": 2.1739959716796874e-05,
      "step": 22796,
      "training_step_time": 7.766300439834595
    },
    {
      "epoch": 2.1740913391113282e-05,
      "model_forward_time": 1.3284063339233398,
      "step": 22797
    },
    {
      "epoch": 2.1740913391113282e-05,
      "step": 22797,
      "training_step_time": 7.774134397506714
    },
    {
      "epoch": 2.1741867065429686e-05,
      "model_forward_time": 1.6973872184753418,
      "step": 22798
    },
    {
      "epoch": 2.1741867065429686e-05,
      "step": 22798,
      "training_step_time": 7.8128650188446045
    },
    {
      "epoch": 2.1742820739746093e-05,
      "model_forward_time": 1.3214154243469238,
      "step": 22799
    },
    {
      "epoch": 2.1742820739746093e-05,
      "step": 22799,
      "training_step_time": 8.16710376739502
    },
    {
      "epoch": 2.17437744140625e-05,
      "grad_norm": 0.116899773478508,
      "learning_rate": 9.158587782305683e-06,
      "loss": 0.0632,
      "step": 22800
    },
    {
      "epoch": 2.17437744140625e-05,
      "model_forward_time": 1.6988105773925781,
      "step": 22800
    },
    {
      "dynamics_loss_avg": 0.05198708306998014,
      "epoch": 2.17437744140625e-05,
      "step": 22800
    },
    {
      "action_loss_avg": 0.002266369294375181,
      "epoch": 2.17437744140625e-05,
      "step": 22800
    },
    {
      "epoch": 2.17437744140625e-05,
      "step": 22800,
      "training_step_time": 9.065085887908936
    },
    {
      "epoch": 2.1744728088378905e-05,
      "model_forward_time": 1.7017419338226318,
      "step": 22801
    },
    {
      "epoch": 2.1744728088378905e-05,
      "step": 22801,
      "training_step_time": 8.879477977752686
    },
    {
      "epoch": 2.1745681762695312e-05,
      "model_forward_time": 1.690417766571045,
      "step": 22802
    },
    {
      "epoch": 2.1745681762695312e-05,
      "step": 22802,
      "training_step_time": 9.846971035003662
    },
    {
      "epoch": 2.174663543701172e-05,
      "model_forward_time": 1.7542753219604492,
      "step": 22803
    },
    {
      "epoch": 2.174663543701172e-05,
      "step": 22803,
      "training_step_time": 7.9126763343811035
    },
    {
      "epoch": 2.1747589111328124e-05,
      "model_forward_time": 1.7040505409240723,
      "step": 22804
    },
    {
      "epoch": 2.1747589111328124e-05,
      "step": 22804,
      "training_step_time": 7.821786403656006
    },
    {
      "epoch": 2.174854278564453e-05,
      "model_forward_time": 2.2146594524383545,
      "step": 22805
    },
    {
      "epoch": 2.174854278564453e-05,
      "step": 22805,
      "training_step_time": 7.7742884159088135
    },
    {
      "epoch": 2.174949645996094e-05,
      "model_forward_time": 1.7148303985595703,
      "step": 22806
    },
    {
      "epoch": 2.174949645996094e-05,
      "step": 22806,
      "training_step_time": 7.785445690155029
    },
    {
      "epoch": 2.1750450134277342e-05,
      "model_forward_time": 1.6999170780181885,
      "step": 22807
    },
    {
      "epoch": 2.1750450134277342e-05,
      "step": 22807,
      "training_step_time": 7.764696359634399
    },
    {
      "epoch": 2.175140380859375e-05,
      "model_forward_time": 1.6993601322174072,
      "step": 22808
    },
    {
      "epoch": 2.175140380859375e-05,
      "step": 22808,
      "training_step_time": 8.160646915435791
    },
    {
      "epoch": 2.1752357482910157e-05,
      "model_forward_time": 1.707921028137207,
      "step": 22809
    },
    {
      "epoch": 2.1752357482910157e-05,
      "step": 22809,
      "training_step_time": 8.65759825706482
    },
    {
      "epoch": 2.175331115722656e-05,
      "grad_norm": 0.11901005357503891,
      "learning_rate": 9.157669550179393e-06,
      "loss": 0.0605,
      "step": 22810
    },
    {
      "epoch": 2.175331115722656e-05,
      "model_forward_time": 1.693382978439331,
      "step": 22810
    },
    {
      "dynamics_loss_avg": 0.048740978725254534,
      "epoch": 2.175331115722656e-05,
      "step": 22810
    },
    {
      "action_loss_avg": 0.003195076243719086,
      "epoch": 2.175331115722656e-05,
      "step": 22810
    },
    {
      "epoch": 2.175331115722656e-05,
      "step": 22810,
      "training_step_time": 8.102222919464111
    },
    {
      "epoch": 2.175426483154297e-05,
      "model_forward_time": 1.7126061916351318,
      "step": 22811
    },
    {
      "epoch": 2.175426483154297e-05,
      "step": 22811,
      "training_step_time": 10.345738887786865
    },
    {
      "epoch": 2.1755218505859376e-05,
      "model_forward_time": 1.725203275680542,
      "step": 22812
    },
    {
      "epoch": 2.1755218505859376e-05,
      "step": 22812,
      "training_step_time": 8.891343116760254
    },
    {
      "epoch": 2.175617218017578e-05,
      "model_forward_time": 1.6983511447906494,
      "step": 22813
    },
    {
      "epoch": 2.175617218017578e-05,
      "step": 22813,
      "training_step_time": 7.73912787437439
    },
    {
      "epoch": 2.1757125854492188e-05,
      "model_forward_time": 1.6892309188842773,
      "step": 22814
    },
    {
      "epoch": 2.1757125854492188e-05,
      "step": 22814,
      "training_step_time": 7.743785858154297
    },
    {
      "epoch": 2.1758079528808595e-05,
      "model_forward_time": 2.244704246520996,
      "step": 22815
    },
    {
      "epoch": 2.1758079528808595e-05,
      "step": 22815,
      "training_step_time": 7.342741250991821
    },
    {
      "epoch": 2.1759033203125e-05,
      "model_forward_time": 1.6943926811218262,
      "step": 22816
    },
    {
      "epoch": 2.1759033203125e-05,
      "step": 22816,
      "training_step_time": 7.712159156799316
    },
    {
      "epoch": 2.1759986877441406e-05,
      "model_forward_time": 1.6719202995300293,
      "step": 22817
    },
    {
      "epoch": 2.1759986877441406e-05,
      "step": 22817,
      "training_step_time": 8.060399770736694
    },
    {
      "epoch": 2.1760940551757814e-05,
      "model_forward_time": 1.6874587535858154,
      "step": 22818
    },
    {
      "epoch": 2.1760940551757814e-05,
      "step": 22818,
      "training_step_time": 8.45934009552002
    },
    {
      "epoch": 2.1761894226074218e-05,
      "model_forward_time": 1.6753735542297363,
      "step": 22819
    },
    {
      "epoch": 2.1761894226074218e-05,
      "step": 22819,
      "training_step_time": 10.102770328521729
    },
    {
      "epoch": 2.1762847900390625e-05,
      "grad_norm": 0.1365048736333847,
      "learning_rate": 9.156750863376609e-06,
      "loss": 0.0613,
      "step": 22820
    },
    {
      "epoch": 2.1762847900390625e-05,
      "model_forward_time": 1.684297800064087,
      "step": 22820
    },
    {
      "dynamics_loss_avg": 0.05688571659848094,
      "epoch": 2.1762847900390625e-05,
      "step": 22820
    },
    {
      "action_loss_avg": 0.0019384565704967827,
      "epoch": 2.1762847900390625e-05,
      "step": 22820
    },
    {
      "epoch": 2.1762847900390625e-05,
      "step": 22820,
      "training_step_time": 7.7465660572052
    },
    {
      "epoch": 2.1763801574707033e-05,
      "model_forward_time": 1.715778112411499,
      "step": 22821
    },
    {
      "epoch": 2.1763801574707033e-05,
      "step": 22821,
      "training_step_time": 7.8304853439331055
    },
    {
      "epoch": 2.1764755249023437e-05,
      "model_forward_time": 1.6846706867218018,
      "step": 22822
    },
    {
      "epoch": 2.1764755249023437e-05,
      "step": 22822,
      "training_step_time": 7.792007684707642
    },
    {
      "epoch": 2.1765708923339844e-05,
      "model_forward_time": 1.6856276988983154,
      "step": 22823
    },
    {
      "epoch": 2.1765708923339844e-05,
      "step": 22823,
      "training_step_time": 7.65686821937561
    },
    {
      "epoch": 2.176666259765625e-05,
      "model_forward_time": 1.8203787803649902,
      "step": 22824
    },
    {
      "epoch": 2.176666259765625e-05,
      "step": 22824,
      "training_step_time": 7.400225639343262
    },
    {
      "epoch": 2.1767616271972656e-05,
      "model_forward_time": 1.6842315196990967,
      "step": 22825
    },
    {
      "epoch": 2.1767616271972656e-05,
      "step": 22825,
      "training_step_time": 7.760299444198608
    },
    {
      "epoch": 2.1768569946289063e-05,
      "model_forward_time": 1.6916940212249756,
      "step": 22826
    },
    {
      "epoch": 2.1768569946289063e-05,
      "step": 22826,
      "training_step_time": 8.077161073684692
    },
    {
      "epoch": 2.176952362060547e-05,
      "model_forward_time": 1.6866462230682373,
      "step": 22827
    },
    {
      "epoch": 2.176952362060547e-05,
      "step": 22827,
      "training_step_time": 7.8318891525268555
    },
    {
      "epoch": 2.1770477294921874e-05,
      "model_forward_time": 1.683516263961792,
      "step": 22828
    },
    {
      "epoch": 2.1770477294921874e-05,
      "step": 22828,
      "training_step_time": 8.656702756881714
    },
    {
      "epoch": 2.1771430969238282e-05,
      "model_forward_time": 1.6815905570983887,
      "step": 22829
    },
    {
      "epoch": 2.1771430969238282e-05,
      "step": 22829,
      "training_step_time": 9.508427381515503
    },
    {
      "epoch": 2.1772384643554686e-05,
      "grad_norm": 0.10808756202459335,
      "learning_rate": 9.1558317219978e-06,
      "loss": 0.0627,
      "step": 22830
    },
    {
      "epoch": 2.1772384643554686e-05,
      "model_forward_time": 1.7188210487365723,
      "step": 22830
    },
    {
      "dynamics_loss_avg": 0.06286577396094799,
      "epoch": 2.1772384643554686e-05,
      "step": 22830
    },
    {
      "action_loss_avg": 0.0022721717716194688,
      "epoch": 2.1772384643554686e-05,
      "step": 22830
    },
    {
      "epoch": 2.1772384643554686e-05,
      "step": 22830,
      "training_step_time": 7.876920938491821
    },
    {
      "epoch": 2.1773338317871093e-05,
      "model_forward_time": 1.4088146686553955,
      "step": 22831
    },
    {
      "epoch": 2.1773338317871093e-05,
      "step": 22831,
      "training_step_time": 7.88387656211853
    },
    {
      "epoch": 2.17742919921875e-05,
      "model_forward_time": 1.7384569644927979,
      "step": 22832
    },
    {
      "epoch": 2.17742919921875e-05,
      "step": 22832,
      "training_step_time": 7.752094745635986
    },
    {
      "epoch": 2.1775245666503905e-05,
      "model_forward_time": 2.8679704666137695,
      "step": 22833
    },
    {
      "epoch": 2.1775245666503905e-05,
      "step": 22833,
      "training_step_time": 8.377942323684692
    },
    {
      "epoch": 2.1776199340820312e-05,
      "model_forward_time": 1.766955852508545,
      "step": 22834
    },
    {
      "epoch": 2.1776199340820312e-05,
      "step": 22834,
      "training_step_time": 7.771791934967041
    },
    {
      "epoch": 2.177715301513672e-05,
      "model_forward_time": 1.7649354934692383,
      "step": 22835
    },
    {
      "epoch": 2.177715301513672e-05,
      "step": 22835,
      "training_step_time": 7.846196889877319
    },
    {
      "epoch": 2.1778106689453124e-05,
      "model_forward_time": 1.757936954498291,
      "step": 22836
    },
    {
      "epoch": 2.1778106689453124e-05,
      "step": 22836,
      "training_step_time": 8.005867958068848
    },
    {
      "epoch": 2.177906036376953e-05,
      "model_forward_time": 1.818610668182373,
      "step": 22837
    },
    {
      "epoch": 2.177906036376953e-05,
      "step": 22837,
      "training_step_time": 8.168709516525269
    },
    {
      "epoch": 2.178001403808594e-05,
      "model_forward_time": 1.7578973770141602,
      "step": 22838
    },
    {
      "epoch": 2.178001403808594e-05,
      "step": 22838,
      "training_step_time": 7.753591299057007
    },
    {
      "epoch": 2.1780967712402342e-05,
      "model_forward_time": 1.758052110671997,
      "step": 22839
    },
    {
      "epoch": 2.1780967712402342e-05,
      "step": 22839,
      "training_step_time": 8.865272760391235
    },
    {
      "epoch": 2.178192138671875e-05,
      "grad_norm": 0.14564596116542816,
      "learning_rate": 9.154912126143484e-06,
      "loss": 0.0647,
      "step": 22840
    },
    {
      "epoch": 2.178192138671875e-05,
      "model_forward_time": 1.7600553035736084,
      "step": 22840
    },
    {
      "dynamics_loss_avg": 0.06250545978546143,
      "epoch": 2.178192138671875e-05,
      "step": 22840
    },
    {
      "action_loss_avg": 0.0019479806418530644,
      "epoch": 2.178192138671875e-05,
      "step": 22840
    },
    {
      "epoch": 2.178192138671875e-05,
      "step": 22840,
      "training_step_time": 7.815568923950195
    },
    {
      "epoch": 2.1782875061035157e-05,
      "model_forward_time": 1.7580461502075195,
      "step": 22841
    },
    {
      "epoch": 2.1782875061035157e-05,
      "step": 22841,
      "training_step_time": 7.830013751983643
    },
    {
      "epoch": 2.178382873535156e-05,
      "model_forward_time": 1.7635979652404785,
      "step": 22842
    },
    {
      "epoch": 2.178382873535156e-05,
      "step": 22842,
      "training_step_time": 8.261728048324585
    },
    {
      "epoch": 2.178478240966797e-05,
      "model_forward_time": 1.7608513832092285,
      "step": 22843
    },
    {
      "epoch": 2.178478240966797e-05,
      "step": 22843,
      "training_step_time": 7.536381721496582
    },
    {
      "epoch": 2.1785736083984376e-05,
      "model_forward_time": 1.7582194805145264,
      "step": 22844
    },
    {
      "epoch": 2.1785736083984376e-05,
      "step": 22844,
      "training_step_time": 8.254765748977661
    },
    {
      "epoch": 2.178668975830078e-05,
      "model_forward_time": 1.7611334323883057,
      "step": 22845
    },
    {
      "epoch": 2.178668975830078e-05,
      "step": 22845,
      "training_step_time": 8.058067083358765
    },
    {
      "epoch": 2.1787643432617188e-05,
      "model_forward_time": 1.7985332012176514,
      "step": 22846
    },
    {
      "epoch": 2.1787643432617188e-05,
      "step": 22846,
      "training_step_time": 8.132258653640747
    },
    {
      "epoch": 2.1788597106933595e-05,
      "model_forward_time": 1.758030891418457,
      "step": 22847
    },
    {
      "epoch": 2.1788597106933595e-05,
      "step": 22847,
      "training_step_time": 7.780765533447266
    },
    {
      "epoch": 2.178955078125e-05,
      "model_forward_time": 1.7586100101470947,
      "step": 22848
    },
    {
      "epoch": 2.178955078125e-05,
      "step": 22848,
      "training_step_time": 7.821167945861816
    },
    {
      "epoch": 2.1790504455566406e-05,
      "model_forward_time": 1.7627146244049072,
      "step": 22849
    },
    {
      "epoch": 2.1790504455566406e-05,
      "step": 22849,
      "training_step_time": 7.89341926574707
    },
    {
      "epoch": 2.1791458129882814e-05,
      "grad_norm": 0.15681041777133942,
      "learning_rate": 9.153992075914224e-06,
      "loss": 0.0643,
      "step": 22850
    },
    {
      "epoch": 2.1791458129882814e-05,
      "model_forward_time": 1.7551987171173096,
      "step": 22850
    },
    {
      "dynamics_loss_avg": 0.04602725375443697,
      "epoch": 2.1791458129882814e-05,
      "step": 22850
    },
    {
      "action_loss_avg": 0.0018606032012030483,
      "epoch": 2.1791458129882814e-05,
      "step": 22850
    },
    {
      "epoch": 2.1791458129882814e-05,
      "step": 22850,
      "training_step_time": 8.602766513824463
    },
    {
      "epoch": 2.1792411804199218e-05,
      "model_forward_time": 1.755361557006836,
      "step": 22851
    },
    {
      "epoch": 2.1792411804199218e-05,
      "step": 22851,
      "training_step_time": 7.750708818435669
    },
    {
      "epoch": 2.1793365478515625e-05,
      "model_forward_time": 2.9195187091827393,
      "step": 22852
    },
    {
      "epoch": 2.1793365478515625e-05,
      "step": 22852,
      "training_step_time": 7.858152151107788
    },
    {
      "epoch": 2.1794319152832033e-05,
      "model_forward_time": 1.758070945739746,
      "step": 22853
    },
    {
      "epoch": 2.1794319152832033e-05,
      "step": 22853,
      "training_step_time": 7.7611236572265625
    },
    {
      "epoch": 2.1795272827148437e-05,
      "model_forward_time": 1.7559089660644531,
      "step": 22854
    },
    {
      "epoch": 2.1795272827148437e-05,
      "step": 22854,
      "training_step_time": 8.01164436340332
    },
    {
      "epoch": 2.1796226501464844e-05,
      "model_forward_time": 1.7751686573028564,
      "step": 22855
    },
    {
      "epoch": 2.1796226501464844e-05,
      "step": 22855,
      "training_step_time": 9.94158387184143
    },
    {
      "epoch": 2.179718017578125e-05,
      "model_forward_time": 1.7639522552490234,
      "step": 22856
    },
    {
      "epoch": 2.179718017578125e-05,
      "step": 22856,
      "training_step_time": 9.141624689102173
    },
    {
      "epoch": 2.1798133850097656e-05,
      "model_forward_time": 1.766986608505249,
      "step": 22857
    },
    {
      "epoch": 2.1798133850097656e-05,
      "step": 22857,
      "training_step_time": 8.804621696472168
    },
    {
      "epoch": 2.1799087524414063e-05,
      "model_forward_time": 1.7598938941955566,
      "step": 22858
    },
    {
      "epoch": 2.1799087524414063e-05,
      "step": 22858,
      "training_step_time": 7.85230016708374
    },
    {
      "epoch": 2.180004119873047e-05,
      "model_forward_time": 1.7578680515289307,
      "step": 22859
    },
    {
      "epoch": 2.180004119873047e-05,
      "step": 22859,
      "training_step_time": 8.84970211982727
    },
    {
      "epoch": 2.1800994873046874e-05,
      "grad_norm": 0.1924779862165451,
      "learning_rate": 9.153071571410635e-06,
      "loss": 0.0617,
      "step": 22860
    },
    {
      "epoch": 2.1800994873046874e-05,
      "model_forward_time": 1.7735912799835205,
      "step": 22860
    },
    {
      "dynamics_loss_avg": 0.04400554299354553,
      "epoch": 2.1800994873046874e-05,
      "step": 22860
    },
    {
      "action_loss_avg": 0.002203407394699752,
      "epoch": 2.1800994873046874e-05,
      "step": 22860
    },
    {
      "epoch": 2.1800994873046874e-05,
      "step": 22860,
      "training_step_time": 7.7997825145721436
    },
    {
      "epoch": 2.1801948547363282e-05,
      "model_forward_time": 2.919393301010132,
      "step": 22861
    },
    {
      "epoch": 2.1801948547363282e-05,
      "step": 22861,
      "training_step_time": 7.894029378890991
    },
    {
      "epoch": 2.1802902221679686e-05,
      "model_forward_time": 1.7642035484313965,
      "step": 22862
    },
    {
      "epoch": 2.1802902221679686e-05,
      "step": 22862,
      "training_step_time": 8.339370727539062
    },
    {
      "epoch": 2.1803855895996093e-05,
      "model_forward_time": 1.3986985683441162,
      "step": 22863
    },
    {
      "epoch": 2.1803855895996093e-05,
      "step": 22863,
      "training_step_time": 9.502655267715454
    },
    {
      "epoch": 2.18048095703125e-05,
      "model_forward_time": 1.7669782638549805,
      "step": 22864
    },
    {
      "epoch": 2.18048095703125e-05,
      "step": 22864,
      "training_step_time": 7.860777378082275
    },
    {
      "epoch": 2.1805763244628905e-05,
      "model_forward_time": 1.8690595626831055,
      "step": 22865
    },
    {
      "epoch": 2.1805763244628905e-05,
      "step": 22865,
      "training_step_time": 8.107380867004395
    },
    {
      "epoch": 2.1806716918945312e-05,
      "model_forward_time": 1.7665948867797852,
      "step": 22866
    },
    {
      "epoch": 2.1806716918945312e-05,
      "step": 22866,
      "training_step_time": 8.307265520095825
    },
    {
      "epoch": 2.180767059326172e-05,
      "model_forward_time": 1.7646324634552002,
      "step": 22867
    },
    {
      "epoch": 2.180767059326172e-05,
      "step": 22867,
      "training_step_time": 8.40444302558899
    },
    {
      "epoch": 2.1808624267578124e-05,
      "model_forward_time": 1.7673966884613037,
      "step": 22868
    },
    {
      "epoch": 2.1808624267578124e-05,
      "step": 22868,
      "training_step_time": 7.670370101928711
    },
    {
      "epoch": 2.180957794189453e-05,
      "model_forward_time": 1.7696495056152344,
      "step": 22869
    },
    {
      "epoch": 2.180957794189453e-05,
      "step": 22869,
      "training_step_time": 8.361432552337646
    },
    {
      "epoch": 2.181053161621094e-05,
      "grad_norm": 0.19169479608535767,
      "learning_rate": 9.152150612733385e-06,
      "loss": 0.062,
      "step": 22870
    },
    {
      "epoch": 2.181053161621094e-05,
      "model_forward_time": 2.878243923187256,
      "step": 22870
    },
    {
      "dynamics_loss_avg": 0.05952410576865077,
      "epoch": 2.181053161621094e-05,
      "step": 22870
    },
    {
      "action_loss_avg": 0.0024107916571665554,
      "epoch": 2.181053161621094e-05,
      "step": 22870
    },
    {
      "epoch": 2.181053161621094e-05,
      "step": 22870,
      "training_step_time": 7.8767805099487305
    },
    {
      "epoch": 2.1811485290527342e-05,
      "model_forward_time": 1.7635014057159424,
      "step": 22871
    },
    {
      "epoch": 2.1811485290527342e-05,
      "step": 22871,
      "training_step_time": 8.055789232254028
    },
    {
      "epoch": 2.181243896484375e-05,
      "model_forward_time": 1.7648766040802002,
      "step": 22872
    },
    {
      "epoch": 2.181243896484375e-05,
      "step": 22872,
      "training_step_time": 8.921599626541138
    },
    {
      "epoch": 2.1813392639160157e-05,
      "model_forward_time": 1.7647678852081299,
      "step": 22873
    },
    {
      "epoch": 2.1813392639160157e-05,
      "step": 22873,
      "training_step_time": 8.32202434539795
    },
    {
      "epoch": 2.181434631347656e-05,
      "model_forward_time": 1.8547706604003906,
      "step": 22874
    },
    {
      "epoch": 2.181434631347656e-05,
      "step": 22874,
      "training_step_time": 8.142569065093994
    },
    {
      "epoch": 2.181529998779297e-05,
      "model_forward_time": 1.936434030532837,
      "step": 22875
    },
    {
      "epoch": 2.181529998779297e-05,
      "step": 22875,
      "training_step_time": 8.63482403755188
    },
    {
      "epoch": 2.1816253662109376e-05,
      "model_forward_time": 0.8561482429504395,
      "step": 22876
    },
    {
      "epoch": 2.1816253662109376e-05,
      "step": 22876,
      "training_step_time": 7.883768558502197
    },
    {
      "epoch": 2.181720733642578e-05,
      "model_forward_time": 1.9035837650299072,
      "step": 22877
    },
    {
      "epoch": 2.181720733642578e-05,
      "step": 22877,
      "training_step_time": 8.601385354995728
    },
    {
      "epoch": 2.1818161010742188e-05,
      "model_forward_time": 1.9233078956604004,
      "step": 22878
    },
    {
      "epoch": 2.1818161010742188e-05,
      "step": 22878,
      "training_step_time": 7.914264440536499
    },
    {
      "epoch": 2.1819114685058595e-05,
      "model_forward_time": 1.921778678894043,
      "step": 22879
    },
    {
      "epoch": 2.1819114685058595e-05,
      "step": 22879,
      "training_step_time": 7.761545181274414
    },
    {
      "epoch": 2.1820068359375e-05,
      "grad_norm": 0.14221176505088806,
      "learning_rate": 9.151229199983185e-06,
      "loss": 0.0604,
      "step": 22880
    },
    {
      "epoch": 2.1820068359375e-05,
      "model_forward_time": 2.974029064178467,
      "step": 22880
    },
    {
      "dynamics_loss_avg": 0.05568198449909687,
      "epoch": 2.1820068359375e-05,
      "step": 22880
    },
    {
      "action_loss_avg": 0.0020282984187360853,
      "epoch": 2.1820068359375e-05,
      "step": 22880
    },
    {
      "epoch": 2.1820068359375e-05,
      "step": 22880,
      "training_step_time": 7.979242563247681
    },
    {
      "epoch": 2.1821022033691406e-05,
      "model_forward_time": 1.9070065021514893,
      "step": 22881
    },
    {
      "epoch": 2.1821022033691406e-05,
      "step": 22881,
      "training_step_time": 8.11057424545288
    },
    {
      "epoch": 2.1821975708007814e-05,
      "model_forward_time": 1.8839142322540283,
      "step": 22882
    },
    {
      "epoch": 2.1821975708007814e-05,
      "step": 22882,
      "training_step_time": 8.908313751220703
    },
    {
      "epoch": 2.1822929382324218e-05,
      "model_forward_time": 1.9258346557617188,
      "step": 22883
    },
    {
      "epoch": 2.1822929382324218e-05,
      "step": 22883,
      "training_step_time": 9.020748138427734
    },
    {
      "epoch": 2.1823883056640625e-05,
      "model_forward_time": 1.8850007057189941,
      "step": 22884
    },
    {
      "epoch": 2.1823883056640625e-05,
      "step": 22884,
      "training_step_time": 8.620752334594727
    },
    {
      "epoch": 2.1824836730957033e-05,
      "model_forward_time": 1.8978886604309082,
      "step": 22885
    },
    {
      "epoch": 2.1824836730957033e-05,
      "step": 22885,
      "training_step_time": 7.837202548980713
    },
    {
      "epoch": 2.1825790405273437e-05,
      "model_forward_time": 1.90543532371521,
      "step": 22886
    },
    {
      "epoch": 2.1825790405273437e-05,
      "step": 22886,
      "training_step_time": 8.913509607315063
    },
    {
      "epoch": 2.1826744079589844e-05,
      "model_forward_time": 1.9129242897033691,
      "step": 22887
    },
    {
      "epoch": 2.1826744079589844e-05,
      "step": 22887,
      "training_step_time": 8.048608779907227
    },
    {
      "epoch": 2.182769775390625e-05,
      "model_forward_time": 1.8953971862792969,
      "step": 22888
    },
    {
      "epoch": 2.182769775390625e-05,
      "step": 22888,
      "training_step_time": 7.695423126220703
    },
    {
      "epoch": 2.1828651428222656e-05,
      "model_forward_time": 3.0544540882110596,
      "step": 22889
    },
    {
      "epoch": 2.1828651428222656e-05,
      "step": 22889,
      "training_step_time": 8.031075239181519
    },
    {
      "epoch": 2.1829605102539063e-05,
      "grad_norm": 0.1636998951435089,
      "learning_rate": 9.150307333260802e-06,
      "loss": 0.0637,
      "step": 22890
    },
    {
      "epoch": 2.1829605102539063e-05,
      "model_forward_time": 1.9093713760375977,
      "step": 22890
    },
    {
      "dynamics_loss_avg": 0.04355033207684755,
      "epoch": 2.1829605102539063e-05,
      "step": 22890
    },
    {
      "action_loss_avg": 0.0023201395582873375,
      "epoch": 2.1829605102539063e-05,
      "step": 22890
    },
    {
      "epoch": 2.1829605102539063e-05,
      "step": 22890,
      "training_step_time": 8.036563396453857
    },
    {
      "epoch": 2.1830558776855467e-05,
      "model_forward_time": 1.9387240409851074,
      "step": 22891
    },
    {
      "epoch": 2.1830558776855467e-05,
      "step": 22891,
      "training_step_time": 7.88025164604187
    },
    {
      "epoch": 2.1831512451171874e-05,
      "model_forward_time": 1.9048371315002441,
      "step": 22892
    },
    {
      "epoch": 2.1831512451171874e-05,
      "step": 22892,
      "training_step_time": 8.02327585220337
    },
    {
      "epoch": 2.1832466125488282e-05,
      "model_forward_time": 1.8921332359313965,
      "step": 22893
    },
    {
      "epoch": 2.1832466125488282e-05,
      "step": 22893,
      "training_step_time": 8.480397701263428
    },
    {
      "epoch": 2.1833419799804686e-05,
      "model_forward_time": 1.927215337753296,
      "step": 22894
    },
    {
      "epoch": 2.1833419799804686e-05,
      "step": 22894,
      "training_step_time": 9.525466918945312
    },
    {
      "epoch": 2.1834373474121093e-05,
      "model_forward_time": 1.8888027667999268,
      "step": 22895
    },
    {
      "epoch": 2.1834373474121093e-05,
      "step": 22895,
      "training_step_time": 8.717469930648804
    },
    {
      "epoch": 2.18353271484375e-05,
      "model_forward_time": 1.9160840511322021,
      "step": 22896
    },
    {
      "epoch": 2.18353271484375e-05,
      "step": 22896,
      "training_step_time": 7.881983280181885
    },
    {
      "epoch": 2.1836280822753905e-05,
      "model_forward_time": 1.9280612468719482,
      "step": 22897
    },
    {
      "epoch": 2.1836280822753905e-05,
      "step": 22897,
      "training_step_time": 7.7005393505096436
    },
    {
      "epoch": 2.1837234497070312e-05,
      "model_forward_time": 3.0326437950134277,
      "step": 22898
    },
    {
      "epoch": 2.1837234497070312e-05,
      "step": 22898,
      "training_step_time": 7.751370668411255
    },
    {
      "epoch": 2.183818817138672e-05,
      "model_forward_time": 1.9049453735351562,
      "step": 22899
    },
    {
      "epoch": 2.183818817138672e-05,
      "step": 22899,
      "training_step_time": 8.6174955368042
    },
    {
      "epoch": 2.1839141845703124e-05,
      "grad_norm": 0.1855631023645401,
      "learning_rate": 9.149385012667047e-06,
      "loss": 0.0622,
      "step": 22900
    },
    {
      "epoch": 2.1839141845703124e-05,
      "model_forward_time": 1.9141857624053955,
      "step": 22900
    },
    {
      "dynamics_loss_avg": 0.08391336258500814,
      "epoch": 2.1839141845703124e-05,
      "step": 22900
    },
    {
      "action_loss_avg": 0.0029849310056306423,
      "epoch": 2.1839141845703124e-05,
      "step": 22900
    },
    {
      "epoch": 2.1839141845703124e-05,
      "step": 22900,
      "training_step_time": 8.63550615310669
    },
    {
      "epoch": 2.184009552001953e-05,
      "model_forward_time": 1.8767344951629639,
      "step": 22901
    },
    {
      "epoch": 2.184009552001953e-05,
      "step": 22901,
      "training_step_time": 7.853846788406372
    },
    {
      "epoch": 2.184104919433594e-05,
      "model_forward_time": 1.8982617855072021,
      "step": 22902
    },
    {
      "epoch": 2.184104919433594e-05,
      "step": 22902,
      "training_step_time": 7.8724799156188965
    },
    {
      "epoch": 2.1842002868652343e-05,
      "model_forward_time": 1.937415361404419,
      "step": 22903
    },
    {
      "epoch": 2.1842002868652343e-05,
      "step": 22903,
      "training_step_time": 7.892697095870972
    },
    {
      "epoch": 2.184295654296875e-05,
      "model_forward_time": 1.9173192977905273,
      "step": 22904
    },
    {
      "epoch": 2.184295654296875e-05,
      "step": 22904,
      "training_step_time": 8.605740547180176
    },
    {
      "epoch": 2.1843910217285157e-05,
      "model_forward_time": 1.9129886627197266,
      "step": 22905
    },
    {
      "epoch": 2.1843910217285157e-05,
      "step": 22905,
      "training_step_time": 8.682177066802979
    },
    {
      "epoch": 2.184486389160156e-05,
      "model_forward_time": 1.9055240154266357,
      "step": 22906
    },
    {
      "epoch": 2.184486389160156e-05,
      "step": 22906,
      "training_step_time": 7.885754823684692
    },
    {
      "epoch": 2.184581756591797e-05,
      "model_forward_time": 3.015463352203369,
      "step": 22907
    },
    {
      "epoch": 2.184581756591797e-05,
      "step": 22907,
      "training_step_time": 7.987419605255127
    },
    {
      "epoch": 2.1846771240234376e-05,
      "model_forward_time": 1.9353547096252441,
      "step": 22908
    },
    {
      "epoch": 2.1846771240234376e-05,
      "step": 22908,
      "training_step_time": 7.984677791595459
    },
    {
      "epoch": 2.184772491455078e-05,
      "model_forward_time": 1.9198687076568604,
      "step": 22909
    },
    {
      "epoch": 2.184772491455078e-05,
      "step": 22909,
      "training_step_time": 7.869685888290405
    },
    {
      "epoch": 2.1848678588867188e-05,
      "grad_norm": 0.1719440370798111,
      "learning_rate": 9.148462238302787e-06,
      "loss": 0.0598,
      "step": 22910
    },
    {
      "epoch": 2.1848678588867188e-05,
      "model_forward_time": 1.9275131225585938,
      "step": 22910
    },
    {
      "dynamics_loss_avg": 0.08568558823317289,
      "epoch": 2.1848678588867188e-05,
      "step": 22910
    },
    {
      "action_loss_avg": 0.003701380401616916,
      "epoch": 2.1848678588867188e-05,
      "step": 22910
    },
    {
      "epoch": 2.1848678588867188e-05,
      "step": 22910,
      "training_step_time": 9.690590858459473
    },
    {
      "epoch": 2.1849632263183595e-05,
      "model_forward_time": 1.9272983074188232,
      "step": 22911
    },
    {
      "epoch": 2.1849632263183595e-05,
      "step": 22911,
      "training_step_time": 7.925642490386963
    },
    {
      "epoch": 2.18505859375e-05,
      "model_forward_time": 1.2183425426483154,
      "step": 22912
    },
    {
      "epoch": 2.18505859375e-05,
      "step": 22912,
      "training_step_time": 8.943799495697021
    },
    {
      "epoch": 2.1851539611816406e-05,
      "model_forward_time": 1.9166967868804932,
      "step": 22913
    },
    {
      "epoch": 2.1851539611816406e-05,
      "step": 22913,
      "training_step_time": 7.841269493103027
    },
    {
      "epoch": 2.1852493286132814e-05,
      "model_forward_time": 1.8953971862792969,
      "step": 22914
    },
    {
      "epoch": 2.1852493286132814e-05,
      "step": 22914,
      "training_step_time": 7.865925073623657
    },
    {
      "epoch": 2.1853446960449218e-05,
      "model_forward_time": 1.9174830913543701,
      "step": 22915
    },
    {
      "epoch": 2.1853446960449218e-05,
      "step": 22915,
      "training_step_time": 7.904202222824097
    },
    {
      "epoch": 2.1854400634765625e-05,
      "model_forward_time": 1.9068353176116943,
      "step": 22916
    },
    {
      "epoch": 2.1854400634765625e-05,
      "step": 22916,
      "training_step_time": 7.791691541671753
    },
    {
      "epoch": 2.1855354309082033e-05,
      "model_forward_time": 3.0242183208465576,
      "step": 22917
    },
    {
      "epoch": 2.1855354309082033e-05,
      "step": 22917,
      "training_step_time": 8.024980306625366
    },
    {
      "epoch": 2.1856307983398437e-05,
      "model_forward_time": 1.936424732208252,
      "step": 22918
    },
    {
      "epoch": 2.1856307983398437e-05,
      "step": 22918,
      "training_step_time": 8.711692333221436
    },
    {
      "epoch": 2.1857261657714844e-05,
      "model_forward_time": 1.9372057914733887,
      "step": 22919
    },
    {
      "epoch": 2.1857261657714844e-05,
      "step": 22919,
      "training_step_time": 7.7925403118133545
    },
    {
      "epoch": 2.185821533203125e-05,
      "grad_norm": 0.14436247944831848,
      "learning_rate": 9.147539010268935e-06,
      "loss": 0.0604,
      "step": 22920
    },
    {
      "epoch": 2.185821533203125e-05,
      "model_forward_time": 1.9398436546325684,
      "step": 22920
    },
    {
      "dynamics_loss_avg": 0.05891843987628818,
      "epoch": 2.185821533203125e-05,
      "step": 22920
    },
    {
      "action_loss_avg": 0.0022817802499048414,
      "epoch": 2.185821533203125e-05,
      "step": 22920
    },
    {
      "epoch": 2.185821533203125e-05,
      "step": 22920,
      "training_step_time": 8.531316041946411
    },
    {
      "epoch": 2.1859169006347656e-05,
      "model_forward_time": 1.9454402923583984,
      "step": 22921
    },
    {
      "epoch": 2.1859169006347656e-05,
      "step": 22921,
      "training_step_time": 8.438157558441162
    },
    {
      "epoch": 2.1860122680664063e-05,
      "model_forward_time": 1.9289298057556152,
      "step": 22922
    },
    {
      "epoch": 2.1860122680664063e-05,
      "step": 22922,
      "training_step_time": 7.824576139450073
    },
    {
      "epoch": 2.1861076354980467e-05,
      "model_forward_time": 1.9212183952331543,
      "step": 22923
    },
    {
      "epoch": 2.1861076354980467e-05,
      "step": 22923,
      "training_step_time": 7.884968042373657
    },
    {
      "epoch": 2.1862030029296875e-05,
      "model_forward_time": 1.927173376083374,
      "step": 22924
    },
    {
      "epoch": 2.1862030029296875e-05,
      "step": 22924,
      "training_step_time": 7.850522756576538
    },
    {
      "epoch": 2.1862983703613282e-05,
      "model_forward_time": 1.9479153156280518,
      "step": 22925
    },
    {
      "epoch": 2.1862983703613282e-05,
      "step": 22925,
      "training_step_time": 7.736002206802368
    },
    {
      "epoch": 2.1863937377929686e-05,
      "model_forward_time": 3.0187466144561768,
      "step": 22926
    },
    {
      "epoch": 2.1863937377929686e-05,
      "step": 22926,
      "training_step_time": 7.736643552780151
    },
    {
      "epoch": 2.1864891052246093e-05,
      "model_forward_time": 1.8708646297454834,
      "step": 22927
    },
    {
      "epoch": 2.1864891052246093e-05,
      "step": 22927,
      "training_step_time": 8.533685207366943
    },
    {
      "epoch": 2.18658447265625e-05,
      "model_forward_time": 1.915755271911621,
      "step": 22928
    },
    {
      "epoch": 2.18658447265625e-05,
      "step": 22928,
      "training_step_time": 7.7530717849731445
    },
    {
      "epoch": 2.1866798400878905e-05,
      "model_forward_time": 1.8951151371002197,
      "step": 22929
    },
    {
      "epoch": 2.1866798400878905e-05,
      "step": 22929,
      "training_step_time": 9.35072112083435
    },
    {
      "epoch": 2.1867752075195312e-05,
      "grad_norm": 0.18369540572166443,
      "learning_rate": 9.14661532866645e-06,
      "loss": 0.0585,
      "step": 22930
    },
    {
      "epoch": 2.1867752075195312e-05,
      "model_forward_time": 1.9363667964935303,
      "step": 22930
    },
    {
      "dynamics_loss_avg": 0.06076948270201683,
      "epoch": 2.1867752075195312e-05,
      "step": 22930
    },
    {
      "action_loss_avg": 0.0032577560225035994,
      "epoch": 2.1867752075195312e-05,
      "step": 22930
    },
    {
      "epoch": 2.1867752075195312e-05,
      "step": 22930,
      "training_step_time": 9.051141023635864
    },
    {
      "epoch": 2.186870574951172e-05,
      "model_forward_time": 1.9555919170379639,
      "step": 22931
    },
    {
      "epoch": 2.186870574951172e-05,
      "step": 22931,
      "training_step_time": 8.649657964706421
    },
    {
      "epoch": 2.1869659423828124e-05,
      "model_forward_time": 1.940613031387329,
      "step": 22932
    },
    {
      "epoch": 2.1869659423828124e-05,
      "step": 22932,
      "training_step_time": 7.892796039581299
    },
    {
      "epoch": 2.187061309814453e-05,
      "model_forward_time": 1.9356038570404053,
      "step": 22933
    },
    {
      "epoch": 2.187061309814453e-05,
      "step": 22933,
      "training_step_time": 8.513251543045044
    },
    {
      "epoch": 2.187156677246094e-05,
      "model_forward_time": 1.9021391868591309,
      "step": 22934
    },
    {
      "epoch": 2.187156677246094e-05,
      "step": 22934,
      "training_step_time": 7.733489751815796
    },
    {
      "epoch": 2.1872520446777343e-05,
      "model_forward_time": 3.019502878189087,
      "step": 22935
    },
    {
      "epoch": 2.1872520446777343e-05,
      "step": 22935,
      "training_step_time": 8.023826122283936
    },
    {
      "epoch": 2.187347412109375e-05,
      "model_forward_time": 1.539050817489624,
      "step": 22936
    },
    {
      "epoch": 2.187347412109375e-05,
      "step": 22936,
      "training_step_time": 7.8386828899383545
    },
    {
      "epoch": 2.1874427795410157e-05,
      "model_forward_time": 1.93186616897583,
      "step": 22937
    },
    {
      "epoch": 2.1874427795410157e-05,
      "step": 22937,
      "training_step_time": 7.9281980991363525
    },
    {
      "epoch": 2.187538146972656e-05,
      "model_forward_time": 1.9190750122070312,
      "step": 22938
    },
    {
      "epoch": 2.187538146972656e-05,
      "step": 22938,
      "training_step_time": 8.019853353500366
    },
    {
      "epoch": 2.187633514404297e-05,
      "model_forward_time": 1.9032788276672363,
      "step": 22939
    },
    {
      "epoch": 2.187633514404297e-05,
      "step": 22939,
      "training_step_time": 7.874293565750122
    },
    {
      "epoch": 2.1877288818359376e-05,
      "grad_norm": 0.1854265183210373,
      "learning_rate": 9.14569119359635e-06,
      "loss": 0.058,
      "step": 22940
    },
    {
      "epoch": 2.1877288818359376e-05,
      "model_forward_time": 1.9188814163208008,
      "step": 22940
    },
    {
      "dynamics_loss_avg": 0.07152226464822889,
      "epoch": 2.1877288818359376e-05,
      "step": 22940
    },
    {
      "action_loss_avg": 0.0023377498728223145,
      "epoch": 2.1877288818359376e-05,
      "step": 22940
    },
    {
      "epoch": 2.1877288818359376e-05,
      "step": 22940,
      "training_step_time": 7.906188249588013
    },
    {
      "epoch": 2.187824249267578e-05,
      "model_forward_time": 1.9042918682098389,
      "step": 22941
    },
    {
      "epoch": 2.187824249267578e-05,
      "step": 22941,
      "training_step_time": 8.72774076461792
    },
    {
      "epoch": 2.1879196166992188e-05,
      "model_forward_time": 1.944284200668335,
      "step": 22942
    },
    {
      "epoch": 2.1879196166992188e-05,
      "step": 22942,
      "training_step_time": 7.992762327194214
    },
    {
      "epoch": 2.1880149841308595e-05,
      "model_forward_time": 1.9232380390167236,
      "step": 22943
    },
    {
      "epoch": 2.1880149841308595e-05,
      "step": 22943,
      "training_step_time": 7.779924631118774
    },
    {
      "epoch": 2.1881103515625e-05,
      "model_forward_time": 1.940373420715332,
      "step": 22944
    },
    {
      "epoch": 2.1881103515625e-05,
      "step": 22944,
      "training_step_time": 7.74750280380249
    },
    {
      "epoch": 2.1882057189941407e-05,
      "model_forward_time": 2.995321035385132,
      "step": 22945
    },
    {
      "epoch": 2.1882057189941407e-05,
      "step": 22945,
      "training_step_time": 8.096943140029907
    },
    {
      "epoch": 2.1883010864257814e-05,
      "model_forward_time": 1.9298169612884521,
      "step": 22946
    },
    {
      "epoch": 2.1883010864257814e-05,
      "step": 22946,
      "training_step_time": 7.867071151733398
    },
    {
      "epoch": 2.1883964538574218e-05,
      "model_forward_time": 1.945070505142212,
      "step": 22947
    },
    {
      "epoch": 2.1883964538574218e-05,
      "step": 22947,
      "training_step_time": 8.978912830352783
    },
    {
      "epoch": 2.1884918212890625e-05,
      "model_forward_time": 1.9250023365020752,
      "step": 22948
    },
    {
      "epoch": 2.1884918212890625e-05,
      "step": 22948,
      "training_step_time": 8.598782777786255
    },
    {
      "epoch": 2.1885871887207033e-05,
      "model_forward_time": 1.9095780849456787,
      "step": 22949
    },
    {
      "epoch": 2.1885871887207033e-05,
      "step": 22949,
      "training_step_time": 8.368037462234497
    },
    {
      "epoch": 2.1886825561523437e-05,
      "grad_norm": 0.1798926144838333,
      "learning_rate": 9.144766605159692e-06,
      "loss": 0.0589,
      "step": 22950
    },
    {
      "epoch": 2.1886825561523437e-05,
      "model_forward_time": 1.899139404296875,
      "step": 22950
    },
    {
      "dynamics_loss_avg": 0.046569026075303555,
      "epoch": 2.1886825561523437e-05,
      "step": 22950
    },
    {
      "action_loss_avg": 0.0018693594611249864,
      "epoch": 2.1886825561523437e-05,
      "step": 22950
    },
    {
      "epoch": 2.1886825561523437e-05,
      "step": 22950,
      "training_step_time": 9.35236120223999
    },
    {
      "epoch": 2.1887779235839844e-05,
      "model_forward_time": 1.1961586475372314,
      "step": 22951
    },
    {
      "epoch": 2.1887779235839844e-05,
      "step": 22951,
      "training_step_time": 7.916925430297852
    },
    {
      "epoch": 2.188873291015625e-05,
      "model_forward_time": 1.918792724609375,
      "step": 22952
    },
    {
      "epoch": 2.188873291015625e-05,
      "step": 22952,
      "training_step_time": 8.029382705688477
    },
    {
      "epoch": 2.1889686584472656e-05,
      "model_forward_time": 1.9337167739868164,
      "step": 22953
    },
    {
      "epoch": 2.1889686584472656e-05,
      "step": 22953,
      "training_step_time": 7.768187999725342
    },
    {
      "epoch": 2.1890640258789063e-05,
      "model_forward_time": 3.040743589401245,
      "step": 22954
    },
    {
      "epoch": 2.1890640258789063e-05,
      "step": 22954,
      "training_step_time": 8.906753063201904
    },
    {
      "epoch": 2.1891593933105467e-05,
      "model_forward_time": 1.9330408573150635,
      "step": 22955
    },
    {
      "epoch": 2.1891593933105467e-05,
      "step": 22955,
      "training_step_time": 8.804242849349976
    },
    {
      "epoch": 2.1892547607421875e-05,
      "model_forward_time": 1.9377479553222656,
      "step": 22956
    },
    {
      "epoch": 2.1892547607421875e-05,
      "step": 22956,
      "training_step_time": 7.924485921859741
    },
    {
      "epoch": 2.1893501281738282e-05,
      "model_forward_time": 1.9123501777648926,
      "step": 22957
    },
    {
      "epoch": 2.1893501281738282e-05,
      "step": 22957,
      "training_step_time": 7.8916847705841064
    },
    {
      "epoch": 2.1894454956054686e-05,
      "model_forward_time": 1.9345042705535889,
      "step": 22958
    },
    {
      "epoch": 2.1894454956054686e-05,
      "step": 22958,
      "training_step_time": 8.568725109100342
    },
    {
      "epoch": 2.1895408630371093e-05,
      "model_forward_time": 1.9078853130340576,
      "step": 22959
    },
    {
      "epoch": 2.1895408630371093e-05,
      "step": 22959,
      "training_step_time": 7.880897760391235
    },
    {
      "epoch": 2.18963623046875e-05,
      "grad_norm": 0.1617116928100586,
      "learning_rate": 9.14384156345759e-06,
      "loss": 0.0596,
      "step": 22960
    },
    {
      "epoch": 2.18963623046875e-05,
      "model_forward_time": 1.9325735569000244,
      "step": 22960
    },
    {
      "dynamics_loss_avg": 0.041093862149864435,
      "epoch": 2.18963623046875e-05,
      "step": 22960
    },
    {
      "action_loss_avg": 0.001847659022314474,
      "epoch": 2.18963623046875e-05,
      "step": 22960
    },
    {
      "epoch": 2.18963623046875e-05,
      "step": 22960,
      "training_step_time": 7.858193397521973
    },
    {
      "epoch": 2.1897315979003905e-05,
      "model_forward_time": 1.9282753467559814,
      "step": 22961
    },
    {
      "epoch": 2.1897315979003905e-05,
      "step": 22961,
      "training_step_time": 7.969263076782227
    },
    {
      "epoch": 2.1898269653320312e-05,
      "model_forward_time": 1.905073642730713,
      "step": 22962
    },
    {
      "epoch": 2.1898269653320312e-05,
      "step": 22962,
      "training_step_time": 8.741462469100952
    },
    {
      "epoch": 2.189922332763672e-05,
      "model_forward_time": 3.017664909362793,
      "step": 22963
    },
    {
      "epoch": 2.189922332763672e-05,
      "step": 22963,
      "training_step_time": 8.003564357757568
    },
    {
      "epoch": 2.1900177001953124e-05,
      "model_forward_time": 1.9262642860412598,
      "step": 22964
    },
    {
      "epoch": 2.1900177001953124e-05,
      "step": 22964,
      "training_step_time": 8.23319125175476
    },
    {
      "epoch": 2.190113067626953e-05,
      "model_forward_time": 1.9435088634490967,
      "step": 22965
    },
    {
      "epoch": 2.190113067626953e-05,
      "step": 22965,
      "training_step_time": 8.68902587890625
    },
    {
      "epoch": 2.190208435058594e-05,
      "model_forward_time": 1.953831434249878,
      "step": 22966
    },
    {
      "epoch": 2.190208435058594e-05,
      "step": 22966,
      "training_step_time": 9.826905488967896
    },
    {
      "epoch": 2.1903038024902343e-05,
      "model_forward_time": 1.9319398403167725,
      "step": 22967
    },
    {
      "epoch": 2.1903038024902343e-05,
      "step": 22967,
      "training_step_time": 8.507420539855957
    },
    {
      "epoch": 2.190399169921875e-05,
      "model_forward_time": 1.9145185947418213,
      "step": 22968
    },
    {
      "epoch": 2.190399169921875e-05,
      "step": 22968,
      "training_step_time": 7.7804176807403564
    },
    {
      "epoch": 2.1904945373535157e-05,
      "model_forward_time": 1.896331787109375,
      "step": 22969
    },
    {
      "epoch": 2.1904945373535157e-05,
      "step": 22969,
      "training_step_time": 7.825019359588623
    },
    {
      "epoch": 2.190589904785156e-05,
      "grad_norm": 0.16950900852680206,
      "learning_rate": 9.142916068591204e-06,
      "loss": 0.0609,
      "step": 22970
    },
    {
      "epoch": 2.190589904785156e-05,
      "model_forward_time": 1.9510471820831299,
      "step": 22970
    },
    {
      "dynamics_loss_avg": 0.04688735799863934,
      "epoch": 2.190589904785156e-05,
      "step": 22970
    },
    {
      "action_loss_avg": 0.0031638075510272757,
      "epoch": 2.190589904785156e-05,
      "step": 22970
    },
    {
      "epoch": 2.190589904785156e-05,
      "step": 22970,
      "training_step_time": 8.055395364761353
    },
    {
      "epoch": 2.190685272216797e-05,
      "model_forward_time": 1.9223253726959229,
      "step": 22971
    },
    {
      "epoch": 2.190685272216797e-05,
      "step": 22971,
      "training_step_time": 7.774507522583008
    },
    {
      "epoch": 2.1907806396484376e-05,
      "model_forward_time": 1.9115891456604004,
      "step": 22972
    },
    {
      "epoch": 2.1907806396484376e-05,
      "step": 22972,
      "training_step_time": 8.251556158065796
    },
    {
      "epoch": 2.190876007080078e-05,
      "model_forward_time": 1.934725046157837,
      "step": 22973
    },
    {
      "epoch": 2.190876007080078e-05,
      "step": 22973,
      "training_step_time": 8.102205276489258
    },
    {
      "epoch": 2.1909713745117188e-05,
      "model_forward_time": 1.9250497817993164,
      "step": 22974
    },
    {
      "epoch": 2.1909713745117188e-05,
      "step": 22974,
      "training_step_time": 7.883788585662842
    },
    {
      "epoch": 2.1910667419433595e-05,
      "model_forward_time": 1.5333616733551025,
      "step": 22975
    },
    {
      "epoch": 2.1910667419433595e-05,
      "step": 22975,
      "training_step_time": 8.605924844741821
    },
    {
      "epoch": 2.191162109375e-05,
      "model_forward_time": 1.9252102375030518,
      "step": 22976
    },
    {
      "epoch": 2.191162109375e-05,
      "step": 22976,
      "training_step_time": 7.81260347366333
    },
    {
      "epoch": 2.1912574768066407e-05,
      "model_forward_time": 1.9359018802642822,
      "step": 22977
    },
    {
      "epoch": 2.1912574768066407e-05,
      "step": 22977,
      "training_step_time": 8.93633770942688
    },
    {
      "epoch": 2.1913528442382814e-05,
      "model_forward_time": 1.9315965175628662,
      "step": 22978
    },
    {
      "epoch": 2.1913528442382814e-05,
      "step": 22978,
      "training_step_time": 7.799027442932129
    },
    {
      "epoch": 2.1914482116699218e-05,
      "model_forward_time": 1.9386744499206543,
      "step": 22979
    },
    {
      "epoch": 2.1914482116699218e-05,
      "step": 22979,
      "training_step_time": 8.027117252349854
    },
    {
      "epoch": 2.1915435791015625e-05,
      "grad_norm": 0.1689952313899994,
      "learning_rate": 9.141990120661745e-06,
      "loss": 0.0628,
      "step": 22980
    },
    {
      "epoch": 2.1915435791015625e-05,
      "model_forward_time": 1.891416311264038,
      "step": 22980
    },
    {
      "dynamics_loss_avg": 0.05639292281121015,
      "epoch": 2.1915435791015625e-05,
      "step": 22980
    },
    {
      "action_loss_avg": 0.0031906130781862883,
      "epoch": 2.1915435791015625e-05,
      "step": 22980
    },
    {
      "epoch": 2.1915435791015625e-05,
      "step": 22980,
      "training_step_time": 7.871138334274292
    },
    {
      "epoch": 2.1916389465332033e-05,
      "model_forward_time": 1.9104375839233398,
      "step": 22981
    },
    {
      "epoch": 2.1916389465332033e-05,
      "step": 22981,
      "training_step_time": 7.746044397354126
    },
    {
      "epoch": 2.1917343139648437e-05,
      "model_forward_time": 3.0342326164245605,
      "step": 22982
    },
    {
      "epoch": 2.1917343139648437e-05,
      "step": 22982,
      "training_step_time": 8.821669340133667
    },
    {
      "epoch": 2.1918296813964844e-05,
      "model_forward_time": 1.9320800304412842,
      "step": 22983
    },
    {
      "epoch": 2.1918296813964844e-05,
      "step": 22983,
      "training_step_time": 7.916717529296875
    },
    {
      "epoch": 2.191925048828125e-05,
      "model_forward_time": 1.9382612705230713,
      "step": 22984
    },
    {
      "epoch": 2.191925048828125e-05,
      "step": 22984,
      "training_step_time": 8.708561897277832
    },
    {
      "epoch": 2.1920204162597656e-05,
      "model_forward_time": 1.916562795639038,
      "step": 22985
    },
    {
      "epoch": 2.1920204162597656e-05,
      "step": 22985,
      "training_step_time": 8.82473874092102
    },
    {
      "epoch": 2.1921157836914063e-05,
      "model_forward_time": 1.9038798809051514,
      "step": 22986
    },
    {
      "epoch": 2.1921157836914063e-05,
      "step": 22986,
      "training_step_time": 8.63914442062378
    },
    {
      "epoch": 2.1922111511230467e-05,
      "model_forward_time": 1.894235372543335,
      "step": 22987
    },
    {
      "epoch": 2.1922111511230467e-05,
      "step": 22987,
      "training_step_time": 8.73843240737915
    },
    {
      "epoch": 2.1923065185546875e-05,
      "model_forward_time": 1.9152812957763672,
      "step": 22988
    },
    {
      "epoch": 2.1923065185546875e-05,
      "step": 22988,
      "training_step_time": 7.928417444229126
    },
    {
      "epoch": 2.1924018859863282e-05,
      "model_forward_time": 1.9304518699645996,
      "step": 22989
    },
    {
      "epoch": 2.1924018859863282e-05,
      "step": 22989,
      "training_step_time": 7.933234453201294
    },
    {
      "epoch": 2.1924972534179686e-05,
      "grad_norm": 0.11572879552841187,
      "learning_rate": 9.141063719770475e-06,
      "loss": 0.0653,
      "step": 22990
    },
    {
      "epoch": 2.1924972534179686e-05,
      "model_forward_time": 1.9207689762115479,
      "step": 22990
    },
    {
      "dynamics_loss_avg": 0.047678387351334096,
      "epoch": 2.1924972534179686e-05,
      "step": 22990
    },
    {
      "action_loss_avg": 0.0029132054245565087,
      "epoch": 2.1924972534179686e-05,
      "step": 22990
    },
    {
      "epoch": 2.1924972534179686e-05,
      "step": 22990,
      "training_step_time": 8.456478595733643
    },
    {
      "epoch": 2.1925926208496093e-05,
      "model_forward_time": 2.9881651401519775,
      "step": 22991
    },
    {
      "epoch": 2.1925926208496093e-05,
      "step": 22991,
      "training_step_time": 8.135732173919678
    },
    {
      "epoch": 2.19268798828125e-05,
      "model_forward_time": 1.9034371376037598,
      "step": 22992
    },
    {
      "epoch": 2.19268798828125e-05,
      "step": 22992,
      "training_step_time": 8.412412881851196
    },
    {
      "epoch": 2.1927833557128905e-05,
      "model_forward_time": 1.9087049961090088,
      "step": 22993
    },
    {
      "epoch": 2.1927833557128905e-05,
      "step": 22993,
      "training_step_time": 9.032999753952026
    },
    {
      "epoch": 2.1928787231445312e-05,
      "model_forward_time": 1.9110848903656006,
      "step": 22994
    },
    {
      "epoch": 2.1928787231445312e-05,
      "step": 22994,
      "training_step_time": 7.928397178649902
    },
    {
      "epoch": 2.192974090576172e-05,
      "model_forward_time": 1.7657184600830078,
      "step": 22995
    },
    {
      "epoch": 2.192974090576172e-05,
      "step": 22995,
      "training_step_time": 9.948806524276733
    },
    {
      "epoch": 2.1930694580078124e-05,
      "model_forward_time": 1.7727501392364502,
      "step": 22996
    },
    {
      "epoch": 2.1930694580078124e-05,
      "step": 22996,
      "training_step_time": 8.954637050628662
    },
    {
      "epoch": 2.193164825439453e-05,
      "model_forward_time": 1.7884039878845215,
      "step": 22997
    },
    {
      "epoch": 2.193164825439453e-05,
      "step": 22997,
      "training_step_time": 8.008098602294922
    },
    {
      "epoch": 2.193260192871094e-05,
      "model_forward_time": 1.7503023147583008,
      "step": 22998
    },
    {
      "epoch": 2.193260192871094e-05,
      "step": 22998,
      "training_step_time": 8.62480354309082
    },
    {
      "epoch": 2.1933555603027343e-05,
      "model_forward_time": 1.7677013874053955,
      "step": 22999
    },
    {
      "epoch": 2.1933555603027343e-05,
      "step": 22999,
      "training_step_time": 7.750528335571289
    },
    {
      "epoch": 2.193450927734375e-05,
      "grad_norm": 0.14791631698608398,
      "learning_rate": 9.140136866018704e-06,
      "loss": 0.063,
      "step": 23000
    },
    {
      "epoch": 2.193450927734375e-05,
      "model_forward_time": 1.75862455368042,
      "step": 23000
    },
    {
      "dynamics_loss_avg": 0.044949709065258504,
      "epoch": 2.193450927734375e-05,
      "step": 23000
    },
    {
      "action_loss_avg": 0.0018384049035375938,
      "epoch": 2.193450927734375e-05,
      "step": 23000
    },
    {
      "epoch": 2.193450927734375e-05,
      "step": 23000,
      "training_step_time": 7.913820505142212
    },
    {
      "epoch": 2.1935462951660157e-05,
      "model_forward_time": 1.773303508758545,
      "step": 23001
    },
    {
      "epoch": 2.1935462951660157e-05,
      "step": 23001,
      "training_step_time": 10.515773296356201
    },
    {
      "epoch": 2.193641662597656e-05,
      "model_forward_time": 1.8020334243774414,
      "step": 23002
    },
    {
      "epoch": 2.193641662597656e-05,
      "step": 23002,
      "training_step_time": 8.426777124404907
    },
    {
      "epoch": 2.193737030029297e-05,
      "model_forward_time": 1.7374377250671387,
      "step": 23003
    },
    {
      "epoch": 2.193737030029297e-05,
      "step": 23003,
      "training_step_time": 8.702723503112793
    },
    {
      "epoch": 2.1938323974609376e-05,
      "model_forward_time": 1.7646098136901855,
      "step": 23004
    },
    {
      "epoch": 2.1938323974609376e-05,
      "step": 23004,
      "training_step_time": 9.851234197616577
    },
    {
      "epoch": 2.193927764892578e-05,
      "model_forward_time": 1.7594940662384033,
      "step": 23005
    },
    {
      "epoch": 2.193927764892578e-05,
      "step": 23005,
      "training_step_time": 7.838857889175415
    },
    {
      "epoch": 2.1940231323242188e-05,
      "model_forward_time": 1.757894515991211,
      "step": 23006
    },
    {
      "epoch": 2.1940231323242188e-05,
      "step": 23006,
      "training_step_time": 8.697944164276123
    },
    {
      "epoch": 2.1941184997558595e-05,
      "model_forward_time": 1.7756202220916748,
      "step": 23007
    },
    {
      "epoch": 2.1941184997558595e-05,
      "step": 23007,
      "training_step_time": 8.915659427642822
    },
    {
      "epoch": 2.1942138671875e-05,
      "model_forward_time": 1.758829116821289,
      "step": 23008
    },
    {
      "epoch": 2.1942138671875e-05,
      "step": 23008,
      "training_step_time": 7.801154613494873
    },
    {
      "epoch": 2.1943092346191407e-05,
      "model_forward_time": 3.357576608657837,
      "step": 23009
    },
    {
      "epoch": 2.1943092346191407e-05,
      "step": 23009,
      "training_step_time": 8.365881443023682
    },
    {
      "epoch": 2.1944046020507814e-05,
      "grad_norm": 0.1773262619972229,
      "learning_rate": 9.13920955950779e-06,
      "loss": 0.0664,
      "step": 23010
    },
    {
      "epoch": 2.1944046020507814e-05,
      "model_forward_time": 1.7427315711975098,
      "step": 23010
    },
    {
      "dynamics_loss_avg": 0.04843714358285069,
      "epoch": 2.1944046020507814e-05,
      "step": 23010
    },
    {
      "action_loss_avg": 0.0025294548366218807,
      "epoch": 2.1944046020507814e-05,
      "step": 23010
    },
    {
      "epoch": 2.1944046020507814e-05,
      "step": 23010,
      "training_step_time": 8.483476161956787
    },
    {
      "epoch": 2.1944999694824218e-05,
      "model_forward_time": 1.7471215724945068,
      "step": 23011
    },
    {
      "epoch": 2.1944999694824218e-05,
      "step": 23011,
      "training_step_time": 9.948758125305176
    },
    {
      "epoch": 2.1945953369140625e-05,
      "model_forward_time": 1.3812644481658936,
      "step": 23012
    },
    {
      "epoch": 2.1945953369140625e-05,
      "step": 23012,
      "training_step_time": 8.520171165466309
    },
    {
      "epoch": 2.1946907043457033e-05,
      "model_forward_time": 1.7708590030670166,
      "step": 23013
    },
    {
      "epoch": 2.1946907043457033e-05,
      "step": 23013,
      "training_step_time": 8.70887565612793
    },
    {
      "epoch": 2.1947860717773437e-05,
      "model_forward_time": 1.7787117958068848,
      "step": 23014
    },
    {
      "epoch": 2.1947860717773437e-05,
      "step": 23014,
      "training_step_time": 9.39052963256836
    },
    {
      "epoch": 2.1948814392089844e-05,
      "model_forward_time": 1.7724263668060303,
      "step": 23015
    },
    {
      "epoch": 2.1948814392089844e-05,
      "step": 23015,
      "training_step_time": 6.794700860977173
    },
    {
      "epoch": 2.194976806640625e-05,
      "model_forward_time": 1.768660306930542,
      "step": 23016
    },
    {
      "epoch": 2.194976806640625e-05,
      "step": 23016,
      "training_step_time": 6.976829290390015
    },
    {
      "epoch": 2.1950721740722656e-05,
      "model_forward_time": 1.7873904705047607,
      "step": 23017
    },
    {
      "epoch": 2.1950721740722656e-05,
      "step": 23017,
      "training_step_time": 6.969187259674072
    },
    {
      "epoch": 2.1951675415039063e-05,
      "model_forward_time": 1.7743089199066162,
      "step": 23018
    },
    {
      "epoch": 2.1951675415039063e-05,
      "step": 23018,
      "training_step_time": 6.9765307903289795
    },
    {
      "epoch": 2.1952629089355467e-05,
      "model_forward_time": 1.7767109870910645,
      "step": 23019
    },
    {
      "epoch": 2.1952629089355467e-05,
      "step": 23019,
      "training_step_time": 6.950867414474487
    },
    {
      "epoch": 2.1953582763671875e-05,
      "grad_norm": 0.23983052372932434,
      "learning_rate": 9.13828180033914e-06,
      "loss": 0.0672,
      "step": 23020
    },
    {
      "epoch": 2.1953582763671875e-05,
      "model_forward_time": 1.775902509689331,
      "step": 23020
    },
    {
      "dynamics_loss_avg": 0.06521718855947256,
      "epoch": 2.1953582763671875e-05,
      "step": 23020
    },
    {
      "action_loss_avg": 0.003061743750004098,
      "epoch": 2.1953582763671875e-05,
      "step": 23020
    },
    {
      "epoch": 2.1953582763671875e-05,
      "step": 23020,
      "training_step_time": 6.941195726394653
    },
    {
      "epoch": 2.1954536437988282e-05,
      "model_forward_time": 1.776064395904541,
      "step": 23021
    },
    {
      "epoch": 2.1954536437988282e-05,
      "step": 23021,
      "training_step_time": 6.938990831375122
    },
    {
      "epoch": 2.1955490112304686e-05,
      "model_forward_time": 1.7697670459747314,
      "step": 23022
    },
    {
      "epoch": 2.1955490112304686e-05,
      "step": 23022,
      "training_step_time": 7.878565073013306
    },
    {
      "epoch": 2.1956443786621093e-05,
      "model_forward_time": 1.7749755382537842,
      "step": 23023
    },
    {
      "epoch": 2.1956443786621093e-05,
      "step": 23023,
      "training_step_time": 6.903828144073486
    },
    {
      "epoch": 2.19573974609375e-05,
      "model_forward_time": 1.7710354328155518,
      "step": 23024
    },
    {
      "epoch": 2.19573974609375e-05,
      "step": 23024,
      "training_step_time": 8.24179196357727
    },
    {
      "epoch": 2.1958351135253905e-05,
      "model_forward_time": 1.77028489112854,
      "step": 23025
    },
    {
      "epoch": 2.1958351135253905e-05,
      "step": 23025,
      "training_step_time": 6.935583829879761
    },
    {
      "epoch": 2.1959304809570312e-05,
      "model_forward_time": 1.7669498920440674,
      "step": 23026
    },
    {
      "epoch": 2.1959304809570312e-05,
      "step": 23026,
      "training_step_time": 6.941004037857056
    },
    {
      "epoch": 2.196025848388672e-05,
      "model_forward_time": 1.7715754508972168,
      "step": 23027
    },
    {
      "epoch": 2.196025848388672e-05,
      "step": 23027,
      "training_step_time": 6.938274145126343
    },
    {
      "epoch": 2.1961212158203124e-05,
      "model_forward_time": 1.7700104713439941,
      "step": 23028
    },
    {
      "epoch": 2.1961212158203124e-05,
      "step": 23028,
      "training_step_time": 6.898820877075195
    },
    {
      "epoch": 2.196216583251953e-05,
      "model_forward_time": 1.770007610321045,
      "step": 23029
    },
    {
      "epoch": 2.196216583251953e-05,
      "step": 23029,
      "training_step_time": 6.932833194732666
    },
    {
      "epoch": 2.196311950683594e-05,
      "grad_norm": 0.19513681530952454,
      "learning_rate": 9.137353588614213e-06,
      "loss": 0.0696,
      "step": 23030
    },
    {
      "epoch": 2.196311950683594e-05,
      "model_forward_time": 1.7818331718444824,
      "step": 23030
    },
    {
      "dynamics_loss_avg": 0.08016902022063732,
      "epoch": 2.196311950683594e-05,
      "step": 23030
    },
    {
      "action_loss_avg": 0.0033925792668014765,
      "epoch": 2.196311950683594e-05,
      "step": 23030
    },
    {
      "epoch": 2.196311950683594e-05,
      "step": 23030,
      "training_step_time": 6.96291184425354
    },
    {
      "epoch": 2.1964073181152343e-05,
      "model_forward_time": 1.757256031036377,
      "step": 23031
    },
    {
      "epoch": 2.1964073181152343e-05,
      "step": 23031,
      "training_step_time": 6.943930387496948
    },
    {
      "epoch": 2.196502685546875e-05,
      "model_forward_time": 2.861063241958618,
      "step": 23032
    },
    {
      "epoch": 2.196502685546875e-05,
      "step": 23032,
      "training_step_time": 7.825877666473389
    },
    {
      "epoch": 2.1965980529785157e-05,
      "model_forward_time": 1.7879445552825928,
      "step": 23033
    },
    {
      "epoch": 2.1965980529785157e-05,
      "step": 23033,
      "training_step_time": 7.911041736602783
    },
    {
      "epoch": 2.196693420410156e-05,
      "model_forward_time": 1.7783498764038086,
      "step": 23034
    },
    {
      "epoch": 2.196693420410156e-05,
      "step": 23034,
      "training_step_time": 7.935772180557251
    },
    {
      "epoch": 2.196788787841797e-05,
      "model_forward_time": 1.7284777164459229,
      "step": 23035
    },
    {
      "epoch": 2.196788787841797e-05,
      "step": 23035,
      "training_step_time": 6.960466623306274
    },
    {
      "epoch": 2.1968841552734376e-05,
      "model_forward_time": 1.7606146335601807,
      "step": 23036
    },
    {
      "epoch": 2.1968841552734376e-05,
      "step": 23036,
      "training_step_time": 6.977025032043457
    },
    {
      "epoch": 2.196979522705078e-05,
      "model_forward_time": 1.7532985210418701,
      "step": 23037
    },
    {
      "epoch": 2.196979522705078e-05,
      "step": 23037,
      "training_step_time": 6.95267391204834
    },
    {
      "epoch": 2.1970748901367188e-05,
      "model_forward_time": 1.7639987468719482,
      "step": 23038
    },
    {
      "epoch": 2.1970748901367188e-05,
      "step": 23038,
      "training_step_time": 6.923511981964111
    },
    {
      "epoch": 2.1971702575683595e-05,
      "model_forward_time": 1.7464985847473145,
      "step": 23039
    },
    {
      "epoch": 2.1971702575683595e-05,
      "step": 23039,
      "training_step_time": 6.894287586212158
    },
    {
      "epoch": 2.197265625e-05,
      "grad_norm": 0.1923864632844925,
      "learning_rate": 9.13642492443452e-06,
      "loss": 0.0624,
      "step": 23040
    },
    {
      "epoch": 2.197265625e-05,
      "model_forward_time": 1.7434639930725098,
      "step": 23040
    },
    {
      "dynamics_loss_avg": 0.06302741914987564,
      "epoch": 2.197265625e-05,
      "step": 23040
    },
    {
      "action_loss_avg": 0.00323935296619311,
      "epoch": 2.197265625e-05,
      "step": 23040
    },
    {
      "epoch": 2.197265625e-05,
      "step": 23040,
      "training_step_time": 6.796327829360962
    },
    {
      "epoch": 2.1973609924316407e-05,
      "model_forward_time": 2.8723459243774414,
      "step": 23041
    },
    {
      "epoch": 2.1973609924316407e-05,
      "step": 23041,
      "training_step_time": 7.922950506210327
    },
    {
      "epoch": 2.1974563598632814e-05,
      "model_forward_time": 1.772122859954834,
      "step": 23042
    },
    {
      "epoch": 2.1974563598632814e-05,
      "step": 23042,
      "training_step_time": 9.473990678787231
    },
    {
      "epoch": 2.1975517272949218e-05,
      "model_forward_time": 1.7418317794799805,
      "step": 23043
    },
    {
      "epoch": 2.1975517272949218e-05,
      "step": 23043,
      "training_step_time": 7.5736002922058105
    },
    {
      "epoch": 2.1976470947265625e-05,
      "model_forward_time": 1.751617670059204,
      "step": 23044
    },
    {
      "epoch": 2.1976470947265625e-05,
      "step": 23044,
      "training_step_time": 7.8448450565338135
    },
    {
      "epoch": 2.1977424621582033e-05,
      "model_forward_time": 1.7944679260253906,
      "step": 23045
    },
    {
      "epoch": 2.1977424621582033e-05,
      "step": 23045,
      "training_step_time": 6.922931909561157
    },
    {
      "epoch": 2.1978378295898437e-05,
      "model_forward_time": 1.7626590728759766,
      "step": 23046
    },
    {
      "epoch": 2.1978378295898437e-05,
      "step": 23046,
      "training_step_time": 6.887654781341553
    },
    {
      "epoch": 2.1979331970214844e-05,
      "model_forward_time": 1.7378029823303223,
      "step": 23047
    },
    {
      "epoch": 2.1979331970214844e-05,
      "step": 23047,
      "training_step_time": 6.918580770492554
    },
    {
      "epoch": 2.198028564453125e-05,
      "model_forward_time": 1.7487273216247559,
      "step": 23048
    },
    {
      "epoch": 2.198028564453125e-05,
      "step": 23048,
      "training_step_time": 6.963984727859497
    },
    {
      "epoch": 2.1981239318847656e-05,
      "model_forward_time": 1.8032822608947754,
      "step": 23049
    },
    {
      "epoch": 2.1981239318847656e-05,
      "step": 23049,
      "training_step_time": 6.941437005996704
    },
    {
      "epoch": 2.1982192993164063e-05,
      "grad_norm": 0.25814083218574524,
      "learning_rate": 9.135495807901615e-06,
      "loss": 0.0634,
      "step": 23050
    },
    {
      "epoch": 2.1982192993164063e-05,
      "model_forward_time": 2.8830249309539795,
      "step": 23050
    },
    {
      "dynamics_loss_avg": 0.0659223148599267,
      "epoch": 2.1982192993164063e-05,
      "step": 23050
    },
    {
      "action_loss_avg": 0.0034737173235043885,
      "epoch": 2.1982192993164063e-05,
      "step": 23050
    },
    {
      "epoch": 2.1982192993164063e-05,
      "step": 23050,
      "training_step_time": 7.865663766860962
    },
    {
      "epoch": 2.1983146667480467e-05,
      "model_forward_time": 1.842221736907959,
      "step": 23051
    },
    {
      "epoch": 2.1983146667480467e-05,
      "step": 23051,
      "training_step_time": 11.142841339111328
    },
    {
      "epoch": 2.1984100341796875e-05,
      "model_forward_time": 1.7495088577270508,
      "step": 23052
    },
    {
      "epoch": 2.1984100341796875e-05,
      "step": 23052,
      "training_step_time": 7.976479530334473
    },
    {
      "epoch": 2.1985054016113282e-05,
      "model_forward_time": 1.7411468029022217,
      "step": 23053
    },
    {
      "epoch": 2.1985054016113282e-05,
      "step": 23053,
      "training_step_time": 7.8537914752960205
    },
    {
      "epoch": 2.1986007690429686e-05,
      "model_forward_time": 1.7434298992156982,
      "step": 23054
    },
    {
      "epoch": 2.1986007690429686e-05,
      "step": 23054,
      "training_step_time": 6.962024927139282
    },
    {
      "epoch": 2.1986961364746093e-05,
      "model_forward_time": 1.7428951263427734,
      "step": 23055
    },
    {
      "epoch": 2.1986961364746093e-05,
      "step": 23055,
      "training_step_time": 6.959037780761719
    },
    {
      "epoch": 2.19879150390625e-05,
      "model_forward_time": 1.746997594833374,
      "step": 23056
    },
    {
      "epoch": 2.19879150390625e-05,
      "step": 23056,
      "training_step_time": 6.961038589477539
    },
    {
      "epoch": 2.1988868713378905e-05,
      "model_forward_time": 1.7465667724609375,
      "step": 23057
    },
    {
      "epoch": 2.1988868713378905e-05,
      "step": 23057,
      "training_step_time": 6.865708827972412
    },
    {
      "epoch": 2.1989822387695312e-05,
      "model_forward_time": 1.7230968475341797,
      "step": 23058
    },
    {
      "epoch": 2.1989822387695312e-05,
      "step": 23058,
      "training_step_time": 6.895289182662964
    },
    {
      "epoch": 2.199077606201172e-05,
      "model_forward_time": 2.8622748851776123,
      "step": 23059
    },
    {
      "epoch": 2.199077606201172e-05,
      "step": 23059,
      "training_step_time": 7.927648067474365
    },
    {
      "epoch": 2.1991729736328124e-05,
      "grad_norm": 0.23781001567840576,
      "learning_rate": 9.134566239117107e-06,
      "loss": 0.0643,
      "step": 23060
    },
    {
      "epoch": 2.1991729736328124e-05,
      "model_forward_time": 1.7518274784088135,
      "step": 23060
    },
    {
      "dynamics_loss_avg": 0.07477644374594092,
      "epoch": 2.1991729736328124e-05,
      "step": 23060
    },
    {
      "action_loss_avg": 0.0032879871490877123,
      "epoch": 2.1991729736328124e-05,
      "step": 23060
    },
    {
      "epoch": 2.1991729736328124e-05,
      "step": 23060,
      "training_step_time": 8.569824934005737
    },
    {
      "epoch": 2.199268341064453e-05,
      "model_forward_time": 1.7310152053833008,
      "step": 23061
    },
    {
      "epoch": 2.199268341064453e-05,
      "step": 23061,
      "training_step_time": 7.940576076507568
    },
    {
      "epoch": 2.199363708496094e-05,
      "model_forward_time": 1.738877773284912,
      "step": 23062
    },
    {
      "epoch": 2.199363708496094e-05,
      "step": 23062,
      "training_step_time": 7.8457512855529785
    },
    {
      "epoch": 2.1994590759277343e-05,
      "model_forward_time": 1.7340812683105469,
      "step": 23063
    },
    {
      "epoch": 2.1994590759277343e-05,
      "step": 23063,
      "training_step_time": 6.908384561538696
    },
    {
      "epoch": 2.199554443359375e-05,
      "model_forward_time": 1.7288122177124023,
      "step": 23064
    },
    {
      "epoch": 2.199554443359375e-05,
      "step": 23064,
      "training_step_time": 6.904025316238403
    },
    {
      "epoch": 2.1996498107910157e-05,
      "model_forward_time": 1.7398123741149902,
      "step": 23065
    },
    {
      "epoch": 2.1996498107910157e-05,
      "step": 23065,
      "training_step_time": 6.945725202560425
    },
    {
      "epoch": 2.199745178222656e-05,
      "model_forward_time": 1.739166021347046,
      "step": 23066
    },
    {
      "epoch": 2.199745178222656e-05,
      "step": 23066,
      "training_step_time": 6.925512075424194
    },
    {
      "epoch": 2.199840545654297e-05,
      "model_forward_time": 1.735426425933838,
      "step": 23067
    },
    {
      "epoch": 2.199840545654297e-05,
      "step": 23067,
      "training_step_time": 6.894988298416138
    },
    {
      "epoch": 2.1999359130859376e-05,
      "model_forward_time": 2.8572945594787598,
      "step": 23068
    },
    {
      "epoch": 2.1999359130859376e-05,
      "step": 23068,
      "training_step_time": 7.675128221511841
    },
    {
      "epoch": 2.200031280517578e-05,
      "model_forward_time": 1.7332112789154053,
      "step": 23069
    },
    {
      "epoch": 2.200031280517578e-05,
      "step": 23069,
      "training_step_time": 7.91057562828064
    },
    {
      "epoch": 2.2001266479492188e-05,
      "grad_norm": 0.24615852534770966,
      "learning_rate": 9.133636218182652e-06,
      "loss": 0.0624,
      "step": 23070
    },
    {
      "epoch": 2.2001266479492188e-05,
      "model_forward_time": 1.7352592945098877,
      "step": 23070
    },
    {
      "dynamics_loss_avg": 0.06958203129470349,
      "epoch": 2.2001266479492188e-05,
      "step": 23070
    },
    {
      "action_loss_avg": 0.0026960478513501584,
      "epoch": 2.2001266479492188e-05,
      "step": 23070
    },
    {
      "epoch": 2.2001266479492188e-05,
      "step": 23070,
      "training_step_time": 8.088550329208374
    },
    {
      "epoch": 2.2002220153808595e-05,
      "model_forward_time": 1.7081067562103271,
      "step": 23071
    },
    {
      "epoch": 2.2002220153808595e-05,
      "step": 23071,
      "training_step_time": 7.425259828567505
    },
    {
      "epoch": 2.2003173828125e-05,
      "model_forward_time": 1.7002208232879639,
      "step": 23072
    },
    {
      "epoch": 2.2003173828125e-05,
      "step": 23072,
      "training_step_time": 7.745121240615845
    },
    {
      "epoch": 2.2004127502441407e-05,
      "model_forward_time": 1.710758924484253,
      "step": 23073
    },
    {
      "epoch": 2.2004127502441407e-05,
      "step": 23073,
      "training_step_time": 6.975130796432495
    },
    {
      "epoch": 2.2005081176757814e-05,
      "model_forward_time": 1.7233765125274658,
      "step": 23074
    },
    {
      "epoch": 2.2005081176757814e-05,
      "step": 23074,
      "training_step_time": 7.006889343261719
    },
    {
      "epoch": 2.2006034851074218e-05,
      "model_forward_time": 1.7227365970611572,
      "step": 23075
    },
    {
      "epoch": 2.2006034851074218e-05,
      "step": 23075,
      "training_step_time": 6.963749170303345
    },
    {
      "epoch": 2.2006988525390625e-05,
      "model_forward_time": 1.7107996940612793,
      "step": 23076
    },
    {
      "epoch": 2.2006988525390625e-05,
      "step": 23076,
      "training_step_time": 7.060620307922363
    },
    {
      "epoch": 2.2007942199707033e-05,
      "model_forward_time": 1.7136690616607666,
      "step": 23077
    },
    {
      "epoch": 2.2007942199707033e-05,
      "step": 23077,
      "training_step_time": 8.533756732940674
    },
    {
      "epoch": 2.2008895874023437e-05,
      "model_forward_time": 1.7860395908355713,
      "step": 23078
    },
    {
      "epoch": 2.2008895874023437e-05,
      "step": 23078,
      "training_step_time": 8.000679969787598
    },
    {
      "epoch": 2.2009849548339844e-05,
      "model_forward_time": 1.712836742401123,
      "step": 23079
    },
    {
      "epoch": 2.2009849548339844e-05,
      "step": 23079,
      "training_step_time": 8.020066022872925
    },
    {
      "epoch": 2.201080322265625e-05,
      "grad_norm": 0.21688392758369446,
      "learning_rate": 9.132705745199953e-06,
      "loss": 0.0615,
      "step": 23080
    },
    {
      "epoch": 2.201080322265625e-05,
      "model_forward_time": 1.7065961360931396,
      "step": 23080
    },
    {
      "dynamics_loss_avg": 0.05427489373832941,
      "epoch": 2.201080322265625e-05,
      "step": 23080
    },
    {
      "action_loss_avg": 0.003921848325990141,
      "epoch": 2.201080322265625e-05,
      "step": 23080
    },
    {
      "epoch": 2.201080322265625e-05,
      "step": 23080,
      "training_step_time": 8.04524540901184
    },
    {
      "epoch": 2.2011756896972656e-05,
      "model_forward_time": 1.7115869522094727,
      "step": 23081
    },
    {
      "epoch": 2.2011756896972656e-05,
      "step": 23081,
      "training_step_time": 7.020471811294556
    },
    {
      "epoch": 2.2012710571289063e-05,
      "model_forward_time": 1.6960456371307373,
      "step": 23082
    },
    {
      "epoch": 2.2012710571289063e-05,
      "step": 23082,
      "training_step_time": 7.754804372787476
    },
    {
      "epoch": 2.2013664245605467e-05,
      "model_forward_time": 1.7087013721466064,
      "step": 23083
    },
    {
      "epoch": 2.2013664245605467e-05,
      "step": 23083,
      "training_step_time": 6.986449718475342
    },
    {
      "epoch": 2.2014617919921875e-05,
      "model_forward_time": 1.716480016708374,
      "step": 23084
    },
    {
      "epoch": 2.2014617919921875e-05,
      "step": 23084,
      "training_step_time": 6.956306219100952
    },
    {
      "epoch": 2.2015571594238282e-05,
      "model_forward_time": 1.7117176055908203,
      "step": 23085
    },
    {
      "epoch": 2.2015571594238282e-05,
      "step": 23085,
      "training_step_time": 6.998986005783081
    },
    {
      "epoch": 2.2016525268554686e-05,
      "model_forward_time": 1.7088923454284668,
      "step": 23086
    },
    {
      "epoch": 2.2016525268554686e-05,
      "step": 23086,
      "training_step_time": 7.016015529632568
    },
    {
      "epoch": 2.2017478942871094e-05,
      "model_forward_time": 2.3375115394592285,
      "step": 23087
    },
    {
      "epoch": 2.2017478942871094e-05,
      "step": 23087,
      "training_step_time": 7.908789396286011
    },
    {
      "epoch": 2.20184326171875e-05,
      "model_forward_time": 1.7173035144805908,
      "step": 23088
    },
    {
      "epoch": 2.20184326171875e-05,
      "step": 23088,
      "training_step_time": 12.767868995666504
    },
    {
      "epoch": 2.2019386291503905e-05,
      "model_forward_time": 1.7138001918792725,
      "step": 23089
    },
    {
      "epoch": 2.2019386291503905e-05,
      "step": 23089,
      "training_step_time": 8.075815439224243
    },
    {
      "epoch": 2.2020339965820312e-05,
      "grad_norm": 0.151445209980011,
      "learning_rate": 9.131774820270769e-06,
      "loss": 0.0657,
      "step": 23090
    },
    {
      "epoch": 2.2020339965820312e-05,
      "model_forward_time": 1.6919260025024414,
      "step": 23090
    },
    {
      "dynamics_loss_avg": 0.060002457816153765,
      "epoch": 2.2020339965820312e-05,
      "step": 23090
    },
    {
      "action_loss_avg": 0.003495169326197356,
      "epoch": 2.2020339965820312e-05,
      "step": 23090
    },
    {
      "epoch": 2.2020339965820312e-05,
      "step": 23090,
      "training_step_time": 6.926621198654175
    },
    {
      "epoch": 2.202129364013672e-05,
      "model_forward_time": 1.7126154899597168,
      "step": 23091
    },
    {
      "epoch": 2.202129364013672e-05,
      "step": 23091,
      "training_step_time": 7.910431861877441
    },
    {
      "epoch": 2.2022247314453124e-05,
      "model_forward_time": 1.7129099369049072,
      "step": 23092
    },
    {
      "epoch": 2.2022247314453124e-05,
      "step": 23092,
      "training_step_time": 7.0058677196502686
    },
    {
      "epoch": 2.202320098876953e-05,
      "model_forward_time": 1.7118427753448486,
      "step": 23093
    },
    {
      "epoch": 2.202320098876953e-05,
      "step": 23093,
      "training_step_time": 7.019463300704956
    },
    {
      "epoch": 2.202415466308594e-05,
      "model_forward_time": 1.7053604125976562,
      "step": 23094
    },
    {
      "epoch": 2.202415466308594e-05,
      "step": 23094,
      "training_step_time": 6.970041275024414
    },
    {
      "epoch": 2.2025108337402343e-05,
      "model_forward_time": 1.715221881866455,
      "step": 23095
    },
    {
      "epoch": 2.2025108337402343e-05,
      "step": 23095,
      "training_step_time": 6.9891722202301025
    },
    {
      "epoch": 2.202606201171875e-05,
      "model_forward_time": 2.3419322967529297,
      "step": 23096
    },
    {
      "epoch": 2.202606201171875e-05,
      "step": 23096,
      "training_step_time": 7.891126871109009
    },
    {
      "epoch": 2.2027015686035157e-05,
      "model_forward_time": 1.7134811878204346,
      "step": 23097
    },
    {
      "epoch": 2.2027015686035157e-05,
      "step": 23097,
      "training_step_time": 12.593520164489746
    },
    {
      "epoch": 2.202796936035156e-05,
      "model_forward_time": 1.7154974937438965,
      "step": 23098
    },
    {
      "epoch": 2.202796936035156e-05,
      "step": 23098,
      "training_step_time": 8.082590103149414
    },
    {
      "epoch": 2.202892303466797e-05,
      "model_forward_time": 1.7212803363800049,
      "step": 23099
    },
    {
      "epoch": 2.202892303466797e-05,
      "step": 23099,
      "training_step_time": 7.835446834564209
    },
    {
      "epoch": 2.2029876708984376e-05,
      "grad_norm": 0.16604572534561157,
      "learning_rate": 9.130843443496902e-06,
      "loss": 0.0627,
      "step": 23100
    },
    {
      "epoch": 2.2029876708984376e-05,
      "model_forward_time": 1.730222463607788,
      "step": 23100
    },
    {
      "dynamics_loss_avg": 0.07055513672530651,
      "epoch": 2.2029876708984376e-05,
      "step": 23100
    },
    {
      "action_loss_avg": 0.004434049146948382,
      "epoch": 2.2029876708984376e-05,
      "step": 23100
    },
    {
      "epoch": 2.2029876708984376e-05,
      "step": 23100,
      "training_step_time": 7.852905750274658
    },
    {
      "epoch": 2.203083038330078e-05,
      "model_forward_time": 1.713076114654541,
      "step": 23101
    },
    {
      "epoch": 2.203083038330078e-05,
      "step": 23101,
      "training_step_time": 6.941547870635986
    },
    {
      "epoch": 2.2031784057617188e-05,
      "model_forward_time": 1.718477725982666,
      "step": 23102
    },
    {
      "epoch": 2.2031784057617188e-05,
      "step": 23102,
      "training_step_time": 6.992100715637207
    },
    {
      "epoch": 2.2032737731933595e-05,
      "model_forward_time": 1.7193613052368164,
      "step": 23103
    },
    {
      "epoch": 2.2032737731933595e-05,
      "step": 23103,
      "training_step_time": 6.784354209899902
    },
    {
      "epoch": 2.203369140625e-05,
      "model_forward_time": 1.721858263015747,
      "step": 23104
    },
    {
      "epoch": 2.203369140625e-05,
      "step": 23104,
      "training_step_time": 6.920310020446777
    },
    {
      "epoch": 2.2034645080566407e-05,
      "model_forward_time": 2.30600905418396,
      "step": 23105
    },
    {
      "epoch": 2.2034645080566407e-05,
      "step": 23105,
      "training_step_time": 7.856360912322998
    },
    {
      "epoch": 2.2035598754882814e-05,
      "model_forward_time": 1.7181122303009033,
      "step": 23106
    },
    {
      "epoch": 2.2035598754882814e-05,
      "step": 23106,
      "training_step_time": 10.867736577987671
    },
    {
      "epoch": 2.2036552429199218e-05,
      "model_forward_time": 1.7166776657104492,
      "step": 23107
    },
    {
      "epoch": 2.2036552429199218e-05,
      "step": 23107,
      "training_step_time": 8.034556865692139
    },
    {
      "epoch": 2.2037506103515626e-05,
      "model_forward_time": 1.7245736122131348,
      "step": 23108
    },
    {
      "epoch": 2.2037506103515626e-05,
      "step": 23108,
      "training_step_time": 7.85324239730835
    },
    {
      "epoch": 2.2038459777832033e-05,
      "model_forward_time": 1.715226173400879,
      "step": 23109
    },
    {
      "epoch": 2.2038459777832033e-05,
      "step": 23109,
      "training_step_time": 6.9682722091674805
    },
    {
      "epoch": 2.2039413452148437e-05,
      "grad_norm": 0.22222678363323212,
      "learning_rate": 9.129911614980206e-06,
      "loss": 0.0603,
      "step": 23110
    },
    {
      "epoch": 2.2039413452148437e-05,
      "model_forward_time": 1.708505392074585,
      "step": 23110
    },
    {
      "dynamics_loss_avg": 0.06671283338218928,
      "epoch": 2.2039413452148437e-05,
      "step": 23110
    },
    {
      "action_loss_avg": 0.003773220826406032,
      "epoch": 2.2039413452148437e-05,
      "step": 23110
    },
    {
      "epoch": 2.2039413452148437e-05,
      "step": 23110,
      "training_step_time": 7.959851503372192
    },
    {
      "epoch": 2.2040367126464844e-05,
      "model_forward_time": 1.7137434482574463,
      "step": 23111
    },
    {
      "epoch": 2.2040367126464844e-05,
      "step": 23111,
      "training_step_time": 6.985983371734619
    },
    {
      "epoch": 2.204132080078125e-05,
      "model_forward_time": 1.7266936302185059,
      "step": 23112
    },
    {
      "epoch": 2.204132080078125e-05,
      "step": 23112,
      "training_step_time": 7.0026233196258545
    },
    {
      "epoch": 2.2042274475097656e-05,
      "model_forward_time": 1.6911633014678955,
      "step": 23113
    },
    {
      "epoch": 2.2042274475097656e-05,
      "step": 23113,
      "training_step_time": 6.872109413146973
    },
    {
      "epoch": 2.2043228149414063e-05,
      "model_forward_time": 2.2522387504577637,
      "step": 23114
    },
    {
      "epoch": 2.2043228149414063e-05,
      "step": 23114,
      "training_step_time": 8.315153121948242
    },
    {
      "epoch": 2.2044181823730467e-05,
      "model_forward_time": 1.3458755016326904,
      "step": 23115
    },
    {
      "epoch": 2.2044181823730467e-05,
      "step": 23115,
      "training_step_time": 9.870048522949219
    },
    {
      "epoch": 2.2045135498046875e-05,
      "model_forward_time": 1.7361335754394531,
      "step": 23116
    },
    {
      "epoch": 2.2045135498046875e-05,
      "step": 23116,
      "training_step_time": 8.041089057922363
    },
    {
      "epoch": 2.2046089172363282e-05,
      "model_forward_time": 1.7403855323791504,
      "step": 23117
    },
    {
      "epoch": 2.2046089172363282e-05,
      "step": 23117,
      "training_step_time": 7.8397510051727295
    },
    {
      "epoch": 2.2047042846679686e-05,
      "model_forward_time": 1.7287938594818115,
      "step": 23118
    },
    {
      "epoch": 2.2047042846679686e-05,
      "step": 23118,
      "training_step_time": 6.991772413253784
    },
    {
      "epoch": 2.2047996520996094e-05,
      "model_forward_time": 1.7292814254760742,
      "step": 23119
    },
    {
      "epoch": 2.2047996520996094e-05,
      "step": 23119,
      "training_step_time": 7.90777325630188
    },
    {
      "epoch": 2.20489501953125e-05,
      "grad_norm": 0.2752336263656616,
      "learning_rate": 9.128979334822586e-06,
      "loss": 0.0647,
      "step": 23120
    },
    {
      "epoch": 2.20489501953125e-05,
      "model_forward_time": 1.7235679626464844,
      "step": 23120
    },
    {
      "dynamics_loss_avg": 0.05648498740047216,
      "epoch": 2.20489501953125e-05,
      "step": 23120
    },
    {
      "action_loss_avg": 0.0047175329353194686,
      "epoch": 2.20489501953125e-05,
      "step": 23120
    },
    {
      "epoch": 2.20489501953125e-05,
      "step": 23120,
      "training_step_time": 6.957341194152832
    },
    {
      "epoch": 2.2049903869628905e-05,
      "model_forward_time": 1.7383403778076172,
      "step": 23121
    },
    {
      "epoch": 2.2049903869628905e-05,
      "step": 23121,
      "training_step_time": 7.016077280044556
    },
    {
      "epoch": 2.2050857543945312e-05,
      "model_forward_time": 1.726562261581421,
      "step": 23122
    },
    {
      "epoch": 2.2050857543945312e-05,
      "step": 23122,
      "training_step_time": 6.979647874832153
    },
    {
      "epoch": 2.205181121826172e-05,
      "model_forward_time": 1.7245876789093018,
      "step": 23123
    },
    {
      "epoch": 2.205181121826172e-05,
      "step": 23123,
      "training_step_time": 7.2822651863098145
    },
    {
      "epoch": 2.2052764892578124e-05,
      "model_forward_time": 2.3248820304870605,
      "step": 23124
    },
    {
      "epoch": 2.2052764892578124e-05,
      "step": 23124,
      "training_step_time": 7.845884799957275
    },
    {
      "epoch": 2.205371856689453e-05,
      "model_forward_time": 1.7337040901184082,
      "step": 23125
    },
    {
      "epoch": 2.205371856689453e-05,
      "step": 23125,
      "training_step_time": 7.981580018997192
    },
    {
      "epoch": 2.205467224121094e-05,
      "model_forward_time": 1.718552827835083,
      "step": 23126
    },
    {
      "epoch": 2.205467224121094e-05,
      "step": 23126,
      "training_step_time": 7.909888505935669
    },
    {
      "epoch": 2.2055625915527343e-05,
      "model_forward_time": 1.7198870182037354,
      "step": 23127
    },
    {
      "epoch": 2.2055625915527343e-05,
      "step": 23127,
      "training_step_time": 6.9440062046051025
    },
    {
      "epoch": 2.205657958984375e-05,
      "model_forward_time": 1.7280645370483398,
      "step": 23128
    },
    {
      "epoch": 2.205657958984375e-05,
      "step": 23128,
      "training_step_time": 7.795568943023682
    },
    {
      "epoch": 2.2057533264160158e-05,
      "model_forward_time": 1.721574306488037,
      "step": 23129
    },
    {
      "epoch": 2.2057533264160158e-05,
      "step": 23129,
      "training_step_time": 6.916093826293945
    },
    {
      "epoch": 2.205848693847656e-05,
      "grad_norm": 0.18780647218227386,
      "learning_rate": 9.128046603125992e-06,
      "loss": 0.0651,
      "step": 23130
    },
    {
      "epoch": 2.205848693847656e-05,
      "model_forward_time": 1.718982219696045,
      "step": 23130
    },
    {
      "dynamics_loss_avg": 0.059633240289986136,
      "epoch": 2.205848693847656e-05,
      "step": 23130
    },
    {
      "action_loss_avg": 0.002700350759550929,
      "epoch": 2.205848693847656e-05,
      "step": 23130
    },
    {
      "epoch": 2.205848693847656e-05,
      "step": 23130,
      "training_step_time": 6.900636434555054
    },
    {
      "epoch": 2.205944061279297e-05,
      "model_forward_time": 1.7201600074768066,
      "step": 23131
    },
    {
      "epoch": 2.205944061279297e-05,
      "step": 23131,
      "training_step_time": 6.967636585235596
    },
    {
      "epoch": 2.2060394287109376e-05,
      "model_forward_time": 1.7257990837097168,
      "step": 23132
    },
    {
      "epoch": 2.2060394287109376e-05,
      "step": 23132,
      "training_step_time": 7.327599048614502
    },
    {
      "epoch": 2.206134796142578e-05,
      "model_forward_time": 2.2911391258239746,
      "step": 23133
    },
    {
      "epoch": 2.206134796142578e-05,
      "step": 23133,
      "training_step_time": 8.522449493408203
    },
    {
      "epoch": 2.2062301635742188e-05,
      "model_forward_time": 0.9931516647338867,
      "step": 23134
    },
    {
      "epoch": 2.2062301635742188e-05,
      "step": 23134,
      "training_step_time": 9.736875295639038
    },
    {
      "epoch": 2.2063255310058595e-05,
      "model_forward_time": 1.7197742462158203,
      "step": 23135
    },
    {
      "epoch": 2.2063255310058595e-05,
      "step": 23135,
      "training_step_time": 7.971799612045288
    },
    {
      "epoch": 2.2064208984375e-05,
      "model_forward_time": 1.7179481983184814,
      "step": 23136
    },
    {
      "epoch": 2.2064208984375e-05,
      "step": 23136,
      "training_step_time": 7.8008787631988525
    },
    {
      "epoch": 2.2065162658691407e-05,
      "model_forward_time": 1.7257540225982666,
      "step": 23137
    },
    {
      "epoch": 2.2065162658691407e-05,
      "step": 23137,
      "training_step_time": 6.743878364562988
    },
    {
      "epoch": 2.2066116333007814e-05,
      "model_forward_time": 1.7235004901885986,
      "step": 23138
    },
    {
      "epoch": 2.2066116333007814e-05,
      "step": 23138,
      "training_step_time": 7.864259958267212
    },
    {
      "epoch": 2.2067070007324218e-05,
      "model_forward_time": 1.7236568927764893,
      "step": 23139
    },
    {
      "epoch": 2.2067070007324218e-05,
      "step": 23139,
      "training_step_time": 6.946058034896851
    },
    {
      "epoch": 2.2068023681640626e-05,
      "grad_norm": 0.11490949243307114,
      "learning_rate": 9.12711341999243e-06,
      "loss": 0.0655,
      "step": 23140
    },
    {
      "epoch": 2.2068023681640626e-05,
      "model_forward_time": 1.7171916961669922,
      "step": 23140
    },
    {
      "dynamics_loss_avg": 0.06034660488367081,
      "epoch": 2.2068023681640626e-05,
      "step": 23140
    },
    {
      "action_loss_avg": 0.002371583110652864,
      "epoch": 2.2068023681640626e-05,
      "step": 23140
    },
    {
      "epoch": 2.2068023681640626e-05,
      "step": 23140,
      "training_step_time": 6.939507722854614
    },
    {
      "epoch": 2.206897735595703e-05,
      "model_forward_time": 1.7347607612609863,
      "step": 23141
    },
    {
      "epoch": 2.206897735595703e-05,
      "step": 23141,
      "training_step_time": 7.3297059535980225
    },
    {
      "epoch": 2.2069931030273437e-05,
      "model_forward_time": 2.304138422012329,
      "step": 23142
    },
    {
      "epoch": 2.2069931030273437e-05,
      "step": 23142,
      "training_step_time": 8.750176191329956
    },
    {
      "epoch": 2.2070884704589844e-05,
      "model_forward_time": 1.718846321105957,
      "step": 23143
    },
    {
      "epoch": 2.2070884704589844e-05,
      "step": 23143,
      "training_step_time": 9.311631441116333
    },
    {
      "epoch": 2.207183837890625e-05,
      "model_forward_time": 1.718275785446167,
      "step": 23144
    },
    {
      "epoch": 2.207183837890625e-05,
      "step": 23144,
      "training_step_time": 8.810071229934692
    },
    {
      "epoch": 2.2072792053222656e-05,
      "model_forward_time": 1.7291383743286133,
      "step": 23145
    },
    {
      "epoch": 2.2072792053222656e-05,
      "step": 23145,
      "training_step_time": 7.733345985412598
    },
    {
      "epoch": 2.2073745727539063e-05,
      "model_forward_time": 1.7229797840118408,
      "step": 23146
    },
    {
      "epoch": 2.2073745727539063e-05,
      "step": 23146,
      "training_step_time": 6.91838264465332
    },
    {
      "epoch": 2.2074699401855467e-05,
      "model_forward_time": 1.7221601009368896,
      "step": 23147
    },
    {
      "epoch": 2.2074699401855467e-05,
      "step": 23147,
      "training_step_time": 7.810096263885498
    },
    {
      "epoch": 2.2075653076171875e-05,
      "model_forward_time": 1.7183077335357666,
      "step": 23148
    },
    {
      "epoch": 2.2075653076171875e-05,
      "step": 23148,
      "training_step_time": 6.889018297195435
    },
    {
      "epoch": 2.2076606750488282e-05,
      "model_forward_time": 1.7335381507873535,
      "step": 23149
    },
    {
      "epoch": 2.2076606750488282e-05,
      "step": 23149,
      "training_step_time": 6.69009256362915
    },
    {
      "epoch": 2.2077560424804686e-05,
      "grad_norm": 0.17965218424797058,
      "learning_rate": 9.126179785523949e-06,
      "loss": 0.0676,
      "step": 23150
    },
    {
      "epoch": 2.2077560424804686e-05,
      "model_forward_time": 1.7110521793365479,
      "step": 23150
    },
    {
      "dynamics_loss_avg": 0.05245340745896101,
      "epoch": 2.2077560424804686e-05,
      "step": 23150
    },
    {
      "action_loss_avg": 0.00256919979583472,
      "epoch": 2.2077560424804686e-05,
      "step": 23150
    },
    {
      "epoch": 2.2077560424804686e-05,
      "step": 23150,
      "training_step_time": 6.891663074493408
    },
    {
      "epoch": 2.2078514099121094e-05,
      "model_forward_time": 1.7540314197540283,
      "step": 23151
    },
    {
      "epoch": 2.2078514099121094e-05,
      "step": 23151,
      "training_step_time": 10.02500605583191
    },
    {
      "epoch": 2.20794677734375e-05,
      "model_forward_time": 1.7109901905059814,
      "step": 23152
    },
    {
      "epoch": 2.20794677734375e-05,
      "step": 23152,
      "training_step_time": 8.625095129013062
    },
    {
      "epoch": 2.2080421447753905e-05,
      "model_forward_time": 1.6981208324432373,
      "step": 23153
    },
    {
      "epoch": 2.2080421447753905e-05,
      "step": 23153,
      "training_step_time": 8.502132177352905
    },
    {
      "epoch": 2.2081375122070312e-05,
      "model_forward_time": 1.711165428161621,
      "step": 23154
    },
    {
      "epoch": 2.2081375122070312e-05,
      "step": 23154,
      "training_step_time": 7.839041709899902
    },
    {
      "epoch": 2.208232879638672e-05,
      "model_forward_time": 1.7116291522979736,
      "step": 23155
    },
    {
      "epoch": 2.208232879638672e-05,
      "step": 23155,
      "training_step_time": 6.972118616104126
    },
    {
      "epoch": 2.2083282470703124e-05,
      "model_forward_time": 1.7190141677856445,
      "step": 23156
    },
    {
      "epoch": 2.2083282470703124e-05,
      "step": 23156,
      "training_step_time": 7.8406970500946045
    },
    {
      "epoch": 2.208423614501953e-05,
      "model_forward_time": 1.7069387435913086,
      "step": 23157
    },
    {
      "epoch": 2.208423614501953e-05,
      "step": 23157,
      "training_step_time": 6.941579341888428
    },
    {
      "epoch": 2.208518981933594e-05,
      "model_forward_time": 1.704772710800171,
      "step": 23158
    },
    {
      "epoch": 2.208518981933594e-05,
      "step": 23158,
      "training_step_time": 6.940845966339111
    },
    {
      "epoch": 2.2086143493652343e-05,
      "model_forward_time": 0.9922211170196533,
      "step": 23159
    },
    {
      "epoch": 2.2086143493652343e-05,
      "step": 23159,
      "training_step_time": 6.966227054595947
    },
    {
      "epoch": 2.208709716796875e-05,
      "grad_norm": 0.1801489144563675,
      "learning_rate": 9.125245699822651e-06,
      "loss": 0.066,
      "step": 23160
    },
    {
      "epoch": 2.208709716796875e-05,
      "model_forward_time": 1.7082476615905762,
      "step": 23160
    },
    {
      "dynamics_loss_avg": 0.04623666713014245,
      "epoch": 2.208709716796875e-05,
      "step": 23160
    },
    {
      "action_loss_avg": 0.0034502645721659064,
      "epoch": 2.208709716796875e-05,
      "step": 23160
    },
    {
      "epoch": 2.208709716796875e-05,
      "step": 23160,
      "training_step_time": 9.052107810974121
    },
    {
      "epoch": 2.2088050842285158e-05,
      "model_forward_time": 2.258369207382202,
      "step": 23161
    },
    {
      "epoch": 2.2088050842285158e-05,
      "step": 23161,
      "training_step_time": 9.240640640258789
    },
    {
      "epoch": 2.208900451660156e-05,
      "model_forward_time": 1.6939146518707275,
      "step": 23162
    },
    {
      "epoch": 2.208900451660156e-05,
      "step": 23162,
      "training_step_time": 7.962062358856201
    },
    {
      "epoch": 2.208995819091797e-05,
      "model_forward_time": 1.7011573314666748,
      "step": 23163
    },
    {
      "epoch": 2.208995819091797e-05,
      "step": 23163,
      "training_step_time": 7.857917547225952
    },
    {
      "epoch": 2.2090911865234376e-05,
      "model_forward_time": 1.7053742408752441,
      "step": 23164
    },
    {
      "epoch": 2.2090911865234376e-05,
      "step": 23164,
      "training_step_time": 7.591280221939087
    },
    {
      "epoch": 2.209186553955078e-05,
      "model_forward_time": 1.7091584205627441,
      "step": 23165
    },
    {
      "epoch": 2.209186553955078e-05,
      "step": 23165,
      "training_step_time": 7.832667589187622
    },
    {
      "epoch": 2.2092819213867188e-05,
      "model_forward_time": 1.7161304950714111,
      "step": 23166
    },
    {
      "epoch": 2.2092819213867188e-05,
      "step": 23166,
      "training_step_time": 6.876345157623291
    },
    {
      "epoch": 2.2093772888183595e-05,
      "model_forward_time": 1.7093753814697266,
      "step": 23167
    },
    {
      "epoch": 2.2093772888183595e-05,
      "step": 23167,
      "training_step_time": 6.914112329483032
    },
    {
      "epoch": 2.20947265625e-05,
      "model_forward_time": 1.7110047340393066,
      "step": 23168
    },
    {
      "epoch": 2.20947265625e-05,
      "step": 23168,
      "training_step_time": 6.892865896224976
    },
    {
      "epoch": 2.2095680236816407e-05,
      "model_forward_time": 1.7176077365875244,
      "step": 23169
    },
    {
      "epoch": 2.2095680236816407e-05,
      "step": 23169,
      "training_step_time": 8.137061595916748
    },
    {
      "epoch": 2.2096633911132814e-05,
      "grad_norm": 0.21644191443920135,
      "learning_rate": 9.124311162990684e-06,
      "loss": 0.0617,
      "step": 23170
    },
    {
      "epoch": 2.2096633911132814e-05,
      "model_forward_time": 2.2885844707489014,
      "step": 23170
    },
    {
      "dynamics_loss_avg": 0.05698457174003124,
      "epoch": 2.2096633911132814e-05,
      "step": 23170
    },
    {
      "action_loss_avg": 0.002477187894692179,
      "epoch": 2.2096633911132814e-05,
      "step": 23170
    },
    {
      "epoch": 2.2096633911132814e-05,
      "step": 23170,
      "training_step_time": 9.708568334579468
    },
    {
      "epoch": 2.2097587585449218e-05,
      "model_forward_time": 1.711129903793335,
      "step": 23171
    },
    {
      "epoch": 2.2097587585449218e-05,
      "step": 23171,
      "training_step_time": 8.609729766845703
    },
    {
      "epoch": 2.2098541259765626e-05,
      "model_forward_time": 1.7136907577514648,
      "step": 23172
    },
    {
      "epoch": 2.2098541259765626e-05,
      "step": 23172,
      "training_step_time": 9.013964176177979
    },
    {
      "epoch": 2.209949493408203e-05,
      "model_forward_time": 1.711430311203003,
      "step": 23173
    },
    {
      "epoch": 2.209949493408203e-05,
      "step": 23173,
      "training_step_time": 7.574191570281982
    },
    {
      "epoch": 2.2100448608398437e-05,
      "model_forward_time": 1.7224178314208984,
      "step": 23174
    },
    {
      "epoch": 2.2100448608398437e-05,
      "step": 23174,
      "training_step_time": 7.6275646686553955
    },
    {
      "epoch": 2.2101402282714844e-05,
      "model_forward_time": 1.717489242553711,
      "step": 23175
    },
    {
      "epoch": 2.2101402282714844e-05,
      "step": 23175,
      "training_step_time": 7.15649938583374
    },
    {
      "epoch": 2.210235595703125e-05,
      "model_forward_time": 1.712782382965088,
      "step": 23176
    },
    {
      "epoch": 2.210235595703125e-05,
      "step": 23176,
      "training_step_time": 7.122015476226807
    },
    {
      "epoch": 2.2103309631347656e-05,
      "model_forward_time": 1.7097930908203125,
      "step": 23177
    },
    {
      "epoch": 2.2103309631347656e-05,
      "step": 23177,
      "training_step_time": 7.1134514808654785
    },
    {
      "epoch": 2.2104263305664063e-05,
      "model_forward_time": 1.714163064956665,
      "step": 23178
    },
    {
      "epoch": 2.2104263305664063e-05,
      "step": 23178,
      "training_step_time": 7.389580249786377
    },
    {
      "epoch": 2.2105216979980467e-05,
      "model_forward_time": 2.311293125152588,
      "step": 23179
    },
    {
      "epoch": 2.2105216979980467e-05,
      "step": 23179,
      "training_step_time": 9.817436456680298
    },
    {
      "epoch": 2.2106170654296875e-05,
      "grad_norm": 0.1617991328239441,
      "learning_rate": 9.12337617513025e-06,
      "loss": 0.0624,
      "step": 23180
    },
    {
      "epoch": 2.2106170654296875e-05,
      "model_forward_time": 1.711423635482788,
      "step": 23180
    },
    {
      "dynamics_loss_avg": 0.06384306028485298,
      "epoch": 2.2106170654296875e-05,
      "step": 23180
    },
    {
      "action_loss_avg": 0.002994062675861642,
      "epoch": 2.2106170654296875e-05,
      "step": 23180
    },
    {
      "epoch": 2.2106170654296875e-05,
      "step": 23180,
      "training_step_time": 9.979915380477905
    },
    {
      "epoch": 2.2107124328613282e-05,
      "model_forward_time": 1.7084555625915527,
      "step": 23181
    },
    {
      "epoch": 2.2107124328613282e-05,
      "step": 23181,
      "training_step_time": 9.459912061691284
    },
    {
      "epoch": 2.2108078002929686e-05,
      "model_forward_time": 1.7080414295196533,
      "step": 23182
    },
    {
      "epoch": 2.2108078002929686e-05,
      "step": 23182,
      "training_step_time": 7.506119966506958
    },
    {
      "epoch": 2.2109031677246094e-05,
      "model_forward_time": 1.7145428657531738,
      "step": 23183
    },
    {
      "epoch": 2.2109031677246094e-05,
      "step": 23183,
      "training_step_time": 7.146547794342041
    },
    {
      "epoch": 2.21099853515625e-05,
      "model_forward_time": 1.709249496459961,
      "step": 23184
    },
    {
      "epoch": 2.21099853515625e-05,
      "step": 23184,
      "training_step_time": 7.947567701339722
    },
    {
      "epoch": 2.2110939025878905e-05,
      "model_forward_time": 1.717193603515625,
      "step": 23185
    },
    {
      "epoch": 2.2110939025878905e-05,
      "step": 23185,
      "training_step_time": 7.088515758514404
    },
    {
      "epoch": 2.2111892700195312e-05,
      "model_forward_time": 1.717069387435913,
      "step": 23186
    },
    {
      "epoch": 2.2111892700195312e-05,
      "step": 23186,
      "training_step_time": 7.108063697814941
    },
    {
      "epoch": 2.211284637451172e-05,
      "model_forward_time": 1.7144489288330078,
      "step": 23187
    },
    {
      "epoch": 2.211284637451172e-05,
      "step": 23187,
      "training_step_time": 7.338048458099365
    },
    {
      "epoch": 2.2113800048828124e-05,
      "model_forward_time": 1.799652338027954,
      "step": 23188
    },
    {
      "epoch": 2.2113800048828124e-05,
      "step": 23188,
      "training_step_time": 9.090736627578735
    },
    {
      "epoch": 2.211475372314453e-05,
      "model_forward_time": 1.714857816696167,
      "step": 23189
    },
    {
      "epoch": 2.211475372314453e-05,
      "step": 23189,
      "training_step_time": 11.373936891555786
    },
    {
      "epoch": 2.211570739746094e-05,
      "grad_norm": 0.17119093239307404,
      "learning_rate": 9.122440736343597e-06,
      "loss": 0.059,
      "step": 23190
    },
    {
      "epoch": 2.211570739746094e-05,
      "model_forward_time": 1.7197208404541016,
      "step": 23190
    },
    {
      "dynamics_loss_avg": 0.06550046373158694,
      "epoch": 2.211570739746094e-05,
      "step": 23190
    },
    {
      "action_loss_avg": 0.005256654502591118,
      "epoch": 2.211570739746094e-05,
      "step": 23190
    },
    {
      "epoch": 2.211570739746094e-05,
      "step": 23190,
      "training_step_time": 8.158722877502441
    },
    {
      "epoch": 2.2116661071777343e-05,
      "model_forward_time": 1.7224986553192139,
      "step": 23191
    },
    {
      "epoch": 2.2116661071777343e-05,
      "step": 23191,
      "training_step_time": 7.589972019195557
    },
    {
      "epoch": 2.211761474609375e-05,
      "model_forward_time": 1.7270066738128662,
      "step": 23192
    },
    {
      "epoch": 2.211761474609375e-05,
      "step": 23192,
      "training_step_time": 7.661220550537109
    },
    {
      "epoch": 2.2118568420410158e-05,
      "model_forward_time": 1.7237913608551025,
      "step": 23193
    },
    {
      "epoch": 2.2118568420410158e-05,
      "step": 23193,
      "training_step_time": 7.895660638809204
    },
    {
      "epoch": 2.211952209472656e-05,
      "model_forward_time": 1.703756332397461,
      "step": 23194
    },
    {
      "epoch": 2.211952209472656e-05,
      "step": 23194,
      "training_step_time": 7.090306043624878
    },
    {
      "epoch": 2.212047576904297e-05,
      "model_forward_time": 1.7228209972381592,
      "step": 23195
    },
    {
      "epoch": 2.212047576904297e-05,
      "step": 23195,
      "training_step_time": 7.090487003326416
    },
    {
      "epoch": 2.2121429443359376e-05,
      "model_forward_time": 1.7249093055725098,
      "step": 23196
    },
    {
      "epoch": 2.2121429443359376e-05,
      "step": 23196,
      "training_step_time": 7.303352355957031
    },
    {
      "epoch": 2.212238311767578e-05,
      "model_forward_time": 1.804955005645752,
      "step": 23197
    },
    {
      "epoch": 2.212238311767578e-05,
      "step": 23197,
      "training_step_time": 9.82509160041809
    },
    {
      "epoch": 2.2123336791992188e-05,
      "model_forward_time": 2.24674391746521,
      "step": 23198
    },
    {
      "epoch": 2.2123336791992188e-05,
      "step": 23198,
      "training_step_time": 8.394114971160889
    },
    {
      "epoch": 2.2124290466308595e-05,
      "model_forward_time": 1.7091431617736816,
      "step": 23199
    },
    {
      "epoch": 2.2124290466308595e-05,
      "step": 23199,
      "training_step_time": 10.294265031814575
    },
    {
      "epoch": 2.2125244140625e-05,
      "grad_norm": 0.21005670726299286,
      "learning_rate": 9.121504846733021e-06,
      "loss": 0.0611,
      "step": 23200
    },
    {
      "epoch": 2.2125244140625e-05,
      "model_forward_time": 1.7076709270477295,
      "step": 23200
    },
    {
      "dynamics_loss_avg": 0.0507527980953455,
      "epoch": 2.2125244140625e-05,
      "step": 23200
    },
    {
      "action_loss_avg": 0.0025550701771862804,
      "epoch": 2.2125244140625e-05,
      "step": 23200
    },
    {
      "epoch": 2.2125244140625e-05,
      "step": 23200,
      "training_step_time": 8.13908839225769
    },
    {
      "epoch": 2.2126197814941407e-05,
      "model_forward_time": 1.718156099319458,
      "step": 23201
    },
    {
      "epoch": 2.2126197814941407e-05,
      "step": 23201,
      "training_step_time": 7.594792366027832
    },
    {
      "epoch": 2.2127151489257814e-05,
      "model_forward_time": 1.7232191562652588,
      "step": 23202
    },
    {
      "epoch": 2.2127151489257814e-05,
      "step": 23202,
      "training_step_time": 7.923767328262329
    },
    {
      "epoch": 2.2128105163574218e-05,
      "model_forward_time": 1.7309324741363525,
      "step": 23203
    },
    {
      "epoch": 2.2128105163574218e-05,
      "step": 23203,
      "training_step_time": 7.312469005584717
    },
    {
      "epoch": 2.2129058837890626e-05,
      "model_forward_time": 1.7130296230316162,
      "step": 23204
    },
    {
      "epoch": 2.2129058837890626e-05,
      "step": 23204,
      "training_step_time": 7.231001138687134
    },
    {
      "epoch": 2.213001251220703e-05,
      "model_forward_time": 1.7266619205474854,
      "step": 23205
    },
    {
      "epoch": 2.213001251220703e-05,
      "step": 23205,
      "training_step_time": 7.210790157318115
    },
    {
      "epoch": 2.2130966186523437e-05,
      "model_forward_time": 1.8022825717926025,
      "step": 23206
    },
    {
      "epoch": 2.2130966186523437e-05,
      "step": 23206,
      "training_step_time": 7.88044285774231
    },
    {
      "epoch": 2.2131919860839844e-05,
      "model_forward_time": 2.2662863731384277,
      "step": 23207
    },
    {
      "epoch": 2.2131919860839844e-05,
      "step": 23207,
      "training_step_time": 8.558810234069824
    },
    {
      "epoch": 2.213287353515625e-05,
      "model_forward_time": 1.721362590789795,
      "step": 23208
    },
    {
      "epoch": 2.213287353515625e-05,
      "step": 23208,
      "training_step_time": 10.635462999343872
    },
    {
      "epoch": 2.2133827209472656e-05,
      "model_forward_time": 1.7267425060272217,
      "step": 23209
    },
    {
      "epoch": 2.2133827209472656e-05,
      "step": 23209,
      "training_step_time": 8.260969400405884
    },
    {
      "epoch": 2.2134780883789063e-05,
      "grad_norm": 0.27943292260169983,
      "learning_rate": 9.120568506400873e-06,
      "loss": 0.0603,
      "step": 23210
    },
    {
      "epoch": 2.2134780883789063e-05,
      "model_forward_time": 1.7100427150726318,
      "step": 23210
    },
    {
      "dynamics_loss_avg": 0.07871763855218887,
      "epoch": 2.2134780883789063e-05,
      "step": 23210
    },
    {
      "action_loss_avg": 0.0032507582916878166,
      "epoch": 2.2134780883789063e-05,
      "step": 23210
    },
    {
      "epoch": 2.2134780883789063e-05,
      "step": 23210,
      "training_step_time": 7.5190346240997314
    },
    {
      "epoch": 2.2135734558105467e-05,
      "model_forward_time": 1.7180721759796143,
      "step": 23211
    },
    {
      "epoch": 2.2135734558105467e-05,
      "step": 23211,
      "training_step_time": 7.823972702026367
    },
    {
      "epoch": 2.2136688232421875e-05,
      "model_forward_time": 1.7253634929656982,
      "step": 23212
    },
    {
      "epoch": 2.2136688232421875e-05,
      "step": 23212,
      "training_step_time": 7.196829795837402
    },
    {
      "epoch": 2.2137641906738282e-05,
      "model_forward_time": 1.7153704166412354,
      "step": 23213
    },
    {
      "epoch": 2.2137641906738282e-05,
      "step": 23213,
      "training_step_time": 7.150254249572754
    },
    {
      "epoch": 2.2138595581054686e-05,
      "model_forward_time": 1.714188814163208,
      "step": 23214
    },
    {
      "epoch": 2.2138595581054686e-05,
      "step": 23214,
      "training_step_time": 7.200757026672363
    },
    {
      "epoch": 2.2139549255371094e-05,
      "model_forward_time": 1.7903475761413574,
      "step": 23215
    },
    {
      "epoch": 2.2139549255371094e-05,
      "step": 23215,
      "training_step_time": 7.895026206970215
    },
    {
      "epoch": 2.21405029296875e-05,
      "model_forward_time": 2.2586770057678223,
      "step": 23216
    },
    {
      "epoch": 2.21405029296875e-05,
      "step": 23216,
      "training_step_time": 7.908244609832764
    },
    {
      "epoch": 2.2141456604003905e-05,
      "model_forward_time": 1.7122306823730469,
      "step": 23217
    },
    {
      "epoch": 2.2141456604003905e-05,
      "step": 23217,
      "training_step_time": 8.893728017807007
    },
    {
      "epoch": 2.2142410278320312e-05,
      "model_forward_time": 1.7155463695526123,
      "step": 23218
    },
    {
      "epoch": 2.2142410278320312e-05,
      "step": 23218,
      "training_step_time": 8.66506838798523
    },
    {
      "epoch": 2.214336395263672e-05,
      "model_forward_time": 1.704817295074463,
      "step": 23219
    },
    {
      "epoch": 2.214336395263672e-05,
      "step": 23219,
      "training_step_time": 7.856014728546143
    },
    {
      "epoch": 2.2144317626953124e-05,
      "grad_norm": 0.19370457530021667,
      "learning_rate": 9.119631715449548e-06,
      "loss": 0.0619,
      "step": 23220
    },
    {
      "epoch": 2.2144317626953124e-05,
      "model_forward_time": 1.7101314067840576,
      "step": 23220
    },
    {
      "dynamics_loss_avg": 0.07229398004710674,
      "epoch": 2.2144317626953124e-05,
      "step": 23220
    },
    {
      "action_loss_avg": 0.0025339803542010486,
      "epoch": 2.2144317626953124e-05,
      "step": 23220
    },
    {
      "epoch": 2.2144317626953124e-05,
      "step": 23220,
      "training_step_time": 7.548490524291992
    },
    {
      "epoch": 2.214527130126953e-05,
      "model_forward_time": 1.7113358974456787,
      "step": 23221
    },
    {
      "epoch": 2.214527130126953e-05,
      "step": 23221,
      "training_step_time": 7.8379130363464355
    },
    {
      "epoch": 2.214622497558594e-05,
      "model_forward_time": 1.7133188247680664,
      "step": 23222
    },
    {
      "epoch": 2.214622497558594e-05,
      "step": 23222,
      "training_step_time": 7.097838640213013
    },
    {
      "epoch": 2.2147178649902343e-05,
      "model_forward_time": 1.7059457302093506,
      "step": 23223
    },
    {
      "epoch": 2.2147178649902343e-05,
      "step": 23223,
      "training_step_time": 7.10426926612854
    },
    {
      "epoch": 2.214813232421875e-05,
      "model_forward_time": 1.7882606983184814,
      "step": 23224
    },
    {
      "epoch": 2.214813232421875e-05,
      "step": 23224,
      "training_step_time": 7.8089587688446045
    },
    {
      "epoch": 2.2149085998535158e-05,
      "model_forward_time": 2.2696309089660645,
      "step": 23225
    },
    {
      "epoch": 2.2149085998535158e-05,
      "step": 23225,
      "training_step_time": 8.507919788360596
    },
    {
      "epoch": 2.215003967285156e-05,
      "model_forward_time": 1.7196252346038818,
      "step": 23226
    },
    {
      "epoch": 2.215003967285156e-05,
      "step": 23226,
      "training_step_time": 7.878697872161865
    },
    {
      "epoch": 2.215099334716797e-05,
      "model_forward_time": 1.716184139251709,
      "step": 23227
    },
    {
      "epoch": 2.215099334716797e-05,
      "step": 23227,
      "training_step_time": 8.1347336769104
    },
    {
      "epoch": 2.2151947021484376e-05,
      "model_forward_time": 1.3297076225280762,
      "step": 23228
    },
    {
      "epoch": 2.2151947021484376e-05,
      "step": 23228,
      "training_step_time": 7.820136308670044
    },
    {
      "epoch": 2.215290069580078e-05,
      "model_forward_time": 1.7187843322753906,
      "step": 23229
    },
    {
      "epoch": 2.215290069580078e-05,
      "step": 23229,
      "training_step_time": 7.622205972671509
    },
    {
      "epoch": 2.2153854370117188e-05,
      "grad_norm": 0.17418718338012695,
      "learning_rate": 9.118694473981494e-06,
      "loss": 0.061,
      "step": 23230
    },
    {
      "epoch": 2.2153854370117188e-05,
      "model_forward_time": 1.3399219512939453,
      "step": 23230
    },
    {
      "dynamics_loss_avg": 0.06424222514033318,
      "epoch": 2.2153854370117188e-05,
      "step": 23230
    },
    {
      "action_loss_avg": 0.002507994679035619,
      "epoch": 2.2153854370117188e-05,
      "step": 23230
    },
    {
      "epoch": 2.2153854370117188e-05,
      "step": 23230,
      "training_step_time": 7.934819221496582
    },
    {
      "epoch": 2.2154808044433595e-05,
      "model_forward_time": 1.714808702468872,
      "step": 23231
    },
    {
      "epoch": 2.2154808044433595e-05,
      "step": 23231,
      "training_step_time": 7.088783025741577
    },
    {
      "epoch": 2.215576171875e-05,
      "model_forward_time": 1.721980333328247,
      "step": 23232
    },
    {
      "epoch": 2.215576171875e-05,
      "step": 23232,
      "training_step_time": 7.138926982879639
    },
    {
      "epoch": 2.2156715393066407e-05,
      "model_forward_time": 1.4136674404144287,
      "step": 23233
    },
    {
      "epoch": 2.2156715393066407e-05,
      "step": 23233,
      "training_step_time": 7.782622575759888
    },
    {
      "epoch": 2.2157669067382814e-05,
      "model_forward_time": 1.718689203262329,
      "step": 23234
    },
    {
      "epoch": 2.2157669067382814e-05,
      "step": 23234,
      "training_step_time": 9.320683002471924
    },
    {
      "epoch": 2.2158622741699218e-05,
      "model_forward_time": 2.271611213684082,
      "step": 23235
    },
    {
      "epoch": 2.2158622741699218e-05,
      "step": 23235,
      "training_step_time": 10.23125410079956
    },
    {
      "epoch": 2.2159576416015626e-05,
      "model_forward_time": 1.712012767791748,
      "step": 23236
    },
    {
      "epoch": 2.2159576416015626e-05,
      "step": 23236,
      "training_step_time": 8.253848314285278
    },
    {
      "epoch": 2.216053009033203e-05,
      "model_forward_time": 1.7169480323791504,
      "step": 23237
    },
    {
      "epoch": 2.216053009033203e-05,
      "step": 23237,
      "training_step_time": 7.893815279006958
    },
    {
      "epoch": 2.2161483764648437e-05,
      "model_forward_time": 1.712554931640625,
      "step": 23238
    },
    {
      "epoch": 2.2161483764648437e-05,
      "step": 23238,
      "training_step_time": 7.525019645690918
    },
    {
      "epoch": 2.2162437438964844e-05,
      "model_forward_time": 1.72200345993042,
      "step": 23239
    },
    {
      "epoch": 2.2162437438964844e-05,
      "step": 23239,
      "training_step_time": 7.884643077850342
    },
    {
      "epoch": 2.216339111328125e-05,
      "grad_norm": 0.23439037799835205,
      "learning_rate": 9.117756782099202e-06,
      "loss": 0.0596,
      "step": 23240
    },
    {
      "epoch": 2.216339111328125e-05,
      "model_forward_time": 1.7074685096740723,
      "step": 23240
    },
    {
      "dynamics_loss_avg": 0.06067925598472357,
      "epoch": 2.216339111328125e-05,
      "step": 23240
    },
    {
      "action_loss_avg": 0.003525136719690636,
      "epoch": 2.216339111328125e-05,
      "step": 23240
    },
    {
      "epoch": 2.216339111328125e-05,
      "step": 23240,
      "training_step_time": 7.094013690948486
    },
    {
      "epoch": 2.2164344787597656e-05,
      "model_forward_time": 1.7159810066223145,
      "step": 23241
    },
    {
      "epoch": 2.2164344787597656e-05,
      "step": 23241,
      "training_step_time": 7.116825819015503
    },
    {
      "epoch": 2.2165298461914063e-05,
      "model_forward_time": 1.803938627243042,
      "step": 23242
    },
    {
      "epoch": 2.2165298461914063e-05,
      "step": 23242,
      "training_step_time": 7.7699501514434814
    },
    {
      "epoch": 2.2166252136230467e-05,
      "model_forward_time": 1.7221453189849854,
      "step": 23243
    },
    {
      "epoch": 2.2166252136230467e-05,
      "step": 23243,
      "training_step_time": 9.277759313583374
    },
    {
      "epoch": 2.2167205810546875e-05,
      "model_forward_time": 2.2694168090820312,
      "step": 23244
    },
    {
      "epoch": 2.2167205810546875e-05,
      "step": 23244,
      "training_step_time": 9.936430931091309
    },
    {
      "epoch": 2.2168159484863282e-05,
      "model_forward_time": 1.7116155624389648,
      "step": 23245
    },
    {
      "epoch": 2.2168159484863282e-05,
      "step": 23245,
      "training_step_time": 8.890662431716919
    },
    {
      "epoch": 2.2169113159179686e-05,
      "model_forward_time": 1.7240993976593018,
      "step": 23246
    },
    {
      "epoch": 2.2169113159179686e-05,
      "step": 23246,
      "training_step_time": 8.251567602157593
    },
    {
      "epoch": 2.2170066833496094e-05,
      "model_forward_time": 1.7194170951843262,
      "step": 23247
    },
    {
      "epoch": 2.2170066833496094e-05,
      "step": 23247,
      "training_step_time": 7.471506118774414
    },
    {
      "epoch": 2.21710205078125e-05,
      "model_forward_time": 1.7190802097320557,
      "step": 23248
    },
    {
      "epoch": 2.21710205078125e-05,
      "step": 23248,
      "training_step_time": 7.643207550048828
    },
    {
      "epoch": 2.2171974182128905e-05,
      "model_forward_time": 1.7235746383666992,
      "step": 23249
    },
    {
      "epoch": 2.2171974182128905e-05,
      "step": 23249,
      "training_step_time": 7.90201735496521
    },
    {
      "epoch": 2.2172927856445312e-05,
      "grad_norm": 0.19188982248306274,
      "learning_rate": 9.11681863990522e-06,
      "loss": 0.063,
      "step": 23250
    },
    {
      "epoch": 2.2172927856445312e-05,
      "model_forward_time": 1.7180874347686768,
      "step": 23250
    },
    {
      "dynamics_loss_avg": 0.06898904768750072,
      "epoch": 2.2172927856445312e-05,
      "step": 23250
    },
    {
      "action_loss_avg": 0.0037422877503558993,
      "epoch": 2.2172927856445312e-05,
      "step": 23250
    },
    {
      "epoch": 2.2172927856445312e-05,
      "step": 23250,
      "training_step_time": 7.063316822052002
    },
    {
      "epoch": 2.217388153076172e-05,
      "model_forward_time": 1.8049108982086182,
      "step": 23251
    },
    {
      "epoch": 2.217388153076172e-05,
      "step": 23251,
      "training_step_time": 7.767966985702515
    },
    {
      "epoch": 2.2174835205078124e-05,
      "model_forward_time": 1.7254531383514404,
      "step": 23252
    },
    {
      "epoch": 2.2174835205078124e-05,
      "step": 23252,
      "training_step_time": 8.63466215133667
    },
    {
      "epoch": 2.217578887939453e-05,
      "model_forward_time": 2.284420967102051,
      "step": 23253
    },
    {
      "epoch": 2.217578887939453e-05,
      "step": 23253,
      "training_step_time": 8.49571681022644
    },
    {
      "epoch": 2.217674255371094e-05,
      "model_forward_time": 1.7188947200775146,
      "step": 23254
    },
    {
      "epoch": 2.217674255371094e-05,
      "step": 23254,
      "training_step_time": 9.702352046966553
    },
    {
      "epoch": 2.2177696228027343e-05,
      "model_forward_time": 1.335998296737671,
      "step": 23255
    },
    {
      "epoch": 2.2177696228027343e-05,
      "step": 23255,
      "training_step_time": 8.655954837799072
    },
    {
      "epoch": 2.217864990234375e-05,
      "model_forward_time": 1.724557876586914,
      "step": 23256
    },
    {
      "epoch": 2.217864990234375e-05,
      "step": 23256,
      "training_step_time": 7.966370582580566
    },
    {
      "epoch": 2.2179603576660158e-05,
      "model_forward_time": 1.7226250171661377,
      "step": 23257
    },
    {
      "epoch": 2.2179603576660158e-05,
      "step": 23257,
      "training_step_time": 7.477471828460693
    },
    {
      "epoch": 2.218055725097656e-05,
      "model_forward_time": 1.7181758880615234,
      "step": 23258
    },
    {
      "epoch": 2.218055725097656e-05,
      "step": 23258,
      "training_step_time": 7.875425338745117
    },
    {
      "epoch": 2.218151092529297e-05,
      "model_forward_time": 1.727114200592041,
      "step": 23259
    },
    {
      "epoch": 2.218151092529297e-05,
      "step": 23259,
      "training_step_time": 7.117910623550415
    },
    {
      "epoch": 2.2182464599609376e-05,
      "grad_norm": 0.16365379095077515,
      "learning_rate": 9.115880047502142e-06,
      "loss": 0.0558,
      "step": 23260
    },
    {
      "epoch": 2.2182464599609376e-05,
      "model_forward_time": 1.7927494049072266,
      "step": 23260
    },
    {
      "dynamics_loss_avg": 0.05863861460238695,
      "epoch": 2.2182464599609376e-05,
      "step": 23260
    },
    {
      "action_loss_avg": 0.002523989457404241,
      "epoch": 2.2182464599609376e-05,
      "step": 23260
    },
    {
      "epoch": 2.2182464599609376e-05,
      "step": 23260,
      "training_step_time": 7.754290819168091
    },
    {
      "epoch": 2.218341827392578e-05,
      "model_forward_time": 1.7141928672790527,
      "step": 23261
    },
    {
      "epoch": 2.218341827392578e-05,
      "step": 23261,
      "training_step_time": 7.802759885787964
    },
    {
      "epoch": 2.2184371948242188e-05,
      "model_forward_time": 1.7226743698120117,
      "step": 23262
    },
    {
      "epoch": 2.2184371948242188e-05,
      "step": 23262,
      "training_step_time": 7.842102766036987
    },
    {
      "epoch": 2.2185325622558595e-05,
      "model_forward_time": 2.259315013885498,
      "step": 23263
    },
    {
      "epoch": 2.2185325622558595e-05,
      "step": 23263,
      "training_step_time": 9.756357192993164
    },
    {
      "epoch": 2.2186279296875e-05,
      "model_forward_time": 1.7279791831970215,
      "step": 23264
    },
    {
      "epoch": 2.2186279296875e-05,
      "step": 23264,
      "training_step_time": 8.13731575012207
    },
    {
      "epoch": 2.2187232971191407e-05,
      "model_forward_time": 1.7220547199249268,
      "step": 23265
    },
    {
      "epoch": 2.2187232971191407e-05,
      "step": 23265,
      "training_step_time": 7.870673894882202
    },
    {
      "epoch": 2.218818664550781e-05,
      "model_forward_time": 1.7196812629699707,
      "step": 23266
    },
    {
      "epoch": 2.218818664550781e-05,
      "step": 23266,
      "training_step_time": 7.442709684371948
    },
    {
      "epoch": 2.2189140319824218e-05,
      "model_forward_time": 1.7245728969573975,
      "step": 23267
    },
    {
      "epoch": 2.2189140319824218e-05,
      "step": 23267,
      "training_step_time": 7.864147663116455
    },
    {
      "epoch": 2.2190093994140626e-05,
      "model_forward_time": 1.7207677364349365,
      "step": 23268
    },
    {
      "epoch": 2.2190093994140626e-05,
      "step": 23268,
      "training_step_time": 7.092342138290405
    },
    {
      "epoch": 2.219104766845703e-05,
      "model_forward_time": 1.773425579071045,
      "step": 23269
    },
    {
      "epoch": 2.219104766845703e-05,
      "step": 23269,
      "training_step_time": 7.809493780136108
    },
    {
      "epoch": 2.2192001342773437e-05,
      "grad_norm": 0.16273267567157745,
      "learning_rate": 9.11494100499261e-06,
      "loss": 0.0571,
      "step": 23270
    },
    {
      "epoch": 2.2192001342773437e-05,
      "model_forward_time": 1.7003495693206787,
      "step": 23270
    },
    {
      "dynamics_loss_avg": 0.05738543067127466,
      "epoch": 2.2192001342773437e-05,
      "step": 23270
    },
    {
      "action_loss_avg": 0.004656403686385602,
      "epoch": 2.2192001342773437e-05,
      "step": 23270
    },
    {
      "epoch": 2.2192001342773437e-05,
      "step": 23270,
      "training_step_time": 7.741548776626587
    },
    {
      "epoch": 2.2192955017089845e-05,
      "model_forward_time": 1.7143023014068604,
      "step": 23271
    },
    {
      "epoch": 2.2192955017089845e-05,
      "step": 23271,
      "training_step_time": 7.70708441734314
    },
    {
      "epoch": 2.219390869140625e-05,
      "model_forward_time": 2.255455255508423,
      "step": 23272
    },
    {
      "epoch": 2.219390869140625e-05,
      "step": 23272,
      "training_step_time": 8.283200740814209
    },
    {
      "epoch": 2.2194862365722656e-05,
      "model_forward_time": 1.7142324447631836,
      "step": 23273
    },
    {
      "epoch": 2.2194862365722656e-05,
      "step": 23273,
      "training_step_time": 8.705507755279541
    },
    {
      "epoch": 2.2195816040039063e-05,
      "model_forward_time": 1.330932378768921,
      "step": 23274
    },
    {
      "epoch": 2.2195816040039063e-05,
      "step": 23274,
      "training_step_time": 7.865094423294067
    },
    {
      "epoch": 2.2196769714355467e-05,
      "model_forward_time": 1.7126438617706299,
      "step": 23275
    },
    {
      "epoch": 2.2196769714355467e-05,
      "step": 23275,
      "training_step_time": 7.399500131607056
    },
    {
      "epoch": 2.2197723388671875e-05,
      "model_forward_time": 1.7123844623565674,
      "step": 23276
    },
    {
      "epoch": 2.2197723388671875e-05,
      "step": 23276,
      "training_step_time": 7.850362777709961
    },
    {
      "epoch": 2.2198677062988282e-05,
      "model_forward_time": 0.9894373416900635,
      "step": 23277
    },
    {
      "epoch": 2.2198677062988282e-05,
      "step": 23277,
      "training_step_time": 7.109813690185547
    },
    {
      "epoch": 2.2199630737304686e-05,
      "model_forward_time": 1.335143804550171,
      "step": 23278
    },
    {
      "epoch": 2.2199630737304686e-05,
      "step": 23278,
      "training_step_time": 7.855052471160889
    },
    {
      "epoch": 2.2200584411621094e-05,
      "model_forward_time": 1.7166647911071777,
      "step": 23279
    },
    {
      "epoch": 2.2200584411621094e-05,
      "step": 23279,
      "training_step_time": 8.340746402740479
    },
    {
      "epoch": 2.22015380859375e-05,
      "grad_norm": 0.23546788096427917,
      "learning_rate": 9.114001512479316e-06,
      "loss": 0.0597,
      "step": 23280
    },
    {
      "epoch": 2.22015380859375e-05,
      "model_forward_time": 1.7092669010162354,
      "step": 23280
    },
    {
      "dynamics_loss_avg": 0.049503929913043976,
      "epoch": 2.22015380859375e-05,
      "step": 23280
    },
    {
      "action_loss_avg": 0.0026776461570989342,
      "epoch": 2.22015380859375e-05,
      "step": 23280
    },
    {
      "epoch": 2.22015380859375e-05,
      "step": 23280,
      "training_step_time": 9.563210248947144
    },
    {
      "epoch": 2.2202491760253905e-05,
      "model_forward_time": 2.2469520568847656,
      "step": 23281
    },
    {
      "epoch": 2.2202491760253905e-05,
      "step": 23281,
      "training_step_time": 8.968372821807861
    },
    {
      "epoch": 2.2203445434570313e-05,
      "model_forward_time": 1.7197749614715576,
      "step": 23282
    },
    {
      "epoch": 2.2203445434570313e-05,
      "step": 23282,
      "training_step_time": 11.355382919311523
    },
    {
      "epoch": 2.220439910888672e-05,
      "model_forward_time": 1.7147600650787354,
      "step": 23283
    },
    {
      "epoch": 2.220439910888672e-05,
      "step": 23283,
      "training_step_time": 8.593391180038452
    },
    {
      "epoch": 2.2205352783203124e-05,
      "model_forward_time": 1.720839262008667,
      "step": 23284
    },
    {
      "epoch": 2.2205352783203124e-05,
      "step": 23284,
      "training_step_time": 7.463709115982056
    },
    {
      "epoch": 2.220630645751953e-05,
      "model_forward_time": 1.719191551208496,
      "step": 23285
    },
    {
      "epoch": 2.220630645751953e-05,
      "step": 23285,
      "training_step_time": 7.6002748012542725
    },
    {
      "epoch": 2.220726013183594e-05,
      "model_forward_time": 1.7244701385498047,
      "step": 23286
    },
    {
      "epoch": 2.220726013183594e-05,
      "step": 23286,
      "training_step_time": 7.924617528915405
    },
    {
      "epoch": 2.2208213806152343e-05,
      "model_forward_time": 1.7189559936523438,
      "step": 23287
    },
    {
      "epoch": 2.2208213806152343e-05,
      "step": 23287,
      "training_step_time": 7.77843713760376
    },
    {
      "epoch": 2.220916748046875e-05,
      "model_forward_time": 1.7118713855743408,
      "step": 23288
    },
    {
      "epoch": 2.220916748046875e-05,
      "step": 23288,
      "training_step_time": 7.785524129867554
    },
    {
      "epoch": 2.2210121154785158e-05,
      "model_forward_time": 1.7191979885101318,
      "step": 23289
    },
    {
      "epoch": 2.2210121154785158e-05,
      "step": 23289,
      "training_step_time": 7.895576238632202
    },
    {
      "epoch": 2.221107482910156e-05,
      "grad_norm": 0.20932024717330933,
      "learning_rate": 9.113061570065002e-06,
      "loss": 0.0619,
      "step": 23290
    },
    {
      "epoch": 2.221107482910156e-05,
      "model_forward_time": 1.7198593616485596,
      "step": 23290
    },
    {
      "dynamics_loss_avg": 0.06433183532208205,
      "epoch": 2.221107482910156e-05,
      "step": 23290
    },
    {
      "action_loss_avg": 0.0033865738660097122,
      "epoch": 2.221107482910156e-05,
      "step": 23290
    },
    {
      "epoch": 2.221107482910156e-05,
      "step": 23290,
      "training_step_time": 9.930304288864136
    },
    {
      "epoch": 2.221202850341797e-05,
      "model_forward_time": 2.26369571685791,
      "step": 23291
    },
    {
      "epoch": 2.221202850341797e-05,
      "step": 23291,
      "training_step_time": 7.740433692932129
    },
    {
      "epoch": 2.2212982177734377e-05,
      "model_forward_time": 1.7080714702606201,
      "step": 23292
    },
    {
      "epoch": 2.2212982177734377e-05,
      "step": 23292,
      "training_step_time": 8.189505100250244
    },
    {
      "epoch": 2.221393585205078e-05,
      "model_forward_time": 1.7157719135284424,
      "step": 23293
    },
    {
      "epoch": 2.221393585205078e-05,
      "step": 23293,
      "training_step_time": 7.907219886779785
    },
    {
      "epoch": 2.2214889526367188e-05,
      "model_forward_time": 1.716064691543579,
      "step": 23294
    },
    {
      "epoch": 2.2214889526367188e-05,
      "step": 23294,
      "training_step_time": 7.61256217956543
    },
    {
      "epoch": 2.2215843200683595e-05,
      "model_forward_time": 1.7170383930206299,
      "step": 23295
    },
    {
      "epoch": 2.2215843200683595e-05,
      "step": 23295,
      "training_step_time": 7.971376657485962
    },
    {
      "epoch": 2.2216796875e-05,
      "model_forward_time": 1.7297334671020508,
      "step": 23296
    },
    {
      "epoch": 2.2216796875e-05,
      "step": 23296,
      "training_step_time": 7.500433683395386
    },
    {
      "epoch": 2.2217750549316407e-05,
      "model_forward_time": 1.7939321994781494,
      "step": 23297
    },
    {
      "epoch": 2.2217750549316407e-05,
      "step": 23297,
      "training_step_time": 7.804167985916138
    },
    {
      "epoch": 2.221870422363281e-05,
      "model_forward_time": 1.7071590423583984,
      "step": 23298
    },
    {
      "epoch": 2.221870422363281e-05,
      "step": 23298,
      "training_step_time": 7.851130247116089
    },
    {
      "epoch": 2.2219657897949218e-05,
      "model_forward_time": 1.7188239097595215,
      "step": 23299
    },
    {
      "epoch": 2.2219657897949218e-05,
      "step": 23299,
      "training_step_time": 9.06708574295044
    },
    {
      "epoch": 2.2220611572265626e-05,
      "grad_norm": 0.2368205338716507,
      "learning_rate": 9.11212117785246e-06,
      "loss": 0.0587,
      "step": 23300
    },
    {
      "epoch": 2.2220611572265626e-05,
      "model_forward_time": 2.2593674659729004,
      "step": 23300
    },
    {
      "dynamics_loss_avg": 0.04861948089674115,
      "epoch": 2.2220611572265626e-05,
      "step": 23300
    },
    {
      "action_loss_avg": 0.003591121768113226,
      "epoch": 2.2220611572265626e-05,
      "step": 23300
    },
    {
      "epoch": 2.2220611572265626e-05,
      "step": 23300,
      "training_step_time": 9.677498579025269
    },
    {
      "epoch": 2.222156524658203e-05,
      "model_forward_time": 1.708141565322876,
      "step": 23301
    },
    {
      "epoch": 2.222156524658203e-05,
      "step": 23301,
      "training_step_time": 8.134701013565063
    },
    {
      "epoch": 2.2222518920898437e-05,
      "model_forward_time": 1.7175543308258057,
      "step": 23302
    },
    {
      "epoch": 2.2222518920898437e-05,
      "step": 23302,
      "training_step_time": 7.96269679069519
    },
    {
      "epoch": 2.2223472595214845e-05,
      "model_forward_time": 1.7129313945770264,
      "step": 23303
    },
    {
      "epoch": 2.2223472595214845e-05,
      "step": 23303,
      "training_step_time": 7.6447038650512695
    },
    {
      "epoch": 2.222442626953125e-05,
      "model_forward_time": 1.7174737453460693,
      "step": 23304
    },
    {
      "epoch": 2.222442626953125e-05,
      "step": 23304,
      "training_step_time": 7.905495643615723
    },
    {
      "epoch": 2.2225379943847656e-05,
      "model_forward_time": 1.7140588760375977,
      "step": 23305
    },
    {
      "epoch": 2.2225379943847656e-05,
      "step": 23305,
      "training_step_time": 7.419193983078003
    },
    {
      "epoch": 2.2226333618164063e-05,
      "model_forward_time": 1.781651258468628,
      "step": 23306
    },
    {
      "epoch": 2.2226333618164063e-05,
      "step": 23306,
      "training_step_time": 7.847486972808838
    },
    {
      "epoch": 2.2227287292480467e-05,
      "model_forward_time": 1.7114925384521484,
      "step": 23307
    },
    {
      "epoch": 2.2227287292480467e-05,
      "step": 23307,
      "training_step_time": 7.809309720993042
    },
    {
      "epoch": 2.2228240966796875e-05,
      "model_forward_time": 1.7181837558746338,
      "step": 23308
    },
    {
      "epoch": 2.2228240966796875e-05,
      "step": 23308,
      "training_step_time": 7.88636040687561
    },
    {
      "epoch": 2.2229194641113282e-05,
      "model_forward_time": 2.269580841064453,
      "step": 23309
    },
    {
      "epoch": 2.2229194641113282e-05,
      "step": 23309,
      "training_step_time": 8.897275924682617
    },
    {
      "epoch": 2.2230148315429686e-05,
      "grad_norm": 0.1781751662492752,
      "learning_rate": 9.111180335944528e-06,
      "loss": 0.062,
      "step": 23310
    },
    {
      "epoch": 2.2230148315429686e-05,
      "model_forward_time": 1.7062022686004639,
      "step": 23310
    },
    {
      "dynamics_loss_avg": 0.056095081567764285,
      "epoch": 2.2230148315429686e-05,
      "step": 23310
    },
    {
      "action_loss_avg": 0.00336216336581856,
      "epoch": 2.2230148315429686e-05,
      "step": 23310
    },
    {
      "epoch": 2.2230148315429686e-05,
      "step": 23310,
      "training_step_time": 8.163418292999268
    },
    {
      "epoch": 2.2231101989746094e-05,
      "model_forward_time": 1.717491626739502,
      "step": 23311
    },
    {
      "epoch": 2.2231101989746094e-05,
      "step": 23311,
      "training_step_time": 8.613512754440308
    },
    {
      "epoch": 2.22320556640625e-05,
      "model_forward_time": 1.7165660858154297,
      "step": 23312
    },
    {
      "epoch": 2.22320556640625e-05,
      "step": 23312,
      "training_step_time": 7.5644636154174805
    },
    {
      "epoch": 2.2233009338378905e-05,
      "model_forward_time": 1.7114877700805664,
      "step": 23313
    },
    {
      "epoch": 2.2233009338378905e-05,
      "step": 23313,
      "training_step_time": 7.908909559249878
    },
    {
      "epoch": 2.2233963012695313e-05,
      "model_forward_time": 1.723816156387329,
      "step": 23314
    },
    {
      "epoch": 2.2233963012695313e-05,
      "step": 23314,
      "training_step_time": 7.109972953796387
    },
    {
      "epoch": 2.223491668701172e-05,
      "model_forward_time": 1.7611713409423828,
      "step": 23315
    },
    {
      "epoch": 2.223491668701172e-05,
      "step": 23315,
      "training_step_time": 7.835176229476929
    },
    {
      "epoch": 2.2235870361328124e-05,
      "model_forward_time": 1.7137157917022705,
      "step": 23316
    },
    {
      "epoch": 2.2235870361328124e-05,
      "step": 23316,
      "training_step_time": 8.413230895996094
    },
    {
      "epoch": 2.223682403564453e-05,
      "model_forward_time": 1.7276637554168701,
      "step": 23317
    },
    {
      "epoch": 2.223682403564453e-05,
      "step": 23317,
      "training_step_time": 9.560015678405762
    },
    {
      "epoch": 2.223777770996094e-05,
      "model_forward_time": 2.271679162979126,
      "step": 23318
    },
    {
      "epoch": 2.223777770996094e-05,
      "step": 23318,
      "training_step_time": 11.447043180465698
    },
    {
      "epoch": 2.2238731384277343e-05,
      "model_forward_time": 1.7150256633758545,
      "step": 23319
    },
    {
      "epoch": 2.2238731384277343e-05,
      "step": 23319,
      "training_step_time": 8.692377805709839
    },
    {
      "epoch": 2.223968505859375e-05,
      "grad_norm": 0.16954109072685242,
      "learning_rate": 9.110239044444093e-06,
      "loss": 0.0632,
      "step": 23320
    },
    {
      "epoch": 2.223968505859375e-05,
      "model_forward_time": 1.7113583087921143,
      "step": 23320
    },
    {
      "dynamics_loss_avg": 0.0915251012891531,
      "epoch": 2.223968505859375e-05,
      "step": 23320
    },
    {
      "action_loss_avg": 0.004489580076187849,
      "epoch": 2.223968505859375e-05,
      "step": 23320
    },
    {
      "epoch": 2.223968505859375e-05,
      "step": 23320,
      "training_step_time": 7.940307140350342
    },
    {
      "epoch": 2.2240638732910158e-05,
      "model_forward_time": 1.7141163349151611,
      "step": 23321
    },
    {
      "epoch": 2.2240638732910158e-05,
      "step": 23321,
      "training_step_time": 8.090983390808105
    },
    {
      "epoch": 2.2241592407226562e-05,
      "model_forward_time": 1.7231366634368896,
      "step": 23322
    },
    {
      "epoch": 2.2241592407226562e-05,
      "step": 23322,
      "training_step_time": 7.924614906311035
    },
    {
      "epoch": 2.224254608154297e-05,
      "model_forward_time": 1.714141845703125,
      "step": 23323
    },
    {
      "epoch": 2.224254608154297e-05,
      "step": 23323,
      "training_step_time": 7.0860536098480225
    },
    {
      "epoch": 2.2243499755859377e-05,
      "model_forward_time": 1.7414231300354004,
      "step": 23324
    },
    {
      "epoch": 2.2243499755859377e-05,
      "step": 23324,
      "training_step_time": 8.321381092071533
    },
    {
      "epoch": 2.224445343017578e-05,
      "model_forward_time": 1.7242295742034912,
      "step": 23325
    },
    {
      "epoch": 2.224445343017578e-05,
      "step": 23325,
      "training_step_time": 8.637407302856445
    },
    {
      "epoch": 2.2245407104492188e-05,
      "model_forward_time": 1.7159099578857422,
      "step": 23326
    },
    {
      "epoch": 2.2245407104492188e-05,
      "step": 23326,
      "training_step_time": 8.866044998168945
    },
    {
      "epoch": 2.2246360778808595e-05,
      "model_forward_time": 2.27236008644104,
      "step": 23327
    },
    {
      "epoch": 2.2246360778808595e-05,
      "step": 23327,
      "training_step_time": 9.353395462036133
    },
    {
      "epoch": 2.2247314453125e-05,
      "model_forward_time": 1.7105894088745117,
      "step": 23328
    },
    {
      "epoch": 2.2247314453125e-05,
      "step": 23328,
      "training_step_time": 8.799426794052124
    },
    {
      "epoch": 2.2248268127441407e-05,
      "model_forward_time": 1.714519739151001,
      "step": 23329
    },
    {
      "epoch": 2.2248268127441407e-05,
      "step": 23329,
      "training_step_time": 8.08757996559143
    },
    {
      "epoch": 2.224922180175781e-05,
      "grad_norm": 0.27200499176979065,
      "learning_rate": 9.109297303454098e-06,
      "loss": 0.0637,
      "step": 23330
    },
    {
      "epoch": 2.224922180175781e-05,
      "model_forward_time": 1.7088959217071533,
      "step": 23330
    },
    {
      "dynamics_loss_avg": 0.04601858574897051,
      "epoch": 2.224922180175781e-05,
      "step": 23330
    },
    {
      "action_loss_avg": 0.002430034521967173,
      "epoch": 2.224922180175781e-05,
      "step": 23330
    },
    {
      "epoch": 2.224922180175781e-05,
      "step": 23330,
      "training_step_time": 7.798502206802368
    },
    {
      "epoch": 2.2250175476074218e-05,
      "model_forward_time": 1.7093701362609863,
      "step": 23331
    },
    {
      "epoch": 2.2250175476074218e-05,
      "step": 23331,
      "training_step_time": 7.828492164611816
    },
    {
      "epoch": 2.2251129150390626e-05,
      "model_forward_time": 1.7017595767974854,
      "step": 23332
    },
    {
      "epoch": 2.2251129150390626e-05,
      "step": 23332,
      "training_step_time": 7.0607826709747314
    },
    {
      "epoch": 2.225208282470703e-05,
      "model_forward_time": 1.721243143081665,
      "step": 23333
    },
    {
      "epoch": 2.225208282470703e-05,
      "step": 23333,
      "training_step_time": 8.272863388061523
    },
    {
      "epoch": 2.2253036499023437e-05,
      "model_forward_time": 1.7124006748199463,
      "step": 23334
    },
    {
      "epoch": 2.2253036499023437e-05,
      "step": 23334,
      "training_step_time": 7.730397701263428
    },
    {
      "epoch": 2.2253990173339845e-05,
      "model_forward_time": 1.7131190299987793,
      "step": 23335
    },
    {
      "epoch": 2.2253990173339845e-05,
      "step": 23335,
      "training_step_time": 7.607248783111572
    },
    {
      "epoch": 2.225494384765625e-05,
      "model_forward_time": 1.7181451320648193,
      "step": 23336
    },
    {
      "epoch": 2.225494384765625e-05,
      "step": 23336,
      "training_step_time": 9.177344560623169
    },
    {
      "epoch": 2.2255897521972656e-05,
      "model_forward_time": 1.7192082405090332,
      "step": 23337
    },
    {
      "epoch": 2.2255897521972656e-05,
      "step": 23337,
      "training_step_time": 8.495665073394775
    },
    {
      "epoch": 2.2256851196289063e-05,
      "model_forward_time": 1.710794448852539,
      "step": 23338
    },
    {
      "epoch": 2.2256851196289063e-05,
      "step": 23338,
      "training_step_time": 7.98951268196106
    },
    {
      "epoch": 2.2257804870605467e-05,
      "model_forward_time": 1.7213382720947266,
      "step": 23339
    },
    {
      "epoch": 2.2257804870605467e-05,
      "step": 23339,
      "training_step_time": 7.709932565689087
    },
    {
      "epoch": 2.2258758544921875e-05,
      "grad_norm": 0.14188961684703827,
      "learning_rate": 9.108355113077527e-06,
      "loss": 0.0636,
      "step": 23340
    },
    {
      "epoch": 2.2258758544921875e-05,
      "model_forward_time": 1.7125568389892578,
      "step": 23340
    },
    {
      "dynamics_loss_avg": 0.06393809858709573,
      "epoch": 2.2258758544921875e-05,
      "step": 23340
    },
    {
      "action_loss_avg": 0.003421005536802113,
      "epoch": 2.2258758544921875e-05,
      "step": 23340
    },
    {
      "epoch": 2.2258758544921875e-05,
      "step": 23340,
      "training_step_time": 7.351043701171875
    },
    {
      "epoch": 2.2259712219238282e-05,
      "model_forward_time": 1.7174060344696045,
      "step": 23341
    },
    {
      "epoch": 2.2259712219238282e-05,
      "step": 23341,
      "training_step_time": 7.3366687297821045
    },
    {
      "epoch": 2.2260665893554686e-05,
      "model_forward_time": 1.7139227390289307,
      "step": 23342
    },
    {
      "epoch": 2.2260665893554686e-05,
      "step": 23342,
      "training_step_time": 7.321900367736816
    },
    {
      "epoch": 2.2261619567871094e-05,
      "model_forward_time": 1.7944400310516357,
      "step": 23343
    },
    {
      "epoch": 2.2261619567871094e-05,
      "step": 23343,
      "training_step_time": 7.681953191757202
    },
    {
      "epoch": 2.22625732421875e-05,
      "model_forward_time": 1.7073533535003662,
      "step": 23344
    },
    {
      "epoch": 2.22625732421875e-05,
      "step": 23344,
      "training_step_time": 8.386873483657837
    },
    {
      "epoch": 2.2263526916503905e-05,
      "model_forward_time": 1.7181117534637451,
      "step": 23345
    },
    {
      "epoch": 2.2263526916503905e-05,
      "step": 23345,
      "training_step_time": 8.419100046157837
    },
    {
      "epoch": 2.2264480590820313e-05,
      "model_forward_time": 1.8735299110412598,
      "step": 23346
    },
    {
      "epoch": 2.2264480590820313e-05,
      "step": 23346,
      "training_step_time": 9.716354608535767
    },
    {
      "epoch": 2.226543426513672e-05,
      "model_forward_time": 1.7302792072296143,
      "step": 23347
    },
    {
      "epoch": 2.226543426513672e-05,
      "step": 23347,
      "training_step_time": 7.860480785369873
    },
    {
      "epoch": 2.2266387939453124e-05,
      "model_forward_time": 1.7136974334716797,
      "step": 23348
    },
    {
      "epoch": 2.2266387939453124e-05,
      "step": 23348,
      "training_step_time": 7.697451829910278
    },
    {
      "epoch": 2.226734161376953e-05,
      "model_forward_time": 1.7174293994903564,
      "step": 23349
    },
    {
      "epoch": 2.226734161376953e-05,
      "step": 23349,
      "training_step_time": 7.269274473190308
    },
    {
      "epoch": 2.226829528808594e-05,
      "grad_norm": 0.15717867016792297,
      "learning_rate": 9.10741247341742e-06,
      "loss": 0.0645,
      "step": 23350
    },
    {
      "epoch": 2.226829528808594e-05,
      "model_forward_time": 1.7070903778076172,
      "step": 23350
    },
    {
      "dynamics_loss_avg": 0.06840321905910969,
      "epoch": 2.226829528808594e-05,
      "step": 23350
    },
    {
      "action_loss_avg": 0.0037380144814960657,
      "epoch": 2.226829528808594e-05,
      "step": 23350
    },
    {
      "epoch": 2.226829528808594e-05,
      "step": 23350,
      "training_step_time": 7.721543788909912
    },
    {
      "epoch": 2.2269248962402343e-05,
      "model_forward_time": 1.7234277725219727,
      "step": 23351
    },
    {
      "epoch": 2.2269248962402343e-05,
      "step": 23351,
      "training_step_time": 7.515310049057007
    },
    {
      "epoch": 2.227020263671875e-05,
      "model_forward_time": 1.7900114059448242,
      "step": 23352
    },
    {
      "epoch": 2.227020263671875e-05,
      "step": 23352,
      "training_step_time": 7.680483818054199
    },
    {
      "epoch": 2.2271156311035158e-05,
      "model_forward_time": 1.7088775634765625,
      "step": 23353
    },
    {
      "epoch": 2.2271156311035158e-05,
      "step": 23353,
      "training_step_time": 7.6374101638793945
    },
    {
      "epoch": 2.2272109985351562e-05,
      "model_forward_time": 1.7191212177276611,
      "step": 23354
    },
    {
      "epoch": 2.2272109985351562e-05,
      "step": 23354,
      "training_step_time": 8.290825605392456
    },
    {
      "epoch": 2.227306365966797e-05,
      "model_forward_time": 2.2641549110412598,
      "step": 23355
    },
    {
      "epoch": 2.227306365966797e-05,
      "step": 23355,
      "training_step_time": 7.762300252914429
    },
    {
      "epoch": 2.2274017333984377e-05,
      "model_forward_time": 1.7141447067260742,
      "step": 23356
    },
    {
      "epoch": 2.2274017333984377e-05,
      "step": 23356,
      "training_step_time": 8.42939019203186
    },
    {
      "epoch": 2.227497100830078e-05,
      "model_forward_time": 1.7181475162506104,
      "step": 23357
    },
    {
      "epoch": 2.227497100830078e-05,
      "step": 23357,
      "training_step_time": 8.922390222549438
    },
    {
      "epoch": 2.2275924682617188e-05,
      "model_forward_time": 1.7155323028564453,
      "step": 23358
    },
    {
      "epoch": 2.2275924682617188e-05,
      "step": 23358,
      "training_step_time": 7.2159905433654785
    },
    {
      "epoch": 2.2276878356933595e-05,
      "model_forward_time": 1.727048635482788,
      "step": 23359
    },
    {
      "epoch": 2.2276878356933595e-05,
      "step": 23359,
      "training_step_time": 7.666569948196411
    },
    {
      "epoch": 2.227783203125e-05,
      "grad_norm": 0.17415431141853333,
      "learning_rate": 9.106469384576858e-06,
      "loss": 0.0608,
      "step": 23360
    },
    {
      "epoch": 2.227783203125e-05,
      "model_forward_time": 1.711273431777954,
      "step": 23360
    },
    {
      "dynamics_loss_avg": 0.06319192964583635,
      "epoch": 2.227783203125e-05,
      "step": 23360
    },
    {
      "action_loss_avg": 0.0035680167842656375,
      "epoch": 2.227783203125e-05,
      "step": 23360
    },
    {
      "epoch": 2.227783203125e-05,
      "step": 23360,
      "training_step_time": 6.802066326141357
    },
    {
      "epoch": 2.2278785705566407e-05,
      "model_forward_time": 1.7712087631225586,
      "step": 23361
    },
    {
      "epoch": 2.2278785705566407e-05,
      "step": 23361,
      "training_step_time": 8.624377489089966
    },
    {
      "epoch": 2.227973937988281e-05,
      "model_forward_time": 1.714482069015503,
      "step": 23362
    },
    {
      "epoch": 2.227973937988281e-05,
      "step": 23362,
      "training_step_time": 7.591683626174927
    },
    {
      "epoch": 2.228069305419922e-05,
      "model_forward_time": 1.7155141830444336,
      "step": 23363
    },
    {
      "epoch": 2.228069305419922e-05,
      "step": 23363,
      "training_step_time": 7.630671977996826
    },
    {
      "epoch": 2.2281646728515626e-05,
      "model_forward_time": 2.2539165019989014,
      "step": 23364
    },
    {
      "epoch": 2.2281646728515626e-05,
      "step": 23364,
      "training_step_time": 8.664366006851196
    },
    {
      "epoch": 2.228260040283203e-05,
      "model_forward_time": 1.7373125553131104,
      "step": 23365
    },
    {
      "epoch": 2.228260040283203e-05,
      "step": 23365,
      "training_step_time": 7.605889558792114
    },
    {
      "epoch": 2.2283554077148437e-05,
      "model_forward_time": 1.7074732780456543,
      "step": 23366
    },
    {
      "epoch": 2.2283554077148437e-05,
      "step": 23366,
      "training_step_time": 7.83379602432251
    },
    {
      "epoch": 2.2284507751464845e-05,
      "model_forward_time": 1.721846342086792,
      "step": 23367
    },
    {
      "epoch": 2.2284507751464845e-05,
      "step": 23367,
      "training_step_time": 7.7011027336120605
    },
    {
      "epoch": 2.228546142578125e-05,
      "model_forward_time": 1.7186799049377441,
      "step": 23368
    },
    {
      "epoch": 2.228546142578125e-05,
      "step": 23368,
      "training_step_time": 7.346510648727417
    },
    {
      "epoch": 2.2286415100097656e-05,
      "model_forward_time": 1.7094204425811768,
      "step": 23369
    },
    {
      "epoch": 2.2286415100097656e-05,
      "step": 23369,
      "training_step_time": 7.750515460968018
    },
    {
      "epoch": 2.2287368774414063e-05,
      "grad_norm": 0.13248075544834137,
      "learning_rate": 9.105525846658978e-06,
      "loss": 0.0598,
      "step": 23370
    },
    {
      "epoch": 2.2287368774414063e-05,
      "model_forward_time": 1.7479138374328613,
      "step": 23370
    },
    {
      "dynamics_loss_avg": 0.05383977070450783,
      "epoch": 2.2287368774414063e-05,
      "step": 23370
    },
    {
      "action_loss_avg": 0.002321326971286908,
      "epoch": 2.2287368774414063e-05,
      "step": 23370
    },
    {
      "epoch": 2.2287368774414063e-05,
      "step": 23370,
      "training_step_time": 7.5956175327301025
    },
    {
      "epoch": 2.2288322448730467e-05,
      "model_forward_time": 1.72066068649292,
      "step": 23371
    },
    {
      "epoch": 2.2288322448730467e-05,
      "step": 23371,
      "training_step_time": 9.119806051254272
    },
    {
      "epoch": 2.2289276123046875e-05,
      "model_forward_time": 1.7109858989715576,
      "step": 23372
    },
    {
      "epoch": 2.2289276123046875e-05,
      "step": 23372,
      "training_step_time": 8.13390827178955
    },
    {
      "epoch": 2.2290229797363282e-05,
      "model_forward_time": 2.2717862129211426,
      "step": 23373
    },
    {
      "epoch": 2.2290229797363282e-05,
      "step": 23373,
      "training_step_time": 9.05932879447937
    },
    {
      "epoch": 2.2291183471679686e-05,
      "model_forward_time": 1.7209465503692627,
      "step": 23374
    },
    {
      "epoch": 2.2291183471679686e-05,
      "step": 23374,
      "training_step_time": 8.714066982269287
    },
    {
      "epoch": 2.2292137145996094e-05,
      "model_forward_time": 1.7098877429962158,
      "step": 23375
    },
    {
      "epoch": 2.2292137145996094e-05,
      "step": 23375,
      "training_step_time": 7.634198427200317
    },
    {
      "epoch": 2.22930908203125e-05,
      "model_forward_time": 1.717139720916748,
      "step": 23376
    },
    {
      "epoch": 2.22930908203125e-05,
      "step": 23376,
      "training_step_time": 7.739493370056152
    },
    {
      "epoch": 2.2294044494628905e-05,
      "model_forward_time": 1.7090251445770264,
      "step": 23377
    },
    {
      "epoch": 2.2294044494628905e-05,
      "step": 23377,
      "training_step_time": 7.175810098648071
    },
    {
      "epoch": 2.2294998168945313e-05,
      "model_forward_time": 1.7229809761047363,
      "step": 23378
    },
    {
      "epoch": 2.2294998168945313e-05,
      "step": 23378,
      "training_step_time": 7.689657926559448
    },
    {
      "epoch": 2.229595184326172e-05,
      "model_forward_time": 1.7252919673919678,
      "step": 23379
    },
    {
      "epoch": 2.229595184326172e-05,
      "step": 23379,
      "training_step_time": 7.562644958496094
    },
    {
      "epoch": 2.2296905517578124e-05,
      "grad_norm": 0.12090595811605453,
      "learning_rate": 9.104581859766965e-06,
      "loss": 0.0628,
      "step": 23380
    },
    {
      "epoch": 2.2296905517578124e-05,
      "model_forward_time": 1.7138936519622803,
      "step": 23380
    },
    {
      "dynamics_loss_avg": 0.06391103640198707,
      "epoch": 2.2296905517578124e-05,
      "step": 23380
    },
    {
      "action_loss_avg": 0.0038280656561255457,
      "epoch": 2.2296905517578124e-05,
      "step": 23380
    },
    {
      "epoch": 2.2296905517578124e-05,
      "step": 23380,
      "training_step_time": 7.55704665184021
    },
    {
      "epoch": 2.229785919189453e-05,
      "model_forward_time": 1.7184433937072754,
      "step": 23381
    },
    {
      "epoch": 2.229785919189453e-05,
      "step": 23381,
      "training_step_time": 9.88556718826294
    },
    {
      "epoch": 2.229881286621094e-05,
      "model_forward_time": 2.252575397491455,
      "step": 23382
    },
    {
      "epoch": 2.229881286621094e-05,
      "step": 23382,
      "training_step_time": 7.64984917640686
    },
    {
      "epoch": 2.2299766540527343e-05,
      "model_forward_time": 1.7249658107757568,
      "step": 23383
    },
    {
      "epoch": 2.2299766540527343e-05,
      "step": 23383,
      "training_step_time": 8.395172357559204
    },
    {
      "epoch": 2.230072021484375e-05,
      "model_forward_time": 1.709744930267334,
      "step": 23384
    },
    {
      "epoch": 2.230072021484375e-05,
      "step": 23384,
      "training_step_time": 7.905589580535889
    },
    {
      "epoch": 2.2301673889160158e-05,
      "model_forward_time": 1.7125093936920166,
      "step": 23385
    },
    {
      "epoch": 2.2301673889160158e-05,
      "step": 23385,
      "training_step_time": 7.68613600730896
    },
    {
      "epoch": 2.2302627563476562e-05,
      "model_forward_time": 1.7319927215576172,
      "step": 23386
    },
    {
      "epoch": 2.2302627563476562e-05,
      "step": 23386,
      "training_step_time": 7.21773886680603
    },
    {
      "epoch": 2.230358123779297e-05,
      "model_forward_time": 1.714615821838379,
      "step": 23387
    },
    {
      "epoch": 2.230358123779297e-05,
      "step": 23387,
      "training_step_time": 7.646886110305786
    },
    {
      "epoch": 2.2304534912109377e-05,
      "model_forward_time": 1.7182724475860596,
      "step": 23388
    },
    {
      "epoch": 2.2304534912109377e-05,
      "step": 23388,
      "training_step_time": 7.613334894180298
    },
    {
      "epoch": 2.230548858642578e-05,
      "model_forward_time": 1.7186226844787598,
      "step": 23389
    },
    {
      "epoch": 2.230548858642578e-05,
      "step": 23389,
      "training_step_time": 7.650010585784912
    },
    {
      "epoch": 2.2306442260742188e-05,
      "grad_norm": 0.11796171963214874,
      "learning_rate": 9.103637424004051e-06,
      "loss": 0.059,
      "step": 23390
    },
    {
      "epoch": 2.2306442260742188e-05,
      "model_forward_time": 1.7057545185089111,
      "step": 23390
    },
    {
      "dynamics_loss_avg": 0.05577486101537943,
      "epoch": 2.2306442260742188e-05,
      "step": 23390
    },
    {
      "action_loss_avg": 0.003968478617025539,
      "epoch": 2.2306442260742188e-05,
      "step": 23390
    },
    {
      "epoch": 2.2306442260742188e-05,
      "step": 23390,
      "training_step_time": 8.203054666519165
    },
    {
      "epoch": 2.2307395935058592e-05,
      "model_forward_time": 1.704676628112793,
      "step": 23391
    },
    {
      "epoch": 2.2307395935058592e-05,
      "step": 23391,
      "training_step_time": 7.739374399185181
    },
    {
      "epoch": 2.2308349609375e-05,
      "model_forward_time": 1.8594186305999756,
      "step": 23392
    },
    {
      "epoch": 2.2308349609375e-05,
      "step": 23392,
      "training_step_time": 9.392348766326904
    },
    {
      "epoch": 2.2309303283691407e-05,
      "model_forward_time": 1.7170600891113281,
      "step": 23393
    },
    {
      "epoch": 2.2309303283691407e-05,
      "step": 23393,
      "training_step_time": 7.665418386459351
    },
    {
      "epoch": 2.231025695800781e-05,
      "model_forward_time": 1.709758996963501,
      "step": 23394
    },
    {
      "epoch": 2.231025695800781e-05,
      "step": 23394,
      "training_step_time": 8.948159456253052
    },
    {
      "epoch": 2.231121063232422e-05,
      "model_forward_time": 1.7027907371520996,
      "step": 23395
    },
    {
      "epoch": 2.231121063232422e-05,
      "step": 23395,
      "training_step_time": 7.209122180938721
    },
    {
      "epoch": 2.2312164306640626e-05,
      "model_forward_time": 1.7043352127075195,
      "step": 23396
    },
    {
      "epoch": 2.2312164306640626e-05,
      "step": 23396,
      "training_step_time": 7.520405530929565
    },
    {
      "epoch": 2.231311798095703e-05,
      "model_forward_time": 1.7112281322479248,
      "step": 23397
    },
    {
      "epoch": 2.231311798095703e-05,
      "step": 23397,
      "training_step_time": 7.506075382232666
    },
    {
      "epoch": 2.2314071655273437e-05,
      "model_forward_time": 1.7674486637115479,
      "step": 23398
    },
    {
      "epoch": 2.2314071655273437e-05,
      "step": 23398,
      "training_step_time": 7.5828142166137695
    },
    {
      "epoch": 2.2315025329589845e-05,
      "model_forward_time": 1.700465440750122,
      "step": 23399
    },
    {
      "epoch": 2.2315025329589845e-05,
      "step": 23399,
      "training_step_time": 8.378047704696655
    },
    {
      "epoch": 2.231597900390625e-05,
      "grad_norm": 0.13403917849063873,
      "learning_rate": 9.102692539473519e-06,
      "loss": 0.0557,
      "step": 23400
    },
    {
      "epoch": 2.231597900390625e-05,
      "model_forward_time": 1.7429530620574951,
      "step": 23400
    },
    {
      "dynamics_loss_avg": 0.07047867849469185,
      "epoch": 2.231597900390625e-05,
      "step": 23400
    },
    {
      "action_loss_avg": 0.004569559759693221,
      "epoch": 2.231597900390625e-05,
      "step": 23400
    },
    {
      "epoch": 2.231597900390625e-05,
      "step": 23400,
      "training_step_time": 8.603603601455688
    },
    {
      "epoch": 2.2316932678222656e-05,
      "model_forward_time": 2.910677194595337,
      "step": 23401
    },
    {
      "epoch": 2.2316932678222656e-05,
      "step": 23401,
      "training_step_time": 8.55235505104065
    },
    {
      "epoch": 2.2317886352539063e-05,
      "model_forward_time": 1.750558614730835,
      "step": 23402
    },
    {
      "epoch": 2.2317886352539063e-05,
      "step": 23402,
      "training_step_time": 9.28277587890625
    },
    {
      "epoch": 2.2318840026855468e-05,
      "model_forward_time": 1.748270034790039,
      "step": 23403
    },
    {
      "epoch": 2.2318840026855468e-05,
      "step": 23403,
      "training_step_time": 8.748913764953613
    },
    {
      "epoch": 2.2319793701171875e-05,
      "model_forward_time": 1.7467377185821533,
      "step": 23404
    },
    {
      "epoch": 2.2319793701171875e-05,
      "step": 23404,
      "training_step_time": 7.569498062133789
    },
    {
      "epoch": 2.2320747375488282e-05,
      "model_forward_time": 1.7502567768096924,
      "step": 23405
    },
    {
      "epoch": 2.2320747375488282e-05,
      "step": 23405,
      "training_step_time": 6.782505989074707
    },
    {
      "epoch": 2.2321701049804686e-05,
      "model_forward_time": 1.7311394214630127,
      "step": 23406
    },
    {
      "epoch": 2.2321701049804686e-05,
      "step": 23406,
      "training_step_time": 7.728009939193726
    },
    {
      "epoch": 2.2322654724121094e-05,
      "model_forward_time": 1.7355117797851562,
      "step": 23407
    },
    {
      "epoch": 2.2322654724121094e-05,
      "step": 23407,
      "training_step_time": 7.633152484893799
    },
    {
      "epoch": 2.23236083984375e-05,
      "model_forward_time": 1.7420244216918945,
      "step": 23408
    },
    {
      "epoch": 2.23236083984375e-05,
      "step": 23408,
      "training_step_time": 8.499492168426514
    },
    {
      "epoch": 2.2324562072753905e-05,
      "model_forward_time": 1.72849702835083,
      "step": 23409
    },
    {
      "epoch": 2.2324562072753905e-05,
      "step": 23409,
      "training_step_time": 8.225238800048828
    },
    {
      "epoch": 2.2325515747070313e-05,
      "grad_norm": 0.16118605434894562,
      "learning_rate": 9.101747206278698e-06,
      "loss": 0.0587,
      "step": 23410
    },
    {
      "epoch": 2.2325515747070313e-05,
      "model_forward_time": 2.878993511199951,
      "step": 23410
    },
    {
      "dynamics_loss_avg": 0.05216172095388174,
      "epoch": 2.2325515747070313e-05,
      "step": 23410
    },
    {
      "action_loss_avg": 0.0023702716222032904,
      "epoch": 2.2325515747070313e-05,
      "step": 23410
    },
    {
      "epoch": 2.2325515747070313e-05,
      "step": 23410,
      "training_step_time": 9.203159093856812
    },
    {
      "epoch": 2.232646942138672e-05,
      "model_forward_time": 1.739492654800415,
      "step": 23411
    },
    {
      "epoch": 2.232646942138672e-05,
      "step": 23411,
      "training_step_time": 8.51718020439148
    },
    {
      "epoch": 2.2327423095703124e-05,
      "model_forward_time": 1.7431273460388184,
      "step": 23412
    },
    {
      "epoch": 2.2327423095703124e-05,
      "step": 23412,
      "training_step_time": 7.838452100753784
    },
    {
      "epoch": 2.232837677001953e-05,
      "model_forward_time": 1.7448217868804932,
      "step": 23413
    },
    {
      "epoch": 2.232837677001953e-05,
      "step": 23413,
      "training_step_time": 7.664191246032715
    },
    {
      "epoch": 2.232933044433594e-05,
      "model_forward_time": 1.7415797710418701,
      "step": 23414
    },
    {
      "epoch": 2.232933044433594e-05,
      "step": 23414,
      "training_step_time": 6.789217233657837
    },
    {
      "epoch": 2.2330284118652343e-05,
      "model_forward_time": 1.7485706806182861,
      "step": 23415
    },
    {
      "epoch": 2.2330284118652343e-05,
      "step": 23415,
      "training_step_time": 7.493785858154297
    },
    {
      "epoch": 2.233123779296875e-05,
      "model_forward_time": 1.744572639465332,
      "step": 23416
    },
    {
      "epoch": 2.233123779296875e-05,
      "step": 23416,
      "training_step_time": 8.224634170532227
    },
    {
      "epoch": 2.2332191467285158e-05,
      "model_forward_time": 1.762063980102539,
      "step": 23417
    },
    {
      "epoch": 2.2332191467285158e-05,
      "step": 23417,
      "training_step_time": 8.197983264923096
    },
    {
      "epoch": 2.2333145141601562e-05,
      "model_forward_time": 1.7602965831756592,
      "step": 23418
    },
    {
      "epoch": 2.2333145141601562e-05,
      "step": 23418,
      "training_step_time": 8.368239402770996
    },
    {
      "epoch": 2.233409881591797e-05,
      "model_forward_time": 2.8551719188690186,
      "step": 23419
    },
    {
      "epoch": 2.233409881591797e-05,
      "step": 23419,
      "training_step_time": 9.660789489746094
    },
    {
      "epoch": 2.2335052490234377e-05,
      "grad_norm": 0.124728262424469,
      "learning_rate": 9.100801424522968e-06,
      "loss": 0.0604,
      "step": 23420
    },
    {
      "epoch": 2.2335052490234377e-05,
      "model_forward_time": 1.7567095756530762,
      "step": 23420
    },
    {
      "dynamics_loss_avg": 0.05463692583143711,
      "epoch": 2.2335052490234377e-05,
      "step": 23420
    },
    {
      "action_loss_avg": 0.0029759365133941174,
      "epoch": 2.2335052490234377e-05,
      "step": 23420
    },
    {
      "epoch": 2.2335052490234377e-05,
      "step": 23420,
      "training_step_time": 7.669029474258423
    },
    {
      "epoch": 2.233600616455078e-05,
      "model_forward_time": 1.758561611175537,
      "step": 23421
    },
    {
      "epoch": 2.233600616455078e-05,
      "step": 23421,
      "training_step_time": 8.352583169937134
    },
    {
      "epoch": 2.2336959838867188e-05,
      "model_forward_time": 1.7547788619995117,
      "step": 23422
    },
    {
      "epoch": 2.2336959838867188e-05,
      "step": 23422,
      "training_step_time": 7.654508590698242
    },
    {
      "epoch": 2.2337913513183592e-05,
      "model_forward_time": 1.7568888664245605,
      "step": 23423
    },
    {
      "epoch": 2.2337913513183592e-05,
      "step": 23423,
      "training_step_time": 7.222382068634033
    },
    {
      "epoch": 2.23388671875e-05,
      "model_forward_time": 1.7528190612792969,
      "step": 23424
    },
    {
      "epoch": 2.23388671875e-05,
      "step": 23424,
      "training_step_time": 7.515112638473511
    },
    {
      "epoch": 2.2339820861816407e-05,
      "model_forward_time": 1.394728660583496,
      "step": 23425
    },
    {
      "epoch": 2.2339820861816407e-05,
      "step": 23425,
      "training_step_time": 7.664692640304565
    },
    {
      "epoch": 2.234077453613281e-05,
      "model_forward_time": 1.7505877017974854,
      "step": 23426
    },
    {
      "epoch": 2.234077453613281e-05,
      "step": 23426,
      "training_step_time": 7.651096343994141
    },
    {
      "epoch": 2.234172821044922e-05,
      "model_forward_time": 1.8567500114440918,
      "step": 23427
    },
    {
      "epoch": 2.234172821044922e-05,
      "step": 23427,
      "training_step_time": 7.670115232467651
    },
    {
      "epoch": 2.2342681884765626e-05,
      "model_forward_time": 1.766153335571289,
      "step": 23428
    },
    {
      "epoch": 2.2342681884765626e-05,
      "step": 23428,
      "training_step_time": 8.156042575836182
    },
    {
      "epoch": 2.234363555908203e-05,
      "model_forward_time": 2.9323320388793945,
      "step": 23429
    },
    {
      "epoch": 2.234363555908203e-05,
      "step": 23429,
      "training_step_time": 7.713673114776611
    },
    {
      "epoch": 2.2344589233398437e-05,
      "grad_norm": 0.1278773993253708,
      "learning_rate": 9.09985519430976e-06,
      "loss": 0.0576,
      "step": 23430
    },
    {
      "epoch": 2.2344589233398437e-05,
      "model_forward_time": 1.7606689929962158,
      "step": 23430
    },
    {
      "dynamics_loss_avg": 0.061522855423390865,
      "epoch": 2.2344589233398437e-05,
      "step": 23430
    },
    {
      "action_loss_avg": 0.003274581750156358,
      "epoch": 2.2344589233398437e-05,
      "step": 23430
    },
    {
      "epoch": 2.2344589233398437e-05,
      "step": 23430,
      "training_step_time": 7.666607618331909
    },
    {
      "epoch": 2.2345542907714845e-05,
      "model_forward_time": 1.7738690376281738,
      "step": 23431
    },
    {
      "epoch": 2.2345542907714845e-05,
      "step": 23431,
      "training_step_time": 7.785904884338379
    },
    {
      "epoch": 2.234649658203125e-05,
      "model_forward_time": 1.7579624652862549,
      "step": 23432
    },
    {
      "epoch": 2.234649658203125e-05,
      "step": 23432,
      "training_step_time": 7.196115016937256
    },
    {
      "epoch": 2.2347450256347656e-05,
      "model_forward_time": 1.7649412155151367,
      "step": 23433
    },
    {
      "epoch": 2.2347450256347656e-05,
      "step": 23433,
      "training_step_time": 7.4937684535980225
    },
    {
      "epoch": 2.2348403930664064e-05,
      "model_forward_time": 1.7664380073547363,
      "step": 23434
    },
    {
      "epoch": 2.2348403930664064e-05,
      "step": 23434,
      "training_step_time": 7.597834348678589
    },
    {
      "epoch": 2.2349357604980468e-05,
      "model_forward_time": 1.7580609321594238,
      "step": 23435
    },
    {
      "epoch": 2.2349357604980468e-05,
      "step": 23435,
      "training_step_time": 7.624163866043091
    },
    {
      "epoch": 2.2350311279296875e-05,
      "model_forward_time": 1.8489391803741455,
      "step": 23436
    },
    {
      "epoch": 2.2350311279296875e-05,
      "step": 23436,
      "training_step_time": 7.582671165466309
    },
    {
      "epoch": 2.2351264953613282e-05,
      "model_forward_time": 1.774911642074585,
      "step": 23437
    },
    {
      "epoch": 2.2351264953613282e-05,
      "step": 23437,
      "training_step_time": 8.843183517456055
    },
    {
      "epoch": 2.2352218627929686e-05,
      "model_forward_time": 2.9427738189697266,
      "step": 23438
    },
    {
      "epoch": 2.2352218627929686e-05,
      "step": 23438,
      "training_step_time": 10.42308497428894
    },
    {
      "epoch": 2.2353172302246094e-05,
      "model_forward_time": 1.7426722049713135,
      "step": 23439
    },
    {
      "epoch": 2.2353172302246094e-05,
      "step": 23439,
      "training_step_time": 7.607486009597778
    },
    {
      "epoch": 2.23541259765625e-05,
      "grad_norm": 0.13317185640335083,
      "learning_rate": 9.098908515742553e-06,
      "loss": 0.0588,
      "step": 23440
    },
    {
      "epoch": 2.23541259765625e-05,
      "model_forward_time": 1.7569849491119385,
      "step": 23440
    },
    {
      "dynamics_loss_avg": 0.0557192225009203,
      "epoch": 2.23541259765625e-05,
      "step": 23440
    },
    {
      "action_loss_avg": 0.0024025667051319035,
      "epoch": 2.23541259765625e-05,
      "step": 23440
    },
    {
      "epoch": 2.23541259765625e-05,
      "step": 23440,
      "training_step_time": 7.870156764984131
    },
    {
      "epoch": 2.2355079650878905e-05,
      "model_forward_time": 1.7714829444885254,
      "step": 23441
    },
    {
      "epoch": 2.2355079650878905e-05,
      "step": 23441,
      "training_step_time": 7.582207918167114
    },
    {
      "epoch": 2.2356033325195313e-05,
      "model_forward_time": 1.7599127292633057,
      "step": 23442
    },
    {
      "epoch": 2.2356033325195313e-05,
      "step": 23442,
      "training_step_time": 7.47126317024231
    },
    {
      "epoch": 2.235698699951172e-05,
      "model_forward_time": 1.7654550075531006,
      "step": 23443
    },
    {
      "epoch": 2.235698699951172e-05,
      "step": 23443,
      "training_step_time": 8.66970682144165
    },
    {
      "epoch": 2.2357940673828124e-05,
      "model_forward_time": 1.7644097805023193,
      "step": 23444
    },
    {
      "epoch": 2.2357940673828124e-05,
      "step": 23444,
      "training_step_time": 7.564527988433838
    },
    {
      "epoch": 2.235889434814453e-05,
      "model_forward_time": 1.8188636302947998,
      "step": 23445
    },
    {
      "epoch": 2.235889434814453e-05,
      "step": 23445,
      "training_step_time": 8.301952838897705
    },
    {
      "epoch": 2.235984802246094e-05,
      "model_forward_time": 1.7713971138000488,
      "step": 23446
    },
    {
      "epoch": 2.235984802246094e-05,
      "step": 23446,
      "training_step_time": 8.313435077667236
    },
    {
      "epoch": 2.2360801696777343e-05,
      "model_forward_time": 2.8897621631622314,
      "step": 23447
    },
    {
      "epoch": 2.2360801696777343e-05,
      "step": 23447,
      "training_step_time": 9.358724117279053
    },
    {
      "epoch": 2.236175537109375e-05,
      "model_forward_time": 1.7559468746185303,
      "step": 23448
    },
    {
      "epoch": 2.236175537109375e-05,
      "step": 23448,
      "training_step_time": 7.576624155044556
    },
    {
      "epoch": 2.2362709045410158e-05,
      "model_forward_time": 1.7710692882537842,
      "step": 23449
    },
    {
      "epoch": 2.2362709045410158e-05,
      "step": 23449,
      "training_step_time": 7.80858039855957
    },
    {
      "epoch": 2.2363662719726562e-05,
      "grad_norm": 0.15890300273895264,
      "learning_rate": 9.097961388924875e-06,
      "loss": 0.0602,
      "step": 23450
    },
    {
      "epoch": 2.2363662719726562e-05,
      "model_forward_time": 1.77357816696167,
      "step": 23450
    },
    {
      "dynamics_loss_avg": 0.05547786429524422,
      "epoch": 2.2363662719726562e-05,
      "step": 23450
    },
    {
      "action_loss_avg": 0.0032008474809117615,
      "epoch": 2.2363662719726562e-05,
      "step": 23450
    },
    {
      "epoch": 2.2363662719726562e-05,
      "step": 23450,
      "training_step_time": 7.627510070800781
    },
    {
      "epoch": 2.236461639404297e-05,
      "model_forward_time": 1.7634360790252686,
      "step": 23451
    },
    {
      "epoch": 2.236461639404297e-05,
      "step": 23451,
      "training_step_time": 7.464986801147461
    },
    {
      "epoch": 2.2365570068359377e-05,
      "model_forward_time": 1.7696564197540283,
      "step": 23452
    },
    {
      "epoch": 2.2365570068359377e-05,
      "step": 23452,
      "training_step_time": 7.600894451141357
    },
    {
      "epoch": 2.236652374267578e-05,
      "model_forward_time": 1.7747607231140137,
      "step": 23453
    },
    {
      "epoch": 2.236652374267578e-05,
      "step": 23453,
      "training_step_time": 8.625241756439209
    },
    {
      "epoch": 2.2367477416992188e-05,
      "model_forward_time": 1.7725751399993896,
      "step": 23454
    },
    {
      "epoch": 2.2367477416992188e-05,
      "step": 23454,
      "training_step_time": 7.664399862289429
    },
    {
      "epoch": 2.2368431091308592e-05,
      "model_forward_time": 1.7661919593811035,
      "step": 23455
    },
    {
      "epoch": 2.2368431091308592e-05,
      "step": 23455,
      "training_step_time": 8.186574459075928
    },
    {
      "epoch": 2.2369384765625e-05,
      "model_forward_time": 2.8965322971343994,
      "step": 23456
    },
    {
      "epoch": 2.2369384765625e-05,
      "step": 23456,
      "training_step_time": 7.631665945053101
    },
    {
      "epoch": 2.2370338439941407e-05,
      "model_forward_time": 1.7602834701538086,
      "step": 23457
    },
    {
      "epoch": 2.2370338439941407e-05,
      "step": 23457,
      "training_step_time": 9.155379295349121
    },
    {
      "epoch": 2.237129211425781e-05,
      "model_forward_time": 1.762270212173462,
      "step": 23458
    },
    {
      "epoch": 2.237129211425781e-05,
      "step": 23458,
      "training_step_time": 7.773677110671997
    },
    {
      "epoch": 2.237224578857422e-05,
      "model_forward_time": 1.7592647075653076,
      "step": 23459
    },
    {
      "epoch": 2.237224578857422e-05,
      "step": 23459,
      "training_step_time": 7.57036566734314
    },
    {
      "epoch": 2.2373199462890626e-05,
      "grad_norm": 0.15120156109333038,
      "learning_rate": 9.097013813960299e-06,
      "loss": 0.061,
      "step": 23460
    },
    {
      "epoch": 2.2373199462890626e-05,
      "model_forward_time": 1.767500638961792,
      "step": 23460
    },
    {
      "dynamics_loss_avg": 0.05320488791912794,
      "epoch": 2.2373199462890626e-05,
      "step": 23460
    },
    {
      "action_loss_avg": 0.0038362654973752796,
      "epoch": 2.2373199462890626e-05,
      "step": 23460
    },
    {
      "epoch": 2.2373199462890626e-05,
      "step": 23460,
      "training_step_time": 7.437870025634766
    },
    {
      "epoch": 2.237415313720703e-05,
      "model_forward_time": 1.7726314067840576,
      "step": 23461
    },
    {
      "epoch": 2.237415313720703e-05,
      "step": 23461,
      "training_step_time": 7.619577407836914
    },
    {
      "epoch": 2.2375106811523437e-05,
      "model_forward_time": 1.7579100131988525,
      "step": 23462
    },
    {
      "epoch": 2.2375106811523437e-05,
      "step": 23462,
      "training_step_time": 7.673564672470093
    },
    {
      "epoch": 2.2376060485839845e-05,
      "model_forward_time": 1.762739896774292,
      "step": 23463
    },
    {
      "epoch": 2.2376060485839845e-05,
      "step": 23463,
      "training_step_time": 7.613199234008789
    },
    {
      "epoch": 2.237701416015625e-05,
      "model_forward_time": 1.8928465843200684,
      "step": 23464
    },
    {
      "epoch": 2.237701416015625e-05,
      "step": 23464,
      "training_step_time": 8.3028883934021
    },
    {
      "epoch": 2.2377967834472656e-05,
      "model_forward_time": 2.8797109127044678,
      "step": 23465
    },
    {
      "epoch": 2.2377967834472656e-05,
      "step": 23465,
      "training_step_time": 8.654998302459717
    },
    {
      "epoch": 2.2378921508789064e-05,
      "model_forward_time": 1.7773010730743408,
      "step": 23466
    },
    {
      "epoch": 2.2378921508789064e-05,
      "step": 23466,
      "training_step_time": 7.664902687072754
    },
    {
      "epoch": 2.2379875183105468e-05,
      "model_forward_time": 1.7646501064300537,
      "step": 23467
    },
    {
      "epoch": 2.2379875183105468e-05,
      "step": 23467,
      "training_step_time": 8.654831409454346
    },
    {
      "epoch": 2.2380828857421875e-05,
      "model_forward_time": 1.7607808113098145,
      "step": 23468
    },
    {
      "epoch": 2.2380828857421875e-05,
      "step": 23468,
      "training_step_time": 7.628392696380615
    },
    {
      "epoch": 2.2381782531738282e-05,
      "model_forward_time": 1.7642590999603271,
      "step": 23469
    },
    {
      "epoch": 2.2381782531738282e-05,
      "step": 23469,
      "training_step_time": 7.423223495483398
    },
    {
      "epoch": 2.2382736206054686e-05,
      "grad_norm": 0.09743522107601166,
      "learning_rate": 9.096065790952451e-06,
      "loss": 0.0617,
      "step": 23470
    },
    {
      "epoch": 2.2382736206054686e-05,
      "model_forward_time": 1.7615759372711182,
      "step": 23470
    },
    {
      "dynamics_loss_avg": 0.04254537355154753,
      "epoch": 2.2382736206054686e-05,
      "step": 23470
    },
    {
      "action_loss_avg": 0.002670432551531121,
      "epoch": 2.2382736206054686e-05,
      "step": 23470
    },
    {
      "epoch": 2.2382736206054686e-05,
      "step": 23470,
      "training_step_time": 7.256927490234375
    },
    {
      "epoch": 2.2383689880371094e-05,
      "model_forward_time": 1.7599501609802246,
      "step": 23471
    },
    {
      "epoch": 2.2383689880371094e-05,
      "step": 23471,
      "training_step_time": 8.611355543136597
    },
    {
      "epoch": 2.23846435546875e-05,
      "model_forward_time": 1.7529852390289307,
      "step": 23472
    },
    {
      "epoch": 2.23846435546875e-05,
      "step": 23472,
      "training_step_time": 7.6285295486450195
    },
    {
      "epoch": 2.2385597229003905e-05,
      "model_forward_time": 1.8771965503692627,
      "step": 23473
    },
    {
      "epoch": 2.2385597229003905e-05,
      "step": 23473,
      "training_step_time": 8.950585842132568
    },
    {
      "epoch": 2.2386550903320313e-05,
      "model_forward_time": 1.7689588069915771,
      "step": 23474
    },
    {
      "epoch": 2.2386550903320313e-05,
      "step": 23474,
      "training_step_time": 9.030824661254883
    },
    {
      "epoch": 2.238750457763672e-05,
      "model_forward_time": 1.7599937915802002,
      "step": 23475
    },
    {
      "epoch": 2.238750457763672e-05,
      "step": 23475,
      "training_step_time": 7.614819049835205
    },
    {
      "epoch": 2.2388458251953124e-05,
      "model_forward_time": 1.766965627670288,
      "step": 23476
    },
    {
      "epoch": 2.2388458251953124e-05,
      "step": 23476,
      "training_step_time": 7.63171124458313
    },
    {
      "epoch": 2.238941192626953e-05,
      "model_forward_time": 1.7592005729675293,
      "step": 23477
    },
    {
      "epoch": 2.238941192626953e-05,
      "step": 23477,
      "training_step_time": 7.689559459686279
    },
    {
      "epoch": 2.239036560058594e-05,
      "model_forward_time": 1.7605676651000977,
      "step": 23478
    },
    {
      "epoch": 2.239036560058594e-05,
      "step": 23478,
      "training_step_time": 8.227109432220459
    },
    {
      "epoch": 2.2391319274902343e-05,
      "model_forward_time": 1.7703609466552734,
      "step": 23479
    },
    {
      "epoch": 2.2391319274902343e-05,
      "step": 23479,
      "training_step_time": 7.570056438446045
    },
    {
      "epoch": 2.239227294921875e-05,
      "grad_norm": 0.18003268539905548,
      "learning_rate": 9.095117320005007e-06,
      "loss": 0.0607,
      "step": 23480
    },
    {
      "epoch": 2.239227294921875e-05,
      "model_forward_time": 1.7657623291015625,
      "step": 23480
    },
    {
      "dynamics_loss_avg": 0.053265373967587945,
      "epoch": 2.239227294921875e-05,
      "step": 23480
    },
    {
      "action_loss_avg": 0.004397912276908755,
      "epoch": 2.239227294921875e-05,
      "step": 23480
    },
    {
      "epoch": 2.239227294921875e-05,
      "step": 23480,
      "training_step_time": 8.45947551727295
    },
    {
      "epoch": 2.2393226623535158e-05,
      "model_forward_time": 1.7594444751739502,
      "step": 23481
    },
    {
      "epoch": 2.2393226623535158e-05,
      "step": 23481,
      "training_step_time": 7.633842945098877
    },
    {
      "epoch": 2.2394180297851562e-05,
      "model_forward_time": 1.8982949256896973,
      "step": 23482
    },
    {
      "epoch": 2.2394180297851562e-05,
      "step": 23482,
      "training_step_time": 7.66860294342041
    },
    {
      "epoch": 2.239513397216797e-05,
      "model_forward_time": 1.755483865737915,
      "step": 23483
    },
    {
      "epoch": 2.239513397216797e-05,
      "step": 23483,
      "training_step_time": 10.030671834945679
    },
    {
      "epoch": 2.2396087646484377e-05,
      "model_forward_time": 2.9456732273101807,
      "step": 23484
    },
    {
      "epoch": 2.2396087646484377e-05,
      "step": 23484,
      "training_step_time": 7.706777572631836
    },
    {
      "epoch": 2.239704132080078e-05,
      "model_forward_time": 1.7525274753570557,
      "step": 23485
    },
    {
      "epoch": 2.239704132080078e-05,
      "step": 23485,
      "training_step_time": 7.5762553215026855
    },
    {
      "epoch": 2.2397994995117188e-05,
      "model_forward_time": 1.7550890445709229,
      "step": 23486
    },
    {
      "epoch": 2.2397994995117188e-05,
      "step": 23486,
      "training_step_time": 7.643211841583252
    },
    {
      "epoch": 2.2398948669433592e-05,
      "model_forward_time": 1.7790043354034424,
      "step": 23487
    },
    {
      "epoch": 2.2398948669433592e-05,
      "step": 23487,
      "training_step_time": 7.602446556091309
    },
    {
      "epoch": 2.239990234375e-05,
      "model_forward_time": 1.740312099456787,
      "step": 23488
    },
    {
      "epoch": 2.239990234375e-05,
      "step": 23488,
      "training_step_time": 7.17336368560791
    },
    {
      "epoch": 2.2400856018066407e-05,
      "model_forward_time": 1.7494761943817139,
      "step": 23489
    },
    {
      "epoch": 2.2400856018066407e-05,
      "step": 23489,
      "training_step_time": 7.608770370483398
    },
    {
      "epoch": 2.240180969238281e-05,
      "grad_norm": 0.13407711684703827,
      "learning_rate": 9.094168401221692e-06,
      "loss": 0.0639,
      "step": 23490
    },
    {
      "epoch": 2.240180969238281e-05,
      "model_forward_time": 1.7533433437347412,
      "step": 23490
    },
    {
      "dynamics_loss_avg": 0.05395672544836998,
      "epoch": 2.240180969238281e-05,
      "step": 23490
    },
    {
      "action_loss_avg": 0.002131819468922913,
      "epoch": 2.240180969238281e-05,
      "step": 23490
    },
    {
      "epoch": 2.240180969238281e-05,
      "step": 23490,
      "training_step_time": 8.436471462249756
    },
    {
      "epoch": 2.240276336669922e-05,
      "model_forward_time": 1.8693723678588867,
      "step": 23491
    },
    {
      "epoch": 2.240276336669922e-05,
      "step": 23491,
      "training_step_time": 8.486379861831665
    },
    {
      "epoch": 2.2403717041015626e-05,
      "model_forward_time": 1.751396894454956,
      "step": 23492
    },
    {
      "epoch": 2.2403717041015626e-05,
      "step": 23492,
      "training_step_time": 8.332859516143799
    },
    {
      "epoch": 2.240467071533203e-05,
      "model_forward_time": 2.9244253635406494,
      "step": 23493
    },
    {
      "epoch": 2.240467071533203e-05,
      "step": 23493,
      "training_step_time": 9.77270770072937
    },
    {
      "epoch": 2.2405624389648437e-05,
      "model_forward_time": 1.753833293914795,
      "step": 23494
    },
    {
      "epoch": 2.2405624389648437e-05,
      "step": 23494,
      "training_step_time": 8.495249509811401
    },
    {
      "epoch": 2.2406578063964845e-05,
      "model_forward_time": 1.7607481479644775,
      "step": 23495
    },
    {
      "epoch": 2.2406578063964845e-05,
      "step": 23495,
      "training_step_time": 9.211531162261963
    },
    {
      "epoch": 2.240753173828125e-05,
      "model_forward_time": 1.7545888423919678,
      "step": 23496
    },
    {
      "epoch": 2.240753173828125e-05,
      "step": 23496,
      "training_step_time": 7.603411912918091
    },
    {
      "epoch": 2.2408485412597656e-05,
      "model_forward_time": 1.7629954814910889,
      "step": 23497
    },
    {
      "epoch": 2.2408485412597656e-05,
      "step": 23497,
      "training_step_time": 7.208204984664917
    },
    {
      "epoch": 2.2409439086914064e-05,
      "model_forward_time": 1.7634294033050537,
      "step": 23498
    },
    {
      "epoch": 2.2409439086914064e-05,
      "step": 23498,
      "training_step_time": 7.621400833129883
    },
    {
      "epoch": 2.2410392761230468e-05,
      "model_forward_time": 1.7398569583892822,
      "step": 23499
    },
    {
      "epoch": 2.2410392761230468e-05,
      "step": 23499,
      "training_step_time": 8.519066572189331
    },
    {
      "epoch": 2.2411346435546875e-05,
      "grad_norm": 0.11171755939722061,
      "learning_rate": 9.093219034706274e-06,
      "loss": 0.0648,
      "step": 23500
    },
    {
      "epoch": 2.2411346435546875e-05,
      "model_forward_time": 1.8764991760253906,
      "step": 23500
    },
    {
      "dynamics_loss_avg": 0.05839511901140213,
      "epoch": 2.2411346435546875e-05,
      "step": 23500
    },
    {
      "action_loss_avg": 0.0018355786451138556,
      "epoch": 2.2411346435546875e-05,
      "step": 23500
    },
    {
      "epoch": 2.2411346435546875e-05,
      "step": 23500,
      "training_step_time": 8.357688903808594
    },
    {
      "epoch": 2.2412300109863282e-05,
      "model_forward_time": 1.7437148094177246,
      "step": 23501
    },
    {
      "epoch": 2.2412300109863282e-05,
      "step": 23501,
      "training_step_time": 8.188286542892456
    },
    {
      "epoch": 2.2413253784179686e-05,
      "model_forward_time": 2.9382731914520264,
      "step": 23502
    },
    {
      "epoch": 2.2413253784179686e-05,
      "step": 23502,
      "training_step_time": 9.023150444030762
    },
    {
      "epoch": 2.2414207458496094e-05,
      "model_forward_time": 1.749678373336792,
      "step": 23503
    },
    {
      "epoch": 2.2414207458496094e-05,
      "step": 23503,
      "training_step_time": 7.658809423446655
    },
    {
      "epoch": 2.24151611328125e-05,
      "model_forward_time": 1.7495520114898682,
      "step": 23504
    },
    {
      "epoch": 2.24151611328125e-05,
      "step": 23504,
      "training_step_time": 7.7100989818573
    },
    {
      "epoch": 2.2416114807128905e-05,
      "model_forward_time": 1.750844955444336,
      "step": 23505
    },
    {
      "epoch": 2.2416114807128905e-05,
      "step": 23505,
      "training_step_time": 7.6680848598480225
    },
    {
      "epoch": 2.2417068481445313e-05,
      "model_forward_time": 1.7580065727233887,
      "step": 23506
    },
    {
      "epoch": 2.2417068481445313e-05,
      "step": 23506,
      "training_step_time": 7.568033456802368
    },
    {
      "epoch": 2.241802215576172e-05,
      "model_forward_time": 1.7570807933807373,
      "step": 23507
    },
    {
      "epoch": 2.241802215576172e-05,
      "step": 23507,
      "training_step_time": 7.68452525138855
    },
    {
      "epoch": 2.2418975830078124e-05,
      "model_forward_time": 1.7548620700836182,
      "step": 23508
    },
    {
      "epoch": 2.2418975830078124e-05,
      "step": 23508,
      "training_step_time": 7.553647756576538
    },
    {
      "epoch": 2.241992950439453e-05,
      "model_forward_time": 1.8680427074432373,
      "step": 23509
    },
    {
      "epoch": 2.241992950439453e-05,
      "step": 23509,
      "training_step_time": 7.696805715560913
    },
    {
      "epoch": 2.242088317871094e-05,
      "grad_norm": 0.15924052894115448,
      "learning_rate": 9.092269220562578e-06,
      "loss": 0.061,
      "step": 23510
    },
    {
      "epoch": 2.242088317871094e-05,
      "model_forward_time": 1.7649078369140625,
      "step": 23510
    },
    {
      "dynamics_loss_avg": 0.05292968638241291,
      "epoch": 2.242088317871094e-05,
      "step": 23510
    },
    {
      "action_loss_avg": 0.003221789037343115,
      "epoch": 2.242088317871094e-05,
      "step": 23510
    },
    {
      "epoch": 2.242088317871094e-05,
      "step": 23510,
      "training_step_time": 9.854641675949097
    },
    {
      "epoch": 2.2421836853027343e-05,
      "model_forward_time": 2.9053447246551514,
      "step": 23511
    },
    {
      "epoch": 2.2421836853027343e-05,
      "step": 23511,
      "training_step_time": 9.904275894165039
    },
    {
      "epoch": 2.242279052734375e-05,
      "model_forward_time": 1.7690620422363281,
      "step": 23512
    },
    {
      "epoch": 2.242279052734375e-05,
      "step": 23512,
      "training_step_time": 9.559625148773193
    },
    {
      "epoch": 2.2423744201660158e-05,
      "model_forward_time": 1.7593088150024414,
      "step": 23513
    },
    {
      "epoch": 2.2423744201660158e-05,
      "step": 23513,
      "training_step_time": 8.341394424438477
    },
    {
      "epoch": 2.2424697875976562e-05,
      "model_forward_time": 1.7719173431396484,
      "step": 23514
    },
    {
      "epoch": 2.2424697875976562e-05,
      "step": 23514,
      "training_step_time": 7.6330246925354
    },
    {
      "epoch": 2.242565155029297e-05,
      "model_forward_time": 1.7617709636688232,
      "step": 23515
    },
    {
      "epoch": 2.242565155029297e-05,
      "step": 23515,
      "training_step_time": 7.5292603969573975
    },
    {
      "epoch": 2.2426605224609373e-05,
      "model_forward_time": 1.7638766765594482,
      "step": 23516
    },
    {
      "epoch": 2.2426605224609373e-05,
      "step": 23516,
      "training_step_time": 7.611003875732422
    },
    {
      "epoch": 2.242755889892578e-05,
      "model_forward_time": 1.7685785293579102,
      "step": 23517
    },
    {
      "epoch": 2.242755889892578e-05,
      "step": 23517,
      "training_step_time": 7.505959510803223
    },
    {
      "epoch": 2.2428512573242188e-05,
      "model_forward_time": 1.876255989074707,
      "step": 23518
    },
    {
      "epoch": 2.2428512573242188e-05,
      "step": 23518,
      "training_step_time": 7.669949293136597
    },
    {
      "epoch": 2.2429466247558592e-05,
      "model_forward_time": 1.7701160907745361,
      "step": 23519
    },
    {
      "epoch": 2.2429466247558592e-05,
      "step": 23519,
      "training_step_time": 8.413041830062866
    },
    {
      "epoch": 2.2430419921875e-05,
      "grad_norm": 0.14193888008594513,
      "learning_rate": 9.091318958894471e-06,
      "loss": 0.0647,
      "step": 23520
    },
    {
      "epoch": 2.2430419921875e-05,
      "model_forward_time": 2.890305995941162,
      "step": 23520
    },
    {
      "dynamics_loss_avg": 0.04289424903690815,
      "epoch": 2.2430419921875e-05,
      "step": 23520
    },
    {
      "action_loss_avg": 0.002667578449472785,
      "epoch": 2.2430419921875e-05,
      "step": 23520
    },
    {
      "epoch": 2.2430419921875e-05,
      "step": 23520,
      "training_step_time": 8.731029272079468
    },
    {
      "epoch": 2.2431373596191407e-05,
      "model_forward_time": 1.7656965255737305,
      "step": 23521
    },
    {
      "epoch": 2.2431373596191407e-05,
      "step": 23521,
      "training_step_time": 7.722659587860107
    },
    {
      "epoch": 2.243232727050781e-05,
      "model_forward_time": 1.7769660949707031,
      "step": 23522
    },
    {
      "epoch": 2.243232727050781e-05,
      "step": 23522,
      "training_step_time": 7.719813108444214
    },
    {
      "epoch": 2.243328094482422e-05,
      "model_forward_time": 1.7613255977630615,
      "step": 23523
    },
    {
      "epoch": 2.243328094482422e-05,
      "step": 23523,
      "training_step_time": 8.256502628326416
    },
    {
      "epoch": 2.2434234619140626e-05,
      "model_forward_time": 1.7596361637115479,
      "step": 23524
    },
    {
      "epoch": 2.2434234619140626e-05,
      "step": 23524,
      "training_step_time": 8.327861785888672
    },
    {
      "epoch": 2.243518829345703e-05,
      "model_forward_time": 1.7653100490570068,
      "step": 23525
    },
    {
      "epoch": 2.243518829345703e-05,
      "step": 23525,
      "training_step_time": 7.262036323547363
    },
    {
      "epoch": 2.2436141967773437e-05,
      "model_forward_time": 1.7529006004333496,
      "step": 23526
    },
    {
      "epoch": 2.2436141967773437e-05,
      "step": 23526,
      "training_step_time": 7.725752592086792
    },
    {
      "epoch": 2.2437095642089845e-05,
      "model_forward_time": 1.879352331161499,
      "step": 23527
    },
    {
      "epoch": 2.2437095642089845e-05,
      "step": 23527,
      "training_step_time": 7.737179517745972
    },
    {
      "epoch": 2.243804931640625e-05,
      "model_forward_time": 1.769371509552002,
      "step": 23528
    },
    {
      "epoch": 2.243804931640625e-05,
      "step": 23528,
      "training_step_time": 8.809303522109985
    },
    {
      "epoch": 2.2439002990722656e-05,
      "model_forward_time": 1.7575733661651611,
      "step": 23529
    },
    {
      "epoch": 2.2439002990722656e-05,
      "step": 23529,
      "training_step_time": 9.844921112060547
    },
    {
      "epoch": 2.2439956665039064e-05,
      "grad_norm": 0.279424250125885,
      "learning_rate": 9.090368249805874e-06,
      "loss": 0.0656,
      "step": 23530
    },
    {
      "epoch": 2.2439956665039064e-05,
      "model_forward_time": 1.7706043720245361,
      "step": 23530
    },
    {
      "dynamics_loss_avg": 0.07065866962075233,
      "epoch": 2.2439956665039064e-05,
      "step": 23530
    },
    {
      "action_loss_avg": 0.002874066133517772,
      "epoch": 2.2439956665039064e-05,
      "step": 23530
    },
    {
      "epoch": 2.2439956665039064e-05,
      "step": 23530,
      "training_step_time": 7.746674299240112
    },
    {
      "epoch": 2.2440910339355468e-05,
      "model_forward_time": 1.757699966430664,
      "step": 23531
    },
    {
      "epoch": 2.2440910339355468e-05,
      "step": 23531,
      "training_step_time": 8.526506662368774
    },
    {
      "epoch": 2.2441864013671875e-05,
      "model_forward_time": 1.780644178390503,
      "step": 23532
    },
    {
      "epoch": 2.2441864013671875e-05,
      "step": 23532,
      "training_step_time": 7.739023208618164
    },
    {
      "epoch": 2.2442817687988282e-05,
      "model_forward_time": 1.3981683254241943,
      "step": 23533
    },
    {
      "epoch": 2.2442817687988282e-05,
      "step": 23533,
      "training_step_time": 7.648119926452637
    },
    {
      "epoch": 2.2443771362304686e-05,
      "model_forward_time": 1.7573843002319336,
      "step": 23534
    },
    {
      "epoch": 2.2443771362304686e-05,
      "step": 23534,
      "training_step_time": 7.198785066604614
    },
    {
      "epoch": 2.2444725036621094e-05,
      "model_forward_time": 1.0789763927459717,
      "step": 23535
    },
    {
      "epoch": 2.2444725036621094e-05,
      "step": 23535,
      "training_step_time": 7.720627784729004
    },
    {
      "epoch": 2.24456787109375e-05,
      "model_forward_time": 1.8902034759521484,
      "step": 23536
    },
    {
      "epoch": 2.24456787109375e-05,
      "step": 23536,
      "training_step_time": 7.694073677062988
    },
    {
      "epoch": 2.2446632385253905e-05,
      "model_forward_time": 1.7582414150238037,
      "step": 23537
    },
    {
      "epoch": 2.2446632385253905e-05,
      "step": 23537,
      "training_step_time": 8.303663730621338
    },
    {
      "epoch": 2.2447586059570313e-05,
      "model_forward_time": 1.7681150436401367,
      "step": 23538
    },
    {
      "epoch": 2.2447586059570313e-05,
      "step": 23538,
      "training_step_time": 8.399798393249512
    },
    {
      "epoch": 2.244853973388672e-05,
      "model_forward_time": 2.910712242126465,
      "step": 23539
    },
    {
      "epoch": 2.244853973388672e-05,
      "step": 23539,
      "training_step_time": 8.563392162322998
    },
    {
      "epoch": 2.2449493408203124e-05,
      "grad_norm": 0.20817802846431732,
      "learning_rate": 9.089417093400754e-06,
      "loss": 0.0665,
      "step": 23540
    },
    {
      "epoch": 2.2449493408203124e-05,
      "model_forward_time": 1.768235206604004,
      "step": 23540
    },
    {
      "dynamics_loss_avg": 0.06564713940024376,
      "epoch": 2.2449493408203124e-05,
      "step": 23540
    },
    {
      "action_loss_avg": 0.0032710890751332044,
      "epoch": 2.2449493408203124e-05,
      "step": 23540
    },
    {
      "epoch": 2.2449493408203124e-05,
      "step": 23540,
      "training_step_time": 7.672998428344727
    },
    {
      "epoch": 2.245044708251953e-05,
      "model_forward_time": 1.7632577419281006,
      "step": 23541
    },
    {
      "epoch": 2.245044708251953e-05,
      "step": 23541,
      "training_step_time": 7.828877210617065
    },
    {
      "epoch": 2.245140075683594e-05,
      "model_forward_time": 1.7532720565795898,
      "step": 23542
    },
    {
      "epoch": 2.245140075683594e-05,
      "step": 23542,
      "training_step_time": 7.639072895050049
    },
    {
      "epoch": 2.2452354431152343e-05,
      "model_forward_time": 1.7695176601409912,
      "step": 23543
    },
    {
      "epoch": 2.2452354431152343e-05,
      "step": 23543,
      "training_step_time": 7.577287435531616
    },
    {
      "epoch": 2.245330810546875e-05,
      "model_forward_time": 1.7422869205474854,
      "step": 23544
    },
    {
      "epoch": 2.245330810546875e-05,
      "step": 23544,
      "training_step_time": 7.619768142700195
    },
    {
      "epoch": 2.2454261779785158e-05,
      "model_forward_time": 1.8805007934570312,
      "step": 23545
    },
    {
      "epoch": 2.2454261779785158e-05,
      "step": 23545,
      "training_step_time": 9.442260265350342
    },
    {
      "epoch": 2.2455215454101562e-05,
      "model_forward_time": 1.7505311965942383,
      "step": 23546
    },
    {
      "epoch": 2.2455215454101562e-05,
      "step": 23546,
      "training_step_time": 7.656413316726685
    },
    {
      "epoch": 2.245616912841797e-05,
      "model_forward_time": 1.7680926322937012,
      "step": 23547
    },
    {
      "epoch": 2.245616912841797e-05,
      "step": 23547,
      "training_step_time": 7.669044494628906
    },
    {
      "epoch": 2.2457122802734373e-05,
      "model_forward_time": 2.932882308959961,
      "step": 23548
    },
    {
      "epoch": 2.2457122802734373e-05,
      "step": 23548,
      "training_step_time": 8.305782318115234
    },
    {
      "epoch": 2.245807647705078e-05,
      "model_forward_time": 1.7581748962402344,
      "step": 23549
    },
    {
      "epoch": 2.245807647705078e-05,
      "step": 23549,
      "training_step_time": 7.754262208938599
    },
    {
      "epoch": 2.2459030151367188e-05,
      "grad_norm": 0.261305034160614,
      "learning_rate": 9.088465489783131e-06,
      "loss": 0.066,
      "step": 23550
    },
    {
      "epoch": 2.2459030151367188e-05,
      "model_forward_time": 1.762101173400879,
      "step": 23550
    },
    {
      "dynamics_loss_avg": 0.06751274652779102,
      "epoch": 2.2459030151367188e-05,
      "step": 23550
    },
    {
      "action_loss_avg": 0.0047255891375243666,
      "epoch": 2.2459030151367188e-05,
      "step": 23550
    },
    {
      "epoch": 2.2459030151367188e-05,
      "step": 23550,
      "training_step_time": 7.6841137409210205
    },
    {
      "epoch": 2.2459983825683592e-05,
      "model_forward_time": 1.7532947063446045,
      "step": 23551
    },
    {
      "epoch": 2.2459983825683592e-05,
      "step": 23551,
      "training_step_time": 8.59280776977539
    },
    {
      "epoch": 2.24609375e-05,
      "model_forward_time": 1.760486364364624,
      "step": 23552
    },
    {
      "epoch": 2.24609375e-05,
      "step": 23552,
      "training_step_time": 7.546108722686768
    },
    {
      "epoch": 2.2461891174316407e-05,
      "model_forward_time": 1.7535066604614258,
      "step": 23553
    },
    {
      "epoch": 2.2461891174316407e-05,
      "step": 23553,
      "training_step_time": 7.706294298171997
    },
    {
      "epoch": 2.246284484863281e-05,
      "model_forward_time": 1.881030797958374,
      "step": 23554
    },
    {
      "epoch": 2.246284484863281e-05,
      "step": 23554,
      "training_step_time": 7.666018009185791
    },
    {
      "epoch": 2.246379852294922e-05,
      "model_forward_time": 1.7503671646118164,
      "step": 23555
    },
    {
      "epoch": 2.246379852294922e-05,
      "step": 23555,
      "training_step_time": 7.616108655929565
    },
    {
      "epoch": 2.2464752197265626e-05,
      "model_forward_time": 1.7556507587432861,
      "step": 23556
    },
    {
      "epoch": 2.2464752197265626e-05,
      "step": 23556,
      "training_step_time": 8.389400959014893
    },
    {
      "epoch": 2.246570587158203e-05,
      "model_forward_time": 1.7620866298675537,
      "step": 23557
    },
    {
      "epoch": 2.246570587158203e-05,
      "step": 23557,
      "training_step_time": 9.348119020462036
    },
    {
      "epoch": 2.2466659545898437e-05,
      "model_forward_time": 1.7512357234954834,
      "step": 23558
    },
    {
      "epoch": 2.2466659545898437e-05,
      "step": 23558,
      "training_step_time": 8.717188835144043
    },
    {
      "epoch": 2.2467613220214845e-05,
      "model_forward_time": 1.7510969638824463,
      "step": 23559
    },
    {
      "epoch": 2.2467613220214845e-05,
      "step": 23559,
      "training_step_time": 8.498355150222778
    },
    {
      "epoch": 2.246856689453125e-05,
      "grad_norm": 0.22057829797267914,
      "learning_rate": 9.087513439057068e-06,
      "loss": 0.0652,
      "step": 23560
    },
    {
      "epoch": 2.246856689453125e-05,
      "model_forward_time": 1.7548606395721436,
      "step": 23560
    },
    {
      "dynamics_loss_avg": 0.08128064861521125,
      "epoch": 2.246856689453125e-05,
      "step": 23560
    },
    {
      "action_loss_avg": 0.003982522356091067,
      "epoch": 2.246856689453125e-05,
      "step": 23560
    },
    {
      "epoch": 2.246856689453125e-05,
      "step": 23560,
      "training_step_time": 8.093408107757568
    },
    {
      "epoch": 2.2469520568847656e-05,
      "model_forward_time": 1.7589225769042969,
      "step": 23561
    },
    {
      "epoch": 2.2469520568847656e-05,
      "step": 23561,
      "training_step_time": 8.643337965011597
    },
    {
      "epoch": 2.2470474243164064e-05,
      "model_forward_time": 1.753824234008789,
      "step": 23562
    },
    {
      "epoch": 2.2470474243164064e-05,
      "step": 23562,
      "training_step_time": 7.535770654678345
    },
    {
      "epoch": 2.2471427917480468e-05,
      "model_forward_time": 1.8848707675933838,
      "step": 23563
    },
    {
      "epoch": 2.2471427917480468e-05,
      "step": 23563,
      "training_step_time": 8.228206157684326
    },
    {
      "epoch": 2.2472381591796875e-05,
      "model_forward_time": 1.7581372261047363,
      "step": 23564
    },
    {
      "epoch": 2.2472381591796875e-05,
      "step": 23564,
      "training_step_time": 8.557155132293701
    },
    {
      "epoch": 2.2473335266113282e-05,
      "model_forward_time": 1.752537727355957,
      "step": 23565
    },
    {
      "epoch": 2.2473335266113282e-05,
      "step": 23565,
      "training_step_time": 7.729271650314331
    },
    {
      "epoch": 2.2474288940429686e-05,
      "model_forward_time": 1.7679443359375,
      "step": 23566
    },
    {
      "epoch": 2.2474288940429686e-05,
      "step": 23566,
      "training_step_time": 8.588899374008179
    },
    {
      "epoch": 2.2475242614746094e-05,
      "model_forward_time": 2.9032142162323,
      "step": 23567
    },
    {
      "epoch": 2.2475242614746094e-05,
      "step": 23567,
      "training_step_time": 9.274522542953491
    },
    {
      "epoch": 2.24761962890625e-05,
      "model_forward_time": 1.7530276775360107,
      "step": 23568
    },
    {
      "epoch": 2.24761962890625e-05,
      "step": 23568,
      "training_step_time": 8.857657194137573
    },
    {
      "epoch": 2.2477149963378905e-05,
      "model_forward_time": 1.7458913326263428,
      "step": 23569
    },
    {
      "epoch": 2.2477149963378905e-05,
      "step": 23569,
      "training_step_time": 7.694727420806885
    },
    {
      "epoch": 2.2478103637695313e-05,
      "grad_norm": 0.20954695343971252,
      "learning_rate": 9.08656094132668e-06,
      "loss": 0.0658,
      "step": 23570
    },
    {
      "epoch": 2.2478103637695313e-05,
      "model_forward_time": 1.7473077774047852,
      "step": 23570
    },
    {
      "dynamics_loss_avg": 0.07054356895387173,
      "epoch": 2.2478103637695313e-05,
      "step": 23570
    },
    {
      "action_loss_avg": 0.0033604456228204073,
      "epoch": 2.2478103637695313e-05,
      "step": 23570
    },
    {
      "epoch": 2.2478103637695313e-05,
      "step": 23570,
      "training_step_time": 7.729336500167847
    },
    {
      "epoch": 2.247905731201172e-05,
      "model_forward_time": 1.770592212677002,
      "step": 23571
    },
    {
      "epoch": 2.247905731201172e-05,
      "step": 23571,
      "training_step_time": 7.658914566040039
    },
    {
      "epoch": 2.2480010986328124e-05,
      "model_forward_time": 1.878986120223999,
      "step": 23572
    },
    {
      "epoch": 2.2480010986328124e-05,
      "step": 23572,
      "training_step_time": 8.539948225021362
    },
    {
      "epoch": 2.248096466064453e-05,
      "model_forward_time": 1.7524995803833008,
      "step": 23573
    },
    {
      "epoch": 2.248096466064453e-05,
      "step": 23573,
      "training_step_time": 8.260332345962524
    },
    {
      "epoch": 2.248191833496094e-05,
      "model_forward_time": 1.7532453536987305,
      "step": 23574
    },
    {
      "epoch": 2.248191833496094e-05,
      "step": 23574,
      "training_step_time": 10.195799112319946
    },
    {
      "epoch": 2.2482872009277343e-05,
      "model_forward_time": 1.7416820526123047,
      "step": 23575
    },
    {
      "epoch": 2.2482872009277343e-05,
      "step": 23575,
      "training_step_time": 8.6845121383667
    },
    {
      "epoch": 2.248382568359375e-05,
      "model_forward_time": 2.9441075325012207,
      "step": 23576
    },
    {
      "epoch": 2.248382568359375e-05,
      "step": 23576,
      "training_step_time": 7.719342470169067
    },
    {
      "epoch": 2.2484779357910158e-05,
      "model_forward_time": 1.7552237510681152,
      "step": 23577
    },
    {
      "epoch": 2.2484779357910158e-05,
      "step": 23577,
      "training_step_time": 8.364060878753662
    },
    {
      "epoch": 2.2485733032226562e-05,
      "model_forward_time": 1.7485771179199219,
      "step": 23578
    },
    {
      "epoch": 2.2485733032226562e-05,
      "step": 23578,
      "training_step_time": 7.715733289718628
    },
    {
      "epoch": 2.248668670654297e-05,
      "model_forward_time": 1.7449922561645508,
      "step": 23579
    },
    {
      "epoch": 2.248668670654297e-05,
      "step": 23579,
      "training_step_time": 7.644906997680664
    },
    {
      "epoch": 2.2487640380859373e-05,
      "grad_norm": 0.18531285226345062,
      "learning_rate": 9.085607996696135e-06,
      "loss": 0.0648,
      "step": 23580
    },
    {
      "epoch": 2.2487640380859373e-05,
      "model_forward_time": 1.75960373878479,
      "step": 23580
    },
    {
      "dynamics_loss_avg": 0.05531027521938085,
      "epoch": 2.2487640380859373e-05,
      "step": 23580
    },
    {
      "action_loss_avg": 0.0028996474255109208,
      "epoch": 2.2487640380859373e-05,
      "step": 23580
    },
    {
      "epoch": 2.2487640380859373e-05,
      "step": 23580,
      "training_step_time": 8.348553657531738
    },
    {
      "epoch": 2.248859405517578e-05,
      "model_forward_time": 1.8817918300628662,
      "step": 23581
    },
    {
      "epoch": 2.248859405517578e-05,
      "step": 23581,
      "training_step_time": 7.719093084335327
    },
    {
      "epoch": 2.2489547729492188e-05,
      "model_forward_time": 1.747666835784912,
      "step": 23582
    },
    {
      "epoch": 2.2489547729492188e-05,
      "step": 23582,
      "training_step_time": 8.431645393371582
    },
    {
      "epoch": 2.2490501403808592e-05,
      "model_forward_time": 1.7540044784545898,
      "step": 23583
    },
    {
      "epoch": 2.2490501403808592e-05,
      "step": 23583,
      "training_step_time": 7.673064708709717
    },
    {
      "epoch": 2.2491455078125e-05,
      "model_forward_time": 1.7539398670196533,
      "step": 23584
    },
    {
      "epoch": 2.2491455078125e-05,
      "step": 23584,
      "training_step_time": 8.97388219833374
    },
    {
      "epoch": 2.2492408752441407e-05,
      "model_forward_time": 2.9077188968658447,
      "step": 23585
    },
    {
      "epoch": 2.2492408752441407e-05,
      "step": 23585,
      "training_step_time": 7.67350172996521
    },
    {
      "epoch": 2.249336242675781e-05,
      "model_forward_time": 1.74674391746521,
      "step": 23586
    },
    {
      "epoch": 2.249336242675781e-05,
      "step": 23586,
      "training_step_time": 8.645628452301025
    },
    {
      "epoch": 2.249431610107422e-05,
      "model_forward_time": 1.7644522190093994,
      "step": 23587
    },
    {
      "epoch": 2.249431610107422e-05,
      "step": 23587,
      "training_step_time": 8.496639966964722
    },
    {
      "epoch": 2.2495269775390626e-05,
      "model_forward_time": 1.7497787475585938,
      "step": 23588
    },
    {
      "epoch": 2.2495269775390626e-05,
      "step": 23588,
      "training_step_time": 7.60474705696106
    },
    {
      "epoch": 2.249622344970703e-05,
      "model_forward_time": 1.75766921043396,
      "step": 23589
    },
    {
      "epoch": 2.249622344970703e-05,
      "step": 23589,
      "training_step_time": 7.5035765171051025
    },
    {
      "epoch": 2.2497177124023437e-05,
      "grad_norm": 0.169878289103508,
      "learning_rate": 9.08465460526964e-06,
      "loss": 0.064,
      "step": 23590
    },
    {
      "epoch": 2.2497177124023437e-05,
      "model_forward_time": 1.8586680889129639,
      "step": 23590
    },
    {
      "dynamics_loss_avg": 0.0761667551472783,
      "epoch": 2.2497177124023437e-05,
      "step": 23590
    },
    {
      "action_loss_avg": 0.0030361995915882288,
      "epoch": 2.2497177124023437e-05,
      "step": 23590
    },
    {
      "epoch": 2.2497177124023437e-05,
      "step": 23590,
      "training_step_time": 7.667433500289917
    },
    {
      "epoch": 2.2498130798339845e-05,
      "model_forward_time": 1.7657909393310547,
      "step": 23591
    },
    {
      "epoch": 2.2498130798339845e-05,
      "step": 23591,
      "training_step_time": 7.411540269851685
    },
    {
      "epoch": 2.249908447265625e-05,
      "model_forward_time": 1.7521488666534424,
      "step": 23592
    },
    {
      "epoch": 2.249908447265625e-05,
      "step": 23592,
      "training_step_time": 8.870728015899658
    },
    {
      "epoch": 2.2500038146972656e-05,
      "model_forward_time": 1.7541882991790771,
      "step": 23593
    },
    {
      "epoch": 2.2500038146972656e-05,
      "step": 23593,
      "training_step_time": 8.616581678390503
    },
    {
      "epoch": 2.2500991821289064e-05,
      "model_forward_time": 2.8963730335235596,
      "step": 23594
    },
    {
      "epoch": 2.2500991821289064e-05,
      "step": 23594,
      "training_step_time": 8.608325242996216
    },
    {
      "epoch": 2.2501945495605468e-05,
      "model_forward_time": 1.7568233013153076,
      "step": 23595
    },
    {
      "epoch": 2.2501945495605468e-05,
      "step": 23595,
      "training_step_time": 7.655825138092041
    },
    {
      "epoch": 2.2502899169921875e-05,
      "model_forward_time": 1.7624590396881104,
      "step": 23596
    },
    {
      "epoch": 2.2502899169921875e-05,
      "step": 23596,
      "training_step_time": 7.760206699371338
    },
    {
      "epoch": 2.2503852844238282e-05,
      "model_forward_time": 1.7508678436279297,
      "step": 23597
    },
    {
      "epoch": 2.2503852844238282e-05,
      "step": 23597,
      "training_step_time": 7.799785137176514
    },
    {
      "epoch": 2.2504806518554687e-05,
      "model_forward_time": 1.7580931186676025,
      "step": 23598
    },
    {
      "epoch": 2.2504806518554687e-05,
      "step": 23598,
      "training_step_time": 7.6908838748931885
    },
    {
      "epoch": 2.2505760192871094e-05,
      "model_forward_time": 1.8116629123687744,
      "step": 23599
    },
    {
      "epoch": 2.2505760192871094e-05,
      "step": 23599,
      "training_step_time": 7.612705707550049
    },
    {
      "epoch": 2.25067138671875e-05,
      "grad_norm": 0.14118120074272156,
      "learning_rate": 9.083700767151458e-06,
      "loss": 0.0647,
      "step": 23600
    },
    {
      "epoch": 2.25067138671875e-05,
      "model_forward_time": 1.760688066482544,
      "step": 23600
    },
    {
      "dynamics_loss_avg": 0.0670248493552208,
      "epoch": 2.25067138671875e-05,
      "step": 23600
    },
    {
      "action_loss_avg": 0.003567692101933062,
      "epoch": 2.25067138671875e-05,
      "step": 23600
    },
    {
      "epoch": 2.25067138671875e-05,
      "step": 23600,
      "training_step_time": 7.4257118701934814
    },
    {
      "epoch": 2.2507667541503905e-05,
      "model_forward_time": 1.7676506042480469,
      "step": 23601
    },
    {
      "epoch": 2.2507667541503905e-05,
      "step": 23601,
      "training_step_time": 8.325265169143677
    },
    {
      "epoch": 2.2508621215820313e-05,
      "model_forward_time": 1.7601892948150635,
      "step": 23602
    },
    {
      "epoch": 2.2508621215820313e-05,
      "step": 23602,
      "training_step_time": 7.598107814788818
    },
    {
      "epoch": 2.250957489013672e-05,
      "model_forward_time": 2.9261953830718994,
      "step": 23603
    },
    {
      "epoch": 2.250957489013672e-05,
      "step": 23603,
      "training_step_time": 7.767444610595703
    },
    {
      "epoch": 2.2510528564453124e-05,
      "model_forward_time": 1.766237497329712,
      "step": 23604
    },
    {
      "epoch": 2.2510528564453124e-05,
      "step": 23604,
      "training_step_time": 8.50135350227356
    },
    {
      "epoch": 2.251148223876953e-05,
      "model_forward_time": 1.7462778091430664,
      "step": 23605
    },
    {
      "epoch": 2.251148223876953e-05,
      "step": 23605,
      "training_step_time": 7.77332329750061
    },
    {
      "epoch": 2.251243591308594e-05,
      "model_forward_time": 1.7512145042419434,
      "step": 23606
    },
    {
      "epoch": 2.251243591308594e-05,
      "step": 23606,
      "training_step_time": 7.663861513137817
    },
    {
      "epoch": 2.2513389587402343e-05,
      "model_forward_time": 1.7542073726654053,
      "step": 23607
    },
    {
      "epoch": 2.2513389587402343e-05,
      "step": 23607,
      "training_step_time": 7.651485443115234
    },
    {
      "epoch": 2.251434326171875e-05,
      "model_forward_time": 1.7827916145324707,
      "step": 23608
    },
    {
      "epoch": 2.251434326171875e-05,
      "step": 23608,
      "training_step_time": 7.60015082359314
    },
    {
      "epoch": 2.2515296936035158e-05,
      "model_forward_time": 1.7574403285980225,
      "step": 23609
    },
    {
      "epoch": 2.2515296936035158e-05,
      "step": 23609,
      "training_step_time": 7.656443119049072
    },
    {
      "epoch": 2.2516250610351562e-05,
      "grad_norm": 0.17853008210659027,
      "learning_rate": 9.082746482445899e-06,
      "loss": 0.0629,
      "step": 23610
    },
    {
      "epoch": 2.2516250610351562e-05,
      "model_forward_time": 1.7559947967529297,
      "step": 23610
    },
    {
      "dynamics_loss_avg": 0.05947162713855505,
      "epoch": 2.2516250610351562e-05,
      "step": 23610
    },
    {
      "action_loss_avg": 0.002494979614857584,
      "epoch": 2.2516250610351562e-05,
      "step": 23610
    },
    {
      "epoch": 2.2516250610351562e-05,
      "step": 23610,
      "training_step_time": 7.621202707290649
    },
    {
      "epoch": 2.251720428466797e-05,
      "model_forward_time": 1.7453839778900146,
      "step": 23611
    },
    {
      "epoch": 2.251720428466797e-05,
      "step": 23611,
      "training_step_time": 8.303894758224487
    },
    {
      "epoch": 2.2518157958984373e-05,
      "model_forward_time": 1.7484238147735596,
      "step": 23612
    },
    {
      "epoch": 2.2518157958984373e-05,
      "step": 23612,
      "training_step_time": 8.677980661392212
    },
    {
      "epoch": 2.251911163330078e-05,
      "model_forward_time": 1.7521748542785645,
      "step": 23613
    },
    {
      "epoch": 2.251911163330078e-05,
      "step": 23613,
      "training_step_time": 7.648998975753784
    },
    {
      "epoch": 2.2520065307617188e-05,
      "model_forward_time": 1.7607839107513428,
      "step": 23614
    },
    {
      "epoch": 2.2520065307617188e-05,
      "step": 23614,
      "training_step_time": 8.560352325439453
    },
    {
      "epoch": 2.2521018981933592e-05,
      "model_forward_time": 1.7634215354919434,
      "step": 23615
    },
    {
      "epoch": 2.2521018981933592e-05,
      "step": 23615,
      "training_step_time": 7.681943655014038
    },
    {
      "epoch": 2.252197265625e-05,
      "model_forward_time": 1.754535436630249,
      "step": 23616
    },
    {
      "epoch": 2.252197265625e-05,
      "step": 23616,
      "training_step_time": 7.596034288406372
    },
    {
      "epoch": 2.2522926330566407e-05,
      "model_forward_time": 1.7511963844299316,
      "step": 23617
    },
    {
      "epoch": 2.2522926330566407e-05,
      "step": 23617,
      "training_step_time": 7.581312656402588
    },
    {
      "epoch": 2.252388000488281e-05,
      "model_forward_time": 1.755784273147583,
      "step": 23618
    },
    {
      "epoch": 2.252388000488281e-05,
      "step": 23618,
      "training_step_time": 7.644644260406494
    },
    {
      "epoch": 2.252483367919922e-05,
      "model_forward_time": 1.754420518875122,
      "step": 23619
    },
    {
      "epoch": 2.252483367919922e-05,
      "step": 23619,
      "training_step_time": 9.363945484161377
    },
    {
      "epoch": 2.2525787353515626e-05,
      "grad_norm": 0.13687388598918915,
      "learning_rate": 9.081791751257325e-06,
      "loss": 0.0629,
      "step": 23620
    },
    {
      "epoch": 2.2525787353515626e-05,
      "model_forward_time": 1.7637054920196533,
      "step": 23620
    },
    {
      "dynamics_loss_avg": 0.05954916626214981,
      "epoch": 2.2525787353515626e-05,
      "step": 23620
    },
    {
      "action_loss_avg": 0.003114854119485244,
      "epoch": 2.2525787353515626e-05,
      "step": 23620
    },
    {
      "epoch": 2.2525787353515626e-05,
      "step": 23620,
      "training_step_time": 8.633739948272705
    },
    {
      "epoch": 2.252674102783203e-05,
      "model_forward_time": 1.758779764175415,
      "step": 23621
    },
    {
      "epoch": 2.252674102783203e-05,
      "step": 23621,
      "training_step_time": 7.614518165588379
    },
    {
      "epoch": 2.2527694702148437e-05,
      "model_forward_time": 2.968881368637085,
      "step": 23622
    },
    {
      "epoch": 2.2527694702148437e-05,
      "step": 23622,
      "training_step_time": 8.636983633041382
    },
    {
      "epoch": 2.2528648376464845e-05,
      "model_forward_time": 1.7583415508270264,
      "step": 23623
    },
    {
      "epoch": 2.2528648376464845e-05,
      "step": 23623,
      "training_step_time": 8.501625299453735
    },
    {
      "epoch": 2.252960205078125e-05,
      "model_forward_time": 1.7561790943145752,
      "step": 23624
    },
    {
      "epoch": 2.252960205078125e-05,
      "step": 23624,
      "training_step_time": 7.691784381866455
    },
    {
      "epoch": 2.2530555725097656e-05,
      "model_forward_time": 1.7552595138549805,
      "step": 23625
    },
    {
      "epoch": 2.2530555725097656e-05,
      "step": 23625,
      "training_step_time": 8.528307676315308
    },
    {
      "epoch": 2.2531509399414064e-05,
      "model_forward_time": 1.7580156326293945,
      "step": 23626
    },
    {
      "epoch": 2.2531509399414064e-05,
      "step": 23626,
      "training_step_time": 7.543006658554077
    },
    {
      "epoch": 2.2532463073730468e-05,
      "model_forward_time": 1.8849818706512451,
      "step": 23627
    },
    {
      "epoch": 2.2532463073730468e-05,
      "step": 23627,
      "training_step_time": 7.6677141189575195
    },
    {
      "epoch": 2.2533416748046875e-05,
      "model_forward_time": 1.750654697418213,
      "step": 23628
    },
    {
      "epoch": 2.2533416748046875e-05,
      "step": 23628,
      "training_step_time": 7.519595384597778
    },
    {
      "epoch": 2.2534370422363283e-05,
      "model_forward_time": 1.7530598640441895,
      "step": 23629
    },
    {
      "epoch": 2.2534370422363283e-05,
      "step": 23629,
      "training_step_time": 9.196265935897827
    },
    {
      "epoch": 2.2535324096679687e-05,
      "grad_norm": 0.1779731661081314,
      "learning_rate": 9.080836573690141e-06,
      "loss": 0.0664,
      "step": 23630
    },
    {
      "epoch": 2.2535324096679687e-05,
      "model_forward_time": 1.773329257965088,
      "step": 23630
    },
    {
      "dynamics_loss_avg": 0.06328814458101988,
      "epoch": 2.2535324096679687e-05,
      "step": 23630
    },
    {
      "action_loss_avg": 0.002580996317556128,
      "epoch": 2.2535324096679687e-05,
      "step": 23630
    },
    {
      "epoch": 2.2535324096679687e-05,
      "step": 23630,
      "training_step_time": 9.25243353843689
    },
    {
      "epoch": 2.2536277770996094e-05,
      "model_forward_time": 2.951608180999756,
      "step": 23631
    },
    {
      "epoch": 2.2536277770996094e-05,
      "step": 23631,
      "training_step_time": 8.421226739883423
    },
    {
      "epoch": 2.25372314453125e-05,
      "model_forward_time": 1.761176347732544,
      "step": 23632
    },
    {
      "epoch": 2.25372314453125e-05,
      "step": 23632,
      "training_step_time": 7.692241668701172
    },
    {
      "epoch": 2.2538185119628905e-05,
      "model_forward_time": 1.7539753913879395,
      "step": 23633
    },
    {
      "epoch": 2.2538185119628905e-05,
      "step": 23633,
      "training_step_time": 7.665099382400513
    },
    {
      "epoch": 2.2539138793945313e-05,
      "model_forward_time": 1.7524237632751465,
      "step": 23634
    },
    {
      "epoch": 2.2539138793945313e-05,
      "step": 23634,
      "training_step_time": 8.463430643081665
    },
    {
      "epoch": 2.254009246826172e-05,
      "model_forward_time": 1.756803035736084,
      "step": 23635
    },
    {
      "epoch": 2.254009246826172e-05,
      "step": 23635,
      "training_step_time": 7.556626796722412
    },
    {
      "epoch": 2.2541046142578124e-05,
      "model_forward_time": 1.8487396240234375,
      "step": 23636
    },
    {
      "epoch": 2.2541046142578124e-05,
      "step": 23636,
      "training_step_time": 7.615248203277588
    },
    {
      "epoch": 2.254199981689453e-05,
      "model_forward_time": 1.7509775161743164,
      "step": 23637
    },
    {
      "epoch": 2.254199981689453e-05,
      "step": 23637,
      "training_step_time": 8.442457675933838
    },
    {
      "epoch": 2.254295349121094e-05,
      "model_forward_time": 1.7744171619415283,
      "step": 23638
    },
    {
      "epoch": 2.254295349121094e-05,
      "step": 23638,
      "training_step_time": 8.294683933258057
    },
    {
      "epoch": 2.2543907165527343e-05,
      "model_forward_time": 1.7535741329193115,
      "step": 23639
    },
    {
      "epoch": 2.2543907165527343e-05,
      "step": 23639,
      "training_step_time": 7.422396659851074
    },
    {
      "epoch": 2.254486083984375e-05,
      "grad_norm": 0.15305736660957336,
      "learning_rate": 9.079880949848804e-06,
      "loss": 0.0631,
      "step": 23640
    },
    {
      "epoch": 2.254486083984375e-05,
      "model_forward_time": 2.919576644897461,
      "step": 23640
    },
    {
      "dynamics_loss_avg": 0.0658013179898262,
      "epoch": 2.254486083984375e-05,
      "step": 23640
    },
    {
      "action_loss_avg": 0.002711760735837743,
      "epoch": 2.254486083984375e-05,
      "step": 23640
    },
    {
      "epoch": 2.254486083984375e-05,
      "step": 23640,
      "training_step_time": 8.742194414138794
    },
    {
      "epoch": 2.2545814514160155e-05,
      "model_forward_time": 1.758984088897705,
      "step": 23641
    },
    {
      "epoch": 2.2545814514160155e-05,
      "step": 23641,
      "training_step_time": 7.635451078414917
    },
    {
      "epoch": 2.2546768188476562e-05,
      "model_forward_time": 1.7465968132019043,
      "step": 23642
    },
    {
      "epoch": 2.2546768188476562e-05,
      "step": 23642,
      "training_step_time": 8.599765539169312
    },
    {
      "epoch": 2.254772186279297e-05,
      "model_forward_time": 1.7557218074798584,
      "step": 23643
    },
    {
      "epoch": 2.254772186279297e-05,
      "step": 23643,
      "training_step_time": 7.333903789520264
    },
    {
      "epoch": 2.2548675537109373e-05,
      "model_forward_time": 1.7436254024505615,
      "step": 23644
    },
    {
      "epoch": 2.2548675537109373e-05,
      "step": 23644,
      "training_step_time": 7.650008678436279
    },
    {
      "epoch": 2.254962921142578e-05,
      "model_forward_time": 1.8017175197601318,
      "step": 23645
    },
    {
      "epoch": 2.254962921142578e-05,
      "step": 23645,
      "training_step_time": 7.634459972381592
    },
    {
      "epoch": 2.2550582885742188e-05,
      "model_forward_time": 1.7532105445861816,
      "step": 23646
    },
    {
      "epoch": 2.2550582885742188e-05,
      "step": 23646,
      "training_step_time": 8.100293636322021
    },
    {
      "epoch": 2.2551536560058592e-05,
      "model_forward_time": 1.7515101432800293,
      "step": 23647
    },
    {
      "epoch": 2.2551536560058592e-05,
      "step": 23647,
      "training_step_time": 9.16141390800476
    },
    {
      "epoch": 2.2552490234375e-05,
      "model_forward_time": 1.763031005859375,
      "step": 23648
    },
    {
      "epoch": 2.2552490234375e-05,
      "step": 23648,
      "training_step_time": 7.598093748092651
    },
    {
      "epoch": 2.2553443908691407e-05,
      "model_forward_time": 2.882838249206543,
      "step": 23649
    },
    {
      "epoch": 2.2553443908691407e-05,
      "step": 23649,
      "training_step_time": 8.61365795135498
    },
    {
      "epoch": 2.255439758300781e-05,
      "grad_norm": 0.19982856512069702,
      "learning_rate": 9.078924879837823e-06,
      "loss": 0.0634,
      "step": 23650
    },
    {
      "epoch": 2.255439758300781e-05,
      "model_forward_time": 1.398545503616333,
      "step": 23650
    },
    {
      "dynamics_loss_avg": 0.058098511397838594,
      "epoch": 2.255439758300781e-05,
      "step": 23650
    },
    {
      "action_loss_avg": 0.0025743258418515326,
      "epoch": 2.255439758300781e-05,
      "step": 23650
    },
    {
      "epoch": 2.255439758300781e-05,
      "step": 23650,
      "training_step_time": 9.131847381591797
    },
    {
      "epoch": 2.255535125732422e-05,
      "model_forward_time": 1.7543470859527588,
      "step": 23651
    },
    {
      "epoch": 2.255535125732422e-05,
      "step": 23651,
      "training_step_time": 8.26729702949524
    },
    {
      "epoch": 2.2556304931640626e-05,
      "model_forward_time": 1.7596187591552734,
      "step": 23652
    },
    {
      "epoch": 2.2556304931640626e-05,
      "step": 23652,
      "training_step_time": 7.706578254699707
    },
    {
      "epoch": 2.255725860595703e-05,
      "model_forward_time": 1.7540736198425293,
      "step": 23653
    },
    {
      "epoch": 2.255725860595703e-05,
      "step": 23653,
      "training_step_time": 7.6211912631988525
    },
    {
      "epoch": 2.2558212280273437e-05,
      "model_forward_time": 1.7977900505065918,
      "step": 23654
    },
    {
      "epoch": 2.2558212280273437e-05,
      "step": 23654,
      "training_step_time": 7.575881719589233
    },
    {
      "epoch": 2.2559165954589845e-05,
      "model_forward_time": 1.7500226497650146,
      "step": 23655
    },
    {
      "epoch": 2.2559165954589845e-05,
      "step": 23655,
      "training_step_time": 8.57705569267273
    },
    {
      "epoch": 2.256011962890625e-05,
      "model_forward_time": 1.7468109130859375,
      "step": 23656
    },
    {
      "epoch": 2.256011962890625e-05,
      "step": 23656,
      "training_step_time": 8.458050012588501
    },
    {
      "epoch": 2.2561073303222656e-05,
      "model_forward_time": 1.7533082962036133,
      "step": 23657
    },
    {
      "epoch": 2.2561073303222656e-05,
      "step": 23657,
      "training_step_time": 7.636045217514038
    },
    {
      "epoch": 2.2562026977539064e-05,
      "model_forward_time": 1.7538132667541504,
      "step": 23658
    },
    {
      "epoch": 2.2562026977539064e-05,
      "step": 23658,
      "training_step_time": 8.76137638092041
    },
    {
      "epoch": 2.2562980651855468e-05,
      "model_forward_time": 2.2764174938201904,
      "step": 23659
    },
    {
      "epoch": 2.2562980651855468e-05,
      "step": 23659,
      "training_step_time": 7.6452765464782715
    },
    {
      "epoch": 2.2563934326171875e-05,
      "grad_norm": 0.13624288141727448,
      "learning_rate": 9.077968363761746e-06,
      "loss": 0.0667,
      "step": 23660
    },
    {
      "epoch": 2.2563934326171875e-05,
      "model_forward_time": 1.759709358215332,
      "step": 23660
    },
    {
      "dynamics_loss_avg": 0.08247871063649655,
      "epoch": 2.2563934326171875e-05,
      "step": 23660
    },
    {
      "action_loss_avg": 0.005352962436154485,
      "epoch": 2.2563934326171875e-05,
      "step": 23660
    },
    {
      "epoch": 2.2563934326171875e-05,
      "step": 23660,
      "training_step_time": 7.773305177688599
    },
    {
      "epoch": 2.2564888000488283e-05,
      "model_forward_time": 1.7521922588348389,
      "step": 23661
    },
    {
      "epoch": 2.2564888000488283e-05,
      "step": 23661,
      "training_step_time": 8.810148000717163
    },
    {
      "epoch": 2.2565841674804687e-05,
      "model_forward_time": 1.7503504753112793,
      "step": 23662
    },
    {
      "epoch": 2.2565841674804687e-05,
      "step": 23662,
      "training_step_time": 8.154340505599976
    },
    {
      "epoch": 2.2566795349121094e-05,
      "model_forward_time": 1.7701663970947266,
      "step": 23663
    },
    {
      "epoch": 2.2566795349121094e-05,
      "step": 23663,
      "training_step_time": 7.646616458892822
    },
    {
      "epoch": 2.25677490234375e-05,
      "model_forward_time": 1.3910307884216309,
      "step": 23664
    },
    {
      "epoch": 2.25677490234375e-05,
      "step": 23664,
      "training_step_time": 8.479253768920898
    },
    {
      "epoch": 2.2568702697753905e-05,
      "model_forward_time": 1.7590372562408447,
      "step": 23665
    },
    {
      "epoch": 2.2568702697753905e-05,
      "step": 23665,
      "training_step_time": 8.387868404388428
    },
    {
      "epoch": 2.2569656372070313e-05,
      "model_forward_time": 1.7360775470733643,
      "step": 23666
    },
    {
      "epoch": 2.2569656372070313e-05,
      "step": 23666,
      "training_step_time": 7.67371678352356
    },
    {
      "epoch": 2.257061004638672e-05,
      "model_forward_time": 1.7500531673431396,
      "step": 23667
    },
    {
      "epoch": 2.257061004638672e-05,
      "step": 23667,
      "training_step_time": 8.299427509307861
    },
    {
      "epoch": 2.2571563720703124e-05,
      "model_forward_time": 2.9360384941101074,
      "step": 23668
    },
    {
      "epoch": 2.2571563720703124e-05,
      "step": 23668,
      "training_step_time": 8.957518100738525
    },
    {
      "epoch": 2.2572517395019532e-05,
      "model_forward_time": 1.751415491104126,
      "step": 23669
    },
    {
      "epoch": 2.2572517395019532e-05,
      "step": 23669,
      "training_step_time": 7.8009867668151855
    },
    {
      "epoch": 2.257347106933594e-05,
      "grad_norm": 0.18815957009792328,
      "learning_rate": 9.077011401725182e-06,
      "loss": 0.0621,
      "step": 23670
    },
    {
      "epoch": 2.257347106933594e-05,
      "model_forward_time": 1.7611522674560547,
      "step": 23670
    },
    {
      "dynamics_loss_avg": 0.06213454883545637,
      "epoch": 2.257347106933594e-05,
      "step": 23670
    },
    {
      "action_loss_avg": 0.003951001120731235,
      "epoch": 2.257347106933594e-05,
      "step": 23670
    },
    {
      "epoch": 2.257347106933594e-05,
      "step": 23670,
      "training_step_time": 7.7116734981536865
    },
    {
      "epoch": 2.2574424743652343e-05,
      "model_forward_time": 1.7475838661193848,
      "step": 23671
    },
    {
      "epoch": 2.2574424743652343e-05,
      "step": 23671,
      "training_step_time": 7.741934776306152
    },
    {
      "epoch": 2.257537841796875e-05,
      "model_forward_time": 1.7590057849884033,
      "step": 23672
    },
    {
      "epoch": 2.257537841796875e-05,
      "step": 23672,
      "training_step_time": 7.680298328399658
    },
    {
      "epoch": 2.2576332092285155e-05,
      "model_forward_time": 1.8843908309936523,
      "step": 23673
    },
    {
      "epoch": 2.2576332092285155e-05,
      "step": 23673,
      "training_step_time": 7.748202800750732
    },
    {
      "epoch": 2.2577285766601562e-05,
      "model_forward_time": 1.7549891471862793,
      "step": 23674
    },
    {
      "epoch": 2.2577285766601562e-05,
      "step": 23674,
      "training_step_time": 7.704941987991333
    },
    {
      "epoch": 2.257823944091797e-05,
      "model_forward_time": 1.751662254333496,
      "step": 23675
    },
    {
      "epoch": 2.257823944091797e-05,
      "step": 23675,
      "training_step_time": 8.68919324874878
    },
    {
      "epoch": 2.2579193115234373e-05,
      "model_forward_time": 1.7503762245178223,
      "step": 23676
    },
    {
      "epoch": 2.2579193115234373e-05,
      "step": 23676,
      "training_step_time": 8.343253135681152
    },
    {
      "epoch": 2.258014678955078e-05,
      "model_forward_time": 2.916254758834839,
      "step": 23677
    },
    {
      "epoch": 2.258014678955078e-05,
      "step": 23677,
      "training_step_time": 8.780521392822266
    },
    {
      "epoch": 2.2581100463867188e-05,
      "model_forward_time": 1.7660155296325684,
      "step": 23678
    },
    {
      "epoch": 2.2581100463867188e-05,
      "step": 23678,
      "training_step_time": 8.727977991104126
    },
    {
      "epoch": 2.2582054138183592e-05,
      "model_forward_time": 1.757310390472412,
      "step": 23679
    },
    {
      "epoch": 2.2582054138183592e-05,
      "step": 23679,
      "training_step_time": 8.446024417877197
    },
    {
      "epoch": 2.25830078125e-05,
      "grad_norm": 0.22636747360229492,
      "learning_rate": 9.07605399383278e-06,
      "loss": 0.0621,
      "step": 23680
    },
    {
      "epoch": 2.25830078125e-05,
      "model_forward_time": 1.767507553100586,
      "step": 23680
    },
    {
      "dynamics_loss_avg": 0.050548067595809695,
      "epoch": 2.25830078125e-05,
      "step": 23680
    },
    {
      "action_loss_avg": 0.0024850519432220606,
      "epoch": 2.25830078125e-05,
      "step": 23680
    },
    {
      "epoch": 2.25830078125e-05,
      "step": 23680,
      "training_step_time": 7.3998703956604
    },
    {
      "epoch": 2.2583961486816407e-05,
      "model_forward_time": 1.7853379249572754,
      "step": 23681
    },
    {
      "epoch": 2.2583961486816407e-05,
      "step": 23681,
      "training_step_time": 8.974545240402222
    },
    {
      "epoch": 2.258491516113281e-05,
      "model_forward_time": 1.8643498420715332,
      "step": 23682
    },
    {
      "epoch": 2.258491516113281e-05,
      "step": 23682,
      "training_step_time": 7.636113405227661
    },
    {
      "epoch": 2.258586883544922e-05,
      "model_forward_time": 1.7607276439666748,
      "step": 23683
    },
    {
      "epoch": 2.258586883544922e-05,
      "step": 23683,
      "training_step_time": 7.731870889663696
    },
    {
      "epoch": 2.2586822509765626e-05,
      "model_forward_time": 1.7492477893829346,
      "step": 23684
    },
    {
      "epoch": 2.2586822509765626e-05,
      "step": 23684,
      "training_step_time": 7.634558200836182
    },
    {
      "epoch": 2.258777618408203e-05,
      "model_forward_time": 1.7495214939117432,
      "step": 23685
    },
    {
      "epoch": 2.258777618408203e-05,
      "step": 23685,
      "training_step_time": 8.366290807723999
    },
    {
      "epoch": 2.2588729858398437e-05,
      "model_forward_time": 2.908071994781494,
      "step": 23686
    },
    {
      "epoch": 2.2588729858398437e-05,
      "step": 23686,
      "training_step_time": 7.748762845993042
    },
    {
      "epoch": 2.2589683532714845e-05,
      "model_forward_time": 1.7487683296203613,
      "step": 23687
    },
    {
      "epoch": 2.2589683532714845e-05,
      "step": 23687,
      "training_step_time": 8.390901327133179
    },
    {
      "epoch": 2.259063720703125e-05,
      "model_forward_time": 1.7540369033813477,
      "step": 23688
    },
    {
      "epoch": 2.259063720703125e-05,
      "step": 23688,
      "training_step_time": 8.955981254577637
    },
    {
      "epoch": 2.2591590881347656e-05,
      "model_forward_time": 1.747497797012329,
      "step": 23689
    },
    {
      "epoch": 2.2591590881347656e-05,
      "step": 23689,
      "training_step_time": 9.326107025146484
    },
    {
      "epoch": 2.2592544555664064e-05,
      "grad_norm": 0.22741036117076874,
      "learning_rate": 9.075096140189243e-06,
      "loss": 0.0598,
      "step": 23690
    },
    {
      "epoch": 2.2592544555664064e-05,
      "model_forward_time": 1.7672662734985352,
      "step": 23690
    },
    {
      "dynamics_loss_avg": 0.056913467682898046,
      "epoch": 2.2592544555664064e-05,
      "step": 23690
    },
    {
      "action_loss_avg": 0.002729387651197612,
      "epoch": 2.2592544555664064e-05,
      "step": 23690
    },
    {
      "epoch": 2.2592544555664064e-05,
      "step": 23690,
      "training_step_time": 6.824368953704834
    },
    {
      "epoch": 2.2593498229980468e-05,
      "model_forward_time": 1.7482264041900635,
      "step": 23691
    },
    {
      "epoch": 2.2593498229980468e-05,
      "step": 23691,
      "training_step_time": 6.7763872146606445
    },
    {
      "epoch": 2.2594451904296875e-05,
      "model_forward_time": 1.7529997825622559,
      "step": 23692
    },
    {
      "epoch": 2.2594451904296875e-05,
      "step": 23692,
      "training_step_time": 6.824777364730835
    },
    {
      "epoch": 2.2595405578613283e-05,
      "model_forward_time": 1.753235101699829,
      "step": 23693
    },
    {
      "epoch": 2.2595405578613283e-05,
      "step": 23693,
      "training_step_time": 6.907890319824219
    },
    {
      "epoch": 2.2596359252929687e-05,
      "model_forward_time": 1.7489511966705322,
      "step": 23694
    },
    {
      "epoch": 2.2596359252929687e-05,
      "step": 23694,
      "training_step_time": 6.910106897354126
    },
    {
      "epoch": 2.2597312927246094e-05,
      "model_forward_time": 1.7686011791229248,
      "step": 23695
    },
    {
      "epoch": 2.2597312927246094e-05,
      "step": 23695,
      "training_step_time": 6.87066125869751
    },
    {
      "epoch": 2.25982666015625e-05,
      "model_forward_time": 1.7562239170074463,
      "step": 23696
    },
    {
      "epoch": 2.25982666015625e-05,
      "step": 23696,
      "training_step_time": 6.851218938827515
    },
    {
      "epoch": 2.2599220275878905e-05,
      "model_forward_time": 1.7516882419586182,
      "step": 23697
    },
    {
      "epoch": 2.2599220275878905e-05,
      "step": 23697,
      "training_step_time": 6.836663484573364
    },
    {
      "epoch": 2.2600173950195313e-05,
      "model_forward_time": 2.889052152633667,
      "step": 23698
    },
    {
      "epoch": 2.2600173950195313e-05,
      "step": 23698,
      "training_step_time": 7.6966712474823
    },
    {
      "epoch": 2.260112762451172e-05,
      "model_forward_time": 1.7603847980499268,
      "step": 23699
    },
    {
      "epoch": 2.260112762451172e-05,
      "step": 23699,
      "training_step_time": 7.828812122344971
    },
    {
      "epoch": 2.2602081298828124e-05,
      "grad_norm": 0.2591152489185333,
      "learning_rate": 9.074137840899319e-06,
      "loss": 0.0613,
      "step": 23700
    },
    {
      "epoch": 2.2602081298828124e-05,
      "model_forward_time": 1.7601690292358398,
      "step": 23700
    },
    {
      "dynamics_loss_avg": 0.07250229120254517,
      "epoch": 2.2602081298828124e-05,
      "step": 23700
    },
    {
      "action_loss_avg": 0.0030141769908368587,
      "epoch": 2.2602081298828124e-05,
      "step": 23700
    },
    {
      "epoch": 2.2602081298828124e-05,
      "step": 23700,
      "training_step_time": 6.826220512390137
    },
    {
      "epoch": 2.2603034973144532e-05,
      "model_forward_time": 1.3999881744384766,
      "step": 23701
    },
    {
      "epoch": 2.2603034973144532e-05,
      "step": 23701,
      "training_step_time": 6.837226629257202
    },
    {
      "epoch": 2.260398864746094e-05,
      "model_forward_time": 1.4015278816223145,
      "step": 23702
    },
    {
      "epoch": 2.260398864746094e-05,
      "step": 23702,
      "training_step_time": 6.811887264251709
    },
    {
      "epoch": 2.2604942321777343e-05,
      "model_forward_time": 1.7464182376861572,
      "step": 23703
    },
    {
      "epoch": 2.2604942321777343e-05,
      "step": 23703,
      "training_step_time": 6.814532995223999
    },
    {
      "epoch": 2.260589599609375e-05,
      "model_forward_time": 1.7516553401947021,
      "step": 23704
    },
    {
      "epoch": 2.260589599609375e-05,
      "step": 23704,
      "training_step_time": 6.849017381668091
    },
    {
      "epoch": 2.2606849670410155e-05,
      "model_forward_time": 1.7622389793395996,
      "step": 23705
    },
    {
      "epoch": 2.2606849670410155e-05,
      "step": 23705,
      "training_step_time": 6.836574077606201
    },
    {
      "epoch": 2.2607803344726562e-05,
      "model_forward_time": 1.7556006908416748,
      "step": 23706
    },
    {
      "epoch": 2.2607803344726562e-05,
      "step": 23706,
      "training_step_time": 6.82787561416626
    },
    {
      "epoch": 2.260875701904297e-05,
      "model_forward_time": 2.8886961936950684,
      "step": 23707
    },
    {
      "epoch": 2.260875701904297e-05,
      "step": 23707,
      "training_step_time": 7.712170600891113
    },
    {
      "epoch": 2.2609710693359373e-05,
      "model_forward_time": 1.7534208297729492,
      "step": 23708
    },
    {
      "epoch": 2.2609710693359373e-05,
      "step": 23708,
      "training_step_time": 7.762660264968872
    },
    {
      "epoch": 2.261066436767578e-05,
      "model_forward_time": 1.7658541202545166,
      "step": 23709
    },
    {
      "epoch": 2.261066436767578e-05,
      "step": 23709,
      "training_step_time": 7.512741327285767
    },
    {
      "epoch": 2.261161804199219e-05,
      "grad_norm": 0.17221182584762573,
      "learning_rate": 9.073179096067804e-06,
      "loss": 0.0661,
      "step": 23710
    },
    {
      "epoch": 2.261161804199219e-05,
      "model_forward_time": 1.7655260562896729,
      "step": 23710
    },
    {
      "dynamics_loss_avg": 0.05854031592607498,
      "epoch": 2.261161804199219e-05,
      "step": 23710
    },
    {
      "action_loss_avg": 0.0022028077743016183,
      "epoch": 2.261161804199219e-05,
      "step": 23710
    },
    {
      "epoch": 2.261161804199219e-05,
      "step": 23710,
      "training_step_time": 6.875404596328735
    },
    {
      "epoch": 2.2612571716308592e-05,
      "model_forward_time": 1.749387502670288,
      "step": 23711
    },
    {
      "epoch": 2.2612571716308592e-05,
      "step": 23711,
      "training_step_time": 6.848543405532837
    },
    {
      "epoch": 2.2613525390625e-05,
      "model_forward_time": 1.7529103755950928,
      "step": 23712
    },
    {
      "epoch": 2.2613525390625e-05,
      "step": 23712,
      "training_step_time": 6.835299730300903
    },
    {
      "epoch": 2.2614479064941407e-05,
      "model_forward_time": 1.7481160163879395,
      "step": 23713
    },
    {
      "epoch": 2.2614479064941407e-05,
      "step": 23713,
      "training_step_time": 6.86214542388916
    },
    {
      "epoch": 2.261543273925781e-05,
      "model_forward_time": 1.7597241401672363,
      "step": 23714
    },
    {
      "epoch": 2.261543273925781e-05,
      "step": 23714,
      "training_step_time": 6.795928001403809
    },
    {
      "epoch": 2.261638641357422e-05,
      "model_forward_time": 1.7471632957458496,
      "step": 23715
    },
    {
      "epoch": 2.261638641357422e-05,
      "step": 23715,
      "training_step_time": 6.834665298461914
    },
    {
      "epoch": 2.2617340087890626e-05,
      "model_forward_time": 2.884413003921509,
      "step": 23716
    },
    {
      "epoch": 2.2617340087890626e-05,
      "step": 23716,
      "training_step_time": 8.58499264717102
    },
    {
      "epoch": 2.261829376220703e-05,
      "model_forward_time": 1.74892258644104,
      "step": 23717
    },
    {
      "epoch": 2.261829376220703e-05,
      "step": 23717,
      "training_step_time": 8.903294086456299
    },
    {
      "epoch": 2.2619247436523437e-05,
      "model_forward_time": 1.747002363204956,
      "step": 23718
    },
    {
      "epoch": 2.2619247436523437e-05,
      "step": 23718,
      "training_step_time": 7.7007973194122314
    },
    {
      "epoch": 2.2620201110839845e-05,
      "model_forward_time": 1.7550036907196045,
      "step": 23719
    },
    {
      "epoch": 2.2620201110839845e-05,
      "step": 23719,
      "training_step_time": 6.8145363330841064
    },
    {
      "epoch": 2.262115478515625e-05,
      "grad_norm": 0.17193560302257538,
      "learning_rate": 9.072219905799549e-06,
      "loss": 0.0678,
      "step": 23720
    },
    {
      "epoch": 2.262115478515625e-05,
      "model_forward_time": 1.7462189197540283,
      "step": 23720
    },
    {
      "dynamics_loss_avg": 0.07467676475644111,
      "epoch": 2.262115478515625e-05,
      "step": 23720
    },
    {
      "action_loss_avg": 0.0026362419943325222,
      "epoch": 2.262115478515625e-05,
      "step": 23720
    },
    {
      "epoch": 2.262115478515625e-05,
      "step": 23720,
      "training_step_time": 6.8568525314331055
    },
    {
      "epoch": 2.2622108459472656e-05,
      "model_forward_time": 1.7532382011413574,
      "step": 23721
    },
    {
      "epoch": 2.2622108459472656e-05,
      "step": 23721,
      "training_step_time": 6.857825040817261
    },
    {
      "epoch": 2.2623062133789064e-05,
      "model_forward_time": 1.3987362384796143,
      "step": 23722
    },
    {
      "epoch": 2.2623062133789064e-05,
      "step": 23722,
      "training_step_time": 6.881223201751709
    },
    {
      "epoch": 2.2624015808105468e-05,
      "model_forward_time": 1.7545621395111084,
      "step": 23723
    },
    {
      "epoch": 2.2624015808105468e-05,
      "step": 23723,
      "training_step_time": 6.8429625034332275
    },
    {
      "epoch": 2.2624969482421875e-05,
      "model_forward_time": 1.7568652629852295,
      "step": 23724
    },
    {
      "epoch": 2.2624969482421875e-05,
      "step": 23724,
      "training_step_time": 6.834606885910034
    },
    {
      "epoch": 2.2625923156738283e-05,
      "model_forward_time": 1.7996835708618164,
      "step": 23725
    },
    {
      "epoch": 2.2625923156738283e-05,
      "step": 23725,
      "training_step_time": 7.622228145599365
    },
    {
      "epoch": 2.2626876831054687e-05,
      "model_forward_time": 2.9009456634521484,
      "step": 23726
    },
    {
      "epoch": 2.2626876831054687e-05,
      "step": 23726,
      "training_step_time": 9.980176448822021
    },
    {
      "epoch": 2.2627830505371094e-05,
      "model_forward_time": 1.7573668956756592,
      "step": 23727
    },
    {
      "epoch": 2.2627830505371094e-05,
      "step": 23727,
      "training_step_time": 7.7541844844818115
    },
    {
      "epoch": 2.26287841796875e-05,
      "model_forward_time": 1.7797937393188477,
      "step": 23728
    },
    {
      "epoch": 2.26287841796875e-05,
      "step": 23728,
      "training_step_time": 7.146135330200195
    },
    {
      "epoch": 2.2629737854003905e-05,
      "model_forward_time": 1.3810334205627441,
      "step": 23729
    },
    {
      "epoch": 2.2629737854003905e-05,
      "step": 23729,
      "training_step_time": 6.770617723464966
    },
    {
      "epoch": 2.2630691528320313e-05,
      "grad_norm": 0.1625833511352539,
      "learning_rate": 9.071260270199447e-06,
      "loss": 0.0685,
      "step": 23730
    },
    {
      "epoch": 2.2630691528320313e-05,
      "model_forward_time": 1.754051685333252,
      "step": 23730
    },
    {
      "dynamics_loss_avg": 0.07898994721472263,
      "epoch": 2.2630691528320313e-05,
      "step": 23730
    },
    {
      "action_loss_avg": 0.0025398771045729517,
      "epoch": 2.2630691528320313e-05,
      "step": 23730
    },
    {
      "epoch": 2.2630691528320313e-05,
      "step": 23730,
      "training_step_time": 6.8430726528167725
    },
    {
      "epoch": 2.263164520263672e-05,
      "model_forward_time": 1.752047061920166,
      "step": 23731
    },
    {
      "epoch": 2.263164520263672e-05,
      "step": 23731,
      "training_step_time": 6.799154996871948
    },
    {
      "epoch": 2.2632598876953124e-05,
      "model_forward_time": 1.7488915920257568,
      "step": 23732
    },
    {
      "epoch": 2.2632598876953124e-05,
      "step": 23732,
      "training_step_time": 6.782134056091309
    },
    {
      "epoch": 2.2633552551269532e-05,
      "model_forward_time": 1.764007806777954,
      "step": 23733
    },
    {
      "epoch": 2.2633552551269532e-05,
      "step": 23733,
      "training_step_time": 6.856168985366821
    },
    {
      "epoch": 2.263450622558594e-05,
      "model_forward_time": 1.776646614074707,
      "step": 23734
    },
    {
      "epoch": 2.263450622558594e-05,
      "step": 23734,
      "training_step_time": 6.824656963348389
    },
    {
      "epoch": 2.2635459899902343e-05,
      "model_forward_time": 2.875802993774414,
      "step": 23735
    },
    {
      "epoch": 2.2635459899902343e-05,
      "step": 23735,
      "training_step_time": 8.772095918655396
    },
    {
      "epoch": 2.263641357421875e-05,
      "model_forward_time": 1.7561571598052979,
      "step": 23736
    },
    {
      "epoch": 2.263641357421875e-05,
      "step": 23736,
      "training_step_time": 7.772418737411499
    },
    {
      "epoch": 2.2637367248535155e-05,
      "model_forward_time": 1.7574403285980225,
      "step": 23737
    },
    {
      "epoch": 2.2637367248535155e-05,
      "step": 23737,
      "training_step_time": 7.173444986343384
    },
    {
      "epoch": 2.2638320922851562e-05,
      "model_forward_time": 1.752655029296875,
      "step": 23738
    },
    {
      "epoch": 2.2638320922851562e-05,
      "step": 23738,
      "training_step_time": 6.852665424346924
    },
    {
      "epoch": 2.263927459716797e-05,
      "model_forward_time": 1.7565088272094727,
      "step": 23739
    },
    {
      "epoch": 2.263927459716797e-05,
      "step": 23739,
      "training_step_time": 6.865382671356201
    },
    {
      "epoch": 2.2640228271484374e-05,
      "grad_norm": 0.122226782143116,
      "learning_rate": 9.070300189372442e-06,
      "loss": 0.0628,
      "step": 23740
    },
    {
      "epoch": 2.2640228271484374e-05,
      "model_forward_time": 1.4042837619781494,
      "step": 23740
    },
    {
      "dynamics_loss_avg": 0.04618058502674103,
      "epoch": 2.2640228271484374e-05,
      "step": 23740
    },
    {
      "action_loss_avg": 0.00136874383315444,
      "epoch": 2.2640228271484374e-05,
      "step": 23740
    },
    {
      "epoch": 2.2640228271484374e-05,
      "step": 23740,
      "training_step_time": 6.866503477096558
    },
    {
      "epoch": 2.264118194580078e-05,
      "model_forward_time": 1.7671020030975342,
      "step": 23741
    },
    {
      "epoch": 2.264118194580078e-05,
      "step": 23741,
      "training_step_time": 6.859184503555298
    },
    {
      "epoch": 2.264213562011719e-05,
      "model_forward_time": 1.7616510391235352,
      "step": 23742
    },
    {
      "epoch": 2.264213562011719e-05,
      "step": 23742,
      "training_step_time": 6.842570543289185
    },
    {
      "epoch": 2.2643089294433592e-05,
      "model_forward_time": 1.7768607139587402,
      "step": 23743
    },
    {
      "epoch": 2.2643089294433592e-05,
      "step": 23743,
      "training_step_time": 6.8499250411987305
    },
    {
      "epoch": 2.264404296875e-05,
      "model_forward_time": 2.8652589321136475,
      "step": 23744
    },
    {
      "epoch": 2.264404296875e-05,
      "step": 23744,
      "training_step_time": 8.743838548660278
    },
    {
      "epoch": 2.2644996643066407e-05,
      "model_forward_time": 1.7659087181091309,
      "step": 23745
    },
    {
      "epoch": 2.2644996643066407e-05,
      "step": 23745,
      "training_step_time": 7.869450569152832
    },
    {
      "epoch": 2.264595031738281e-05,
      "model_forward_time": 1.7578456401824951,
      "step": 23746
    },
    {
      "epoch": 2.264595031738281e-05,
      "step": 23746,
      "training_step_time": 7.558565378189087
    },
    {
      "epoch": 2.264690399169922e-05,
      "model_forward_time": 1.7604684829711914,
      "step": 23747
    },
    {
      "epoch": 2.264690399169922e-05,
      "step": 23747,
      "training_step_time": 6.650582790374756
    },
    {
      "epoch": 2.2647857666015626e-05,
      "model_forward_time": 1.7565457820892334,
      "step": 23748
    },
    {
      "epoch": 2.2647857666015626e-05,
      "step": 23748,
      "training_step_time": 6.764401435852051
    },
    {
      "epoch": 2.264881134033203e-05,
      "model_forward_time": 1.7642040252685547,
      "step": 23749
    },
    {
      "epoch": 2.264881134033203e-05,
      "step": 23749,
      "training_step_time": 6.8029444217681885
    },
    {
      "epoch": 2.2649765014648438e-05,
      "grad_norm": 0.16057957708835602,
      "learning_rate": 9.06933966342353e-06,
      "loss": 0.0637,
      "step": 23750
    },
    {
      "epoch": 2.2649765014648438e-05,
      "model_forward_time": 1.766561508178711,
      "step": 23750
    },
    {
      "dynamics_loss_avg": 0.06843407154083252,
      "epoch": 2.2649765014648438e-05,
      "step": 23750
    },
    {
      "action_loss_avg": 0.003957957914099097,
      "epoch": 2.2649765014648438e-05,
      "step": 23750
    },
    {
      "epoch": 2.2649765014648438e-05,
      "step": 23750,
      "training_step_time": 6.759906768798828
    },
    {
      "epoch": 2.2650718688964845e-05,
      "model_forward_time": 1.7562611103057861,
      "step": 23751
    },
    {
      "epoch": 2.2650718688964845e-05,
      "step": 23751,
      "training_step_time": 6.8375489711761475
    },
    {
      "epoch": 2.265167236328125e-05,
      "model_forward_time": 1.7508151531219482,
      "step": 23752
    },
    {
      "epoch": 2.265167236328125e-05,
      "step": 23752,
      "training_step_time": 6.8186256885528564
    },
    {
      "epoch": 2.2652626037597656e-05,
      "model_forward_time": 2.892611265182495,
      "step": 23753
    },
    {
      "epoch": 2.2652626037597656e-05,
      "step": 23753,
      "training_step_time": 7.707762002944946
    },
    {
      "epoch": 2.2653579711914064e-05,
      "model_forward_time": 1.7616138458251953,
      "step": 23754
    },
    {
      "epoch": 2.2653579711914064e-05,
      "step": 23754,
      "training_step_time": 8.529043197631836
    },
    {
      "epoch": 2.2654533386230468e-05,
      "model_forward_time": 1.768984079360962,
      "step": 23755
    },
    {
      "epoch": 2.2654533386230468e-05,
      "step": 23755,
      "training_step_time": 7.569242000579834
    },
    {
      "epoch": 2.2655487060546875e-05,
      "model_forward_time": 1.7456600666046143,
      "step": 23756
    },
    {
      "epoch": 2.2655487060546875e-05,
      "step": 23756,
      "training_step_time": 6.870656490325928
    },
    {
      "epoch": 2.2656440734863283e-05,
      "model_forward_time": 1.7575647830963135,
      "step": 23757
    },
    {
      "epoch": 2.2656440734863283e-05,
      "step": 23757,
      "training_step_time": 6.782888889312744
    },
    {
      "epoch": 2.2657394409179687e-05,
      "model_forward_time": 1.759078025817871,
      "step": 23758
    },
    {
      "epoch": 2.2657394409179687e-05,
      "step": 23758,
      "training_step_time": 6.746172189712524
    },
    {
      "epoch": 2.2658348083496094e-05,
      "model_forward_time": 1.753779411315918,
      "step": 23759
    },
    {
      "epoch": 2.2658348083496094e-05,
      "step": 23759,
      "training_step_time": 6.818320035934448
    },
    {
      "epoch": 2.26593017578125e-05,
      "grad_norm": 0.186894029378891,
      "learning_rate": 9.068378692457748e-06,
      "loss": 0.0613,
      "step": 23760
    },
    {
      "epoch": 2.26593017578125e-05,
      "model_forward_time": 1.751335859298706,
      "step": 23760
    },
    {
      "dynamics_loss_avg": 0.06886585988104343,
      "epoch": 2.26593017578125e-05,
      "step": 23760
    },
    {
      "action_loss_avg": 0.004685660591349006,
      "epoch": 2.26593017578125e-05,
      "step": 23760
    },
    {
      "epoch": 2.26593017578125e-05,
      "step": 23760,
      "training_step_time": 6.8528032302856445
    },
    {
      "epoch": 2.2660255432128906e-05,
      "model_forward_time": 1.7596087455749512,
      "step": 23761
    },
    {
      "epoch": 2.2660255432128906e-05,
      "step": 23761,
      "training_step_time": 6.770004749298096
    },
    {
      "epoch": 2.2661209106445313e-05,
      "model_forward_time": 1.7714519500732422,
      "step": 23762
    },
    {
      "epoch": 2.2661209106445313e-05,
      "step": 23762,
      "training_step_time": 7.660472631454468
    },
    {
      "epoch": 2.266216278076172e-05,
      "model_forward_time": 2.9196391105651855,
      "step": 23763
    },
    {
      "epoch": 2.266216278076172e-05,
      "step": 23763,
      "training_step_time": 10.963398456573486
    },
    {
      "epoch": 2.2663116455078124e-05,
      "model_forward_time": 1.3890528678894043,
      "step": 23764
    },
    {
      "epoch": 2.2663116455078124e-05,
      "step": 23764,
      "training_step_time": 7.630438566207886
    },
    {
      "epoch": 2.2664070129394532e-05,
      "model_forward_time": 1.7641260623931885,
      "step": 23765
    },
    {
      "epoch": 2.2664070129394532e-05,
      "step": 23765,
      "training_step_time": 7.502185583114624
    },
    {
      "epoch": 2.2665023803710936e-05,
      "model_forward_time": 1.7574028968811035,
      "step": 23766
    },
    {
      "epoch": 2.2665023803710936e-05,
      "step": 23766,
      "training_step_time": 6.82442045211792
    },
    {
      "epoch": 2.2665977478027343e-05,
      "model_forward_time": 1.7567169666290283,
      "step": 23767
    },
    {
      "epoch": 2.2665977478027343e-05,
      "step": 23767,
      "training_step_time": 6.8564372062683105
    },
    {
      "epoch": 2.266693115234375e-05,
      "model_forward_time": 1.760502815246582,
      "step": 23768
    },
    {
      "epoch": 2.266693115234375e-05,
      "step": 23768,
      "training_step_time": 6.885763883590698
    },
    {
      "epoch": 2.2667884826660155e-05,
      "model_forward_time": 1.670130968093872,
      "step": 23769
    },
    {
      "epoch": 2.2667884826660155e-05,
      "step": 23769,
      "training_step_time": 6.838603496551514
    },
    {
      "epoch": 2.2668838500976562e-05,
      "grad_norm": 0.14471064507961273,
      "learning_rate": 9.06741727658019e-06,
      "loss": 0.0566,
      "step": 23770
    },
    {
      "epoch": 2.2668838500976562e-05,
      "model_forward_time": 1.6772990226745605,
      "step": 23770
    },
    {
      "dynamics_loss_avg": 0.05526532959192991,
      "epoch": 2.2668838500976562e-05,
      "step": 23770
    },
    {
      "action_loss_avg": 0.0026256693177856504,
      "epoch": 2.2668838500976562e-05,
      "step": 23770
    },
    {
      "epoch": 2.2668838500976562e-05,
      "step": 23770,
      "training_step_time": 6.876570463180542
    },
    {
      "epoch": 2.266979217529297e-05,
      "model_forward_time": 1.696728229522705,
      "step": 23771
    },
    {
      "epoch": 2.266979217529297e-05,
      "step": 23771,
      "training_step_time": 8.71773648262024
    },
    {
      "epoch": 2.2670745849609374e-05,
      "model_forward_time": 2.2148683071136475,
      "step": 23772
    },
    {
      "epoch": 2.2670745849609374e-05,
      "step": 23772,
      "training_step_time": 10.394137620925903
    },
    {
      "epoch": 2.267169952392578e-05,
      "model_forward_time": 1.6774108409881592,
      "step": 23773
    },
    {
      "epoch": 2.267169952392578e-05,
      "step": 23773,
      "training_step_time": 7.741777181625366
    },
    {
      "epoch": 2.267265319824219e-05,
      "model_forward_time": 1.6848361492156982,
      "step": 23774
    },
    {
      "epoch": 2.267265319824219e-05,
      "step": 23774,
      "training_step_time": 7.476589679718018
    },
    {
      "epoch": 2.2673606872558592e-05,
      "model_forward_time": 1.678436279296875,
      "step": 23775
    },
    {
      "epoch": 2.2673606872558592e-05,
      "step": 23775,
      "training_step_time": 6.801845550537109
    },
    {
      "epoch": 2.2674560546875e-05,
      "model_forward_time": 1.6779510974884033,
      "step": 23776
    },
    {
      "epoch": 2.2674560546875e-05,
      "step": 23776,
      "training_step_time": 6.899479389190674
    },
    {
      "epoch": 2.2675514221191407e-05,
      "model_forward_time": 1.680079698562622,
      "step": 23777
    },
    {
      "epoch": 2.2675514221191407e-05,
      "step": 23777,
      "training_step_time": 6.888346433639526
    },
    {
      "epoch": 2.267646789550781e-05,
      "model_forward_time": 1.6789605617523193,
      "step": 23778
    },
    {
      "epoch": 2.267646789550781e-05,
      "step": 23778,
      "training_step_time": 6.8818039894104
    },
    {
      "epoch": 2.267742156982422e-05,
      "model_forward_time": 1.68379807472229,
      "step": 23779
    },
    {
      "epoch": 2.267742156982422e-05,
      "step": 23779,
      "training_step_time": 6.900158643722534
    },
    {
      "epoch": 2.2678375244140626e-05,
      "grad_norm": 0.10482251644134521,
      "learning_rate": 9.066455415895993e-06,
      "loss": 0.0597,
      "step": 23780
    },
    {
      "epoch": 2.2678375244140626e-05,
      "model_forward_time": 1.6995816230773926,
      "step": 23780
    },
    {
      "dynamics_loss_avg": 0.0806418489664793,
      "epoch": 2.2678375244140626e-05,
      "step": 23780
    },
    {
      "action_loss_avg": 0.004383546509779989,
      "epoch": 2.2678375244140626e-05,
      "step": 23780
    },
    {
      "epoch": 2.2678375244140626e-05,
      "step": 23780,
      "training_step_time": 7.274131536483765
    },
    {
      "epoch": 2.267932891845703e-05,
      "model_forward_time": 2.220730781555176,
      "step": 23781
    },
    {
      "epoch": 2.267932891845703e-05,
      "step": 23781,
      "training_step_time": 8.810991048812866
    },
    {
      "epoch": 2.2680282592773438e-05,
      "model_forward_time": 1.6823585033416748,
      "step": 23782
    },
    {
      "epoch": 2.2680282592773438e-05,
      "step": 23782,
      "training_step_time": 7.779916524887085
    },
    {
      "epoch": 2.2681236267089845e-05,
      "model_forward_time": 1.680901050567627,
      "step": 23783
    },
    {
      "epoch": 2.2681236267089845e-05,
      "step": 23783,
      "training_step_time": 7.764162302017212
    },
    {
      "epoch": 2.268218994140625e-05,
      "model_forward_time": 1.6709222793579102,
      "step": 23784
    },
    {
      "epoch": 2.268218994140625e-05,
      "step": 23784,
      "training_step_time": 6.850203037261963
    },
    {
      "epoch": 2.2683143615722656e-05,
      "model_forward_time": 1.6784098148345947,
      "step": 23785
    },
    {
      "epoch": 2.2683143615722656e-05,
      "step": 23785,
      "training_step_time": 6.872020244598389
    },
    {
      "epoch": 2.2684097290039064e-05,
      "model_forward_time": 1.6772122383117676,
      "step": 23786
    },
    {
      "epoch": 2.2684097290039064e-05,
      "step": 23786,
      "training_step_time": 6.862796306610107
    },
    {
      "epoch": 2.2685050964355468e-05,
      "model_forward_time": 1.6811726093292236,
      "step": 23787
    },
    {
      "epoch": 2.2685050964355468e-05,
      "step": 23787,
      "training_step_time": 6.861065626144409
    },
    {
      "epoch": 2.2686004638671875e-05,
      "model_forward_time": 1.690399169921875,
      "step": 23788
    },
    {
      "epoch": 2.2686004638671875e-05,
      "step": 23788,
      "training_step_time": 6.852703332901001
    },
    {
      "epoch": 2.2686958312988283e-05,
      "model_forward_time": 1.676900863647461,
      "step": 23789
    },
    {
      "epoch": 2.2686958312988283e-05,
      "step": 23789,
      "training_step_time": 6.859029293060303
    },
    {
      "epoch": 2.2687911987304687e-05,
      "grad_norm": 0.14845427870750427,
      "learning_rate": 9.065493110510346e-06,
      "loss": 0.0607,
      "step": 23790
    },
    {
      "epoch": 2.2687911987304687e-05,
      "model_forward_time": 2.230543851852417,
      "step": 23790
    },
    {
      "dynamics_loss_avg": 0.06170671135187149,
      "epoch": 2.2687911987304687e-05,
      "step": 23790
    },
    {
      "action_loss_avg": 0.0021508899633772673,
      "epoch": 2.2687911987304687e-05,
      "step": 23790
    },
    {
      "epoch": 2.2687911987304687e-05,
      "step": 23790,
      "training_step_time": 9.58436369895935
    },
    {
      "epoch": 2.2688865661621094e-05,
      "model_forward_time": 1.6816956996917725,
      "step": 23791
    },
    {
      "epoch": 2.2688865661621094e-05,
      "step": 23791,
      "training_step_time": 9.28056001663208
    },
    {
      "epoch": 2.26898193359375e-05,
      "model_forward_time": 1.6780502796173096,
      "step": 23792
    },
    {
      "epoch": 2.26898193359375e-05,
      "step": 23792,
      "training_step_time": 7.76850152015686
    },
    {
      "epoch": 2.2690773010253906e-05,
      "model_forward_time": 1.6789896488189697,
      "step": 23793
    },
    {
      "epoch": 2.2690773010253906e-05,
      "step": 23793,
      "training_step_time": 6.897469997406006
    },
    {
      "epoch": 2.2691726684570313e-05,
      "model_forward_time": 1.679490327835083,
      "step": 23794
    },
    {
      "epoch": 2.2691726684570313e-05,
      "step": 23794,
      "training_step_time": 6.799667835235596
    },
    {
      "epoch": 2.269268035888672e-05,
      "model_forward_time": 1.679137945175171,
      "step": 23795
    },
    {
      "epoch": 2.269268035888672e-05,
      "step": 23795,
      "training_step_time": 6.819461822509766
    },
    {
      "epoch": 2.2693634033203124e-05,
      "model_forward_time": 1.678760051727295,
      "step": 23796
    },
    {
      "epoch": 2.2693634033203124e-05,
      "step": 23796,
      "training_step_time": 6.837135553359985
    },
    {
      "epoch": 2.2694587707519532e-05,
      "model_forward_time": 1.6815056800842285,
      "step": 23797
    },
    {
      "epoch": 2.2694587707519532e-05,
      "step": 23797,
      "training_step_time": 6.871795177459717
    },
    {
      "epoch": 2.2695541381835936e-05,
      "model_forward_time": 1.6847193241119385,
      "step": 23798
    },
    {
      "epoch": 2.2695541381835936e-05,
      "step": 23798,
      "training_step_time": 6.891259431838989
    },
    {
      "epoch": 2.2696495056152343e-05,
      "model_forward_time": 1.7329018115997314,
      "step": 23799
    },
    {
      "epoch": 2.2696495056152343e-05,
      "step": 23799,
      "training_step_time": 8.593396186828613
    },
    {
      "epoch": 2.269744873046875e-05,
      "grad_norm": 0.1686464548110962,
      "learning_rate": 9.064530360528484e-06,
      "loss": 0.06,
      "step": 23800
    },
    {
      "epoch": 2.269744873046875e-05,
      "model_forward_time": 1.6742277145385742,
      "step": 23800
    },
    {
      "dynamics_loss_avg": 0.06612538807094097,
      "epoch": 2.269744873046875e-05,
      "step": 23800
    },
    {
      "action_loss_avg": 0.003619985398836434,
      "epoch": 2.269744873046875e-05,
      "step": 23800
    },
    {
      "epoch": 2.269744873046875e-05,
      "step": 23800,
      "training_step_time": 8.94255018234253
    },
    {
      "epoch": 2.2698402404785155e-05,
      "model_forward_time": 1.6680314540863037,
      "step": 23801
    },
    {
      "epoch": 2.2698402404785155e-05,
      "step": 23801,
      "training_step_time": 7.801044702529907
    },
    {
      "epoch": 2.2699356079101562e-05,
      "model_forward_time": 1.6786222457885742,
      "step": 23802
    },
    {
      "epoch": 2.2699356079101562e-05,
      "step": 23802,
      "training_step_time": 7.581096887588501
    },
    {
      "epoch": 2.270030975341797e-05,
      "model_forward_time": 1.6798090934753418,
      "step": 23803
    },
    {
      "epoch": 2.270030975341797e-05,
      "step": 23803,
      "training_step_time": 6.955120325088501
    },
    {
      "epoch": 2.2701263427734374e-05,
      "model_forward_time": 1.6831023693084717,
      "step": 23804
    },
    {
      "epoch": 2.2701263427734374e-05,
      "step": 23804,
      "training_step_time": 6.894813776016235
    },
    {
      "epoch": 2.270221710205078e-05,
      "model_forward_time": 1.678312063217163,
      "step": 23805
    },
    {
      "epoch": 2.270221710205078e-05,
      "step": 23805,
      "training_step_time": 6.904383182525635
    },
    {
      "epoch": 2.270317077636719e-05,
      "model_forward_time": 1.6778807640075684,
      "step": 23806
    },
    {
      "epoch": 2.270317077636719e-05,
      "step": 23806,
      "training_step_time": 6.866828918457031
    },
    {
      "epoch": 2.2704124450683592e-05,
      "model_forward_time": 1.6789324283599854,
      "step": 23807
    },
    {
      "epoch": 2.2704124450683592e-05,
      "step": 23807,
      "training_step_time": 6.872194766998291
    },
    {
      "epoch": 2.2705078125e-05,
      "model_forward_time": 1.6986377239227295,
      "step": 23808
    },
    {
      "epoch": 2.2705078125e-05,
      "step": 23808,
      "training_step_time": 7.87241005897522
    },
    {
      "epoch": 2.2706031799316407e-05,
      "model_forward_time": 2.212120532989502,
      "step": 23809
    },
    {
      "epoch": 2.2706031799316407e-05,
      "step": 23809,
      "training_step_time": 9.33123779296875
    },
    {
      "epoch": 2.270698547363281e-05,
      "grad_norm": 0.18645691871643066,
      "learning_rate": 9.063567166055695e-06,
      "loss": 0.0565,
      "step": 23810
    },
    {
      "epoch": 2.270698547363281e-05,
      "model_forward_time": 1.675405502319336,
      "step": 23810
    },
    {
      "dynamics_loss_avg": 0.04518593288958073,
      "epoch": 2.270698547363281e-05,
      "step": 23810
    },
    {
      "action_loss_avg": 0.002384070202242583,
      "epoch": 2.270698547363281e-05,
      "step": 23810
    },
    {
      "epoch": 2.270698547363281e-05,
      "step": 23810,
      "training_step_time": 7.736458778381348
    },
    {
      "epoch": 2.270793914794922e-05,
      "model_forward_time": 1.6789703369140625,
      "step": 23811
    },
    {
      "epoch": 2.270793914794922e-05,
      "step": 23811,
      "training_step_time": 7.632082223892212
    },
    {
      "epoch": 2.2708892822265626e-05,
      "model_forward_time": 1.6783173084259033,
      "step": 23812
    },
    {
      "epoch": 2.2708892822265626e-05,
      "step": 23812,
      "training_step_time": 6.880887508392334
    },
    {
      "epoch": 2.270984649658203e-05,
      "model_forward_time": 1.6821274757385254,
      "step": 23813
    },
    {
      "epoch": 2.270984649658203e-05,
      "step": 23813,
      "training_step_time": 6.910833358764648
    },
    {
      "epoch": 2.2710800170898438e-05,
      "model_forward_time": 1.6800851821899414,
      "step": 23814
    },
    {
      "epoch": 2.2710800170898438e-05,
      "step": 23814,
      "training_step_time": 6.856874465942383
    },
    {
      "epoch": 2.2711753845214845e-05,
      "model_forward_time": 1.6818912029266357,
      "step": 23815
    },
    {
      "epoch": 2.2711753845214845e-05,
      "step": 23815,
      "training_step_time": 6.833198308944702
    },
    {
      "epoch": 2.271270751953125e-05,
      "model_forward_time": 1.6848418712615967,
      "step": 23816
    },
    {
      "epoch": 2.271270751953125e-05,
      "step": 23816,
      "training_step_time": 6.828465700149536
    },
    {
      "epoch": 2.2713661193847656e-05,
      "model_forward_time": 1.6803081035614014,
      "step": 23817
    },
    {
      "epoch": 2.2713661193847656e-05,
      "step": 23817,
      "training_step_time": 9.365480184555054
    },
    {
      "epoch": 2.2714614868164064e-05,
      "model_forward_time": 2.2126801013946533,
      "step": 23818
    },
    {
      "epoch": 2.2714614868164064e-05,
      "step": 23818,
      "training_step_time": 8.633000373840332
    },
    {
      "epoch": 2.2715568542480468e-05,
      "model_forward_time": 1.6795570850372314,
      "step": 23819
    },
    {
      "epoch": 2.2715568542480468e-05,
      "step": 23819,
      "training_step_time": 9.66724419593811
    },
    {
      "epoch": 2.2716522216796875e-05,
      "grad_norm": 0.16135311126708984,
      "learning_rate": 9.06260352719731e-06,
      "loss": 0.06,
      "step": 23820
    },
    {
      "epoch": 2.2716522216796875e-05,
      "model_forward_time": 1.6830322742462158,
      "step": 23820
    },
    {
      "dynamics_loss_avg": 0.06024901531636715,
      "epoch": 2.2716522216796875e-05,
      "step": 23820
    },
    {
      "action_loss_avg": 0.0029852737265173347,
      "epoch": 2.2716522216796875e-05,
      "step": 23820
    },
    {
      "epoch": 2.2716522216796875e-05,
      "step": 23820,
      "training_step_time": 7.543245553970337
    },
    {
      "epoch": 2.2717475891113283e-05,
      "model_forward_time": 1.6811482906341553,
      "step": 23821
    },
    {
      "epoch": 2.2717475891113283e-05,
      "step": 23821,
      "training_step_time": 6.7984044551849365
    },
    {
      "epoch": 2.2718429565429687e-05,
      "model_forward_time": 1.678675651550293,
      "step": 23822
    },
    {
      "epoch": 2.2718429565429687e-05,
      "step": 23822,
      "training_step_time": 6.8595311641693115
    },
    {
      "epoch": 2.2719383239746094e-05,
      "model_forward_time": 1.6731336116790771,
      "step": 23823
    },
    {
      "epoch": 2.2719383239746094e-05,
      "step": 23823,
      "training_step_time": 6.7663633823394775
    },
    {
      "epoch": 2.27203369140625e-05,
      "model_forward_time": 1.6797926425933838,
      "step": 23824
    },
    {
      "epoch": 2.27203369140625e-05,
      "step": 23824,
      "training_step_time": 6.747602224349976
    },
    {
      "epoch": 2.2721290588378906e-05,
      "model_forward_time": 1.6881020069122314,
      "step": 23825
    },
    {
      "epoch": 2.2721290588378906e-05,
      "step": 23825,
      "training_step_time": 6.855574369430542
    },
    {
      "epoch": 2.2722244262695313e-05,
      "model_forward_time": 1.6701693534851074,
      "step": 23826
    },
    {
      "epoch": 2.2722244262695313e-05,
      "step": 23826,
      "training_step_time": 9.27113127708435
    },
    {
      "epoch": 2.272319793701172e-05,
      "model_forward_time": 2.2214629650115967,
      "step": 23827
    },
    {
      "epoch": 2.272319793701172e-05,
      "step": 23827,
      "training_step_time": 10.673158407211304
    },
    {
      "epoch": 2.2724151611328124e-05,
      "model_forward_time": 1.6830921173095703,
      "step": 23828
    },
    {
      "epoch": 2.2724151611328124e-05,
      "step": 23828,
      "training_step_time": 8.741333723068237
    },
    {
      "epoch": 2.2725105285644532e-05,
      "model_forward_time": 1.6852500438690186,
      "step": 23829
    },
    {
      "epoch": 2.2725105285644532e-05,
      "step": 23829,
      "training_step_time": 7.679096221923828
    },
    {
      "epoch": 2.2726058959960936e-05,
      "grad_norm": 0.17833112180233002,
      "learning_rate": 9.06163944405871e-06,
      "loss": 0.0599,
      "step": 23830
    },
    {
      "epoch": 2.2726058959960936e-05,
      "model_forward_time": 1.6769695281982422,
      "step": 23830
    },
    {
      "dynamics_loss_avg": 0.05438297055661678,
      "epoch": 2.2726058959960936e-05,
      "step": 23830
    },
    {
      "action_loss_avg": 0.0022601260629016907,
      "epoch": 2.2726058959960936e-05,
      "step": 23830
    },
    {
      "epoch": 2.2726058959960936e-05,
      "step": 23830,
      "training_step_time": 7.188723564147949
    },
    {
      "epoch": 2.2727012634277343e-05,
      "model_forward_time": 1.6808075904846191,
      "step": 23831
    },
    {
      "epoch": 2.2727012634277343e-05,
      "step": 23831,
      "training_step_time": 6.741145133972168
    },
    {
      "epoch": 2.272796630859375e-05,
      "model_forward_time": 1.678490161895752,
      "step": 23832
    },
    {
      "epoch": 2.272796630859375e-05,
      "step": 23832,
      "training_step_time": 6.745378494262695
    },
    {
      "epoch": 2.2728919982910155e-05,
      "model_forward_time": 1.6907923221588135,
      "step": 23833
    },
    {
      "epoch": 2.2728919982910155e-05,
      "step": 23833,
      "training_step_time": 6.859872341156006
    },
    {
      "epoch": 2.2729873657226562e-05,
      "model_forward_time": 1.682373046875,
      "step": 23834
    },
    {
      "epoch": 2.2729873657226562e-05,
      "step": 23834,
      "training_step_time": 6.79142427444458
    },
    {
      "epoch": 2.273082733154297e-05,
      "model_forward_time": 1.6827890872955322,
      "step": 23835
    },
    {
      "epoch": 2.273082733154297e-05,
      "step": 23835,
      "training_step_time": 7.70223593711853
    },
    {
      "epoch": 2.2731781005859374e-05,
      "model_forward_time": 2.250765085220337,
      "step": 23836
    },
    {
      "epoch": 2.2731781005859374e-05,
      "step": 23836,
      "training_step_time": 8.813499212265015
    },
    {
      "epoch": 2.273273468017578e-05,
      "model_forward_time": 1.6871092319488525,
      "step": 23837
    },
    {
      "epoch": 2.273273468017578e-05,
      "step": 23837,
      "training_step_time": 10.975073337554932
    },
    {
      "epoch": 2.273368835449219e-05,
      "model_forward_time": 1.6843228340148926,
      "step": 23838
    },
    {
      "epoch": 2.273368835449219e-05,
      "step": 23838,
      "training_step_time": 7.68537974357605
    },
    {
      "epoch": 2.2734642028808592e-05,
      "model_forward_time": 1.6835031509399414,
      "step": 23839
    },
    {
      "epoch": 2.2734642028808592e-05,
      "step": 23839,
      "training_step_time": 7.464325666427612
    },
    {
      "epoch": 2.2735595703125e-05,
      "grad_norm": 0.0997936874628067,
      "learning_rate": 9.060674916745328e-06,
      "loss": 0.0645,
      "step": 23840
    },
    {
      "epoch": 2.2735595703125e-05,
      "model_forward_time": 1.6780214309692383,
      "step": 23840
    },
    {
      "dynamics_loss_avg": 0.06577829904854297,
      "epoch": 2.2735595703125e-05,
      "step": 23840
    },
    {
      "action_loss_avg": 0.0023142014164477588,
      "epoch": 2.2735595703125e-05,
      "step": 23840
    },
    {
      "epoch": 2.2735595703125e-05,
      "step": 23840,
      "training_step_time": 6.827317714691162
    },
    {
      "epoch": 2.2736549377441407e-05,
      "model_forward_time": 1.6867356300354004,
      "step": 23841
    },
    {
      "epoch": 2.2736549377441407e-05,
      "step": 23841,
      "training_step_time": 6.798612356185913
    },
    {
      "epoch": 2.273750305175781e-05,
      "model_forward_time": 1.3182930946350098,
      "step": 23842
    },
    {
      "epoch": 2.273750305175781e-05,
      "step": 23842,
      "training_step_time": 6.860234260559082
    },
    {
      "epoch": 2.273845672607422e-05,
      "model_forward_time": 1.6902122497558594,
      "step": 23843
    },
    {
      "epoch": 2.273845672607422e-05,
      "step": 23843,
      "training_step_time": 6.875118732452393
    },
    {
      "epoch": 2.2739410400390626e-05,
      "model_forward_time": 1.6869854927062988,
      "step": 23844
    },
    {
      "epoch": 2.2739410400390626e-05,
      "step": 23844,
      "training_step_time": 7.458265066146851
    },
    {
      "epoch": 2.274036407470703e-05,
      "model_forward_time": 2.2551393508911133,
      "step": 23845
    },
    {
      "epoch": 2.274036407470703e-05,
      "step": 23845,
      "training_step_time": 10.833488702774048
    },
    {
      "epoch": 2.2741317749023438e-05,
      "model_forward_time": 1.6896250247955322,
      "step": 23846
    },
    {
      "epoch": 2.2741317749023438e-05,
      "step": 23846,
      "training_step_time": 8.633357763290405
    },
    {
      "epoch": 2.2742271423339845e-05,
      "model_forward_time": 1.6915581226348877,
      "step": 23847
    },
    {
      "epoch": 2.2742271423339845e-05,
      "step": 23847,
      "training_step_time": 10.639560222625732
    },
    {
      "epoch": 2.274322509765625e-05,
      "model_forward_time": 1.6854074001312256,
      "step": 23848
    },
    {
      "epoch": 2.274322509765625e-05,
      "step": 23848,
      "training_step_time": 7.53665828704834
    },
    {
      "epoch": 2.2744178771972656e-05,
      "model_forward_time": 1.6839985847473145,
      "step": 23849
    },
    {
      "epoch": 2.2744178771972656e-05,
      "step": 23849,
      "training_step_time": 6.982589244842529
    },
    {
      "epoch": 2.2745132446289064e-05,
      "grad_norm": 0.1520998775959015,
      "learning_rate": 9.059709945362641e-06,
      "loss": 0.0633,
      "step": 23850
    },
    {
      "epoch": 2.2745132446289064e-05,
      "model_forward_time": 1.685737133026123,
      "step": 23850
    },
    {
      "dynamics_loss_avg": 0.05236140768975019,
      "epoch": 2.2745132446289064e-05,
      "step": 23850
    },
    {
      "action_loss_avg": 0.002350822230800986,
      "epoch": 2.2745132446289064e-05,
      "step": 23850
    },
    {
      "epoch": 2.2745132446289064e-05,
      "step": 23850,
      "training_step_time": 6.946735858917236
    },
    {
      "epoch": 2.2746086120605468e-05,
      "model_forward_time": 1.6819913387298584,
      "step": 23851
    },
    {
      "epoch": 2.2746086120605468e-05,
      "step": 23851,
      "training_step_time": 6.947875022888184
    },
    {
      "epoch": 2.2747039794921875e-05,
      "model_forward_time": 1.6871626377105713,
      "step": 23852
    },
    {
      "epoch": 2.2747039794921875e-05,
      "step": 23852,
      "training_step_time": 6.9782164096832275
    },
    {
      "epoch": 2.2747993469238283e-05,
      "model_forward_time": 1.686697006225586,
      "step": 23853
    },
    {
      "epoch": 2.2747993469238283e-05,
      "step": 23853,
      "training_step_time": 7.504994630813599
    },
    {
      "epoch": 2.2748947143554687e-05,
      "model_forward_time": 1.7166554927825928,
      "step": 23854
    },
    {
      "epoch": 2.2748947143554687e-05,
      "step": 23854,
      "training_step_time": 8.811745643615723
    },
    {
      "epoch": 2.2749900817871094e-05,
      "model_forward_time": 2.2317535877227783,
      "step": 23855
    },
    {
      "epoch": 2.2749900817871094e-05,
      "step": 23855,
      "training_step_time": 9.056824207305908
    },
    {
      "epoch": 2.27508544921875e-05,
      "model_forward_time": 1.686619758605957,
      "step": 23856
    },
    {
      "epoch": 2.27508544921875e-05,
      "step": 23856,
      "training_step_time": 8.434990644454956
    },
    {
      "epoch": 2.2751808166503906e-05,
      "model_forward_time": 1.6857852935791016,
      "step": 23857
    },
    {
      "epoch": 2.2751808166503906e-05,
      "step": 23857,
      "training_step_time": 7.674529790878296
    },
    {
      "epoch": 2.2752761840820313e-05,
      "model_forward_time": 1.693894386291504,
      "step": 23858
    },
    {
      "epoch": 2.2752761840820313e-05,
      "step": 23858,
      "training_step_time": 7.286325454711914
    },
    {
      "epoch": 2.275371551513672e-05,
      "model_forward_time": 1.317014217376709,
      "step": 23859
    },
    {
      "epoch": 2.275371551513672e-05,
      "step": 23859,
      "training_step_time": 6.921895265579224
    },
    {
      "epoch": 2.2754669189453124e-05,
      "grad_norm": 0.1302536278963089,
      "learning_rate": 9.05874453001618e-06,
      "loss": 0.0642,
      "step": 23860
    },
    {
      "epoch": 2.2754669189453124e-05,
      "model_forward_time": 1.6840589046478271,
      "step": 23860
    },
    {
      "dynamics_loss_avg": 0.06014846041798592,
      "epoch": 2.2754669189453124e-05,
      "step": 23860
    },
    {
      "action_loss_avg": 0.002930901845684275,
      "epoch": 2.2754669189453124e-05,
      "step": 23860
    },
    {
      "epoch": 2.2754669189453124e-05,
      "step": 23860,
      "training_step_time": 6.914905071258545
    },
    {
      "epoch": 2.2755622863769532e-05,
      "model_forward_time": 1.687330722808838,
      "step": 23861
    },
    {
      "epoch": 2.2755622863769532e-05,
      "step": 23861,
      "training_step_time": 6.92186975479126
    },
    {
      "epoch": 2.2756576538085936e-05,
      "model_forward_time": 1.6848657131195068,
      "step": 23862
    },
    {
      "epoch": 2.2756576538085936e-05,
      "step": 23862,
      "training_step_time": 7.486845016479492
    },
    {
      "epoch": 2.2757530212402343e-05,
      "model_forward_time": 1.702476978302002,
      "step": 23863
    },
    {
      "epoch": 2.2757530212402343e-05,
      "step": 23863,
      "training_step_time": 7.766194820404053
    },
    {
      "epoch": 2.275848388671875e-05,
      "model_forward_time": 2.2276995182037354,
      "step": 23864
    },
    {
      "epoch": 2.275848388671875e-05,
      "step": 23864,
      "training_step_time": 9.783382415771484
    },
    {
      "epoch": 2.2759437561035155e-05,
      "model_forward_time": 1.6752233505249023,
      "step": 23865
    },
    {
      "epoch": 2.2759437561035155e-05,
      "step": 23865,
      "training_step_time": 8.785490036010742
    },
    {
      "epoch": 2.2760391235351562e-05,
      "model_forward_time": 1.6857385635375977,
      "step": 23866
    },
    {
      "epoch": 2.2760391235351562e-05,
      "step": 23866,
      "training_step_time": 7.702730417251587
    },
    {
      "epoch": 2.276134490966797e-05,
      "model_forward_time": 1.69224214553833,
      "step": 23867
    },
    {
      "epoch": 2.276134490966797e-05,
      "step": 23867,
      "training_step_time": 7.472368240356445
    },
    {
      "epoch": 2.2762298583984374e-05,
      "model_forward_time": 1.6836142539978027,
      "step": 23868
    },
    {
      "epoch": 2.2762298583984374e-05,
      "step": 23868,
      "training_step_time": 6.881069183349609
    },
    {
      "epoch": 2.276325225830078e-05,
      "model_forward_time": 1.689378261566162,
      "step": 23869
    },
    {
      "epoch": 2.276325225830078e-05,
      "step": 23869,
      "training_step_time": 6.86362361907959
    },
    {
      "epoch": 2.276420593261719e-05,
      "grad_norm": 0.20909716188907623,
      "learning_rate": 9.057778670811517e-06,
      "loss": 0.0664,
      "step": 23870
    },
    {
      "epoch": 2.276420593261719e-05,
      "model_forward_time": 1.6788103580474854,
      "step": 23870
    },
    {
      "dynamics_loss_avg": 0.06881728209555149,
      "epoch": 2.276420593261719e-05,
      "step": 23870
    },
    {
      "action_loss_avg": 0.0027808088227175174,
      "epoch": 2.276420593261719e-05,
      "step": 23870
    },
    {
      "epoch": 2.276420593261719e-05,
      "step": 23870,
      "training_step_time": 6.853212594985962
    },
    {
      "epoch": 2.2765159606933592e-05,
      "model_forward_time": 1.6914560794830322,
      "step": 23871
    },
    {
      "epoch": 2.2765159606933592e-05,
      "step": 23871,
      "training_step_time": 7.423419237136841
    },
    {
      "epoch": 2.276611328125e-05,
      "model_forward_time": 1.6928410530090332,
      "step": 23872
    },
    {
      "epoch": 2.276611328125e-05,
      "step": 23872,
      "training_step_time": 7.69965672492981
    },
    {
      "epoch": 2.2767066955566407e-05,
      "model_forward_time": 2.2315149307250977,
      "step": 23873
    },
    {
      "epoch": 2.2767066955566407e-05,
      "step": 23873,
      "training_step_time": 8.548521041870117
    },
    {
      "epoch": 2.276802062988281e-05,
      "model_forward_time": 1.6873931884765625,
      "step": 23874
    },
    {
      "epoch": 2.276802062988281e-05,
      "step": 23874,
      "training_step_time": 7.988305568695068
    },
    {
      "epoch": 2.276897430419922e-05,
      "model_forward_time": 1.3004276752471924,
      "step": 23875
    },
    {
      "epoch": 2.276897430419922e-05,
      "step": 23875,
      "training_step_time": 8.425840377807617
    },
    {
      "epoch": 2.2769927978515626e-05,
      "model_forward_time": 1.6871299743652344,
      "step": 23876
    },
    {
      "epoch": 2.2769927978515626e-05,
      "step": 23876,
      "training_step_time": 7.364156484603882
    },
    {
      "epoch": 2.277088165283203e-05,
      "model_forward_time": 0.9772214889526367,
      "step": 23877
    },
    {
      "epoch": 2.277088165283203e-05,
      "step": 23877,
      "training_step_time": 6.865692853927612
    },
    {
      "epoch": 2.2771835327148438e-05,
      "model_forward_time": 1.6900701522827148,
      "step": 23878
    },
    {
      "epoch": 2.2771835327148438e-05,
      "step": 23878,
      "training_step_time": 6.88253927230835
    },
    {
      "epoch": 2.2772789001464845e-05,
      "model_forward_time": 1.6877529621124268,
      "step": 23879
    },
    {
      "epoch": 2.2772789001464845e-05,
      "step": 23879,
      "training_step_time": 6.864226818084717
    },
    {
      "epoch": 2.277374267578125e-05,
      "grad_norm": 0.16236180067062378,
      "learning_rate": 9.05681236785428e-06,
      "loss": 0.0688,
      "step": 23880
    },
    {
      "epoch": 2.277374267578125e-05,
      "model_forward_time": 1.6803171634674072,
      "step": 23880
    },
    {
      "dynamics_loss_avg": 0.061518175154924394,
      "epoch": 2.277374267578125e-05,
      "step": 23880
    },
    {
      "action_loss_avg": 0.004199582780711353,
      "epoch": 2.277374267578125e-05,
      "step": 23880
    },
    {
      "epoch": 2.277374267578125e-05,
      "step": 23880,
      "training_step_time": 7.084986925125122
    },
    {
      "epoch": 2.2774696350097656e-05,
      "model_forward_time": 1.687882423400879,
      "step": 23881
    },
    {
      "epoch": 2.2774696350097656e-05,
      "step": 23881,
      "training_step_time": 8.568257331848145
    },
    {
      "epoch": 2.2775650024414064e-05,
      "model_forward_time": 1.750969648361206,
      "step": 23882
    },
    {
      "epoch": 2.2775650024414064e-05,
      "step": 23882,
      "training_step_time": 8.713696718215942
    },
    {
      "epoch": 2.2776603698730468e-05,
      "model_forward_time": 2.2294888496398926,
      "step": 23883
    },
    {
      "epoch": 2.2776603698730468e-05,
      "step": 23883,
      "training_step_time": 8.860814094543457
    },
    {
      "epoch": 2.2777557373046875e-05,
      "model_forward_time": 1.6917226314544678,
      "step": 23884
    },
    {
      "epoch": 2.2777557373046875e-05,
      "step": 23884,
      "training_step_time": 7.751380920410156
    },
    {
      "epoch": 2.2778511047363283e-05,
      "model_forward_time": 1.6878867149353027,
      "step": 23885
    },
    {
      "epoch": 2.2778511047363283e-05,
      "step": 23885,
      "training_step_time": 7.433642387390137
    },
    {
      "epoch": 2.2779464721679687e-05,
      "model_forward_time": 1.6874468326568604,
      "step": 23886
    },
    {
      "epoch": 2.2779464721679687e-05,
      "step": 23886,
      "training_step_time": 6.839648485183716
    },
    {
      "epoch": 2.2780418395996094e-05,
      "model_forward_time": 1.6876442432403564,
      "step": 23887
    },
    {
      "epoch": 2.2780418395996094e-05,
      "step": 23887,
      "training_step_time": 6.855576038360596
    },
    {
      "epoch": 2.27813720703125e-05,
      "model_forward_time": 1.6897773742675781,
      "step": 23888
    },
    {
      "epoch": 2.27813720703125e-05,
      "step": 23888,
      "training_step_time": 6.86558723449707
    },
    {
      "epoch": 2.2782325744628906e-05,
      "model_forward_time": 1.68556547164917,
      "step": 23889
    },
    {
      "epoch": 2.2782325744628906e-05,
      "step": 23889,
      "training_step_time": 6.874081611633301
    },
    {
      "epoch": 2.2783279418945313e-05,
      "grad_norm": 0.14278043806552887,
      "learning_rate": 9.055845621250144e-06,
      "loss": 0.0626,
      "step": 23890
    },
    {
      "epoch": 2.2783279418945313e-05,
      "model_forward_time": 1.6826796531677246,
      "step": 23890
    },
    {
      "dynamics_loss_avg": 0.06799421459436417,
      "epoch": 2.2783279418945313e-05,
      "step": 23890
    },
    {
      "action_loss_avg": 0.0026199817832093684,
      "epoch": 2.2783279418945313e-05,
      "step": 23890
    },
    {
      "epoch": 2.2783279418945313e-05,
      "step": 23890,
      "training_step_time": 8.13487195968628
    },
    {
      "epoch": 2.2784233093261717e-05,
      "model_forward_time": 1.7378902435302734,
      "step": 23891
    },
    {
      "epoch": 2.2784233093261717e-05,
      "step": 23891,
      "training_step_time": 8.319315433502197
    },
    {
      "epoch": 2.2785186767578124e-05,
      "model_forward_time": 2.2220327854156494,
      "step": 23892
    },
    {
      "epoch": 2.2785186767578124e-05,
      "step": 23892,
      "training_step_time": 7.989872694015503
    },
    {
      "epoch": 2.2786140441894532e-05,
      "model_forward_time": 1.6902377605438232,
      "step": 23893
    },
    {
      "epoch": 2.2786140441894532e-05,
      "step": 23893,
      "training_step_time": 7.786804914474487
    },
    {
      "epoch": 2.2787094116210936e-05,
      "model_forward_time": 1.689307451248169,
      "step": 23894
    },
    {
      "epoch": 2.2787094116210936e-05,
      "step": 23894,
      "training_step_time": 7.425502061843872
    },
    {
      "epoch": 2.2788047790527343e-05,
      "model_forward_time": 1.6835978031158447,
      "step": 23895
    },
    {
      "epoch": 2.2788047790527343e-05,
      "step": 23895,
      "training_step_time": 7.45973801612854
    },
    {
      "epoch": 2.278900146484375e-05,
      "model_forward_time": 1.6889703273773193,
      "step": 23896
    },
    {
      "epoch": 2.278900146484375e-05,
      "step": 23896,
      "training_step_time": 6.869741916656494
    },
    {
      "epoch": 2.2789955139160155e-05,
      "model_forward_time": 1.685920000076294,
      "step": 23897
    },
    {
      "epoch": 2.2789955139160155e-05,
      "step": 23897,
      "training_step_time": 6.871926784515381
    },
    {
      "epoch": 2.2790908813476562e-05,
      "model_forward_time": 1.6898682117462158,
      "step": 23898
    },
    {
      "epoch": 2.2790908813476562e-05,
      "step": 23898,
      "training_step_time": 6.863729953765869
    },
    {
      "epoch": 2.279186248779297e-05,
      "model_forward_time": 1.686645746231079,
      "step": 23899
    },
    {
      "epoch": 2.279186248779297e-05,
      "step": 23899,
      "training_step_time": 7.52809739112854
    },
    {
      "epoch": 2.2792816162109374e-05,
      "grad_norm": 0.10941147804260254,
      "learning_rate": 9.054878431104825e-06,
      "loss": 0.0621,
      "step": 23900
    },
    {
      "epoch": 2.2792816162109374e-05,
      "model_forward_time": 1.712656021118164,
      "step": 23900
    },
    {
      "dynamics_loss_avg": 0.06119363680481911,
      "epoch": 2.2792816162109374e-05,
      "step": 23900
    },
    {
      "action_loss_avg": 0.002800161496270448,
      "epoch": 2.2792816162109374e-05,
      "step": 23900
    },
    {
      "epoch": 2.2792816162109374e-05,
      "step": 23900,
      "training_step_time": 8.19804048538208
    },
    {
      "epoch": 2.279376983642578e-05,
      "model_forward_time": 2.2149813175201416,
      "step": 23901
    },
    {
      "epoch": 2.279376983642578e-05,
      "step": 23901,
      "training_step_time": 8.913808107376099
    },
    {
      "epoch": 2.279472351074219e-05,
      "model_forward_time": 1.6837809085845947,
      "step": 23902
    },
    {
      "epoch": 2.279472351074219e-05,
      "step": 23902,
      "training_step_time": 7.724691867828369
    },
    {
      "epoch": 2.2795677185058593e-05,
      "model_forward_time": 1.6823604106903076,
      "step": 23903
    },
    {
      "epoch": 2.2795677185058593e-05,
      "step": 23903,
      "training_step_time": 8.494913578033447
    },
    {
      "epoch": 2.2796630859375e-05,
      "model_forward_time": 1.6843392848968506,
      "step": 23904
    },
    {
      "epoch": 2.2796630859375e-05,
      "step": 23904,
      "training_step_time": 7.507962465286255
    },
    {
      "epoch": 2.2797584533691407e-05,
      "model_forward_time": 1.6814751625061035,
      "step": 23905
    },
    {
      "epoch": 2.2797584533691407e-05,
      "step": 23905,
      "training_step_time": 6.956470012664795
    },
    {
      "epoch": 2.279853820800781e-05,
      "model_forward_time": 1.6865744590759277,
      "step": 23906
    },
    {
      "epoch": 2.279853820800781e-05,
      "step": 23906,
      "training_step_time": 6.919979810714722
    },
    {
      "epoch": 2.279949188232422e-05,
      "model_forward_time": 1.6849584579467773,
      "step": 23907
    },
    {
      "epoch": 2.279949188232422e-05,
      "step": 23907,
      "training_step_time": 6.977052927017212
    },
    {
      "epoch": 2.2800445556640626e-05,
      "model_forward_time": 1.6935482025146484,
      "step": 23908
    },
    {
      "epoch": 2.2800445556640626e-05,
      "step": 23908,
      "training_step_time": 8.078711032867432
    },
    {
      "epoch": 2.280139923095703e-05,
      "model_forward_time": 1.6891648769378662,
      "step": 23909
    },
    {
      "epoch": 2.280139923095703e-05,
      "step": 23909,
      "training_step_time": 7.634799957275391
    },
    {
      "epoch": 2.2802352905273438e-05,
      "grad_norm": 0.12399518489837646,
      "learning_rate": 9.0539107975241e-06,
      "loss": 0.0632,
      "step": 23910
    },
    {
      "epoch": 2.2802352905273438e-05,
      "model_forward_time": 2.215425491333008,
      "step": 23910
    },
    {
      "dynamics_loss_avg": 0.08994033392518759,
      "epoch": 2.2802352905273438e-05,
      "step": 23910
    },
    {
      "action_loss_avg": 0.003880411898717284,
      "epoch": 2.2802352905273438e-05,
      "step": 23910
    },
    {
      "epoch": 2.2802352905273438e-05,
      "step": 23910,
      "training_step_time": 9.658769607543945
    },
    {
      "epoch": 2.2803306579589845e-05,
      "model_forward_time": 1.6850130558013916,
      "step": 23911
    },
    {
      "epoch": 2.2803306579589845e-05,
      "step": 23911,
      "training_step_time": 8.619850158691406
    },
    {
      "epoch": 2.280426025390625e-05,
      "model_forward_time": 1.681807518005371,
      "step": 23912
    },
    {
      "epoch": 2.280426025390625e-05,
      "step": 23912,
      "training_step_time": 7.656979084014893
    },
    {
      "epoch": 2.2805213928222656e-05,
      "model_forward_time": 1.6886944770812988,
      "step": 23913
    },
    {
      "epoch": 2.2805213928222656e-05,
      "step": 23913,
      "training_step_time": 7.449572563171387
    },
    {
      "epoch": 2.2806167602539064e-05,
      "model_forward_time": 1.6869518756866455,
      "step": 23914
    },
    {
      "epoch": 2.2806167602539064e-05,
      "step": 23914,
      "training_step_time": 6.85851263999939
    },
    {
      "epoch": 2.2807121276855468e-05,
      "model_forward_time": 1.683929204940796,
      "step": 23915
    },
    {
      "epoch": 2.2807121276855468e-05,
      "step": 23915,
      "training_step_time": 6.858675003051758
    },
    {
      "epoch": 2.2808074951171875e-05,
      "model_forward_time": 1.681786060333252,
      "step": 23916
    },
    {
      "epoch": 2.2808074951171875e-05,
      "step": 23916,
      "training_step_time": 6.849176645278931
    },
    {
      "epoch": 2.2809028625488283e-05,
      "model_forward_time": 1.683237075805664,
      "step": 23917
    },
    {
      "epoch": 2.2809028625488283e-05,
      "step": 23917,
      "training_step_time": 7.657662630081177
    },
    {
      "epoch": 2.2809982299804687e-05,
      "model_forward_time": 1.6793229579925537,
      "step": 23918
    },
    {
      "epoch": 2.2809982299804687e-05,
      "step": 23918,
      "training_step_time": 7.7226269245147705
    },
    {
      "epoch": 2.2810935974121094e-05,
      "model_forward_time": 2.258622884750366,
      "step": 23919
    },
    {
      "epoch": 2.2810935974121094e-05,
      "step": 23919,
      "training_step_time": 8.377508163452148
    },
    {
      "epoch": 2.28118896484375e-05,
      "grad_norm": 0.10975944995880127,
      "learning_rate": 9.052942720613784e-06,
      "loss": 0.0627,
      "step": 23920
    },
    {
      "epoch": 2.28118896484375e-05,
      "model_forward_time": 1.318195104598999,
      "step": 23920
    },
    {
      "dynamics_loss_avg": 0.07601252309978009,
      "epoch": 2.28118896484375e-05,
      "step": 23920
    },
    {
      "action_loss_avg": 0.002506999342585914,
      "epoch": 2.28118896484375e-05,
      "step": 23920
    },
    {
      "epoch": 2.28118896484375e-05,
      "step": 23920,
      "training_step_time": 8.492204666137695
    },
    {
      "epoch": 2.2812843322753906e-05,
      "model_forward_time": 1.6848547458648682,
      "step": 23921
    },
    {
      "epoch": 2.2812843322753906e-05,
      "step": 23921,
      "training_step_time": 8.127954959869385
    },
    {
      "epoch": 2.2813796997070313e-05,
      "model_forward_time": 1.683260440826416,
      "step": 23922
    },
    {
      "epoch": 2.2813796997070313e-05,
      "step": 23922,
      "training_step_time": 7.697003126144409
    },
    {
      "epoch": 2.2814750671386717e-05,
      "model_forward_time": 1.6839921474456787,
      "step": 23923
    },
    {
      "epoch": 2.2814750671386717e-05,
      "step": 23923,
      "training_step_time": 6.873464107513428
    },
    {
      "epoch": 2.2815704345703125e-05,
      "model_forward_time": 1.6874661445617676,
      "step": 23924
    },
    {
      "epoch": 2.2815704345703125e-05,
      "step": 23924,
      "training_step_time": 6.774933576583862
    },
    {
      "epoch": 2.2816658020019532e-05,
      "model_forward_time": 1.6849472522735596,
      "step": 23925
    },
    {
      "epoch": 2.2816658020019532e-05,
      "step": 23925,
      "training_step_time": 6.842803239822388
    },
    {
      "epoch": 2.2817611694335936e-05,
      "model_forward_time": 1.6860640048980713,
      "step": 23926
    },
    {
      "epoch": 2.2817611694335936e-05,
      "step": 23926,
      "training_step_time": 7.412535667419434
    },
    {
      "epoch": 2.2818565368652343e-05,
      "model_forward_time": 1.6907987594604492,
      "step": 23927
    },
    {
      "epoch": 2.2818565368652343e-05,
      "step": 23927,
      "training_step_time": 7.7002787590026855
    },
    {
      "epoch": 2.281951904296875e-05,
      "model_forward_time": 1.737847089767456,
      "step": 23928
    },
    {
      "epoch": 2.281951904296875e-05,
      "step": 23928,
      "training_step_time": 9.498351573944092
    },
    {
      "epoch": 2.2820472717285155e-05,
      "model_forward_time": 2.2258546352386475,
      "step": 23929
    },
    {
      "epoch": 2.2820472717285155e-05,
      "step": 23929,
      "training_step_time": 9.026954889297485
    },
    {
      "epoch": 2.2821426391601562e-05,
      "grad_norm": 0.15175671875476837,
      "learning_rate": 9.051974200479745e-06,
      "loss": 0.0575,
      "step": 23930
    },
    {
      "epoch": 2.2821426391601562e-05,
      "model_forward_time": 1.6808040142059326,
      "step": 23930
    },
    {
      "dynamics_loss_avg": 0.06952466145157814,
      "epoch": 2.2821426391601562e-05,
      "step": 23930
    },
    {
      "action_loss_avg": 0.0026100035407580434,
      "epoch": 2.2821426391601562e-05,
      "step": 23930
    },
    {
      "epoch": 2.2821426391601562e-05,
      "step": 23930,
      "training_step_time": 7.706602573394775
    },
    {
      "epoch": 2.282238006591797e-05,
      "model_forward_time": 1.688849925994873,
      "step": 23931
    },
    {
      "epoch": 2.282238006591797e-05,
      "step": 23931,
      "training_step_time": 7.593359470367432
    },
    {
      "epoch": 2.2823333740234374e-05,
      "model_forward_time": 1.69034743309021,
      "step": 23932
    },
    {
      "epoch": 2.2823333740234374e-05,
      "step": 23932,
      "training_step_time": 6.886476278305054
    },
    {
      "epoch": 2.282428741455078e-05,
      "model_forward_time": 1.6830792427062988,
      "step": 23933
    },
    {
      "epoch": 2.282428741455078e-05,
      "step": 23933,
      "training_step_time": 6.861773490905762
    },
    {
      "epoch": 2.282524108886719e-05,
      "model_forward_time": 1.6846468448638916,
      "step": 23934
    },
    {
      "epoch": 2.282524108886719e-05,
      "step": 23934,
      "training_step_time": 6.877480506896973
    },
    {
      "epoch": 2.2826194763183593e-05,
      "model_forward_time": 1.685591697692871,
      "step": 23935
    },
    {
      "epoch": 2.2826194763183593e-05,
      "step": 23935,
      "training_step_time": 7.411060571670532
    },
    {
      "epoch": 2.28271484375e-05,
      "model_forward_time": 1.683190107345581,
      "step": 23936
    },
    {
      "epoch": 2.28271484375e-05,
      "step": 23936,
      "training_step_time": 8.428248882293701
    },
    {
      "epoch": 2.2828102111816407e-05,
      "model_forward_time": 1.741363525390625,
      "step": 23937
    },
    {
      "epoch": 2.2828102111816407e-05,
      "step": 23937,
      "training_step_time": 8.723477125167847
    },
    {
      "epoch": 2.282905578613281e-05,
      "model_forward_time": 2.231808662414551,
      "step": 23938
    },
    {
      "epoch": 2.282905578613281e-05,
      "step": 23938,
      "training_step_time": 8.26036024093628
    },
    {
      "epoch": 2.283000946044922e-05,
      "model_forward_time": 1.683804988861084,
      "step": 23939
    },
    {
      "epoch": 2.283000946044922e-05,
      "step": 23939,
      "training_step_time": 7.935032844543457
    },
    {
      "epoch": 2.2830963134765626e-05,
      "grad_norm": 0.12360621988773346,
      "learning_rate": 9.051005237227901e-06,
      "loss": 0.0598,
      "step": 23940
    },
    {
      "epoch": 2.2830963134765626e-05,
      "model_forward_time": 1.6816041469573975,
      "step": 23940
    },
    {
      "dynamics_loss_avg": 0.06320590320974588,
      "epoch": 2.2830963134765626e-05,
      "step": 23940
    },
    {
      "action_loss_avg": 0.003325730824144557,
      "epoch": 2.2830963134765626e-05,
      "step": 23940
    },
    {
      "epoch": 2.2830963134765626e-05,
      "step": 23940,
      "training_step_time": 7.667365312576294
    },
    {
      "epoch": 2.283191680908203e-05,
      "model_forward_time": 1.6835167407989502,
      "step": 23941
    },
    {
      "epoch": 2.283191680908203e-05,
      "step": 23941,
      "training_step_time": 7.752461671829224
    },
    {
      "epoch": 2.2832870483398438e-05,
      "model_forward_time": 1.6852941513061523,
      "step": 23942
    },
    {
      "epoch": 2.2832870483398438e-05,
      "step": 23942,
      "training_step_time": 6.860026836395264
    },
    {
      "epoch": 2.2833824157714845e-05,
      "model_forward_time": 1.6860861778259277,
      "step": 23943
    },
    {
      "epoch": 2.2833824157714845e-05,
      "step": 23943,
      "training_step_time": 6.901758193969727
    },
    {
      "epoch": 2.283477783203125e-05,
      "model_forward_time": 1.6871366500854492,
      "step": 23944
    },
    {
      "epoch": 2.283477783203125e-05,
      "step": 23944,
      "training_step_time": 7.769612550735474
    },
    {
      "epoch": 2.2835731506347657e-05,
      "model_forward_time": 1.6852669715881348,
      "step": 23945
    },
    {
      "epoch": 2.2835731506347657e-05,
      "step": 23945,
      "training_step_time": 7.720505714416504
    },
    {
      "epoch": 2.2836685180664064e-05,
      "model_forward_time": 1.6950268745422363,
      "step": 23946
    },
    {
      "epoch": 2.2836685180664064e-05,
      "step": 23946,
      "training_step_time": 8.645723104476929
    },
    {
      "epoch": 2.2837638854980468e-05,
      "model_forward_time": 2.2236788272857666,
      "step": 23947
    },
    {
      "epoch": 2.2837638854980468e-05,
      "step": 23947,
      "training_step_time": 9.581258535385132
    },
    {
      "epoch": 2.2838592529296875e-05,
      "model_forward_time": 1.6878278255462646,
      "step": 23948
    },
    {
      "epoch": 2.2838592529296875e-05,
      "step": 23948,
      "training_step_time": 9.099400281906128
    },
    {
      "epoch": 2.2839546203613283e-05,
      "model_forward_time": 1.6881048679351807,
      "step": 23949
    },
    {
      "epoch": 2.2839546203613283e-05,
      "step": 23949,
      "training_step_time": 8.079546689987183
    },
    {
      "epoch": 2.2840499877929687e-05,
      "grad_norm": 0.15512079000473022,
      "learning_rate": 9.050035830964214e-06,
      "loss": 0.063,
      "step": 23950
    },
    {
      "epoch": 2.2840499877929687e-05,
      "model_forward_time": 1.682525634765625,
      "step": 23950
    },
    {
      "dynamics_loss_avg": 0.08222409151494503,
      "epoch": 2.2840499877929687e-05,
      "step": 23950
    },
    {
      "action_loss_avg": 0.004272143560228869,
      "epoch": 2.2840499877929687e-05,
      "step": 23950
    },
    {
      "epoch": 2.2840499877929687e-05,
      "step": 23950,
      "training_step_time": 7.733208179473877
    },
    {
      "epoch": 2.2841453552246094e-05,
      "model_forward_time": 1.6878318786621094,
      "step": 23951
    },
    {
      "epoch": 2.2841453552246094e-05,
      "step": 23951,
      "training_step_time": 6.846404790878296
    },
    {
      "epoch": 2.28424072265625e-05,
      "model_forward_time": 1.3044586181640625,
      "step": 23952
    },
    {
      "epoch": 2.28424072265625e-05,
      "step": 23952,
      "training_step_time": 6.8504674434661865
    },
    {
      "epoch": 2.2843360900878906e-05,
      "model_forward_time": 1.6829807758331299,
      "step": 23953
    },
    {
      "epoch": 2.2843360900878906e-05,
      "step": 23953,
      "training_step_time": 7.224311113357544
    },
    {
      "epoch": 2.2844314575195313e-05,
      "model_forward_time": 1.3134455680847168,
      "step": 23954
    },
    {
      "epoch": 2.2844314575195313e-05,
      "step": 23954,
      "training_step_time": 8.65911865234375
    },
    {
      "epoch": 2.2845268249511717e-05,
      "model_forward_time": 1.6988334655761719,
      "step": 23955
    },
    {
      "epoch": 2.2845268249511717e-05,
      "step": 23955,
      "training_step_time": 9.542830467224121
    },
    {
      "epoch": 2.2846221923828125e-05,
      "model_forward_time": 1.709519624710083,
      "step": 23956
    },
    {
      "epoch": 2.2846221923828125e-05,
      "step": 23956,
      "training_step_time": 8.853526830673218
    },
    {
      "epoch": 2.2847175598144532e-05,
      "model_forward_time": 1.6874995231628418,
      "step": 23957
    },
    {
      "epoch": 2.2847175598144532e-05,
      "step": 23957,
      "training_step_time": 8.057332754135132
    },
    {
      "epoch": 2.2848129272460936e-05,
      "model_forward_time": 1.6843645572662354,
      "step": 23958
    },
    {
      "epoch": 2.2848129272460936e-05,
      "step": 23958,
      "training_step_time": 7.712602138519287
    },
    {
      "epoch": 2.2849082946777343e-05,
      "model_forward_time": 1.684891939163208,
      "step": 23959
    },
    {
      "epoch": 2.2849082946777343e-05,
      "step": 23959,
      "training_step_time": 7.8040993213653564
    },
    {
      "epoch": 2.285003662109375e-05,
      "grad_norm": 0.22656984627246857,
      "learning_rate": 9.049065981794699e-06,
      "loss": 0.0631,
      "step": 23960
    },
    {
      "epoch": 2.285003662109375e-05,
      "model_forward_time": 1.2969970703125,
      "step": 23960
    },
    {
      "dynamics_loss_avg": 0.06159686055034399,
      "epoch": 2.285003662109375e-05,
      "step": 23960
    },
    {
      "action_loss_avg": 0.00220281999791041,
      "epoch": 2.285003662109375e-05,
      "step": 23960
    },
    {
      "epoch": 2.285003662109375e-05,
      "step": 23960,
      "training_step_time": 6.905488014221191
    },
    {
      "epoch": 2.2850990295410155e-05,
      "model_forward_time": 1.6829838752746582,
      "step": 23961
    },
    {
      "epoch": 2.2850990295410155e-05,
      "step": 23961,
      "training_step_time": 6.935909986495972
    },
    {
      "epoch": 2.2851943969726562e-05,
      "model_forward_time": 1.6848211288452148,
      "step": 23962
    },
    {
      "epoch": 2.2851943969726562e-05,
      "step": 23962,
      "training_step_time": 7.267305374145508
    },
    {
      "epoch": 2.285289764404297e-05,
      "model_forward_time": 1.6847124099731445,
      "step": 23963
    },
    {
      "epoch": 2.285289764404297e-05,
      "step": 23963,
      "training_step_time": 7.669025659561157
    },
    {
      "epoch": 2.2853851318359374e-05,
      "model_forward_time": 1.6862397193908691,
      "step": 23964
    },
    {
      "epoch": 2.2853851318359374e-05,
      "step": 23964,
      "training_step_time": 8.670140743255615
    },
    {
      "epoch": 2.285480499267578e-05,
      "model_forward_time": 1.7030982971191406,
      "step": 23965
    },
    {
      "epoch": 2.285480499267578e-05,
      "step": 23965,
      "training_step_time": 8.403883457183838
    },
    {
      "epoch": 2.285575866699219e-05,
      "model_forward_time": 2.230313301086426,
      "step": 23966
    },
    {
      "epoch": 2.285575866699219e-05,
      "step": 23966,
      "training_step_time": 7.957462787628174
    },
    {
      "epoch": 2.2856712341308593e-05,
      "model_forward_time": 1.6896448135375977,
      "step": 23967
    },
    {
      "epoch": 2.2856712341308593e-05,
      "step": 23967,
      "training_step_time": 7.68621039390564
    },
    {
      "epoch": 2.2857666015625e-05,
      "model_forward_time": 1.6821022033691406,
      "step": 23968
    },
    {
      "epoch": 2.2857666015625e-05,
      "step": 23968,
      "training_step_time": 7.817879915237427
    },
    {
      "epoch": 2.2858619689941407e-05,
      "model_forward_time": 1.6847155094146729,
      "step": 23969
    },
    {
      "epoch": 2.2858619689941407e-05,
      "step": 23969,
      "training_step_time": 7.0007898807525635
    },
    {
      "epoch": 2.285957336425781e-05,
      "grad_norm": 0.17068061232566833,
      "learning_rate": 9.048095689825415e-06,
      "loss": 0.0657,
      "step": 23970
    },
    {
      "epoch": 2.285957336425781e-05,
      "model_forward_time": 1.6834466457366943,
      "step": 23970
    },
    {
      "dynamics_loss_avg": 0.05952016767114401,
      "epoch": 2.285957336425781e-05,
      "step": 23970
    },
    {
      "action_loss_avg": 0.002752334880642593,
      "epoch": 2.285957336425781e-05,
      "step": 23970
    },
    {
      "epoch": 2.285957336425781e-05,
      "step": 23970,
      "training_step_time": 6.941744804382324
    },
    {
      "epoch": 2.286052703857422e-05,
      "model_forward_time": 1.6832704544067383,
      "step": 23971
    },
    {
      "epoch": 2.286052703857422e-05,
      "step": 23971,
      "training_step_time": 7.3811705112457275
    },
    {
      "epoch": 2.2861480712890626e-05,
      "model_forward_time": 1.684577465057373,
      "step": 23972
    },
    {
      "epoch": 2.2861480712890626e-05,
      "step": 23972,
      "training_step_time": 7.71149468421936
    },
    {
      "epoch": 2.286243438720703e-05,
      "model_forward_time": 1.6847515106201172,
      "step": 23973
    },
    {
      "epoch": 2.286243438720703e-05,
      "step": 23973,
      "training_step_time": 7.821508884429932
    },
    {
      "epoch": 2.2863388061523438e-05,
      "model_forward_time": 1.7024056911468506,
      "step": 23974
    },
    {
      "epoch": 2.2863388061523438e-05,
      "step": 23974,
      "training_step_time": 8.489752531051636
    },
    {
      "epoch": 2.2864341735839845e-05,
      "model_forward_time": 2.229121685028076,
      "step": 23975
    },
    {
      "epoch": 2.2864341735839845e-05,
      "step": 23975,
      "training_step_time": 11.545182466506958
    },
    {
      "epoch": 2.286529541015625e-05,
      "model_forward_time": 1.683239459991455,
      "step": 23976
    },
    {
      "epoch": 2.286529541015625e-05,
      "step": 23976,
      "training_step_time": 7.789703130722046
    },
    {
      "epoch": 2.2866249084472657e-05,
      "model_forward_time": 1.683579921722412,
      "step": 23977
    },
    {
      "epoch": 2.2866249084472657e-05,
      "step": 23977,
      "training_step_time": 7.852375745773315
    },
    {
      "epoch": 2.2867202758789064e-05,
      "model_forward_time": 1.6846504211425781,
      "step": 23978
    },
    {
      "epoch": 2.2867202758789064e-05,
      "step": 23978,
      "training_step_time": 7.833571195602417
    },
    {
      "epoch": 2.2868156433105468e-05,
      "model_forward_time": 1.6793127059936523,
      "step": 23979
    },
    {
      "epoch": 2.2868156433105468e-05,
      "step": 23979,
      "training_step_time": 7.0414674282073975
    },
    {
      "epoch": 2.2869110107421875e-05,
      "grad_norm": 0.18726998567581177,
      "learning_rate": 9.047124955162472e-06,
      "loss": 0.0703,
      "step": 23980
    },
    {
      "epoch": 2.2869110107421875e-05,
      "model_forward_time": 1.6797728538513184,
      "step": 23980
    },
    {
      "dynamics_loss_avg": 0.057423264533281324,
      "epoch": 2.2869110107421875e-05,
      "step": 23980
    },
    {
      "action_loss_avg": 0.0017593229946214705,
      "epoch": 2.2869110107421875e-05,
      "step": 23980
    },
    {
      "epoch": 2.2869110107421875e-05,
      "step": 23980,
      "training_step_time": 7.2184624671936035
    },
    {
      "epoch": 2.2870063781738283e-05,
      "model_forward_time": 1.6894416809082031,
      "step": 23981
    },
    {
      "epoch": 2.2870063781738283e-05,
      "step": 23981,
      "training_step_time": 7.689223289489746
    },
    {
      "epoch": 2.2871017456054687e-05,
      "model_forward_time": 1.6823606491088867,
      "step": 23982
    },
    {
      "epoch": 2.2871017456054687e-05,
      "step": 23982,
      "training_step_time": 8.333361387252808
    },
    {
      "epoch": 2.2871971130371094e-05,
      "model_forward_time": 1.6939525604248047,
      "step": 23983
    },
    {
      "epoch": 2.2871971130371094e-05,
      "step": 23983,
      "training_step_time": 8.762027502059937
    },
    {
      "epoch": 2.28729248046875e-05,
      "model_forward_time": 2.261929988861084,
      "step": 23984
    },
    {
      "epoch": 2.28729248046875e-05,
      "step": 23984,
      "training_step_time": 9.452646017074585
    },
    {
      "epoch": 2.2873878479003906e-05,
      "model_forward_time": 1.6796789169311523,
      "step": 23985
    },
    {
      "epoch": 2.2873878479003906e-05,
      "step": 23985,
      "training_step_time": 8.333279848098755
    },
    {
      "epoch": 2.2874832153320313e-05,
      "model_forward_time": 1.6850004196166992,
      "step": 23986
    },
    {
      "epoch": 2.2874832153320313e-05,
      "step": 23986,
      "training_step_time": 8.585267782211304
    },
    {
      "epoch": 2.2875785827636717e-05,
      "model_forward_time": 1.308983325958252,
      "step": 23987
    },
    {
      "epoch": 2.2875785827636717e-05,
      "step": 23987,
      "training_step_time": 7.826778888702393
    },
    {
      "epoch": 2.2876739501953125e-05,
      "model_forward_time": 1.6852376461029053,
      "step": 23988
    },
    {
      "epoch": 2.2876739501953125e-05,
      "step": 23988,
      "training_step_time": 7.128340721130371
    },
    {
      "epoch": 2.2877693176269532e-05,
      "model_forward_time": 1.6866405010223389,
      "step": 23989
    },
    {
      "epoch": 2.2877693176269532e-05,
      "step": 23989,
      "training_step_time": 7.136099338531494
    },
    {
      "epoch": 2.2878646850585936e-05,
      "grad_norm": 0.14845837652683258,
      "learning_rate": 9.04615377791203e-06,
      "loss": 0.0632,
      "step": 23990
    },
    {
      "epoch": 2.2878646850585936e-05,
      "model_forward_time": 1.679945468902588,
      "step": 23990
    },
    {
      "dynamics_loss_avg": 0.047688013315200804,
      "epoch": 2.2878646850585936e-05,
      "step": 23990
    },
    {
      "action_loss_avg": 0.0022850509791169315,
      "epoch": 2.2878646850585936e-05,
      "step": 23990
    },
    {
      "epoch": 2.2878646850585936e-05,
      "step": 23990,
      "training_step_time": 8.667466640472412
    },
    {
      "epoch": 2.2879600524902343e-05,
      "model_forward_time": 1.689276933670044,
      "step": 23991
    },
    {
      "epoch": 2.2879600524902343e-05,
      "step": 23991,
      "training_step_time": 8.123172760009766
    },
    {
      "epoch": 2.288055419921875e-05,
      "model_forward_time": 1.6860978603363037,
      "step": 23992
    },
    {
      "epoch": 2.288055419921875e-05,
      "step": 23992,
      "training_step_time": 7.936918020248413
    },
    {
      "epoch": 2.2881507873535155e-05,
      "model_forward_time": 1.6906802654266357,
      "step": 23993
    },
    {
      "epoch": 2.2881507873535155e-05,
      "step": 23993,
      "training_step_time": 8.496302366256714
    },
    {
      "epoch": 2.2882461547851562e-05,
      "model_forward_time": 1.6814584732055664,
      "step": 23994
    },
    {
      "epoch": 2.2882461547851562e-05,
      "step": 23994,
      "training_step_time": 9.463075160980225
    },
    {
      "epoch": 2.288341522216797e-05,
      "model_forward_time": 1.6915209293365479,
      "step": 23995
    },
    {
      "epoch": 2.288341522216797e-05,
      "step": 23995,
      "training_step_time": 7.777276039123535
    },
    {
      "epoch": 2.2884368896484374e-05,
      "model_forward_time": 1.6799817085266113,
      "step": 23996
    },
    {
      "epoch": 2.2884368896484374e-05,
      "step": 23996,
      "training_step_time": 7.741900682449341
    },
    {
      "epoch": 2.288532257080078e-05,
      "model_forward_time": 1.7024171352386475,
      "step": 23997
    },
    {
      "epoch": 2.288532257080078e-05,
      "step": 23997,
      "training_step_time": 7.5554444789886475
    },
    {
      "epoch": 2.288627624511719e-05,
      "model_forward_time": 1.7026078701019287,
      "step": 23998
    },
    {
      "epoch": 2.288627624511719e-05,
      "step": 23998,
      "training_step_time": 7.187994718551636
    },
    {
      "epoch": 2.2887229919433593e-05,
      "model_forward_time": 1.7205145359039307,
      "step": 23999
    },
    {
      "epoch": 2.2887229919433593e-05,
      "step": 23999,
      "training_step_time": 7.782276630401611
    },
    {
      "epoch": 2.288818359375e-05,
      "grad_norm": 0.19388064742088318,
      "learning_rate": 9.045182158180293e-06,
      "loss": 0.0655,
      "step": 24000
    },
    {
      "epoch": 2.288818359375e-05,
      "model_forward_time": 1.793381690979004,
      "step": 24000
    },
    {
      "dynamics_loss_avg": 0.07017698287963867,
      "epoch": 2.288818359375e-05,
      "step": 24000
    },
    {
      "action_loss_avg": 0.003134751063771546,
      "epoch": 2.288818359375e-05,
      "step": 24000
    },
    {
      "epoch": 2.288818359375e-05,
      "step": 24000,
      "training_step_time": 7.972697734832764
    },
    {
      "epoch": 2.2889137268066407e-05,
      "model_forward_time": 1.7684404850006104,
      "step": 24001
    },
    {
      "epoch": 2.2889137268066407e-05,
      "step": 24001,
      "training_step_time": 9.58673357963562
    },
    {
      "epoch": 2.289009094238281e-05,
      "model_forward_time": 3.421433687210083,
      "step": 24002
    },
    {
      "epoch": 2.289009094238281e-05,
      "step": 24002,
      "training_step_time": 10.45565676689148
    },
    {
      "epoch": 2.289104461669922e-05,
      "model_forward_time": 1.7555413246154785,
      "step": 24003
    },
    {
      "epoch": 2.289104461669922e-05,
      "step": 24003,
      "training_step_time": 11.023760080337524
    },
    {
      "epoch": 2.2891998291015626e-05,
      "model_forward_time": 1.7578985691070557,
      "step": 24004
    },
    {
      "epoch": 2.2891998291015626e-05,
      "step": 24004,
      "training_step_time": 8.635738372802734
    },
    {
      "epoch": 2.289295196533203e-05,
      "model_forward_time": 1.7634072303771973,
      "step": 24005
    },
    {
      "epoch": 2.289295196533203e-05,
      "step": 24005,
      "training_step_time": 7.712889194488525
    },
    {
      "epoch": 2.2893905639648438e-05,
      "model_forward_time": 1.7548394203186035,
      "step": 24006
    },
    {
      "epoch": 2.2893905639648438e-05,
      "step": 24006,
      "training_step_time": 7.925585985183716
    },
    {
      "epoch": 2.2894859313964845e-05,
      "model_forward_time": 1.759833574295044,
      "step": 24007
    },
    {
      "epoch": 2.2894859313964845e-05,
      "step": 24007,
      "training_step_time": 6.922904968261719
    },
    {
      "epoch": 2.289581298828125e-05,
      "model_forward_time": 1.7537930011749268,
      "step": 24008
    },
    {
      "epoch": 2.289581298828125e-05,
      "step": 24008,
      "training_step_time": 7.761029243469238
    },
    {
      "epoch": 2.2896766662597657e-05,
      "model_forward_time": 1.7446918487548828,
      "step": 24009
    },
    {
      "epoch": 2.2896766662597657e-05,
      "step": 24009,
      "training_step_time": 8.640632390975952
    },
    {
      "epoch": 2.2897720336914064e-05,
      "grad_norm": 0.18989357352256775,
      "learning_rate": 9.044210096073517e-06,
      "loss": 0.0657,
      "step": 24010
    },
    {
      "epoch": 2.2897720336914064e-05,
      "model_forward_time": 1.7549607753753662,
      "step": 24010
    },
    {
      "dynamics_loss_avg": 0.05556350387632847,
      "epoch": 2.2897720336914064e-05,
      "step": 24010
    },
    {
      "action_loss_avg": 0.002586662844987586,
      "epoch": 2.2897720336914064e-05,
      "step": 24010
    },
    {
      "epoch": 2.2897720336914064e-05,
      "step": 24010,
      "training_step_time": 8.769778966903687
    },
    {
      "epoch": 2.2898674011230468e-05,
      "model_forward_time": 2.9144463539123535,
      "step": 24011
    },
    {
      "epoch": 2.2898674011230468e-05,
      "step": 24011,
      "training_step_time": 10.832986116409302
    },
    {
      "epoch": 2.2899627685546875e-05,
      "model_forward_time": 1.7584011554718018,
      "step": 24012
    },
    {
      "epoch": 2.2899627685546875e-05,
      "step": 24012,
      "training_step_time": 9.132790327072144
    },
    {
      "epoch": 2.2900581359863283e-05,
      "model_forward_time": 1.7513129711151123,
      "step": 24013
    },
    {
      "epoch": 2.2900581359863283e-05,
      "step": 24013,
      "training_step_time": 7.839317798614502
    },
    {
      "epoch": 2.2901535034179687e-05,
      "model_forward_time": 1.7633624076843262,
      "step": 24014
    },
    {
      "epoch": 2.2901535034179687e-05,
      "step": 24014,
      "training_step_time": 7.676285266876221
    },
    {
      "epoch": 2.2902488708496094e-05,
      "model_forward_time": 1.7676045894622803,
      "step": 24015
    },
    {
      "epoch": 2.2902488708496094e-05,
      "step": 24015,
      "training_step_time": 7.85286283493042
    },
    {
      "epoch": 2.29034423828125e-05,
      "model_forward_time": 1.7598800659179688,
      "step": 24016
    },
    {
      "epoch": 2.29034423828125e-05,
      "step": 24016,
      "training_step_time": 6.9315595626831055
    },
    {
      "epoch": 2.2904396057128906e-05,
      "model_forward_time": 1.754988670349121,
      "step": 24017
    },
    {
      "epoch": 2.2904396057128906e-05,
      "step": 24017,
      "training_step_time": 7.763675689697266
    },
    {
      "epoch": 2.2905349731445313e-05,
      "model_forward_time": 1.759248971939087,
      "step": 24018
    },
    {
      "epoch": 2.2905349731445313e-05,
      "step": 24018,
      "training_step_time": 7.610130071640015
    },
    {
      "epoch": 2.2906303405761717e-05,
      "model_forward_time": 1.7560789585113525,
      "step": 24019
    },
    {
      "epoch": 2.2906303405761717e-05,
      "step": 24019,
      "training_step_time": 8.55604887008667
    },
    {
      "epoch": 2.2907257080078125e-05,
      "grad_norm": 0.23360107839107513,
      "learning_rate": 9.043237591698006e-06,
      "loss": 0.066,
      "step": 24020
    },
    {
      "epoch": 2.2907257080078125e-05,
      "model_forward_time": 1.7511873245239258,
      "step": 24020
    },
    {
      "dynamics_loss_avg": 0.06422387678176164,
      "epoch": 2.2907257080078125e-05,
      "step": 24020
    },
    {
      "action_loss_avg": 0.0027252429514192044,
      "epoch": 2.2907257080078125e-05,
      "step": 24020
    },
    {
      "epoch": 2.2907257080078125e-05,
      "step": 24020,
      "training_step_time": 9.350786447525024
    },
    {
      "epoch": 2.2908210754394532e-05,
      "model_forward_time": 1.7511522769927979,
      "step": 24021
    },
    {
      "epoch": 2.2908210754394532e-05,
      "step": 24021,
      "training_step_time": 9.005702495574951
    },
    {
      "epoch": 2.2909164428710936e-05,
      "model_forward_time": 1.7632789611816406,
      "step": 24022
    },
    {
      "epoch": 2.2909164428710936e-05,
      "step": 24022,
      "training_step_time": 8.352405071258545
    },
    {
      "epoch": 2.2910118103027343e-05,
      "model_forward_time": 1.7547414302825928,
      "step": 24023
    },
    {
      "epoch": 2.2910118103027343e-05,
      "step": 24023,
      "training_step_time": 7.778999090194702
    },
    {
      "epoch": 2.291107177734375e-05,
      "model_forward_time": 1.7527356147766113,
      "step": 24024
    },
    {
      "epoch": 2.291107177734375e-05,
      "step": 24024,
      "training_step_time": 7.836429119110107
    },
    {
      "epoch": 2.2912025451660155e-05,
      "model_forward_time": 1.7542080879211426,
      "step": 24025
    },
    {
      "epoch": 2.2912025451660155e-05,
      "step": 24025,
      "training_step_time": 6.96994161605835
    },
    {
      "epoch": 2.2912979125976562e-05,
      "model_forward_time": 1.7619941234588623,
      "step": 24026
    },
    {
      "epoch": 2.2912979125976562e-05,
      "step": 24026,
      "training_step_time": 7.742720127105713
    },
    {
      "epoch": 2.291393280029297e-05,
      "model_forward_time": 1.7520687580108643,
      "step": 24027
    },
    {
      "epoch": 2.291393280029297e-05,
      "step": 24027,
      "training_step_time": 7.577683687210083
    },
    {
      "epoch": 2.2914886474609374e-05,
      "model_forward_time": 1.7649552822113037,
      "step": 24028
    },
    {
      "epoch": 2.2914886474609374e-05,
      "step": 24028,
      "training_step_time": 7.841689825057983
    },
    {
      "epoch": 2.291584014892578e-05,
      "model_forward_time": 1.7589595317840576,
      "step": 24029
    },
    {
      "epoch": 2.291584014892578e-05,
      "step": 24029,
      "training_step_time": 9.658473014831543
    },
    {
      "epoch": 2.291679382324219e-05,
      "grad_norm": 0.17498207092285156,
      "learning_rate": 9.04226464516011e-06,
      "loss": 0.0656,
      "step": 24030
    },
    {
      "epoch": 2.291679382324219e-05,
      "model_forward_time": 2.9179461002349854,
      "step": 24030
    },
    {
      "dynamics_loss_avg": 0.04759016316384077,
      "epoch": 2.291679382324219e-05,
      "step": 24030
    },
    {
      "action_loss_avg": 0.0029783590696752073,
      "epoch": 2.291679382324219e-05,
      "step": 24030
    },
    {
      "epoch": 2.291679382324219e-05,
      "step": 24030,
      "training_step_time": 8.63824987411499
    },
    {
      "epoch": 2.2917747497558593e-05,
      "model_forward_time": 1.7547950744628906,
      "step": 24031
    },
    {
      "epoch": 2.2917747497558593e-05,
      "step": 24031,
      "training_step_time": 8.583239793777466
    },
    {
      "epoch": 2.2918701171875e-05,
      "model_forward_time": 1.758763313293457,
      "step": 24032
    },
    {
      "epoch": 2.2918701171875e-05,
      "step": 24032,
      "training_step_time": 7.8351662158966064
    },
    {
      "epoch": 2.2919654846191407e-05,
      "model_forward_time": 1.7499301433563232,
      "step": 24033
    },
    {
      "epoch": 2.2919654846191407e-05,
      "step": 24033,
      "training_step_time": 7.756531715393066
    },
    {
      "epoch": 2.292060852050781e-05,
      "model_forward_time": 1.7558631896972656,
      "step": 24034
    },
    {
      "epoch": 2.292060852050781e-05,
      "step": 24034,
      "training_step_time": 7.943188190460205
    },
    {
      "epoch": 2.292156219482422e-05,
      "model_forward_time": 1.7657406330108643,
      "step": 24035
    },
    {
      "epoch": 2.292156219482422e-05,
      "step": 24035,
      "training_step_time": 7.468707799911499
    },
    {
      "epoch": 2.2922515869140626e-05,
      "model_forward_time": 1.764779806137085,
      "step": 24036
    },
    {
      "epoch": 2.2922515869140626e-05,
      "step": 24036,
      "training_step_time": 7.804807186126709
    },
    {
      "epoch": 2.292346954345703e-05,
      "model_forward_time": 1.7528245449066162,
      "step": 24037
    },
    {
      "epoch": 2.292346954345703e-05,
      "step": 24037,
      "training_step_time": 7.857170104980469
    },
    {
      "epoch": 2.2924423217773438e-05,
      "model_forward_time": 1.7655811309814453,
      "step": 24038
    },
    {
      "epoch": 2.2924423217773438e-05,
      "step": 24038,
      "training_step_time": 8.569079399108887
    },
    {
      "epoch": 2.2925376892089845e-05,
      "model_forward_time": 2.8966712951660156,
      "step": 24039
    },
    {
      "epoch": 2.2925376892089845e-05,
      "step": 24039,
      "training_step_time": 8.761404991149902
    },
    {
      "epoch": 2.292633056640625e-05,
      "grad_norm": 0.251882404088974,
      "learning_rate": 9.041291256566229e-06,
      "loss": 0.0661,
      "step": 24040
    },
    {
      "epoch": 2.292633056640625e-05,
      "model_forward_time": 1.763923168182373,
      "step": 24040
    },
    {
      "dynamics_loss_avg": 0.06697846418246627,
      "epoch": 2.292633056640625e-05,
      "step": 24040
    },
    {
      "action_loss_avg": 0.004072679404634983,
      "epoch": 2.292633056640625e-05,
      "step": 24040
    },
    {
      "epoch": 2.292633056640625e-05,
      "step": 24040,
      "training_step_time": 8.988993644714355
    },
    {
      "epoch": 2.2927284240722657e-05,
      "model_forward_time": 1.775939702987671,
      "step": 24041
    },
    {
      "epoch": 2.2927284240722657e-05,
      "step": 24041,
      "training_step_time": 7.887922525405884
    },
    {
      "epoch": 2.2928237915039064e-05,
      "model_forward_time": 1.7570676803588867,
      "step": 24042
    },
    {
      "epoch": 2.2928237915039064e-05,
      "step": 24042,
      "training_step_time": 7.758168935775757
    },
    {
      "epoch": 2.2929191589355468e-05,
      "model_forward_time": 1.762462854385376,
      "step": 24043
    },
    {
      "epoch": 2.2929191589355468e-05,
      "step": 24043,
      "training_step_time": 7.5300986766815186
    },
    {
      "epoch": 2.2930145263671875e-05,
      "model_forward_time": 1.7511491775512695,
      "step": 24044
    },
    {
      "epoch": 2.2930145263671875e-05,
      "step": 24044,
      "training_step_time": 7.484510183334351
    },
    {
      "epoch": 2.2931098937988283e-05,
      "model_forward_time": 1.7640457153320312,
      "step": 24045
    },
    {
      "epoch": 2.2931098937988283e-05,
      "step": 24045,
      "training_step_time": 7.856940031051636
    },
    {
      "epoch": 2.2932052612304687e-05,
      "model_forward_time": 1.7743022441864014,
      "step": 24046
    },
    {
      "epoch": 2.2932052612304687e-05,
      "step": 24046,
      "training_step_time": 8.370043516159058
    },
    {
      "epoch": 2.2933006286621094e-05,
      "model_forward_time": 1.757309913635254,
      "step": 24047
    },
    {
      "epoch": 2.2933006286621094e-05,
      "step": 24047,
      "training_step_time": 7.909186363220215
    },
    {
      "epoch": 2.29339599609375e-05,
      "model_forward_time": 2.9953720569610596,
      "step": 24048
    },
    {
      "epoch": 2.29339599609375e-05,
      "step": 24048,
      "training_step_time": 8.80394721031189
    },
    {
      "epoch": 2.2934913635253906e-05,
      "model_forward_time": 1.7467784881591797,
      "step": 24049
    },
    {
      "epoch": 2.2934913635253906e-05,
      "step": 24049,
      "training_step_time": 7.816229581832886
    },
    {
      "epoch": 2.2935867309570313e-05,
      "grad_norm": 0.17421476542949677,
      "learning_rate": 9.040317426022814e-06,
      "loss": 0.0652,
      "step": 24050
    },
    {
      "epoch": 2.2935867309570313e-05,
      "model_forward_time": 1.7238454818725586,
      "step": 24050
    },
    {
      "dynamics_loss_avg": 0.06361201535910369,
      "epoch": 2.2935867309570313e-05,
      "step": 24050
    },
    {
      "action_loss_avg": 0.0030782932415604593,
      "epoch": 2.2935867309570313e-05,
      "step": 24050
    },
    {
      "epoch": 2.2935867309570313e-05,
      "step": 24050,
      "training_step_time": 7.84685206413269
    },
    {
      "epoch": 2.2936820983886717e-05,
      "model_forward_time": 1.7507920265197754,
      "step": 24051
    },
    {
      "epoch": 2.2936820983886717e-05,
      "step": 24051,
      "training_step_time": 7.741147756576538
    },
    {
      "epoch": 2.2937774658203125e-05,
      "model_forward_time": 1.7643375396728516,
      "step": 24052
    },
    {
      "epoch": 2.2937774658203125e-05,
      "step": 24052,
      "training_step_time": 7.915451288223267
    },
    {
      "epoch": 2.2938728332519532e-05,
      "model_forward_time": 1.7402186393737793,
      "step": 24053
    },
    {
      "epoch": 2.2938728332519532e-05,
      "step": 24053,
      "training_step_time": 7.014176368713379
    },
    {
      "epoch": 2.2939682006835936e-05,
      "model_forward_time": 1.7512164115905762,
      "step": 24054
    },
    {
      "epoch": 2.2939682006835936e-05,
      "step": 24054,
      "training_step_time": 7.75702691078186
    },
    {
      "epoch": 2.2940635681152343e-05,
      "model_forward_time": 1.7467973232269287,
      "step": 24055
    },
    {
      "epoch": 2.2940635681152343e-05,
      "step": 24055,
      "training_step_time": 8.44959044456482
    },
    {
      "epoch": 2.294158935546875e-05,
      "model_forward_time": 1.7494275569915771,
      "step": 24056
    },
    {
      "epoch": 2.294158935546875e-05,
      "step": 24056,
      "training_step_time": 9.585906028747559
    },
    {
      "epoch": 2.2942543029785155e-05,
      "model_forward_time": 2.8674187660217285,
      "step": 24057
    },
    {
      "epoch": 2.2942543029785155e-05,
      "step": 24057,
      "training_step_time": 9.356622457504272
    },
    {
      "epoch": 2.2943496704101562e-05,
      "model_forward_time": 1.8920791149139404,
      "step": 24058
    },
    {
      "epoch": 2.2943496704101562e-05,
      "step": 24058,
      "training_step_time": 7.826704978942871
    },
    {
      "epoch": 2.294445037841797e-05,
      "model_forward_time": 1.400982141494751,
      "step": 24059
    },
    {
      "epoch": 2.294445037841797e-05,
      "step": 24059,
      "training_step_time": 7.859400510787964
    },
    {
      "epoch": 2.2945404052734374e-05,
      "grad_norm": 0.20099444687366486,
      "learning_rate": 9.03934315363636e-06,
      "loss": 0.0664,
      "step": 24060
    },
    {
      "epoch": 2.2945404052734374e-05,
      "model_forward_time": 1.7311959266662598,
      "step": 24060
    },
    {
      "dynamics_loss_avg": 0.06588717475533486,
      "epoch": 2.2945404052734374e-05,
      "step": 24060
    },
    {
      "action_loss_avg": 0.002720988041255623,
      "epoch": 2.2945404052734374e-05,
      "step": 24060
    },
    {
      "epoch": 2.2945404052734374e-05,
      "step": 24060,
      "training_step_time": 7.848126173019409
    },
    {
      "epoch": 2.294635772705078e-05,
      "model_forward_time": 1.7776687145233154,
      "step": 24061
    },
    {
      "epoch": 2.294635772705078e-05,
      "step": 24061,
      "training_step_time": 8.010343313217163
    },
    {
      "epoch": 2.294731140136719e-05,
      "model_forward_time": 1.0510118007659912,
      "step": 24062
    },
    {
      "epoch": 2.294731140136719e-05,
      "step": 24062,
      "training_step_time": 7.103802442550659
    },
    {
      "epoch": 2.2948265075683593e-05,
      "model_forward_time": 1.7638542652130127,
      "step": 24063
    },
    {
      "epoch": 2.2948265075683593e-05,
      "step": 24063,
      "training_step_time": 8.46025800704956
    },
    {
      "epoch": 2.294921875e-05,
      "model_forward_time": 1.7672970294952393,
      "step": 24064
    },
    {
      "epoch": 2.294921875e-05,
      "step": 24064,
      "training_step_time": 7.855697393417358
    },
    {
      "epoch": 2.2950172424316407e-05,
      "model_forward_time": 1.7544033527374268,
      "step": 24065
    },
    {
      "epoch": 2.2950172424316407e-05,
      "step": 24065,
      "training_step_time": 9.146766424179077
    },
    {
      "epoch": 2.295112609863281e-05,
      "model_forward_time": 1.7692525386810303,
      "step": 24066
    },
    {
      "epoch": 2.295112609863281e-05,
      "step": 24066,
      "training_step_time": 8.97760534286499
    },
    {
      "epoch": 2.295207977294922e-05,
      "model_forward_time": 2.9269649982452393,
      "step": 24067
    },
    {
      "epoch": 2.295207977294922e-05,
      "step": 24067,
      "training_step_time": 8.024762153625488
    },
    {
      "epoch": 2.2953033447265626e-05,
      "model_forward_time": 1.7583727836608887,
      "step": 24068
    },
    {
      "epoch": 2.2953033447265626e-05,
      "step": 24068,
      "training_step_time": 7.929356098175049
    },
    {
      "epoch": 2.295398712158203e-05,
      "model_forward_time": 1.7509589195251465,
      "step": 24069
    },
    {
      "epoch": 2.295398712158203e-05,
      "step": 24069,
      "training_step_time": 7.761576414108276
    },
    {
      "epoch": 2.2954940795898438e-05,
      "grad_norm": 0.21462300419807434,
      "learning_rate": 9.03836843951341e-06,
      "loss": 0.0669,
      "step": 24070
    },
    {
      "epoch": 2.2954940795898438e-05,
      "model_forward_time": 1.7397713661193848,
      "step": 24070
    },
    {
      "dynamics_loss_avg": 0.062011612392961976,
      "epoch": 2.2954940795898438e-05,
      "step": 24070
    },
    {
      "action_loss_avg": 0.002673623833106831,
      "epoch": 2.2954940795898438e-05,
      "step": 24070
    },
    {
      "epoch": 2.2954940795898438e-05,
      "step": 24070,
      "training_step_time": 7.876824855804443
    },
    {
      "epoch": 2.2955894470214845e-05,
      "model_forward_time": 1.749084711074829,
      "step": 24071
    },
    {
      "epoch": 2.2955894470214845e-05,
      "step": 24071,
      "training_step_time": 7.953442573547363
    },
    {
      "epoch": 2.295684814453125e-05,
      "model_forward_time": 1.762929916381836,
      "step": 24072
    },
    {
      "epoch": 2.295684814453125e-05,
      "step": 24072,
      "training_step_time": 7.807870626449585
    },
    {
      "epoch": 2.2957801818847657e-05,
      "model_forward_time": 1.754349708557129,
      "step": 24073
    },
    {
      "epoch": 2.2957801818847657e-05,
      "step": 24073,
      "training_step_time": 7.690088272094727
    },
    {
      "epoch": 2.2958755493164064e-05,
      "model_forward_time": 1.7777881622314453,
      "step": 24074
    },
    {
      "epoch": 2.2958755493164064e-05,
      "step": 24074,
      "training_step_time": 7.8814942836761475
    },
    {
      "epoch": 2.2959709167480468e-05,
      "model_forward_time": 1.7571094036102295,
      "step": 24075
    },
    {
      "epoch": 2.2959709167480468e-05,
      "step": 24075,
      "training_step_time": 8.421844482421875
    },
    {
      "epoch": 2.2960662841796875e-05,
      "model_forward_time": 2.928173303604126,
      "step": 24076
    },
    {
      "epoch": 2.2960662841796875e-05,
      "step": 24076,
      "training_step_time": 8.966204404830933
    },
    {
      "epoch": 2.2961616516113283e-05,
      "model_forward_time": 1.7407212257385254,
      "step": 24077
    },
    {
      "epoch": 2.2961616516113283e-05,
      "step": 24077,
      "training_step_time": 7.864778518676758
    },
    {
      "epoch": 2.2962570190429687e-05,
      "model_forward_time": 1.7531013488769531,
      "step": 24078
    },
    {
      "epoch": 2.2962570190429687e-05,
      "step": 24078,
      "training_step_time": 8.695834636688232
    },
    {
      "epoch": 2.2963523864746094e-05,
      "model_forward_time": 1.759674310684204,
      "step": 24079
    },
    {
      "epoch": 2.2963523864746094e-05,
      "step": 24079,
      "training_step_time": 7.76241660118103
    },
    {
      "epoch": 2.29644775390625e-05,
      "grad_norm": 0.1275862157344818,
      "learning_rate": 9.03739328376056e-06,
      "loss": 0.065,
      "step": 24080
    },
    {
      "epoch": 2.29644775390625e-05,
      "model_forward_time": 1.774641752243042,
      "step": 24080
    },
    {
      "dynamics_loss_avg": 0.06907125748693943,
      "epoch": 2.29644775390625e-05,
      "step": 24080
    },
    {
      "action_loss_avg": 0.00403159277047962,
      "epoch": 2.29644775390625e-05,
      "step": 24080
    },
    {
      "epoch": 2.29644775390625e-05,
      "step": 24080,
      "training_step_time": 7.953461647033691
    },
    {
      "epoch": 2.2965431213378906e-05,
      "model_forward_time": 1.7604811191558838,
      "step": 24081
    },
    {
      "epoch": 2.2965431213378906e-05,
      "step": 24081,
      "training_step_time": 7.804166078567505
    },
    {
      "epoch": 2.2966384887695313e-05,
      "model_forward_time": 1.7606980800628662,
      "step": 24082
    },
    {
      "epoch": 2.2966384887695313e-05,
      "step": 24082,
      "training_step_time": 7.6779303550720215
    },
    {
      "epoch": 2.2967338562011717e-05,
      "model_forward_time": 1.7553579807281494,
      "step": 24083
    },
    {
      "epoch": 2.2967338562011717e-05,
      "step": 24083,
      "training_step_time": 8.550217151641846
    },
    {
      "epoch": 2.2968292236328125e-05,
      "model_forward_time": 1.7610175609588623,
      "step": 24084
    },
    {
      "epoch": 2.2968292236328125e-05,
      "step": 24084,
      "training_step_time": 8.555450201034546
    },
    {
      "epoch": 2.2969245910644532e-05,
      "model_forward_time": 2.9369845390319824,
      "step": 24085
    },
    {
      "epoch": 2.2969245910644532e-05,
      "step": 24085,
      "training_step_time": 8.665282011032104
    },
    {
      "epoch": 2.2970199584960936e-05,
      "model_forward_time": 1.7821743488311768,
      "step": 24086
    },
    {
      "epoch": 2.2970199584960936e-05,
      "step": 24086,
      "training_step_time": 7.885920524597168
    },
    {
      "epoch": 2.2971153259277344e-05,
      "model_forward_time": 1.7580411434173584,
      "step": 24087
    },
    {
      "epoch": 2.2971153259277344e-05,
      "step": 24087,
      "training_step_time": 7.635622978210449
    },
    {
      "epoch": 2.297210693359375e-05,
      "model_forward_time": 1.766404628753662,
      "step": 24088
    },
    {
      "epoch": 2.297210693359375e-05,
      "step": 24088,
      "training_step_time": 7.5904388427734375
    },
    {
      "epoch": 2.2973060607910155e-05,
      "model_forward_time": 1.7595243453979492,
      "step": 24089
    },
    {
      "epoch": 2.2973060607910155e-05,
      "step": 24089,
      "training_step_time": 7.976446628570557
    },
    {
      "epoch": 2.2974014282226562e-05,
      "grad_norm": 0.17329226434230804,
      "learning_rate": 9.03641768648445e-06,
      "loss": 0.0656,
      "step": 24090
    },
    {
      "epoch": 2.2974014282226562e-05,
      "model_forward_time": 1.7679123878479004,
      "step": 24090
    },
    {
      "dynamics_loss_avg": 0.07823829371482134,
      "epoch": 2.2974014282226562e-05,
      "step": 24090
    },
    {
      "action_loss_avg": 0.004409936827141791,
      "epoch": 2.2974014282226562e-05,
      "step": 24090
    },
    {
      "epoch": 2.2974014282226562e-05,
      "step": 24090,
      "training_step_time": 7.895867824554443
    },
    {
      "epoch": 2.297496795654297e-05,
      "model_forward_time": 1.7626838684082031,
      "step": 24091
    },
    {
      "epoch": 2.297496795654297e-05,
      "step": 24091,
      "training_step_time": 7.764442205429077
    },
    {
      "epoch": 2.2975921630859374e-05,
      "model_forward_time": 1.7580134868621826,
      "step": 24092
    },
    {
      "epoch": 2.2975921630859374e-05,
      "step": 24092,
      "training_step_time": 8.023614645004272
    },
    {
      "epoch": 2.297687530517578e-05,
      "model_forward_time": 1.7502689361572266,
      "step": 24093
    },
    {
      "epoch": 2.297687530517578e-05,
      "step": 24093,
      "training_step_time": 7.938107967376709
    },
    {
      "epoch": 2.297782897949219e-05,
      "model_forward_time": 2.857120990753174,
      "step": 24094
    },
    {
      "epoch": 2.297782897949219e-05,
      "step": 24094,
      "training_step_time": 10.48887324333191
    },
    {
      "epoch": 2.2978782653808593e-05,
      "model_forward_time": 1.8516440391540527,
      "step": 24095
    },
    {
      "epoch": 2.2978782653808593e-05,
      "step": 24095,
      "training_step_time": 8.916043996810913
    },
    {
      "epoch": 2.2979736328125e-05,
      "model_forward_time": 1.7702281475067139,
      "step": 24096
    },
    {
      "epoch": 2.2979736328125e-05,
      "step": 24096,
      "training_step_time": 8.975611686706543
    },
    {
      "epoch": 2.2980690002441407e-05,
      "model_forward_time": 1.7633419036865234,
      "step": 24097
    },
    {
      "epoch": 2.2980690002441407e-05,
      "step": 24097,
      "training_step_time": 7.795659303665161
    },
    {
      "epoch": 2.298164367675781e-05,
      "model_forward_time": 1.7606308460235596,
      "step": 24098
    },
    {
      "epoch": 2.298164367675781e-05,
      "step": 24098,
      "training_step_time": 7.650757789611816
    },
    {
      "epoch": 2.298259735107422e-05,
      "model_forward_time": 1.7629218101501465,
      "step": 24099
    },
    {
      "epoch": 2.298259735107422e-05,
      "step": 24099,
      "training_step_time": 8.58250617980957
    },
    {
      "epoch": 2.2983551025390626e-05,
      "grad_norm": 0.1676110476255417,
      "learning_rate": 9.035441647791772e-06,
      "loss": 0.0669,
      "step": 24100
    },
    {
      "epoch": 2.2983551025390626e-05,
      "model_forward_time": 1.7622249126434326,
      "step": 24100
    },
    {
      "dynamics_loss_avg": 0.07272866703569889,
      "epoch": 2.2983551025390626e-05,
      "step": 24100
    },
    {
      "action_loss_avg": 0.0028238211874850095,
      "epoch": 2.2983551025390626e-05,
      "step": 24100
    },
    {
      "epoch": 2.2983551025390626e-05,
      "step": 24100,
      "training_step_time": 7.367194890975952
    },
    {
      "epoch": 2.298450469970703e-05,
      "model_forward_time": 1.7596070766448975,
      "step": 24101
    },
    {
      "epoch": 2.298450469970703e-05,
      "step": 24101,
      "training_step_time": 8.608605146408081
    },
    {
      "epoch": 2.2985458374023438e-05,
      "model_forward_time": 1.759979248046875,
      "step": 24102
    },
    {
      "epoch": 2.2985458374023438e-05,
      "step": 24102,
      "training_step_time": 9.256473302841187
    },
    {
      "epoch": 2.2986412048339845e-05,
      "model_forward_time": 1.7641582489013672,
      "step": 24103
    },
    {
      "epoch": 2.2986412048339845e-05,
      "step": 24103,
      "training_step_time": 9.513776779174805
    },
    {
      "epoch": 2.298736572265625e-05,
      "model_forward_time": 2.9874486923217773,
      "step": 24104
    },
    {
      "epoch": 2.298736572265625e-05,
      "step": 24104,
      "training_step_time": 9.02400517463684
    },
    {
      "epoch": 2.2988319396972657e-05,
      "model_forward_time": 1.7683746814727783,
      "step": 24105
    },
    {
      "epoch": 2.2988319396972657e-05,
      "step": 24105,
      "training_step_time": 8.578256607055664
    },
    {
      "epoch": 2.2989273071289064e-05,
      "model_forward_time": 1.761918306350708,
      "step": 24106
    },
    {
      "epoch": 2.2989273071289064e-05,
      "step": 24106,
      "training_step_time": 7.828128099441528
    },
    {
      "epoch": 2.2990226745605468e-05,
      "model_forward_time": 1.7734148502349854,
      "step": 24107
    },
    {
      "epoch": 2.2990226745605468e-05,
      "step": 24107,
      "training_step_time": 7.816952466964722
    },
    {
      "epoch": 2.2991180419921876e-05,
      "model_forward_time": 1.7829468250274658,
      "step": 24108
    },
    {
      "epoch": 2.2991180419921876e-05,
      "step": 24108,
      "training_step_time": 7.975220441818237
    },
    {
      "epoch": 2.2992134094238283e-05,
      "model_forward_time": 1.7501909732818604,
      "step": 24109
    },
    {
      "epoch": 2.2992134094238283e-05,
      "step": 24109,
      "training_step_time": 7.397641658782959
    },
    {
      "epoch": 2.2993087768554687e-05,
      "grad_norm": 0.12161432951688766,
      "learning_rate": 9.034465167789262e-06,
      "loss": 0.0668,
      "step": 24110
    },
    {
      "epoch": 2.2993087768554687e-05,
      "model_forward_time": 1.7649624347686768,
      "step": 24110
    },
    {
      "dynamics_loss_avg": 0.07564546931535006,
      "epoch": 2.2993087768554687e-05,
      "step": 24110
    },
    {
      "action_loss_avg": 0.002586666273418814,
      "epoch": 2.2993087768554687e-05,
      "step": 24110
    },
    {
      "epoch": 2.2993087768554687e-05,
      "step": 24110,
      "training_step_time": 8.551249504089355
    },
    {
      "epoch": 2.2994041442871094e-05,
      "model_forward_time": 1.7524657249450684,
      "step": 24111
    },
    {
      "epoch": 2.2994041442871094e-05,
      "step": 24111,
      "training_step_time": 7.9683942794799805
    },
    {
      "epoch": 2.29949951171875e-05,
      "model_forward_time": 1.7488441467285156,
      "step": 24112
    },
    {
      "epoch": 2.29949951171875e-05,
      "step": 24112,
      "training_step_time": 8.071853637695312
    },
    {
      "epoch": 2.2995948791503906e-05,
      "model_forward_time": 2.97298526763916,
      "step": 24113
    },
    {
      "epoch": 2.2995948791503906e-05,
      "step": 24113,
      "training_step_time": 9.01004409790039
    },
    {
      "epoch": 2.2996902465820313e-05,
      "model_forward_time": 1.761472463607788,
      "step": 24114
    },
    {
      "epoch": 2.2996902465820313e-05,
      "step": 24114,
      "training_step_time": 8.832431077957153
    },
    {
      "epoch": 2.2997856140136717e-05,
      "model_forward_time": 1.7592453956604004,
      "step": 24115
    },
    {
      "epoch": 2.2997856140136717e-05,
      "step": 24115,
      "training_step_time": 7.808072328567505
    },
    {
      "epoch": 2.2998809814453125e-05,
      "model_forward_time": 1.7641010284423828,
      "step": 24116
    },
    {
      "epoch": 2.2998809814453125e-05,
      "step": 24116,
      "training_step_time": 8.513468503952026
    },
    {
      "epoch": 2.2999763488769532e-05,
      "model_forward_time": 1.7531394958496094,
      "step": 24117
    },
    {
      "epoch": 2.2999763488769532e-05,
      "step": 24117,
      "training_step_time": 7.907219409942627
    },
    {
      "epoch": 2.3000717163085936e-05,
      "model_forward_time": 1.7587547302246094,
      "step": 24118
    },
    {
      "epoch": 2.3000717163085936e-05,
      "step": 24118,
      "training_step_time": 7.4418559074401855
    },
    {
      "epoch": 2.3001670837402344e-05,
      "model_forward_time": 1.7620139122009277,
      "step": 24119
    },
    {
      "epoch": 2.3001670837402344e-05,
      "step": 24119,
      "training_step_time": 7.395655155181885
    },
    {
      "epoch": 2.300262451171875e-05,
      "grad_norm": 0.18386733531951904,
      "learning_rate": 9.033488246583706e-06,
      "loss": 0.0639,
      "step": 24120
    },
    {
      "epoch": 2.300262451171875e-05,
      "model_forward_time": 1.774705171585083,
      "step": 24120
    },
    {
      "dynamics_loss_avg": 0.06778514180332422,
      "epoch": 2.300262451171875e-05,
      "step": 24120
    },
    {
      "action_loss_avg": 0.0023699066718108953,
      "epoch": 2.300262451171875e-05,
      "step": 24120
    },
    {
      "epoch": 2.300262451171875e-05,
      "step": 24120,
      "training_step_time": 9.481185913085938
    },
    {
      "epoch": 2.3003578186035155e-05,
      "model_forward_time": 1.7414770126342773,
      "step": 24121
    },
    {
      "epoch": 2.3003578186035155e-05,
      "step": 24121,
      "training_step_time": 10.161965370178223
    },
    {
      "epoch": 2.3004531860351562e-05,
      "model_forward_time": 2.9998371601104736,
      "step": 24122
    },
    {
      "epoch": 2.3004531860351562e-05,
      "step": 24122,
      "training_step_time": 7.9431164264678955
    },
    {
      "epoch": 2.300548553466797e-05,
      "model_forward_time": 1.7604076862335205,
      "step": 24123
    },
    {
      "epoch": 2.300548553466797e-05,
      "step": 24123,
      "training_step_time": 8.004301071166992
    },
    {
      "epoch": 2.3006439208984374e-05,
      "model_forward_time": 1.7586414813995361,
      "step": 24124
    },
    {
      "epoch": 2.3006439208984374e-05,
      "step": 24124,
      "training_step_time": 7.861103057861328
    },
    {
      "epoch": 2.300739288330078e-05,
      "model_forward_time": 1.7634198665618896,
      "step": 24125
    },
    {
      "epoch": 2.300739288330078e-05,
      "step": 24125,
      "training_step_time": 7.714802503585815
    },
    {
      "epoch": 2.300834655761719e-05,
      "model_forward_time": 1.7568323612213135,
      "step": 24126
    },
    {
      "epoch": 2.300834655761719e-05,
      "step": 24126,
      "training_step_time": 7.907965898513794
    },
    {
      "epoch": 2.3009300231933593e-05,
      "model_forward_time": 1.7662434577941895,
      "step": 24127
    },
    {
      "epoch": 2.3009300231933593e-05,
      "step": 24127,
      "training_step_time": 7.231624126434326
    },
    {
      "epoch": 2.301025390625e-05,
      "model_forward_time": 1.765026330947876,
      "step": 24128
    },
    {
      "epoch": 2.301025390625e-05,
      "step": 24128,
      "training_step_time": 7.327723741531372
    },
    {
      "epoch": 2.3011207580566408e-05,
      "model_forward_time": 1.7763888835906982,
      "step": 24129
    },
    {
      "epoch": 2.3011207580566408e-05,
      "step": 24129,
      "training_step_time": 7.958181619644165
    },
    {
      "epoch": 2.301216125488281e-05,
      "grad_norm": 0.11289335787296295,
      "learning_rate": 9.032510884281942e-06,
      "loss": 0.0634,
      "step": 24130
    },
    {
      "epoch": 2.301216125488281e-05,
      "model_forward_time": 1.761225700378418,
      "step": 24130
    },
    {
      "dynamics_loss_avg": 0.053443195298314095,
      "epoch": 2.301216125488281e-05,
      "step": 24130
    },
    {
      "action_loss_avg": 0.0020933509629685433,
      "epoch": 2.301216125488281e-05,
      "step": 24130
    },
    {
      "epoch": 2.301216125488281e-05,
      "step": 24130,
      "training_step_time": 8.898201704025269
    },
    {
      "epoch": 2.301311492919922e-05,
      "model_forward_time": 2.8733901977539062,
      "step": 24131
    },
    {
      "epoch": 2.301311492919922e-05,
      "step": 24131,
      "training_step_time": 8.965611934661865
    },
    {
      "epoch": 2.3014068603515626e-05,
      "model_forward_time": 1.869992971420288,
      "step": 24132
    },
    {
      "epoch": 2.3014068603515626e-05,
      "step": 24132,
      "training_step_time": 9.141459226608276
    },
    {
      "epoch": 2.301502227783203e-05,
      "model_forward_time": 1.7621257305145264,
      "step": 24133
    },
    {
      "epoch": 2.301502227783203e-05,
      "step": 24133,
      "training_step_time": 7.882822513580322
    },
    {
      "epoch": 2.3015975952148438e-05,
      "model_forward_time": 1.7539660930633545,
      "step": 24134
    },
    {
      "epoch": 2.3015975952148438e-05,
      "step": 24134,
      "training_step_time": 8.454331159591675
    },
    {
      "epoch": 2.3016929626464845e-05,
      "model_forward_time": 1.7693636417388916,
      "step": 24135
    },
    {
      "epoch": 2.3016929626464845e-05,
      "step": 24135,
      "training_step_time": 7.956981658935547
    },
    {
      "epoch": 2.301788330078125e-05,
      "model_forward_time": 1.7672619819641113,
      "step": 24136
    },
    {
      "epoch": 2.301788330078125e-05,
      "step": 24136,
      "training_step_time": 7.413489103317261
    },
    {
      "epoch": 2.3018836975097657e-05,
      "model_forward_time": 1.760545253753662,
      "step": 24137
    },
    {
      "epoch": 2.3018836975097657e-05,
      "step": 24137,
      "training_step_time": 7.347318887710571
    },
    {
      "epoch": 2.3019790649414064e-05,
      "model_forward_time": 1.761580228805542,
      "step": 24138
    },
    {
      "epoch": 2.3019790649414064e-05,
      "step": 24138,
      "training_step_time": 9.328305721282959
    },
    {
      "epoch": 2.3020744323730468e-05,
      "model_forward_time": 1.7649025917053223,
      "step": 24139
    },
    {
      "epoch": 2.3020744323730468e-05,
      "step": 24139,
      "training_step_time": 9.63233494758606
    },
    {
      "epoch": 2.3021697998046876e-05,
      "grad_norm": 0.15265272557735443,
      "learning_rate": 9.03153308099085e-06,
      "loss": 0.0664,
      "step": 24140
    },
    {
      "epoch": 2.3021697998046876e-05,
      "model_forward_time": 2.8804492950439453,
      "step": 24140
    },
    {
      "dynamics_loss_avg": 0.07044050619006156,
      "epoch": 2.3021697998046876e-05,
      "step": 24140
    },
    {
      "action_loss_avg": 0.0026376981113571674,
      "epoch": 2.3021697998046876e-05,
      "step": 24140
    },
    {
      "epoch": 2.3021697998046876e-05,
      "step": 24140,
      "training_step_time": 9.574678659439087
    },
    {
      "epoch": 2.302265167236328e-05,
      "model_forward_time": 1.8111283779144287,
      "step": 24141
    },
    {
      "epoch": 2.302265167236328e-05,
      "step": 24141,
      "training_step_time": 7.776946067810059
    },
    {
      "epoch": 2.3023605346679687e-05,
      "model_forward_time": 1.7612478733062744,
      "step": 24142
    },
    {
      "epoch": 2.3023605346679687e-05,
      "step": 24142,
      "training_step_time": 8.344407320022583
    },
    {
      "epoch": 2.3024559020996094e-05,
      "model_forward_time": 1.7656936645507812,
      "step": 24143
    },
    {
      "epoch": 2.3024559020996094e-05,
      "step": 24143,
      "training_step_time": 8.717938899993896
    },
    {
      "epoch": 2.30255126953125e-05,
      "model_forward_time": 1.7667226791381836,
      "step": 24144
    },
    {
      "epoch": 2.30255126953125e-05,
      "step": 24144,
      "training_step_time": 7.7435126304626465
    },
    {
      "epoch": 2.3026466369628906e-05,
      "model_forward_time": 1.765035629272461,
      "step": 24145
    },
    {
      "epoch": 2.3026466369628906e-05,
      "step": 24145,
      "training_step_time": 6.848869800567627
    },
    {
      "epoch": 2.3027420043945313e-05,
      "model_forward_time": 1.7609949111938477,
      "step": 24146
    },
    {
      "epoch": 2.3027420043945313e-05,
      "step": 24146,
      "training_step_time": 7.134824275970459
    },
    {
      "epoch": 2.3028373718261717e-05,
      "model_forward_time": 1.7591519355773926,
      "step": 24147
    },
    {
      "epoch": 2.3028373718261717e-05,
      "step": 24147,
      "training_step_time": 8.248924255371094
    },
    {
      "epoch": 2.3029327392578125e-05,
      "model_forward_time": 1.7600202560424805,
      "step": 24148
    },
    {
      "epoch": 2.3029327392578125e-05,
      "step": 24148,
      "training_step_time": 7.860453128814697
    },
    {
      "epoch": 2.3030281066894532e-05,
      "model_forward_time": 1.7585594654083252,
      "step": 24149
    },
    {
      "epoch": 2.3030281066894532e-05,
      "step": 24149,
      "training_step_time": 10.311781167984009
    },
    {
      "epoch": 2.3031234741210936e-05,
      "grad_norm": 0.215109720826149,
      "learning_rate": 9.030554836817358e-06,
      "loss": 0.0647,
      "step": 24150
    },
    {
      "epoch": 2.3031234741210936e-05,
      "model_forward_time": 1.7736496925354004,
      "step": 24150
    },
    {
      "dynamics_loss_avg": 0.0631051518023014,
      "epoch": 2.3031234741210936e-05,
      "step": 24150
    },
    {
      "action_loss_avg": 0.00247433491749689,
      "epoch": 2.3031234741210936e-05,
      "step": 24150
    },
    {
      "epoch": 2.3031234741210936e-05,
      "step": 24150,
      "training_step_time": 8.717773675918579
    },
    {
      "epoch": 2.3032188415527344e-05,
      "model_forward_time": 1.763390064239502,
      "step": 24151
    },
    {
      "epoch": 2.3032188415527344e-05,
      "step": 24151,
      "training_step_time": 7.91853141784668
    },
    {
      "epoch": 2.303314208984375e-05,
      "model_forward_time": 1.7631971836090088,
      "step": 24152
    },
    {
      "epoch": 2.303314208984375e-05,
      "step": 24152,
      "training_step_time": 7.845226287841797
    },
    {
      "epoch": 2.3034095764160155e-05,
      "model_forward_time": 1.732182264328003,
      "step": 24153
    },
    {
      "epoch": 2.3034095764160155e-05,
      "step": 24153,
      "training_step_time": 7.693866491317749
    },
    {
      "epoch": 2.3035049438476562e-05,
      "model_forward_time": 1.7624921798706055,
      "step": 24154
    },
    {
      "epoch": 2.3035049438476562e-05,
      "step": 24154,
      "training_step_time": 7.757220268249512
    },
    {
      "epoch": 2.303600311279297e-05,
      "model_forward_time": 1.7641475200653076,
      "step": 24155
    },
    {
      "epoch": 2.303600311279297e-05,
      "step": 24155,
      "training_step_time": 7.24375057220459
    },
    {
      "epoch": 2.3036956787109374e-05,
      "model_forward_time": 1.768376350402832,
      "step": 24156
    },
    {
      "epoch": 2.3036956787109374e-05,
      "step": 24156,
      "training_step_time": 7.668294906616211
    },
    {
      "epoch": 2.303791046142578e-05,
      "model_forward_time": 1.7638421058654785,
      "step": 24157
    },
    {
      "epoch": 2.303791046142578e-05,
      "step": 24157,
      "training_step_time": 7.88030743598938
    },
    {
      "epoch": 2.303886413574219e-05,
      "model_forward_time": 1.7731497287750244,
      "step": 24158
    },
    {
      "epoch": 2.303886413574219e-05,
      "step": 24158,
      "training_step_time": 9.549703121185303
    },
    {
      "epoch": 2.3039817810058593e-05,
      "model_forward_time": 2.916477680206299,
      "step": 24159
    },
    {
      "epoch": 2.3039817810058593e-05,
      "step": 24159,
      "training_step_time": 10.533333539962769
    },
    {
      "epoch": 2.3040771484375e-05,
      "grad_norm": 0.16834944486618042,
      "learning_rate": 9.02957615186845e-06,
      "loss": 0.0658,
      "step": 24160
    },
    {
      "epoch": 2.3040771484375e-05,
      "model_forward_time": 1.7332711219787598,
      "step": 24160
    },
    {
      "dynamics_loss_avg": 0.0609321067109704,
      "epoch": 2.3040771484375e-05,
      "step": 24160
    },
    {
      "action_loss_avg": 0.00409938560333103,
      "epoch": 2.3040771484375e-05,
      "step": 24160
    },
    {
      "epoch": 2.3040771484375e-05,
      "step": 24160,
      "training_step_time": 8.525765657424927
    },
    {
      "epoch": 2.3041725158691408e-05,
      "model_forward_time": 1.7456717491149902,
      "step": 24161
    },
    {
      "epoch": 2.3041725158691408e-05,
      "step": 24161,
      "training_step_time": 7.9036853313446045
    },
    {
      "epoch": 2.304267883300781e-05,
      "model_forward_time": 1.7465460300445557,
      "step": 24162
    },
    {
      "epoch": 2.304267883300781e-05,
      "step": 24162,
      "training_step_time": 7.627700567245483
    },
    {
      "epoch": 2.304363250732422e-05,
      "model_forward_time": 1.763132095336914,
      "step": 24163
    },
    {
      "epoch": 2.304363250732422e-05,
      "step": 24163,
      "training_step_time": 7.727911949157715
    },
    {
      "epoch": 2.3044586181640626e-05,
      "model_forward_time": 1.7432749271392822,
      "step": 24164
    },
    {
      "epoch": 2.3044586181640626e-05,
      "step": 24164,
      "training_step_time": 7.7536468505859375
    },
    {
      "epoch": 2.304553985595703e-05,
      "model_forward_time": 1.7540168762207031,
      "step": 24165
    },
    {
      "epoch": 2.304553985595703e-05,
      "step": 24165,
      "training_step_time": 7.491770505905151
    },
    {
      "epoch": 2.3046493530273438e-05,
      "model_forward_time": 1.7524292469024658,
      "step": 24166
    },
    {
      "epoch": 2.3046493530273438e-05,
      "step": 24166,
      "training_step_time": 7.841893196105957
    },
    {
      "epoch": 2.3047447204589845e-05,
      "model_forward_time": 1.7455058097839355,
      "step": 24167
    },
    {
      "epoch": 2.3047447204589845e-05,
      "step": 24167,
      "training_step_time": 8.572952032089233
    },
    {
      "epoch": 2.304840087890625e-05,
      "model_forward_time": 2.884544610977173,
      "step": 24168
    },
    {
      "epoch": 2.304840087890625e-05,
      "step": 24168,
      "training_step_time": 8.744087219238281
    },
    {
      "epoch": 2.3049354553222657e-05,
      "model_forward_time": 1.839813232421875,
      "step": 24169
    },
    {
      "epoch": 2.3049354553222657e-05,
      "step": 24169,
      "training_step_time": 8.298167943954468
    },
    {
      "epoch": 2.3050308227539064e-05,
      "grad_norm": 0.16505727171897888,
      "learning_rate": 9.028597026251156e-06,
      "loss": 0.0626,
      "step": 24170
    },
    {
      "epoch": 2.3050308227539064e-05,
      "model_forward_time": 1.7306058406829834,
      "step": 24170
    },
    {
      "dynamics_loss_avg": 0.055390076152980326,
      "epoch": 2.3050308227539064e-05,
      "step": 24170
    },
    {
      "action_loss_avg": 0.0037275137729011476,
      "epoch": 2.3050308227539064e-05,
      "step": 24170
    },
    {
      "epoch": 2.3050308227539064e-05,
      "step": 24170,
      "training_step_time": 7.464260816574097
    },
    {
      "epoch": 2.3051261901855468e-05,
      "model_forward_time": 1.7516961097717285,
      "step": 24171
    },
    {
      "epoch": 2.3051261901855468e-05,
      "step": 24171,
      "training_step_time": 7.799186944961548
    },
    {
      "epoch": 2.3052215576171876e-05,
      "model_forward_time": 1.7513654232025146,
      "step": 24172
    },
    {
      "epoch": 2.3052215576171876e-05,
      "step": 24172,
      "training_step_time": 7.715254306793213
    },
    {
      "epoch": 2.305316925048828e-05,
      "model_forward_time": 1.7464323043823242,
      "step": 24173
    },
    {
      "epoch": 2.305316925048828e-05,
      "step": 24173,
      "training_step_time": 6.773102521896362
    },
    {
      "epoch": 2.3054122924804687e-05,
      "model_forward_time": 1.750798225402832,
      "step": 24174
    },
    {
      "epoch": 2.3054122924804687e-05,
      "step": 24174,
      "training_step_time": 7.543853759765625
    },
    {
      "epoch": 2.3055076599121094e-05,
      "model_forward_time": 1.750413417816162,
      "step": 24175
    },
    {
      "epoch": 2.3055076599121094e-05,
      "step": 24175,
      "training_step_time": 7.75646185874939
    },
    {
      "epoch": 2.30560302734375e-05,
      "model_forward_time": 1.7472093105316162,
      "step": 24176
    },
    {
      "epoch": 2.30560302734375e-05,
      "step": 24176,
      "training_step_time": 8.640190601348877
    },
    {
      "epoch": 2.3056983947753906e-05,
      "model_forward_time": 2.887559175491333,
      "step": 24177
    },
    {
      "epoch": 2.3056983947753906e-05,
      "step": 24177,
      "training_step_time": 9.687128067016602
    },
    {
      "epoch": 2.3057937622070313e-05,
      "model_forward_time": 1.820486068725586,
      "step": 24178
    },
    {
      "epoch": 2.3057937622070313e-05,
      "step": 24178,
      "training_step_time": 7.672991037368774
    },
    {
      "epoch": 2.3058891296386717e-05,
      "model_forward_time": 1.7543120384216309,
      "step": 24179
    },
    {
      "epoch": 2.3058891296386717e-05,
      "step": 24179,
      "training_step_time": 7.776640892028809
    },
    {
      "epoch": 2.3059844970703125e-05,
      "grad_norm": 0.20426826179027557,
      "learning_rate": 9.027617460072547e-06,
      "loss": 0.0632,
      "step": 24180
    },
    {
      "epoch": 2.3059844970703125e-05,
      "model_forward_time": 1.7376155853271484,
      "step": 24180
    },
    {
      "dynamics_loss_avg": 0.054245694819837806,
      "epoch": 2.3059844970703125e-05,
      "step": 24180
    },
    {
      "action_loss_avg": 0.002644616225734353,
      "epoch": 2.3059844970703125e-05,
      "step": 24180
    },
    {
      "epoch": 2.3059844970703125e-05,
      "step": 24180,
      "training_step_time": 7.783634185791016
    },
    {
      "epoch": 2.3060798645019532e-05,
      "model_forward_time": 1.7460434436798096,
      "step": 24181
    },
    {
      "epoch": 2.3060798645019532e-05,
      "step": 24181,
      "training_step_time": 7.62748908996582
    },
    {
      "epoch": 2.3061752319335936e-05,
      "model_forward_time": 1.750270128250122,
      "step": 24182
    },
    {
      "epoch": 2.3061752319335936e-05,
      "step": 24182,
      "training_step_time": 6.783012628555298
    },
    {
      "epoch": 2.3062705993652344e-05,
      "model_forward_time": 1.7418608665466309,
      "step": 24183
    },
    {
      "epoch": 2.3062705993652344e-05,
      "step": 24183,
      "training_step_time": 7.571130990982056
    },
    {
      "epoch": 2.306365966796875e-05,
      "model_forward_time": 1.7431187629699707,
      "step": 24184
    },
    {
      "epoch": 2.306365966796875e-05,
      "step": 24184,
      "training_step_time": 7.752028942108154
    },
    {
      "epoch": 2.3064613342285155e-05,
      "model_forward_time": 1.768380880355835,
      "step": 24185
    },
    {
      "epoch": 2.3064613342285155e-05,
      "step": 24185,
      "training_step_time": 7.7507500648498535
    },
    {
      "epoch": 2.3065567016601562e-05,
      "model_forward_time": 2.8445913791656494,
      "step": 24186
    },
    {
      "epoch": 2.3065567016601562e-05,
      "step": 24186,
      "training_step_time": 7.761486768722534
    },
    {
      "epoch": 2.306652069091797e-05,
      "model_forward_time": 1.768953800201416,
      "step": 24187
    },
    {
      "epoch": 2.306652069091797e-05,
      "step": 24187,
      "training_step_time": 8.8935067653656
    },
    {
      "epoch": 2.3067474365234374e-05,
      "model_forward_time": 1.7411527633666992,
      "step": 24188
    },
    {
      "epoch": 2.3067474365234374e-05,
      "step": 24188,
      "training_step_time": 8.540980577468872
    },
    {
      "epoch": 2.306842803955078e-05,
      "model_forward_time": 1.7441060543060303,
      "step": 24189
    },
    {
      "epoch": 2.306842803955078e-05,
      "step": 24189,
      "training_step_time": 7.56789231300354
    },
    {
      "epoch": 2.306938171386719e-05,
      "grad_norm": 0.14758753776550293,
      "learning_rate": 9.026637453439746e-06,
      "loss": 0.0637,
      "step": 24190
    },
    {
      "epoch": 2.306938171386719e-05,
      "model_forward_time": 1.7516868114471436,
      "step": 24190
    },
    {
      "dynamics_loss_avg": 0.06887640506029129,
      "epoch": 2.306938171386719e-05,
      "step": 24190
    },
    {
      "action_loss_avg": 0.0034460874856449665,
      "epoch": 2.306938171386719e-05,
      "step": 24190
    },
    {
      "epoch": 2.306938171386719e-05,
      "step": 24190,
      "training_step_time": 7.634671449661255
    },
    {
      "epoch": 2.3070335388183593e-05,
      "model_forward_time": 1.7538127899169922,
      "step": 24191
    },
    {
      "epoch": 2.3070335388183593e-05,
      "step": 24191,
      "training_step_time": 7.82333517074585
    },
    {
      "epoch": 2.30712890625e-05,
      "model_forward_time": 1.7538597583770752,
      "step": 24192
    },
    {
      "epoch": 2.30712890625e-05,
      "step": 24192,
      "training_step_time": 7.572632789611816
    },
    {
      "epoch": 2.3072242736816408e-05,
      "model_forward_time": 1.7454710006713867,
      "step": 24193
    },
    {
      "epoch": 2.3072242736816408e-05,
      "step": 24193,
      "training_step_time": 8.753980875015259
    },
    {
      "epoch": 2.307319641113281e-05,
      "model_forward_time": 1.7398815155029297,
      "step": 24194
    },
    {
      "epoch": 2.307319641113281e-05,
      "step": 24194,
      "training_step_time": 9.306062698364258
    },
    {
      "epoch": 2.307415008544922e-05,
      "model_forward_time": 2.8213658332824707,
      "step": 24195
    },
    {
      "epoch": 2.307415008544922e-05,
      "step": 24195,
      "training_step_time": 11.39645004272461
    },
    {
      "epoch": 2.3075103759765626e-05,
      "model_forward_time": 1.7497637271881104,
      "step": 24196
    },
    {
      "epoch": 2.3075103759765626e-05,
      "step": 24196,
      "training_step_time": 8.939598560333252
    },
    {
      "epoch": 2.307605743408203e-05,
      "model_forward_time": 1.8761110305786133,
      "step": 24197
    },
    {
      "epoch": 2.307605743408203e-05,
      "step": 24197,
      "training_step_time": 8.550084352493286
    },
    {
      "epoch": 2.3077011108398438e-05,
      "model_forward_time": 1.7563910484313965,
      "step": 24198
    },
    {
      "epoch": 2.3077011108398438e-05,
      "step": 24198,
      "training_step_time": 8.681039571762085
    },
    {
      "epoch": 2.3077964782714845e-05,
      "model_forward_time": 1.747986078262329,
      "step": 24199
    },
    {
      "epoch": 2.3077964782714845e-05,
      "step": 24199,
      "training_step_time": 7.829547882080078
    },
    {
      "epoch": 2.307891845703125e-05,
      "grad_norm": 0.1421235352754593,
      "learning_rate": 9.025657006459927e-06,
      "loss": 0.0675,
      "step": 24200
    },
    {
      "epoch": 2.307891845703125e-05,
      "model_forward_time": 1.7462129592895508,
      "step": 24200
    },
    {
      "dynamics_loss_avg": 0.06620339043438435,
      "epoch": 2.307891845703125e-05,
      "step": 24200
    },
    {
      "action_loss_avg": 0.0022097576584201306,
      "epoch": 2.307891845703125e-05,
      "step": 24200
    },
    {
      "epoch": 2.307891845703125e-05,
      "step": 24200,
      "training_step_time": 7.595044374465942
    },
    {
      "epoch": 2.3079872131347657e-05,
      "model_forward_time": 1.7364346981048584,
      "step": 24201
    },
    {
      "epoch": 2.3079872131347657e-05,
      "step": 24201,
      "training_step_time": 7.575222969055176
    },
    {
      "epoch": 2.3080825805664064e-05,
      "model_forward_time": 1.7442584037780762,
      "step": 24202
    },
    {
      "epoch": 2.3080825805664064e-05,
      "step": 24202,
      "training_step_time": 7.54044246673584
    },
    {
      "epoch": 2.3081779479980468e-05,
      "model_forward_time": 1.3975019454956055,
      "step": 24203
    },
    {
      "epoch": 2.3081779479980468e-05,
      "step": 24203,
      "training_step_time": 9.08533000946045
    },
    {
      "epoch": 2.3082733154296876e-05,
      "model_forward_time": 2.8607523441314697,
      "step": 24204
    },
    {
      "epoch": 2.3082733154296876e-05,
      "step": 24204,
      "training_step_time": 7.708055734634399
    },
    {
      "epoch": 2.308368682861328e-05,
      "model_forward_time": 1.7484631538391113,
      "step": 24205
    },
    {
      "epoch": 2.308368682861328e-05,
      "step": 24205,
      "training_step_time": 8.405152797698975
    },
    {
      "epoch": 2.3084640502929687e-05,
      "model_forward_time": 1.8675811290740967,
      "step": 24206
    },
    {
      "epoch": 2.3084640502929687e-05,
      "step": 24206,
      "training_step_time": 7.759876489639282
    },
    {
      "epoch": 2.3085594177246094e-05,
      "model_forward_time": 1.7429885864257812,
      "step": 24207
    },
    {
      "epoch": 2.3085594177246094e-05,
      "step": 24207,
      "training_step_time": 7.59141993522644
    },
    {
      "epoch": 2.30865478515625e-05,
      "model_forward_time": 1.7463738918304443,
      "step": 24208
    },
    {
      "epoch": 2.30865478515625e-05,
      "step": 24208,
      "training_step_time": 7.716633319854736
    },
    {
      "epoch": 2.3087501525878906e-05,
      "model_forward_time": 1.7577693462371826,
      "step": 24209
    },
    {
      "epoch": 2.3087501525878906e-05,
      "step": 24209,
      "training_step_time": 7.608987331390381
    },
    {
      "epoch": 2.3088455200195313e-05,
      "grad_norm": 0.12139511853456497,
      "learning_rate": 9.024676119240312e-06,
      "loss": 0.0618,
      "step": 24210
    },
    {
      "epoch": 2.3088455200195313e-05,
      "model_forward_time": 1.754011869430542,
      "step": 24210
    },
    {
      "dynamics_loss_avg": 0.05466340519487858,
      "epoch": 2.3088455200195313e-05,
      "step": 24210
    },
    {
      "action_loss_avg": 0.00228124464629218,
      "epoch": 2.3088455200195313e-05,
      "step": 24210
    },
    {
      "epoch": 2.3088455200195313e-05,
      "step": 24210,
      "training_step_time": 7.4875328540802
    },
    {
      "epoch": 2.3089408874511717e-05,
      "model_forward_time": 1.7531938552856445,
      "step": 24211
    },
    {
      "epoch": 2.3089408874511717e-05,
      "step": 24211,
      "training_step_time": 7.265792608261108
    },
    {
      "epoch": 2.3090362548828125e-05,
      "model_forward_time": 1.7494642734527588,
      "step": 24212
    },
    {
      "epoch": 2.3090362548828125e-05,
      "step": 24212,
      "training_step_time": 9.436801195144653
    },
    {
      "epoch": 2.3091316223144532e-05,
      "model_forward_time": 2.9037632942199707,
      "step": 24213
    },
    {
      "epoch": 2.3091316223144532e-05,
      "step": 24213,
      "training_step_time": 8.826562881469727
    },
    {
      "epoch": 2.3092269897460936e-05,
      "model_forward_time": 1.7486085891723633,
      "step": 24214
    },
    {
      "epoch": 2.3092269897460936e-05,
      "step": 24214,
      "training_step_time": 7.78769588470459
    },
    {
      "epoch": 2.3093223571777344e-05,
      "model_forward_time": 1.8501605987548828,
      "step": 24215
    },
    {
      "epoch": 2.3093223571777344e-05,
      "step": 24215,
      "training_step_time": 7.863801956176758
    },
    {
      "epoch": 2.309417724609375e-05,
      "model_forward_time": 1.753676176071167,
      "step": 24216
    },
    {
      "epoch": 2.309417724609375e-05,
      "step": 24216,
      "training_step_time": 7.735564947128296
    },
    {
      "epoch": 2.3095130920410155e-05,
      "model_forward_time": 1.7430098056793213,
      "step": 24217
    },
    {
      "epoch": 2.3095130920410155e-05,
      "step": 24217,
      "training_step_time": 7.681980133056641
    },
    {
      "epoch": 2.3096084594726562e-05,
      "model_forward_time": 1.7439525127410889,
      "step": 24218
    },
    {
      "epoch": 2.3096084594726562e-05,
      "step": 24218,
      "training_step_time": 7.662659645080566
    },
    {
      "epoch": 2.309703826904297e-05,
      "model_forward_time": 1.7589235305786133,
      "step": 24219
    },
    {
      "epoch": 2.309703826904297e-05,
      "step": 24219,
      "training_step_time": 7.7440712451934814
    },
    {
      "epoch": 2.3097991943359374e-05,
      "grad_norm": 0.19209884107112885,
      "learning_rate": 9.023694791888167e-06,
      "loss": 0.0646,
      "step": 24220
    },
    {
      "epoch": 2.3097991943359374e-05,
      "model_forward_time": 1.753965139389038,
      "step": 24220
    },
    {
      "dynamics_loss_avg": 0.06772661171853542,
      "epoch": 2.3097991943359374e-05,
      "step": 24220
    },
    {
      "action_loss_avg": 0.002624538727104664,
      "epoch": 2.3097991943359374e-05,
      "step": 24220
    },
    {
      "epoch": 2.3097991943359374e-05,
      "step": 24220,
      "training_step_time": 7.271543025970459
    },
    {
      "epoch": 2.309894561767578e-05,
      "model_forward_time": 1.7480015754699707,
      "step": 24221
    },
    {
      "epoch": 2.309894561767578e-05,
      "step": 24221,
      "training_step_time": 8.523384094238281
    },
    {
      "epoch": 2.309989929199219e-05,
      "model_forward_time": 2.876573324203491,
      "step": 24222
    },
    {
      "epoch": 2.309989929199219e-05,
      "step": 24222,
      "training_step_time": 8.233757734298706
    },
    {
      "epoch": 2.3100852966308593e-05,
      "model_forward_time": 1.7524223327636719,
      "step": 24223
    },
    {
      "epoch": 2.3100852966308593e-05,
      "step": 24223,
      "training_step_time": 7.678330183029175
    },
    {
      "epoch": 2.3101806640625e-05,
      "model_forward_time": 1.8207004070281982,
      "step": 24224
    },
    {
      "epoch": 2.3101806640625e-05,
      "step": 24224,
      "training_step_time": 8.34940481185913
    },
    {
      "epoch": 2.3102760314941408e-05,
      "model_forward_time": 1.7490887641906738,
      "step": 24225
    },
    {
      "epoch": 2.3102760314941408e-05,
      "step": 24225,
      "training_step_time": 7.576494216918945
    },
    {
      "epoch": 2.310371398925781e-05,
      "model_forward_time": 1.746103286743164,
      "step": 24226
    },
    {
      "epoch": 2.310371398925781e-05,
      "step": 24226,
      "training_step_time": 8.003434658050537
    },
    {
      "epoch": 2.310466766357422e-05,
      "model_forward_time": 1.7478761672973633,
      "step": 24227
    },
    {
      "epoch": 2.310466766357422e-05,
      "step": 24227,
      "training_step_time": 7.537966012954712
    },
    {
      "epoch": 2.3105621337890626e-05,
      "model_forward_time": 1.746079444885254,
      "step": 24228
    },
    {
      "epoch": 2.3105621337890626e-05,
      "step": 24228,
      "training_step_time": 7.657302618026733
    },
    {
      "epoch": 2.310657501220703e-05,
      "model_forward_time": 1.752434492111206,
      "step": 24229
    },
    {
      "epoch": 2.310657501220703e-05,
      "step": 24229,
      "training_step_time": 7.570274114608765
    },
    {
      "epoch": 2.3107528686523438e-05,
      "grad_norm": 0.1733914315700531,
      "learning_rate": 9.022713024510807e-06,
      "loss": 0.0661,
      "step": 24230
    },
    {
      "epoch": 2.3107528686523438e-05,
      "model_forward_time": 1.7665433883666992,
      "step": 24230
    },
    {
      "dynamics_loss_avg": 0.0760772679001093,
      "epoch": 2.3107528686523438e-05,
      "step": 24230
    },
    {
      "action_loss_avg": 0.003458538034465164,
      "epoch": 2.3107528686523438e-05,
      "step": 24230
    },
    {
      "epoch": 2.3107528686523438e-05,
      "step": 24230,
      "training_step_time": 8.414798974990845
    },
    {
      "epoch": 2.3108482360839845e-05,
      "model_forward_time": 2.8655459880828857,
      "step": 24231
    },
    {
      "epoch": 2.3108482360839845e-05,
      "step": 24231,
      "training_step_time": 10.562591314315796
    },
    {
      "epoch": 2.310943603515625e-05,
      "model_forward_time": 1.764176368713379,
      "step": 24232
    },
    {
      "epoch": 2.310943603515625e-05,
      "step": 24232,
      "training_step_time": 7.69060492515564
    },
    {
      "epoch": 2.3110389709472657e-05,
      "model_forward_time": 1.819218397140503,
      "step": 24233
    },
    {
      "epoch": 2.3110389709472657e-05,
      "step": 24233,
      "training_step_time": 8.798163890838623
    },
    {
      "epoch": 2.3111343383789064e-05,
      "model_forward_time": 1.7464599609375,
      "step": 24234
    },
    {
      "epoch": 2.3111343383789064e-05,
      "step": 24234,
      "training_step_time": 7.811035394668579
    },
    {
      "epoch": 2.3112297058105468e-05,
      "model_forward_time": 1.773179054260254,
      "step": 24235
    },
    {
      "epoch": 2.3112297058105468e-05,
      "step": 24235,
      "training_step_time": 8.337660312652588
    },
    {
      "epoch": 2.3113250732421876e-05,
      "model_forward_time": 1.7670013904571533,
      "step": 24236
    },
    {
      "epoch": 2.3113250732421876e-05,
      "step": 24236,
      "training_step_time": 7.737705707550049
    },
    {
      "epoch": 2.311420440673828e-05,
      "model_forward_time": 1.7624835968017578,
      "step": 24237
    },
    {
      "epoch": 2.311420440673828e-05,
      "step": 24237,
      "training_step_time": 7.780625104904175
    },
    {
      "epoch": 2.3115158081054687e-05,
      "model_forward_time": 1.7666189670562744,
      "step": 24238
    },
    {
      "epoch": 2.3115158081054687e-05,
      "step": 24238,
      "training_step_time": 7.123822212219238
    },
    {
      "epoch": 2.3116111755371094e-05,
      "model_forward_time": 1.7544119358062744,
      "step": 24239
    },
    {
      "epoch": 2.3116111755371094e-05,
      "step": 24239,
      "training_step_time": 8.236209154129028
    },
    {
      "epoch": 2.31170654296875e-05,
      "grad_norm": 0.26821398735046387,
      "learning_rate": 9.021730817215601e-06,
      "loss": 0.0649,
      "step": 24240
    },
    {
      "epoch": 2.31170654296875e-05,
      "model_forward_time": 1.768200159072876,
      "step": 24240
    },
    {
      "dynamics_loss_avg": 0.06491380389779806,
      "epoch": 2.31170654296875e-05,
      "step": 24240
    },
    {
      "action_loss_avg": 0.003566627367399633,
      "epoch": 2.31170654296875e-05,
      "step": 24240
    },
    {
      "epoch": 2.31170654296875e-05,
      "step": 24240,
      "training_step_time": 7.739078998565674
    },
    {
      "epoch": 2.3118019104003906e-05,
      "model_forward_time": 2.9149975776672363,
      "step": 24241
    },
    {
      "epoch": 2.3118019104003906e-05,
      "step": 24241,
      "training_step_time": 7.752109050750732
    },
    {
      "epoch": 2.3118972778320313e-05,
      "model_forward_time": 1.7589666843414307,
      "step": 24242
    },
    {
      "epoch": 2.3118972778320313e-05,
      "step": 24242,
      "training_step_time": 8.82054090499878
    },
    {
      "epoch": 2.3119926452636717e-05,
      "model_forward_time": 1.8689992427825928,
      "step": 24243
    },
    {
      "epoch": 2.3119926452636717e-05,
      "step": 24243,
      "training_step_time": 7.774123668670654
    },
    {
      "epoch": 2.3120880126953125e-05,
      "model_forward_time": 1.7624297142028809,
      "step": 24244
    },
    {
      "epoch": 2.3120880126953125e-05,
      "step": 24244,
      "training_step_time": 7.646908283233643
    },
    {
      "epoch": 2.3121833801269532e-05,
      "model_forward_time": 1.759469985961914,
      "step": 24245
    },
    {
      "epoch": 2.3121833801269532e-05,
      "step": 24245,
      "training_step_time": 7.7798943519592285
    },
    {
      "epoch": 2.3122787475585936e-05,
      "model_forward_time": 1.7600643634796143,
      "step": 24246
    },
    {
      "epoch": 2.3122787475585936e-05,
      "step": 24246,
      "training_step_time": 7.672273397445679
    },
    {
      "epoch": 2.3123741149902344e-05,
      "model_forward_time": 1.7312614917755127,
      "step": 24247
    },
    {
      "epoch": 2.3123741149902344e-05,
      "step": 24247,
      "training_step_time": 7.15585732460022
    },
    {
      "epoch": 2.312469482421875e-05,
      "model_forward_time": 1.735506534576416,
      "step": 24248
    },
    {
      "epoch": 2.312469482421875e-05,
      "step": 24248,
      "training_step_time": 7.643707513809204
    },
    {
      "epoch": 2.3125648498535155e-05,
      "model_forward_time": 1.730189323425293,
      "step": 24249
    },
    {
      "epoch": 2.3125648498535155e-05,
      "step": 24249,
      "training_step_time": 8.55322551727295
    },
    {
      "epoch": 2.3126602172851562e-05,
      "grad_norm": 0.21085427701473236,
      "learning_rate": 9.02074817010996e-06,
      "loss": 0.0695,
      "step": 24250
    },
    {
      "epoch": 2.3126602172851562e-05,
      "model_forward_time": 2.3043477535247803,
      "step": 24250
    },
    {
      "dynamics_loss_avg": 0.05360662415623665,
      "epoch": 2.3126602172851562e-05,
      "step": 24250
    },
    {
      "action_loss_avg": 0.0012509584950748831,
      "epoch": 2.3126602172851562e-05,
      "step": 24250
    },
    {
      "epoch": 2.3126602172851562e-05,
      "step": 24250,
      "training_step_time": 9.306643962860107
    },
    {
      "epoch": 2.312755584716797e-05,
      "model_forward_time": 1.726196527481079,
      "step": 24251
    },
    {
      "epoch": 2.312755584716797e-05,
      "step": 24251,
      "training_step_time": 8.679707288742065
    },
    {
      "epoch": 2.3128509521484374e-05,
      "model_forward_time": 1.779829502105713,
      "step": 24252
    },
    {
      "epoch": 2.3128509521484374e-05,
      "step": 24252,
      "training_step_time": 7.7709949016571045
    },
    {
      "epoch": 2.312946319580078e-05,
      "model_forward_time": 1.3523240089416504,
      "step": 24253
    },
    {
      "epoch": 2.312946319580078e-05,
      "step": 24253,
      "training_step_time": 7.669118404388428
    },
    {
      "epoch": 2.313041687011719e-05,
      "model_forward_time": 1.7112534046173096,
      "step": 24254
    },
    {
      "epoch": 2.313041687011719e-05,
      "step": 24254,
      "training_step_time": 7.724700212478638
    },
    {
      "epoch": 2.3131370544433593e-05,
      "model_forward_time": 1.715357780456543,
      "step": 24255
    },
    {
      "epoch": 2.3131370544433593e-05,
      "step": 24255,
      "training_step_time": 7.647408485412598
    },
    {
      "epoch": 2.313232421875e-05,
      "model_forward_time": 1.7110047340393066,
      "step": 24256
    },
    {
      "epoch": 2.313232421875e-05,
      "step": 24256,
      "training_step_time": 7.777026414871216
    },
    {
      "epoch": 2.3133277893066408e-05,
      "model_forward_time": 1.7093007564544678,
      "step": 24257
    },
    {
      "epoch": 2.3133277893066408e-05,
      "step": 24257,
      "training_step_time": 7.612128734588623
    },
    {
      "epoch": 2.313423156738281e-05,
      "model_forward_time": 1.7173402309417725,
      "step": 24258
    },
    {
      "epoch": 2.313423156738281e-05,
      "step": 24258,
      "training_step_time": 7.737412452697754
    },
    {
      "epoch": 2.313518524169922e-05,
      "model_forward_time": 1.8759384155273438,
      "step": 24259
    },
    {
      "epoch": 2.313518524169922e-05,
      "step": 24259,
      "training_step_time": 7.693624019622803
    },
    {
      "epoch": 2.3136138916015626e-05,
      "grad_norm": 0.22750061750411987,
      "learning_rate": 9.019765083301342e-06,
      "loss": 0.0659,
      "step": 24260
    },
    {
      "epoch": 2.3136138916015626e-05,
      "model_forward_time": 1.717975378036499,
      "step": 24260
    },
    {
      "dynamics_loss_avg": 0.055114371702075005,
      "epoch": 2.3136138916015626e-05,
      "step": 24260
    },
    {
      "action_loss_avg": 0.002642699901480228,
      "epoch": 2.3136138916015626e-05,
      "step": 24260
    },
    {
      "epoch": 2.3136138916015626e-05,
      "step": 24260,
      "training_step_time": 8.693431615829468
    },
    {
      "epoch": 2.313709259033203e-05,
      "model_forward_time": 1.7700002193450928,
      "step": 24261
    },
    {
      "epoch": 2.313709259033203e-05,
      "step": 24261,
      "training_step_time": 9.22432017326355
    },
    {
      "epoch": 2.3138046264648438e-05,
      "model_forward_time": 1.7123115062713623,
      "step": 24262
    },
    {
      "epoch": 2.3138046264648438e-05,
      "step": 24262,
      "training_step_time": 7.8857948780059814
    },
    {
      "epoch": 2.3138999938964845e-05,
      "model_forward_time": 1.7203662395477295,
      "step": 24263
    },
    {
      "epoch": 2.3138999938964845e-05,
      "step": 24263,
      "training_step_time": 8.729371547698975
    },
    {
      "epoch": 2.313995361328125e-05,
      "model_forward_time": 1.714353322982788,
      "step": 24264
    },
    {
      "epoch": 2.313995361328125e-05,
      "step": 24264,
      "training_step_time": 7.695498943328857
    },
    {
      "epoch": 2.3140907287597657e-05,
      "model_forward_time": 1.718515157699585,
      "step": 24265
    },
    {
      "epoch": 2.3140907287597657e-05,
      "step": 24265,
      "training_step_time": 8.954182147979736
    },
    {
      "epoch": 2.314186096191406e-05,
      "model_forward_time": 1.701521396636963,
      "step": 24266
    },
    {
      "epoch": 2.314186096191406e-05,
      "step": 24266,
      "training_step_time": 7.346920490264893
    },
    {
      "epoch": 2.3142814636230468e-05,
      "model_forward_time": 1.7006101608276367,
      "step": 24267
    },
    {
      "epoch": 2.3142814636230468e-05,
      "step": 24267,
      "training_step_time": 8.532360792160034
    },
    {
      "epoch": 2.3143768310546876e-05,
      "model_forward_time": 2.294509172439575,
      "step": 24268
    },
    {
      "epoch": 2.3143768310546876e-05,
      "step": 24268,
      "training_step_time": 8.56328010559082
    },
    {
      "epoch": 2.314472198486328e-05,
      "model_forward_time": 1.7150630950927734,
      "step": 24269
    },
    {
      "epoch": 2.314472198486328e-05,
      "step": 24269,
      "training_step_time": 7.83322548866272
    },
    {
      "epoch": 2.3145675659179687e-05,
      "grad_norm": 0.23773697018623352,
      "learning_rate": 9.01878155689726e-06,
      "loss": 0.0627,
      "step": 24270
    },
    {
      "epoch": 2.3145675659179687e-05,
      "model_forward_time": 1.7739129066467285,
      "step": 24270
    },
    {
      "dynamics_loss_avg": 0.05536963660269976,
      "epoch": 2.3145675659179687e-05,
      "step": 24270
    },
    {
      "action_loss_avg": 0.005064100224990398,
      "epoch": 2.3145675659179687e-05,
      "step": 24270
    },
    {
      "epoch": 2.3145675659179687e-05,
      "step": 24270,
      "training_step_time": 7.9315690994262695
    },
    {
      "epoch": 2.3146629333496095e-05,
      "model_forward_time": 1.7162835597991943,
      "step": 24271
    },
    {
      "epoch": 2.3146629333496095e-05,
      "step": 24271,
      "training_step_time": 8.851853370666504
    },
    {
      "epoch": 2.31475830078125e-05,
      "model_forward_time": 1.7270348072052002,
      "step": 24272
    },
    {
      "epoch": 2.31475830078125e-05,
      "step": 24272,
      "training_step_time": 7.726801872253418
    },
    {
      "epoch": 2.3148536682128906e-05,
      "model_forward_time": 1.720341682434082,
      "step": 24273
    },
    {
      "epoch": 2.3148536682128906e-05,
      "step": 24273,
      "training_step_time": 8.436121225357056
    },
    {
      "epoch": 2.3149490356445313e-05,
      "model_forward_time": 1.7089972496032715,
      "step": 24274
    },
    {
      "epoch": 2.3149490356445313e-05,
      "step": 24274,
      "training_step_time": 7.804772138595581
    },
    {
      "epoch": 2.3150444030761717e-05,
      "model_forward_time": 1.7189910411834717,
      "step": 24275
    },
    {
      "epoch": 2.3150444030761717e-05,
      "step": 24275,
      "training_step_time": 7.98710298538208
    },
    {
      "epoch": 2.3151397705078125e-05,
      "model_forward_time": 1.7212309837341309,
      "step": 24276
    },
    {
      "epoch": 2.3151397705078125e-05,
      "step": 24276,
      "training_step_time": 7.6894848346710205
    },
    {
      "epoch": 2.3152351379394532e-05,
      "model_forward_time": 1.7144427299499512,
      "step": 24277
    },
    {
      "epoch": 2.3152351379394532e-05,
      "step": 24277,
      "training_step_time": 10.455664157867432
    },
    {
      "epoch": 2.3153305053710936e-05,
      "model_forward_time": 1.7202470302581787,
      "step": 24278
    },
    {
      "epoch": 2.3153305053710936e-05,
      "step": 24278,
      "training_step_time": 7.826565742492676
    },
    {
      "epoch": 2.3154258728027344e-05,
      "model_forward_time": 1.350525140762329,
      "step": 24279
    },
    {
      "epoch": 2.3154258728027344e-05,
      "step": 24279,
      "training_step_time": 8.926674127578735
    },
    {
      "epoch": 2.315521240234375e-05,
      "grad_norm": 0.20476779341697693,
      "learning_rate": 9.017797591005269e-06,
      "loss": 0.064,
      "step": 24280
    },
    {
      "epoch": 2.315521240234375e-05,
      "model_forward_time": 1.717602014541626,
      "step": 24280
    },
    {
      "dynamics_loss_avg": 0.06863492541015148,
      "epoch": 2.315521240234375e-05,
      "step": 24280
    },
    {
      "action_loss_avg": 0.004744384670630097,
      "epoch": 2.315521240234375e-05,
      "step": 24280
    },
    {
      "epoch": 2.315521240234375e-05,
      "step": 24280,
      "training_step_time": 7.730878114700317
    },
    {
      "epoch": 2.3156166076660155e-05,
      "model_forward_time": 1.712583065032959,
      "step": 24281
    },
    {
      "epoch": 2.3156166076660155e-05,
      "step": 24281,
      "training_step_time": 7.616739273071289
    },
    {
      "epoch": 2.3157119750976563e-05,
      "model_forward_time": 1.714977741241455,
      "step": 24282
    },
    {
      "epoch": 2.3157119750976563e-05,
      "step": 24282,
      "training_step_time": 7.857701778411865
    },
    {
      "epoch": 2.315807342529297e-05,
      "model_forward_time": 1.3515245914459229,
      "step": 24283
    },
    {
      "epoch": 2.315807342529297e-05,
      "step": 24283,
      "training_step_time": 7.697068214416504
    },
    {
      "epoch": 2.3159027099609374e-05,
      "model_forward_time": 1.7000062465667725,
      "step": 24284
    },
    {
      "epoch": 2.3159027099609374e-05,
      "step": 24284,
      "training_step_time": 7.121383905410767
    },
    {
      "epoch": 2.315998077392578e-05,
      "model_forward_time": 1.7236096858978271,
      "step": 24285
    },
    {
      "epoch": 2.315998077392578e-05,
      "step": 24285,
      "training_step_time": 8.301918268203735
    },
    {
      "epoch": 2.316093444824219e-05,
      "model_forward_time": 1.7077410221099854,
      "step": 24286
    },
    {
      "epoch": 2.316093444824219e-05,
      "step": 24286,
      "training_step_time": 7.794215679168701
    },
    {
      "epoch": 2.3161888122558593e-05,
      "model_forward_time": 2.2807633876800537,
      "step": 24287
    },
    {
      "epoch": 2.3161888122558593e-05,
      "step": 24287,
      "training_step_time": 8.719146251678467
    },
    {
      "epoch": 2.3162841796875e-05,
      "model_forward_time": 1.7353196144104004,
      "step": 24288
    },
    {
      "epoch": 2.3162841796875e-05,
      "step": 24288,
      "training_step_time": 8.575429916381836
    },
    {
      "epoch": 2.3163795471191408e-05,
      "model_forward_time": 1.708940029144287,
      "step": 24289
    },
    {
      "epoch": 2.3163795471191408e-05,
      "step": 24289,
      "training_step_time": 7.788584470748901
    },
    {
      "epoch": 2.316474914550781e-05,
      "grad_norm": 0.1932709813117981,
      "learning_rate": 9.016813185732974e-06,
      "loss": 0.0657,
      "step": 24290
    },
    {
      "epoch": 2.316474914550781e-05,
      "model_forward_time": 1.7350223064422607,
      "step": 24290
    },
    {
      "dynamics_loss_avg": 0.05716219078749418,
      "epoch": 2.316474914550781e-05,
      "step": 24290
    },
    {
      "action_loss_avg": 0.0053413627727422865,
      "epoch": 2.316474914550781e-05,
      "step": 24290
    },
    {
      "epoch": 2.316474914550781e-05,
      "step": 24290,
      "training_step_time": 8.659119844436646
    },
    {
      "epoch": 2.316570281982422e-05,
      "model_forward_time": 1.7202775478363037,
      "step": 24291
    },
    {
      "epoch": 2.316570281982422e-05,
      "step": 24291,
      "training_step_time": 7.845137357711792
    },
    {
      "epoch": 2.3166656494140627e-05,
      "model_forward_time": 1.732320785522461,
      "step": 24292
    },
    {
      "epoch": 2.3166656494140627e-05,
      "step": 24292,
      "training_step_time": 7.600287199020386
    },
    {
      "epoch": 2.316761016845703e-05,
      "model_forward_time": 1.718947410583496,
      "step": 24293
    },
    {
      "epoch": 2.316761016845703e-05,
      "step": 24293,
      "training_step_time": 7.170873403549194
    },
    {
      "epoch": 2.3168563842773438e-05,
      "model_forward_time": 1.7211527824401855,
      "step": 24294
    },
    {
      "epoch": 2.3168563842773438e-05,
      "step": 24294,
      "training_step_time": 7.6845316886901855
    },
    {
      "epoch": 2.3169517517089845e-05,
      "model_forward_time": 1.713667869567871,
      "step": 24295
    },
    {
      "epoch": 2.3169517517089845e-05,
      "step": 24295,
      "training_step_time": 9.599982738494873
    },
    {
      "epoch": 2.317047119140625e-05,
      "model_forward_time": 2.266794204711914,
      "step": 24296
    },
    {
      "epoch": 2.317047119140625e-05,
      "step": 24296,
      "training_step_time": 7.675420045852661
    },
    {
      "epoch": 2.3171424865722657e-05,
      "model_forward_time": 1.7178676128387451,
      "step": 24297
    },
    {
      "epoch": 2.3171424865722657e-05,
      "step": 24297,
      "training_step_time": 9.344475030899048
    },
    {
      "epoch": 2.317237854003906e-05,
      "model_forward_time": 1.7709107398986816,
      "step": 24298
    },
    {
      "epoch": 2.317237854003906e-05,
      "step": 24298,
      "training_step_time": 8.88007140159607
    },
    {
      "epoch": 2.3173332214355468e-05,
      "model_forward_time": 1.7210144996643066,
      "step": 24299
    },
    {
      "epoch": 2.3173332214355468e-05,
      "step": 24299,
      "training_step_time": 7.653924942016602
    },
    {
      "epoch": 2.3174285888671876e-05,
      "grad_norm": 0.21720007061958313,
      "learning_rate": 9.015828341188026e-06,
      "loss": 0.0684,
      "step": 24300
    },
    {
      "epoch": 2.3174285888671876e-05,
      "model_forward_time": 1.7116048336029053,
      "step": 24300
    },
    {
      "dynamics_loss_avg": 0.059175951220095155,
      "epoch": 2.3174285888671876e-05,
      "step": 24300
    },
    {
      "action_loss_avg": 0.0029604112438391893,
      "epoch": 2.3174285888671876e-05,
      "step": 24300
    },
    {
      "epoch": 2.3174285888671876e-05,
      "step": 24300,
      "training_step_time": 7.671877861022949
    },
    {
      "epoch": 2.317523956298828e-05,
      "model_forward_time": 1.7166695594787598,
      "step": 24301
    },
    {
      "epoch": 2.317523956298828e-05,
      "step": 24301,
      "training_step_time": 7.613860368728638
    },
    {
      "epoch": 2.3176193237304687e-05,
      "model_forward_time": 1.725114345550537,
      "step": 24302
    },
    {
      "epoch": 2.3176193237304687e-05,
      "step": 24302,
      "training_step_time": 7.744590520858765
    },
    {
      "epoch": 2.3177146911621095e-05,
      "model_forward_time": 1.7160253524780273,
      "step": 24303
    },
    {
      "epoch": 2.3177146911621095e-05,
      "step": 24303,
      "training_step_time": 7.605074882507324
    },
    {
      "epoch": 2.31781005859375e-05,
      "model_forward_time": 1.714564561843872,
      "step": 24304
    },
    {
      "epoch": 2.31781005859375e-05,
      "step": 24304,
      "training_step_time": 7.738391399383545
    },
    {
      "epoch": 2.3179054260253906e-05,
      "model_forward_time": 2.2901456356048584,
      "step": 24305
    },
    {
      "epoch": 2.3179054260253906e-05,
      "step": 24305,
      "training_step_time": 9.023206949234009
    },
    {
      "epoch": 2.3180007934570313e-05,
      "model_forward_time": 1.7109897136688232,
      "step": 24306
    },
    {
      "epoch": 2.3180007934570313e-05,
      "step": 24306,
      "training_step_time": 7.675048828125
    },
    {
      "epoch": 2.3180961608886717e-05,
      "model_forward_time": 1.7674505710601807,
      "step": 24307
    },
    {
      "epoch": 2.3180961608886717e-05,
      "step": 24307,
      "training_step_time": 9.366690635681152
    },
    {
      "epoch": 2.3181915283203125e-05,
      "model_forward_time": 1.7154626846313477,
      "step": 24308
    },
    {
      "epoch": 2.3181915283203125e-05,
      "step": 24308,
      "training_step_time": 7.644737958908081
    },
    {
      "epoch": 2.3182868957519532e-05,
      "model_forward_time": 1.3474721908569336,
      "step": 24309
    },
    {
      "epoch": 2.3182868957519532e-05,
      "step": 24309,
      "training_step_time": 7.7537031173706055
    },
    {
      "epoch": 2.3183822631835936e-05,
      "grad_norm": 0.18501949310302734,
      "learning_rate": 9.014843057478131e-06,
      "loss": 0.0645,
      "step": 24310
    },
    {
      "epoch": 2.3183822631835936e-05,
      "model_forward_time": 1.724790096282959,
      "step": 24310
    },
    {
      "dynamics_loss_avg": 0.07327438499778509,
      "epoch": 2.3183822631835936e-05,
      "step": 24310
    },
    {
      "action_loss_avg": 0.0039639735594391825,
      "epoch": 2.3183822631835936e-05,
      "step": 24310
    },
    {
      "epoch": 2.3183822631835936e-05,
      "step": 24310,
      "training_step_time": 8.57732605934143
    },
    {
      "epoch": 2.3184776306152344e-05,
      "model_forward_time": 1.7122952938079834,
      "step": 24311
    },
    {
      "epoch": 2.3184776306152344e-05,
      "step": 24311,
      "training_step_time": 7.775789260864258
    },
    {
      "epoch": 2.318572998046875e-05,
      "model_forward_time": 1.7104196548461914,
      "step": 24312
    },
    {
      "epoch": 2.318572998046875e-05,
      "step": 24312,
      "training_step_time": 7.57011604309082
    },
    {
      "epoch": 2.3186683654785155e-05,
      "model_forward_time": 1.727858543395996,
      "step": 24313
    },
    {
      "epoch": 2.3186683654785155e-05,
      "step": 24313,
      "training_step_time": 8.333791017532349
    },
    {
      "epoch": 2.3187637329101563e-05,
      "model_forward_time": 1.7140929698944092,
      "step": 24314
    },
    {
      "epoch": 2.3187637329101563e-05,
      "step": 24314,
      "training_step_time": 7.633755445480347
    },
    {
      "epoch": 2.318859100341797e-05,
      "model_forward_time": 2.2785465717315674,
      "step": 24315
    },
    {
      "epoch": 2.318859100341797e-05,
      "step": 24315,
      "training_step_time": 9.112396478652954
    },
    {
      "epoch": 2.3189544677734374e-05,
      "model_forward_time": 1.7489962577819824,
      "step": 24316
    },
    {
      "epoch": 2.3189544677734374e-05,
      "step": 24316,
      "training_step_time": 7.883121013641357
    },
    {
      "epoch": 2.319049835205078e-05,
      "model_forward_time": 1.7171235084533691,
      "step": 24317
    },
    {
      "epoch": 2.319049835205078e-05,
      "step": 24317,
      "training_step_time": 7.7720208168029785
    },
    {
      "epoch": 2.319145202636719e-05,
      "model_forward_time": 1.7364630699157715,
      "step": 24318
    },
    {
      "epoch": 2.319145202636719e-05,
      "step": 24318,
      "training_step_time": 7.5507800579071045
    },
    {
      "epoch": 2.3192405700683593e-05,
      "model_forward_time": 1.7159411907196045,
      "step": 24319
    },
    {
      "epoch": 2.3192405700683593e-05,
      "step": 24319,
      "training_step_time": 7.7979209423065186
    },
    {
      "epoch": 2.3193359375e-05,
      "grad_norm": 0.1694221943616867,
      "learning_rate": 9.013857334711033e-06,
      "loss": 0.0676,
      "step": 24320
    },
    {
      "epoch": 2.3193359375e-05,
      "model_forward_time": 1.7183277606964111,
      "step": 24320
    },
    {
      "dynamics_loss_avg": 0.06128615317866206,
      "epoch": 2.3193359375e-05,
      "step": 24320
    },
    {
      "action_loss_avg": 0.0021175373083679007,
      "epoch": 2.3193359375e-05,
      "step": 24320
    },
    {
      "epoch": 2.3193359375e-05,
      "step": 24320,
      "training_step_time": 7.786034345626831
    },
    {
      "epoch": 2.3194313049316408e-05,
      "model_forward_time": 1.359436273574829,
      "step": 24321
    },
    {
      "epoch": 2.3194313049316408e-05,
      "step": 24321,
      "training_step_time": 7.568830728530884
    },
    {
      "epoch": 2.3195266723632812e-05,
      "model_forward_time": 1.7111332416534424,
      "step": 24322
    },
    {
      "epoch": 2.3195266723632812e-05,
      "step": 24322,
      "training_step_time": 8.646625757217407
    },
    {
      "epoch": 2.319622039794922e-05,
      "model_forward_time": 1.7210915088653564,
      "step": 24323
    },
    {
      "epoch": 2.319622039794922e-05,
      "step": 24323,
      "training_step_time": 8.3756742477417
    },
    {
      "epoch": 2.3197174072265627e-05,
      "model_forward_time": 2.26257061958313,
      "step": 24324
    },
    {
      "epoch": 2.3197174072265627e-05,
      "step": 24324,
      "training_step_time": 7.689486265182495
    },
    {
      "epoch": 2.319812774658203e-05,
      "model_forward_time": 1.7721490859985352,
      "step": 24325
    },
    {
      "epoch": 2.319812774658203e-05,
      "step": 24325,
      "training_step_time": 7.833758354187012
    },
    {
      "epoch": 2.3199081420898438e-05,
      "model_forward_time": 1.7258741855621338,
      "step": 24326
    },
    {
      "epoch": 2.3199081420898438e-05,
      "step": 24326,
      "training_step_time": 7.811201333999634
    },
    {
      "epoch": 2.3200035095214845e-05,
      "model_forward_time": 1.7218799591064453,
      "step": 24327
    },
    {
      "epoch": 2.3200035095214845e-05,
      "step": 24327,
      "training_step_time": 7.749557971954346
    },
    {
      "epoch": 2.320098876953125e-05,
      "model_forward_time": 1.7186918258666992,
      "step": 24328
    },
    {
      "epoch": 2.320098876953125e-05,
      "step": 24328,
      "training_step_time": 7.707665205001831
    },
    {
      "epoch": 2.3201942443847657e-05,
      "model_forward_time": 1.720296859741211,
      "step": 24329
    },
    {
      "epoch": 2.3201942443847657e-05,
      "step": 24329,
      "training_step_time": 7.76310658454895
    },
    {
      "epoch": 2.320289611816406e-05,
      "grad_norm": 0.21937154233455658,
      "learning_rate": 9.012871172994533e-06,
      "loss": 0.063,
      "step": 24330
    },
    {
      "epoch": 2.320289611816406e-05,
      "model_forward_time": 1.72160005569458,
      "step": 24330
    },
    {
      "dynamics_loss_avg": 0.06863092631101608,
      "epoch": 2.320289611816406e-05,
      "step": 24330
    },
    {
      "action_loss_avg": 0.0035514947725459934,
      "epoch": 2.320289611816406e-05,
      "step": 24330
    },
    {
      "epoch": 2.320289611816406e-05,
      "step": 24330,
      "training_step_time": 7.565385818481445
    },
    {
      "epoch": 2.3203849792480468e-05,
      "model_forward_time": 1.7205348014831543,
      "step": 24331
    },
    {
      "epoch": 2.3203849792480468e-05,
      "step": 24331,
      "training_step_time": 8.452092409133911
    },
    {
      "epoch": 2.3204803466796876e-05,
      "model_forward_time": 1.710298776626587,
      "step": 24332
    },
    {
      "epoch": 2.3204803466796876e-05,
      "step": 24332,
      "training_step_time": 9.523292064666748
    },
    {
      "epoch": 2.320575714111328e-05,
      "model_forward_time": 2.2960095405578613,
      "step": 24333
    },
    {
      "epoch": 2.320575714111328e-05,
      "step": 24333,
      "training_step_time": 10.168517827987671
    },
    {
      "epoch": 2.3206710815429687e-05,
      "model_forward_time": 1.7666263580322266,
      "step": 24334
    },
    {
      "epoch": 2.3206710815429687e-05,
      "step": 24334,
      "training_step_time": 8.271069765090942
    },
    {
      "epoch": 2.3207664489746095e-05,
      "model_forward_time": 1.7249515056610107,
      "step": 24335
    },
    {
      "epoch": 2.3207664489746095e-05,
      "step": 24335,
      "training_step_time": 8.825704336166382
    },
    {
      "epoch": 2.32086181640625e-05,
      "model_forward_time": 1.715573787689209,
      "step": 24336
    },
    {
      "epoch": 2.32086181640625e-05,
      "step": 24336,
      "training_step_time": 8.332277536392212
    },
    {
      "epoch": 2.3209571838378906e-05,
      "model_forward_time": 1.7204155921936035,
      "step": 24337
    },
    {
      "epoch": 2.3209571838378906e-05,
      "step": 24337,
      "training_step_time": 7.700362920761108
    },
    {
      "epoch": 2.3210525512695313e-05,
      "model_forward_time": 1.71421217918396,
      "step": 24338
    },
    {
      "epoch": 2.3210525512695313e-05,
      "step": 24338,
      "training_step_time": 8.437250137329102
    },
    {
      "epoch": 2.3211479187011717e-05,
      "model_forward_time": 1.7131261825561523,
      "step": 24339
    },
    {
      "epoch": 2.3211479187011717e-05,
      "step": 24339,
      "training_step_time": 7.46963095664978
    },
    {
      "epoch": 2.3212432861328125e-05,
      "grad_norm": 0.13819043338298798,
      "learning_rate": 9.011884572436476e-06,
      "loss": 0.0661,
      "step": 24340
    },
    {
      "epoch": 2.3212432861328125e-05,
      "model_forward_time": 1.720820426940918,
      "step": 24340
    },
    {
      "dynamics_loss_avg": 0.06257977783679962,
      "epoch": 2.3212432861328125e-05,
      "step": 24340
    },
    {
      "action_loss_avg": 0.0029449866531649606,
      "epoch": 2.3212432861328125e-05,
      "step": 24340
    },
    {
      "epoch": 2.3212432861328125e-05,
      "step": 24340,
      "training_step_time": 7.69622540473938
    },
    {
      "epoch": 2.3213386535644532e-05,
      "model_forward_time": 1.7182199954986572,
      "step": 24341
    },
    {
      "epoch": 2.3213386535644532e-05,
      "step": 24341,
      "training_step_time": 8.5660240650177
    },
    {
      "epoch": 2.3214340209960936e-05,
      "model_forward_time": 2.2900962829589844,
      "step": 24342
    },
    {
      "epoch": 2.3214340209960936e-05,
      "step": 24342,
      "training_step_time": 9.298805713653564
    },
    {
      "epoch": 2.3215293884277344e-05,
      "model_forward_time": 1.7717368602752686,
      "step": 24343
    },
    {
      "epoch": 2.3215293884277344e-05,
      "step": 24343,
      "training_step_time": 8.15902829170227
    },
    {
      "epoch": 2.321624755859375e-05,
      "model_forward_time": 1.7189414501190186,
      "step": 24344
    },
    {
      "epoch": 2.321624755859375e-05,
      "step": 24344,
      "training_step_time": 7.946398496627808
    },
    {
      "epoch": 2.3217201232910155e-05,
      "model_forward_time": 1.7181801795959473,
      "step": 24345
    },
    {
      "epoch": 2.3217201232910155e-05,
      "step": 24345,
      "training_step_time": 7.803037643432617
    },
    {
      "epoch": 2.3218154907226563e-05,
      "model_forward_time": 1.7238211631774902,
      "step": 24346
    },
    {
      "epoch": 2.3218154907226563e-05,
      "step": 24346,
      "training_step_time": 7.71051287651062
    },
    {
      "epoch": 2.321910858154297e-05,
      "model_forward_time": 1.7236864566802979,
      "step": 24347
    },
    {
      "epoch": 2.321910858154297e-05,
      "step": 24347,
      "training_step_time": 7.781166315078735
    },
    {
      "epoch": 2.3220062255859374e-05,
      "model_forward_time": 1.7126455307006836,
      "step": 24348
    },
    {
      "epoch": 2.3220062255859374e-05,
      "step": 24348,
      "training_step_time": 7.763101816177368
    },
    {
      "epoch": 2.322101593017578e-05,
      "model_forward_time": 1.7228264808654785,
      "step": 24349
    },
    {
      "epoch": 2.322101593017578e-05,
      "step": 24349,
      "training_step_time": 7.666823863983154
    },
    {
      "epoch": 2.322196960449219e-05,
      "grad_norm": 0.17394909262657166,
      "learning_rate": 9.010897533144754e-06,
      "loss": 0.0621,
      "step": 24350
    },
    {
      "epoch": 2.322196960449219e-05,
      "model_forward_time": 1.7268505096435547,
      "step": 24350
    },
    {
      "dynamics_loss_avg": 0.044677579589188096,
      "epoch": 2.322196960449219e-05,
      "step": 24350
    },
    {
      "action_loss_avg": 0.0013805228052660822,
      "epoch": 2.322196960449219e-05,
      "step": 24350
    },
    {
      "epoch": 2.322196960449219e-05,
      "step": 24350,
      "training_step_time": 9.080735683441162
    },
    {
      "epoch": 2.3222923278808593e-05,
      "model_forward_time": 1.7174286842346191,
      "step": 24351
    },
    {
      "epoch": 2.3222923278808593e-05,
      "step": 24351,
      "training_step_time": 9.985842943191528
    },
    {
      "epoch": 2.3223876953125e-05,
      "model_forward_time": 1.7168176174163818,
      "step": 24352
    },
    {
      "epoch": 2.3223876953125e-05,
      "step": 24352,
      "training_step_time": 6.747920513153076
    },
    {
      "epoch": 2.3224830627441408e-05,
      "model_forward_time": 1.7205955982208252,
      "step": 24353
    },
    {
      "epoch": 2.3224830627441408e-05,
      "step": 24353,
      "training_step_time": 6.745771169662476
    },
    {
      "epoch": 2.3225784301757812e-05,
      "model_forward_time": 1.7231500148773193,
      "step": 24354
    },
    {
      "epoch": 2.3225784301757812e-05,
      "step": 24354,
      "training_step_time": 6.7231245040893555
    },
    {
      "epoch": 2.322673797607422e-05,
      "model_forward_time": 1.7233390808105469,
      "step": 24355
    },
    {
      "epoch": 2.322673797607422e-05,
      "step": 24355,
      "training_step_time": 6.740548849105835
    },
    {
      "epoch": 2.3227691650390627e-05,
      "model_forward_time": 1.717482089996338,
      "step": 24356
    },
    {
      "epoch": 2.3227691650390627e-05,
      "step": 24356,
      "training_step_time": 6.734065294265747
    },
    {
      "epoch": 2.322864532470703e-05,
      "model_forward_time": 1.715040683746338,
      "step": 24357
    },
    {
      "epoch": 2.322864532470703e-05,
      "step": 24357,
      "training_step_time": 6.7108683586120605
    },
    {
      "epoch": 2.3229598999023438e-05,
      "model_forward_time": 1.7189054489135742,
      "step": 24358
    },
    {
      "epoch": 2.3229598999023438e-05,
      "step": 24358,
      "training_step_time": 6.700240612030029
    },
    {
      "epoch": 2.3230552673339845e-05,
      "model_forward_time": 1.7155747413635254,
      "step": 24359
    },
    {
      "epoch": 2.3230552673339845e-05,
      "step": 24359,
      "training_step_time": 8.226723670959473
    },
    {
      "epoch": 2.323150634765625e-05,
      "grad_norm": 0.1737658530473709,
      "learning_rate": 9.009910055227306e-06,
      "loss": 0.0595,
      "step": 24360
    },
    {
      "epoch": 2.323150634765625e-05,
      "model_forward_time": 1.7074408531188965,
      "step": 24360
    },
    {
      "dynamics_loss_avg": 0.06517473589628935,
      "epoch": 2.323150634765625e-05,
      "step": 24360
    },
    {
      "action_loss_avg": 0.003002109908265993,
      "epoch": 2.323150634765625e-05,
      "step": 24360
    },
    {
      "epoch": 2.323150634765625e-05,
      "step": 24360,
      "training_step_time": 6.990416049957275
    },
    {
      "epoch": 2.3232460021972657e-05,
      "model_forward_time": 1.693694829940796,
      "step": 24361
    },
    {
      "epoch": 2.3232460021972657e-05,
      "step": 24361,
      "training_step_time": 7.777284383773804
    },
    {
      "epoch": 2.323341369628906e-05,
      "model_forward_time": 1.6874659061431885,
      "step": 24362
    },
    {
      "epoch": 2.323341369628906e-05,
      "step": 24362,
      "training_step_time": 6.743091821670532
    },
    {
      "epoch": 2.323436737060547e-05,
      "model_forward_time": 1.6924421787261963,
      "step": 24363
    },
    {
      "epoch": 2.323436737060547e-05,
      "step": 24363,
      "training_step_time": 6.734239816665649
    },
    {
      "epoch": 2.3235321044921876e-05,
      "model_forward_time": 1.6900315284729004,
      "step": 24364
    },
    {
      "epoch": 2.3235321044921876e-05,
      "step": 24364,
      "training_step_time": 6.767023324966431
    },
    {
      "epoch": 2.323627471923828e-05,
      "model_forward_time": 1.6879734992980957,
      "step": 24365
    },
    {
      "epoch": 2.323627471923828e-05,
      "step": 24365,
      "training_step_time": 6.746337890625
    },
    {
      "epoch": 2.3237228393554687e-05,
      "model_forward_time": 1.6916589736938477,
      "step": 24366
    },
    {
      "epoch": 2.3237228393554687e-05,
      "step": 24366,
      "training_step_time": 6.767857074737549
    },
    {
      "epoch": 2.3238182067871095e-05,
      "model_forward_time": 1.6893303394317627,
      "step": 24367
    },
    {
      "epoch": 2.3238182067871095e-05,
      "step": 24367,
      "training_step_time": 6.74107813835144
    },
    {
      "epoch": 2.32391357421875e-05,
      "model_forward_time": 1.686631441116333,
      "step": 24368
    },
    {
      "epoch": 2.32391357421875e-05,
      "step": 24368,
      "training_step_time": 7.311823844909668
    },
    {
      "epoch": 2.3240089416503906e-05,
      "model_forward_time": 1.723930835723877,
      "step": 24369
    },
    {
      "epoch": 2.3240089416503906e-05,
      "step": 24369,
      "training_step_time": 7.565343618392944
    },
    {
      "epoch": 2.3241043090820313e-05,
      "grad_norm": 0.1902020126581192,
      "learning_rate": 9.008922138792125e-06,
      "loss": 0.0628,
      "step": 24370
    },
    {
      "epoch": 2.3241043090820313e-05,
      "model_forward_time": 1.6855907440185547,
      "step": 24370
    },
    {
      "dynamics_loss_avg": 0.06683089453727006,
      "epoch": 2.3241043090820313e-05,
      "step": 24370
    },
    {
      "action_loss_avg": 0.005366589000914246,
      "epoch": 2.3241043090820313e-05,
      "step": 24370
    },
    {
      "epoch": 2.3241043090820313e-05,
      "step": 24370,
      "training_step_time": 7.816865682601929
    },
    {
      "epoch": 2.3241996765136717e-05,
      "model_forward_time": 1.7052485942840576,
      "step": 24371
    },
    {
      "epoch": 2.3241996765136717e-05,
      "step": 24371,
      "training_step_time": 6.738396644592285
    },
    {
      "epoch": 2.3242950439453125e-05,
      "model_forward_time": 1.690932273864746,
      "step": 24372
    },
    {
      "epoch": 2.3242950439453125e-05,
      "step": 24372,
      "training_step_time": 6.776194095611572
    },
    {
      "epoch": 2.3243904113769532e-05,
      "model_forward_time": 1.6955111026763916,
      "step": 24373
    },
    {
      "epoch": 2.3243904113769532e-05,
      "step": 24373,
      "training_step_time": 6.8057286739349365
    },
    {
      "epoch": 2.3244857788085936e-05,
      "model_forward_time": 1.7442631721496582,
      "step": 24374
    },
    {
      "epoch": 2.3244857788085936e-05,
      "step": 24374,
      "training_step_time": 6.857757329940796
    },
    {
      "epoch": 2.3245811462402344e-05,
      "model_forward_time": 1.7014870643615723,
      "step": 24375
    },
    {
      "epoch": 2.3245811462402344e-05,
      "step": 24375,
      "training_step_time": 6.785830497741699
    },
    {
      "epoch": 2.324676513671875e-05,
      "model_forward_time": 1.705702781677246,
      "step": 24376
    },
    {
      "epoch": 2.324676513671875e-05,
      "step": 24376,
      "training_step_time": 6.779160499572754
    },
    {
      "epoch": 2.3247718811035155e-05,
      "model_forward_time": 1.7222111225128174,
      "step": 24377
    },
    {
      "epoch": 2.3247718811035155e-05,
      "step": 24377,
      "training_step_time": 6.819396018981934
    },
    {
      "epoch": 2.3248672485351563e-05,
      "model_forward_time": 2.240994453430176,
      "step": 24378
    },
    {
      "epoch": 2.3248672485351563e-05,
      "step": 24378,
      "training_step_time": 7.585907936096191
    },
    {
      "epoch": 2.324962615966797e-05,
      "model_forward_time": 1.697601556777954,
      "step": 24379
    },
    {
      "epoch": 2.324962615966797e-05,
      "step": 24379,
      "training_step_time": 7.772137641906738
    },
    {
      "epoch": 2.3250579833984374e-05,
      "grad_norm": 0.18058253824710846,
      "learning_rate": 9.007933783947245e-06,
      "loss": 0.0617,
      "step": 24380
    },
    {
      "epoch": 2.3250579833984374e-05,
      "model_forward_time": 1.7259228229522705,
      "step": 24380
    },
    {
      "dynamics_loss_avg": 0.056797729898244144,
      "epoch": 2.3250579833984374e-05,
      "step": 24380
    },
    {
      "action_loss_avg": 0.0023854793515056373,
      "epoch": 2.3250579833984374e-05,
      "step": 24380
    },
    {
      "epoch": 2.3250579833984374e-05,
      "step": 24380,
      "training_step_time": 7.224889755249023
    },
    {
      "epoch": 2.325153350830078e-05,
      "model_forward_time": 1.699422836303711,
      "step": 24381
    },
    {
      "epoch": 2.325153350830078e-05,
      "step": 24381,
      "training_step_time": 6.8781561851501465
    },
    {
      "epoch": 2.325248718261719e-05,
      "model_forward_time": 1.6975529193878174,
      "step": 24382
    },
    {
      "epoch": 2.325248718261719e-05,
      "step": 24382,
      "training_step_time": 6.7925732135772705
    },
    {
      "epoch": 2.3253440856933593e-05,
      "model_forward_time": 1.6953849792480469,
      "step": 24383
    },
    {
      "epoch": 2.3253440856933593e-05,
      "step": 24383,
      "training_step_time": 6.791845798492432
    },
    {
      "epoch": 2.325439453125e-05,
      "model_forward_time": 1.6974658966064453,
      "step": 24384
    },
    {
      "epoch": 2.325439453125e-05,
      "step": 24384,
      "training_step_time": 6.748638153076172
    },
    {
      "epoch": 2.3255348205566408e-05,
      "model_forward_time": 1.6936218738555908,
      "step": 24385
    },
    {
      "epoch": 2.3255348205566408e-05,
      "step": 24385,
      "training_step_time": 6.785263776779175
    },
    {
      "epoch": 2.3256301879882812e-05,
      "model_forward_time": 1.6992297172546387,
      "step": 24386
    },
    {
      "epoch": 2.3256301879882812e-05,
      "step": 24386,
      "training_step_time": 6.81618332862854
    },
    {
      "epoch": 2.325725555419922e-05,
      "model_forward_time": 2.224504232406616,
      "step": 24387
    },
    {
      "epoch": 2.325725555419922e-05,
      "step": 24387,
      "training_step_time": 8.300185680389404
    },
    {
      "epoch": 2.3258209228515627e-05,
      "model_forward_time": 1.6977546215057373,
      "step": 24388
    },
    {
      "epoch": 2.3258209228515627e-05,
      "step": 24388,
      "training_step_time": 10.74692440032959
    },
    {
      "epoch": 2.325916290283203e-05,
      "model_forward_time": 1.6945500373840332,
      "step": 24389
    },
    {
      "epoch": 2.325916290283203e-05,
      "step": 24389,
      "training_step_time": 7.571634531021118
    },
    {
      "epoch": 2.3260116577148438e-05,
      "grad_norm": 0.12508638203144073,
      "learning_rate": 9.006944990800753e-06,
      "loss": 0.0631,
      "step": 24390
    },
    {
      "epoch": 2.3260116577148438e-05,
      "model_forward_time": 1.7046244144439697,
      "step": 24390
    },
    {
      "dynamics_loss_avg": 0.06389503628015518,
      "epoch": 2.3260116577148438e-05,
      "step": 24390
    },
    {
      "action_loss_avg": 0.00259675569832325,
      "epoch": 2.3260116577148438e-05,
      "step": 24390
    },
    {
      "epoch": 2.3260116577148438e-05,
      "step": 24390,
      "training_step_time": 6.770596265792847
    },
    {
      "epoch": 2.3261070251464842e-05,
      "model_forward_time": 1.7026588916778564,
      "step": 24391
    },
    {
      "epoch": 2.3261070251464842e-05,
      "step": 24391,
      "training_step_time": 6.811375379562378
    },
    {
      "epoch": 2.326202392578125e-05,
      "model_forward_time": 1.6849493980407715,
      "step": 24392
    },
    {
      "epoch": 2.326202392578125e-05,
      "step": 24392,
      "training_step_time": 6.81619930267334
    },
    {
      "epoch": 2.3262977600097657e-05,
      "model_forward_time": 1.6972403526306152,
      "step": 24393
    },
    {
      "epoch": 2.3262977600097657e-05,
      "step": 24393,
      "training_step_time": 6.762872934341431
    },
    {
      "epoch": 2.326393127441406e-05,
      "model_forward_time": 1.6953227519989014,
      "step": 24394
    },
    {
      "epoch": 2.326393127441406e-05,
      "step": 24394,
      "training_step_time": 6.7688047885894775
    },
    {
      "epoch": 2.326488494873047e-05,
      "model_forward_time": 1.6961345672607422,
      "step": 24395
    },
    {
      "epoch": 2.326488494873047e-05,
      "step": 24395,
      "training_step_time": 6.7616002559661865
    },
    {
      "epoch": 2.3265838623046876e-05,
      "model_forward_time": 2.2318313121795654,
      "step": 24396
    },
    {
      "epoch": 2.3265838623046876e-05,
      "step": 24396,
      "training_step_time": 7.6056530475616455
    },
    {
      "epoch": 2.326679229736328e-05,
      "model_forward_time": 1.750180721282959,
      "step": 24397
    },
    {
      "epoch": 2.326679229736328e-05,
      "step": 24397,
      "training_step_time": 9.806729793548584
    },
    {
      "epoch": 2.3267745971679687e-05,
      "model_forward_time": 1.699228286743164,
      "step": 24398
    },
    {
      "epoch": 2.3267745971679687e-05,
      "step": 24398,
      "training_step_time": 7.676759958267212
    },
    {
      "epoch": 2.3268699645996095e-05,
      "model_forward_time": 1.6984062194824219,
      "step": 24399
    },
    {
      "epoch": 2.3268699645996095e-05,
      "step": 24399,
      "training_step_time": 6.765057802200317
    },
    {
      "epoch": 2.32696533203125e-05,
      "grad_norm": 0.16781507432460785,
      "learning_rate": 9.005955759460779e-06,
      "loss": 0.0643,
      "step": 24400
    },
    {
      "epoch": 2.32696533203125e-05,
      "model_forward_time": 1.6983730792999268,
      "step": 24400
    },
    {
      "dynamics_loss_avg": 0.06175668686628342,
      "epoch": 2.32696533203125e-05,
      "step": 24400
    },
    {
      "action_loss_avg": 0.003472151863388717,
      "epoch": 2.32696533203125e-05,
      "step": 24400
    },
    {
      "epoch": 2.32696533203125e-05,
      "step": 24400,
      "training_step_time": 6.742882251739502
    },
    {
      "epoch": 2.3270606994628906e-05,
      "model_forward_time": 1.6952049732208252,
      "step": 24401
    },
    {
      "epoch": 2.3270606994628906e-05,
      "step": 24401,
      "training_step_time": 6.742860794067383
    },
    {
      "epoch": 2.3271560668945313e-05,
      "model_forward_time": 1.7010486125946045,
      "step": 24402
    },
    {
      "epoch": 2.3271560668945313e-05,
      "step": 24402,
      "training_step_time": 6.7810423374176025
    },
    {
      "epoch": 2.3272514343261718e-05,
      "model_forward_time": 1.701632022857666,
      "step": 24403
    },
    {
      "epoch": 2.3272514343261718e-05,
      "step": 24403,
      "training_step_time": 6.816353797912598
    },
    {
      "epoch": 2.3273468017578125e-05,
      "model_forward_time": 1.7036340236663818,
      "step": 24404
    },
    {
      "epoch": 2.3273468017578125e-05,
      "step": 24404,
      "training_step_time": 6.760014295578003
    },
    {
      "epoch": 2.3274421691894532e-05,
      "model_forward_time": 2.2282655239105225,
      "step": 24405
    },
    {
      "epoch": 2.3274421691894532e-05,
      "step": 24405,
      "training_step_time": 7.566951274871826
    },
    {
      "epoch": 2.3275375366210936e-05,
      "model_forward_time": 1.7484493255615234,
      "step": 24406
    },
    {
      "epoch": 2.3275375366210936e-05,
      "step": 24406,
      "training_step_time": 9.818815231323242
    },
    {
      "epoch": 2.3276329040527344e-05,
      "model_forward_time": 1.6960203647613525,
      "step": 24407
    },
    {
      "epoch": 2.3276329040527344e-05,
      "step": 24407,
      "training_step_time": 8.143500089645386
    },
    {
      "epoch": 2.327728271484375e-05,
      "model_forward_time": 1.6915466785430908,
      "step": 24408
    },
    {
      "epoch": 2.327728271484375e-05,
      "step": 24408,
      "training_step_time": 6.777391672134399
    },
    {
      "epoch": 2.3278236389160155e-05,
      "model_forward_time": 1.701247215270996,
      "step": 24409
    },
    {
      "epoch": 2.3278236389160155e-05,
      "step": 24409,
      "training_step_time": 6.741654396057129
    },
    {
      "epoch": 2.3279190063476563e-05,
      "grad_norm": 0.1701127588748932,
      "learning_rate": 9.004966090035507e-06,
      "loss": 0.067,
      "step": 24410
    },
    {
      "epoch": 2.3279190063476563e-05,
      "model_forward_time": 1.756673812866211,
      "step": 24410
    },
    {
      "dynamics_loss_avg": 0.055081347841769455,
      "epoch": 2.3279190063476563e-05,
      "step": 24410
    },
    {
      "action_loss_avg": 0.002811649558134377,
      "epoch": 2.3279190063476563e-05,
      "step": 24410
    },
    {
      "epoch": 2.3279190063476563e-05,
      "step": 24410,
      "training_step_time": 6.743959188461304
    },
    {
      "epoch": 2.328014373779297e-05,
      "model_forward_time": 1.723771095275879,
      "step": 24411
    },
    {
      "epoch": 2.328014373779297e-05,
      "step": 24411,
      "training_step_time": 6.737893342971802
    },
    {
      "epoch": 2.3281097412109374e-05,
      "model_forward_time": 1.694852352142334,
      "step": 24412
    },
    {
      "epoch": 2.3281097412109374e-05,
      "step": 24412,
      "training_step_time": 6.757298231124878
    },
    {
      "epoch": 2.328205108642578e-05,
      "model_forward_time": 1.694697618484497,
      "step": 24413
    },
    {
      "epoch": 2.328205108642578e-05,
      "step": 24413,
      "training_step_time": 6.765542268753052
    },
    {
      "epoch": 2.328300476074219e-05,
      "model_forward_time": 2.220987558364868,
      "step": 24414
    },
    {
      "epoch": 2.328300476074219e-05,
      "step": 24414,
      "training_step_time": 7.494095087051392
    },
    {
      "epoch": 2.3283958435058593e-05,
      "model_forward_time": 1.7328383922576904,
      "step": 24415
    },
    {
      "epoch": 2.3283958435058593e-05,
      "step": 24415,
      "training_step_time": 9.395440816879272
    },
    {
      "epoch": 2.3284912109375e-05,
      "model_forward_time": 1.7651591300964355,
      "step": 24416
    },
    {
      "epoch": 2.3284912109375e-05,
      "step": 24416,
      "training_step_time": 9.161900043487549
    },
    {
      "epoch": 2.3285865783691408e-05,
      "model_forward_time": 1.7144014835357666,
      "step": 24417
    },
    {
      "epoch": 2.3285865783691408e-05,
      "step": 24417,
      "training_step_time": 7.376113176345825
    },
    {
      "epoch": 2.3286819458007812e-05,
      "model_forward_time": 1.7009150981903076,
      "step": 24418
    },
    {
      "epoch": 2.3286819458007812e-05,
      "step": 24418,
      "training_step_time": 6.763562440872192
    },
    {
      "epoch": 2.328777313232422e-05,
      "model_forward_time": 1.715099811553955,
      "step": 24419
    },
    {
      "epoch": 2.328777313232422e-05,
      "step": 24419,
      "training_step_time": 6.680844783782959
    },
    {
      "epoch": 2.3288726806640627e-05,
      "grad_norm": 0.15648359060287476,
      "learning_rate": 9.003975982633166e-06,
      "loss": 0.0631,
      "step": 24420
    },
    {
      "epoch": 2.3288726806640627e-05,
      "model_forward_time": 1.704448938369751,
      "step": 24420
    },
    {
      "dynamics_loss_avg": 0.06712592486292124,
      "epoch": 2.3288726806640627e-05,
      "step": 24420
    },
    {
      "action_loss_avg": 0.002750743634533137,
      "epoch": 2.3288726806640627e-05,
      "step": 24420
    },
    {
      "epoch": 2.3288726806640627e-05,
      "step": 24420,
      "training_step_time": 6.748265504837036
    },
    {
      "epoch": 2.328968048095703e-05,
      "model_forward_time": 1.7070345878601074,
      "step": 24421
    },
    {
      "epoch": 2.328968048095703e-05,
      "step": 24421,
      "training_step_time": 6.780461549758911
    },
    {
      "epoch": 2.3290634155273438e-05,
      "model_forward_time": 1.711233377456665,
      "step": 24422
    },
    {
      "epoch": 2.3290634155273438e-05,
      "step": 24422,
      "training_step_time": 6.769613265991211
    },
    {
      "epoch": 2.3291587829589842e-05,
      "model_forward_time": 1.7016334533691406,
      "step": 24423
    },
    {
      "epoch": 2.3291587829589842e-05,
      "step": 24423,
      "training_step_time": 7.753036737442017
    },
    {
      "epoch": 2.329254150390625e-05,
      "model_forward_time": 1.697838306427002,
      "step": 24424
    },
    {
      "epoch": 2.329254150390625e-05,
      "step": 24424,
      "training_step_time": 8.73842716217041
    },
    {
      "epoch": 2.3293495178222657e-05,
      "model_forward_time": 1.761603832244873,
      "step": 24425
    },
    {
      "epoch": 2.3293495178222657e-05,
      "step": 24425,
      "training_step_time": 9.154435396194458
    },
    {
      "epoch": 2.329444885253906e-05,
      "model_forward_time": 1.7031524181365967,
      "step": 24426
    },
    {
      "epoch": 2.329444885253906e-05,
      "step": 24426,
      "training_step_time": 7.590683221817017
    },
    {
      "epoch": 2.329540252685547e-05,
      "model_forward_time": 1.7071585655212402,
      "step": 24427
    },
    {
      "epoch": 2.329540252685547e-05,
      "step": 24427,
      "training_step_time": 6.744060039520264
    },
    {
      "epoch": 2.3296356201171876e-05,
      "model_forward_time": 1.7007725238800049,
      "step": 24428
    },
    {
      "epoch": 2.3296356201171876e-05,
      "step": 24428,
      "training_step_time": 6.760143756866455
    },
    {
      "epoch": 2.329730987548828e-05,
      "model_forward_time": 1.702580451965332,
      "step": 24429
    },
    {
      "epoch": 2.329730987548828e-05,
      "step": 24429,
      "training_step_time": 6.759416818618774
    },
    {
      "epoch": 2.3298263549804687e-05,
      "grad_norm": 0.13490773737430573,
      "learning_rate": 9.00298543736203e-06,
      "loss": 0.0654,
      "step": 24430
    },
    {
      "epoch": 2.3298263549804687e-05,
      "model_forward_time": 1.7037458419799805,
      "step": 24430
    },
    {
      "dynamics_loss_avg": 0.055452752579003575,
      "epoch": 2.3298263549804687e-05,
      "step": 24430
    },
    {
      "action_loss_avg": 0.0028262509062187747,
      "epoch": 2.3298263549804687e-05,
      "step": 24430
    },
    {
      "epoch": 2.3298263549804687e-05,
      "step": 24430,
      "training_step_time": 6.750315427780151
    },
    {
      "epoch": 2.3299217224121095e-05,
      "model_forward_time": 1.700312852859497,
      "step": 24431
    },
    {
      "epoch": 2.3299217224121095e-05,
      "step": 24431,
      "training_step_time": 6.736889600753784
    },
    {
      "epoch": 2.33001708984375e-05,
      "model_forward_time": 1.6971909999847412,
      "step": 24432
    },
    {
      "epoch": 2.33001708984375e-05,
      "step": 24432,
      "training_step_time": 6.76305079460144
    },
    {
      "epoch": 2.3301124572753906e-05,
      "model_forward_time": 2.2292189598083496,
      "step": 24433
    },
    {
      "epoch": 2.3301124572753906e-05,
      "step": 24433,
      "training_step_time": 9.476414203643799
    },
    {
      "epoch": 2.3302078247070314e-05,
      "model_forward_time": 1.767228603363037,
      "step": 24434
    },
    {
      "epoch": 2.3302078247070314e-05,
      "step": 24434,
      "training_step_time": 11.003955602645874
    },
    {
      "epoch": 2.3303031921386718e-05,
      "model_forward_time": 1.7033600807189941,
      "step": 24435
    },
    {
      "epoch": 2.3303031921386718e-05,
      "step": 24435,
      "training_step_time": 7.637269496917725
    },
    {
      "epoch": 2.3303985595703125e-05,
      "model_forward_time": 1.700756549835205,
      "step": 24436
    },
    {
      "epoch": 2.3303985595703125e-05,
      "step": 24436,
      "training_step_time": 6.775935411453247
    },
    {
      "epoch": 2.3304939270019532e-05,
      "model_forward_time": 1.7082035541534424,
      "step": 24437
    },
    {
      "epoch": 2.3304939270019532e-05,
      "step": 24437,
      "training_step_time": 6.70975661277771
    },
    {
      "epoch": 2.3305892944335936e-05,
      "model_forward_time": 1.7060065269470215,
      "step": 24438
    },
    {
      "epoch": 2.3305892944335936e-05,
      "step": 24438,
      "training_step_time": 6.744980573654175
    },
    {
      "epoch": 2.3306846618652344e-05,
      "model_forward_time": 1.7004263401031494,
      "step": 24439
    },
    {
      "epoch": 2.3306846618652344e-05,
      "step": 24439,
      "training_step_time": 6.7076191902160645
    },
    {
      "epoch": 2.330780029296875e-05,
      "grad_norm": 0.15097612142562866,
      "learning_rate": 9.001994454330428e-06,
      "loss": 0.0661,
      "step": 24440
    },
    {
      "epoch": 2.330780029296875e-05,
      "model_forward_time": 1.7011704444885254,
      "step": 24440
    },
    {
      "dynamics_loss_avg": 0.05483628418296575,
      "epoch": 2.330780029296875e-05,
      "step": 24440
    },
    {
      "action_loss_avg": 0.0018146994931157677,
      "epoch": 2.330780029296875e-05,
      "step": 24440
    },
    {
      "epoch": 2.330780029296875e-05,
      "step": 24440,
      "training_step_time": 6.712047576904297
    },
    {
      "epoch": 2.3308753967285155e-05,
      "model_forward_time": 1.6954503059387207,
      "step": 24441
    },
    {
      "epoch": 2.3308753967285155e-05,
      "step": 24441,
      "training_step_time": 6.758099794387817
    },
    {
      "epoch": 2.3309707641601563e-05,
      "model_forward_time": 2.2459163665771484,
      "step": 24442
    },
    {
      "epoch": 2.3309707641601563e-05,
      "step": 24442,
      "training_step_time": 8.771141767501831
    },
    {
      "epoch": 2.331066131591797e-05,
      "model_forward_time": 1.750473976135254,
      "step": 24443
    },
    {
      "epoch": 2.331066131591797e-05,
      "step": 24443,
      "training_step_time": 9.425965785980225
    },
    {
      "epoch": 2.3311614990234374e-05,
      "model_forward_time": 1.7743139266967773,
      "step": 24444
    },
    {
      "epoch": 2.3311614990234374e-05,
      "step": 24444,
      "training_step_time": 7.714483261108398
    },
    {
      "epoch": 2.331256866455078e-05,
      "model_forward_time": 1.7024617195129395,
      "step": 24445
    },
    {
      "epoch": 2.331256866455078e-05,
      "step": 24445,
      "training_step_time": 7.445594787597656
    },
    {
      "epoch": 2.331352233886719e-05,
      "model_forward_time": 1.7065589427947998,
      "step": 24446
    },
    {
      "epoch": 2.331352233886719e-05,
      "step": 24446,
      "training_step_time": 6.755941867828369
    },
    {
      "epoch": 2.3314476013183593e-05,
      "model_forward_time": 1.7065708637237549,
      "step": 24447
    },
    {
      "epoch": 2.3314476013183593e-05,
      "step": 24447,
      "training_step_time": 6.739386081695557
    },
    {
      "epoch": 2.33154296875e-05,
      "model_forward_time": 1.7029192447662354,
      "step": 24448
    },
    {
      "epoch": 2.33154296875e-05,
      "step": 24448,
      "training_step_time": 6.734646320343018
    },
    {
      "epoch": 2.3316383361816408e-05,
      "model_forward_time": 1.7115261554718018,
      "step": 24449
    },
    {
      "epoch": 2.3316383361816408e-05,
      "step": 24449,
      "training_step_time": 6.747603416442871
    },
    {
      "epoch": 2.3317337036132812e-05,
      "grad_norm": 0.11417398601770401,
      "learning_rate": 9.001003033646728e-06,
      "loss": 0.0622,
      "step": 24450
    },
    {
      "epoch": 2.3317337036132812e-05,
      "model_forward_time": 1.7053420543670654,
      "step": 24450
    },
    {
      "dynamics_loss_avg": 0.07033118773251772,
      "epoch": 2.3317337036132812e-05,
      "step": 24450
    },
    {
      "action_loss_avg": 0.0030311131617054344,
      "epoch": 2.3317337036132812e-05,
      "step": 24450
    },
    {
      "epoch": 2.3317337036132812e-05,
      "step": 24450,
      "training_step_time": 6.733277797698975
    },
    {
      "epoch": 2.331829071044922e-05,
      "model_forward_time": 2.220280885696411,
      "step": 24451
    },
    {
      "epoch": 2.331829071044922e-05,
      "step": 24451,
      "training_step_time": 8.723304986953735
    },
    {
      "epoch": 2.3319244384765627e-05,
      "model_forward_time": 1.7068190574645996,
      "step": 24452
    },
    {
      "epoch": 2.3319244384765627e-05,
      "step": 24452,
      "training_step_time": 8.711326122283936
    },
    {
      "epoch": 2.332019805908203e-05,
      "model_forward_time": 1.7486262321472168,
      "step": 24453
    },
    {
      "epoch": 2.332019805908203e-05,
      "step": 24453,
      "training_step_time": 7.6872642040252686
    },
    {
      "epoch": 2.3321151733398438e-05,
      "model_forward_time": 1.6926579475402832,
      "step": 24454
    },
    {
      "epoch": 2.3321151733398438e-05,
      "step": 24454,
      "training_step_time": 7.75401496887207
    },
    {
      "epoch": 2.3322105407714842e-05,
      "model_forward_time": 1.7126572132110596,
      "step": 24455
    },
    {
      "epoch": 2.3322105407714842e-05,
      "step": 24455,
      "training_step_time": 6.812075138092041
    },
    {
      "epoch": 2.332305908203125e-05,
      "model_forward_time": 1.7492411136627197,
      "step": 24456
    },
    {
      "epoch": 2.332305908203125e-05,
      "step": 24456,
      "training_step_time": 6.750744104385376
    },
    {
      "epoch": 2.3324012756347657e-05,
      "model_forward_time": 1.7039952278137207,
      "step": 24457
    },
    {
      "epoch": 2.3324012756347657e-05,
      "step": 24457,
      "training_step_time": 6.7716145515441895
    },
    {
      "epoch": 2.332496643066406e-05,
      "model_forward_time": 1.6904852390289307,
      "step": 24458
    },
    {
      "epoch": 2.332496643066406e-05,
      "step": 24458,
      "training_step_time": 6.727999448776245
    },
    {
      "epoch": 2.332592010498047e-05,
      "model_forward_time": 1.6746594905853271,
      "step": 24459
    },
    {
      "epoch": 2.332592010498047e-05,
      "step": 24459,
      "training_step_time": 6.752686262130737
    },
    {
      "epoch": 2.3326873779296876e-05,
      "grad_norm": 0.21598480641841888,
      "learning_rate": 9.00001117541935e-06,
      "loss": 0.0618,
      "step": 24460
    },
    {
      "epoch": 2.3326873779296876e-05,
      "model_forward_time": 2.236837148666382,
      "step": 24460
    },
    {
      "dynamics_loss_avg": 0.05057541560381651,
      "epoch": 2.3326873779296876e-05,
      "step": 24460
    },
    {
      "action_loss_avg": 0.002524622739292681,
      "epoch": 2.3326873779296876e-05,
      "step": 24460
    },
    {
      "epoch": 2.3326873779296876e-05,
      "step": 24460,
      "training_step_time": 9.41415786743164
    },
    {
      "epoch": 2.332782745361328e-05,
      "model_forward_time": 1.6802740097045898,
      "step": 24461
    },
    {
      "epoch": 2.332782745361328e-05,
      "step": 24461,
      "training_step_time": 8.520119667053223
    },
    {
      "epoch": 2.3328781127929687e-05,
      "model_forward_time": 1.683699131011963,
      "step": 24462
    },
    {
      "epoch": 2.3328781127929687e-05,
      "step": 24462,
      "training_step_time": 8.38204550743103
    },
    {
      "epoch": 2.3329734802246095e-05,
      "model_forward_time": 1.6915771961212158,
      "step": 24463
    },
    {
      "epoch": 2.3329734802246095e-05,
      "step": 24463,
      "training_step_time": 7.760433912277222
    },
    {
      "epoch": 2.33306884765625e-05,
      "model_forward_time": 1.690699577331543,
      "step": 24464
    },
    {
      "epoch": 2.33306884765625e-05,
      "step": 24464,
      "training_step_time": 6.722613096237183
    },
    {
      "epoch": 2.3331642150878906e-05,
      "model_forward_time": 1.6959452629089355,
      "step": 24465
    },
    {
      "epoch": 2.3331642150878906e-05,
      "step": 24465,
      "training_step_time": 6.749753952026367
    },
    {
      "epoch": 2.3332595825195314e-05,
      "model_forward_time": 1.687190294265747,
      "step": 24466
    },
    {
      "epoch": 2.3332595825195314e-05,
      "step": 24466,
      "training_step_time": 6.780924558639526
    },
    {
      "epoch": 2.3333549499511718e-05,
      "model_forward_time": 1.6891417503356934,
      "step": 24467
    },
    {
      "epoch": 2.3333549499511718e-05,
      "step": 24467,
      "training_step_time": 6.780468702316284
    },
    {
      "epoch": 2.3334503173828125e-05,
      "model_forward_time": 1.6878843307495117,
      "step": 24468
    },
    {
      "epoch": 2.3334503173828125e-05,
      "step": 24468,
      "training_step_time": 6.735990285873413
    },
    {
      "epoch": 2.3335456848144532e-05,
      "model_forward_time": 2.2163515090942383,
      "step": 24469
    },
    {
      "epoch": 2.3335456848144532e-05,
      "step": 24469,
      "training_step_time": 8.527497053146362
    },
    {
      "epoch": 2.3336410522460936e-05,
      "grad_norm": 0.21132485568523407,
      "learning_rate": 8.999018879756765e-06,
      "loss": 0.0639,
      "step": 24470
    },
    {
      "epoch": 2.3336410522460936e-05,
      "model_forward_time": 1.6843879222869873,
      "step": 24470
    },
    {
      "dynamics_loss_avg": 0.06799098439514636,
      "epoch": 2.3336410522460936e-05,
      "step": 24470
    },
    {
      "action_loss_avg": 0.003452114120591432,
      "epoch": 2.3336410522460936e-05,
      "step": 24470
    },
    {
      "epoch": 2.3336410522460936e-05,
      "step": 24470,
      "training_step_time": 9.83154010772705
    },
    {
      "epoch": 2.3337364196777344e-05,
      "model_forward_time": 1.6889739036560059,
      "step": 24471
    },
    {
      "epoch": 2.3337364196777344e-05,
      "step": 24471,
      "training_step_time": 8.80657434463501
    },
    {
      "epoch": 2.333831787109375e-05,
      "model_forward_time": 1.7013413906097412,
      "step": 24472
    },
    {
      "epoch": 2.333831787109375e-05,
      "step": 24472,
      "training_step_time": 7.723144292831421
    },
    {
      "epoch": 2.3339271545410155e-05,
      "model_forward_time": 1.6890015602111816,
      "step": 24473
    },
    {
      "epoch": 2.3339271545410155e-05,
      "step": 24473,
      "training_step_time": 6.836348533630371
    },
    {
      "epoch": 2.3340225219726563e-05,
      "model_forward_time": 1.6857666969299316,
      "step": 24474
    },
    {
      "epoch": 2.3340225219726563e-05,
      "step": 24474,
      "training_step_time": 6.8155906200408936
    },
    {
      "epoch": 2.334117889404297e-05,
      "model_forward_time": 1.6932086944580078,
      "step": 24475
    },
    {
      "epoch": 2.334117889404297e-05,
      "step": 24475,
      "training_step_time": 6.812049627304077
    },
    {
      "epoch": 2.3342132568359374e-05,
      "model_forward_time": 1.6906497478485107,
      "step": 24476
    },
    {
      "epoch": 2.3342132568359374e-05,
      "step": 24476,
      "training_step_time": 6.8008294105529785
    },
    {
      "epoch": 2.334308624267578e-05,
      "model_forward_time": 1.6864173412322998,
      "step": 24477
    },
    {
      "epoch": 2.334308624267578e-05,
      "step": 24477,
      "training_step_time": 6.82269811630249
    },
    {
      "epoch": 2.334403991699219e-05,
      "model_forward_time": 1.6906909942626953,
      "step": 24478
    },
    {
      "epoch": 2.334403991699219e-05,
      "step": 24478,
      "training_step_time": 8.139668464660645
    },
    {
      "epoch": 2.3344993591308593e-05,
      "model_forward_time": 2.2278385162353516,
      "step": 24479
    },
    {
      "epoch": 2.3344993591308593e-05,
      "step": 24479,
      "training_step_time": 8.947019815444946
    },
    {
      "epoch": 2.3345947265625e-05,
      "grad_norm": 0.17266499996185303,
      "learning_rate": 8.998026146767486e-06,
      "loss": 0.066,
      "step": 24480
    },
    {
      "epoch": 2.3345947265625e-05,
      "model_forward_time": 1.694439172744751,
      "step": 24480
    },
    {
      "dynamics_loss_avg": 0.06662932056933642,
      "epoch": 2.3345947265625e-05,
      "step": 24480
    },
    {
      "action_loss_avg": 0.0036303522007074206,
      "epoch": 2.3345947265625e-05,
      "step": 24480
    },
    {
      "epoch": 2.3345947265625e-05,
      "step": 24480,
      "training_step_time": 8.485394716262817
    },
    {
      "epoch": 2.3346900939941408e-05,
      "model_forward_time": 1.6959950923919678,
      "step": 24481
    },
    {
      "epoch": 2.3346900939941408e-05,
      "step": 24481,
      "training_step_time": 8.75691556930542
    },
    {
      "epoch": 2.3347854614257812e-05,
      "model_forward_time": 1.6892411708831787,
      "step": 24482
    },
    {
      "epoch": 2.3347854614257812e-05,
      "step": 24482,
      "training_step_time": 7.676019191741943
    },
    {
      "epoch": 2.334880828857422e-05,
      "model_forward_time": 1.6871349811553955,
      "step": 24483
    },
    {
      "epoch": 2.334880828857422e-05,
      "step": 24483,
      "training_step_time": 6.840402603149414
    },
    {
      "epoch": 2.3349761962890627e-05,
      "model_forward_time": 1.6842844486236572,
      "step": 24484
    },
    {
      "epoch": 2.3349761962890627e-05,
      "step": 24484,
      "training_step_time": 6.863727569580078
    },
    {
      "epoch": 2.335071563720703e-05,
      "model_forward_time": 1.693558931350708,
      "step": 24485
    },
    {
      "epoch": 2.335071563720703e-05,
      "step": 24485,
      "training_step_time": 6.837887525558472
    },
    {
      "epoch": 2.3351669311523438e-05,
      "model_forward_time": 1.6960501670837402,
      "step": 24486
    },
    {
      "epoch": 2.3351669311523438e-05,
      "step": 24486,
      "training_step_time": 6.864675045013428
    },
    {
      "epoch": 2.3352622985839842e-05,
      "model_forward_time": 1.6976053714752197,
      "step": 24487
    },
    {
      "epoch": 2.3352622985839842e-05,
      "step": 24487,
      "training_step_time": 8.649466514587402
    },
    {
      "epoch": 2.335357666015625e-05,
      "model_forward_time": 2.231349468231201,
      "step": 24488
    },
    {
      "epoch": 2.335357666015625e-05,
      "step": 24488,
      "training_step_time": 8.448564529418945
    },
    {
      "epoch": 2.3354530334472657e-05,
      "model_forward_time": 1.6948575973510742,
      "step": 24489
    },
    {
      "epoch": 2.3354530334472657e-05,
      "step": 24489,
      "training_step_time": 7.8974902629852295
    },
    {
      "epoch": 2.335548400878906e-05,
      "grad_norm": 0.1965828239917755,
      "learning_rate": 8.99703297656008e-06,
      "loss": 0.0658,
      "step": 24490
    },
    {
      "epoch": 2.335548400878906e-05,
      "model_forward_time": 1.682605266571045,
      "step": 24490
    },
    {
      "dynamics_loss_avg": 0.06096114162355661,
      "epoch": 2.335548400878906e-05,
      "step": 24490
    },
    {
      "action_loss_avg": 0.0028999918431509286,
      "epoch": 2.335548400878906e-05,
      "step": 24490
    },
    {
      "epoch": 2.335548400878906e-05,
      "step": 24490,
      "training_step_time": 8.965033292770386
    },
    {
      "epoch": 2.335643768310547e-05,
      "model_forward_time": 1.6862261295318604,
      "step": 24491
    },
    {
      "epoch": 2.335643768310547e-05,
      "step": 24491,
      "training_step_time": 7.733193397521973
    },
    {
      "epoch": 2.3357391357421876e-05,
      "model_forward_time": 1.6919541358947754,
      "step": 24492
    },
    {
      "epoch": 2.3357391357421876e-05,
      "step": 24492,
      "training_step_time": 6.86122727394104
    },
    {
      "epoch": 2.335834503173828e-05,
      "model_forward_time": 1.320037841796875,
      "step": 24493
    },
    {
      "epoch": 2.335834503173828e-05,
      "step": 24493,
      "training_step_time": 6.880202770233154
    },
    {
      "epoch": 2.3359298706054687e-05,
      "model_forward_time": 1.6892011165618896,
      "step": 24494
    },
    {
      "epoch": 2.3359298706054687e-05,
      "step": 24494,
      "training_step_time": 6.903411149978638
    },
    {
      "epoch": 2.3360252380371095e-05,
      "model_forward_time": 1.6883106231689453,
      "step": 24495
    },
    {
      "epoch": 2.3360252380371095e-05,
      "step": 24495,
      "training_step_time": 6.936352729797363
    },
    {
      "epoch": 2.33612060546875e-05,
      "model_forward_time": 1.6856353282928467,
      "step": 24496
    },
    {
      "epoch": 2.33612060546875e-05,
      "step": 24496,
      "training_step_time": 7.5734827518463135
    },
    {
      "epoch": 2.3362159729003906e-05,
      "model_forward_time": 2.2328107357025146,
      "step": 24497
    },
    {
      "epoch": 2.3362159729003906e-05,
      "step": 24497,
      "training_step_time": 7.903965950012207
    },
    {
      "epoch": 2.3363113403320314e-05,
      "model_forward_time": 1.6952471733093262,
      "step": 24498
    },
    {
      "epoch": 2.3363113403320314e-05,
      "step": 24498,
      "training_step_time": 9.54877257347107
    },
    {
      "epoch": 2.3364067077636718e-05,
      "model_forward_time": 1.6839725971221924,
      "step": 24499
    },
    {
      "epoch": 2.3364067077636718e-05,
      "step": 24499,
      "training_step_time": 8.304714679718018
    },
    {
      "epoch": 2.3365020751953125e-05,
      "grad_norm": 0.23396919667720795,
      "learning_rate": 8.996039369243158e-06,
      "loss": 0.0611,
      "step": 24500
    },
    {
      "epoch": 2.3365020751953125e-05,
      "model_forward_time": 1.7072315216064453,
      "step": 24500
    },
    {
      "dynamics_loss_avg": 0.06617196723818779,
      "epoch": 2.3365020751953125e-05,
      "step": 24500
    },
    {
      "action_loss_avg": 0.002879921777639538,
      "epoch": 2.3365020751953125e-05,
      "step": 24500
    },
    {
      "epoch": 2.3365020751953125e-05,
      "step": 24500,
      "training_step_time": 7.830768823623657
    },
    {
      "epoch": 2.3365974426269532e-05,
      "model_forward_time": 1.675750970840454,
      "step": 24501
    },
    {
      "epoch": 2.3365974426269532e-05,
      "step": 24501,
      "training_step_time": 7.164491176605225
    },
    {
      "epoch": 2.3366928100585936e-05,
      "model_forward_time": 1.687727689743042,
      "step": 24502
    },
    {
      "epoch": 2.3366928100585936e-05,
      "step": 24502,
      "training_step_time": 6.84794020652771
    },
    {
      "epoch": 2.3367881774902344e-05,
      "model_forward_time": 1.6870148181915283,
      "step": 24503
    },
    {
      "epoch": 2.3367881774902344e-05,
      "step": 24503,
      "training_step_time": 6.815974712371826
    },
    {
      "epoch": 2.336883544921875e-05,
      "model_forward_time": 1.6911227703094482,
      "step": 24504
    },
    {
      "epoch": 2.336883544921875e-05,
      "step": 24504,
      "training_step_time": 6.7918541431427
    },
    {
      "epoch": 2.3369789123535155e-05,
      "model_forward_time": 1.7049319744110107,
      "step": 24505
    },
    {
      "epoch": 2.3369789123535155e-05,
      "step": 24505,
      "training_step_time": 7.565863370895386
    },
    {
      "epoch": 2.3370742797851563e-05,
      "model_forward_time": 2.257018804550171,
      "step": 24506
    },
    {
      "epoch": 2.3370742797851563e-05,
      "step": 24506,
      "training_step_time": 9.040872573852539
    },
    {
      "epoch": 2.337169647216797e-05,
      "model_forward_time": 1.68743896484375,
      "step": 24507
    },
    {
      "epoch": 2.337169647216797e-05,
      "step": 24507,
      "training_step_time": 9.404630422592163
    },
    {
      "epoch": 2.3372650146484374e-05,
      "model_forward_time": 1.7011961936950684,
      "step": 24508
    },
    {
      "epoch": 2.3372650146484374e-05,
      "step": 24508,
      "training_step_time": 8.647317886352539
    },
    {
      "epoch": 2.337360382080078e-05,
      "model_forward_time": 1.7094347476959229,
      "step": 24509
    },
    {
      "epoch": 2.337360382080078e-05,
      "step": 24509,
      "training_step_time": 7.770537853240967
    },
    {
      "epoch": 2.337455749511719e-05,
      "grad_norm": 0.19934146106243134,
      "learning_rate": 8.995045324925378e-06,
      "loss": 0.0592,
      "step": 24510
    },
    {
      "epoch": 2.337455749511719e-05,
      "model_forward_time": 1.6862506866455078,
      "step": 24510
    },
    {
      "dynamics_loss_avg": 0.05786687843501568,
      "epoch": 2.337455749511719e-05,
      "step": 24510
    },
    {
      "action_loss_avg": 0.003469628596212715,
      "epoch": 2.337455749511719e-05,
      "step": 24510
    },
    {
      "epoch": 2.337455749511719e-05,
      "step": 24510,
      "training_step_time": 7.036584377288818
    },
    {
      "epoch": 2.3375511169433593e-05,
      "model_forward_time": 1.6902656555175781,
      "step": 24511
    },
    {
      "epoch": 2.3375511169433593e-05,
      "step": 24511,
      "training_step_time": 6.784870386123657
    },
    {
      "epoch": 2.337646484375e-05,
      "model_forward_time": 1.6913890838623047,
      "step": 24512
    },
    {
      "epoch": 2.337646484375e-05,
      "step": 24512,
      "training_step_time": 6.767969369888306
    },
    {
      "epoch": 2.3377418518066408e-05,
      "model_forward_time": 1.6946825981140137,
      "step": 24513
    },
    {
      "epoch": 2.3377418518066408e-05,
      "step": 24513,
      "training_step_time": 6.849996328353882
    },
    {
      "epoch": 2.3378372192382812e-05,
      "model_forward_time": 1.6950247287750244,
      "step": 24514
    },
    {
      "epoch": 2.3378372192382812e-05,
      "step": 24514,
      "training_step_time": 8.096840143203735
    },
    {
      "epoch": 2.337932586669922e-05,
      "model_forward_time": 1.6933352947235107,
      "step": 24515
    },
    {
      "epoch": 2.337932586669922e-05,
      "step": 24515,
      "training_step_time": 9.429352283477783
    },
    {
      "epoch": 2.3380279541015623e-05,
      "model_forward_time": 1.684539794921875,
      "step": 24516
    },
    {
      "epoch": 2.3380279541015623e-05,
      "step": 24516,
      "training_step_time": 7.875350713729858
    },
    {
      "epoch": 2.338123321533203e-05,
      "model_forward_time": 1.7078638076782227,
      "step": 24517
    },
    {
      "epoch": 2.338123321533203e-05,
      "step": 24517,
      "training_step_time": 8.31336498260498
    },
    {
      "epoch": 2.3382186889648438e-05,
      "model_forward_time": 1.702406644821167,
      "step": 24518
    },
    {
      "epoch": 2.3382186889648438e-05,
      "step": 24518,
      "training_step_time": 8.530577898025513
    },
    {
      "epoch": 2.3383140563964842e-05,
      "model_forward_time": 1.6780948638916016,
      "step": 24519
    },
    {
      "epoch": 2.3383140563964842e-05,
      "step": 24519,
      "training_step_time": 7.632819652557373
    },
    {
      "epoch": 2.338409423828125e-05,
      "grad_norm": 0.19535064697265625,
      "learning_rate": 8.994050843715447e-06,
      "loss": 0.0628,
      "step": 24520
    },
    {
      "epoch": 2.338409423828125e-05,
      "model_forward_time": 1.6839752197265625,
      "step": 24520
    },
    {
      "dynamics_loss_avg": 0.07841946184635162,
      "epoch": 2.338409423828125e-05,
      "step": 24520
    },
    {
      "action_loss_avg": 0.002656492794631049,
      "epoch": 2.338409423828125e-05,
      "step": 24520
    },
    {
      "epoch": 2.338409423828125e-05,
      "step": 24520,
      "training_step_time": 6.85050368309021
    },
    {
      "epoch": 2.3385047912597657e-05,
      "model_forward_time": 1.6923527717590332,
      "step": 24521
    },
    {
      "epoch": 2.3385047912597657e-05,
      "step": 24521,
      "training_step_time": 6.844205141067505
    },
    {
      "epoch": 2.338600158691406e-05,
      "model_forward_time": 1.6971633434295654,
      "step": 24522
    },
    {
      "epoch": 2.338600158691406e-05,
      "step": 24522,
      "training_step_time": 6.802794694900513
    },
    {
      "epoch": 2.338695526123047e-05,
      "model_forward_time": 1.6919307708740234,
      "step": 24523
    },
    {
      "epoch": 2.338695526123047e-05,
      "step": 24523,
      "training_step_time": 8.262020587921143
    },
    {
      "epoch": 2.3387908935546876e-05,
      "model_forward_time": 1.681849718093872,
      "step": 24524
    },
    {
      "epoch": 2.3387908935546876e-05,
      "step": 24524,
      "training_step_time": 8.270657062530518
    },
    {
      "epoch": 2.338886260986328e-05,
      "model_forward_time": 2.218242883682251,
      "step": 24525
    },
    {
      "epoch": 2.338886260986328e-05,
      "step": 24525,
      "training_step_time": 7.820814609527588
    },
    {
      "epoch": 2.3389816284179687e-05,
      "model_forward_time": 1.6952905654907227,
      "step": 24526
    },
    {
      "epoch": 2.3389816284179687e-05,
      "step": 24526,
      "training_step_time": 10.913019180297852
    },
    {
      "epoch": 2.3390769958496095e-05,
      "model_forward_time": 1.6934478282928467,
      "step": 24527
    },
    {
      "epoch": 2.3390769958496095e-05,
      "step": 24527,
      "training_step_time": 8.688546895980835
    },
    {
      "epoch": 2.33917236328125e-05,
      "model_forward_time": 1.7046926021575928,
      "step": 24528
    },
    {
      "epoch": 2.33917236328125e-05,
      "step": 24528,
      "training_step_time": 7.688571929931641
    },
    {
      "epoch": 2.3392677307128906e-05,
      "model_forward_time": 1.6941428184509277,
      "step": 24529
    },
    {
      "epoch": 2.3392677307128906e-05,
      "step": 24529,
      "training_step_time": 7.1365907192230225
    },
    {
      "epoch": 2.3393630981445314e-05,
      "grad_norm": 0.18110805749893188,
      "learning_rate": 8.99305592572212e-06,
      "loss": 0.063,
      "step": 24530
    },
    {
      "epoch": 2.3393630981445314e-05,
      "model_forward_time": 1.7023451328277588,
      "step": 24530
    },
    {
      "dynamics_loss_avg": 0.06562230139970779,
      "epoch": 2.3393630981445314e-05,
      "step": 24530
    },
    {
      "action_loss_avg": 0.0019496070337481797,
      "epoch": 2.3393630981445314e-05,
      "step": 24530
    },
    {
      "epoch": 2.3393630981445314e-05,
      "step": 24530,
      "training_step_time": 6.794957160949707
    },
    {
      "epoch": 2.3394584655761718e-05,
      "model_forward_time": 1.6936662197113037,
      "step": 24531
    },
    {
      "epoch": 2.3394584655761718e-05,
      "step": 24531,
      "training_step_time": 6.807176351547241
    },
    {
      "epoch": 2.3395538330078125e-05,
      "model_forward_time": 1.6956238746643066,
      "step": 24532
    },
    {
      "epoch": 2.3395538330078125e-05,
      "step": 24532,
      "training_step_time": 7.333922863006592
    },
    {
      "epoch": 2.3396492004394532e-05,
      "model_forward_time": 1.68263578414917,
      "step": 24533
    },
    {
      "epoch": 2.3396492004394532e-05,
      "step": 24533,
      "training_step_time": 7.636560916900635
    },
    {
      "epoch": 2.3397445678710936e-05,
      "model_forward_time": 2.2136895656585693,
      "step": 24534
    },
    {
      "epoch": 2.3397445678710936e-05,
      "step": 24534,
      "training_step_time": 8.632758617401123
    },
    {
      "epoch": 2.3398399353027344e-05,
      "model_forward_time": 1.696908712387085,
      "step": 24535
    },
    {
      "epoch": 2.3398399353027344e-05,
      "step": 24535,
      "training_step_time": 8.80495285987854
    },
    {
      "epoch": 2.339935302734375e-05,
      "model_forward_time": 1.6963846683502197,
      "step": 24536
    },
    {
      "epoch": 2.339935302734375e-05,
      "step": 24536,
      "training_step_time": 7.7232677936553955
    },
    {
      "epoch": 2.3400306701660155e-05,
      "model_forward_time": 1.7207112312316895,
      "step": 24537
    },
    {
      "epoch": 2.3400306701660155e-05,
      "step": 24537,
      "training_step_time": 7.726860046386719
    },
    {
      "epoch": 2.3401260375976563e-05,
      "model_forward_time": 1.707996129989624,
      "step": 24538
    },
    {
      "epoch": 2.3401260375976563e-05,
      "step": 24538,
      "training_step_time": 7.70496678352356
    },
    {
      "epoch": 2.340221405029297e-05,
      "model_forward_time": 1.6821796894073486,
      "step": 24539
    },
    {
      "epoch": 2.340221405029297e-05,
      "step": 24539,
      "training_step_time": 6.813603401184082
    },
    {
      "epoch": 2.3403167724609374e-05,
      "grad_norm": 0.19602032005786896,
      "learning_rate": 8.992060571054203e-06,
      "loss": 0.0626,
      "step": 24540
    },
    {
      "epoch": 2.3403167724609374e-05,
      "model_forward_time": 1.6926853656768799,
      "step": 24540
    },
    {
      "dynamics_loss_avg": 0.06475914008915425,
      "epoch": 2.3403167724609374e-05,
      "step": 24540
    },
    {
      "action_loss_avg": 0.0030674351030029355,
      "epoch": 2.3403167724609374e-05,
      "step": 24540
    },
    {
      "epoch": 2.3403167724609374e-05,
      "step": 24540,
      "training_step_time": 6.789673566818237
    },
    {
      "epoch": 2.340412139892578e-05,
      "model_forward_time": 1.696223497390747,
      "step": 24541
    },
    {
      "epoch": 2.340412139892578e-05,
      "step": 24541,
      "training_step_time": 7.309318780899048
    },
    {
      "epoch": 2.340507507324219e-05,
      "model_forward_time": 1.7005462646484375,
      "step": 24542
    },
    {
      "epoch": 2.340507507324219e-05,
      "step": 24542,
      "training_step_time": 7.639572858810425
    },
    {
      "epoch": 2.3406028747558593e-05,
      "model_forward_time": 2.2326724529266357,
      "step": 24543
    },
    {
      "epoch": 2.3406028747558593e-05,
      "step": 24543,
      "training_step_time": 7.887974739074707
    },
    {
      "epoch": 2.3406982421875e-05,
      "model_forward_time": 1.6956202983856201,
      "step": 24544
    },
    {
      "epoch": 2.3406982421875e-05,
      "step": 24544,
      "training_step_time": 8.521450996398926
    },
    {
      "epoch": 2.3407936096191408e-05,
      "model_forward_time": 1.7233822345733643,
      "step": 24545
    },
    {
      "epoch": 2.3407936096191408e-05,
      "step": 24545,
      "training_step_time": 7.864741325378418
    },
    {
      "epoch": 2.3408889770507812e-05,
      "model_forward_time": 1.7027366161346436,
      "step": 24546
    },
    {
      "epoch": 2.3408889770507812e-05,
      "step": 24546,
      "training_step_time": 7.846508741378784
    },
    {
      "epoch": 2.340984344482422e-05,
      "model_forward_time": 1.7017767429351807,
      "step": 24547
    },
    {
      "epoch": 2.340984344482422e-05,
      "step": 24547,
      "training_step_time": 7.71211838722229
    },
    {
      "epoch": 2.3410797119140623e-05,
      "model_forward_time": 1.69020676612854,
      "step": 24548
    },
    {
      "epoch": 2.3410797119140623e-05,
      "step": 24548,
      "training_step_time": 7.223806619644165
    },
    {
      "epoch": 2.341175079345703e-05,
      "model_forward_time": 1.695552110671997,
      "step": 24549
    },
    {
      "epoch": 2.341175079345703e-05,
      "step": 24549,
      "training_step_time": 6.863368272781372
    },
    {
      "epoch": 2.3412704467773438e-05,
      "grad_norm": 0.17353469133377075,
      "learning_rate": 8.991064779820542e-06,
      "loss": 0.0641,
      "step": 24550
    },
    {
      "epoch": 2.3412704467773438e-05,
      "model_forward_time": 1.6853089332580566,
      "step": 24550
    },
    {
      "dynamics_loss_avg": 0.05699722561985254,
      "epoch": 2.3412704467773438e-05,
      "step": 24550
    },
    {
      "action_loss_avg": 0.0019410782668273896,
      "epoch": 2.3412704467773438e-05,
      "step": 24550
    },
    {
      "epoch": 2.3412704467773438e-05,
      "step": 24550,
      "training_step_time": 7.308021068572998
    },
    {
      "epoch": 2.3413658142089842e-05,
      "model_forward_time": 1.7057485580444336,
      "step": 24551
    },
    {
      "epoch": 2.3413658142089842e-05,
      "step": 24551,
      "training_step_time": 7.554277420043945
    },
    {
      "epoch": 2.341461181640625e-05,
      "model_forward_time": 2.246615409851074,
      "step": 24552
    },
    {
      "epoch": 2.341461181640625e-05,
      "step": 24552,
      "training_step_time": 8.404746770858765
    },
    {
      "epoch": 2.3415565490722657e-05,
      "model_forward_time": 1.6987509727478027,
      "step": 24553
    },
    {
      "epoch": 2.3415565490722657e-05,
      "step": 24553,
      "training_step_time": 8.692174434661865
    },
    {
      "epoch": 2.341651916503906e-05,
      "model_forward_time": 1.724123239517212,
      "step": 24554
    },
    {
      "epoch": 2.341651916503906e-05,
      "step": 24554,
      "training_step_time": 8.371798992156982
    },
    {
      "epoch": 2.341747283935547e-05,
      "model_forward_time": 1.6922035217285156,
      "step": 24555
    },
    {
      "epoch": 2.341747283935547e-05,
      "step": 24555,
      "training_step_time": 8.29475998878479
    },
    {
      "epoch": 2.3418426513671876e-05,
      "model_forward_time": 1.6919221878051758,
      "step": 24556
    },
    {
      "epoch": 2.3418426513671876e-05,
      "step": 24556,
      "training_step_time": 7.698327541351318
    },
    {
      "epoch": 2.341938018798828e-05,
      "model_forward_time": 1.6857976913452148,
      "step": 24557
    },
    {
      "epoch": 2.341938018798828e-05,
      "step": 24557,
      "training_step_time": 7.184170722961426
    },
    {
      "epoch": 2.3420333862304687e-05,
      "model_forward_time": 1.6874799728393555,
      "step": 24558
    },
    {
      "epoch": 2.3420333862304687e-05,
      "step": 24558,
      "training_step_time": 6.818044900894165
    },
    {
      "epoch": 2.3421287536621095e-05,
      "model_forward_time": 1.6875426769256592,
      "step": 24559
    },
    {
      "epoch": 2.3421287536621095e-05,
      "step": 24559,
      "training_step_time": 7.303989887237549
    },
    {
      "epoch": 2.34222412109375e-05,
      "grad_norm": 0.15905189514160156,
      "learning_rate": 8.990068552130037e-06,
      "loss": 0.0649,
      "step": 24560
    },
    {
      "epoch": 2.34222412109375e-05,
      "model_forward_time": 1.6892979145050049,
      "step": 24560
    },
    {
      "dynamics_loss_avg": 0.05008216015994549,
      "epoch": 2.34222412109375e-05,
      "step": 24560
    },
    {
      "action_loss_avg": 0.0019271217577625066,
      "epoch": 2.34222412109375e-05,
      "step": 24560
    },
    {
      "epoch": 2.34222412109375e-05,
      "step": 24560,
      "training_step_time": 8.339802265167236
    },
    {
      "epoch": 2.3423194885253906e-05,
      "model_forward_time": 2.2494587898254395,
      "step": 24561
    },
    {
      "epoch": 2.3423194885253906e-05,
      "step": 24561,
      "training_step_time": 7.580295562744141
    },
    {
      "epoch": 2.3424148559570314e-05,
      "model_forward_time": 1.6867363452911377,
      "step": 24562
    },
    {
      "epoch": 2.3424148559570314e-05,
      "step": 24562,
      "training_step_time": 9.930782556533813
    },
    {
      "epoch": 2.3425102233886718e-05,
      "model_forward_time": 1.6988410949707031,
      "step": 24563
    },
    {
      "epoch": 2.3425102233886718e-05,
      "step": 24563,
      "training_step_time": 8.151706457138062
    },
    {
      "epoch": 2.3426055908203125e-05,
      "model_forward_time": 1.6905994415283203,
      "step": 24564
    },
    {
      "epoch": 2.3426055908203125e-05,
      "step": 24564,
      "training_step_time": 7.621095895767212
    },
    {
      "epoch": 2.3427009582519532e-05,
      "model_forward_time": 1.7099955081939697,
      "step": 24565
    },
    {
      "epoch": 2.3427009582519532e-05,
      "step": 24565,
      "training_step_time": 7.295431613922119
    },
    {
      "epoch": 2.3427963256835936e-05,
      "model_forward_time": 1.6883723735809326,
      "step": 24566
    },
    {
      "epoch": 2.3427963256835936e-05,
      "step": 24566,
      "training_step_time": 7.134768486022949
    },
    {
      "epoch": 2.3428916931152344e-05,
      "model_forward_time": 1.6911559104919434,
      "step": 24567
    },
    {
      "epoch": 2.3428916931152344e-05,
      "step": 24567,
      "training_step_time": 6.670873403549194
    },
    {
      "epoch": 2.342987060546875e-05,
      "model_forward_time": 1.6869847774505615,
      "step": 24568
    },
    {
      "epoch": 2.342987060546875e-05,
      "step": 24568,
      "training_step_time": 6.76446533203125
    },
    {
      "epoch": 2.3430824279785155e-05,
      "model_forward_time": 1.6966602802276611,
      "step": 24569
    },
    {
      "epoch": 2.3430824279785155e-05,
      "step": 24569,
      "training_step_time": 8.286737442016602
    },
    {
      "epoch": 2.3431777954101563e-05,
      "grad_norm": 0.1659531593322754,
      "learning_rate": 8.989071888091633e-06,
      "loss": 0.0619,
      "step": 24570
    },
    {
      "epoch": 2.3431777954101563e-05,
      "model_forward_time": 1.6894547939300537,
      "step": 24570
    },
    {
      "dynamics_loss_avg": 0.04635070264339447,
      "epoch": 2.3431777954101563e-05,
      "step": 24570
    },
    {
      "action_loss_avg": 0.002064118825364858,
      "epoch": 2.3431777954101563e-05,
      "step": 24570
    },
    {
      "epoch": 2.3431777954101563e-05,
      "step": 24570,
      "training_step_time": 7.474606037139893
    },
    {
      "epoch": 2.343273162841797e-05,
      "model_forward_time": 2.230109930038452,
      "step": 24571
    },
    {
      "epoch": 2.343273162841797e-05,
      "step": 24571,
      "training_step_time": 8.320621967315674
    },
    {
      "epoch": 2.3433685302734374e-05,
      "model_forward_time": 1.6939024925231934,
      "step": 24572
    },
    {
      "epoch": 2.3433685302734374e-05,
      "step": 24572,
      "training_step_time": 9.598689079284668
    },
    {
      "epoch": 2.343463897705078e-05,
      "model_forward_time": 1.7008237838745117,
      "step": 24573
    },
    {
      "epoch": 2.343463897705078e-05,
      "step": 24573,
      "training_step_time": 7.6876654624938965
    },
    {
      "epoch": 2.343559265136719e-05,
      "model_forward_time": 1.7189586162567139,
      "step": 24574
    },
    {
      "epoch": 2.343559265136719e-05,
      "step": 24574,
      "training_step_time": 7.647534608840942
    },
    {
      "epoch": 2.3436546325683593e-05,
      "model_forward_time": 1.6872096061706543,
      "step": 24575
    },
    {
      "epoch": 2.3436546325683593e-05,
      "step": 24575,
      "training_step_time": 7.612288951873779
    },
    {
      "epoch": 2.34375e-05,
      "model_forward_time": 1.693084716796875,
      "step": 24576
    },
    {
      "epoch": 2.34375e-05,
      "step": 24576,
      "training_step_time": 6.683487892150879
    },
    {
      "epoch": 2.3438453674316408e-05,
      "model_forward_time": 1.6985821723937988,
      "step": 24577
    },
    {
      "epoch": 2.3438453674316408e-05,
      "step": 24577,
      "training_step_time": 6.7351953983306885
    },
    {
      "epoch": 2.3439407348632812e-05,
      "model_forward_time": 1.7017254829406738,
      "step": 24578
    },
    {
      "epoch": 2.3439407348632812e-05,
      "step": 24578,
      "training_step_time": 7.2254533767700195
    },
    {
      "epoch": 2.344036102294922e-05,
      "model_forward_time": 1.698136806488037,
      "step": 24579
    },
    {
      "epoch": 2.344036102294922e-05,
      "step": 24579,
      "training_step_time": 8.745304584503174
    },
    {
      "epoch": 2.3441314697265623e-05,
      "grad_norm": 0.1472497135400772,
      "learning_rate": 8.988074787814328e-06,
      "loss": 0.068,
      "step": 24580
    },
    {
      "epoch": 2.3441314697265623e-05,
      "model_forward_time": 2.254807949066162,
      "step": 24580
    },
    {
      "dynamics_loss_avg": 0.06753668319433928,
      "epoch": 2.3441314697265623e-05,
      "step": 24580
    },
    {
      "action_loss_avg": 0.00413784014963312,
      "epoch": 2.3441314697265623e-05,
      "step": 24580
    },
    {
      "epoch": 2.3441314697265623e-05,
      "step": 24580,
      "training_step_time": 8.808185577392578
    },
    {
      "epoch": 2.344226837158203e-05,
      "model_forward_time": 1.7087349891662598,
      "step": 24581
    },
    {
      "epoch": 2.344226837158203e-05,
      "step": 24581,
      "training_step_time": 8.475275039672852
    },
    {
      "epoch": 2.3443222045898438e-05,
      "model_forward_time": 1.7000622749328613,
      "step": 24582
    },
    {
      "epoch": 2.3443222045898438e-05,
      "step": 24582,
      "training_step_time": 7.63022255897522
    },
    {
      "epoch": 2.3444175720214842e-05,
      "model_forward_time": 1.7663557529449463,
      "step": 24583
    },
    {
      "epoch": 2.3444175720214842e-05,
      "step": 24583,
      "training_step_time": 7.651948928833008
    },
    {
      "epoch": 2.344512939453125e-05,
      "model_forward_time": 1.7068021297454834,
      "step": 24584
    },
    {
      "epoch": 2.344512939453125e-05,
      "step": 24584,
      "training_step_time": 7.484947681427002
    },
    {
      "epoch": 2.3446083068847657e-05,
      "model_forward_time": 1.7132091522216797,
      "step": 24585
    },
    {
      "epoch": 2.3446083068847657e-05,
      "step": 24585,
      "training_step_time": 7.096501111984253
    },
    {
      "epoch": 2.344703674316406e-05,
      "model_forward_time": 1.7197437286376953,
      "step": 24586
    },
    {
      "epoch": 2.344703674316406e-05,
      "step": 24586,
      "training_step_time": 6.7145140171051025
    },
    {
      "epoch": 2.344799041748047e-05,
      "model_forward_time": 1.7022273540496826,
      "step": 24587
    },
    {
      "epoch": 2.344799041748047e-05,
      "step": 24587,
      "training_step_time": 7.51112174987793
    },
    {
      "epoch": 2.3448944091796876e-05,
      "model_forward_time": 1.7033517360687256,
      "step": 24588
    },
    {
      "epoch": 2.3448944091796876e-05,
      "step": 24588,
      "training_step_time": 7.668962478637695
    },
    {
      "epoch": 2.344989776611328e-05,
      "model_forward_time": 1.8720746040344238,
      "step": 24589
    },
    {
      "epoch": 2.344989776611328e-05,
      "step": 24589,
      "training_step_time": 7.834439039230347
    },
    {
      "epoch": 2.3450851440429687e-05,
      "grad_norm": 0.15067218244075775,
      "learning_rate": 8.987077251407159e-06,
      "loss": 0.0658,
      "step": 24590
    },
    {
      "epoch": 2.3450851440429687e-05,
      "model_forward_time": 1.710052728652954,
      "step": 24590
    },
    {
      "dynamics_loss_avg": 0.07531149219721556,
      "epoch": 2.3450851440429687e-05,
      "step": 24590
    },
    {
      "action_loss_avg": 0.0031757940363604576,
      "epoch": 2.3450851440429687e-05,
      "step": 24590
    },
    {
      "epoch": 2.3450851440429687e-05,
      "step": 24590,
      "training_step_time": 7.621313571929932
    },
    {
      "epoch": 2.3451805114746095e-05,
      "model_forward_time": 1.7352612018585205,
      "step": 24591
    },
    {
      "epoch": 2.3451805114746095e-05,
      "step": 24591,
      "training_step_time": 9.277039051055908
    },
    {
      "epoch": 2.34527587890625e-05,
      "model_forward_time": 1.7630541324615479,
      "step": 24592
    },
    {
      "epoch": 2.34527587890625e-05,
      "step": 24592,
      "training_step_time": 8.52912712097168
    },
    {
      "epoch": 2.3453712463378906e-05,
      "model_forward_time": 1.7107632160186768,
      "step": 24593
    },
    {
      "epoch": 2.3453712463378906e-05,
      "step": 24593,
      "training_step_time": 6.763270854949951
    },
    {
      "epoch": 2.3454666137695314e-05,
      "model_forward_time": 1.7088840007781982,
      "step": 24594
    },
    {
      "epoch": 2.3454666137695314e-05,
      "step": 24594,
      "training_step_time": 7.567426681518555
    },
    {
      "epoch": 2.3455619812011718e-05,
      "model_forward_time": 1.7174386978149414,
      "step": 24595
    },
    {
      "epoch": 2.3455619812011718e-05,
      "step": 24595,
      "training_step_time": 6.7895283699035645
    },
    {
      "epoch": 2.3456573486328125e-05,
      "model_forward_time": 1.7041974067687988,
      "step": 24596
    },
    {
      "epoch": 2.3456573486328125e-05,
      "step": 24596,
      "training_step_time": 7.298335552215576
    },
    {
      "epoch": 2.3457527160644532e-05,
      "model_forward_time": 1.7109043598175049,
      "step": 24597
    },
    {
      "epoch": 2.3457527160644532e-05,
      "step": 24597,
      "training_step_time": 7.587996006011963
    },
    {
      "epoch": 2.3458480834960937e-05,
      "model_forward_time": 2.24440860748291,
      "step": 24598
    },
    {
      "epoch": 2.3458480834960937e-05,
      "step": 24598,
      "training_step_time": 8.960146188735962
    },
    {
      "epoch": 2.3459434509277344e-05,
      "model_forward_time": 1.7044661045074463,
      "step": 24599
    },
    {
      "epoch": 2.3459434509277344e-05,
      "step": 24599,
      "training_step_time": 10.928932189941406
    },
    {
      "epoch": 2.346038818359375e-05,
      "grad_norm": 0.1383911371231079,
      "learning_rate": 8.986079278979216e-06,
      "loss": 0.0623,
      "step": 24600
    },
    {
      "epoch": 2.346038818359375e-05,
      "model_forward_time": 1.7079684734344482,
      "step": 24600
    },
    {
      "dynamics_loss_avg": 0.04692316045984626,
      "epoch": 2.346038818359375e-05,
      "step": 24600
    },
    {
      "action_loss_avg": 0.0023574228573124854,
      "epoch": 2.346038818359375e-05,
      "step": 24600
    },
    {
      "epoch": 2.346038818359375e-05,
      "step": 24600,
      "training_step_time": 9.705988645553589
    },
    {
      "epoch": 2.3461341857910155e-05,
      "model_forward_time": 1.748124122619629,
      "step": 24601
    },
    {
      "epoch": 2.3461341857910155e-05,
      "step": 24601,
      "training_step_time": 7.70147180557251
    },
    {
      "epoch": 2.3462295532226563e-05,
      "model_forward_time": 1.7087421417236328,
      "step": 24602
    },
    {
      "epoch": 2.3462295532226563e-05,
      "step": 24602,
      "training_step_time": 7.112769842147827
    },
    {
      "epoch": 2.346324920654297e-05,
      "model_forward_time": 1.710557460784912,
      "step": 24603
    },
    {
      "epoch": 2.346324920654297e-05,
      "step": 24603,
      "training_step_time": 7.6652443408966064
    },
    {
      "epoch": 2.3464202880859374e-05,
      "model_forward_time": 1.7032017707824707,
      "step": 24604
    },
    {
      "epoch": 2.3464202880859374e-05,
      "step": 24604,
      "training_step_time": 6.769172668457031
    },
    {
      "epoch": 2.346515655517578e-05,
      "model_forward_time": 1.7241718769073486,
      "step": 24605
    },
    {
      "epoch": 2.346515655517578e-05,
      "step": 24605,
      "training_step_time": 7.285057306289673
    },
    {
      "epoch": 2.346611022949219e-05,
      "model_forward_time": 1.7055203914642334,
      "step": 24606
    },
    {
      "epoch": 2.346611022949219e-05,
      "step": 24606,
      "training_step_time": 7.589632987976074
    },
    {
      "epoch": 2.3467063903808593e-05,
      "model_forward_time": 1.705446720123291,
      "step": 24607
    },
    {
      "epoch": 2.3467063903808593e-05,
      "step": 24607,
      "training_step_time": 9.292388200759888
    },
    {
      "epoch": 2.3468017578125e-05,
      "model_forward_time": 1.7082288265228271,
      "step": 24608
    },
    {
      "epoch": 2.3468017578125e-05,
      "step": 24608,
      "training_step_time": 10.014115810394287
    },
    {
      "epoch": 2.3468971252441408e-05,
      "model_forward_time": 1.7075088024139404,
      "step": 24609
    },
    {
      "epoch": 2.3468971252441408e-05,
      "step": 24609,
      "training_step_time": 8.405937194824219
    },
    {
      "epoch": 2.3469924926757812e-05,
      "grad_norm": 0.18782484531402588,
      "learning_rate": 8.985080870639635e-06,
      "loss": 0.0638,
      "step": 24610
    },
    {
      "epoch": 2.3469924926757812e-05,
      "model_forward_time": 1.7198941707611084,
      "step": 24610
    },
    {
      "dynamics_loss_avg": 0.06297292169183492,
      "epoch": 2.3469924926757812e-05,
      "step": 24610
    },
    {
      "action_loss_avg": 0.005013368371874094,
      "epoch": 2.3469924926757812e-05,
      "step": 24610
    },
    {
      "epoch": 2.3469924926757812e-05,
      "step": 24610,
      "training_step_time": 7.693342685699463
    },
    {
      "epoch": 2.347087860107422e-05,
      "model_forward_time": 1.705230474472046,
      "step": 24611
    },
    {
      "epoch": 2.347087860107422e-05,
      "step": 24611,
      "training_step_time": 7.797892808914185
    },
    {
      "epoch": 2.3471832275390623e-05,
      "model_forward_time": 1.7190420627593994,
      "step": 24612
    },
    {
      "epoch": 2.3471832275390623e-05,
      "step": 24612,
      "training_step_time": 7.651966333389282
    },
    {
      "epoch": 2.347278594970703e-05,
      "model_forward_time": 1.706883430480957,
      "step": 24613
    },
    {
      "epoch": 2.347278594970703e-05,
      "step": 24613,
      "training_step_time": 6.77315092086792
    },
    {
      "epoch": 2.3473739624023438e-05,
      "model_forward_time": 1.7093172073364258,
      "step": 24614
    },
    {
      "epoch": 2.3473739624023438e-05,
      "step": 24614,
      "training_step_time": 7.401667356491089
    },
    {
      "epoch": 2.3474693298339842e-05,
      "model_forward_time": 1.7055456638336182,
      "step": 24615
    },
    {
      "epoch": 2.3474693298339842e-05,
      "step": 24615,
      "training_step_time": 7.400599002838135
    },
    {
      "epoch": 2.347564697265625e-05,
      "model_forward_time": 1.7150354385375977,
      "step": 24616
    },
    {
      "epoch": 2.347564697265625e-05,
      "step": 24616,
      "training_step_time": 8.5538170337677
    },
    {
      "epoch": 2.3476600646972657e-05,
      "model_forward_time": 2.2563564777374268,
      "step": 24617
    },
    {
      "epoch": 2.3476600646972657e-05,
      "step": 24617,
      "training_step_time": 9.376091241836548
    },
    {
      "epoch": 2.347755432128906e-05,
      "model_forward_time": 1.7107973098754883,
      "step": 24618
    },
    {
      "epoch": 2.347755432128906e-05,
      "step": 24618,
      "training_step_time": 9.326944589614868
    },
    {
      "epoch": 2.347850799560547e-05,
      "model_forward_time": 1.7145421504974365,
      "step": 24619
    },
    {
      "epoch": 2.347850799560547e-05,
      "step": 24619,
      "training_step_time": 7.696648120880127
    },
    {
      "epoch": 2.3479461669921876e-05,
      "grad_norm": 0.15470437705516815,
      "learning_rate": 8.984082026497603e-06,
      "loss": 0.0625,
      "step": 24620
    },
    {
      "epoch": 2.3479461669921876e-05,
      "model_forward_time": 1.7171776294708252,
      "step": 24620
    },
    {
      "dynamics_loss_avg": 0.051619569025933745,
      "epoch": 2.3479461669921876e-05,
      "step": 24620
    },
    {
      "action_loss_avg": 0.0026320390112232416,
      "epoch": 2.3479461669921876e-05,
      "step": 24620
    },
    {
      "epoch": 2.3479461669921876e-05,
      "step": 24620,
      "training_step_time": 7.694993495941162
    },
    {
      "epoch": 2.348041534423828e-05,
      "model_forward_time": 1.7078921794891357,
      "step": 24621
    },
    {
      "epoch": 2.348041534423828e-05,
      "step": 24621,
      "training_step_time": 7.5943145751953125
    },
    {
      "epoch": 2.3481369018554687e-05,
      "model_forward_time": 1.705855131149292,
      "step": 24622
    },
    {
      "epoch": 2.3481369018554687e-05,
      "step": 24622,
      "training_step_time": 7.235460519790649
    },
    {
      "epoch": 2.3482322692871095e-05,
      "model_forward_time": 1.7138409614562988,
      "step": 24623
    },
    {
      "epoch": 2.3482322692871095e-05,
      "step": 24623,
      "training_step_time": 7.200159072875977
    },
    {
      "epoch": 2.34832763671875e-05,
      "model_forward_time": 1.7046658992767334,
      "step": 24624
    },
    {
      "epoch": 2.34832763671875e-05,
      "step": 24624,
      "training_step_time": 7.161664009094238
    },
    {
      "epoch": 2.3484230041503906e-05,
      "model_forward_time": 1.7069461345672607,
      "step": 24625
    },
    {
      "epoch": 2.3484230041503906e-05,
      "step": 24625,
      "training_step_time": 9.503187894821167
    },
    {
      "epoch": 2.3485183715820314e-05,
      "model_forward_time": 2.254315137863159,
      "step": 24626
    },
    {
      "epoch": 2.3485183715820314e-05,
      "step": 24626,
      "training_step_time": 7.904337167739868
    },
    {
      "epoch": 2.3486137390136718e-05,
      "model_forward_time": 1.7004294395446777,
      "step": 24627
    },
    {
      "epoch": 2.3486137390136718e-05,
      "step": 24627,
      "training_step_time": 8.725773096084595
    },
    {
      "epoch": 2.3487091064453125e-05,
      "model_forward_time": 1.6858394145965576,
      "step": 24628
    },
    {
      "epoch": 2.3487091064453125e-05,
      "step": 24628,
      "training_step_time": 7.649110794067383
    },
    {
      "epoch": 2.3488044738769533e-05,
      "model_forward_time": 1.8162877559661865,
      "step": 24629
    },
    {
      "epoch": 2.3488044738769533e-05,
      "step": 24629,
      "training_step_time": 7.6894543170928955
    },
    {
      "epoch": 2.3488998413085937e-05,
      "grad_norm": 0.2203194797039032,
      "learning_rate": 8.983082746662351e-06,
      "loss": 0.0629,
      "step": 24630
    },
    {
      "epoch": 2.3488998413085937e-05,
      "model_forward_time": 1.822960376739502,
      "step": 24630
    },
    {
      "dynamics_loss_avg": 0.06594648472964763,
      "epoch": 2.3488998413085937e-05,
      "step": 24630
    },
    {
      "action_loss_avg": 0.006523378007113933,
      "epoch": 2.3488998413085937e-05,
      "step": 24630
    },
    {
      "epoch": 2.3488998413085937e-05,
      "step": 24630,
      "training_step_time": 8.37684440612793
    },
    {
      "epoch": 2.3489952087402344e-05,
      "model_forward_time": 1.8245255947113037,
      "step": 24631
    },
    {
      "epoch": 2.3489952087402344e-05,
      "step": 24631,
      "training_step_time": 7.2222816944122314
    },
    {
      "epoch": 2.349090576171875e-05,
      "model_forward_time": 1.8272686004638672,
      "step": 24632
    },
    {
      "epoch": 2.349090576171875e-05,
      "step": 24632,
      "training_step_time": 7.150749206542969
    },
    {
      "epoch": 2.3491859436035155e-05,
      "model_forward_time": 1.8226699829101562,
      "step": 24633
    },
    {
      "epoch": 2.3491859436035155e-05,
      "step": 24633,
      "training_step_time": 7.29088282585144
    },
    {
      "epoch": 2.3492813110351563e-05,
      "model_forward_time": 1.824470043182373,
      "step": 24634
    },
    {
      "epoch": 2.3492813110351563e-05,
      "step": 24634,
      "training_step_time": 7.6083550453186035
    },
    {
      "epoch": 2.349376678466797e-05,
      "model_forward_time": 2.3495635986328125,
      "step": 24635
    },
    {
      "epoch": 2.349376678466797e-05,
      "step": 24635,
      "training_step_time": 7.902592420578003
    },
    {
      "epoch": 2.3494720458984374e-05,
      "model_forward_time": 1.7454042434692383,
      "step": 24636
    },
    {
      "epoch": 2.3494720458984374e-05,
      "step": 24636,
      "training_step_time": 7.76387882232666
    },
    {
      "epoch": 2.349567413330078e-05,
      "model_forward_time": 1.7368996143341064,
      "step": 24637
    },
    {
      "epoch": 2.349567413330078e-05,
      "step": 24637,
      "training_step_time": 8.218553304672241
    },
    {
      "epoch": 2.349662780761719e-05,
      "model_forward_time": 1.7287077903747559,
      "step": 24638
    },
    {
      "epoch": 2.349662780761719e-05,
      "step": 24638,
      "training_step_time": 8.094542026519775
    },
    {
      "epoch": 2.3497581481933593e-05,
      "model_forward_time": 1.7250638008117676,
      "step": 24639
    },
    {
      "epoch": 2.3497581481933593e-05,
      "step": 24639,
      "training_step_time": 7.621338844299316
    },
    {
      "epoch": 2.349853515625e-05,
      "grad_norm": 0.30137109756469727,
      "learning_rate": 8.982083031243156e-06,
      "loss": 0.0616,
      "step": 24640
    },
    {
      "epoch": 2.349853515625e-05,
      "model_forward_time": 1.707775354385376,
      "step": 24640
    },
    {
      "dynamics_loss_avg": 0.055557834543287755,
      "epoch": 2.349853515625e-05,
      "step": 24640
    },
    {
      "action_loss_avg": 0.0036185478733386845,
      "epoch": 2.349853515625e-05,
      "step": 24640
    },
    {
      "epoch": 2.349853515625e-05,
      "step": 24640,
      "training_step_time": 7.59237003326416
    },
    {
      "epoch": 2.3499488830566405e-05,
      "model_forward_time": 1.7318072319030762,
      "step": 24641
    },
    {
      "epoch": 2.3499488830566405e-05,
      "step": 24641,
      "training_step_time": 7.445699214935303
    },
    {
      "epoch": 2.3500442504882812e-05,
      "model_forward_time": 1.731632947921753,
      "step": 24642
    },
    {
      "epoch": 2.3500442504882812e-05,
      "step": 24642,
      "training_step_time": 7.227826118469238
    },
    {
      "epoch": 2.350139617919922e-05,
      "model_forward_time": 1.7268428802490234,
      "step": 24643
    },
    {
      "epoch": 2.350139617919922e-05,
      "step": 24643,
      "training_step_time": 7.602056264877319
    },
    {
      "epoch": 2.3502349853515623e-05,
      "model_forward_time": 2.2877840995788574,
      "step": 24644
    },
    {
      "epoch": 2.3502349853515623e-05,
      "step": 24644,
      "training_step_time": 8.84527039527893
    },
    {
      "epoch": 2.350330352783203e-05,
      "model_forward_time": 1.7266881465911865,
      "step": 24645
    },
    {
      "epoch": 2.350330352783203e-05,
      "step": 24645,
      "training_step_time": 10.039777517318726
    },
    {
      "epoch": 2.3504257202148438e-05,
      "model_forward_time": 1.7238385677337646,
      "step": 24646
    },
    {
      "epoch": 2.3504257202148438e-05,
      "step": 24646,
      "training_step_time": 8.620981454849243
    },
    {
      "epoch": 2.3505210876464842e-05,
      "model_forward_time": 1.7173526287078857,
      "step": 24647
    },
    {
      "epoch": 2.3505210876464842e-05,
      "step": 24647,
      "training_step_time": 7.58412504196167
    },
    {
      "epoch": 2.350616455078125e-05,
      "model_forward_time": 1.7390098571777344,
      "step": 24648
    },
    {
      "epoch": 2.350616455078125e-05,
      "step": 24648,
      "training_step_time": 8.080230474472046
    },
    {
      "epoch": 2.3507118225097657e-05,
      "model_forward_time": 1.723966360092163,
      "step": 24649
    },
    {
      "epoch": 2.3507118225097657e-05,
      "step": 24649,
      "training_step_time": 7.538553476333618
    },
    {
      "epoch": 2.350807189941406e-05,
      "grad_norm": 0.16244550049304962,
      "learning_rate": 8.98108288034935e-06,
      "loss": 0.0622,
      "step": 24650
    },
    {
      "epoch": 2.350807189941406e-05,
      "model_forward_time": 1.71360182762146,
      "step": 24650
    },
    {
      "dynamics_loss_avg": 0.06432686559855938,
      "epoch": 2.350807189941406e-05,
      "step": 24650
    },
    {
      "action_loss_avg": 0.0025641954387538137,
      "epoch": 2.350807189941406e-05,
      "step": 24650
    },
    {
      "epoch": 2.350807189941406e-05,
      "step": 24650,
      "training_step_time": 7.422739505767822
    },
    {
      "epoch": 2.350902557373047e-05,
      "model_forward_time": 1.7311859130859375,
      "step": 24651
    },
    {
      "epoch": 2.350902557373047e-05,
      "step": 24651,
      "training_step_time": 7.165992736816406
    },
    {
      "epoch": 2.3509979248046876e-05,
      "model_forward_time": 1.7349331378936768,
      "step": 24652
    },
    {
      "epoch": 2.3509979248046876e-05,
      "step": 24652,
      "training_step_time": 8.647136688232422
    },
    {
      "epoch": 2.351093292236328e-05,
      "model_forward_time": 2.268040418624878,
      "step": 24653
    },
    {
      "epoch": 2.351093292236328e-05,
      "step": 24653,
      "training_step_time": 8.492037773132324
    },
    {
      "epoch": 2.3511886596679687e-05,
      "model_forward_time": 1.732799768447876,
      "step": 24654
    },
    {
      "epoch": 2.3511886596679687e-05,
      "step": 24654,
      "training_step_time": 9.581459522247314
    },
    {
      "epoch": 2.3512840270996095e-05,
      "model_forward_time": 1.722348928451538,
      "step": 24655
    },
    {
      "epoch": 2.3512840270996095e-05,
      "step": 24655,
      "training_step_time": 8.215914964675903
    },
    {
      "epoch": 2.35137939453125e-05,
      "model_forward_time": 1.8015189170837402,
      "step": 24656
    },
    {
      "epoch": 2.35137939453125e-05,
      "step": 24656,
      "training_step_time": 7.671596050262451
    },
    {
      "epoch": 2.3514747619628906e-05,
      "model_forward_time": 1.7229821681976318,
      "step": 24657
    },
    {
      "epoch": 2.3514747619628906e-05,
      "step": 24657,
      "training_step_time": 7.963888645172119
    },
    {
      "epoch": 2.3515701293945314e-05,
      "model_forward_time": 1.728290319442749,
      "step": 24658
    },
    {
      "epoch": 2.3515701293945314e-05,
      "step": 24658,
      "training_step_time": 7.4992499351501465
    },
    {
      "epoch": 2.3516654968261718e-05,
      "model_forward_time": 1.7290265560150146,
      "step": 24659
    },
    {
      "epoch": 2.3516654968261718e-05,
      "step": 24659,
      "training_step_time": 7.453978776931763
    },
    {
      "epoch": 2.3517608642578125e-05,
      "grad_norm": 0.2252872884273529,
      "learning_rate": 8.980082294090305e-06,
      "loss": 0.0571,
      "step": 24660
    },
    {
      "epoch": 2.3517608642578125e-05,
      "model_forward_time": 1.715501070022583,
      "step": 24660
    },
    {
      "dynamics_loss_avg": 0.0621535137295723,
      "epoch": 2.3517608642578125e-05,
      "step": 24660
    },
    {
      "action_loss_avg": 0.003965129674179479,
      "epoch": 2.3517608642578125e-05,
      "step": 24660
    },
    {
      "epoch": 2.3517608642578125e-05,
      "step": 24660,
      "training_step_time": 6.741799831390381
    },
    {
      "epoch": 2.3518562316894533e-05,
      "model_forward_time": 1.7245063781738281,
      "step": 24661
    },
    {
      "epoch": 2.3518562316894533e-05,
      "step": 24661,
      "training_step_time": 8.506467819213867
    },
    {
      "epoch": 2.3519515991210937e-05,
      "model_forward_time": 1.736402988433838,
      "step": 24662
    },
    {
      "epoch": 2.3519515991210937e-05,
      "step": 24662,
      "training_step_time": 9.449951171875
    },
    {
      "epoch": 2.3520469665527344e-05,
      "model_forward_time": 2.2605488300323486,
      "step": 24663
    },
    {
      "epoch": 2.3520469665527344e-05,
      "step": 24663,
      "training_step_time": 9.166265726089478
    },
    {
      "epoch": 2.352142333984375e-05,
      "model_forward_time": 1.7261669635772705,
      "step": 24664
    },
    {
      "epoch": 2.352142333984375e-05,
      "step": 24664,
      "training_step_time": 7.721230506896973
    },
    {
      "epoch": 2.3522377014160155e-05,
      "model_forward_time": 1.7991409301757812,
      "step": 24665
    },
    {
      "epoch": 2.3522377014160155e-05,
      "step": 24665,
      "training_step_time": 8.75710415840149
    },
    {
      "epoch": 2.3523330688476563e-05,
      "model_forward_time": 1.7160897254943848,
      "step": 24666
    },
    {
      "epoch": 2.3523330688476563e-05,
      "step": 24666,
      "training_step_time": 7.528908967971802
    },
    {
      "epoch": 2.352428436279297e-05,
      "model_forward_time": 1.7175705432891846,
      "step": 24667
    },
    {
      "epoch": 2.352428436279297e-05,
      "step": 24667,
      "training_step_time": 7.829706907272339
    },
    {
      "epoch": 2.3525238037109374e-05,
      "model_forward_time": 1.7226767539978027,
      "step": 24668
    },
    {
      "epoch": 2.3525238037109374e-05,
      "step": 24668,
      "training_step_time": 7.5989580154418945
    },
    {
      "epoch": 2.3526191711425782e-05,
      "model_forward_time": 1.7269303798675537,
      "step": 24669
    },
    {
      "epoch": 2.3526191711425782e-05,
      "step": 24669,
      "training_step_time": 6.782026290893555
    },
    {
      "epoch": 2.352714538574219e-05,
      "grad_norm": 0.16668400168418884,
      "learning_rate": 8.979081272575443e-06,
      "loss": 0.0629,
      "step": 24670
    },
    {
      "epoch": 2.352714538574219e-05,
      "model_forward_time": 1.340944528579712,
      "step": 24670
    },
    {
      "dynamics_loss_avg": 0.067115568369627,
      "epoch": 2.352714538574219e-05,
      "step": 24670
    },
    {
      "action_loss_avg": 0.0028447415214031936,
      "epoch": 2.352714538574219e-05,
      "step": 24670
    },
    {
      "epoch": 2.352714538574219e-05,
      "step": 24670,
      "training_step_time": 7.561429023742676
    },
    {
      "epoch": 2.3528099060058593e-05,
      "model_forward_time": 1.717566728591919,
      "step": 24671
    },
    {
      "epoch": 2.3528099060058593e-05,
      "step": 24671,
      "training_step_time": 8.455207586288452
    },
    {
      "epoch": 2.3529052734375e-05,
      "model_forward_time": 2.284865617752075,
      "step": 24672
    },
    {
      "epoch": 2.3529052734375e-05,
      "step": 24672,
      "training_step_time": 8.714785099029541
    },
    {
      "epoch": 2.3530006408691405e-05,
      "model_forward_time": 1.7190699577331543,
      "step": 24673
    },
    {
      "epoch": 2.3530006408691405e-05,
      "step": 24673,
      "training_step_time": 8.611859321594238
    },
    {
      "epoch": 2.3530960083007812e-05,
      "model_forward_time": 1.7938203811645508,
      "step": 24674
    },
    {
      "epoch": 2.3530960083007812e-05,
      "step": 24674,
      "training_step_time": 7.694624185562134
    },
    {
      "epoch": 2.353191375732422e-05,
      "model_forward_time": 1.7100791931152344,
      "step": 24675
    },
    {
      "epoch": 2.353191375732422e-05,
      "step": 24675,
      "training_step_time": 7.566851615905762
    },
    {
      "epoch": 2.3532867431640623e-05,
      "model_forward_time": 1.7143990993499756,
      "step": 24676
    },
    {
      "epoch": 2.3532867431640623e-05,
      "step": 24676,
      "training_step_time": 7.91931414604187
    },
    {
      "epoch": 2.353382110595703e-05,
      "model_forward_time": 1.7327880859375,
      "step": 24677
    },
    {
      "epoch": 2.353382110595703e-05,
      "step": 24677,
      "training_step_time": 7.644509792327881
    },
    {
      "epoch": 2.3534774780273438e-05,
      "model_forward_time": 1.7226548194885254,
      "step": 24678
    },
    {
      "epoch": 2.3534774780273438e-05,
      "step": 24678,
      "training_step_time": 6.816240072250366
    },
    {
      "epoch": 2.3535728454589842e-05,
      "model_forward_time": 1.7187604904174805,
      "step": 24679
    },
    {
      "epoch": 2.3535728454589842e-05,
      "step": 24679,
      "training_step_time": 7.567851781845093
    },
    {
      "epoch": 2.353668212890625e-05,
      "grad_norm": 0.20327992737293243,
      "learning_rate": 8.978079815914236e-06,
      "loss": 0.0622,
      "step": 24680
    },
    {
      "epoch": 2.353668212890625e-05,
      "model_forward_time": 1.71612548828125,
      "step": 24680
    },
    {
      "dynamics_loss_avg": 0.0811070816591382,
      "epoch": 2.353668212890625e-05,
      "step": 24680
    },
    {
      "action_loss_avg": 0.002784089924534783,
      "epoch": 2.353668212890625e-05,
      "step": 24680
    },
    {
      "epoch": 2.353668212890625e-05,
      "step": 24680,
      "training_step_time": 7.643869400024414
    },
    {
      "epoch": 2.3537635803222657e-05,
      "model_forward_time": 2.2389185428619385,
      "step": 24681
    },
    {
      "epoch": 2.3537635803222657e-05,
      "step": 24681,
      "training_step_time": 8.702203512191772
    },
    {
      "epoch": 2.353858947753906e-05,
      "model_forward_time": 1.746030569076538,
      "step": 24682
    },
    {
      "epoch": 2.353858947753906e-05,
      "step": 24682,
      "training_step_time": 8.322502136230469
    },
    {
      "epoch": 2.353954315185547e-05,
      "model_forward_time": 1.7318110466003418,
      "step": 24683
    },
    {
      "epoch": 2.353954315185547e-05,
      "step": 24683,
      "training_step_time": 9.175017833709717
    },
    {
      "epoch": 2.3540496826171876e-05,
      "model_forward_time": 1.7109663486480713,
      "step": 24684
    },
    {
      "epoch": 2.3540496826171876e-05,
      "step": 24684,
      "training_step_time": 7.585526704788208
    },
    {
      "epoch": 2.354145050048828e-05,
      "model_forward_time": 1.7156178951263428,
      "step": 24685
    },
    {
      "epoch": 2.354145050048828e-05,
      "step": 24685,
      "training_step_time": 7.887912273406982
    },
    {
      "epoch": 2.3542404174804687e-05,
      "model_forward_time": 1.7473852634429932,
      "step": 24686
    },
    {
      "epoch": 2.3542404174804687e-05,
      "step": 24686,
      "training_step_time": 7.646647214889526
    },
    {
      "epoch": 2.3543357849121095e-05,
      "model_forward_time": 1.7217001914978027,
      "step": 24687
    },
    {
      "epoch": 2.3543357849121095e-05,
      "step": 24687,
      "training_step_time": 7.204357624053955
    },
    {
      "epoch": 2.35443115234375e-05,
      "model_forward_time": 1.7165746688842773,
      "step": 24688
    },
    {
      "epoch": 2.35443115234375e-05,
      "step": 24688,
      "training_step_time": 7.427340030670166
    },
    {
      "epoch": 2.3545265197753906e-05,
      "model_forward_time": 1.7256615161895752,
      "step": 24689
    },
    {
      "epoch": 2.3545265197753906e-05,
      "step": 24689,
      "training_step_time": 7.611419200897217
    },
    {
      "epoch": 2.3546218872070314e-05,
      "grad_norm": 0.21062399446964264,
      "learning_rate": 8.977077924216202e-06,
      "loss": 0.06,
      "step": 24690
    },
    {
      "epoch": 2.3546218872070314e-05,
      "model_forward_time": 2.301628589630127,
      "step": 24690
    },
    {
      "dynamics_loss_avg": 0.05269428361207247,
      "epoch": 2.3546218872070314e-05,
      "step": 24690
    },
    {
      "action_loss_avg": 0.002425817592302337,
      "epoch": 2.3546218872070314e-05,
      "step": 24690
    },
    {
      "epoch": 2.3546218872070314e-05,
      "step": 24690,
      "training_step_time": 8.871749877929688
    },
    {
      "epoch": 2.3547172546386718e-05,
      "model_forward_time": 1.7335095405578613,
      "step": 24691
    },
    {
      "epoch": 2.3547172546386718e-05,
      "step": 24691,
      "training_step_time": 8.58805227279663
    },
    {
      "epoch": 2.3548126220703125e-05,
      "model_forward_time": 1.7029926776885986,
      "step": 24692
    },
    {
      "epoch": 2.3548126220703125e-05,
      "step": 24692,
      "training_step_time": 8.215009450912476
    },
    {
      "epoch": 2.3549079895019533e-05,
      "model_forward_time": 1.7160682678222656,
      "step": 24693
    },
    {
      "epoch": 2.3549079895019533e-05,
      "step": 24693,
      "training_step_time": 8.365955829620361
    },
    {
      "epoch": 2.3550033569335937e-05,
      "model_forward_time": 1.70025634765625,
      "step": 24694
    },
    {
      "epoch": 2.3550033569335937e-05,
      "step": 24694,
      "training_step_time": 7.54915189743042
    },
    {
      "epoch": 2.3550987243652344e-05,
      "model_forward_time": 1.7038114070892334,
      "step": 24695
    },
    {
      "epoch": 2.3550987243652344e-05,
      "step": 24695,
      "training_step_time": 7.732760906219482
    },
    {
      "epoch": 2.355194091796875e-05,
      "model_forward_time": 1.7056372165679932,
      "step": 24696
    },
    {
      "epoch": 2.355194091796875e-05,
      "step": 24696,
      "training_step_time": 6.748241424560547
    },
    {
      "epoch": 2.3552894592285155e-05,
      "model_forward_time": 1.7041940689086914,
      "step": 24697
    },
    {
      "epoch": 2.3552894592285155e-05,
      "step": 24697,
      "training_step_time": 8.046387195587158
    },
    {
      "epoch": 2.3553848266601563e-05,
      "model_forward_time": 1.6994547843933105,
      "step": 24698
    },
    {
      "epoch": 2.3553848266601563e-05,
      "step": 24698,
      "training_step_time": 8.369426965713501
    },
    {
      "epoch": 2.355480194091797e-05,
      "model_forward_time": 2.277076244354248,
      "step": 24699
    },
    {
      "epoch": 2.355480194091797e-05,
      "step": 24699,
      "training_step_time": 9.3669912815094
    },
    {
      "epoch": 2.3555755615234374e-05,
      "grad_norm": 0.17916209995746613,
      "learning_rate": 8.976075597590906e-06,
      "loss": 0.0631,
      "step": 24700
    },
    {
      "epoch": 2.3555755615234374e-05,
      "model_forward_time": 1.6923959255218506,
      "step": 24700
    },
    {
      "dynamics_loss_avg": 0.05419346056878567,
      "epoch": 2.3555755615234374e-05,
      "step": 24700
    },
    {
      "action_loss_avg": 0.002118765522027388,
      "epoch": 2.3555755615234374e-05,
      "step": 24700
    },
    {
      "epoch": 2.3555755615234374e-05,
      "step": 24700,
      "training_step_time": 8.231352090835571
    },
    {
      "epoch": 2.3556709289550782e-05,
      "model_forward_time": 1.7043416500091553,
      "step": 24701
    },
    {
      "epoch": 2.3556709289550782e-05,
      "step": 24701,
      "training_step_time": 7.745654582977295
    },
    {
      "epoch": 2.355766296386719e-05,
      "model_forward_time": 1.7280597686767578,
      "step": 24702
    },
    {
      "epoch": 2.355766296386719e-05,
      "step": 24702,
      "training_step_time": 7.786600112915039
    },
    {
      "epoch": 2.3558616638183593e-05,
      "model_forward_time": 1.7016596794128418,
      "step": 24703
    },
    {
      "epoch": 2.3558616638183593e-05,
      "step": 24703,
      "training_step_time": 8.170321226119995
    },
    {
      "epoch": 2.35595703125e-05,
      "model_forward_time": 1.703038215637207,
      "step": 24704
    },
    {
      "epoch": 2.35595703125e-05,
      "step": 24704,
      "training_step_time": 8.00566816329956
    },
    {
      "epoch": 2.3560523986816405e-05,
      "model_forward_time": 1.6970536708831787,
      "step": 24705
    },
    {
      "epoch": 2.3560523986816405e-05,
      "step": 24705,
      "training_step_time": 7.616325378417969
    },
    {
      "epoch": 2.3561477661132812e-05,
      "model_forward_time": 1.6999471187591553,
      "step": 24706
    },
    {
      "epoch": 2.3561477661132812e-05,
      "step": 24706,
      "training_step_time": 7.485984802246094
    },
    {
      "epoch": 2.356243133544922e-05,
      "model_forward_time": 1.705308437347412,
      "step": 24707
    },
    {
      "epoch": 2.356243133544922e-05,
      "step": 24707,
      "training_step_time": 7.7669517993927
    },
    {
      "epoch": 2.3563385009765623e-05,
      "model_forward_time": 1.7030258178710938,
      "step": 24708
    },
    {
      "epoch": 2.3563385009765623e-05,
      "step": 24708,
      "training_step_time": 8.302942276000977
    },
    {
      "epoch": 2.356433868408203e-05,
      "model_forward_time": 1.7049975395202637,
      "step": 24709
    },
    {
      "epoch": 2.356433868408203e-05,
      "step": 24709,
      "training_step_time": 7.782138347625732
    },
    {
      "epoch": 2.356529235839844e-05,
      "grad_norm": 0.15621644258499146,
      "learning_rate": 8.975072836147959e-06,
      "loss": 0.0615,
      "step": 24710
    },
    {
      "epoch": 2.356529235839844e-05,
      "model_forward_time": 1.691798448562622,
      "step": 24710
    },
    {
      "dynamics_loss_avg": 0.05770222283899784,
      "epoch": 2.356529235839844e-05,
      "step": 24710
    },
    {
      "action_loss_avg": 0.0038687430176651107,
      "epoch": 2.356529235839844e-05,
      "step": 24710
    },
    {
      "epoch": 2.356529235839844e-05,
      "step": 24710,
      "training_step_time": 7.733356952667236
    },
    {
      "epoch": 2.3566246032714842e-05,
      "model_forward_time": 1.7116076946258545,
      "step": 24711
    },
    {
      "epoch": 2.3566246032714842e-05,
      "step": 24711,
      "training_step_time": 9.357570886611938
    },
    {
      "epoch": 2.356719970703125e-05,
      "model_forward_time": 1.7148394584655762,
      "step": 24712
    },
    {
      "epoch": 2.356719970703125e-05,
      "step": 24712,
      "training_step_time": 7.635547637939453
    },
    {
      "epoch": 2.3568153381347657e-05,
      "model_forward_time": 1.6983599662780762,
      "step": 24713
    },
    {
      "epoch": 2.3568153381347657e-05,
      "step": 24713,
      "training_step_time": 7.755085468292236
    },
    {
      "epoch": 2.356910705566406e-05,
      "model_forward_time": 1.6929864883422852,
      "step": 24714
    },
    {
      "epoch": 2.356910705566406e-05,
      "step": 24714,
      "training_step_time": 7.889237642288208
    },
    {
      "epoch": 2.357006072998047e-05,
      "model_forward_time": 1.7010560035705566,
      "step": 24715
    },
    {
      "epoch": 2.357006072998047e-05,
      "step": 24715,
      "training_step_time": 7.429821252822876
    },
    {
      "epoch": 2.3571014404296876e-05,
      "model_forward_time": 1.6996228694915771,
      "step": 24716
    },
    {
      "epoch": 2.3571014404296876e-05,
      "step": 24716,
      "training_step_time": 7.741708517074585
    },
    {
      "epoch": 2.357196807861328e-05,
      "model_forward_time": 1.6955294609069824,
      "step": 24717
    },
    {
      "epoch": 2.357196807861328e-05,
      "step": 24717,
      "training_step_time": 8.43977427482605
    },
    {
      "epoch": 2.3572921752929687e-05,
      "model_forward_time": 2.256012439727783,
      "step": 24718
    },
    {
      "epoch": 2.3572921752929687e-05,
      "step": 24718,
      "training_step_time": 9.329222202301025
    },
    {
      "epoch": 2.3573875427246095e-05,
      "model_forward_time": 1.6972954273223877,
      "step": 24719
    },
    {
      "epoch": 2.3573875427246095e-05,
      "step": 24719,
      "training_step_time": 8.445797681808472
    },
    {
      "epoch": 2.35748291015625e-05,
      "grad_norm": 0.13668036460876465,
      "learning_rate": 8.974069639997025e-06,
      "loss": 0.0641,
      "step": 24720
    },
    {
      "epoch": 2.35748291015625e-05,
      "model_forward_time": 1.7066049575805664,
      "step": 24720
    },
    {
      "dynamics_loss_avg": 0.05924070905894041,
      "epoch": 2.35748291015625e-05,
      "step": 24720
    },
    {
      "action_loss_avg": 0.002277348272036761,
      "epoch": 2.35748291015625e-05,
      "step": 24720
    },
    {
      "epoch": 2.35748291015625e-05,
      "step": 24720,
      "training_step_time": 8.778624296188354
    },
    {
      "epoch": 2.3575782775878906e-05,
      "model_forward_time": 1.7100915908813477,
      "step": 24721
    },
    {
      "epoch": 2.3575782775878906e-05,
      "step": 24721,
      "training_step_time": 7.692523717880249
    },
    {
      "epoch": 2.3576736450195314e-05,
      "model_forward_time": 1.70332932472229,
      "step": 24722
    },
    {
      "epoch": 2.3576736450195314e-05,
      "step": 24722,
      "training_step_time": 7.698593378067017
    },
    {
      "epoch": 2.3577690124511718e-05,
      "model_forward_time": 1.7010207176208496,
      "step": 24723
    },
    {
      "epoch": 2.3577690124511718e-05,
      "step": 24723,
      "training_step_time": 7.624603033065796
    },
    {
      "epoch": 2.3578643798828125e-05,
      "model_forward_time": 1.6957342624664307,
      "step": 24724
    },
    {
      "epoch": 2.3578643798828125e-05,
      "step": 24724,
      "training_step_time": 7.43411922454834
    },
    {
      "epoch": 2.3579597473144533e-05,
      "model_forward_time": 1.7036900520324707,
      "step": 24725
    },
    {
      "epoch": 2.3579597473144533e-05,
      "step": 24725,
      "training_step_time": 8.766867876052856
    },
    {
      "epoch": 2.3580551147460937e-05,
      "model_forward_time": 1.6804628372192383,
      "step": 24726
    },
    {
      "epoch": 2.3580551147460937e-05,
      "step": 24726,
      "training_step_time": 7.727896213531494
    },
    {
      "epoch": 2.3581504821777344e-05,
      "model_forward_time": 2.253380298614502,
      "step": 24727
    },
    {
      "epoch": 2.3581504821777344e-05,
      "step": 24727,
      "training_step_time": 8.481005907058716
    },
    {
      "epoch": 2.358245849609375e-05,
      "model_forward_time": 1.6941726207733154,
      "step": 24728
    },
    {
      "epoch": 2.358245849609375e-05,
      "step": 24728,
      "training_step_time": 7.75331711769104
    },
    {
      "epoch": 2.3583412170410155e-05,
      "model_forward_time": 1.6950156688690186,
      "step": 24729
    },
    {
      "epoch": 2.3583412170410155e-05,
      "step": 24729,
      "training_step_time": 8.3124520778656
    },
    {
      "epoch": 2.3584365844726563e-05,
      "grad_norm": 0.14648735523223877,
      "learning_rate": 8.973066009247808e-06,
      "loss": 0.0633,
      "step": 24730
    },
    {
      "epoch": 2.3584365844726563e-05,
      "model_forward_time": 1.7032036781311035,
      "step": 24730
    },
    {
      "dynamics_loss_avg": 0.07242080681025982,
      "epoch": 2.3584365844726563e-05,
      "step": 24730
    },
    {
      "action_loss_avg": 0.0032943602185696364,
      "epoch": 2.3584365844726563e-05,
      "step": 24730
    },
    {
      "epoch": 2.3584365844726563e-05,
      "step": 24730,
      "training_step_time": 9.19725251197815
    },
    {
      "epoch": 2.358531951904297e-05,
      "model_forward_time": 1.6904873847961426,
      "step": 24731
    },
    {
      "epoch": 2.358531951904297e-05,
      "step": 24731,
      "training_step_time": 8.262214183807373
    },
    {
      "epoch": 2.3586273193359374e-05,
      "model_forward_time": 1.697500467300415,
      "step": 24732
    },
    {
      "epoch": 2.3586273193359374e-05,
      "step": 24732,
      "training_step_time": 7.770689010620117
    },
    {
      "epoch": 2.3587226867675782e-05,
      "model_forward_time": 1.694084644317627,
      "step": 24733
    },
    {
      "epoch": 2.3587226867675782e-05,
      "step": 24733,
      "training_step_time": 7.413462400436401
    },
    {
      "epoch": 2.358818054199219e-05,
      "model_forward_time": 1.6919825077056885,
      "step": 24734
    },
    {
      "epoch": 2.358818054199219e-05,
      "step": 24734,
      "training_step_time": 7.681299448013306
    },
    {
      "epoch": 2.3589134216308593e-05,
      "model_forward_time": 1.7023451328277588,
      "step": 24735
    },
    {
      "epoch": 2.3589134216308593e-05,
      "step": 24735,
      "training_step_time": 7.684929847717285
    },
    {
      "epoch": 2.3590087890625e-05,
      "model_forward_time": 2.258399248123169,
      "step": 24736
    },
    {
      "epoch": 2.3590087890625e-05,
      "step": 24736,
      "training_step_time": 7.635892629623413
    },
    {
      "epoch": 2.3591041564941405e-05,
      "model_forward_time": 1.6925604343414307,
      "step": 24737
    },
    {
      "epoch": 2.3591041564941405e-05,
      "step": 24737,
      "training_step_time": 7.662470102310181
    },
    {
      "epoch": 2.3591995239257812e-05,
      "model_forward_time": 1.7049312591552734,
      "step": 24738
    },
    {
      "epoch": 2.3591995239257812e-05,
      "step": 24738,
      "training_step_time": 7.686647653579712
    },
    {
      "epoch": 2.359294891357422e-05,
      "model_forward_time": 1.6968624591827393,
      "step": 24739
    },
    {
      "epoch": 2.359294891357422e-05,
      "step": 24739,
      "training_step_time": 7.694288015365601
    },
    {
      "epoch": 2.3593902587890624e-05,
      "grad_norm": 0.18047712743282318,
      "learning_rate": 8.972061944010066e-06,
      "loss": 0.0591,
      "step": 24740
    },
    {
      "epoch": 2.3593902587890624e-05,
      "model_forward_time": 1.6901147365570068,
      "step": 24740
    },
    {
      "dynamics_loss_avg": 0.057087868079543115,
      "epoch": 2.3593902587890624e-05,
      "step": 24740
    },
    {
      "action_loss_avg": 0.003301367023959756,
      "epoch": 2.3593902587890624e-05,
      "step": 24740
    },
    {
      "epoch": 2.3593902587890624e-05,
      "step": 24740,
      "training_step_time": 7.5904717445373535
    },
    {
      "epoch": 2.359485626220703e-05,
      "model_forward_time": 1.6990838050842285,
      "step": 24741
    },
    {
      "epoch": 2.359485626220703e-05,
      "step": 24741,
      "training_step_time": 8.624404668807983
    },
    {
      "epoch": 2.359580993652344e-05,
      "model_forward_time": 1.69120454788208,
      "step": 24742
    },
    {
      "epoch": 2.359580993652344e-05,
      "step": 24742,
      "training_step_time": 7.837046384811401
    },
    {
      "epoch": 2.3596763610839842e-05,
      "model_forward_time": 1.6959192752838135,
      "step": 24743
    },
    {
      "epoch": 2.3596763610839842e-05,
      "step": 24743,
      "training_step_time": 7.359475612640381
    },
    {
      "epoch": 2.359771728515625e-05,
      "model_forward_time": 1.6944725513458252,
      "step": 24744
    },
    {
      "epoch": 2.359771728515625e-05,
      "step": 24744,
      "training_step_time": 8.44839859008789
    },
    {
      "epoch": 2.3598670959472657e-05,
      "model_forward_time": 2.256972074508667,
      "step": 24745
    },
    {
      "epoch": 2.3598670959472657e-05,
      "step": 24745,
      "training_step_time": 8.75456428527832
    },
    {
      "epoch": 2.359962463378906e-05,
      "model_forward_time": 1.6986134052276611,
      "step": 24746
    },
    {
      "epoch": 2.359962463378906e-05,
      "step": 24746,
      "training_step_time": 7.66309666633606
    },
    {
      "epoch": 2.360057830810547e-05,
      "model_forward_time": 1.6889853477478027,
      "step": 24747
    },
    {
      "epoch": 2.360057830810547e-05,
      "step": 24747,
      "training_step_time": 9.077930927276611
    },
    {
      "epoch": 2.3601531982421876e-05,
      "model_forward_time": 1.71067476272583,
      "step": 24748
    },
    {
      "epoch": 2.3601531982421876e-05,
      "step": 24748,
      "training_step_time": 8.63843584060669
    },
    {
      "epoch": 2.360248565673828e-05,
      "model_forward_time": 1.7124507427215576,
      "step": 24749
    },
    {
      "epoch": 2.360248565673828e-05,
      "step": 24749,
      "training_step_time": 7.563657522201538
    },
    {
      "epoch": 2.3603439331054688e-05,
      "grad_norm": 0.20988360047340393,
      "learning_rate": 8.971057444393603e-06,
      "loss": 0.0617,
      "step": 24750
    },
    {
      "epoch": 2.3603439331054688e-05,
      "model_forward_time": 1.6912100315093994,
      "step": 24750
    },
    {
      "dynamics_loss_avg": 0.07115535214543342,
      "epoch": 2.3603439331054688e-05,
      "step": 24750
    },
    {
      "action_loss_avg": 0.002718998969066888,
      "epoch": 2.3603439331054688e-05,
      "step": 24750
    },
    {
      "epoch": 2.3603439331054688e-05,
      "step": 24750,
      "training_step_time": 7.773252010345459
    },
    {
      "epoch": 2.3604393005371095e-05,
      "model_forward_time": 1.694075107574463,
      "step": 24751
    },
    {
      "epoch": 2.3604393005371095e-05,
      "step": 24751,
      "training_step_time": 7.5556113719940186
    },
    {
      "epoch": 2.36053466796875e-05,
      "model_forward_time": 1.6925864219665527,
      "step": 24752
    },
    {
      "epoch": 2.36053466796875e-05,
      "step": 24752,
      "training_step_time": 7.212648868560791
    },
    {
      "epoch": 2.3606300354003906e-05,
      "model_forward_time": 1.6984035968780518,
      "step": 24753
    },
    {
      "epoch": 2.3606300354003906e-05,
      "step": 24753,
      "training_step_time": 7.683934450149536
    },
    {
      "epoch": 2.3607254028320314e-05,
      "model_forward_time": 2.2331368923187256,
      "step": 24754
    },
    {
      "epoch": 2.3607254028320314e-05,
      "step": 24754,
      "training_step_time": 7.652080774307251
    },
    {
      "epoch": 2.3608207702636718e-05,
      "model_forward_time": 1.7178351879119873,
      "step": 24755
    },
    {
      "epoch": 2.3608207702636718e-05,
      "step": 24755,
      "training_step_time": 7.564612150192261
    },
    {
      "epoch": 2.3609161376953125e-05,
      "model_forward_time": 1.7037227153778076,
      "step": 24756
    },
    {
      "epoch": 2.3609161376953125e-05,
      "step": 24756,
      "training_step_time": 7.583776950836182
    },
    {
      "epoch": 2.3610115051269533e-05,
      "model_forward_time": 1.7476420402526855,
      "step": 24757
    },
    {
      "epoch": 2.3610115051269533e-05,
      "step": 24757,
      "training_step_time": 8.247579336166382
    },
    {
      "epoch": 2.3611068725585937e-05,
      "model_forward_time": 1.7054390907287598,
      "step": 24758
    },
    {
      "epoch": 2.3611068725585937e-05,
      "step": 24758,
      "training_step_time": 8.034560441970825
    },
    {
      "epoch": 2.3612022399902344e-05,
      "model_forward_time": 1.7141358852386475,
      "step": 24759
    },
    {
      "epoch": 2.3612022399902344e-05,
      "step": 24759,
      "training_step_time": 7.202120304107666
    },
    {
      "epoch": 2.361297607421875e-05,
      "grad_norm": 0.12304548919200897,
      "learning_rate": 8.970052510508268e-06,
      "loss": 0.0624,
      "step": 24760
    },
    {
      "epoch": 2.361297607421875e-05,
      "model_forward_time": 1.711045265197754,
      "step": 24760
    },
    {
      "dynamics_loss_avg": 0.06323083862662315,
      "epoch": 2.361297607421875e-05,
      "step": 24760
    },
    {
      "action_loss_avg": 0.0028312432812526824,
      "epoch": 2.361297607421875e-05,
      "step": 24760
    },
    {
      "epoch": 2.361297607421875e-05,
      "step": 24760,
      "training_step_time": 7.543557405471802
    },
    {
      "epoch": 2.3613929748535156e-05,
      "model_forward_time": 1.7098639011383057,
      "step": 24761
    },
    {
      "epoch": 2.3613929748535156e-05,
      "step": 24761,
      "training_step_time": 7.24834942817688
    },
    {
      "epoch": 2.3614883422851563e-05,
      "model_forward_time": 1.7105815410614014,
      "step": 24762
    },
    {
      "epoch": 2.3614883422851563e-05,
      "step": 24762,
      "training_step_time": 7.625777244567871
    },
    {
      "epoch": 2.361583709716797e-05,
      "model_forward_time": 1.7115447521209717,
      "step": 24763
    },
    {
      "epoch": 2.361583709716797e-05,
      "step": 24763,
      "training_step_time": 8.650490760803223
    },
    {
      "epoch": 2.3616790771484374e-05,
      "model_forward_time": 2.252385377883911,
      "step": 24764
    },
    {
      "epoch": 2.3616790771484374e-05,
      "step": 24764,
      "training_step_time": 7.5549445152282715
    },
    {
      "epoch": 2.3617744445800782e-05,
      "model_forward_time": 1.7125639915466309,
      "step": 24765
    },
    {
      "epoch": 2.3617744445800782e-05,
      "step": 24765,
      "training_step_time": 8.21008038520813
    },
    {
      "epoch": 2.3618698120117186e-05,
      "model_forward_time": 1.7212097644805908,
      "step": 24766
    },
    {
      "epoch": 2.3618698120117186e-05,
      "step": 24766,
      "training_step_time": 7.553825855255127
    },
    {
      "epoch": 2.3619651794433593e-05,
      "model_forward_time": 1.7114176750183105,
      "step": 24767
    },
    {
      "epoch": 2.3619651794433593e-05,
      "step": 24767,
      "training_step_time": 8.636705875396729
    },
    {
      "epoch": 2.362060546875e-05,
      "model_forward_time": 1.6892273426055908,
      "step": 24768
    },
    {
      "epoch": 2.362060546875e-05,
      "step": 24768,
      "training_step_time": 7.529181480407715
    },
    {
      "epoch": 2.3621559143066405e-05,
      "model_forward_time": 1.7100298404693604,
      "step": 24769
    },
    {
      "epoch": 2.3621559143066405e-05,
      "step": 24769,
      "training_step_time": 7.575420618057251
    },
    {
      "epoch": 2.3622512817382812e-05,
      "grad_norm": 0.1815594732761383,
      "learning_rate": 8.969047142463959e-06,
      "loss": 0.0588,
      "step": 24770
    },
    {
      "epoch": 2.3622512817382812e-05,
      "model_forward_time": 1.6945781707763672,
      "step": 24770
    },
    {
      "dynamics_loss_avg": 0.05393807291984558,
      "epoch": 2.3622512817382812e-05,
      "step": 24770
    },
    {
      "action_loss_avg": 0.002388437482295558,
      "epoch": 2.3622512817382812e-05,
      "step": 24770
    },
    {
      "epoch": 2.3622512817382812e-05,
      "step": 24770,
      "training_step_time": 7.220196485519409
    },
    {
      "epoch": 2.362346649169922e-05,
      "model_forward_time": 1.7001197338104248,
      "step": 24771
    },
    {
      "epoch": 2.362346649169922e-05,
      "step": 24771,
      "training_step_time": 7.65231466293335
    },
    {
      "epoch": 2.3624420166015624e-05,
      "model_forward_time": 1.7075951099395752,
      "step": 24772
    },
    {
      "epoch": 2.3624420166015624e-05,
      "step": 24772,
      "training_step_time": 7.642813205718994
    },
    {
      "epoch": 2.362537384033203e-05,
      "model_forward_time": 2.2646281719207764,
      "step": 24773
    },
    {
      "epoch": 2.362537384033203e-05,
      "step": 24773,
      "training_step_time": 7.681143283843994
    },
    {
      "epoch": 2.362632751464844e-05,
      "model_forward_time": 1.7140979766845703,
      "step": 24774
    },
    {
      "epoch": 2.362632751464844e-05,
      "step": 24774,
      "training_step_time": 9.494579553604126
    },
    {
      "epoch": 2.3627281188964842e-05,
      "model_forward_time": 1.7001185417175293,
      "step": 24775
    },
    {
      "epoch": 2.3627281188964842e-05,
      "step": 24775,
      "training_step_time": 9.359840869903564
    },
    {
      "epoch": 2.362823486328125e-05,
      "model_forward_time": 1.7232379913330078,
      "step": 24776
    },
    {
      "epoch": 2.362823486328125e-05,
      "step": 24776,
      "training_step_time": 7.653502941131592
    },
    {
      "epoch": 2.3629188537597657e-05,
      "model_forward_time": 1.7038514614105225,
      "step": 24777
    },
    {
      "epoch": 2.3629188537597657e-05,
      "step": 24777,
      "training_step_time": 7.492984294891357
    },
    {
      "epoch": 2.363014221191406e-05,
      "model_forward_time": 1.7073779106140137,
      "step": 24778
    },
    {
      "epoch": 2.363014221191406e-05,
      "step": 24778,
      "training_step_time": 7.643795728683472
    },
    {
      "epoch": 2.363109588623047e-05,
      "model_forward_time": 1.7072694301605225,
      "step": 24779
    },
    {
      "epoch": 2.363109588623047e-05,
      "step": 24779,
      "training_step_time": 8.259780883789062
    },
    {
      "epoch": 2.3632049560546876e-05,
      "grad_norm": 0.12226279824972153,
      "learning_rate": 8.968041340370622e-06,
      "loss": 0.0624,
      "step": 24780
    },
    {
      "epoch": 2.3632049560546876e-05,
      "model_forward_time": 1.6911396980285645,
      "step": 24780
    },
    {
      "dynamics_loss_avg": 0.05525638908147812,
      "epoch": 2.3632049560546876e-05,
      "step": 24780
    },
    {
      "action_loss_avg": 0.002282954810652882,
      "epoch": 2.3632049560546876e-05,
      "step": 24780
    },
    {
      "epoch": 2.3632049560546876e-05,
      "step": 24780,
      "training_step_time": 7.614809513092041
    },
    {
      "epoch": 2.363300323486328e-05,
      "model_forward_time": 1.7093040943145752,
      "step": 24781
    },
    {
      "epoch": 2.363300323486328e-05,
      "step": 24781,
      "training_step_time": 7.609692573547363
    },
    {
      "epoch": 2.3633956909179688e-05,
      "model_forward_time": 1.8695476055145264,
      "step": 24782
    },
    {
      "epoch": 2.3633956909179688e-05,
      "step": 24782,
      "training_step_time": 7.621503591537476
    },
    {
      "epoch": 2.3634910583496095e-05,
      "model_forward_time": 1.7476332187652588,
      "step": 24783
    },
    {
      "epoch": 2.3634910583496095e-05,
      "step": 24783,
      "training_step_time": 9.026731729507446
    },
    {
      "epoch": 2.36358642578125e-05,
      "model_forward_time": 1.7594773769378662,
      "step": 24784
    },
    {
      "epoch": 2.36358642578125e-05,
      "step": 24784,
      "training_step_time": 8.82028865814209
    },
    {
      "epoch": 2.3636817932128906e-05,
      "model_forward_time": 1.7861583232879639,
      "step": 24785
    },
    {
      "epoch": 2.3636817932128906e-05,
      "step": 24785,
      "training_step_time": 8.568438529968262
    },
    {
      "epoch": 2.3637771606445314e-05,
      "model_forward_time": 1.7649953365325928,
      "step": 24786
    },
    {
      "epoch": 2.3637771606445314e-05,
      "step": 24786,
      "training_step_time": 8.570460796356201
    },
    {
      "epoch": 2.3638725280761718e-05,
      "model_forward_time": 1.7605059146881104,
      "step": 24787
    },
    {
      "epoch": 2.3638725280761718e-05,
      "step": 24787,
      "training_step_time": 7.5305492877960205
    },
    {
      "epoch": 2.3639678955078125e-05,
      "model_forward_time": 1.7578909397125244,
      "step": 24788
    },
    {
      "epoch": 2.3639678955078125e-05,
      "step": 24788,
      "training_step_time": 8.006176471710205
    },
    {
      "epoch": 2.3640632629394533e-05,
      "model_forward_time": 1.75970458984375,
      "step": 24789
    },
    {
      "epoch": 2.3640632629394533e-05,
      "step": 24789,
      "training_step_time": 8.831841945648193
    },
    {
      "epoch": 2.3641586303710937e-05,
      "grad_norm": 0.12310564517974854,
      "learning_rate": 8.96703510433825e-06,
      "loss": 0.0629,
      "step": 24790
    },
    {
      "epoch": 2.3641586303710937e-05,
      "model_forward_time": 1.381105899810791,
      "step": 24790
    },
    {
      "dynamics_loss_avg": 0.06880815867334604,
      "epoch": 2.3641586303710937e-05,
      "step": 24790
    },
    {
      "action_loss_avg": 0.0021191761596128346,
      "epoch": 2.3641586303710937e-05,
      "step": 24790
    },
    {
      "epoch": 2.3641586303710937e-05,
      "step": 24790,
      "training_step_time": 7.659195899963379
    },
    {
      "epoch": 2.3642539978027344e-05,
      "model_forward_time": 2.8209893703460693,
      "step": 24791
    },
    {
      "epoch": 2.3642539978027344e-05,
      "step": 24791,
      "training_step_time": 8.716707229614258
    },
    {
      "epoch": 2.364349365234375e-05,
      "model_forward_time": 1.7403316497802734,
      "step": 24792
    },
    {
      "epoch": 2.364349365234375e-05,
      "step": 24792,
      "training_step_time": 9.092211246490479
    },
    {
      "epoch": 2.3644447326660156e-05,
      "model_forward_time": 1.7700729370117188,
      "step": 24793
    },
    {
      "epoch": 2.3644447326660156e-05,
      "step": 24793,
      "training_step_time": 8.276863098144531
    },
    {
      "epoch": 2.3645401000976563e-05,
      "model_forward_time": 1.7471728324890137,
      "step": 24794
    },
    {
      "epoch": 2.3645401000976563e-05,
      "step": 24794,
      "training_step_time": 7.604159832000732
    },
    {
      "epoch": 2.364635467529297e-05,
      "model_forward_time": 1.759934425354004,
      "step": 24795
    },
    {
      "epoch": 2.364635467529297e-05,
      "step": 24795,
      "training_step_time": 8.455639600753784
    },
    {
      "epoch": 2.3647308349609374e-05,
      "model_forward_time": 1.7521262168884277,
      "step": 24796
    },
    {
      "epoch": 2.3647308349609374e-05,
      "step": 24796,
      "training_step_time": 7.48699164390564
    },
    {
      "epoch": 2.3648262023925782e-05,
      "model_forward_time": 1.7559881210327148,
      "step": 24797
    },
    {
      "epoch": 2.3648262023925782e-05,
      "step": 24797,
      "training_step_time": 7.197509765625
    },
    {
      "epoch": 2.3649215698242186e-05,
      "model_forward_time": 1.378563404083252,
      "step": 24798
    },
    {
      "epoch": 2.3649215698242186e-05,
      "step": 24798,
      "training_step_time": 7.5664613246917725
    },
    {
      "epoch": 2.3650169372558593e-05,
      "model_forward_time": 1.743401050567627,
      "step": 24799
    },
    {
      "epoch": 2.3650169372558593e-05,
      "step": 24799,
      "training_step_time": 8.478983402252197
    },
    {
      "epoch": 2.3651123046875e-05,
      "grad_norm": 0.18398351967334747,
      "learning_rate": 8.966028434476884e-06,
      "loss": 0.0607,
      "step": 24800
    },
    {
      "epoch": 2.3651123046875e-05,
      "model_forward_time": 1.7425899505615234,
      "step": 24800
    },
    {
      "dynamics_loss_avg": 0.04879548829048872,
      "epoch": 2.3651123046875e-05,
      "step": 24800
    },
    {
      "action_loss_avg": 0.0019920508726499974,
      "epoch": 2.3651123046875e-05,
      "step": 24800
    },
    {
      "epoch": 2.3651123046875e-05,
      "step": 24800,
      "training_step_time": 9.362612009048462
    },
    {
      "epoch": 2.3652076721191405e-05,
      "model_forward_time": 2.90866756439209,
      "step": 24801
    },
    {
      "epoch": 2.3652076721191405e-05,
      "step": 24801,
      "training_step_time": 9.330916166305542
    },
    {
      "epoch": 2.3653030395507812e-05,
      "model_forward_time": 1.7532124519348145,
      "step": 24802
    },
    {
      "epoch": 2.3653030395507812e-05,
      "step": 24802,
      "training_step_time": 7.588054418563843
    },
    {
      "epoch": 2.365398406982422e-05,
      "model_forward_time": 1.7597222328186035,
      "step": 24803
    },
    {
      "epoch": 2.365398406982422e-05,
      "step": 24803,
      "training_step_time": 7.778462171554565
    },
    {
      "epoch": 2.3654937744140624e-05,
      "model_forward_time": 1.8665854930877686,
      "step": 24804
    },
    {
      "epoch": 2.3654937744140624e-05,
      "step": 24804,
      "training_step_time": 8.522302150726318
    },
    {
      "epoch": 2.365589141845703e-05,
      "model_forward_time": 1.7594547271728516,
      "step": 24805
    },
    {
      "epoch": 2.365589141845703e-05,
      "step": 24805,
      "training_step_time": 7.597989797592163
    },
    {
      "epoch": 2.365684509277344e-05,
      "model_forward_time": 1.7557549476623535,
      "step": 24806
    },
    {
      "epoch": 2.365684509277344e-05,
      "step": 24806,
      "training_step_time": 7.727841377258301
    },
    {
      "epoch": 2.3657798767089842e-05,
      "model_forward_time": 1.746222734451294,
      "step": 24807
    },
    {
      "epoch": 2.3657798767089842e-05,
      "step": 24807,
      "training_step_time": 7.684142351150513
    },
    {
      "epoch": 2.365875244140625e-05,
      "model_forward_time": 1.7527565956115723,
      "step": 24808
    },
    {
      "epoch": 2.365875244140625e-05,
      "step": 24808,
      "training_step_time": 7.744202375411987
    },
    {
      "epoch": 2.3659706115722657e-05,
      "model_forward_time": 1.751213550567627,
      "step": 24809
    },
    {
      "epoch": 2.3659706115722657e-05,
      "step": 24809,
      "training_step_time": 7.7682600021362305
    },
    {
      "epoch": 2.366065979003906e-05,
      "grad_norm": 0.20084445178508759,
      "learning_rate": 8.96502133089661e-06,
      "loss": 0.0635,
      "step": 24810
    },
    {
      "epoch": 2.366065979003906e-05,
      "model_forward_time": 2.8700082302093506,
      "step": 24810
    },
    {
      "dynamics_loss_avg": 0.06338135432451963,
      "epoch": 2.366065979003906e-05,
      "step": 24810
    },
    {
      "action_loss_avg": 0.00299809476127848,
      "epoch": 2.366065979003906e-05,
      "step": 24810
    },
    {
      "epoch": 2.366065979003906e-05,
      "step": 24810,
      "training_step_time": 7.738612413406372
    },
    {
      "epoch": 2.366161346435547e-05,
      "model_forward_time": 1.7597575187683105,
      "step": 24811
    },
    {
      "epoch": 2.366161346435547e-05,
      "step": 24811,
      "training_step_time": 7.68670392036438
    },
    {
      "epoch": 2.3662567138671876e-05,
      "model_forward_time": 1.772181749343872,
      "step": 24812
    },
    {
      "epoch": 2.3662567138671876e-05,
      "step": 24812,
      "training_step_time": 8.292738676071167
    },
    {
      "epoch": 2.366352081298828e-05,
      "model_forward_time": 1.8114345073699951,
      "step": 24813
    },
    {
      "epoch": 2.366352081298828e-05,
      "step": 24813,
      "training_step_time": 8.232662439346313
    },
    {
      "epoch": 2.3664474487304688e-05,
      "model_forward_time": 1.7708027362823486,
      "step": 24814
    },
    {
      "epoch": 2.3664474487304688e-05,
      "step": 24814,
      "training_step_time": 7.6139726638793945
    },
    {
      "epoch": 2.3665428161621095e-05,
      "model_forward_time": 1.7462129592895508,
      "step": 24815
    },
    {
      "epoch": 2.3665428161621095e-05,
      "step": 24815,
      "training_step_time": 7.781925916671753
    },
    {
      "epoch": 2.36663818359375e-05,
      "model_forward_time": 1.7479398250579834,
      "step": 24816
    },
    {
      "epoch": 2.36663818359375e-05,
      "step": 24816,
      "training_step_time": 7.701385021209717
    },
    {
      "epoch": 2.3667335510253906e-05,
      "model_forward_time": 1.7566370964050293,
      "step": 24817
    },
    {
      "epoch": 2.3667335510253906e-05,
      "step": 24817,
      "training_step_time": 7.668304920196533
    },
    {
      "epoch": 2.3668289184570314e-05,
      "model_forward_time": 1.7408676147460938,
      "step": 24818
    },
    {
      "epoch": 2.3668289184570314e-05,
      "step": 24818,
      "training_step_time": 9.527319431304932
    },
    {
      "epoch": 2.3669242858886718e-05,
      "model_forward_time": 2.8602218627929688,
      "step": 24819
    },
    {
      "epoch": 2.3669242858886718e-05,
      "step": 24819,
      "training_step_time": 7.786704063415527
    },
    {
      "epoch": 2.3670196533203125e-05,
      "grad_norm": 0.22624710202217102,
      "learning_rate": 8.964013793707564e-06,
      "loss": 0.0629,
      "step": 24820
    },
    {
      "epoch": 2.3670196533203125e-05,
      "model_forward_time": 1.7953553199768066,
      "step": 24820
    },
    {
      "dynamics_loss_avg": 0.06569296363741159,
      "epoch": 2.3670196533203125e-05,
      "step": 24820
    },
    {
      "action_loss_avg": 0.002449174097273499,
      "epoch": 2.3670196533203125e-05,
      "step": 24820
    },
    {
      "epoch": 2.3670196533203125e-05,
      "step": 24820,
      "training_step_time": 7.69430685043335
    },
    {
      "epoch": 2.3671150207519533e-05,
      "model_forward_time": 1.7103848457336426,
      "step": 24821
    },
    {
      "epoch": 2.3671150207519533e-05,
      "step": 24821,
      "training_step_time": 9.57752013206482
    },
    {
      "epoch": 2.3672103881835937e-05,
      "model_forward_time": 1.7245712280273438,
      "step": 24822
    },
    {
      "epoch": 2.3672103881835937e-05,
      "step": 24822,
      "training_step_time": 7.73237681388855
    },
    {
      "epoch": 2.3673057556152344e-05,
      "model_forward_time": 1.714771032333374,
      "step": 24823
    },
    {
      "epoch": 2.3673057556152344e-05,
      "step": 24823,
      "training_step_time": 7.755695104598999
    },
    {
      "epoch": 2.367401123046875e-05,
      "model_forward_time": 1.7164599895477295,
      "step": 24824
    },
    {
      "epoch": 2.367401123046875e-05,
      "step": 24824,
      "training_step_time": 7.745987892150879
    },
    {
      "epoch": 2.3674964904785156e-05,
      "model_forward_time": 1.7155795097351074,
      "step": 24825
    },
    {
      "epoch": 2.3674964904785156e-05,
      "step": 24825,
      "training_step_time": 7.6902875900268555
    },
    {
      "epoch": 2.3675918579101563e-05,
      "model_forward_time": 1.710728406906128,
      "step": 24826
    },
    {
      "epoch": 2.3675918579101563e-05,
      "step": 24826,
      "training_step_time": 7.653374671936035
    },
    {
      "epoch": 2.367687225341797e-05,
      "model_forward_time": 1.7141969203948975,
      "step": 24827
    },
    {
      "epoch": 2.367687225341797e-05,
      "step": 24827,
      "training_step_time": 8.197593688964844
    },
    {
      "epoch": 2.3677825927734374e-05,
      "model_forward_time": 1.8633410930633545,
      "step": 24828
    },
    {
      "epoch": 2.3677825927734374e-05,
      "step": 24828,
      "training_step_time": 8.47128438949585
    },
    {
      "epoch": 2.3678779602050782e-05,
      "model_forward_time": 1.7179803848266602,
      "step": 24829
    },
    {
      "epoch": 2.3678779602050782e-05,
      "step": 24829,
      "training_step_time": 8.635738611221313
    },
    {
      "epoch": 2.3679733276367186e-05,
      "grad_norm": 0.17777717113494873,
      "learning_rate": 8.963005823019932e-06,
      "loss": 0.0629,
      "step": 24830
    },
    {
      "epoch": 2.3679733276367186e-05,
      "model_forward_time": 1.700171709060669,
      "step": 24830
    },
    {
      "dynamics_loss_avg": 0.057823021151125434,
      "epoch": 2.3679733276367186e-05,
      "step": 24830
    },
    {
      "action_loss_avg": 0.002028585982043296,
      "epoch": 2.3679733276367186e-05,
      "step": 24830
    },
    {
      "epoch": 2.3679733276367186e-05,
      "step": 24830,
      "training_step_time": 7.6974711418151855
    },
    {
      "epoch": 2.3680686950683593e-05,
      "model_forward_time": 1.7187905311584473,
      "step": 24831
    },
    {
      "epoch": 2.3680686950683593e-05,
      "step": 24831,
      "training_step_time": 7.71977686882019
    },
    {
      "epoch": 2.3681640625e-05,
      "model_forward_time": 1.7096257209777832,
      "step": 24832
    },
    {
      "epoch": 2.3681640625e-05,
      "step": 24832,
      "training_step_time": 7.652787446975708
    },
    {
      "epoch": 2.3682594299316405e-05,
      "model_forward_time": 1.705653190612793,
      "step": 24833
    },
    {
      "epoch": 2.3682594299316405e-05,
      "step": 24833,
      "training_step_time": 7.535089015960693
    },
    {
      "epoch": 2.3683547973632812e-05,
      "model_forward_time": 1.7003569602966309,
      "step": 24834
    },
    {
      "epoch": 2.3683547973632812e-05,
      "step": 24834,
      "training_step_time": 7.829695224761963
    },
    {
      "epoch": 2.368450164794922e-05,
      "model_forward_time": 1.7178866863250732,
      "step": 24835
    },
    {
      "epoch": 2.368450164794922e-05,
      "step": 24835,
      "training_step_time": 8.08361554145813
    },
    {
      "epoch": 2.3685455322265624e-05,
      "model_forward_time": 1.7111611366271973,
      "step": 24836
    },
    {
      "epoch": 2.3685455322265624e-05,
      "step": 24836,
      "training_step_time": 8.703129291534424
    },
    {
      "epoch": 2.368640899658203e-05,
      "model_forward_time": 1.7097647190093994,
      "step": 24837
    },
    {
      "epoch": 2.368640899658203e-05,
      "step": 24837,
      "training_step_time": 7.672762393951416
    },
    {
      "epoch": 2.368736267089844e-05,
      "model_forward_time": 2.273016929626465,
      "step": 24838
    },
    {
      "epoch": 2.368736267089844e-05,
      "step": 24838,
      "training_step_time": 7.5558555126190186
    },
    {
      "epoch": 2.3688316345214842e-05,
      "model_forward_time": 1.7088966369628906,
      "step": 24839
    },
    {
      "epoch": 2.3688316345214842e-05,
      "step": 24839,
      "training_step_time": 7.697749614715576
    },
    {
      "epoch": 2.368927001953125e-05,
      "grad_norm": 0.1351853311061859,
      "learning_rate": 8.961997418943939e-06,
      "loss": 0.06,
      "step": 24840
    },
    {
      "epoch": 2.368927001953125e-05,
      "model_forward_time": 1.7109849452972412,
      "step": 24840
    },
    {
      "dynamics_loss_avg": 0.06769832437857985,
      "epoch": 2.368927001953125e-05,
      "step": 24840
    },
    {
      "action_loss_avg": 0.004175775445764885,
      "epoch": 2.368927001953125e-05,
      "step": 24840
    },
    {
      "epoch": 2.368927001953125e-05,
      "step": 24840,
      "training_step_time": 7.697130441665649
    },
    {
      "epoch": 2.3690223693847657e-05,
      "model_forward_time": 1.7125344276428223,
      "step": 24841
    },
    {
      "epoch": 2.3690223693847657e-05,
      "step": 24841,
      "training_step_time": 7.71693754196167
    },
    {
      "epoch": 2.369117736816406e-05,
      "model_forward_time": 1.7101466655731201,
      "step": 24842
    },
    {
      "epoch": 2.369117736816406e-05,
      "step": 24842,
      "training_step_time": 7.546049118041992
    },
    {
      "epoch": 2.369213104248047e-05,
      "model_forward_time": 1.714970350265503,
      "step": 24843
    },
    {
      "epoch": 2.369213104248047e-05,
      "step": 24843,
      "training_step_time": 8.70405125617981
    },
    {
      "epoch": 2.3693084716796876e-05,
      "model_forward_time": 1.7173388004302979,
      "step": 24844
    },
    {
      "epoch": 2.3693084716796876e-05,
      "step": 24844,
      "training_step_time": 7.573938369750977
    },
    {
      "epoch": 2.369403839111328e-05,
      "model_forward_time": 1.7149393558502197,
      "step": 24845
    },
    {
      "epoch": 2.369403839111328e-05,
      "step": 24845,
      "training_step_time": 7.675413608551025
    },
    {
      "epoch": 2.3694992065429688e-05,
      "model_forward_time": 1.7314906120300293,
      "step": 24846
    },
    {
      "epoch": 2.3694992065429688e-05,
      "step": 24846,
      "training_step_time": 9.053287744522095
    },
    {
      "epoch": 2.3695945739746095e-05,
      "model_forward_time": 2.292442798614502,
      "step": 24847
    },
    {
      "epoch": 2.3695945739746095e-05,
      "step": 24847,
      "training_step_time": 8.642723798751831
    },
    {
      "epoch": 2.36968994140625e-05,
      "model_forward_time": 1.7012059688568115,
      "step": 24848
    },
    {
      "epoch": 2.36968994140625e-05,
      "step": 24848,
      "training_step_time": 8.718507766723633
    },
    {
      "epoch": 2.3697853088378906e-05,
      "model_forward_time": 1.3231167793273926,
      "step": 24849
    },
    {
      "epoch": 2.3697853088378906e-05,
      "step": 24849,
      "training_step_time": 8.250322103500366
    },
    {
      "epoch": 2.3698806762695314e-05,
      "grad_norm": 0.16132815182209015,
      "learning_rate": 8.960988581589866e-06,
      "loss": 0.0618,
      "step": 24850
    },
    {
      "epoch": 2.3698806762695314e-05,
      "model_forward_time": 1.7157297134399414,
      "step": 24850
    },
    {
      "dynamics_loss_avg": 0.049431861564517024,
      "epoch": 2.3698806762695314e-05,
      "step": 24850
    },
    {
      "action_loss_avg": 0.00207570954808034,
      "epoch": 2.3698806762695314e-05,
      "step": 24850
    },
    {
      "epoch": 2.3698806762695314e-05,
      "step": 24850,
      "training_step_time": 7.702070951461792
    },
    {
      "epoch": 2.3699760437011718e-05,
      "model_forward_time": 1.70975923538208,
      "step": 24851
    },
    {
      "epoch": 2.3699760437011718e-05,
      "step": 24851,
      "training_step_time": 7.575071811676025
    },
    {
      "epoch": 2.3700714111328125e-05,
      "model_forward_time": 1.7064704895019531,
      "step": 24852
    },
    {
      "epoch": 2.3700714111328125e-05,
      "step": 24852,
      "training_step_time": 7.6663124561309814
    },
    {
      "epoch": 2.3701667785644533e-05,
      "model_forward_time": 1.7159161567687988,
      "step": 24853
    },
    {
      "epoch": 2.3701667785644533e-05,
      "step": 24853,
      "training_step_time": 7.634748935699463
    },
    {
      "epoch": 2.3702621459960937e-05,
      "model_forward_time": 1.704996109008789,
      "step": 24854
    },
    {
      "epoch": 2.3702621459960937e-05,
      "step": 24854,
      "training_step_time": 7.723509311676025
    },
    {
      "epoch": 2.3703575134277344e-05,
      "model_forward_time": 1.7128968238830566,
      "step": 24855
    },
    {
      "epoch": 2.3703575134277344e-05,
      "step": 24855,
      "training_step_time": 7.7729504108428955
    },
    {
      "epoch": 2.370452880859375e-05,
      "model_forward_time": 2.2808663845062256,
      "step": 24856
    },
    {
      "epoch": 2.370452880859375e-05,
      "step": 24856,
      "training_step_time": 8.341274499893188
    },
    {
      "epoch": 2.3705482482910156e-05,
      "model_forward_time": 1.7160179615020752,
      "step": 24857
    },
    {
      "epoch": 2.3705482482910156e-05,
      "step": 24857,
      "training_step_time": 7.6611714363098145
    },
    {
      "epoch": 2.3706436157226563e-05,
      "model_forward_time": 1.7093610763549805,
      "step": 24858
    },
    {
      "epoch": 2.3706436157226563e-05,
      "step": 24858,
      "training_step_time": 7.696784257888794
    },
    {
      "epoch": 2.370738983154297e-05,
      "model_forward_time": 1.7203631401062012,
      "step": 24859
    },
    {
      "epoch": 2.370738983154297e-05,
      "step": 24859,
      "training_step_time": 7.703415632247925
    },
    {
      "epoch": 2.3708343505859374e-05,
      "grad_norm": 0.15502186119556427,
      "learning_rate": 8.959979311068037e-06,
      "loss": 0.0589,
      "step": 24860
    },
    {
      "epoch": 2.3708343505859374e-05,
      "model_forward_time": 1.7027959823608398,
      "step": 24860
    },
    {
      "dynamics_loss_avg": 0.05541989933699369,
      "epoch": 2.3708343505859374e-05,
      "step": 24860
    },
    {
      "action_loss_avg": 0.0025398606609087437,
      "epoch": 2.3708343505859374e-05,
      "step": 24860
    },
    {
      "epoch": 2.3708343505859374e-05,
      "step": 24860,
      "training_step_time": 7.728860139846802
    },
    {
      "epoch": 2.3709297180175782e-05,
      "model_forward_time": 1.7082667350769043,
      "step": 24861
    },
    {
      "epoch": 2.3709297180175782e-05,
      "step": 24861,
      "training_step_time": 7.130630731582642
    },
    {
      "epoch": 2.3710250854492186e-05,
      "model_forward_time": 1.7112205028533936,
      "step": 24862
    },
    {
      "epoch": 2.3710250854492186e-05,
      "step": 24862,
      "training_step_time": 7.735912084579468
    },
    {
      "epoch": 2.3711204528808593e-05,
      "model_forward_time": 1.7136001586914062,
      "step": 24863
    },
    {
      "epoch": 2.3711204528808593e-05,
      "step": 24863,
      "training_step_time": 8.535142421722412
    },
    {
      "epoch": 2.3712158203125e-05,
      "model_forward_time": 1.7078914642333984,
      "step": 24864
    },
    {
      "epoch": 2.3712158203125e-05,
      "step": 24864,
      "training_step_time": 7.697096586227417
    },
    {
      "epoch": 2.3713111877441405e-05,
      "model_forward_time": 2.259122848510742,
      "step": 24865
    },
    {
      "epoch": 2.3713111877441405e-05,
      "step": 24865,
      "training_step_time": 8.67379379272461
    },
    {
      "epoch": 2.3714065551757812e-05,
      "model_forward_time": 1.7215142250061035,
      "step": 24866
    },
    {
      "epoch": 2.3714065551757812e-05,
      "step": 24866,
      "training_step_time": 7.679100751876831
    },
    {
      "epoch": 2.371501922607422e-05,
      "model_forward_time": 1.710946798324585,
      "step": 24867
    },
    {
      "epoch": 2.371501922607422e-05,
      "step": 24867,
      "training_step_time": 8.651573896408081
    },
    {
      "epoch": 2.3715972900390624e-05,
      "model_forward_time": 1.7148418426513672,
      "step": 24868
    },
    {
      "epoch": 2.3715972900390624e-05,
      "step": 24868,
      "training_step_time": 7.666209697723389
    },
    {
      "epoch": 2.371692657470703e-05,
      "model_forward_time": 1.7084877490997314,
      "step": 24869
    },
    {
      "epoch": 2.371692657470703e-05,
      "step": 24869,
      "training_step_time": 7.770431995391846
    },
    {
      "epoch": 2.371788024902344e-05,
      "grad_norm": 0.1445644348859787,
      "learning_rate": 8.958969607488823e-06,
      "loss": 0.0586,
      "step": 24870
    },
    {
      "epoch": 2.371788024902344e-05,
      "model_forward_time": 1.6794958114624023,
      "step": 24870
    },
    {
      "dynamics_loss_avg": 0.06349738761782646,
      "epoch": 2.371788024902344e-05,
      "step": 24870
    },
    {
      "action_loss_avg": 0.00254861896391958,
      "epoch": 2.371788024902344e-05,
      "step": 24870
    },
    {
      "epoch": 2.371788024902344e-05,
      "step": 24870,
      "training_step_time": 7.260403394699097
    },
    {
      "epoch": 2.3718833923339842e-05,
      "model_forward_time": 1.6868631839752197,
      "step": 24871
    },
    {
      "epoch": 2.3718833923339842e-05,
      "step": 24871,
      "training_step_time": 7.697549819946289
    },
    {
      "epoch": 2.371978759765625e-05,
      "model_forward_time": 1.684713363647461,
      "step": 24872
    },
    {
      "epoch": 2.371978759765625e-05,
      "step": 24872,
      "training_step_time": 8.706554174423218
    },
    {
      "epoch": 2.3720741271972657e-05,
      "model_forward_time": 1.697484016418457,
      "step": 24873
    },
    {
      "epoch": 2.3720741271972657e-05,
      "step": 24873,
      "training_step_time": 8.794870138168335
    },
    {
      "epoch": 2.372169494628906e-05,
      "model_forward_time": 1.6872997283935547,
      "step": 24874
    },
    {
      "epoch": 2.372169494628906e-05,
      "step": 24874,
      "training_step_time": 7.755268812179565
    },
    {
      "epoch": 2.372264862060547e-05,
      "model_forward_time": 2.2425537109375,
      "step": 24875
    },
    {
      "epoch": 2.372264862060547e-05,
      "step": 24875,
      "training_step_time": 8.148659467697144
    },
    {
      "epoch": 2.3723602294921876e-05,
      "model_forward_time": 1.7058768272399902,
      "step": 24876
    },
    {
      "epoch": 2.3723602294921876e-05,
      "step": 24876,
      "training_step_time": 7.682008743286133
    },
    {
      "epoch": 2.372455596923828e-05,
      "model_forward_time": 1.7062811851501465,
      "step": 24877
    },
    {
      "epoch": 2.372455596923828e-05,
      "step": 24877,
      "training_step_time": 8.242632865905762
    },
    {
      "epoch": 2.3725509643554688e-05,
      "model_forward_time": 1.7098946571350098,
      "step": 24878
    },
    {
      "epoch": 2.3725509643554688e-05,
      "step": 24878,
      "training_step_time": 7.699641466140747
    },
    {
      "epoch": 2.3726463317871095e-05,
      "model_forward_time": 1.6778051853179932,
      "step": 24879
    },
    {
      "epoch": 2.3726463317871095e-05,
      "step": 24879,
      "training_step_time": 7.56668758392334
    },
    {
      "epoch": 2.37274169921875e-05,
      "grad_norm": 0.16847434639930725,
      "learning_rate": 8.957959470962647e-06,
      "loss": 0.0615,
      "step": 24880
    },
    {
      "epoch": 2.37274169921875e-05,
      "model_forward_time": 1.7024714946746826,
      "step": 24880
    },
    {
      "dynamics_loss_avg": 0.05968657787889242,
      "epoch": 2.37274169921875e-05,
      "step": 24880
    },
    {
      "action_loss_avg": 0.001999460376100615,
      "epoch": 2.37274169921875e-05,
      "step": 24880
    },
    {
      "epoch": 2.37274169921875e-05,
      "step": 24880,
      "training_step_time": 7.70631217956543
    },
    {
      "epoch": 2.3728370666503906e-05,
      "model_forward_time": 1.7179896831512451,
      "step": 24881
    },
    {
      "epoch": 2.3728370666503906e-05,
      "step": 24881,
      "training_step_time": 7.6850426197052
    },
    {
      "epoch": 2.3729324340820314e-05,
      "model_forward_time": 1.713066816329956,
      "step": 24882
    },
    {
      "epoch": 2.3729324340820314e-05,
      "step": 24882,
      "training_step_time": 8.38283395767212
    },
    {
      "epoch": 2.3730278015136718e-05,
      "model_forward_time": 1.7136664390563965,
      "step": 24883
    },
    {
      "epoch": 2.3730278015136718e-05,
      "step": 24883,
      "training_step_time": 8.458942651748657
    },
    {
      "epoch": 2.3731231689453125e-05,
      "model_forward_time": 2.2621750831604004,
      "step": 24884
    },
    {
      "epoch": 2.3731231689453125e-05,
      "step": 24884,
      "training_step_time": 7.673992156982422
    },
    {
      "epoch": 2.3732185363769533e-05,
      "model_forward_time": 1.7120099067687988,
      "step": 24885
    },
    {
      "epoch": 2.3732185363769533e-05,
      "step": 24885,
      "training_step_time": 7.709072828292847
    },
    {
      "epoch": 2.3733139038085937e-05,
      "model_forward_time": 1.7195701599121094,
      "step": 24886
    },
    {
      "epoch": 2.3733139038085937e-05,
      "step": 24886,
      "training_step_time": 8.868380308151245
    },
    {
      "epoch": 2.3734092712402344e-05,
      "model_forward_time": 1.7025330066680908,
      "step": 24887
    },
    {
      "epoch": 2.3734092712402344e-05,
      "step": 24887,
      "training_step_time": 7.819446563720703
    },
    {
      "epoch": 2.373504638671875e-05,
      "model_forward_time": 1.716130256652832,
      "step": 24888
    },
    {
      "epoch": 2.373504638671875e-05,
      "step": 24888,
      "training_step_time": 7.778759479522705
    },
    {
      "epoch": 2.3736000061035156e-05,
      "model_forward_time": 1.7121367454528809,
      "step": 24889
    },
    {
      "epoch": 2.3736000061035156e-05,
      "step": 24889,
      "training_step_time": 9.264559030532837
    },
    {
      "epoch": 2.3736953735351563e-05,
      "grad_norm": 0.1164194792509079,
      "learning_rate": 8.956948901599972e-06,
      "loss": 0.0586,
      "step": 24890
    },
    {
      "epoch": 2.3736953735351563e-05,
      "model_forward_time": 1.6947100162506104,
      "step": 24890
    },
    {
      "dynamics_loss_avg": 0.0490273293107748,
      "epoch": 2.3736953735351563e-05,
      "step": 24890
    },
    {
      "action_loss_avg": 0.0022292998095508666,
      "epoch": 2.3736953735351563e-05,
      "step": 24890
    },
    {
      "epoch": 2.3736953735351563e-05,
      "step": 24890,
      "training_step_time": 8.622748851776123
    },
    {
      "epoch": 2.3737907409667967e-05,
      "model_forward_time": 1.6974983215332031,
      "step": 24891
    },
    {
      "epoch": 2.3737907409667967e-05,
      "step": 24891,
      "training_step_time": 7.74524188041687
    },
    {
      "epoch": 2.3738861083984374e-05,
      "model_forward_time": 1.6968367099761963,
      "step": 24892
    },
    {
      "epoch": 2.3738861083984374e-05,
      "step": 24892,
      "training_step_time": 8.806782722473145
    },
    {
      "epoch": 2.3739814758300782e-05,
      "model_forward_time": 2.275265693664551,
      "step": 24893
    },
    {
      "epoch": 2.3739814758300782e-05,
      "step": 24893,
      "training_step_time": 7.681028127670288
    },
    {
      "epoch": 2.3740768432617186e-05,
      "model_forward_time": 1.6986942291259766,
      "step": 24894
    },
    {
      "epoch": 2.3740768432617186e-05,
      "step": 24894,
      "training_step_time": 9.288772106170654
    },
    {
      "epoch": 2.3741722106933593e-05,
      "model_forward_time": 1.7015578746795654,
      "step": 24895
    },
    {
      "epoch": 2.3741722106933593e-05,
      "step": 24895,
      "training_step_time": 7.773633718490601
    },
    {
      "epoch": 2.374267578125e-05,
      "model_forward_time": 1.705444574356079,
      "step": 24896
    },
    {
      "epoch": 2.374267578125e-05,
      "step": 24896,
      "training_step_time": 7.662311553955078
    },
    {
      "epoch": 2.3743629455566405e-05,
      "model_forward_time": 1.7035729885101318,
      "step": 24897
    },
    {
      "epoch": 2.3743629455566405e-05,
      "step": 24897,
      "training_step_time": 7.823460578918457
    },
    {
      "epoch": 2.3744583129882812e-05,
      "model_forward_time": 1.7058632373809814,
      "step": 24898
    },
    {
      "epoch": 2.3744583129882812e-05,
      "step": 24898,
      "training_step_time": 7.7408506870269775
    },
    {
      "epoch": 2.374553680419922e-05,
      "model_forward_time": 1.6951682567596436,
      "step": 24899
    },
    {
      "epoch": 2.374553680419922e-05,
      "step": 24899,
      "training_step_time": 7.882472038269043
    },
    {
      "epoch": 2.3746490478515624e-05,
      "grad_norm": 0.10898245871067047,
      "learning_rate": 8.955937899511315e-06,
      "loss": 0.0582,
      "step": 24900
    },
    {
      "epoch": 2.3746490478515624e-05,
      "model_forward_time": 1.6915700435638428,
      "step": 24900
    },
    {
      "dynamics_loss_avg": 0.050135312043130396,
      "epoch": 2.3746490478515624e-05,
      "step": 24900
    },
    {
      "action_loss_avg": 0.002440100465901196,
      "epoch": 2.3746490478515624e-05,
      "step": 24900
    },
    {
      "epoch": 2.3746490478515624e-05,
      "step": 24900,
      "training_step_time": 7.824177503585815
    },
    {
      "epoch": 2.374744415283203e-05,
      "model_forward_time": 1.7080464363098145,
      "step": 24901
    },
    {
      "epoch": 2.374744415283203e-05,
      "step": 24901,
      "training_step_time": 8.81929087638855
    },
    {
      "epoch": 2.374839782714844e-05,
      "model_forward_time": 1.858332633972168,
      "step": 24902
    },
    {
      "epoch": 2.374839782714844e-05,
      "step": 24902,
      "training_step_time": 9.488856315612793
    },
    {
      "epoch": 2.3749351501464843e-05,
      "model_forward_time": 1.7059776782989502,
      "step": 24903
    },
    {
      "epoch": 2.3749351501464843e-05,
      "step": 24903,
      "training_step_time": 9.306551456451416
    },
    {
      "epoch": 2.375030517578125e-05,
      "model_forward_time": 1.700453758239746,
      "step": 24904
    },
    {
      "epoch": 2.375030517578125e-05,
      "step": 24904,
      "training_step_time": 7.738372802734375
    },
    {
      "epoch": 2.3751258850097657e-05,
      "model_forward_time": 1.726860523223877,
      "step": 24905
    },
    {
      "epoch": 2.3751258850097657e-05,
      "step": 24905,
      "training_step_time": 8.684881687164307
    },
    {
      "epoch": 2.375221252441406e-05,
      "model_forward_time": 1.6980762481689453,
      "step": 24906
    },
    {
      "epoch": 2.375221252441406e-05,
      "step": 24906,
      "training_step_time": 7.730737209320068
    },
    {
      "epoch": 2.375316619873047e-05,
      "model_forward_time": 1.7041747570037842,
      "step": 24907
    },
    {
      "epoch": 2.375316619873047e-05,
      "step": 24907,
      "training_step_time": 7.682713508605957
    },
    {
      "epoch": 2.3754119873046876e-05,
      "model_forward_time": 1.698847770690918,
      "step": 24908
    },
    {
      "epoch": 2.3754119873046876e-05,
      "step": 24908,
      "training_step_time": 7.773975849151611
    },
    {
      "epoch": 2.375507354736328e-05,
      "model_forward_time": 1.7045691013336182,
      "step": 24909
    },
    {
      "epoch": 2.375507354736328e-05,
      "step": 24909,
      "training_step_time": 7.7563700675964355
    },
    {
      "epoch": 2.3756027221679688e-05,
      "grad_norm": 0.12736476957798004,
      "learning_rate": 8.954926464807238e-06,
      "loss": 0.0621,
      "step": 24910
    },
    {
      "epoch": 2.3756027221679688e-05,
      "model_forward_time": 1.700833797454834,
      "step": 24910
    },
    {
      "dynamics_loss_avg": 0.056279781833291055,
      "epoch": 2.3756027221679688e-05,
      "step": 24910
    },
    {
      "action_loss_avg": 0.002861837606178597,
      "epoch": 2.3756027221679688e-05,
      "step": 24910
    },
    {
      "epoch": 2.3756027221679688e-05,
      "step": 24910,
      "training_step_time": 7.6153953075408936
    },
    {
      "epoch": 2.3756980895996095e-05,
      "model_forward_time": 2.2330527305603027,
      "step": 24911
    },
    {
      "epoch": 2.3756980895996095e-05,
      "step": 24911,
      "training_step_time": 7.79578709602356
    },
    {
      "epoch": 2.37579345703125e-05,
      "model_forward_time": 1.7286274433135986,
      "step": 24912
    },
    {
      "epoch": 2.37579345703125e-05,
      "step": 24912,
      "training_step_time": 8.326573371887207
    },
    {
      "epoch": 2.3758888244628906e-05,
      "model_forward_time": 1.7058994770050049,
      "step": 24913
    },
    {
      "epoch": 2.3758888244628906e-05,
      "step": 24913,
      "training_step_time": 8.620211601257324
    },
    {
      "epoch": 2.3759841918945314e-05,
      "model_forward_time": 1.7046737670898438,
      "step": 24914
    },
    {
      "epoch": 2.3759841918945314e-05,
      "step": 24914,
      "training_step_time": 9.073796510696411
    },
    {
      "epoch": 2.3760795593261718e-05,
      "model_forward_time": 1.7014341354370117,
      "step": 24915
    },
    {
      "epoch": 2.3760795593261718e-05,
      "step": 24915,
      "training_step_time": 7.669227838516235
    },
    {
      "epoch": 2.3761749267578125e-05,
      "model_forward_time": 1.712040662765503,
      "step": 24916
    },
    {
      "epoch": 2.3761749267578125e-05,
      "step": 24916,
      "training_step_time": 8.125067472457886
    },
    {
      "epoch": 2.3762702941894533e-05,
      "model_forward_time": 1.3174476623535156,
      "step": 24917
    },
    {
      "epoch": 2.3762702941894533e-05,
      "step": 24917,
      "training_step_time": 8.347566604614258
    },
    {
      "epoch": 2.3763656616210937e-05,
      "model_forward_time": 1.7130603790283203,
      "step": 24918
    },
    {
      "epoch": 2.3763656616210937e-05,
      "step": 24918,
      "training_step_time": 7.112403392791748
    },
    {
      "epoch": 2.3764610290527344e-05,
      "model_forward_time": 1.7032811641693115,
      "step": 24919
    },
    {
      "epoch": 2.3764610290527344e-05,
      "step": 24919,
      "training_step_time": 7.760297775268555
    },
    {
      "epoch": 2.376556396484375e-05,
      "grad_norm": 0.15856996178627014,
      "learning_rate": 8.953914597598347e-06,
      "loss": 0.0636,
      "step": 24920
    },
    {
      "epoch": 2.376556396484375e-05,
      "model_forward_time": 1.6990044116973877,
      "step": 24920
    },
    {
      "dynamics_loss_avg": 0.04898442644625902,
      "epoch": 2.376556396484375e-05,
      "step": 24920
    },
    {
      "action_loss_avg": 0.0017337249941192568,
      "epoch": 2.376556396484375e-05,
      "step": 24920
    },
    {
      "epoch": 2.376556396484375e-05,
      "step": 24920,
      "training_step_time": 8.487900972366333
    },
    {
      "epoch": 2.3766517639160156e-05,
      "model_forward_time": 2.257399320602417,
      "step": 24921
    },
    {
      "epoch": 2.3766517639160156e-05,
      "step": 24921,
      "training_step_time": 8.677542448043823
    },
    {
      "epoch": 2.3767471313476563e-05,
      "model_forward_time": 1.7023282051086426,
      "step": 24922
    },
    {
      "epoch": 2.3767471313476563e-05,
      "step": 24922,
      "training_step_time": 8.55888557434082
    },
    {
      "epoch": 2.3768424987792967e-05,
      "model_forward_time": 1.7088086605072021,
      "step": 24923
    },
    {
      "epoch": 2.3768424987792967e-05,
      "step": 24923,
      "training_step_time": 7.619289398193359
    },
    {
      "epoch": 2.3769378662109375e-05,
      "model_forward_time": 1.7081050872802734,
      "step": 24924
    },
    {
      "epoch": 2.3769378662109375e-05,
      "step": 24924,
      "training_step_time": 7.732266902923584
    },
    {
      "epoch": 2.3770332336425782e-05,
      "model_forward_time": 1.7057149410247803,
      "step": 24925
    },
    {
      "epoch": 2.3770332336425782e-05,
      "step": 24925,
      "training_step_time": 7.703234910964966
    },
    {
      "epoch": 2.3771286010742186e-05,
      "model_forward_time": 1.7106153964996338,
      "step": 24926
    },
    {
      "epoch": 2.3771286010742186e-05,
      "step": 24926,
      "training_step_time": 7.644545316696167
    },
    {
      "epoch": 2.3772239685058593e-05,
      "model_forward_time": 1.7059032917022705,
      "step": 24927
    },
    {
      "epoch": 2.3772239685058593e-05,
      "step": 24927,
      "training_step_time": 7.785688638687134
    },
    {
      "epoch": 2.3773193359375e-05,
      "model_forward_time": 1.6980359554290771,
      "step": 24928
    },
    {
      "epoch": 2.3773193359375e-05,
      "step": 24928,
      "training_step_time": 7.749479532241821
    },
    {
      "epoch": 2.3774147033691405e-05,
      "model_forward_time": 1.7086970806121826,
      "step": 24929
    },
    {
      "epoch": 2.3774147033691405e-05,
      "step": 24929,
      "training_step_time": 8.814946174621582
    },
    {
      "epoch": 2.3775100708007812e-05,
      "grad_norm": 0.10672606527805328,
      "learning_rate": 8.952902297995302e-06,
      "loss": 0.0609,
      "step": 24930
    },
    {
      "epoch": 2.3775100708007812e-05,
      "model_forward_time": 1.8677105903625488,
      "step": 24930
    },
    {
      "dynamics_loss_avg": 0.05340635813772678,
      "epoch": 2.3775100708007812e-05,
      "step": 24930
    },
    {
      "action_loss_avg": 0.003581881080754101,
      "epoch": 2.3775100708007812e-05,
      "step": 24930
    },
    {
      "epoch": 2.3775100708007812e-05,
      "step": 24930,
      "training_step_time": 7.672159671783447
    },
    {
      "epoch": 2.377605438232422e-05,
      "model_forward_time": 1.7010164260864258,
      "step": 24931
    },
    {
      "epoch": 2.377605438232422e-05,
      "step": 24931,
      "training_step_time": 7.739266872406006
    },
    {
      "epoch": 2.3777008056640624e-05,
      "model_forward_time": 0.9769058227539062,
      "step": 24932
    },
    {
      "epoch": 2.3777008056640624e-05,
      "step": 24932,
      "training_step_time": 7.564927816390991
    },
    {
      "epoch": 2.377796173095703e-05,
      "model_forward_time": 1.7039387226104736,
      "step": 24933
    },
    {
      "epoch": 2.377796173095703e-05,
      "step": 24933,
      "training_step_time": 7.572360277175903
    },
    {
      "epoch": 2.377891540527344e-05,
      "model_forward_time": 1.713944435119629,
      "step": 24934
    },
    {
      "epoch": 2.377891540527344e-05,
      "step": 24934,
      "training_step_time": 7.646234750747681
    },
    {
      "epoch": 2.3779869079589843e-05,
      "model_forward_time": 1.316746473312378,
      "step": 24935
    },
    {
      "epoch": 2.3779869079589843e-05,
      "step": 24935,
      "training_step_time": 7.875531196594238
    },
    {
      "epoch": 2.378082275390625e-05,
      "model_forward_time": 1.703279972076416,
      "step": 24936
    },
    {
      "epoch": 2.378082275390625e-05,
      "step": 24936,
      "training_step_time": 7.681894779205322
    },
    {
      "epoch": 2.3781776428222657e-05,
      "model_forward_time": 1.7120416164398193,
      "step": 24937
    },
    {
      "epoch": 2.3781776428222657e-05,
      "step": 24937,
      "training_step_time": 8.327208042144775
    },
    {
      "epoch": 2.378273010253906e-05,
      "model_forward_time": 1.6995258331298828,
      "step": 24938
    },
    {
      "epoch": 2.378273010253906e-05,
      "step": 24938,
      "training_step_time": 8.224158525466919
    },
    {
      "epoch": 2.378368377685547e-05,
      "model_forward_time": 1.7226238250732422,
      "step": 24939
    },
    {
      "epoch": 2.378368377685547e-05,
      "step": 24939,
      "training_step_time": 7.568359375
    },
    {
      "epoch": 2.3784637451171876e-05,
      "grad_norm": 0.1524246633052826,
      "learning_rate": 8.951889566108805e-06,
      "loss": 0.0642,
      "step": 24940
    },
    {
      "epoch": 2.3784637451171876e-05,
      "model_forward_time": 2.2569167613983154,
      "step": 24940
    },
    {
      "dynamics_loss_avg": 0.0596330595202744,
      "epoch": 2.3784637451171876e-05,
      "step": 24940
    },
    {
      "action_loss_avg": 0.0016078577318694443,
      "epoch": 2.3784637451171876e-05,
      "step": 24940
    },
    {
      "epoch": 2.3784637451171876e-05,
      "step": 24940,
      "training_step_time": 10.336490154266357
    },
    {
      "epoch": 2.378559112548828e-05,
      "model_forward_time": 1.690425157546997,
      "step": 24941
    },
    {
      "epoch": 2.378559112548828e-05,
      "step": 24941,
      "training_step_time": 7.658429384231567
    },
    {
      "epoch": 2.3786544799804688e-05,
      "model_forward_time": 1.6986563205718994,
      "step": 24942
    },
    {
      "epoch": 2.3786544799804688e-05,
      "step": 24942,
      "training_step_time": 7.620595932006836
    },
    {
      "epoch": 2.3787498474121095e-05,
      "model_forward_time": 1.7138142585754395,
      "step": 24943
    },
    {
      "epoch": 2.3787498474121095e-05,
      "step": 24943,
      "training_step_time": 8.708192110061646
    },
    {
      "epoch": 2.37884521484375e-05,
      "model_forward_time": 1.6994044780731201,
      "step": 24944
    },
    {
      "epoch": 2.37884521484375e-05,
      "step": 24944,
      "training_step_time": 7.671154737472534
    },
    {
      "epoch": 2.3789405822753907e-05,
      "model_forward_time": 1.6938250064849854,
      "step": 24945
    },
    {
      "epoch": 2.3789405822753907e-05,
      "step": 24945,
      "training_step_time": 7.732797145843506
    },
    {
      "epoch": 2.3790359497070314e-05,
      "model_forward_time": 1.6965117454528809,
      "step": 24946
    },
    {
      "epoch": 2.3790359497070314e-05,
      "step": 24946,
      "training_step_time": 7.731005668640137
    },
    {
      "epoch": 2.3791313171386718e-05,
      "model_forward_time": 1.6997087001800537,
      "step": 24947
    },
    {
      "epoch": 2.3791313171386718e-05,
      "step": 24947,
      "training_step_time": 7.650306940078735
    },
    {
      "epoch": 2.3792266845703125e-05,
      "model_forward_time": 1.6974763870239258,
      "step": 24948
    },
    {
      "epoch": 2.3792266845703125e-05,
      "step": 24948,
      "training_step_time": 9.483378410339355
    },
    {
      "epoch": 2.3793220520019533e-05,
      "model_forward_time": 2.2543416023254395,
      "step": 24949
    },
    {
      "epoch": 2.3793220520019533e-05,
      "step": 24949,
      "training_step_time": 9.290562152862549
    },
    {
      "epoch": 2.3794174194335937e-05,
      "grad_norm": 0.17887380719184875,
      "learning_rate": 8.950876402049606e-06,
      "loss": 0.0595,
      "step": 24950
    },
    {
      "epoch": 2.3794174194335937e-05,
      "model_forward_time": 1.6793162822723389,
      "step": 24950
    },
    {
      "dynamics_loss_avg": 0.06829503700137138,
      "epoch": 2.3794174194335937e-05,
      "step": 24950
    },
    {
      "action_loss_avg": 0.002207429747795686,
      "epoch": 2.3794174194335937e-05,
      "step": 24950
    },
    {
      "epoch": 2.3794174194335937e-05,
      "step": 24950,
      "training_step_time": 8.327033042907715
    },
    {
      "epoch": 2.3795127868652344e-05,
      "model_forward_time": 1.6978075504302979,
      "step": 24951
    },
    {
      "epoch": 2.3795127868652344e-05,
      "step": 24951,
      "training_step_time": 7.672366142272949
    },
    {
      "epoch": 2.379608154296875e-05,
      "model_forward_time": 1.6929876804351807,
      "step": 24952
    },
    {
      "epoch": 2.379608154296875e-05,
      "step": 24952,
      "training_step_time": 7.609163284301758
    },
    {
      "epoch": 2.3797035217285156e-05,
      "model_forward_time": 1.6930515766143799,
      "step": 24953
    },
    {
      "epoch": 2.3797035217285156e-05,
      "step": 24953,
      "training_step_time": 7.628926992416382
    },
    {
      "epoch": 2.3797988891601563e-05,
      "model_forward_time": 1.6949419975280762,
      "step": 24954
    },
    {
      "epoch": 2.3797988891601563e-05,
      "step": 24954,
      "training_step_time": 7.657285690307617
    },
    {
      "epoch": 2.3798942565917967e-05,
      "model_forward_time": 1.6931657791137695,
      "step": 24955
    },
    {
      "epoch": 2.3798942565917967e-05,
      "step": 24955,
      "training_step_time": 7.748739242553711
    },
    {
      "epoch": 2.3799896240234375e-05,
      "model_forward_time": 1.692765712738037,
      "step": 24956
    },
    {
      "epoch": 2.3799896240234375e-05,
      "step": 24956,
      "training_step_time": 8.708566904067993
    },
    {
      "epoch": 2.3800849914550782e-05,
      "model_forward_time": 1.703538417816162,
      "step": 24957
    },
    {
      "epoch": 2.3800849914550782e-05,
      "step": 24957,
      "training_step_time": 8.444819211959839
    },
    {
      "epoch": 2.3801803588867186e-05,
      "model_forward_time": 2.232501745223999,
      "step": 24958
    },
    {
      "epoch": 2.3801803588867186e-05,
      "step": 24958,
      "training_step_time": 8.924172639846802
    },
    {
      "epoch": 2.3802757263183593e-05,
      "model_forward_time": 1.6974902153015137,
      "step": 24959
    },
    {
      "epoch": 2.3802757263183593e-05,
      "step": 24959,
      "training_step_time": 8.381824254989624
    },
    {
      "epoch": 2.38037109375e-05,
      "grad_norm": 0.10860420763492584,
      "learning_rate": 8.949862805928505e-06,
      "loss": 0.0638,
      "step": 24960
    },
    {
      "epoch": 2.38037109375e-05,
      "model_forward_time": 1.6910107135772705,
      "step": 24960
    },
    {
      "dynamics_loss_avg": 0.063235904276371,
      "epoch": 2.38037109375e-05,
      "step": 24960
    },
    {
      "action_loss_avg": 0.0024154479848220944,
      "epoch": 2.38037109375e-05,
      "step": 24960
    },
    {
      "epoch": 2.38037109375e-05,
      "step": 24960,
      "training_step_time": 7.595606565475464
    },
    {
      "epoch": 2.3804664611816405e-05,
      "model_forward_time": 1.6920084953308105,
      "step": 24961
    },
    {
      "epoch": 2.3804664611816405e-05,
      "step": 24961,
      "training_step_time": 8.33023977279663
    },
    {
      "epoch": 2.3805618286132812e-05,
      "model_forward_time": 1.690021276473999,
      "step": 24962
    },
    {
      "epoch": 2.3805618286132812e-05,
      "step": 24962,
      "training_step_time": 8.335569381713867
    },
    {
      "epoch": 2.380657196044922e-05,
      "model_forward_time": 1.6931586265563965,
      "step": 24963
    },
    {
      "epoch": 2.380657196044922e-05,
      "step": 24963,
      "training_step_time": 7.615753412246704
    },
    {
      "epoch": 2.3807525634765624e-05,
      "model_forward_time": 1.6996300220489502,
      "step": 24964
    },
    {
      "epoch": 2.3807525634765624e-05,
      "step": 24964,
      "training_step_time": 8.107768535614014
    },
    {
      "epoch": 2.380847930908203e-05,
      "model_forward_time": 1.6945364475250244,
      "step": 24965
    },
    {
      "epoch": 2.380847930908203e-05,
      "step": 24965,
      "training_step_time": 8.484118700027466
    },
    {
      "epoch": 2.380943298339844e-05,
      "model_forward_time": 1.6940274238586426,
      "step": 24966
    },
    {
      "epoch": 2.380943298339844e-05,
      "step": 24966,
      "training_step_time": 8.69993782043457
    },
    {
      "epoch": 2.3810386657714843e-05,
      "model_forward_time": 2.241682529449463,
      "step": 24967
    },
    {
      "epoch": 2.3810386657714843e-05,
      "step": 24967,
      "training_step_time": 9.26832890510559
    },
    {
      "epoch": 2.381134033203125e-05,
      "model_forward_time": 1.6988589763641357,
      "step": 24968
    },
    {
      "epoch": 2.381134033203125e-05,
      "step": 24968,
      "training_step_time": 8.217148542404175
    },
    {
      "epoch": 2.3812294006347657e-05,
      "model_forward_time": 1.6938786506652832,
      "step": 24969
    },
    {
      "epoch": 2.3812294006347657e-05,
      "step": 24969,
      "training_step_time": 8.492821216583252
    },
    {
      "epoch": 2.381324768066406e-05,
      "grad_norm": 0.2228284329175949,
      "learning_rate": 8.948848777856342e-06,
      "loss": 0.0618,
      "step": 24970
    },
    {
      "epoch": 2.381324768066406e-05,
      "model_forward_time": 1.681190013885498,
      "step": 24970
    },
    {
      "dynamics_loss_avg": 0.059846892952919006,
      "epoch": 2.381324768066406e-05,
      "step": 24970
    },
    {
      "action_loss_avg": 0.00235234581050463,
      "epoch": 2.381324768066406e-05,
      "step": 24970
    },
    {
      "epoch": 2.381324768066406e-05,
      "step": 24970,
      "training_step_time": 8.585161209106445
    },
    {
      "epoch": 2.381420135498047e-05,
      "model_forward_time": 1.7145106792449951,
      "step": 24971
    },
    {
      "epoch": 2.381420135498047e-05,
      "step": 24971,
      "training_step_time": 8.615278005599976
    },
    {
      "epoch": 2.3815155029296876e-05,
      "model_forward_time": 1.6982171535491943,
      "step": 24972
    },
    {
      "epoch": 2.3815155029296876e-05,
      "step": 24972,
      "training_step_time": 7.5479416847229
    },
    {
      "epoch": 2.381610870361328e-05,
      "model_forward_time": 1.317056655883789,
      "step": 24973
    },
    {
      "epoch": 2.381610870361328e-05,
      "step": 24973,
      "training_step_time": 7.348775625228882
    },
    {
      "epoch": 2.3817062377929688e-05,
      "model_forward_time": 1.695091724395752,
      "step": 24974
    },
    {
      "epoch": 2.3817062377929688e-05,
      "step": 24974,
      "training_step_time": 8.76303768157959
    },
    {
      "epoch": 2.3818016052246095e-05,
      "model_forward_time": 1.6914875507354736,
      "step": 24975
    },
    {
      "epoch": 2.3818016052246095e-05,
      "step": 24975,
      "training_step_time": 7.65181303024292
    },
    {
      "epoch": 2.38189697265625e-05,
      "model_forward_time": 2.2120091915130615,
      "step": 24976
    },
    {
      "epoch": 2.38189697265625e-05,
      "step": 24976,
      "training_step_time": 8.30947470664978
    },
    {
      "epoch": 2.3819923400878907e-05,
      "model_forward_time": 1.6899683475494385,
      "step": 24977
    },
    {
      "epoch": 2.3819923400878907e-05,
      "step": 24977,
      "training_step_time": 8.931853294372559
    },
    {
      "epoch": 2.3820877075195314e-05,
      "model_forward_time": 1.6799325942993164,
      "step": 24978
    },
    {
      "epoch": 2.3820877075195314e-05,
      "step": 24978,
      "training_step_time": 7.514007329940796
    },
    {
      "epoch": 2.3821830749511718e-05,
      "model_forward_time": 1.693209171295166,
      "step": 24979
    },
    {
      "epoch": 2.3821830749511718e-05,
      "step": 24979,
      "training_step_time": 7.604986667633057
    },
    {
      "epoch": 2.3822784423828125e-05,
      "grad_norm": 0.17983588576316833,
      "learning_rate": 8.947834317944017e-06,
      "loss": 0.0616,
      "step": 24980
    },
    {
      "epoch": 2.3822784423828125e-05,
      "model_forward_time": 1.6925232410430908,
      "step": 24980
    },
    {
      "dynamics_loss_avg": 0.05784398764371872,
      "epoch": 2.3822784423828125e-05,
      "step": 24980
    },
    {
      "action_loss_avg": 0.002438599761808291,
      "epoch": 2.3822784423828125e-05,
      "step": 24980
    },
    {
      "epoch": 2.3822784423828125e-05,
      "step": 24980,
      "training_step_time": 7.68585205078125
    },
    {
      "epoch": 2.3823738098144533e-05,
      "model_forward_time": 1.6970198154449463,
      "step": 24981
    },
    {
      "epoch": 2.3823738098144533e-05,
      "step": 24981,
      "training_step_time": 8.43367600440979
    },
    {
      "epoch": 2.3824691772460937e-05,
      "model_forward_time": 1.7003319263458252,
      "step": 24982
    },
    {
      "epoch": 2.3824691772460937e-05,
      "step": 24982,
      "training_step_time": 7.598318815231323
    },
    {
      "epoch": 2.3825645446777344e-05,
      "model_forward_time": 1.7041189670562744,
      "step": 24983
    },
    {
      "epoch": 2.3825645446777344e-05,
      "step": 24983,
      "training_step_time": 8.226120471954346
    },
    {
      "epoch": 2.382659912109375e-05,
      "model_forward_time": 1.7047677040100098,
      "step": 24984
    },
    {
      "epoch": 2.382659912109375e-05,
      "step": 24984,
      "training_step_time": 8.203855514526367
    },
    {
      "epoch": 2.3827552795410156e-05,
      "model_forward_time": 1.727919578552246,
      "step": 24985
    },
    {
      "epoch": 2.3827552795410156e-05,
      "step": 24985,
      "training_step_time": 7.578036546707153
    },
    {
      "epoch": 2.3828506469726563e-05,
      "model_forward_time": 2.243140459060669,
      "step": 24986
    },
    {
      "epoch": 2.3828506469726563e-05,
      "step": 24986,
      "training_step_time": 10.258912801742554
    },
    {
      "epoch": 2.3829460144042967e-05,
      "model_forward_time": 1.7030599117279053,
      "step": 24987
    },
    {
      "epoch": 2.3829460144042967e-05,
      "step": 24987,
      "training_step_time": 8.17933177947998
    },
    {
      "epoch": 2.3830413818359375e-05,
      "model_forward_time": 1.6979210376739502,
      "step": 24988
    },
    {
      "epoch": 2.3830413818359375e-05,
      "step": 24988,
      "training_step_time": 7.550271034240723
    },
    {
      "epoch": 2.3831367492675782e-05,
      "model_forward_time": 1.711090326309204,
      "step": 24989
    },
    {
      "epoch": 2.3831367492675782e-05,
      "step": 24989,
      "training_step_time": 7.578195333480835
    },
    {
      "epoch": 2.3832321166992186e-05,
      "grad_norm": 0.12821455299854279,
      "learning_rate": 8.946819426302467e-06,
      "loss": 0.0581,
      "step": 24990
    },
    {
      "epoch": 2.3832321166992186e-05,
      "model_forward_time": 1.6980993747711182,
      "step": 24990
    },
    {
      "dynamics_loss_avg": 0.06240985710173845,
      "epoch": 2.3832321166992186e-05,
      "step": 24990
    },
    {
      "action_loss_avg": 0.0028413286127033643,
      "epoch": 2.3832321166992186e-05,
      "step": 24990
    },
    {
      "epoch": 2.3832321166992186e-05,
      "step": 24990,
      "training_step_time": 7.595221519470215
    },
    {
      "epoch": 2.3833274841308593e-05,
      "model_forward_time": 1.7008919715881348,
      "step": 24991
    },
    {
      "epoch": 2.3833274841308593e-05,
      "step": 24991,
      "training_step_time": 7.564181327819824
    },
    {
      "epoch": 2.3834228515625e-05,
      "model_forward_time": 1.7017500400543213,
      "step": 24992
    },
    {
      "epoch": 2.3834228515625e-05,
      "step": 24992,
      "training_step_time": 8.286923170089722
    },
    {
      "epoch": 2.3835182189941405e-05,
      "model_forward_time": 1.7037835121154785,
      "step": 24993
    },
    {
      "epoch": 2.3835182189941405e-05,
      "step": 24993,
      "training_step_time": 8.678527355194092
    },
    {
      "epoch": 2.3836135864257812e-05,
      "model_forward_time": 1.700699806213379,
      "step": 24994
    },
    {
      "epoch": 2.3836135864257812e-05,
      "step": 24994,
      "training_step_time": 7.797089338302612
    },
    {
      "epoch": 2.383708953857422e-05,
      "model_forward_time": 2.25036358833313,
      "step": 24995
    },
    {
      "epoch": 2.383708953857422e-05,
      "step": 24995,
      "training_step_time": 9.208827257156372
    },
    {
      "epoch": 2.3838043212890624e-05,
      "model_forward_time": 1.7029550075531006,
      "step": 24996
    },
    {
      "epoch": 2.3838043212890624e-05,
      "step": 24996,
      "training_step_time": 8.625735521316528
    },
    {
      "epoch": 2.383899688720703e-05,
      "model_forward_time": 1.7108736038208008,
      "step": 24997
    },
    {
      "epoch": 2.383899688720703e-05,
      "step": 24997,
      "training_step_time": 8.472311019897461
    },
    {
      "epoch": 2.383995056152344e-05,
      "model_forward_time": 1.7163805961608887,
      "step": 24998
    },
    {
      "epoch": 2.383995056152344e-05,
      "step": 24998,
      "training_step_time": 7.787766933441162
    },
    {
      "epoch": 2.3840904235839843e-05,
      "model_forward_time": 1.7224879264831543,
      "step": 24999
    },
    {
      "epoch": 2.3840904235839843e-05,
      "step": 24999,
      "training_step_time": 7.648972034454346
    },
    {
      "epoch": 2.384185791015625e-05,
      "grad_norm": 0.10169017314910889,
      "learning_rate": 8.945804103042677e-06,
      "loss": 0.0566,
      "step": 25000
    },
    {
      "epoch": 2.384185791015625e-05,
      "model_forward_time": 1.7364671230316162,
      "step": 25000
    },
    {
      "dynamics_loss_avg": 0.055879909358918664,
      "epoch": 2.384185791015625e-05,
      "step": 25000
    },
    {
      "action_loss_avg": 0.0018119389045750723,
      "epoch": 2.384185791015625e-05,
      "step": 25000
    },
    {
      "epoch": 2.384185791015625e-05,
      "step": 25000,
      "training_step_time": 6.731069803237915
    },
    {
      "epoch": 2.3842811584472657e-05,
      "model_forward_time": 1.7508513927459717,
      "step": 25001
    },
    {
      "epoch": 2.3842811584472657e-05,
      "step": 25001,
      "training_step_time": 9.533736228942871
    },
    {
      "epoch": 2.384376525878906e-05,
      "model_forward_time": 1.7502028942108154,
      "step": 25002
    },
    {
      "epoch": 2.384376525878906e-05,
      "step": 25002,
      "training_step_time": 10.74450421333313
    },
    {
      "epoch": 2.384471893310547e-05,
      "model_forward_time": 3.288520336151123,
      "step": 25003
    },
    {
      "epoch": 2.384471893310547e-05,
      "step": 25003,
      "training_step_time": 10.781368017196655
    },
    {
      "epoch": 2.3845672607421876e-05,
      "model_forward_time": 1.7333319187164307,
      "step": 25004
    },
    {
      "epoch": 2.3845672607421876e-05,
      "step": 25004,
      "training_step_time": 8.601665735244751
    },
    {
      "epoch": 2.384662628173828e-05,
      "model_forward_time": 1.7419319152832031,
      "step": 25005
    },
    {
      "epoch": 2.384662628173828e-05,
      "step": 25005,
      "training_step_time": 8.604573488235474
    },
    {
      "epoch": 2.3847579956054688e-05,
      "model_forward_time": 1.744983434677124,
      "step": 25006
    },
    {
      "epoch": 2.3847579956054688e-05,
      "step": 25006,
      "training_step_time": 10.066373825073242
    },
    {
      "epoch": 2.3848533630371095e-05,
      "model_forward_time": 1.750725507736206,
      "step": 25007
    },
    {
      "epoch": 2.3848533630371095e-05,
      "step": 25007,
      "training_step_time": 8.26586127281189
    },
    {
      "epoch": 2.38494873046875e-05,
      "model_forward_time": 1.7373566627502441,
      "step": 25008
    },
    {
      "epoch": 2.38494873046875e-05,
      "step": 25008,
      "training_step_time": 8.87293791770935
    },
    {
      "epoch": 2.3850440979003907e-05,
      "model_forward_time": 1.7470743656158447,
      "step": 25009
    },
    {
      "epoch": 2.3850440979003907e-05,
      "step": 25009,
      "training_step_time": 8.14394736289978
    },
    {
      "epoch": 2.3851394653320314e-05,
      "grad_norm": 0.19261299073696136,
      "learning_rate": 8.944788348275682e-06,
      "loss": 0.0611,
      "step": 25010
    },
    {
      "epoch": 2.3851394653320314e-05,
      "model_forward_time": 1.7424442768096924,
      "step": 25010
    },
    {
      "dynamics_loss_avg": 0.05276972837746143,
      "epoch": 2.3851394653320314e-05,
      "step": 25010
    },
    {
      "action_loss_avg": 0.0026117973204236476,
      "epoch": 2.3851394653320314e-05,
      "step": 25010
    },
    {
      "epoch": 2.3851394653320314e-05,
      "step": 25010,
      "training_step_time": 8.927115201950073
    },
    {
      "epoch": 2.3852348327636718e-05,
      "model_forward_time": 1.7513232231140137,
      "step": 25011
    },
    {
      "epoch": 2.3852348327636718e-05,
      "step": 25011,
      "training_step_time": 9.694796085357666
    },
    {
      "epoch": 2.3853302001953125e-05,
      "model_forward_time": 1.7450263500213623,
      "step": 25012
    },
    {
      "epoch": 2.3853302001953125e-05,
      "step": 25012,
      "training_step_time": 7.718665838241577
    },
    {
      "epoch": 2.3854255676269533e-05,
      "model_forward_time": 2.8813483715057373,
      "step": 25013
    },
    {
      "epoch": 2.3854255676269533e-05,
      "step": 25013,
      "training_step_time": 8.67441725730896
    },
    {
      "epoch": 2.3855209350585937e-05,
      "model_forward_time": 1.743518590927124,
      "step": 25014
    },
    {
      "epoch": 2.3855209350585937e-05,
      "step": 25014,
      "training_step_time": 8.686329364776611
    },
    {
      "epoch": 2.3856163024902344e-05,
      "model_forward_time": 1.7440447807312012,
      "step": 25015
    },
    {
      "epoch": 2.3856163024902344e-05,
      "step": 25015,
      "training_step_time": 9.402897834777832
    },
    {
      "epoch": 2.385711669921875e-05,
      "model_forward_time": 1.7601721286773682,
      "step": 25016
    },
    {
      "epoch": 2.385711669921875e-05,
      "step": 25016,
      "training_step_time": 6.838853120803833
    },
    {
      "epoch": 2.3858070373535156e-05,
      "model_forward_time": 1.758815050125122,
      "step": 25017
    },
    {
      "epoch": 2.3858070373535156e-05,
      "step": 25017,
      "training_step_time": 6.812703371047974
    },
    {
      "epoch": 2.3859024047851563e-05,
      "model_forward_time": 1.7641425132751465,
      "step": 25018
    },
    {
      "epoch": 2.3859024047851563e-05,
      "step": 25018,
      "training_step_time": 6.861149311065674
    },
    {
      "epoch": 2.3859977722167967e-05,
      "model_forward_time": 1.7530276775360107,
      "step": 25019
    },
    {
      "epoch": 2.3859977722167967e-05,
      "step": 25019,
      "training_step_time": 6.7863609790802
    },
    {
      "epoch": 2.3860931396484375e-05,
      "grad_norm": 0.15902601182460785,
      "learning_rate": 8.943772162112566e-06,
      "loss": 0.0624,
      "step": 25020
    },
    {
      "epoch": 2.3860931396484375e-05,
      "model_forward_time": 1.7585668563842773,
      "step": 25020
    },
    {
      "dynamics_loss_avg": 0.057367294654250146,
      "epoch": 2.3860931396484375e-05,
      "step": 25020
    },
    {
      "action_loss_avg": 0.0025827394623775036,
      "epoch": 2.3860931396484375e-05,
      "step": 25020
    },
    {
      "epoch": 2.3860931396484375e-05,
      "step": 25020,
      "training_step_time": 6.853980302810669
    },
    {
      "epoch": 2.3861885070800782e-05,
      "model_forward_time": 1.7685585021972656,
      "step": 25021
    },
    {
      "epoch": 2.3861885070800782e-05,
      "step": 25021,
      "training_step_time": 6.789028644561768
    },
    {
      "epoch": 2.3862838745117186e-05,
      "model_forward_time": 1.799978494644165,
      "step": 25022
    },
    {
      "epoch": 2.3862838745117186e-05,
      "step": 25022,
      "training_step_time": 6.8214430809021
    },
    {
      "epoch": 2.3863792419433593e-05,
      "model_forward_time": 1.7708823680877686,
      "step": 25023
    },
    {
      "epoch": 2.3863792419433593e-05,
      "step": 25023,
      "training_step_time": 7.31861138343811
    },
    {
      "epoch": 2.386474609375e-05,
      "model_forward_time": 2.8542423248291016,
      "step": 25024
    },
    {
      "epoch": 2.386474609375e-05,
      "step": 25024,
      "training_step_time": 7.625951766967773
    },
    {
      "epoch": 2.3865699768066405e-05,
      "model_forward_time": 1.7685952186584473,
      "step": 25025
    },
    {
      "epoch": 2.3865699768066405e-05,
      "step": 25025,
      "training_step_time": 8.88448429107666
    },
    {
      "epoch": 2.3866653442382812e-05,
      "model_forward_time": 1.7870206832885742,
      "step": 25026
    },
    {
      "epoch": 2.3866653442382812e-05,
      "step": 25026,
      "training_step_time": 6.768862724304199
    },
    {
      "epoch": 2.386760711669922e-05,
      "model_forward_time": 1.7623445987701416,
      "step": 25027
    },
    {
      "epoch": 2.386760711669922e-05,
      "step": 25027,
      "training_step_time": 6.779263257980347
    },
    {
      "epoch": 2.3868560791015624e-05,
      "model_forward_time": 1.7637600898742676,
      "step": 25028
    },
    {
      "epoch": 2.3868560791015624e-05,
      "step": 25028,
      "training_step_time": 6.7854323387146
    },
    {
      "epoch": 2.386951446533203e-05,
      "model_forward_time": 1.7548470497131348,
      "step": 25029
    },
    {
      "epoch": 2.386951446533203e-05,
      "step": 25029,
      "training_step_time": 6.787319183349609
    },
    {
      "epoch": 2.387046813964844e-05,
      "grad_norm": 0.11836883425712585,
      "learning_rate": 8.942755544664454e-06,
      "loss": 0.0591,
      "step": 25030
    },
    {
      "epoch": 2.387046813964844e-05,
      "model_forward_time": 1.760704755783081,
      "step": 25030
    },
    {
      "dynamics_loss_avg": 0.05542513392865658,
      "epoch": 2.387046813964844e-05,
      "step": 25030
    },
    {
      "action_loss_avg": 0.0023510319762863217,
      "epoch": 2.387046813964844e-05,
      "step": 25030
    },
    {
      "epoch": 2.387046813964844e-05,
      "step": 25030,
      "training_step_time": 6.8380725383758545
    },
    {
      "epoch": 2.3871421813964843e-05,
      "model_forward_time": 1.7626864910125732,
      "step": 25031
    },
    {
      "epoch": 2.3871421813964843e-05,
      "step": 25031,
      "training_step_time": 6.8157103061676025
    },
    {
      "epoch": 2.387237548828125e-05,
      "model_forward_time": 1.7574667930603027,
      "step": 25032
    },
    {
      "epoch": 2.387237548828125e-05,
      "step": 25032,
      "training_step_time": 7.304841041564941
    },
    {
      "epoch": 2.3873329162597657e-05,
      "model_forward_time": 2.997105360031128,
      "step": 25033
    },
    {
      "epoch": 2.3873329162597657e-05,
      "step": 25033,
      "training_step_time": 7.78430700302124
    },
    {
      "epoch": 2.387428283691406e-05,
      "model_forward_time": 1.924443244934082,
      "step": 25034
    },
    {
      "epoch": 2.387428283691406e-05,
      "step": 25034,
      "training_step_time": 7.821080207824707
    },
    {
      "epoch": 2.387523651123047e-05,
      "model_forward_time": 1.9299688339233398,
      "step": 25035
    },
    {
      "epoch": 2.387523651123047e-05,
      "step": 25035,
      "training_step_time": 7.72556471824646
    },
    {
      "epoch": 2.3876190185546876e-05,
      "model_forward_time": 1.9423460960388184,
      "step": 25036
    },
    {
      "epoch": 2.3876190185546876e-05,
      "step": 25036,
      "training_step_time": 6.867740154266357
    },
    {
      "epoch": 2.387714385986328e-05,
      "model_forward_time": 1.9248051643371582,
      "step": 25037
    },
    {
      "epoch": 2.387714385986328e-05,
      "step": 25037,
      "training_step_time": 6.813791751861572
    },
    {
      "epoch": 2.3878097534179688e-05,
      "model_forward_time": 1.5835752487182617,
      "step": 25038
    },
    {
      "epoch": 2.3878097534179688e-05,
      "step": 25038,
      "training_step_time": 6.8252480030059814
    },
    {
      "epoch": 2.3879051208496095e-05,
      "model_forward_time": 1.9116215705871582,
      "step": 25039
    },
    {
      "epoch": 2.3879051208496095e-05,
      "step": 25039,
      "training_step_time": 6.819445610046387
    },
    {
      "epoch": 2.38800048828125e-05,
      "grad_norm": 0.1488543599843979,
      "learning_rate": 8.941738496042525e-06,
      "loss": 0.0636,
      "step": 25040
    },
    {
      "epoch": 2.38800048828125e-05,
      "model_forward_time": 1.9255280494689941,
      "step": 25040
    },
    {
      "dynamics_loss_avg": 0.048961430042982104,
      "epoch": 2.38800048828125e-05,
      "step": 25040
    },
    {
      "action_loss_avg": 0.002273789222817868,
      "epoch": 2.38800048828125e-05,
      "step": 25040
    },
    {
      "epoch": 2.38800048828125e-05,
      "step": 25040,
      "training_step_time": 6.809607028961182
    },
    {
      "epoch": 2.3880958557128907e-05,
      "model_forward_time": 1.90218186378479,
      "step": 25041
    },
    {
      "epoch": 2.3880958557128907e-05,
      "step": 25041,
      "training_step_time": 6.854403018951416
    },
    {
      "epoch": 2.3881912231445314e-05,
      "model_forward_time": 3.0170035362243652,
      "step": 25042
    },
    {
      "epoch": 2.3881912231445314e-05,
      "step": 25042,
      "training_step_time": 7.835042953491211
    },
    {
      "epoch": 2.3882865905761718e-05,
      "model_forward_time": 1.9225378036499023,
      "step": 25043
    },
    {
      "epoch": 2.3882865905761718e-05,
      "step": 25043,
      "training_step_time": 8.466893434524536
    },
    {
      "epoch": 2.3883819580078125e-05,
      "model_forward_time": 1.9277729988098145,
      "step": 25044
    },
    {
      "epoch": 2.3883819580078125e-05,
      "step": 25044,
      "training_step_time": 7.767371892929077
    },
    {
      "epoch": 2.3884773254394533e-05,
      "model_forward_time": 1.9316215515136719,
      "step": 25045
    },
    {
      "epoch": 2.3884773254394533e-05,
      "step": 25045,
      "training_step_time": 6.7972400188446045
    },
    {
      "epoch": 2.3885726928710937e-05,
      "model_forward_time": 1.931962251663208,
      "step": 25046
    },
    {
      "epoch": 2.3885726928710937e-05,
      "step": 25046,
      "training_step_time": 6.809913873672485
    },
    {
      "epoch": 2.3886680603027344e-05,
      "model_forward_time": 1.9156639575958252,
      "step": 25047
    },
    {
      "epoch": 2.3886680603027344e-05,
      "step": 25047,
      "training_step_time": 6.951304912567139
    },
    {
      "epoch": 2.388763427734375e-05,
      "model_forward_time": 1.9268906116485596,
      "step": 25048
    },
    {
      "epoch": 2.388763427734375e-05,
      "step": 25048,
      "training_step_time": 6.819656133651733
    },
    {
      "epoch": 2.3888587951660156e-05,
      "model_forward_time": 1.942573070526123,
      "step": 25049
    },
    {
      "epoch": 2.3888587951660156e-05,
      "step": 25049,
      "training_step_time": 6.808899641036987
    },
    {
      "epoch": 2.3889541625976563e-05,
      "grad_norm": 0.1809457242488861,
      "learning_rate": 8.940721016358e-06,
      "loss": 0.0648,
      "step": 25050
    },
    {
      "epoch": 2.3889541625976563e-05,
      "model_forward_time": 1.9334297180175781,
      "step": 25050
    },
    {
      "dynamics_loss_avg": 0.06887234561145306,
      "epoch": 2.3889541625976563e-05,
      "step": 25050
    },
    {
      "action_loss_avg": 0.0035764163010753693,
      "epoch": 2.3889541625976563e-05,
      "step": 25050
    },
    {
      "epoch": 2.3889541625976563e-05,
      "step": 25050,
      "training_step_time": 6.8434975147247314
    },
    {
      "epoch": 2.3890495300292967e-05,
      "model_forward_time": 3.04225754737854,
      "step": 25051
    },
    {
      "epoch": 2.3890495300292967e-05,
      "step": 25051,
      "training_step_time": 8.561652421951294
    },
    {
      "epoch": 2.3891448974609375e-05,
      "model_forward_time": 1.9291589260101318,
      "step": 25052
    },
    {
      "epoch": 2.3891448974609375e-05,
      "step": 25052,
      "training_step_time": 10.87575387954712
    },
    {
      "epoch": 2.3892402648925782e-05,
      "model_forward_time": 1.919280767440796,
      "step": 25053
    },
    {
      "epoch": 2.3892402648925782e-05,
      "step": 25053,
      "training_step_time": 7.803907632827759
    },
    {
      "epoch": 2.3893356323242186e-05,
      "model_forward_time": 1.9277105331420898,
      "step": 25054
    },
    {
      "epoch": 2.3893356323242186e-05,
      "step": 25054,
      "training_step_time": 6.83030104637146
    },
    {
      "epoch": 2.3894309997558593e-05,
      "model_forward_time": 1.917116403579712,
      "step": 25055
    },
    {
      "epoch": 2.3894309997558593e-05,
      "step": 25055,
      "training_step_time": 6.763044595718384
    },
    {
      "epoch": 2.3895263671875e-05,
      "model_forward_time": 1.9338009357452393,
      "step": 25056
    },
    {
      "epoch": 2.3895263671875e-05,
      "step": 25056,
      "training_step_time": 6.75354528427124
    },
    {
      "epoch": 2.3896217346191405e-05,
      "model_forward_time": 1.9248812198638916,
      "step": 25057
    },
    {
      "epoch": 2.3896217346191405e-05,
      "step": 25057,
      "training_step_time": 6.862866640090942
    },
    {
      "epoch": 2.3897171020507812e-05,
      "model_forward_time": 1.9605865478515625,
      "step": 25058
    },
    {
      "epoch": 2.3897171020507812e-05,
      "step": 25058,
      "training_step_time": 6.847620964050293
    },
    {
      "epoch": 2.389812469482422e-05,
      "model_forward_time": 1.9383723735809326,
      "step": 25059
    },
    {
      "epoch": 2.389812469482422e-05,
      "step": 25059,
      "training_step_time": 6.750992298126221
    },
    {
      "epoch": 2.3899078369140624e-05,
      "grad_norm": 0.16657771170139313,
      "learning_rate": 8.939703105722148e-06,
      "loss": 0.0637,
      "step": 25060
    },
    {
      "epoch": 2.3899078369140624e-05,
      "model_forward_time": 3.051382303237915,
      "step": 25060
    },
    {
      "dynamics_loss_avg": 0.06479019392281771,
      "epoch": 2.3899078369140624e-05,
      "step": 25060
    },
    {
      "action_loss_avg": 0.002240410045487806,
      "epoch": 2.3899078369140624e-05,
      "step": 25060
    },
    {
      "epoch": 2.3899078369140624e-05,
      "step": 25060,
      "training_step_time": 7.817953824996948
    },
    {
      "epoch": 2.390003204345703e-05,
      "model_forward_time": 1.921525478363037,
      "step": 25061
    },
    {
      "epoch": 2.390003204345703e-05,
      "step": 25061,
      "training_step_time": 9.320061683654785
    },
    {
      "epoch": 2.390098571777344e-05,
      "model_forward_time": 1.575774908065796,
      "step": 25062
    },
    {
      "epoch": 2.390098571777344e-05,
      "step": 25062,
      "training_step_time": 7.632107257843018
    },
    {
      "epoch": 2.3901939392089843e-05,
      "model_forward_time": 1.9306910037994385,
      "step": 25063
    },
    {
      "epoch": 2.3901939392089843e-05,
      "step": 25063,
      "training_step_time": 6.804898977279663
    },
    {
      "epoch": 2.390289306640625e-05,
      "model_forward_time": 1.901961326599121,
      "step": 25064
    },
    {
      "epoch": 2.390289306640625e-05,
      "step": 25064,
      "training_step_time": 6.7719995975494385
    },
    {
      "epoch": 2.3903846740722657e-05,
      "model_forward_time": 1.9134693145751953,
      "step": 25065
    },
    {
      "epoch": 2.3903846740722657e-05,
      "step": 25065,
      "training_step_time": 6.8080034255981445
    },
    {
      "epoch": 2.390480041503906e-05,
      "model_forward_time": 1.9244060516357422,
      "step": 25066
    },
    {
      "epoch": 2.390480041503906e-05,
      "step": 25066,
      "training_step_time": 6.789950132369995
    },
    {
      "epoch": 2.390575408935547e-05,
      "model_forward_time": 1.9170410633087158,
      "step": 25067
    },
    {
      "epoch": 2.390575408935547e-05,
      "step": 25067,
      "training_step_time": 6.80185341835022
    },
    {
      "epoch": 2.3906707763671876e-05,
      "model_forward_time": 1.926853895187378,
      "step": 25068
    },
    {
      "epoch": 2.3906707763671876e-05,
      "step": 25068,
      "training_step_time": 6.770783185958862
    },
    {
      "epoch": 2.390766143798828e-05,
      "model_forward_time": 1.933398962020874,
      "step": 25069
    },
    {
      "epoch": 2.390766143798828e-05,
      "step": 25069,
      "training_step_time": 7.726759910583496
    },
    {
      "epoch": 2.3908615112304688e-05,
      "grad_norm": 0.17091652750968933,
      "learning_rate": 8.93868476424629e-06,
      "loss": 0.0647,
      "step": 25070
    },
    {
      "epoch": 2.3908615112304688e-05,
      "model_forward_time": 2.9872982501983643,
      "step": 25070
    },
    {
      "dynamics_loss_avg": 0.04735347032546997,
      "epoch": 2.3908615112304688e-05,
      "step": 25070
    },
    {
      "action_loss_avg": 0.0016801644262159244,
      "epoch": 2.3908615112304688e-05,
      "step": 25070
    },
    {
      "epoch": 2.3908615112304688e-05,
      "step": 25070,
      "training_step_time": 8.316584348678589
    },
    {
      "epoch": 2.3909568786621095e-05,
      "model_forward_time": 1.912308692932129,
      "step": 25071
    },
    {
      "epoch": 2.3909568786621095e-05,
      "step": 25071,
      "training_step_time": 7.866776943206787
    },
    {
      "epoch": 2.39105224609375e-05,
      "model_forward_time": 1.940157175064087,
      "step": 25072
    },
    {
      "epoch": 2.39105224609375e-05,
      "step": 25072,
      "training_step_time": 7.6659510135650635
    },
    {
      "epoch": 2.3911476135253907e-05,
      "model_forward_time": 1.934103012084961,
      "step": 25073
    },
    {
      "epoch": 2.3911476135253907e-05,
      "step": 25073,
      "training_step_time": 6.798874139785767
    },
    {
      "epoch": 2.3912429809570314e-05,
      "model_forward_time": 1.9402837753295898,
      "step": 25074
    },
    {
      "epoch": 2.3912429809570314e-05,
      "step": 25074,
      "training_step_time": 6.82498025894165
    },
    {
      "epoch": 2.3913383483886718e-05,
      "model_forward_time": 1.9248406887054443,
      "step": 25075
    },
    {
      "epoch": 2.3913383483886718e-05,
      "step": 25075,
      "training_step_time": 6.822130441665649
    },
    {
      "epoch": 2.3914337158203125e-05,
      "model_forward_time": 1.943920612335205,
      "step": 25076
    },
    {
      "epoch": 2.3914337158203125e-05,
      "step": 25076,
      "training_step_time": 6.844005107879639
    },
    {
      "epoch": 2.3915290832519533e-05,
      "model_forward_time": 1.9005208015441895,
      "step": 25077
    },
    {
      "epoch": 2.3915290832519533e-05,
      "step": 25077,
      "training_step_time": 6.730596303939819
    },
    {
      "epoch": 2.3916244506835937e-05,
      "model_forward_time": 1.927030086517334,
      "step": 25078
    },
    {
      "epoch": 2.3916244506835937e-05,
      "step": 25078,
      "training_step_time": 7.633147239685059
    },
    {
      "epoch": 2.3917198181152344e-05,
      "model_forward_time": 2.6815693378448486,
      "step": 25079
    },
    {
      "epoch": 2.3917198181152344e-05,
      "step": 25079,
      "training_step_time": 7.780579090118408
    },
    {
      "epoch": 2.391815185546875e-05,
      "grad_norm": 0.14102508127689362,
      "learning_rate": 8.937665992041786e-06,
      "loss": 0.061,
      "step": 25080
    },
    {
      "epoch": 2.391815185546875e-05,
      "model_forward_time": 1.9324345588684082,
      "step": 25080
    },
    {
      "dynamics_loss_avg": 0.06098635317757726,
      "epoch": 2.391815185546875e-05,
      "step": 25080
    },
    {
      "action_loss_avg": 0.0034250265976879745,
      "epoch": 2.391815185546875e-05,
      "step": 25080
    },
    {
      "epoch": 2.391815185546875e-05,
      "step": 25080,
      "training_step_time": 8.801980018615723
    },
    {
      "epoch": 2.3919105529785156e-05,
      "model_forward_time": 1.9308035373687744,
      "step": 25081
    },
    {
      "epoch": 2.3919105529785156e-05,
      "step": 25081,
      "training_step_time": 7.698888778686523
    },
    {
      "epoch": 2.3920059204101563e-05,
      "model_forward_time": 1.9257776737213135,
      "step": 25082
    },
    {
      "epoch": 2.3920059204101563e-05,
      "step": 25082,
      "training_step_time": 6.809703350067139
    },
    {
      "epoch": 2.3921012878417967e-05,
      "model_forward_time": 1.9549837112426758,
      "step": 25083
    },
    {
      "epoch": 2.3921012878417967e-05,
      "step": 25083,
      "training_step_time": 6.805832386016846
    },
    {
      "epoch": 2.3921966552734375e-05,
      "model_forward_time": 1.9375638961791992,
      "step": 25084
    },
    {
      "epoch": 2.3921966552734375e-05,
      "step": 25084,
      "training_step_time": 6.804182529449463
    },
    {
      "epoch": 2.3922920227050782e-05,
      "model_forward_time": 1.9308805465698242,
      "step": 25085
    },
    {
      "epoch": 2.3922920227050782e-05,
      "step": 25085,
      "training_step_time": 6.831911325454712
    },
    {
      "epoch": 2.3923873901367186e-05,
      "model_forward_time": 1.9320662021636963,
      "step": 25086
    },
    {
      "epoch": 2.3923873901367186e-05,
      "step": 25086,
      "training_step_time": 6.815191268920898
    },
    {
      "epoch": 2.3924827575683594e-05,
      "model_forward_time": 1.938488245010376,
      "step": 25087
    },
    {
      "epoch": 2.3924827575683594e-05,
      "step": 25087,
      "training_step_time": 6.7783660888671875
    },
    {
      "epoch": 2.392578125e-05,
      "model_forward_time": 3.065899610519409,
      "step": 25088
    },
    {
      "epoch": 2.392578125e-05,
      "step": 25088,
      "training_step_time": 7.853846549987793
    },
    {
      "epoch": 2.3926734924316405e-05,
      "model_forward_time": 1.9243652820587158,
      "step": 25089
    },
    {
      "epoch": 2.3926734924316405e-05,
      "step": 25089,
      "training_step_time": 9.79729175567627
    },
    {
      "epoch": 2.3927688598632812e-05,
      "grad_norm": 0.12839946150779724,
      "learning_rate": 8.936646789220051e-06,
      "loss": 0.0638,
      "step": 25090
    },
    {
      "epoch": 2.3927688598632812e-05,
      "model_forward_time": 1.9439773559570312,
      "step": 25090
    },
    {
      "dynamics_loss_avg": 0.03942689951509237,
      "epoch": 2.3927688598632812e-05,
      "step": 25090
    },
    {
      "action_loss_avg": 0.0012153763876995073,
      "epoch": 2.3927688598632812e-05,
      "step": 25090
    },
    {
      "epoch": 2.3927688598632812e-05,
      "step": 25090,
      "training_step_time": 7.765718936920166
    },
    {
      "epoch": 2.392864227294922e-05,
      "model_forward_time": 1.9213147163391113,
      "step": 25091
    },
    {
      "epoch": 2.392864227294922e-05,
      "step": 25091,
      "training_step_time": 6.799633741378784
    },
    {
      "epoch": 2.3929595947265624e-05,
      "model_forward_time": 1.9429643154144287,
      "step": 25092
    },
    {
      "epoch": 2.3929595947265624e-05,
      "step": 25092,
      "training_step_time": 6.80203914642334
    },
    {
      "epoch": 2.393054962158203e-05,
      "model_forward_time": 1.9262726306915283,
      "step": 25093
    },
    {
      "epoch": 2.393054962158203e-05,
      "step": 25093,
      "training_step_time": 6.813807725906372
    },
    {
      "epoch": 2.393150329589844e-05,
      "model_forward_time": 1.930734395980835,
      "step": 25094
    },
    {
      "epoch": 2.393150329589844e-05,
      "step": 25094,
      "training_step_time": 6.782961130142212
    },
    {
      "epoch": 2.3932456970214843e-05,
      "model_forward_time": 1.9374098777770996,
      "step": 25095
    },
    {
      "epoch": 2.3932456970214843e-05,
      "step": 25095,
      "training_step_time": 6.727035045623779
    },
    {
      "epoch": 2.393341064453125e-05,
      "model_forward_time": 1.9303977489471436,
      "step": 25096
    },
    {
      "epoch": 2.393341064453125e-05,
      "step": 25096,
      "training_step_time": 6.750105381011963
    },
    {
      "epoch": 2.3934364318847657e-05,
      "model_forward_time": 3.0277116298675537,
      "step": 25097
    },
    {
      "epoch": 2.3934364318847657e-05,
      "step": 25097,
      "training_step_time": 7.790459632873535
    },
    {
      "epoch": 2.393531799316406e-05,
      "model_forward_time": 1.9238486289978027,
      "step": 25098
    },
    {
      "epoch": 2.393531799316406e-05,
      "step": 25098,
      "training_step_time": 9.627575159072876
    },
    {
      "epoch": 2.393627166748047e-05,
      "model_forward_time": 1.9283428192138672,
      "step": 25099
    },
    {
      "epoch": 2.393627166748047e-05,
      "step": 25099,
      "training_step_time": 7.798033714294434
    },
    {
      "epoch": 2.3937225341796876e-05,
      "grad_norm": 0.15510404109954834,
      "learning_rate": 8.93562715589254e-06,
      "loss": 0.0642,
      "step": 25100
    },
    {
      "epoch": 2.3937225341796876e-05,
      "model_forward_time": 1.9325199127197266,
      "step": 25100
    },
    {
      "dynamics_loss_avg": 0.04284580871462822,
      "epoch": 2.3937225341796876e-05,
      "step": 25100
    },
    {
      "action_loss_avg": 0.0019083470106124878,
      "epoch": 2.3937225341796876e-05,
      "step": 25100
    },
    {
      "epoch": 2.3937225341796876e-05,
      "step": 25100,
      "training_step_time": 7.6169257164001465
    },
    {
      "epoch": 2.393817901611328e-05,
      "model_forward_time": 1.9448204040527344,
      "step": 25101
    },
    {
      "epoch": 2.393817901611328e-05,
      "step": 25101,
      "training_step_time": 6.786848068237305
    },
    {
      "epoch": 2.3939132690429688e-05,
      "model_forward_time": 1.9207432270050049,
      "step": 25102
    },
    {
      "epoch": 2.3939132690429688e-05,
      "step": 25102,
      "training_step_time": 6.747115612030029
    },
    {
      "epoch": 2.3940086364746095e-05,
      "model_forward_time": 1.9326930046081543,
      "step": 25103
    },
    {
      "epoch": 2.3940086364746095e-05,
      "step": 25103,
      "training_step_time": 6.758625507354736
    },
    {
      "epoch": 2.39410400390625e-05,
      "model_forward_time": 1.904139518737793,
      "step": 25104
    },
    {
      "epoch": 2.39410400390625e-05,
      "step": 25104,
      "training_step_time": 6.767688035964966
    },
    {
      "epoch": 2.3941993713378907e-05,
      "model_forward_time": 1.922978401184082,
      "step": 25105
    },
    {
      "epoch": 2.3941993713378907e-05,
      "step": 25105,
      "training_step_time": 6.7242162227630615
    },
    {
      "epoch": 2.3942947387695314e-05,
      "model_forward_time": 3.0395774841308594,
      "step": 25106
    },
    {
      "epoch": 2.3942947387695314e-05,
      "step": 25106,
      "training_step_time": 7.744620084762573
    },
    {
      "epoch": 2.3943901062011718e-05,
      "model_forward_time": 1.9347245693206787,
      "step": 25107
    },
    {
      "epoch": 2.3943901062011718e-05,
      "step": 25107,
      "training_step_time": 10.181153059005737
    },
    {
      "epoch": 2.3944854736328126e-05,
      "model_forward_time": 1.9216971397399902,
      "step": 25108
    },
    {
      "epoch": 2.3944854736328126e-05,
      "step": 25108,
      "training_step_time": 8.268854856491089
    },
    {
      "epoch": 2.3945808410644533e-05,
      "model_forward_time": 1.9331161975860596,
      "step": 25109
    },
    {
      "epoch": 2.3945808410644533e-05,
      "step": 25109,
      "training_step_time": 7.609643936157227
    },
    {
      "epoch": 2.3946762084960937e-05,
      "grad_norm": 0.13830432295799255,
      "learning_rate": 8.934607092170763e-06,
      "loss": 0.0629,
      "step": 25110
    },
    {
      "epoch": 2.3946762084960937e-05,
      "model_forward_time": 1.9199421405792236,
      "step": 25110
    },
    {
      "dynamics_loss_avg": 0.06655137725174427,
      "epoch": 2.3946762084960937e-05,
      "step": 25110
    },
    {
      "action_loss_avg": 0.0028109935868997128,
      "epoch": 2.3946762084960937e-05,
      "step": 25110
    },
    {
      "epoch": 2.3946762084960937e-05,
      "step": 25110,
      "training_step_time": 6.732128381729126
    },
    {
      "epoch": 2.3947715759277344e-05,
      "model_forward_time": 1.9077401161193848,
      "step": 25111
    },
    {
      "epoch": 2.3947715759277344e-05,
      "step": 25111,
      "training_step_time": 6.7491724491119385
    },
    {
      "epoch": 2.394866943359375e-05,
      "model_forward_time": 1.9093544483184814,
      "step": 25112
    },
    {
      "epoch": 2.394866943359375e-05,
      "step": 25112,
      "training_step_time": 6.708985328674316
    },
    {
      "epoch": 2.3949623107910156e-05,
      "model_forward_time": 1.9386520385742188,
      "step": 25113
    },
    {
      "epoch": 2.3949623107910156e-05,
      "step": 25113,
      "training_step_time": 6.752128839492798
    },
    {
      "epoch": 2.3950576782226563e-05,
      "model_forward_time": 1.9360015392303467,
      "step": 25114
    },
    {
      "epoch": 2.3950576782226563e-05,
      "step": 25114,
      "training_step_time": 6.726487159729004
    },
    {
      "epoch": 2.3951530456542967e-05,
      "model_forward_time": 1.9363858699798584,
      "step": 25115
    },
    {
      "epoch": 2.3951530456542967e-05,
      "step": 25115,
      "training_step_time": 7.730502605438232
    },
    {
      "epoch": 2.3952484130859375e-05,
      "model_forward_time": 3.01377534866333,
      "step": 25116
    },
    {
      "epoch": 2.3952484130859375e-05,
      "step": 25116,
      "training_step_time": 7.785484790802002
    },
    {
      "epoch": 2.3953437805175782e-05,
      "model_forward_time": 1.9217336177825928,
      "step": 25117
    },
    {
      "epoch": 2.3953437805175782e-05,
      "step": 25117,
      "training_step_time": 7.6726861000061035
    },
    {
      "epoch": 2.3954391479492186e-05,
      "model_forward_time": 1.920433759689331,
      "step": 25118
    },
    {
      "epoch": 2.3954391479492186e-05,
      "step": 25118,
      "training_step_time": 7.673014163970947
    },
    {
      "epoch": 2.3955345153808594e-05,
      "model_forward_time": 1.90006422996521,
      "step": 25119
    },
    {
      "epoch": 2.3955345153808594e-05,
      "step": 25119,
      "training_step_time": 6.744521379470825
    },
    {
      "epoch": 2.3956298828125e-05,
      "grad_norm": 0.13985639810562134,
      "learning_rate": 8.933586598166267e-06,
      "loss": 0.0637,
      "step": 25120
    },
    {
      "epoch": 2.3956298828125e-05,
      "model_forward_time": 1.7712571620941162,
      "step": 25120
    },
    {
      "dynamics_loss_avg": 0.054723909310996535,
      "epoch": 2.3956298828125e-05,
      "step": 25120
    },
    {
      "action_loss_avg": 0.002961195487296209,
      "epoch": 2.3956298828125e-05,
      "step": 25120
    },
    {
      "epoch": 2.3956298828125e-05,
      "step": 25120,
      "training_step_time": 6.769350051879883
    },
    {
      "epoch": 2.3957252502441405e-05,
      "model_forward_time": 1.7674973011016846,
      "step": 25121
    },
    {
      "epoch": 2.3957252502441405e-05,
      "step": 25121,
      "training_step_time": 6.826879501342773
    },
    {
      "epoch": 2.3958206176757812e-05,
      "model_forward_time": 1.7616353034973145,
      "step": 25122
    },
    {
      "epoch": 2.3958206176757812e-05,
      "step": 25122,
      "training_step_time": 6.763803482055664
    },
    {
      "epoch": 2.395915985107422e-05,
      "model_forward_time": 1.7690989971160889,
      "step": 25123
    },
    {
      "epoch": 2.395915985107422e-05,
      "step": 25123,
      "training_step_time": 6.765485048294067
    },
    {
      "epoch": 2.3960113525390624e-05,
      "model_forward_time": 1.8305654525756836,
      "step": 25124
    },
    {
      "epoch": 2.3960113525390624e-05,
      "step": 25124,
      "training_step_time": 7.736645936965942
    },
    {
      "epoch": 2.396106719970703e-05,
      "model_forward_time": 2.937544822692871,
      "step": 25125
    },
    {
      "epoch": 2.396106719970703e-05,
      "step": 25125,
      "training_step_time": 11.44573450088501
    },
    {
      "epoch": 2.396202087402344e-05,
      "model_forward_time": 1.7581899166107178,
      "step": 25126
    },
    {
      "epoch": 2.396202087402344e-05,
      "step": 25126,
      "training_step_time": 9.314586639404297
    },
    {
      "epoch": 2.3962974548339843e-05,
      "model_forward_time": 1.7572619915008545,
      "step": 25127
    },
    {
      "epoch": 2.3962974548339843e-05,
      "step": 25127,
      "training_step_time": 7.659418106079102
    },
    {
      "epoch": 2.396392822265625e-05,
      "model_forward_time": 1.7624154090881348,
      "step": 25128
    },
    {
      "epoch": 2.396392822265625e-05,
      "step": 25128,
      "training_step_time": 6.754666090011597
    },
    {
      "epoch": 2.3964881896972658e-05,
      "model_forward_time": 1.7607274055480957,
      "step": 25129
    },
    {
      "epoch": 2.3964881896972658e-05,
      "step": 25129,
      "training_step_time": 6.7679383754730225
    },
    {
      "epoch": 2.396583557128906e-05,
      "grad_norm": 0.15974688529968262,
      "learning_rate": 8.932565673990656e-06,
      "loss": 0.0621,
      "step": 25130
    },
    {
      "epoch": 2.396583557128906e-05,
      "model_forward_time": 1.769244909286499,
      "step": 25130
    },
    {
      "dynamics_loss_avg": 0.05196238104254007,
      "epoch": 2.396583557128906e-05,
      "step": 25130
    },
    {
      "action_loss_avg": 0.001424538911669515,
      "epoch": 2.396583557128906e-05,
      "step": 25130
    },
    {
      "epoch": 2.396583557128906e-05,
      "step": 25130,
      "training_step_time": 6.756218910217285
    },
    {
      "epoch": 2.396678924560547e-05,
      "model_forward_time": 1.4055418968200684,
      "step": 25131
    },
    {
      "epoch": 2.396678924560547e-05,
      "step": 25131,
      "training_step_time": 6.786448955535889
    },
    {
      "epoch": 2.3967742919921876e-05,
      "model_forward_time": 1.7621824741363525,
      "step": 25132
    },
    {
      "epoch": 2.3967742919921876e-05,
      "step": 25132,
      "training_step_time": 6.724366903305054
    },
    {
      "epoch": 2.396869659423828e-05,
      "model_forward_time": 1.8281700611114502,
      "step": 25133
    },
    {
      "epoch": 2.396869659423828e-05,
      "step": 25133,
      "training_step_time": 8.674919366836548
    },
    {
      "epoch": 2.3969650268554688e-05,
      "model_forward_time": 2.8787343502044678,
      "step": 25134
    },
    {
      "epoch": 2.3969650268554688e-05,
      "step": 25134,
      "training_step_time": 8.301105976104736
    },
    {
      "epoch": 2.3970603942871095e-05,
      "model_forward_time": 1.7556140422821045,
      "step": 25135
    },
    {
      "epoch": 2.3970603942871095e-05,
      "step": 25135,
      "training_step_time": 10.511367797851562
    },
    {
      "epoch": 2.39715576171875e-05,
      "model_forward_time": 1.7552917003631592,
      "step": 25136
    },
    {
      "epoch": 2.39715576171875e-05,
      "step": 25136,
      "training_step_time": 8.82395339012146
    },
    {
      "epoch": 2.3972511291503907e-05,
      "model_forward_time": 1.7590148448944092,
      "step": 25137
    },
    {
      "epoch": 2.3972511291503907e-05,
      "step": 25137,
      "training_step_time": 7.1934731006622314
    },
    {
      "epoch": 2.3973464965820314e-05,
      "model_forward_time": 1.7607054710388184,
      "step": 25138
    },
    {
      "epoch": 2.3973464965820314e-05,
      "step": 25138,
      "training_step_time": 6.708758354187012
    },
    {
      "epoch": 2.3974418640136718e-05,
      "model_forward_time": 1.7608404159545898,
      "step": 25139
    },
    {
      "epoch": 2.3974418640136718e-05,
      "step": 25139,
      "training_step_time": 6.781572103500366
    },
    {
      "epoch": 2.3975372314453126e-05,
      "grad_norm": 0.15696658194065094,
      "learning_rate": 8.931544319755574e-06,
      "loss": 0.0621,
      "step": 25140
    },
    {
      "epoch": 2.3975372314453126e-05,
      "model_forward_time": 1.7702696323394775,
      "step": 25140
    },
    {
      "dynamics_loss_avg": 0.06523876767605544,
      "epoch": 2.3975372314453126e-05,
      "step": 25140
    },
    {
      "action_loss_avg": 0.0023259438457898796,
      "epoch": 2.3975372314453126e-05,
      "step": 25140
    },
    {
      "epoch": 2.3975372314453126e-05,
      "step": 25140,
      "training_step_time": 6.80531644821167
    },
    {
      "epoch": 2.397632598876953e-05,
      "model_forward_time": 1.7655072212219238,
      "step": 25141
    },
    {
      "epoch": 2.397632598876953e-05,
      "step": 25141,
      "training_step_time": 6.715158939361572
    },
    {
      "epoch": 2.3977279663085937e-05,
      "model_forward_time": 1.4306881427764893,
      "step": 25142
    },
    {
      "epoch": 2.3977279663085937e-05,
      "step": 25142,
      "training_step_time": 7.725668668746948
    },
    {
      "epoch": 2.3978233337402344e-05,
      "model_forward_time": 2.8243186473846436,
      "step": 25143
    },
    {
      "epoch": 2.3978233337402344e-05,
      "step": 25143,
      "training_step_time": 8.792963981628418
    },
    {
      "epoch": 2.397918701171875e-05,
      "model_forward_time": 1.7624626159667969,
      "step": 25144
    },
    {
      "epoch": 2.397918701171875e-05,
      "step": 25144,
      "training_step_time": 8.897128582000732
    },
    {
      "epoch": 2.3980140686035156e-05,
      "model_forward_time": 1.7599494457244873,
      "step": 25145
    },
    {
      "epoch": 2.3980140686035156e-05,
      "step": 25145,
      "training_step_time": 7.77698016166687
    },
    {
      "epoch": 2.3981094360351563e-05,
      "model_forward_time": 1.7705957889556885,
      "step": 25146
    },
    {
      "epoch": 2.3981094360351563e-05,
      "step": 25146,
      "training_step_time": 7.740780353546143
    },
    {
      "epoch": 2.3982048034667967e-05,
      "model_forward_time": 1.758516550064087,
      "step": 25147
    },
    {
      "epoch": 2.3982048034667967e-05,
      "step": 25147,
      "training_step_time": 6.81154465675354
    },
    {
      "epoch": 2.3983001708984375e-05,
      "model_forward_time": 1.7685515880584717,
      "step": 25148
    },
    {
      "epoch": 2.3983001708984375e-05,
      "step": 25148,
      "training_step_time": 6.836496353149414
    },
    {
      "epoch": 2.3983955383300782e-05,
      "model_forward_time": 1.7639961242675781,
      "step": 25149
    },
    {
      "epoch": 2.3983955383300782e-05,
      "step": 25149,
      "training_step_time": 6.8520588874816895
    },
    {
      "epoch": 2.3984909057617186e-05,
      "grad_norm": 0.23910941183567047,
      "learning_rate": 8.930522535572718e-06,
      "loss": 0.0626,
      "step": 25150
    },
    {
      "epoch": 2.3984909057617186e-05,
      "model_forward_time": 1.7687268257141113,
      "step": 25150
    },
    {
      "dynamics_loss_avg": 0.0577405882999301,
      "epoch": 2.3984909057617186e-05,
      "step": 25150
    },
    {
      "action_loss_avg": 0.0022630740189924837,
      "epoch": 2.3984909057617186e-05,
      "step": 25150
    },
    {
      "epoch": 2.3984909057617186e-05,
      "step": 25150,
      "training_step_time": 6.848432540893555
    },
    {
      "epoch": 2.3985862731933594e-05,
      "model_forward_time": 1.799408197402954,
      "step": 25151
    },
    {
      "epoch": 2.3985862731933594e-05,
      "step": 25151,
      "training_step_time": 7.781281232833862
    },
    {
      "epoch": 2.398681640625e-05,
      "model_forward_time": 1.7598373889923096,
      "step": 25152
    },
    {
      "epoch": 2.398681640625e-05,
      "step": 25152,
      "training_step_time": 8.775876998901367
    },
    {
      "epoch": 2.3987770080566405e-05,
      "model_forward_time": 2.8891639709472656,
      "step": 25153
    },
    {
      "epoch": 2.3987770080566405e-05,
      "step": 25153,
      "training_step_time": 9.847180843353271
    },
    {
      "epoch": 2.3988723754882812e-05,
      "model_forward_time": 1.7345874309539795,
      "step": 25154
    },
    {
      "epoch": 2.3988723754882812e-05,
      "step": 25154,
      "training_step_time": 8.502328634262085
    },
    {
      "epoch": 2.398967742919922e-05,
      "model_forward_time": 1.736952304840088,
      "step": 25155
    },
    {
      "epoch": 2.398967742919922e-05,
      "step": 25155,
      "training_step_time": 7.673684597015381
    },
    {
      "epoch": 2.3990631103515624e-05,
      "model_forward_time": 1.7369792461395264,
      "step": 25156
    },
    {
      "epoch": 2.3990631103515624e-05,
      "step": 25156,
      "training_step_time": 6.789344787597656
    },
    {
      "epoch": 2.399158477783203e-05,
      "model_forward_time": 1.3657174110412598,
      "step": 25157
    },
    {
      "epoch": 2.399158477783203e-05,
      "step": 25157,
      "training_step_time": 6.775943279266357
    },
    {
      "epoch": 2.399253845214844e-05,
      "model_forward_time": 1.7401752471923828,
      "step": 25158
    },
    {
      "epoch": 2.399253845214844e-05,
      "step": 25158,
      "training_step_time": 6.796553373336792
    },
    {
      "epoch": 2.3993492126464843e-05,
      "model_forward_time": 1.7383184432983398,
      "step": 25159
    },
    {
      "epoch": 2.3993492126464843e-05,
      "step": 25159,
      "training_step_time": 6.772127628326416
    },
    {
      "epoch": 2.399444580078125e-05,
      "grad_norm": 0.20071689784526825,
      "learning_rate": 8.929500321553825e-06,
      "loss": 0.0619,
      "step": 25160
    },
    {
      "epoch": 2.399444580078125e-05,
      "model_forward_time": 1.3755543231964111,
      "step": 25160
    },
    {
      "dynamics_loss_avg": 0.061091473512351516,
      "epoch": 2.399444580078125e-05,
      "step": 25160
    },
    {
      "action_loss_avg": 0.002507037058239803,
      "epoch": 2.399444580078125e-05,
      "step": 25160
    },
    {
      "epoch": 2.399444580078125e-05,
      "step": 25160,
      "training_step_time": 7.7821900844573975
    },
    {
      "epoch": 2.3995399475097658e-05,
      "model_forward_time": 1.7362210750579834,
      "step": 25161
    },
    {
      "epoch": 2.3995399475097658e-05,
      "step": 25161,
      "training_step_time": 8.794918060302734
    },
    {
      "epoch": 2.399635314941406e-05,
      "model_forward_time": 2.8449301719665527,
      "step": 25162
    },
    {
      "epoch": 2.399635314941406e-05,
      "step": 25162,
      "training_step_time": 7.830271482467651
    },
    {
      "epoch": 2.399730682373047e-05,
      "model_forward_time": 1.7365155220031738,
      "step": 25163
    },
    {
      "epoch": 2.399730682373047e-05,
      "step": 25163,
      "training_step_time": 8.76625680923462
    },
    {
      "epoch": 2.3998260498046876e-05,
      "model_forward_time": 1.7343559265136719,
      "step": 25164
    },
    {
      "epoch": 2.3998260498046876e-05,
      "step": 25164,
      "training_step_time": 7.684887409210205
    },
    {
      "epoch": 2.399921417236328e-05,
      "model_forward_time": 1.738370418548584,
      "step": 25165
    },
    {
      "epoch": 2.399921417236328e-05,
      "step": 25165,
      "training_step_time": 6.801763296127319
    },
    {
      "epoch": 2.4000167846679688e-05,
      "model_forward_time": 1.73537015914917,
      "step": 25166
    },
    {
      "epoch": 2.4000167846679688e-05,
      "step": 25166,
      "training_step_time": 6.822248697280884
    },
    {
      "epoch": 2.4001121520996095e-05,
      "model_forward_time": 1.7320189476013184,
      "step": 25167
    },
    {
      "epoch": 2.4001121520996095e-05,
      "step": 25167,
      "training_step_time": 6.845339059829712
    },
    {
      "epoch": 2.40020751953125e-05,
      "model_forward_time": 1.7369136810302734,
      "step": 25168
    },
    {
      "epoch": 2.40020751953125e-05,
      "step": 25168,
      "training_step_time": 6.8327977657318115
    },
    {
      "epoch": 2.4003028869628907e-05,
      "model_forward_time": 1.740894079208374,
      "step": 25169
    },
    {
      "epoch": 2.4003028869628907e-05,
      "step": 25169,
      "training_step_time": 7.801709175109863
    },
    {
      "epoch": 2.4003982543945314e-05,
      "grad_norm": 0.19685450196266174,
      "learning_rate": 8.928477677810687e-06,
      "loss": 0.0661,
      "step": 25170
    },
    {
      "epoch": 2.4003982543945314e-05,
      "model_forward_time": 1.813730239868164,
      "step": 25170
    },
    {
      "dynamics_loss_avg": 0.06028915960341692,
      "epoch": 2.4003982543945314e-05,
      "step": 25170
    },
    {
      "action_loss_avg": 0.0018856156500987708,
      "epoch": 2.4003982543945314e-05,
      "step": 25170
    },
    {
      "epoch": 2.4003982543945314e-05,
      "step": 25170,
      "training_step_time": 8.21162462234497
    },
    {
      "epoch": 2.4004936218261718e-05,
      "model_forward_time": 2.848867654800415,
      "step": 25171
    },
    {
      "epoch": 2.4004936218261718e-05,
      "step": 25171,
      "training_step_time": 9.32284140586853
    },
    {
      "epoch": 2.4005889892578126e-05,
      "model_forward_time": 1.7470908164978027,
      "step": 25172
    },
    {
      "epoch": 2.4005889892578126e-05,
      "step": 25172,
      "training_step_time": 8.26699185371399
    },
    {
      "epoch": 2.400684356689453e-05,
      "model_forward_time": 1.7489197254180908,
      "step": 25173
    },
    {
      "epoch": 2.400684356689453e-05,
      "step": 25173,
      "training_step_time": 8.736616849899292
    },
    {
      "epoch": 2.4007797241210937e-05,
      "model_forward_time": 1.7475597858428955,
      "step": 25174
    },
    {
      "epoch": 2.4007797241210937e-05,
      "step": 25174,
      "training_step_time": 7.203927993774414
    },
    {
      "epoch": 2.4008750915527344e-05,
      "model_forward_time": 1.7544670104980469,
      "step": 25175
    },
    {
      "epoch": 2.4008750915527344e-05,
      "step": 25175,
      "training_step_time": 6.80604887008667
    },
    {
      "epoch": 2.400970458984375e-05,
      "model_forward_time": 1.393815040588379,
      "step": 25176
    },
    {
      "epoch": 2.400970458984375e-05,
      "step": 25176,
      "training_step_time": 6.793318033218384
    },
    {
      "epoch": 2.4010658264160156e-05,
      "model_forward_time": 1.7462289333343506,
      "step": 25177
    },
    {
      "epoch": 2.4010658264160156e-05,
      "step": 25177,
      "training_step_time": 6.802466869354248
    },
    {
      "epoch": 2.4011611938476563e-05,
      "model_forward_time": 1.7448914051055908,
      "step": 25178
    },
    {
      "epoch": 2.4011611938476563e-05,
      "step": 25178,
      "training_step_time": 7.744684219360352
    },
    {
      "epoch": 2.4012565612792967e-05,
      "model_forward_time": 1.8109495639801025,
      "step": 25179
    },
    {
      "epoch": 2.4012565612792967e-05,
      "step": 25179,
      "training_step_time": 7.66005802154541
    },
    {
      "epoch": 2.4013519287109375e-05,
      "grad_norm": 0.199364572763443,
      "learning_rate": 8.927454604455137e-06,
      "loss": 0.0642,
      "step": 25180
    },
    {
      "epoch": 2.4013519287109375e-05,
      "model_forward_time": 1.7366328239440918,
      "step": 25180
    },
    {
      "dynamics_loss_avg": 0.04770181905478239,
      "epoch": 2.4013519287109375e-05,
      "step": 25180
    },
    {
      "action_loss_avg": 0.0026501757267396895,
      "epoch": 2.4013519287109375e-05,
      "step": 25180
    },
    {
      "epoch": 2.4013519287109375e-05,
      "step": 25180,
      "training_step_time": 8.778450012207031
    },
    {
      "epoch": 2.4014472961425782e-05,
      "model_forward_time": 2.535616636276245,
      "step": 25181
    },
    {
      "epoch": 2.4014472961425782e-05,
      "step": 25181,
      "training_step_time": 8.712454795837402
    },
    {
      "epoch": 2.4015426635742186e-05,
      "model_forward_time": 1.7483563423156738,
      "step": 25182
    },
    {
      "epoch": 2.4015426635742186e-05,
      "step": 25182,
      "training_step_time": 7.737666130065918
    },
    {
      "epoch": 2.4016380310058594e-05,
      "model_forward_time": 1.7505526542663574,
      "step": 25183
    },
    {
      "epoch": 2.4016380310058594e-05,
      "step": 25183,
      "training_step_time": 7.649171352386475
    },
    {
      "epoch": 2.4017333984375e-05,
      "model_forward_time": 1.75321626663208,
      "step": 25184
    },
    {
      "epoch": 2.4017333984375e-05,
      "step": 25184,
      "training_step_time": 6.836812734603882
    },
    {
      "epoch": 2.4018287658691405e-05,
      "model_forward_time": 1.7555196285247803,
      "step": 25185
    },
    {
      "epoch": 2.4018287658691405e-05,
      "step": 25185,
      "training_step_time": 6.790071487426758
    },
    {
      "epoch": 2.4019241333007812e-05,
      "model_forward_time": 1.7503561973571777,
      "step": 25186
    },
    {
      "epoch": 2.4019241333007812e-05,
      "step": 25186,
      "training_step_time": 6.799313545227051
    },
    {
      "epoch": 2.402019500732422e-05,
      "model_forward_time": 1.7461981773376465,
      "step": 25187
    },
    {
      "epoch": 2.402019500732422e-05,
      "step": 25187,
      "training_step_time": 7.782623529434204
    },
    {
      "epoch": 2.4021148681640624e-05,
      "model_forward_time": 1.8260345458984375,
      "step": 25188
    },
    {
      "epoch": 2.4021148681640624e-05,
      "step": 25188,
      "training_step_time": 8.283612251281738
    },
    {
      "epoch": 2.402210235595703e-05,
      "model_forward_time": 1.7537755966186523,
      "step": 25189
    },
    {
      "epoch": 2.402210235595703e-05,
      "step": 25189,
      "training_step_time": 8.76885437965393
    },
    {
      "epoch": 2.402305603027344e-05,
      "grad_norm": 0.19472333788871765,
      "learning_rate": 8.926431101599054e-06,
      "loss": 0.0653,
      "step": 25190
    },
    {
      "epoch": 2.402305603027344e-05,
      "model_forward_time": 2.911194324493408,
      "step": 25190
    },
    {
      "dynamics_loss_avg": 0.05137282367795706,
      "epoch": 2.402305603027344e-05,
      "step": 25190
    },
    {
      "action_loss_avg": 0.003869569778908044,
      "epoch": 2.402305603027344e-05,
      "step": 25190
    },
    {
      "epoch": 2.402305603027344e-05,
      "step": 25190,
      "training_step_time": 9.834825277328491
    },
    {
      "epoch": 2.4024009704589843e-05,
      "model_forward_time": 1.7486381530761719,
      "step": 25191
    },
    {
      "epoch": 2.4024009704589843e-05,
      "step": 25191,
      "training_step_time": 8.473320484161377
    },
    {
      "epoch": 2.402496337890625e-05,
      "model_forward_time": 1.7568910121917725,
      "step": 25192
    },
    {
      "epoch": 2.402496337890625e-05,
      "step": 25192,
      "training_step_time": 7.789235353469849
    },
    {
      "epoch": 2.4025917053222658e-05,
      "model_forward_time": 1.7530584335327148,
      "step": 25193
    },
    {
      "epoch": 2.4025917053222658e-05,
      "step": 25193,
      "training_step_time": 7.763733625411987
    },
    {
      "epoch": 2.402687072753906e-05,
      "model_forward_time": 1.745903491973877,
      "step": 25194
    },
    {
      "epoch": 2.402687072753906e-05,
      "step": 25194,
      "training_step_time": 6.919855117797852
    },
    {
      "epoch": 2.402782440185547e-05,
      "model_forward_time": 1.749953269958496,
      "step": 25195
    },
    {
      "epoch": 2.402782440185547e-05,
      "step": 25195,
      "training_step_time": 6.828192949295044
    },
    {
      "epoch": 2.4028778076171876e-05,
      "model_forward_time": 1.7489683628082275,
      "step": 25196
    },
    {
      "epoch": 2.4028778076171876e-05,
      "step": 25196,
      "training_step_time": 7.896473169326782
    },
    {
      "epoch": 2.402973175048828e-05,
      "model_forward_time": 1.794811725616455,
      "step": 25197
    },
    {
      "epoch": 2.402973175048828e-05,
      "step": 25197,
      "training_step_time": 9.3137526512146
    },
    {
      "epoch": 2.4030685424804688e-05,
      "model_forward_time": 1.74711275100708,
      "step": 25198
    },
    {
      "epoch": 2.4030685424804688e-05,
      "step": 25198,
      "training_step_time": 8.916454792022705
    },
    {
      "epoch": 2.4031639099121095e-05,
      "model_forward_time": 2.8789846897125244,
      "step": 25199
    },
    {
      "epoch": 2.4031639099121095e-05,
      "step": 25199,
      "training_step_time": 8.378811836242676
    },
    {
      "epoch": 2.40325927734375e-05,
      "grad_norm": 0.24238227307796478,
      "learning_rate": 8.92540716935437e-06,
      "loss": 0.0635,
      "step": 25200
    },
    {
      "epoch": 2.40325927734375e-05,
      "model_forward_time": 1.7642381191253662,
      "step": 25200
    },
    {
      "dynamics_loss_avg": 0.06216162256896496,
      "epoch": 2.40325927734375e-05,
      "step": 25200
    },
    {
      "action_loss_avg": 0.0018561062577646225,
      "epoch": 2.40325927734375e-05,
      "step": 25200
    },
    {
      "epoch": 2.40325927734375e-05,
      "step": 25200,
      "training_step_time": 8.460206270217896
    },
    {
      "epoch": 2.4033546447753907e-05,
      "model_forward_time": 1.7494268417358398,
      "step": 25201
    },
    {
      "epoch": 2.4033546447753907e-05,
      "step": 25201,
      "training_step_time": 7.590523958206177
    },
    {
      "epoch": 2.4034500122070314e-05,
      "model_forward_time": 1.7520318031311035,
      "step": 25202
    },
    {
      "epoch": 2.4034500122070314e-05,
      "step": 25202,
      "training_step_time": 7.5497143268585205
    },
    {
      "epoch": 2.4035453796386718e-05,
      "model_forward_time": 1.740795373916626,
      "step": 25203
    },
    {
      "epoch": 2.4035453796386718e-05,
      "step": 25203,
      "training_step_time": 6.788533449172974
    },
    {
      "epoch": 2.4036407470703126e-05,
      "model_forward_time": 1.7476420402526855,
      "step": 25204
    },
    {
      "epoch": 2.4036407470703126e-05,
      "step": 25204,
      "training_step_time": 6.749904632568359
    },
    {
      "epoch": 2.403736114501953e-05,
      "model_forward_time": 1.7496895790100098,
      "step": 25205
    },
    {
      "epoch": 2.403736114501953e-05,
      "step": 25205,
      "training_step_time": 7.763633728027344
    },
    {
      "epoch": 2.4038314819335937e-05,
      "model_forward_time": 1.7719769477844238,
      "step": 25206
    },
    {
      "epoch": 2.4038314819335937e-05,
      "step": 25206,
      "training_step_time": 8.258026361465454
    },
    {
      "epoch": 2.4039268493652344e-05,
      "model_forward_time": 1.7272677421569824,
      "step": 25207
    },
    {
      "epoch": 2.4039268493652344e-05,
      "step": 25207,
      "training_step_time": 7.838925123214722
    },
    {
      "epoch": 2.404022216796875e-05,
      "model_forward_time": 2.8528316020965576,
      "step": 25208
    },
    {
      "epoch": 2.404022216796875e-05,
      "step": 25208,
      "training_step_time": 7.668859004974365
    },
    {
      "epoch": 2.4041175842285156e-05,
      "model_forward_time": 1.3736529350280762,
      "step": 25209
    },
    {
      "epoch": 2.4041175842285156e-05,
      "step": 25209,
      "training_step_time": 8.386336088180542
    },
    {
      "epoch": 2.4042129516601563e-05,
      "grad_norm": 0.18829011917114258,
      "learning_rate": 8.92438280783306e-06,
      "loss": 0.062,
      "step": 25210
    },
    {
      "epoch": 2.4042129516601563e-05,
      "model_forward_time": 1.7390055656433105,
      "step": 25210
    },
    {
      "dynamics_loss_avg": 0.05732662901282311,
      "epoch": 2.4042129516601563e-05,
      "step": 25210
    },
    {
      "action_loss_avg": 0.00320809866534546,
      "epoch": 2.4042129516601563e-05,
      "step": 25210
    },
    {
      "epoch": 2.4042129516601563e-05,
      "step": 25210,
      "training_step_time": 8.290273189544678
    },
    {
      "epoch": 2.4043083190917967e-05,
      "model_forward_time": 1.7492105960845947,
      "step": 25211
    },
    {
      "epoch": 2.4043083190917967e-05,
      "step": 25211,
      "training_step_time": 7.618561267852783
    },
    {
      "epoch": 2.4044036865234375e-05,
      "model_forward_time": 1.7418081760406494,
      "step": 25212
    },
    {
      "epoch": 2.4044036865234375e-05,
      "step": 25212,
      "training_step_time": 6.761817932128906
    },
    {
      "epoch": 2.4044990539550782e-05,
      "model_forward_time": 1.727773666381836,
      "step": 25213
    },
    {
      "epoch": 2.4044990539550782e-05,
      "step": 25213,
      "training_step_time": 6.763864994049072
    },
    {
      "epoch": 2.4045944213867186e-05,
      "model_forward_time": 1.740753173828125,
      "step": 25214
    },
    {
      "epoch": 2.4045944213867186e-05,
      "step": 25214,
      "training_step_time": 7.763964891433716
    },
    {
      "epoch": 2.4046897888183594e-05,
      "model_forward_time": 1.3739054203033447,
      "step": 25215
    },
    {
      "epoch": 2.4046897888183594e-05,
      "step": 25215,
      "training_step_time": 7.650660991668701
    },
    {
      "epoch": 2.40478515625e-05,
      "model_forward_time": 1.7613279819488525,
      "step": 25216
    },
    {
      "epoch": 2.40478515625e-05,
      "step": 25216,
      "training_step_time": 7.814492225646973
    },
    {
      "epoch": 2.4048805236816405e-05,
      "model_forward_time": 1.7689447402954102,
      "step": 25217
    },
    {
      "epoch": 2.4048805236816405e-05,
      "step": 25217,
      "training_step_time": 9.713917255401611
    },
    {
      "epoch": 2.4049758911132812e-05,
      "model_forward_time": 2.9225869178771973,
      "step": 25218
    },
    {
      "epoch": 2.4049758911132812e-05,
      "step": 25218,
      "training_step_time": 9.734268426895142
    },
    {
      "epoch": 2.405071258544922e-05,
      "model_forward_time": 1.74143648147583,
      "step": 25219
    },
    {
      "epoch": 2.405071258544922e-05,
      "step": 25219,
      "training_step_time": 8.704200267791748
    },
    {
      "epoch": 2.4051666259765624e-05,
      "grad_norm": 0.22623729705810547,
      "learning_rate": 8.923358017147145e-06,
      "loss": 0.0606,
      "step": 25220
    },
    {
      "epoch": 2.4051666259765624e-05,
      "model_forward_time": 1.7438547611236572,
      "step": 25220
    },
    {
      "dynamics_loss_avg": 0.06024393737316132,
      "epoch": 2.4051666259765624e-05,
      "step": 25220
    },
    {
      "action_loss_avg": 0.002296121558174491,
      "epoch": 2.4051666259765624e-05,
      "step": 25220
    },
    {
      "epoch": 2.4051666259765624e-05,
      "step": 25220,
      "training_step_time": 7.637938737869263
    },
    {
      "epoch": 2.405261993408203e-05,
      "model_forward_time": 1.7439076900482178,
      "step": 25221
    },
    {
      "epoch": 2.405261993408203e-05,
      "step": 25221,
      "training_step_time": 6.7806077003479
    },
    {
      "epoch": 2.405357360839844e-05,
      "model_forward_time": 1.7436118125915527,
      "step": 25222
    },
    {
      "epoch": 2.405357360839844e-05,
      "step": 25222,
      "training_step_time": 6.768861293792725
    },
    {
      "epoch": 2.4054527282714843e-05,
      "model_forward_time": 1.730855941772461,
      "step": 25223
    },
    {
      "epoch": 2.4054527282714843e-05,
      "step": 25223,
      "training_step_time": 7.448377847671509
    },
    {
      "epoch": 2.405548095703125e-05,
      "model_forward_time": 1.7370409965515137,
      "step": 25224
    },
    {
      "epoch": 2.405548095703125e-05,
      "step": 25224,
      "training_step_time": 7.873049974441528
    },
    {
      "epoch": 2.4056434631347658e-05,
      "model_forward_time": 1.385221004486084,
      "step": 25225
    },
    {
      "epoch": 2.4056434631347658e-05,
      "step": 25225,
      "training_step_time": 8.24212121963501
    },
    {
      "epoch": 2.405738830566406e-05,
      "model_forward_time": 1.7589116096496582,
      "step": 25226
    },
    {
      "epoch": 2.405738830566406e-05,
      "step": 25226,
      "training_step_time": 9.492296695709229
    },
    {
      "epoch": 2.405834197998047e-05,
      "model_forward_time": 2.494948625564575,
      "step": 25227
    },
    {
      "epoch": 2.405834197998047e-05,
      "step": 25227,
      "training_step_time": 8.299731016159058
    },
    {
      "epoch": 2.4059295654296876e-05,
      "model_forward_time": 1.7418851852416992,
      "step": 25228
    },
    {
      "epoch": 2.4059295654296876e-05,
      "step": 25228,
      "training_step_time": 7.6516289710998535
    },
    {
      "epoch": 2.406024932861328e-05,
      "model_forward_time": 1.745884656906128,
      "step": 25229
    },
    {
      "epoch": 2.406024932861328e-05,
      "step": 25229,
      "training_step_time": 7.594357967376709
    },
    {
      "epoch": 2.4061203002929688e-05,
      "grad_norm": 0.17409200966358185,
      "learning_rate": 8.922332797408697e-06,
      "loss": 0.0594,
      "step": 25230
    },
    {
      "epoch": 2.4061203002929688e-05,
      "model_forward_time": 1.7424240112304688,
      "step": 25230
    },
    {
      "dynamics_loss_avg": 0.06170909889042377,
      "epoch": 2.4061203002929688e-05,
      "step": 25230
    },
    {
      "action_loss_avg": 0.002296969317831099,
      "epoch": 2.4061203002929688e-05,
      "step": 25230
    },
    {
      "epoch": 2.4061203002929688e-05,
      "step": 25230,
      "training_step_time": 6.815784931182861
    },
    {
      "epoch": 2.4062156677246095e-05,
      "model_forward_time": 1.7330958843231201,
      "step": 25231
    },
    {
      "epoch": 2.4062156677246095e-05,
      "step": 25231,
      "training_step_time": 6.75365424156189
    },
    {
      "epoch": 2.40631103515625e-05,
      "model_forward_time": 1.7350444793701172,
      "step": 25232
    },
    {
      "epoch": 2.40631103515625e-05,
      "step": 25232,
      "training_step_time": 7.409522771835327
    },
    {
      "epoch": 2.4064064025878907e-05,
      "model_forward_time": 1.7366750240325928,
      "step": 25233
    },
    {
      "epoch": 2.4064064025878907e-05,
      "step": 25233,
      "training_step_time": 8.797069549560547
    },
    {
      "epoch": 2.4065017700195314e-05,
      "model_forward_time": 1.7497012615203857,
      "step": 25234
    },
    {
      "epoch": 2.4065017700195314e-05,
      "step": 25234,
      "training_step_time": 9.682978868484497
    },
    {
      "epoch": 2.4065971374511718e-05,
      "model_forward_time": 1.7449166774749756,
      "step": 25235
    },
    {
      "epoch": 2.4065971374511718e-05,
      "step": 25235,
      "training_step_time": 9.020527124404907
    },
    {
      "epoch": 2.4066925048828126e-05,
      "model_forward_time": 1.7314653396606445,
      "step": 25236
    },
    {
      "epoch": 2.4066925048828126e-05,
      "step": 25236,
      "training_step_time": 8.259608745574951
    },
    {
      "epoch": 2.406787872314453e-05,
      "model_forward_time": 1.7405040264129639,
      "step": 25237
    },
    {
      "epoch": 2.406787872314453e-05,
      "step": 25237,
      "training_step_time": 8.514297485351562
    },
    {
      "epoch": 2.4068832397460937e-05,
      "model_forward_time": 1.7366054058074951,
      "step": 25238
    },
    {
      "epoch": 2.4068832397460937e-05,
      "step": 25238,
      "training_step_time": 7.788606405258179
    },
    {
      "epoch": 2.4069786071777344e-05,
      "model_forward_time": 1.7347724437713623,
      "step": 25239
    },
    {
      "epoch": 2.4069786071777344e-05,
      "step": 25239,
      "training_step_time": 7.67905330657959
    },
    {
      "epoch": 2.407073974609375e-05,
      "grad_norm": 0.15712223947048187,
      "learning_rate": 8.921307148729832e-06,
      "loss": 0.061,
      "step": 25240
    },
    {
      "epoch": 2.407073974609375e-05,
      "model_forward_time": 1.7291512489318848,
      "step": 25240
    },
    {
      "dynamics_loss_avg": 0.05612546131014824,
      "epoch": 2.407073974609375e-05,
      "step": 25240
    },
    {
      "action_loss_avg": 0.002618194546084851,
      "epoch": 2.407073974609375e-05,
      "step": 25240
    },
    {
      "epoch": 2.407073974609375e-05,
      "step": 25240,
      "training_step_time": 6.813523054122925
    },
    {
      "epoch": 2.4071693420410156e-05,
      "model_forward_time": 1.7513980865478516,
      "step": 25241
    },
    {
      "epoch": 2.4071693420410156e-05,
      "step": 25241,
      "training_step_time": 7.410946607589722
    },
    {
      "epoch": 2.4072647094726563e-05,
      "model_forward_time": 1.7465107440948486,
      "step": 25242
    },
    {
      "epoch": 2.4072647094726563e-05,
      "step": 25242,
      "training_step_time": 7.718729734420776
    },
    {
      "epoch": 2.4073600769042967e-05,
      "model_forward_time": 1.752856969833374,
      "step": 25243
    },
    {
      "epoch": 2.4073600769042967e-05,
      "step": 25243,
      "training_step_time": 7.741361618041992
    },
    {
      "epoch": 2.4074554443359375e-05,
      "model_forward_time": 1.7444202899932861,
      "step": 25244
    },
    {
      "epoch": 2.4074554443359375e-05,
      "step": 25244,
      "training_step_time": 7.899554252624512
    },
    {
      "epoch": 2.4075508117675782e-05,
      "model_forward_time": 1.7750732898712158,
      "step": 25245
    },
    {
      "epoch": 2.4075508117675782e-05,
      "step": 25245,
      "training_step_time": 9.32302451133728
    },
    {
      "epoch": 2.4076461791992186e-05,
      "model_forward_time": 1.7334377765655518,
      "step": 25246
    },
    {
      "epoch": 2.4076461791992186e-05,
      "step": 25246,
      "training_step_time": 10.232509136199951
    },
    {
      "epoch": 2.4077415466308594e-05,
      "model_forward_time": 1.7367796897888184,
      "step": 25247
    },
    {
      "epoch": 2.4077415466308594e-05,
      "step": 25247,
      "training_step_time": 8.305335998535156
    },
    {
      "epoch": 2.4078369140625e-05,
      "model_forward_time": 1.3777754306793213,
      "step": 25248
    },
    {
      "epoch": 2.4078369140625e-05,
      "step": 25248,
      "training_step_time": 7.669081687927246
    },
    {
      "epoch": 2.4079322814941405e-05,
      "model_forward_time": 1.7433297634124756,
      "step": 25249
    },
    {
      "epoch": 2.4079322814941405e-05,
      "step": 25249,
      "training_step_time": 6.764788866043091
    },
    {
      "epoch": 2.4080276489257812e-05,
      "grad_norm": 0.12271972000598907,
      "learning_rate": 8.920281071222712e-06,
      "loss": 0.0628,
      "step": 25250
    },
    {
      "epoch": 2.4080276489257812e-05,
      "model_forward_time": 1.7339844703674316,
      "step": 25250
    },
    {
      "dynamics_loss_avg": 0.060151030868291856,
      "epoch": 2.4080276489257812e-05,
      "step": 25250
    },
    {
      "action_loss_avg": 0.002695204905467108,
      "epoch": 2.4080276489257812e-05,
      "step": 25250
    },
    {
      "epoch": 2.4080276489257812e-05,
      "step": 25250,
      "training_step_time": 7.388625860214233
    },
    {
      "epoch": 2.408123016357422e-05,
      "model_forward_time": 1.7358558177947998,
      "step": 25251
    },
    {
      "epoch": 2.408123016357422e-05,
      "step": 25251,
      "training_step_time": 8.174371242523193
    },
    {
      "epoch": 2.4082183837890624e-05,
      "model_forward_time": 1.7366299629211426,
      "step": 25252
    },
    {
      "epoch": 2.4082183837890624e-05,
      "step": 25252,
      "training_step_time": 8.208513021469116
    },
    {
      "epoch": 2.408313751220703e-05,
      "model_forward_time": 1.7390656471252441,
      "step": 25253
    },
    {
      "epoch": 2.408313751220703e-05,
      "step": 25253,
      "training_step_time": 9.093658924102783
    },
    {
      "epoch": 2.408409118652344e-05,
      "model_forward_time": 1.7708048820495605,
      "step": 25254
    },
    {
      "epoch": 2.408409118652344e-05,
      "step": 25254,
      "training_step_time": 9.484877586364746
    },
    {
      "epoch": 2.4085044860839843e-05,
      "model_forward_time": 2.926137685775757,
      "step": 25255
    },
    {
      "epoch": 2.4085044860839843e-05,
      "step": 25255,
      "training_step_time": 9.558240413665771
    },
    {
      "epoch": 2.408599853515625e-05,
      "model_forward_time": 1.7404217720031738,
      "step": 25256
    },
    {
      "epoch": 2.408599853515625e-05,
      "step": 25256,
      "training_step_time": 7.7472920417785645
    },
    {
      "epoch": 2.4086952209472658e-05,
      "model_forward_time": 1.7322709560394287,
      "step": 25257
    },
    {
      "epoch": 2.4086952209472658e-05,
      "step": 25257,
      "training_step_time": 7.729722499847412
    },
    {
      "epoch": 2.408790588378906e-05,
      "model_forward_time": 1.7338337898254395,
      "step": 25258
    },
    {
      "epoch": 2.408790588378906e-05,
      "step": 25258,
      "training_step_time": 6.810077428817749
    },
    {
      "epoch": 2.408885955810547e-05,
      "model_forward_time": 1.366934061050415,
      "step": 25259
    },
    {
      "epoch": 2.408885955810547e-05,
      "step": 25259,
      "training_step_time": 7.496757984161377
    },
    {
      "epoch": 2.4089813232421876e-05,
      "grad_norm": 0.17465627193450928,
      "learning_rate": 8.919254564999548e-06,
      "loss": 0.0638,
      "step": 25260
    },
    {
      "epoch": 2.4089813232421876e-05,
      "model_forward_time": 1.733152151107788,
      "step": 25260
    },
    {
      "dynamics_loss_avg": 0.051049523428082465,
      "epoch": 2.4089813232421876e-05,
      "step": 25260
    },
    {
      "action_loss_avg": 0.0019039824663195759,
      "epoch": 2.4089813232421876e-05,
      "step": 25260
    },
    {
      "epoch": 2.4089813232421876e-05,
      "step": 25260,
      "training_step_time": 8.699995994567871
    },
    {
      "epoch": 2.409076690673828e-05,
      "model_forward_time": 1.7478210926055908,
      "step": 25261
    },
    {
      "epoch": 2.409076690673828e-05,
      "step": 25261,
      "training_step_time": 7.698570489883423
    },
    {
      "epoch": 2.4091720581054688e-05,
      "model_forward_time": 1.7530109882354736,
      "step": 25262
    },
    {
      "epoch": 2.4091720581054688e-05,
      "step": 25262,
      "training_step_time": 8.464476823806763
    },
    {
      "epoch": 2.4092674255371095e-05,
      "model_forward_time": 1.74772047996521,
      "step": 25263
    },
    {
      "epoch": 2.4092674255371095e-05,
      "step": 25263,
      "training_step_time": 8.321561813354492
    },
    {
      "epoch": 2.40936279296875e-05,
      "model_forward_time": 2.936467409133911,
      "step": 25264
    },
    {
      "epoch": 2.40936279296875e-05,
      "step": 25264,
      "training_step_time": 8.85969614982605
    },
    {
      "epoch": 2.4094581604003907e-05,
      "model_forward_time": 1.7337620258331299,
      "step": 25265
    },
    {
      "epoch": 2.4094581604003907e-05,
      "step": 25265,
      "training_step_time": 7.663123846054077
    },
    {
      "epoch": 2.409553527832031e-05,
      "model_forward_time": 1.7527337074279785,
      "step": 25266
    },
    {
      "epoch": 2.409553527832031e-05,
      "step": 25266,
      "training_step_time": 7.772760391235352
    },
    {
      "epoch": 2.4096488952636718e-05,
      "model_forward_time": 1.7382214069366455,
      "step": 25267
    },
    {
      "epoch": 2.4096488952636718e-05,
      "step": 25267,
      "training_step_time": 7.203884840011597
    },
    {
      "epoch": 2.4097442626953126e-05,
      "model_forward_time": 1.7413392066955566,
      "step": 25268
    },
    {
      "epoch": 2.4097442626953126e-05,
      "step": 25268,
      "training_step_time": 7.457093715667725
    },
    {
      "epoch": 2.409839630126953e-05,
      "model_forward_time": 1.7381298542022705,
      "step": 25269
    },
    {
      "epoch": 2.409839630126953e-05,
      "step": 25269,
      "training_step_time": 7.760919570922852
    },
    {
      "epoch": 2.4099349975585937e-05,
      "grad_norm": 0.18838652968406677,
      "learning_rate": 8.918227630172598e-06,
      "loss": 0.068,
      "step": 25270
    },
    {
      "epoch": 2.4099349975585937e-05,
      "model_forward_time": 1.748913288116455,
      "step": 25270
    },
    {
      "dynamics_loss_avg": 0.07105429824441671,
      "epoch": 2.4099349975585937e-05,
      "step": 25270
    },
    {
      "action_loss_avg": 0.002953406982123852,
      "epoch": 2.4099349975585937e-05,
      "step": 25270
    },
    {
      "epoch": 2.4099349975585937e-05,
      "step": 25270,
      "training_step_time": 7.602341175079346
    },
    {
      "epoch": 2.4100303649902345e-05,
      "model_forward_time": 1.7540631294250488,
      "step": 25271
    },
    {
      "epoch": 2.4100303649902345e-05,
      "step": 25271,
      "training_step_time": 7.605438947677612
    },
    {
      "epoch": 2.410125732421875e-05,
      "model_forward_time": 1.7578997611999512,
      "step": 25272
    },
    {
      "epoch": 2.410125732421875e-05,
      "step": 25272,
      "training_step_time": 8.446106195449829
    },
    {
      "epoch": 2.4102210998535156e-05,
      "model_forward_time": 2.9103848934173584,
      "step": 25273
    },
    {
      "epoch": 2.4102210998535156e-05,
      "step": 25273,
      "training_step_time": 9.431370496749878
    },
    {
      "epoch": 2.4103164672851563e-05,
      "model_forward_time": 1.741682529449463,
      "step": 25274
    },
    {
      "epoch": 2.4103164672851563e-05,
      "step": 25274,
      "training_step_time": 8.651223182678223
    },
    {
      "epoch": 2.4104118347167967e-05,
      "model_forward_time": 1.7357189655303955,
      "step": 25275
    },
    {
      "epoch": 2.4104118347167967e-05,
      "step": 25275,
      "training_step_time": 7.827970027923584
    },
    {
      "epoch": 2.4105072021484375e-05,
      "model_forward_time": 1.7366998195648193,
      "step": 25276
    },
    {
      "epoch": 2.4105072021484375e-05,
      "step": 25276,
      "training_step_time": 7.530047655105591
    },
    {
      "epoch": 2.4106025695800782e-05,
      "model_forward_time": 1.7536213397979736,
      "step": 25277
    },
    {
      "epoch": 2.4106025695800782e-05,
      "step": 25277,
      "training_step_time": 7.3809590339660645
    },
    {
      "epoch": 2.4106979370117186e-05,
      "model_forward_time": 1.7342119216918945,
      "step": 25278
    },
    {
      "epoch": 2.4106979370117186e-05,
      "step": 25278,
      "training_step_time": 7.728459119796753
    },
    {
      "epoch": 2.4107933044433594e-05,
      "model_forward_time": 1.742189884185791,
      "step": 25279
    },
    {
      "epoch": 2.4107933044433594e-05,
      "step": 25279,
      "training_step_time": 7.350478887557983
    },
    {
      "epoch": 2.410888671875e-05,
      "grad_norm": 0.21353356540203094,
      "learning_rate": 8.917200266854164e-06,
      "loss": 0.0659,
      "step": 25280
    },
    {
      "epoch": 2.410888671875e-05,
      "model_forward_time": 1.7598052024841309,
      "step": 25280
    },
    {
      "dynamics_loss_avg": 0.04949018079787493,
      "epoch": 2.410888671875e-05,
      "step": 25280
    },
    {
      "action_loss_avg": 0.0023459034506231546,
      "epoch": 2.410888671875e-05,
      "step": 25280
    },
    {
      "epoch": 2.410888671875e-05,
      "step": 25280,
      "training_step_time": 7.691632986068726
    },
    {
      "epoch": 2.4109840393066405e-05,
      "model_forward_time": 1.746086597442627,
      "step": 25281
    },
    {
      "epoch": 2.4109840393066405e-05,
      "step": 25281,
      "training_step_time": 8.25657033920288
    },
    {
      "epoch": 2.4110794067382813e-05,
      "model_forward_time": 2.8773961067199707,
      "step": 25282
    },
    {
      "epoch": 2.4110794067382813e-05,
      "step": 25282,
      "training_step_time": 8.81735897064209
    },
    {
      "epoch": 2.411174774169922e-05,
      "model_forward_time": 1.7383553981781006,
      "step": 25283
    },
    {
      "epoch": 2.411174774169922e-05,
      "step": 25283,
      "training_step_time": 8.67291259765625
    },
    {
      "epoch": 2.4112701416015624e-05,
      "model_forward_time": 1.7475152015686035,
      "step": 25284
    },
    {
      "epoch": 2.4112701416015624e-05,
      "step": 25284,
      "training_step_time": 8.354950428009033
    },
    {
      "epoch": 2.411365509033203e-05,
      "model_forward_time": 1.7313249111175537,
      "step": 25285
    },
    {
      "epoch": 2.411365509033203e-05,
      "step": 25285,
      "training_step_time": 7.700973033905029
    },
    {
      "epoch": 2.411460876464844e-05,
      "model_forward_time": 1.7386448383331299,
      "step": 25286
    },
    {
      "epoch": 2.411460876464844e-05,
      "step": 25286,
      "training_step_time": 7.530148506164551
    },
    {
      "epoch": 2.4115562438964843e-05,
      "model_forward_time": 1.746715784072876,
      "step": 25287
    },
    {
      "epoch": 2.4115562438964843e-05,
      "step": 25287,
      "training_step_time": 8.56873869895935
    },
    {
      "epoch": 2.411651611328125e-05,
      "model_forward_time": 1.734792947769165,
      "step": 25288
    },
    {
      "epoch": 2.411651611328125e-05,
      "step": 25288,
      "training_step_time": 7.536374568939209
    },
    {
      "epoch": 2.4117469787597658e-05,
      "model_forward_time": 1.7471542358398438,
      "step": 25289
    },
    {
      "epoch": 2.4117469787597658e-05,
      "step": 25289,
      "training_step_time": 8.310779333114624
    },
    {
      "epoch": 2.411842346191406e-05,
      "grad_norm": 0.17847789824008942,
      "learning_rate": 8.9161724751566e-06,
      "loss": 0.0641,
      "step": 25290
    },
    {
      "epoch": 2.411842346191406e-05,
      "model_forward_time": 1.737628698348999,
      "step": 25290
    },
    {
      "dynamics_loss_avg": 0.07270762510597706,
      "epoch": 2.411842346191406e-05,
      "step": 25290
    },
    {
      "action_loss_avg": 0.0036264186492189765,
      "epoch": 2.411842346191406e-05,
      "step": 25290
    },
    {
      "epoch": 2.411842346191406e-05,
      "step": 25290,
      "training_step_time": 8.990638732910156
    },
    {
      "epoch": 2.411937713623047e-05,
      "model_forward_time": 2.8706793785095215,
      "step": 25291
    },
    {
      "epoch": 2.411937713623047e-05,
      "step": 25291,
      "training_step_time": 7.7995383739471436
    },
    {
      "epoch": 2.4120330810546877e-05,
      "model_forward_time": 1.7360131740570068,
      "step": 25292
    },
    {
      "epoch": 2.4120330810546877e-05,
      "step": 25292,
      "training_step_time": 8.368125677108765
    },
    {
      "epoch": 2.412128448486328e-05,
      "model_forward_time": 1.7334115505218506,
      "step": 25293
    },
    {
      "epoch": 2.412128448486328e-05,
      "step": 25293,
      "training_step_time": 8.738250494003296
    },
    {
      "epoch": 2.4122238159179688e-05,
      "model_forward_time": 1.736696720123291,
      "step": 25294
    },
    {
      "epoch": 2.4122238159179688e-05,
      "step": 25294,
      "training_step_time": 7.6808366775512695
    },
    {
      "epoch": 2.4123191833496095e-05,
      "model_forward_time": 1.7401776313781738,
      "step": 25295
    },
    {
      "epoch": 2.4123191833496095e-05,
      "step": 25295,
      "training_step_time": 7.57792592048645
    },
    {
      "epoch": 2.41241455078125e-05,
      "model_forward_time": 1.7082905769348145,
      "step": 25296
    },
    {
      "epoch": 2.41241455078125e-05,
      "step": 25296,
      "training_step_time": 7.746271848678589
    },
    {
      "epoch": 2.4125099182128907e-05,
      "model_forward_time": 1.7337498664855957,
      "step": 25297
    },
    {
      "epoch": 2.4125099182128907e-05,
      "step": 25297,
      "training_step_time": 7.382275342941284
    },
    {
      "epoch": 2.412605285644531e-05,
      "model_forward_time": 1.7286977767944336,
      "step": 25298
    },
    {
      "epoch": 2.412605285644531e-05,
      "step": 25298,
      "training_step_time": 8.173554420471191
    },
    {
      "epoch": 2.4127006530761718e-05,
      "model_forward_time": 1.7496490478515625,
      "step": 25299
    },
    {
      "epoch": 2.4127006530761718e-05,
      "step": 25299,
      "training_step_time": 7.869067907333374
    },
    {
      "epoch": 2.4127960205078126e-05,
      "grad_norm": 0.18770359456539154,
      "learning_rate": 8.915144255192302e-06,
      "loss": 0.0657,
      "step": 25300
    },
    {
      "epoch": 2.4127960205078126e-05,
      "model_forward_time": 1.7526395320892334,
      "step": 25300
    },
    {
      "dynamics_loss_avg": 0.0597396019846201,
      "epoch": 2.4127960205078126e-05,
      "step": 25300
    },
    {
      "action_loss_avg": 0.0028054329042788593,
      "epoch": 2.4127960205078126e-05,
      "step": 25300
    },
    {
      "epoch": 2.4127960205078126e-05,
      "step": 25300,
      "training_step_time": 8.145228862762451
    },
    {
      "epoch": 2.412891387939453e-05,
      "model_forward_time": 1.7237179279327393,
      "step": 25301
    },
    {
      "epoch": 2.412891387939453e-05,
      "step": 25301,
      "training_step_time": 7.675225734710693
    },
    {
      "epoch": 2.4129867553710937e-05,
      "model_forward_time": 1.7239313125610352,
      "step": 25302
    },
    {
      "epoch": 2.4129867553710937e-05,
      "step": 25302,
      "training_step_time": 7.629887580871582
    },
    {
      "epoch": 2.4130821228027345e-05,
      "model_forward_time": 1.737309217453003,
      "step": 25303
    },
    {
      "epoch": 2.4130821228027345e-05,
      "step": 25303,
      "training_step_time": 7.613196611404419
    },
    {
      "epoch": 2.413177490234375e-05,
      "model_forward_time": 1.7172763347625732,
      "step": 25304
    },
    {
      "epoch": 2.413177490234375e-05,
      "step": 25304,
      "training_step_time": 7.560218811035156
    },
    {
      "epoch": 2.4132728576660156e-05,
      "model_forward_time": 1.7213561534881592,
      "step": 25305
    },
    {
      "epoch": 2.4132728576660156e-05,
      "step": 25305,
      "training_step_time": 7.661184787750244
    },
    {
      "epoch": 2.4133682250976563e-05,
      "model_forward_time": 1.7205166816711426,
      "step": 25306
    },
    {
      "epoch": 2.4133682250976563e-05,
      "step": 25306,
      "training_step_time": 7.386188745498657
    },
    {
      "epoch": 2.4134635925292967e-05,
      "model_forward_time": 1.737710952758789,
      "step": 25307
    },
    {
      "epoch": 2.4134635925292967e-05,
      "step": 25307,
      "training_step_time": 7.625895261764526
    },
    {
      "epoch": 2.4135589599609375e-05,
      "model_forward_time": 1.3787636756896973,
      "step": 25308
    },
    {
      "epoch": 2.4135589599609375e-05,
      "step": 25308,
      "training_step_time": 8.660191059112549
    },
    {
      "epoch": 2.4136543273925782e-05,
      "model_forward_time": 1.7450520992279053,
      "step": 25309
    },
    {
      "epoch": 2.4136543273925782e-05,
      "step": 25309,
      "training_step_time": 10.354136943817139
    },
    {
      "epoch": 2.4137496948242186e-05,
      "grad_norm": 0.17934207618236542,
      "learning_rate": 8.914115607073713e-06,
      "loss": 0.0632,
      "step": 25310
    },
    {
      "epoch": 2.4137496948242186e-05,
      "model_forward_time": 2.9275128841400146,
      "step": 25310
    },
    {
      "dynamics_loss_avg": 0.05227291192859411,
      "epoch": 2.4137496948242186e-05,
      "step": 25310
    },
    {
      "action_loss_avg": 0.003023110725916922,
      "epoch": 2.4137496948242186e-05,
      "step": 25310
    },
    {
      "epoch": 2.4137496948242186e-05,
      "step": 25310,
      "training_step_time": 9.206888675689697
    },
    {
      "epoch": 2.4138450622558594e-05,
      "model_forward_time": 1.7253012657165527,
      "step": 25311
    },
    {
      "epoch": 2.4138450622558594e-05,
      "step": 25311,
      "training_step_time": 8.609055042266846
    },
    {
      "epoch": 2.4139404296875e-05,
      "model_forward_time": 1.7415766716003418,
      "step": 25312
    },
    {
      "epoch": 2.4139404296875e-05,
      "step": 25312,
      "training_step_time": 7.736117124557495
    },
    {
      "epoch": 2.4140357971191405e-05,
      "model_forward_time": 1.7336313724517822,
      "step": 25313
    },
    {
      "epoch": 2.4140357971191405e-05,
      "step": 25313,
      "training_step_time": 7.5440545082092285
    },
    {
      "epoch": 2.4141311645507813e-05,
      "model_forward_time": 1.7234406471252441,
      "step": 25314
    },
    {
      "epoch": 2.4141311645507813e-05,
      "step": 25314,
      "training_step_time": 7.699260234832764
    },
    {
      "epoch": 2.414226531982422e-05,
      "model_forward_time": 1.7380034923553467,
      "step": 25315
    },
    {
      "epoch": 2.414226531982422e-05,
      "step": 25315,
      "training_step_time": 8.00849962234497
    },
    {
      "epoch": 2.4143218994140624e-05,
      "model_forward_time": 1.7231719493865967,
      "step": 25316
    },
    {
      "epoch": 2.4143218994140624e-05,
      "step": 25316,
      "training_step_time": 8.561383962631226
    },
    {
      "epoch": 2.414417266845703e-05,
      "model_forward_time": 1.7376716136932373,
      "step": 25317
    },
    {
      "epoch": 2.414417266845703e-05,
      "step": 25317,
      "training_step_time": 8.56766128540039
    },
    {
      "epoch": 2.414512634277344e-05,
      "model_forward_time": 1.7530789375305176,
      "step": 25318
    },
    {
      "epoch": 2.414512634277344e-05,
      "step": 25318,
      "training_step_time": 8.401707887649536
    },
    {
      "epoch": 2.4146080017089843e-05,
      "model_forward_time": 2.8737270832061768,
      "step": 25319
    },
    {
      "epoch": 2.4146080017089843e-05,
      "step": 25319,
      "training_step_time": 7.607556104660034
    },
    {
      "epoch": 2.414703369140625e-05,
      "grad_norm": 0.1392635554075241,
      "learning_rate": 8.913086530913327e-06,
      "loss": 0.0666,
      "step": 25320
    },
    {
      "epoch": 2.414703369140625e-05,
      "model_forward_time": 1.7288415431976318,
      "step": 25320
    },
    {
      "dynamics_loss_avg": 0.07892647869884968,
      "epoch": 2.414703369140625e-05,
      "step": 25320
    },
    {
      "action_loss_avg": 0.003209521932876669,
      "epoch": 2.414703369140625e-05,
      "step": 25320
    },
    {
      "epoch": 2.414703369140625e-05,
      "step": 25320,
      "training_step_time": 7.637590169906616
    },
    {
      "epoch": 2.4147987365722658e-05,
      "model_forward_time": 1.7308008670806885,
      "step": 25321
    },
    {
      "epoch": 2.4147987365722658e-05,
      "step": 25321,
      "training_step_time": 7.644014835357666
    },
    {
      "epoch": 2.4148941040039062e-05,
      "model_forward_time": 1.7256481647491455,
      "step": 25322
    },
    {
      "epoch": 2.4148941040039062e-05,
      "step": 25322,
      "training_step_time": 7.657408237457275
    },
    {
      "epoch": 2.414989471435547e-05,
      "model_forward_time": 1.714533805847168,
      "step": 25323
    },
    {
      "epoch": 2.414989471435547e-05,
      "step": 25323,
      "training_step_time": 7.733616352081299
    },
    {
      "epoch": 2.4150848388671877e-05,
      "model_forward_time": 1.7256062030792236,
      "step": 25324
    },
    {
      "epoch": 2.4150848388671877e-05,
      "step": 25324,
      "training_step_time": 7.404191255569458
    },
    {
      "epoch": 2.415180206298828e-05,
      "model_forward_time": 1.35829758644104,
      "step": 25325
    },
    {
      "epoch": 2.415180206298828e-05,
      "step": 25325,
      "training_step_time": 7.6140077114105225
    },
    {
      "epoch": 2.4152755737304688e-05,
      "model_forward_time": 1.735328197479248,
      "step": 25326
    },
    {
      "epoch": 2.4152755737304688e-05,
      "step": 25326,
      "training_step_time": 8.667253017425537
    },
    {
      "epoch": 2.4153709411621095e-05,
      "model_forward_time": 1.7315385341644287,
      "step": 25327
    },
    {
      "epoch": 2.4153709411621095e-05,
      "step": 25327,
      "training_step_time": 8.31583023071289
    },
    {
      "epoch": 2.41546630859375e-05,
      "model_forward_time": 2.8773984909057617,
      "step": 25328
    },
    {
      "epoch": 2.41546630859375e-05,
      "step": 25328,
      "training_step_time": 9.997339487075806
    },
    {
      "epoch": 2.4155616760253907e-05,
      "model_forward_time": 1.7121882438659668,
      "step": 25329
    },
    {
      "epoch": 2.4155616760253907e-05,
      "step": 25329,
      "training_step_time": 8.444154262542725
    },
    {
      "epoch": 2.415657043457031e-05,
      "grad_norm": 0.17914196848869324,
      "learning_rate": 8.912057026823681e-06,
      "loss": 0.0646,
      "step": 25330
    },
    {
      "epoch": 2.415657043457031e-05,
      "model_forward_time": 1.7271790504455566,
      "step": 25330
    },
    {
      "dynamics_loss_avg": 0.05637156385928392,
      "epoch": 2.415657043457031e-05,
      "step": 25330
    },
    {
      "action_loss_avg": 0.001796679839026183,
      "epoch": 2.415657043457031e-05,
      "step": 25330
    },
    {
      "epoch": 2.415657043457031e-05,
      "step": 25330,
      "training_step_time": 7.591446161270142
    },
    {
      "epoch": 2.4157524108886718e-05,
      "model_forward_time": 1.7182016372680664,
      "step": 25331
    },
    {
      "epoch": 2.4157524108886718e-05,
      "step": 25331,
      "training_step_time": 7.688563108444214
    },
    {
      "epoch": 2.4158477783203126e-05,
      "model_forward_time": 1.7286877632141113,
      "step": 25332
    },
    {
      "epoch": 2.4158477783203126e-05,
      "step": 25332,
      "training_step_time": 7.76851487159729
    },
    {
      "epoch": 2.415943145751953e-05,
      "model_forward_time": 1.7204177379608154,
      "step": 25333
    },
    {
      "epoch": 2.415943145751953e-05,
      "step": 25333,
      "training_step_time": 7.34788179397583
    },
    {
      "epoch": 2.4160385131835937e-05,
      "model_forward_time": 1.7269482612609863,
      "step": 25334
    },
    {
      "epoch": 2.4160385131835937e-05,
      "step": 25334,
      "training_step_time": 7.5698158740997314
    },
    {
      "epoch": 2.4161338806152345e-05,
      "model_forward_time": 1.7303426265716553,
      "step": 25335
    },
    {
      "epoch": 2.4161338806152345e-05,
      "step": 25335,
      "training_step_time": 9.564866304397583
    },
    {
      "epoch": 2.416229248046875e-05,
      "model_forward_time": 1.7364978790283203,
      "step": 25336
    },
    {
      "epoch": 2.416229248046875e-05,
      "step": 25336,
      "training_step_time": 8.558071613311768
    },
    {
      "epoch": 2.4163246154785156e-05,
      "model_forward_time": 2.792288303375244,
      "step": 25337
    },
    {
      "epoch": 2.4163246154785156e-05,
      "step": 25337,
      "training_step_time": 9.593571901321411
    },
    {
      "epoch": 2.4164199829101563e-05,
      "model_forward_time": 1.729369878768921,
      "step": 25338
    },
    {
      "epoch": 2.4164199829101563e-05,
      "step": 25338,
      "training_step_time": 9.391898393630981
    },
    {
      "epoch": 2.4165153503417967e-05,
      "model_forward_time": 1.7269701957702637,
      "step": 25339
    },
    {
      "epoch": 2.4165153503417967e-05,
      "step": 25339,
      "training_step_time": 9.261900186538696
    },
    {
      "epoch": 2.4166107177734375e-05,
      "grad_norm": 0.17812348902225494,
      "learning_rate": 8.91102709491736e-06,
      "loss": 0.0617,
      "step": 25340
    },
    {
      "epoch": 2.4166107177734375e-05,
      "model_forward_time": 1.7131164073944092,
      "step": 25340
    },
    {
      "dynamics_loss_avg": 0.05331863341853023,
      "epoch": 2.4166107177734375e-05,
      "step": 25340
    },
    {
      "action_loss_avg": 0.0022759640531148763,
      "epoch": 2.4166107177734375e-05,
      "step": 25340
    },
    {
      "epoch": 2.4166107177734375e-05,
      "step": 25340,
      "training_step_time": 7.690785884857178
    },
    {
      "epoch": 2.4167060852050782e-05,
      "model_forward_time": 1.729266881942749,
      "step": 25341
    },
    {
      "epoch": 2.4167060852050782e-05,
      "step": 25341,
      "training_step_time": 8.035630941390991
    },
    {
      "epoch": 2.4168014526367186e-05,
      "model_forward_time": 1.728498935699463,
      "step": 25342
    },
    {
      "epoch": 2.4168014526367186e-05,
      "step": 25342,
      "training_step_time": 7.244460582733154
    },
    {
      "epoch": 2.4168968200683594e-05,
      "model_forward_time": 1.7234821319580078,
      "step": 25343
    },
    {
      "epoch": 2.4168968200683594e-05,
      "step": 25343,
      "training_step_time": 7.332627534866333
    },
    {
      "epoch": 2.4169921875e-05,
      "model_forward_time": 1.7455885410308838,
      "step": 25344
    },
    {
      "epoch": 2.4169921875e-05,
      "step": 25344,
      "training_step_time": 8.18002462387085
    },
    {
      "epoch": 2.4170875549316405e-05,
      "model_forward_time": 1.7268757820129395,
      "step": 25345
    },
    {
      "epoch": 2.4170875549316405e-05,
      "step": 25345,
      "training_step_time": 9.192735195159912
    },
    {
      "epoch": 2.4171829223632813e-05,
      "model_forward_time": 1.3718478679656982,
      "step": 25346
    },
    {
      "epoch": 2.4171829223632813e-05,
      "step": 25346,
      "training_step_time": 9.065969944000244
    },
    {
      "epoch": 2.417278289794922e-05,
      "model_forward_time": 2.900108814239502,
      "step": 25347
    },
    {
      "epoch": 2.417278289794922e-05,
      "step": 25347,
      "training_step_time": 9.17483639717102
    },
    {
      "epoch": 2.4173736572265624e-05,
      "model_forward_time": 1.722308874130249,
      "step": 25348
    },
    {
      "epoch": 2.4173736572265624e-05,
      "step": 25348,
      "training_step_time": 7.505572080612183
    },
    {
      "epoch": 2.417469024658203e-05,
      "model_forward_time": 1.7242472171783447,
      "step": 25349
    },
    {
      "epoch": 2.417469024658203e-05,
      "step": 25349,
      "training_step_time": 7.669291734695435
    },
    {
      "epoch": 2.417564392089844e-05,
      "grad_norm": 0.15492931008338928,
      "learning_rate": 8.909996735306996e-06,
      "loss": 0.0608,
      "step": 25350
    },
    {
      "epoch": 2.417564392089844e-05,
      "model_forward_time": 1.7109715938568115,
      "step": 25350
    },
    {
      "dynamics_loss_avg": 0.07478687101975083,
      "epoch": 2.417564392089844e-05,
      "step": 25350
    },
    {
      "action_loss_avg": 0.004081892664544284,
      "epoch": 2.417564392089844e-05,
      "step": 25350
    },
    {
      "epoch": 2.417564392089844e-05,
      "step": 25350,
      "training_step_time": 7.523393392562866
    },
    {
      "epoch": 2.4176597595214843e-05,
      "model_forward_time": 1.7215619087219238,
      "step": 25351
    },
    {
      "epoch": 2.4176597595214843e-05,
      "step": 25351,
      "training_step_time": 7.618694067001343
    },
    {
      "epoch": 2.417755126953125e-05,
      "model_forward_time": 1.357560396194458,
      "step": 25352
    },
    {
      "epoch": 2.417755126953125e-05,
      "step": 25352,
      "training_step_time": 7.507106304168701
    },
    {
      "epoch": 2.4178504943847658e-05,
      "model_forward_time": 1.743852138519287,
      "step": 25353
    },
    {
      "epoch": 2.4178504943847658e-05,
      "step": 25353,
      "training_step_time": 8.316333293914795
    },
    {
      "epoch": 2.4179458618164062e-05,
      "model_forward_time": 1.7402055263519287,
      "step": 25354
    },
    {
      "epoch": 2.4179458618164062e-05,
      "step": 25354,
      "training_step_time": 8.668707132339478
    },
    {
      "epoch": 2.418041229248047e-05,
      "model_forward_time": 1.757678508758545,
      "step": 25355
    },
    {
      "epoch": 2.418041229248047e-05,
      "step": 25355,
      "training_step_time": 7.633066654205322
    },
    {
      "epoch": 2.4181365966796877e-05,
      "model_forward_time": 2.8657386302948,
      "step": 25356
    },
    {
      "epoch": 2.4181365966796877e-05,
      "step": 25356,
      "training_step_time": 7.656792402267456
    },
    {
      "epoch": 2.418231964111328e-05,
      "model_forward_time": 1.724862813949585,
      "step": 25357
    },
    {
      "epoch": 2.418231964111328e-05,
      "step": 25357,
      "training_step_time": 8.145476341247559
    },
    {
      "epoch": 2.4183273315429688e-05,
      "model_forward_time": 1.716341495513916,
      "step": 25358
    },
    {
      "epoch": 2.4183273315429688e-05,
      "step": 25358,
      "training_step_time": 7.598587274551392
    },
    {
      "epoch": 2.4184226989746095e-05,
      "model_forward_time": 1.7271668910980225,
      "step": 25359
    },
    {
      "epoch": 2.4184226989746095e-05,
      "step": 25359,
      "training_step_time": 7.502945184707642
    },
    {
      "epoch": 2.41851806640625e-05,
      "grad_norm": 0.1504306048154831,
      "learning_rate": 8.908965948105269e-06,
      "loss": 0.0644,
      "step": 25360
    },
    {
      "epoch": 2.41851806640625e-05,
      "model_forward_time": 1.7121036052703857,
      "step": 25360
    },
    {
      "dynamics_loss_avg": 0.05636303862556815,
      "epoch": 2.41851806640625e-05,
      "step": 25360
    },
    {
      "action_loss_avg": 0.0023131982277845966,
      "epoch": 2.41851806640625e-05,
      "step": 25360
    },
    {
      "epoch": 2.41851806640625e-05,
      "step": 25360,
      "training_step_time": 7.609297513961792
    },
    {
      "epoch": 2.4186134338378907e-05,
      "model_forward_time": 1.7249836921691895,
      "step": 25361
    },
    {
      "epoch": 2.4186134338378907e-05,
      "step": 25361,
      "training_step_time": 7.331126689910889
    },
    {
      "epoch": 2.418708801269531e-05,
      "model_forward_time": 1.3727679252624512,
      "step": 25362
    },
    {
      "epoch": 2.418708801269531e-05,
      "step": 25362,
      "training_step_time": 7.550939083099365
    },
    {
      "epoch": 2.418804168701172e-05,
      "model_forward_time": 1.7138941287994385,
      "step": 25363
    },
    {
      "epoch": 2.418804168701172e-05,
      "step": 25363,
      "training_step_time": 7.673813819885254
    },
    {
      "epoch": 2.4188995361328126e-05,
      "model_forward_time": 1.755136251449585,
      "step": 25364
    },
    {
      "epoch": 2.4188995361328126e-05,
      "step": 25364,
      "training_step_time": 7.5576558113098145
    },
    {
      "epoch": 2.418994903564453e-05,
      "model_forward_time": 2.7889344692230225,
      "step": 25365
    },
    {
      "epoch": 2.418994903564453e-05,
      "step": 25365,
      "training_step_time": 8.494024991989136
    },
    {
      "epoch": 2.4190902709960937e-05,
      "model_forward_time": 1.7310447692871094,
      "step": 25366
    },
    {
      "epoch": 2.4190902709960937e-05,
      "step": 25366,
      "training_step_time": 8.094774007797241
    },
    {
      "epoch": 2.4191856384277345e-05,
      "model_forward_time": 1.7259998321533203,
      "step": 25367
    },
    {
      "epoch": 2.4191856384277345e-05,
      "step": 25367,
      "training_step_time": 8.192234516143799
    },
    {
      "epoch": 2.419281005859375e-05,
      "model_forward_time": 1.720552682876587,
      "step": 25368
    },
    {
      "epoch": 2.419281005859375e-05,
      "step": 25368,
      "training_step_time": 7.484704971313477
    },
    {
      "epoch": 2.4193763732910156e-05,
      "model_forward_time": 1.716733694076538,
      "step": 25369
    },
    {
      "epoch": 2.4193763732910156e-05,
      "step": 25369,
      "training_step_time": 7.736843585968018
    },
    {
      "epoch": 2.4194717407226563e-05,
      "grad_norm": 0.14581656455993652,
      "learning_rate": 8.907934733424902e-06,
      "loss": 0.0581,
      "step": 25370
    },
    {
      "epoch": 2.4194717407226563e-05,
      "model_forward_time": 1.7266099452972412,
      "step": 25370
    },
    {
      "dynamics_loss_avg": 0.06270832903683185,
      "epoch": 2.4194717407226563e-05,
      "step": 25370
    },
    {
      "action_loss_avg": 0.003194930308382027,
      "epoch": 2.4194717407226563e-05,
      "step": 25370
    },
    {
      "epoch": 2.4194717407226563e-05,
      "step": 25370,
      "training_step_time": 9.291248798370361
    },
    {
      "epoch": 2.4195671081542967e-05,
      "model_forward_time": 1.7599372863769531,
      "step": 25371
    },
    {
      "epoch": 2.4195671081542967e-05,
      "step": 25371,
      "training_step_time": 7.54259467124939
    },
    {
      "epoch": 2.4196624755859375e-05,
      "model_forward_time": 1.7129185199737549,
      "step": 25372
    },
    {
      "epoch": 2.4196624755859375e-05,
      "step": 25372,
      "training_step_time": 8.3125638961792
    },
    {
      "epoch": 2.4197578430175782e-05,
      "model_forward_time": 1.7506015300750732,
      "step": 25373
    },
    {
      "epoch": 2.4197578430175782e-05,
      "step": 25373,
      "training_step_time": 8.873296737670898
    },
    {
      "epoch": 2.4198532104492186e-05,
      "model_forward_time": 1.7154364585876465,
      "step": 25374
    },
    {
      "epoch": 2.4198532104492186e-05,
      "step": 25374,
      "training_step_time": 8.44171142578125
    },
    {
      "epoch": 2.4199485778808594e-05,
      "model_forward_time": 2.9265666007995605,
      "step": 25375
    },
    {
      "epoch": 2.4199485778808594e-05,
      "step": 25375,
      "training_step_time": 8.4446382522583
    },
    {
      "epoch": 2.4200439453125e-05,
      "model_forward_time": 1.712702751159668,
      "step": 25376
    },
    {
      "epoch": 2.4200439453125e-05,
      "step": 25376,
      "training_step_time": 7.543659925460815
    },
    {
      "epoch": 2.4201393127441405e-05,
      "model_forward_time": 1.717238187789917,
      "step": 25377
    },
    {
      "epoch": 2.4201393127441405e-05,
      "step": 25377,
      "training_step_time": 7.694955587387085
    },
    {
      "epoch": 2.4202346801757813e-05,
      "model_forward_time": 1.723942756652832,
      "step": 25378
    },
    {
      "epoch": 2.4202346801757813e-05,
      "step": 25378,
      "training_step_time": 7.795150518417358
    },
    {
      "epoch": 2.420330047607422e-05,
      "model_forward_time": 1.715991735458374,
      "step": 25379
    },
    {
      "epoch": 2.420330047607422e-05,
      "step": 25379,
      "training_step_time": 7.409099340438843
    },
    {
      "epoch": 2.4204254150390624e-05,
      "grad_norm": 0.11336450278759003,
      "learning_rate": 8.906903091378667e-06,
      "loss": 0.061,
      "step": 25380
    },
    {
      "epoch": 2.4204254150390624e-05,
      "model_forward_time": 1.7149860858917236,
      "step": 25380
    },
    {
      "dynamics_loss_avg": 0.04960634633898735,
      "epoch": 2.4204254150390624e-05,
      "step": 25380
    },
    {
      "action_loss_avg": 0.0017568040639162064,
      "epoch": 2.4204254150390624e-05,
      "step": 25380
    },
    {
      "epoch": 2.4204254150390624e-05,
      "step": 25380,
      "training_step_time": 7.5325891971588135
    },
    {
      "epoch": 2.420520782470703e-05,
      "model_forward_time": 1.7540771961212158,
      "step": 25381
    },
    {
      "epoch": 2.420520782470703e-05,
      "step": 25381,
      "training_step_time": 9.42827820777893
    },
    {
      "epoch": 2.420616149902344e-05,
      "model_forward_time": 1.3633091449737549,
      "step": 25382
    },
    {
      "epoch": 2.420616149902344e-05,
      "step": 25382,
      "training_step_time": 8.078390836715698
    },
    {
      "epoch": 2.4207115173339843e-05,
      "model_forward_time": 1.70931077003479,
      "step": 25383
    },
    {
      "epoch": 2.4207115173339843e-05,
      "step": 25383,
      "training_step_time": 8.883344650268555
    },
    {
      "epoch": 2.420806884765625e-05,
      "model_forward_time": 2.860508918762207,
      "step": 25384
    },
    {
      "epoch": 2.420806884765625e-05,
      "step": 25384,
      "training_step_time": 7.602729082107544
    },
    {
      "epoch": 2.4209022521972658e-05,
      "model_forward_time": 1.7213997840881348,
      "step": 25385
    },
    {
      "epoch": 2.4209022521972658e-05,
      "step": 25385,
      "training_step_time": 9.479898452758789
    },
    {
      "epoch": 2.4209976196289062e-05,
      "model_forward_time": 1.7243030071258545,
      "step": 25386
    },
    {
      "epoch": 2.4209976196289062e-05,
      "step": 25386,
      "training_step_time": 7.56157374382019
    },
    {
      "epoch": 2.421092987060547e-05,
      "model_forward_time": 1.7236449718475342,
      "step": 25387
    },
    {
      "epoch": 2.421092987060547e-05,
      "step": 25387,
      "training_step_time": 7.733824968338013
    },
    {
      "epoch": 2.4211883544921877e-05,
      "model_forward_time": 1.730863332748413,
      "step": 25388
    },
    {
      "epoch": 2.4211883544921877e-05,
      "step": 25388,
      "training_step_time": 7.1563801765441895
    },
    {
      "epoch": 2.421283721923828e-05,
      "model_forward_time": 1.707087516784668,
      "step": 25389
    },
    {
      "epoch": 2.421283721923828e-05,
      "step": 25389,
      "training_step_time": 7.540755033493042
    },
    {
      "epoch": 2.4213790893554688e-05,
      "grad_norm": 0.13837331533432007,
      "learning_rate": 8.905871022079384e-06,
      "loss": 0.0614,
      "step": 25390
    },
    {
      "epoch": 2.4213790893554688e-05,
      "model_forward_time": 1.7250573635101318,
      "step": 25390
    },
    {
      "dynamics_loss_avg": 0.05737001709640026,
      "epoch": 2.4213790893554688e-05,
      "step": 25390
    },
    {
      "action_loss_avg": 0.0023935328994411976,
      "epoch": 2.4213790893554688e-05,
      "step": 25390
    },
    {
      "epoch": 2.4213790893554688e-05,
      "step": 25390,
      "training_step_time": 7.751638412475586
    },
    {
      "epoch": 2.4214744567871092e-05,
      "model_forward_time": 1.728996992111206,
      "step": 25391
    },
    {
      "epoch": 2.4214744567871092e-05,
      "step": 25391,
      "training_step_time": 8.797924518585205
    },
    {
      "epoch": 2.42156982421875e-05,
      "model_forward_time": 1.7170727252960205,
      "step": 25392
    },
    {
      "epoch": 2.42156982421875e-05,
      "step": 25392,
      "training_step_time": 8.186530828475952
    },
    {
      "epoch": 2.4216651916503907e-05,
      "model_forward_time": 2.8365638256073,
      "step": 25393
    },
    {
      "epoch": 2.4216651916503907e-05,
      "step": 25393,
      "training_step_time": 8.665149688720703
    },
    {
      "epoch": 2.421760559082031e-05,
      "model_forward_time": 1.7171921730041504,
      "step": 25394
    },
    {
      "epoch": 2.421760559082031e-05,
      "step": 25394,
      "training_step_time": 9.703174352645874
    },
    {
      "epoch": 2.421855926513672e-05,
      "model_forward_time": 1.719409704208374,
      "step": 25395
    },
    {
      "epoch": 2.421855926513672e-05,
      "step": 25395,
      "training_step_time": 7.590260982513428
    },
    {
      "epoch": 2.4219512939453126e-05,
      "model_forward_time": 1.7139577865600586,
      "step": 25396
    },
    {
      "epoch": 2.4219512939453126e-05,
      "step": 25396,
      "training_step_time": 7.755874395370483
    },
    {
      "epoch": 2.422046661376953e-05,
      "model_forward_time": 1.7141847610473633,
      "step": 25397
    },
    {
      "epoch": 2.422046661376953e-05,
      "step": 25397,
      "training_step_time": 7.186652898788452
    },
    {
      "epoch": 2.4221420288085937e-05,
      "model_forward_time": 1.7083392143249512,
      "step": 25398
    },
    {
      "epoch": 2.4221420288085937e-05,
      "step": 25398,
      "training_step_time": 8.589720726013184
    },
    {
      "epoch": 2.4222373962402345e-05,
      "model_forward_time": 1.7355365753173828,
      "step": 25399
    },
    {
      "epoch": 2.4222373962402345e-05,
      "step": 25399,
      "training_step_time": 7.668214797973633
    },
    {
      "epoch": 2.422332763671875e-05,
      "grad_norm": 0.15940508246421814,
      "learning_rate": 8.904838525639921e-06,
      "loss": 0.0623,
      "step": 25400
    },
    {
      "epoch": 2.422332763671875e-05,
      "model_forward_time": 1.3614461421966553,
      "step": 25400
    },
    {
      "dynamics_loss_avg": 0.060861959122121334,
      "epoch": 2.422332763671875e-05,
      "step": 25400
    },
    {
      "action_loss_avg": 0.0022722617839463057,
      "epoch": 2.422332763671875e-05,
      "step": 25400
    },
    {
      "epoch": 2.422332763671875e-05,
      "step": 25400,
      "training_step_time": 9.454964876174927
    },
    {
      "epoch": 2.4224281311035156e-05,
      "model_forward_time": 1.7144169807434082,
      "step": 25401
    },
    {
      "epoch": 2.4224281311035156e-05,
      "step": 25401,
      "training_step_time": 8.194236516952515
    },
    {
      "epoch": 2.4225234985351563e-05,
      "model_forward_time": 1.7102484703063965,
      "step": 25402
    },
    {
      "epoch": 2.4225234985351563e-05,
      "step": 25402,
      "training_step_time": 8.929887294769287
    },
    {
      "epoch": 2.4226188659667968e-05,
      "model_forward_time": 1.3511738777160645,
      "step": 25403
    },
    {
      "epoch": 2.4226188659667968e-05,
      "step": 25403,
      "training_step_time": 8.46407175064087
    },
    {
      "epoch": 2.4227142333984375e-05,
      "model_forward_time": 1.3633103370666504,
      "step": 25404
    },
    {
      "epoch": 2.4227142333984375e-05,
      "step": 25404,
      "training_step_time": 7.665304899215698
    },
    {
      "epoch": 2.4228096008300782e-05,
      "model_forward_time": 1.7190518379211426,
      "step": 25405
    },
    {
      "epoch": 2.4228096008300782e-05,
      "step": 25405,
      "training_step_time": 7.683391571044922
    },
    {
      "epoch": 2.4229049682617186e-05,
      "model_forward_time": 1.72320556640625,
      "step": 25406
    },
    {
      "epoch": 2.4229049682617186e-05,
      "step": 25406,
      "training_step_time": 7.762802839279175
    },
    {
      "epoch": 2.4230003356933594e-05,
      "model_forward_time": 1.7252771854400635,
      "step": 25407
    },
    {
      "epoch": 2.4230003356933594e-05,
      "step": 25407,
      "training_step_time": 7.557296514511108
    },
    {
      "epoch": 2.423095703125e-05,
      "model_forward_time": 1.746206521987915,
      "step": 25408
    },
    {
      "epoch": 2.423095703125e-05,
      "step": 25408,
      "training_step_time": 7.714504957199097
    },
    {
      "epoch": 2.4231910705566405e-05,
      "model_forward_time": 1.7235803604125977,
      "step": 25409
    },
    {
      "epoch": 2.4231910705566405e-05,
      "step": 25409,
      "training_step_time": 8.973127841949463
    },
    {
      "epoch": 2.4232864379882813e-05,
      "grad_norm": 0.15231451392173767,
      "learning_rate": 8.903805602173186e-06,
      "loss": 0.0592,
      "step": 25410
    },
    {
      "epoch": 2.4232864379882813e-05,
      "model_forward_time": 1.7487123012542725,
      "step": 25410
    },
    {
      "dynamics_loss_avg": 0.04625028036534786,
      "epoch": 2.4232864379882813e-05,
      "step": 25410
    },
    {
      "action_loss_avg": 0.0018512082227971406,
      "epoch": 2.4232864379882813e-05,
      "step": 25410
    },
    {
      "epoch": 2.4232864379882813e-05,
      "step": 25410,
      "training_step_time": 8.441845655441284
    },
    {
      "epoch": 2.423381805419922e-05,
      "model_forward_time": 1.7121305465698242,
      "step": 25411
    },
    {
      "epoch": 2.423381805419922e-05,
      "step": 25411,
      "training_step_time": 8.381605625152588
    },
    {
      "epoch": 2.4234771728515624e-05,
      "model_forward_time": 2.879384994506836,
      "step": 25412
    },
    {
      "epoch": 2.4234771728515624e-05,
      "step": 25412,
      "training_step_time": 7.632538318634033
    },
    {
      "epoch": 2.423572540283203e-05,
      "model_forward_time": 1.3516128063201904,
      "step": 25413
    },
    {
      "epoch": 2.423572540283203e-05,
      "step": 25413,
      "training_step_time": 7.627639532089233
    },
    {
      "epoch": 2.423667907714844e-05,
      "model_forward_time": 1.7350096702575684,
      "step": 25414
    },
    {
      "epoch": 2.423667907714844e-05,
      "step": 25414,
      "training_step_time": 7.754370927810669
    },
    {
      "epoch": 2.4237632751464843e-05,
      "model_forward_time": 1.7184267044067383,
      "step": 25415
    },
    {
      "epoch": 2.4237632751464843e-05,
      "step": 25415,
      "training_step_time": 8.400353908538818
    },
    {
      "epoch": 2.423858642578125e-05,
      "model_forward_time": 1.7285990715026855,
      "step": 25416
    },
    {
      "epoch": 2.423858642578125e-05,
      "step": 25416,
      "training_step_time": 7.57571268081665
    },
    {
      "epoch": 2.4239540100097658e-05,
      "model_forward_time": 1.742551326751709,
      "step": 25417
    },
    {
      "epoch": 2.4239540100097658e-05,
      "step": 25417,
      "training_step_time": 7.798506021499634
    },
    {
      "epoch": 2.4240493774414062e-05,
      "model_forward_time": 1.7271106243133545,
      "step": 25418
    },
    {
      "epoch": 2.4240493774414062e-05,
      "step": 25418,
      "training_step_time": 8.273810863494873
    },
    {
      "epoch": 2.424144744873047e-05,
      "model_forward_time": 1.7602722644805908,
      "step": 25419
    },
    {
      "epoch": 2.424144744873047e-05,
      "step": 25419,
      "training_step_time": 8.916386842727661
    },
    {
      "epoch": 2.4242401123046877e-05,
      "grad_norm": 0.1354459822177887,
      "learning_rate": 8.902772251792138e-06,
      "loss": 0.0648,
      "step": 25420
    },
    {
      "epoch": 2.4242401123046877e-05,
      "model_forward_time": 1.7109224796295166,
      "step": 25420
    },
    {
      "dynamics_loss_avg": 0.0519241064786911,
      "epoch": 2.4242401123046877e-05,
      "step": 25420
    },
    {
      "action_loss_avg": 0.0016524268488865347,
      "epoch": 2.4242401123046877e-05,
      "step": 25420
    },
    {
      "epoch": 2.4242401123046877e-05,
      "step": 25420,
      "training_step_time": 9.166696310043335
    },
    {
      "epoch": 2.424335479736328e-05,
      "model_forward_time": 2.8372626304626465,
      "step": 25421
    },
    {
      "epoch": 2.424335479736328e-05,
      "step": 25421,
      "training_step_time": 8.573567390441895
    },
    {
      "epoch": 2.4244308471679688e-05,
      "model_forward_time": 1.7233095169067383,
      "step": 25422
    },
    {
      "epoch": 2.4244308471679688e-05,
      "step": 25422,
      "training_step_time": 7.600302219390869
    },
    {
      "epoch": 2.4245262145996092e-05,
      "model_forward_time": 1.7229251861572266,
      "step": 25423
    },
    {
      "epoch": 2.4245262145996092e-05,
      "step": 25423,
      "training_step_time": 7.700915813446045
    },
    {
      "epoch": 2.42462158203125e-05,
      "model_forward_time": 1.716900110244751,
      "step": 25424
    },
    {
      "epoch": 2.42462158203125e-05,
      "step": 25424,
      "training_step_time": 7.734350204467773
    },
    {
      "epoch": 2.4247169494628907e-05,
      "model_forward_time": 1.730578899383545,
      "step": 25425
    },
    {
      "epoch": 2.4247169494628907e-05,
      "step": 25425,
      "training_step_time": 7.633315801620483
    },
    {
      "epoch": 2.424812316894531e-05,
      "model_forward_time": 1.749446153640747,
      "step": 25426
    },
    {
      "epoch": 2.424812316894531e-05,
      "step": 25426,
      "training_step_time": 7.829885721206665
    },
    {
      "epoch": 2.424907684326172e-05,
      "model_forward_time": 1.7177131175994873,
      "step": 25427
    },
    {
      "epoch": 2.424907684326172e-05,
      "step": 25427,
      "training_step_time": 7.7119598388671875
    },
    {
      "epoch": 2.4250030517578126e-05,
      "model_forward_time": 1.7339630126953125,
      "step": 25428
    },
    {
      "epoch": 2.4250030517578126e-05,
      "step": 25428,
      "training_step_time": 8.457412958145142
    },
    {
      "epoch": 2.425098419189453e-05,
      "model_forward_time": 1.7293701171875,
      "step": 25429
    },
    {
      "epoch": 2.425098419189453e-05,
      "step": 25429,
      "training_step_time": 7.708279132843018
    },
    {
      "epoch": 2.4251937866210937e-05,
      "grad_norm": 0.18793044984340668,
      "learning_rate": 8.901738474609786e-06,
      "loss": 0.0593,
      "step": 25430
    },
    {
      "epoch": 2.4251937866210937e-05,
      "model_forward_time": 1.7220942974090576,
      "step": 25430
    },
    {
      "dynamics_loss_avg": 0.053337080776691435,
      "epoch": 2.4251937866210937e-05,
      "step": 25430
    },
    {
      "action_loss_avg": 0.0018886187637690455,
      "epoch": 2.4251937866210937e-05,
      "step": 25430
    },
    {
      "epoch": 2.4251937866210937e-05,
      "step": 25430,
      "training_step_time": 8.13602614402771
    },
    {
      "epoch": 2.4252891540527345e-05,
      "model_forward_time": 1.7302219867706299,
      "step": 25431
    },
    {
      "epoch": 2.4252891540527345e-05,
      "step": 25431,
      "training_step_time": 8.199665546417236
    },
    {
      "epoch": 2.425384521484375e-05,
      "model_forward_time": 1.7271578311920166,
      "step": 25432
    },
    {
      "epoch": 2.425384521484375e-05,
      "step": 25432,
      "training_step_time": 7.6227569580078125
    },
    {
      "epoch": 2.4254798889160156e-05,
      "model_forward_time": 1.723637342453003,
      "step": 25433
    },
    {
      "epoch": 2.4254798889160156e-05,
      "step": 25433,
      "training_step_time": 7.72380256652832
    },
    {
      "epoch": 2.4255752563476564e-05,
      "model_forward_time": 1.7274599075317383,
      "step": 25434
    },
    {
      "epoch": 2.4255752563476564e-05,
      "step": 25434,
      "training_step_time": 7.626876592636108
    },
    {
      "epoch": 2.4256706237792968e-05,
      "model_forward_time": 1.732414960861206,
      "step": 25435
    },
    {
      "epoch": 2.4256706237792968e-05,
      "step": 25435,
      "training_step_time": 7.755040168762207
    },
    {
      "epoch": 2.4257659912109375e-05,
      "model_forward_time": 1.7184572219848633,
      "step": 25436
    },
    {
      "epoch": 2.4257659912109375e-05,
      "step": 25436,
      "training_step_time": 8.559677124023438
    },
    {
      "epoch": 2.4258613586425782e-05,
      "model_forward_time": 1.7277123928070068,
      "step": 25437
    },
    {
      "epoch": 2.4258613586425782e-05,
      "step": 25437,
      "training_step_time": 7.676466703414917
    },
    {
      "epoch": 2.4259567260742186e-05,
      "model_forward_time": 1.7205407619476318,
      "step": 25438
    },
    {
      "epoch": 2.4259567260742186e-05,
      "step": 25438,
      "training_step_time": 7.733837842941284
    },
    {
      "epoch": 2.4260520935058594e-05,
      "model_forward_time": 1.7264783382415771,
      "step": 25439
    },
    {
      "epoch": 2.4260520935058594e-05,
      "step": 25439,
      "training_step_time": 8.480902433395386
    },
    {
      "epoch": 2.4261474609375e-05,
      "grad_norm": 0.19564999639987946,
      "learning_rate": 8.90070427073918e-06,
      "loss": 0.0626,
      "step": 25440
    },
    {
      "epoch": 2.4261474609375e-05,
      "model_forward_time": 2.9615116119384766,
      "step": 25440
    },
    {
      "dynamics_loss_avg": 0.05630157813429833,
      "epoch": 2.4261474609375e-05,
      "step": 25440
    },
    {
      "action_loss_avg": 0.0019144670892274006,
      "epoch": 2.4261474609375e-05,
      "step": 25440
    },
    {
      "epoch": 2.4261474609375e-05,
      "step": 25440,
      "training_step_time": 7.791313409805298
    },
    {
      "epoch": 2.4262428283691405e-05,
      "model_forward_time": 1.7126331329345703,
      "step": 25441
    },
    {
      "epoch": 2.4262428283691405e-05,
      "step": 25441,
      "training_step_time": 8.429685831069946
    },
    {
      "epoch": 2.4263381958007813e-05,
      "model_forward_time": 1.723813772201538,
      "step": 25442
    },
    {
      "epoch": 2.4263381958007813e-05,
      "step": 25442,
      "training_step_time": 8.717123031616211
    },
    {
      "epoch": 2.426433563232422e-05,
      "model_forward_time": 1.3599073886871338,
      "step": 25443
    },
    {
      "epoch": 2.426433563232422e-05,
      "step": 25443,
      "training_step_time": 7.709493160247803
    },
    {
      "epoch": 2.4265289306640624e-05,
      "model_forward_time": 1.745542049407959,
      "step": 25444
    },
    {
      "epoch": 2.4265289306640624e-05,
      "step": 25444,
      "training_step_time": 7.804940462112427
    },
    {
      "epoch": 2.426624298095703e-05,
      "model_forward_time": 1.732990026473999,
      "step": 25445
    },
    {
      "epoch": 2.426624298095703e-05,
      "step": 25445,
      "training_step_time": 7.766552209854126
    },
    {
      "epoch": 2.426719665527344e-05,
      "model_forward_time": 1.7199702262878418,
      "step": 25446
    },
    {
      "epoch": 2.426719665527344e-05,
      "step": 25446,
      "training_step_time": 8.223860502243042
    },
    {
      "epoch": 2.4268150329589843e-05,
      "model_forward_time": 1.7121670246124268,
      "step": 25447
    },
    {
      "epoch": 2.4268150329589843e-05,
      "step": 25447,
      "training_step_time": 7.877993822097778
    },
    {
      "epoch": 2.426910400390625e-05,
      "model_forward_time": 1.7193753719329834,
      "step": 25448
    },
    {
      "epoch": 2.426910400390625e-05,
      "step": 25448,
      "training_step_time": 8.635643720626831
    },
    {
      "epoch": 2.4270057678222658e-05,
      "model_forward_time": 2.850573778152466,
      "step": 25449
    },
    {
      "epoch": 2.4270057678222658e-05,
      "step": 25449,
      "training_step_time": 8.549618482589722
    },
    {
      "epoch": 2.4271011352539062e-05,
      "grad_norm": 0.17582771182060242,
      "learning_rate": 8.89966964029342e-06,
      "loss": 0.0682,
      "step": 25450
    },
    {
      "epoch": 2.4271011352539062e-05,
      "model_forward_time": 1.7218530178070068,
      "step": 25450
    },
    {
      "dynamics_loss_avg": 0.06898904554545879,
      "epoch": 2.4271011352539062e-05,
      "step": 25450
    },
    {
      "action_loss_avg": 0.0021475695772096514,
      "epoch": 2.4271011352539062e-05,
      "step": 25450
    },
    {
      "epoch": 2.4271011352539062e-05,
      "step": 25450,
      "training_step_time": 8.625607967376709
    },
    {
      "epoch": 2.427196502685547e-05,
      "model_forward_time": 1.7278189659118652,
      "step": 25451
    },
    {
      "epoch": 2.427196502685547e-05,
      "step": 25451,
      "training_step_time": 8.72849988937378
    },
    {
      "epoch": 2.4272918701171877e-05,
      "model_forward_time": 1.7260637283325195,
      "step": 25452
    },
    {
      "epoch": 2.4272918701171877e-05,
      "step": 25452,
      "training_step_time": 7.5878870487213135
    },
    {
      "epoch": 2.427387237548828e-05,
      "model_forward_time": 1.7511188983917236,
      "step": 25453
    },
    {
      "epoch": 2.427387237548828e-05,
      "step": 25453,
      "training_step_time": 7.674392461776733
    },
    {
      "epoch": 2.4274826049804688e-05,
      "model_forward_time": 1.7251603603363037,
      "step": 25454
    },
    {
      "epoch": 2.4274826049804688e-05,
      "step": 25454,
      "training_step_time": 8.543200731277466
    },
    {
      "epoch": 2.4275779724121092e-05,
      "model_forward_time": 1.7200970649719238,
      "step": 25455
    },
    {
      "epoch": 2.4275779724121092e-05,
      "step": 25455,
      "training_step_time": 8.54649305343628
    },
    {
      "epoch": 2.42767333984375e-05,
      "model_forward_time": 1.7522549629211426,
      "step": 25456
    },
    {
      "epoch": 2.42767333984375e-05,
      "step": 25456,
      "training_step_time": 8.544427394866943
    },
    {
      "epoch": 2.4277687072753907e-05,
      "model_forward_time": 1.7279434204101562,
      "step": 25457
    },
    {
      "epoch": 2.4277687072753907e-05,
      "step": 25457,
      "training_step_time": 7.638490438461304
    },
    {
      "epoch": 2.427864074707031e-05,
      "model_forward_time": 2.832240104675293,
      "step": 25458
    },
    {
      "epoch": 2.427864074707031e-05,
      "step": 25458,
      "training_step_time": 8.635499477386475
    },
    {
      "epoch": 2.427959442138672e-05,
      "model_forward_time": 1.7201437950134277,
      "step": 25459
    },
    {
      "epoch": 2.427959442138672e-05,
      "step": 25459,
      "training_step_time": 7.5977442264556885
    },
    {
      "epoch": 2.4280548095703126e-05,
      "grad_norm": 0.21333900094032288,
      "learning_rate": 8.898634583385653e-06,
      "loss": 0.0634,
      "step": 25460
    },
    {
      "epoch": 2.4280548095703126e-05,
      "model_forward_time": 1.7172601222991943,
      "step": 25460
    },
    {
      "dynamics_loss_avg": 0.039466348849236964,
      "epoch": 2.4280548095703126e-05,
      "step": 25460
    },
    {
      "action_loss_avg": 0.0010955155710689723,
      "epoch": 2.4280548095703126e-05,
      "step": 25460
    },
    {
      "epoch": 2.4280548095703126e-05,
      "step": 25460,
      "training_step_time": 7.719062805175781
    },
    {
      "epoch": 2.428150177001953e-05,
      "model_forward_time": 1.7195537090301514,
      "step": 25461
    },
    {
      "epoch": 2.428150177001953e-05,
      "step": 25461,
      "training_step_time": 7.6322691440582275
    },
    {
      "epoch": 2.4282455444335937e-05,
      "model_forward_time": 1.7366881370544434,
      "step": 25462
    },
    {
      "epoch": 2.4282455444335937e-05,
      "step": 25462,
      "training_step_time": 8.715851783752441
    },
    {
      "epoch": 2.4283409118652345e-05,
      "model_forward_time": 1.7246510982513428,
      "step": 25463
    },
    {
      "epoch": 2.4283409118652345e-05,
      "step": 25463,
      "training_step_time": 7.758347511291504
    },
    {
      "epoch": 2.428436279296875e-05,
      "model_forward_time": 1.711390733718872,
      "step": 25464
    },
    {
      "epoch": 2.428436279296875e-05,
      "step": 25464,
      "training_step_time": 7.699918508529663
    },
    {
      "epoch": 2.4285316467285156e-05,
      "model_forward_time": 1.7275772094726562,
      "step": 25465
    },
    {
      "epoch": 2.4285316467285156e-05,
      "step": 25465,
      "training_step_time": 7.701992988586426
    },
    {
      "epoch": 2.4286270141601564e-05,
      "model_forward_time": 1.713942289352417,
      "step": 25466
    },
    {
      "epoch": 2.4286270141601564e-05,
      "step": 25466,
      "training_step_time": 8.135028600692749
    },
    {
      "epoch": 2.4287223815917968e-05,
      "model_forward_time": 1.3592915534973145,
      "step": 25467
    },
    {
      "epoch": 2.4287223815917968e-05,
      "step": 25467,
      "training_step_time": 8.200541973114014
    },
    {
      "epoch": 2.4288177490234375e-05,
      "model_forward_time": 1.719423770904541,
      "step": 25468
    },
    {
      "epoch": 2.4288177490234375e-05,
      "step": 25468,
      "training_step_time": 8.220805406570435
    },
    {
      "epoch": 2.4289131164550782e-05,
      "model_forward_time": 1.7146453857421875,
      "step": 25469
    },
    {
      "epoch": 2.4289131164550782e-05,
      "step": 25469,
      "training_step_time": 7.671375751495361
    },
    {
      "epoch": 2.4290084838867186e-05,
      "grad_norm": 0.146428182721138,
      "learning_rate": 8.897599100129066e-06,
      "loss": 0.0625,
      "step": 25470
    },
    {
      "epoch": 2.4290084838867186e-05,
      "model_forward_time": 1.7114253044128418,
      "step": 25470
    },
    {
      "dynamics_loss_avg": 0.06975531335920096,
      "epoch": 2.4290084838867186e-05,
      "step": 25470
    },
    {
      "action_loss_avg": 0.0026788100658450276,
      "epoch": 2.4290084838867186e-05,
      "step": 25470
    },
    {
      "epoch": 2.4290084838867186e-05,
      "step": 25470,
      "training_step_time": 7.735178470611572
    },
    {
      "epoch": 2.4291038513183594e-05,
      "model_forward_time": 1.730356216430664,
      "step": 25471
    },
    {
      "epoch": 2.4291038513183594e-05,
      "step": 25471,
      "training_step_time": 7.817902326583862
    },
    {
      "epoch": 2.42919921875e-05,
      "model_forward_time": 1.7180216312408447,
      "step": 25472
    },
    {
      "epoch": 2.42919921875e-05,
      "step": 25472,
      "training_step_time": 8.721585750579834
    },
    {
      "epoch": 2.4292945861816405e-05,
      "model_forward_time": 1.7210781574249268,
      "step": 25473
    },
    {
      "epoch": 2.4292945861816405e-05,
      "step": 25473,
      "training_step_time": 7.73328971862793
    },
    {
      "epoch": 2.4293899536132813e-05,
      "model_forward_time": 1.7366845607757568,
      "step": 25474
    },
    {
      "epoch": 2.4293899536132813e-05,
      "step": 25474,
      "training_step_time": 8.25847578048706
    },
    {
      "epoch": 2.429485321044922e-05,
      "model_forward_time": 1.7185795307159424,
      "step": 25475
    },
    {
      "epoch": 2.429485321044922e-05,
      "step": 25475,
      "training_step_time": 7.657639741897583
    },
    {
      "epoch": 2.4295806884765624e-05,
      "model_forward_time": 1.7128589153289795,
      "step": 25476
    },
    {
      "epoch": 2.4295806884765624e-05,
      "step": 25476,
      "training_step_time": 7.779354572296143
    },
    {
      "epoch": 2.429676055908203e-05,
      "model_forward_time": 2.895852565765381,
      "step": 25477
    },
    {
      "epoch": 2.429676055908203e-05,
      "step": 25477,
      "training_step_time": 8.750408172607422
    },
    {
      "epoch": 2.429771423339844e-05,
      "model_forward_time": 1.7182316780090332,
      "step": 25478
    },
    {
      "epoch": 2.429771423339844e-05,
      "step": 25478,
      "training_step_time": 8.612223148345947
    },
    {
      "epoch": 2.4298667907714843e-05,
      "model_forward_time": 1.7256689071655273,
      "step": 25479
    },
    {
      "epoch": 2.4298667907714843e-05,
      "step": 25479,
      "training_step_time": 8.737193584442139
    },
    {
      "epoch": 2.429962158203125e-05,
      "grad_norm": 0.12876544892787933,
      "learning_rate": 8.896563190636904e-06,
      "loss": 0.0665,
      "step": 25480
    },
    {
      "epoch": 2.429962158203125e-05,
      "model_forward_time": 1.7274370193481445,
      "step": 25480
    },
    {
      "dynamics_loss_avg": 0.04194351620972157,
      "epoch": 2.429962158203125e-05,
      "step": 25480
    },
    {
      "action_loss_avg": 0.0010479024611413478,
      "epoch": 2.429962158203125e-05,
      "step": 25480
    },
    {
      "epoch": 2.429962158203125e-05,
      "step": 25480,
      "training_step_time": 7.682478666305542
    },
    {
      "epoch": 2.4300575256347658e-05,
      "model_forward_time": 1.713994026184082,
      "step": 25481
    },
    {
      "epoch": 2.4300575256347658e-05,
      "step": 25481,
      "training_step_time": 7.743287086486816
    },
    {
      "epoch": 2.4301528930664062e-05,
      "model_forward_time": 1.7159035205841064,
      "step": 25482
    },
    {
      "epoch": 2.4301528930664062e-05,
      "step": 25482,
      "training_step_time": 8.18652081489563
    },
    {
      "epoch": 2.430248260498047e-05,
      "model_forward_time": 1.7143008708953857,
      "step": 25483
    },
    {
      "epoch": 2.430248260498047e-05,
      "step": 25483,
      "training_step_time": 9.35280728340149
    },
    {
      "epoch": 2.4303436279296877e-05,
      "model_forward_time": 1.7197990417480469,
      "step": 25484
    },
    {
      "epoch": 2.4303436279296877e-05,
      "step": 25484,
      "training_step_time": 7.777721881866455
    },
    {
      "epoch": 2.430438995361328e-05,
      "model_forward_time": 1.714806079864502,
      "step": 25485
    },
    {
      "epoch": 2.430438995361328e-05,
      "step": 25485,
      "training_step_time": 9.274622678756714
    },
    {
      "epoch": 2.4305343627929688e-05,
      "model_forward_time": 2.905853033065796,
      "step": 25486
    },
    {
      "epoch": 2.4305343627929688e-05,
      "step": 25486,
      "training_step_time": 8.754113674163818
    },
    {
      "epoch": 2.4306297302246092e-05,
      "model_forward_time": 1.7183995246887207,
      "step": 25487
    },
    {
      "epoch": 2.4306297302246092e-05,
      "step": 25487,
      "training_step_time": 8.51547122001648
    },
    {
      "epoch": 2.43072509765625e-05,
      "model_forward_time": 1.7218284606933594,
      "step": 25488
    },
    {
      "epoch": 2.43072509765625e-05,
      "step": 25488,
      "training_step_time": 7.826128005981445
    },
    {
      "epoch": 2.4308204650878907e-05,
      "model_forward_time": 1.7159624099731445,
      "step": 25489
    },
    {
      "epoch": 2.4308204650878907e-05,
      "step": 25489,
      "training_step_time": 7.741144895553589
    },
    {
      "epoch": 2.430915832519531e-05,
      "grad_norm": 0.18688316643238068,
      "learning_rate": 8.895526855022448e-06,
      "loss": 0.0684,
      "step": 25490
    },
    {
      "epoch": 2.430915832519531e-05,
      "model_forward_time": 1.7285706996917725,
      "step": 25490
    },
    {
      "dynamics_loss_avg": 0.06414593253284692,
      "epoch": 2.430915832519531e-05,
      "step": 25490
    },
    {
      "action_loss_avg": 0.0017905188724398613,
      "epoch": 2.430915832519531e-05,
      "step": 25490
    },
    {
      "epoch": 2.430915832519531e-05,
      "step": 25490,
      "training_step_time": 7.7253334522247314
    },
    {
      "epoch": 2.431011199951172e-05,
      "model_forward_time": 1.7259433269500732,
      "step": 25491
    },
    {
      "epoch": 2.431011199951172e-05,
      "step": 25491,
      "training_step_time": 10.550110101699829
    },
    {
      "epoch": 2.4311065673828126e-05,
      "model_forward_time": 1.7265980243682861,
      "step": 25492
    },
    {
      "epoch": 2.4311065673828126e-05,
      "step": 25492,
      "training_step_time": 7.796741008758545
    },
    {
      "epoch": 2.431201934814453e-05,
      "model_forward_time": 1.741682767868042,
      "step": 25493
    },
    {
      "epoch": 2.431201934814453e-05,
      "step": 25493,
      "training_step_time": 7.781362771987915
    },
    {
      "epoch": 2.4312973022460937e-05,
      "model_forward_time": 1.7137768268585205,
      "step": 25494
    },
    {
      "epoch": 2.4312973022460937e-05,
      "step": 25494,
      "training_step_time": 8.56043791770935
    },
    {
      "epoch": 2.4313926696777345e-05,
      "model_forward_time": 2.8736445903778076,
      "step": 25495
    },
    {
      "epoch": 2.4313926696777345e-05,
      "step": 25495,
      "training_step_time": 7.568863868713379
    },
    {
      "epoch": 2.431488037109375e-05,
      "model_forward_time": 1.7231793403625488,
      "step": 25496
    },
    {
      "epoch": 2.431488037109375e-05,
      "step": 25496,
      "training_step_time": 8.514325141906738
    },
    {
      "epoch": 2.4315834045410156e-05,
      "model_forward_time": 1.7244203090667725,
      "step": 25497
    },
    {
      "epoch": 2.4315834045410156e-05,
      "step": 25497,
      "training_step_time": 7.673352479934692
    },
    {
      "epoch": 2.4316787719726564e-05,
      "model_forward_time": 1.7192573547363281,
      "step": 25498
    },
    {
      "epoch": 2.4316787719726564e-05,
      "step": 25498,
      "training_step_time": 7.6269145011901855
    },
    {
      "epoch": 2.4317741394042968e-05,
      "model_forward_time": 1.7518057823181152,
      "step": 25499
    },
    {
      "epoch": 2.4317741394042968e-05,
      "step": 25499,
      "training_step_time": 7.653871059417725
    },
    {
      "epoch": 2.4318695068359375e-05,
      "grad_norm": 0.19552592933177948,
      "learning_rate": 8.894490093399034e-06,
      "loss": 0.0624,
      "step": 25500
    },
    {
      "epoch": 2.4318695068359375e-05,
      "model_forward_time": 1.7170696258544922,
      "step": 25500
    },
    {
      "dynamics_loss_avg": 0.0686732992529869,
      "epoch": 2.4318695068359375e-05,
      "step": 25500
    },
    {
      "action_loss_avg": 0.0035082547081401573,
      "epoch": 2.4318695068359375e-05,
      "step": 25500
    },
    {
      "epoch": 2.4318695068359375e-05,
      "step": 25500,
      "training_step_time": 7.710762977600098
    },
    {
      "epoch": 2.4319648742675782e-05,
      "model_forward_time": 1.717153549194336,
      "step": 25501
    },
    {
      "epoch": 2.4319648742675782e-05,
      "step": 25501,
      "training_step_time": 8.755716562271118
    },
    {
      "epoch": 2.4320602416992186e-05,
      "model_forward_time": 1.745762825012207,
      "step": 25502
    },
    {
      "epoch": 2.4320602416992186e-05,
      "step": 25502,
      "training_step_time": 9.263437986373901
    },
    {
      "epoch": 2.4321556091308594e-05,
      "model_forward_time": 1.7175383567810059,
      "step": 25503
    },
    {
      "epoch": 2.4321556091308594e-05,
      "step": 25503,
      "training_step_time": 9.356935977935791
    },
    {
      "epoch": 2.4322509765625e-05,
      "model_forward_time": 2.8606173992156982,
      "step": 25504
    },
    {
      "epoch": 2.4322509765625e-05,
      "step": 25504,
      "training_step_time": 8.497431993484497
    },
    {
      "epoch": 2.4323463439941405e-05,
      "model_forward_time": 1.7192001342773438,
      "step": 25505
    },
    {
      "epoch": 2.4323463439941405e-05,
      "step": 25505,
      "training_step_time": 7.587868690490723
    },
    {
      "epoch": 2.4324417114257813e-05,
      "model_forward_time": 1.7241275310516357,
      "step": 25506
    },
    {
      "epoch": 2.4324417114257813e-05,
      "step": 25506,
      "training_step_time": 7.657790899276733
    },
    {
      "epoch": 2.432537078857422e-05,
      "model_forward_time": 1.7316949367523193,
      "step": 25507
    },
    {
      "epoch": 2.432537078857422e-05,
      "step": 25507,
      "training_step_time": 7.773480653762817
    },
    {
      "epoch": 2.4326324462890624e-05,
      "model_forward_time": 1.7393081188201904,
      "step": 25508
    },
    {
      "epoch": 2.4326324462890624e-05,
      "step": 25508,
      "training_step_time": 7.4203760623931885
    },
    {
      "epoch": 2.432727813720703e-05,
      "model_forward_time": 1.718106746673584,
      "step": 25509
    },
    {
      "epoch": 2.432727813720703e-05,
      "step": 25509,
      "training_step_time": 7.780054569244385
    },
    {
      "epoch": 2.432823181152344e-05,
      "grad_norm": 0.17134739458560944,
      "learning_rate": 8.893452905880035e-06,
      "loss": 0.0625,
      "step": 25510
    },
    {
      "epoch": 2.432823181152344e-05,
      "model_forward_time": 1.7305772304534912,
      "step": 25510
    },
    {
      "dynamics_loss_avg": 0.054611852020025255,
      "epoch": 2.432823181152344e-05,
      "step": 25510
    },
    {
      "action_loss_avg": 0.002528334961971268,
      "epoch": 2.432823181152344e-05,
      "step": 25510
    },
    {
      "epoch": 2.432823181152344e-05,
      "step": 25510,
      "training_step_time": 7.6647789478302
    },
    {
      "epoch": 2.4329185485839843e-05,
      "model_forward_time": 1.7391552925109863,
      "step": 25511
    },
    {
      "epoch": 2.4329185485839843e-05,
      "step": 25511,
      "training_step_time": 8.461904287338257
    },
    {
      "epoch": 2.433013916015625e-05,
      "model_forward_time": 1.7344138622283936,
      "step": 25512
    },
    {
      "epoch": 2.433013916015625e-05,
      "step": 25512,
      "training_step_time": 7.674624443054199
    },
    {
      "epoch": 2.4331092834472658e-05,
      "model_forward_time": 2.8099777698516846,
      "step": 25513
    },
    {
      "epoch": 2.4331092834472658e-05,
      "step": 25513,
      "training_step_time": 9.654770851135254
    },
    {
      "epoch": 2.4332046508789062e-05,
      "model_forward_time": 1.7169702053070068,
      "step": 25514
    },
    {
      "epoch": 2.4332046508789062e-05,
      "step": 25514,
      "training_step_time": 9.197763204574585
    },
    {
      "epoch": 2.433300018310547e-05,
      "model_forward_time": 1.7239601612091064,
      "step": 25515
    },
    {
      "epoch": 2.433300018310547e-05,
      "step": 25515,
      "training_step_time": 7.55638313293457
    },
    {
      "epoch": 2.4333953857421873e-05,
      "model_forward_time": 1.7203974723815918,
      "step": 25516
    },
    {
      "epoch": 2.4333953857421873e-05,
      "step": 25516,
      "training_step_time": 7.67183780670166
    },
    {
      "epoch": 2.433490753173828e-05,
      "model_forward_time": 1.7382466793060303,
      "step": 25517
    },
    {
      "epoch": 2.433490753173828e-05,
      "step": 25517,
      "training_step_time": 8.371023416519165
    },
    {
      "epoch": 2.4335861206054688e-05,
      "model_forward_time": 1.7105927467346191,
      "step": 25518
    },
    {
      "epoch": 2.4335861206054688e-05,
      "step": 25518,
      "training_step_time": 7.738694667816162
    },
    {
      "epoch": 2.4336814880371092e-05,
      "model_forward_time": 1.7292921543121338,
      "step": 25519
    },
    {
      "epoch": 2.4336814880371092e-05,
      "step": 25519,
      "training_step_time": 7.673164129257202
    },
    {
      "epoch": 2.43377685546875e-05,
      "grad_norm": 0.19522599875926971,
      "learning_rate": 8.892415292578882e-06,
      "loss": 0.0652,
      "step": 25520
    },
    {
      "epoch": 2.43377685546875e-05,
      "model_forward_time": 1.7303147315979004,
      "step": 25520
    },
    {
      "dynamics_loss_avg": 0.06245699934661388,
      "epoch": 2.43377685546875e-05,
      "step": 25520
    },
    {
      "action_loss_avg": 0.0032570795097853988,
      "epoch": 2.43377685546875e-05,
      "step": 25520
    },
    {
      "epoch": 2.43377685546875e-05,
      "step": 25520,
      "training_step_time": 8.699933290481567
    },
    {
      "epoch": 2.4338722229003907e-05,
      "model_forward_time": 1.727529764175415,
      "step": 25521
    },
    {
      "epoch": 2.4338722229003907e-05,
      "step": 25521,
      "training_step_time": 7.924204349517822
    },
    {
      "epoch": 2.433967590332031e-05,
      "model_forward_time": 1.7152838706970215,
      "step": 25522
    },
    {
      "epoch": 2.433967590332031e-05,
      "step": 25522,
      "training_step_time": 9.171928405761719
    },
    {
      "epoch": 2.434062957763672e-05,
      "model_forward_time": 1.7387325763702393,
      "step": 25523
    },
    {
      "epoch": 2.434062957763672e-05,
      "step": 25523,
      "training_step_time": 7.628492832183838
    },
    {
      "epoch": 2.4341583251953126e-05,
      "model_forward_time": 1.735255479812622,
      "step": 25524
    },
    {
      "epoch": 2.4341583251953126e-05,
      "step": 25524,
      "training_step_time": 7.621539354324341
    },
    {
      "epoch": 2.434253692626953e-05,
      "model_forward_time": 1.7298493385314941,
      "step": 25525
    },
    {
      "epoch": 2.434253692626953e-05,
      "step": 25525,
      "training_step_time": 7.65060830116272
    },
    {
      "epoch": 2.4343490600585937e-05,
      "model_forward_time": 1.7508265972137451,
      "step": 25526
    },
    {
      "epoch": 2.4343490600585937e-05,
      "step": 25526,
      "training_step_time": 7.538695812225342
    },
    {
      "epoch": 2.4344444274902345e-05,
      "model_forward_time": 1.3726561069488525,
      "step": 25527
    },
    {
      "epoch": 2.4344444274902345e-05,
      "step": 25527,
      "training_step_time": 8.744933605194092
    },
    {
      "epoch": 2.434539794921875e-05,
      "model_forward_time": 1.739166498184204,
      "step": 25528
    },
    {
      "epoch": 2.434539794921875e-05,
      "step": 25528,
      "training_step_time": 8.3847074508667
    },
    {
      "epoch": 2.4346351623535156e-05,
      "model_forward_time": 1.7370507717132568,
      "step": 25529
    },
    {
      "epoch": 2.4346351623535156e-05,
      "step": 25529,
      "training_step_time": 8.277525186538696
    },
    {
      "epoch": 2.4347305297851564e-05,
      "grad_norm": 0.1899106353521347,
      "learning_rate": 8.891377253609045e-06,
      "loss": 0.0633,
      "step": 25530
    },
    {
      "epoch": 2.4347305297851564e-05,
      "model_forward_time": 1.7288792133331299,
      "step": 25530
    },
    {
      "dynamics_loss_avg": 0.06400146149098873,
      "epoch": 2.4347305297851564e-05,
      "step": 25530
    },
    {
      "action_loss_avg": 0.002436855819541961,
      "epoch": 2.4347305297851564e-05,
      "step": 25530
    },
    {
      "epoch": 2.4347305297851564e-05,
      "step": 25530,
      "training_step_time": 8.609361410140991
    },
    {
      "epoch": 2.4348258972167968e-05,
      "model_forward_time": 1.7319531440734863,
      "step": 25531
    },
    {
      "epoch": 2.4348258972167968e-05,
      "step": 25531,
      "training_step_time": 8.72263479232788
    },
    {
      "epoch": 2.4349212646484375e-05,
      "model_forward_time": 2.8954451084136963,
      "step": 25532
    },
    {
      "epoch": 2.4349212646484375e-05,
      "step": 25532,
      "training_step_time": 8.340253114700317
    },
    {
      "epoch": 2.4350166320800782e-05,
      "model_forward_time": 1.738868236541748,
      "step": 25533
    },
    {
      "epoch": 2.4350166320800782e-05,
      "step": 25533,
      "training_step_time": 7.5791099071502686
    },
    {
      "epoch": 2.4351119995117186e-05,
      "model_forward_time": 1.7335262298583984,
      "step": 25534
    },
    {
      "epoch": 2.4351119995117186e-05,
      "step": 25534,
      "training_step_time": 8.280051946640015
    },
    {
      "epoch": 2.4352073669433594e-05,
      "model_forward_time": 1.7439239025115967,
      "step": 25535
    },
    {
      "epoch": 2.4352073669433594e-05,
      "step": 25535,
      "training_step_time": 7.753471374511719
    },
    {
      "epoch": 2.435302734375e-05,
      "model_forward_time": 1.395312786102295,
      "step": 25536
    },
    {
      "epoch": 2.435302734375e-05,
      "step": 25536,
      "training_step_time": 7.720517158508301
    },
    {
      "epoch": 2.4353981018066405e-05,
      "model_forward_time": 1.7318572998046875,
      "step": 25537
    },
    {
      "epoch": 2.4353981018066405e-05,
      "step": 25537,
      "training_step_time": 8.57063913345337
    },
    {
      "epoch": 2.4354934692382813e-05,
      "model_forward_time": 1.7418172359466553,
      "step": 25538
    },
    {
      "epoch": 2.4354934692382813e-05,
      "step": 25538,
      "training_step_time": 8.452197074890137
    },
    {
      "epoch": 2.435588836669922e-05,
      "model_forward_time": 1.760324478149414,
      "step": 25539
    },
    {
      "epoch": 2.435588836669922e-05,
      "step": 25539,
      "training_step_time": 8.827226161956787
    },
    {
      "epoch": 2.4356842041015624e-05,
      "grad_norm": 0.1241852194070816,
      "learning_rate": 8.890338789084043e-06,
      "loss": 0.063,
      "step": 25540
    },
    {
      "epoch": 2.4356842041015624e-05,
      "model_forward_time": 1.7300128936767578,
      "step": 25540
    },
    {
      "dynamics_loss_avg": 0.04622608572244644,
      "epoch": 2.4356842041015624e-05,
      "step": 25540
    },
    {
      "action_loss_avg": 0.001975785882677883,
      "epoch": 2.4356842041015624e-05,
      "step": 25540
    },
    {
      "epoch": 2.4356842041015624e-05,
      "step": 25540,
      "training_step_time": 9.371356964111328
    },
    {
      "epoch": 2.435779571533203e-05,
      "model_forward_time": 2.8936924934387207,
      "step": 25541
    },
    {
      "epoch": 2.435779571533203e-05,
      "step": 25541,
      "training_step_time": 7.7689478397369385
    },
    {
      "epoch": 2.435874938964844e-05,
      "model_forward_time": 1.7512664794921875,
      "step": 25542
    },
    {
      "epoch": 2.435874938964844e-05,
      "step": 25542,
      "training_step_time": 7.693944931030273
    },
    {
      "epoch": 2.4359703063964843e-05,
      "model_forward_time": 1.738008975982666,
      "step": 25543
    },
    {
      "epoch": 2.4359703063964843e-05,
      "step": 25543,
      "training_step_time": 7.8472676277160645
    },
    {
      "epoch": 2.436065673828125e-05,
      "model_forward_time": 1.734192132949829,
      "step": 25544
    },
    {
      "epoch": 2.436065673828125e-05,
      "step": 25544,
      "training_step_time": 8.726553916931152
    },
    {
      "epoch": 2.4361610412597658e-05,
      "model_forward_time": 1.7653229236602783,
      "step": 25545
    },
    {
      "epoch": 2.4361610412597658e-05,
      "step": 25545,
      "training_step_time": 7.646878719329834
    },
    {
      "epoch": 2.4362564086914062e-05,
      "model_forward_time": 1.7372305393218994,
      "step": 25546
    },
    {
      "epoch": 2.4362564086914062e-05,
      "step": 25546,
      "training_step_time": 7.589674472808838
    },
    {
      "epoch": 2.436351776123047e-05,
      "model_forward_time": 1.735194206237793,
      "step": 25547
    },
    {
      "epoch": 2.436351776123047e-05,
      "step": 25547,
      "training_step_time": 9.03489089012146
    },
    {
      "epoch": 2.4364471435546873e-05,
      "model_forward_time": 1.7500543594360352,
      "step": 25548
    },
    {
      "epoch": 2.4364471435546873e-05,
      "step": 25548,
      "training_step_time": 9.816687107086182
    },
    {
      "epoch": 2.436542510986328e-05,
      "model_forward_time": 1.738077163696289,
      "step": 25549
    },
    {
      "epoch": 2.436542510986328e-05,
      "step": 25549,
      "training_step_time": 7.661344528198242
    },
    {
      "epoch": 2.4366378784179688e-05,
      "grad_norm": 0.16836217045783997,
      "learning_rate": 8.88929989911744e-06,
      "loss": 0.0621,
      "step": 25550
    },
    {
      "epoch": 2.4366378784179688e-05,
      "model_forward_time": 2.833627700805664,
      "step": 25550
    },
    {
      "dynamics_loss_avg": 0.04084340250119567,
      "epoch": 2.4366378784179688e-05,
      "step": 25550
    },
    {
      "action_loss_avg": 0.0015418591676279903,
      "epoch": 2.4366378784179688e-05,
      "step": 25550
    },
    {
      "epoch": 2.4366378784179688e-05,
      "step": 25550,
      "training_step_time": 7.74594783782959
    },
    {
      "epoch": 2.4367332458496092e-05,
      "model_forward_time": 1.737901210784912,
      "step": 25551
    },
    {
      "epoch": 2.4367332458496092e-05,
      "step": 25551,
      "training_step_time": 7.707057952880859
    },
    {
      "epoch": 2.43682861328125e-05,
      "model_forward_time": 1.7325565814971924,
      "step": 25552
    },
    {
      "epoch": 2.43682861328125e-05,
      "step": 25552,
      "training_step_time": 8.75992226600647
    },
    {
      "epoch": 2.4369239807128907e-05,
      "model_forward_time": 1.736215353012085,
      "step": 25553
    },
    {
      "epoch": 2.4369239807128907e-05,
      "step": 25553,
      "training_step_time": 7.722179174423218
    },
    {
      "epoch": 2.437019348144531e-05,
      "model_forward_time": 1.7610056400299072,
      "step": 25554
    },
    {
      "epoch": 2.437019348144531e-05,
      "step": 25554,
      "training_step_time": 8.351954936981201
    },
    {
      "epoch": 2.437114715576172e-05,
      "model_forward_time": 1.738335371017456,
      "step": 25555
    },
    {
      "epoch": 2.437114715576172e-05,
      "step": 25555,
      "training_step_time": 7.793557643890381
    },
    {
      "epoch": 2.4372100830078126e-05,
      "model_forward_time": 1.7350049018859863,
      "step": 25556
    },
    {
      "epoch": 2.4372100830078126e-05,
      "step": 25556,
      "training_step_time": 7.752892017364502
    },
    {
      "epoch": 2.437305450439453e-05,
      "model_forward_time": 1.746490478515625,
      "step": 25557
    },
    {
      "epoch": 2.437305450439453e-05,
      "step": 25557,
      "training_step_time": 8.526808738708496
    },
    {
      "epoch": 2.4374008178710937e-05,
      "model_forward_time": 1.728219747543335,
      "step": 25558
    },
    {
      "epoch": 2.4374008178710937e-05,
      "step": 25558,
      "training_step_time": 7.898995637893677
    },
    {
      "epoch": 2.4374961853027345e-05,
      "model_forward_time": 1.7491803169250488,
      "step": 25559
    },
    {
      "epoch": 2.4374961853027345e-05,
      "step": 25559,
      "training_step_time": 9.05092740058899
    },
    {
      "epoch": 2.437591552734375e-05,
      "grad_norm": 0.19617469608783722,
      "learning_rate": 8.888260583822848e-06,
      "loss": 0.0655,
      "step": 25560
    },
    {
      "epoch": 2.437591552734375e-05,
      "model_forward_time": 1.7393090724945068,
      "step": 25560
    },
    {
      "dynamics_loss_avg": 0.056299262680113316,
      "epoch": 2.437591552734375e-05,
      "step": 25560
    },
    {
      "action_loss_avg": 0.0016713083197828383,
      "epoch": 2.437591552734375e-05,
      "step": 25560
    },
    {
      "epoch": 2.437591552734375e-05,
      "step": 25560,
      "training_step_time": 8.305393934249878
    },
    {
      "epoch": 2.4376869201660156e-05,
      "model_forward_time": 1.74190354347229,
      "step": 25561
    },
    {
      "epoch": 2.4376869201660156e-05,
      "step": 25561,
      "training_step_time": 8.640043497085571
    },
    {
      "epoch": 2.4377822875976564e-05,
      "model_forward_time": 1.7421419620513916,
      "step": 25562
    },
    {
      "epoch": 2.4377822875976564e-05,
      "step": 25562,
      "training_step_time": 7.7603983879089355
    },
    {
      "epoch": 2.4378776550292968e-05,
      "model_forward_time": 1.7604856491088867,
      "step": 25563
    },
    {
      "epoch": 2.4378776550292968e-05,
      "step": 25563,
      "training_step_time": 7.54174280166626
    },
    {
      "epoch": 2.4379730224609375e-05,
      "model_forward_time": 1.7395079135894775,
      "step": 25564
    },
    {
      "epoch": 2.4379730224609375e-05,
      "step": 25564,
      "training_step_time": 7.836963653564453
    },
    {
      "epoch": 2.4380683898925782e-05,
      "model_forward_time": 1.7259695529937744,
      "step": 25565
    },
    {
      "epoch": 2.4380683898925782e-05,
      "step": 25565,
      "training_step_time": 7.632804870605469
    },
    {
      "epoch": 2.4381637573242186e-05,
      "model_forward_time": 1.73408842086792,
      "step": 25566
    },
    {
      "epoch": 2.4381637573242186e-05,
      "step": 25566,
      "training_step_time": 9.308113813400269
    },
    {
      "epoch": 2.4382591247558594e-05,
      "model_forward_time": 1.350785255432129,
      "step": 25567
    },
    {
      "epoch": 2.4382591247558594e-05,
      "step": 25567,
      "training_step_time": 7.9825050830841064
    },
    {
      "epoch": 2.4383544921875e-05,
      "model_forward_time": 1.736966848373413,
      "step": 25568
    },
    {
      "epoch": 2.4383544921875e-05,
      "step": 25568,
      "training_step_time": 9.700790882110596
    },
    {
      "epoch": 2.4384498596191405e-05,
      "model_forward_time": 2.8671305179595947,
      "step": 25569
    },
    {
      "epoch": 2.4384498596191405e-05,
      "step": 25569,
      "training_step_time": 7.688408851623535
    },
    {
      "epoch": 2.4385452270507813e-05,
      "grad_norm": 0.16868185997009277,
      "learning_rate": 8.887220843313922e-06,
      "loss": 0.0621,
      "step": 25570
    },
    {
      "epoch": 2.4385452270507813e-05,
      "model_forward_time": 1.7194440364837646,
      "step": 25570
    },
    {
      "dynamics_loss_avg": 0.058961762115359306,
      "epoch": 2.4385452270507813e-05,
      "step": 25570
    },
    {
      "action_loss_avg": 0.0021697296120692044,
      "epoch": 2.4385452270507813e-05,
      "step": 25570
    },
    {
      "epoch": 2.4385452270507813e-05,
      "step": 25570,
      "training_step_time": 8.224281787872314
    },
    {
      "epoch": 2.438640594482422e-05,
      "model_forward_time": 1.7196831703186035,
      "step": 25571
    },
    {
      "epoch": 2.438640594482422e-05,
      "step": 25571,
      "training_step_time": 7.813435792922974
    },
    {
      "epoch": 2.4387359619140624e-05,
      "model_forward_time": 1.7191803455352783,
      "step": 25572
    },
    {
      "epoch": 2.4387359619140624e-05,
      "step": 25572,
      "training_step_time": 7.659014463424683
    },
    {
      "epoch": 2.438831329345703e-05,
      "model_forward_time": 1.7416071891784668,
      "step": 25573
    },
    {
      "epoch": 2.438831329345703e-05,
      "step": 25573,
      "training_step_time": 7.733792066574097
    },
    {
      "epoch": 2.438926696777344e-05,
      "model_forward_time": 1.7151501178741455,
      "step": 25574
    },
    {
      "epoch": 2.438926696777344e-05,
      "step": 25574,
      "training_step_time": 7.697216510772705
    },
    {
      "epoch": 2.4390220642089843e-05,
      "model_forward_time": 1.7169063091278076,
      "step": 25575
    },
    {
      "epoch": 2.4390220642089843e-05,
      "step": 25575,
      "training_step_time": 7.728978872299194
    },
    {
      "epoch": 2.439117431640625e-05,
      "model_forward_time": 1.7357442378997803,
      "step": 25576
    },
    {
      "epoch": 2.439117431640625e-05,
      "step": 25576,
      "training_step_time": 8.629317283630371
    },
    {
      "epoch": 2.4392127990722658e-05,
      "model_forward_time": 1.734705924987793,
      "step": 25577
    },
    {
      "epoch": 2.4392127990722658e-05,
      "step": 25577,
      "training_step_time": 7.768986940383911
    },
    {
      "epoch": 2.4393081665039062e-05,
      "model_forward_time": 2.8778271675109863,
      "step": 25578
    },
    {
      "epoch": 2.4393081665039062e-05,
      "step": 25578,
      "training_step_time": 8.494247674942017
    },
    {
      "epoch": 2.439403533935547e-05,
      "model_forward_time": 1.7183289527893066,
      "step": 25579
    },
    {
      "epoch": 2.439403533935547e-05,
      "step": 25579,
      "training_step_time": 8.429847002029419
    },
    {
      "epoch": 2.4394989013671873e-05,
      "grad_norm": 0.1702268272638321,
      "learning_rate": 8.88618067770437e-06,
      "loss": 0.0599,
      "step": 25580
    },
    {
      "epoch": 2.4394989013671873e-05,
      "model_forward_time": 1.7105400562286377,
      "step": 25580
    },
    {
      "dynamics_loss_avg": 0.04604191426187754,
      "epoch": 2.4394989013671873e-05,
      "step": 25580
    },
    {
      "action_loss_avg": 0.0019299104693345725,
      "epoch": 2.4394989013671873e-05,
      "step": 25580
    },
    {
      "epoch": 2.4394989013671873e-05,
      "step": 25580,
      "training_step_time": 7.879586935043335
    },
    {
      "epoch": 2.439594268798828e-05,
      "model_forward_time": 1.722463846206665,
      "step": 25581
    },
    {
      "epoch": 2.439594268798828e-05,
      "step": 25581,
      "training_step_time": 7.5947794914245605
    },
    {
      "epoch": 2.4396896362304688e-05,
      "model_forward_time": 1.7362217903137207,
      "step": 25582
    },
    {
      "epoch": 2.4396896362304688e-05,
      "step": 25582,
      "training_step_time": 7.714210033416748
    },
    {
      "epoch": 2.4397850036621092e-05,
      "model_forward_time": 1.7223079204559326,
      "step": 25583
    },
    {
      "epoch": 2.4397850036621092e-05,
      "step": 25583,
      "training_step_time": 8.614463806152344
    },
    {
      "epoch": 2.43988037109375e-05,
      "model_forward_time": 1.7154018878936768,
      "step": 25584
    },
    {
      "epoch": 2.43988037109375e-05,
      "step": 25584,
      "training_step_time": 8.266923666000366
    },
    {
      "epoch": 2.4399757385253907e-05,
      "model_forward_time": 1.7423362731933594,
      "step": 25585
    },
    {
      "epoch": 2.4399757385253907e-05,
      "step": 25585,
      "training_step_time": 8.591617345809937
    },
    {
      "epoch": 2.440071105957031e-05,
      "model_forward_time": 1.7161283493041992,
      "step": 25586
    },
    {
      "epoch": 2.440071105957031e-05,
      "step": 25586,
      "training_step_time": 8.307887077331543
    },
    {
      "epoch": 2.440166473388672e-05,
      "model_forward_time": 2.8357105255126953,
      "step": 25587
    },
    {
      "epoch": 2.440166473388672e-05,
      "step": 25587,
      "training_step_time": 7.691597938537598
    },
    {
      "epoch": 2.4402618408203126e-05,
      "model_forward_time": 1.7107641696929932,
      "step": 25588
    },
    {
      "epoch": 2.4402618408203126e-05,
      "step": 25588,
      "training_step_time": 9.848659753799438
    },
    {
      "epoch": 2.440357208251953e-05,
      "model_forward_time": 1.723996639251709,
      "step": 25589
    },
    {
      "epoch": 2.440357208251953e-05,
      "step": 25589,
      "training_step_time": 7.770958423614502
    },
    {
      "epoch": 2.4404525756835937e-05,
      "grad_norm": 0.12732833623886108,
      "learning_rate": 8.885140087107943e-06,
      "loss": 0.0605,
      "step": 25590
    },
    {
      "epoch": 2.4404525756835937e-05,
      "model_forward_time": 1.3468363285064697,
      "step": 25590
    },
    {
      "dynamics_loss_avg": 0.058871899545192716,
      "epoch": 2.4404525756835937e-05,
      "step": 25590
    },
    {
      "action_loss_avg": 0.002758751908550039,
      "epoch": 2.4404525756835937e-05,
      "step": 25590
    },
    {
      "epoch": 2.4404525756835937e-05,
      "step": 25590,
      "training_step_time": 7.7107579708099365
    },
    {
      "epoch": 2.4405479431152345e-05,
      "model_forward_time": 1.7410447597503662,
      "step": 25591
    },
    {
      "epoch": 2.4405479431152345e-05,
      "step": 25591,
      "training_step_time": 7.791870355606079
    },
    {
      "epoch": 2.440643310546875e-05,
      "model_forward_time": 1.7161943912506104,
      "step": 25592
    },
    {
      "epoch": 2.440643310546875e-05,
      "step": 25592,
      "training_step_time": 8.661532402038574
    },
    {
      "epoch": 2.4407386779785156e-05,
      "model_forward_time": 1.703733205795288,
      "step": 25593
    },
    {
      "epoch": 2.4407386779785156e-05,
      "step": 25593,
      "training_step_time": 7.725428104400635
    },
    {
      "epoch": 2.4408340454101564e-05,
      "model_forward_time": 1.7283897399902344,
      "step": 25594
    },
    {
      "epoch": 2.4408340454101564e-05,
      "step": 25594,
      "training_step_time": 7.940983295440674
    },
    {
      "epoch": 2.4409294128417968e-05,
      "model_forward_time": 1.3454809188842773,
      "step": 25595
    },
    {
      "epoch": 2.4409294128417968e-05,
      "step": 25595,
      "training_step_time": 7.917421102523804
    },
    {
      "epoch": 2.4410247802734375e-05,
      "model_forward_time": 2.8632614612579346,
      "step": 25596
    },
    {
      "epoch": 2.4410247802734375e-05,
      "step": 25596,
      "training_step_time": 8.728975057601929
    },
    {
      "epoch": 2.4411201477050782e-05,
      "model_forward_time": 1.715808629989624,
      "step": 25597
    },
    {
      "epoch": 2.4411201477050782e-05,
      "step": 25597,
      "training_step_time": 7.887984037399292
    },
    {
      "epoch": 2.4412155151367187e-05,
      "model_forward_time": 1.720201015472412,
      "step": 25598
    },
    {
      "epoch": 2.4412155151367187e-05,
      "step": 25598,
      "training_step_time": 8.848083019256592
    },
    {
      "epoch": 2.4413108825683594e-05,
      "model_forward_time": 1.7201581001281738,
      "step": 25599
    },
    {
      "epoch": 2.4413108825683594e-05,
      "step": 25599,
      "training_step_time": 7.739179849624634
    },
    {
      "epoch": 2.44140625e-05,
      "grad_norm": 0.22895580530166626,
      "learning_rate": 8.884099071638437e-06,
      "loss": 0.0664,
      "step": 25600
    },
    {
      "epoch": 2.44140625e-05,
      "model_forward_time": 1.7233753204345703,
      "step": 25600
    },
    {
      "dynamics_loss_avg": 0.06420712247490883,
      "epoch": 2.44140625e-05,
      "step": 25600
    },
    {
      "action_loss_avg": 0.0023642044165171684,
      "epoch": 2.44140625e-05,
      "step": 25600
    },
    {
      "epoch": 2.44140625e-05,
      "step": 25600,
      "training_step_time": 7.645071983337402
    },
    {
      "epoch": 2.4415016174316405e-05,
      "model_forward_time": 1.7209150791168213,
      "step": 25601
    },
    {
      "epoch": 2.4415016174316405e-05,
      "step": 25601,
      "training_step_time": 7.637786149978638
    },
    {
      "epoch": 2.4415969848632813e-05,
      "model_forward_time": 1.7269878387451172,
      "step": 25602
    },
    {
      "epoch": 2.4415969848632813e-05,
      "step": 25602,
      "training_step_time": 7.816159248352051
    },
    {
      "epoch": 2.441692352294922e-05,
      "model_forward_time": 1.3678336143493652,
      "step": 25603
    },
    {
      "epoch": 2.441692352294922e-05,
      "step": 25603,
      "training_step_time": 7.968015909194946
    },
    {
      "epoch": 2.4417877197265624e-05,
      "model_forward_time": 1.7191486358642578,
      "step": 25604
    },
    {
      "epoch": 2.4417877197265624e-05,
      "step": 25604,
      "training_step_time": 8.486224174499512
    },
    {
      "epoch": 2.441883087158203e-05,
      "model_forward_time": 2.8395984172821045,
      "step": 25605
    },
    {
      "epoch": 2.441883087158203e-05,
      "step": 25605,
      "training_step_time": 8.500874996185303
    },
    {
      "epoch": 2.441978454589844e-05,
      "model_forward_time": 1.7386119365692139,
      "step": 25606
    },
    {
      "epoch": 2.441978454589844e-05,
      "step": 25606,
      "training_step_time": 7.869839191436768
    },
    {
      "epoch": 2.4420738220214843e-05,
      "model_forward_time": 1.733726978302002,
      "step": 25607
    },
    {
      "epoch": 2.4420738220214843e-05,
      "step": 25607,
      "training_step_time": 9.594560623168945
    },
    {
      "epoch": 2.442169189453125e-05,
      "model_forward_time": 1.7289893627166748,
      "step": 25608
    },
    {
      "epoch": 2.442169189453125e-05,
      "step": 25608,
      "training_step_time": 7.704766035079956
    },
    {
      "epoch": 2.4422645568847658e-05,
      "model_forward_time": 1.7141265869140625,
      "step": 25609
    },
    {
      "epoch": 2.4422645568847658e-05,
      "step": 25609,
      "training_step_time": 8.57969355583191
    },
    {
      "epoch": 2.4423599243164062e-05,
      "grad_norm": 0.28213605284690857,
      "learning_rate": 8.883057631409694e-06,
      "loss": 0.067,
      "step": 25610
    },
    {
      "epoch": 2.4423599243164062e-05,
      "model_forward_time": 1.7534925937652588,
      "step": 25610
    },
    {
      "dynamics_loss_avg": 0.04906334765255451,
      "epoch": 2.4423599243164062e-05,
      "step": 25610
    },
    {
      "action_loss_avg": 0.0019217247201595456,
      "epoch": 2.4423599243164062e-05,
      "step": 25610
    },
    {
      "epoch": 2.4423599243164062e-05,
      "step": 25610,
      "training_step_time": 7.614060163497925
    },
    {
      "epoch": 2.442455291748047e-05,
      "model_forward_time": 1.0344972610473633,
      "step": 25611
    },
    {
      "epoch": 2.442455291748047e-05,
      "step": 25611,
      "training_step_time": 7.751773357391357
    },
    {
      "epoch": 2.4425506591796873e-05,
      "model_forward_time": 1.7249937057495117,
      "step": 25612
    },
    {
      "epoch": 2.4425506591796873e-05,
      "step": 25612,
      "training_step_time": 8.692385911941528
    },
    {
      "epoch": 2.442646026611328e-05,
      "model_forward_time": 1.727553367614746,
      "step": 25613
    },
    {
      "epoch": 2.442646026611328e-05,
      "step": 25613,
      "training_step_time": 8.022770643234253
    },
    {
      "epoch": 2.4427413940429688e-05,
      "model_forward_time": 1.738034963607788,
      "step": 25614
    },
    {
      "epoch": 2.4427413940429688e-05,
      "step": 25614,
      "training_step_time": 7.755049467086792
    },
    {
      "epoch": 2.4428367614746092e-05,
      "model_forward_time": 2.881316661834717,
      "step": 25615
    },
    {
      "epoch": 2.4428367614746092e-05,
      "step": 25615,
      "training_step_time": 7.781064987182617
    },
    {
      "epoch": 2.44293212890625e-05,
      "model_forward_time": 1.7332885265350342,
      "step": 25616
    },
    {
      "epoch": 2.44293212890625e-05,
      "step": 25616,
      "training_step_time": 9.367072343826294
    },
    {
      "epoch": 2.4430274963378907e-05,
      "model_forward_time": 1.7358078956604004,
      "step": 25617
    },
    {
      "epoch": 2.4430274963378907e-05,
      "step": 25617,
      "training_step_time": 7.695634603500366
    },
    {
      "epoch": 2.443122863769531e-05,
      "model_forward_time": 1.7180323600769043,
      "step": 25618
    },
    {
      "epoch": 2.443122863769531e-05,
      "step": 25618,
      "training_step_time": 8.717585802078247
    },
    {
      "epoch": 2.443218231201172e-05,
      "model_forward_time": 1.7430386543273926,
      "step": 25619
    },
    {
      "epoch": 2.443218231201172e-05,
      "step": 25619,
      "training_step_time": 7.64274525642395
    },
    {
      "epoch": 2.4433135986328126e-05,
      "grad_norm": 0.21573658287525177,
      "learning_rate": 8.882015766535608e-06,
      "loss": 0.0672,
      "step": 25620
    },
    {
      "epoch": 2.4433135986328126e-05,
      "model_forward_time": 1.7273507118225098,
      "step": 25620
    },
    {
      "dynamics_loss_avg": 0.046457532420754434,
      "epoch": 2.4433135986328126e-05,
      "step": 25620
    },
    {
      "action_loss_avg": 0.001479630611720495,
      "epoch": 2.4433135986328126e-05,
      "step": 25620
    },
    {
      "epoch": 2.4433135986328126e-05,
      "step": 25620,
      "training_step_time": 7.735836744308472
    },
    {
      "epoch": 2.443408966064453e-05,
      "model_forward_time": 1.725433111190796,
      "step": 25621
    },
    {
      "epoch": 2.443408966064453e-05,
      "step": 25621,
      "training_step_time": 9.14200735092163
    },
    {
      "epoch": 2.4435043334960937e-05,
      "model_forward_time": 1.7649385929107666,
      "step": 25622
    },
    {
      "epoch": 2.4435043334960937e-05,
      "step": 25622,
      "training_step_time": 9.045557737350464
    },
    {
      "epoch": 2.4435997009277345e-05,
      "model_forward_time": 1.7183504104614258,
      "step": 25623
    },
    {
      "epoch": 2.4435997009277345e-05,
      "step": 25623,
      "training_step_time": 8.768600702285767
    },
    {
      "epoch": 2.443695068359375e-05,
      "model_forward_time": 2.8713057041168213,
      "step": 25624
    },
    {
      "epoch": 2.443695068359375e-05,
      "step": 25624,
      "training_step_time": 8.733680725097656
    },
    {
      "epoch": 2.4437904357910156e-05,
      "model_forward_time": 1.363738775253296,
      "step": 25625
    },
    {
      "epoch": 2.4437904357910156e-05,
      "step": 25625,
      "training_step_time": 7.8669984340667725
    },
    {
      "epoch": 2.4438858032226564e-05,
      "model_forward_time": 1.7263388633728027,
      "step": 25626
    },
    {
      "epoch": 2.4438858032226564e-05,
      "step": 25626,
      "training_step_time": 8.866094589233398
    },
    {
      "epoch": 2.4439811706542968e-05,
      "model_forward_time": 1.7298872470855713,
      "step": 25627
    },
    {
      "epoch": 2.4439811706542968e-05,
      "step": 25627,
      "training_step_time": 7.8581743240356445
    },
    {
      "epoch": 2.4440765380859375e-05,
      "model_forward_time": 1.7407128810882568,
      "step": 25628
    },
    {
      "epoch": 2.4440765380859375e-05,
      "step": 25628,
      "training_step_time": 7.661487340927124
    },
    {
      "epoch": 2.4441719055175783e-05,
      "model_forward_time": 1.729647159576416,
      "step": 25629
    },
    {
      "epoch": 2.4441719055175783e-05,
      "step": 25629,
      "training_step_time": 8.790141820907593
    },
    {
      "epoch": 2.4442672729492187e-05,
      "grad_norm": 0.23290245234966278,
      "learning_rate": 8.880973477130115e-06,
      "loss": 0.0681,
      "step": 25630
    },
    {
      "epoch": 2.4442672729492187e-05,
      "model_forward_time": 1.717829704284668,
      "step": 25630
    },
    {
      "dynamics_loss_avg": 0.07424698136746884,
      "epoch": 2.4442672729492187e-05,
      "step": 25630
    },
    {
      "action_loss_avg": 0.003275883535388857,
      "epoch": 2.4442672729492187e-05,
      "step": 25630
    },
    {
      "epoch": 2.4442672729492187e-05,
      "step": 25630,
      "training_step_time": 7.647894620895386
    },
    {
      "epoch": 2.4443626403808594e-05,
      "model_forward_time": 1.750532627105713,
      "step": 25631
    },
    {
      "epoch": 2.4443626403808594e-05,
      "step": 25631,
      "training_step_time": 9.098331212997437
    },
    {
      "epoch": 2.4444580078125e-05,
      "model_forward_time": 1.3549704551696777,
      "step": 25632
    },
    {
      "epoch": 2.4444580078125e-05,
      "step": 25632,
      "training_step_time": 7.716842889785767
    },
    {
      "epoch": 2.4445533752441405e-05,
      "model_forward_time": 1.7280659675598145,
      "step": 25633
    },
    {
      "epoch": 2.4445533752441405e-05,
      "step": 25633,
      "training_step_time": 8.276237726211548
    },
    {
      "epoch": 2.4446487426757813e-05,
      "model_forward_time": 1.7243404388427734,
      "step": 25634
    },
    {
      "epoch": 2.4446487426757813e-05,
      "step": 25634,
      "training_step_time": 8.455683708190918
    },
    {
      "epoch": 2.444744110107422e-05,
      "model_forward_time": 1.723400592803955,
      "step": 25635
    },
    {
      "epoch": 2.444744110107422e-05,
      "step": 25635,
      "training_step_time": 7.847935438156128
    },
    {
      "epoch": 2.4448394775390624e-05,
      "model_forward_time": 1.7273454666137695,
      "step": 25636
    },
    {
      "epoch": 2.4448394775390624e-05,
      "step": 25636,
      "training_step_time": 7.8199121952056885
    },
    {
      "epoch": 2.444934844970703e-05,
      "model_forward_time": 1.7471675872802734,
      "step": 25637
    },
    {
      "epoch": 2.444934844970703e-05,
      "step": 25637,
      "training_step_time": 7.685986042022705
    },
    {
      "epoch": 2.445030212402344e-05,
      "model_forward_time": 1.7277731895446777,
      "step": 25638
    },
    {
      "epoch": 2.445030212402344e-05,
      "step": 25638,
      "training_step_time": 8.608938455581665
    },
    {
      "epoch": 2.4451255798339843e-05,
      "model_forward_time": 1.713597297668457,
      "step": 25639
    },
    {
      "epoch": 2.4451255798339843e-05,
      "step": 25639,
      "training_step_time": 7.682787656784058
    },
    {
      "epoch": 2.445220947265625e-05,
      "grad_norm": 0.1675204634666443,
      "learning_rate": 8.879930763307198e-06,
      "loss": 0.0637,
      "step": 25640
    },
    {
      "epoch": 2.445220947265625e-05,
      "model_forward_time": 1.7503228187561035,
      "step": 25640
    },
    {
      "dynamics_loss_avg": 0.0468309722840786,
      "epoch": 2.445220947265625e-05,
      "step": 25640
    },
    {
      "action_loss_avg": 0.0021023155364673584,
      "epoch": 2.445220947265625e-05,
      "step": 25640
    },
    {
      "epoch": 2.445220947265625e-05,
      "step": 25640,
      "training_step_time": 8.829605102539062
    },
    {
      "epoch": 2.4453163146972655e-05,
      "model_forward_time": 1.730027198791504,
      "step": 25641
    },
    {
      "epoch": 2.4453163146972655e-05,
      "step": 25641,
      "training_step_time": 7.786300897598267
    },
    {
      "epoch": 2.4454116821289062e-05,
      "model_forward_time": 1.712101936340332,
      "step": 25642
    },
    {
      "epoch": 2.4454116821289062e-05,
      "step": 25642,
      "training_step_time": 7.785626649856567
    },
    {
      "epoch": 2.445507049560547e-05,
      "model_forward_time": 2.2762537002563477,
      "step": 25643
    },
    {
      "epoch": 2.445507049560547e-05,
      "step": 25643,
      "training_step_time": 7.784818172454834
    },
    {
      "epoch": 2.4456024169921873e-05,
      "model_forward_time": 1.720388650894165,
      "step": 25644
    },
    {
      "epoch": 2.4456024169921873e-05,
      "step": 25644,
      "training_step_time": 7.760110139846802
    },
    {
      "epoch": 2.445697784423828e-05,
      "model_forward_time": 1.715524435043335,
      "step": 25645
    },
    {
      "epoch": 2.445697784423828e-05,
      "step": 25645,
      "training_step_time": 7.749375343322754
    },
    {
      "epoch": 2.4457931518554688e-05,
      "model_forward_time": 1.7182667255401611,
      "step": 25646
    },
    {
      "epoch": 2.4457931518554688e-05,
      "step": 25646,
      "training_step_time": 8.408458948135376
    },
    {
      "epoch": 2.4458885192871092e-05,
      "model_forward_time": 1.7236888408660889,
      "step": 25647
    },
    {
      "epoch": 2.4458885192871092e-05,
      "step": 25647,
      "training_step_time": 7.779034376144409
    },
    {
      "epoch": 2.44598388671875e-05,
      "model_forward_time": 1.7159714698791504,
      "step": 25648
    },
    {
      "epoch": 2.44598388671875e-05,
      "step": 25648,
      "training_step_time": 7.817633152008057
    },
    {
      "epoch": 2.4460792541503907e-05,
      "model_forward_time": 1.7444121837615967,
      "step": 25649
    },
    {
      "epoch": 2.4460792541503907e-05,
      "step": 25649,
      "training_step_time": 7.824605941772461
    },
    {
      "epoch": 2.446174621582031e-05,
      "grad_norm": 0.13991862535476685,
      "learning_rate": 8.878887625180885e-06,
      "loss": 0.0599,
      "step": 25650
    },
    {
      "epoch": 2.446174621582031e-05,
      "model_forward_time": 1.7504205703735352,
      "step": 25650
    },
    {
      "dynamics_loss_avg": 0.059225056506693366,
      "epoch": 2.446174621582031e-05,
      "step": 25650
    },
    {
      "action_loss_avg": 0.0026567810331471265,
      "epoch": 2.446174621582031e-05,
      "step": 25650
    },
    {
      "epoch": 2.446174621582031e-05,
      "step": 25650,
      "training_step_time": 7.814164400100708
    },
    {
      "epoch": 2.446269989013672e-05,
      "model_forward_time": 1.7176475524902344,
      "step": 25651
    },
    {
      "epoch": 2.446269989013672e-05,
      "step": 25651,
      "training_step_time": 7.913281679153442
    },
    {
      "epoch": 2.4463653564453126e-05,
      "model_forward_time": 2.25896954536438,
      "step": 25652
    },
    {
      "epoch": 2.4463653564453126e-05,
      "step": 25652,
      "training_step_time": 7.898262977600098
    },
    {
      "epoch": 2.446460723876953e-05,
      "model_forward_time": 1.7153093814849854,
      "step": 25653
    },
    {
      "epoch": 2.446460723876953e-05,
      "step": 25653,
      "training_step_time": 7.893101692199707
    },
    {
      "epoch": 2.4465560913085937e-05,
      "model_forward_time": 1.724222183227539,
      "step": 25654
    },
    {
      "epoch": 2.4465560913085937e-05,
      "step": 25654,
      "training_step_time": 7.839278221130371
    },
    {
      "epoch": 2.4466514587402345e-05,
      "model_forward_time": 1.7163994312286377,
      "step": 25655
    },
    {
      "epoch": 2.4466514587402345e-05,
      "step": 25655,
      "training_step_time": 7.685408115386963
    },
    {
      "epoch": 2.446746826171875e-05,
      "model_forward_time": 1.7188289165496826,
      "step": 25656
    },
    {
      "epoch": 2.446746826171875e-05,
      "step": 25656,
      "training_step_time": 8.788217544555664
    },
    {
      "epoch": 2.4468421936035156e-05,
      "model_forward_time": 1.7204880714416504,
      "step": 25657
    },
    {
      "epoch": 2.4468421936035156e-05,
      "step": 25657,
      "training_step_time": 7.799779653549194
    },
    {
      "epoch": 2.4469375610351564e-05,
      "model_forward_time": 1.7231674194335938,
      "step": 25658
    },
    {
      "epoch": 2.4469375610351564e-05,
      "step": 25658,
      "training_step_time": 8.046467781066895
    },
    {
      "epoch": 2.4470329284667968e-05,
      "model_forward_time": 1.7341060638427734,
      "step": 25659
    },
    {
      "epoch": 2.4470329284667968e-05,
      "step": 25659,
      "training_step_time": 7.788680553436279
    },
    {
      "epoch": 2.4471282958984375e-05,
      "grad_norm": 0.14637137949466705,
      "learning_rate": 8.877844062865253e-06,
      "loss": 0.0623,
      "step": 25660
    },
    {
      "epoch": 2.4471282958984375e-05,
      "model_forward_time": 1.7162117958068848,
      "step": 25660
    },
    {
      "dynamics_loss_avg": 0.0653272426687181,
      "epoch": 2.4471282958984375e-05,
      "step": 25660
    },
    {
      "action_loss_avg": 0.0021351155621232463,
      "epoch": 2.4471282958984375e-05,
      "step": 25660
    },
    {
      "epoch": 2.4471282958984375e-05,
      "step": 25660,
      "training_step_time": 7.761980295181274
    },
    {
      "epoch": 2.4472236633300783e-05,
      "model_forward_time": 2.2628395557403564,
      "step": 25661
    },
    {
      "epoch": 2.4472236633300783e-05,
      "step": 25661,
      "training_step_time": 7.901363372802734
    },
    {
      "epoch": 2.4473190307617187e-05,
      "model_forward_time": 1.7221863269805908,
      "step": 25662
    },
    {
      "epoch": 2.4473190307617187e-05,
      "step": 25662,
      "training_step_time": 8.966190814971924
    },
    {
      "epoch": 2.4474143981933594e-05,
      "model_forward_time": 1.716404914855957,
      "step": 25663
    },
    {
      "epoch": 2.4474143981933594e-05,
      "step": 25663,
      "training_step_time": 8.81674575805664
    },
    {
      "epoch": 2.447509765625e-05,
      "model_forward_time": 1.3434956073760986,
      "step": 25664
    },
    {
      "epoch": 2.447509765625e-05,
      "step": 25664,
      "training_step_time": 8.236716270446777
    },
    {
      "epoch": 2.4476051330566405e-05,
      "model_forward_time": 1.7154581546783447,
      "step": 25665
    },
    {
      "epoch": 2.4476051330566405e-05,
      "step": 25665,
      "training_step_time": 7.771586179733276
    },
    {
      "epoch": 2.4477005004882813e-05,
      "model_forward_time": 1.717489242553711,
      "step": 25666
    },
    {
      "epoch": 2.4477005004882813e-05,
      "step": 25666,
      "training_step_time": 7.881361961364746
    },
    {
      "epoch": 2.447795867919922e-05,
      "model_forward_time": 1.7210946083068848,
      "step": 25667
    },
    {
      "epoch": 2.447795867919922e-05,
      "step": 25667,
      "training_step_time": 7.912087917327881
    },
    {
      "epoch": 2.4478912353515624e-05,
      "model_forward_time": 1.7211945056915283,
      "step": 25668
    },
    {
      "epoch": 2.4478912353515624e-05,
      "step": 25668,
      "training_step_time": 8.474773406982422
    },
    {
      "epoch": 2.4479866027832032e-05,
      "model_forward_time": 1.717071533203125,
      "step": 25669
    },
    {
      "epoch": 2.4479866027832032e-05,
      "step": 25669,
      "training_step_time": 7.827106714248657
    },
    {
      "epoch": 2.448081970214844e-05,
      "grad_norm": 0.17799122631549835,
      "learning_rate": 8.876800076474425e-06,
      "loss": 0.0648,
      "step": 25670
    },
    {
      "epoch": 2.448081970214844e-05,
      "model_forward_time": 2.2805230617523193,
      "step": 25670
    },
    {
      "dynamics_loss_avg": 0.06260746102780104,
      "epoch": 2.448081970214844e-05,
      "step": 25670
    },
    {
      "action_loss_avg": 0.0028227363945916295,
      "epoch": 2.448081970214844e-05,
      "step": 25670
    },
    {
      "epoch": 2.448081970214844e-05,
      "step": 25670,
      "training_step_time": 7.756454229354858
    },
    {
      "epoch": 2.4481773376464843e-05,
      "model_forward_time": 1.7161979675292969,
      "step": 25671
    },
    {
      "epoch": 2.4481773376464843e-05,
      "step": 25671,
      "training_step_time": 8.400238990783691
    },
    {
      "epoch": 2.448272705078125e-05,
      "model_forward_time": 1.7262051105499268,
      "step": 25672
    },
    {
      "epoch": 2.448272705078125e-05,
      "step": 25672,
      "training_step_time": 8.379959344863892
    },
    {
      "epoch": 2.4483680725097655e-05,
      "model_forward_time": 1.72023606300354,
      "step": 25673
    },
    {
      "epoch": 2.4483680725097655e-05,
      "step": 25673,
      "training_step_time": 7.846040487289429
    },
    {
      "epoch": 2.4484634399414062e-05,
      "model_forward_time": 1.7116978168487549,
      "step": 25674
    },
    {
      "epoch": 2.4484634399414062e-05,
      "step": 25674,
      "training_step_time": 7.867973566055298
    },
    {
      "epoch": 2.448558807373047e-05,
      "model_forward_time": 1.7114861011505127,
      "step": 25675
    },
    {
      "epoch": 2.448558807373047e-05,
      "step": 25675,
      "training_step_time": 7.761461019515991
    },
    {
      "epoch": 2.4486541748046873e-05,
      "model_forward_time": 1.7263457775115967,
      "step": 25676
    },
    {
      "epoch": 2.4486541748046873e-05,
      "step": 25676,
      "training_step_time": 7.735141754150391
    },
    {
      "epoch": 2.448749542236328e-05,
      "model_forward_time": 1.712805986404419,
      "step": 25677
    },
    {
      "epoch": 2.448749542236328e-05,
      "step": 25677,
      "training_step_time": 8.58286738395691
    },
    {
      "epoch": 2.4488449096679688e-05,
      "model_forward_time": 1.758216381072998,
      "step": 25678
    },
    {
      "epoch": 2.4488449096679688e-05,
      "step": 25678,
      "training_step_time": 8.196125030517578
    },
    {
      "epoch": 2.4489402770996092e-05,
      "model_forward_time": 2.2583608627319336,
      "step": 25679
    },
    {
      "epoch": 2.4489402770996092e-05,
      "step": 25679,
      "training_step_time": 8.55666184425354
    },
    {
      "epoch": 2.44903564453125e-05,
      "grad_norm": 0.15905839204788208,
      "learning_rate": 8.875755666122567e-06,
      "loss": 0.0603,
      "step": 25680
    },
    {
      "epoch": 2.44903564453125e-05,
      "model_forward_time": 1.7251229286193848,
      "step": 25680
    },
    {
      "dynamics_loss_avg": 0.06141958553344011,
      "epoch": 2.44903564453125e-05,
      "step": 25680
    },
    {
      "action_loss_avg": 0.0017081223719287663,
      "epoch": 2.44903564453125e-05,
      "step": 25680
    },
    {
      "epoch": 2.44903564453125e-05,
      "step": 25680,
      "training_step_time": 7.799693584442139
    },
    {
      "epoch": 2.4491310119628907e-05,
      "model_forward_time": 1.730471134185791,
      "step": 25681
    },
    {
      "epoch": 2.4491310119628907e-05,
      "step": 25681,
      "training_step_time": 7.770204782485962
    },
    {
      "epoch": 2.449226379394531e-05,
      "model_forward_time": 1.7367055416107178,
      "step": 25682
    },
    {
      "epoch": 2.449226379394531e-05,
      "step": 25682,
      "training_step_time": 7.794127464294434
    },
    {
      "epoch": 2.449321746826172e-05,
      "model_forward_time": 1.7232441902160645,
      "step": 25683
    },
    {
      "epoch": 2.449321746826172e-05,
      "step": 25683,
      "training_step_time": 7.695091247558594
    },
    {
      "epoch": 2.4494171142578126e-05,
      "model_forward_time": 1.7144029140472412,
      "step": 25684
    },
    {
      "epoch": 2.4494171142578126e-05,
      "step": 25684,
      "training_step_time": 8.74656343460083
    },
    {
      "epoch": 2.449512481689453e-05,
      "model_forward_time": 1.7181627750396729,
      "step": 25685
    },
    {
      "epoch": 2.449512481689453e-05,
      "step": 25685,
      "training_step_time": 7.651429176330566
    },
    {
      "epoch": 2.4496078491210937e-05,
      "model_forward_time": 1.7205827236175537,
      "step": 25686
    },
    {
      "epoch": 2.4496078491210937e-05,
      "step": 25686,
      "training_step_time": 8.815546751022339
    },
    {
      "epoch": 2.4497032165527345e-05,
      "model_forward_time": 1.773162841796875,
      "step": 25687
    },
    {
      "epoch": 2.4497032165527345e-05,
      "step": 25687,
      "training_step_time": 7.6860644817352295
    },
    {
      "epoch": 2.449798583984375e-05,
      "model_forward_time": 1.719006061553955,
      "step": 25688
    },
    {
      "epoch": 2.449798583984375e-05,
      "step": 25688,
      "training_step_time": 7.618317365646362
    },
    {
      "epoch": 2.4498939514160156e-05,
      "model_forward_time": 2.252562999725342,
      "step": 25689
    },
    {
      "epoch": 2.4498939514160156e-05,
      "step": 25689,
      "training_step_time": 8.381953954696655
    },
    {
      "epoch": 2.4499893188476564e-05,
      "grad_norm": 0.18565264344215393,
      "learning_rate": 8.874710831923901e-06,
      "loss": 0.0639,
      "step": 25690
    },
    {
      "epoch": 2.4499893188476564e-05,
      "model_forward_time": 1.7179207801818848,
      "step": 25690
    },
    {
      "dynamics_loss_avg": 0.05605225041508675,
      "epoch": 2.4499893188476564e-05,
      "step": 25690
    },
    {
      "action_loss_avg": 0.002308748132782057,
      "epoch": 2.4499893188476564e-05,
      "step": 25690
    },
    {
      "epoch": 2.4499893188476564e-05,
      "step": 25690,
      "training_step_time": 7.737638235092163
    },
    {
      "epoch": 2.4500846862792968e-05,
      "model_forward_time": 1.7142727375030518,
      "step": 25691
    },
    {
      "epoch": 2.4500846862792968e-05,
      "step": 25691,
      "training_step_time": 9.492133140563965
    },
    {
      "epoch": 2.4501800537109375e-05,
      "model_forward_time": 1.7587647438049316,
      "step": 25692
    },
    {
      "epoch": 2.4501800537109375e-05,
      "step": 25692,
      "training_step_time": 8.240363597869873
    },
    {
      "epoch": 2.4502754211425783e-05,
      "model_forward_time": 1.7179369926452637,
      "step": 25693
    },
    {
      "epoch": 2.4502754211425783e-05,
      "step": 25693,
      "training_step_time": 6.752256631851196
    },
    {
      "epoch": 2.4503707885742187e-05,
      "model_forward_time": 1.7210688591003418,
      "step": 25694
    },
    {
      "epoch": 2.4503707885742187e-05,
      "step": 25694,
      "training_step_time": 6.803426027297974
    },
    {
      "epoch": 2.4504661560058594e-05,
      "model_forward_time": 1.7202844619750977,
      "step": 25695
    },
    {
      "epoch": 2.4504661560058594e-05,
      "step": 25695,
      "training_step_time": 6.788939952850342
    },
    {
      "epoch": 2.4505615234375e-05,
      "model_forward_time": 1.7208178043365479,
      "step": 25696
    },
    {
      "epoch": 2.4505615234375e-05,
      "step": 25696,
      "training_step_time": 6.76862645149231
    },
    {
      "epoch": 2.4506568908691405e-05,
      "model_forward_time": 1.3761680126190186,
      "step": 25697
    },
    {
      "epoch": 2.4506568908691405e-05,
      "step": 25697,
      "training_step_time": 6.775734901428223
    },
    {
      "epoch": 2.4507522583007813e-05,
      "model_forward_time": 1.7229056358337402,
      "step": 25698
    },
    {
      "epoch": 2.4507522583007813e-05,
      "step": 25698,
      "training_step_time": 6.78563117980957
    },
    {
      "epoch": 2.450847625732422e-05,
      "model_forward_time": 1.7189311981201172,
      "step": 25699
    },
    {
      "epoch": 2.450847625732422e-05,
      "step": 25699,
      "training_step_time": 6.837074279785156
    },
    {
      "epoch": 2.4509429931640624e-05,
      "grad_norm": 0.1349741369485855,
      "learning_rate": 8.87366557399268e-06,
      "loss": 0.062,
      "step": 25700
    },
    {
      "epoch": 2.4509429931640624e-05,
      "model_forward_time": 1.7147951126098633,
      "step": 25700
    },
    {
      "dynamics_loss_avg": 0.05279802149161696,
      "epoch": 2.4509429931640624e-05,
      "step": 25700
    },
    {
      "action_loss_avg": 0.0014675104524940253,
      "epoch": 2.4509429931640624e-05,
      "step": 25700
    },
    {
      "epoch": 2.4509429931640624e-05,
      "step": 25700,
      "training_step_time": 7.749894142150879
    },
    {
      "epoch": 2.4510383605957032e-05,
      "model_forward_time": 2.275446653366089,
      "step": 25701
    },
    {
      "epoch": 2.4510383605957032e-05,
      "step": 25701,
      "training_step_time": 8.664223909378052
    },
    {
      "epoch": 2.451133728027344e-05,
      "model_forward_time": 1.7161586284637451,
      "step": 25702
    },
    {
      "epoch": 2.451133728027344e-05,
      "step": 25702,
      "training_step_time": 7.786275386810303
    },
    {
      "epoch": 2.4512290954589843e-05,
      "model_forward_time": 1.7180626392364502,
      "step": 25703
    },
    {
      "epoch": 2.4512290954589843e-05,
      "step": 25703,
      "training_step_time": 6.76424765586853
    },
    {
      "epoch": 2.451324462890625e-05,
      "model_forward_time": 1.7252843379974365,
      "step": 25704
    },
    {
      "epoch": 2.451324462890625e-05,
      "step": 25704,
      "training_step_time": 6.764650344848633
    },
    {
      "epoch": 2.4514198303222655e-05,
      "model_forward_time": 1.7231175899505615,
      "step": 25705
    },
    {
      "epoch": 2.4514198303222655e-05,
      "step": 25705,
      "training_step_time": 6.775392055511475
    },
    {
      "epoch": 2.4515151977539062e-05,
      "model_forward_time": 1.7188663482666016,
      "step": 25706
    },
    {
      "epoch": 2.4515151977539062e-05,
      "step": 25706,
      "training_step_time": 6.743883371353149
    },
    {
      "epoch": 2.451610565185547e-05,
      "model_forward_time": 1.7173724174499512,
      "step": 25707
    },
    {
      "epoch": 2.451610565185547e-05,
      "step": 25707,
      "training_step_time": 6.771028995513916
    },
    {
      "epoch": 2.4517059326171873e-05,
      "model_forward_time": 1.7061858177185059,
      "step": 25708
    },
    {
      "epoch": 2.4517059326171873e-05,
      "step": 25708,
      "training_step_time": 6.7865753173828125
    },
    {
      "epoch": 2.451801300048828e-05,
      "model_forward_time": 1.7107501029968262,
      "step": 25709
    },
    {
      "epoch": 2.451801300048828e-05,
      "step": 25709,
      "training_step_time": 6.7595374584198
    },
    {
      "epoch": 2.451896667480469e-05,
      "grad_norm": 0.19687825441360474,
      "learning_rate": 8.872619892443218e-06,
      "loss": 0.0604,
      "step": 25710
    },
    {
      "epoch": 2.451896667480469e-05,
      "model_forward_time": 2.2899115085601807,
      "step": 25710
    },
    {
      "dynamics_loss_avg": 0.05861461628228426,
      "epoch": 2.451896667480469e-05,
      "step": 25710
    },
    {
      "action_loss_avg": 0.0018832873029168695,
      "epoch": 2.451896667480469e-05,
      "step": 25710
    },
    {
      "epoch": 2.451896667480469e-05,
      "step": 25710,
      "training_step_time": 7.490076780319214
    },
    {
      "epoch": 2.4519920349121092e-05,
      "model_forward_time": 1.7261924743652344,
      "step": 25711
    },
    {
      "epoch": 2.4519920349121092e-05,
      "step": 25711,
      "training_step_time": 7.778635025024414
    },
    {
      "epoch": 2.45208740234375e-05,
      "model_forward_time": 1.724585771560669,
      "step": 25712
    },
    {
      "epoch": 2.45208740234375e-05,
      "step": 25712,
      "training_step_time": 7.296208381652832
    },
    {
      "epoch": 2.4521827697753907e-05,
      "model_forward_time": 1.7193574905395508,
      "step": 25713
    },
    {
      "epoch": 2.4521827697753907e-05,
      "step": 25713,
      "training_step_time": 6.794344663619995
    },
    {
      "epoch": 2.452278137207031e-05,
      "model_forward_time": 1.7265403270721436,
      "step": 25714
    },
    {
      "epoch": 2.452278137207031e-05,
      "step": 25714,
      "training_step_time": 6.789820909500122
    },
    {
      "epoch": 2.452373504638672e-05,
      "model_forward_time": 1.7214791774749756,
      "step": 25715
    },
    {
      "epoch": 2.452373504638672e-05,
      "step": 25715,
      "training_step_time": 6.800105810165405
    },
    {
      "epoch": 2.4524688720703126e-05,
      "model_forward_time": 1.7194240093231201,
      "step": 25716
    },
    {
      "epoch": 2.4524688720703126e-05,
      "step": 25716,
      "training_step_time": 6.800654888153076
    },
    {
      "epoch": 2.452564239501953e-05,
      "model_forward_time": 1.7277615070343018,
      "step": 25717
    },
    {
      "epoch": 2.452564239501953e-05,
      "step": 25717,
      "training_step_time": 6.756520748138428
    },
    {
      "epoch": 2.4526596069335937e-05,
      "model_forward_time": 1.7213022708892822,
      "step": 25718
    },
    {
      "epoch": 2.4526596069335937e-05,
      "step": 25718,
      "training_step_time": 6.827120542526245
    },
    {
      "epoch": 2.4527549743652345e-05,
      "model_forward_time": 2.2840328216552734,
      "step": 25719
    },
    {
      "epoch": 2.4527549743652345e-05,
      "step": 25719,
      "training_step_time": 7.653949499130249
    },
    {
      "epoch": 2.452850341796875e-05,
      "grad_norm": 0.1553110033273697,
      "learning_rate": 8.871573787389865e-06,
      "loss": 0.0628,
      "step": 25720
    },
    {
      "epoch": 2.452850341796875e-05,
      "model_forward_time": 1.7173330783843994,
      "step": 25720
    },
    {
      "dynamics_loss_avg": 0.05120149441063404,
      "epoch": 2.452850341796875e-05,
      "step": 25720
    },
    {
      "action_loss_avg": 0.001616812334395945,
      "epoch": 2.452850341796875e-05,
      "step": 25720
    },
    {
      "epoch": 2.452850341796875e-05,
      "step": 25720,
      "training_step_time": 8.808173656463623
    },
    {
      "epoch": 2.4529457092285156e-05,
      "model_forward_time": 1.7205564975738525,
      "step": 25721
    },
    {
      "epoch": 2.4529457092285156e-05,
      "step": 25721,
      "training_step_time": 7.529989957809448
    },
    {
      "epoch": 2.4530410766601564e-05,
      "model_forward_time": 1.7324886322021484,
      "step": 25722
    },
    {
      "epoch": 2.4530410766601564e-05,
      "step": 25722,
      "training_step_time": 6.774138927459717
    },
    {
      "epoch": 2.4531364440917968e-05,
      "model_forward_time": 1.7186260223388672,
      "step": 25723
    },
    {
      "epoch": 2.4531364440917968e-05,
      "step": 25723,
      "training_step_time": 6.816898584365845
    },
    {
      "epoch": 2.4532318115234375e-05,
      "model_forward_time": 1.7196378707885742,
      "step": 25724
    },
    {
      "epoch": 2.4532318115234375e-05,
      "step": 25724,
      "training_step_time": 6.796083211898804
    },
    {
      "epoch": 2.4533271789550783e-05,
      "model_forward_time": 1.7201361656188965,
      "step": 25725
    },
    {
      "epoch": 2.4533271789550783e-05,
      "step": 25725,
      "training_step_time": 6.776966094970703
    },
    {
      "epoch": 2.4534225463867187e-05,
      "model_forward_time": 1.7158057689666748,
      "step": 25726
    },
    {
      "epoch": 2.4534225463867187e-05,
      "step": 25726,
      "training_step_time": 6.781067609786987
    },
    {
      "epoch": 2.4535179138183594e-05,
      "model_forward_time": 1.3599791526794434,
      "step": 25727
    },
    {
      "epoch": 2.4535179138183594e-05,
      "step": 25727,
      "training_step_time": 6.76944899559021
    },
    {
      "epoch": 2.45361328125e-05,
      "model_forward_time": 2.2741897106170654,
      "step": 25728
    },
    {
      "epoch": 2.45361328125e-05,
      "step": 25728,
      "training_step_time": 7.602048397064209
    },
    {
      "epoch": 2.4537086486816405e-05,
      "model_forward_time": 1.7168128490447998,
      "step": 25729
    },
    {
      "epoch": 2.4537086486816405e-05,
      "step": 25729,
      "training_step_time": 10.027027606964111
    },
    {
      "epoch": 2.4538040161132813e-05,
      "grad_norm": 0.14993929862976074,
      "learning_rate": 8.870527258947025e-06,
      "loss": 0.063,
      "step": 25730
    },
    {
      "epoch": 2.4538040161132813e-05,
      "model_forward_time": 1.7198092937469482,
      "step": 25730
    },
    {
      "dynamics_loss_avg": 0.05104335248470306,
      "epoch": 2.4538040161132813e-05,
      "step": 25730
    },
    {
      "action_loss_avg": 0.0020551680237986146,
      "epoch": 2.4538040161132813e-05,
      "step": 25730
    },
    {
      "epoch": 2.4538040161132813e-05,
      "step": 25730,
      "training_step_time": 7.799095869064331
    },
    {
      "epoch": 2.453899383544922e-05,
      "model_forward_time": 1.7589833736419678,
      "step": 25731
    },
    {
      "epoch": 2.453899383544922e-05,
      "step": 25731,
      "training_step_time": 7.120055675506592
    },
    {
      "epoch": 2.4539947509765624e-05,
      "model_forward_time": 1.7250175476074219,
      "step": 25732
    },
    {
      "epoch": 2.4539947509765624e-05,
      "step": 25732,
      "training_step_time": 6.828960418701172
    },
    {
      "epoch": 2.4540901184082032e-05,
      "model_forward_time": 1.722801923751831,
      "step": 25733
    },
    {
      "epoch": 2.4540901184082032e-05,
      "step": 25733,
      "training_step_time": 6.765254735946655
    },
    {
      "epoch": 2.454185485839844e-05,
      "model_forward_time": 1.722196102142334,
      "step": 25734
    },
    {
      "epoch": 2.454185485839844e-05,
      "step": 25734,
      "training_step_time": 6.778802156448364
    },
    {
      "epoch": 2.4542808532714843e-05,
      "model_forward_time": 1.7215447425842285,
      "step": 25735
    },
    {
      "epoch": 2.4542808532714843e-05,
      "step": 25735,
      "training_step_time": 6.786565780639648
    },
    {
      "epoch": 2.454376220703125e-05,
      "model_forward_time": 1.7232277393341064,
      "step": 25736
    },
    {
      "epoch": 2.454376220703125e-05,
      "step": 25736,
      "training_step_time": 6.789198875427246
    },
    {
      "epoch": 2.4544715881347655e-05,
      "model_forward_time": 1.710798740386963,
      "step": 25737
    },
    {
      "epoch": 2.4544715881347655e-05,
      "step": 25737,
      "training_step_time": 8.07488465309143
    },
    {
      "epoch": 2.4545669555664062e-05,
      "model_forward_time": 1.7073230743408203,
      "step": 25738
    },
    {
      "epoch": 2.4545669555664062e-05,
      "step": 25738,
      "training_step_time": 9.549150943756104
    },
    {
      "epoch": 2.454662322998047e-05,
      "model_forward_time": 1.724858283996582,
      "step": 25739
    },
    {
      "epoch": 2.454662322998047e-05,
      "step": 25739,
      "training_step_time": 7.8289666175842285
    },
    {
      "epoch": 2.4547576904296874e-05,
      "grad_norm": 0.12722931802272797,
      "learning_rate": 8.869480307229144e-06,
      "loss": 0.0621,
      "step": 25740
    },
    {
      "epoch": 2.4547576904296874e-05,
      "model_forward_time": 1.7266271114349365,
      "step": 25740
    },
    {
      "dynamics_loss_avg": 0.05698955897241831,
      "epoch": 2.4547576904296874e-05,
      "step": 25740
    },
    {
      "action_loss_avg": 0.002091381704667583,
      "epoch": 2.4547576904296874e-05,
      "step": 25740
    },
    {
      "epoch": 2.4547576904296874e-05,
      "step": 25740,
      "training_step_time": 7.083669662475586
    },
    {
      "epoch": 2.454853057861328e-05,
      "model_forward_time": 1.7300777435302734,
      "step": 25741
    },
    {
      "epoch": 2.454853057861328e-05,
      "step": 25741,
      "training_step_time": 6.77439022064209
    },
    {
      "epoch": 2.454948425292969e-05,
      "model_forward_time": 1.7199337482452393,
      "step": 25742
    },
    {
      "epoch": 2.454948425292969e-05,
      "step": 25742,
      "training_step_time": 6.7801899909973145
    },
    {
      "epoch": 2.4550437927246092e-05,
      "model_forward_time": 1.7203850746154785,
      "step": 25743
    },
    {
      "epoch": 2.4550437927246092e-05,
      "step": 25743,
      "training_step_time": 6.783477306365967
    },
    {
      "epoch": 2.45513916015625e-05,
      "model_forward_time": 1.7249395847320557,
      "step": 25744
    },
    {
      "epoch": 2.45513916015625e-05,
      "step": 25744,
      "training_step_time": 6.76151704788208
    },
    {
      "epoch": 2.4552345275878907e-05,
      "model_forward_time": 1.7155656814575195,
      "step": 25745
    },
    {
      "epoch": 2.4552345275878907e-05,
      "step": 25745,
      "training_step_time": 6.764951705932617
    },
    {
      "epoch": 2.455329895019531e-05,
      "model_forward_time": 1.7255332469940186,
      "step": 25746
    },
    {
      "epoch": 2.455329895019531e-05,
      "step": 25746,
      "training_step_time": 7.458327770233154
    },
    {
      "epoch": 2.455425262451172e-05,
      "model_forward_time": 2.2993693351745605,
      "step": 25747
    },
    {
      "epoch": 2.455425262451172e-05,
      "step": 25747,
      "training_step_time": 8.97936463356018
    },
    {
      "epoch": 2.4555206298828126e-05,
      "model_forward_time": 1.7224318981170654,
      "step": 25748
    },
    {
      "epoch": 2.4555206298828126e-05,
      "step": 25748,
      "training_step_time": 7.770806550979614
    },
    {
      "epoch": 2.455615997314453e-05,
      "model_forward_time": 1.7151525020599365,
      "step": 25749
    },
    {
      "epoch": 2.455615997314453e-05,
      "step": 25749,
      "training_step_time": 7.567108869552612
    },
    {
      "epoch": 2.4557113647460938e-05,
      "grad_norm": 0.22847715020179749,
      "learning_rate": 8.868432932350712e-06,
      "loss": 0.0632,
      "step": 25750
    },
    {
      "epoch": 2.4557113647460938e-05,
      "model_forward_time": 1.7248690128326416,
      "step": 25750
    },
    {
      "dynamics_loss_avg": 0.05823212154209614,
      "epoch": 2.4557113647460938e-05,
      "step": 25750
    },
    {
      "action_loss_avg": 0.00272060080897063,
      "epoch": 2.4557113647460938e-05,
      "step": 25750
    },
    {
      "epoch": 2.4557113647460938e-05,
      "step": 25750,
      "training_step_time": 6.71356201171875
    },
    {
      "epoch": 2.4558067321777345e-05,
      "model_forward_time": 1.7274165153503418,
      "step": 25751
    },
    {
      "epoch": 2.4558067321777345e-05,
      "step": 25751,
      "training_step_time": 6.727255344390869
    },
    {
      "epoch": 2.455902099609375e-05,
      "model_forward_time": 1.7267897129058838,
      "step": 25752
    },
    {
      "epoch": 2.455902099609375e-05,
      "step": 25752,
      "training_step_time": 6.756802558898926
    },
    {
      "epoch": 2.4559974670410156e-05,
      "model_forward_time": 1.7191390991210938,
      "step": 25753
    },
    {
      "epoch": 2.4559974670410156e-05,
      "step": 25753,
      "training_step_time": 6.750784873962402
    },
    {
      "epoch": 2.4560928344726564e-05,
      "model_forward_time": 1.7219159603118896,
      "step": 25754
    },
    {
      "epoch": 2.4560928344726564e-05,
      "step": 25754,
      "training_step_time": 6.730240345001221
    },
    {
      "epoch": 2.4561882019042968e-05,
      "model_forward_time": 1.7199277877807617,
      "step": 25755
    },
    {
      "epoch": 2.4561882019042968e-05,
      "step": 25755,
      "training_step_time": 7.7600038051605225
    },
    {
      "epoch": 2.4562835693359375e-05,
      "model_forward_time": 2.314296007156372,
      "step": 25756
    },
    {
      "epoch": 2.4562835693359375e-05,
      "step": 25756,
      "training_step_time": 10.053002119064331
    },
    {
      "epoch": 2.4563789367675783e-05,
      "model_forward_time": 1.717562198638916,
      "step": 25757
    },
    {
      "epoch": 2.4563789367675783e-05,
      "step": 25757,
      "training_step_time": 7.820262432098389
    },
    {
      "epoch": 2.4564743041992187e-05,
      "model_forward_time": 1.7165358066558838,
      "step": 25758
    },
    {
      "epoch": 2.4564743041992187e-05,
      "step": 25758,
      "training_step_time": 7.6182849407196045
    },
    {
      "epoch": 2.4565696716308594e-05,
      "model_forward_time": 1.7227654457092285,
      "step": 25759
    },
    {
      "epoch": 2.4565696716308594e-05,
      "step": 25759,
      "training_step_time": 6.750395774841309
    },
    {
      "epoch": 2.4566650390625e-05,
      "grad_norm": 0.24704772233963013,
      "learning_rate": 8.867385134426272e-06,
      "loss": 0.0634,
      "step": 25760
    },
    {
      "epoch": 2.4566650390625e-05,
      "model_forward_time": 1.8904438018798828,
      "step": 25760
    },
    {
      "dynamics_loss_avg": 0.05298967957496643,
      "epoch": 2.4566650390625e-05,
      "step": 25760
    },
    {
      "action_loss_avg": 0.001902040105778724,
      "epoch": 2.4566650390625e-05,
      "step": 25760
    },
    {
      "epoch": 2.4566650390625e-05,
      "step": 25760,
      "training_step_time": 6.739432096481323
    },
    {
      "epoch": 2.4567604064941406e-05,
      "model_forward_time": 1.903862714767456,
      "step": 25761
    },
    {
      "epoch": 2.4567604064941406e-05,
      "step": 25761,
      "training_step_time": 6.7212114334106445
    },
    {
      "epoch": 2.4568557739257813e-05,
      "model_forward_time": 1.9267680644989014,
      "step": 25762
    },
    {
      "epoch": 2.4568557739257813e-05,
      "step": 25762,
      "training_step_time": 6.77977442741394
    },
    {
      "epoch": 2.456951141357422e-05,
      "model_forward_time": 1.923027515411377,
      "step": 25763
    },
    {
      "epoch": 2.456951141357422e-05,
      "step": 25763,
      "training_step_time": 6.740466594696045
    },
    {
      "epoch": 2.4570465087890624e-05,
      "model_forward_time": 1.9083480834960938,
      "step": 25764
    },
    {
      "epoch": 2.4570465087890624e-05,
      "step": 25764,
      "training_step_time": 6.738123893737793
    },
    {
      "epoch": 2.4571418762207032e-05,
      "model_forward_time": 2.9695537090301514,
      "step": 25765
    },
    {
      "epoch": 2.4571418762207032e-05,
      "step": 25765,
      "training_step_time": 7.708138465881348
    },
    {
      "epoch": 2.4572372436523436e-05,
      "model_forward_time": 1.9198188781738281,
      "step": 25766
    },
    {
      "epoch": 2.4572372436523436e-05,
      "step": 25766,
      "training_step_time": 9.630394220352173
    },
    {
      "epoch": 2.4573326110839843e-05,
      "model_forward_time": 1.9060330390930176,
      "step": 25767
    },
    {
      "epoch": 2.4573326110839843e-05,
      "step": 25767,
      "training_step_time": 7.589025020599365
    },
    {
      "epoch": 2.457427978515625e-05,
      "model_forward_time": 1.9096238613128662,
      "step": 25768
    },
    {
      "epoch": 2.457427978515625e-05,
      "step": 25768,
      "training_step_time": 7.018789291381836
    },
    {
      "epoch": 2.4575233459472655e-05,
      "model_forward_time": 1.9057533740997314,
      "step": 25769
    },
    {
      "epoch": 2.4575233459472655e-05,
      "step": 25769,
      "training_step_time": 6.735637187957764
    },
    {
      "epoch": 2.4576187133789062e-05,
      "grad_norm": 0.22829411923885345,
      "learning_rate": 8.866336913570408e-06,
      "loss": 0.0648,
      "step": 25770
    },
    {
      "epoch": 2.4576187133789062e-05,
      "model_forward_time": 1.92988920211792,
      "step": 25770
    },
    {
      "dynamics_loss_avg": 0.053471446223557,
      "epoch": 2.4576187133789062e-05,
      "step": 25770
    },
    {
      "action_loss_avg": 0.0016538811207283289,
      "epoch": 2.4576187133789062e-05,
      "step": 25770
    },
    {
      "epoch": 2.4576187133789062e-05,
      "step": 25770,
      "training_step_time": 6.7184576988220215
    },
    {
      "epoch": 2.457714080810547e-05,
      "model_forward_time": 1.9252796173095703,
      "step": 25771
    },
    {
      "epoch": 2.457714080810547e-05,
      "step": 25771,
      "training_step_time": 6.756237268447876
    },
    {
      "epoch": 2.4578094482421874e-05,
      "model_forward_time": 1.9113109111785889,
      "step": 25772
    },
    {
      "epoch": 2.4578094482421874e-05,
      "step": 25772,
      "training_step_time": 6.710317611694336
    },
    {
      "epoch": 2.457904815673828e-05,
      "model_forward_time": 1.9267637729644775,
      "step": 25773
    },
    {
      "epoch": 2.457904815673828e-05,
      "step": 25773,
      "training_step_time": 6.712686538696289
    },
    {
      "epoch": 2.458000183105469e-05,
      "model_forward_time": 3.0424275398254395,
      "step": 25774
    },
    {
      "epoch": 2.458000183105469e-05,
      "step": 25774,
      "training_step_time": 9.279043912887573
    },
    {
      "epoch": 2.4580955505371092e-05,
      "model_forward_time": 1.9249298572540283,
      "step": 25775
    },
    {
      "epoch": 2.4580955505371092e-05,
      "step": 25775,
      "training_step_time": 10.218031406402588
    },
    {
      "epoch": 2.45819091796875e-05,
      "model_forward_time": 1.9127061367034912,
      "step": 25776
    },
    {
      "epoch": 2.45819091796875e-05,
      "step": 25776,
      "training_step_time": 7.659467458724976
    },
    {
      "epoch": 2.4582862854003907e-05,
      "model_forward_time": 1.9203410148620605,
      "step": 25777
    },
    {
      "epoch": 2.4582862854003907e-05,
      "step": 25777,
      "training_step_time": 7.5014472007751465
    },
    {
      "epoch": 2.458381652832031e-05,
      "model_forward_time": 1.8986179828643799,
      "step": 25778
    },
    {
      "epoch": 2.458381652832031e-05,
      "step": 25778,
      "training_step_time": 6.79865288734436
    },
    {
      "epoch": 2.458477020263672e-05,
      "model_forward_time": 1.9112751483917236,
      "step": 25779
    },
    {
      "epoch": 2.458477020263672e-05,
      "step": 25779,
      "training_step_time": 6.756906270980835
    },
    {
      "epoch": 2.4585723876953126e-05,
      "grad_norm": 0.13051724433898926,
      "learning_rate": 8.865288269897752e-06,
      "loss": 0.0611,
      "step": 25780
    },
    {
      "epoch": 2.4585723876953126e-05,
      "model_forward_time": 1.9098806381225586,
      "step": 25780
    },
    {
      "dynamics_loss_avg": 0.061418299470096825,
      "epoch": 2.4585723876953126e-05,
      "step": 25780
    },
    {
      "action_loss_avg": 0.0018961341556860134,
      "epoch": 2.4585723876953126e-05,
      "step": 25780
    },
    {
      "epoch": 2.4585723876953126e-05,
      "step": 25780,
      "training_step_time": 6.709837436676025
    },
    {
      "epoch": 2.458667755126953e-05,
      "model_forward_time": 1.8985099792480469,
      "step": 25781
    },
    {
      "epoch": 2.458667755126953e-05,
      "step": 25781,
      "training_step_time": 6.692681789398193
    },
    {
      "epoch": 2.4587631225585938e-05,
      "model_forward_time": 1.9245493412017822,
      "step": 25782
    },
    {
      "epoch": 2.4587631225585938e-05,
      "step": 25782,
      "training_step_time": 6.701568841934204
    },
    {
      "epoch": 2.4588584899902345e-05,
      "model_forward_time": 3.028433084487915,
      "step": 25783
    },
    {
      "epoch": 2.4588584899902345e-05,
      "step": 25783,
      "training_step_time": 7.794262647628784
    },
    {
      "epoch": 2.458953857421875e-05,
      "model_forward_time": 1.9061312675476074,
      "step": 25784
    },
    {
      "epoch": 2.458953857421875e-05,
      "step": 25784,
      "training_step_time": 10.695732831954956
    },
    {
      "epoch": 2.4590492248535156e-05,
      "model_forward_time": 1.9229545593261719,
      "step": 25785
    },
    {
      "epoch": 2.4590492248535156e-05,
      "step": 25785,
      "training_step_time": 7.755503177642822
    },
    {
      "epoch": 2.4591445922851564e-05,
      "model_forward_time": 1.914365530014038,
      "step": 25786
    },
    {
      "epoch": 2.4591445922851564e-05,
      "step": 25786,
      "training_step_time": 7.559417724609375
    },
    {
      "epoch": 2.4592399597167968e-05,
      "model_forward_time": 1.9106299877166748,
      "step": 25787
    },
    {
      "epoch": 2.4592399597167968e-05,
      "step": 25787,
      "training_step_time": 6.694677352905273
    },
    {
      "epoch": 2.4593353271484375e-05,
      "model_forward_time": 1.9041368961334229,
      "step": 25788
    },
    {
      "epoch": 2.4593353271484375e-05,
      "step": 25788,
      "training_step_time": 6.750119686126709
    },
    {
      "epoch": 2.4594306945800783e-05,
      "model_forward_time": 1.924323558807373,
      "step": 25789
    },
    {
      "epoch": 2.4594306945800783e-05,
      "step": 25789,
      "training_step_time": 6.6489479541778564
    },
    {
      "epoch": 2.4595260620117187e-05,
      "grad_norm": 0.18602658808231354,
      "learning_rate": 8.864239203522982e-06,
      "loss": 0.0621,
      "step": 25790
    },
    {
      "epoch": 2.4595260620117187e-05,
      "model_forward_time": 1.9163882732391357,
      "step": 25790
    },
    {
      "dynamics_loss_avg": 0.045733976550400256,
      "epoch": 2.4595260620117187e-05,
      "step": 25790
    },
    {
      "action_loss_avg": 0.0016948567470535636,
      "epoch": 2.4595260620117187e-05,
      "step": 25790
    },
    {
      "epoch": 2.4595260620117187e-05,
      "step": 25790,
      "training_step_time": 6.709742784500122
    },
    {
      "epoch": 2.4596214294433594e-05,
      "model_forward_time": 1.9204282760620117,
      "step": 25791
    },
    {
      "epoch": 2.4596214294433594e-05,
      "step": 25791,
      "training_step_time": 6.727886199951172
    },
    {
      "epoch": 2.459716796875e-05,
      "model_forward_time": 1.9153838157653809,
      "step": 25792
    },
    {
      "epoch": 2.459716796875e-05,
      "step": 25792,
      "training_step_time": 7.596899509429932
    },
    {
      "epoch": 2.4598121643066406e-05,
      "model_forward_time": 2.963261604309082,
      "step": 25793
    },
    {
      "epoch": 2.4598121643066406e-05,
      "step": 25793,
      "training_step_time": 8.714888572692871
    },
    {
      "epoch": 2.4599075317382813e-05,
      "model_forward_time": 1.9199092388153076,
      "step": 25794
    },
    {
      "epoch": 2.4599075317382813e-05,
      "step": 25794,
      "training_step_time": 9.510159254074097
    },
    {
      "epoch": 2.460002899169922e-05,
      "model_forward_time": 1.9123954772949219,
      "step": 25795
    },
    {
      "epoch": 2.460002899169922e-05,
      "step": 25795,
      "training_step_time": 7.5880446434021
    },
    {
      "epoch": 2.4600982666015624e-05,
      "model_forward_time": 1.9344806671142578,
      "step": 25796
    },
    {
      "epoch": 2.4600982666015624e-05,
      "step": 25796,
      "training_step_time": 7.515158414840698
    },
    {
      "epoch": 2.4601936340332032e-05,
      "model_forward_time": 1.9198381900787354,
      "step": 25797
    },
    {
      "epoch": 2.4601936340332032e-05,
      "step": 25797,
      "training_step_time": 6.716077566146851
    },
    {
      "epoch": 2.4602890014648436e-05,
      "model_forward_time": 1.912843942642212,
      "step": 25798
    },
    {
      "epoch": 2.4602890014648436e-05,
      "step": 25798,
      "training_step_time": 6.710599184036255
    },
    {
      "epoch": 2.4603843688964843e-05,
      "model_forward_time": 1.9185383319854736,
      "step": 25799
    },
    {
      "epoch": 2.4603843688964843e-05,
      "step": 25799,
      "training_step_time": 6.766185522079468
    },
    {
      "epoch": 2.460479736328125e-05,
      "grad_norm": 0.1493614912033081,
      "learning_rate": 8.863189714560823e-06,
      "loss": 0.0651,
      "step": 25800
    },
    {
      "epoch": 2.460479736328125e-05,
      "model_forward_time": 1.922691822052002,
      "step": 25800
    },
    {
      "dynamics_loss_avg": 0.06423512510955334,
      "epoch": 2.460479736328125e-05,
      "step": 25800
    },
    {
      "action_loss_avg": 0.002447692607529461,
      "epoch": 2.460479736328125e-05,
      "step": 25800
    },
    {
      "epoch": 2.460479736328125e-05,
      "step": 25800,
      "training_step_time": 6.773477077484131
    },
    {
      "epoch": 2.4605751037597655e-05,
      "model_forward_time": 1.900261640548706,
      "step": 25801
    },
    {
      "epoch": 2.4605751037597655e-05,
      "step": 25801,
      "training_step_time": 7.725954532623291
    },
    {
      "epoch": 2.4606704711914062e-05,
      "model_forward_time": 2.9956257343292236,
      "step": 25802
    },
    {
      "epoch": 2.4606704711914062e-05,
      "step": 25802,
      "training_step_time": 8.304138898849487
    },
    {
      "epoch": 2.460765838623047e-05,
      "model_forward_time": 1.9096543788909912,
      "step": 25803
    },
    {
      "epoch": 2.460765838623047e-05,
      "step": 25803,
      "training_step_time": 9.38033938407898
    },
    {
      "epoch": 2.4608612060546874e-05,
      "model_forward_time": 1.9156877994537354,
      "step": 25804
    },
    {
      "epoch": 2.4608612060546874e-05,
      "step": 25804,
      "training_step_time": 7.549916744232178
    },
    {
      "epoch": 2.460956573486328e-05,
      "model_forward_time": 1.8987839221954346,
      "step": 25805
    },
    {
      "epoch": 2.460956573486328e-05,
      "step": 25805,
      "training_step_time": 7.441347122192383
    },
    {
      "epoch": 2.461051940917969e-05,
      "model_forward_time": 1.8985388278961182,
      "step": 25806
    },
    {
      "epoch": 2.461051940917969e-05,
      "step": 25806,
      "training_step_time": 6.735034704208374
    },
    {
      "epoch": 2.4611473083496092e-05,
      "model_forward_time": 1.9155070781707764,
      "step": 25807
    },
    {
      "epoch": 2.4611473083496092e-05,
      "step": 25807,
      "training_step_time": 6.703341960906982
    },
    {
      "epoch": 2.46124267578125e-05,
      "model_forward_time": 1.9210233688354492,
      "step": 25808
    },
    {
      "epoch": 2.46124267578125e-05,
      "step": 25808,
      "training_step_time": 6.740902423858643
    },
    {
      "epoch": 2.4613380432128907e-05,
      "model_forward_time": 1.9021108150482178,
      "step": 25809
    },
    {
      "epoch": 2.4613380432128907e-05,
      "step": 25809,
      "training_step_time": 6.758853912353516
    },
    {
      "epoch": 2.461433410644531e-05,
      "grad_norm": 0.19063885509967804,
      "learning_rate": 8.862139803126043e-06,
      "loss": 0.0624,
      "step": 25810
    },
    {
      "epoch": 2.461433410644531e-05,
      "model_forward_time": 1.9077837467193604,
      "step": 25810
    },
    {
      "dynamics_loss_avg": 0.05889131184667349,
      "epoch": 2.461433410644531e-05,
      "step": 25810
    },
    {
      "action_loss_avg": 0.002432285901159048,
      "epoch": 2.461433410644531e-05,
      "step": 25810
    },
    {
      "epoch": 2.461433410644531e-05,
      "step": 25810,
      "training_step_time": 8.410973072052002
    },
    {
      "epoch": 2.461528778076172e-05,
      "model_forward_time": 3.011679172515869,
      "step": 25811
    },
    {
      "epoch": 2.461528778076172e-05,
      "step": 25811,
      "training_step_time": 10.78371262550354
    },
    {
      "epoch": 2.4616241455078126e-05,
      "model_forward_time": 1.906531810760498,
      "step": 25812
    },
    {
      "epoch": 2.4616241455078126e-05,
      "step": 25812,
      "training_step_time": 10.588209390640259
    },
    {
      "epoch": 2.461719512939453e-05,
      "model_forward_time": 1.8980741500854492,
      "step": 25813
    },
    {
      "epoch": 2.461719512939453e-05,
      "step": 25813,
      "training_step_time": 8.199225664138794
    },
    {
      "epoch": 2.4618148803710938e-05,
      "model_forward_time": 1.882420301437378,
      "step": 25814
    },
    {
      "epoch": 2.4618148803710938e-05,
      "step": 25814,
      "training_step_time": 7.49403715133667
    },
    {
      "epoch": 2.4619102478027345e-05,
      "model_forward_time": 1.554062843322754,
      "step": 25815
    },
    {
      "epoch": 2.4619102478027345e-05,
      "step": 25815,
      "training_step_time": 6.771090745925903
    },
    {
      "epoch": 2.462005615234375e-05,
      "model_forward_time": 1.907691478729248,
      "step": 25816
    },
    {
      "epoch": 2.462005615234375e-05,
      "step": 25816,
      "training_step_time": 6.756630897521973
    },
    {
      "epoch": 2.4621009826660156e-05,
      "model_forward_time": 1.9180548191070557,
      "step": 25817
    },
    {
      "epoch": 2.4621009826660156e-05,
      "step": 25817,
      "training_step_time": 6.716676235198975
    },
    {
      "epoch": 2.4621963500976564e-05,
      "model_forward_time": 1.9256820678710938,
      "step": 25818
    },
    {
      "epoch": 2.4621963500976564e-05,
      "step": 25818,
      "training_step_time": 6.768763065338135
    },
    {
      "epoch": 2.4622917175292968e-05,
      "model_forward_time": 1.5490045547485352,
      "step": 25819
    },
    {
      "epoch": 2.4622917175292968e-05,
      "step": 25819,
      "training_step_time": 8.77487564086914
    },
    {
      "epoch": 2.4623870849609375e-05,
      "grad_norm": 0.14612062275409698,
      "learning_rate": 8.861089469333464e-06,
      "loss": 0.0612,
      "step": 25820
    },
    {
      "epoch": 2.4623870849609375e-05,
      "model_forward_time": 2.9725165367126465,
      "step": 25820
    },
    {
      "dynamics_loss_avg": 0.06275369096547365,
      "epoch": 2.4623870849609375e-05,
      "step": 25820
    },
    {
      "action_loss_avg": 0.0028415729524567724,
      "epoch": 2.4623870849609375e-05,
      "step": 25820
    },
    {
      "epoch": 2.4623870849609375e-05,
      "step": 25820,
      "training_step_time": 8.2670738697052
    },
    {
      "epoch": 2.4624824523925783e-05,
      "model_forward_time": 1.9104599952697754,
      "step": 25821
    },
    {
      "epoch": 2.4624824523925783e-05,
      "step": 25821,
      "training_step_time": 10.049363613128662
    },
    {
      "epoch": 2.4625778198242187e-05,
      "model_forward_time": 1.9143810272216797,
      "step": 25822
    },
    {
      "epoch": 2.4625778198242187e-05,
      "step": 25822,
      "training_step_time": 7.831086874008179
    },
    {
      "epoch": 2.4626731872558594e-05,
      "model_forward_time": 1.918691873550415,
      "step": 25823
    },
    {
      "epoch": 2.4626731872558594e-05,
      "step": 25823,
      "training_step_time": 7.6373584270477295
    },
    {
      "epoch": 2.4627685546875e-05,
      "model_forward_time": 1.557084321975708,
      "step": 25824
    },
    {
      "epoch": 2.4627685546875e-05,
      "step": 25824,
      "training_step_time": 6.813265562057495
    },
    {
      "epoch": 2.4628639221191406e-05,
      "model_forward_time": 1.8886537551879883,
      "step": 25825
    },
    {
      "epoch": 2.4628639221191406e-05,
      "step": 25825,
      "training_step_time": 6.700376749038696
    },
    {
      "epoch": 2.4629592895507813e-05,
      "model_forward_time": 1.9114348888397217,
      "step": 25826
    },
    {
      "epoch": 2.4629592895507813e-05,
      "step": 25826,
      "training_step_time": 6.720148324966431
    },
    {
      "epoch": 2.463054656982422e-05,
      "model_forward_time": 1.885650396347046,
      "step": 25827
    },
    {
      "epoch": 2.463054656982422e-05,
      "step": 25827,
      "training_step_time": 6.684110879898071
    },
    {
      "epoch": 2.4631500244140624e-05,
      "model_forward_time": 1.8743295669555664,
      "step": 25828
    },
    {
      "epoch": 2.4631500244140624e-05,
      "step": 25828,
      "training_step_time": 7.584576606750488
    },
    {
      "epoch": 2.4632453918457032e-05,
      "model_forward_time": 1.9173359870910645,
      "step": 25829
    },
    {
      "epoch": 2.4632453918457032e-05,
      "step": 25829,
      "training_step_time": 8.420945644378662
    },
    {
      "epoch": 2.4633407592773436e-05,
      "grad_norm": 0.2188105285167694,
      "learning_rate": 8.860038713297945e-06,
      "loss": 0.0608,
      "step": 25830
    },
    {
      "epoch": 2.4633407592773436e-05,
      "model_forward_time": 2.9904117584228516,
      "step": 25830
    },
    {
      "dynamics_loss_avg": 0.052791187353432176,
      "epoch": 2.4633407592773436e-05,
      "step": 25830
    },
    {
      "action_loss_avg": 0.0024616677372250708,
      "epoch": 2.4633407592773436e-05,
      "step": 25830
    },
    {
      "epoch": 2.4633407592773436e-05,
      "step": 25830,
      "training_step_time": 8.418061256408691
    },
    {
      "epoch": 2.4634361267089843e-05,
      "model_forward_time": 1.9085133075714111,
      "step": 25831
    },
    {
      "epoch": 2.4634361267089843e-05,
      "step": 25831,
      "training_step_time": 7.785965919494629
    },
    {
      "epoch": 2.463531494140625e-05,
      "model_forward_time": 1.9219741821289062,
      "step": 25832
    },
    {
      "epoch": 2.463531494140625e-05,
      "step": 25832,
      "training_step_time": 7.683272838592529
    },
    {
      "epoch": 2.4636268615722655e-05,
      "model_forward_time": 1.9076135158538818,
      "step": 25833
    },
    {
      "epoch": 2.4636268615722655e-05,
      "step": 25833,
      "training_step_time": 7.510009527206421
    },
    {
      "epoch": 2.4637222290039062e-05,
      "model_forward_time": 1.9176485538482666,
      "step": 25834
    },
    {
      "epoch": 2.4637222290039062e-05,
      "step": 25834,
      "training_step_time": 6.779713869094849
    },
    {
      "epoch": 2.463817596435547e-05,
      "model_forward_time": 1.9223203659057617,
      "step": 25835
    },
    {
      "epoch": 2.463817596435547e-05,
      "step": 25835,
      "training_step_time": 6.8064892292022705
    },
    {
      "epoch": 2.4639129638671874e-05,
      "model_forward_time": 1.903167724609375,
      "step": 25836
    },
    {
      "epoch": 2.4639129638671874e-05,
      "step": 25836,
      "training_step_time": 6.819788217544556
    },
    {
      "epoch": 2.464008331298828e-05,
      "model_forward_time": 1.9256818294525146,
      "step": 25837
    },
    {
      "epoch": 2.464008331298828e-05,
      "step": 25837,
      "training_step_time": 7.600724935531616
    },
    {
      "epoch": 2.464103698730469e-05,
      "model_forward_time": 1.919231653213501,
      "step": 25838
    },
    {
      "epoch": 2.464103698730469e-05,
      "step": 25838,
      "training_step_time": 8.759890079498291
    },
    {
      "epoch": 2.4641990661621092e-05,
      "model_forward_time": 3.0110106468200684,
      "step": 25839
    },
    {
      "epoch": 2.4641990661621092e-05,
      "step": 25839,
      "training_step_time": 8.829048156738281
    },
    {
      "epoch": 2.46429443359375e-05,
      "grad_norm": 0.1762496531009674,
      "learning_rate": 8.858987535134395e-06,
      "loss": 0.0612,
      "step": 25840
    },
    {
      "epoch": 2.46429443359375e-05,
      "model_forward_time": 1.9205996990203857,
      "step": 25840
    },
    {
      "dynamics_loss_avg": 0.048203762620687485,
      "epoch": 2.46429443359375e-05,
      "step": 25840
    },
    {
      "action_loss_avg": 0.0016526831052033231,
      "epoch": 2.46429443359375e-05,
      "step": 25840
    },
    {
      "epoch": 2.46429443359375e-05,
      "step": 25840,
      "training_step_time": 7.896395444869995
    },
    {
      "epoch": 2.4643898010253907e-05,
      "model_forward_time": 1.8977422714233398,
      "step": 25841
    },
    {
      "epoch": 2.4643898010253907e-05,
      "step": 25841,
      "training_step_time": 8.62369155883789
    },
    {
      "epoch": 2.464485168457031e-05,
      "model_forward_time": 1.8992514610290527,
      "step": 25842
    },
    {
      "epoch": 2.464485168457031e-05,
      "step": 25842,
      "training_step_time": 7.268624782562256
    },
    {
      "epoch": 2.464580535888672e-05,
      "model_forward_time": 1.8993892669677734,
      "step": 25843
    },
    {
      "epoch": 2.464580535888672e-05,
      "step": 25843,
      "training_step_time": 7.517794847488403
    },
    {
      "epoch": 2.4646759033203126e-05,
      "model_forward_time": 1.911566972732544,
      "step": 25844
    },
    {
      "epoch": 2.4646759033203126e-05,
      "step": 25844,
      "training_step_time": 6.808602571487427
    },
    {
      "epoch": 2.464771270751953e-05,
      "model_forward_time": 1.9051330089569092,
      "step": 25845
    },
    {
      "epoch": 2.464771270751953e-05,
      "step": 25845,
      "training_step_time": 6.809688568115234
    },
    {
      "epoch": 2.4648666381835938e-05,
      "model_forward_time": 1.9198634624481201,
      "step": 25846
    },
    {
      "epoch": 2.4648666381835938e-05,
      "step": 25846,
      "training_step_time": 7.6695640087127686
    },
    {
      "epoch": 2.4649620056152345e-05,
      "model_forward_time": 1.901620864868164,
      "step": 25847
    },
    {
      "epoch": 2.4649620056152345e-05,
      "step": 25847,
      "training_step_time": 8.795816898345947
    },
    {
      "epoch": 2.465057373046875e-05,
      "model_forward_time": 3.019763469696045,
      "step": 25848
    },
    {
      "epoch": 2.465057373046875e-05,
      "step": 25848,
      "training_step_time": 9.190560817718506
    },
    {
      "epoch": 2.4651527404785156e-05,
      "model_forward_time": 1.9124963283538818,
      "step": 25849
    },
    {
      "epoch": 2.4651527404785156e-05,
      "step": 25849,
      "training_step_time": 9.92031717300415
    },
    {
      "epoch": 2.4652481079101564e-05,
      "grad_norm": 0.12316511571407318,
      "learning_rate": 8.857935934957769e-06,
      "loss": 0.0582,
      "step": 25850
    },
    {
      "epoch": 2.4652481079101564e-05,
      "model_forward_time": 1.9210233688354492,
      "step": 25850
    },
    {
      "dynamics_loss_avg": 0.06417124047875404,
      "epoch": 2.4652481079101564e-05,
      "step": 25850
    },
    {
      "action_loss_avg": 0.0017335776938125491,
      "epoch": 2.4652481079101564e-05,
      "step": 25850
    },
    {
      "epoch": 2.4652481079101564e-05,
      "step": 25850,
      "training_step_time": 7.667623996734619
    },
    {
      "epoch": 2.4653434753417968e-05,
      "model_forward_time": 1.9099667072296143,
      "step": 25851
    },
    {
      "epoch": 2.4653434753417968e-05,
      "step": 25851,
      "training_step_time": 7.633840799331665
    },
    {
      "epoch": 2.4654388427734375e-05,
      "model_forward_time": 1.9126558303833008,
      "step": 25852
    },
    {
      "epoch": 2.4654388427734375e-05,
      "step": 25852,
      "training_step_time": 7.55447793006897
    },
    {
      "epoch": 2.4655342102050783e-05,
      "model_forward_time": 1.8806326389312744,
      "step": 25853
    },
    {
      "epoch": 2.4655342102050783e-05,
      "step": 25853,
      "training_step_time": 6.766199350357056
    },
    {
      "epoch": 2.4656295776367187e-05,
      "model_forward_time": 1.910015344619751,
      "step": 25854
    },
    {
      "epoch": 2.4656295776367187e-05,
      "step": 25854,
      "training_step_time": 6.777821063995361
    },
    {
      "epoch": 2.4657249450683594e-05,
      "model_forward_time": 1.8981940746307373,
      "step": 25855
    },
    {
      "epoch": 2.4657249450683594e-05,
      "step": 25855,
      "training_step_time": 7.652784585952759
    },
    {
      "epoch": 2.4658203125e-05,
      "model_forward_time": 1.9177863597869873,
      "step": 25856
    },
    {
      "epoch": 2.4658203125e-05,
      "step": 25856,
      "training_step_time": 8.846055269241333
    },
    {
      "epoch": 2.4659156799316406e-05,
      "model_forward_time": 3.003551483154297,
      "step": 25857
    },
    {
      "epoch": 2.4659156799316406e-05,
      "step": 25857,
      "training_step_time": 7.784711599349976
    },
    {
      "epoch": 2.4660110473632813e-05,
      "model_forward_time": 1.9011526107788086,
      "step": 25858
    },
    {
      "epoch": 2.4660110473632813e-05,
      "step": 25858,
      "training_step_time": 8.437806367874146
    },
    {
      "epoch": 2.466106414794922e-05,
      "model_forward_time": 1.918670654296875,
      "step": 25859
    },
    {
      "epoch": 2.466106414794922e-05,
      "step": 25859,
      "training_step_time": 7.637858867645264
    },
    {
      "epoch": 2.4662017822265624e-05,
      "grad_norm": 0.15472333133220673,
      "learning_rate": 8.85688391288307e-06,
      "loss": 0.0606,
      "step": 25860
    },
    {
      "epoch": 2.4662017822265624e-05,
      "model_forward_time": 1.916832447052002,
      "step": 25860
    },
    {
      "dynamics_loss_avg": 0.06662985347211361,
      "epoch": 2.4662017822265624e-05,
      "step": 25860
    },
    {
      "action_loss_avg": 0.0033377601619577037,
      "epoch": 2.4662017822265624e-05,
      "step": 25860
    },
    {
      "epoch": 2.4662017822265624e-05,
      "step": 25860,
      "training_step_time": 7.649655103683472
    },
    {
      "epoch": 2.4662971496582032e-05,
      "model_forward_time": 1.9262609481811523,
      "step": 25861
    },
    {
      "epoch": 2.4662971496582032e-05,
      "step": 25861,
      "training_step_time": 7.482388496398926
    },
    {
      "epoch": 2.4663925170898436e-05,
      "model_forward_time": 1.909660816192627,
      "step": 25862
    },
    {
      "epoch": 2.4663925170898436e-05,
      "step": 25862,
      "training_step_time": 6.776515960693359
    },
    {
      "epoch": 2.4664878845214843e-05,
      "model_forward_time": 1.9194614887237549,
      "step": 25863
    },
    {
      "epoch": 2.4664878845214843e-05,
      "step": 25863,
      "training_step_time": 6.7692201137542725
    },
    {
      "epoch": 2.466583251953125e-05,
      "model_forward_time": 1.923008918762207,
      "step": 25864
    },
    {
      "epoch": 2.466583251953125e-05,
      "step": 25864,
      "training_step_time": 7.687718868255615
    },
    {
      "epoch": 2.4666786193847655e-05,
      "model_forward_time": 1.9098296165466309,
      "step": 25865
    },
    {
      "epoch": 2.4666786193847655e-05,
      "step": 25865,
      "training_step_time": 8.376953363418579
    },
    {
      "epoch": 2.4667739868164062e-05,
      "model_forward_time": 2.9634249210357666,
      "step": 25866
    },
    {
      "epoch": 2.4667739868164062e-05,
      "step": 25866,
      "training_step_time": 9.442177772521973
    },
    {
      "epoch": 2.466869354248047e-05,
      "model_forward_time": 1.9116506576538086,
      "step": 25867
    },
    {
      "epoch": 2.466869354248047e-05,
      "step": 25867,
      "training_step_time": 9.569722414016724
    },
    {
      "epoch": 2.4669647216796874e-05,
      "model_forward_time": 1.9121065139770508,
      "step": 25868
    },
    {
      "epoch": 2.4669647216796874e-05,
      "step": 25868,
      "training_step_time": 7.6792168617248535
    },
    {
      "epoch": 2.467060089111328e-05,
      "model_forward_time": 1.9064879417419434,
      "step": 25869
    },
    {
      "epoch": 2.467060089111328e-05,
      "step": 25869,
      "training_step_time": 7.632015228271484
    },
    {
      "epoch": 2.467155456542969e-05,
      "grad_norm": 0.2534886598587036,
      "learning_rate": 8.855831469025347e-06,
      "loss": 0.0612,
      "step": 25870
    },
    {
      "epoch": 2.467155456542969e-05,
      "model_forward_time": 1.8976340293884277,
      "step": 25870
    },
    {
      "dynamics_loss_avg": 0.05813244618475437,
      "epoch": 2.467155456542969e-05,
      "step": 25870
    },
    {
      "action_loss_avg": 0.003134105203207582,
      "epoch": 2.467155456542969e-05,
      "step": 25870
    },
    {
      "epoch": 2.467155456542969e-05,
      "step": 25870,
      "training_step_time": 6.6795618534088135
    },
    {
      "epoch": 2.4672508239746092e-05,
      "model_forward_time": 1.536268711090088,
      "step": 25871
    },
    {
      "epoch": 2.4672508239746092e-05,
      "step": 25871,
      "training_step_time": 7.520674705505371
    },
    {
      "epoch": 2.46734619140625e-05,
      "model_forward_time": 1.9139456748962402,
      "step": 25872
    },
    {
      "epoch": 2.46734619140625e-05,
      "step": 25872,
      "training_step_time": 6.8101842403411865
    },
    {
      "epoch": 2.4674415588378907e-05,
      "model_forward_time": 1.9129185676574707,
      "step": 25873
    },
    {
      "epoch": 2.4674415588378907e-05,
      "step": 25873,
      "training_step_time": 7.751112699508667
    },
    {
      "epoch": 2.467536926269531e-05,
      "model_forward_time": 1.8928766250610352,
      "step": 25874
    },
    {
      "epoch": 2.467536926269531e-05,
      "step": 25874,
      "training_step_time": 8.927134275436401
    },
    {
      "epoch": 2.467632293701172e-05,
      "model_forward_time": 1.9103469848632812,
      "step": 25875
    },
    {
      "epoch": 2.467632293701172e-05,
      "step": 25875,
      "training_step_time": 8.357241868972778
    },
    {
      "epoch": 2.4677276611328126e-05,
      "model_forward_time": 2.9903039932250977,
      "step": 25876
    },
    {
      "epoch": 2.4677276611328126e-05,
      "step": 25876,
      "training_step_time": 9.021092176437378
    },
    {
      "epoch": 2.467823028564453e-05,
      "model_forward_time": 1.9104268550872803,
      "step": 25877
    },
    {
      "epoch": 2.467823028564453e-05,
      "step": 25877,
      "training_step_time": 7.792874813079834
    },
    {
      "epoch": 2.4679183959960938e-05,
      "model_forward_time": 1.9089608192443848,
      "step": 25878
    },
    {
      "epoch": 2.4679183959960938e-05,
      "step": 25878,
      "training_step_time": 8.657135725021362
    },
    {
      "epoch": 2.4680137634277345e-05,
      "model_forward_time": 1.9139747619628906,
      "step": 25879
    },
    {
      "epoch": 2.4680137634277345e-05,
      "step": 25879,
      "training_step_time": 7.645522594451904
    },
    {
      "epoch": 2.468109130859375e-05,
      "grad_norm": 0.14315906167030334,
      "learning_rate": 8.85477860349969e-06,
      "loss": 0.0629,
      "step": 25880
    },
    {
      "epoch": 2.468109130859375e-05,
      "model_forward_time": 1.8954105377197266,
      "step": 25880
    },
    {
      "dynamics_loss_avg": 0.06691017020493746,
      "epoch": 2.468109130859375e-05,
      "step": 25880
    },
    {
      "action_loss_avg": 0.0032180230133235454,
      "epoch": 2.468109130859375e-05,
      "step": 25880
    },
    {
      "epoch": 2.468109130859375e-05,
      "step": 25880,
      "training_step_time": 7.5710608959198
    },
    {
      "epoch": 2.4682044982910156e-05,
      "model_forward_time": 1.9196867942810059,
      "step": 25881
    },
    {
      "epoch": 2.4682044982910156e-05,
      "step": 25881,
      "training_step_time": 6.787043809890747
    },
    {
      "epoch": 2.4682998657226564e-05,
      "model_forward_time": 1.8903989791870117,
      "step": 25882
    },
    {
      "epoch": 2.4682998657226564e-05,
      "step": 25882,
      "training_step_time": 7.528330564498901
    },
    {
      "epoch": 2.4683952331542968e-05,
      "model_forward_time": 1.9200501441955566,
      "step": 25883
    },
    {
      "epoch": 2.4683952331542968e-05,
      "step": 25883,
      "training_step_time": 7.632895469665527
    },
    {
      "epoch": 2.4684906005859375e-05,
      "model_forward_time": 1.9160425662994385,
      "step": 25884
    },
    {
      "epoch": 2.4684906005859375e-05,
      "step": 25884,
      "training_step_time": 9.692662715911865
    },
    {
      "epoch": 2.4685859680175783e-05,
      "model_forward_time": 3.0571107864379883,
      "step": 25885
    },
    {
      "epoch": 2.4685859680175783e-05,
      "step": 25885,
      "training_step_time": 8.4473135471344
    },
    {
      "epoch": 2.4686813354492187e-05,
      "model_forward_time": 1.9000577926635742,
      "step": 25886
    },
    {
      "epoch": 2.4686813354492187e-05,
      "step": 25886,
      "training_step_time": 9.888720750808716
    },
    {
      "epoch": 2.4687767028808594e-05,
      "model_forward_time": 1.9298591613769531,
      "step": 25887
    },
    {
      "epoch": 2.4687767028808594e-05,
      "step": 25887,
      "training_step_time": 8.455911874771118
    },
    {
      "epoch": 2.4688720703125e-05,
      "model_forward_time": 1.9097225666046143,
      "step": 25888
    },
    {
      "epoch": 2.4688720703125e-05,
      "step": 25888,
      "training_step_time": 7.555579900741577
    },
    {
      "epoch": 2.4689674377441406e-05,
      "model_forward_time": 1.9267923831939697,
      "step": 25889
    },
    {
      "epoch": 2.4689674377441406e-05,
      "step": 25889,
      "training_step_time": 7.507425785064697
    },
    {
      "epoch": 2.4690628051757813e-05,
      "grad_norm": 0.1239733099937439,
      "learning_rate": 8.85372531642124e-06,
      "loss": 0.0608,
      "step": 25890
    },
    {
      "epoch": 2.4690628051757813e-05,
      "model_forward_time": 1.9230735301971436,
      "step": 25890
    },
    {
      "dynamics_loss_avg": 0.043231149390339854,
      "epoch": 2.4690628051757813e-05,
      "step": 25890
    },
    {
      "action_loss_avg": 0.0015103989688213914,
      "epoch": 2.4690628051757813e-05,
      "step": 25890
    },
    {
      "epoch": 2.4690628051757813e-05,
      "step": 25890,
      "training_step_time": 6.789116382598877
    },
    {
      "epoch": 2.4691581726074217e-05,
      "model_forward_time": 1.8834528923034668,
      "step": 25891
    },
    {
      "epoch": 2.4691581726074217e-05,
      "step": 25891,
      "training_step_time": 7.597042798995972
    },
    {
      "epoch": 2.4692535400390624e-05,
      "model_forward_time": 1.8943939208984375,
      "step": 25892
    },
    {
      "epoch": 2.4692535400390624e-05,
      "step": 25892,
      "training_step_time": 7.633928537368774
    },
    {
      "epoch": 2.4693489074707032e-05,
      "model_forward_time": 1.9175090789794922,
      "step": 25893
    },
    {
      "epoch": 2.4693489074707032e-05,
      "step": 25893,
      "training_step_time": 7.712908983230591
    },
    {
      "epoch": 2.4694442749023436e-05,
      "model_forward_time": 2.609571695327759,
      "step": 25894
    },
    {
      "epoch": 2.4694442749023436e-05,
      "step": 25894,
      "training_step_time": 7.703932285308838
    },
    {
      "epoch": 2.4695396423339843e-05,
      "model_forward_time": 1.9284417629241943,
      "step": 25895
    },
    {
      "epoch": 2.4695396423339843e-05,
      "step": 25895,
      "training_step_time": 10.078006982803345
    },
    {
      "epoch": 2.469635009765625e-05,
      "model_forward_time": 1.9008944034576416,
      "step": 25896
    },
    {
      "epoch": 2.469635009765625e-05,
      "step": 25896,
      "training_step_time": 7.680617332458496
    },
    {
      "epoch": 2.4697303771972655e-05,
      "model_forward_time": 1.916778564453125,
      "step": 25897
    },
    {
      "epoch": 2.4697303771972655e-05,
      "step": 25897,
      "training_step_time": 7.559388637542725
    },
    {
      "epoch": 2.4698257446289062e-05,
      "model_forward_time": 1.9165740013122559,
      "step": 25898
    },
    {
      "epoch": 2.4698257446289062e-05,
      "step": 25898,
      "training_step_time": 7.857736825942993
    },
    {
      "epoch": 2.469921112060547e-05,
      "model_forward_time": 1.8976306915283203,
      "step": 25899
    },
    {
      "epoch": 2.469921112060547e-05,
      "step": 25899,
      "training_step_time": 6.818255186080933
    },
    {
      "epoch": 2.4700164794921874e-05,
      "grad_norm": 0.12465137243270874,
      "learning_rate": 8.852671607905185e-06,
      "loss": 0.0588,
      "step": 25900
    },
    {
      "epoch": 2.4700164794921874e-05,
      "model_forward_time": 1.905350923538208,
      "step": 25900
    },
    {
      "dynamics_loss_avg": 0.04080715477466583,
      "epoch": 2.4700164794921874e-05,
      "step": 25900
    },
    {
      "action_loss_avg": 0.0012571147701237351,
      "epoch": 2.4700164794921874e-05,
      "step": 25900
    },
    {
      "epoch": 2.4700164794921874e-05,
      "step": 25900,
      "training_step_time": 7.8274781703948975
    },
    {
      "epoch": 2.470111846923828e-05,
      "model_forward_time": 1.5410878658294678,
      "step": 25901
    },
    {
      "epoch": 2.470111846923828e-05,
      "step": 25901,
      "training_step_time": 7.598175525665283
    },
    {
      "epoch": 2.470207214355469e-05,
      "model_forward_time": 1.9148118495941162,
      "step": 25902
    },
    {
      "epoch": 2.470207214355469e-05,
      "step": 25902,
      "training_step_time": 7.781116247177124
    },
    {
      "epoch": 2.4703025817871093e-05,
      "model_forward_time": 1.9088313579559326,
      "step": 25903
    },
    {
      "epoch": 2.4703025817871093e-05,
      "step": 25903,
      "training_step_time": 9.303381204605103
    },
    {
      "epoch": 2.47039794921875e-05,
      "model_forward_time": 2.917541027069092,
      "step": 25904
    },
    {
      "epoch": 2.47039794921875e-05,
      "step": 25904,
      "training_step_time": 8.072935104370117
    },
    {
      "epoch": 2.4704933166503907e-05,
      "model_forward_time": 1.8996834754943848,
      "step": 25905
    },
    {
      "epoch": 2.4704933166503907e-05,
      "step": 25905,
      "training_step_time": 7.6083173751831055
    },
    {
      "epoch": 2.470588684082031e-05,
      "model_forward_time": 1.8920044898986816,
      "step": 25906
    },
    {
      "epoch": 2.470588684082031e-05,
      "step": 25906,
      "training_step_time": 7.567210674285889
    },
    {
      "epoch": 2.470684051513672e-05,
      "model_forward_time": 1.9089534282684326,
      "step": 25907
    },
    {
      "epoch": 2.470684051513672e-05,
      "step": 25907,
      "training_step_time": 7.032171249389648
    },
    {
      "epoch": 2.4707794189453126e-05,
      "model_forward_time": 1.9119718074798584,
      "step": 25908
    },
    {
      "epoch": 2.4707794189453126e-05,
      "step": 25908,
      "training_step_time": 7.544457912445068
    },
    {
      "epoch": 2.470874786376953e-05,
      "model_forward_time": 1.886718988418579,
      "step": 25909
    },
    {
      "epoch": 2.470874786376953e-05,
      "step": 25909,
      "training_step_time": 7.742065668106079
    },
    {
      "epoch": 2.4709701538085938e-05,
      "grad_norm": 0.22544503211975098,
      "learning_rate": 8.851617478066754e-06,
      "loss": 0.063,
      "step": 25910
    },
    {
      "epoch": 2.4709701538085938e-05,
      "model_forward_time": 1.9060218334197998,
      "step": 25910
    },
    {
      "dynamics_loss_avg": 0.04488468961790204,
      "epoch": 2.4709701538085938e-05,
      "step": 25910
    },
    {
      "action_loss_avg": 0.0016976975137367844,
      "epoch": 2.4709701538085938e-05,
      "step": 25910
    },
    {
      "epoch": 2.4709701538085938e-05,
      "step": 25910,
      "training_step_time": 8.171745538711548
    },
    {
      "epoch": 2.4710655212402345e-05,
      "model_forward_time": 1.9063241481781006,
      "step": 25911
    },
    {
      "epoch": 2.4710655212402345e-05,
      "step": 25911,
      "training_step_time": 9.462819814682007
    },
    {
      "epoch": 2.471160888671875e-05,
      "model_forward_time": 1.9070351123809814,
      "step": 25912
    },
    {
      "epoch": 2.471160888671875e-05,
      "step": 25912,
      "training_step_time": 7.797682523727417
    },
    {
      "epoch": 2.4712562561035156e-05,
      "model_forward_time": 2.97959566116333,
      "step": 25913
    },
    {
      "epoch": 2.4712562561035156e-05,
      "step": 25913,
      "training_step_time": 8.85913610458374
    },
    {
      "epoch": 2.4713516235351564e-05,
      "model_forward_time": 1.8972444534301758,
      "step": 25914
    },
    {
      "epoch": 2.4713516235351564e-05,
      "step": 25914,
      "training_step_time": 8.252405405044556
    },
    {
      "epoch": 2.4714469909667968e-05,
      "model_forward_time": 1.9050419330596924,
      "step": 25915
    },
    {
      "epoch": 2.4714469909667968e-05,
      "step": 25915,
      "training_step_time": 8.166569232940674
    },
    {
      "epoch": 2.4715423583984375e-05,
      "model_forward_time": 1.8833892345428467,
      "step": 25916
    },
    {
      "epoch": 2.4715423583984375e-05,
      "step": 25916,
      "training_step_time": 7.291850805282593
    },
    {
      "epoch": 2.4716377258300783e-05,
      "model_forward_time": 1.902754783630371,
      "step": 25917
    },
    {
      "epoch": 2.4716377258300783e-05,
      "step": 25917,
      "training_step_time": 7.533475875854492
    },
    {
      "epoch": 2.4717330932617187e-05,
      "model_forward_time": 1.9146339893341064,
      "step": 25918
    },
    {
      "epoch": 2.4717330932617187e-05,
      "step": 25918,
      "training_step_time": 7.441058158874512
    },
    {
      "epoch": 2.4718284606933594e-05,
      "model_forward_time": 1.9016494750976562,
      "step": 25919
    },
    {
      "epoch": 2.4718284606933594e-05,
      "step": 25919,
      "training_step_time": 7.7810022830963135
    },
    {
      "epoch": 2.471923828125e-05,
      "grad_norm": 0.13951531052589417,
      "learning_rate": 8.850562927021227e-06,
      "loss": 0.0637,
      "step": 25920
    },
    {
      "epoch": 2.471923828125e-05,
      "model_forward_time": 1.5562670230865479,
      "step": 25920
    },
    {
      "dynamics_loss_avg": 0.05720321349799633,
      "epoch": 2.471923828125e-05,
      "step": 25920
    },
    {
      "action_loss_avg": 0.001559290470322594,
      "epoch": 2.471923828125e-05,
      "step": 25920
    },
    {
      "epoch": 2.471923828125e-05,
      "step": 25920,
      "training_step_time": 7.6426098346710205
    },
    {
      "epoch": 2.4720191955566406e-05,
      "model_forward_time": 1.9090781211853027,
      "step": 25921
    },
    {
      "epoch": 2.4720191955566406e-05,
      "step": 25921,
      "training_step_time": 8.345758438110352
    },
    {
      "epoch": 2.4721145629882813e-05,
      "model_forward_time": 2.591329574584961,
      "step": 25922
    },
    {
      "epoch": 2.4721145629882813e-05,
      "step": 25922,
      "training_step_time": 9.03596019744873
    },
    {
      "epoch": 2.4722099304199217e-05,
      "model_forward_time": 1.9225528240203857,
      "step": 25923
    },
    {
      "epoch": 2.4722099304199217e-05,
      "step": 25923,
      "training_step_time": 7.79940128326416
    },
    {
      "epoch": 2.4723052978515625e-05,
      "model_forward_time": 1.9181523323059082,
      "step": 25924
    },
    {
      "epoch": 2.4723052978515625e-05,
      "step": 25924,
      "training_step_time": 7.57466197013855
    },
    {
      "epoch": 2.4724006652832032e-05,
      "model_forward_time": 1.889784574508667,
      "step": 25925
    },
    {
      "epoch": 2.4724006652832032e-05,
      "step": 25925,
      "training_step_time": 7.566791772842407
    },
    {
      "epoch": 2.4724960327148436e-05,
      "model_forward_time": 1.9023730754852295,
      "step": 25926
    },
    {
      "epoch": 2.4724960327148436e-05,
      "step": 25926,
      "training_step_time": 7.504283428192139
    },
    {
      "epoch": 2.4725914001464843e-05,
      "model_forward_time": 1.875347375869751,
      "step": 25927
    },
    {
      "epoch": 2.4725914001464843e-05,
      "step": 25927,
      "training_step_time": 7.4457619190216064
    },
    {
      "epoch": 2.472686767578125e-05,
      "model_forward_time": 1.8943040370941162,
      "step": 25928
    },
    {
      "epoch": 2.472686767578125e-05,
      "step": 25928,
      "training_step_time": 7.801994562149048
    },
    {
      "epoch": 2.4727821350097655e-05,
      "model_forward_time": 1.852592945098877,
      "step": 25929
    },
    {
      "epoch": 2.4727821350097655e-05,
      "step": 25929,
      "training_step_time": 7.632269859313965
    },
    {
      "epoch": 2.4728775024414062e-05,
      "grad_norm": 0.17022207379341125,
      "learning_rate": 8.849507954883928e-06,
      "loss": 0.0625,
      "step": 25930
    },
    {
      "epoch": 2.4728775024414062e-05,
      "model_forward_time": 1.888188123703003,
      "step": 25930
    },
    {
      "dynamics_loss_avg": 0.045673217810690406,
      "epoch": 2.4728775024414062e-05,
      "step": 25930
    },
    {
      "action_loss_avg": 0.0020688660995801912,
      "epoch": 2.4728775024414062e-05,
      "step": 25930
    },
    {
      "epoch": 2.4728775024414062e-05,
      "step": 25930,
      "training_step_time": 8.632965326309204
    },
    {
      "epoch": 2.472972869873047e-05,
      "model_forward_time": 2.952867269515991,
      "step": 25931
    },
    {
      "epoch": 2.472972869873047e-05,
      "step": 25931,
      "training_step_time": 8.44414234161377
    },
    {
      "epoch": 2.4730682373046874e-05,
      "model_forward_time": 1.7345800399780273,
      "step": 25932
    },
    {
      "epoch": 2.4730682373046874e-05,
      "step": 25932,
      "training_step_time": 10.308333158493042
    },
    {
      "epoch": 2.473163604736328e-05,
      "model_forward_time": 1.7340083122253418,
      "step": 25933
    },
    {
      "epoch": 2.473163604736328e-05,
      "step": 25933,
      "training_step_time": 7.61358380317688
    },
    {
      "epoch": 2.473258972167969e-05,
      "model_forward_time": 1.733734130859375,
      "step": 25934
    },
    {
      "epoch": 2.473258972167969e-05,
      "step": 25934,
      "training_step_time": 7.5145018100738525
    },
    {
      "epoch": 2.4733543395996093e-05,
      "model_forward_time": 1.719783067703247,
      "step": 25935
    },
    {
      "epoch": 2.4733543395996093e-05,
      "step": 25935,
      "training_step_time": 7.4244139194488525
    },
    {
      "epoch": 2.47344970703125e-05,
      "model_forward_time": 1.7325043678283691,
      "step": 25936
    },
    {
      "epoch": 2.47344970703125e-05,
      "step": 25936,
      "training_step_time": 7.454088449478149
    },
    {
      "epoch": 2.4735450744628907e-05,
      "model_forward_time": 1.7181572914123535,
      "step": 25937
    },
    {
      "epoch": 2.4735450744628907e-05,
      "step": 25937,
      "training_step_time": 7.826102256774902
    },
    {
      "epoch": 2.473640441894531e-05,
      "model_forward_time": 1.7297382354736328,
      "step": 25938
    },
    {
      "epoch": 2.473640441894531e-05,
      "step": 25938,
      "training_step_time": 7.611117362976074
    },
    {
      "epoch": 2.473735809326172e-05,
      "model_forward_time": 1.7196269035339355,
      "step": 25939
    },
    {
      "epoch": 2.473735809326172e-05,
      "step": 25939,
      "training_step_time": 9.0631742477417
    },
    {
      "epoch": 2.4738311767578126e-05,
      "grad_norm": 0.1451239287853241,
      "learning_rate": 8.848452561770227e-06,
      "loss": 0.0645,
      "step": 25940
    },
    {
      "epoch": 2.4738311767578126e-05,
      "model_forward_time": 2.8836920261383057,
      "step": 25940
    },
    {
      "dynamics_loss_avg": 0.06837036088109016,
      "epoch": 2.4738311767578126e-05,
      "step": 25940
    },
    {
      "action_loss_avg": 0.0026906709477771073,
      "epoch": 2.4738311767578126e-05,
      "step": 25940
    },
    {
      "epoch": 2.4738311767578126e-05,
      "step": 25940,
      "training_step_time": 8.785243034362793
    },
    {
      "epoch": 2.473926544189453e-05,
      "model_forward_time": 1.729461908340454,
      "step": 25941
    },
    {
      "epoch": 2.473926544189453e-05,
      "step": 25941,
      "training_step_time": 9.70744013786316
    },
    {
      "epoch": 2.4740219116210938e-05,
      "model_forward_time": 1.7281584739685059,
      "step": 25942
    },
    {
      "epoch": 2.4740219116210938e-05,
      "step": 25942,
      "training_step_time": 8.760217666625977
    },
    {
      "epoch": 2.4741172790527345e-05,
      "model_forward_time": 1.7351596355438232,
      "step": 25943
    },
    {
      "epoch": 2.4741172790527345e-05,
      "step": 25943,
      "training_step_time": 7.5524375438690186
    },
    {
      "epoch": 2.474212646484375e-05,
      "model_forward_time": 1.7358646392822266,
      "step": 25944
    },
    {
      "epoch": 2.474212646484375e-05,
      "step": 25944,
      "training_step_time": 7.385780096054077
    },
    {
      "epoch": 2.4743080139160157e-05,
      "model_forward_time": 1.7399744987487793,
      "step": 25945
    },
    {
      "epoch": 2.4743080139160157e-05,
      "step": 25945,
      "training_step_time": 7.52806830406189
    },
    {
      "epoch": 2.4744033813476564e-05,
      "model_forward_time": 1.7357511520385742,
      "step": 25946
    },
    {
      "epoch": 2.4744033813476564e-05,
      "step": 25946,
      "training_step_time": 8.830122709274292
    },
    {
      "epoch": 2.4744987487792968e-05,
      "model_forward_time": 1.7332684993743896,
      "step": 25947
    },
    {
      "epoch": 2.4744987487792968e-05,
      "step": 25947,
      "training_step_time": 7.5962934494018555
    },
    {
      "epoch": 2.4745941162109375e-05,
      "model_forward_time": 1.7265477180480957,
      "step": 25948
    },
    {
      "epoch": 2.4745941162109375e-05,
      "step": 25948,
      "training_step_time": 8.658238887786865
    },
    {
      "epoch": 2.4746894836425783e-05,
      "model_forward_time": 2.8898606300354004,
      "step": 25949
    },
    {
      "epoch": 2.4746894836425783e-05,
      "step": 25949,
      "training_step_time": 8.383563756942749
    },
    {
      "epoch": 2.4747848510742187e-05,
      "grad_norm": 0.11186212301254272,
      "learning_rate": 8.847396747795538e-06,
      "loss": 0.0652,
      "step": 25950
    },
    {
      "epoch": 2.4747848510742187e-05,
      "model_forward_time": 1.7150053977966309,
      "step": 25950
    },
    {
      "dynamics_loss_avg": 0.05113890813663602,
      "epoch": 2.4747848510742187e-05,
      "step": 25950
    },
    {
      "action_loss_avg": 0.002362562681082636,
      "epoch": 2.4747848510742187e-05,
      "step": 25950
    },
    {
      "epoch": 2.4747848510742187e-05,
      "step": 25950,
      "training_step_time": 7.9209630489349365
    },
    {
      "epoch": 2.4748802185058594e-05,
      "model_forward_time": 1.7298541069030762,
      "step": 25951
    },
    {
      "epoch": 2.4748802185058594e-05,
      "step": 25951,
      "training_step_time": 8.364360332489014
    },
    {
      "epoch": 2.4749755859375e-05,
      "model_forward_time": 1.7152199745178223,
      "step": 25952
    },
    {
      "epoch": 2.4749755859375e-05,
      "step": 25952,
      "training_step_time": 7.616017818450928
    },
    {
      "epoch": 2.4750709533691406e-05,
      "model_forward_time": 1.7298715114593506,
      "step": 25953
    },
    {
      "epoch": 2.4750709533691406e-05,
      "step": 25953,
      "training_step_time": 7.378061771392822
    },
    {
      "epoch": 2.4751663208007813e-05,
      "model_forward_time": 1.7172131538391113,
      "step": 25954
    },
    {
      "epoch": 2.4751663208007813e-05,
      "step": 25954,
      "training_step_time": 7.493332386016846
    },
    {
      "epoch": 2.4752616882324217e-05,
      "model_forward_time": 1.3535046577453613,
      "step": 25955
    },
    {
      "epoch": 2.4752616882324217e-05,
      "step": 25955,
      "training_step_time": 7.764142751693726
    },
    {
      "epoch": 2.4753570556640625e-05,
      "model_forward_time": 1.7238342761993408,
      "step": 25956
    },
    {
      "epoch": 2.4753570556640625e-05,
      "step": 25956,
      "training_step_time": 7.610616683959961
    },
    {
      "epoch": 2.4754524230957032e-05,
      "model_forward_time": 1.7300240993499756,
      "step": 25957
    },
    {
      "epoch": 2.4754524230957032e-05,
      "step": 25957,
      "training_step_time": 7.624035835266113
    },
    {
      "epoch": 2.4755477905273436e-05,
      "model_forward_time": 1.3796663284301758,
      "step": 25958
    },
    {
      "epoch": 2.4755477905273436e-05,
      "step": 25958,
      "training_step_time": 7.742479085922241
    },
    {
      "epoch": 2.4756431579589843e-05,
      "model_forward_time": 2.9237332344055176,
      "step": 25959
    },
    {
      "epoch": 2.4756431579589843e-05,
      "step": 25959,
      "training_step_time": 10.565494060516357
    },
    {
      "epoch": 2.475738525390625e-05,
      "grad_norm": 0.11058700084686279,
      "learning_rate": 8.846340513075327e-06,
      "loss": 0.0671,
      "step": 25960
    },
    {
      "epoch": 2.475738525390625e-05,
      "model_forward_time": 1.7198312282562256,
      "step": 25960
    },
    {
      "dynamics_loss_avg": 0.07039918471127748,
      "epoch": 2.475738525390625e-05,
      "step": 25960
    },
    {
      "action_loss_avg": 0.00202923733741045,
      "epoch": 2.475738525390625e-05,
      "step": 25960
    },
    {
      "epoch": 2.475738525390625e-05,
      "step": 25960,
      "training_step_time": 8.730486154556274
    },
    {
      "epoch": 2.4758338928222655e-05,
      "model_forward_time": 1.7266745567321777,
      "step": 25961
    },
    {
      "epoch": 2.4758338928222655e-05,
      "step": 25961,
      "training_step_time": 7.575193405151367
    },
    {
      "epoch": 2.4759292602539062e-05,
      "model_forward_time": 1.713757038116455,
      "step": 25962
    },
    {
      "epoch": 2.4759292602539062e-05,
      "step": 25962,
      "training_step_time": 7.584678411483765
    },
    {
      "epoch": 2.476024627685547e-05,
      "model_forward_time": 1.7218105792999268,
      "step": 25963
    },
    {
      "epoch": 2.476024627685547e-05,
      "step": 25963,
      "training_step_time": 7.529191493988037
    },
    {
      "epoch": 2.4761199951171874e-05,
      "model_forward_time": 1.725508689880371,
      "step": 25964
    },
    {
      "epoch": 2.4761199951171874e-05,
      "step": 25964,
      "training_step_time": 6.780606985092163
    },
    {
      "epoch": 2.476215362548828e-05,
      "model_forward_time": 1.7275195121765137,
      "step": 25965
    },
    {
      "epoch": 2.476215362548828e-05,
      "step": 25965,
      "training_step_time": 7.68680477142334
    },
    {
      "epoch": 2.476310729980469e-05,
      "model_forward_time": 1.725515365600586,
      "step": 25966
    },
    {
      "epoch": 2.476310729980469e-05,
      "step": 25966,
      "training_step_time": 7.633455276489258
    },
    {
      "epoch": 2.4764060974121093e-05,
      "model_forward_time": 1.7722725868225098,
      "step": 25967
    },
    {
      "epoch": 2.4764060974121093e-05,
      "step": 25967,
      "training_step_time": 8.250859022140503
    },
    {
      "epoch": 2.47650146484375e-05,
      "model_forward_time": 2.862806558609009,
      "step": 25968
    },
    {
      "epoch": 2.47650146484375e-05,
      "step": 25968,
      "training_step_time": 10.26372480392456
    },
    {
      "epoch": 2.4765968322753907e-05,
      "model_forward_time": 1.7246112823486328,
      "step": 25969
    },
    {
      "epoch": 2.4765968322753907e-05,
      "step": 25969,
      "training_step_time": 9.434626340866089
    },
    {
      "epoch": 2.476692199707031e-05,
      "grad_norm": 0.1721980720758438,
      "learning_rate": 8.845283857725098e-06,
      "loss": 0.0659,
      "step": 25970
    },
    {
      "epoch": 2.476692199707031e-05,
      "model_forward_time": 1.716731071472168,
      "step": 25970
    },
    {
      "dynamics_loss_avg": 0.058977166563272475,
      "epoch": 2.476692199707031e-05,
      "step": 25970
    },
    {
      "action_loss_avg": 0.0016558688919758424,
      "epoch": 2.476692199707031e-05,
      "step": 25970
    },
    {
      "epoch": 2.476692199707031e-05,
      "step": 25970,
      "training_step_time": 8.664785623550415
    },
    {
      "epoch": 2.476787567138672e-05,
      "model_forward_time": 1.7368741035461426,
      "step": 25971
    },
    {
      "epoch": 2.476787567138672e-05,
      "step": 25971,
      "training_step_time": 7.532190322875977
    },
    {
      "epoch": 2.4768829345703126e-05,
      "model_forward_time": 1.7243542671203613,
      "step": 25972
    },
    {
      "epoch": 2.4768829345703126e-05,
      "step": 25972,
      "training_step_time": 7.473100423812866
    },
    {
      "epoch": 2.476978302001953e-05,
      "model_forward_time": 1.7458112239837646,
      "step": 25973
    },
    {
      "epoch": 2.476978302001953e-05,
      "step": 25973,
      "training_step_time": 6.76338791847229
    },
    {
      "epoch": 2.4770736694335938e-05,
      "model_forward_time": 1.7537469863891602,
      "step": 25974
    },
    {
      "epoch": 2.4770736694335938e-05,
      "step": 25974,
      "training_step_time": 7.781946897506714
    },
    {
      "epoch": 2.4771690368652345e-05,
      "model_forward_time": 1.7328248023986816,
      "step": 25975
    },
    {
      "epoch": 2.4771690368652345e-05,
      "step": 25975,
      "training_step_time": 7.571533679962158
    },
    {
      "epoch": 2.477264404296875e-05,
      "model_forward_time": 1.7266731262207031,
      "step": 25976
    },
    {
      "epoch": 2.477264404296875e-05,
      "step": 25976,
      "training_step_time": 8.596503496170044
    },
    {
      "epoch": 2.4773597717285157e-05,
      "model_forward_time": 2.89298677444458,
      "step": 25977
    },
    {
      "epoch": 2.4773597717285157e-05,
      "step": 25977,
      "training_step_time": 7.775965690612793
    },
    {
      "epoch": 2.4774551391601564e-05,
      "model_forward_time": 1.7260444164276123,
      "step": 25978
    },
    {
      "epoch": 2.4774551391601564e-05,
      "step": 25978,
      "training_step_time": 8.656772375106812
    },
    {
      "epoch": 2.4775505065917968e-05,
      "model_forward_time": 1.7349262237548828,
      "step": 25979
    },
    {
      "epoch": 2.4775505065917968e-05,
      "step": 25979,
      "training_step_time": 7.691948175430298
    },
    {
      "epoch": 2.4776458740234375e-05,
      "grad_norm": 0.23856770992279053,
      "learning_rate": 8.84422678186041e-06,
      "loss": 0.0667,
      "step": 25980
    },
    {
      "epoch": 2.4776458740234375e-05,
      "model_forward_time": 1.717379093170166,
      "step": 25980
    },
    {
      "dynamics_loss_avg": 0.06245613433420658,
      "epoch": 2.4776458740234375e-05,
      "step": 25980
    },
    {
      "action_loss_avg": 0.0023851730104070155,
      "epoch": 2.4776458740234375e-05,
      "step": 25980
    },
    {
      "epoch": 2.4776458740234375e-05,
      "step": 25980,
      "training_step_time": 7.5834009647369385
    },
    {
      "epoch": 2.4777412414550783e-05,
      "model_forward_time": 1.7228786945343018,
      "step": 25981
    },
    {
      "epoch": 2.4777412414550783e-05,
      "step": 25981,
      "training_step_time": 7.523356676101685
    },
    {
      "epoch": 2.4778366088867187e-05,
      "model_forward_time": 1.7266745567321777,
      "step": 25982
    },
    {
      "epoch": 2.4778366088867187e-05,
      "step": 25982,
      "training_step_time": 7.580313205718994
    },
    {
      "epoch": 2.4779319763183594e-05,
      "model_forward_time": 1.7291240692138672,
      "step": 25983
    },
    {
      "epoch": 2.4779319763183594e-05,
      "step": 25983,
      "training_step_time": 7.827840089797974
    },
    {
      "epoch": 2.47802734375e-05,
      "model_forward_time": 1.7226455211639404,
      "step": 25984
    },
    {
      "epoch": 2.47802734375e-05,
      "step": 25984,
      "training_step_time": 7.474109888076782
    },
    {
      "epoch": 2.4781227111816406e-05,
      "model_forward_time": 1.728548288345337,
      "step": 25985
    },
    {
      "epoch": 2.4781227111816406e-05,
      "step": 25985,
      "training_step_time": 10.10480284690857
    },
    {
      "epoch": 2.4782180786132813e-05,
      "model_forward_time": 2.906548500061035,
      "step": 25986
    },
    {
      "epoch": 2.4782180786132813e-05,
      "step": 25986,
      "training_step_time": 8.412830352783203
    },
    {
      "epoch": 2.4783134460449217e-05,
      "model_forward_time": 1.7243902683258057,
      "step": 25987
    },
    {
      "epoch": 2.4783134460449217e-05,
      "step": 25987,
      "training_step_time": 7.870678186416626
    },
    {
      "epoch": 2.4784088134765625e-05,
      "model_forward_time": 1.721311330795288,
      "step": 25988
    },
    {
      "epoch": 2.4784088134765625e-05,
      "step": 25988,
      "training_step_time": 7.6960649490356445
    },
    {
      "epoch": 2.4785041809082032e-05,
      "model_forward_time": 1.3586103916168213,
      "step": 25989
    },
    {
      "epoch": 2.4785041809082032e-05,
      "step": 25989,
      "training_step_time": 7.4850029945373535
    },
    {
      "epoch": 2.4785995483398436e-05,
      "grad_norm": 0.20440132915973663,
      "learning_rate": 8.84316928559686e-06,
      "loss": 0.0623,
      "step": 25990
    },
    {
      "epoch": 2.4785995483398436e-05,
      "model_forward_time": 1.7170729637145996,
      "step": 25990
    },
    {
      "dynamics_loss_avg": 0.05011558197438717,
      "epoch": 2.4785995483398436e-05,
      "step": 25990
    },
    {
      "action_loss_avg": 0.0019478041969705373,
      "epoch": 2.4785995483398436e-05,
      "step": 25990
    },
    {
      "epoch": 2.4785995483398436e-05,
      "step": 25990,
      "training_step_time": 7.580148458480835
    },
    {
      "epoch": 2.4786949157714843e-05,
      "model_forward_time": 1.7311663627624512,
      "step": 25991
    },
    {
      "epoch": 2.4786949157714843e-05,
      "step": 25991,
      "training_step_time": 7.573922872543335
    },
    {
      "epoch": 2.478790283203125e-05,
      "model_forward_time": 1.7239291667938232,
      "step": 25992
    },
    {
      "epoch": 2.478790283203125e-05,
      "step": 25992,
      "training_step_time": 7.774472713470459
    },
    {
      "epoch": 2.4788856506347655e-05,
      "model_forward_time": 1.718322515487671,
      "step": 25993
    },
    {
      "epoch": 2.4788856506347655e-05,
      "step": 25993,
      "training_step_time": 7.567060232162476
    },
    {
      "epoch": 2.4789810180664062e-05,
      "model_forward_time": 1.723252773284912,
      "step": 25994
    },
    {
      "epoch": 2.4789810180664062e-05,
      "step": 25994,
      "training_step_time": 9.789140462875366
    },
    {
      "epoch": 2.479076385498047e-05,
      "model_forward_time": 1.848982572555542,
      "step": 25995
    },
    {
      "epoch": 2.479076385498047e-05,
      "step": 25995,
      "training_step_time": 7.65356183052063
    },
    {
      "epoch": 2.4791717529296874e-05,
      "model_forward_time": 2.8434531688690186,
      "step": 25996
    },
    {
      "epoch": 2.4791717529296874e-05,
      "step": 25996,
      "training_step_time": 9.13095235824585
    },
    {
      "epoch": 2.479267120361328e-05,
      "model_forward_time": 1.7567381858825684,
      "step": 25997
    },
    {
      "epoch": 2.479267120361328e-05,
      "step": 25997,
      "training_step_time": 8.55498218536377
    },
    {
      "epoch": 2.479362487792969e-05,
      "model_forward_time": 1.7598211765289307,
      "step": 25998
    },
    {
      "epoch": 2.479362487792969e-05,
      "step": 25998,
      "training_step_time": 8.734278678894043
    },
    {
      "epoch": 2.4794578552246093e-05,
      "model_forward_time": 1.720902919769287,
      "step": 25999
    },
    {
      "epoch": 2.4794578552246093e-05,
      "step": 25999,
      "training_step_time": 7.68729043006897
    },
    {
      "epoch": 2.47955322265625e-05,
      "grad_norm": 0.16631951928138733,
      "learning_rate": 8.842111369050095e-06,
      "loss": 0.0572,
      "step": 26000
    },
    {
      "epoch": 2.47955322265625e-05,
      "model_forward_time": 1.758134365081787,
      "step": 26000
    },
    {
      "dynamics_loss_avg": 0.05682923598214984,
      "epoch": 2.47955322265625e-05,
      "step": 26000
    },
    {
      "action_loss_avg": 0.002559489069972187,
      "epoch": 2.47955322265625e-05,
      "step": 26000
    },
    {
      "epoch": 2.47955322265625e-05,
      "step": 26000,
      "training_step_time": 7.764256715774536
    },
    {
      "epoch": 2.4796485900878907e-05,
      "model_forward_time": 1.7361867427825928,
      "step": 26001
    },
    {
      "epoch": 2.4796485900878907e-05,
      "step": 26001,
      "training_step_time": 7.730175018310547
    },
    {
      "epoch": 2.479743957519531e-05,
      "model_forward_time": 1.7394030094146729,
      "step": 26002
    },
    {
      "epoch": 2.479743957519531e-05,
      "step": 26002,
      "training_step_time": 8.255814552307129
    },
    {
      "epoch": 2.479839324951172e-05,
      "model_forward_time": 1.7444815635681152,
      "step": 26003
    },
    {
      "epoch": 2.479839324951172e-05,
      "step": 26003,
      "training_step_time": 9.606379270553589
    },
    {
      "epoch": 2.4799346923828126e-05,
      "model_forward_time": 3.264822006225586,
      "step": 26004
    },
    {
      "epoch": 2.4799346923828126e-05,
      "step": 26004,
      "training_step_time": 10.847103357315063
    },
    {
      "epoch": 2.480030059814453e-05,
      "model_forward_time": 1.7318239212036133,
      "step": 26005
    },
    {
      "epoch": 2.480030059814453e-05,
      "step": 26005,
      "training_step_time": 9.478752851486206
    },
    {
      "epoch": 2.4801254272460938e-05,
      "model_forward_time": 1.0320558547973633,
      "step": 26006
    },
    {
      "epoch": 2.4801254272460938e-05,
      "step": 26006,
      "training_step_time": 9.960103273391724
    },
    {
      "epoch": 2.4802207946777345e-05,
      "model_forward_time": 1.736750602722168,
      "step": 26007
    },
    {
      "epoch": 2.4802207946777345e-05,
      "step": 26007,
      "training_step_time": 8.284883499145508
    },
    {
      "epoch": 2.480316162109375e-05,
      "model_forward_time": 1.7276442050933838,
      "step": 26008
    },
    {
      "epoch": 2.480316162109375e-05,
      "step": 26008,
      "training_step_time": 7.5982465744018555
    },
    {
      "epoch": 2.4804115295410157e-05,
      "model_forward_time": 1.7380492687225342,
      "step": 26009
    },
    {
      "epoch": 2.4804115295410157e-05,
      "step": 26009,
      "training_step_time": 8.439757347106934
    },
    {
      "epoch": 2.4805068969726564e-05,
      "grad_norm": 0.16122271120548248,
      "learning_rate": 8.841053032335809e-06,
      "loss": 0.0602,
      "step": 26010
    },
    {
      "epoch": 2.4805068969726564e-05,
      "model_forward_time": 1.7287633419036865,
      "step": 26010
    },
    {
      "dynamics_loss_avg": 0.050531489215791224,
      "epoch": 2.4805068969726564e-05,
      "step": 26010
    },
    {
      "action_loss_avg": 0.0023711080313660203,
      "epoch": 2.4805068969726564e-05,
      "step": 26010
    },
    {
      "epoch": 2.4805068969726564e-05,
      "step": 26010,
      "training_step_time": 7.5889668464660645
    },
    {
      "epoch": 2.4806022644042968e-05,
      "model_forward_time": 1.7521486282348633,
      "step": 26011
    },
    {
      "epoch": 2.4806022644042968e-05,
      "step": 26011,
      "training_step_time": 7.698638677597046
    },
    {
      "epoch": 2.4806976318359375e-05,
      "model_forward_time": 1.3717079162597656,
      "step": 26012
    },
    {
      "epoch": 2.4806976318359375e-05,
      "step": 26012,
      "training_step_time": 8.398041486740112
    },
    {
      "epoch": 2.4807929992675783e-05,
      "model_forward_time": 1.8520634174346924,
      "step": 26013
    },
    {
      "epoch": 2.4807929992675783e-05,
      "step": 26013,
      "training_step_time": 8.77751636505127
    },
    {
      "epoch": 2.4808883666992187e-05,
      "model_forward_time": 2.894296169281006,
      "step": 26014
    },
    {
      "epoch": 2.4808883666992187e-05,
      "step": 26014,
      "training_step_time": 8.698633432388306
    },
    {
      "epoch": 2.4809837341308594e-05,
      "model_forward_time": 1.7355997562408447,
      "step": 26015
    },
    {
      "epoch": 2.4809837341308594e-05,
      "step": 26015,
      "training_step_time": 8.548092603683472
    },
    {
      "epoch": 2.4810791015625e-05,
      "model_forward_time": 1.733109951019287,
      "step": 26016
    },
    {
      "epoch": 2.4810791015625e-05,
      "step": 26016,
      "training_step_time": 8.383945941925049
    },
    {
      "epoch": 2.4811744689941406e-05,
      "model_forward_time": 1.7293031215667725,
      "step": 26017
    },
    {
      "epoch": 2.4811744689941406e-05,
      "step": 26017,
      "training_step_time": 7.5578436851501465
    },
    {
      "epoch": 2.4812698364257813e-05,
      "model_forward_time": 1.7301318645477295,
      "step": 26018
    },
    {
      "epoch": 2.4812698364257813e-05,
      "step": 26018,
      "training_step_time": 7.522704362869263
    },
    {
      "epoch": 2.4813652038574217e-05,
      "model_forward_time": 1.7358670234680176,
      "step": 26019
    },
    {
      "epoch": 2.4813652038574217e-05,
      "step": 26019,
      "training_step_time": 7.530566692352295
    },
    {
      "epoch": 2.4814605712890625e-05,
      "grad_norm": 0.13058266043663025,
      "learning_rate": 8.839994275569735e-06,
      "loss": 0.0601,
      "step": 26020
    },
    {
      "epoch": 2.4814605712890625e-05,
      "model_forward_time": 1.7362008094787598,
      "step": 26020
    },
    {
      "dynamics_loss_avg": 0.061694913543760777,
      "epoch": 2.4814605712890625e-05,
      "step": 26020
    },
    {
      "action_loss_avg": 0.0023003211070317777,
      "epoch": 2.4814605712890625e-05,
      "step": 26020
    },
    {
      "epoch": 2.4814605712890625e-05,
      "step": 26020,
      "training_step_time": 7.629646062850952
    },
    {
      "epoch": 2.4815559387207032e-05,
      "model_forward_time": 1.7273457050323486,
      "step": 26021
    },
    {
      "epoch": 2.4815559387207032e-05,
      "step": 26021,
      "training_step_time": 7.757516145706177
    },
    {
      "epoch": 2.4816513061523436e-05,
      "model_forward_time": 1.8261632919311523,
      "step": 26022
    },
    {
      "epoch": 2.4816513061523436e-05,
      "step": 26022,
      "training_step_time": 8.95501971244812
    },
    {
      "epoch": 2.4817466735839843e-05,
      "model_forward_time": 2.8730340003967285,
      "step": 26023
    },
    {
      "epoch": 2.4817466735839843e-05,
      "step": 26023,
      "training_step_time": 8.645245552062988
    },
    {
      "epoch": 2.481842041015625e-05,
      "model_forward_time": 1.7303338050842285,
      "step": 26024
    },
    {
      "epoch": 2.481842041015625e-05,
      "step": 26024,
      "training_step_time": 8.788583040237427
    },
    {
      "epoch": 2.4819374084472655e-05,
      "model_forward_time": 1.0216524600982666,
      "step": 26025
    },
    {
      "epoch": 2.4819374084472655e-05,
      "step": 26025,
      "training_step_time": 7.6171417236328125
    },
    {
      "epoch": 2.4820327758789062e-05,
      "model_forward_time": 1.7330658435821533,
      "step": 26026
    },
    {
      "epoch": 2.4820327758789062e-05,
      "step": 26026,
      "training_step_time": 7.561342239379883
    },
    {
      "epoch": 2.482128143310547e-05,
      "model_forward_time": 1.73152756690979,
      "step": 26027
    },
    {
      "epoch": 2.482128143310547e-05,
      "step": 26027,
      "training_step_time": 8.424667596817017
    },
    {
      "epoch": 2.4822235107421874e-05,
      "model_forward_time": 1.7295289039611816,
      "step": 26028
    },
    {
      "epoch": 2.4822235107421874e-05,
      "step": 26028,
      "training_step_time": 7.521167278289795
    },
    {
      "epoch": 2.482318878173828e-05,
      "model_forward_time": 1.7324328422546387,
      "step": 26029
    },
    {
      "epoch": 2.482318878173828e-05,
      "step": 26029,
      "training_step_time": 7.626365900039673
    },
    {
      "epoch": 2.482414245605469e-05,
      "grad_norm": 0.1701318770647049,
      "learning_rate": 8.838935098867663e-06,
      "loss": 0.0582,
      "step": 26030
    },
    {
      "epoch": 2.482414245605469e-05,
      "model_forward_time": 1.7401974201202393,
      "step": 26030
    },
    {
      "dynamics_loss_avg": 0.054022568836808206,
      "epoch": 2.482414245605469e-05,
      "step": 26030
    },
    {
      "action_loss_avg": 0.0033039201865904034,
      "epoch": 2.482414245605469e-05,
      "step": 26030
    },
    {
      "epoch": 2.482414245605469e-05,
      "step": 26030,
      "training_step_time": 7.737353801727295
    },
    {
      "epoch": 2.4825096130371093e-05,
      "model_forward_time": 1.798851490020752,
      "step": 26031
    },
    {
      "epoch": 2.4825096130371093e-05,
      "step": 26031,
      "training_step_time": 8.172398328781128
    },
    {
      "epoch": 2.48260498046875e-05,
      "model_forward_time": 1.7415826320648193,
      "step": 26032
    },
    {
      "epoch": 2.48260498046875e-05,
      "step": 26032,
      "training_step_time": 9.128219604492188
    },
    {
      "epoch": 2.4827003479003907e-05,
      "model_forward_time": 1.7395901679992676,
      "step": 26033
    },
    {
      "epoch": 2.4827003479003907e-05,
      "step": 26033,
      "training_step_time": 7.878256797790527
    },
    {
      "epoch": 2.482795715332031e-05,
      "model_forward_time": 1.7280158996582031,
      "step": 26034
    },
    {
      "epoch": 2.482795715332031e-05,
      "step": 26034,
      "training_step_time": 9.031108379364014
    },
    {
      "epoch": 2.482891082763672e-05,
      "model_forward_time": 1.73524808883667,
      "step": 26035
    },
    {
      "epoch": 2.482891082763672e-05,
      "step": 26035,
      "training_step_time": 8.718575477600098
    },
    {
      "epoch": 2.4829864501953126e-05,
      "model_forward_time": 1.7288939952850342,
      "step": 26036
    },
    {
      "epoch": 2.4829864501953126e-05,
      "step": 26036,
      "training_step_time": 7.405040264129639
    },
    {
      "epoch": 2.483081817626953e-05,
      "model_forward_time": 1.7338652610778809,
      "step": 26037
    },
    {
      "epoch": 2.483081817626953e-05,
      "step": 26037,
      "training_step_time": 7.573632717132568
    },
    {
      "epoch": 2.4831771850585938e-05,
      "model_forward_time": 1.7396929264068604,
      "step": 26038
    },
    {
      "epoch": 2.4831771850585938e-05,
      "step": 26038,
      "training_step_time": 7.6462602615356445
    },
    {
      "epoch": 2.4832725524902345e-05,
      "model_forward_time": 1.7319581508636475,
      "step": 26039
    },
    {
      "epoch": 2.4832725524902345e-05,
      "step": 26039,
      "training_step_time": 7.644449472427368
    },
    {
      "epoch": 2.483367919921875e-05,
      "grad_norm": 0.20544478297233582,
      "learning_rate": 8.83787550234542e-06,
      "loss": 0.0604,
      "step": 26040
    },
    {
      "epoch": 2.483367919921875e-05,
      "model_forward_time": 1.7291584014892578,
      "step": 26040
    },
    {
      "dynamics_loss_avg": 0.07056141868233681,
      "epoch": 2.483367919921875e-05,
      "step": 26040
    },
    {
      "action_loss_avg": 0.003158557100687176,
      "epoch": 2.483367919921875e-05,
      "step": 26040
    },
    {
      "epoch": 2.483367919921875e-05,
      "step": 26040,
      "training_step_time": 8.188114643096924
    },
    {
      "epoch": 2.4834632873535157e-05,
      "model_forward_time": 1.7350902557373047,
      "step": 26041
    },
    {
      "epoch": 2.4834632873535157e-05,
      "step": 26041,
      "training_step_time": 9.18205738067627
    },
    {
      "epoch": 2.4835586547851564e-05,
      "model_forward_time": 2.8194386959075928,
      "step": 26042
    },
    {
      "epoch": 2.4835586547851564e-05,
      "step": 26042,
      "training_step_time": 8.853960514068604
    },
    {
      "epoch": 2.4836540222167968e-05,
      "model_forward_time": 1.728119134902954,
      "step": 26043
    },
    {
      "epoch": 2.4836540222167968e-05,
      "step": 26043,
      "training_step_time": 9.134271621704102
    },
    {
      "epoch": 2.4837493896484375e-05,
      "model_forward_time": 1.7292215824127197,
      "step": 26044
    },
    {
      "epoch": 2.4837493896484375e-05,
      "step": 26044,
      "training_step_time": 8.233946561813354
    },
    {
      "epoch": 2.4838447570800783e-05,
      "model_forward_time": 1.7599401473999023,
      "step": 26045
    },
    {
      "epoch": 2.4838447570800783e-05,
      "step": 26045,
      "training_step_time": 7.587521076202393
    },
    {
      "epoch": 2.4839401245117187e-05,
      "model_forward_time": 1.7308006286621094,
      "step": 26046
    },
    {
      "epoch": 2.4839401245117187e-05,
      "step": 26046,
      "training_step_time": 7.491231679916382
    },
    {
      "epoch": 2.4840354919433594e-05,
      "model_forward_time": 1.7359428405761719,
      "step": 26047
    },
    {
      "epoch": 2.4840354919433594e-05,
      "step": 26047,
      "training_step_time": 8.817130088806152
    },
    {
      "epoch": 2.484130859375e-05,
      "model_forward_time": 1.7325658798217773,
      "step": 26048
    },
    {
      "epoch": 2.484130859375e-05,
      "step": 26048,
      "training_step_time": 8.427565097808838
    },
    {
      "epoch": 2.4842262268066406e-05,
      "model_forward_time": 1.7274551391601562,
      "step": 26049
    },
    {
      "epoch": 2.4842262268066406e-05,
      "step": 26049,
      "training_step_time": 7.658435106277466
    },
    {
      "epoch": 2.4843215942382813e-05,
      "grad_norm": 0.12124539166688919,
      "learning_rate": 8.83681548611888e-06,
      "loss": 0.0601,
      "step": 26050
    },
    {
      "epoch": 2.4843215942382813e-05,
      "model_forward_time": 1.836916446685791,
      "step": 26050
    },
    {
      "dynamics_loss_avg": 0.05258841700851917,
      "epoch": 2.4843215942382813e-05,
      "step": 26050
    },
    {
      "action_loss_avg": 0.002574084117077291,
      "epoch": 2.4843215942382813e-05,
      "step": 26050
    },
    {
      "epoch": 2.4843215942382813e-05,
      "step": 26050,
      "training_step_time": 7.619810581207275
    },
    {
      "epoch": 2.4844169616699217e-05,
      "model_forward_time": 2.854153633117676,
      "step": 26051
    },
    {
      "epoch": 2.4844169616699217e-05,
      "step": 26051,
      "training_step_time": 9.408889055252075
    },
    {
      "epoch": 2.4845123291015625e-05,
      "model_forward_time": 1.7145581245422363,
      "step": 26052
    },
    {
      "epoch": 2.4845123291015625e-05,
      "step": 26052,
      "training_step_time": 7.953427076339722
    },
    {
      "epoch": 2.4846076965332032e-05,
      "model_forward_time": 1.739999771118164,
      "step": 26053
    },
    {
      "epoch": 2.4846076965332032e-05,
      "step": 26053,
      "training_step_time": 7.680891752243042
    },
    {
      "epoch": 2.4847030639648436e-05,
      "model_forward_time": 1.7352008819580078,
      "step": 26054
    },
    {
      "epoch": 2.4847030639648436e-05,
      "step": 26054,
      "training_step_time": 7.595767498016357
    },
    {
      "epoch": 2.4847984313964843e-05,
      "model_forward_time": 1.7311108112335205,
      "step": 26055
    },
    {
      "epoch": 2.4847984313964843e-05,
      "step": 26055,
      "training_step_time": 7.540410757064819
    },
    {
      "epoch": 2.484893798828125e-05,
      "model_forward_time": 1.7379019260406494,
      "step": 26056
    },
    {
      "epoch": 2.484893798828125e-05,
      "step": 26056,
      "training_step_time": 7.700313329696655
    },
    {
      "epoch": 2.4849891662597655e-05,
      "model_forward_time": 1.737802505493164,
      "step": 26057
    },
    {
      "epoch": 2.4849891662597655e-05,
      "step": 26057,
      "training_step_time": 8.138425350189209
    },
    {
      "epoch": 2.4850845336914062e-05,
      "model_forward_time": 1.7408552169799805,
      "step": 26058
    },
    {
      "epoch": 2.4850845336914062e-05,
      "step": 26058,
      "training_step_time": 7.5805699825286865
    },
    {
      "epoch": 2.485179901123047e-05,
      "model_forward_time": 1.8165953159332275,
      "step": 26059
    },
    {
      "epoch": 2.485179901123047e-05,
      "step": 26059,
      "training_step_time": 7.677568674087524
    },
    {
      "epoch": 2.4852752685546874e-05,
      "grad_norm": 0.1565054953098297,
      "learning_rate": 8.835755050303968e-06,
      "loss": 0.062,
      "step": 26060
    },
    {
      "epoch": 2.4852752685546874e-05,
      "model_forward_time": 2.8761746883392334,
      "step": 26060
    },
    {
      "dynamics_loss_avg": 0.06427512429654599,
      "epoch": 2.4852752685546874e-05,
      "step": 26060
    },
    {
      "action_loss_avg": 0.0019007851427886636,
      "epoch": 2.4852752685546874e-05,
      "step": 26060
    },
    {
      "epoch": 2.4852752685546874e-05,
      "step": 26060,
      "training_step_time": 8.821883916854858
    },
    {
      "epoch": 2.485370635986328e-05,
      "model_forward_time": 1.8039700984954834,
      "step": 26061
    },
    {
      "epoch": 2.485370635986328e-05,
      "step": 26061,
      "training_step_time": 9.357783317565918
    },
    {
      "epoch": 2.485466003417969e-05,
      "model_forward_time": 1.7366702556610107,
      "step": 26062
    },
    {
      "epoch": 2.485466003417969e-05,
      "step": 26062,
      "training_step_time": 7.759347915649414
    },
    {
      "epoch": 2.4855613708496093e-05,
      "model_forward_time": 1.7413208484649658,
      "step": 26063
    },
    {
      "epoch": 2.4855613708496093e-05,
      "step": 26063,
      "training_step_time": 7.609564781188965
    },
    {
      "epoch": 2.48565673828125e-05,
      "model_forward_time": 1.7413661479949951,
      "step": 26064
    },
    {
      "epoch": 2.48565673828125e-05,
      "step": 26064,
      "training_step_time": 7.63102388381958
    },
    {
      "epoch": 2.4857521057128907e-05,
      "model_forward_time": 1.7345926761627197,
      "step": 26065
    },
    {
      "epoch": 2.4857521057128907e-05,
      "step": 26065,
      "training_step_time": 7.631822109222412
    },
    {
      "epoch": 2.485847473144531e-05,
      "model_forward_time": 1.7386524677276611,
      "step": 26066
    },
    {
      "epoch": 2.485847473144531e-05,
      "step": 26066,
      "training_step_time": 7.651796817779541
    },
    {
      "epoch": 2.485942840576172e-05,
      "model_forward_time": 1.739428997039795,
      "step": 26067
    },
    {
      "epoch": 2.485942840576172e-05,
      "step": 26067,
      "training_step_time": 7.694352865219116
    },
    {
      "epoch": 2.4860382080078126e-05,
      "model_forward_time": 1.7380564212799072,
      "step": 26068
    },
    {
      "epoch": 2.4860382080078126e-05,
      "step": 26068,
      "training_step_time": 9.550999402999878
    },
    {
      "epoch": 2.486133575439453e-05,
      "model_forward_time": 2.877196788787842,
      "step": 26069
    },
    {
      "epoch": 2.486133575439453e-05,
      "step": 26069,
      "training_step_time": 7.650108814239502
    },
    {
      "epoch": 2.4862289428710938e-05,
      "grad_norm": 0.18165163695812225,
      "learning_rate": 8.834694195016653e-06,
      "loss": 0.064,
      "step": 26070
    },
    {
      "epoch": 2.4862289428710938e-05,
      "model_forward_time": 1.732858419418335,
      "step": 26070
    },
    {
      "dynamics_loss_avg": 0.05938875339925289,
      "epoch": 2.4862289428710938e-05,
      "step": 26070
    },
    {
      "action_loss_avg": 0.0034019886516034603,
      "epoch": 2.4862289428710938e-05,
      "step": 26070
    },
    {
      "epoch": 2.4862289428710938e-05,
      "step": 26070,
      "training_step_time": 8.521172523498535
    },
    {
      "epoch": 2.4863243103027345e-05,
      "model_forward_time": 1.7329134941101074,
      "step": 26071
    },
    {
      "epoch": 2.4863243103027345e-05,
      "step": 26071,
      "training_step_time": 8.262221813201904
    },
    {
      "epoch": 2.486419677734375e-05,
      "model_forward_time": 1.7438292503356934,
      "step": 26072
    },
    {
      "epoch": 2.486419677734375e-05,
      "step": 26072,
      "training_step_time": 8.583790302276611
    },
    {
      "epoch": 2.4865150451660157e-05,
      "model_forward_time": 1.7438900470733643,
      "step": 26073
    },
    {
      "epoch": 2.4865150451660157e-05,
      "step": 26073,
      "training_step_time": 7.552263498306274
    },
    {
      "epoch": 2.4866104125976564e-05,
      "model_forward_time": 1.7511663436889648,
      "step": 26074
    },
    {
      "epoch": 2.4866104125976564e-05,
      "step": 26074,
      "training_step_time": 7.547827482223511
    },
    {
      "epoch": 2.4867057800292968e-05,
      "model_forward_time": 1.7340281009674072,
      "step": 26075
    },
    {
      "epoch": 2.4867057800292968e-05,
      "step": 26075,
      "training_step_time": 7.603804111480713
    },
    {
      "epoch": 2.4868011474609375e-05,
      "model_forward_time": 1.7414307594299316,
      "step": 26076
    },
    {
      "epoch": 2.4868011474609375e-05,
      "step": 26076,
      "training_step_time": 8.177051305770874
    },
    {
      "epoch": 2.4868965148925783e-05,
      "model_forward_time": 1.7337839603424072,
      "step": 26077
    },
    {
      "epoch": 2.4868965148925783e-05,
      "step": 26077,
      "training_step_time": 7.636990308761597
    },
    {
      "epoch": 2.4869918823242187e-05,
      "model_forward_time": 2.92881441116333,
      "step": 26078
    },
    {
      "epoch": 2.4869918823242187e-05,
      "step": 26078,
      "training_step_time": 8.681309700012207
    },
    {
      "epoch": 2.4870872497558594e-05,
      "model_forward_time": 1.7382960319519043,
      "step": 26079
    },
    {
      "epoch": 2.4870872497558594e-05,
      "step": 26079,
      "training_step_time": 9.674437999725342
    },
    {
      "epoch": 2.4871826171875e-05,
      "grad_norm": 0.1360369473695755,
      "learning_rate": 8.833632920372943e-06,
      "loss": 0.0639,
      "step": 26080
    },
    {
      "epoch": 2.4871826171875e-05,
      "model_forward_time": 1.742361068725586,
      "step": 26080
    },
    {
      "dynamics_loss_avg": 0.05182660492137074,
      "epoch": 2.4871826171875e-05,
      "step": 26080
    },
    {
      "action_loss_avg": 0.0024606720893643797,
      "epoch": 2.4871826171875e-05,
      "step": 26080
    },
    {
      "epoch": 2.4871826171875e-05,
      "step": 26080,
      "training_step_time": 7.9670937061309814
    },
    {
      "epoch": 2.4872779846191406e-05,
      "model_forward_time": 1.7207810878753662,
      "step": 26081
    },
    {
      "epoch": 2.4872779846191406e-05,
      "step": 26081,
      "training_step_time": 8.554966688156128
    },
    {
      "epoch": 2.4873733520507813e-05,
      "model_forward_time": 1.721630334854126,
      "step": 26082
    },
    {
      "epoch": 2.4873733520507813e-05,
      "step": 26082,
      "training_step_time": 7.566447019577026
    },
    {
      "epoch": 2.4874687194824217e-05,
      "model_forward_time": 1.7346274852752686,
      "step": 26083
    },
    {
      "epoch": 2.4874687194824217e-05,
      "step": 26083,
      "training_step_time": 7.484164714813232
    },
    {
      "epoch": 2.4875640869140625e-05,
      "model_forward_time": 1.7291667461395264,
      "step": 26084
    },
    {
      "epoch": 2.4875640869140625e-05,
      "step": 26084,
      "training_step_time": 7.673220872879028
    },
    {
      "epoch": 2.4876594543457032e-05,
      "model_forward_time": 1.7229604721069336,
      "step": 26085
    },
    {
      "epoch": 2.4876594543457032e-05,
      "step": 26085,
      "training_step_time": 7.58799409866333
    },
    {
      "epoch": 2.4877548217773436e-05,
      "model_forward_time": 1.7233431339263916,
      "step": 26086
    },
    {
      "epoch": 2.4877548217773436e-05,
      "step": 26086,
      "training_step_time": 9.175546646118164
    },
    {
      "epoch": 2.4878501892089844e-05,
      "model_forward_time": 2.9176018238067627,
      "step": 26087
    },
    {
      "epoch": 2.4878501892089844e-05,
      "step": 26087,
      "training_step_time": 8.657334804534912
    },
    {
      "epoch": 2.487945556640625e-05,
      "model_forward_time": 1.7254929542541504,
      "step": 26088
    },
    {
      "epoch": 2.487945556640625e-05,
      "step": 26088,
      "training_step_time": 9.050816297531128
    },
    {
      "epoch": 2.4880409240722655e-05,
      "model_forward_time": 1.738239049911499,
      "step": 26089
    },
    {
      "epoch": 2.4880409240722655e-05,
      "step": 26089,
      "training_step_time": 7.938734769821167
    },
    {
      "epoch": 2.4881362915039062e-05,
      "grad_norm": 0.14856483042240143,
      "learning_rate": 8.832571226488904e-06,
      "loss": 0.0638,
      "step": 26090
    },
    {
      "epoch": 2.4881362915039062e-05,
      "model_forward_time": 1.7243521213531494,
      "step": 26090
    },
    {
      "dynamics_loss_avg": 0.06926181018352509,
      "epoch": 2.4881362915039062e-05,
      "step": 26090
    },
    {
      "action_loss_avg": 0.0029221713601145895,
      "epoch": 2.4881362915039062e-05,
      "step": 26090
    },
    {
      "epoch": 2.4881362915039062e-05,
      "step": 26090,
      "training_step_time": 7.670695066452026
    },
    {
      "epoch": 2.488231658935547e-05,
      "model_forward_time": 1.7157487869262695,
      "step": 26091
    },
    {
      "epoch": 2.488231658935547e-05,
      "step": 26091,
      "training_step_time": 7.5570080280303955
    },
    {
      "epoch": 2.4883270263671874e-05,
      "model_forward_time": 1.725956678390503,
      "step": 26092
    },
    {
      "epoch": 2.4883270263671874e-05,
      "step": 26092,
      "training_step_time": 7.538747787475586
    },
    {
      "epoch": 2.488422393798828e-05,
      "model_forward_time": 1.0177195072174072,
      "step": 26093
    },
    {
      "epoch": 2.488422393798828e-05,
      "step": 26093,
      "training_step_time": 7.6404759883880615
    },
    {
      "epoch": 2.488517761230469e-05,
      "model_forward_time": 1.7221379280090332,
      "step": 26094
    },
    {
      "epoch": 2.488517761230469e-05,
      "step": 26094,
      "training_step_time": 7.642052173614502
    },
    {
      "epoch": 2.4886131286621093e-05,
      "model_forward_time": 1.734100103378296,
      "step": 26095
    },
    {
      "epoch": 2.4886131286621093e-05,
      "step": 26095,
      "training_step_time": 8.734831809997559
    },
    {
      "epoch": 2.48870849609375e-05,
      "model_forward_time": 2.891238212585449,
      "step": 26096
    },
    {
      "epoch": 2.48870849609375e-05,
      "step": 26096,
      "training_step_time": 7.711407423019409
    },
    {
      "epoch": 2.4888038635253907e-05,
      "model_forward_time": 1.7228639125823975,
      "step": 26097
    },
    {
      "epoch": 2.4888038635253907e-05,
      "step": 26097,
      "training_step_time": 8.67623233795166
    },
    {
      "epoch": 2.488899230957031e-05,
      "model_forward_time": 1.7514472007751465,
      "step": 26098
    },
    {
      "epoch": 2.488899230957031e-05,
      "step": 26098,
      "training_step_time": 8.783491134643555
    },
    {
      "epoch": 2.488994598388672e-05,
      "model_forward_time": 1.7264914512634277,
      "step": 26099
    },
    {
      "epoch": 2.488994598388672e-05,
      "step": 26099,
      "training_step_time": 7.660909652709961
    },
    {
      "epoch": 2.4890899658203126e-05,
      "grad_norm": 0.16534188389778137,
      "learning_rate": 8.831509113480634e-06,
      "loss": 0.0628,
      "step": 26100
    },
    {
      "epoch": 2.4890899658203126e-05,
      "model_forward_time": 1.7302579879760742,
      "step": 26100
    },
    {
      "dynamics_loss_avg": 0.051035930216312406,
      "epoch": 2.4890899658203126e-05,
      "step": 26100
    },
    {
      "action_loss_avg": 0.0020604462653864173,
      "epoch": 2.4890899658203126e-05,
      "step": 26100
    },
    {
      "epoch": 2.4890899658203126e-05,
      "step": 26100,
      "training_step_time": 7.501551866531372
    },
    {
      "epoch": 2.489185333251953e-05,
      "model_forward_time": 1.7297520637512207,
      "step": 26101
    },
    {
      "epoch": 2.489185333251953e-05,
      "step": 26101,
      "training_step_time": 7.036892414093018
    },
    {
      "epoch": 2.4892807006835938e-05,
      "model_forward_time": 1.7197389602661133,
      "step": 26102
    },
    {
      "epoch": 2.4892807006835938e-05,
      "step": 26102,
      "training_step_time": 7.618396043777466
    },
    {
      "epoch": 2.4893760681152345e-05,
      "model_forward_time": 1.7197494506835938,
      "step": 26103
    },
    {
      "epoch": 2.4893760681152345e-05,
      "step": 26103,
      "training_step_time": 8.426660299301147
    },
    {
      "epoch": 2.489471435546875e-05,
      "model_forward_time": 1.7259764671325684,
      "step": 26104
    },
    {
      "epoch": 2.489471435546875e-05,
      "step": 26104,
      "training_step_time": 7.626520872116089
    },
    {
      "epoch": 2.4895668029785157e-05,
      "model_forward_time": 2.891151189804077,
      "step": 26105
    },
    {
      "epoch": 2.4895668029785157e-05,
      "step": 26105,
      "training_step_time": 8.755831003189087
    },
    {
      "epoch": 2.4896621704101564e-05,
      "model_forward_time": 1.725667953491211,
      "step": 26106
    },
    {
      "epoch": 2.4896621704101564e-05,
      "step": 26106,
      "training_step_time": 7.690825939178467
    },
    {
      "epoch": 2.4897575378417968e-05,
      "model_forward_time": 1.740164041519165,
      "step": 26107
    },
    {
      "epoch": 2.4897575378417968e-05,
      "step": 26107,
      "training_step_time": 7.649315595626831
    },
    {
      "epoch": 2.4898529052734376e-05,
      "model_forward_time": 1.7166798114776611,
      "step": 26108
    },
    {
      "epoch": 2.4898529052734376e-05,
      "step": 26108,
      "training_step_time": 8.788007736206055
    },
    {
      "epoch": 2.4899482727050783e-05,
      "model_forward_time": 1.7310645580291748,
      "step": 26109
    },
    {
      "epoch": 2.4899482727050783e-05,
      "step": 26109,
      "training_step_time": 7.639132976531982
    },
    {
      "epoch": 2.4900436401367187e-05,
      "grad_norm": 0.14919474720954895,
      "learning_rate": 8.83044658146429e-06,
      "loss": 0.0638,
      "step": 26110
    },
    {
      "epoch": 2.4900436401367187e-05,
      "model_forward_time": 1.728722333908081,
      "step": 26110
    },
    {
      "dynamics_loss_avg": 0.04596470557153225,
      "epoch": 2.4900436401367187e-05,
      "step": 26110
    },
    {
      "action_loss_avg": 0.0019448953098617494,
      "epoch": 2.4900436401367187e-05,
      "step": 26110
    },
    {
      "epoch": 2.4900436401367187e-05,
      "step": 26110,
      "training_step_time": 7.582033634185791
    },
    {
      "epoch": 2.4901390075683594e-05,
      "model_forward_time": 1.7236213684082031,
      "step": 26111
    },
    {
      "epoch": 2.4901390075683594e-05,
      "step": 26111,
      "training_step_time": 7.604887008666992
    },
    {
      "epoch": 2.490234375e-05,
      "model_forward_time": 1.7190165519714355,
      "step": 26112
    },
    {
      "epoch": 2.490234375e-05,
      "step": 26112,
      "training_step_time": 7.548767805099487
    },
    {
      "epoch": 2.4903297424316406e-05,
      "model_forward_time": 1.7238008975982666,
      "step": 26113
    },
    {
      "epoch": 2.4903297424316406e-05,
      "step": 26113,
      "training_step_time": 7.727663278579712
    },
    {
      "epoch": 2.4904251098632813e-05,
      "model_forward_time": 2.881636142730713,
      "step": 26114
    },
    {
      "epoch": 2.4904251098632813e-05,
      "step": 26114,
      "training_step_time": 8.64949631690979
    },
    {
      "epoch": 2.4905204772949217e-05,
      "model_forward_time": 1.7347931861877441,
      "step": 26115
    },
    {
      "epoch": 2.4905204772949217e-05,
      "step": 26115,
      "training_step_time": 9.544153213500977
    },
    {
      "epoch": 2.4906158447265625e-05,
      "model_forward_time": 1.733764410018921,
      "step": 26116
    },
    {
      "epoch": 2.4906158447265625e-05,
      "step": 26116,
      "training_step_time": 8.753053903579712
    },
    {
      "epoch": 2.4907112121582032e-05,
      "model_forward_time": 1.7193846702575684,
      "step": 26117
    },
    {
      "epoch": 2.4907112121582032e-05,
      "step": 26117,
      "training_step_time": 9.607048988342285
    },
    {
      "epoch": 2.4908065795898436e-05,
      "model_forward_time": 1.7215445041656494,
      "step": 26118
    },
    {
      "epoch": 2.4908065795898436e-05,
      "step": 26118,
      "training_step_time": 7.661603927612305
    },
    {
      "epoch": 2.4909019470214844e-05,
      "model_forward_time": 1.7256660461425781,
      "step": 26119
    },
    {
      "epoch": 2.4909019470214844e-05,
      "step": 26119,
      "training_step_time": 7.646103382110596
    },
    {
      "epoch": 2.490997314453125e-05,
      "grad_norm": 0.15292678773403168,
      "learning_rate": 8.829383630556067e-06,
      "loss": 0.0612,
      "step": 26120
    },
    {
      "epoch": 2.490997314453125e-05,
      "model_forward_time": 1.7226316928863525,
      "step": 26120
    },
    {
      "dynamics_loss_avg": 0.07040216103196144,
      "epoch": 2.490997314453125e-05,
      "step": 26120
    },
    {
      "action_loss_avg": 0.0028083729383070024,
      "epoch": 2.490997314453125e-05,
      "step": 26120
    },
    {
      "epoch": 2.490997314453125e-05,
      "step": 26120,
      "training_step_time": 7.664460897445679
    },
    {
      "epoch": 2.4910926818847655e-05,
      "model_forward_time": 1.727175235748291,
      "step": 26121
    },
    {
      "epoch": 2.4910926818847655e-05,
      "step": 26121,
      "training_step_time": 8.15271258354187
    },
    {
      "epoch": 2.4911880493164062e-05,
      "model_forward_time": 1.7276687622070312,
      "step": 26122
    },
    {
      "epoch": 2.4911880493164062e-05,
      "step": 26122,
      "training_step_time": 7.719446420669556
    },
    {
      "epoch": 2.491283416748047e-05,
      "model_forward_time": 1.720245122909546,
      "step": 26123
    },
    {
      "epoch": 2.491283416748047e-05,
      "step": 26123,
      "training_step_time": 9.149915933609009
    },
    {
      "epoch": 2.4913787841796874e-05,
      "model_forward_time": 1.7448241710662842,
      "step": 26124
    },
    {
      "epoch": 2.4913787841796874e-05,
      "step": 26124,
      "training_step_time": 8.415217638015747
    },
    {
      "epoch": 2.491474151611328e-05,
      "model_forward_time": 1.743995189666748,
      "step": 26125
    },
    {
      "epoch": 2.491474151611328e-05,
      "step": 26125,
      "training_step_time": 7.7261669635772705
    },
    {
      "epoch": 2.491569519042969e-05,
      "model_forward_time": 1.726736307144165,
      "step": 26126
    },
    {
      "epoch": 2.491569519042969e-05,
      "step": 26126,
      "training_step_time": 7.882989406585693
    },
    {
      "epoch": 2.4916648864746093e-05,
      "model_forward_time": 1.723886489868164,
      "step": 26127
    },
    {
      "epoch": 2.4916648864746093e-05,
      "step": 26127,
      "training_step_time": 8.458343982696533
    },
    {
      "epoch": 2.49176025390625e-05,
      "model_forward_time": 1.7315049171447754,
      "step": 26128
    },
    {
      "epoch": 2.49176025390625e-05,
      "step": 26128,
      "training_step_time": 7.548013687133789
    },
    {
      "epoch": 2.4918556213378908e-05,
      "model_forward_time": 1.7208847999572754,
      "step": 26129
    },
    {
      "epoch": 2.4918556213378908e-05,
      "step": 26129,
      "training_step_time": 7.521117210388184
    },
    {
      "epoch": 2.491950988769531e-05,
      "grad_norm": 0.1290828436613083,
      "learning_rate": 8.828320260872207e-06,
      "loss": 0.0626,
      "step": 26130
    },
    {
      "epoch": 2.491950988769531e-05,
      "model_forward_time": 1.7243289947509766,
      "step": 26130
    },
    {
      "dynamics_loss_avg": 0.052102820947766304,
      "epoch": 2.491950988769531e-05,
      "step": 26130
    },
    {
      "action_loss_avg": 0.00239855496911332,
      "epoch": 2.491950988769531e-05,
      "step": 26130
    },
    {
      "epoch": 2.491950988769531e-05,
      "step": 26130,
      "training_step_time": 7.493250370025635
    },
    {
      "epoch": 2.492046356201172e-05,
      "model_forward_time": 1.731309175491333,
      "step": 26131
    },
    {
      "epoch": 2.492046356201172e-05,
      "step": 26131,
      "training_step_time": 7.615786790847778
    },
    {
      "epoch": 2.4921417236328126e-05,
      "model_forward_time": 1.7257347106933594,
      "step": 26132
    },
    {
      "epoch": 2.4921417236328126e-05,
      "step": 26132,
      "training_step_time": 8.045024394989014
    },
    {
      "epoch": 2.492237091064453e-05,
      "model_forward_time": 1.7447781562805176,
      "step": 26133
    },
    {
      "epoch": 2.492237091064453e-05,
      "step": 26133,
      "training_step_time": 9.50786566734314
    },
    {
      "epoch": 2.4923324584960938e-05,
      "model_forward_time": 1.745621681213379,
      "step": 26134
    },
    {
      "epoch": 2.4923324584960938e-05,
      "step": 26134,
      "training_step_time": 8.408136367797852
    },
    {
      "epoch": 2.4924278259277345e-05,
      "model_forward_time": 1.7279393672943115,
      "step": 26135
    },
    {
      "epoch": 2.4924278259277345e-05,
      "step": 26135,
      "training_step_time": 8.658977270126343
    },
    {
      "epoch": 2.492523193359375e-05,
      "model_forward_time": 1.7278389930725098,
      "step": 26136
    },
    {
      "epoch": 2.492523193359375e-05,
      "step": 26136,
      "training_step_time": 8.265112400054932
    },
    {
      "epoch": 2.4926185607910157e-05,
      "model_forward_time": 1.7202167510986328,
      "step": 26137
    },
    {
      "epoch": 2.4926185607910157e-05,
      "step": 26137,
      "training_step_time": 8.430611848831177
    },
    {
      "epoch": 2.4927139282226564e-05,
      "model_forward_time": 1.7117772102355957,
      "step": 26138
    },
    {
      "epoch": 2.4927139282226564e-05,
      "step": 26138,
      "training_step_time": 7.439134120941162
    },
    {
      "epoch": 2.4928092956542968e-05,
      "model_forward_time": 1.7295312881469727,
      "step": 26139
    },
    {
      "epoch": 2.4928092956542968e-05,
      "step": 26139,
      "training_step_time": 9.102053880691528
    },
    {
      "epoch": 2.4929046630859376e-05,
      "grad_norm": 0.15894904732704163,
      "learning_rate": 8.827256472529e-06,
      "loss": 0.0631,
      "step": 26140
    },
    {
      "epoch": 2.4929046630859376e-05,
      "model_forward_time": 1.7248778343200684,
      "step": 26140
    },
    {
      "dynamics_loss_avg": 0.07147791804745793,
      "epoch": 2.4929046630859376e-05,
      "step": 26140
    },
    {
      "action_loss_avg": 0.004134652108768932,
      "epoch": 2.4929046630859376e-05,
      "step": 26140
    },
    {
      "epoch": 2.4929046630859376e-05,
      "step": 26140,
      "training_step_time": 7.625586986541748
    },
    {
      "epoch": 2.493000030517578e-05,
      "model_forward_time": 1.7268526554107666,
      "step": 26141
    },
    {
      "epoch": 2.493000030517578e-05,
      "step": 26141,
      "training_step_time": 8.138059377670288
    },
    {
      "epoch": 2.4930953979492187e-05,
      "model_forward_time": 1.7419378757476807,
      "step": 26142
    },
    {
      "epoch": 2.4930953979492187e-05,
      "step": 26142,
      "training_step_time": 7.6813437938690186
    },
    {
      "epoch": 2.4931907653808594e-05,
      "model_forward_time": 1.7479043006896973,
      "step": 26143
    },
    {
      "epoch": 2.4931907653808594e-05,
      "step": 26143,
      "training_step_time": 8.531516075134277
    },
    {
      "epoch": 2.4932861328125e-05,
      "model_forward_time": 1.7273168563842773,
      "step": 26144
    },
    {
      "epoch": 2.4932861328125e-05,
      "step": 26144,
      "training_step_time": 7.648025035858154
    },
    {
      "epoch": 2.4933815002441406e-05,
      "model_forward_time": 1.7274749279022217,
      "step": 26145
    },
    {
      "epoch": 2.4933815002441406e-05,
      "step": 26145,
      "training_step_time": 8.588029146194458
    },
    {
      "epoch": 2.4934768676757813e-05,
      "model_forward_time": 1.7309815883636475,
      "step": 26146
    },
    {
      "epoch": 2.4934768676757813e-05,
      "step": 26146,
      "training_step_time": 7.7017598152160645
    },
    {
      "epoch": 2.4935722351074217e-05,
      "model_forward_time": 1.721311092376709,
      "step": 26147
    },
    {
      "epoch": 2.4935722351074217e-05,
      "step": 26147,
      "training_step_time": 6.9988853931427
    },
    {
      "epoch": 2.4936676025390625e-05,
      "model_forward_time": 1.7381486892700195,
      "step": 26148
    },
    {
      "epoch": 2.4936676025390625e-05,
      "step": 26148,
      "training_step_time": 7.595582723617554
    },
    {
      "epoch": 2.4937629699707032e-05,
      "model_forward_time": 1.7290658950805664,
      "step": 26149
    },
    {
      "epoch": 2.4937629699707032e-05,
      "step": 26149,
      "training_step_time": 7.639132261276245
    },
    {
      "epoch": 2.4938583374023436e-05,
      "grad_norm": 0.19659768044948578,
      "learning_rate": 8.826192265642778e-06,
      "loss": 0.0619,
      "step": 26150
    },
    {
      "epoch": 2.4938583374023436e-05,
      "model_forward_time": 1.7276999950408936,
      "step": 26150
    },
    {
      "dynamics_loss_avg": 0.057042353041470054,
      "epoch": 2.4938583374023436e-05,
      "step": 26150
    },
    {
      "action_loss_avg": 0.003167193732224405,
      "epoch": 2.4938583374023436e-05,
      "step": 26150
    },
    {
      "epoch": 2.4938583374023436e-05,
      "step": 26150,
      "training_step_time": 7.636049032211304
    },
    {
      "epoch": 2.4939537048339844e-05,
      "model_forward_time": 2.987407922744751,
      "step": 26151
    },
    {
      "epoch": 2.4939537048339844e-05,
      "step": 26151,
      "training_step_time": 8.595699787139893
    },
    {
      "epoch": 2.494049072265625e-05,
      "model_forward_time": 1.7385141849517822,
      "step": 26152
    },
    {
      "epoch": 2.494049072265625e-05,
      "step": 26152,
      "training_step_time": 8.277090787887573
    },
    {
      "epoch": 2.4941444396972655e-05,
      "model_forward_time": 1.7278659343719482,
      "step": 26153
    },
    {
      "epoch": 2.4941444396972655e-05,
      "step": 26153,
      "training_step_time": 9.396112442016602
    },
    {
      "epoch": 2.4942398071289062e-05,
      "model_forward_time": 1.7269775867462158,
      "step": 26154
    },
    {
      "epoch": 2.4942398071289062e-05,
      "step": 26154,
      "training_step_time": 8.488501787185669
    },
    {
      "epoch": 2.494335174560547e-05,
      "model_forward_time": 1.729264259338379,
      "step": 26155
    },
    {
      "epoch": 2.494335174560547e-05,
      "step": 26155,
      "training_step_time": 8.16992998123169
    },
    {
      "epoch": 2.4944305419921874e-05,
      "model_forward_time": 1.728100299835205,
      "step": 26156
    },
    {
      "epoch": 2.4944305419921874e-05,
      "step": 26156,
      "training_step_time": 7.539815425872803
    },
    {
      "epoch": 2.494525909423828e-05,
      "model_forward_time": 1.726902961730957,
      "step": 26157
    },
    {
      "epoch": 2.494525909423828e-05,
      "step": 26157,
      "training_step_time": 7.524762868881226
    },
    {
      "epoch": 2.494621276855469e-05,
      "model_forward_time": 1.7320196628570557,
      "step": 26158
    },
    {
      "epoch": 2.494621276855469e-05,
      "step": 26158,
      "training_step_time": 8.143254518508911
    },
    {
      "epoch": 2.4947166442871093e-05,
      "model_forward_time": 1.7256996631622314,
      "step": 26159
    },
    {
      "epoch": 2.4947166442871093e-05,
      "step": 26159,
      "training_step_time": 7.602630853652954
    },
    {
      "epoch": 2.49481201171875e-05,
      "grad_norm": 0.16697543859481812,
      "learning_rate": 8.825127640329923e-06,
      "loss": 0.0584,
      "step": 26160
    },
    {
      "epoch": 2.49481201171875e-05,
      "model_forward_time": 2.929612874984741,
      "step": 26160
    },
    {
      "dynamics_loss_avg": 0.05191420707851648,
      "epoch": 2.49481201171875e-05,
      "step": 26160
    },
    {
      "action_loss_avg": 0.0025008938391692936,
      "epoch": 2.49481201171875e-05,
      "step": 26160
    },
    {
      "epoch": 2.49481201171875e-05,
      "step": 26160,
      "training_step_time": 9.54931092262268
    },
    {
      "epoch": 2.4949073791503908e-05,
      "model_forward_time": 1.7460734844207764,
      "step": 26161
    },
    {
      "epoch": 2.4949073791503908e-05,
      "step": 26161,
      "training_step_time": 7.561345338821411
    },
    {
      "epoch": 2.495002746582031e-05,
      "model_forward_time": 1.726745367050171,
      "step": 26162
    },
    {
      "epoch": 2.495002746582031e-05,
      "step": 26162,
      "training_step_time": 8.892227172851562
    },
    {
      "epoch": 2.495098114013672e-05,
      "model_forward_time": 1.7244136333465576,
      "step": 26163
    },
    {
      "epoch": 2.495098114013672e-05,
      "step": 26163,
      "training_step_time": 9.530527830123901
    },
    {
      "epoch": 2.4951934814453126e-05,
      "model_forward_time": 1.725327730178833,
      "step": 26164
    },
    {
      "epoch": 2.4951934814453126e-05,
      "step": 26164,
      "training_step_time": 7.749398469924927
    },
    {
      "epoch": 2.495288848876953e-05,
      "model_forward_time": 1.7279176712036133,
      "step": 26165
    },
    {
      "epoch": 2.495288848876953e-05,
      "step": 26165,
      "training_step_time": 7.571133613586426
    },
    {
      "epoch": 2.4953842163085938e-05,
      "model_forward_time": 1.718381404876709,
      "step": 26166
    },
    {
      "epoch": 2.4953842163085938e-05,
      "step": 26166,
      "training_step_time": 8.094509363174438
    },
    {
      "epoch": 2.4954795837402345e-05,
      "model_forward_time": 1.7292754650115967,
      "step": 26167
    },
    {
      "epoch": 2.4954795837402345e-05,
      "step": 26167,
      "training_step_time": 8.608467102050781
    },
    {
      "epoch": 2.495574951171875e-05,
      "model_forward_time": 1.7141218185424805,
      "step": 26168
    },
    {
      "epoch": 2.495574951171875e-05,
      "step": 26168,
      "training_step_time": 7.6481194496154785
    },
    {
      "epoch": 2.4956703186035157e-05,
      "model_forward_time": 2.8493247032165527,
      "step": 26169
    },
    {
      "epoch": 2.4956703186035157e-05,
      "step": 26169,
      "training_step_time": 8.68133282661438
    },
    {
      "epoch": 2.4957656860351564e-05,
      "grad_norm": 0.11275291442871094,
      "learning_rate": 8.824062596706861e-06,
      "loss": 0.0632,
      "step": 26170
    },
    {
      "epoch": 2.4957656860351564e-05,
      "model_forward_time": 1.82094144821167,
      "step": 26170
    },
    {
      "dynamics_loss_avg": 0.059562383219599724,
      "epoch": 2.4957656860351564e-05,
      "step": 26170
    },
    {
      "action_loss_avg": 0.0036038137855939566,
      "epoch": 2.4957656860351564e-05,
      "step": 26170
    },
    {
      "epoch": 2.4957656860351564e-05,
      "step": 26170,
      "training_step_time": 7.616314888000488
    },
    {
      "epoch": 2.4958610534667968e-05,
      "model_forward_time": 1.7284982204437256,
      "step": 26171
    },
    {
      "epoch": 2.4958610534667968e-05,
      "step": 26171,
      "training_step_time": 7.608084201812744
    },
    {
      "epoch": 2.4959564208984376e-05,
      "model_forward_time": 1.727663516998291,
      "step": 26172
    },
    {
      "epoch": 2.4959564208984376e-05,
      "step": 26172,
      "training_step_time": 7.6190338134765625
    },
    {
      "epoch": 2.496051788330078e-05,
      "model_forward_time": 1.733483076095581,
      "step": 26173
    },
    {
      "epoch": 2.496051788330078e-05,
      "step": 26173,
      "training_step_time": 7.790558099746704
    },
    {
      "epoch": 2.4961471557617187e-05,
      "model_forward_time": 1.7269339561462402,
      "step": 26174
    },
    {
      "epoch": 2.4961471557617187e-05,
      "step": 26174,
      "training_step_time": 7.5249924659729
    },
    {
      "epoch": 2.4962425231933594e-05,
      "model_forward_time": 1.7324743270874023,
      "step": 26175
    },
    {
      "epoch": 2.4962425231933594e-05,
      "step": 26175,
      "training_step_time": 7.549037456512451
    },
    {
      "epoch": 2.496337890625e-05,
      "model_forward_time": 1.7284958362579346,
      "step": 26176
    },
    {
      "epoch": 2.496337890625e-05,
      "step": 26176,
      "training_step_time": 7.625939846038818
    },
    {
      "epoch": 2.4964332580566406e-05,
      "model_forward_time": 1.7228412628173828,
      "step": 26177
    },
    {
      "epoch": 2.4964332580566406e-05,
      "step": 26177,
      "training_step_time": 8.150387048721313
    },
    {
      "epoch": 2.4965286254882813e-05,
      "model_forward_time": 2.8931305408477783,
      "step": 26178
    },
    {
      "epoch": 2.4965286254882813e-05,
      "step": 26178,
      "training_step_time": 8.14591383934021
    },
    {
      "epoch": 2.4966239929199217e-05,
      "model_forward_time": 1.759596586227417,
      "step": 26179
    },
    {
      "epoch": 2.4966239929199217e-05,
      "step": 26179,
      "training_step_time": 8.171674251556396
    },
    {
      "epoch": 2.4967193603515625e-05,
      "grad_norm": 0.162335604429245,
      "learning_rate": 8.822997134890063e-06,
      "loss": 0.0631,
      "step": 26180
    },
    {
      "epoch": 2.4967193603515625e-05,
      "model_forward_time": 1.7318158149719238,
      "step": 26180
    },
    {
      "dynamics_loss_avg": 0.06746781282126904,
      "epoch": 2.4967193603515625e-05,
      "step": 26180
    },
    {
      "action_loss_avg": 0.0027423907886259258,
      "epoch": 2.4967193603515625e-05,
      "step": 26180
    },
    {
      "epoch": 2.4967193603515625e-05,
      "step": 26180,
      "training_step_time": 7.646693468093872
    },
    {
      "epoch": 2.4968147277832032e-05,
      "model_forward_time": 1.7329206466674805,
      "step": 26181
    },
    {
      "epoch": 2.4968147277832032e-05,
      "step": 26181,
      "training_step_time": 8.132404327392578
    },
    {
      "epoch": 2.4969100952148436e-05,
      "model_forward_time": 1.723543643951416,
      "step": 26182
    },
    {
      "epoch": 2.4969100952148436e-05,
      "step": 26182,
      "training_step_time": 7.844196557998657
    },
    {
      "epoch": 2.4970054626464844e-05,
      "model_forward_time": 1.7292077541351318,
      "step": 26183
    },
    {
      "epoch": 2.4970054626464844e-05,
      "step": 26183,
      "training_step_time": 7.552131414413452
    },
    {
      "epoch": 2.497100830078125e-05,
      "model_forward_time": 1.7328376770019531,
      "step": 26184
    },
    {
      "epoch": 2.497100830078125e-05,
      "step": 26184,
      "training_step_time": 7.480625629425049
    },
    {
      "epoch": 2.4971961975097655e-05,
      "model_forward_time": 1.7273471355438232,
      "step": 26185
    },
    {
      "epoch": 2.4971961975097655e-05,
      "step": 26185,
      "training_step_time": 8.096060276031494
    },
    {
      "epoch": 2.4972915649414062e-05,
      "model_forward_time": 1.7258992195129395,
      "step": 26186
    },
    {
      "epoch": 2.4972915649414062e-05,
      "step": 26186,
      "training_step_time": 7.485703229904175
    },
    {
      "epoch": 2.497386932373047e-05,
      "model_forward_time": 2.8580281734466553,
      "step": 26187
    },
    {
      "epoch": 2.497386932373047e-05,
      "step": 26187,
      "training_step_time": 8.630163669586182
    },
    {
      "epoch": 2.4974822998046874e-05,
      "model_forward_time": 1.7441368103027344,
      "step": 26188
    },
    {
      "epoch": 2.4974822998046874e-05,
      "step": 26188,
      "training_step_time": 8.546925067901611
    },
    {
      "epoch": 2.497577667236328e-05,
      "model_forward_time": 1.7292835712432861,
      "step": 26189
    },
    {
      "epoch": 2.497577667236328e-05,
      "step": 26189,
      "training_step_time": 9.462761163711548
    },
    {
      "epoch": 2.497673034667969e-05,
      "grad_norm": 0.15255185961723328,
      "learning_rate": 8.821931254996044e-06,
      "loss": 0.0629,
      "step": 26190
    },
    {
      "epoch": 2.497673034667969e-05,
      "model_forward_time": 1.7327687740325928,
      "step": 26190
    },
    {
      "dynamics_loss_avg": 0.0426748531870544,
      "epoch": 2.497673034667969e-05,
      "step": 26190
    },
    {
      "action_loss_avg": 0.0026445537514518946,
      "epoch": 2.497673034667969e-05,
      "step": 26190
    },
    {
      "epoch": 2.497673034667969e-05,
      "step": 26190,
      "training_step_time": 8.248928546905518
    },
    {
      "epoch": 2.4977684020996093e-05,
      "model_forward_time": 1.729306697845459,
      "step": 26191
    },
    {
      "epoch": 2.4977684020996093e-05,
      "step": 26191,
      "training_step_time": 8.401537895202637
    },
    {
      "epoch": 2.49786376953125e-05,
      "model_forward_time": 1.726299524307251,
      "step": 26192
    },
    {
      "epoch": 2.49786376953125e-05,
      "step": 26192,
      "training_step_time": 8.282185077667236
    },
    {
      "epoch": 2.4979591369628908e-05,
      "model_forward_time": 1.7248616218566895,
      "step": 26193
    },
    {
      "epoch": 2.4979591369628908e-05,
      "step": 26193,
      "training_step_time": 7.55274772644043
    },
    {
      "epoch": 2.498054504394531e-05,
      "model_forward_time": 1.7277002334594727,
      "step": 26194
    },
    {
      "epoch": 2.498054504394531e-05,
      "step": 26194,
      "training_step_time": 7.575062990188599
    },
    {
      "epoch": 2.498149871826172e-05,
      "model_forward_time": 1.7318966388702393,
      "step": 26195
    },
    {
      "epoch": 2.498149871826172e-05,
      "step": 26195,
      "training_step_time": 8.159119367599487
    },
    {
      "epoch": 2.4982452392578126e-05,
      "model_forward_time": 1.7303450107574463,
      "step": 26196
    },
    {
      "epoch": 2.4982452392578126e-05,
      "step": 26196,
      "training_step_time": 8.151947975158691
    },
    {
      "epoch": 2.498340606689453e-05,
      "model_forward_time": 1.7494885921478271,
      "step": 26197
    },
    {
      "epoch": 2.498340606689453e-05,
      "step": 26197,
      "training_step_time": 7.64915657043457
    },
    {
      "epoch": 2.4984359741210938e-05,
      "model_forward_time": 1.731513261795044,
      "step": 26198
    },
    {
      "epoch": 2.4984359741210938e-05,
      "step": 26198,
      "training_step_time": 9.673654317855835
    },
    {
      "epoch": 2.4985313415527345e-05,
      "model_forward_time": 1.7271842956542969,
      "step": 26199
    },
    {
      "epoch": 2.4985313415527345e-05,
      "step": 26199,
      "training_step_time": 10.135128259658813
    },
    {
      "epoch": 2.498626708984375e-05,
      "grad_norm": 0.13343501091003418,
      "learning_rate": 8.820864957141371e-06,
      "loss": 0.0605,
      "step": 26200
    },
    {
      "epoch": 2.498626708984375e-05,
      "model_forward_time": 1.7232062816619873,
      "step": 26200
    },
    {
      "dynamics_loss_avg": 0.04653594810515642,
      "epoch": 2.498626708984375e-05,
      "step": 26200
    },
    {
      "action_loss_avg": 0.0020168424176517873,
      "epoch": 2.498626708984375e-05,
      "step": 26200
    },
    {
      "epoch": 2.498626708984375e-05,
      "step": 26200,
      "training_step_time": 8.295233011245728
    },
    {
      "epoch": 2.4987220764160157e-05,
      "model_forward_time": 1.724393606185913,
      "step": 26201
    },
    {
      "epoch": 2.4987220764160157e-05,
      "step": 26201,
      "training_step_time": 8.768350601196289
    },
    {
      "epoch": 2.4988174438476564e-05,
      "model_forward_time": 1.7292211055755615,
      "step": 26202
    },
    {
      "epoch": 2.4988174438476564e-05,
      "step": 26202,
      "training_step_time": 7.61430025100708
    },
    {
      "epoch": 2.4989128112792968e-05,
      "model_forward_time": 1.7304613590240479,
      "step": 26203
    },
    {
      "epoch": 2.4989128112792968e-05,
      "step": 26203,
      "training_step_time": 7.527939081192017
    },
    {
      "epoch": 2.4990081787109376e-05,
      "model_forward_time": 1.725879192352295,
      "step": 26204
    },
    {
      "epoch": 2.4990081787109376e-05,
      "step": 26204,
      "training_step_time": 7.613981008529663
    },
    {
      "epoch": 2.499103546142578e-05,
      "model_forward_time": 1.7261381149291992,
      "step": 26205
    },
    {
      "epoch": 2.499103546142578e-05,
      "step": 26205,
      "training_step_time": 7.636200189590454
    },
    {
      "epoch": 2.4991989135742187e-05,
      "model_forward_time": 2.8521554470062256,
      "step": 26206
    },
    {
      "epoch": 2.4991989135742187e-05,
      "step": 26206,
      "training_step_time": 8.56273365020752
    },
    {
      "epoch": 2.4992942810058594e-05,
      "model_forward_time": 1.7330844402313232,
      "step": 26207
    },
    {
      "epoch": 2.4992942810058594e-05,
      "step": 26207,
      "training_step_time": 8.26925277709961
    },
    {
      "epoch": 2.4993896484375e-05,
      "model_forward_time": 1.7293167114257812,
      "step": 26208
    },
    {
      "epoch": 2.4993896484375e-05,
      "step": 26208,
      "training_step_time": 8.586430311203003
    },
    {
      "epoch": 2.4994850158691406e-05,
      "model_forward_time": 1.7277629375457764,
      "step": 26209
    },
    {
      "epoch": 2.4994850158691406e-05,
      "step": 26209,
      "training_step_time": 8.498096227645874
    },
    {
      "epoch": 2.4995803833007813e-05,
      "grad_norm": 0.15792283415794373,
      "learning_rate": 8.81979824144265e-06,
      "loss": 0.0626,
      "step": 26210
    },
    {
      "epoch": 2.4995803833007813e-05,
      "model_forward_time": 1.7252271175384521,
      "step": 26210
    },
    {
      "dynamics_loss_avg": 0.05850144829601049,
      "epoch": 2.4995803833007813e-05,
      "step": 26210
    },
    {
      "action_loss_avg": 0.002564325404819101,
      "epoch": 2.4995803833007813e-05,
      "step": 26210
    },
    {
      "epoch": 2.4995803833007813e-05,
      "step": 26210,
      "training_step_time": 7.8901472091674805
    },
    {
      "epoch": 2.4996757507324217e-05,
      "model_forward_time": 1.7239363193511963,
      "step": 26211
    },
    {
      "epoch": 2.4996757507324217e-05,
      "step": 26211,
      "training_step_time": 7.688344955444336
    },
    {
      "epoch": 2.4997711181640625e-05,
      "model_forward_time": 1.7214610576629639,
      "step": 26212
    },
    {
      "epoch": 2.4997711181640625e-05,
      "step": 26212,
      "training_step_time": 7.619613885879517
    },
    {
      "epoch": 2.4998664855957032e-05,
      "model_forward_time": 1.7324309349060059,
      "step": 26213
    },
    {
      "epoch": 2.4998664855957032e-05,
      "step": 26213,
      "training_step_time": 7.6824071407318115
    },
    {
      "epoch": 2.4999618530273436e-05,
      "model_forward_time": 1.0137903690338135,
      "step": 26214
    },
    {
      "epoch": 2.4999618530273436e-05,
      "step": 26214,
      "training_step_time": 7.584251642227173
    },
    {
      "epoch": 2.5000572204589844e-05,
      "model_forward_time": 2.851402521133423,
      "step": 26215
    },
    {
      "epoch": 2.5000572204589844e-05,
      "step": 26215,
      "training_step_time": 8.14781928062439
    },
    {
      "epoch": 2.500152587890625e-05,
      "model_forward_time": 1.7371439933776855,
      "step": 26216
    },
    {
      "epoch": 2.500152587890625e-05,
      "step": 26216,
      "training_step_time": 7.713176488876343
    },
    {
      "epoch": 2.5002479553222655e-05,
      "model_forward_time": 1.0236666202545166,
      "step": 26217
    },
    {
      "epoch": 2.5002479553222655e-05,
      "step": 26217,
      "training_step_time": 8.65199589729309
    },
    {
      "epoch": 2.5003433227539062e-05,
      "model_forward_time": 1.7265217304229736,
      "step": 26218
    },
    {
      "epoch": 2.5003433227539062e-05,
      "step": 26218,
      "training_step_time": 7.639275312423706
    },
    {
      "epoch": 2.500438690185547e-05,
      "model_forward_time": 1.7243502140045166,
      "step": 26219
    },
    {
      "epoch": 2.500438690185547e-05,
      "step": 26219,
      "training_step_time": 9.877158880233765
    },
    {
      "epoch": 2.5005340576171874e-05,
      "grad_norm": 0.1521034687757492,
      "learning_rate": 8.818731108016535e-06,
      "loss": 0.0668,
      "step": 26220
    },
    {
      "epoch": 2.5005340576171874e-05,
      "model_forward_time": 1.7204711437225342,
      "step": 26220
    },
    {
      "dynamics_loss_avg": 0.07013738006353379,
      "epoch": 2.5005340576171874e-05,
      "step": 26220
    },
    {
      "action_loss_avg": 0.002324450493324548,
      "epoch": 2.5005340576171874e-05,
      "step": 26220
    },
    {
      "epoch": 2.5005340576171874e-05,
      "step": 26220,
      "training_step_time": 8.76936388015747
    },
    {
      "epoch": 2.500629425048828e-05,
      "model_forward_time": 1.7245604991912842,
      "step": 26221
    },
    {
      "epoch": 2.500629425048828e-05,
      "step": 26221,
      "training_step_time": 8.639660596847534
    },
    {
      "epoch": 2.500724792480469e-05,
      "model_forward_time": 1.7242894172668457,
      "step": 26222
    },
    {
      "epoch": 2.500724792480469e-05,
      "step": 26222,
      "training_step_time": 7.74267053604126
    },
    {
      "epoch": 2.5008201599121093e-05,
      "model_forward_time": 1.7249586582183838,
      "step": 26223
    },
    {
      "epoch": 2.5008201599121093e-05,
      "step": 26223,
      "training_step_time": 7.4949750900268555
    },
    {
      "epoch": 2.50091552734375e-05,
      "model_forward_time": 1.7553596496582031,
      "step": 26224
    },
    {
      "epoch": 2.50091552734375e-05,
      "step": 26224,
      "training_step_time": 7.807475805282593
    },
    {
      "epoch": 2.5010108947753908e-05,
      "model_forward_time": 3.025637149810791,
      "step": 26225
    },
    {
      "epoch": 2.5010108947753908e-05,
      "step": 26225,
      "training_step_time": 7.967555284500122
    },
    {
      "epoch": 2.501106262207031e-05,
      "model_forward_time": 1.7266438007354736,
      "step": 26226
    },
    {
      "epoch": 2.501106262207031e-05,
      "step": 26226,
      "training_step_time": 7.803161144256592
    },
    {
      "epoch": 2.501201629638672e-05,
      "model_forward_time": 1.727313756942749,
      "step": 26227
    },
    {
      "epoch": 2.501201629638672e-05,
      "step": 26227,
      "training_step_time": 7.84702467918396
    },
    {
      "epoch": 2.5012969970703126e-05,
      "model_forward_time": 1.727869987487793,
      "step": 26228
    },
    {
      "epoch": 2.5012969970703126e-05,
      "step": 26228,
      "training_step_time": 7.770235061645508
    },
    {
      "epoch": 2.501392364501953e-05,
      "model_forward_time": 1.7352566719055176,
      "step": 26229
    },
    {
      "epoch": 2.501392364501953e-05,
      "step": 26229,
      "training_step_time": 7.958019971847534
    },
    {
      "epoch": 2.5014877319335938e-05,
      "grad_norm": 0.17470014095306396,
      "learning_rate": 8.81766355697973e-06,
      "loss": 0.0622,
      "step": 26230
    },
    {
      "epoch": 2.5014877319335938e-05,
      "model_forward_time": 1.7219040393829346,
      "step": 26230
    },
    {
      "dynamics_loss_avg": 0.0535304743796587,
      "epoch": 2.5014877319335938e-05,
      "step": 26230
    },
    {
      "action_loss_avg": 0.001853839319664985,
      "epoch": 2.5014877319335938e-05,
      "step": 26230
    },
    {
      "epoch": 2.5014877319335938e-05,
      "step": 26230,
      "training_step_time": 7.797879695892334
    },
    {
      "epoch": 2.5015830993652345e-05,
      "model_forward_time": 1.729987621307373,
      "step": 26231
    },
    {
      "epoch": 2.5015830993652345e-05,
      "step": 26231,
      "training_step_time": 7.827145099639893
    },
    {
      "epoch": 2.501678466796875e-05,
      "model_forward_time": 1.724050760269165,
      "step": 26232
    },
    {
      "epoch": 2.501678466796875e-05,
      "step": 26232,
      "training_step_time": 8.169293642044067
    },
    {
      "epoch": 2.5017738342285157e-05,
      "model_forward_time": 1.7423450946807861,
      "step": 26233
    },
    {
      "epoch": 2.5017738342285157e-05,
      "step": 26233,
      "training_step_time": 7.867898464202881
    },
    {
      "epoch": 2.5018692016601564e-05,
      "model_forward_time": 3.000537633895874,
      "step": 26234
    },
    {
      "epoch": 2.5018692016601564e-05,
      "step": 26234,
      "training_step_time": 8.815001964569092
    },
    {
      "epoch": 2.5019645690917968e-05,
      "model_forward_time": 1.7242152690887451,
      "step": 26235
    },
    {
      "epoch": 2.5019645690917968e-05,
      "step": 26235,
      "training_step_time": 8.799789905548096
    },
    {
      "epoch": 2.5020599365234376e-05,
      "model_forward_time": 1.7268152236938477,
      "step": 26236
    },
    {
      "epoch": 2.5020599365234376e-05,
      "step": 26236,
      "training_step_time": 8.567565441131592
    },
    {
      "epoch": 2.502155303955078e-05,
      "model_forward_time": 1.7240192890167236,
      "step": 26237
    },
    {
      "epoch": 2.502155303955078e-05,
      "step": 26237,
      "training_step_time": 8.810638427734375
    },
    {
      "epoch": 2.5022506713867187e-05,
      "model_forward_time": 1.7323286533355713,
      "step": 26238
    },
    {
      "epoch": 2.5022506713867187e-05,
      "step": 26238,
      "training_step_time": 7.966694116592407
    },
    {
      "epoch": 2.5023460388183594e-05,
      "model_forward_time": 1.726351261138916,
      "step": 26239
    },
    {
      "epoch": 2.5023460388183594e-05,
      "step": 26239,
      "training_step_time": 7.8297553062438965
    },
    {
      "epoch": 2.50244140625e-05,
      "grad_norm": 0.18082663416862488,
      "learning_rate": 8.816595588448977e-06,
      "loss": 0.0608,
      "step": 26240
    },
    {
      "epoch": 2.50244140625e-05,
      "model_forward_time": 1.7312870025634766,
      "step": 26240
    },
    {
      "dynamics_loss_avg": 0.048523404821753505,
      "epoch": 2.50244140625e-05,
      "step": 26240
    },
    {
      "action_loss_avg": 0.0018732006836216896,
      "epoch": 2.50244140625e-05,
      "step": 26240
    },
    {
      "epoch": 2.50244140625e-05,
      "step": 26240,
      "training_step_time": 7.738945484161377
    },
    {
      "epoch": 2.5025367736816406e-05,
      "model_forward_time": 1.7225677967071533,
      "step": 26241
    },
    {
      "epoch": 2.5025367736816406e-05,
      "step": 26241,
      "training_step_time": 7.526937007904053
    },
    {
      "epoch": 2.5026321411132813e-05,
      "model_forward_time": 1.7309529781341553,
      "step": 26242
    },
    {
      "epoch": 2.5026321411132813e-05,
      "step": 26242,
      "training_step_time": 7.727932691574097
    },
    {
      "epoch": 2.5027275085449217e-05,
      "model_forward_time": 2.8299973011016846,
      "step": 26243
    },
    {
      "epoch": 2.5027275085449217e-05,
      "step": 26243,
      "training_step_time": 7.785269737243652
    },
    {
      "epoch": 2.5028228759765625e-05,
      "model_forward_time": 1.7455697059631348,
      "step": 26244
    },
    {
      "epoch": 2.5028228759765625e-05,
      "step": 26244,
      "training_step_time": 8.729142427444458
    },
    {
      "epoch": 2.5029182434082032e-05,
      "model_forward_time": 1.7219574451446533,
      "step": 26245
    },
    {
      "epoch": 2.5029182434082032e-05,
      "step": 26245,
      "training_step_time": 8.732714891433716
    },
    {
      "epoch": 2.5030136108398436e-05,
      "model_forward_time": 1.7177746295928955,
      "step": 26246
    },
    {
      "epoch": 2.5030136108398436e-05,
      "step": 26246,
      "training_step_time": 7.556612730026245
    },
    {
      "epoch": 2.5031089782714844e-05,
      "model_forward_time": 1.7133581638336182,
      "step": 26247
    },
    {
      "epoch": 2.5031089782714844e-05,
      "step": 26247,
      "training_step_time": 8.44619083404541
    },
    {
      "epoch": 2.503204345703125e-05,
      "model_forward_time": 1.7169256210327148,
      "step": 26248
    },
    {
      "epoch": 2.503204345703125e-05,
      "step": 26248,
      "training_step_time": 7.569157123565674
    },
    {
      "epoch": 2.5032997131347655e-05,
      "model_forward_time": 1.7171151638031006,
      "step": 26249
    },
    {
      "epoch": 2.5032997131347655e-05,
      "step": 26249,
      "training_step_time": 7.552887678146362
    },
    {
      "epoch": 2.5033950805664062e-05,
      "grad_norm": 0.29200470447540283,
      "learning_rate": 8.81552720254107e-06,
      "loss": 0.063,
      "step": 26250
    },
    {
      "epoch": 2.5033950805664062e-05,
      "model_forward_time": 1.7253680229187012,
      "step": 26250
    },
    {
      "dynamics_loss_avg": 0.053288739360868934,
      "epoch": 2.5033950805664062e-05,
      "step": 26250
    },
    {
      "action_loss_avg": 0.002979805268114433,
      "epoch": 2.5033950805664062e-05,
      "step": 26250
    },
    {
      "epoch": 2.5033950805664062e-05,
      "step": 26250,
      "training_step_time": 7.4272987842559814
    },
    {
      "epoch": 2.503490447998047e-05,
      "model_forward_time": 1.7248492240905762,
      "step": 26251
    },
    {
      "epoch": 2.503490447998047e-05,
      "step": 26251,
      "training_step_time": 7.4791882038116455
    },
    {
      "epoch": 2.5035858154296874e-05,
      "model_forward_time": 2.818063259124756,
      "step": 26252
    },
    {
      "epoch": 2.5035858154296874e-05,
      "step": 26252,
      "training_step_time": 8.517330169677734
    },
    {
      "epoch": 2.503681182861328e-05,
      "model_forward_time": 1.7257936000823975,
      "step": 26253
    },
    {
      "epoch": 2.503681182861328e-05,
      "step": 26253,
      "training_step_time": 7.607778549194336
    },
    {
      "epoch": 2.503776550292969e-05,
      "model_forward_time": 1.7144584655761719,
      "step": 26254
    },
    {
      "epoch": 2.503776550292969e-05,
      "step": 26254,
      "training_step_time": 7.691803932189941
    },
    {
      "epoch": 2.5038719177246093e-05,
      "model_forward_time": 1.7092499732971191,
      "step": 26255
    },
    {
      "epoch": 2.5038719177246093e-05,
      "step": 26255,
      "training_step_time": 7.636714935302734
    },
    {
      "epoch": 2.50396728515625e-05,
      "model_forward_time": 1.7152361869812012,
      "step": 26256
    },
    {
      "epoch": 2.50396728515625e-05,
      "step": 26256,
      "training_step_time": 8.289217472076416
    },
    {
      "epoch": 2.5040626525878908e-05,
      "model_forward_time": 1.7103497982025146,
      "step": 26257
    },
    {
      "epoch": 2.5040626525878908e-05,
      "step": 26257,
      "training_step_time": 8.751252889633179
    },
    {
      "epoch": 2.504158020019531e-05,
      "model_forward_time": 1.7085599899291992,
      "step": 26258
    },
    {
      "epoch": 2.504158020019531e-05,
      "step": 26258,
      "training_step_time": 7.584180116653442
    },
    {
      "epoch": 2.504253387451172e-05,
      "model_forward_time": 1.7173957824707031,
      "step": 26259
    },
    {
      "epoch": 2.504253387451172e-05,
      "step": 26259,
      "training_step_time": 7.608794927597046
    },
    {
      "epoch": 2.5043487548828126e-05,
      "grad_norm": 0.16927316784858704,
      "learning_rate": 8.814458399372843e-06,
      "loss": 0.0666,
      "step": 26260
    },
    {
      "epoch": 2.5043487548828126e-05,
      "model_forward_time": 1.7357487678527832,
      "step": 26260
    },
    {
      "dynamics_loss_avg": 0.056601461954414845,
      "epoch": 2.5043487548828126e-05,
      "step": 26260
    },
    {
      "action_loss_avg": 0.002318214863771573,
      "epoch": 2.5043487548828126e-05,
      "step": 26260
    },
    {
      "epoch": 2.5043487548828126e-05,
      "step": 26260,
      "training_step_time": 7.468412637710571
    },
    {
      "epoch": 2.504444122314453e-05,
      "model_forward_time": 2.8176374435424805,
      "step": 26261
    },
    {
      "epoch": 2.504444122314453e-05,
      "step": 26261,
      "training_step_time": 7.637390375137329
    },
    {
      "epoch": 2.5045394897460938e-05,
      "model_forward_time": 1.7227025032043457,
      "step": 26262
    },
    {
      "epoch": 2.5045394897460938e-05,
      "step": 26262,
      "training_step_time": 7.598231792449951
    },
    {
      "epoch": 2.5046348571777345e-05,
      "model_forward_time": 1.7138278484344482,
      "step": 26263
    },
    {
      "epoch": 2.5046348571777345e-05,
      "step": 26263,
      "training_step_time": 8.311437129974365
    },
    {
      "epoch": 2.504730224609375e-05,
      "model_forward_time": 1.7154855728149414,
      "step": 26264
    },
    {
      "epoch": 2.504730224609375e-05,
      "step": 26264,
      "training_step_time": 7.646756887435913
    },
    {
      "epoch": 2.5048255920410157e-05,
      "model_forward_time": 1.711921215057373,
      "step": 26265
    },
    {
      "epoch": 2.5048255920410157e-05,
      "step": 26265,
      "training_step_time": 8.692440032958984
    },
    {
      "epoch": 2.504920959472656e-05,
      "model_forward_time": 1.7136693000793457,
      "step": 26266
    },
    {
      "epoch": 2.504920959472656e-05,
      "step": 26266,
      "training_step_time": 7.89653205871582
    },
    {
      "epoch": 2.5050163269042968e-05,
      "model_forward_time": 1.7085797786712646,
      "step": 26267
    },
    {
      "epoch": 2.5050163269042968e-05,
      "step": 26267,
      "training_step_time": 7.576086759567261
    },
    {
      "epoch": 2.5051116943359376e-05,
      "model_forward_time": 1.7145154476165771,
      "step": 26268
    },
    {
      "epoch": 2.5051116943359376e-05,
      "step": 26268,
      "training_step_time": 7.292582273483276
    },
    {
      "epoch": 2.505207061767578e-05,
      "model_forward_time": 1.7262589931488037,
      "step": 26269
    },
    {
      "epoch": 2.505207061767578e-05,
      "step": 26269,
      "training_step_time": 7.658358335494995
    },
    {
      "epoch": 2.5053024291992187e-05,
      "grad_norm": 0.18092328310012817,
      "learning_rate": 8.813389179061182e-06,
      "loss": 0.0605,
      "step": 26270
    },
    {
      "epoch": 2.5053024291992187e-05,
      "model_forward_time": 1.7090928554534912,
      "step": 26270
    },
    {
      "dynamics_loss_avg": 0.06297233533114195,
      "epoch": 2.5053024291992187e-05,
      "step": 26270
    },
    {
      "action_loss_avg": 0.003049201605608687,
      "epoch": 2.5053024291992187e-05,
      "step": 26270
    },
    {
      "epoch": 2.5053024291992187e-05,
      "step": 26270,
      "training_step_time": 9.149642705917358
    },
    {
      "epoch": 2.5053977966308595e-05,
      "model_forward_time": 1.7189369201660156,
      "step": 26271
    },
    {
      "epoch": 2.5053977966308595e-05,
      "step": 26271,
      "training_step_time": 7.671921253204346
    },
    {
      "epoch": 2.5054931640625e-05,
      "model_forward_time": 1.714521884918213,
      "step": 26272
    },
    {
      "epoch": 2.5054931640625e-05,
      "step": 26272,
      "training_step_time": 8.774066925048828
    },
    {
      "epoch": 2.5055885314941406e-05,
      "model_forward_time": 1.7122414112091064,
      "step": 26273
    },
    {
      "epoch": 2.5055885314941406e-05,
      "step": 26273,
      "training_step_time": 7.6125218868255615
    },
    {
      "epoch": 2.5056838989257813e-05,
      "model_forward_time": 1.7151529788970947,
      "step": 26274
    },
    {
      "epoch": 2.5056838989257813e-05,
      "step": 26274,
      "training_step_time": 8.37468695640564
    },
    {
      "epoch": 2.5057792663574217e-05,
      "model_forward_time": 1.7135038375854492,
      "step": 26275
    },
    {
      "epoch": 2.5057792663574217e-05,
      "step": 26275,
      "training_step_time": 7.839825868606567
    },
    {
      "epoch": 2.5058746337890625e-05,
      "model_forward_time": 1.711104393005371,
      "step": 26276
    },
    {
      "epoch": 2.5058746337890625e-05,
      "step": 26276,
      "training_step_time": 9.52185869216919
    },
    {
      "epoch": 2.5059700012207032e-05,
      "model_forward_time": 1.710846185684204,
      "step": 26277
    },
    {
      "epoch": 2.5059700012207032e-05,
      "step": 26277,
      "training_step_time": 7.40911340713501
    },
    {
      "epoch": 2.5060653686523436e-05,
      "model_forward_time": 1.7356898784637451,
      "step": 26278
    },
    {
      "epoch": 2.5060653686523436e-05,
      "step": 26278,
      "training_step_time": 8.368700742721558
    },
    {
      "epoch": 2.5061607360839844e-05,
      "model_forward_time": 1.7163188457489014,
      "step": 26279
    },
    {
      "epoch": 2.5061607360839844e-05,
      "step": 26279,
      "training_step_time": 7.676150560379028
    },
    {
      "epoch": 2.506256103515625e-05,
      "grad_norm": 0.14976802468299866,
      "learning_rate": 8.812319541723012e-06,
      "loss": 0.0645,
      "step": 26280
    },
    {
      "epoch": 2.506256103515625e-05,
      "model_forward_time": 2.9307196140289307,
      "step": 26280
    },
    {
      "dynamics_loss_avg": 0.07005872838199138,
      "epoch": 2.506256103515625e-05,
      "step": 26280
    },
    {
      "action_loss_avg": 0.003369357861811295,
      "epoch": 2.506256103515625e-05,
      "step": 26280
    },
    {
      "epoch": 2.506256103515625e-05,
      "step": 26280,
      "training_step_time": 8.260827779769897
    },
    {
      "epoch": 2.5063514709472655e-05,
      "model_forward_time": 1.724989891052246,
      "step": 26281
    },
    {
      "epoch": 2.5063514709472655e-05,
      "step": 26281,
      "training_step_time": 7.6517956256866455
    },
    {
      "epoch": 2.5064468383789063e-05,
      "model_forward_time": 1.7094697952270508,
      "step": 26282
    },
    {
      "epoch": 2.5064468383789063e-05,
      "step": 26282,
      "training_step_time": 9.22795820236206
    },
    {
      "epoch": 2.506542205810547e-05,
      "model_forward_time": 1.7118334770202637,
      "step": 26283
    },
    {
      "epoch": 2.506542205810547e-05,
      "step": 26283,
      "training_step_time": 8.300963163375854
    },
    {
      "epoch": 2.5066375732421874e-05,
      "model_forward_time": 1.7105612754821777,
      "step": 26284
    },
    {
      "epoch": 2.5066375732421874e-05,
      "step": 26284,
      "training_step_time": 7.641348361968994
    },
    {
      "epoch": 2.506732940673828e-05,
      "model_forward_time": 1.7154896259307861,
      "step": 26285
    },
    {
      "epoch": 2.506732940673828e-05,
      "step": 26285,
      "training_step_time": 8.924742698669434
    },
    {
      "epoch": 2.506828308105469e-05,
      "model_forward_time": 1.708634853363037,
      "step": 26286
    },
    {
      "epoch": 2.506828308105469e-05,
      "step": 26286,
      "training_step_time": 7.511705636978149
    },
    {
      "epoch": 2.5069236755371093e-05,
      "model_forward_time": 1.732682466506958,
      "step": 26287
    },
    {
      "epoch": 2.5069236755371093e-05,
      "step": 26287,
      "training_step_time": 7.629374742507935
    },
    {
      "epoch": 2.50701904296875e-05,
      "model_forward_time": 1.708589792251587,
      "step": 26288
    },
    {
      "epoch": 2.50701904296875e-05,
      "step": 26288,
      "training_step_time": 8.736694812774658
    },
    {
      "epoch": 2.5071144104003908e-05,
      "model_forward_time": 2.8673863410949707,
      "step": 26289
    },
    {
      "epoch": 2.5071144104003908e-05,
      "step": 26289,
      "training_step_time": 7.681634426116943
    },
    {
      "epoch": 2.507209777832031e-05,
      "grad_norm": 0.12507471442222595,
      "learning_rate": 8.811249487475308e-06,
      "loss": 0.0609,
      "step": 26290
    },
    {
      "epoch": 2.507209777832031e-05,
      "model_forward_time": 1.7075200080871582,
      "step": 26290
    },
    {
      "dynamics_loss_avg": 0.049229791201651094,
      "epoch": 2.507209777832031e-05,
      "step": 26290
    },
    {
      "action_loss_avg": 0.0044650242431089286,
      "epoch": 2.507209777832031e-05,
      "step": 26290
    },
    {
      "epoch": 2.507209777832031e-05,
      "step": 26290,
      "training_step_time": 7.638505935668945
    },
    {
      "epoch": 2.507305145263672e-05,
      "model_forward_time": 1.7147367000579834,
      "step": 26291
    },
    {
      "epoch": 2.507305145263672e-05,
      "step": 26291,
      "training_step_time": 8.677770376205444
    },
    {
      "epoch": 2.5074005126953127e-05,
      "model_forward_time": 1.7075788974761963,
      "step": 26292
    },
    {
      "epoch": 2.5074005126953127e-05,
      "step": 26292,
      "training_step_time": 8.54703950881958
    },
    {
      "epoch": 2.507495880126953e-05,
      "model_forward_time": 1.7127091884613037,
      "step": 26293
    },
    {
      "epoch": 2.507495880126953e-05,
      "step": 26293,
      "training_step_time": 7.654617786407471
    },
    {
      "epoch": 2.5075912475585938e-05,
      "model_forward_time": 1.716278076171875,
      "step": 26294
    },
    {
      "epoch": 2.5075912475585938e-05,
      "step": 26294,
      "training_step_time": 7.749913692474365
    },
    {
      "epoch": 2.5076866149902345e-05,
      "model_forward_time": 1.7037367820739746,
      "step": 26295
    },
    {
      "epoch": 2.5076866149902345e-05,
      "step": 26295,
      "training_step_time": 7.701727628707886
    },
    {
      "epoch": 2.507781982421875e-05,
      "model_forward_time": 1.73199462890625,
      "step": 26296
    },
    {
      "epoch": 2.507781982421875e-05,
      "step": 26296,
      "training_step_time": 7.582072496414185
    },
    {
      "epoch": 2.5078773498535157e-05,
      "model_forward_time": 1.3356389999389648,
      "step": 26297
    },
    {
      "epoch": 2.5078773498535157e-05,
      "step": 26297,
      "training_step_time": 7.654564619064331
    },
    {
      "epoch": 2.507972717285156e-05,
      "model_forward_time": 2.8395979404449463,
      "step": 26298
    },
    {
      "epoch": 2.507972717285156e-05,
      "step": 26298,
      "training_step_time": 7.647681951522827
    },
    {
      "epoch": 2.5080680847167968e-05,
      "model_forward_time": 1.7072296142578125,
      "step": 26299
    },
    {
      "epoch": 2.5080680847167968e-05,
      "step": 26299,
      "training_step_time": 8.368956327438354
    },
    {
      "epoch": 2.5081634521484376e-05,
      "grad_norm": 0.15964603424072266,
      "learning_rate": 8.810179016435092e-06,
      "loss": 0.0608,
      "step": 26300
    },
    {
      "epoch": 2.5081634521484376e-05,
      "model_forward_time": 1.7420687675476074,
      "step": 26300
    },
    {
      "dynamics_loss_avg": 0.06907858680933714,
      "epoch": 2.5081634521484376e-05,
      "step": 26300
    },
    {
      "action_loss_avg": 0.003724228497594595,
      "epoch": 2.5081634521484376e-05,
      "step": 26300
    },
    {
      "epoch": 2.5081634521484376e-05,
      "step": 26300,
      "training_step_time": 8.74143099784851
    },
    {
      "epoch": 2.508258819580078e-05,
      "model_forward_time": 1.7089900970458984,
      "step": 26301
    },
    {
      "epoch": 2.508258819580078e-05,
      "step": 26301,
      "training_step_time": 7.620795726776123
    },
    {
      "epoch": 2.5083541870117187e-05,
      "model_forward_time": 1.7102839946746826,
      "step": 26302
    },
    {
      "epoch": 2.5083541870117187e-05,
      "step": 26302,
      "training_step_time": 8.812552452087402
    },
    {
      "epoch": 2.5084495544433595e-05,
      "model_forward_time": 1.7205698490142822,
      "step": 26303
    },
    {
      "epoch": 2.5084495544433595e-05,
      "step": 26303,
      "training_step_time": 7.894201040267944
    },
    {
      "epoch": 2.508544921875e-05,
      "model_forward_time": 1.7288644313812256,
      "step": 26304
    },
    {
      "epoch": 2.508544921875e-05,
      "step": 26304,
      "training_step_time": 7.492722034454346
    },
    {
      "epoch": 2.5086402893066406e-05,
      "model_forward_time": 1.7543342113494873,
      "step": 26305
    },
    {
      "epoch": 2.5086402893066406e-05,
      "step": 26305,
      "training_step_time": 8.70784878730774
    },
    {
      "epoch": 2.5087356567382813e-05,
      "model_forward_time": 1.735767126083374,
      "step": 26306
    },
    {
      "epoch": 2.5087356567382813e-05,
      "step": 26306,
      "training_step_time": 7.667649507522583
    },
    {
      "epoch": 2.5088310241699217e-05,
      "model_forward_time": 2.836603879928589,
      "step": 26307
    },
    {
      "epoch": 2.5088310241699217e-05,
      "step": 26307,
      "training_step_time": 8.634018421173096
    },
    {
      "epoch": 2.5089263916015625e-05,
      "model_forward_time": 1.7265605926513672,
      "step": 26308
    },
    {
      "epoch": 2.5089263916015625e-05,
      "step": 26308,
      "training_step_time": 7.652677059173584
    },
    {
      "epoch": 2.5090217590332032e-05,
      "model_forward_time": 1.7451293468475342,
      "step": 26309
    },
    {
      "epoch": 2.5090217590332032e-05,
      "step": 26309,
      "training_step_time": 8.34107780456543
    },
    {
      "epoch": 2.5091171264648436e-05,
      "grad_norm": 0.17628313601016998,
      "learning_rate": 8.809108128719428e-06,
      "loss": 0.0615,
      "step": 26310
    },
    {
      "epoch": 2.5091171264648436e-05,
      "model_forward_time": 1.7329413890838623,
      "step": 26310
    },
    {
      "dynamics_loss_avg": 0.06004298888146877,
      "epoch": 2.5091171264648436e-05,
      "step": 26310
    },
    {
      "action_loss_avg": 0.002869499451480806,
      "epoch": 2.5091171264648436e-05,
      "step": 26310
    },
    {
      "epoch": 2.5091171264648436e-05,
      "step": 26310,
      "training_step_time": 8.598093032836914
    },
    {
      "epoch": 2.5092124938964844e-05,
      "model_forward_time": 1.7383577823638916,
      "step": 26311
    },
    {
      "epoch": 2.5092124938964844e-05,
      "step": 26311,
      "training_step_time": 8.358927011489868
    },
    {
      "epoch": 2.509307861328125e-05,
      "model_forward_time": 1.7384991645812988,
      "step": 26312
    },
    {
      "epoch": 2.509307861328125e-05,
      "step": 26312,
      "training_step_time": 8.448393821716309
    },
    {
      "epoch": 2.5094032287597655e-05,
      "model_forward_time": 1.7288355827331543,
      "step": 26313
    },
    {
      "epoch": 2.5094032287597655e-05,
      "step": 26313,
      "training_step_time": 7.581625461578369
    },
    {
      "epoch": 2.5094985961914063e-05,
      "model_forward_time": 1.743255853652954,
      "step": 26314
    },
    {
      "epoch": 2.5094985961914063e-05,
      "step": 26314,
      "training_step_time": 7.706333875656128
    },
    {
      "epoch": 2.509593963623047e-05,
      "model_forward_time": 1.7341067790985107,
      "step": 26315
    },
    {
      "epoch": 2.509593963623047e-05,
      "step": 26315,
      "training_step_time": 7.665026903152466
    },
    {
      "epoch": 2.5096893310546874e-05,
      "model_forward_time": 1.7403090000152588,
      "step": 26316
    },
    {
      "epoch": 2.5096893310546874e-05,
      "step": 26316,
      "training_step_time": 8.435725688934326
    },
    {
      "epoch": 2.509784698486328e-05,
      "model_forward_time": 2.9151318073272705,
      "step": 26317
    },
    {
      "epoch": 2.509784698486328e-05,
      "step": 26317,
      "training_step_time": 7.727531433105469
    },
    {
      "epoch": 2.509880065917969e-05,
      "model_forward_time": 1.7412843704223633,
      "step": 26318
    },
    {
      "epoch": 2.509880065917969e-05,
      "step": 26318,
      "training_step_time": 8.6831533908844
    },
    {
      "epoch": 2.5099754333496093e-05,
      "model_forward_time": 1.7291297912597656,
      "step": 26319
    },
    {
      "epoch": 2.5099754333496093e-05,
      "step": 26319,
      "training_step_time": 8.936370611190796
    },
    {
      "epoch": 2.51007080078125e-05,
      "grad_norm": 0.15643273293972015,
      "learning_rate": 8.808036824445424e-06,
      "loss": 0.0572,
      "step": 26320
    },
    {
      "epoch": 2.51007080078125e-05,
      "model_forward_time": 1.7338464260101318,
      "step": 26320
    },
    {
      "dynamics_loss_avg": 0.06099719256162643,
      "epoch": 2.51007080078125e-05,
      "step": 26320
    },
    {
      "action_loss_avg": 0.0018709497759118675,
      "epoch": 2.51007080078125e-05,
      "step": 26320
    },
    {
      "epoch": 2.51007080078125e-05,
      "step": 26320,
      "training_step_time": 8.652257919311523
    },
    {
      "epoch": 2.5101661682128908e-05,
      "model_forward_time": 1.7318241596221924,
      "step": 26321
    },
    {
      "epoch": 2.5101661682128908e-05,
      "step": 26321,
      "training_step_time": 7.972665786743164
    },
    {
      "epoch": 2.5102615356445312e-05,
      "model_forward_time": 1.736140251159668,
      "step": 26322
    },
    {
      "epoch": 2.5102615356445312e-05,
      "step": 26322,
      "training_step_time": 8.584275484085083
    },
    {
      "epoch": 2.510356903076172e-05,
      "model_forward_time": 1.7655658721923828,
      "step": 26323
    },
    {
      "epoch": 2.510356903076172e-05,
      "step": 26323,
      "training_step_time": 8.50972580909729
    },
    {
      "epoch": 2.5104522705078127e-05,
      "model_forward_time": 1.7202413082122803,
      "step": 26324
    },
    {
      "epoch": 2.5104522705078127e-05,
      "step": 26324,
      "training_step_time": 7.687566041946411
    },
    {
      "epoch": 2.510547637939453e-05,
      "model_forward_time": 1.7336006164550781,
      "step": 26325
    },
    {
      "epoch": 2.510547637939453e-05,
      "step": 26325,
      "training_step_time": 8.600596189498901
    },
    {
      "epoch": 2.5106430053710938e-05,
      "model_forward_time": 2.8396472930908203,
      "step": 26326
    },
    {
      "epoch": 2.5106430053710938e-05,
      "step": 26326,
      "training_step_time": 8.254548788070679
    },
    {
      "epoch": 2.5107383728027345e-05,
      "model_forward_time": 1.733250617980957,
      "step": 26327
    },
    {
      "epoch": 2.5107383728027345e-05,
      "step": 26327,
      "training_step_time": 8.799221992492676
    },
    {
      "epoch": 2.510833740234375e-05,
      "model_forward_time": 1.7318098545074463,
      "step": 26328
    },
    {
      "epoch": 2.510833740234375e-05,
      "step": 26328,
      "training_step_time": 7.624755859375
    },
    {
      "epoch": 2.5109291076660157e-05,
      "model_forward_time": 1.7394905090332031,
      "step": 26329
    },
    {
      "epoch": 2.5109291076660157e-05,
      "step": 26329,
      "training_step_time": 9.002099752426147
    },
    {
      "epoch": 2.511024475097656e-05,
      "grad_norm": 0.18640419840812683,
      "learning_rate": 8.806965103730238e-06,
      "loss": 0.0573,
      "step": 26330
    },
    {
      "epoch": 2.511024475097656e-05,
      "model_forward_time": 1.7333481311798096,
      "step": 26330
    },
    {
      "dynamics_loss_avg": 0.044863061234354976,
      "epoch": 2.511024475097656e-05,
      "step": 26330
    },
    {
      "action_loss_avg": 0.0018016777699813246,
      "epoch": 2.511024475097656e-05,
      "step": 26330
    },
    {
      "epoch": 2.511024475097656e-05,
      "step": 26330,
      "training_step_time": 7.596823215484619
    },
    {
      "epoch": 2.5111198425292968e-05,
      "model_forward_time": 1.7307968139648438,
      "step": 26331
    },
    {
      "epoch": 2.5111198425292968e-05,
      "step": 26331,
      "training_step_time": 8.840128421783447
    },
    {
      "epoch": 2.5112152099609376e-05,
      "model_forward_time": 1.7316498756408691,
      "step": 26332
    },
    {
      "epoch": 2.5112152099609376e-05,
      "step": 26332,
      "training_step_time": 7.55015230178833
    },
    {
      "epoch": 2.511310577392578e-05,
      "model_forward_time": 1.7349190711975098,
      "step": 26333
    },
    {
      "epoch": 2.511310577392578e-05,
      "step": 26333,
      "training_step_time": 7.637593746185303
    },
    {
      "epoch": 2.5114059448242187e-05,
      "model_forward_time": 1.736236333847046,
      "step": 26334
    },
    {
      "epoch": 2.5114059448242187e-05,
      "step": 26334,
      "training_step_time": 8.159057855606079
    },
    {
      "epoch": 2.5115013122558595e-05,
      "model_forward_time": 2.917036294937134,
      "step": 26335
    },
    {
      "epoch": 2.5115013122558595e-05,
      "step": 26335,
      "training_step_time": 7.752483367919922
    },
    {
      "epoch": 2.5115966796875e-05,
      "model_forward_time": 1.7317514419555664,
      "step": 26336
    },
    {
      "epoch": 2.5115966796875e-05,
      "step": 26336,
      "training_step_time": 9.827798843383789
    },
    {
      "epoch": 2.5116920471191406e-05,
      "model_forward_time": 1.7306406497955322,
      "step": 26337
    },
    {
      "epoch": 2.5116920471191406e-05,
      "step": 26337,
      "training_step_time": 7.7683165073394775
    },
    {
      "epoch": 2.5117874145507813e-05,
      "model_forward_time": 1.7366135120391846,
      "step": 26338
    },
    {
      "epoch": 2.5117874145507813e-05,
      "step": 26338,
      "training_step_time": 7.714136123657227
    },
    {
      "epoch": 2.5118827819824217e-05,
      "model_forward_time": 1.73795485496521,
      "step": 26339
    },
    {
      "epoch": 2.5118827819824217e-05,
      "step": 26339,
      "training_step_time": 7.696134090423584
    },
    {
      "epoch": 2.5119781494140625e-05,
      "grad_norm": 0.14629127085208893,
      "learning_rate": 8.805892966691074e-06,
      "loss": 0.0569,
      "step": 26340
    },
    {
      "epoch": 2.5119781494140625e-05,
      "model_forward_time": 1.7396843433380127,
      "step": 26340
    },
    {
      "dynamics_loss_avg": 0.05981626082211733,
      "epoch": 2.5119781494140625e-05,
      "step": 26340
    },
    {
      "action_loss_avg": 0.003358887601643801,
      "epoch": 2.5119781494140625e-05,
      "step": 26340
    },
    {
      "epoch": 2.5119781494140625e-05,
      "step": 26340,
      "training_step_time": 7.983242034912109
    },
    {
      "epoch": 2.5120735168457032e-05,
      "model_forward_time": 1.7344179153442383,
      "step": 26341
    },
    {
      "epoch": 2.5120735168457032e-05,
      "step": 26341,
      "training_step_time": 7.614598274230957
    },
    {
      "epoch": 2.5121688842773436e-05,
      "model_forward_time": 1.7506701946258545,
      "step": 26342
    },
    {
      "epoch": 2.5121688842773436e-05,
      "step": 26342,
      "training_step_time": 8.44876766204834
    },
    {
      "epoch": 2.5122642517089844e-05,
      "model_forward_time": 1.733088493347168,
      "step": 26343
    },
    {
      "epoch": 2.5122642517089844e-05,
      "step": 26343,
      "training_step_time": 7.664754867553711
    },
    {
      "epoch": 2.512359619140625e-05,
      "model_forward_time": 2.834913492202759,
      "step": 26344
    },
    {
      "epoch": 2.512359619140625e-05,
      "step": 26344,
      "training_step_time": 7.690704584121704
    },
    {
      "epoch": 2.5124549865722655e-05,
      "model_forward_time": 1.7302603721618652,
      "step": 26345
    },
    {
      "epoch": 2.5124549865722655e-05,
      "step": 26345,
      "training_step_time": 7.597821474075317
    },
    {
      "epoch": 2.5125503540039063e-05,
      "model_forward_time": 1.7316069602966309,
      "step": 26346
    },
    {
      "epoch": 2.5125503540039063e-05,
      "step": 26346,
      "training_step_time": 7.838991641998291
    },
    {
      "epoch": 2.512645721435547e-05,
      "model_forward_time": 1.7333531379699707,
      "step": 26347
    },
    {
      "epoch": 2.512645721435547e-05,
      "step": 26347,
      "training_step_time": 8.365755081176758
    },
    {
      "epoch": 2.5127410888671874e-05,
      "model_forward_time": 1.3522343635559082,
      "step": 26348
    },
    {
      "epoch": 2.5127410888671874e-05,
      "step": 26348,
      "training_step_time": 7.709803581237793
    },
    {
      "epoch": 2.512836456298828e-05,
      "model_forward_time": 1.732987403869629,
      "step": 26349
    },
    {
      "epoch": 2.512836456298828e-05,
      "step": 26349,
      "training_step_time": 7.908533334732056
    },
    {
      "epoch": 2.512931823730469e-05,
      "grad_norm": 0.21456439793109894,
      "learning_rate": 8.804820413445176e-06,
      "loss": 0.0626,
      "step": 26350
    },
    {
      "epoch": 2.512931823730469e-05,
      "model_forward_time": 1.7279980182647705,
      "step": 26350
    },
    {
      "dynamics_loss_avg": 0.059540740214288235,
      "epoch": 2.512931823730469e-05,
      "step": 26350
    },
    {
      "action_loss_avg": 0.003705073962919414,
      "epoch": 2.512931823730469e-05,
      "step": 26350
    },
    {
      "epoch": 2.512931823730469e-05,
      "step": 26350,
      "training_step_time": 7.5826849937438965
    },
    {
      "epoch": 2.5130271911621093e-05,
      "model_forward_time": 1.7535254955291748,
      "step": 26351
    },
    {
      "epoch": 2.5130271911621093e-05,
      "step": 26351,
      "training_step_time": 8.762603521347046
    },
    {
      "epoch": 2.51312255859375e-05,
      "model_forward_time": 1.7284061908721924,
      "step": 26352
    },
    {
      "epoch": 2.51312255859375e-05,
      "step": 26352,
      "training_step_time": 9.181617975234985
    },
    {
      "epoch": 2.5132179260253908e-05,
      "model_forward_time": 1.726818323135376,
      "step": 26353
    },
    {
      "epoch": 2.5132179260253908e-05,
      "step": 26353,
      "training_step_time": 6.8078837394714355
    },
    {
      "epoch": 2.5133132934570312e-05,
      "model_forward_time": 1.7350149154663086,
      "step": 26354
    },
    {
      "epoch": 2.5133132934570312e-05,
      "step": 26354,
      "training_step_time": 6.783249855041504
    },
    {
      "epoch": 2.513408660888672e-05,
      "model_forward_time": 1.3607094287872314,
      "step": 26355
    },
    {
      "epoch": 2.513408660888672e-05,
      "step": 26355,
      "training_step_time": 6.741567373275757
    },
    {
      "epoch": 2.5135040283203127e-05,
      "model_forward_time": 1.7288999557495117,
      "step": 26356
    },
    {
      "epoch": 2.5135040283203127e-05,
      "step": 26356,
      "training_step_time": 6.788630485534668
    },
    {
      "epoch": 2.513599395751953e-05,
      "model_forward_time": 1.730846881866455,
      "step": 26357
    },
    {
      "epoch": 2.513599395751953e-05,
      "step": 26357,
      "training_step_time": 6.813085556030273
    },
    {
      "epoch": 2.5136947631835938e-05,
      "model_forward_time": 1.7302651405334473,
      "step": 26358
    },
    {
      "epoch": 2.5136947631835938e-05,
      "step": 26358,
      "training_step_time": 6.773225545883179
    },
    {
      "epoch": 2.5137901306152345e-05,
      "model_forward_time": 1.7330079078674316,
      "step": 26359
    },
    {
      "epoch": 2.5137901306152345e-05,
      "step": 26359,
      "training_step_time": 6.72979211807251
    },
    {
      "epoch": 2.513885498046875e-05,
      "grad_norm": 0.14902006089687347,
      "learning_rate": 8.803747444109839e-06,
      "loss": 0.0613,
      "step": 26360
    },
    {
      "epoch": 2.513885498046875e-05,
      "model_forward_time": 1.735278844833374,
      "step": 26360
    },
    {
      "dynamics_loss_avg": 0.06881909370422364,
      "epoch": 2.513885498046875e-05,
      "step": 26360
    },
    {
      "action_loss_avg": 0.0022085049364250152,
      "epoch": 2.513885498046875e-05,
      "step": 26360
    },
    {
      "epoch": 2.513885498046875e-05,
      "step": 26360,
      "training_step_time": 6.9539430141448975
    },
    {
      "epoch": 2.5139808654785157e-05,
      "model_forward_time": 2.8633999824523926,
      "step": 26361
    },
    {
      "epoch": 2.5139808654785157e-05,
      "step": 26361,
      "training_step_time": 7.77021050453186
    },
    {
      "epoch": 2.514076232910156e-05,
      "model_forward_time": 1.7345890998840332,
      "step": 26362
    },
    {
      "epoch": 2.514076232910156e-05,
      "step": 26362,
      "training_step_time": 7.861090660095215
    },
    {
      "epoch": 2.514171600341797e-05,
      "model_forward_time": 1.727998971939087,
      "step": 26363
    },
    {
      "epoch": 2.514171600341797e-05,
      "step": 26363,
      "training_step_time": 6.810610294342041
    },
    {
      "epoch": 2.5142669677734376e-05,
      "model_forward_time": 1.7285747528076172,
      "step": 26364
    },
    {
      "epoch": 2.5142669677734376e-05,
      "step": 26364,
      "training_step_time": 6.796751022338867
    },
    {
      "epoch": 2.514362335205078e-05,
      "model_forward_time": 1.7292380332946777,
      "step": 26365
    },
    {
      "epoch": 2.514362335205078e-05,
      "step": 26365,
      "training_step_time": 6.762730360031128
    },
    {
      "epoch": 2.5144577026367187e-05,
      "model_forward_time": 1.696275234222412,
      "step": 26366
    },
    {
      "epoch": 2.5144577026367187e-05,
      "step": 26366,
      "training_step_time": 6.67944073677063
    },
    {
      "epoch": 2.5145530700683595e-05,
      "model_forward_time": 0.9970667362213135,
      "step": 26367
    },
    {
      "epoch": 2.5145530700683595e-05,
      "step": 26367,
      "training_step_time": 6.804757118225098
    },
    {
      "epoch": 2.5146484375e-05,
      "model_forward_time": 1.7149708271026611,
      "step": 26368
    },
    {
      "epoch": 2.5146484375e-05,
      "step": 26368,
      "training_step_time": 6.829800605773926
    },
    {
      "epoch": 2.5147438049316406e-05,
      "model_forward_time": 1.7116661071777344,
      "step": 26369
    },
    {
      "epoch": 2.5147438049316406e-05,
      "step": 26369,
      "training_step_time": 6.817921876907349
    },
    {
      "epoch": 2.5148391723632813e-05,
      "grad_norm": 0.15645530819892883,
      "learning_rate": 8.802674058802398e-06,
      "loss": 0.0645,
      "step": 26370
    },
    {
      "epoch": 2.5148391723632813e-05,
      "model_forward_time": 2.821772575378418,
      "step": 26370
    },
    {
      "dynamics_loss_avg": 0.05482571441680193,
      "epoch": 2.5148391723632813e-05,
      "step": 26370
    },
    {
      "action_loss_avg": 0.003183246951084584,
      "epoch": 2.5148391723632813e-05,
      "step": 26370
    },
    {
      "epoch": 2.5148391723632813e-05,
      "step": 26370,
      "training_step_time": 7.635315895080566
    },
    {
      "epoch": 2.5149345397949217e-05,
      "model_forward_time": 1.7163548469543457,
      "step": 26371
    },
    {
      "epoch": 2.5149345397949217e-05,
      "step": 26371,
      "training_step_time": 7.7934370040893555
    },
    {
      "epoch": 2.5150299072265625e-05,
      "model_forward_time": 1.7167418003082275,
      "step": 26372
    },
    {
      "epoch": 2.5150299072265625e-05,
      "step": 26372,
      "training_step_time": 7.954192638397217
    },
    {
      "epoch": 2.5151252746582032e-05,
      "model_forward_time": 1.7066786289215088,
      "step": 26373
    },
    {
      "epoch": 2.5151252746582032e-05,
      "step": 26373,
      "training_step_time": 6.851196527481079
    },
    {
      "epoch": 2.5152206420898436e-05,
      "model_forward_time": 1.7125012874603271,
      "step": 26374
    },
    {
      "epoch": 2.5152206420898436e-05,
      "step": 26374,
      "training_step_time": 6.867193222045898
    },
    {
      "epoch": 2.5153160095214844e-05,
      "model_forward_time": 1.7209789752960205,
      "step": 26375
    },
    {
      "epoch": 2.5153160095214844e-05,
      "step": 26375,
      "training_step_time": 6.780080318450928
    },
    {
      "epoch": 2.515411376953125e-05,
      "model_forward_time": 1.7102258205413818,
      "step": 26376
    },
    {
      "epoch": 2.515411376953125e-05,
      "step": 26376,
      "training_step_time": 6.793583393096924
    },
    {
      "epoch": 2.5155067443847655e-05,
      "model_forward_time": 1.7090644836425781,
      "step": 26377
    },
    {
      "epoch": 2.5155067443847655e-05,
      "step": 26377,
      "training_step_time": 6.773067235946655
    },
    {
      "epoch": 2.5156021118164063e-05,
      "model_forward_time": 1.7367503643035889,
      "step": 26378
    },
    {
      "epoch": 2.5156021118164063e-05,
      "step": 26378,
      "training_step_time": 6.777586936950684
    },
    {
      "epoch": 2.515697479248047e-05,
      "model_forward_time": 2.8517889976501465,
      "step": 26379
    },
    {
      "epoch": 2.515697479248047e-05,
      "step": 26379,
      "training_step_time": 7.681809186935425
    },
    {
      "epoch": 2.5157928466796874e-05,
      "grad_norm": 0.18573300540447235,
      "learning_rate": 8.801600257640242e-06,
      "loss": 0.0632,
      "step": 26380
    },
    {
      "epoch": 2.5157928466796874e-05,
      "model_forward_time": 1.7352774143218994,
      "step": 26380
    },
    {
      "dynamics_loss_avg": 0.046154855750501156,
      "epoch": 2.5157928466796874e-05,
      "step": 26380
    },
    {
      "action_loss_avg": 0.001931105088442564,
      "epoch": 2.5157928466796874e-05,
      "step": 26380
    },
    {
      "epoch": 2.5157928466796874e-05,
      "step": 26380,
      "training_step_time": 7.7893712520599365
    },
    {
      "epoch": 2.515888214111328e-05,
      "model_forward_time": 1.7331652641296387,
      "step": 26381
    },
    {
      "epoch": 2.515888214111328e-05,
      "step": 26381,
      "training_step_time": 7.898364067077637
    },
    {
      "epoch": 2.515983581542969e-05,
      "model_forward_time": 1.7358827590942383,
      "step": 26382
    },
    {
      "epoch": 2.515983581542969e-05,
      "step": 26382,
      "training_step_time": 6.796891212463379
    },
    {
      "epoch": 2.5160789489746093e-05,
      "model_forward_time": 1.6970434188842773,
      "step": 26383
    },
    {
      "epoch": 2.5160789489746093e-05,
      "step": 26383,
      "training_step_time": 6.8203911781311035
    },
    {
      "epoch": 2.51617431640625e-05,
      "model_forward_time": 1.7284023761749268,
      "step": 26384
    },
    {
      "epoch": 2.51617431640625e-05,
      "step": 26384,
      "training_step_time": 6.806930780410767
    },
    {
      "epoch": 2.5162696838378908e-05,
      "model_forward_time": 1.7306640148162842,
      "step": 26385
    },
    {
      "epoch": 2.5162696838378908e-05,
      "step": 26385,
      "training_step_time": 6.7530200481414795
    },
    {
      "epoch": 2.5163650512695312e-05,
      "model_forward_time": 1.733928918838501,
      "step": 26386
    },
    {
      "epoch": 2.5163650512695312e-05,
      "step": 26386,
      "training_step_time": 6.799181699752808
    },
    {
      "epoch": 2.516460418701172e-05,
      "model_forward_time": 1.7361059188842773,
      "step": 26387
    },
    {
      "epoch": 2.516460418701172e-05,
      "step": 26387,
      "training_step_time": 6.833292007446289
    },
    {
      "epoch": 2.5165557861328127e-05,
      "model_forward_time": 1.7621865272521973,
      "step": 26388
    },
    {
      "epoch": 2.5165557861328127e-05,
      "step": 26388,
      "training_step_time": 7.548280715942383
    },
    {
      "epoch": 2.516651153564453e-05,
      "model_forward_time": 2.9336328506469727,
      "step": 26389
    },
    {
      "epoch": 2.516651153564453e-05,
      "step": 26389,
      "training_step_time": 9.682328939437866
    },
    {
      "epoch": 2.5167465209960938e-05,
      "grad_norm": 0.18048247694969177,
      "learning_rate": 8.800526040740795e-06,
      "loss": 0.0654,
      "step": 26390
    },
    {
      "epoch": 2.5167465209960938e-05,
      "model_forward_time": 1.733144760131836,
      "step": 26390
    },
    {
      "dynamics_loss_avg": 0.05242996923625469,
      "epoch": 2.5167465209960938e-05,
      "step": 26390
    },
    {
      "action_loss_avg": 0.0024088857986498623,
      "epoch": 2.5167465209960938e-05,
      "step": 26390
    },
    {
      "epoch": 2.5167465209960938e-05,
      "step": 26390,
      "training_step_time": 7.704897165298462
    },
    {
      "epoch": 2.5168418884277342e-05,
      "model_forward_time": 1.7356936931610107,
      "step": 26391
    },
    {
      "epoch": 2.5168418884277342e-05,
      "step": 26391,
      "training_step_time": 7.90028977394104
    },
    {
      "epoch": 2.516937255859375e-05,
      "model_forward_time": 1.727888584136963,
      "step": 26392
    },
    {
      "epoch": 2.516937255859375e-05,
      "step": 26392,
      "training_step_time": 6.831422328948975
    },
    {
      "epoch": 2.5170326232910157e-05,
      "model_forward_time": 1.7398505210876465,
      "step": 26393
    },
    {
      "epoch": 2.5170326232910157e-05,
      "step": 26393,
      "training_step_time": 6.794207811355591
    },
    {
      "epoch": 2.517127990722656e-05,
      "model_forward_time": 1.7362751960754395,
      "step": 26394
    },
    {
      "epoch": 2.517127990722656e-05,
      "step": 26394,
      "training_step_time": 6.78946852684021
    },
    {
      "epoch": 2.517223358154297e-05,
      "model_forward_time": 1.7374229431152344,
      "step": 26395
    },
    {
      "epoch": 2.517223358154297e-05,
      "step": 26395,
      "training_step_time": 6.785413026809692
    },
    {
      "epoch": 2.5173187255859376e-05,
      "model_forward_time": 1.6998882293701172,
      "step": 26396
    },
    {
      "epoch": 2.5173187255859376e-05,
      "step": 26396,
      "training_step_time": 6.824932813644409
    },
    {
      "epoch": 2.517414093017578e-05,
      "model_forward_time": 1.7699427604675293,
      "step": 26397
    },
    {
      "epoch": 2.517414093017578e-05,
      "step": 26397,
      "training_step_time": 7.5244903564453125
    },
    {
      "epoch": 2.5175094604492187e-05,
      "model_forward_time": 2.916275978088379,
      "step": 26398
    },
    {
      "epoch": 2.5175094604492187e-05,
      "step": 26398,
      "training_step_time": 8.756980895996094
    },
    {
      "epoch": 2.5176048278808595e-05,
      "model_forward_time": 1.733567237854004,
      "step": 26399
    },
    {
      "epoch": 2.5176048278808595e-05,
      "step": 26399,
      "training_step_time": 7.700205564498901
    },
    {
      "epoch": 2.5177001953125e-05,
      "grad_norm": 0.1303386241197586,
      "learning_rate": 8.799451408221536e-06,
      "loss": 0.0625,
      "step": 26400
    },
    {
      "epoch": 2.5177001953125e-05,
      "model_forward_time": 1.7337050437927246,
      "step": 26400
    },
    {
      "dynamics_loss_avg": 0.05576127003878355,
      "epoch": 2.5177001953125e-05,
      "step": 26400
    },
    {
      "action_loss_avg": 0.0022968345205299557,
      "epoch": 2.5177001953125e-05,
      "step": 26400
    },
    {
      "epoch": 2.5177001953125e-05,
      "step": 26400,
      "training_step_time": 7.822350263595581
    },
    {
      "epoch": 2.5177955627441406e-05,
      "model_forward_time": 1.7416884899139404,
      "step": 26401
    },
    {
      "epoch": 2.5177955627441406e-05,
      "step": 26401,
      "training_step_time": 6.785877704620361
    },
    {
      "epoch": 2.5178909301757813e-05,
      "model_forward_time": 1.7347331047058105,
      "step": 26402
    },
    {
      "epoch": 2.5178909301757813e-05,
      "step": 26402,
      "training_step_time": 6.762516975402832
    },
    {
      "epoch": 2.5179862976074218e-05,
      "model_forward_time": 1.7292346954345703,
      "step": 26403
    },
    {
      "epoch": 2.5179862976074218e-05,
      "step": 26403,
      "training_step_time": 6.635260820388794
    },
    {
      "epoch": 2.5180816650390625e-05,
      "model_forward_time": 1.7351016998291016,
      "step": 26404
    },
    {
      "epoch": 2.5180816650390625e-05,
      "step": 26404,
      "training_step_time": 6.753871202468872
    },
    {
      "epoch": 2.5181770324707032e-05,
      "model_forward_time": 1.7408170700073242,
      "step": 26405
    },
    {
      "epoch": 2.5181770324707032e-05,
      "step": 26405,
      "training_step_time": 6.76779580116272
    },
    {
      "epoch": 2.5182723999023436e-05,
      "model_forward_time": 1.7380058765411377,
      "step": 26406
    },
    {
      "epoch": 2.5182723999023436e-05,
      "step": 26406,
      "training_step_time": 7.377254247665405
    },
    {
      "epoch": 2.5183677673339844e-05,
      "model_forward_time": 2.855701208114624,
      "step": 26407
    },
    {
      "epoch": 2.5183677673339844e-05,
      "step": 26407,
      "training_step_time": 8.694469213485718
    },
    {
      "epoch": 2.518463134765625e-05,
      "model_forward_time": 1.7373387813568115,
      "step": 26408
    },
    {
      "epoch": 2.518463134765625e-05,
      "step": 26408,
      "training_step_time": 7.720556974411011
    },
    {
      "epoch": 2.5185585021972655e-05,
      "model_forward_time": 1.743814468383789,
      "step": 26409
    },
    {
      "epoch": 2.5185585021972655e-05,
      "step": 26409,
      "training_step_time": 7.860840320587158
    },
    {
      "epoch": 2.5186538696289063e-05,
      "grad_norm": 0.1783503293991089,
      "learning_rate": 8.798376360199981e-06,
      "loss": 0.0607,
      "step": 26410
    },
    {
      "epoch": 2.5186538696289063e-05,
      "model_forward_time": 1.733335018157959,
      "step": 26410
    },
    {
      "dynamics_loss_avg": 0.05772977452725172,
      "epoch": 2.5186538696289063e-05,
      "step": 26410
    },
    {
      "action_loss_avg": 0.002582191053079441,
      "epoch": 2.5186538696289063e-05,
      "step": 26410
    },
    {
      "epoch": 2.5186538696289063e-05,
      "step": 26410,
      "training_step_time": 6.778513431549072
    },
    {
      "epoch": 2.518749237060547e-05,
      "model_forward_time": 1.744471788406372,
      "step": 26411
    },
    {
      "epoch": 2.518749237060547e-05,
      "step": 26411,
      "training_step_time": 6.757516860961914
    },
    {
      "epoch": 2.5188446044921874e-05,
      "model_forward_time": 1.7424027919769287,
      "step": 26412
    },
    {
      "epoch": 2.5188446044921874e-05,
      "step": 26412,
      "training_step_time": 6.758919954299927
    },
    {
      "epoch": 2.518939971923828e-05,
      "model_forward_time": 1.7359964847564697,
      "step": 26413
    },
    {
      "epoch": 2.518939971923828e-05,
      "step": 26413,
      "training_step_time": 6.728935480117798
    },
    {
      "epoch": 2.519035339355469e-05,
      "model_forward_time": 1.7031440734863281,
      "step": 26414
    },
    {
      "epoch": 2.519035339355469e-05,
      "step": 26414,
      "training_step_time": 6.739886522293091
    },
    {
      "epoch": 2.5191307067871093e-05,
      "model_forward_time": 1.7169699668884277,
      "step": 26415
    },
    {
      "epoch": 2.5191307067871093e-05,
      "step": 26415,
      "training_step_time": 6.793606758117676
    },
    {
      "epoch": 2.51922607421875e-05,
      "model_forward_time": 2.90901255607605,
      "step": 26416
    },
    {
      "epoch": 2.51922607421875e-05,
      "step": 26416,
      "training_step_time": 8.50598406791687
    },
    {
      "epoch": 2.5193214416503908e-05,
      "model_forward_time": 1.7148246765136719,
      "step": 26417
    },
    {
      "epoch": 2.5193214416503908e-05,
      "step": 26417,
      "training_step_time": 10.270953893661499
    },
    {
      "epoch": 2.5194168090820312e-05,
      "model_forward_time": 1.7136781215667725,
      "step": 26418
    },
    {
      "epoch": 2.5194168090820312e-05,
      "step": 26418,
      "training_step_time": 7.943588018417358
    },
    {
      "epoch": 2.519512176513672e-05,
      "model_forward_time": 1.714918613433838,
      "step": 26419
    },
    {
      "epoch": 2.519512176513672e-05,
      "step": 26419,
      "training_step_time": 6.7431910037994385
    },
    {
      "epoch": 2.5196075439453127e-05,
      "grad_norm": 0.23315300047397614,
      "learning_rate": 8.797300896793702e-06,
      "loss": 0.062,
      "step": 26420
    },
    {
      "epoch": 2.5196075439453127e-05,
      "model_forward_time": 1.7143421173095703,
      "step": 26420
    },
    {
      "dynamics_loss_avg": 0.055139639787375926,
      "epoch": 2.5196075439453127e-05,
      "step": 26420
    },
    {
      "action_loss_avg": 0.002488813517265953,
      "epoch": 2.5196075439453127e-05,
      "step": 26420
    },
    {
      "epoch": 2.5196075439453127e-05,
      "step": 26420,
      "training_step_time": 6.788507461547852
    },
    {
      "epoch": 2.519702911376953e-05,
      "model_forward_time": 1.7158503532409668,
      "step": 26421
    },
    {
      "epoch": 2.519702911376953e-05,
      "step": 26421,
      "training_step_time": 6.789414405822754
    },
    {
      "epoch": 2.5197982788085938e-05,
      "model_forward_time": 1.7157001495361328,
      "step": 26422
    },
    {
      "epoch": 2.5197982788085938e-05,
      "step": 26422,
      "training_step_time": 6.75747275352478
    },
    {
      "epoch": 2.5198936462402342e-05,
      "model_forward_time": 1.7164435386657715,
      "step": 26423
    },
    {
      "epoch": 2.5198936462402342e-05,
      "step": 26423,
      "training_step_time": 6.756420850753784
    },
    {
      "epoch": 2.519989013671875e-05,
      "model_forward_time": 1.7180366516113281,
      "step": 26424
    },
    {
      "epoch": 2.519989013671875e-05,
      "step": 26424,
      "training_step_time": 6.798238754272461
    },
    {
      "epoch": 2.5200843811035157e-05,
      "model_forward_time": 2.8871498107910156,
      "step": 26425
    },
    {
      "epoch": 2.5200843811035157e-05,
      "step": 26425,
      "training_step_time": 7.702221393585205
    },
    {
      "epoch": 2.520179748535156e-05,
      "model_forward_time": 1.71305513381958,
      "step": 26426
    },
    {
      "epoch": 2.520179748535156e-05,
      "step": 26426,
      "training_step_time": 8.532804727554321
    },
    {
      "epoch": 2.520275115966797e-05,
      "model_forward_time": 1.7149841785430908,
      "step": 26427
    },
    {
      "epoch": 2.520275115966797e-05,
      "step": 26427,
      "training_step_time": 8.233951568603516
    },
    {
      "epoch": 2.5203704833984376e-05,
      "model_forward_time": 1.7240431308746338,
      "step": 26428
    },
    {
      "epoch": 2.5203704833984376e-05,
      "step": 26428,
      "training_step_time": 6.751981258392334
    },
    {
      "epoch": 2.520465850830078e-05,
      "model_forward_time": 1.705108404159546,
      "step": 26429
    },
    {
      "epoch": 2.520465850830078e-05,
      "step": 26429,
      "training_step_time": 6.7819740772247314
    },
    {
      "epoch": 2.5205612182617187e-05,
      "grad_norm": 0.34088537096977234,
      "learning_rate": 8.796225018120302e-06,
      "loss": 0.0594,
      "step": 26430
    },
    {
      "epoch": 2.5205612182617187e-05,
      "model_forward_time": 1.7104296684265137,
      "step": 26430
    },
    {
      "dynamics_loss_avg": 0.07544884216040373,
      "epoch": 2.5205612182617187e-05,
      "step": 26430
    },
    {
      "action_loss_avg": 0.0030331047251820563,
      "epoch": 2.5205612182617187e-05,
      "step": 26430
    },
    {
      "epoch": 2.5205612182617187e-05,
      "step": 26430,
      "training_step_time": 6.7827308177948
    },
    {
      "epoch": 2.5206565856933595e-05,
      "model_forward_time": 1.726196050643921,
      "step": 26431
    },
    {
      "epoch": 2.5206565856933595e-05,
      "step": 26431,
      "training_step_time": 6.796297073364258
    },
    {
      "epoch": 2.520751953125e-05,
      "model_forward_time": 1.7148921489715576,
      "step": 26432
    },
    {
      "epoch": 2.520751953125e-05,
      "step": 26432,
      "training_step_time": 6.784631013870239
    },
    {
      "epoch": 2.5208473205566406e-05,
      "model_forward_time": 1.721123218536377,
      "step": 26433
    },
    {
      "epoch": 2.5208473205566406e-05,
      "step": 26433,
      "training_step_time": 6.779833078384399
    },
    {
      "epoch": 2.5209426879882814e-05,
      "model_forward_time": 1.8086771965026855,
      "step": 26434
    },
    {
      "epoch": 2.5209426879882814e-05,
      "step": 26434,
      "training_step_time": 7.52019476890564
    },
    {
      "epoch": 2.5210380554199218e-05,
      "model_forward_time": 2.941000461578369,
      "step": 26435
    },
    {
      "epoch": 2.5210380554199218e-05,
      "step": 26435,
      "training_step_time": 10.27976393699646
    },
    {
      "epoch": 2.5211334228515625e-05,
      "model_forward_time": 1.717428207397461,
      "step": 26436
    },
    {
      "epoch": 2.5211334228515625e-05,
      "step": 26436,
      "training_step_time": 7.866238355636597
    },
    {
      "epoch": 2.5212287902832032e-05,
      "model_forward_time": 1.7084455490112305,
      "step": 26437
    },
    {
      "epoch": 2.5212287902832032e-05,
      "step": 26437,
      "training_step_time": 7.998005628585815
    },
    {
      "epoch": 2.5213241577148436e-05,
      "model_forward_time": 1.7144019603729248,
      "step": 26438
    },
    {
      "epoch": 2.5213241577148436e-05,
      "step": 26438,
      "training_step_time": 6.802097797393799
    },
    {
      "epoch": 2.5214195251464844e-05,
      "model_forward_time": 1.711679458618164,
      "step": 26439
    },
    {
      "epoch": 2.5214195251464844e-05,
      "step": 26439,
      "training_step_time": 6.8121209144592285
    },
    {
      "epoch": 2.521514892578125e-05,
      "grad_norm": 0.15025220811367035,
      "learning_rate": 8.795148724297444e-06,
      "loss": 0.0628,
      "step": 26440
    },
    {
      "epoch": 2.521514892578125e-05,
      "model_forward_time": 1.7126314640045166,
      "step": 26440
    },
    {
      "dynamics_loss_avg": 0.0432385042309761,
      "epoch": 2.521514892578125e-05,
      "step": 26440
    },
    {
      "action_loss_avg": 0.002712826116476208,
      "epoch": 2.521514892578125e-05,
      "step": 26440
    },
    {
      "epoch": 2.521514892578125e-05,
      "step": 26440,
      "training_step_time": 6.765411138534546
    },
    {
      "epoch": 2.5216102600097655e-05,
      "model_forward_time": 1.71360182762146,
      "step": 26441
    },
    {
      "epoch": 2.5216102600097655e-05,
      "step": 26441,
      "training_step_time": 6.7844603061676025
    },
    {
      "epoch": 2.5217056274414063e-05,
      "model_forward_time": 1.7182493209838867,
      "step": 26442
    },
    {
      "epoch": 2.5217056274414063e-05,
      "step": 26442,
      "training_step_time": 6.8095009326934814
    },
    {
      "epoch": 2.521800994873047e-05,
      "model_forward_time": 1.7552595138549805,
      "step": 26443
    },
    {
      "epoch": 2.521800994873047e-05,
      "step": 26443,
      "training_step_time": 7.564892768859863
    },
    {
      "epoch": 2.5218963623046874e-05,
      "model_forward_time": 2.9167635440826416,
      "step": 26444
    },
    {
      "epoch": 2.5218963623046874e-05,
      "step": 26444,
      "training_step_time": 9.383000135421753
    },
    {
      "epoch": 2.521991729736328e-05,
      "model_forward_time": 1.7366890907287598,
      "step": 26445
    },
    {
      "epoch": 2.521991729736328e-05,
      "step": 26445,
      "training_step_time": 8.367857217788696
    },
    {
      "epoch": 2.522087097167969e-05,
      "model_forward_time": 1.7292721271514893,
      "step": 26446
    },
    {
      "epoch": 2.522087097167969e-05,
      "step": 26446,
      "training_step_time": 7.843574285507202
    },
    {
      "epoch": 2.5221824645996093e-05,
      "model_forward_time": 1.73445725440979,
      "step": 26447
    },
    {
      "epoch": 2.5221824645996093e-05,
      "step": 26447,
      "training_step_time": 6.799602746963501
    },
    {
      "epoch": 2.52227783203125e-05,
      "model_forward_time": 1.7389838695526123,
      "step": 26448
    },
    {
      "epoch": 2.52227783203125e-05,
      "step": 26448,
      "training_step_time": 6.821293354034424
    },
    {
      "epoch": 2.5223731994628908e-05,
      "model_forward_time": 1.6997177600860596,
      "step": 26449
    },
    {
      "epoch": 2.5223731994628908e-05,
      "step": 26449,
      "training_step_time": 6.833899974822998
    },
    {
      "epoch": 2.5224685668945312e-05,
      "grad_norm": 0.15369193255901337,
      "learning_rate": 8.794072015442826e-06,
      "loss": 0.0613,
      "step": 26450
    },
    {
      "epoch": 2.5224685668945312e-05,
      "model_forward_time": 1.7125887870788574,
      "step": 26450
    },
    {
      "dynamics_loss_avg": 0.08526793085038661,
      "epoch": 2.5224685668945312e-05,
      "step": 26450
    },
    {
      "action_loss_avg": 0.0039225775515660645,
      "epoch": 2.5224685668945312e-05,
      "step": 26450
    },
    {
      "epoch": 2.5224685668945312e-05,
      "step": 26450,
      "training_step_time": 6.804640054702759
    },
    {
      "epoch": 2.522563934326172e-05,
      "model_forward_time": 1.7186360359191895,
      "step": 26451
    },
    {
      "epoch": 2.522563934326172e-05,
      "step": 26451,
      "training_step_time": 6.758533239364624
    },
    {
      "epoch": 2.5226593017578127e-05,
      "model_forward_time": 1.7374653816223145,
      "step": 26452
    },
    {
      "epoch": 2.5226593017578127e-05,
      "step": 26452,
      "training_step_time": 7.528774738311768
    },
    {
      "epoch": 2.522754669189453e-05,
      "model_forward_time": 2.885732650756836,
      "step": 26453
    },
    {
      "epoch": 2.522754669189453e-05,
      "step": 26453,
      "training_step_time": 8.455338478088379
    },
    {
      "epoch": 2.5228500366210938e-05,
      "model_forward_time": 1.7188527584075928,
      "step": 26454
    },
    {
      "epoch": 2.5228500366210938e-05,
      "step": 26454,
      "training_step_time": 9.442269325256348
    },
    {
      "epoch": 2.5229454040527342e-05,
      "model_forward_time": 1.7009704113006592,
      "step": 26455
    },
    {
      "epoch": 2.5229454040527342e-05,
      "step": 26455,
      "training_step_time": 7.8735716342926025
    },
    {
      "epoch": 2.523040771484375e-05,
      "model_forward_time": 1.7242977619171143,
      "step": 26456
    },
    {
      "epoch": 2.523040771484375e-05,
      "step": 26456,
      "training_step_time": 6.807209253311157
    },
    {
      "epoch": 2.5231361389160157e-05,
      "model_forward_time": 1.3306725025177002,
      "step": 26457
    },
    {
      "epoch": 2.5231361389160157e-05,
      "step": 26457,
      "training_step_time": 6.730736494064331
    },
    {
      "epoch": 2.523231506347656e-05,
      "model_forward_time": 1.7111496925354004,
      "step": 26458
    },
    {
      "epoch": 2.523231506347656e-05,
      "step": 26458,
      "training_step_time": 6.727668762207031
    },
    {
      "epoch": 2.523326873779297e-05,
      "model_forward_time": 1.712317943572998,
      "step": 26459
    },
    {
      "epoch": 2.523326873779297e-05,
      "step": 26459,
      "training_step_time": 6.768748998641968
    },
    {
      "epoch": 2.5234222412109376e-05,
      "grad_norm": 0.1288900226354599,
      "learning_rate": 8.792994891674198e-06,
      "loss": 0.0633,
      "step": 26460
    },
    {
      "epoch": 2.5234222412109376e-05,
      "model_forward_time": 1.7126269340515137,
      "step": 26460
    },
    {
      "dynamics_loss_avg": 0.07187723815441131,
      "epoch": 2.5234222412109376e-05,
      "step": 26460
    },
    {
      "action_loss_avg": 0.0028978219022974374,
      "epoch": 2.5234222412109376e-05,
      "step": 26460
    },
    {
      "epoch": 2.5234222412109376e-05,
      "step": 26460,
      "training_step_time": 6.8164238929748535
    },
    {
      "epoch": 2.523517608642578e-05,
      "model_forward_time": 1.7483158111572266,
      "step": 26461
    },
    {
      "epoch": 2.523517608642578e-05,
      "step": 26461,
      "training_step_time": 7.50211763381958
    },
    {
      "epoch": 2.5236129760742187e-05,
      "model_forward_time": 2.850642442703247,
      "step": 26462
    },
    {
      "epoch": 2.5236129760742187e-05,
      "step": 26462,
      "training_step_time": 7.769603252410889
    },
    {
      "epoch": 2.5237083435058595e-05,
      "model_forward_time": 1.7123098373413086,
      "step": 26463
    },
    {
      "epoch": 2.5237083435058595e-05,
      "step": 26463,
      "training_step_time": 10.569827795028687
    },
    {
      "epoch": 2.5238037109375e-05,
      "model_forward_time": 1.7172391414642334,
      "step": 26464
    },
    {
      "epoch": 2.5238037109375e-05,
      "step": 26464,
      "training_step_time": 7.700166940689087
    },
    {
      "epoch": 2.5238990783691406e-05,
      "model_forward_time": 1.7133805751800537,
      "step": 26465
    },
    {
      "epoch": 2.5238990783691406e-05,
      "step": 26465,
      "training_step_time": 7.9418816566467285
    },
    {
      "epoch": 2.5239944458007814e-05,
      "model_forward_time": 1.7228877544403076,
      "step": 26466
    },
    {
      "epoch": 2.5239944458007814e-05,
      "step": 26466,
      "training_step_time": 6.7763378620147705
    },
    {
      "epoch": 2.5240898132324218e-05,
      "model_forward_time": 1.7346270084381104,
      "step": 26467
    },
    {
      "epoch": 2.5240898132324218e-05,
      "step": 26467,
      "training_step_time": 6.779102802276611
    },
    {
      "epoch": 2.5241851806640625e-05,
      "model_forward_time": 1.730926275253296,
      "step": 26468
    },
    {
      "epoch": 2.5241851806640625e-05,
      "step": 26468,
      "training_step_time": 6.788060903549194
    },
    {
      "epoch": 2.5242805480957032e-05,
      "model_forward_time": 1.7426996231079102,
      "step": 26469
    },
    {
      "epoch": 2.5242805480957032e-05,
      "step": 26469,
      "training_step_time": 6.883181810379028
    },
    {
      "epoch": 2.5243759155273436e-05,
      "grad_norm": 0.159791961312294,
      "learning_rate": 8.79191735310935e-06,
      "loss": 0.0609,
      "step": 26470
    },
    {
      "epoch": 2.5243759155273436e-05,
      "model_forward_time": 1.7484612464904785,
      "step": 26470
    },
    {
      "dynamics_loss_avg": 0.06445251367986202,
      "epoch": 2.5243759155273436e-05,
      "step": 26470
    },
    {
      "action_loss_avg": 0.0032750325626693666,
      "epoch": 2.5243759155273436e-05,
      "step": 26470
    },
    {
      "epoch": 2.5243759155273436e-05,
      "step": 26470,
      "training_step_time": 7.091135501861572
    },
    {
      "epoch": 2.5244712829589844e-05,
      "model_forward_time": 1.7620933055877686,
      "step": 26471
    },
    {
      "epoch": 2.5244712829589844e-05,
      "step": 26471,
      "training_step_time": 9.188895463943481
    },
    {
      "epoch": 2.524566650390625e-05,
      "model_forward_time": 1.7297146320343018,
      "step": 26472
    },
    {
      "epoch": 2.524566650390625e-05,
      "step": 26472,
      "training_step_time": 9.695931673049927
    },
    {
      "epoch": 2.5246620178222655e-05,
      "model_forward_time": 1.734302043914795,
      "step": 26473
    },
    {
      "epoch": 2.5246620178222655e-05,
      "step": 26473,
      "training_step_time": 7.719437599182129
    },
    {
      "epoch": 2.5247573852539063e-05,
      "model_forward_time": 1.7381439208984375,
      "step": 26474
    },
    {
      "epoch": 2.5247573852539063e-05,
      "step": 26474,
      "training_step_time": 7.9277184009552
    },
    {
      "epoch": 2.524852752685547e-05,
      "model_forward_time": 1.697319507598877,
      "step": 26475
    },
    {
      "epoch": 2.524852752685547e-05,
      "step": 26475,
      "training_step_time": 6.805738210678101
    },
    {
      "epoch": 2.5249481201171874e-05,
      "model_forward_time": 1.7091553211212158,
      "step": 26476
    },
    {
      "epoch": 2.5249481201171874e-05,
      "step": 26476,
      "training_step_time": 6.820884943008423
    },
    {
      "epoch": 2.525043487548828e-05,
      "model_forward_time": 1.7191479206085205,
      "step": 26477
    },
    {
      "epoch": 2.525043487548828e-05,
      "step": 26477,
      "training_step_time": 6.795119285583496
    },
    {
      "epoch": 2.525138854980469e-05,
      "model_forward_time": 1.7178826332092285,
      "step": 26478
    },
    {
      "epoch": 2.525138854980469e-05,
      "step": 26478,
      "training_step_time": 6.765315771102905
    },
    {
      "epoch": 2.5252342224121093e-05,
      "model_forward_time": 1.7316420078277588,
      "step": 26479
    },
    {
      "epoch": 2.5252342224121093e-05,
      "step": 26479,
      "training_step_time": 7.023534297943115
    },
    {
      "epoch": 2.52532958984375e-05,
      "grad_norm": 0.14310671389102936,
      "learning_rate": 8.790839399866122e-06,
      "loss": 0.0628,
      "step": 26480
    },
    {
      "epoch": 2.52532958984375e-05,
      "model_forward_time": 1.7408292293548584,
      "step": 26480
    },
    {
      "dynamics_loss_avg": 0.07774662375450134,
      "epoch": 2.52532958984375e-05,
      "step": 26480
    },
    {
      "action_loss_avg": 0.003529524884652346,
      "epoch": 2.52532958984375e-05,
      "step": 26480
    },
    {
      "epoch": 2.52532958984375e-05,
      "step": 26480,
      "training_step_time": 8.272985458374023
    },
    {
      "epoch": 2.5254249572753908e-05,
      "model_forward_time": 2.9036457538604736,
      "step": 26481
    },
    {
      "epoch": 2.5254249572753908e-05,
      "step": 26481,
      "training_step_time": 11.451237201690674
    },
    {
      "epoch": 2.5255203247070312e-05,
      "model_forward_time": 1.7083935737609863,
      "step": 26482
    },
    {
      "epoch": 2.5255203247070312e-05,
      "step": 26482,
      "training_step_time": 7.823835372924805
    },
    {
      "epoch": 2.525615692138672e-05,
      "model_forward_time": 1.7102739810943604,
      "step": 26483
    },
    {
      "epoch": 2.525615692138672e-05,
      "step": 26483,
      "training_step_time": 7.8802855014801025
    },
    {
      "epoch": 2.5257110595703127e-05,
      "model_forward_time": 1.714677333831787,
      "step": 26484
    },
    {
      "epoch": 2.5257110595703127e-05,
      "step": 26484,
      "training_step_time": 6.830575466156006
    },
    {
      "epoch": 2.525806427001953e-05,
      "model_forward_time": 1.7207744121551514,
      "step": 26485
    },
    {
      "epoch": 2.525806427001953e-05,
      "step": 26485,
      "training_step_time": 6.803303956985474
    },
    {
      "epoch": 2.5259017944335938e-05,
      "model_forward_time": 1.710003137588501,
      "step": 26486
    },
    {
      "epoch": 2.5259017944335938e-05,
      "step": 26486,
      "training_step_time": 6.863422155380249
    },
    {
      "epoch": 2.5259971618652342e-05,
      "model_forward_time": 1.705575704574585,
      "step": 26487
    },
    {
      "epoch": 2.5259971618652342e-05,
      "step": 26487,
      "training_step_time": 6.838903427124023
    },
    {
      "epoch": 2.526092529296875e-05,
      "model_forward_time": 1.7556090354919434,
      "step": 26488
    },
    {
      "epoch": 2.526092529296875e-05,
      "step": 26488,
      "training_step_time": 7.1169209480285645
    },
    {
      "epoch": 2.5261878967285157e-05,
      "model_forward_time": 1.7177388668060303,
      "step": 26489
    },
    {
      "epoch": 2.5261878967285157e-05,
      "step": 26489,
      "training_step_time": 8.774035930633545
    },
    {
      "epoch": 2.526283264160156e-05,
      "grad_norm": 0.13359296321868896,
      "learning_rate": 8.789761032062398e-06,
      "loss": 0.0616,
      "step": 26490
    },
    {
      "epoch": 2.526283264160156e-05,
      "model_forward_time": 2.8870368003845215,
      "step": 26490
    },
    {
      "dynamics_loss_avg": 0.04221677351742983,
      "epoch": 2.526283264160156e-05,
      "step": 26490
    },
    {
      "action_loss_avg": 0.0016885666991584003,
      "epoch": 2.526283264160156e-05,
      "step": 26490
    },
    {
      "epoch": 2.526283264160156e-05,
      "step": 26490,
      "training_step_time": 10.102850198745728
    },
    {
      "epoch": 2.526378631591797e-05,
      "model_forward_time": 1.715144157409668,
      "step": 26491
    },
    {
      "epoch": 2.526378631591797e-05,
      "step": 26491,
      "training_step_time": 8.519603967666626
    },
    {
      "epoch": 2.5264739990234376e-05,
      "model_forward_time": 1.72200345993042,
      "step": 26492
    },
    {
      "epoch": 2.5264739990234376e-05,
      "step": 26492,
      "training_step_time": 7.662426948547363
    },
    {
      "epoch": 2.526569366455078e-05,
      "model_forward_time": 1.714855432510376,
      "step": 26493
    },
    {
      "epoch": 2.526569366455078e-05,
      "step": 26493,
      "training_step_time": 7.830077648162842
    },
    {
      "epoch": 2.5266647338867187e-05,
      "model_forward_time": 1.714770793914795,
      "step": 26494
    },
    {
      "epoch": 2.5266647338867187e-05,
      "step": 26494,
      "training_step_time": 6.799825429916382
    },
    {
      "epoch": 2.5267601013183595e-05,
      "model_forward_time": 1.7152318954467773,
      "step": 26495
    },
    {
      "epoch": 2.5267601013183595e-05,
      "step": 26495,
      "training_step_time": 6.841279029846191
    },
    {
      "epoch": 2.52685546875e-05,
      "model_forward_time": 1.717939853668213,
      "step": 26496
    },
    {
      "epoch": 2.52685546875e-05,
      "step": 26496,
      "training_step_time": 6.782106637954712
    },
    {
      "epoch": 2.5269508361816406e-05,
      "model_forward_time": 1.7279808521270752,
      "step": 26497
    },
    {
      "epoch": 2.5269508361816406e-05,
      "step": 26497,
      "training_step_time": 7.024479150772095
    },
    {
      "epoch": 2.5270462036132814e-05,
      "model_forward_time": 1.7239601612091064,
      "step": 26498
    },
    {
      "epoch": 2.5270462036132814e-05,
      "step": 26498,
      "training_step_time": 7.660460710525513
    },
    {
      "epoch": 2.5271415710449218e-05,
      "model_forward_time": 2.8116886615753174,
      "step": 26499
    },
    {
      "epoch": 2.5271415710449218e-05,
      "step": 26499,
      "training_step_time": 10.081957817077637
    },
    {
      "epoch": 2.5272369384765625e-05,
      "grad_norm": 0.14788715541362762,
      "learning_rate": 8.788682249816103e-06,
      "loss": 0.0655,
      "step": 26500
    },
    {
      "epoch": 2.5272369384765625e-05,
      "model_forward_time": 1.6991193294525146,
      "step": 26500
    },
    {
      "dynamics_loss_avg": 0.07111103860661387,
      "epoch": 2.5272369384765625e-05,
      "step": 26500
    },
    {
      "action_loss_avg": 0.0031331971869803963,
      "epoch": 2.5272369384765625e-05,
      "step": 26500
    },
    {
      "epoch": 2.5272369384765625e-05,
      "step": 26500,
      "training_step_time": 8.954124212265015
    },
    {
      "epoch": 2.5273323059082032e-05,
      "model_forward_time": 1.714435338973999,
      "step": 26501
    },
    {
      "epoch": 2.5273323059082032e-05,
      "step": 26501,
      "training_step_time": 8.409494400024414
    },
    {
      "epoch": 2.5274276733398436e-05,
      "model_forward_time": 1.7192504405975342,
      "step": 26502
    },
    {
      "epoch": 2.5274276733398436e-05,
      "step": 26502,
      "training_step_time": 7.890431642532349
    },
    {
      "epoch": 2.5275230407714844e-05,
      "model_forward_time": 1.7144584655761719,
      "step": 26503
    },
    {
      "epoch": 2.5275230407714844e-05,
      "step": 26503,
      "training_step_time": 6.787086725234985
    },
    {
      "epoch": 2.527618408203125e-05,
      "model_forward_time": 1.718151569366455,
      "step": 26504
    },
    {
      "epoch": 2.527618408203125e-05,
      "step": 26504,
      "training_step_time": 6.781488418579102
    },
    {
      "epoch": 2.5277137756347655e-05,
      "model_forward_time": 1.717473030090332,
      "step": 26505
    },
    {
      "epoch": 2.5277137756347655e-05,
      "step": 26505,
      "training_step_time": 6.779504776000977
    },
    {
      "epoch": 2.5278091430664063e-05,
      "model_forward_time": 1.7232205867767334,
      "step": 26506
    },
    {
      "epoch": 2.5278091430664063e-05,
      "step": 26506,
      "training_step_time": 7.303223371505737
    },
    {
      "epoch": 2.527904510498047e-05,
      "model_forward_time": 1.7198050022125244,
      "step": 26507
    },
    {
      "epoch": 2.527904510498047e-05,
      "step": 26507,
      "training_step_time": 7.711774826049805
    },
    {
      "epoch": 2.5279998779296874e-05,
      "model_forward_time": 2.8313148021698,
      "step": 26508
    },
    {
      "epoch": 2.5279998779296874e-05,
      "step": 26508,
      "training_step_time": 7.6299147605896
    },
    {
      "epoch": 2.528095245361328e-05,
      "model_forward_time": 1.709923505783081,
      "step": 26509
    },
    {
      "epoch": 2.528095245361328e-05,
      "step": 26509,
      "training_step_time": 9.347535610198975
    },
    {
      "epoch": 2.528190612792969e-05,
      "grad_norm": 0.11965473741292953,
      "learning_rate": 8.787603053245215e-06,
      "loss": 0.0668,
      "step": 26510
    },
    {
      "epoch": 2.528190612792969e-05,
      "model_forward_time": 1.716125726699829,
      "step": 26510
    },
    {
      "dynamics_loss_avg": 0.05107177253812552,
      "epoch": 2.528190612792969e-05,
      "step": 26510
    },
    {
      "action_loss_avg": 0.0021351480856537817,
      "epoch": 2.528190612792969e-05,
      "step": 26510
    },
    {
      "epoch": 2.528190612792969e-05,
      "step": 26510,
      "training_step_time": 7.770723104476929
    },
    {
      "epoch": 2.5282859802246093e-05,
      "model_forward_time": 1.7147436141967773,
      "step": 26511
    },
    {
      "epoch": 2.5282859802246093e-05,
      "step": 26511,
      "training_step_time": 7.693030118942261
    },
    {
      "epoch": 2.52838134765625e-05,
      "model_forward_time": 1.7185721397399902,
      "step": 26512
    },
    {
      "epoch": 2.52838134765625e-05,
      "step": 26512,
      "training_step_time": 6.76804518699646
    },
    {
      "epoch": 2.5284767150878908e-05,
      "model_forward_time": 1.7175581455230713,
      "step": 26513
    },
    {
      "epoch": 2.5284767150878908e-05,
      "step": 26513,
      "training_step_time": 6.768883466720581
    },
    {
      "epoch": 2.5285720825195312e-05,
      "model_forward_time": 1.715409755706787,
      "step": 26514
    },
    {
      "epoch": 2.5285720825195312e-05,
      "step": 26514,
      "training_step_time": 6.78655481338501
    },
    {
      "epoch": 2.528667449951172e-05,
      "model_forward_time": 1.7132182121276855,
      "step": 26515
    },
    {
      "epoch": 2.528667449951172e-05,
      "step": 26515,
      "training_step_time": 6.84757399559021
    },
    {
      "epoch": 2.5287628173828123e-05,
      "model_forward_time": 1.7489683628082275,
      "step": 26516
    },
    {
      "epoch": 2.5287628173828123e-05,
      "step": 26516,
      "training_step_time": 7.685719728469849
    },
    {
      "epoch": 2.528858184814453e-05,
      "model_forward_time": 2.821129560470581,
      "step": 26517
    },
    {
      "epoch": 2.528858184814453e-05,
      "step": 26517,
      "training_step_time": 8.368139028549194
    },
    {
      "epoch": 2.5289535522460938e-05,
      "model_forward_time": 1.7106852531433105,
      "step": 26518
    },
    {
      "epoch": 2.5289535522460938e-05,
      "step": 26518,
      "training_step_time": 11.660248279571533
    },
    {
      "epoch": 2.5290489196777342e-05,
      "model_forward_time": 1.7021551132202148,
      "step": 26519
    },
    {
      "epoch": 2.5290489196777342e-05,
      "step": 26519,
      "training_step_time": 7.757148742675781
    },
    {
      "epoch": 2.529144287109375e-05,
      "grad_norm": 0.12234289944171906,
      "learning_rate": 8.78652344246775e-06,
      "loss": 0.0678,
      "step": 26520
    },
    {
      "epoch": 2.529144287109375e-05,
      "model_forward_time": 1.7154865264892578,
      "step": 26520
    },
    {
      "dynamics_loss_avg": 0.049088595621287825,
      "epoch": 2.529144287109375e-05,
      "step": 26520
    },
    {
      "action_loss_avg": 0.001474994095042348,
      "epoch": 2.529144287109375e-05,
      "step": 26520
    },
    {
      "epoch": 2.529144287109375e-05,
      "step": 26520,
      "training_step_time": 8.002612352371216
    },
    {
      "epoch": 2.5292396545410157e-05,
      "model_forward_time": 1.7105696201324463,
      "step": 26521
    },
    {
      "epoch": 2.5292396545410157e-05,
      "step": 26521,
      "training_step_time": 6.826208829879761
    },
    {
      "epoch": 2.529335021972656e-05,
      "model_forward_time": 1.7064094543457031,
      "step": 26522
    },
    {
      "epoch": 2.529335021972656e-05,
      "step": 26522,
      "training_step_time": 6.793689727783203
    },
    {
      "epoch": 2.529430389404297e-05,
      "model_forward_time": 1.7234392166137695,
      "step": 26523
    },
    {
      "epoch": 2.529430389404297e-05,
      "step": 26523,
      "training_step_time": 6.840652704238892
    },
    {
      "epoch": 2.5295257568359376e-05,
      "model_forward_time": 1.717848777770996,
      "step": 26524
    },
    {
      "epoch": 2.5295257568359376e-05,
      "step": 26524,
      "training_step_time": 6.849385499954224
    },
    {
      "epoch": 2.529621124267578e-05,
      "model_forward_time": 1.745553970336914,
      "step": 26525
    },
    {
      "epoch": 2.529621124267578e-05,
      "step": 26525,
      "training_step_time": 7.6967689990997314
    },
    {
      "epoch": 2.5297164916992187e-05,
      "model_forward_time": 1.7322824001312256,
      "step": 26526
    },
    {
      "epoch": 2.5297164916992187e-05,
      "step": 26526,
      "training_step_time": 8.603165864944458
    },
    {
      "epoch": 2.5298118591308595e-05,
      "model_forward_time": 1.7147409915924072,
      "step": 26527
    },
    {
      "epoch": 2.5298118591308595e-05,
      "step": 26527,
      "training_step_time": 9.624119281768799
    },
    {
      "epoch": 2.5299072265625e-05,
      "model_forward_time": 1.715620517730713,
      "step": 26528
    },
    {
      "epoch": 2.5299072265625e-05,
      "step": 26528,
      "training_step_time": 10.013628721237183
    },
    {
      "epoch": 2.5300025939941406e-05,
      "model_forward_time": 1.7195112705230713,
      "step": 26529
    },
    {
      "epoch": 2.5300025939941406e-05,
      "step": 26529,
      "training_step_time": 8.650118350982666
    },
    {
      "epoch": 2.5300979614257814e-05,
      "grad_norm": 0.1054312139749527,
      "learning_rate": 8.785443417601776e-06,
      "loss": 0.0652,
      "step": 26530
    },
    {
      "epoch": 2.5300979614257814e-05,
      "model_forward_time": 1.7099590301513672,
      "step": 26530
    },
    {
      "dynamics_loss_avg": 0.052464764937758444,
      "epoch": 2.5300979614257814e-05,
      "step": 26530
    },
    {
      "action_loss_avg": 0.002232610702048987,
      "epoch": 2.5300979614257814e-05,
      "step": 26530
    },
    {
      "epoch": 2.5300979614257814e-05,
      "step": 26530,
      "training_step_time": 7.736555099487305
    },
    {
      "epoch": 2.5301933288574218e-05,
      "model_forward_time": 1.7088735103607178,
      "step": 26531
    },
    {
      "epoch": 2.5301933288574218e-05,
      "step": 26531,
      "training_step_time": 6.844073295593262
    },
    {
      "epoch": 2.5302886962890625e-05,
      "model_forward_time": 1.7180240154266357,
      "step": 26532
    },
    {
      "epoch": 2.5302886962890625e-05,
      "step": 26532,
      "training_step_time": 6.835004091262817
    },
    {
      "epoch": 2.5303840637207032e-05,
      "model_forward_time": 1.7102396488189697,
      "step": 26533
    },
    {
      "epoch": 2.5303840637207032e-05,
      "step": 26533,
      "training_step_time": 6.825969219207764
    },
    {
      "epoch": 2.5304794311523436e-05,
      "model_forward_time": 1.7382540702819824,
      "step": 26534
    },
    {
      "epoch": 2.5304794311523436e-05,
      "step": 26534,
      "training_step_time": 7.732501745223999
    },
    {
      "epoch": 2.5305747985839844e-05,
      "model_forward_time": 1.7275876998901367,
      "step": 26535
    },
    {
      "epoch": 2.5305747985839844e-05,
      "step": 26535,
      "training_step_time": 8.360551595687866
    },
    {
      "epoch": 2.530670166015625e-05,
      "model_forward_time": 2.87750244140625,
      "step": 26536
    },
    {
      "epoch": 2.530670166015625e-05,
      "step": 26536,
      "training_step_time": 10.040566682815552
    },
    {
      "epoch": 2.5307655334472655e-05,
      "model_forward_time": 1.715088129043579,
      "step": 26537
    },
    {
      "epoch": 2.5307655334472655e-05,
      "step": 26537,
      "training_step_time": 8.658860206604004
    },
    {
      "epoch": 2.5308609008789063e-05,
      "model_forward_time": 1.7104952335357666,
      "step": 26538
    },
    {
      "epoch": 2.5308609008789063e-05,
      "step": 26538,
      "training_step_time": 8.55159330368042
    },
    {
      "epoch": 2.530956268310547e-05,
      "model_forward_time": 1.71403169631958,
      "step": 26539
    },
    {
      "epoch": 2.530956268310547e-05,
      "step": 26539,
      "training_step_time": 7.693370819091797
    },
    {
      "epoch": 2.5310516357421874e-05,
      "grad_norm": 0.215381920337677,
      "learning_rate": 8.784362978765401e-06,
      "loss": 0.0623,
      "step": 26540
    },
    {
      "epoch": 2.5310516357421874e-05,
      "model_forward_time": 1.7082064151763916,
      "step": 26540
    },
    {
      "dynamics_loss_avg": 0.07096748426556587,
      "epoch": 2.5310516357421874e-05,
      "step": 26540
    },
    {
      "action_loss_avg": 0.0032193067483603953,
      "epoch": 2.5310516357421874e-05,
      "step": 26540
    },
    {
      "epoch": 2.5310516357421874e-05,
      "step": 26540,
      "training_step_time": 6.861759424209595
    },
    {
      "epoch": 2.531147003173828e-05,
      "model_forward_time": 1.714442253112793,
      "step": 26541
    },
    {
      "epoch": 2.531147003173828e-05,
      "step": 26541,
      "training_step_time": 6.7908008098602295
    },
    {
      "epoch": 2.531242370605469e-05,
      "model_forward_time": 1.716170072555542,
      "step": 26542
    },
    {
      "epoch": 2.531242370605469e-05,
      "step": 26542,
      "training_step_time": 6.858124256134033
    },
    {
      "epoch": 2.5313377380371093e-05,
      "model_forward_time": 1.7351088523864746,
      "step": 26543
    },
    {
      "epoch": 2.5313377380371093e-05,
      "step": 26543,
      "training_step_time": 7.822454214096069
    },
    {
      "epoch": 2.53143310546875e-05,
      "model_forward_time": 1.7325637340545654,
      "step": 26544
    },
    {
      "epoch": 2.53143310546875e-05,
      "step": 26544,
      "training_step_time": 8.692307472229004
    },
    {
      "epoch": 2.5315284729003908e-05,
      "model_forward_time": 2.8688535690307617,
      "step": 26545
    },
    {
      "epoch": 2.5315284729003908e-05,
      "step": 26545,
      "training_step_time": 9.25392460823059
    },
    {
      "epoch": 2.5316238403320312e-05,
      "model_forward_time": 1.704927682876587,
      "step": 26546
    },
    {
      "epoch": 2.5316238403320312e-05,
      "step": 26546,
      "training_step_time": 8.829060077667236
    },
    {
      "epoch": 2.531719207763672e-05,
      "model_forward_time": 1.7122931480407715,
      "step": 26547
    },
    {
      "epoch": 2.531719207763672e-05,
      "step": 26547,
      "training_step_time": 7.810282230377197
    },
    {
      "epoch": 2.5318145751953123e-05,
      "model_forward_time": 1.7091467380523682,
      "step": 26548
    },
    {
      "epoch": 2.5318145751953123e-05,
      "step": 26548,
      "training_step_time": 7.49941349029541
    },
    {
      "epoch": 2.531909942626953e-05,
      "model_forward_time": 1.7054355144500732,
      "step": 26549
    },
    {
      "epoch": 2.531909942626953e-05,
      "step": 26549,
      "training_step_time": 6.8643457889556885
    },
    {
      "epoch": 2.5320053100585938e-05,
      "grad_norm": 0.17912663519382477,
      "learning_rate": 8.78328212607678e-06,
      "loss": 0.0628,
      "step": 26550
    },
    {
      "epoch": 2.5320053100585938e-05,
      "model_forward_time": 1.708488941192627,
      "step": 26550
    },
    {
      "dynamics_loss_avg": 0.058143934980034825,
      "epoch": 2.5320053100585938e-05,
      "step": 26550
    },
    {
      "action_loss_avg": 0.0024766066926531494,
      "epoch": 2.5320053100585938e-05,
      "step": 26550
    },
    {
      "epoch": 2.5320053100585938e-05,
      "step": 26550,
      "training_step_time": 6.892549753189087
    },
    {
      "epoch": 2.5321006774902342e-05,
      "model_forward_time": 1.712052345275879,
      "step": 26551
    },
    {
      "epoch": 2.5321006774902342e-05,
      "step": 26551,
      "training_step_time": 6.89445948600769
    },
    {
      "epoch": 2.532196044921875e-05,
      "model_forward_time": 1.712421178817749,
      "step": 26552
    },
    {
      "epoch": 2.532196044921875e-05,
      "step": 26552,
      "training_step_time": 9.24104118347168
    },
    {
      "epoch": 2.5322914123535157e-05,
      "model_forward_time": 1.7247283458709717,
      "step": 26553
    },
    {
      "epoch": 2.5322914123535157e-05,
      "step": 26553,
      "training_step_time": 8.55199909210205
    },
    {
      "epoch": 2.532386779785156e-05,
      "model_forward_time": 2.8579370975494385,
      "step": 26554
    },
    {
      "epoch": 2.532386779785156e-05,
      "step": 26554,
      "training_step_time": 8.232840776443481
    },
    {
      "epoch": 2.532482147216797e-05,
      "model_forward_time": 1.7093069553375244,
      "step": 26555
    },
    {
      "epoch": 2.532482147216797e-05,
      "step": 26555,
      "training_step_time": 9.596976518630981
    },
    {
      "epoch": 2.5325775146484376e-05,
      "model_forward_time": 1.717947006225586,
      "step": 26556
    },
    {
      "epoch": 2.5325775146484376e-05,
      "step": 26556,
      "training_step_time": 7.7894086837768555
    },
    {
      "epoch": 2.532672882080078e-05,
      "model_forward_time": 1.7193512916564941,
      "step": 26557
    },
    {
      "epoch": 2.532672882080078e-05,
      "step": 26557,
      "training_step_time": 7.4306700229644775
    },
    {
      "epoch": 2.5327682495117187e-05,
      "model_forward_time": 1.7103192806243896,
      "step": 26558
    },
    {
      "epoch": 2.5327682495117187e-05,
      "step": 26558,
      "training_step_time": 7.081241846084595
    },
    {
      "epoch": 2.5328636169433595e-05,
      "model_forward_time": 1.7109603881835938,
      "step": 26559
    },
    {
      "epoch": 2.5328636169433595e-05,
      "step": 26559,
      "training_step_time": 6.811969757080078
    },
    {
      "epoch": 2.532958984375e-05,
      "grad_norm": 0.15085266530513763,
      "learning_rate": 8.782200859654113e-06,
      "loss": 0.061,
      "step": 26560
    },
    {
      "epoch": 2.532958984375e-05,
      "model_forward_time": 1.7145435810089111,
      "step": 26560
    },
    {
      "dynamics_loss_avg": 0.05702867228537798,
      "epoch": 2.532958984375e-05,
      "step": 26560
    },
    {
      "action_loss_avg": 0.0021916491445153953,
      "epoch": 2.532958984375e-05,
      "step": 26560
    },
    {
      "epoch": 2.532958984375e-05,
      "step": 26560,
      "training_step_time": 6.889306306838989
    },
    {
      "epoch": 2.5330543518066406e-05,
      "model_forward_time": 1.7109005451202393,
      "step": 26561
    },
    {
      "epoch": 2.5330543518066406e-05,
      "step": 26561,
      "training_step_time": 7.705755233764648
    },
    {
      "epoch": 2.5331497192382814e-05,
      "model_forward_time": 1.7518229484558105,
      "step": 26562
    },
    {
      "epoch": 2.5331497192382814e-05,
      "step": 26562,
      "training_step_time": 7.729521751403809
    },
    {
      "epoch": 2.5332450866699218e-05,
      "model_forward_time": 2.8640236854553223,
      "step": 26563
    },
    {
      "epoch": 2.5332450866699218e-05,
      "step": 26563,
      "training_step_time": 7.745559453964233
    },
    {
      "epoch": 2.5333404541015625e-05,
      "model_forward_time": 1.7055981159210205,
      "step": 26564
    },
    {
      "epoch": 2.5333404541015625e-05,
      "step": 26564,
      "training_step_time": 10.096107959747314
    },
    {
      "epoch": 2.5334358215332032e-05,
      "model_forward_time": 1.7138848304748535,
      "step": 26565
    },
    {
      "epoch": 2.5334358215332032e-05,
      "step": 26565,
      "training_step_time": 9.709333419799805
    },
    {
      "epoch": 2.5335311889648436e-05,
      "model_forward_time": 1.7167248725891113,
      "step": 26566
    },
    {
      "epoch": 2.5335311889648436e-05,
      "step": 26566,
      "training_step_time": 7.647004842758179
    },
    {
      "epoch": 2.5336265563964844e-05,
      "model_forward_time": 1.7145798206329346,
      "step": 26567
    },
    {
      "epoch": 2.5336265563964844e-05,
      "step": 26567,
      "training_step_time": 7.158836126327515
    },
    {
      "epoch": 2.533721923828125e-05,
      "model_forward_time": 1.7104403972625732,
      "step": 26568
    },
    {
      "epoch": 2.533721923828125e-05,
      "step": 26568,
      "training_step_time": 6.928028345108032
    },
    {
      "epoch": 2.5338172912597655e-05,
      "model_forward_time": 1.7135872840881348,
      "step": 26569
    },
    {
      "epoch": 2.5338172912597655e-05,
      "step": 26569,
      "training_step_time": 6.845012187957764
    },
    {
      "epoch": 2.5339126586914063e-05,
      "grad_norm": 0.18134558200836182,
      "learning_rate": 8.781119179615647e-06,
      "loss": 0.0651,
      "step": 26570
    },
    {
      "epoch": 2.5339126586914063e-05,
      "model_forward_time": 1.720423698425293,
      "step": 26570
    },
    {
      "dynamics_loss_avg": 0.05058034639805555,
      "epoch": 2.5339126586914063e-05,
      "step": 26570
    },
    {
      "action_loss_avg": 0.0017920621961820872,
      "epoch": 2.5339126586914063e-05,
      "step": 26570
    },
    {
      "epoch": 2.5339126586914063e-05,
      "step": 26570,
      "training_step_time": 6.914184808731079
    },
    {
      "epoch": 2.534008026123047e-05,
      "model_forward_time": 1.7597246170043945,
      "step": 26571
    },
    {
      "epoch": 2.534008026123047e-05,
      "step": 26571,
      "training_step_time": 8.429697036743164
    },
    {
      "epoch": 2.5341033935546874e-05,
      "model_forward_time": 2.8648691177368164,
      "step": 26572
    },
    {
      "epoch": 2.5341033935546874e-05,
      "step": 26572,
      "training_step_time": 8.170238018035889
    },
    {
      "epoch": 2.534198760986328e-05,
      "model_forward_time": 1.7309978008270264,
      "step": 26573
    },
    {
      "epoch": 2.534198760986328e-05,
      "step": 26573,
      "training_step_time": 10.329896450042725
    },
    {
      "epoch": 2.534294128417969e-05,
      "model_forward_time": 1.7346417903900146,
      "step": 26574
    },
    {
      "epoch": 2.534294128417969e-05,
      "step": 26574,
      "training_step_time": 8.626144409179688
    },
    {
      "epoch": 2.5343894958496093e-05,
      "model_forward_time": 1.729862928390503,
      "step": 26575
    },
    {
      "epoch": 2.5343894958496093e-05,
      "step": 26575,
      "training_step_time": 7.636773109436035
    },
    {
      "epoch": 2.53448486328125e-05,
      "model_forward_time": 1.7391855716705322,
      "step": 26576
    },
    {
      "epoch": 2.53448486328125e-05,
      "step": 26576,
      "training_step_time": 7.417132616043091
    },
    {
      "epoch": 2.5345802307128908e-05,
      "model_forward_time": 1.7334270477294922,
      "step": 26577
    },
    {
      "epoch": 2.5345802307128908e-05,
      "step": 26577,
      "training_step_time": 6.848755359649658
    },
    {
      "epoch": 2.5346755981445312e-05,
      "model_forward_time": 1.7402260303497314,
      "step": 26578
    },
    {
      "epoch": 2.5346755981445312e-05,
      "step": 26578,
      "training_step_time": 6.8372132778167725
    },
    {
      "epoch": 2.534770965576172e-05,
      "model_forward_time": 1.7332308292388916,
      "step": 26579
    },
    {
      "epoch": 2.534770965576172e-05,
      "step": 26579,
      "training_step_time": 6.81517481803894
    },
    {
      "epoch": 2.5348663330078123e-05,
      "grad_norm": 0.20233525335788727,
      "learning_rate": 8.780037086079674e-06,
      "loss": 0.062,
      "step": 26580
    },
    {
      "epoch": 2.5348663330078123e-05,
      "model_forward_time": 1.7809832096099854,
      "step": 26580
    },
    {
      "dynamics_loss_avg": 0.05440694075077772,
      "epoch": 2.5348663330078123e-05,
      "step": 26580
    },
    {
      "action_loss_avg": 0.0041529601730871946,
      "epoch": 2.5348663330078123e-05,
      "step": 26580
    },
    {
      "epoch": 2.5348663330078123e-05,
      "step": 26580,
      "training_step_time": 7.710156679153442
    },
    {
      "epoch": 2.534961700439453e-05,
      "model_forward_time": 1.763261079788208,
      "step": 26581
    },
    {
      "epoch": 2.534961700439453e-05,
      "step": 26581,
      "training_step_time": 9.030219554901123
    },
    {
      "epoch": 2.5350570678710938e-05,
      "model_forward_time": 1.7303082942962646,
      "step": 26582
    },
    {
      "epoch": 2.5350570678710938e-05,
      "step": 26582,
      "training_step_time": 9.371251583099365
    },
    {
      "epoch": 2.5351524353027342e-05,
      "model_forward_time": 1.7309391498565674,
      "step": 26583
    },
    {
      "epoch": 2.5351524353027342e-05,
      "step": 26583,
      "training_step_time": 9.835461378097534
    },
    {
      "epoch": 2.535247802734375e-05,
      "model_forward_time": 1.7359414100646973,
      "step": 26584
    },
    {
      "epoch": 2.535247802734375e-05,
      "step": 26584,
      "training_step_time": 8.344915866851807
    },
    {
      "epoch": 2.5353431701660157e-05,
      "model_forward_time": 1.7341036796569824,
      "step": 26585
    },
    {
      "epoch": 2.5353431701660157e-05,
      "step": 26585,
      "training_step_time": 7.3620758056640625
    },
    {
      "epoch": 2.535438537597656e-05,
      "model_forward_time": 1.7349293231964111,
      "step": 26586
    },
    {
      "epoch": 2.535438537597656e-05,
      "step": 26586,
      "training_step_time": 6.779362201690674
    },
    {
      "epoch": 2.535533905029297e-05,
      "model_forward_time": 1.7288968563079834,
      "step": 26587
    },
    {
      "epoch": 2.535533905029297e-05,
      "step": 26587,
      "training_step_time": 6.696406841278076
    },
    {
      "epoch": 2.5356292724609376e-05,
      "model_forward_time": 1.7339932918548584,
      "step": 26588
    },
    {
      "epoch": 2.5356292724609376e-05,
      "step": 26588,
      "training_step_time": 6.7990570068359375
    },
    {
      "epoch": 2.535724639892578e-05,
      "model_forward_time": 1.7325456142425537,
      "step": 26589
    },
    {
      "epoch": 2.535724639892578e-05,
      "step": 26589,
      "training_step_time": 7.749108791351318
    },
    {
      "epoch": 2.5358200073242187e-05,
      "grad_norm": 0.133420929312706,
      "learning_rate": 8.778954579164527e-06,
      "loss": 0.0629,
      "step": 26590
    },
    {
      "epoch": 2.5358200073242187e-05,
      "model_forward_time": 1.7439160346984863,
      "step": 26590
    },
    {
      "dynamics_loss_avg": 0.07084221169352531,
      "epoch": 2.5358200073242187e-05,
      "step": 26590
    },
    {
      "action_loss_avg": 0.002857800724450499,
      "epoch": 2.5358200073242187e-05,
      "step": 26590
    },
    {
      "epoch": 2.5358200073242187e-05,
      "step": 26590,
      "training_step_time": 7.653856992721558
    },
    {
      "epoch": 2.5359153747558595e-05,
      "model_forward_time": 2.956329584121704,
      "step": 26591
    },
    {
      "epoch": 2.5359153747558595e-05,
      "step": 26591,
      "training_step_time": 10.306079149246216
    },
    {
      "epoch": 2.5360107421875e-05,
      "model_forward_time": 1.7309527397155762,
      "step": 26592
    },
    {
      "epoch": 2.5360107421875e-05,
      "step": 26592,
      "training_step_time": 8.610556840896606
    },
    {
      "epoch": 2.5361061096191406e-05,
      "model_forward_time": 1.7368924617767334,
      "step": 26593
    },
    {
      "epoch": 2.5361061096191406e-05,
      "step": 26593,
      "training_step_time": 7.761708974838257
    },
    {
      "epoch": 2.5362014770507814e-05,
      "model_forward_time": 1.7278695106506348,
      "step": 26594
    },
    {
      "epoch": 2.5362014770507814e-05,
      "step": 26594,
      "training_step_time": 8.51882266998291
    },
    {
      "epoch": 2.5362968444824218e-05,
      "model_forward_time": 1.735771894454956,
      "step": 26595
    },
    {
      "epoch": 2.5362968444824218e-05,
      "step": 26595,
      "training_step_time": 6.806033134460449
    },
    {
      "epoch": 2.5363922119140625e-05,
      "model_forward_time": 1.7378454208374023,
      "step": 26596
    },
    {
      "epoch": 2.5363922119140625e-05,
      "step": 26596,
      "training_step_time": 6.8218138217926025
    },
    {
      "epoch": 2.5364875793457032e-05,
      "model_forward_time": 1.73899245262146,
      "step": 26597
    },
    {
      "epoch": 2.5364875793457032e-05,
      "step": 26597,
      "training_step_time": 6.816419363021851
    },
    {
      "epoch": 2.5365829467773437e-05,
      "model_forward_time": 1.739469051361084,
      "step": 26598
    },
    {
      "epoch": 2.5365829467773437e-05,
      "step": 26598,
      "training_step_time": 7.762451648712158
    },
    {
      "epoch": 2.5366783142089844e-05,
      "model_forward_time": 1.7916603088378906,
      "step": 26599
    },
    {
      "epoch": 2.5366783142089844e-05,
      "step": 26599,
      "training_step_time": 7.70698881149292
    },
    {
      "epoch": 2.536773681640625e-05,
      "grad_norm": 0.16458973288536072,
      "learning_rate": 8.777871658988589e-06,
      "loss": 0.0621,
      "step": 26600
    },
    {
      "epoch": 2.536773681640625e-05,
      "model_forward_time": 2.898495674133301,
      "step": 26600
    },
    {
      "dynamics_loss_avg": 0.06788431834429502,
      "epoch": 2.536773681640625e-05,
      "step": 26600
    },
    {
      "action_loss_avg": 0.0023181181983090936,
      "epoch": 2.536773681640625e-05,
      "step": 26600
    },
    {
      "epoch": 2.536773681640625e-05,
      "step": 26600,
      "training_step_time": 8.976975917816162
    },
    {
      "epoch": 2.5368690490722655e-05,
      "model_forward_time": 1.7314209938049316,
      "step": 26601
    },
    {
      "epoch": 2.5368690490722655e-05,
      "step": 26601,
      "training_step_time": 10.305309057235718
    },
    {
      "epoch": 2.5369644165039063e-05,
      "model_forward_time": 1.3593971729278564,
      "step": 26602
    },
    {
      "epoch": 2.5369644165039063e-05,
      "step": 26602,
      "training_step_time": 8.456520557403564
    },
    {
      "epoch": 2.537059783935547e-05,
      "model_forward_time": 1.7349109649658203,
      "step": 26603
    },
    {
      "epoch": 2.537059783935547e-05,
      "step": 26603,
      "training_step_time": 8.474885702133179
    },
    {
      "epoch": 2.5371551513671874e-05,
      "model_forward_time": 1.737095832824707,
      "step": 26604
    },
    {
      "epoch": 2.5371551513671874e-05,
      "step": 26604,
      "training_step_time": 7.130443572998047
    },
    {
      "epoch": 2.537250518798828e-05,
      "model_forward_time": 1.7311515808105469,
      "step": 26605
    },
    {
      "epoch": 2.537250518798828e-05,
      "step": 26605,
      "training_step_time": 6.824311256408691
    },
    {
      "epoch": 2.537345886230469e-05,
      "model_forward_time": 1.730593204498291,
      "step": 26606
    },
    {
      "epoch": 2.537345886230469e-05,
      "step": 26606,
      "training_step_time": 6.910720109939575
    },
    {
      "epoch": 2.5374412536621093e-05,
      "model_forward_time": 1.7363872528076172,
      "step": 26607
    },
    {
      "epoch": 2.5374412536621093e-05,
      "step": 26607,
      "training_step_time": 7.735029220581055
    },
    {
      "epoch": 2.53753662109375e-05,
      "model_forward_time": 1.7571115493774414,
      "step": 26608
    },
    {
      "epoch": 2.53753662109375e-05,
      "step": 26608,
      "training_step_time": 7.673151731491089
    },
    {
      "epoch": 2.5376319885253908e-05,
      "model_forward_time": 2.852046251296997,
      "step": 26609
    },
    {
      "epoch": 2.5376319885253908e-05,
      "step": 26609,
      "training_step_time": 9.300292015075684
    },
    {
      "epoch": 2.5377273559570312e-05,
      "grad_norm": 0.17525768280029297,
      "learning_rate": 8.776788325670285e-06,
      "loss": 0.0622,
      "step": 26610
    },
    {
      "epoch": 2.5377273559570312e-05,
      "model_forward_time": 1.7319145202636719,
      "step": 26610
    },
    {
      "dynamics_loss_avg": 0.05467678140848875,
      "epoch": 2.5377273559570312e-05,
      "step": 26610
    },
    {
      "action_loss_avg": 0.0020294765912694857,
      "epoch": 2.5377273559570312e-05,
      "step": 26610
    },
    {
      "epoch": 2.5377273559570312e-05,
      "step": 26610,
      "training_step_time": 8.723207950592041
    },
    {
      "epoch": 2.537822723388672e-05,
      "model_forward_time": 1.7346522808074951,
      "step": 26611
    },
    {
      "epoch": 2.537822723388672e-05,
      "step": 26611,
      "training_step_time": 7.762784242630005
    },
    {
      "epoch": 2.5379180908203123e-05,
      "model_forward_time": 1.7321109771728516,
      "step": 26612
    },
    {
      "epoch": 2.5379180908203123e-05,
      "step": 26612,
      "training_step_time": 8.295028448104858
    },
    {
      "epoch": 2.538013458251953e-05,
      "model_forward_time": 1.7396280765533447,
      "step": 26613
    },
    {
      "epoch": 2.538013458251953e-05,
      "step": 26613,
      "training_step_time": 7.405122518539429
    },
    {
      "epoch": 2.5381088256835938e-05,
      "model_forward_time": 1.7373096942901611,
      "step": 26614
    },
    {
      "epoch": 2.5381088256835938e-05,
      "step": 26614,
      "training_step_time": 6.841024875640869
    },
    {
      "epoch": 2.5382041931152342e-05,
      "model_forward_time": 1.7343120574951172,
      "step": 26615
    },
    {
      "epoch": 2.5382041931152342e-05,
      "step": 26615,
      "training_step_time": 6.844214677810669
    },
    {
      "epoch": 2.538299560546875e-05,
      "model_forward_time": 1.737440586090088,
      "step": 26616
    },
    {
      "epoch": 2.538299560546875e-05,
      "step": 26616,
      "training_step_time": 7.617339849472046
    },
    {
      "epoch": 2.5383949279785157e-05,
      "model_forward_time": 1.7518129348754883,
      "step": 26617
    },
    {
      "epoch": 2.5383949279785157e-05,
      "step": 26617,
      "training_step_time": 8.64953064918518
    },
    {
      "epoch": 2.538490295410156e-05,
      "model_forward_time": 2.836587429046631,
      "step": 26618
    },
    {
      "epoch": 2.538490295410156e-05,
      "step": 26618,
      "training_step_time": 8.565223693847656
    },
    {
      "epoch": 2.538585662841797e-05,
      "model_forward_time": 1.7294442653656006,
      "step": 26619
    },
    {
      "epoch": 2.538585662841797e-05,
      "step": 26619,
      "training_step_time": 10.257458209991455
    },
    {
      "epoch": 2.5386810302734376e-05,
      "grad_norm": 0.17668680846691132,
      "learning_rate": 8.77570457932809e-06,
      "loss": 0.0629,
      "step": 26620
    },
    {
      "epoch": 2.5386810302734376e-05,
      "model_forward_time": 1.7300984859466553,
      "step": 26620
    },
    {
      "dynamics_loss_avg": 0.04211130887269974,
      "epoch": 2.5386810302734376e-05,
      "step": 26620
    },
    {
      "action_loss_avg": 0.001271883855224587,
      "epoch": 2.5386810302734376e-05,
      "step": 26620
    },
    {
      "epoch": 2.5386810302734376e-05,
      "step": 26620,
      "training_step_time": 8.621062755584717
    },
    {
      "epoch": 2.538776397705078e-05,
      "model_forward_time": 1.7366173267364502,
      "step": 26621
    },
    {
      "epoch": 2.538776397705078e-05,
      "step": 26621,
      "training_step_time": 7.743294715881348
    },
    {
      "epoch": 2.5388717651367187e-05,
      "model_forward_time": 1.7327606678009033,
      "step": 26622
    },
    {
      "epoch": 2.5388717651367187e-05,
      "step": 26622,
      "training_step_time": 7.349581956863403
    },
    {
      "epoch": 2.5389671325683595e-05,
      "model_forward_time": 1.7401564121246338,
      "step": 26623
    },
    {
      "epoch": 2.5389671325683595e-05,
      "step": 26623,
      "training_step_time": 6.794084310531616
    },
    {
      "epoch": 2.5390625e-05,
      "model_forward_time": 1.7361316680908203,
      "step": 26624
    },
    {
      "epoch": 2.5390625e-05,
      "step": 26624,
      "training_step_time": 6.804259300231934
    },
    {
      "epoch": 2.5391578674316406e-05,
      "model_forward_time": 1.7297108173370361,
      "step": 26625
    },
    {
      "epoch": 2.5391578674316406e-05,
      "step": 26625,
      "training_step_time": 7.357749938964844
    },
    {
      "epoch": 2.5392532348632814e-05,
      "model_forward_time": 1.7390375137329102,
      "step": 26626
    },
    {
      "epoch": 2.5392532348632814e-05,
      "step": 26626,
      "training_step_time": 7.692194223403931
    },
    {
      "epoch": 2.5393486022949218e-05,
      "model_forward_time": 1.7636160850524902,
      "step": 26627
    },
    {
      "epoch": 2.5393486022949218e-05,
      "step": 26627,
      "training_step_time": 7.565781354904175
    },
    {
      "epoch": 2.5394439697265625e-05,
      "model_forward_time": 2.9096875190734863,
      "step": 26628
    },
    {
      "epoch": 2.5394439697265625e-05,
      "step": 26628,
      "training_step_time": 8.664474248886108
    },
    {
      "epoch": 2.5395393371582033e-05,
      "model_forward_time": 1.737910509109497,
      "step": 26629
    },
    {
      "epoch": 2.5395393371582033e-05,
      "step": 26629,
      "training_step_time": 9.053296566009521
    },
    {
      "epoch": 2.5396347045898437e-05,
      "grad_norm": 0.1428593248128891,
      "learning_rate": 8.774620420080517e-06,
      "loss": 0.0615,
      "step": 26630
    },
    {
      "epoch": 2.5396347045898437e-05,
      "model_forward_time": 1.727447271347046,
      "step": 26630
    },
    {
      "dynamics_loss_avg": 0.057060695625841616,
      "epoch": 2.5396347045898437e-05,
      "step": 26630
    },
    {
      "action_loss_avg": 0.002870458411052823,
      "epoch": 2.5396347045898437e-05,
      "step": 26630
    },
    {
      "epoch": 2.5396347045898437e-05,
      "step": 26630,
      "training_step_time": 7.682180881500244
    },
    {
      "epoch": 2.5397300720214844e-05,
      "model_forward_time": 1.7292149066925049,
      "step": 26631
    },
    {
      "epoch": 2.5397300720214844e-05,
      "step": 26631,
      "training_step_time": 7.481361389160156
    },
    {
      "epoch": 2.539825439453125e-05,
      "model_forward_time": 1.736769437789917,
      "step": 26632
    },
    {
      "epoch": 2.539825439453125e-05,
      "step": 26632,
      "training_step_time": 7.029557228088379
    },
    {
      "epoch": 2.5399208068847655e-05,
      "model_forward_time": 1.7348101139068604,
      "step": 26633
    },
    {
      "epoch": 2.5399208068847655e-05,
      "step": 26633,
      "training_step_time": 6.792855739593506
    },
    {
      "epoch": 2.5400161743164063e-05,
      "model_forward_time": 1.737581491470337,
      "step": 26634
    },
    {
      "epoch": 2.5400161743164063e-05,
      "step": 26634,
      "training_step_time": 7.372753858566284
    },
    {
      "epoch": 2.540111541748047e-05,
      "model_forward_time": 1.7239151000976562,
      "step": 26635
    },
    {
      "epoch": 2.540111541748047e-05,
      "step": 26635,
      "training_step_time": 8.135818481445312
    },
    {
      "epoch": 2.5402069091796874e-05,
      "model_forward_time": 1.7594096660614014,
      "step": 26636
    },
    {
      "epoch": 2.5402069091796874e-05,
      "step": 26636,
      "training_step_time": 7.591601848602295
    },
    {
      "epoch": 2.540302276611328e-05,
      "model_forward_time": 2.9429922103881836,
      "step": 26637
    },
    {
      "epoch": 2.540302276611328e-05,
      "step": 26637,
      "training_step_time": 8.6753830909729
    },
    {
      "epoch": 2.540397644042969e-05,
      "model_forward_time": 1.7043790817260742,
      "step": 26638
    },
    {
      "epoch": 2.540397644042969e-05,
      "step": 26638,
      "training_step_time": 8.710381984710693
    },
    {
      "epoch": 2.5404930114746093e-05,
      "model_forward_time": 1.708425521850586,
      "step": 26639
    },
    {
      "epoch": 2.5404930114746093e-05,
      "step": 26639,
      "training_step_time": 8.81596565246582
    },
    {
      "epoch": 2.54058837890625e-05,
      "grad_norm": 0.11097157001495361,
      "learning_rate": 8.773535848046131e-06,
      "loss": 0.0623,
      "step": 26640
    },
    {
      "epoch": 2.54058837890625e-05,
      "model_forward_time": 1.7061753273010254,
      "step": 26640
    },
    {
      "dynamics_loss_avg": 0.07186689861118793,
      "epoch": 2.54058837890625e-05,
      "step": 26640
    },
    {
      "action_loss_avg": 0.003387359992484562,
      "epoch": 2.54058837890625e-05,
      "step": 26640
    },
    {
      "epoch": 2.54058837890625e-05,
      "step": 26640,
      "training_step_time": 7.573051691055298
    },
    {
      "epoch": 2.5406837463378905e-05,
      "model_forward_time": 1.710465669631958,
      "step": 26641
    },
    {
      "epoch": 2.5406837463378905e-05,
      "step": 26641,
      "training_step_time": 7.178609132766724
    },
    {
      "epoch": 2.5407791137695312e-05,
      "model_forward_time": 1.7135448455810547,
      "step": 26642
    },
    {
      "epoch": 2.5407791137695312e-05,
      "step": 26642,
      "training_step_time": 6.697600364685059
    },
    {
      "epoch": 2.540874481201172e-05,
      "model_forward_time": 1.716486930847168,
      "step": 26643
    },
    {
      "epoch": 2.540874481201172e-05,
      "step": 26643,
      "training_step_time": 7.3391642570495605
    },
    {
      "epoch": 2.5409698486328123e-05,
      "model_forward_time": 1.710930585861206,
      "step": 26644
    },
    {
      "epoch": 2.5409698486328123e-05,
      "step": 26644,
      "training_step_time": 7.725318908691406
    },
    {
      "epoch": 2.541065216064453e-05,
      "model_forward_time": 1.7756071090698242,
      "step": 26645
    },
    {
      "epoch": 2.541065216064453e-05,
      "step": 26645,
      "training_step_time": 8.44118618965149
    },
    {
      "epoch": 2.5411605834960938e-05,
      "model_forward_time": 2.8881444931030273,
      "step": 26646
    },
    {
      "epoch": 2.5411605834960938e-05,
      "step": 26646,
      "training_step_time": 8.642153263092041
    },
    {
      "epoch": 2.5412559509277342e-05,
      "model_forward_time": 1.705132007598877,
      "step": 26647
    },
    {
      "epoch": 2.5412559509277342e-05,
      "step": 26647,
      "training_step_time": 8.713370561599731
    },
    {
      "epoch": 2.541351318359375e-05,
      "model_forward_time": 1.7111682891845703,
      "step": 26648
    },
    {
      "epoch": 2.541351318359375e-05,
      "step": 26648,
      "training_step_time": 7.660487651824951
    },
    {
      "epoch": 2.5414466857910157e-05,
      "model_forward_time": 1.7053933143615723,
      "step": 26649
    },
    {
      "epoch": 2.5414466857910157e-05,
      "step": 26649,
      "training_step_time": 8.412890434265137
    },
    {
      "epoch": 2.541542053222656e-05,
      "grad_norm": 0.11511335521936417,
      "learning_rate": 8.772450863343538e-06,
      "loss": 0.0602,
      "step": 26650
    },
    {
      "epoch": 2.541542053222656e-05,
      "model_forward_time": 1.706691026687622,
      "step": 26650
    },
    {
      "dynamics_loss_avg": 0.06545752789825202,
      "epoch": 2.541542053222656e-05,
      "step": 26650
    },
    {
      "action_loss_avg": 0.0030036999378353357,
      "epoch": 2.541542053222656e-05,
      "step": 26650
    },
    {
      "epoch": 2.541542053222656e-05,
      "step": 26650,
      "training_step_time": 7.190412759780884
    },
    {
      "epoch": 2.541637420654297e-05,
      "model_forward_time": 1.711963176727295,
      "step": 26651
    },
    {
      "epoch": 2.541637420654297e-05,
      "step": 26651,
      "training_step_time": 6.766069650650024
    },
    {
      "epoch": 2.5417327880859376e-05,
      "model_forward_time": 1.7127137184143066,
      "step": 26652
    },
    {
      "epoch": 2.5417327880859376e-05,
      "step": 26652,
      "training_step_time": 7.385308265686035
    },
    {
      "epoch": 2.541828155517578e-05,
      "model_forward_time": 1.7168467044830322,
      "step": 26653
    },
    {
      "epoch": 2.541828155517578e-05,
      "step": 26653,
      "training_step_time": 7.654092073440552
    },
    {
      "epoch": 2.5419235229492187e-05,
      "model_forward_time": 1.7602465152740479,
      "step": 26654
    },
    {
      "epoch": 2.5419235229492187e-05,
      "step": 26654,
      "training_step_time": 7.647838592529297
    },
    {
      "epoch": 2.5420188903808595e-05,
      "model_forward_time": 2.8663570880889893,
      "step": 26655
    },
    {
      "epoch": 2.5420188903808595e-05,
      "step": 26655,
      "training_step_time": 7.661565065383911
    },
    {
      "epoch": 2.5421142578125e-05,
      "model_forward_time": 1.7160723209381104,
      "step": 26656
    },
    {
      "epoch": 2.5421142578125e-05,
      "step": 26656,
      "training_step_time": 10.088584661483765
    },
    {
      "epoch": 2.5422096252441406e-05,
      "model_forward_time": 1.7135510444641113,
      "step": 26657
    },
    {
      "epoch": 2.5422096252441406e-05,
      "step": 26657,
      "training_step_time": 8.195815563201904
    },
    {
      "epoch": 2.5423049926757814e-05,
      "model_forward_time": 1.7224035263061523,
      "step": 26658
    },
    {
      "epoch": 2.5423049926757814e-05,
      "step": 26658,
      "training_step_time": 7.709398984909058
    },
    {
      "epoch": 2.5424003601074218e-05,
      "model_forward_time": 1.7209839820861816,
      "step": 26659
    },
    {
      "epoch": 2.5424003601074218e-05,
      "step": 26659,
      "training_step_time": 7.326372861862183
    },
    {
      "epoch": 2.5424957275390625e-05,
      "grad_norm": 0.16979490220546722,
      "learning_rate": 8.77136546609139e-06,
      "loss": 0.0618,
      "step": 26660
    },
    {
      "epoch": 2.5424957275390625e-05,
      "model_forward_time": 1.7152159214019775,
      "step": 26660
    },
    {
      "dynamics_loss_avg": 0.051892487518489364,
      "epoch": 2.5424957275390625e-05,
      "step": 26660
    },
    {
      "action_loss_avg": 0.0020622353826183827,
      "epoch": 2.5424957275390625e-05,
      "step": 26660
    },
    {
      "epoch": 2.5424957275390625e-05,
      "step": 26660,
      "training_step_time": 7.048422336578369
    },
    {
      "epoch": 2.5425910949707033e-05,
      "model_forward_time": 1.719470739364624,
      "step": 26661
    },
    {
      "epoch": 2.5425910949707033e-05,
      "step": 26661,
      "training_step_time": 7.374099254608154
    },
    {
      "epoch": 2.5426864624023437e-05,
      "model_forward_time": 1.7175183296203613,
      "step": 26662
    },
    {
      "epoch": 2.5426864624023437e-05,
      "step": 26662,
      "training_step_time": 7.564208030700684
    },
    {
      "epoch": 2.5427818298339844e-05,
      "model_forward_time": 1.7506184577941895,
      "step": 26663
    },
    {
      "epoch": 2.5427818298339844e-05,
      "step": 26663,
      "training_step_time": 7.747970104217529
    },
    {
      "epoch": 2.542877197265625e-05,
      "model_forward_time": 2.832359790802002,
      "step": 26664
    },
    {
      "epoch": 2.542877197265625e-05,
      "step": 26664,
      "training_step_time": 8.609527826309204
    },
    {
      "epoch": 2.5429725646972655e-05,
      "model_forward_time": 1.715679407119751,
      "step": 26665
    },
    {
      "epoch": 2.5429725646972655e-05,
      "step": 26665,
      "training_step_time": 9.055191993713379
    },
    {
      "epoch": 2.5430679321289063e-05,
      "model_forward_time": 1.7122657299041748,
      "step": 26666
    },
    {
      "epoch": 2.5430679321289063e-05,
      "step": 26666,
      "training_step_time": 9.564351558685303
    },
    {
      "epoch": 2.543163299560547e-05,
      "model_forward_time": 1.7177422046661377,
      "step": 26667
    },
    {
      "epoch": 2.543163299560547e-05,
      "step": 26667,
      "training_step_time": 8.396703481674194
    },
    {
      "epoch": 2.5432586669921874e-05,
      "model_forward_time": 1.7112441062927246,
      "step": 26668
    },
    {
      "epoch": 2.5432586669921874e-05,
      "step": 26668,
      "training_step_time": 7.691376447677612
    },
    {
      "epoch": 2.5433540344238282e-05,
      "model_forward_time": 1.722060203552246,
      "step": 26669
    },
    {
      "epoch": 2.5433540344238282e-05,
      "step": 26669,
      "training_step_time": 7.113773345947266
    },
    {
      "epoch": 2.543449401855469e-05,
      "grad_norm": 0.1631767600774765,
      "learning_rate": 8.770279656408386e-06,
      "loss": 0.0602,
      "step": 26670
    },
    {
      "epoch": 2.543449401855469e-05,
      "model_forward_time": 1.7092785835266113,
      "step": 26670
    },
    {
      "dynamics_loss_avg": 0.06288804486393929,
      "epoch": 2.543449401855469e-05,
      "step": 26670
    },
    {
      "action_loss_avg": 0.0031431945564690976,
      "epoch": 2.543449401855469e-05,
      "step": 26670
    },
    {
      "epoch": 2.543449401855469e-05,
      "step": 26670,
      "training_step_time": 7.400902509689331
    },
    {
      "epoch": 2.5435447692871093e-05,
      "model_forward_time": 1.7157905101776123,
      "step": 26671
    },
    {
      "epoch": 2.5435447692871093e-05,
      "step": 26671,
      "training_step_time": 7.360893487930298
    },
    {
      "epoch": 2.54364013671875e-05,
      "model_forward_time": 1.7235219478607178,
      "step": 26672
    },
    {
      "epoch": 2.54364013671875e-05,
      "step": 26672,
      "training_step_time": 8.473938703536987
    },
    {
      "epoch": 2.5437355041503905e-05,
      "model_forward_time": 2.7902660369873047,
      "step": 26673
    },
    {
      "epoch": 2.5437355041503905e-05,
      "step": 26673,
      "training_step_time": 7.616747617721558
    },
    {
      "epoch": 2.5438308715820312e-05,
      "model_forward_time": 1.7125794887542725,
      "step": 26674
    },
    {
      "epoch": 2.5438308715820312e-05,
      "step": 26674,
      "training_step_time": 7.709600448608398
    },
    {
      "epoch": 2.543926239013672e-05,
      "model_forward_time": 1.7052834033966064,
      "step": 26675
    },
    {
      "epoch": 2.543926239013672e-05,
      "step": 26675,
      "training_step_time": 8.543157577514648
    },
    {
      "epoch": 2.5440216064453123e-05,
      "model_forward_time": 1.7113635540008545,
      "step": 26676
    },
    {
      "epoch": 2.5440216064453123e-05,
      "step": 26676,
      "training_step_time": 7.755549192428589
    },
    {
      "epoch": 2.544116973876953e-05,
      "model_forward_time": 1.8987019062042236,
      "step": 26677
    },
    {
      "epoch": 2.544116973876953e-05,
      "step": 26677,
      "training_step_time": 7.7953362464904785
    },
    {
      "epoch": 2.5442123413085938e-05,
      "model_forward_time": 1.8716511726379395,
      "step": 26678
    },
    {
      "epoch": 2.5442123413085938e-05,
      "step": 26678,
      "training_step_time": 7.107048988342285
    },
    {
      "epoch": 2.5443077087402342e-05,
      "model_forward_time": 1.8947129249572754,
      "step": 26679
    },
    {
      "epoch": 2.5443077087402342e-05,
      "step": 26679,
      "training_step_time": 7.69694185256958
    },
    {
      "epoch": 2.544403076171875e-05,
      "grad_norm": 0.16324767470359802,
      "learning_rate": 8.769193434413266e-06,
      "loss": 0.0676,
      "step": 26680
    },
    {
      "epoch": 2.544403076171875e-05,
      "model_forward_time": 1.7847647666931152,
      "step": 26680
    },
    {
      "dynamics_loss_avg": 0.055565685592591764,
      "epoch": 2.544403076171875e-05,
      "step": 26680
    },
    {
      "action_loss_avg": 0.0013159395195543766,
      "epoch": 2.544403076171875e-05,
      "step": 26680
    },
    {
      "epoch": 2.544403076171875e-05,
      "step": 26680,
      "training_step_time": 7.360640287399292
    },
    {
      "epoch": 2.5444984436035157e-05,
      "model_forward_time": 1.723031759262085,
      "step": 26681
    },
    {
      "epoch": 2.5444984436035157e-05,
      "step": 26681,
      "training_step_time": 8.789731979370117
    },
    {
      "epoch": 2.544593811035156e-05,
      "model_forward_time": 1.7347829341888428,
      "step": 26682
    },
    {
      "epoch": 2.544593811035156e-05,
      "step": 26682,
      "training_step_time": 7.594831466674805
    },
    {
      "epoch": 2.544689178466797e-05,
      "model_forward_time": 2.878520965576172,
      "step": 26683
    },
    {
      "epoch": 2.544689178466797e-05,
      "step": 26683,
      "training_step_time": 8.309863328933716
    },
    {
      "epoch": 2.5447845458984376e-05,
      "model_forward_time": 1.7427937984466553,
      "step": 26684
    },
    {
      "epoch": 2.5447845458984376e-05,
      "step": 26684,
      "training_step_time": 8.615203380584717
    },
    {
      "epoch": 2.544879913330078e-05,
      "model_forward_time": 1.7243502140045166,
      "step": 26685
    },
    {
      "epoch": 2.544879913330078e-05,
      "step": 26685,
      "training_step_time": 7.79481315612793
    },
    {
      "epoch": 2.5449752807617187e-05,
      "model_forward_time": 1.7229552268981934,
      "step": 26686
    },
    {
      "epoch": 2.5449752807617187e-05,
      "step": 26686,
      "training_step_time": 7.778970718383789
    },
    {
      "epoch": 2.5450706481933595e-05,
      "model_forward_time": 1.7349677085876465,
      "step": 26687
    },
    {
      "epoch": 2.5450706481933595e-05,
      "step": 26687,
      "training_step_time": 7.126665115356445
    },
    {
      "epoch": 2.545166015625e-05,
      "model_forward_time": 1.723900556564331,
      "step": 26688
    },
    {
      "epoch": 2.545166015625e-05,
      "step": 26688,
      "training_step_time": 6.8761773109436035
    },
    {
      "epoch": 2.5452613830566406e-05,
      "model_forward_time": 1.7228407859802246,
      "step": 26689
    },
    {
      "epoch": 2.5452613830566406e-05,
      "step": 26689,
      "training_step_time": 7.400278568267822
    },
    {
      "epoch": 2.5453567504882814e-05,
      "grad_norm": 0.1868395060300827,
      "learning_rate": 8.76810680022482e-06,
      "loss": 0.0625,
      "step": 26690
    },
    {
      "epoch": 2.5453567504882814e-05,
      "model_forward_time": 1.7036669254302979,
      "step": 26690
    },
    {
      "dynamics_loss_avg": 0.058256948925554754,
      "epoch": 2.5453567504882814e-05,
      "step": 26690
    },
    {
      "action_loss_avg": 0.0021920446335570885,
      "epoch": 2.5453567504882814e-05,
      "step": 26690
    },
    {
      "epoch": 2.5453567504882814e-05,
      "step": 26690,
      "training_step_time": 7.646904230117798
    },
    {
      "epoch": 2.5454521179199218e-05,
      "model_forward_time": 1.746253490447998,
      "step": 26691
    },
    {
      "epoch": 2.5454521179199218e-05,
      "step": 26691,
      "training_step_time": 7.635898113250732
    },
    {
      "epoch": 2.5455474853515625e-05,
      "model_forward_time": 2.876098394393921,
      "step": 26692
    },
    {
      "epoch": 2.5455474853515625e-05,
      "step": 26692,
      "training_step_time": 8.55758810043335
    },
    {
      "epoch": 2.5456428527832033e-05,
      "model_forward_time": 1.725599765777588,
      "step": 26693
    },
    {
      "epoch": 2.5456428527832033e-05,
      "step": 26693,
      "training_step_time": 8.309959411621094
    },
    {
      "epoch": 2.5457382202148437e-05,
      "model_forward_time": 1.7241978645324707,
      "step": 26694
    },
    {
      "epoch": 2.5457382202148437e-05,
      "step": 26694,
      "training_step_time": 7.849798917770386
    },
    {
      "epoch": 2.5458335876464844e-05,
      "model_forward_time": 1.7255213260650635,
      "step": 26695
    },
    {
      "epoch": 2.5458335876464844e-05,
      "step": 26695,
      "training_step_time": 7.715807914733887
    },
    {
      "epoch": 2.545928955078125e-05,
      "model_forward_time": 1.7258539199829102,
      "step": 26696
    },
    {
      "epoch": 2.545928955078125e-05,
      "step": 26696,
      "training_step_time": 7.353429555892944
    },
    {
      "epoch": 2.5460243225097655e-05,
      "model_forward_time": 1.726238489151001,
      "step": 26697
    },
    {
      "epoch": 2.5460243225097655e-05,
      "step": 26697,
      "training_step_time": 7.072556972503662
    },
    {
      "epoch": 2.5461196899414063e-05,
      "model_forward_time": 1.7199864387512207,
      "step": 26698
    },
    {
      "epoch": 2.5461196899414063e-05,
      "step": 26698,
      "training_step_time": 7.3881776332855225
    },
    {
      "epoch": 2.546215057373047e-05,
      "model_forward_time": 1.72444486618042,
      "step": 26699
    },
    {
      "epoch": 2.546215057373047e-05,
      "step": 26699,
      "training_step_time": 7.63213586807251
    },
    {
      "epoch": 2.5463104248046874e-05,
      "grad_norm": 0.20624420046806335,
      "learning_rate": 8.767019753961878e-06,
      "loss": 0.0623,
      "step": 26700
    },
    {
      "epoch": 2.5463104248046874e-05,
      "model_forward_time": 1.7080614566802979,
      "step": 26700
    },
    {
      "dynamics_loss_avg": 0.053922314569354056,
      "epoch": 2.5463104248046874e-05,
      "step": 26700
    },
    {
      "action_loss_avg": 0.002318661811295897,
      "epoch": 2.5463104248046874e-05,
      "step": 26700
    },
    {
      "epoch": 2.5463104248046874e-05,
      "step": 26700,
      "training_step_time": 7.720235824584961
    },
    {
      "epoch": 2.5464057922363282e-05,
      "model_forward_time": 2.8777081966400146,
      "step": 26701
    },
    {
      "epoch": 2.5464057922363282e-05,
      "step": 26701,
      "training_step_time": 7.719439744949341
    },
    {
      "epoch": 2.546501159667969e-05,
      "model_forward_time": 1.341160535812378,
      "step": 26702
    },
    {
      "epoch": 2.546501159667969e-05,
      "step": 26702,
      "training_step_time": 8.712517976760864
    },
    {
      "epoch": 2.5465965270996093e-05,
      "model_forward_time": 1.7273495197296143,
      "step": 26703
    },
    {
      "epoch": 2.5465965270996093e-05,
      "step": 26703,
      "training_step_time": 8.157162427902222
    },
    {
      "epoch": 2.54669189453125e-05,
      "model_forward_time": 1.7229406833648682,
      "step": 26704
    },
    {
      "epoch": 2.54669189453125e-05,
      "step": 26704,
      "training_step_time": 7.686223745346069
    },
    {
      "epoch": 2.5467872619628905e-05,
      "model_forward_time": 1.7223076820373535,
      "step": 26705
    },
    {
      "epoch": 2.5467872619628905e-05,
      "step": 26705,
      "training_step_time": 7.445093631744385
    },
    {
      "epoch": 2.5468826293945312e-05,
      "model_forward_time": 1.7268764972686768,
      "step": 26706
    },
    {
      "epoch": 2.5468826293945312e-05,
      "step": 26706,
      "training_step_time": 7.092002630233765
    },
    {
      "epoch": 2.546977996826172e-05,
      "model_forward_time": 1.7239060401916504,
      "step": 26707
    },
    {
      "epoch": 2.546977996826172e-05,
      "step": 26707,
      "training_step_time": 7.422830820083618
    },
    {
      "epoch": 2.5470733642578123e-05,
      "model_forward_time": 1.77290678024292,
      "step": 26708
    },
    {
      "epoch": 2.5470733642578123e-05,
      "step": 26708,
      "training_step_time": 8.617098331451416
    },
    {
      "epoch": 2.547168731689453e-05,
      "model_forward_time": 1.7306807041168213,
      "step": 26709
    },
    {
      "epoch": 2.547168731689453e-05,
      "step": 26709,
      "training_step_time": 7.738178730010986
    },
    {
      "epoch": 2.547264099121094e-05,
      "grad_norm": 0.1604059934616089,
      "learning_rate": 8.765932295743322e-06,
      "loss": 0.0606,
      "step": 26710
    },
    {
      "epoch": 2.547264099121094e-05,
      "model_forward_time": 2.8148722648620605,
      "step": 26710
    },
    {
      "dynamics_loss_avg": 0.052918584644794465,
      "epoch": 2.547264099121094e-05,
      "step": 26710
    },
    {
      "action_loss_avg": 0.003536240046378225,
      "epoch": 2.547264099121094e-05,
      "step": 26710
    },
    {
      "epoch": 2.547264099121094e-05,
      "step": 26710,
      "training_step_time": 8.577408790588379
    },
    {
      "epoch": 2.5473594665527342e-05,
      "model_forward_time": 1.834841012954712,
      "step": 26711
    },
    {
      "epoch": 2.5473594665527342e-05,
      "step": 26711,
      "training_step_time": 8.409992933273315
    },
    {
      "epoch": 2.547454833984375e-05,
      "model_forward_time": 1.7241666316986084,
      "step": 26712
    },
    {
      "epoch": 2.547454833984375e-05,
      "step": 26712,
      "training_step_time": 7.806396484375
    },
    {
      "epoch": 2.5475502014160157e-05,
      "model_forward_time": 1.727365493774414,
      "step": 26713
    },
    {
      "epoch": 2.5475502014160157e-05,
      "step": 26713,
      "training_step_time": 8.474086284637451
    },
    {
      "epoch": 2.547645568847656e-05,
      "model_forward_time": 1.7248892784118652,
      "step": 26714
    },
    {
      "epoch": 2.547645568847656e-05,
      "step": 26714,
      "training_step_time": 7.56619930267334
    },
    {
      "epoch": 2.547740936279297e-05,
      "model_forward_time": 1.7159297466278076,
      "step": 26715
    },
    {
      "epoch": 2.547740936279297e-05,
      "step": 26715,
      "training_step_time": 7.363903522491455
    },
    {
      "epoch": 2.5478363037109376e-05,
      "model_forward_time": 1.7190241813659668,
      "step": 26716
    },
    {
      "epoch": 2.5478363037109376e-05,
      "step": 26716,
      "training_step_time": 7.348975658416748
    },
    {
      "epoch": 2.547931671142578e-05,
      "model_forward_time": 1.7244658470153809,
      "step": 26717
    },
    {
      "epoch": 2.547931671142578e-05,
      "step": 26717,
      "training_step_time": 7.694680213928223
    },
    {
      "epoch": 2.5480270385742187e-05,
      "model_forward_time": 1.7165329456329346,
      "step": 26718
    },
    {
      "epoch": 2.5480270385742187e-05,
      "step": 26718,
      "training_step_time": 7.684697151184082
    },
    {
      "epoch": 2.5481224060058595e-05,
      "model_forward_time": 1.7137706279754639,
      "step": 26719
    },
    {
      "epoch": 2.5481224060058595e-05,
      "step": 26719,
      "training_step_time": 8.670904636383057
    },
    {
      "epoch": 2.5482177734375e-05,
      "grad_norm": 0.21452836692333221,
      "learning_rate": 8.76484442568807e-06,
      "loss": 0.0632,
      "step": 26720
    },
    {
      "epoch": 2.5482177734375e-05,
      "model_forward_time": 2.914597988128662,
      "step": 26720
    },
    {
      "dynamics_loss_avg": 0.05814490802586079,
      "epoch": 2.5482177734375e-05,
      "step": 26720
    },
    {
      "action_loss_avg": 0.002659921027952805,
      "epoch": 2.5482177734375e-05,
      "step": 26720
    },
    {
      "epoch": 2.5482177734375e-05,
      "step": 26720,
      "training_step_time": 9.520087718963623
    },
    {
      "epoch": 2.5483131408691406e-05,
      "model_forward_time": 1.5199592113494873,
      "step": 26721
    },
    {
      "epoch": 2.5483131408691406e-05,
      "step": 26721,
      "training_step_time": 9.077976942062378
    },
    {
      "epoch": 2.5484085083007814e-05,
      "model_forward_time": 1.8752326965332031,
      "step": 26722
    },
    {
      "epoch": 2.5484085083007814e-05,
      "step": 26722,
      "training_step_time": 7.768991947174072
    },
    {
      "epoch": 2.5485038757324218e-05,
      "model_forward_time": 1.8949244022369385,
      "step": 26723
    },
    {
      "epoch": 2.5485038757324218e-05,
      "step": 26723,
      "training_step_time": 7.805029392242432
    },
    {
      "epoch": 2.5485992431640625e-05,
      "model_forward_time": 1.9032785892486572,
      "step": 26724
    },
    {
      "epoch": 2.5485992431640625e-05,
      "step": 26724,
      "training_step_time": 7.450108051300049
    },
    {
      "epoch": 2.5486946105957033e-05,
      "model_forward_time": 1.7477798461914062,
      "step": 26725
    },
    {
      "epoch": 2.5486946105957033e-05,
      "step": 26725,
      "training_step_time": 7.43857216835022
    },
    {
      "epoch": 2.5487899780273437e-05,
      "model_forward_time": 1.7217822074890137,
      "step": 26726
    },
    {
      "epoch": 2.5487899780273437e-05,
      "step": 26726,
      "training_step_time": 7.636734247207642
    },
    {
      "epoch": 2.5488853454589844e-05,
      "model_forward_time": 1.726832389831543,
      "step": 26727
    },
    {
      "epoch": 2.5488853454589844e-05,
      "step": 26727,
      "training_step_time": 7.673879623413086
    },
    {
      "epoch": 2.548980712890625e-05,
      "model_forward_time": 1.7200608253479004,
      "step": 26728
    },
    {
      "epoch": 2.548980712890625e-05,
      "step": 26728,
      "training_step_time": 7.751910924911499
    },
    {
      "epoch": 2.5490760803222655e-05,
      "model_forward_time": 2.9256644248962402,
      "step": 26729
    },
    {
      "epoch": 2.5490760803222655e-05,
      "step": 26729,
      "training_step_time": 8.703336000442505
    },
    {
      "epoch": 2.5491714477539063e-05,
      "grad_norm": 0.14605097472667694,
      "learning_rate": 8.763756143915093e-06,
      "loss": 0.0628,
      "step": 26730
    },
    {
      "epoch": 2.5491714477539063e-05,
      "model_forward_time": 1.7371973991394043,
      "step": 26730
    },
    {
      "dynamics_loss_avg": 0.0480956737883389,
      "epoch": 2.5491714477539063e-05,
      "step": 26730
    },
    {
      "action_loss_avg": 0.0018270444154040888,
      "epoch": 2.5491714477539063e-05,
      "step": 26730
    },
    {
      "epoch": 2.5491714477539063e-05,
      "step": 26730,
      "training_step_time": 8.678203821182251
    },
    {
      "epoch": 2.549266815185547e-05,
      "model_forward_time": 1.7224671840667725,
      "step": 26731
    },
    {
      "epoch": 2.549266815185547e-05,
      "step": 26731,
      "training_step_time": 7.774407863616943
    },
    {
      "epoch": 2.5493621826171874e-05,
      "model_forward_time": 1.7129144668579102,
      "step": 26732
    },
    {
      "epoch": 2.5493621826171874e-05,
      "step": 26732,
      "training_step_time": 7.70633339881897
    },
    {
      "epoch": 2.5494575500488282e-05,
      "model_forward_time": 1.7181878089904785,
      "step": 26733
    },
    {
      "epoch": 2.5494575500488282e-05,
      "step": 26733,
      "training_step_time": 7.385066509246826
    },
    {
      "epoch": 2.549552917480469e-05,
      "model_forward_time": 1.014073371887207,
      "step": 26734
    },
    {
      "epoch": 2.549552917480469e-05,
      "step": 26734,
      "training_step_time": 7.339619874954224
    },
    {
      "epoch": 2.5496482849121093e-05,
      "model_forward_time": 1.3589272499084473,
      "step": 26735
    },
    {
      "epoch": 2.5496482849121093e-05,
      "step": 26735,
      "training_step_time": 7.607857704162598
    },
    {
      "epoch": 2.54974365234375e-05,
      "model_forward_time": 1.7270135879516602,
      "step": 26736
    },
    {
      "epoch": 2.54974365234375e-05,
      "step": 26736,
      "training_step_time": 7.65642237663269
    },
    {
      "epoch": 2.5498390197753905e-05,
      "model_forward_time": 1.719520092010498,
      "step": 26737
    },
    {
      "epoch": 2.5498390197753905e-05,
      "step": 26737,
      "training_step_time": 7.728050470352173
    },
    {
      "epoch": 2.5499343872070312e-05,
      "model_forward_time": 2.8205275535583496,
      "step": 26738
    },
    {
      "epoch": 2.5499343872070312e-05,
      "step": 26738,
      "training_step_time": 9.448627710342407
    },
    {
      "epoch": 2.550029754638672e-05,
      "model_forward_time": 1.7308166027069092,
      "step": 26739
    },
    {
      "epoch": 2.550029754638672e-05,
      "step": 26739,
      "training_step_time": 9.24781584739685
    },
    {
      "epoch": 2.5501251220703124e-05,
      "grad_norm": 0.137893408536911,
      "learning_rate": 8.7626674505434e-06,
      "loss": 0.062,
      "step": 26740
    },
    {
      "epoch": 2.5501251220703124e-05,
      "model_forward_time": 1.7406346797943115,
      "step": 26740
    },
    {
      "dynamics_loss_avg": 0.05905160494148731,
      "epoch": 2.5501251220703124e-05,
      "step": 26740
    },
    {
      "action_loss_avg": 0.0025024738046340645,
      "epoch": 2.5501251220703124e-05,
      "step": 26740
    },
    {
      "epoch": 2.5501251220703124e-05,
      "step": 26740,
      "training_step_time": 8.251270771026611
    },
    {
      "epoch": 2.550220489501953e-05,
      "model_forward_time": 1.7344605922698975,
      "step": 26741
    },
    {
      "epoch": 2.550220489501953e-05,
      "step": 26741,
      "training_step_time": 7.814343690872192
    },
    {
      "epoch": 2.550315856933594e-05,
      "model_forward_time": 1.7381582260131836,
      "step": 26742
    },
    {
      "epoch": 2.550315856933594e-05,
      "step": 26742,
      "training_step_time": 7.541435718536377
    },
    {
      "epoch": 2.5504112243652342e-05,
      "model_forward_time": 1.7315340042114258,
      "step": 26743
    },
    {
      "epoch": 2.5504112243652342e-05,
      "step": 26743,
      "training_step_time": 7.138928413391113
    },
    {
      "epoch": 2.550506591796875e-05,
      "model_forward_time": 1.732828140258789,
      "step": 26744
    },
    {
      "epoch": 2.550506591796875e-05,
      "step": 26744,
      "training_step_time": 7.660937786102295
    },
    {
      "epoch": 2.5506019592285157e-05,
      "model_forward_time": 1.730943202972412,
      "step": 26745
    },
    {
      "epoch": 2.5506019592285157e-05,
      "step": 26745,
      "training_step_time": 7.645848751068115
    },
    {
      "epoch": 2.550697326660156e-05,
      "model_forward_time": 1.7294225692749023,
      "step": 26746
    },
    {
      "epoch": 2.550697326660156e-05,
      "step": 26746,
      "training_step_time": 9.074566841125488
    },
    {
      "epoch": 2.550792694091797e-05,
      "model_forward_time": 1.7278761863708496,
      "step": 26747
    },
    {
      "epoch": 2.550792694091797e-05,
      "step": 26747,
      "training_step_time": 7.655100345611572
    },
    {
      "epoch": 2.5508880615234376e-05,
      "model_forward_time": 2.896617889404297,
      "step": 26748
    },
    {
      "epoch": 2.5508880615234376e-05,
      "step": 26748,
      "training_step_time": 8.759000778198242
    },
    {
      "epoch": 2.550983428955078e-05,
      "model_forward_time": 1.7301688194274902,
      "step": 26749
    },
    {
      "epoch": 2.550983428955078e-05,
      "step": 26749,
      "training_step_time": 8.217977523803711
    },
    {
      "epoch": 2.5510787963867188e-05,
      "grad_norm": 0.20868416130542755,
      "learning_rate": 8.761578345692055e-06,
      "loss": 0.0589,
      "step": 26750
    },
    {
      "epoch": 2.5510787963867188e-05,
      "model_forward_time": 1.7441179752349854,
      "step": 26750
    },
    {
      "dynamics_loss_avg": 0.056486133486032486,
      "epoch": 2.5510787963867188e-05,
      "step": 26750
    },
    {
      "action_loss_avg": 0.0033191770868143066,
      "epoch": 2.5510787963867188e-05,
      "step": 26750
    },
    {
      "epoch": 2.5510787963867188e-05,
      "step": 26750,
      "training_step_time": 9.26937985420227
    },
    {
      "epoch": 2.5511741638183595e-05,
      "model_forward_time": 1.7355046272277832,
      "step": 26751
    },
    {
      "epoch": 2.5511741638183595e-05,
      "step": 26751,
      "training_step_time": 7.505693435668945
    },
    {
      "epoch": 2.55126953125e-05,
      "model_forward_time": 1.7363219261169434,
      "step": 26752
    },
    {
      "epoch": 2.55126953125e-05,
      "step": 26752,
      "training_step_time": 7.328484535217285
    },
    {
      "epoch": 2.5513648986816406e-05,
      "model_forward_time": 1.7396619319915771,
      "step": 26753
    },
    {
      "epoch": 2.5513648986816406e-05,
      "step": 26753,
      "training_step_time": 7.590219020843506
    },
    {
      "epoch": 2.5514602661132814e-05,
      "model_forward_time": 1.7356059551239014,
      "step": 26754
    },
    {
      "epoch": 2.5514602661132814e-05,
      "step": 26754,
      "training_step_time": 7.542049169540405
    },
    {
      "epoch": 2.5515556335449218e-05,
      "model_forward_time": 1.7334885597229004,
      "step": 26755
    },
    {
      "epoch": 2.5515556335449218e-05,
      "step": 26755,
      "training_step_time": 7.67439603805542
    },
    {
      "epoch": 2.5516510009765625e-05,
      "model_forward_time": 1.7296695709228516,
      "step": 26756
    },
    {
      "epoch": 2.5516510009765625e-05,
      "step": 26756,
      "training_step_time": 8.259911060333252
    },
    {
      "epoch": 2.5517463684082033e-05,
      "model_forward_time": 2.8973681926727295,
      "step": 26757
    },
    {
      "epoch": 2.5517463684082033e-05,
      "step": 26757,
      "training_step_time": 9.624800443649292
    },
    {
      "epoch": 2.5518417358398437e-05,
      "model_forward_time": 1.7309722900390625,
      "step": 26758
    },
    {
      "epoch": 2.5518417358398437e-05,
      "step": 26758,
      "training_step_time": 8.579842805862427
    },
    {
      "epoch": 2.5519371032714844e-05,
      "model_forward_time": 1.731879472732544,
      "step": 26759
    },
    {
      "epoch": 2.5519371032714844e-05,
      "step": 26759,
      "training_step_time": 7.742440462112427
    },
    {
      "epoch": 2.552032470703125e-05,
      "grad_norm": 0.13902607560157776,
      "learning_rate": 8.760488829480155e-06,
      "loss": 0.0611,
      "step": 26760
    },
    {
      "epoch": 2.552032470703125e-05,
      "model_forward_time": 1.732743501663208,
      "step": 26760
    },
    {
      "dynamics_loss_avg": 0.07608421966433525,
      "epoch": 2.552032470703125e-05,
      "step": 26760
    },
    {
      "action_loss_avg": 0.003646028373623267,
      "epoch": 2.552032470703125e-05,
      "step": 26760
    },
    {
      "epoch": 2.552032470703125e-05,
      "step": 26760,
      "training_step_time": 8.31272578239441
    },
    {
      "epoch": 2.5521278381347656e-05,
      "model_forward_time": 1.7439777851104736,
      "step": 26761
    },
    {
      "epoch": 2.5521278381347656e-05,
      "step": 26761,
      "training_step_time": 8.4119291305542
    },
    {
      "epoch": 2.5522232055664063e-05,
      "model_forward_time": 1.3655998706817627,
      "step": 26762
    },
    {
      "epoch": 2.5522232055664063e-05,
      "step": 26762,
      "training_step_time": 7.641716480255127
    },
    {
      "epoch": 2.552318572998047e-05,
      "model_forward_time": 1.7335050106048584,
      "step": 26763
    },
    {
      "epoch": 2.552318572998047e-05,
      "step": 26763,
      "training_step_time": 7.502293586730957
    },
    {
      "epoch": 2.5524139404296874e-05,
      "model_forward_time": 1.7414379119873047,
      "step": 26764
    },
    {
      "epoch": 2.5524139404296874e-05,
      "step": 26764,
      "training_step_time": 7.6554999351501465
    },
    {
      "epoch": 2.5525093078613282e-05,
      "model_forward_time": 1.7340946197509766,
      "step": 26765
    },
    {
      "epoch": 2.5525093078613282e-05,
      "step": 26765,
      "training_step_time": 8.489473104476929
    },
    {
      "epoch": 2.5526046752929686e-05,
      "model_forward_time": 2.9053356647491455,
      "step": 26766
    },
    {
      "epoch": 2.5526046752929686e-05,
      "step": 26766,
      "training_step_time": 7.746046543121338
    },
    {
      "epoch": 2.5527000427246093e-05,
      "model_forward_time": 1.7319629192352295,
      "step": 26767
    },
    {
      "epoch": 2.5527000427246093e-05,
      "step": 26767,
      "training_step_time": 8.621136903762817
    },
    {
      "epoch": 2.55279541015625e-05,
      "model_forward_time": 1.7286460399627686,
      "step": 26768
    },
    {
      "epoch": 2.55279541015625e-05,
      "step": 26768,
      "training_step_time": 7.696776390075684
    },
    {
      "epoch": 2.5528907775878905e-05,
      "model_forward_time": 1.7307968139648438,
      "step": 26769
    },
    {
      "epoch": 2.5528907775878905e-05,
      "step": 26769,
      "training_step_time": 7.756287336349487
    },
    {
      "epoch": 2.5529861450195312e-05,
      "grad_norm": 0.1488082855939865,
      "learning_rate": 8.759398902026855e-06,
      "loss": 0.0603,
      "step": 26770
    },
    {
      "epoch": 2.5529861450195312e-05,
      "model_forward_time": 1.7434196472167969,
      "step": 26770
    },
    {
      "dynamics_loss_avg": 0.06077310405671597,
      "epoch": 2.5529861450195312e-05,
      "step": 26770
    },
    {
      "action_loss_avg": 0.0027085201814770698,
      "epoch": 2.5529861450195312e-05,
      "step": 26770
    },
    {
      "epoch": 2.5529861450195312e-05,
      "step": 26770,
      "training_step_time": 7.370948791503906
    },
    {
      "epoch": 2.553081512451172e-05,
      "model_forward_time": 1.73311185836792,
      "step": 26771
    },
    {
      "epoch": 2.553081512451172e-05,
      "step": 26771,
      "training_step_time": 8.11675477027893
    },
    {
      "epoch": 2.5531768798828124e-05,
      "model_forward_time": 1.7424688339233398,
      "step": 26772
    },
    {
      "epoch": 2.5531768798828124e-05,
      "step": 26772,
      "training_step_time": 7.0718536376953125
    },
    {
      "epoch": 2.553272247314453e-05,
      "model_forward_time": 1.735551118850708,
      "step": 26773
    },
    {
      "epoch": 2.553272247314453e-05,
      "step": 26773,
      "training_step_time": 8.451193571090698
    },
    {
      "epoch": 2.553367614746094e-05,
      "model_forward_time": 1.7352676391601562,
      "step": 26774
    },
    {
      "epoch": 2.553367614746094e-05,
      "step": 26774,
      "training_step_time": 8.64172887802124
    },
    {
      "epoch": 2.5534629821777342e-05,
      "model_forward_time": 2.917978286743164,
      "step": 26775
    },
    {
      "epoch": 2.5534629821777342e-05,
      "step": 26775,
      "training_step_time": 7.67661190032959
    },
    {
      "epoch": 2.553558349609375e-05,
      "model_forward_time": 1.7308855056762695,
      "step": 26776
    },
    {
      "epoch": 2.553558349609375e-05,
      "step": 26776,
      "training_step_time": 7.5658063888549805
    },
    {
      "epoch": 2.5536537170410157e-05,
      "model_forward_time": 1.3712730407714844,
      "step": 26777
    },
    {
      "epoch": 2.5536537170410157e-05,
      "step": 26777,
      "training_step_time": 8.251173496246338
    },
    {
      "epoch": 2.553749084472656e-05,
      "model_forward_time": 1.7320024967193604,
      "step": 26778
    },
    {
      "epoch": 2.553749084472656e-05,
      "step": 26778,
      "training_step_time": 7.685980796813965
    },
    {
      "epoch": 2.553844451904297e-05,
      "model_forward_time": 1.7642385959625244,
      "step": 26779
    },
    {
      "epoch": 2.553844451904297e-05,
      "step": 26779,
      "training_step_time": 8.100108861923218
    },
    {
      "epoch": 2.5539398193359376e-05,
      "grad_norm": 0.11092756688594818,
      "learning_rate": 8.758308563451338e-06,
      "loss": 0.0603,
      "step": 26780
    },
    {
      "epoch": 2.5539398193359376e-05,
      "model_forward_time": 1.3622212409973145,
      "step": 26780
    },
    {
      "dynamics_loss_avg": 0.06404986307024955,
      "epoch": 2.5539398193359376e-05,
      "step": 26780
    },
    {
      "action_loss_avg": 0.002414851900539361,
      "epoch": 2.5539398193359376e-05,
      "step": 26780
    },
    {
      "epoch": 2.5539398193359376e-05,
      "step": 26780,
      "training_step_time": 8.300827503204346
    },
    {
      "epoch": 2.554035186767578e-05,
      "model_forward_time": 1.729172706604004,
      "step": 26781
    },
    {
      "epoch": 2.554035186767578e-05,
      "step": 26781,
      "training_step_time": 7.32097053527832
    },
    {
      "epoch": 2.5541305541992188e-05,
      "model_forward_time": 1.7206354141235352,
      "step": 26782
    },
    {
      "epoch": 2.5541305541992188e-05,
      "step": 26782,
      "training_step_time": 8.649678945541382
    },
    {
      "epoch": 2.5542259216308595e-05,
      "model_forward_time": 1.7251741886138916,
      "step": 26783
    },
    {
      "epoch": 2.5542259216308595e-05,
      "step": 26783,
      "training_step_time": 7.487261056900024
    },
    {
      "epoch": 2.5543212890625e-05,
      "model_forward_time": 1.7283375263214111,
      "step": 26784
    },
    {
      "epoch": 2.5543212890625e-05,
      "step": 26784,
      "training_step_time": 8.108721494674683
    },
    {
      "epoch": 2.5544166564941406e-05,
      "model_forward_time": 1.8510842323303223,
      "step": 26785
    },
    {
      "epoch": 2.5544166564941406e-05,
      "step": 26785,
      "training_step_time": 7.632005214691162
    },
    {
      "epoch": 2.5545120239257814e-05,
      "model_forward_time": 1.727391242980957,
      "step": 26786
    },
    {
      "epoch": 2.5545120239257814e-05,
      "step": 26786,
      "training_step_time": 8.155485391616821
    },
    {
      "epoch": 2.5546073913574218e-05,
      "model_forward_time": 1.7243692874908447,
      "step": 26787
    },
    {
      "epoch": 2.5546073913574218e-05,
      "step": 26787,
      "training_step_time": 7.637315034866333
    },
    {
      "epoch": 2.5547027587890625e-05,
      "model_forward_time": 1.708509922027588,
      "step": 26788
    },
    {
      "epoch": 2.5547027587890625e-05,
      "step": 26788,
      "training_step_time": 8.295132398605347
    },
    {
      "epoch": 2.5547981262207033e-05,
      "model_forward_time": 1.7288734912872314,
      "step": 26789
    },
    {
      "epoch": 2.5547981262207033e-05,
      "step": 26789,
      "training_step_time": 7.448429584503174
    },
    {
      "epoch": 2.5548934936523437e-05,
      "grad_norm": 0.1837417483329773,
      "learning_rate": 8.757217813872849e-06,
      "loss": 0.0591,
      "step": 26790
    },
    {
      "epoch": 2.5548934936523437e-05,
      "model_forward_time": 1.0103394985198975,
      "step": 26790
    },
    {
      "dynamics_loss_avg": 0.05229575727134943,
      "epoch": 2.5548934936523437e-05,
      "step": 26790
    },
    {
      "action_loss_avg": 0.0022183302498888224,
      "epoch": 2.5548934936523437e-05,
      "step": 26790
    },
    {
      "epoch": 2.5548934936523437e-05,
      "step": 26790,
      "training_step_time": 7.465535879135132
    },
    {
      "epoch": 2.5549888610839844e-05,
      "model_forward_time": 1.7279913425445557,
      "step": 26791
    },
    {
      "epoch": 2.5549888610839844e-05,
      "step": 26791,
      "training_step_time": 7.626181125640869
    },
    {
      "epoch": 2.555084228515625e-05,
      "model_forward_time": 1.7245466709136963,
      "step": 26792
    },
    {
      "epoch": 2.555084228515625e-05,
      "step": 26792,
      "training_step_time": 7.690049409866333
    },
    {
      "epoch": 2.5551795959472656e-05,
      "model_forward_time": 1.7284796237945557,
      "step": 26793
    },
    {
      "epoch": 2.5551795959472656e-05,
      "step": 26793,
      "training_step_time": 7.680562496185303
    },
    {
      "epoch": 2.5552749633789063e-05,
      "model_forward_time": 2.8570690155029297,
      "step": 26794
    },
    {
      "epoch": 2.5552749633789063e-05,
      "step": 26794,
      "training_step_time": 9.296243906021118
    },
    {
      "epoch": 2.555370330810547e-05,
      "model_forward_time": 1.7246227264404297,
      "step": 26795
    },
    {
      "epoch": 2.555370330810547e-05,
      "step": 26795,
      "training_step_time": 8.658222913742065
    },
    {
      "epoch": 2.5554656982421874e-05,
      "model_forward_time": 1.7256534099578857,
      "step": 26796
    },
    {
      "epoch": 2.5554656982421874e-05,
      "step": 26796,
      "training_step_time": 7.771371364593506
    },
    {
      "epoch": 2.5555610656738282e-05,
      "model_forward_time": 1.7212846279144287,
      "step": 26797
    },
    {
      "epoch": 2.5555610656738282e-05,
      "step": 26797,
      "training_step_time": 7.629828453063965
    },
    {
      "epoch": 2.5556564331054686e-05,
      "model_forward_time": 1.7246854305267334,
      "step": 26798
    },
    {
      "epoch": 2.5556564331054686e-05,
      "step": 26798,
      "training_step_time": 7.590969800949097
    },
    {
      "epoch": 2.5557518005371093e-05,
      "model_forward_time": 1.7252576351165771,
      "step": 26799
    },
    {
      "epoch": 2.5557518005371093e-05,
      "step": 26799,
      "training_step_time": 7.696996688842773
    },
    {
      "epoch": 2.55584716796875e-05,
      "grad_norm": 0.16931220889091492,
      "learning_rate": 8.75612665341067e-06,
      "loss": 0.058,
      "step": 26800
    },
    {
      "epoch": 2.55584716796875e-05,
      "model_forward_time": 1.7223386764526367,
      "step": 26800
    },
    {
      "dynamics_loss_avg": 0.05745923314243555,
      "epoch": 2.55584716796875e-05,
      "step": 26800
    },
    {
      "action_loss_avg": 0.0033773628587368877,
      "epoch": 2.55584716796875e-05,
      "step": 26800
    },
    {
      "epoch": 2.55584716796875e-05,
      "step": 26800,
      "training_step_time": 7.746028184890747
    },
    {
      "epoch": 2.5559425354003905e-05,
      "model_forward_time": 1.720982313156128,
      "step": 26801
    },
    {
      "epoch": 2.5559425354003905e-05,
      "step": 26801,
      "training_step_time": 7.678360223770142
    },
    {
      "epoch": 2.5560379028320312e-05,
      "model_forward_time": 1.7196619510650635,
      "step": 26802
    },
    {
      "epoch": 2.5560379028320312e-05,
      "step": 26802,
      "training_step_time": 8.575411319732666
    },
    {
      "epoch": 2.556133270263672e-05,
      "model_forward_time": 2.853631019592285,
      "step": 26803
    },
    {
      "epoch": 2.556133270263672e-05,
      "step": 26803,
      "training_step_time": 7.729165315628052
    },
    {
      "epoch": 2.5562286376953124e-05,
      "model_forward_time": 1.718775987625122,
      "step": 26804
    },
    {
      "epoch": 2.5562286376953124e-05,
      "step": 26804,
      "training_step_time": 7.807446479797363
    },
    {
      "epoch": 2.556324005126953e-05,
      "model_forward_time": 1.7152981758117676,
      "step": 26805
    },
    {
      "epoch": 2.556324005126953e-05,
      "step": 26805,
      "training_step_time": 7.72496485710144
    },
    {
      "epoch": 2.556419372558594e-05,
      "model_forward_time": 1.7202990055084229,
      "step": 26806
    },
    {
      "epoch": 2.556419372558594e-05,
      "step": 26806,
      "training_step_time": 7.770172357559204
    },
    {
      "epoch": 2.5565147399902342e-05,
      "model_forward_time": 1.0133106708526611,
      "step": 26807
    },
    {
      "epoch": 2.5565147399902342e-05,
      "step": 26807,
      "training_step_time": 8.26050877571106
    },
    {
      "epoch": 2.556610107421875e-05,
      "model_forward_time": 1.719651699066162,
      "step": 26808
    },
    {
      "epoch": 2.556610107421875e-05,
      "step": 26808,
      "training_step_time": 7.713842868804932
    },
    {
      "epoch": 2.5567054748535157e-05,
      "model_forward_time": 1.7218399047851562,
      "step": 26809
    },
    {
      "epoch": 2.5567054748535157e-05,
      "step": 26809,
      "training_step_time": 7.731755018234253
    },
    {
      "epoch": 2.556800842285156e-05,
      "grad_norm": 0.1827101856470108,
      "learning_rate": 8.755035082184126e-06,
      "loss": 0.0602,
      "step": 26810
    },
    {
      "epoch": 2.556800842285156e-05,
      "model_forward_time": 1.7236406803131104,
      "step": 26810
    },
    {
      "dynamics_loss_avg": 0.07158677838742733,
      "epoch": 2.556800842285156e-05,
      "step": 26810
    },
    {
      "action_loss_avg": 0.0030342555721290408,
      "epoch": 2.556800842285156e-05,
      "step": 26810
    },
    {
      "epoch": 2.556800842285156e-05,
      "step": 26810,
      "training_step_time": 7.742075204849243
    },
    {
      "epoch": 2.556896209716797e-05,
      "model_forward_time": 1.7255854606628418,
      "step": 26811
    },
    {
      "epoch": 2.556896209716797e-05,
      "step": 26811,
      "training_step_time": 7.723918199539185
    },
    {
      "epoch": 2.5569915771484376e-05,
      "model_forward_time": 1.7767722606658936,
      "step": 26812
    },
    {
      "epoch": 2.5569915771484376e-05,
      "step": 26812,
      "training_step_time": 10.09854793548584
    },
    {
      "epoch": 2.557086944580078e-05,
      "model_forward_time": 2.847383499145508,
      "step": 26813
    },
    {
      "epoch": 2.557086944580078e-05,
      "step": 26813,
      "training_step_time": 8.805442094802856
    },
    {
      "epoch": 2.5571823120117188e-05,
      "model_forward_time": 1.7227773666381836,
      "step": 26814
    },
    {
      "epoch": 2.5571823120117188e-05,
      "step": 26814,
      "training_step_time": 7.784214496612549
    },
    {
      "epoch": 2.5572776794433595e-05,
      "model_forward_time": 1.733309030532837,
      "step": 26815
    },
    {
      "epoch": 2.5572776794433595e-05,
      "step": 26815,
      "training_step_time": 7.767294883728027
    },
    {
      "epoch": 2.557373046875e-05,
      "model_forward_time": 1.7287075519561768,
      "step": 26816
    },
    {
      "epoch": 2.557373046875e-05,
      "step": 26816,
      "training_step_time": 8.473331928253174
    },
    {
      "epoch": 2.5574684143066406e-05,
      "model_forward_time": 1.7216293811798096,
      "step": 26817
    },
    {
      "epoch": 2.5574684143066406e-05,
      "step": 26817,
      "training_step_time": 7.652889728546143
    },
    {
      "epoch": 2.5575637817382814e-05,
      "model_forward_time": 1.724947452545166,
      "step": 26818
    },
    {
      "epoch": 2.5575637817382814e-05,
      "step": 26818,
      "training_step_time": 7.714738368988037
    },
    {
      "epoch": 2.5576591491699218e-05,
      "model_forward_time": 1.7327938079833984,
      "step": 26819
    },
    {
      "epoch": 2.5576591491699218e-05,
      "step": 26819,
      "training_step_time": 7.680890321731567
    },
    {
      "epoch": 2.5577545166015625e-05,
      "grad_norm": 0.16199353337287903,
      "learning_rate": 8.753943100312593e-06,
      "loss": 0.062,
      "step": 26820
    },
    {
      "epoch": 2.5577545166015625e-05,
      "model_forward_time": 1.7124099731445312,
      "step": 26820
    },
    {
      "dynamics_loss_avg": 0.0576271690428257,
      "epoch": 2.5577545166015625e-05,
      "step": 26820
    },
    {
      "action_loss_avg": 0.0026660633680876344,
      "epoch": 2.5577545166015625e-05,
      "step": 26820
    },
    {
      "epoch": 2.5577545166015625e-05,
      "step": 26820,
      "training_step_time": 7.704853057861328
    },
    {
      "epoch": 2.5578498840332033e-05,
      "model_forward_time": 1.746687889099121,
      "step": 26821
    },
    {
      "epoch": 2.5578498840332033e-05,
      "step": 26821,
      "training_step_time": 7.724803447723389
    },
    {
      "epoch": 2.5579452514648437e-05,
      "model_forward_time": 2.871694326400757,
      "step": 26822
    },
    {
      "epoch": 2.5579452514648437e-05,
      "step": 26822,
      "training_step_time": 8.673977375030518
    },
    {
      "epoch": 2.5580406188964844e-05,
      "model_forward_time": 1.7251396179199219,
      "step": 26823
    },
    {
      "epoch": 2.5580406188964844e-05,
      "step": 26823,
      "training_step_time": 7.777392387390137
    },
    {
      "epoch": 2.558135986328125e-05,
      "model_forward_time": 1.7241730690002441,
      "step": 26824
    },
    {
      "epoch": 2.558135986328125e-05,
      "step": 26824,
      "training_step_time": 7.678397178649902
    },
    {
      "epoch": 2.5582313537597656e-05,
      "model_forward_time": 1.7247343063354492,
      "step": 26825
    },
    {
      "epoch": 2.5582313537597656e-05,
      "step": 26825,
      "training_step_time": 7.786252498626709
    },
    {
      "epoch": 2.5583267211914063e-05,
      "model_forward_time": 1.723231554031372,
      "step": 26826
    },
    {
      "epoch": 2.5583267211914063e-05,
      "step": 26826,
      "training_step_time": 7.568756818771362
    },
    {
      "epoch": 2.558422088623047e-05,
      "model_forward_time": 1.7315752506256104,
      "step": 26827
    },
    {
      "epoch": 2.558422088623047e-05,
      "step": 26827,
      "training_step_time": 8.130428552627563
    },
    {
      "epoch": 2.5585174560546874e-05,
      "model_forward_time": 1.364485263824463,
      "step": 26828
    },
    {
      "epoch": 2.5585174560546874e-05,
      "step": 26828,
      "training_step_time": 7.638874292373657
    },
    {
      "epoch": 2.5586128234863282e-05,
      "model_forward_time": 1.7313506603240967,
      "step": 26829
    },
    {
      "epoch": 2.5586128234863282e-05,
      "step": 26829,
      "training_step_time": 7.651547908782959
    },
    {
      "epoch": 2.5587081909179686e-05,
      "grad_norm": 0.15586036443710327,
      "learning_rate": 8.752850707915484e-06,
      "loss": 0.0627,
      "step": 26830
    },
    {
      "epoch": 2.5587081909179686e-05,
      "model_forward_time": 1.7369110584259033,
      "step": 26830
    },
    {
      "dynamics_loss_avg": 0.05804514242336154,
      "epoch": 2.5587081909179686e-05,
      "step": 26830
    },
    {
      "action_loss_avg": 0.0018368854478467255,
      "epoch": 2.5587081909179686e-05,
      "step": 26830
    },
    {
      "epoch": 2.5587081909179686e-05,
      "step": 26830,
      "training_step_time": 8.54064393043518
    },
    {
      "epoch": 2.5588035583496093e-05,
      "model_forward_time": 2.911412000656128,
      "step": 26831
    },
    {
      "epoch": 2.5588035583496093e-05,
      "step": 26831,
      "training_step_time": 8.600560665130615
    },
    {
      "epoch": 2.55889892578125e-05,
      "model_forward_time": 1.7264657020568848,
      "step": 26832
    },
    {
      "epoch": 2.55889892578125e-05,
      "step": 26832,
      "training_step_time": 7.7701451778411865
    },
    {
      "epoch": 2.5589942932128905e-05,
      "model_forward_time": 1.7312953472137451,
      "step": 26833
    },
    {
      "epoch": 2.5589942932128905e-05,
      "step": 26833,
      "training_step_time": 7.86299991607666
    },
    {
      "epoch": 2.5590896606445312e-05,
      "model_forward_time": 1.7274386882781982,
      "step": 26834
    },
    {
      "epoch": 2.5590896606445312e-05,
      "step": 26834,
      "training_step_time": 8.424786567687988
    },
    {
      "epoch": 2.559185028076172e-05,
      "model_forward_time": 1.7328639030456543,
      "step": 26835
    },
    {
      "epoch": 2.559185028076172e-05,
      "step": 26835,
      "training_step_time": 8.962007522583008
    },
    {
      "epoch": 2.5592803955078124e-05,
      "model_forward_time": 1.744657278060913,
      "step": 26836
    },
    {
      "epoch": 2.5592803955078124e-05,
      "step": 26836,
      "training_step_time": 7.77090859413147
    },
    {
      "epoch": 2.559375762939453e-05,
      "model_forward_time": 1.7294561862945557,
      "step": 26837
    },
    {
      "epoch": 2.559375762939453e-05,
      "step": 26837,
      "training_step_time": 8.358085632324219
    },
    {
      "epoch": 2.559471130371094e-05,
      "model_forward_time": 1.7302989959716797,
      "step": 26838
    },
    {
      "epoch": 2.559471130371094e-05,
      "step": 26838,
      "training_step_time": 8.616102457046509
    },
    {
      "epoch": 2.5595664978027342e-05,
      "model_forward_time": 1.7308599948883057,
      "step": 26839
    },
    {
      "epoch": 2.5595664978027342e-05,
      "step": 26839,
      "training_step_time": 7.841612339019775
    },
    {
      "epoch": 2.559661865234375e-05,
      "grad_norm": 0.19233083724975586,
      "learning_rate": 8.751757905112264e-06,
      "loss": 0.0629,
      "step": 26840
    },
    {
      "epoch": 2.559661865234375e-05,
      "model_forward_time": 2.894197702407837,
      "step": 26840
    },
    {
      "dynamics_loss_avg": 0.056041070632636546,
      "epoch": 2.559661865234375e-05,
      "step": 26840
    },
    {
      "action_loss_avg": 0.003479045466519892,
      "epoch": 2.559661865234375e-05,
      "step": 26840
    },
    {
      "epoch": 2.559661865234375e-05,
      "step": 26840,
      "training_step_time": 8.937026977539062
    },
    {
      "epoch": 2.5597572326660157e-05,
      "model_forward_time": 1.7272617816925049,
      "step": 26841
    },
    {
      "epoch": 2.5597572326660157e-05,
      "step": 26841,
      "training_step_time": 7.751231908798218
    },
    {
      "epoch": 2.559852600097656e-05,
      "model_forward_time": 1.734684944152832,
      "step": 26842
    },
    {
      "epoch": 2.559852600097656e-05,
      "step": 26842,
      "training_step_time": 7.692137002944946
    },
    {
      "epoch": 2.559947967529297e-05,
      "model_forward_time": 1.7201557159423828,
      "step": 26843
    },
    {
      "epoch": 2.559947967529297e-05,
      "step": 26843,
      "training_step_time": 7.595533847808838
    },
    {
      "epoch": 2.5600433349609376e-05,
      "model_forward_time": 1.7565431594848633,
      "step": 26844
    },
    {
      "epoch": 2.5600433349609376e-05,
      "step": 26844,
      "training_step_time": 7.660296678543091
    },
    {
      "epoch": 2.560138702392578e-05,
      "model_forward_time": 1.7292149066925049,
      "step": 26845
    },
    {
      "epoch": 2.560138702392578e-05,
      "step": 26845,
      "training_step_time": 7.6696937084198
    },
    {
      "epoch": 2.5602340698242188e-05,
      "model_forward_time": 1.7323105335235596,
      "step": 26846
    },
    {
      "epoch": 2.5602340698242188e-05,
      "step": 26846,
      "training_step_time": 7.699862957000732
    },
    {
      "epoch": 2.5603294372558595e-05,
      "model_forward_time": 1.737792730331421,
      "step": 26847
    },
    {
      "epoch": 2.5603294372558595e-05,
      "step": 26847,
      "training_step_time": 7.656335353851318
    },
    {
      "epoch": 2.5604248046875e-05,
      "model_forward_time": 1.7304577827453613,
      "step": 26848
    },
    {
      "epoch": 2.5604248046875e-05,
      "step": 26848,
      "training_step_time": 9.428212642669678
    },
    {
      "epoch": 2.5605201721191406e-05,
      "model_forward_time": 2.876034736633301,
      "step": 26849
    },
    {
      "epoch": 2.5605201721191406e-05,
      "step": 26849,
      "training_step_time": 9.997619390487671
    },
    {
      "epoch": 2.5606155395507814e-05,
      "grad_norm": 0.14104297757148743,
      "learning_rate": 8.75066469202244e-06,
      "loss": 0.0584,
      "step": 26850
    },
    {
      "epoch": 2.5606155395507814e-05,
      "model_forward_time": 1.7324821949005127,
      "step": 26850
    },
    {
      "dynamics_loss_avg": 0.06307563558220863,
      "epoch": 2.5606155395507814e-05,
      "step": 26850
    },
    {
      "action_loss_avg": 0.0023549064062535763,
      "epoch": 2.5606155395507814e-05,
      "step": 26850
    },
    {
      "epoch": 2.5606155395507814e-05,
      "step": 26850,
      "training_step_time": 8.487176895141602
    },
    {
      "epoch": 2.5607109069824218e-05,
      "model_forward_time": 1.726797103881836,
      "step": 26851
    },
    {
      "epoch": 2.5607109069824218e-05,
      "step": 26851,
      "training_step_time": 8.379013061523438
    },
    {
      "epoch": 2.5608062744140625e-05,
      "model_forward_time": 1.7375879287719727,
      "step": 26852
    },
    {
      "epoch": 2.5608062744140625e-05,
      "step": 26852,
      "training_step_time": 7.530215263366699
    },
    {
      "epoch": 2.5609016418457033e-05,
      "model_forward_time": 1.723809003829956,
      "step": 26853
    },
    {
      "epoch": 2.5609016418457033e-05,
      "step": 26853,
      "training_step_time": 7.403185129165649
    },
    {
      "epoch": 2.5609970092773437e-05,
      "model_forward_time": 1.7283520698547363,
      "step": 26854
    },
    {
      "epoch": 2.5609970092773437e-05,
      "step": 26854,
      "training_step_time": 7.634108304977417
    },
    {
      "epoch": 2.5610923767089844e-05,
      "model_forward_time": 1.7284214496612549,
      "step": 26855
    },
    {
      "epoch": 2.5610923767089844e-05,
      "step": 26855,
      "training_step_time": 7.579628944396973
    },
    {
      "epoch": 2.561187744140625e-05,
      "model_forward_time": 1.7290880680084229,
      "step": 26856
    },
    {
      "epoch": 2.561187744140625e-05,
      "step": 26856,
      "training_step_time": 8.281639814376831
    },
    {
      "epoch": 2.5612831115722656e-05,
      "model_forward_time": 1.7343695163726807,
      "step": 26857
    },
    {
      "epoch": 2.5612831115722656e-05,
      "step": 26857,
      "training_step_time": 9.399134397506714
    },
    {
      "epoch": 2.5613784790039063e-05,
      "model_forward_time": 1.7705421447753906,
      "step": 26858
    },
    {
      "epoch": 2.5613784790039063e-05,
      "step": 26858,
      "training_step_time": 10.442903757095337
    },
    {
      "epoch": 2.561473846435547e-05,
      "model_forward_time": 1.7294261455535889,
      "step": 26859
    },
    {
      "epoch": 2.561473846435547e-05,
      "step": 26859,
      "training_step_time": 8.653272867202759
    },
    {
      "epoch": 2.5615692138671874e-05,
      "grad_norm": 0.19634005427360535,
      "learning_rate": 8.749571068765568e-06,
      "loss": 0.063,
      "step": 26860
    },
    {
      "epoch": 2.5615692138671874e-05,
      "model_forward_time": 1.7242600917816162,
      "step": 26860
    },
    {
      "dynamics_loss_avg": 0.06729240901768208,
      "epoch": 2.5615692138671874e-05,
      "step": 26860
    },
    {
      "action_loss_avg": 0.0037206541164778173,
      "epoch": 2.5615692138671874e-05,
      "step": 26860
    },
    {
      "epoch": 2.5615692138671874e-05,
      "step": 26860,
      "training_step_time": 8.218055248260498
    },
    {
      "epoch": 2.5616645812988282e-05,
      "model_forward_time": 1.731698989868164,
      "step": 26861
    },
    {
      "epoch": 2.5616645812988282e-05,
      "step": 26861,
      "training_step_time": 8.48218584060669
    },
    {
      "epoch": 2.5617599487304686e-05,
      "model_forward_time": 1.733248233795166,
      "step": 26862
    },
    {
      "epoch": 2.5617599487304686e-05,
      "step": 26862,
      "training_step_time": 7.352198123931885
    },
    {
      "epoch": 2.5618553161621093e-05,
      "model_forward_time": 1.7324271202087402,
      "step": 26863
    },
    {
      "epoch": 2.5618553161621093e-05,
      "step": 26863,
      "training_step_time": 7.758566617965698
    },
    {
      "epoch": 2.56195068359375e-05,
      "model_forward_time": 1.7314677238464355,
      "step": 26864
    },
    {
      "epoch": 2.56195068359375e-05,
      "step": 26864,
      "training_step_time": 7.673593521118164
    },
    {
      "epoch": 2.5620460510253905e-05,
      "model_forward_time": 1.733396291732788,
      "step": 26865
    },
    {
      "epoch": 2.5620460510253905e-05,
      "step": 26865,
      "training_step_time": 8.718005657196045
    },
    {
      "epoch": 2.5621414184570312e-05,
      "model_forward_time": 1.7297282218933105,
      "step": 26866
    },
    {
      "epoch": 2.5621414184570312e-05,
      "step": 26866,
      "training_step_time": 7.713520050048828
    },
    {
      "epoch": 2.562236785888672e-05,
      "model_forward_time": 1.7371885776519775,
      "step": 26867
    },
    {
      "epoch": 2.562236785888672e-05,
      "step": 26867,
      "training_step_time": 8.583112239837646
    },
    {
      "epoch": 2.5623321533203124e-05,
      "model_forward_time": 2.8813838958740234,
      "step": 26868
    },
    {
      "epoch": 2.5623321533203124e-05,
      "step": 26868,
      "training_step_time": 8.771040916442871
    },
    {
      "epoch": 2.562427520751953e-05,
      "model_forward_time": 1.7300119400024414,
      "step": 26869
    },
    {
      "epoch": 2.562427520751953e-05,
      "step": 26869,
      "training_step_time": 8.366405487060547
    },
    {
      "epoch": 2.562522888183594e-05,
      "grad_norm": 0.1412806212902069,
      "learning_rate": 8.748477035461237e-06,
      "loss": 0.059,
      "step": 26870
    },
    {
      "epoch": 2.562522888183594e-05,
      "model_forward_time": 1.7341868877410889,
      "step": 26870
    },
    {
      "dynamics_loss_avg": 0.03576932922005653,
      "epoch": 2.562522888183594e-05,
      "step": 26870
    },
    {
      "action_loss_avg": 0.0019865554582793265,
      "epoch": 2.562522888183594e-05,
      "step": 26870
    },
    {
      "epoch": 2.562522888183594e-05,
      "step": 26870,
      "training_step_time": 7.693568229675293
    },
    {
      "epoch": 2.5626182556152342e-05,
      "model_forward_time": 1.731313943862915,
      "step": 26871
    },
    {
      "epoch": 2.5626182556152342e-05,
      "step": 26871,
      "training_step_time": 7.586097717285156
    },
    {
      "epoch": 2.562713623046875e-05,
      "model_forward_time": 1.7268691062927246,
      "step": 26872
    },
    {
      "epoch": 2.562713623046875e-05,
      "step": 26872,
      "training_step_time": 7.699073791503906
    },
    {
      "epoch": 2.5628089904785157e-05,
      "model_forward_time": 1.7307846546173096,
      "step": 26873
    },
    {
      "epoch": 2.5628089904785157e-05,
      "step": 26873,
      "training_step_time": 7.656013011932373
    },
    {
      "epoch": 2.562904357910156e-05,
      "model_forward_time": 1.7280442714691162,
      "step": 26874
    },
    {
      "epoch": 2.562904357910156e-05,
      "step": 26874,
      "training_step_time": 8.671970129013062
    },
    {
      "epoch": 2.562999725341797e-05,
      "model_forward_time": 1.729569435119629,
      "step": 26875
    },
    {
      "epoch": 2.562999725341797e-05,
      "step": 26875,
      "training_step_time": 7.553622484207153
    },
    {
      "epoch": 2.5630950927734376e-05,
      "model_forward_time": 1.7378613948822021,
      "step": 26876
    },
    {
      "epoch": 2.5630950927734376e-05,
      "step": 26876,
      "training_step_time": 8.53513789176941
    },
    {
      "epoch": 2.563190460205078e-05,
      "model_forward_time": 2.865204095840454,
      "step": 26877
    },
    {
      "epoch": 2.563190460205078e-05,
      "step": 26877,
      "training_step_time": 7.773591995239258
    },
    {
      "epoch": 2.5632858276367188e-05,
      "model_forward_time": 1.729269027709961,
      "step": 26878
    },
    {
      "epoch": 2.5632858276367188e-05,
      "step": 26878,
      "training_step_time": 7.7839624881744385
    },
    {
      "epoch": 2.5633811950683595e-05,
      "model_forward_time": 1.7342517375946045,
      "step": 26879
    },
    {
      "epoch": 2.5633811950683595e-05,
      "step": 26879,
      "training_step_time": 7.752635955810547
    },
    {
      "epoch": 2.5634765625e-05,
      "grad_norm": 0.15135693550109863,
      "learning_rate": 8.747382592229095e-06,
      "loss": 0.0613,
      "step": 26880
    },
    {
      "epoch": 2.5634765625e-05,
      "model_forward_time": 1.7270989418029785,
      "step": 26880
    },
    {
      "dynamics_loss_avg": 0.04810806512832642,
      "epoch": 2.5634765625e-05,
      "step": 26880
    },
    {
      "action_loss_avg": 0.0023536267108283937,
      "epoch": 2.5634765625e-05,
      "step": 26880
    },
    {
      "epoch": 2.5634765625e-05,
      "step": 26880,
      "training_step_time": 7.560044288635254
    },
    {
      "epoch": 2.5635719299316406e-05,
      "model_forward_time": 1.7434594631195068,
      "step": 26881
    },
    {
      "epoch": 2.5635719299316406e-05,
      "step": 26881,
      "training_step_time": 7.642754077911377
    },
    {
      "epoch": 2.5636672973632814e-05,
      "model_forward_time": 1.7175076007843018,
      "step": 26882
    },
    {
      "epoch": 2.5636672973632814e-05,
      "step": 26882,
      "training_step_time": 7.846292495727539
    },
    {
      "epoch": 2.5637626647949218e-05,
      "model_forward_time": 1.7254834175109863,
      "step": 26883
    },
    {
      "epoch": 2.5637626647949218e-05,
      "step": 26883,
      "training_step_time": 7.827185153961182
    },
    {
      "epoch": 2.5638580322265625e-05,
      "model_forward_time": 1.0229036808013916,
      "step": 26884
    },
    {
      "epoch": 2.5638580322265625e-05,
      "step": 26884,
      "training_step_time": 7.862246513366699
    },
    {
      "epoch": 2.5639533996582033e-05,
      "model_forward_time": 1.7432167530059814,
      "step": 26885
    },
    {
      "epoch": 2.5639533996582033e-05,
      "step": 26885,
      "training_step_time": 7.725444555282593
    },
    {
      "epoch": 2.5640487670898437e-05,
      "model_forward_time": 1.8426728248596191,
      "step": 26886
    },
    {
      "epoch": 2.5640487670898437e-05,
      "step": 26886,
      "training_step_time": 8.92075228691101
    },
    {
      "epoch": 2.5641441345214844e-05,
      "model_forward_time": 1.7367722988128662,
      "step": 26887
    },
    {
      "epoch": 2.5641441345214844e-05,
      "step": 26887,
      "training_step_time": 9.43507432937622
    },
    {
      "epoch": 2.564239501953125e-05,
      "model_forward_time": 1.7284204959869385,
      "step": 26888
    },
    {
      "epoch": 2.564239501953125e-05,
      "step": 26888,
      "training_step_time": 8.679439067840576
    },
    {
      "epoch": 2.5643348693847656e-05,
      "model_forward_time": 1.7364647388458252,
      "step": 26889
    },
    {
      "epoch": 2.5643348693847656e-05,
      "step": 26889,
      "training_step_time": 7.755166530609131
    },
    {
      "epoch": 2.5644302368164063e-05,
      "grad_norm": 0.14381662011146545,
      "learning_rate": 8.746287739188828e-06,
      "loss": 0.06,
      "step": 26890
    },
    {
      "epoch": 2.5644302368164063e-05,
      "model_forward_time": 1.7322845458984375,
      "step": 26890
    },
    {
      "dynamics_loss_avg": 0.0634774660691619,
      "epoch": 2.5644302368164063e-05,
      "step": 26890
    },
    {
      "action_loss_avg": 0.002768591226777062,
      "epoch": 2.5644302368164063e-05,
      "step": 26890
    },
    {
      "epoch": 2.5644302368164063e-05,
      "step": 26890,
      "training_step_time": 7.736727237701416
    },
    {
      "epoch": 2.5645256042480467e-05,
      "model_forward_time": 1.7566730976104736,
      "step": 26891
    },
    {
      "epoch": 2.5645256042480467e-05,
      "step": 26891,
      "training_step_time": 7.881620645523071
    },
    {
      "epoch": 2.5646209716796874e-05,
      "model_forward_time": 1.728679895401001,
      "step": 26892
    },
    {
      "epoch": 2.5646209716796874e-05,
      "step": 26892,
      "training_step_time": 7.799771547317505
    },
    {
      "epoch": 2.5647163391113282e-05,
      "model_forward_time": 1.7340867519378662,
      "step": 26893
    },
    {
      "epoch": 2.5647163391113282e-05,
      "step": 26893,
      "training_step_time": 7.790262222290039
    },
    {
      "epoch": 2.5648117065429686e-05,
      "model_forward_time": 1.7291371822357178,
      "step": 26894
    },
    {
      "epoch": 2.5648117065429686e-05,
      "step": 26894,
      "training_step_time": 8.41986083984375
    },
    {
      "epoch": 2.5649070739746093e-05,
      "model_forward_time": 1.7775614261627197,
      "step": 26895
    },
    {
      "epoch": 2.5649070739746093e-05,
      "step": 26895,
      "training_step_time": 8.86514949798584
    },
    {
      "epoch": 2.56500244140625e-05,
      "model_forward_time": 2.9210410118103027,
      "step": 26896
    },
    {
      "epoch": 2.56500244140625e-05,
      "step": 26896,
      "training_step_time": 7.794673442840576
    },
    {
      "epoch": 2.5650978088378905e-05,
      "model_forward_time": 1.7321429252624512,
      "step": 26897
    },
    {
      "epoch": 2.5650978088378905e-05,
      "step": 26897,
      "training_step_time": 8.719078540802002
    },
    {
      "epoch": 2.5651931762695312e-05,
      "model_forward_time": 1.7335457801818848,
      "step": 26898
    },
    {
      "epoch": 2.5651931762695312e-05,
      "step": 26898,
      "training_step_time": 7.627975702285767
    },
    {
      "epoch": 2.565288543701172e-05,
      "model_forward_time": 1.7336187362670898,
      "step": 26899
    },
    {
      "epoch": 2.565288543701172e-05,
      "step": 26899,
      "training_step_time": 7.79115104675293
    },
    {
      "epoch": 2.5653839111328124e-05,
      "grad_norm": 0.11011626571416855,
      "learning_rate": 8.745192476460165e-06,
      "loss": 0.0616,
      "step": 26900
    },
    {
      "epoch": 2.5653839111328124e-05,
      "model_forward_time": 1.7391314506530762,
      "step": 26900
    },
    {
      "dynamics_loss_avg": 0.0612797724083066,
      "epoch": 2.5653839111328124e-05,
      "step": 26900
    },
    {
      "action_loss_avg": 0.0038144302321597933,
      "epoch": 2.5653839111328124e-05,
      "step": 26900
    },
    {
      "epoch": 2.5653839111328124e-05,
      "step": 26900,
      "training_step_time": 7.81381893157959
    },
    {
      "epoch": 2.565479278564453e-05,
      "model_forward_time": 1.7262656688690186,
      "step": 26901
    },
    {
      "epoch": 2.565479278564453e-05,
      "step": 26901,
      "training_step_time": 7.7922563552856445
    },
    {
      "epoch": 2.565574645996094e-05,
      "model_forward_time": 1.7342865467071533,
      "step": 26902
    },
    {
      "epoch": 2.565574645996094e-05,
      "step": 26902,
      "training_step_time": 7.794888019561768
    },
    {
      "epoch": 2.5656700134277343e-05,
      "model_forward_time": 1.7272961139678955,
      "step": 26903
    },
    {
      "epoch": 2.5656700134277343e-05,
      "step": 26903,
      "training_step_time": 7.808435678482056
    },
    {
      "epoch": 2.565765380859375e-05,
      "model_forward_time": 1.7621266841888428,
      "step": 26904
    },
    {
      "epoch": 2.565765380859375e-05,
      "step": 26904,
      "training_step_time": 7.785990476608276
    },
    {
      "epoch": 2.5658607482910157e-05,
      "model_forward_time": 2.8696188926696777,
      "step": 26905
    },
    {
      "epoch": 2.5658607482910157e-05,
      "step": 26905,
      "training_step_time": 8.579273462295532
    },
    {
      "epoch": 2.565956115722656e-05,
      "model_forward_time": 1.7315189838409424,
      "step": 26906
    },
    {
      "epoch": 2.565956115722656e-05,
      "step": 26906,
      "training_step_time": 7.805123329162598
    },
    {
      "epoch": 2.566051483154297e-05,
      "model_forward_time": 1.7321014404296875,
      "step": 26907
    },
    {
      "epoch": 2.566051483154297e-05,
      "step": 26907,
      "training_step_time": 7.747660875320435
    },
    {
      "epoch": 2.5661468505859376e-05,
      "model_forward_time": 1.727189064025879,
      "step": 26908
    },
    {
      "epoch": 2.5661468505859376e-05,
      "step": 26908,
      "training_step_time": 7.542317628860474
    },
    {
      "epoch": 2.566242218017578e-05,
      "model_forward_time": 1.7304294109344482,
      "step": 26909
    },
    {
      "epoch": 2.566242218017578e-05,
      "step": 26909,
      "training_step_time": 7.627307891845703
    },
    {
      "epoch": 2.5663375854492188e-05,
      "grad_norm": 0.1785232573747635,
      "learning_rate": 8.744096804162882e-06,
      "loss": 0.0631,
      "step": 26910
    },
    {
      "epoch": 2.5663375854492188e-05,
      "model_forward_time": 1.719076156616211,
      "step": 26910
    },
    {
      "dynamics_loss_avg": 0.058607651852071285,
      "epoch": 2.5663375854492188e-05,
      "step": 26910
    },
    {
      "action_loss_avg": 0.002465671510435641,
      "epoch": 2.5663375854492188e-05,
      "step": 26910
    },
    {
      "epoch": 2.5663375854492188e-05,
      "step": 26910,
      "training_step_time": 7.729874849319458
    },
    {
      "epoch": 2.5664329528808595e-05,
      "model_forward_time": 1.7282614707946777,
      "step": 26911
    },
    {
      "epoch": 2.5664329528808595e-05,
      "step": 26911,
      "training_step_time": 8.273449897766113
    },
    {
      "epoch": 2.5665283203125e-05,
      "model_forward_time": 1.7254843711853027,
      "step": 26912
    },
    {
      "epoch": 2.5665283203125e-05,
      "step": 26912,
      "training_step_time": 8.430572271347046
    },
    {
      "epoch": 2.5666236877441406e-05,
      "model_forward_time": 1.7317595481872559,
      "step": 26913
    },
    {
      "epoch": 2.5666236877441406e-05,
      "step": 26913,
      "training_step_time": 10.165063381195068
    },
    {
      "epoch": 2.5667190551757814e-05,
      "model_forward_time": 2.91121768951416,
      "step": 26914
    },
    {
      "epoch": 2.5667190551757814e-05,
      "step": 26914,
      "training_step_time": 7.78938627243042
    },
    {
      "epoch": 2.5668144226074218e-05,
      "model_forward_time": 1.7307074069976807,
      "step": 26915
    },
    {
      "epoch": 2.5668144226074218e-05,
      "step": 26915,
      "training_step_time": 8.604138135910034
    },
    {
      "epoch": 2.5669097900390625e-05,
      "model_forward_time": 1.729494333267212,
      "step": 26916
    },
    {
      "epoch": 2.5669097900390625e-05,
      "step": 26916,
      "training_step_time": 7.832887887954712
    },
    {
      "epoch": 2.5670051574707033e-05,
      "model_forward_time": 1.7369718551635742,
      "step": 26917
    },
    {
      "epoch": 2.5670051574707033e-05,
      "step": 26917,
      "training_step_time": 7.555474519729614
    },
    {
      "epoch": 2.5671005249023437e-05,
      "model_forward_time": 1.7272989749908447,
      "step": 26918
    },
    {
      "epoch": 2.5671005249023437e-05,
      "step": 26918,
      "training_step_time": 8.349234342575073
    },
    {
      "epoch": 2.5671958923339844e-05,
      "model_forward_time": 1.7309067249298096,
      "step": 26919
    },
    {
      "epoch": 2.5671958923339844e-05,
      "step": 26919,
      "training_step_time": 7.7290873527526855
    },
    {
      "epoch": 2.567291259765625e-05,
      "grad_norm": 0.1337306797504425,
      "learning_rate": 8.743000722416804e-06,
      "loss": 0.0607,
      "step": 26920
    },
    {
      "epoch": 2.567291259765625e-05,
      "model_forward_time": 1.7218272686004639,
      "step": 26920
    },
    {
      "dynamics_loss_avg": 0.041361203975975515,
      "epoch": 2.567291259765625e-05,
      "step": 26920
    },
    {
      "action_loss_avg": 0.0019283144967630506,
      "epoch": 2.567291259765625e-05,
      "step": 26920
    },
    {
      "epoch": 2.567291259765625e-05,
      "step": 26920,
      "training_step_time": 8.346457242965698
    },
    {
      "epoch": 2.5673866271972656e-05,
      "model_forward_time": 1.7209587097167969,
      "step": 26921
    },
    {
      "epoch": 2.5673866271972656e-05,
      "step": 26921,
      "training_step_time": 9.84457802772522
    },
    {
      "epoch": 2.5674819946289063e-05,
      "model_forward_time": 1.7226364612579346,
      "step": 26922
    },
    {
      "epoch": 2.5674819946289063e-05,
      "step": 26922,
      "training_step_time": 8.358060359954834
    },
    {
      "epoch": 2.5675773620605467e-05,
      "model_forward_time": 2.8901865482330322,
      "step": 26923
    },
    {
      "epoch": 2.5675773620605467e-05,
      "step": 26923,
      "training_step_time": 8.460192203521729
    },
    {
      "epoch": 2.5676727294921875e-05,
      "model_forward_time": 1.7439568042755127,
      "step": 26924
    },
    {
      "epoch": 2.5676727294921875e-05,
      "step": 26924,
      "training_step_time": 9.590791940689087
    },
    {
      "epoch": 2.5677680969238282e-05,
      "model_forward_time": 1.7318449020385742,
      "step": 26925
    },
    {
      "epoch": 2.5677680969238282e-05,
      "step": 26925,
      "training_step_time": 8.219851970672607
    },
    {
      "epoch": 2.5678634643554686e-05,
      "model_forward_time": 1.7292938232421875,
      "step": 26926
    },
    {
      "epoch": 2.5678634643554686e-05,
      "step": 26926,
      "training_step_time": 7.515427350997925
    },
    {
      "epoch": 2.5679588317871093e-05,
      "model_forward_time": 1.7301373481750488,
      "step": 26927
    },
    {
      "epoch": 2.5679588317871093e-05,
      "step": 26927,
      "training_step_time": 8.318603038787842
    },
    {
      "epoch": 2.56805419921875e-05,
      "model_forward_time": 1.731360912322998,
      "step": 26928
    },
    {
      "epoch": 2.56805419921875e-05,
      "step": 26928,
      "training_step_time": 7.858957290649414
    },
    {
      "epoch": 2.5681495666503905e-05,
      "model_forward_time": 1.7322807312011719,
      "step": 26929
    },
    {
      "epoch": 2.5681495666503905e-05,
      "step": 26929,
      "training_step_time": 8.31244707107544
    },
    {
      "epoch": 2.5682449340820312e-05,
      "grad_norm": 0.21931719779968262,
      "learning_rate": 8.741904231341794e-06,
      "loss": 0.0626,
      "step": 26930
    },
    {
      "epoch": 2.5682449340820312e-05,
      "model_forward_time": 1.7461645603179932,
      "step": 26930
    },
    {
      "dynamics_loss_avg": 0.049892483279109,
      "epoch": 2.5682449340820312e-05,
      "step": 26930
    },
    {
      "action_loss_avg": 0.002110674825962633,
      "epoch": 2.5682449340820312e-05,
      "step": 26930
    },
    {
      "epoch": 2.5682449340820312e-05,
      "step": 26930,
      "training_step_time": 7.785963773727417
    },
    {
      "epoch": 2.568340301513672e-05,
      "model_forward_time": 1.720867395401001,
      "step": 26931
    },
    {
      "epoch": 2.568340301513672e-05,
      "step": 26931,
      "training_step_time": 7.865549087524414
    },
    {
      "epoch": 2.5684356689453124e-05,
      "model_forward_time": 1.8248980045318604,
      "step": 26932
    },
    {
      "epoch": 2.5684356689453124e-05,
      "step": 26932,
      "training_step_time": 8.352189540863037
    },
    {
      "epoch": 2.568531036376953e-05,
      "model_forward_time": 1.7311639785766602,
      "step": 26933
    },
    {
      "epoch": 2.568531036376953e-05,
      "step": 26933,
      "training_step_time": 8.37142300605774
    },
    {
      "epoch": 2.568626403808594e-05,
      "model_forward_time": 1.7255768775939941,
      "step": 26934
    },
    {
      "epoch": 2.568626403808594e-05,
      "step": 26934,
      "training_step_time": 7.693803071975708
    },
    {
      "epoch": 2.5687217712402343e-05,
      "model_forward_time": 1.7322287559509277,
      "step": 26935
    },
    {
      "epoch": 2.5687217712402343e-05,
      "step": 26935,
      "training_step_time": 7.64947772026062
    },
    {
      "epoch": 2.568817138671875e-05,
      "model_forward_time": 1.7293128967285156,
      "step": 26936
    },
    {
      "epoch": 2.568817138671875e-05,
      "step": 26936,
      "training_step_time": 8.221421241760254
    },
    {
      "epoch": 2.5689125061035157e-05,
      "model_forward_time": 1.7308826446533203,
      "step": 26937
    },
    {
      "epoch": 2.5689125061035157e-05,
      "step": 26937,
      "training_step_time": 7.711897611618042
    },
    {
      "epoch": 2.569007873535156e-05,
      "model_forward_time": 1.7335050106048584,
      "step": 26938
    },
    {
      "epoch": 2.569007873535156e-05,
      "step": 26938,
      "training_step_time": 7.612283229827881
    },
    {
      "epoch": 2.569103240966797e-05,
      "model_forward_time": 1.737985372543335,
      "step": 26939
    },
    {
      "epoch": 2.569103240966797e-05,
      "step": 26939,
      "training_step_time": 7.637385606765747
    },
    {
      "epoch": 2.5691986083984376e-05,
      "grad_norm": 0.1260290890932083,
      "learning_rate": 8.740807331057762e-06,
      "loss": 0.0632,
      "step": 26940
    },
    {
      "epoch": 2.5691986083984376e-05,
      "model_forward_time": 1.7238779067993164,
      "step": 26940
    },
    {
      "dynamics_loss_avg": 0.05709137171506882,
      "epoch": 2.5691986083984376e-05,
      "step": 26940
    },
    {
      "action_loss_avg": 0.002516104152891785,
      "epoch": 2.5691986083984376e-05,
      "step": 26940
    },
    {
      "epoch": 2.5691986083984376e-05,
      "step": 26940,
      "training_step_time": 7.653970241546631
    },
    {
      "epoch": 2.569293975830078e-05,
      "model_forward_time": 1.794663429260254,
      "step": 26941
    },
    {
      "epoch": 2.569293975830078e-05,
      "step": 26941,
      "training_step_time": 7.706802845001221
    },
    {
      "epoch": 2.5693893432617188e-05,
      "model_forward_time": 2.8788516521453857,
      "step": 26942
    },
    {
      "epoch": 2.5693893432617188e-05,
      "step": 26942,
      "training_step_time": 7.910889148712158
    },
    {
      "epoch": 2.5694847106933595e-05,
      "model_forward_time": 1.7321388721466064,
      "step": 26943
    },
    {
      "epoch": 2.5694847106933595e-05,
      "step": 26943,
      "training_step_time": 8.29092288017273
    },
    {
      "epoch": 2.569580078125e-05,
      "model_forward_time": 1.731673240661621,
      "step": 26944
    },
    {
      "epoch": 2.569580078125e-05,
      "step": 26944,
      "training_step_time": 7.687387466430664
    },
    {
      "epoch": 2.5696754455566407e-05,
      "model_forward_time": 1.7278647422790527,
      "step": 26945
    },
    {
      "epoch": 2.5696754455566407e-05,
      "step": 26945,
      "training_step_time": 7.5833916664123535
    },
    {
      "epoch": 2.5697708129882814e-05,
      "model_forward_time": 1.7307133674621582,
      "step": 26946
    },
    {
      "epoch": 2.5697708129882814e-05,
      "step": 26946,
      "training_step_time": 7.674041271209717
    },
    {
      "epoch": 2.5698661804199218e-05,
      "model_forward_time": 1.727989673614502,
      "step": 26947
    },
    {
      "epoch": 2.5698661804199218e-05,
      "step": 26947,
      "training_step_time": 8.06033730506897
    },
    {
      "epoch": 2.5699615478515625e-05,
      "model_forward_time": 1.727065086364746,
      "step": 26948
    },
    {
      "epoch": 2.5699615478515625e-05,
      "step": 26948,
      "training_step_time": 8.602979183197021
    },
    {
      "epoch": 2.5700569152832033e-05,
      "model_forward_time": 1.729494333267212,
      "step": 26949
    },
    {
      "epoch": 2.5700569152832033e-05,
      "step": 26949,
      "training_step_time": 7.7059056758880615
    },
    {
      "epoch": 2.5701522827148437e-05,
      "grad_norm": 0.12461968511343002,
      "learning_rate": 8.739710021684666e-06,
      "loss": 0.0651,
      "step": 26950
    },
    {
      "epoch": 2.5701522827148437e-05,
      "model_forward_time": 1.7637317180633545,
      "step": 26950
    },
    {
      "dynamics_loss_avg": 0.07278220504522323,
      "epoch": 2.5701522827148437e-05,
      "step": 26950
    },
    {
      "action_loss_avg": 0.0025072745280340313,
      "epoch": 2.5701522827148437e-05,
      "step": 26950
    },
    {
      "epoch": 2.5701522827148437e-05,
      "step": 26950,
      "training_step_time": 8.870853900909424
    },
    {
      "epoch": 2.5702476501464844e-05,
      "model_forward_time": 2.8378913402557373,
      "step": 26951
    },
    {
      "epoch": 2.5702476501464844e-05,
      "step": 26951,
      "training_step_time": 8.45219087600708
    },
    {
      "epoch": 2.570343017578125e-05,
      "model_forward_time": 1.7316749095916748,
      "step": 26952
    },
    {
      "epoch": 2.570343017578125e-05,
      "step": 26952,
      "training_step_time": 8.752107620239258
    },
    {
      "epoch": 2.5704383850097656e-05,
      "model_forward_time": 1.7248823642730713,
      "step": 26953
    },
    {
      "epoch": 2.5704383850097656e-05,
      "step": 26953,
      "training_step_time": 7.724839210510254
    },
    {
      "epoch": 2.5705337524414063e-05,
      "model_forward_time": 1.7286772727966309,
      "step": 26954
    },
    {
      "epoch": 2.5705337524414063e-05,
      "step": 26954,
      "training_step_time": 7.589309453964233
    },
    {
      "epoch": 2.5706291198730467e-05,
      "model_forward_time": 1.793870449066162,
      "step": 26955
    },
    {
      "epoch": 2.5706291198730467e-05,
      "step": 26955,
      "training_step_time": 8.179429054260254
    },
    {
      "epoch": 2.5707244873046875e-05,
      "model_forward_time": 1.7302563190460205,
      "step": 26956
    },
    {
      "epoch": 2.5707244873046875e-05,
      "step": 26956,
      "training_step_time": 7.640625953674316
    },
    {
      "epoch": 2.5708198547363282e-05,
      "model_forward_time": 1.7344825267791748,
      "step": 26957
    },
    {
      "epoch": 2.5708198547363282e-05,
      "step": 26957,
      "training_step_time": 7.776599168777466
    },
    {
      "epoch": 2.5709152221679686e-05,
      "model_forward_time": 1.731633186340332,
      "step": 26958
    },
    {
      "epoch": 2.5709152221679686e-05,
      "step": 26958,
      "training_step_time": 7.773817300796509
    },
    {
      "epoch": 2.5710105895996093e-05,
      "model_forward_time": 1.724426507949829,
      "step": 26959
    },
    {
      "epoch": 2.5710105895996093e-05,
      "step": 26959,
      "training_step_time": 7.760707378387451
    },
    {
      "epoch": 2.57110595703125e-05,
      "grad_norm": 0.12719959020614624,
      "learning_rate": 8.738612303342504e-06,
      "loss": 0.0656,
      "step": 26960
    },
    {
      "epoch": 2.57110595703125e-05,
      "model_forward_time": 2.882781744003296,
      "step": 26960
    },
    {
      "dynamics_loss_avg": 0.05567399244755507,
      "epoch": 2.57110595703125e-05,
      "step": 26960
    },
    {
      "action_loss_avg": 0.0031908113218378277,
      "epoch": 2.57110595703125e-05,
      "step": 26960
    },
    {
      "epoch": 2.57110595703125e-05,
      "step": 26960,
      "training_step_time": 7.817622184753418
    },
    {
      "epoch": 2.5712013244628905e-05,
      "model_forward_time": 1.7375454902648926,
      "step": 26961
    },
    {
      "epoch": 2.5712013244628905e-05,
      "step": 26961,
      "training_step_time": 7.798185348510742
    },
    {
      "epoch": 2.5712966918945312e-05,
      "model_forward_time": 1.7233123779296875,
      "step": 26962
    },
    {
      "epoch": 2.5712966918945312e-05,
      "step": 26962,
      "training_step_time": 10.14031457901001
    },
    {
      "epoch": 2.571392059326172e-05,
      "model_forward_time": 1.7318649291992188,
      "step": 26963
    },
    {
      "epoch": 2.571392059326172e-05,
      "step": 26963,
      "training_step_time": 7.611651182174683
    },
    {
      "epoch": 2.5714874267578124e-05,
      "model_forward_time": 1.7253897190093994,
      "step": 26964
    },
    {
      "epoch": 2.5714874267578124e-05,
      "step": 26964,
      "training_step_time": 7.73103404045105
    },
    {
      "epoch": 2.571582794189453e-05,
      "model_forward_time": 1.7273521423339844,
      "step": 26965
    },
    {
      "epoch": 2.571582794189453e-05,
      "step": 26965,
      "training_step_time": 7.657727479934692
    },
    {
      "epoch": 2.571678161621094e-05,
      "model_forward_time": 1.7240467071533203,
      "step": 26966
    },
    {
      "epoch": 2.571678161621094e-05,
      "step": 26966,
      "training_step_time": 8.68593454360962
    },
    {
      "epoch": 2.5717735290527343e-05,
      "model_forward_time": 1.7278223037719727,
      "step": 26967
    },
    {
      "epoch": 2.5717735290527343e-05,
      "step": 26967,
      "training_step_time": 7.7234947681427
    },
    {
      "epoch": 2.571868896484375e-05,
      "model_forward_time": 1.732191562652588,
      "step": 26968
    },
    {
      "epoch": 2.571868896484375e-05,
      "step": 26968,
      "training_step_time": 7.637748956680298
    },
    {
      "epoch": 2.5719642639160157e-05,
      "model_forward_time": 2.8700804710388184,
      "step": 26969
    },
    {
      "epoch": 2.5719642639160157e-05,
      "step": 26969,
      "training_step_time": 7.820899248123169
    },
    {
      "epoch": 2.572059631347656e-05,
      "grad_norm": 0.1430027335882187,
      "learning_rate": 8.73751417615132e-06,
      "loss": 0.0646,
      "step": 26970
    },
    {
      "epoch": 2.572059631347656e-05,
      "model_forward_time": 1.7309987545013428,
      "step": 26970
    },
    {
      "dynamics_loss_avg": 0.06263378746807576,
      "epoch": 2.572059631347656e-05,
      "step": 26970
    },
    {
      "action_loss_avg": 0.002165930307819508,
      "epoch": 2.572059631347656e-05,
      "step": 26970
    },
    {
      "epoch": 2.572059631347656e-05,
      "step": 26970,
      "training_step_time": 7.844020843505859
    },
    {
      "epoch": 2.572154998779297e-05,
      "model_forward_time": 1.732447862625122,
      "step": 26971
    },
    {
      "epoch": 2.572154998779297e-05,
      "step": 26971,
      "training_step_time": 10.014132022857666
    },
    {
      "epoch": 2.5722503662109376e-05,
      "model_forward_time": 1.7396256923675537,
      "step": 26972
    },
    {
      "epoch": 2.5722503662109376e-05,
      "step": 26972,
      "training_step_time": 8.314450979232788
    },
    {
      "epoch": 2.572345733642578e-05,
      "model_forward_time": 1.7327709197998047,
      "step": 26973
    },
    {
      "epoch": 2.572345733642578e-05,
      "step": 26973,
      "training_step_time": 7.581519603729248
    },
    {
      "epoch": 2.5724411010742188e-05,
      "model_forward_time": 1.7267096042633057,
      "step": 26974
    },
    {
      "epoch": 2.5724411010742188e-05,
      "step": 26974,
      "training_step_time": 7.644058704376221
    },
    {
      "epoch": 2.5725364685058595e-05,
      "model_forward_time": 1.7290260791778564,
      "step": 26975
    },
    {
      "epoch": 2.5725364685058595e-05,
      "step": 26975,
      "training_step_time": 8.477519035339355
    },
    {
      "epoch": 2.5726318359375e-05,
      "model_forward_time": 1.7364239692687988,
      "step": 26976
    },
    {
      "epoch": 2.5726318359375e-05,
      "step": 26976,
      "training_step_time": 7.699801921844482
    },
    {
      "epoch": 2.5727272033691407e-05,
      "model_forward_time": 1.7206861972808838,
      "step": 26977
    },
    {
      "epoch": 2.5727272033691407e-05,
      "step": 26977,
      "training_step_time": 8.331480741500854
    },
    {
      "epoch": 2.5728225708007814e-05,
      "model_forward_time": 2.8739354610443115,
      "step": 26978
    },
    {
      "epoch": 2.5728225708007814e-05,
      "step": 26978,
      "training_step_time": 7.696730852127075
    },
    {
      "epoch": 2.5729179382324218e-05,
      "model_forward_time": 1.7299809455871582,
      "step": 26979
    },
    {
      "epoch": 2.5729179382324218e-05,
      "step": 26979,
      "training_step_time": 8.777166604995728
    },
    {
      "epoch": 2.5730133056640625e-05,
      "grad_norm": 0.17578554153442383,
      "learning_rate": 8.736415640231208e-06,
      "loss": 0.0645,
      "step": 26980
    },
    {
      "epoch": 2.5730133056640625e-05,
      "model_forward_time": 1.7358906269073486,
      "step": 26980
    },
    {
      "dynamics_loss_avg": 0.054516743682324884,
      "epoch": 2.5730133056640625e-05,
      "step": 26980
    },
    {
      "action_loss_avg": 0.002631111803930253,
      "epoch": 2.5730133056640625e-05,
      "step": 26980
    },
    {
      "epoch": 2.5730133056640625e-05,
      "step": 26980,
      "training_step_time": 7.605219841003418
    },
    {
      "epoch": 2.5731086730957033e-05,
      "model_forward_time": 1.7258973121643066,
      "step": 26981
    },
    {
      "epoch": 2.5731086730957033e-05,
      "step": 26981,
      "training_step_time": 7.6681227684021
    },
    {
      "epoch": 2.5732040405273437e-05,
      "model_forward_time": 1.7329738140106201,
      "step": 26982
    },
    {
      "epoch": 2.5732040405273437e-05,
      "step": 26982,
      "training_step_time": 7.644489765167236
    },
    {
      "epoch": 2.5732994079589844e-05,
      "model_forward_time": 1.733128547668457,
      "step": 26983
    },
    {
      "epoch": 2.5732994079589844e-05,
      "step": 26983,
      "training_step_time": 7.634342670440674
    },
    {
      "epoch": 2.573394775390625e-05,
      "model_forward_time": 1.7327346801757812,
      "step": 26984
    },
    {
      "epoch": 2.573394775390625e-05,
      "step": 26984,
      "training_step_time": 7.60151219367981
    },
    {
      "epoch": 2.5734901428222656e-05,
      "model_forward_time": 1.729130744934082,
      "step": 26985
    },
    {
      "epoch": 2.5734901428222656e-05,
      "step": 26985,
      "training_step_time": 7.68968939781189
    },
    {
      "epoch": 2.5735855102539063e-05,
      "model_forward_time": 1.7298617362976074,
      "step": 26986
    },
    {
      "epoch": 2.5735855102539063e-05,
      "step": 26986,
      "training_step_time": 7.7294299602508545
    },
    {
      "epoch": 2.5736808776855467e-05,
      "model_forward_time": 1.7604780197143555,
      "step": 26987
    },
    {
      "epoch": 2.5736808776855467e-05,
      "step": 26987,
      "training_step_time": 8.546406745910645
    },
    {
      "epoch": 2.5737762451171875e-05,
      "model_forward_time": 2.855476140975952,
      "step": 26988
    },
    {
      "epoch": 2.5737762451171875e-05,
      "step": 26988,
      "training_step_time": 7.653867959976196
    },
    {
      "epoch": 2.5738716125488282e-05,
      "model_forward_time": 1.7583942413330078,
      "step": 26989
    },
    {
      "epoch": 2.5738716125488282e-05,
      "step": 26989,
      "training_step_time": 8.537561178207397
    },
    {
      "epoch": 2.5739669799804686e-05,
      "grad_norm": 0.14837698638439178,
      "learning_rate": 8.735316695702298e-06,
      "loss": 0.0627,
      "step": 26990
    },
    {
      "epoch": 2.5739669799804686e-05,
      "model_forward_time": 1.7366671562194824,
      "step": 26990
    },
    {
      "dynamics_loss_avg": 0.05866119675338268,
      "epoch": 2.5739669799804686e-05,
      "step": 26990
    },
    {
      "action_loss_avg": 0.002119782587396912,
      "epoch": 2.5739669799804686e-05,
      "step": 26990
    },
    {
      "epoch": 2.5739669799804686e-05,
      "step": 26990,
      "training_step_time": 7.718420028686523
    },
    {
      "epoch": 2.5740623474121093e-05,
      "model_forward_time": 1.7390000820159912,
      "step": 26991
    },
    {
      "epoch": 2.5740623474121093e-05,
      "step": 26991,
      "training_step_time": 7.652922630310059
    },
    {
      "epoch": 2.57415771484375e-05,
      "model_forward_time": 1.7287592887878418,
      "step": 26992
    },
    {
      "epoch": 2.57415771484375e-05,
      "step": 26992,
      "training_step_time": 7.68206262588501
    },
    {
      "epoch": 2.5742530822753905e-05,
      "model_forward_time": 1.736781358718872,
      "step": 26993
    },
    {
      "epoch": 2.5742530822753905e-05,
      "step": 26993,
      "training_step_time": 7.684519529342651
    },
    {
      "epoch": 2.5743484497070312e-05,
      "model_forward_time": 1.738600730895996,
      "step": 26994
    },
    {
      "epoch": 2.5743484497070312e-05,
      "step": 26994,
      "training_step_time": 7.789006948471069
    },
    {
      "epoch": 2.574443817138672e-05,
      "model_forward_time": 1.7299673557281494,
      "step": 26995
    },
    {
      "epoch": 2.574443817138672e-05,
      "step": 26995,
      "training_step_time": 8.766646385192871
    },
    {
      "epoch": 2.5745391845703124e-05,
      "model_forward_time": 1.7377283573150635,
      "step": 26996
    },
    {
      "epoch": 2.5745391845703124e-05,
      "step": 26996,
      "training_step_time": 10.760032653808594
    },
    {
      "epoch": 2.574634552001953e-05,
      "model_forward_time": 2.948046922683716,
      "step": 26997
    },
    {
      "epoch": 2.574634552001953e-05,
      "step": 26997,
      "training_step_time": 8.651821851730347
    },
    {
      "epoch": 2.574729919433594e-05,
      "model_forward_time": 1.749830961227417,
      "step": 26998
    },
    {
      "epoch": 2.574729919433594e-05,
      "step": 26998,
      "training_step_time": 7.928771734237671
    },
    {
      "epoch": 2.5748252868652343e-05,
      "model_forward_time": 1.7741260528564453,
      "step": 26999
    },
    {
      "epoch": 2.5748252868652343e-05,
      "step": 26999,
      "training_step_time": 7.604187726974487
    },
    {
      "epoch": 2.574920654296875e-05,
      "grad_norm": 0.16191178560256958,
      "learning_rate": 8.734217342684768e-06,
      "loss": 0.0636,
      "step": 27000
    },
    {
      "epoch": 2.574920654296875e-05,
      "model_forward_time": 1.7691984176635742,
      "step": 27000
    },
    {
      "dynamics_loss_avg": 0.052867460809648034,
      "epoch": 2.574920654296875e-05,
      "step": 27000
    },
    {
      "action_loss_avg": 0.0033859350660350174,
      "epoch": 2.574920654296875e-05,
      "step": 27000
    },
    {
      "epoch": 2.574920654296875e-05,
      "step": 27000,
      "training_step_time": 7.858731746673584
    },
    {
      "epoch": 2.5750160217285157e-05,
      "model_forward_time": 1.7205677032470703,
      "step": 27001
    },
    {
      "epoch": 2.5750160217285157e-05,
      "step": 27001,
      "training_step_time": 9.396167755126953
    },
    {
      "epoch": 2.575111389160156e-05,
      "model_forward_time": 1.720337152481079,
      "step": 27002
    },
    {
      "epoch": 2.575111389160156e-05,
      "step": 27002,
      "training_step_time": 7.6496803760528564
    },
    {
      "epoch": 2.575206756591797e-05,
      "model_forward_time": 1.7230441570281982,
      "step": 27003
    },
    {
      "epoch": 2.575206756591797e-05,
      "step": 27003,
      "training_step_time": 8.715532541275024
    },
    {
      "epoch": 2.5753021240234376e-05,
      "model_forward_time": 1.7164766788482666,
      "step": 27004
    },
    {
      "epoch": 2.5753021240234376e-05,
      "step": 27004,
      "training_step_time": 10.813629865646362
    },
    {
      "epoch": 2.575397491455078e-05,
      "model_forward_time": 3.2719268798828125,
      "step": 27005
    },
    {
      "epoch": 2.575397491455078e-05,
      "step": 27005,
      "training_step_time": 10.734114408493042
    },
    {
      "epoch": 2.5754928588867188e-05,
      "model_forward_time": 1.864116907119751,
      "step": 27006
    },
    {
      "epoch": 2.5754928588867188e-05,
      "step": 27006,
      "training_step_time": 11.962333917617798
    },
    {
      "epoch": 2.5755882263183595e-05,
      "model_forward_time": 1.7172579765319824,
      "step": 27007
    },
    {
      "epoch": 2.5755882263183595e-05,
      "step": 27007,
      "training_step_time": 10.488494157791138
    },
    {
      "epoch": 2.57568359375e-05,
      "model_forward_time": 1.7141704559326172,
      "step": 27008
    },
    {
      "epoch": 2.57568359375e-05,
      "step": 27008,
      "training_step_time": 9.51390266418457
    },
    {
      "epoch": 2.5757789611816407e-05,
      "model_forward_time": 1.7193350791931152,
      "step": 27009
    },
    {
      "epoch": 2.5757789611816407e-05,
      "step": 27009,
      "training_step_time": 7.628952503204346
    },
    {
      "epoch": 2.5758743286132814e-05,
      "grad_norm": 0.13440975546836853,
      "learning_rate": 8.733117581298846e-06,
      "loss": 0.0604,
      "step": 27010
    },
    {
      "epoch": 2.5758743286132814e-05,
      "model_forward_time": 1.7229549884796143,
      "step": 27010
    },
    {
      "dynamics_loss_avg": 0.06388344503939151,
      "epoch": 2.5758743286132814e-05,
      "step": 27010
    },
    {
      "action_loss_avg": 0.0033195487689226867,
      "epoch": 2.5758743286132814e-05,
      "step": 27010
    },
    {
      "epoch": 2.5758743286132814e-05,
      "step": 27010,
      "training_step_time": 9.053396224975586
    },
    {
      "epoch": 2.5759696960449218e-05,
      "model_forward_time": 1.7328271865844727,
      "step": 27011
    },
    {
      "epoch": 2.5759696960449218e-05,
      "step": 27011,
      "training_step_time": 8.731801271438599
    },
    {
      "epoch": 2.5760650634765625e-05,
      "model_forward_time": 1.7321372032165527,
      "step": 27012
    },
    {
      "epoch": 2.5760650634765625e-05,
      "step": 27012,
      "training_step_time": 9.804076433181763
    },
    {
      "epoch": 2.5761604309082033e-05,
      "model_forward_time": 1.7300660610198975,
      "step": 27013
    },
    {
      "epoch": 2.5761604309082033e-05,
      "step": 27013,
      "training_step_time": 8.69159722328186
    },
    {
      "epoch": 2.5762557983398437e-05,
      "model_forward_time": 2.8058178424835205,
      "step": 27014
    },
    {
      "epoch": 2.5762557983398437e-05,
      "step": 27014,
      "training_step_time": 8.44130802154541
    },
    {
      "epoch": 2.5763511657714844e-05,
      "model_forward_time": 1.9127304553985596,
      "step": 27015
    },
    {
      "epoch": 2.5763511657714844e-05,
      "step": 27015,
      "training_step_time": 10.575289964675903
    },
    {
      "epoch": 2.576446533203125e-05,
      "model_forward_time": 1.8930950164794922,
      "step": 27016
    },
    {
      "epoch": 2.576446533203125e-05,
      "step": 27016,
      "training_step_time": 7.734895467758179
    },
    {
      "epoch": 2.5765419006347656e-05,
      "model_forward_time": 1.5028855800628662,
      "step": 27017
    },
    {
      "epoch": 2.5765419006347656e-05,
      "step": 27017,
      "training_step_time": 6.712443113327026
    },
    {
      "epoch": 2.5766372680664063e-05,
      "model_forward_time": 1.8709132671356201,
      "step": 27018
    },
    {
      "epoch": 2.5766372680664063e-05,
      "step": 27018,
      "training_step_time": 6.767801523208618
    },
    {
      "epoch": 2.5767326354980467e-05,
      "model_forward_time": 1.5327820777893066,
      "step": 27019
    },
    {
      "epoch": 2.5767326354980467e-05,
      "step": 27019,
      "training_step_time": 6.738818645477295
    },
    {
      "epoch": 2.5768280029296875e-05,
      "grad_norm": 0.19770343601703644,
      "learning_rate": 8.732017411664797e-06,
      "loss": 0.0629,
      "step": 27020
    },
    {
      "epoch": 2.5768280029296875e-05,
      "model_forward_time": 1.8882102966308594,
      "step": 27020
    },
    {
      "dynamics_loss_avg": 0.08275018837302923,
      "epoch": 2.5768280029296875e-05,
      "step": 27020
    },
    {
      "action_loss_avg": 0.0031917959277052432,
      "epoch": 2.5768280029296875e-05,
      "step": 27020
    },
    {
      "epoch": 2.5768280029296875e-05,
      "step": 27020,
      "training_step_time": 6.7688539028167725
    },
    {
      "epoch": 2.5769233703613282e-05,
      "model_forward_time": 1.8830780982971191,
      "step": 27021
    },
    {
      "epoch": 2.5769233703613282e-05,
      "step": 27021,
      "training_step_time": 6.755951881408691
    },
    {
      "epoch": 2.5770187377929686e-05,
      "model_forward_time": 1.8973069190979004,
      "step": 27022
    },
    {
      "epoch": 2.5770187377929686e-05,
      "step": 27022,
      "training_step_time": 6.727777719497681
    },
    {
      "epoch": 2.5771141052246093e-05,
      "model_forward_time": 1.8759839534759521,
      "step": 27023
    },
    {
      "epoch": 2.5771141052246093e-05,
      "step": 27023,
      "training_step_time": 6.676387071609497
    },
    {
      "epoch": 2.57720947265625e-05,
      "model_forward_time": 1.907850980758667,
      "step": 27024
    },
    {
      "epoch": 2.57720947265625e-05,
      "step": 27024,
      "training_step_time": 9.836348295211792
    },
    {
      "epoch": 2.5773048400878905e-05,
      "model_forward_time": 2.9296276569366455,
      "step": 27025
    },
    {
      "epoch": 2.5773048400878905e-05,
      "step": 27025,
      "training_step_time": 7.783307313919067
    },
    {
      "epoch": 2.5774002075195312e-05,
      "model_forward_time": 1.8716325759887695,
      "step": 27026
    },
    {
      "epoch": 2.5774002075195312e-05,
      "step": 27026,
      "training_step_time": 6.8583502769470215
    },
    {
      "epoch": 2.577495574951172e-05,
      "model_forward_time": 1.7628271579742432,
      "step": 27027
    },
    {
      "epoch": 2.577495574951172e-05,
      "step": 27027,
      "training_step_time": 6.8223936557769775
    },
    {
      "epoch": 2.5775909423828124e-05,
      "model_forward_time": 1.7296416759490967,
      "step": 27028
    },
    {
      "epoch": 2.5775909423828124e-05,
      "step": 27028,
      "training_step_time": 6.822173595428467
    },
    {
      "epoch": 2.577686309814453e-05,
      "model_forward_time": 1.7215588092803955,
      "step": 27029
    },
    {
      "epoch": 2.577686309814453e-05,
      "step": 27029,
      "training_step_time": 6.79474401473999
    },
    {
      "epoch": 2.577781677246094e-05,
      "grad_norm": 0.197677344083786,
      "learning_rate": 8.730916833902936e-06,
      "loss": 0.0634,
      "step": 27030
    },
    {
      "epoch": 2.577781677246094e-05,
      "model_forward_time": 1.7202603816986084,
      "step": 27030
    },
    {
      "dynamics_loss_avg": 0.0692346164956689,
      "epoch": 2.577781677246094e-05,
      "step": 27030
    },
    {
      "action_loss_avg": 0.005516275687841699,
      "epoch": 2.577781677246094e-05,
      "step": 27030
    },
    {
      "epoch": 2.577781677246094e-05,
      "step": 27030,
      "training_step_time": 6.8270180225372314
    },
    {
      "epoch": 2.5778770446777343e-05,
      "model_forward_time": 1.702007532119751,
      "step": 27031
    },
    {
      "epoch": 2.5778770446777343e-05,
      "step": 27031,
      "training_step_time": 6.845132350921631
    },
    {
      "epoch": 2.577972412109375e-05,
      "model_forward_time": 1.3422126770019531,
      "step": 27032
    },
    {
      "epoch": 2.577972412109375e-05,
      "step": 27032,
      "training_step_time": 6.871904373168945
    },
    {
      "epoch": 2.5780677795410157e-05,
      "model_forward_time": 1.3448681831359863,
      "step": 27033
    },
    {
      "epoch": 2.5780677795410157e-05,
      "step": 27033,
      "training_step_time": 6.814161062240601
    },
    {
      "epoch": 2.578163146972656e-05,
      "model_forward_time": 1.3561065196990967,
      "step": 27034
    },
    {
      "epoch": 2.578163146972656e-05,
      "step": 27034,
      "training_step_time": 7.767542362213135
    },
    {
      "epoch": 2.578258514404297e-05,
      "model_forward_time": 2.4632999897003174,
      "step": 27035
    },
    {
      "epoch": 2.578258514404297e-05,
      "step": 27035,
      "training_step_time": 7.22655725479126
    },
    {
      "epoch": 2.5783538818359376e-05,
      "model_forward_time": 1.712824821472168,
      "step": 27036
    },
    {
      "epoch": 2.5783538818359376e-05,
      "step": 27036,
      "training_step_time": 6.874366998672485
    },
    {
      "epoch": 2.578449249267578e-05,
      "model_forward_time": 1.7267546653747559,
      "step": 27037
    },
    {
      "epoch": 2.578449249267578e-05,
      "step": 27037,
      "training_step_time": 6.774147033691406
    },
    {
      "epoch": 2.5785446166992188e-05,
      "model_forward_time": 1.7169697284698486,
      "step": 27038
    },
    {
      "epoch": 2.5785446166992188e-05,
      "step": 27038,
      "training_step_time": 6.853008031845093
    },
    {
      "epoch": 2.5786399841308595e-05,
      "model_forward_time": 1.7362732887268066,
      "step": 27039
    },
    {
      "epoch": 2.5786399841308595e-05,
      "step": 27039,
      "training_step_time": 6.842054843902588
    },
    {
      "epoch": 2.5787353515625e-05,
      "grad_norm": 0.20417872071266174,
      "learning_rate": 8.729815848133617e-06,
      "loss": 0.0623,
      "step": 27040
    },
    {
      "epoch": 2.5787353515625e-05,
      "model_forward_time": 1.715449571609497,
      "step": 27040
    },
    {
      "dynamics_loss_avg": 0.0626811746507883,
      "epoch": 2.5787353515625e-05,
      "step": 27040
    },
    {
      "action_loss_avg": 0.004621558036888018,
      "epoch": 2.5787353515625e-05,
      "step": 27040
    },
    {
      "epoch": 2.5787353515625e-05,
      "step": 27040,
      "training_step_time": 6.872849464416504
    },
    {
      "epoch": 2.5788307189941407e-05,
      "model_forward_time": 1.7236437797546387,
      "step": 27041
    },
    {
      "epoch": 2.5788307189941407e-05,
      "step": 27041,
      "training_step_time": 6.797238826751709
    },
    {
      "epoch": 2.5789260864257814e-05,
      "model_forward_time": 1.3481378555297852,
      "step": 27042
    },
    {
      "epoch": 2.5789260864257814e-05,
      "step": 27042,
      "training_step_time": 6.8620336055755615
    },
    {
      "epoch": 2.5790214538574218e-05,
      "model_forward_time": 1.7187695503234863,
      "step": 27043
    },
    {
      "epoch": 2.5790214538574218e-05,
      "step": 27043,
      "training_step_time": 7.8287482261657715
    },
    {
      "epoch": 2.5791168212890625e-05,
      "model_forward_time": 2.9372987747192383,
      "step": 27044
    },
    {
      "epoch": 2.5791168212890625e-05,
      "step": 27044,
      "training_step_time": 7.746856689453125
    },
    {
      "epoch": 2.5792121887207033e-05,
      "model_forward_time": 1.7193357944488525,
      "step": 27045
    },
    {
      "epoch": 2.5792121887207033e-05,
      "step": 27045,
      "training_step_time": 6.813169479370117
    },
    {
      "epoch": 2.5793075561523437e-05,
      "model_forward_time": 1.721658706665039,
      "step": 27046
    },
    {
      "epoch": 2.5793075561523437e-05,
      "step": 27046,
      "training_step_time": 6.91241717338562
    },
    {
      "epoch": 2.5794029235839844e-05,
      "model_forward_time": 1.3587050437927246,
      "step": 27047
    },
    {
      "epoch": 2.5794029235839844e-05,
      "step": 27047,
      "training_step_time": 6.901014566421509
    },
    {
      "epoch": 2.579498291015625e-05,
      "model_forward_time": 1.7075417041778564,
      "step": 27048
    },
    {
      "epoch": 2.579498291015625e-05,
      "step": 27048,
      "training_step_time": 6.833940505981445
    },
    {
      "epoch": 2.5795936584472656e-05,
      "model_forward_time": 1.7503955364227295,
      "step": 27049
    },
    {
      "epoch": 2.5795936584472656e-05,
      "step": 27049,
      "training_step_time": 6.879392385482788
    },
    {
      "epoch": 2.5796890258789063e-05,
      "grad_norm": 0.16508032381534576,
      "learning_rate": 8.728714454477247e-06,
      "loss": 0.0629,
      "step": 27050
    },
    {
      "epoch": 2.5796890258789063e-05,
      "model_forward_time": 1.7209391593933105,
      "step": 27050
    },
    {
      "dynamics_loss_avg": 0.0454709317535162,
      "epoch": 2.5796890258789063e-05,
      "step": 27050
    },
    {
      "action_loss_avg": 0.001722321956185624,
      "epoch": 2.5796890258789063e-05,
      "step": 27050
    },
    {
      "epoch": 2.5796890258789063e-05,
      "step": 27050,
      "training_step_time": 6.887469291687012
    },
    {
      "epoch": 2.5797843933105467e-05,
      "model_forward_time": 1.7171707153320312,
      "step": 27051
    },
    {
      "epoch": 2.5797843933105467e-05,
      "step": 27051,
      "training_step_time": 6.84810471534729
    },
    {
      "epoch": 2.5798797607421875e-05,
      "model_forward_time": 1.7113425731658936,
      "step": 27052
    },
    {
      "epoch": 2.5798797607421875e-05,
      "step": 27052,
      "training_step_time": 9.3834810256958
    },
    {
      "epoch": 2.5799751281738282e-05,
      "model_forward_time": 2.9075655937194824,
      "step": 27053
    },
    {
      "epoch": 2.5799751281738282e-05,
      "step": 27053,
      "training_step_time": 7.712857484817505
    },
    {
      "epoch": 2.5800704956054686e-05,
      "model_forward_time": 1.7341747283935547,
      "step": 27054
    },
    {
      "epoch": 2.5800704956054686e-05,
      "step": 27054,
      "training_step_time": 6.839986324310303
    },
    {
      "epoch": 2.5801658630371093e-05,
      "model_forward_time": 1.7214715480804443,
      "step": 27055
    },
    {
      "epoch": 2.5801658630371093e-05,
      "step": 27055,
      "training_step_time": 6.9072957038879395
    },
    {
      "epoch": 2.58026123046875e-05,
      "model_forward_time": 1.715101718902588,
      "step": 27056
    },
    {
      "epoch": 2.58026123046875e-05,
      "step": 27056,
      "training_step_time": 6.8863365650177
    },
    {
      "epoch": 2.5803565979003905e-05,
      "model_forward_time": 1.7251980304718018,
      "step": 27057
    },
    {
      "epoch": 2.5803565979003905e-05,
      "step": 27057,
      "training_step_time": 6.862654685974121
    },
    {
      "epoch": 2.5804519653320312e-05,
      "model_forward_time": 1.7347254753112793,
      "step": 27058
    },
    {
      "epoch": 2.5804519653320312e-05,
      "step": 27058,
      "training_step_time": 6.8564698696136475
    },
    {
      "epoch": 2.580547332763672e-05,
      "model_forward_time": 1.710784912109375,
      "step": 27059
    },
    {
      "epoch": 2.580547332763672e-05,
      "step": 27059,
      "training_step_time": 6.883767604827881
    },
    {
      "epoch": 2.5806427001953124e-05,
      "grad_norm": 0.1960562914609909,
      "learning_rate": 8.72761265305427e-06,
      "loss": 0.0656,
      "step": 27060
    },
    {
      "epoch": 2.5806427001953124e-05,
      "model_forward_time": 1.7016491889953613,
      "step": 27060
    },
    {
      "dynamics_loss_avg": 0.06220539128407836,
      "epoch": 2.5806427001953124e-05,
      "step": 27060
    },
    {
      "action_loss_avg": 0.002883210708387196,
      "epoch": 2.5806427001953124e-05,
      "step": 27060
    },
    {
      "epoch": 2.5806427001953124e-05,
      "step": 27060,
      "training_step_time": 6.90285062789917
    },
    {
      "epoch": 2.580738067626953e-05,
      "model_forward_time": 1.710057258605957,
      "step": 27061
    },
    {
      "epoch": 2.580738067626953e-05,
      "step": 27061,
      "training_step_time": 11.35168981552124
    },
    {
      "epoch": 2.580833435058594e-05,
      "model_forward_time": 2.429063320159912,
      "step": 27062
    },
    {
      "epoch": 2.580833435058594e-05,
      "step": 27062,
      "training_step_time": 7.644928216934204
    },
    {
      "epoch": 2.5809288024902343e-05,
      "model_forward_time": 1.7185337543487549,
      "step": 27063
    },
    {
      "epoch": 2.5809288024902343e-05,
      "step": 27063,
      "training_step_time": 6.896411657333374
    },
    {
      "epoch": 2.581024169921875e-05,
      "model_forward_time": 1.7128913402557373,
      "step": 27064
    },
    {
      "epoch": 2.581024169921875e-05,
      "step": 27064,
      "training_step_time": 6.920142650604248
    },
    {
      "epoch": 2.5811195373535157e-05,
      "model_forward_time": 1.7141673564910889,
      "step": 27065
    },
    {
      "epoch": 2.5811195373535157e-05,
      "step": 27065,
      "training_step_time": 6.905587911605835
    },
    {
      "epoch": 2.581214904785156e-05,
      "model_forward_time": 1.7115757465362549,
      "step": 27066
    },
    {
      "epoch": 2.581214904785156e-05,
      "step": 27066,
      "training_step_time": 6.803913354873657
    },
    {
      "epoch": 2.581310272216797e-05,
      "model_forward_time": 1.7042176723480225,
      "step": 27067
    },
    {
      "epoch": 2.581310272216797e-05,
      "step": 27067,
      "training_step_time": 6.826693773269653
    },
    {
      "epoch": 2.5814056396484376e-05,
      "model_forward_time": 1.7108404636383057,
      "step": 27068
    },
    {
      "epoch": 2.5814056396484376e-05,
      "step": 27068,
      "training_step_time": 6.825410842895508
    },
    {
      "epoch": 2.581501007080078e-05,
      "model_forward_time": 1.3473546504974365,
      "step": 27069
    },
    {
      "epoch": 2.581501007080078e-05,
      "step": 27069,
      "training_step_time": 6.8314831256866455
    },
    {
      "epoch": 2.5815963745117188e-05,
      "grad_norm": 0.1663714349269867,
      "learning_rate": 8.726510443985176e-06,
      "loss": 0.0655,
      "step": 27070
    },
    {
      "epoch": 2.5815963745117188e-05,
      "model_forward_time": 1.7127571105957031,
      "step": 27070
    },
    {
      "dynamics_loss_avg": 0.04769946932792664,
      "epoch": 2.5815963745117188e-05,
      "step": 27070
    },
    {
      "action_loss_avg": 0.003595656336983666,
      "epoch": 2.5815963745117188e-05,
      "step": 27070
    },
    {
      "epoch": 2.5815963745117188e-05,
      "step": 27070,
      "training_step_time": 8.707986116409302
    },
    {
      "epoch": 2.5816917419433595e-05,
      "model_forward_time": 1.803778886795044,
      "step": 27071
    },
    {
      "epoch": 2.5816917419433595e-05,
      "step": 27071,
      "training_step_time": 7.762221097946167
    },
    {
      "epoch": 2.581787109375e-05,
      "model_forward_time": 2.8968214988708496,
      "step": 27072
    },
    {
      "epoch": 2.581787109375e-05,
      "step": 27072,
      "training_step_time": 7.694477558135986
    },
    {
      "epoch": 2.5818824768066407e-05,
      "model_forward_time": 1.7117934226989746,
      "step": 27073
    },
    {
      "epoch": 2.5818824768066407e-05,
      "step": 27073,
      "training_step_time": 6.899766683578491
    },
    {
      "epoch": 2.5819778442382814e-05,
      "model_forward_time": 1.7138304710388184,
      "step": 27074
    },
    {
      "epoch": 2.5819778442382814e-05,
      "step": 27074,
      "training_step_time": 6.871053218841553
    },
    {
      "epoch": 2.5820732116699218e-05,
      "model_forward_time": 1.7087507247924805,
      "step": 27075
    },
    {
      "epoch": 2.5820732116699218e-05,
      "step": 27075,
      "training_step_time": 6.811784267425537
    },
    {
      "epoch": 2.5821685791015625e-05,
      "model_forward_time": 1.7085726261138916,
      "step": 27076
    },
    {
      "epoch": 2.5821685791015625e-05,
      "step": 27076,
      "training_step_time": 6.831425666809082
    },
    {
      "epoch": 2.5822639465332033e-05,
      "model_forward_time": 1.7077224254608154,
      "step": 27077
    },
    {
      "epoch": 2.5822639465332033e-05,
      "step": 27077,
      "training_step_time": 6.836893796920776
    },
    {
      "epoch": 2.5823593139648437e-05,
      "model_forward_time": 1.7140963077545166,
      "step": 27078
    },
    {
      "epoch": 2.5823593139648437e-05,
      "step": 27078,
      "training_step_time": 6.855525732040405
    },
    {
      "epoch": 2.5824546813964844e-05,
      "model_forward_time": 1.7079124450683594,
      "step": 27079
    },
    {
      "epoch": 2.5824546813964844e-05,
      "step": 27079,
      "training_step_time": 8.668838262557983
    },
    {
      "epoch": 2.582550048828125e-05,
      "grad_norm": 0.15812183916568756,
      "learning_rate": 8.725407827390504e-06,
      "loss": 0.0624,
      "step": 27080
    },
    {
      "epoch": 2.582550048828125e-05,
      "model_forward_time": 1.7746167182922363,
      "step": 27080
    },
    {
      "dynamics_loss_avg": 0.0664119467139244,
      "epoch": 2.582550048828125e-05,
      "step": 27080
    },
    {
      "action_loss_avg": 0.002583070460241288,
      "epoch": 2.582550048828125e-05,
      "step": 27080
    },
    {
      "epoch": 2.582550048828125e-05,
      "step": 27080,
      "training_step_time": 9.918541431427002
    },
    {
      "epoch": 2.5826454162597656e-05,
      "model_forward_time": 2.854191541671753,
      "step": 27081
    },
    {
      "epoch": 2.5826454162597656e-05,
      "step": 27081,
      "training_step_time": 7.608360052108765
    },
    {
      "epoch": 2.5827407836914063e-05,
      "model_forward_time": 1.3311409950256348,
      "step": 27082
    },
    {
      "epoch": 2.5827407836914063e-05,
      "step": 27082,
      "training_step_time": 6.764561414718628
    },
    {
      "epoch": 2.5828361511230467e-05,
      "model_forward_time": 1.7056074142456055,
      "step": 27083
    },
    {
      "epoch": 2.5828361511230467e-05,
      "step": 27083,
      "training_step_time": 6.771895408630371
    },
    {
      "epoch": 2.5829315185546875e-05,
      "model_forward_time": 1.7204692363739014,
      "step": 27084
    },
    {
      "epoch": 2.5829315185546875e-05,
      "step": 27084,
      "training_step_time": 6.789174318313599
    },
    {
      "epoch": 2.5830268859863282e-05,
      "model_forward_time": 1.725348711013794,
      "step": 27085
    },
    {
      "epoch": 2.5830268859863282e-05,
      "step": 27085,
      "training_step_time": 6.803464412689209
    },
    {
      "epoch": 2.5831222534179686e-05,
      "model_forward_time": 1.7277424335479736,
      "step": 27086
    },
    {
      "epoch": 2.5831222534179686e-05,
      "step": 27086,
      "training_step_time": 6.829730749130249
    },
    {
      "epoch": 2.5832176208496094e-05,
      "model_forward_time": 1.721559762954712,
      "step": 27087
    },
    {
      "epoch": 2.5832176208496094e-05,
      "step": 27087,
      "training_step_time": 6.7595930099487305
    },
    {
      "epoch": 2.58331298828125e-05,
      "model_forward_time": 1.7276766300201416,
      "step": 27088
    },
    {
      "epoch": 2.58331298828125e-05,
      "step": 27088,
      "training_step_time": 8.509507179260254
    },
    {
      "epoch": 2.5834083557128905e-05,
      "model_forward_time": 1.3721990585327148,
      "step": 27089
    },
    {
      "epoch": 2.5834083557128905e-05,
      "step": 27089,
      "training_step_time": 7.780465841293335
    },
    {
      "epoch": 2.5835037231445312e-05,
      "grad_norm": 0.19049882888793945,
      "learning_rate": 8.724304803390833e-06,
      "loss": 0.0649,
      "step": 27090
    },
    {
      "epoch": 2.5835037231445312e-05,
      "model_forward_time": 2.8743629455566406,
      "step": 27090
    },
    {
      "dynamics_loss_avg": 0.07877439893782139,
      "epoch": 2.5835037231445312e-05,
      "step": 27090
    },
    {
      "action_loss_avg": 0.0034017295052763074,
      "epoch": 2.5835037231445312e-05,
      "step": 27090
    },
    {
      "epoch": 2.5835037231445312e-05,
      "step": 27090,
      "training_step_time": 7.681225538253784
    },
    {
      "epoch": 2.583599090576172e-05,
      "model_forward_time": 1.7284629344940186,
      "step": 27091
    },
    {
      "epoch": 2.583599090576172e-05,
      "step": 27091,
      "training_step_time": 7.1965532302856445
    },
    {
      "epoch": 2.5836944580078124e-05,
      "model_forward_time": 1.726957082748413,
      "step": 27092
    },
    {
      "epoch": 2.5836944580078124e-05,
      "step": 27092,
      "training_step_time": 6.791350841522217
    },
    {
      "epoch": 2.583789825439453e-05,
      "model_forward_time": 1.7286064624786377,
      "step": 27093
    },
    {
      "epoch": 2.583789825439453e-05,
      "step": 27093,
      "training_step_time": 6.815500974655151
    },
    {
      "epoch": 2.583885192871094e-05,
      "model_forward_time": 1.7637832164764404,
      "step": 27094
    },
    {
      "epoch": 2.583885192871094e-05,
      "step": 27094,
      "training_step_time": 6.840536117553711
    },
    {
      "epoch": 2.5839805603027343e-05,
      "model_forward_time": 1.7281455993652344,
      "step": 27095
    },
    {
      "epoch": 2.5839805603027343e-05,
      "step": 27095,
      "training_step_time": 6.835549831390381
    },
    {
      "epoch": 2.584075927734375e-05,
      "model_forward_time": 1.7236504554748535,
      "step": 27096
    },
    {
      "epoch": 2.584075927734375e-05,
      "step": 27096,
      "training_step_time": 6.797835111618042
    },
    {
      "epoch": 2.5841712951660157e-05,
      "model_forward_time": 1.7234177589416504,
      "step": 27097
    },
    {
      "epoch": 2.5841712951660157e-05,
      "step": 27097,
      "training_step_time": 7.5939013957977295
    },
    {
      "epoch": 2.584266662597656e-05,
      "model_forward_time": 1.7352697849273682,
      "step": 27098
    },
    {
      "epoch": 2.584266662597656e-05,
      "step": 27098,
      "training_step_time": 10.280489683151245
    },
    {
      "epoch": 2.584362030029297e-05,
      "model_forward_time": 1.753025770187378,
      "step": 27099
    },
    {
      "epoch": 2.584362030029297e-05,
      "step": 27099,
      "training_step_time": 9.269324779510498
    },
    {
      "epoch": 2.5844573974609376e-05,
      "grad_norm": 0.2075870931148529,
      "learning_rate": 8.723201372106789e-06,
      "loss": 0.0647,
      "step": 27100
    },
    {
      "epoch": 2.5844573974609376e-05,
      "model_forward_time": 2.9624550342559814,
      "step": 27100
    },
    {
      "dynamics_loss_avg": 0.060062171705067155,
      "epoch": 2.5844573974609376e-05,
      "step": 27100
    },
    {
      "action_loss_avg": 0.002011017507174984,
      "epoch": 2.5844573974609376e-05,
      "step": 27100
    },
    {
      "epoch": 2.5844573974609376e-05,
      "step": 27100,
      "training_step_time": 7.736230850219727
    },
    {
      "epoch": 2.584552764892578e-05,
      "model_forward_time": 1.7255797386169434,
      "step": 27101
    },
    {
      "epoch": 2.584552764892578e-05,
      "step": 27101,
      "training_step_time": 6.811081409454346
    },
    {
      "epoch": 2.5846481323242188e-05,
      "model_forward_time": 1.7267985343933105,
      "step": 27102
    },
    {
      "epoch": 2.5846481323242188e-05,
      "step": 27102,
      "training_step_time": 6.786176681518555
    },
    {
      "epoch": 2.5847434997558595e-05,
      "model_forward_time": 1.725292444229126,
      "step": 27103
    },
    {
      "epoch": 2.5847434997558595e-05,
      "step": 27103,
      "training_step_time": 6.802385330200195
    },
    {
      "epoch": 2.5848388671875e-05,
      "model_forward_time": 1.72621750831604,
      "step": 27104
    },
    {
      "epoch": 2.5848388671875e-05,
      "step": 27104,
      "training_step_time": 6.806288719177246
    },
    {
      "epoch": 2.5849342346191407e-05,
      "model_forward_time": 1.730647325515747,
      "step": 27105
    },
    {
      "epoch": 2.5849342346191407e-05,
      "step": 27105,
      "training_step_time": 6.843533754348755
    },
    {
      "epoch": 2.5850296020507814e-05,
      "model_forward_time": 1.730912208557129,
      "step": 27106
    },
    {
      "epoch": 2.5850296020507814e-05,
      "step": 27106,
      "training_step_time": 7.662299156188965
    },
    {
      "epoch": 2.5851249694824218e-05,
      "model_forward_time": 1.7216851711273193,
      "step": 27107
    },
    {
      "epoch": 2.5851249694824218e-05,
      "step": 27107,
      "training_step_time": 10.386556625366211
    },
    {
      "epoch": 2.5852203369140626e-05,
      "model_forward_time": 1.8115994930267334,
      "step": 27108
    },
    {
      "epoch": 2.5852203369140626e-05,
      "step": 27108,
      "training_step_time": 7.727456569671631
    },
    {
      "epoch": 2.5853157043457033e-05,
      "model_forward_time": 2.835380792617798,
      "step": 27109
    },
    {
      "epoch": 2.5853157043457033e-05,
      "step": 27109,
      "training_step_time": 7.700786352157593
    },
    {
      "epoch": 2.5854110717773437e-05,
      "grad_norm": 0.19094860553741455,
      "learning_rate": 8.722097533659038e-06,
      "loss": 0.0632,
      "step": 27110
    },
    {
      "epoch": 2.5854110717773437e-05,
      "model_forward_time": 1.7048888206481934,
      "step": 27110
    },
    {
      "dynamics_loss_avg": 0.05892421323806048,
      "epoch": 2.5854110717773437e-05,
      "step": 27110
    },
    {
      "action_loss_avg": 0.0035103269678074867,
      "epoch": 2.5854110717773437e-05,
      "step": 27110
    },
    {
      "epoch": 2.5854110717773437e-05,
      "step": 27110,
      "training_step_time": 6.890419960021973
    },
    {
      "epoch": 2.5855064392089844e-05,
      "model_forward_time": 1.709916591644287,
      "step": 27111
    },
    {
      "epoch": 2.5855064392089844e-05,
      "step": 27111,
      "training_step_time": 6.879817247390747
    },
    {
      "epoch": 2.585601806640625e-05,
      "model_forward_time": 1.7084579467773438,
      "step": 27112
    },
    {
      "epoch": 2.585601806640625e-05,
      "step": 27112,
      "training_step_time": 6.9348814487457275
    },
    {
      "epoch": 2.5856971740722656e-05,
      "model_forward_time": 1.7018351554870605,
      "step": 27113
    },
    {
      "epoch": 2.5856971740722656e-05,
      "step": 27113,
      "training_step_time": 6.905313014984131
    },
    {
      "epoch": 2.5857925415039063e-05,
      "model_forward_time": 1.7105519771575928,
      "step": 27114
    },
    {
      "epoch": 2.5857925415039063e-05,
      "step": 27114,
      "training_step_time": 6.919149160385132
    },
    {
      "epoch": 2.5858879089355467e-05,
      "model_forward_time": 1.3426625728607178,
      "step": 27115
    },
    {
      "epoch": 2.5858879089355467e-05,
      "step": 27115,
      "training_step_time": 7.773993253707886
    },
    {
      "epoch": 2.5859832763671875e-05,
      "model_forward_time": 1.710198163986206,
      "step": 27116
    },
    {
      "epoch": 2.5859832763671875e-05,
      "step": 27116,
      "training_step_time": 9.781696557998657
    },
    {
      "epoch": 2.5860786437988282e-05,
      "model_forward_time": 1.783635139465332,
      "step": 27117
    },
    {
      "epoch": 2.5860786437988282e-05,
      "step": 27117,
      "training_step_time": 8.473522901535034
    },
    {
      "epoch": 2.5861740112304686e-05,
      "model_forward_time": 2.856428861618042,
      "step": 27118
    },
    {
      "epoch": 2.5861740112304686e-05,
      "step": 27118,
      "training_step_time": 7.6870574951171875
    },
    {
      "epoch": 2.5862693786621094e-05,
      "model_forward_time": 1.7086501121520996,
      "step": 27119
    },
    {
      "epoch": 2.5862693786621094e-05,
      "step": 27119,
      "training_step_time": 7.34155535697937
    },
    {
      "epoch": 2.58636474609375e-05,
      "grad_norm": 0.16765722632408142,
      "learning_rate": 8.720993288168299e-06,
      "loss": 0.0661,
      "step": 27120
    },
    {
      "epoch": 2.58636474609375e-05,
      "model_forward_time": 1.7019660472869873,
      "step": 27120
    },
    {
      "dynamics_loss_avg": 0.06863425839692354,
      "epoch": 2.58636474609375e-05,
      "step": 27120
    },
    {
      "action_loss_avg": 0.0022456308593973517,
      "epoch": 2.58636474609375e-05,
      "step": 27120
    },
    {
      "epoch": 2.58636474609375e-05,
      "step": 27120,
      "training_step_time": 6.901720285415649
    },
    {
      "epoch": 2.5864601135253905e-05,
      "model_forward_time": 1.7083961963653564,
      "step": 27121
    },
    {
      "epoch": 2.5864601135253905e-05,
      "step": 27121,
      "training_step_time": 6.891310214996338
    },
    {
      "epoch": 2.5865554809570312e-05,
      "model_forward_time": 1.714463233947754,
      "step": 27122
    },
    {
      "epoch": 2.5865554809570312e-05,
      "step": 27122,
      "training_step_time": 6.8199381828308105
    },
    {
      "epoch": 2.586650848388672e-05,
      "model_forward_time": 1.7076447010040283,
      "step": 27123
    },
    {
      "epoch": 2.586650848388672e-05,
      "step": 27123,
      "training_step_time": 6.8668053150177
    },
    {
      "epoch": 2.5867462158203124e-05,
      "model_forward_time": 1.7116775512695312,
      "step": 27124
    },
    {
      "epoch": 2.5867462158203124e-05,
      "step": 27124,
      "training_step_time": 7.659351825714111
    },
    {
      "epoch": 2.586841583251953e-05,
      "model_forward_time": 1.7050840854644775,
      "step": 27125
    },
    {
      "epoch": 2.586841583251953e-05,
      "step": 27125,
      "training_step_time": 8.621638298034668
    },
    {
      "epoch": 2.586936950683594e-05,
      "model_forward_time": 1.7553026676177979,
      "step": 27126
    },
    {
      "epoch": 2.586936950683594e-05,
      "step": 27126,
      "training_step_time": 8.593348026275635
    },
    {
      "epoch": 2.5870323181152343e-05,
      "model_forward_time": 2.8496222496032715,
      "step": 27127
    },
    {
      "epoch": 2.5870323181152343e-05,
      "step": 27127,
      "training_step_time": 7.640540599822998
    },
    {
      "epoch": 2.587127685546875e-05,
      "model_forward_time": 1.7085342407226562,
      "step": 27128
    },
    {
      "epoch": 2.587127685546875e-05,
      "step": 27128,
      "training_step_time": 7.298632860183716
    },
    {
      "epoch": 2.5872230529785158e-05,
      "model_forward_time": 1.7073547840118408,
      "step": 27129
    },
    {
      "epoch": 2.5872230529785158e-05,
      "step": 27129,
      "training_step_time": 6.763439178466797
    },
    {
      "epoch": 2.587318420410156e-05,
      "grad_norm": 0.19165383279323578,
      "learning_rate": 8.719888635755327e-06,
      "loss": 0.0637,
      "step": 27130
    },
    {
      "epoch": 2.587318420410156e-05,
      "model_forward_time": 1.7085340023040771,
      "step": 27130
    },
    {
      "dynamics_loss_avg": 0.06293107913807035,
      "epoch": 2.587318420410156e-05,
      "step": 27130
    },
    {
      "action_loss_avg": 0.003825060013332404,
      "epoch": 2.587318420410156e-05,
      "step": 27130
    },
    {
      "epoch": 2.587318420410156e-05,
      "step": 27130,
      "training_step_time": 6.782384395599365
    },
    {
      "epoch": 2.587413787841797e-05,
      "model_forward_time": 1.7098579406738281,
      "step": 27131
    },
    {
      "epoch": 2.587413787841797e-05,
      "step": 27131,
      "training_step_time": 6.7633209228515625
    },
    {
      "epoch": 2.5875091552734376e-05,
      "model_forward_time": 1.7076129913330078,
      "step": 27132
    },
    {
      "epoch": 2.5875091552734376e-05,
      "step": 27132,
      "training_step_time": 6.832566976547241
    },
    {
      "epoch": 2.587604522705078e-05,
      "model_forward_time": 1.704479455947876,
      "step": 27133
    },
    {
      "epoch": 2.587604522705078e-05,
      "step": 27133,
      "training_step_time": 7.6283793449401855
    },
    {
      "epoch": 2.5876998901367188e-05,
      "model_forward_time": 1.7102537155151367,
      "step": 27134
    },
    {
      "epoch": 2.5876998901367188e-05,
      "step": 27134,
      "training_step_time": 9.676655769348145
    },
    {
      "epoch": 2.5877952575683595e-05,
      "model_forward_time": 1.7297029495239258,
      "step": 27135
    },
    {
      "epoch": 2.5877952575683595e-05,
      "step": 27135,
      "training_step_time": 8.583792448043823
    },
    {
      "epoch": 2.587890625e-05,
      "model_forward_time": 2.843770980834961,
      "step": 27136
    },
    {
      "epoch": 2.587890625e-05,
      "step": 27136,
      "training_step_time": 7.61467719078064
    },
    {
      "epoch": 2.5879859924316407e-05,
      "model_forward_time": 1.7210900783538818,
      "step": 27137
    },
    {
      "epoch": 2.5879859924316407e-05,
      "step": 27137,
      "training_step_time": 7.296599388122559
    },
    {
      "epoch": 2.5880813598632814e-05,
      "model_forward_time": 1.7240386009216309,
      "step": 27138
    },
    {
      "epoch": 2.5880813598632814e-05,
      "step": 27138,
      "training_step_time": 6.763700723648071
    },
    {
      "epoch": 2.5881767272949218e-05,
      "model_forward_time": 1.3479397296905518,
      "step": 27139
    },
    {
      "epoch": 2.5881767272949218e-05,
      "step": 27139,
      "training_step_time": 6.824892044067383
    },
    {
      "epoch": 2.5882720947265626e-05,
      "grad_norm": 0.12492873519659042,
      "learning_rate": 8.718783576540928e-06,
      "loss": 0.0641,
      "step": 27140
    },
    {
      "epoch": 2.5882720947265626e-05,
      "model_forward_time": 1.723792552947998,
      "step": 27140
    },
    {
      "dynamics_loss_avg": 0.07265150025486947,
      "epoch": 2.5882720947265626e-05,
      "step": 27140
    },
    {
      "action_loss_avg": 0.003638288576621562,
      "epoch": 2.5882720947265626e-05,
      "step": 27140
    },
    {
      "epoch": 2.5882720947265626e-05,
      "step": 27140,
      "training_step_time": 6.800096750259399
    },
    {
      "epoch": 2.588367462158203e-05,
      "model_forward_time": 1.7292044162750244,
      "step": 27141
    },
    {
      "epoch": 2.588367462158203e-05,
      "step": 27141,
      "training_step_time": 6.7807910442352295
    },
    {
      "epoch": 2.5884628295898437e-05,
      "model_forward_time": 1.7249906063079834,
      "step": 27142
    },
    {
      "epoch": 2.5884628295898437e-05,
      "step": 27142,
      "training_step_time": 7.621301651000977
    },
    {
      "epoch": 2.5885581970214844e-05,
      "model_forward_time": 1.7258141040802002,
      "step": 27143
    },
    {
      "epoch": 2.5885581970214844e-05,
      "step": 27143,
      "training_step_time": 7.574769735336304
    },
    {
      "epoch": 2.588653564453125e-05,
      "model_forward_time": 1.7210760116577148,
      "step": 27144
    },
    {
      "epoch": 2.588653564453125e-05,
      "step": 27144,
      "training_step_time": 7.75697135925293
    },
    {
      "epoch": 2.5887489318847656e-05,
      "model_forward_time": 1.8253605365753174,
      "step": 27145
    },
    {
      "epoch": 2.5887489318847656e-05,
      "step": 27145,
      "training_step_time": 7.673588037490845
    },
    {
      "epoch": 2.5888442993164063e-05,
      "model_forward_time": 2.1457338333129883,
      "step": 27146
    },
    {
      "epoch": 2.5888442993164063e-05,
      "step": 27146,
      "training_step_time": 7.521506309509277
    },
    {
      "epoch": 2.5889396667480467e-05,
      "model_forward_time": 1.7232012748718262,
      "step": 27147
    },
    {
      "epoch": 2.5889396667480467e-05,
      "step": 27147,
      "training_step_time": 6.847071170806885
    },
    {
      "epoch": 2.5890350341796875e-05,
      "model_forward_time": 1.708486557006836,
      "step": 27148
    },
    {
      "epoch": 2.5890350341796875e-05,
      "step": 27148,
      "training_step_time": 6.803901433944702
    },
    {
      "epoch": 2.5891304016113282e-05,
      "model_forward_time": 1.7085020542144775,
      "step": 27149
    },
    {
      "epoch": 2.5891304016113282e-05,
      "step": 27149,
      "training_step_time": 6.816231966018677
    },
    {
      "epoch": 2.5892257690429686e-05,
      "grad_norm": 0.1574796587228775,
      "learning_rate": 8.717678110645948e-06,
      "loss": 0.0637,
      "step": 27150
    },
    {
      "epoch": 2.5892257690429686e-05,
      "model_forward_time": 1.708883285522461,
      "step": 27150
    },
    {
      "dynamics_loss_avg": 0.05174504742026329,
      "epoch": 2.5892257690429686e-05,
      "step": 27150
    },
    {
      "action_loss_avg": 0.0018987160758115352,
      "epoch": 2.5892257690429686e-05,
      "step": 27150
    },
    {
      "epoch": 2.5892257690429686e-05,
      "step": 27150,
      "training_step_time": 6.786969423294067
    },
    {
      "epoch": 2.5893211364746094e-05,
      "model_forward_time": 1.7096707820892334,
      "step": 27151
    },
    {
      "epoch": 2.5893211364746094e-05,
      "step": 27151,
      "training_step_time": 7.590389251708984
    },
    {
      "epoch": 2.58941650390625e-05,
      "model_forward_time": 1.7135531902313232,
      "step": 27152
    },
    {
      "epoch": 2.58941650390625e-05,
      "step": 27152,
      "training_step_time": 8.608075618743896
    },
    {
      "epoch": 2.5895118713378905e-05,
      "model_forward_time": 1.7103374004364014,
      "step": 27153
    },
    {
      "epoch": 2.5895118713378905e-05,
      "step": 27153,
      "training_step_time": 9.922447443008423
    },
    {
      "epoch": 2.5896072387695312e-05,
      "model_forward_time": 1.7973449230194092,
      "step": 27154
    },
    {
      "epoch": 2.5896072387695312e-05,
      "step": 27154,
      "training_step_time": 7.675693511962891
    },
    {
      "epoch": 2.589702606201172e-05,
      "model_forward_time": 2.8426027297973633,
      "step": 27155
    },
    {
      "epoch": 2.589702606201172e-05,
      "step": 27155,
      "training_step_time": 7.5907769203186035
    },
    {
      "epoch": 2.5897979736328124e-05,
      "model_forward_time": 1.7140343189239502,
      "step": 27156
    },
    {
      "epoch": 2.5897979736328124e-05,
      "step": 27156,
      "training_step_time": 6.799879550933838
    },
    {
      "epoch": 2.589893341064453e-05,
      "model_forward_time": 1.7100579738616943,
      "step": 27157
    },
    {
      "epoch": 2.589893341064453e-05,
      "step": 27157,
      "training_step_time": 6.7956554889678955
    },
    {
      "epoch": 2.589988708496094e-05,
      "model_forward_time": 1.7151334285736084,
      "step": 27158
    },
    {
      "epoch": 2.589988708496094e-05,
      "step": 27158,
      "training_step_time": 6.80581259727478
    },
    {
      "epoch": 2.5900840759277343e-05,
      "model_forward_time": 1.715639352798462,
      "step": 27159
    },
    {
      "epoch": 2.5900840759277343e-05,
      "step": 27159,
      "training_step_time": 6.827441215515137
    },
    {
      "epoch": 2.590179443359375e-05,
      "grad_norm": 0.15073366463184357,
      "learning_rate": 8.71657223819128e-06,
      "loss": 0.0668,
      "step": 27160
    },
    {
      "epoch": 2.590179443359375e-05,
      "model_forward_time": 1.7196123600006104,
      "step": 27160
    },
    {
      "dynamics_loss_avg": 0.0502005584537983,
      "epoch": 2.590179443359375e-05,
      "step": 27160
    },
    {
      "action_loss_avg": 0.002745262096868828,
      "epoch": 2.590179443359375e-05,
      "step": 27160
    },
    {
      "epoch": 2.590179443359375e-05,
      "step": 27160,
      "training_step_time": 7.600029706954956
    },
    {
      "epoch": 2.5902748107910158e-05,
      "model_forward_time": 1.7089498043060303,
      "step": 27161
    },
    {
      "epoch": 2.5902748107910158e-05,
      "step": 27161,
      "training_step_time": 7.524138689041138
    },
    {
      "epoch": 2.590370178222656e-05,
      "model_forward_time": 1.7079551219940186,
      "step": 27162
    },
    {
      "epoch": 2.590370178222656e-05,
      "step": 27162,
      "training_step_time": 10.789320945739746
    },
    {
      "epoch": 2.590465545654297e-05,
      "model_forward_time": 1.375223159790039,
      "step": 27163
    },
    {
      "epoch": 2.590465545654297e-05,
      "step": 27163,
      "training_step_time": 8.543878078460693
    },
    {
      "epoch": 2.5905609130859376e-05,
      "model_forward_time": 1.3419930934906006,
      "step": 27164
    },
    {
      "epoch": 2.5905609130859376e-05,
      "step": 27164,
      "training_step_time": 7.5351033210754395
    },
    {
      "epoch": 2.590656280517578e-05,
      "model_forward_time": 2.904629945755005,
      "step": 27165
    },
    {
      "epoch": 2.590656280517578e-05,
      "step": 27165,
      "training_step_time": 7.620463848114014
    },
    {
      "epoch": 2.5907516479492188e-05,
      "model_forward_time": 1.7224206924438477,
      "step": 27166
    },
    {
      "epoch": 2.5907516479492188e-05,
      "step": 27166,
      "training_step_time": 6.767805099487305
    },
    {
      "epoch": 2.5908470153808595e-05,
      "model_forward_time": 1.7111492156982422,
      "step": 27167
    },
    {
      "epoch": 2.5908470153808595e-05,
      "step": 27167,
      "training_step_time": 6.773727655410767
    },
    {
      "epoch": 2.5909423828125e-05,
      "model_forward_time": 1.704637050628662,
      "step": 27168
    },
    {
      "epoch": 2.5909423828125e-05,
      "step": 27168,
      "training_step_time": 6.836073160171509
    },
    {
      "epoch": 2.5910377502441407e-05,
      "model_forward_time": 1.7068507671356201,
      "step": 27169
    },
    {
      "epoch": 2.5910377502441407e-05,
      "step": 27169,
      "training_step_time": 7.518432140350342
    },
    {
      "epoch": 2.5911331176757814e-05,
      "grad_norm": 0.10747520625591278,
      "learning_rate": 8.715465959297858e-06,
      "loss": 0.0642,
      "step": 27170
    },
    {
      "epoch": 2.5911331176757814e-05,
      "model_forward_time": 1.6952359676361084,
      "step": 27170
    },
    {
      "dynamics_loss_avg": 0.053987585753202436,
      "epoch": 2.5911331176757814e-05,
      "step": 27170
    },
    {
      "action_loss_avg": 0.0025900213629938664,
      "epoch": 2.5911331176757814e-05,
      "step": 27170
    },
    {
      "epoch": 2.5911331176757814e-05,
      "step": 27170,
      "training_step_time": 8.550982475280762
    },
    {
      "epoch": 2.5912284851074218e-05,
      "model_forward_time": 1.706268310546875,
      "step": 27171
    },
    {
      "epoch": 2.5912284851074218e-05,
      "step": 27171,
      "training_step_time": 9.694964170455933
    },
    {
      "epoch": 2.5913238525390626e-05,
      "model_forward_time": 0.999772310256958,
      "step": 27172
    },
    {
      "epoch": 2.5913238525390626e-05,
      "step": 27172,
      "training_step_time": 9.458632230758667
    },
    {
      "epoch": 2.591419219970703e-05,
      "model_forward_time": 1.7147390842437744,
      "step": 27173
    },
    {
      "epoch": 2.591419219970703e-05,
      "step": 27173,
      "training_step_time": 7.572220325469971
    },
    {
      "epoch": 2.5915145874023437e-05,
      "model_forward_time": 2.8376989364624023,
      "step": 27174
    },
    {
      "epoch": 2.5915145874023437e-05,
      "step": 27174,
      "training_step_time": 7.59822416305542
    },
    {
      "epoch": 2.5916099548339844e-05,
      "model_forward_time": 1.7038118839263916,
      "step": 27175
    },
    {
      "epoch": 2.5916099548339844e-05,
      "step": 27175,
      "training_step_time": 6.7736780643463135
    },
    {
      "epoch": 2.591705322265625e-05,
      "model_forward_time": 1.7087323665618896,
      "step": 27176
    },
    {
      "epoch": 2.591705322265625e-05,
      "step": 27176,
      "training_step_time": 6.776416301727295
    },
    {
      "epoch": 2.5918006896972656e-05,
      "model_forward_time": 1.7063767910003662,
      "step": 27177
    },
    {
      "epoch": 2.5918006896972656e-05,
      "step": 27177,
      "training_step_time": 6.833104372024536
    },
    {
      "epoch": 2.5918960571289063e-05,
      "model_forward_time": 1.7096576690673828,
      "step": 27178
    },
    {
      "epoch": 2.5918960571289063e-05,
      "step": 27178,
      "training_step_time": 6.84009313583374
    },
    {
      "epoch": 2.5919914245605467e-05,
      "model_forward_time": 1.707794427871704,
      "step": 27179
    },
    {
      "epoch": 2.5919914245605467e-05,
      "step": 27179,
      "training_step_time": 8.645330667495728
    },
    {
      "epoch": 2.5920867919921875e-05,
      "grad_norm": 0.1907981038093567,
      "learning_rate": 8.714359274086665e-06,
      "loss": 0.0645,
      "step": 27180
    },
    {
      "epoch": 2.5920867919921875e-05,
      "model_forward_time": 1.7142562866210938,
      "step": 27180
    },
    {
      "dynamics_loss_avg": 0.06145771704614163,
      "epoch": 2.5920867919921875e-05,
      "step": 27180
    },
    {
      "action_loss_avg": 0.0020731189579237252,
      "epoch": 2.5920867919921875e-05,
      "step": 27180
    },
    {
      "epoch": 2.5920867919921875e-05,
      "step": 27180,
      "training_step_time": 8.155550003051758
    },
    {
      "epoch": 2.5921821594238282e-05,
      "model_forward_time": 1.7196681499481201,
      "step": 27181
    },
    {
      "epoch": 2.5921821594238282e-05,
      "step": 27181,
      "training_step_time": 9.809403657913208
    },
    {
      "epoch": 2.5922775268554686e-05,
      "model_forward_time": 1.7251088619232178,
      "step": 27182
    },
    {
      "epoch": 2.5922775268554686e-05,
      "step": 27182,
      "training_step_time": 8.400045156478882
    },
    {
      "epoch": 2.5923728942871094e-05,
      "model_forward_time": 2.840749502182007,
      "step": 27183
    },
    {
      "epoch": 2.5923728942871094e-05,
      "step": 27183,
      "training_step_time": 7.616870164871216
    },
    {
      "epoch": 2.59246826171875e-05,
      "model_forward_time": 1.707115888595581,
      "step": 27184
    },
    {
      "epoch": 2.59246826171875e-05,
      "step": 27184,
      "training_step_time": 6.849409580230713
    },
    {
      "epoch": 2.5925636291503905e-05,
      "model_forward_time": 1.7096712589263916,
      "step": 27185
    },
    {
      "epoch": 2.5925636291503905e-05,
      "step": 27185,
      "training_step_time": 6.828283786773682
    },
    {
      "epoch": 2.5926589965820312e-05,
      "model_forward_time": 1.7163515090942383,
      "step": 27186
    },
    {
      "epoch": 2.5926589965820312e-05,
      "step": 27186,
      "training_step_time": 6.8629419803619385
    },
    {
      "epoch": 2.592754364013672e-05,
      "model_forward_time": 1.3444457054138184,
      "step": 27187
    },
    {
      "epoch": 2.592754364013672e-05,
      "step": 27187,
      "training_step_time": 6.9590678215026855
    },
    {
      "epoch": 2.5928497314453124e-05,
      "model_forward_time": 1.7221622467041016,
      "step": 27188
    },
    {
      "epoch": 2.5928497314453124e-05,
      "step": 27188,
      "training_step_time": 7.713745594024658
    },
    {
      "epoch": 2.592945098876953e-05,
      "model_forward_time": 1.7208843231201172,
      "step": 27189
    },
    {
      "epoch": 2.592945098876953e-05,
      "step": 27189,
      "training_step_time": 8.817070007324219
    },
    {
      "epoch": 2.593040466308594e-05,
      "grad_norm": 0.1379684954881668,
      "learning_rate": 8.713252182678727e-06,
      "loss": 0.0648,
      "step": 27190
    },
    {
      "epoch": 2.593040466308594e-05,
      "model_forward_time": 1.725104808807373,
      "step": 27190
    },
    {
      "dynamics_loss_avg": 0.06952359303832054,
      "epoch": 2.593040466308594e-05,
      "step": 27190
    },
    {
      "action_loss_avg": 0.0031644987175241113,
      "epoch": 2.593040466308594e-05,
      "step": 27190
    },
    {
      "epoch": 2.593040466308594e-05,
      "step": 27190,
      "training_step_time": 9.99367070198059
    },
    {
      "epoch": 2.5931358337402343e-05,
      "model_forward_time": 1.7337632179260254,
      "step": 27191
    },
    {
      "epoch": 2.5931358337402343e-05,
      "step": 27191,
      "training_step_time": 8.37463903427124
    },
    {
      "epoch": 2.593231201171875e-05,
      "model_forward_time": 1.7537977695465088,
      "step": 27192
    },
    {
      "epoch": 2.593231201171875e-05,
      "step": 27192,
      "training_step_time": 7.659818649291992
    },
    {
      "epoch": 2.5933265686035158e-05,
      "model_forward_time": 2.2697994709014893,
      "step": 27193
    },
    {
      "epoch": 2.5933265686035158e-05,
      "step": 27193,
      "training_step_time": 7.146142244338989
    },
    {
      "epoch": 2.593421936035156e-05,
      "model_forward_time": 1.725379228591919,
      "step": 27194
    },
    {
      "epoch": 2.593421936035156e-05,
      "step": 27194,
      "training_step_time": 6.944140672683716
    },
    {
      "epoch": 2.593517303466797e-05,
      "model_forward_time": 1.7475202083587646,
      "step": 27195
    },
    {
      "epoch": 2.593517303466797e-05,
      "step": 27195,
      "training_step_time": 6.942912578582764
    },
    {
      "epoch": 2.5936126708984376e-05,
      "model_forward_time": 1.7387831211090088,
      "step": 27196
    },
    {
      "epoch": 2.5936126708984376e-05,
      "step": 27196,
      "training_step_time": 6.963945627212524
    },
    {
      "epoch": 2.593708038330078e-05,
      "model_forward_time": 1.7307116985321045,
      "step": 27197
    },
    {
      "epoch": 2.593708038330078e-05,
      "step": 27197,
      "training_step_time": 8.748577117919922
    },
    {
      "epoch": 2.5938034057617188e-05,
      "model_forward_time": 1.740847110748291,
      "step": 27198
    },
    {
      "epoch": 2.5938034057617188e-05,
      "step": 27198,
      "training_step_time": 7.786230087280273
    },
    {
      "epoch": 2.5938987731933595e-05,
      "model_forward_time": 1.7091338634490967,
      "step": 27199
    },
    {
      "epoch": 2.5938987731933595e-05,
      "step": 27199,
      "training_step_time": 9.623099088668823
    },
    {
      "epoch": 2.593994140625e-05,
      "grad_norm": 0.1318991333246231,
      "learning_rate": 8.712144685195112e-06,
      "loss": 0.0609,
      "step": 27200
    },
    {
      "epoch": 2.593994140625e-05,
      "model_forward_time": 1.7182965278625488,
      "step": 27200
    },
    {
      "dynamics_loss_avg": 0.054697268083691596,
      "epoch": 2.593994140625e-05,
      "step": 27200
    },
    {
      "action_loss_avg": 0.0031372412107884885,
      "epoch": 2.593994140625e-05,
      "step": 27200
    },
    {
      "epoch": 2.593994140625e-05,
      "step": 27200,
      "training_step_time": 8.478200435638428
    },
    {
      "epoch": 2.5940895080566407e-05,
      "model_forward_time": 1.7286159992218018,
      "step": 27201
    },
    {
      "epoch": 2.5940895080566407e-05,
      "step": 27201,
      "training_step_time": 8.519004106521606
    },
    {
      "epoch": 2.5941848754882814e-05,
      "model_forward_time": 2.31531023979187,
      "step": 27202
    },
    {
      "epoch": 2.5941848754882814e-05,
      "step": 27202,
      "training_step_time": 7.33874773979187
    },
    {
      "epoch": 2.5942802429199218e-05,
      "model_forward_time": 1.7374627590179443,
      "step": 27203
    },
    {
      "epoch": 2.5942802429199218e-05,
      "step": 27203,
      "training_step_time": 6.96966552734375
    },
    {
      "epoch": 2.5943756103515626e-05,
      "model_forward_time": 1.7427215576171875,
      "step": 27204
    },
    {
      "epoch": 2.5943756103515626e-05,
      "step": 27204,
      "training_step_time": 6.901081323623657
    },
    {
      "epoch": 2.594470977783203e-05,
      "model_forward_time": 1.7459077835083008,
      "step": 27205
    },
    {
      "epoch": 2.594470977783203e-05,
      "step": 27205,
      "training_step_time": 6.892039060592651
    },
    {
      "epoch": 2.5945663452148437e-05,
      "model_forward_time": 1.735001564025879,
      "step": 27206
    },
    {
      "epoch": 2.5945663452148437e-05,
      "step": 27206,
      "training_step_time": 7.696648597717285
    },
    {
      "epoch": 2.5946617126464844e-05,
      "model_forward_time": 1.7402629852294922,
      "step": 27207
    },
    {
      "epoch": 2.5946617126464844e-05,
      "step": 27207,
      "training_step_time": 8.256048917770386
    },
    {
      "epoch": 2.594757080078125e-05,
      "model_forward_time": 1.7408936023712158,
      "step": 27208
    },
    {
      "epoch": 2.594757080078125e-05,
      "step": 27208,
      "training_step_time": 10.67443060874939
    },
    {
      "epoch": 2.5948524475097656e-05,
      "model_forward_time": 1.7335388660430908,
      "step": 27209
    },
    {
      "epoch": 2.5948524475097656e-05,
      "step": 27209,
      "training_step_time": 7.684084892272949
    },
    {
      "epoch": 2.5949478149414063e-05,
      "grad_norm": 0.15909072756767273,
      "learning_rate": 8.711036781756936e-06,
      "loss": 0.0596,
      "step": 27210
    },
    {
      "epoch": 2.5949478149414063e-05,
      "model_forward_time": 1.726442813873291,
      "step": 27210
    },
    {
      "dynamics_loss_avg": 0.06326148509979249,
      "epoch": 2.5949478149414063e-05,
      "step": 27210
    },
    {
      "action_loss_avg": 0.00290350922732614,
      "epoch": 2.5949478149414063e-05,
      "step": 27210
    },
    {
      "epoch": 2.5949478149414063e-05,
      "step": 27210,
      "training_step_time": 8.934523105621338
    },
    {
      "epoch": 2.5950431823730467e-05,
      "model_forward_time": 2.3032748699188232,
      "step": 27211
    },
    {
      "epoch": 2.5950431823730467e-05,
      "step": 27211,
      "training_step_time": 7.322988271713257
    },
    {
      "epoch": 2.5951385498046875e-05,
      "model_forward_time": 1.7507507801055908,
      "step": 27212
    },
    {
      "epoch": 2.5951385498046875e-05,
      "step": 27212,
      "training_step_time": 6.914011001586914
    },
    {
      "epoch": 2.5952339172363282e-05,
      "model_forward_time": 1.744877815246582,
      "step": 27213
    },
    {
      "epoch": 2.5952339172363282e-05,
      "step": 27213,
      "training_step_time": 6.923456907272339
    },
    {
      "epoch": 2.5953292846679686e-05,
      "model_forward_time": 1.7389843463897705,
      "step": 27214
    },
    {
      "epoch": 2.5953292846679686e-05,
      "step": 27214,
      "training_step_time": 6.9117138385772705
    },
    {
      "epoch": 2.5954246520996094e-05,
      "model_forward_time": 1.7434420585632324,
      "step": 27215
    },
    {
      "epoch": 2.5954246520996094e-05,
      "step": 27215,
      "training_step_time": 7.630134582519531
    },
    {
      "epoch": 2.59552001953125e-05,
      "model_forward_time": 1.7396368980407715,
      "step": 27216
    },
    {
      "epoch": 2.59552001953125e-05,
      "step": 27216,
      "training_step_time": 7.683416128158569
    },
    {
      "epoch": 2.5956153869628905e-05,
      "model_forward_time": 1.7324178218841553,
      "step": 27217
    },
    {
      "epoch": 2.5956153869628905e-05,
      "step": 27217,
      "training_step_time": 9.61214280128479
    },
    {
      "epoch": 2.5957107543945312e-05,
      "model_forward_time": 1.7072360515594482,
      "step": 27218
    },
    {
      "epoch": 2.5957107543945312e-05,
      "step": 27218,
      "training_step_time": 8.711242198944092
    },
    {
      "epoch": 2.595806121826172e-05,
      "model_forward_time": 1.7259397506713867,
      "step": 27219
    },
    {
      "epoch": 2.595806121826172e-05,
      "step": 27219,
      "training_step_time": 8.256283044815063
    },
    {
      "epoch": 2.5959014892578124e-05,
      "grad_norm": 0.11546111106872559,
      "learning_rate": 8.709928472485357e-06,
      "loss": 0.058,
      "step": 27220
    },
    {
      "epoch": 2.5959014892578124e-05,
      "model_forward_time": 1.9356441497802734,
      "step": 27220
    },
    {
      "dynamics_loss_avg": 0.0748192884027958,
      "epoch": 2.5959014892578124e-05,
      "step": 27220
    },
    {
      "action_loss_avg": 0.003941562108229846,
      "epoch": 2.5959014892578124e-05,
      "step": 27220
    },
    {
      "epoch": 2.5959014892578124e-05,
      "step": 27220,
      "training_step_time": 7.325239181518555
    },
    {
      "epoch": 2.595996856689453e-05,
      "model_forward_time": 1.742222547531128,
      "step": 27221
    },
    {
      "epoch": 2.595996856689453e-05,
      "step": 27221,
      "training_step_time": 6.908117294311523
    },
    {
      "epoch": 2.596092224121094e-05,
      "model_forward_time": 1.7352159023284912,
      "step": 27222
    },
    {
      "epoch": 2.596092224121094e-05,
      "step": 27222,
      "training_step_time": 6.900192022323608
    },
    {
      "epoch": 2.5961875915527343e-05,
      "model_forward_time": 1.73677659034729,
      "step": 27223
    },
    {
      "epoch": 2.5961875915527343e-05,
      "step": 27223,
      "training_step_time": 6.879579305648804
    },
    {
      "epoch": 2.596282958984375e-05,
      "model_forward_time": 1.7360596656799316,
      "step": 27224
    },
    {
      "epoch": 2.596282958984375e-05,
      "step": 27224,
      "training_step_time": 7.986144781112671
    },
    {
      "epoch": 2.5963783264160158e-05,
      "model_forward_time": 1.3772554397583008,
      "step": 27225
    },
    {
      "epoch": 2.5963783264160158e-05,
      "step": 27225,
      "training_step_time": 7.705183029174805
    },
    {
      "epoch": 2.596473693847656e-05,
      "model_forward_time": 1.3622286319732666,
      "step": 27226
    },
    {
      "epoch": 2.596473693847656e-05,
      "step": 27226,
      "training_step_time": 9.911895513534546
    },
    {
      "epoch": 2.596569061279297e-05,
      "model_forward_time": 1.7324845790863037,
      "step": 27227
    },
    {
      "epoch": 2.596569061279297e-05,
      "step": 27227,
      "training_step_time": 8.531235933303833
    },
    {
      "epoch": 2.5966644287109376e-05,
      "model_forward_time": 1.040452003479004,
      "step": 27228
    },
    {
      "epoch": 2.5966644287109376e-05,
      "step": 27228,
      "training_step_time": 10.45741319656372
    },
    {
      "epoch": 2.596759796142578e-05,
      "model_forward_time": 1.761498212814331,
      "step": 27229
    },
    {
      "epoch": 2.596759796142578e-05,
      "step": 27229,
      "training_step_time": 7.604563236236572
    },
    {
      "epoch": 2.5968551635742188e-05,
      "grad_norm": 0.1318897157907486,
      "learning_rate": 8.708819757501579e-06,
      "loss": 0.0581,
      "step": 27230
    },
    {
      "epoch": 2.5968551635742188e-05,
      "model_forward_time": 2.2855963706970215,
      "step": 27230
    },
    {
      "dynamics_loss_avg": 0.07061147335916758,
      "epoch": 2.5968551635742188e-05,
      "step": 27230
    },
    {
      "action_loss_avg": 0.0032868333626538517,
      "epoch": 2.5968551635742188e-05,
      "step": 27230
    },
    {
      "epoch": 2.5968551635742188e-05,
      "step": 27230,
      "training_step_time": 7.078149080276489
    },
    {
      "epoch": 2.5969505310058595e-05,
      "model_forward_time": 1.7349975109100342,
      "step": 27231
    },
    {
      "epoch": 2.5969505310058595e-05,
      "step": 27231,
      "training_step_time": 6.819153070449829
    },
    {
      "epoch": 2.5970458984375e-05,
      "model_forward_time": 1.7390820980072021,
      "step": 27232
    },
    {
      "epoch": 2.5970458984375e-05,
      "step": 27232,
      "training_step_time": 6.855851411819458
    },
    {
      "epoch": 2.5971412658691407e-05,
      "model_forward_time": 1.7381162643432617,
      "step": 27233
    },
    {
      "epoch": 2.5971412658691407e-05,
      "step": 27233,
      "training_step_time": 7.4074387550354
    },
    {
      "epoch": 2.5972366333007814e-05,
      "model_forward_time": 1.7366187572479248,
      "step": 27234
    },
    {
      "epoch": 2.5972366333007814e-05,
      "step": 27234,
      "training_step_time": 8.704864263534546
    },
    {
      "epoch": 2.5973320007324218e-05,
      "model_forward_time": 1.3749494552612305,
      "step": 27235
    },
    {
      "epoch": 2.5973320007324218e-05,
      "step": 27235,
      "training_step_time": 8.447812795639038
    },
    {
      "epoch": 2.5974273681640626e-05,
      "model_forward_time": 1.7372636795043945,
      "step": 27236
    },
    {
      "epoch": 2.5974273681640626e-05,
      "step": 27236,
      "training_step_time": 7.802679777145386
    },
    {
      "epoch": 2.597522735595703e-05,
      "model_forward_time": 1.731738567352295,
      "step": 27237
    },
    {
      "epoch": 2.597522735595703e-05,
      "step": 27237,
      "training_step_time": 8.212681770324707
    },
    {
      "epoch": 2.5976181030273437e-05,
      "model_forward_time": 1.7429721355438232,
      "step": 27238
    },
    {
      "epoch": 2.5976181030273437e-05,
      "step": 27238,
      "training_step_time": 7.608882427215576
    },
    {
      "epoch": 2.5977134704589844e-05,
      "model_forward_time": 1.739903450012207,
      "step": 27239
    },
    {
      "epoch": 2.5977134704589844e-05,
      "step": 27239,
      "training_step_time": 8.349029541015625
    },
    {
      "epoch": 2.597808837890625e-05,
      "grad_norm": 0.15148477256298065,
      "learning_rate": 8.707710636926847e-06,
      "loss": 0.0567,
      "step": 27240
    },
    {
      "epoch": 2.597808837890625e-05,
      "model_forward_time": 1.7404465675354004,
      "step": 27240
    },
    {
      "dynamics_loss_avg": 0.05386130250990391,
      "epoch": 2.597808837890625e-05,
      "step": 27240
    },
    {
      "action_loss_avg": 0.0015443141397554427,
      "epoch": 2.597808837890625e-05,
      "step": 27240
    },
    {
      "epoch": 2.597808837890625e-05,
      "step": 27240,
      "training_step_time": 6.890058517456055
    },
    {
      "epoch": 2.5979042053222656e-05,
      "model_forward_time": 1.7192792892456055,
      "step": 27241
    },
    {
      "epoch": 2.5979042053222656e-05,
      "step": 27241,
      "training_step_time": 6.890635967254639
    },
    {
      "epoch": 2.5979995727539063e-05,
      "model_forward_time": 1.742119550704956,
      "step": 27242
    },
    {
      "epoch": 2.5979995727539063e-05,
      "step": 27242,
      "training_step_time": 7.446588516235352
    },
    {
      "epoch": 2.5980949401855467e-05,
      "model_forward_time": 1.7410309314727783,
      "step": 27243
    },
    {
      "epoch": 2.5980949401855467e-05,
      "step": 27243,
      "training_step_time": 7.712245464324951
    },
    {
      "epoch": 2.5981903076171875e-05,
      "model_forward_time": 1.7424886226654053,
      "step": 27244
    },
    {
      "epoch": 2.5981903076171875e-05,
      "step": 27244,
      "training_step_time": 8.68914532661438
    },
    {
      "epoch": 2.5982856750488282e-05,
      "model_forward_time": 1.7452163696289062,
      "step": 27245
    },
    {
      "epoch": 2.5982856750488282e-05,
      "step": 27245,
      "training_step_time": 9.345086336135864
    },
    {
      "epoch": 2.5983810424804686e-05,
      "model_forward_time": 1.7454490661621094,
      "step": 27246
    },
    {
      "epoch": 2.5983810424804686e-05,
      "step": 27246,
      "training_step_time": 9.57478642463684
    },
    {
      "epoch": 2.5984764099121094e-05,
      "model_forward_time": 1.7411408424377441,
      "step": 27247
    },
    {
      "epoch": 2.5984764099121094e-05,
      "step": 27247,
      "training_step_time": 7.783022165298462
    },
    {
      "epoch": 2.59857177734375e-05,
      "model_forward_time": 1.7966718673706055,
      "step": 27248
    },
    {
      "epoch": 2.59857177734375e-05,
      "step": 27248,
      "training_step_time": 7.3639976978302
    },
    {
      "epoch": 2.5986671447753905e-05,
      "model_forward_time": 2.2832789421081543,
      "step": 27249
    },
    {
      "epoch": 2.5986671447753905e-05,
      "step": 27249,
      "training_step_time": 7.163650751113892
    },
    {
      "epoch": 2.5987625122070312e-05,
      "grad_norm": 0.14253707230091095,
      "learning_rate": 8.706601110882456e-06,
      "loss": 0.0588,
      "step": 27250
    },
    {
      "epoch": 2.5987625122070312e-05,
      "model_forward_time": 1.728208303451538,
      "step": 27250
    },
    {
      "dynamics_loss_avg": 0.05610227659344673,
      "epoch": 2.5987625122070312e-05,
      "step": 27250
    },
    {
      "action_loss_avg": 0.0027754724840633573,
      "epoch": 2.5987625122070312e-05,
      "step": 27250
    },
    {
      "epoch": 2.5987625122070312e-05,
      "step": 27250,
      "training_step_time": 6.959144115447998
    },
    {
      "epoch": 2.598857879638672e-05,
      "model_forward_time": 1.3718147277832031,
      "step": 27251
    },
    {
      "epoch": 2.598857879638672e-05,
      "step": 27251,
      "training_step_time": 7.4808714389801025
    },
    {
      "epoch": 2.5989532470703124e-05,
      "model_forward_time": 1.7234489917755127,
      "step": 27252
    },
    {
      "epoch": 2.5989532470703124e-05,
      "step": 27252,
      "training_step_time": 7.732324600219727
    },
    {
      "epoch": 2.599048614501953e-05,
      "model_forward_time": 1.7236480712890625,
      "step": 27253
    },
    {
      "epoch": 2.599048614501953e-05,
      "step": 27253,
      "training_step_time": 7.77885365486145
    },
    {
      "epoch": 2.599143981933594e-05,
      "model_forward_time": 1.7266526222229004,
      "step": 27254
    },
    {
      "epoch": 2.599143981933594e-05,
      "step": 27254,
      "training_step_time": 8.425930976867676
    },
    {
      "epoch": 2.5992393493652343e-05,
      "model_forward_time": 1.7175374031066895,
      "step": 27255
    },
    {
      "epoch": 2.5992393493652343e-05,
      "step": 27255,
      "training_step_time": 9.2478666305542
    },
    {
      "epoch": 2.599334716796875e-05,
      "model_forward_time": 1.7304563522338867,
      "step": 27256
    },
    {
      "epoch": 2.599334716796875e-05,
      "step": 27256,
      "training_step_time": 7.865832328796387
    },
    {
      "epoch": 2.5994300842285158e-05,
      "model_forward_time": 1.776505708694458,
      "step": 27257
    },
    {
      "epoch": 2.5994300842285158e-05,
      "step": 27257,
      "training_step_time": 7.359625577926636
    },
    {
      "epoch": 2.599525451660156e-05,
      "model_forward_time": 2.2633206844329834,
      "step": 27258
    },
    {
      "epoch": 2.599525451660156e-05,
      "step": 27258,
      "training_step_time": 7.1070237159729
    },
    {
      "epoch": 2.599620819091797e-05,
      "model_forward_time": 1.7200267314910889,
      "step": 27259
    },
    {
      "epoch": 2.599620819091797e-05,
      "step": 27259,
      "training_step_time": 6.899963855743408
    },
    {
      "epoch": 2.5997161865234376e-05,
      "grad_norm": 0.11805540323257446,
      "learning_rate": 8.705491179489738e-06,
      "loss": 0.0569,
      "step": 27260
    },
    {
      "epoch": 2.5997161865234376e-05,
      "model_forward_time": 1.7291548252105713,
      "step": 27260
    },
    {
      "dynamics_loss_avg": 0.05243471059948206,
      "epoch": 2.5997161865234376e-05,
      "step": 27260
    },
    {
      "action_loss_avg": 0.0015593477350194008,
      "epoch": 2.5997161865234376e-05,
      "step": 27260
    },
    {
      "epoch": 2.5997161865234376e-05,
      "step": 27260,
      "training_step_time": 7.583930253982544
    },
    {
      "epoch": 2.599811553955078e-05,
      "model_forward_time": 1.7411484718322754,
      "step": 27261
    },
    {
      "epoch": 2.599811553955078e-05,
      "step": 27261,
      "training_step_time": 7.637479543685913
    },
    {
      "epoch": 2.5999069213867188e-05,
      "model_forward_time": 1.7219326496124268,
      "step": 27262
    },
    {
      "epoch": 2.5999069213867188e-05,
      "step": 27262,
      "training_step_time": 7.749286890029907
    },
    {
      "epoch": 2.6000022888183595e-05,
      "model_forward_time": 1.7122063636779785,
      "step": 27263
    },
    {
      "epoch": 2.6000022888183595e-05,
      "step": 27263,
      "training_step_time": 9.758687257766724
    },
    {
      "epoch": 2.60009765625e-05,
      "model_forward_time": 1.339590072631836,
      "step": 27264
    },
    {
      "epoch": 2.60009765625e-05,
      "step": 27264,
      "training_step_time": 8.812109231948853
    },
    {
      "epoch": 2.6001930236816407e-05,
      "model_forward_time": 1.7200829982757568,
      "step": 27265
    },
    {
      "epoch": 2.6001930236816407e-05,
      "step": 27265,
      "training_step_time": 8.268842458724976
    },
    {
      "epoch": 2.600288391113281e-05,
      "model_forward_time": 1.7300631999969482,
      "step": 27266
    },
    {
      "epoch": 2.600288391113281e-05,
      "step": 27266,
      "training_step_time": 7.578003168106079
    },
    {
      "epoch": 2.6003837585449218e-05,
      "model_forward_time": 2.2547082901000977,
      "step": 27267
    },
    {
      "epoch": 2.6003837585449218e-05,
      "step": 27267,
      "training_step_time": 7.327630996704102
    },
    {
      "epoch": 2.6004791259765626e-05,
      "model_forward_time": 1.693037748336792,
      "step": 27268
    },
    {
      "epoch": 2.6004791259765626e-05,
      "step": 27268,
      "training_step_time": 6.864951133728027
    },
    {
      "epoch": 2.600574493408203e-05,
      "model_forward_time": 1.702005386352539,
      "step": 27269
    },
    {
      "epoch": 2.600574493408203e-05,
      "step": 27269,
      "training_step_time": 6.893257141113281
    },
    {
      "epoch": 2.6006698608398437e-05,
      "grad_norm": 0.1807839423418045,
      "learning_rate": 8.704380842870078e-06,
      "loss": 0.0609,
      "step": 27270
    },
    {
      "epoch": 2.6006698608398437e-05,
      "model_forward_time": 0.9917914867401123,
      "step": 27270
    },
    {
      "dynamics_loss_avg": 0.05791416075080633,
      "epoch": 2.6006698608398437e-05,
      "step": 27270
    },
    {
      "action_loss_avg": 0.0016516550444066525,
      "epoch": 2.6006698608398437e-05,
      "step": 27270
    },
    {
      "epoch": 2.6006698608398437e-05,
      "step": 27270,
      "training_step_time": 7.559987306594849
    },
    {
      "epoch": 2.6007652282714845e-05,
      "model_forward_time": 1.7106659412384033,
      "step": 27271
    },
    {
      "epoch": 2.6007652282714845e-05,
      "step": 27271,
      "training_step_time": 7.81518292427063
    },
    {
      "epoch": 2.600860595703125e-05,
      "model_forward_time": 1.7185158729553223,
      "step": 27272
    },
    {
      "epoch": 2.600860595703125e-05,
      "step": 27272,
      "training_step_time": 9.31230878829956
    },
    {
      "epoch": 2.6009559631347656e-05,
      "model_forward_time": 1.7248342037200928,
      "step": 27273
    },
    {
      "epoch": 2.6009559631347656e-05,
      "step": 27273,
      "training_step_time": 11.030421257019043
    },
    {
      "epoch": 2.6010513305664063e-05,
      "model_forward_time": 1.679025411605835,
      "step": 27274
    },
    {
      "epoch": 2.6010513305664063e-05,
      "step": 27274,
      "training_step_time": 9.637831211090088
    },
    {
      "epoch": 2.6011466979980467e-05,
      "model_forward_time": 1.725525140762329,
      "step": 27275
    },
    {
      "epoch": 2.6011466979980467e-05,
      "step": 27275,
      "training_step_time": 7.723154544830322
    },
    {
      "epoch": 2.6012420654296875e-05,
      "model_forward_time": 1.7183632850646973,
      "step": 27276
    },
    {
      "epoch": 2.6012420654296875e-05,
      "step": 27276,
      "training_step_time": 7.364945650100708
    },
    {
      "epoch": 2.6013374328613282e-05,
      "model_forward_time": 2.2512948513031006,
      "step": 27277
    },
    {
      "epoch": 2.6013374328613282e-05,
      "step": 27277,
      "training_step_time": 7.095296859741211
    },
    {
      "epoch": 2.6014328002929686e-05,
      "model_forward_time": 1.7062218189239502,
      "step": 27278
    },
    {
      "epoch": 2.6014328002929686e-05,
      "step": 27278,
      "training_step_time": 6.861844062805176
    },
    {
      "epoch": 2.6015281677246094e-05,
      "model_forward_time": 1.711064338684082,
      "step": 27279
    },
    {
      "epoch": 2.6015281677246094e-05,
      "step": 27279,
      "training_step_time": 7.61080527305603
    },
    {
      "epoch": 2.60162353515625e-05,
      "grad_norm": 0.1493029147386551,
      "learning_rate": 8.703270101144897e-06,
      "loss": 0.0627,
      "step": 27280
    },
    {
      "epoch": 2.60162353515625e-05,
      "model_forward_time": 1.7180192470550537,
      "step": 27280
    },
    {
      "dynamics_loss_avg": 0.05156911127269268,
      "epoch": 2.60162353515625e-05,
      "step": 27280
    },
    {
      "action_loss_avg": 0.0020769166352692992,
      "epoch": 2.60162353515625e-05,
      "step": 27280
    },
    {
      "epoch": 2.60162353515625e-05,
      "step": 27280,
      "training_step_time": 7.699413776397705
    },
    {
      "epoch": 2.6017189025878905e-05,
      "model_forward_time": 1.7233965396881104,
      "step": 27281
    },
    {
      "epoch": 2.6017189025878905e-05,
      "step": 27281,
      "training_step_time": 7.8541412353515625
    },
    {
      "epoch": 2.6018142700195313e-05,
      "model_forward_time": 1.7079145908355713,
      "step": 27282
    },
    {
      "epoch": 2.6018142700195313e-05,
      "step": 27282,
      "training_step_time": 8.56676197052002
    },
    {
      "epoch": 2.601909637451172e-05,
      "model_forward_time": 1.7097995281219482,
      "step": 27283
    },
    {
      "epoch": 2.601909637451172e-05,
      "step": 27283,
      "training_step_time": 9.206334352493286
    },
    {
      "epoch": 2.6020050048828124e-05,
      "model_forward_time": 1.7107164859771729,
      "step": 27284
    },
    {
      "epoch": 2.6020050048828124e-05,
      "step": 27284,
      "training_step_time": 8.606749773025513
    },
    {
      "epoch": 2.602100372314453e-05,
      "model_forward_time": 1.7401750087738037,
      "step": 27285
    },
    {
      "epoch": 2.602100372314453e-05,
      "step": 27285,
      "training_step_time": 7.300381183624268
    },
    {
      "epoch": 2.602195739746094e-05,
      "model_forward_time": 2.263479709625244,
      "step": 27286
    },
    {
      "epoch": 2.602195739746094e-05,
      "step": 27286,
      "training_step_time": 7.078731298446655
    },
    {
      "epoch": 2.6022911071777343e-05,
      "model_forward_time": 1.7379474639892578,
      "step": 27287
    },
    {
      "epoch": 2.6022911071777343e-05,
      "step": 27287,
      "training_step_time": 6.872305393218994
    },
    {
      "epoch": 2.602386474609375e-05,
      "model_forward_time": 1.7226555347442627,
      "step": 27288
    },
    {
      "epoch": 2.602386474609375e-05,
      "step": 27288,
      "training_step_time": 8.331108093261719
    },
    {
      "epoch": 2.6024818420410158e-05,
      "model_forward_time": 1.7239925861358643,
      "step": 27289
    },
    {
      "epoch": 2.6024818420410158e-05,
      "step": 27289,
      "training_step_time": 8.72021222114563
    },
    {
      "epoch": 2.602577209472656e-05,
      "grad_norm": 0.1648625284433365,
      "learning_rate": 8.702158954435664e-06,
      "loss": 0.0603,
      "step": 27290
    },
    {
      "epoch": 2.602577209472656e-05,
      "model_forward_time": 1.7157347202301025,
      "step": 27290
    },
    {
      "dynamics_loss_avg": 0.07776683494448662,
      "epoch": 2.602577209472656e-05,
      "step": 27290
    },
    {
      "action_loss_avg": 0.003049049683613703,
      "epoch": 2.602577209472656e-05,
      "step": 27290
    },
    {
      "epoch": 2.602577209472656e-05,
      "step": 27290,
      "training_step_time": 8.685695886611938
    },
    {
      "epoch": 2.602672576904297e-05,
      "model_forward_time": 1.7439377307891846,
      "step": 27291
    },
    {
      "epoch": 2.602672576904297e-05,
      "step": 27291,
      "training_step_time": 10.1518714427948
    },
    {
      "epoch": 2.6027679443359377e-05,
      "model_forward_time": 1.7782886028289795,
      "step": 27292
    },
    {
      "epoch": 2.6027679443359377e-05,
      "step": 27292,
      "training_step_time": 7.8661110401153564
    },
    {
      "epoch": 2.602863311767578e-05,
      "model_forward_time": 1.7430424690246582,
      "step": 27293
    },
    {
      "epoch": 2.602863311767578e-05,
      "step": 27293,
      "training_step_time": 7.681487560272217
    },
    {
      "epoch": 2.6029586791992188e-05,
      "model_forward_time": 1.7221426963806152,
      "step": 27294
    },
    {
      "epoch": 2.6029586791992188e-05,
      "step": 27294,
      "training_step_time": 7.618168830871582
    },
    {
      "epoch": 2.6030540466308595e-05,
      "model_forward_time": 2.2855045795440674,
      "step": 27295
    },
    {
      "epoch": 2.6030540466308595e-05,
      "step": 27295,
      "training_step_time": 7.378920555114746
    },
    {
      "epoch": 2.6031494140625e-05,
      "model_forward_time": 1.727815866470337,
      "step": 27296
    },
    {
      "epoch": 2.6031494140625e-05,
      "step": 27296,
      "training_step_time": 6.911072254180908
    },
    {
      "epoch": 2.6032447814941407e-05,
      "model_forward_time": 1.7421038150787354,
      "step": 27297
    },
    {
      "epoch": 2.6032447814941407e-05,
      "step": 27297,
      "training_step_time": 7.430231094360352
    },
    {
      "epoch": 2.603340148925781e-05,
      "model_forward_time": 1.7206151485443115,
      "step": 27298
    },
    {
      "epoch": 2.603340148925781e-05,
      "step": 27298,
      "training_step_time": 7.706724166870117
    },
    {
      "epoch": 2.6034355163574218e-05,
      "model_forward_time": 1.7222058773040771,
      "step": 27299
    },
    {
      "epoch": 2.6034355163574218e-05,
      "step": 27299,
      "training_step_time": 7.803283452987671
    },
    {
      "epoch": 2.6035308837890626e-05,
      "grad_norm": 0.170816570520401,
      "learning_rate": 8.701047402863897e-06,
      "loss": 0.0602,
      "step": 27300
    },
    {
      "epoch": 2.6035308837890626e-05,
      "model_forward_time": 1.7282476425170898,
      "step": 27300
    },
    {
      "dynamics_loss_avg": 0.03723128093406558,
      "epoch": 2.6035308837890626e-05,
      "step": 27300
    },
    {
      "action_loss_avg": 0.001496726943878457,
      "epoch": 2.6035308837890626e-05,
      "step": 27300
    },
    {
      "epoch": 2.6035308837890626e-05,
      "step": 27300,
      "training_step_time": 9.549195051193237
    },
    {
      "epoch": 2.603626251220703e-05,
      "model_forward_time": 1.7776837348937988,
      "step": 27301
    },
    {
      "epoch": 2.603626251220703e-05,
      "step": 27301,
      "training_step_time": 8.60872769355774
    },
    {
      "epoch": 2.6037216186523437e-05,
      "model_forward_time": 1.732537031173706,
      "step": 27302
    },
    {
      "epoch": 2.6037216186523437e-05,
      "step": 27302,
      "training_step_time": 8.797436475753784
    },
    {
      "epoch": 2.6038169860839845e-05,
      "model_forward_time": 1.7268071174621582,
      "step": 27303
    },
    {
      "epoch": 2.6038169860839845e-05,
      "step": 27303,
      "training_step_time": 7.764089822769165
    },
    {
      "epoch": 2.603912353515625e-05,
      "model_forward_time": 2.253783941268921,
      "step": 27304
    },
    {
      "epoch": 2.603912353515625e-05,
      "step": 27304,
      "training_step_time": 7.358581066131592
    },
    {
      "epoch": 2.6040077209472656e-05,
      "model_forward_time": 1.7259705066680908,
      "step": 27305
    },
    {
      "epoch": 2.6040077209472656e-05,
      "step": 27305,
      "training_step_time": 7.000784397125244
    },
    {
      "epoch": 2.6041030883789063e-05,
      "model_forward_time": 1.751603126525879,
      "step": 27306
    },
    {
      "epoch": 2.6041030883789063e-05,
      "step": 27306,
      "training_step_time": 6.914287567138672
    },
    {
      "epoch": 2.6041984558105467e-05,
      "model_forward_time": 1.7287440299987793,
      "step": 27307
    },
    {
      "epoch": 2.6041984558105467e-05,
      "step": 27307,
      "training_step_time": 7.6576831340789795
    },
    {
      "epoch": 2.6042938232421875e-05,
      "model_forward_time": 1.7342534065246582,
      "step": 27308
    },
    {
      "epoch": 2.6042938232421875e-05,
      "step": 27308,
      "training_step_time": 7.8569653034210205
    },
    {
      "epoch": 2.6043891906738282e-05,
      "model_forward_time": 1.7297179698944092,
      "step": 27309
    },
    {
      "epoch": 2.6043891906738282e-05,
      "step": 27309,
      "training_step_time": 8.92108154296875
    },
    {
      "epoch": 2.6044845581054686e-05,
      "grad_norm": 0.1726163625717163,
      "learning_rate": 8.699935446551149e-06,
      "loss": 0.062,
      "step": 27310
    },
    {
      "epoch": 2.6044845581054686e-05,
      "model_forward_time": 1.8015923500061035,
      "step": 27310
    },
    {
      "dynamics_loss_avg": 0.06715730018913746,
      "epoch": 2.6044845581054686e-05,
      "step": 27310
    },
    {
      "action_loss_avg": 0.0043882250785827635,
      "epoch": 2.6044845581054686e-05,
      "step": 27310
    },
    {
      "epoch": 2.6044845581054686e-05,
      "step": 27310,
      "training_step_time": 9.796581029891968
    },
    {
      "epoch": 2.6045799255371094e-05,
      "model_forward_time": 1.7534828186035156,
      "step": 27311
    },
    {
      "epoch": 2.6045799255371094e-05,
      "step": 27311,
      "training_step_time": 7.870307922363281
    },
    {
      "epoch": 2.60467529296875e-05,
      "model_forward_time": 1.7575843334197998,
      "step": 27312
    },
    {
      "epoch": 2.60467529296875e-05,
      "step": 27312,
      "training_step_time": 9.289498329162598
    },
    {
      "epoch": 2.6047706604003905e-05,
      "model_forward_time": 1.7531843185424805,
      "step": 27313
    },
    {
      "epoch": 2.6047706604003905e-05,
      "step": 27313,
      "training_step_time": 7.406967878341675
    },
    {
      "epoch": 2.6048660278320313e-05,
      "model_forward_time": 2.276195764541626,
      "step": 27314
    },
    {
      "epoch": 2.6048660278320313e-05,
      "step": 27314,
      "training_step_time": 7.153008937835693
    },
    {
      "epoch": 2.604961395263672e-05,
      "model_forward_time": 1.764094352722168,
      "step": 27315
    },
    {
      "epoch": 2.604961395263672e-05,
      "step": 27315,
      "training_step_time": 6.8202056884765625
    },
    {
      "epoch": 2.6050567626953124e-05,
      "model_forward_time": 1.7507014274597168,
      "step": 27316
    },
    {
      "epoch": 2.6050567626953124e-05,
      "step": 27316,
      "training_step_time": 7.441403388977051
    },
    {
      "epoch": 2.605152130126953e-05,
      "model_forward_time": 1.7390532493591309,
      "step": 27317
    },
    {
      "epoch": 2.605152130126953e-05,
      "step": 27317,
      "training_step_time": 7.82554030418396
    },
    {
      "epoch": 2.605247497558594e-05,
      "model_forward_time": 1.7516350746154785,
      "step": 27318
    },
    {
      "epoch": 2.605247497558594e-05,
      "step": 27318,
      "training_step_time": 8.791541814804077
    },
    {
      "epoch": 2.6053428649902343e-05,
      "model_forward_time": 1.7894964218139648,
      "step": 27319
    },
    {
      "epoch": 2.6053428649902343e-05,
      "step": 27319,
      "training_step_time": 8.440297842025757
    },
    {
      "epoch": 2.605438232421875e-05,
      "grad_norm": 0.18436382710933685,
      "learning_rate": 8.698823085619022e-06,
      "loss": 0.0595,
      "step": 27320
    },
    {
      "epoch": 2.605438232421875e-05,
      "model_forward_time": 1.7392852306365967,
      "step": 27320
    },
    {
      "dynamics_loss_avg": 0.04385603386908769,
      "epoch": 2.605438232421875e-05,
      "step": 27320
    },
    {
      "action_loss_avg": 0.003128060733433813,
      "epoch": 2.605438232421875e-05,
      "step": 27320
    },
    {
      "epoch": 2.605438232421875e-05,
      "step": 27320,
      "training_step_time": 9.22486686706543
    },
    {
      "epoch": 2.6055335998535158e-05,
      "model_forward_time": 1.7679407596588135,
      "step": 27321
    },
    {
      "epoch": 2.6055335998535158e-05,
      "step": 27321,
      "training_step_time": 8.592390060424805
    },
    {
      "epoch": 2.6056289672851562e-05,
      "model_forward_time": 1.760728359222412,
      "step": 27322
    },
    {
      "epoch": 2.6056289672851562e-05,
      "step": 27322,
      "training_step_time": 7.559792995452881
    },
    {
      "epoch": 2.605724334716797e-05,
      "model_forward_time": 2.3200182914733887,
      "step": 27323
    },
    {
      "epoch": 2.605724334716797e-05,
      "step": 27323,
      "training_step_time": 7.2629640102386475
    },
    {
      "epoch": 2.6058197021484377e-05,
      "model_forward_time": 1.381990909576416,
      "step": 27324
    },
    {
      "epoch": 2.6058197021484377e-05,
      "step": 27324,
      "training_step_time": 6.935683012008667
    },
    {
      "epoch": 2.605915069580078e-05,
      "model_forward_time": 1.7534167766571045,
      "step": 27325
    },
    {
      "epoch": 2.605915069580078e-05,
      "step": 27325,
      "training_step_time": 7.474104166030884
    },
    {
      "epoch": 2.6060104370117188e-05,
      "model_forward_time": 1.7670187950134277,
      "step": 27326
    },
    {
      "epoch": 2.6060104370117188e-05,
      "step": 27326,
      "training_step_time": 8.353208541870117
    },
    {
      "epoch": 2.6061058044433595e-05,
      "model_forward_time": 1.754378318786621,
      "step": 27327
    },
    {
      "epoch": 2.6061058044433595e-05,
      "step": 27327,
      "training_step_time": 7.687541246414185
    },
    {
      "epoch": 2.606201171875e-05,
      "model_forward_time": 1.7803163528442383,
      "step": 27328
    },
    {
      "epoch": 2.606201171875e-05,
      "step": 27328,
      "training_step_time": 8.937551975250244
    },
    {
      "epoch": 2.6062965393066407e-05,
      "model_forward_time": 1.7474274635314941,
      "step": 27329
    },
    {
      "epoch": 2.6062965393066407e-05,
      "step": 27329,
      "training_step_time": 8.756865978240967
    },
    {
      "epoch": 2.606391906738281e-05,
      "grad_norm": 0.17468266189098358,
      "learning_rate": 8.697710320189166e-06,
      "loss": 0.0662,
      "step": 27330
    },
    {
      "epoch": 2.606391906738281e-05,
      "model_forward_time": 1.7530007362365723,
      "step": 27330
    },
    {
      "dynamics_loss_avg": 0.0514687180519104,
      "epoch": 2.606391906738281e-05,
      "step": 27330
    },
    {
      "action_loss_avg": 0.003732117696199566,
      "epoch": 2.606391906738281e-05,
      "step": 27330
    },
    {
      "epoch": 2.606391906738281e-05,
      "step": 27330,
      "training_step_time": 7.720641613006592
    },
    {
      "epoch": 2.6064872741699218e-05,
      "model_forward_time": 1.7447261810302734,
      "step": 27331
    },
    {
      "epoch": 2.6064872741699218e-05,
      "step": 27331,
      "training_step_time": 7.697709083557129
    },
    {
      "epoch": 2.6065826416015626e-05,
      "model_forward_time": 2.3415117263793945,
      "step": 27332
    },
    {
      "epoch": 2.6065826416015626e-05,
      "step": 27332,
      "training_step_time": 7.394133567810059
    },
    {
      "epoch": 2.606678009033203e-05,
      "model_forward_time": 1.7623498439788818,
      "step": 27333
    },
    {
      "epoch": 2.606678009033203e-05,
      "step": 27333,
      "training_step_time": 6.917793273925781
    },
    {
      "epoch": 2.6067733764648437e-05,
      "model_forward_time": 1.7562854290008545,
      "step": 27334
    },
    {
      "epoch": 2.6067733764648437e-05,
      "step": 27334,
      "training_step_time": 7.471411943435669
    },
    {
      "epoch": 2.6068687438964845e-05,
      "model_forward_time": 1.7689299583435059,
      "step": 27335
    },
    {
      "epoch": 2.6068687438964845e-05,
      "step": 27335,
      "training_step_time": 7.761149168014526
    },
    {
      "epoch": 2.606964111328125e-05,
      "model_forward_time": 1.756464958190918,
      "step": 27336
    },
    {
      "epoch": 2.606964111328125e-05,
      "step": 27336,
      "training_step_time": 9.64661169052124
    },
    {
      "epoch": 2.6070594787597656e-05,
      "model_forward_time": 1.7918205261230469,
      "step": 27337
    },
    {
      "epoch": 2.6070594787597656e-05,
      "step": 27337,
      "training_step_time": 9.413370132446289
    },
    {
      "epoch": 2.6071548461914063e-05,
      "model_forward_time": 1.7588756084442139,
      "step": 27338
    },
    {
      "epoch": 2.6071548461914063e-05,
      "step": 27338,
      "training_step_time": 8.033661842346191
    },
    {
      "epoch": 2.6072502136230467e-05,
      "model_forward_time": 1.7462091445922852,
      "step": 27339
    },
    {
      "epoch": 2.6072502136230467e-05,
      "step": 27339,
      "training_step_time": 8.762619733810425
    },
    {
      "epoch": 2.6073455810546875e-05,
      "grad_norm": 0.10779841244220734,
      "learning_rate": 8.696597150383269e-06,
      "loss": 0.0646,
      "step": 27340
    },
    {
      "epoch": 2.6073455810546875e-05,
      "model_forward_time": 1.7406070232391357,
      "step": 27340
    },
    {
      "dynamics_loss_avg": 0.04673394463025034,
      "epoch": 2.6073455810546875e-05,
      "step": 27340
    },
    {
      "action_loss_avg": 0.0020752330194227396,
      "epoch": 2.6073455810546875e-05,
      "step": 27340
    },
    {
      "epoch": 2.6073455810546875e-05,
      "step": 27340,
      "training_step_time": 7.71635627746582
    },
    {
      "epoch": 2.6074409484863282e-05,
      "model_forward_time": 2.302520751953125,
      "step": 27341
    },
    {
      "epoch": 2.6074409484863282e-05,
      "step": 27341,
      "training_step_time": 7.33807635307312
    },
    {
      "epoch": 2.6075363159179686e-05,
      "model_forward_time": 1.7466301918029785,
      "step": 27342
    },
    {
      "epoch": 2.6075363159179686e-05,
      "step": 27342,
      "training_step_time": 6.901210308074951
    },
    {
      "epoch": 2.6076316833496094e-05,
      "model_forward_time": 1.7779278755187988,
      "step": 27343
    },
    {
      "epoch": 2.6076316833496094e-05,
      "step": 27343,
      "training_step_time": 7.5111916065216064
    },
    {
      "epoch": 2.60772705078125e-05,
      "model_forward_time": 1.7663865089416504,
      "step": 27344
    },
    {
      "epoch": 2.60772705078125e-05,
      "step": 27344,
      "training_step_time": 7.692366600036621
    },
    {
      "epoch": 2.6078224182128905e-05,
      "model_forward_time": 1.7621068954467773,
      "step": 27345
    },
    {
      "epoch": 2.6078224182128905e-05,
      "step": 27345,
      "training_step_time": 7.808401107788086
    },
    {
      "epoch": 2.6079177856445313e-05,
      "model_forward_time": 1.7659494876861572,
      "step": 27346
    },
    {
      "epoch": 2.6079177856445313e-05,
      "step": 27346,
      "training_step_time": 10.512314319610596
    },
    {
      "epoch": 2.608013153076172e-05,
      "model_forward_time": 1.7581532001495361,
      "step": 27347
    },
    {
      "epoch": 2.608013153076172e-05,
      "step": 27347,
      "training_step_time": 9.087259531021118
    },
    {
      "epoch": 2.6081085205078124e-05,
      "model_forward_time": 1.7775568962097168,
      "step": 27348
    },
    {
      "epoch": 2.6081085205078124e-05,
      "step": 27348,
      "training_step_time": 7.844781398773193
    },
    {
      "epoch": 2.608203887939453e-05,
      "model_forward_time": 1.7561085224151611,
      "step": 27349
    },
    {
      "epoch": 2.608203887939453e-05,
      "step": 27349,
      "training_step_time": 7.6852476596832275
    },
    {
      "epoch": 2.608299255371094e-05,
      "grad_norm": 0.22261250019073486,
      "learning_rate": 8.695483576323062e-06,
      "loss": 0.067,
      "step": 27350
    },
    {
      "epoch": 2.608299255371094e-05,
      "model_forward_time": 1.7464313507080078,
      "step": 27350
    },
    {
      "dynamics_loss_avg": 0.07595942467451096,
      "epoch": 2.608299255371094e-05,
      "step": 27350
    },
    {
      "action_loss_avg": 0.00304693229845725,
      "epoch": 2.608299255371094e-05,
      "step": 27350
    },
    {
      "epoch": 2.608299255371094e-05,
      "step": 27350,
      "training_step_time": 8.749786376953125
    },
    {
      "epoch": 2.6083946228027343e-05,
      "model_forward_time": 1.7567658424377441,
      "step": 27351
    },
    {
      "epoch": 2.6083946228027343e-05,
      "step": 27351,
      "training_step_time": 6.847741365432739
    },
    {
      "epoch": 2.608489990234375e-05,
      "model_forward_time": 1.7585694789886475,
      "step": 27352
    },
    {
      "epoch": 2.608489990234375e-05,
      "step": 27352,
      "training_step_time": 7.424799680709839
    },
    {
      "epoch": 2.6085853576660158e-05,
      "model_forward_time": 1.755800485610962,
      "step": 27353
    },
    {
      "epoch": 2.6085853576660158e-05,
      "step": 27353,
      "training_step_time": 7.805132150650024
    },
    {
      "epoch": 2.6086807250976562e-05,
      "model_forward_time": 1.7640161514282227,
      "step": 27354
    },
    {
      "epoch": 2.6086807250976562e-05,
      "step": 27354,
      "training_step_time": 7.779735803604126
    },
    {
      "epoch": 2.608776092529297e-05,
      "model_forward_time": 1.7285735607147217,
      "step": 27355
    },
    {
      "epoch": 2.608776092529297e-05,
      "step": 27355,
      "training_step_time": 8.758923768997192
    },
    {
      "epoch": 2.6088714599609377e-05,
      "model_forward_time": 1.7760956287384033,
      "step": 27356
    },
    {
      "epoch": 2.6088714599609377e-05,
      "step": 27356,
      "training_step_time": 10.032483100891113
    },
    {
      "epoch": 2.608966827392578e-05,
      "model_forward_time": 1.739659070968628,
      "step": 27357
    },
    {
      "epoch": 2.608966827392578e-05,
      "step": 27357,
      "training_step_time": 8.82041311264038
    },
    {
      "epoch": 2.6090621948242188e-05,
      "model_forward_time": 1.7429747581481934,
      "step": 27358
    },
    {
      "epoch": 2.6090621948242188e-05,
      "step": 27358,
      "training_step_time": 7.70424222946167
    },
    {
      "epoch": 2.6091575622558595e-05,
      "model_forward_time": 1.3705527782440186,
      "step": 27359
    },
    {
      "epoch": 2.6091575622558595e-05,
      "step": 27359,
      "training_step_time": 7.615550518035889
    },
    {
      "epoch": 2.6092529296875e-05,
      "grad_norm": 0.1687946617603302,
      "learning_rate": 8.69436959813033e-06,
      "loss": 0.0667,
      "step": 27360
    },
    {
      "epoch": 2.6092529296875e-05,
      "model_forward_time": 2.2726473808288574,
      "step": 27360
    },
    {
      "dynamics_loss_avg": 0.05065382570028305,
      "epoch": 2.6092529296875e-05,
      "step": 27360
    },
    {
      "action_loss_avg": 0.001508397510042414,
      "epoch": 2.6092529296875e-05,
      "step": 27360
    },
    {
      "epoch": 2.6092529296875e-05,
      "step": 27360,
      "training_step_time": 7.060316562652588
    },
    {
      "epoch": 2.6093482971191407e-05,
      "model_forward_time": 1.7453250885009766,
      "step": 27361
    },
    {
      "epoch": 2.6093482971191407e-05,
      "step": 27361,
      "training_step_time": 7.323096513748169
    },
    {
      "epoch": 2.609443664550781e-05,
      "model_forward_time": 1.3691349029541016,
      "step": 27362
    },
    {
      "epoch": 2.609443664550781e-05,
      "step": 27362,
      "training_step_time": 7.698894262313843
    },
    {
      "epoch": 2.609539031982422e-05,
      "model_forward_time": 1.7323341369628906,
      "step": 27363
    },
    {
      "epoch": 2.609539031982422e-05,
      "step": 27363,
      "training_step_time": 7.805238246917725
    },
    {
      "epoch": 2.6096343994140626e-05,
      "model_forward_time": 1.7427282333374023,
      "step": 27364
    },
    {
      "epoch": 2.6096343994140626e-05,
      "step": 27364,
      "training_step_time": 7.639511585235596
    },
    {
      "epoch": 2.609729766845703e-05,
      "model_forward_time": 1.7849986553192139,
      "step": 27365
    },
    {
      "epoch": 2.609729766845703e-05,
      "step": 27365,
      "training_step_time": 7.772347450256348
    },
    {
      "epoch": 2.6098251342773437e-05,
      "model_forward_time": 1.7282307147979736,
      "step": 27366
    },
    {
      "epoch": 2.6098251342773437e-05,
      "step": 27366,
      "training_step_time": 8.86645221710205
    },
    {
      "epoch": 2.6099205017089845e-05,
      "model_forward_time": 1.7469847202301025,
      "step": 27367
    },
    {
      "epoch": 2.6099205017089845e-05,
      "step": 27367,
      "training_step_time": 7.756620168685913
    },
    {
      "epoch": 2.610015869140625e-05,
      "model_forward_time": 1.7398560047149658,
      "step": 27368
    },
    {
      "epoch": 2.610015869140625e-05,
      "step": 27368,
      "training_step_time": 7.289862871170044
    },
    {
      "epoch": 2.6101112365722656e-05,
      "model_forward_time": 2.289621591567993,
      "step": 27369
    },
    {
      "epoch": 2.6101112365722656e-05,
      "step": 27369,
      "training_step_time": 7.683396577835083
    },
    {
      "epoch": 2.6102066040039063e-05,
      "grad_norm": 0.1581689864397049,
      "learning_rate": 8.693255215926893e-06,
      "loss": 0.0661,
      "step": 27370
    },
    {
      "epoch": 2.6102066040039063e-05,
      "model_forward_time": 1.7317380905151367,
      "step": 27370
    },
    {
      "dynamics_loss_avg": 0.06276242006570101,
      "epoch": 2.6102066040039063e-05,
      "step": 27370
    },
    {
      "action_loss_avg": 0.0024225197965279223,
      "epoch": 2.6102066040039063e-05,
      "step": 27370
    },
    {
      "epoch": 2.6102066040039063e-05,
      "step": 27370,
      "training_step_time": 7.267747640609741
    },
    {
      "epoch": 2.6103019714355467e-05,
      "model_forward_time": 1.7449018955230713,
      "step": 27371
    },
    {
      "epoch": 2.6103019714355467e-05,
      "step": 27371,
      "training_step_time": 7.60563063621521
    },
    {
      "epoch": 2.6103973388671875e-05,
      "model_forward_time": 1.7338414192199707,
      "step": 27372
    },
    {
      "epoch": 2.6103973388671875e-05,
      "step": 27372,
      "training_step_time": 7.707545757293701
    },
    {
      "epoch": 2.6104927062988282e-05,
      "model_forward_time": 1.7590575218200684,
      "step": 27373
    },
    {
      "epoch": 2.6104927062988282e-05,
      "step": 27373,
      "training_step_time": 7.645148038864136
    },
    {
      "epoch": 2.6105880737304686e-05,
      "model_forward_time": 1.7779018878936768,
      "step": 27374
    },
    {
      "epoch": 2.6105880737304686e-05,
      "step": 27374,
      "training_step_time": 7.787833213806152
    },
    {
      "epoch": 2.6106834411621094e-05,
      "model_forward_time": 1.7405455112457275,
      "step": 27375
    },
    {
      "epoch": 2.6106834411621094e-05,
      "step": 27375,
      "training_step_time": 9.63517141342163
    },
    {
      "epoch": 2.61077880859375e-05,
      "model_forward_time": 1.753617763519287,
      "step": 27376
    },
    {
      "epoch": 2.61077880859375e-05,
      "step": 27376,
      "training_step_time": 8.402791023254395
    },
    {
      "epoch": 2.6108741760253905e-05,
      "model_forward_time": 1.746027946472168,
      "step": 27377
    },
    {
      "epoch": 2.6108741760253905e-05,
      "step": 27377,
      "training_step_time": 7.4131340980529785
    },
    {
      "epoch": 2.6109695434570313e-05,
      "model_forward_time": 2.3122775554656982,
      "step": 27378
    },
    {
      "epoch": 2.6109695434570313e-05,
      "step": 27378,
      "training_step_time": 7.602409362792969
    },
    {
      "epoch": 2.611064910888672e-05,
      "model_forward_time": 1.3793728351593018,
      "step": 27379
    },
    {
      "epoch": 2.611064910888672e-05,
      "step": 27379,
      "training_step_time": 7.392691135406494
    },
    {
      "epoch": 2.6111602783203124e-05,
      "grad_norm": 0.19490613043308258,
      "learning_rate": 8.692140429834618e-06,
      "loss": 0.0646,
      "step": 27380
    },
    {
      "epoch": 2.6111602783203124e-05,
      "model_forward_time": 1.731544017791748,
      "step": 27380
    },
    {
      "dynamics_loss_avg": 0.06288001965731382,
      "epoch": 2.6111602783203124e-05,
      "step": 27380
    },
    {
      "action_loss_avg": 0.002884558163350448,
      "epoch": 2.6111602783203124e-05,
      "step": 27380
    },
    {
      "epoch": 2.6111602783203124e-05,
      "step": 27380,
      "training_step_time": 7.654590368270874
    },
    {
      "epoch": 2.611255645751953e-05,
      "model_forward_time": 1.735792636871338,
      "step": 27381
    },
    {
      "epoch": 2.611255645751953e-05,
      "step": 27381,
      "training_step_time": 9.95327115058899
    },
    {
      "epoch": 2.611351013183594e-05,
      "model_forward_time": 1.7466986179351807,
      "step": 27382
    },
    {
      "epoch": 2.611351013183594e-05,
      "step": 27382,
      "training_step_time": 8.538443326950073
    },
    {
      "epoch": 2.6114463806152343e-05,
      "model_forward_time": 1.3752870559692383,
      "step": 27383
    },
    {
      "epoch": 2.6114463806152343e-05,
      "step": 27383,
      "training_step_time": 7.790687322616577
    },
    {
      "epoch": 2.611541748046875e-05,
      "model_forward_time": 1.7456679344177246,
      "step": 27384
    },
    {
      "epoch": 2.611541748046875e-05,
      "step": 27384,
      "training_step_time": 8.287299156188965
    },
    {
      "epoch": 2.6116371154785158e-05,
      "model_forward_time": 1.7400314807891846,
      "step": 27385
    },
    {
      "epoch": 2.6116371154785158e-05,
      "step": 27385,
      "training_step_time": 7.694979190826416
    },
    {
      "epoch": 2.6117324829101562e-05,
      "model_forward_time": 1.7482404708862305,
      "step": 27386
    },
    {
      "epoch": 2.6117324829101562e-05,
      "step": 27386,
      "training_step_time": 7.661998271942139
    },
    {
      "epoch": 2.611827850341797e-05,
      "model_forward_time": 1.7475264072418213,
      "step": 27387
    },
    {
      "epoch": 2.611827850341797e-05,
      "step": 27387,
      "training_step_time": 8.607146501541138
    },
    {
      "epoch": 2.6119232177734377e-05,
      "model_forward_time": 1.74544095993042,
      "step": 27388
    },
    {
      "epoch": 2.6119232177734377e-05,
      "step": 27388,
      "training_step_time": 7.254881143569946
    },
    {
      "epoch": 2.612018585205078e-05,
      "model_forward_time": 1.7498323917388916,
      "step": 27389
    },
    {
      "epoch": 2.612018585205078e-05,
      "step": 27389,
      "training_step_time": 7.665289878845215
    },
    {
      "epoch": 2.6121139526367188e-05,
      "grad_norm": 0.13336825370788574,
      "learning_rate": 8.691025239975415e-06,
      "loss": 0.0662,
      "step": 27390
    },
    {
      "epoch": 2.6121139526367188e-05,
      "model_forward_time": 1.3459384441375732,
      "step": 27390
    },
    {
      "dynamics_loss_avg": 0.05672012753784657,
      "epoch": 2.6121139526367188e-05,
      "step": 27390
    },
    {
      "action_loss_avg": 0.00282578564947471,
      "epoch": 2.6121139526367188e-05,
      "step": 27390
    },
    {
      "epoch": 2.6121139526367188e-05,
      "step": 27390,
      "training_step_time": 7.61282205581665
    },
    {
      "epoch": 2.6122093200683592e-05,
      "model_forward_time": 1.7307779788970947,
      "step": 27391
    },
    {
      "epoch": 2.6122093200683592e-05,
      "step": 27391,
      "training_step_time": 8.488759279251099
    },
    {
      "epoch": 2.6123046875e-05,
      "model_forward_time": 1.715308666229248,
      "step": 27392
    },
    {
      "epoch": 2.6123046875e-05,
      "step": 27392,
      "training_step_time": 8.896950721740723
    },
    {
      "epoch": 2.6124000549316407e-05,
      "model_forward_time": 1.7203080654144287,
      "step": 27393
    },
    {
      "epoch": 2.6124000549316407e-05,
      "step": 27393,
      "training_step_time": 8.746013164520264
    },
    {
      "epoch": 2.612495422363281e-05,
      "model_forward_time": 1.7380502223968506,
      "step": 27394
    },
    {
      "epoch": 2.612495422363281e-05,
      "step": 27394,
      "training_step_time": 7.828256607055664
    },
    {
      "epoch": 2.612590789794922e-05,
      "model_forward_time": 1.74029541015625,
      "step": 27395
    },
    {
      "epoch": 2.612590789794922e-05,
      "step": 27395,
      "training_step_time": 8.692776203155518
    },
    {
      "epoch": 2.6126861572265626e-05,
      "model_forward_time": 1.7274794578552246,
      "step": 27396
    },
    {
      "epoch": 2.6126861572265626e-05,
      "step": 27396,
      "training_step_time": 8.118694543838501
    },
    {
      "epoch": 2.612781524658203e-05,
      "model_forward_time": 2.9267168045043945,
      "step": 27397
    },
    {
      "epoch": 2.612781524658203e-05,
      "step": 27397,
      "training_step_time": 7.756715297698975
    },
    {
      "epoch": 2.6128768920898437e-05,
      "model_forward_time": 1.7253804206848145,
      "step": 27398
    },
    {
      "epoch": 2.6128768920898437e-05,
      "step": 27398,
      "training_step_time": 7.762362241744995
    },
    {
      "epoch": 2.6129722595214845e-05,
      "model_forward_time": 1.7258665561676025,
      "step": 27399
    },
    {
      "epoch": 2.6129722595214845e-05,
      "step": 27399,
      "training_step_time": 8.584672212600708
    },
    {
      "epoch": 2.613067626953125e-05,
      "grad_norm": 0.13889844715595245,
      "learning_rate": 8.689909646471243e-06,
      "loss": 0.0671,
      "step": 27400
    },
    {
      "epoch": 2.613067626953125e-05,
      "model_forward_time": 1.7264769077301025,
      "step": 27400
    },
    {
      "dynamics_loss_avg": 0.07289397604763508,
      "epoch": 2.613067626953125e-05,
      "step": 27400
    },
    {
      "action_loss_avg": 0.004128050548024476,
      "epoch": 2.613067626953125e-05,
      "step": 27400
    },
    {
      "epoch": 2.613067626953125e-05,
      "step": 27400,
      "training_step_time": 7.68044900894165
    },
    {
      "epoch": 2.6131629943847656e-05,
      "model_forward_time": 1.7259459495544434,
      "step": 27401
    },
    {
      "epoch": 2.6131629943847656e-05,
      "step": 27401,
      "training_step_time": 7.6715967655181885
    },
    {
      "epoch": 2.6132583618164063e-05,
      "model_forward_time": 1.7135789394378662,
      "step": 27402
    },
    {
      "epoch": 2.6132583618164063e-05,
      "step": 27402,
      "training_step_time": 7.874422073364258
    },
    {
      "epoch": 2.6133537292480468e-05,
      "model_forward_time": 1.7550671100616455,
      "step": 27403
    },
    {
      "epoch": 2.6133537292480468e-05,
      "step": 27403,
      "training_step_time": 7.760725975036621
    },
    {
      "epoch": 2.6134490966796875e-05,
      "model_forward_time": 1.7336311340332031,
      "step": 27404
    },
    {
      "epoch": 2.6134490966796875e-05,
      "step": 27404,
      "training_step_time": 7.6726274490356445
    },
    {
      "epoch": 2.6135444641113282e-05,
      "model_forward_time": 1.7258586883544922,
      "step": 27405
    },
    {
      "epoch": 2.6135444641113282e-05,
      "step": 27405,
      "training_step_time": 7.402037858963013
    },
    {
      "epoch": 2.6136398315429686e-05,
      "model_forward_time": 2.8992462158203125,
      "step": 27406
    },
    {
      "epoch": 2.6136398315429686e-05,
      "step": 27406,
      "training_step_time": 7.672901391983032
    },
    {
      "epoch": 2.6137351989746094e-05,
      "model_forward_time": 1.7234158515930176,
      "step": 27407
    },
    {
      "epoch": 2.6137351989746094e-05,
      "step": 27407,
      "training_step_time": 7.738101005554199
    },
    {
      "epoch": 2.61383056640625e-05,
      "model_forward_time": 1.7255759239196777,
      "step": 27408
    },
    {
      "epoch": 2.61383056640625e-05,
      "step": 27408,
      "training_step_time": 7.5692737102508545
    },
    {
      "epoch": 2.6139259338378905e-05,
      "model_forward_time": 1.726653814315796,
      "step": 27409
    },
    {
      "epoch": 2.6139259338378905e-05,
      "step": 27409,
      "training_step_time": 7.621955394744873
    },
    {
      "epoch": 2.6140213012695313e-05,
      "grad_norm": 0.1024492010474205,
      "learning_rate": 8.6887936494441e-06,
      "loss": 0.0661,
      "step": 27410
    },
    {
      "epoch": 2.6140213012695313e-05,
      "model_forward_time": 1.7290372848510742,
      "step": 27410
    },
    {
      "dynamics_loss_avg": 0.05946505293250084,
      "epoch": 2.6140213012695313e-05,
      "step": 27410
    },
    {
      "action_loss_avg": 0.0026889803062658756,
      "epoch": 2.6140213012695313e-05,
      "step": 27410
    },
    {
      "epoch": 2.6140213012695313e-05,
      "step": 27410,
      "training_step_time": 9.25493049621582
    },
    {
      "epoch": 2.614116668701172e-05,
      "model_forward_time": 1.7262699604034424,
      "step": 27411
    },
    {
      "epoch": 2.614116668701172e-05,
      "step": 27411,
      "training_step_time": 8.961040735244751
    },
    {
      "epoch": 2.6142120361328124e-05,
      "model_forward_time": 1.7369580268859863,
      "step": 27412
    },
    {
      "epoch": 2.6142120361328124e-05,
      "step": 27412,
      "training_step_time": 8.980305671691895
    },
    {
      "epoch": 2.614307403564453e-05,
      "model_forward_time": 1.7261290550231934,
      "step": 27413
    },
    {
      "epoch": 2.614307403564453e-05,
      "step": 27413,
      "training_step_time": 7.687352180480957
    },
    {
      "epoch": 2.614402770996094e-05,
      "model_forward_time": 1.7253999710083008,
      "step": 27414
    },
    {
      "epoch": 2.614402770996094e-05,
      "step": 27414,
      "training_step_time": 7.446531295776367
    },
    {
      "epoch": 2.6144981384277343e-05,
      "model_forward_time": 2.8789141178131104,
      "step": 27415
    },
    {
      "epoch": 2.6144981384277343e-05,
      "step": 27415,
      "training_step_time": 7.677954912185669
    },
    {
      "epoch": 2.614593505859375e-05,
      "model_forward_time": 1.7278568744659424,
      "step": 27416
    },
    {
      "epoch": 2.614593505859375e-05,
      "step": 27416,
      "training_step_time": 7.705219507217407
    },
    {
      "epoch": 2.6146888732910158e-05,
      "model_forward_time": 1.362131118774414,
      "step": 27417
    },
    {
      "epoch": 2.6146888732910158e-05,
      "step": 27417,
      "training_step_time": 8.408698081970215
    },
    {
      "epoch": 2.6147842407226562e-05,
      "model_forward_time": 1.7232766151428223,
      "step": 27418
    },
    {
      "epoch": 2.6147842407226562e-05,
      "step": 27418,
      "training_step_time": 8.579249143600464
    },
    {
      "epoch": 2.614879608154297e-05,
      "model_forward_time": 1.7294843196868896,
      "step": 27419
    },
    {
      "epoch": 2.614879608154297e-05,
      "step": 27419,
      "training_step_time": 7.646403789520264
    },
    {
      "epoch": 2.6149749755859377e-05,
      "grad_norm": 0.15592429041862488,
      "learning_rate": 8.687677249016028e-06,
      "loss": 0.0658,
      "step": 27420
    },
    {
      "epoch": 2.6149749755859377e-05,
      "model_forward_time": 1.7257671356201172,
      "step": 27420
    },
    {
      "dynamics_loss_avg": 0.053833124414086345,
      "epoch": 2.6149749755859377e-05,
      "step": 27420
    },
    {
      "action_loss_avg": 0.0023098915873561053,
      "epoch": 2.6149749755859377e-05,
      "step": 27420
    },
    {
      "epoch": 2.6149749755859377e-05,
      "step": 27420,
      "training_step_time": 10.666592597961426
    },
    {
      "epoch": 2.615070343017578e-05,
      "model_forward_time": 1.7541711330413818,
      "step": 27421
    },
    {
      "epoch": 2.615070343017578e-05,
      "step": 27421,
      "training_step_time": 9.022012948989868
    },
    {
      "epoch": 2.6151657104492188e-05,
      "model_forward_time": 1.7294995784759521,
      "step": 27422
    },
    {
      "epoch": 2.6151657104492188e-05,
      "step": 27422,
      "training_step_time": 9.705180406570435
    },
    {
      "epoch": 2.6152610778808592e-05,
      "model_forward_time": 1.7430996894836426,
      "step": 27423
    },
    {
      "epoch": 2.6152610778808592e-05,
      "step": 27423,
      "training_step_time": 7.463740348815918
    },
    {
      "epoch": 2.6153564453125e-05,
      "model_forward_time": 1.7235450744628906,
      "step": 27424
    },
    {
      "epoch": 2.6153564453125e-05,
      "step": 27424,
      "training_step_time": 7.582864761352539
    },
    {
      "epoch": 2.6154518127441407e-05,
      "model_forward_time": 2.848365068435669,
      "step": 27425
    },
    {
      "epoch": 2.6154518127441407e-05,
      "step": 27425,
      "training_step_time": 7.712198495864868
    },
    {
      "epoch": 2.615547180175781e-05,
      "model_forward_time": 1.3543589115142822,
      "step": 27426
    },
    {
      "epoch": 2.615547180175781e-05,
      "step": 27426,
      "training_step_time": 7.554395914077759
    },
    {
      "epoch": 2.615642547607422e-05,
      "model_forward_time": 1.7276339530944824,
      "step": 27427
    },
    {
      "epoch": 2.615642547607422e-05,
      "step": 27427,
      "training_step_time": 8.16271686553955
    },
    {
      "epoch": 2.6157379150390626e-05,
      "model_forward_time": 1.7276866436004639,
      "step": 27428
    },
    {
      "epoch": 2.6157379150390626e-05,
      "step": 27428,
      "training_step_time": 7.747762680053711
    },
    {
      "epoch": 2.615833282470703e-05,
      "model_forward_time": 1.7287085056304932,
      "step": 27429
    },
    {
      "epoch": 2.615833282470703e-05,
      "step": 27429,
      "training_step_time": 9.729387044906616
    },
    {
      "epoch": 2.6159286499023437e-05,
      "grad_norm": 0.17067882418632507,
      "learning_rate": 8.686560445309118e-06,
      "loss": 0.0657,
      "step": 27430
    },
    {
      "epoch": 2.6159286499023437e-05,
      "model_forward_time": 1.7234890460968018,
      "step": 27430
    },
    {
      "dynamics_loss_avg": 0.052961761876940724,
      "epoch": 2.6159286499023437e-05,
      "step": 27430
    },
    {
      "action_loss_avg": 0.0033408492046874015,
      "epoch": 2.6159286499023437e-05,
      "step": 27430
    },
    {
      "epoch": 2.6159286499023437e-05,
      "step": 27430,
      "training_step_time": 9.278056621551514
    },
    {
      "epoch": 2.6160240173339845e-05,
      "model_forward_time": 1.730766773223877,
      "step": 27431
    },
    {
      "epoch": 2.6160240173339845e-05,
      "step": 27431,
      "training_step_time": 9.236672401428223
    },
    {
      "epoch": 2.616119384765625e-05,
      "model_forward_time": 1.7289609909057617,
      "step": 27432
    },
    {
      "epoch": 2.616119384765625e-05,
      "step": 27432,
      "training_step_time": 7.6553795337677
    },
    {
      "epoch": 2.6162147521972656e-05,
      "model_forward_time": 1.7521724700927734,
      "step": 27433
    },
    {
      "epoch": 2.6162147521972656e-05,
      "step": 27433,
      "training_step_time": 7.57351279258728
    },
    {
      "epoch": 2.6163101196289064e-05,
      "model_forward_time": 2.904792070388794,
      "step": 27434
    },
    {
      "epoch": 2.6163101196289064e-05,
      "step": 27434,
      "training_step_time": 7.644169092178345
    },
    {
      "epoch": 2.6164054870605468e-05,
      "model_forward_time": 1.7233386039733887,
      "step": 27435
    },
    {
      "epoch": 2.6164054870605468e-05,
      "step": 27435,
      "training_step_time": 8.264952898025513
    },
    {
      "epoch": 2.6165008544921875e-05,
      "model_forward_time": 1.7429611682891846,
      "step": 27436
    },
    {
      "epoch": 2.6165008544921875e-05,
      "step": 27436,
      "training_step_time": 7.602243423461914
    },
    {
      "epoch": 2.6165962219238282e-05,
      "model_forward_time": 1.7393288612365723,
      "step": 27437
    },
    {
      "epoch": 2.6165962219238282e-05,
      "step": 27437,
      "training_step_time": 8.560799598693848
    },
    {
      "epoch": 2.6166915893554686e-05,
      "model_forward_time": 1.73917555809021,
      "step": 27438
    },
    {
      "epoch": 2.6166915893554686e-05,
      "step": 27438,
      "training_step_time": 9.691862106323242
    },
    {
      "epoch": 2.6167869567871094e-05,
      "model_forward_time": 1.7404534816741943,
      "step": 27439
    },
    {
      "epoch": 2.6167869567871094e-05,
      "step": 27439,
      "training_step_time": 9.080041885375977
    },
    {
      "epoch": 2.61688232421875e-05,
      "grad_norm": 0.2068965882062912,
      "learning_rate": 8.6854432384455e-06,
      "loss": 0.0615,
      "step": 27440
    },
    {
      "epoch": 2.61688232421875e-05,
      "model_forward_time": 1.8214936256408691,
      "step": 27440
    },
    {
      "dynamics_loss_avg": 0.05692869089543819,
      "epoch": 2.61688232421875e-05,
      "step": 27440
    },
    {
      "action_loss_avg": 0.0023874628474004566,
      "epoch": 2.61688232421875e-05,
      "step": 27440
    },
    {
      "epoch": 2.61688232421875e-05,
      "step": 27440,
      "training_step_time": 7.799200773239136
    },
    {
      "epoch": 2.6169776916503905e-05,
      "model_forward_time": 1.7412662506103516,
      "step": 27441
    },
    {
      "epoch": 2.6169776916503905e-05,
      "step": 27441,
      "training_step_time": 7.650602340698242
    },
    {
      "epoch": 2.6170730590820313e-05,
      "model_forward_time": 1.740372896194458,
      "step": 27442
    },
    {
      "epoch": 2.6170730590820313e-05,
      "step": 27442,
      "training_step_time": 7.997141122817993
    },
    {
      "epoch": 2.617168426513672e-05,
      "model_forward_time": 2.901352643966675,
      "step": 27443
    },
    {
      "epoch": 2.617168426513672e-05,
      "step": 27443,
      "training_step_time": 7.691171646118164
    },
    {
      "epoch": 2.6172637939453124e-05,
      "model_forward_time": 1.74464750289917,
      "step": 27444
    },
    {
      "epoch": 2.6172637939453124e-05,
      "step": 27444,
      "training_step_time": 7.693598747253418
    },
    {
      "epoch": 2.617359161376953e-05,
      "model_forward_time": 1.7389237880706787,
      "step": 27445
    },
    {
      "epoch": 2.617359161376953e-05,
      "step": 27445,
      "training_step_time": 7.7026917934417725
    },
    {
      "epoch": 2.617454528808594e-05,
      "model_forward_time": 1.742750644683838,
      "step": 27446
    },
    {
      "epoch": 2.617454528808594e-05,
      "step": 27446,
      "training_step_time": 7.528636455535889
    },
    {
      "epoch": 2.6175498962402343e-05,
      "model_forward_time": 1.7454864978790283,
      "step": 27447
    },
    {
      "epoch": 2.6175498962402343e-05,
      "step": 27447,
      "training_step_time": 7.714688301086426
    },
    {
      "epoch": 2.617645263671875e-05,
      "model_forward_time": 1.7337548732757568,
      "step": 27448
    },
    {
      "epoch": 2.617645263671875e-05,
      "step": 27448,
      "training_step_time": 9.519390106201172
    },
    {
      "epoch": 2.6177406311035158e-05,
      "model_forward_time": 1.815535545349121,
      "step": 27449
    },
    {
      "epoch": 2.6177406311035158e-05,
      "step": 27449,
      "training_step_time": 8.678749084472656
    },
    {
      "epoch": 2.6178359985351562e-05,
      "grad_norm": 0.17739009857177734,
      "learning_rate": 8.68432562854735e-06,
      "loss": 0.0611,
      "step": 27450
    },
    {
      "epoch": 2.6178359985351562e-05,
      "model_forward_time": 1.7491767406463623,
      "step": 27450
    },
    {
      "dynamics_loss_avg": 0.05923137664794922,
      "epoch": 2.6178359985351562e-05,
      "step": 27450
    },
    {
      "action_loss_avg": 0.002826468972489238,
      "epoch": 2.6178359985351562e-05,
      "step": 27450
    },
    {
      "epoch": 2.6178359985351562e-05,
      "step": 27450,
      "training_step_time": 7.665441036224365
    },
    {
      "epoch": 2.617931365966797e-05,
      "model_forward_time": 1.740231990814209,
      "step": 27451
    },
    {
      "epoch": 2.617931365966797e-05,
      "step": 27451,
      "training_step_time": 7.401915550231934
    },
    {
      "epoch": 2.6180267333984377e-05,
      "model_forward_time": 2.847339153289795,
      "step": 27452
    },
    {
      "epoch": 2.6180267333984377e-05,
      "step": 27452,
      "training_step_time": 7.579244375228882
    },
    {
      "epoch": 2.618122100830078e-05,
      "model_forward_time": 1.755256175994873,
      "step": 27453
    },
    {
      "epoch": 2.618122100830078e-05,
      "step": 27453,
      "training_step_time": 7.708309650421143
    },
    {
      "epoch": 2.6182174682617188e-05,
      "model_forward_time": 1.7414567470550537,
      "step": 27454
    },
    {
      "epoch": 2.6182174682617188e-05,
      "step": 27454,
      "training_step_time": 7.591001033782959
    },
    {
      "epoch": 2.6183128356933592e-05,
      "model_forward_time": 1.73854660987854,
      "step": 27455
    },
    {
      "epoch": 2.6183128356933592e-05,
      "step": 27455,
      "training_step_time": 8.66523814201355
    },
    {
      "epoch": 2.618408203125e-05,
      "model_forward_time": 1.7444305419921875,
      "step": 27456
    },
    {
      "epoch": 2.618408203125e-05,
      "step": 27456,
      "training_step_time": 8.205749988555908
    },
    {
      "epoch": 2.6185035705566407e-05,
      "model_forward_time": 1.740295171737671,
      "step": 27457
    },
    {
      "epoch": 2.6185035705566407e-05,
      "step": 27457,
      "training_step_time": 7.762936115264893
    },
    {
      "epoch": 2.618598937988281e-05,
      "model_forward_time": 1.8113934993743896,
      "step": 27458
    },
    {
      "epoch": 2.618598937988281e-05,
      "step": 27458,
      "training_step_time": 8.792349815368652
    },
    {
      "epoch": 2.618694305419922e-05,
      "model_forward_time": 1.741572618484497,
      "step": 27459
    },
    {
      "epoch": 2.618694305419922e-05,
      "step": 27459,
      "training_step_time": 8.409364461898804
    },
    {
      "epoch": 2.6187896728515626e-05,
      "grad_norm": 0.15146875381469727,
      "learning_rate": 8.683207615736889e-06,
      "loss": 0.0636,
      "step": 27460
    },
    {
      "epoch": 2.6187896728515626e-05,
      "model_forward_time": 1.7383456230163574,
      "step": 27460
    },
    {
      "dynamics_loss_avg": 0.05283893095329404,
      "epoch": 2.6187896728515626e-05,
      "step": 27460
    },
    {
      "action_loss_avg": 0.002930843812646344,
      "epoch": 2.6187896728515626e-05,
      "step": 27460
    },
    {
      "epoch": 2.6187896728515626e-05,
      "step": 27460,
      "training_step_time": 7.6819024085998535
    },
    {
      "epoch": 2.618885040283203e-05,
      "model_forward_time": 2.876682758331299,
      "step": 27461
    },
    {
      "epoch": 2.618885040283203e-05,
      "step": 27461,
      "training_step_time": 7.726781129837036
    },
    {
      "epoch": 2.6189804077148437e-05,
      "model_forward_time": 1.7400541305541992,
      "step": 27462
    },
    {
      "epoch": 2.6189804077148437e-05,
      "step": 27462,
      "training_step_time": 7.76855993270874
    },
    {
      "epoch": 2.6190757751464845e-05,
      "model_forward_time": 1.3719546794891357,
      "step": 27463
    },
    {
      "epoch": 2.6190757751464845e-05,
      "step": 27463,
      "training_step_time": 7.5792906284332275
    },
    {
      "epoch": 2.619171142578125e-05,
      "model_forward_time": 1.7463219165802002,
      "step": 27464
    },
    {
      "epoch": 2.619171142578125e-05,
      "step": 27464,
      "training_step_time": 7.723909378051758
    },
    {
      "epoch": 2.6192665100097656e-05,
      "model_forward_time": 1.7412450313568115,
      "step": 27465
    },
    {
      "epoch": 2.6192665100097656e-05,
      "step": 27465,
      "training_step_time": 7.803773880004883
    },
    {
      "epoch": 2.6193618774414064e-05,
      "model_forward_time": 1.7370047569274902,
      "step": 27466
    },
    {
      "epoch": 2.6193618774414064e-05,
      "step": 27466,
      "training_step_time": 8.349448442459106
    },
    {
      "epoch": 2.6194572448730468e-05,
      "model_forward_time": 1.7856521606445312,
      "step": 27467
    },
    {
      "epoch": 2.6194572448730468e-05,
      "step": 27467,
      "training_step_time": 9.695534229278564
    },
    {
      "epoch": 2.6195526123046875e-05,
      "model_forward_time": 1.0331182479858398,
      "step": 27468
    },
    {
      "epoch": 2.6195526123046875e-05,
      "step": 27468,
      "training_step_time": 8.539046287536621
    },
    {
      "epoch": 2.6196479797363282e-05,
      "model_forward_time": 1.7377359867095947,
      "step": 27469
    },
    {
      "epoch": 2.6196479797363282e-05,
      "step": 27469,
      "training_step_time": 7.680124044418335
    },
    {
      "epoch": 2.6197433471679686e-05,
      "grad_norm": 0.17854785919189453,
      "learning_rate": 8.682089200136378e-06,
      "loss": 0.064,
      "step": 27470
    },
    {
      "epoch": 2.6197433471679686e-05,
      "model_forward_time": 1.7378296852111816,
      "step": 27470
    },
    {
      "dynamics_loss_avg": 0.04875671612098813,
      "epoch": 2.6197433471679686e-05,
      "step": 27470
    },
    {
      "action_loss_avg": 0.0019709457323187963,
      "epoch": 2.6197433471679686e-05,
      "step": 27470
    },
    {
      "epoch": 2.6197433471679686e-05,
      "step": 27470,
      "training_step_time": 7.677055358886719
    },
    {
      "epoch": 2.6198387145996094e-05,
      "model_forward_time": 2.8925280570983887,
      "step": 27471
    },
    {
      "epoch": 2.6198387145996094e-05,
      "step": 27471,
      "training_step_time": 8.242004156112671
    },
    {
      "epoch": 2.61993408203125e-05,
      "model_forward_time": 1.7427089214324951,
      "step": 27472
    },
    {
      "epoch": 2.61993408203125e-05,
      "step": 27472,
      "training_step_time": 7.616291284561157
    },
    {
      "epoch": 2.6200294494628905e-05,
      "model_forward_time": 1.7381067276000977,
      "step": 27473
    },
    {
      "epoch": 2.6200294494628905e-05,
      "step": 27473,
      "training_step_time": 7.704359531402588
    },
    {
      "epoch": 2.6201248168945313e-05,
      "model_forward_time": 1.7573800086975098,
      "step": 27474
    },
    {
      "epoch": 2.6201248168945313e-05,
      "step": 27474,
      "training_step_time": 8.18799376487732
    },
    {
      "epoch": 2.620220184326172e-05,
      "model_forward_time": 1.7394945621490479,
      "step": 27475
    },
    {
      "epoch": 2.620220184326172e-05,
      "step": 27475,
      "training_step_time": 8.43502950668335
    },
    {
      "epoch": 2.6203155517578124e-05,
      "model_forward_time": 1.0324621200561523,
      "step": 27476
    },
    {
      "epoch": 2.6203155517578124e-05,
      "step": 27476,
      "training_step_time": 8.680612802505493
    },
    {
      "epoch": 2.620410919189453e-05,
      "model_forward_time": 1.7379155158996582,
      "step": 27477
    },
    {
      "epoch": 2.620410919189453e-05,
      "step": 27477,
      "training_step_time": 8.238536834716797
    },
    {
      "epoch": 2.620506286621094e-05,
      "model_forward_time": 1.7437810897827148,
      "step": 27478
    },
    {
      "epoch": 2.620506286621094e-05,
      "step": 27478,
      "training_step_time": 7.6828813552856445
    },
    {
      "epoch": 2.6206016540527343e-05,
      "model_forward_time": 1.7439632415771484,
      "step": 27479
    },
    {
      "epoch": 2.6206016540527343e-05,
      "step": 27479,
      "training_step_time": 7.329334497451782
    },
    {
      "epoch": 2.620697021484375e-05,
      "grad_norm": 0.18912188708782196,
      "learning_rate": 8.680970381868131e-06,
      "loss": 0.0616,
      "step": 27480
    },
    {
      "epoch": 2.620697021484375e-05,
      "model_forward_time": 2.825812816619873,
      "step": 27480
    },
    {
      "dynamics_loss_avg": 0.060141774266958235,
      "epoch": 2.620697021484375e-05,
      "step": 27480
    },
    {
      "action_loss_avg": 0.0036252794438041747,
      "epoch": 2.620697021484375e-05,
      "step": 27480
    },
    {
      "epoch": 2.620697021484375e-05,
      "step": 27480,
      "training_step_time": 7.686105966567993
    },
    {
      "epoch": 2.6207923889160158e-05,
      "model_forward_time": 1.742065191268921,
      "step": 27481
    },
    {
      "epoch": 2.6207923889160158e-05,
      "step": 27481,
      "training_step_time": 7.740090370178223
    },
    {
      "epoch": 2.6208877563476562e-05,
      "model_forward_time": 1.7389187812805176,
      "step": 27482
    },
    {
      "epoch": 2.6208877563476562e-05,
      "step": 27482,
      "training_step_time": 7.713025093078613
    },
    {
      "epoch": 2.620983123779297e-05,
      "model_forward_time": 1.7369744777679443,
      "step": 27483
    },
    {
      "epoch": 2.620983123779297e-05,
      "step": 27483,
      "training_step_time": 8.652838945388794
    },
    {
      "epoch": 2.6210784912109377e-05,
      "model_forward_time": 1.7332828044891357,
      "step": 27484
    },
    {
      "epoch": 2.6210784912109377e-05,
      "step": 27484,
      "training_step_time": 8.444786071777344
    },
    {
      "epoch": 2.621173858642578e-05,
      "model_forward_time": 1.7335915565490723,
      "step": 27485
    },
    {
      "epoch": 2.621173858642578e-05,
      "step": 27485,
      "training_step_time": 8.923800945281982
    },
    {
      "epoch": 2.6212692260742188e-05,
      "model_forward_time": 1.8010823726654053,
      "step": 27486
    },
    {
      "epoch": 2.6212692260742188e-05,
      "step": 27486,
      "training_step_time": 8.481662034988403
    },
    {
      "epoch": 2.6213645935058592e-05,
      "model_forward_time": 1.7436234951019287,
      "step": 27487
    },
    {
      "epoch": 2.6213645935058592e-05,
      "step": 27487,
      "training_step_time": 7.7469542026519775
    },
    {
      "epoch": 2.6214599609375e-05,
      "model_forward_time": 1.731905221939087,
      "step": 27488
    },
    {
      "epoch": 2.6214599609375e-05,
      "step": 27488,
      "training_step_time": 7.651896715164185
    },
    {
      "epoch": 2.6215553283691407e-05,
      "model_forward_time": 1.7463576793670654,
      "step": 27489
    },
    {
      "epoch": 2.6215553283691407e-05,
      "step": 27489,
      "training_step_time": 7.720420598983765
    },
    {
      "epoch": 2.621650695800781e-05,
      "grad_norm": 0.13223044574260712,
      "learning_rate": 8.679851161054498e-06,
      "loss": 0.0649,
      "step": 27490
    },
    {
      "epoch": 2.621650695800781e-05,
      "model_forward_time": 2.9009218215942383,
      "step": 27490
    },
    {
      "dynamics_loss_avg": 0.06661536060273647,
      "epoch": 2.621650695800781e-05,
      "step": 27490
    },
    {
      "action_loss_avg": 0.002471672062529251,
      "epoch": 2.621650695800781e-05,
      "step": 27490
    },
    {
      "epoch": 2.621650695800781e-05,
      "step": 27490,
      "training_step_time": 7.783901691436768
    },
    {
      "epoch": 2.621746063232422e-05,
      "model_forward_time": 1.7446656227111816,
      "step": 27491
    },
    {
      "epoch": 2.621746063232422e-05,
      "step": 27491,
      "training_step_time": 8.451284885406494
    },
    {
      "epoch": 2.6218414306640626e-05,
      "model_forward_time": 1.7402980327606201,
      "step": 27492
    },
    {
      "epoch": 2.6218414306640626e-05,
      "step": 27492,
      "training_step_time": 7.67753005027771
    },
    {
      "epoch": 2.621936798095703e-05,
      "model_forward_time": 1.7421762943267822,
      "step": 27493
    },
    {
      "epoch": 2.621936798095703e-05,
      "step": 27493,
      "training_step_time": 9.138046503067017
    },
    {
      "epoch": 2.6220321655273437e-05,
      "model_forward_time": 1.7333595752716064,
      "step": 27494
    },
    {
      "epoch": 2.6220321655273437e-05,
      "step": 27494,
      "training_step_time": 7.723965406417847
    },
    {
      "epoch": 2.6221275329589845e-05,
      "model_forward_time": 1.8079860210418701,
      "step": 27495
    },
    {
      "epoch": 2.6221275329589845e-05,
      "step": 27495,
      "training_step_time": 7.709545612335205
    },
    {
      "epoch": 2.622222900390625e-05,
      "model_forward_time": 1.7307989597320557,
      "step": 27496
    },
    {
      "epoch": 2.622222900390625e-05,
      "step": 27496,
      "training_step_time": 8.49168062210083
    },
    {
      "epoch": 2.6223182678222656e-05,
      "model_forward_time": 1.7333135604858398,
      "step": 27497
    },
    {
      "epoch": 2.6223182678222656e-05,
      "step": 27497,
      "training_step_time": 7.53132438659668
    },
    {
      "epoch": 2.6224136352539064e-05,
      "model_forward_time": 1.7374341487884521,
      "step": 27498
    },
    {
      "epoch": 2.6224136352539064e-05,
      "step": 27498,
      "training_step_time": 7.076096773147583
    },
    {
      "epoch": 2.6225090026855468e-05,
      "model_forward_time": 2.8625664710998535,
      "step": 27499
    },
    {
      "epoch": 2.6225090026855468e-05,
      "step": 27499,
      "training_step_time": 8.638837337493896
    },
    {
      "epoch": 2.6226043701171875e-05,
      "grad_norm": 0.1308167725801468,
      "learning_rate": 8.678731537817873e-06,
      "loss": 0.0613,
      "step": 27500
    },
    {
      "epoch": 2.6226043701171875e-05,
      "model_forward_time": 1.7392911911010742,
      "step": 27500
    },
    {
      "dynamics_loss_avg": 0.0481606675311923,
      "epoch": 2.6226043701171875e-05,
      "step": 27500
    },
    {
      "action_loss_avg": 0.0022049768012948334,
      "epoch": 2.6226043701171875e-05,
      "step": 27500
    },
    {
      "epoch": 2.6226043701171875e-05,
      "step": 27500,
      "training_step_time": 8.54560399055481
    },
    {
      "epoch": 2.6226997375488282e-05,
      "model_forward_time": 1.737334966659546,
      "step": 27501
    },
    {
      "epoch": 2.6226997375488282e-05,
      "step": 27501,
      "training_step_time": 7.643188238143921
    },
    {
      "epoch": 2.6227951049804686e-05,
      "model_forward_time": 1.7392704486846924,
      "step": 27502
    },
    {
      "epoch": 2.6227951049804686e-05,
      "step": 27502,
      "training_step_time": 7.775755882263184
    },
    {
      "epoch": 2.6228904724121094e-05,
      "model_forward_time": 1.7359087467193604,
      "step": 27503
    },
    {
      "epoch": 2.6228904724121094e-05,
      "step": 27503,
      "training_step_time": 7.729344129562378
    },
    {
      "epoch": 2.62298583984375e-05,
      "model_forward_time": 1.808126449584961,
      "step": 27504
    },
    {
      "epoch": 2.62298583984375e-05,
      "step": 27504,
      "training_step_time": 9.333089113235474
    },
    {
      "epoch": 2.6230812072753905e-05,
      "model_forward_time": 1.7517812252044678,
      "step": 27505
    },
    {
      "epoch": 2.6230812072753905e-05,
      "step": 27505,
      "training_step_time": 7.696512460708618
    },
    {
      "epoch": 2.6231765747070313e-05,
      "model_forward_time": 1.7321257591247559,
      "step": 27506
    },
    {
      "epoch": 2.6231765747070313e-05,
      "step": 27506,
      "training_step_time": 7.778763771057129
    },
    {
      "epoch": 2.623271942138672e-05,
      "model_forward_time": 1.7455775737762451,
      "step": 27507
    },
    {
      "epoch": 2.623271942138672e-05,
      "step": 27507,
      "training_step_time": 7.269359588623047
    },
    {
      "epoch": 2.6233673095703124e-05,
      "model_forward_time": 2.897794008255005,
      "step": 27508
    },
    {
      "epoch": 2.6233673095703124e-05,
      "step": 27508,
      "training_step_time": 7.69970965385437
    },
    {
      "epoch": 2.623462677001953e-05,
      "model_forward_time": 1.7441747188568115,
      "step": 27509
    },
    {
      "epoch": 2.623462677001953e-05,
      "step": 27509,
      "training_step_time": 8.441259860992432
    },
    {
      "epoch": 2.623558044433594e-05,
      "grad_norm": 0.19177868962287903,
      "learning_rate": 8.677611512280697e-06,
      "loss": 0.0633,
      "step": 27510
    },
    {
      "epoch": 2.623558044433594e-05,
      "model_forward_time": 1.7211124897003174,
      "step": 27510
    },
    {
      "dynamics_loss_avg": 0.0665081387385726,
      "epoch": 2.623558044433594e-05,
      "step": 27510
    },
    {
      "action_loss_avg": 0.0032868479611352086,
      "epoch": 2.623558044433594e-05,
      "step": 27510
    },
    {
      "epoch": 2.623558044433594e-05,
      "step": 27510,
      "training_step_time": 7.726099014282227
    },
    {
      "epoch": 2.6236534118652343e-05,
      "model_forward_time": 1.7347242832183838,
      "step": 27511
    },
    {
      "epoch": 2.6236534118652343e-05,
      "step": 27511,
      "training_step_time": 8.650035381317139
    },
    {
      "epoch": 2.623748779296875e-05,
      "model_forward_time": 1.7336599826812744,
      "step": 27512
    },
    {
      "epoch": 2.623748779296875e-05,
      "step": 27512,
      "training_step_time": 8.192315101623535
    },
    {
      "epoch": 2.6238441467285158e-05,
      "model_forward_time": 1.0268027782440186,
      "step": 27513
    },
    {
      "epoch": 2.6238441467285158e-05,
      "step": 27513,
      "training_step_time": 10.04000973701477
    },
    {
      "epoch": 2.6239395141601562e-05,
      "model_forward_time": 1.7640907764434814,
      "step": 27514
    },
    {
      "epoch": 2.6239395141601562e-05,
      "step": 27514,
      "training_step_time": 8.517964601516724
    },
    {
      "epoch": 2.624034881591797e-05,
      "model_forward_time": 1.7637693881988525,
      "step": 27515
    },
    {
      "epoch": 2.624034881591797e-05,
      "step": 27515,
      "training_step_time": 7.930908441543579
    },
    {
      "epoch": 2.6241302490234373e-05,
      "model_forward_time": 1.7386672496795654,
      "step": 27516
    },
    {
      "epoch": 2.6241302490234373e-05,
      "step": 27516,
      "training_step_time": 7.575325965881348
    },
    {
      "epoch": 2.624225616455078e-05,
      "model_forward_time": 1.7396621704101562,
      "step": 27517
    },
    {
      "epoch": 2.624225616455078e-05,
      "step": 27517,
      "training_step_time": 8.633903741836548
    },
    {
      "epoch": 2.6243209838867188e-05,
      "model_forward_time": 1.735656976699829,
      "step": 27518
    },
    {
      "epoch": 2.6243209838867188e-05,
      "step": 27518,
      "training_step_time": 8.52236795425415
    },
    {
      "epoch": 2.6244163513183592e-05,
      "model_forward_time": 1.7257170677185059,
      "step": 27519
    },
    {
      "epoch": 2.6244163513183592e-05,
      "step": 27519,
      "training_step_time": 7.654012680053711
    },
    {
      "epoch": 2.62451171875e-05,
      "grad_norm": 0.15201513469219208,
      "learning_rate": 8.676491084565456e-06,
      "loss": 0.0637,
      "step": 27520
    },
    {
      "epoch": 2.62451171875e-05,
      "model_forward_time": 1.738361120223999,
      "step": 27520
    },
    {
      "dynamics_loss_avg": 0.06728851106017827,
      "epoch": 2.62451171875e-05,
      "step": 27520
    },
    {
      "action_loss_avg": 0.002368409326300025,
      "epoch": 2.62451171875e-05,
      "step": 27520
    },
    {
      "epoch": 2.62451171875e-05,
      "step": 27520,
      "training_step_time": 8.534198999404907
    },
    {
      "epoch": 2.6246070861816407e-05,
      "model_forward_time": 1.7371172904968262,
      "step": 27521
    },
    {
      "epoch": 2.6246070861816407e-05,
      "step": 27521,
      "training_step_time": 8.38138484954834
    },
    {
      "epoch": 2.624702453613281e-05,
      "model_forward_time": 1.747222661972046,
      "step": 27522
    },
    {
      "epoch": 2.624702453613281e-05,
      "step": 27522,
      "training_step_time": 10.045578479766846
    },
    {
      "epoch": 2.624797821044922e-05,
      "model_forward_time": 1.831197738647461,
      "step": 27523
    },
    {
      "epoch": 2.624797821044922e-05,
      "step": 27523,
      "training_step_time": 8.50310730934143
    },
    {
      "epoch": 2.6248931884765626e-05,
      "model_forward_time": 1.7409093379974365,
      "step": 27524
    },
    {
      "epoch": 2.6248931884765626e-05,
      "step": 27524,
      "training_step_time": 8.438569068908691
    },
    {
      "epoch": 2.624988555908203e-05,
      "model_forward_time": 1.7376680374145508,
      "step": 27525
    },
    {
      "epoch": 2.624988555908203e-05,
      "step": 27525,
      "training_step_time": 7.614696502685547
    },
    {
      "epoch": 2.6250839233398437e-05,
      "model_forward_time": 1.7461752891540527,
      "step": 27526
    },
    {
      "epoch": 2.6250839233398437e-05,
      "step": 27526,
      "training_step_time": 7.595122814178467
    },
    {
      "epoch": 2.6251792907714845e-05,
      "model_forward_time": 2.8611528873443604,
      "step": 27527
    },
    {
      "epoch": 2.6251792907714845e-05,
      "step": 27527,
      "training_step_time": 7.634469270706177
    },
    {
      "epoch": 2.625274658203125e-05,
      "model_forward_time": 1.740837812423706,
      "step": 27528
    },
    {
      "epoch": 2.625274658203125e-05,
      "step": 27528,
      "training_step_time": 8.588696241378784
    },
    {
      "epoch": 2.6253700256347656e-05,
      "model_forward_time": 1.7452709674835205,
      "step": 27529
    },
    {
      "epoch": 2.6253700256347656e-05,
      "step": 27529,
      "training_step_time": 8.690049171447754
    },
    {
      "epoch": 2.6254653930664064e-05,
      "grad_norm": 0.13969464600086212,
      "learning_rate": 8.675370254794678e-06,
      "loss": 0.0658,
      "step": 27530
    },
    {
      "epoch": 2.6254653930664064e-05,
      "model_forward_time": 1.7367017269134521,
      "step": 27530
    },
    {
      "dynamics_loss_avg": 0.055385091342031956,
      "epoch": 2.6254653930664064e-05,
      "step": 27530
    },
    {
      "action_loss_avg": 0.0021547104290220885,
      "epoch": 2.6254653930664064e-05,
      "step": 27530
    },
    {
      "epoch": 2.6254653930664064e-05,
      "step": 27530,
      "training_step_time": 8.632620573043823
    },
    {
      "epoch": 2.6255607604980468e-05,
      "model_forward_time": 1.7451016902923584,
      "step": 27531
    },
    {
      "epoch": 2.6255607604980468e-05,
      "step": 27531,
      "training_step_time": 8.193567752838135
    },
    {
      "epoch": 2.6256561279296875e-05,
      "model_forward_time": 1.8025846481323242,
      "step": 27532
    },
    {
      "epoch": 2.6256561279296875e-05,
      "step": 27532,
      "training_step_time": 8.54100751876831
    },
    {
      "epoch": 2.6257514953613282e-05,
      "model_forward_time": 1.741903305053711,
      "step": 27533
    },
    {
      "epoch": 2.6257514953613282e-05,
      "step": 27533,
      "training_step_time": 9.36324691772461
    },
    {
      "epoch": 2.6258468627929686e-05,
      "model_forward_time": 1.7460126876831055,
      "step": 27534
    },
    {
      "epoch": 2.6258468627929686e-05,
      "step": 27534,
      "training_step_time": 7.689919471740723
    },
    {
      "epoch": 2.6259422302246094e-05,
      "model_forward_time": 1.7444982528686523,
      "step": 27535
    },
    {
      "epoch": 2.6259422302246094e-05,
      "step": 27535,
      "training_step_time": 7.666261672973633
    },
    {
      "epoch": 2.62603759765625e-05,
      "model_forward_time": 2.8888697624206543,
      "step": 27536
    },
    {
      "epoch": 2.62603759765625e-05,
      "step": 27536,
      "training_step_time": 7.75273871421814
    },
    {
      "epoch": 2.6261329650878905e-05,
      "model_forward_time": 1.7419850826263428,
      "step": 27537
    },
    {
      "epoch": 2.6261329650878905e-05,
      "step": 27537,
      "training_step_time": 7.634826898574829
    },
    {
      "epoch": 2.6262283325195313e-05,
      "model_forward_time": 1.7301204204559326,
      "step": 27538
    },
    {
      "epoch": 2.6262283325195313e-05,
      "step": 27538,
      "training_step_time": 7.76941704750061
    },
    {
      "epoch": 2.626323699951172e-05,
      "model_forward_time": 1.7457258701324463,
      "step": 27539
    },
    {
      "epoch": 2.626323699951172e-05,
      "step": 27539,
      "training_step_time": 8.716801166534424
    },
    {
      "epoch": 2.6264190673828124e-05,
      "grad_norm": 0.21833617985248566,
      "learning_rate": 8.674249023090934e-06,
      "loss": 0.064,
      "step": 27540
    },
    {
      "epoch": 2.6264190673828124e-05,
      "model_forward_time": 1.7442924976348877,
      "step": 27540
    },
    {
      "dynamics_loss_avg": 0.055361091159284115,
      "epoch": 2.6264190673828124e-05,
      "step": 27540
    },
    {
      "action_loss_avg": 0.0029645390750374645,
      "epoch": 2.6264190673828124e-05,
      "step": 27540
    },
    {
      "epoch": 2.6264190673828124e-05,
      "step": 27540,
      "training_step_time": 9.40298080444336
    },
    {
      "epoch": 2.626514434814453e-05,
      "model_forward_time": 1.8096683025360107,
      "step": 27541
    },
    {
      "epoch": 2.626514434814453e-05,
      "step": 27541,
      "training_step_time": 9.630964040756226
    },
    {
      "epoch": 2.626609802246094e-05,
      "model_forward_time": 1.747253179550171,
      "step": 27542
    },
    {
      "epoch": 2.626609802246094e-05,
      "step": 27542,
      "training_step_time": 8.863727331161499
    },
    {
      "epoch": 2.6267051696777343e-05,
      "model_forward_time": 1.7363715171813965,
      "step": 27543
    },
    {
      "epoch": 2.6267051696777343e-05,
      "step": 27543,
      "training_step_time": 7.951455593109131
    },
    {
      "epoch": 2.626800537109375e-05,
      "model_forward_time": 1.740293264389038,
      "step": 27544
    },
    {
      "epoch": 2.626800537109375e-05,
      "step": 27544,
      "training_step_time": 7.705036401748657
    },
    {
      "epoch": 2.6268959045410158e-05,
      "model_forward_time": 2.9018795490264893,
      "step": 27545
    },
    {
      "epoch": 2.6268959045410158e-05,
      "step": 27545,
      "training_step_time": 7.7506773471832275
    },
    {
      "epoch": 2.6269912719726562e-05,
      "model_forward_time": 1.7385096549987793,
      "step": 27546
    },
    {
      "epoch": 2.6269912719726562e-05,
      "step": 27546,
      "training_step_time": 7.630040884017944
    },
    {
      "epoch": 2.627086639404297e-05,
      "model_forward_time": 1.7368931770324707,
      "step": 27547
    },
    {
      "epoch": 2.627086639404297e-05,
      "step": 27547,
      "training_step_time": 8.192346572875977
    },
    {
      "epoch": 2.6271820068359373e-05,
      "model_forward_time": 1.7307460308074951,
      "step": 27548
    },
    {
      "epoch": 2.6271820068359373e-05,
      "step": 27548,
      "training_step_time": 8.773336410522461
    },
    {
      "epoch": 2.627277374267578e-05,
      "model_forward_time": 1.7367863655090332,
      "step": 27549
    },
    {
      "epoch": 2.627277374267578e-05,
      "step": 27549,
      "training_step_time": 8.166279792785645
    },
    {
      "epoch": 2.6273727416992188e-05,
      "grad_norm": 0.18316981196403503,
      "learning_rate": 8.673127389576843e-06,
      "loss": 0.0649,
      "step": 27550
    },
    {
      "epoch": 2.6273727416992188e-05,
      "model_forward_time": 1.7799222469329834,
      "step": 27550
    },
    {
      "dynamics_loss_avg": 0.07549723479896783,
      "epoch": 2.6273727416992188e-05,
      "step": 27550
    },
    {
      "action_loss_avg": 0.003200993570499122,
      "epoch": 2.6273727416992188e-05,
      "step": 27550
    },
    {
      "epoch": 2.6273727416992188e-05,
      "step": 27550,
      "training_step_time": 7.842385292053223
    },
    {
      "epoch": 2.6274681091308592e-05,
      "model_forward_time": 1.7363183498382568,
      "step": 27551
    },
    {
      "epoch": 2.6274681091308592e-05,
      "step": 27551,
      "training_step_time": 7.789090394973755
    },
    {
      "epoch": 2.6275634765625e-05,
      "model_forward_time": 1.7400877475738525,
      "step": 27552
    },
    {
      "epoch": 2.6275634765625e-05,
      "step": 27552,
      "training_step_time": 7.984256267547607
    },
    {
      "epoch": 2.6276588439941407e-05,
      "model_forward_time": 1.7395117282867432,
      "step": 27553
    },
    {
      "epoch": 2.6276588439941407e-05,
      "step": 27553,
      "training_step_time": 7.6429054737091064
    },
    {
      "epoch": 2.627754211425781e-05,
      "model_forward_time": 2.9004032611846924,
      "step": 27554
    },
    {
      "epoch": 2.627754211425781e-05,
      "step": 27554,
      "training_step_time": 7.73381233215332
    },
    {
      "epoch": 2.627849578857422e-05,
      "model_forward_time": 1.7352776527404785,
      "step": 27555
    },
    {
      "epoch": 2.627849578857422e-05,
      "step": 27555,
      "training_step_time": 7.601586103439331
    },
    {
      "epoch": 2.6279449462890626e-05,
      "model_forward_time": 1.735485553741455,
      "step": 27556
    },
    {
      "epoch": 2.6279449462890626e-05,
      "step": 27556,
      "training_step_time": 7.682660341262817
    },
    {
      "epoch": 2.628040313720703e-05,
      "model_forward_time": 1.7293212413787842,
      "step": 27557
    },
    {
      "epoch": 2.628040313720703e-05,
      "step": 27557,
      "training_step_time": 7.814368724822998
    },
    {
      "epoch": 2.6281356811523437e-05,
      "model_forward_time": 1.7391202449798584,
      "step": 27558
    },
    {
      "epoch": 2.6281356811523437e-05,
      "step": 27558,
      "training_step_time": 8.178512573242188
    },
    {
      "epoch": 2.6282310485839845e-05,
      "model_forward_time": 1.733710527420044,
      "step": 27559
    },
    {
      "epoch": 2.6282310485839845e-05,
      "step": 27559,
      "training_step_time": 9.270381689071655
    },
    {
      "epoch": 2.628326416015625e-05,
      "grad_norm": 0.12033085525035858,
      "learning_rate": 8.672005354375061e-06,
      "loss": 0.0615,
      "step": 27560
    },
    {
      "epoch": 2.628326416015625e-05,
      "model_forward_time": 1.7367351055145264,
      "step": 27560
    },
    {
      "dynamics_loss_avg": 0.05431896336376667,
      "epoch": 2.628326416015625e-05,
      "step": 27560
    },
    {
      "action_loss_avg": 0.002369359147269279,
      "epoch": 2.628326416015625e-05,
      "step": 27560
    },
    {
      "epoch": 2.628326416015625e-05,
      "step": 27560,
      "training_step_time": 7.821903705596924
    },
    {
      "epoch": 2.6284217834472656e-05,
      "model_forward_time": 1.7360751628875732,
      "step": 27561
    },
    {
      "epoch": 2.6284217834472656e-05,
      "step": 27561,
      "training_step_time": 9.340001583099365
    },
    {
      "epoch": 2.6285171508789064e-05,
      "model_forward_time": 1.7366583347320557,
      "step": 27562
    },
    {
      "epoch": 2.6285171508789064e-05,
      "step": 27562,
      "training_step_time": 8.512657880783081
    },
    {
      "epoch": 2.6286125183105468e-05,
      "model_forward_time": 2.8541111946105957,
      "step": 27563
    },
    {
      "epoch": 2.6286125183105468e-05,
      "step": 27563,
      "training_step_time": 7.694123268127441
    },
    {
      "epoch": 2.6287078857421875e-05,
      "model_forward_time": 1.764251947402954,
      "step": 27564
    },
    {
      "epoch": 2.6287078857421875e-05,
      "step": 27564,
      "training_step_time": 7.597520589828491
    },
    {
      "epoch": 2.6288032531738282e-05,
      "model_forward_time": 1.7684037685394287,
      "step": 27565
    },
    {
      "epoch": 2.6288032531738282e-05,
      "step": 27565,
      "training_step_time": 9.257212162017822
    },
    {
      "epoch": 2.6288986206054686e-05,
      "model_forward_time": 1.7330965995788574,
      "step": 27566
    },
    {
      "epoch": 2.6288986206054686e-05,
      "step": 27566,
      "training_step_time": 9.631819725036621
    },
    {
      "epoch": 2.6289939880371094e-05,
      "model_forward_time": 1.7336299419403076,
      "step": 27567
    },
    {
      "epoch": 2.6289939880371094e-05,
      "step": 27567,
      "training_step_time": 7.6809797286987305
    },
    {
      "epoch": 2.62908935546875e-05,
      "model_forward_time": 1.7261183261871338,
      "step": 27568
    },
    {
      "epoch": 2.62908935546875e-05,
      "step": 27568,
      "training_step_time": 9.701675415039062
    },
    {
      "epoch": 2.6291847229003905e-05,
      "model_forward_time": 1.8043982982635498,
      "step": 27569
    },
    {
      "epoch": 2.6291847229003905e-05,
      "step": 27569,
      "training_step_time": 7.77060079574585
    },
    {
      "epoch": 2.6292800903320313e-05,
      "grad_norm": 0.12270218878984451,
      "learning_rate": 8.670882917608296e-06,
      "loss": 0.0648,
      "step": 27570
    },
    {
      "epoch": 2.6292800903320313e-05,
      "model_forward_time": 1.7359659671783447,
      "step": 27570
    },
    {
      "dynamics_loss_avg": 0.06683998061344028,
      "epoch": 2.6292800903320313e-05,
      "step": 27570
    },
    {
      "action_loss_avg": 0.004529123229440301,
      "epoch": 2.6292800903320313e-05,
      "step": 27570
    },
    {
      "epoch": 2.6292800903320313e-05,
      "step": 27570,
      "training_step_time": 7.707934141159058
    },
    {
      "epoch": 2.629375457763672e-05,
      "model_forward_time": 1.7293341159820557,
      "step": 27571
    },
    {
      "epoch": 2.629375457763672e-05,
      "step": 27571,
      "training_step_time": 8.734750032424927
    },
    {
      "epoch": 2.6294708251953124e-05,
      "model_forward_time": 2.8462741374969482,
      "step": 27572
    },
    {
      "epoch": 2.6294708251953124e-05,
      "step": 27572,
      "training_step_time": 7.659815788269043
    },
    {
      "epoch": 2.629566192626953e-05,
      "model_forward_time": 1.734006643295288,
      "step": 27573
    },
    {
      "epoch": 2.629566192626953e-05,
      "step": 27573,
      "training_step_time": 7.575289964675903
    },
    {
      "epoch": 2.629661560058594e-05,
      "model_forward_time": 1.7465400695800781,
      "step": 27574
    },
    {
      "epoch": 2.629661560058594e-05,
      "step": 27574,
      "training_step_time": 7.664305686950684
    },
    {
      "epoch": 2.6297569274902343e-05,
      "model_forward_time": 1.7373933792114258,
      "step": 27575
    },
    {
      "epoch": 2.6297569274902343e-05,
      "step": 27575,
      "training_step_time": 7.72957968711853
    },
    {
      "epoch": 2.629852294921875e-05,
      "model_forward_time": 1.7466042041778564,
      "step": 27576
    },
    {
      "epoch": 2.629852294921875e-05,
      "step": 27576,
      "training_step_time": 7.6762635707855225
    },
    {
      "epoch": 2.6299476623535158e-05,
      "model_forward_time": 1.7301244735717773,
      "step": 27577
    },
    {
      "epoch": 2.6299476623535158e-05,
      "step": 27577,
      "training_step_time": 8.849901914596558
    },
    {
      "epoch": 2.6300430297851562e-05,
      "model_forward_time": 1.8013715744018555,
      "step": 27578
    },
    {
      "epoch": 2.6300430297851562e-05,
      "step": 27578,
      "training_step_time": 7.753648042678833
    },
    {
      "epoch": 2.630138397216797e-05,
      "model_forward_time": 1.744398832321167,
      "step": 27579
    },
    {
      "epoch": 2.630138397216797e-05,
      "step": 27579,
      "training_step_time": 7.747247695922852
    },
    {
      "epoch": 2.6302337646484373e-05,
      "grad_norm": 0.14533731341362,
      "learning_rate": 8.669760079399293e-06,
      "loss": 0.0608,
      "step": 27580
    },
    {
      "epoch": 2.6302337646484373e-05,
      "model_forward_time": 1.7667911052703857,
      "step": 27580
    },
    {
      "dynamics_loss_avg": 0.06607544422149658,
      "epoch": 2.6302337646484373e-05,
      "step": 27580
    },
    {
      "action_loss_avg": 0.0022045872174203396,
      "epoch": 2.6302337646484373e-05,
      "step": 27580
    },
    {
      "epoch": 2.6302337646484373e-05,
      "step": 27580,
      "training_step_time": 7.719097137451172
    },
    {
      "epoch": 2.630329132080078e-05,
      "model_forward_time": 2.9033796787261963,
      "step": 27581
    },
    {
      "epoch": 2.630329132080078e-05,
      "step": 27581,
      "training_step_time": 7.696122169494629
    },
    {
      "epoch": 2.6304244995117188e-05,
      "model_forward_time": 1.7448530197143555,
      "step": 27582
    },
    {
      "epoch": 2.6304244995117188e-05,
      "step": 27582,
      "training_step_time": 7.610411882400513
    },
    {
      "epoch": 2.6305198669433592e-05,
      "model_forward_time": 1.7420899868011475,
      "step": 27583
    },
    {
      "epoch": 2.6305198669433592e-05,
      "step": 27583,
      "training_step_time": 7.724560737609863
    },
    {
      "epoch": 2.630615234375e-05,
      "model_forward_time": 1.7623114585876465,
      "step": 27584
    },
    {
      "epoch": 2.630615234375e-05,
      "step": 27584,
      "training_step_time": 7.676819801330566
    },
    {
      "epoch": 2.6307106018066407e-05,
      "model_forward_time": 1.7460224628448486,
      "step": 27585
    },
    {
      "epoch": 2.6307106018066407e-05,
      "step": 27585,
      "training_step_time": 8.562983512878418
    },
    {
      "epoch": 2.630805969238281e-05,
      "model_forward_time": 1.7345595359802246,
      "step": 27586
    },
    {
      "epoch": 2.630805969238281e-05,
      "step": 27586,
      "training_step_time": 8.1539945602417
    },
    {
      "epoch": 2.630901336669922e-05,
      "model_forward_time": 1.738832950592041,
      "step": 27587
    },
    {
      "epoch": 2.630901336669922e-05,
      "step": 27587,
      "training_step_time": 8.393016815185547
    },
    {
      "epoch": 2.6309967041015626e-05,
      "model_forward_time": 1.8167152404785156,
      "step": 27588
    },
    {
      "epoch": 2.6309967041015626e-05,
      "step": 27588,
      "training_step_time": 7.719980001449585
    },
    {
      "epoch": 2.631092071533203e-05,
      "model_forward_time": 1.7380542755126953,
      "step": 27589
    },
    {
      "epoch": 2.631092071533203e-05,
      "step": 27589,
      "training_step_time": 8.664906978607178
    },
    {
      "epoch": 2.6311874389648437e-05,
      "grad_norm": 0.19514937698841095,
      "learning_rate": 8.668636839870845e-06,
      "loss": 0.0622,
      "step": 27590
    },
    {
      "epoch": 2.6311874389648437e-05,
      "model_forward_time": 2.5175445079803467,
      "step": 27590
    },
    {
      "dynamics_loss_avg": 0.0676440903916955,
      "epoch": 2.6311874389648437e-05,
      "step": 27590
    },
    {
      "action_loss_avg": 0.0033065575698856264,
      "epoch": 2.6311874389648437e-05,
      "step": 27590
    },
    {
      "epoch": 2.6311874389648437e-05,
      "step": 27590,
      "training_step_time": 7.604000091552734
    },
    {
      "epoch": 2.6312828063964845e-05,
      "model_forward_time": 1.367685317993164,
      "step": 27591
    },
    {
      "epoch": 2.6312828063964845e-05,
      "step": 27591,
      "training_step_time": 7.608221769332886
    },
    {
      "epoch": 2.631378173828125e-05,
      "model_forward_time": 1.7420709133148193,
      "step": 27592
    },
    {
      "epoch": 2.631378173828125e-05,
      "step": 27592,
      "training_step_time": 8.206429243087769
    },
    {
      "epoch": 2.6314735412597656e-05,
      "model_forward_time": 1.7371325492858887,
      "step": 27593
    },
    {
      "epoch": 2.6314735412597656e-05,
      "step": 27593,
      "training_step_time": 7.726949214935303
    },
    {
      "epoch": 2.6315689086914064e-05,
      "model_forward_time": 1.7421746253967285,
      "step": 27594
    },
    {
      "epoch": 2.6315689086914064e-05,
      "step": 27594,
      "training_step_time": 8.560978889465332
    },
    {
      "epoch": 2.6316642761230468e-05,
      "model_forward_time": 1.7447075843811035,
      "step": 27595
    },
    {
      "epoch": 2.6316642761230468e-05,
      "step": 27595,
      "training_step_time": 8.316352605819702
    },
    {
      "epoch": 2.6317596435546875e-05,
      "model_forward_time": 1.7311785221099854,
      "step": 27596
    },
    {
      "epoch": 2.6317596435546875e-05,
      "step": 27596,
      "training_step_time": 10.262678384780884
    },
    {
      "epoch": 2.6318550109863282e-05,
      "model_forward_time": 1.7403693199157715,
      "step": 27597
    },
    {
      "epoch": 2.6318550109863282e-05,
      "step": 27597,
      "training_step_time": 7.743750810623169
    },
    {
      "epoch": 2.6319503784179687e-05,
      "model_forward_time": 1.7462222576141357,
      "step": 27598
    },
    {
      "epoch": 2.6319503784179687e-05,
      "step": 27598,
      "training_step_time": 7.860276699066162
    },
    {
      "epoch": 2.6320457458496094e-05,
      "model_forward_time": 2.887927532196045,
      "step": 27599
    },
    {
      "epoch": 2.6320457458496094e-05,
      "step": 27599,
      "training_step_time": 7.726028203964233
    },
    {
      "epoch": 2.63214111328125e-05,
      "grad_norm": 0.11206085234880447,
      "learning_rate": 8.66751319914579e-06,
      "loss": 0.0597,
      "step": 27600
    },
    {
      "epoch": 2.63214111328125e-05,
      "model_forward_time": 1.748246669769287,
      "step": 27600
    },
    {
      "dynamics_loss_avg": 0.04082359112799168,
      "epoch": 2.63214111328125e-05,
      "step": 27600
    },
    {
      "action_loss_avg": 0.0019186086719855666,
      "epoch": 2.63214111328125e-05,
      "step": 27600
    },
    {
      "epoch": 2.63214111328125e-05,
      "step": 27600,
      "training_step_time": 7.59353232383728
    },
    {
      "epoch": 2.6322364807128905e-05,
      "model_forward_time": 1.3536081314086914,
      "step": 27601
    },
    {
      "epoch": 2.6322364807128905e-05,
      "step": 27601,
      "training_step_time": 7.519177436828613
    },
    {
      "epoch": 2.6323318481445313e-05,
      "model_forward_time": 1.7439661026000977,
      "step": 27602
    },
    {
      "epoch": 2.6323318481445313e-05,
      "step": 27602,
      "training_step_time": 7.808600425720215
    },
    {
      "epoch": 2.632427215576172e-05,
      "model_forward_time": 1.7402377128601074,
      "step": 27603
    },
    {
      "epoch": 2.632427215576172e-05,
      "step": 27603,
      "training_step_time": 8.304675102233887
    },
    {
      "epoch": 2.6325225830078124e-05,
      "model_forward_time": 1.737198829650879,
      "step": 27604
    },
    {
      "epoch": 2.6325225830078124e-05,
      "step": 27604,
      "training_step_time": 7.676321744918823
    },
    {
      "epoch": 2.632617950439453e-05,
      "model_forward_time": 1.7441823482513428,
      "step": 27605
    },
    {
      "epoch": 2.632617950439453e-05,
      "step": 27605,
      "training_step_time": 8.575628995895386
    },
    {
      "epoch": 2.632713317871094e-05,
      "model_forward_time": 1.814462423324585,
      "step": 27606
    },
    {
      "epoch": 2.632713317871094e-05,
      "step": 27606,
      "training_step_time": 8.52707576751709
    },
    {
      "epoch": 2.6328086853027343e-05,
      "model_forward_time": 1.7412469387054443,
      "step": 27607
    },
    {
      "epoch": 2.6328086853027343e-05,
      "step": 27607,
      "training_step_time": 7.726175785064697
    },
    {
      "epoch": 2.632904052734375e-05,
      "model_forward_time": 2.849414587020874,
      "step": 27608
    },
    {
      "epoch": 2.632904052734375e-05,
      "step": 27608,
      "training_step_time": 7.807190895080566
    },
    {
      "epoch": 2.6329994201660158e-05,
      "model_forward_time": 1.7407019138336182,
      "step": 27609
    },
    {
      "epoch": 2.6329994201660158e-05,
      "step": 27609,
      "training_step_time": 7.637292146682739
    },
    {
      "epoch": 2.6330947875976562e-05,
      "grad_norm": 0.13833919167518616,
      "learning_rate": 8.666389157347002e-06,
      "loss": 0.0597,
      "step": 27610
    },
    {
      "epoch": 2.6330947875976562e-05,
      "model_forward_time": 1.7138116359710693,
      "step": 27610
    },
    {
      "dynamics_loss_avg": 0.06611717846244573,
      "epoch": 2.6330947875976562e-05,
      "step": 27610
    },
    {
      "action_loss_avg": 0.0021420433040475473,
      "epoch": 2.6330947875976562e-05,
      "step": 27610
    },
    {
      "epoch": 2.6330947875976562e-05,
      "step": 27610,
      "training_step_time": 7.6773681640625
    },
    {
      "epoch": 2.633190155029297e-05,
      "model_forward_time": 1.7350521087646484,
      "step": 27611
    },
    {
      "epoch": 2.633190155029297e-05,
      "step": 27611,
      "training_step_time": 7.640859365463257
    },
    {
      "epoch": 2.6332855224609373e-05,
      "model_forward_time": 1.73016357421875,
      "step": 27612
    },
    {
      "epoch": 2.6332855224609373e-05,
      "step": 27612,
      "training_step_time": 7.768841743469238
    },
    {
      "epoch": 2.633380889892578e-05,
      "model_forward_time": 1.7290265560150146,
      "step": 27613
    },
    {
      "epoch": 2.633380889892578e-05,
      "step": 27613,
      "training_step_time": 7.556323766708374
    },
    {
      "epoch": 2.6334762573242188e-05,
      "model_forward_time": 1.7315912246704102,
      "step": 27614
    },
    {
      "epoch": 2.6334762573242188e-05,
      "step": 27614,
      "training_step_time": 8.509397745132446
    },
    {
      "epoch": 2.6335716247558592e-05,
      "model_forward_time": 1.808983325958252,
      "step": 27615
    },
    {
      "epoch": 2.6335716247558592e-05,
      "step": 27615,
      "training_step_time": 7.76686954498291
    },
    {
      "epoch": 2.6336669921875e-05,
      "model_forward_time": 1.731764793395996,
      "step": 27616
    },
    {
      "epoch": 2.6336669921875e-05,
      "step": 27616,
      "training_step_time": 7.631626605987549
    },
    {
      "epoch": 2.6337623596191407e-05,
      "model_forward_time": 1.7574779987335205,
      "step": 27617
    },
    {
      "epoch": 2.6337623596191407e-05,
      "step": 27617,
      "training_step_time": 8.208842515945435
    },
    {
      "epoch": 2.633857727050781e-05,
      "model_forward_time": 1.7296648025512695,
      "step": 27618
    },
    {
      "epoch": 2.633857727050781e-05,
      "step": 27618,
      "training_step_time": 7.550886154174805
    },
    {
      "epoch": 2.633953094482422e-05,
      "model_forward_time": 1.0219144821166992,
      "step": 27619
    },
    {
      "epoch": 2.633953094482422e-05,
      "step": 27619,
      "training_step_time": 7.546885967254639
    },
    {
      "epoch": 2.6340484619140626e-05,
      "grad_norm": 0.16343605518341064,
      "learning_rate": 8.66526471459741e-06,
      "loss": 0.0602,
      "step": 27620
    },
    {
      "epoch": 2.6340484619140626e-05,
      "model_forward_time": 1.731663465499878,
      "step": 27620
    },
    {
      "dynamics_loss_avg": 0.06268335264176131,
      "epoch": 2.6340484619140626e-05,
      "step": 27620
    },
    {
      "action_loss_avg": 0.0018700691405683755,
      "epoch": 2.6340484619140626e-05,
      "step": 27620
    },
    {
      "epoch": 2.6340484619140626e-05,
      "step": 27620,
      "training_step_time": 8.376863241195679
    },
    {
      "epoch": 2.634143829345703e-05,
      "model_forward_time": 1.7417938709259033,
      "step": 27621
    },
    {
      "epoch": 2.634143829345703e-05,
      "step": 27621,
      "training_step_time": 8.279033184051514
    },
    {
      "epoch": 2.6342391967773437e-05,
      "model_forward_time": 1.7444422245025635,
      "step": 27622
    },
    {
      "epoch": 2.6342391967773437e-05,
      "step": 27622,
      "training_step_time": 7.644444704055786
    },
    {
      "epoch": 2.6343345642089845e-05,
      "model_forward_time": 1.7366971969604492,
      "step": 27623
    },
    {
      "epoch": 2.6343345642089845e-05,
      "step": 27623,
      "training_step_time": 8.763740301132202
    },
    {
      "epoch": 2.634429931640625e-05,
      "model_forward_time": 1.8098385334014893,
      "step": 27624
    },
    {
      "epoch": 2.634429931640625e-05,
      "step": 27624,
      "training_step_time": 9.117357969284058
    },
    {
      "epoch": 2.6345252990722656e-05,
      "model_forward_time": 1.7438881397247314,
      "step": 27625
    },
    {
      "epoch": 2.6345252990722656e-05,
      "step": 27625,
      "training_step_time": 7.69091272354126
    },
    {
      "epoch": 2.6346206665039064e-05,
      "model_forward_time": 1.7451012134552002,
      "step": 27626
    },
    {
      "epoch": 2.6346206665039064e-05,
      "step": 27626,
      "training_step_time": 7.9648144245147705
    },
    {
      "epoch": 2.6347160339355468e-05,
      "model_forward_time": 2.8915750980377197,
      "step": 27627
    },
    {
      "epoch": 2.6347160339355468e-05,
      "step": 27627,
      "training_step_time": 7.7088048458099365
    },
    {
      "epoch": 2.6348114013671875e-05,
      "model_forward_time": 1.7422244548797607,
      "step": 27628
    },
    {
      "epoch": 2.6348114013671875e-05,
      "step": 27628,
      "training_step_time": 8.624061346054077
    },
    {
      "epoch": 2.6349067687988283e-05,
      "model_forward_time": 1.7438135147094727,
      "step": 27629
    },
    {
      "epoch": 2.6349067687988283e-05,
      "step": 27629,
      "training_step_time": 7.545644760131836
    },
    {
      "epoch": 2.6350021362304687e-05,
      "grad_norm": 0.18884363770484924,
      "learning_rate": 8.664139871019979e-06,
      "loss": 0.0632,
      "step": 27630
    },
    {
      "epoch": 2.6350021362304687e-05,
      "model_forward_time": 1.7160096168518066,
      "step": 27630
    },
    {
      "dynamics_loss_avg": 0.06586258802562953,
      "epoch": 2.6350021362304687e-05,
      "step": 27630
    },
    {
      "action_loss_avg": 0.002774504758417606,
      "epoch": 2.6350021362304687e-05,
      "step": 27630
    },
    {
      "epoch": 2.6350021362304687e-05,
      "step": 27630,
      "training_step_time": 8.4837007522583
    },
    {
      "epoch": 2.6350975036621094e-05,
      "model_forward_time": 1.7311758995056152,
      "step": 27631
    },
    {
      "epoch": 2.6350975036621094e-05,
      "step": 27631,
      "training_step_time": 8.651033878326416
    },
    {
      "epoch": 2.63519287109375e-05,
      "model_forward_time": 1.7276873588562012,
      "step": 27632
    },
    {
      "epoch": 2.63519287109375e-05,
      "step": 27632,
      "training_step_time": 8.87554931640625
    },
    {
      "epoch": 2.6352882385253905e-05,
      "model_forward_time": 1.7607190608978271,
      "step": 27633
    },
    {
      "epoch": 2.6352882385253905e-05,
      "step": 27633,
      "training_step_time": 8.556957721710205
    },
    {
      "epoch": 2.6353836059570313e-05,
      "model_forward_time": 1.7310967445373535,
      "step": 27634
    },
    {
      "epoch": 2.6353836059570313e-05,
      "step": 27634,
      "training_step_time": 7.808773040771484
    },
    {
      "epoch": 2.635478973388672e-05,
      "model_forward_time": 1.73868727684021,
      "step": 27635
    },
    {
      "epoch": 2.635478973388672e-05,
      "step": 27635,
      "training_step_time": 7.898420572280884
    },
    {
      "epoch": 2.6355743408203124e-05,
      "model_forward_time": 2.873835802078247,
      "step": 27636
    },
    {
      "epoch": 2.6355743408203124e-05,
      "step": 27636,
      "training_step_time": 7.7209248542785645
    },
    {
      "epoch": 2.635669708251953e-05,
      "model_forward_time": 1.3757755756378174,
      "step": 27637
    },
    {
      "epoch": 2.635669708251953e-05,
      "step": 27637,
      "training_step_time": 7.591612100601196
    },
    {
      "epoch": 2.635765075683594e-05,
      "model_forward_time": 1.7322394847869873,
      "step": 27638
    },
    {
      "epoch": 2.635765075683594e-05,
      "step": 27638,
      "training_step_time": 7.605982542037964
    },
    {
      "epoch": 2.6358604431152343e-05,
      "model_forward_time": 1.735541582107544,
      "step": 27639
    },
    {
      "epoch": 2.6358604431152343e-05,
      "step": 27639,
      "training_step_time": 8.765403747558594
    },
    {
      "epoch": 2.635955810546875e-05,
      "grad_norm": 0.21006067097187042,
      "learning_rate": 8.663014626737723e-06,
      "loss": 0.0624,
      "step": 27640
    },
    {
      "epoch": 2.635955810546875e-05,
      "model_forward_time": 1.7396647930145264,
      "step": 27640
    },
    {
      "dynamics_loss_avg": 0.05864176331087947,
      "epoch": 2.635955810546875e-05,
      "step": 27640
    },
    {
      "action_loss_avg": 0.003024147532414645,
      "epoch": 2.635955810546875e-05,
      "step": 27640
    },
    {
      "epoch": 2.635955810546875e-05,
      "step": 27640,
      "training_step_time": 7.5822248458862305
    },
    {
      "epoch": 2.6360511779785155e-05,
      "model_forward_time": 1.7402276992797852,
      "step": 27641
    },
    {
      "epoch": 2.6360511779785155e-05,
      "step": 27641,
      "training_step_time": 7.562317371368408
    },
    {
      "epoch": 2.6361465454101562e-05,
      "model_forward_time": 1.7464919090270996,
      "step": 27642
    },
    {
      "epoch": 2.6361465454101562e-05,
      "step": 27642,
      "training_step_time": 8.86020016670227
    },
    {
      "epoch": 2.636241912841797e-05,
      "model_forward_time": 1.7603671550750732,
      "step": 27643
    },
    {
      "epoch": 2.636241912841797e-05,
      "step": 27643,
      "training_step_time": 7.757658004760742
    },
    {
      "epoch": 2.6363372802734373e-05,
      "model_forward_time": 1.7453892230987549,
      "step": 27644
    },
    {
      "epoch": 2.6363372802734373e-05,
      "step": 27644,
      "training_step_time": 7.564999580383301
    },
    {
      "epoch": 2.636432647705078e-05,
      "model_forward_time": 1.3737077713012695,
      "step": 27645
    },
    {
      "epoch": 2.636432647705078e-05,
      "step": 27645,
      "training_step_time": 7.561632394790649
    },
    {
      "epoch": 2.6365280151367188e-05,
      "model_forward_time": 2.8633158206939697,
      "step": 27646
    },
    {
      "epoch": 2.6365280151367188e-05,
      "step": 27646,
      "training_step_time": 7.579191446304321
    },
    {
      "epoch": 2.6366233825683592e-05,
      "model_forward_time": 1.7448546886444092,
      "step": 27647
    },
    {
      "epoch": 2.6366233825683592e-05,
      "step": 27647,
      "training_step_time": 7.544986248016357
    },
    {
      "epoch": 2.63671875e-05,
      "model_forward_time": 1.7407219409942627,
      "step": 27648
    },
    {
      "epoch": 2.63671875e-05,
      "step": 27648,
      "training_step_time": 7.618247032165527
    },
    {
      "epoch": 2.6368141174316407e-05,
      "model_forward_time": 1.7331335544586182,
      "step": 27649
    },
    {
      "epoch": 2.6368141174316407e-05,
      "step": 27649,
      "training_step_time": 9.425129413604736
    },
    {
      "epoch": 2.636909484863281e-05,
      "grad_norm": 0.2321227788925171,
      "learning_rate": 8.661888981873692e-06,
      "loss": 0.0595,
      "step": 27650
    },
    {
      "epoch": 2.636909484863281e-05,
      "model_forward_time": 1.3533217906951904,
      "step": 27650
    },
    {
      "dynamics_loss_avg": 0.03943336308002472,
      "epoch": 2.636909484863281e-05,
      "step": 27650
    },
    {
      "action_loss_avg": 0.0019299422740004957,
      "epoch": 2.636909484863281e-05,
      "step": 27650
    },
    {
      "epoch": 2.636909484863281e-05,
      "step": 27650,
      "training_step_time": 7.633016109466553
    },
    {
      "epoch": 2.637004852294922e-05,
      "model_forward_time": 1.7188720703125,
      "step": 27651
    },
    {
      "epoch": 2.637004852294922e-05,
      "step": 27651,
      "training_step_time": 7.882537603378296
    },
    {
      "epoch": 2.6371002197265626e-05,
      "model_forward_time": 1.794619083404541,
      "step": 27652
    },
    {
      "epoch": 2.6371002197265626e-05,
      "step": 27652,
      "training_step_time": 8.450366973876953
    },
    {
      "epoch": 2.637195587158203e-05,
      "model_forward_time": 1.7176697254180908,
      "step": 27653
    },
    {
      "epoch": 2.637195587158203e-05,
      "step": 27653,
      "training_step_time": 7.796989917755127
    },
    {
      "epoch": 2.6372909545898437e-05,
      "model_forward_time": 1.721064567565918,
      "step": 27654
    },
    {
      "epoch": 2.6372909545898437e-05,
      "step": 27654,
      "training_step_time": 8.872249841690063
    },
    {
      "epoch": 2.6373863220214845e-05,
      "model_forward_time": 2.8835599422454834,
      "step": 27655
    },
    {
      "epoch": 2.6373863220214845e-05,
      "step": 27655,
      "training_step_time": 7.753778457641602
    },
    {
      "epoch": 2.637481689453125e-05,
      "model_forward_time": 1.7193629741668701,
      "step": 27656
    },
    {
      "epoch": 2.637481689453125e-05,
      "step": 27656,
      "training_step_time": 7.683709144592285
    },
    {
      "epoch": 2.6375770568847656e-05,
      "model_forward_time": 1.7172026634216309,
      "step": 27657
    },
    {
      "epoch": 2.6375770568847656e-05,
      "step": 27657,
      "training_step_time": 7.8223772048950195
    },
    {
      "epoch": 2.6376724243164064e-05,
      "model_forward_time": 1.7249879837036133,
      "step": 27658
    },
    {
      "epoch": 2.6376724243164064e-05,
      "step": 27658,
      "training_step_time": 7.7939722537994385
    },
    {
      "epoch": 2.6377677917480468e-05,
      "model_forward_time": 1.71962571144104,
      "step": 27659
    },
    {
      "epoch": 2.6377677917480468e-05,
      "step": 27659,
      "training_step_time": 7.657230377197266
    },
    {
      "epoch": 2.6378631591796875e-05,
      "grad_norm": 0.11526019126176834,
      "learning_rate": 8.660762936550989e-06,
      "loss": 0.0586,
      "step": 27660
    },
    {
      "epoch": 2.6378631591796875e-05,
      "model_forward_time": 1.7085518836975098,
      "step": 27660
    },
    {
      "dynamics_loss_avg": 0.06375823505222797,
      "epoch": 2.6378631591796875e-05,
      "step": 27660
    },
    {
      "action_loss_avg": 0.0026992106810212134,
      "epoch": 2.6378631591796875e-05,
      "step": 27660
    },
    {
      "epoch": 2.6378631591796875e-05,
      "step": 27660,
      "training_step_time": 9.64961576461792
    },
    {
      "epoch": 2.6379585266113283e-05,
      "model_forward_time": 1.8061649799346924,
      "step": 27661
    },
    {
      "epoch": 2.6379585266113283e-05,
      "step": 27661,
      "training_step_time": 7.905689477920532
    },
    {
      "epoch": 2.6380538940429687e-05,
      "model_forward_time": 1.7452373504638672,
      "step": 27662
    },
    {
      "epoch": 2.6380538940429687e-05,
      "step": 27662,
      "training_step_time": 7.853025197982788
    },
    {
      "epoch": 2.6381492614746094e-05,
      "model_forward_time": 1.7435064315795898,
      "step": 27663
    },
    {
      "epoch": 2.6381492614746094e-05,
      "step": 27663,
      "training_step_time": 7.952466726303101
    },
    {
      "epoch": 2.63824462890625e-05,
      "model_forward_time": 2.8716561794281006,
      "step": 27664
    },
    {
      "epoch": 2.63824462890625e-05,
      "step": 27664,
      "training_step_time": 7.732110023498535
    },
    {
      "epoch": 2.6383399963378905e-05,
      "model_forward_time": 1.743544340133667,
      "step": 27665
    },
    {
      "epoch": 2.6383399963378905e-05,
      "step": 27665,
      "training_step_time": 7.623687505722046
    },
    {
      "epoch": 2.6384353637695313e-05,
      "model_forward_time": 1.739267110824585,
      "step": 27666
    },
    {
      "epoch": 2.6384353637695313e-05,
      "step": 27666,
      "training_step_time": 7.817121744155884
    },
    {
      "epoch": 2.638530731201172e-05,
      "model_forward_time": 1.7467401027679443,
      "step": 27667
    },
    {
      "epoch": 2.638530731201172e-05,
      "step": 27667,
      "training_step_time": 9.401015043258667
    },
    {
      "epoch": 2.6386260986328124e-05,
      "model_forward_time": 1.7427306175231934,
      "step": 27668
    },
    {
      "epoch": 2.6386260986328124e-05,
      "step": 27668,
      "training_step_time": 7.75743842124939
    },
    {
      "epoch": 2.6387214660644532e-05,
      "model_forward_time": 1.73240327835083,
      "step": 27669
    },
    {
      "epoch": 2.6387214660644532e-05,
      "step": 27669,
      "training_step_time": 8.801086664199829
    },
    {
      "epoch": 2.638816833496094e-05,
      "grad_norm": 0.19641409814357758,
      "learning_rate": 8.659636490892753e-06,
      "loss": 0.0596,
      "step": 27670
    },
    {
      "epoch": 2.638816833496094e-05,
      "model_forward_time": 1.7979748249053955,
      "step": 27670
    },
    {
      "dynamics_loss_avg": 0.06605840362608432,
      "epoch": 2.638816833496094e-05,
      "step": 27670
    },
    {
      "action_loss_avg": 0.0018286153121152894,
      "epoch": 2.638816833496094e-05,
      "step": 27670
    },
    {
      "epoch": 2.638816833496094e-05,
      "step": 27670,
      "training_step_time": 7.906354665756226
    },
    {
      "epoch": 2.6389122009277343e-05,
      "model_forward_time": 1.7462894916534424,
      "step": 27671
    },
    {
      "epoch": 2.6389122009277343e-05,
      "step": 27671,
      "training_step_time": 8.809054613113403
    },
    {
      "epoch": 2.639007568359375e-05,
      "model_forward_time": 1.7470145225524902,
      "step": 27672
    },
    {
      "epoch": 2.639007568359375e-05,
      "step": 27672,
      "training_step_time": 7.85813570022583
    },
    {
      "epoch": 2.6391029357910155e-05,
      "model_forward_time": 2.856074810028076,
      "step": 27673
    },
    {
      "epoch": 2.6391029357910155e-05,
      "step": 27673,
      "training_step_time": 7.668668031692505
    },
    {
      "epoch": 2.6391983032226562e-05,
      "model_forward_time": 1.7409968376159668,
      "step": 27674
    },
    {
      "epoch": 2.6391983032226562e-05,
      "step": 27674,
      "training_step_time": 7.591888666152954
    },
    {
      "epoch": 2.639293670654297e-05,
      "model_forward_time": 1.3662080764770508,
      "step": 27675
    },
    {
      "epoch": 2.639293670654297e-05,
      "step": 27675,
      "training_step_time": 7.753589630126953
    },
    {
      "epoch": 2.6393890380859373e-05,
      "model_forward_time": 1.7478580474853516,
      "step": 27676
    },
    {
      "epoch": 2.6393890380859373e-05,
      "step": 27676,
      "training_step_time": 7.856217861175537
    },
    {
      "epoch": 2.639484405517578e-05,
      "model_forward_time": 1.7385499477386475,
      "step": 27677
    },
    {
      "epoch": 2.639484405517578e-05,
      "step": 27677,
      "training_step_time": 7.762354373931885
    },
    {
      "epoch": 2.6395797729492188e-05,
      "model_forward_time": 1.7410593032836914,
      "step": 27678
    },
    {
      "epoch": 2.6395797729492188e-05,
      "step": 27678,
      "training_step_time": 9.76662015914917
    },
    {
      "epoch": 2.6396751403808592e-05,
      "model_forward_time": 1.7680702209472656,
      "step": 27679
    },
    {
      "epoch": 2.6396751403808592e-05,
      "step": 27679,
      "training_step_time": 8.390887022018433
    },
    {
      "epoch": 2.6397705078125e-05,
      "grad_norm": 0.19159337878227234,
      "learning_rate": 8.658509645022174e-06,
      "loss": 0.0648,
      "step": 27680
    },
    {
      "epoch": 2.6397705078125e-05,
      "model_forward_time": 1.7391033172607422,
      "step": 27680
    },
    {
      "dynamics_loss_avg": 0.05526230214163661,
      "epoch": 2.6397705078125e-05,
      "step": 27680
    },
    {
      "action_loss_avg": 0.00236997134052217,
      "epoch": 2.6397705078125e-05,
      "step": 27680
    },
    {
      "epoch": 2.6397705078125e-05,
      "step": 27680,
      "training_step_time": 8.970346927642822
    },
    {
      "epoch": 2.6398658752441407e-05,
      "model_forward_time": 1.7472214698791504,
      "step": 27681
    },
    {
      "epoch": 2.6398658752441407e-05,
      "step": 27681,
      "training_step_time": 8.717323303222656
    },
    {
      "epoch": 2.639961242675781e-05,
      "model_forward_time": 1.7436084747314453,
      "step": 27682
    },
    {
      "epoch": 2.639961242675781e-05,
      "step": 27682,
      "training_step_time": 7.950249195098877
    },
    {
      "epoch": 2.640056610107422e-05,
      "model_forward_time": 2.910917282104492,
      "step": 27683
    },
    {
      "epoch": 2.640056610107422e-05,
      "step": 27683,
      "training_step_time": 7.705561399459839
    },
    {
      "epoch": 2.6401519775390626e-05,
      "model_forward_time": 1.7740733623504639,
      "step": 27684
    },
    {
      "epoch": 2.6401519775390626e-05,
      "step": 27684,
      "training_step_time": 7.645021677017212
    },
    {
      "epoch": 2.640247344970703e-05,
      "model_forward_time": 1.7468023300170898,
      "step": 27685
    },
    {
      "epoch": 2.640247344970703e-05,
      "step": 27685,
      "training_step_time": 7.740780353546143
    },
    {
      "epoch": 2.6403427124023437e-05,
      "model_forward_time": 1.7478203773498535,
      "step": 27686
    },
    {
      "epoch": 2.6403427124023437e-05,
      "step": 27686,
      "training_step_time": 8.522190809249878
    },
    {
      "epoch": 2.6404380798339845e-05,
      "model_forward_time": 1.3608176708221436,
      "step": 27687
    },
    {
      "epoch": 2.6404380798339845e-05,
      "step": 27687,
      "training_step_time": 8.673328399658203
    },
    {
      "epoch": 2.640533447265625e-05,
      "model_forward_time": 1.7435576915740967,
      "step": 27688
    },
    {
      "epoch": 2.640533447265625e-05,
      "step": 27688,
      "training_step_time": 7.748054265975952
    },
    {
      "epoch": 2.6406288146972656e-05,
      "model_forward_time": 1.819382905960083,
      "step": 27689
    },
    {
      "epoch": 2.6406288146972656e-05,
      "step": 27689,
      "training_step_time": 7.830536365509033
    },
    {
      "epoch": 2.6407241821289064e-05,
      "grad_norm": 0.1216978058218956,
      "learning_rate": 8.65738239906248e-06,
      "loss": 0.0628,
      "step": 27690
    },
    {
      "epoch": 2.6407241821289064e-05,
      "model_forward_time": 1.741236925125122,
      "step": 27690
    },
    {
      "dynamics_loss_avg": 0.08012615256011486,
      "epoch": 2.6407241821289064e-05,
      "step": 27690
    },
    {
      "action_loss_avg": 0.00282443207106553,
      "epoch": 2.6407241821289064e-05,
      "step": 27690
    },
    {
      "epoch": 2.6407241821289064e-05,
      "step": 27690,
      "training_step_time": 8.461500644683838
    },
    {
      "epoch": 2.6408195495605468e-05,
      "model_forward_time": 1.7370655536651611,
      "step": 27691
    },
    {
      "epoch": 2.6408195495605468e-05,
      "step": 27691,
      "training_step_time": 7.927365303039551
    },
    {
      "epoch": 2.6409149169921875e-05,
      "model_forward_time": 2.882377862930298,
      "step": 27692
    },
    {
      "epoch": 2.6409149169921875e-05,
      "step": 27692,
      "training_step_time": 8.701180219650269
    },
    {
      "epoch": 2.6410102844238283e-05,
      "model_forward_time": 1.7443203926086426,
      "step": 27693
    },
    {
      "epoch": 2.6410102844238283e-05,
      "step": 27693,
      "training_step_time": 7.57419490814209
    },
    {
      "epoch": 2.6411056518554687e-05,
      "model_forward_time": 1.7382068634033203,
      "step": 27694
    },
    {
      "epoch": 2.6411056518554687e-05,
      "step": 27694,
      "training_step_time": 7.696481466293335
    },
    {
      "epoch": 2.6412010192871094e-05,
      "model_forward_time": 1.740276575088501,
      "step": 27695
    },
    {
      "epoch": 2.6412010192871094e-05,
      "step": 27695,
      "training_step_time": 8.818185329437256
    },
    {
      "epoch": 2.64129638671875e-05,
      "model_forward_time": 1.738743543624878,
      "step": 27696
    },
    {
      "epoch": 2.64129638671875e-05,
      "step": 27696,
      "training_step_time": 9.837889671325684
    },
    {
      "epoch": 2.6413917541503905e-05,
      "model_forward_time": 1.7466614246368408,
      "step": 27697
    },
    {
      "epoch": 2.6413917541503905e-05,
      "step": 27697,
      "training_step_time": 6.8485376834869385
    },
    {
      "epoch": 2.6414871215820313e-05,
      "model_forward_time": 1.743227481842041,
      "step": 27698
    },
    {
      "epoch": 2.6414871215820313e-05,
      "step": 27698,
      "training_step_time": 6.949456214904785
    },
    {
      "epoch": 2.641582489013672e-05,
      "model_forward_time": 1.7456746101379395,
      "step": 27699
    },
    {
      "epoch": 2.641582489013672e-05,
      "step": 27699,
      "training_step_time": 6.86232328414917
    },
    {
      "epoch": 2.6416778564453124e-05,
      "grad_norm": 0.10560062527656555,
      "learning_rate": 8.656254753136948e-06,
      "loss": 0.062,
      "step": 27700
    },
    {
      "epoch": 2.6416778564453124e-05,
      "model_forward_time": 1.7382807731628418,
      "step": 27700
    },
    {
      "dynamics_loss_avg": 0.05807083323597908,
      "epoch": 2.6416778564453124e-05,
      "step": 27700
    },
    {
      "action_loss_avg": 0.0018860475451219828,
      "epoch": 2.6416778564453124e-05,
      "step": 27700
    },
    {
      "epoch": 2.6416778564453124e-05,
      "step": 27700,
      "training_step_time": 6.891705513000488
    },
    {
      "epoch": 2.6417732238769532e-05,
      "model_forward_time": 1.7433838844299316,
      "step": 27701
    },
    {
      "epoch": 2.6417732238769532e-05,
      "step": 27701,
      "training_step_time": 6.929659366607666
    },
    {
      "epoch": 2.641868591308594e-05,
      "model_forward_time": 1.7444331645965576,
      "step": 27702
    },
    {
      "epoch": 2.641868591308594e-05,
      "step": 27702,
      "training_step_time": 6.84612774848938
    },
    {
      "epoch": 2.6419639587402343e-05,
      "model_forward_time": 1.7438766956329346,
      "step": 27703
    },
    {
      "epoch": 2.6419639587402343e-05,
      "step": 27703,
      "training_step_time": 6.883871793746948
    },
    {
      "epoch": 2.642059326171875e-05,
      "model_forward_time": 1.7378108501434326,
      "step": 27704
    },
    {
      "epoch": 2.642059326171875e-05,
      "step": 27704,
      "training_step_time": 6.850056171417236
    },
    {
      "epoch": 2.6421546936035155e-05,
      "model_forward_time": 2.8861653804779053,
      "step": 27705
    },
    {
      "epoch": 2.6421546936035155e-05,
      "step": 27705,
      "training_step_time": 7.729725122451782
    },
    {
      "epoch": 2.6422500610351562e-05,
      "model_forward_time": 1.7421021461486816,
      "step": 27706
    },
    {
      "epoch": 2.6422500610351562e-05,
      "step": 27706,
      "training_step_time": 7.9950573444366455
    },
    {
      "epoch": 2.642345428466797e-05,
      "model_forward_time": 1.7426517009735107,
      "step": 27707
    },
    {
      "epoch": 2.642345428466797e-05,
      "step": 27707,
      "training_step_time": 6.880682468414307
    },
    {
      "epoch": 2.6424407958984373e-05,
      "model_forward_time": 1.7351975440979004,
      "step": 27708
    },
    {
      "epoch": 2.6424407958984373e-05,
      "step": 27708,
      "training_step_time": 6.845135927200317
    },
    {
      "epoch": 2.642536163330078e-05,
      "model_forward_time": 1.7448101043701172,
      "step": 27709
    },
    {
      "epoch": 2.642536163330078e-05,
      "step": 27709,
      "training_step_time": 6.852912187576294
    },
    {
      "epoch": 2.642631530761719e-05,
      "grad_norm": 0.1266087293624878,
      "learning_rate": 8.65512670736889e-06,
      "loss": 0.0585,
      "step": 27710
    },
    {
      "epoch": 2.642631530761719e-05,
      "model_forward_time": 1.743783950805664,
      "step": 27710
    },
    {
      "dynamics_loss_avg": 0.05573341902345419,
      "epoch": 2.642631530761719e-05,
      "step": 27710
    },
    {
      "action_loss_avg": 0.003121629200177267,
      "epoch": 2.642631530761719e-05,
      "step": 27710
    },
    {
      "epoch": 2.642631530761719e-05,
      "step": 27710,
      "training_step_time": 6.907628297805786
    },
    {
      "epoch": 2.6427268981933592e-05,
      "model_forward_time": 1.7461678981781006,
      "step": 27711
    },
    {
      "epoch": 2.6427268981933592e-05,
      "step": 27711,
      "training_step_time": 6.852078676223755
    },
    {
      "epoch": 2.642822265625e-05,
      "model_forward_time": 1.742394208908081,
      "step": 27712
    },
    {
      "epoch": 2.642822265625e-05,
      "step": 27712,
      "training_step_time": 6.83677864074707
    },
    {
      "epoch": 2.6429176330566407e-05,
      "model_forward_time": 1.7434751987457275,
      "step": 27713
    },
    {
      "epoch": 2.6429176330566407e-05,
      "step": 27713,
      "training_step_time": 7.8535239696502686
    },
    {
      "epoch": 2.643013000488281e-05,
      "model_forward_time": 2.9402425289154053,
      "step": 27714
    },
    {
      "epoch": 2.643013000488281e-05,
      "step": 27714,
      "training_step_time": 7.793696403503418
    },
    {
      "epoch": 2.643108367919922e-05,
      "model_forward_time": 1.7446568012237549,
      "step": 27715
    },
    {
      "epoch": 2.643108367919922e-05,
      "step": 27715,
      "training_step_time": 9.004355669021606
    },
    {
      "epoch": 2.6432037353515626e-05,
      "model_forward_time": 1.7405779361724854,
      "step": 27716
    },
    {
      "epoch": 2.6432037353515626e-05,
      "step": 27716,
      "training_step_time": 6.864320993423462
    },
    {
      "epoch": 2.643299102783203e-05,
      "model_forward_time": 1.7483067512512207,
      "step": 27717
    },
    {
      "epoch": 2.643299102783203e-05,
      "step": 27717,
      "training_step_time": 6.8958775997161865
    },
    {
      "epoch": 2.6433944702148437e-05,
      "model_forward_time": 1.744131088256836,
      "step": 27718
    },
    {
      "epoch": 2.6433944702148437e-05,
      "step": 27718,
      "training_step_time": 6.925260305404663
    },
    {
      "epoch": 2.6434898376464845e-05,
      "model_forward_time": 1.7469704151153564,
      "step": 27719
    },
    {
      "epoch": 2.6434898376464845e-05,
      "step": 27719,
      "training_step_time": 6.892723798751831
    },
    {
      "epoch": 2.643585205078125e-05,
      "grad_norm": 0.11013459414243698,
      "learning_rate": 8.653998261881672e-06,
      "loss": 0.064,
      "step": 27720
    },
    {
      "epoch": 2.643585205078125e-05,
      "model_forward_time": 1.7382586002349854,
      "step": 27720
    },
    {
      "dynamics_loss_avg": 0.04669016692787409,
      "epoch": 2.643585205078125e-05,
      "step": 27720
    },
    {
      "action_loss_avg": 0.0031653579906560482,
      "epoch": 2.643585205078125e-05,
      "step": 27720
    },
    {
      "epoch": 2.643585205078125e-05,
      "step": 27720,
      "training_step_time": 6.901818037033081
    },
    {
      "epoch": 2.6436805725097656e-05,
      "model_forward_time": 1.0320820808410645,
      "step": 27721
    },
    {
      "epoch": 2.6436805725097656e-05,
      "step": 27721,
      "training_step_time": 6.844562530517578
    },
    {
      "epoch": 2.6437759399414064e-05,
      "model_forward_time": 1.7287280559539795,
      "step": 27722
    },
    {
      "epoch": 2.6437759399414064e-05,
      "step": 27722,
      "training_step_time": 6.877543926239014
    },
    {
      "epoch": 2.6438713073730468e-05,
      "model_forward_time": 2.875474452972412,
      "step": 27723
    },
    {
      "epoch": 2.6438713073730468e-05,
      "step": 27723,
      "training_step_time": 7.778269529342651
    },
    {
      "epoch": 2.6439666748046875e-05,
      "model_forward_time": 1.7278170585632324,
      "step": 27724
    },
    {
      "epoch": 2.6439666748046875e-05,
      "step": 27724,
      "training_step_time": 7.8420257568359375
    },
    {
      "epoch": 2.6440620422363283e-05,
      "model_forward_time": 1.7293226718902588,
      "step": 27725
    },
    {
      "epoch": 2.6440620422363283e-05,
      "step": 27725,
      "training_step_time": 7.715028762817383
    },
    {
      "epoch": 2.6441574096679687e-05,
      "model_forward_time": 1.7337968349456787,
      "step": 27726
    },
    {
      "epoch": 2.6441574096679687e-05,
      "step": 27726,
      "training_step_time": 6.855488538742065
    },
    {
      "epoch": 2.6442527770996094e-05,
      "model_forward_time": 1.7280364036560059,
      "step": 27727
    },
    {
      "epoch": 2.6442527770996094e-05,
      "step": 27727,
      "training_step_time": 6.936911344528198
    },
    {
      "epoch": 2.64434814453125e-05,
      "model_forward_time": 1.7329654693603516,
      "step": 27728
    },
    {
      "epoch": 2.64434814453125e-05,
      "step": 27728,
      "training_step_time": 6.85059118270874
    },
    {
      "epoch": 2.6444435119628905e-05,
      "model_forward_time": 1.7351250648498535,
      "step": 27729
    },
    {
      "epoch": 2.6444435119628905e-05,
      "step": 27729,
      "training_step_time": 6.892546892166138
    },
    {
      "epoch": 2.6445388793945313e-05,
      "grad_norm": 0.10672350972890854,
      "learning_rate": 8.6528694167987e-06,
      "loss": 0.0564,
      "step": 27730
    },
    {
      "epoch": 2.6445388793945313e-05,
      "model_forward_time": 1.729032039642334,
      "step": 27730
    },
    {
      "dynamics_loss_avg": 0.05779152382165194,
      "epoch": 2.6445388793945313e-05,
      "step": 27730
    },
    {
      "action_loss_avg": 0.0023056435282342138,
      "epoch": 2.6445388793945313e-05,
      "step": 27730
    },
    {
      "epoch": 2.6445388793945313e-05,
      "step": 27730,
      "training_step_time": 6.834657430648804
    },
    {
      "epoch": 2.644634246826172e-05,
      "model_forward_time": 1.7242417335510254,
      "step": 27731
    },
    {
      "epoch": 2.644634246826172e-05,
      "step": 27731,
      "training_step_time": 6.861793279647827
    },
    {
      "epoch": 2.6447296142578124e-05,
      "model_forward_time": 2.855437994003296,
      "step": 27732
    },
    {
      "epoch": 2.6447296142578124e-05,
      "step": 27732,
      "training_step_time": 7.600677490234375
    },
    {
      "epoch": 2.6448249816894532e-05,
      "model_forward_time": 1.7243211269378662,
      "step": 27733
    },
    {
      "epoch": 2.6448249816894532e-05,
      "step": 27733,
      "training_step_time": 9.449665307998657
    },
    {
      "epoch": 2.644920349121094e-05,
      "model_forward_time": 1.3569600582122803,
      "step": 27734
    },
    {
      "epoch": 2.644920349121094e-05,
      "step": 27734,
      "training_step_time": 7.844137907028198
    },
    {
      "epoch": 2.6450157165527343e-05,
      "model_forward_time": 1.7264924049377441,
      "step": 27735
    },
    {
      "epoch": 2.6450157165527343e-05,
      "step": 27735,
      "training_step_time": 6.881213903427124
    },
    {
      "epoch": 2.645111083984375e-05,
      "model_forward_time": 1.728126049041748,
      "step": 27736
    },
    {
      "epoch": 2.645111083984375e-05,
      "step": 27736,
      "training_step_time": 6.827042818069458
    },
    {
      "epoch": 2.6452064514160155e-05,
      "model_forward_time": 1.7252154350280762,
      "step": 27737
    },
    {
      "epoch": 2.6452064514160155e-05,
      "step": 27737,
      "training_step_time": 6.85756778717041
    },
    {
      "epoch": 2.6453018188476562e-05,
      "model_forward_time": 1.7229945659637451,
      "step": 27738
    },
    {
      "epoch": 2.6453018188476562e-05,
      "step": 27738,
      "training_step_time": 6.809755563735962
    },
    {
      "epoch": 2.645397186279297e-05,
      "model_forward_time": 1.7252440452575684,
      "step": 27739
    },
    {
      "epoch": 2.645397186279297e-05,
      "step": 27739,
      "training_step_time": 6.877011060714722
    },
    {
      "epoch": 2.6454925537109374e-05,
      "grad_norm": 0.10333047062158585,
      "learning_rate": 8.651740172243416e-06,
      "loss": 0.0607,
      "step": 27740
    },
    {
      "epoch": 2.6454925537109374e-05,
      "model_forward_time": 1.7219200134277344,
      "step": 27740
    },
    {
      "dynamics_loss_avg": 0.055623373854905366,
      "epoch": 2.6454925537109374e-05,
      "step": 27740
    },
    {
      "action_loss_avg": 0.002337916917167604,
      "epoch": 2.6454925537109374e-05,
      "step": 27740
    },
    {
      "epoch": 2.6454925537109374e-05,
      "step": 27740,
      "training_step_time": 6.882404804229736
    },
    {
      "epoch": 2.645587921142578e-05,
      "model_forward_time": 1.3848609924316406,
      "step": 27741
    },
    {
      "epoch": 2.645587921142578e-05,
      "step": 27741,
      "training_step_time": 7.466289758682251
    },
    {
      "epoch": 2.645683288574219e-05,
      "model_forward_time": 2.9125020503997803,
      "step": 27742
    },
    {
      "epoch": 2.645683288574219e-05,
      "step": 27742,
      "training_step_time": 10.097548723220825
    },
    {
      "epoch": 2.6457786560058592e-05,
      "model_forward_time": 1.7276899814605713,
      "step": 27743
    },
    {
      "epoch": 2.6457786560058592e-05,
      "step": 27743,
      "training_step_time": 7.776357412338257
    },
    {
      "epoch": 2.6458740234375e-05,
      "model_forward_time": 1.7271735668182373,
      "step": 27744
    },
    {
      "epoch": 2.6458740234375e-05,
      "step": 27744,
      "training_step_time": 6.869628190994263
    },
    {
      "epoch": 2.6459693908691407e-05,
      "model_forward_time": 1.730013370513916,
      "step": 27745
    },
    {
      "epoch": 2.6459693908691407e-05,
      "step": 27745,
      "training_step_time": 6.8534018993377686
    },
    {
      "epoch": 2.646064758300781e-05,
      "model_forward_time": 1.7231178283691406,
      "step": 27746
    },
    {
      "epoch": 2.646064758300781e-05,
      "step": 27746,
      "training_step_time": 6.849562644958496
    },
    {
      "epoch": 2.646160125732422e-05,
      "model_forward_time": 1.73056960105896,
      "step": 27747
    },
    {
      "epoch": 2.646160125732422e-05,
      "step": 27747,
      "training_step_time": 6.806647539138794
    },
    {
      "epoch": 2.6462554931640626e-05,
      "model_forward_time": 1.73093581199646,
      "step": 27748
    },
    {
      "epoch": 2.6462554931640626e-05,
      "step": 27748,
      "training_step_time": 6.859338760375977
    },
    {
      "epoch": 2.646350860595703e-05,
      "model_forward_time": 1.7241299152374268,
      "step": 27749
    },
    {
      "epoch": 2.646350860595703e-05,
      "step": 27749,
      "training_step_time": 6.840847730636597
    },
    {
      "epoch": 2.6464462280273438e-05,
      "grad_norm": 0.21432502567768097,
      "learning_rate": 8.65061052833932e-06,
      "loss": 0.0587,
      "step": 27750
    },
    {
      "epoch": 2.6464462280273438e-05,
      "model_forward_time": 1.7515366077423096,
      "step": 27750
    },
    {
      "dynamics_loss_avg": 0.0647929273545742,
      "epoch": 2.6464462280273438e-05,
      "step": 27750
    },
    {
      "action_loss_avg": 0.002162134199170396,
      "epoch": 2.6464462280273438e-05,
      "step": 27750
    },
    {
      "epoch": 2.6464462280273438e-05,
      "step": 27750,
      "training_step_time": 7.470085382461548
    },
    {
      "epoch": 2.6465415954589845e-05,
      "model_forward_time": 2.8456954956054688,
      "step": 27751
    },
    {
      "epoch": 2.6465415954589845e-05,
      "step": 27751,
      "training_step_time": 8.632770776748657
    },
    {
      "epoch": 2.646636962890625e-05,
      "model_forward_time": 1.7275099754333496,
      "step": 27752
    },
    {
      "epoch": 2.646636962890625e-05,
      "step": 27752,
      "training_step_time": 8.807047128677368
    },
    {
      "epoch": 2.6467323303222656e-05,
      "model_forward_time": 1.7229132652282715,
      "step": 27753
    },
    {
      "epoch": 2.6467323303222656e-05,
      "step": 27753,
      "training_step_time": 7.61491847038269
    },
    {
      "epoch": 2.6468276977539064e-05,
      "model_forward_time": 1.7257251739501953,
      "step": 27754
    },
    {
      "epoch": 2.6468276977539064e-05,
      "step": 27754,
      "training_step_time": 6.870548486709595
    },
    {
      "epoch": 2.6469230651855468e-05,
      "model_forward_time": 1.7264723777770996,
      "step": 27755
    },
    {
      "epoch": 2.6469230651855468e-05,
      "step": 27755,
      "training_step_time": 6.877330303192139
    },
    {
      "epoch": 2.6470184326171875e-05,
      "model_forward_time": 1.7331268787384033,
      "step": 27756
    },
    {
      "epoch": 2.6470184326171875e-05,
      "step": 27756,
      "training_step_time": 6.870047569274902
    },
    {
      "epoch": 2.6471138000488283e-05,
      "model_forward_time": 1.7354021072387695,
      "step": 27757
    },
    {
      "epoch": 2.6471138000488283e-05,
      "step": 27757,
      "training_step_time": 6.824643611907959
    },
    {
      "epoch": 2.6472091674804687e-05,
      "model_forward_time": 1.7331602573394775,
      "step": 27758
    },
    {
      "epoch": 2.6472091674804687e-05,
      "step": 27758,
      "training_step_time": 6.866174221038818
    },
    {
      "epoch": 2.6473045349121094e-05,
      "model_forward_time": 1.736128568649292,
      "step": 27759
    },
    {
      "epoch": 2.6473045349121094e-05,
      "step": 27759,
      "training_step_time": 7.444851398468018
    },
    {
      "epoch": 2.64739990234375e-05,
      "grad_norm": 0.13491109013557434,
      "learning_rate": 8.649480485209945e-06,
      "loss": 0.0578,
      "step": 27760
    },
    {
      "epoch": 2.64739990234375e-05,
      "model_forward_time": 2.8473706245422363,
      "step": 27760
    },
    {
      "dynamics_loss_avg": 0.06913639921694995,
      "epoch": 2.64739990234375e-05,
      "step": 27760
    },
    {
      "action_loss_avg": 0.0028296487056650222,
      "epoch": 2.64739990234375e-05,
      "step": 27760
    },
    {
      "epoch": 2.64739990234375e-05,
      "step": 27760,
      "training_step_time": 10.636923551559448
    },
    {
      "epoch": 2.6474952697753906e-05,
      "model_forward_time": 1.7238214015960693,
      "step": 27761
    },
    {
      "epoch": 2.6474952697753906e-05,
      "step": 27761,
      "training_step_time": 8.819101572036743
    },
    {
      "epoch": 2.6475906372070313e-05,
      "model_forward_time": 1.730173110961914,
      "step": 27762
    },
    {
      "epoch": 2.6475906372070313e-05,
      "step": 27762,
      "training_step_time": 7.599564790725708
    },
    {
      "epoch": 2.647686004638672e-05,
      "model_forward_time": 1.7283473014831543,
      "step": 27763
    },
    {
      "epoch": 2.647686004638672e-05,
      "step": 27763,
      "training_step_time": 7.331097364425659
    },
    {
      "epoch": 2.6477813720703124e-05,
      "model_forward_time": 1.7341210842132568,
      "step": 27764
    },
    {
      "epoch": 2.6477813720703124e-05,
      "step": 27764,
      "training_step_time": 6.8925065994262695
    },
    {
      "epoch": 2.6478767395019532e-05,
      "model_forward_time": 1.7322251796722412,
      "step": 27765
    },
    {
      "epoch": 2.6478767395019532e-05,
      "step": 27765,
      "training_step_time": 6.84283185005188
    },
    {
      "epoch": 2.6479721069335936e-05,
      "model_forward_time": 1.7254977226257324,
      "step": 27766
    },
    {
      "epoch": 2.6479721069335936e-05,
      "step": 27766,
      "training_step_time": 6.873899221420288
    },
    {
      "epoch": 2.6480674743652343e-05,
      "model_forward_time": 1.7254273891448975,
      "step": 27767
    },
    {
      "epoch": 2.6480674743652343e-05,
      "step": 27767,
      "training_step_time": 6.940814971923828
    },
    {
      "epoch": 2.648162841796875e-05,
      "model_forward_time": 1.731804609298706,
      "step": 27768
    },
    {
      "epoch": 2.648162841796875e-05,
      "step": 27768,
      "training_step_time": 6.9296605587005615
    },
    {
      "epoch": 2.6482582092285155e-05,
      "model_forward_time": 1.77341890335083,
      "step": 27769
    },
    {
      "epoch": 2.6482582092285155e-05,
      "step": 27769,
      "training_step_time": 9.294449806213379
    },
    {
      "epoch": 2.6483535766601562e-05,
      "grad_norm": 0.14473557472229004,
      "learning_rate": 8.648350042978872e-06,
      "loss": 0.0624,
      "step": 27770
    },
    {
      "epoch": 2.6483535766601562e-05,
      "model_forward_time": 1.7286670207977295,
      "step": 27770
    },
    {
      "dynamics_loss_avg": 0.056976545974612235,
      "epoch": 2.6483535766601562e-05,
      "step": 27770
    },
    {
      "action_loss_avg": 0.0020320092036854474,
      "epoch": 2.6483535766601562e-05,
      "step": 27770
    },
    {
      "epoch": 2.6483535766601562e-05,
      "step": 27770,
      "training_step_time": 9.870119571685791
    },
    {
      "epoch": 2.648448944091797e-05,
      "model_forward_time": 1.731139898300171,
      "step": 27771
    },
    {
      "epoch": 2.648448944091797e-05,
      "step": 27771,
      "training_step_time": 7.7545082569122314
    },
    {
      "epoch": 2.6485443115234374e-05,
      "model_forward_time": 1.7275722026824951,
      "step": 27772
    },
    {
      "epoch": 2.6485443115234374e-05,
      "step": 27772,
      "training_step_time": 7.220738172531128
    },
    {
      "epoch": 2.648639678955078e-05,
      "model_forward_time": 1.7432866096496582,
      "step": 27773
    },
    {
      "epoch": 2.648639678955078e-05,
      "step": 27773,
      "training_step_time": 6.920936584472656
    },
    {
      "epoch": 2.648735046386719e-05,
      "model_forward_time": 1.7331428527832031,
      "step": 27774
    },
    {
      "epoch": 2.648735046386719e-05,
      "step": 27774,
      "training_step_time": 6.867356777191162
    },
    {
      "epoch": 2.6488304138183592e-05,
      "model_forward_time": 1.7367377281188965,
      "step": 27775
    },
    {
      "epoch": 2.6488304138183592e-05,
      "step": 27775,
      "training_step_time": 6.86104679107666
    },
    {
      "epoch": 2.64892578125e-05,
      "model_forward_time": 1.7323307991027832,
      "step": 27776
    },
    {
      "epoch": 2.64892578125e-05,
      "step": 27776,
      "training_step_time": 6.830726623535156
    },
    {
      "epoch": 2.6490211486816407e-05,
      "model_forward_time": 1.7281992435455322,
      "step": 27777
    },
    {
      "epoch": 2.6490211486816407e-05,
      "step": 27777,
      "training_step_time": 6.835880994796753
    },
    {
      "epoch": 2.649116516113281e-05,
      "model_forward_time": 1.7453563213348389,
      "step": 27778
    },
    {
      "epoch": 2.649116516113281e-05,
      "step": 27778,
      "training_step_time": 7.708120346069336
    },
    {
      "epoch": 2.649211883544922e-05,
      "model_forward_time": 2.8958983421325684,
      "step": 27779
    },
    {
      "epoch": 2.649211883544922e-05,
      "step": 27779,
      "training_step_time": 9.410046339035034
    },
    {
      "epoch": 2.6493072509765626e-05,
      "grad_norm": 0.12873443961143494,
      "learning_rate": 8.64721920176972e-06,
      "loss": 0.0611,
      "step": 27780
    },
    {
      "epoch": 2.6493072509765626e-05,
      "model_forward_time": 1.738593339920044,
      "step": 27780
    },
    {
      "dynamics_loss_avg": 0.07464914489537477,
      "epoch": 2.6493072509765626e-05,
      "step": 27780
    },
    {
      "action_loss_avg": 0.003565267170779407,
      "epoch": 2.6493072509765626e-05,
      "step": 27780
    },
    {
      "epoch": 2.6493072509765626e-05,
      "step": 27780,
      "training_step_time": 8.358571290969849
    },
    {
      "epoch": 2.649402618408203e-05,
      "model_forward_time": 1.730635643005371,
      "step": 27781
    },
    {
      "epoch": 2.649402618408203e-05,
      "step": 27781,
      "training_step_time": 7.338253736495972
    },
    {
      "epoch": 2.6494979858398438e-05,
      "model_forward_time": 1.7284772396087646,
      "step": 27782
    },
    {
      "epoch": 2.6494979858398438e-05,
      "step": 27782,
      "training_step_time": 6.8467676639556885
    },
    {
      "epoch": 2.6495933532714845e-05,
      "model_forward_time": 1.7255117893218994,
      "step": 27783
    },
    {
      "epoch": 2.6495933532714845e-05,
      "step": 27783,
      "training_step_time": 6.817677736282349
    },
    {
      "epoch": 2.649688720703125e-05,
      "model_forward_time": 1.733126163482666,
      "step": 27784
    },
    {
      "epoch": 2.649688720703125e-05,
      "step": 27784,
      "training_step_time": 6.823138475418091
    },
    {
      "epoch": 2.6497840881347656e-05,
      "model_forward_time": 1.7030818462371826,
      "step": 27785
    },
    {
      "epoch": 2.6497840881347656e-05,
      "step": 27785,
      "training_step_time": 6.832371234893799
    },
    {
      "epoch": 2.6498794555664064e-05,
      "model_forward_time": 1.7121508121490479,
      "step": 27786
    },
    {
      "epoch": 2.6498794555664064e-05,
      "step": 27786,
      "training_step_time": 6.865063428878784
    },
    {
      "epoch": 2.6499748229980468e-05,
      "model_forward_time": 1.7270488739013672,
      "step": 27787
    },
    {
      "epoch": 2.6499748229980468e-05,
      "step": 27787,
      "training_step_time": 7.691280364990234
    },
    {
      "epoch": 2.6500701904296875e-05,
      "model_forward_time": 2.857679843902588,
      "step": 27788
    },
    {
      "epoch": 2.6500701904296875e-05,
      "step": 27788,
      "training_step_time": 10.525389909744263
    },
    {
      "epoch": 2.6501655578613283e-05,
      "model_forward_time": 1.716153860092163,
      "step": 27789
    },
    {
      "epoch": 2.6501655578613283e-05,
      "step": 27789,
      "training_step_time": 8.854635953903198
    },
    {
      "epoch": 2.6502609252929687e-05,
      "grad_norm": 0.22016851603984833,
      "learning_rate": 8.646087961706165e-06,
      "loss": 0.0572,
      "step": 27790
    },
    {
      "epoch": 2.6502609252929687e-05,
      "model_forward_time": 1.7077219486236572,
      "step": 27790
    },
    {
      "dynamics_loss_avg": 0.04257559645920992,
      "epoch": 2.6502609252929687e-05,
      "step": 27790
    },
    {
      "action_loss_avg": 0.0017906818131450564,
      "epoch": 2.6502609252929687e-05,
      "step": 27790
    },
    {
      "epoch": 2.6502609252929687e-05,
      "step": 27790,
      "training_step_time": 7.750467777252197
    },
    {
      "epoch": 2.6503562927246094e-05,
      "model_forward_time": 1.706153392791748,
      "step": 27791
    },
    {
      "epoch": 2.6503562927246094e-05,
      "step": 27791,
      "training_step_time": 6.8546600341796875
    },
    {
      "epoch": 2.65045166015625e-05,
      "model_forward_time": 1.3319587707519531,
      "step": 27792
    },
    {
      "epoch": 2.65045166015625e-05,
      "step": 27792,
      "training_step_time": 6.843634128570557
    },
    {
      "epoch": 2.6505470275878906e-05,
      "model_forward_time": 1.7079887390136719,
      "step": 27793
    },
    {
      "epoch": 2.6505470275878906e-05,
      "step": 27793,
      "training_step_time": 6.9264936447143555
    },
    {
      "epoch": 2.6506423950195313e-05,
      "model_forward_time": 1.7081389427185059,
      "step": 27794
    },
    {
      "epoch": 2.6506423950195313e-05,
      "step": 27794,
      "training_step_time": 6.912582159042358
    },
    {
      "epoch": 2.650737762451172e-05,
      "model_forward_time": 1.7073934078216553,
      "step": 27795
    },
    {
      "epoch": 2.650737762451172e-05,
      "step": 27795,
      "training_step_time": 6.908294916152954
    },
    {
      "epoch": 2.6508331298828124e-05,
      "model_forward_time": 1.7142143249511719,
      "step": 27796
    },
    {
      "epoch": 2.6508331298828124e-05,
      "step": 27796,
      "training_step_time": 9.07920527458191
    },
    {
      "epoch": 2.6509284973144532e-05,
      "model_forward_time": 2.869948387145996,
      "step": 27797
    },
    {
      "epoch": 2.6509284973144532e-05,
      "step": 27797,
      "training_step_time": 8.796269178390503
    },
    {
      "epoch": 2.6510238647460936e-05,
      "model_forward_time": 1.7071824073791504,
      "step": 27798
    },
    {
      "epoch": 2.6510238647460936e-05,
      "step": 27798,
      "training_step_time": 7.853121757507324
    },
    {
      "epoch": 2.6511192321777343e-05,
      "model_forward_time": 1.7053568363189697,
      "step": 27799
    },
    {
      "epoch": 2.6511192321777343e-05,
      "step": 27799,
      "training_step_time": 7.72838282585144
    },
    {
      "epoch": 2.651214599609375e-05,
      "grad_norm": 0.23293475806713104,
      "learning_rate": 8.644956322911908e-06,
      "loss": 0.0617,
      "step": 27800
    },
    {
      "epoch": 2.651214599609375e-05,
      "model_forward_time": 1.3360130786895752,
      "step": 27800
    },
    {
      "dynamics_loss_avg": 0.06591282635927201,
      "epoch": 2.651214599609375e-05,
      "step": 27800
    },
    {
      "action_loss_avg": 0.002780910092405975,
      "epoch": 2.651214599609375e-05,
      "step": 27800
    },
    {
      "epoch": 2.651214599609375e-05,
      "step": 27800,
      "training_step_time": 7.338832855224609
    },
    {
      "epoch": 2.6513099670410155e-05,
      "model_forward_time": 1.7090957164764404,
      "step": 27801
    },
    {
      "epoch": 2.6513099670410155e-05,
      "step": 27801,
      "training_step_time": 6.88218879699707
    },
    {
      "epoch": 2.6514053344726562e-05,
      "model_forward_time": 1.7139008045196533,
      "step": 27802
    },
    {
      "epoch": 2.6514053344726562e-05,
      "step": 27802,
      "training_step_time": 6.909766674041748
    },
    {
      "epoch": 2.651500701904297e-05,
      "model_forward_time": 1.7056002616882324,
      "step": 27803
    },
    {
      "epoch": 2.651500701904297e-05,
      "step": 27803,
      "training_step_time": 6.903810739517212
    },
    {
      "epoch": 2.6515960693359374e-05,
      "model_forward_time": 1.7097067832946777,
      "step": 27804
    },
    {
      "epoch": 2.6515960693359374e-05,
      "step": 27804,
      "training_step_time": 6.8907999992370605
    },
    {
      "epoch": 2.651691436767578e-05,
      "model_forward_time": 1.7185072898864746,
      "step": 27805
    },
    {
      "epoch": 2.651691436767578e-05,
      "step": 27805,
      "training_step_time": 8.574054479598999
    },
    {
      "epoch": 2.651786804199219e-05,
      "model_forward_time": 2.7986814975738525,
      "step": 27806
    },
    {
      "epoch": 2.651786804199219e-05,
      "step": 27806,
      "training_step_time": 9.611085414886475
    },
    {
      "epoch": 2.6518821716308592e-05,
      "model_forward_time": 1.7096378803253174,
      "step": 27807
    },
    {
      "epoch": 2.6518821716308592e-05,
      "step": 27807,
      "training_step_time": 11.321656942367554
    },
    {
      "epoch": 2.6519775390625e-05,
      "model_forward_time": 1.7070832252502441,
      "step": 27808
    },
    {
      "epoch": 2.6519775390625e-05,
      "step": 27808,
      "training_step_time": 7.698323965072632
    },
    {
      "epoch": 2.6520729064941407e-05,
      "model_forward_time": 1.7088494300842285,
      "step": 27809
    },
    {
      "epoch": 2.6520729064941407e-05,
      "step": 27809,
      "training_step_time": 7.751702785491943
    },
    {
      "epoch": 2.652168273925781e-05,
      "grad_norm": 0.17540258169174194,
      "learning_rate": 8.64382428551071e-06,
      "loss": 0.0574,
      "step": 27810
    },
    {
      "epoch": 2.652168273925781e-05,
      "model_forward_time": 1.716456413269043,
      "step": 27810
    },
    {
      "dynamics_loss_avg": 0.07103925701230765,
      "epoch": 2.652168273925781e-05,
      "step": 27810
    },
    {
      "action_loss_avg": 0.0034445223514921962,
      "epoch": 2.652168273925781e-05,
      "step": 27810
    },
    {
      "epoch": 2.652168273925781e-05,
      "step": 27810,
      "training_step_time": 6.893686771392822
    },
    {
      "epoch": 2.652263641357422e-05,
      "model_forward_time": 1.7114856243133545,
      "step": 27811
    },
    {
      "epoch": 2.652263641357422e-05,
      "step": 27811,
      "training_step_time": 6.884809494018555
    },
    {
      "epoch": 2.6523590087890626e-05,
      "model_forward_time": 1.7063448429107666,
      "step": 27812
    },
    {
      "epoch": 2.6523590087890626e-05,
      "step": 27812,
      "training_step_time": 6.830152750015259
    },
    {
      "epoch": 2.652454376220703e-05,
      "model_forward_time": 1.7102727890014648,
      "step": 27813
    },
    {
      "epoch": 2.652454376220703e-05,
      "step": 27813,
      "training_step_time": 6.886640310287476
    },
    {
      "epoch": 2.6525497436523438e-05,
      "model_forward_time": 1.7329761981964111,
      "step": 27814
    },
    {
      "epoch": 2.6525497436523438e-05,
      "step": 27814,
      "training_step_time": 7.750437259674072
    },
    {
      "epoch": 2.6526451110839845e-05,
      "model_forward_time": 2.881939172744751,
      "step": 27815
    },
    {
      "epoch": 2.6526451110839845e-05,
      "step": 27815,
      "training_step_time": 9.70689082145691
    },
    {
      "epoch": 2.652740478515625e-05,
      "model_forward_time": 1.7350506782531738,
      "step": 27816
    },
    {
      "epoch": 2.652740478515625e-05,
      "step": 27816,
      "training_step_time": 11.090559720993042
    },
    {
      "epoch": 2.6528358459472656e-05,
      "model_forward_time": 1.7261648178100586,
      "step": 27817
    },
    {
      "epoch": 2.6528358459472656e-05,
      "step": 27817,
      "training_step_time": 7.899033308029175
    },
    {
      "epoch": 2.6529312133789064e-05,
      "model_forward_time": 1.7254536151885986,
      "step": 27818
    },
    {
      "epoch": 2.6529312133789064e-05,
      "step": 27818,
      "training_step_time": 8.464099884033203
    },
    {
      "epoch": 2.6530265808105468e-05,
      "model_forward_time": 1.7330725193023682,
      "step": 27819
    },
    {
      "epoch": 2.6530265808105468e-05,
      "step": 27819,
      "training_step_time": 6.888821601867676
    },
    {
      "epoch": 2.6531219482421875e-05,
      "grad_norm": 0.21464020013809204,
      "learning_rate": 8.642691849626364e-06,
      "loss": 0.0621,
      "step": 27820
    },
    {
      "epoch": 2.6531219482421875e-05,
      "model_forward_time": 1.7258341312408447,
      "step": 27820
    },
    {
      "dynamics_loss_avg": 0.050056496635079384,
      "epoch": 2.6531219482421875e-05,
      "step": 27820
    },
    {
      "action_loss_avg": 0.0027113750285934655,
      "epoch": 2.6531219482421875e-05,
      "step": 27820
    },
    {
      "epoch": 2.6531219482421875e-05,
      "step": 27820,
      "training_step_time": 6.894030332565308
    },
    {
      "epoch": 2.6532173156738283e-05,
      "model_forward_time": 1.7236180305480957,
      "step": 27821
    },
    {
      "epoch": 2.6532173156738283e-05,
      "step": 27821,
      "training_step_time": 6.849085330963135
    },
    {
      "epoch": 2.6533126831054687e-05,
      "model_forward_time": 1.7252063751220703,
      "step": 27822
    },
    {
      "epoch": 2.6533126831054687e-05,
      "step": 27822,
      "training_step_time": 6.856395959854126
    },
    {
      "epoch": 2.6534080505371094e-05,
      "model_forward_time": 1.733597993850708,
      "step": 27823
    },
    {
      "epoch": 2.6534080505371094e-05,
      "step": 27823,
      "training_step_time": 7.721625328063965
    },
    {
      "epoch": 2.65350341796875e-05,
      "model_forward_time": 1.7427537441253662,
      "step": 27824
    },
    {
      "epoch": 2.65350341796875e-05,
      "step": 27824,
      "training_step_time": 9.441722869873047
    },
    {
      "epoch": 2.6535987854003906e-05,
      "model_forward_time": 2.926957607269287,
      "step": 27825
    },
    {
      "epoch": 2.6535987854003906e-05,
      "step": 27825,
      "training_step_time": 9.905761480331421
    },
    {
      "epoch": 2.6536941528320313e-05,
      "model_forward_time": 1.7288987636566162,
      "step": 27826
    },
    {
      "epoch": 2.6536941528320313e-05,
      "step": 27826,
      "training_step_time": 8.374364137649536
    },
    {
      "epoch": 2.653789520263672e-05,
      "model_forward_time": 1.730654239654541,
      "step": 27827
    },
    {
      "epoch": 2.653789520263672e-05,
      "step": 27827,
      "training_step_time": 7.708837509155273
    },
    {
      "epoch": 2.6538848876953124e-05,
      "model_forward_time": 1.7304844856262207,
      "step": 27828
    },
    {
      "epoch": 2.6538848876953124e-05,
      "step": 27828,
      "training_step_time": 7.307236671447754
    },
    {
      "epoch": 2.6539802551269532e-05,
      "model_forward_time": 1.7350025177001953,
      "step": 27829
    },
    {
      "epoch": 2.6539802551269532e-05,
      "step": 27829,
      "training_step_time": 6.818073749542236
    },
    {
      "epoch": 2.6540756225585936e-05,
      "grad_norm": 0.20063835382461548,
      "learning_rate": 8.641559015382716e-06,
      "loss": 0.0566,
      "step": 27830
    },
    {
      "epoch": 2.6540756225585936e-05,
      "model_forward_time": 1.72385573387146,
      "step": 27830
    },
    {
      "dynamics_loss_avg": 0.0572813268750906,
      "epoch": 2.6540756225585936e-05,
      "step": 27830
    },
    {
      "action_loss_avg": 0.002235326904337853,
      "epoch": 2.6540756225585936e-05,
      "step": 27830
    },
    {
      "epoch": 2.6540756225585936e-05,
      "step": 27830,
      "training_step_time": 6.814800262451172
    },
    {
      "epoch": 2.6541709899902343e-05,
      "model_forward_time": 1.7274422645568848,
      "step": 27831
    },
    {
      "epoch": 2.6541709899902343e-05,
      "step": 27831,
      "training_step_time": 6.856608629226685
    },
    {
      "epoch": 2.654266357421875e-05,
      "model_forward_time": 1.728855848312378,
      "step": 27832
    },
    {
      "epoch": 2.654266357421875e-05,
      "step": 27832,
      "training_step_time": 7.75108528137207
    },
    {
      "epoch": 2.6543617248535155e-05,
      "model_forward_time": 1.7998998165130615,
      "step": 27833
    },
    {
      "epoch": 2.6543617248535155e-05,
      "step": 27833,
      "training_step_time": 8.548132419586182
    },
    {
      "epoch": 2.6544570922851562e-05,
      "model_forward_time": 2.9213199615478516,
      "step": 27834
    },
    {
      "epoch": 2.6544570922851562e-05,
      "step": 27834,
      "training_step_time": 9.37388563156128
    },
    {
      "epoch": 2.654552459716797e-05,
      "model_forward_time": 1.7386512756347656,
      "step": 27835
    },
    {
      "epoch": 2.654552459716797e-05,
      "step": 27835,
      "training_step_time": 9.505688428878784
    },
    {
      "epoch": 2.6546478271484374e-05,
      "model_forward_time": 1.7349600791931152,
      "step": 27836
    },
    {
      "epoch": 2.6546478271484374e-05,
      "step": 27836,
      "training_step_time": 7.642976999282837
    },
    {
      "epoch": 2.654743194580078e-05,
      "model_forward_time": 1.7279024124145508,
      "step": 27837
    },
    {
      "epoch": 2.654743194580078e-05,
      "step": 27837,
      "training_step_time": 7.6090264320373535
    },
    {
      "epoch": 2.654838562011719e-05,
      "model_forward_time": 1.7315926551818848,
      "step": 27838
    },
    {
      "epoch": 2.654838562011719e-05,
      "step": 27838,
      "training_step_time": 6.896543264389038
    },
    {
      "epoch": 2.6549339294433592e-05,
      "model_forward_time": 1.7294015884399414,
      "step": 27839
    },
    {
      "epoch": 2.6549339294433592e-05,
      "step": 27839,
      "training_step_time": 6.855619668960571
    },
    {
      "epoch": 2.655029296875e-05,
      "grad_norm": 0.10017336905002594,
      "learning_rate": 8.64042578290365e-06,
      "loss": 0.0595,
      "step": 27840
    },
    {
      "epoch": 2.655029296875e-05,
      "model_forward_time": 1.7307779788970947,
      "step": 27840
    },
    {
      "dynamics_loss_avg": 0.058201977610588075,
      "epoch": 2.655029296875e-05,
      "step": 27840
    },
    {
      "action_loss_avg": 0.0027224737510550766,
      "epoch": 2.655029296875e-05,
      "step": 27840
    },
    {
      "epoch": 2.655029296875e-05,
      "step": 27840,
      "training_step_time": 6.919924974441528
    },
    {
      "epoch": 2.6551246643066407e-05,
      "model_forward_time": 1.72800612449646,
      "step": 27841
    },
    {
      "epoch": 2.6551246643066407e-05,
      "step": 27841,
      "training_step_time": 7.687788724899292
    },
    {
      "epoch": 2.655220031738281e-05,
      "model_forward_time": 1.7549316883087158,
      "step": 27842
    },
    {
      "epoch": 2.655220031738281e-05,
      "step": 27842,
      "training_step_time": 7.691791772842407
    },
    {
      "epoch": 2.655315399169922e-05,
      "model_forward_time": 2.913062810897827,
      "step": 27843
    },
    {
      "epoch": 2.655315399169922e-05,
      "step": 27843,
      "training_step_time": 10.284252882003784
    },
    {
      "epoch": 2.6554107666015626e-05,
      "model_forward_time": 1.7302403450012207,
      "step": 27844
    },
    {
      "epoch": 2.6554107666015626e-05,
      "step": 27844,
      "training_step_time": 8.738418340682983
    },
    {
      "epoch": 2.655506134033203e-05,
      "model_forward_time": 1.7340583801269531,
      "step": 27845
    },
    {
      "epoch": 2.655506134033203e-05,
      "step": 27845,
      "training_step_time": 7.7558934688568115
    },
    {
      "epoch": 2.6556015014648438e-05,
      "model_forward_time": 1.7403521537780762,
      "step": 27846
    },
    {
      "epoch": 2.6556015014648438e-05,
      "step": 27846,
      "training_step_time": 7.695433616638184
    },
    {
      "epoch": 2.6556968688964845e-05,
      "model_forward_time": 1.7284293174743652,
      "step": 27847
    },
    {
      "epoch": 2.6556968688964845e-05,
      "step": 27847,
      "training_step_time": 6.8540284633636475
    },
    {
      "epoch": 2.655792236328125e-05,
      "model_forward_time": 1.7111003398895264,
      "step": 27848
    },
    {
      "epoch": 2.655792236328125e-05,
      "step": 27848,
      "training_step_time": 6.854992389678955
    },
    {
      "epoch": 2.6558876037597656e-05,
      "model_forward_time": 1.7254679203033447,
      "step": 27849
    },
    {
      "epoch": 2.6558876037597656e-05,
      "step": 27849,
      "training_step_time": 6.834193706512451
    },
    {
      "epoch": 2.6559829711914064e-05,
      "grad_norm": 0.16938385367393494,
      "learning_rate": 8.639292152313092e-06,
      "loss": 0.0603,
      "step": 27850
    },
    {
      "epoch": 2.6559829711914064e-05,
      "model_forward_time": 1.3497014045715332,
      "step": 27850
    },
    {
      "dynamics_loss_avg": 0.042349426075816154,
      "epoch": 2.6559829711914064e-05,
      "step": 27850
    },
    {
      "action_loss_avg": 0.0017037284909747542,
      "epoch": 2.6559829711914064e-05,
      "step": 27850
    },
    {
      "epoch": 2.6559829711914064e-05,
      "step": 27850,
      "training_step_time": 7.693031072616577
    },
    {
      "epoch": 2.6560783386230468e-05,
      "model_forward_time": 1.718259572982788,
      "step": 27851
    },
    {
      "epoch": 2.6560783386230468e-05,
      "step": 27851,
      "training_step_time": 7.633135557174683
    },
    {
      "epoch": 2.6561737060546875e-05,
      "model_forward_time": 2.287058115005493,
      "step": 27852
    },
    {
      "epoch": 2.6561737060546875e-05,
      "step": 27852,
      "training_step_time": 9.81784176826477
    },
    {
      "epoch": 2.6562690734863283e-05,
      "model_forward_time": 1.727243423461914,
      "step": 27853
    },
    {
      "epoch": 2.6562690734863283e-05,
      "step": 27853,
      "training_step_time": 8.847969770431519
    },
    {
      "epoch": 2.6563644409179687e-05,
      "model_forward_time": 1.7217562198638916,
      "step": 27854
    },
    {
      "epoch": 2.6563644409179687e-05,
      "step": 27854,
      "training_step_time": 7.783880710601807
    },
    {
      "epoch": 2.6564598083496094e-05,
      "model_forward_time": 1.7408452033996582,
      "step": 27855
    },
    {
      "epoch": 2.6564598083496094e-05,
      "step": 27855,
      "training_step_time": 7.734982967376709
    },
    {
      "epoch": 2.65655517578125e-05,
      "model_forward_time": 1.7217793464660645,
      "step": 27856
    },
    {
      "epoch": 2.65655517578125e-05,
      "step": 27856,
      "training_step_time": 7.300280332565308
    },
    {
      "epoch": 2.6566505432128906e-05,
      "model_forward_time": 1.7250621318817139,
      "step": 27857
    },
    {
      "epoch": 2.6566505432128906e-05,
      "step": 27857,
      "training_step_time": 6.85653281211853
    },
    {
      "epoch": 2.6567459106445313e-05,
      "model_forward_time": 1.7384593486785889,
      "step": 27858
    },
    {
      "epoch": 2.6567459106445313e-05,
      "step": 27858,
      "training_step_time": 6.8735198974609375
    },
    {
      "epoch": 2.656841278076172e-05,
      "model_forward_time": 1.7238171100616455,
      "step": 27859
    },
    {
      "epoch": 2.656841278076172e-05,
      "step": 27859,
      "training_step_time": 7.73502254486084
    },
    {
      "epoch": 2.6569366455078124e-05,
      "grad_norm": 0.17368149757385254,
      "learning_rate": 8.638158123735015e-06,
      "loss": 0.0572,
      "step": 27860
    },
    {
      "epoch": 2.6569366455078124e-05,
      "model_forward_time": 1.7208349704742432,
      "step": 27860
    },
    {
      "dynamics_loss_avg": 0.05837739240378141,
      "epoch": 2.6569366455078124e-05,
      "step": 27860
    },
    {
      "action_loss_avg": 0.0016911384591367096,
      "epoch": 2.6569366455078124e-05,
      "step": 27860
    },
    {
      "epoch": 2.6569366455078124e-05,
      "step": 27860,
      "training_step_time": 7.660404682159424
    },
    {
      "epoch": 2.6570320129394532e-05,
      "model_forward_time": 1.7231287956237793,
      "step": 27861
    },
    {
      "epoch": 2.6570320129394532e-05,
      "step": 27861,
      "training_step_time": 9.953428030014038
    },
    {
      "epoch": 2.6571273803710936e-05,
      "model_forward_time": 1.7157366275787354,
      "step": 27862
    },
    {
      "epoch": 2.6571273803710936e-05,
      "step": 27862,
      "training_step_time": 7.851224184036255
    },
    {
      "epoch": 2.6572227478027343e-05,
      "model_forward_time": 1.721836805343628,
      "step": 27863
    },
    {
      "epoch": 2.6572227478027343e-05,
      "step": 27863,
      "training_step_time": 7.834959030151367
    },
    {
      "epoch": 2.657318115234375e-05,
      "model_forward_time": 1.7250587940216064,
      "step": 27864
    },
    {
      "epoch": 2.657318115234375e-05,
      "step": 27864,
      "training_step_time": 7.654239177703857
    },
    {
      "epoch": 2.6574134826660155e-05,
      "model_forward_time": 1.7208342552185059,
      "step": 27865
    },
    {
      "epoch": 2.6574134826660155e-05,
      "step": 27865,
      "training_step_time": 7.33914852142334
    },
    {
      "epoch": 2.6575088500976562e-05,
      "model_forward_time": 1.7254912853240967,
      "step": 27866
    },
    {
      "epoch": 2.6575088500976562e-05,
      "step": 27866,
      "training_step_time": 6.801110506057739
    },
    {
      "epoch": 2.657604217529297e-05,
      "model_forward_time": 1.745187520980835,
      "step": 27867
    },
    {
      "epoch": 2.657604217529297e-05,
      "step": 27867,
      "training_step_time": 6.887890338897705
    },
    {
      "epoch": 2.6576995849609374e-05,
      "model_forward_time": 1.7350540161132812,
      "step": 27868
    },
    {
      "epoch": 2.6576995849609374e-05,
      "step": 27868,
      "training_step_time": 7.712851047515869
    },
    {
      "epoch": 2.657794952392578e-05,
      "model_forward_time": 1.7169249057769775,
      "step": 27869
    },
    {
      "epoch": 2.657794952392578e-05,
      "step": 27869,
      "training_step_time": 7.633627891540527
    },
    {
      "epoch": 2.657890319824219e-05,
      "grad_norm": 0.11579027026891708,
      "learning_rate": 8.637023697293436e-06,
      "loss": 0.0575,
      "step": 27870
    },
    {
      "epoch": 2.657890319824219e-05,
      "model_forward_time": 1.7220821380615234,
      "step": 27870
    },
    {
      "dynamics_loss_avg": 0.04390841796994209,
      "epoch": 2.657890319824219e-05,
      "step": 27870
    },
    {
      "action_loss_avg": 0.0015083196922205389,
      "epoch": 2.657890319824219e-05,
      "step": 27870
    },
    {
      "epoch": 2.657890319824219e-05,
      "step": 27870,
      "training_step_time": 8.259031295776367
    },
    {
      "epoch": 2.6579856872558592e-05,
      "model_forward_time": 2.32120418548584,
      "step": 27871
    },
    {
      "epoch": 2.6579856872558592e-05,
      "step": 27871,
      "training_step_time": 7.895214319229126
    },
    {
      "epoch": 2.6580810546875e-05,
      "model_forward_time": 1.7155380249023438,
      "step": 27872
    },
    {
      "epoch": 2.6580810546875e-05,
      "step": 27872,
      "training_step_time": 8.970024347305298
    },
    {
      "epoch": 2.6581764221191407e-05,
      "model_forward_time": 1.7293074131011963,
      "step": 27873
    },
    {
      "epoch": 2.6581764221191407e-05,
      "step": 27873,
      "training_step_time": 7.7797815799713135
    },
    {
      "epoch": 2.658271789550781e-05,
      "model_forward_time": 1.3691625595092773,
      "step": 27874
    },
    {
      "epoch": 2.658271789550781e-05,
      "step": 27874,
      "training_step_time": 8.432057857513428
    },
    {
      "epoch": 2.658367156982422e-05,
      "model_forward_time": 1.720529556274414,
      "step": 27875
    },
    {
      "epoch": 2.658367156982422e-05,
      "step": 27875,
      "training_step_time": 6.96676778793335
    },
    {
      "epoch": 2.6584625244140626e-05,
      "model_forward_time": 1.7197329998016357,
      "step": 27876
    },
    {
      "epoch": 2.6584625244140626e-05,
      "step": 27876,
      "training_step_time": 6.955944776535034
    },
    {
      "epoch": 2.658557891845703e-05,
      "model_forward_time": 1.7197303771972656,
      "step": 27877
    },
    {
      "epoch": 2.658557891845703e-05,
      "step": 27877,
      "training_step_time": 7.703861713409424
    },
    {
      "epoch": 2.6586532592773438e-05,
      "model_forward_time": 1.7234175205230713,
      "step": 27878
    },
    {
      "epoch": 2.6586532592773438e-05,
      "step": 27878,
      "training_step_time": 7.617880344390869
    },
    {
      "epoch": 2.6587486267089845e-05,
      "model_forward_time": 1.769700288772583,
      "step": 27879
    },
    {
      "epoch": 2.6587486267089845e-05,
      "step": 27879,
      "training_step_time": 8.369688272476196
    },
    {
      "epoch": 2.658843994140625e-05,
      "grad_norm": 0.13875360786914825,
      "learning_rate": 8.635888873112414e-06,
      "loss": 0.0581,
      "step": 27880
    },
    {
      "epoch": 2.658843994140625e-05,
      "model_forward_time": 2.3071560859680176,
      "step": 27880
    },
    {
      "dynamics_loss_avg": 0.05650992188602686,
      "epoch": 2.658843994140625e-05,
      "step": 27880
    },
    {
      "action_loss_avg": 0.003386977512855083,
      "epoch": 2.658843994140625e-05,
      "step": 27880
    },
    {
      "epoch": 2.658843994140625e-05,
      "step": 27880,
      "training_step_time": 8.264781475067139
    },
    {
      "epoch": 2.6589393615722656e-05,
      "model_forward_time": 1.724646806716919,
      "step": 27881
    },
    {
      "epoch": 2.6589393615722656e-05,
      "step": 27881,
      "training_step_time": 8.882972240447998
    },
    {
      "epoch": 2.6590347290039064e-05,
      "model_forward_time": 1.7214908599853516,
      "step": 27882
    },
    {
      "epoch": 2.6590347290039064e-05,
      "step": 27882,
      "training_step_time": 7.727362155914307
    },
    {
      "epoch": 2.6591300964355468e-05,
      "model_forward_time": 1.7173352241516113,
      "step": 27883
    },
    {
      "epoch": 2.6591300964355468e-05,
      "step": 27883,
      "training_step_time": 7.750576734542847
    },
    {
      "epoch": 2.6592254638671875e-05,
      "model_forward_time": 1.727097511291504,
      "step": 27884
    },
    {
      "epoch": 2.6592254638671875e-05,
      "step": 27884,
      "training_step_time": 7.278979301452637
    },
    {
      "epoch": 2.6593208312988283e-05,
      "model_forward_time": 1.7231521606445312,
      "step": 27885
    },
    {
      "epoch": 2.6593208312988283e-05,
      "step": 27885,
      "training_step_time": 6.980100631713867
    },
    {
      "epoch": 2.6594161987304687e-05,
      "model_forward_time": 1.7225298881530762,
      "step": 27886
    },
    {
      "epoch": 2.6594161987304687e-05,
      "step": 27886,
      "training_step_time": 7.723702907562256
    },
    {
      "epoch": 2.6595115661621094e-05,
      "model_forward_time": 1.7207891941070557,
      "step": 27887
    },
    {
      "epoch": 2.6595115661621094e-05,
      "step": 27887,
      "training_step_time": 7.623464822769165
    },
    {
      "epoch": 2.65960693359375e-05,
      "model_forward_time": 1.7280631065368652,
      "step": 27888
    },
    {
      "epoch": 2.65960693359375e-05,
      "step": 27888,
      "training_step_time": 7.561882495880127
    },
    {
      "epoch": 2.6597023010253906e-05,
      "model_forward_time": 2.3253026008605957,
      "step": 27889
    },
    {
      "epoch": 2.6597023010253906e-05,
      "step": 27889,
      "training_step_time": 9.539212226867676
    },
    {
      "epoch": 2.6597976684570313e-05,
      "grad_norm": 0.14687125384807587,
      "learning_rate": 8.634753651316053e-06,
      "loss": 0.0635,
      "step": 27890
    },
    {
      "epoch": 2.6597976684570313e-05,
      "model_forward_time": 1.7227206230163574,
      "step": 27890
    },
    {
      "dynamics_loss_avg": 0.05586208440363407,
      "epoch": 2.6597976684570313e-05,
      "step": 27890
    },
    {
      "action_loss_avg": 0.0022074237815104423,
      "epoch": 2.6597976684570313e-05,
      "step": 27890
    },
    {
      "epoch": 2.6597976684570313e-05,
      "step": 27890,
      "training_step_time": 7.803613901138306
    },
    {
      "epoch": 2.6598930358886717e-05,
      "model_forward_time": 1.731126070022583,
      "step": 27891
    },
    {
      "epoch": 2.6598930358886717e-05,
      "step": 27891,
      "training_step_time": 7.824353218078613
    },
    {
      "epoch": 2.6599884033203124e-05,
      "model_forward_time": 1.7227988243103027,
      "step": 27892
    },
    {
      "epoch": 2.6599884033203124e-05,
      "step": 27892,
      "training_step_time": 7.164464235305786
    },
    {
      "epoch": 2.6600837707519532e-05,
      "model_forward_time": 1.7484228610992432,
      "step": 27893
    },
    {
      "epoch": 2.6600837707519532e-05,
      "step": 27893,
      "training_step_time": 7.704613208770752
    },
    {
      "epoch": 2.6601791381835936e-05,
      "model_forward_time": 1.7288398742675781,
      "step": 27894
    },
    {
      "epoch": 2.6601791381835936e-05,
      "step": 27894,
      "training_step_time": 7.000009059906006
    },
    {
      "epoch": 2.6602745056152343e-05,
      "model_forward_time": 1.7355525493621826,
      "step": 27895
    },
    {
      "epoch": 2.6602745056152343e-05,
      "step": 27895,
      "training_step_time": 7.759738206863403
    },
    {
      "epoch": 2.660369873046875e-05,
      "model_forward_time": 1.7390108108520508,
      "step": 27896
    },
    {
      "epoch": 2.660369873046875e-05,
      "step": 27896,
      "training_step_time": 8.09708857536316
    },
    {
      "epoch": 2.6604652404785155e-05,
      "model_forward_time": 1.7305731773376465,
      "step": 27897
    },
    {
      "epoch": 2.6604652404785155e-05,
      "step": 27897,
      "training_step_time": 7.671108245849609
    },
    {
      "epoch": 2.6605606079101562e-05,
      "model_forward_time": 2.3182411193847656,
      "step": 27898
    },
    {
      "epoch": 2.6605606079101562e-05,
      "step": 27898,
      "training_step_time": 8.775722980499268
    },
    {
      "epoch": 2.660655975341797e-05,
      "model_forward_time": 1.723438024520874,
      "step": 27899
    },
    {
      "epoch": 2.660655975341797e-05,
      "step": 27899,
      "training_step_time": 7.683989524841309
    },
    {
      "epoch": 2.6607513427734374e-05,
      "grad_norm": 0.16390302777290344,
      "learning_rate": 8.633618032028495e-06,
      "loss": 0.0614,
      "step": 27900
    },
    {
      "epoch": 2.6607513427734374e-05,
      "model_forward_time": 1.7300965785980225,
      "step": 27900
    },
    {
      "dynamics_loss_avg": 0.05123917479068041,
      "epoch": 2.6607513427734374e-05,
      "step": 27900
    },
    {
      "action_loss_avg": 0.001571072600199841,
      "epoch": 2.6607513427734374e-05,
      "step": 27900
    },
    {
      "epoch": 2.6607513427734374e-05,
      "step": 27900,
      "training_step_time": 8.40731930732727
    },
    {
      "epoch": 2.660846710205078e-05,
      "model_forward_time": 1.7324833869934082,
      "step": 27901
    },
    {
      "epoch": 2.660846710205078e-05,
      "step": 27901,
      "training_step_time": 7.635700464248657
    },
    {
      "epoch": 2.660942077636719e-05,
      "model_forward_time": 1.7205314636230469,
      "step": 27902
    },
    {
      "epoch": 2.660942077636719e-05,
      "step": 27902,
      "training_step_time": 7.688710451126099
    },
    {
      "epoch": 2.6610374450683593e-05,
      "model_forward_time": 1.732926845550537,
      "step": 27903
    },
    {
      "epoch": 2.6610374450683593e-05,
      "step": 27903,
      "training_step_time": 6.890429735183716
    },
    {
      "epoch": 2.6611328125e-05,
      "model_forward_time": 1.3614580631256104,
      "step": 27904
    },
    {
      "epoch": 2.6611328125e-05,
      "step": 27904,
      "training_step_time": 7.6413893699646
    },
    {
      "epoch": 2.6612281799316407e-05,
      "model_forward_time": 1.7263731956481934,
      "step": 27905
    },
    {
      "epoch": 2.6612281799316407e-05,
      "step": 27905,
      "training_step_time": 7.5338006019592285
    },
    {
      "epoch": 2.661323547363281e-05,
      "model_forward_time": 1.7225518226623535,
      "step": 27906
    },
    {
      "epoch": 2.661323547363281e-05,
      "step": 27906,
      "training_step_time": 7.611726999282837
    },
    {
      "epoch": 2.661418914794922e-05,
      "model_forward_time": 1.7124111652374268,
      "step": 27907
    },
    {
      "epoch": 2.661418914794922e-05,
      "step": 27907,
      "training_step_time": 7.669852018356323
    },
    {
      "epoch": 2.6615142822265626e-05,
      "model_forward_time": 2.304598808288574,
      "step": 27908
    },
    {
      "epoch": 2.6615142822265626e-05,
      "step": 27908,
      "training_step_time": 7.809313535690308
    },
    {
      "epoch": 2.661609649658203e-05,
      "model_forward_time": 1.7220499515533447,
      "step": 27909
    },
    {
      "epoch": 2.661609649658203e-05,
      "step": 27909,
      "training_step_time": 9.524247646331787
    },
    {
      "epoch": 2.6617050170898438e-05,
      "grad_norm": 0.12606166303157806,
      "learning_rate": 8.632482015373934e-06,
      "loss": 0.061,
      "step": 27910
    },
    {
      "epoch": 2.6617050170898438e-05,
      "model_forward_time": 1.7268083095550537,
      "step": 27910
    },
    {
      "dynamics_loss_avg": 0.06630445625633001,
      "epoch": 2.6617050170898438e-05,
      "step": 27910
    },
    {
      "action_loss_avg": 0.004336229583714157,
      "epoch": 2.6617050170898438e-05,
      "step": 27910
    },
    {
      "epoch": 2.6617050170898438e-05,
      "step": 27910,
      "training_step_time": 8.242409706115723
    },
    {
      "epoch": 2.6618003845214845e-05,
      "model_forward_time": 1.7259838581085205,
      "step": 27911
    },
    {
      "epoch": 2.6618003845214845e-05,
      "step": 27911,
      "training_step_time": 7.640529632568359
    },
    {
      "epoch": 2.661895751953125e-05,
      "model_forward_time": 1.7257790565490723,
      "step": 27912
    },
    {
      "epoch": 2.661895751953125e-05,
      "step": 27912,
      "training_step_time": 7.240025043487549
    },
    {
      "epoch": 2.6619911193847656e-05,
      "model_forward_time": 1.7255897521972656,
      "step": 27913
    },
    {
      "epoch": 2.6619911193847656e-05,
      "step": 27913,
      "training_step_time": 6.979203462600708
    },
    {
      "epoch": 2.6620864868164064e-05,
      "model_forward_time": 1.727489709854126,
      "step": 27914
    },
    {
      "epoch": 2.6620864868164064e-05,
      "step": 27914,
      "training_step_time": 7.637339353561401
    },
    {
      "epoch": 2.6621818542480468e-05,
      "model_forward_time": 1.7222373485565186,
      "step": 27915
    },
    {
      "epoch": 2.6621818542480468e-05,
      "step": 27915,
      "training_step_time": 7.542502403259277
    },
    {
      "epoch": 2.6622772216796875e-05,
      "model_forward_time": 1.737553596496582,
      "step": 27916
    },
    {
      "epoch": 2.6622772216796875e-05,
      "step": 27916,
      "training_step_time": 8.867126226425171
    },
    {
      "epoch": 2.6623725891113283e-05,
      "model_forward_time": 2.313807725906372,
      "step": 27917
    },
    {
      "epoch": 2.6623725891113283e-05,
      "step": 27917,
      "training_step_time": 8.913285255432129
    },
    {
      "epoch": 2.6624679565429687e-05,
      "model_forward_time": 1.750988245010376,
      "step": 27918
    },
    {
      "epoch": 2.6624679565429687e-05,
      "step": 27918,
      "training_step_time": 7.717604637145996
    },
    {
      "epoch": 2.6625633239746094e-05,
      "model_forward_time": 1.7260463237762451,
      "step": 27919
    },
    {
      "epoch": 2.6625633239746094e-05,
      "step": 27919,
      "training_step_time": 7.724142789840698
    },
    {
      "epoch": 2.66265869140625e-05,
      "grad_norm": 0.12270068377256393,
      "learning_rate": 8.631345601476601e-06,
      "loss": 0.0603,
      "step": 27920
    },
    {
      "epoch": 2.66265869140625e-05,
      "model_forward_time": 1.7237935066223145,
      "step": 27920
    },
    {
      "dynamics_loss_avg": 0.05915395300835371,
      "epoch": 2.66265869140625e-05,
      "step": 27920
    },
    {
      "action_loss_avg": 0.0030630120483692734,
      "epoch": 2.66265869140625e-05,
      "step": 27920
    },
    {
      "epoch": 2.66265869140625e-05,
      "step": 27920,
      "training_step_time": 7.552630424499512
    },
    {
      "epoch": 2.6627540588378906e-05,
      "model_forward_time": 1.7169556617736816,
      "step": 27921
    },
    {
      "epoch": 2.6627540588378906e-05,
      "step": 27921,
      "training_step_time": 7.223131895065308
    },
    {
      "epoch": 2.6628494262695313e-05,
      "model_forward_time": 1.7405126094818115,
      "step": 27922
    },
    {
      "epoch": 2.6628494262695313e-05,
      "step": 27922,
      "training_step_time": 6.964781284332275
    },
    {
      "epoch": 2.6629447937011717e-05,
      "model_forward_time": 1.3626329898834229,
      "step": 27923
    },
    {
      "epoch": 2.6629447937011717e-05,
      "step": 27923,
      "training_step_time": 7.535585403442383
    },
    {
      "epoch": 2.6630401611328125e-05,
      "model_forward_time": 1.7218647003173828,
      "step": 27924
    },
    {
      "epoch": 2.6630401611328125e-05,
      "step": 27924,
      "training_step_time": 9.103405475616455
    },
    {
      "epoch": 2.6631355285644532e-05,
      "model_forward_time": 1.7165470123291016,
      "step": 27925
    },
    {
      "epoch": 2.6631355285644532e-05,
      "step": 27925,
      "training_step_time": 9.180789709091187
    },
    {
      "epoch": 2.6632308959960936e-05,
      "model_forward_time": 2.319117546081543,
      "step": 27926
    },
    {
      "epoch": 2.6632308959960936e-05,
      "step": 27926,
      "training_step_time": 9.389390230178833
    },
    {
      "epoch": 2.6633262634277343e-05,
      "model_forward_time": 1.7198834419250488,
      "step": 27927
    },
    {
      "epoch": 2.6633262634277343e-05,
      "step": 27927,
      "training_step_time": 7.76946759223938
    },
    {
      "epoch": 2.663421630859375e-05,
      "model_forward_time": 1.722240924835205,
      "step": 27928
    },
    {
      "epoch": 2.663421630859375e-05,
      "step": 27928,
      "training_step_time": 7.803058385848999
    },
    {
      "epoch": 2.6635169982910155e-05,
      "model_forward_time": 1.7241384983062744,
      "step": 27929
    },
    {
      "epoch": 2.6635169982910155e-05,
      "step": 27929,
      "training_step_time": 7.605244398117065
    },
    {
      "epoch": 2.6636123657226562e-05,
      "grad_norm": 0.15179625153541565,
      "learning_rate": 8.630208790460771e-06,
      "loss": 0.0598,
      "step": 27930
    },
    {
      "epoch": 2.6636123657226562e-05,
      "model_forward_time": 1.7200300693511963,
      "step": 27930
    },
    {
      "dynamics_loss_avg": 0.05238588526844978,
      "epoch": 2.6636123657226562e-05,
      "step": 27930
    },
    {
      "action_loss_avg": 0.0012765954365022481,
      "epoch": 2.6636123657226562e-05,
      "step": 27930
    },
    {
      "epoch": 2.6636123657226562e-05,
      "step": 27930,
      "training_step_time": 7.681758642196655
    },
    {
      "epoch": 2.663707733154297e-05,
      "model_forward_time": 1.7220063209533691,
      "step": 27931
    },
    {
      "epoch": 2.663707733154297e-05,
      "step": 27931,
      "training_step_time": 7.008000612258911
    },
    {
      "epoch": 2.6638031005859374e-05,
      "model_forward_time": 1.7258336544036865,
      "step": 27932
    },
    {
      "epoch": 2.6638031005859374e-05,
      "step": 27932,
      "training_step_time": 8.564183235168457
    },
    {
      "epoch": 2.663898468017578e-05,
      "model_forward_time": 1.7213127613067627,
      "step": 27933
    },
    {
      "epoch": 2.663898468017578e-05,
      "step": 27933,
      "training_step_time": 8.680551767349243
    },
    {
      "epoch": 2.663993835449219e-05,
      "model_forward_time": 1.7260873317718506,
      "step": 27934
    },
    {
      "epoch": 2.663993835449219e-05,
      "step": 27934,
      "training_step_time": 8.54940676689148
    },
    {
      "epoch": 2.6640892028808593e-05,
      "model_forward_time": 2.271874189376831,
      "step": 27935
    },
    {
      "epoch": 2.6640892028808593e-05,
      "step": 27935,
      "training_step_time": 8.490455627441406
    },
    {
      "epoch": 2.6641845703125e-05,
      "model_forward_time": 1.7217676639556885,
      "step": 27936
    },
    {
      "epoch": 2.6641845703125e-05,
      "step": 27936,
      "training_step_time": 9.418662548065186
    },
    {
      "epoch": 2.6642799377441407e-05,
      "model_forward_time": 1.7199010848999023,
      "step": 27937
    },
    {
      "epoch": 2.6642799377441407e-05,
      "step": 27937,
      "training_step_time": 8.756170988082886
    },
    {
      "epoch": 2.664375305175781e-05,
      "model_forward_time": 1.7200915813446045,
      "step": 27938
    },
    {
      "epoch": 2.664375305175781e-05,
      "step": 27938,
      "training_step_time": 7.6347949504852295
    },
    {
      "epoch": 2.664470672607422e-05,
      "model_forward_time": 1.719181776046753,
      "step": 27939
    },
    {
      "epoch": 2.664470672607422e-05,
      "step": 27939,
      "training_step_time": 7.697050333023071
    },
    {
      "epoch": 2.6645660400390626e-05,
      "grad_norm": 0.18973411619663239,
      "learning_rate": 8.629071582450767e-06,
      "loss": 0.0593,
      "step": 27940
    },
    {
      "epoch": 2.6645660400390626e-05,
      "model_forward_time": 1.724562644958496,
      "step": 27940
    },
    {
      "dynamics_loss_avg": 0.0566972172819078,
      "epoch": 2.6645660400390626e-05,
      "step": 27940
    },
    {
      "action_loss_avg": 0.0015524400398135186,
      "epoch": 2.6645660400390626e-05,
      "step": 27940
    },
    {
      "epoch": 2.6645660400390626e-05,
      "step": 27940,
      "training_step_time": 6.980857849121094
    },
    {
      "epoch": 2.664661407470703e-05,
      "model_forward_time": 1.726917028427124,
      "step": 27941
    },
    {
      "epoch": 2.664661407470703e-05,
      "step": 27941,
      "training_step_time": 7.575557231903076
    },
    {
      "epoch": 2.6647567749023438e-05,
      "model_forward_time": 1.722174882888794,
      "step": 27942
    },
    {
      "epoch": 2.6647567749023438e-05,
      "step": 27942,
      "training_step_time": 7.565316677093506
    },
    {
      "epoch": 2.6648521423339845e-05,
      "model_forward_time": 1.7198872566223145,
      "step": 27943
    },
    {
      "epoch": 2.6648521423339845e-05,
      "step": 27943,
      "training_step_time": 7.6985743045806885
    },
    {
      "epoch": 2.664947509765625e-05,
      "model_forward_time": 1.71974778175354,
      "step": 27944
    },
    {
      "epoch": 2.664947509765625e-05,
      "step": 27944,
      "training_step_time": 9.470350503921509
    },
    {
      "epoch": 2.6650428771972657e-05,
      "model_forward_time": 1.7761292457580566,
      "step": 27945
    },
    {
      "epoch": 2.6650428771972657e-05,
      "step": 27945,
      "training_step_time": 9.526683807373047
    },
    {
      "epoch": 2.6651382446289064e-05,
      "model_forward_time": 1.722928524017334,
      "step": 27946
    },
    {
      "epoch": 2.6651382446289064e-05,
      "step": 27946,
      "training_step_time": 9.029172897338867
    },
    {
      "epoch": 2.6652336120605468e-05,
      "model_forward_time": 1.7180969715118408,
      "step": 27947
    },
    {
      "epoch": 2.6652336120605468e-05,
      "step": 27947,
      "training_step_time": 7.682296514511108
    },
    {
      "epoch": 2.6653289794921875e-05,
      "model_forward_time": 1.7232251167297363,
      "step": 27948
    },
    {
      "epoch": 2.6653289794921875e-05,
      "step": 27948,
      "training_step_time": 7.949530124664307
    },
    {
      "epoch": 2.6654243469238283e-05,
      "model_forward_time": 1.7171905040740967,
      "step": 27949
    },
    {
      "epoch": 2.6654243469238283e-05,
      "step": 27949,
      "training_step_time": 6.9528114795684814
    },
    {
      "epoch": 2.6655197143554687e-05,
      "grad_norm": 0.15922965109348297,
      "learning_rate": 8.627933977570951e-06,
      "loss": 0.0567,
      "step": 27950
    },
    {
      "epoch": 2.6655197143554687e-05,
      "model_forward_time": 1.7174203395843506,
      "step": 27950
    },
    {
      "dynamics_loss_avg": 0.05465193334966898,
      "epoch": 2.6655197143554687e-05,
      "step": 27950
    },
    {
      "action_loss_avg": 0.0033914836298208685,
      "epoch": 2.6655197143554687e-05,
      "step": 27950
    },
    {
      "epoch": 2.6655197143554687e-05,
      "step": 27950,
      "training_step_time": 7.489968776702881
    },
    {
      "epoch": 2.6656150817871094e-05,
      "model_forward_time": 1.725841999053955,
      "step": 27951
    },
    {
      "epoch": 2.6656150817871094e-05,
      "step": 27951,
      "training_step_time": 7.534485816955566
    },
    {
      "epoch": 2.66571044921875e-05,
      "model_forward_time": 1.7171440124511719,
      "step": 27952
    },
    {
      "epoch": 2.66571044921875e-05,
      "step": 27952,
      "training_step_time": 8.523534059524536
    },
    {
      "epoch": 2.6658058166503906e-05,
      "model_forward_time": 1.7180895805358887,
      "step": 27953
    },
    {
      "epoch": 2.6658058166503906e-05,
      "step": 27953,
      "training_step_time": 8.666177034378052
    },
    {
      "epoch": 2.6659011840820313e-05,
      "model_forward_time": 2.328596591949463,
      "step": 27954
    },
    {
      "epoch": 2.6659011840820313e-05,
      "step": 27954,
      "training_step_time": 8.695860624313354
    },
    {
      "epoch": 2.6659965515136717e-05,
      "model_forward_time": 1.7207057476043701,
      "step": 27955
    },
    {
      "epoch": 2.6659965515136717e-05,
      "step": 27955,
      "training_step_time": 7.6271607875823975
    },
    {
      "epoch": 2.6660919189453125e-05,
      "model_forward_time": 1.726144552230835,
      "step": 27956
    },
    {
      "epoch": 2.6660919189453125e-05,
      "step": 27956,
      "training_step_time": 7.730335235595703
    },
    {
      "epoch": 2.6661872863769532e-05,
      "model_forward_time": 1.7226364612579346,
      "step": 27957
    },
    {
      "epoch": 2.6661872863769532e-05,
      "step": 27957,
      "training_step_time": 7.107901096343994
    },
    {
      "epoch": 2.6662826538085936e-05,
      "model_forward_time": 1.7185354232788086,
      "step": 27958
    },
    {
      "epoch": 2.6662826538085936e-05,
      "step": 27958,
      "training_step_time": 7.739101886749268
    },
    {
      "epoch": 2.6663780212402343e-05,
      "model_forward_time": 1.7179906368255615,
      "step": 27959
    },
    {
      "epoch": 2.6663780212402343e-05,
      "step": 27959,
      "training_step_time": 8.06771993637085
    },
    {
      "epoch": 2.666473388671875e-05,
      "grad_norm": 0.14547264575958252,
      "learning_rate": 8.626795975945729e-06,
      "loss": 0.0607,
      "step": 27960
    },
    {
      "epoch": 2.666473388671875e-05,
      "model_forward_time": 1.7150602340698242,
      "step": 27960
    },
    {
      "dynamics_loss_avg": 0.05940716490149498,
      "epoch": 2.666473388671875e-05,
      "step": 27960
    },
    {
      "action_loss_avg": 0.002270543095073663,
      "epoch": 2.666473388671875e-05,
      "step": 27960
    },
    {
      "epoch": 2.666473388671875e-05,
      "step": 27960,
      "training_step_time": 9.18263053894043
    },
    {
      "epoch": 2.6665687561035155e-05,
      "model_forward_time": 1.350992202758789,
      "step": 27961
    },
    {
      "epoch": 2.6665687561035155e-05,
      "step": 27961,
      "training_step_time": 9.150677919387817
    },
    {
      "epoch": 2.6666641235351562e-05,
      "model_forward_time": 1.7131316661834717,
      "step": 27962
    },
    {
      "epoch": 2.6666641235351562e-05,
      "step": 27962,
      "training_step_time": 8.380784034729004
    },
    {
      "epoch": 2.666759490966797e-05,
      "model_forward_time": 2.331432342529297,
      "step": 27963
    },
    {
      "epoch": 2.666759490966797e-05,
      "step": 27963,
      "training_step_time": 8.349151372909546
    },
    {
      "epoch": 2.6668548583984374e-05,
      "model_forward_time": 1.7167317867279053,
      "step": 27964
    },
    {
      "epoch": 2.6668548583984374e-05,
      "step": 27964,
      "training_step_time": 7.62007212638855
    },
    {
      "epoch": 2.666950225830078e-05,
      "model_forward_time": 1.7187631130218506,
      "step": 27965
    },
    {
      "epoch": 2.666950225830078e-05,
      "step": 27965,
      "training_step_time": 9.426753044128418
    },
    {
      "epoch": 2.667045593261719e-05,
      "model_forward_time": 1.7159838676452637,
      "step": 27966
    },
    {
      "epoch": 2.667045593261719e-05,
      "step": 27966,
      "training_step_time": 7.54842734336853
    },
    {
      "epoch": 2.6671409606933593e-05,
      "model_forward_time": 1.7085390090942383,
      "step": 27967
    },
    {
      "epoch": 2.6671409606933593e-05,
      "step": 27967,
      "training_step_time": 7.309374570846558
    },
    {
      "epoch": 2.667236328125e-05,
      "model_forward_time": 1.715014934539795,
      "step": 27968
    },
    {
      "epoch": 2.667236328125e-05,
      "step": 27968,
      "training_step_time": 7.58073616027832
    },
    {
      "epoch": 2.6673316955566407e-05,
      "model_forward_time": 0.9988994598388672,
      "step": 27969
    },
    {
      "epoch": 2.6673316955566407e-05,
      "step": 27969,
      "training_step_time": 7.599383354187012
    },
    {
      "epoch": 2.667427062988281e-05,
      "grad_norm": 0.13943655788898468,
      "learning_rate": 8.625657577699552e-06,
      "loss": 0.0584,
      "step": 27970
    },
    {
      "epoch": 2.667427062988281e-05,
      "model_forward_time": 1.7050139904022217,
      "step": 27970
    },
    {
      "dynamics_loss_avg": 0.0659952286630869,
      "epoch": 2.667427062988281e-05,
      "step": 27970
    },
    {
      "action_loss_avg": 0.0032583283784333615,
      "epoch": 2.667427062988281e-05,
      "step": 27970
    },
    {
      "epoch": 2.667427062988281e-05,
      "step": 27970,
      "training_step_time": 7.614614963531494
    },
    {
      "epoch": 2.667522430419922e-05,
      "model_forward_time": 1.7089946269989014,
      "step": 27971
    },
    {
      "epoch": 2.667522430419922e-05,
      "step": 27971,
      "training_step_time": 9.231200218200684
    },
    {
      "epoch": 2.6676177978515626e-05,
      "model_forward_time": 2.3209750652313232,
      "step": 27972
    },
    {
      "epoch": 2.6676177978515626e-05,
      "step": 27972,
      "training_step_time": 8.271925687789917
    },
    {
      "epoch": 2.667713165283203e-05,
      "model_forward_time": 1.7097158432006836,
      "step": 27973
    },
    {
      "epoch": 2.667713165283203e-05,
      "step": 27973,
      "training_step_time": 7.659418821334839
    },
    {
      "epoch": 2.6678085327148438e-05,
      "model_forward_time": 1.7053155899047852,
      "step": 27974
    },
    {
      "epoch": 2.6678085327148438e-05,
      "step": 27974,
      "training_step_time": 8.550367832183838
    },
    {
      "epoch": 2.6679039001464845e-05,
      "model_forward_time": 1.7044572830200195,
      "step": 27975
    },
    {
      "epoch": 2.6679039001464845e-05,
      "step": 27975,
      "training_step_time": 7.700849294662476
    },
    {
      "epoch": 2.667999267578125e-05,
      "model_forward_time": 1.7077901363372803,
      "step": 27976
    },
    {
      "epoch": 2.667999267578125e-05,
      "step": 27976,
      "training_step_time": 7.265049457550049
    },
    {
      "epoch": 2.6680946350097657e-05,
      "model_forward_time": 1.705416202545166,
      "step": 27977
    },
    {
      "epoch": 2.6680946350097657e-05,
      "step": 27977,
      "training_step_time": 7.614095211029053
    },
    {
      "epoch": 2.6681900024414064e-05,
      "model_forward_time": 1.7065331935882568,
      "step": 27978
    },
    {
      "epoch": 2.6681900024414064e-05,
      "step": 27978,
      "training_step_time": 7.60242772102356
    },
    {
      "epoch": 2.6682853698730468e-05,
      "model_forward_time": 1.7105352878570557,
      "step": 27979
    },
    {
      "epoch": 2.6682853698730468e-05,
      "step": 27979,
      "training_step_time": 8.301462650299072
    },
    {
      "epoch": 2.6683807373046875e-05,
      "grad_norm": 0.1455271691083908,
      "learning_rate": 8.624518782956915e-06,
      "loss": 0.0587,
      "step": 27980
    },
    {
      "epoch": 2.6683807373046875e-05,
      "model_forward_time": 1.7013659477233887,
      "step": 27980
    },
    {
      "dynamics_loss_avg": 0.05596879068762064,
      "epoch": 2.6683807373046875e-05,
      "step": 27980
    },
    {
      "action_loss_avg": 0.0026308898348361255,
      "epoch": 2.6683807373046875e-05,
      "step": 27980
    },
    {
      "epoch": 2.6683807373046875e-05,
      "step": 27980,
      "training_step_time": 8.701896667480469
    },
    {
      "epoch": 2.6684761047363283e-05,
      "model_forward_time": 2.2840945720672607,
      "step": 27981
    },
    {
      "epoch": 2.6684761047363283e-05,
      "step": 27981,
      "training_step_time": 10.845458269119263
    },
    {
      "epoch": 2.6685714721679687e-05,
      "model_forward_time": 1.7043044567108154,
      "step": 27982
    },
    {
      "epoch": 2.6685714721679687e-05,
      "step": 27982,
      "training_step_time": 7.8102171421051025
    },
    {
      "epoch": 2.6686668395996094e-05,
      "model_forward_time": 1.708921194076538,
      "step": 27983
    },
    {
      "epoch": 2.6686668395996094e-05,
      "step": 27983,
      "training_step_time": 7.650688648223877
    },
    {
      "epoch": 2.66876220703125e-05,
      "model_forward_time": 1.709456205368042,
      "step": 27984
    },
    {
      "epoch": 2.66876220703125e-05,
      "step": 27984,
      "training_step_time": 7.757867097854614
    },
    {
      "epoch": 2.6688575744628906e-05,
      "model_forward_time": 1.7059881687164307,
      "step": 27985
    },
    {
      "epoch": 2.6688575744628906e-05,
      "step": 27985,
      "training_step_time": 7.019810199737549
    },
    {
      "epoch": 2.6689529418945313e-05,
      "model_forward_time": 1.708770751953125,
      "step": 27986
    },
    {
      "epoch": 2.6689529418945313e-05,
      "step": 27986,
      "training_step_time": 7.601302862167358
    },
    {
      "epoch": 2.6690483093261717e-05,
      "model_forward_time": 1.7009708881378174,
      "step": 27987
    },
    {
      "epoch": 2.6690483093261717e-05,
      "step": 27987,
      "training_step_time": 7.614315748214722
    },
    {
      "epoch": 2.6691436767578125e-05,
      "model_forward_time": 1.7109498977661133,
      "step": 27988
    },
    {
      "epoch": 2.6691436767578125e-05,
      "step": 27988,
      "training_step_time": 7.6779820919036865
    },
    {
      "epoch": 2.6692390441894532e-05,
      "model_forward_time": 1.7008445262908936,
      "step": 27989
    },
    {
      "epoch": 2.6692390441894532e-05,
      "step": 27989,
      "training_step_time": 7.789514780044556
    },
    {
      "epoch": 2.6693344116210936e-05,
      "grad_norm": 0.4329024851322174,
      "learning_rate": 8.62337959184235e-06,
      "loss": 0.0622,
      "step": 27990
    },
    {
      "epoch": 2.6693344116210936e-05,
      "model_forward_time": 2.2416934967041016,
      "step": 27990
    },
    {
      "dynamics_loss_avg": 0.06023823190480471,
      "epoch": 2.6693344116210936e-05,
      "step": 27990
    },
    {
      "action_loss_avg": 0.0020030934363603594,
      "epoch": 2.6693344116210936e-05,
      "step": 27990
    },
    {
      "epoch": 2.6693344116210936e-05,
      "step": 27990,
      "training_step_time": 9.216328382492065
    },
    {
      "epoch": 2.6694297790527343e-05,
      "model_forward_time": 1.7602183818817139,
      "step": 27991
    },
    {
      "epoch": 2.6694297790527343e-05,
      "step": 27991,
      "training_step_time": 7.7089455127716064
    },
    {
      "epoch": 2.669525146484375e-05,
      "model_forward_time": 1.3455111980438232,
      "step": 27992
    },
    {
      "epoch": 2.669525146484375e-05,
      "step": 27992,
      "training_step_time": 7.641048192977905
    },
    {
      "epoch": 2.6696205139160155e-05,
      "model_forward_time": 1.703495979309082,
      "step": 27993
    },
    {
      "epoch": 2.6696205139160155e-05,
      "step": 27993,
      "training_step_time": 7.7414710521698
    },
    {
      "epoch": 2.6697158813476562e-05,
      "model_forward_time": 1.7039456367492676,
      "step": 27994
    },
    {
      "epoch": 2.6697158813476562e-05,
      "step": 27994,
      "training_step_time": 7.086383104324341
    },
    {
      "epoch": 2.669811248779297e-05,
      "model_forward_time": 1.707334041595459,
      "step": 27995
    },
    {
      "epoch": 2.669811248779297e-05,
      "step": 27995,
      "training_step_time": 7.700985908508301
    },
    {
      "epoch": 2.6699066162109374e-05,
      "model_forward_time": 1.6980373859405518,
      "step": 27996
    },
    {
      "epoch": 2.6699066162109374e-05,
      "step": 27996,
      "training_step_time": 7.580953359603882
    },
    {
      "epoch": 2.670001983642578e-05,
      "model_forward_time": 1.7239189147949219,
      "step": 27997
    },
    {
      "epoch": 2.670001983642578e-05,
      "step": 27997,
      "training_step_time": 7.670383930206299
    },
    {
      "epoch": 2.670097351074219e-05,
      "model_forward_time": 1.737734079360962,
      "step": 27998
    },
    {
      "epoch": 2.670097351074219e-05,
      "step": 27998,
      "training_step_time": 8.8208909034729
    },
    {
      "epoch": 2.6701927185058593e-05,
      "model_forward_time": 1.7128798961639404,
      "step": 27999
    },
    {
      "epoch": 2.6701927185058593e-05,
      "step": 27999,
      "training_step_time": 8.569144487380981
    },
    {
      "epoch": 2.6702880859375e-05,
      "grad_norm": 0.14210253953933716,
      "learning_rate": 8.622240004480441e-06,
      "loss": 0.0589,
      "step": 28000
    },
    {
      "epoch": 2.6702880859375e-05,
      "model_forward_time": 1.7496201992034912,
      "step": 28000
    },
    {
      "dynamics_loss_avg": 0.047004624456167224,
      "epoch": 2.6702880859375e-05,
      "step": 28000
    },
    {
      "action_loss_avg": 0.001565083078457974,
      "epoch": 2.6702880859375e-05,
      "step": 28000
    },
    {
      "epoch": 2.6702880859375e-05,
      "step": 28000,
      "training_step_time": 9.130595445632935
    },
    {
      "epoch": 2.6703834533691407e-05,
      "model_forward_time": 1.7013866901397705,
      "step": 28001
    },
    {
      "epoch": 2.6703834533691407e-05,
      "step": 28001,
      "training_step_time": 8.928495168685913
    },
    {
      "epoch": 2.670478820800781e-05,
      "model_forward_time": 1.7227084636688232,
      "step": 28002
    },
    {
      "epoch": 2.670478820800781e-05,
      "step": 28002,
      "training_step_time": 7.849090337753296
    },
    {
      "epoch": 2.670574188232422e-05,
      "model_forward_time": 1.7153573036193848,
      "step": 28003
    },
    {
      "epoch": 2.670574188232422e-05,
      "step": 28003,
      "training_step_time": 7.905299663543701
    },
    {
      "epoch": 2.6706695556640626e-05,
      "model_forward_time": 1.720249891281128,
      "step": 28004
    },
    {
      "epoch": 2.6706695556640626e-05,
      "step": 28004,
      "training_step_time": 8.98007321357727
    },
    {
      "epoch": 2.670764923095703e-05,
      "model_forward_time": 1.3581318855285645,
      "step": 28005
    },
    {
      "epoch": 2.670764923095703e-05,
      "step": 28005,
      "training_step_time": 7.744336843490601
    },
    {
      "epoch": 2.6708602905273438e-05,
      "model_forward_time": 1.7186894416809082,
      "step": 28006
    },
    {
      "epoch": 2.6708602905273438e-05,
      "step": 28006,
      "training_step_time": 8.51647138595581
    },
    {
      "epoch": 2.6709556579589845e-05,
      "model_forward_time": 1.7169241905212402,
      "step": 28007
    },
    {
      "epoch": 2.6709556579589845e-05,
      "step": 28007,
      "training_step_time": 9.963739395141602
    },
    {
      "epoch": 2.671051025390625e-05,
      "model_forward_time": 1.7167818546295166,
      "step": 28008
    },
    {
      "epoch": 2.671051025390625e-05,
      "step": 28008,
      "training_step_time": 10.396292209625244
    },
    {
      "epoch": 2.6711463928222657e-05,
      "model_forward_time": 1.7738621234893799,
      "step": 28009
    },
    {
      "epoch": 2.6711463928222657e-05,
      "step": 28009,
      "training_step_time": 8.569334506988525
    },
    {
      "epoch": 2.6712417602539064e-05,
      "grad_norm": 0.12806336581707,
      "learning_rate": 8.621100020995813e-06,
      "loss": 0.059,
      "step": 28010
    },
    {
      "epoch": 2.6712417602539064e-05,
      "model_forward_time": 1.7169091701507568,
      "step": 28010
    },
    {
      "dynamics_loss_avg": 0.0668840941041708,
      "epoch": 2.6712417602539064e-05,
      "step": 28010
    },
    {
      "action_loss_avg": 0.002477605582680553,
      "epoch": 2.6712417602539064e-05,
      "step": 28010
    },
    {
      "epoch": 2.6712417602539064e-05,
      "step": 28010,
      "training_step_time": 8.775688886642456
    },
    {
      "epoch": 2.6713371276855468e-05,
      "model_forward_time": 1.70835280418396,
      "step": 28011
    },
    {
      "epoch": 2.6713371276855468e-05,
      "step": 28011,
      "training_step_time": 7.866480350494385
    },
    {
      "epoch": 2.6714324951171875e-05,
      "model_forward_time": 1.7106552124023438,
      "step": 28012
    },
    {
      "epoch": 2.6714324951171875e-05,
      "step": 28012,
      "training_step_time": 7.936040878295898
    },
    {
      "epoch": 2.6715278625488283e-05,
      "model_forward_time": 1.7784092426300049,
      "step": 28013
    },
    {
      "epoch": 2.6715278625488283e-05,
      "step": 28013,
      "training_step_time": 8.99648118019104
    },
    {
      "epoch": 2.6716232299804687e-05,
      "model_forward_time": 1.7166996002197266,
      "step": 28014
    },
    {
      "epoch": 2.6716232299804687e-05,
      "step": 28014,
      "training_step_time": 7.698885202407837
    },
    {
      "epoch": 2.6717185974121094e-05,
      "model_forward_time": 1.7181727886199951,
      "step": 28015
    },
    {
      "epoch": 2.6717185974121094e-05,
      "step": 28015,
      "training_step_time": 7.805060386657715
    },
    {
      "epoch": 2.67181396484375e-05,
      "model_forward_time": 1.716165542602539,
      "step": 28016
    },
    {
      "epoch": 2.67181396484375e-05,
      "step": 28016,
      "training_step_time": 9.937953472137451
    },
    {
      "epoch": 2.6719093322753906e-05,
      "model_forward_time": 1.7213149070739746,
      "step": 28017
    },
    {
      "epoch": 2.6719093322753906e-05,
      "step": 28017,
      "training_step_time": 7.96716570854187
    },
    {
      "epoch": 2.6720046997070313e-05,
      "model_forward_time": 2.328386068344116,
      "step": 28018
    },
    {
      "epoch": 2.6720046997070313e-05,
      "step": 28018,
      "training_step_time": 8.01385760307312
    },
    {
      "epoch": 2.6721000671386717e-05,
      "model_forward_time": 1.7100114822387695,
      "step": 28019
    },
    {
      "epoch": 2.6721000671386717e-05,
      "step": 28019,
      "training_step_time": 7.981307506561279
    },
    {
      "epoch": 2.6721954345703125e-05,
      "grad_norm": 0.17841023206710815,
      "learning_rate": 8.619959641513132e-06,
      "loss": 0.0635,
      "step": 28020
    },
    {
      "epoch": 2.6721954345703125e-05,
      "model_forward_time": 1.7178232669830322,
      "step": 28020
    },
    {
      "dynamics_loss_avg": 0.06314936317503453,
      "epoch": 2.6721954345703125e-05,
      "step": 28020
    },
    {
      "action_loss_avg": 0.0023966907523572446,
      "epoch": 2.6721954345703125e-05,
      "step": 28020
    },
    {
      "epoch": 2.6721954345703125e-05,
      "step": 28020,
      "training_step_time": 7.923085689544678
    },
    {
      "epoch": 2.6722908020019532e-05,
      "model_forward_time": 1.7112932205200195,
      "step": 28021
    },
    {
      "epoch": 2.6722908020019532e-05,
      "step": 28021,
      "training_step_time": 7.95209264755249
    },
    {
      "epoch": 2.6723861694335936e-05,
      "model_forward_time": 1.7198774814605713,
      "step": 28022
    },
    {
      "epoch": 2.6723861694335936e-05,
      "step": 28022,
      "training_step_time": 7.002179384231567
    },
    {
      "epoch": 2.6724815368652343e-05,
      "model_forward_time": 1.7192802429199219,
      "step": 28023
    },
    {
      "epoch": 2.6724815368652343e-05,
      "step": 28023,
      "training_step_time": 7.97471022605896
    },
    {
      "epoch": 2.672576904296875e-05,
      "model_forward_time": 1.7111566066741943,
      "step": 28024
    },
    {
      "epoch": 2.672576904296875e-05,
      "step": 28024,
      "training_step_time": 7.780811071395874
    },
    {
      "epoch": 2.6726722717285155e-05,
      "model_forward_time": 1.7176449298858643,
      "step": 28025
    },
    {
      "epoch": 2.6726722717285155e-05,
      "step": 28025,
      "training_step_time": 8.919251203536987
    },
    {
      "epoch": 2.6727676391601562e-05,
      "model_forward_time": 1.7228343486785889,
      "step": 28026
    },
    {
      "epoch": 2.6727676391601562e-05,
      "step": 28026,
      "training_step_time": 8.86747670173645
    },
    {
      "epoch": 2.672863006591797e-05,
      "model_forward_time": 2.352370023727417,
      "step": 28027
    },
    {
      "epoch": 2.672863006591797e-05,
      "step": 28027,
      "training_step_time": 8.878438234329224
    },
    {
      "epoch": 2.6729583740234374e-05,
      "model_forward_time": 1.7192251682281494,
      "step": 28028
    },
    {
      "epoch": 2.6729583740234374e-05,
      "step": 28028,
      "training_step_time": 8.899391174316406
    },
    {
      "epoch": 2.673053741455078e-05,
      "model_forward_time": 1.7140028476715088,
      "step": 28029
    },
    {
      "epoch": 2.673053741455078e-05,
      "step": 28029,
      "training_step_time": 8.953539609909058
    },
    {
      "epoch": 2.673149108886719e-05,
      "grad_norm": 0.23080405592918396,
      "learning_rate": 8.618818866157105e-06,
      "loss": 0.0605,
      "step": 28030
    },
    {
      "epoch": 2.673149108886719e-05,
      "model_forward_time": 1.7144570350646973,
      "step": 28030
    },
    {
      "dynamics_loss_avg": 0.04557662568986416,
      "epoch": 2.673149108886719e-05,
      "step": 28030
    },
    {
      "action_loss_avg": 0.0015234792488627135,
      "epoch": 2.673149108886719e-05,
      "step": 28030
    },
    {
      "epoch": 2.673149108886719e-05,
      "step": 28030,
      "training_step_time": 7.921138048171997
    },
    {
      "epoch": 2.6732444763183593e-05,
      "model_forward_time": 1.7136449813842773,
      "step": 28031
    },
    {
      "epoch": 2.6732444763183593e-05,
      "step": 28031,
      "training_step_time": 7.046110391616821
    },
    {
      "epoch": 2.67333984375e-05,
      "model_forward_time": 1.7180736064910889,
      "step": 28032
    },
    {
      "epoch": 2.67333984375e-05,
      "step": 28032,
      "training_step_time": 7.9715821743011475
    },
    {
      "epoch": 2.6734352111816407e-05,
      "model_forward_time": 1.7170372009277344,
      "step": 28033
    },
    {
      "epoch": 2.6734352111816407e-05,
      "step": 28033,
      "training_step_time": 7.839864015579224
    },
    {
      "epoch": 2.673530578613281e-05,
      "model_forward_time": 1.716933250427246,
      "step": 28034
    },
    {
      "epoch": 2.673530578613281e-05,
      "step": 28034,
      "training_step_time": 7.861549377441406
    },
    {
      "epoch": 2.673625946044922e-05,
      "model_forward_time": 1.7189648151397705,
      "step": 28035
    },
    {
      "epoch": 2.673625946044922e-05,
      "step": 28035,
      "training_step_time": 7.945261716842651
    },
    {
      "epoch": 2.6737213134765626e-05,
      "model_forward_time": 2.3449912071228027,
      "step": 28036
    },
    {
      "epoch": 2.6737213134765626e-05,
      "step": 28036,
      "training_step_time": 8.755437850952148
    },
    {
      "epoch": 2.673816680908203e-05,
      "model_forward_time": 1.7151434421539307,
      "step": 28037
    },
    {
      "epoch": 2.673816680908203e-05,
      "step": 28037,
      "training_step_time": 8.829248189926147
    },
    {
      "epoch": 2.6739120483398438e-05,
      "model_forward_time": 1.7218892574310303,
      "step": 28038
    },
    {
      "epoch": 2.6739120483398438e-05,
      "step": 28038,
      "training_step_time": 7.95316743850708
    },
    {
      "epoch": 2.6740074157714845e-05,
      "model_forward_time": 1.7256155014038086,
      "step": 28039
    },
    {
      "epoch": 2.6740074157714845e-05,
      "step": 28039,
      "training_step_time": 8.774537086486816
    },
    {
      "epoch": 2.674102783203125e-05,
      "grad_norm": 0.2034156769514084,
      "learning_rate": 8.617677695052488e-06,
      "loss": 0.0629,
      "step": 28040
    },
    {
      "epoch": 2.674102783203125e-05,
      "model_forward_time": 1.7221956253051758,
      "step": 28040
    },
    {
      "dynamics_loss_avg": 0.07233957182615995,
      "epoch": 2.674102783203125e-05,
      "step": 28040
    },
    {
      "action_loss_avg": 0.0024078629998257385,
      "epoch": 2.674102783203125e-05,
      "step": 28040
    },
    {
      "epoch": 2.674102783203125e-05,
      "step": 28040,
      "training_step_time": 7.775526285171509
    },
    {
      "epoch": 2.6741981506347657e-05,
      "model_forward_time": 1.721848726272583,
      "step": 28041
    },
    {
      "epoch": 2.6741981506347657e-05,
      "step": 28041,
      "training_step_time": 7.974883556365967
    },
    {
      "epoch": 2.6742935180664064e-05,
      "model_forward_time": 1.7317054271697998,
      "step": 28042
    },
    {
      "epoch": 2.6742935180664064e-05,
      "step": 28042,
      "training_step_time": 7.782544374465942
    },
    {
      "epoch": 2.6743888854980468e-05,
      "model_forward_time": 1.718782901763916,
      "step": 28043
    },
    {
      "epoch": 2.6743888854980468e-05,
      "step": 28043,
      "training_step_time": 7.840355634689331
    },
    {
      "epoch": 2.6744842529296875e-05,
      "model_forward_time": 1.7238247394561768,
      "step": 28044
    },
    {
      "epoch": 2.6744842529296875e-05,
      "step": 28044,
      "training_step_time": 7.908869981765747
    },
    {
      "epoch": 2.6745796203613283e-05,
      "model_forward_time": 2.2974202632904053,
      "step": 28045
    },
    {
      "epoch": 2.6745796203613283e-05,
      "step": 28045,
      "training_step_time": 9.317552328109741
    },
    {
      "epoch": 2.6746749877929687e-05,
      "model_forward_time": 1.776350736618042,
      "step": 28046
    },
    {
      "epoch": 2.6746749877929687e-05,
      "step": 28046,
      "training_step_time": 9.114473342895508
    },
    {
      "epoch": 2.6747703552246094e-05,
      "model_forward_time": 1.7186453342437744,
      "step": 28047
    },
    {
      "epoch": 2.6747703552246094e-05,
      "step": 28047,
      "training_step_time": 7.8891260623931885
    },
    {
      "epoch": 2.67486572265625e-05,
      "model_forward_time": 1.717668056488037,
      "step": 28048
    },
    {
      "epoch": 2.67486572265625e-05,
      "step": 28048,
      "training_step_time": 8.410884380340576
    },
    {
      "epoch": 2.6749610900878906e-05,
      "model_forward_time": 1.7221918106079102,
      "step": 28049
    },
    {
      "epoch": 2.6749610900878906e-05,
      "step": 28049,
      "training_step_time": 7.562695264816284
    },
    {
      "epoch": 2.6750564575195313e-05,
      "grad_norm": 0.21212299168109894,
      "learning_rate": 8.616536128324078e-06,
      "loss": 0.0638,
      "step": 28050
    },
    {
      "epoch": 2.6750564575195313e-05,
      "model_forward_time": 1.7242379188537598,
      "step": 28050
    },
    {
      "dynamics_loss_avg": 0.06416304707527161,
      "epoch": 2.6750564575195313e-05,
      "step": 28050
    },
    {
      "action_loss_avg": 0.0021414496120996775,
      "epoch": 2.6750564575195313e-05,
      "step": 28050
    },
    {
      "epoch": 2.6750564575195313e-05,
      "step": 28050,
      "training_step_time": 7.888325214385986
    },
    {
      "epoch": 2.6751518249511717e-05,
      "model_forward_time": 1.721325397491455,
      "step": 28051
    },
    {
      "epoch": 2.6751518249511717e-05,
      "step": 28051,
      "training_step_time": 7.807392597198486
    },
    {
      "epoch": 2.6752471923828125e-05,
      "model_forward_time": 1.7299294471740723,
      "step": 28052
    },
    {
      "epoch": 2.6752471923828125e-05,
      "step": 28052,
      "training_step_time": 7.841579914093018
    },
    {
      "epoch": 2.6753425598144532e-05,
      "model_forward_time": 1.7250807285308838,
      "step": 28053
    },
    {
      "epoch": 2.6753425598144532e-05,
      "step": 28053,
      "training_step_time": 7.888093948364258
    },
    {
      "epoch": 2.6754379272460936e-05,
      "model_forward_time": 1.7185163497924805,
      "step": 28054
    },
    {
      "epoch": 2.6754379272460936e-05,
      "step": 28054,
      "training_step_time": 8.450851202011108
    },
    {
      "epoch": 2.6755332946777343e-05,
      "model_forward_time": 1.7863943576812744,
      "step": 28055
    },
    {
      "epoch": 2.6755332946777343e-05,
      "step": 28055,
      "training_step_time": 9.049249172210693
    },
    {
      "epoch": 2.675628662109375e-05,
      "model_forward_time": 1.7258274555206299,
      "step": 28056
    },
    {
      "epoch": 2.675628662109375e-05,
      "step": 28056,
      "training_step_time": 7.956156969070435
    },
    {
      "epoch": 2.6757240295410155e-05,
      "model_forward_time": 1.7242138385772705,
      "step": 28057
    },
    {
      "epoch": 2.6757240295410155e-05,
      "step": 28057,
      "training_step_time": 7.879328966140747
    },
    {
      "epoch": 2.6758193969726562e-05,
      "model_forward_time": 1.7176024913787842,
      "step": 28058
    },
    {
      "epoch": 2.6758193969726562e-05,
      "step": 28058,
      "training_step_time": 7.817941427230835
    },
    {
      "epoch": 2.675914764404297e-05,
      "model_forward_time": 1.7415080070495605,
      "step": 28059
    },
    {
      "epoch": 2.675914764404297e-05,
      "step": 28059,
      "training_step_time": 7.881863832473755
    },
    {
      "epoch": 2.6760101318359374e-05,
      "grad_norm": 0.2134634256362915,
      "learning_rate": 8.615394166096712e-06,
      "loss": 0.0632,
      "step": 28060
    },
    {
      "epoch": 2.6760101318359374e-05,
      "model_forward_time": 1.726611614227295,
      "step": 28060
    },
    {
      "dynamics_loss_avg": 0.049406135082244874,
      "epoch": 2.6760101318359374e-05,
      "step": 28060
    },
    {
      "action_loss_avg": 0.0024241280916612595,
      "epoch": 2.6760101318359374e-05,
      "step": 28060
    },
    {
      "epoch": 2.6760101318359374e-05,
      "step": 28060,
      "training_step_time": 7.89154577255249
    },
    {
      "epoch": 2.676105499267578e-05,
      "model_forward_time": 1.7109439373016357,
      "step": 28061
    },
    {
      "epoch": 2.676105499267578e-05,
      "step": 28061,
      "training_step_time": 7.856269359588623
    },
    {
      "epoch": 2.676200866699219e-05,
      "model_forward_time": 1.7047886848449707,
      "step": 28062
    },
    {
      "epoch": 2.676200866699219e-05,
      "step": 28062,
      "training_step_time": 8.869150638580322
    },
    {
      "epoch": 2.6762962341308593e-05,
      "model_forward_time": 1.7068114280700684,
      "step": 28063
    },
    {
      "epoch": 2.6762962341308593e-05,
      "step": 28063,
      "training_step_time": 9.920149564743042
    },
    {
      "epoch": 2.6763916015625e-05,
      "model_forward_time": 1.5801894664764404,
      "step": 28064
    },
    {
      "epoch": 2.6763916015625e-05,
      "step": 28064,
      "training_step_time": 8.592439651489258
    },
    {
      "epoch": 2.6764869689941407e-05,
      "model_forward_time": 1.7276835441589355,
      "step": 28065
    },
    {
      "epoch": 2.6764869689941407e-05,
      "step": 28065,
      "training_step_time": 7.9295430183410645
    },
    {
      "epoch": 2.676582336425781e-05,
      "model_forward_time": 1.7302837371826172,
      "step": 28066
    },
    {
      "epoch": 2.676582336425781e-05,
      "step": 28066,
      "training_step_time": 7.901321649551392
    },
    {
      "epoch": 2.676677703857422e-05,
      "model_forward_time": 1.7268996238708496,
      "step": 28067
    },
    {
      "epoch": 2.676677703857422e-05,
      "step": 28067,
      "training_step_time": 7.87026309967041
    },
    {
      "epoch": 2.6767730712890626e-05,
      "model_forward_time": 1.7220492362976074,
      "step": 28068
    },
    {
      "epoch": 2.6767730712890626e-05,
      "step": 28068,
      "training_step_time": 7.5092689990997314
    },
    {
      "epoch": 2.676868438720703e-05,
      "model_forward_time": 1.7415332794189453,
      "step": 28069
    },
    {
      "epoch": 2.676868438720703e-05,
      "step": 28069,
      "training_step_time": 8.020797491073608
    },
    {
      "epoch": 2.6769638061523438e-05,
      "grad_norm": 0.14229415357112885,
      "learning_rate": 8.614251808495278e-06,
      "loss": 0.0638,
      "step": 28070
    },
    {
      "epoch": 2.6769638061523438e-05,
      "model_forward_time": 1.7216434478759766,
      "step": 28070
    },
    {
      "dynamics_loss_avg": 0.04541862141340971,
      "epoch": 2.6769638061523438e-05,
      "step": 28070
    },
    {
      "action_loss_avg": 0.002306513348594308,
      "epoch": 2.6769638061523438e-05,
      "step": 28070
    },
    {
      "epoch": 2.6769638061523438e-05,
      "step": 28070,
      "training_step_time": 7.819372653961182
    },
    {
      "epoch": 2.6770591735839845e-05,
      "model_forward_time": 1.7187120914459229,
      "step": 28071
    },
    {
      "epoch": 2.6770591735839845e-05,
      "step": 28071,
      "training_step_time": 8.007652521133423
    },
    {
      "epoch": 2.677154541015625e-05,
      "model_forward_time": 1.7278382778167725,
      "step": 28072
    },
    {
      "epoch": 2.677154541015625e-05,
      "step": 28072,
      "training_step_time": 7.932119131088257
    },
    {
      "epoch": 2.6772499084472657e-05,
      "model_forward_time": 2.345743417739868,
      "step": 28073
    },
    {
      "epoch": 2.6772499084472657e-05,
      "step": 28073,
      "training_step_time": 9.678367853164673
    },
    {
      "epoch": 2.6773452758789064e-05,
      "model_forward_time": 1.726958990097046,
      "step": 28074
    },
    {
      "epoch": 2.6773452758789064e-05,
      "step": 28074,
      "training_step_time": 8.754783391952515
    },
    {
      "epoch": 2.6774406433105468e-05,
      "model_forward_time": 1.7275011539459229,
      "step": 28075
    },
    {
      "epoch": 2.6774406433105468e-05,
      "step": 28075,
      "training_step_time": 7.78225564956665
    },
    {
      "epoch": 2.6775360107421875e-05,
      "model_forward_time": 1.716367244720459,
      "step": 28076
    },
    {
      "epoch": 2.6775360107421875e-05,
      "step": 28076,
      "training_step_time": 7.842576026916504
    },
    {
      "epoch": 2.6776313781738283e-05,
      "model_forward_time": 1.7142796516418457,
      "step": 28077
    },
    {
      "epoch": 2.6776313781738283e-05,
      "step": 28077,
      "training_step_time": 7.85227108001709
    },
    {
      "epoch": 2.6777267456054687e-05,
      "model_forward_time": 1.7274909019470215,
      "step": 28078
    },
    {
      "epoch": 2.6777267456054687e-05,
      "step": 28078,
      "training_step_time": 7.876155614852905
    },
    {
      "epoch": 2.6778221130371094e-05,
      "model_forward_time": 1.7219796180725098,
      "step": 28079
    },
    {
      "epoch": 2.6778221130371094e-05,
      "step": 28079,
      "training_step_time": 7.6987457275390625
    },
    {
      "epoch": 2.67791748046875e-05,
      "grad_norm": 0.11683499813079834,
      "learning_rate": 8.6131090556447e-06,
      "loss": 0.0634,
      "step": 28080
    },
    {
      "epoch": 2.67791748046875e-05,
      "model_forward_time": 1.726398229598999,
      "step": 28080
    },
    {
      "dynamics_loss_avg": 0.06320112836547195,
      "epoch": 2.67791748046875e-05,
      "step": 28080
    },
    {
      "action_loss_avg": 0.002373782107315492,
      "epoch": 2.67791748046875e-05,
      "step": 28080
    },
    {
      "epoch": 2.67791748046875e-05,
      "step": 28080,
      "training_step_time": 8.30139684677124
    },
    {
      "epoch": 2.6780128479003906e-05,
      "model_forward_time": 1.730231761932373,
      "step": 28081
    },
    {
      "epoch": 2.6780128479003906e-05,
      "step": 28081,
      "training_step_time": 7.866819143295288
    },
    {
      "epoch": 2.6781082153320313e-05,
      "model_forward_time": 2.276277780532837,
      "step": 28082
    },
    {
      "epoch": 2.6781082153320313e-05,
      "step": 28082,
      "training_step_time": 8.908254384994507
    },
    {
      "epoch": 2.6782035827636717e-05,
      "model_forward_time": 1.7874763011932373,
      "step": 28083
    },
    {
      "epoch": 2.6782035827636717e-05,
      "step": 28083,
      "training_step_time": 9.554339170455933
    },
    {
      "epoch": 2.6782989501953125e-05,
      "model_forward_time": 1.7072017192840576,
      "step": 28084
    },
    {
      "epoch": 2.6782989501953125e-05,
      "step": 28084,
      "training_step_time": 8.49275279045105
    },
    {
      "epoch": 2.6783943176269532e-05,
      "model_forward_time": 1.7114155292510986,
      "step": 28085
    },
    {
      "epoch": 2.6783943176269532e-05,
      "step": 28085,
      "training_step_time": 8.308695316314697
    },
    {
      "epoch": 2.6784896850585936e-05,
      "model_forward_time": 1.713029384613037,
      "step": 28086
    },
    {
      "epoch": 2.6784896850585936e-05,
      "step": 28086,
      "training_step_time": 7.352158546447754
    },
    {
      "epoch": 2.6785850524902344e-05,
      "model_forward_time": 1.7090826034545898,
      "step": 28087
    },
    {
      "epoch": 2.6785850524902344e-05,
      "step": 28087,
      "training_step_time": 7.950752019882202
    },
    {
      "epoch": 2.678680419921875e-05,
      "model_forward_time": 1.7108864784240723,
      "step": 28088
    },
    {
      "epoch": 2.678680419921875e-05,
      "step": 28088,
      "training_step_time": 7.771599531173706
    },
    {
      "epoch": 2.6787757873535155e-05,
      "model_forward_time": 1.7142901420593262,
      "step": 28089
    },
    {
      "epoch": 2.6787757873535155e-05,
      "step": 28089,
      "training_step_time": 7.780415773391724
    },
    {
      "epoch": 2.6788711547851562e-05,
      "grad_norm": 0.10142569988965988,
      "learning_rate": 8.611965907669948e-06,
      "loss": 0.0609,
      "step": 28090
    },
    {
      "epoch": 2.6788711547851562e-05,
      "model_forward_time": 1.7161273956298828,
      "step": 28090
    },
    {
      "dynamics_loss_avg": 0.07821676284074783,
      "epoch": 2.6788711547851562e-05,
      "step": 28090
    },
    {
      "action_loss_avg": 0.0039691944839432836,
      "epoch": 2.6788711547851562e-05,
      "step": 28090
    },
    {
      "epoch": 2.6788711547851562e-05,
      "step": 28090,
      "training_step_time": 8.920538663864136
    },
    {
      "epoch": 2.678966522216797e-05,
      "model_forward_time": 1.7096717357635498,
      "step": 28091
    },
    {
      "epoch": 2.678966522216797e-05,
      "step": 28091,
      "training_step_time": 9.743182182312012
    },
    {
      "epoch": 2.6790618896484374e-05,
      "model_forward_time": 2.35221529006958,
      "step": 28092
    },
    {
      "epoch": 2.6790618896484374e-05,
      "step": 28092,
      "training_step_time": 8.894735336303711
    },
    {
      "epoch": 2.679157257080078e-05,
      "model_forward_time": 1.7148816585540771,
      "step": 28093
    },
    {
      "epoch": 2.679157257080078e-05,
      "step": 28093,
      "training_step_time": 8.724819660186768
    },
    {
      "epoch": 2.679252624511719e-05,
      "model_forward_time": 1.713444471359253,
      "step": 28094
    },
    {
      "epoch": 2.679252624511719e-05,
      "step": 28094,
      "training_step_time": 7.781759738922119
    },
    {
      "epoch": 2.6793479919433593e-05,
      "model_forward_time": 1.7159638404846191,
      "step": 28095
    },
    {
      "epoch": 2.6793479919433593e-05,
      "step": 28095,
      "training_step_time": 7.621597051620483
    },
    {
      "epoch": 2.679443359375e-05,
      "model_forward_time": 1.7261736392974854,
      "step": 28096
    },
    {
      "epoch": 2.679443359375e-05,
      "step": 28096,
      "training_step_time": 7.970871925354004
    },
    {
      "epoch": 2.6795387268066407e-05,
      "model_forward_time": 1.7142853736877441,
      "step": 28097
    },
    {
      "epoch": 2.6795387268066407e-05,
      "step": 28097,
      "training_step_time": 7.7062647342681885
    },
    {
      "epoch": 2.679634094238281e-05,
      "model_forward_time": 1.719369649887085,
      "step": 28098
    },
    {
      "epoch": 2.679634094238281e-05,
      "step": 28098,
      "training_step_time": 8.769843578338623
    },
    {
      "epoch": 2.679729461669922e-05,
      "model_forward_time": 1.7187726497650146,
      "step": 28099
    },
    {
      "epoch": 2.679729461669922e-05,
      "step": 28099,
      "training_step_time": 8.352768182754517
    },
    {
      "epoch": 2.6798248291015626e-05,
      "grad_norm": 0.124835304915905,
      "learning_rate": 8.610822364696035e-06,
      "loss": 0.0572,
      "step": 28100
    },
    {
      "epoch": 2.6798248291015626e-05,
      "model_forward_time": 1.7156527042388916,
      "step": 28100
    },
    {
      "dynamics_loss_avg": 0.04662053119391203,
      "epoch": 2.6798248291015626e-05,
      "step": 28100
    },
    {
      "action_loss_avg": 0.0022850007517263293,
      "epoch": 2.6798248291015626e-05,
      "step": 28100
    },
    {
      "epoch": 2.6798248291015626e-05,
      "step": 28100,
      "training_step_time": 7.826101541519165
    },
    {
      "epoch": 2.679920196533203e-05,
      "model_forward_time": 2.328341245651245,
      "step": 28101
    },
    {
      "epoch": 2.679920196533203e-05,
      "step": 28101,
      "training_step_time": 9.117857933044434
    },
    {
      "epoch": 2.6800155639648438e-05,
      "model_forward_time": 1.7258892059326172,
      "step": 28102
    },
    {
      "epoch": 2.6800155639648438e-05,
      "step": 28102,
      "training_step_time": 7.844332933425903
    },
    {
      "epoch": 2.6801109313964845e-05,
      "model_forward_time": 1.7091889381408691,
      "step": 28103
    },
    {
      "epoch": 2.6801109313964845e-05,
      "step": 28103,
      "training_step_time": 8.224511861801147
    },
    {
      "epoch": 2.680206298828125e-05,
      "model_forward_time": 1.7109363079071045,
      "step": 28104
    },
    {
      "epoch": 2.680206298828125e-05,
      "step": 28104,
      "training_step_time": 7.476654529571533
    },
    {
      "epoch": 2.6803016662597657e-05,
      "model_forward_time": 1.7589285373687744,
      "step": 28105
    },
    {
      "epoch": 2.6803016662597657e-05,
      "step": 28105,
      "training_step_time": 7.882424831390381
    },
    {
      "epoch": 2.6803970336914064e-05,
      "model_forward_time": 1.351569652557373,
      "step": 28106
    },
    {
      "epoch": 2.6803970336914064e-05,
      "step": 28106,
      "training_step_time": 7.749487638473511
    },
    {
      "epoch": 2.6804924011230468e-05,
      "model_forward_time": 1.7180900573730469,
      "step": 28107
    },
    {
      "epoch": 2.6804924011230468e-05,
      "step": 28107,
      "training_step_time": 7.764786958694458
    },
    {
      "epoch": 2.6805877685546876e-05,
      "model_forward_time": 1.7176575660705566,
      "step": 28108
    },
    {
      "epoch": 2.6805877685546876e-05,
      "step": 28108,
      "training_step_time": 8.332446336746216
    },
    {
      "epoch": 2.6806831359863283e-05,
      "model_forward_time": 1.7161660194396973,
      "step": 28109
    },
    {
      "epoch": 2.6806831359863283e-05,
      "step": 28109,
      "training_step_time": 8.57594108581543
    },
    {
      "epoch": 2.6807785034179687e-05,
      "grad_norm": 0.118777796626091,
      "learning_rate": 8.609678426848017e-06,
      "loss": 0.06,
      "step": 28110
    },
    {
      "epoch": 2.6807785034179687e-05,
      "model_forward_time": 2.2718594074249268,
      "step": 28110
    },
    {
      "dynamics_loss_avg": 0.06602118015289307,
      "epoch": 2.6807785034179687e-05,
      "step": 28110
    },
    {
      "action_loss_avg": 0.0026183077658060937,
      "epoch": 2.6807785034179687e-05,
      "step": 28110
    },
    {
      "epoch": 2.6807785034179687e-05,
      "step": 28110,
      "training_step_time": 8.033944129943848
    },
    {
      "epoch": 2.6808738708496094e-05,
      "model_forward_time": 1.7774326801300049,
      "step": 28111
    },
    {
      "epoch": 2.6808738708496094e-05,
      "step": 28111,
      "training_step_time": 7.897937297821045
    },
    {
      "epoch": 2.68096923828125e-05,
      "model_forward_time": 1.7177462577819824,
      "step": 28112
    },
    {
      "epoch": 2.68096923828125e-05,
      "step": 28112,
      "training_step_time": 8.623297214508057
    },
    {
      "epoch": 2.6810646057128906e-05,
      "model_forward_time": 1.7164337635040283,
      "step": 28113
    },
    {
      "epoch": 2.6810646057128906e-05,
      "step": 28113,
      "training_step_time": 7.85577654838562
    },
    {
      "epoch": 2.6811599731445313e-05,
      "model_forward_time": 1.713437557220459,
      "step": 28114
    },
    {
      "epoch": 2.6811599731445313e-05,
      "step": 28114,
      "training_step_time": 7.829553127288818
    },
    {
      "epoch": 2.6812553405761717e-05,
      "model_forward_time": 1.7206766605377197,
      "step": 28115
    },
    {
      "epoch": 2.6812553405761717e-05,
      "step": 28115,
      "training_step_time": 7.775456428527832
    },
    {
      "epoch": 2.6813507080078125e-05,
      "model_forward_time": 1.7359542846679688,
      "step": 28116
    },
    {
      "epoch": 2.6813507080078125e-05,
      "step": 28116,
      "training_step_time": 7.762388706207275
    },
    {
      "epoch": 2.6814460754394532e-05,
      "model_forward_time": 1.7224328517913818,
      "step": 28117
    },
    {
      "epoch": 2.6814460754394532e-05,
      "step": 28117,
      "training_step_time": 8.317651748657227
    },
    {
      "epoch": 2.6815414428710936e-05,
      "model_forward_time": 1.7068841457366943,
      "step": 28118
    },
    {
      "epoch": 2.6815414428710936e-05,
      "step": 28118,
      "training_step_time": 8.387025356292725
    },
    {
      "epoch": 2.6816368103027344e-05,
      "model_forward_time": 2.2568869590759277,
      "step": 28119
    },
    {
      "epoch": 2.6816368103027344e-05,
      "step": 28119,
      "training_step_time": 8.808930397033691
    },
    {
      "epoch": 2.681732177734375e-05,
      "grad_norm": 0.15531349182128906,
      "learning_rate": 8.608534094250991e-06,
      "loss": 0.058,
      "step": 28120
    },
    {
      "epoch": 2.681732177734375e-05,
      "model_forward_time": 1.7687711715698242,
      "step": 28120
    },
    {
      "dynamics_loss_avg": 0.07805030718445778,
      "epoch": 2.681732177734375e-05,
      "step": 28120
    },
    {
      "action_loss_avg": 0.0021098932600580158,
      "epoch": 2.681732177734375e-05,
      "step": 28120
    },
    {
      "epoch": 2.681732177734375e-05,
      "step": 28120,
      "training_step_time": 7.896804332733154
    },
    {
      "epoch": 2.6818275451660155e-05,
      "model_forward_time": 1.7083971500396729,
      "step": 28121
    },
    {
      "epoch": 2.6818275451660155e-05,
      "step": 28121,
      "training_step_time": 7.770306825637817
    },
    {
      "epoch": 2.6819229125976562e-05,
      "model_forward_time": 1.7033939361572266,
      "step": 28122
    },
    {
      "epoch": 2.6819229125976562e-05,
      "step": 28122,
      "training_step_time": 7.827590227127075
    },
    {
      "epoch": 2.682018280029297e-05,
      "model_forward_time": 1.7155451774597168,
      "step": 28123
    },
    {
      "epoch": 2.682018280029297e-05,
      "step": 28123,
      "training_step_time": 7.824181318283081
    },
    {
      "epoch": 2.6821136474609374e-05,
      "model_forward_time": 1.7065482139587402,
      "step": 28124
    },
    {
      "epoch": 2.6821136474609374e-05,
      "step": 28124,
      "training_step_time": 7.777074575424194
    },
    {
      "epoch": 2.682209014892578e-05,
      "model_forward_time": 1.7065870761871338,
      "step": 28125
    },
    {
      "epoch": 2.682209014892578e-05,
      "step": 28125,
      "training_step_time": 8.606221914291382
    },
    {
      "epoch": 2.682304382324219e-05,
      "model_forward_time": 1.7102947235107422,
      "step": 28126
    },
    {
      "epoch": 2.682304382324219e-05,
      "step": 28126,
      "training_step_time": 8.483858108520508
    },
    {
      "epoch": 2.6823997497558593e-05,
      "model_forward_time": 1.7090916633605957,
      "step": 28127
    },
    {
      "epoch": 2.6823997497558593e-05,
      "step": 28127,
      "training_step_time": 8.973370790481567
    },
    {
      "epoch": 2.6824951171875e-05,
      "model_forward_time": 2.2625887393951416,
      "step": 28128
    },
    {
      "epoch": 2.6824951171875e-05,
      "step": 28128,
      "training_step_time": 9.385973453521729
    },
    {
      "epoch": 2.6825904846191408e-05,
      "model_forward_time": 1.7508177757263184,
      "step": 28129
    },
    {
      "epoch": 2.6825904846191408e-05,
      "step": 28129,
      "training_step_time": 8.576499700546265
    },
    {
      "epoch": 2.682685852050781e-05,
      "grad_norm": 0.11386708915233612,
      "learning_rate": 8.607389367030103e-06,
      "loss": 0.0568,
      "step": 28130
    },
    {
      "epoch": 2.682685852050781e-05,
      "model_forward_time": 1.7066307067871094,
      "step": 28130
    },
    {
      "dynamics_loss_avg": 0.05329115055501461,
      "epoch": 2.682685852050781e-05,
      "step": 28130
    },
    {
      "action_loss_avg": 0.003248928254470229,
      "epoch": 2.682685852050781e-05,
      "step": 28130
    },
    {
      "epoch": 2.682685852050781e-05,
      "step": 28130,
      "training_step_time": 7.825827360153198
    },
    {
      "epoch": 2.682781219482422e-05,
      "model_forward_time": 1.7036030292510986,
      "step": 28131
    },
    {
      "epoch": 2.682781219482422e-05,
      "step": 28131,
      "training_step_time": 8.407137393951416
    },
    {
      "epoch": 2.6828765869140626e-05,
      "model_forward_time": 1.7083089351654053,
      "step": 28132
    },
    {
      "epoch": 2.6828765869140626e-05,
      "step": 28132,
      "training_step_time": 7.681596040725708
    },
    {
      "epoch": 2.682971954345703e-05,
      "model_forward_time": 0.9870412349700928,
      "step": 28133
    },
    {
      "epoch": 2.682971954345703e-05,
      "step": 28133,
      "training_step_time": 7.876615762710571
    },
    {
      "epoch": 2.6830673217773438e-05,
      "model_forward_time": 1.709195613861084,
      "step": 28134
    },
    {
      "epoch": 2.6830673217773438e-05,
      "step": 28134,
      "training_step_time": 7.703692436218262
    },
    {
      "epoch": 2.6831626892089845e-05,
      "model_forward_time": 1.7029201984405518,
      "step": 28135
    },
    {
      "epoch": 2.6831626892089845e-05,
      "step": 28135,
      "training_step_time": 7.779382944107056
    },
    {
      "epoch": 2.683258056640625e-05,
      "model_forward_time": 1.7043144702911377,
      "step": 28136
    },
    {
      "epoch": 2.683258056640625e-05,
      "step": 28136,
      "training_step_time": 7.84457540512085
    },
    {
      "epoch": 2.6833534240722657e-05,
      "model_forward_time": 1.7065203189849854,
      "step": 28137
    },
    {
      "epoch": 2.6833534240722657e-05,
      "step": 28137,
      "training_step_time": 7.896347999572754
    },
    {
      "epoch": 2.6834487915039064e-05,
      "model_forward_time": 2.2753684520721436,
      "step": 28138
    },
    {
      "epoch": 2.6834487915039064e-05,
      "step": 28138,
      "training_step_time": 7.991152763366699
    },
    {
      "epoch": 2.6835441589355468e-05,
      "model_forward_time": 1.7115647792816162,
      "step": 28139
    },
    {
      "epoch": 2.6835441589355468e-05,
      "step": 28139,
      "training_step_time": 7.919703245162964
    },
    {
      "epoch": 2.6836395263671876e-05,
      "grad_norm": 0.1280268132686615,
      "learning_rate": 8.60624424531054e-06,
      "loss": 0.0571,
      "step": 28140
    },
    {
      "epoch": 2.6836395263671876e-05,
      "model_forward_time": 1.004927396774292,
      "step": 28140
    },
    {
      "dynamics_loss_avg": 0.06275015529245138,
      "epoch": 2.6836395263671876e-05,
      "step": 28140
    },
    {
      "action_loss_avg": 0.0024053762579569595,
      "epoch": 2.6836395263671876e-05,
      "step": 28140
    },
    {
      "epoch": 2.6836395263671876e-05,
      "step": 28140,
      "training_step_time": 8.733818292617798
    },
    {
      "epoch": 2.683734893798828e-05,
      "model_forward_time": 1.7119274139404297,
      "step": 28141
    },
    {
      "epoch": 2.683734893798828e-05,
      "step": 28141,
      "training_step_time": 7.454300880432129
    },
    {
      "epoch": 2.6838302612304687e-05,
      "model_forward_time": 1.7034895420074463,
      "step": 28142
    },
    {
      "epoch": 2.6838302612304687e-05,
      "step": 28142,
      "training_step_time": 8.600220918655396
    },
    {
      "epoch": 2.6839256286621094e-05,
      "model_forward_time": 1.698702096939087,
      "step": 28143
    },
    {
      "epoch": 2.6839256286621094e-05,
      "step": 28143,
      "training_step_time": 7.731387376785278
    },
    {
      "epoch": 2.68402099609375e-05,
      "model_forward_time": 1.7037732601165771,
      "step": 28144
    },
    {
      "epoch": 2.68402099609375e-05,
      "step": 28144,
      "training_step_time": 8.621318578720093
    },
    {
      "epoch": 2.6841163635253906e-05,
      "model_forward_time": 1.7538259029388428,
      "step": 28145
    },
    {
      "epoch": 2.6841163635253906e-05,
      "step": 28145,
      "training_step_time": 8.420177936553955
    },
    {
      "epoch": 2.6842117309570313e-05,
      "model_forward_time": 1.6966047286987305,
      "step": 28146
    },
    {
      "epoch": 2.6842117309570313e-05,
      "step": 28146,
      "training_step_time": 7.937114715576172
    },
    {
      "epoch": 2.6843070983886717e-05,
      "model_forward_time": 2.2543649673461914,
      "step": 28147
    },
    {
      "epoch": 2.6843070983886717e-05,
      "step": 28147,
      "training_step_time": 9.213852643966675
    },
    {
      "epoch": 2.6844024658203125e-05,
      "model_forward_time": 1.7592363357543945,
      "step": 28148
    },
    {
      "epoch": 2.6844024658203125e-05,
      "step": 28148,
      "training_step_time": 7.983145713806152
    },
    {
      "epoch": 2.6844978332519532e-05,
      "model_forward_time": 1.7104413509368896,
      "step": 28149
    },
    {
      "epoch": 2.6844978332519532e-05,
      "step": 28149,
      "training_step_time": 7.907393932342529
    },
    {
      "epoch": 2.6845932006835936e-05,
      "grad_norm": 0.1612854152917862,
      "learning_rate": 8.605098729217525e-06,
      "loss": 0.061,
      "step": 28150
    },
    {
      "epoch": 2.6845932006835936e-05,
      "model_forward_time": 1.700361967086792,
      "step": 28150
    },
    {
      "dynamics_loss_avg": 0.07283200044184923,
      "epoch": 2.6845932006835936e-05,
      "step": 28150
    },
    {
      "action_loss_avg": 0.0019715457689017057,
      "epoch": 2.6845932006835936e-05,
      "step": 28150
    },
    {
      "epoch": 2.6845932006835936e-05,
      "step": 28150,
      "training_step_time": 8.377223253250122
    },
    {
      "epoch": 2.6846885681152344e-05,
      "model_forward_time": 1.7059509754180908,
      "step": 28151
    },
    {
      "epoch": 2.6846885681152344e-05,
      "step": 28151,
      "training_step_time": 7.9144861698150635
    },
    {
      "epoch": 2.684783935546875e-05,
      "model_forward_time": 1.703805923461914,
      "step": 28152
    },
    {
      "epoch": 2.684783935546875e-05,
      "step": 28152,
      "training_step_time": 7.654223442077637
    },
    {
      "epoch": 2.6848793029785155e-05,
      "model_forward_time": 1.6998887062072754,
      "step": 28153
    },
    {
      "epoch": 2.6848793029785155e-05,
      "step": 28153,
      "training_step_time": 8.509595394134521
    },
    {
      "epoch": 2.6849746704101562e-05,
      "model_forward_time": 1.7067971229553223,
      "step": 28154
    },
    {
      "epoch": 2.6849746704101562e-05,
      "step": 28154,
      "training_step_time": 7.758184909820557
    },
    {
      "epoch": 2.685070037841797e-05,
      "model_forward_time": 1.7022089958190918,
      "step": 28155
    },
    {
      "epoch": 2.685070037841797e-05,
      "step": 28155,
      "training_step_time": 8.861003398895264
    },
    {
      "epoch": 2.6851654052734374e-05,
      "model_forward_time": 2.2516579627990723,
      "step": 28156
    },
    {
      "epoch": 2.6851654052734374e-05,
      "step": 28156,
      "training_step_time": 9.490949392318726
    },
    {
      "epoch": 2.685260772705078e-05,
      "model_forward_time": 1.7624988555908203,
      "step": 28157
    },
    {
      "epoch": 2.685260772705078e-05,
      "step": 28157,
      "training_step_time": 7.880253314971924
    },
    {
      "epoch": 2.685356140136719e-05,
      "model_forward_time": 1.699450969696045,
      "step": 28158
    },
    {
      "epoch": 2.685356140136719e-05,
      "step": 28158,
      "training_step_time": 7.567533016204834
    },
    {
      "epoch": 2.6854515075683593e-05,
      "model_forward_time": 1.711106538772583,
      "step": 28159
    },
    {
      "epoch": 2.6854515075683593e-05,
      "step": 28159,
      "training_step_time": 7.828954458236694
    },
    {
      "epoch": 2.685546875e-05,
      "grad_norm": 0.0924961194396019,
      "learning_rate": 8.603952818876335e-06,
      "loss": 0.0562,
      "step": 28160
    },
    {
      "epoch": 2.685546875e-05,
      "model_forward_time": 1.703892469406128,
      "step": 28160
    },
    {
      "dynamics_loss_avg": 0.056308533437550067,
      "epoch": 2.685546875e-05,
      "step": 28160
    },
    {
      "action_loss_avg": 0.002223197423154488,
      "epoch": 2.685546875e-05,
      "step": 28160
    },
    {
      "epoch": 2.685546875e-05,
      "step": 28160,
      "training_step_time": 7.82676887512207
    },
    {
      "epoch": 2.6856422424316408e-05,
      "model_forward_time": 1.703932523727417,
      "step": 28161
    },
    {
      "epoch": 2.6856422424316408e-05,
      "step": 28161,
      "training_step_time": 9.723420143127441
    },
    {
      "epoch": 2.685737609863281e-05,
      "model_forward_time": 1.701206922531128,
      "step": 28162
    },
    {
      "epoch": 2.685737609863281e-05,
      "step": 28162,
      "training_step_time": 7.778799533843994
    },
    {
      "epoch": 2.685832977294922e-05,
      "model_forward_time": 1.7075262069702148,
      "step": 28163
    },
    {
      "epoch": 2.685832977294922e-05,
      "step": 28163,
      "training_step_time": 7.817070245742798
    },
    {
      "epoch": 2.6859283447265626e-05,
      "model_forward_time": 1.6983602046966553,
      "step": 28164
    },
    {
      "epoch": 2.6859283447265626e-05,
      "step": 28164,
      "training_step_time": 8.663498401641846
    },
    {
      "epoch": 2.686023712158203e-05,
      "model_forward_time": 1.7030308246612549,
      "step": 28165
    },
    {
      "epoch": 2.686023712158203e-05,
      "step": 28165,
      "training_step_time": 9.44114875793457
    },
    {
      "epoch": 2.6861190795898438e-05,
      "model_forward_time": 1.730475902557373,
      "step": 28166
    },
    {
      "epoch": 2.6861190795898438e-05,
      "step": 28166,
      "training_step_time": 8.410728931427002
    },
    {
      "epoch": 2.6862144470214845e-05,
      "model_forward_time": 1.6951854228973389,
      "step": 28167
    },
    {
      "epoch": 2.6862144470214845e-05,
      "step": 28167,
      "training_step_time": 7.650934934616089
    },
    {
      "epoch": 2.686309814453125e-05,
      "model_forward_time": 1.7040338516235352,
      "step": 28168
    },
    {
      "epoch": 2.686309814453125e-05,
      "step": 28168,
      "training_step_time": 7.907679080963135
    },
    {
      "epoch": 2.6864051818847657e-05,
      "model_forward_time": 1.6985950469970703,
      "step": 28169
    },
    {
      "epoch": 2.6864051818847657e-05,
      "step": 28169,
      "training_step_time": 7.8382344245910645
    },
    {
      "epoch": 2.6865005493164064e-05,
      "grad_norm": 0.30664893984794617,
      "learning_rate": 8.602806514412282e-06,
      "loss": 0.0576,
      "step": 28170
    },
    {
      "epoch": 2.6865005493164064e-05,
      "model_forward_time": 1.6971328258514404,
      "step": 28170
    },
    {
      "dynamics_loss_avg": 0.044502115435898305,
      "epoch": 2.6865005493164064e-05,
      "step": 28170
    },
    {
      "action_loss_avg": 0.00189538182457909,
      "epoch": 2.6865005493164064e-05,
      "step": 28170
    },
    {
      "epoch": 2.6865005493164064e-05,
      "step": 28170,
      "training_step_time": 7.499208211898804
    },
    {
      "epoch": 2.6865959167480468e-05,
      "model_forward_time": 1.6973848342895508,
      "step": 28171
    },
    {
      "epoch": 2.6865959167480468e-05,
      "step": 28171,
      "training_step_time": 7.802712678909302
    },
    {
      "epoch": 2.6866912841796876e-05,
      "model_forward_time": 1.69966721534729,
      "step": 28172
    },
    {
      "epoch": 2.6866912841796876e-05,
      "step": 28172,
      "training_step_time": 8.811501026153564
    },
    {
      "epoch": 2.686786651611328e-05,
      "model_forward_time": 1.7033066749572754,
      "step": 28173
    },
    {
      "epoch": 2.686786651611328e-05,
      "step": 28173,
      "training_step_time": 8.470787763595581
    },
    {
      "epoch": 2.6868820190429687e-05,
      "model_forward_time": 1.6971516609191895,
      "step": 28174
    },
    {
      "epoch": 2.6868820190429687e-05,
      "step": 28174,
      "training_step_time": 8.674056053161621
    },
    {
      "epoch": 2.6869773864746094e-05,
      "model_forward_time": 2.2795205116271973,
      "step": 28175
    },
    {
      "epoch": 2.6869773864746094e-05,
      "step": 28175,
      "training_step_time": 8.651322841644287
    },
    {
      "epoch": 2.68707275390625e-05,
      "model_forward_time": 1.6916263103485107,
      "step": 28176
    },
    {
      "epoch": 2.68707275390625e-05,
      "step": 28176,
      "training_step_time": 8.748558521270752
    },
    {
      "epoch": 2.6871681213378906e-05,
      "model_forward_time": 1.6939244270324707,
      "step": 28177
    },
    {
      "epoch": 2.6871681213378906e-05,
      "step": 28177,
      "training_step_time": 7.887512922286987
    },
    {
      "epoch": 2.6872634887695313e-05,
      "model_forward_time": 1.3301591873168945,
      "step": 28178
    },
    {
      "epoch": 2.6872634887695313e-05,
      "step": 28178,
      "training_step_time": 7.470797300338745
    },
    {
      "epoch": 2.6873588562011717e-05,
      "model_forward_time": 1.6964805126190186,
      "step": 28179
    },
    {
      "epoch": 2.6873588562011717e-05,
      "step": 28179,
      "training_step_time": 7.898759603500366
    },
    {
      "epoch": 2.6874542236328125e-05,
      "grad_norm": 0.11559145897626877,
      "learning_rate": 8.601659815950726e-06,
      "loss": 0.0607,
      "step": 28180
    },
    {
      "epoch": 2.6874542236328125e-05,
      "model_forward_time": 1.7033824920654297,
      "step": 28180
    },
    {
      "dynamics_loss_avg": 0.04884372968226671,
      "epoch": 2.6874542236328125e-05,
      "step": 28180
    },
    {
      "action_loss_avg": 0.002180008395225741,
      "epoch": 2.6874542236328125e-05,
      "step": 28180
    },
    {
      "epoch": 2.6874542236328125e-05,
      "step": 28180,
      "training_step_time": 7.9287989139556885
    },
    {
      "epoch": 2.6875495910644532e-05,
      "model_forward_time": 1.704042673110962,
      "step": 28181
    },
    {
      "epoch": 2.6875495910644532e-05,
      "step": 28181,
      "training_step_time": 8.67831015586853
    },
    {
      "epoch": 2.6876449584960936e-05,
      "model_forward_time": 1.695068359375,
      "step": 28182
    },
    {
      "epoch": 2.6876449584960936e-05,
      "step": 28182,
      "training_step_time": 8.923331022262573
    },
    {
      "epoch": 2.6877403259277344e-05,
      "model_forward_time": 1.7012338638305664,
      "step": 28183
    },
    {
      "epoch": 2.6877403259277344e-05,
      "step": 28183,
      "training_step_time": 9.106767416000366
    },
    {
      "epoch": 2.687835693359375e-05,
      "model_forward_time": 2.273275136947632,
      "step": 28184
    },
    {
      "epoch": 2.687835693359375e-05,
      "step": 28184,
      "training_step_time": 8.566628456115723
    },
    {
      "epoch": 2.6879310607910155e-05,
      "model_forward_time": 1.7580416202545166,
      "step": 28185
    },
    {
      "epoch": 2.6879310607910155e-05,
      "step": 28185,
      "training_step_time": 7.98401665687561
    },
    {
      "epoch": 2.6880264282226562e-05,
      "model_forward_time": 1.7067906856536865,
      "step": 28186
    },
    {
      "epoch": 2.6880264282226562e-05,
      "step": 28186,
      "training_step_time": 7.853945255279541
    },
    {
      "epoch": 2.688121795654297e-05,
      "model_forward_time": 1.7019548416137695,
      "step": 28187
    },
    {
      "epoch": 2.688121795654297e-05,
      "step": 28187,
      "training_step_time": 7.4206719398498535
    },
    {
      "epoch": 2.6882171630859374e-05,
      "model_forward_time": 1.7009649276733398,
      "step": 28188
    },
    {
      "epoch": 2.6882171630859374e-05,
      "step": 28188,
      "training_step_time": 8.94034719467163
    },
    {
      "epoch": 2.688312530517578e-05,
      "model_forward_time": 1.7069289684295654,
      "step": 28189
    },
    {
      "epoch": 2.688312530517578e-05,
      "step": 28189,
      "training_step_time": 7.899625778198242
    },
    {
      "epoch": 2.688407897949219e-05,
      "grad_norm": 0.17865075170993805,
      "learning_rate": 8.600512723617067e-06,
      "loss": 0.0623,
      "step": 28190
    },
    {
      "epoch": 2.688407897949219e-05,
      "model_forward_time": 1.7026281356811523,
      "step": 28190
    },
    {
      "dynamics_loss_avg": 0.07395892553031444,
      "epoch": 2.688407897949219e-05,
      "step": 28190
    },
    {
      "action_loss_avg": 0.0028647058992646635,
      "epoch": 2.688407897949219e-05,
      "step": 28190
    },
    {
      "epoch": 2.688407897949219e-05,
      "step": 28190,
      "training_step_time": 7.774817228317261
    },
    {
      "epoch": 2.6885032653808593e-05,
      "model_forward_time": 1.7062268257141113,
      "step": 28191
    },
    {
      "epoch": 2.6885032653808593e-05,
      "step": 28191,
      "training_step_time": 9.845861673355103
    },
    {
      "epoch": 2.6885986328125e-05,
      "model_forward_time": 1.799138069152832,
      "step": 28192
    },
    {
      "epoch": 2.6885986328125e-05,
      "step": 28192,
      "training_step_time": 8.721593618392944
    },
    {
      "epoch": 2.6886940002441408e-05,
      "model_forward_time": 2.829967737197876,
      "step": 28193
    },
    {
      "epoch": 2.6886940002441408e-05,
      "step": 28193,
      "training_step_time": 10.251577377319336
    },
    {
      "epoch": 2.688789367675781e-05,
      "model_forward_time": 1.7364389896392822,
      "step": 28194
    },
    {
      "epoch": 2.688789367675781e-05,
      "step": 28194,
      "training_step_time": 9.381669044494629
    },
    {
      "epoch": 2.688884735107422e-05,
      "model_forward_time": 1.744800329208374,
      "step": 28195
    },
    {
      "epoch": 2.688884735107422e-05,
      "step": 28195,
      "training_step_time": 8.334845542907715
    },
    {
      "epoch": 2.6889801025390626e-05,
      "model_forward_time": 1.7424392700195312,
      "step": 28196
    },
    {
      "epoch": 2.6889801025390626e-05,
      "step": 28196,
      "training_step_time": 7.77534294128418
    },
    {
      "epoch": 2.689075469970703e-05,
      "model_forward_time": 1.7408668994903564,
      "step": 28197
    },
    {
      "epoch": 2.689075469970703e-05,
      "step": 28197,
      "training_step_time": 8.32276725769043
    },
    {
      "epoch": 2.6891708374023438e-05,
      "model_forward_time": 1.7404417991638184,
      "step": 28198
    },
    {
      "epoch": 2.6891708374023438e-05,
      "step": 28198,
      "training_step_time": 8.553523063659668
    },
    {
      "epoch": 2.6892662048339845e-05,
      "model_forward_time": 1.7360742092132568,
      "step": 28199
    },
    {
      "epoch": 2.6892662048339845e-05,
      "step": 28199,
      "training_step_time": 7.760973691940308
    },
    {
      "epoch": 2.689361572265625e-05,
      "grad_norm": 0.18459433317184448,
      "learning_rate": 8.599365237536751e-06,
      "loss": 0.0577,
      "step": 28200
    },
    {
      "epoch": 2.689361572265625e-05,
      "model_forward_time": 1.7413620948791504,
      "step": 28200
    },
    {
      "dynamics_loss_avg": 0.05442420486360788,
      "epoch": 2.689361572265625e-05,
      "step": 28200
    },
    {
      "action_loss_avg": 0.001968765677884221,
      "epoch": 2.689361572265625e-05,
      "step": 28200
    },
    {
      "epoch": 2.689361572265625e-05,
      "step": 28200,
      "training_step_time": 8.89193320274353
    },
    {
      "epoch": 2.6894569396972657e-05,
      "model_forward_time": 1.7858266830444336,
      "step": 28201
    },
    {
      "epoch": 2.6894569396972657e-05,
      "step": 28201,
      "training_step_time": 7.944892168045044
    },
    {
      "epoch": 2.6895523071289064e-05,
      "model_forward_time": 2.839646339416504,
      "step": 28202
    },
    {
      "epoch": 2.6895523071289064e-05,
      "step": 28202,
      "training_step_time": 9.454930067062378
    },
    {
      "epoch": 2.6896476745605468e-05,
      "model_forward_time": 1.7359514236450195,
      "step": 28203
    },
    {
      "epoch": 2.6896476745605468e-05,
      "step": 28203,
      "training_step_time": 7.835411310195923
    },
    {
      "epoch": 2.6897430419921876e-05,
      "model_forward_time": 1.7425758838653564,
      "step": 28204
    },
    {
      "epoch": 2.6897430419921876e-05,
      "step": 28204,
      "training_step_time": 8.747034311294556
    },
    {
      "epoch": 2.689838409423828e-05,
      "model_forward_time": 1.3745923042297363,
      "step": 28205
    },
    {
      "epoch": 2.689838409423828e-05,
      "step": 28205,
      "training_step_time": 7.860590696334839
    },
    {
      "epoch": 2.6899337768554687e-05,
      "model_forward_time": 1.7368664741516113,
      "step": 28206
    },
    {
      "epoch": 2.6899337768554687e-05,
      "step": 28206,
      "training_step_time": 7.780405759811401
    },
    {
      "epoch": 2.6900291442871094e-05,
      "model_forward_time": 1.7438361644744873,
      "step": 28207
    },
    {
      "epoch": 2.6900291442871094e-05,
      "step": 28207,
      "training_step_time": 7.915660619735718
    },
    {
      "epoch": 2.69012451171875e-05,
      "model_forward_time": 1.7361564636230469,
      "step": 28208
    },
    {
      "epoch": 2.69012451171875e-05,
      "step": 28208,
      "training_step_time": 7.8036768436431885
    },
    {
      "epoch": 2.6902198791503906e-05,
      "model_forward_time": 1.7339181900024414,
      "step": 28209
    },
    {
      "epoch": 2.6902198791503906e-05,
      "step": 28209,
      "training_step_time": 7.859572410583496
    },
    {
      "epoch": 2.6903152465820313e-05,
      "grad_norm": 0.18302124738693237,
      "learning_rate": 8.598217357835264e-06,
      "loss": 0.0642,
      "step": 28210
    },
    {
      "epoch": 2.6903152465820313e-05,
      "model_forward_time": 1.3730618953704834,
      "step": 28210
    },
    {
      "dynamics_loss_avg": 0.059835558757185936,
      "epoch": 2.6903152465820313e-05,
      "step": 28210
    },
    {
      "action_loss_avg": 0.0019797429791651665,
      "epoch": 2.6903152465820313e-05,
      "step": 28210
    },
    {
      "epoch": 2.6903152465820313e-05,
      "step": 28210,
      "training_step_time": 8.45075011253357
    },
    {
      "epoch": 2.6904106140136717e-05,
      "model_forward_time": 1.7479045391082764,
      "step": 28211
    },
    {
      "epoch": 2.6904106140136717e-05,
      "step": 28211,
      "training_step_time": 8.614909648895264
    },
    {
      "epoch": 2.6905059814453125e-05,
      "model_forward_time": 2.9084525108337402,
      "step": 28212
    },
    {
      "epoch": 2.6905059814453125e-05,
      "step": 28212,
      "training_step_time": 7.778743743896484
    },
    {
      "epoch": 2.6906013488769532e-05,
      "model_forward_time": 1.740461826324463,
      "step": 28213
    },
    {
      "epoch": 2.6906013488769532e-05,
      "step": 28213,
      "training_step_time": 8.323320865631104
    },
    {
      "epoch": 2.6906967163085936e-05,
      "model_forward_time": 1.7435212135314941,
      "step": 28214
    },
    {
      "epoch": 2.6906967163085936e-05,
      "step": 28214,
      "training_step_time": 7.745515584945679
    },
    {
      "epoch": 2.6907920837402344e-05,
      "model_forward_time": 1.7325255870819092,
      "step": 28215
    },
    {
      "epoch": 2.6907920837402344e-05,
      "step": 28215,
      "training_step_time": 7.646799564361572
    },
    {
      "epoch": 2.690887451171875e-05,
      "model_forward_time": 1.7340335845947266,
      "step": 28216
    },
    {
      "epoch": 2.690887451171875e-05,
      "step": 28216,
      "training_step_time": 7.731338739395142
    },
    {
      "epoch": 2.6909828186035155e-05,
      "model_forward_time": 1.7430591583251953,
      "step": 28217
    },
    {
      "epoch": 2.6909828186035155e-05,
      "step": 28217,
      "training_step_time": 8.241921186447144
    },
    {
      "epoch": 2.6910781860351562e-05,
      "model_forward_time": 1.7399792671203613,
      "step": 28218
    },
    {
      "epoch": 2.6910781860351562e-05,
      "step": 28218,
      "training_step_time": 8.762980461120605
    },
    {
      "epoch": 2.691173553466797e-05,
      "model_forward_time": 1.7381105422973633,
      "step": 28219
    },
    {
      "epoch": 2.691173553466797e-05,
      "step": 28219,
      "training_step_time": 8.990598678588867
    },
    {
      "epoch": 2.6912689208984374e-05,
      "grad_norm": 0.20494785904884338,
      "learning_rate": 8.597069084638135e-06,
      "loss": 0.0612,
      "step": 28220
    },
    {
      "epoch": 2.6912689208984374e-05,
      "model_forward_time": 1.797921895980835,
      "step": 28220
    },
    {
      "dynamics_loss_avg": 0.052890246361494066,
      "epoch": 2.6912689208984374e-05,
      "step": 28220
    },
    {
      "action_loss_avg": 0.002357016602763906,
      "epoch": 2.6912689208984374e-05,
      "step": 28220
    },
    {
      "epoch": 2.6912689208984374e-05,
      "step": 28220,
      "training_step_time": 10.431612491607666
    },
    {
      "epoch": 2.691364288330078e-05,
      "model_forward_time": 2.8798086643218994,
      "step": 28221
    },
    {
      "epoch": 2.691364288330078e-05,
      "step": 28221,
      "training_step_time": 7.877673149108887
    },
    {
      "epoch": 2.691459655761719e-05,
      "model_forward_time": 1.7327086925506592,
      "step": 28222
    },
    {
      "epoch": 2.691459655761719e-05,
      "step": 28222,
      "training_step_time": 7.85906195640564
    },
    {
      "epoch": 2.6915550231933593e-05,
      "model_forward_time": 1.7322232723236084,
      "step": 28223
    },
    {
      "epoch": 2.6915550231933593e-05,
      "step": 28223,
      "training_step_time": 9.260034561157227
    },
    {
      "epoch": 2.691650390625e-05,
      "model_forward_time": 1.7423043251037598,
      "step": 28224
    },
    {
      "epoch": 2.691650390625e-05,
      "step": 28224,
      "training_step_time": 8.40441346168518
    },
    {
      "epoch": 2.6917457580566408e-05,
      "model_forward_time": 1.7617530822753906,
      "step": 28225
    },
    {
      "epoch": 2.6917457580566408e-05,
      "step": 28225,
      "training_step_time": 8.652652740478516
    },
    {
      "epoch": 2.691841125488281e-05,
      "model_forward_time": 1.7400791645050049,
      "step": 28226
    },
    {
      "epoch": 2.691841125488281e-05,
      "step": 28226,
      "training_step_time": 7.6832053661346436
    },
    {
      "epoch": 2.691936492919922e-05,
      "model_forward_time": 1.7327244281768799,
      "step": 28227
    },
    {
      "epoch": 2.691936492919922e-05,
      "step": 28227,
      "training_step_time": 7.735928058624268
    },
    {
      "epoch": 2.6920318603515626e-05,
      "model_forward_time": 1.736163854598999,
      "step": 28228
    },
    {
      "epoch": 2.6920318603515626e-05,
      "step": 28228,
      "training_step_time": 7.879936456680298
    },
    {
      "epoch": 2.692127227783203e-05,
      "model_forward_time": 1.8022348880767822,
      "step": 28229
    },
    {
      "epoch": 2.692127227783203e-05,
      "step": 28229,
      "training_step_time": 7.7611753940582275
    },
    {
      "epoch": 2.6922225952148438e-05,
      "grad_norm": 0.18471848964691162,
      "learning_rate": 8.59592041807094e-06,
      "loss": 0.0657,
      "step": 28230
    },
    {
      "epoch": 2.6922225952148438e-05,
      "model_forward_time": 2.8719558715820312,
      "step": 28230
    },
    {
      "dynamics_loss_avg": 0.057837747130542995,
      "epoch": 2.6922225952148438e-05,
      "step": 28230
    },
    {
      "action_loss_avg": 0.0024842581071425228,
      "epoch": 2.6922225952148438e-05,
      "step": 28230
    },
    {
      "epoch": 2.6922225952148438e-05,
      "step": 28230,
      "training_step_time": 9.662024021148682
    },
    {
      "epoch": 2.6923179626464845e-05,
      "model_forward_time": 1.7393417358398438,
      "step": 28231
    },
    {
      "epoch": 2.6923179626464845e-05,
      "step": 28231,
      "training_step_time": 7.763920545578003
    },
    {
      "epoch": 2.692413330078125e-05,
      "model_forward_time": 1.7345242500305176,
      "step": 28232
    },
    {
      "epoch": 2.692413330078125e-05,
      "step": 28232,
      "training_step_time": 8.524815797805786
    },
    {
      "epoch": 2.6925086975097657e-05,
      "model_forward_time": 1.7387192249298096,
      "step": 28233
    },
    {
      "epoch": 2.6925086975097657e-05,
      "step": 28233,
      "training_step_time": 7.7790796756744385
    },
    {
      "epoch": 2.6926040649414064e-05,
      "model_forward_time": 1.7367148399353027,
      "step": 28234
    },
    {
      "epoch": 2.6926040649414064e-05,
      "step": 28234,
      "training_step_time": 8.791898012161255
    },
    {
      "epoch": 2.6926994323730468e-05,
      "model_forward_time": 1.7345495223999023,
      "step": 28235
    },
    {
      "epoch": 2.6926994323730468e-05,
      "step": 28235,
      "training_step_time": 7.686025381088257
    },
    {
      "epoch": 2.6927947998046876e-05,
      "model_forward_time": 1.7357463836669922,
      "step": 28236
    },
    {
      "epoch": 2.6927947998046876e-05,
      "step": 28236,
      "training_step_time": 7.795385122299194
    },
    {
      "epoch": 2.692890167236328e-05,
      "model_forward_time": 1.7380802631378174,
      "step": 28237
    },
    {
      "epoch": 2.692890167236328e-05,
      "step": 28237,
      "training_step_time": 8.373547554016113
    },
    {
      "epoch": 2.6929855346679687e-05,
      "model_forward_time": 1.8019144535064697,
      "step": 28238
    },
    {
      "epoch": 2.6929855346679687e-05,
      "step": 28238,
      "training_step_time": 7.855783939361572
    },
    {
      "epoch": 2.6930809020996094e-05,
      "model_forward_time": 2.8332481384277344,
      "step": 28239
    },
    {
      "epoch": 2.6930809020996094e-05,
      "step": 28239,
      "training_step_time": 9.629124402999878
    },
    {
      "epoch": 2.69317626953125e-05,
      "grad_norm": 0.2117459774017334,
      "learning_rate": 8.594771358259295e-06,
      "loss": 0.061,
      "step": 28240
    },
    {
      "epoch": 2.69317626953125e-05,
      "model_forward_time": 1.7280144691467285,
      "step": 28240
    },
    {
      "dynamics_loss_avg": 0.04511715900152922,
      "epoch": 2.69317626953125e-05,
      "step": 28240
    },
    {
      "action_loss_avg": 0.0014857925823889672,
      "epoch": 2.69317626953125e-05,
      "step": 28240
    },
    {
      "epoch": 2.69317626953125e-05,
      "step": 28240,
      "training_step_time": 8.617259979248047
    },
    {
      "epoch": 2.6932716369628906e-05,
      "model_forward_time": 1.737471580505371,
      "step": 28241
    },
    {
      "epoch": 2.6932716369628906e-05,
      "step": 28241,
      "training_step_time": 8.513241052627563
    },
    {
      "epoch": 2.6933670043945313e-05,
      "model_forward_time": 1.742030382156372,
      "step": 28242
    },
    {
      "epoch": 2.6933670043945313e-05,
      "step": 28242,
      "training_step_time": 7.784132480621338
    },
    {
      "epoch": 2.6934623718261717e-05,
      "model_forward_time": 1.7351479530334473,
      "step": 28243
    },
    {
      "epoch": 2.6934623718261717e-05,
      "step": 28243,
      "training_step_time": 7.712791681289673
    },
    {
      "epoch": 2.6935577392578125e-05,
      "model_forward_time": 1.3741865158081055,
      "step": 28244
    },
    {
      "epoch": 2.6935577392578125e-05,
      "step": 28244,
      "training_step_time": 7.764374017715454
    },
    {
      "epoch": 2.6936531066894532e-05,
      "model_forward_time": 1.7379491329193115,
      "step": 28245
    },
    {
      "epoch": 2.6936531066894532e-05,
      "step": 28245,
      "training_step_time": 7.781846761703491
    },
    {
      "epoch": 2.6937484741210936e-05,
      "model_forward_time": 1.7358081340789795,
      "step": 28246
    },
    {
      "epoch": 2.6937484741210936e-05,
      "step": 28246,
      "training_step_time": 7.865959644317627
    },
    {
      "epoch": 2.6938438415527344e-05,
      "model_forward_time": 1.7718970775604248,
      "step": 28247
    },
    {
      "epoch": 2.6938438415527344e-05,
      "step": 28247,
      "training_step_time": 7.8779520988464355
    },
    {
      "epoch": 2.693939208984375e-05,
      "model_forward_time": 1.7357697486877441,
      "step": 28248
    },
    {
      "epoch": 2.693939208984375e-05,
      "step": 28248,
      "training_step_time": 7.78695011138916
    },
    {
      "epoch": 2.6940345764160155e-05,
      "model_forward_time": 2.932060718536377,
      "step": 28249
    },
    {
      "epoch": 2.6940345764160155e-05,
      "step": 28249,
      "training_step_time": 8.854919672012329
    },
    {
      "epoch": 2.6941299438476562e-05,
      "grad_norm": 0.24577419459819794,
      "learning_rate": 8.593621905328857e-06,
      "loss": 0.0628,
      "step": 28250
    },
    {
      "epoch": 2.6941299438476562e-05,
      "model_forward_time": 1.7364094257354736,
      "step": 28250
    },
    {
      "dynamics_loss_avg": 0.05167727246880531,
      "epoch": 2.6941299438476562e-05,
      "step": 28250
    },
    {
      "action_loss_avg": 0.0020110490848310294,
      "epoch": 2.6941299438476562e-05,
      "step": 28250
    },
    {
      "epoch": 2.6941299438476562e-05,
      "step": 28250,
      "training_step_time": 8.798248052597046
    },
    {
      "epoch": 2.694225311279297e-05,
      "model_forward_time": 1.7327766418457031,
      "step": 28251
    },
    {
      "epoch": 2.694225311279297e-05,
      "step": 28251,
      "training_step_time": 8.290976285934448
    },
    {
      "epoch": 2.6943206787109374e-05,
      "model_forward_time": 1.7392594814300537,
      "step": 28252
    },
    {
      "epoch": 2.6943206787109374e-05,
      "step": 28252,
      "training_step_time": 7.717998743057251
    },
    {
      "epoch": 2.694416046142578e-05,
      "model_forward_time": 1.7440054416656494,
      "step": 28253
    },
    {
      "epoch": 2.694416046142578e-05,
      "step": 28253,
      "training_step_time": 7.871467351913452
    },
    {
      "epoch": 2.694511413574219e-05,
      "model_forward_time": 1.7374918460845947,
      "step": 28254
    },
    {
      "epoch": 2.694511413574219e-05,
      "step": 28254,
      "training_step_time": 7.789637088775635
    },
    {
      "epoch": 2.6946067810058593e-05,
      "model_forward_time": 1.7423474788665771,
      "step": 28255
    },
    {
      "epoch": 2.6946067810058593e-05,
      "step": 28255,
      "training_step_time": 9.51228404045105
    },
    {
      "epoch": 2.6947021484375e-05,
      "model_forward_time": 1.7326509952545166,
      "step": 28256
    },
    {
      "epoch": 2.6947021484375e-05,
      "step": 28256,
      "training_step_time": 8.363276720046997
    },
    {
      "epoch": 2.6947975158691408e-05,
      "model_forward_time": 1.7397394180297852,
      "step": 28257
    },
    {
      "epoch": 2.6947975158691408e-05,
      "step": 28257,
      "training_step_time": 9.476024389266968
    },
    {
      "epoch": 2.694892883300781e-05,
      "model_forward_time": 2.840123176574707,
      "step": 28258
    },
    {
      "epoch": 2.694892883300781e-05,
      "step": 28258,
      "training_step_time": 7.829550266265869
    },
    {
      "epoch": 2.694988250732422e-05,
      "model_forward_time": 1.742607831954956,
      "step": 28259
    },
    {
      "epoch": 2.694988250732422e-05,
      "step": 28259,
      "training_step_time": 7.925098419189453
    },
    {
      "epoch": 2.6950836181640626e-05,
      "grad_norm": 0.21213574707508087,
      "learning_rate": 8.592472059405332e-06,
      "loss": 0.0638,
      "step": 28260
    },
    {
      "epoch": 2.6950836181640626e-05,
      "model_forward_time": 1.7583084106445312,
      "step": 28260
    },
    {
      "dynamics_loss_avg": 0.07360602226108312,
      "epoch": 2.6950836181640626e-05,
      "step": 28260
    },
    {
      "action_loss_avg": 0.0028049829765222968,
      "epoch": 2.6950836181640626e-05,
      "step": 28260
    },
    {
      "epoch": 2.6950836181640626e-05,
      "step": 28260,
      "training_step_time": 7.737385511398315
    },
    {
      "epoch": 2.695178985595703e-05,
      "model_forward_time": 1.7365236282348633,
      "step": 28261
    },
    {
      "epoch": 2.695178985595703e-05,
      "step": 28261,
      "training_step_time": 7.567126750946045
    },
    {
      "epoch": 2.6952743530273438e-05,
      "model_forward_time": 1.7283508777618408,
      "step": 28262
    },
    {
      "epoch": 2.6952743530273438e-05,
      "step": 28262,
      "training_step_time": 7.850548982620239
    },
    {
      "epoch": 2.6953697204589845e-05,
      "model_forward_time": 1.7400100231170654,
      "step": 28263
    },
    {
      "epoch": 2.6953697204589845e-05,
      "step": 28263,
      "training_step_time": 7.747636079788208
    },
    {
      "epoch": 2.695465087890625e-05,
      "model_forward_time": 1.7649116516113281,
      "step": 28264
    },
    {
      "epoch": 2.695465087890625e-05,
      "step": 28264,
      "training_step_time": 9.370970249176025
    },
    {
      "epoch": 2.6955604553222657e-05,
      "model_forward_time": 1.7386324405670166,
      "step": 28265
    },
    {
      "epoch": 2.6955604553222657e-05,
      "step": 28265,
      "training_step_time": 7.774794340133667
    },
    {
      "epoch": 2.695655822753906e-05,
      "model_forward_time": 1.8022739887237549,
      "step": 28266
    },
    {
      "epoch": 2.695655822753906e-05,
      "step": 28266,
      "training_step_time": 7.897031784057617
    },
    {
      "epoch": 2.6957511901855468e-05,
      "model_forward_time": 2.843379020690918,
      "step": 28267
    },
    {
      "epoch": 2.6957511901855468e-05,
      "step": 28267,
      "training_step_time": 8.656896114349365
    },
    {
      "epoch": 2.6958465576171876e-05,
      "model_forward_time": 1.7326791286468506,
      "step": 28268
    },
    {
      "epoch": 2.6958465576171876e-05,
      "step": 28268,
      "training_step_time": 7.838428497314453
    },
    {
      "epoch": 2.695941925048828e-05,
      "model_forward_time": 1.7489221096038818,
      "step": 28269
    },
    {
      "epoch": 2.695941925048828e-05,
      "step": 28269,
      "training_step_time": 7.831878662109375
    },
    {
      "epoch": 2.6960372924804687e-05,
      "grad_norm": 0.25640588998794556,
      "learning_rate": 8.591321820614465e-06,
      "loss": 0.0612,
      "step": 28270
    },
    {
      "epoch": 2.6960372924804687e-05,
      "model_forward_time": 1.7345805168151855,
      "step": 28270
    },
    {
      "dynamics_loss_avg": 0.06728582829236984,
      "epoch": 2.6960372924804687e-05,
      "step": 28270
    },
    {
      "action_loss_avg": 0.0029418001300655305,
      "epoch": 2.6960372924804687e-05,
      "step": 28270
    },
    {
      "epoch": 2.6960372924804687e-05,
      "step": 28270,
      "training_step_time": 7.823878765106201
    },
    {
      "epoch": 2.6961326599121095e-05,
      "model_forward_time": 1.7368924617767334,
      "step": 28271
    },
    {
      "epoch": 2.6961326599121095e-05,
      "step": 28271,
      "training_step_time": 7.858432292938232
    },
    {
      "epoch": 2.69622802734375e-05,
      "model_forward_time": 1.7546792030334473,
      "step": 28272
    },
    {
      "epoch": 2.69622802734375e-05,
      "step": 28272,
      "training_step_time": 7.833743333816528
    },
    {
      "epoch": 2.6963233947753906e-05,
      "model_forward_time": 1.7376580238342285,
      "step": 28273
    },
    {
      "epoch": 2.6963233947753906e-05,
      "step": 28273,
      "training_step_time": 7.846546411514282
    },
    {
      "epoch": 2.6964187622070313e-05,
      "model_forward_time": 1.7393808364868164,
      "step": 28274
    },
    {
      "epoch": 2.6964187622070313e-05,
      "step": 28274,
      "training_step_time": 9.905771493911743
    },
    {
      "epoch": 2.6965141296386717e-05,
      "model_forward_time": 1.7648224830627441,
      "step": 28275
    },
    {
      "epoch": 2.6965141296386717e-05,
      "step": 28275,
      "training_step_time": 8.339359998703003
    },
    {
      "epoch": 2.6966094970703125e-05,
      "model_forward_time": 2.8620247840881348,
      "step": 28276
    },
    {
      "epoch": 2.6966094970703125e-05,
      "step": 28276,
      "training_step_time": 7.825641870498657
    },
    {
      "epoch": 2.6967048645019532e-05,
      "model_forward_time": 1.7470171451568604,
      "step": 28277
    },
    {
      "epoch": 2.6967048645019532e-05,
      "step": 28277,
      "training_step_time": 7.866196155548096
    },
    {
      "epoch": 2.6968002319335936e-05,
      "model_forward_time": 1.7539093494415283,
      "step": 28278
    },
    {
      "epoch": 2.6968002319335936e-05,
      "step": 28278,
      "training_step_time": 7.818674087524414
    },
    {
      "epoch": 2.6968955993652344e-05,
      "model_forward_time": 1.749650239944458,
      "step": 28279
    },
    {
      "epoch": 2.6968955993652344e-05,
      "step": 28279,
      "training_step_time": 7.907248497009277
    },
    {
      "epoch": 2.696990966796875e-05,
      "grad_norm": 0.15659338235855103,
      "learning_rate": 8.590171189082042e-06,
      "loss": 0.0627,
      "step": 28280
    },
    {
      "epoch": 2.696990966796875e-05,
      "model_forward_time": 1.757063865661621,
      "step": 28280
    },
    {
      "dynamics_loss_avg": 0.045847002696245906,
      "epoch": 2.696990966796875e-05,
      "step": 28280
    },
    {
      "action_loss_avg": 0.0016783884260803462,
      "epoch": 2.696990966796875e-05,
      "step": 28280
    },
    {
      "epoch": 2.696990966796875e-05,
      "step": 28280,
      "training_step_time": 8.540358781814575
    },
    {
      "epoch": 2.6970863342285155e-05,
      "model_forward_time": 1.7568867206573486,
      "step": 28281
    },
    {
      "epoch": 2.6970863342285155e-05,
      "step": 28281,
      "training_step_time": 7.840630054473877
    },
    {
      "epoch": 2.6971817016601563e-05,
      "model_forward_time": 1.7526721954345703,
      "step": 28282
    },
    {
      "epoch": 2.6971817016601563e-05,
      "step": 28282,
      "training_step_time": 7.875591278076172
    },
    {
      "epoch": 2.697277069091797e-05,
      "model_forward_time": 1.7458536624908447,
      "step": 28283
    },
    {
      "epoch": 2.697277069091797e-05,
      "step": 28283,
      "training_step_time": 7.7545835971832275
    },
    {
      "epoch": 2.6973724365234374e-05,
      "model_forward_time": 1.7498583793640137,
      "step": 28284
    },
    {
      "epoch": 2.6973724365234374e-05,
      "step": 28284,
      "training_step_time": 8.541825532913208
    },
    {
      "epoch": 2.697467803955078e-05,
      "model_forward_time": 1.7504122257232666,
      "step": 28285
    },
    {
      "epoch": 2.697467803955078e-05,
      "step": 28285,
      "training_step_time": 11.262120485305786
    },
    {
      "epoch": 2.697563171386719e-05,
      "model_forward_time": 1.746274471282959,
      "step": 28286
    },
    {
      "epoch": 2.697563171386719e-05,
      "step": 28286,
      "training_step_time": 7.920135974884033
    },
    {
      "epoch": 2.6976585388183593e-05,
      "model_forward_time": 1.7474658489227295,
      "step": 28287
    },
    {
      "epoch": 2.6976585388183593e-05,
      "step": 28287,
      "training_step_time": 7.839036464691162
    },
    {
      "epoch": 2.69775390625e-05,
      "model_forward_time": 1.7511920928955078,
      "step": 28288
    },
    {
      "epoch": 2.69775390625e-05,
      "step": 28288,
      "training_step_time": 8.65363097190857
    },
    {
      "epoch": 2.6978492736816408e-05,
      "model_forward_time": 1.7479827404022217,
      "step": 28289
    },
    {
      "epoch": 2.6978492736816408e-05,
      "step": 28289,
      "training_step_time": 7.689406871795654
    },
    {
      "epoch": 2.697944641113281e-05,
      "grad_norm": 0.20752982795238495,
      "learning_rate": 8.589020164933895e-06,
      "loss": 0.0626,
      "step": 28290
    },
    {
      "epoch": 2.697944641113281e-05,
      "model_forward_time": 1.758378505706787,
      "step": 28290
    },
    {
      "dynamics_loss_avg": 0.06424328126013279,
      "epoch": 2.697944641113281e-05,
      "step": 28290
    },
    {
      "action_loss_avg": 0.0029400946863461287,
      "epoch": 2.697944641113281e-05,
      "step": 28290
    },
    {
      "epoch": 2.697944641113281e-05,
      "step": 28290,
      "training_step_time": 7.832324028015137
    },
    {
      "epoch": 2.698040008544922e-05,
      "model_forward_time": 1.7492713928222656,
      "step": 28291
    },
    {
      "epoch": 2.698040008544922e-05,
      "step": 28291,
      "training_step_time": 7.8460893630981445
    },
    {
      "epoch": 2.6981353759765627e-05,
      "model_forward_time": 1.7459399700164795,
      "step": 28292
    },
    {
      "epoch": 2.6981353759765627e-05,
      "step": 28292,
      "training_step_time": 8.792189836502075
    },
    {
      "epoch": 2.698230743408203e-05,
      "model_forward_time": 1.7485828399658203,
      "step": 28293
    },
    {
      "epoch": 2.698230743408203e-05,
      "step": 28293,
      "training_step_time": 8.41580581665039
    },
    {
      "epoch": 2.6983261108398438e-05,
      "model_forward_time": 1.7974498271942139,
      "step": 28294
    },
    {
      "epoch": 2.6983261108398438e-05,
      "step": 28294,
      "training_step_time": 9.534943580627441
    },
    {
      "epoch": 2.6984214782714845e-05,
      "model_forward_time": 2.8842055797576904,
      "step": 28295
    },
    {
      "epoch": 2.6984214782714845e-05,
      "step": 28295,
      "training_step_time": 8.635087251663208
    },
    {
      "epoch": 2.698516845703125e-05,
      "model_forward_time": 1.7472360134124756,
      "step": 28296
    },
    {
      "epoch": 2.698516845703125e-05,
      "step": 28296,
      "training_step_time": 8.237834215164185
    },
    {
      "epoch": 2.6986122131347657e-05,
      "model_forward_time": 1.7518718242645264,
      "step": 28297
    },
    {
      "epoch": 2.6986122131347657e-05,
      "step": 28297,
      "training_step_time": 7.666199207305908
    },
    {
      "epoch": 2.698707580566406e-05,
      "model_forward_time": 1.7440059185028076,
      "step": 28298
    },
    {
      "epoch": 2.698707580566406e-05,
      "step": 28298,
      "training_step_time": 8.33713173866272
    },
    {
      "epoch": 2.6988029479980468e-05,
      "model_forward_time": 1.7344913482666016,
      "step": 28299
    },
    {
      "epoch": 2.6988029479980468e-05,
      "step": 28299,
      "training_step_time": 8.841440916061401
    },
    {
      "epoch": 2.6988983154296876e-05,
      "grad_norm": 0.17455238103866577,
      "learning_rate": 8.5878687482959e-06,
      "loss": 0.0665,
      "step": 28300
    },
    {
      "epoch": 2.6988983154296876e-05,
      "model_forward_time": 1.7563972473144531,
      "step": 28300
    },
    {
      "dynamics_loss_avg": 0.0598704693838954,
      "epoch": 2.6988983154296876e-05,
      "step": 28300
    },
    {
      "action_loss_avg": 0.00152076703379862,
      "epoch": 2.6988983154296876e-05,
      "step": 28300
    },
    {
      "epoch": 2.6988983154296876e-05,
      "step": 28300,
      "training_step_time": 7.874620676040649
    },
    {
      "epoch": 2.698993682861328e-05,
      "model_forward_time": 1.7467873096466064,
      "step": 28301
    },
    {
      "epoch": 2.698993682861328e-05,
      "step": 28301,
      "training_step_time": 7.748943567276001
    },
    {
      "epoch": 2.6990890502929687e-05,
      "model_forward_time": 1.7575721740722656,
      "step": 28302
    },
    {
      "epoch": 2.6990890502929687e-05,
      "step": 28302,
      "training_step_time": 7.928934097290039
    },
    {
      "epoch": 2.6991844177246095e-05,
      "model_forward_time": 1.8195030689239502,
      "step": 28303
    },
    {
      "epoch": 2.6991844177246095e-05,
      "step": 28303,
      "training_step_time": 9.38503384590149
    },
    {
      "epoch": 2.69927978515625e-05,
      "model_forward_time": 2.875335216522217,
      "step": 28304
    },
    {
      "epoch": 2.69927978515625e-05,
      "step": 28304,
      "training_step_time": 8.804269075393677
    },
    {
      "epoch": 2.6993751525878906e-05,
      "model_forward_time": 1.754286766052246,
      "step": 28305
    },
    {
      "epoch": 2.6993751525878906e-05,
      "step": 28305,
      "training_step_time": 7.819446802139282
    },
    {
      "epoch": 2.6994705200195313e-05,
      "model_forward_time": 1.7559154033660889,
      "step": 28306
    },
    {
      "epoch": 2.6994705200195313e-05,
      "step": 28306,
      "training_step_time": 8.788256645202637
    },
    {
      "epoch": 2.6995658874511717e-05,
      "model_forward_time": 1.3919129371643066,
      "step": 28307
    },
    {
      "epoch": 2.6995658874511717e-05,
      "step": 28307,
      "training_step_time": 7.249370098114014
    },
    {
      "epoch": 2.6996612548828125e-05,
      "model_forward_time": 1.7539103031158447,
      "step": 28308
    },
    {
      "epoch": 2.6996612548828125e-05,
      "step": 28308,
      "training_step_time": 7.8021039962768555
    },
    {
      "epoch": 2.6997566223144532e-05,
      "model_forward_time": 1.7503364086151123,
      "step": 28309
    },
    {
      "epoch": 2.6997566223144532e-05,
      "step": 28309,
      "training_step_time": 8.780891180038452
    },
    {
      "epoch": 2.6998519897460936e-05,
      "grad_norm": 0.11159399896860123,
      "learning_rate": 8.58671693929397e-06,
      "loss": 0.0612,
      "step": 28310
    },
    {
      "epoch": 2.6998519897460936e-05,
      "model_forward_time": 1.7508671283721924,
      "step": 28310
    },
    {
      "dynamics_loss_avg": 0.07007334176450968,
      "epoch": 2.6998519897460936e-05,
      "step": 28310
    },
    {
      "action_loss_avg": 0.003157562104752287,
      "epoch": 2.6998519897460936e-05,
      "step": 28310
    },
    {
      "epoch": 2.6998519897460936e-05,
      "step": 28310,
      "training_step_time": 7.760371685028076
    },
    {
      "epoch": 2.6999473571777344e-05,
      "model_forward_time": 1.7592053413391113,
      "step": 28311
    },
    {
      "epoch": 2.6999473571777344e-05,
      "step": 28311,
      "training_step_time": 8.828211545944214
    },
    {
      "epoch": 2.700042724609375e-05,
      "model_forward_time": 1.8191101551055908,
      "step": 28312
    },
    {
      "epoch": 2.700042724609375e-05,
      "step": 28312,
      "training_step_time": 11.658851623535156
    },
    {
      "epoch": 2.7001380920410155e-05,
      "model_forward_time": 2.9071712493896484,
      "step": 28313
    },
    {
      "epoch": 2.7001380920410155e-05,
      "step": 28313,
      "training_step_time": 8.898072481155396
    },
    {
      "epoch": 2.7002334594726563e-05,
      "model_forward_time": 1.7569572925567627,
      "step": 28314
    },
    {
      "epoch": 2.7002334594726563e-05,
      "step": 28314,
      "training_step_time": 7.876360654830933
    },
    {
      "epoch": 2.700328826904297e-05,
      "model_forward_time": 1.7833364009857178,
      "step": 28315
    },
    {
      "epoch": 2.700328826904297e-05,
      "step": 28315,
      "training_step_time": 7.797282457351685
    },
    {
      "epoch": 2.7004241943359374e-05,
      "model_forward_time": 1.7457146644592285,
      "step": 28316
    },
    {
      "epoch": 2.7004241943359374e-05,
      "step": 28316,
      "training_step_time": 7.7674174308776855
    },
    {
      "epoch": 2.700519561767578e-05,
      "model_forward_time": 1.7494158744812012,
      "step": 28317
    },
    {
      "epoch": 2.700519561767578e-05,
      "step": 28317,
      "training_step_time": 7.781510829925537
    },
    {
      "epoch": 2.700614929199219e-05,
      "model_forward_time": 1.7509732246398926,
      "step": 28318
    },
    {
      "epoch": 2.700614929199219e-05,
      "step": 28318,
      "training_step_time": 7.85512113571167
    },
    {
      "epoch": 2.7007102966308593e-05,
      "model_forward_time": 1.7424063682556152,
      "step": 28319
    },
    {
      "epoch": 2.7007102966308593e-05,
      "step": 28319,
      "training_step_time": 7.825018405914307
    },
    {
      "epoch": 2.7008056640625e-05,
      "grad_norm": 0.15754516422748566,
      "learning_rate": 8.58556473805407e-06,
      "loss": 0.0638,
      "step": 28320
    },
    {
      "epoch": 2.7008056640625e-05,
      "model_forward_time": 1.7464146614074707,
      "step": 28320
    },
    {
      "dynamics_loss_avg": 0.06306052729487419,
      "epoch": 2.7008056640625e-05,
      "step": 28320
    },
    {
      "action_loss_avg": 0.0031975981895811857,
      "epoch": 2.7008056640625e-05,
      "step": 28320
    },
    {
      "epoch": 2.7008056640625e-05,
      "step": 28320,
      "training_step_time": 7.699367046356201
    },
    {
      "epoch": 2.7009010314941408e-05,
      "model_forward_time": 1.7549984455108643,
      "step": 28321
    },
    {
      "epoch": 2.7009010314941408e-05,
      "step": 28321,
      "training_step_time": 8.512760400772095
    },
    {
      "epoch": 2.7009963989257812e-05,
      "model_forward_time": 2.887639284133911,
      "step": 28322
    },
    {
      "epoch": 2.7009963989257812e-05,
      "step": 28322,
      "training_step_time": 7.828145742416382
    },
    {
      "epoch": 2.701091766357422e-05,
      "model_forward_time": 1.7481415271759033,
      "step": 28323
    },
    {
      "epoch": 2.701091766357422e-05,
      "step": 28323,
      "training_step_time": 7.797920227050781
    },
    {
      "epoch": 2.7011871337890627e-05,
      "model_forward_time": 1.7384748458862305,
      "step": 28324
    },
    {
      "epoch": 2.7011871337890627e-05,
      "step": 28324,
      "training_step_time": 7.789612770080566
    },
    {
      "epoch": 2.701282501220703e-05,
      "model_forward_time": 1.7493634223937988,
      "step": 28325
    },
    {
      "epoch": 2.701282501220703e-05,
      "step": 28325,
      "training_step_time": 7.818723440170288
    },
    {
      "epoch": 2.7013778686523438e-05,
      "model_forward_time": 1.7428843975067139,
      "step": 28326
    },
    {
      "epoch": 2.7013778686523438e-05,
      "step": 28326,
      "training_step_time": 7.755015134811401
    },
    {
      "epoch": 2.7014732360839845e-05,
      "model_forward_time": 1.7419025897979736,
      "step": 28327
    },
    {
      "epoch": 2.7014732360839845e-05,
      "step": 28327,
      "training_step_time": 7.864633798599243
    },
    {
      "epoch": 2.701568603515625e-05,
      "model_forward_time": 1.7531416416168213,
      "step": 28328
    },
    {
      "epoch": 2.701568603515625e-05,
      "step": 28328,
      "training_step_time": 7.824434518814087
    },
    {
      "epoch": 2.7016639709472657e-05,
      "model_forward_time": 1.7458059787750244,
      "step": 28329
    },
    {
      "epoch": 2.7016639709472657e-05,
      "step": 28329,
      "training_step_time": 7.683311462402344
    },
    {
      "epoch": 2.701759338378906e-05,
      "grad_norm": 0.20608893036842346,
      "learning_rate": 8.584412144702203e-06,
      "loss": 0.0628,
      "step": 28330
    },
    {
      "epoch": 2.701759338378906e-05,
      "model_forward_time": 1.7389233112335205,
      "step": 28330
    },
    {
      "dynamics_loss_avg": 0.05471017770469189,
      "epoch": 2.701759338378906e-05,
      "step": 28330
    },
    {
      "action_loss_avg": 0.0020226267573889346,
      "epoch": 2.701759338378906e-05,
      "step": 28330
    },
    {
      "epoch": 2.701759338378906e-05,
      "step": 28330,
      "training_step_time": 8.39713430404663
    },
    {
      "epoch": 2.7018547058105468e-05,
      "model_forward_time": 2.9216794967651367,
      "step": 28331
    },
    {
      "epoch": 2.7018547058105468e-05,
      "step": 28331,
      "training_step_time": 8.686325311660767
    },
    {
      "epoch": 2.7019500732421876e-05,
      "model_forward_time": 1.7419230937957764,
      "step": 28332
    },
    {
      "epoch": 2.7019500732421876e-05,
      "step": 28332,
      "training_step_time": 7.769379138946533
    },
    {
      "epoch": 2.702045440673828e-05,
      "model_forward_time": 1.745964527130127,
      "step": 28333
    },
    {
      "epoch": 2.702045440673828e-05,
      "step": 28333,
      "training_step_time": 8.956037759780884
    },
    {
      "epoch": 2.7021408081054687e-05,
      "model_forward_time": 1.7498142719268799,
      "step": 28334
    },
    {
      "epoch": 2.7021408081054687e-05,
      "step": 28334,
      "training_step_time": 7.767447233200073
    },
    {
      "epoch": 2.7022361755371095e-05,
      "model_forward_time": 1.7511343955993652,
      "step": 28335
    },
    {
      "epoch": 2.7022361755371095e-05,
      "step": 28335,
      "training_step_time": 7.633917331695557
    },
    {
      "epoch": 2.70233154296875e-05,
      "model_forward_time": 1.7509665489196777,
      "step": 28336
    },
    {
      "epoch": 2.70233154296875e-05,
      "step": 28336,
      "training_step_time": 7.805932998657227
    },
    {
      "epoch": 2.7024269104003906e-05,
      "model_forward_time": 1.7516837120056152,
      "step": 28337
    },
    {
      "epoch": 2.7024269104003906e-05,
      "step": 28337,
      "training_step_time": 7.7653374671936035
    },
    {
      "epoch": 2.7025222778320313e-05,
      "model_forward_time": 1.7479491233825684,
      "step": 28338
    },
    {
      "epoch": 2.7025222778320313e-05,
      "step": 28338,
      "training_step_time": 8.457487106323242
    },
    {
      "epoch": 2.7026176452636717e-05,
      "model_forward_time": 1.7571544647216797,
      "step": 28339
    },
    {
      "epoch": 2.7026176452636717e-05,
      "step": 28339,
      "training_step_time": 7.798216342926025
    },
    {
      "epoch": 2.7027130126953125e-05,
      "grad_norm": 0.19630372524261475,
      "learning_rate": 8.58325915936441e-06,
      "loss": 0.0605,
      "step": 28340
    },
    {
      "epoch": 2.7027130126953125e-05,
      "model_forward_time": 2.9590706825256348,
      "step": 28340
    },
    {
      "dynamics_loss_avg": 0.06218602042645216,
      "epoch": 2.7027130126953125e-05,
      "step": 28340
    },
    {
      "action_loss_avg": 0.002053026505745947,
      "epoch": 2.7027130126953125e-05,
      "step": 28340
    },
    {
      "epoch": 2.7027130126953125e-05,
      "step": 28340,
      "training_step_time": 7.831353187561035
    },
    {
      "epoch": 2.7028083801269532e-05,
      "model_forward_time": 1.7481234073638916,
      "step": 28341
    },
    {
      "epoch": 2.7028083801269532e-05,
      "step": 28341,
      "training_step_time": 8.505075454711914
    },
    {
      "epoch": 2.7029037475585936e-05,
      "model_forward_time": 1.7525050640106201,
      "step": 28342
    },
    {
      "epoch": 2.7029037475585936e-05,
      "step": 28342,
      "training_step_time": 7.530643463134766
    },
    {
      "epoch": 2.7029991149902344e-05,
      "model_forward_time": 1.751692771911621,
      "step": 28343
    },
    {
      "epoch": 2.7029991149902344e-05,
      "step": 28343,
      "training_step_time": 7.769082546234131
    },
    {
      "epoch": 2.703094482421875e-05,
      "model_forward_time": 1.7498507499694824,
      "step": 28344
    },
    {
      "epoch": 2.703094482421875e-05,
      "step": 28344,
      "training_step_time": 7.685120344161987
    },
    {
      "epoch": 2.7031898498535155e-05,
      "model_forward_time": 1.74839186668396,
      "step": 28345
    },
    {
      "epoch": 2.7031898498535155e-05,
      "step": 28345,
      "training_step_time": 7.835176229476929
    },
    {
      "epoch": 2.7032852172851563e-05,
      "model_forward_time": 1.7421844005584717,
      "step": 28346
    },
    {
      "epoch": 2.7032852172851563e-05,
      "step": 28346,
      "training_step_time": 8.803173065185547
    },
    {
      "epoch": 2.703380584716797e-05,
      "model_forward_time": 1.7508516311645508,
      "step": 28347
    },
    {
      "epoch": 2.703380584716797e-05,
      "step": 28347,
      "training_step_time": 9.586714744567871
    },
    {
      "epoch": 2.7034759521484374e-05,
      "model_forward_time": 2.8964366912841797,
      "step": 28348
    },
    {
      "epoch": 2.7034759521484374e-05,
      "step": 28348,
      "training_step_time": 10.284000635147095
    },
    {
      "epoch": 2.703571319580078e-05,
      "model_forward_time": 1.7440273761749268,
      "step": 28349
    },
    {
      "epoch": 2.703571319580078e-05,
      "step": 28349,
      "training_step_time": 6.91273832321167
    },
    {
      "epoch": 2.703666687011719e-05,
      "grad_norm": 0.21000945568084717,
      "learning_rate": 8.582105782166784e-06,
      "loss": 0.0587,
      "step": 28350
    },
    {
      "epoch": 2.703666687011719e-05,
      "model_forward_time": 1.7394182682037354,
      "step": 28350
    },
    {
      "dynamics_loss_avg": 0.058012879081070425,
      "epoch": 2.703666687011719e-05,
      "step": 28350
    },
    {
      "action_loss_avg": 0.002253508218564093,
      "epoch": 2.703666687011719e-05,
      "step": 28350
    },
    {
      "epoch": 2.703666687011719e-05,
      "step": 28350,
      "training_step_time": 6.811794281005859
    },
    {
      "epoch": 2.7037620544433593e-05,
      "model_forward_time": 1.7415592670440674,
      "step": 28351
    },
    {
      "epoch": 2.7037620544433593e-05,
      "step": 28351,
      "training_step_time": 6.937283515930176
    },
    {
      "epoch": 2.703857421875e-05,
      "model_forward_time": 1.737797498703003,
      "step": 28352
    },
    {
      "epoch": 2.703857421875e-05,
      "step": 28352,
      "training_step_time": 6.846070051193237
    },
    {
      "epoch": 2.7039527893066408e-05,
      "model_forward_time": 1.7465829849243164,
      "step": 28353
    },
    {
      "epoch": 2.7039527893066408e-05,
      "step": 28353,
      "training_step_time": 6.954110383987427
    },
    {
      "epoch": 2.7040481567382812e-05,
      "model_forward_time": 1.741774320602417,
      "step": 28354
    },
    {
      "epoch": 2.7040481567382812e-05,
      "step": 28354,
      "training_step_time": 6.968319654464722
    },
    {
      "epoch": 2.704143524169922e-05,
      "model_forward_time": 1.7340967655181885,
      "step": 28355
    },
    {
      "epoch": 2.704143524169922e-05,
      "step": 28355,
      "training_step_time": 6.928027391433716
    },
    {
      "epoch": 2.7042388916015627e-05,
      "model_forward_time": 1.7500617504119873,
      "step": 28356
    },
    {
      "epoch": 2.7042388916015627e-05,
      "step": 28356,
      "training_step_time": 6.9603636264801025
    },
    {
      "epoch": 2.704334259033203e-05,
      "model_forward_time": 2.9513301849365234,
      "step": 28357
    },
    {
      "epoch": 2.704334259033203e-05,
      "step": 28357,
      "training_step_time": 9.965520858764648
    },
    {
      "epoch": 2.7044296264648438e-05,
      "model_forward_time": 1.7381856441497803,
      "step": 28358
    },
    {
      "epoch": 2.7044296264648438e-05,
      "step": 28358,
      "training_step_time": 7.723952531814575
    },
    {
      "epoch": 2.7045249938964845e-05,
      "model_forward_time": 1.7445569038391113,
      "step": 28359
    },
    {
      "epoch": 2.7045249938964845e-05,
      "step": 28359,
      "training_step_time": 6.811243534088135
    },
    {
      "epoch": 2.704620361328125e-05,
      "grad_norm": 0.263940691947937,
      "learning_rate": 8.580952013235456e-06,
      "loss": 0.0619,
      "step": 28360
    },
    {
      "epoch": 2.704620361328125e-05,
      "model_forward_time": 1.747988224029541,
      "step": 28360
    },
    {
      "dynamics_loss_avg": 0.06082594282925129,
      "epoch": 2.704620361328125e-05,
      "step": 28360
    },
    {
      "action_loss_avg": 0.002912213880335912,
      "epoch": 2.704620361328125e-05,
      "step": 28360
    },
    {
      "epoch": 2.704620361328125e-05,
      "step": 28360,
      "training_step_time": 6.910602569580078
    },
    {
      "epoch": 2.7047157287597657e-05,
      "model_forward_time": 1.7441823482513428,
      "step": 28361
    },
    {
      "epoch": 2.7047157287597657e-05,
      "step": 28361,
      "training_step_time": 6.868935585021973
    },
    {
      "epoch": 2.704811096191406e-05,
      "model_forward_time": 1.7414405345916748,
      "step": 28362
    },
    {
      "epoch": 2.704811096191406e-05,
      "step": 28362,
      "training_step_time": 6.907376050949097
    },
    {
      "epoch": 2.704906463623047e-05,
      "model_forward_time": 1.7378172874450684,
      "step": 28363
    },
    {
      "epoch": 2.704906463623047e-05,
      "step": 28363,
      "training_step_time": 6.865837812423706
    },
    {
      "epoch": 2.7050018310546876e-05,
      "model_forward_time": 1.7465121746063232,
      "step": 28364
    },
    {
      "epoch": 2.7050018310546876e-05,
      "step": 28364,
      "training_step_time": 6.85823917388916
    },
    {
      "epoch": 2.705097198486328e-05,
      "model_forward_time": 1.7477447986602783,
      "step": 28365
    },
    {
      "epoch": 2.705097198486328e-05,
      "step": 28365,
      "training_step_time": 6.834102630615234
    },
    {
      "epoch": 2.7051925659179687e-05,
      "model_forward_time": 2.5045225620269775,
      "step": 28366
    },
    {
      "epoch": 2.7051925659179687e-05,
      "step": 28366,
      "training_step_time": 7.230213403701782
    },
    {
      "epoch": 2.7052879333496095e-05,
      "model_forward_time": 1.740943193435669,
      "step": 28367
    },
    {
      "epoch": 2.7052879333496095e-05,
      "step": 28367,
      "training_step_time": 7.727489233016968
    },
    {
      "epoch": 2.70538330078125e-05,
      "model_forward_time": 1.7548911571502686,
      "step": 28368
    },
    {
      "epoch": 2.70538330078125e-05,
      "step": 28368,
      "training_step_time": 6.920908689498901
    },
    {
      "epoch": 2.7054786682128906e-05,
      "model_forward_time": 1.743776798248291,
      "step": 28369
    },
    {
      "epoch": 2.7054786682128906e-05,
      "step": 28369,
      "training_step_time": 6.872037887573242
    },
    {
      "epoch": 2.7055740356445313e-05,
      "grad_norm": 0.20959191024303436,
      "learning_rate": 8.579797852696596e-06,
      "loss": 0.0619,
      "step": 28370
    },
    {
      "epoch": 2.7055740356445313e-05,
      "model_forward_time": 1.7383098602294922,
      "step": 28370
    },
    {
      "dynamics_loss_avg": 0.05522266272455454,
      "epoch": 2.7055740356445313e-05,
      "step": 28370
    },
    {
      "action_loss_avg": 0.0025447876541875303,
      "epoch": 2.7055740356445313e-05,
      "step": 28370
    },
    {
      "epoch": 2.7055740356445313e-05,
      "step": 28370,
      "training_step_time": 6.912755727767944
    },
    {
      "epoch": 2.7056694030761717e-05,
      "model_forward_time": 1.7358717918395996,
      "step": 28371
    },
    {
      "epoch": 2.7056694030761717e-05,
      "step": 28371,
      "training_step_time": 6.8716065883636475
    },
    {
      "epoch": 2.7057647705078125e-05,
      "model_forward_time": 1.7426090240478516,
      "step": 28372
    },
    {
      "epoch": 2.7057647705078125e-05,
      "step": 28372,
      "training_step_time": 6.876060724258423
    },
    {
      "epoch": 2.7058601379394532e-05,
      "model_forward_time": 1.7623310089111328,
      "step": 28373
    },
    {
      "epoch": 2.7058601379394532e-05,
      "step": 28373,
      "training_step_time": 6.862882614135742
    },
    {
      "epoch": 2.7059555053710936e-05,
      "model_forward_time": 1.7371344566345215,
      "step": 28374
    },
    {
      "epoch": 2.7059555053710936e-05,
      "step": 28374,
      "training_step_time": 6.866623878479004
    },
    {
      "epoch": 2.7060508728027344e-05,
      "model_forward_time": 2.8810150623321533,
      "step": 28375
    },
    {
      "epoch": 2.7060508728027344e-05,
      "step": 28375,
      "training_step_time": 7.627654552459717
    },
    {
      "epoch": 2.706146240234375e-05,
      "model_forward_time": 1.745154857635498,
      "step": 28376
    },
    {
      "epoch": 2.706146240234375e-05,
      "step": 28376,
      "training_step_time": 8.661460876464844
    },
    {
      "epoch": 2.7062416076660155e-05,
      "model_forward_time": 1.7629520893096924,
      "step": 28377
    },
    {
      "epoch": 2.7062416076660155e-05,
      "step": 28377,
      "training_step_time": 7.674360275268555
    },
    {
      "epoch": 2.7063369750976563e-05,
      "model_forward_time": 1.7393441200256348,
      "step": 28378
    },
    {
      "epoch": 2.7063369750976563e-05,
      "step": 28378,
      "training_step_time": 6.877151727676392
    },
    {
      "epoch": 2.706432342529297e-05,
      "model_forward_time": 1.7441751956939697,
      "step": 28379
    },
    {
      "epoch": 2.706432342529297e-05,
      "step": 28379,
      "training_step_time": 6.877064943313599
    },
    {
      "epoch": 2.7065277099609374e-05,
      "grad_norm": 0.1801327019929886,
      "learning_rate": 8.578643300676429e-06,
      "loss": 0.0607,
      "step": 28380
    },
    {
      "epoch": 2.7065277099609374e-05,
      "model_forward_time": 1.744706153869629,
      "step": 28380
    },
    {
      "dynamics_loss_avg": 0.07880077138543129,
      "epoch": 2.7065277099609374e-05,
      "step": 28380
    },
    {
      "action_loss_avg": 0.002830775664187968,
      "epoch": 2.7065277099609374e-05,
      "step": 28380
    },
    {
      "epoch": 2.7065277099609374e-05,
      "step": 28380,
      "training_step_time": 6.862095355987549
    },
    {
      "epoch": 2.706623077392578e-05,
      "model_forward_time": 1.7464187145233154,
      "step": 28381
    },
    {
      "epoch": 2.706623077392578e-05,
      "step": 28381,
      "training_step_time": 6.832053899765015
    },
    {
      "epoch": 2.706718444824219e-05,
      "model_forward_time": 1.7465662956237793,
      "step": 28382
    },
    {
      "epoch": 2.706718444824219e-05,
      "step": 28382,
      "training_step_time": 6.86353325843811
    },
    {
      "epoch": 2.7068138122558593e-05,
      "model_forward_time": 1.7407305240631104,
      "step": 28383
    },
    {
      "epoch": 2.7068138122558593e-05,
      "step": 28383,
      "training_step_time": 6.812004089355469
    },
    {
      "epoch": 2.7069091796875e-05,
      "model_forward_time": 1.7388291358947754,
      "step": 28384
    },
    {
      "epoch": 2.7069091796875e-05,
      "step": 28384,
      "training_step_time": 8.330553770065308
    },
    {
      "epoch": 2.7070045471191408e-05,
      "model_forward_time": 1.8118762969970703,
      "step": 28385
    },
    {
      "epoch": 2.7070045471191408e-05,
      "step": 28385,
      "training_step_time": 9.77034854888916
    },
    {
      "epoch": 2.7070999145507812e-05,
      "model_forward_time": 1.7383437156677246,
      "step": 28386
    },
    {
      "epoch": 2.7070999145507812e-05,
      "step": 28386,
      "training_step_time": 7.684262037277222
    },
    {
      "epoch": 2.707195281982422e-05,
      "model_forward_time": 1.736731767654419,
      "step": 28387
    },
    {
      "epoch": 2.707195281982422e-05,
      "step": 28387,
      "training_step_time": 6.8205695152282715
    },
    {
      "epoch": 2.7072906494140627e-05,
      "model_forward_time": 1.7355639934539795,
      "step": 28388
    },
    {
      "epoch": 2.7072906494140627e-05,
      "step": 28388,
      "training_step_time": 6.872948884963989
    },
    {
      "epoch": 2.707386016845703e-05,
      "model_forward_time": 1.7735424041748047,
      "step": 28389
    },
    {
      "epoch": 2.707386016845703e-05,
      "step": 28389,
      "training_step_time": 6.7795960903167725
    },
    {
      "epoch": 2.7074813842773438e-05,
      "grad_norm": 0.22666582465171814,
      "learning_rate": 8.577488357301209e-06,
      "loss": 0.0625,
      "step": 28390
    },
    {
      "epoch": 2.7074813842773438e-05,
      "model_forward_time": 1.7510926723480225,
      "step": 28390
    },
    {
      "dynamics_loss_avg": 0.06912083812057972,
      "epoch": 2.7074813842773438e-05,
      "step": 28390
    },
    {
      "action_loss_avg": 0.0023304757312871517,
      "epoch": 2.7074813842773438e-05,
      "step": 28390
    },
    {
      "epoch": 2.7074813842773438e-05,
      "step": 28390,
      "training_step_time": 6.819491386413574
    },
    {
      "epoch": 2.7075767517089842e-05,
      "model_forward_time": 1.7436902523040771,
      "step": 28391
    },
    {
      "epoch": 2.7075767517089842e-05,
      "step": 28391,
      "training_step_time": 6.82572340965271
    },
    {
      "epoch": 2.707672119140625e-05,
      "model_forward_time": 1.7410457134246826,
      "step": 28392
    },
    {
      "epoch": 2.707672119140625e-05,
      "step": 28392,
      "training_step_time": 6.791833877563477
    },
    {
      "epoch": 2.7077674865722657e-05,
      "model_forward_time": 1.7467527389526367,
      "step": 28393
    },
    {
      "epoch": 2.7077674865722657e-05,
      "step": 28393,
      "training_step_time": 6.764404535293579
    },
    {
      "epoch": 2.707862854003906e-05,
      "model_forward_time": 2.968242883682251,
      "step": 28394
    },
    {
      "epoch": 2.707862854003906e-05,
      "step": 28394,
      "training_step_time": 10.270118236541748
    },
    {
      "epoch": 2.707958221435547e-05,
      "model_forward_time": 1.7479944229125977,
      "step": 28395
    },
    {
      "epoch": 2.707958221435547e-05,
      "step": 28395,
      "training_step_time": 8.210335731506348
    },
    {
      "epoch": 2.7080535888671876e-05,
      "model_forward_time": 1.7333073616027832,
      "step": 28396
    },
    {
      "epoch": 2.7080535888671876e-05,
      "step": 28396,
      "training_step_time": 6.799253463745117
    },
    {
      "epoch": 2.708148956298828e-05,
      "model_forward_time": 1.7427115440368652,
      "step": 28397
    },
    {
      "epoch": 2.708148956298828e-05,
      "step": 28397,
      "training_step_time": 6.775734901428223
    },
    {
      "epoch": 2.7082443237304687e-05,
      "model_forward_time": 1.811462640762329,
      "step": 28398
    },
    {
      "epoch": 2.7082443237304687e-05,
      "step": 28398,
      "training_step_time": 6.771886587142944
    },
    {
      "epoch": 2.7083396911621095e-05,
      "model_forward_time": 1.7399346828460693,
      "step": 28399
    },
    {
      "epoch": 2.7083396911621095e-05,
      "step": 28399,
      "training_step_time": 6.793828248977661
    },
    {
      "epoch": 2.70843505859375e-05,
      "grad_norm": 0.22249875962734222,
      "learning_rate": 8.576333022697244e-06,
      "loss": 0.063,
      "step": 28400
    },
    {
      "epoch": 2.70843505859375e-05,
      "model_forward_time": 1.748413324356079,
      "step": 28400
    },
    {
      "dynamics_loss_avg": 0.05580791523680091,
      "epoch": 2.70843505859375e-05,
      "step": 28400
    },
    {
      "action_loss_avg": 0.001819643733324483,
      "epoch": 2.70843505859375e-05,
      "step": 28400
    },
    {
      "epoch": 2.70843505859375e-05,
      "step": 28400,
      "training_step_time": 6.780614614486694
    },
    {
      "epoch": 2.7085304260253906e-05,
      "model_forward_time": 1.7484889030456543,
      "step": 28401
    },
    {
      "epoch": 2.7085304260253906e-05,
      "step": 28401,
      "training_step_time": 6.796848773956299
    },
    {
      "epoch": 2.7086257934570313e-05,
      "model_forward_time": 1.743407964706421,
      "step": 28402
    },
    {
      "epoch": 2.7086257934570313e-05,
      "step": 28402,
      "training_step_time": 6.8254289627075195
    },
    {
      "epoch": 2.7087211608886718e-05,
      "model_forward_time": 2.8870415687561035,
      "step": 28403
    },
    {
      "epoch": 2.7087211608886718e-05,
      "step": 28403,
      "training_step_time": 8.204485177993774
    },
    {
      "epoch": 2.7088165283203125e-05,
      "model_forward_time": 1.7996628284454346,
      "step": 28404
    },
    {
      "epoch": 2.7088165283203125e-05,
      "step": 28404,
      "training_step_time": 7.691575288772583
    },
    {
      "epoch": 2.7089118957519532e-05,
      "model_forward_time": 1.7383978366851807,
      "step": 28405
    },
    {
      "epoch": 2.7089118957519532e-05,
      "step": 28405,
      "training_step_time": 7.715478897094727
    },
    {
      "epoch": 2.7090072631835936e-05,
      "model_forward_time": 1.738487720489502,
      "step": 28406
    },
    {
      "epoch": 2.7090072631835936e-05,
      "step": 28406,
      "training_step_time": 6.818852424621582
    },
    {
      "epoch": 2.7091026306152344e-05,
      "model_forward_time": 1.741426944732666,
      "step": 28407
    },
    {
      "epoch": 2.7091026306152344e-05,
      "step": 28407,
      "training_step_time": 6.780324459075928
    },
    {
      "epoch": 2.709197998046875e-05,
      "model_forward_time": 1.7364954948425293,
      "step": 28408
    },
    {
      "epoch": 2.709197998046875e-05,
      "step": 28408,
      "training_step_time": 6.783787250518799
    },
    {
      "epoch": 2.7092933654785155e-05,
      "model_forward_time": 1.7410576343536377,
      "step": 28409
    },
    {
      "epoch": 2.7092933654785155e-05,
      "step": 28409,
      "training_step_time": 6.820619821548462
    },
    {
      "epoch": 2.7093887329101563e-05,
      "grad_norm": 0.1580856740474701,
      "learning_rate": 8.575177296990874e-06,
      "loss": 0.0619,
      "step": 28410
    },
    {
      "epoch": 2.7093887329101563e-05,
      "model_forward_time": 1.7403035163879395,
      "step": 28410
    },
    {
      "dynamics_loss_avg": 0.04203309351578355,
      "epoch": 2.7093887329101563e-05,
      "step": 28410
    },
    {
      "action_loss_avg": 0.0016004467266611754,
      "epoch": 2.7093887329101563e-05,
      "step": 28410
    },
    {
      "epoch": 2.7093887329101563e-05,
      "step": 28410,
      "training_step_time": 6.816651821136475
    },
    {
      "epoch": 2.709484100341797e-05,
      "model_forward_time": 1.7353639602661133,
      "step": 28411
    },
    {
      "epoch": 2.709484100341797e-05,
      "step": 28411,
      "training_step_time": 6.842352867126465
    },
    {
      "epoch": 2.7095794677734374e-05,
      "model_forward_time": 2.9429357051849365,
      "step": 28412
    },
    {
      "epoch": 2.7095794677734374e-05,
      "step": 28412,
      "training_step_time": 7.714171886444092
    },
    {
      "epoch": 2.709674835205078e-05,
      "model_forward_time": 1.8084022998809814,
      "step": 28413
    },
    {
      "epoch": 2.709674835205078e-05,
      "step": 28413,
      "training_step_time": 9.325579404830933
    },
    {
      "epoch": 2.709770202636719e-05,
      "model_forward_time": 1.7599194049835205,
      "step": 28414
    },
    {
      "epoch": 2.709770202636719e-05,
      "step": 28414,
      "training_step_time": 7.665140628814697
    },
    {
      "epoch": 2.7098655700683593e-05,
      "model_forward_time": 1.7334976196289062,
      "step": 28415
    },
    {
      "epoch": 2.7098655700683593e-05,
      "step": 28415,
      "training_step_time": 6.842429876327515
    },
    {
      "epoch": 2.7099609375e-05,
      "model_forward_time": 1.3789663314819336,
      "step": 28416
    },
    {
      "epoch": 2.7099609375e-05,
      "step": 28416,
      "training_step_time": 6.8557257652282715
    },
    {
      "epoch": 2.7100563049316408e-05,
      "model_forward_time": 1.7403841018676758,
      "step": 28417
    },
    {
      "epoch": 2.7100563049316408e-05,
      "step": 28417,
      "training_step_time": 6.842265605926514
    },
    {
      "epoch": 2.7101516723632812e-05,
      "model_forward_time": 1.3718688488006592,
      "step": 28418
    },
    {
      "epoch": 2.7101516723632812e-05,
      "step": 28418,
      "training_step_time": 6.826082229614258
    },
    {
      "epoch": 2.710247039794922e-05,
      "model_forward_time": 1.7409074306488037,
      "step": 28419
    },
    {
      "epoch": 2.710247039794922e-05,
      "step": 28419,
      "training_step_time": 6.796586275100708
    },
    {
      "epoch": 2.7103424072265627e-05,
      "grad_norm": 0.15569981932640076,
      "learning_rate": 8.57402118030849e-06,
      "loss": 0.0634,
      "step": 28420
    },
    {
      "epoch": 2.7103424072265627e-05,
      "model_forward_time": 1.745096206665039,
      "step": 28420
    },
    {
      "dynamics_loss_avg": 0.061016672756522894,
      "epoch": 2.7103424072265627e-05,
      "step": 28420
    },
    {
      "action_loss_avg": 0.002384119655471295,
      "epoch": 2.7103424072265627e-05,
      "step": 28420
    },
    {
      "epoch": 2.7103424072265627e-05,
      "step": 28420,
      "training_step_time": 6.860999584197998
    },
    {
      "epoch": 2.710437774658203e-05,
      "model_forward_time": 2.8944952487945557,
      "step": 28421
    },
    {
      "epoch": 2.710437774658203e-05,
      "step": 28421,
      "training_step_time": 8.661571264266968
    },
    {
      "epoch": 2.7105331420898438e-05,
      "model_forward_time": 1.791581630706787,
      "step": 28422
    },
    {
      "epoch": 2.7105331420898438e-05,
      "step": 28422,
      "training_step_time": 8.783539056777954
    },
    {
      "epoch": 2.7106285095214842e-05,
      "model_forward_time": 1.744504451751709,
      "step": 28423
    },
    {
      "epoch": 2.7106285095214842e-05,
      "step": 28423,
      "training_step_time": 7.627561330795288
    },
    {
      "epoch": 2.710723876953125e-05,
      "model_forward_time": 1.7445580959320068,
      "step": 28424
    },
    {
      "epoch": 2.710723876953125e-05,
      "step": 28424,
      "training_step_time": 7.618624925613403
    },
    {
      "epoch": 2.7108192443847657e-05,
      "model_forward_time": 1.7389891147613525,
      "step": 28425
    },
    {
      "epoch": 2.7108192443847657e-05,
      "step": 28425,
      "training_step_time": 6.7960286140441895
    },
    {
      "epoch": 2.710914611816406e-05,
      "model_forward_time": 1.736924648284912,
      "step": 28426
    },
    {
      "epoch": 2.710914611816406e-05,
      "step": 28426,
      "training_step_time": 6.772446393966675
    },
    {
      "epoch": 2.711009979248047e-05,
      "model_forward_time": 1.7397031784057617,
      "step": 28427
    },
    {
      "epoch": 2.711009979248047e-05,
      "step": 28427,
      "training_step_time": 6.789661169052124
    },
    {
      "epoch": 2.7111053466796876e-05,
      "model_forward_time": 1.7401502132415771,
      "step": 28428
    },
    {
      "epoch": 2.7111053466796876e-05,
      "step": 28428,
      "training_step_time": 6.848795175552368
    },
    {
      "epoch": 2.711200714111328e-05,
      "model_forward_time": 1.7386808395385742,
      "step": 28429
    },
    {
      "epoch": 2.711200714111328e-05,
      "step": 28429,
      "training_step_time": 6.839853525161743
    },
    {
      "epoch": 2.7112960815429687e-05,
      "grad_norm": 0.15971267223358154,
      "learning_rate": 8.572864672776524e-06,
      "loss": 0.0638,
      "step": 28430
    },
    {
      "epoch": 2.7112960815429687e-05,
      "model_forward_time": 2.8698816299438477,
      "step": 28430
    },
    {
      "dynamics_loss_avg": 0.0648605264723301,
      "epoch": 2.7112960815429687e-05,
      "step": 28430
    },
    {
      "action_loss_avg": 0.0020905482117086647,
      "epoch": 2.7112960815429687e-05,
      "step": 28430
    },
    {
      "epoch": 2.7112960815429687e-05,
      "step": 28430,
      "training_step_time": 7.681220054626465
    },
    {
      "epoch": 2.7113914489746095e-05,
      "model_forward_time": 1.7613804340362549,
      "step": 28431
    },
    {
      "epoch": 2.7113914489746095e-05,
      "step": 28431,
      "training_step_time": 8.583687543869019
    },
    {
      "epoch": 2.71148681640625e-05,
      "model_forward_time": 1.7364602088928223,
      "step": 28432
    },
    {
      "epoch": 2.71148681640625e-05,
      "step": 28432,
      "training_step_time": 8.268383979797363
    },
    {
      "epoch": 2.7115821838378906e-05,
      "model_forward_time": 1.7110295295715332,
      "step": 28433
    },
    {
      "epoch": 2.7115821838378906e-05,
      "step": 28433,
      "training_step_time": 6.835714101791382
    },
    {
      "epoch": 2.7116775512695314e-05,
      "model_forward_time": 1.7327661514282227,
      "step": 28434
    },
    {
      "epoch": 2.7116775512695314e-05,
      "step": 28434,
      "training_step_time": 6.811308145523071
    },
    {
      "epoch": 2.7117729187011718e-05,
      "model_forward_time": 1.741112470626831,
      "step": 28435
    },
    {
      "epoch": 2.7117729187011718e-05,
      "step": 28435,
      "training_step_time": 6.862843751907349
    },
    {
      "epoch": 2.7118682861328125e-05,
      "model_forward_time": 1.7443740367889404,
      "step": 28436
    },
    {
      "epoch": 2.7118682861328125e-05,
      "step": 28436,
      "training_step_time": 6.8533406257629395
    },
    {
      "epoch": 2.7119636535644532e-05,
      "model_forward_time": 1.7314379215240479,
      "step": 28437
    },
    {
      "epoch": 2.7119636535644532e-05,
      "step": 28437,
      "training_step_time": 6.875513076782227
    },
    {
      "epoch": 2.7120590209960936e-05,
      "model_forward_time": 1.7197234630584717,
      "step": 28438
    },
    {
      "epoch": 2.7120590209960936e-05,
      "step": 28438,
      "training_step_time": 6.870943307876587
    },
    {
      "epoch": 2.7121543884277344e-05,
      "model_forward_time": 2.8283116817474365,
      "step": 28439
    },
    {
      "epoch": 2.7121543884277344e-05,
      "step": 28439,
      "training_step_time": 7.697232484817505
    },
    {
      "epoch": 2.712249755859375e-05,
      "grad_norm": 0.12095106393098831,
      "learning_rate": 8.571707774521447e-06,
      "loss": 0.0661,
      "step": 28440
    },
    {
      "epoch": 2.712249755859375e-05,
      "model_forward_time": 1.7077088356018066,
      "step": 28440
    },
    {
      "dynamics_loss_avg": 0.05340702654793859,
      "epoch": 2.712249755859375e-05,
      "step": 28440
    },
    {
      "action_loss_avg": 0.0014829534862656145,
      "epoch": 2.712249755859375e-05,
      "step": 28440
    },
    {
      "epoch": 2.712249755859375e-05,
      "step": 28440,
      "training_step_time": 11.624521732330322
    },
    {
      "epoch": 2.7123451232910155e-05,
      "model_forward_time": 1.4560182094573975,
      "step": 28441
    },
    {
      "epoch": 2.7123451232910155e-05,
      "step": 28441,
      "training_step_time": 8.677346467971802
    },
    {
      "epoch": 2.7124404907226563e-05,
      "model_forward_time": 1.7305090427398682,
      "step": 28442
    },
    {
      "epoch": 2.7124404907226563e-05,
      "step": 28442,
      "training_step_time": 7.199024200439453
    },
    {
      "epoch": 2.712535858154297e-05,
      "model_forward_time": 1.730393409729004,
      "step": 28443
    },
    {
      "epoch": 2.712535858154297e-05,
      "step": 28443,
      "training_step_time": 7.6040940284729
    },
    {
      "epoch": 2.7126312255859374e-05,
      "model_forward_time": 1.7309257984161377,
      "step": 28444
    },
    {
      "epoch": 2.7126312255859374e-05,
      "step": 28444,
      "training_step_time": 6.853123664855957
    },
    {
      "epoch": 2.712726593017578e-05,
      "model_forward_time": 1.7269394397735596,
      "step": 28445
    },
    {
      "epoch": 2.712726593017578e-05,
      "step": 28445,
      "training_step_time": 6.85381555557251
    },
    {
      "epoch": 2.712821960449219e-05,
      "model_forward_time": 1.7335608005523682,
      "step": 28446
    },
    {
      "epoch": 2.712821960449219e-05,
      "step": 28446,
      "training_step_time": 6.813436985015869
    },
    {
      "epoch": 2.7129173278808593e-05,
      "model_forward_time": 1.7287561893463135,
      "step": 28447
    },
    {
      "epoch": 2.7129173278808593e-05,
      "step": 28447,
      "training_step_time": 6.800284147262573
    },
    {
      "epoch": 2.7130126953125e-05,
      "model_forward_time": 1.7317936420440674,
      "step": 28448
    },
    {
      "epoch": 2.7130126953125e-05,
      "step": 28448,
      "training_step_time": 7.621313095092773
    },
    {
      "epoch": 2.7131080627441408e-05,
      "model_forward_time": 2.89521861076355,
      "step": 28449
    },
    {
      "epoch": 2.7131080627441408e-05,
      "step": 28449,
      "training_step_time": 11.621633291244507
    },
    {
      "epoch": 2.7132034301757812e-05,
      "grad_norm": 0.1651238203048706,
      "learning_rate": 8.57055048566978e-06,
      "loss": 0.0595,
      "step": 28450
    },
    {
      "epoch": 2.7132034301757812e-05,
      "model_forward_time": 1.7985622882843018,
      "step": 28450
    },
    {
      "dynamics_loss_avg": 0.06424378417432308,
      "epoch": 2.7132034301757812e-05,
      "step": 28450
    },
    {
      "action_loss_avg": 0.0028524849331006408,
      "epoch": 2.7132034301757812e-05,
      "step": 28450
    },
    {
      "epoch": 2.7132034301757812e-05,
      "step": 28450,
      "training_step_time": 9.45601487159729
    },
    {
      "epoch": 2.713298797607422e-05,
      "model_forward_time": 1.7345950603485107,
      "step": 28451
    },
    {
      "epoch": 2.713298797607422e-05,
      "step": 28451,
      "training_step_time": 7.6789727210998535
    },
    {
      "epoch": 2.7133941650390627e-05,
      "model_forward_time": 1.7308845520019531,
      "step": 28452
    },
    {
      "epoch": 2.7133941650390627e-05,
      "step": 28452,
      "training_step_time": 7.566030979156494
    },
    {
      "epoch": 2.713489532470703e-05,
      "model_forward_time": 1.730402946472168,
      "step": 28453
    },
    {
      "epoch": 2.713489532470703e-05,
      "step": 28453,
      "training_step_time": 6.791367530822754
    },
    {
      "epoch": 2.7135848999023438e-05,
      "model_forward_time": 1.7297017574310303,
      "step": 28454
    },
    {
      "epoch": 2.7135848999023438e-05,
      "step": 28454,
      "training_step_time": 6.790514707565308
    },
    {
      "epoch": 2.7136802673339842e-05,
      "model_forward_time": 1.7358872890472412,
      "step": 28455
    },
    {
      "epoch": 2.7136802673339842e-05,
      "step": 28455,
      "training_step_time": 6.827126741409302
    },
    {
      "epoch": 2.713775634765625e-05,
      "model_forward_time": 1.7332911491394043,
      "step": 28456
    },
    {
      "epoch": 2.713775634765625e-05,
      "step": 28456,
      "training_step_time": 6.7902610301971436
    },
    {
      "epoch": 2.7138710021972657e-05,
      "model_forward_time": 1.7335584163665771,
      "step": 28457
    },
    {
      "epoch": 2.7138710021972657e-05,
      "step": 28457,
      "training_step_time": 7.635180234909058
    },
    {
      "epoch": 2.713966369628906e-05,
      "model_forward_time": 2.8890178203582764,
      "step": 28458
    },
    {
      "epoch": 2.713966369628906e-05,
      "step": 28458,
      "training_step_time": 8.713330030441284
    },
    {
      "epoch": 2.714061737060547e-05,
      "model_forward_time": 1.7912113666534424,
      "step": 28459
    },
    {
      "epoch": 2.714061737060547e-05,
      "step": 28459,
      "training_step_time": 10.530050277709961
    },
    {
      "epoch": 2.7141571044921876e-05,
      "grad_norm": 0.15744292736053467,
      "learning_rate": 8.569392806348079e-06,
      "loss": 0.0597,
      "step": 28460
    },
    {
      "epoch": 2.7141571044921876e-05,
      "model_forward_time": 1.7110364437103271,
      "step": 28460
    },
    {
      "dynamics_loss_avg": 0.06586714405566454,
      "epoch": 2.7141571044921876e-05,
      "step": 28460
    },
    {
      "action_loss_avg": 0.0021034836187027394,
      "epoch": 2.7141571044921876e-05,
      "step": 28460
    },
    {
      "epoch": 2.7141571044921876e-05,
      "step": 28460,
      "training_step_time": 7.623173236846924
    },
    {
      "epoch": 2.714252471923828e-05,
      "model_forward_time": 1.7286055088043213,
      "step": 28461
    },
    {
      "epoch": 2.714252471923828e-05,
      "step": 28461,
      "training_step_time": 7.571640491485596
    },
    {
      "epoch": 2.7143478393554687e-05,
      "model_forward_time": 1.793372631072998,
      "step": 28462
    },
    {
      "epoch": 2.7143478393554687e-05,
      "step": 28462,
      "training_step_time": 6.820321321487427
    },
    {
      "epoch": 2.7144432067871095e-05,
      "model_forward_time": 1.7251529693603516,
      "step": 28463
    },
    {
      "epoch": 2.7144432067871095e-05,
      "step": 28463,
      "training_step_time": 6.830095052719116
    },
    {
      "epoch": 2.71453857421875e-05,
      "model_forward_time": 1.7317891120910645,
      "step": 28464
    },
    {
      "epoch": 2.71453857421875e-05,
      "step": 28464,
      "training_step_time": 6.850400924682617
    },
    {
      "epoch": 2.7146339416503906e-05,
      "model_forward_time": 1.7324872016906738,
      "step": 28465
    },
    {
      "epoch": 2.7146339416503906e-05,
      "step": 28465,
      "training_step_time": 6.856337308883667
    },
    {
      "epoch": 2.7147293090820314e-05,
      "model_forward_time": 1.729322910308838,
      "step": 28466
    },
    {
      "epoch": 2.7147293090820314e-05,
      "step": 28466,
      "training_step_time": 7.660104751586914
    },
    {
      "epoch": 2.7148246765136718e-05,
      "model_forward_time": 2.8487138748168945,
      "step": 28467
    },
    {
      "epoch": 2.7148246765136718e-05,
      "step": 28467,
      "training_step_time": 8.680912017822266
    },
    {
      "epoch": 2.7149200439453125e-05,
      "model_forward_time": 1.7244188785552979,
      "step": 28468
    },
    {
      "epoch": 2.7149200439453125e-05,
      "step": 28468,
      "training_step_time": 9.79163932800293
    },
    {
      "epoch": 2.7150154113769532e-05,
      "model_forward_time": 1.7279808521270752,
      "step": 28469
    },
    {
      "epoch": 2.7150154113769532e-05,
      "step": 28469,
      "training_step_time": 7.642155885696411
    },
    {
      "epoch": 2.7151107788085936e-05,
      "grad_norm": 0.14420713484287262,
      "learning_rate": 8.568234736682948e-06,
      "loss": 0.0633,
      "step": 28470
    },
    {
      "epoch": 2.7151107788085936e-05,
      "model_forward_time": 1.7281107902526855,
      "step": 28470
    },
    {
      "dynamics_loss_avg": 0.06893339194357395,
      "epoch": 2.7151107788085936e-05,
      "step": 28470
    },
    {
      "action_loss_avg": 0.001974350231466815,
      "epoch": 2.7151107788085936e-05,
      "step": 28470
    },
    {
      "epoch": 2.7151107788085936e-05,
      "step": 28470,
      "training_step_time": 8.059684991836548
    },
    {
      "epoch": 2.7152061462402344e-05,
      "model_forward_time": 1.7288529872894287,
      "step": 28471
    },
    {
      "epoch": 2.7152061462402344e-05,
      "step": 28471,
      "training_step_time": 6.806538343429565
    },
    {
      "epoch": 2.715301513671875e-05,
      "model_forward_time": 1.7266051769256592,
      "step": 28472
    },
    {
      "epoch": 2.715301513671875e-05,
      "step": 28472,
      "training_step_time": 6.794668197631836
    },
    {
      "epoch": 2.7153968811035155e-05,
      "model_forward_time": 1.7207748889923096,
      "step": 28473
    },
    {
      "epoch": 2.7153968811035155e-05,
      "step": 28473,
      "training_step_time": 6.827930450439453
    },
    {
      "epoch": 2.7154922485351563e-05,
      "model_forward_time": 1.7355625629425049,
      "step": 28474
    },
    {
      "epoch": 2.7154922485351563e-05,
      "step": 28474,
      "training_step_time": 6.805022478103638
    },
    {
      "epoch": 2.715587615966797e-05,
      "model_forward_time": 1.730724573135376,
      "step": 28475
    },
    {
      "epoch": 2.715587615966797e-05,
      "step": 28475,
      "training_step_time": 7.631443738937378
    },
    {
      "epoch": 2.7156829833984374e-05,
      "model_forward_time": 2.823972463607788,
      "step": 28476
    },
    {
      "epoch": 2.7156829833984374e-05,
      "step": 28476,
      "training_step_time": 9.464622259140015
    },
    {
      "epoch": 2.715778350830078e-05,
      "model_forward_time": 1.7270007133483887,
      "step": 28477
    },
    {
      "epoch": 2.715778350830078e-05,
      "step": 28477,
      "training_step_time": 9.757477283477783
    },
    {
      "epoch": 2.715873718261719e-05,
      "model_forward_time": 1.7754414081573486,
      "step": 28478
    },
    {
      "epoch": 2.715873718261719e-05,
      "step": 28478,
      "training_step_time": 8.521931886672974
    },
    {
      "epoch": 2.7159690856933593e-05,
      "model_forward_time": 1.7287063598632812,
      "step": 28479
    },
    {
      "epoch": 2.7159690856933593e-05,
      "step": 28479,
      "training_step_time": 8.008152484893799
    },
    {
      "epoch": 2.716064453125e-05,
      "grad_norm": 0.15002648532390594,
      "learning_rate": 8.56707627680103e-06,
      "loss": 0.0588,
      "step": 28480
    },
    {
      "epoch": 2.716064453125e-05,
      "model_forward_time": 1.7229084968566895,
      "step": 28480
    },
    {
      "dynamics_loss_avg": 0.059071360249072315,
      "epoch": 2.716064453125e-05,
      "step": 28480
    },
    {
      "action_loss_avg": 0.0021798893401864918,
      "epoch": 2.716064453125e-05,
      "step": 28480
    },
    {
      "epoch": 2.716064453125e-05,
      "step": 28480,
      "training_step_time": 6.806392669677734
    },
    {
      "epoch": 2.7161598205566408e-05,
      "model_forward_time": 1.7350492477416992,
      "step": 28481
    },
    {
      "epoch": 2.7161598205566408e-05,
      "step": 28481,
      "training_step_time": 6.7830798625946045
    },
    {
      "epoch": 2.7162551879882812e-05,
      "model_forward_time": 1.7293713092803955,
      "step": 28482
    },
    {
      "epoch": 2.7162551879882812e-05,
      "step": 28482,
      "training_step_time": 6.815353870391846
    },
    {
      "epoch": 2.716350555419922e-05,
      "model_forward_time": 1.7249529361724854,
      "step": 28483
    },
    {
      "epoch": 2.716350555419922e-05,
      "step": 28483,
      "training_step_time": 6.761990547180176
    },
    {
      "epoch": 2.7164459228515627e-05,
      "model_forward_time": 1.737666368484497,
      "step": 28484
    },
    {
      "epoch": 2.7164459228515627e-05,
      "step": 28484,
      "training_step_time": 7.68405818939209
    },
    {
      "epoch": 2.716541290283203e-05,
      "model_forward_time": 2.8040454387664795,
      "step": 28485
    },
    {
      "epoch": 2.716541290283203e-05,
      "step": 28485,
      "training_step_time": 7.61745548248291
    },
    {
      "epoch": 2.7166366577148438e-05,
      "model_forward_time": 1.7184722423553467,
      "step": 28486
    },
    {
      "epoch": 2.7166366577148438e-05,
      "step": 28486,
      "training_step_time": 9.326835870742798
    },
    {
      "epoch": 2.7167320251464842e-05,
      "model_forward_time": 1.7905192375183105,
      "step": 28487
    },
    {
      "epoch": 2.7167320251464842e-05,
      "step": 28487,
      "training_step_time": 7.817180871963501
    },
    {
      "epoch": 2.716827392578125e-05,
      "model_forward_time": 1.7293634414672852,
      "step": 28488
    },
    {
      "epoch": 2.716827392578125e-05,
      "step": 28488,
      "training_step_time": 7.559544801712036
    },
    {
      "epoch": 2.7169227600097657e-05,
      "model_forward_time": 1.7368619441986084,
      "step": 28489
    },
    {
      "epoch": 2.7169227600097657e-05,
      "step": 28489,
      "training_step_time": 7.601301908493042
    },
    {
      "epoch": 2.717018127441406e-05,
      "grad_norm": 0.20507076382637024,
      "learning_rate": 8.565917426829013e-06,
      "loss": 0.0577,
      "step": 28490
    },
    {
      "epoch": 2.717018127441406e-05,
      "model_forward_time": 1.736039638519287,
      "step": 28490
    },
    {
      "dynamics_loss_avg": 0.07264125514775514,
      "epoch": 2.717018127441406e-05,
      "step": 28490
    },
    {
      "action_loss_avg": 0.002508901944383979,
      "epoch": 2.717018127441406e-05,
      "step": 28490
    },
    {
      "epoch": 2.717018127441406e-05,
      "step": 28490,
      "training_step_time": 6.818129301071167
    },
    {
      "epoch": 2.717113494873047e-05,
      "model_forward_time": 1.7308204174041748,
      "step": 28491
    },
    {
      "epoch": 2.717113494873047e-05,
      "step": 28491,
      "training_step_time": 6.818023681640625
    },
    {
      "epoch": 2.7172088623046876e-05,
      "model_forward_time": 1.3566231727600098,
      "step": 28492
    },
    {
      "epoch": 2.7172088623046876e-05,
      "step": 28492,
      "training_step_time": 6.81424880027771
    },
    {
      "epoch": 2.717304229736328e-05,
      "model_forward_time": 1.7297697067260742,
      "step": 28493
    },
    {
      "epoch": 2.717304229736328e-05,
      "step": 28493,
      "training_step_time": 7.742764472961426
    },
    {
      "epoch": 2.7173995971679687e-05,
      "model_forward_time": 1.7299301624298096,
      "step": 28494
    },
    {
      "epoch": 2.7173995971679687e-05,
      "step": 28494,
      "training_step_time": 8.051891803741455
    },
    {
      "epoch": 2.7174949645996095e-05,
      "model_forward_time": 2.9013471603393555,
      "step": 28495
    },
    {
      "epoch": 2.7174949645996095e-05,
      "step": 28495,
      "training_step_time": 8.702134370803833
    },
    {
      "epoch": 2.71759033203125e-05,
      "model_forward_time": 1.7920894622802734,
      "step": 28496
    },
    {
      "epoch": 2.71759033203125e-05,
      "step": 28496,
      "training_step_time": 9.508275270462036
    },
    {
      "epoch": 2.7176856994628906e-05,
      "model_forward_time": 1.728712558746338,
      "step": 28497
    },
    {
      "epoch": 2.7176856994628906e-05,
      "step": 28497,
      "training_step_time": 7.672063827514648
    },
    {
      "epoch": 2.7177810668945314e-05,
      "model_forward_time": 1.73637056350708,
      "step": 28498
    },
    {
      "epoch": 2.7177810668945314e-05,
      "step": 28498,
      "training_step_time": 7.555262804031372
    },
    {
      "epoch": 2.7178764343261718e-05,
      "model_forward_time": 1.7248272895812988,
      "step": 28499
    },
    {
      "epoch": 2.7178764343261718e-05,
      "step": 28499,
      "training_step_time": 6.776242733001709
    },
    {
      "epoch": 2.7179718017578125e-05,
      "grad_norm": 0.1507953554391861,
      "learning_rate": 8.564758186893629e-06,
      "loss": 0.062,
      "step": 28500
    },
    {
      "epoch": 2.7179718017578125e-05,
      "model_forward_time": 1.7102031707763672,
      "step": 28500
    },
    {
      "dynamics_loss_avg": 0.05959013216197491,
      "epoch": 2.7179718017578125e-05,
      "step": 28500
    },
    {
      "action_loss_avg": 0.001752719096839428,
      "epoch": 2.7179718017578125e-05,
      "step": 28500
    },
    {
      "epoch": 2.7179718017578125e-05,
      "step": 28500,
      "training_step_time": 6.799015283584595
    },
    {
      "epoch": 2.7180671691894532e-05,
      "model_forward_time": 1.727804183959961,
      "step": 28501
    },
    {
      "epoch": 2.7180671691894532e-05,
      "step": 28501,
      "training_step_time": 6.784229516983032
    },
    {
      "epoch": 2.7181625366210936e-05,
      "model_forward_time": 1.727128028869629,
      "step": 28502
    },
    {
      "epoch": 2.7181625366210936e-05,
      "step": 28502,
      "training_step_time": 7.672231435775757
    },
    {
      "epoch": 2.7182579040527344e-05,
      "model_forward_time": 1.724179983139038,
      "step": 28503
    },
    {
      "epoch": 2.7182579040527344e-05,
      "step": 28503,
      "training_step_time": 7.469259023666382
    },
    {
      "epoch": 2.718353271484375e-05,
      "model_forward_time": 2.8453221321105957,
      "step": 28504
    },
    {
      "epoch": 2.718353271484375e-05,
      "step": 28504,
      "training_step_time": 8.537516593933105
    },
    {
      "epoch": 2.7184486389160155e-05,
      "model_forward_time": 1.7730557918548584,
      "step": 28505
    },
    {
      "epoch": 2.7184486389160155e-05,
      "step": 28505,
      "training_step_time": 9.621835947036743
    },
    {
      "epoch": 2.7185440063476563e-05,
      "model_forward_time": 1.7251930236816406,
      "step": 28506
    },
    {
      "epoch": 2.7185440063476563e-05,
      "step": 28506,
      "training_step_time": 7.664572238922119
    },
    {
      "epoch": 2.718639373779297e-05,
      "model_forward_time": 1.7249858379364014,
      "step": 28507
    },
    {
      "epoch": 2.718639373779297e-05,
      "step": 28507,
      "training_step_time": 7.421823740005493
    },
    {
      "epoch": 2.7187347412109374e-05,
      "model_forward_time": 1.7274775505065918,
      "step": 28508
    },
    {
      "epoch": 2.7187347412109374e-05,
      "step": 28508,
      "training_step_time": 7.636115789413452
    },
    {
      "epoch": 2.718830108642578e-05,
      "model_forward_time": 1.7704503536224365,
      "step": 28509
    },
    {
      "epoch": 2.718830108642578e-05,
      "step": 28509,
      "training_step_time": 6.999696254730225
    },
    {
      "epoch": 2.718925476074219e-05,
      "grad_norm": 0.10998854041099548,
      "learning_rate": 8.56359855712165e-06,
      "loss": 0.0569,
      "step": 28510
    },
    {
      "epoch": 2.718925476074219e-05,
      "model_forward_time": 1.7222795486450195,
      "step": 28510
    },
    {
      "dynamics_loss_avg": 0.06596845872700215,
      "epoch": 2.718925476074219e-05,
      "step": 28510
    },
    {
      "action_loss_avg": 0.0021615192235913128,
      "epoch": 2.718925476074219e-05,
      "step": 28510
    },
    {
      "epoch": 2.718925476074219e-05,
      "step": 28510,
      "training_step_time": 6.8477184772491455
    },
    {
      "epoch": 2.7190208435058593e-05,
      "model_forward_time": 1.7221300601959229,
      "step": 28511
    },
    {
      "epoch": 2.7190208435058593e-05,
      "step": 28511,
      "training_step_time": 7.795314311981201
    },
    {
      "epoch": 2.7191162109375e-05,
      "model_forward_time": 1.3678717613220215,
      "step": 28512
    },
    {
      "epoch": 2.7191162109375e-05,
      "step": 28512,
      "training_step_time": 7.643410682678223
    },
    {
      "epoch": 2.7192115783691408e-05,
      "model_forward_time": 2.842041254043579,
      "step": 28513
    },
    {
      "epoch": 2.7192115783691408e-05,
      "step": 28513,
      "training_step_time": 9.43088412284851
    },
    {
      "epoch": 2.7193069458007812e-05,
      "model_forward_time": 1.7330360412597656,
      "step": 28514
    },
    {
      "epoch": 2.7193069458007812e-05,
      "step": 28514,
      "training_step_time": 9.765217304229736
    },
    {
      "epoch": 2.719402313232422e-05,
      "model_forward_time": 1.7247850894927979,
      "step": 28515
    },
    {
      "epoch": 2.719402313232422e-05,
      "step": 28515,
      "training_step_time": 7.837226152420044
    },
    {
      "epoch": 2.7194976806640623e-05,
      "model_forward_time": 1.750659704208374,
      "step": 28516
    },
    {
      "epoch": 2.7194976806640623e-05,
      "step": 28516,
      "training_step_time": 7.744429349899292
    },
    {
      "epoch": 2.719593048095703e-05,
      "model_forward_time": 1.729290246963501,
      "step": 28517
    },
    {
      "epoch": 2.719593048095703e-05,
      "step": 28517,
      "training_step_time": 7.681563138961792
    },
    {
      "epoch": 2.7196884155273438e-05,
      "model_forward_time": 1.7235634326934814,
      "step": 28518
    },
    {
      "epoch": 2.7196884155273438e-05,
      "step": 28518,
      "training_step_time": 6.850642204284668
    },
    {
      "epoch": 2.7197837829589842e-05,
      "model_forward_time": 1.7231979370117188,
      "step": 28519
    },
    {
      "epoch": 2.7197837829589842e-05,
      "step": 28519,
      "training_step_time": 6.995096206665039
    },
    {
      "epoch": 2.719879150390625e-05,
      "grad_norm": 0.1092173308134079,
      "learning_rate": 8.562438537639888e-06,
      "loss": 0.0566,
      "step": 28520
    },
    {
      "epoch": 2.719879150390625e-05,
      "model_forward_time": 1.722620964050293,
      "step": 28520
    },
    {
      "dynamics_loss_avg": 0.047132961452007294,
      "epoch": 2.719879150390625e-05,
      "step": 28520
    },
    {
      "action_loss_avg": 0.0019006116664968431,
      "epoch": 2.719879150390625e-05,
      "step": 28520
    },
    {
      "epoch": 2.719879150390625e-05,
      "step": 28520,
      "training_step_time": 7.851822137832642
    },
    {
      "epoch": 2.7199745178222657e-05,
      "model_forward_time": 1.7254040241241455,
      "step": 28521
    },
    {
      "epoch": 2.7199745178222657e-05,
      "step": 28521,
      "training_step_time": 7.761054515838623
    },
    {
      "epoch": 2.720069885253906e-05,
      "model_forward_time": 1.7293360233306885,
      "step": 28522
    },
    {
      "epoch": 2.720069885253906e-05,
      "step": 28522,
      "training_step_time": 9.278861999511719
    },
    {
      "epoch": 2.720165252685547e-05,
      "model_forward_time": 1.7241275310516357,
      "step": 28523
    },
    {
      "epoch": 2.720165252685547e-05,
      "step": 28523,
      "training_step_time": 9.290202379226685
    },
    {
      "epoch": 2.7202606201171876e-05,
      "model_forward_time": 1.787381887435913,
      "step": 28524
    },
    {
      "epoch": 2.7202606201171876e-05,
      "step": 28524,
      "training_step_time": 8.779768705368042
    },
    {
      "epoch": 2.720355987548828e-05,
      "model_forward_time": 1.7298877239227295,
      "step": 28525
    },
    {
      "epoch": 2.720355987548828e-05,
      "step": 28525,
      "training_step_time": 7.678441762924194
    },
    {
      "epoch": 2.7204513549804687e-05,
      "model_forward_time": 1.7289385795593262,
      "step": 28526
    },
    {
      "epoch": 2.7204513549804687e-05,
      "step": 28526,
      "training_step_time": 7.648917198181152
    },
    {
      "epoch": 2.7205467224121095e-05,
      "model_forward_time": 1.7249987125396729,
      "step": 28527
    },
    {
      "epoch": 2.7205467224121095e-05,
      "step": 28527,
      "training_step_time": 6.9328203201293945
    },
    {
      "epoch": 2.72064208984375e-05,
      "model_forward_time": 1.7249372005462646,
      "step": 28528
    },
    {
      "epoch": 2.72064208984375e-05,
      "step": 28528,
      "training_step_time": 6.965000867843628
    },
    {
      "epoch": 2.7207374572753906e-05,
      "model_forward_time": 1.7276394367218018,
      "step": 28529
    },
    {
      "epoch": 2.7207374572753906e-05,
      "step": 28529,
      "training_step_time": 7.835453748703003
    },
    {
      "epoch": 2.7208328247070314e-05,
      "grad_norm": 0.1531497836112976,
      "learning_rate": 8.561278128575206e-06,
      "loss": 0.0602,
      "step": 28530
    },
    {
      "epoch": 2.7208328247070314e-05,
      "model_forward_time": 1.7338123321533203,
      "step": 28530
    },
    {
      "dynamics_loss_avg": 0.057316462974995375,
      "epoch": 2.7208328247070314e-05,
      "step": 28530
    },
    {
      "action_loss_avg": 0.0015863036242080852,
      "epoch": 2.7208328247070314e-05,
      "step": 28530
    },
    {
      "epoch": 2.7208328247070314e-05,
      "step": 28530,
      "training_step_time": 7.295725107192993
    },
    {
      "epoch": 2.7209281921386718e-05,
      "model_forward_time": 1.7289092540740967,
      "step": 28531
    },
    {
      "epoch": 2.7209281921386718e-05,
      "step": 28531,
      "training_step_time": 9.559308290481567
    },
    {
      "epoch": 2.7210235595703125e-05,
      "model_forward_time": 2.8734397888183594,
      "step": 28532
    },
    {
      "epoch": 2.7210235595703125e-05,
      "step": 28532,
      "training_step_time": 10.519785165786743
    },
    {
      "epoch": 2.7211189270019532e-05,
      "model_forward_time": 1.79486083984375,
      "step": 28533
    },
    {
      "epoch": 2.7211189270019532e-05,
      "step": 28533,
      "training_step_time": 8.30072021484375
    },
    {
      "epoch": 2.7212142944335936e-05,
      "model_forward_time": 1.7241060733795166,
      "step": 28534
    },
    {
      "epoch": 2.7212142944335936e-05,
      "step": 28534,
      "training_step_time": 7.632391452789307
    },
    {
      "epoch": 2.7213096618652344e-05,
      "model_forward_time": 1.7281906604766846,
      "step": 28535
    },
    {
      "epoch": 2.7213096618652344e-05,
      "step": 28535,
      "training_step_time": 7.612346887588501
    },
    {
      "epoch": 2.721405029296875e-05,
      "model_forward_time": 1.723323106765747,
      "step": 28536
    },
    {
      "epoch": 2.721405029296875e-05,
      "step": 28536,
      "training_step_time": 7.553311586380005
    },
    {
      "epoch": 2.7215003967285155e-05,
      "model_forward_time": 1.7335364818572998,
      "step": 28537
    },
    {
      "epoch": 2.7215003967285155e-05,
      "step": 28537,
      "training_step_time": 6.846443176269531
    },
    {
      "epoch": 2.7215957641601563e-05,
      "model_forward_time": 1.7249693870544434,
      "step": 28538
    },
    {
      "epoch": 2.7215957641601563e-05,
      "step": 28538,
      "training_step_time": 7.718819618225098
    },
    {
      "epoch": 2.721691131591797e-05,
      "model_forward_time": 1.728102684020996,
      "step": 28539
    },
    {
      "epoch": 2.721691131591797e-05,
      "step": 28539,
      "training_step_time": 7.566156625747681
    },
    {
      "epoch": 2.7217864990234374e-05,
      "grad_norm": 0.12888392806053162,
      "learning_rate": 8.5601173300545e-06,
      "loss": 0.0571,
      "step": 28540
    },
    {
      "epoch": 2.7217864990234374e-05,
      "model_forward_time": 1.7384486198425293,
      "step": 28540
    },
    {
      "dynamics_loss_avg": 0.053446244820952414,
      "epoch": 2.7217864990234374e-05,
      "step": 28540
    },
    {
      "action_loss_avg": 0.002107762114610523,
      "epoch": 2.7217864990234374e-05,
      "step": 28540
    },
    {
      "epoch": 2.7217864990234374e-05,
      "step": 28540,
      "training_step_time": 7.557333707809448
    },
    {
      "epoch": 2.721881866455078e-05,
      "model_forward_time": 2.9421613216400146,
      "step": 28541
    },
    {
      "epoch": 2.721881866455078e-05,
      "step": 28541,
      "training_step_time": 9.16983699798584
    },
    {
      "epoch": 2.721977233886719e-05,
      "model_forward_time": 1.789292573928833,
      "step": 28542
    },
    {
      "epoch": 2.721977233886719e-05,
      "step": 28542,
      "training_step_time": 7.671643257141113
    },
    {
      "epoch": 2.7220726013183593e-05,
      "model_forward_time": 1.7246391773223877,
      "step": 28543
    },
    {
      "epoch": 2.7220726013183593e-05,
      "step": 28543,
      "training_step_time": 7.663290977478027
    },
    {
      "epoch": 2.72216796875e-05,
      "model_forward_time": 1.7227108478546143,
      "step": 28544
    },
    {
      "epoch": 2.72216796875e-05,
      "step": 28544,
      "training_step_time": 7.580949544906616
    },
    {
      "epoch": 2.7222633361816408e-05,
      "model_forward_time": 1.3649990558624268,
      "step": 28545
    },
    {
      "epoch": 2.7222633361816408e-05,
      "step": 28545,
      "training_step_time": 7.232820272445679
    },
    {
      "epoch": 2.7223587036132812e-05,
      "model_forward_time": 1.7279291152954102,
      "step": 28546
    },
    {
      "epoch": 2.7223587036132812e-05,
      "step": 28546,
      "training_step_time": 6.844207525253296
    },
    {
      "epoch": 2.722454071044922e-05,
      "model_forward_time": 1.7278549671173096,
      "step": 28547
    },
    {
      "epoch": 2.722454071044922e-05,
      "step": 28547,
      "training_step_time": 7.700519800186157
    },
    {
      "epoch": 2.7225494384765623e-05,
      "model_forward_time": 1.729193925857544,
      "step": 28548
    },
    {
      "epoch": 2.7225494384765623e-05,
      "step": 28548,
      "training_step_time": 6.854154825210571
    },
    {
      "epoch": 2.722644805908203e-05,
      "model_forward_time": 1.7323403358459473,
      "step": 28549
    },
    {
      "epoch": 2.722644805908203e-05,
      "step": 28549,
      "training_step_time": 8.488884449005127
    },
    {
      "epoch": 2.7227401733398438e-05,
      "grad_norm": 0.12591196596622467,
      "learning_rate": 8.558956142204718e-06,
      "loss": 0.0572,
      "step": 28550
    },
    {
      "epoch": 2.7227401733398438e-05,
      "model_forward_time": 2.85532546043396,
      "step": 28550
    },
    {
      "dynamics_loss_avg": 0.07040509507060051,
      "epoch": 2.7227401733398438e-05,
      "step": 28550
    },
    {
      "action_loss_avg": 0.00255436705192551,
      "epoch": 2.7227401733398438e-05,
      "step": 28550
    },
    {
      "epoch": 2.7227401733398438e-05,
      "step": 28550,
      "training_step_time": 9.626614332199097
    },
    {
      "epoch": 2.7228355407714842e-05,
      "model_forward_time": 1.7638800144195557,
      "step": 28551
    },
    {
      "epoch": 2.7228355407714842e-05,
      "step": 28551,
      "training_step_time": 8.253538131713867
    },
    {
      "epoch": 2.722930908203125e-05,
      "model_forward_time": 1.7270162105560303,
      "step": 28552
    },
    {
      "epoch": 2.722930908203125e-05,
      "step": 28552,
      "training_step_time": 7.670320272445679
    },
    {
      "epoch": 2.7230262756347657e-05,
      "model_forward_time": 1.7129647731781006,
      "step": 28553
    },
    {
      "epoch": 2.7230262756347657e-05,
      "step": 28553,
      "training_step_time": 8.277120351791382
    },
    {
      "epoch": 2.723121643066406e-05,
      "model_forward_time": 1.7315852642059326,
      "step": 28554
    },
    {
      "epoch": 2.723121643066406e-05,
      "step": 28554,
      "training_step_time": 7.222856521606445
    },
    {
      "epoch": 2.723217010498047e-05,
      "model_forward_time": 1.7371933460235596,
      "step": 28555
    },
    {
      "epoch": 2.723217010498047e-05,
      "step": 28555,
      "training_step_time": 6.792731523513794
    },
    {
      "epoch": 2.7233123779296876e-05,
      "model_forward_time": 1.7386608123779297,
      "step": 28556
    },
    {
      "epoch": 2.7233123779296876e-05,
      "step": 28556,
      "training_step_time": 7.619307518005371
    },
    {
      "epoch": 2.723407745361328e-05,
      "model_forward_time": 1.7328031063079834,
      "step": 28557
    },
    {
      "epoch": 2.723407745361328e-05,
      "step": 28557,
      "training_step_time": 6.810589551925659
    },
    {
      "epoch": 2.7235031127929687e-05,
      "model_forward_time": 1.7364773750305176,
      "step": 28558
    },
    {
      "epoch": 2.7235031127929687e-05,
      "step": 28558,
      "training_step_time": 7.61638331413269
    },
    {
      "epoch": 2.7235984802246095e-05,
      "model_forward_time": 2.791188955307007,
      "step": 28559
    },
    {
      "epoch": 2.7235984802246095e-05,
      "step": 28559,
      "training_step_time": 7.681954622268677
    },
    {
      "epoch": 2.72369384765625e-05,
      "grad_norm": 0.13263562321662903,
      "learning_rate": 8.557794565152839e-06,
      "loss": 0.0598,
      "step": 28560
    },
    {
      "epoch": 2.72369384765625e-05,
      "model_forward_time": 1.717163324356079,
      "step": 28560
    },
    {
      "dynamics_loss_avg": 0.06628482043743134,
      "epoch": 2.72369384765625e-05,
      "step": 28560
    },
    {
      "action_loss_avg": 0.001924255135236308,
      "epoch": 2.72369384765625e-05,
      "step": 28560
    },
    {
      "epoch": 2.72369384765625e-05,
      "step": 28560,
      "training_step_time": 10.057342767715454
    },
    {
      "epoch": 2.7237892150878906e-05,
      "model_forward_time": 1.7310082912445068,
      "step": 28561
    },
    {
      "epoch": 2.7237892150878906e-05,
      "step": 28561,
      "training_step_time": 7.813608646392822
    },
    {
      "epoch": 2.7238845825195314e-05,
      "model_forward_time": 1.7343409061431885,
      "step": 28562
    },
    {
      "epoch": 2.7238845825195314e-05,
      "step": 28562,
      "training_step_time": 7.656419992446899
    },
    {
      "epoch": 2.7239799499511718e-05,
      "model_forward_time": 1.3650596141815186,
      "step": 28563
    },
    {
      "epoch": 2.7239799499511718e-05,
      "step": 28563,
      "training_step_time": 7.555685520172119
    },
    {
      "epoch": 2.7240753173828125e-05,
      "model_forward_time": 1.7409942150115967,
      "step": 28564
    },
    {
      "epoch": 2.7240753173828125e-05,
      "step": 28564,
      "training_step_time": 7.1014392375946045
    },
    {
      "epoch": 2.7241706848144532e-05,
      "model_forward_time": 1.7223012447357178,
      "step": 28565
    },
    {
      "epoch": 2.7241706848144532e-05,
      "step": 28565,
      "training_step_time": 7.623274803161621
    },
    {
      "epoch": 2.7242660522460936e-05,
      "model_forward_time": 1.7389154434204102,
      "step": 28566
    },
    {
      "epoch": 2.7242660522460936e-05,
      "step": 28566,
      "training_step_time": 7.649946212768555
    },
    {
      "epoch": 2.7243614196777344e-05,
      "model_forward_time": 1.7264766693115234,
      "step": 28567
    },
    {
      "epoch": 2.7243614196777344e-05,
      "step": 28567,
      "training_step_time": 7.608379602432251
    },
    {
      "epoch": 2.724456787109375e-05,
      "model_forward_time": 1.5532350540161133,
      "step": 28568
    },
    {
      "epoch": 2.724456787109375e-05,
      "step": 28568,
      "training_step_time": 8.90663766860962
    },
    {
      "epoch": 2.7245521545410155e-05,
      "model_forward_time": 2.97485089302063,
      "step": 28569
    },
    {
      "epoch": 2.7245521545410155e-05,
      "step": 28569,
      "training_step_time": 8.786904096603394
    },
    {
      "epoch": 2.7246475219726563e-05,
      "grad_norm": 0.12241468578577042,
      "learning_rate": 8.5566325990259e-06,
      "loss": 0.061,
      "step": 28570
    },
    {
      "epoch": 2.7246475219726563e-05,
      "model_forward_time": 1.8651039600372314,
      "step": 28570
    },
    {
      "dynamics_loss_avg": 0.07508110608905554,
      "epoch": 2.7246475219726563e-05,
      "step": 28570
    },
    {
      "action_loss_avg": 0.0023772896907757967,
      "epoch": 2.7246475219726563e-05,
      "step": 28570
    },
    {
      "epoch": 2.7246475219726563e-05,
      "step": 28570,
      "training_step_time": 8.405118465423584
    },
    {
      "epoch": 2.724742889404297e-05,
      "model_forward_time": 1.9084703922271729,
      "step": 28571
    },
    {
      "epoch": 2.724742889404297e-05,
      "step": 28571,
      "training_step_time": 7.605119943618774
    },
    {
      "epoch": 2.7248382568359374e-05,
      "model_forward_time": 1.8882372379302979,
      "step": 28572
    },
    {
      "epoch": 2.7248382568359374e-05,
      "step": 28572,
      "training_step_time": 7.552941560745239
    },
    {
      "epoch": 2.724933624267578e-05,
      "model_forward_time": 1.913351058959961,
      "step": 28573
    },
    {
      "epoch": 2.724933624267578e-05,
      "step": 28573,
      "training_step_time": 6.989052057266235
    },
    {
      "epoch": 2.725028991699219e-05,
      "model_forward_time": 1.8925769329071045,
      "step": 28574
    },
    {
      "epoch": 2.725028991699219e-05,
      "step": 28574,
      "training_step_time": 7.576837062835693
    },
    {
      "epoch": 2.7251243591308593e-05,
      "model_forward_time": 1.8606460094451904,
      "step": 28575
    },
    {
      "epoch": 2.7251243591308593e-05,
      "step": 28575,
      "training_step_time": 7.632663726806641
    },
    {
      "epoch": 2.7252197265625e-05,
      "model_forward_time": 1.7897632122039795,
      "step": 28576
    },
    {
      "epoch": 2.7252197265625e-05,
      "step": 28576,
      "training_step_time": 8.239476680755615
    },
    {
      "epoch": 2.7253150939941408e-05,
      "model_forward_time": 1.7388098239898682,
      "step": 28577
    },
    {
      "epoch": 2.7253150939941408e-05,
      "step": 28577,
      "training_step_time": 8.219353914260864
    },
    {
      "epoch": 2.7254104614257812e-05,
      "model_forward_time": 2.895524740219116,
      "step": 28578
    },
    {
      "epoch": 2.7254104614257812e-05,
      "step": 28578,
      "training_step_time": 9.69690489768982
    },
    {
      "epoch": 2.725505828857422e-05,
      "model_forward_time": 1.7563061714172363,
      "step": 28579
    },
    {
      "epoch": 2.725505828857422e-05,
      "step": 28579,
      "training_step_time": 8.61341404914856
    },
    {
      "epoch": 2.7256011962890623e-05,
      "grad_norm": 0.1490059494972229,
      "learning_rate": 8.555470243950963e-06,
      "loss": 0.0615,
      "step": 28580
    },
    {
      "epoch": 2.7256011962890623e-05,
      "model_forward_time": 1.7254235744476318,
      "step": 28580
    },
    {
      "dynamics_loss_avg": 0.06432566307485103,
      "epoch": 2.7256011962890623e-05,
      "step": 28580
    },
    {
      "action_loss_avg": 0.0025614233629312365,
      "epoch": 2.7256011962890623e-05,
      "step": 28580
    },
    {
      "epoch": 2.7256011962890623e-05,
      "step": 28580,
      "training_step_time": 7.688171625137329
    },
    {
      "epoch": 2.725696563720703e-05,
      "model_forward_time": 1.7395834922790527,
      "step": 28581
    },
    {
      "epoch": 2.725696563720703e-05,
      "step": 28581,
      "training_step_time": 7.547720670700073
    },
    {
      "epoch": 2.7257919311523438e-05,
      "model_forward_time": 1.741987705230713,
      "step": 28582
    },
    {
      "epoch": 2.7257919311523438e-05,
      "step": 28582,
      "training_step_time": 7.142877578735352
    },
    {
      "epoch": 2.7258872985839842e-05,
      "model_forward_time": 1.7426397800445557,
      "step": 28583
    },
    {
      "epoch": 2.7258872985839842e-05,
      "step": 28583,
      "training_step_time": 7.546560525894165
    },
    {
      "epoch": 2.725982666015625e-05,
      "model_forward_time": 1.7245988845825195,
      "step": 28584
    },
    {
      "epoch": 2.725982666015625e-05,
      "step": 28584,
      "training_step_time": 7.636733055114746
    },
    {
      "epoch": 2.7260780334472657e-05,
      "model_forward_time": 1.365718126296997,
      "step": 28585
    },
    {
      "epoch": 2.7260780334472657e-05,
      "step": 28585,
      "training_step_time": 7.3830931186676025
    },
    {
      "epoch": 2.726173400878906e-05,
      "model_forward_time": 1.7373571395874023,
      "step": 28586
    },
    {
      "epoch": 2.726173400878906e-05,
      "step": 28586,
      "training_step_time": 9.377990007400513
    },
    {
      "epoch": 2.726268768310547e-05,
      "model_forward_time": 2.83034610748291,
      "step": 28587
    },
    {
      "epoch": 2.726268768310547e-05,
      "step": 28587,
      "training_step_time": 9.778948783874512
    },
    {
      "epoch": 2.7263641357421876e-05,
      "model_forward_time": 1.741964340209961,
      "step": 28588
    },
    {
      "epoch": 2.7263641357421876e-05,
      "step": 28588,
      "training_step_time": 9.754754304885864
    },
    {
      "epoch": 2.726459503173828e-05,
      "model_forward_time": 1.1024844646453857,
      "step": 28589
    },
    {
      "epoch": 2.726459503173828e-05,
      "step": 28589,
      "training_step_time": 8.45872688293457
    },
    {
      "epoch": 2.7265548706054687e-05,
      "grad_norm": 0.11991918087005615,
      "learning_rate": 8.55430750005515e-06,
      "loss": 0.0645,
      "step": 28590
    },
    {
      "epoch": 2.7265548706054687e-05,
      "model_forward_time": 1.7241361141204834,
      "step": 28590
    },
    {
      "dynamics_loss_avg": 0.050006844103336334,
      "epoch": 2.7265548706054687e-05,
      "step": 28590
    },
    {
      "action_loss_avg": 0.001970074698328972,
      "epoch": 2.7265548706054687e-05,
      "step": 28590
    },
    {
      "epoch": 2.7265548706054687e-05,
      "step": 28590,
      "training_step_time": 6.8556554317474365
    },
    {
      "epoch": 2.7266502380371095e-05,
      "model_forward_time": 1.7422056198120117,
      "step": 28591
    },
    {
      "epoch": 2.7266502380371095e-05,
      "step": 28591,
      "training_step_time": 7.478827953338623
    },
    {
      "epoch": 2.72674560546875e-05,
      "model_forward_time": 1.7339146137237549,
      "step": 28592
    },
    {
      "epoch": 2.72674560546875e-05,
      "step": 28592,
      "training_step_time": 7.610933542251587
    },
    {
      "epoch": 2.7268409729003906e-05,
      "model_forward_time": 1.7196831703186035,
      "step": 28593
    },
    {
      "epoch": 2.7268409729003906e-05,
      "step": 28593,
      "training_step_time": 7.620420217514038
    },
    {
      "epoch": 2.7269363403320314e-05,
      "model_forward_time": 1.7419250011444092,
      "step": 28594
    },
    {
      "epoch": 2.7269363403320314e-05,
      "step": 28594,
      "training_step_time": 7.451269149780273
    },
    {
      "epoch": 2.7270317077636718e-05,
      "model_forward_time": 1.740276575088501,
      "step": 28595
    },
    {
      "epoch": 2.7270317077636718e-05,
      "step": 28595,
      "training_step_time": 7.678104877471924
    },
    {
      "epoch": 2.7271270751953125e-05,
      "model_forward_time": 1.737894058227539,
      "step": 28596
    },
    {
      "epoch": 2.7271270751953125e-05,
      "step": 28596,
      "training_step_time": 9.047297477722168
    },
    {
      "epoch": 2.7272224426269532e-05,
      "model_forward_time": 2.5193490982055664,
      "step": 28597
    },
    {
      "epoch": 2.7272224426269532e-05,
      "step": 28597,
      "training_step_time": 9.139908790588379
    },
    {
      "epoch": 2.7273178100585937e-05,
      "model_forward_time": 1.8355839252471924,
      "step": 28598
    },
    {
      "epoch": 2.7273178100585937e-05,
      "step": 28598,
      "training_step_time": 7.665678977966309
    },
    {
      "epoch": 2.7274131774902344e-05,
      "model_forward_time": 1.7372586727142334,
      "step": 28599
    },
    {
      "epoch": 2.7274131774902344e-05,
      "step": 28599,
      "training_step_time": 7.491038084030151
    },
    {
      "epoch": 2.727508544921875e-05,
      "grad_norm": 0.167654350399971,
      "learning_rate": 8.55314436746561e-06,
      "loss": 0.0599,
      "step": 28600
    },
    {
      "epoch": 2.727508544921875e-05,
      "model_forward_time": 1.7301373481750488,
      "step": 28600
    },
    {
      "dynamics_loss_avg": 0.06229640059173107,
      "epoch": 2.727508544921875e-05,
      "step": 28600
    },
    {
      "action_loss_avg": 0.003316784603521228,
      "epoch": 2.727508544921875e-05,
      "step": 28600
    },
    {
      "epoch": 2.727508544921875e-05,
      "step": 28600,
      "training_step_time": 7.481147050857544
    },
    {
      "epoch": 2.7276039123535155e-05,
      "model_forward_time": 1.7338519096374512,
      "step": 28601
    },
    {
      "epoch": 2.7276039123535155e-05,
      "step": 28601,
      "training_step_time": 7.622293710708618
    },
    {
      "epoch": 2.7276992797851563e-05,
      "model_forward_time": 1.7368438243865967,
      "step": 28602
    },
    {
      "epoch": 2.7276992797851563e-05,
      "step": 28602,
      "training_step_time": 7.572123765945435
    },
    {
      "epoch": 2.727794647216797e-05,
      "model_forward_time": 1.7393996715545654,
      "step": 28603
    },
    {
      "epoch": 2.727794647216797e-05,
      "step": 28603,
      "training_step_time": 7.501085996627808
    },
    {
      "epoch": 2.7278900146484374e-05,
      "model_forward_time": 1.7465016841888428,
      "step": 28604
    },
    {
      "epoch": 2.7278900146484374e-05,
      "step": 28604,
      "training_step_time": 10.227679252624512
    },
    {
      "epoch": 2.727985382080078e-05,
      "model_forward_time": 1.7440845966339111,
      "step": 28605
    },
    {
      "epoch": 2.727985382080078e-05,
      "step": 28605,
      "training_step_time": 8.445232629776001
    },
    {
      "epoch": 2.728080749511719e-05,
      "model_forward_time": 2.8540842533111572,
      "step": 28606
    },
    {
      "epoch": 2.728080749511719e-05,
      "step": 28606,
      "training_step_time": 7.775344610214233
    },
    {
      "epoch": 2.7281761169433593e-05,
      "model_forward_time": 1.8227708339691162,
      "step": 28607
    },
    {
      "epoch": 2.7281761169433593e-05,
      "step": 28607,
      "training_step_time": 8.212499618530273
    },
    {
      "epoch": 2.728271484375e-05,
      "model_forward_time": 1.7376067638397217,
      "step": 28608
    },
    {
      "epoch": 2.728271484375e-05,
      "step": 28608,
      "training_step_time": 7.618547439575195
    },
    {
      "epoch": 2.7283668518066408e-05,
      "model_forward_time": 1.7419371604919434,
      "step": 28609
    },
    {
      "epoch": 2.7283668518066408e-05,
      "step": 28609,
      "training_step_time": 7.5637171268463135
    },
    {
      "epoch": 2.7284622192382812e-05,
      "grad_norm": 0.10750886797904968,
      "learning_rate": 8.551980846309545e-06,
      "loss": 0.0608,
      "step": 28610
    },
    {
      "epoch": 2.7284622192382812e-05,
      "model_forward_time": 1.7274413108825684,
      "step": 28610
    },
    {
      "dynamics_loss_avg": 0.06323528997600078,
      "epoch": 2.7284622192382812e-05,
      "step": 28610
    },
    {
      "action_loss_avg": 0.0019131802604533733,
      "epoch": 2.7284622192382812e-05,
      "step": 28610
    },
    {
      "epoch": 2.7284622192382812e-05,
      "step": 28610,
      "training_step_time": 7.269511699676514
    },
    {
      "epoch": 2.728557586669922e-05,
      "model_forward_time": 1.7397613525390625,
      "step": 28611
    },
    {
      "epoch": 2.728557586669922e-05,
      "step": 28611,
      "training_step_time": 7.612816333770752
    },
    {
      "epoch": 2.7286529541015623e-05,
      "model_forward_time": 1.7294063568115234,
      "step": 28612
    },
    {
      "epoch": 2.7286529541015623e-05,
      "step": 28612,
      "training_step_time": 7.547682523727417
    },
    {
      "epoch": 2.728748321533203e-05,
      "model_forward_time": 1.7428288459777832,
      "step": 28613
    },
    {
      "epoch": 2.728748321533203e-05,
      "step": 28613,
      "training_step_time": 8.45676302909851
    },
    {
      "epoch": 2.7288436889648438e-05,
      "model_forward_time": 1.7395875453948975,
      "step": 28614
    },
    {
      "epoch": 2.7288436889648438e-05,
      "step": 28614,
      "training_step_time": 9.044671535491943
    },
    {
      "epoch": 2.7289390563964842e-05,
      "model_forward_time": 1.7424702644348145,
      "step": 28615
    },
    {
      "epoch": 2.7289390563964842e-05,
      "step": 28615,
      "training_step_time": 9.30618929862976
    },
    {
      "epoch": 2.729034423828125e-05,
      "model_forward_time": 1.798126220703125,
      "step": 28616
    },
    {
      "epoch": 2.729034423828125e-05,
      "step": 28616,
      "training_step_time": 8.592066049575806
    },
    {
      "epoch": 2.7291297912597657e-05,
      "model_forward_time": 1.7444672584533691,
      "step": 28617
    },
    {
      "epoch": 2.7291297912597657e-05,
      "step": 28617,
      "training_step_time": 7.69521427154541
    },
    {
      "epoch": 2.729225158691406e-05,
      "model_forward_time": 1.741093635559082,
      "step": 28618
    },
    {
      "epoch": 2.729225158691406e-05,
      "step": 28618,
      "training_step_time": 8.076493501663208
    },
    {
      "epoch": 2.729320526123047e-05,
      "model_forward_time": 1.7384815216064453,
      "step": 28619
    },
    {
      "epoch": 2.729320526123047e-05,
      "step": 28619,
      "training_step_time": 7.28991961479187
    },
    {
      "epoch": 2.7294158935546876e-05,
      "grad_norm": 0.10140784829854965,
      "learning_rate": 8.550816936714194e-06,
      "loss": 0.0606,
      "step": 28620
    },
    {
      "epoch": 2.7294158935546876e-05,
      "model_forward_time": 1.7305574417114258,
      "step": 28620
    },
    {
      "dynamics_loss_avg": 0.07208395935595036,
      "epoch": 2.7294158935546876e-05,
      "step": 28620
    },
    {
      "action_loss_avg": 0.0023650761984754355,
      "epoch": 2.7294158935546876e-05,
      "step": 28620
    },
    {
      "epoch": 2.7294158935546876e-05,
      "step": 28620,
      "training_step_time": 7.699862718582153
    },
    {
      "epoch": 2.729511260986328e-05,
      "model_forward_time": 1.7413301467895508,
      "step": 28621
    },
    {
      "epoch": 2.729511260986328e-05,
      "step": 28621,
      "training_step_time": 7.7168049812316895
    },
    {
      "epoch": 2.7296066284179687e-05,
      "model_forward_time": 1.7370796203613281,
      "step": 28622
    },
    {
      "epoch": 2.7296066284179687e-05,
      "step": 28622,
      "training_step_time": 7.492823362350464
    },
    {
      "epoch": 2.7297019958496095e-05,
      "model_forward_time": 1.7146832942962646,
      "step": 28623
    },
    {
      "epoch": 2.7297019958496095e-05,
      "step": 28623,
      "training_step_time": 8.418680906295776
    },
    {
      "epoch": 2.72979736328125e-05,
      "model_forward_time": 1.735966444015503,
      "step": 28624
    },
    {
      "epoch": 2.72979736328125e-05,
      "step": 28624,
      "training_step_time": 9.18764877319336
    },
    {
      "epoch": 2.7298927307128906e-05,
      "model_forward_time": 2.9973442554473877,
      "step": 28625
    },
    {
      "epoch": 2.7298927307128906e-05,
      "step": 28625,
      "training_step_time": 7.769457101821899
    },
    {
      "epoch": 2.7299880981445314e-05,
      "model_forward_time": 1.7250523567199707,
      "step": 28626
    },
    {
      "epoch": 2.7299880981445314e-05,
      "step": 28626,
      "training_step_time": 7.614821434020996
    },
    {
      "epoch": 2.7300834655761718e-05,
      "model_forward_time": 1.7269127368927002,
      "step": 28627
    },
    {
      "epoch": 2.7300834655761718e-05,
      "step": 28627,
      "training_step_time": 7.15475058555603
    },
    {
      "epoch": 2.7301788330078125e-05,
      "model_forward_time": 1.7975080013275146,
      "step": 28628
    },
    {
      "epoch": 2.7301788330078125e-05,
      "step": 28628,
      "training_step_time": 7.494589805603027
    },
    {
      "epoch": 2.7302742004394533e-05,
      "model_forward_time": 1.7349433898925781,
      "step": 28629
    },
    {
      "epoch": 2.7302742004394533e-05,
      "step": 28629,
      "training_step_time": 7.583831071853638
    },
    {
      "epoch": 2.7303695678710937e-05,
      "grad_norm": 0.1292572170495987,
      "learning_rate": 8.549652638806842e-06,
      "loss": 0.0612,
      "step": 28630
    },
    {
      "epoch": 2.7303695678710937e-05,
      "model_forward_time": 1.7645649909973145,
      "step": 28630
    },
    {
      "dynamics_loss_avg": 0.059722240641713144,
      "epoch": 2.7303695678710937e-05,
      "step": 28630
    },
    {
      "action_loss_avg": 0.0016664766008034348,
      "epoch": 2.7303695678710937e-05,
      "step": 28630
    },
    {
      "epoch": 2.7303695678710937e-05,
      "step": 28630,
      "training_step_time": 7.711104393005371
    },
    {
      "epoch": 2.7304649353027344e-05,
      "model_forward_time": 1.7314136028289795,
      "step": 28631
    },
    {
      "epoch": 2.7304649353027344e-05,
      "step": 28631,
      "training_step_time": 7.473132848739624
    },
    {
      "epoch": 2.730560302734375e-05,
      "model_forward_time": 1.734673261642456,
      "step": 28632
    },
    {
      "epoch": 2.730560302734375e-05,
      "step": 28632,
      "training_step_time": 7.682974338531494
    },
    {
      "epoch": 2.7306556701660155e-05,
      "model_forward_time": 1.7276411056518555,
      "step": 28633
    },
    {
      "epoch": 2.7306556701660155e-05,
      "step": 28633,
      "training_step_time": 9.482403993606567
    },
    {
      "epoch": 2.7307510375976563e-05,
      "model_forward_time": 2.911600112915039,
      "step": 28634
    },
    {
      "epoch": 2.7307510375976563e-05,
      "step": 28634,
      "training_step_time": 7.731562614440918
    },
    {
      "epoch": 2.730846405029297e-05,
      "model_forward_time": 1.8257408142089844,
      "step": 28635
    },
    {
      "epoch": 2.730846405029297e-05,
      "step": 28635,
      "training_step_time": 8.632935523986816
    },
    {
      "epoch": 2.7309417724609374e-05,
      "model_forward_time": 1.7272605895996094,
      "step": 28636
    },
    {
      "epoch": 2.7309417724609374e-05,
      "step": 28636,
      "training_step_time": 7.2823710441589355
    },
    {
      "epoch": 2.731037139892578e-05,
      "model_forward_time": 1.7327296733856201,
      "step": 28637
    },
    {
      "epoch": 2.731037139892578e-05,
      "step": 28637,
      "training_step_time": 7.515944480895996
    },
    {
      "epoch": 2.731132507324219e-05,
      "model_forward_time": 1.7287375926971436,
      "step": 28638
    },
    {
      "epoch": 2.731132507324219e-05,
      "step": 28638,
      "training_step_time": 7.711353778839111
    },
    {
      "epoch": 2.7312278747558593e-05,
      "model_forward_time": 1.7282378673553467,
      "step": 28639
    },
    {
      "epoch": 2.7312278747558593e-05,
      "step": 28639,
      "training_step_time": 7.643898963928223
    },
    {
      "epoch": 2.7313232421875e-05,
      "grad_norm": 0.11001899093389511,
      "learning_rate": 8.548487952714813e-06,
      "loss": 0.0583,
      "step": 28640
    },
    {
      "epoch": 2.7313232421875e-05,
      "model_forward_time": 1.7207632064819336,
      "step": 28640
    },
    {
      "dynamics_loss_avg": 0.07028856631368399,
      "epoch": 2.7313232421875e-05,
      "step": 28640
    },
    {
      "action_loss_avg": 0.0021102450380567463,
      "epoch": 2.7313232421875e-05,
      "step": 28640
    },
    {
      "epoch": 2.7313232421875e-05,
      "step": 28640,
      "training_step_time": 7.505021810531616
    },
    {
      "epoch": 2.7314186096191405e-05,
      "model_forward_time": 1.7338979244232178,
      "step": 28641
    },
    {
      "epoch": 2.7314186096191405e-05,
      "step": 28641,
      "training_step_time": 7.658618211746216
    },
    {
      "epoch": 2.7315139770507812e-05,
      "model_forward_time": 1.733245849609375,
      "step": 28642
    },
    {
      "epoch": 2.7315139770507812e-05,
      "step": 28642,
      "training_step_time": 9.073529958724976
    },
    {
      "epoch": 2.731609344482422e-05,
      "model_forward_time": 2.511378526687622,
      "step": 28643
    },
    {
      "epoch": 2.731609344482422e-05,
      "step": 28643,
      "training_step_time": 10.057217359542847
    },
    {
      "epoch": 2.7317047119140623e-05,
      "model_forward_time": 1.7830734252929688,
      "step": 28644
    },
    {
      "epoch": 2.7317047119140623e-05,
      "step": 28644,
      "training_step_time": 7.670485734939575
    },
    {
      "epoch": 2.731800079345703e-05,
      "model_forward_time": 1.7386488914489746,
      "step": 28645
    },
    {
      "epoch": 2.731800079345703e-05,
      "step": 28645,
      "training_step_time": 7.667125701904297
    },
    {
      "epoch": 2.7318954467773438e-05,
      "model_forward_time": 1.7368137836456299,
      "step": 28646
    },
    {
      "epoch": 2.7318954467773438e-05,
      "step": 28646,
      "training_step_time": 7.435034513473511
    },
    {
      "epoch": 2.7319908142089842e-05,
      "model_forward_time": 1.7375094890594482,
      "step": 28647
    },
    {
      "epoch": 2.7319908142089842e-05,
      "step": 28647,
      "training_step_time": 7.442541599273682
    },
    {
      "epoch": 2.732086181640625e-05,
      "model_forward_time": 1.737403154373169,
      "step": 28648
    },
    {
      "epoch": 2.732086181640625e-05,
      "step": 28648,
      "training_step_time": 7.223836183547974
    },
    {
      "epoch": 2.7321815490722657e-05,
      "model_forward_time": 1.7413287162780762,
      "step": 28649
    },
    {
      "epoch": 2.7321815490722657e-05,
      "step": 28649,
      "training_step_time": 7.425961494445801
    },
    {
      "epoch": 2.732276916503906e-05,
      "grad_norm": 0.15346863865852356,
      "learning_rate": 8.547322878565479e-06,
      "loss": 0.0604,
      "step": 28650
    },
    {
      "epoch": 2.732276916503906e-05,
      "model_forward_time": 1.7451982498168945,
      "step": 28650
    },
    {
      "dynamics_loss_avg": 0.06394348293542862,
      "epoch": 2.732276916503906e-05,
      "step": 28650
    },
    {
      "action_loss_avg": 0.002768072229810059,
      "epoch": 2.732276916503906e-05,
      "step": 28650
    },
    {
      "epoch": 2.732276916503906e-05,
      "step": 28650,
      "training_step_time": 8.518391609191895
    },
    {
      "epoch": 2.732372283935547e-05,
      "model_forward_time": 1.752681016921997,
      "step": 28651
    },
    {
      "epoch": 2.732372283935547e-05,
      "step": 28651,
      "training_step_time": 7.757633924484253
    },
    {
      "epoch": 2.7324676513671876e-05,
      "model_forward_time": 2.862171173095703,
      "step": 28652
    },
    {
      "epoch": 2.7324676513671876e-05,
      "step": 28652,
      "training_step_time": 8.557913541793823
    },
    {
      "epoch": 2.732563018798828e-05,
      "model_forward_time": 1.775954008102417,
      "step": 28653
    },
    {
      "epoch": 2.732563018798828e-05,
      "step": 28653,
      "training_step_time": 7.671594858169556
    },
    {
      "epoch": 2.7326583862304687e-05,
      "model_forward_time": 1.7509362697601318,
      "step": 28654
    },
    {
      "epoch": 2.7326583862304687e-05,
      "step": 28654,
      "training_step_time": 7.643219947814941
    },
    {
      "epoch": 2.7327537536621095e-05,
      "model_forward_time": 1.3868865966796875,
      "step": 28655
    },
    {
      "epoch": 2.7327537536621095e-05,
      "step": 28655,
      "training_step_time": 7.461032152175903
    },
    {
      "epoch": 2.73284912109375e-05,
      "model_forward_time": 1.7802734375,
      "step": 28656
    },
    {
      "epoch": 2.73284912109375e-05,
      "step": 28656,
      "training_step_time": 7.351026773452759
    },
    {
      "epoch": 2.7329444885253906e-05,
      "model_forward_time": 1.747765302658081,
      "step": 28657
    },
    {
      "epoch": 2.7329444885253906e-05,
      "step": 28657,
      "training_step_time": 7.638626337051392
    },
    {
      "epoch": 2.7330398559570314e-05,
      "model_forward_time": 1.7523159980773926,
      "step": 28658
    },
    {
      "epoch": 2.7330398559570314e-05,
      "step": 28658,
      "training_step_time": 7.382328987121582
    },
    {
      "epoch": 2.7331352233886718e-05,
      "model_forward_time": 1.7226121425628662,
      "step": 28659
    },
    {
      "epoch": 2.7331352233886718e-05,
      "step": 28659,
      "training_step_time": 8.339869260787964
    },
    {
      "epoch": 2.7332305908203125e-05,
      "grad_norm": 0.14877185225486755,
      "learning_rate": 8.546157416486246e-06,
      "loss": 0.0591,
      "step": 28660
    },
    {
      "epoch": 2.7332305908203125e-05,
      "model_forward_time": 1.7231600284576416,
      "step": 28660
    },
    {
      "dynamics_loss_avg": 0.06223974619060755,
      "epoch": 2.7332305908203125e-05,
      "step": 28660
    },
    {
      "action_loss_avg": 0.0025638345745392145,
      "epoch": 2.7332305908203125e-05,
      "step": 28660
    },
    {
      "epoch": 2.7332305908203125e-05,
      "step": 28660,
      "training_step_time": 8.772052526473999
    },
    {
      "epoch": 2.7333259582519533e-05,
      "model_forward_time": 1.7263813018798828,
      "step": 28661
    },
    {
      "epoch": 2.7333259582519533e-05,
      "step": 28661,
      "training_step_time": 8.714967966079712
    },
    {
      "epoch": 2.7334213256835937e-05,
      "model_forward_time": 2.8702023029327393,
      "step": 28662
    },
    {
      "epoch": 2.7334213256835937e-05,
      "step": 28662,
      "training_step_time": 7.647683382034302
    },
    {
      "epoch": 2.7335166931152344e-05,
      "model_forward_time": 1.8774664402008057,
      "step": 28663
    },
    {
      "epoch": 2.7335166931152344e-05,
      "step": 28663,
      "training_step_time": 8.920961380004883
    },
    {
      "epoch": 2.733612060546875e-05,
      "model_forward_time": 1.7516510486602783,
      "step": 28664
    },
    {
      "epoch": 2.733612060546875e-05,
      "step": 28664,
      "training_step_time": 7.444355010986328
    },
    {
      "epoch": 2.7337074279785155e-05,
      "model_forward_time": 1.7410197257995605,
      "step": 28665
    },
    {
      "epoch": 2.7337074279785155e-05,
      "step": 28665,
      "training_step_time": 7.184160470962524
    },
    {
      "epoch": 2.7338027954101563e-05,
      "model_forward_time": 1.751831293106079,
      "step": 28666
    },
    {
      "epoch": 2.7338027954101563e-05,
      "step": 28666,
      "training_step_time": 7.5745697021484375
    },
    {
      "epoch": 2.733898162841797e-05,
      "model_forward_time": 1.7493388652801514,
      "step": 28667
    },
    {
      "epoch": 2.733898162841797e-05,
      "step": 28667,
      "training_step_time": 7.382936477661133
    },
    {
      "epoch": 2.7339935302734374e-05,
      "model_forward_time": 1.7543649673461914,
      "step": 28668
    },
    {
      "epoch": 2.7339935302734374e-05,
      "step": 28668,
      "training_step_time": 8.238281488418579
    },
    {
      "epoch": 2.7340888977050782e-05,
      "model_forward_time": 1.7338593006134033,
      "step": 28669
    },
    {
      "epoch": 2.7340888977050782e-05,
      "step": 28669,
      "training_step_time": 7.729201793670654
    },
    {
      "epoch": 2.734184265136719e-05,
      "grad_norm": 0.18019999563694,
      "learning_rate": 8.54499156660457e-06,
      "loss": 0.06,
      "step": 28670
    },
    {
      "epoch": 2.734184265136719e-05,
      "model_forward_time": 1.7319941520690918,
      "step": 28670
    },
    {
      "dynamics_loss_avg": 0.0736486529931426,
      "epoch": 2.734184265136719e-05,
      "step": 28670
    },
    {
      "action_loss_avg": 0.0021958457422442735,
      "epoch": 2.734184265136719e-05,
      "step": 28670
    },
    {
      "epoch": 2.734184265136719e-05,
      "step": 28670,
      "training_step_time": 8.392010927200317
    },
    {
      "epoch": 2.7342796325683593e-05,
      "model_forward_time": 2.8917503356933594,
      "step": 28671
    },
    {
      "epoch": 2.7342796325683593e-05,
      "step": 28671,
      "training_step_time": 8.584730625152588
    },
    {
      "epoch": 2.734375e-05,
      "model_forward_time": 1.842503309249878,
      "step": 28672
    },
    {
      "epoch": 2.734375e-05,
      "step": 28672,
      "training_step_time": 7.941486835479736
    },
    {
      "epoch": 2.7344703674316405e-05,
      "model_forward_time": 1.753864049911499,
      "step": 28673
    },
    {
      "epoch": 2.7344703674316405e-05,
      "step": 28673,
      "training_step_time": 7.64577579498291
    },
    {
      "epoch": 2.7345657348632812e-05,
      "model_forward_time": 1.756565809249878,
      "step": 28674
    },
    {
      "epoch": 2.7345657348632812e-05,
      "step": 28674,
      "training_step_time": 7.460778474807739
    },
    {
      "epoch": 2.734661102294922e-05,
      "model_forward_time": 1.3918602466583252,
      "step": 28675
    },
    {
      "epoch": 2.734661102294922e-05,
      "step": 28675,
      "training_step_time": 7.688958168029785
    },
    {
      "epoch": 2.7347564697265623e-05,
      "model_forward_time": 1.7782411575317383,
      "step": 28676
    },
    {
      "epoch": 2.7347564697265623e-05,
      "step": 28676,
      "training_step_time": 6.9756128787994385
    },
    {
      "epoch": 2.734851837158203e-05,
      "model_forward_time": 1.7565639019012451,
      "step": 28677
    },
    {
      "epoch": 2.734851837158203e-05,
      "step": 28677,
      "training_step_time": 7.8436360359191895
    },
    {
      "epoch": 2.7349472045898438e-05,
      "model_forward_time": 1.756864309310913,
      "step": 28678
    },
    {
      "epoch": 2.7349472045898438e-05,
      "step": 28678,
      "training_step_time": 8.683440446853638
    },
    {
      "epoch": 2.7350425720214842e-05,
      "model_forward_time": 1.761399269104004,
      "step": 28679
    },
    {
      "epoch": 2.7350425720214842e-05,
      "step": 28679,
      "training_step_time": 8.351043939590454
    },
    {
      "epoch": 2.735137939453125e-05,
      "grad_norm": 0.1520150750875473,
      "learning_rate": 8.543825329047947e-06,
      "loss": 0.0632,
      "step": 28680
    },
    {
      "epoch": 2.735137939453125e-05,
      "model_forward_time": 2.5353808403015137,
      "step": 28680
    },
    {
      "dynamics_loss_avg": 0.07674533408135176,
      "epoch": 2.735137939453125e-05,
      "step": 28680
    },
    {
      "action_loss_avg": 0.0030472322599962355,
      "epoch": 2.735137939453125e-05,
      "step": 28680
    },
    {
      "epoch": 2.735137939453125e-05,
      "step": 28680,
      "training_step_time": 8.596505880355835
    },
    {
      "epoch": 2.7352333068847657e-05,
      "model_forward_time": 1.7966175079345703,
      "step": 28681
    },
    {
      "epoch": 2.7352333068847657e-05,
      "step": 28681,
      "training_step_time": 7.744693279266357
    },
    {
      "epoch": 2.735328674316406e-05,
      "model_forward_time": 1.7569730281829834,
      "step": 28682
    },
    {
      "epoch": 2.735328674316406e-05,
      "step": 28682,
      "training_step_time": 7.940628290176392
    },
    {
      "epoch": 2.735424041748047e-05,
      "model_forward_time": 1.7534220218658447,
      "step": 28683
    },
    {
      "epoch": 2.735424041748047e-05,
      "step": 28683,
      "training_step_time": 7.359839916229248
    },
    {
      "epoch": 2.7355194091796876e-05,
      "model_forward_time": 1.7558016777038574,
      "step": 28684
    },
    {
      "epoch": 2.7355194091796876e-05,
      "step": 28684,
      "training_step_time": 7.598244905471802
    },
    {
      "epoch": 2.735614776611328e-05,
      "model_forward_time": 1.7619190216064453,
      "step": 28685
    },
    {
      "epoch": 2.735614776611328e-05,
      "step": 28685,
      "training_step_time": 6.856480598449707
    },
    {
      "epoch": 2.7357101440429687e-05,
      "model_forward_time": 1.37856125831604,
      "step": 28686
    },
    {
      "epoch": 2.7357101440429687e-05,
      "step": 28686,
      "training_step_time": 7.889911651611328
    },
    {
      "epoch": 2.7358055114746095e-05,
      "model_forward_time": 1.3977196216583252,
      "step": 28687
    },
    {
      "epoch": 2.7358055114746095e-05,
      "step": 28687,
      "training_step_time": 8.667431116104126
    },
    {
      "epoch": 2.73590087890625e-05,
      "model_forward_time": 1.7873148918151855,
      "step": 28688
    },
    {
      "epoch": 2.73590087890625e-05,
      "step": 28688,
      "training_step_time": 7.783386707305908
    },
    {
      "epoch": 2.7359962463378906e-05,
      "model_forward_time": 1.7569637298583984,
      "step": 28689
    },
    {
      "epoch": 2.7359962463378906e-05,
      "step": 28689,
      "training_step_time": 9.736462354660034
    },
    {
      "epoch": 2.7360916137695314e-05,
      "grad_norm": 0.13113197684288025,
      "learning_rate": 8.542658703943914e-06,
      "loss": 0.059,
      "step": 28690
    },
    {
      "epoch": 2.7360916137695314e-05,
      "model_forward_time": 2.50760555267334,
      "step": 28690
    },
    {
      "dynamics_loss_avg": 0.06830875221639872,
      "epoch": 2.7360916137695314e-05,
      "step": 28690
    },
    {
      "action_loss_avg": 0.0016968924028333277,
      "epoch": 2.7360916137695314e-05,
      "step": 28690
    },
    {
      "epoch": 2.7360916137695314e-05,
      "step": 28690,
      "training_step_time": 7.640274524688721
    },
    {
      "epoch": 2.7361869812011718e-05,
      "model_forward_time": 1.8764257431030273,
      "step": 28691
    },
    {
      "epoch": 2.7361869812011718e-05,
      "step": 28691,
      "training_step_time": 7.894938230514526
    },
    {
      "epoch": 2.7362823486328125e-05,
      "model_forward_time": 1.754951000213623,
      "step": 28692
    },
    {
      "epoch": 2.7362823486328125e-05,
      "step": 28692,
      "training_step_time": 7.314247369766235
    },
    {
      "epoch": 2.7363777160644533e-05,
      "model_forward_time": 1.7587757110595703,
      "step": 28693
    },
    {
      "epoch": 2.7363777160644533e-05,
      "step": 28693,
      "training_step_time": 7.4310290813446045
    },
    {
      "epoch": 2.7364730834960937e-05,
      "model_forward_time": 1.7517881393432617,
      "step": 28694
    },
    {
      "epoch": 2.7364730834960937e-05,
      "step": 28694,
      "training_step_time": 7.669870138168335
    },
    {
      "epoch": 2.7365684509277344e-05,
      "model_forward_time": 1.7569818496704102,
      "step": 28695
    },
    {
      "epoch": 2.7365684509277344e-05,
      "step": 28695,
      "training_step_time": 7.775164842605591
    },
    {
      "epoch": 2.736663818359375e-05,
      "model_forward_time": 1.753305196762085,
      "step": 28696
    },
    {
      "epoch": 2.736663818359375e-05,
      "step": 28696,
      "training_step_time": 8.317784309387207
    },
    {
      "epoch": 2.7367591857910155e-05,
      "model_forward_time": 1.7581164836883545,
      "step": 28697
    },
    {
      "epoch": 2.7367591857910155e-05,
      "step": 28697,
      "training_step_time": 7.7182533740997314
    },
    {
      "epoch": 2.7368545532226563e-05,
      "model_forward_time": 1.817138433456421,
      "step": 28698
    },
    {
      "epoch": 2.7368545532226563e-05,
      "step": 28698,
      "training_step_time": 8.5392165184021
    },
    {
      "epoch": 2.736949920654297e-05,
      "model_forward_time": 2.4898126125335693,
      "step": 28699
    },
    {
      "epoch": 2.736949920654297e-05,
      "step": 28699,
      "training_step_time": 8.633968591690063
    },
    {
      "epoch": 2.7370452880859374e-05,
      "grad_norm": 0.1380985528230667,
      "learning_rate": 8.54149169142005e-06,
      "loss": 0.0661,
      "step": 28700
    },
    {
      "epoch": 2.7370452880859374e-05,
      "model_forward_time": 1.8072125911712646,
      "step": 28700
    },
    {
      "dynamics_loss_avg": 0.07962941024452448,
      "epoch": 2.7370452880859374e-05,
      "step": 28700
    },
    {
      "action_loss_avg": 0.002752904291264713,
      "epoch": 2.7370452880859374e-05,
      "step": 28700
    },
    {
      "epoch": 2.7370452880859374e-05,
      "step": 28700,
      "training_step_time": 8.747334718704224
    },
    {
      "epoch": 2.7371406555175782e-05,
      "model_forward_time": 1.7511842250823975,
      "step": 28701
    },
    {
      "epoch": 2.7371406555175782e-05,
      "step": 28701,
      "training_step_time": 7.381258249282837
    },
    {
      "epoch": 2.737236022949219e-05,
      "model_forward_time": 1.7481184005737305,
      "step": 28702
    },
    {
      "epoch": 2.737236022949219e-05,
      "step": 28702,
      "training_step_time": 7.555567264556885
    },
    {
      "epoch": 2.7373313903808593e-05,
      "model_forward_time": 1.7464511394500732,
      "step": 28703
    },
    {
      "epoch": 2.7373313903808593e-05,
      "step": 28703,
      "training_step_time": 7.714148283004761
    },
    {
      "epoch": 2.7374267578125e-05,
      "model_forward_time": 1.7557849884033203,
      "step": 28704
    },
    {
      "epoch": 2.7374267578125e-05,
      "step": 28704,
      "training_step_time": 7.8205649852752686
    },
    {
      "epoch": 2.7375221252441405e-05,
      "model_forward_time": 1.75600004196167,
      "step": 28705
    },
    {
      "epoch": 2.7375221252441405e-05,
      "step": 28705,
      "training_step_time": 8.334293842315674
    },
    {
      "epoch": 2.7376174926757812e-05,
      "model_forward_time": 1.7588212490081787,
      "step": 28706
    },
    {
      "epoch": 2.7376174926757812e-05,
      "step": 28706,
      "training_step_time": 8.38558578491211
    },
    {
      "epoch": 2.737712860107422e-05,
      "model_forward_time": 1.7502381801605225,
      "step": 28707
    },
    {
      "epoch": 2.737712860107422e-05,
      "step": 28707,
      "training_step_time": 10.166560411453247
    },
    {
      "epoch": 2.7378082275390623e-05,
      "model_forward_time": 2.843472480773926,
      "step": 28708
    },
    {
      "epoch": 2.7378082275390623e-05,
      "step": 28708,
      "training_step_time": 7.607024192810059
    },
    {
      "epoch": 2.737903594970703e-05,
      "model_forward_time": 1.7774412631988525,
      "step": 28709
    },
    {
      "epoch": 2.737903594970703e-05,
      "step": 28709,
      "training_step_time": 7.565382957458496
    },
    {
      "epoch": 2.737998962402344e-05,
      "grad_norm": 0.14311343431472778,
      "learning_rate": 8.540324291603981e-06,
      "loss": 0.0616,
      "step": 28710
    },
    {
      "epoch": 2.737998962402344e-05,
      "model_forward_time": 1.7421181201934814,
      "step": 28710
    },
    {
      "dynamics_loss_avg": 0.0493349552154541,
      "epoch": 2.737998962402344e-05,
      "step": 28710
    },
    {
      "action_loss_avg": 0.0017340623307973146,
      "epoch": 2.737998962402344e-05,
      "step": 28710
    },
    {
      "epoch": 2.737998962402344e-05,
      "step": 28710,
      "training_step_time": 7.814350843429565
    },
    {
      "epoch": 2.7380943298339842e-05,
      "model_forward_time": 1.7469902038574219,
      "step": 28711
    },
    {
      "epoch": 2.7380943298339842e-05,
      "step": 28711,
      "training_step_time": 7.499787330627441
    },
    {
      "epoch": 2.738189697265625e-05,
      "model_forward_time": 1.7456543445587158,
      "step": 28712
    },
    {
      "epoch": 2.738189697265625e-05,
      "step": 28712,
      "training_step_time": 7.811464071273804
    },
    {
      "epoch": 2.7382850646972657e-05,
      "model_forward_time": 1.7554466724395752,
      "step": 28713
    },
    {
      "epoch": 2.7382850646972657e-05,
      "step": 28713,
      "training_step_time": 7.62798285484314
    },
    {
      "epoch": 2.738380432128906e-05,
      "model_forward_time": 1.7610082626342773,
      "step": 28714
    },
    {
      "epoch": 2.738380432128906e-05,
      "step": 28714,
      "training_step_time": 7.57038950920105
    },
    {
      "epoch": 2.738475799560547e-05,
      "model_forward_time": 1.7485158443450928,
      "step": 28715
    },
    {
      "epoch": 2.738475799560547e-05,
      "step": 28715,
      "training_step_time": 8.162906169891357
    },
    {
      "epoch": 2.7385711669921876e-05,
      "model_forward_time": 1.7611267566680908,
      "step": 28716
    },
    {
      "epoch": 2.7385711669921876e-05,
      "step": 28716,
      "training_step_time": 9.590440511703491
    },
    {
      "epoch": 2.738666534423828e-05,
      "model_forward_time": 1.3705213069915771,
      "step": 28717
    },
    {
      "epoch": 2.738666534423828e-05,
      "step": 28717,
      "training_step_time": 9.052005052566528
    },
    {
      "epoch": 2.7387619018554687e-05,
      "model_forward_time": 2.9083054065704346,
      "step": 28718
    },
    {
      "epoch": 2.7387619018554687e-05,
      "step": 28718,
      "training_step_time": 7.683148145675659
    },
    {
      "epoch": 2.7388572692871095e-05,
      "model_forward_time": 1.7766950130462646,
      "step": 28719
    },
    {
      "epoch": 2.7388572692871095e-05,
      "step": 28719,
      "training_step_time": 8.576687097549438
    },
    {
      "epoch": 2.73895263671875e-05,
      "grad_norm": 0.13013917207717896,
      "learning_rate": 8.539156504623369e-06,
      "loss": 0.0626,
      "step": 28720
    },
    {
      "epoch": 2.73895263671875e-05,
      "model_forward_time": 1.7483439445495605,
      "step": 28720
    },
    {
      "dynamics_loss_avg": 0.05644531138241291,
      "epoch": 2.73895263671875e-05,
      "step": 28720
    },
    {
      "action_loss_avg": 0.0016328782658092678,
      "epoch": 2.73895263671875e-05,
      "step": 28720
    },
    {
      "epoch": 2.73895263671875e-05,
      "step": 28720,
      "training_step_time": 7.163849830627441
    },
    {
      "epoch": 2.7390480041503906e-05,
      "model_forward_time": 1.7482240200042725,
      "step": 28721
    },
    {
      "epoch": 2.7390480041503906e-05,
      "step": 28721,
      "training_step_time": 7.752109050750732
    },
    {
      "epoch": 2.7391433715820314e-05,
      "model_forward_time": 1.7438223361968994,
      "step": 28722
    },
    {
      "epoch": 2.7391433715820314e-05,
      "step": 28722,
      "training_step_time": 7.622647762298584
    },
    {
      "epoch": 2.7392387390136718e-05,
      "model_forward_time": 1.7543203830718994,
      "step": 28723
    },
    {
      "epoch": 2.7392387390136718e-05,
      "step": 28723,
      "training_step_time": 7.525566101074219
    },
    {
      "epoch": 2.7393341064453125e-05,
      "model_forward_time": 1.7493515014648438,
      "step": 28724
    },
    {
      "epoch": 2.7393341064453125e-05,
      "step": 28724,
      "training_step_time": 10.671308279037476
    },
    {
      "epoch": 2.7394294738769533e-05,
      "model_forward_time": 1.754340648651123,
      "step": 28725
    },
    {
      "epoch": 2.7394294738769533e-05,
      "step": 28725,
      "training_step_time": 9.220078229904175
    },
    {
      "epoch": 2.7395248413085937e-05,
      "model_forward_time": 1.7466404438018799,
      "step": 28726
    },
    {
      "epoch": 2.7395248413085937e-05,
      "step": 28726,
      "training_step_time": 7.747192859649658
    },
    {
      "epoch": 2.7396202087402344e-05,
      "model_forward_time": 2.9127590656280518,
      "step": 28727
    },
    {
      "epoch": 2.7396202087402344e-05,
      "step": 28727,
      "training_step_time": 7.700388431549072
    },
    {
      "epoch": 2.739715576171875e-05,
      "model_forward_time": 1.7674546241760254,
      "step": 28728
    },
    {
      "epoch": 2.739715576171875e-05,
      "step": 28728,
      "training_step_time": 8.006843090057373
    },
    {
      "epoch": 2.7398109436035155e-05,
      "model_forward_time": 1.7522790431976318,
      "step": 28729
    },
    {
      "epoch": 2.7398109436035155e-05,
      "step": 28729,
      "training_step_time": 7.406405925750732
    },
    {
      "epoch": 2.7399063110351563e-05,
      "grad_norm": 0.11294440925121307,
      "learning_rate": 8.537988330605923e-06,
      "loss": 0.0619,
      "step": 28730
    },
    {
      "epoch": 2.7399063110351563e-05,
      "model_forward_time": 1.751375675201416,
      "step": 28730
    },
    {
      "dynamics_loss_avg": 0.07391264718025922,
      "epoch": 2.7399063110351563e-05,
      "step": 28730
    },
    {
      "action_loss_avg": 0.0024147035379428416,
      "epoch": 2.7399063110351563e-05,
      "step": 28730
    },
    {
      "epoch": 2.7399063110351563e-05,
      "step": 28730,
      "training_step_time": 7.914139986038208
    },
    {
      "epoch": 2.740001678466797e-05,
      "model_forward_time": 1.7561769485473633,
      "step": 28731
    },
    {
      "epoch": 2.740001678466797e-05,
      "step": 28731,
      "training_step_time": 7.7558913230896
    },
    {
      "epoch": 2.7400970458984374e-05,
      "model_forward_time": 1.7412500381469727,
      "step": 28732
    },
    {
      "epoch": 2.7400970458984374e-05,
      "step": 28732,
      "training_step_time": 8.213906049728394
    },
    {
      "epoch": 2.7401924133300782e-05,
      "model_forward_time": 1.7515947818756104,
      "step": 28733
    },
    {
      "epoch": 2.7401924133300782e-05,
      "step": 28733,
      "training_step_time": 8.633903980255127
    },
    {
      "epoch": 2.740287780761719e-05,
      "model_forward_time": 1.7478446960449219,
      "step": 28734
    },
    {
      "epoch": 2.740287780761719e-05,
      "step": 28734,
      "training_step_time": 8.688201427459717
    },
    {
      "epoch": 2.7403831481933593e-05,
      "model_forward_time": 1.7450847625732422,
      "step": 28735
    },
    {
      "epoch": 2.7403831481933593e-05,
      "step": 28735,
      "training_step_time": 7.803913831710815
    },
    {
      "epoch": 2.740478515625e-05,
      "model_forward_time": 2.887726306915283,
      "step": 28736
    },
    {
      "epoch": 2.740478515625e-05,
      "step": 28736,
      "training_step_time": 7.805246591567993
    },
    {
      "epoch": 2.7405738830566405e-05,
      "model_forward_time": 1.7476961612701416,
      "step": 28737
    },
    {
      "epoch": 2.7405738830566405e-05,
      "step": 28737,
      "training_step_time": 8.657640218734741
    },
    {
      "epoch": 2.7406692504882812e-05,
      "model_forward_time": 1.749753713607788,
      "step": 28738
    },
    {
      "epoch": 2.7406692504882812e-05,
      "step": 28738,
      "training_step_time": 7.458456754684448
    },
    {
      "epoch": 2.740764617919922e-05,
      "model_forward_time": 1.7497351169586182,
      "step": 28739
    },
    {
      "epoch": 2.740764617919922e-05,
      "step": 28739,
      "training_step_time": 8.49514651298523
    },
    {
      "epoch": 2.7408599853515624e-05,
      "grad_norm": 0.10611816495656967,
      "learning_rate": 8.536819769679393e-06,
      "loss": 0.061,
      "step": 28740
    },
    {
      "epoch": 2.7408599853515624e-05,
      "model_forward_time": 1.7435994148254395,
      "step": 28740
    },
    {
      "dynamics_loss_avg": 0.07332855556160212,
      "epoch": 2.7408599853515624e-05,
      "step": 28740
    },
    {
      "action_loss_avg": 0.0027637804974801838,
      "epoch": 2.7408599853515624e-05,
      "step": 28740
    },
    {
      "epoch": 2.7408599853515624e-05,
      "step": 28740,
      "training_step_time": 7.884096145629883
    },
    {
      "epoch": 2.740955352783203e-05,
      "model_forward_time": 1.74363374710083,
      "step": 28741
    },
    {
      "epoch": 2.740955352783203e-05,
      "step": 28741,
      "training_step_time": 7.6734583377838135
    },
    {
      "epoch": 2.741050720214844e-05,
      "model_forward_time": 1.738293170928955,
      "step": 28742
    },
    {
      "epoch": 2.741050720214844e-05,
      "step": 28742,
      "training_step_time": 7.7639477252960205
    },
    {
      "epoch": 2.7411460876464842e-05,
      "model_forward_time": 1.7483320236206055,
      "step": 28743
    },
    {
      "epoch": 2.7411460876464842e-05,
      "step": 28743,
      "training_step_time": 8.896599769592285
    },
    {
      "epoch": 2.741241455078125e-05,
      "model_forward_time": 1.7335302829742432,
      "step": 28744
    },
    {
      "epoch": 2.741241455078125e-05,
      "step": 28744,
      "training_step_time": 8.861800909042358
    },
    {
      "epoch": 2.7413368225097657e-05,
      "model_forward_time": 2.8895108699798584,
      "step": 28745
    },
    {
      "epoch": 2.7413368225097657e-05,
      "step": 28745,
      "training_step_time": 8.304409980773926
    },
    {
      "epoch": 2.741432189941406e-05,
      "model_forward_time": 1.7371771335601807,
      "step": 28746
    },
    {
      "epoch": 2.741432189941406e-05,
      "step": 28746,
      "training_step_time": 7.874910593032837
    },
    {
      "epoch": 2.741527557373047e-05,
      "model_forward_time": 1.7789180278778076,
      "step": 28747
    },
    {
      "epoch": 2.741527557373047e-05,
      "step": 28747,
      "training_step_time": 7.690483570098877
    },
    {
      "epoch": 2.7416229248046876e-05,
      "model_forward_time": 1.7476561069488525,
      "step": 28748
    },
    {
      "epoch": 2.7416229248046876e-05,
      "step": 28748,
      "training_step_time": 7.313584804534912
    },
    {
      "epoch": 2.741718292236328e-05,
      "model_forward_time": 1.7463762760162354,
      "step": 28749
    },
    {
      "epoch": 2.741718292236328e-05,
      "step": 28749,
      "training_step_time": 7.878274202346802
    },
    {
      "epoch": 2.7418136596679688e-05,
      "grad_norm": 0.2230745255947113,
      "learning_rate": 8.53565082197157e-06,
      "loss": 0.064,
      "step": 28750
    },
    {
      "epoch": 2.7418136596679688e-05,
      "model_forward_time": 1.743027925491333,
      "step": 28750
    },
    {
      "dynamics_loss_avg": 0.05258856005966663,
      "epoch": 2.7418136596679688e-05,
      "step": 28750
    },
    {
      "action_loss_avg": 0.0016623503877781331,
      "epoch": 2.7418136596679688e-05,
      "step": 28750
    },
    {
      "epoch": 2.7418136596679688e-05,
      "step": 28750,
      "training_step_time": 7.737508535385132
    },
    {
      "epoch": 2.7419090270996095e-05,
      "model_forward_time": 1.746469497680664,
      "step": 28751
    },
    {
      "epoch": 2.7419090270996095e-05,
      "step": 28751,
      "training_step_time": 7.799896001815796
    },
    {
      "epoch": 2.74200439453125e-05,
      "model_forward_time": 1.7493524551391602,
      "step": 28752
    },
    {
      "epoch": 2.74200439453125e-05,
      "step": 28752,
      "training_step_time": 9.71044373512268
    },
    {
      "epoch": 2.7420997619628906e-05,
      "model_forward_time": 1.7512397766113281,
      "step": 28753
    },
    {
      "epoch": 2.7420997619628906e-05,
      "step": 28753,
      "training_step_time": 8.531614065170288
    },
    {
      "epoch": 2.7421951293945314e-05,
      "model_forward_time": 1.7318742275238037,
      "step": 28754
    },
    {
      "epoch": 2.7421951293945314e-05,
      "step": 28754,
      "training_step_time": 8.336012840270996
    },
    {
      "epoch": 2.7422904968261718e-05,
      "model_forward_time": 1.7691702842712402,
      "step": 28755
    },
    {
      "epoch": 2.7422904968261718e-05,
      "step": 28755,
      "training_step_time": 8.814100503921509
    },
    {
      "epoch": 2.7423858642578125e-05,
      "model_forward_time": 1.789794921875,
      "step": 28756
    },
    {
      "epoch": 2.7423858642578125e-05,
      "step": 28756,
      "training_step_time": 8.11879277229309
    },
    {
      "epoch": 2.7424812316894533e-05,
      "model_forward_time": 1.7514560222625732,
      "step": 28757
    },
    {
      "epoch": 2.7424812316894533e-05,
      "step": 28757,
      "training_step_time": 7.410225868225098
    },
    {
      "epoch": 2.7425765991210937e-05,
      "model_forward_time": 1.757284164428711,
      "step": 28758
    },
    {
      "epoch": 2.7425765991210937e-05,
      "step": 28758,
      "training_step_time": 7.903015375137329
    },
    {
      "epoch": 2.7426719665527344e-05,
      "model_forward_time": 1.7475378513336182,
      "step": 28759
    },
    {
      "epoch": 2.7426719665527344e-05,
      "step": 28759,
      "training_step_time": 7.765779733657837
    },
    {
      "epoch": 2.742767333984375e-05,
      "grad_norm": 0.17761453986167908,
      "learning_rate": 8.534481487610289e-06,
      "loss": 0.0606,
      "step": 28760
    },
    {
      "epoch": 2.742767333984375e-05,
      "model_forward_time": 1.7496614456176758,
      "step": 28760
    },
    {
      "dynamics_loss_avg": 0.06419871132820845,
      "epoch": 2.742767333984375e-05,
      "step": 28760
    },
    {
      "action_loss_avg": 0.0020179734099656345,
      "epoch": 2.742767333984375e-05,
      "step": 28760
    },
    {
      "epoch": 2.742767333984375e-05,
      "step": 28760,
      "training_step_time": 7.842661380767822
    },
    {
      "epoch": 2.7428627014160156e-05,
      "model_forward_time": 1.744791030883789,
      "step": 28761
    },
    {
      "epoch": 2.7428627014160156e-05,
      "step": 28761,
      "training_step_time": 8.382593870162964
    },
    {
      "epoch": 2.7429580688476563e-05,
      "model_forward_time": 1.7530195713043213,
      "step": 28762
    },
    {
      "epoch": 2.7429580688476563e-05,
      "step": 28762,
      "training_step_time": 8.77848505973816
    },
    {
      "epoch": 2.743053436279297e-05,
      "model_forward_time": 1.7477340698242188,
      "step": 28763
    },
    {
      "epoch": 2.743053436279297e-05,
      "step": 28763,
      "training_step_time": 8.31742000579834
    },
    {
      "epoch": 2.7431488037109374e-05,
      "model_forward_time": 2.925103187561035,
      "step": 28764
    },
    {
      "epoch": 2.7431488037109374e-05,
      "step": 28764,
      "training_step_time": 7.793558359146118
    },
    {
      "epoch": 2.7432441711425782e-05,
      "model_forward_time": 1.754793643951416,
      "step": 28765
    },
    {
      "epoch": 2.7432441711425782e-05,
      "step": 28765,
      "training_step_time": 8.914480686187744
    },
    {
      "epoch": 2.7433395385742186e-05,
      "model_forward_time": 1.7520461082458496,
      "step": 28766
    },
    {
      "epoch": 2.7433395385742186e-05,
      "step": 28766,
      "training_step_time": 7.341795921325684
    },
    {
      "epoch": 2.7434349060058593e-05,
      "model_forward_time": 1.7478077411651611,
      "step": 28767
    },
    {
      "epoch": 2.7434349060058593e-05,
      "step": 28767,
      "training_step_time": 7.812998294830322
    },
    {
      "epoch": 2.7435302734375e-05,
      "model_forward_time": 1.7457754611968994,
      "step": 28768
    },
    {
      "epoch": 2.7435302734375e-05,
      "step": 28768,
      "training_step_time": 7.588990926742554
    },
    {
      "epoch": 2.7436256408691405e-05,
      "model_forward_time": 1.753687858581543,
      "step": 28769
    },
    {
      "epoch": 2.7436256408691405e-05,
      "step": 28769,
      "training_step_time": 7.534221649169922
    },
    {
      "epoch": 2.7437210083007812e-05,
      "grad_norm": 0.25964632630348206,
      "learning_rate": 8.533311766723428e-06,
      "loss": 0.0642,
      "step": 28770
    },
    {
      "epoch": 2.7437210083007812e-05,
      "model_forward_time": 1.753593921661377,
      "step": 28770
    },
    {
      "dynamics_loss_avg": 0.05128184985369444,
      "epoch": 2.7437210083007812e-05,
      "step": 28770
    },
    {
      "action_loss_avg": 0.0017320132872555404,
      "epoch": 2.7437210083007812e-05,
      "step": 28770
    },
    {
      "epoch": 2.7437210083007812e-05,
      "step": 28770,
      "training_step_time": 8.413711071014404
    },
    {
      "epoch": 2.743816375732422e-05,
      "model_forward_time": 1.7537109851837158,
      "step": 28771
    },
    {
      "epoch": 2.743816375732422e-05,
      "step": 28771,
      "training_step_time": 8.186710119247437
    },
    {
      "epoch": 2.7439117431640624e-05,
      "model_forward_time": 1.804746150970459,
      "step": 28772
    },
    {
      "epoch": 2.7439117431640624e-05,
      "step": 28772,
      "training_step_time": 8.742665529251099
    },
    {
      "epoch": 2.744007110595703e-05,
      "model_forward_time": 2.9265284538269043,
      "step": 28773
    },
    {
      "epoch": 2.744007110595703e-05,
      "step": 28773,
      "training_step_time": 8.76215386390686
    },
    {
      "epoch": 2.744102478027344e-05,
      "model_forward_time": 1.7629427909851074,
      "step": 28774
    },
    {
      "epoch": 2.744102478027344e-05,
      "step": 28774,
      "training_step_time": 7.9031901359558105
    },
    {
      "epoch": 2.7441978454589842e-05,
      "model_forward_time": 1.8273956775665283,
      "step": 28775
    },
    {
      "epoch": 2.7441978454589842e-05,
      "step": 28775,
      "training_step_time": 7.711178541183472
    },
    {
      "epoch": 2.744293212890625e-05,
      "model_forward_time": 1.7595994472503662,
      "step": 28776
    },
    {
      "epoch": 2.744293212890625e-05,
      "step": 28776,
      "training_step_time": 8.754372835159302
    },
    {
      "epoch": 2.7443885803222657e-05,
      "model_forward_time": 1.7565364837646484,
      "step": 28777
    },
    {
      "epoch": 2.7443885803222657e-05,
      "step": 28777,
      "training_step_time": 7.604955673217773
    },
    {
      "epoch": 2.744483947753906e-05,
      "model_forward_time": 1.754547119140625,
      "step": 28778
    },
    {
      "epoch": 2.744483947753906e-05,
      "step": 28778,
      "training_step_time": 7.643332004547119
    },
    {
      "epoch": 2.744579315185547e-05,
      "model_forward_time": 1.7584764957427979,
      "step": 28779
    },
    {
      "epoch": 2.744579315185547e-05,
      "step": 28779,
      "training_step_time": 8.625038385391235
    },
    {
      "epoch": 2.7446746826171876e-05,
      "grad_norm": 0.1649562120437622,
      "learning_rate": 8.532141659438902e-06,
      "loss": 0.0624,
      "step": 28780
    },
    {
      "epoch": 2.7446746826171876e-05,
      "model_forward_time": 1.910400390625,
      "step": 28780
    },
    {
      "dynamics_loss_avg": 0.07933737225830555,
      "epoch": 2.7446746826171876e-05,
      "step": 28780
    },
    {
      "action_loss_avg": 0.0025311885692644864,
      "epoch": 2.7446746826171876e-05,
      "step": 28780
    },
    {
      "epoch": 2.7446746826171876e-05,
      "step": 28780,
      "training_step_time": 8.786009550094604
    },
    {
      "epoch": 2.744770050048828e-05,
      "model_forward_time": 1.9209764003753662,
      "step": 28781
    },
    {
      "epoch": 2.744770050048828e-05,
      "step": 28781,
      "training_step_time": 9.673792123794556
    },
    {
      "epoch": 2.7448654174804688e-05,
      "model_forward_time": 3.028308629989624,
      "step": 28782
    },
    {
      "epoch": 2.7448654174804688e-05,
      "step": 28782,
      "training_step_time": 7.879990339279175
    },
    {
      "epoch": 2.7449607849121095e-05,
      "model_forward_time": 1.571425199508667,
      "step": 28783
    },
    {
      "epoch": 2.7449607849121095e-05,
      "step": 28783,
      "training_step_time": 7.83365535736084
    },
    {
      "epoch": 2.74505615234375e-05,
      "model_forward_time": 1.9197008609771729,
      "step": 28784
    },
    {
      "epoch": 2.74505615234375e-05,
      "step": 28784,
      "training_step_time": 7.6875128746032715
    },
    {
      "epoch": 2.7451515197753906e-05,
      "model_forward_time": 1.9223589897155762,
      "step": 28785
    },
    {
      "epoch": 2.7451515197753906e-05,
      "step": 28785,
      "training_step_time": 7.2863993644714355
    },
    {
      "epoch": 2.7452468872070314e-05,
      "model_forward_time": 1.9327189922332764,
      "step": 28786
    },
    {
      "epoch": 2.7452468872070314e-05,
      "step": 28786,
      "training_step_time": 7.789731979370117
    },
    {
      "epoch": 2.7453422546386718e-05,
      "model_forward_time": 1.9140527248382568,
      "step": 28787
    },
    {
      "epoch": 2.7453422546386718e-05,
      "step": 28787,
      "training_step_time": 8.576624870300293
    },
    {
      "epoch": 2.7454376220703125e-05,
      "model_forward_time": 1.89540433883667,
      "step": 28788
    },
    {
      "epoch": 2.7454376220703125e-05,
      "step": 28788,
      "training_step_time": 8.69571566581726
    },
    {
      "epoch": 2.7455329895019533e-05,
      "model_forward_time": 1.9084084033966064,
      "step": 28789
    },
    {
      "epoch": 2.7455329895019533e-05,
      "step": 28789,
      "training_step_time": 7.791245698928833
    },
    {
      "epoch": 2.7456283569335937e-05,
      "grad_norm": 0.13307175040245056,
      "learning_rate": 8.530971165884676e-06,
      "loss": 0.0646,
      "step": 28790
    },
    {
      "epoch": 2.7456283569335937e-05,
      "model_forward_time": 1.904435157775879,
      "step": 28790
    },
    {
      "dynamics_loss_avg": 0.06259132027626038,
      "epoch": 2.7456283569335937e-05,
      "step": 28790
    },
    {
      "action_loss_avg": 0.0016563816054258496,
      "epoch": 2.7456283569335937e-05,
      "step": 28790
    },
    {
      "epoch": 2.7456283569335937e-05,
      "step": 28790,
      "training_step_time": 8.39767837524414
    },
    {
      "epoch": 2.7457237243652344e-05,
      "model_forward_time": 2.968512535095215,
      "step": 28791
    },
    {
      "epoch": 2.7457237243652344e-05,
      "step": 28791,
      "training_step_time": 8.517133474349976
    },
    {
      "epoch": 2.745819091796875e-05,
      "model_forward_time": 1.937312126159668,
      "step": 28792
    },
    {
      "epoch": 2.745819091796875e-05,
      "step": 28792,
      "training_step_time": 8.58368182182312
    },
    {
      "epoch": 2.7459144592285156e-05,
      "model_forward_time": 1.9365298748016357,
      "step": 28793
    },
    {
      "epoch": 2.7459144592285156e-05,
      "step": 28793,
      "training_step_time": 7.864950180053711
    },
    {
      "epoch": 2.7460098266601563e-05,
      "model_forward_time": 1.9341726303100586,
      "step": 28794
    },
    {
      "epoch": 2.7460098266601563e-05,
      "step": 28794,
      "training_step_time": 8.327386617660522
    },
    {
      "epoch": 2.746105194091797e-05,
      "model_forward_time": 1.7874960899353027,
      "step": 28795
    },
    {
      "epoch": 2.746105194091797e-05,
      "step": 28795,
      "training_step_time": 7.724990129470825
    },
    {
      "epoch": 2.7462005615234374e-05,
      "model_forward_time": 1.7450411319732666,
      "step": 28796
    },
    {
      "epoch": 2.7462005615234374e-05,
      "step": 28796,
      "training_step_time": 7.632016897201538
    },
    {
      "epoch": 2.7462959289550782e-05,
      "model_forward_time": 1.748786449432373,
      "step": 28797
    },
    {
      "epoch": 2.7462959289550782e-05,
      "step": 28797,
      "training_step_time": 7.688993215560913
    },
    {
      "epoch": 2.7463912963867186e-05,
      "model_forward_time": 1.7392377853393555,
      "step": 28798
    },
    {
      "epoch": 2.7463912963867186e-05,
      "step": 28798,
      "training_step_time": 7.743653059005737
    },
    {
      "epoch": 2.7464866638183593e-05,
      "model_forward_time": 1.7444562911987305,
      "step": 28799
    },
    {
      "epoch": 2.7464866638183593e-05,
      "step": 28799,
      "training_step_time": 7.790262222290039
    },
    {
      "epoch": 2.74658203125e-05,
      "grad_norm": 0.14890894293785095,
      "learning_rate": 8.529800286188752e-06,
      "loss": 0.0621,
      "step": 28800
    },
    {
      "epoch": 2.74658203125e-05,
      "model_forward_time": 1.7539162635803223,
      "step": 28800
    },
    {
      "dynamics_loss_avg": 0.06274452824145556,
      "epoch": 2.74658203125e-05,
      "step": 28800
    },
    {
      "action_loss_avg": 0.0018515406060032546,
      "epoch": 2.74658203125e-05,
      "step": 28800
    },
    {
      "epoch": 2.74658203125e-05,
      "step": 28800,
      "training_step_time": 8.170456647872925
    },
    {
      "epoch": 2.7466773986816405e-05,
      "model_forward_time": 2.8964552879333496,
      "step": 28801
    },
    {
      "epoch": 2.7466773986816405e-05,
      "step": 28801,
      "training_step_time": 8.606561660766602
    },
    {
      "epoch": 2.7467727661132812e-05,
      "model_forward_time": 1.757528305053711,
      "step": 28802
    },
    {
      "epoch": 2.7467727661132812e-05,
      "step": 28802,
      "training_step_time": 7.947945833206177
    },
    {
      "epoch": 2.746868133544922e-05,
      "model_forward_time": 1.7459347248077393,
      "step": 28803
    },
    {
      "epoch": 2.746868133544922e-05,
      "step": 28803,
      "training_step_time": 7.567893981933594
    },
    {
      "epoch": 2.7469635009765624e-05,
      "model_forward_time": 1.7507829666137695,
      "step": 28804
    },
    {
      "epoch": 2.7469635009765624e-05,
      "step": 28804,
      "training_step_time": 7.717356204986572
    },
    {
      "epoch": 2.747058868408203e-05,
      "model_forward_time": 1.7409751415252686,
      "step": 28805
    },
    {
      "epoch": 2.747058868408203e-05,
      "step": 28805,
      "training_step_time": 8.435896873474121
    },
    {
      "epoch": 2.747154235839844e-05,
      "model_forward_time": 1.7476673126220703,
      "step": 28806
    },
    {
      "epoch": 2.747154235839844e-05,
      "step": 28806,
      "training_step_time": 8.700510501861572
    },
    {
      "epoch": 2.7472496032714842e-05,
      "model_forward_time": 1.7517762184143066,
      "step": 28807
    },
    {
      "epoch": 2.7472496032714842e-05,
      "step": 28807,
      "training_step_time": 8.367217540740967
    },
    {
      "epoch": 2.747344970703125e-05,
      "model_forward_time": 1.7418954372406006,
      "step": 28808
    },
    {
      "epoch": 2.747344970703125e-05,
      "step": 28808,
      "training_step_time": 8.28059434890747
    },
    {
      "epoch": 2.7474403381347657e-05,
      "model_forward_time": 1.7519681453704834,
      "step": 28809
    },
    {
      "epoch": 2.7474403381347657e-05,
      "step": 28809,
      "training_step_time": 8.189128637313843
    },
    {
      "epoch": 2.747535705566406e-05,
      "grad_norm": 0.1315270960330963,
      "learning_rate": 8.528629020479176e-06,
      "loss": 0.0658,
      "step": 28810
    },
    {
      "epoch": 2.747535705566406e-05,
      "model_forward_time": 2.8890273571014404,
      "step": 28810
    },
    {
      "dynamics_loss_avg": 0.07091436516493559,
      "epoch": 2.747535705566406e-05,
      "step": 28810
    },
    {
      "action_loss_avg": 0.002148643517284654,
      "epoch": 2.747535705566406e-05,
      "step": 28810
    },
    {
      "epoch": 2.747535705566406e-05,
      "step": 28810,
      "training_step_time": 8.321897029876709
    },
    {
      "epoch": 2.747631072998047e-05,
      "model_forward_time": 1.753187894821167,
      "step": 28811
    },
    {
      "epoch": 2.747631072998047e-05,
      "step": 28811,
      "training_step_time": 7.8635032176971436
    },
    {
      "epoch": 2.7477264404296876e-05,
      "model_forward_time": 1.7349953651428223,
      "step": 28812
    },
    {
      "epoch": 2.7477264404296876e-05,
      "step": 28812,
      "training_step_time": 7.5747222900390625
    },
    {
      "epoch": 2.747821807861328e-05,
      "model_forward_time": 1.70603609085083,
      "step": 28813
    },
    {
      "epoch": 2.747821807861328e-05,
      "step": 28813,
      "training_step_time": 7.75750994682312
    },
    {
      "epoch": 2.7479171752929688e-05,
      "model_forward_time": 1.7236719131469727,
      "step": 28814
    },
    {
      "epoch": 2.7479171752929688e-05,
      "step": 28814,
      "training_step_time": 7.538761377334595
    },
    {
      "epoch": 2.7480125427246095e-05,
      "model_forward_time": 1.7250876426696777,
      "step": 28815
    },
    {
      "epoch": 2.7480125427246095e-05,
      "step": 28815,
      "training_step_time": 7.659365177154541
    },
    {
      "epoch": 2.74810791015625e-05,
      "model_forward_time": 1.7211031913757324,
      "step": 28816
    },
    {
      "epoch": 2.74810791015625e-05,
      "step": 28816,
      "training_step_time": 8.301080703735352
    },
    {
      "epoch": 2.7482032775878906e-05,
      "model_forward_time": 1.715569019317627,
      "step": 28817
    },
    {
      "epoch": 2.7482032775878906e-05,
      "step": 28817,
      "training_step_time": 7.800281047821045
    },
    {
      "epoch": 2.7482986450195314e-05,
      "model_forward_time": 1.7180898189544678,
      "step": 28818
    },
    {
      "epoch": 2.7482986450195314e-05,
      "step": 28818,
      "training_step_time": 8.639917850494385
    },
    {
      "epoch": 2.7483940124511718e-05,
      "model_forward_time": 2.894854784011841,
      "step": 28819
    },
    {
      "epoch": 2.7483940124511718e-05,
      "step": 28819,
      "training_step_time": 8.190271615982056
    },
    {
      "epoch": 2.7484893798828125e-05,
      "grad_norm": 0.15430788695812225,
      "learning_rate": 8.527457368884033e-06,
      "loss": 0.0656,
      "step": 28820
    },
    {
      "epoch": 2.7484893798828125e-05,
      "model_forward_time": 1.72159743309021,
      "step": 28820
    },
    {
      "dynamics_loss_avg": 0.06015551500022411,
      "epoch": 2.7484893798828125e-05,
      "step": 28820
    },
    {
      "action_loss_avg": 0.0017480822280049324,
      "epoch": 2.7484893798828125e-05,
      "step": 28820
    },
    {
      "epoch": 2.7484893798828125e-05,
      "step": 28820,
      "training_step_time": 7.621569633483887
    },
    {
      "epoch": 2.7485847473144533e-05,
      "model_forward_time": 1.7353034019470215,
      "step": 28821
    },
    {
      "epoch": 2.7485847473144533e-05,
      "step": 28821,
      "training_step_time": 7.744548797607422
    },
    {
      "epoch": 2.7486801147460937e-05,
      "model_forward_time": 1.7234227657318115,
      "step": 28822
    },
    {
      "epoch": 2.7486801147460937e-05,
      "step": 28822,
      "training_step_time": 7.5828447341918945
    },
    {
      "epoch": 2.7487754821777344e-05,
      "model_forward_time": 1.722945213317871,
      "step": 28823
    },
    {
      "epoch": 2.7487754821777344e-05,
      "step": 28823,
      "training_step_time": 7.347034454345703
    },
    {
      "epoch": 2.748870849609375e-05,
      "model_forward_time": 1.7140252590179443,
      "step": 28824
    },
    {
      "epoch": 2.748870849609375e-05,
      "step": 28824,
      "training_step_time": 7.491818428039551
    },
    {
      "epoch": 2.7489662170410156e-05,
      "model_forward_time": 1.7219908237457275,
      "step": 28825
    },
    {
      "epoch": 2.7489662170410156e-05,
      "step": 28825,
      "training_step_time": 7.631479024887085
    },
    {
      "epoch": 2.7490615844726563e-05,
      "model_forward_time": 1.0257248878479004,
      "step": 28826
    },
    {
      "epoch": 2.7490615844726563e-05,
      "step": 28826,
      "training_step_time": 7.757333278656006
    },
    {
      "epoch": 2.749156951904297e-05,
      "model_forward_time": 1.7209446430206299,
      "step": 28827
    },
    {
      "epoch": 2.749156951904297e-05,
      "step": 28827,
      "training_step_time": 9.05794644355774
    },
    {
      "epoch": 2.7492523193359374e-05,
      "model_forward_time": 2.8256659507751465,
      "step": 28828
    },
    {
      "epoch": 2.7492523193359374e-05,
      "step": 28828,
      "training_step_time": 7.628270864486694
    },
    {
      "epoch": 2.7493476867675782e-05,
      "model_forward_time": 1.7196669578552246,
      "step": 28829
    },
    {
      "epoch": 2.7493476867675782e-05,
      "step": 28829,
      "training_step_time": 7.607149600982666
    },
    {
      "epoch": 2.7494430541992186e-05,
      "grad_norm": 0.1835385262966156,
      "learning_rate": 8.52628533153146e-06,
      "loss": 0.0649,
      "step": 28830
    },
    {
      "epoch": 2.7494430541992186e-05,
      "model_forward_time": 1.7370648384094238,
      "step": 28830
    },
    {
      "dynamics_loss_avg": 0.06346879787743091,
      "epoch": 2.7494430541992186e-05,
      "step": 28830
    },
    {
      "action_loss_avg": 0.0017235586419701576,
      "epoch": 2.7494430541992186e-05,
      "step": 28830
    },
    {
      "epoch": 2.7494430541992186e-05,
      "step": 28830,
      "training_step_time": 7.7577972412109375
    },
    {
      "epoch": 2.7495384216308593e-05,
      "model_forward_time": 1.7303121089935303,
      "step": 28831
    },
    {
      "epoch": 2.7495384216308593e-05,
      "step": 28831,
      "training_step_time": 7.625094413757324
    },
    {
      "epoch": 2.7496337890625e-05,
      "model_forward_time": 1.7257380485534668,
      "step": 28832
    },
    {
      "epoch": 2.7496337890625e-05,
      "step": 28832,
      "training_step_time": 7.793649435043335
    },
    {
      "epoch": 2.7497291564941405e-05,
      "model_forward_time": 1.720116138458252,
      "step": 28833
    },
    {
      "epoch": 2.7497291564941405e-05,
      "step": 28833,
      "training_step_time": 7.478803396224976
    },
    {
      "epoch": 2.7498245239257812e-05,
      "model_forward_time": 1.7181446552276611,
      "step": 28834
    },
    {
      "epoch": 2.7498245239257812e-05,
      "step": 28834,
      "training_step_time": 7.57986044883728
    },
    {
      "epoch": 2.749919891357422e-05,
      "model_forward_time": 1.7224667072296143,
      "step": 28835
    },
    {
      "epoch": 2.749919891357422e-05,
      "step": 28835,
      "training_step_time": 9.241275548934937
    },
    {
      "epoch": 2.7500152587890624e-05,
      "model_forward_time": 1.7198779582977295,
      "step": 28836
    },
    {
      "epoch": 2.7500152587890624e-05,
      "step": 28836,
      "training_step_time": 9.47936487197876
    },
    {
      "epoch": 2.750110626220703e-05,
      "model_forward_time": 1.7235462665557861,
      "step": 28837
    },
    {
      "epoch": 2.750110626220703e-05,
      "step": 28837,
      "training_step_time": 8.502264499664307
    },
    {
      "epoch": 2.750205993652344e-05,
      "model_forward_time": 2.905724048614502,
      "step": 28838
    },
    {
      "epoch": 2.750205993652344e-05,
      "step": 28838,
      "training_step_time": 7.661660671234131
    },
    {
      "epoch": 2.7503013610839842e-05,
      "model_forward_time": 1.728560447692871,
      "step": 28839
    },
    {
      "epoch": 2.7503013610839842e-05,
      "step": 28839,
      "training_step_time": 7.897518634796143
    },
    {
      "epoch": 2.750396728515625e-05,
      "grad_norm": 0.1479046493768692,
      "learning_rate": 8.525112908549622e-06,
      "loss": 0.0642,
      "step": 28840
    },
    {
      "epoch": 2.750396728515625e-05,
      "model_forward_time": 1.7263898849487305,
      "step": 28840
    },
    {
      "dynamics_loss_avg": 0.08013410121202469,
      "epoch": 2.750396728515625e-05,
      "step": 28840
    },
    {
      "action_loss_avg": 0.002706129045691341,
      "epoch": 2.750396728515625e-05,
      "step": 28840
    },
    {
      "epoch": 2.750396728515625e-05,
      "step": 28840,
      "training_step_time": 7.54357385635376
    },
    {
      "epoch": 2.7504920959472657e-05,
      "model_forward_time": 1.7256348133087158,
      "step": 28841
    },
    {
      "epoch": 2.7504920959472657e-05,
      "step": 28841,
      "training_step_time": 7.735010147094727
    },
    {
      "epoch": 2.750587463378906e-05,
      "model_forward_time": 1.7205414772033691,
      "step": 28842
    },
    {
      "epoch": 2.750587463378906e-05,
      "step": 28842,
      "training_step_time": 7.4721667766571045
    },
    {
      "epoch": 2.750682830810547e-05,
      "model_forward_time": 1.7242577075958252,
      "step": 28843
    },
    {
      "epoch": 2.750682830810547e-05,
      "step": 28843,
      "training_step_time": 8.098112106323242
    },
    {
      "epoch": 2.7507781982421876e-05,
      "model_forward_time": 1.739961862564087,
      "step": 28844
    },
    {
      "epoch": 2.7507781982421876e-05,
      "step": 28844,
      "training_step_time": 9.329081296920776
    },
    {
      "epoch": 2.750873565673828e-05,
      "model_forward_time": 1.7429702281951904,
      "step": 28845
    },
    {
      "epoch": 2.750873565673828e-05,
      "step": 28845,
      "training_step_time": 8.405333518981934
    },
    {
      "epoch": 2.7509689331054688e-05,
      "model_forward_time": 1.740809679031372,
      "step": 28846
    },
    {
      "epoch": 2.7509689331054688e-05,
      "step": 28846,
      "training_step_time": 8.050708293914795
    },
    {
      "epoch": 2.7510643005371095e-05,
      "model_forward_time": 2.872748613357544,
      "step": 28847
    },
    {
      "epoch": 2.7510643005371095e-05,
      "step": 28847,
      "training_step_time": 7.634628534317017
    },
    {
      "epoch": 2.75115966796875e-05,
      "model_forward_time": 1.7393369674682617,
      "step": 28848
    },
    {
      "epoch": 2.75115966796875e-05,
      "step": 28848,
      "training_step_time": 7.771108865737915
    },
    {
      "epoch": 2.7512550354003906e-05,
      "model_forward_time": 1.7392010688781738,
      "step": 28849
    },
    {
      "epoch": 2.7512550354003906e-05,
      "step": 28849,
      "training_step_time": 7.603283166885376
    },
    {
      "epoch": 2.7513504028320314e-05,
      "grad_norm": 0.1255629062652588,
      "learning_rate": 8.523940100066735e-06,
      "loss": 0.0608,
      "step": 28850
    },
    {
      "epoch": 2.7513504028320314e-05,
      "model_forward_time": 1.7220065593719482,
      "step": 28850
    },
    {
      "dynamics_loss_avg": 0.07086958698928356,
      "epoch": 2.7513504028320314e-05,
      "step": 28850
    },
    {
      "action_loss_avg": 0.0017879000428365544,
      "epoch": 2.7513504028320314e-05,
      "step": 28850
    },
    {
      "epoch": 2.7513504028320314e-05,
      "step": 28850,
      "training_step_time": 7.793053865432739
    },
    {
      "epoch": 2.7514457702636718e-05,
      "model_forward_time": 1.733915090560913,
      "step": 28851
    },
    {
      "epoch": 2.7514457702636718e-05,
      "step": 28851,
      "training_step_time": 8.00028395652771
    },
    {
      "epoch": 2.7515411376953125e-05,
      "model_forward_time": 1.7413358688354492,
      "step": 28852
    },
    {
      "epoch": 2.7515411376953125e-05,
      "step": 28852,
      "training_step_time": 7.647444009780884
    },
    {
      "epoch": 2.7516365051269533e-05,
      "model_forward_time": 1.742572546005249,
      "step": 28853
    },
    {
      "epoch": 2.7516365051269533e-05,
      "step": 28853,
      "training_step_time": 8.394293308258057
    },
    {
      "epoch": 2.7517318725585937e-05,
      "model_forward_time": 1.7434601783752441,
      "step": 28854
    },
    {
      "epoch": 2.7517318725585937e-05,
      "step": 28854,
      "training_step_time": 7.789679288864136
    },
    {
      "epoch": 2.7518272399902344e-05,
      "model_forward_time": 1.0294115543365479,
      "step": 28855
    },
    {
      "epoch": 2.7518272399902344e-05,
      "step": 28855,
      "training_step_time": 7.600584030151367
    },
    {
      "epoch": 2.751922607421875e-05,
      "model_forward_time": 1.8078956604003906,
      "step": 28856
    },
    {
      "epoch": 2.751922607421875e-05,
      "step": 28856,
      "training_step_time": 8.14310097694397
    },
    {
      "epoch": 2.7520179748535156e-05,
      "model_forward_time": 1.7406692504882812,
      "step": 28857
    },
    {
      "epoch": 2.7520179748535156e-05,
      "step": 28857,
      "training_step_time": 7.801162958145142
    },
    {
      "epoch": 2.7521133422851563e-05,
      "model_forward_time": 1.7402772903442383,
      "step": 28858
    },
    {
      "epoch": 2.7521133422851563e-05,
      "step": 28858,
      "training_step_time": 7.6638712882995605
    },
    {
      "epoch": 2.752208709716797e-05,
      "model_forward_time": 1.3781213760375977,
      "step": 28859
    },
    {
      "epoch": 2.752208709716797e-05,
      "step": 28859,
      "training_step_time": 8.101325511932373
    },
    {
      "epoch": 2.7523040771484374e-05,
      "grad_norm": 0.13727113604545593,
      "learning_rate": 8.52276690621106e-06,
      "loss": 0.0592,
      "step": 28860
    },
    {
      "epoch": 2.7523040771484374e-05,
      "model_forward_time": 1.725585699081421,
      "step": 28860
    },
    {
      "dynamics_loss_avg": 0.05271125864237547,
      "epoch": 2.7523040771484374e-05,
      "step": 28860
    },
    {
      "action_loss_avg": 0.002414898993447423,
      "epoch": 2.7523040771484374e-05,
      "step": 28860
    },
    {
      "epoch": 2.7523040771484374e-05,
      "step": 28860,
      "training_step_time": 7.588911533355713
    },
    {
      "epoch": 2.7523994445800782e-05,
      "model_forward_time": 1.7380938529968262,
      "step": 28861
    },
    {
      "epoch": 2.7523994445800782e-05,
      "step": 28861,
      "training_step_time": 7.627344846725464
    },
    {
      "epoch": 2.7524948120117186e-05,
      "model_forward_time": 1.7387866973876953,
      "step": 28862
    },
    {
      "epoch": 2.7524948120117186e-05,
      "step": 28862,
      "training_step_time": 8.385910511016846
    },
    {
      "epoch": 2.7525901794433593e-05,
      "model_forward_time": 1.7414178848266602,
      "step": 28863
    },
    {
      "epoch": 2.7525901794433593e-05,
      "step": 28863,
      "training_step_time": 8.518607139587402
    },
    {
      "epoch": 2.752685546875e-05,
      "model_forward_time": 1.7392487525939941,
      "step": 28864
    },
    {
      "epoch": 2.752685546875e-05,
      "step": 28864,
      "training_step_time": 9.430763006210327
    },
    {
      "epoch": 2.7527809143066405e-05,
      "model_forward_time": 1.4383981227874756,
      "step": 28865
    },
    {
      "epoch": 2.7527809143066405e-05,
      "step": 28865,
      "training_step_time": 8.249475479125977
    },
    {
      "epoch": 2.7528762817382812e-05,
      "model_forward_time": 2.8466286659240723,
      "step": 28866
    },
    {
      "epoch": 2.7528762817382812e-05,
      "step": 28866,
      "training_step_time": 7.9093592166900635
    },
    {
      "epoch": 2.752971649169922e-05,
      "model_forward_time": 1.7352449893951416,
      "step": 28867
    },
    {
      "epoch": 2.752971649169922e-05,
      "step": 28867,
      "training_step_time": 7.726568698883057
    },
    {
      "epoch": 2.7530670166015624e-05,
      "model_forward_time": 1.7439546585083008,
      "step": 28868
    },
    {
      "epoch": 2.7530670166015624e-05,
      "step": 28868,
      "training_step_time": 7.774589776992798
    },
    {
      "epoch": 2.753162384033203e-05,
      "model_forward_time": 1.747638463973999,
      "step": 28869
    },
    {
      "epoch": 2.753162384033203e-05,
      "step": 28869,
      "training_step_time": 7.59378981590271
    },
    {
      "epoch": 2.753257751464844e-05,
      "grad_norm": 0.1439456045627594,
      "learning_rate": 8.52159332711089e-06,
      "loss": 0.0589,
      "step": 28870
    },
    {
      "epoch": 2.753257751464844e-05,
      "model_forward_time": 1.7254674434661865,
      "step": 28870
    },
    {
      "dynamics_loss_avg": 0.06620540656149387,
      "epoch": 2.753257751464844e-05,
      "step": 28870
    },
    {
      "action_loss_avg": 0.0018074147403240204,
      "epoch": 2.753257751464844e-05,
      "step": 28870
    },
    {
      "epoch": 2.753257751464844e-05,
      "step": 28870,
      "training_step_time": 8.225024700164795
    },
    {
      "epoch": 2.7533531188964842e-05,
      "model_forward_time": 1.7306883335113525,
      "step": 28871
    },
    {
      "epoch": 2.7533531188964842e-05,
      "step": 28871,
      "training_step_time": 8.436902284622192
    },
    {
      "epoch": 2.753448486328125e-05,
      "model_forward_time": 1.7410142421722412,
      "step": 28872
    },
    {
      "epoch": 2.753448486328125e-05,
      "step": 28872,
      "training_step_time": 9.485129594802856
    },
    {
      "epoch": 2.7535438537597657e-05,
      "model_forward_time": 1.7359163761138916,
      "step": 28873
    },
    {
      "epoch": 2.7535438537597657e-05,
      "step": 28873,
      "training_step_time": 7.610947608947754
    },
    {
      "epoch": 2.753639221191406e-05,
      "model_forward_time": 1.8157176971435547,
      "step": 28874
    },
    {
      "epoch": 2.753639221191406e-05,
      "step": 28874,
      "training_step_time": 7.59429931640625
    },
    {
      "epoch": 2.753734588623047e-05,
      "model_forward_time": 2.8467230796813965,
      "step": 28875
    },
    {
      "epoch": 2.753734588623047e-05,
      "step": 28875,
      "training_step_time": 7.941590070724487
    },
    {
      "epoch": 2.7538299560546876e-05,
      "model_forward_time": 1.7333402633666992,
      "step": 28876
    },
    {
      "epoch": 2.7538299560546876e-05,
      "step": 28876,
      "training_step_time": 7.657575607299805
    },
    {
      "epoch": 2.753925323486328e-05,
      "model_forward_time": 1.734001636505127,
      "step": 28877
    },
    {
      "epoch": 2.753925323486328e-05,
      "step": 28877,
      "training_step_time": 7.540493488311768
    },
    {
      "epoch": 2.7540206909179688e-05,
      "model_forward_time": 1.7356491088867188,
      "step": 28878
    },
    {
      "epoch": 2.7540206909179688e-05,
      "step": 28878,
      "training_step_time": 8.92650318145752
    },
    {
      "epoch": 2.7541160583496095e-05,
      "model_forward_time": 1.7416210174560547,
      "step": 28879
    },
    {
      "epoch": 2.7541160583496095e-05,
      "step": 28879,
      "training_step_time": 7.538066625595093
    },
    {
      "epoch": 2.75421142578125e-05,
      "grad_norm": 0.11078549921512604,
      "learning_rate": 8.52041936289457e-06,
      "loss": 0.0602,
      "step": 28880
    },
    {
      "epoch": 2.75421142578125e-05,
      "model_forward_time": 1.7262027263641357,
      "step": 28880
    },
    {
      "dynamics_loss_avg": 0.05557032972574234,
      "epoch": 2.75421142578125e-05,
      "step": 28880
    },
    {
      "action_loss_avg": 0.0016613415209576488,
      "epoch": 2.75421142578125e-05,
      "step": 28880
    },
    {
      "epoch": 2.75421142578125e-05,
      "step": 28880,
      "training_step_time": 8.163370847702026
    },
    {
      "epoch": 2.7543067932128906e-05,
      "model_forward_time": 1.746108055114746,
      "step": 28881
    },
    {
      "epoch": 2.7543067932128906e-05,
      "step": 28881,
      "training_step_time": 8.885850191116333
    },
    {
      "epoch": 2.7544021606445314e-05,
      "model_forward_time": 1.7613356113433838,
      "step": 28882
    },
    {
      "epoch": 2.7544021606445314e-05,
      "step": 28882,
      "training_step_time": 8.471348524093628
    },
    {
      "epoch": 2.7544975280761718e-05,
      "model_forward_time": 1.7908780574798584,
      "step": 28883
    },
    {
      "epoch": 2.7544975280761718e-05,
      "step": 28883,
      "training_step_time": 8.473725318908691
    },
    {
      "epoch": 2.7545928955078125e-05,
      "model_forward_time": 2.821465253829956,
      "step": 28884
    },
    {
      "epoch": 2.7545928955078125e-05,
      "step": 28884,
      "training_step_time": 7.599025249481201
    },
    {
      "epoch": 2.7546882629394533e-05,
      "model_forward_time": 1.731194019317627,
      "step": 28885
    },
    {
      "epoch": 2.7546882629394533e-05,
      "step": 28885,
      "training_step_time": 7.819909334182739
    },
    {
      "epoch": 2.7547836303710937e-05,
      "model_forward_time": 1.724196195602417,
      "step": 28886
    },
    {
      "epoch": 2.7547836303710937e-05,
      "step": 28886,
      "training_step_time": 7.590260982513428
    },
    {
      "epoch": 2.7548789978027344e-05,
      "model_forward_time": 1.7235581874847412,
      "step": 28887
    },
    {
      "epoch": 2.7548789978027344e-05,
      "step": 28887,
      "training_step_time": 7.827822923660278
    },
    {
      "epoch": 2.754974365234375e-05,
      "model_forward_time": 1.726400375366211,
      "step": 28888
    },
    {
      "epoch": 2.754974365234375e-05,
      "step": 28888,
      "training_step_time": 7.712540864944458
    },
    {
      "epoch": 2.7550697326660156e-05,
      "model_forward_time": 1.7386410236358643,
      "step": 28889
    },
    {
      "epoch": 2.7550697326660156e-05,
      "step": 28889,
      "training_step_time": 8.634471416473389
    },
    {
      "epoch": 2.7551651000976563e-05,
      "grad_norm": 0.14737360179424286,
      "learning_rate": 8.51924501369048e-06,
      "loss": 0.0598,
      "step": 28890
    },
    {
      "epoch": 2.7551651000976563e-05,
      "model_forward_time": 1.7178666591644287,
      "step": 28890
    },
    {
      "dynamics_loss_avg": 0.0613831302151084,
      "epoch": 2.7551651000976563e-05,
      "step": 28890
    },
    {
      "action_loss_avg": 0.0022857985110022127,
      "epoch": 2.7551651000976563e-05,
      "step": 28890
    },
    {
      "epoch": 2.7551651000976563e-05,
      "step": 28890,
      "training_step_time": 8.213098049163818
    },
    {
      "epoch": 2.7552604675292967e-05,
      "model_forward_time": 1.7209734916687012,
      "step": 28891
    },
    {
      "epoch": 2.7552604675292967e-05,
      "step": 28891,
      "training_step_time": 9.687751293182373
    },
    {
      "epoch": 2.7553558349609374e-05,
      "model_forward_time": 1.7651512622833252,
      "step": 28892
    },
    {
      "epoch": 2.7553558349609374e-05,
      "step": 28892,
      "training_step_time": 8.527327060699463
    },
    {
      "epoch": 2.7554512023925782e-05,
      "model_forward_time": 1.7936556339263916,
      "step": 28893
    },
    {
      "epoch": 2.7554512023925782e-05,
      "step": 28893,
      "training_step_time": 8.336771965026855
    },
    {
      "epoch": 2.7555465698242186e-05,
      "model_forward_time": 2.476515769958496,
      "step": 28894
    },
    {
      "epoch": 2.7555465698242186e-05,
      "step": 28894,
      "training_step_time": 7.780737638473511
    },
    {
      "epoch": 2.7556419372558593e-05,
      "model_forward_time": 1.7328410148620605,
      "step": 28895
    },
    {
      "epoch": 2.7556419372558593e-05,
      "step": 28895,
      "training_step_time": 7.668537378311157
    },
    {
      "epoch": 2.7557373046875e-05,
      "model_forward_time": 1.7377119064331055,
      "step": 28896
    },
    {
      "epoch": 2.7557373046875e-05,
      "step": 28896,
      "training_step_time": 7.929933071136475
    },
    {
      "epoch": 2.7558326721191405e-05,
      "model_forward_time": 1.7304224967956543,
      "step": 28897
    },
    {
      "epoch": 2.7558326721191405e-05,
      "step": 28897,
      "training_step_time": 8.286935806274414
    },
    {
      "epoch": 2.7559280395507812e-05,
      "model_forward_time": 1.7251696586608887,
      "step": 28898
    },
    {
      "epoch": 2.7559280395507812e-05,
      "step": 28898,
      "training_step_time": 7.565119981765747
    },
    {
      "epoch": 2.756023406982422e-05,
      "model_forward_time": 1.7030975818634033,
      "step": 28899
    },
    {
      "epoch": 2.756023406982422e-05,
      "step": 28899,
      "training_step_time": 10.320370435714722
    },
    {
      "epoch": 2.7561187744140624e-05,
      "grad_norm": 0.14618705213069916,
      "learning_rate": 8.518070279627047e-06,
      "loss": 0.0603,
      "step": 28900
    },
    {
      "epoch": 2.7561187744140624e-05,
      "model_forward_time": 1.7105538845062256,
      "step": 28900
    },
    {
      "dynamics_loss_avg": 0.052598774805665015,
      "epoch": 2.7561187744140624e-05,
      "step": 28900
    },
    {
      "action_loss_avg": 0.0018690465600229799,
      "epoch": 2.7561187744140624e-05,
      "step": 28900
    },
    {
      "epoch": 2.7561187744140624e-05,
      "step": 28900,
      "training_step_time": 7.778481960296631
    },
    {
      "epoch": 2.756214141845703e-05,
      "model_forward_time": 1.7759888172149658,
      "step": 28901
    },
    {
      "epoch": 2.756214141845703e-05,
      "step": 28901,
      "training_step_time": 8.339510917663574
    },
    {
      "epoch": 2.756309509277344e-05,
      "model_forward_time": 1.7616422176361084,
      "step": 28902
    },
    {
      "epoch": 2.756309509277344e-05,
      "step": 28902,
      "training_step_time": 7.573659896850586
    },
    {
      "epoch": 2.7564048767089843e-05,
      "model_forward_time": 2.825421094894409,
      "step": 28903
    },
    {
      "epoch": 2.7564048767089843e-05,
      "step": 28903,
      "training_step_time": 7.8516082763671875
    },
    {
      "epoch": 2.756500244140625e-05,
      "model_forward_time": 1.7002286911010742,
      "step": 28904
    },
    {
      "epoch": 2.756500244140625e-05,
      "step": 28904,
      "training_step_time": 8.621882438659668
    },
    {
      "epoch": 2.7565956115722657e-05,
      "model_forward_time": 1.7339518070220947,
      "step": 28905
    },
    {
      "epoch": 2.7565956115722657e-05,
      "step": 28905,
      "training_step_time": 7.645090579986572
    },
    {
      "epoch": 2.756690979003906e-05,
      "model_forward_time": 1.734618902206421,
      "step": 28906
    },
    {
      "epoch": 2.756690979003906e-05,
      "step": 28906,
      "training_step_time": 7.40846586227417
    },
    {
      "epoch": 2.756786346435547e-05,
      "model_forward_time": 1.7353191375732422,
      "step": 28907
    },
    {
      "epoch": 2.756786346435547e-05,
      "step": 28907,
      "training_step_time": 8.405824899673462
    },
    {
      "epoch": 2.7568817138671876e-05,
      "model_forward_time": 1.7343804836273193,
      "step": 28908
    },
    {
      "epoch": 2.7568817138671876e-05,
      "step": 28908,
      "training_step_time": 9.46018934249878
    },
    {
      "epoch": 2.756977081298828e-05,
      "model_forward_time": 1.3513009548187256,
      "step": 28909
    },
    {
      "epoch": 2.756977081298828e-05,
      "step": 28909,
      "training_step_time": 7.863832235336304
    },
    {
      "epoch": 2.7570724487304688e-05,
      "grad_norm": 0.11297168582677841,
      "learning_rate": 8.516895160832738e-06,
      "loss": 0.0601,
      "step": 28910
    },
    {
      "epoch": 2.7570724487304688e-05,
      "model_forward_time": 1.7132744789123535,
      "step": 28910
    },
    {
      "dynamics_loss_avg": 0.06806609053164721,
      "epoch": 2.7570724487304688e-05,
      "step": 28910
    },
    {
      "action_loss_avg": 0.002169762353878468,
      "epoch": 2.7570724487304688e-05,
      "step": 28910
    },
    {
      "epoch": 2.7570724487304688e-05,
      "step": 28910,
      "training_step_time": 7.603803873062134
    },
    {
      "epoch": 2.7571678161621095e-05,
      "model_forward_time": 1.7336230278015137,
      "step": 28911
    },
    {
      "epoch": 2.7571678161621095e-05,
      "step": 28911,
      "training_step_time": 8.155571937561035
    },
    {
      "epoch": 2.75726318359375e-05,
      "model_forward_time": 2.814588785171509,
      "step": 28912
    },
    {
      "epoch": 2.75726318359375e-05,
      "step": 28912,
      "training_step_time": 8.732041835784912
    },
    {
      "epoch": 2.7573585510253906e-05,
      "model_forward_time": 1.734541654586792,
      "step": 28913
    },
    {
      "epoch": 2.7573585510253906e-05,
      "step": 28913,
      "training_step_time": 7.672622203826904
    },
    {
      "epoch": 2.7574539184570314e-05,
      "model_forward_time": 1.7260470390319824,
      "step": 28914
    },
    {
      "epoch": 2.7574539184570314e-05,
      "step": 28914,
      "training_step_time": 7.772300481796265
    },
    {
      "epoch": 2.7575492858886718e-05,
      "model_forward_time": 1.7241098880767822,
      "step": 28915
    },
    {
      "epoch": 2.7575492858886718e-05,
      "step": 28915,
      "training_step_time": 7.436683416366577
    },
    {
      "epoch": 2.7576446533203125e-05,
      "model_forward_time": 1.0336549282073975,
      "step": 28916
    },
    {
      "epoch": 2.7576446533203125e-05,
      "step": 28916,
      "training_step_time": 8.377958297729492
    },
    {
      "epoch": 2.7577400207519533e-05,
      "model_forward_time": 1.7138102054595947,
      "step": 28917
    },
    {
      "epoch": 2.7577400207519533e-05,
      "step": 28917,
      "training_step_time": 8.584100008010864
    },
    {
      "epoch": 2.7578353881835937e-05,
      "model_forward_time": 1.7302517890930176,
      "step": 28918
    },
    {
      "epoch": 2.7578353881835937e-05,
      "step": 28918,
      "training_step_time": 9.069273471832275
    },
    {
      "epoch": 2.7579307556152344e-05,
      "model_forward_time": 1.7275986671447754,
      "step": 28919
    },
    {
      "epoch": 2.7579307556152344e-05,
      "step": 28919,
      "training_step_time": 8.661369800567627
    },
    {
      "epoch": 2.758026123046875e-05,
      "grad_norm": 0.11016005277633667,
      "learning_rate": 8.515719657436061e-06,
      "loss": 0.0627,
      "step": 28920
    },
    {
      "epoch": 2.758026123046875e-05,
      "model_forward_time": 1.7439329624176025,
      "step": 28920
    },
    {
      "dynamics_loss_avg": 0.04202251480892301,
      "epoch": 2.758026123046875e-05,
      "step": 28920
    },
    {
      "action_loss_avg": 0.0007928861276013776,
      "epoch": 2.758026123046875e-05,
      "step": 28920
    },
    {
      "epoch": 2.758026123046875e-05,
      "step": 28920,
      "training_step_time": 8.730680704116821
    },
    {
      "epoch": 2.7581214904785156e-05,
      "model_forward_time": 1.7920773029327393,
      "step": 28921
    },
    {
      "epoch": 2.7581214904785156e-05,
      "step": 28921,
      "training_step_time": 7.876506090164185
    },
    {
      "epoch": 2.7582168579101563e-05,
      "model_forward_time": 2.9059243202209473,
      "step": 28922
    },
    {
      "epoch": 2.7582168579101563e-05,
      "step": 28922,
      "training_step_time": 8.7574622631073
    },
    {
      "epoch": 2.7583122253417967e-05,
      "model_forward_time": 1.7325372695922852,
      "step": 28923
    },
    {
      "epoch": 2.7583122253417967e-05,
      "step": 28923,
      "training_step_time": 7.829027414321899
    },
    {
      "epoch": 2.7584075927734375e-05,
      "model_forward_time": 1.7011244297027588,
      "step": 28924
    },
    {
      "epoch": 2.7584075927734375e-05,
      "step": 28924,
      "training_step_time": 7.513463258743286
    },
    {
      "epoch": 2.7585029602050782e-05,
      "model_forward_time": 1.7321271896362305,
      "step": 28925
    },
    {
      "epoch": 2.7585029602050782e-05,
      "step": 28925,
      "training_step_time": 7.695355176925659
    },
    {
      "epoch": 2.7585983276367186e-05,
      "model_forward_time": 1.738976240158081,
      "step": 28926
    },
    {
      "epoch": 2.7585983276367186e-05,
      "step": 28926,
      "training_step_time": 8.662317276000977
    },
    {
      "epoch": 2.7586936950683593e-05,
      "model_forward_time": 1.7247309684753418,
      "step": 28927
    },
    {
      "epoch": 2.7586936950683593e-05,
      "step": 28927,
      "training_step_time": 9.630287885665894
    },
    {
      "epoch": 2.7587890625e-05,
      "model_forward_time": 1.7316534519195557,
      "step": 28928
    },
    {
      "epoch": 2.7587890625e-05,
      "step": 28928,
      "training_step_time": 8.300276517868042
    },
    {
      "epoch": 2.7588844299316405e-05,
      "model_forward_time": 1.7368769645690918,
      "step": 28929
    },
    {
      "epoch": 2.7588844299316405e-05,
      "step": 28929,
      "training_step_time": 7.563310384750366
    },
    {
      "epoch": 2.7589797973632812e-05,
      "grad_norm": 0.18386419117450714,
      "learning_rate": 8.51454376956557e-06,
      "loss": 0.0614,
      "step": 28930
    },
    {
      "epoch": 2.7589797973632812e-05,
      "model_forward_time": 1.7838928699493408,
      "step": 28930
    },
    {
      "dynamics_loss_avg": 0.057919522374868394,
      "epoch": 2.7589797973632812e-05,
      "step": 28930
    },
    {
      "action_loss_avg": 0.001885898906039074,
      "epoch": 2.7589797973632812e-05,
      "step": 28930
    },
    {
      "epoch": 2.7589797973632812e-05,
      "step": 28930,
      "training_step_time": 7.9035584926605225
    },
    {
      "epoch": 2.759075164794922e-05,
      "model_forward_time": 2.875339984893799,
      "step": 28931
    },
    {
      "epoch": 2.759075164794922e-05,
      "step": 28931,
      "training_step_time": 7.624929666519165
    },
    {
      "epoch": 2.7591705322265624e-05,
      "model_forward_time": 1.7388317584991455,
      "step": 28932
    },
    {
      "epoch": 2.7591705322265624e-05,
      "step": 28932,
      "training_step_time": 7.758603811264038
    },
    {
      "epoch": 2.759265899658203e-05,
      "model_forward_time": 1.7342281341552734,
      "step": 28933
    },
    {
      "epoch": 2.759265899658203e-05,
      "step": 28933,
      "training_step_time": 7.425481081008911
    },
    {
      "epoch": 2.759361267089844e-05,
      "model_forward_time": 1.7317397594451904,
      "step": 28934
    },
    {
      "epoch": 2.759361267089844e-05,
      "step": 28934,
      "training_step_time": 7.666987895965576
    },
    {
      "epoch": 2.7594566345214843e-05,
      "model_forward_time": 1.72987699508667,
      "step": 28935
    },
    {
      "epoch": 2.7594566345214843e-05,
      "step": 28935,
      "training_step_time": 7.573986291885376
    },
    {
      "epoch": 2.759552001953125e-05,
      "model_forward_time": 1.7046256065368652,
      "step": 28936
    },
    {
      "epoch": 2.759552001953125e-05,
      "step": 28936,
      "training_step_time": 7.746669292449951
    },
    {
      "epoch": 2.7596473693847657e-05,
      "model_forward_time": 1.7293589115142822,
      "step": 28937
    },
    {
      "epoch": 2.7596473693847657e-05,
      "step": 28937,
      "training_step_time": 7.810733795166016
    },
    {
      "epoch": 2.759742736816406e-05,
      "model_forward_time": 1.7276771068572998,
      "step": 28938
    },
    {
      "epoch": 2.759742736816406e-05,
      "step": 28938,
      "training_step_time": 7.65422797203064
    },
    {
      "epoch": 2.759838104248047e-05,
      "model_forward_time": 1.778541088104248,
      "step": 28939
    },
    {
      "epoch": 2.759838104248047e-05,
      "step": 28939,
      "training_step_time": 10.30695390701294
    },
    {
      "epoch": 2.7599334716796876e-05,
      "grad_norm": 0.13575655221939087,
      "learning_rate": 8.513367497349854e-06,
      "loss": 0.06,
      "step": 28940
    },
    {
      "epoch": 2.7599334716796876e-05,
      "model_forward_time": 2.83482027053833,
      "step": 28940
    },
    {
      "dynamics_loss_avg": 0.06201448384672403,
      "epoch": 2.7599334716796876e-05,
      "step": 28940
    },
    {
      "action_loss_avg": 0.001712759054498747,
      "epoch": 2.7599334716796876e-05,
      "step": 28940
    },
    {
      "epoch": 2.7599334716796876e-05,
      "step": 28940,
      "training_step_time": 7.6179563999176025
    },
    {
      "epoch": 2.760028839111328e-05,
      "model_forward_time": 1.7198688983917236,
      "step": 28941
    },
    {
      "epoch": 2.760028839111328e-05,
      "step": 28941,
      "training_step_time": 8.784181118011475
    },
    {
      "epoch": 2.7601242065429688e-05,
      "model_forward_time": 1.72837495803833,
      "step": 28942
    },
    {
      "epoch": 2.7601242065429688e-05,
      "step": 28942,
      "training_step_time": 7.959837913513184
    },
    {
      "epoch": 2.7602195739746095e-05,
      "model_forward_time": 1.7373783588409424,
      "step": 28943
    },
    {
      "epoch": 2.7602195739746095e-05,
      "step": 28943,
      "training_step_time": 7.4936323165893555
    },
    {
      "epoch": 2.76031494140625e-05,
      "model_forward_time": 1.7355704307556152,
      "step": 28944
    },
    {
      "epoch": 2.76031494140625e-05,
      "step": 28944,
      "training_step_time": 8.304661512374878
    },
    {
      "epoch": 2.7604103088378907e-05,
      "model_forward_time": 1.726820468902588,
      "step": 28945
    },
    {
      "epoch": 2.7604103088378907e-05,
      "step": 28945,
      "training_step_time": 7.726134777069092
    },
    {
      "epoch": 2.7605056762695314e-05,
      "model_forward_time": 1.7240145206451416,
      "step": 28946
    },
    {
      "epoch": 2.7605056762695314e-05,
      "step": 28946,
      "training_step_time": 7.743411302566528
    },
    {
      "epoch": 2.7606010437011718e-05,
      "model_forward_time": 1.7280879020690918,
      "step": 28947
    },
    {
      "epoch": 2.7606010437011718e-05,
      "step": 28947,
      "training_step_time": 7.6173436641693115
    },
    {
      "epoch": 2.7606964111328125e-05,
      "model_forward_time": 1.7448813915252686,
      "step": 28948
    },
    {
      "epoch": 2.7606964111328125e-05,
      "step": 28948,
      "training_step_time": 8.388856410980225
    },
    {
      "epoch": 2.7607917785644533e-05,
      "model_forward_time": 2.8342790603637695,
      "step": 28949
    },
    {
      "epoch": 2.7607917785644533e-05,
      "step": 28949,
      "training_step_time": 7.8389036655426025
    },
    {
      "epoch": 2.7608871459960937e-05,
      "grad_norm": 0.1884067952632904,
      "learning_rate": 8.512190840917551e-06,
      "loss": 0.0628,
      "step": 28950
    },
    {
      "epoch": 2.7608871459960937e-05,
      "model_forward_time": 1.7145841121673584,
      "step": 28950
    },
    {
      "dynamics_loss_avg": 0.04446749137714505,
      "epoch": 2.7608871459960937e-05,
      "step": 28950
    },
    {
      "action_loss_avg": 0.0016892596206162125,
      "epoch": 2.7608871459960937e-05,
      "step": 28950
    },
    {
      "epoch": 2.7608871459960937e-05,
      "step": 28950,
      "training_step_time": 7.843174934387207
    },
    {
      "epoch": 2.7609825134277344e-05,
      "model_forward_time": 1.7245686054229736,
      "step": 28951
    },
    {
      "epoch": 2.7609825134277344e-05,
      "step": 28951,
      "training_step_time": 7.549185276031494
    },
    {
      "epoch": 2.761077880859375e-05,
      "model_forward_time": 1.7331888675689697,
      "step": 28952
    },
    {
      "epoch": 2.761077880859375e-05,
      "step": 28952,
      "training_step_time": 8.29179573059082
    },
    {
      "epoch": 2.7611732482910156e-05,
      "model_forward_time": 1.7378859519958496,
      "step": 28953
    },
    {
      "epoch": 2.7611732482910156e-05,
      "step": 28953,
      "training_step_time": 8.363699436187744
    },
    {
      "epoch": 2.7612686157226563e-05,
      "model_forward_time": 1.7329955101013184,
      "step": 28954
    },
    {
      "epoch": 2.7612686157226563e-05,
      "step": 28954,
      "training_step_time": 7.727689981460571
    },
    {
      "epoch": 2.7613639831542967e-05,
      "model_forward_time": 1.7337119579315186,
      "step": 28955
    },
    {
      "epoch": 2.7613639831542967e-05,
      "step": 28955,
      "training_step_time": 8.407483339309692
    },
    {
      "epoch": 2.7614593505859375e-05,
      "model_forward_time": 1.7305889129638672,
      "step": 28956
    },
    {
      "epoch": 2.7614593505859375e-05,
      "step": 28956,
      "training_step_time": 7.644830226898193
    },
    {
      "epoch": 2.7615547180175782e-05,
      "model_forward_time": 1.740382194519043,
      "step": 28957
    },
    {
      "epoch": 2.7615547180175782e-05,
      "step": 28957,
      "training_step_time": 8.579460144042969
    },
    {
      "epoch": 2.7616500854492186e-05,
      "model_forward_time": 1.7887217998504639,
      "step": 28958
    },
    {
      "epoch": 2.7616500854492186e-05,
      "step": 28958,
      "training_step_time": 8.40346074104309
    },
    {
      "epoch": 2.7617454528808593e-05,
      "model_forward_time": 1.718529224395752,
      "step": 28959
    },
    {
      "epoch": 2.7617454528808593e-05,
      "step": 28959,
      "training_step_time": 7.737817764282227
    },
    {
      "epoch": 2.7618408203125e-05,
      "grad_norm": 0.16702541708946228,
      "learning_rate": 8.511013800397338e-06,
      "loss": 0.0598,
      "step": 28960
    },
    {
      "epoch": 2.7618408203125e-05,
      "model_forward_time": 1.7120583057403564,
      "step": 28960
    },
    {
      "dynamics_loss_avg": 0.0694210322573781,
      "epoch": 2.7618408203125e-05,
      "step": 28960
    },
    {
      "action_loss_avg": 0.0032763037306722254,
      "epoch": 2.7618408203125e-05,
      "step": 28960
    },
    {
      "epoch": 2.7618408203125e-05,
      "step": 28960,
      "training_step_time": 7.637913942337036
    },
    {
      "epoch": 2.7619361877441405e-05,
      "model_forward_time": 1.7569642066955566,
      "step": 28961
    },
    {
      "epoch": 2.7619361877441405e-05,
      "step": 28961,
      "training_step_time": 8.314841270446777
    },
    {
      "epoch": 2.7620315551757812e-05,
      "model_forward_time": 1.7240488529205322,
      "step": 28962
    },
    {
      "epoch": 2.7620315551757812e-05,
      "step": 28962,
      "training_step_time": 7.666287899017334
    },
    {
      "epoch": 2.762126922607422e-05,
      "model_forward_time": 1.7293448448181152,
      "step": 28963
    },
    {
      "epoch": 2.762126922607422e-05,
      "step": 28963,
      "training_step_time": 9.498944282531738
    },
    {
      "epoch": 2.7622222900390624e-05,
      "model_forward_time": 1.7315831184387207,
      "step": 28964
    },
    {
      "epoch": 2.7622222900390624e-05,
      "step": 28964,
      "training_step_time": 8.536028861999512
    },
    {
      "epoch": 2.762317657470703e-05,
      "model_forward_time": 1.733445405960083,
      "step": 28965
    },
    {
      "epoch": 2.762317657470703e-05,
      "step": 28965,
      "training_step_time": 8.287593126296997
    },
    {
      "epoch": 2.762413024902344e-05,
      "model_forward_time": 1.7329151630401611,
      "step": 28966
    },
    {
      "epoch": 2.762413024902344e-05,
      "step": 28966,
      "training_step_time": 8.189888000488281
    },
    {
      "epoch": 2.7625083923339843e-05,
      "model_forward_time": 1.7978360652923584,
      "step": 28967
    },
    {
      "epoch": 2.7625083923339843e-05,
      "step": 28967,
      "training_step_time": 7.861925840377808
    },
    {
      "epoch": 2.762603759765625e-05,
      "model_forward_time": 2.8642170429229736,
      "step": 28968
    },
    {
      "epoch": 2.762603759765625e-05,
      "step": 28968,
      "training_step_time": 7.866677284240723
    },
    {
      "epoch": 2.7626991271972657e-05,
      "model_forward_time": 1.7336609363555908,
      "step": 28969
    },
    {
      "epoch": 2.7626991271972657e-05,
      "step": 28969,
      "training_step_time": 7.704375743865967
    },
    {
      "epoch": 2.762794494628906e-05,
      "grad_norm": 0.18518120050430298,
      "learning_rate": 8.509836375917938e-06,
      "loss": 0.061,
      "step": 28970
    },
    {
      "epoch": 2.762794494628906e-05,
      "model_forward_time": 1.7093446254730225,
      "step": 28970
    },
    {
      "dynamics_loss_avg": 0.05426286216825247,
      "epoch": 2.762794494628906e-05,
      "step": 28970
    },
    {
      "action_loss_avg": 0.0020267291867639868,
      "epoch": 2.762794494628906e-05,
      "step": 28970
    },
    {
      "epoch": 2.762794494628906e-05,
      "step": 28970,
      "training_step_time": 7.367032051086426
    },
    {
      "epoch": 2.762889862060547e-05,
      "model_forward_time": 1.7260119915008545,
      "step": 28971
    },
    {
      "epoch": 2.762889862060547e-05,
      "step": 28971,
      "training_step_time": 8.494945049285889
    },
    {
      "epoch": 2.7629852294921876e-05,
      "model_forward_time": 1.7325265407562256,
      "step": 28972
    },
    {
      "epoch": 2.7629852294921876e-05,
      "step": 28972,
      "training_step_time": 8.770653486251831
    },
    {
      "epoch": 2.763080596923828e-05,
      "model_forward_time": 1.734915018081665,
      "step": 28973
    },
    {
      "epoch": 2.763080596923828e-05,
      "step": 28973,
      "training_step_time": 7.7994513511657715
    },
    {
      "epoch": 2.7631759643554688e-05,
      "model_forward_time": 1.7325689792633057,
      "step": 28974
    },
    {
      "epoch": 2.7631759643554688e-05,
      "step": 28974,
      "training_step_time": 9.872668743133545
    },
    {
      "epoch": 2.7632713317871095e-05,
      "model_forward_time": 1.7244210243225098,
      "step": 28975
    },
    {
      "epoch": 2.7632713317871095e-05,
      "step": 28975,
      "training_step_time": 8.71299409866333
    },
    {
      "epoch": 2.76336669921875e-05,
      "model_forward_time": 1.795781135559082,
      "step": 28976
    },
    {
      "epoch": 2.76336669921875e-05,
      "step": 28976,
      "training_step_time": 8.670732498168945
    },
    {
      "epoch": 2.7634620666503907e-05,
      "model_forward_time": 2.8699777126312256,
      "step": 28977
    },
    {
      "epoch": 2.7634620666503907e-05,
      "step": 28977,
      "training_step_time": 7.867701768875122
    },
    {
      "epoch": 2.7635574340820314e-05,
      "model_forward_time": 1.726815938949585,
      "step": 28978
    },
    {
      "epoch": 2.7635574340820314e-05,
      "step": 28978,
      "training_step_time": 8.360194683074951
    },
    {
      "epoch": 2.7636528015136718e-05,
      "model_forward_time": 1.7326300144195557,
      "step": 28979
    },
    {
      "epoch": 2.7636528015136718e-05,
      "step": 28979,
      "training_step_time": 7.218102693557739
    },
    {
      "epoch": 2.7637481689453125e-05,
      "grad_norm": 0.16277970373630524,
      "learning_rate": 8.508658567608105e-06,
      "loss": 0.0598,
      "step": 28980
    },
    {
      "epoch": 2.7637481689453125e-05,
      "model_forward_time": 1.7088723182678223,
      "step": 28980
    },
    {
      "dynamics_loss_avg": 0.06367385163903236,
      "epoch": 2.7637481689453125e-05,
      "step": 28980
    },
    {
      "action_loss_avg": 0.0016032267827540635,
      "epoch": 2.7637481689453125e-05,
      "step": 28980
    },
    {
      "epoch": 2.7637481689453125e-05,
      "step": 28980,
      "training_step_time": 7.541942119598389
    },
    {
      "epoch": 2.7638435363769533e-05,
      "model_forward_time": 1.700587511062622,
      "step": 28981
    },
    {
      "epoch": 2.7638435363769533e-05,
      "step": 28981,
      "training_step_time": 8.885116577148438
    },
    {
      "epoch": 2.7639389038085937e-05,
      "model_forward_time": 1.7216966152191162,
      "step": 28982
    },
    {
      "epoch": 2.7639389038085937e-05,
      "step": 28982,
      "training_step_time": 7.8307905197143555
    },
    {
      "epoch": 2.7640342712402344e-05,
      "model_forward_time": 1.7116971015930176,
      "step": 28983
    },
    {
      "epoch": 2.7640342712402344e-05,
      "step": 28983,
      "training_step_time": 7.8381874561309814
    },
    {
      "epoch": 2.764129638671875e-05,
      "model_forward_time": 1.7318921089172363,
      "step": 28984
    },
    {
      "epoch": 2.764129638671875e-05,
      "step": 28984,
      "training_step_time": 7.770880937576294
    },
    {
      "epoch": 2.7642250061035156e-05,
      "model_forward_time": 1.7577290534973145,
      "step": 28985
    },
    {
      "epoch": 2.7642250061035156e-05,
      "step": 28985,
      "training_step_time": 8.28666067123413
    },
    {
      "epoch": 2.7643203735351563e-05,
      "model_forward_time": 2.8313348293304443,
      "step": 28986
    },
    {
      "epoch": 2.7643203735351563e-05,
      "step": 28986,
      "training_step_time": 8.391340970993042
    },
    {
      "epoch": 2.7644157409667967e-05,
      "model_forward_time": 1.7271161079406738,
      "step": 28987
    },
    {
      "epoch": 2.7644157409667967e-05,
      "step": 28987,
      "training_step_time": 7.6847310066223145
    },
    {
      "epoch": 2.7645111083984375e-05,
      "model_forward_time": 1.7257769107818604,
      "step": 28988
    },
    {
      "epoch": 2.7645111083984375e-05,
      "step": 28988,
      "training_step_time": 7.245275020599365
    },
    {
      "epoch": 2.7646064758300782e-05,
      "model_forward_time": 1.7270770072937012,
      "step": 28989
    },
    {
      "epoch": 2.7646064758300782e-05,
      "step": 28989,
      "training_step_time": 8.517598867416382
    },
    {
      "epoch": 2.7647018432617186e-05,
      "grad_norm": 0.15415406227111816,
      "learning_rate": 8.507480375596647e-06,
      "loss": 0.0616,
      "step": 28990
    },
    {
      "epoch": 2.7647018432617186e-05,
      "model_forward_time": 1.7147037982940674,
      "step": 28990
    },
    {
      "dynamics_loss_avg": 0.06551346834748983,
      "epoch": 2.7647018432617186e-05,
      "step": 28990
    },
    {
      "action_loss_avg": 0.0021467821672558783,
      "epoch": 2.7647018432617186e-05,
      "step": 28990
    },
    {
      "epoch": 2.7647018432617186e-05,
      "step": 28990,
      "training_step_time": 7.698608160018921
    },
    {
      "epoch": 2.7647972106933593e-05,
      "model_forward_time": 1.7186684608459473,
      "step": 28991
    },
    {
      "epoch": 2.7647972106933593e-05,
      "step": 28991,
      "training_step_time": 7.830347061157227
    },
    {
      "epoch": 2.764892578125e-05,
      "model_forward_time": 1.7322187423706055,
      "step": 28992
    },
    {
      "epoch": 2.764892578125e-05,
      "step": 28992,
      "training_step_time": 8.765684843063354
    },
    {
      "epoch": 2.7649879455566405e-05,
      "model_forward_time": 1.356114387512207,
      "step": 28993
    },
    {
      "epoch": 2.7649879455566405e-05,
      "step": 28993,
      "training_step_time": 7.679377555847168
    },
    {
      "epoch": 2.7650833129882812e-05,
      "model_forward_time": 1.7236440181732178,
      "step": 28994
    },
    {
      "epoch": 2.7650833129882812e-05,
      "step": 28994,
      "training_step_time": 8.697518825531006
    },
    {
      "epoch": 2.765178680419922e-05,
      "model_forward_time": 2.8349316120147705,
      "step": 28995
    },
    {
      "epoch": 2.765178680419922e-05,
      "step": 28995,
      "training_step_time": 7.718170881271362
    },
    {
      "epoch": 2.7652740478515624e-05,
      "model_forward_time": 1.7319364547729492,
      "step": 28996
    },
    {
      "epoch": 2.7652740478515624e-05,
      "step": 28996,
      "training_step_time": 7.774757623672485
    },
    {
      "epoch": 2.765369415283203e-05,
      "model_forward_time": 1.7582104206085205,
      "step": 28997
    },
    {
      "epoch": 2.765369415283203e-05,
      "step": 28997,
      "training_step_time": 7.687924385070801
    },
    {
      "epoch": 2.765464782714844e-05,
      "model_forward_time": 1.7135109901428223,
      "step": 28998
    },
    {
      "epoch": 2.765464782714844e-05,
      "step": 28998,
      "training_step_time": 7.800869941711426
    },
    {
      "epoch": 2.7655601501464843e-05,
      "model_forward_time": 1.7203824520111084,
      "step": 28999
    },
    {
      "epoch": 2.7655601501464843e-05,
      "step": 28999,
      "training_step_time": 9.042030334472656
    },
    {
      "epoch": 2.765655517578125e-05,
      "grad_norm": 0.13709941506385803,
      "learning_rate": 8.506301800012408e-06,
      "loss": 0.062,
      "step": 29000
    },
    {
      "epoch": 2.765655517578125e-05,
      "model_forward_time": 1.7999968528747559,
      "step": 29000
    },
    {
      "dynamics_loss_avg": 0.08098820745944976,
      "epoch": 2.765655517578125e-05,
      "step": 29000
    },
    {
      "action_loss_avg": 0.0024234833137597887,
      "epoch": 2.765655517578125e-05,
      "step": 29000
    },
    {
      "epoch": 2.765655517578125e-05,
      "step": 29000,
      "training_step_time": 9.85345721244812
    },
    {
      "epoch": 2.7657508850097657e-05,
      "model_forward_time": 1.7771711349487305,
      "step": 29001
    },
    {
      "epoch": 2.7657508850097657e-05,
      "step": 29001,
      "training_step_time": 7.703062295913696
    },
    {
      "epoch": 2.765846252441406e-05,
      "model_forward_time": 1.7885205745697021,
      "step": 29002
    },
    {
      "epoch": 2.765846252441406e-05,
      "step": 29002,
      "training_step_time": 7.772270441055298
    },
    {
      "epoch": 2.765941619873047e-05,
      "model_forward_time": 2.8115041255950928,
      "step": 29003
    },
    {
      "epoch": 2.765941619873047e-05,
      "step": 29003,
      "training_step_time": 9.19068169593811
    },
    {
      "epoch": 2.7660369873046876e-05,
      "model_forward_time": 1.7736608982086182,
      "step": 29004
    },
    {
      "epoch": 2.7660369873046876e-05,
      "step": 29004,
      "training_step_time": 8.582278490066528
    },
    {
      "epoch": 2.766132354736328e-05,
      "model_forward_time": 1.77046537399292,
      "step": 29005
    },
    {
      "epoch": 2.766132354736328e-05,
      "step": 29005,
      "training_step_time": 7.733193397521973
    },
    {
      "epoch": 2.7662277221679688e-05,
      "model_forward_time": 1.7770321369171143,
      "step": 29006
    },
    {
      "epoch": 2.7662277221679688e-05,
      "step": 29006,
      "training_step_time": 7.491063594818115
    },
    {
      "epoch": 2.7663230895996095e-05,
      "model_forward_time": 1.8272459506988525,
      "step": 29007
    },
    {
      "epoch": 2.7663230895996095e-05,
      "step": 29007,
      "training_step_time": 9.031567096710205
    },
    {
      "epoch": 2.76641845703125e-05,
      "model_forward_time": 1.7721049785614014,
      "step": 29008
    },
    {
      "epoch": 2.76641845703125e-05,
      "step": 29008,
      "training_step_time": 10.042452812194824
    },
    {
      "epoch": 2.7665138244628907e-05,
      "model_forward_time": 1.7559852600097656,
      "step": 29009
    },
    {
      "epoch": 2.7665138244628907e-05,
      "step": 29009,
      "training_step_time": 8.749364137649536
    },
    {
      "epoch": 2.7666091918945314e-05,
      "grad_norm": 0.15345902740955353,
      "learning_rate": 8.505122840984278e-06,
      "loss": 0.0638,
      "step": 29010
    },
    {
      "epoch": 2.7666091918945314e-05,
      "model_forward_time": 1.7869806289672852,
      "step": 29010
    },
    {
      "dynamics_loss_avg": 0.05609255880117416,
      "epoch": 2.7666091918945314e-05,
      "step": 29010
    },
    {
      "action_loss_avg": 0.0015432494576089085,
      "epoch": 2.7666091918945314e-05,
      "step": 29010
    },
    {
      "epoch": 2.7666091918945314e-05,
      "step": 29010,
      "training_step_time": 6.797333002090454
    },
    {
      "epoch": 2.7667045593261718e-05,
      "model_forward_time": 1.7755334377288818,
      "step": 29011
    },
    {
      "epoch": 2.7667045593261718e-05,
      "step": 29011,
      "training_step_time": 6.753556728363037
    },
    {
      "epoch": 2.7667999267578125e-05,
      "model_forward_time": 1.7674229145050049,
      "step": 29012
    },
    {
      "epoch": 2.7667999267578125e-05,
      "step": 29012,
      "training_step_time": 6.748249530792236
    },
    {
      "epoch": 2.7668952941894533e-05,
      "model_forward_time": 1.7626323699951172,
      "step": 29013
    },
    {
      "epoch": 2.7668952941894533e-05,
      "step": 29013,
      "training_step_time": 6.757244348526001
    },
    {
      "epoch": 2.7669906616210937e-05,
      "model_forward_time": 1.7695202827453613,
      "step": 29014
    },
    {
      "epoch": 2.7669906616210937e-05,
      "step": 29014,
      "training_step_time": 6.747359037399292
    },
    {
      "epoch": 2.7670860290527344e-05,
      "model_forward_time": 1.7818374633789062,
      "step": 29015
    },
    {
      "epoch": 2.7670860290527344e-05,
      "step": 29015,
      "training_step_time": 6.741145849227905
    },
    {
      "epoch": 2.767181396484375e-05,
      "model_forward_time": 1.7829248905181885,
      "step": 29016
    },
    {
      "epoch": 2.767181396484375e-05,
      "step": 29016,
      "training_step_time": 6.728703498840332
    },
    {
      "epoch": 2.7672767639160156e-05,
      "model_forward_time": 1.7599494457244873,
      "step": 29017
    },
    {
      "epoch": 2.7672767639160156e-05,
      "step": 29017,
      "training_step_time": 6.699554204940796
    },
    {
      "epoch": 2.7673721313476563e-05,
      "model_forward_time": 2.371462106704712,
      "step": 29018
    },
    {
      "epoch": 2.7673721313476563e-05,
      "step": 29018,
      "training_step_time": 8.64337968826294
    },
    {
      "epoch": 2.7674674987792967e-05,
      "model_forward_time": 1.4038352966308594,
      "step": 29019
    },
    {
      "epoch": 2.7674674987792967e-05,
      "step": 29019,
      "training_step_time": 7.641440391540527
    },
    {
      "epoch": 2.7675628662109375e-05,
      "grad_norm": 0.1454603523015976,
      "learning_rate": 8.503943498641184e-06,
      "loss": 0.0628,
      "step": 29020
    },
    {
      "epoch": 2.7675628662109375e-05,
      "model_forward_time": 1.7508788108825684,
      "step": 29020
    },
    {
      "dynamics_loss_avg": 0.0613242506980896,
      "epoch": 2.7675628662109375e-05,
      "step": 29020
    },
    {
      "action_loss_avg": 0.002275942167034373,
      "epoch": 2.7675628662109375e-05,
      "step": 29020
    },
    {
      "epoch": 2.7675628662109375e-05,
      "step": 29020,
      "training_step_time": 6.7695701122283936
    },
    {
      "epoch": 2.7676582336425782e-05,
      "model_forward_time": 1.7917418479919434,
      "step": 29021
    },
    {
      "epoch": 2.7676582336425782e-05,
      "step": 29021,
      "training_step_time": 6.776254892349243
    },
    {
      "epoch": 2.7677536010742186e-05,
      "model_forward_time": 1.7648985385894775,
      "step": 29022
    },
    {
      "epoch": 2.7677536010742186e-05,
      "step": 29022,
      "training_step_time": 6.784137010574341
    },
    {
      "epoch": 2.7678489685058593e-05,
      "model_forward_time": 1.7730417251586914,
      "step": 29023
    },
    {
      "epoch": 2.7678489685058593e-05,
      "step": 29023,
      "training_step_time": 6.750302314758301
    },
    {
      "epoch": 2.7679443359375e-05,
      "model_forward_time": 1.76776123046875,
      "step": 29024
    },
    {
      "epoch": 2.7679443359375e-05,
      "step": 29024,
      "training_step_time": 6.753549098968506
    },
    {
      "epoch": 2.7680397033691405e-05,
      "model_forward_time": 1.774531602859497,
      "step": 29025
    },
    {
      "epoch": 2.7680397033691405e-05,
      "step": 29025,
      "training_step_time": 6.738451719284058
    },
    {
      "epoch": 2.7681350708007812e-05,
      "model_forward_time": 1.768362045288086,
      "step": 29026
    },
    {
      "epoch": 2.7681350708007812e-05,
      "step": 29026,
      "training_step_time": 6.790321350097656
    },
    {
      "epoch": 2.768230438232422e-05,
      "model_forward_time": 2.374488115310669,
      "step": 29027
    },
    {
      "epoch": 2.768230438232422e-05,
      "step": 29027,
      "training_step_time": 7.450534343719482
    },
    {
      "epoch": 2.7683258056640624e-05,
      "model_forward_time": 1.7728540897369385,
      "step": 29028
    },
    {
      "epoch": 2.7683258056640624e-05,
      "step": 29028,
      "training_step_time": 7.742655515670776
    },
    {
      "epoch": 2.768421173095703e-05,
      "model_forward_time": 1.7635493278503418,
      "step": 29029
    },
    {
      "epoch": 2.768421173095703e-05,
      "step": 29029,
      "training_step_time": 6.73870062828064
    },
    {
      "epoch": 2.768516540527344e-05,
      "grad_norm": 0.15790405869483948,
      "learning_rate": 8.502763773112095e-06,
      "loss": 0.0644,
      "step": 29030
    },
    {
      "epoch": 2.768516540527344e-05,
      "model_forward_time": 1.781111478805542,
      "step": 29030
    },
    {
      "dynamics_loss_avg": 0.056903337314724925,
      "epoch": 2.768516540527344e-05,
      "step": 29030
    },
    {
      "action_loss_avg": 0.002301145938690752,
      "epoch": 2.768516540527344e-05,
      "step": 29030
    },
    {
      "epoch": 2.768516540527344e-05,
      "step": 29030,
      "training_step_time": 6.745004892349243
    },
    {
      "epoch": 2.7686119079589843e-05,
      "model_forward_time": 1.7629892826080322,
      "step": 29031
    },
    {
      "epoch": 2.7686119079589843e-05,
      "step": 29031,
      "training_step_time": 6.749032974243164
    },
    {
      "epoch": 2.768707275390625e-05,
      "model_forward_time": 1.7738170623779297,
      "step": 29032
    },
    {
      "epoch": 2.768707275390625e-05,
      "step": 29032,
      "training_step_time": 6.740677833557129
    },
    {
      "epoch": 2.7688026428222657e-05,
      "model_forward_time": 1.757887363433838,
      "step": 29033
    },
    {
      "epoch": 2.7688026428222657e-05,
      "step": 29033,
      "training_step_time": 6.753369092941284
    },
    {
      "epoch": 2.768898010253906e-05,
      "model_forward_time": 1.7682266235351562,
      "step": 29034
    },
    {
      "epoch": 2.768898010253906e-05,
      "step": 29034,
      "training_step_time": 6.728531360626221
    },
    {
      "epoch": 2.768993377685547e-05,
      "model_forward_time": 1.7654798030853271,
      "step": 29035
    },
    {
      "epoch": 2.768993377685547e-05,
      "step": 29035,
      "training_step_time": 6.753388404846191
    },
    {
      "epoch": 2.7690887451171876e-05,
      "model_forward_time": 2.3778138160705566,
      "step": 29036
    },
    {
      "epoch": 2.7690887451171876e-05,
      "step": 29036,
      "training_step_time": 7.451353311538696
    },
    {
      "epoch": 2.769184112548828e-05,
      "model_forward_time": 1.759784460067749,
      "step": 29037
    },
    {
      "epoch": 2.769184112548828e-05,
      "step": 29037,
      "training_step_time": 9.442054271697998
    },
    {
      "epoch": 2.7692794799804688e-05,
      "model_forward_time": 1.7694175243377686,
      "step": 29038
    },
    {
      "epoch": 2.7692794799804688e-05,
      "step": 29038,
      "training_step_time": 7.698771238327026
    },
    {
      "epoch": 2.7693748474121095e-05,
      "model_forward_time": 1.766526222229004,
      "step": 29039
    },
    {
      "epoch": 2.7693748474121095e-05,
      "step": 29039,
      "training_step_time": 6.694756269454956
    },
    {
      "epoch": 2.76947021484375e-05,
      "grad_norm": 0.14055046439170837,
      "learning_rate": 8.501583664526026e-06,
      "loss": 0.0616,
      "step": 29040
    },
    {
      "epoch": 2.76947021484375e-05,
      "model_forward_time": 1.767745018005371,
      "step": 29040
    },
    {
      "dynamics_loss_avg": 0.04798412965610623,
      "epoch": 2.76947021484375e-05,
      "step": 29040
    },
    {
      "action_loss_avg": 0.0018383037124294788,
      "epoch": 2.76947021484375e-05,
      "step": 29040
    },
    {
      "epoch": 2.76947021484375e-05,
      "step": 29040,
      "training_step_time": 6.691235303878784
    },
    {
      "epoch": 2.7695655822753907e-05,
      "model_forward_time": 1.7690846920013428,
      "step": 29041
    },
    {
      "epoch": 2.7695655822753907e-05,
      "step": 29041,
      "training_step_time": 6.731688976287842
    },
    {
      "epoch": 2.7696609497070314e-05,
      "model_forward_time": 1.7697317600250244,
      "step": 29042
    },
    {
      "epoch": 2.7696609497070314e-05,
      "step": 29042,
      "training_step_time": 6.716545581817627
    },
    {
      "epoch": 2.7697563171386718e-05,
      "model_forward_time": 1.7750647068023682,
      "step": 29043
    },
    {
      "epoch": 2.7697563171386718e-05,
      "step": 29043,
      "training_step_time": 6.732206344604492
    },
    {
      "epoch": 2.7698516845703125e-05,
      "model_forward_time": 1.7915637493133545,
      "step": 29044
    },
    {
      "epoch": 2.7698516845703125e-05,
      "step": 29044,
      "training_step_time": 6.702305316925049
    },
    {
      "epoch": 2.7699470520019533e-05,
      "model_forward_time": 2.3510797023773193,
      "step": 29045
    },
    {
      "epoch": 2.7699470520019533e-05,
      "step": 29045,
      "training_step_time": 7.107821464538574
    },
    {
      "epoch": 2.7700424194335937e-05,
      "model_forward_time": 1.7637715339660645,
      "step": 29046
    },
    {
      "epoch": 2.7700424194335937e-05,
      "step": 29046,
      "training_step_time": 8.475298404693604
    },
    {
      "epoch": 2.7701377868652344e-05,
      "model_forward_time": 1.7607100009918213,
      "step": 29047
    },
    {
      "epoch": 2.7701377868652344e-05,
      "step": 29047,
      "training_step_time": 7.680837392807007
    },
    {
      "epoch": 2.770233154296875e-05,
      "model_forward_time": 1.7665793895721436,
      "step": 29048
    },
    {
      "epoch": 2.770233154296875e-05,
      "step": 29048,
      "training_step_time": 6.683758020401001
    },
    {
      "epoch": 2.7703285217285156e-05,
      "model_forward_time": 1.7742347717285156,
      "step": 29049
    },
    {
      "epoch": 2.7703285217285156e-05,
      "step": 29049,
      "training_step_time": 6.698210716247559
    },
    {
      "epoch": 2.7704238891601563e-05,
      "grad_norm": 0.20474383234977722,
      "learning_rate": 8.500403173012033e-06,
      "loss": 0.0653,
      "step": 29050
    },
    {
      "epoch": 2.7704238891601563e-05,
      "model_forward_time": 1.759606122970581,
      "step": 29050
    },
    {
      "dynamics_loss_avg": 0.07024642080068588,
      "epoch": 2.7704238891601563e-05,
      "step": 29050
    },
    {
      "action_loss_avg": 0.002874019474256784,
      "epoch": 2.7704238891601563e-05,
      "step": 29050
    },
    {
      "epoch": 2.7704238891601563e-05,
      "step": 29050,
      "training_step_time": 6.684579849243164
    },
    {
      "epoch": 2.7705192565917967e-05,
      "model_forward_time": 1.7532544136047363,
      "step": 29051
    },
    {
      "epoch": 2.7705192565917967e-05,
      "step": 29051,
      "training_step_time": 6.7058610916137695
    },
    {
      "epoch": 2.7706146240234375e-05,
      "model_forward_time": 1.767723560333252,
      "step": 29052
    },
    {
      "epoch": 2.7706146240234375e-05,
      "step": 29052,
      "training_step_time": 6.682891130447388
    },
    {
      "epoch": 2.7707099914550782e-05,
      "model_forward_time": 1.7057771682739258,
      "step": 29053
    },
    {
      "epoch": 2.7707099914550782e-05,
      "step": 29053,
      "training_step_time": 6.71043062210083
    },
    {
      "epoch": 2.7708053588867186e-05,
      "model_forward_time": 2.341980457305908,
      "step": 29054
    },
    {
      "epoch": 2.7708053588867186e-05,
      "step": 29054,
      "training_step_time": 7.111090183258057
    },
    {
      "epoch": 2.7709007263183593e-05,
      "model_forward_time": 1.7525708675384521,
      "step": 29055
    },
    {
      "epoch": 2.7709007263183593e-05,
      "step": 29055,
      "training_step_time": 10.806089401245117
    },
    {
      "epoch": 2.77099609375e-05,
      "model_forward_time": 1.753648281097412,
      "step": 29056
    },
    {
      "epoch": 2.77099609375e-05,
      "step": 29056,
      "training_step_time": 7.7118120193481445
    },
    {
      "epoch": 2.7710914611816405e-05,
      "model_forward_time": 1.7569196224212646,
      "step": 29057
    },
    {
      "epoch": 2.7710914611816405e-05,
      "step": 29057,
      "training_step_time": 7.646280527114868
    },
    {
      "epoch": 2.7711868286132812e-05,
      "model_forward_time": 1.7604081630706787,
      "step": 29058
    },
    {
      "epoch": 2.7711868286132812e-05,
      "step": 29058,
      "training_step_time": 6.715596675872803
    },
    {
      "epoch": 2.771282196044922e-05,
      "model_forward_time": 1.772904872894287,
      "step": 29059
    },
    {
      "epoch": 2.771282196044922e-05,
      "step": 29059,
      "training_step_time": 6.735937118530273
    },
    {
      "epoch": 2.7713775634765624e-05,
      "grad_norm": 0.13516885042190552,
      "learning_rate": 8.499222298699213e-06,
      "loss": 0.0629,
      "step": 29060
    },
    {
      "epoch": 2.7713775634765624e-05,
      "model_forward_time": 1.7478864192962646,
      "step": 29060
    },
    {
      "dynamics_loss_avg": 0.05810121260583401,
      "epoch": 2.7713775634765624e-05,
      "step": 29060
    },
    {
      "action_loss_avg": 0.004682928667170927,
      "epoch": 2.7713775634765624e-05,
      "step": 29060
    },
    {
      "epoch": 2.7713775634765624e-05,
      "step": 29060,
      "training_step_time": 6.734222173690796
    },
    {
      "epoch": 2.771472930908203e-05,
      "model_forward_time": 1.7558879852294922,
      "step": 29061
    },
    {
      "epoch": 2.771472930908203e-05,
      "step": 29061,
      "training_step_time": 6.720332384109497
    },
    {
      "epoch": 2.771568298339844e-05,
      "model_forward_time": 1.7538893222808838,
      "step": 29062
    },
    {
      "epoch": 2.771568298339844e-05,
      "step": 29062,
      "training_step_time": 6.715259552001953
    },
    {
      "epoch": 2.7716636657714843e-05,
      "model_forward_time": 1.7652010917663574,
      "step": 29063
    },
    {
      "epoch": 2.7716636657714843e-05,
      "step": 29063,
      "training_step_time": 6.9951331615448
    },
    {
      "epoch": 2.771759033203125e-05,
      "model_forward_time": 2.348421812057495,
      "step": 29064
    },
    {
      "epoch": 2.771759033203125e-05,
      "step": 29064,
      "training_step_time": 9.688485860824585
    },
    {
      "epoch": 2.7718544006347657e-05,
      "model_forward_time": 1.7633392810821533,
      "step": 29065
    },
    {
      "epoch": 2.7718544006347657e-05,
      "step": 29065,
      "training_step_time": 7.659206867218018
    },
    {
      "epoch": 2.771949768066406e-05,
      "model_forward_time": 1.7600162029266357,
      "step": 29066
    },
    {
      "epoch": 2.771949768066406e-05,
      "step": 29066,
      "training_step_time": 7.556099891662598
    },
    {
      "epoch": 2.772045135498047e-05,
      "model_forward_time": 1.766444444656372,
      "step": 29067
    },
    {
      "epoch": 2.772045135498047e-05,
      "step": 29067,
      "training_step_time": 6.7572808265686035
    },
    {
      "epoch": 2.7721405029296876e-05,
      "model_forward_time": 1.7493896484375,
      "step": 29068
    },
    {
      "epoch": 2.7721405029296876e-05,
      "step": 29068,
      "training_step_time": 6.707586050033569
    },
    {
      "epoch": 2.772235870361328e-05,
      "model_forward_time": 1.7718260288238525,
      "step": 29069
    },
    {
      "epoch": 2.772235870361328e-05,
      "step": 29069,
      "training_step_time": 6.7651207447052
    },
    {
      "epoch": 2.7723312377929688e-05,
      "grad_norm": 0.17391353845596313,
      "learning_rate": 8.4980410417167e-06,
      "loss": 0.0634,
      "step": 29070
    },
    {
      "epoch": 2.7723312377929688e-05,
      "model_forward_time": 1.039414644241333,
      "step": 29070
    },
    {
      "dynamics_loss_avg": 0.05663739591836929,
      "epoch": 2.7723312377929688e-05,
      "step": 29070
    },
    {
      "action_loss_avg": 0.0015482151706237347,
      "epoch": 2.7723312377929688e-05,
      "step": 29070
    },
    {
      "epoch": 2.7723312377929688e-05,
      "step": 29070,
      "training_step_time": 6.736853361129761
    },
    {
      "epoch": 2.7724266052246095e-05,
      "model_forward_time": 1.768756628036499,
      "step": 29071
    },
    {
      "epoch": 2.7724266052246095e-05,
      "step": 29071,
      "training_step_time": 6.7269446849823
    },
    {
      "epoch": 2.77252197265625e-05,
      "model_forward_time": 1.7683000564575195,
      "step": 29072
    },
    {
      "epoch": 2.77252197265625e-05,
      "step": 29072,
      "training_step_time": 6.765747308731079
    },
    {
      "epoch": 2.7726173400878907e-05,
      "model_forward_time": 2.3492956161499023,
      "step": 29073
    },
    {
      "epoch": 2.7726173400878907e-05,
      "step": 29073,
      "training_step_time": 10.571362972259521
    },
    {
      "epoch": 2.7727127075195314e-05,
      "model_forward_time": 1.7668073177337646,
      "step": 29074
    },
    {
      "epoch": 2.7727127075195314e-05,
      "step": 29074,
      "training_step_time": 7.800476312637329
    },
    {
      "epoch": 2.7728080749511718e-05,
      "model_forward_time": 1.7625436782836914,
      "step": 29075
    },
    {
      "epoch": 2.7728080749511718e-05,
      "step": 29075,
      "training_step_time": 7.59846830368042
    },
    {
      "epoch": 2.7729034423828125e-05,
      "model_forward_time": 1.7678070068359375,
      "step": 29076
    },
    {
      "epoch": 2.7729034423828125e-05,
      "step": 29076,
      "training_step_time": 7.539596796035767
    },
    {
      "epoch": 2.7729988098144533e-05,
      "model_forward_time": 1.7666807174682617,
      "step": 29077
    },
    {
      "epoch": 2.7729988098144533e-05,
      "step": 29077,
      "training_step_time": 6.688750505447388
    },
    {
      "epoch": 2.7730941772460937e-05,
      "model_forward_time": 1.7767798900604248,
      "step": 29078
    },
    {
      "epoch": 2.7730941772460937e-05,
      "step": 29078,
      "training_step_time": 6.744734287261963
    },
    {
      "epoch": 2.7731895446777344e-05,
      "model_forward_time": 1.7670118808746338,
      "step": 29079
    },
    {
      "epoch": 2.7731895446777344e-05,
      "step": 29079,
      "training_step_time": 6.73556113243103
    },
    {
      "epoch": 2.773284912109375e-05,
      "grad_norm": 0.28787416219711304,
      "learning_rate": 8.49685940219368e-06,
      "loss": 0.0598,
      "step": 29080
    },
    {
      "epoch": 2.773284912109375e-05,
      "model_forward_time": 1.7634203433990479,
      "step": 29080
    },
    {
      "dynamics_loss_avg": 0.0730127764865756,
      "epoch": 2.773284912109375e-05,
      "step": 29080
    },
    {
      "action_loss_avg": 0.003482423326931894,
      "epoch": 2.773284912109375e-05,
      "step": 29080
    },
    {
      "epoch": 2.773284912109375e-05,
      "step": 29080,
      "training_step_time": 6.757055759429932
    },
    {
      "epoch": 2.7733802795410156e-05,
      "model_forward_time": 1.7578699588775635,
      "step": 29081
    },
    {
      "epoch": 2.7733802795410156e-05,
      "step": 29081,
      "training_step_time": 6.735466003417969
    },
    {
      "epoch": 2.7734756469726563e-05,
      "model_forward_time": 2.354811191558838,
      "step": 29082
    },
    {
      "epoch": 2.7734756469726563e-05,
      "step": 29082,
      "training_step_time": 8.578114748001099
    },
    {
      "epoch": 2.7735710144042967e-05,
      "model_forward_time": 1.7667019367218018,
      "step": 29083
    },
    {
      "epoch": 2.7735710144042967e-05,
      "step": 29083,
      "training_step_time": 11.122091293334961
    },
    {
      "epoch": 2.7736663818359375e-05,
      "model_forward_time": 1.7713844776153564,
      "step": 29084
    },
    {
      "epoch": 2.7736663818359375e-05,
      "step": 29084,
      "training_step_time": 7.595473766326904
    },
    {
      "epoch": 2.7737617492675782e-05,
      "model_forward_time": 1.7680737972259521,
      "step": 29085
    },
    {
      "epoch": 2.7737617492675782e-05,
      "step": 29085,
      "training_step_time": 7.663820028305054
    },
    {
      "epoch": 2.7738571166992186e-05,
      "model_forward_time": 1.7610523700714111,
      "step": 29086
    },
    {
      "epoch": 2.7738571166992186e-05,
      "step": 29086,
      "training_step_time": 6.843481779098511
    },
    {
      "epoch": 2.7739524841308594e-05,
      "model_forward_time": 1.764005184173584,
      "step": 29087
    },
    {
      "epoch": 2.7739524841308594e-05,
      "step": 29087,
      "training_step_time": 6.764148712158203
    },
    {
      "epoch": 2.7740478515625e-05,
      "model_forward_time": 1.4037885665893555,
      "step": 29088
    },
    {
      "epoch": 2.7740478515625e-05,
      "step": 29088,
      "training_step_time": 6.760857105255127
    },
    {
      "epoch": 2.7741432189941405e-05,
      "model_forward_time": 1.7723326683044434,
      "step": 29089
    },
    {
      "epoch": 2.7741432189941405e-05,
      "step": 29089,
      "training_step_time": 6.778054237365723
    },
    {
      "epoch": 2.7742385864257812e-05,
      "grad_norm": 0.15450143814086914,
      "learning_rate": 8.495677380259375e-06,
      "loss": 0.0606,
      "step": 29090
    },
    {
      "epoch": 2.7742385864257812e-05,
      "model_forward_time": 1.752234697341919,
      "step": 29090
    },
    {
      "dynamics_loss_avg": 0.06255062837153673,
      "epoch": 2.7742385864257812e-05,
      "step": 29090
    },
    {
      "action_loss_avg": 0.002144214854342863,
      "epoch": 2.7742385864257812e-05,
      "step": 29090
    },
    {
      "epoch": 2.7742385864257812e-05,
      "step": 29090,
      "training_step_time": 6.76265549659729
    },
    {
      "epoch": 2.774333953857422e-05,
      "model_forward_time": 1.8026089668273926,
      "step": 29091
    },
    {
      "epoch": 2.774333953857422e-05,
      "step": 29091,
      "training_step_time": 8.782165050506592
    },
    {
      "epoch": 2.7744293212890624e-05,
      "model_forward_time": 2.3440043926239014,
      "step": 29092
    },
    {
      "epoch": 2.7744293212890624e-05,
      "step": 29092,
      "training_step_time": 8.694982767105103
    },
    {
      "epoch": 2.774524688720703e-05,
      "model_forward_time": 1.7709171772003174,
      "step": 29093
    },
    {
      "epoch": 2.774524688720703e-05,
      "step": 29093,
      "training_step_time": 8.107036828994751
    },
    {
      "epoch": 2.774620056152344e-05,
      "model_forward_time": 1.771639108657837,
      "step": 29094
    },
    {
      "epoch": 2.774620056152344e-05,
      "step": 29094,
      "training_step_time": 7.816854953765869
    },
    {
      "epoch": 2.7747154235839843e-05,
      "model_forward_time": 1.750910997390747,
      "step": 29095
    },
    {
      "epoch": 2.7747154235839843e-05,
      "step": 29095,
      "training_step_time": 6.817084074020386
    },
    {
      "epoch": 2.774810791015625e-05,
      "model_forward_time": 1.7409579753875732,
      "step": 29096
    },
    {
      "epoch": 2.774810791015625e-05,
      "step": 29096,
      "training_step_time": 6.980064630508423
    },
    {
      "epoch": 2.7749061584472657e-05,
      "model_forward_time": 1.7337679862976074,
      "step": 29097
    },
    {
      "epoch": 2.7749061584472657e-05,
      "step": 29097,
      "training_step_time": 6.958866596221924
    },
    {
      "epoch": 2.775001525878906e-05,
      "model_forward_time": 1.732529640197754,
      "step": 29098
    },
    {
      "epoch": 2.775001525878906e-05,
      "step": 29098,
      "training_step_time": 6.857923269271851
    },
    {
      "epoch": 2.775096893310547e-05,
      "model_forward_time": 1.7312967777252197,
      "step": 29099
    },
    {
      "epoch": 2.775096893310547e-05,
      "step": 29099,
      "training_step_time": 6.844707250595093
    },
    {
      "epoch": 2.7751922607421876e-05,
      "grad_norm": 0.17239809036254883,
      "learning_rate": 8.494494976043045e-06,
      "loss": 0.0606,
      "step": 29100
    },
    {
      "epoch": 2.7751922607421876e-05,
      "model_forward_time": 1.7433695793151855,
      "step": 29100
    },
    {
      "dynamics_loss_avg": 0.0592167004942894,
      "epoch": 2.7751922607421876e-05,
      "step": 29100
    },
    {
      "action_loss_avg": 0.002896430803230032,
      "epoch": 2.7751922607421876e-05,
      "step": 29100
    },
    {
      "epoch": 2.7751922607421876e-05,
      "step": 29100,
      "training_step_time": 8.539083480834961
    },
    {
      "epoch": 2.775287628173828e-05,
      "model_forward_time": 2.8586361408233643,
      "step": 29101
    },
    {
      "epoch": 2.775287628173828e-05,
      "step": 29101,
      "training_step_time": 9.651029586791992
    },
    {
      "epoch": 2.7753829956054688e-05,
      "model_forward_time": 1.371342658996582,
      "step": 29102
    },
    {
      "epoch": 2.7753829956054688e-05,
      "step": 29102,
      "training_step_time": 8.015334367752075
    },
    {
      "epoch": 2.7754783630371095e-05,
      "model_forward_time": 1.7336606979370117,
      "step": 29103
    },
    {
      "epoch": 2.7754783630371095e-05,
      "step": 29103,
      "training_step_time": 7.6879942417144775
    },
    {
      "epoch": 2.77557373046875e-05,
      "model_forward_time": 1.7334489822387695,
      "step": 29104
    },
    {
      "epoch": 2.77557373046875e-05,
      "step": 29104,
      "training_step_time": 6.86005425453186
    },
    {
      "epoch": 2.7756690979003907e-05,
      "model_forward_time": 1.7322320938110352,
      "step": 29105
    },
    {
      "epoch": 2.7756690979003907e-05,
      "step": 29105,
      "training_step_time": 6.872002840042114
    },
    {
      "epoch": 2.7757644653320314e-05,
      "model_forward_time": 1.7358100414276123,
      "step": 29106
    },
    {
      "epoch": 2.7757644653320314e-05,
      "step": 29106,
      "training_step_time": 6.905760765075684
    },
    {
      "epoch": 2.7758598327636718e-05,
      "model_forward_time": 1.7338612079620361,
      "step": 29107
    },
    {
      "epoch": 2.7758598327636718e-05,
      "step": 29107,
      "training_step_time": 6.8956992626190186
    },
    {
      "epoch": 2.7759552001953126e-05,
      "model_forward_time": 1.7331507205963135,
      "step": 29108
    },
    {
      "epoch": 2.7759552001953126e-05,
      "step": 29108,
      "training_step_time": 6.9114906787872314
    },
    {
      "epoch": 2.7760505676269533e-05,
      "model_forward_time": 1.7426645755767822,
      "step": 29109
    },
    {
      "epoch": 2.7760505676269533e-05,
      "step": 29109,
      "training_step_time": 8.978451490402222
    },
    {
      "epoch": 2.7761459350585937e-05,
      "grad_norm": 0.15000270307064056,
      "learning_rate": 8.493312189673999e-06,
      "loss": 0.0618,
      "step": 29110
    },
    {
      "epoch": 2.7761459350585937e-05,
      "model_forward_time": 2.9283204078674316,
      "step": 29110
    },
    {
      "dynamics_loss_avg": 0.05307102613151073,
      "epoch": 2.7761459350585937e-05,
      "step": 29110
    },
    {
      "action_loss_avg": 0.0015133153065107763,
      "epoch": 2.7761459350585937e-05,
      "step": 29110
    },
    {
      "epoch": 2.7761459350585937e-05,
      "step": 29110,
      "training_step_time": 8.402463912963867
    },
    {
      "epoch": 2.7762413024902344e-05,
      "model_forward_time": 1.7365520000457764,
      "step": 29111
    },
    {
      "epoch": 2.7762413024902344e-05,
      "step": 29111,
      "training_step_time": 7.805369853973389
    },
    {
      "epoch": 2.776336669921875e-05,
      "model_forward_time": 1.7350068092346191,
      "step": 29112
    },
    {
      "epoch": 2.776336669921875e-05,
      "step": 29112,
      "training_step_time": 8.424034357070923
    },
    {
      "epoch": 2.7764320373535156e-05,
      "model_forward_time": 1.7392032146453857,
      "step": 29113
    },
    {
      "epoch": 2.7764320373535156e-05,
      "step": 29113,
      "training_step_time": 6.757303237915039
    },
    {
      "epoch": 2.7765274047851563e-05,
      "model_forward_time": 1.73885178565979,
      "step": 29114
    },
    {
      "epoch": 2.7765274047851563e-05,
      "step": 29114,
      "training_step_time": 6.797322034835815
    },
    {
      "epoch": 2.7766227722167967e-05,
      "model_forward_time": 1.734708309173584,
      "step": 29115
    },
    {
      "epoch": 2.7766227722167967e-05,
      "step": 29115,
      "training_step_time": 6.6505348682403564
    },
    {
      "epoch": 2.7767181396484375e-05,
      "model_forward_time": 1.7383038997650146,
      "step": 29116
    },
    {
      "epoch": 2.7767181396484375e-05,
      "step": 29116,
      "training_step_time": 6.694659948348999
    },
    {
      "epoch": 2.7768135070800782e-05,
      "model_forward_time": 1.7381505966186523,
      "step": 29117
    },
    {
      "epoch": 2.7768135070800782e-05,
      "step": 29117,
      "training_step_time": 6.665858507156372
    },
    {
      "epoch": 2.7769088745117186e-05,
      "model_forward_time": 1.7447757720947266,
      "step": 29118
    },
    {
      "epoch": 2.7769088745117186e-05,
      "step": 29118,
      "training_step_time": 7.496198892593384
    },
    {
      "epoch": 2.7770042419433594e-05,
      "model_forward_time": 2.5192649364471436,
      "step": 29119
    },
    {
      "epoch": 2.7770042419433594e-05,
      "step": 29119,
      "training_step_time": 8.601712465286255
    },
    {
      "epoch": 2.777099609375e-05,
      "grad_norm": 0.15083526074886322,
      "learning_rate": 8.492129021281585e-06,
      "loss": 0.0636,
      "step": 29120
    },
    {
      "epoch": 2.777099609375e-05,
      "model_forward_time": 1.735550880432129,
      "step": 29120
    },
    {
      "dynamics_loss_avg": 0.04716106317937374,
      "epoch": 2.777099609375e-05,
      "step": 29120
    },
    {
      "action_loss_avg": 0.0018311269348487258,
      "epoch": 2.777099609375e-05,
      "step": 29120
    },
    {
      "epoch": 2.777099609375e-05,
      "step": 29120,
      "training_step_time": 9.44807767868042
    },
    {
      "epoch": 2.7771949768066405e-05,
      "model_forward_time": 1.7381327152252197,
      "step": 29121
    },
    {
      "epoch": 2.7771949768066405e-05,
      "step": 29121,
      "training_step_time": 7.79885721206665
    },
    {
      "epoch": 2.7772903442382812e-05,
      "model_forward_time": 1.743746280670166,
      "step": 29122
    },
    {
      "epoch": 2.7772903442382812e-05,
      "step": 29122,
      "training_step_time": 7.601653099060059
    },
    {
      "epoch": 2.777385711669922e-05,
      "model_forward_time": 1.7430493831634521,
      "step": 29123
    },
    {
      "epoch": 2.777385711669922e-05,
      "step": 29123,
      "training_step_time": 6.752146482467651
    },
    {
      "epoch": 2.7774810791015624e-05,
      "model_forward_time": 1.7398293018341064,
      "step": 29124
    },
    {
      "epoch": 2.7774810791015624e-05,
      "step": 29124,
      "training_step_time": 6.777569770812988
    },
    {
      "epoch": 2.777576446533203e-05,
      "model_forward_time": 1.739234209060669,
      "step": 29125
    },
    {
      "epoch": 2.777576446533203e-05,
      "step": 29125,
      "training_step_time": 6.759669542312622
    },
    {
      "epoch": 2.777671813964844e-05,
      "model_forward_time": 1.7414238452911377,
      "step": 29126
    },
    {
      "epoch": 2.777671813964844e-05,
      "step": 29126,
      "training_step_time": 6.808602809906006
    },
    {
      "epoch": 2.7777671813964843e-05,
      "model_forward_time": 1.7340304851531982,
      "step": 29127
    },
    {
      "epoch": 2.7777671813964843e-05,
      "step": 29127,
      "training_step_time": 7.535779714584351
    },
    {
      "epoch": 2.777862548828125e-05,
      "model_forward_time": 1.8155386447906494,
      "step": 29128
    },
    {
      "epoch": 2.777862548828125e-05,
      "step": 29128,
      "training_step_time": 8.860517501831055
    },
    {
      "epoch": 2.7779579162597658e-05,
      "model_forward_time": 2.912243127822876,
      "step": 29129
    },
    {
      "epoch": 2.7779579162597658e-05,
      "step": 29129,
      "training_step_time": 9.278574705123901
    },
    {
      "epoch": 2.778053283691406e-05,
      "grad_norm": 0.21715512871742249,
      "learning_rate": 8.49094547099519e-06,
      "loss": 0.0642,
      "step": 29130
    },
    {
      "epoch": 2.778053283691406e-05,
      "model_forward_time": 1.744960069656372,
      "step": 29130
    },
    {
      "dynamics_loss_avg": 0.056866628117859364,
      "epoch": 2.778053283691406e-05,
      "step": 29130
    },
    {
      "action_loss_avg": 0.0026687397330533713,
      "epoch": 2.778053283691406e-05,
      "step": 29130
    },
    {
      "epoch": 2.778053283691406e-05,
      "step": 29130,
      "training_step_time": 7.807122707366943
    },
    {
      "epoch": 2.778148651123047e-05,
      "model_forward_time": 1.7448065280914307,
      "step": 29131
    },
    {
      "epoch": 2.778148651123047e-05,
      "step": 29131,
      "training_step_time": 7.6479811668396
    },
    {
      "epoch": 2.7782440185546876e-05,
      "model_forward_time": 1.7373359203338623,
      "step": 29132
    },
    {
      "epoch": 2.7782440185546876e-05,
      "step": 29132,
      "training_step_time": 6.7252583503723145
    },
    {
      "epoch": 2.778339385986328e-05,
      "model_forward_time": 1.7334752082824707,
      "step": 29133
    },
    {
      "epoch": 2.778339385986328e-05,
      "step": 29133,
      "training_step_time": 6.735658407211304
    },
    {
      "epoch": 2.7784347534179688e-05,
      "model_forward_time": 1.73592209815979,
      "step": 29134
    },
    {
      "epoch": 2.7784347534179688e-05,
      "step": 29134,
      "training_step_time": 6.757227182388306
    },
    {
      "epoch": 2.7785301208496095e-05,
      "model_forward_time": 1.3566486835479736,
      "step": 29135
    },
    {
      "epoch": 2.7785301208496095e-05,
      "step": 29135,
      "training_step_time": 6.726044654846191
    },
    {
      "epoch": 2.77862548828125e-05,
      "model_forward_time": 1.7360520362854004,
      "step": 29136
    },
    {
      "epoch": 2.77862548828125e-05,
      "step": 29136,
      "training_step_time": 7.520642995834351
    },
    {
      "epoch": 2.7787208557128907e-05,
      "model_forward_time": 1.7532298564910889,
      "step": 29137
    },
    {
      "epoch": 2.7787208557128907e-05,
      "step": 29137,
      "training_step_time": 8.587558031082153
    },
    {
      "epoch": 2.7788162231445314e-05,
      "model_forward_time": 2.8359122276306152,
      "step": 29138
    },
    {
      "epoch": 2.7788162231445314e-05,
      "step": 29138,
      "training_step_time": 8.664786100387573
    },
    {
      "epoch": 2.7789115905761718e-05,
      "model_forward_time": 1.7319602966308594,
      "step": 29139
    },
    {
      "epoch": 2.7789115905761718e-05,
      "step": 29139,
      "training_step_time": 7.641173839569092
    },
    {
      "epoch": 2.7790069580078126e-05,
      "grad_norm": 0.11653042584657669,
      "learning_rate": 8.489761538944248e-06,
      "loss": 0.0626,
      "step": 29140
    },
    {
      "epoch": 2.7790069580078126e-05,
      "model_forward_time": 1.734882116317749,
      "step": 29140
    },
    {
      "dynamics_loss_avg": 0.05806495286524296,
      "epoch": 2.7790069580078126e-05,
      "step": 29140
    },
    {
      "action_loss_avg": 0.002207105988054536,
      "epoch": 2.7790069580078126e-05,
      "step": 29140
    },
    {
      "epoch": 2.7790069580078126e-05,
      "step": 29140,
      "training_step_time": 7.54840612411499
    },
    {
      "epoch": 2.779102325439453e-05,
      "model_forward_time": 1.7335083484649658,
      "step": 29141
    },
    {
      "epoch": 2.779102325439453e-05,
      "step": 29141,
      "training_step_time": 6.713327646255493
    },
    {
      "epoch": 2.7791976928710937e-05,
      "model_forward_time": 1.7251574993133545,
      "step": 29142
    },
    {
      "epoch": 2.7791976928710937e-05,
      "step": 29142,
      "training_step_time": 6.7395477294921875
    },
    {
      "epoch": 2.7792930603027344e-05,
      "model_forward_time": 1.7402124404907227,
      "step": 29143
    },
    {
      "epoch": 2.7792930603027344e-05,
      "step": 29143,
      "training_step_time": 6.723901033401489
    },
    {
      "epoch": 2.779388427734375e-05,
      "model_forward_time": 1.7114553451538086,
      "step": 29144
    },
    {
      "epoch": 2.779388427734375e-05,
      "step": 29144,
      "training_step_time": 6.710287094116211
    },
    {
      "epoch": 2.7794837951660156e-05,
      "model_forward_time": 1.0172159671783447,
      "step": 29145
    },
    {
      "epoch": 2.7794837951660156e-05,
      "step": 29145,
      "training_step_time": 7.524846315383911
    },
    {
      "epoch": 2.7795791625976563e-05,
      "model_forward_time": 1.732445478439331,
      "step": 29146
    },
    {
      "epoch": 2.7795791625976563e-05,
      "step": 29146,
      "training_step_time": 8.357326030731201
    },
    {
      "epoch": 2.7796745300292967e-05,
      "model_forward_time": 1.754758596420288,
      "step": 29147
    },
    {
      "epoch": 2.7796745300292967e-05,
      "step": 29147,
      "training_step_time": 8.74805736541748
    },
    {
      "epoch": 2.7797698974609375e-05,
      "model_forward_time": 1.731156826019287,
      "step": 29148
    },
    {
      "epoch": 2.7797698974609375e-05,
      "step": 29148,
      "training_step_time": 8.706824541091919
    },
    {
      "epoch": 2.7798652648925782e-05,
      "model_forward_time": 1.738753318786621,
      "step": 29149
    },
    {
      "epoch": 2.7798652648925782e-05,
      "step": 29149,
      "training_step_time": 7.508314609527588
    },
    {
      "epoch": 2.7799606323242186e-05,
      "grad_norm": 0.15641023218631744,
      "learning_rate": 8.48857722525823e-06,
      "loss": 0.0594,
      "step": 29150
    },
    {
      "epoch": 2.7799606323242186e-05,
      "model_forward_time": 1.724226713180542,
      "step": 29150
    },
    {
      "dynamics_loss_avg": 0.059926499426364896,
      "epoch": 2.7799606323242186e-05,
      "step": 29150
    },
    {
      "action_loss_avg": 0.005333604186307639,
      "epoch": 2.7799606323242186e-05,
      "step": 29150
    },
    {
      "epoch": 2.7799606323242186e-05,
      "step": 29150,
      "training_step_time": 7.130021095275879
    },
    {
      "epoch": 2.7800559997558594e-05,
      "model_forward_time": 1.7330703735351562,
      "step": 29151
    },
    {
      "epoch": 2.7800559997558594e-05,
      "step": 29151,
      "training_step_time": 6.6780335903167725
    },
    {
      "epoch": 2.7801513671875e-05,
      "model_forward_time": 1.7386796474456787,
      "step": 29152
    },
    {
      "epoch": 2.7801513671875e-05,
      "step": 29152,
      "training_step_time": 6.683550119400024
    },
    {
      "epoch": 2.7802467346191405e-05,
      "model_forward_time": 1.7375428676605225,
      "step": 29153
    },
    {
      "epoch": 2.7802467346191405e-05,
      "step": 29153,
      "training_step_time": 6.765605926513672
    },
    {
      "epoch": 2.7803421020507812e-05,
      "model_forward_time": 1.7346813678741455,
      "step": 29154
    },
    {
      "epoch": 2.7803421020507812e-05,
      "step": 29154,
      "training_step_time": 7.662146329879761
    },
    {
      "epoch": 2.780437469482422e-05,
      "model_forward_time": 1.7336997985839844,
      "step": 29155
    },
    {
      "epoch": 2.780437469482422e-05,
      "step": 29155,
      "training_step_time": 8.471491813659668
    },
    {
      "epoch": 2.7805328369140624e-05,
      "model_forward_time": 1.7529120445251465,
      "step": 29156
    },
    {
      "epoch": 2.7805328369140624e-05,
      "step": 29156,
      "training_step_time": 9.177096843719482
    },
    {
      "epoch": 2.780628204345703e-05,
      "model_forward_time": 2.921233892440796,
      "step": 29157
    },
    {
      "epoch": 2.780628204345703e-05,
      "step": 29157,
      "training_step_time": 8.513732671737671
    },
    {
      "epoch": 2.780723571777344e-05,
      "model_forward_time": 1.7336151599884033,
      "step": 29158
    },
    {
      "epoch": 2.780723571777344e-05,
      "step": 29158,
      "training_step_time": 7.784040212631226
    },
    {
      "epoch": 2.7808189392089843e-05,
      "model_forward_time": 1.7347581386566162,
      "step": 29159
    },
    {
      "epoch": 2.7808189392089843e-05,
      "step": 29159,
      "training_step_time": 7.756582498550415
    },
    {
      "epoch": 2.780914306640625e-05,
      "grad_norm": 0.14937302470207214,
      "learning_rate": 8.487392530066651e-06,
      "loss": 0.0615,
      "step": 29160
    },
    {
      "epoch": 2.780914306640625e-05,
      "model_forward_time": 1.728529930114746,
      "step": 29160
    },
    {
      "dynamics_loss_avg": 0.05522876139730215,
      "epoch": 2.780914306640625e-05,
      "step": 29160
    },
    {
      "action_loss_avg": 0.001568809460150078,
      "epoch": 2.780914306640625e-05,
      "step": 29160
    },
    {
      "epoch": 2.780914306640625e-05,
      "step": 29160,
      "training_step_time": 6.804904937744141
    },
    {
      "epoch": 2.7810096740722658e-05,
      "model_forward_time": 0.6907758712768555,
      "step": 29161
    },
    {
      "epoch": 2.7810096740722658e-05,
      "step": 29161,
      "training_step_time": 6.721081495285034
    },
    {
      "epoch": 2.781105041503906e-05,
      "model_forward_time": 1.7268831729888916,
      "step": 29162
    },
    {
      "epoch": 2.781105041503906e-05,
      "step": 29162,
      "training_step_time": 6.726582765579224
    },
    {
      "epoch": 2.781200408935547e-05,
      "model_forward_time": 1.7363955974578857,
      "step": 29163
    },
    {
      "epoch": 2.781200408935547e-05,
      "step": 29163,
      "training_step_time": 7.586397171020508
    },
    {
      "epoch": 2.7812957763671876e-05,
      "model_forward_time": 1.7352056503295898,
      "step": 29164
    },
    {
      "epoch": 2.7812957763671876e-05,
      "step": 29164,
      "training_step_time": 7.769057273864746
    },
    {
      "epoch": 2.781391143798828e-05,
      "model_forward_time": 1.7479310035705566,
      "step": 29165
    },
    {
      "epoch": 2.781391143798828e-05,
      "step": 29165,
      "training_step_time": 8.651153802871704
    },
    {
      "epoch": 2.7814865112304688e-05,
      "model_forward_time": 1.7459900379180908,
      "step": 29166
    },
    {
      "epoch": 2.7814865112304688e-05,
      "step": 29166,
      "training_step_time": 9.556141376495361
    },
    {
      "epoch": 2.7815818786621095e-05,
      "model_forward_time": 1.7300291061401367,
      "step": 29167
    },
    {
      "epoch": 2.7815818786621095e-05,
      "step": 29167,
      "training_step_time": 7.718748331069946
    },
    {
      "epoch": 2.78167724609375e-05,
      "model_forward_time": 1.7358229160308838,
      "step": 29168
    },
    {
      "epoch": 2.78167724609375e-05,
      "step": 29168,
      "training_step_time": 7.6770336627960205
    },
    {
      "epoch": 2.7817726135253907e-05,
      "model_forward_time": 1.7361750602722168,
      "step": 29169
    },
    {
      "epoch": 2.7817726135253907e-05,
      "step": 29169,
      "training_step_time": 6.696916818618774
    },
    {
      "epoch": 2.7818679809570314e-05,
      "grad_norm": 0.13142237067222595,
      "learning_rate": 8.48620745349907e-06,
      "loss": 0.0552,
      "step": 29170
    },
    {
      "epoch": 2.7818679809570314e-05,
      "model_forward_time": 1.7350213527679443,
      "step": 29170
    },
    {
      "dynamics_loss_avg": 0.0677266463637352,
      "epoch": 2.7818679809570314e-05,
      "step": 29170
    },
    {
      "action_loss_avg": 0.002712129458086565,
      "epoch": 2.7818679809570314e-05,
      "step": 29170
    },
    {
      "epoch": 2.7818679809570314e-05,
      "step": 29170,
      "training_step_time": 6.714181900024414
    },
    {
      "epoch": 2.7819633483886718e-05,
      "model_forward_time": 1.7366931438446045,
      "step": 29171
    },
    {
      "epoch": 2.7819633483886718e-05,
      "step": 29171,
      "training_step_time": 6.723135709762573
    },
    {
      "epoch": 2.7820587158203126e-05,
      "model_forward_time": 1.7368223667144775,
      "step": 29172
    },
    {
      "epoch": 2.7820587158203126e-05,
      "step": 29172,
      "training_step_time": 8.498390197753906
    },
    {
      "epoch": 2.782154083251953e-05,
      "model_forward_time": 1.737567663192749,
      "step": 29173
    },
    {
      "epoch": 2.782154083251953e-05,
      "step": 29173,
      "training_step_time": 7.665610313415527
    },
    {
      "epoch": 2.7822494506835937e-05,
      "model_forward_time": 1.7526068687438965,
      "step": 29174
    },
    {
      "epoch": 2.7822494506835937e-05,
      "step": 29174,
      "training_step_time": 7.5822999477386475
    },
    {
      "epoch": 2.7823448181152344e-05,
      "model_forward_time": 1.7450511455535889,
      "step": 29175
    },
    {
      "epoch": 2.7823448181152344e-05,
      "step": 29175,
      "training_step_time": 8.779157876968384
    },
    {
      "epoch": 2.782440185546875e-05,
      "model_forward_time": 2.9242591857910156,
      "step": 29176
    },
    {
      "epoch": 2.782440185546875e-05,
      "step": 29176,
      "training_step_time": 7.691287517547607
    },
    {
      "epoch": 2.7825355529785156e-05,
      "model_forward_time": 1.719193458557129,
      "step": 29177
    },
    {
      "epoch": 2.7825355529785156e-05,
      "step": 29177,
      "training_step_time": 7.214181184768677
    },
    {
      "epoch": 2.7826309204101563e-05,
      "model_forward_time": 1.737917423248291,
      "step": 29178
    },
    {
      "epoch": 2.7826309204101563e-05,
      "step": 29178,
      "training_step_time": 7.575108289718628
    },
    {
      "epoch": 2.7827262878417967e-05,
      "model_forward_time": 1.7302768230438232,
      "step": 29179
    },
    {
      "epoch": 2.7827262878417967e-05,
      "step": 29179,
      "training_step_time": 6.673715829849243
    },
    {
      "epoch": 2.7828216552734375e-05,
      "grad_norm": 0.14733268320560455,
      "learning_rate": 8.485021995685081e-06,
      "loss": 0.0604,
      "step": 29180
    },
    {
      "epoch": 2.7828216552734375e-05,
      "model_forward_time": 1.7328426837921143,
      "step": 29180
    },
    {
      "dynamics_loss_avg": 0.054300214163959026,
      "epoch": 2.7828216552734375e-05,
      "step": 29180
    },
    {
      "action_loss_avg": 0.0016775445197708904,
      "epoch": 2.7828216552734375e-05,
      "step": 29180
    },
    {
      "epoch": 2.7828216552734375e-05,
      "step": 29180,
      "training_step_time": 6.731323957443237
    },
    {
      "epoch": 2.7829170227050782e-05,
      "model_forward_time": 1.7291958332061768,
      "step": 29181
    },
    {
      "epoch": 2.7829170227050782e-05,
      "step": 29181,
      "training_step_time": 7.474995851516724
    },
    {
      "epoch": 2.7830123901367186e-05,
      "model_forward_time": 1.7407801151275635,
      "step": 29182
    },
    {
      "epoch": 2.7830123901367186e-05,
      "step": 29182,
      "training_step_time": 7.6569249629974365
    },
    {
      "epoch": 2.7831077575683594e-05,
      "model_forward_time": 1.7614858150482178,
      "step": 29183
    },
    {
      "epoch": 2.7831077575683594e-05,
      "step": 29183,
      "training_step_time": 8.288886070251465
    },
    {
      "epoch": 2.783203125e-05,
      "model_forward_time": 1.7487969398498535,
      "step": 29184
    },
    {
      "epoch": 2.783203125e-05,
      "step": 29184,
      "training_step_time": 7.672912120819092
    },
    {
      "epoch": 2.7832984924316405e-05,
      "model_forward_time": 2.8608243465423584,
      "step": 29185
    },
    {
      "epoch": 2.7832984924316405e-05,
      "step": 29185,
      "training_step_time": 8.766375064849854
    },
    {
      "epoch": 2.7833938598632812e-05,
      "model_forward_time": 1.7414507865905762,
      "step": 29186
    },
    {
      "epoch": 2.7833938598632812e-05,
      "step": 29186,
      "training_step_time": 7.585507869720459
    },
    {
      "epoch": 2.783489227294922e-05,
      "model_forward_time": 1.7281091213226318,
      "step": 29187
    },
    {
      "epoch": 2.783489227294922e-05,
      "step": 29187,
      "training_step_time": 7.5746424198150635
    },
    {
      "epoch": 2.7835845947265624e-05,
      "model_forward_time": 1.7343692779541016,
      "step": 29188
    },
    {
      "epoch": 2.7835845947265624e-05,
      "step": 29188,
      "training_step_time": 6.69184684753418
    },
    {
      "epoch": 2.783679962158203e-05,
      "model_forward_time": 1.73160719871521,
      "step": 29189
    },
    {
      "epoch": 2.783679962158203e-05,
      "step": 29189,
      "training_step_time": 6.745621681213379
    },
    {
      "epoch": 2.783775329589844e-05,
      "grad_norm": 0.09477473795413971,
      "learning_rate": 8.483836156754328e-06,
      "loss": 0.0555,
      "step": 29190
    },
    {
      "epoch": 2.783775329589844e-05,
      "model_forward_time": 1.7298612594604492,
      "step": 29190
    },
    {
      "dynamics_loss_avg": 0.05229340735822916,
      "epoch": 2.783775329589844e-05,
      "step": 29190
    },
    {
      "action_loss_avg": 0.0021287286479491742,
      "epoch": 2.783775329589844e-05,
      "step": 29190
    },
    {
      "epoch": 2.783775329589844e-05,
      "step": 29190,
      "training_step_time": 7.489861726760864
    },
    {
      "epoch": 2.7838706970214843e-05,
      "model_forward_time": 1.7310502529144287,
      "step": 29191
    },
    {
      "epoch": 2.7838706970214843e-05,
      "step": 29191,
      "training_step_time": 8.583845853805542
    },
    {
      "epoch": 2.783966064453125e-05,
      "model_forward_time": 1.751774787902832,
      "step": 29192
    },
    {
      "epoch": 2.783966064453125e-05,
      "step": 29192,
      "training_step_time": 9.781617164611816
    },
    {
      "epoch": 2.7840614318847658e-05,
      "model_forward_time": 1.7579247951507568,
      "step": 29193
    },
    {
      "epoch": 2.7840614318847658e-05,
      "step": 29193,
      "training_step_time": 8.662015199661255
    },
    {
      "epoch": 2.784156799316406e-05,
      "model_forward_time": 2.886977195739746,
      "step": 29194
    },
    {
      "epoch": 2.784156799316406e-05,
      "step": 29194,
      "training_step_time": 7.705620527267456
    },
    {
      "epoch": 2.784252166748047e-05,
      "model_forward_time": 1.7153332233428955,
      "step": 29195
    },
    {
      "epoch": 2.784252166748047e-05,
      "step": 29195,
      "training_step_time": 7.5732409954071045
    },
    {
      "epoch": 2.7843475341796876e-05,
      "model_forward_time": 1.7230281829833984,
      "step": 29196
    },
    {
      "epoch": 2.7843475341796876e-05,
      "step": 29196,
      "training_step_time": 7.174489498138428
    },
    {
      "epoch": 2.784442901611328e-05,
      "model_forward_time": 1.713066577911377,
      "step": 29197
    },
    {
      "epoch": 2.784442901611328e-05,
      "step": 29197,
      "training_step_time": 6.640647649765015
    },
    {
      "epoch": 2.7845382690429688e-05,
      "model_forward_time": 1.7154860496520996,
      "step": 29198
    },
    {
      "epoch": 2.7845382690429688e-05,
      "step": 29198,
      "training_step_time": 6.656325817108154
    },
    {
      "epoch": 2.7846336364746095e-05,
      "model_forward_time": 1.6985056400299072,
      "step": 29199
    },
    {
      "epoch": 2.7846336364746095e-05,
      "step": 29199,
      "training_step_time": 7.33629846572876
    },
    {
      "epoch": 2.78472900390625e-05,
      "grad_norm": 0.12745901942253113,
      "learning_rate": 8.482649936836491e-06,
      "loss": 0.0588,
      "step": 29200
    },
    {
      "epoch": 2.78472900390625e-05,
      "model_forward_time": 1.7166106700897217,
      "step": 29200
    },
    {
      "dynamics_loss_avg": 0.04618074381724,
      "epoch": 2.78472900390625e-05,
      "step": 29200
    },
    {
      "action_loss_avg": 0.0016487006971146912,
      "epoch": 2.78472900390625e-05,
      "step": 29200
    },
    {
      "epoch": 2.78472900390625e-05,
      "step": 29200,
      "training_step_time": 7.565610408782959
    },
    {
      "epoch": 2.7848243713378907e-05,
      "model_forward_time": 1.7316911220550537,
      "step": 29201
    },
    {
      "epoch": 2.7848243713378907e-05,
      "step": 29201,
      "training_step_time": 8.46632194519043
    },
    {
      "epoch": 2.7849197387695314e-05,
      "model_forward_time": 1.7361993789672852,
      "step": 29202
    },
    {
      "epoch": 2.7849197387695314e-05,
      "step": 29202,
      "training_step_time": 8.271241903305054
    },
    {
      "epoch": 2.7850151062011718e-05,
      "model_forward_time": 2.816710948944092,
      "step": 29203
    },
    {
      "epoch": 2.7850151062011718e-05,
      "step": 29203,
      "training_step_time": 9.731021404266357
    },
    {
      "epoch": 2.7851104736328126e-05,
      "model_forward_time": 1.7206757068634033,
      "step": 29204
    },
    {
      "epoch": 2.7851104736328126e-05,
      "step": 29204,
      "training_step_time": 7.570587873458862
    },
    {
      "epoch": 2.785205841064453e-05,
      "model_forward_time": 1.7236924171447754,
      "step": 29205
    },
    {
      "epoch": 2.785205841064453e-05,
      "step": 29205,
      "training_step_time": 7.559511184692383
    },
    {
      "epoch": 2.7853012084960937e-05,
      "model_forward_time": 1.7190327644348145,
      "step": 29206
    },
    {
      "epoch": 2.7853012084960937e-05,
      "step": 29206,
      "training_step_time": 6.624568462371826
    },
    {
      "epoch": 2.7853965759277344e-05,
      "model_forward_time": 1.718702793121338,
      "step": 29207
    },
    {
      "epoch": 2.7853965759277344e-05,
      "step": 29207,
      "training_step_time": 6.625316143035889
    },
    {
      "epoch": 2.785491943359375e-05,
      "model_forward_time": 1.7151341438293457,
      "step": 29208
    },
    {
      "epoch": 2.785491943359375e-05,
      "step": 29208,
      "training_step_time": 7.430057525634766
    },
    {
      "epoch": 2.7855873107910156e-05,
      "model_forward_time": 1.7263176441192627,
      "step": 29209
    },
    {
      "epoch": 2.7855873107910156e-05,
      "step": 29209,
      "training_step_time": 7.4625585079193115
    },
    {
      "epoch": 2.7856826782226563e-05,
      "grad_norm": 0.11463917046785355,
      "learning_rate": 8.481463336061293e-06,
      "loss": 0.0609,
      "step": 29210
    },
    {
      "epoch": 2.7856826782226563e-05,
      "model_forward_time": 1.719451904296875,
      "step": 29210
    },
    {
      "dynamics_loss_avg": 0.051232786290347575,
      "epoch": 2.7856826782226563e-05,
      "step": 29210
    },
    {
      "action_loss_avg": 0.0018974307575263083,
      "epoch": 2.7856826782226563e-05,
      "step": 29210
    },
    {
      "epoch": 2.7856826782226563e-05,
      "step": 29210,
      "training_step_time": 8.47462511062622
    },
    {
      "epoch": 2.7857780456542967e-05,
      "model_forward_time": 1.8020656108856201,
      "step": 29211
    },
    {
      "epoch": 2.7857780456542967e-05,
      "step": 29211,
      "training_step_time": 9.450350046157837
    },
    {
      "epoch": 2.7858734130859375e-05,
      "model_forward_time": 1.7127766609191895,
      "step": 29212
    },
    {
      "epoch": 2.7858734130859375e-05,
      "step": 29212,
      "training_step_time": 11.004793405532837
    },
    {
      "epoch": 2.7859687805175782e-05,
      "model_forward_time": 1.7192230224609375,
      "step": 29213
    },
    {
      "epoch": 2.7859687805175782e-05,
      "step": 29213,
      "training_step_time": 7.6690146923065186
    },
    {
      "epoch": 2.7860641479492186e-05,
      "model_forward_time": 1.365466833114624,
      "step": 29214
    },
    {
      "epoch": 2.7860641479492186e-05,
      "step": 29214,
      "training_step_time": 7.575382947921753
    },
    {
      "epoch": 2.7861595153808594e-05,
      "model_forward_time": 1.7262804508209229,
      "step": 29215
    },
    {
      "epoch": 2.7861595153808594e-05,
      "step": 29215,
      "training_step_time": 6.704535722732544
    },
    {
      "epoch": 2.7862548828125e-05,
      "model_forward_time": 1.3572094440460205,
      "step": 29216
    },
    {
      "epoch": 2.7862548828125e-05,
      "step": 29216,
      "training_step_time": 6.665351390838623
    },
    {
      "epoch": 2.7863502502441405e-05,
      "model_forward_time": 1.7089338302612305,
      "step": 29217
    },
    {
      "epoch": 2.7863502502441405e-05,
      "step": 29217,
      "training_step_time": 7.448454141616821
    },
    {
      "epoch": 2.7864456176757812e-05,
      "model_forward_time": 1.7151970863342285,
      "step": 29218
    },
    {
      "epoch": 2.7864456176757812e-05,
      "step": 29218,
      "training_step_time": 8.438350915908813
    },
    {
      "epoch": 2.786540985107422e-05,
      "model_forward_time": 1.7171711921691895,
      "step": 29219
    },
    {
      "epoch": 2.786540985107422e-05,
      "step": 29219,
      "training_step_time": 7.6390061378479
    },
    {
      "epoch": 2.7866363525390624e-05,
      "grad_norm": 0.13979050517082214,
      "learning_rate": 8.480276354558497e-06,
      "loss": 0.06,
      "step": 29220
    },
    {
      "epoch": 2.7866363525390624e-05,
      "model_forward_time": 1.8047709465026855,
      "step": 29220
    },
    {
      "dynamics_loss_avg": 0.055353238340467216,
      "epoch": 2.7866363525390624e-05,
      "step": 29220
    },
    {
      "action_loss_avg": 0.0021702597150579095,
      "epoch": 2.7866363525390624e-05,
      "step": 29220
    },
    {
      "epoch": 2.7866363525390624e-05,
      "step": 29220,
      "training_step_time": 7.629392147064209
    },
    {
      "epoch": 2.786731719970703e-05,
      "model_forward_time": 1.7195026874542236,
      "step": 29221
    },
    {
      "epoch": 2.786731719970703e-05,
      "step": 29221,
      "training_step_time": 8.72459363937378
    },
    {
      "epoch": 2.786827087402344e-05,
      "model_forward_time": 2.473442316055298,
      "step": 29222
    },
    {
      "epoch": 2.786827087402344e-05,
      "step": 29222,
      "training_step_time": 7.6736884117126465
    },
    {
      "epoch": 2.7869224548339843e-05,
      "model_forward_time": 1.7192275524139404,
      "step": 29223
    },
    {
      "epoch": 2.7869224548339843e-05,
      "step": 29223,
      "training_step_time": 7.584559917449951
    },
    {
      "epoch": 2.787017822265625e-05,
      "model_forward_time": 1.7159204483032227,
      "step": 29224
    },
    {
      "epoch": 2.787017822265625e-05,
      "step": 29224,
      "training_step_time": 7.557071685791016
    },
    {
      "epoch": 2.7871131896972658e-05,
      "model_forward_time": 1.7176921367645264,
      "step": 29225
    },
    {
      "epoch": 2.7871131896972658e-05,
      "step": 29225,
      "training_step_time": 6.678713083267212
    },
    {
      "epoch": 2.787208557128906e-05,
      "model_forward_time": 1.7172877788543701,
      "step": 29226
    },
    {
      "epoch": 2.787208557128906e-05,
      "step": 29226,
      "training_step_time": 7.460236310958862
    },
    {
      "epoch": 2.787303924560547e-05,
      "model_forward_time": 1.7213220596313477,
      "step": 29227
    },
    {
      "epoch": 2.787303924560547e-05,
      "step": 29227,
      "training_step_time": 7.906993627548218
    },
    {
      "epoch": 2.7873992919921876e-05,
      "model_forward_time": 1.7181012630462646,
      "step": 29228
    },
    {
      "epoch": 2.7873992919921876e-05,
      "step": 29228,
      "training_step_time": 7.559831619262695
    },
    {
      "epoch": 2.787494659423828e-05,
      "model_forward_time": 1.804335117340088,
      "step": 29229
    },
    {
      "epoch": 2.787494659423828e-05,
      "step": 29229,
      "training_step_time": 7.657864332199097
    },
    {
      "epoch": 2.7875900268554688e-05,
      "grad_norm": 0.11428940296173096,
      "learning_rate": 8.479088992457913e-06,
      "loss": 0.0589,
      "step": 29230
    },
    {
      "epoch": 2.7875900268554688e-05,
      "model_forward_time": 1.7189958095550537,
      "step": 29230
    },
    {
      "dynamics_loss_avg": 0.05459529459476471,
      "epoch": 2.7875900268554688e-05,
      "step": 29230
    },
    {
      "action_loss_avg": 0.00200025849044323,
      "epoch": 2.7875900268554688e-05,
      "step": 29230
    },
    {
      "epoch": 2.7875900268554688e-05,
      "step": 29230,
      "training_step_time": 11.99381160736084
    },
    {
      "epoch": 2.7876853942871095e-05,
      "model_forward_time": 2.823725938796997,
      "step": 29231
    },
    {
      "epoch": 2.7876853942871095e-05,
      "step": 29231,
      "training_step_time": 8.642755031585693
    },
    {
      "epoch": 2.78778076171875e-05,
      "model_forward_time": 1.72206711769104,
      "step": 29232
    },
    {
      "epoch": 2.78778076171875e-05,
      "step": 29232,
      "training_step_time": 7.578789234161377
    },
    {
      "epoch": 2.7878761291503907e-05,
      "model_forward_time": 1.7148330211639404,
      "step": 29233
    },
    {
      "epoch": 2.7878761291503907e-05,
      "step": 29233,
      "training_step_time": 7.667084455490112
    },
    {
      "epoch": 2.7879714965820314e-05,
      "model_forward_time": 1.719766616821289,
      "step": 29234
    },
    {
      "epoch": 2.7879714965820314e-05,
      "step": 29234,
      "training_step_time": 6.750688552856445
    },
    {
      "epoch": 2.7880668640136718e-05,
      "model_forward_time": 0.6869964599609375,
      "step": 29235
    },
    {
      "epoch": 2.7880668640136718e-05,
      "step": 29235,
      "training_step_time": 7.4474263191223145
    },
    {
      "epoch": 2.7881622314453126e-05,
      "model_forward_time": 1.719191551208496,
      "step": 29236
    },
    {
      "epoch": 2.7881622314453126e-05,
      "step": 29236,
      "training_step_time": 8.243870735168457
    },
    {
      "epoch": 2.788257598876953e-05,
      "model_forward_time": 1.7214186191558838,
      "step": 29237
    },
    {
      "epoch": 2.788257598876953e-05,
      "step": 29237,
      "training_step_time": 9.425659656524658
    },
    {
      "epoch": 2.7883529663085937e-05,
      "model_forward_time": 1.801912546157837,
      "step": 29238
    },
    {
      "epoch": 2.7883529663085937e-05,
      "step": 29238,
      "training_step_time": 11.24075722694397
    },
    {
      "epoch": 2.7884483337402344e-05,
      "model_forward_time": 1.7154383659362793,
      "step": 29239
    },
    {
      "epoch": 2.7884483337402344e-05,
      "step": 29239,
      "training_step_time": 8.761638879776001
    },
    {
      "epoch": 2.788543701171875e-05,
      "grad_norm": 0.15378060936927795,
      "learning_rate": 8.477901249889388e-06,
      "loss": 0.0636,
      "step": 29240
    },
    {
      "epoch": 2.788543701171875e-05,
      "model_forward_time": 1.7133359909057617,
      "step": 29240
    },
    {
      "dynamics_loss_avg": 0.06303909355774522,
      "epoch": 2.788543701171875e-05,
      "step": 29240
    },
    {
      "action_loss_avg": 0.0020193062257021664,
      "epoch": 2.788543701171875e-05,
      "step": 29240
    },
    {
      "epoch": 2.788543701171875e-05,
      "step": 29240,
      "training_step_time": 7.553747177124023
    },
    {
      "epoch": 2.7886390686035156e-05,
      "model_forward_time": 2.8543763160705566,
      "step": 29241
    },
    {
      "epoch": 2.7886390686035156e-05,
      "step": 29241,
      "training_step_time": 7.60048770904541
    },
    {
      "epoch": 2.7887344360351563e-05,
      "model_forward_time": 1.718181848526001,
      "step": 29242
    },
    {
      "epoch": 2.7887344360351563e-05,
      "step": 29242,
      "training_step_time": 7.532237529754639
    },
    {
      "epoch": 2.7888298034667967e-05,
      "model_forward_time": 1.7241544723510742,
      "step": 29243
    },
    {
      "epoch": 2.7888298034667967e-05,
      "step": 29243,
      "training_step_time": 6.651981592178345
    },
    {
      "epoch": 2.7889251708984375e-05,
      "model_forward_time": 1.721332311630249,
      "step": 29244
    },
    {
      "epoch": 2.7889251708984375e-05,
      "step": 29244,
      "training_step_time": 7.428741216659546
    },
    {
      "epoch": 2.7890205383300782e-05,
      "model_forward_time": 1.7136385440826416,
      "step": 29245
    },
    {
      "epoch": 2.7890205383300782e-05,
      "step": 29245,
      "training_step_time": 7.399252414703369
    },
    {
      "epoch": 2.7891159057617186e-05,
      "model_forward_time": 1.7171003818511963,
      "step": 29246
    },
    {
      "epoch": 2.7891159057617186e-05,
      "step": 29246,
      "training_step_time": 7.4814746379852295
    },
    {
      "epoch": 2.7892112731933594e-05,
      "model_forward_time": 1.7997291088104248,
      "step": 29247
    },
    {
      "epoch": 2.7892112731933594e-05,
      "step": 29247,
      "training_step_time": 8.411634683609009
    },
    {
      "epoch": 2.789306640625e-05,
      "model_forward_time": 1.718672752380371,
      "step": 29248
    },
    {
      "epoch": 2.789306640625e-05,
      "step": 29248,
      "training_step_time": 9.404554605484009
    },
    {
      "epoch": 2.7894020080566405e-05,
      "model_forward_time": 1.7159080505371094,
      "step": 29249
    },
    {
      "epoch": 2.7894020080566405e-05,
      "step": 29249,
      "training_step_time": 8.269630193710327
    },
    {
      "epoch": 2.7894973754882812e-05,
      "grad_norm": 0.18794767558574677,
      "learning_rate": 8.476713126982809e-06,
      "loss": 0.0579,
      "step": 29250
    },
    {
      "epoch": 2.7894973754882812e-05,
      "model_forward_time": 2.8484957218170166,
      "step": 29250
    },
    {
      "dynamics_loss_avg": 0.047365473583340645,
      "epoch": 2.7894973754882812e-05,
      "step": 29250
    },
    {
      "action_loss_avg": 0.0022073474712669848,
      "epoch": 2.7894973754882812e-05,
      "step": 29250
    },
    {
      "epoch": 2.7894973754882812e-05,
      "step": 29250,
      "training_step_time": 7.708340883255005
    },
    {
      "epoch": 2.789592742919922e-05,
      "model_forward_time": 1.711735725402832,
      "step": 29251
    },
    {
      "epoch": 2.789592742919922e-05,
      "step": 29251,
      "training_step_time": 7.564810752868652
    },
    {
      "epoch": 2.7896881103515624e-05,
      "model_forward_time": 1.705038070678711,
      "step": 29252
    },
    {
      "epoch": 2.7896881103515624e-05,
      "step": 29252,
      "training_step_time": 7.663717031478882
    },
    {
      "epoch": 2.789783477783203e-05,
      "model_forward_time": 1.7127797603607178,
      "step": 29253
    },
    {
      "epoch": 2.789783477783203e-05,
      "step": 29253,
      "training_step_time": 7.5774641036987305
    },
    {
      "epoch": 2.789878845214844e-05,
      "model_forward_time": 1.7198562622070312,
      "step": 29254
    },
    {
      "epoch": 2.789878845214844e-05,
      "step": 29254,
      "training_step_time": 7.445088863372803
    },
    {
      "epoch": 2.7899742126464843e-05,
      "model_forward_time": 1.7234628200531006,
      "step": 29255
    },
    {
      "epoch": 2.7899742126464843e-05,
      "step": 29255,
      "training_step_time": 7.550403118133545
    },
    {
      "epoch": 2.790069580078125e-05,
      "model_forward_time": 1.8068037033081055,
      "step": 29256
    },
    {
      "epoch": 2.790069580078125e-05,
      "step": 29256,
      "training_step_time": 9.465887308120728
    },
    {
      "epoch": 2.7901649475097658e-05,
      "model_forward_time": 1.7218608856201172,
      "step": 29257
    },
    {
      "epoch": 2.7901649475097658e-05,
      "step": 29257,
      "training_step_time": 7.722050428390503
    },
    {
      "epoch": 2.790260314941406e-05,
      "model_forward_time": 1.711888074874878,
      "step": 29258
    },
    {
      "epoch": 2.790260314941406e-05,
      "step": 29258,
      "training_step_time": 9.845186233520508
    },
    {
      "epoch": 2.790355682373047e-05,
      "model_forward_time": 2.815042495727539,
      "step": 29259
    },
    {
      "epoch": 2.790355682373047e-05,
      "step": 29259,
      "training_step_time": 7.659809827804565
    },
    {
      "epoch": 2.7904510498046876e-05,
      "grad_norm": 0.13708116114139557,
      "learning_rate": 8.475524623868112e-06,
      "loss": 0.0608,
      "step": 29260
    },
    {
      "epoch": 2.7904510498046876e-05,
      "model_forward_time": 1.7209186553955078,
      "step": 29260
    },
    {
      "dynamics_loss_avg": 0.04674726538360119,
      "epoch": 2.7904510498046876e-05,
      "step": 29260
    },
    {
      "action_loss_avg": 0.002445676166098565,
      "epoch": 2.7904510498046876e-05,
      "step": 29260
    },
    {
      "epoch": 2.7904510498046876e-05,
      "step": 29260,
      "training_step_time": 7.647268533706665
    },
    {
      "epoch": 2.790546417236328e-05,
      "model_forward_time": 1.7078404426574707,
      "step": 29261
    },
    {
      "epoch": 2.790546417236328e-05,
      "step": 29261,
      "training_step_time": 7.695338010787964
    },
    {
      "epoch": 2.7906417846679688e-05,
      "model_forward_time": 1.715813159942627,
      "step": 29262
    },
    {
      "epoch": 2.7906417846679688e-05,
      "step": 29262,
      "training_step_time": 7.620753526687622
    },
    {
      "epoch": 2.7907371520996095e-05,
      "model_forward_time": 1.7163453102111816,
      "step": 29263
    },
    {
      "epoch": 2.7907371520996095e-05,
      "step": 29263,
      "training_step_time": 7.528080224990845
    },
    {
      "epoch": 2.79083251953125e-05,
      "model_forward_time": 1.714829444885254,
      "step": 29264
    },
    {
      "epoch": 2.79083251953125e-05,
      "step": 29264,
      "training_step_time": 7.53458309173584
    },
    {
      "epoch": 2.7909278869628907e-05,
      "model_forward_time": 1.7992010116577148,
      "step": 29265
    },
    {
      "epoch": 2.7909278869628907e-05,
      "step": 29265,
      "training_step_time": 7.653483629226685
    },
    {
      "epoch": 2.791023254394531e-05,
      "model_forward_time": 1.719226598739624,
      "step": 29266
    },
    {
      "epoch": 2.791023254394531e-05,
      "step": 29266,
      "training_step_time": 7.741391897201538
    },
    {
      "epoch": 2.7911186218261718e-05,
      "model_forward_time": 1.7158441543579102,
      "step": 29267
    },
    {
      "epoch": 2.7911186218261718e-05,
      "step": 29267,
      "training_step_time": 10.358726024627686
    },
    {
      "epoch": 2.7912139892578126e-05,
      "model_forward_time": 1.7253386974334717,
      "step": 29268
    },
    {
      "epoch": 2.7912139892578126e-05,
      "step": 29268,
      "training_step_time": 8.75330638885498
    },
    {
      "epoch": 2.791309356689453e-05,
      "model_forward_time": 1.7148187160491943,
      "step": 29269
    },
    {
      "epoch": 2.791309356689453e-05,
      "step": 29269,
      "training_step_time": 7.694444179534912
    },
    {
      "epoch": 2.7914047241210937e-05,
      "grad_norm": 0.17577563226222992,
      "learning_rate": 8.474335740675266e-06,
      "loss": 0.0588,
      "step": 29270
    },
    {
      "epoch": 2.7914047241210937e-05,
      "model_forward_time": 1.723351001739502,
      "step": 29270
    },
    {
      "dynamics_loss_avg": 0.06596229802817107,
      "epoch": 2.7914047241210937e-05,
      "step": 29270
    },
    {
      "action_loss_avg": 0.0022289571235887705,
      "epoch": 2.7914047241210937e-05,
      "step": 29270
    },
    {
      "epoch": 2.7914047241210937e-05,
      "step": 29270,
      "training_step_time": 7.6713738441467285
    },
    {
      "epoch": 2.7915000915527345e-05,
      "model_forward_time": 1.7163963317871094,
      "step": 29271
    },
    {
      "epoch": 2.7915000915527345e-05,
      "step": 29271,
      "training_step_time": 7.60054612159729
    },
    {
      "epoch": 2.791595458984375e-05,
      "model_forward_time": 1.7174575328826904,
      "step": 29272
    },
    {
      "epoch": 2.791595458984375e-05,
      "step": 29272,
      "training_step_time": 7.233814477920532
    },
    {
      "epoch": 2.7916908264160156e-05,
      "model_forward_time": 1.718613862991333,
      "step": 29273
    },
    {
      "epoch": 2.7916908264160156e-05,
      "step": 29273,
      "training_step_time": 7.476840257644653
    },
    {
      "epoch": 2.7917861938476563e-05,
      "model_forward_time": 1.7758007049560547,
      "step": 29274
    },
    {
      "epoch": 2.7917861938476563e-05,
      "step": 29274,
      "training_step_time": 8.338240623474121
    },
    {
      "epoch": 2.7918815612792967e-05,
      "model_forward_time": 1.7118523120880127,
      "step": 29275
    },
    {
      "epoch": 2.7918815612792967e-05,
      "step": 29275,
      "training_step_time": 7.681291580200195
    },
    {
      "epoch": 2.7919769287109375e-05,
      "model_forward_time": 1.3597078323364258,
      "step": 29276
    },
    {
      "epoch": 2.7919769287109375e-05,
      "step": 29276,
      "training_step_time": 9.369953155517578
    },
    {
      "epoch": 2.7920722961425782e-05,
      "model_forward_time": 1.7309579849243164,
      "step": 29277
    },
    {
      "epoch": 2.7920722961425782e-05,
      "step": 29277,
      "training_step_time": 8.15166449546814
    },
    {
      "epoch": 2.7921676635742186e-05,
      "model_forward_time": 2.9195756912231445,
      "step": 29278
    },
    {
      "epoch": 2.7921676635742186e-05,
      "step": 29278,
      "training_step_time": 7.651660203933716
    },
    {
      "epoch": 2.7922630310058594e-05,
      "model_forward_time": 1.7292594909667969,
      "step": 29279
    },
    {
      "epoch": 2.7922630310058594e-05,
      "step": 29279,
      "training_step_time": 7.220748424530029
    },
    {
      "epoch": 2.7923583984375e-05,
      "grad_norm": 0.16683341562747955,
      "learning_rate": 8.473146477534288e-06,
      "loss": 0.0592,
      "step": 29280
    },
    {
      "epoch": 2.7923583984375e-05,
      "model_forward_time": 1.7343096733093262,
      "step": 29280
    },
    {
      "dynamics_loss_avg": 0.0627482570707798,
      "epoch": 2.7923583984375e-05,
      "step": 29280
    },
    {
      "action_loss_avg": 0.002272293868009001,
      "epoch": 2.7923583984375e-05,
      "step": 29280
    },
    {
      "epoch": 2.7923583984375e-05,
      "step": 29280,
      "training_step_time": 7.587771654129028
    },
    {
      "epoch": 2.7924537658691405e-05,
      "model_forward_time": 1.3537564277648926,
      "step": 29281
    },
    {
      "epoch": 2.7924537658691405e-05,
      "step": 29281,
      "training_step_time": 7.155413627624512
    },
    {
      "epoch": 2.7925491333007813e-05,
      "model_forward_time": 1.7283246517181396,
      "step": 29282
    },
    {
      "epoch": 2.7925491333007813e-05,
      "step": 29282,
      "training_step_time": 7.42784309387207
    },
    {
      "epoch": 2.792644500732422e-05,
      "model_forward_time": 1.7663989067077637,
      "step": 29283
    },
    {
      "epoch": 2.792644500732422e-05,
      "step": 29283,
      "training_step_time": 9.701459169387817
    },
    {
      "epoch": 2.7927398681640624e-05,
      "model_forward_time": 1.7349228858947754,
      "step": 29284
    },
    {
      "epoch": 2.7927398681640624e-05,
      "step": 29284,
      "training_step_time": 7.6166768074035645
    },
    {
      "epoch": 2.792835235595703e-05,
      "model_forward_time": 1.7400562763214111,
      "step": 29285
    },
    {
      "epoch": 2.792835235595703e-05,
      "step": 29285,
      "training_step_time": 8.283049821853638
    },
    {
      "epoch": 2.792930603027344e-05,
      "model_forward_time": 1.7166447639465332,
      "step": 29286
    },
    {
      "epoch": 2.792930603027344e-05,
      "step": 29286,
      "training_step_time": 8.677417516708374
    },
    {
      "epoch": 2.7930259704589843e-05,
      "model_forward_time": 2.868165969848633,
      "step": 29287
    },
    {
      "epoch": 2.7930259704589843e-05,
      "step": 29287,
      "training_step_time": 7.676124572753906
    },
    {
      "epoch": 2.793121337890625e-05,
      "model_forward_time": 1.7360515594482422,
      "step": 29288
    },
    {
      "epoch": 2.793121337890625e-05,
      "step": 29288,
      "training_step_time": 7.239905595779419
    },
    {
      "epoch": 2.7932167053222658e-05,
      "model_forward_time": 1.731872797012329,
      "step": 29289
    },
    {
      "epoch": 2.7932167053222658e-05,
      "step": 29289,
      "training_step_time": 7.696284294128418
    },
    {
      "epoch": 2.793312072753906e-05,
      "grad_norm": 0.18358591198921204,
      "learning_rate": 8.471956834575233e-06,
      "loss": 0.0594,
      "step": 29290
    },
    {
      "epoch": 2.793312072753906e-05,
      "model_forward_time": 1.725616455078125,
      "step": 29290
    },
    {
      "dynamics_loss_avg": 0.06672687381505966,
      "epoch": 2.793312072753906e-05,
      "step": 29290
    },
    {
      "action_loss_avg": 0.0032413989654742183,
      "epoch": 2.793312072753906e-05,
      "step": 29290
    },
    {
      "epoch": 2.793312072753906e-05,
      "step": 29290,
      "training_step_time": 7.171261310577393
    },
    {
      "epoch": 2.793407440185547e-05,
      "model_forward_time": 1.737889289855957,
      "step": 29291
    },
    {
      "epoch": 2.793407440185547e-05,
      "step": 29291,
      "training_step_time": 7.452035665512085
    },
    {
      "epoch": 2.7935028076171877e-05,
      "model_forward_time": 1.734543800354004,
      "step": 29292
    },
    {
      "epoch": 2.7935028076171877e-05,
      "step": 29292,
      "training_step_time": 7.543483018875122
    },
    {
      "epoch": 2.793598175048828e-05,
      "model_forward_time": 1.7600626945495605,
      "step": 29293
    },
    {
      "epoch": 2.793598175048828e-05,
      "step": 29293,
      "training_step_time": 7.608613967895508
    },
    {
      "epoch": 2.7936935424804688e-05,
      "model_forward_time": 1.3608105182647705,
      "step": 29294
    },
    {
      "epoch": 2.7936935424804688e-05,
      "step": 29294,
      "training_step_time": 9.235124111175537
    },
    {
      "epoch": 2.7937889099121095e-05,
      "model_forward_time": 1.7257664203643799,
      "step": 29295
    },
    {
      "epoch": 2.7937889099121095e-05,
      "step": 29295,
      "training_step_time": 9.752368927001953
    },
    {
      "epoch": 2.79388427734375e-05,
      "model_forward_time": 2.801500082015991,
      "step": 29296
    },
    {
      "epoch": 2.79388427734375e-05,
      "step": 29296,
      "training_step_time": 8.628494262695312
    },
    {
      "epoch": 2.7939796447753907e-05,
      "model_forward_time": 1.723560094833374,
      "step": 29297
    },
    {
      "epoch": 2.7939796447753907e-05,
      "step": 29297,
      "training_step_time": 7.239164352416992
    },
    {
      "epoch": 2.794075012207031e-05,
      "model_forward_time": 1.7284414768218994,
      "step": 29298
    },
    {
      "epoch": 2.794075012207031e-05,
      "step": 29298,
      "training_step_time": 7.621535301208496
    },
    {
      "epoch": 2.7941703796386718e-05,
      "model_forward_time": 1.735579013824463,
      "step": 29299
    },
    {
      "epoch": 2.7941703796386718e-05,
      "step": 29299,
      "training_step_time": 6.720181465148926
    },
    {
      "epoch": 2.7942657470703126e-05,
      "grad_norm": 0.21113450825214386,
      "learning_rate": 8.470766811928198e-06,
      "loss": 0.0583,
      "step": 29300
    },
    {
      "epoch": 2.7942657470703126e-05,
      "model_forward_time": 1.7261154651641846,
      "step": 29300
    },
    {
      "dynamics_loss_avg": 0.0544411214068532,
      "epoch": 2.7942657470703126e-05,
      "step": 29300
    },
    {
      "action_loss_avg": 0.0025103876600041986,
      "epoch": 2.7942657470703126e-05,
      "step": 29300
    },
    {
      "epoch": 2.7942657470703126e-05,
      "step": 29300,
      "training_step_time": 7.477720260620117
    },
    {
      "epoch": 2.794361114501953e-05,
      "model_forward_time": 1.752044916152954,
      "step": 29301
    },
    {
      "epoch": 2.794361114501953e-05,
      "step": 29301,
      "training_step_time": 8.583321332931519
    },
    {
      "epoch": 2.7944564819335937e-05,
      "model_forward_time": 1.7487893104553223,
      "step": 29302
    },
    {
      "epoch": 2.7944564819335937e-05,
      "step": 29302,
      "training_step_time": 7.5814738273620605
    },
    {
      "epoch": 2.7945518493652345e-05,
      "model_forward_time": 1.7342350482940674,
      "step": 29303
    },
    {
      "epoch": 2.7945518493652345e-05,
      "step": 29303,
      "training_step_time": 8.22339391708374
    },
    {
      "epoch": 2.794647216796875e-05,
      "model_forward_time": 1.7357702255249023,
      "step": 29304
    },
    {
      "epoch": 2.794647216796875e-05,
      "step": 29304,
      "training_step_time": 9.271020412445068
    },
    {
      "epoch": 2.7947425842285156e-05,
      "model_forward_time": 1.7269797325134277,
      "step": 29305
    },
    {
      "epoch": 2.7947425842285156e-05,
      "step": 29305,
      "training_step_time": 9.134442567825317
    },
    {
      "epoch": 2.7948379516601563e-05,
      "model_forward_time": 2.924546480178833,
      "step": 29306
    },
    {
      "epoch": 2.7948379516601563e-05,
      "step": 29306,
      "training_step_time": 7.678441762924194
    },
    {
      "epoch": 2.7949333190917967e-05,
      "model_forward_time": 1.731020450592041,
      "step": 29307
    },
    {
      "epoch": 2.7949333190917967e-05,
      "step": 29307,
      "training_step_time": 8.029422044754028
    },
    {
      "epoch": 2.7950286865234375e-05,
      "model_forward_time": 1.7337703704833984,
      "step": 29308
    },
    {
      "epoch": 2.7950286865234375e-05,
      "step": 29308,
      "training_step_time": 6.687750816345215
    },
    {
      "epoch": 2.7951240539550782e-05,
      "model_forward_time": 1.7317090034484863,
      "step": 29309
    },
    {
      "epoch": 2.7951240539550782e-05,
      "step": 29309,
      "training_step_time": 8.342559337615967
    },
    {
      "epoch": 2.7952194213867186e-05,
      "grad_norm": 0.21436861157417297,
      "learning_rate": 8.469576409723322e-06,
      "loss": 0.0569,
      "step": 29310
    },
    {
      "epoch": 2.7952194213867186e-05,
      "model_forward_time": 1.721825361251831,
      "step": 29310
    },
    {
      "dynamics_loss_avg": 0.04886166397482157,
      "epoch": 2.7952194213867186e-05,
      "step": 29310
    },
    {
      "action_loss_avg": 0.00211033271625638,
      "epoch": 2.7952194213867186e-05,
      "step": 29310
    },
    {
      "epoch": 2.7952194213867186e-05,
      "step": 29310,
      "training_step_time": 7.550038814544678
    },
    {
      "epoch": 2.7953147888183594e-05,
      "model_forward_time": 1.728771686553955,
      "step": 29311
    },
    {
      "epoch": 2.7953147888183594e-05,
      "step": 29311,
      "training_step_time": 8.335627555847168
    },
    {
      "epoch": 2.79541015625e-05,
      "model_forward_time": 1.7356972694396973,
      "step": 29312
    },
    {
      "epoch": 2.79541015625e-05,
      "step": 29312,
      "training_step_time": 7.652388572692871
    },
    {
      "epoch": 2.7955055236816405e-05,
      "model_forward_time": 1.737968921661377,
      "step": 29313
    },
    {
      "epoch": 2.7955055236816405e-05,
      "step": 29313,
      "training_step_time": 7.675187826156616
    },
    {
      "epoch": 2.7956008911132813e-05,
      "model_forward_time": 1.7346000671386719,
      "step": 29314
    },
    {
      "epoch": 2.7956008911132813e-05,
      "step": 29314,
      "training_step_time": 7.649227142333984
    },
    {
      "epoch": 2.795696258544922e-05,
      "model_forward_time": 2.898298740386963,
      "step": 29315
    },
    {
      "epoch": 2.795696258544922e-05,
      "step": 29315,
      "training_step_time": 7.614569187164307
    },
    {
      "epoch": 2.7957916259765624e-05,
      "model_forward_time": 1.7305374145507812,
      "step": 29316
    },
    {
      "epoch": 2.7957916259765624e-05,
      "step": 29316,
      "training_step_time": 7.997729301452637
    },
    {
      "epoch": 2.795886993408203e-05,
      "model_forward_time": 1.7268354892730713,
      "step": 29317
    },
    {
      "epoch": 2.795886993408203e-05,
      "step": 29317,
      "training_step_time": 6.654655933380127
    },
    {
      "epoch": 2.795982360839844e-05,
      "model_forward_time": 1.7358803749084473,
      "step": 29318
    },
    {
      "epoch": 2.795982360839844e-05,
      "step": 29318,
      "training_step_time": 7.334798574447632
    },
    {
      "epoch": 2.7960777282714843e-05,
      "model_forward_time": 1.7334918975830078,
      "step": 29319
    },
    {
      "epoch": 2.7960777282714843e-05,
      "step": 29319,
      "training_step_time": 8.594161987304688
    },
    {
      "epoch": 2.796173095703125e-05,
      "grad_norm": 0.2566193640232086,
      "learning_rate": 8.468385628090788e-06,
      "loss": 0.0615,
      "step": 29320
    },
    {
      "epoch": 2.796173095703125e-05,
      "model_forward_time": 1.73018217086792,
      "step": 29320
    },
    {
      "dynamics_loss_avg": 0.057760967127978805,
      "epoch": 2.796173095703125e-05,
      "step": 29320
    },
    {
      "action_loss_avg": 0.002706093346932903,
      "epoch": 2.796173095703125e-05,
      "step": 29320
    },
    {
      "epoch": 2.796173095703125e-05,
      "step": 29320,
      "training_step_time": 8.525830745697021
    },
    {
      "epoch": 2.7962684631347658e-05,
      "model_forward_time": 1.7614924907684326,
      "step": 29321
    },
    {
      "epoch": 2.7962684631347658e-05,
      "step": 29321,
      "training_step_time": 7.683285236358643
    },
    {
      "epoch": 2.7963638305664062e-05,
      "model_forward_time": 1.735961675643921,
      "step": 29322
    },
    {
      "epoch": 2.7963638305664062e-05,
      "step": 29322,
      "training_step_time": 8.728003978729248
    },
    {
      "epoch": 2.796459197998047e-05,
      "model_forward_time": 1.7298176288604736,
      "step": 29323
    },
    {
      "epoch": 2.796459197998047e-05,
      "step": 29323,
      "training_step_time": 7.642760753631592
    },
    {
      "epoch": 2.7965545654296877e-05,
      "model_forward_time": 2.9105920791625977,
      "step": 29324
    },
    {
      "epoch": 2.7965545654296877e-05,
      "step": 29324,
      "training_step_time": 8.220590591430664
    },
    {
      "epoch": 2.796649932861328e-05,
      "model_forward_time": 1.735574722290039,
      "step": 29325
    },
    {
      "epoch": 2.796649932861328e-05,
      "step": 29325,
      "training_step_time": 7.457401990890503
    },
    {
      "epoch": 2.7967453002929688e-05,
      "model_forward_time": 1.7360715866088867,
      "step": 29326
    },
    {
      "epoch": 2.7967453002929688e-05,
      "step": 29326,
      "training_step_time": 7.5693724155426025
    },
    {
      "epoch": 2.7968406677246095e-05,
      "model_forward_time": 1.7352893352508545,
      "step": 29327
    },
    {
      "epoch": 2.7968406677246095e-05,
      "step": 29327,
      "training_step_time": 7.336971044540405
    },
    {
      "epoch": 2.79693603515625e-05,
      "model_forward_time": 1.7352244853973389,
      "step": 29328
    },
    {
      "epoch": 2.79693603515625e-05,
      "step": 29328,
      "training_step_time": 7.479536771774292
    },
    {
      "epoch": 2.7970314025878907e-05,
      "model_forward_time": 1.7652199268341064,
      "step": 29329
    },
    {
      "epoch": 2.7970314025878907e-05,
      "step": 29329,
      "training_step_time": 8.22994875907898
    },
    {
      "epoch": 2.797126770019531e-05,
      "grad_norm": 0.20829425752162933,
      "learning_rate": 8.467194467160815e-06,
      "loss": 0.0627,
      "step": 29330
    },
    {
      "epoch": 2.797126770019531e-05,
      "model_forward_time": 1.747495174407959,
      "step": 29330
    },
    {
      "dynamics_loss_avg": 0.05110194440931082,
      "epoch": 2.797126770019531e-05,
      "step": 29330
    },
    {
      "action_loss_avg": 0.0015090553468326106,
      "epoch": 2.797126770019531e-05,
      "step": 29330
    },
    {
      "epoch": 2.797126770019531e-05,
      "step": 29330,
      "training_step_time": 8.807339191436768
    },
    {
      "epoch": 2.7972221374511718e-05,
      "model_forward_time": 1.732706069946289,
      "step": 29331
    },
    {
      "epoch": 2.7972221374511718e-05,
      "step": 29331,
      "training_step_time": 8.194377899169922
    },
    {
      "epoch": 2.7973175048828126e-05,
      "model_forward_time": 1.7373430728912354,
      "step": 29332
    },
    {
      "epoch": 2.7973175048828126e-05,
      "step": 29332,
      "training_step_time": 10.748126745223999
    },
    {
      "epoch": 2.797412872314453e-05,
      "model_forward_time": 2.891472578048706,
      "step": 29333
    },
    {
      "epoch": 2.797412872314453e-05,
      "step": 29333,
      "training_step_time": 8.656414270401001
    },
    {
      "epoch": 2.7975082397460937e-05,
      "model_forward_time": 1.7359158992767334,
      "step": 29334
    },
    {
      "epoch": 2.7975082397460937e-05,
      "step": 29334,
      "training_step_time": 7.469491243362427
    },
    {
      "epoch": 2.7976036071777345e-05,
      "model_forward_time": 1.3543715476989746,
      "step": 29335
    },
    {
      "epoch": 2.7976036071777345e-05,
      "step": 29335,
      "training_step_time": 7.5852391719818115
    },
    {
      "epoch": 2.797698974609375e-05,
      "model_forward_time": 1.7306389808654785,
      "step": 29336
    },
    {
      "epoch": 2.797698974609375e-05,
      "step": 29336,
      "training_step_time": 7.308046340942383
    },
    {
      "epoch": 2.7977943420410156e-05,
      "model_forward_time": 1.7319352626800537,
      "step": 29337
    },
    {
      "epoch": 2.7977943420410156e-05,
      "step": 29337,
      "training_step_time": 7.482421159744263
    },
    {
      "epoch": 2.7978897094726563e-05,
      "model_forward_time": 1.7572886943817139,
      "step": 29338
    },
    {
      "epoch": 2.7978897094726563e-05,
      "step": 29338,
      "training_step_time": 7.705791473388672
    },
    {
      "epoch": 2.7979850769042967e-05,
      "model_forward_time": 1.7600297927856445,
      "step": 29339
    },
    {
      "epoch": 2.7979850769042967e-05,
      "step": 29339,
      "training_step_time": 7.750561952590942
    },
    {
      "epoch": 2.7980804443359375e-05,
      "grad_norm": 0.25077196955680847,
      "learning_rate": 8.466002927063668e-06,
      "loss": 0.0589,
      "step": 29340
    },
    {
      "epoch": 2.7980804443359375e-05,
      "model_forward_time": 1.728261947631836,
      "step": 29340
    },
    {
      "dynamics_loss_avg": 0.0624468881636858,
      "epoch": 2.7980804443359375e-05,
      "step": 29340
    },
    {
      "action_loss_avg": 0.0020705276110675186,
      "epoch": 2.7980804443359375e-05,
      "step": 29340
    },
    {
      "epoch": 2.7980804443359375e-05,
      "step": 29340,
      "training_step_time": 9.377628564834595
    },
    {
      "epoch": 2.7981758117675782e-05,
      "model_forward_time": 1.7298469543457031,
      "step": 29341
    },
    {
      "epoch": 2.7981758117675782e-05,
      "step": 29341,
      "training_step_time": 7.619793176651001
    },
    {
      "epoch": 2.7982711791992186e-05,
      "model_forward_time": 2.817253828048706,
      "step": 29342
    },
    {
      "epoch": 2.7982711791992186e-05,
      "step": 29342,
      "training_step_time": 7.647095203399658
    },
    {
      "epoch": 2.7983665466308594e-05,
      "model_forward_time": 1.7253212928771973,
      "step": 29343
    },
    {
      "epoch": 2.7983665466308594e-05,
      "step": 29343,
      "training_step_time": 8.168667078018188
    },
    {
      "epoch": 2.7984619140625e-05,
      "model_forward_time": 1.7386407852172852,
      "step": 29344
    },
    {
      "epoch": 2.7984619140625e-05,
      "step": 29344,
      "training_step_time": 7.710721015930176
    },
    {
      "epoch": 2.7985572814941405e-05,
      "model_forward_time": 1.7333145141601562,
      "step": 29345
    },
    {
      "epoch": 2.7985572814941405e-05,
      "step": 29345,
      "training_step_time": 7.123133182525635
    },
    {
      "epoch": 2.7986526489257813e-05,
      "model_forward_time": 1.7270176410675049,
      "step": 29346
    },
    {
      "epoch": 2.7986526489257813e-05,
      "step": 29346,
      "training_step_time": 7.349820375442505
    },
    {
      "epoch": 2.798748016357422e-05,
      "model_forward_time": 1.733799934387207,
      "step": 29347
    },
    {
      "epoch": 2.798748016357422e-05,
      "step": 29347,
      "training_step_time": 7.53492283821106
    },
    {
      "epoch": 2.7988433837890624e-05,
      "model_forward_time": 1.7366313934326172,
      "step": 29348
    },
    {
      "epoch": 2.7988433837890624e-05,
      "step": 29348,
      "training_step_time": 9.115008115768433
    },
    {
      "epoch": 2.798938751220703e-05,
      "model_forward_time": 1.7223172187805176,
      "step": 29349
    },
    {
      "epoch": 2.798938751220703e-05,
      "step": 29349,
      "training_step_time": 9.127767562866211
    },
    {
      "epoch": 2.799034118652344e-05,
      "grad_norm": 0.2695276737213135,
      "learning_rate": 8.464811007929652e-06,
      "loss": 0.0608,
      "step": 29350
    },
    {
      "epoch": 2.799034118652344e-05,
      "model_forward_time": 1.7162256240844727,
      "step": 29350
    },
    {
      "dynamics_loss_avg": 0.04027692135423422,
      "epoch": 2.799034118652344e-05,
      "step": 29350
    },
    {
      "action_loss_avg": 0.0017651188012678177,
      "epoch": 2.799034118652344e-05,
      "step": 29350
    },
    {
      "epoch": 2.799034118652344e-05,
      "step": 29350,
      "training_step_time": 7.650105714797974
    },
    {
      "epoch": 2.7991294860839843e-05,
      "model_forward_time": 1.7149319648742676,
      "step": 29351
    },
    {
      "epoch": 2.7991294860839843e-05,
      "step": 29351,
      "training_step_time": 7.678756237030029
    },
    {
      "epoch": 2.799224853515625e-05,
      "model_forward_time": 2.903547763824463,
      "step": 29352
    },
    {
      "epoch": 2.799224853515625e-05,
      "step": 29352,
      "training_step_time": 7.69464111328125
    },
    {
      "epoch": 2.7993202209472658e-05,
      "model_forward_time": 1.7165586948394775,
      "step": 29353
    },
    {
      "epoch": 2.7993202209472658e-05,
      "step": 29353,
      "training_step_time": 7.689998626708984
    },
    {
      "epoch": 2.7994155883789062e-05,
      "model_forward_time": 1.7201104164123535,
      "step": 29354
    },
    {
      "epoch": 2.7994155883789062e-05,
      "step": 29354,
      "training_step_time": 7.099688529968262
    },
    {
      "epoch": 2.799510955810547e-05,
      "model_forward_time": 1.7170994281768799,
      "step": 29355
    },
    {
      "epoch": 2.799510955810547e-05,
      "step": 29355,
      "training_step_time": 8.235952854156494
    },
    {
      "epoch": 2.7996063232421877e-05,
      "model_forward_time": 1.7378885746002197,
      "step": 29356
    },
    {
      "epoch": 2.7996063232421877e-05,
      "step": 29356,
      "training_step_time": 8.527176856994629
    },
    {
      "epoch": 2.799701690673828e-05,
      "model_forward_time": 1.3801283836364746,
      "step": 29357
    },
    {
      "epoch": 2.799701690673828e-05,
      "step": 29357,
      "training_step_time": 8.702828407287598
    },
    {
      "epoch": 2.7997970581054688e-05,
      "model_forward_time": 1.7119526863098145,
      "step": 29358
    },
    {
      "epoch": 2.7997970581054688e-05,
      "step": 29358,
      "training_step_time": 7.6641013622283936
    },
    {
      "epoch": 2.7998924255371095e-05,
      "model_forward_time": 1.7259023189544678,
      "step": 29359
    },
    {
      "epoch": 2.7998924255371095e-05,
      "step": 29359,
      "training_step_time": 7.720487833023071
    },
    {
      "epoch": 2.79998779296875e-05,
      "grad_norm": 0.17949312925338745,
      "learning_rate": 8.463618709889115e-06,
      "loss": 0.0626,
      "step": 29360
    },
    {
      "epoch": 2.79998779296875e-05,
      "model_forward_time": 1.7155802249908447,
      "step": 29360
    },
    {
      "dynamics_loss_avg": 0.048189377039670946,
      "epoch": 2.79998779296875e-05,
      "step": 29360
    },
    {
      "action_loss_avg": 0.0016425690060714259,
      "epoch": 2.79998779296875e-05,
      "step": 29360
    },
    {
      "epoch": 2.79998779296875e-05,
      "step": 29360,
      "training_step_time": 7.727865695953369
    },
    {
      "epoch": 2.8000831604003907e-05,
      "model_forward_time": 2.8142497539520264,
      "step": 29361
    },
    {
      "epoch": 2.8000831604003907e-05,
      "step": 29361,
      "training_step_time": 7.76017951965332
    },
    {
      "epoch": 2.800178527832031e-05,
      "model_forward_time": 1.724238634109497,
      "step": 29362
    },
    {
      "epoch": 2.800178527832031e-05,
      "step": 29362,
      "training_step_time": 7.662700176239014
    },
    {
      "epoch": 2.800273895263672e-05,
      "model_forward_time": 1.7254869937896729,
      "step": 29363
    },
    {
      "epoch": 2.800273895263672e-05,
      "step": 29363,
      "training_step_time": 6.900481224060059
    },
    {
      "epoch": 2.8003692626953126e-05,
      "model_forward_time": 1.7253038883209229,
      "step": 29364
    },
    {
      "epoch": 2.8003692626953126e-05,
      "step": 29364,
      "training_step_time": 7.491281986236572
    },
    {
      "epoch": 2.800464630126953e-05,
      "model_forward_time": 1.7158527374267578,
      "step": 29365
    },
    {
      "epoch": 2.800464630126953e-05,
      "step": 29365,
      "training_step_time": 9.31007981300354
    },
    {
      "epoch": 2.8005599975585937e-05,
      "model_forward_time": 1.7347924709320068,
      "step": 29366
    },
    {
      "epoch": 2.8005599975585937e-05,
      "step": 29366,
      "training_step_time": 7.658547878265381
    },
    {
      "epoch": 2.8006553649902345e-05,
      "model_forward_time": 1.7194855213165283,
      "step": 29367
    },
    {
      "epoch": 2.8006553649902345e-05,
      "step": 29367,
      "training_step_time": 8.678045988082886
    },
    {
      "epoch": 2.800750732421875e-05,
      "model_forward_time": 1.7313225269317627,
      "step": 29368
    },
    {
      "epoch": 2.800750732421875e-05,
      "step": 29368,
      "training_step_time": 9.368457317352295
    },
    {
      "epoch": 2.8008460998535156e-05,
      "model_forward_time": 1.7355015277862549,
      "step": 29369
    },
    {
      "epoch": 2.8008460998535156e-05,
      "step": 29369,
      "training_step_time": 7.659049034118652
    },
    {
      "epoch": 2.8009414672851563e-05,
      "grad_norm": 0.1239028349518776,
      "learning_rate": 8.462426033072442e-06,
      "loss": 0.0583,
      "step": 29370
    },
    {
      "epoch": 2.8009414672851563e-05,
      "model_forward_time": 2.85799241065979,
      "step": 29370
    },
    {
      "dynamics_loss_avg": 0.049328366108238694,
      "epoch": 2.8009414672851563e-05,
      "step": 29370
    },
    {
      "action_loss_avg": 0.0023552555474452676,
      "epoch": 2.8009414672851563e-05,
      "step": 29370
    },
    {
      "epoch": 2.8009414672851563e-05,
      "step": 29370,
      "training_step_time": 7.625316858291626
    },
    {
      "epoch": 2.8010368347167967e-05,
      "model_forward_time": 1.7286479473114014,
      "step": 29371
    },
    {
      "epoch": 2.8010368347167967e-05,
      "step": 29371,
      "training_step_time": 7.511480808258057
    },
    {
      "epoch": 2.8011322021484375e-05,
      "model_forward_time": 1.7421135902404785,
      "step": 29372
    },
    {
      "epoch": 2.8011322021484375e-05,
      "step": 29372,
      "training_step_time": 7.573558807373047
    },
    {
      "epoch": 2.8012275695800782e-05,
      "model_forward_time": 1.025602102279663,
      "step": 29373
    },
    {
      "epoch": 2.8012275695800782e-05,
      "step": 29373,
      "training_step_time": 7.363897085189819
    },
    {
      "epoch": 2.8013229370117186e-05,
      "model_forward_time": 1.733964204788208,
      "step": 29374
    },
    {
      "epoch": 2.8013229370117186e-05,
      "step": 29374,
      "training_step_time": 8.526237726211548
    },
    {
      "epoch": 2.8014183044433594e-05,
      "model_forward_time": 1.81795334815979,
      "step": 29375
    },
    {
      "epoch": 2.8014183044433594e-05,
      "step": 29375,
      "training_step_time": 7.58492636680603
    },
    {
      "epoch": 2.801513671875e-05,
      "model_forward_time": 1.7127659320831299,
      "step": 29376
    },
    {
      "epoch": 2.801513671875e-05,
      "step": 29376,
      "training_step_time": 8.232290029525757
    },
    {
      "epoch": 2.8016090393066405e-05,
      "model_forward_time": 1.724668264389038,
      "step": 29377
    },
    {
      "epoch": 2.8016090393066405e-05,
      "step": 29377,
      "training_step_time": 7.7267749309539795
    },
    {
      "epoch": 2.8017044067382813e-05,
      "model_forward_time": 1.7155530452728271,
      "step": 29378
    },
    {
      "epoch": 2.8017044067382813e-05,
      "step": 29378,
      "training_step_time": 8.216477155685425
    },
    {
      "epoch": 2.801799774169922e-05,
      "model_forward_time": 1.7171554565429688,
      "step": 29379
    },
    {
      "epoch": 2.801799774169922e-05,
      "step": 29379,
      "training_step_time": 8.186382532119751
    },
    {
      "epoch": 2.8018951416015624e-05,
      "grad_norm": 0.1856945902109146,
      "learning_rate": 8.461232977610063e-06,
      "loss": 0.0589,
      "step": 29380
    },
    {
      "epoch": 2.8018951416015624e-05,
      "model_forward_time": 2.877936601638794,
      "step": 29380
    },
    {
      "dynamics_loss_avg": 0.0739217009395361,
      "epoch": 2.8018951416015624e-05,
      "step": 29380
    },
    {
      "action_loss_avg": 0.0025839458918198942,
      "epoch": 2.8018951416015624e-05,
      "step": 29380
    },
    {
      "epoch": 2.8018951416015624e-05,
      "step": 29380,
      "training_step_time": 7.655871152877808
    },
    {
      "epoch": 2.801990509033203e-05,
      "model_forward_time": 1.717836618423462,
      "step": 29381
    },
    {
      "epoch": 2.801990509033203e-05,
      "step": 29381,
      "training_step_time": 7.689060926437378
    },
    {
      "epoch": 2.802085876464844e-05,
      "model_forward_time": 1.7235572338104248,
      "step": 29382
    },
    {
      "epoch": 2.802085876464844e-05,
      "step": 29382,
      "training_step_time": 8.196408748626709
    },
    {
      "epoch": 2.8021812438964843e-05,
      "model_forward_time": 1.7079224586486816,
      "step": 29383
    },
    {
      "epoch": 2.8021812438964843e-05,
      "step": 29383,
      "training_step_time": 7.526630401611328
    },
    {
      "epoch": 2.802276611328125e-05,
      "model_forward_time": 1.7867937088012695,
      "step": 29384
    },
    {
      "epoch": 2.802276611328125e-05,
      "step": 29384,
      "training_step_time": 7.688719749450684
    },
    {
      "epoch": 2.8023719787597658e-05,
      "model_forward_time": 1.7134463787078857,
      "step": 29385
    },
    {
      "epoch": 2.8023719787597658e-05,
      "step": 29385,
      "training_step_time": 8.501617193222046
    },
    {
      "epoch": 2.8024673461914062e-05,
      "model_forward_time": 1.71561861038208,
      "step": 29386
    },
    {
      "epoch": 2.8024673461914062e-05,
      "step": 29386,
      "training_step_time": 8.408849954605103
    },
    {
      "epoch": 2.802562713623047e-05,
      "model_forward_time": 1.7232930660247803,
      "step": 29387
    },
    {
      "epoch": 2.802562713623047e-05,
      "step": 29387,
      "training_step_time": 8.425236701965332
    },
    {
      "epoch": 2.8026580810546877e-05,
      "model_forward_time": 1.7317185401916504,
      "step": 29388
    },
    {
      "epoch": 2.8026580810546877e-05,
      "step": 29388,
      "training_step_time": 8.508872032165527
    },
    {
      "epoch": 2.802753448486328e-05,
      "model_forward_time": 2.869720935821533,
      "step": 29389
    },
    {
      "epoch": 2.802753448486328e-05,
      "step": 29389,
      "training_step_time": 9.362470865249634
    },
    {
      "epoch": 2.8028488159179688e-05,
      "grad_norm": 0.15154896676540375,
      "learning_rate": 8.46003954363245e-06,
      "loss": 0.0599,
      "step": 29390
    },
    {
      "epoch": 2.8028488159179688e-05,
      "model_forward_time": 1.7350363731384277,
      "step": 29390
    },
    {
      "dynamics_loss_avg": 0.044798655062913896,
      "epoch": 2.8028488159179688e-05,
      "step": 29390
    },
    {
      "action_loss_avg": 0.001785698306048289,
      "epoch": 2.8028488159179688e-05,
      "step": 29390
    },
    {
      "epoch": 2.8028488159179688e-05,
      "step": 29390,
      "training_step_time": 7.731551170349121
    },
    {
      "epoch": 2.8029441833496092e-05,
      "model_forward_time": 1.733579397201538,
      "step": 29391
    },
    {
      "epoch": 2.8029441833496092e-05,
      "step": 29391,
      "training_step_time": 7.374114513397217
    },
    {
      "epoch": 2.80303955078125e-05,
      "model_forward_time": 1.7355437278747559,
      "step": 29392
    },
    {
      "epoch": 2.80303955078125e-05,
      "step": 29392,
      "training_step_time": 8.523512363433838
    },
    {
      "epoch": 2.8031349182128907e-05,
      "model_forward_time": 1.7703578472137451,
      "step": 29393
    },
    {
      "epoch": 2.8031349182128907e-05,
      "step": 29393,
      "training_step_time": 7.695277214050293
    },
    {
      "epoch": 2.803230285644531e-05,
      "model_forward_time": 1.7316319942474365,
      "step": 29394
    },
    {
      "epoch": 2.803230285644531e-05,
      "step": 29394,
      "training_step_time": 8.546578168869019
    },
    {
      "epoch": 2.803325653076172e-05,
      "model_forward_time": 1.7363102436065674,
      "step": 29395
    },
    {
      "epoch": 2.803325653076172e-05,
      "step": 29395,
      "training_step_time": 8.941459894180298
    },
    {
      "epoch": 2.8034210205078126e-05,
      "model_forward_time": 1.7303757667541504,
      "step": 29396
    },
    {
      "epoch": 2.8034210205078126e-05,
      "step": 29396,
      "training_step_time": 8.932247400283813
    },
    {
      "epoch": 2.803516387939453e-05,
      "model_forward_time": 1.7320241928100586,
      "step": 29397
    },
    {
      "epoch": 2.803516387939453e-05,
      "step": 29397,
      "training_step_time": 8.231974601745605
    },
    {
      "epoch": 2.8036117553710937e-05,
      "model_forward_time": 2.860684633255005,
      "step": 29398
    },
    {
      "epoch": 2.8036117553710937e-05,
      "step": 29398,
      "training_step_time": 7.63071346282959
    },
    {
      "epoch": 2.8037071228027345e-05,
      "model_forward_time": 1.7256786823272705,
      "step": 29399
    },
    {
      "epoch": 2.8037071228027345e-05,
      "step": 29399,
      "training_step_time": 7.666430473327637
    },
    {
      "epoch": 2.803802490234375e-05,
      "grad_norm": 0.2087671011686325,
      "learning_rate": 8.458845731270116e-06,
      "loss": 0.0578,
      "step": 29400
    },
    {
      "epoch": 2.803802490234375e-05,
      "model_forward_time": 1.7168385982513428,
      "step": 29400
    },
    {
      "dynamics_loss_avg": 0.06152935903519392,
      "epoch": 2.803802490234375e-05,
      "step": 29400
    },
    {
      "action_loss_avg": 0.004182613338343799,
      "epoch": 2.803802490234375e-05,
      "step": 29400
    },
    {
      "epoch": 2.803802490234375e-05,
      "step": 29400,
      "training_step_time": 7.363794326782227
    },
    {
      "epoch": 2.8038978576660156e-05,
      "model_forward_time": 1.7212002277374268,
      "step": 29401
    },
    {
      "epoch": 2.8038978576660156e-05,
      "step": 29401,
      "training_step_time": 7.495685338973999
    },
    {
      "epoch": 2.8039932250976563e-05,
      "model_forward_time": 1.752058506011963,
      "step": 29402
    },
    {
      "epoch": 2.8039932250976563e-05,
      "step": 29402,
      "training_step_time": 8.175400972366333
    },
    {
      "epoch": 2.8040885925292968e-05,
      "model_forward_time": 1.7400104999542236,
      "step": 29403
    },
    {
      "epoch": 2.8040885925292968e-05,
      "step": 29403,
      "training_step_time": 7.647177219390869
    },
    {
      "epoch": 2.8041839599609375e-05,
      "model_forward_time": 1.7303597927093506,
      "step": 29404
    },
    {
      "epoch": 2.8041839599609375e-05,
      "step": 29404,
      "training_step_time": 7.688105583190918
    },
    {
      "epoch": 2.8042793273925782e-05,
      "model_forward_time": 1.739063024520874,
      "step": 29405
    },
    {
      "epoch": 2.8042793273925782e-05,
      "step": 29405,
      "training_step_time": 7.83230996131897
    },
    {
      "epoch": 2.8043746948242186e-05,
      "model_forward_time": 1.7424590587615967,
      "step": 29406
    },
    {
      "epoch": 2.8043746948242186e-05,
      "step": 29406,
      "training_step_time": 8.237752199172974
    },
    {
      "epoch": 2.8044700622558594e-05,
      "model_forward_time": 2.858470916748047,
      "step": 29407
    },
    {
      "epoch": 2.8044700622558594e-05,
      "step": 29407,
      "training_step_time": 9.049092054367065
    },
    {
      "epoch": 2.8045654296875e-05,
      "model_forward_time": 1.7279717922210693,
      "step": 29408
    },
    {
      "epoch": 2.8045654296875e-05,
      "step": 29408,
      "training_step_time": 7.580846071243286
    },
    {
      "epoch": 2.8046607971191405e-05,
      "model_forward_time": 1.7337374687194824,
      "step": 29409
    },
    {
      "epoch": 2.8046607971191405e-05,
      "step": 29409,
      "training_step_time": 7.329481363296509
    },
    {
      "epoch": 2.8047561645507813e-05,
      "grad_norm": 0.20349977910518646,
      "learning_rate": 8.457651540653612e-06,
      "loss": 0.059,
      "step": 29410
    },
    {
      "epoch": 2.8047561645507813e-05,
      "model_forward_time": 1.355987787246704,
      "step": 29410
    },
    {
      "dynamics_loss_avg": 0.07432201839983463,
      "epoch": 2.8047561645507813e-05,
      "step": 29410
    },
    {
      "action_loss_avg": 0.0028337146883131935,
      "epoch": 2.8047561645507813e-05,
      "step": 29410
    },
    {
      "epoch": 2.8047561645507813e-05,
      "step": 29410,
      "training_step_time": 7.545113801956177
    },
    {
      "epoch": 2.804851531982422e-05,
      "model_forward_time": 1.742126703262329,
      "step": 29411
    },
    {
      "epoch": 2.804851531982422e-05,
      "step": 29411,
      "training_step_time": 9.272452116012573
    },
    {
      "epoch": 2.8049468994140624e-05,
      "model_forward_time": 1.7196426391601562,
      "step": 29412
    },
    {
      "epoch": 2.8049468994140624e-05,
      "step": 29412,
      "training_step_time": 7.42665958404541
    },
    {
      "epoch": 2.805042266845703e-05,
      "model_forward_time": 1.7130558490753174,
      "step": 29413
    },
    {
      "epoch": 2.805042266845703e-05,
      "step": 29413,
      "training_step_time": 7.7018883228302
    },
    {
      "epoch": 2.805137634277344e-05,
      "model_forward_time": 1.723174810409546,
      "step": 29414
    },
    {
      "epoch": 2.805137634277344e-05,
      "step": 29414,
      "training_step_time": 9.17943787574768
    },
    {
      "epoch": 2.8052330017089843e-05,
      "model_forward_time": 1.7154240608215332,
      "step": 29415
    },
    {
      "epoch": 2.8052330017089843e-05,
      "step": 29415,
      "training_step_time": 9.49676251411438
    },
    {
      "epoch": 2.805328369140625e-05,
      "model_forward_time": 1.7282662391662598,
      "step": 29416
    },
    {
      "epoch": 2.805328369140625e-05,
      "step": 29416,
      "training_step_time": 8.232394218444824
    },
    {
      "epoch": 2.8054237365722658e-05,
      "model_forward_time": 1.7118537425994873,
      "step": 29417
    },
    {
      "epoch": 2.8054237365722658e-05,
      "step": 29417,
      "training_step_time": 7.592972278594971
    },
    {
      "epoch": 2.8055191040039062e-05,
      "model_forward_time": 1.7199175357818604,
      "step": 29418
    },
    {
      "epoch": 2.8055191040039062e-05,
      "step": 29418,
      "training_step_time": 7.636346101760864
    },
    {
      "epoch": 2.805614471435547e-05,
      "model_forward_time": 1.7401728630065918,
      "step": 29419
    },
    {
      "epoch": 2.805614471435547e-05,
      "step": 29419,
      "training_step_time": 7.496936798095703
    },
    {
      "epoch": 2.8057098388671877e-05,
      "grad_norm": 0.2151399850845337,
      "learning_rate": 8.456456971913533e-06,
      "loss": 0.0551,
      "step": 29420
    },
    {
      "epoch": 2.8057098388671877e-05,
      "model_forward_time": 1.751485824584961,
      "step": 29420
    },
    {
      "dynamics_loss_avg": 0.04047384951263666,
      "epoch": 2.8057098388671877e-05,
      "step": 29420
    },
    {
      "action_loss_avg": 0.0016404446796514094,
      "epoch": 2.8057098388671877e-05,
      "step": 29420
    },
    {
      "epoch": 2.8057098388671877e-05,
      "step": 29420,
      "training_step_time": 7.667252779006958
    },
    {
      "epoch": 2.805805206298828e-05,
      "model_forward_time": 1.7651116847991943,
      "step": 29421
    },
    {
      "epoch": 2.805805206298828e-05,
      "step": 29421,
      "training_step_time": 8.587807416915894
    },
    {
      "epoch": 2.8059005737304688e-05,
      "model_forward_time": 1.729088544845581,
      "step": 29422
    },
    {
      "epoch": 2.8059005737304688e-05,
      "step": 29422,
      "training_step_time": 8.771559238433838
    },
    {
      "epoch": 2.8059959411621092e-05,
      "model_forward_time": 1.7324612140655518,
      "step": 29423
    },
    {
      "epoch": 2.8059959411621092e-05,
      "step": 29423,
      "training_step_time": 7.902727127075195
    },
    {
      "epoch": 2.80609130859375e-05,
      "model_forward_time": 1.7472457885742188,
      "step": 29424
    },
    {
      "epoch": 2.80609130859375e-05,
      "step": 29424,
      "training_step_time": 9.56154203414917
    },
    {
      "epoch": 2.8061866760253907e-05,
      "model_forward_time": 1.7324378490447998,
      "step": 29425
    },
    {
      "epoch": 2.8061866760253907e-05,
      "step": 29425,
      "training_step_time": 8.198251008987427
    },
    {
      "epoch": 2.806282043457031e-05,
      "model_forward_time": 2.8911778926849365,
      "step": 29426
    },
    {
      "epoch": 2.806282043457031e-05,
      "step": 29426,
      "training_step_time": 7.66576886177063
    },
    {
      "epoch": 2.806377410888672e-05,
      "model_forward_time": 1.7367563247680664,
      "step": 29427
    },
    {
      "epoch": 2.806377410888672e-05,
      "step": 29427,
      "training_step_time": 7.65052604675293
    },
    {
      "epoch": 2.8064727783203126e-05,
      "model_forward_time": 1.7284581661224365,
      "step": 29428
    },
    {
      "epoch": 2.8064727783203126e-05,
      "step": 29428,
      "training_step_time": 7.528720855712891
    },
    {
      "epoch": 2.806568145751953e-05,
      "model_forward_time": 1.7512485980987549,
      "step": 29429
    },
    {
      "epoch": 2.806568145751953e-05,
      "step": 29429,
      "training_step_time": 7.679804563522339
    },
    {
      "epoch": 2.8066635131835937e-05,
      "grad_norm": 0.17727340757846832,
      "learning_rate": 8.455262025180517e-06,
      "loss": 0.0581,
      "step": 29430
    },
    {
      "epoch": 2.8066635131835937e-05,
      "model_forward_time": 1.787459373474121,
      "step": 29430
    },
    {
      "dynamics_loss_avg": 0.057096042670309544,
      "epoch": 2.8066635131835937e-05,
      "step": 29430
    },
    {
      "action_loss_avg": 0.0021186536294408144,
      "epoch": 2.8066635131835937e-05,
      "step": 29430
    },
    {
      "epoch": 2.8066635131835937e-05,
      "step": 29430,
      "training_step_time": 8.478500127792358
    },
    {
      "epoch": 2.8067588806152345e-05,
      "model_forward_time": 1.7347393035888672,
      "step": 29431
    },
    {
      "epoch": 2.8067588806152345e-05,
      "step": 29431,
      "training_step_time": 9.185543775558472
    },
    {
      "epoch": 2.806854248046875e-05,
      "model_forward_time": 1.7311058044433594,
      "step": 29432
    },
    {
      "epoch": 2.806854248046875e-05,
      "step": 29432,
      "training_step_time": 11.132330894470215
    },
    {
      "epoch": 2.8069496154785156e-05,
      "model_forward_time": 1.701425313949585,
      "step": 29433
    },
    {
      "epoch": 2.8069496154785156e-05,
      "step": 29433,
      "training_step_time": 7.807873010635376
    },
    {
      "epoch": 2.8070449829101564e-05,
      "model_forward_time": 1.723175287246704,
      "step": 29434
    },
    {
      "epoch": 2.8070449829101564e-05,
      "step": 29434,
      "training_step_time": 8.242806196212769
    },
    {
      "epoch": 2.8071403503417968e-05,
      "model_forward_time": 2.8536393642425537,
      "step": 29435
    },
    {
      "epoch": 2.8071403503417968e-05,
      "step": 29435,
      "training_step_time": 8.489997386932373
    },
    {
      "epoch": 2.8072357177734375e-05,
      "model_forward_time": 1.7211813926696777,
      "step": 29436
    },
    {
      "epoch": 2.8072357177734375e-05,
      "step": 29436,
      "training_step_time": 7.746954679489136
    },
    {
      "epoch": 2.8073310852050782e-05,
      "model_forward_time": 1.7152409553527832,
      "step": 29437
    },
    {
      "epoch": 2.8073310852050782e-05,
      "step": 29437,
      "training_step_time": 7.558534145355225
    },
    {
      "epoch": 2.8074264526367186e-05,
      "model_forward_time": 1.7351760864257812,
      "step": 29438
    },
    {
      "epoch": 2.8074264526367186e-05,
      "step": 29438,
      "training_step_time": 9.255812406539917
    },
    {
      "epoch": 2.8075218200683594e-05,
      "model_forward_time": 1.7373788356781006,
      "step": 29439
    },
    {
      "epoch": 2.8075218200683594e-05,
      "step": 29439,
      "training_step_time": 7.546972990036011
    },
    {
      "epoch": 2.8076171875e-05,
      "grad_norm": 0.09809967875480652,
      "learning_rate": 8.454066700585241e-06,
      "loss": 0.0572,
      "step": 29440
    },
    {
      "epoch": 2.8076171875e-05,
      "model_forward_time": 1.7294166088104248,
      "step": 29440
    },
    {
      "dynamics_loss_avg": 0.0643046822398901,
      "epoch": 2.8076171875e-05,
      "step": 29440
    },
    {
      "action_loss_avg": 0.0029963236302137375,
      "epoch": 2.8076171875e-05,
      "step": 29440
    },
    {
      "epoch": 2.8076171875e-05,
      "step": 29440,
      "training_step_time": 8.376005172729492
    },
    {
      "epoch": 2.8077125549316405e-05,
      "model_forward_time": 1.7185840606689453,
      "step": 29441
    },
    {
      "epoch": 2.8077125549316405e-05,
      "step": 29441,
      "training_step_time": 7.778034925460815
    },
    {
      "epoch": 2.8078079223632813e-05,
      "model_forward_time": 1.719764232635498,
      "step": 29442
    },
    {
      "epoch": 2.8078079223632813e-05,
      "step": 29442,
      "training_step_time": 8.879763841629028
    },
    {
      "epoch": 2.807903289794922e-05,
      "model_forward_time": 1.720094919204712,
      "step": 29443
    },
    {
      "epoch": 2.807903289794922e-05,
      "step": 29443,
      "training_step_time": 8.971806526184082
    },
    {
      "epoch": 2.8079986572265624e-05,
      "model_forward_time": 2.923692464828491,
      "step": 29444
    },
    {
      "epoch": 2.8079986572265624e-05,
      "step": 29444,
      "training_step_time": 7.681371450424194
    },
    {
      "epoch": 2.808094024658203e-05,
      "model_forward_time": 1.7196433544158936,
      "step": 29445
    },
    {
      "epoch": 2.808094024658203e-05,
      "step": 29445,
      "training_step_time": 7.596991539001465
    },
    {
      "epoch": 2.808189392089844e-05,
      "model_forward_time": 1.720851182937622,
      "step": 29446
    },
    {
      "epoch": 2.808189392089844e-05,
      "step": 29446,
      "training_step_time": 7.559279918670654
    },
    {
      "epoch": 2.8082847595214843e-05,
      "model_forward_time": 1.7498326301574707,
      "step": 29447
    },
    {
      "epoch": 2.8082847595214843e-05,
      "step": 29447,
      "training_step_time": 8.69222092628479
    },
    {
      "epoch": 2.808380126953125e-05,
      "model_forward_time": 1.7535052299499512,
      "step": 29448
    },
    {
      "epoch": 2.808380126953125e-05,
      "step": 29448,
      "training_step_time": 7.677758455276489
    },
    {
      "epoch": 2.8084754943847658e-05,
      "model_forward_time": 1.7198538780212402,
      "step": 29449
    },
    {
      "epoch": 2.8084754943847658e-05,
      "step": 29449,
      "training_step_time": 9.578253746032715
    },
    {
      "epoch": 2.8085708618164062e-05,
      "grad_norm": 0.12041456252336502,
      "learning_rate": 8.452870998258423e-06,
      "loss": 0.059,
      "step": 29450
    },
    {
      "epoch": 2.8085708618164062e-05,
      "model_forward_time": 1.7100541591644287,
      "step": 29450
    },
    {
      "dynamics_loss_avg": 0.05458555752411485,
      "epoch": 2.8085708618164062e-05,
      "step": 29450
    },
    {
      "action_loss_avg": 0.0021010040625697004,
      "epoch": 2.8085708618164062e-05,
      "step": 29450
    },
    {
      "epoch": 2.8085708618164062e-05,
      "step": 29450,
      "training_step_time": 9.283414840698242
    },
    {
      "epoch": 2.808666229248047e-05,
      "model_forward_time": 1.7199180126190186,
      "step": 29451
    },
    {
      "epoch": 2.808666229248047e-05,
      "step": 29451,
      "training_step_time": 8.999679565429688
    },
    {
      "epoch": 2.8087615966796877e-05,
      "model_forward_time": 1.717261552810669,
      "step": 29452
    },
    {
      "epoch": 2.8087615966796877e-05,
      "step": 29452,
      "training_step_time": 7.699803829193115
    },
    {
      "epoch": 2.808856964111328e-05,
      "model_forward_time": 2.8061444759368896,
      "step": 29453
    },
    {
      "epoch": 2.808856964111328e-05,
      "step": 29453,
      "training_step_time": 7.6867523193359375
    },
    {
      "epoch": 2.8089523315429688e-05,
      "model_forward_time": 1.7234795093536377,
      "step": 29454
    },
    {
      "epoch": 2.8089523315429688e-05,
      "step": 29454,
      "training_step_time": 7.599776983261108
    },
    {
      "epoch": 2.8090476989746092e-05,
      "model_forward_time": 1.702573537826538,
      "step": 29455
    },
    {
      "epoch": 2.8090476989746092e-05,
      "step": 29455,
      "training_step_time": 7.60306715965271
    },
    {
      "epoch": 2.80914306640625e-05,
      "model_forward_time": 1.7282872200012207,
      "step": 29456
    },
    {
      "epoch": 2.80914306640625e-05,
      "step": 29456,
      "training_step_time": 7.717698574066162
    },
    {
      "epoch": 2.8092384338378907e-05,
      "model_forward_time": 1.8119287490844727,
      "step": 29457
    },
    {
      "epoch": 2.8092384338378907e-05,
      "step": 29457,
      "training_step_time": 8.422903299331665
    },
    {
      "epoch": 2.809333801269531e-05,
      "model_forward_time": 1.714571237564087,
      "step": 29458
    },
    {
      "epoch": 2.809333801269531e-05,
      "step": 29458,
      "training_step_time": 7.610644340515137
    },
    {
      "epoch": 2.809429168701172e-05,
      "model_forward_time": 1.717334270477295,
      "step": 29459
    },
    {
      "epoch": 2.809429168701172e-05,
      "step": 29459,
      "training_step_time": 8.32295560836792
    },
    {
      "epoch": 2.8095245361328126e-05,
      "grad_norm": 0.2391611784696579,
      "learning_rate": 8.451674918330825e-06,
      "loss": 0.058,
      "step": 29460
    },
    {
      "epoch": 2.8095245361328126e-05,
      "model_forward_time": 1.7237732410430908,
      "step": 29460
    },
    {
      "dynamics_loss_avg": 0.0428492097184062,
      "epoch": 2.8095245361328126e-05,
      "step": 29460
    },
    {
      "action_loss_avg": 0.0017640864942222833,
      "epoch": 2.8095245361328126e-05,
      "step": 29460
    },
    {
      "epoch": 2.8095245361328126e-05,
      "step": 29460,
      "training_step_time": 7.695525884628296
    },
    {
      "epoch": 2.809619903564453e-05,
      "model_forward_time": 1.714686632156372,
      "step": 29461
    },
    {
      "epoch": 2.809619903564453e-05,
      "step": 29461,
      "training_step_time": 7.6402246952056885
    },
    {
      "epoch": 2.8097152709960937e-05,
      "model_forward_time": 1.717282772064209,
      "step": 29462
    },
    {
      "epoch": 2.8097152709960937e-05,
      "step": 29462,
      "training_step_time": 11.21277141571045
    },
    {
      "epoch": 2.8098106384277345e-05,
      "model_forward_time": 1.7239480018615723,
      "step": 29463
    },
    {
      "epoch": 2.8098106384277345e-05,
      "step": 29463,
      "training_step_time": 7.563304662704468
    },
    {
      "epoch": 2.809906005859375e-05,
      "model_forward_time": 1.7218594551086426,
      "step": 29464
    },
    {
      "epoch": 2.809906005859375e-05,
      "step": 29464,
      "training_step_time": 7.68670392036438
    },
    {
      "epoch": 2.8100013732910156e-05,
      "model_forward_time": 1.7225477695465088,
      "step": 29465
    },
    {
      "epoch": 2.8100013732910156e-05,
      "step": 29465,
      "training_step_time": 7.714891195297241
    },
    {
      "epoch": 2.8100967407226564e-05,
      "model_forward_time": 1.8055534362792969,
      "step": 29466
    },
    {
      "epoch": 2.8100967407226564e-05,
      "step": 29466,
      "training_step_time": 7.572339296340942
    },
    {
      "epoch": 2.8101921081542968e-05,
      "model_forward_time": 1.7157378196716309,
      "step": 29467
    },
    {
      "epoch": 2.8101921081542968e-05,
      "step": 29467,
      "training_step_time": 7.603870153427124
    },
    {
      "epoch": 2.8102874755859375e-05,
      "model_forward_time": 1.7220406532287598,
      "step": 29468
    },
    {
      "epoch": 2.8102874755859375e-05,
      "step": 29468,
      "training_step_time": 9.759098768234253
    },
    {
      "epoch": 2.8103828430175782e-05,
      "model_forward_time": 1.7190711498260498,
      "step": 29469
    },
    {
      "epoch": 2.8103828430175782e-05,
      "step": 29469,
      "training_step_time": 7.608867645263672
    },
    {
      "epoch": 2.8104782104492186e-05,
      "grad_norm": 0.1370311975479126,
      "learning_rate": 8.450478460933247e-06,
      "loss": 0.0545,
      "step": 29470
    },
    {
      "epoch": 2.8104782104492186e-05,
      "model_forward_time": 1.7203607559204102,
      "step": 29470
    },
    {
      "dynamics_loss_avg": 0.05252471435815096,
      "epoch": 2.8104782104492186e-05,
      "step": 29470
    },
    {
      "action_loss_avg": 0.002556688612094149,
      "epoch": 2.8104782104492186e-05,
      "step": 29470
    },
    {
      "epoch": 2.8104782104492186e-05,
      "step": 29470,
      "training_step_time": 8.094700574874878
    },
    {
      "epoch": 2.8105735778808594e-05,
      "model_forward_time": 1.7138786315917969,
      "step": 29471
    },
    {
      "epoch": 2.8105735778808594e-05,
      "step": 29471,
      "training_step_time": 8.356744289398193
    },
    {
      "epoch": 2.8106689453125e-05,
      "model_forward_time": 2.91343092918396,
      "step": 29472
    },
    {
      "epoch": 2.8106689453125e-05,
      "step": 29472,
      "training_step_time": 8.224838256835938
    },
    {
      "epoch": 2.8107643127441405e-05,
      "model_forward_time": 1.7175803184509277,
      "step": 29473
    },
    {
      "epoch": 2.8107643127441405e-05,
      "step": 29473,
      "training_step_time": 7.2742979526519775
    },
    {
      "epoch": 2.8108596801757813e-05,
      "model_forward_time": 1.7179813385009766,
      "step": 29474
    },
    {
      "epoch": 2.8108596801757813e-05,
      "step": 29474,
      "training_step_time": 7.72077202796936
    },
    {
      "epoch": 2.810955047607422e-05,
      "model_forward_time": 1.7965266704559326,
      "step": 29475
    },
    {
      "epoch": 2.810955047607422e-05,
      "step": 29475,
      "training_step_time": 7.534425735473633
    },
    {
      "epoch": 2.8110504150390624e-05,
      "model_forward_time": 1.7185487747192383,
      "step": 29476
    },
    {
      "epoch": 2.8110504150390624e-05,
      "step": 29476,
      "training_step_time": 7.577582359313965
    },
    {
      "epoch": 2.811145782470703e-05,
      "model_forward_time": 1.70489501953125,
      "step": 29477
    },
    {
      "epoch": 2.811145782470703e-05,
      "step": 29477,
      "training_step_time": 8.705766439437866
    },
    {
      "epoch": 2.811241149902344e-05,
      "model_forward_time": 1.7266199588775635,
      "step": 29478
    },
    {
      "epoch": 2.811241149902344e-05,
      "step": 29478,
      "training_step_time": 9.024518251419067
    },
    {
      "epoch": 2.8113365173339843e-05,
      "model_forward_time": 1.7154512405395508,
      "step": 29479
    },
    {
      "epoch": 2.8113365173339843e-05,
      "step": 29479,
      "training_step_time": 8.606494426727295
    },
    {
      "epoch": 2.811431884765625e-05,
      "grad_norm": 0.09360352903604507,
      "learning_rate": 8.449281626196533e-06,
      "loss": 0.0595,
      "step": 29480
    },
    {
      "epoch": 2.811431884765625e-05,
      "model_forward_time": 1.7225770950317383,
      "step": 29480
    },
    {
      "dynamics_loss_avg": 0.05713452734053135,
      "epoch": 2.811431884765625e-05,
      "step": 29480
    },
    {
      "action_loss_avg": 0.002796717343153432,
      "epoch": 2.811431884765625e-05,
      "step": 29480
    },
    {
      "epoch": 2.811431884765625e-05,
      "step": 29480,
      "training_step_time": 8.581928968429565
    },
    {
      "epoch": 2.8115272521972658e-05,
      "model_forward_time": 2.8712193965911865,
      "step": 29481
    },
    {
      "epoch": 2.8115272521972658e-05,
      "step": 29481,
      "training_step_time": 8.302646398544312
    },
    {
      "epoch": 2.8116226196289062e-05,
      "model_forward_time": 1.717097282409668,
      "step": 29482
    },
    {
      "epoch": 2.8116226196289062e-05,
      "step": 29482,
      "training_step_time": 7.649676084518433
    },
    {
      "epoch": 2.811717987060547e-05,
      "model_forward_time": 1.721254587173462,
      "step": 29483
    },
    {
      "epoch": 2.811717987060547e-05,
      "step": 29483,
      "training_step_time": 7.680893898010254
    },
    {
      "epoch": 2.8118133544921877e-05,
      "model_forward_time": 1.7968273162841797,
      "step": 29484
    },
    {
      "epoch": 2.8118133544921877e-05,
      "step": 29484,
      "training_step_time": 8.569040536880493
    },
    {
      "epoch": 2.811908721923828e-05,
      "model_forward_time": 1.732581377029419,
      "step": 29485
    },
    {
      "epoch": 2.811908721923828e-05,
      "step": 29485,
      "training_step_time": 8.379497289657593
    },
    {
      "epoch": 2.8120040893554688e-05,
      "model_forward_time": 1.7153418064117432,
      "step": 29486
    },
    {
      "epoch": 2.8120040893554688e-05,
      "step": 29486,
      "training_step_time": 7.6673126220703125
    },
    {
      "epoch": 2.8120994567871092e-05,
      "model_forward_time": 1.7217514514923096,
      "step": 29487
    },
    {
      "epoch": 2.8120994567871092e-05,
      "step": 29487,
      "training_step_time": 7.737730503082275
    },
    {
      "epoch": 2.81219482421875e-05,
      "model_forward_time": 1.7261135578155518,
      "step": 29488
    },
    {
      "epoch": 2.81219482421875e-05,
      "step": 29488,
      "training_step_time": 10.063645124435425
    },
    {
      "epoch": 2.8122901916503907e-05,
      "model_forward_time": 1.7147469520568848,
      "step": 29489
    },
    {
      "epoch": 2.8122901916503907e-05,
      "step": 29489,
      "training_step_time": 8.688798189163208
    },
    {
      "epoch": 2.812385559082031e-05,
      "grad_norm": 0.14455054700374603,
      "learning_rate": 8.448084414251565e-06,
      "loss": 0.0584,
      "step": 29490
    },
    {
      "epoch": 2.812385559082031e-05,
      "model_forward_time": 2.8490240573883057,
      "step": 29490
    },
    {
      "dynamics_loss_avg": 0.04934039581567049,
      "epoch": 2.812385559082031e-05,
      "step": 29490
    },
    {
      "action_loss_avg": 0.001895683741895482,
      "epoch": 2.812385559082031e-05,
      "step": 29490
    },
    {
      "epoch": 2.812385559082031e-05,
      "step": 29490,
      "training_step_time": 8.249384880065918
    },
    {
      "epoch": 2.812480926513672e-05,
      "model_forward_time": 1.7193024158477783,
      "step": 29491
    },
    {
      "epoch": 2.812480926513672e-05,
      "step": 29491,
      "training_step_time": 7.593569755554199
    },
    {
      "epoch": 2.8125762939453126e-05,
      "model_forward_time": 1.7146143913269043,
      "step": 29492
    },
    {
      "epoch": 2.8125762939453126e-05,
      "step": 29492,
      "training_step_time": 7.633819103240967
    },
    {
      "epoch": 2.812671661376953e-05,
      "model_forward_time": 1.755248785018921,
      "step": 29493
    },
    {
      "epoch": 2.812671661376953e-05,
      "step": 29493,
      "training_step_time": 7.569290399551392
    },
    {
      "epoch": 2.8127670288085937e-05,
      "model_forward_time": 1.7248406410217285,
      "step": 29494
    },
    {
      "epoch": 2.8127670288085937e-05,
      "step": 29494,
      "training_step_time": 7.565479516983032
    },
    {
      "epoch": 2.8128623962402345e-05,
      "model_forward_time": 1.7690997123718262,
      "step": 29495
    },
    {
      "epoch": 2.8128623962402345e-05,
      "step": 29495,
      "training_step_time": 7.545926570892334
    },
    {
      "epoch": 2.812957763671875e-05,
      "model_forward_time": 1.723787784576416,
      "step": 29496
    },
    {
      "epoch": 2.812957763671875e-05,
      "step": 29496,
      "training_step_time": 7.783057451248169
    },
    {
      "epoch": 2.8130531311035156e-05,
      "model_forward_time": 1.7115576267242432,
      "step": 29497
    },
    {
      "epoch": 2.8130531311035156e-05,
      "step": 29497,
      "training_step_time": 7.652025461196899
    },
    {
      "epoch": 2.8131484985351564e-05,
      "model_forward_time": 1.7271826267242432,
      "step": 29498
    },
    {
      "epoch": 2.8131484985351564e-05,
      "step": 29498,
      "training_step_time": 7.652278184890747
    },
    {
      "epoch": 2.8132438659667968e-05,
      "model_forward_time": 2.830613136291504,
      "step": 29499
    },
    {
      "epoch": 2.8132438659667968e-05,
      "step": 29499,
      "training_step_time": 7.648945093154907
    },
    {
      "epoch": 2.8133392333984375e-05,
      "grad_norm": 0.16374506056308746,
      "learning_rate": 8.446886825229271e-06,
      "loss": 0.0622,
      "step": 29500
    },
    {
      "epoch": 2.8133392333984375e-05,
      "model_forward_time": 1.7215352058410645,
      "step": 29500
    },
    {
      "dynamics_loss_avg": 0.047007841803133485,
      "epoch": 2.8133392333984375e-05,
      "step": 29500
    },
    {
      "action_loss_avg": 0.0024966585187939927,
      "epoch": 2.8133392333984375e-05,
      "step": 29500
    },
    {
      "epoch": 2.8133392333984375e-05,
      "step": 29500,
      "training_step_time": 7.5794358253479
    },
    {
      "epoch": 2.8134346008300782e-05,
      "model_forward_time": 1.7155213356018066,
      "step": 29501
    },
    {
      "epoch": 2.8134346008300782e-05,
      "step": 29501,
      "training_step_time": 7.799535274505615
    },
    {
      "epoch": 2.8135299682617186e-05,
      "model_forward_time": 1.731321096420288,
      "step": 29502
    },
    {
      "epoch": 2.8135299682617186e-05,
      "step": 29502,
      "training_step_time": 7.5697643756866455
    },
    {
      "epoch": 2.8136253356933594e-05,
      "model_forward_time": 1.7309679985046387,
      "step": 29503
    },
    {
      "epoch": 2.8136253356933594e-05,
      "step": 29503,
      "training_step_time": 7.535632371902466
    },
    {
      "epoch": 2.813720703125e-05,
      "model_forward_time": 1.7206661701202393,
      "step": 29504
    },
    {
      "epoch": 2.813720703125e-05,
      "step": 29504,
      "training_step_time": 8.093455791473389
    },
    {
      "epoch": 2.8138160705566405e-05,
      "model_forward_time": 1.721832275390625,
      "step": 29505
    },
    {
      "epoch": 2.8138160705566405e-05,
      "step": 29505,
      "training_step_time": 8.620556116104126
    },
    {
      "epoch": 2.8139114379882813e-05,
      "model_forward_time": 1.712517499923706,
      "step": 29506
    },
    {
      "epoch": 2.8139114379882813e-05,
      "step": 29506,
      "training_step_time": 8.734902620315552
    },
    {
      "epoch": 2.814006805419922e-05,
      "model_forward_time": 1.7286796569824219,
      "step": 29507
    },
    {
      "epoch": 2.814006805419922e-05,
      "step": 29507,
      "training_step_time": 8.187000513076782
    },
    {
      "epoch": 2.8141021728515624e-05,
      "model_forward_time": 2.798220157623291,
      "step": 29508
    },
    {
      "epoch": 2.8141021728515624e-05,
      "step": 29508,
      "training_step_time": 7.596302509307861
    },
    {
      "epoch": 2.814197540283203e-05,
      "model_forward_time": 1.7174835205078125,
      "step": 29509
    },
    {
      "epoch": 2.814197540283203e-05,
      "step": 29509,
      "training_step_time": 8.675673246383667
    },
    {
      "epoch": 2.814292907714844e-05,
      "grad_norm": 0.17751309275627136,
      "learning_rate": 8.445688859260615e-06,
      "loss": 0.0581,
      "step": 29510
    },
    {
      "epoch": 2.814292907714844e-05,
      "model_forward_time": 1.7162840366363525,
      "step": 29510
    },
    {
      "dynamics_loss_avg": 0.03837542720139027,
      "epoch": 2.814292907714844e-05,
      "step": 29510
    },
    {
      "action_loss_avg": 0.0013318685319973155,
      "epoch": 2.814292907714844e-05,
      "step": 29510
    },
    {
      "epoch": 2.814292907714844e-05,
      "step": 29510,
      "training_step_time": 7.560133457183838
    },
    {
      "epoch": 2.8143882751464843e-05,
      "model_forward_time": 1.7358829975128174,
      "step": 29511
    },
    {
      "epoch": 2.8143882751464843e-05,
      "step": 29511,
      "training_step_time": 8.399208784103394
    },
    {
      "epoch": 2.814483642578125e-05,
      "model_forward_time": 1.748929500579834,
      "step": 29512
    },
    {
      "epoch": 2.814483642578125e-05,
      "step": 29512,
      "training_step_time": 7.566553831100464
    },
    {
      "epoch": 2.8145790100097658e-05,
      "model_forward_time": 1.7192537784576416,
      "step": 29513
    },
    {
      "epoch": 2.8145790100097658e-05,
      "step": 29513,
      "training_step_time": 8.34099006652832
    },
    {
      "epoch": 2.8146743774414062e-05,
      "model_forward_time": 1.719667673110962,
      "step": 29514
    },
    {
      "epoch": 2.8146743774414062e-05,
      "step": 29514,
      "training_step_time": 8.811734914779663
    },
    {
      "epoch": 2.814769744873047e-05,
      "model_forward_time": 1.7221198081970215,
      "step": 29515
    },
    {
      "epoch": 2.814769744873047e-05,
      "step": 29515,
      "training_step_time": 7.628998756408691
    },
    {
      "epoch": 2.8148651123046873e-05,
      "model_forward_time": 1.7182679176330566,
      "step": 29516
    },
    {
      "epoch": 2.8148651123046873e-05,
      "step": 29516,
      "training_step_time": 8.551689624786377
    },
    {
      "epoch": 2.814960479736328e-05,
      "model_forward_time": 1.7307143211364746,
      "step": 29517
    },
    {
      "epoch": 2.814960479736328e-05,
      "step": 29517,
      "training_step_time": 8.908597230911255
    },
    {
      "epoch": 2.8150558471679688e-05,
      "model_forward_time": 2.898350238800049,
      "step": 29518
    },
    {
      "epoch": 2.8150558471679688e-05,
      "step": 29518,
      "training_step_time": 7.7359209060668945
    },
    {
      "epoch": 2.8151512145996092e-05,
      "model_forward_time": 1.3425891399383545,
      "step": 29519
    },
    {
      "epoch": 2.8151512145996092e-05,
      "step": 29519,
      "training_step_time": 7.7324864864349365
    },
    {
      "epoch": 2.81524658203125e-05,
      "grad_norm": 0.17110304534435272,
      "learning_rate": 8.444490516476606e-06,
      "loss": 0.0616,
      "step": 29520
    },
    {
      "epoch": 2.81524658203125e-05,
      "model_forward_time": 1.7196075916290283,
      "step": 29520
    },
    {
      "dynamics_loss_avg": 0.051426134258508685,
      "epoch": 2.81524658203125e-05,
      "step": 29520
    },
    {
      "action_loss_avg": 0.001676173077430576,
      "epoch": 2.81524658203125e-05,
      "step": 29520
    },
    {
      "epoch": 2.81524658203125e-05,
      "step": 29520,
      "training_step_time": 8.47385859489441
    },
    {
      "epoch": 2.8153419494628907e-05,
      "model_forward_time": 1.7347052097320557,
      "step": 29521
    },
    {
      "epoch": 2.8153419494628907e-05,
      "step": 29521,
      "training_step_time": 7.627305269241333
    },
    {
      "epoch": 2.815437316894531e-05,
      "model_forward_time": 1.726529598236084,
      "step": 29522
    },
    {
      "epoch": 2.815437316894531e-05,
      "step": 29522,
      "training_step_time": 8.754218339920044
    },
    {
      "epoch": 2.815532684326172e-05,
      "model_forward_time": 1.704758644104004,
      "step": 29523
    },
    {
      "epoch": 2.815532684326172e-05,
      "step": 29523,
      "training_step_time": 10.818479776382446
    },
    {
      "epoch": 2.8156280517578126e-05,
      "model_forward_time": 1.7169194221496582,
      "step": 29524
    },
    {
      "epoch": 2.8156280517578126e-05,
      "step": 29524,
      "training_step_time": 8.62026572227478
    },
    {
      "epoch": 2.815723419189453e-05,
      "model_forward_time": 1.7159576416015625,
      "step": 29525
    },
    {
      "epoch": 2.815723419189453e-05,
      "step": 29525,
      "training_step_time": 9.641195058822632
    },
    {
      "epoch": 2.8158187866210937e-05,
      "model_forward_time": 1.7216923236846924,
      "step": 29526
    },
    {
      "epoch": 2.8158187866210937e-05,
      "step": 29526,
      "training_step_time": 8.572479724884033
    },
    {
      "epoch": 2.8159141540527345e-05,
      "model_forward_time": 2.925475597381592,
      "step": 29527
    },
    {
      "epoch": 2.8159141540527345e-05,
      "step": 29527,
      "training_step_time": 8.34636640548706
    },
    {
      "epoch": 2.816009521484375e-05,
      "model_forward_time": 1.7134265899658203,
      "step": 29528
    },
    {
      "epoch": 2.816009521484375e-05,
      "step": 29528,
      "training_step_time": 7.73655891418457
    },
    {
      "epoch": 2.8161048889160156e-05,
      "model_forward_time": 1.7196292877197266,
      "step": 29529
    },
    {
      "epoch": 2.8161048889160156e-05,
      "step": 29529,
      "training_step_time": 7.6404407024383545
    },
    {
      "epoch": 2.8162002563476564e-05,
      "grad_norm": 0.1586403101682663,
      "learning_rate": 8.443291797008294e-06,
      "loss": 0.0668,
      "step": 29530
    },
    {
      "epoch": 2.8162002563476564e-05,
      "model_forward_time": 1.7519826889038086,
      "step": 29530
    },
    {
      "dynamics_loss_avg": 0.07790728155523538,
      "epoch": 2.8162002563476564e-05,
      "step": 29530
    },
    {
      "action_loss_avg": 0.0029114449745975433,
      "epoch": 2.8162002563476564e-05,
      "step": 29530
    },
    {
      "epoch": 2.8162002563476564e-05,
      "step": 29530,
      "training_step_time": 8.190067768096924
    },
    {
      "epoch": 2.8162956237792968e-05,
      "model_forward_time": 1.726529836654663,
      "step": 29531
    },
    {
      "epoch": 2.8162956237792968e-05,
      "step": 29531,
      "training_step_time": 7.737102746963501
    },
    {
      "epoch": 2.8163909912109375e-05,
      "model_forward_time": 1.7237684726715088,
      "step": 29532
    },
    {
      "epoch": 2.8163909912109375e-05,
      "step": 29532,
      "training_step_time": 8.781147956848145
    },
    {
      "epoch": 2.8164863586425782e-05,
      "model_forward_time": 1.711496114730835,
      "step": 29533
    },
    {
      "epoch": 2.8164863586425782e-05,
      "step": 29533,
      "training_step_time": 9.431320905685425
    },
    {
      "epoch": 2.8165817260742186e-05,
      "model_forward_time": 1.7209272384643555,
      "step": 29534
    },
    {
      "epoch": 2.8165817260742186e-05,
      "step": 29534,
      "training_step_time": 9.371201038360596
    },
    {
      "epoch": 2.8166770935058594e-05,
      "model_forward_time": 1.713686227798462,
      "step": 29535
    },
    {
      "epoch": 2.8166770935058594e-05,
      "step": 29535,
      "training_step_time": 8.579096555709839
    },
    {
      "epoch": 2.8167724609375e-05,
      "model_forward_time": 2.8724255561828613,
      "step": 29536
    },
    {
      "epoch": 2.8167724609375e-05,
      "step": 29536,
      "training_step_time": 7.757951498031616
    },
    {
      "epoch": 2.8168678283691405e-05,
      "model_forward_time": 1.7254600524902344,
      "step": 29537
    },
    {
      "epoch": 2.8168678283691405e-05,
      "step": 29537,
      "training_step_time": 7.637974500656128
    },
    {
      "epoch": 2.8169631958007813e-05,
      "model_forward_time": 1.7221460342407227,
      "step": 29538
    },
    {
      "epoch": 2.8169631958007813e-05,
      "step": 29538,
      "training_step_time": 8.576836109161377
    },
    {
      "epoch": 2.817058563232422e-05,
      "model_forward_time": 1.833998441696167,
      "step": 29539
    },
    {
      "epoch": 2.817058563232422e-05,
      "step": 29539,
      "training_step_time": 7.634179353713989
    },
    {
      "epoch": 2.8171539306640624e-05,
      "grad_norm": 0.17800624668598175,
      "learning_rate": 8.442092700986766e-06,
      "loss": 0.0612,
      "step": 29540
    },
    {
      "epoch": 2.8171539306640624e-05,
      "model_forward_time": 1.7132327556610107,
      "step": 29540
    },
    {
      "dynamics_loss_avg": 0.05999172795563936,
      "epoch": 2.8171539306640624e-05,
      "step": 29540
    },
    {
      "action_loss_avg": 0.0018797632888890802,
      "epoch": 2.8171539306640624e-05,
      "step": 29540
    },
    {
      "epoch": 2.8171539306640624e-05,
      "step": 29540,
      "training_step_time": 8.729815006256104
    },
    {
      "epoch": 2.817249298095703e-05,
      "model_forward_time": 1.7335865497589111,
      "step": 29541
    },
    {
      "epoch": 2.817249298095703e-05,
      "step": 29541,
      "training_step_time": 7.6482093334198
    },
    {
      "epoch": 2.817344665527344e-05,
      "model_forward_time": 1.7255394458770752,
      "step": 29542
    },
    {
      "epoch": 2.817344665527344e-05,
      "step": 29542,
      "training_step_time": 7.820095539093018
    },
    {
      "epoch": 2.8174400329589843e-05,
      "model_forward_time": 1.7239353656768799,
      "step": 29543
    },
    {
      "epoch": 2.8174400329589843e-05,
      "step": 29543,
      "training_step_time": 8.305389642715454
    },
    {
      "epoch": 2.817535400390625e-05,
      "model_forward_time": 1.739739179611206,
      "step": 29544
    },
    {
      "epoch": 2.817535400390625e-05,
      "step": 29544,
      "training_step_time": 9.256982326507568
    },
    {
      "epoch": 2.8176307678222658e-05,
      "model_forward_time": 2.8530149459838867,
      "step": 29545
    },
    {
      "epoch": 2.8176307678222658e-05,
      "step": 29545,
      "training_step_time": 8.57039761543274
    },
    {
      "epoch": 2.8177261352539062e-05,
      "model_forward_time": 1.7333359718322754,
      "step": 29546
    },
    {
      "epoch": 2.8177261352539062e-05,
      "step": 29546,
      "training_step_time": 7.677054405212402
    },
    {
      "epoch": 2.817821502685547e-05,
      "model_forward_time": 1.7227528095245361,
      "step": 29547
    },
    {
      "epoch": 2.817821502685547e-05,
      "step": 29547,
      "training_step_time": 7.786454916000366
    },
    {
      "epoch": 2.8179168701171873e-05,
      "model_forward_time": 1.7967560291290283,
      "step": 29548
    },
    {
      "epoch": 2.8179168701171873e-05,
      "step": 29548,
      "training_step_time": 7.583611726760864
    },
    {
      "epoch": 2.818012237548828e-05,
      "model_forward_time": 1.7262816429138184,
      "step": 29549
    },
    {
      "epoch": 2.818012237548828e-05,
      "step": 29549,
      "training_step_time": 7.694045782089233
    },
    {
      "epoch": 2.8181076049804688e-05,
      "grad_norm": 0.13420939445495605,
      "learning_rate": 8.440893228543157e-06,
      "loss": 0.064,
      "step": 29550
    },
    {
      "epoch": 2.8181076049804688e-05,
      "model_forward_time": 1.7297589778900146,
      "step": 29550
    },
    {
      "dynamics_loss_avg": 0.05047077629715204,
      "epoch": 2.8181076049804688e-05,
      "step": 29550
    },
    {
      "action_loss_avg": 0.0024397669942118226,
      "epoch": 2.8181076049804688e-05,
      "step": 29550
    },
    {
      "epoch": 2.8181076049804688e-05,
      "step": 29550,
      "training_step_time": 7.642120599746704
    },
    {
      "epoch": 2.8182029724121092e-05,
      "model_forward_time": 1.7262396812438965,
      "step": 29551
    },
    {
      "epoch": 2.8182029724121092e-05,
      "step": 29551,
      "training_step_time": 8.444664478302002
    },
    {
      "epoch": 2.81829833984375e-05,
      "model_forward_time": 1.7276039123535156,
      "step": 29552
    },
    {
      "epoch": 2.81829833984375e-05,
      "step": 29552,
      "training_step_time": 9.665084600448608
    },
    {
      "epoch": 2.8183937072753907e-05,
      "model_forward_time": 1.3529624938964844,
      "step": 29553
    },
    {
      "epoch": 2.8183937072753907e-05,
      "step": 29553,
      "training_step_time": 9.09137511253357
    },
    {
      "epoch": 2.818489074707031e-05,
      "model_forward_time": 2.80692720413208,
      "step": 29554
    },
    {
      "epoch": 2.818489074707031e-05,
      "step": 29554,
      "training_step_time": 7.578314781188965
    },
    {
      "epoch": 2.818584442138672e-05,
      "model_forward_time": 1.7164325714111328,
      "step": 29555
    },
    {
      "epoch": 2.818584442138672e-05,
      "step": 29555,
      "training_step_time": 8.986835479736328
    },
    {
      "epoch": 2.8186798095703126e-05,
      "model_forward_time": 1.7242772579193115,
      "step": 29556
    },
    {
      "epoch": 2.8186798095703126e-05,
      "step": 29556,
      "training_step_time": 7.673143148422241
    },
    {
      "epoch": 2.818775177001953e-05,
      "model_forward_time": 1.775635004043579,
      "step": 29557
    },
    {
      "epoch": 2.818775177001953e-05,
      "step": 29557,
      "training_step_time": 7.524877071380615
    },
    {
      "epoch": 2.8188705444335937e-05,
      "model_forward_time": 1.7169218063354492,
      "step": 29558
    },
    {
      "epoch": 2.8188705444335937e-05,
      "step": 29558,
      "training_step_time": 9.299054622650146
    },
    {
      "epoch": 2.8189659118652345e-05,
      "model_forward_time": 1.7217638492584229,
      "step": 29559
    },
    {
      "epoch": 2.8189659118652345e-05,
      "step": 29559,
      "training_step_time": 8.182237148284912
    },
    {
      "epoch": 2.819061279296875e-05,
      "grad_norm": 0.12622594833374023,
      "learning_rate": 8.439693379808637e-06,
      "loss": 0.0604,
      "step": 29560
    },
    {
      "epoch": 2.819061279296875e-05,
      "model_forward_time": 1.3612158298492432,
      "step": 29560
    },
    {
      "dynamics_loss_avg": 0.04845090862363577,
      "epoch": 2.819061279296875e-05,
      "step": 29560
    },
    {
      "action_loss_avg": 0.003559759707422927,
      "epoch": 2.819061279296875e-05,
      "step": 29560
    },
    {
      "epoch": 2.819061279296875e-05,
      "step": 29560,
      "training_step_time": 7.73646879196167
    },
    {
      "epoch": 2.8191566467285156e-05,
      "model_forward_time": 1.7328407764434814,
      "step": 29561
    },
    {
      "epoch": 2.8191566467285156e-05,
      "step": 29561,
      "training_step_time": 7.703106164932251
    },
    {
      "epoch": 2.8192520141601564e-05,
      "model_forward_time": 1.7282893657684326,
      "step": 29562
    },
    {
      "epoch": 2.8192520141601564e-05,
      "step": 29562,
      "training_step_time": 7.836406707763672
    },
    {
      "epoch": 2.8193473815917968e-05,
      "model_forward_time": 1.728391408920288,
      "step": 29563
    },
    {
      "epoch": 2.8193473815917968e-05,
      "step": 29563,
      "training_step_time": 8.71950364112854
    },
    {
      "epoch": 2.8194427490234375e-05,
      "model_forward_time": 2.862855911254883,
      "step": 29564
    },
    {
      "epoch": 2.8194427490234375e-05,
      "step": 29564,
      "training_step_time": 7.651450872421265
    },
    {
      "epoch": 2.8195381164550782e-05,
      "model_forward_time": 1.7188992500305176,
      "step": 29565
    },
    {
      "epoch": 2.8195381164550782e-05,
      "step": 29565,
      "training_step_time": 8.331254482269287
    },
    {
      "epoch": 2.8196334838867186e-05,
      "model_forward_time": 1.7502331733703613,
      "step": 29566
    },
    {
      "epoch": 2.8196334838867186e-05,
      "step": 29566,
      "training_step_time": 7.515076160430908
    },
    {
      "epoch": 2.8197288513183594e-05,
      "model_forward_time": 1.7291662693023682,
      "step": 29567
    },
    {
      "epoch": 2.8197288513183594e-05,
      "step": 29567,
      "training_step_time": 7.631757974624634
    },
    {
      "epoch": 2.81982421875e-05,
      "model_forward_time": 1.7276461124420166,
      "step": 29568
    },
    {
      "epoch": 2.81982421875e-05,
      "step": 29568,
      "training_step_time": 7.594332933425903
    },
    {
      "epoch": 2.8199195861816405e-05,
      "model_forward_time": 1.7239494323730469,
      "step": 29569
    },
    {
      "epoch": 2.8199195861816405e-05,
      "step": 29569,
      "training_step_time": 7.627605199813843
    },
    {
      "epoch": 2.8200149536132813e-05,
      "grad_norm": 0.18612715601921082,
      "learning_rate": 8.43849315491442e-06,
      "loss": 0.0692,
      "step": 29570
    },
    {
      "epoch": 2.8200149536132813e-05,
      "model_forward_time": 1.7275798320770264,
      "step": 29570
    },
    {
      "dynamics_loss_avg": 0.0588624220341444,
      "epoch": 2.8200149536132813e-05,
      "step": 29570
    },
    {
      "action_loss_avg": 0.0019383188686333598,
      "epoch": 2.8200149536132813e-05,
      "step": 29570
    },
    {
      "epoch": 2.8200149536132813e-05,
      "step": 29570,
      "training_step_time": 7.636271953582764
    },
    {
      "epoch": 2.820110321044922e-05,
      "model_forward_time": 1.3572447299957275,
      "step": 29571
    },
    {
      "epoch": 2.820110321044922e-05,
      "step": 29571,
      "training_step_time": 7.7519402503967285
    },
    {
      "epoch": 2.8202056884765624e-05,
      "model_forward_time": 1.7159414291381836,
      "step": 29572
    },
    {
      "epoch": 2.8202056884765624e-05,
      "step": 29572,
      "training_step_time": 8.602295160293579
    },
    {
      "epoch": 2.820301055908203e-05,
      "model_forward_time": 2.2972583770751953,
      "step": 29573
    },
    {
      "epoch": 2.820301055908203e-05,
      "step": 29573,
      "training_step_time": 7.626738786697388
    },
    {
      "epoch": 2.820396423339844e-05,
      "model_forward_time": 0.6925532817840576,
      "step": 29574
    },
    {
      "epoch": 2.820396423339844e-05,
      "step": 29574,
      "training_step_time": 7.578039169311523
    },
    {
      "epoch": 2.8204917907714843e-05,
      "model_forward_time": 1.7538301944732666,
      "step": 29575
    },
    {
      "epoch": 2.8204917907714843e-05,
      "step": 29575,
      "training_step_time": 8.60591745376587
    },
    {
      "epoch": 2.820587158203125e-05,
      "model_forward_time": 1.8817076683044434,
      "step": 29576
    },
    {
      "epoch": 2.820587158203125e-05,
      "step": 29576,
      "training_step_time": 7.594712972640991
    },
    {
      "epoch": 2.8206825256347658e-05,
      "model_forward_time": 1.741152048110962,
      "step": 29577
    },
    {
      "epoch": 2.8206825256347658e-05,
      "step": 29577,
      "training_step_time": 7.57685661315918
    },
    {
      "epoch": 2.8207778930664062e-05,
      "model_forward_time": 1.7613029479980469,
      "step": 29578
    },
    {
      "epoch": 2.8207778930664062e-05,
      "step": 29578,
      "training_step_time": 7.687501668930054
    },
    {
      "epoch": 2.820873260498047e-05,
      "model_forward_time": 1.743375301361084,
      "step": 29579
    },
    {
      "epoch": 2.820873260498047e-05,
      "step": 29579,
      "training_step_time": 8.61226224899292
    },
    {
      "epoch": 2.8209686279296873e-05,
      "grad_norm": 0.2212650626897812,
      "learning_rate": 8.437292553991763e-06,
      "loss": 0.065,
      "step": 29580
    },
    {
      "epoch": 2.8209686279296873e-05,
      "model_forward_time": 1.746074914932251,
      "step": 29580
    },
    {
      "dynamics_loss_avg": 0.07440764531493187,
      "epoch": 2.8209686279296873e-05,
      "step": 29580
    },
    {
      "action_loss_avg": 0.002985308435745537,
      "epoch": 2.8209686279296873e-05,
      "step": 29580
    },
    {
      "epoch": 2.8209686279296873e-05,
      "step": 29580,
      "training_step_time": 7.6956703662872314
    },
    {
      "epoch": 2.821063995361328e-05,
      "model_forward_time": 1.7569506168365479,
      "step": 29581
    },
    {
      "epoch": 2.821063995361328e-05,
      "step": 29581,
      "training_step_time": 8.65646243095398
    },
    {
      "epoch": 2.8211593627929688e-05,
      "model_forward_time": 1.740699052810669,
      "step": 29582
    },
    {
      "epoch": 2.8211593627929688e-05,
      "step": 29582,
      "training_step_time": 8.403958082199097
    },
    {
      "epoch": 2.8212547302246092e-05,
      "model_forward_time": 2.916534662246704,
      "step": 29583
    },
    {
      "epoch": 2.8212547302246092e-05,
      "step": 29583,
      "training_step_time": 7.689037799835205
    },
    {
      "epoch": 2.82135009765625e-05,
      "model_forward_time": 1.7393543720245361,
      "step": 29584
    },
    {
      "epoch": 2.82135009765625e-05,
      "step": 29584,
      "training_step_time": 7.611623764038086
    },
    {
      "epoch": 2.8214454650878907e-05,
      "model_forward_time": 1.860264539718628,
      "step": 29585
    },
    {
      "epoch": 2.8214454650878907e-05,
      "step": 29585,
      "training_step_time": 7.556737422943115
    },
    {
      "epoch": 2.821540832519531e-05,
      "model_forward_time": 1.748037576675415,
      "step": 29586
    },
    {
      "epoch": 2.821540832519531e-05,
      "step": 29586,
      "training_step_time": 7.666374206542969
    },
    {
      "epoch": 2.821636199951172e-05,
      "model_forward_time": 1.7414700984954834,
      "step": 29587
    },
    {
      "epoch": 2.821636199951172e-05,
      "step": 29587,
      "training_step_time": 8.50804328918457
    },
    {
      "epoch": 2.8217315673828126e-05,
      "model_forward_time": 1.7498297691345215,
      "step": 29588
    },
    {
      "epoch": 2.8217315673828126e-05,
      "step": 29588,
      "training_step_time": 8.631325483322144
    },
    {
      "epoch": 2.821826934814453e-05,
      "model_forward_time": 1.7280914783477783,
      "step": 29589
    },
    {
      "epoch": 2.821826934814453e-05,
      "step": 29589,
      "training_step_time": 8.279993295669556
    },
    {
      "epoch": 2.8219223022460937e-05,
      "grad_norm": 0.21432001888751984,
      "learning_rate": 8.43609157717196e-06,
      "loss": 0.0632,
      "step": 29590
    },
    {
      "epoch": 2.8219223022460937e-05,
      "model_forward_time": 1.7415285110473633,
      "step": 29590
    },
    {
      "dynamics_loss_avg": 0.0645890288054943,
      "epoch": 2.8219223022460937e-05,
      "step": 29590
    },
    {
      "action_loss_avg": 0.0032830028329044582,
      "epoch": 2.8219223022460937e-05,
      "step": 29590
    },
    {
      "epoch": 2.8219223022460937e-05,
      "step": 29590,
      "training_step_time": 7.689257383346558
    },
    {
      "epoch": 2.8220176696777345e-05,
      "model_forward_time": 1.7473711967468262,
      "step": 29591
    },
    {
      "epoch": 2.8220176696777345e-05,
      "step": 29591,
      "training_step_time": 8.590504884719849
    },
    {
      "epoch": 2.822113037109375e-05,
      "model_forward_time": 2.8928654193878174,
      "step": 29592
    },
    {
      "epoch": 2.822113037109375e-05,
      "step": 29592,
      "training_step_time": 9.547372102737427
    },
    {
      "epoch": 2.8222084045410156e-05,
      "model_forward_time": 1.7434892654418945,
      "step": 29593
    },
    {
      "epoch": 2.8222084045410156e-05,
      "step": 29593,
      "training_step_time": 8.131980419158936
    },
    {
      "epoch": 2.8223037719726564e-05,
      "model_forward_time": 1.8156986236572266,
      "step": 29594
    },
    {
      "epoch": 2.8223037719726564e-05,
      "step": 29594,
      "training_step_time": 7.5510029792785645
    },
    {
      "epoch": 2.8223991394042968e-05,
      "model_forward_time": 1.7359702587127686,
      "step": 29595
    },
    {
      "epoch": 2.8223991394042968e-05,
      "step": 29595,
      "training_step_time": 7.641377687454224
    },
    {
      "epoch": 2.8224945068359375e-05,
      "model_forward_time": 1.7418100833892822,
      "step": 29596
    },
    {
      "epoch": 2.8224945068359375e-05,
      "step": 29596,
      "training_step_time": 7.871584177017212
    },
    {
      "epoch": 2.8225898742675782e-05,
      "model_forward_time": 1.7360544204711914,
      "step": 29597
    },
    {
      "epoch": 2.8225898742675782e-05,
      "step": 29597,
      "training_step_time": 9.53361701965332
    },
    {
      "epoch": 2.8226852416992187e-05,
      "model_forward_time": 1.7404236793518066,
      "step": 29598
    },
    {
      "epoch": 2.8226852416992187e-05,
      "step": 29598,
      "training_step_time": 8.241237163543701
    },
    {
      "epoch": 2.8227806091308594e-05,
      "model_forward_time": 1.7388865947723389,
      "step": 29599
    },
    {
      "epoch": 2.8227806091308594e-05,
      "step": 29599,
      "training_step_time": 8.536206722259521
    },
    {
      "epoch": 2.8228759765625e-05,
      "grad_norm": 0.125672847032547,
      "learning_rate": 8.434890224586348e-06,
      "loss": 0.0642,
      "step": 29600
    },
    {
      "epoch": 2.8228759765625e-05,
      "model_forward_time": 1.7543387413024902,
      "step": 29600
    },
    {
      "dynamics_loss_avg": 0.03751222100108862,
      "epoch": 2.8228759765625e-05,
      "step": 29600
    },
    {
      "action_loss_avg": 0.0012628473428776487,
      "epoch": 2.8228759765625e-05,
      "step": 29600
    },
    {
      "epoch": 2.8228759765625e-05,
      "step": 29600,
      "training_step_time": 7.751535654067993
    },
    {
      "epoch": 2.8229713439941405e-05,
      "model_forward_time": 2.853978395462036,
      "step": 29601
    },
    {
      "epoch": 2.8229713439941405e-05,
      "step": 29601,
      "training_step_time": 7.664585113525391
    },
    {
      "epoch": 2.8230667114257813e-05,
      "model_forward_time": 1.7372157573699951,
      "step": 29602
    },
    {
      "epoch": 2.8230667114257813e-05,
      "step": 29602,
      "training_step_time": 7.6431262493133545
    },
    {
      "epoch": 2.823162078857422e-05,
      "model_forward_time": 1.8024146556854248,
      "step": 29603
    },
    {
      "epoch": 2.823162078857422e-05,
      "step": 29603,
      "training_step_time": 7.665765047073364
    },
    {
      "epoch": 2.8232574462890624e-05,
      "model_forward_time": 1.7396392822265625,
      "step": 29604
    },
    {
      "epoch": 2.8232574462890624e-05,
      "step": 29604,
      "training_step_time": 8.26621961593628
    },
    {
      "epoch": 2.823352813720703e-05,
      "model_forward_time": 1.7388100624084473,
      "step": 29605
    },
    {
      "epoch": 2.823352813720703e-05,
      "step": 29605,
      "training_step_time": 7.824190139770508
    },
    {
      "epoch": 2.823448181152344e-05,
      "model_forward_time": 1.7536489963531494,
      "step": 29606
    },
    {
      "epoch": 2.823448181152344e-05,
      "step": 29606,
      "training_step_time": 7.684732675552368
    },
    {
      "epoch": 2.8235435485839843e-05,
      "model_forward_time": 1.7372064590454102,
      "step": 29607
    },
    {
      "epoch": 2.8235435485839843e-05,
      "step": 29607,
      "training_step_time": 7.74730658531189
    },
    {
      "epoch": 2.823638916015625e-05,
      "model_forward_time": 1.7352404594421387,
      "step": 29608
    },
    {
      "epoch": 2.823638916015625e-05,
      "step": 29608,
      "training_step_time": 7.685209035873413
    },
    {
      "epoch": 2.8237342834472658e-05,
      "model_forward_time": 1.741722822189331,
      "step": 29609
    },
    {
      "epoch": 2.8237342834472658e-05,
      "step": 29609,
      "training_step_time": 8.275688409805298
    },
    {
      "epoch": 2.8238296508789062e-05,
      "grad_norm": 0.1402169167995453,
      "learning_rate": 8.433688496366304e-06,
      "loss": 0.0654,
      "step": 29610
    },
    {
      "epoch": 2.8238296508789062e-05,
      "model_forward_time": 2.8564281463623047,
      "step": 29610
    },
    {
      "dynamics_loss_avg": 0.05673555405810475,
      "epoch": 2.8238296508789062e-05,
      "step": 29610
    },
    {
      "action_loss_avg": 0.0031075791630428284,
      "epoch": 2.8238296508789062e-05,
      "step": 29610
    },
    {
      "epoch": 2.8238296508789062e-05,
      "step": 29610,
      "training_step_time": 7.700787544250488
    },
    {
      "epoch": 2.823925018310547e-05,
      "model_forward_time": 1.745811939239502,
      "step": 29611
    },
    {
      "epoch": 2.823925018310547e-05,
      "step": 29611,
      "training_step_time": 8.479217052459717
    },
    {
      "epoch": 2.8240203857421873e-05,
      "model_forward_time": 1.775635004043579,
      "step": 29612
    },
    {
      "epoch": 2.8240203857421873e-05,
      "step": 29612,
      "training_step_time": 7.658324241638184
    },
    {
      "epoch": 2.824115753173828e-05,
      "model_forward_time": 1.7378156185150146,
      "step": 29613
    },
    {
      "epoch": 2.824115753173828e-05,
      "step": 29613,
      "training_step_time": 8.550265789031982
    },
    {
      "epoch": 2.8242111206054688e-05,
      "model_forward_time": 1.7578606605529785,
      "step": 29614
    },
    {
      "epoch": 2.8242111206054688e-05,
      "step": 29614,
      "training_step_time": 8.338255405426025
    },
    {
      "epoch": 2.8243064880371092e-05,
      "model_forward_time": 1.743617296218872,
      "step": 29615
    },
    {
      "epoch": 2.8243064880371092e-05,
      "step": 29615,
      "training_step_time": 8.239725589752197
    },
    {
      "epoch": 2.82440185546875e-05,
      "model_forward_time": 1.7399296760559082,
      "step": 29616
    },
    {
      "epoch": 2.82440185546875e-05,
      "step": 29616,
      "training_step_time": 7.7560341358184814
    },
    {
      "epoch": 2.8244972229003907e-05,
      "model_forward_time": 1.747910737991333,
      "step": 29617
    },
    {
      "epoch": 2.8244972229003907e-05,
      "step": 29617,
      "training_step_time": 8.361670017242432
    },
    {
      "epoch": 2.824592590332031e-05,
      "model_forward_time": 1.7401783466339111,
      "step": 29618
    },
    {
      "epoch": 2.824592590332031e-05,
      "step": 29618,
      "training_step_time": 7.7451982498168945
    },
    {
      "epoch": 2.824687957763672e-05,
      "model_forward_time": 1.7318053245544434,
      "step": 29619
    },
    {
      "epoch": 2.824687957763672e-05,
      "step": 29619,
      "training_step_time": 8.158087491989136
    },
    {
      "epoch": 2.8247833251953126e-05,
      "grad_norm": 0.1296529620885849,
      "learning_rate": 8.432486392643249e-06,
      "loss": 0.0614,
      "step": 29620
    },
    {
      "epoch": 2.8247833251953126e-05,
      "model_forward_time": 1.7364249229431152,
      "step": 29620
    },
    {
      "dynamics_loss_avg": 0.06582457534968852,
      "epoch": 2.8247833251953126e-05,
      "step": 29620
    },
    {
      "action_loss_avg": 0.002837920928141102,
      "epoch": 2.8247833251953126e-05,
      "step": 29620
    },
    {
      "epoch": 2.8247833251953126e-05,
      "step": 29620,
      "training_step_time": 7.616903066635132
    },
    {
      "epoch": 2.824878692626953e-05,
      "model_forward_time": 1.7378602027893066,
      "step": 29621
    },
    {
      "epoch": 2.824878692626953e-05,
      "step": 29621,
      "training_step_time": 7.722566366195679
    },
    {
      "epoch": 2.8249740600585937e-05,
      "model_forward_time": 1.748110055923462,
      "step": 29622
    },
    {
      "epoch": 2.8249740600585937e-05,
      "step": 29622,
      "training_step_time": 8.1985023021698
    },
    {
      "epoch": 2.8250694274902345e-05,
      "model_forward_time": 1.7376606464385986,
      "step": 29623
    },
    {
      "epoch": 2.8250694274902345e-05,
      "step": 29623,
      "training_step_time": 7.704646587371826
    },
    {
      "epoch": 2.825164794921875e-05,
      "model_forward_time": 1.7367310523986816,
      "step": 29624
    },
    {
      "epoch": 2.825164794921875e-05,
      "step": 29624,
      "training_step_time": 8.700925827026367
    },
    {
      "epoch": 2.8252601623535156e-05,
      "model_forward_time": 1.7388849258422852,
      "step": 29625
    },
    {
      "epoch": 2.8252601623535156e-05,
      "step": 29625,
      "training_step_time": 8.15148401260376
    },
    {
      "epoch": 2.8253555297851564e-05,
      "model_forward_time": 1.7379004955291748,
      "step": 29626
    },
    {
      "epoch": 2.8253555297851564e-05,
      "step": 29626,
      "training_step_time": 7.571962594985962
    },
    {
      "epoch": 2.8254508972167968e-05,
      "model_forward_time": 1.7411000728607178,
      "step": 29627
    },
    {
      "epoch": 2.8254508972167968e-05,
      "step": 29627,
      "training_step_time": 8.699404001235962
    },
    {
      "epoch": 2.8255462646484375e-05,
      "model_forward_time": 1.7345366477966309,
      "step": 29628
    },
    {
      "epoch": 2.8255462646484375e-05,
      "step": 29628,
      "training_step_time": 7.691570043563843
    },
    {
      "epoch": 2.8256416320800783e-05,
      "model_forward_time": 2.8981847763061523,
      "step": 29629
    },
    {
      "epoch": 2.8256416320800783e-05,
      "step": 29629,
      "training_step_time": 7.629277229309082
    },
    {
      "epoch": 2.8257369995117187e-05,
      "grad_norm": 0.17417725920677185,
      "learning_rate": 8.431283913548642e-06,
      "loss": 0.066,
      "step": 29630
    },
    {
      "epoch": 2.8257369995117187e-05,
      "model_forward_time": 1.8928570747375488,
      "step": 29630
    },
    {
      "dynamics_loss_avg": 0.06953426338732242,
      "epoch": 2.8257369995117187e-05,
      "step": 29630
    },
    {
      "action_loss_avg": 0.0030084094847552477,
      "epoch": 2.8257369995117187e-05,
      "step": 29630
    },
    {
      "epoch": 2.8257369995117187e-05,
      "step": 29630,
      "training_step_time": 7.585003137588501
    },
    {
      "epoch": 2.8258323669433594e-05,
      "model_forward_time": 1.922734022140503,
      "step": 29631
    },
    {
      "epoch": 2.8258323669433594e-05,
      "step": 29631,
      "training_step_time": 7.530442714691162
    },
    {
      "epoch": 2.825927734375e-05,
      "model_forward_time": 1.8951401710510254,
      "step": 29632
    },
    {
      "epoch": 2.825927734375e-05,
      "step": 29632,
      "training_step_time": 7.635572195053101
    },
    {
      "epoch": 2.8260231018066405e-05,
      "model_forward_time": 1.9223434925079346,
      "step": 29633
    },
    {
      "epoch": 2.8260231018066405e-05,
      "step": 29633,
      "training_step_time": 7.593641519546509
    },
    {
      "epoch": 2.8261184692382813e-05,
      "model_forward_time": 1.9140312671661377,
      "step": 29634
    },
    {
      "epoch": 2.8261184692382813e-05,
      "step": 29634,
      "training_step_time": 8.755163431167603
    },
    {
      "epoch": 2.826213836669922e-05,
      "model_forward_time": 1.919825792312622,
      "step": 29635
    },
    {
      "epoch": 2.826213836669922e-05,
      "step": 29635,
      "training_step_time": 7.5960986614227295
    },
    {
      "epoch": 2.8263092041015624e-05,
      "model_forward_time": 1.9141261577606201,
      "step": 29636
    },
    {
      "epoch": 2.8263092041015624e-05,
      "step": 29636,
      "training_step_time": 9.078977346420288
    },
    {
      "epoch": 2.826404571533203e-05,
      "model_forward_time": 1.8993291854858398,
      "step": 29637
    },
    {
      "epoch": 2.826404571533203e-05,
      "step": 29637,
      "training_step_time": 7.763249635696411
    },
    {
      "epoch": 2.826499938964844e-05,
      "model_forward_time": 2.977421283721924,
      "step": 29638
    },
    {
      "epoch": 2.826499938964844e-05,
      "step": 29638,
      "training_step_time": 7.782967567443848
    },
    {
      "epoch": 2.8265953063964843e-05,
      "model_forward_time": 1.815244197845459,
      "step": 29639
    },
    {
      "epoch": 2.8265953063964843e-05,
      "step": 29639,
      "training_step_time": 9.259277105331421
    },
    {
      "epoch": 2.826690673828125e-05,
      "grad_norm": 0.2073514759540558,
      "learning_rate": 8.430081059213985e-06,
      "loss": 0.0601,
      "step": 29640
    },
    {
      "epoch": 2.826690673828125e-05,
      "model_forward_time": 1.8268389701843262,
      "step": 29640
    },
    {
      "dynamics_loss_avg": 0.05154881421476602,
      "epoch": 2.826690673828125e-05,
      "step": 29640
    },
    {
      "action_loss_avg": 0.0021603681554552166,
      "epoch": 2.826690673828125e-05,
      "step": 29640
    },
    {
      "epoch": 2.826690673828125e-05,
      "step": 29640,
      "training_step_time": 7.618407487869263
    },
    {
      "epoch": 2.8267860412597655e-05,
      "model_forward_time": 1.7336359024047852,
      "step": 29641
    },
    {
      "epoch": 2.8267860412597655e-05,
      "step": 29641,
      "training_step_time": 8.642391443252563
    },
    {
      "epoch": 2.8268814086914062e-05,
      "model_forward_time": 1.7389857769012451,
      "step": 29642
    },
    {
      "epoch": 2.8268814086914062e-05,
      "step": 29642,
      "training_step_time": 7.564216613769531
    },
    {
      "epoch": 2.826976776123047e-05,
      "model_forward_time": 1.7880337238311768,
      "step": 29643
    },
    {
      "epoch": 2.826976776123047e-05,
      "step": 29643,
      "training_step_time": 7.812968015670776
    },
    {
      "epoch": 2.8270721435546873e-05,
      "model_forward_time": 1.727647066116333,
      "step": 29644
    },
    {
      "epoch": 2.8270721435546873e-05,
      "step": 29644,
      "training_step_time": 8.482994794845581
    },
    {
      "epoch": 2.827167510986328e-05,
      "model_forward_time": 1.7306079864501953,
      "step": 29645
    },
    {
      "epoch": 2.827167510986328e-05,
      "step": 29645,
      "training_step_time": 8.18480110168457
    },
    {
      "epoch": 2.8272628784179688e-05,
      "model_forward_time": 1.7347278594970703,
      "step": 29646
    },
    {
      "epoch": 2.8272628784179688e-05,
      "step": 29646,
      "training_step_time": 8.611997842788696
    },
    {
      "epoch": 2.8273582458496092e-05,
      "model_forward_time": 2.9170219898223877,
      "step": 29647
    },
    {
      "epoch": 2.8273582458496092e-05,
      "step": 29647,
      "training_step_time": 7.662735939025879
    },
    {
      "epoch": 2.82745361328125e-05,
      "model_forward_time": 1.7344629764556885,
      "step": 29648
    },
    {
      "epoch": 2.82745361328125e-05,
      "step": 29648,
      "training_step_time": 7.575817108154297
    },
    {
      "epoch": 2.8275489807128907e-05,
      "model_forward_time": 1.7787716388702393,
      "step": 29649
    },
    {
      "epoch": 2.8275489807128907e-05,
      "step": 29649,
      "training_step_time": 7.6143975257873535
    },
    {
      "epoch": 2.827644348144531e-05,
      "grad_norm": 0.17019058763980865,
      "learning_rate": 8.428877829770823e-06,
      "loss": 0.0586,
      "step": 29650
    },
    {
      "epoch": 2.827644348144531e-05,
      "model_forward_time": 1.7234196662902832,
      "step": 29650
    },
    {
      "dynamics_loss_avg": 0.046663338877260686,
      "epoch": 2.827644348144531e-05,
      "step": 29650
    },
    {
      "action_loss_avg": 0.0025805416342336686,
      "epoch": 2.827644348144531e-05,
      "step": 29650
    },
    {
      "epoch": 2.827644348144531e-05,
      "step": 29650,
      "training_step_time": 7.621499300003052
    },
    {
      "epoch": 2.827739715576172e-05,
      "model_forward_time": 1.7307589054107666,
      "step": 29651
    },
    {
      "epoch": 2.827739715576172e-05,
      "step": 29651,
      "training_step_time": 8.120794534683228
    },
    {
      "epoch": 2.8278350830078126e-05,
      "model_forward_time": 1.7272210121154785,
      "step": 29652
    },
    {
      "epoch": 2.8278350830078126e-05,
      "step": 29652,
      "training_step_time": 7.796379804611206
    },
    {
      "epoch": 2.827930450439453e-05,
      "model_forward_time": 1.7306509017944336,
      "step": 29653
    },
    {
      "epoch": 2.827930450439453e-05,
      "step": 29653,
      "training_step_time": 7.5664825439453125
    },
    {
      "epoch": 2.8280258178710937e-05,
      "model_forward_time": 1.7287960052490234,
      "step": 29654
    },
    {
      "epoch": 2.8280258178710937e-05,
      "step": 29654,
      "training_step_time": 7.608222007751465
    },
    {
      "epoch": 2.8281211853027345e-05,
      "model_forward_time": 1.3800017833709717,
      "step": 29655
    },
    {
      "epoch": 2.8281211853027345e-05,
      "step": 29655,
      "training_step_time": 7.647010803222656
    },
    {
      "epoch": 2.828216552734375e-05,
      "model_forward_time": 2.855191469192505,
      "step": 29656
    },
    {
      "epoch": 2.828216552734375e-05,
      "step": 29656,
      "training_step_time": 9.068462133407593
    },
    {
      "epoch": 2.8283119201660156e-05,
      "model_forward_time": 1.7422645092010498,
      "step": 29657
    },
    {
      "epoch": 2.8283119201660156e-05,
      "step": 29657,
      "training_step_time": 7.556897878646851
    },
    {
      "epoch": 2.8284072875976564e-05,
      "model_forward_time": 1.7329130172729492,
      "step": 29658
    },
    {
      "epoch": 2.8284072875976564e-05,
      "step": 29658,
      "training_step_time": 8.347527265548706
    },
    {
      "epoch": 2.8285026550292968e-05,
      "model_forward_time": 1.7341663837432861,
      "step": 29659
    },
    {
      "epoch": 2.8285026550292968e-05,
      "step": 29659,
      "training_step_time": 7.616934537887573
    },
    {
      "epoch": 2.8285980224609375e-05,
      "grad_norm": 0.1420113891363144,
      "learning_rate": 8.427674225350735e-06,
      "loss": 0.0592,
      "step": 29660
    },
    {
      "epoch": 2.8285980224609375e-05,
      "model_forward_time": 1.7392642498016357,
      "step": 29660
    },
    {
      "dynamics_loss_avg": 0.05319566857069731,
      "epoch": 2.8285980224609375e-05,
      "step": 29660
    },
    {
      "action_loss_avg": 0.0029300399852218106,
      "epoch": 2.8285980224609375e-05,
      "step": 29660
    },
    {
      "epoch": 2.8285980224609375e-05,
      "step": 29660,
      "training_step_time": 8.189087390899658
    },
    {
      "epoch": 2.8286933898925783e-05,
      "model_forward_time": 1.7320921421051025,
      "step": 29661
    },
    {
      "epoch": 2.8286933898925783e-05,
      "step": 29661,
      "training_step_time": 9.676577091217041
    },
    {
      "epoch": 2.8287887573242187e-05,
      "model_forward_time": 1.7390270233154297,
      "step": 29662
    },
    {
      "epoch": 2.8287887573242187e-05,
      "step": 29662,
      "training_step_time": 7.632514476776123
    },
    {
      "epoch": 2.8288841247558594e-05,
      "model_forward_time": 1.7318201065063477,
      "step": 29663
    },
    {
      "epoch": 2.8288841247558594e-05,
      "step": 29663,
      "training_step_time": 8.640064716339111
    },
    {
      "epoch": 2.8289794921875e-05,
      "model_forward_time": 1.742264986038208,
      "step": 29664
    },
    {
      "epoch": 2.8289794921875e-05,
      "step": 29664,
      "training_step_time": 7.738976001739502
    },
    {
      "epoch": 2.8290748596191405e-05,
      "model_forward_time": 1.733039140701294,
      "step": 29665
    },
    {
      "epoch": 2.8290748596191405e-05,
      "step": 29665,
      "training_step_time": 7.8349504470825195
    },
    {
      "epoch": 2.8291702270507813e-05,
      "model_forward_time": 2.8608250617980957,
      "step": 29666
    },
    {
      "epoch": 2.8291702270507813e-05,
      "step": 29666,
      "training_step_time": 7.6784985065460205
    },
    {
      "epoch": 2.829265594482422e-05,
      "model_forward_time": 1.348085880279541,
      "step": 29667
    },
    {
      "epoch": 2.829265594482422e-05,
      "step": 29667,
      "training_step_time": 7.721282482147217
    },
    {
      "epoch": 2.8293609619140624e-05,
      "model_forward_time": 1.7931413650512695,
      "step": 29668
    },
    {
      "epoch": 2.8293609619140624e-05,
      "step": 29668,
      "training_step_time": 7.680039167404175
    },
    {
      "epoch": 2.8294563293457032e-05,
      "model_forward_time": 1.7246711254119873,
      "step": 29669
    },
    {
      "epoch": 2.8294563293457032e-05,
      "step": 29669,
      "training_step_time": 7.640463829040527
    },
    {
      "epoch": 2.829551696777344e-05,
      "grad_norm": 0.11542894691228867,
      "learning_rate": 8.426470246085347e-06,
      "loss": 0.059,
      "step": 29670
    },
    {
      "epoch": 2.829551696777344e-05,
      "model_forward_time": 1.7268879413604736,
      "step": 29670
    },
    {
      "dynamics_loss_avg": 0.0702035741880536,
      "epoch": 2.829551696777344e-05,
      "step": 29670
    },
    {
      "action_loss_avg": 0.003370429371716455,
      "epoch": 2.829551696777344e-05,
      "step": 29670
    },
    {
      "epoch": 2.829551696777344e-05,
      "step": 29670,
      "training_step_time": 7.854552507400513
    },
    {
      "epoch": 2.8296470642089843e-05,
      "model_forward_time": 1.7238802909851074,
      "step": 29671
    },
    {
      "epoch": 2.8296470642089843e-05,
      "step": 29671,
      "training_step_time": 7.869932174682617
    },
    {
      "epoch": 2.829742431640625e-05,
      "model_forward_time": 1.7293500900268555,
      "step": 29672
    },
    {
      "epoch": 2.829742431640625e-05,
      "step": 29672,
      "training_step_time": 7.778414011001587
    },
    {
      "epoch": 2.8298377990722655e-05,
      "model_forward_time": 1.730437994003296,
      "step": 29673
    },
    {
      "epoch": 2.8298377990722655e-05,
      "step": 29673,
      "training_step_time": 9.46626615524292
    },
    {
      "epoch": 2.8299331665039062e-05,
      "model_forward_time": 1.7304139137268066,
      "step": 29674
    },
    {
      "epoch": 2.8299331665039062e-05,
      "step": 29674,
      "training_step_time": 7.876322269439697
    },
    {
      "epoch": 2.830028533935547e-05,
      "model_forward_time": 2.965496063232422,
      "step": 29675
    },
    {
      "epoch": 2.830028533935547e-05,
      "step": 29675,
      "training_step_time": 7.801529884338379
    },
    {
      "epoch": 2.8301239013671873e-05,
      "model_forward_time": 1.7444262504577637,
      "step": 29676
    },
    {
      "epoch": 2.8301239013671873e-05,
      "step": 29676,
      "training_step_time": 7.696086645126343
    },
    {
      "epoch": 2.830219268798828e-05,
      "model_forward_time": 1.7847130298614502,
      "step": 29677
    },
    {
      "epoch": 2.830219268798828e-05,
      "step": 29677,
      "training_step_time": 7.739642143249512
    },
    {
      "epoch": 2.8303146362304688e-05,
      "model_forward_time": 1.735548496246338,
      "step": 29678
    },
    {
      "epoch": 2.8303146362304688e-05,
      "step": 29678,
      "training_step_time": 8.222235202789307
    },
    {
      "epoch": 2.8304100036621092e-05,
      "model_forward_time": 1.7319304943084717,
      "step": 29679
    },
    {
      "epoch": 2.8304100036621092e-05,
      "step": 29679,
      "training_step_time": 8.762251377105713
    },
    {
      "epoch": 2.83050537109375e-05,
      "grad_norm": 0.10459334403276443,
      "learning_rate": 8.425265892106325e-06,
      "loss": 0.0552,
      "step": 29680
    },
    {
      "epoch": 2.83050537109375e-05,
      "model_forward_time": 1.7687809467315674,
      "step": 29680
    },
    {
      "dynamics_loss_avg": 0.055489514209330085,
      "epoch": 2.83050537109375e-05,
      "step": 29680
    },
    {
      "action_loss_avg": 0.0031779946642927826,
      "epoch": 2.83050537109375e-05,
      "step": 29680
    },
    {
      "epoch": 2.83050537109375e-05,
      "step": 29680,
      "training_step_time": 7.662474155426025
    },
    {
      "epoch": 2.8306007385253907e-05,
      "model_forward_time": 1.7366175651550293,
      "step": 29681
    },
    {
      "epoch": 2.8306007385253907e-05,
      "step": 29681,
      "training_step_time": 7.574344158172607
    },
    {
      "epoch": 2.830696105957031e-05,
      "model_forward_time": 1.7250785827636719,
      "step": 29682
    },
    {
      "epoch": 2.830696105957031e-05,
      "step": 29682,
      "training_step_time": 8.67466926574707
    },
    {
      "epoch": 2.830791473388672e-05,
      "model_forward_time": 1.7414145469665527,
      "step": 29683
    },
    {
      "epoch": 2.830791473388672e-05,
      "step": 29683,
      "training_step_time": 7.743695974349976
    },
    {
      "epoch": 2.8308868408203126e-05,
      "model_forward_time": 2.905700445175171,
      "step": 29684
    },
    {
      "epoch": 2.8308868408203126e-05,
      "step": 29684,
      "training_step_time": 7.647212743759155
    },
    {
      "epoch": 2.830982208251953e-05,
      "model_forward_time": 1.7325465679168701,
      "step": 29685
    },
    {
      "epoch": 2.830982208251953e-05,
      "step": 29685,
      "training_step_time": 7.6156251430511475
    },
    {
      "epoch": 2.8310775756835937e-05,
      "model_forward_time": 1.7938551902770996,
      "step": 29686
    },
    {
      "epoch": 2.8310775756835937e-05,
      "step": 29686,
      "training_step_time": 7.682287931442261
    },
    {
      "epoch": 2.8311729431152345e-05,
      "model_forward_time": 1.7363333702087402,
      "step": 29687
    },
    {
      "epoch": 2.8311729431152345e-05,
      "step": 29687,
      "training_step_time": 7.582131385803223
    },
    {
      "epoch": 2.831268310546875e-05,
      "model_forward_time": 1.733717679977417,
      "step": 29688
    },
    {
      "epoch": 2.831268310546875e-05,
      "step": 29688,
      "training_step_time": 8.358448505401611
    },
    {
      "epoch": 2.8313636779785156e-05,
      "model_forward_time": 1.7342545986175537,
      "step": 29689
    },
    {
      "epoch": 2.8313636779785156e-05,
      "step": 29689,
      "training_step_time": 7.64864706993103
    },
    {
      "epoch": 2.8314590454101564e-05,
      "grad_norm": 0.1721225529909134,
      "learning_rate": 8.424061163545375e-06,
      "loss": 0.0622,
      "step": 29690
    },
    {
      "epoch": 2.8314590454101564e-05,
      "model_forward_time": 2.3336548805236816,
      "step": 29690
    },
    {
      "dynamics_loss_avg": 0.05211301064118743,
      "epoch": 2.8314590454101564e-05,
      "step": 29690
    },
    {
      "action_loss_avg": 0.002137565557495691,
      "epoch": 2.8314590454101564e-05,
      "step": 29690
    },
    {
      "epoch": 2.8314590454101564e-05,
      "step": 29690,
      "training_step_time": 7.647233724594116
    },
    {
      "epoch": 2.8315544128417968e-05,
      "model_forward_time": 1.7287886142730713,
      "step": 29691
    },
    {
      "epoch": 2.8315544128417968e-05,
      "step": 29691,
      "training_step_time": 8.582847833633423
    },
    {
      "epoch": 2.8316497802734375e-05,
      "model_forward_time": 1.7364907264709473,
      "step": 29692
    },
    {
      "epoch": 2.8316497802734375e-05,
      "step": 29692,
      "training_step_time": 6.798476457595825
    },
    {
      "epoch": 2.8317451477050783e-05,
      "model_forward_time": 1.7313134670257568,
      "step": 29693
    },
    {
      "epoch": 2.8317451477050783e-05,
      "step": 29693,
      "training_step_time": 6.814582109451294
    },
    {
      "epoch": 2.8318405151367187e-05,
      "model_forward_time": 1.7296993732452393,
      "step": 29694
    },
    {
      "epoch": 2.8318405151367187e-05,
      "step": 29694,
      "training_step_time": 6.778288841247559
    },
    {
      "epoch": 2.8319358825683594e-05,
      "model_forward_time": 1.7303013801574707,
      "step": 29695
    },
    {
      "epoch": 2.8319358825683594e-05,
      "step": 29695,
      "training_step_time": 6.808711290359497
    },
    {
      "epoch": 2.83203125e-05,
      "model_forward_time": 1.7287673950195312,
      "step": 29696
    },
    {
      "epoch": 2.83203125e-05,
      "step": 29696,
      "training_step_time": 6.8059492111206055
    },
    {
      "epoch": 2.8321266174316405e-05,
      "model_forward_time": 1.7336430549621582,
      "step": 29697
    },
    {
      "epoch": 2.8321266174316405e-05,
      "step": 29697,
      "training_step_time": 6.798389196395874
    },
    {
      "epoch": 2.8322219848632813e-05,
      "model_forward_time": 1.7347452640533447,
      "step": 29698
    },
    {
      "epoch": 2.8322219848632813e-05,
      "step": 29698,
      "training_step_time": 6.795315504074097
    },
    {
      "epoch": 2.832317352294922e-05,
      "model_forward_time": 1.7377915382385254,
      "step": 29699
    },
    {
      "epoch": 2.832317352294922e-05,
      "step": 29699,
      "training_step_time": 6.776468515396118
    },
    {
      "epoch": 2.8324127197265624e-05,
      "grad_norm": 0.1523837000131607,
      "learning_rate": 8.422856060534243e-06,
      "loss": 0.0618,
      "step": 29700
    },
    {
      "epoch": 2.8324127197265624e-05,
      "model_forward_time": 2.832562208175659,
      "step": 29700
    },
    {
      "dynamics_loss_avg": 0.05908578038215637,
      "epoch": 2.8324127197265624e-05,
      "step": 29700
    },
    {
      "action_loss_avg": 0.0019508756522554904,
      "epoch": 2.8324127197265624e-05,
      "step": 29700
    },
    {
      "epoch": 2.8324127197265624e-05,
      "step": 29700,
      "training_step_time": 10.557013988494873
    },
    {
      "epoch": 2.8325080871582032e-05,
      "model_forward_time": 1.7907981872558594,
      "step": 29701
    },
    {
      "epoch": 2.8325080871582032e-05,
      "step": 29701,
      "training_step_time": 8.397341966629028
    },
    {
      "epoch": 2.832603454589844e-05,
      "model_forward_time": 1.737135648727417,
      "step": 29702
    },
    {
      "epoch": 2.832603454589844e-05,
      "step": 29702,
      "training_step_time": 6.808578729629517
    },
    {
      "epoch": 2.8326988220214843e-05,
      "model_forward_time": 1.7367453575134277,
      "step": 29703
    },
    {
      "epoch": 2.8326988220214843e-05,
      "step": 29703,
      "training_step_time": 6.813209772109985
    },
    {
      "epoch": 2.832794189453125e-05,
      "model_forward_time": 1.7289419174194336,
      "step": 29704
    },
    {
      "epoch": 2.832794189453125e-05,
      "step": 29704,
      "training_step_time": 6.8051910400390625
    },
    {
      "epoch": 2.8328895568847655e-05,
      "model_forward_time": 1.7392852306365967,
      "step": 29705
    },
    {
      "epoch": 2.8328895568847655e-05,
      "step": 29705,
      "training_step_time": 6.843278646469116
    },
    {
      "epoch": 2.8329849243164062e-05,
      "model_forward_time": 1.739386796951294,
      "step": 29706
    },
    {
      "epoch": 2.8329849243164062e-05,
      "step": 29706,
      "training_step_time": 6.8142125606536865
    },
    {
      "epoch": 2.833080291748047e-05,
      "model_forward_time": 1.7340586185455322,
      "step": 29707
    },
    {
      "epoch": 2.833080291748047e-05,
      "step": 29707,
      "training_step_time": 6.766205072402954
    },
    {
      "epoch": 2.8331756591796873e-05,
      "model_forward_time": 1.7235081195831299,
      "step": 29708
    },
    {
      "epoch": 2.8331756591796873e-05,
      "step": 29708,
      "training_step_time": 6.790848016738892
    },
    {
      "epoch": 2.833271026611328e-05,
      "model_forward_time": 2.8755693435668945,
      "step": 29709
    },
    {
      "epoch": 2.833271026611328e-05,
      "step": 29709,
      "training_step_time": 8.43022632598877
    },
    {
      "epoch": 2.833366394042969e-05,
      "grad_norm": 0.2130202203989029,
      "learning_rate": 8.42165058320472e-06,
      "loss": 0.0599,
      "step": 29710
    },
    {
      "epoch": 2.833366394042969e-05,
      "model_forward_time": 1.7689917087554932,
      "step": 29710
    },
    {
      "dynamics_loss_avg": 0.08021412584930658,
      "epoch": 2.833366394042969e-05,
      "step": 29710
    },
    {
      "action_loss_avg": 0.0029392266413196923,
      "epoch": 2.833366394042969e-05,
      "step": 29710
    },
    {
      "epoch": 2.833366394042969e-05,
      "step": 29710,
      "training_step_time": 7.82484245300293
    },
    {
      "epoch": 2.8334617614746092e-05,
      "model_forward_time": 1.7221641540527344,
      "step": 29711
    },
    {
      "epoch": 2.8334617614746092e-05,
      "step": 29711,
      "training_step_time": 7.126293420791626
    },
    {
      "epoch": 2.83355712890625e-05,
      "model_forward_time": 1.7231390476226807,
      "step": 29712
    },
    {
      "epoch": 2.83355712890625e-05,
      "step": 29712,
      "training_step_time": 6.806821823120117
    },
    {
      "epoch": 2.8336524963378907e-05,
      "model_forward_time": 1.7277541160583496,
      "step": 29713
    },
    {
      "epoch": 2.8336524963378907e-05,
      "step": 29713,
      "training_step_time": 6.802170515060425
    },
    {
      "epoch": 2.833747863769531e-05,
      "model_forward_time": 1.7351574897766113,
      "step": 29714
    },
    {
      "epoch": 2.833747863769531e-05,
      "step": 29714,
      "training_step_time": 6.78763222694397
    },
    {
      "epoch": 2.833843231201172e-05,
      "model_forward_time": 1.7296092510223389,
      "step": 29715
    },
    {
      "epoch": 2.833843231201172e-05,
      "step": 29715,
      "training_step_time": 6.747233152389526
    },
    {
      "epoch": 2.8339385986328126e-05,
      "model_forward_time": 1.733485460281372,
      "step": 29716
    },
    {
      "epoch": 2.8339385986328126e-05,
      "step": 29716,
      "training_step_time": 6.793585538864136
    },
    {
      "epoch": 2.834033966064453e-05,
      "model_forward_time": 1.7297492027282715,
      "step": 29717
    },
    {
      "epoch": 2.834033966064453e-05,
      "step": 29717,
      "training_step_time": 6.827006578445435
    },
    {
      "epoch": 2.8341293334960937e-05,
      "model_forward_time": 2.944443464279175,
      "step": 29718
    },
    {
      "epoch": 2.8341293334960937e-05,
      "step": 29718,
      "training_step_time": 7.776083707809448
    },
    {
      "epoch": 2.8342247009277345e-05,
      "model_forward_time": 1.7362351417541504,
      "step": 29719
    },
    {
      "epoch": 2.8342247009277345e-05,
      "step": 29719,
      "training_step_time": 8.915381669998169
    },
    {
      "epoch": 2.834320068359375e-05,
      "grad_norm": 0.16904693841934204,
      "learning_rate": 8.420444731688632e-06,
      "loss": 0.0595,
      "step": 29720
    },
    {
      "epoch": 2.834320068359375e-05,
      "model_forward_time": 1.7224225997924805,
      "step": 29720
    },
    {
      "dynamics_loss_avg": 0.04294368224218488,
      "epoch": 2.834320068359375e-05,
      "step": 29720
    },
    {
      "action_loss_avg": 0.0015419404371641576,
      "epoch": 2.834320068359375e-05,
      "step": 29720
    },
    {
      "epoch": 2.834320068359375e-05,
      "step": 29720,
      "training_step_time": 7.580102920532227
    },
    {
      "epoch": 2.8344154357910156e-05,
      "model_forward_time": 1.727917194366455,
      "step": 29721
    },
    {
      "epoch": 2.8344154357910156e-05,
      "step": 29721,
      "training_step_time": 6.835465669631958
    },
    {
      "epoch": 2.8345108032226564e-05,
      "model_forward_time": 1.7353742122650146,
      "step": 29722
    },
    {
      "epoch": 2.8345108032226564e-05,
      "step": 29722,
      "training_step_time": 6.791460990905762
    },
    {
      "epoch": 2.8346061706542968e-05,
      "model_forward_time": 1.7270698547363281,
      "step": 29723
    },
    {
      "epoch": 2.8346061706542968e-05,
      "step": 29723,
      "training_step_time": 6.761925458908081
    },
    {
      "epoch": 2.8347015380859375e-05,
      "model_forward_time": 1.7297513484954834,
      "step": 29724
    },
    {
      "epoch": 2.8347015380859375e-05,
      "step": 29724,
      "training_step_time": 6.8013386726379395
    },
    {
      "epoch": 2.8347969055175783e-05,
      "model_forward_time": 1.7260277271270752,
      "step": 29725
    },
    {
      "epoch": 2.8347969055175783e-05,
      "step": 29725,
      "training_step_time": 6.794588327407837
    },
    {
      "epoch": 2.8348922729492187e-05,
      "model_forward_time": 1.7987947463989258,
      "step": 29726
    },
    {
      "epoch": 2.8348922729492187e-05,
      "step": 29726,
      "training_step_time": 6.803833246231079
    },
    {
      "epoch": 2.8349876403808594e-05,
      "model_forward_time": 2.82999849319458,
      "step": 29727
    },
    {
      "epoch": 2.8349876403808594e-05,
      "step": 29727,
      "training_step_time": 7.818143844604492
    },
    {
      "epoch": 2.8350830078125e-05,
      "model_forward_time": 1.8274803161621094,
      "step": 29728
    },
    {
      "epoch": 2.8350830078125e-05,
      "step": 29728,
      "training_step_time": 9.238517045974731
    },
    {
      "epoch": 2.8351783752441405e-05,
      "model_forward_time": 1.7196102142333984,
      "step": 29729
    },
    {
      "epoch": 2.8351783752441405e-05,
      "step": 29729,
      "training_step_time": 7.733854532241821
    },
    {
      "epoch": 2.8352737426757813e-05,
      "grad_norm": 0.16740524768829346,
      "learning_rate": 8.419238506117852e-06,
      "loss": 0.0611,
      "step": 29730
    },
    {
      "epoch": 2.8352737426757813e-05,
      "model_forward_time": 1.7319622039794922,
      "step": 29730
    },
    {
      "dynamics_loss_avg": 0.051671336963772775,
      "epoch": 2.8352737426757813e-05,
      "step": 29730
    },
    {
      "action_loss_avg": 0.001934076426550746,
      "epoch": 2.8352737426757813e-05,
      "step": 29730
    },
    {
      "epoch": 2.8352737426757813e-05,
      "step": 29730,
      "training_step_time": 6.789184331893921
    },
    {
      "epoch": 2.835369110107422e-05,
      "model_forward_time": 1.7563128471374512,
      "step": 29731
    },
    {
      "epoch": 2.835369110107422e-05,
      "step": 29731,
      "training_step_time": 6.809210300445557
    },
    {
      "epoch": 2.8354644775390624e-05,
      "model_forward_time": 1.7146427631378174,
      "step": 29732
    },
    {
      "epoch": 2.8354644775390624e-05,
      "step": 29732,
      "training_step_time": 6.810967445373535
    },
    {
      "epoch": 2.8355598449707032e-05,
      "model_forward_time": 1.724073886871338,
      "step": 29733
    },
    {
      "epoch": 2.8355598449707032e-05,
      "step": 29733,
      "training_step_time": 6.7913124561309814
    },
    {
      "epoch": 2.835655212402344e-05,
      "model_forward_time": 1.726093053817749,
      "step": 29734
    },
    {
      "epoch": 2.835655212402344e-05,
      "step": 29734,
      "training_step_time": 6.882552862167358
    },
    {
      "epoch": 2.8357505798339843e-05,
      "model_forward_time": 1.7366352081298828,
      "step": 29735
    },
    {
      "epoch": 2.8357505798339843e-05,
      "step": 29735,
      "training_step_time": 6.8232643604278564
    },
    {
      "epoch": 2.835845947265625e-05,
      "model_forward_time": 2.8081650733947754,
      "step": 29736
    },
    {
      "epoch": 2.835845947265625e-05,
      "step": 29736,
      "training_step_time": 7.769970178604126
    },
    {
      "epoch": 2.8359413146972655e-05,
      "model_forward_time": 1.7652819156646729,
      "step": 29737
    },
    {
      "epoch": 2.8359413146972655e-05,
      "step": 29737,
      "training_step_time": 12.373091697692871
    },
    {
      "epoch": 2.8360366821289062e-05,
      "model_forward_time": 1.721153974533081,
      "step": 29738
    },
    {
      "epoch": 2.8360366821289062e-05,
      "step": 29738,
      "training_step_time": 7.819573640823364
    },
    {
      "epoch": 2.836132049560547e-05,
      "model_forward_time": 1.731971025466919,
      "step": 29739
    },
    {
      "epoch": 2.836132049560547e-05,
      "step": 29739,
      "training_step_time": 7.194998264312744
    },
    {
      "epoch": 2.8362274169921874e-05,
      "grad_norm": 0.16936735808849335,
      "learning_rate": 8.41803190662429e-06,
      "loss": 0.0613,
      "step": 29740
    },
    {
      "epoch": 2.8362274169921874e-05,
      "model_forward_time": 1.728395938873291,
      "step": 29740
    },
    {
      "dynamics_loss_avg": 0.06021592728793621,
      "epoch": 2.8362274169921874e-05,
      "step": 29740
    },
    {
      "action_loss_avg": 0.002087508258409798,
      "epoch": 2.8362274169921874e-05,
      "step": 29740
    },
    {
      "epoch": 2.8362274169921874e-05,
      "step": 29740,
      "training_step_time": 6.766845226287842
    },
    {
      "epoch": 2.836322784423828e-05,
      "model_forward_time": 1.7215962409973145,
      "step": 29741
    },
    {
      "epoch": 2.836322784423828e-05,
      "step": 29741,
      "training_step_time": 6.80739688873291
    },
    {
      "epoch": 2.836418151855469e-05,
      "model_forward_time": 1.7269043922424316,
      "step": 29742
    },
    {
      "epoch": 2.836418151855469e-05,
      "step": 29742,
      "training_step_time": 6.8029844760894775
    },
    {
      "epoch": 2.8365135192871092e-05,
      "model_forward_time": 1.720853567123413,
      "step": 29743
    },
    {
      "epoch": 2.8365135192871092e-05,
      "step": 29743,
      "training_step_time": 6.81168794631958
    },
    {
      "epoch": 2.83660888671875e-05,
      "model_forward_time": 1.7318108081817627,
      "step": 29744
    },
    {
      "epoch": 2.83660888671875e-05,
      "step": 29744,
      "training_step_time": 6.89131760597229
    },
    {
      "epoch": 2.8367042541503907e-05,
      "model_forward_time": 1.7305290699005127,
      "step": 29745
    },
    {
      "epoch": 2.8367042541503907e-05,
      "step": 29745,
      "training_step_time": 7.792139291763306
    },
    {
      "epoch": 2.836799621582031e-05,
      "model_forward_time": 2.9363138675689697,
      "step": 29746
    },
    {
      "epoch": 2.836799621582031e-05,
      "step": 29746,
      "training_step_time": 9.332858324050903
    },
    {
      "epoch": 2.836894989013672e-05,
      "model_forward_time": 1.7385399341583252,
      "step": 29747
    },
    {
      "epoch": 2.836894989013672e-05,
      "step": 29747,
      "training_step_time": 7.87946629524231
    },
    {
      "epoch": 2.8369903564453126e-05,
      "model_forward_time": 1.7239773273468018,
      "step": 29748
    },
    {
      "epoch": 2.8369903564453126e-05,
      "step": 29748,
      "training_step_time": 7.627842903137207
    },
    {
      "epoch": 2.837085723876953e-05,
      "model_forward_time": 1.7211532592773438,
      "step": 29749
    },
    {
      "epoch": 2.837085723876953e-05,
      "step": 29749,
      "training_step_time": 6.869868516921997
    },
    {
      "epoch": 2.8371810913085938e-05,
      "grad_norm": 0.17397540807724,
      "learning_rate": 8.416824933339898e-06,
      "loss": 0.0572,
      "step": 29750
    },
    {
      "epoch": 2.8371810913085938e-05,
      "model_forward_time": 1.7243549823760986,
      "step": 29750
    },
    {
      "dynamics_loss_avg": 0.04773691762238741,
      "epoch": 2.8371810913085938e-05,
      "step": 29750
    },
    {
      "action_loss_avg": 0.001743779529351741,
      "epoch": 2.8371810913085938e-05,
      "step": 29750
    },
    {
      "epoch": 2.8371810913085938e-05,
      "step": 29750,
      "training_step_time": 6.81547474861145
    },
    {
      "epoch": 2.8372764587402345e-05,
      "model_forward_time": 1.7333853244781494,
      "step": 29751
    },
    {
      "epoch": 2.8372764587402345e-05,
      "step": 29751,
      "training_step_time": 6.831228733062744
    },
    {
      "epoch": 2.837371826171875e-05,
      "model_forward_time": 1.730774164199829,
      "step": 29752
    },
    {
      "epoch": 2.837371826171875e-05,
      "step": 29752,
      "training_step_time": 6.85708475112915
    },
    {
      "epoch": 2.8374671936035156e-05,
      "model_forward_time": 1.7285432815551758,
      "step": 29753
    },
    {
      "epoch": 2.8374671936035156e-05,
      "step": 29753,
      "training_step_time": 6.821689128875732
    },
    {
      "epoch": 2.8375625610351564e-05,
      "model_forward_time": 1.722355604171753,
      "step": 29754
    },
    {
      "epoch": 2.8375625610351564e-05,
      "step": 29754,
      "training_step_time": 7.733999490737915
    },
    {
      "epoch": 2.8376579284667968e-05,
      "model_forward_time": 2.9708824157714844,
      "step": 29755
    },
    {
      "epoch": 2.8376579284667968e-05,
      "step": 29755,
      "training_step_time": 9.371016263961792
    },
    {
      "epoch": 2.8377532958984375e-05,
      "model_forward_time": 1.8401257991790771,
      "step": 29756
    },
    {
      "epoch": 2.8377532958984375e-05,
      "step": 29756,
      "training_step_time": 7.926882266998291
    },
    {
      "epoch": 2.8378486633300783e-05,
      "model_forward_time": 1.718630075454712,
      "step": 29757
    },
    {
      "epoch": 2.8378486633300783e-05,
      "step": 29757,
      "training_step_time": 7.635567903518677
    },
    {
      "epoch": 2.8379440307617187e-05,
      "model_forward_time": 1.732621431350708,
      "step": 29758
    },
    {
      "epoch": 2.8379440307617187e-05,
      "step": 29758,
      "training_step_time": 6.81879734992981
    },
    {
      "epoch": 2.8380393981933594e-05,
      "model_forward_time": 1.7288672924041748,
      "step": 29759
    },
    {
      "epoch": 2.8380393981933594e-05,
      "step": 29759,
      "training_step_time": 6.816856861114502
    },
    {
      "epoch": 2.838134765625e-05,
      "grad_norm": 0.17371635138988495,
      "learning_rate": 8.415617586396667e-06,
      "loss": 0.0551,
      "step": 29760
    },
    {
      "epoch": 2.838134765625e-05,
      "model_forward_time": 1.7283685207366943,
      "step": 29760
    },
    {
      "dynamics_loss_avg": 0.06609180215746165,
      "epoch": 2.838134765625e-05,
      "step": 29760
    },
    {
      "action_loss_avg": 0.0022576696530450134,
      "epoch": 2.838134765625e-05,
      "step": 29760
    },
    {
      "epoch": 2.838134765625e-05,
      "step": 29760,
      "training_step_time": 6.866483211517334
    },
    {
      "epoch": 2.8382301330566406e-05,
      "model_forward_time": 1.7274723052978516,
      "step": 29761
    },
    {
      "epoch": 2.8382301330566406e-05,
      "step": 29761,
      "training_step_time": 6.825803995132446
    },
    {
      "epoch": 2.8383255004882813e-05,
      "model_forward_time": 1.7207224369049072,
      "step": 29762
    },
    {
      "epoch": 2.8383255004882813e-05,
      "step": 29762,
      "training_step_time": 6.758094787597656
    },
    {
      "epoch": 2.838420867919922e-05,
      "model_forward_time": 1.7306437492370605,
      "step": 29763
    },
    {
      "epoch": 2.838420867919922e-05,
      "step": 29763,
      "training_step_time": 7.6390321254730225
    },
    {
      "epoch": 2.8385162353515624e-05,
      "model_forward_time": 2.8771378993988037,
      "step": 29764
    },
    {
      "epoch": 2.8385162353515624e-05,
      "step": 29764,
      "training_step_time": 8.474661350250244
    },
    {
      "epoch": 2.8386116027832032e-05,
      "model_forward_time": 1.8259365558624268,
      "step": 29765
    },
    {
      "epoch": 2.8386116027832032e-05,
      "step": 29765,
      "training_step_time": 9.214237451553345
    },
    {
      "epoch": 2.8387069702148436e-05,
      "model_forward_time": 1.7560489177703857,
      "step": 29766
    },
    {
      "epoch": 2.8387069702148436e-05,
      "step": 29766,
      "training_step_time": 7.71487021446228
    },
    {
      "epoch": 2.8388023376464843e-05,
      "model_forward_time": 1.7611134052276611,
      "step": 29767
    },
    {
      "epoch": 2.8388023376464843e-05,
      "step": 29767,
      "training_step_time": 6.84852933883667
    },
    {
      "epoch": 2.838897705078125e-05,
      "model_forward_time": 1.357133388519287,
      "step": 29768
    },
    {
      "epoch": 2.838897705078125e-05,
      "step": 29768,
      "training_step_time": 6.861493349075317
    },
    {
      "epoch": 2.8389930725097655e-05,
      "model_forward_time": 1.7353532314300537,
      "step": 29769
    },
    {
      "epoch": 2.8389930725097655e-05,
      "step": 29769,
      "training_step_time": 6.789097547531128
    },
    {
      "epoch": 2.8390884399414062e-05,
      "grad_norm": 0.13605865836143494,
      "learning_rate": 8.414409865926632e-06,
      "loss": 0.0555,
      "step": 29770
    },
    {
      "epoch": 2.8390884399414062e-05,
      "model_forward_time": 1.770369529724121,
      "step": 29770
    },
    {
      "dynamics_loss_avg": 0.05600575637072325,
      "epoch": 2.8390884399414062e-05,
      "step": 29770
    },
    {
      "action_loss_avg": 0.0024460827524308117,
      "epoch": 2.8390884399414062e-05,
      "step": 29770
    },
    {
      "epoch": 2.8390884399414062e-05,
      "step": 29770,
      "training_step_time": 6.79072642326355
    },
    {
      "epoch": 2.839183807373047e-05,
      "model_forward_time": 1.730109691619873,
      "step": 29771
    },
    {
      "epoch": 2.839183807373047e-05,
      "step": 29771,
      "training_step_time": 6.807722806930542
    },
    {
      "epoch": 2.8392791748046874e-05,
      "model_forward_time": 1.7245557308197021,
      "step": 29772
    },
    {
      "epoch": 2.8392791748046874e-05,
      "step": 29772,
      "training_step_time": 6.8638014793396
    },
    {
      "epoch": 2.839374542236328e-05,
      "model_forward_time": 2.8491358757019043,
      "step": 29773
    },
    {
      "epoch": 2.839374542236328e-05,
      "step": 29773,
      "training_step_time": 9.1669340133667
    },
    {
      "epoch": 2.839469909667969e-05,
      "model_forward_time": 1.789576768875122,
      "step": 29774
    },
    {
      "epoch": 2.839469909667969e-05,
      "step": 29774,
      "training_step_time": 9.062436580657959
    },
    {
      "epoch": 2.8395652770996092e-05,
      "model_forward_time": 1.7187814712524414,
      "step": 29775
    },
    {
      "epoch": 2.8395652770996092e-05,
      "step": 29775,
      "training_step_time": 8.336377620697021
    },
    {
      "epoch": 2.83966064453125e-05,
      "model_forward_time": 1.7152907848358154,
      "step": 29776
    },
    {
      "epoch": 2.83966064453125e-05,
      "step": 29776,
      "training_step_time": 7.522540330886841
    },
    {
      "epoch": 2.8397560119628907e-05,
      "model_forward_time": 1.7236425876617432,
      "step": 29777
    },
    {
      "epoch": 2.8397560119628907e-05,
      "step": 29777,
      "training_step_time": 6.819642543792725
    },
    {
      "epoch": 2.839851379394531e-05,
      "model_forward_time": 1.7270424365997314,
      "step": 29778
    },
    {
      "epoch": 2.839851379394531e-05,
      "step": 29778,
      "training_step_time": 6.837764024734497
    },
    {
      "epoch": 2.839946746826172e-05,
      "model_forward_time": 1.72475266456604,
      "step": 29779
    },
    {
      "epoch": 2.839946746826172e-05,
      "step": 29779,
      "training_step_time": 6.823164701461792
    },
    {
      "epoch": 2.8400421142578126e-05,
      "grad_norm": 0.12559698522090912,
      "learning_rate": 8.413201772061867e-06,
      "loss": 0.0566,
      "step": 29780
    },
    {
      "epoch": 2.8400421142578126e-05,
      "model_forward_time": 1.734609603881836,
      "step": 29780
    },
    {
      "dynamics_loss_avg": 0.05209017153829336,
      "epoch": 2.8400421142578126e-05,
      "step": 29780
    },
    {
      "action_loss_avg": 0.0016064321578596718,
      "epoch": 2.8400421142578126e-05,
      "step": 29780
    },
    {
      "epoch": 2.8400421142578126e-05,
      "step": 29780,
      "training_step_time": 6.841149806976318
    },
    {
      "epoch": 2.840137481689453e-05,
      "model_forward_time": 1.7213068008422852,
      "step": 29781
    },
    {
      "epoch": 2.840137481689453e-05,
      "step": 29781,
      "training_step_time": 6.84323263168335
    },
    {
      "epoch": 2.8402328491210938e-05,
      "model_forward_time": 1.7210693359375,
      "step": 29782
    },
    {
      "epoch": 2.8402328491210938e-05,
      "step": 29782,
      "training_step_time": 8.82474946975708
    },
    {
      "epoch": 2.8403282165527345e-05,
      "model_forward_time": 1.747765064239502,
      "step": 29783
    },
    {
      "epoch": 2.8403282165527345e-05,
      "step": 29783,
      "training_step_time": 9.782551527023315
    },
    {
      "epoch": 2.840423583984375e-05,
      "model_forward_time": 1.7219555377960205,
      "step": 29784
    },
    {
      "epoch": 2.840423583984375e-05,
      "step": 29784,
      "training_step_time": 7.772649049758911
    },
    {
      "epoch": 2.8405189514160156e-05,
      "model_forward_time": 1.7290441989898682,
      "step": 29785
    },
    {
      "epoch": 2.8405189514160156e-05,
      "step": 29785,
      "training_step_time": 7.724414110183716
    },
    {
      "epoch": 2.8406143188476564e-05,
      "model_forward_time": 1.726820945739746,
      "step": 29786
    },
    {
      "epoch": 2.8406143188476564e-05,
      "step": 29786,
      "training_step_time": 6.871448993682861
    },
    {
      "epoch": 2.8407096862792968e-05,
      "model_forward_time": 1.7427325248718262,
      "step": 29787
    },
    {
      "epoch": 2.8407096862792968e-05,
      "step": 29787,
      "training_step_time": 6.858670473098755
    },
    {
      "epoch": 2.8408050537109375e-05,
      "model_forward_time": 1.735748052597046,
      "step": 29788
    },
    {
      "epoch": 2.8408050537109375e-05,
      "step": 29788,
      "training_step_time": 6.799248933792114
    },
    {
      "epoch": 2.8409004211425783e-05,
      "model_forward_time": 1.7221741676330566,
      "step": 29789
    },
    {
      "epoch": 2.8409004211425783e-05,
      "step": 29789,
      "training_step_time": 6.875396490097046
    },
    {
      "epoch": 2.8409957885742187e-05,
      "grad_norm": 0.10410703718662262,
      "learning_rate": 8.41199330493449e-06,
      "loss": 0.0582,
      "step": 29790
    },
    {
      "epoch": 2.8409957885742187e-05,
      "model_forward_time": 1.7333619594573975,
      "step": 29790
    },
    {
      "dynamics_loss_avg": 0.06113616097718477,
      "epoch": 2.8409957885742187e-05,
      "step": 29790
    },
    {
      "action_loss_avg": 0.0026792044402100144,
      "epoch": 2.8409957885742187e-05,
      "step": 29790
    },
    {
      "epoch": 2.8409957885742187e-05,
      "step": 29790,
      "training_step_time": 6.901430130004883
    },
    {
      "epoch": 2.8410911560058594e-05,
      "model_forward_time": 1.7410647869110107,
      "step": 29791
    },
    {
      "epoch": 2.8410911560058594e-05,
      "step": 29791,
      "training_step_time": 8.610467195510864
    },
    {
      "epoch": 2.8411865234375e-05,
      "model_forward_time": 2.2860774993896484,
      "step": 29792
    },
    {
      "epoch": 2.8411865234375e-05,
      "step": 29792,
      "training_step_time": 8.711090087890625
    },
    {
      "epoch": 2.8412818908691406e-05,
      "model_forward_time": 1.7800421714782715,
      "step": 29793
    },
    {
      "epoch": 2.8412818908691406e-05,
      "step": 29793,
      "training_step_time": 8.708219051361084
    },
    {
      "epoch": 2.8413772583007813e-05,
      "model_forward_time": 1.722743034362793,
      "step": 29794
    },
    {
      "epoch": 2.8413772583007813e-05,
      "step": 29794,
      "training_step_time": 7.723412990570068
    },
    {
      "epoch": 2.841472625732422e-05,
      "model_forward_time": 1.7283556461334229,
      "step": 29795
    },
    {
      "epoch": 2.841472625732422e-05,
      "step": 29795,
      "training_step_time": 6.817576885223389
    },
    {
      "epoch": 2.8415679931640624e-05,
      "model_forward_time": 1.725590705871582,
      "step": 29796
    },
    {
      "epoch": 2.8415679931640624e-05,
      "step": 29796,
      "training_step_time": 6.8128886222839355
    },
    {
      "epoch": 2.8416633605957032e-05,
      "model_forward_time": 1.721726894378662,
      "step": 29797
    },
    {
      "epoch": 2.8416633605957032e-05,
      "step": 29797,
      "training_step_time": 6.832069158554077
    },
    {
      "epoch": 2.8417587280273436e-05,
      "model_forward_time": 1.7173008918762207,
      "step": 29798
    },
    {
      "epoch": 2.8417587280273436e-05,
      "step": 29798,
      "training_step_time": 6.7503931522369385
    },
    {
      "epoch": 2.8418540954589843e-05,
      "model_forward_time": 1.722747564315796,
      "step": 29799
    },
    {
      "epoch": 2.8418540954589843e-05,
      "step": 29799,
      "training_step_time": 6.806317329406738
    },
    {
      "epoch": 2.841949462890625e-05,
      "grad_norm": 0.1099306270480156,
      "learning_rate": 8.410784464676653e-06,
      "loss": 0.0553,
      "step": 29800
    },
    {
      "epoch": 2.841949462890625e-05,
      "model_forward_time": 1.3698267936706543,
      "step": 29800
    },
    {
      "dynamics_loss_avg": 0.05781285334378481,
      "epoch": 2.841949462890625e-05,
      "step": 29800
    },
    {
      "action_loss_avg": 0.002414391393540427,
      "epoch": 2.841949462890625e-05,
      "step": 29800
    },
    {
      "epoch": 2.841949462890625e-05,
      "step": 29800,
      "training_step_time": 8.285957098007202
    },
    {
      "epoch": 2.8420448303222655e-05,
      "model_forward_time": 2.2730555534362793,
      "step": 29801
    },
    {
      "epoch": 2.8420448303222655e-05,
      "step": 29801,
      "training_step_time": 8.458161115646362
    },
    {
      "epoch": 2.8421401977539062e-05,
      "model_forward_time": 1.7539117336273193,
      "step": 29802
    },
    {
      "epoch": 2.8421401977539062e-05,
      "step": 29802,
      "training_step_time": 8.781841278076172
    },
    {
      "epoch": 2.842235565185547e-05,
      "model_forward_time": 1.7248830795288086,
      "step": 29803
    },
    {
      "epoch": 2.842235565185547e-05,
      "step": 29803,
      "training_step_time": 8.40989875793457
    },
    {
      "epoch": 2.8423309326171874e-05,
      "model_forward_time": 1.7209663391113281,
      "step": 29804
    },
    {
      "epoch": 2.8423309326171874e-05,
      "step": 29804,
      "training_step_time": 6.791745662689209
    },
    {
      "epoch": 2.842426300048828e-05,
      "model_forward_time": 1.7155060768127441,
      "step": 29805
    },
    {
      "epoch": 2.842426300048828e-05,
      "step": 29805,
      "training_step_time": 6.823639392852783
    },
    {
      "epoch": 2.842521667480469e-05,
      "model_forward_time": 1.7295985221862793,
      "step": 29806
    },
    {
      "epoch": 2.842521667480469e-05,
      "step": 29806,
      "training_step_time": 6.8033881187438965
    },
    {
      "epoch": 2.8426170349121092e-05,
      "model_forward_time": 1.026984691619873,
      "step": 29807
    },
    {
      "epoch": 2.8426170349121092e-05,
      "step": 29807,
      "training_step_time": 6.808583736419678
    },
    {
      "epoch": 2.84271240234375e-05,
      "model_forward_time": 1.722233772277832,
      "step": 29808
    },
    {
      "epoch": 2.84271240234375e-05,
      "step": 29808,
      "training_step_time": 6.792839527130127
    },
    {
      "epoch": 2.8428077697753907e-05,
      "model_forward_time": 1.7383599281311035,
      "step": 29809
    },
    {
      "epoch": 2.8428077697753907e-05,
      "step": 29809,
      "training_step_time": 7.534291982650757
    },
    {
      "epoch": 2.842903137207031e-05,
      "grad_norm": 0.14803996682167053,
      "learning_rate": 8.409575251420556e-06,
      "loss": 0.0568,
      "step": 29810
    },
    {
      "epoch": 2.842903137207031e-05,
      "model_forward_time": 1.728562593460083,
      "step": 29810
    },
    {
      "dynamics_loss_avg": 0.04646044271066785,
      "epoch": 2.842903137207031e-05,
      "step": 29810
    },
    {
      "action_loss_avg": 0.0019522219066857361,
      "epoch": 2.842903137207031e-05,
      "step": 29810
    },
    {
      "epoch": 2.842903137207031e-05,
      "step": 29810,
      "training_step_time": 9.156244039535522
    },
    {
      "epoch": 2.842998504638672e-05,
      "model_forward_time": 1.7235052585601807,
      "step": 29811
    },
    {
      "epoch": 2.842998504638672e-05,
      "step": 29811,
      "training_step_time": 9.685734987258911
    },
    {
      "epoch": 2.8430938720703126e-05,
      "model_forward_time": 1.719282865524292,
      "step": 29812
    },
    {
      "epoch": 2.8430938720703126e-05,
      "step": 29812,
      "training_step_time": 7.83551812171936
    },
    {
      "epoch": 2.843189239501953e-05,
      "model_forward_time": 1.7260775566101074,
      "step": 29813
    },
    {
      "epoch": 2.843189239501953e-05,
      "step": 29813,
      "training_step_time": 7.602019548416138
    },
    {
      "epoch": 2.8432846069335938e-05,
      "model_forward_time": 1.7191600799560547,
      "step": 29814
    },
    {
      "epoch": 2.8432846069335938e-05,
      "step": 29814,
      "training_step_time": 6.867604494094849
    },
    {
      "epoch": 2.8433799743652345e-05,
      "model_forward_time": 1.7300238609313965,
      "step": 29815
    },
    {
      "epoch": 2.8433799743652345e-05,
      "step": 29815,
      "training_step_time": 6.803088426589966
    },
    {
      "epoch": 2.843475341796875e-05,
      "model_forward_time": 1.7116692066192627,
      "step": 29816
    },
    {
      "epoch": 2.843475341796875e-05,
      "step": 29816,
      "training_step_time": 6.761080026626587
    },
    {
      "epoch": 2.8435707092285156e-05,
      "model_forward_time": 1.7262604236602783,
      "step": 29817
    },
    {
      "epoch": 2.8435707092285156e-05,
      "step": 29817,
      "training_step_time": 6.819153308868408
    },
    {
      "epoch": 2.8436660766601564e-05,
      "model_forward_time": 1.724196195602417,
      "step": 29818
    },
    {
      "epoch": 2.8436660766601564e-05,
      "step": 29818,
      "training_step_time": 6.833921670913696
    },
    {
      "epoch": 2.8437614440917968e-05,
      "model_forward_time": 1.7216448783874512,
      "step": 29819
    },
    {
      "epoch": 2.8437614440917968e-05,
      "step": 29819,
      "training_step_time": 8.658251285552979
    },
    {
      "epoch": 2.8438568115234375e-05,
      "grad_norm": 0.11643999069929123,
      "learning_rate": 8.408365665298435e-06,
      "loss": 0.0559,
      "step": 29820
    },
    {
      "epoch": 2.8438568115234375e-05,
      "model_forward_time": 2.2811343669891357,
      "step": 29820
    },
    {
      "dynamics_loss_avg": 0.044128250889480115,
      "epoch": 2.8438568115234375e-05,
      "step": 29820
    },
    {
      "action_loss_avg": 0.0018844841164536775,
      "epoch": 2.8438568115234375e-05,
      "step": 29820
    },
    {
      "epoch": 2.8438568115234375e-05,
      "step": 29820,
      "training_step_time": 9.365062952041626
    },
    {
      "epoch": 2.8439521789550783e-05,
      "model_forward_time": 1.7785274982452393,
      "step": 29821
    },
    {
      "epoch": 2.8439521789550783e-05,
      "step": 29821,
      "training_step_time": 8.71090054512024
    },
    {
      "epoch": 2.8440475463867187e-05,
      "model_forward_time": 1.7358860969543457,
      "step": 29822
    },
    {
      "epoch": 2.8440475463867187e-05,
      "step": 29822,
      "training_step_time": 7.641706466674805
    },
    {
      "epoch": 2.8441429138183594e-05,
      "model_forward_time": 1.739142656326294,
      "step": 29823
    },
    {
      "epoch": 2.8441429138183594e-05,
      "step": 29823,
      "training_step_time": 6.811196327209473
    },
    {
      "epoch": 2.84423828125e-05,
      "model_forward_time": 1.730360746383667,
      "step": 29824
    },
    {
      "epoch": 2.84423828125e-05,
      "step": 29824,
      "training_step_time": 6.7969746589660645
    },
    {
      "epoch": 2.8443336486816406e-05,
      "model_forward_time": 1.7454447746276855,
      "step": 29825
    },
    {
      "epoch": 2.8443336486816406e-05,
      "step": 29825,
      "training_step_time": 6.868903398513794
    },
    {
      "epoch": 2.8444290161132813e-05,
      "model_forward_time": 1.7625503540039062,
      "step": 29826
    },
    {
      "epoch": 2.8444290161132813e-05,
      "step": 29826,
      "training_step_time": 6.810212850570679
    },
    {
      "epoch": 2.844524383544922e-05,
      "model_forward_time": 1.7366881370544434,
      "step": 29827
    },
    {
      "epoch": 2.844524383544922e-05,
      "step": 29827,
      "training_step_time": 6.82021689414978
    },
    {
      "epoch": 2.8446197509765624e-05,
      "model_forward_time": 1.7327618598937988,
      "step": 29828
    },
    {
      "epoch": 2.8446197509765624e-05,
      "step": 29828,
      "training_step_time": 7.607673645019531
    },
    {
      "epoch": 2.8447151184082032e-05,
      "model_forward_time": 2.2729198932647705,
      "step": 29829
    },
    {
      "epoch": 2.8447151184082032e-05,
      "step": 29829,
      "training_step_time": 10.79148530960083
    },
    {
      "epoch": 2.8448104858398436e-05,
      "grad_norm": 0.1314101219177246,
      "learning_rate": 8.40715570644257e-06,
      "loss": 0.0545,
      "step": 29830
    },
    {
      "epoch": 2.8448104858398436e-05,
      "model_forward_time": 1.7933168411254883,
      "step": 29830
    },
    {
      "dynamics_loss_avg": 0.04401713516563177,
      "epoch": 2.8448104858398436e-05,
      "step": 29830
    },
    {
      "action_loss_avg": 0.002695844357367605,
      "epoch": 2.8448104858398436e-05,
      "step": 29830
    },
    {
      "epoch": 2.8448104858398436e-05,
      "step": 29830,
      "training_step_time": 8.961609601974487
    },
    {
      "epoch": 2.8449058532714843e-05,
      "model_forward_time": 1.7137694358825684,
      "step": 29831
    },
    {
      "epoch": 2.8449058532714843e-05,
      "step": 29831,
      "training_step_time": 7.740660905838013
    },
    {
      "epoch": 2.845001220703125e-05,
      "model_forward_time": 1.7318146228790283,
      "step": 29832
    },
    {
      "epoch": 2.845001220703125e-05,
      "step": 29832,
      "training_step_time": 6.841440200805664
    },
    {
      "epoch": 2.8450965881347655e-05,
      "model_forward_time": 1.7321393489837646,
      "step": 29833
    },
    {
      "epoch": 2.8450965881347655e-05,
      "step": 29833,
      "training_step_time": 6.786632061004639
    },
    {
      "epoch": 2.8451919555664062e-05,
      "model_forward_time": 1.7378904819488525,
      "step": 29834
    },
    {
      "epoch": 2.8451919555664062e-05,
      "step": 29834,
      "training_step_time": 6.808381795883179
    },
    {
      "epoch": 2.845287322998047e-05,
      "model_forward_time": 1.7363042831420898,
      "step": 29835
    },
    {
      "epoch": 2.845287322998047e-05,
      "step": 29835,
      "training_step_time": 6.836941480636597
    },
    {
      "epoch": 2.8453826904296874e-05,
      "model_forward_time": 1.738893747329712,
      "step": 29836
    },
    {
      "epoch": 2.8453826904296874e-05,
      "step": 29836,
      "training_step_time": 6.807397842407227
    },
    {
      "epoch": 2.845478057861328e-05,
      "model_forward_time": 1.7489299774169922,
      "step": 29837
    },
    {
      "epoch": 2.845478057861328e-05,
      "step": 29837,
      "training_step_time": 7.656187057495117
    },
    {
      "epoch": 2.845573425292969e-05,
      "model_forward_time": 2.2753121852874756,
      "step": 29838
    },
    {
      "epoch": 2.845573425292969e-05,
      "step": 29838,
      "training_step_time": 9.585811376571655
    },
    {
      "epoch": 2.8456687927246092e-05,
      "model_forward_time": 1.7535135746002197,
      "step": 29839
    },
    {
      "epoch": 2.8456687927246092e-05,
      "step": 29839,
      "training_step_time": 10.828901052474976
    },
    {
      "epoch": 2.84576416015625e-05,
      "grad_norm": 0.12592627108097076,
      "learning_rate": 8.40594537498528e-06,
      "loss": 0.0582,
      "step": 29840
    },
    {
      "epoch": 2.84576416015625e-05,
      "model_forward_time": 1.7373721599578857,
      "step": 29840
    },
    {
      "dynamics_loss_avg": 0.05117297638207674,
      "epoch": 2.84576416015625e-05,
      "step": 29840
    },
    {
      "action_loss_avg": 0.0013704576413147151,
      "epoch": 2.84576416015625e-05,
      "step": 29840
    },
    {
      "epoch": 2.84576416015625e-05,
      "step": 29840,
      "training_step_time": 7.699479579925537
    },
    {
      "epoch": 2.8458595275878907e-05,
      "model_forward_time": 1.3693814277648926,
      "step": 29841
    },
    {
      "epoch": 2.8458595275878907e-05,
      "step": 29841,
      "training_step_time": 7.127326488494873
    },
    {
      "epoch": 2.845954895019531e-05,
      "model_forward_time": 1.7312850952148438,
      "step": 29842
    },
    {
      "epoch": 2.845954895019531e-05,
      "step": 29842,
      "training_step_time": 6.835047483444214
    },
    {
      "epoch": 2.846050262451172e-05,
      "model_forward_time": 1.7398779392242432,
      "step": 29843
    },
    {
      "epoch": 2.846050262451172e-05,
      "step": 29843,
      "training_step_time": 6.836834669113159
    },
    {
      "epoch": 2.8461456298828126e-05,
      "model_forward_time": 1.7383832931518555,
      "step": 29844
    },
    {
      "epoch": 2.8461456298828126e-05,
      "step": 29844,
      "training_step_time": 6.948344945907593
    },
    {
      "epoch": 2.846240997314453e-05,
      "model_forward_time": 1.7372050285339355,
      "step": 29845
    },
    {
      "epoch": 2.846240997314453e-05,
      "step": 29845,
      "training_step_time": 6.840887546539307
    },
    {
      "epoch": 2.8463363647460938e-05,
      "model_forward_time": 1.732072353363037,
      "step": 29846
    },
    {
      "epoch": 2.8463363647460938e-05,
      "step": 29846,
      "training_step_time": 7.694379806518555
    },
    {
      "epoch": 2.8464317321777345e-05,
      "model_forward_time": 2.2744970321655273,
      "step": 29847
    },
    {
      "epoch": 2.8464317321777345e-05,
      "step": 29847,
      "training_step_time": 8.401934385299683
    },
    {
      "epoch": 2.846527099609375e-05,
      "model_forward_time": 1.7390458583831787,
      "step": 29848
    },
    {
      "epoch": 2.846527099609375e-05,
      "step": 29848,
      "training_step_time": 9.778934001922607
    },
    {
      "epoch": 2.8466224670410156e-05,
      "model_forward_time": 1.7947096824645996,
      "step": 29849
    },
    {
      "epoch": 2.8466224670410156e-05,
      "step": 29849,
      "training_step_time": 8.424899339675903
    },
    {
      "epoch": 2.8467178344726564e-05,
      "grad_norm": 0.1545836627483368,
      "learning_rate": 8.404734671058924e-06,
      "loss": 0.0622,
      "step": 29850
    },
    {
      "epoch": 2.8467178344726564e-05,
      "model_forward_time": 1.736865758895874,
      "step": 29850
    },
    {
      "dynamics_loss_avg": 0.06656696125864983,
      "epoch": 2.8467178344726564e-05,
      "step": 29850
    },
    {
      "action_loss_avg": 0.0019290722062578424,
      "epoch": 2.8467178344726564e-05,
      "step": 29850
    },
    {
      "epoch": 2.8467178344726564e-05,
      "step": 29850,
      "training_step_time": 8.505213260650635
    },
    {
      "epoch": 2.8468132019042968e-05,
      "model_forward_time": 1.7402427196502686,
      "step": 29851
    },
    {
      "epoch": 2.8468132019042968e-05,
      "step": 29851,
      "training_step_time": 6.84600567817688
    },
    {
      "epoch": 2.8469085693359375e-05,
      "model_forward_time": 1.719926357269287,
      "step": 29852
    },
    {
      "epoch": 2.8469085693359375e-05,
      "step": 29852,
      "training_step_time": 6.856937646865845
    },
    {
      "epoch": 2.8470039367675783e-05,
      "model_forward_time": 1.7415146827697754,
      "step": 29853
    },
    {
      "epoch": 2.8470039367675783e-05,
      "step": 29853,
      "training_step_time": 6.852604389190674
    },
    {
      "epoch": 2.8470993041992187e-05,
      "model_forward_time": 1.3811371326446533,
      "step": 29854
    },
    {
      "epoch": 2.8470993041992187e-05,
      "step": 29854,
      "training_step_time": 6.85625696182251
    },
    {
      "epoch": 2.8471946716308594e-05,
      "model_forward_time": 1.7425835132598877,
      "step": 29855
    },
    {
      "epoch": 2.8471946716308594e-05,
      "step": 29855,
      "training_step_time": 7.452989816665649
    },
    {
      "epoch": 2.8472900390625e-05,
      "model_forward_time": 2.335480213165283,
      "step": 29856
    },
    {
      "epoch": 2.8472900390625e-05,
      "step": 29856,
      "training_step_time": 10.23560357093811
    },
    {
      "epoch": 2.8473854064941406e-05,
      "model_forward_time": 1.7459826469421387,
      "step": 29857
    },
    {
      "epoch": 2.8473854064941406e-05,
      "step": 29857,
      "training_step_time": 8.755963802337646
    },
    {
      "epoch": 2.8474807739257813e-05,
      "model_forward_time": 1.8079586029052734,
      "step": 29858
    },
    {
      "epoch": 2.8474807739257813e-05,
      "step": 29858,
      "training_step_time": 7.917805433273315
    },
    {
      "epoch": 2.847576141357422e-05,
      "model_forward_time": 1.7496886253356934,
      "step": 29859
    },
    {
      "epoch": 2.847576141357422e-05,
      "step": 29859,
      "training_step_time": 7.658434867858887
    },
    {
      "epoch": 2.8476715087890624e-05,
      "grad_norm": 0.17590966820716858,
      "learning_rate": 8.403523594795903e-06,
      "loss": 0.0602,
      "step": 29860
    },
    {
      "epoch": 2.8476715087890624e-05,
      "model_forward_time": 1.7481603622436523,
      "step": 29860
    },
    {
      "dynamics_loss_avg": 0.052474785782396796,
      "epoch": 2.8476715087890624e-05,
      "step": 29860
    },
    {
      "action_loss_avg": 0.0014877120498567819,
      "epoch": 2.8476715087890624e-05,
      "step": 29860
    },
    {
      "epoch": 2.8476715087890624e-05,
      "step": 29860,
      "training_step_time": 7.5166473388671875
    },
    {
      "epoch": 2.8477668762207032e-05,
      "model_forward_time": 1.7440483570098877,
      "step": 29861
    },
    {
      "epoch": 2.8477668762207032e-05,
      "step": 29861,
      "training_step_time": 6.815739631652832
    },
    {
      "epoch": 2.8478622436523436e-05,
      "model_forward_time": 1.742598295211792,
      "step": 29862
    },
    {
      "epoch": 2.8478622436523436e-05,
      "step": 29862,
      "training_step_time": 6.7201151847839355
    },
    {
      "epoch": 2.8479576110839843e-05,
      "model_forward_time": 1.7438082695007324,
      "step": 29863
    },
    {
      "epoch": 2.8479576110839843e-05,
      "step": 29863,
      "training_step_time": 6.801671266555786
    },
    {
      "epoch": 2.848052978515625e-05,
      "model_forward_time": 1.7224431037902832,
      "step": 29864
    },
    {
      "epoch": 2.848052978515625e-05,
      "step": 29864,
      "training_step_time": 7.0746424198150635
    },
    {
      "epoch": 2.8481483459472655e-05,
      "model_forward_time": 2.3404295444488525,
      "step": 29865
    },
    {
      "epoch": 2.8481483459472655e-05,
      "step": 29865,
      "training_step_time": 8.24363088607788
    },
    {
      "epoch": 2.8482437133789062e-05,
      "model_forward_time": 1.7459487915039062,
      "step": 29866
    },
    {
      "epoch": 2.8482437133789062e-05,
      "step": 29866,
      "training_step_time": 11.516430377960205
    },
    {
      "epoch": 2.848339080810547e-05,
      "model_forward_time": 1.7937898635864258,
      "step": 29867
    },
    {
      "epoch": 2.848339080810547e-05,
      "step": 29867,
      "training_step_time": 8.619027614593506
    },
    {
      "epoch": 2.8484344482421874e-05,
      "model_forward_time": 1.7299621105194092,
      "step": 29868
    },
    {
      "epoch": 2.8484344482421874e-05,
      "step": 29868,
      "training_step_time": 7.7403528690338135
    },
    {
      "epoch": 2.848529815673828e-05,
      "model_forward_time": 1.7420282363891602,
      "step": 29869
    },
    {
      "epoch": 2.848529815673828e-05,
      "step": 29869,
      "training_step_time": 7.514347076416016
    },
    {
      "epoch": 2.848625183105469e-05,
      "grad_norm": 0.1542198359966278,
      "learning_rate": 8.402312146328659e-06,
      "loss": 0.0617,
      "step": 29870
    },
    {
      "epoch": 2.848625183105469e-05,
      "model_forward_time": 1.7334916591644287,
      "step": 29870
    },
    {
      "dynamics_loss_avg": 0.056164107657969,
      "epoch": 2.848625183105469e-05,
      "step": 29870
    },
    {
      "action_loss_avg": 0.002134186233161017,
      "epoch": 2.848625183105469e-05,
      "step": 29870
    },
    {
      "epoch": 2.848625183105469e-05,
      "step": 29870,
      "training_step_time": 6.697881460189819
    },
    {
      "epoch": 2.8487205505371092e-05,
      "model_forward_time": 1.7488415241241455,
      "step": 29871
    },
    {
      "epoch": 2.8487205505371092e-05,
      "step": 29871,
      "training_step_time": 6.823259353637695
    },
    {
      "epoch": 2.84881591796875e-05,
      "model_forward_time": 1.731421709060669,
      "step": 29872
    },
    {
      "epoch": 2.84881591796875e-05,
      "step": 29872,
      "training_step_time": 6.809943199157715
    },
    {
      "epoch": 2.8489112854003907e-05,
      "model_forward_time": 1.736823320388794,
      "step": 29873
    },
    {
      "epoch": 2.8489112854003907e-05,
      "step": 29873,
      "training_step_time": 7.091511487960815
    },
    {
      "epoch": 2.849006652832031e-05,
      "model_forward_time": 2.294921398162842,
      "step": 29874
    },
    {
      "epoch": 2.849006652832031e-05,
      "step": 29874,
      "training_step_time": 8.428341627120972
    },
    {
      "epoch": 2.849102020263672e-05,
      "model_forward_time": 1.7342495918273926,
      "step": 29875
    },
    {
      "epoch": 2.849102020263672e-05,
      "step": 29875,
      "training_step_time": 9.207245111465454
    },
    {
      "epoch": 2.8491973876953126e-05,
      "model_forward_time": 1.7757115364074707,
      "step": 29876
    },
    {
      "epoch": 2.8491973876953126e-05,
      "step": 29876,
      "training_step_time": 9.54299283027649
    },
    {
      "epoch": 2.849292755126953e-05,
      "model_forward_time": 1.7426199913024902,
      "step": 29877
    },
    {
      "epoch": 2.849292755126953e-05,
      "step": 29877,
      "training_step_time": 7.670451879501343
    },
    {
      "epoch": 2.8493881225585938e-05,
      "model_forward_time": 1.7196781635284424,
      "step": 29878
    },
    {
      "epoch": 2.8493881225585938e-05,
      "step": 29878,
      "training_step_time": 8.419023990631104
    },
    {
      "epoch": 2.8494834899902345e-05,
      "model_forward_time": 1.737793207168579,
      "step": 29879
    },
    {
      "epoch": 2.8494834899902345e-05,
      "step": 29879,
      "training_step_time": 6.70739221572876
    },
    {
      "epoch": 2.849578857421875e-05,
      "grad_norm": 0.2480129599571228,
      "learning_rate": 8.401100325789675e-06,
      "loss": 0.0621,
      "step": 29880
    },
    {
      "epoch": 2.849578857421875e-05,
      "model_forward_time": 1.7335762977600098,
      "step": 29880
    },
    {
      "dynamics_loss_avg": 0.039266332611441614,
      "epoch": 2.849578857421875e-05,
      "step": 29880
    },
    {
      "action_loss_avg": 0.001401835144497454,
      "epoch": 2.849578857421875e-05,
      "step": 29880
    },
    {
      "epoch": 2.849578857421875e-05,
      "step": 29880,
      "training_step_time": 6.696105718612671
    },
    {
      "epoch": 2.8496742248535156e-05,
      "model_forward_time": 1.7414345741271973,
      "step": 29881
    },
    {
      "epoch": 2.8496742248535156e-05,
      "step": 29881,
      "training_step_time": 6.749968767166138
    },
    {
      "epoch": 2.8497695922851564e-05,
      "model_forward_time": 1.740419864654541,
      "step": 29882
    },
    {
      "epoch": 2.8497695922851564e-05,
      "step": 29882,
      "training_step_time": 7.021915674209595
    },
    {
      "epoch": 2.8498649597167968e-05,
      "model_forward_time": 1.0340633392333984,
      "step": 29883
    },
    {
      "epoch": 2.8498649597167968e-05,
      "step": 29883,
      "training_step_time": 9.30417013168335
    },
    {
      "epoch": 2.8499603271484375e-05,
      "model_forward_time": 2.2986338138580322,
      "step": 29884
    },
    {
      "epoch": 2.8499603271484375e-05,
      "step": 29884,
      "training_step_time": 10.276341438293457
    },
    {
      "epoch": 2.8500556945800783e-05,
      "model_forward_time": 1.751159429550171,
      "step": 29885
    },
    {
      "epoch": 2.8500556945800783e-05,
      "step": 29885,
      "training_step_time": 9.62532377243042
    },
    {
      "epoch": 2.8501510620117187e-05,
      "model_forward_time": 1.7218780517578125,
      "step": 29886
    },
    {
      "epoch": 2.8501510620117187e-05,
      "step": 29886,
      "training_step_time": 7.753263235092163
    },
    {
      "epoch": 2.8502464294433594e-05,
      "model_forward_time": 1.7436432838439941,
      "step": 29887
    },
    {
      "epoch": 2.8502464294433594e-05,
      "step": 29887,
      "training_step_time": 7.580734968185425
    },
    {
      "epoch": 2.850341796875e-05,
      "model_forward_time": 1.7393884658813477,
      "step": 29888
    },
    {
      "epoch": 2.850341796875e-05,
      "step": 29888,
      "training_step_time": 7.443425893783569
    },
    {
      "epoch": 2.8504371643066406e-05,
      "model_forward_time": 1.724280595779419,
      "step": 29889
    },
    {
      "epoch": 2.8504371643066406e-05,
      "step": 29889,
      "training_step_time": 6.742435455322266
    },
    {
      "epoch": 2.8505325317382813e-05,
      "grad_norm": 0.18197280168533325,
      "learning_rate": 8.39988813331147e-06,
      "loss": 0.0614,
      "step": 29890
    },
    {
      "epoch": 2.8505325317382813e-05,
      "model_forward_time": 1.7353878021240234,
      "step": 29890
    },
    {
      "dynamics_loss_avg": 0.05610068216919899,
      "epoch": 2.8505325317382813e-05,
      "step": 29890
    },
    {
      "action_loss_avg": 0.0018783227598760277,
      "epoch": 2.8505325317382813e-05,
      "step": 29890
    },
    {
      "epoch": 2.8505325317382813e-05,
      "step": 29890,
      "training_step_time": 6.695531368255615
    },
    {
      "epoch": 2.8506278991699217e-05,
      "model_forward_time": 1.7387449741363525,
      "step": 29891
    },
    {
      "epoch": 2.8506278991699217e-05,
      "step": 29891,
      "training_step_time": 6.742037296295166
    },
    {
      "epoch": 2.8507232666015624e-05,
      "model_forward_time": 1.7377378940582275,
      "step": 29892
    },
    {
      "epoch": 2.8507232666015624e-05,
      "step": 29892,
      "training_step_time": 7.5927488803863525
    },
    {
      "epoch": 2.8508186340332032e-05,
      "model_forward_time": 2.254422664642334,
      "step": 29893
    },
    {
      "epoch": 2.8508186340332032e-05,
      "step": 29893,
      "training_step_time": 8.297751903533936
    },
    {
      "epoch": 2.8509140014648436e-05,
      "model_forward_time": 1.7102484703063965,
      "step": 29894
    },
    {
      "epoch": 2.8509140014648436e-05,
      "step": 29894,
      "training_step_time": 7.6253862380981445
    },
    {
      "epoch": 2.8510093688964843e-05,
      "model_forward_time": 1.7693336009979248,
      "step": 29895
    },
    {
      "epoch": 2.8510093688964843e-05,
      "step": 29895,
      "training_step_time": 7.716170787811279
    },
    {
      "epoch": 2.851104736328125e-05,
      "model_forward_time": 1.7130260467529297,
      "step": 29896
    },
    {
      "epoch": 2.851104736328125e-05,
      "step": 29896,
      "training_step_time": 7.226507902145386
    },
    {
      "epoch": 2.8512001037597655e-05,
      "model_forward_time": 1.7171313762664795,
      "step": 29897
    },
    {
      "epoch": 2.8512001037597655e-05,
      "step": 29897,
      "training_step_time": 7.553532838821411
    },
    {
      "epoch": 2.8512954711914062e-05,
      "model_forward_time": 1.7107253074645996,
      "step": 29898
    },
    {
      "epoch": 2.8512954711914062e-05,
      "step": 29898,
      "training_step_time": 6.983131170272827
    },
    {
      "epoch": 2.851390838623047e-05,
      "model_forward_time": 1.7283408641815186,
      "step": 29899
    },
    {
      "epoch": 2.851390838623047e-05,
      "step": 29899,
      "training_step_time": 6.739251613616943
    },
    {
      "epoch": 2.8514862060546874e-05,
      "grad_norm": 0.12186727672815323,
      "learning_rate": 8.398675569026613e-06,
      "loss": 0.0601,
      "step": 29900
    },
    {
      "epoch": 2.8514862060546874e-05,
      "model_forward_time": 1.714308500289917,
      "step": 29900
    },
    {
      "dynamics_loss_avg": 0.055760775320231916,
      "epoch": 2.8514862060546874e-05,
      "step": 29900
    },
    {
      "action_loss_avg": 0.0026434007508214564,
      "epoch": 2.8514862060546874e-05,
      "step": 29900
    },
    {
      "epoch": 2.8514862060546874e-05,
      "step": 29900,
      "training_step_time": 7.020375967025757
    },
    {
      "epoch": 2.851581573486328e-05,
      "model_forward_time": 1.7166144847869873,
      "step": 29901
    },
    {
      "epoch": 2.851581573486328e-05,
      "step": 29901,
      "training_step_time": 8.003145694732666
    },
    {
      "epoch": 2.851676940917969e-05,
      "model_forward_time": 2.2922935485839844,
      "step": 29902
    },
    {
      "epoch": 2.851676940917969e-05,
      "step": 29902,
      "training_step_time": 7.6988396644592285
    },
    {
      "epoch": 2.8517723083496093e-05,
      "model_forward_time": 1.017852544784546,
      "step": 29903
    },
    {
      "epoch": 2.8517723083496093e-05,
      "step": 29903,
      "training_step_time": 8.3251633644104
    },
    {
      "epoch": 2.85186767578125e-05,
      "model_forward_time": 1.7733063697814941,
      "step": 29904
    },
    {
      "epoch": 2.85186767578125e-05,
      "step": 29904,
      "training_step_time": 9.425427675247192
    },
    {
      "epoch": 2.8519630432128907e-05,
      "model_forward_time": 1.7285733222961426,
      "step": 29905
    },
    {
      "epoch": 2.8519630432128907e-05,
      "step": 29905,
      "training_step_time": 7.699692249298096
    },
    {
      "epoch": 2.852058410644531e-05,
      "model_forward_time": 1.7244186401367188,
      "step": 29906
    },
    {
      "epoch": 2.852058410644531e-05,
      "step": 29906,
      "training_step_time": 7.63033390045166
    },
    {
      "epoch": 2.852153778076172e-05,
      "model_forward_time": 1.7246532440185547,
      "step": 29907
    },
    {
      "epoch": 2.852153778076172e-05,
      "step": 29907,
      "training_step_time": 7.509535551071167
    },
    {
      "epoch": 2.8522491455078126e-05,
      "model_forward_time": 1.7375786304473877,
      "step": 29908
    },
    {
      "epoch": 2.8522491455078126e-05,
      "step": 29908,
      "training_step_time": 6.807365894317627
    },
    {
      "epoch": 2.852344512939453e-05,
      "model_forward_time": 1.7293028831481934,
      "step": 29909
    },
    {
      "epoch": 2.852344512939453e-05,
      "step": 29909,
      "training_step_time": 7.122757911682129
    },
    {
      "epoch": 2.8524398803710938e-05,
      "grad_norm": 0.10023579746484756,
      "learning_rate": 8.397462633067705e-06,
      "loss": 0.0576,
      "step": 29910
    },
    {
      "epoch": 2.8524398803710938e-05,
      "model_forward_time": 1.726381778717041,
      "step": 29910
    },
    {
      "dynamics_loss_avg": 0.05506120175123215,
      "epoch": 2.8524398803710938e-05,
      "step": 29910
    },
    {
      "action_loss_avg": 0.0025442295707762243,
      "epoch": 2.8524398803710938e-05,
      "step": 29910
    },
    {
      "epoch": 2.8524398803710938e-05,
      "step": 29910,
      "training_step_time": 7.506869554519653
    },
    {
      "epoch": 2.8525352478027345e-05,
      "model_forward_time": 1.7510805130004883,
      "step": 29911
    },
    {
      "epoch": 2.8525352478027345e-05,
      "step": 29911,
      "training_step_time": 8.53817629814148
    },
    {
      "epoch": 2.852630615234375e-05,
      "model_forward_time": 1.9522321224212646,
      "step": 29912
    },
    {
      "epoch": 2.852630615234375e-05,
      "step": 29912,
      "training_step_time": 8.659010171890259
    },
    {
      "epoch": 2.8527259826660156e-05,
      "model_forward_time": 1.8087139129638672,
      "step": 29913
    },
    {
      "epoch": 2.8527259826660156e-05,
      "step": 29913,
      "training_step_time": 7.750594139099121
    },
    {
      "epoch": 2.8528213500976564e-05,
      "model_forward_time": 1.7602741718292236,
      "step": 29914
    },
    {
      "epoch": 2.8528213500976564e-05,
      "step": 29914,
      "training_step_time": 7.7313268184661865
    },
    {
      "epoch": 2.8529167175292968e-05,
      "model_forward_time": 1.3732788562774658,
      "step": 29915
    },
    {
      "epoch": 2.8529167175292968e-05,
      "step": 29915,
      "training_step_time": 7.558483123779297
    },
    {
      "epoch": 2.8530120849609375e-05,
      "model_forward_time": 1.7488162517547607,
      "step": 29916
    },
    {
      "epoch": 2.8530120849609375e-05,
      "step": 29916,
      "training_step_time": 7.4857566356658936
    },
    {
      "epoch": 2.8531074523925783e-05,
      "model_forward_time": 1.7580814361572266,
      "step": 29917
    },
    {
      "epoch": 2.8531074523925783e-05,
      "step": 29917,
      "training_step_time": 6.821739196777344
    },
    {
      "epoch": 2.8532028198242187e-05,
      "model_forward_time": 1.754457712173462,
      "step": 29918
    },
    {
      "epoch": 2.8532028198242187e-05,
      "step": 29918,
      "training_step_time": 7.096503496170044
    },
    {
      "epoch": 2.8532981872558594e-05,
      "model_forward_time": 1.7554600238800049,
      "step": 29919
    },
    {
      "epoch": 2.8532981872558594e-05,
      "step": 29919,
      "training_step_time": 7.291717529296875
    },
    {
      "epoch": 2.8533935546875e-05,
      "grad_norm": 0.11396773159503937,
      "learning_rate": 8.396249325567391e-06,
      "loss": 0.0607,
      "step": 29920
    },
    {
      "epoch": 2.8533935546875e-05,
      "model_forward_time": 1.7472755908966064,
      "step": 29920
    },
    {
      "dynamics_loss_avg": 0.05792681332677603,
      "epoch": 2.8533935546875e-05,
      "step": 29920
    },
    {
      "action_loss_avg": 0.0023285418108571323,
      "epoch": 2.8533935546875e-05,
      "step": 29920
    },
    {
      "epoch": 2.8533935546875e-05,
      "step": 29920,
      "training_step_time": 7.698570966720581
    },
    {
      "epoch": 2.8534889221191406e-05,
      "model_forward_time": 2.311408042907715,
      "step": 29921
    },
    {
      "epoch": 2.8534889221191406e-05,
      "step": 29921,
      "training_step_time": 8.412550926208496
    },
    {
      "epoch": 2.8535842895507813e-05,
      "model_forward_time": 1.7811012268066406,
      "step": 29922
    },
    {
      "epoch": 2.8535842895507813e-05,
      "step": 29922,
      "training_step_time": 9.360897541046143
    },
    {
      "epoch": 2.8536796569824217e-05,
      "model_forward_time": 1.7459502220153809,
      "step": 29923
    },
    {
      "epoch": 2.8536796569824217e-05,
      "step": 29923,
      "training_step_time": 7.822811841964722
    },
    {
      "epoch": 2.8537750244140625e-05,
      "model_forward_time": 1.7192738056182861,
      "step": 29924
    },
    {
      "epoch": 2.8537750244140625e-05,
      "step": 29924,
      "training_step_time": 7.641574859619141
    },
    {
      "epoch": 2.8538703918457032e-05,
      "model_forward_time": 1.7790207862854004,
      "step": 29925
    },
    {
      "epoch": 2.8538703918457032e-05,
      "step": 29925,
      "training_step_time": 7.253377914428711
    },
    {
      "epoch": 2.8539657592773436e-05,
      "model_forward_time": 1.0216975212097168,
      "step": 29926
    },
    {
      "epoch": 2.8539657592773436e-05,
      "step": 29926,
      "training_step_time": 7.490926027297974
    },
    {
      "epoch": 2.8540611267089843e-05,
      "model_forward_time": 1.7315559387207031,
      "step": 29927
    },
    {
      "epoch": 2.8540611267089843e-05,
      "step": 29927,
      "training_step_time": 7.248707056045532
    },
    {
      "epoch": 2.854156494140625e-05,
      "model_forward_time": 1.746187448501587,
      "step": 29928
    },
    {
      "epoch": 2.854156494140625e-05,
      "step": 29928,
      "training_step_time": 7.422503232955933
    },
    {
      "epoch": 2.8542518615722655e-05,
      "model_forward_time": 1.7330312728881836,
      "step": 29929
    },
    {
      "epoch": 2.8542518615722655e-05,
      "step": 29929,
      "training_step_time": 7.666375160217285
    },
    {
      "epoch": 2.8543472290039062e-05,
      "grad_norm": 0.15125367045402527,
      "learning_rate": 8.395035646658356e-06,
      "loss": 0.0576,
      "step": 29930
    },
    {
      "epoch": 2.8543472290039062e-05,
      "model_forward_time": 1.7356057167053223,
      "step": 29930
    },
    {
      "dynamics_loss_avg": 0.04373201048001647,
      "epoch": 2.8543472290039062e-05,
      "step": 29930
    },
    {
      "action_loss_avg": 0.001317036314867437,
      "epoch": 2.8543472290039062e-05,
      "step": 29930
    },
    {
      "epoch": 2.8543472290039062e-05,
      "step": 29930,
      "training_step_time": 9.200457334518433
    },
    {
      "epoch": 2.854442596435547e-05,
      "model_forward_time": 2.27601957321167,
      "step": 29931
    },
    {
      "epoch": 2.854442596435547e-05,
      "step": 29931,
      "training_step_time": 9.265782594680786
    },
    {
      "epoch": 2.8545379638671874e-05,
      "model_forward_time": 1.7829616069793701,
      "step": 29932
    },
    {
      "epoch": 2.8545379638671874e-05,
      "step": 29932,
      "training_step_time": 8.91869831085205
    },
    {
      "epoch": 2.854633331298828e-05,
      "model_forward_time": 1.7324881553649902,
      "step": 29933
    },
    {
      "epoch": 2.854633331298828e-05,
      "step": 29933,
      "training_step_time": 8.637039422988892
    },
    {
      "epoch": 2.854728698730469e-05,
      "model_forward_time": 1.7294278144836426,
      "step": 29934
    },
    {
      "epoch": 2.854728698730469e-05,
      "step": 29934,
      "training_step_time": 7.623108148574829
    },
    {
      "epoch": 2.8548240661621093e-05,
      "model_forward_time": 1.722914695739746,
      "step": 29935
    },
    {
      "epoch": 2.8548240661621093e-05,
      "step": 29935,
      "training_step_time": 7.516547203063965
    },
    {
      "epoch": 2.85491943359375e-05,
      "model_forward_time": 1.7252728939056396,
      "step": 29936
    },
    {
      "epoch": 2.85491943359375e-05,
      "step": 29936,
      "training_step_time": 6.857976675033569
    },
    {
      "epoch": 2.8550148010253907e-05,
      "model_forward_time": 1.3639512062072754,
      "step": 29937
    },
    {
      "epoch": 2.8550148010253907e-05,
      "step": 29937,
      "training_step_time": 7.36211371421814
    },
    {
      "epoch": 2.855110168457031e-05,
      "model_forward_time": 1.7340004444122314,
      "step": 29938
    },
    {
      "epoch": 2.855110168457031e-05,
      "step": 29938,
      "training_step_time": 7.629744052886963
    },
    {
      "epoch": 2.855205535888672e-05,
      "model_forward_time": 1.726926565170288,
      "step": 29939
    },
    {
      "epoch": 2.855205535888672e-05,
      "step": 29939,
      "training_step_time": 10.288451671600342
    },
    {
      "epoch": 2.8553009033203126e-05,
      "grad_norm": 0.12113361805677414,
      "learning_rate": 8.39382159647333e-06,
      "loss": 0.0558,
      "step": 29940
    },
    {
      "epoch": 2.8553009033203126e-05,
      "model_forward_time": 2.2874653339385986,
      "step": 29940
    },
    {
      "dynamics_loss_avg": 0.05212421417236328,
      "epoch": 2.8553009033203126e-05,
      "step": 29940
    },
    {
      "action_loss_avg": 0.002786433562869206,
      "epoch": 2.8553009033203126e-05,
      "step": 29940
    },
    {
      "epoch": 2.8553009033203126e-05,
      "step": 29940,
      "training_step_time": 7.674234628677368
    },
    {
      "epoch": 2.855396270751953e-05,
      "model_forward_time": 1.762430191040039,
      "step": 29941
    },
    {
      "epoch": 2.855396270751953e-05,
      "step": 29941,
      "training_step_time": 8.635599851608276
    },
    {
      "epoch": 2.8554916381835938e-05,
      "model_forward_time": 1.7236676216125488,
      "step": 29942
    },
    {
      "epoch": 2.8554916381835938e-05,
      "step": 29942,
      "training_step_time": 7.794036865234375
    },
    {
      "epoch": 2.8555870056152345e-05,
      "model_forward_time": 1.7289066314697266,
      "step": 29943
    },
    {
      "epoch": 2.8555870056152345e-05,
      "step": 29943,
      "training_step_time": 7.736665964126587
    },
    {
      "epoch": 2.855682373046875e-05,
      "model_forward_time": 1.7437143325805664,
      "step": 29944
    },
    {
      "epoch": 2.855682373046875e-05,
      "step": 29944,
      "training_step_time": 7.651755332946777
    },
    {
      "epoch": 2.8557777404785157e-05,
      "model_forward_time": 1.7347195148468018,
      "step": 29945
    },
    {
      "epoch": 2.8557777404785157e-05,
      "step": 29945,
      "training_step_time": 6.9561543464660645
    },
    {
      "epoch": 2.8558731079101564e-05,
      "model_forward_time": 1.7290194034576416,
      "step": 29946
    },
    {
      "epoch": 2.8558731079101564e-05,
      "step": 29946,
      "training_step_time": 7.296902656555176
    },
    {
      "epoch": 2.8559684753417968e-05,
      "model_forward_time": 1.727663516998291,
      "step": 29947
    },
    {
      "epoch": 2.8559684753417968e-05,
      "step": 29947,
      "training_step_time": 7.661683559417725
    },
    {
      "epoch": 2.8560638427734375e-05,
      "model_forward_time": 1.7327847480773926,
      "step": 29948
    },
    {
      "epoch": 2.8560638427734375e-05,
      "step": 29948,
      "training_step_time": 7.746777296066284
    },
    {
      "epoch": 2.8561592102050783e-05,
      "model_forward_time": 2.32475209236145,
      "step": 29949
    },
    {
      "epoch": 2.8561592102050783e-05,
      "step": 29949,
      "training_step_time": 10.253241777420044
    },
    {
      "epoch": 2.8562545776367187e-05,
      "grad_norm": 0.14131267368793488,
      "learning_rate": 8.392607175145076e-06,
      "loss": 0.0587,
      "step": 29950
    },
    {
      "epoch": 2.8562545776367187e-05,
      "model_forward_time": 1.7457504272460938,
      "step": 29950
    },
    {
      "dynamics_loss_avg": 0.060847511328756806,
      "epoch": 2.8562545776367187e-05,
      "step": 29950
    },
    {
      "action_loss_avg": 0.0017019645543769001,
      "epoch": 2.8562545776367187e-05,
      "step": 29950
    },
    {
      "epoch": 2.8562545776367187e-05,
      "step": 29950,
      "training_step_time": 8.203068971633911
    },
    {
      "epoch": 2.8563499450683594e-05,
      "model_forward_time": 1.749161720275879,
      "step": 29951
    },
    {
      "epoch": 2.8563499450683594e-05,
      "step": 29951,
      "training_step_time": 7.722055912017822
    },
    {
      "epoch": 2.8564453125e-05,
      "model_forward_time": 1.7951619625091553,
      "step": 29952
    },
    {
      "epoch": 2.8564453125e-05,
      "step": 29952,
      "training_step_time": 7.224480628967285
    },
    {
      "epoch": 2.8565406799316406e-05,
      "model_forward_time": 1.763962984085083,
      "step": 29953
    },
    {
      "epoch": 2.8565406799316406e-05,
      "step": 29953,
      "training_step_time": 7.460800886154175
    },
    {
      "epoch": 2.8566360473632813e-05,
      "model_forward_time": 1.7540504932403564,
      "step": 29954
    },
    {
      "epoch": 2.8566360473632813e-05,
      "step": 29954,
      "training_step_time": 6.855615615844727
    },
    {
      "epoch": 2.8567314147949217e-05,
      "model_forward_time": 1.7554476261138916,
      "step": 29955
    },
    {
      "epoch": 2.8567314147949217e-05,
      "step": 29955,
      "training_step_time": 7.306575059890747
    },
    {
      "epoch": 2.8568267822265625e-05,
      "model_forward_time": 1.7523987293243408,
      "step": 29956
    },
    {
      "epoch": 2.8568267822265625e-05,
      "step": 29956,
      "training_step_time": 8.227927923202515
    },
    {
      "epoch": 2.8569221496582032e-05,
      "model_forward_time": 1.7484047412872314,
      "step": 29957
    },
    {
      "epoch": 2.8569221496582032e-05,
      "step": 29957,
      "training_step_time": 7.756763935089111
    },
    {
      "epoch": 2.8570175170898436e-05,
      "model_forward_time": 1.7475578784942627,
      "step": 29958
    },
    {
      "epoch": 2.8570175170898436e-05,
      "step": 29958,
      "training_step_time": 12.216950416564941
    },
    {
      "epoch": 2.8571128845214843e-05,
      "model_forward_time": 1.7442855834960938,
      "step": 29959
    },
    {
      "epoch": 2.8571128845214843e-05,
      "step": 29959,
      "training_step_time": 8.642763376235962
    },
    {
      "epoch": 2.857208251953125e-05,
      "grad_norm": 0.11429423093795776,
      "learning_rate": 8.3913923828064e-06,
      "loss": 0.0607,
      "step": 29960
    },
    {
      "epoch": 2.857208251953125e-05,
      "model_forward_time": 1.7673602104187012,
      "step": 29960
    },
    {
      "dynamics_loss_avg": 0.05228427033871412,
      "epoch": 2.857208251953125e-05,
      "step": 29960
    },
    {
      "action_loss_avg": 0.0014757345445104875,
      "epoch": 2.857208251953125e-05,
      "step": 29960
    },
    {
      "epoch": 2.857208251953125e-05,
      "step": 29960,
      "training_step_time": 8.834757804870605
    },
    {
      "epoch": 2.8573036193847655e-05,
      "model_forward_time": 1.388479232788086,
      "step": 29961
    },
    {
      "epoch": 2.8573036193847655e-05,
      "step": 29961,
      "training_step_time": 7.237023115158081
    },
    {
      "epoch": 2.8573989868164062e-05,
      "model_forward_time": 1.7477469444274902,
      "step": 29962
    },
    {
      "epoch": 2.8573989868164062e-05,
      "step": 29962,
      "training_step_time": 7.698753118515015
    },
    {
      "epoch": 2.857494354248047e-05,
      "model_forward_time": 1.7455155849456787,
      "step": 29963
    },
    {
      "epoch": 2.857494354248047e-05,
      "step": 29963,
      "training_step_time": 7.495616912841797
    },
    {
      "epoch": 2.8575897216796874e-05,
      "model_forward_time": 1.7394685745239258,
      "step": 29964
    },
    {
      "epoch": 2.8575897216796874e-05,
      "step": 29964,
      "training_step_time": 7.281040191650391
    },
    {
      "epoch": 2.857685089111328e-05,
      "model_forward_time": 1.7457835674285889,
      "step": 29965
    },
    {
      "epoch": 2.857685089111328e-05,
      "step": 29965,
      "training_step_time": 8.116875648498535
    },
    {
      "epoch": 2.857780456542969e-05,
      "model_forward_time": 1.784940481185913,
      "step": 29966
    },
    {
      "epoch": 2.857780456542969e-05,
      "step": 29966,
      "training_step_time": 8.165469646453857
    },
    {
      "epoch": 2.8578758239746093e-05,
      "model_forward_time": 1.7547218799591064,
      "step": 29967
    },
    {
      "epoch": 2.8578758239746093e-05,
      "step": 29967,
      "training_step_time": 8.595976829528809
    },
    {
      "epoch": 2.85797119140625e-05,
      "model_forward_time": 2.354630708694458,
      "step": 29968
    },
    {
      "epoch": 2.85797119140625e-05,
      "step": 29968,
      "training_step_time": 7.603521347045898
    },
    {
      "epoch": 2.8580665588378907e-05,
      "model_forward_time": 1.7489373683929443,
      "step": 29969
    },
    {
      "epoch": 2.8580665588378907e-05,
      "step": 29969,
      "training_step_time": 8.572595834732056
    },
    {
      "epoch": 2.858161926269531e-05,
      "grad_norm": 0.1530161052942276,
      "learning_rate": 8.390177219590153e-06,
      "loss": 0.057,
      "step": 29970
    },
    {
      "epoch": 2.858161926269531e-05,
      "model_forward_time": 1.809842586517334,
      "step": 29970
    },
    {
      "dynamics_loss_avg": 0.047003046795725825,
      "epoch": 2.858161926269531e-05,
      "step": 29970
    },
    {
      "action_loss_avg": 0.001783314395288471,
      "epoch": 2.858161926269531e-05,
      "step": 29970
    },
    {
      "epoch": 2.858161926269531e-05,
      "step": 29970,
      "training_step_time": 7.688815593719482
    },
    {
      "epoch": 2.858257293701172e-05,
      "model_forward_time": 1.7427866458892822,
      "step": 29971
    },
    {
      "epoch": 2.858257293701172e-05,
      "step": 29971,
      "training_step_time": 7.6242616176605225
    },
    {
      "epoch": 2.8583526611328126e-05,
      "model_forward_time": 1.7561028003692627,
      "step": 29972
    },
    {
      "epoch": 2.8583526611328126e-05,
      "step": 29972,
      "training_step_time": 7.5000996589660645
    },
    {
      "epoch": 2.858448028564453e-05,
      "model_forward_time": 1.768794298171997,
      "step": 29973
    },
    {
      "epoch": 2.858448028564453e-05,
      "step": 29973,
      "training_step_time": 7.294215440750122
    },
    {
      "epoch": 2.8585433959960938e-05,
      "model_forward_time": 1.7452623844146729,
      "step": 29974
    },
    {
      "epoch": 2.8585433959960938e-05,
      "step": 29974,
      "training_step_time": 8.375807762145996
    },
    {
      "epoch": 2.8586387634277345e-05,
      "model_forward_time": 1.7390501499176025,
      "step": 29975
    },
    {
      "epoch": 2.8586387634277345e-05,
      "step": 29975,
      "training_step_time": 8.382706880569458
    },
    {
      "epoch": 2.858734130859375e-05,
      "model_forward_time": 1.74472975730896,
      "step": 29976
    },
    {
      "epoch": 2.858734130859375e-05,
      "step": 29976,
      "training_step_time": 8.490190505981445
    },
    {
      "epoch": 2.8588294982910157e-05,
      "model_forward_time": 2.323038101196289,
      "step": 29977
    },
    {
      "epoch": 2.8588294982910157e-05,
      "step": 29977,
      "training_step_time": 8.77267599105835
    },
    {
      "epoch": 2.8589248657226564e-05,
      "model_forward_time": 1.7458157539367676,
      "step": 29978
    },
    {
      "epoch": 2.8589248657226564e-05,
      "step": 29978,
      "training_step_time": 7.600225448608398
    },
    {
      "epoch": 2.8590202331542968e-05,
      "model_forward_time": 1.737760066986084,
      "step": 29979
    },
    {
      "epoch": 2.8590202331542968e-05,
      "step": 29979,
      "training_step_time": 7.890708923339844
    },
    {
      "epoch": 2.8591156005859375e-05,
      "grad_norm": 0.1171560287475586,
      "learning_rate": 8.388961685629222e-06,
      "loss": 0.0605,
      "step": 29980
    },
    {
      "epoch": 2.8591156005859375e-05,
      "model_forward_time": 1.7264471054077148,
      "step": 29980
    },
    {
      "dynamics_loss_avg": 0.058057887479662895,
      "epoch": 2.8591156005859375e-05,
      "step": 29980
    },
    {
      "action_loss_avg": 0.0014604280702769756,
      "epoch": 2.8591156005859375e-05,
      "step": 29980
    },
    {
      "epoch": 2.8591156005859375e-05,
      "step": 29980,
      "training_step_time": 7.494710206985474
    },
    {
      "epoch": 2.8592109680175783e-05,
      "model_forward_time": 1.7310895919799805,
      "step": 29981
    },
    {
      "epoch": 2.8592109680175783e-05,
      "step": 29981,
      "training_step_time": 7.411471366882324
    },
    {
      "epoch": 2.8593063354492187e-05,
      "model_forward_time": 1.7304487228393555,
      "step": 29982
    },
    {
      "epoch": 2.8593063354492187e-05,
      "step": 29982,
      "training_step_time": 7.201646566390991
    },
    {
      "epoch": 2.8594017028808594e-05,
      "model_forward_time": 1.7212905883789062,
      "step": 29983
    },
    {
      "epoch": 2.8594017028808594e-05,
      "step": 29983,
      "training_step_time": 8.300066947937012
    },
    {
      "epoch": 2.8594970703125e-05,
      "model_forward_time": 1.7290992736816406,
      "step": 29984
    },
    {
      "epoch": 2.8594970703125e-05,
      "step": 29984,
      "training_step_time": 8.50037932395935
    },
    {
      "epoch": 2.8595924377441406e-05,
      "model_forward_time": 1.7300066947937012,
      "step": 29985
    },
    {
      "epoch": 2.8595924377441406e-05,
      "step": 29985,
      "training_step_time": 9.092249631881714
    },
    {
      "epoch": 2.8596878051757813e-05,
      "model_forward_time": 2.2903189659118652,
      "step": 29986
    },
    {
      "epoch": 2.8596878051757813e-05,
      "step": 29986,
      "training_step_time": 7.818192481994629
    },
    {
      "epoch": 2.8597831726074217e-05,
      "model_forward_time": 1.725550889968872,
      "step": 29987
    },
    {
      "epoch": 2.8597831726074217e-05,
      "step": 29987,
      "training_step_time": 7.720987558364868
    },
    {
      "epoch": 2.8598785400390625e-05,
      "model_forward_time": 1.7305254936218262,
      "step": 29988
    },
    {
      "epoch": 2.8598785400390625e-05,
      "step": 29988,
      "training_step_time": 9.94519305229187
    },
    {
      "epoch": 2.8599739074707032e-05,
      "model_forward_time": 1.7264783382415771,
      "step": 29989
    },
    {
      "epoch": 2.8599739074707032e-05,
      "step": 29989,
      "training_step_time": 7.217511415481567
    },
    {
      "epoch": 2.8600692749023436e-05,
      "grad_norm": 0.17630723118782043,
      "learning_rate": 8.387745781056537e-06,
      "loss": 0.0601,
      "step": 29990
    },
    {
      "epoch": 2.8600692749023436e-05,
      "model_forward_time": 1.752225399017334,
      "step": 29990
    },
    {
      "dynamics_loss_avg": 0.053470654599368575,
      "epoch": 2.8600692749023436e-05,
      "step": 29990
    },
    {
      "action_loss_avg": 0.0019717678951565176,
      "epoch": 2.8600692749023436e-05,
      "step": 29990
    },
    {
      "epoch": 2.8600692749023436e-05,
      "step": 29990,
      "training_step_time": 7.480695724487305
    },
    {
      "epoch": 2.8601646423339843e-05,
      "model_forward_time": 1.7311980724334717,
      "step": 29991
    },
    {
      "epoch": 2.8601646423339843e-05,
      "step": 29991,
      "training_step_time": 7.125730514526367
    },
    {
      "epoch": 2.860260009765625e-05,
      "model_forward_time": 1.7312531471252441,
      "step": 29992
    },
    {
      "epoch": 2.860260009765625e-05,
      "step": 29992,
      "training_step_time": 7.655427932739258
    },
    {
      "epoch": 2.8603553771972655e-05,
      "model_forward_time": 1.7238037586212158,
      "step": 29993
    },
    {
      "epoch": 2.8603553771972655e-05,
      "step": 29993,
      "training_step_time": 8.351023435592651
    },
    {
      "epoch": 2.8604507446289062e-05,
      "model_forward_time": 1.7296876907348633,
      "step": 29994
    },
    {
      "epoch": 2.8604507446289062e-05,
      "step": 29994,
      "training_step_time": 8.552118062973022
    },
    {
      "epoch": 2.860546112060547e-05,
      "model_forward_time": 2.261056423187256,
      "step": 29995
    },
    {
      "epoch": 2.860546112060547e-05,
      "step": 29995,
      "training_step_time": 7.813213348388672
    },
    {
      "epoch": 2.8606414794921874e-05,
      "model_forward_time": 1.725299596786499,
      "step": 29996
    },
    {
      "epoch": 2.8606414794921874e-05,
      "step": 29996,
      "training_step_time": 8.746125221252441
    },
    {
      "epoch": 2.860736846923828e-05,
      "model_forward_time": 1.7483391761779785,
      "step": 29997
    },
    {
      "epoch": 2.860736846923828e-05,
      "step": 29997,
      "training_step_time": 7.859089374542236
    },
    {
      "epoch": 2.860832214355469e-05,
      "model_forward_time": 1.8142142295837402,
      "step": 29998
    },
    {
      "epoch": 2.860832214355469e-05,
      "step": 29998,
      "training_step_time": 8.034870386123657
    },
    {
      "epoch": 2.8609275817871093e-05,
      "model_forward_time": 1.724611520767212,
      "step": 29999
    },
    {
      "epoch": 2.8609275817871093e-05,
      "step": 29999,
      "training_step_time": 8.13072919845581
    },
    {
      "epoch": 2.86102294921875e-05,
      "grad_norm": 0.13105125725269318,
      "learning_rate": 8.386529506005066e-06,
      "loss": 0.0613,
      "step": 30000
    },
    {
      "epoch": 2.86102294921875e-05,
      "model_forward_time": 1.768705129623413,
      "step": 30000
    },
    {
      "dynamics_loss_avg": 0.04917270615696907,
      "epoch": 2.86102294921875e-05,
      "step": 30000
    },
    {
      "action_loss_avg": 0.0016153169912286104,
      "epoch": 2.86102294921875e-05,
      "step": 30000
    },
    {
      "epoch": 2.86102294921875e-05,
      "step": 30000,
      "training_step_time": 6.5970025062561035
    },
    {
      "epoch": 2.8611183166503907e-05,
      "model_forward_time": 1.7475268840789795,
      "step": 30001
    },
    {
      "epoch": 2.8611183166503907e-05,
      "step": 30001,
      "training_step_time": 7.650733232498169
    },
    {
      "epoch": 2.861213684082031e-05,
      "model_forward_time": 1.7347044944763184,
      "step": 30002
    },
    {
      "epoch": 2.861213684082031e-05,
      "step": 30002,
      "training_step_time": 9.212775230407715
    },
    {
      "epoch": 2.861309051513672e-05,
      "model_forward_time": 1.721158504486084,
      "step": 30003
    },
    {
      "epoch": 2.861309051513672e-05,
      "step": 30003,
      "training_step_time": 9.533322095870972
    },
    {
      "epoch": 2.8614044189453126e-05,
      "model_forward_time": 3.2682855129241943,
      "step": 30004
    },
    {
      "epoch": 2.8614044189453126e-05,
      "step": 30004,
      "training_step_time": 10.981175184249878
    },
    {
      "epoch": 2.861499786376953e-05,
      "model_forward_time": 1.7066981792449951,
      "step": 30005
    },
    {
      "epoch": 2.861499786376953e-05,
      "step": 30005,
      "training_step_time": 8.554633378982544
    },
    {
      "epoch": 2.8615951538085938e-05,
      "model_forward_time": 1.7283804416656494,
      "step": 30006
    },
    {
      "epoch": 2.8615951538085938e-05,
      "step": 30006,
      "training_step_time": 7.680720090866089
    },
    {
      "epoch": 2.8616905212402345e-05,
      "model_forward_time": 1.7434923648834229,
      "step": 30007
    },
    {
      "epoch": 2.8616905212402345e-05,
      "step": 30007,
      "training_step_time": 7.816186428070068
    },
    {
      "epoch": 2.861785888671875e-05,
      "model_forward_time": 1.7272472381591797,
      "step": 30008
    },
    {
      "epoch": 2.861785888671875e-05,
      "step": 30008,
      "training_step_time": 7.379945278167725
    },
    {
      "epoch": 2.8618812561035157e-05,
      "model_forward_time": 1.7285499572753906,
      "step": 30009
    },
    {
      "epoch": 2.8618812561035157e-05,
      "step": 30009,
      "training_step_time": 7.061063289642334
    },
    {
      "epoch": 2.8619766235351564e-05,
      "grad_norm": 0.16663497686386108,
      "learning_rate": 8.385312860607821e-06,
      "loss": 0.0654,
      "step": 30010
    },
    {
      "epoch": 2.8619766235351564e-05,
      "model_forward_time": 1.7237648963928223,
      "step": 30010
    },
    {
      "dynamics_loss_avg": 0.05547346044331789,
      "epoch": 2.8619766235351564e-05,
      "step": 30010
    },
    {
      "action_loss_avg": 0.001921177317854017,
      "epoch": 2.8619766235351564e-05,
      "step": 30010
    },
    {
      "epoch": 2.8619766235351564e-05,
      "step": 30010,
      "training_step_time": 8.106504201889038
    },
    {
      "epoch": 2.8620719909667968e-05,
      "model_forward_time": 1.7249045372009277,
      "step": 30011
    },
    {
      "epoch": 2.8620719909667968e-05,
      "step": 30011,
      "training_step_time": 8.065935850143433
    },
    {
      "epoch": 2.8621673583984375e-05,
      "model_forward_time": 1.723646640777588,
      "step": 30012
    },
    {
      "epoch": 2.8621673583984375e-05,
      "step": 30012,
      "training_step_time": 8.579109907150269
    },
    {
      "epoch": 2.8622627258300783e-05,
      "model_forward_time": 2.8424525260925293,
      "step": 30013
    },
    {
      "epoch": 2.8622627258300783e-05,
      "step": 30013,
      "training_step_time": 9.844517946243286
    },
    {
      "epoch": 2.8623580932617187e-05,
      "model_forward_time": 1.724259853363037,
      "step": 30014
    },
    {
      "epoch": 2.8623580932617187e-05,
      "step": 30014,
      "training_step_time": 8.618373155593872
    },
    {
      "epoch": 2.8624534606933594e-05,
      "model_forward_time": 1.7246074676513672,
      "step": 30015
    },
    {
      "epoch": 2.8624534606933594e-05,
      "step": 30015,
      "training_step_time": 7.703153133392334
    },
    {
      "epoch": 2.862548828125e-05,
      "model_forward_time": 1.3560965061187744,
      "step": 30016
    },
    {
      "epoch": 2.862548828125e-05,
      "step": 30016,
      "training_step_time": 7.718135833740234
    },
    {
      "epoch": 2.8626441955566406e-05,
      "model_forward_time": 1.7314162254333496,
      "step": 30017
    },
    {
      "epoch": 2.8626441955566406e-05,
      "step": 30017,
      "training_step_time": 7.147404432296753
    },
    {
      "epoch": 2.8627395629882813e-05,
      "model_forward_time": 1.7352964878082275,
      "step": 30018
    },
    {
      "epoch": 2.8627395629882813e-05,
      "step": 30018,
      "training_step_time": 7.387775659561157
    },
    {
      "epoch": 2.8628349304199217e-05,
      "model_forward_time": 1.7276933193206787,
      "step": 30019
    },
    {
      "epoch": 2.8628349304199217e-05,
      "step": 30019,
      "training_step_time": 7.825798749923706
    },
    {
      "epoch": 2.8629302978515625e-05,
      "grad_norm": 0.15574969351291656,
      "learning_rate": 8.384095844997849e-06,
      "loss": 0.062,
      "step": 30020
    },
    {
      "epoch": 2.8629302978515625e-05,
      "model_forward_time": 1.7225921154022217,
      "step": 30020
    },
    {
      "dynamics_loss_avg": 0.06303579043596982,
      "epoch": 2.8629302978515625e-05,
      "step": 30020
    },
    {
      "action_loss_avg": 0.002030645025661215,
      "epoch": 2.8629302978515625e-05,
      "step": 30020
    },
    {
      "epoch": 2.8629302978515625e-05,
      "step": 30020,
      "training_step_time": 8.362491130828857
    },
    {
      "epoch": 2.8630256652832032e-05,
      "model_forward_time": 1.7315423488616943,
      "step": 30021
    },
    {
      "epoch": 2.8630256652832032e-05,
      "step": 30021,
      "training_step_time": 10.618609428405762
    },
    {
      "epoch": 2.8631210327148436e-05,
      "model_forward_time": 1.7611515522003174,
      "step": 30022
    },
    {
      "epoch": 2.8631210327148436e-05,
      "step": 30022,
      "training_step_time": 8.418442964553833
    },
    {
      "epoch": 2.8632164001464843e-05,
      "model_forward_time": 2.9408979415893555,
      "step": 30023
    },
    {
      "epoch": 2.8632164001464843e-05,
      "step": 30023,
      "training_step_time": 7.72580623626709
    },
    {
      "epoch": 2.863311767578125e-05,
      "model_forward_time": 1.729429006576538,
      "step": 30024
    },
    {
      "epoch": 2.863311767578125e-05,
      "step": 30024,
      "training_step_time": 8.279063701629639
    },
    {
      "epoch": 2.8634071350097655e-05,
      "model_forward_time": 1.7251043319702148,
      "step": 30025
    },
    {
      "epoch": 2.8634071350097655e-05,
      "step": 30025,
      "training_step_time": 7.868000745773315
    },
    {
      "epoch": 2.8635025024414062e-05,
      "model_forward_time": 1.7168188095092773,
      "step": 30026
    },
    {
      "epoch": 2.8635025024414062e-05,
      "step": 30026,
      "training_step_time": 7.55226469039917
    },
    {
      "epoch": 2.863597869873047e-05,
      "model_forward_time": 1.724397897720337,
      "step": 30027
    },
    {
      "epoch": 2.863597869873047e-05,
      "step": 30027,
      "training_step_time": 7.062389612197876
    },
    {
      "epoch": 2.8636932373046874e-05,
      "model_forward_time": 1.7388079166412354,
      "step": 30028
    },
    {
      "epoch": 2.8636932373046874e-05,
      "step": 30028,
      "training_step_time": 7.229531288146973
    },
    {
      "epoch": 2.863788604736328e-05,
      "model_forward_time": 1.7500572204589844,
      "step": 30029
    },
    {
      "epoch": 2.863788604736328e-05,
      "step": 30029,
      "training_step_time": 7.55879545211792
    },
    {
      "epoch": 2.863883972167969e-05,
      "grad_norm": 0.16091032326221466,
      "learning_rate": 8.382878459308246e-06,
      "loss": 0.0647,
      "step": 30030
    },
    {
      "epoch": 2.863883972167969e-05,
      "model_forward_time": 1.72393798828125,
      "step": 30030
    },
    {
      "dynamics_loss_avg": 0.055154654942452905,
      "epoch": 2.863883972167969e-05,
      "step": 30030
    },
    {
      "action_loss_avg": 0.003047146866447292,
      "epoch": 2.863883972167969e-05,
      "step": 30030
    },
    {
      "epoch": 2.863883972167969e-05,
      "step": 30030,
      "training_step_time": 7.4270429611206055
    },
    {
      "epoch": 2.8639793395996093e-05,
      "model_forward_time": 1.388035535812378,
      "step": 30031
    },
    {
      "epoch": 2.8639793395996093e-05,
      "step": 30031,
      "training_step_time": 9.12447190284729
    },
    {
      "epoch": 2.86407470703125e-05,
      "model_forward_time": 2.8865089416503906,
      "step": 30032
    },
    {
      "epoch": 2.86407470703125e-05,
      "step": 30032,
      "training_step_time": 7.654252290725708
    },
    {
      "epoch": 2.8641700744628907e-05,
      "model_forward_time": 1.7254209518432617,
      "step": 30033
    },
    {
      "epoch": 2.8641700744628907e-05,
      "step": 30033,
      "training_step_time": 8.177706718444824
    },
    {
      "epoch": 2.864265441894531e-05,
      "model_forward_time": 1.7377309799194336,
      "step": 30034
    },
    {
      "epoch": 2.864265441894531e-05,
      "step": 30034,
      "training_step_time": 8.974241256713867
    },
    {
      "epoch": 2.864360809326172e-05,
      "model_forward_time": 1.7241361141204834,
      "step": 30035
    },
    {
      "epoch": 2.864360809326172e-05,
      "step": 30035,
      "training_step_time": 7.659740447998047
    },
    {
      "epoch": 2.8644561767578126e-05,
      "model_forward_time": 1.7398102283477783,
      "step": 30036
    },
    {
      "epoch": 2.8644561767578126e-05,
      "step": 30036,
      "training_step_time": 6.753648519515991
    },
    {
      "epoch": 2.864551544189453e-05,
      "model_forward_time": 1.7325661182403564,
      "step": 30037
    },
    {
      "epoch": 2.864551544189453e-05,
      "step": 30037,
      "training_step_time": 7.590442180633545
    },
    {
      "epoch": 2.8646469116210938e-05,
      "model_forward_time": 1.7502658367156982,
      "step": 30038
    },
    {
      "epoch": 2.8646469116210938e-05,
      "step": 30038,
      "training_step_time": 7.567044734954834
    },
    {
      "epoch": 2.8647422790527345e-05,
      "model_forward_time": 1.730997085571289,
      "step": 30039
    },
    {
      "epoch": 2.8647422790527345e-05,
      "step": 30039,
      "training_step_time": 9.139892339706421
    },
    {
      "epoch": 2.864837646484375e-05,
      "grad_norm": 0.20151786506175995,
      "learning_rate": 8.381660703672138e-06,
      "loss": 0.0644,
      "step": 30040
    },
    {
      "epoch": 2.864837646484375e-05,
      "model_forward_time": 1.7552664279937744,
      "step": 30040
    },
    {
      "dynamics_loss_avg": 0.04613782912492752,
      "epoch": 2.864837646484375e-05,
      "step": 30040
    },
    {
      "action_loss_avg": 0.0014181594189722091,
      "epoch": 2.864837646484375e-05,
      "step": 30040
    },
    {
      "epoch": 2.864837646484375e-05,
      "step": 30040,
      "training_step_time": 9.658265829086304
    },
    {
      "epoch": 2.8649330139160157e-05,
      "model_forward_time": 2.8742003440856934,
      "step": 30041
    },
    {
      "epoch": 2.8649330139160157e-05,
      "step": 30041,
      "training_step_time": 9.318519592285156
    },
    {
      "epoch": 2.8650283813476564e-05,
      "model_forward_time": 1.3556036949157715,
      "step": 30042
    },
    {
      "epoch": 2.8650283813476564e-05,
      "step": 30042,
      "training_step_time": 8.666881561279297
    },
    {
      "epoch": 2.8651237487792968e-05,
      "model_forward_time": 1.7174761295318604,
      "step": 30043
    },
    {
      "epoch": 2.8651237487792968e-05,
      "step": 30043,
      "training_step_time": 7.833131790161133
    },
    {
      "epoch": 2.8652191162109375e-05,
      "model_forward_time": 1.7258117198944092,
      "step": 30044
    },
    {
      "epoch": 2.8652191162109375e-05,
      "step": 30044,
      "training_step_time": 7.789204835891724
    },
    {
      "epoch": 2.8653144836425783e-05,
      "model_forward_time": 1.7164921760559082,
      "step": 30045
    },
    {
      "epoch": 2.8653144836425783e-05,
      "step": 30045,
      "training_step_time": 7.005436897277832
    },
    {
      "epoch": 2.8654098510742187e-05,
      "model_forward_time": 1.7186684608459473,
      "step": 30046
    },
    {
      "epoch": 2.8654098510742187e-05,
      "step": 30046,
      "training_step_time": 7.642978668212891
    },
    {
      "epoch": 2.8655052185058594e-05,
      "model_forward_time": 1.7398080825805664,
      "step": 30047
    },
    {
      "epoch": 2.8655052185058594e-05,
      "step": 30047,
      "training_step_time": 8.470503807067871
    },
    {
      "epoch": 2.8656005859375e-05,
      "model_forward_time": 1.7202138900756836,
      "step": 30048
    },
    {
      "epoch": 2.8656005859375e-05,
      "step": 30048,
      "training_step_time": 8.413247346878052
    },
    {
      "epoch": 2.8656959533691406e-05,
      "model_forward_time": 1.7440695762634277,
      "step": 30049
    },
    {
      "epoch": 2.8656959533691406e-05,
      "step": 30049,
      "training_step_time": 8.339076519012451
    },
    {
      "epoch": 2.8657913208007813e-05,
      "grad_norm": 0.21722523868083954,
      "learning_rate": 8.380442578222701e-06,
      "loss": 0.0668,
      "step": 30050
    },
    {
      "epoch": 2.8657913208007813e-05,
      "model_forward_time": 1.7207574844360352,
      "step": 30050
    },
    {
      "dynamics_loss_avg": 0.05073646446689963,
      "epoch": 2.8657913208007813e-05,
      "step": 30050
    },
    {
      "action_loss_avg": 0.002211108039773535,
      "epoch": 2.8657913208007813e-05,
      "step": 30050
    },
    {
      "epoch": 2.8657913208007813e-05,
      "step": 30050,
      "training_step_time": 9.625728368759155
    },
    {
      "epoch": 2.8658866882324217e-05,
      "model_forward_time": 1.7214465141296387,
      "step": 30051
    },
    {
      "epoch": 2.8658866882324217e-05,
      "step": 30051,
      "training_step_time": 8.105754137039185
    },
    {
      "epoch": 2.8659820556640625e-05,
      "model_forward_time": 1.7233049869537354,
      "step": 30052
    },
    {
      "epoch": 2.8659820556640625e-05,
      "step": 30052,
      "training_step_time": 7.718520641326904
    },
    {
      "epoch": 2.8660774230957032e-05,
      "model_forward_time": 1.7204267978668213,
      "step": 30053
    },
    {
      "epoch": 2.8660774230957032e-05,
      "step": 30053,
      "training_step_time": 7.613176584243774
    },
    {
      "epoch": 2.8661727905273436e-05,
      "model_forward_time": 1.718998908996582,
      "step": 30054
    },
    {
      "epoch": 2.8661727905273436e-05,
      "step": 30054,
      "training_step_time": 6.81173038482666
    },
    {
      "epoch": 2.8662681579589843e-05,
      "model_forward_time": 1.7107913494110107,
      "step": 30055
    },
    {
      "epoch": 2.8662681579589843e-05,
      "step": 30055,
      "training_step_time": 7.428880453109741
    },
    {
      "epoch": 2.866363525390625e-05,
      "model_forward_time": 1.7311153411865234,
      "step": 30056
    },
    {
      "epoch": 2.866363525390625e-05,
      "step": 30056,
      "training_step_time": 7.160962820053101
    },
    {
      "epoch": 2.8664588928222655e-05,
      "model_forward_time": 1.7207701206207275,
      "step": 30057
    },
    {
      "epoch": 2.8664588928222655e-05,
      "step": 30057,
      "training_step_time": 7.526978492736816
    },
    {
      "epoch": 2.8665542602539062e-05,
      "model_forward_time": 1.7368130683898926,
      "step": 30058
    },
    {
      "epoch": 2.8665542602539062e-05,
      "step": 30058,
      "training_step_time": 8.498088359832764
    },
    {
      "epoch": 2.866649627685547e-05,
      "model_forward_time": 1.7188999652862549,
      "step": 30059
    },
    {
      "epoch": 2.866649627685547e-05,
      "step": 30059,
      "training_step_time": 8.48856806755066
    },
    {
      "epoch": 2.8667449951171874e-05,
      "grad_norm": 0.20685341954231262,
      "learning_rate": 8.379224083093148e-06,
      "loss": 0.0632,
      "step": 30060
    },
    {
      "epoch": 2.8667449951171874e-05,
      "model_forward_time": 2.912389039993286,
      "step": 30060
    },
    {
      "dynamics_loss_avg": 0.05745799280703068,
      "epoch": 2.8667449951171874e-05,
      "step": 30060
    },
    {
      "action_loss_avg": 0.0032232712721452115,
      "epoch": 2.8667449951171874e-05,
      "step": 30060
    },
    {
      "epoch": 2.8667449951171874e-05,
      "step": 30060,
      "training_step_time": 8.730952262878418
    },
    {
      "epoch": 2.866840362548828e-05,
      "model_forward_time": 1.7179086208343506,
      "step": 30061
    },
    {
      "epoch": 2.866840362548828e-05,
      "step": 30061,
      "training_step_time": 8.174673795700073
    },
    {
      "epoch": 2.866935729980469e-05,
      "model_forward_time": 1.7203361988067627,
      "step": 30062
    },
    {
      "epoch": 2.866935729980469e-05,
      "step": 30062,
      "training_step_time": 7.662594556808472
    },
    {
      "epoch": 2.8670310974121093e-05,
      "model_forward_time": 1.7317464351654053,
      "step": 30063
    },
    {
      "epoch": 2.8670310974121093e-05,
      "step": 30063,
      "training_step_time": 7.527719497680664
    },
    {
      "epoch": 2.86712646484375e-05,
      "model_forward_time": 1.7377145290374756,
      "step": 30064
    },
    {
      "epoch": 2.86712646484375e-05,
      "step": 30064,
      "training_step_time": 7.928052186965942
    },
    {
      "epoch": 2.8672218322753907e-05,
      "model_forward_time": 1.7179827690124512,
      "step": 30065
    },
    {
      "epoch": 2.8672218322753907e-05,
      "step": 30065,
      "training_step_time": 7.617919683456421
    },
    {
      "epoch": 2.867317199707031e-05,
      "model_forward_time": 1.7151551246643066,
      "step": 30066
    },
    {
      "epoch": 2.867317199707031e-05,
      "step": 30066,
      "training_step_time": 9.465611696243286
    },
    {
      "epoch": 2.867412567138672e-05,
      "model_forward_time": 1.8450465202331543,
      "step": 30067
    },
    {
      "epoch": 2.867412567138672e-05,
      "step": 30067,
      "training_step_time": 8.427682638168335
    },
    {
      "epoch": 2.8675079345703126e-05,
      "model_forward_time": 1.7123064994812012,
      "step": 30068
    },
    {
      "epoch": 2.8675079345703126e-05,
      "step": 30068,
      "training_step_time": 8.577497005462646
    },
    {
      "epoch": 2.867603302001953e-05,
      "model_forward_time": 2.8639349937438965,
      "step": 30069
    },
    {
      "epoch": 2.867603302001953e-05,
      "step": 30069,
      "training_step_time": 8.508227348327637
    },
    {
      "epoch": 2.8676986694335938e-05,
      "grad_norm": 0.16086764633655548,
      "learning_rate": 8.378005218416728e-06,
      "loss": 0.063,
      "step": 30070
    },
    {
      "epoch": 2.8676986694335938e-05,
      "model_forward_time": 1.7225263118743896,
      "step": 30070
    },
    {
      "dynamics_loss_avg": 0.06583125479519367,
      "epoch": 2.8676986694335938e-05,
      "step": 30070
    },
    {
      "action_loss_avg": 0.0036271188640967012,
      "epoch": 2.8676986694335938e-05,
      "step": 30070
    },
    {
      "epoch": 2.8676986694335938e-05,
      "step": 30070,
      "training_step_time": 7.618150949478149
    },
    {
      "epoch": 2.8677940368652345e-05,
      "model_forward_time": 1.714977502822876,
      "step": 30071
    },
    {
      "epoch": 2.8677940368652345e-05,
      "step": 30071,
      "training_step_time": 7.624737024307251
    },
    {
      "epoch": 2.867889404296875e-05,
      "model_forward_time": 1.714585781097412,
      "step": 30072
    },
    {
      "epoch": 2.867889404296875e-05,
      "step": 30072,
      "training_step_time": 7.441351890563965
    },
    {
      "epoch": 2.8679847717285157e-05,
      "model_forward_time": 1.719848394393921,
      "step": 30073
    },
    {
      "epoch": 2.8679847717285157e-05,
      "step": 30073,
      "training_step_time": 7.347397327423096
    },
    {
      "epoch": 2.8680801391601564e-05,
      "model_forward_time": 1.7200078964233398,
      "step": 30074
    },
    {
      "epoch": 2.8680801391601564e-05,
      "step": 30074,
      "training_step_time": 7.680467367172241
    },
    {
      "epoch": 2.8681755065917968e-05,
      "model_forward_time": 1.7246229648590088,
      "step": 30075
    },
    {
      "epoch": 2.8681755065917968e-05,
      "step": 30075,
      "training_step_time": 8.051181077957153
    },
    {
      "epoch": 2.8682708740234375e-05,
      "model_forward_time": 1.7290174961090088,
      "step": 30076
    },
    {
      "epoch": 2.8682708740234375e-05,
      "step": 30076,
      "training_step_time": 7.600605010986328
    },
    {
      "epoch": 2.8683662414550783e-05,
      "model_forward_time": 1.7126588821411133,
      "step": 30077
    },
    {
      "epoch": 2.8683662414550783e-05,
      "step": 30077,
      "training_step_time": 9.206312656402588
    },
    {
      "epoch": 2.8684616088867187e-05,
      "model_forward_time": 2.8187108039855957,
      "step": 30078
    },
    {
      "epoch": 2.8684616088867187e-05,
      "step": 30078,
      "training_step_time": 8.231240272521973
    },
    {
      "epoch": 2.8685569763183594e-05,
      "model_forward_time": 1.7078795433044434,
      "step": 30079
    },
    {
      "epoch": 2.8685569763183594e-05,
      "step": 30079,
      "training_step_time": 8.550047397613525
    },
    {
      "epoch": 2.86865234375e-05,
      "grad_norm": 0.12322249263525009,
      "learning_rate": 8.376785984326735e-06,
      "loss": 0.0578,
      "step": 30080
    },
    {
      "epoch": 2.86865234375e-05,
      "model_forward_time": 1.7058615684509277,
      "step": 30080
    },
    {
      "dynamics_loss_avg": 0.05757550708949566,
      "epoch": 2.86865234375e-05,
      "step": 30080
    },
    {
      "action_loss_avg": 0.001995681563857943,
      "epoch": 2.86865234375e-05,
      "step": 30080
    },
    {
      "epoch": 2.86865234375e-05,
      "step": 30080,
      "training_step_time": 7.682966470718384
    },
    {
      "epoch": 2.8687477111816406e-05,
      "model_forward_time": 1.7062866687774658,
      "step": 30081
    },
    {
      "epoch": 2.8687477111816406e-05,
      "step": 30081,
      "training_step_time": 7.641521453857422
    },
    {
      "epoch": 2.8688430786132813e-05,
      "model_forward_time": 1.6982126235961914,
      "step": 30082
    },
    {
      "epoch": 2.8688430786132813e-05,
      "step": 30082,
      "training_step_time": 7.586910247802734
    },
    {
      "epoch": 2.8689384460449217e-05,
      "model_forward_time": 1.7093985080718994,
      "step": 30083
    },
    {
      "epoch": 2.8689384460449217e-05,
      "step": 30083,
      "training_step_time": 7.675207614898682
    },
    {
      "epoch": 2.8690338134765625e-05,
      "model_forward_time": 1.725804090499878,
      "step": 30084
    },
    {
      "epoch": 2.8690338134765625e-05,
      "step": 30084,
      "training_step_time": 7.5301103591918945
    },
    {
      "epoch": 2.8691291809082032e-05,
      "model_forward_time": 1.7350716590881348,
      "step": 30085
    },
    {
      "epoch": 2.8691291809082032e-05,
      "step": 30085,
      "training_step_time": 8.775178670883179
    },
    {
      "epoch": 2.8692245483398436e-05,
      "model_forward_time": 1.7197864055633545,
      "step": 30086
    },
    {
      "epoch": 2.8692245483398436e-05,
      "step": 30086,
      "training_step_time": 10.032898426055908
    },
    {
      "epoch": 2.8693199157714844e-05,
      "model_forward_time": 2.8345043659210205,
      "step": 30087
    },
    {
      "epoch": 2.8693199157714844e-05,
      "step": 30087,
      "training_step_time": 7.673069715499878
    },
    {
      "epoch": 2.869415283203125e-05,
      "model_forward_time": 1.708991527557373,
      "step": 30088
    },
    {
      "epoch": 2.869415283203125e-05,
      "step": 30088,
      "training_step_time": 7.777352571487427
    },
    {
      "epoch": 2.8695106506347655e-05,
      "model_forward_time": 1.7051680088043213,
      "step": 30089
    },
    {
      "epoch": 2.8695106506347655e-05,
      "step": 30089,
      "training_step_time": 7.758447647094727
    },
    {
      "epoch": 2.8696060180664062e-05,
      "grad_norm": 0.2841937839984894,
      "learning_rate": 8.375566380956506e-06,
      "loss": 0.0619,
      "step": 30090
    },
    {
      "epoch": 2.8696060180664062e-05,
      "model_forward_time": 1.6966395378112793,
      "step": 30090
    },
    {
      "dynamics_loss_avg": 0.06128625348210335,
      "epoch": 2.8696060180664062e-05,
      "step": 30090
    },
    {
      "action_loss_avg": 0.0016126398579217494,
      "epoch": 2.8696060180664062e-05,
      "step": 30090
    },
    {
      "epoch": 2.8696060180664062e-05,
      "step": 30090,
      "training_step_time": 7.707845687866211
    },
    {
      "epoch": 2.869701385498047e-05,
      "model_forward_time": 1.7142424583435059,
      "step": 30091
    },
    {
      "epoch": 2.869701385498047e-05,
      "step": 30091,
      "training_step_time": 7.589753150939941
    },
    {
      "epoch": 2.8697967529296874e-05,
      "model_forward_time": 1.703537940979004,
      "step": 30092
    },
    {
      "epoch": 2.8697967529296874e-05,
      "step": 30092,
      "training_step_time": 7.548489093780518
    },
    {
      "epoch": 2.869892120361328e-05,
      "model_forward_time": 1.727168321609497,
      "step": 30093
    },
    {
      "epoch": 2.869892120361328e-05,
      "step": 30093,
      "training_step_time": 7.7083399295806885
    },
    {
      "epoch": 2.869987487792969e-05,
      "model_forward_time": 1.7344129085540771,
      "step": 30094
    },
    {
      "epoch": 2.869987487792969e-05,
      "step": 30094,
      "training_step_time": 7.6701273918151855
    },
    {
      "epoch": 2.8700828552246093e-05,
      "model_forward_time": 1.7155935764312744,
      "step": 30095
    },
    {
      "epoch": 2.8700828552246093e-05,
      "step": 30095,
      "training_step_time": 9.08664345741272
    },
    {
      "epoch": 2.87017822265625e-05,
      "model_forward_time": 2.7916767597198486,
      "step": 30096
    },
    {
      "epoch": 2.87017822265625e-05,
      "step": 30096,
      "training_step_time": 9.705932855606079
    },
    {
      "epoch": 2.8702735900878907e-05,
      "model_forward_time": 1.7349655628204346,
      "step": 30097
    },
    {
      "epoch": 2.8702735900878907e-05,
      "step": 30097,
      "training_step_time": 7.694962739944458
    },
    {
      "epoch": 2.870368957519531e-05,
      "model_forward_time": 1.741236925125122,
      "step": 30098
    },
    {
      "epoch": 2.870368957519531e-05,
      "step": 30098,
      "training_step_time": 8.599166870117188
    },
    {
      "epoch": 2.870464324951172e-05,
      "model_forward_time": 1.7386958599090576,
      "step": 30099
    },
    {
      "epoch": 2.870464324951172e-05,
      "step": 30099,
      "training_step_time": 7.69651985168457
    },
    {
      "epoch": 2.8705596923828126e-05,
      "grad_norm": 0.2785869240760803,
      "learning_rate": 8.37434640843941e-06,
      "loss": 0.0594,
      "step": 30100
    },
    {
      "epoch": 2.8705596923828126e-05,
      "model_forward_time": 1.7352163791656494,
      "step": 30100
    },
    {
      "dynamics_loss_avg": 0.05198254138231277,
      "epoch": 2.8705596923828126e-05,
      "step": 30100
    },
    {
      "action_loss_avg": 0.002319074055412784,
      "epoch": 2.8705596923828126e-05,
      "step": 30100
    },
    {
      "epoch": 2.8705596923828126e-05,
      "step": 30100,
      "training_step_time": 6.8962790966033936
    },
    {
      "epoch": 2.870655059814453e-05,
      "model_forward_time": 1.7130746841430664,
      "step": 30101
    },
    {
      "epoch": 2.870655059814453e-05,
      "step": 30101,
      "training_step_time": 7.676566123962402
    },
    {
      "epoch": 2.8707504272460938e-05,
      "model_forward_time": 1.7346057891845703,
      "step": 30102
    },
    {
      "epoch": 2.8707504272460938e-05,
      "step": 30102,
      "training_step_time": 8.267149448394775
    },
    {
      "epoch": 2.8708457946777345e-05,
      "model_forward_time": 1.7468147277832031,
      "step": 30103
    },
    {
      "epoch": 2.8708457946777345e-05,
      "step": 30103,
      "training_step_time": 8.681670188903809
    },
    {
      "epoch": 2.870941162109375e-05,
      "model_forward_time": 1.7296392917633057,
      "step": 30104
    },
    {
      "epoch": 2.870941162109375e-05,
      "step": 30104,
      "training_step_time": 7.710690021514893
    },
    {
      "epoch": 2.8710365295410157e-05,
      "model_forward_time": 1.369983434677124,
      "step": 30105
    },
    {
      "epoch": 2.8710365295410157e-05,
      "step": 30105,
      "training_step_time": 8.52978515625
    },
    {
      "epoch": 2.8711318969726564e-05,
      "model_forward_time": 2.8677291870117188,
      "step": 30106
    },
    {
      "epoch": 2.8711318969726564e-05,
      "step": 30106,
      "training_step_time": 7.72327446937561
    },
    {
      "epoch": 2.8712272644042968e-05,
      "model_forward_time": 1.7380175590515137,
      "step": 30107
    },
    {
      "epoch": 2.8712272644042968e-05,
      "step": 30107,
      "training_step_time": 8.561730146408081
    },
    {
      "epoch": 2.8713226318359376e-05,
      "model_forward_time": 1.730210304260254,
      "step": 30108
    },
    {
      "epoch": 2.8713226318359376e-05,
      "step": 30108,
      "training_step_time": 7.716708660125732
    },
    {
      "epoch": 2.8714179992675783e-05,
      "model_forward_time": 1.7314083576202393,
      "step": 30109
    },
    {
      "epoch": 2.8714179992675783e-05,
      "step": 30109,
      "training_step_time": 7.3436830043792725
    },
    {
      "epoch": 2.8715133666992187e-05,
      "grad_norm": 0.20254097878932953,
      "learning_rate": 8.373126066908868e-06,
      "loss": 0.0655,
      "step": 30110
    },
    {
      "epoch": 2.8715133666992187e-05,
      "model_forward_time": 1.732954740524292,
      "step": 30110
    },
    {
      "dynamics_loss_avg": 0.06288115698844195,
      "epoch": 2.8715133666992187e-05,
      "step": 30110
    },
    {
      "action_loss_avg": 0.003187156093190424,
      "epoch": 2.8715133666992187e-05,
      "step": 30110
    },
    {
      "epoch": 2.8715133666992187e-05,
      "step": 30110,
      "training_step_time": 8.113399505615234
    },
    {
      "epoch": 2.8716087341308594e-05,
      "model_forward_time": 1.7188527584075928,
      "step": 30111
    },
    {
      "epoch": 2.8716087341308594e-05,
      "step": 30111,
      "training_step_time": 7.6741015911102295
    },
    {
      "epoch": 2.8717041015625e-05,
      "model_forward_time": 1.7662687301635742,
      "step": 30112
    },
    {
      "epoch": 2.8717041015625e-05,
      "step": 30112,
      "training_step_time": 7.6710662841796875
    },
    {
      "epoch": 2.8717994689941406e-05,
      "model_forward_time": 1.73333740234375,
      "step": 30113
    },
    {
      "epoch": 2.8717994689941406e-05,
      "step": 30113,
      "training_step_time": 8.412575483322144
    },
    {
      "epoch": 2.8718948364257813e-05,
      "model_forward_time": 1.7239022254943848,
      "step": 30114
    },
    {
      "epoch": 2.8718948364257813e-05,
      "step": 30114,
      "training_step_time": 9.332250356674194
    },
    {
      "epoch": 2.8719902038574217e-05,
      "model_forward_time": 2.8901901245117188,
      "step": 30115
    },
    {
      "epoch": 2.8719902038574217e-05,
      "step": 30115,
      "training_step_time": 7.746703147888184
    },
    {
      "epoch": 2.8720855712890625e-05,
      "model_forward_time": 1.731964111328125,
      "step": 30116
    },
    {
      "epoch": 2.8720855712890625e-05,
      "step": 30116,
      "training_step_time": 7.690043687820435
    },
    {
      "epoch": 2.8721809387207032e-05,
      "model_forward_time": 1.7376351356506348,
      "step": 30117
    },
    {
      "epoch": 2.8721809387207032e-05,
      "step": 30117,
      "training_step_time": 8.385361194610596
    },
    {
      "epoch": 2.8722763061523436e-05,
      "model_forward_time": 1.7302663326263428,
      "step": 30118
    },
    {
      "epoch": 2.8722763061523436e-05,
      "step": 30118,
      "training_step_time": 7.314197301864624
    },
    {
      "epoch": 2.8723716735839844e-05,
      "model_forward_time": 1.7387573719024658,
      "step": 30119
    },
    {
      "epoch": 2.8723716735839844e-05,
      "step": 30119,
      "training_step_time": 8.20166802406311
    },
    {
      "epoch": 2.872467041015625e-05,
      "grad_norm": 0.12337897717952728,
      "learning_rate": 8.371905356498326e-06,
      "loss": 0.0617,
      "step": 30120
    },
    {
      "epoch": 2.872467041015625e-05,
      "model_forward_time": 1.719226598739624,
      "step": 30120
    },
    {
      "dynamics_loss_avg": 0.05407321490347385,
      "epoch": 2.872467041015625e-05,
      "step": 30120
    },
    {
      "action_loss_avg": 0.0014093740668613465,
      "epoch": 2.872467041015625e-05,
      "step": 30120
    },
    {
      "epoch": 2.872467041015625e-05,
      "step": 30120,
      "training_step_time": 7.698647737503052
    },
    {
      "epoch": 2.8725624084472655e-05,
      "model_forward_time": 1.7495944499969482,
      "step": 30121
    },
    {
      "epoch": 2.8725624084472655e-05,
      "step": 30121,
      "training_step_time": 9.528404951095581
    },
    {
      "epoch": 2.8726577758789062e-05,
      "model_forward_time": 1.7366094589233398,
      "step": 30122
    },
    {
      "epoch": 2.8726577758789062e-05,
      "step": 30122,
      "training_step_time": 10.01507019996643
    },
    {
      "epoch": 2.872753143310547e-05,
      "model_forward_time": 1.736135482788086,
      "step": 30123
    },
    {
      "epoch": 2.872753143310547e-05,
      "step": 30123,
      "training_step_time": 9.07723617553711
    },
    {
      "epoch": 2.8728485107421874e-05,
      "model_forward_time": 2.8783271312713623,
      "step": 30124
    },
    {
      "epoch": 2.8728485107421874e-05,
      "step": 30124,
      "training_step_time": 8.738210201263428
    },
    {
      "epoch": 2.872943878173828e-05,
      "model_forward_time": 1.7161715030670166,
      "step": 30125
    },
    {
      "epoch": 2.872943878173828e-05,
      "step": 30125,
      "training_step_time": 8.492019891738892
    },
    {
      "epoch": 2.873039245605469e-05,
      "model_forward_time": 1.7345545291900635,
      "step": 30126
    },
    {
      "epoch": 2.873039245605469e-05,
      "step": 30126,
      "training_step_time": 7.784962892532349
    },
    {
      "epoch": 2.8731346130371093e-05,
      "model_forward_time": 1.0306415557861328,
      "step": 30127
    },
    {
      "epoch": 2.8731346130371093e-05,
      "step": 30127,
      "training_step_time": 7.559280157089233
    },
    {
      "epoch": 2.87322998046875e-05,
      "model_forward_time": 1.7171361446380615,
      "step": 30128
    },
    {
      "epoch": 2.87322998046875e-05,
      "step": 30128,
      "training_step_time": 7.623521566390991
    },
    {
      "epoch": 2.8733253479003908e-05,
      "model_forward_time": 1.7301485538482666,
      "step": 30129
    },
    {
      "epoch": 2.8733253479003908e-05,
      "step": 30129,
      "training_step_time": 7.6427531242370605
    },
    {
      "epoch": 2.873420715332031e-05,
      "grad_norm": 0.16429933905601501,
      "learning_rate": 8.370684277341288e-06,
      "loss": 0.0617,
      "step": 30130
    },
    {
      "epoch": 2.873420715332031e-05,
      "model_forward_time": 1.3891069889068604,
      "step": 30130
    },
    {
      "dynamics_loss_avg": 0.05374221112579107,
      "epoch": 2.873420715332031e-05,
      "step": 30130
    },
    {
      "action_loss_avg": 0.0021370893795392477,
      "epoch": 2.873420715332031e-05,
      "step": 30130
    },
    {
      "epoch": 2.873420715332031e-05,
      "step": 30130,
      "training_step_time": 7.613965749740601
    },
    {
      "epoch": 2.873516082763672e-05,
      "model_forward_time": 1.7670636177062988,
      "step": 30131
    },
    {
      "epoch": 2.873516082763672e-05,
      "step": 30131,
      "training_step_time": 7.710321426391602
    },
    {
      "epoch": 2.8736114501953126e-05,
      "model_forward_time": 1.7301604747772217,
      "step": 30132
    },
    {
      "epoch": 2.8736114501953126e-05,
      "step": 30132,
      "training_step_time": 8.72900652885437
    },
    {
      "epoch": 2.873706817626953e-05,
      "model_forward_time": 1.7312180995941162,
      "step": 30133
    },
    {
      "epoch": 2.873706817626953e-05,
      "step": 30133,
      "training_step_time": 9.17499589920044
    },
    {
      "epoch": 2.8738021850585938e-05,
      "model_forward_time": 2.8789570331573486,
      "step": 30134
    },
    {
      "epoch": 2.8738021850585938e-05,
      "step": 30134,
      "training_step_time": 8.53432035446167
    },
    {
      "epoch": 2.8738975524902345e-05,
      "model_forward_time": 1.7283656597137451,
      "step": 30135
    },
    {
      "epoch": 2.8738975524902345e-05,
      "step": 30135,
      "training_step_time": 7.782310247421265
    },
    {
      "epoch": 2.873992919921875e-05,
      "model_forward_time": 1.727400302886963,
      "step": 30136
    },
    {
      "epoch": 2.873992919921875e-05,
      "step": 30136,
      "training_step_time": 7.638283967971802
    },
    {
      "epoch": 2.8740882873535157e-05,
      "model_forward_time": 1.7332451343536377,
      "step": 30137
    },
    {
      "epoch": 2.8740882873535157e-05,
      "step": 30137,
      "training_step_time": 7.598332405090332
    },
    {
      "epoch": 2.8741836547851564e-05,
      "model_forward_time": 1.7346856594085693,
      "step": 30138
    },
    {
      "epoch": 2.8741836547851564e-05,
      "step": 30138,
      "training_step_time": 7.482968330383301
    },
    {
      "epoch": 2.8742790222167968e-05,
      "model_forward_time": 1.7522554397583008,
      "step": 30139
    },
    {
      "epoch": 2.8742790222167968e-05,
      "step": 30139,
      "training_step_time": 8.732871532440186
    },
    {
      "epoch": 2.8743743896484376e-05,
      "grad_norm": 0.1599133163690567,
      "learning_rate": 8.369462829571283e-06,
      "loss": 0.0623,
      "step": 30140
    },
    {
      "epoch": 2.8743743896484376e-05,
      "model_forward_time": 1.7415664196014404,
      "step": 30140
    },
    {
      "dynamics_loss_avg": 0.06712031401693822,
      "epoch": 2.8743743896484376e-05,
      "step": 30140
    },
    {
      "action_loss_avg": 0.0024830407288391142,
      "epoch": 2.8743743896484376e-05,
      "step": 30140
    },
    {
      "epoch": 2.8743743896484376e-05,
      "step": 30140,
      "training_step_time": 7.656256198883057
    },
    {
      "epoch": 2.874469757080078e-05,
      "model_forward_time": 1.726243495941162,
      "step": 30141
    },
    {
      "epoch": 2.874469757080078e-05,
      "step": 30141,
      "training_step_time": 7.673222303390503
    },
    {
      "epoch": 2.8745651245117187e-05,
      "model_forward_time": 1.727250099182129,
      "step": 30142
    },
    {
      "epoch": 2.8745651245117187e-05,
      "step": 30142,
      "training_step_time": 8.78976321220398
    },
    {
      "epoch": 2.8746604919433594e-05,
      "model_forward_time": 2.8860082626342773,
      "step": 30143
    },
    {
      "epoch": 2.8746604919433594e-05,
      "step": 30143,
      "training_step_time": 7.746860027313232
    },
    {
      "epoch": 2.874755859375e-05,
      "model_forward_time": 1.7410597801208496,
      "step": 30144
    },
    {
      "epoch": 2.874755859375e-05,
      "step": 30144,
      "training_step_time": 8.85010290145874
    },
    {
      "epoch": 2.8748512268066406e-05,
      "model_forward_time": 1.729677677154541,
      "step": 30145
    },
    {
      "epoch": 2.8748512268066406e-05,
      "step": 30145,
      "training_step_time": 8.43685531616211
    },
    {
      "epoch": 2.8749465942382813e-05,
      "model_forward_time": 1.713742733001709,
      "step": 30146
    },
    {
      "epoch": 2.8749465942382813e-05,
      "step": 30146,
      "training_step_time": 6.884997606277466
    },
    {
      "epoch": 2.8750419616699217e-05,
      "model_forward_time": 1.73244047164917,
      "step": 30147
    },
    {
      "epoch": 2.8750419616699217e-05,
      "step": 30147,
      "training_step_time": 7.646989583969116
    },
    {
      "epoch": 2.8751373291015625e-05,
      "model_forward_time": 1.7498700618743896,
      "step": 30148
    },
    {
      "epoch": 2.8751373291015625e-05,
      "step": 30148,
      "training_step_time": 7.813899040222168
    },
    {
      "epoch": 2.8752326965332032e-05,
      "model_forward_time": 1.7514057159423828,
      "step": 30149
    },
    {
      "epoch": 2.8752326965332032e-05,
      "step": 30149,
      "training_step_time": 7.625326633453369
    },
    {
      "epoch": 2.8753280639648436e-05,
      "grad_norm": 0.10435540229082108,
      "learning_rate": 8.36824101332189e-06,
      "loss": 0.0632,
      "step": 30150
    },
    {
      "epoch": 2.8753280639648436e-05,
      "model_forward_time": 1.7193870544433594,
      "step": 30150
    },
    {
      "dynamics_loss_avg": 0.05572537761181593,
      "epoch": 2.8753280639648436e-05,
      "step": 30150
    },
    {
      "action_loss_avg": 0.0017594260716577992,
      "epoch": 2.8753280639648436e-05,
      "step": 30150
    },
    {
      "epoch": 2.8753280639648436e-05,
      "step": 30150,
      "training_step_time": 9.26310110092163
    },
    {
      "epoch": 2.8754234313964844e-05,
      "model_forward_time": 1.7320849895477295,
      "step": 30151
    },
    {
      "epoch": 2.8754234313964844e-05,
      "step": 30151,
      "training_step_time": 9.058847188949585
    },
    {
      "epoch": 2.875518798828125e-05,
      "model_forward_time": 2.8664169311523438,
      "step": 30152
    },
    {
      "epoch": 2.875518798828125e-05,
      "step": 30152,
      "training_step_time": 9.103893280029297
    },
    {
      "epoch": 2.8756141662597655e-05,
      "model_forward_time": 1.7381508350372314,
      "step": 30153
    },
    {
      "epoch": 2.8756141662597655e-05,
      "step": 30153,
      "training_step_time": 7.704118967056274
    },
    {
      "epoch": 2.8757095336914062e-05,
      "model_forward_time": 1.7292070388793945,
      "step": 30154
    },
    {
      "epoch": 2.8757095336914062e-05,
      "step": 30154,
      "training_step_time": 7.745009183883667
    },
    {
      "epoch": 2.875804901123047e-05,
      "model_forward_time": 1.7322149276733398,
      "step": 30155
    },
    {
      "epoch": 2.875804901123047e-05,
      "step": 30155,
      "training_step_time": 7.281466007232666
    },
    {
      "epoch": 2.8759002685546874e-05,
      "model_forward_time": 1.7356817722320557,
      "step": 30156
    },
    {
      "epoch": 2.8759002685546874e-05,
      "step": 30156,
      "training_step_time": 7.623660326004028
    },
    {
      "epoch": 2.875995635986328e-05,
      "model_forward_time": 1.76021146774292,
      "step": 30157
    },
    {
      "epoch": 2.875995635986328e-05,
      "step": 30157,
      "training_step_time": 7.685765027999878
    },
    {
      "epoch": 2.876091003417969e-05,
      "model_forward_time": 1.7318828105926514,
      "step": 30158
    },
    {
      "epoch": 2.876091003417969e-05,
      "step": 30158,
      "training_step_time": 9.43345022201538
    },
    {
      "epoch": 2.8761863708496093e-05,
      "model_forward_time": 1.727919340133667,
      "step": 30159
    },
    {
      "epoch": 2.8761863708496093e-05,
      "step": 30159,
      "training_step_time": 7.6307291984558105
    },
    {
      "epoch": 2.87628173828125e-05,
      "grad_norm": 0.11596537381410599,
      "learning_rate": 8.367018828726721e-06,
      "loss": 0.0642,
      "step": 30160
    },
    {
      "epoch": 2.87628173828125e-05,
      "model_forward_time": 1.7223472595214844,
      "step": 30160
    },
    {
      "dynamics_loss_avg": 0.05397859439253807,
      "epoch": 2.87628173828125e-05,
      "step": 30160
    },
    {
      "action_loss_avg": 0.002091687230858952,
      "epoch": 2.87628173828125e-05,
      "step": 30160
    },
    {
      "epoch": 2.87628173828125e-05,
      "step": 30160,
      "training_step_time": 8.20986032485962
    },
    {
      "epoch": 2.8763771057128908e-05,
      "model_forward_time": 2.816422462463379,
      "step": 30161
    },
    {
      "epoch": 2.8763771057128908e-05,
      "step": 30161,
      "training_step_time": 9.28708267211914
    },
    {
      "epoch": 2.876472473144531e-05,
      "model_forward_time": 1.7353019714355469,
      "step": 30162
    },
    {
      "epoch": 2.876472473144531e-05,
      "step": 30162,
      "training_step_time": 7.756914854049683
    },
    {
      "epoch": 2.876567840576172e-05,
      "model_forward_time": 1.7331125736236572,
      "step": 30163
    },
    {
      "epoch": 2.876567840576172e-05,
      "step": 30163,
      "training_step_time": 7.63534688949585
    },
    {
      "epoch": 2.8766632080078126e-05,
      "model_forward_time": 1.7309670448303223,
      "step": 30164
    },
    {
      "epoch": 2.8766632080078126e-05,
      "step": 30164,
      "training_step_time": 7.286903142929077
    },
    {
      "epoch": 2.876758575439453e-05,
      "model_forward_time": 1.73807954788208,
      "step": 30165
    },
    {
      "epoch": 2.876758575439453e-05,
      "step": 30165,
      "training_step_time": 7.350444316864014
    },
    {
      "epoch": 2.8768539428710938e-05,
      "model_forward_time": 1.723616361618042,
      "step": 30166
    },
    {
      "epoch": 2.8768539428710938e-05,
      "step": 30166,
      "training_step_time": 7.642182350158691
    },
    {
      "epoch": 2.8769493103027345e-05,
      "model_forward_time": 1.7529075145721436,
      "step": 30167
    },
    {
      "epoch": 2.8769493103027345e-05,
      "step": 30167,
      "training_step_time": 8.540024757385254
    },
    {
      "epoch": 2.877044677734375e-05,
      "model_forward_time": 1.7611470222473145,
      "step": 30168
    },
    {
      "epoch": 2.877044677734375e-05,
      "step": 30168,
      "training_step_time": 8.668010234832764
    },
    {
      "epoch": 2.8771400451660157e-05,
      "model_forward_time": 1.7268569469451904,
      "step": 30169
    },
    {
      "epoch": 2.8771400451660157e-05,
      "step": 30169,
      "training_step_time": 7.667499303817749
    },
    {
      "epoch": 2.8772354125976564e-05,
      "grad_norm": 0.15014994144439697,
      "learning_rate": 8.365796275919437e-06,
      "loss": 0.0623,
      "step": 30170
    },
    {
      "epoch": 2.8772354125976564e-05,
      "model_forward_time": 1.7239646911621094,
      "step": 30170
    },
    {
      "dynamics_loss_avg": 0.044454066269099715,
      "epoch": 2.8772354125976564e-05,
      "step": 30170
    },
    {
      "action_loss_avg": 0.0019443878263700753,
      "epoch": 2.8772354125976564e-05,
      "step": 30170
    },
    {
      "epoch": 2.8772354125976564e-05,
      "step": 30170,
      "training_step_time": 8.500734090805054
    },
    {
      "epoch": 2.8773307800292968e-05,
      "model_forward_time": 2.925440549850464,
      "step": 30171
    },
    {
      "epoch": 2.8773307800292968e-05,
      "step": 30171,
      "training_step_time": 8.790706634521484
    },
    {
      "epoch": 2.8774261474609376e-05,
      "model_forward_time": 1.729480266571045,
      "step": 30172
    },
    {
      "epoch": 2.8774261474609376e-05,
      "step": 30172,
      "training_step_time": 7.666458368301392
    },
    {
      "epoch": 2.877521514892578e-05,
      "model_forward_time": 1.7408132553100586,
      "step": 30173
    },
    {
      "epoch": 2.877521514892578e-05,
      "step": 30173,
      "training_step_time": 7.355097770690918
    },
    {
      "epoch": 2.8776168823242187e-05,
      "model_forward_time": 1.7271580696105957,
      "step": 30174
    },
    {
      "epoch": 2.8776168823242187e-05,
      "step": 30174,
      "training_step_time": 6.886519193649292
    },
    {
      "epoch": 2.8777122497558594e-05,
      "model_forward_time": 1.74003005027771,
      "step": 30175
    },
    {
      "epoch": 2.8777122497558594e-05,
      "step": 30175,
      "training_step_time": 7.636059522628784
    },
    {
      "epoch": 2.8778076171875e-05,
      "model_forward_time": 1.756988525390625,
      "step": 30176
    },
    {
      "epoch": 2.8778076171875e-05,
      "step": 30176,
      "training_step_time": 7.663052320480347
    },
    {
      "epoch": 2.8779029846191406e-05,
      "model_forward_time": 1.7608973979949951,
      "step": 30177
    },
    {
      "epoch": 2.8779029846191406e-05,
      "step": 30177,
      "training_step_time": 7.667909383773804
    },
    {
      "epoch": 2.8779983520507813e-05,
      "model_forward_time": 1.731736421585083,
      "step": 30178
    },
    {
      "epoch": 2.8779983520507813e-05,
      "step": 30178,
      "training_step_time": 7.6399688720703125
    },
    {
      "epoch": 2.8780937194824217e-05,
      "model_forward_time": 1.717029333114624,
      "step": 30179
    },
    {
      "epoch": 2.8780937194824217e-05,
      "step": 30179,
      "training_step_time": 8.416063785552979
    },
    {
      "epoch": 2.8781890869140625e-05,
      "grad_norm": 0.13604727387428284,
      "learning_rate": 8.364573355033733e-06,
      "loss": 0.0652,
      "step": 30180
    },
    {
      "epoch": 2.8781890869140625e-05,
      "model_forward_time": 2.886957883834839,
      "step": 30180
    },
    {
      "dynamics_loss_avg": 0.0659462008625269,
      "epoch": 2.8781890869140625e-05,
      "step": 30180
    },
    {
      "action_loss_avg": 0.003080712066730484,
      "epoch": 2.8781890869140625e-05,
      "step": 30180
    },
    {
      "epoch": 2.8781890869140625e-05,
      "step": 30180,
      "training_step_time": 7.770951509475708
    },
    {
      "epoch": 2.8782844543457032e-05,
      "model_forward_time": 1.7376456260681152,
      "step": 30181
    },
    {
      "epoch": 2.8782844543457032e-05,
      "step": 30181,
      "training_step_time": 7.63765025138855
    },
    {
      "epoch": 2.8783798217773436e-05,
      "model_forward_time": 1.7278118133544922,
      "step": 30182
    },
    {
      "epoch": 2.8783798217773436e-05,
      "step": 30182,
      "training_step_time": 7.684306621551514
    },
    {
      "epoch": 2.8784751892089844e-05,
      "model_forward_time": 1.7348289489746094,
      "step": 30183
    },
    {
      "epoch": 2.8784751892089844e-05,
      "step": 30183,
      "training_step_time": 7.328760623931885
    },
    {
      "epoch": 2.878570556640625e-05,
      "model_forward_time": 1.7365715503692627,
      "step": 30184
    },
    {
      "epoch": 2.878570556640625e-05,
      "step": 30184,
      "training_step_time": 7.623698949813843
    },
    {
      "epoch": 2.8786659240722655e-05,
      "model_forward_time": 1.7530484199523926,
      "step": 30185
    },
    {
      "epoch": 2.8786659240722655e-05,
      "step": 30185,
      "training_step_time": 7.621838092803955
    },
    {
      "epoch": 2.8787612915039062e-05,
      "model_forward_time": 1.7537074089050293,
      "step": 30186
    },
    {
      "epoch": 2.8787612915039062e-05,
      "step": 30186,
      "training_step_time": 7.588818788528442
    },
    {
      "epoch": 2.878856658935547e-05,
      "model_forward_time": 1.7289984226226807,
      "step": 30187
    },
    {
      "epoch": 2.878856658935547e-05,
      "step": 30187,
      "training_step_time": 8.217893123626709
    },
    {
      "epoch": 2.8789520263671874e-05,
      "model_forward_time": 1.7397022247314453,
      "step": 30188
    },
    {
      "epoch": 2.8789520263671874e-05,
      "step": 30188,
      "training_step_time": 7.6934654712677
    },
    {
      "epoch": 2.879047393798828e-05,
      "model_forward_time": 2.8766562938690186,
      "step": 30189
    },
    {
      "epoch": 2.879047393798828e-05,
      "step": 30189,
      "training_step_time": 7.721072196960449
    },
    {
      "epoch": 2.879142761230469e-05,
      "grad_norm": 0.12483975291252136,
      "learning_rate": 8.363350066203347e-06,
      "loss": 0.0633,
      "step": 30190
    },
    {
      "epoch": 2.879142761230469e-05,
      "model_forward_time": 1.734736680984497,
      "step": 30190
    },
    {
      "dynamics_loss_avg": 0.05692113768309355,
      "epoch": 2.879142761230469e-05,
      "step": 30190
    },
    {
      "action_loss_avg": 0.0021944004110991955,
      "epoch": 2.879142761230469e-05,
      "step": 30190
    },
    {
      "epoch": 2.879142761230469e-05,
      "step": 30190,
      "training_step_time": 8.500411987304688
    },
    {
      "epoch": 2.8792381286621093e-05,
      "model_forward_time": 1.736626386642456,
      "step": 30191
    },
    {
      "epoch": 2.8792381286621093e-05,
      "step": 30191,
      "training_step_time": 7.74436354637146
    },
    {
      "epoch": 2.87933349609375e-05,
      "model_forward_time": 1.7184736728668213,
      "step": 30192
    },
    {
      "epoch": 2.87933349609375e-05,
      "step": 30192,
      "training_step_time": 7.345842599868774
    },
    {
      "epoch": 2.8794288635253908e-05,
      "model_forward_time": 1.737722396850586,
      "step": 30193
    },
    {
      "epoch": 2.8794288635253908e-05,
      "step": 30193,
      "training_step_time": 7.582773923873901
    },
    {
      "epoch": 2.879524230957031e-05,
      "model_forward_time": 1.739793300628662,
      "step": 30194
    },
    {
      "epoch": 2.879524230957031e-05,
      "step": 30194,
      "training_step_time": 7.655318260192871
    },
    {
      "epoch": 2.879619598388672e-05,
      "model_forward_time": 1.767591953277588,
      "step": 30195
    },
    {
      "epoch": 2.879619598388672e-05,
      "step": 30195,
      "training_step_time": 7.619939565658569
    },
    {
      "epoch": 2.8797149658203126e-05,
      "model_forward_time": 1.354442834854126,
      "step": 30196
    },
    {
      "epoch": 2.8797149658203126e-05,
      "step": 30196,
      "training_step_time": 7.648133039474487
    },
    {
      "epoch": 2.879810333251953e-05,
      "model_forward_time": 1.7320318222045898,
      "step": 30197
    },
    {
      "epoch": 2.879810333251953e-05,
      "step": 30197,
      "training_step_time": 7.6820924282073975
    },
    {
      "epoch": 2.8799057006835938e-05,
      "model_forward_time": 2.8426711559295654,
      "step": 30198
    },
    {
      "epoch": 2.8799057006835938e-05,
      "step": 30198,
      "training_step_time": 7.791961193084717
    },
    {
      "epoch": 2.8800010681152345e-05,
      "model_forward_time": 1.727135419845581,
      "step": 30199
    },
    {
      "epoch": 2.8800010681152345e-05,
      "step": 30199,
      "training_step_time": 7.6272852420806885
    },
    {
      "epoch": 2.880096435546875e-05,
      "grad_norm": 0.17766274511814117,
      "learning_rate": 8.362126409562053e-06,
      "loss": 0.063,
      "step": 30200
    },
    {
      "epoch": 2.880096435546875e-05,
      "model_forward_time": 1.714224100112915,
      "step": 30200
    },
    {
      "dynamics_loss_avg": 0.06096534878015518,
      "epoch": 2.880096435546875e-05,
      "step": 30200
    },
    {
      "action_loss_avg": 0.00277939090738073,
      "epoch": 2.880096435546875e-05,
      "step": 30200
    },
    {
      "epoch": 2.880096435546875e-05,
      "step": 30200,
      "training_step_time": 7.661800384521484
    },
    {
      "epoch": 2.8801918029785157e-05,
      "model_forward_time": 1.723799705505371,
      "step": 30201
    },
    {
      "epoch": 2.8801918029785157e-05,
      "step": 30201,
      "training_step_time": 7.397507905960083
    },
    {
      "epoch": 2.8802871704101564e-05,
      "model_forward_time": 1.7314400672912598,
      "step": 30202
    },
    {
      "epoch": 2.8802871704101564e-05,
      "step": 30202,
      "training_step_time": 8.174192190170288
    },
    {
      "epoch": 2.8803825378417968e-05,
      "model_forward_time": 1.723172664642334,
      "step": 30203
    },
    {
      "epoch": 2.8803825378417968e-05,
      "step": 30203,
      "training_step_time": 8.10641074180603
    },
    {
      "epoch": 2.8804779052734376e-05,
      "model_forward_time": 1.7433385848999023,
      "step": 30204
    },
    {
      "epoch": 2.8804779052734376e-05,
      "step": 30204,
      "training_step_time": 8.608466625213623
    },
    {
      "epoch": 2.880573272705078e-05,
      "model_forward_time": 1.7199864387512207,
      "step": 30205
    },
    {
      "epoch": 2.880573272705078e-05,
      "step": 30205,
      "training_step_time": 9.528473615646362
    },
    {
      "epoch": 2.8806686401367187e-05,
      "model_forward_time": 1.715428113937378,
      "step": 30206
    },
    {
      "epoch": 2.8806686401367187e-05,
      "step": 30206,
      "training_step_time": 7.76254677772522
    },
    {
      "epoch": 2.8807640075683594e-05,
      "model_forward_time": 1.7245948314666748,
      "step": 30207
    },
    {
      "epoch": 2.8807640075683594e-05,
      "step": 30207,
      "training_step_time": 8.880942106246948
    },
    {
      "epoch": 2.880859375e-05,
      "model_forward_time": 1.7334063053131104,
      "step": 30208
    },
    {
      "epoch": 2.880859375e-05,
      "step": 30208,
      "training_step_time": 8.808704376220703
    },
    {
      "epoch": 2.8809547424316406e-05,
      "model_forward_time": 1.7191689014434814,
      "step": 30209
    },
    {
      "epoch": 2.8809547424316406e-05,
      "step": 30209,
      "training_step_time": 8.06432819366455
    },
    {
      "epoch": 2.8810501098632813e-05,
      "grad_norm": 0.19326043128967285,
      "learning_rate": 8.360902385243668e-06,
      "loss": 0.0652,
      "step": 30210
    },
    {
      "epoch": 2.8810501098632813e-05,
      "model_forward_time": 1.732135534286499,
      "step": 30210
    },
    {
      "dynamics_loss_avg": 0.0649828715249896,
      "epoch": 2.8810501098632813e-05,
      "step": 30210
    },
    {
      "action_loss_avg": 0.0024046899168752136,
      "epoch": 2.8810501098632813e-05,
      "step": 30210
    },
    {
      "epoch": 2.8810501098632813e-05,
      "step": 30210,
      "training_step_time": 7.703451871871948
    },
    {
      "epoch": 2.8811454772949217e-05,
      "model_forward_time": 1.713172197341919,
      "step": 30211
    },
    {
      "epoch": 2.8811454772949217e-05,
      "step": 30211,
      "training_step_time": 7.401098966598511
    },
    {
      "epoch": 2.8812408447265625e-05,
      "model_forward_time": 1.7214486598968506,
      "step": 30212
    },
    {
      "epoch": 2.8812408447265625e-05,
      "step": 30212,
      "training_step_time": 7.461758136749268
    },
    {
      "epoch": 2.8813362121582032e-05,
      "model_forward_time": 1.7995872497558594,
      "step": 30213
    },
    {
      "epoch": 2.8813362121582032e-05,
      "step": 30213,
      "training_step_time": 8.454104661941528
    },
    {
      "epoch": 2.8814315795898436e-05,
      "model_forward_time": 1.7313015460968018,
      "step": 30214
    },
    {
      "epoch": 2.8814315795898436e-05,
      "step": 30214,
      "training_step_time": 8.348206043243408
    },
    {
      "epoch": 2.8815269470214844e-05,
      "model_forward_time": 1.7208421230316162,
      "step": 30215
    },
    {
      "epoch": 2.8815269470214844e-05,
      "step": 30215,
      "training_step_time": 7.639043807983398
    },
    {
      "epoch": 2.881622314453125e-05,
      "model_forward_time": 1.7254548072814941,
      "step": 30216
    },
    {
      "epoch": 2.881622314453125e-05,
      "step": 30216,
      "training_step_time": 9.29718017578125
    },
    {
      "epoch": 2.8817176818847655e-05,
      "model_forward_time": 2.8982760906219482,
      "step": 30217
    },
    {
      "epoch": 2.8817176818847655e-05,
      "step": 30217,
      "training_step_time": 8.561401128768921
    },
    {
      "epoch": 2.8818130493164062e-05,
      "model_forward_time": 1.7215862274169922,
      "step": 30218
    },
    {
      "epoch": 2.8818130493164062e-05,
      "step": 30218,
      "training_step_time": 7.607062339782715
    },
    {
      "epoch": 2.881908416748047e-05,
      "model_forward_time": 1.730118989944458,
      "step": 30219
    },
    {
      "epoch": 2.881908416748047e-05,
      "step": 30219,
      "training_step_time": 7.857837200164795
    },
    {
      "epoch": 2.8820037841796874e-05,
      "grad_norm": 0.17090651392936707,
      "learning_rate": 8.359677993382052e-06,
      "loss": 0.0603,
      "step": 30220
    },
    {
      "epoch": 2.8820037841796874e-05,
      "model_forward_time": 1.7197744846343994,
      "step": 30220
    },
    {
      "dynamics_loss_avg": 0.05712557882070542,
      "epoch": 2.8820037841796874e-05,
      "step": 30220
    },
    {
      "action_loss_avg": 0.002081958006601781,
      "epoch": 2.8820037841796874e-05,
      "step": 30220
    },
    {
      "epoch": 2.8820037841796874e-05,
      "step": 30220,
      "training_step_time": 7.4388747215271
    },
    {
      "epoch": 2.882099151611328e-05,
      "model_forward_time": 1.7334675788879395,
      "step": 30221
    },
    {
      "epoch": 2.882099151611328e-05,
      "step": 30221,
      "training_step_time": 7.676929473876953
    },
    {
      "epoch": 2.882194519042969e-05,
      "model_forward_time": 1.7683825492858887,
      "step": 30222
    },
    {
      "epoch": 2.882194519042969e-05,
      "step": 30222,
      "training_step_time": 8.550113201141357
    },
    {
      "epoch": 2.8822898864746093e-05,
      "model_forward_time": 1.7200257778167725,
      "step": 30223
    },
    {
      "epoch": 2.8822898864746093e-05,
      "step": 30223,
      "training_step_time": 7.648004770278931
    },
    {
      "epoch": 2.88238525390625e-05,
      "model_forward_time": 1.7247636318206787,
      "step": 30224
    },
    {
      "epoch": 2.88238525390625e-05,
      "step": 30224,
      "training_step_time": 8.451537609100342
    },
    {
      "epoch": 2.8824806213378908e-05,
      "model_forward_time": 1.7240948677062988,
      "step": 30225
    },
    {
      "epoch": 2.8824806213378908e-05,
      "step": 30225,
      "training_step_time": 8.76110291481018
    },
    {
      "epoch": 2.882575988769531e-05,
      "model_forward_time": 2.88389253616333,
      "step": 30226
    },
    {
      "epoch": 2.882575988769531e-05,
      "step": 30226,
      "training_step_time": 7.914718866348267
    },
    {
      "epoch": 2.882671356201172e-05,
      "model_forward_time": 1.7275125980377197,
      "step": 30227
    },
    {
      "epoch": 2.882671356201172e-05,
      "step": 30227,
      "training_step_time": 7.646519422531128
    },
    {
      "epoch": 2.8827667236328126e-05,
      "model_forward_time": 1.725637435913086,
      "step": 30228
    },
    {
      "epoch": 2.8827667236328126e-05,
      "step": 30228,
      "training_step_time": 7.161490201950073
    },
    {
      "epoch": 2.882862091064453e-05,
      "model_forward_time": 1.7261731624603271,
      "step": 30229
    },
    {
      "epoch": 2.882862091064453e-05,
      "step": 30229,
      "training_step_time": 7.527961492538452
    },
    {
      "epoch": 2.8829574584960938e-05,
      "grad_norm": 0.11444814503192902,
      "learning_rate": 8.358453234111104e-06,
      "loss": 0.0626,
      "step": 30230
    },
    {
      "epoch": 2.8829574584960938e-05,
      "model_forward_time": 1.7401037216186523,
      "step": 30230
    },
    {
      "dynamics_loss_avg": 0.051404801104217766,
      "epoch": 2.8829574584960938e-05,
      "step": 30230
    },
    {
      "action_loss_avg": 0.0013759117980953306,
      "epoch": 2.8829574584960938e-05,
      "step": 30230
    },
    {
      "epoch": 2.8829574584960938e-05,
      "step": 30230,
      "training_step_time": 7.5435802936553955
    },
    {
      "epoch": 2.8830528259277345e-05,
      "model_forward_time": 1.7400472164154053,
      "step": 30231
    },
    {
      "epoch": 2.8830528259277345e-05,
      "step": 30231,
      "training_step_time": 8.785235404968262
    },
    {
      "epoch": 2.883148193359375e-05,
      "model_forward_time": 1.7434210777282715,
      "step": 30232
    },
    {
      "epoch": 2.883148193359375e-05,
      "step": 30232,
      "training_step_time": 7.646017789840698
    },
    {
      "epoch": 2.8832435607910157e-05,
      "model_forward_time": 1.7361631393432617,
      "step": 30233
    },
    {
      "epoch": 2.8832435607910157e-05,
      "step": 30233,
      "training_step_time": 9.009551048278809
    },
    {
      "epoch": 2.8833389282226564e-05,
      "model_forward_time": 1.7355964183807373,
      "step": 30234
    },
    {
      "epoch": 2.8833389282226564e-05,
      "step": 30234,
      "training_step_time": 8.790306806564331
    },
    {
      "epoch": 2.8834342956542968e-05,
      "model_forward_time": 2.283435583114624,
      "step": 30235
    },
    {
      "epoch": 2.8834342956542968e-05,
      "step": 30235,
      "training_step_time": 8.902536392211914
    },
    {
      "epoch": 2.8835296630859376e-05,
      "model_forward_time": 1.782433271408081,
      "step": 30236
    },
    {
      "epoch": 2.8835296630859376e-05,
      "step": 30236,
      "training_step_time": 7.814220428466797
    },
    {
      "epoch": 2.883625030517578e-05,
      "model_forward_time": 1.7438669204711914,
      "step": 30237
    },
    {
      "epoch": 2.883625030517578e-05,
      "step": 30237,
      "training_step_time": 7.7221739292144775
    },
    {
      "epoch": 2.8837203979492187e-05,
      "model_forward_time": 1.7741291522979736,
      "step": 30238
    },
    {
      "epoch": 2.8837203979492187e-05,
      "step": 30238,
      "training_step_time": 7.559529066085815
    },
    {
      "epoch": 2.8838157653808594e-05,
      "model_forward_time": 1.7405989170074463,
      "step": 30239
    },
    {
      "epoch": 2.8838157653808594e-05,
      "step": 30239,
      "training_step_time": 8.398486375808716
    },
    {
      "epoch": 2.8839111328125e-05,
      "grad_norm": 0.18408066034317017,
      "learning_rate": 8.357228107564756e-06,
      "loss": 0.059,
      "step": 30240
    },
    {
      "epoch": 2.8839111328125e-05,
      "model_forward_time": 1.7289443016052246,
      "step": 30240
    },
    {
      "dynamics_loss_avg": 0.04227426731958985,
      "epoch": 2.8839111328125e-05,
      "step": 30240
    },
    {
      "action_loss_avg": 0.0019897647900506852,
      "epoch": 2.8839111328125e-05,
      "step": 30240
    },
    {
      "epoch": 2.8839111328125e-05,
      "step": 30240,
      "training_step_time": 9.073925733566284
    },
    {
      "epoch": 2.8840065002441406e-05,
      "model_forward_time": 1.733457326889038,
      "step": 30241
    },
    {
      "epoch": 2.8840065002441406e-05,
      "step": 30241,
      "training_step_time": 8.424926519393921
    },
    {
      "epoch": 2.8841018676757813e-05,
      "model_forward_time": 1.7344744205474854,
      "step": 30242
    },
    {
      "epoch": 2.8841018676757813e-05,
      "step": 30242,
      "training_step_time": 7.6962339878082275
    },
    {
      "epoch": 2.8841972351074217e-05,
      "model_forward_time": 1.7362289428710938,
      "step": 30243
    },
    {
      "epoch": 2.8841972351074217e-05,
      "step": 30243,
      "training_step_time": 9.246692419052124
    },
    {
      "epoch": 2.8842926025390625e-05,
      "model_forward_time": 2.2998414039611816,
      "step": 30244
    },
    {
      "epoch": 2.8842926025390625e-05,
      "step": 30244,
      "training_step_time": 9.602166175842285
    },
    {
      "epoch": 2.8843879699707032e-05,
      "model_forward_time": 1.736778974533081,
      "step": 30245
    },
    {
      "epoch": 2.8843879699707032e-05,
      "step": 30245,
      "training_step_time": 7.904855251312256
    },
    {
      "epoch": 2.8844833374023436e-05,
      "model_forward_time": 1.7348482608795166,
      "step": 30246
    },
    {
      "epoch": 2.8844833374023436e-05,
      "step": 30246,
      "training_step_time": 8.171534776687622
    },
    {
      "epoch": 2.8845787048339844e-05,
      "model_forward_time": 1.7583084106445312,
      "step": 30247
    },
    {
      "epoch": 2.8845787048339844e-05,
      "step": 30247,
      "training_step_time": 7.469999074935913
    },
    {
      "epoch": 2.884674072265625e-05,
      "model_forward_time": 1.7365617752075195,
      "step": 30248
    },
    {
      "epoch": 2.884674072265625e-05,
      "step": 30248,
      "training_step_time": 8.175096273422241
    },
    {
      "epoch": 2.8847694396972655e-05,
      "model_forward_time": 1.7315001487731934,
      "step": 30249
    },
    {
      "epoch": 2.8847694396972655e-05,
      "step": 30249,
      "training_step_time": 7.449075698852539
    },
    {
      "epoch": 2.8848648071289062e-05,
      "grad_norm": 0.15623867511749268,
      "learning_rate": 8.356002613876993e-06,
      "loss": 0.058,
      "step": 30250
    },
    {
      "epoch": 2.8848648071289062e-05,
      "model_forward_time": 1.7173881530761719,
      "step": 30250
    },
    {
      "dynamics_loss_avg": 0.06512595470994711,
      "epoch": 2.8848648071289062e-05,
      "step": 30250
    },
    {
      "action_loss_avg": 0.0038282696972601117,
      "epoch": 2.8848648071289062e-05,
      "step": 30250
    },
    {
      "epoch": 2.8848648071289062e-05,
      "step": 30250,
      "training_step_time": 8.757184505462646
    },
    {
      "epoch": 2.884960174560547e-05,
      "model_forward_time": 1.7272655963897705,
      "step": 30251
    },
    {
      "epoch": 2.884960174560547e-05,
      "step": 30251,
      "training_step_time": 9.480016708374023
    },
    {
      "epoch": 2.8850555419921874e-05,
      "model_forward_time": 1.7321789264678955,
      "step": 30252
    },
    {
      "epoch": 2.8850555419921874e-05,
      "step": 30252,
      "training_step_time": 9.371153593063354
    },
    {
      "epoch": 2.885150909423828e-05,
      "model_forward_time": 2.284515857696533,
      "step": 30253
    },
    {
      "epoch": 2.885150909423828e-05,
      "step": 30253,
      "training_step_time": 7.805814266204834
    },
    {
      "epoch": 2.885246276855469e-05,
      "model_forward_time": 1.7425575256347656,
      "step": 30254
    },
    {
      "epoch": 2.885246276855469e-05,
      "step": 30254,
      "training_step_time": 8.674573183059692
    },
    {
      "epoch": 2.8853416442871093e-05,
      "model_forward_time": 1.7394509315490723,
      "step": 30255
    },
    {
      "epoch": 2.8853416442871093e-05,
      "step": 30255,
      "training_step_time": 8.545562982559204
    },
    {
      "epoch": 2.88543701171875e-05,
      "model_forward_time": 1.749218225479126,
      "step": 30256
    },
    {
      "epoch": 2.88543701171875e-05,
      "step": 30256,
      "training_step_time": 7.636617660522461
    },
    {
      "epoch": 2.8855323791503908e-05,
      "model_forward_time": 1.7462875843048096,
      "step": 30257
    },
    {
      "epoch": 2.8855323791503908e-05,
      "step": 30257,
      "training_step_time": 7.1418702602386475
    },
    {
      "epoch": 2.885627746582031e-05,
      "model_forward_time": 1.7353014945983887,
      "step": 30258
    },
    {
      "epoch": 2.885627746582031e-05,
      "step": 30258,
      "training_step_time": 7.624804496765137
    },
    {
      "epoch": 2.885723114013672e-05,
      "model_forward_time": 1.7337546348571777,
      "step": 30259
    },
    {
      "epoch": 2.885723114013672e-05,
      "step": 30259,
      "training_step_time": 7.756179332733154
    },
    {
      "epoch": 2.8858184814453126e-05,
      "grad_norm": 0.21624624729156494,
      "learning_rate": 8.35477675318183e-06,
      "loss": 0.0598,
      "step": 30260
    },
    {
      "epoch": 2.8858184814453126e-05,
      "model_forward_time": 1.7669763565063477,
      "step": 30260
    },
    {
      "dynamics_loss_avg": 0.06895461082458496,
      "epoch": 2.8858184814453126e-05,
      "step": 30260
    },
    {
      "action_loss_avg": 0.0034186986391432582,
      "epoch": 2.8858184814453126e-05,
      "step": 30260
    },
    {
      "epoch": 2.8858184814453126e-05,
      "step": 30260,
      "training_step_time": 9.29372787475586
    },
    {
      "epoch": 2.885913848876953e-05,
      "model_forward_time": 1.7433273792266846,
      "step": 30261
    },
    {
      "epoch": 2.885913848876953e-05,
      "step": 30261,
      "training_step_time": 8.660857200622559
    },
    {
      "epoch": 2.8860092163085938e-05,
      "model_forward_time": 1.7436728477478027,
      "step": 30262
    },
    {
      "epoch": 2.8860092163085938e-05,
      "step": 30262,
      "training_step_time": 9.402250289916992
    },
    {
      "epoch": 2.8861045837402345e-05,
      "model_forward_time": 1.3759138584136963,
      "step": 30263
    },
    {
      "epoch": 2.8861045837402345e-05,
      "step": 30263,
      "training_step_time": 7.858831882476807
    },
    {
      "epoch": 2.886199951171875e-05,
      "model_forward_time": 1.7449402809143066,
      "step": 30264
    },
    {
      "epoch": 2.886199951171875e-05,
      "step": 30264,
      "training_step_time": 7.613401412963867
    },
    {
      "epoch": 2.8862953186035157e-05,
      "model_forward_time": 1.7259571552276611,
      "step": 30265
    },
    {
      "epoch": 2.8862953186035157e-05,
      "step": 30265,
      "training_step_time": 7.701417922973633
    },
    {
      "epoch": 2.886390686035156e-05,
      "model_forward_time": 1.7692639827728271,
      "step": 30266
    },
    {
      "epoch": 2.886390686035156e-05,
      "step": 30266,
      "training_step_time": 7.293192148208618
    },
    {
      "epoch": 2.8864860534667968e-05,
      "model_forward_time": 1.7232892513275146,
      "step": 30267
    },
    {
      "epoch": 2.8864860534667968e-05,
      "step": 30267,
      "training_step_time": 7.615830421447754
    },
    {
      "epoch": 2.8865814208984376e-05,
      "model_forward_time": 1.7063961029052734,
      "step": 30268
    },
    {
      "epoch": 2.8865814208984376e-05,
      "step": 30268,
      "training_step_time": 7.746963262557983
    },
    {
      "epoch": 2.886676788330078e-05,
      "model_forward_time": 1.714829921722412,
      "step": 30269
    },
    {
      "epoch": 2.886676788330078e-05,
      "step": 30269,
      "training_step_time": 8.75345516204834
    },
    {
      "epoch": 2.8867721557617187e-05,
      "grad_norm": 0.1835341453552246,
      "learning_rate": 8.353550525613323e-06,
      "loss": 0.0562,
      "step": 30270
    },
    {
      "epoch": 2.8867721557617187e-05,
      "model_forward_time": 1.7183105945587158,
      "step": 30270
    },
    {
      "dynamics_loss_avg": 0.057384404726326464,
      "epoch": 2.8867721557617187e-05,
      "step": 30270
    },
    {
      "action_loss_avg": 0.002400836523156613,
      "epoch": 2.8867721557617187e-05,
      "step": 30270
    },
    {
      "epoch": 2.8867721557617187e-05,
      "step": 30270,
      "training_step_time": 8.226768732070923
    },
    {
      "epoch": 2.8868675231933595e-05,
      "model_forward_time": 1.718552589416504,
      "step": 30271
    },
    {
      "epoch": 2.8868675231933595e-05,
      "step": 30271,
      "training_step_time": 8.583361148834229
    },
    {
      "epoch": 2.886962890625e-05,
      "model_forward_time": 2.2684569358825684,
      "step": 30272
    },
    {
      "epoch": 2.886962890625e-05,
      "step": 30272,
      "training_step_time": 8.517631769180298
    },
    {
      "epoch": 2.8870582580566406e-05,
      "model_forward_time": 1.7182049751281738,
      "step": 30273
    },
    {
      "epoch": 2.8870582580566406e-05,
      "step": 30273,
      "training_step_time": 9.659265756607056
    },
    {
      "epoch": 2.8871536254882813e-05,
      "model_forward_time": 1.7134017944335938,
      "step": 30274
    },
    {
      "epoch": 2.8871536254882813e-05,
      "step": 30274,
      "training_step_time": 7.711479902267456
    },
    {
      "epoch": 2.8872489929199217e-05,
      "model_forward_time": 1.7667810916900635,
      "step": 30275
    },
    {
      "epoch": 2.8872489929199217e-05,
      "step": 30275,
      "training_step_time": 7.289727210998535
    },
    {
      "epoch": 2.8873443603515625e-05,
      "model_forward_time": 1.7224197387695312,
      "step": 30276
    },
    {
      "epoch": 2.8873443603515625e-05,
      "step": 30276,
      "training_step_time": 8.936266660690308
    },
    {
      "epoch": 2.8874397277832032e-05,
      "model_forward_time": 1.7244760990142822,
      "step": 30277
    },
    {
      "epoch": 2.8874397277832032e-05,
      "step": 30277,
      "training_step_time": 7.697908639907837
    },
    {
      "epoch": 2.8875350952148436e-05,
      "model_forward_time": 1.712001085281372,
      "step": 30278
    },
    {
      "epoch": 2.8875350952148436e-05,
      "step": 30278,
      "training_step_time": 8.896002531051636
    },
    {
      "epoch": 2.8876304626464844e-05,
      "model_forward_time": 1.7251172065734863,
      "step": 30279
    },
    {
      "epoch": 2.8876304626464844e-05,
      "step": 30279,
      "training_step_time": 8.765712261199951
    },
    {
      "epoch": 2.887725830078125e-05,
      "grad_norm": 0.14131832122802734,
      "learning_rate": 8.352323931305572e-06,
      "loss": 0.0596,
      "step": 30280
    },
    {
      "epoch": 2.887725830078125e-05,
      "model_forward_time": 1.723954200744629,
      "step": 30280
    },
    {
      "dynamics_loss_avg": 0.060791267454624175,
      "epoch": 2.887725830078125e-05,
      "step": 30280
    },
    {
      "action_loss_avg": 0.0027647327457088976,
      "epoch": 2.887725830078125e-05,
      "step": 30280
    },
    {
      "epoch": 2.887725830078125e-05,
      "step": 30280,
      "training_step_time": 8.157507181167603
    },
    {
      "epoch": 2.8878211975097655e-05,
      "model_forward_time": 2.259016990661621,
      "step": 30281
    },
    {
      "epoch": 2.8878211975097655e-05,
      "step": 30281,
      "training_step_time": 8.57431411743164
    },
    {
      "epoch": 2.8879165649414063e-05,
      "model_forward_time": 1.7199244499206543,
      "step": 30282
    },
    {
      "epoch": 2.8879165649414063e-05,
      "step": 30282,
      "training_step_time": 7.759871006011963
    },
    {
      "epoch": 2.888011932373047e-05,
      "model_forward_time": 1.7261207103729248,
      "step": 30283
    },
    {
      "epoch": 2.888011932373047e-05,
      "step": 30283,
      "training_step_time": 8.726712703704834
    },
    {
      "epoch": 2.8881072998046874e-05,
      "model_forward_time": 1.729524850845337,
      "step": 30284
    },
    {
      "epoch": 2.8881072998046874e-05,
      "step": 30284,
      "training_step_time": 7.304358243942261
    },
    {
      "epoch": 2.888202667236328e-05,
      "model_forward_time": 1.7170679569244385,
      "step": 30285
    },
    {
      "epoch": 2.888202667236328e-05,
      "step": 30285,
      "training_step_time": 7.643038749694824
    },
    {
      "epoch": 2.888298034667969e-05,
      "model_forward_time": 1.7125635147094727,
      "step": 30286
    },
    {
      "epoch": 2.888298034667969e-05,
      "step": 30286,
      "training_step_time": 7.627069711685181
    },
    {
      "epoch": 2.8883934020996093e-05,
      "model_forward_time": 1.7178194522857666,
      "step": 30287
    },
    {
      "epoch": 2.8883934020996093e-05,
      "step": 30287,
      "training_step_time": 7.861717700958252
    },
    {
      "epoch": 2.88848876953125e-05,
      "model_forward_time": 1.7121262550354004,
      "step": 30288
    },
    {
      "epoch": 2.88848876953125e-05,
      "step": 30288,
      "training_step_time": 7.718586206436157
    },
    {
      "epoch": 2.8885841369628908e-05,
      "model_forward_time": 1.7272701263427734,
      "step": 30289
    },
    {
      "epoch": 2.8885841369628908e-05,
      "step": 30289,
      "training_step_time": 9.265540838241577
    },
    {
      "epoch": 2.888679504394531e-05,
      "grad_norm": 0.16015149652957916,
      "learning_rate": 8.351096970392718e-06,
      "loss": 0.0565,
      "step": 30290
    },
    {
      "epoch": 2.888679504394531e-05,
      "model_forward_time": 2.2497777938842773,
      "step": 30290
    },
    {
      "dynamics_loss_avg": 0.06318758837878705,
      "epoch": 2.888679504394531e-05,
      "step": 30290
    },
    {
      "action_loss_avg": 0.00228264182806015,
      "epoch": 2.888679504394531e-05,
      "step": 30290
    },
    {
      "epoch": 2.888679504394531e-05,
      "step": 30290,
      "training_step_time": 10.130774021148682
    },
    {
      "epoch": 2.888774871826172e-05,
      "model_forward_time": 1.7112672328948975,
      "step": 30291
    },
    {
      "epoch": 2.888774871826172e-05,
      "step": 30291,
      "training_step_time": 7.7435948848724365
    },
    {
      "epoch": 2.8888702392578127e-05,
      "model_forward_time": 1.7252557277679443,
      "step": 30292
    },
    {
      "epoch": 2.8888702392578127e-05,
      "step": 30292,
      "training_step_time": 7.662583112716675
    },
    {
      "epoch": 2.888965606689453e-05,
      "model_forward_time": 1.7217214107513428,
      "step": 30293
    },
    {
      "epoch": 2.888965606689453e-05,
      "step": 30293,
      "training_step_time": 8.218284606933594
    },
    {
      "epoch": 2.8890609741210938e-05,
      "model_forward_time": 1.7239789962768555,
      "step": 30294
    },
    {
      "epoch": 2.8890609741210938e-05,
      "step": 30294,
      "training_step_time": 7.70416784286499
    },
    {
      "epoch": 2.8891563415527345e-05,
      "model_forward_time": 1.7111194133758545,
      "step": 30295
    },
    {
      "epoch": 2.8891563415527345e-05,
      "step": 30295,
      "training_step_time": 7.6839776039123535
    },
    {
      "epoch": 2.889251708984375e-05,
      "model_forward_time": 1.7199187278747559,
      "step": 30296
    },
    {
      "epoch": 2.889251708984375e-05,
      "step": 30296,
      "training_step_time": 7.7580366134643555
    },
    {
      "epoch": 2.8893470764160157e-05,
      "model_forward_time": 1.7130932807922363,
      "step": 30297
    },
    {
      "epoch": 2.8893470764160157e-05,
      "step": 30297,
      "training_step_time": 11.54376769065857
    },
    {
      "epoch": 2.889442443847656e-05,
      "model_forward_time": 1.720118761062622,
      "step": 30298
    },
    {
      "epoch": 2.889442443847656e-05,
      "step": 30298,
      "training_step_time": 7.756444931030273
    },
    {
      "epoch": 2.8895378112792968e-05,
      "model_forward_time": 2.2589898109436035,
      "step": 30299
    },
    {
      "epoch": 2.8895378112792968e-05,
      "step": 30299,
      "training_step_time": 7.93238377571106
    },
    {
      "epoch": 2.8896331787109376e-05,
      "grad_norm": 0.17006412148475647,
      "learning_rate": 8.349869643008937e-06,
      "loss": 0.0601,
      "step": 30300
    },
    {
      "epoch": 2.8896331787109376e-05,
      "model_forward_time": 1.7300701141357422,
      "step": 30300
    },
    {
      "dynamics_loss_avg": 0.0529964592307806,
      "epoch": 2.8896331787109376e-05,
      "step": 30300
    },
    {
      "action_loss_avg": 0.002843186288373545,
      "epoch": 2.8896331787109376e-05,
      "step": 30300
    },
    {
      "epoch": 2.8896331787109376e-05,
      "step": 30300,
      "training_step_time": 7.810287237167358
    },
    {
      "epoch": 2.889728546142578e-05,
      "model_forward_time": 1.716425895690918,
      "step": 30301
    },
    {
      "epoch": 2.889728546142578e-05,
      "step": 30301,
      "training_step_time": 7.77246880531311
    },
    {
      "epoch": 2.8898239135742187e-05,
      "model_forward_time": 1.7192614078521729,
      "step": 30302
    },
    {
      "epoch": 2.8898239135742187e-05,
      "step": 30302,
      "training_step_time": 7.552670955657959
    },
    {
      "epoch": 2.8899192810058595e-05,
      "model_forward_time": 1.801128625869751,
      "step": 30303
    },
    {
      "epoch": 2.8899192810058595e-05,
      "step": 30303,
      "training_step_time": 7.804886341094971
    },
    {
      "epoch": 2.8900146484375e-05,
      "model_forward_time": 1.7128558158874512,
      "step": 30304
    },
    {
      "epoch": 2.8900146484375e-05,
      "step": 30304,
      "training_step_time": 7.761536598205566
    },
    {
      "epoch": 2.8901100158691406e-05,
      "model_forward_time": 1.7130491733551025,
      "step": 30305
    },
    {
      "epoch": 2.8901100158691406e-05,
      "step": 30305,
      "training_step_time": 8.531748294830322
    },
    {
      "epoch": 2.8902053833007813e-05,
      "model_forward_time": 1.710888147354126,
      "step": 30306
    },
    {
      "epoch": 2.8902053833007813e-05,
      "step": 30306,
      "training_step_time": 7.807037115097046
    },
    {
      "epoch": 2.8903007507324217e-05,
      "model_forward_time": 1.7224364280700684,
      "step": 30307
    },
    {
      "epoch": 2.8903007507324217e-05,
      "step": 30307,
      "training_step_time": 9.659983158111572
    },
    {
      "epoch": 2.8903961181640625e-05,
      "model_forward_time": 2.259256601333618,
      "step": 30308
    },
    {
      "epoch": 2.8903961181640625e-05,
      "step": 30308,
      "training_step_time": 8.598334074020386
    },
    {
      "epoch": 2.8904914855957032e-05,
      "model_forward_time": 1.7225322723388672,
      "step": 30309
    },
    {
      "epoch": 2.8904914855957032e-05,
      "step": 30309,
      "training_step_time": 8.833873987197876
    },
    {
      "epoch": 2.8905868530273436e-05,
      "grad_norm": 0.1622646301984787,
      "learning_rate": 8.348641949288449e-06,
      "loss": 0.0591,
      "step": 30310
    },
    {
      "epoch": 2.8905868530273436e-05,
      "model_forward_time": 1.7274105548858643,
      "step": 30310
    },
    {
      "dynamics_loss_avg": 0.06509875003248453,
      "epoch": 2.8905868530273436e-05,
      "step": 30310
    },
    {
      "action_loss_avg": 0.001869648537831381,
      "epoch": 2.8905868530273436e-05,
      "step": 30310
    },
    {
      "epoch": 2.8905868530273436e-05,
      "step": 30310,
      "training_step_time": 8.861194372177124
    },
    {
      "epoch": 2.8906822204589844e-05,
      "model_forward_time": 1.7220594882965088,
      "step": 30311
    },
    {
      "epoch": 2.8906822204589844e-05,
      "step": 30311,
      "training_step_time": 7.549216985702515
    },
    {
      "epoch": 2.890777587890625e-05,
      "model_forward_time": 1.7925713062286377,
      "step": 30312
    },
    {
      "epoch": 2.890777587890625e-05,
      "step": 30312,
      "training_step_time": 7.6955366134643555
    },
    {
      "epoch": 2.8908729553222655e-05,
      "model_forward_time": 1.7149803638458252,
      "step": 30313
    },
    {
      "epoch": 2.8908729553222655e-05,
      "step": 30313,
      "training_step_time": 8.223025560379028
    },
    {
      "epoch": 2.8909683227539063e-05,
      "model_forward_time": 1.725250005722046,
      "step": 30314
    },
    {
      "epoch": 2.8909683227539063e-05,
      "step": 30314,
      "training_step_time": 8.411238193511963
    },
    {
      "epoch": 2.891063690185547e-05,
      "model_forward_time": 1.7149410247802734,
      "step": 30315
    },
    {
      "epoch": 2.891063690185547e-05,
      "step": 30315,
      "training_step_time": 8.819984436035156
    },
    {
      "epoch": 2.8911590576171874e-05,
      "model_forward_time": 1.7209596633911133,
      "step": 30316
    },
    {
      "epoch": 2.8911590576171874e-05,
      "step": 30316,
      "training_step_time": 8.316467046737671
    },
    {
      "epoch": 2.891254425048828e-05,
      "model_forward_time": 2.253089666366577,
      "step": 30317
    },
    {
      "epoch": 2.891254425048828e-05,
      "step": 30317,
      "training_step_time": 9.29979944229126
    },
    {
      "epoch": 2.891349792480469e-05,
      "model_forward_time": 1.7328085899353027,
      "step": 30318
    },
    {
      "epoch": 2.891349792480469e-05,
      "step": 30318,
      "training_step_time": 8.735191822052002
    },
    {
      "epoch": 2.8914451599121093e-05,
      "model_forward_time": 1.724731683731079,
      "step": 30319
    },
    {
      "epoch": 2.8914451599121093e-05,
      "step": 30319,
      "training_step_time": 8.794828414916992
    },
    {
      "epoch": 2.89154052734375e-05,
      "grad_norm": 0.14322878420352936,
      "learning_rate": 8.34741388936551e-06,
      "loss": 0.0582,
      "step": 30320
    },
    {
      "epoch": 2.89154052734375e-05,
      "model_forward_time": 1.731924057006836,
      "step": 30320
    },
    {
      "dynamics_loss_avg": 0.05367179457098246,
      "epoch": 2.89154052734375e-05,
      "step": 30320
    },
    {
      "action_loss_avg": 0.002471292979316786,
      "epoch": 2.89154052734375e-05,
      "step": 30320
    },
    {
      "epoch": 2.89154052734375e-05,
      "step": 30320,
      "training_step_time": 7.617002010345459
    },
    {
      "epoch": 2.8916358947753908e-05,
      "model_forward_time": 1.7661561965942383,
      "step": 30321
    },
    {
      "epoch": 2.8916358947753908e-05,
      "step": 30321,
      "training_step_time": 7.773451805114746
    },
    {
      "epoch": 2.8917312622070312e-05,
      "model_forward_time": 1.7130422592163086,
      "step": 30322
    },
    {
      "epoch": 2.8917312622070312e-05,
      "step": 30322,
      "training_step_time": 7.785104751586914
    },
    {
      "epoch": 2.891826629638672e-05,
      "model_forward_time": 1.728184700012207,
      "step": 30323
    },
    {
      "epoch": 2.891826629638672e-05,
      "step": 30323,
      "training_step_time": 7.804910182952881
    },
    {
      "epoch": 2.8919219970703127e-05,
      "model_forward_time": 1.7212345600128174,
      "step": 30324
    },
    {
      "epoch": 2.8919219970703127e-05,
      "step": 30324,
      "training_step_time": 7.916212558746338
    },
    {
      "epoch": 2.892017364501953e-05,
      "model_forward_time": 1.7181897163391113,
      "step": 30325
    },
    {
      "epoch": 2.892017364501953e-05,
      "step": 30325,
      "training_step_time": 8.365919828414917
    },
    {
      "epoch": 2.8921127319335938e-05,
      "model_forward_time": 2.273367166519165,
      "step": 30326
    },
    {
      "epoch": 2.8921127319335938e-05,
      "step": 30326,
      "training_step_time": 8.949562788009644
    },
    {
      "epoch": 2.8922080993652345e-05,
      "model_forward_time": 1.7134768962860107,
      "step": 30327
    },
    {
      "epoch": 2.8922080993652345e-05,
      "step": 30327,
      "training_step_time": 7.804460287094116
    },
    {
      "epoch": 2.892303466796875e-05,
      "model_forward_time": 1.7389626502990723,
      "step": 30328
    },
    {
      "epoch": 2.892303466796875e-05,
      "step": 30328,
      "training_step_time": 9.038255453109741
    },
    {
      "epoch": 2.8923988342285157e-05,
      "model_forward_time": 1.7172446250915527,
      "step": 30329
    },
    {
      "epoch": 2.8923988342285157e-05,
      "step": 30329,
      "training_step_time": 7.73326301574707
    },
    {
      "epoch": 2.892494201660156e-05,
      "grad_norm": 0.12978306412696838,
      "learning_rate": 8.346185463374423e-06,
      "loss": 0.0578,
      "step": 30330
    },
    {
      "epoch": 2.892494201660156e-05,
      "model_forward_time": 1.7323191165924072,
      "step": 30330
    },
    {
      "dynamics_loss_avg": 0.04790108148008585,
      "epoch": 2.892494201660156e-05,
      "step": 30330
    },
    {
      "action_loss_avg": 0.0016655638930387795,
      "epoch": 2.892494201660156e-05,
      "step": 30330
    },
    {
      "epoch": 2.892494201660156e-05,
      "step": 30330,
      "training_step_time": 8.447067737579346
    },
    {
      "epoch": 2.8925895690917968e-05,
      "model_forward_time": 1.7207369804382324,
      "step": 30331
    },
    {
      "epoch": 2.8925895690917968e-05,
      "step": 30331,
      "training_step_time": 7.65697717666626
    },
    {
      "epoch": 2.8926849365234376e-05,
      "model_forward_time": 1.7123889923095703,
      "step": 30332
    },
    {
      "epoch": 2.8926849365234376e-05,
      "step": 30332,
      "training_step_time": 7.733849763870239
    },
    {
      "epoch": 2.892780303955078e-05,
      "model_forward_time": 1.7305681705474854,
      "step": 30333
    },
    {
      "epoch": 2.892780303955078e-05,
      "step": 30333,
      "training_step_time": 7.880265951156616
    },
    {
      "epoch": 2.8928756713867187e-05,
      "model_forward_time": 1.7161967754364014,
      "step": 30334
    },
    {
      "epoch": 2.8928756713867187e-05,
      "step": 30334,
      "training_step_time": 9.647782802581787
    },
    {
      "epoch": 2.8929710388183595e-05,
      "model_forward_time": 2.284088611602783,
      "step": 30335
    },
    {
      "epoch": 2.8929710388183595e-05,
      "step": 30335,
      "training_step_time": 8.86296272277832
    },
    {
      "epoch": 2.89306640625e-05,
      "model_forward_time": 1.7259132862091064,
      "step": 30336
    },
    {
      "epoch": 2.89306640625e-05,
      "step": 30336,
      "training_step_time": 8.780302286148071
    },
    {
      "epoch": 2.8931617736816406e-05,
      "model_forward_time": 1.7321603298187256,
      "step": 30337
    },
    {
      "epoch": 2.8931617736816406e-05,
      "step": 30337,
      "training_step_time": 8.2206130027771
    },
    {
      "epoch": 2.8932571411132813e-05,
      "model_forward_time": 1.7238068580627441,
      "step": 30338
    },
    {
      "epoch": 2.8932571411132813e-05,
      "step": 30338,
      "training_step_time": 7.706875324249268
    },
    {
      "epoch": 2.8933525085449217e-05,
      "model_forward_time": 1.7253391742706299,
      "step": 30339
    },
    {
      "epoch": 2.8933525085449217e-05,
      "step": 30339,
      "training_step_time": 7.80254054069519
    },
    {
      "epoch": 2.8934478759765625e-05,
      "grad_norm": 0.14321789145469666,
      "learning_rate": 8.344956671449524e-06,
      "loss": 0.0624,
      "step": 30340
    },
    {
      "epoch": 2.8934478759765625e-05,
      "model_forward_time": 1.787853479385376,
      "step": 30340
    },
    {
      "dynamics_loss_avg": 0.06679362691938877,
      "epoch": 2.8934478759765625e-05,
      "step": 30340
    },
    {
      "action_loss_avg": 0.0030300505692139267,
      "epoch": 2.8934478759765625e-05,
      "step": 30340
    },
    {
      "epoch": 2.8934478759765625e-05,
      "step": 30340,
      "training_step_time": 7.581084251403809
    },
    {
      "epoch": 2.8935432434082032e-05,
      "model_forward_time": 1.7103393077850342,
      "step": 30341
    },
    {
      "epoch": 2.8935432434082032e-05,
      "step": 30341,
      "training_step_time": 7.710805654525757
    },
    {
      "epoch": 2.8936386108398436e-05,
      "model_forward_time": 1.7106812000274658,
      "step": 30342
    },
    {
      "epoch": 2.8936386108398436e-05,
      "step": 30342,
      "training_step_time": 7.912582874298096
    },
    {
      "epoch": 2.8937339782714844e-05,
      "model_forward_time": 1.7351644039154053,
      "step": 30343
    },
    {
      "epoch": 2.8937339782714844e-05,
      "step": 30343,
      "training_step_time": 9.332127571105957
    },
    {
      "epoch": 2.893829345703125e-05,
      "model_forward_time": 2.245590925216675,
      "step": 30344
    },
    {
      "epoch": 2.893829345703125e-05,
      "step": 30344,
      "training_step_time": 7.85326623916626
    },
    {
      "epoch": 2.8939247131347655e-05,
      "model_forward_time": 1.7386460304260254,
      "step": 30345
    },
    {
      "epoch": 2.8939247131347655e-05,
      "step": 30345,
      "training_step_time": 7.911548614501953
    },
    {
      "epoch": 2.8940200805664063e-05,
      "model_forward_time": 1.7346341609954834,
      "step": 30346
    },
    {
      "epoch": 2.8940200805664063e-05,
      "step": 30346,
      "training_step_time": 8.100968599319458
    },
    {
      "epoch": 2.894115447998047e-05,
      "model_forward_time": 1.7282922267913818,
      "step": 30347
    },
    {
      "epoch": 2.894115447998047e-05,
      "step": 30347,
      "training_step_time": 7.82441258430481
    },
    {
      "epoch": 2.8942108154296874e-05,
      "model_forward_time": 1.7252657413482666,
      "step": 30348
    },
    {
      "epoch": 2.8942108154296874e-05,
      "step": 30348,
      "training_step_time": 7.774901866912842
    },
    {
      "epoch": 2.894306182861328e-05,
      "model_forward_time": 1.786360263824463,
      "step": 30349
    },
    {
      "epoch": 2.894306182861328e-05,
      "step": 30349,
      "training_step_time": 8.619311094284058
    },
    {
      "epoch": 2.894401550292969e-05,
      "grad_norm": 0.12925229966640472,
      "learning_rate": 8.343727513725193e-06,
      "loss": 0.0617,
      "step": 30350
    },
    {
      "epoch": 2.894401550292969e-05,
      "model_forward_time": 1.7295639514923096,
      "step": 30350
    },
    {
      "dynamics_loss_avg": 0.06276123709976673,
      "epoch": 2.894401550292969e-05,
      "step": 30350
    },
    {
      "action_loss_avg": 0.00264619849040173,
      "epoch": 2.894401550292969e-05,
      "step": 30350
    },
    {
      "epoch": 2.894401550292969e-05,
      "step": 30350,
      "training_step_time": 9.39316987991333
    },
    {
      "epoch": 2.8944969177246093e-05,
      "model_forward_time": 1.7287876605987549,
      "step": 30351
    },
    {
      "epoch": 2.8944969177246093e-05,
      "step": 30351,
      "training_step_time": 6.904853820800781
    },
    {
      "epoch": 2.89459228515625e-05,
      "model_forward_time": 1.7339632511138916,
      "step": 30352
    },
    {
      "epoch": 2.89459228515625e-05,
      "step": 30352,
      "training_step_time": 6.938499212265015
    },
    {
      "epoch": 2.8946876525878908e-05,
      "model_forward_time": 1.739161491394043,
      "step": 30353
    },
    {
      "epoch": 2.8946876525878908e-05,
      "step": 30353,
      "training_step_time": 6.975286483764648
    },
    {
      "epoch": 2.8947830200195312e-05,
      "model_forward_time": 1.7190368175506592,
      "step": 30354
    },
    {
      "epoch": 2.8947830200195312e-05,
      "step": 30354,
      "training_step_time": 6.973165273666382
    },
    {
      "epoch": 2.894878387451172e-05,
      "model_forward_time": 1.7204980850219727,
      "step": 30355
    },
    {
      "epoch": 2.894878387451172e-05,
      "step": 30355,
      "training_step_time": 6.961507797241211
    },
    {
      "epoch": 2.8949737548828127e-05,
      "model_forward_time": 1.7229173183441162,
      "step": 30356
    },
    {
      "epoch": 2.8949737548828127e-05,
      "step": 30356,
      "training_step_time": 6.973629951477051
    },
    {
      "epoch": 2.895069122314453e-05,
      "model_forward_time": 1.723548412322998,
      "step": 30357
    },
    {
      "epoch": 2.895069122314453e-05,
      "step": 30357,
      "training_step_time": 6.933613061904907
    },
    {
      "epoch": 2.8951644897460938e-05,
      "model_forward_time": 1.722771167755127,
      "step": 30358
    },
    {
      "epoch": 2.8951644897460938e-05,
      "step": 30358,
      "training_step_time": 8.490936994552612
    },
    {
      "epoch": 2.8952598571777345e-05,
      "model_forward_time": 1.7156834602355957,
      "step": 30359
    },
    {
      "epoch": 2.8952598571777345e-05,
      "step": 30359,
      "training_step_time": 6.868803977966309
    },
    {
      "epoch": 2.895355224609375e-05,
      "grad_norm": 0.1276574581861496,
      "learning_rate": 8.342497990335847e-06,
      "loss": 0.058,
      "step": 30360
    },
    {
      "epoch": 2.895355224609375e-05,
      "model_forward_time": 1.7195842266082764,
      "step": 30360
    },
    {
      "dynamics_loss_avg": 0.05155830616131425,
      "epoch": 2.895355224609375e-05,
      "step": 30360
    },
    {
      "action_loss_avg": 0.0018230554007459432,
      "epoch": 2.895355224609375e-05,
      "step": 30360
    },
    {
      "epoch": 2.895355224609375e-05,
      "step": 30360,
      "training_step_time": 7.95001482963562
    },
    {
      "epoch": 2.8954505920410157e-05,
      "model_forward_time": 1.74061918258667,
      "step": 30361
    },
    {
      "epoch": 2.8954505920410157e-05,
      "step": 30361,
      "training_step_time": 6.8855485916137695
    },
    {
      "epoch": 2.895545959472656e-05,
      "model_forward_time": 1.7114202976226807,
      "step": 30362
    },
    {
      "epoch": 2.895545959472656e-05,
      "step": 30362,
      "training_step_time": 6.925884962081909
    },
    {
      "epoch": 2.895641326904297e-05,
      "model_forward_time": 1.7167778015136719,
      "step": 30363
    },
    {
      "epoch": 2.895641326904297e-05,
      "step": 30363,
      "training_step_time": 6.885050296783447
    },
    {
      "epoch": 2.8957366943359376e-05,
      "model_forward_time": 1.3527672290802002,
      "step": 30364
    },
    {
      "epoch": 2.8957366943359376e-05,
      "step": 30364,
      "training_step_time": 6.8872599601745605
    },
    {
      "epoch": 2.895832061767578e-05,
      "model_forward_time": 1.7311561107635498,
      "step": 30365
    },
    {
      "epoch": 2.895832061767578e-05,
      "step": 30365,
      "training_step_time": 6.881730794906616
    },
    {
      "epoch": 2.8959274291992187e-05,
      "model_forward_time": 1.719905138015747,
      "step": 30366
    },
    {
      "epoch": 2.8959274291992187e-05,
      "step": 30366,
      "training_step_time": 6.884484767913818
    },
    {
      "epoch": 2.8960227966308595e-05,
      "model_forward_time": 1.716552734375,
      "step": 30367
    },
    {
      "epoch": 2.8960227966308595e-05,
      "step": 30367,
      "training_step_time": 6.860193729400635
    },
    {
      "epoch": 2.8961181640625e-05,
      "model_forward_time": 2.2631664276123047,
      "step": 30368
    },
    {
      "epoch": 2.8961181640625e-05,
      "step": 30368,
      "training_step_time": 7.619197845458984
    },
    {
      "epoch": 2.8962135314941406e-05,
      "model_forward_time": 1.728175401687622,
      "step": 30369
    },
    {
      "epoch": 2.8962135314941406e-05,
      "step": 30369,
      "training_step_time": 7.815302133560181
    },
    {
      "epoch": 2.8963088989257813e-05,
      "grad_norm": 0.15799780189990997,
      "learning_rate": 8.341268101415948e-06,
      "loss": 0.0573,
      "step": 30370
    },
    {
      "epoch": 2.8963088989257813e-05,
      "model_forward_time": 1.735262393951416,
      "step": 30370
    },
    {
      "dynamics_loss_avg": 0.0595008485019207,
      "epoch": 2.8963088989257813e-05,
      "step": 30370
    },
    {
      "action_loss_avg": 0.0038322640932165085,
      "epoch": 2.8963088989257813e-05,
      "step": 30370
    },
    {
      "epoch": 2.8963088989257813e-05,
      "step": 30370,
      "training_step_time": 7.1332385540008545
    },
    {
      "epoch": 2.8964042663574217e-05,
      "model_forward_time": 1.736494779586792,
      "step": 30371
    },
    {
      "epoch": 2.8964042663574217e-05,
      "step": 30371,
      "training_step_time": 6.882184982299805
    },
    {
      "epoch": 2.8964996337890625e-05,
      "model_forward_time": 1.7328107357025146,
      "step": 30372
    },
    {
      "epoch": 2.8964996337890625e-05,
      "step": 30372,
      "training_step_time": 6.841672658920288
    },
    {
      "epoch": 2.8965950012207032e-05,
      "model_forward_time": 1.7324600219726562,
      "step": 30373
    },
    {
      "epoch": 2.8965950012207032e-05,
      "step": 30373,
      "training_step_time": 6.887690544128418
    },
    {
      "epoch": 2.8966903686523436e-05,
      "model_forward_time": 1.7406461238861084,
      "step": 30374
    },
    {
      "epoch": 2.8966903686523436e-05,
      "step": 30374,
      "training_step_time": 6.857197284698486
    },
    {
      "epoch": 2.8967857360839844e-05,
      "model_forward_time": 1.7344233989715576,
      "step": 30375
    },
    {
      "epoch": 2.8967857360839844e-05,
      "step": 30375,
      "training_step_time": 6.861243486404419
    },
    {
      "epoch": 2.896881103515625e-05,
      "model_forward_time": 1.7354373931884766,
      "step": 30376
    },
    {
      "epoch": 2.896881103515625e-05,
      "step": 30376,
      "training_step_time": 6.810710430145264
    },
    {
      "epoch": 2.8969764709472655e-05,
      "model_forward_time": 2.2842986583709717,
      "step": 30377
    },
    {
      "epoch": 2.8969764709472655e-05,
      "step": 30377,
      "training_step_time": 7.630691051483154
    },
    {
      "epoch": 2.8970718383789063e-05,
      "model_forward_time": 1.7365684509277344,
      "step": 30378
    },
    {
      "epoch": 2.8970718383789063e-05,
      "step": 30378,
      "training_step_time": 7.806527137756348
    },
    {
      "epoch": 2.897167205810547e-05,
      "model_forward_time": 1.7429368495941162,
      "step": 30379
    },
    {
      "epoch": 2.897167205810547e-05,
      "step": 30379,
      "training_step_time": 7.642395257949829
    },
    {
      "epoch": 2.8972625732421874e-05,
      "grad_norm": 0.1755930632352829,
      "learning_rate": 8.34003784709999e-06,
      "loss": 0.0593,
      "step": 30380
    },
    {
      "epoch": 2.8972625732421874e-05,
      "model_forward_time": 1.732490062713623,
      "step": 30380
    },
    {
      "dynamics_loss_avg": 0.05755150876939297,
      "epoch": 2.8972625732421874e-05,
      "step": 30380
    },
    {
      "action_loss_avg": 0.002765985671430826,
      "epoch": 2.8972625732421874e-05,
      "step": 30380
    },
    {
      "epoch": 2.8972625732421874e-05,
      "step": 30380,
      "training_step_time": 6.8823933601379395
    },
    {
      "epoch": 2.897357940673828e-05,
      "model_forward_time": 1.736588954925537,
      "step": 30381
    },
    {
      "epoch": 2.897357940673828e-05,
      "step": 30381,
      "training_step_time": 6.813160419464111
    },
    {
      "epoch": 2.897453308105469e-05,
      "model_forward_time": 1.731499195098877,
      "step": 30382
    },
    {
      "epoch": 2.897453308105469e-05,
      "step": 30382,
      "training_step_time": 6.821547508239746
    },
    {
      "epoch": 2.8975486755371093e-05,
      "model_forward_time": 1.7462246417999268,
      "step": 30383
    },
    {
      "epoch": 2.8975486755371093e-05,
      "step": 30383,
      "training_step_time": 6.890786170959473
    },
    {
      "epoch": 2.89764404296875e-05,
      "model_forward_time": 1.7420663833618164,
      "step": 30384
    },
    {
      "epoch": 2.89764404296875e-05,
      "step": 30384,
      "training_step_time": 6.835134267807007
    },
    {
      "epoch": 2.8977394104003908e-05,
      "model_forward_time": 1.7397747039794922,
      "step": 30385
    },
    {
      "epoch": 2.8977394104003908e-05,
      "step": 30385,
      "training_step_time": 6.862877607345581
    },
    {
      "epoch": 2.8978347778320312e-05,
      "model_forward_time": 2.301222324371338,
      "step": 30386
    },
    {
      "epoch": 2.8978347778320312e-05,
      "step": 30386,
      "training_step_time": 7.692861318588257
    },
    {
      "epoch": 2.897930145263672e-05,
      "model_forward_time": 1.741715431213379,
      "step": 30387
    },
    {
      "epoch": 2.897930145263672e-05,
      "step": 30387,
      "training_step_time": 10.762397527694702
    },
    {
      "epoch": 2.8980255126953127e-05,
      "model_forward_time": 1.734276294708252,
      "step": 30388
    },
    {
      "epoch": 2.8980255126953127e-05,
      "step": 30388,
      "training_step_time": 7.641331911087036
    },
    {
      "epoch": 2.898120880126953e-05,
      "model_forward_time": 1.735776424407959,
      "step": 30389
    },
    {
      "epoch": 2.898120880126953e-05,
      "step": 30389,
      "training_step_time": 6.860974311828613
    },
    {
      "epoch": 2.8982162475585938e-05,
      "grad_norm": 0.12689051032066345,
      "learning_rate": 8.338807227522516e-06,
      "loss": 0.0579,
      "step": 30390
    },
    {
      "epoch": 2.8982162475585938e-05,
      "model_forward_time": 1.7320396900177002,
      "step": 30390
    },
    {
      "dynamics_loss_avg": 0.048321896232664585,
      "epoch": 2.8982162475585938e-05,
      "step": 30390
    },
    {
      "action_loss_avg": 0.0020839139149757104,
      "epoch": 2.8982162475585938e-05,
      "step": 30390
    },
    {
      "epoch": 2.8982162475585938e-05,
      "step": 30390,
      "training_step_time": 6.815579652786255
    },
    {
      "epoch": 2.8983116149902342e-05,
      "model_forward_time": 1.73699951171875,
      "step": 30391
    },
    {
      "epoch": 2.8983116149902342e-05,
      "step": 30391,
      "training_step_time": 6.861520290374756
    },
    {
      "epoch": 2.898406982421875e-05,
      "model_forward_time": 1.7396011352539062,
      "step": 30392
    },
    {
      "epoch": 2.898406982421875e-05,
      "step": 30392,
      "training_step_time": 6.8336181640625
    },
    {
      "epoch": 2.8985023498535157e-05,
      "model_forward_time": 1.736222505569458,
      "step": 30393
    },
    {
      "epoch": 2.8985023498535157e-05,
      "step": 30393,
      "training_step_time": 6.839025497436523
    },
    {
      "epoch": 2.898597717285156e-05,
      "model_forward_time": 1.7414567470550537,
      "step": 30394
    },
    {
      "epoch": 2.898597717285156e-05,
      "step": 30394,
      "training_step_time": 6.858151197433472
    },
    {
      "epoch": 2.898693084716797e-05,
      "model_forward_time": 2.3022401332855225,
      "step": 30395
    },
    {
      "epoch": 2.898693084716797e-05,
      "step": 30395,
      "training_step_time": 7.664928197860718
    },
    {
      "epoch": 2.8987884521484376e-05,
      "model_forward_time": 1.3666839599609375,
      "step": 30396
    },
    {
      "epoch": 2.8987884521484376e-05,
      "step": 30396,
      "training_step_time": 9.826201677322388
    },
    {
      "epoch": 2.898883819580078e-05,
      "model_forward_time": 1.735349178314209,
      "step": 30397
    },
    {
      "epoch": 2.898883819580078e-05,
      "step": 30397,
      "training_step_time": 8.676860094070435
    },
    {
      "epoch": 2.8989791870117187e-05,
      "model_forward_time": 1.7222249507904053,
      "step": 30398
    },
    {
      "epoch": 2.8989791870117187e-05,
      "step": 30398,
      "training_step_time": 7.161396265029907
    },
    {
      "epoch": 2.8990745544433595e-05,
      "model_forward_time": 1.7485551834106445,
      "step": 30399
    },
    {
      "epoch": 2.8990745544433595e-05,
      "step": 30399,
      "training_step_time": 6.867627382278442
    },
    {
      "epoch": 2.899169921875e-05,
      "grad_norm": 0.11376465857028961,
      "learning_rate": 8.337576242818104e-06,
      "loss": 0.0608,
      "step": 30400
    },
    {
      "epoch": 2.899169921875e-05,
      "model_forward_time": 1.7420365810394287,
      "step": 30400
    },
    {
      "dynamics_loss_avg": 0.056618990190327165,
      "epoch": 2.899169921875e-05,
      "step": 30400
    },
    {
      "action_loss_avg": 0.002141988289076835,
      "epoch": 2.899169921875e-05,
      "step": 30400
    },
    {
      "epoch": 2.899169921875e-05,
      "step": 30400,
      "training_step_time": 6.879053115844727
    },
    {
      "epoch": 2.8992652893066406e-05,
      "model_forward_time": 1.7408862113952637,
      "step": 30401
    },
    {
      "epoch": 2.8992652893066406e-05,
      "step": 30401,
      "training_step_time": 6.863258361816406
    },
    {
      "epoch": 2.8993606567382813e-05,
      "model_forward_time": 1.7264275550842285,
      "step": 30402
    },
    {
      "epoch": 2.8993606567382813e-05,
      "step": 30402,
      "training_step_time": 6.900298595428467
    },
    {
      "epoch": 2.8994560241699218e-05,
      "model_forward_time": 1.7254152297973633,
      "step": 30403
    },
    {
      "epoch": 2.8994560241699218e-05,
      "step": 30403,
      "training_step_time": 6.785114288330078
    },
    {
      "epoch": 2.8995513916015625e-05,
      "model_forward_time": 1.720412254333496,
      "step": 30404
    },
    {
      "epoch": 2.8995513916015625e-05,
      "step": 30404,
      "training_step_time": 7.585913181304932
    },
    {
      "epoch": 2.8996467590332032e-05,
      "model_forward_time": 2.2968461513519287,
      "step": 30405
    },
    {
      "epoch": 2.8996467590332032e-05,
      "step": 30405,
      "training_step_time": 9.656918048858643
    },
    {
      "epoch": 2.8997421264648436e-05,
      "model_forward_time": 1.0367772579193115,
      "step": 30406
    },
    {
      "epoch": 2.8997421264648436e-05,
      "step": 30406,
      "training_step_time": 7.77865743637085
    },
    {
      "epoch": 2.8998374938964844e-05,
      "model_forward_time": 1.7321445941925049,
      "step": 30407
    },
    {
      "epoch": 2.8998374938964844e-05,
      "step": 30407,
      "training_step_time": 7.350139379501343
    },
    {
      "epoch": 2.899932861328125e-05,
      "model_forward_time": 1.7316973209381104,
      "step": 30408
    },
    {
      "epoch": 2.899932861328125e-05,
      "step": 30408,
      "training_step_time": 6.9115331172943115
    },
    {
      "epoch": 2.9000282287597655e-05,
      "model_forward_time": 1.7040212154388428,
      "step": 30409
    },
    {
      "epoch": 2.9000282287597655e-05,
      "step": 30409,
      "training_step_time": 6.911341190338135
    },
    {
      "epoch": 2.9001235961914063e-05,
      "grad_norm": 0.11453185975551605,
      "learning_rate": 8.336344893121372e-06,
      "loss": 0.0625,
      "step": 30410
    },
    {
      "epoch": 2.9001235961914063e-05,
      "model_forward_time": 1.7294297218322754,
      "step": 30410
    },
    {
      "dynamics_loss_avg": 0.06057991255074739,
      "epoch": 2.9001235961914063e-05,
      "step": 30410
    },
    {
      "action_loss_avg": 0.0019532258796971292,
      "epoch": 2.9001235961914063e-05,
      "step": 30410
    },
    {
      "epoch": 2.9001235961914063e-05,
      "step": 30410,
      "training_step_time": 6.884625673294067
    },
    {
      "epoch": 2.900218963623047e-05,
      "model_forward_time": 1.7240159511566162,
      "step": 30411
    },
    {
      "epoch": 2.900218963623047e-05,
      "step": 30411,
      "training_step_time": 6.979768753051758
    },
    {
      "epoch": 2.9003143310546874e-05,
      "model_forward_time": 1.7386374473571777,
      "step": 30412
    },
    {
      "epoch": 2.9003143310546874e-05,
      "step": 30412,
      "training_step_time": 6.938636541366577
    },
    {
      "epoch": 2.900409698486328e-05,
      "model_forward_time": 1.7273414134979248,
      "step": 30413
    },
    {
      "epoch": 2.900409698486328e-05,
      "step": 30413,
      "training_step_time": 7.547159194946289
    },
    {
      "epoch": 2.900505065917969e-05,
      "model_forward_time": 2.292681932449341,
      "step": 30414
    },
    {
      "epoch": 2.900505065917969e-05,
      "step": 30414,
      "training_step_time": 9.700436353683472
    },
    {
      "epoch": 2.9006004333496093e-05,
      "model_forward_time": 1.7355940341949463,
      "step": 30415
    },
    {
      "epoch": 2.9006004333496093e-05,
      "step": 30415,
      "training_step_time": 8.080373287200928
    },
    {
      "epoch": 2.90069580078125e-05,
      "model_forward_time": 1.7382478713989258,
      "step": 30416
    },
    {
      "epoch": 2.90069580078125e-05,
      "step": 30416,
      "training_step_time": 7.617099046707153
    },
    {
      "epoch": 2.9007911682128908e-05,
      "model_forward_time": 1.750901222229004,
      "step": 30417
    },
    {
      "epoch": 2.9007911682128908e-05,
      "step": 30417,
      "training_step_time": 6.956415891647339
    },
    {
      "epoch": 2.9008865356445312e-05,
      "model_forward_time": 1.7367229461669922,
      "step": 30418
    },
    {
      "epoch": 2.9008865356445312e-05,
      "step": 30418,
      "training_step_time": 6.970945596694946
    },
    {
      "epoch": 2.900981903076172e-05,
      "model_forward_time": 1.7364001274108887,
      "step": 30419
    },
    {
      "epoch": 2.900981903076172e-05,
      "step": 30419,
      "training_step_time": 6.971035480499268
    },
    {
      "epoch": 2.9010772705078127e-05,
      "grad_norm": 0.1486188769340515,
      "learning_rate": 8.335113178566978e-06,
      "loss": 0.0625,
      "step": 30420
    },
    {
      "epoch": 2.9010772705078127e-05,
      "model_forward_time": 1.736511468887329,
      "step": 30420
    },
    {
      "dynamics_loss_avg": 0.06860477337613702,
      "epoch": 2.9010772705078127e-05,
      "step": 30420
    },
    {
      "action_loss_avg": 0.002234115049941465,
      "epoch": 2.9010772705078127e-05,
      "step": 30420
    },
    {
      "epoch": 2.9010772705078127e-05,
      "step": 30420,
      "training_step_time": 6.91502571105957
    },
    {
      "epoch": 2.901172637939453e-05,
      "model_forward_time": 1.7351815700531006,
      "step": 30421
    },
    {
      "epoch": 2.901172637939453e-05,
      "step": 30421,
      "training_step_time": 6.93086576461792
    },
    {
      "epoch": 2.9012680053710938e-05,
      "model_forward_time": 1.7428019046783447,
      "step": 30422
    },
    {
      "epoch": 2.9012680053710938e-05,
      "step": 30422,
      "training_step_time": 6.9120917320251465
    },
    {
      "epoch": 2.9013633728027342e-05,
      "model_forward_time": 2.2915029525756836,
      "step": 30423
    },
    {
      "epoch": 2.9013633728027342e-05,
      "step": 30423,
      "training_step_time": 7.73049259185791
    },
    {
      "epoch": 2.901458740234375e-05,
      "model_forward_time": 1.7394955158233643,
      "step": 30424
    },
    {
      "epoch": 2.901458740234375e-05,
      "step": 30424,
      "training_step_time": 8.73206090927124
    },
    {
      "epoch": 2.9015541076660157e-05,
      "model_forward_time": 1.7422091960906982,
      "step": 30425
    },
    {
      "epoch": 2.9015541076660157e-05,
      "step": 30425,
      "training_step_time": 7.601061820983887
    },
    {
      "epoch": 2.901649475097656e-05,
      "model_forward_time": 1.7171704769134521,
      "step": 30426
    },
    {
      "epoch": 2.901649475097656e-05,
      "step": 30426,
      "training_step_time": 7.100586891174316
    },
    {
      "epoch": 2.901744842529297e-05,
      "model_forward_time": 1.7342641353607178,
      "step": 30427
    },
    {
      "epoch": 2.901744842529297e-05,
      "step": 30427,
      "training_step_time": 6.866748809814453
    },
    {
      "epoch": 2.9018402099609376e-05,
      "model_forward_time": 1.7388293743133545,
      "step": 30428
    },
    {
      "epoch": 2.9018402099609376e-05,
      "step": 30428,
      "training_step_time": 6.926791667938232
    },
    {
      "epoch": 2.901935577392578e-05,
      "model_forward_time": 1.7327642440795898,
      "step": 30429
    },
    {
      "epoch": 2.901935577392578e-05,
      "step": 30429,
      "training_step_time": 6.933964729309082
    },
    {
      "epoch": 2.9020309448242187e-05,
      "grad_norm": 0.09363893419504166,
      "learning_rate": 8.333881099289622e-06,
      "loss": 0.0588,
      "step": 30430
    },
    {
      "epoch": 2.9020309448242187e-05,
      "model_forward_time": 1.744422197341919,
      "step": 30430
    },
    {
      "dynamics_loss_avg": 0.04134067054837942,
      "epoch": 2.9020309448242187e-05,
      "step": 30430
    },
    {
      "action_loss_avg": 0.0019636519544292243,
      "epoch": 2.9020309448242187e-05,
      "step": 30430
    },
    {
      "epoch": 2.9020309448242187e-05,
      "step": 30430,
      "training_step_time": 6.704747915267944
    },
    {
      "epoch": 2.9021263122558595e-05,
      "model_forward_time": 1.736532211303711,
      "step": 30431
    },
    {
      "epoch": 2.9021263122558595e-05,
      "step": 30431,
      "training_step_time": 7.865638017654419
    },
    {
      "epoch": 2.9022216796875e-05,
      "model_forward_time": 2.29154372215271,
      "step": 30432
    },
    {
      "epoch": 2.9022216796875e-05,
      "step": 30432,
      "training_step_time": 8.588294267654419
    },
    {
      "epoch": 2.9023170471191406e-05,
      "model_forward_time": 1.7408061027526855,
      "step": 30433
    },
    {
      "epoch": 2.9023170471191406e-05,
      "step": 30433,
      "training_step_time": 9.772705078125
    },
    {
      "epoch": 2.9024124145507814e-05,
      "model_forward_time": 1.7349159717559814,
      "step": 30434
    },
    {
      "epoch": 2.9024124145507814e-05,
      "step": 30434,
      "training_step_time": 7.699666261672974
    },
    {
      "epoch": 2.9025077819824218e-05,
      "model_forward_time": 1.7351586818695068,
      "step": 30435
    },
    {
      "epoch": 2.9025077819824218e-05,
      "step": 30435,
      "training_step_time": 7.179324388504028
    },
    {
      "epoch": 2.9026031494140625e-05,
      "model_forward_time": 1.7296881675720215,
      "step": 30436
    },
    {
      "epoch": 2.9026031494140625e-05,
      "step": 30436,
      "training_step_time": 6.9057297706604
    },
    {
      "epoch": 2.9026985168457032e-05,
      "model_forward_time": 1.7308623790740967,
      "step": 30437
    },
    {
      "epoch": 2.9026985168457032e-05,
      "step": 30437,
      "training_step_time": 6.899245023727417
    },
    {
      "epoch": 2.9027938842773436e-05,
      "model_forward_time": 1.7271809577941895,
      "step": 30438
    },
    {
      "epoch": 2.9027938842773436e-05,
      "step": 30438,
      "training_step_time": 6.905763626098633
    },
    {
      "epoch": 2.9028892517089844e-05,
      "model_forward_time": 1.7324514389038086,
      "step": 30439
    },
    {
      "epoch": 2.9028892517089844e-05,
      "step": 30439,
      "training_step_time": 6.910428762435913
    },
    {
      "epoch": 2.902984619140625e-05,
      "grad_norm": 0.1769757717847824,
      "learning_rate": 8.33264865542404e-06,
      "loss": 0.0635,
      "step": 30440
    },
    {
      "epoch": 2.902984619140625e-05,
      "model_forward_time": 1.7353293895721436,
      "step": 30440
    },
    {
      "dynamics_loss_avg": 0.06623187623918056,
      "epoch": 2.902984619140625e-05,
      "step": 30440
    },
    {
      "action_loss_avg": 0.00299460650421679,
      "epoch": 2.902984619140625e-05,
      "step": 30440
    },
    {
      "epoch": 2.902984619140625e-05,
      "step": 30440,
      "training_step_time": 6.890581369400024
    },
    {
      "epoch": 2.9030799865722655e-05,
      "model_forward_time": 1.7471485137939453,
      "step": 30441
    },
    {
      "epoch": 2.9030799865722655e-05,
      "step": 30441,
      "training_step_time": 8.516435384750366
    },
    {
      "epoch": 2.9031753540039063e-05,
      "model_forward_time": 2.2804696559906006,
      "step": 30442
    },
    {
      "epoch": 2.9031753540039063e-05,
      "step": 30442,
      "training_step_time": 9.966941118240356
    },
    {
      "epoch": 2.903270721435547e-05,
      "model_forward_time": 1.7318050861358643,
      "step": 30443
    },
    {
      "epoch": 2.903270721435547e-05,
      "step": 30443,
      "training_step_time": 7.717032194137573
    },
    {
      "epoch": 2.9033660888671874e-05,
      "model_forward_time": 1.7317471504211426,
      "step": 30444
    },
    {
      "epoch": 2.9033660888671874e-05,
      "step": 30444,
      "training_step_time": 7.1687352657318115
    },
    {
      "epoch": 2.903461456298828e-05,
      "model_forward_time": 1.7434988021850586,
      "step": 30445
    },
    {
      "epoch": 2.903461456298828e-05,
      "step": 30445,
      "training_step_time": 6.901409387588501
    },
    {
      "epoch": 2.903556823730469e-05,
      "model_forward_time": 1.730449914932251,
      "step": 30446
    },
    {
      "epoch": 2.903556823730469e-05,
      "step": 30446,
      "training_step_time": 6.912018299102783
    },
    {
      "epoch": 2.9036521911621093e-05,
      "model_forward_time": 1.7351751327514648,
      "step": 30447
    },
    {
      "epoch": 2.9036521911621093e-05,
      "step": 30447,
      "training_step_time": 6.91407322883606
    },
    {
      "epoch": 2.90374755859375e-05,
      "model_forward_time": 1.7268905639648438,
      "step": 30448
    },
    {
      "epoch": 2.90374755859375e-05,
      "step": 30448,
      "training_step_time": 6.883212089538574
    },
    {
      "epoch": 2.9038429260253908e-05,
      "model_forward_time": 1.739424467086792,
      "step": 30449
    },
    {
      "epoch": 2.9038429260253908e-05,
      "step": 30449,
      "training_step_time": 6.936687469482422
    },
    {
      "epoch": 2.9039382934570312e-05,
      "grad_norm": 0.12845328450202942,
      "learning_rate": 8.331415847105013e-06,
      "loss": 0.065,
      "step": 30450
    },
    {
      "epoch": 2.9039382934570312e-05,
      "model_forward_time": 1.7272298336029053,
      "step": 30450
    },
    {
      "dynamics_loss_avg": 0.04801247827708721,
      "epoch": 2.9039382934570312e-05,
      "step": 30450
    },
    {
      "action_loss_avg": 0.001978655043058097,
      "epoch": 2.9039382934570312e-05,
      "step": 30450
    },
    {
      "epoch": 2.9039382934570312e-05,
      "step": 30450,
      "training_step_time": 9.208219528198242
    },
    {
      "epoch": 2.904033660888672e-05,
      "model_forward_time": 2.280751943588257,
      "step": 30451
    },
    {
      "epoch": 2.904033660888672e-05,
      "step": 30451,
      "training_step_time": 8.734772443771362
    },
    {
      "epoch": 2.9041290283203127e-05,
      "model_forward_time": 1.738358974456787,
      "step": 30452
    },
    {
      "epoch": 2.9041290283203127e-05,
      "step": 30452,
      "training_step_time": 7.737622022628784
    },
    {
      "epoch": 2.904224395751953e-05,
      "model_forward_time": 1.7277493476867676,
      "step": 30453
    },
    {
      "epoch": 2.904224395751953e-05,
      "step": 30453,
      "training_step_time": 7.538705110549927
    },
    {
      "epoch": 2.9043197631835938e-05,
      "model_forward_time": 1.725773572921753,
      "step": 30454
    },
    {
      "epoch": 2.9043197631835938e-05,
      "step": 30454,
      "training_step_time": 6.881747484207153
    },
    {
      "epoch": 2.9044151306152342e-05,
      "model_forward_time": 1.7252576351165771,
      "step": 30455
    },
    {
      "epoch": 2.9044151306152342e-05,
      "step": 30455,
      "training_step_time": 6.926926374435425
    },
    {
      "epoch": 2.904510498046875e-05,
      "model_forward_time": 1.741990566253662,
      "step": 30456
    },
    {
      "epoch": 2.904510498046875e-05,
      "step": 30456,
      "training_step_time": 6.903867721557617
    },
    {
      "epoch": 2.9046058654785157e-05,
      "model_forward_time": 1.7318804264068604,
      "step": 30457
    },
    {
      "epoch": 2.9046058654785157e-05,
      "step": 30457,
      "training_step_time": 6.903869152069092
    },
    {
      "epoch": 2.904701232910156e-05,
      "model_forward_time": 1.7368581295013428,
      "step": 30458
    },
    {
      "epoch": 2.904701232910156e-05,
      "step": 30458,
      "training_step_time": 6.907844066619873
    },
    {
      "epoch": 2.904796600341797e-05,
      "model_forward_time": 1.7147784233093262,
      "step": 30459
    },
    {
      "epoch": 2.904796600341797e-05,
      "step": 30459,
      "training_step_time": 9.569793462753296
    },
    {
      "epoch": 2.9048919677734376e-05,
      "grad_norm": 0.10150685161352158,
      "learning_rate": 8.330182674467358e-06,
      "loss": 0.0623,
      "step": 30460
    },
    {
      "epoch": 2.9048919677734376e-05,
      "model_forward_time": 2.297346353530884,
      "step": 30460
    },
    {
      "dynamics_loss_avg": 0.05724720042198896,
      "epoch": 2.9048919677734376e-05,
      "step": 30460
    },
    {
      "action_loss_avg": 0.0022292464098427446,
      "epoch": 2.9048919677734376e-05,
      "step": 30460
    },
    {
      "epoch": 2.9048919677734376e-05,
      "step": 30460,
      "training_step_time": 8.766767978668213
    },
    {
      "epoch": 2.904987335205078e-05,
      "model_forward_time": 1.7325708866119385,
      "step": 30461
    },
    {
      "epoch": 2.904987335205078e-05,
      "step": 30461,
      "training_step_time": 8.515832901000977
    },
    {
      "epoch": 2.9050827026367187e-05,
      "model_forward_time": 1.7342755794525146,
      "step": 30462
    },
    {
      "epoch": 2.9050827026367187e-05,
      "step": 30462,
      "training_step_time": 7.721426248550415
    },
    {
      "epoch": 2.9051780700683595e-05,
      "model_forward_time": 1.730936050415039,
      "step": 30463
    },
    {
      "epoch": 2.9051780700683595e-05,
      "step": 30463,
      "training_step_time": 6.934291124343872
    },
    {
      "epoch": 2.9052734375e-05,
      "model_forward_time": 1.7287514209747314,
      "step": 30464
    },
    {
      "epoch": 2.9052734375e-05,
      "step": 30464,
      "training_step_time": 6.872599840164185
    },
    {
      "epoch": 2.9053688049316406e-05,
      "model_forward_time": 1.7259645462036133,
      "step": 30465
    },
    {
      "epoch": 2.9053688049316406e-05,
      "step": 30465,
      "training_step_time": 6.940240859985352
    },
    {
      "epoch": 2.9054641723632814e-05,
      "model_forward_time": 1.7370312213897705,
      "step": 30466
    },
    {
      "epoch": 2.9054641723632814e-05,
      "step": 30466,
      "training_step_time": 6.950648546218872
    },
    {
      "epoch": 2.9055595397949218e-05,
      "model_forward_time": 1.7299082279205322,
      "step": 30467
    },
    {
      "epoch": 2.9055595397949218e-05,
      "step": 30467,
      "training_step_time": 6.873730182647705
    },
    {
      "epoch": 2.9056549072265625e-05,
      "model_forward_time": 1.7296247482299805,
      "step": 30468
    },
    {
      "epoch": 2.9056549072265625e-05,
      "step": 30468,
      "training_step_time": 7.813958644866943
    },
    {
      "epoch": 2.9057502746582032e-05,
      "model_forward_time": 2.3019485473632812,
      "step": 30469
    },
    {
      "epoch": 2.9057502746582032e-05,
      "step": 30469,
      "training_step_time": 9.013702392578125
    },
    {
      "epoch": 2.9058456420898436e-05,
      "grad_norm": 0.14010290801525116,
      "learning_rate": 8.32894913764593e-06,
      "loss": 0.0584,
      "step": 30470
    },
    {
      "epoch": 2.9058456420898436e-05,
      "model_forward_time": 1.727541446685791,
      "step": 30470
    },
    {
      "dynamics_loss_avg": 0.04855598881840706,
      "epoch": 2.9058456420898436e-05,
      "step": 30470
    },
    {
      "action_loss_avg": 0.0017498895205790177,
      "epoch": 2.9058456420898436e-05,
      "step": 30470
    },
    {
      "epoch": 2.9058456420898436e-05,
      "step": 30470,
      "training_step_time": 9.401512622833252
    },
    {
      "epoch": 2.9059410095214844e-05,
      "model_forward_time": 1.7355377674102783,
      "step": 30471
    },
    {
      "epoch": 2.9059410095214844e-05,
      "step": 30471,
      "training_step_time": 7.683515548706055
    },
    {
      "epoch": 2.906036376953125e-05,
      "model_forward_time": 1.7256598472595215,
      "step": 30472
    },
    {
      "epoch": 2.906036376953125e-05,
      "step": 30472,
      "training_step_time": 7.187735080718994
    },
    {
      "epoch": 2.9061317443847655e-05,
      "model_forward_time": 1.7309339046478271,
      "step": 30473
    },
    {
      "epoch": 2.9061317443847655e-05,
      "step": 30473,
      "training_step_time": 6.899579763412476
    },
    {
      "epoch": 2.9062271118164063e-05,
      "model_forward_time": 1.7281944751739502,
      "step": 30474
    },
    {
      "epoch": 2.9062271118164063e-05,
      "step": 30474,
      "training_step_time": 6.917662620544434
    },
    {
      "epoch": 2.906322479248047e-05,
      "model_forward_time": 1.7365443706512451,
      "step": 30475
    },
    {
      "epoch": 2.906322479248047e-05,
      "step": 30475,
      "training_step_time": 6.9336090087890625
    },
    {
      "epoch": 2.9064178466796874e-05,
      "model_forward_time": 1.7346773147583008,
      "step": 30476
    },
    {
      "epoch": 2.9064178466796874e-05,
      "step": 30476,
      "training_step_time": 6.967014312744141
    },
    {
      "epoch": 2.906513214111328e-05,
      "model_forward_time": 1.726980447769165,
      "step": 30477
    },
    {
      "epoch": 2.906513214111328e-05,
      "step": 30477,
      "training_step_time": 7.788952827453613
    },
    {
      "epoch": 2.906608581542969e-05,
      "model_forward_time": 2.2882533073425293,
      "step": 30478
    },
    {
      "epoch": 2.906608581542969e-05,
      "step": 30478,
      "training_step_time": 8.540802955627441
    },
    {
      "epoch": 2.9067039489746093e-05,
      "model_forward_time": 1.7298123836517334,
      "step": 30479
    },
    {
      "epoch": 2.9067039489746093e-05,
      "step": 30479,
      "training_step_time": 8.003437995910645
    },
    {
      "epoch": 2.90679931640625e-05,
      "grad_norm": 0.13067539036273956,
      "learning_rate": 8.327715236775634e-06,
      "loss": 0.0626,
      "step": 30480
    },
    {
      "epoch": 2.90679931640625e-05,
      "model_forward_time": 1.730238437652588,
      "step": 30480
    },
    {
      "dynamics_loss_avg": 0.05853575058281422,
      "epoch": 2.90679931640625e-05,
      "step": 30480
    },
    {
      "action_loss_avg": 0.001921339496038854,
      "epoch": 2.90679931640625e-05,
      "step": 30480
    },
    {
      "epoch": 2.90679931640625e-05,
      "step": 30480,
      "training_step_time": 7.744790554046631
    },
    {
      "epoch": 2.9068946838378908e-05,
      "model_forward_time": 1.7274446487426758,
      "step": 30481
    },
    {
      "epoch": 2.9068946838378908e-05,
      "step": 30481,
      "training_step_time": 7.5737738609313965
    },
    {
      "epoch": 2.9069900512695312e-05,
      "model_forward_time": 1.730924129486084,
      "step": 30482
    },
    {
      "epoch": 2.9069900512695312e-05,
      "step": 30482,
      "training_step_time": 6.931288719177246
    },
    {
      "epoch": 2.907085418701172e-05,
      "model_forward_time": 1.7245244979858398,
      "step": 30483
    },
    {
      "epoch": 2.907085418701172e-05,
      "step": 30483,
      "training_step_time": 6.951801061630249
    },
    {
      "epoch": 2.9071807861328127e-05,
      "model_forward_time": 1.7275516986846924,
      "step": 30484
    },
    {
      "epoch": 2.9071807861328127e-05,
      "step": 30484,
      "training_step_time": 6.997964382171631
    },
    {
      "epoch": 2.907276153564453e-05,
      "model_forward_time": 1.7286605834960938,
      "step": 30485
    },
    {
      "epoch": 2.907276153564453e-05,
      "step": 30485,
      "training_step_time": 7.022340297698975
    },
    {
      "epoch": 2.9073715209960938e-05,
      "model_forward_time": 1.784860610961914,
      "step": 30486
    },
    {
      "epoch": 2.9073715209960938e-05,
      "step": 30486,
      "training_step_time": 8.822021245956421
    },
    {
      "epoch": 2.9074668884277342e-05,
      "model_forward_time": 1.7394063472747803,
      "step": 30487
    },
    {
      "epoch": 2.9074668884277342e-05,
      "step": 30487,
      "training_step_time": 9.922115802764893
    },
    {
      "epoch": 2.907562255859375e-05,
      "model_forward_time": 2.3031668663024902,
      "step": 30488
    },
    {
      "epoch": 2.907562255859375e-05,
      "step": 30488,
      "training_step_time": 9.371402740478516
    },
    {
      "epoch": 2.9076576232910157e-05,
      "model_forward_time": 1.7313377857208252,
      "step": 30489
    },
    {
      "epoch": 2.9076576232910157e-05,
      "step": 30489,
      "training_step_time": 8.748151302337646
    },
    {
      "epoch": 2.907752990722656e-05,
      "grad_norm": 0.16409733891487122,
      "learning_rate": 8.326480971991399e-06,
      "loss": 0.0563,
      "step": 30490
    },
    {
      "epoch": 2.907752990722656e-05,
      "model_forward_time": 1.731034517288208,
      "step": 30490
    },
    {
      "dynamics_loss_avg": 0.06672939406707883,
      "epoch": 2.907752990722656e-05,
      "step": 30490
    },
    {
      "action_loss_avg": 0.002246212321915664,
      "epoch": 2.907752990722656e-05,
      "step": 30490
    },
    {
      "epoch": 2.907752990722656e-05,
      "step": 30490,
      "training_step_time": 7.4438090324401855
    },
    {
      "epoch": 2.907848358154297e-05,
      "model_forward_time": 1.7207679748535156,
      "step": 30491
    },
    {
      "epoch": 2.907848358154297e-05,
      "step": 30491,
      "training_step_time": 6.94023323059082
    },
    {
      "epoch": 2.9079437255859376e-05,
      "model_forward_time": 1.7297873497009277,
      "step": 30492
    },
    {
      "epoch": 2.9079437255859376e-05,
      "step": 30492,
      "training_step_time": 6.886467933654785
    },
    {
      "epoch": 2.908039093017578e-05,
      "model_forward_time": 1.0176372528076172,
      "step": 30493
    },
    {
      "epoch": 2.908039093017578e-05,
      "step": 30493,
      "training_step_time": 6.923936128616333
    },
    {
      "epoch": 2.9081344604492187e-05,
      "model_forward_time": 1.7321829795837402,
      "step": 30494
    },
    {
      "epoch": 2.9081344604492187e-05,
      "step": 30494,
      "training_step_time": 6.8993754386901855
    },
    {
      "epoch": 2.9082298278808595e-05,
      "model_forward_time": 1.7338368892669678,
      "step": 30495
    },
    {
      "epoch": 2.9082298278808595e-05,
      "step": 30495,
      "training_step_time": 7.734302759170532
    },
    {
      "epoch": 2.9083251953125e-05,
      "model_forward_time": 1.7157580852508545,
      "step": 30496
    },
    {
      "epoch": 2.9083251953125e-05,
      "step": 30496,
      "training_step_time": 9.364400148391724
    },
    {
      "epoch": 2.9084205627441406e-05,
      "model_forward_time": 2.305854082107544,
      "step": 30497
    },
    {
      "epoch": 2.9084205627441406e-05,
      "step": 30497,
      "training_step_time": 8.056525707244873
    },
    {
      "epoch": 2.9085159301757814e-05,
      "model_forward_time": 1.728656530380249,
      "step": 30498
    },
    {
      "epoch": 2.9085159301757814e-05,
      "step": 30498,
      "training_step_time": 7.758704900741577
    },
    {
      "epoch": 2.9086112976074218e-05,
      "model_forward_time": 1.7325325012207031,
      "step": 30499
    },
    {
      "epoch": 2.9086112976074218e-05,
      "step": 30499,
      "training_step_time": 7.6766743659973145
    },
    {
      "epoch": 2.9087066650390625e-05,
      "grad_norm": 0.17826630175113678,
      "learning_rate": 8.325246343428206e-06,
      "loss": 0.0614,
      "step": 30500
    },
    {
      "epoch": 2.9087066650390625e-05,
      "model_forward_time": 1.723379373550415,
      "step": 30500
    },
    {
      "dynamics_loss_avg": 0.049644806422293186,
      "epoch": 2.9087066650390625e-05,
      "step": 30500
    },
    {
      "action_loss_avg": 0.0021584313712082804,
      "epoch": 2.9087066650390625e-05,
      "step": 30500
    },
    {
      "epoch": 2.9087066650390625e-05,
      "step": 30500,
      "training_step_time": 7.481019496917725
    },
    {
      "epoch": 2.9088020324707032e-05,
      "model_forward_time": 1.731943130493164,
      "step": 30501
    },
    {
      "epoch": 2.9088020324707032e-05,
      "step": 30501,
      "training_step_time": 6.743437051773071
    },
    {
      "epoch": 2.9088973999023436e-05,
      "model_forward_time": 1.7276451587677002,
      "step": 30502
    },
    {
      "epoch": 2.9088973999023436e-05,
      "step": 30502,
      "training_step_time": 6.90589714050293
    },
    {
      "epoch": 2.9089927673339844e-05,
      "model_forward_time": 1.7422230243682861,
      "step": 30503
    },
    {
      "epoch": 2.9089927673339844e-05,
      "step": 30503,
      "training_step_time": 6.895309209823608
    },
    {
      "epoch": 2.909088134765625e-05,
      "model_forward_time": 1.734776496887207,
      "step": 30504
    },
    {
      "epoch": 2.909088134765625e-05,
      "step": 30504,
      "training_step_time": 7.4402711391448975
    },
    {
      "epoch": 2.9091835021972655e-05,
      "model_forward_time": 1.7082009315490723,
      "step": 30505
    },
    {
      "epoch": 2.9091835021972655e-05,
      "step": 30505,
      "training_step_time": 7.731048345565796
    },
    {
      "epoch": 2.9092788696289063e-05,
      "model_forward_time": 2.292741298675537,
      "step": 30506
    },
    {
      "epoch": 2.9092788696289063e-05,
      "step": 30506,
      "training_step_time": 8.792391538619995
    },
    {
      "epoch": 2.909374237060547e-05,
      "model_forward_time": 1.719111442565918,
      "step": 30507
    },
    {
      "epoch": 2.909374237060547e-05,
      "step": 30507,
      "training_step_time": 7.774400234222412
    },
    {
      "epoch": 2.9094696044921874e-05,
      "model_forward_time": 1.7243146896362305,
      "step": 30508
    },
    {
      "epoch": 2.9094696044921874e-05,
      "step": 30508,
      "training_step_time": 8.713783025741577
    },
    {
      "epoch": 2.909564971923828e-05,
      "model_forward_time": 1.7208168506622314,
      "step": 30509
    },
    {
      "epoch": 2.909564971923828e-05,
      "step": 30509,
      "training_step_time": 7.53062891960144
    },
    {
      "epoch": 2.909660339355469e-05,
      "grad_norm": 0.1718030571937561,
      "learning_rate": 8.324011351221074e-06,
      "loss": 0.0595,
      "step": 30510
    },
    {
      "epoch": 2.909660339355469e-05,
      "model_forward_time": 1.7278711795806885,
      "step": 30510
    },
    {
      "dynamics_loss_avg": 0.05112192630767822,
      "epoch": 2.909660339355469e-05,
      "step": 30510
    },
    {
      "action_loss_avg": 0.001974851929116994,
      "epoch": 2.909660339355469e-05,
      "step": 30510
    },
    {
      "epoch": 2.909660339355469e-05,
      "step": 30510,
      "training_step_time": 6.9009315967559814
    },
    {
      "epoch": 2.9097557067871093e-05,
      "model_forward_time": 1.731783151626587,
      "step": 30511
    },
    {
      "epoch": 2.9097557067871093e-05,
      "step": 30511,
      "training_step_time": 6.943552732467651
    },
    {
      "epoch": 2.90985107421875e-05,
      "model_forward_time": 1.727388858795166,
      "step": 30512
    },
    {
      "epoch": 2.90985107421875e-05,
      "step": 30512,
      "training_step_time": 6.904451608657837
    },
    {
      "epoch": 2.9099464416503908e-05,
      "model_forward_time": 1.7378532886505127,
      "step": 30513
    },
    {
      "epoch": 2.9099464416503908e-05,
      "step": 30513,
      "training_step_time": 7.270605802536011
    },
    {
      "epoch": 2.9100418090820312e-05,
      "model_forward_time": 1.726996898651123,
      "step": 30514
    },
    {
      "epoch": 2.9100418090820312e-05,
      "step": 30514,
      "training_step_time": 8.64316701889038
    },
    {
      "epoch": 2.910137176513672e-05,
      "model_forward_time": 2.3164544105529785,
      "step": 30515
    },
    {
      "epoch": 2.910137176513672e-05,
      "step": 30515,
      "training_step_time": 7.841278791427612
    },
    {
      "epoch": 2.9102325439453123e-05,
      "model_forward_time": 1.7384798526763916,
      "step": 30516
    },
    {
      "epoch": 2.9102325439453123e-05,
      "step": 30516,
      "training_step_time": 7.7537078857421875
    },
    {
      "epoch": 2.910327911376953e-05,
      "model_forward_time": 1.7290661334991455,
      "step": 30517
    },
    {
      "epoch": 2.910327911376953e-05,
      "step": 30517,
      "training_step_time": 7.6719160079956055
    },
    {
      "epoch": 2.9104232788085938e-05,
      "model_forward_time": 1.7311604022979736,
      "step": 30518
    },
    {
      "epoch": 2.9104232788085938e-05,
      "step": 30518,
      "training_step_time": 7.503657341003418
    },
    {
      "epoch": 2.9105186462402342e-05,
      "model_forward_time": 1.7386822700500488,
      "step": 30519
    },
    {
      "epoch": 2.9105186462402342e-05,
      "step": 30519,
      "training_step_time": 6.705734729766846
    },
    {
      "epoch": 2.910614013671875e-05,
      "grad_norm": 0.1502188742160797,
      "learning_rate": 8.322775995505057e-06,
      "loss": 0.0596,
      "step": 30520
    },
    {
      "epoch": 2.910614013671875e-05,
      "model_forward_time": 1.750169277191162,
      "step": 30520
    },
    {
      "dynamics_loss_avg": 0.06602713689208031,
      "epoch": 2.910614013671875e-05,
      "step": 30520
    },
    {
      "action_loss_avg": 0.002393280854448676,
      "epoch": 2.910614013671875e-05,
      "step": 30520
    },
    {
      "epoch": 2.910614013671875e-05,
      "step": 30520,
      "training_step_time": 6.754047632217407
    },
    {
      "epoch": 2.9107093811035157e-05,
      "model_forward_time": 1.7266993522644043,
      "step": 30521
    },
    {
      "epoch": 2.9107093811035157e-05,
      "step": 30521,
      "training_step_time": 6.75309681892395
    },
    {
      "epoch": 2.910804748535156e-05,
      "model_forward_time": 1.726848840713501,
      "step": 30522
    },
    {
      "epoch": 2.910804748535156e-05,
      "step": 30522,
      "training_step_time": 7.290431261062622
    },
    {
      "epoch": 2.910900115966797e-05,
      "model_forward_time": 1.367128610610962,
      "step": 30523
    },
    {
      "epoch": 2.910900115966797e-05,
      "step": 30523,
      "training_step_time": 7.7296226024627686
    },
    {
      "epoch": 2.9109954833984376e-05,
      "model_forward_time": 1.7777833938598633,
      "step": 30524
    },
    {
      "epoch": 2.9109954833984376e-05,
      "step": 30524,
      "training_step_time": 8.632634401321411
    },
    {
      "epoch": 2.911090850830078e-05,
      "model_forward_time": 2.2823925018310547,
      "step": 30525
    },
    {
      "epoch": 2.911090850830078e-05,
      "step": 30525,
      "training_step_time": 10.325276374816895
    },
    {
      "epoch": 2.9111862182617187e-05,
      "model_forward_time": 1.7277522087097168,
      "step": 30526
    },
    {
      "epoch": 2.9111862182617187e-05,
      "step": 30526,
      "training_step_time": 7.745136260986328
    },
    {
      "epoch": 2.9112815856933595e-05,
      "model_forward_time": 1.7272922992706299,
      "step": 30527
    },
    {
      "epoch": 2.9112815856933595e-05,
      "step": 30527,
      "training_step_time": 8.32974648475647
    },
    {
      "epoch": 2.911376953125e-05,
      "model_forward_time": 1.7313554286956787,
      "step": 30528
    },
    {
      "epoch": 2.911376953125e-05,
      "step": 30528,
      "training_step_time": 6.714853525161743
    },
    {
      "epoch": 2.9114723205566406e-05,
      "model_forward_time": 1.7300941944122314,
      "step": 30529
    },
    {
      "epoch": 2.9114723205566406e-05,
      "step": 30529,
      "training_step_time": 6.745103359222412
    },
    {
      "epoch": 2.9115676879882814e-05,
      "grad_norm": 0.1676260083913803,
      "learning_rate": 8.321540276415254e-06,
      "loss": 0.0596,
      "step": 30530
    },
    {
      "epoch": 2.9115676879882814e-05,
      "model_forward_time": 1.7289514541625977,
      "step": 30530
    },
    {
      "dynamics_loss_avg": 0.06442481745034456,
      "epoch": 2.9115676879882814e-05,
      "step": 30530
    },
    {
      "action_loss_avg": 0.0017844125715782866,
      "epoch": 2.9115676879882814e-05,
      "step": 30530
    },
    {
      "epoch": 2.9115676879882814e-05,
      "step": 30530,
      "training_step_time": 6.752982139587402
    },
    {
      "epoch": 2.9116630554199218e-05,
      "model_forward_time": 1.7325668334960938,
      "step": 30531
    },
    {
      "epoch": 2.9116630554199218e-05,
      "step": 30531,
      "training_step_time": 7.282033443450928
    },
    {
      "epoch": 2.9117584228515625e-05,
      "model_forward_time": 1.728198766708374,
      "step": 30532
    },
    {
      "epoch": 2.9117584228515625e-05,
      "step": 30532,
      "training_step_time": 9.563781261444092
    },
    {
      "epoch": 2.9118537902832032e-05,
      "model_forward_time": 1.7724223136901855,
      "step": 30533
    },
    {
      "epoch": 2.9118537902832032e-05,
      "step": 30533,
      "training_step_time": 7.79423189163208
    },
    {
      "epoch": 2.9119491577148436e-05,
      "model_forward_time": 2.2784879207611084,
      "step": 30534
    },
    {
      "epoch": 2.9119491577148436e-05,
      "step": 30534,
      "training_step_time": 8.298650026321411
    },
    {
      "epoch": 2.9120445251464844e-05,
      "model_forward_time": 1.7202012538909912,
      "step": 30535
    },
    {
      "epoch": 2.9120445251464844e-05,
      "step": 30535,
      "training_step_time": 7.783332347869873
    },
    {
      "epoch": 2.912139892578125e-05,
      "model_forward_time": 1.721069097518921,
      "step": 30536
    },
    {
      "epoch": 2.912139892578125e-05,
      "step": 30536,
      "training_step_time": 7.700928449630737
    },
    {
      "epoch": 2.9122352600097655e-05,
      "model_forward_time": 1.730520248413086,
      "step": 30537
    },
    {
      "epoch": 2.9122352600097655e-05,
      "step": 30537,
      "training_step_time": 7.558272838592529
    },
    {
      "epoch": 2.9123306274414063e-05,
      "model_forward_time": 1.7181897163391113,
      "step": 30538
    },
    {
      "epoch": 2.9123306274414063e-05,
      "step": 30538,
      "training_step_time": 6.7997119426727295
    },
    {
      "epoch": 2.912425994873047e-05,
      "model_forward_time": 1.6948692798614502,
      "step": 30539
    },
    {
      "epoch": 2.912425994873047e-05,
      "step": 30539,
      "training_step_time": 6.786596059799194
    },
    {
      "epoch": 2.9125213623046874e-05,
      "grad_norm": 0.14089980721473694,
      "learning_rate": 8.320304194086798e-06,
      "loss": 0.0618,
      "step": 30540
    },
    {
      "epoch": 2.9125213623046874e-05,
      "model_forward_time": 1.7013700008392334,
      "step": 30540
    },
    {
      "dynamics_loss_avg": 0.054290874674916265,
      "epoch": 2.9125213623046874e-05,
      "step": 30540
    },
    {
      "action_loss_avg": 0.0014100042404606937,
      "epoch": 2.9125213623046874e-05,
      "step": 30540
    },
    {
      "epoch": 2.9125213623046874e-05,
      "step": 30540,
      "training_step_time": 7.665964126586914
    },
    {
      "epoch": 2.912616729736328e-05,
      "model_forward_time": 1.7056310176849365,
      "step": 30541
    },
    {
      "epoch": 2.912616729736328e-05,
      "step": 30541,
      "training_step_time": 8.804034948348999
    },
    {
      "epoch": 2.912712097167969e-05,
      "model_forward_time": 1.7165675163269043,
      "step": 30542
    },
    {
      "epoch": 2.912712097167969e-05,
      "step": 30542,
      "training_step_time": 9.470993518829346
    },
    {
      "epoch": 2.9128074645996093e-05,
      "model_forward_time": 2.252777576446533,
      "step": 30543
    },
    {
      "epoch": 2.9128074645996093e-05,
      "step": 30543,
      "training_step_time": 10.08568525314331
    },
    {
      "epoch": 2.91290283203125e-05,
      "model_forward_time": 1.7086825370788574,
      "step": 30544
    },
    {
      "epoch": 2.91290283203125e-05,
      "step": 30544,
      "training_step_time": 8.731139183044434
    },
    {
      "epoch": 2.9129981994628908e-05,
      "model_forward_time": 1.7044672966003418,
      "step": 30545
    },
    {
      "epoch": 2.9129981994628908e-05,
      "step": 30545,
      "training_step_time": 7.729295253753662
    },
    {
      "epoch": 2.9130935668945312e-05,
      "model_forward_time": 1.7009167671203613,
      "step": 30546
    },
    {
      "epoch": 2.9130935668945312e-05,
      "step": 30546,
      "training_step_time": 7.537600755691528
    },
    {
      "epoch": 2.913188934326172e-05,
      "model_forward_time": 1.7019543647766113,
      "step": 30547
    },
    {
      "epoch": 2.913188934326172e-05,
      "step": 30547,
      "training_step_time": 6.891808271408081
    },
    {
      "epoch": 2.9132843017578123e-05,
      "model_forward_time": 1.7076997756958008,
      "step": 30548
    },
    {
      "epoch": 2.9132843017578123e-05,
      "step": 30548,
      "training_step_time": 6.861848831176758
    },
    {
      "epoch": 2.913379669189453e-05,
      "model_forward_time": 1.7125482559204102,
      "step": 30549
    },
    {
      "epoch": 2.913379669189453e-05,
      "step": 30549,
      "training_step_time": 6.840747594833374
    },
    {
      "epoch": 2.9134750366210938e-05,
      "grad_norm": 0.13599829375743866,
      "learning_rate": 8.319067748654869e-06,
      "loss": 0.0593,
      "step": 30550
    },
    {
      "epoch": 2.9134750366210938e-05,
      "model_forward_time": 1.709604024887085,
      "step": 30550
    },
    {
      "dynamics_loss_avg": 0.04257468823343515,
      "epoch": 2.9134750366210938e-05,
      "step": 30550
    },
    {
      "action_loss_avg": 0.0023996239295229317,
      "epoch": 2.9134750366210938e-05,
      "step": 30550
    },
    {
      "epoch": 2.9134750366210938e-05,
      "step": 30550,
      "training_step_time": 8.357035636901855
    },
    {
      "epoch": 2.9135704040527342e-05,
      "model_forward_time": 1.746877908706665,
      "step": 30551
    },
    {
      "epoch": 2.9135704040527342e-05,
      "step": 30551,
      "training_step_time": 8.723480463027954
    },
    {
      "epoch": 2.913665771484375e-05,
      "model_forward_time": 2.261094570159912,
      "step": 30552
    },
    {
      "epoch": 2.913665771484375e-05,
      "step": 30552,
      "training_step_time": 7.985862731933594
    },
    {
      "epoch": 2.9137611389160157e-05,
      "model_forward_time": 1.7324283123016357,
      "step": 30553
    },
    {
      "epoch": 2.9137611389160157e-05,
      "step": 30553,
      "training_step_time": 8.068222999572754
    },
    {
      "epoch": 2.913856506347656e-05,
      "model_forward_time": 1.7405822277069092,
      "step": 30554
    },
    {
      "epoch": 2.913856506347656e-05,
      "step": 30554,
      "training_step_time": 9.715620756149292
    },
    {
      "epoch": 2.913951873779297e-05,
      "model_forward_time": 1.7493507862091064,
      "step": 30555
    },
    {
      "epoch": 2.913951873779297e-05,
      "step": 30555,
      "training_step_time": 7.59114146232605
    },
    {
      "epoch": 2.9140472412109376e-05,
      "model_forward_time": 1.7469117641448975,
      "step": 30556
    },
    {
      "epoch": 2.9140472412109376e-05,
      "step": 30556,
      "training_step_time": 7.243518829345703
    },
    {
      "epoch": 2.914142608642578e-05,
      "model_forward_time": 1.7519683837890625,
      "step": 30557
    },
    {
      "epoch": 2.914142608642578e-05,
      "step": 30557,
      "training_step_time": 6.8143675327301025
    },
    {
      "epoch": 2.9142379760742187e-05,
      "model_forward_time": 1.7491400241851807,
      "step": 30558
    },
    {
      "epoch": 2.9142379760742187e-05,
      "step": 30558,
      "training_step_time": 6.807873725891113
    },
    {
      "epoch": 2.9143333435058595e-05,
      "model_forward_time": 1.7515296936035156,
      "step": 30559
    },
    {
      "epoch": 2.9143333435058595e-05,
      "step": 30559,
      "training_step_time": 8.666164636611938
    },
    {
      "epoch": 2.9144287109375e-05,
      "grad_norm": 0.14401620626449585,
      "learning_rate": 8.317830940254682e-06,
      "loss": 0.0604,
      "step": 30560
    },
    {
      "epoch": 2.9144287109375e-05,
      "model_forward_time": 1.7831976413726807,
      "step": 30560
    },
    {
      "dynamics_loss_avg": 0.05323849748820066,
      "epoch": 2.9144287109375e-05,
      "step": 30560
    },
    {
      "action_loss_avg": 0.0022570837085368113,
      "epoch": 2.9144287109375e-05,
      "step": 30560
    },
    {
      "epoch": 2.9144287109375e-05,
      "step": 30560,
      "training_step_time": 8.555982112884521
    },
    {
      "epoch": 2.9145240783691406e-05,
      "model_forward_time": 2.835475206375122,
      "step": 30561
    },
    {
      "epoch": 2.9145240783691406e-05,
      "step": 30561,
      "training_step_time": 8.34936785697937
    },
    {
      "epoch": 2.9146194458007814e-05,
      "model_forward_time": 1.7454273700714111,
      "step": 30562
    },
    {
      "epoch": 2.9146194458007814e-05,
      "step": 30562,
      "training_step_time": 9.026485443115234
    },
    {
      "epoch": 2.9147148132324218e-05,
      "model_forward_time": 1.7522873878479004,
      "step": 30563
    },
    {
      "epoch": 2.9147148132324218e-05,
      "step": 30563,
      "training_step_time": 7.771841287612915
    },
    {
      "epoch": 2.9148101806640625e-05,
      "model_forward_time": 1.7476062774658203,
      "step": 30564
    },
    {
      "epoch": 2.9148101806640625e-05,
      "step": 30564,
      "training_step_time": 7.686838150024414
    },
    {
      "epoch": 2.9149055480957032e-05,
      "model_forward_time": 1.749563217163086,
      "step": 30565
    },
    {
      "epoch": 2.9149055480957032e-05,
      "step": 30565,
      "training_step_time": 7.3038787841796875
    },
    {
      "epoch": 2.9150009155273436e-05,
      "model_forward_time": 1.7449817657470703,
      "step": 30566
    },
    {
      "epoch": 2.9150009155273436e-05,
      "step": 30566,
      "training_step_time": 6.808804035186768
    },
    {
      "epoch": 2.9150962829589844e-05,
      "model_forward_time": 1.746734619140625,
      "step": 30567
    },
    {
      "epoch": 2.9150962829589844e-05,
      "step": 30567,
      "training_step_time": 6.776599645614624
    },
    {
      "epoch": 2.915191650390625e-05,
      "model_forward_time": 1.751103162765503,
      "step": 30568
    },
    {
      "epoch": 2.915191650390625e-05,
      "step": 30568,
      "training_step_time": 7.794931888580322
    },
    {
      "epoch": 2.9152870178222655e-05,
      "model_forward_time": 1.7740299701690674,
      "step": 30569
    },
    {
      "epoch": 2.9152870178222655e-05,
      "step": 30569,
      "training_step_time": 7.737242221832275
    },
    {
      "epoch": 2.9153823852539063e-05,
      "grad_norm": 0.13893409073352814,
      "learning_rate": 8.31659376902149e-06,
      "loss": 0.0612,
      "step": 30570
    },
    {
      "epoch": 2.9153823852539063e-05,
      "model_forward_time": 1.7447431087493896,
      "step": 30570
    },
    {
      "dynamics_loss_avg": 0.04117134194821119,
      "epoch": 2.9153823852539063e-05,
      "step": 30570
    },
    {
      "action_loss_avg": 0.0014206184539943933,
      "epoch": 2.9153823852539063e-05,
      "step": 30570
    },
    {
      "epoch": 2.9153823852539063e-05,
      "step": 30570,
      "training_step_time": 7.701432466506958
    },
    {
      "epoch": 2.915477752685547e-05,
      "model_forward_time": 2.8490726947784424,
      "step": 30571
    },
    {
      "epoch": 2.915477752685547e-05,
      "step": 30571,
      "training_step_time": 8.72780728340149
    },
    {
      "epoch": 2.9155731201171874e-05,
      "model_forward_time": 1.7484381198883057,
      "step": 30572
    },
    {
      "epoch": 2.9155731201171874e-05,
      "step": 30572,
      "training_step_time": 8.401232242584229
    },
    {
      "epoch": 2.915668487548828e-05,
      "model_forward_time": 1.7442834377288818,
      "step": 30573
    },
    {
      "epoch": 2.915668487548828e-05,
      "step": 30573,
      "training_step_time": 7.687965631484985
    },
    {
      "epoch": 2.915763854980469e-05,
      "model_forward_time": 1.7463221549987793,
      "step": 30574
    },
    {
      "epoch": 2.915763854980469e-05,
      "step": 30574,
      "training_step_time": 7.172547817230225
    },
    {
      "epoch": 2.9158592224121093e-05,
      "model_forward_time": 1.7500696182250977,
      "step": 30575
    },
    {
      "epoch": 2.9158592224121093e-05,
      "step": 30575,
      "training_step_time": 6.7457122802734375
    },
    {
      "epoch": 2.91595458984375e-05,
      "model_forward_time": 1.7429389953613281,
      "step": 30576
    },
    {
      "epoch": 2.91595458984375e-05,
      "step": 30576,
      "training_step_time": 6.774276971817017
    },
    {
      "epoch": 2.9160499572753908e-05,
      "model_forward_time": 1.7487072944641113,
      "step": 30577
    },
    {
      "epoch": 2.9160499572753908e-05,
      "step": 30577,
      "training_step_time": 7.619345188140869
    },
    {
      "epoch": 2.9161453247070312e-05,
      "model_forward_time": 1.3758339881896973,
      "step": 30578
    },
    {
      "epoch": 2.9161453247070312e-05,
      "step": 30578,
      "training_step_time": 8.463064670562744
    },
    {
      "epoch": 2.916240692138672e-05,
      "model_forward_time": 1.8144924640655518,
      "step": 30579
    },
    {
      "epoch": 2.916240692138672e-05,
      "step": 30579,
      "training_step_time": 7.676546573638916
    },
    {
      "epoch": 2.9163360595703123e-05,
      "grad_norm": 0.08572911471128464,
      "learning_rate": 8.315356235090592e-06,
      "loss": 0.0572,
      "step": 30580
    },
    {
      "epoch": 2.9163360595703123e-05,
      "model_forward_time": 2.8416967391967773,
      "step": 30580
    },
    {
      "dynamics_loss_avg": 0.05140016488730907,
      "epoch": 2.9163360595703123e-05,
      "step": 30580
    },
    {
      "action_loss_avg": 0.0026117807836271824,
      "epoch": 2.9163360595703123e-05,
      "step": 30580
    },
    {
      "epoch": 2.9163360595703123e-05,
      "step": 30580,
      "training_step_time": 9.472090005874634
    },
    {
      "epoch": 2.916431427001953e-05,
      "model_forward_time": 1.748722791671753,
      "step": 30581
    },
    {
      "epoch": 2.916431427001953e-05,
      "step": 30581,
      "training_step_time": 8.627527236938477
    },
    {
      "epoch": 2.9165267944335938e-05,
      "model_forward_time": 1.745365858078003,
      "step": 30582
    },
    {
      "epoch": 2.9165267944335938e-05,
      "step": 30582,
      "training_step_time": 7.700045824050903
    },
    {
      "epoch": 2.9166221618652342e-05,
      "model_forward_time": 1.735433578491211,
      "step": 30583
    },
    {
      "epoch": 2.9166221618652342e-05,
      "step": 30583,
      "training_step_time": 6.723620414733887
    },
    {
      "epoch": 2.916717529296875e-05,
      "model_forward_time": 1.7315037250518799,
      "step": 30584
    },
    {
      "epoch": 2.916717529296875e-05,
      "step": 30584,
      "training_step_time": 7.201249361038208
    },
    {
      "epoch": 2.9168128967285157e-05,
      "model_forward_time": 1.0238254070281982,
      "step": 30585
    },
    {
      "epoch": 2.9168128967285157e-05,
      "step": 30585,
      "training_step_time": 6.722106456756592
    },
    {
      "epoch": 2.916908264160156e-05,
      "model_forward_time": 1.744997262954712,
      "step": 30586
    },
    {
      "epoch": 2.916908264160156e-05,
      "step": 30586,
      "training_step_time": 7.735993146896362
    },
    {
      "epoch": 2.917003631591797e-05,
      "model_forward_time": 1.807178020477295,
      "step": 30587
    },
    {
      "epoch": 2.917003631591797e-05,
      "step": 30587,
      "training_step_time": 8.342000961303711
    },
    {
      "epoch": 2.9170989990234376e-05,
      "model_forward_time": 1.8074207305908203,
      "step": 30588
    },
    {
      "epoch": 2.9170989990234376e-05,
      "step": 30588,
      "training_step_time": 7.741270065307617
    },
    {
      "epoch": 2.917194366455078e-05,
      "model_forward_time": 2.8061649799346924,
      "step": 30589
    },
    {
      "epoch": 2.917194366455078e-05,
      "step": 30589,
      "training_step_time": 9.350128173828125
    },
    {
      "epoch": 2.9172897338867187e-05,
      "grad_norm": 0.11902987211942673,
      "learning_rate": 8.314118338597319e-06,
      "loss": 0.0592,
      "step": 30590
    },
    {
      "epoch": 2.9172897338867187e-05,
      "model_forward_time": 1.7389140129089355,
      "step": 30590
    },
    {
      "dynamics_loss_avg": 0.0511667225509882,
      "epoch": 2.9172897338867187e-05,
      "step": 30590
    },
    {
      "action_loss_avg": 0.0013941920187789947,
      "epoch": 2.9172897338867187e-05,
      "step": 30590
    },
    {
      "epoch": 2.9172897338867187e-05,
      "step": 30590,
      "training_step_time": 7.735241413116455
    },
    {
      "epoch": 2.9173851013183595e-05,
      "model_forward_time": 1.7380619049072266,
      "step": 30591
    },
    {
      "epoch": 2.9173851013183595e-05,
      "step": 30591,
      "training_step_time": 8.704779624938965
    },
    {
      "epoch": 2.91748046875e-05,
      "model_forward_time": 1.716116189956665,
      "step": 30592
    },
    {
      "epoch": 2.91748046875e-05,
      "step": 30592,
      "training_step_time": 7.161601305007935
    },
    {
      "epoch": 2.9175758361816406e-05,
      "model_forward_time": 1.7150681018829346,
      "step": 30593
    },
    {
      "epoch": 2.9175758361816406e-05,
      "step": 30593,
      "training_step_time": 7.273471117019653
    },
    {
      "epoch": 2.9176712036132814e-05,
      "model_forward_time": 1.732370376586914,
      "step": 30594
    },
    {
      "epoch": 2.9176712036132814e-05,
      "step": 30594,
      "training_step_time": 6.71333122253418
    },
    {
      "epoch": 2.9177665710449218e-05,
      "model_forward_time": 1.7425577640533447,
      "step": 30595
    },
    {
      "epoch": 2.9177665710449218e-05,
      "step": 30595,
      "training_step_time": 7.738317489624023
    },
    {
      "epoch": 2.9178619384765625e-05,
      "model_forward_time": 1.7357993125915527,
      "step": 30596
    },
    {
      "epoch": 2.9178619384765625e-05,
      "step": 30596,
      "training_step_time": 7.693537950515747
    },
    {
      "epoch": 2.9179573059082032e-05,
      "model_forward_time": 1.7598657608032227,
      "step": 30597
    },
    {
      "epoch": 2.9179573059082032e-05,
      "step": 30597,
      "training_step_time": 7.708556652069092
    },
    {
      "epoch": 2.9180526733398437e-05,
      "model_forward_time": 1.760422945022583,
      "step": 30598
    },
    {
      "epoch": 2.9180526733398437e-05,
      "step": 30598,
      "training_step_time": 10.050661087036133
    },
    {
      "epoch": 2.9181480407714844e-05,
      "model_forward_time": 2.849210262298584,
      "step": 30599
    },
    {
      "epoch": 2.9181480407714844e-05,
      "step": 30599,
      "training_step_time": 9.777826309204102
    },
    {
      "epoch": 2.918243408203125e-05,
      "grad_norm": 0.09839080274105072,
      "learning_rate": 8.312880079677049e-06,
      "loss": 0.0576,
      "step": 30600
    },
    {
      "epoch": 2.918243408203125e-05,
      "model_forward_time": 1.7355577945709229,
      "step": 30600
    },
    {
      "dynamics_loss_avg": 0.056401093304157254,
      "epoch": 2.918243408203125e-05,
      "step": 30600
    },
    {
      "action_loss_avg": 0.0022727141564246266,
      "epoch": 2.918243408203125e-05,
      "step": 30600
    },
    {
      "epoch": 2.918243408203125e-05,
      "step": 30600,
      "training_step_time": 7.67711877822876
    },
    {
      "epoch": 2.9183387756347655e-05,
      "model_forward_time": 1.7314589023590088,
      "step": 30601
    },
    {
      "epoch": 2.9183387756347655e-05,
      "step": 30601,
      "training_step_time": 7.17390513420105
    },
    {
      "epoch": 2.9184341430664063e-05,
      "model_forward_time": 1.7496895790100098,
      "step": 30602
    },
    {
      "epoch": 2.9184341430664063e-05,
      "step": 30602,
      "training_step_time": 7.309917211532593
    },
    {
      "epoch": 2.918529510498047e-05,
      "model_forward_time": 1.7291529178619385,
      "step": 30603
    },
    {
      "epoch": 2.918529510498047e-05,
      "step": 30603,
      "training_step_time": 6.700727701187134
    },
    {
      "epoch": 2.9186248779296874e-05,
      "model_forward_time": 1.7399170398712158,
      "step": 30604
    },
    {
      "epoch": 2.9186248779296874e-05,
      "step": 30604,
      "training_step_time": 7.697482109069824
    },
    {
      "epoch": 2.918720245361328e-05,
      "model_forward_time": 1.7286410331726074,
      "step": 30605
    },
    {
      "epoch": 2.918720245361328e-05,
      "step": 30605,
      "training_step_time": 7.783252239227295
    },
    {
      "epoch": 2.918815612792969e-05,
      "model_forward_time": 1.744295597076416,
      "step": 30606
    },
    {
      "epoch": 2.918815612792969e-05,
      "step": 30606,
      "training_step_time": 7.708846807479858
    },
    {
      "epoch": 2.9189109802246093e-05,
      "model_forward_time": 1.7488727569580078,
      "step": 30607
    },
    {
      "epoch": 2.9189109802246093e-05,
      "step": 30607,
      "training_step_time": 8.735697031021118
    },
    {
      "epoch": 2.91900634765625e-05,
      "model_forward_time": 2.901374101638794,
      "step": 30608
    },
    {
      "epoch": 2.91900634765625e-05,
      "step": 30608,
      "training_step_time": 8.308869361877441
    },
    {
      "epoch": 2.9191017150878908e-05,
      "model_forward_time": 1.7449328899383545,
      "step": 30609
    },
    {
      "epoch": 2.9191017150878908e-05,
      "step": 30609,
      "training_step_time": 7.833317041397095
    },
    {
      "epoch": 2.9191970825195312e-05,
      "grad_norm": 0.12947587668895721,
      "learning_rate": 8.311641458465191e-06,
      "loss": 0.0614,
      "step": 30610
    },
    {
      "epoch": 2.9191970825195312e-05,
      "model_forward_time": 1.7395436763763428,
      "step": 30610
    },
    {
      "dynamics_loss_avg": 0.05898567661643028,
      "epoch": 2.9191970825195312e-05,
      "step": 30610
    },
    {
      "action_loss_avg": 0.001857121632201597,
      "epoch": 2.9191970825195312e-05,
      "step": 30610
    },
    {
      "epoch": 2.9191970825195312e-05,
      "step": 30610,
      "training_step_time": 7.593662977218628
    },
    {
      "epoch": 2.919292449951172e-05,
      "model_forward_time": 1.7414045333862305,
      "step": 30611
    },
    {
      "epoch": 2.919292449951172e-05,
      "step": 30611,
      "training_step_time": 6.764117240905762
    },
    {
      "epoch": 2.9193878173828123e-05,
      "model_forward_time": 1.7318189144134521,
      "step": 30612
    },
    {
      "epoch": 2.9193878173828123e-05,
      "step": 30612,
      "training_step_time": 7.227532386779785
    },
    {
      "epoch": 2.919483184814453e-05,
      "model_forward_time": 1.7462832927703857,
      "step": 30613
    },
    {
      "epoch": 2.919483184814453e-05,
      "step": 30613,
      "training_step_time": 7.7293267250061035
    },
    {
      "epoch": 2.9195785522460938e-05,
      "model_forward_time": 1.7353532314300537,
      "step": 30614
    },
    {
      "epoch": 2.9195785522460938e-05,
      "step": 30614,
      "training_step_time": 7.745464324951172
    },
    {
      "epoch": 2.9196739196777342e-05,
      "model_forward_time": 1.742917776107788,
      "step": 30615
    },
    {
      "epoch": 2.9196739196777342e-05,
      "step": 30615,
      "training_step_time": 8.782908201217651
    },
    {
      "epoch": 2.919769287109375e-05,
      "model_forward_time": 1.740825891494751,
      "step": 30616
    },
    {
      "epoch": 2.919769287109375e-05,
      "step": 30616,
      "training_step_time": 7.749662160873413
    },
    {
      "epoch": 2.9198646545410157e-05,
      "model_forward_time": 2.9054267406463623,
      "step": 30617
    },
    {
      "epoch": 2.9198646545410157e-05,
      "step": 30617,
      "training_step_time": 7.787020444869995
    },
    {
      "epoch": 2.919960021972656e-05,
      "model_forward_time": 1.7466180324554443,
      "step": 30618
    },
    {
      "epoch": 2.919960021972656e-05,
      "step": 30618,
      "training_step_time": 7.75516152381897
    },
    {
      "epoch": 2.920055389404297e-05,
      "model_forward_time": 1.731011152267456,
      "step": 30619
    },
    {
      "epoch": 2.920055389404297e-05,
      "step": 30619,
      "training_step_time": 7.730737686157227
    },
    {
      "epoch": 2.9201507568359376e-05,
      "grad_norm": 0.14171257615089417,
      "learning_rate": 8.310402475097205e-06,
      "loss": 0.0597,
      "step": 30620
    },
    {
      "epoch": 2.9201507568359376e-05,
      "model_forward_time": 1.7392024993896484,
      "step": 30620
    },
    {
      "dynamics_loss_avg": 0.06933177225291728,
      "epoch": 2.9201507568359376e-05,
      "step": 30620
    },
    {
      "action_loss_avg": 0.0027677080128341914,
      "epoch": 2.9201507568359376e-05,
      "step": 30620
    },
    {
      "epoch": 2.9201507568359376e-05,
      "step": 30620,
      "training_step_time": 6.764976978302002
    },
    {
      "epoch": 2.920246124267578e-05,
      "model_forward_time": 1.7277138233184814,
      "step": 30621
    },
    {
      "epoch": 2.920246124267578e-05,
      "step": 30621,
      "training_step_time": 7.2420127391815186
    },
    {
      "epoch": 2.9203414916992187e-05,
      "model_forward_time": 1.7375595569610596,
      "step": 30622
    },
    {
      "epoch": 2.9203414916992187e-05,
      "step": 30622,
      "training_step_time": 7.73961067199707
    },
    {
      "epoch": 2.9204368591308595e-05,
      "model_forward_time": 1.7254712581634521,
      "step": 30623
    },
    {
      "epoch": 2.9204368591308595e-05,
      "step": 30623,
      "training_step_time": 7.665414094924927
    },
    {
      "epoch": 2.9205322265625e-05,
      "model_forward_time": 1.7240562438964844,
      "step": 30624
    },
    {
      "epoch": 2.9205322265625e-05,
      "step": 30624,
      "training_step_time": 7.728510856628418
    },
    {
      "epoch": 2.9206275939941406e-05,
      "model_forward_time": 1.8336262702941895,
      "step": 30625
    },
    {
      "epoch": 2.9206275939941406e-05,
      "step": 30625,
      "training_step_time": 9.293182611465454
    },
    {
      "epoch": 2.9207229614257814e-05,
      "model_forward_time": 2.9058456420898438,
      "step": 30626
    },
    {
      "epoch": 2.9207229614257814e-05,
      "step": 30626,
      "training_step_time": 8.725384712219238
    },
    {
      "epoch": 2.9208183288574218e-05,
      "model_forward_time": 1.7313306331634521,
      "step": 30627
    },
    {
      "epoch": 2.9208183288574218e-05,
      "step": 30627,
      "training_step_time": 10.1503005027771
    },
    {
      "epoch": 2.9209136962890625e-05,
      "model_forward_time": 1.7397058010101318,
      "step": 30628
    },
    {
      "epoch": 2.9209136962890625e-05,
      "step": 30628,
      "training_step_time": 7.741727113723755
    },
    {
      "epoch": 2.9210090637207033e-05,
      "model_forward_time": 1.7310607433319092,
      "step": 30629
    },
    {
      "epoch": 2.9210090637207033e-05,
      "step": 30629,
      "training_step_time": 6.764291048049927
    },
    {
      "epoch": 2.9211044311523437e-05,
      "grad_norm": 0.11965182423591614,
      "learning_rate": 8.30916312970858e-06,
      "loss": 0.0616,
      "step": 30630
    },
    {
      "epoch": 2.9211044311523437e-05,
      "model_forward_time": 1.748777151107788,
      "step": 30630
    },
    {
      "dynamics_loss_avg": 0.05225066356360912,
      "epoch": 2.9211044311523437e-05,
      "step": 30630
    },
    {
      "action_loss_avg": 0.0013177001965232193,
      "epoch": 2.9211044311523437e-05,
      "step": 30630
    },
    {
      "epoch": 2.9211044311523437e-05,
      "step": 30630,
      "training_step_time": 7.43181300163269
    },
    {
      "epoch": 2.9211997985839844e-05,
      "model_forward_time": 1.7271811962127686,
      "step": 30631
    },
    {
      "epoch": 2.9211997985839844e-05,
      "step": 30631,
      "training_step_time": 7.656010389328003
    },
    {
      "epoch": 2.921295166015625e-05,
      "model_forward_time": 1.7332115173339844,
      "step": 30632
    },
    {
      "epoch": 2.921295166015625e-05,
      "step": 30632,
      "training_step_time": 7.497915029525757
    },
    {
      "epoch": 2.9213905334472655e-05,
      "model_forward_time": 1.717921495437622,
      "step": 30633
    },
    {
      "epoch": 2.9213905334472655e-05,
      "step": 30633,
      "training_step_time": 7.731873989105225
    },
    {
      "epoch": 2.9214859008789063e-05,
      "model_forward_time": 1.7688093185424805,
      "step": 30634
    },
    {
      "epoch": 2.9214859008789063e-05,
      "step": 30634,
      "training_step_time": 7.651438236236572
    },
    {
      "epoch": 2.921581268310547e-05,
      "model_forward_time": 2.9003608226776123,
      "step": 30635
    },
    {
      "epoch": 2.921581268310547e-05,
      "step": 30635,
      "training_step_time": 8.309493780136108
    },
    {
      "epoch": 2.9216766357421874e-05,
      "model_forward_time": 1.716280221939087,
      "step": 30636
    },
    {
      "epoch": 2.9216766357421874e-05,
      "step": 30636,
      "training_step_time": 9.88066029548645
    },
    {
      "epoch": 2.921772003173828e-05,
      "model_forward_time": 1.716658353805542,
      "step": 30637
    },
    {
      "epoch": 2.921772003173828e-05,
      "step": 30637,
      "training_step_time": 9.465257167816162
    },
    {
      "epoch": 2.921867370605469e-05,
      "model_forward_time": 1.7318651676177979,
      "step": 30638
    },
    {
      "epoch": 2.921867370605469e-05,
      "step": 30638,
      "training_step_time": 6.631286144256592
    },
    {
      "epoch": 2.9219627380371093e-05,
      "model_forward_time": 1.7245995998382568,
      "step": 30639
    },
    {
      "epoch": 2.9219627380371093e-05,
      "step": 30639,
      "training_step_time": 6.744371175765991
    },
    {
      "epoch": 2.92205810546875e-05,
      "grad_norm": 0.12068299949169159,
      "learning_rate": 8.307923422434852e-06,
      "loss": 0.0631,
      "step": 30640
    },
    {
      "epoch": 2.92205810546875e-05,
      "model_forward_time": 1.72062087059021,
      "step": 30640
    },
    {
      "dynamics_loss_avg": 0.04651759508997202,
      "epoch": 2.92205810546875e-05,
      "step": 30640
    },
    {
      "action_loss_avg": 0.0019946889427956195,
      "epoch": 2.92205810546875e-05,
      "step": 30640
    },
    {
      "epoch": 2.92205810546875e-05,
      "step": 30640,
      "training_step_time": 7.69605565071106
    },
    {
      "epoch": 2.9221534729003905e-05,
      "model_forward_time": 1.724440574645996,
      "step": 30641
    },
    {
      "epoch": 2.9221534729003905e-05,
      "step": 30641,
      "training_step_time": 7.538390874862671
    },
    {
      "epoch": 2.9222488403320312e-05,
      "model_forward_time": 1.7226636409759521,
      "step": 30642
    },
    {
      "epoch": 2.9222488403320312e-05,
      "step": 30642,
      "training_step_time": 7.743348836898804
    },
    {
      "epoch": 2.922344207763672e-05,
      "model_forward_time": 1.734346866607666,
      "step": 30643
    },
    {
      "epoch": 2.922344207763672e-05,
      "step": 30643,
      "training_step_time": 8.468384742736816
    },
    {
      "epoch": 2.9224395751953123e-05,
      "model_forward_time": 2.8274505138397217,
      "step": 30644
    },
    {
      "epoch": 2.9224395751953123e-05,
      "step": 30644,
      "training_step_time": 9.48801040649414
    },
    {
      "epoch": 2.922534942626953e-05,
      "model_forward_time": 1.7158713340759277,
      "step": 30645
    },
    {
      "epoch": 2.922534942626953e-05,
      "step": 30645,
      "training_step_time": 9.609738826751709
    },
    {
      "epoch": 2.9226303100585938e-05,
      "model_forward_time": 1.7208616733551025,
      "step": 30646
    },
    {
      "epoch": 2.9226303100585938e-05,
      "step": 30646,
      "training_step_time": 8.884835481643677
    },
    {
      "epoch": 2.9227256774902342e-05,
      "model_forward_time": 1.3658208847045898,
      "step": 30647
    },
    {
      "epoch": 2.9227256774902342e-05,
      "step": 30647,
      "training_step_time": 7.638355016708374
    },
    {
      "epoch": 2.922821044921875e-05,
      "model_forward_time": 1.7290399074554443,
      "step": 30648
    },
    {
      "epoch": 2.922821044921875e-05,
      "step": 30648,
      "training_step_time": 7.053534507751465
    },
    {
      "epoch": 2.9229164123535157e-05,
      "model_forward_time": 1.7305684089660645,
      "step": 30649
    },
    {
      "epoch": 2.9229164123535157e-05,
      "step": 30649,
      "training_step_time": 7.715918064117432
    },
    {
      "epoch": 2.923011779785156e-05,
      "grad_norm": 0.11438727378845215,
      "learning_rate": 8.30668335341159e-06,
      "loss": 0.0603,
      "step": 30650
    },
    {
      "epoch": 2.923011779785156e-05,
      "model_forward_time": 1.7319860458374023,
      "step": 30650
    },
    {
      "dynamics_loss_avg": 0.06803745813667775,
      "epoch": 2.923011779785156e-05,
      "step": 30650
    },
    {
      "action_loss_avg": 0.0023032804194372146,
      "epoch": 2.923011779785156e-05,
      "step": 30650
    },
    {
      "epoch": 2.923011779785156e-05,
      "step": 30650,
      "training_step_time": 7.525421380996704
    },
    {
      "epoch": 2.923107147216797e-05,
      "model_forward_time": 1.7250118255615234,
      "step": 30651
    },
    {
      "epoch": 2.923107147216797e-05,
      "step": 30651,
      "training_step_time": 9.166714191436768
    },
    {
      "epoch": 2.9232025146484376e-05,
      "model_forward_time": 1.7384271621704102,
      "step": 30652
    },
    {
      "epoch": 2.9232025146484376e-05,
      "step": 30652,
      "training_step_time": 7.712438106536865
    },
    {
      "epoch": 2.923297882080078e-05,
      "model_forward_time": 2.8199169635772705,
      "step": 30653
    },
    {
      "epoch": 2.923297882080078e-05,
      "step": 30653,
      "training_step_time": 10.12777042388916
    },
    {
      "epoch": 2.9233932495117187e-05,
      "model_forward_time": 1.7254576683044434,
      "step": 30654
    },
    {
      "epoch": 2.9233932495117187e-05,
      "step": 30654,
      "training_step_time": 8.738450288772583
    },
    {
      "epoch": 2.9234886169433595e-05,
      "model_forward_time": 1.7319269180297852,
      "step": 30655
    },
    {
      "epoch": 2.9234886169433595e-05,
      "step": 30655,
      "training_step_time": 8.44257664680481
    },
    {
      "epoch": 2.923583984375e-05,
      "model_forward_time": 1.3597769737243652,
      "step": 30656
    },
    {
      "epoch": 2.923583984375e-05,
      "step": 30656,
      "training_step_time": 7.654937028884888
    },
    {
      "epoch": 2.9236793518066406e-05,
      "model_forward_time": 1.7270381450653076,
      "step": 30657
    },
    {
      "epoch": 2.9236793518066406e-05,
      "step": 30657,
      "training_step_time": 7.07353949546814
    },
    {
      "epoch": 2.9237747192382814e-05,
      "model_forward_time": 1.36566162109375,
      "step": 30658
    },
    {
      "epoch": 2.9237747192382814e-05,
      "step": 30658,
      "training_step_time": 7.721920490264893
    },
    {
      "epoch": 2.9238700866699218e-05,
      "model_forward_time": 1.722362756729126,
      "step": 30659
    },
    {
      "epoch": 2.9238700866699218e-05,
      "step": 30659,
      "training_step_time": 7.464717149734497
    },
    {
      "epoch": 2.9239654541015625e-05,
      "grad_norm": 0.19853372871875763,
      "learning_rate": 8.305442922774408e-06,
      "loss": 0.0533,
      "step": 30660
    },
    {
      "epoch": 2.9239654541015625e-05,
      "model_forward_time": 1.7401666641235352,
      "step": 30660
    },
    {
      "dynamics_loss_avg": 0.04648403078317642,
      "epoch": 2.9239654541015625e-05,
      "step": 30660
    },
    {
      "action_loss_avg": 0.0016421713167801498,
      "epoch": 2.9239654541015625e-05,
      "step": 30660
    },
    {
      "epoch": 2.9239654541015625e-05,
      "step": 30660,
      "training_step_time": 7.689936876296997
    },
    {
      "epoch": 2.9240608215332033e-05,
      "model_forward_time": 1.7334916591644287,
      "step": 30661
    },
    {
      "epoch": 2.9240608215332033e-05,
      "step": 30661,
      "training_step_time": 8.76863408088684
    },
    {
      "epoch": 2.9241561889648437e-05,
      "model_forward_time": 2.8359391689300537,
      "step": 30662
    },
    {
      "epoch": 2.9241561889648437e-05,
      "step": 30662,
      "training_step_time": 8.326633214950562
    },
    {
      "epoch": 2.9242515563964844e-05,
      "model_forward_time": 1.7371912002563477,
      "step": 30663
    },
    {
      "epoch": 2.9242515563964844e-05,
      "step": 30663,
      "training_step_time": 7.689922094345093
    },
    {
      "epoch": 2.924346923828125e-05,
      "model_forward_time": 1.7301504611968994,
      "step": 30664
    },
    {
      "epoch": 2.924346923828125e-05,
      "step": 30664,
      "training_step_time": 8.416853904724121
    },
    {
      "epoch": 2.9244422912597655e-05,
      "model_forward_time": 1.7307977676391602,
      "step": 30665
    },
    {
      "epoch": 2.9244422912597655e-05,
      "step": 30665,
      "training_step_time": 7.698671102523804
    },
    {
      "epoch": 2.9245376586914063e-05,
      "model_forward_time": 1.724076509475708,
      "step": 30666
    },
    {
      "epoch": 2.9245376586914063e-05,
      "step": 30666,
      "training_step_time": 6.884150743484497
    },
    {
      "epoch": 2.924633026123047e-05,
      "model_forward_time": 1.7342157363891602,
      "step": 30667
    },
    {
      "epoch": 2.924633026123047e-05,
      "step": 30667,
      "training_step_time": 7.801163911819458
    },
    {
      "epoch": 2.9247283935546874e-05,
      "model_forward_time": 1.7189981937408447,
      "step": 30668
    },
    {
      "epoch": 2.9247283935546874e-05,
      "step": 30668,
      "training_step_time": 7.495297908782959
    },
    {
      "epoch": 2.9248237609863282e-05,
      "model_forward_time": 1.7349073886871338,
      "step": 30669
    },
    {
      "epoch": 2.9248237609863282e-05,
      "step": 30669,
      "training_step_time": 7.728314399719238
    },
    {
      "epoch": 2.924919128417969e-05,
      "grad_norm": 0.23789222538471222,
      "learning_rate": 8.304202130658959e-06,
      "loss": 0.0594,
      "step": 30670
    },
    {
      "epoch": 2.924919128417969e-05,
      "model_forward_time": 1.7353558540344238,
      "step": 30670
    },
    {
      "dynamics_loss_avg": 0.06012350656092167,
      "epoch": 2.924919128417969e-05,
      "step": 30670
    },
    {
      "action_loss_avg": 0.0021001028071623296,
      "epoch": 2.924919128417969e-05,
      "step": 30670
    },
    {
      "epoch": 2.924919128417969e-05,
      "step": 30670,
      "training_step_time": 7.782405138015747
    },
    {
      "epoch": 2.9250144958496093e-05,
      "model_forward_time": 1.824758529663086,
      "step": 30671
    },
    {
      "epoch": 2.9250144958496093e-05,
      "step": 30671,
      "training_step_time": 7.74100136756897
    },
    {
      "epoch": 2.92510986328125e-05,
      "model_forward_time": 2.8787577152252197,
      "step": 30672
    },
    {
      "epoch": 2.92510986328125e-05,
      "step": 30672,
      "training_step_time": 9.325892448425293
    },
    {
      "epoch": 2.9252052307128905e-05,
      "model_forward_time": 1.7298197746276855,
      "step": 30673
    },
    {
      "epoch": 2.9252052307128905e-05,
      "step": 30673,
      "training_step_time": 8.253196954727173
    },
    {
      "epoch": 2.9253005981445312e-05,
      "model_forward_time": 1.717655897140503,
      "step": 30674
    },
    {
      "epoch": 2.9253005981445312e-05,
      "step": 30674,
      "training_step_time": 8.742491960525513
    },
    {
      "epoch": 2.925395965576172e-05,
      "model_forward_time": 1.7264983654022217,
      "step": 30675
    },
    {
      "epoch": 2.925395965576172e-05,
      "step": 30675,
      "training_step_time": 7.636938810348511
    },
    {
      "epoch": 2.9254913330078123e-05,
      "model_forward_time": 1.7227609157562256,
      "step": 30676
    },
    {
      "epoch": 2.9254913330078123e-05,
      "step": 30676,
      "training_step_time": 7.733577728271484
    },
    {
      "epoch": 2.925586700439453e-05,
      "model_forward_time": 1.7162368297576904,
      "step": 30677
    },
    {
      "epoch": 2.925586700439453e-05,
      "step": 30677,
      "training_step_time": 7.557523012161255
    },
    {
      "epoch": 2.9256820678710938e-05,
      "model_forward_time": 1.7236087322235107,
      "step": 30678
    },
    {
      "epoch": 2.9256820678710938e-05,
      "step": 30678,
      "training_step_time": 8.991020917892456
    },
    {
      "epoch": 2.9257774353027342e-05,
      "model_forward_time": 1.7144172191619873,
      "step": 30679
    },
    {
      "epoch": 2.9257774353027342e-05,
      "step": 30679,
      "training_step_time": 7.72167181968689
    },
    {
      "epoch": 2.925872802734375e-05,
      "grad_norm": 0.15373721718788147,
      "learning_rate": 8.302960977200932e-06,
      "loss": 0.0606,
      "step": 30680
    },
    {
      "epoch": 2.925872802734375e-05,
      "model_forward_time": 1.818695068359375,
      "step": 30680
    },
    {
      "dynamics_loss_avg": 0.05384685322642326,
      "epoch": 2.925872802734375e-05,
      "step": 30680
    },
    {
      "action_loss_avg": 0.0015827255760086701,
      "epoch": 2.925872802734375e-05,
      "step": 30680
    },
    {
      "epoch": 2.925872802734375e-05,
      "step": 30680,
      "training_step_time": 7.724813938140869
    },
    {
      "epoch": 2.9259681701660157e-05,
      "model_forward_time": 2.86171817779541,
      "step": 30681
    },
    {
      "epoch": 2.9259681701660157e-05,
      "step": 30681,
      "training_step_time": 11.70472240447998
    },
    {
      "epoch": 2.926063537597656e-05,
      "model_forward_time": 1.7230503559112549,
      "step": 30682
    },
    {
      "epoch": 2.926063537597656e-05,
      "step": 30682,
      "training_step_time": 9.561130285263062
    },
    {
      "epoch": 2.926158905029297e-05,
      "model_forward_time": 1.7248618602752686,
      "step": 30683
    },
    {
      "epoch": 2.926158905029297e-05,
      "step": 30683,
      "training_step_time": 7.732126474380493
    },
    {
      "epoch": 2.9262542724609376e-05,
      "model_forward_time": 1.7066223621368408,
      "step": 30684
    },
    {
      "epoch": 2.9262542724609376e-05,
      "step": 30684,
      "training_step_time": 7.6710569858551025
    },
    {
      "epoch": 2.926349639892578e-05,
      "model_forward_time": 1.7327444553375244,
      "step": 30685
    },
    {
      "epoch": 2.926349639892578e-05,
      "step": 30685,
      "training_step_time": 7.8006932735443115
    },
    {
      "epoch": 2.9264450073242187e-05,
      "model_forward_time": 1.0252680778503418,
      "step": 30686
    },
    {
      "epoch": 2.9264450073242187e-05,
      "step": 30686,
      "training_step_time": 7.516557455062866
    },
    {
      "epoch": 2.9265403747558595e-05,
      "model_forward_time": 1.7350335121154785,
      "step": 30687
    },
    {
      "epoch": 2.9265403747558595e-05,
      "step": 30687,
      "training_step_time": 7.707757949829102
    },
    {
      "epoch": 2.9266357421875e-05,
      "model_forward_time": 1.7176885604858398,
      "step": 30688
    },
    {
      "epoch": 2.9266357421875e-05,
      "step": 30688,
      "training_step_time": 7.814824104309082
    },
    {
      "epoch": 2.9267311096191406e-05,
      "model_forward_time": 1.7430973052978516,
      "step": 30689
    },
    {
      "epoch": 2.9267311096191406e-05,
      "step": 30689,
      "training_step_time": 9.886348724365234
    },
    {
      "epoch": 2.9268264770507814e-05,
      "grad_norm": 0.1858121007680893,
      "learning_rate": 8.301719462536059e-06,
      "loss": 0.0615,
      "step": 30690
    },
    {
      "epoch": 2.9268264770507814e-05,
      "model_forward_time": 2.4738080501556396,
      "step": 30690
    },
    {
      "dynamics_loss_avg": 0.05038056895136833,
      "epoch": 2.9268264770507814e-05,
      "step": 30690
    },
    {
      "action_loss_avg": 0.002258849932695739,
      "epoch": 2.9268264770507814e-05,
      "step": 30690
    },
    {
      "epoch": 2.9268264770507814e-05,
      "step": 30690,
      "training_step_time": 8.85905647277832
    },
    {
      "epoch": 2.9269218444824218e-05,
      "model_forward_time": 1.7308406829833984,
      "step": 30691
    },
    {
      "epoch": 2.9269218444824218e-05,
      "step": 30691,
      "training_step_time": 7.723740339279175
    },
    {
      "epoch": 2.9270172119140625e-05,
      "model_forward_time": 1.7216978073120117,
      "step": 30692
    },
    {
      "epoch": 2.9270172119140625e-05,
      "step": 30692,
      "training_step_time": 9.77169680595398
    },
    {
      "epoch": 2.9271125793457033e-05,
      "model_forward_time": 1.7210235595703125,
      "step": 30693
    },
    {
      "epoch": 2.9271125793457033e-05,
      "step": 30693,
      "training_step_time": 7.668560743331909
    },
    {
      "epoch": 2.9272079467773437e-05,
      "model_forward_time": 1.72271728515625,
      "step": 30694
    },
    {
      "epoch": 2.9272079467773437e-05,
      "step": 30694,
      "training_step_time": 7.681877851486206
    },
    {
      "epoch": 2.9273033142089844e-05,
      "model_forward_time": 1.7198834419250488,
      "step": 30695
    },
    {
      "epoch": 2.9273033142089844e-05,
      "step": 30695,
      "training_step_time": 7.498508930206299
    },
    {
      "epoch": 2.927398681640625e-05,
      "model_forward_time": 1.7246534824371338,
      "step": 30696
    },
    {
      "epoch": 2.927398681640625e-05,
      "step": 30696,
      "training_step_time": 7.717787027359009
    },
    {
      "epoch": 2.9274940490722655e-05,
      "model_forward_time": 1.7191319465637207,
      "step": 30697
    },
    {
      "epoch": 2.9274940490722655e-05,
      "step": 30697,
      "training_step_time": 7.74886679649353
    },
    {
      "epoch": 2.9275894165039063e-05,
      "model_forward_time": 1.73431396484375,
      "step": 30698
    },
    {
      "epoch": 2.9275894165039063e-05,
      "step": 30698,
      "training_step_time": 8.558144807815552
    },
    {
      "epoch": 2.927684783935547e-05,
      "model_forward_time": 2.8078980445861816,
      "step": 30699
    },
    {
      "epoch": 2.927684783935547e-05,
      "step": 30699,
      "training_step_time": 8.376001596450806
    },
    {
      "epoch": 2.9277801513671874e-05,
      "grad_norm": 0.16314765810966492,
      "learning_rate": 8.30047758680011e-06,
      "loss": 0.0653,
      "step": 30700
    },
    {
      "epoch": 2.9277801513671874e-05,
      "model_forward_time": 1.7193355560302734,
      "step": 30700
    },
    {
      "dynamics_loss_avg": 0.07083482686430216,
      "epoch": 2.9277801513671874e-05,
      "step": 30700
    },
    {
      "action_loss_avg": 0.0032070587971247734,
      "epoch": 2.9277801513671874e-05,
      "step": 30700
    },
    {
      "epoch": 2.9277801513671874e-05,
      "step": 30700,
      "training_step_time": 7.699579954147339
    },
    {
      "epoch": 2.9278755187988282e-05,
      "model_forward_time": 1.7242650985717773,
      "step": 30701
    },
    {
      "epoch": 2.9278755187988282e-05,
      "step": 30701,
      "training_step_time": 7.694571256637573
    },
    {
      "epoch": 2.927970886230469e-05,
      "model_forward_time": 1.7185070514678955,
      "step": 30702
    },
    {
      "epoch": 2.927970886230469e-05,
      "step": 30702,
      "training_step_time": 7.7316813468933105
    },
    {
      "epoch": 2.9280662536621093e-05,
      "model_forward_time": 1.7174482345581055,
      "step": 30703
    },
    {
      "epoch": 2.9280662536621093e-05,
      "step": 30703,
      "training_step_time": 7.696166753768921
    },
    {
      "epoch": 2.92816162109375e-05,
      "model_forward_time": 1.7150452136993408,
      "step": 30704
    },
    {
      "epoch": 2.92816162109375e-05,
      "step": 30704,
      "training_step_time": 7.477979898452759
    },
    {
      "epoch": 2.9282569885253905e-05,
      "model_forward_time": 1.7168073654174805,
      "step": 30705
    },
    {
      "epoch": 2.9282569885253905e-05,
      "step": 30705,
      "training_step_time": 7.7336485385894775
    },
    {
      "epoch": 2.9283523559570312e-05,
      "model_forward_time": 1.3466413021087646,
      "step": 30706
    },
    {
      "epoch": 2.9283523559570312e-05,
      "step": 30706,
      "training_step_time": 7.713574647903442
    },
    {
      "epoch": 2.928447723388672e-05,
      "model_forward_time": 1.3664767742156982,
      "step": 30707
    },
    {
      "epoch": 2.928447723388672e-05,
      "step": 30707,
      "training_step_time": 7.75573468208313
    },
    {
      "epoch": 2.9285430908203123e-05,
      "model_forward_time": 1.740330696105957,
      "step": 30708
    },
    {
      "epoch": 2.9285430908203123e-05,
      "step": 30708,
      "training_step_time": 8.856077432632446
    },
    {
      "epoch": 2.928638458251953e-05,
      "model_forward_time": 2.8725547790527344,
      "step": 30709
    },
    {
      "epoch": 2.928638458251953e-05,
      "step": 30709,
      "training_step_time": 8.210381746292114
    },
    {
      "epoch": 2.928733825683594e-05,
      "grad_norm": 0.15293683111667633,
      "learning_rate": 8.299235350128893e-06,
      "loss": 0.0609,
      "step": 30710
    },
    {
      "epoch": 2.928733825683594e-05,
      "model_forward_time": 1.7248013019561768,
      "step": 30710
    },
    {
      "dynamics_loss_avg": 0.052801992371678354,
      "epoch": 2.928733825683594e-05,
      "step": 30710
    },
    {
      "action_loss_avg": 0.002341870911186561,
      "epoch": 2.928733825683594e-05,
      "step": 30710
    },
    {
      "epoch": 2.928733825683594e-05,
      "step": 30710,
      "training_step_time": 7.60962986946106
    },
    {
      "epoch": 2.9288291931152342e-05,
      "model_forward_time": 1.7125060558319092,
      "step": 30711
    },
    {
      "epoch": 2.9288291931152342e-05,
      "step": 30711,
      "training_step_time": 7.716486215591431
    },
    {
      "epoch": 2.928924560546875e-05,
      "model_forward_time": 1.7221615314483643,
      "step": 30712
    },
    {
      "epoch": 2.928924560546875e-05,
      "step": 30712,
      "training_step_time": 7.45137619972229
    },
    {
      "epoch": 2.9290199279785157e-05,
      "model_forward_time": 1.7197155952453613,
      "step": 30713
    },
    {
      "epoch": 2.9290199279785157e-05,
      "step": 30713,
      "training_step_time": 7.692172050476074
    },
    {
      "epoch": 2.929115295410156e-05,
      "model_forward_time": 1.7196414470672607,
      "step": 30714
    },
    {
      "epoch": 2.929115295410156e-05,
      "step": 30714,
      "training_step_time": 7.704012870788574
    },
    {
      "epoch": 2.929210662841797e-05,
      "model_forward_time": 1.723304033279419,
      "step": 30715
    },
    {
      "epoch": 2.929210662841797e-05,
      "step": 30715,
      "training_step_time": 7.707732677459717
    },
    {
      "epoch": 2.9293060302734376e-05,
      "model_forward_time": 1.7218403816223145,
      "step": 30716
    },
    {
      "epoch": 2.9293060302734376e-05,
      "step": 30716,
      "training_step_time": 8.27473521232605
    },
    {
      "epoch": 2.929401397705078e-05,
      "model_forward_time": 1.8067176342010498,
      "step": 30717
    },
    {
      "epoch": 2.929401397705078e-05,
      "step": 30717,
      "training_step_time": 7.735238313674927
    },
    {
      "epoch": 2.9294967651367187e-05,
      "model_forward_time": 2.842414140701294,
      "step": 30718
    },
    {
      "epoch": 2.9294967651367187e-05,
      "step": 30718,
      "training_step_time": 8.226178169250488
    },
    {
      "epoch": 2.9295921325683595e-05,
      "model_forward_time": 1.7192015647888184,
      "step": 30719
    },
    {
      "epoch": 2.9295921325683595e-05,
      "step": 30719,
      "training_step_time": 10.054848194122314
    },
    {
      "epoch": 2.9296875e-05,
      "grad_norm": 0.1370890736579895,
      "learning_rate": 8.297992752658258e-06,
      "loss": 0.0656,
      "step": 30720
    },
    {
      "epoch": 2.9296875e-05,
      "model_forward_time": 1.7277333736419678,
      "step": 30720
    },
    {
      "dynamics_loss_avg": 0.06457775346934795,
      "epoch": 2.9296875e-05,
      "step": 30720
    },
    {
      "action_loss_avg": 0.0024275169824250044,
      "epoch": 2.9296875e-05,
      "step": 30720
    },
    {
      "epoch": 2.9296875e-05,
      "step": 30720,
      "training_step_time": 8.561751127243042
    },
    {
      "epoch": 2.9297828674316406e-05,
      "model_forward_time": 1.7213847637176514,
      "step": 30721
    },
    {
      "epoch": 2.9297828674316406e-05,
      "step": 30721,
      "training_step_time": 8.80351185798645
    },
    {
      "epoch": 2.9298782348632814e-05,
      "model_forward_time": 1.7233331203460693,
      "step": 30722
    },
    {
      "epoch": 2.9298782348632814e-05,
      "step": 30722,
      "training_step_time": 7.688122987747192
    },
    {
      "epoch": 2.9299736022949218e-05,
      "model_forward_time": 1.7185969352722168,
      "step": 30723
    },
    {
      "epoch": 2.9299736022949218e-05,
      "step": 30723,
      "training_step_time": 7.701301097869873
    },
    {
      "epoch": 2.9300689697265625e-05,
      "model_forward_time": 1.7295849323272705,
      "step": 30724
    },
    {
      "epoch": 2.9300689697265625e-05,
      "step": 30724,
      "training_step_time": 7.685480833053589
    },
    {
      "epoch": 2.9301643371582033e-05,
      "model_forward_time": 1.723785161972046,
      "step": 30725
    },
    {
      "epoch": 2.9301643371582033e-05,
      "step": 30725,
      "training_step_time": 7.686870336532593
    },
    {
      "epoch": 2.9302597045898437e-05,
      "model_forward_time": 1.7414782047271729,
      "step": 30726
    },
    {
      "epoch": 2.9302597045898437e-05,
      "step": 30726,
      "training_step_time": 8.260610342025757
    },
    {
      "epoch": 2.9303550720214844e-05,
      "model_forward_time": 2.8286049365997314,
      "step": 30727
    },
    {
      "epoch": 2.9303550720214844e-05,
      "step": 30727,
      "training_step_time": 7.683849334716797
    },
    {
      "epoch": 2.930450439453125e-05,
      "model_forward_time": 1.7199804782867432,
      "step": 30728
    },
    {
      "epoch": 2.930450439453125e-05,
      "step": 30728,
      "training_step_time": 7.90859055519104
    },
    {
      "epoch": 2.9305458068847655e-05,
      "model_forward_time": 1.7185986042022705,
      "step": 30729
    },
    {
      "epoch": 2.9305458068847655e-05,
      "step": 30729,
      "training_step_time": 8.642737627029419
    },
    {
      "epoch": 2.9306411743164063e-05,
      "grad_norm": 0.19290614128112793,
      "learning_rate": 8.296749794524095e-06,
      "loss": 0.0632,
      "step": 30730
    },
    {
      "epoch": 2.9306411743164063e-05,
      "model_forward_time": 1.721083164215088,
      "step": 30730
    },
    {
      "dynamics_loss_avg": 0.0511514462530613,
      "epoch": 2.9306411743164063e-05,
      "step": 30730
    },
    {
      "action_loss_avg": 0.0025338713196106253,
      "epoch": 2.9306411743164063e-05,
      "step": 30730
    },
    {
      "epoch": 2.9306411743164063e-05,
      "step": 30730,
      "training_step_time": 7.807831525802612
    },
    {
      "epoch": 2.930736541748047e-05,
      "model_forward_time": 1.7230939865112305,
      "step": 30731
    },
    {
      "epoch": 2.930736541748047e-05,
      "step": 30731,
      "training_step_time": 7.681213617324829
    },
    {
      "epoch": 2.9308319091796874e-05,
      "model_forward_time": 1.7163877487182617,
      "step": 30732
    },
    {
      "epoch": 2.9308319091796874e-05,
      "step": 30732,
      "training_step_time": 7.672563791275024
    },
    {
      "epoch": 2.9309272766113282e-05,
      "model_forward_time": 1.718855381011963,
      "step": 30733
    },
    {
      "epoch": 2.9309272766113282e-05,
      "step": 30733,
      "training_step_time": 7.705129146575928
    },
    {
      "epoch": 2.931022644042969e-05,
      "model_forward_time": 1.7140309810638428,
      "step": 30734
    },
    {
      "epoch": 2.931022644042969e-05,
      "step": 30734,
      "training_step_time": 7.463576316833496
    },
    {
      "epoch": 2.9311180114746093e-05,
      "model_forward_time": 1.7312867641448975,
      "step": 30735
    },
    {
      "epoch": 2.9311180114746093e-05,
      "step": 30735,
      "training_step_time": 8.225585222244263
    },
    {
      "epoch": 2.93121337890625e-05,
      "model_forward_time": 1.7264714241027832,
      "step": 30736
    },
    {
      "epoch": 2.93121337890625e-05,
      "step": 30736,
      "training_step_time": 9.035072565078735
    },
    {
      "epoch": 2.9313087463378905e-05,
      "model_forward_time": 1.7205166816711426,
      "step": 30737
    },
    {
      "epoch": 2.9313087463378905e-05,
      "step": 30737,
      "training_step_time": 7.776561260223389
    },
    {
      "epoch": 2.9314041137695312e-05,
      "model_forward_time": 1.7218005657196045,
      "step": 30738
    },
    {
      "epoch": 2.9314041137695312e-05,
      "step": 30738,
      "training_step_time": 7.767102956771851
    },
    {
      "epoch": 2.931499481201172e-05,
      "model_forward_time": 1.7196590900421143,
      "step": 30739
    },
    {
      "epoch": 2.931499481201172e-05,
      "step": 30739,
      "training_step_time": 8.798212766647339
    },
    {
      "epoch": 2.9315948486328124e-05,
      "grad_norm": 0.17007432878017426,
      "learning_rate": 8.295506475862332e-06,
      "loss": 0.0641,
      "step": 30740
    },
    {
      "epoch": 2.9315948486328124e-05,
      "model_forward_time": 1.7061817646026611,
      "step": 30740
    },
    {
      "dynamics_loss_avg": 0.05485232155770063,
      "epoch": 2.9315948486328124e-05,
      "step": 30740
    },
    {
      "action_loss_avg": 0.00206087400438264,
      "epoch": 2.9315948486328124e-05,
      "step": 30740
    },
    {
      "epoch": 2.9315948486328124e-05,
      "step": 30740,
      "training_step_time": 7.74934983253479
    },
    {
      "epoch": 2.931690216064453e-05,
      "model_forward_time": 1.7161839008331299,
      "step": 30741
    },
    {
      "epoch": 2.931690216064453e-05,
      "step": 30741,
      "training_step_time": 7.715373754501343
    },
    {
      "epoch": 2.931785583496094e-05,
      "model_forward_time": 1.7169363498687744,
      "step": 30742
    },
    {
      "epoch": 2.931785583496094e-05,
      "step": 30742,
      "training_step_time": 7.613108158111572
    },
    {
      "epoch": 2.9318809509277342e-05,
      "model_forward_time": 1.3617208003997803,
      "step": 30743
    },
    {
      "epoch": 2.9318809509277342e-05,
      "step": 30743,
      "training_step_time": 7.424014568328857
    },
    {
      "epoch": 2.931976318359375e-05,
      "model_forward_time": 1.7334671020507812,
      "step": 30744
    },
    {
      "epoch": 2.931976318359375e-05,
      "step": 30744,
      "training_step_time": 8.29589557647705
    },
    {
      "epoch": 2.9320716857910157e-05,
      "model_forward_time": 1.73606538772583,
      "step": 30745
    },
    {
      "epoch": 2.9320716857910157e-05,
      "step": 30745,
      "training_step_time": 9.025276899337769
    },
    {
      "epoch": 2.932167053222656e-05,
      "model_forward_time": 2.8781261444091797,
      "step": 30746
    },
    {
      "epoch": 2.932167053222656e-05,
      "step": 30746,
      "training_step_time": 10.44589614868164
    },
    {
      "epoch": 2.932262420654297e-05,
      "model_forward_time": 1.7175216674804688,
      "step": 30747
    },
    {
      "epoch": 2.932262420654297e-05,
      "step": 30747,
      "training_step_time": 8.405153512954712
    },
    {
      "epoch": 2.9323577880859376e-05,
      "model_forward_time": 1.7270469665527344,
      "step": 30748
    },
    {
      "epoch": 2.9323577880859376e-05,
      "step": 30748,
      "training_step_time": 8.463633298873901
    },
    {
      "epoch": 2.932453155517578e-05,
      "model_forward_time": 1.7127797603607178,
      "step": 30749
    },
    {
      "epoch": 2.932453155517578e-05,
      "step": 30749,
      "training_step_time": 7.622058868408203
    },
    {
      "epoch": 2.9325485229492188e-05,
      "grad_norm": 0.17310070991516113,
      "learning_rate": 8.294262796808934e-06,
      "loss": 0.0646,
      "step": 30750
    },
    {
      "epoch": 2.9325485229492188e-05,
      "model_forward_time": 1.7342681884765625,
      "step": 30750
    },
    {
      "dynamics_loss_avg": 0.04472865480929613,
      "epoch": 2.9325485229492188e-05,
      "step": 30750
    },
    {
      "action_loss_avg": 0.0015421535092173145,
      "epoch": 2.9325485229492188e-05,
      "step": 30750
    },
    {
      "epoch": 2.9325485229492188e-05,
      "step": 30750,
      "training_step_time": 7.612311363220215
    },
    {
      "epoch": 2.9326438903808595e-05,
      "model_forward_time": 1.7124407291412354,
      "step": 30751
    },
    {
      "epoch": 2.9326438903808595e-05,
      "step": 30751,
      "training_step_time": 7.727401256561279
    },
    {
      "epoch": 2.9327392578125e-05,
      "model_forward_time": 1.720310926437378,
      "step": 30752
    },
    {
      "epoch": 2.9327392578125e-05,
      "step": 30752,
      "training_step_time": 7.698776006698608
    },
    {
      "epoch": 2.9328346252441406e-05,
      "model_forward_time": 1.3496272563934326,
      "step": 30753
    },
    {
      "epoch": 2.9328346252441406e-05,
      "step": 30753,
      "training_step_time": 8.656246900558472
    },
    {
      "epoch": 2.9329299926757814e-05,
      "model_forward_time": 1.732652187347412,
      "step": 30754
    },
    {
      "epoch": 2.9329299926757814e-05,
      "step": 30754,
      "training_step_time": 7.578861951828003
    },
    {
      "epoch": 2.9330253601074218e-05,
      "model_forward_time": 2.4664735794067383,
      "step": 30755
    },
    {
      "epoch": 2.9330253601074218e-05,
      "step": 30755,
      "training_step_time": 9.234094619750977
    },
    {
      "epoch": 2.9331207275390625e-05,
      "model_forward_time": 1.7200055122375488,
      "step": 30756
    },
    {
      "epoch": 2.9331207275390625e-05,
      "step": 30756,
      "training_step_time": 7.714510202407837
    },
    {
      "epoch": 2.9332160949707033e-05,
      "model_forward_time": 1.7125718593597412,
      "step": 30757
    },
    {
      "epoch": 2.9332160949707033e-05,
      "step": 30757,
      "training_step_time": 7.593792915344238
    },
    {
      "epoch": 2.9333114624023437e-05,
      "model_forward_time": 1.7193121910095215,
      "step": 30758
    },
    {
      "epoch": 2.9333114624023437e-05,
      "step": 30758,
      "training_step_time": 7.631850957870483
    },
    {
      "epoch": 2.9334068298339844e-05,
      "model_forward_time": 1.7161030769348145,
      "step": 30759
    },
    {
      "epoch": 2.9334068298339844e-05,
      "step": 30759,
      "training_step_time": 8.722126960754395
    },
    {
      "epoch": 2.933502197265625e-05,
      "grad_norm": 0.17374521493911743,
      "learning_rate": 8.29301875749991e-06,
      "loss": 0.0614,
      "step": 30760
    },
    {
      "epoch": 2.933502197265625e-05,
      "model_forward_time": 1.3569204807281494,
      "step": 30760
    },
    {
      "dynamics_loss_avg": 0.05278368555009365,
      "epoch": 2.933502197265625e-05,
      "step": 30760
    },
    {
      "action_loss_avg": 0.0029401599895209074,
      "epoch": 2.933502197265625e-05,
      "step": 30760
    },
    {
      "epoch": 2.933502197265625e-05,
      "step": 30760,
      "training_step_time": 7.783114194869995
    },
    {
      "epoch": 2.9335975646972656e-05,
      "model_forward_time": 1.7280192375183105,
      "step": 30761
    },
    {
      "epoch": 2.9335975646972656e-05,
      "step": 30761,
      "training_step_time": 7.695092678070068
    },
    {
      "epoch": 2.9336929321289063e-05,
      "model_forward_time": 1.720745325088501,
      "step": 30762
    },
    {
      "epoch": 2.9336929321289063e-05,
      "step": 30762,
      "training_step_time": 7.701549291610718
    },
    {
      "epoch": 2.933788299560547e-05,
      "model_forward_time": 1.784592866897583,
      "step": 30763
    },
    {
      "epoch": 2.933788299560547e-05,
      "step": 30763,
      "training_step_time": 8.524444580078125
    },
    {
      "epoch": 2.9338836669921874e-05,
      "model_forward_time": 1.7215349674224854,
      "step": 30764
    },
    {
      "epoch": 2.9338836669921874e-05,
      "step": 30764,
      "training_step_time": 10.21362829208374
    },
    {
      "epoch": 2.9339790344238282e-05,
      "model_forward_time": 2.885413646697998,
      "step": 30765
    },
    {
      "epoch": 2.9339790344238282e-05,
      "step": 30765,
      "training_step_time": 7.832226514816284
    },
    {
      "epoch": 2.9340744018554686e-05,
      "model_forward_time": 1.7164087295532227,
      "step": 30766
    },
    {
      "epoch": 2.9340744018554686e-05,
      "step": 30766,
      "training_step_time": 7.6981964111328125
    },
    {
      "epoch": 2.9341697692871093e-05,
      "model_forward_time": 1.726445198059082,
      "step": 30767
    },
    {
      "epoch": 2.9341697692871093e-05,
      "step": 30767,
      "training_step_time": 8.735333681106567
    },
    {
      "epoch": 2.93426513671875e-05,
      "model_forward_time": 1.7357749938964844,
      "step": 30768
    },
    {
      "epoch": 2.93426513671875e-05,
      "step": 30768,
      "training_step_time": 7.778677225112915
    },
    {
      "epoch": 2.9343605041503905e-05,
      "model_forward_time": 1.7203493118286133,
      "step": 30769
    },
    {
      "epoch": 2.9343605041503905e-05,
      "step": 30769,
      "training_step_time": 7.755876064300537
    },
    {
      "epoch": 2.9344558715820312e-05,
      "grad_norm": 0.16055928170681,
      "learning_rate": 8.291774358071304e-06,
      "loss": 0.0575,
      "step": 30770
    },
    {
      "epoch": 2.9344558715820312e-05,
      "model_forward_time": 1.7211685180664062,
      "step": 30770
    },
    {
      "dynamics_loss_avg": 0.06968978606164455,
      "epoch": 2.9344558715820312e-05,
      "step": 30770
    },
    {
      "action_loss_avg": 0.002028854127274826,
      "epoch": 2.9344558715820312e-05,
      "step": 30770
    },
    {
      "epoch": 2.9344558715820312e-05,
      "step": 30770,
      "training_step_time": 7.688555479049683
    },
    {
      "epoch": 2.934551239013672e-05,
      "model_forward_time": 1.7175633907318115,
      "step": 30771
    },
    {
      "epoch": 2.934551239013672e-05,
      "step": 30771,
      "training_step_time": 7.623195171356201
    },
    {
      "epoch": 2.9346466064453124e-05,
      "model_forward_time": 1.742978572845459,
      "step": 30772
    },
    {
      "epoch": 2.9346466064453124e-05,
      "step": 30772,
      "training_step_time": 7.758294582366943
    },
    {
      "epoch": 2.934741973876953e-05,
      "model_forward_time": 1.7442331314086914,
      "step": 30773
    },
    {
      "epoch": 2.934741973876953e-05,
      "step": 30773,
      "training_step_time": 9.478427648544312
    },
    {
      "epoch": 2.934837341308594e-05,
      "model_forward_time": 2.872730255126953,
      "step": 30774
    },
    {
      "epoch": 2.934837341308594e-05,
      "step": 30774,
      "training_step_time": 7.7020134925842285
    },
    {
      "epoch": 2.9349327087402342e-05,
      "model_forward_time": 1.7268311977386475,
      "step": 30775
    },
    {
      "epoch": 2.9349327087402342e-05,
      "step": 30775,
      "training_step_time": 9.069929122924805
    },
    {
      "epoch": 2.935028076171875e-05,
      "model_forward_time": 1.722555160522461,
      "step": 30776
    },
    {
      "epoch": 2.935028076171875e-05,
      "step": 30776,
      "training_step_time": 8.723284482955933
    },
    {
      "epoch": 2.9351234436035157e-05,
      "model_forward_time": 1.7193191051483154,
      "step": 30777
    },
    {
      "epoch": 2.9351234436035157e-05,
      "step": 30777,
      "training_step_time": 7.651920557022095
    },
    {
      "epoch": 2.935218811035156e-05,
      "model_forward_time": 1.7221109867095947,
      "step": 30778
    },
    {
      "epoch": 2.935218811035156e-05,
      "step": 30778,
      "training_step_time": 7.689604043960571
    },
    {
      "epoch": 2.935314178466797e-05,
      "model_forward_time": 1.7220003604888916,
      "step": 30779
    },
    {
      "epoch": 2.935314178466797e-05,
      "step": 30779,
      "training_step_time": 7.759084224700928
    },
    {
      "epoch": 2.9354095458984376e-05,
      "grad_norm": 0.14412173628807068,
      "learning_rate": 8.290529598659206e-06,
      "loss": 0.0558,
      "step": 30780
    },
    {
      "epoch": 2.9354095458984376e-05,
      "model_forward_time": 1.7232308387756348,
      "step": 30780
    },
    {
      "dynamics_loss_avg": 0.05643521714955568,
      "epoch": 2.9354095458984376e-05,
      "step": 30780
    },
    {
      "action_loss_avg": 0.0025049102056073024,
      "epoch": 2.9354095458984376e-05,
      "step": 30780
    },
    {
      "epoch": 2.9354095458984376e-05,
      "step": 30780,
      "training_step_time": 7.643008708953857
    },
    {
      "epoch": 2.935504913330078e-05,
      "model_forward_time": 1.719627857208252,
      "step": 30781
    },
    {
      "epoch": 2.935504913330078e-05,
      "step": 30781,
      "training_step_time": 7.777617454528809
    },
    {
      "epoch": 2.9356002807617188e-05,
      "model_forward_time": 1.7627477645874023,
      "step": 30782
    },
    {
      "epoch": 2.9356002807617188e-05,
      "step": 30782,
      "training_step_time": 7.714074373245239
    },
    {
      "epoch": 2.9356956481933595e-05,
      "model_forward_time": 2.888676643371582,
      "step": 30783
    },
    {
      "epoch": 2.9356956481933595e-05,
      "step": 30783,
      "training_step_time": 8.361153364181519
    },
    {
      "epoch": 2.935791015625e-05,
      "model_forward_time": 1.7184364795684814,
      "step": 30784
    },
    {
      "epoch": 2.935791015625e-05,
      "step": 30784,
      "training_step_time": 7.6553003787994385
    },
    {
      "epoch": 2.9358863830566406e-05,
      "model_forward_time": 1.7312297821044922,
      "step": 30785
    },
    {
      "epoch": 2.9358863830566406e-05,
      "step": 30785,
      "training_step_time": 7.688187837600708
    },
    {
      "epoch": 2.9359817504882814e-05,
      "model_forward_time": 1.7216618061065674,
      "step": 30786
    },
    {
      "epoch": 2.9359817504882814e-05,
      "step": 30786,
      "training_step_time": 7.664472341537476
    },
    {
      "epoch": 2.9360771179199218e-05,
      "model_forward_time": 1.7264671325683594,
      "step": 30787
    },
    {
      "epoch": 2.9360771179199218e-05,
      "step": 30787,
      "training_step_time": 7.665438652038574
    },
    {
      "epoch": 2.9361724853515625e-05,
      "model_forward_time": 1.7158093452453613,
      "step": 30788
    },
    {
      "epoch": 2.9361724853515625e-05,
      "step": 30788,
      "training_step_time": 7.681795120239258
    },
    {
      "epoch": 2.9362678527832033e-05,
      "model_forward_time": 1.7237048149108887,
      "step": 30789
    },
    {
      "epoch": 2.9362678527832033e-05,
      "step": 30789,
      "training_step_time": 8.396604299545288
    },
    {
      "epoch": 2.9363632202148437e-05,
      "grad_norm": 0.19031323492527008,
      "learning_rate": 8.289284479399737e-06,
      "loss": 0.0619,
      "step": 30790
    },
    {
      "epoch": 2.9363632202148437e-05,
      "model_forward_time": 1.7205567359924316,
      "step": 30790
    },
    {
      "dynamics_loss_avg": 0.06035687159746885,
      "epoch": 2.9363632202148437e-05,
      "step": 30790
    },
    {
      "action_loss_avg": 0.00243604198039975,
      "epoch": 2.9363632202148437e-05,
      "step": 30790
    },
    {
      "epoch": 2.9363632202148437e-05,
      "step": 30790,
      "training_step_time": 7.689422845840454
    },
    {
      "epoch": 2.9364585876464844e-05,
      "model_forward_time": 1.8156719207763672,
      "step": 30791
    },
    {
      "epoch": 2.9364585876464844e-05,
      "step": 30791,
      "training_step_time": 7.67972469329834
    },
    {
      "epoch": 2.936553955078125e-05,
      "model_forward_time": 2.8382625579833984,
      "step": 30792
    },
    {
      "epoch": 2.936553955078125e-05,
      "step": 30792,
      "training_step_time": 7.655281066894531
    },
    {
      "epoch": 2.9366493225097656e-05,
      "model_forward_time": 1.7202353477478027,
      "step": 30793
    },
    {
      "epoch": 2.9366493225097656e-05,
      "step": 30793,
      "training_step_time": 7.674389362335205
    },
    {
      "epoch": 2.9367446899414063e-05,
      "model_forward_time": 1.7197365760803223,
      "step": 30794
    },
    {
      "epoch": 2.9367446899414063e-05,
      "step": 30794,
      "training_step_time": 8.391357183456421
    },
    {
      "epoch": 2.936840057373047e-05,
      "model_forward_time": 1.7210524082183838,
      "step": 30795
    },
    {
      "epoch": 2.936840057373047e-05,
      "step": 30795,
      "training_step_time": 7.702620029449463
    },
    {
      "epoch": 2.9369354248046874e-05,
      "model_forward_time": 1.7226197719573975,
      "step": 30796
    },
    {
      "epoch": 2.9369354248046874e-05,
      "step": 30796,
      "training_step_time": 7.651737928390503
    },
    {
      "epoch": 2.9370307922363282e-05,
      "model_forward_time": 1.719968318939209,
      "step": 30797
    },
    {
      "epoch": 2.9370307922363282e-05,
      "step": 30797,
      "training_step_time": 7.733175992965698
    },
    {
      "epoch": 2.9371261596679686e-05,
      "model_forward_time": 1.7211806774139404,
      "step": 30798
    },
    {
      "epoch": 2.9371261596679686e-05,
      "step": 30798,
      "training_step_time": 7.68008279800415
    },
    {
      "epoch": 2.9372215270996093e-05,
      "model_forward_time": 1.7192845344543457,
      "step": 30799
    },
    {
      "epoch": 2.9372215270996093e-05,
      "step": 30799,
      "training_step_time": 7.737138509750366
    },
    {
      "epoch": 2.93731689453125e-05,
      "grad_norm": 0.16856059432029724,
      "learning_rate": 8.288039000429064e-06,
      "loss": 0.0596,
      "step": 30800
    },
    {
      "epoch": 2.93731689453125e-05,
      "model_forward_time": 1.763315200805664,
      "step": 30800
    },
    {
      "dynamics_loss_avg": 0.07081305924803019,
      "epoch": 2.93731689453125e-05,
      "step": 30800
    },
    {
      "action_loss_avg": 0.0022207826230442153,
      "epoch": 2.93731689453125e-05,
      "step": 30800
    },
    {
      "epoch": 2.93731689453125e-05,
      "step": 30800,
      "training_step_time": 8.506977081298828
    },
    {
      "epoch": 2.9374122619628905e-05,
      "model_forward_time": 2.790696620941162,
      "step": 30801
    },
    {
      "epoch": 2.9374122619628905e-05,
      "step": 30801,
      "training_step_time": 7.743661403656006
    },
    {
      "epoch": 2.9375076293945312e-05,
      "model_forward_time": 1.71919584274292,
      "step": 30802
    },
    {
      "epoch": 2.9375076293945312e-05,
      "step": 30802,
      "training_step_time": 9.65536904335022
    },
    {
      "epoch": 2.937602996826172e-05,
      "model_forward_time": 1.7156195640563965,
      "step": 30803
    },
    {
      "epoch": 2.937602996826172e-05,
      "step": 30803,
      "training_step_time": 8.323636770248413
    },
    {
      "epoch": 2.9376983642578124e-05,
      "model_forward_time": 1.7250142097473145,
      "step": 30804
    },
    {
      "epoch": 2.9376983642578124e-05,
      "step": 30804,
      "training_step_time": 7.789227247238159
    },
    {
      "epoch": 2.937793731689453e-05,
      "model_forward_time": 1.7172484397888184,
      "step": 30805
    },
    {
      "epoch": 2.937793731689453e-05,
      "step": 30805,
      "training_step_time": 7.762129783630371
    },
    {
      "epoch": 2.937889099121094e-05,
      "model_forward_time": 1.71820068359375,
      "step": 30806
    },
    {
      "epoch": 2.937889099121094e-05,
      "step": 30806,
      "training_step_time": 8.626358985900879
    },
    {
      "epoch": 2.9379844665527342e-05,
      "model_forward_time": 1.7187981605529785,
      "step": 30807
    },
    {
      "epoch": 2.9379844665527342e-05,
      "step": 30807,
      "training_step_time": 8.283533811569214
    },
    {
      "epoch": 2.938079833984375e-05,
      "model_forward_time": 1.7174410820007324,
      "step": 30808
    },
    {
      "epoch": 2.938079833984375e-05,
      "step": 30808,
      "training_step_time": 8.652493953704834
    },
    {
      "epoch": 2.9381752014160157e-05,
      "model_forward_time": 1.7315857410430908,
      "step": 30809
    },
    {
      "epoch": 2.9381752014160157e-05,
      "step": 30809,
      "training_step_time": 7.728151798248291
    },
    {
      "epoch": 2.938270568847656e-05,
      "grad_norm": 0.14066053926944733,
      "learning_rate": 8.28679316188339e-06,
      "loss": 0.0538,
      "step": 30810
    },
    {
      "epoch": 2.938270568847656e-05,
      "model_forward_time": 1.7509281635284424,
      "step": 30810
    },
    {
      "dynamics_loss_avg": 0.05243538301438093,
      "epoch": 2.938270568847656e-05,
      "step": 30810
    },
    {
      "action_loss_avg": 0.0020857897063251583,
      "epoch": 2.938270568847656e-05,
      "step": 30810
    },
    {
      "epoch": 2.938270568847656e-05,
      "step": 30810,
      "training_step_time": 9.382458925247192
    },
    {
      "epoch": 2.938365936279297e-05,
      "model_forward_time": 2.5376622676849365,
      "step": 30811
    },
    {
      "epoch": 2.938365936279297e-05,
      "step": 30811,
      "training_step_time": 7.811954736709595
    },
    {
      "epoch": 2.9384613037109376e-05,
      "model_forward_time": 1.7202422618865967,
      "step": 30812
    },
    {
      "epoch": 2.9384613037109376e-05,
      "step": 30812,
      "training_step_time": 7.730652809143066
    },
    {
      "epoch": 2.938556671142578e-05,
      "model_forward_time": 1.7251031398773193,
      "step": 30813
    },
    {
      "epoch": 2.938556671142578e-05,
      "step": 30813,
      "training_step_time": 8.627378225326538
    },
    {
      "epoch": 2.9386520385742188e-05,
      "model_forward_time": 1.7149832248687744,
      "step": 30814
    },
    {
      "epoch": 2.9386520385742188e-05,
      "step": 30814,
      "training_step_time": 7.781858444213867
    },
    {
      "epoch": 2.9387474060058595e-05,
      "model_forward_time": 1.7223691940307617,
      "step": 30815
    },
    {
      "epoch": 2.9387474060058595e-05,
      "step": 30815,
      "training_step_time": 7.806401968002319
    },
    {
      "epoch": 2.9388427734375e-05,
      "model_forward_time": 1.7252726554870605,
      "step": 30816
    },
    {
      "epoch": 2.9388427734375e-05,
      "step": 30816,
      "training_step_time": 7.71602463722229
    },
    {
      "epoch": 2.9389381408691406e-05,
      "model_forward_time": 1.7131445407867432,
      "step": 30817
    },
    {
      "epoch": 2.9389381408691406e-05,
      "step": 30817,
      "training_step_time": 7.890605688095093
    },
    {
      "epoch": 2.9390335083007814e-05,
      "model_forward_time": 1.7310638427734375,
      "step": 30818
    },
    {
      "epoch": 2.9390335083007814e-05,
      "step": 30818,
      "training_step_time": 8.210799932479858
    },
    {
      "epoch": 2.9391288757324218e-05,
      "model_forward_time": 1.7353007793426514,
      "step": 30819
    },
    {
      "epoch": 2.9391288757324218e-05,
      "step": 30819,
      "training_step_time": 9.871647357940674
    },
    {
      "epoch": 2.9392242431640625e-05,
      "grad_norm": 0.18465054035186768,
      "learning_rate": 8.285546963898953e-06,
      "loss": 0.055,
      "step": 30820
    },
    {
      "epoch": 2.9392242431640625e-05,
      "model_forward_time": 2.893428325653076,
      "step": 30820
    },
    {
      "dynamics_loss_avg": 0.07448215615004301,
      "epoch": 2.9392242431640625e-05,
      "step": 30820
    },
    {
      "action_loss_avg": 0.004055006464477628,
      "epoch": 2.9392242431640625e-05,
      "step": 30820
    },
    {
      "epoch": 2.9392242431640625e-05,
      "step": 30820,
      "training_step_time": 9.837405920028687
    },
    {
      "epoch": 2.9393196105957033e-05,
      "model_forward_time": 1.7235524654388428,
      "step": 30821
    },
    {
      "epoch": 2.9393196105957033e-05,
      "step": 30821,
      "training_step_time": 7.7530295848846436
    },
    {
      "epoch": 2.9394149780273437e-05,
      "model_forward_time": 1.7291929721832275,
      "step": 30822
    },
    {
      "epoch": 2.9394149780273437e-05,
      "step": 30822,
      "training_step_time": 8.472941637039185
    },
    {
      "epoch": 2.9395103454589844e-05,
      "model_forward_time": 1.734929084777832,
      "step": 30823
    },
    {
      "epoch": 2.9395103454589844e-05,
      "step": 30823,
      "training_step_time": 7.742522239685059
    },
    {
      "epoch": 2.939605712890625e-05,
      "model_forward_time": 1.7319700717926025,
      "step": 30824
    },
    {
      "epoch": 2.939605712890625e-05,
      "step": 30824,
      "training_step_time": 7.784386873245239
    },
    {
      "epoch": 2.9397010803222656e-05,
      "model_forward_time": 1.7393131256103516,
      "step": 30825
    },
    {
      "epoch": 2.9397010803222656e-05,
      "step": 30825,
      "training_step_time": 7.692171573638916
    },
    {
      "epoch": 2.9397964477539063e-05,
      "model_forward_time": 1.731191873550415,
      "step": 30826
    },
    {
      "epoch": 2.9397964477539063e-05,
      "step": 30826,
      "training_step_time": 8.409545421600342
    },
    {
      "epoch": 2.939891815185547e-05,
      "model_forward_time": 1.727355718612671,
      "step": 30827
    },
    {
      "epoch": 2.939891815185547e-05,
      "step": 30827,
      "training_step_time": 9.481340646743774
    },
    {
      "epoch": 2.9399871826171874e-05,
      "model_forward_time": 1.72969651222229,
      "step": 30828
    },
    {
      "epoch": 2.9399871826171874e-05,
      "step": 30828,
      "training_step_time": 8.695891380310059
    },
    {
      "epoch": 2.9400825500488282e-05,
      "model_forward_time": 2.8644087314605713,
      "step": 30829
    },
    {
      "epoch": 2.9400825500488282e-05,
      "step": 30829,
      "training_step_time": 8.718009948730469
    },
    {
      "epoch": 2.9401779174804686e-05,
      "grad_norm": 0.13616538047790527,
      "learning_rate": 8.284300406612044e-06,
      "loss": 0.06,
      "step": 30830
    },
    {
      "epoch": 2.9401779174804686e-05,
      "model_forward_time": 1.7319977283477783,
      "step": 30830
    },
    {
      "dynamics_loss_avg": 0.054443256556987764,
      "epoch": 2.9401779174804686e-05,
      "step": 30830
    },
    {
      "action_loss_avg": 0.0016335995605913922,
      "epoch": 2.9401779174804686e-05,
      "step": 30830
    },
    {
      "epoch": 2.9401779174804686e-05,
      "step": 30830,
      "training_step_time": 8.342063665390015
    },
    {
      "epoch": 2.9402732849121093e-05,
      "model_forward_time": 1.7452523708343506,
      "step": 30831
    },
    {
      "epoch": 2.9402732849121093e-05,
      "step": 30831,
      "training_step_time": 7.7611048221588135
    },
    {
      "epoch": 2.94036865234375e-05,
      "model_forward_time": 1.7345426082611084,
      "step": 30832
    },
    {
      "epoch": 2.94036865234375e-05,
      "step": 30832,
      "training_step_time": 7.734161615371704
    },
    {
      "epoch": 2.9404640197753905e-05,
      "model_forward_time": 1.736818552017212,
      "step": 30833
    },
    {
      "epoch": 2.9404640197753905e-05,
      "step": 30833,
      "training_step_time": 7.775957107543945
    },
    {
      "epoch": 2.9405593872070312e-05,
      "model_forward_time": 1.7426414489746094,
      "step": 30834
    },
    {
      "epoch": 2.9405593872070312e-05,
      "step": 30834,
      "training_step_time": 8.437134027481079
    },
    {
      "epoch": 2.940654754638672e-05,
      "model_forward_time": 1.733579397201538,
      "step": 30835
    },
    {
      "epoch": 2.940654754638672e-05,
      "step": 30835,
      "training_step_time": 7.772053480148315
    },
    {
      "epoch": 2.9407501220703124e-05,
      "model_forward_time": 1.740116834640503,
      "step": 30836
    },
    {
      "epoch": 2.9407501220703124e-05,
      "step": 30836,
      "training_step_time": 8.296897172927856
    },
    {
      "epoch": 2.940845489501953e-05,
      "model_forward_time": 1.7866497039794922,
      "step": 30837
    },
    {
      "epoch": 2.940845489501953e-05,
      "step": 30837,
      "training_step_time": 8.811357021331787
    },
    {
      "epoch": 2.940940856933594e-05,
      "model_forward_time": 2.8531150817871094,
      "step": 30838
    },
    {
      "epoch": 2.940940856933594e-05,
      "step": 30838,
      "training_step_time": 8.803099870681763
    },
    {
      "epoch": 2.9410362243652342e-05,
      "model_forward_time": 1.7444617748260498,
      "step": 30839
    },
    {
      "epoch": 2.9410362243652342e-05,
      "step": 30839,
      "training_step_time": 8.810499668121338
    },
    {
      "epoch": 2.941131591796875e-05,
      "grad_norm": 0.11541380733251572,
      "learning_rate": 8.283053490158978e-06,
      "loss": 0.0599,
      "step": 30840
    },
    {
      "epoch": 2.941131591796875e-05,
      "model_forward_time": 1.739306926727295,
      "step": 30840
    },
    {
      "dynamics_loss_avg": 0.05061072036623955,
      "epoch": 2.941131591796875e-05,
      "step": 30840
    },
    {
      "action_loss_avg": 0.001809724501799792,
      "epoch": 2.941131591796875e-05,
      "step": 30840
    },
    {
      "epoch": 2.941131591796875e-05,
      "step": 30840,
      "training_step_time": 7.769377946853638
    },
    {
      "epoch": 2.9412269592285157e-05,
      "model_forward_time": 1.7306647300720215,
      "step": 30841
    },
    {
      "epoch": 2.9412269592285157e-05,
      "step": 30841,
      "training_step_time": 7.663687229156494
    },
    {
      "epoch": 2.941322326660156e-05,
      "model_forward_time": 1.738506555557251,
      "step": 30842
    },
    {
      "epoch": 2.941322326660156e-05,
      "step": 30842,
      "training_step_time": 7.817219972610474
    },
    {
      "epoch": 2.941417694091797e-05,
      "model_forward_time": 1.37391996383667,
      "step": 30843
    },
    {
      "epoch": 2.941417694091797e-05,
      "step": 30843,
      "training_step_time": 7.645288944244385
    },
    {
      "epoch": 2.9415130615234376e-05,
      "model_forward_time": 1.7456047534942627,
      "step": 30844
    },
    {
      "epoch": 2.9415130615234376e-05,
      "step": 30844,
      "training_step_time": 8.450945138931274
    },
    {
      "epoch": 2.941608428955078e-05,
      "model_forward_time": 1.746859073638916,
      "step": 30845
    },
    {
      "epoch": 2.941608428955078e-05,
      "step": 30845,
      "training_step_time": 7.606975793838501
    },
    {
      "epoch": 2.9417037963867188e-05,
      "model_forward_time": 1.761129379272461,
      "step": 30846
    },
    {
      "epoch": 2.9417037963867188e-05,
      "step": 30846,
      "training_step_time": 9.656996965408325
    },
    {
      "epoch": 2.9417991638183595e-05,
      "model_forward_time": 1.758476972579956,
      "step": 30847
    },
    {
      "epoch": 2.9417991638183595e-05,
      "step": 30847,
      "training_step_time": 9.310738801956177
    },
    {
      "epoch": 2.94189453125e-05,
      "model_forward_time": 2.9466495513916016,
      "step": 30848
    },
    {
      "epoch": 2.94189453125e-05,
      "step": 30848,
      "training_step_time": 7.82507848739624
    },
    {
      "epoch": 2.9419898986816406e-05,
      "model_forward_time": 1.7493281364440918,
      "step": 30849
    },
    {
      "epoch": 2.9419898986816406e-05,
      "step": 30849,
      "training_step_time": 7.751049280166626
    },
    {
      "epoch": 2.9420852661132814e-05,
      "grad_norm": 0.11289387196302414,
      "learning_rate": 8.28180621467612e-06,
      "loss": 0.0613,
      "step": 30850
    },
    {
      "epoch": 2.9420852661132814e-05,
      "model_forward_time": 1.7403481006622314,
      "step": 30850
    },
    {
      "dynamics_loss_avg": 0.053766193240880965,
      "epoch": 2.9420852661132814e-05,
      "step": 30850
    },
    {
      "action_loss_avg": 0.0018854906840715558,
      "epoch": 2.9420852661132814e-05,
      "step": 30850
    },
    {
      "epoch": 2.9420852661132814e-05,
      "step": 30850,
      "training_step_time": 7.726873159408569
    },
    {
      "epoch": 2.9421806335449218e-05,
      "model_forward_time": 1.7488775253295898,
      "step": 30851
    },
    {
      "epoch": 2.9421806335449218e-05,
      "step": 30851,
      "training_step_time": 7.7960968017578125
    },
    {
      "epoch": 2.9422760009765625e-05,
      "model_forward_time": 1.732923984527588,
      "step": 30852
    },
    {
      "epoch": 2.9422760009765625e-05,
      "step": 30852,
      "training_step_time": 7.5594987869262695
    },
    {
      "epoch": 2.9423713684082033e-05,
      "model_forward_time": 1.7350807189941406,
      "step": 30853
    },
    {
      "epoch": 2.9423713684082033e-05,
      "step": 30853,
      "training_step_time": 7.634820938110352
    },
    {
      "epoch": 2.9424667358398437e-05,
      "model_forward_time": 1.7431740760803223,
      "step": 30854
    },
    {
      "epoch": 2.9424667358398437e-05,
      "step": 30854,
      "training_step_time": 7.723712921142578
    },
    {
      "epoch": 2.9425621032714844e-05,
      "model_forward_time": 1.7511708736419678,
      "step": 30855
    },
    {
      "epoch": 2.9425621032714844e-05,
      "step": 30855,
      "training_step_time": 7.680279016494751
    },
    {
      "epoch": 2.942657470703125e-05,
      "model_forward_time": 1.7522168159484863,
      "step": 30856
    },
    {
      "epoch": 2.942657470703125e-05,
      "step": 30856,
      "training_step_time": 9.850541114807129
    },
    {
      "epoch": 2.9427528381347656e-05,
      "model_forward_time": 2.885991096496582,
      "step": 30857
    },
    {
      "epoch": 2.9427528381347656e-05,
      "step": 30857,
      "training_step_time": 7.7571094036102295
    },
    {
      "epoch": 2.9428482055664063e-05,
      "model_forward_time": 1.7266795635223389,
      "step": 30858
    },
    {
      "epoch": 2.9428482055664063e-05,
      "step": 30858,
      "training_step_time": 7.64202094078064
    },
    {
      "epoch": 2.942943572998047e-05,
      "model_forward_time": 1.7478747367858887,
      "step": 30859
    },
    {
      "epoch": 2.942943572998047e-05,
      "step": 30859,
      "training_step_time": 7.670080661773682
    },
    {
      "epoch": 2.9430389404296874e-05,
      "grad_norm": 0.11073381453752518,
      "learning_rate": 8.280558580299868e-06,
      "loss": 0.0589,
      "step": 30860
    },
    {
      "epoch": 2.9430389404296874e-05,
      "model_forward_time": 1.729886770248413,
      "step": 30860
    },
    {
      "dynamics_loss_avg": 0.05942028388381004,
      "epoch": 2.9430389404296874e-05,
      "step": 30860
    },
    {
      "action_loss_avg": 0.0021039209328591823,
      "epoch": 2.9430389404296874e-05,
      "step": 30860
    },
    {
      "epoch": 2.9430389404296874e-05,
      "step": 30860,
      "training_step_time": 7.803090810775757
    },
    {
      "epoch": 2.9431343078613282e-05,
      "model_forward_time": 1.7337620258331299,
      "step": 30861
    },
    {
      "epoch": 2.9431343078613282e-05,
      "step": 30861,
      "training_step_time": 7.793963432312012
    },
    {
      "epoch": 2.9432296752929686e-05,
      "model_forward_time": 1.7440729141235352,
      "step": 30862
    },
    {
      "epoch": 2.9432296752929686e-05,
      "step": 30862,
      "training_step_time": 7.68452525138855
    },
    {
      "epoch": 2.9433250427246093e-05,
      "model_forward_time": 1.7322051525115967,
      "step": 30863
    },
    {
      "epoch": 2.9433250427246093e-05,
      "step": 30863,
      "training_step_time": 8.661509037017822
    },
    {
      "epoch": 2.94342041015625e-05,
      "model_forward_time": 1.7463974952697754,
      "step": 30864
    },
    {
      "epoch": 2.94342041015625e-05,
      "step": 30864,
      "training_step_time": 7.743179798126221
    },
    {
      "epoch": 2.9435157775878905e-05,
      "model_forward_time": 1.7424304485321045,
      "step": 30865
    },
    {
      "epoch": 2.9435157775878905e-05,
      "step": 30865,
      "training_step_time": 7.934840202331543
    },
    {
      "epoch": 2.9436111450195312e-05,
      "model_forward_time": 2.8749451637268066,
      "step": 30866
    },
    {
      "epoch": 2.9436111450195312e-05,
      "step": 30866,
      "training_step_time": 8.30634331703186
    },
    {
      "epoch": 2.943706512451172e-05,
      "model_forward_time": 1.7228975296020508,
      "step": 30867
    },
    {
      "epoch": 2.943706512451172e-05,
      "step": 30867,
      "training_step_time": 7.766326904296875
    },
    {
      "epoch": 2.9438018798828124e-05,
      "model_forward_time": 1.737678050994873,
      "step": 30868
    },
    {
      "epoch": 2.9438018798828124e-05,
      "step": 30868,
      "training_step_time": 7.716793775558472
    },
    {
      "epoch": 2.943897247314453e-05,
      "model_forward_time": 1.728914499282837,
      "step": 30869
    },
    {
      "epoch": 2.943897247314453e-05,
      "step": 30869,
      "training_step_time": 8.954473495483398
    },
    {
      "epoch": 2.943992614746094e-05,
      "grad_norm": 0.11993439495563507,
      "learning_rate": 8.27931058716666e-06,
      "loss": 0.0564,
      "step": 30870
    },
    {
      "epoch": 2.943992614746094e-05,
      "model_forward_time": 1.7383041381835938,
      "step": 30870
    },
    {
      "dynamics_loss_avg": 0.0670579994097352,
      "epoch": 2.943992614746094e-05,
      "step": 30870
    },
    {
      "action_loss_avg": 0.0016680546366842463,
      "epoch": 2.943992614746094e-05,
      "step": 30870
    },
    {
      "epoch": 2.943992614746094e-05,
      "step": 30870,
      "training_step_time": 7.794141054153442
    },
    {
      "epoch": 2.9440879821777342e-05,
      "model_forward_time": 1.7206485271453857,
      "step": 30871
    },
    {
      "epoch": 2.9440879821777342e-05,
      "step": 30871,
      "training_step_time": 7.674018621444702
    },
    {
      "epoch": 2.944183349609375e-05,
      "model_forward_time": 1.7259130477905273,
      "step": 30872
    },
    {
      "epoch": 2.944183349609375e-05,
      "step": 30872,
      "training_step_time": 7.730490684509277
    },
    {
      "epoch": 2.9442787170410157e-05,
      "model_forward_time": 1.743621826171875,
      "step": 30873
    },
    {
      "epoch": 2.9442787170410157e-05,
      "step": 30873,
      "training_step_time": 8.61298131942749
    },
    {
      "epoch": 2.944374084472656e-05,
      "model_forward_time": 1.742894172668457,
      "step": 30874
    },
    {
      "epoch": 2.944374084472656e-05,
      "step": 30874,
      "training_step_time": 9.580203533172607
    },
    {
      "epoch": 2.944469451904297e-05,
      "model_forward_time": 2.8443870544433594,
      "step": 30875
    },
    {
      "epoch": 2.944469451904297e-05,
      "step": 30875,
      "training_step_time": 8.634828090667725
    },
    {
      "epoch": 2.9445648193359376e-05,
      "model_forward_time": 1.7230093479156494,
      "step": 30876
    },
    {
      "epoch": 2.9445648193359376e-05,
      "step": 30876,
      "training_step_time": 8.450509786605835
    },
    {
      "epoch": 2.944660186767578e-05,
      "model_forward_time": 1.7363951206207275,
      "step": 30877
    },
    {
      "epoch": 2.944660186767578e-05,
      "step": 30877,
      "training_step_time": 8.255800008773804
    },
    {
      "epoch": 2.9447555541992188e-05,
      "model_forward_time": 1.7282218933105469,
      "step": 30878
    },
    {
      "epoch": 2.9447555541992188e-05,
      "step": 30878,
      "training_step_time": 8.908457517623901
    },
    {
      "epoch": 2.9448509216308595e-05,
      "model_forward_time": 1.7460954189300537,
      "step": 30879
    },
    {
      "epoch": 2.9448509216308595e-05,
      "step": 30879,
      "training_step_time": 7.702277660369873
    },
    {
      "epoch": 2.9449462890625e-05,
      "grad_norm": 0.11017516255378723,
      "learning_rate": 8.27806223541298e-06,
      "loss": 0.0591,
      "step": 30880
    },
    {
      "epoch": 2.9449462890625e-05,
      "model_forward_time": 1.7297234535217285,
      "step": 30880
    },
    {
      "dynamics_loss_avg": 0.04112596809864044,
      "epoch": 2.9449462890625e-05,
      "step": 30880
    },
    {
      "action_loss_avg": 0.0011084069992648438,
      "epoch": 2.9449462890625e-05,
      "step": 30880
    },
    {
      "epoch": 2.9449462890625e-05,
      "step": 30880,
      "training_step_time": 8.162670612335205
    },
    {
      "epoch": 2.9450416564941406e-05,
      "model_forward_time": 1.7361302375793457,
      "step": 30881
    },
    {
      "epoch": 2.9450416564941406e-05,
      "step": 30881,
      "training_step_time": 7.934062480926514
    },
    {
      "epoch": 2.9451370239257814e-05,
      "model_forward_time": 1.7241313457489014,
      "step": 30882
    },
    {
      "epoch": 2.9451370239257814e-05,
      "step": 30882,
      "training_step_time": 7.661489248275757
    },
    {
      "epoch": 2.9452323913574218e-05,
      "model_forward_time": 1.7657575607299805,
      "step": 30883
    },
    {
      "epoch": 2.9452323913574218e-05,
      "step": 30883,
      "training_step_time": 8.313615083694458
    },
    {
      "epoch": 2.9453277587890625e-05,
      "model_forward_time": 2.8637900352478027,
      "step": 30884
    },
    {
      "epoch": 2.9453277587890625e-05,
      "step": 30884,
      "training_step_time": 7.746342658996582
    },
    {
      "epoch": 2.9454231262207033e-05,
      "model_forward_time": 1.7315022945404053,
      "step": 30885
    },
    {
      "epoch": 2.9454231262207033e-05,
      "step": 30885,
      "training_step_time": 8.735503435134888
    },
    {
      "epoch": 2.9455184936523437e-05,
      "model_forward_time": 1.7383253574371338,
      "step": 30886
    },
    {
      "epoch": 2.9455184936523437e-05,
      "step": 30886,
      "training_step_time": 8.642893552780151
    },
    {
      "epoch": 2.9456138610839844e-05,
      "model_forward_time": 1.7064151763916016,
      "step": 30887
    },
    {
      "epoch": 2.9456138610839844e-05,
      "step": 30887,
      "training_step_time": 7.744444370269775
    },
    {
      "epoch": 2.945709228515625e-05,
      "model_forward_time": 1.7152478694915771,
      "step": 30888
    },
    {
      "epoch": 2.945709228515625e-05,
      "step": 30888,
      "training_step_time": 7.880646705627441
    },
    {
      "epoch": 2.9458045959472656e-05,
      "model_forward_time": 1.3728547096252441,
      "step": 30889
    },
    {
      "epoch": 2.9458045959472656e-05,
      "step": 30889,
      "training_step_time": 7.754975318908691
    },
    {
      "epoch": 2.9458999633789063e-05,
      "grad_norm": 0.11921379715204239,
      "learning_rate": 8.27681352517534e-06,
      "loss": 0.0609,
      "step": 30890
    },
    {
      "epoch": 2.9458999633789063e-05,
      "model_forward_time": 1.7228951454162598,
      "step": 30890
    },
    {
      "dynamics_loss_avg": 0.054179757740348576,
      "epoch": 2.9458999633789063e-05,
      "step": 30890
    },
    {
      "action_loss_avg": 0.0020165313384495676,
      "epoch": 2.9458999633789063e-05,
      "step": 30890
    },
    {
      "epoch": 2.9458999633789063e-05,
      "step": 30890,
      "training_step_time": 7.490106821060181
    },
    {
      "epoch": 2.9459953308105467e-05,
      "model_forward_time": 1.7262487411499023,
      "step": 30891
    },
    {
      "epoch": 2.9459953308105467e-05,
      "step": 30891,
      "training_step_time": 7.730742692947388
    },
    {
      "epoch": 2.9460906982421874e-05,
      "model_forward_time": 1.754239559173584,
      "step": 30892
    },
    {
      "epoch": 2.9460906982421874e-05,
      "step": 30892,
      "training_step_time": 7.778455018997192
    },
    {
      "epoch": 2.9461860656738282e-05,
      "model_forward_time": 1.7502760887145996,
      "step": 30893
    },
    {
      "epoch": 2.9461860656738282e-05,
      "step": 30893,
      "training_step_time": 7.815238952636719
    },
    {
      "epoch": 2.9462814331054686e-05,
      "model_forward_time": 2.9209444522857666,
      "step": 30894
    },
    {
      "epoch": 2.9462814331054686e-05,
      "step": 30894,
      "training_step_time": 8.376850366592407
    },
    {
      "epoch": 2.9463768005371093e-05,
      "model_forward_time": 1.7412230968475342,
      "step": 30895
    },
    {
      "epoch": 2.9463768005371093e-05,
      "step": 30895,
      "training_step_time": 7.671118497848511
    },
    {
      "epoch": 2.94647216796875e-05,
      "model_forward_time": 1.7383363246917725,
      "step": 30896
    },
    {
      "epoch": 2.94647216796875e-05,
      "step": 30896,
      "training_step_time": 7.687377452850342
    },
    {
      "epoch": 2.9465675354003905e-05,
      "model_forward_time": 1.7365272045135498,
      "step": 30897
    },
    {
      "epoch": 2.9465675354003905e-05,
      "step": 30897,
      "training_step_time": 7.815284729003906
    },
    {
      "epoch": 2.9466629028320312e-05,
      "model_forward_time": 1.7509891986846924,
      "step": 30898
    },
    {
      "epoch": 2.9466629028320312e-05,
      "step": 30898,
      "training_step_time": 7.742149353027344
    },
    {
      "epoch": 2.946758270263672e-05,
      "model_forward_time": 1.7354907989501953,
      "step": 30899
    },
    {
      "epoch": 2.946758270263672e-05,
      "step": 30899,
      "training_step_time": 7.5459654331207275
    },
    {
      "epoch": 2.9468536376953124e-05,
      "grad_norm": 0.15674540400505066,
      "learning_rate": 8.2755644565903e-06,
      "loss": 0.0587,
      "step": 30900
    },
    {
      "epoch": 2.9468536376953124e-05,
      "model_forward_time": 1.7313072681427002,
      "step": 30900
    },
    {
      "dynamics_loss_avg": 0.04245207672938704,
      "epoch": 2.9468536376953124e-05,
      "step": 30900
    },
    {
      "action_loss_avg": 0.0018905966775491835,
      "epoch": 2.9468536376953124e-05,
      "step": 30900
    },
    {
      "epoch": 2.9468536376953124e-05,
      "step": 30900,
      "training_step_time": 7.698470592498779
    },
    {
      "epoch": 2.946949005126953e-05,
      "model_forward_time": 1.739530086517334,
      "step": 30901
    },
    {
      "epoch": 2.946949005126953e-05,
      "step": 30901,
      "training_step_time": 8.503453493118286
    },
    {
      "epoch": 2.947044372558594e-05,
      "model_forward_time": 1.7418837547302246,
      "step": 30902
    },
    {
      "epoch": 2.947044372558594e-05,
      "step": 30902,
      "training_step_time": 9.224872827529907
    },
    {
      "epoch": 2.9471397399902343e-05,
      "model_forward_time": 2.927511692047119,
      "step": 30903
    },
    {
      "epoch": 2.9471397399902343e-05,
      "step": 30903,
      "training_step_time": 7.769360303878784
    },
    {
      "epoch": 2.947235107421875e-05,
      "model_forward_time": 1.7316184043884277,
      "step": 30904
    },
    {
      "epoch": 2.947235107421875e-05,
      "step": 30904,
      "training_step_time": 7.651144742965698
    },
    {
      "epoch": 2.9473304748535157e-05,
      "model_forward_time": 1.7393019199371338,
      "step": 30905
    },
    {
      "epoch": 2.9473304748535157e-05,
      "step": 30905,
      "training_step_time": 7.753431081771851
    },
    {
      "epoch": 2.947425842285156e-05,
      "model_forward_time": 1.7341580390930176,
      "step": 30906
    },
    {
      "epoch": 2.947425842285156e-05,
      "step": 30906,
      "training_step_time": 7.743034839630127
    },
    {
      "epoch": 2.947521209716797e-05,
      "model_forward_time": 1.7280805110931396,
      "step": 30907
    },
    {
      "epoch": 2.947521209716797e-05,
      "step": 30907,
      "training_step_time": 8.48629641532898
    },
    {
      "epoch": 2.9476165771484376e-05,
      "model_forward_time": 1.7372524738311768,
      "step": 30908
    },
    {
      "epoch": 2.9476165771484376e-05,
      "step": 30908,
      "training_step_time": 8.759922742843628
    },
    {
      "epoch": 2.947711944580078e-05,
      "model_forward_time": 1.7279551029205322,
      "step": 30909
    },
    {
      "epoch": 2.947711944580078e-05,
      "step": 30909,
      "training_step_time": 8.580610990524292
    },
    {
      "epoch": 2.9478073120117188e-05,
      "grad_norm": 0.16324657201766968,
      "learning_rate": 8.274315029794456e-06,
      "loss": 0.0612,
      "step": 30910
    },
    {
      "epoch": 2.9478073120117188e-05,
      "model_forward_time": 1.7316405773162842,
      "step": 30910
    },
    {
      "dynamics_loss_avg": 0.06045612059533596,
      "epoch": 2.9478073120117188e-05,
      "step": 30910
    },
    {
      "action_loss_avg": 0.0021426937950309367,
      "epoch": 2.9478073120117188e-05,
      "step": 30910
    },
    {
      "epoch": 2.9478073120117188e-05,
      "step": 30910,
      "training_step_time": 8.798263549804688
    },
    {
      "epoch": 2.9479026794433595e-05,
      "model_forward_time": 1.7358062267303467,
      "step": 30911
    },
    {
      "epoch": 2.9479026794433595e-05,
      "step": 30911,
      "training_step_time": 7.725204944610596
    },
    {
      "epoch": 2.947998046875e-05,
      "model_forward_time": 2.900756597518921,
      "step": 30912
    },
    {
      "epoch": 2.947998046875e-05,
      "step": 30912,
      "training_step_time": 9.547423839569092
    },
    {
      "epoch": 2.9480934143066406e-05,
      "model_forward_time": 1.7254853248596191,
      "step": 30913
    },
    {
      "epoch": 2.9480934143066406e-05,
      "step": 30913,
      "training_step_time": 9.483743667602539
    },
    {
      "epoch": 2.9481887817382814e-05,
      "model_forward_time": 1.737713098526001,
      "step": 30914
    },
    {
      "epoch": 2.9481887817382814e-05,
      "step": 30914,
      "training_step_time": 7.7110185623168945
    },
    {
      "epoch": 2.9482841491699218e-05,
      "model_forward_time": 1.3766183853149414,
      "step": 30915
    },
    {
      "epoch": 2.9482841491699218e-05,
      "step": 30915,
      "training_step_time": 8.752034902572632
    },
    {
      "epoch": 2.9483795166015625e-05,
      "model_forward_time": 1.7444207668304443,
      "step": 30916
    },
    {
      "epoch": 2.9483795166015625e-05,
      "step": 30916,
      "training_step_time": 7.744889497756958
    },
    {
      "epoch": 2.9484748840332033e-05,
      "model_forward_time": 1.7552664279937744,
      "step": 30917
    },
    {
      "epoch": 2.9484748840332033e-05,
      "step": 30917,
      "training_step_time": 7.587970495223999
    },
    {
      "epoch": 2.9485702514648437e-05,
      "model_forward_time": 1.7464728355407715,
      "step": 30918
    },
    {
      "epoch": 2.9485702514648437e-05,
      "step": 30918,
      "training_step_time": 9.103402614593506
    },
    {
      "epoch": 2.9486656188964844e-05,
      "model_forward_time": 1.7241783142089844,
      "step": 30919
    },
    {
      "epoch": 2.9486656188964844e-05,
      "step": 30919,
      "training_step_time": 8.547120094299316
    },
    {
      "epoch": 2.948760986328125e-05,
      "grad_norm": 0.14495648443698883,
      "learning_rate": 8.273065244924443e-06,
      "loss": 0.0597,
      "step": 30920
    },
    {
      "epoch": 2.948760986328125e-05,
      "model_forward_time": 1.8517661094665527,
      "step": 30920
    },
    {
      "dynamics_loss_avg": 0.03969100210815668,
      "epoch": 2.948760986328125e-05,
      "step": 30920
    },
    {
      "action_loss_avg": 0.0021419027441879736,
      "epoch": 2.948760986328125e-05,
      "step": 30920
    },
    {
      "epoch": 2.948760986328125e-05,
      "step": 30920,
      "training_step_time": 7.766034126281738
    },
    {
      "epoch": 2.9488563537597656e-05,
      "model_forward_time": 2.8385415077209473,
      "step": 30921
    },
    {
      "epoch": 2.9488563537597656e-05,
      "step": 30921,
      "training_step_time": 9.38844609260559
    },
    {
      "epoch": 2.9489517211914063e-05,
      "model_forward_time": 1.7481153011322021,
      "step": 30922
    },
    {
      "epoch": 2.9489517211914063e-05,
      "step": 30922,
      "training_step_time": 7.7227373123168945
    },
    {
      "epoch": 2.9490470886230467e-05,
      "model_forward_time": 1.7393174171447754,
      "step": 30923
    },
    {
      "epoch": 2.9490470886230467e-05,
      "step": 30923,
      "training_step_time": 7.75302267074585
    },
    {
      "epoch": 2.9491424560546875e-05,
      "model_forward_time": 1.7382793426513672,
      "step": 30924
    },
    {
      "epoch": 2.9491424560546875e-05,
      "step": 30924,
      "training_step_time": 7.810132741928101
    },
    {
      "epoch": 2.9492378234863282e-05,
      "model_forward_time": 1.7440195083618164,
      "step": 30925
    },
    {
      "epoch": 2.9492378234863282e-05,
      "step": 30925,
      "training_step_time": 7.806918621063232
    },
    {
      "epoch": 2.9493331909179686e-05,
      "model_forward_time": 1.738706111907959,
      "step": 30926
    },
    {
      "epoch": 2.9493331909179686e-05,
      "step": 30926,
      "training_step_time": 7.63386607170105
    },
    {
      "epoch": 2.9494285583496093e-05,
      "model_forward_time": 1.738454818725586,
      "step": 30927
    },
    {
      "epoch": 2.9494285583496093e-05,
      "step": 30927,
      "training_step_time": 7.529192924499512
    },
    {
      "epoch": 2.94952392578125e-05,
      "model_forward_time": 1.7421200275421143,
      "step": 30928
    },
    {
      "epoch": 2.94952392578125e-05,
      "step": 30928,
      "training_step_time": 7.78085732460022
    },
    {
      "epoch": 2.9496192932128905e-05,
      "model_forward_time": 1.8257842063903809,
      "step": 30929
    },
    {
      "epoch": 2.9496192932128905e-05,
      "step": 30929,
      "training_step_time": 9.242532014846802
    },
    {
      "epoch": 2.9497146606445312e-05,
      "grad_norm": 0.14375509321689606,
      "learning_rate": 8.271815102116936e-06,
      "loss": 0.0645,
      "step": 30930
    },
    {
      "epoch": 2.9497146606445312e-05,
      "model_forward_time": 1.7517027854919434,
      "step": 30930
    },
    {
      "dynamics_loss_avg": 0.055357599072158335,
      "epoch": 2.9497146606445312e-05,
      "step": 30930
    },
    {
      "action_loss_avg": 0.001341760324430652,
      "epoch": 2.9497146606445312e-05,
      "step": 30930
    },
    {
      "epoch": 2.9497146606445312e-05,
      "step": 30930,
      "training_step_time": 8.286019325256348
    },
    {
      "epoch": 2.949810028076172e-05,
      "model_forward_time": 1.7475593090057373,
      "step": 30931
    },
    {
      "epoch": 2.949810028076172e-05,
      "step": 30931,
      "training_step_time": 7.719954967498779
    },
    {
      "epoch": 2.9499053955078124e-05,
      "model_forward_time": 1.7385814189910889,
      "step": 30932
    },
    {
      "epoch": 2.9499053955078124e-05,
      "step": 30932,
      "training_step_time": 7.6632983684539795
    },
    {
      "epoch": 2.950000762939453e-05,
      "model_forward_time": 1.736464262008667,
      "step": 30933
    },
    {
      "epoch": 2.950000762939453e-05,
      "step": 30933,
      "training_step_time": 7.692748308181763
    },
    {
      "epoch": 2.950096130371094e-05,
      "model_forward_time": 1.747075080871582,
      "step": 30934
    },
    {
      "epoch": 2.950096130371094e-05,
      "step": 30934,
      "training_step_time": 8.6405611038208
    },
    {
      "epoch": 2.9501914978027343e-05,
      "model_forward_time": 1.7486355304718018,
      "step": 30935
    },
    {
      "epoch": 2.9501914978027343e-05,
      "step": 30935,
      "training_step_time": 7.64111590385437
    },
    {
      "epoch": 2.950286865234375e-05,
      "model_forward_time": 1.7409517765045166,
      "step": 30936
    },
    {
      "epoch": 2.950286865234375e-05,
      "step": 30936,
      "training_step_time": 7.757373332977295
    },
    {
      "epoch": 2.9503822326660157e-05,
      "model_forward_time": 1.7333683967590332,
      "step": 30937
    },
    {
      "epoch": 2.9503822326660157e-05,
      "step": 30937,
      "training_step_time": 7.827314615249634
    },
    {
      "epoch": 2.950477600097656e-05,
      "model_forward_time": 1.816321611404419,
      "step": 30938
    },
    {
      "epoch": 2.950477600097656e-05,
      "step": 30938,
      "training_step_time": 8.764527082443237
    },
    {
      "epoch": 2.950572967529297e-05,
      "model_forward_time": 1.7336835861206055,
      "step": 30939
    },
    {
      "epoch": 2.950572967529297e-05,
      "step": 30939,
      "training_step_time": 9.464876890182495
    },
    {
      "epoch": 2.9506683349609376e-05,
      "grad_norm": 0.11826799064874649,
      "learning_rate": 8.270564601508648e-06,
      "loss": 0.0621,
      "step": 30940
    },
    {
      "epoch": 2.9506683349609376e-05,
      "model_forward_time": 2.952223300933838,
      "step": 30940
    },
    {
      "dynamics_loss_avg": 0.06253547333180905,
      "epoch": 2.9506683349609376e-05,
      "step": 30940
    },
    {
      "action_loss_avg": 0.0026379439426818863,
      "epoch": 2.9506683349609376e-05,
      "step": 30940
    },
    {
      "epoch": 2.9506683349609376e-05,
      "step": 30940,
      "training_step_time": 8.33179759979248
    },
    {
      "epoch": 2.950763702392578e-05,
      "model_forward_time": 1.739626407623291,
      "step": 30941
    },
    {
      "epoch": 2.950763702392578e-05,
      "step": 30941,
      "training_step_time": 7.730441331863403
    },
    {
      "epoch": 2.9508590698242188e-05,
      "model_forward_time": 1.7406187057495117,
      "step": 30942
    },
    {
      "epoch": 2.9508590698242188e-05,
      "step": 30942,
      "training_step_time": 7.833824872970581
    },
    {
      "epoch": 2.9509544372558595e-05,
      "model_forward_time": 1.7457342147827148,
      "step": 30943
    },
    {
      "epoch": 2.9509544372558595e-05,
      "step": 30943,
      "training_step_time": 7.839202165603638
    },
    {
      "epoch": 2.9510498046875e-05,
      "model_forward_time": 1.7370941638946533,
      "step": 30944
    },
    {
      "epoch": 2.9510498046875e-05,
      "step": 30944,
      "training_step_time": 7.738251209259033
    },
    {
      "epoch": 2.9511451721191407e-05,
      "model_forward_time": 1.749293327331543,
      "step": 30945
    },
    {
      "epoch": 2.9511451721191407e-05,
      "step": 30945,
      "training_step_time": 7.72890305519104
    },
    {
      "epoch": 2.9512405395507814e-05,
      "model_forward_time": 1.745337724685669,
      "step": 30946
    },
    {
      "epoch": 2.9512405395507814e-05,
      "step": 30946,
      "training_step_time": 7.826049327850342
    },
    {
      "epoch": 2.9513359069824218e-05,
      "model_forward_time": 1.7751164436340332,
      "step": 30947
    },
    {
      "epoch": 2.9513359069824218e-05,
      "step": 30947,
      "training_step_time": 8.577715396881104
    },
    {
      "epoch": 2.9514312744140625e-05,
      "model_forward_time": 1.7458903789520264,
      "step": 30948
    },
    {
      "epoch": 2.9514312744140625e-05,
      "step": 30948,
      "training_step_time": 8.29627776145935
    },
    {
      "epoch": 2.9515266418457033e-05,
      "model_forward_time": 2.905268430709839,
      "step": 30949
    },
    {
      "epoch": 2.9515266418457033e-05,
      "step": 30949,
      "training_step_time": 9.524362564086914
    },
    {
      "epoch": 2.9516220092773437e-05,
      "grad_norm": 0.18959400057792664,
      "learning_rate": 8.269313743236333e-06,
      "loss": 0.0579,
      "step": 30950
    },
    {
      "epoch": 2.9516220092773437e-05,
      "model_forward_time": 1.7647731304168701,
      "step": 30950
    },
    {
      "dynamics_loss_avg": 0.05904550664126873,
      "epoch": 2.9516220092773437e-05,
      "step": 30950
    },
    {
      "action_loss_avg": 0.0019307176436996088,
      "epoch": 2.9516220092773437e-05,
      "step": 30950
    },
    {
      "epoch": 2.9516220092773437e-05,
      "step": 30950,
      "training_step_time": 7.762569189071655
    },
    {
      "epoch": 2.9517173767089844e-05,
      "model_forward_time": 1.7432620525360107,
      "step": 30951
    },
    {
      "epoch": 2.9517173767089844e-05,
      "step": 30951,
      "training_step_time": 7.72636079788208
    },
    {
      "epoch": 2.951812744140625e-05,
      "model_forward_time": 1.7375304698944092,
      "step": 30952
    },
    {
      "epoch": 2.951812744140625e-05,
      "step": 30952,
      "training_step_time": 8.278936386108398
    },
    {
      "epoch": 2.9519081115722656e-05,
      "model_forward_time": 1.752328872680664,
      "step": 30953
    },
    {
      "epoch": 2.9519081115722656e-05,
      "step": 30953,
      "training_step_time": 7.781793594360352
    },
    {
      "epoch": 2.9520034790039063e-05,
      "model_forward_time": 1.7305338382720947,
      "step": 30954
    },
    {
      "epoch": 2.9520034790039063e-05,
      "step": 30954,
      "training_step_time": 7.7089128494262695
    },
    {
      "epoch": 2.9520988464355467e-05,
      "model_forward_time": 1.7312679290771484,
      "step": 30955
    },
    {
      "epoch": 2.9520988464355467e-05,
      "step": 30955,
      "training_step_time": 7.848236322402954
    },
    {
      "epoch": 2.9521942138671875e-05,
      "model_forward_time": 1.7597975730895996,
      "step": 30956
    },
    {
      "epoch": 2.9521942138671875e-05,
      "step": 30956,
      "training_step_time": 8.643215894699097
    },
    {
      "epoch": 2.9522895812988282e-05,
      "model_forward_time": 1.740804672241211,
      "step": 30957
    },
    {
      "epoch": 2.9522895812988282e-05,
      "step": 30957,
      "training_step_time": 8.861818790435791
    },
    {
      "epoch": 2.9523849487304686e-05,
      "model_forward_time": 2.8345541954040527,
      "step": 30958
    },
    {
      "epoch": 2.9523849487304686e-05,
      "step": 30958,
      "training_step_time": 7.685708045959473
    },
    {
      "epoch": 2.9524803161621093e-05,
      "model_forward_time": 1.7435204982757568,
      "step": 30959
    },
    {
      "epoch": 2.9524803161621093e-05,
      "step": 30959,
      "training_step_time": 7.750709533691406
    },
    {
      "epoch": 2.95257568359375e-05,
      "grad_norm": 0.1703542172908783,
      "learning_rate": 8.268062527436783e-06,
      "loss": 0.0584,
      "step": 30960
    },
    {
      "epoch": 2.95257568359375e-05,
      "model_forward_time": 1.7578823566436768,
      "step": 30960
    },
    {
      "dynamics_loss_avg": 0.05601588226854801,
      "epoch": 2.95257568359375e-05,
      "step": 30960
    },
    {
      "action_loss_avg": 0.0019070308306254446,
      "epoch": 2.95257568359375e-05,
      "step": 30960
    },
    {
      "epoch": 2.95257568359375e-05,
      "step": 30960,
      "training_step_time": 9.059056520462036
    },
    {
      "epoch": 2.9526710510253905e-05,
      "model_forward_time": 1.7379133701324463,
      "step": 30961
    },
    {
      "epoch": 2.9526710510253905e-05,
      "step": 30961,
      "training_step_time": 7.691036701202393
    },
    {
      "epoch": 2.9527664184570312e-05,
      "model_forward_time": 1.7385122776031494,
      "step": 30962
    },
    {
      "epoch": 2.9527664184570312e-05,
      "step": 30962,
      "training_step_time": 7.843541860580444
    },
    {
      "epoch": 2.952861785888672e-05,
      "model_forward_time": 1.7418432235717773,
      "step": 30963
    },
    {
      "epoch": 2.952861785888672e-05,
      "step": 30963,
      "training_step_time": 8.720176696777344
    },
    {
      "epoch": 2.9529571533203124e-05,
      "model_forward_time": 1.7384614944458008,
      "step": 30964
    },
    {
      "epoch": 2.9529571533203124e-05,
      "step": 30964,
      "training_step_time": 8.335533380508423
    },
    {
      "epoch": 2.953052520751953e-05,
      "model_forward_time": 1.3779256343841553,
      "step": 30965
    },
    {
      "epoch": 2.953052520751953e-05,
      "step": 30965,
      "training_step_time": 7.860357761383057
    },
    {
      "epoch": 2.953147888183594e-05,
      "model_forward_time": 1.8681559562683105,
      "step": 30966
    },
    {
      "epoch": 2.953147888183594e-05,
      "step": 30966,
      "training_step_time": 8.967903852462769
    },
    {
      "epoch": 2.9532432556152343e-05,
      "model_forward_time": 2.892082691192627,
      "step": 30967
    },
    {
      "epoch": 2.9532432556152343e-05,
      "step": 30967,
      "training_step_time": 8.367482900619507
    },
    {
      "epoch": 2.953338623046875e-05,
      "model_forward_time": 1.7402093410491943,
      "step": 30968
    },
    {
      "epoch": 2.953338623046875e-05,
      "step": 30968,
      "training_step_time": 8.538082122802734
    },
    {
      "epoch": 2.9534339904785157e-05,
      "model_forward_time": 1.7374377250671387,
      "step": 30969
    },
    {
      "epoch": 2.9534339904785157e-05,
      "step": 30969,
      "training_step_time": 8.676735877990723
    },
    {
      "epoch": 2.953529357910156e-05,
      "grad_norm": 0.15108747780323029,
      "learning_rate": 8.266810954246828e-06,
      "loss": 0.0574,
      "step": 30970
    },
    {
      "epoch": 2.953529357910156e-05,
      "model_forward_time": 1.7524747848510742,
      "step": 30970
    },
    {
      "dynamics_loss_avg": 0.06048636212944984,
      "epoch": 2.953529357910156e-05,
      "step": 30970
    },
    {
      "action_loss_avg": 0.0021276462357491255,
      "epoch": 2.953529357910156e-05,
      "step": 30970
    },
    {
      "epoch": 2.953529357910156e-05,
      "step": 30970,
      "training_step_time": 7.769040822982788
    },
    {
      "epoch": 2.953624725341797e-05,
      "model_forward_time": 1.7413995265960693,
      "step": 30971
    },
    {
      "epoch": 2.953624725341797e-05,
      "step": 30971,
      "training_step_time": 7.888286590576172
    },
    {
      "epoch": 2.9537200927734376e-05,
      "model_forward_time": 1.743276596069336,
      "step": 30972
    },
    {
      "epoch": 2.9537200927734376e-05,
      "step": 30972,
      "training_step_time": 7.490556955337524
    },
    {
      "epoch": 2.953815460205078e-05,
      "model_forward_time": 1.7522928714752197,
      "step": 30973
    },
    {
      "epoch": 2.953815460205078e-05,
      "step": 30973,
      "training_step_time": 9.506406545639038
    },
    {
      "epoch": 2.9539108276367188e-05,
      "model_forward_time": 1.7409179210662842,
      "step": 30974
    },
    {
      "epoch": 2.9539108276367188e-05,
      "step": 30974,
      "training_step_time": 8.228495359420776
    },
    {
      "epoch": 2.9540061950683595e-05,
      "model_forward_time": 1.848923683166504,
      "step": 30975
    },
    {
      "epoch": 2.9540061950683595e-05,
      "step": 30975,
      "training_step_time": 9.604368209838867
    },
    {
      "epoch": 2.9541015625e-05,
      "model_forward_time": 1.766693115234375,
      "step": 30976
    },
    {
      "epoch": 2.9541015625e-05,
      "step": 30976,
      "training_step_time": 7.699396371841431
    },
    {
      "epoch": 2.9541969299316407e-05,
      "model_forward_time": 2.8788504600524902,
      "step": 30977
    },
    {
      "epoch": 2.9541969299316407e-05,
      "step": 30977,
      "training_step_time": 8.631410837173462
    },
    {
      "epoch": 2.9542922973632814e-05,
      "model_forward_time": 1.736849069595337,
      "step": 30978
    },
    {
      "epoch": 2.9542922973632814e-05,
      "step": 30978,
      "training_step_time": 8.71932053565979
    },
    {
      "epoch": 2.9543876647949218e-05,
      "model_forward_time": 1.736030101776123,
      "step": 30979
    },
    {
      "epoch": 2.9543876647949218e-05,
      "step": 30979,
      "training_step_time": 7.740898370742798
    },
    {
      "epoch": 2.9544830322265625e-05,
      "grad_norm": 0.11500006914138794,
      "learning_rate": 8.265559023803337e-06,
      "loss": 0.0589,
      "step": 30980
    },
    {
      "epoch": 2.9544830322265625e-05,
      "model_forward_time": 1.741837978363037,
      "step": 30980
    },
    {
      "dynamics_loss_avg": 0.06094556376338005,
      "epoch": 2.9544830322265625e-05,
      "step": 30980
    },
    {
      "action_loss_avg": 0.0025212862994521855,
      "epoch": 2.9544830322265625e-05,
      "step": 30980
    },
    {
      "epoch": 2.9544830322265625e-05,
      "step": 30980,
      "training_step_time": 7.7958598136901855
    },
    {
      "epoch": 2.9545783996582033e-05,
      "model_forward_time": 1.7398898601531982,
      "step": 30981
    },
    {
      "epoch": 2.9545783996582033e-05,
      "step": 30981,
      "training_step_time": 7.856038570404053
    },
    {
      "epoch": 2.9546737670898437e-05,
      "model_forward_time": 1.7386503219604492,
      "step": 30982
    },
    {
      "epoch": 2.9546737670898437e-05,
      "step": 30982,
      "training_step_time": 7.729666471481323
    },
    {
      "epoch": 2.9547691345214844e-05,
      "model_forward_time": 1.7312486171722412,
      "step": 30983
    },
    {
      "epoch": 2.9547691345214844e-05,
      "step": 30983,
      "training_step_time": 7.769273281097412
    },
    {
      "epoch": 2.954864501953125e-05,
      "model_forward_time": 1.7940194606781006,
      "step": 30984
    },
    {
      "epoch": 2.954864501953125e-05,
      "step": 30984,
      "training_step_time": 7.889995098114014
    },
    {
      "epoch": 2.9549598693847656e-05,
      "model_forward_time": 1.7528414726257324,
      "step": 30985
    },
    {
      "epoch": 2.9549598693847656e-05,
      "step": 30985,
      "training_step_time": 7.794941186904907
    },
    {
      "epoch": 2.9550552368164063e-05,
      "model_forward_time": 2.9423775672912598,
      "step": 30986
    },
    {
      "epoch": 2.9550552368164063e-05,
      "step": 30986,
      "training_step_time": 8.863730430603027
    },
    {
      "epoch": 2.9551506042480467e-05,
      "model_forward_time": 1.7300922870635986,
      "step": 30987
    },
    {
      "epoch": 2.9551506042480467e-05,
      "step": 30987,
      "training_step_time": 7.7430949211120605
    },
    {
      "epoch": 2.9552459716796875e-05,
      "model_forward_time": 1.7421340942382812,
      "step": 30988
    },
    {
      "epoch": 2.9552459716796875e-05,
      "step": 30988,
      "training_step_time": 7.707074165344238
    },
    {
      "epoch": 2.9553413391113282e-05,
      "model_forward_time": 1.7385177612304688,
      "step": 30989
    },
    {
      "epoch": 2.9553413391113282e-05,
      "step": 30989,
      "training_step_time": 7.74618673324585
    },
    {
      "epoch": 2.9554367065429686e-05,
      "grad_norm": 0.12262675166130066,
      "learning_rate": 8.264306736243223e-06,
      "loss": 0.0635,
      "step": 30990
    },
    {
      "epoch": 2.9554367065429686e-05,
      "model_forward_time": 1.7500505447387695,
      "step": 30990
    },
    {
      "dynamics_loss_avg": 0.05772756580263376,
      "epoch": 2.9554367065429686e-05,
      "step": 30990
    },
    {
      "action_loss_avg": 0.0023929397633764894,
      "epoch": 2.9554367065429686e-05,
      "step": 30990
    },
    {
      "epoch": 2.9554367065429686e-05,
      "step": 30990,
      "training_step_time": 7.827037811279297
    },
    {
      "epoch": 2.9555320739746093e-05,
      "model_forward_time": 1.736738920211792,
      "step": 30991
    },
    {
      "epoch": 2.9555320739746093e-05,
      "step": 30991,
      "training_step_time": 8.433404684066772
    },
    {
      "epoch": 2.95562744140625e-05,
      "model_forward_time": 1.7456719875335693,
      "step": 30992
    },
    {
      "epoch": 2.95562744140625e-05,
      "step": 30992,
      "training_step_time": 7.723767518997192
    },
    {
      "epoch": 2.9557228088378905e-05,
      "model_forward_time": 1.7716772556304932,
      "step": 30993
    },
    {
      "epoch": 2.9557228088378905e-05,
      "step": 30993,
      "training_step_time": 7.8672261238098145
    },
    {
      "epoch": 2.9558181762695312e-05,
      "model_forward_time": 1.7332980632781982,
      "step": 30994
    },
    {
      "epoch": 2.9558181762695312e-05,
      "step": 30994,
      "training_step_time": 7.736762762069702
    },
    {
      "epoch": 2.955913543701172e-05,
      "model_forward_time": 2.905876874923706,
      "step": 30995
    },
    {
      "epoch": 2.955913543701172e-05,
      "step": 30995,
      "training_step_time": 8.507313013076782
    },
    {
      "epoch": 2.9560089111328124e-05,
      "model_forward_time": 1.7372112274169922,
      "step": 30996
    },
    {
      "epoch": 2.9560089111328124e-05,
      "step": 30996,
      "training_step_time": 10.332706451416016
    },
    {
      "epoch": 2.956104278564453e-05,
      "model_forward_time": 1.7454571723937988,
      "step": 30997
    },
    {
      "epoch": 2.956104278564453e-05,
      "step": 30997,
      "training_step_time": 9.03672170639038
    },
    {
      "epoch": 2.956199645996094e-05,
      "model_forward_time": 1.7609267234802246,
      "step": 30998
    },
    {
      "epoch": 2.956199645996094e-05,
      "step": 30998,
      "training_step_time": 8.106037855148315
    },
    {
      "epoch": 2.9562950134277343e-05,
      "model_forward_time": 1.7472288608551025,
      "step": 30999
    },
    {
      "epoch": 2.9562950134277343e-05,
      "step": 30999,
      "training_step_time": 8.637513637542725
    },
    {
      "epoch": 2.956390380859375e-05,
      "grad_norm": 0.12994341552257538,
      "learning_rate": 8.263054091703432e-06,
      "loss": 0.0577,
      "step": 31000
    },
    {
      "epoch": 2.956390380859375e-05,
      "model_forward_time": 1.7956433296203613,
      "step": 31000
    },
    {
      "dynamics_loss_avg": 0.047125177644193175,
      "epoch": 2.956390380859375e-05,
      "step": 31000
    },
    {
      "action_loss_avg": 0.0019584326655603945,
      "epoch": 2.956390380859375e-05,
      "step": 31000
    },
    {
      "epoch": 2.956390380859375e-05,
      "step": 31000,
      "training_step_time": 9.214309930801392
    },
    {
      "epoch": 2.9564857482910157e-05,
      "model_forward_time": 1.7621829509735107,
      "step": 31001
    },
    {
      "epoch": 2.9564857482910157e-05,
      "step": 31001,
      "training_step_time": 8.325214147567749
    },
    {
      "epoch": 2.956581115722656e-05,
      "model_forward_time": 1.779750108718872,
      "step": 31002
    },
    {
      "epoch": 2.956581115722656e-05,
      "step": 31002,
      "training_step_time": 7.831873893737793
    },
    {
      "epoch": 2.956676483154297e-05,
      "model_forward_time": 1.7509617805480957,
      "step": 31003
    },
    {
      "epoch": 2.956676483154297e-05,
      "step": 31003,
      "training_step_time": 11.511801719665527
    },
    {
      "epoch": 2.9567718505859376e-05,
      "model_forward_time": 1.742966890335083,
      "step": 31004
    },
    {
      "epoch": 2.9567718505859376e-05,
      "step": 31004,
      "training_step_time": 8.565042495727539
    },
    {
      "epoch": 2.956867218017578e-05,
      "model_forward_time": 1.749373197555542,
      "step": 31005
    },
    {
      "epoch": 2.956867218017578e-05,
      "step": 31005,
      "training_step_time": 8.382211446762085
    },
    {
      "epoch": 2.9569625854492188e-05,
      "model_forward_time": 1.7563598155975342,
      "step": 31006
    },
    {
      "epoch": 2.9569625854492188e-05,
      "step": 31006,
      "training_step_time": 8.6966712474823
    },
    {
      "epoch": 2.9570579528808595e-05,
      "model_forward_time": 1.746004343032837,
      "step": 31007
    },
    {
      "epoch": 2.9570579528808595e-05,
      "step": 31007,
      "training_step_time": 10.318145036697388
    },
    {
      "epoch": 2.9571533203125e-05,
      "model_forward_time": 1.753342628479004,
      "step": 31008
    },
    {
      "epoch": 2.9571533203125e-05,
      "step": 31008,
      "training_step_time": 9.259305477142334
    },
    {
      "epoch": 2.9572486877441407e-05,
      "model_forward_time": 1.7527456283569336,
      "step": 31009
    },
    {
      "epoch": 2.9572486877441407e-05,
      "step": 31009,
      "training_step_time": 8.514493227005005
    },
    {
      "epoch": 2.9573440551757814e-05,
      "grad_norm": 0.11258363723754883,
      "learning_rate": 8.26180109032095e-06,
      "loss": 0.0638,
      "step": 31010
    },
    {
      "epoch": 2.9573440551757814e-05,
      "model_forward_time": 1.7427864074707031,
      "step": 31010
    },
    {
      "dynamics_loss_avg": 0.053169929422438145,
      "epoch": 2.9573440551757814e-05,
      "step": 31010
    },
    {
      "action_loss_avg": 0.0017310297320364044,
      "epoch": 2.9573440551757814e-05,
      "step": 31010
    },
    {
      "epoch": 2.9573440551757814e-05,
      "step": 31010,
      "training_step_time": 9.139953136444092
    },
    {
      "epoch": 2.9574394226074218e-05,
      "model_forward_time": 1.7364511489868164,
      "step": 31011
    },
    {
      "epoch": 2.9574394226074218e-05,
      "step": 31011,
      "training_step_time": 6.911550521850586
    },
    {
      "epoch": 2.9575347900390625e-05,
      "model_forward_time": 1.739466905593872,
      "step": 31012
    },
    {
      "epoch": 2.9575347900390625e-05,
      "step": 31012,
      "training_step_time": 6.91316032409668
    },
    {
      "epoch": 2.9576301574707033e-05,
      "model_forward_time": 1.7337963581085205,
      "step": 31013
    },
    {
      "epoch": 2.9576301574707033e-05,
      "step": 31013,
      "training_step_time": 6.959660530090332
    },
    {
      "epoch": 2.9577255249023437e-05,
      "model_forward_time": 1.7422542572021484,
      "step": 31014
    },
    {
      "epoch": 2.9577255249023437e-05,
      "step": 31014,
      "training_step_time": 6.818954944610596
    },
    {
      "epoch": 2.9578208923339844e-05,
      "model_forward_time": 1.7486109733581543,
      "step": 31015
    },
    {
      "epoch": 2.9578208923339844e-05,
      "step": 31015,
      "training_step_time": 6.943490743637085
    },
    {
      "epoch": 2.957916259765625e-05,
      "model_forward_time": 1.7429921627044678,
      "step": 31016
    },
    {
      "epoch": 2.957916259765625e-05,
      "step": 31016,
      "training_step_time": 6.9720118045806885
    },
    {
      "epoch": 2.9580116271972656e-05,
      "model_forward_time": 1.7365505695343018,
      "step": 31017
    },
    {
      "epoch": 2.9580116271972656e-05,
      "step": 31017,
      "training_step_time": 6.918019533157349
    },
    {
      "epoch": 2.9581069946289063e-05,
      "model_forward_time": 1.7539050579071045,
      "step": 31018
    },
    {
      "epoch": 2.9581069946289063e-05,
      "step": 31018,
      "training_step_time": 6.884992599487305
    },
    {
      "epoch": 2.9582023620605467e-05,
      "model_forward_time": 2.876042604446411,
      "step": 31019
    },
    {
      "epoch": 2.9582023620605467e-05,
      "step": 31019,
      "training_step_time": 7.670525789260864
    },
    {
      "epoch": 2.9582977294921875e-05,
      "grad_norm": 0.14098325371742249,
      "learning_rate": 8.260547732232807e-06,
      "loss": 0.0599,
      "step": 31020
    },
    {
      "epoch": 2.9582977294921875e-05,
      "model_forward_time": 1.7348496913909912,
      "step": 31020
    },
    {
      "dynamics_loss_avg": 0.06128890663385391,
      "epoch": 2.9582977294921875e-05,
      "step": 31020
    },
    {
      "action_loss_avg": 0.003155897167744115,
      "epoch": 2.9582977294921875e-05,
      "step": 31020
    },
    {
      "epoch": 2.9582977294921875e-05,
      "step": 31020,
      "training_step_time": 7.855497598648071
    },
    {
      "epoch": 2.9583930969238282e-05,
      "model_forward_time": 1.7469477653503418,
      "step": 31021
    },
    {
      "epoch": 2.9583930969238282e-05,
      "step": 31021,
      "training_step_time": 7.024221420288086
    },
    {
      "epoch": 2.9584884643554686e-05,
      "model_forward_time": 1.7454569339752197,
      "step": 31022
    },
    {
      "epoch": 2.9584884643554686e-05,
      "step": 31022,
      "training_step_time": 6.996109962463379
    },
    {
      "epoch": 2.9585838317871093e-05,
      "model_forward_time": 1.7387890815734863,
      "step": 31023
    },
    {
      "epoch": 2.9585838317871093e-05,
      "step": 31023,
      "training_step_time": 6.934406042098999
    },
    {
      "epoch": 2.95867919921875e-05,
      "model_forward_time": 1.7439372539520264,
      "step": 31024
    },
    {
      "epoch": 2.95867919921875e-05,
      "step": 31024,
      "training_step_time": 6.918380260467529
    },
    {
      "epoch": 2.9587745666503905e-05,
      "model_forward_time": 1.7357892990112305,
      "step": 31025
    },
    {
      "epoch": 2.9587745666503905e-05,
      "step": 31025,
      "training_step_time": 6.92877197265625
    },
    {
      "epoch": 2.9588699340820312e-05,
      "model_forward_time": 1.734196424484253,
      "step": 31026
    },
    {
      "epoch": 2.9588699340820312e-05,
      "step": 31026,
      "training_step_time": 6.958560466766357
    },
    {
      "epoch": 2.958965301513672e-05,
      "model_forward_time": 1.7571563720703125,
      "step": 31027
    },
    {
      "epoch": 2.958965301513672e-05,
      "step": 31027,
      "training_step_time": 6.962145090103149
    },
    {
      "epoch": 2.9590606689453124e-05,
      "model_forward_time": 2.884770631790161,
      "step": 31028
    },
    {
      "epoch": 2.9590606689453124e-05,
      "step": 31028,
      "training_step_time": 7.81770396232605
    },
    {
      "epoch": 2.959156036376953e-05,
      "model_forward_time": 1.7530558109283447,
      "step": 31029
    },
    {
      "epoch": 2.959156036376953e-05,
      "step": 31029,
      "training_step_time": 7.879678726196289
    },
    {
      "epoch": 2.959251403808594e-05,
      "grad_norm": 0.16344912350177765,
      "learning_rate": 8.259294017576063e-06,
      "loss": 0.067,
      "step": 31030
    },
    {
      "epoch": 2.959251403808594e-05,
      "model_forward_time": 1.755004644393921,
      "step": 31030
    },
    {
      "dynamics_loss_avg": 0.05454190019518137,
      "epoch": 2.959251403808594e-05,
      "step": 31030
    },
    {
      "action_loss_avg": 0.002264406147878617,
      "epoch": 2.959251403808594e-05,
      "step": 31030
    },
    {
      "epoch": 2.959251403808594e-05,
      "step": 31030,
      "training_step_time": 7.72746467590332
    },
    {
      "epoch": 2.9593467712402343e-05,
      "model_forward_time": 1.7504050731658936,
      "step": 31031
    },
    {
      "epoch": 2.9593467712402343e-05,
      "step": 31031,
      "training_step_time": 6.930721044540405
    },
    {
      "epoch": 2.959442138671875e-05,
      "model_forward_time": 1.7461488246917725,
      "step": 31032
    },
    {
      "epoch": 2.959442138671875e-05,
      "step": 31032,
      "training_step_time": 6.869303941726685
    },
    {
      "epoch": 2.9595375061035157e-05,
      "model_forward_time": 1.7591631412506104,
      "step": 31033
    },
    {
      "epoch": 2.9595375061035157e-05,
      "step": 31033,
      "training_step_time": 6.902191162109375
    },
    {
      "epoch": 2.959632873535156e-05,
      "model_forward_time": 1.7550745010375977,
      "step": 31034
    },
    {
      "epoch": 2.959632873535156e-05,
      "step": 31034,
      "training_step_time": 6.872319221496582
    },
    {
      "epoch": 2.959728240966797e-05,
      "model_forward_time": 1.7597839832305908,
      "step": 31035
    },
    {
      "epoch": 2.959728240966797e-05,
      "step": 31035,
      "training_step_time": 6.789806365966797
    },
    {
      "epoch": 2.9598236083984376e-05,
      "model_forward_time": 1.75482177734375,
      "step": 31036
    },
    {
      "epoch": 2.9598236083984376e-05,
      "step": 31036,
      "training_step_time": 6.868826866149902
    },
    {
      "epoch": 2.959918975830078e-05,
      "model_forward_time": 2.8843843936920166,
      "step": 31037
    },
    {
      "epoch": 2.959918975830078e-05,
      "step": 31037,
      "training_step_time": 7.828715801239014
    },
    {
      "epoch": 2.9600143432617188e-05,
      "model_forward_time": 1.9209997653961182,
      "step": 31038
    },
    {
      "epoch": 2.9600143432617188e-05,
      "step": 31038,
      "training_step_time": 9.63057804107666
    },
    {
      "epoch": 2.9601097106933595e-05,
      "model_forward_time": 1.8707644939422607,
      "step": 31039
    },
    {
      "epoch": 2.9601097106933595e-05,
      "step": 31039,
      "training_step_time": 7.833002328872681
    },
    {
      "epoch": 2.960205078125e-05,
      "grad_norm": 0.13971126079559326,
      "learning_rate": 8.258039946487829e-06,
      "loss": 0.069,
      "step": 31040
    },
    {
      "epoch": 2.960205078125e-05,
      "model_forward_time": 1.921079397201538,
      "step": 31040
    },
    {
      "dynamics_loss_avg": 0.05027412120252848,
      "epoch": 2.960205078125e-05,
      "step": 31040
    },
    {
      "action_loss_avg": 0.0016818064439576119,
      "epoch": 2.960205078125e-05,
      "step": 31040
    },
    {
      "epoch": 2.960205078125e-05,
      "step": 31040,
      "training_step_time": 6.79211688041687
    },
    {
      "epoch": 2.9603004455566407e-05,
      "model_forward_time": 1.8908913135528564,
      "step": 31041
    },
    {
      "epoch": 2.9603004455566407e-05,
      "step": 31041,
      "training_step_time": 6.861015319824219
    },
    {
      "epoch": 2.9603958129882814e-05,
      "model_forward_time": 1.9027931690216064,
      "step": 31042
    },
    {
      "epoch": 2.9603958129882814e-05,
      "step": 31042,
      "training_step_time": 6.913000583648682
    },
    {
      "epoch": 2.9604911804199218e-05,
      "model_forward_time": 1.9093027114868164,
      "step": 31043
    },
    {
      "epoch": 2.9604911804199218e-05,
      "step": 31043,
      "training_step_time": 6.901524305343628
    },
    {
      "epoch": 2.9605865478515625e-05,
      "model_forward_time": 1.9109859466552734,
      "step": 31044
    },
    {
      "epoch": 2.9605865478515625e-05,
      "step": 31044,
      "training_step_time": 6.9251604080200195
    },
    {
      "epoch": 2.9606819152832033e-05,
      "model_forward_time": 1.9028875827789307,
      "step": 31045
    },
    {
      "epoch": 2.9606819152832033e-05,
      "step": 31045,
      "training_step_time": 6.858985424041748
    },
    {
      "epoch": 2.9607772827148437e-05,
      "model_forward_time": 3.037884473800659,
      "step": 31046
    },
    {
      "epoch": 2.9607772827148437e-05,
      "step": 31046,
      "training_step_time": 7.811065435409546
    },
    {
      "epoch": 2.9608726501464844e-05,
      "model_forward_time": 1.8704967498779297,
      "step": 31047
    },
    {
      "epoch": 2.9608726501464844e-05,
      "step": 31047,
      "training_step_time": 10.134069681167603
    },
    {
      "epoch": 2.960968017578125e-05,
      "model_forward_time": 1.7397747039794922,
      "step": 31048
    },
    {
      "epoch": 2.960968017578125e-05,
      "step": 31048,
      "training_step_time": 7.966853618621826
    },
    {
      "epoch": 2.9610633850097656e-05,
      "model_forward_time": 1.7274253368377686,
      "step": 31049
    },
    {
      "epoch": 2.9610633850097656e-05,
      "step": 31049,
      "training_step_time": 6.893801212310791
    },
    {
      "epoch": 2.9611587524414063e-05,
      "grad_norm": 0.1356818825006485,
      "learning_rate": 8.256785519105242e-06,
      "loss": 0.0599,
      "step": 31050
    },
    {
      "epoch": 2.9611587524414063e-05,
      "model_forward_time": 1.7323362827301025,
      "step": 31050
    },
    {
      "dynamics_loss_avg": 0.05120303314179182,
      "epoch": 2.9611587524414063e-05,
      "step": 31050
    },
    {
      "action_loss_avg": 0.0024080475384835156,
      "epoch": 2.9611587524414063e-05,
      "step": 31050
    },
    {
      "epoch": 2.9611587524414063e-05,
      "step": 31050,
      "training_step_time": 6.908805847167969
    },
    {
      "epoch": 2.9612541198730467e-05,
      "model_forward_time": 1.7334551811218262,
      "step": 31051
    },
    {
      "epoch": 2.9612541198730467e-05,
      "step": 31051,
      "training_step_time": 6.9232494831085205
    },
    {
      "epoch": 2.9613494873046875e-05,
      "model_forward_time": 1.7439906597137451,
      "step": 31052
    },
    {
      "epoch": 2.9613494873046875e-05,
      "step": 31052,
      "training_step_time": 6.878344535827637
    },
    {
      "epoch": 2.9614448547363282e-05,
      "model_forward_time": 1.74007248878479,
      "step": 31053
    },
    {
      "epoch": 2.9614448547363282e-05,
      "step": 31053,
      "training_step_time": 6.896632432937622
    },
    {
      "epoch": 2.9615402221679686e-05,
      "model_forward_time": 1.774857997894287,
      "step": 31054
    },
    {
      "epoch": 2.9615402221679686e-05,
      "step": 31054,
      "training_step_time": 6.894939422607422
    },
    {
      "epoch": 2.9616355895996093e-05,
      "model_forward_time": 2.867530107498169,
      "step": 31055
    },
    {
      "epoch": 2.9616355895996093e-05,
      "step": 31055,
      "training_step_time": 7.828450679779053
    },
    {
      "epoch": 2.96173095703125e-05,
      "model_forward_time": 1.8541936874389648,
      "step": 31056
    },
    {
      "epoch": 2.96173095703125e-05,
      "step": 31056,
      "training_step_time": 8.502537727355957
    },
    {
      "epoch": 2.9618263244628905e-05,
      "model_forward_time": 1.7585434913635254,
      "step": 31057
    },
    {
      "epoch": 2.9618263244628905e-05,
      "step": 31057,
      "training_step_time": 7.917740821838379
    },
    {
      "epoch": 2.9619216918945312e-05,
      "model_forward_time": 1.765594244003296,
      "step": 31058
    },
    {
      "epoch": 2.9619216918945312e-05,
      "step": 31058,
      "training_step_time": 6.928797245025635
    },
    {
      "epoch": 2.962017059326172e-05,
      "model_forward_time": 1.7428131103515625,
      "step": 31059
    },
    {
      "epoch": 2.962017059326172e-05,
      "step": 31059,
      "training_step_time": 6.905176877975464
    },
    {
      "epoch": 2.9621124267578124e-05,
      "grad_norm": 0.13120834529399872,
      "learning_rate": 8.25553073556549e-06,
      "loss": 0.0623,
      "step": 31060
    },
    {
      "epoch": 2.9621124267578124e-05,
      "model_forward_time": 1.736509084701538,
      "step": 31060
    },
    {
      "dynamics_loss_avg": 0.058886325545609,
      "epoch": 2.9621124267578124e-05,
      "step": 31060
    },
    {
      "action_loss_avg": 0.0022450160642620174,
      "epoch": 2.9621124267578124e-05,
      "step": 31060
    },
    {
      "epoch": 2.9621124267578124e-05,
      "step": 31060,
      "training_step_time": 6.896326303482056
    },
    {
      "epoch": 2.962207794189453e-05,
      "model_forward_time": 1.7593941688537598,
      "step": 31061
    },
    {
      "epoch": 2.962207794189453e-05,
      "step": 31061,
      "training_step_time": 6.901311159133911
    },
    {
      "epoch": 2.962303161621094e-05,
      "model_forward_time": 1.7460048198699951,
      "step": 31062
    },
    {
      "epoch": 2.962303161621094e-05,
      "step": 31062,
      "training_step_time": 6.8520050048828125
    },
    {
      "epoch": 2.9623985290527343e-05,
      "model_forward_time": 1.7527985572814941,
      "step": 31063
    },
    {
      "epoch": 2.9623985290527343e-05,
      "step": 31063,
      "training_step_time": 6.91196870803833
    },
    {
      "epoch": 2.962493896484375e-05,
      "model_forward_time": 1.7599515914916992,
      "step": 31064
    },
    {
      "epoch": 2.962493896484375e-05,
      "step": 31064,
      "training_step_time": 6.9313764572143555
    },
    {
      "epoch": 2.9625892639160157e-05,
      "model_forward_time": 2.9878973960876465,
      "step": 31065
    },
    {
      "epoch": 2.9625892639160157e-05,
      "step": 31065,
      "training_step_time": 8.781478881835938
    },
    {
      "epoch": 2.962684631347656e-05,
      "model_forward_time": 1.7484676837921143,
      "step": 31066
    },
    {
      "epoch": 2.962684631347656e-05,
      "step": 31066,
      "training_step_time": 9.548686027526855
    },
    {
      "epoch": 2.962779998779297e-05,
      "model_forward_time": 1.7462661266326904,
      "step": 31067
    },
    {
      "epoch": 2.962779998779297e-05,
      "step": 31067,
      "training_step_time": 7.9160315990448
    },
    {
      "epoch": 2.9628753662109376e-05,
      "model_forward_time": 1.7346248626708984,
      "step": 31068
    },
    {
      "epoch": 2.9628753662109376e-05,
      "step": 31068,
      "training_step_time": 6.878360986709595
    },
    {
      "epoch": 2.962970733642578e-05,
      "model_forward_time": 1.7402398586273193,
      "step": 31069
    },
    {
      "epoch": 2.962970733642578e-05,
      "step": 31069,
      "training_step_time": 6.973617315292358
    },
    {
      "epoch": 2.9630661010742188e-05,
      "grad_norm": 0.17442524433135986,
      "learning_rate": 8.254275596005787e-06,
      "loss": 0.0678,
      "step": 31070
    },
    {
      "epoch": 2.9630661010742188e-05,
      "model_forward_time": 1.7337889671325684,
      "step": 31070
    },
    {
      "dynamics_loss_avg": 0.058010921254754065,
      "epoch": 2.9630661010742188e-05,
      "step": 31070
    },
    {
      "action_loss_avg": 0.0015360035933554173,
      "epoch": 2.9630661010742188e-05,
      "step": 31070
    },
    {
      "epoch": 2.9630661010742188e-05,
      "step": 31070,
      "training_step_time": 6.935214042663574
    },
    {
      "epoch": 2.9631614685058595e-05,
      "model_forward_time": 1.7542643547058105,
      "step": 31071
    },
    {
      "epoch": 2.9631614685058595e-05,
      "step": 31071,
      "training_step_time": 6.869626045227051
    },
    {
      "epoch": 2.9632568359375e-05,
      "model_forward_time": 1.7384161949157715,
      "step": 31072
    },
    {
      "epoch": 2.9632568359375e-05,
      "step": 31072,
      "training_step_time": 6.9513349533081055
    },
    {
      "epoch": 2.9633522033691407e-05,
      "model_forward_time": 1.7586777210235596,
      "step": 31073
    },
    {
      "epoch": 2.9633522033691407e-05,
      "step": 31073,
      "training_step_time": 6.950786113739014
    },
    {
      "epoch": 2.9634475708007814e-05,
      "model_forward_time": 2.9824092388153076,
      "step": 31074
    },
    {
      "epoch": 2.9634475708007814e-05,
      "step": 31074,
      "training_step_time": 7.971704721450806
    },
    {
      "epoch": 2.9635429382324218e-05,
      "model_forward_time": 1.7502336502075195,
      "step": 31075
    },
    {
      "epoch": 2.9635429382324218e-05,
      "step": 31075,
      "training_step_time": 8.509073257446289
    },
    {
      "epoch": 2.9636383056640625e-05,
      "model_forward_time": 1.747952938079834,
      "step": 31076
    },
    {
      "epoch": 2.9636383056640625e-05,
      "step": 31076,
      "training_step_time": 7.98745059967041
    },
    {
      "epoch": 2.9637336730957033e-05,
      "model_forward_time": 1.7419214248657227,
      "step": 31077
    },
    {
      "epoch": 2.9637336730957033e-05,
      "step": 31077,
      "training_step_time": 6.931766986846924
    },
    {
      "epoch": 2.9638290405273437e-05,
      "model_forward_time": 1.7637286186218262,
      "step": 31078
    },
    {
      "epoch": 2.9638290405273437e-05,
      "step": 31078,
      "training_step_time": 6.863340377807617
    },
    {
      "epoch": 2.9639244079589844e-05,
      "model_forward_time": 1.7460980415344238,
      "step": 31079
    },
    {
      "epoch": 2.9639244079589844e-05,
      "step": 31079,
      "training_step_time": 6.942755699157715
    },
    {
      "epoch": 2.964019775390625e-05,
      "grad_norm": 0.17830894887447357,
      "learning_rate": 8.2530201005634e-06,
      "loss": 0.0671,
      "step": 31080
    },
    {
      "epoch": 2.964019775390625e-05,
      "model_forward_time": 1.7252659797668457,
      "step": 31080
    },
    {
      "dynamics_loss_avg": 0.07014962621033191,
      "epoch": 2.964019775390625e-05,
      "step": 31080
    },
    {
      "action_loss_avg": 0.0017762935312930495,
      "epoch": 2.964019775390625e-05,
      "step": 31080
    },
    {
      "epoch": 2.964019775390625e-05,
      "step": 31080,
      "training_step_time": 6.834075927734375
    },
    {
      "epoch": 2.9641151428222656e-05,
      "model_forward_time": 1.74674391746521,
      "step": 31081
    },
    {
      "epoch": 2.9641151428222656e-05,
      "step": 31081,
      "training_step_time": 6.78593111038208
    },
    {
      "epoch": 2.9642105102539063e-05,
      "model_forward_time": 1.7503626346588135,
      "step": 31082
    },
    {
      "epoch": 2.9642105102539063e-05,
      "step": 31082,
      "training_step_time": 6.7873170375823975
    },
    {
      "epoch": 2.9643058776855467e-05,
      "model_forward_time": 2.8699467182159424,
      "step": 31083
    },
    {
      "epoch": 2.9643058776855467e-05,
      "step": 31083,
      "training_step_time": 7.828017473220825
    },
    {
      "epoch": 2.9644012451171875e-05,
      "model_forward_time": 1.84425687789917,
      "step": 31084
    },
    {
      "epoch": 2.9644012451171875e-05,
      "step": 31084,
      "training_step_time": 8.690020322799683
    },
    {
      "epoch": 2.9644966125488282e-05,
      "model_forward_time": 1.7419564723968506,
      "step": 31085
    },
    {
      "epoch": 2.9644966125488282e-05,
      "step": 31085,
      "training_step_time": 7.833259344100952
    },
    {
      "epoch": 2.9645919799804686e-05,
      "model_forward_time": 1.7479681968688965,
      "step": 31086
    },
    {
      "epoch": 2.9645919799804686e-05,
      "step": 31086,
      "training_step_time": 7.75018048286438
    },
    {
      "epoch": 2.9646873474121094e-05,
      "model_forward_time": 1.7563502788543701,
      "step": 31087
    },
    {
      "epoch": 2.9646873474121094e-05,
      "step": 31087,
      "training_step_time": 6.769336700439453
    },
    {
      "epoch": 2.96478271484375e-05,
      "model_forward_time": 1.7420952320098877,
      "step": 31088
    },
    {
      "epoch": 2.96478271484375e-05,
      "step": 31088,
      "training_step_time": 6.785557270050049
    },
    {
      "epoch": 2.9648780822753905e-05,
      "model_forward_time": 1.735987901687622,
      "step": 31089
    },
    {
      "epoch": 2.9648780822753905e-05,
      "step": 31089,
      "training_step_time": 6.78382420539856
    },
    {
      "epoch": 2.9649734497070312e-05,
      "grad_norm": 0.16905345022678375,
      "learning_rate": 8.251764249375627e-06,
      "loss": 0.0659,
      "step": 31090
    },
    {
      "epoch": 2.9649734497070312e-05,
      "model_forward_time": 1.7265539169311523,
      "step": 31090
    },
    {
      "dynamics_loss_avg": 0.057825027871876954,
      "epoch": 2.9649734497070312e-05,
      "step": 31090
    },
    {
      "action_loss_avg": 0.0023056615609675646,
      "epoch": 2.9649734497070312e-05,
      "step": 31090
    },
    {
      "epoch": 2.9649734497070312e-05,
      "step": 31090,
      "training_step_time": 6.771069526672363
    },
    {
      "epoch": 2.965068817138672e-05,
      "model_forward_time": 1.7710227966308594,
      "step": 31091
    },
    {
      "epoch": 2.965068817138672e-05,
      "step": 31091,
      "training_step_time": 6.80800461769104
    },
    {
      "epoch": 2.9651641845703124e-05,
      "model_forward_time": 2.8776135444641113,
      "step": 31092
    },
    {
      "epoch": 2.9651641845703124e-05,
      "step": 31092,
      "training_step_time": 8.81986927986145
    },
    {
      "epoch": 2.965259552001953e-05,
      "model_forward_time": 1.8338637351989746,
      "step": 31093
    },
    {
      "epoch": 2.965259552001953e-05,
      "step": 31093,
      "training_step_time": 11.590254306793213
    },
    {
      "epoch": 2.965354919433594e-05,
      "model_forward_time": 1.729539155960083,
      "step": 31094
    },
    {
      "epoch": 2.965354919433594e-05,
      "step": 31094,
      "training_step_time": 7.76155948638916
    },
    {
      "epoch": 2.9654502868652343e-05,
      "model_forward_time": 1.7250592708587646,
      "step": 31095
    },
    {
      "epoch": 2.9654502868652343e-05,
      "step": 31095,
      "training_step_time": 7.8409483432769775
    },
    {
      "epoch": 2.965545654296875e-05,
      "model_forward_time": 1.7489118576049805,
      "step": 31096
    },
    {
      "epoch": 2.965545654296875e-05,
      "step": 31096,
      "training_step_time": 6.875014781951904
    },
    {
      "epoch": 2.9656410217285157e-05,
      "model_forward_time": 1.721735954284668,
      "step": 31097
    },
    {
      "epoch": 2.9656410217285157e-05,
      "step": 31097,
      "training_step_time": 6.880151033401489
    },
    {
      "epoch": 2.965736389160156e-05,
      "model_forward_time": 1.7432022094726562,
      "step": 31098
    },
    {
      "epoch": 2.965736389160156e-05,
      "step": 31098,
      "training_step_time": 6.89708137512207
    },
    {
      "epoch": 2.965831756591797e-05,
      "model_forward_time": 1.3840558528900146,
      "step": 31099
    },
    {
      "epoch": 2.965831756591797e-05,
      "step": 31099,
      "training_step_time": 6.859957695007324
    },
    {
      "epoch": 2.9659271240234376e-05,
      "grad_norm": 0.1642327606678009,
      "learning_rate": 8.250508042579803e-06,
      "loss": 0.0646,
      "step": 31100
    },
    {
      "epoch": 2.9659271240234376e-05,
      "model_forward_time": 1.730006217956543,
      "step": 31100
    },
    {
      "dynamics_loss_avg": 0.06296893060207367,
      "epoch": 2.9659271240234376e-05,
      "step": 31100
    },
    {
      "action_loss_avg": 0.002222381846513599,
      "epoch": 2.9659271240234376e-05,
      "step": 31100
    },
    {
      "epoch": 2.9659271240234376e-05,
      "step": 31100,
      "training_step_time": 6.95803689956665
    },
    {
      "epoch": 2.966022491455078e-05,
      "model_forward_time": 2.8074183464050293,
      "step": 31101
    },
    {
      "epoch": 2.966022491455078e-05,
      "step": 31101,
      "training_step_time": 7.768648386001587
    },
    {
      "epoch": 2.9661178588867188e-05,
      "model_forward_time": 1.837914228439331,
      "step": 31102
    },
    {
      "epoch": 2.9661178588867188e-05,
      "step": 31102,
      "training_step_time": 9.729110717773438
    },
    {
      "epoch": 2.9662132263183595e-05,
      "model_forward_time": 1.7500731945037842,
      "step": 31103
    },
    {
      "epoch": 2.9662132263183595e-05,
      "step": 31103,
      "training_step_time": 7.906257152557373
    },
    {
      "epoch": 2.96630859375e-05,
      "model_forward_time": 1.7365844249725342,
      "step": 31104
    },
    {
      "epoch": 2.96630859375e-05,
      "step": 31104,
      "training_step_time": 7.845499277114868
    },
    {
      "epoch": 2.9664039611816407e-05,
      "model_forward_time": 1.747194528579712,
      "step": 31105
    },
    {
      "epoch": 2.9664039611816407e-05,
      "step": 31105,
      "training_step_time": 6.9405107498168945
    },
    {
      "epoch": 2.9664993286132814e-05,
      "model_forward_time": 1.7411234378814697,
      "step": 31106
    },
    {
      "epoch": 2.9664993286132814e-05,
      "step": 31106,
      "training_step_time": 6.912444591522217
    },
    {
      "epoch": 2.9665946960449218e-05,
      "model_forward_time": 1.7407400608062744,
      "step": 31107
    },
    {
      "epoch": 2.9665946960449218e-05,
      "step": 31107,
      "training_step_time": 6.887694835662842
    },
    {
      "epoch": 2.9666900634765626e-05,
      "model_forward_time": 1.733638048171997,
      "step": 31108
    },
    {
      "epoch": 2.9666900634765626e-05,
      "step": 31108,
      "training_step_time": 6.8700385093688965
    },
    {
      "epoch": 2.9667854309082033e-05,
      "model_forward_time": 1.733795404434204,
      "step": 31109
    },
    {
      "epoch": 2.9667854309082033e-05,
      "step": 31109,
      "training_step_time": 6.9020140171051025
    },
    {
      "epoch": 2.9668807983398437e-05,
      "grad_norm": 0.19891047477722168,
      "learning_rate": 8.249251480313307e-06,
      "loss": 0.0647,
      "step": 31110
    },
    {
      "epoch": 2.9668807983398437e-05,
      "model_forward_time": 1.7314600944519043,
      "step": 31110
    },
    {
      "dynamics_loss_avg": 0.07174565084278584,
      "epoch": 2.9668807983398437e-05,
      "step": 31110
    },
    {
      "action_loss_avg": 0.002976826799567789,
      "epoch": 2.9668807983398437e-05,
      "step": 31110
    },
    {
      "epoch": 2.9668807983398437e-05,
      "step": 31110,
      "training_step_time": 6.881465435028076
    },
    {
      "epoch": 2.9669761657714844e-05,
      "model_forward_time": 2.9858760833740234,
      "step": 31111
    },
    {
      "epoch": 2.9669761657714844e-05,
      "step": 31111,
      "training_step_time": 9.937391519546509
    },
    {
      "epoch": 2.967071533203125e-05,
      "model_forward_time": 1.7402963638305664,
      "step": 31112
    },
    {
      "epoch": 2.967071533203125e-05,
      "step": 31112,
      "training_step_time": 7.894277811050415
    },
    {
      "epoch": 2.9671669006347656e-05,
      "model_forward_time": 1.711571455001831,
      "step": 31113
    },
    {
      "epoch": 2.9671669006347656e-05,
      "step": 31113,
      "training_step_time": 7.989048957824707
    },
    {
      "epoch": 2.9672622680664063e-05,
      "model_forward_time": 1.7066600322723389,
      "step": 31114
    },
    {
      "epoch": 2.9672622680664063e-05,
      "step": 31114,
      "training_step_time": 6.859103441238403
    },
    {
      "epoch": 2.9673576354980467e-05,
      "model_forward_time": 1.716273546218872,
      "step": 31115
    },
    {
      "epoch": 2.9673576354980467e-05,
      "step": 31115,
      "training_step_time": 6.841495752334595
    },
    {
      "epoch": 2.9674530029296875e-05,
      "model_forward_time": 1.7131304740905762,
      "step": 31116
    },
    {
      "epoch": 2.9674530029296875e-05,
      "step": 31116,
      "training_step_time": 6.871612310409546
    },
    {
      "epoch": 2.9675483703613282e-05,
      "model_forward_time": 1.708254337310791,
      "step": 31117
    },
    {
      "epoch": 2.9675483703613282e-05,
      "step": 31117,
      "training_step_time": 6.915834903717041
    },
    {
      "epoch": 2.9676437377929686e-05,
      "model_forward_time": 1.7149326801300049,
      "step": 31118
    },
    {
      "epoch": 2.9676437377929686e-05,
      "step": 31118,
      "training_step_time": 6.881545305252075
    },
    {
      "epoch": 2.9677391052246094e-05,
      "model_forward_time": 1.7212893962860107,
      "step": 31119
    },
    {
      "epoch": 2.9677391052246094e-05,
      "step": 31119,
      "training_step_time": 6.807843208312988
    },
    {
      "epoch": 2.96783447265625e-05,
      "grad_norm": 0.1560579538345337,
      "learning_rate": 8.247994562713555e-06,
      "loss": 0.0677,
      "step": 31120
    },
    {
      "epoch": 2.96783447265625e-05,
      "model_forward_time": 2.271517753601074,
      "step": 31120
    },
    {
      "dynamics_loss_avg": 0.0548041770234704,
      "epoch": 2.96783447265625e-05,
      "step": 31120
    },
    {
      "action_loss_avg": 0.0020342716728919187,
      "epoch": 2.96783447265625e-05,
      "step": 31120
    },
    {
      "epoch": 2.96783447265625e-05,
      "step": 31120,
      "training_step_time": 8.80743956565857
    },
    {
      "epoch": 2.9679298400878905e-05,
      "model_forward_time": 1.7453718185424805,
      "step": 31121
    },
    {
      "epoch": 2.9679298400878905e-05,
      "step": 31121,
      "training_step_time": 9.566826581954956
    },
    {
      "epoch": 2.9680252075195312e-05,
      "model_forward_time": 1.724238395690918,
      "step": 31122
    },
    {
      "epoch": 2.9680252075195312e-05,
      "step": 31122,
      "training_step_time": 7.87327766418457
    },
    {
      "epoch": 2.968120574951172e-05,
      "model_forward_time": 1.722445011138916,
      "step": 31123
    },
    {
      "epoch": 2.968120574951172e-05,
      "step": 31123,
      "training_step_time": 7.4923930168151855
    },
    {
      "epoch": 2.9682159423828124e-05,
      "model_forward_time": 1.7202706336975098,
      "step": 31124
    },
    {
      "epoch": 2.9682159423828124e-05,
      "step": 31124,
      "training_step_time": 6.7773213386535645
    },
    {
      "epoch": 2.968311309814453e-05,
      "model_forward_time": 1.720858097076416,
      "step": 31125
    },
    {
      "epoch": 2.968311309814453e-05,
      "step": 31125,
      "training_step_time": 6.904242753982544
    },
    {
      "epoch": 2.968406677246094e-05,
      "model_forward_time": 1.7276017665863037,
      "step": 31126
    },
    {
      "epoch": 2.968406677246094e-05,
      "step": 31126,
      "training_step_time": 6.885624170303345
    },
    {
      "epoch": 2.9685020446777343e-05,
      "model_forward_time": 1.7216377258300781,
      "step": 31127
    },
    {
      "epoch": 2.9685020446777343e-05,
      "step": 31127,
      "training_step_time": 6.840719223022461
    },
    {
      "epoch": 2.968597412109375e-05,
      "model_forward_time": 1.722618818283081,
      "step": 31128
    },
    {
      "epoch": 2.968597412109375e-05,
      "step": 31128,
      "training_step_time": 6.881000757217407
    },
    {
      "epoch": 2.9686927795410158e-05,
      "model_forward_time": 2.278109312057495,
      "step": 31129
    },
    {
      "epoch": 2.9686927795410158e-05,
      "step": 31129,
      "training_step_time": 10.47598910331726
    },
    {
      "epoch": 2.968788146972656e-05,
      "grad_norm": 0.15539869666099548,
      "learning_rate": 8.246737289917998e-06,
      "loss": 0.0606,
      "step": 31130
    },
    {
      "epoch": 2.968788146972656e-05,
      "model_forward_time": 1.7309949398040771,
      "step": 31130
    },
    {
      "dynamics_loss_avg": 0.05358398500829935,
      "epoch": 2.968788146972656e-05,
      "step": 31130
    },
    {
      "action_loss_avg": 0.0022772686555981637,
      "epoch": 2.968788146972656e-05,
      "step": 31130
    },
    {
      "epoch": 2.968788146972656e-05,
      "step": 31130,
      "training_step_time": 11.551742792129517
    },
    {
      "epoch": 2.968883514404297e-05,
      "model_forward_time": 1.7250547409057617,
      "step": 31131
    },
    {
      "epoch": 2.968883514404297e-05,
      "step": 31131,
      "training_step_time": 8.477905750274658
    },
    {
      "epoch": 2.9689788818359376e-05,
      "model_forward_time": 1.7217469215393066,
      "step": 31132
    },
    {
      "epoch": 2.9689788818359376e-05,
      "step": 31132,
      "training_step_time": 7.587630271911621
    },
    {
      "epoch": 2.969074249267578e-05,
      "model_forward_time": 1.7244963645935059,
      "step": 31133
    },
    {
      "epoch": 2.969074249267578e-05,
      "step": 31133,
      "training_step_time": 6.8679492473602295
    },
    {
      "epoch": 2.9691696166992188e-05,
      "model_forward_time": 1.7232341766357422,
      "step": 31134
    },
    {
      "epoch": 2.9691696166992188e-05,
      "step": 31134,
      "training_step_time": 6.862224578857422
    },
    {
      "epoch": 2.9692649841308595e-05,
      "model_forward_time": 1.6966781616210938,
      "step": 31135
    },
    {
      "epoch": 2.9692649841308595e-05,
      "step": 31135,
      "training_step_time": 6.868228197097778
    },
    {
      "epoch": 2.9693603515625e-05,
      "model_forward_time": 1.7176241874694824,
      "step": 31136
    },
    {
      "epoch": 2.9693603515625e-05,
      "step": 31136,
      "training_step_time": 6.825592756271362
    },
    {
      "epoch": 2.9694557189941407e-05,
      "model_forward_time": 1.7218213081359863,
      "step": 31137
    },
    {
      "epoch": 2.9694557189941407e-05,
      "step": 31137,
      "training_step_time": 6.857983112335205
    },
    {
      "epoch": 2.9695510864257814e-05,
      "model_forward_time": 2.286695718765259,
      "step": 31138
    },
    {
      "epoch": 2.9695510864257814e-05,
      "step": 31138,
      "training_step_time": 8.592751502990723
    },
    {
      "epoch": 2.9696464538574218e-05,
      "model_forward_time": 1.7270848751068115,
      "step": 31139
    },
    {
      "epoch": 2.9696464538574218e-05,
      "step": 31139,
      "training_step_time": 9.389309883117676
    },
    {
      "epoch": 2.9697418212890626e-05,
      "grad_norm": 0.1941666603088379,
      "learning_rate": 8.245479662064135e-06,
      "loss": 0.0658,
      "step": 31140
    },
    {
      "epoch": 2.9697418212890626e-05,
      "model_forward_time": 1.7192986011505127,
      "step": 31140
    },
    {
      "dynamics_loss_avg": 0.06808731816709042,
      "epoch": 2.9697418212890626e-05,
      "step": 31140
    },
    {
      "action_loss_avg": 0.002487761914380826,
      "epoch": 2.9697418212890626e-05,
      "step": 31140
    },
    {
      "epoch": 2.9697418212890626e-05,
      "step": 31140,
      "training_step_time": 7.986624479293823
    },
    {
      "epoch": 2.969837188720703e-05,
      "model_forward_time": 1.7200591564178467,
      "step": 31141
    },
    {
      "epoch": 2.969837188720703e-05,
      "step": 31141,
      "training_step_time": 8.008085489273071
    },
    {
      "epoch": 2.9699325561523437e-05,
      "model_forward_time": 1.7189719676971436,
      "step": 31142
    },
    {
      "epoch": 2.9699325561523437e-05,
      "step": 31142,
      "training_step_time": 6.848525285720825
    },
    {
      "epoch": 2.9700279235839844e-05,
      "model_forward_time": 1.7259020805358887,
      "step": 31143
    },
    {
      "epoch": 2.9700279235839844e-05,
      "step": 31143,
      "training_step_time": 6.848270893096924
    },
    {
      "epoch": 2.970123291015625e-05,
      "model_forward_time": 1.7282438278198242,
      "step": 31144
    },
    {
      "epoch": 2.970123291015625e-05,
      "step": 31144,
      "training_step_time": 6.802566766738892
    },
    {
      "epoch": 2.9702186584472656e-05,
      "model_forward_time": 1.718895435333252,
      "step": 31145
    },
    {
      "epoch": 2.9702186584472656e-05,
      "step": 31145,
      "training_step_time": 6.9066267013549805
    },
    {
      "epoch": 2.9703140258789063e-05,
      "model_forward_time": 1.7167649269104004,
      "step": 31146
    },
    {
      "epoch": 2.9703140258789063e-05,
      "step": 31146,
      "training_step_time": 6.893619060516357
    },
    {
      "epoch": 2.9704093933105467e-05,
      "model_forward_time": 2.260345458984375,
      "step": 31147
    },
    {
      "epoch": 2.9704093933105467e-05,
      "step": 31147,
      "training_step_time": 7.615965127944946
    },
    {
      "epoch": 2.9705047607421875e-05,
      "model_forward_time": 1.7254912853240967,
      "step": 31148
    },
    {
      "epoch": 2.9705047607421875e-05,
      "step": 31148,
      "training_step_time": 8.459010601043701
    },
    {
      "epoch": 2.9706001281738282e-05,
      "model_forward_time": 1.7431812286376953,
      "step": 31149
    },
    {
      "epoch": 2.9706001281738282e-05,
      "step": 31149,
      "training_step_time": 10.184098482131958
    },
    {
      "epoch": 2.9706954956054686e-05,
      "grad_norm": 0.22472384572029114,
      "learning_rate": 8.244221679289497e-06,
      "loss": 0.0636,
      "step": 31150
    },
    {
      "epoch": 2.9706954956054686e-05,
      "model_forward_time": 1.7170829772949219,
      "step": 31150
    },
    {
      "dynamics_loss_avg": 0.05690862610936165,
      "epoch": 2.9706954956054686e-05,
      "step": 31150
    },
    {
      "action_loss_avg": 0.0027803086384665223,
      "epoch": 2.9706954956054686e-05,
      "step": 31150
    },
    {
      "epoch": 2.9706954956054686e-05,
      "step": 31150,
      "training_step_time": 7.916081666946411
    },
    {
      "epoch": 2.9707908630371094e-05,
      "model_forward_time": 1.7193059921264648,
      "step": 31151
    },
    {
      "epoch": 2.9707908630371094e-05,
      "step": 31151,
      "training_step_time": 6.919921398162842
    },
    {
      "epoch": 2.97088623046875e-05,
      "model_forward_time": 1.7188951969146729,
      "step": 31152
    },
    {
      "epoch": 2.97088623046875e-05,
      "step": 31152,
      "training_step_time": 6.94198751449585
    },
    {
      "epoch": 2.9709815979003905e-05,
      "model_forward_time": 1.7231152057647705,
      "step": 31153
    },
    {
      "epoch": 2.9709815979003905e-05,
      "step": 31153,
      "training_step_time": 6.901437044143677
    },
    {
      "epoch": 2.9710769653320312e-05,
      "model_forward_time": 1.7177019119262695,
      "step": 31154
    },
    {
      "epoch": 2.9710769653320312e-05,
      "step": 31154,
      "training_step_time": 6.930644512176514
    },
    {
      "epoch": 2.971172332763672e-05,
      "model_forward_time": 1.7235338687896729,
      "step": 31155
    },
    {
      "epoch": 2.971172332763672e-05,
      "step": 31155,
      "training_step_time": 6.936832666397095
    },
    {
      "epoch": 2.9712677001953124e-05,
      "model_forward_time": 1.718881368637085,
      "step": 31156
    },
    {
      "epoch": 2.9712677001953124e-05,
      "step": 31156,
      "training_step_time": 7.596841812133789
    },
    {
      "epoch": 2.971363067626953e-05,
      "model_forward_time": 2.269318103790283,
      "step": 31157
    },
    {
      "epoch": 2.971363067626953e-05,
      "step": 31157,
      "training_step_time": 8.577821493148804
    },
    {
      "epoch": 2.971458435058594e-05,
      "model_forward_time": 1.7331607341766357,
      "step": 31158
    },
    {
      "epoch": 2.971458435058594e-05,
      "step": 31158,
      "training_step_time": 11.09525179862976
    },
    {
      "epoch": 2.9715538024902343e-05,
      "model_forward_time": 1.7212598323822021,
      "step": 31159
    },
    {
      "epoch": 2.9715538024902343e-05,
      "step": 31159,
      "training_step_time": 7.866995334625244
    },
    {
      "epoch": 2.971649169921875e-05,
      "grad_norm": 0.14967718720436096,
      "learning_rate": 8.24296334173165e-06,
      "loss": 0.065,
      "step": 31160
    },
    {
      "epoch": 2.971649169921875e-05,
      "model_forward_time": 1.7118914127349854,
      "step": 31160
    },
    {
      "dynamics_loss_avg": 0.05711023285984993,
      "epoch": 2.971649169921875e-05,
      "step": 31160
    },
    {
      "action_loss_avg": 0.0020581655378919097,
      "epoch": 2.971649169921875e-05,
      "step": 31160
    },
    {
      "epoch": 2.971649169921875e-05,
      "step": 31160,
      "training_step_time": 7.57700777053833
    },
    {
      "epoch": 2.9717445373535158e-05,
      "model_forward_time": 1.7231502532958984,
      "step": 31161
    },
    {
      "epoch": 2.9717445373535158e-05,
      "step": 31161,
      "training_step_time": 6.930384397506714
    },
    {
      "epoch": 2.971839904785156e-05,
      "model_forward_time": 1.699683666229248,
      "step": 31162
    },
    {
      "epoch": 2.971839904785156e-05,
      "step": 31162,
      "training_step_time": 6.950850248336792
    },
    {
      "epoch": 2.971935272216797e-05,
      "model_forward_time": 1.7157855033874512,
      "step": 31163
    },
    {
      "epoch": 2.971935272216797e-05,
      "step": 31163,
      "training_step_time": 6.911728143692017
    },
    {
      "epoch": 2.9720306396484376e-05,
      "model_forward_time": 1.719090223312378,
      "step": 31164
    },
    {
      "epoch": 2.9720306396484376e-05,
      "step": 31164,
      "training_step_time": 6.85863733291626
    },
    {
      "epoch": 2.972126007080078e-05,
      "model_forward_time": 1.7138192653656006,
      "step": 31165
    },
    {
      "epoch": 2.972126007080078e-05,
      "step": 31165,
      "training_step_time": 7.632653474807739
    },
    {
      "epoch": 2.9722213745117188e-05,
      "model_forward_time": 2.2473268508911133,
      "step": 31166
    },
    {
      "epoch": 2.9722213745117188e-05,
      "step": 31166,
      "training_step_time": 8.86048412322998
    },
    {
      "epoch": 2.9723167419433595e-05,
      "model_forward_time": 1.7300124168395996,
      "step": 31167
    },
    {
      "epoch": 2.9723167419433595e-05,
      "step": 31167,
      "training_step_time": 10.798759460449219
    },
    {
      "epoch": 2.972412109375e-05,
      "model_forward_time": 1.7136693000793457,
      "step": 31168
    },
    {
      "epoch": 2.972412109375e-05,
      "step": 31168,
      "training_step_time": 8.521879196166992
    },
    {
      "epoch": 2.9725074768066407e-05,
      "model_forward_time": 1.717803716659546,
      "step": 31169
    },
    {
      "epoch": 2.9725074768066407e-05,
      "step": 31169,
      "training_step_time": 8.032942056655884
    },
    {
      "epoch": 2.9726028442382814e-05,
      "grad_norm": 0.15563040971755981,
      "learning_rate": 8.24170464952821e-06,
      "loss": 0.0651,
      "step": 31170
    },
    {
      "epoch": 2.9726028442382814e-05,
      "model_forward_time": 1.7072596549987793,
      "step": 31170
    },
    {
      "dynamics_loss_avg": 0.06465978994965553,
      "epoch": 2.9726028442382814e-05,
      "step": 31170
    },
    {
      "action_loss_avg": 0.002264987037051469,
      "epoch": 2.9726028442382814e-05,
      "step": 31170
    },
    {
      "epoch": 2.9726028442382814e-05,
      "step": 31170,
      "training_step_time": 6.906433343887329
    },
    {
      "epoch": 2.9726982116699218e-05,
      "model_forward_time": 1.7089149951934814,
      "step": 31171
    },
    {
      "epoch": 2.9726982116699218e-05,
      "step": 31171,
      "training_step_time": 6.926934719085693
    },
    {
      "epoch": 2.9727935791015626e-05,
      "model_forward_time": 1.708782434463501,
      "step": 31172
    },
    {
      "epoch": 2.9727935791015626e-05,
      "step": 31172,
      "training_step_time": 6.913607358932495
    },
    {
      "epoch": 2.972888946533203e-05,
      "model_forward_time": 1.720538854598999,
      "step": 31173
    },
    {
      "epoch": 2.972888946533203e-05,
      "step": 31173,
      "training_step_time": 6.862157106399536
    },
    {
      "epoch": 2.9729843139648437e-05,
      "model_forward_time": 1.7131218910217285,
      "step": 31174
    },
    {
      "epoch": 2.9729843139648437e-05,
      "step": 31174,
      "training_step_time": 7.571101427078247
    },
    {
      "epoch": 2.9730796813964844e-05,
      "model_forward_time": 2.2469420433044434,
      "step": 31175
    },
    {
      "epoch": 2.9730796813964844e-05,
      "step": 31175,
      "training_step_time": 9.227516889572144
    },
    {
      "epoch": 2.973175048828125e-05,
      "model_forward_time": 1.7271738052368164,
      "step": 31176
    },
    {
      "epoch": 2.973175048828125e-05,
      "step": 31176,
      "training_step_time": 8.959498882293701
    },
    {
      "epoch": 2.9732704162597656e-05,
      "model_forward_time": 1.7079002857208252,
      "step": 31177
    },
    {
      "epoch": 2.9732704162597656e-05,
      "step": 31177,
      "training_step_time": 8.614433526992798
    },
    {
      "epoch": 2.9733657836914063e-05,
      "model_forward_time": 1.7154655456542969,
      "step": 31178
    },
    {
      "epoch": 2.9733657836914063e-05,
      "step": 31178,
      "training_step_time": 7.879013776779175
    },
    {
      "epoch": 2.9734611511230467e-05,
      "model_forward_time": 1.7124278545379639,
      "step": 31179
    },
    {
      "epoch": 2.9734611511230467e-05,
      "step": 31179,
      "training_step_time": 6.822950124740601
    },
    {
      "epoch": 2.9735565185546875e-05,
      "grad_norm": 0.15871857106685638,
      "learning_rate": 8.240445602816824e-06,
      "loss": 0.0617,
      "step": 31180
    },
    {
      "epoch": 2.9735565185546875e-05,
      "model_forward_time": 1.7048158645629883,
      "step": 31180
    },
    {
      "dynamics_loss_avg": 0.06060162633657455,
      "epoch": 2.9735565185546875e-05,
      "step": 31180
    },
    {
      "action_loss_avg": 0.002392719779163599,
      "epoch": 2.9735565185546875e-05,
      "step": 31180
    },
    {
      "epoch": 2.9735565185546875e-05,
      "step": 31180,
      "training_step_time": 6.809524059295654
    },
    {
      "epoch": 2.9736518859863282e-05,
      "model_forward_time": 1.7109010219573975,
      "step": 31181
    },
    {
      "epoch": 2.9736518859863282e-05,
      "step": 31181,
      "training_step_time": 6.846345663070679
    },
    {
      "epoch": 2.9737472534179686e-05,
      "model_forward_time": 1.7167677879333496,
      "step": 31182
    },
    {
      "epoch": 2.9737472534179686e-05,
      "step": 31182,
      "training_step_time": 6.827118873596191
    },
    {
      "epoch": 2.9738426208496094e-05,
      "model_forward_time": 1.7252962589263916,
      "step": 31183
    },
    {
      "epoch": 2.9738426208496094e-05,
      "step": 31183,
      "training_step_time": 7.576175928115845
    },
    {
      "epoch": 2.97393798828125e-05,
      "model_forward_time": 2.28584885597229,
      "step": 31184
    },
    {
      "epoch": 2.97393798828125e-05,
      "step": 31184,
      "training_step_time": 8.421100378036499
    },
    {
      "epoch": 2.9740333557128905e-05,
      "model_forward_time": 1.7286407947540283,
      "step": 31185
    },
    {
      "epoch": 2.9740333557128905e-05,
      "step": 31185,
      "training_step_time": 9.46684741973877
    },
    {
      "epoch": 2.9741287231445312e-05,
      "model_forward_time": 1.735295295715332,
      "step": 31186
    },
    {
      "epoch": 2.9741287231445312e-05,
      "step": 31186,
      "training_step_time": 7.943278789520264
    },
    {
      "epoch": 2.974224090576172e-05,
      "model_forward_time": 1.745581865310669,
      "step": 31187
    },
    {
      "epoch": 2.974224090576172e-05,
      "step": 31187,
      "training_step_time": 9.732158184051514
    },
    {
      "epoch": 2.9743194580078124e-05,
      "model_forward_time": 1.7326698303222656,
      "step": 31188
    },
    {
      "epoch": 2.9743194580078124e-05,
      "step": 31188,
      "training_step_time": 6.892943382263184
    },
    {
      "epoch": 2.974414825439453e-05,
      "model_forward_time": 1.7340211868286133,
      "step": 31189
    },
    {
      "epoch": 2.974414825439453e-05,
      "step": 31189,
      "training_step_time": 6.9446702003479
    },
    {
      "epoch": 2.974510192871094e-05,
      "grad_norm": 0.1477406919002533,
      "learning_rate": 8.239186201735178e-06,
      "loss": 0.0641,
      "step": 31190
    },
    {
      "epoch": 2.974510192871094e-05,
      "model_forward_time": 1.7360010147094727,
      "step": 31190
    },
    {
      "dynamics_loss_avg": 0.059094236604869364,
      "epoch": 2.974510192871094e-05,
      "step": 31190
    },
    {
      "action_loss_avg": 0.0020376636588480324,
      "epoch": 2.974510192871094e-05,
      "step": 31190
    },
    {
      "epoch": 2.974510192871094e-05,
      "step": 31190,
      "training_step_time": 6.921590805053711
    },
    {
      "epoch": 2.9746055603027343e-05,
      "model_forward_time": 1.732865810394287,
      "step": 31191
    },
    {
      "epoch": 2.9746055603027343e-05,
      "step": 31191,
      "training_step_time": 6.956918239593506
    },
    {
      "epoch": 2.974700927734375e-05,
      "model_forward_time": 1.7356865406036377,
      "step": 31192
    },
    {
      "epoch": 2.974700927734375e-05,
      "step": 31192,
      "training_step_time": 7.4863362312316895
    },
    {
      "epoch": 2.9747962951660158e-05,
      "model_forward_time": 2.300248146057129,
      "step": 31193
    },
    {
      "epoch": 2.9747962951660158e-05,
      "step": 31193,
      "training_step_time": 8.728271484375
    },
    {
      "epoch": 2.974891662597656e-05,
      "model_forward_time": 1.7304294109344482,
      "step": 31194
    },
    {
      "epoch": 2.974891662597656e-05,
      "step": 31194,
      "training_step_time": 9.040793657302856
    },
    {
      "epoch": 2.974987030029297e-05,
      "model_forward_time": 1.7341039180755615,
      "step": 31195
    },
    {
      "epoch": 2.974987030029297e-05,
      "step": 31195,
      "training_step_time": 10.729612827301025
    },
    {
      "epoch": 2.9750823974609376e-05,
      "model_forward_time": 1.7293410301208496,
      "step": 31196
    },
    {
      "epoch": 2.9750823974609376e-05,
      "step": 31196,
      "training_step_time": 8.88145899772644
    },
    {
      "epoch": 2.975177764892578e-05,
      "model_forward_time": 1.7345876693725586,
      "step": 31197
    },
    {
      "epoch": 2.975177764892578e-05,
      "step": 31197,
      "training_step_time": 7.530518293380737
    },
    {
      "epoch": 2.9752731323242188e-05,
      "model_forward_time": 1.7323362827301025,
      "step": 31198
    },
    {
      "epoch": 2.9752731323242188e-05,
      "step": 31198,
      "training_step_time": 6.869268894195557
    },
    {
      "epoch": 2.9753684997558595e-05,
      "model_forward_time": 1.7127320766448975,
      "step": 31199
    },
    {
      "epoch": 2.9753684997558595e-05,
      "step": 31199,
      "training_step_time": 6.84114670753479
    },
    {
      "epoch": 2.9754638671875e-05,
      "grad_norm": 0.16332247853279114,
      "learning_rate": 8.237926446420999e-06,
      "loss": 0.0605,
      "step": 31200
    },
    {
      "epoch": 2.9754638671875e-05,
      "model_forward_time": 1.7190353870391846,
      "step": 31200
    },
    {
      "dynamics_loss_avg": 0.06582723874598742,
      "epoch": 2.9754638671875e-05,
      "step": 31200
    },
    {
      "action_loss_avg": 0.002294071135111153,
      "epoch": 2.9754638671875e-05,
      "step": 31200
    },
    {
      "epoch": 2.9754638671875e-05,
      "step": 31200,
      "training_step_time": 6.894607067108154
    },
    {
      "epoch": 2.9755592346191407e-05,
      "model_forward_time": 1.7313215732574463,
      "step": 31201
    },
    {
      "epoch": 2.9755592346191407e-05,
      "step": 31201,
      "training_step_time": 7.513723611831665
    },
    {
      "epoch": 2.9756546020507814e-05,
      "model_forward_time": 1.724968671798706,
      "step": 31202
    },
    {
      "epoch": 2.9756546020507814e-05,
      "step": 31202,
      "training_step_time": 8.265583038330078
    },
    {
      "epoch": 2.9757499694824218e-05,
      "model_forward_time": 1.719994306564331,
      "step": 31203
    },
    {
      "epoch": 2.9757499694824218e-05,
      "step": 31203,
      "training_step_time": 8.549900770187378
    },
    {
      "epoch": 2.9758453369140626e-05,
      "model_forward_time": 1.810577392578125,
      "step": 31204
    },
    {
      "epoch": 2.9758453369140626e-05,
      "step": 31204,
      "training_step_time": 9.344305038452148
    },
    {
      "epoch": 2.975940704345703e-05,
      "model_forward_time": 1.7325806617736816,
      "step": 31205
    },
    {
      "epoch": 2.975940704345703e-05,
      "step": 31205,
      "training_step_time": 9.686132192611694
    },
    {
      "epoch": 2.9760360717773437e-05,
      "model_forward_time": 1.7325634956359863,
      "step": 31206
    },
    {
      "epoch": 2.9760360717773437e-05,
      "step": 31206,
      "training_step_time": 7.557898283004761
    },
    {
      "epoch": 2.9761314392089844e-05,
      "model_forward_time": 1.7334699630737305,
      "step": 31207
    },
    {
      "epoch": 2.9761314392089844e-05,
      "step": 31207,
      "training_step_time": 6.8592307567596436
    },
    {
      "epoch": 2.976226806640625e-05,
      "model_forward_time": 1.738396167755127,
      "step": 31208
    },
    {
      "epoch": 2.976226806640625e-05,
      "step": 31208,
      "training_step_time": 6.816855192184448
    },
    {
      "epoch": 2.9763221740722656e-05,
      "model_forward_time": 1.7455251216888428,
      "step": 31209
    },
    {
      "epoch": 2.9763221740722656e-05,
      "step": 31209,
      "training_step_time": 6.872378349304199
    },
    {
      "epoch": 2.9764175415039063e-05,
      "grad_norm": 0.14313599467277527,
      "learning_rate": 8.236666337012052e-06,
      "loss": 0.0643,
      "step": 31210
    },
    {
      "epoch": 2.9764175415039063e-05,
      "model_forward_time": 1.7307353019714355,
      "step": 31210
    },
    {
      "dynamics_loss_avg": 0.05601233337074518,
      "epoch": 2.9764175415039063e-05,
      "step": 31210
    },
    {
      "action_loss_avg": 0.0017310724710114301,
      "epoch": 2.9764175415039063e-05,
      "step": 31210
    },
    {
      "epoch": 2.9764175415039063e-05,
      "step": 31210,
      "training_step_time": 7.519543886184692
    },
    {
      "epoch": 2.9765129089355467e-05,
      "model_forward_time": 1.7173538208007812,
      "step": 31211
    },
    {
      "epoch": 2.9765129089355467e-05,
      "step": 31211,
      "training_step_time": 7.733806610107422
    },
    {
      "epoch": 2.9766082763671875e-05,
      "model_forward_time": 1.8795382976531982,
      "step": 31212
    },
    {
      "epoch": 2.9766082763671875e-05,
      "step": 31212,
      "training_step_time": 8.574433088302612
    },
    {
      "epoch": 2.9767036437988282e-05,
      "model_forward_time": 1.7954349517822266,
      "step": 31213
    },
    {
      "epoch": 2.9767036437988282e-05,
      "step": 31213,
      "training_step_time": 8.960710287094116
    },
    {
      "epoch": 2.9767990112304686e-05,
      "model_forward_time": 1.7365353107452393,
      "step": 31214
    },
    {
      "epoch": 2.9767990112304686e-05,
      "step": 31214,
      "training_step_time": 7.96433687210083
    },
    {
      "epoch": 2.9768943786621094e-05,
      "model_forward_time": 1.723799228668213,
      "step": 31215
    },
    {
      "epoch": 2.9768943786621094e-05,
      "step": 31215,
      "training_step_time": 7.7764410972595215
    },
    {
      "epoch": 2.97698974609375e-05,
      "model_forward_time": 1.7204315662384033,
      "step": 31216
    },
    {
      "epoch": 2.97698974609375e-05,
      "step": 31216,
      "training_step_time": 7.197155714035034
    },
    {
      "epoch": 2.9770851135253905e-05,
      "model_forward_time": 1.726966142654419,
      "step": 31217
    },
    {
      "epoch": 2.9770851135253905e-05,
      "step": 31217,
      "training_step_time": 6.919250726699829
    },
    {
      "epoch": 2.9771804809570312e-05,
      "model_forward_time": 1.3502886295318604,
      "step": 31218
    },
    {
      "epoch": 2.9771804809570312e-05,
      "step": 31218,
      "training_step_time": 6.917158603668213
    },
    {
      "epoch": 2.977275848388672e-05,
      "model_forward_time": 1.725886344909668,
      "step": 31219
    },
    {
      "epoch": 2.977275848388672e-05,
      "step": 31219,
      "training_step_time": 6.994909286499023
    },
    {
      "epoch": 2.9773712158203124e-05,
      "grad_norm": 0.13350684940814972,
      "learning_rate": 8.235405873646141e-06,
      "loss": 0.0611,
      "step": 31220
    },
    {
      "epoch": 2.9773712158203124e-05,
      "model_forward_time": 1.7109248638153076,
      "step": 31220
    },
    {
      "dynamics_loss_avg": 0.05475369952619076,
      "epoch": 2.9773712158203124e-05,
      "step": 31220
    },
    {
      "action_loss_avg": 0.0018705346039496363,
      "epoch": 2.9773712158203124e-05,
      "step": 31220
    },
    {
      "epoch": 2.9773712158203124e-05,
      "step": 31220,
      "training_step_time": 8.296556234359741
    },
    {
      "epoch": 2.977466583251953e-05,
      "model_forward_time": 2.255115032196045,
      "step": 31221
    },
    {
      "epoch": 2.977466583251953e-05,
      "step": 31221,
      "training_step_time": 8.541623592376709
    },
    {
      "epoch": 2.977561950683594e-05,
      "model_forward_time": 1.789888858795166,
      "step": 31222
    },
    {
      "epoch": 2.977561950683594e-05,
      "step": 31222,
      "training_step_time": 8.36753797531128
    },
    {
      "epoch": 2.9776573181152343e-05,
      "model_forward_time": 1.7267827987670898,
      "step": 31223
    },
    {
      "epoch": 2.9776573181152343e-05,
      "step": 31223,
      "training_step_time": 7.863570690155029
    },
    {
      "epoch": 2.977752685546875e-05,
      "model_forward_time": 1.738450527191162,
      "step": 31224
    },
    {
      "epoch": 2.977752685546875e-05,
      "step": 31224,
      "training_step_time": 7.792363166809082
    },
    {
      "epoch": 2.9778480529785158e-05,
      "model_forward_time": 1.7258830070495605,
      "step": 31225
    },
    {
      "epoch": 2.9778480529785158e-05,
      "step": 31225,
      "training_step_time": 7.169359922409058
    },
    {
      "epoch": 2.977943420410156e-05,
      "model_forward_time": 1.7243070602416992,
      "step": 31226
    },
    {
      "epoch": 2.977943420410156e-05,
      "step": 31226,
      "training_step_time": 6.845620393753052
    },
    {
      "epoch": 2.978038787841797e-05,
      "model_forward_time": 1.7349731922149658,
      "step": 31227
    },
    {
      "epoch": 2.978038787841797e-05,
      "step": 31227,
      "training_step_time": 6.8838653564453125
    },
    {
      "epoch": 2.9781341552734376e-05,
      "model_forward_time": 1.7266113758087158,
      "step": 31228
    },
    {
      "epoch": 2.9781341552734376e-05,
      "step": 31228,
      "training_step_time": 6.8545355796813965
    },
    {
      "epoch": 2.978229522705078e-05,
      "model_forward_time": 1.7225747108459473,
      "step": 31229
    },
    {
      "epoch": 2.978229522705078e-05,
      "step": 31229,
      "training_step_time": 7.64846658706665
    },
    {
      "epoch": 2.9783248901367188e-05,
      "grad_norm": 0.15244098007678986,
      "learning_rate": 8.234145056461107e-06,
      "loss": 0.0591,
      "step": 31230
    },
    {
      "epoch": 2.9783248901367188e-05,
      "model_forward_time": 2.2760069370269775,
      "step": 31230
    },
    {
      "dynamics_loss_avg": 0.06312052197754384,
      "epoch": 2.9783248901367188e-05,
      "step": 31230
    },
    {
      "action_loss_avg": 0.002625470663770102,
      "epoch": 2.9783248901367188e-05,
      "step": 31230
    },
    {
      "epoch": 2.9783248901367188e-05,
      "step": 31230,
      "training_step_time": 8.590514421463013
    },
    {
      "epoch": 2.9784202575683595e-05,
      "model_forward_time": 1.7657287120819092,
      "step": 31231
    },
    {
      "epoch": 2.9784202575683595e-05,
      "step": 31231,
      "training_step_time": 8.807233810424805
    },
    {
      "epoch": 2.978515625e-05,
      "model_forward_time": 1.7249703407287598,
      "step": 31232
    },
    {
      "epoch": 2.978515625e-05,
      "step": 31232,
      "training_step_time": 7.946589231491089
    },
    {
      "epoch": 2.9786109924316407e-05,
      "model_forward_time": 1.7346000671386719,
      "step": 31233
    },
    {
      "epoch": 2.9786109924316407e-05,
      "step": 31233,
      "training_step_time": 7.94927191734314
    },
    {
      "epoch": 2.9787063598632814e-05,
      "model_forward_time": 1.7225391864776611,
      "step": 31234
    },
    {
      "epoch": 2.9787063598632814e-05,
      "step": 31234,
      "training_step_time": 7.623589754104614
    },
    {
      "epoch": 2.9788017272949218e-05,
      "model_forward_time": 1.7227413654327393,
      "step": 31235
    },
    {
      "epoch": 2.9788017272949218e-05,
      "step": 31235,
      "training_step_time": 6.913630962371826
    },
    {
      "epoch": 2.9788970947265626e-05,
      "model_forward_time": 1.7214787006378174,
      "step": 31236
    },
    {
      "epoch": 2.9788970947265626e-05,
      "step": 31236,
      "training_step_time": 6.939539432525635
    },
    {
      "epoch": 2.978992462158203e-05,
      "model_forward_time": 1.725534439086914,
      "step": 31237
    },
    {
      "epoch": 2.978992462158203e-05,
      "step": 31237,
      "training_step_time": 6.9298481941223145
    },
    {
      "epoch": 2.9790878295898437e-05,
      "model_forward_time": 1.7240705490112305,
      "step": 31238
    },
    {
      "epoch": 2.9790878295898437e-05,
      "step": 31238,
      "training_step_time": 7.514082193374634
    },
    {
      "epoch": 2.9791831970214844e-05,
      "model_forward_time": 1.7221429347991943,
      "step": 31239
    },
    {
      "epoch": 2.9791831970214844e-05,
      "step": 31239,
      "training_step_time": 9.54435396194458
    },
    {
      "epoch": 2.979278564453125e-05,
      "grad_norm": 0.11943486332893372,
      "learning_rate": 8.232883885594831e-06,
      "loss": 0.0603,
      "step": 31240
    },
    {
      "epoch": 2.979278564453125e-05,
      "model_forward_time": 1.71944260597229,
      "step": 31240
    },
    {
      "dynamics_loss_avg": 0.05022594928741455,
      "epoch": 2.979278564453125e-05,
      "step": 31240
    },
    {
      "action_loss_avg": 0.0014611933336709626,
      "epoch": 2.979278564453125e-05,
      "step": 31240
    },
    {
      "epoch": 2.979278564453125e-05,
      "step": 31240,
      "training_step_time": 9.635966300964355
    },
    {
      "epoch": 2.9793739318847656e-05,
      "model_forward_time": 1.7293510437011719,
      "step": 31241
    },
    {
      "epoch": 2.9793739318847656e-05,
      "step": 31241,
      "training_step_time": 7.921607732772827
    },
    {
      "epoch": 2.9794692993164063e-05,
      "model_forward_time": 1.7244172096252441,
      "step": 31242
    },
    {
      "epoch": 2.9794692993164063e-05,
      "step": 31242,
      "training_step_time": 8.449203729629517
    },
    {
      "epoch": 2.9795646667480467e-05,
      "model_forward_time": 1.7409074306488037,
      "step": 31243
    },
    {
      "epoch": 2.9795646667480467e-05,
      "step": 31243,
      "training_step_time": 7.65151047706604
    },
    {
      "epoch": 2.9796600341796875e-05,
      "model_forward_time": 1.7390618324279785,
      "step": 31244
    },
    {
      "epoch": 2.9796600341796875e-05,
      "step": 31244,
      "training_step_time": 7.24553918838501
    },
    {
      "epoch": 2.9797554016113282e-05,
      "model_forward_time": 1.7556941509246826,
      "step": 31245
    },
    {
      "epoch": 2.9797554016113282e-05,
      "step": 31245,
      "training_step_time": 6.883901596069336
    },
    {
      "epoch": 2.9798507690429686e-05,
      "model_forward_time": 1.7136294841766357,
      "step": 31246
    },
    {
      "epoch": 2.9798507690429686e-05,
      "step": 31246,
      "training_step_time": 6.87419581413269
    },
    {
      "epoch": 2.9799461364746094e-05,
      "model_forward_time": 1.7217206954956055,
      "step": 31247
    },
    {
      "epoch": 2.9799461364746094e-05,
      "step": 31247,
      "training_step_time": 7.461735963821411
    },
    {
      "epoch": 2.98004150390625e-05,
      "model_forward_time": 1.73187255859375,
      "step": 31248
    },
    {
      "epoch": 2.98004150390625e-05,
      "step": 31248,
      "training_step_time": 8.057827234268188
    },
    {
      "epoch": 2.9801368713378905e-05,
      "model_forward_time": 2.2621395587921143,
      "step": 31249
    },
    {
      "epoch": 2.9801368713378905e-05,
      "step": 31249,
      "training_step_time": 8.394523859024048
    },
    {
      "epoch": 2.9802322387695312e-05,
      "grad_norm": 0.17729660868644714,
      "learning_rate": 8.231622361185236e-06,
      "loss": 0.0656,
      "step": 31250
    },
    {
      "epoch": 2.9802322387695312e-05,
      "model_forward_time": 1.793532371520996,
      "step": 31250
    },
    {
      "dynamics_loss_avg": 0.059009181708097456,
      "epoch": 2.9802322387695312e-05,
      "step": 31250
    },
    {
      "action_loss_avg": 0.0024716398795135317,
      "epoch": 2.9802322387695312e-05,
      "step": 31250
    },
    {
      "epoch": 2.9802322387695312e-05,
      "step": 31250,
      "training_step_time": 10.779179334640503
    },
    {
      "epoch": 2.980327606201172e-05,
      "model_forward_time": 1.7233858108520508,
      "step": 31251
    },
    {
      "epoch": 2.980327606201172e-05,
      "step": 31251,
      "training_step_time": 7.690492630004883
    },
    {
      "epoch": 2.9804229736328124e-05,
      "model_forward_time": 1.7331087589263916,
      "step": 31252
    },
    {
      "epoch": 2.9804229736328124e-05,
      "step": 31252,
      "training_step_time": 7.673234462738037
    },
    {
      "epoch": 2.980518341064453e-05,
      "model_forward_time": 1.7268011569976807,
      "step": 31253
    },
    {
      "epoch": 2.980518341064453e-05,
      "step": 31253,
      "training_step_time": 7.078978061676025
    },
    {
      "epoch": 2.980613708496094e-05,
      "model_forward_time": 1.721120834350586,
      "step": 31254
    },
    {
      "epoch": 2.980613708496094e-05,
      "step": 31254,
      "training_step_time": 6.727947235107422
    },
    {
      "epoch": 2.9807090759277343e-05,
      "model_forward_time": 1.7271745204925537,
      "step": 31255
    },
    {
      "epoch": 2.9807090759277343e-05,
      "step": 31255,
      "training_step_time": 6.732969760894775
    },
    {
      "epoch": 2.980804443359375e-05,
      "model_forward_time": 1.7276687622070312,
      "step": 31256
    },
    {
      "epoch": 2.980804443359375e-05,
      "step": 31256,
      "training_step_time": 7.099742650985718
    },
    {
      "epoch": 2.9808998107910158e-05,
      "model_forward_time": 1.7373874187469482,
      "step": 31257
    },
    {
      "epoch": 2.9808998107910158e-05,
      "step": 31257,
      "training_step_time": 7.396597385406494
    },
    {
      "epoch": 2.980995178222656e-05,
      "model_forward_time": 2.253598213195801,
      "step": 31258
    },
    {
      "epoch": 2.980995178222656e-05,
      "step": 31258,
      "training_step_time": 8.728997707366943
    },
    {
      "epoch": 2.981090545654297e-05,
      "model_forward_time": 1.7966606616973877,
      "step": 31259
    },
    {
      "epoch": 2.981090545654297e-05,
      "step": 31259,
      "training_step_time": 9.994566440582275
    },
    {
      "epoch": 2.9811859130859376e-05,
      "grad_norm": 0.16747815907001495,
      "learning_rate": 8.230360483370277e-06,
      "loss": 0.0595,
      "step": 31260
    },
    {
      "epoch": 2.9811859130859376e-05,
      "model_forward_time": 1.722830057144165,
      "step": 31260
    },
    {
      "dynamics_loss_avg": 0.04669608399271965,
      "epoch": 2.9811859130859376e-05,
      "step": 31260
    },
    {
      "action_loss_avg": 0.0015342554746894167,
      "epoch": 2.9811859130859376e-05,
      "step": 31260
    },
    {
      "epoch": 2.9811859130859376e-05,
      "step": 31260,
      "training_step_time": 7.786264181137085
    },
    {
      "epoch": 2.981281280517578e-05,
      "model_forward_time": 1.7290680408477783,
      "step": 31261
    },
    {
      "epoch": 2.981281280517578e-05,
      "step": 31261,
      "training_step_time": 7.797788619995117
    },
    {
      "epoch": 2.9813766479492188e-05,
      "model_forward_time": 1.7433602809906006,
      "step": 31262
    },
    {
      "epoch": 2.9813766479492188e-05,
      "step": 31262,
      "training_step_time": 7.472665071487427
    },
    {
      "epoch": 2.9814720153808595e-05,
      "model_forward_time": 1.7257096767425537,
      "step": 31263
    },
    {
      "epoch": 2.9814720153808595e-05,
      "step": 31263,
      "training_step_time": 6.729954957962036
    },
    {
      "epoch": 2.9815673828125e-05,
      "model_forward_time": 1.7282378673553467,
      "step": 31264
    },
    {
      "epoch": 2.9815673828125e-05,
      "step": 31264,
      "training_step_time": 6.707774877548218
    },
    {
      "epoch": 2.9816627502441407e-05,
      "model_forward_time": 1.7238848209381104,
      "step": 31265
    },
    {
      "epoch": 2.9816627502441407e-05,
      "step": 31265,
      "training_step_time": 6.767740249633789
    },
    {
      "epoch": 2.981758117675781e-05,
      "model_forward_time": 1.732919692993164,
      "step": 31266
    },
    {
      "epoch": 2.981758117675781e-05,
      "step": 31266,
      "training_step_time": 7.3875792026519775
    },
    {
      "epoch": 2.9818534851074218e-05,
      "model_forward_time": 2.2542293071746826,
      "step": 31267
    },
    {
      "epoch": 2.9818534851074218e-05,
      "step": 31267,
      "training_step_time": 7.682776212692261
    },
    {
      "epoch": 2.9819488525390626e-05,
      "model_forward_time": 1.7279393672943115,
      "step": 31268
    },
    {
      "epoch": 2.9819488525390626e-05,
      "step": 31268,
      "training_step_time": 7.7477312088012695
    },
    {
      "epoch": 2.982044219970703e-05,
      "model_forward_time": 1.7252366542816162,
      "step": 31269
    },
    {
      "epoch": 2.982044219970703e-05,
      "step": 31269,
      "training_step_time": 8.654965162277222
    },
    {
      "epoch": 2.9821395874023437e-05,
      "grad_norm": 0.14646963775157928,
      "learning_rate": 8.229098252287954e-06,
      "loss": 0.0647,
      "step": 31270
    },
    {
      "epoch": 2.9821395874023437e-05,
      "model_forward_time": 1.7179982662200928,
      "step": 31270
    },
    {
      "dynamics_loss_avg": 0.06680058687925339,
      "epoch": 2.9821395874023437e-05,
      "step": 31270
    },
    {
      "action_loss_avg": 0.002333093079505488,
      "epoch": 2.9821395874023437e-05,
      "step": 31270
    },
    {
      "epoch": 2.9821395874023437e-05,
      "step": 31270,
      "training_step_time": 7.677867889404297
    },
    {
      "epoch": 2.9822349548339845e-05,
      "model_forward_time": 1.7260475158691406,
      "step": 31271
    },
    {
      "epoch": 2.9822349548339845e-05,
      "step": 31271,
      "training_step_time": 7.453241586685181
    },
    {
      "epoch": 2.982330322265625e-05,
      "model_forward_time": 1.7340586185455322,
      "step": 31272
    },
    {
      "epoch": 2.982330322265625e-05,
      "step": 31272,
      "training_step_time": 6.707999229431152
    },
    {
      "epoch": 2.9824256896972656e-05,
      "model_forward_time": 1.728907585144043,
      "step": 31273
    },
    {
      "epoch": 2.9824256896972656e-05,
      "step": 31273,
      "training_step_time": 6.7648680210113525
    },
    {
      "epoch": 2.9825210571289063e-05,
      "model_forward_time": 1.7240703105926514,
      "step": 31274
    },
    {
      "epoch": 2.9825210571289063e-05,
      "step": 31274,
      "training_step_time": 6.74059534072876
    },
    {
      "epoch": 2.9826164245605467e-05,
      "model_forward_time": 1.7213609218597412,
      "step": 31275
    },
    {
      "epoch": 2.9826164245605467e-05,
      "step": 31275,
      "training_step_time": 7.375860691070557
    },
    {
      "epoch": 2.9827117919921875e-05,
      "model_forward_time": 2.247614622116089,
      "step": 31276
    },
    {
      "epoch": 2.9827117919921875e-05,
      "step": 31276,
      "training_step_time": 8.727710008621216
    },
    {
      "epoch": 2.9828071594238282e-05,
      "model_forward_time": 1.7277765274047852,
      "step": 31277
    },
    {
      "epoch": 2.9828071594238282e-05,
      "step": 31277,
      "training_step_time": 7.759247541427612
    },
    {
      "epoch": 2.9829025268554686e-05,
      "model_forward_time": 1.7220408916473389,
      "step": 31278
    },
    {
      "epoch": 2.9829025268554686e-05,
      "step": 31278,
      "training_step_time": 8.254340648651123
    },
    {
      "epoch": 2.9829978942871094e-05,
      "model_forward_time": 1.7326915264129639,
      "step": 31279
    },
    {
      "epoch": 2.9829978942871094e-05,
      "step": 31279,
      "training_step_time": 7.646401643753052
    },
    {
      "epoch": 2.98309326171875e-05,
      "grad_norm": 0.1595175713300705,
      "learning_rate": 8.2278356680763e-06,
      "loss": 0.0579,
      "step": 31280
    },
    {
      "epoch": 2.98309326171875e-05,
      "model_forward_time": 1.7199139595031738,
      "step": 31280
    },
    {
      "dynamics_loss_avg": 0.05842998940497637,
      "epoch": 2.98309326171875e-05,
      "step": 31280
    },
    {
      "action_loss_avg": 0.002739274478517473,
      "epoch": 2.98309326171875e-05,
      "step": 31280
    },
    {
      "epoch": 2.98309326171875e-05,
      "step": 31280,
      "training_step_time": 7.520842552185059
    },
    {
      "epoch": 2.9831886291503905e-05,
      "model_forward_time": 1.7285852432250977,
      "step": 31281
    },
    {
      "epoch": 2.9831886291503905e-05,
      "step": 31281,
      "training_step_time": 7.0488996505737305
    },
    {
      "epoch": 2.9832839965820313e-05,
      "model_forward_time": 1.7226097583770752,
      "step": 31282
    },
    {
      "epoch": 2.9832839965820313e-05,
      "step": 31282,
      "training_step_time": 6.710580825805664
    },
    {
      "epoch": 2.983379364013672e-05,
      "model_forward_time": 1.7369182109832764,
      "step": 31283
    },
    {
      "epoch": 2.983379364013672e-05,
      "step": 31283,
      "training_step_time": 6.682512044906616
    },
    {
      "epoch": 2.9834747314453124e-05,
      "model_forward_time": 1.7262656688690186,
      "step": 31284
    },
    {
      "epoch": 2.9834747314453124e-05,
      "step": 31284,
      "training_step_time": 7.36583948135376
    },
    {
      "epoch": 2.983570098876953e-05,
      "model_forward_time": 1.7231509685516357,
      "step": 31285
    },
    {
      "epoch": 2.983570098876953e-05,
      "step": 31285,
      "training_step_time": 9.14176893234253
    },
    {
      "epoch": 2.983665466308594e-05,
      "model_forward_time": 1.7317378520965576,
      "step": 31286
    },
    {
      "epoch": 2.983665466308594e-05,
      "step": 31286,
      "training_step_time": 9.147430896759033
    },
    {
      "epoch": 2.9837608337402343e-05,
      "model_forward_time": 1.7837224006652832,
      "step": 31287
    },
    {
      "epoch": 2.9837608337402343e-05,
      "step": 31287,
      "training_step_time": 10.301599264144897
    },
    {
      "epoch": 2.983856201171875e-05,
      "model_forward_time": 1.7201006412506104,
      "step": 31288
    },
    {
      "epoch": 2.983856201171875e-05,
      "step": 31288,
      "training_step_time": 7.647719621658325
    },
    {
      "epoch": 2.9839515686035158e-05,
      "model_forward_time": 1.721968650817871,
      "step": 31289
    },
    {
      "epoch": 2.9839515686035158e-05,
      "step": 31289,
      "training_step_time": 7.5272674560546875
    },
    {
      "epoch": 2.984046936035156e-05,
      "grad_norm": 0.18210655450820923,
      "learning_rate": 8.22657273087339e-06,
      "loss": 0.0624,
      "step": 31290
    },
    {
      "epoch": 2.984046936035156e-05,
      "model_forward_time": 1.7203192710876465,
      "step": 31290
    },
    {
      "dynamics_loss_avg": 0.0489455783739686,
      "epoch": 2.984046936035156e-05,
      "step": 31290
    },
    {
      "action_loss_avg": 0.0017107309657149016,
      "epoch": 2.984046936035156e-05,
      "step": 31290
    },
    {
      "epoch": 2.984046936035156e-05,
      "step": 31290,
      "training_step_time": 7.0623016357421875
    },
    {
      "epoch": 2.984142303466797e-05,
      "model_forward_time": 1.7303447723388672,
      "step": 31291
    },
    {
      "epoch": 2.984142303466797e-05,
      "step": 31291,
      "training_step_time": 6.695928573608398
    },
    {
      "epoch": 2.9842376708984377e-05,
      "model_forward_time": 1.7219834327697754,
      "step": 31292
    },
    {
      "epoch": 2.9842376708984377e-05,
      "step": 31292,
      "training_step_time": 6.741800785064697
    },
    {
      "epoch": 2.984333038330078e-05,
      "model_forward_time": 1.7172915935516357,
      "step": 31293
    },
    {
      "epoch": 2.984333038330078e-05,
      "step": 31293,
      "training_step_time": 7.459827661514282
    },
    {
      "epoch": 2.9844284057617188e-05,
      "model_forward_time": 1.7159969806671143,
      "step": 31294
    },
    {
      "epoch": 2.9844284057617188e-05,
      "step": 31294,
      "training_step_time": 8.667986154556274
    },
    {
      "epoch": 2.9845237731933595e-05,
      "model_forward_time": 2.253641366958618,
      "step": 31295
    },
    {
      "epoch": 2.9845237731933595e-05,
      "step": 31295,
      "training_step_time": 8.351347208023071
    },
    {
      "epoch": 2.984619140625e-05,
      "model_forward_time": 1.752582311630249,
      "step": 31296
    },
    {
      "epoch": 2.984619140625e-05,
      "step": 31296,
      "training_step_time": 7.670441627502441
    },
    {
      "epoch": 2.9847145080566407e-05,
      "model_forward_time": 1.7335684299468994,
      "step": 31297
    },
    {
      "epoch": 2.9847145080566407e-05,
      "step": 31297,
      "training_step_time": 8.644215822219849
    },
    {
      "epoch": 2.984809875488281e-05,
      "model_forward_time": 1.7175796031951904,
      "step": 31298
    },
    {
      "epoch": 2.984809875488281e-05,
      "step": 31298,
      "training_step_time": 8.438543319702148
    },
    {
      "epoch": 2.9849052429199218e-05,
      "model_forward_time": 1.7093994617462158,
      "step": 31299
    },
    {
      "epoch": 2.9849052429199218e-05,
      "step": 31299,
      "training_step_time": 7.286622047424316
    },
    {
      "epoch": 2.9850006103515626e-05,
      "grad_norm": 0.16843971610069275,
      "learning_rate": 8.225309440817337e-06,
      "loss": 0.0608,
      "step": 31300
    },
    {
      "epoch": 2.9850006103515626e-05,
      "model_forward_time": 1.719954490661621,
      "step": 31300
    },
    {
      "dynamics_loss_avg": 0.056148717552423476,
      "epoch": 2.9850006103515626e-05,
      "step": 31300
    },
    {
      "action_loss_avg": 0.0026324101781938225,
      "epoch": 2.9850006103515626e-05,
      "step": 31300
    },
    {
      "epoch": 2.9850006103515626e-05,
      "step": 31300,
      "training_step_time": 6.657868146896362
    },
    {
      "epoch": 2.985095977783203e-05,
      "model_forward_time": 1.7217872142791748,
      "step": 31301
    },
    {
      "epoch": 2.985095977783203e-05,
      "step": 31301,
      "training_step_time": 6.707834482192993
    },
    {
      "epoch": 2.9851913452148437e-05,
      "model_forward_time": 1.7239227294921875,
      "step": 31302
    },
    {
      "epoch": 2.9851913452148437e-05,
      "step": 31302,
      "training_step_time": 8.043199300765991
    },
    {
      "epoch": 2.9852867126464845e-05,
      "model_forward_time": 1.7247216701507568,
      "step": 31303
    },
    {
      "epoch": 2.9852867126464845e-05,
      "step": 31303,
      "training_step_time": 8.44250750541687
    },
    {
      "epoch": 2.985382080078125e-05,
      "model_forward_time": 2.2582106590270996,
      "step": 31304
    },
    {
      "epoch": 2.985382080078125e-05,
      "step": 31304,
      "training_step_time": 9.324798583984375
    },
    {
      "epoch": 2.9854774475097656e-05,
      "model_forward_time": 1.737206220626831,
      "step": 31305
    },
    {
      "epoch": 2.9854774475097656e-05,
      "step": 31305,
      "training_step_time": 7.729318857192993
    },
    {
      "epoch": 2.9855728149414063e-05,
      "model_forward_time": 1.7153716087341309,
      "step": 31306
    },
    {
      "epoch": 2.9855728149414063e-05,
      "step": 31306,
      "training_step_time": 7.685394048690796
    },
    {
      "epoch": 2.9856681823730467e-05,
      "model_forward_time": 1.714280366897583,
      "step": 31307
    },
    {
      "epoch": 2.9856681823730467e-05,
      "step": 31307,
      "training_step_time": 7.651912689208984
    },
    {
      "epoch": 2.9857635498046875e-05,
      "model_forward_time": 1.714730978012085,
      "step": 31308
    },
    {
      "epoch": 2.9857635498046875e-05,
      "step": 31308,
      "training_step_time": 7.377161979675293
    },
    {
      "epoch": 2.9858589172363282e-05,
      "model_forward_time": 1.726470947265625,
      "step": 31309
    },
    {
      "epoch": 2.9858589172363282e-05,
      "step": 31309,
      "training_step_time": 7.111096382141113
    },
    {
      "epoch": 2.9859542846679686e-05,
      "grad_norm": 0.17947357892990112,
      "learning_rate": 8.224045798046294e-06,
      "loss": 0.0615,
      "step": 31310
    },
    {
      "epoch": 2.9859542846679686e-05,
      "model_forward_time": 1.7187070846557617,
      "step": 31310
    },
    {
      "dynamics_loss_avg": 0.047638097777962685,
      "epoch": 2.9859542846679686e-05,
      "step": 31310
    },
    {
      "action_loss_avg": 0.0021862175432033836,
      "epoch": 2.9859542846679686e-05,
      "step": 31310
    },
    {
      "epoch": 2.9859542846679686e-05,
      "step": 31310,
      "training_step_time": 6.73581862449646
    },
    {
      "epoch": 2.9860496520996094e-05,
      "model_forward_time": 1.7174766063690186,
      "step": 31311
    },
    {
      "epoch": 2.9860496520996094e-05,
      "step": 31311,
      "training_step_time": 7.2924275398254395
    },
    {
      "epoch": 2.98614501953125e-05,
      "model_forward_time": 1.7237789630889893,
      "step": 31312
    },
    {
      "epoch": 2.98614501953125e-05,
      "step": 31312,
      "training_step_time": 7.688182830810547
    },
    {
      "epoch": 2.9862403869628905e-05,
      "model_forward_time": 2.254033088684082,
      "step": 31313
    },
    {
      "epoch": 2.9862403869628905e-05,
      "step": 31313,
      "training_step_time": 8.476989507675171
    },
    {
      "epoch": 2.9863357543945313e-05,
      "model_forward_time": 1.771958589553833,
      "step": 31314
    },
    {
      "epoch": 2.9863357543945313e-05,
      "step": 31314,
      "training_step_time": 9.685755491256714
    },
    {
      "epoch": 2.986431121826172e-05,
      "model_forward_time": 1.7921879291534424,
      "step": 31315
    },
    {
      "epoch": 2.986431121826172e-05,
      "step": 31315,
      "training_step_time": 9.861333847045898
    },
    {
      "epoch": 2.9865264892578124e-05,
      "model_forward_time": 1.7203779220581055,
      "step": 31316
    },
    {
      "epoch": 2.9865264892578124e-05,
      "step": 31316,
      "training_step_time": 8.322141885757446
    },
    {
      "epoch": 2.986621856689453e-05,
      "model_forward_time": 1.7154312133789062,
      "step": 31317
    },
    {
      "epoch": 2.986621856689453e-05,
      "step": 31317,
      "training_step_time": 7.4111528396606445
    },
    {
      "epoch": 2.986717224121094e-05,
      "model_forward_time": 1.7252912521362305,
      "step": 31318
    },
    {
      "epoch": 2.986717224121094e-05,
      "step": 31318,
      "training_step_time": 6.690919399261475
    },
    {
      "epoch": 2.9868125915527343e-05,
      "model_forward_time": 1.7250564098358154,
      "step": 31319
    },
    {
      "epoch": 2.9868125915527343e-05,
      "step": 31319,
      "training_step_time": 6.739567995071411
    },
    {
      "epoch": 2.986907958984375e-05,
      "grad_norm": 0.18287402391433716,
      "learning_rate": 8.22278180269845e-06,
      "loss": 0.0649,
      "step": 31320
    },
    {
      "epoch": 2.986907958984375e-05,
      "model_forward_time": 1.7313013076782227,
      "step": 31320
    },
    {
      "dynamics_loss_avg": 0.05662708543241024,
      "epoch": 2.986907958984375e-05,
      "step": 31320
    },
    {
      "action_loss_avg": 0.001824531829333864,
      "epoch": 2.986907958984375e-05,
      "step": 31320
    },
    {
      "epoch": 2.986907958984375e-05,
      "step": 31320,
      "training_step_time": 7.320944309234619
    },
    {
      "epoch": 2.9870033264160158e-05,
      "model_forward_time": 1.7252185344696045,
      "step": 31321
    },
    {
      "epoch": 2.9870033264160158e-05,
      "step": 31321,
      "training_step_time": 7.763928651809692
    },
    {
      "epoch": 2.9870986938476562e-05,
      "model_forward_time": 2.256770372390747,
      "step": 31322
    },
    {
      "epoch": 2.9870986938476562e-05,
      "step": 31322,
      "training_step_time": 9.43947148323059
    },
    {
      "epoch": 2.987194061279297e-05,
      "model_forward_time": 1.7282865047454834,
      "step": 31323
    },
    {
      "epoch": 2.987194061279297e-05,
      "step": 31323,
      "training_step_time": 9.688643455505371
    },
    {
      "epoch": 2.9872894287109377e-05,
      "model_forward_time": 1.7276082038879395,
      "step": 31324
    },
    {
      "epoch": 2.9872894287109377e-05,
      "step": 31324,
      "training_step_time": 8.28327989578247
    },
    {
      "epoch": 2.987384796142578e-05,
      "model_forward_time": 1.793288230895996,
      "step": 31325
    },
    {
      "epoch": 2.987384796142578e-05,
      "step": 31325,
      "training_step_time": 7.599381923675537
    },
    {
      "epoch": 2.9874801635742188e-05,
      "model_forward_time": 1.7170629501342773,
      "step": 31326
    },
    {
      "epoch": 2.9874801635742188e-05,
      "step": 31326,
      "training_step_time": 7.411924123764038
    },
    {
      "epoch": 2.9875755310058595e-05,
      "model_forward_time": 1.723473072052002,
      "step": 31327
    },
    {
      "epoch": 2.9875755310058595e-05,
      "step": 31327,
      "training_step_time": 7.272143840789795
    },
    {
      "epoch": 2.9876708984375e-05,
      "model_forward_time": 1.7148637771606445,
      "step": 31328
    },
    {
      "epoch": 2.9876708984375e-05,
      "step": 31328,
      "training_step_time": 6.683220386505127
    },
    {
      "epoch": 2.9877662658691407e-05,
      "model_forward_time": 1.7295467853546143,
      "step": 31329
    },
    {
      "epoch": 2.9877662658691407e-05,
      "step": 31329,
      "training_step_time": 7.212870836257935
    },
    {
      "epoch": 2.987861633300781e-05,
      "grad_norm": 0.16945238411426544,
      "learning_rate": 8.221517454912032e-06,
      "loss": 0.0655,
      "step": 31330
    },
    {
      "epoch": 2.987861633300781e-05,
      "model_forward_time": 1.713310718536377,
      "step": 31330
    },
    {
      "dynamics_loss_avg": 0.05951168965548277,
      "epoch": 2.987861633300781e-05,
      "step": 31330
    },
    {
      "action_loss_avg": 0.002495614840881899,
      "epoch": 2.987861633300781e-05,
      "step": 31330
    },
    {
      "epoch": 2.987861633300781e-05,
      "step": 31330,
      "training_step_time": 8.27649450302124
    },
    {
      "epoch": 2.9879570007324218e-05,
      "model_forward_time": 2.2556169033050537,
      "step": 31331
    },
    {
      "epoch": 2.9879570007324218e-05,
      "step": 31331,
      "training_step_time": 8.394170761108398
    },
    {
      "epoch": 2.9880523681640626e-05,
      "model_forward_time": 1.7211768627166748,
      "step": 31332
    },
    {
      "epoch": 2.9880523681640626e-05,
      "step": 31332,
      "training_step_time": 8.278948307037354
    },
    {
      "epoch": 2.988147735595703e-05,
      "model_forward_time": 1.7371551990509033,
      "step": 31333
    },
    {
      "epoch": 2.988147735595703e-05,
      "step": 31333,
      "training_step_time": 8.589750528335571
    },
    {
      "epoch": 2.9882431030273437e-05,
      "model_forward_time": 1.7802813053131104,
      "step": 31334
    },
    {
      "epoch": 2.9882431030273437e-05,
      "step": 31334,
      "training_step_time": 7.6208930015563965
    },
    {
      "epoch": 2.9883384704589845e-05,
      "model_forward_time": 1.7300660610198975,
      "step": 31335
    },
    {
      "epoch": 2.9883384704589845e-05,
      "step": 31335,
      "training_step_time": 7.465502738952637
    },
    {
      "epoch": 2.988433837890625e-05,
      "model_forward_time": 1.737705945968628,
      "step": 31336
    },
    {
      "epoch": 2.988433837890625e-05,
      "step": 31336,
      "training_step_time": 7.267215728759766
    },
    {
      "epoch": 2.9885292053222656e-05,
      "model_forward_time": 1.737783432006836,
      "step": 31337
    },
    {
      "epoch": 2.9885292053222656e-05,
      "step": 31337,
      "training_step_time": 6.717331409454346
    },
    {
      "epoch": 2.9886245727539063e-05,
      "model_forward_time": 1.7384998798370361,
      "step": 31338
    },
    {
      "epoch": 2.9886245727539063e-05,
      "step": 31338,
      "training_step_time": 7.361774682998657
    },
    {
      "epoch": 2.9887199401855467e-05,
      "model_forward_time": 1.7330262660980225,
      "step": 31339
    },
    {
      "epoch": 2.9887199401855467e-05,
      "step": 31339,
      "training_step_time": 7.739444017410278
    },
    {
      "epoch": 2.9888153076171875e-05,
      "grad_norm": 0.15710687637329102,
      "learning_rate": 8.220252754825308e-06,
      "loss": 0.0653,
      "step": 31340
    },
    {
      "epoch": 2.9888153076171875e-05,
      "model_forward_time": 1.740901231765747,
      "step": 31340
    },
    {
      "dynamics_loss_avg": 0.055876602232456204,
      "epoch": 2.9888153076171875e-05,
      "step": 31340
    },
    {
      "action_loss_avg": 0.0023780359711963682,
      "epoch": 2.9888153076171875e-05,
      "step": 31340
    },
    {
      "epoch": 2.9888153076171875e-05,
      "step": 31340,
      "training_step_time": 8.090336561203003
    },
    {
      "epoch": 2.9889106750488282e-05,
      "model_forward_time": 2.261756420135498,
      "step": 31341
    },
    {
      "epoch": 2.9889106750488282e-05,
      "step": 31341,
      "training_step_time": 8.826237678527832
    },
    {
      "epoch": 2.9890060424804686e-05,
      "model_forward_time": 1.7449800968170166,
      "step": 31342
    },
    {
      "epoch": 2.9890060424804686e-05,
      "step": 31342,
      "training_step_time": 8.723265409469604
    },
    {
      "epoch": 2.9891014099121094e-05,
      "model_forward_time": 1.7606606483459473,
      "step": 31343
    },
    {
      "epoch": 2.9891014099121094e-05,
      "step": 31343,
      "training_step_time": 8.67033052444458
    },
    {
      "epoch": 2.98919677734375e-05,
      "model_forward_time": 1.742100477218628,
      "step": 31344
    },
    {
      "epoch": 2.98919677734375e-05,
      "step": 31344,
      "training_step_time": 8.320826530456543
    },
    {
      "epoch": 2.9892921447753905e-05,
      "model_forward_time": 1.735966682434082,
      "step": 31345
    },
    {
      "epoch": 2.9892921447753905e-05,
      "step": 31345,
      "training_step_time": 7.47080659866333
    },
    {
      "epoch": 2.9893875122070313e-05,
      "model_forward_time": 1.7320811748504639,
      "step": 31346
    },
    {
      "epoch": 2.9893875122070313e-05,
      "step": 31346,
      "training_step_time": 6.766605615615845
    },
    {
      "epoch": 2.989482879638672e-05,
      "model_forward_time": 1.7345707416534424,
      "step": 31347
    },
    {
      "epoch": 2.989482879638672e-05,
      "step": 31347,
      "training_step_time": 7.339049816131592
    },
    {
      "epoch": 2.9895782470703124e-05,
      "model_forward_time": 1.7369325160980225,
      "step": 31348
    },
    {
      "epoch": 2.9895782470703124e-05,
      "step": 31348,
      "training_step_time": 7.732176780700684
    },
    {
      "epoch": 2.989673614501953e-05,
      "model_forward_time": 1.734299659729004,
      "step": 31349
    },
    {
      "epoch": 2.989673614501953e-05,
      "step": 31349,
      "training_step_time": 7.484485864639282
    },
    {
      "epoch": 2.989768981933594e-05,
      "grad_norm": 0.19061994552612305,
      "learning_rate": 8.218987702576586e-06,
      "loss": 0.069,
      "step": 31350
    },
    {
      "epoch": 2.989768981933594e-05,
      "model_forward_time": 2.2600574493408203,
      "step": 31350
    },
    {
      "dynamics_loss_avg": 0.07289712317287922,
      "epoch": 2.989768981933594e-05,
      "step": 31350
    },
    {
      "action_loss_avg": 0.002639966021524742,
      "epoch": 2.989768981933594e-05,
      "step": 31350
    },
    {
      "epoch": 2.989768981933594e-05,
      "step": 31350,
      "training_step_time": 7.779776096343994
    },
    {
      "epoch": 2.9898643493652343e-05,
      "model_forward_time": 1.7358620166778564,
      "step": 31351
    },
    {
      "epoch": 2.9898643493652343e-05,
      "step": 31351,
      "training_step_time": 7.724858999252319
    },
    {
      "epoch": 2.989959716796875e-05,
      "model_forward_time": 1.7606384754180908,
      "step": 31352
    },
    {
      "epoch": 2.989959716796875e-05,
      "step": 31352,
      "training_step_time": 7.660315275192261
    },
    {
      "epoch": 2.9900550842285158e-05,
      "model_forward_time": 1.7328464984893799,
      "step": 31353
    },
    {
      "epoch": 2.9900550842285158e-05,
      "step": 31353,
      "training_step_time": 8.34722900390625
    },
    {
      "epoch": 2.9901504516601562e-05,
      "model_forward_time": 1.7486004829406738,
      "step": 31354
    },
    {
      "epoch": 2.9901504516601562e-05,
      "step": 31354,
      "training_step_time": 7.479619979858398
    },
    {
      "epoch": 2.990245819091797e-05,
      "model_forward_time": 1.7370953559875488,
      "step": 31355
    },
    {
      "epoch": 2.990245819091797e-05,
      "step": 31355,
      "training_step_time": 7.120856761932373
    },
    {
      "epoch": 2.9903411865234377e-05,
      "model_forward_time": 1.7347631454467773,
      "step": 31356
    },
    {
      "epoch": 2.9903411865234377e-05,
      "step": 31356,
      "training_step_time": 7.248311281204224
    },
    {
      "epoch": 2.990436553955078e-05,
      "model_forward_time": 1.7427539825439453,
      "step": 31357
    },
    {
      "epoch": 2.990436553955078e-05,
      "step": 31357,
      "training_step_time": 7.576189756393433
    },
    {
      "epoch": 2.9905319213867188e-05,
      "model_forward_time": 1.7354710102081299,
      "step": 31358
    },
    {
      "epoch": 2.9905319213867188e-05,
      "step": 31358,
      "training_step_time": 8.52177119255066
    },
    {
      "epoch": 2.9906272888183595e-05,
      "model_forward_time": 2.268887758255005,
      "step": 31359
    },
    {
      "epoch": 2.9906272888183595e-05,
      "step": 31359,
      "training_step_time": 7.673932075500488
    },
    {
      "epoch": 2.99072265625e-05,
      "grad_norm": 0.21774376928806305,
      "learning_rate": 8.217722298304209e-06,
      "loss": 0.0643,
      "step": 31360
    },
    {
      "epoch": 2.99072265625e-05,
      "model_forward_time": 1.739234209060669,
      "step": 31360
    },
    {
      "dynamics_loss_avg": 0.05206774491816759,
      "epoch": 2.99072265625e-05,
      "step": 31360
    },
    {
      "action_loss_avg": 0.0016988797404337674,
      "epoch": 2.99072265625e-05,
      "step": 31360
    },
    {
      "epoch": 2.99072265625e-05,
      "step": 31360,
      "training_step_time": 8.862818479537964
    },
    {
      "epoch": 2.9908180236816407e-05,
      "model_forward_time": 1.7383038997650146,
      "step": 31361
    },
    {
      "epoch": 2.9908180236816407e-05,
      "step": 31361,
      "training_step_time": 8.696545600891113
    },
    {
      "epoch": 2.990913391113281e-05,
      "model_forward_time": 1.8112473487854004,
      "step": 31362
    },
    {
      "epoch": 2.990913391113281e-05,
      "step": 31362,
      "training_step_time": 7.679900646209717
    },
    {
      "epoch": 2.991008758544922e-05,
      "model_forward_time": 1.7324776649475098,
      "step": 31363
    },
    {
      "epoch": 2.991008758544922e-05,
      "step": 31363,
      "training_step_time": 7.626674652099609
    },
    {
      "epoch": 2.9911041259765626e-05,
      "model_forward_time": 1.7307839393615723,
      "step": 31364
    },
    {
      "epoch": 2.9911041259765626e-05,
      "step": 31364,
      "training_step_time": 7.088347911834717
    },
    {
      "epoch": 2.991199493408203e-05,
      "model_forward_time": 1.7429449558258057,
      "step": 31365
    },
    {
      "epoch": 2.991199493408203e-05,
      "step": 31365,
      "training_step_time": 6.777266502380371
    },
    {
      "epoch": 2.9912948608398437e-05,
      "model_forward_time": 1.730992317199707,
      "step": 31366
    },
    {
      "epoch": 2.9912948608398437e-05,
      "step": 31366,
      "training_step_time": 7.518633604049683
    },
    {
      "epoch": 2.9913902282714845e-05,
      "model_forward_time": 1.7362010478973389,
      "step": 31367
    },
    {
      "epoch": 2.9913902282714845e-05,
      "step": 31367,
      "training_step_time": 7.84842324256897
    },
    {
      "epoch": 2.991485595703125e-05,
      "model_forward_time": 2.2886762619018555,
      "step": 31368
    },
    {
      "epoch": 2.991485595703125e-05,
      "step": 31368,
      "training_step_time": 8.381015062332153
    },
    {
      "epoch": 2.9915809631347656e-05,
      "model_forward_time": 1.7276864051818848,
      "step": 31369
    },
    {
      "epoch": 2.9915809631347656e-05,
      "step": 31369,
      "training_step_time": 8.532208442687988
    },
    {
      "epoch": 2.9916763305664063e-05,
      "grad_norm": 0.1384572684764862,
      "learning_rate": 8.216456542146556e-06,
      "loss": 0.0663,
      "step": 31370
    },
    {
      "epoch": 2.9916763305664063e-05,
      "model_forward_time": 1.7387197017669678,
      "step": 31370
    },
    {
      "dynamics_loss_avg": 0.06435731761157512,
      "epoch": 2.9916763305664063e-05,
      "step": 31370
    },
    {
      "action_loss_avg": 0.0023212425061501563,
      "epoch": 2.9916763305664063e-05,
      "step": 31370
    },
    {
      "epoch": 2.9916763305664063e-05,
      "step": 31370,
      "training_step_time": 10.692155599594116
    },
    {
      "epoch": 2.9917716979980467e-05,
      "model_forward_time": 1.7171523571014404,
      "step": 31371
    },
    {
      "epoch": 2.9917716979980467e-05,
      "step": 31371,
      "training_step_time": 8.662437438964844
    },
    {
      "epoch": 2.9918670654296875e-05,
      "model_forward_time": 1.7211194038391113,
      "step": 31372
    },
    {
      "epoch": 2.9918670654296875e-05,
      "step": 31372,
      "training_step_time": 7.584178686141968
    },
    {
      "epoch": 2.9919624328613282e-05,
      "model_forward_time": 1.7212395668029785,
      "step": 31373
    },
    {
      "epoch": 2.9919624328613282e-05,
      "step": 31373,
      "training_step_time": 7.06789493560791
    },
    {
      "epoch": 2.9920578002929686e-05,
      "model_forward_time": 1.7148733139038086,
      "step": 31374
    },
    {
      "epoch": 2.9920578002929686e-05,
      "step": 31374,
      "training_step_time": 6.779489755630493
    },
    {
      "epoch": 2.9921531677246094e-05,
      "model_forward_time": 1.717043161392212,
      "step": 31375
    },
    {
      "epoch": 2.9921531677246094e-05,
      "step": 31375,
      "training_step_time": 7.613078594207764
    },
    {
      "epoch": 2.99224853515625e-05,
      "model_forward_time": 1.716432809829712,
      "step": 31376
    },
    {
      "epoch": 2.99224853515625e-05,
      "step": 31376,
      "training_step_time": 7.804906606674194
    },
    {
      "epoch": 2.9923439025878905e-05,
      "model_forward_time": 2.8176565170288086,
      "step": 31377
    },
    {
      "epoch": 2.9923439025878905e-05,
      "step": 31377,
      "training_step_time": 7.667543888092041
    },
    {
      "epoch": 2.9924392700195313e-05,
      "model_forward_time": 1.7166755199432373,
      "step": 31378
    },
    {
      "epoch": 2.9924392700195313e-05,
      "step": 31378,
      "training_step_time": 9.69045615196228
    },
    {
      "epoch": 2.992534637451172e-05,
      "model_forward_time": 1.7465784549713135,
      "step": 31379
    },
    {
      "epoch": 2.992534637451172e-05,
      "step": 31379,
      "training_step_time": 8.387965679168701
    },
    {
      "epoch": 2.9926300048828124e-05,
      "grad_norm": 0.1796121895313263,
      "learning_rate": 8.215190434242055e-06,
      "loss": 0.0641,
      "step": 31380
    },
    {
      "epoch": 2.9926300048828124e-05,
      "model_forward_time": 1.3494653701782227,
      "step": 31380
    },
    {
      "dynamics_loss_avg": 0.0598106911405921,
      "epoch": 2.9926300048828124e-05,
      "step": 31380
    },
    {
      "action_loss_avg": 0.0016597742564044894,
      "epoch": 2.9926300048828124e-05,
      "step": 31380
    },
    {
      "epoch": 2.9926300048828124e-05,
      "step": 31380,
      "training_step_time": 8.438342094421387
    },
    {
      "epoch": 2.992725372314453e-05,
      "model_forward_time": 1.7209744453430176,
      "step": 31381
    },
    {
      "epoch": 2.992725372314453e-05,
      "step": 31381,
      "training_step_time": 7.648769855499268
    },
    {
      "epoch": 2.992820739746094e-05,
      "model_forward_time": 1.7197070121765137,
      "step": 31382
    },
    {
      "epoch": 2.992820739746094e-05,
      "step": 31382,
      "training_step_time": 7.199312925338745
    },
    {
      "epoch": 2.9929161071777343e-05,
      "model_forward_time": 1.7155652046203613,
      "step": 31383
    },
    {
      "epoch": 2.9929161071777343e-05,
      "step": 31383,
      "training_step_time": 6.722798824310303
    },
    {
      "epoch": 2.993011474609375e-05,
      "model_forward_time": 1.7129602432250977,
      "step": 31384
    },
    {
      "epoch": 2.993011474609375e-05,
      "step": 31384,
      "training_step_time": 7.538497686386108
    },
    {
      "epoch": 2.9931068420410158e-05,
      "model_forward_time": 1.722034215927124,
      "step": 31385
    },
    {
      "epoch": 2.9931068420410158e-05,
      "step": 31385,
      "training_step_time": 8.02096438407898
    },
    {
      "epoch": 2.9932022094726562e-05,
      "model_forward_time": 1.7214081287384033,
      "step": 31386
    },
    {
      "epoch": 2.9932022094726562e-05,
      "step": 31386,
      "training_step_time": 7.607179403305054
    },
    {
      "epoch": 2.993297576904297e-05,
      "model_forward_time": 2.8973426818847656,
      "step": 31387
    },
    {
      "epoch": 2.993297576904297e-05,
      "step": 31387,
      "training_step_time": 7.677266836166382
    },
    {
      "epoch": 2.9933929443359377e-05,
      "model_forward_time": 1.7383975982666016,
      "step": 31388
    },
    {
      "epoch": 2.9933929443359377e-05,
      "step": 31388,
      "training_step_time": 7.680719614028931
    },
    {
      "epoch": 2.993488311767578e-05,
      "model_forward_time": 1.7134020328521729,
      "step": 31389
    },
    {
      "epoch": 2.993488311767578e-05,
      "step": 31389,
      "training_step_time": 8.203806400299072
    },
    {
      "epoch": 2.9935836791992188e-05,
      "grad_norm": 0.22909919917583466,
      "learning_rate": 8.213923974729162e-06,
      "loss": 0.064,
      "step": 31390
    },
    {
      "epoch": 2.9935836791992188e-05,
      "model_forward_time": 1.7132842540740967,
      "step": 31390
    },
    {
      "dynamics_loss_avg": 0.059349178336560726,
      "epoch": 2.9935836791992188e-05,
      "step": 31390
    },
    {
      "action_loss_avg": 0.001953816445893608,
      "epoch": 2.9935836791992188e-05,
      "step": 31390
    },
    {
      "epoch": 2.9935836791992188e-05,
      "step": 31390,
      "training_step_time": 7.605690002441406
    },
    {
      "epoch": 2.9936790466308592e-05,
      "model_forward_time": 1.3426032066345215,
      "step": 31391
    },
    {
      "epoch": 2.9936790466308592e-05,
      "step": 31391,
      "training_step_time": 7.188822507858276
    },
    {
      "epoch": 2.9937744140625e-05,
      "model_forward_time": 1.6999258995056152,
      "step": 31392
    },
    {
      "epoch": 2.9937744140625e-05,
      "step": 31392,
      "training_step_time": 7.111653804779053
    },
    {
      "epoch": 2.9938697814941407e-05,
      "model_forward_time": 1.7199387550354004,
      "step": 31393
    },
    {
      "epoch": 2.9938697814941407e-05,
      "step": 31393,
      "training_step_time": 7.36508846282959
    },
    {
      "epoch": 2.993965148925781e-05,
      "model_forward_time": 1.7205963134765625,
      "step": 31394
    },
    {
      "epoch": 2.993965148925781e-05,
      "step": 31394,
      "training_step_time": 8.473634004592896
    },
    {
      "epoch": 2.994060516357422e-05,
      "model_forward_time": 1.725022315979004,
      "step": 31395
    },
    {
      "epoch": 2.994060516357422e-05,
      "step": 31395,
      "training_step_time": 7.543313503265381
    },
    {
      "epoch": 2.9941558837890626e-05,
      "model_forward_time": 2.831690549850464,
      "step": 31396
    },
    {
      "epoch": 2.9941558837890626e-05,
      "step": 31396,
      "training_step_time": 7.723388433456421
    },
    {
      "epoch": 2.994251251220703e-05,
      "model_forward_time": 1.734299659729004,
      "step": 31397
    },
    {
      "epoch": 2.994251251220703e-05,
      "step": 31397,
      "training_step_time": 8.718886137008667
    },
    {
      "epoch": 2.9943466186523437e-05,
      "model_forward_time": 1.7185163497924805,
      "step": 31398
    },
    {
      "epoch": 2.9943466186523437e-05,
      "step": 31398,
      "training_step_time": 8.290925025939941
    },
    {
      "epoch": 2.9944419860839845e-05,
      "model_forward_time": 1.7235453128814697,
      "step": 31399
    },
    {
      "epoch": 2.9944419860839845e-05,
      "step": 31399,
      "training_step_time": 7.632302284240723
    },
    {
      "epoch": 2.994537353515625e-05,
      "grad_norm": 0.2416037768125534,
      "learning_rate": 8.212657163746372e-06,
      "loss": 0.0607,
      "step": 31400
    },
    {
      "epoch": 2.994537353515625e-05,
      "model_forward_time": 1.3630008697509766,
      "step": 31400
    },
    {
      "dynamics_loss_avg": 0.06242802515625954,
      "epoch": 2.994537353515625e-05,
      "step": 31400
    },
    {
      "action_loss_avg": 0.0017920447047799827,
      "epoch": 2.994537353515625e-05,
      "step": 31400
    },
    {
      "epoch": 2.994537353515625e-05,
      "step": 31400,
      "training_step_time": 7.443155765533447
    },
    {
      "epoch": 2.9946327209472656e-05,
      "model_forward_time": 1.7121703624725342,
      "step": 31401
    },
    {
      "epoch": 2.9946327209472656e-05,
      "step": 31401,
      "training_step_time": 7.127507448196411
    },
    {
      "epoch": 2.9947280883789063e-05,
      "model_forward_time": 1.7202353477478027,
      "step": 31402
    },
    {
      "epoch": 2.9947280883789063e-05,
      "step": 31402,
      "training_step_time": 7.2479088306427
    },
    {
      "epoch": 2.9948234558105468e-05,
      "model_forward_time": 1.7211792469024658,
      "step": 31403
    },
    {
      "epoch": 2.9948234558105468e-05,
      "step": 31403,
      "training_step_time": 7.583093881607056
    },
    {
      "epoch": 2.9949188232421875e-05,
      "model_forward_time": 1.7245433330535889,
      "step": 31404
    },
    {
      "epoch": 2.9949188232421875e-05,
      "step": 31404,
      "training_step_time": 7.7053446769714355
    },
    {
      "epoch": 2.9950141906738282e-05,
      "model_forward_time": 2.81164288520813,
      "step": 31405
    },
    {
      "epoch": 2.9950141906738282e-05,
      "step": 31405,
      "training_step_time": 8.65363097190857
    },
    {
      "epoch": 2.9951095581054686e-05,
      "model_forward_time": 1.723304033279419,
      "step": 31406
    },
    {
      "epoch": 2.9951095581054686e-05,
      "step": 31406,
      "training_step_time": 7.725545406341553
    },
    {
      "epoch": 2.9952049255371094e-05,
      "model_forward_time": 1.7206494808197021,
      "step": 31407
    },
    {
      "epoch": 2.9952049255371094e-05,
      "step": 31407,
      "training_step_time": 8.890058755874634
    },
    {
      "epoch": 2.99530029296875e-05,
      "model_forward_time": 1.721001148223877,
      "step": 31408
    },
    {
      "epoch": 2.99530029296875e-05,
      "step": 31408,
      "training_step_time": 7.558094263076782
    },
    {
      "epoch": 2.9953956604003905e-05,
      "model_forward_time": 1.7202632427215576,
      "step": 31409
    },
    {
      "epoch": 2.9953956604003905e-05,
      "step": 31409,
      "training_step_time": 7.449705123901367
    },
    {
      "epoch": 2.9954910278320313e-05,
      "grad_norm": 0.22509722411632538,
      "learning_rate": 8.211390001432226e-06,
      "loss": 0.0644,
      "step": 31410
    },
    {
      "epoch": 2.9954910278320313e-05,
      "model_forward_time": 1.7235698699951172,
      "step": 31410
    },
    {
      "dynamics_loss_avg": 0.07918908894062042,
      "epoch": 2.9954910278320313e-05,
      "step": 31410
    },
    {
      "action_loss_avg": 0.0026281191851012407,
      "epoch": 2.9954910278320313e-05,
      "step": 31410
    },
    {
      "epoch": 2.9954910278320313e-05,
      "step": 31410,
      "training_step_time": 7.026297330856323
    },
    {
      "epoch": 2.995586395263672e-05,
      "model_forward_time": 1.7219593524932861,
      "step": 31411
    },
    {
      "epoch": 2.995586395263672e-05,
      "step": 31411,
      "training_step_time": 7.088756322860718
    },
    {
      "epoch": 2.9956817626953124e-05,
      "model_forward_time": 1.7250304222106934,
      "step": 31412
    },
    {
      "epoch": 2.9956817626953124e-05,
      "step": 31412,
      "training_step_time": 7.4927449226379395
    },
    {
      "epoch": 2.995777130126953e-05,
      "model_forward_time": 1.714493751525879,
      "step": 31413
    },
    {
      "epoch": 2.995777130126953e-05,
      "step": 31413,
      "training_step_time": 8.36765432357788
    },
    {
      "epoch": 2.995872497558594e-05,
      "model_forward_time": 1.7500584125518799,
      "step": 31414
    },
    {
      "epoch": 2.995872497558594e-05,
      "step": 31414,
      "training_step_time": 8.931329727172852
    },
    {
      "epoch": 2.9959678649902343e-05,
      "model_forward_time": 1.7266244888305664,
      "step": 31415
    },
    {
      "epoch": 2.9959678649902343e-05,
      "step": 31415,
      "training_step_time": 7.8410139083862305
    },
    {
      "epoch": 2.996063232421875e-05,
      "model_forward_time": 1.739605188369751,
      "step": 31416
    },
    {
      "epoch": 2.996063232421875e-05,
      "step": 31416,
      "training_step_time": 8.618372201919556
    },
    {
      "epoch": 2.9961585998535158e-05,
      "model_forward_time": 1.7183706760406494,
      "step": 31417
    },
    {
      "epoch": 2.9961585998535158e-05,
      "step": 31417,
      "training_step_time": 8.834325313568115
    },
    {
      "epoch": 2.9962539672851562e-05,
      "model_forward_time": 1.720989465713501,
      "step": 31418
    },
    {
      "epoch": 2.9962539672851562e-05,
      "step": 31418,
      "training_step_time": 7.67125391960144
    },
    {
      "epoch": 2.996349334716797e-05,
      "model_forward_time": 1.7208616733551025,
      "step": 31419
    },
    {
      "epoch": 2.996349334716797e-05,
      "step": 31419,
      "training_step_time": 7.1809608936309814
    },
    {
      "epoch": 2.9964447021484377e-05,
      "grad_norm": 0.1847393959760666,
      "learning_rate": 8.210122487925299e-06,
      "loss": 0.0599,
      "step": 31420
    },
    {
      "epoch": 2.9964447021484377e-05,
      "model_forward_time": 1.7238008975982666,
      "step": 31420
    },
    {
      "dynamics_loss_avg": 0.04780634101480245,
      "epoch": 2.9964447021484377e-05,
      "step": 31420
    },
    {
      "action_loss_avg": 0.0022936715453397484,
      "epoch": 2.9964447021484377e-05,
      "step": 31420
    },
    {
      "epoch": 2.9964447021484377e-05,
      "step": 31420,
      "training_step_time": 7.207994699478149
    },
    {
      "epoch": 2.996540069580078e-05,
      "model_forward_time": 1.7219762802124023,
      "step": 31421
    },
    {
      "epoch": 2.996540069580078e-05,
      "step": 31421,
      "training_step_time": 7.437623977661133
    },
    {
      "epoch": 2.9966354370117188e-05,
      "model_forward_time": 1.7215516567230225,
      "step": 31422
    },
    {
      "epoch": 2.9966354370117188e-05,
      "step": 31422,
      "training_step_time": 7.883042812347412
    },
    {
      "epoch": 2.9967308044433592e-05,
      "model_forward_time": 1.7269337177276611,
      "step": 31423
    },
    {
      "epoch": 2.9967308044433592e-05,
      "step": 31423,
      "training_step_time": 8.31789255142212
    },
    {
      "epoch": 2.996826171875e-05,
      "model_forward_time": 2.8803939819335938,
      "step": 31424
    },
    {
      "epoch": 2.996826171875e-05,
      "step": 31424,
      "training_step_time": 7.789314270019531
    },
    {
      "epoch": 2.9969215393066407e-05,
      "model_forward_time": 1.7501130104064941,
      "step": 31425
    },
    {
      "epoch": 2.9969215393066407e-05,
      "step": 31425,
      "training_step_time": 7.8474812507629395
    },
    {
      "epoch": 2.997016906738281e-05,
      "model_forward_time": 1.720296859741211,
      "step": 31426
    },
    {
      "epoch": 2.997016906738281e-05,
      "step": 31426,
      "training_step_time": 8.691104650497437
    },
    {
      "epoch": 2.997112274169922e-05,
      "model_forward_time": 1.7227766513824463,
      "step": 31427
    },
    {
      "epoch": 2.997112274169922e-05,
      "step": 31427,
      "training_step_time": 7.688881874084473
    },
    {
      "epoch": 2.9972076416015626e-05,
      "model_forward_time": 1.7169911861419678,
      "step": 31428
    },
    {
      "epoch": 2.9972076416015626e-05,
      "step": 31428,
      "training_step_time": 7.585232734680176
    },
    {
      "epoch": 2.997303009033203e-05,
      "model_forward_time": 1.7167255878448486,
      "step": 31429
    },
    {
      "epoch": 2.997303009033203e-05,
      "step": 31429,
      "training_step_time": 7.216315984725952
    },
    {
      "epoch": 2.9973983764648437e-05,
      "grad_norm": 0.17508631944656372,
      "learning_rate": 8.208854623364202e-06,
      "loss": 0.0654,
      "step": 31430
    },
    {
      "epoch": 2.9973983764648437e-05,
      "model_forward_time": 1.7279348373413086,
      "step": 31430
    },
    {
      "dynamics_loss_avg": 0.06260786950588226,
      "epoch": 2.9973983764648437e-05,
      "step": 31430
    },
    {
      "action_loss_avg": 0.0025718971621245144,
      "epoch": 2.9973983764648437e-05,
      "step": 31430
    },
    {
      "epoch": 2.9973983764648437e-05,
      "step": 31430,
      "training_step_time": 7.462342262268066
    },
    {
      "epoch": 2.9974937438964845e-05,
      "model_forward_time": 1.7209062576293945,
      "step": 31431
    },
    {
      "epoch": 2.9974937438964845e-05,
      "step": 31431,
      "training_step_time": 7.8505470752716064
    },
    {
      "epoch": 2.997589111328125e-05,
      "model_forward_time": 1.7247333526611328,
      "step": 31432
    },
    {
      "epoch": 2.997589111328125e-05,
      "step": 31432,
      "training_step_time": 7.771581411361694
    },
    {
      "epoch": 2.9976844787597656e-05,
      "model_forward_time": 2.9151244163513184,
      "step": 31433
    },
    {
      "epoch": 2.9976844787597656e-05,
      "step": 31433,
      "training_step_time": 8.679725408554077
    },
    {
      "epoch": 2.9977798461914064e-05,
      "model_forward_time": 1.7408084869384766,
      "step": 31434
    },
    {
      "epoch": 2.9977798461914064e-05,
      "step": 31434,
      "training_step_time": 8.898314237594604
    },
    {
      "epoch": 2.9978752136230468e-05,
      "model_forward_time": 1.7215919494628906,
      "step": 31435
    },
    {
      "epoch": 2.9978752136230468e-05,
      "step": 31435,
      "training_step_time": 7.889963626861572
    },
    {
      "epoch": 2.9979705810546875e-05,
      "model_forward_time": 1.725395679473877,
      "step": 31436
    },
    {
      "epoch": 2.9979705810546875e-05,
      "step": 31436,
      "training_step_time": 8.329630851745605
    },
    {
      "epoch": 2.9980659484863282e-05,
      "model_forward_time": 1.7214834690093994,
      "step": 31437
    },
    {
      "epoch": 2.9980659484863282e-05,
      "step": 31437,
      "training_step_time": 7.639131546020508
    },
    {
      "epoch": 2.9981613159179686e-05,
      "model_forward_time": 1.7108078002929688,
      "step": 31438
    },
    {
      "epoch": 2.9981613159179686e-05,
      "step": 31438,
      "training_step_time": 7.323865175247192
    },
    {
      "epoch": 2.9982566833496094e-05,
      "model_forward_time": 1.7228517532348633,
      "step": 31439
    },
    {
      "epoch": 2.9982566833496094e-05,
      "step": 31439,
      "training_step_time": 7.59425163269043
    },
    {
      "epoch": 2.99835205078125e-05,
      "grad_norm": 0.1481824368238449,
      "learning_rate": 8.207586407887588e-06,
      "loss": 0.0608,
      "step": 31440
    },
    {
      "epoch": 2.99835205078125e-05,
      "model_forward_time": 1.7297236919403076,
      "step": 31440
    },
    {
      "dynamics_loss_avg": 0.048685078509151936,
      "epoch": 2.99835205078125e-05,
      "step": 31440
    },
    {
      "action_loss_avg": 0.0021219627233222127,
      "epoch": 2.99835205078125e-05,
      "step": 31440
    },
    {
      "epoch": 2.99835205078125e-05,
      "step": 31440,
      "training_step_time": 8.047972440719604
    },
    {
      "epoch": 2.9984474182128905e-05,
      "model_forward_time": 1.7236824035644531,
      "step": 31441
    },
    {
      "epoch": 2.9984474182128905e-05,
      "step": 31441,
      "training_step_time": 9.209924697875977
    },
    {
      "epoch": 2.9985427856445313e-05,
      "model_forward_time": 2.835131883621216,
      "step": 31442
    },
    {
      "epoch": 2.9985427856445313e-05,
      "step": 31442,
      "training_step_time": 8.818352460861206
    },
    {
      "epoch": 2.998638153076172e-05,
      "model_forward_time": 1.7591660022735596,
      "step": 31443
    },
    {
      "epoch": 2.998638153076172e-05,
      "step": 31443,
      "training_step_time": 8.492007732391357
    },
    {
      "epoch": 2.9987335205078124e-05,
      "model_forward_time": 1.7163991928100586,
      "step": 31444
    },
    {
      "epoch": 2.9987335205078124e-05,
      "step": 31444,
      "training_step_time": 8.548756837844849
    },
    {
      "epoch": 2.998828887939453e-05,
      "model_forward_time": 1.7211580276489258,
      "step": 31445
    },
    {
      "epoch": 2.998828887939453e-05,
      "step": 31445,
      "training_step_time": 7.839318037033081
    },
    {
      "epoch": 2.998924255371094e-05,
      "model_forward_time": 1.723078727722168,
      "step": 31446
    },
    {
      "epoch": 2.998924255371094e-05,
      "step": 31446,
      "training_step_time": 7.772645950317383
    },
    {
      "epoch": 2.9990196228027343e-05,
      "model_forward_time": 1.7217166423797607,
      "step": 31447
    },
    {
      "epoch": 2.9990196228027343e-05,
      "step": 31447,
      "training_step_time": 7.274349927902222
    },
    {
      "epoch": 2.999114990234375e-05,
      "model_forward_time": 1.7215185165405273,
      "step": 31448
    },
    {
      "epoch": 2.999114990234375e-05,
      "step": 31448,
      "training_step_time": 7.567256927490234
    },
    {
      "epoch": 2.9992103576660158e-05,
      "model_forward_time": 1.7251291275024414,
      "step": 31449
    },
    {
      "epoch": 2.9992103576660158e-05,
      "step": 31449,
      "training_step_time": 8.734144449234009
    },
    {
      "epoch": 2.9993057250976562e-05,
      "grad_norm": 0.14675132930278778,
      "learning_rate": 8.206317841634148e-06,
      "loss": 0.0614,
      "step": 31450
    },
    {
      "epoch": 2.9993057250976562e-05,
      "model_forward_time": 1.726081132888794,
      "step": 31450
    },
    {
      "dynamics_loss_avg": 0.07309104353189469,
      "epoch": 2.9993057250976562e-05,
      "step": 31450
    },
    {
      "action_loss_avg": 0.0034844840527512135,
      "epoch": 2.9993057250976562e-05,
      "step": 31450
    },
    {
      "epoch": 2.9993057250976562e-05,
      "step": 31450,
      "training_step_time": 7.773489713668823
    },
    {
      "epoch": 2.999401092529297e-05,
      "model_forward_time": 2.8464245796203613,
      "step": 31451
    },
    {
      "epoch": 2.999401092529297e-05,
      "step": 31451,
      "training_step_time": 7.859302997589111
    },
    {
      "epoch": 2.9994964599609377e-05,
      "model_forward_time": 1.7133152484893799,
      "step": 31452
    },
    {
      "epoch": 2.9994964599609377e-05,
      "step": 31452,
      "training_step_time": 8.664633512496948
    },
    {
      "epoch": 2.999591827392578e-05,
      "model_forward_time": 1.7221872806549072,
      "step": 31453
    },
    {
      "epoch": 2.999591827392578e-05,
      "step": 31453,
      "training_step_time": 9.027984380722046
    },
    {
      "epoch": 2.9996871948242188e-05,
      "model_forward_time": 1.3591876029968262,
      "step": 31454
    },
    {
      "epoch": 2.9996871948242188e-05,
      "step": 31454,
      "training_step_time": 7.888577938079834
    },
    {
      "epoch": 2.9997825622558592e-05,
      "model_forward_time": 1.725280523300171,
      "step": 31455
    },
    {
      "epoch": 2.9997825622558592e-05,
      "step": 31455,
      "training_step_time": 7.8382720947265625
    },
    {
      "epoch": 2.9998779296875e-05,
      "model_forward_time": 1.7166752815246582,
      "step": 31456
    },
    {
      "epoch": 2.9998779296875e-05,
      "step": 31456,
      "training_step_time": 7.278952598571777
    },
    {
      "epoch": 2.9999732971191407e-05,
      "model_forward_time": 1.7229385375976562,
      "step": 31457
    },
    {
      "epoch": 2.9999732971191407e-05,
      "step": 31457,
      "training_step_time": 7.5370705127716064
    },
    {
      "epoch": 3.000068664550781e-05,
      "model_forward_time": 1.7221651077270508,
      "step": 31458
    },
    {
      "epoch": 3.000068664550781e-05,
      "step": 31458,
      "training_step_time": 8.255839824676514
    },
    {
      "epoch": 3.000164031982422e-05,
      "model_forward_time": 1.7214608192443848,
      "step": 31459
    },
    {
      "epoch": 3.000164031982422e-05,
      "step": 31459,
      "training_step_time": 7.891506910324097
    },
    {
      "epoch": 3.0002593994140626e-05,
      "grad_norm": 0.1710795909166336,
      "learning_rate": 8.20504892474261e-06,
      "loss": 0.0586,
      "step": 31460
    },
    {
      "epoch": 3.0002593994140626e-05,
      "model_forward_time": 2.8696117401123047,
      "step": 31460
    },
    {
      "dynamics_loss_avg": 0.060532407090067866,
      "epoch": 3.0002593994140626e-05,
      "step": 31460
    },
    {
      "action_loss_avg": 0.0024670693586813287,
      "epoch": 3.0002593994140626e-05,
      "step": 31460
    },
    {
      "epoch": 3.0002593994140626e-05,
      "step": 31460,
      "training_step_time": 9.44926142692566
    },
    {
      "epoch": 3.000354766845703e-05,
      "model_forward_time": 1.7260282039642334,
      "step": 31461
    },
    {
      "epoch": 3.000354766845703e-05,
      "step": 31461,
      "training_step_time": 10.22182035446167
    },
    {
      "epoch": 3.0004501342773437e-05,
      "model_forward_time": 1.7190747261047363,
      "step": 31462
    },
    {
      "epoch": 3.0004501342773437e-05,
      "step": 31462,
      "training_step_time": 7.990035057067871
    },
    {
      "epoch": 3.0005455017089845e-05,
      "model_forward_time": 1.713975191116333,
      "step": 31463
    },
    {
      "epoch": 3.0005455017089845e-05,
      "step": 31463,
      "training_step_time": 7.914476633071899
    },
    {
      "epoch": 3.000640869140625e-05,
      "model_forward_time": 1.7206816673278809,
      "step": 31464
    },
    {
      "epoch": 3.000640869140625e-05,
      "step": 31464,
      "training_step_time": 7.809052467346191
    },
    {
      "epoch": 3.0007362365722656e-05,
      "model_forward_time": 1.7206602096557617,
      "step": 31465
    },
    {
      "epoch": 3.0007362365722656e-05,
      "step": 31465,
      "training_step_time": 7.482663631439209
    },
    {
      "epoch": 3.0008316040039064e-05,
      "model_forward_time": 1.721034049987793,
      "step": 31466
    },
    {
      "epoch": 3.0008316040039064e-05,
      "step": 31466,
      "training_step_time": 7.540131568908691
    },
    {
      "epoch": 3.0009269714355468e-05,
      "model_forward_time": 1.7028894424438477,
      "step": 31467
    },
    {
      "epoch": 3.0009269714355468e-05,
      "step": 31467,
      "training_step_time": 7.668856143951416
    },
    {
      "epoch": 3.0010223388671875e-05,
      "model_forward_time": 1.712996244430542,
      "step": 31468
    },
    {
      "epoch": 3.0010223388671875e-05,
      "step": 31468,
      "training_step_time": 8.789148092269897
    },
    {
      "epoch": 3.0011177062988282e-05,
      "model_forward_time": 2.880213975906372,
      "step": 31469
    },
    {
      "epoch": 3.0011177062988282e-05,
      "step": 31469,
      "training_step_time": 7.814697504043579
    },
    {
      "epoch": 3.0012130737304686e-05,
      "grad_norm": 0.15351015329360962,
      "learning_rate": 8.203779657351738e-06,
      "loss": 0.0614,
      "step": 31470
    },
    {
      "epoch": 3.0012130737304686e-05,
      "model_forward_time": 1.7213456630706787,
      "step": 31470
    },
    {
      "dynamics_loss_avg": 0.06248972807079554,
      "epoch": 3.0012130737304686e-05,
      "step": 31470
    },
    {
      "action_loss_avg": 0.0020404353854246438,
      "epoch": 3.0012130737304686e-05,
      "step": 31470
    },
    {
      "epoch": 3.0012130737304686e-05,
      "step": 31470,
      "training_step_time": 8.444298028945923
    },
    {
      "epoch": 3.0013084411621094e-05,
      "model_forward_time": 1.7134580612182617,
      "step": 31471
    },
    {
      "epoch": 3.0013084411621094e-05,
      "step": 31471,
      "training_step_time": 10.006866693496704
    },
    {
      "epoch": 3.00140380859375e-05,
      "model_forward_time": 1.70930814743042,
      "step": 31472
    },
    {
      "epoch": 3.00140380859375e-05,
      "step": 31472,
      "training_step_time": 7.8479163646698
    },
    {
      "epoch": 3.0014991760253905e-05,
      "model_forward_time": 1.7170395851135254,
      "step": 31473
    },
    {
      "epoch": 3.0014991760253905e-05,
      "step": 31473,
      "training_step_time": 8.479700803756714
    },
    {
      "epoch": 3.0015945434570313e-05,
      "model_forward_time": 1.730492353439331,
      "step": 31474
    },
    {
      "epoch": 3.0015945434570313e-05,
      "step": 31474,
      "training_step_time": 7.474883317947388
    },
    {
      "epoch": 3.001689910888672e-05,
      "model_forward_time": 1.7161602973937988,
      "step": 31475
    },
    {
      "epoch": 3.001689910888672e-05,
      "step": 31475,
      "training_step_time": 7.487051010131836
    },
    {
      "epoch": 3.0017852783203124e-05,
      "model_forward_time": 1.7180414199829102,
      "step": 31476
    },
    {
      "epoch": 3.0017852783203124e-05,
      "step": 31476,
      "training_step_time": 7.729282379150391
    },
    {
      "epoch": 3.001880645751953e-05,
      "model_forward_time": 1.7142188549041748,
      "step": 31477
    },
    {
      "epoch": 3.001880645751953e-05,
      "step": 31477,
      "training_step_time": 7.861432075500488
    },
    {
      "epoch": 3.001976013183594e-05,
      "model_forward_time": 2.8750481605529785,
      "step": 31478
    },
    {
      "epoch": 3.001976013183594e-05,
      "step": 31478,
      "training_step_time": 7.798575401306152
    },
    {
      "epoch": 3.0020713806152343e-05,
      "model_forward_time": 1.717912197113037,
      "step": 31479
    },
    {
      "epoch": 3.0020713806152343e-05,
      "step": 31479,
      "training_step_time": 9.883538246154785
    },
    {
      "epoch": 3.002166748046875e-05,
      "grad_norm": 0.18483799695968628,
      "learning_rate": 8.20251003960034e-06,
      "loss": 0.0671,
      "step": 31480
    },
    {
      "epoch": 3.002166748046875e-05,
      "model_forward_time": 1.3703422546386719,
      "step": 31480
    },
    {
      "dynamics_loss_avg": 0.06306775920093059,
      "epoch": 3.002166748046875e-05,
      "step": 31480
    },
    {
      "action_loss_avg": 0.001980882242787629,
      "epoch": 3.002166748046875e-05,
      "step": 31480
    },
    {
      "epoch": 3.002166748046875e-05,
      "step": 31480,
      "training_step_time": 9.311956644058228
    },
    {
      "epoch": 3.0022621154785158e-05,
      "model_forward_time": 1.7203631401062012,
      "step": 31481
    },
    {
      "epoch": 3.0022621154785158e-05,
      "step": 31481,
      "training_step_time": 8.396296262741089
    },
    {
      "epoch": 3.0023574829101562e-05,
      "model_forward_time": 1.7234315872192383,
      "step": 31482
    },
    {
      "epoch": 3.0023574829101562e-05,
      "step": 31482,
      "training_step_time": 7.776418447494507
    },
    {
      "epoch": 3.002452850341797e-05,
      "model_forward_time": 1.715958833694458,
      "step": 31483
    },
    {
      "epoch": 3.002452850341797e-05,
      "step": 31483,
      "training_step_time": 7.735071659088135
    },
    {
      "epoch": 3.0025482177734377e-05,
      "model_forward_time": 1.7232234477996826,
      "step": 31484
    },
    {
      "epoch": 3.0025482177734377e-05,
      "step": 31484,
      "training_step_time": 7.509862422943115
    },
    {
      "epoch": 3.002643585205078e-05,
      "model_forward_time": 1.7227473258972168,
      "step": 31485
    },
    {
      "epoch": 3.002643585205078e-05,
      "step": 31485,
      "training_step_time": 8.430346250534058
    },
    {
      "epoch": 3.0027389526367188e-05,
      "model_forward_time": 1.7175064086914062,
      "step": 31486
    },
    {
      "epoch": 3.0027389526367188e-05,
      "step": 31486,
      "training_step_time": 7.903221368789673
    },
    {
      "epoch": 3.0028343200683592e-05,
      "model_forward_time": 2.8303940296173096,
      "step": 31487
    },
    {
      "epoch": 3.0028343200683592e-05,
      "step": 31487,
      "training_step_time": 7.813634157180786
    },
    {
      "epoch": 3.0029296875e-05,
      "model_forward_time": 1.7158007621765137,
      "step": 31488
    },
    {
      "epoch": 3.0029296875e-05,
      "step": 31488,
      "training_step_time": 8.705589532852173
    },
    {
      "epoch": 3.0030250549316407e-05,
      "model_forward_time": 1.7328999042510986,
      "step": 31489
    },
    {
      "epoch": 3.0030250549316407e-05,
      "step": 31489,
      "training_step_time": 8.454355001449585
    },
    {
      "epoch": 3.003120422363281e-05,
      "grad_norm": 0.1790241003036499,
      "learning_rate": 8.201240071627259e-06,
      "loss": 0.0636,
      "step": 31490
    },
    {
      "epoch": 3.003120422363281e-05,
      "model_forward_time": 1.7211689949035645,
      "step": 31490
    },
    {
      "dynamics_loss_avg": 0.05280344448983669,
      "epoch": 3.003120422363281e-05,
      "step": 31490
    },
    {
      "action_loss_avg": 0.0020374901418108494,
      "epoch": 3.003120422363281e-05,
      "step": 31490
    },
    {
      "epoch": 3.003120422363281e-05,
      "step": 31490,
      "training_step_time": 9.533010721206665
    },
    {
      "epoch": 3.003215789794922e-05,
      "model_forward_time": 1.7144999504089355,
      "step": 31491
    },
    {
      "epoch": 3.003215789794922e-05,
      "step": 31491,
      "training_step_time": 8.590177297592163
    },
    {
      "epoch": 3.0033111572265626e-05,
      "model_forward_time": 1.7207887172698975,
      "step": 31492
    },
    {
      "epoch": 3.0033111572265626e-05,
      "step": 31492,
      "training_step_time": 7.823724269866943
    },
    {
      "epoch": 3.003406524658203e-05,
      "model_forward_time": 1.7038085460662842,
      "step": 31493
    },
    {
      "epoch": 3.003406524658203e-05,
      "step": 31493,
      "training_step_time": 7.670799970626831
    },
    {
      "epoch": 3.0035018920898437e-05,
      "model_forward_time": 1.725402593612671,
      "step": 31494
    },
    {
      "epoch": 3.0035018920898437e-05,
      "step": 31494,
      "training_step_time": 7.470142364501953
    },
    {
      "epoch": 3.0035972595214845e-05,
      "model_forward_time": 1.7190167903900146,
      "step": 31495
    },
    {
      "epoch": 3.0035972595214845e-05,
      "step": 31495,
      "training_step_time": 7.899336099624634
    },
    {
      "epoch": 3.003692626953125e-05,
      "model_forward_time": 1.7209415435791016,
      "step": 31496
    },
    {
      "epoch": 3.003692626953125e-05,
      "step": 31496,
      "training_step_time": 7.825817108154297
    },
    {
      "epoch": 3.0037879943847656e-05,
      "model_forward_time": 2.8527371883392334,
      "step": 31497
    },
    {
      "epoch": 3.0037879943847656e-05,
      "step": 31497,
      "training_step_time": 8.893692970275879
    },
    {
      "epoch": 3.0038833618164064e-05,
      "model_forward_time": 1.72391676902771,
      "step": 31498
    },
    {
      "epoch": 3.0038833618164064e-05,
      "step": 31498,
      "training_step_time": 7.984612703323364
    },
    {
      "epoch": 3.0039787292480468e-05,
      "model_forward_time": 1.7271225452423096,
      "step": 31499
    },
    {
      "epoch": 3.0039787292480468e-05,
      "step": 31499,
      "training_step_time": 7.917252540588379
    },
    {
      "epoch": 3.0040740966796875e-05,
      "grad_norm": 0.12800465524196625,
      "learning_rate": 8.199969753571377e-06,
      "loss": 0.063,
      "step": 31500
    },
    {
      "epoch": 3.0040740966796875e-05,
      "model_forward_time": 1.7208468914031982,
      "step": 31500
    },
    {
      "dynamics_loss_avg": 0.043593081459403035,
      "epoch": 3.0040740966796875e-05,
      "step": 31500
    },
    {
      "action_loss_avg": 0.002632545359665528,
      "epoch": 3.0040740966796875e-05,
      "step": 31500
    },
    {
      "epoch": 3.0040740966796875e-05,
      "step": 31500,
      "training_step_time": 7.90706729888916
    },
    {
      "epoch": 3.0041694641113282e-05,
      "model_forward_time": 1.7150561809539795,
      "step": 31501
    },
    {
      "epoch": 3.0041694641113282e-05,
      "step": 31501,
      "training_step_time": 7.7373692989349365
    },
    {
      "epoch": 3.0042648315429686e-05,
      "model_forward_time": 1.724656105041504,
      "step": 31502
    },
    {
      "epoch": 3.0042648315429686e-05,
      "step": 31502,
      "training_step_time": 7.600671768188477
    },
    {
      "epoch": 3.0043601989746094e-05,
      "model_forward_time": 1.7162461280822754,
      "step": 31503
    },
    {
      "epoch": 3.0043601989746094e-05,
      "step": 31503,
      "training_step_time": 7.2566094398498535
    },
    {
      "epoch": 3.00445556640625e-05,
      "model_forward_time": 1.7269575595855713,
      "step": 31504
    },
    {
      "epoch": 3.00445556640625e-05,
      "step": 31504,
      "training_step_time": 7.912670373916626
    },
    {
      "epoch": 3.0045509338378905e-05,
      "model_forward_time": 1.7226293087005615,
      "step": 31505
    },
    {
      "epoch": 3.0045509338378905e-05,
      "step": 31505,
      "training_step_time": 8.722758769989014
    },
    {
      "epoch": 3.0046463012695313e-05,
      "model_forward_time": 2.8384788036346436,
      "step": 31506
    },
    {
      "epoch": 3.0046463012695313e-05,
      "step": 31506,
      "training_step_time": 8.711029291152954
    },
    {
      "epoch": 3.004741668701172e-05,
      "model_forward_time": 1.7168540954589844,
      "step": 31507
    },
    {
      "epoch": 3.004741668701172e-05,
      "step": 31507,
      "training_step_time": 8.948736667633057
    },
    {
      "epoch": 3.0048370361328124e-05,
      "model_forward_time": 1.7096083164215088,
      "step": 31508
    },
    {
      "epoch": 3.0048370361328124e-05,
      "step": 31508,
      "training_step_time": 7.974219560623169
    },
    {
      "epoch": 3.004932403564453e-05,
      "model_forward_time": 1.7117445468902588,
      "step": 31509
    },
    {
      "epoch": 3.004932403564453e-05,
      "step": 31509,
      "training_step_time": 8.915282249450684
    },
    {
      "epoch": 3.005027770996094e-05,
      "grad_norm": 0.21189989149570465,
      "learning_rate": 8.198699085571616e-06,
      "loss": 0.0643,
      "step": 31510
    },
    {
      "epoch": 3.005027770996094e-05,
      "model_forward_time": 1.7296028137207031,
      "step": 31510
    },
    {
      "dynamics_loss_avg": 0.05123872496187687,
      "epoch": 3.005027770996094e-05,
      "step": 31510
    },
    {
      "action_loss_avg": 0.0019753452914301308,
      "epoch": 3.005027770996094e-05,
      "step": 31510
    },
    {
      "epoch": 3.005027770996094e-05,
      "step": 31510,
      "training_step_time": 7.8248584270477295
    },
    {
      "epoch": 3.0051231384277343e-05,
      "model_forward_time": 1.7266380786895752,
      "step": 31511
    },
    {
      "epoch": 3.0051231384277343e-05,
      "step": 31511,
      "training_step_time": 7.617260932922363
    },
    {
      "epoch": 3.005218505859375e-05,
      "model_forward_time": 1.7210772037506104,
      "step": 31512
    },
    {
      "epoch": 3.005218505859375e-05,
      "step": 31512,
      "training_step_time": 7.3225696086883545
    },
    {
      "epoch": 3.0053138732910158e-05,
      "model_forward_time": 1.720181941986084,
      "step": 31513
    },
    {
      "epoch": 3.0053138732910158e-05,
      "step": 31513,
      "training_step_time": 8.075074672698975
    },
    {
      "epoch": 3.0054092407226562e-05,
      "model_forward_time": 1.7248902320861816,
      "step": 31514
    },
    {
      "epoch": 3.0054092407226562e-05,
      "step": 31514,
      "training_step_time": 8.657118797302246
    },
    {
      "epoch": 3.005504608154297e-05,
      "model_forward_time": 2.8925487995147705,
      "step": 31515
    },
    {
      "epoch": 3.005504608154297e-05,
      "step": 31515,
      "training_step_time": 8.590651273727417
    },
    {
      "epoch": 3.0055999755859373e-05,
      "model_forward_time": 1.7097043991088867,
      "step": 31516
    },
    {
      "epoch": 3.0055999755859373e-05,
      "step": 31516,
      "training_step_time": 8.77534794807434
    },
    {
      "epoch": 3.005695343017578e-05,
      "model_forward_time": 1.742192268371582,
      "step": 31517
    },
    {
      "epoch": 3.005695343017578e-05,
      "step": 31517,
      "training_step_time": 7.8623573780059814
    },
    {
      "epoch": 3.0057907104492188e-05,
      "model_forward_time": 1.7175462245941162,
      "step": 31518
    },
    {
      "epoch": 3.0057907104492188e-05,
      "step": 31518,
      "training_step_time": 7.876430988311768
    },
    {
      "epoch": 3.0058860778808592e-05,
      "model_forward_time": 1.70859694480896,
      "step": 31519
    },
    {
      "epoch": 3.0058860778808592e-05,
      "step": 31519,
      "training_step_time": 7.901110887527466
    },
    {
      "epoch": 3.0059814453125e-05,
      "grad_norm": 0.17526616156101227,
      "learning_rate": 8.197428067766929e-06,
      "loss": 0.0631,
      "step": 31520
    },
    {
      "epoch": 3.0059814453125e-05,
      "model_forward_time": 1.713376522064209,
      "step": 31520
    },
    {
      "dynamics_loss_avg": 0.05018486902117729,
      "epoch": 3.0059814453125e-05,
      "step": 31520
    },
    {
      "action_loss_avg": 0.0030710546649061143,
      "epoch": 3.0059814453125e-05,
      "step": 31520
    },
    {
      "epoch": 3.0059814453125e-05,
      "step": 31520,
      "training_step_time": 7.61259126663208
    },
    {
      "epoch": 3.0060768127441407e-05,
      "model_forward_time": 1.7159628868103027,
      "step": 31521
    },
    {
      "epoch": 3.0060768127441407e-05,
      "step": 31521,
      "training_step_time": 7.23688530921936
    },
    {
      "epoch": 3.006172180175781e-05,
      "model_forward_time": 1.723602533340454,
      "step": 31522
    },
    {
      "epoch": 3.006172180175781e-05,
      "step": 31522,
      "training_step_time": 7.921381711959839
    },
    {
      "epoch": 3.006267547607422e-05,
      "model_forward_time": 1.716491460800171,
      "step": 31523
    },
    {
      "epoch": 3.006267547607422e-05,
      "step": 31523,
      "training_step_time": 7.813645124435425
    },
    {
      "epoch": 3.0063629150390626e-05,
      "model_forward_time": 2.8940393924713135,
      "step": 31524
    },
    {
      "epoch": 3.0063629150390626e-05,
      "step": 31524,
      "training_step_time": 7.838335752487183
    },
    {
      "epoch": 3.006458282470703e-05,
      "model_forward_time": 1.7170376777648926,
      "step": 31525
    },
    {
      "epoch": 3.006458282470703e-05,
      "step": 31525,
      "training_step_time": 9.27888011932373
    },
    {
      "epoch": 3.0065536499023437e-05,
      "model_forward_time": 1.736670732498169,
      "step": 31526
    },
    {
      "epoch": 3.0065536499023437e-05,
      "step": 31526,
      "training_step_time": 7.895663022994995
    },
    {
      "epoch": 3.0066490173339845e-05,
      "model_forward_time": 1.7179536819458008,
      "step": 31527
    },
    {
      "epoch": 3.0066490173339845e-05,
      "step": 31527,
      "training_step_time": 7.871108293533325
    },
    {
      "epoch": 3.006744384765625e-05,
      "model_forward_time": 1.7081501483917236,
      "step": 31528
    },
    {
      "epoch": 3.006744384765625e-05,
      "step": 31528,
      "training_step_time": 8.852427244186401
    },
    {
      "epoch": 3.0068397521972656e-05,
      "model_forward_time": 1.713609218597412,
      "step": 31529
    },
    {
      "epoch": 3.0068397521972656e-05,
      "step": 31529,
      "training_step_time": 7.63267970085144
    },
    {
      "epoch": 3.0069351196289064e-05,
      "grad_norm": 0.15696552395820618,
      "learning_rate": 8.196156700296316e-06,
      "loss": 0.0625,
      "step": 31530
    },
    {
      "epoch": 3.0069351196289064e-05,
      "model_forward_time": 1.7147209644317627,
      "step": 31530
    },
    {
      "dynamics_loss_avg": 0.07598623223602771,
      "epoch": 3.0069351196289064e-05,
      "step": 31530
    },
    {
      "action_loss_avg": 0.0017081431287806481,
      "epoch": 3.0069351196289064e-05,
      "step": 31530
    },
    {
      "epoch": 3.0069351196289064e-05,
      "step": 31530,
      "training_step_time": 7.265743255615234
    },
    {
      "epoch": 3.0070304870605468e-05,
      "model_forward_time": 1.717332124710083,
      "step": 31531
    },
    {
      "epoch": 3.0070304870605468e-05,
      "step": 31531,
      "training_step_time": 8.865705966949463
    },
    {
      "epoch": 3.0071258544921875e-05,
      "model_forward_time": 1.7199840545654297,
      "step": 31532
    },
    {
      "epoch": 3.0071258544921875e-05,
      "step": 31532,
      "training_step_time": 7.7408506870269775
    },
    {
      "epoch": 3.0072212219238282e-05,
      "model_forward_time": 2.8211090564727783,
      "step": 31533
    },
    {
      "epoch": 3.0072212219238282e-05,
      "step": 31533,
      "training_step_time": 8.31651496887207
    },
    {
      "epoch": 3.0073165893554686e-05,
      "model_forward_time": 1.7326228618621826,
      "step": 31534
    },
    {
      "epoch": 3.0073165893554686e-05,
      "step": 31534,
      "training_step_time": 10.39340090751648
    },
    {
      "epoch": 3.0074119567871094e-05,
      "model_forward_time": 1.7232229709625244,
      "step": 31535
    },
    {
      "epoch": 3.0074119567871094e-05,
      "step": 31535,
      "training_step_time": 7.901670455932617
    },
    {
      "epoch": 3.00750732421875e-05,
      "model_forward_time": 1.7231214046478271,
      "step": 31536
    },
    {
      "epoch": 3.00750732421875e-05,
      "step": 31536,
      "training_step_time": 7.710427522659302
    },
    {
      "epoch": 3.0076026916503905e-05,
      "model_forward_time": 1.7117714881896973,
      "step": 31537
    },
    {
      "epoch": 3.0076026916503905e-05,
      "step": 31537,
      "training_step_time": 7.901414632797241
    },
    {
      "epoch": 3.0076980590820313e-05,
      "model_forward_time": 1.715993881225586,
      "step": 31538
    },
    {
      "epoch": 3.0076980590820313e-05,
      "step": 31538,
      "training_step_time": 8.414023160934448
    },
    {
      "epoch": 3.007793426513672e-05,
      "model_forward_time": 1.3577029705047607,
      "step": 31539
    },
    {
      "epoch": 3.007793426513672e-05,
      "step": 31539,
      "training_step_time": 7.871331453323364
    },
    {
      "epoch": 3.0078887939453124e-05,
      "grad_norm": 0.16615521907806396,
      "learning_rate": 8.194884983298815e-06,
      "loss": 0.0607,
      "step": 31540
    },
    {
      "epoch": 3.0078887939453124e-05,
      "model_forward_time": 1.706465244293213,
      "step": 31540
    },
    {
      "dynamics_loss_avg": 0.07644505612552166,
      "epoch": 3.0078887939453124e-05,
      "step": 31540
    },
    {
      "action_loss_avg": 0.0026631381595507266,
      "epoch": 3.0078887939453124e-05,
      "step": 31540
    },
    {
      "epoch": 3.0078887939453124e-05,
      "step": 31540,
      "training_step_time": 7.93860650062561
    },
    {
      "epoch": 3.007984161376953e-05,
      "model_forward_time": 1.7224762439727783,
      "step": 31541
    },
    {
      "epoch": 3.007984161376953e-05,
      "step": 31541,
      "training_step_time": 7.734474182128906
    },
    {
      "epoch": 3.008079528808594e-05,
      "model_forward_time": 1.3565218448638916,
      "step": 31542
    },
    {
      "epoch": 3.008079528808594e-05,
      "step": 31542,
      "training_step_time": 7.864084720611572
    },
    {
      "epoch": 3.0081748962402343e-05,
      "model_forward_time": 2.8227150440216064,
      "step": 31543
    },
    {
      "epoch": 3.0081748962402343e-05,
      "step": 31543,
      "training_step_time": 8.753150463104248
    },
    {
      "epoch": 3.008270263671875e-05,
      "model_forward_time": 1.7180078029632568,
      "step": 31544
    },
    {
      "epoch": 3.008270263671875e-05,
      "step": 31544,
      "training_step_time": 8.380872249603271
    },
    {
      "epoch": 3.0083656311035158e-05,
      "model_forward_time": 1.7625043392181396,
      "step": 31545
    },
    {
      "epoch": 3.0083656311035158e-05,
      "step": 31545,
      "training_step_time": 8.306689500808716
    },
    {
      "epoch": 3.0084609985351562e-05,
      "model_forward_time": 1.7174177169799805,
      "step": 31546
    },
    {
      "epoch": 3.0084609985351562e-05,
      "step": 31546,
      "training_step_time": 7.854674816131592
    },
    {
      "epoch": 3.008556365966797e-05,
      "model_forward_time": 1.7210311889648438,
      "step": 31547
    },
    {
      "epoch": 3.008556365966797e-05,
      "step": 31547,
      "training_step_time": 7.95884895324707
    },
    {
      "epoch": 3.0086517333984373e-05,
      "model_forward_time": 1.7245404720306396,
      "step": 31548
    },
    {
      "epoch": 3.0086517333984373e-05,
      "step": 31548,
      "training_step_time": 7.5978803634643555
    },
    {
      "epoch": 3.008747100830078e-05,
      "model_forward_time": 1.7140312194824219,
      "step": 31549
    },
    {
      "epoch": 3.008747100830078e-05,
      "step": 31549,
      "training_step_time": 8.067428827285767
    },
    {
      "epoch": 3.0088424682617188e-05,
      "grad_norm": 0.133378267288208,
      "learning_rate": 8.19361291691349e-06,
      "loss": 0.0648,
      "step": 31550
    },
    {
      "epoch": 3.0088424682617188e-05,
      "model_forward_time": 1.7078022956848145,
      "step": 31550
    },
    {
      "dynamics_loss_avg": 0.041645352728664875,
      "epoch": 3.0088424682617188e-05,
      "step": 31550
    },
    {
      "action_loss_avg": 0.0009928963088896125,
      "epoch": 3.0088424682617188e-05,
      "step": 31550
    },
    {
      "epoch": 3.0088424682617188e-05,
      "step": 31550,
      "training_step_time": 9.279154777526855
    },
    {
      "epoch": 3.0089378356933592e-05,
      "model_forward_time": 1.696932315826416,
      "step": 31551
    },
    {
      "epoch": 3.0089378356933592e-05,
      "step": 31551,
      "training_step_time": 8.721408367156982
    },
    {
      "epoch": 3.009033203125e-05,
      "model_forward_time": 2.7992665767669678,
      "step": 31552
    },
    {
      "epoch": 3.009033203125e-05,
      "step": 31552,
      "training_step_time": 10.34743356704712
    },
    {
      "epoch": 3.0091285705566407e-05,
      "model_forward_time": 1.7232041358947754,
      "step": 31553
    },
    {
      "epoch": 3.0091285705566407e-05,
      "step": 31553,
      "training_step_time": 7.9855334758758545
    },
    {
      "epoch": 3.009223937988281e-05,
      "model_forward_time": 1.7382769584655762,
      "step": 31554
    },
    {
      "epoch": 3.009223937988281e-05,
      "step": 31554,
      "training_step_time": 8.870256185531616
    },
    {
      "epoch": 3.009319305419922e-05,
      "model_forward_time": 1.720477819442749,
      "step": 31555
    },
    {
      "epoch": 3.009319305419922e-05,
      "step": 31555,
      "training_step_time": 8.483342409133911
    },
    {
      "epoch": 3.0094146728515626e-05,
      "model_forward_time": 1.718869924545288,
      "step": 31556
    },
    {
      "epoch": 3.0094146728515626e-05,
      "step": 31556,
      "training_step_time": 9.730482339859009
    },
    {
      "epoch": 3.009510040283203e-05,
      "model_forward_time": 1.7173316478729248,
      "step": 31557
    },
    {
      "epoch": 3.009510040283203e-05,
      "step": 31557,
      "training_step_time": 7.581280946731567
    },
    {
      "epoch": 3.0096054077148437e-05,
      "model_forward_time": 1.7164416313171387,
      "step": 31558
    },
    {
      "epoch": 3.0096054077148437e-05,
      "step": 31558,
      "training_step_time": 7.922058820724487
    },
    {
      "epoch": 3.0097007751464845e-05,
      "model_forward_time": 1.7155742645263672,
      "step": 31559
    },
    {
      "epoch": 3.0097007751464845e-05,
      "step": 31559,
      "training_step_time": 7.672111988067627
    },
    {
      "epoch": 3.009796142578125e-05,
      "grad_norm": 0.14498700201511383,
      "learning_rate": 8.192340501279464e-06,
      "loss": 0.0579,
      "step": 31560
    },
    {
      "epoch": 3.009796142578125e-05,
      "model_forward_time": 1.7183446884155273,
      "step": 31560
    },
    {
      "dynamics_loss_avg": 0.04851927608251572,
      "epoch": 3.009796142578125e-05,
      "step": 31560
    },
    {
      "action_loss_avg": 0.001847319380613044,
      "epoch": 3.009796142578125e-05,
      "step": 31560
    },
    {
      "epoch": 3.009796142578125e-05,
      "step": 31560,
      "training_step_time": 7.824496030807495
    },
    {
      "epoch": 3.0098915100097656e-05,
      "model_forward_time": 2.816556453704834,
      "step": 31561
    },
    {
      "epoch": 3.0098915100097656e-05,
      "step": 31561,
      "training_step_time": 7.786334276199341
    },
    {
      "epoch": 3.0099868774414064e-05,
      "model_forward_time": 1.718191385269165,
      "step": 31562
    },
    {
      "epoch": 3.0099868774414064e-05,
      "step": 31562,
      "training_step_time": 8.592910528182983
    },
    {
      "epoch": 3.0100822448730468e-05,
      "model_forward_time": 1.7531731128692627,
      "step": 31563
    },
    {
      "epoch": 3.0100822448730468e-05,
      "step": 31563,
      "training_step_time": 7.8055524826049805
    },
    {
      "epoch": 3.0101776123046875e-05,
      "model_forward_time": 1.7169570922851562,
      "step": 31564
    },
    {
      "epoch": 3.0101776123046875e-05,
      "step": 31564,
      "training_step_time": 8.883527278900146
    },
    {
      "epoch": 3.0102729797363282e-05,
      "model_forward_time": 1.709761142730713,
      "step": 31565
    },
    {
      "epoch": 3.0102729797363282e-05,
      "step": 31565,
      "training_step_time": 8.901102304458618
    },
    {
      "epoch": 3.0103683471679686e-05,
      "model_forward_time": 1.71384859085083,
      "step": 31566
    },
    {
      "epoch": 3.0103683471679686e-05,
      "step": 31566,
      "training_step_time": 7.805877447128296
    },
    {
      "epoch": 3.0104637145996094e-05,
      "model_forward_time": 1.7171940803527832,
      "step": 31567
    },
    {
      "epoch": 3.0104637145996094e-05,
      "step": 31567,
      "training_step_time": 8.062829494476318
    },
    {
      "epoch": 3.01055908203125e-05,
      "model_forward_time": 1.7064361572265625,
      "step": 31568
    },
    {
      "epoch": 3.01055908203125e-05,
      "step": 31568,
      "training_step_time": 7.555817604064941
    },
    {
      "epoch": 3.0106544494628905e-05,
      "model_forward_time": 1.7170605659484863,
      "step": 31569
    },
    {
      "epoch": 3.0106544494628905e-05,
      "step": 31569,
      "training_step_time": 7.861233472824097
    },
    {
      "epoch": 3.0107498168945313e-05,
      "grad_norm": 0.20357128977775574,
      "learning_rate": 8.191067736535876e-06,
      "loss": 0.057,
      "step": 31570
    },
    {
      "epoch": 3.0107498168945313e-05,
      "model_forward_time": 2.4380831718444824,
      "step": 31570
    },
    {
      "dynamics_loss_avg": 0.05432132016867399,
      "epoch": 3.0107498168945313e-05,
      "step": 31570
    },
    {
      "action_loss_avg": 0.002167790534440428,
      "epoch": 3.0107498168945313e-05,
      "step": 31570
    },
    {
      "epoch": 3.0107498168945313e-05,
      "step": 31570,
      "training_step_time": 8.300492763519287
    },
    {
      "epoch": 3.010845184326172e-05,
      "model_forward_time": 1.7168333530426025,
      "step": 31571
    },
    {
      "epoch": 3.010845184326172e-05,
      "step": 31571,
      "training_step_time": 7.830399990081787
    },
    {
      "epoch": 3.0109405517578124e-05,
      "model_forward_time": 1.7386903762817383,
      "step": 31572
    },
    {
      "epoch": 3.0109405517578124e-05,
      "step": 31572,
      "training_step_time": 9.35063099861145
    },
    {
      "epoch": 3.011035919189453e-05,
      "model_forward_time": 1.7223107814788818,
      "step": 31573
    },
    {
      "epoch": 3.011035919189453e-05,
      "step": 31573,
      "training_step_time": 8.326959133148193
    },
    {
      "epoch": 3.011131286621094e-05,
      "model_forward_time": 1.7173922061920166,
      "step": 31574
    },
    {
      "epoch": 3.011131286621094e-05,
      "step": 31574,
      "training_step_time": 8.480466365814209
    },
    {
      "epoch": 3.0112266540527343e-05,
      "model_forward_time": 1.7180626392364502,
      "step": 31575
    },
    {
      "epoch": 3.0112266540527343e-05,
      "step": 31575,
      "training_step_time": 7.909536123275757
    },
    {
      "epoch": 3.011322021484375e-05,
      "model_forward_time": 1.7207458019256592,
      "step": 31576
    },
    {
      "epoch": 3.011322021484375e-05,
      "step": 31576,
      "training_step_time": 8.005149602890015
    },
    {
      "epoch": 3.0114173889160158e-05,
      "model_forward_time": 1.7276086807250977,
      "step": 31577
    },
    {
      "epoch": 3.0114173889160158e-05,
      "step": 31577,
      "training_step_time": 7.216816186904907
    },
    {
      "epoch": 3.0115127563476562e-05,
      "model_forward_time": 1.7252826690673828,
      "step": 31578
    },
    {
      "epoch": 3.0115127563476562e-05,
      "step": 31578,
      "training_step_time": 8.5928373336792
    },
    {
      "epoch": 3.011608123779297e-05,
      "model_forward_time": 1.7386376857757568,
      "step": 31579
    },
    {
      "epoch": 3.011608123779297e-05,
      "step": 31579,
      "training_step_time": 8.210213422775269
    },
    {
      "epoch": 3.0117034912109373e-05,
      "grad_norm": 0.11134441196918488,
      "learning_rate": 8.18979462282192e-06,
      "loss": 0.0591,
      "step": 31580
    },
    {
      "epoch": 3.0117034912109373e-05,
      "model_forward_time": 2.8770787715911865,
      "step": 31580
    },
    {
      "dynamics_loss_avg": 0.06582768671214581,
      "epoch": 3.0117034912109373e-05,
      "step": 31580
    },
    {
      "action_loss_avg": 0.0020604112069122493,
      "epoch": 3.0117034912109373e-05,
      "step": 31580
    },
    {
      "epoch": 3.0117034912109373e-05,
      "step": 31580,
      "training_step_time": 8.868135213851929
    },
    {
      "epoch": 3.011798858642578e-05,
      "model_forward_time": 1.7168104648590088,
      "step": 31581
    },
    {
      "epoch": 3.011798858642578e-05,
      "step": 31581,
      "training_step_time": 8.641318559646606
    },
    {
      "epoch": 3.0118942260742188e-05,
      "model_forward_time": 1.7074742317199707,
      "step": 31582
    },
    {
      "epoch": 3.0118942260742188e-05,
      "step": 31582,
      "training_step_time": 8.029672622680664
    },
    {
      "epoch": 3.0119895935058592e-05,
      "model_forward_time": 1.7173337936401367,
      "step": 31583
    },
    {
      "epoch": 3.0119895935058592e-05,
      "step": 31583,
      "training_step_time": 7.813345432281494
    },
    {
      "epoch": 3.0120849609375e-05,
      "model_forward_time": 1.7211956977844238,
      "step": 31584
    },
    {
      "epoch": 3.0120849609375e-05,
      "step": 31584,
      "training_step_time": 7.957194805145264
    },
    {
      "epoch": 3.0121803283691407e-05,
      "model_forward_time": 1.7206406593322754,
      "step": 31585
    },
    {
      "epoch": 3.0121803283691407e-05,
      "step": 31585,
      "training_step_time": 7.953373432159424
    },
    {
      "epoch": 3.012275695800781e-05,
      "model_forward_time": 1.7176964282989502,
      "step": 31586
    },
    {
      "epoch": 3.012275695800781e-05,
      "step": 31586,
      "training_step_time": 7.329669237136841
    },
    {
      "epoch": 3.012371063232422e-05,
      "model_forward_time": 1.7238504886627197,
      "step": 31587
    },
    {
      "epoch": 3.012371063232422e-05,
      "step": 31587,
      "training_step_time": 7.9139814376831055
    },
    {
      "epoch": 3.0124664306640626e-05,
      "model_forward_time": 1.7646088600158691,
      "step": 31588
    },
    {
      "epoch": 3.0124664306640626e-05,
      "step": 31588,
      "training_step_time": 7.82492208480835
    },
    {
      "epoch": 3.012561798095703e-05,
      "model_forward_time": 2.953195095062256,
      "step": 31589
    },
    {
      "epoch": 3.012561798095703e-05,
      "step": 31589,
      "training_step_time": 7.951970815658569
    },
    {
      "epoch": 3.0126571655273437e-05,
      "grad_norm": 0.1892339438199997,
      "learning_rate": 8.188521160276818e-06,
      "loss": 0.0573,
      "step": 31590
    },
    {
      "epoch": 3.0126571655273437e-05,
      "model_forward_time": 1.7377400398254395,
      "step": 31590
    },
    {
      "dynamics_loss_avg": 0.06736465953290463,
      "epoch": 3.0126571655273437e-05,
      "step": 31590
    },
    {
      "action_loss_avg": 0.0025590162025764585,
      "epoch": 3.0126571655273437e-05,
      "step": 31590
    },
    {
      "epoch": 3.0126571655273437e-05,
      "step": 31590,
      "training_step_time": 8.430359363555908
    },
    {
      "epoch": 3.0127525329589845e-05,
      "model_forward_time": 1.763733148574829,
      "step": 31591
    },
    {
      "epoch": 3.0127525329589845e-05,
      "step": 31591,
      "training_step_time": 7.913081407546997
    },
    {
      "epoch": 3.012847900390625e-05,
      "model_forward_time": 1.7419440746307373,
      "step": 31592
    },
    {
      "epoch": 3.012847900390625e-05,
      "step": 31592,
      "training_step_time": 7.945254564285278
    },
    {
      "epoch": 3.0129432678222656e-05,
      "model_forward_time": 1.729102373123169,
      "step": 31593
    },
    {
      "epoch": 3.0129432678222656e-05,
      "step": 31593,
      "training_step_time": 7.963835954666138
    },
    {
      "epoch": 3.0130386352539064e-05,
      "model_forward_time": 1.7358348369598389,
      "step": 31594
    },
    {
      "epoch": 3.0130386352539064e-05,
      "step": 31594,
      "training_step_time": 8.126819372177124
    },
    {
      "epoch": 3.0131340026855468e-05,
      "model_forward_time": 1.7285079956054688,
      "step": 31595
    },
    {
      "epoch": 3.0131340026855468e-05,
      "step": 31595,
      "training_step_time": 7.351445436477661
    },
    {
      "epoch": 3.0132293701171875e-05,
      "model_forward_time": 1.7437820434570312,
      "step": 31596
    },
    {
      "epoch": 3.0132293701171875e-05,
      "step": 31596,
      "training_step_time": 7.822535753250122
    },
    {
      "epoch": 3.0133247375488282e-05,
      "model_forward_time": 1.0392754077911377,
      "step": 31597
    },
    {
      "epoch": 3.0133247375488282e-05,
      "step": 31597,
      "training_step_time": 7.80663800239563
    },
    {
      "epoch": 3.0134201049804687e-05,
      "model_forward_time": 2.8379440307617188,
      "step": 31598
    },
    {
      "epoch": 3.0134201049804687e-05,
      "step": 31598,
      "training_step_time": 8.362343072891235
    },
    {
      "epoch": 3.0135154724121094e-05,
      "model_forward_time": 1.7487223148345947,
      "step": 31599
    },
    {
      "epoch": 3.0135154724121094e-05,
      "step": 31599,
      "training_step_time": 9.441613674163818
    },
    {
      "epoch": 3.01361083984375e-05,
      "grad_norm": 0.11849971860647202,
      "learning_rate": 8.187247349039837e-06,
      "loss": 0.0583,
      "step": 31600
    },
    {
      "epoch": 3.01361083984375e-05,
      "model_forward_time": 1.7577643394470215,
      "step": 31600
    },
    {
      "dynamics_loss_avg": 0.046497960761189464,
      "epoch": 3.01361083984375e-05,
      "step": 31600
    },
    {
      "action_loss_avg": 0.0022087643010308966,
      "epoch": 3.01361083984375e-05,
      "step": 31600
    },
    {
      "epoch": 3.01361083984375e-05,
      "step": 31600,
      "training_step_time": 7.806359767913818
    },
    {
      "epoch": 3.0137062072753905e-05,
      "model_forward_time": 1.7409663200378418,
      "step": 31601
    },
    {
      "epoch": 3.0137062072753905e-05,
      "step": 31601,
      "training_step_time": 7.806312084197998
    },
    {
      "epoch": 3.0138015747070313e-05,
      "model_forward_time": 1.736497163772583,
      "step": 31602
    },
    {
      "epoch": 3.0138015747070313e-05,
      "step": 31602,
      "training_step_time": 7.862088680267334
    },
    {
      "epoch": 3.013896942138672e-05,
      "model_forward_time": 1.7437613010406494,
      "step": 31603
    },
    {
      "epoch": 3.013896942138672e-05,
      "step": 31603,
      "training_step_time": 9.089785099029541
    },
    {
      "epoch": 3.0139923095703124e-05,
      "model_forward_time": 1.7390131950378418,
      "step": 31604
    },
    {
      "epoch": 3.0139923095703124e-05,
      "step": 31604,
      "training_step_time": 7.365910530090332
    },
    {
      "epoch": 3.014087677001953e-05,
      "model_forward_time": 1.747112512588501,
      "step": 31605
    },
    {
      "epoch": 3.014087677001953e-05,
      "step": 31605,
      "training_step_time": 8.781927585601807
    },
    {
      "epoch": 3.014183044433594e-05,
      "model_forward_time": 1.756889820098877,
      "step": 31606
    },
    {
      "epoch": 3.014183044433594e-05,
      "step": 31606,
      "training_step_time": 7.808279275894165
    },
    {
      "epoch": 3.0142784118652343e-05,
      "model_forward_time": 1.7395260334014893,
      "step": 31607
    },
    {
      "epoch": 3.0142784118652343e-05,
      "step": 31607,
      "training_step_time": 7.814626216888428
    },
    {
      "epoch": 3.014373779296875e-05,
      "model_forward_time": 2.942335605621338,
      "step": 31608
    },
    {
      "epoch": 3.014373779296875e-05,
      "step": 31608,
      "training_step_time": 9.006423950195312
    },
    {
      "epoch": 3.0144691467285158e-05,
      "model_forward_time": 1.7706542015075684,
      "step": 31609
    },
    {
      "epoch": 3.0144691467285158e-05,
      "step": 31609,
      "training_step_time": 9.701480627059937
    },
    {
      "epoch": 3.0145645141601562e-05,
      "grad_norm": 0.14213576912879944,
      "learning_rate": 8.18597318925028e-06,
      "loss": 0.0604,
      "step": 31610
    },
    {
      "epoch": 3.0145645141601562e-05,
      "model_forward_time": 1.743077278137207,
      "step": 31610
    },
    {
      "dynamics_loss_avg": 0.06426881700754165,
      "epoch": 3.0145645141601562e-05,
      "step": 31610
    },
    {
      "action_loss_avg": 0.0019928972003981473,
      "epoch": 3.0145645141601562e-05,
      "step": 31610
    },
    {
      "epoch": 3.0145645141601562e-05,
      "step": 31610,
      "training_step_time": 8.588232040405273
    },
    {
      "epoch": 3.014659881591797e-05,
      "model_forward_time": 1.733863115310669,
      "step": 31611
    },
    {
      "epoch": 3.014659881591797e-05,
      "step": 31611,
      "training_step_time": 9.294447183609009
    },
    {
      "epoch": 3.0147552490234373e-05,
      "model_forward_time": 1.747553825378418,
      "step": 31612
    },
    {
      "epoch": 3.0147552490234373e-05,
      "step": 31612,
      "training_step_time": 8.749706506729126
    },
    {
      "epoch": 3.014850616455078e-05,
      "model_forward_time": 1.7491590976715088,
      "step": 31613
    },
    {
      "epoch": 3.014850616455078e-05,
      "step": 31613,
      "training_step_time": 7.815234422683716
    },
    {
      "epoch": 3.0149459838867188e-05,
      "model_forward_time": 1.7364959716796875,
      "step": 31614
    },
    {
      "epoch": 3.0149459838867188e-05,
      "step": 31614,
      "training_step_time": 8.284393072128296
    },
    {
      "epoch": 3.0150413513183592e-05,
      "model_forward_time": 1.745164155960083,
      "step": 31615
    },
    {
      "epoch": 3.0150413513183592e-05,
      "step": 31615,
      "training_step_time": 8.79400110244751
    },
    {
      "epoch": 3.01513671875e-05,
      "model_forward_time": 1.740950345993042,
      "step": 31616
    },
    {
      "epoch": 3.01513671875e-05,
      "step": 31616,
      "training_step_time": 8.362460136413574
    },
    {
      "epoch": 3.0152320861816407e-05,
      "model_forward_time": 2.926934242248535,
      "step": 31617
    },
    {
      "epoch": 3.0152320861816407e-05,
      "step": 31617,
      "training_step_time": 9.90737271308899
    },
    {
      "epoch": 3.015327453613281e-05,
      "model_forward_time": 1.7586617469787598,
      "step": 31618
    },
    {
      "epoch": 3.015327453613281e-05,
      "step": 31618,
      "training_step_time": 8.648835897445679
    },
    {
      "epoch": 3.015422821044922e-05,
      "model_forward_time": 1.7371652126312256,
      "step": 31619
    },
    {
      "epoch": 3.015422821044922e-05,
      "step": 31619,
      "training_step_time": 7.798925161361694
    },
    {
      "epoch": 3.0155181884765626e-05,
      "grad_norm": 0.2522779107093811,
      "learning_rate": 8.184698681047482e-06,
      "loss": 0.0627,
      "step": 31620
    },
    {
      "epoch": 3.0155181884765626e-05,
      "model_forward_time": 1.7424805164337158,
      "step": 31620
    },
    {
      "dynamics_loss_avg": 0.06492916811257601,
      "epoch": 3.0155181884765626e-05,
      "step": 31620
    },
    {
      "action_loss_avg": 0.0023416537907905875,
      "epoch": 3.0155181884765626e-05,
      "step": 31620
    },
    {
      "epoch": 3.0155181884765626e-05,
      "step": 31620,
      "training_step_time": 7.8202736377716064
    },
    {
      "epoch": 3.015613555908203e-05,
      "model_forward_time": 1.7271595001220703,
      "step": 31621
    },
    {
      "epoch": 3.015613555908203e-05,
      "step": 31621,
      "training_step_time": 7.853828191757202
    },
    {
      "epoch": 3.0157089233398437e-05,
      "model_forward_time": 1.7437553405761719,
      "step": 31622
    },
    {
      "epoch": 3.0157089233398437e-05,
      "step": 31622,
      "training_step_time": 7.794679403305054
    },
    {
      "epoch": 3.0158042907714845e-05,
      "model_forward_time": 1.737973928451538,
      "step": 31623
    },
    {
      "epoch": 3.0158042907714845e-05,
      "step": 31623,
      "training_step_time": 7.719743728637695
    },
    {
      "epoch": 3.015899658203125e-05,
      "model_forward_time": 1.747999668121338,
      "step": 31624
    },
    {
      "epoch": 3.015899658203125e-05,
      "step": 31624,
      "training_step_time": 7.6886537075042725
    },
    {
      "epoch": 3.0159950256347656e-05,
      "model_forward_time": 1.7639315128326416,
      "step": 31625
    },
    {
      "epoch": 3.0159950256347656e-05,
      "step": 31625,
      "training_step_time": 7.612983226776123
    },
    {
      "epoch": 3.0160903930664064e-05,
      "model_forward_time": 2.9879608154296875,
      "step": 31626
    },
    {
      "epoch": 3.0160903930664064e-05,
      "step": 31626,
      "training_step_time": 7.899993181228638
    },
    {
      "epoch": 3.0161857604980468e-05,
      "model_forward_time": 1.7432923316955566,
      "step": 31627
    },
    {
      "epoch": 3.0161857604980468e-05,
      "step": 31627,
      "training_step_time": 9.360495805740356
    },
    {
      "epoch": 3.0162811279296875e-05,
      "model_forward_time": 1.7498369216918945,
      "step": 31628
    },
    {
      "epoch": 3.0162811279296875e-05,
      "step": 31628,
      "training_step_time": 9.278658151626587
    },
    {
      "epoch": 3.0163764953613283e-05,
      "model_forward_time": 1.7320172786712646,
      "step": 31629
    },
    {
      "epoch": 3.0163764953613283e-05,
      "step": 31629,
      "training_step_time": 7.839995622634888
    },
    {
      "epoch": 3.0164718627929687e-05,
      "grad_norm": 0.17386817932128906,
      "learning_rate": 8.183423824570826e-06,
      "loss": 0.0609,
      "step": 31630
    },
    {
      "epoch": 3.0164718627929687e-05,
      "model_forward_time": 1.7381646633148193,
      "step": 31630
    },
    {
      "dynamics_loss_avg": 0.06493067611008882,
      "epoch": 3.0164718627929687e-05,
      "step": 31630
    },
    {
      "action_loss_avg": 0.0018854663765523582,
      "epoch": 3.0164718627929687e-05,
      "step": 31630
    },
    {
      "epoch": 3.0164718627929687e-05,
      "step": 31630,
      "training_step_time": 7.774916172027588
    },
    {
      "epoch": 3.0165672302246094e-05,
      "model_forward_time": 1.7401654720306396,
      "step": 31631
    },
    {
      "epoch": 3.0165672302246094e-05,
      "step": 31631,
      "training_step_time": 7.907749891281128
    },
    {
      "epoch": 3.01666259765625e-05,
      "model_forward_time": 1.7420282363891602,
      "step": 31632
    },
    {
      "epoch": 3.01666259765625e-05,
      "step": 31632,
      "training_step_time": 7.650803089141846
    },
    {
      "epoch": 3.0167579650878905e-05,
      "model_forward_time": 1.7396602630615234,
      "step": 31633
    },
    {
      "epoch": 3.0167579650878905e-05,
      "step": 31633,
      "training_step_time": 8.518820762634277
    },
    {
      "epoch": 3.0168533325195313e-05,
      "model_forward_time": 1.7691738605499268,
      "step": 31634
    },
    {
      "epoch": 3.0168533325195313e-05,
      "step": 31634,
      "training_step_time": 8.100622177124023
    },
    {
      "epoch": 3.016948699951172e-05,
      "model_forward_time": 2.8925323486328125,
      "step": 31635
    },
    {
      "epoch": 3.016948699951172e-05,
      "step": 31635,
      "training_step_time": 7.759658336639404
    },
    {
      "epoch": 3.0170440673828124e-05,
      "model_forward_time": 1.748176097869873,
      "step": 31636
    },
    {
      "epoch": 3.0170440673828124e-05,
      "step": 31636,
      "training_step_time": 8.925530195236206
    },
    {
      "epoch": 3.017139434814453e-05,
      "model_forward_time": 1.77907395362854,
      "step": 31637
    },
    {
      "epoch": 3.017139434814453e-05,
      "step": 31637,
      "training_step_time": 7.826215505599976
    },
    {
      "epoch": 3.017234802246094e-05,
      "model_forward_time": 1.7355284690856934,
      "step": 31638
    },
    {
      "epoch": 3.017234802246094e-05,
      "step": 31638,
      "training_step_time": 8.427269697189331
    },
    {
      "epoch": 3.0173301696777343e-05,
      "model_forward_time": 1.7417290210723877,
      "step": 31639
    },
    {
      "epoch": 3.0173301696777343e-05,
      "step": 31639,
      "training_step_time": 8.483346939086914
    },
    {
      "epoch": 3.017425537109375e-05,
      "grad_norm": 0.13902513682842255,
      "learning_rate": 8.182148619959725e-06,
      "loss": 0.0631,
      "step": 31640
    },
    {
      "epoch": 3.017425537109375e-05,
      "model_forward_time": 1.7288110256195068,
      "step": 31640
    },
    {
      "dynamics_loss_avg": 0.061508013121783736,
      "epoch": 3.017425537109375e-05,
      "step": 31640
    },
    {
      "action_loss_avg": 0.0026483542460482568,
      "epoch": 3.017425537109375e-05,
      "step": 31640
    },
    {
      "epoch": 3.017425537109375e-05,
      "step": 31640,
      "training_step_time": 7.810096025466919
    },
    {
      "epoch": 3.0175209045410155e-05,
      "model_forward_time": 1.7493846416473389,
      "step": 31641
    },
    {
      "epoch": 3.0175209045410155e-05,
      "step": 31641,
      "training_step_time": 7.634202480316162
    },
    {
      "epoch": 3.0176162719726562e-05,
      "model_forward_time": 1.7376043796539307,
      "step": 31642
    },
    {
      "epoch": 3.0176162719726562e-05,
      "step": 31642,
      "training_step_time": 7.757547855377197
    },
    {
      "epoch": 3.017711639404297e-05,
      "model_forward_time": 1.7549970149993896,
      "step": 31643
    },
    {
      "epoch": 3.017711639404297e-05,
      "step": 31643,
      "training_step_time": 7.637010097503662
    },
    {
      "epoch": 3.0178070068359373e-05,
      "model_forward_time": 2.8988916873931885,
      "step": 31644
    },
    {
      "epoch": 3.0178070068359373e-05,
      "step": 31644,
      "training_step_time": 7.769491910934448
    },
    {
      "epoch": 3.017902374267578e-05,
      "model_forward_time": 1.7358129024505615,
      "step": 31645
    },
    {
      "epoch": 3.017902374267578e-05,
      "step": 31645,
      "training_step_time": 7.719485759735107
    },
    {
      "epoch": 3.0179977416992188e-05,
      "model_forward_time": 1.760495901107788,
      "step": 31646
    },
    {
      "epoch": 3.0179977416992188e-05,
      "step": 31646,
      "training_step_time": 7.767213582992554
    },
    {
      "epoch": 3.0180931091308592e-05,
      "model_forward_time": 1.7343051433563232,
      "step": 31647
    },
    {
      "epoch": 3.0180931091308592e-05,
      "step": 31647,
      "training_step_time": 8.718863010406494
    },
    {
      "epoch": 3.0181884765625e-05,
      "model_forward_time": 1.7355473041534424,
      "step": 31648
    },
    {
      "epoch": 3.0181884765625e-05,
      "step": 31648,
      "training_step_time": 8.822010517120361
    },
    {
      "epoch": 3.0182838439941407e-05,
      "model_forward_time": 1.732074499130249,
      "step": 31649
    },
    {
      "epoch": 3.0182838439941407e-05,
      "step": 31649,
      "training_step_time": 7.7410454750061035
    },
    {
      "epoch": 3.018379211425781e-05,
      "grad_norm": 0.15463382005691528,
      "learning_rate": 8.180873067353637e-06,
      "loss": 0.0601,
      "step": 31650
    },
    {
      "epoch": 3.018379211425781e-05,
      "model_forward_time": 1.7410547733306885,
      "step": 31650
    },
    {
      "dynamics_loss_avg": 0.03991431165486574,
      "epoch": 3.018379211425781e-05,
      "step": 31650
    },
    {
      "action_loss_avg": 0.0013509922922821716,
      "epoch": 3.018379211425781e-05,
      "step": 31650
    },
    {
      "epoch": 3.018379211425781e-05,
      "step": 31650,
      "training_step_time": 7.721873760223389
    },
    {
      "epoch": 3.018474578857422e-05,
      "model_forward_time": 1.7277109622955322,
      "step": 31651
    },
    {
      "epoch": 3.018474578857422e-05,
      "step": 31651,
      "training_step_time": 7.701282978057861
    },
    {
      "epoch": 3.0185699462890626e-05,
      "model_forward_time": 1.7399697303771973,
      "step": 31652
    },
    {
      "epoch": 3.0185699462890626e-05,
      "step": 31652,
      "training_step_time": 8.508243560791016
    },
    {
      "epoch": 3.018665313720703e-05,
      "model_forward_time": 2.458430051803589,
      "step": 31653
    },
    {
      "epoch": 3.018665313720703e-05,
      "step": 31653,
      "training_step_time": 8.934659719467163
    },
    {
      "epoch": 3.0187606811523437e-05,
      "model_forward_time": 1.7236683368682861,
      "step": 31654
    },
    {
      "epoch": 3.0187606811523437e-05,
      "step": 31654,
      "training_step_time": 7.8391430377960205
    },
    {
      "epoch": 3.0188560485839845e-05,
      "model_forward_time": 1.751570224761963,
      "step": 31655
    },
    {
      "epoch": 3.0188560485839845e-05,
      "step": 31655,
      "training_step_time": 8.572907209396362
    },
    {
      "epoch": 3.018951416015625e-05,
      "model_forward_time": 1.7590830326080322,
      "step": 31656
    },
    {
      "epoch": 3.018951416015625e-05,
      "step": 31656,
      "training_step_time": 7.9519078731536865
    },
    {
      "epoch": 3.0190467834472656e-05,
      "model_forward_time": 1.7357470989227295,
      "step": 31657
    },
    {
      "epoch": 3.0190467834472656e-05,
      "step": 31657,
      "training_step_time": 9.0388343334198
    },
    {
      "epoch": 3.0191421508789064e-05,
      "model_forward_time": 1.7393584251403809,
      "step": 31658
    },
    {
      "epoch": 3.0191421508789064e-05,
      "step": 31658,
      "training_step_time": 7.865679979324341
    },
    {
      "epoch": 3.0192375183105468e-05,
      "model_forward_time": 1.7354273796081543,
      "step": 31659
    },
    {
      "epoch": 3.0192375183105468e-05,
      "step": 31659,
      "training_step_time": 7.679560899734497
    },
    {
      "epoch": 3.0193328857421875e-05,
      "grad_norm": 0.20172418653964996,
      "learning_rate": 8.179597166892052e-06,
      "loss": 0.0597,
      "step": 31660
    },
    {
      "epoch": 3.0193328857421875e-05,
      "model_forward_time": 1.7456495761871338,
      "step": 31660
    },
    {
      "dynamics_loss_avg": 0.05783975757658481,
      "epoch": 3.0193328857421875e-05,
      "step": 31660
    },
    {
      "action_loss_avg": 0.0017025688895955682,
      "epoch": 3.0193328857421875e-05,
      "step": 31660
    },
    {
      "epoch": 3.0193328857421875e-05,
      "step": 31660,
      "training_step_time": 7.524073600769043
    },
    {
      "epoch": 3.0194282531738283e-05,
      "model_forward_time": 1.740535020828247,
      "step": 31661
    },
    {
      "epoch": 3.0194282531738283e-05,
      "step": 31661,
      "training_step_time": 8.5164213180542
    },
    {
      "epoch": 3.0195236206054687e-05,
      "model_forward_time": 1.8534183502197266,
      "step": 31662
    },
    {
      "epoch": 3.0195236206054687e-05,
      "step": 31662,
      "training_step_time": 7.606246709823608
    },
    {
      "epoch": 3.0196189880371094e-05,
      "model_forward_time": 2.8867409229278564,
      "step": 31663
    },
    {
      "epoch": 3.0196189880371094e-05,
      "step": 31663,
      "training_step_time": 7.781898021697998
    },
    {
      "epoch": 3.01971435546875e-05,
      "model_forward_time": 1.7441508769989014,
      "step": 31664
    },
    {
      "epoch": 3.01971435546875e-05,
      "step": 31664,
      "training_step_time": 7.69969916343689
    },
    {
      "epoch": 3.0198097229003905e-05,
      "model_forward_time": 1.7434754371643066,
      "step": 31665
    },
    {
      "epoch": 3.0198097229003905e-05,
      "step": 31665,
      "training_step_time": 8.54589033126831
    },
    {
      "epoch": 3.0199050903320313e-05,
      "model_forward_time": 1.7351207733154297,
      "step": 31666
    },
    {
      "epoch": 3.0199050903320313e-05,
      "step": 31666,
      "training_step_time": 8.585689783096313
    },
    {
      "epoch": 3.020000457763672e-05,
      "model_forward_time": 1.741590976715088,
      "step": 31667
    },
    {
      "epoch": 3.020000457763672e-05,
      "step": 31667,
      "training_step_time": 7.989497661590576
    },
    {
      "epoch": 3.0200958251953124e-05,
      "model_forward_time": 1.7356047630310059,
      "step": 31668
    },
    {
      "epoch": 3.0200958251953124e-05,
      "step": 31668,
      "training_step_time": 8.351973295211792
    },
    {
      "epoch": 3.0201911926269532e-05,
      "model_forward_time": 1.7462749481201172,
      "step": 31669
    },
    {
      "epoch": 3.0201911926269532e-05,
      "step": 31669,
      "training_step_time": 7.797008037567139
    },
    {
      "epoch": 3.020286560058594e-05,
      "grad_norm": 0.19630040228366852,
      "learning_rate": 8.178320918714502e-06,
      "loss": 0.063,
      "step": 31670
    },
    {
      "epoch": 3.020286560058594e-05,
      "model_forward_time": 1.7439298629760742,
      "step": 31670
    },
    {
      "dynamics_loss_avg": 0.07008000016212464,
      "epoch": 3.020286560058594e-05,
      "step": 31670
    },
    {
      "action_loss_avg": 0.00261458259774372,
      "epoch": 3.020286560058594e-05,
      "step": 31670
    },
    {
      "epoch": 3.020286560058594e-05,
      "step": 31670,
      "training_step_time": 8.424056053161621
    },
    {
      "epoch": 3.0203819274902343e-05,
      "model_forward_time": 1.848071813583374,
      "step": 31671
    },
    {
      "epoch": 3.0203819274902343e-05,
      "step": 31671,
      "training_step_time": 8.551535844802856
    },
    {
      "epoch": 3.020477294921875e-05,
      "model_forward_time": 2.8691554069519043,
      "step": 31672
    },
    {
      "epoch": 3.020477294921875e-05,
      "step": 31672,
      "training_step_time": 8.867624759674072
    },
    {
      "epoch": 3.0205726623535155e-05,
      "model_forward_time": 1.749013900756836,
      "step": 31673
    },
    {
      "epoch": 3.0205726623535155e-05,
      "step": 31673,
      "training_step_time": 8.745510578155518
    },
    {
      "epoch": 3.0206680297851562e-05,
      "model_forward_time": 1.7428388595581055,
      "step": 31674
    },
    {
      "epoch": 3.0206680297851562e-05,
      "step": 31674,
      "training_step_time": 7.523876667022705
    },
    {
      "epoch": 3.020763397216797e-05,
      "model_forward_time": 1.7431814670562744,
      "step": 31675
    },
    {
      "epoch": 3.020763397216797e-05,
      "step": 31675,
      "training_step_time": 7.796262502670288
    },
    {
      "epoch": 3.0208587646484373e-05,
      "model_forward_time": 1.734929084777832,
      "step": 31676
    },
    {
      "epoch": 3.0208587646484373e-05,
      "step": 31676,
      "training_step_time": 8.067054748535156
    },
    {
      "epoch": 3.020954132080078e-05,
      "model_forward_time": 1.734121561050415,
      "step": 31677
    },
    {
      "epoch": 3.020954132080078e-05,
      "step": 31677,
      "training_step_time": 7.771491050720215
    },
    {
      "epoch": 3.0210494995117188e-05,
      "model_forward_time": 1.7551329135894775,
      "step": 31678
    },
    {
      "epoch": 3.0210494995117188e-05,
      "step": 31678,
      "training_step_time": 7.832465171813965
    },
    {
      "epoch": 3.0211448669433592e-05,
      "model_forward_time": 1.7401175498962402,
      "step": 31679
    },
    {
      "epoch": 3.0211448669433592e-05,
      "step": 31679,
      "training_step_time": 7.757103443145752
    },
    {
      "epoch": 3.021240234375e-05,
      "grad_norm": 0.17083227634429932,
      "learning_rate": 8.177044322960554e-06,
      "loss": 0.0587,
      "step": 31680
    },
    {
      "epoch": 3.021240234375e-05,
      "model_forward_time": 1.8368284702301025,
      "step": 31680
    },
    {
      "dynamics_loss_avg": 0.04090636745095253,
      "epoch": 3.021240234375e-05,
      "step": 31680
    },
    {
      "action_loss_avg": 0.001920185866765678,
      "epoch": 3.021240234375e-05,
      "step": 31680
    },
    {
      "epoch": 3.021240234375e-05,
      "step": 31680,
      "training_step_time": 7.651148319244385
    },
    {
      "epoch": 3.0213356018066407e-05,
      "model_forward_time": 2.8916444778442383,
      "step": 31681
    },
    {
      "epoch": 3.0213356018066407e-05,
      "step": 31681,
      "training_step_time": 9.637166023254395
    },
    {
      "epoch": 3.021430969238281e-05,
      "model_forward_time": 1.7365446090698242,
      "step": 31682
    },
    {
      "epoch": 3.021430969238281e-05,
      "step": 31682,
      "training_step_time": 7.865136623382568
    },
    {
      "epoch": 3.021526336669922e-05,
      "model_forward_time": 1.7459888458251953,
      "step": 31683
    },
    {
      "epoch": 3.021526336669922e-05,
      "step": 31683,
      "training_step_time": 9.627583980560303
    },
    {
      "epoch": 3.0216217041015626e-05,
      "model_forward_time": 1.7396650314331055,
      "step": 31684
    },
    {
      "epoch": 3.0216217041015626e-05,
      "step": 31684,
      "training_step_time": 8.780287742614746
    },
    {
      "epoch": 3.021717071533203e-05,
      "model_forward_time": 1.7497270107269287,
      "step": 31685
    },
    {
      "epoch": 3.021717071533203e-05,
      "step": 31685,
      "training_step_time": 8.440492868423462
    },
    {
      "epoch": 3.0218124389648437e-05,
      "model_forward_time": 1.7395904064178467,
      "step": 31686
    },
    {
      "epoch": 3.0218124389648437e-05,
      "step": 31686,
      "training_step_time": 7.792536973953247
    },
    {
      "epoch": 3.0219078063964845e-05,
      "model_forward_time": 1.7231309413909912,
      "step": 31687
    },
    {
      "epoch": 3.0219078063964845e-05,
      "step": 31687,
      "training_step_time": 7.644067049026489
    },
    {
      "epoch": 3.022003173828125e-05,
      "model_forward_time": 1.7429308891296387,
      "step": 31688
    },
    {
      "epoch": 3.022003173828125e-05,
      "step": 31688,
      "training_step_time": 8.308130264282227
    },
    {
      "epoch": 3.0220985412597656e-05,
      "model_forward_time": 1.8161330223083496,
      "step": 31689
    },
    {
      "epoch": 3.0220985412597656e-05,
      "step": 31689,
      "training_step_time": 7.655878782272339
    },
    {
      "epoch": 3.0221939086914064e-05,
      "grad_norm": 0.14663130044937134,
      "learning_rate": 8.175767379769817e-06,
      "loss": 0.0613,
      "step": 31690
    },
    {
      "epoch": 3.0221939086914064e-05,
      "model_forward_time": 2.85585880279541,
      "step": 31690
    },
    {
      "dynamics_loss_avg": 0.05995971839874983,
      "epoch": 3.0221939086914064e-05,
      "step": 31690
    },
    {
      "action_loss_avg": 0.0018723649132880381,
      "epoch": 3.0221939086914064e-05,
      "step": 31690
    },
    {
      "epoch": 3.0221939086914064e-05,
      "step": 31690,
      "training_step_time": 7.8164567947387695
    },
    {
      "epoch": 3.0222892761230468e-05,
      "model_forward_time": 1.7393202781677246,
      "step": 31691
    },
    {
      "epoch": 3.0222892761230468e-05,
      "step": 31691,
      "training_step_time": 8.898270845413208
    },
    {
      "epoch": 3.0223846435546875e-05,
      "model_forward_time": 1.7619264125823975,
      "step": 31692
    },
    {
      "epoch": 3.0223846435546875e-05,
      "step": 31692,
      "training_step_time": 9.491898536682129
    },
    {
      "epoch": 3.0224800109863283e-05,
      "model_forward_time": 1.7414333820343018,
      "step": 31693
    },
    {
      "epoch": 3.0224800109863283e-05,
      "step": 31693,
      "training_step_time": 6.886816501617432
    },
    {
      "epoch": 3.0225753784179687e-05,
      "model_forward_time": 1.7184333801269531,
      "step": 31694
    },
    {
      "epoch": 3.0225753784179687e-05,
      "step": 31694,
      "training_step_time": 6.938995599746704
    },
    {
      "epoch": 3.0226707458496094e-05,
      "model_forward_time": 1.736741304397583,
      "step": 31695
    },
    {
      "epoch": 3.0226707458496094e-05,
      "step": 31695,
      "training_step_time": 6.900197744369507
    },
    {
      "epoch": 3.02276611328125e-05,
      "model_forward_time": 1.7416484355926514,
      "step": 31696
    },
    {
      "epoch": 3.02276611328125e-05,
      "step": 31696,
      "training_step_time": 6.899333715438843
    },
    {
      "epoch": 3.0228614807128905e-05,
      "model_forward_time": 1.7366948127746582,
      "step": 31697
    },
    {
      "epoch": 3.0228614807128905e-05,
      "step": 31697,
      "training_step_time": 6.932511568069458
    },
    {
      "epoch": 3.0229568481445313e-05,
      "model_forward_time": 1.7255635261535645,
      "step": 31698
    },
    {
      "epoch": 3.0229568481445313e-05,
      "step": 31698,
      "training_step_time": 6.91839861869812
    },
    {
      "epoch": 3.023052215576172e-05,
      "model_forward_time": 1.733144998550415,
      "step": 31699
    },
    {
      "epoch": 3.023052215576172e-05,
      "step": 31699,
      "training_step_time": 6.886788845062256
    },
    {
      "epoch": 3.0231475830078124e-05,
      "grad_norm": 0.1699400544166565,
      "learning_rate": 8.174490089281932e-06,
      "loss": 0.0594,
      "step": 31700
    },
    {
      "epoch": 3.0231475830078124e-05,
      "model_forward_time": 1.735823631286621,
      "step": 31700
    },
    {
      "dynamics_loss_avg": 0.053645457699894906,
      "epoch": 3.0231475830078124e-05,
      "step": 31700
    },
    {
      "action_loss_avg": 0.0020728293049614875,
      "epoch": 3.0231475830078124e-05,
      "step": 31700
    },
    {
      "epoch": 3.0231475830078124e-05,
      "step": 31700,
      "training_step_time": 6.829315900802612
    },
    {
      "epoch": 3.0232429504394532e-05,
      "model_forward_time": 2.8850550651550293,
      "step": 31701
    },
    {
      "epoch": 3.0232429504394532e-05,
      "step": 31701,
      "training_step_time": 7.831676483154297
    },
    {
      "epoch": 3.023338317871094e-05,
      "model_forward_time": 1.7462542057037354,
      "step": 31702
    },
    {
      "epoch": 3.023338317871094e-05,
      "step": 31702,
      "training_step_time": 7.969107151031494
    },
    {
      "epoch": 3.0234336853027343e-05,
      "model_forward_time": 1.7401771545410156,
      "step": 31703
    },
    {
      "epoch": 3.0234336853027343e-05,
      "step": 31703,
      "training_step_time": 6.895550012588501
    },
    {
      "epoch": 3.023529052734375e-05,
      "model_forward_time": 1.742358922958374,
      "step": 31704
    },
    {
      "epoch": 3.023529052734375e-05,
      "step": 31704,
      "training_step_time": 6.898415565490723
    },
    {
      "epoch": 3.0236244201660155e-05,
      "model_forward_time": 1.7385237216949463,
      "step": 31705
    },
    {
      "epoch": 3.0236244201660155e-05,
      "step": 31705,
      "training_step_time": 6.898846626281738
    },
    {
      "epoch": 3.0237197875976562e-05,
      "model_forward_time": 1.7411704063415527,
      "step": 31706
    },
    {
      "epoch": 3.0237197875976562e-05,
      "step": 31706,
      "training_step_time": 6.874954462051392
    },
    {
      "epoch": 3.023815155029297e-05,
      "model_forward_time": 1.7411305904388428,
      "step": 31707
    },
    {
      "epoch": 3.023815155029297e-05,
      "step": 31707,
      "training_step_time": 6.863753318786621
    },
    {
      "epoch": 3.0239105224609373e-05,
      "model_forward_time": 1.7478868961334229,
      "step": 31708
    },
    {
      "epoch": 3.0239105224609373e-05,
      "step": 31708,
      "training_step_time": 6.903712034225464
    },
    {
      "epoch": 3.024005889892578e-05,
      "model_forward_time": 1.7462356090545654,
      "step": 31709
    },
    {
      "epoch": 3.024005889892578e-05,
      "step": 31709,
      "training_step_time": 6.863550186157227
    },
    {
      "epoch": 3.024101257324219e-05,
      "grad_norm": 0.18286553025245667,
      "learning_rate": 8.173212451636585e-06,
      "loss": 0.0615,
      "step": 31710
    },
    {
      "epoch": 3.024101257324219e-05,
      "model_forward_time": 2.8807637691497803,
      "step": 31710
    },
    {
      "dynamics_loss_avg": 0.056627367809414866,
      "epoch": 3.024101257324219e-05,
      "step": 31710
    },
    {
      "action_loss_avg": 0.002028899057768285,
      "epoch": 3.024101257324219e-05,
      "step": 31710
    },
    {
      "epoch": 3.024101257324219e-05,
      "step": 31710,
      "training_step_time": 7.839284420013428
    },
    {
      "epoch": 3.0241966247558592e-05,
      "model_forward_time": 1.7463996410369873,
      "step": 31711
    },
    {
      "epoch": 3.0241966247558592e-05,
      "step": 31711,
      "training_step_time": 7.97327446937561
    },
    {
      "epoch": 3.0242919921875e-05,
      "model_forward_time": 1.7421762943267822,
      "step": 31712
    },
    {
      "epoch": 3.0242919921875e-05,
      "step": 31712,
      "training_step_time": 7.511950731277466
    },
    {
      "epoch": 3.0243873596191407e-05,
      "model_forward_time": 1.7512946128845215,
      "step": 31713
    },
    {
      "epoch": 3.0243873596191407e-05,
      "step": 31713,
      "training_step_time": 6.820278167724609
    },
    {
      "epoch": 3.024482727050781e-05,
      "model_forward_time": 1.7486793994903564,
      "step": 31714
    },
    {
      "epoch": 3.024482727050781e-05,
      "step": 31714,
      "training_step_time": 6.893857955932617
    },
    {
      "epoch": 3.024578094482422e-05,
      "model_forward_time": 1.7480113506317139,
      "step": 31715
    },
    {
      "epoch": 3.024578094482422e-05,
      "step": 31715,
      "training_step_time": 6.8780882358551025
    },
    {
      "epoch": 3.0246734619140626e-05,
      "model_forward_time": 1.737222671508789,
      "step": 31716
    },
    {
      "epoch": 3.0246734619140626e-05,
      "step": 31716,
      "training_step_time": 6.830574989318848
    },
    {
      "epoch": 3.024768829345703e-05,
      "model_forward_time": 1.7248561382293701,
      "step": 31717
    },
    {
      "epoch": 3.024768829345703e-05,
      "step": 31717,
      "training_step_time": 6.719537258148193
    },
    {
      "epoch": 3.0248641967773437e-05,
      "model_forward_time": 1.7359843254089355,
      "step": 31718
    },
    {
      "epoch": 3.0248641967773437e-05,
      "step": 31718,
      "training_step_time": 6.882831335067749
    },
    {
      "epoch": 3.0249595642089845e-05,
      "model_forward_time": 2.872709274291992,
      "step": 31719
    },
    {
      "epoch": 3.0249595642089845e-05,
      "step": 31719,
      "training_step_time": 7.803602933883667
    },
    {
      "epoch": 3.025054931640625e-05,
      "grad_norm": 0.1601119041442871,
      "learning_rate": 8.171934466973494e-06,
      "loss": 0.0608,
      "step": 31720
    },
    {
      "epoch": 3.025054931640625e-05,
      "model_forward_time": 1.7482459545135498,
      "step": 31720
    },
    {
      "dynamics_loss_avg": 0.05493312980979681,
      "epoch": 3.025054931640625e-05,
      "step": 31720
    },
    {
      "action_loss_avg": 0.0017227401083800941,
      "epoch": 3.025054931640625e-05,
      "step": 31720
    },
    {
      "epoch": 3.025054931640625e-05,
      "step": 31720,
      "training_step_time": 7.9513397216796875
    },
    {
      "epoch": 3.0251502990722656e-05,
      "model_forward_time": 1.738720417022705,
      "step": 31721
    },
    {
      "epoch": 3.0251502990722656e-05,
      "step": 31721,
      "training_step_time": 7.665802478790283
    },
    {
      "epoch": 3.0252456665039064e-05,
      "model_forward_time": 1.7312545776367188,
      "step": 31722
    },
    {
      "epoch": 3.0252456665039064e-05,
      "step": 31722,
      "training_step_time": 6.864028215408325
    },
    {
      "epoch": 3.0253410339355468e-05,
      "model_forward_time": 1.7343809604644775,
      "step": 31723
    },
    {
      "epoch": 3.0253410339355468e-05,
      "step": 31723,
      "training_step_time": 6.886829137802124
    },
    {
      "epoch": 3.0254364013671875e-05,
      "model_forward_time": 1.7266879081726074,
      "step": 31724
    },
    {
      "epoch": 3.0254364013671875e-05,
      "step": 31724,
      "training_step_time": 6.883814096450806
    },
    {
      "epoch": 3.0255317687988283e-05,
      "model_forward_time": 1.7414782047271729,
      "step": 31725
    },
    {
      "epoch": 3.0255317687988283e-05,
      "step": 31725,
      "training_step_time": 6.860524892807007
    },
    {
      "epoch": 3.0256271362304687e-05,
      "model_forward_time": 1.7301206588745117,
      "step": 31726
    },
    {
      "epoch": 3.0256271362304687e-05,
      "step": 31726,
      "training_step_time": 6.900656700134277
    },
    {
      "epoch": 3.0257225036621094e-05,
      "model_forward_time": 1.7382009029388428,
      "step": 31727
    },
    {
      "epoch": 3.0257225036621094e-05,
      "step": 31727,
      "training_step_time": 6.8621039390563965
    },
    {
      "epoch": 3.02581787109375e-05,
      "model_forward_time": 2.8609728813171387,
      "step": 31728
    },
    {
      "epoch": 3.02581787109375e-05,
      "step": 31728,
      "training_step_time": 7.762055158615112
    },
    {
      "epoch": 3.0259132385253905e-05,
      "model_forward_time": 1.741192102432251,
      "step": 31729
    },
    {
      "epoch": 3.0259132385253905e-05,
      "step": 31729,
      "training_step_time": 9.995208978652954
    },
    {
      "epoch": 3.0260086059570313e-05,
      "grad_norm": 0.18622605502605438,
      "learning_rate": 8.170656135432419e-06,
      "loss": 0.0612,
      "step": 31730
    },
    {
      "epoch": 3.0260086059570313e-05,
      "model_forward_time": 1.7400531768798828,
      "step": 31730
    },
    {
      "dynamics_loss_avg": 0.05772790824994445,
      "epoch": 3.0260086059570313e-05,
      "step": 31730
    },
    {
      "action_loss_avg": 0.0024027500476222485,
      "epoch": 3.0260086059570313e-05,
      "step": 31730
    },
    {
      "epoch": 3.0260086059570313e-05,
      "step": 31730,
      "training_step_time": 7.68407678604126
    },
    {
      "epoch": 3.026103973388672e-05,
      "model_forward_time": 1.7393443584442139,
      "step": 31731
    },
    {
      "epoch": 3.026103973388672e-05,
      "step": 31731,
      "training_step_time": 6.765134334564209
    },
    {
      "epoch": 3.0261993408203124e-05,
      "model_forward_time": 1.7321600914001465,
      "step": 31732
    },
    {
      "epoch": 3.0261993408203124e-05,
      "step": 31732,
      "training_step_time": 6.80328369140625
    },
    {
      "epoch": 3.0262947082519532e-05,
      "model_forward_time": 1.7374472618103027,
      "step": 31733
    },
    {
      "epoch": 3.0262947082519532e-05,
      "step": 31733,
      "training_step_time": 6.807072162628174
    },
    {
      "epoch": 3.026390075683594e-05,
      "model_forward_time": 1.7289824485778809,
      "step": 31734
    },
    {
      "epoch": 3.026390075683594e-05,
      "step": 31734,
      "training_step_time": 6.77541184425354
    },
    {
      "epoch": 3.0264854431152343e-05,
      "model_forward_time": 1.7386674880981445,
      "step": 31735
    },
    {
      "epoch": 3.0264854431152343e-05,
      "step": 31735,
      "training_step_time": 6.806185722351074
    },
    {
      "epoch": 3.026580810546875e-05,
      "model_forward_time": 1.7363667488098145,
      "step": 31736
    },
    {
      "epoch": 3.026580810546875e-05,
      "step": 31736,
      "training_step_time": 6.793016195297241
    },
    {
      "epoch": 3.0266761779785155e-05,
      "model_forward_time": 2.8749067783355713,
      "step": 31737
    },
    {
      "epoch": 3.0266761779785155e-05,
      "step": 31737,
      "training_step_time": 7.733252286911011
    },
    {
      "epoch": 3.0267715454101562e-05,
      "model_forward_time": 1.7243342399597168,
      "step": 31738
    },
    {
      "epoch": 3.0267715454101562e-05,
      "step": 31738,
      "training_step_time": 11.860224962234497
    },
    {
      "epoch": 3.026866912841797e-05,
      "model_forward_time": 1.7363812923431396,
      "step": 31739
    },
    {
      "epoch": 3.026866912841797e-05,
      "step": 31739,
      "training_step_time": 8.460837364196777
    },
    {
      "epoch": 3.0269622802734374e-05,
      "grad_norm": 0.18997734785079956,
      "learning_rate": 8.169377457153155e-06,
      "loss": 0.0643,
      "step": 31740
    },
    {
      "epoch": 3.0269622802734374e-05,
      "model_forward_time": 1.7403583526611328,
      "step": 31740
    },
    {
      "dynamics_loss_avg": 0.05767479091882706,
      "epoch": 3.0269622802734374e-05,
      "step": 31740
    },
    {
      "action_loss_avg": 0.00189230005198624,
      "epoch": 3.0269622802734374e-05,
      "step": 31740
    },
    {
      "epoch": 3.0269622802734374e-05,
      "step": 31740,
      "training_step_time": 6.751413106918335
    },
    {
      "epoch": 3.027057647705078e-05,
      "model_forward_time": 1.736983060836792,
      "step": 31741
    },
    {
      "epoch": 3.027057647705078e-05,
      "step": 31741,
      "training_step_time": 6.744931697845459
    },
    {
      "epoch": 3.027153015136719e-05,
      "model_forward_time": 1.7397651672363281,
      "step": 31742
    },
    {
      "epoch": 3.027153015136719e-05,
      "step": 31742,
      "training_step_time": 6.800216197967529
    },
    {
      "epoch": 3.0272483825683592e-05,
      "model_forward_time": 1.7372932434082031,
      "step": 31743
    },
    {
      "epoch": 3.0272483825683592e-05,
      "step": 31743,
      "training_step_time": 6.767383337020874
    },
    {
      "epoch": 3.02734375e-05,
      "model_forward_time": 1.7290961742401123,
      "step": 31744
    },
    {
      "epoch": 3.02734375e-05,
      "step": 31744,
      "training_step_time": 6.8244194984436035
    },
    {
      "epoch": 3.0274391174316407e-05,
      "model_forward_time": 1.7380280494689941,
      "step": 31745
    },
    {
      "epoch": 3.0274391174316407e-05,
      "step": 31745,
      "training_step_time": 6.803755283355713
    },
    {
      "epoch": 3.027534484863281e-05,
      "model_forward_time": 1.7264437675476074,
      "step": 31746
    },
    {
      "epoch": 3.027534484863281e-05,
      "step": 31746,
      "training_step_time": 7.263187646865845
    },
    {
      "epoch": 3.027629852294922e-05,
      "model_forward_time": 1.834871530532837,
      "step": 31747
    },
    {
      "epoch": 3.027629852294922e-05,
      "step": 31747,
      "training_step_time": 9.710196495056152
    },
    {
      "epoch": 3.0277252197265626e-05,
      "model_forward_time": 1.7523491382598877,
      "step": 31748
    },
    {
      "epoch": 3.0277252197265626e-05,
      "step": 31748,
      "training_step_time": 8.596071243286133
    },
    {
      "epoch": 3.027820587158203e-05,
      "model_forward_time": 1.7297418117523193,
      "step": 31749
    },
    {
      "epoch": 3.027820587158203e-05,
      "step": 31749,
      "training_step_time": 7.675867319107056
    },
    {
      "epoch": 3.0279159545898438e-05,
      "grad_norm": 0.14435890316963196,
      "learning_rate": 8.16809843227554e-06,
      "loss": 0.0636,
      "step": 31750
    },
    {
      "epoch": 3.0279159545898438e-05,
      "model_forward_time": 1.7390482425689697,
      "step": 31750
    },
    {
      "dynamics_loss_avg": 0.05216356590390205,
      "epoch": 3.0279159545898438e-05,
      "step": 31750
    },
    {
      "action_loss_avg": 0.001979075057897717,
      "epoch": 3.0279159545898438e-05,
      "step": 31750
    },
    {
      "epoch": 3.0279159545898438e-05,
      "step": 31750,
      "training_step_time": 6.854375123977661
    },
    {
      "epoch": 3.0280113220214845e-05,
      "model_forward_time": 1.3740098476409912,
      "step": 31751
    },
    {
      "epoch": 3.0280113220214845e-05,
      "step": 31751,
      "training_step_time": 6.848668336868286
    },
    {
      "epoch": 3.028106689453125e-05,
      "model_forward_time": 1.739163875579834,
      "step": 31752
    },
    {
      "epoch": 3.028106689453125e-05,
      "step": 31752,
      "training_step_time": 6.865251541137695
    },
    {
      "epoch": 3.0282020568847656e-05,
      "model_forward_time": 1.7345046997070312,
      "step": 31753
    },
    {
      "epoch": 3.0282020568847656e-05,
      "step": 31753,
      "training_step_time": 6.8613502979278564
    },
    {
      "epoch": 3.0282974243164064e-05,
      "model_forward_time": 1.7409355640411377,
      "step": 31754
    },
    {
      "epoch": 3.0282974243164064e-05,
      "step": 31754,
      "training_step_time": 6.844255685806274
    },
    {
      "epoch": 3.0283927917480468e-05,
      "model_forward_time": 1.734086513519287,
      "step": 31755
    },
    {
      "epoch": 3.0283927917480468e-05,
      "step": 31755,
      "training_step_time": 6.876591205596924
    },
    {
      "epoch": 3.0284881591796875e-05,
      "model_forward_time": 2.8270516395568848,
      "step": 31756
    },
    {
      "epoch": 3.0284881591796875e-05,
      "step": 31756,
      "training_step_time": 10.447384357452393
    },
    {
      "epoch": 3.0285835266113283e-05,
      "model_forward_time": 1.7311875820159912,
      "step": 31757
    },
    {
      "epoch": 3.0285835266113283e-05,
      "step": 31757,
      "training_step_time": 7.783928632736206
    },
    {
      "epoch": 3.0286788940429687e-05,
      "model_forward_time": 1.7301299571990967,
      "step": 31758
    },
    {
      "epoch": 3.0286788940429687e-05,
      "step": 31758,
      "training_step_time": 7.7077717781066895
    },
    {
      "epoch": 3.0287742614746094e-05,
      "model_forward_time": 1.738023042678833,
      "step": 31759
    },
    {
      "epoch": 3.0287742614746094e-05,
      "step": 31759,
      "training_step_time": 6.789129734039307
    },
    {
      "epoch": 3.02886962890625e-05,
      "grad_norm": 0.1530570685863495,
      "learning_rate": 8.166819060939441e-06,
      "loss": 0.0664,
      "step": 31760
    },
    {
      "epoch": 3.02886962890625e-05,
      "model_forward_time": 1.7451589107513428,
      "step": 31760
    },
    {
      "dynamics_loss_avg": 0.040846428275108336,
      "epoch": 3.02886962890625e-05,
      "step": 31760
    },
    {
      "action_loss_avg": 0.001330720860278234,
      "epoch": 3.02886962890625e-05,
      "step": 31760
    },
    {
      "epoch": 3.02886962890625e-05,
      "step": 31760,
      "training_step_time": 6.83390474319458
    },
    {
      "epoch": 3.0289649963378906e-05,
      "model_forward_time": 1.7296218872070312,
      "step": 31761
    },
    {
      "epoch": 3.0289649963378906e-05,
      "step": 31761,
      "training_step_time": 6.809430122375488
    },
    {
      "epoch": 3.0290603637695313e-05,
      "model_forward_time": 1.7373735904693604,
      "step": 31762
    },
    {
      "epoch": 3.0290603637695313e-05,
      "step": 31762,
      "training_step_time": 6.8136351108551025
    },
    {
      "epoch": 3.029155731201172e-05,
      "model_forward_time": 1.7397570610046387,
      "step": 31763
    },
    {
      "epoch": 3.029155731201172e-05,
      "step": 31763,
      "training_step_time": 6.769129276275635
    },
    {
      "epoch": 3.0292510986328124e-05,
      "model_forward_time": 1.738978385925293,
      "step": 31764
    },
    {
      "epoch": 3.0292510986328124e-05,
      "step": 31764,
      "training_step_time": 6.769359827041626
    },
    {
      "epoch": 3.0293464660644532e-05,
      "model_forward_time": 2.8440403938293457,
      "step": 31765
    },
    {
      "epoch": 3.0293464660644532e-05,
      "step": 31765,
      "training_step_time": 10.15884017944336
    },
    {
      "epoch": 3.0294418334960936e-05,
      "model_forward_time": 1.7390432357788086,
      "step": 31766
    },
    {
      "epoch": 3.0294418334960936e-05,
      "step": 31766,
      "training_step_time": 9.605992078781128
    },
    {
      "epoch": 3.0295372009277343e-05,
      "model_forward_time": 1.7296712398529053,
      "step": 31767
    },
    {
      "epoch": 3.0295372009277343e-05,
      "step": 31767,
      "training_step_time": 7.776026725769043
    },
    {
      "epoch": 3.029632568359375e-05,
      "model_forward_time": 1.733245849609375,
      "step": 31768
    },
    {
      "epoch": 3.029632568359375e-05,
      "step": 31768,
      "training_step_time": 6.896927356719971
    },
    {
      "epoch": 3.0297279357910155e-05,
      "model_forward_time": 1.7316079139709473,
      "step": 31769
    },
    {
      "epoch": 3.0297279357910155e-05,
      "step": 31769,
      "training_step_time": 6.915022134780884
    },
    {
      "epoch": 3.0298233032226562e-05,
      "grad_norm": 0.1599198281764984,
      "learning_rate": 8.165539343284773e-06,
      "loss": 0.0643,
      "step": 31770
    },
    {
      "epoch": 3.0298233032226562e-05,
      "model_forward_time": 1.7417123317718506,
      "step": 31770
    },
    {
      "dynamics_loss_avg": 0.0543407091870904,
      "epoch": 3.0298233032226562e-05,
      "step": 31770
    },
    {
      "action_loss_avg": 0.001697322295513004,
      "epoch": 3.0298233032226562e-05,
      "step": 31770
    },
    {
      "epoch": 3.0298233032226562e-05,
      "step": 31770,
      "training_step_time": 6.9443817138671875
    },
    {
      "epoch": 3.029918670654297e-05,
      "model_forward_time": 1.7367663383483887,
      "step": 31771
    },
    {
      "epoch": 3.029918670654297e-05,
      "step": 31771,
      "training_step_time": 6.943917512893677
    },
    {
      "epoch": 3.0300140380859374e-05,
      "model_forward_time": 1.7298381328582764,
      "step": 31772
    },
    {
      "epoch": 3.0300140380859374e-05,
      "step": 31772,
      "training_step_time": 6.930754661560059
    },
    {
      "epoch": 3.030109405517578e-05,
      "model_forward_time": 1.7355873584747314,
      "step": 31773
    },
    {
      "epoch": 3.030109405517578e-05,
      "step": 31773,
      "training_step_time": 6.9285993576049805
    },
    {
      "epoch": 3.030204772949219e-05,
      "model_forward_time": 2.8351776599884033,
      "step": 31774
    },
    {
      "epoch": 3.030204772949219e-05,
      "step": 31774,
      "training_step_time": 8.025240182876587
    },
    {
      "epoch": 3.0303001403808592e-05,
      "model_forward_time": 1.7290380001068115,
      "step": 31775
    },
    {
      "epoch": 3.0303001403808592e-05,
      "step": 31775,
      "training_step_time": 9.904228687286377
    },
    {
      "epoch": 3.0303955078125e-05,
      "model_forward_time": 1.7206735610961914,
      "step": 31776
    },
    {
      "epoch": 3.0303955078125e-05,
      "step": 31776,
      "training_step_time": 7.867670297622681
    },
    {
      "epoch": 3.0304908752441407e-05,
      "model_forward_time": 1.7327029705047607,
      "step": 31777
    },
    {
      "epoch": 3.0304908752441407e-05,
      "step": 31777,
      "training_step_time": 6.921956777572632
    },
    {
      "epoch": 3.030586242675781e-05,
      "model_forward_time": 1.7305684089660645,
      "step": 31778
    },
    {
      "epoch": 3.030586242675781e-05,
      "step": 31778,
      "training_step_time": 6.848487377166748
    },
    {
      "epoch": 3.030681610107422e-05,
      "model_forward_time": 1.741074562072754,
      "step": 31779
    },
    {
      "epoch": 3.030681610107422e-05,
      "step": 31779,
      "training_step_time": 6.899847745895386
    },
    {
      "epoch": 3.0307769775390626e-05,
      "grad_norm": 0.11118815094232559,
      "learning_rate": 8.16425927945148e-06,
      "loss": 0.0661,
      "step": 31780
    },
    {
      "epoch": 3.0307769775390626e-05,
      "model_forward_time": 1.7521610260009766,
      "step": 31780
    },
    {
      "dynamics_loss_avg": 0.0547485513612628,
      "epoch": 3.0307769775390626e-05,
      "step": 31780
    },
    {
      "action_loss_avg": 0.0015141361276619137,
      "epoch": 3.0307769775390626e-05,
      "step": 31780
    },
    {
      "epoch": 3.0307769775390626e-05,
      "step": 31780,
      "training_step_time": 6.871756553649902
    },
    {
      "epoch": 3.030872344970703e-05,
      "model_forward_time": 1.7236714363098145,
      "step": 31781
    },
    {
      "epoch": 3.030872344970703e-05,
      "step": 31781,
      "training_step_time": 6.898952007293701
    },
    {
      "epoch": 3.0309677124023438e-05,
      "model_forward_time": 1.737513542175293,
      "step": 31782
    },
    {
      "epoch": 3.0309677124023438e-05,
      "step": 31782,
      "training_step_time": 6.9761199951171875
    },
    {
      "epoch": 3.0310630798339845e-05,
      "model_forward_time": 2.840632677078247,
      "step": 31783
    },
    {
      "epoch": 3.0310630798339845e-05,
      "step": 31783,
      "training_step_time": 8.912123441696167
    },
    {
      "epoch": 3.031158447265625e-05,
      "model_forward_time": 1.7332661151885986,
      "step": 31784
    },
    {
      "epoch": 3.031158447265625e-05,
      "step": 31784,
      "training_step_time": 9.882924318313599
    },
    {
      "epoch": 3.0312538146972656e-05,
      "model_forward_time": 1.7395398616790771,
      "step": 31785
    },
    {
      "epoch": 3.0312538146972656e-05,
      "step": 31785,
      "training_step_time": 9.626879453659058
    },
    {
      "epoch": 3.0313491821289064e-05,
      "model_forward_time": 1.7428247928619385,
      "step": 31786
    },
    {
      "epoch": 3.0313491821289064e-05,
      "step": 31786,
      "training_step_time": 7.8752593994140625
    },
    {
      "epoch": 3.0314445495605468e-05,
      "model_forward_time": 1.7593388557434082,
      "step": 31787
    },
    {
      "epoch": 3.0314445495605468e-05,
      "step": 31787,
      "training_step_time": 6.8937108516693115
    },
    {
      "epoch": 3.0315399169921875e-05,
      "model_forward_time": 1.739591121673584,
      "step": 31788
    },
    {
      "epoch": 3.0315399169921875e-05,
      "step": 31788,
      "training_step_time": 6.946598291397095
    },
    {
      "epoch": 3.0316352844238283e-05,
      "model_forward_time": 1.7427442073822021,
      "step": 31789
    },
    {
      "epoch": 3.0316352844238283e-05,
      "step": 31789,
      "training_step_time": 6.905847072601318
    },
    {
      "epoch": 3.0317306518554687e-05,
      "grad_norm": 0.1545802652835846,
      "learning_rate": 8.162978869579551e-06,
      "loss": 0.0661,
      "step": 31790
    },
    {
      "epoch": 3.0317306518554687e-05,
      "model_forward_time": 1.7430684566497803,
      "step": 31790
    },
    {
      "dynamics_loss_avg": 0.06838607341051102,
      "epoch": 3.0317306518554687e-05,
      "step": 31790
    },
    {
      "action_loss_avg": 0.0018749105161987244,
      "epoch": 3.0317306518554687e-05,
      "step": 31790
    },
    {
      "epoch": 3.0317306518554687e-05,
      "step": 31790,
      "training_step_time": 6.915445566177368
    },
    {
      "epoch": 3.0318260192871094e-05,
      "model_forward_time": 1.7336838245391846,
      "step": 31791
    },
    {
      "epoch": 3.0318260192871094e-05,
      "step": 31791,
      "training_step_time": 6.93543004989624
    },
    {
      "epoch": 3.03192138671875e-05,
      "model_forward_time": 1.742366075515747,
      "step": 31792
    },
    {
      "epoch": 3.03192138671875e-05,
      "step": 31792,
      "training_step_time": 7.965298175811768
    },
    {
      "epoch": 3.0320167541503906e-05,
      "model_forward_time": 2.841146230697632,
      "step": 31793
    },
    {
      "epoch": 3.0320167541503906e-05,
      "step": 31793,
      "training_step_time": 11.243908882141113
    },
    {
      "epoch": 3.0321121215820313e-05,
      "model_forward_time": 1.748295545578003,
      "step": 31794
    },
    {
      "epoch": 3.0321121215820313e-05,
      "step": 31794,
      "training_step_time": 9.528897762298584
    },
    {
      "epoch": 3.032207489013672e-05,
      "model_forward_time": 1.7198169231414795,
      "step": 31795
    },
    {
      "epoch": 3.032207489013672e-05,
      "step": 31795,
      "training_step_time": 8.025206327438354
    },
    {
      "epoch": 3.0323028564453124e-05,
      "model_forward_time": 1.7305662631988525,
      "step": 31796
    },
    {
      "epoch": 3.0323028564453124e-05,
      "step": 31796,
      "training_step_time": 6.903731346130371
    },
    {
      "epoch": 3.0323982238769532e-05,
      "model_forward_time": 1.7247581481933594,
      "step": 31797
    },
    {
      "epoch": 3.0323982238769532e-05,
      "step": 31797,
      "training_step_time": 6.913147449493408
    },
    {
      "epoch": 3.0324935913085936e-05,
      "model_forward_time": 1.7341701984405518,
      "step": 31798
    },
    {
      "epoch": 3.0324935913085936e-05,
      "step": 31798,
      "training_step_time": 6.927030086517334
    },
    {
      "epoch": 3.0325889587402343e-05,
      "model_forward_time": 1.3587777614593506,
      "step": 31799
    },
    {
      "epoch": 3.0325889587402343e-05,
      "step": 31799,
      "training_step_time": 6.889392375946045
    },
    {
      "epoch": 3.032684326171875e-05,
      "grad_norm": 0.1490498036146164,
      "learning_rate": 8.161698113809007e-06,
      "loss": 0.0657,
      "step": 31800
    },
    {
      "epoch": 3.032684326171875e-05,
      "model_forward_time": 1.7217674255371094,
      "step": 31800
    },
    {
      "dynamics_loss_avg": 0.05113299936056137,
      "epoch": 3.032684326171875e-05,
      "step": 31800
    },
    {
      "action_loss_avg": 0.002271846344228834,
      "epoch": 3.032684326171875e-05,
      "step": 31800
    },
    {
      "epoch": 3.032684326171875e-05,
      "step": 31800,
      "training_step_time": 6.923451662063599
    },
    {
      "epoch": 3.0327796936035155e-05,
      "model_forward_time": 1.7094957828521729,
      "step": 31801
    },
    {
      "epoch": 3.0327796936035155e-05,
      "step": 31801,
      "training_step_time": 7.860040664672852
    },
    {
      "epoch": 3.0328750610351562e-05,
      "model_forward_time": 2.812816858291626,
      "step": 31802
    },
    {
      "epoch": 3.0328750610351562e-05,
      "step": 31802,
      "training_step_time": 8.49704909324646
    },
    {
      "epoch": 3.032970428466797e-05,
      "model_forward_time": 1.7531275749206543,
      "step": 31803
    },
    {
      "epoch": 3.032970428466797e-05,
      "step": 31803,
      "training_step_time": 8.81028437614441
    },
    {
      "epoch": 3.0330657958984374e-05,
      "model_forward_time": 1.726029872894287,
      "step": 31804
    },
    {
      "epoch": 3.0330657958984374e-05,
      "step": 31804,
      "training_step_time": 8.001756191253662
    },
    {
      "epoch": 3.033161163330078e-05,
      "model_forward_time": 1.7235376834869385,
      "step": 31805
    },
    {
      "epoch": 3.033161163330078e-05,
      "step": 31805,
      "training_step_time": 7.353703022003174
    },
    {
      "epoch": 3.033256530761719e-05,
      "model_forward_time": 1.7189362049102783,
      "step": 31806
    },
    {
      "epoch": 3.033256530761719e-05,
      "step": 31806,
      "training_step_time": 6.869697332382202
    },
    {
      "epoch": 3.0333518981933592e-05,
      "model_forward_time": 1.7198762893676758,
      "step": 31807
    },
    {
      "epoch": 3.0333518981933592e-05,
      "step": 31807,
      "training_step_time": 6.863996267318726
    },
    {
      "epoch": 3.033447265625e-05,
      "model_forward_time": 1.7257444858551025,
      "step": 31808
    },
    {
      "epoch": 3.033447265625e-05,
      "step": 31808,
      "training_step_time": 6.872547626495361
    },
    {
      "epoch": 3.0335426330566407e-05,
      "model_forward_time": 1.7182185649871826,
      "step": 31809
    },
    {
      "epoch": 3.0335426330566407e-05,
      "step": 31809,
      "training_step_time": 6.919950723648071
    },
    {
      "epoch": 3.033638000488281e-05,
      "grad_norm": 0.19370917975902557,
      "learning_rate": 8.160417012279911e-06,
      "loss": 0.0617,
      "step": 31810
    },
    {
      "epoch": 3.033638000488281e-05,
      "model_forward_time": 1.7198214530944824,
      "step": 31810
    },
    {
      "dynamics_loss_avg": 0.057933570258319377,
      "epoch": 3.033638000488281e-05,
      "step": 31810
    },
    {
      "action_loss_avg": 0.003059963881969452,
      "epoch": 3.033638000488281e-05,
      "step": 31810
    },
    {
      "epoch": 3.033638000488281e-05,
      "step": 31810,
      "training_step_time": 8.014087438583374
    },
    {
      "epoch": 3.033733367919922e-05,
      "model_forward_time": 2.8161134719848633,
      "step": 31811
    },
    {
      "epoch": 3.033733367919922e-05,
      "step": 31811,
      "training_step_time": 9.94640827178955
    },
    {
      "epoch": 3.0338287353515626e-05,
      "model_forward_time": 1.7451038360595703,
      "step": 31812
    },
    {
      "epoch": 3.0338287353515626e-05,
      "step": 31812,
      "training_step_time": 10.476930618286133
    },
    {
      "epoch": 3.033924102783203e-05,
      "model_forward_time": 1.7227025032043457,
      "step": 31813
    },
    {
      "epoch": 3.033924102783203e-05,
      "step": 31813,
      "training_step_time": 8.888391971588135
    },
    {
      "epoch": 3.0340194702148438e-05,
      "model_forward_time": 1.7118208408355713,
      "step": 31814
    },
    {
      "epoch": 3.0340194702148438e-05,
      "step": 31814,
      "training_step_time": 7.734071731567383
    },
    {
      "epoch": 3.0341148376464845e-05,
      "model_forward_time": 1.7190203666687012,
      "step": 31815
    },
    {
      "epoch": 3.0341148376464845e-05,
      "step": 31815,
      "training_step_time": 6.891882658004761
    },
    {
      "epoch": 3.034210205078125e-05,
      "model_forward_time": 1.0100517272949219,
      "step": 31816
    },
    {
      "epoch": 3.034210205078125e-05,
      "step": 31816,
      "training_step_time": 6.89475417137146
    },
    {
      "epoch": 3.0343055725097656e-05,
      "model_forward_time": 1.7421627044677734,
      "step": 31817
    },
    {
      "epoch": 3.0343055725097656e-05,
      "step": 31817,
      "training_step_time": 6.966010808944702
    },
    {
      "epoch": 3.0344009399414064e-05,
      "model_forward_time": 1.7320573329925537,
      "step": 31818
    },
    {
      "epoch": 3.0344009399414064e-05,
      "step": 31818,
      "training_step_time": 6.848519325256348
    },
    {
      "epoch": 3.0344963073730468e-05,
      "model_forward_time": 1.7417724132537842,
      "step": 31819
    },
    {
      "epoch": 3.0344963073730468e-05,
      "step": 31819,
      "training_step_time": 7.998260021209717
    },
    {
      "epoch": 3.0345916748046875e-05,
      "grad_norm": 0.12998852133750916,
      "learning_rate": 8.159135565132363e-06,
      "loss": 0.0633,
      "step": 31820
    },
    {
      "epoch": 3.0345916748046875e-05,
      "model_forward_time": 1.747049331665039,
      "step": 31820
    },
    {
      "dynamics_loss_avg": 0.05123728979378939,
      "epoch": 3.0345916748046875e-05,
      "step": 31820
    },
    {
      "action_loss_avg": 0.001603709752089344,
      "epoch": 3.0345916748046875e-05,
      "step": 31820
    },
    {
      "epoch": 3.0345916748046875e-05,
      "step": 31820,
      "training_step_time": 8.422594547271729
    },
    {
      "epoch": 3.0346870422363283e-05,
      "model_forward_time": 2.9147841930389404,
      "step": 31821
    },
    {
      "epoch": 3.0346870422363283e-05,
      "step": 31821,
      "training_step_time": 9.23005223274231
    },
    {
      "epoch": 3.0347824096679687e-05,
      "model_forward_time": 1.734309434890747,
      "step": 31822
    },
    {
      "epoch": 3.0347824096679687e-05,
      "step": 31822,
      "training_step_time": 7.8594536781311035
    },
    {
      "epoch": 3.0348777770996094e-05,
      "model_forward_time": 1.7363231182098389,
      "step": 31823
    },
    {
      "epoch": 3.0348777770996094e-05,
      "step": 31823,
      "training_step_time": 7.809841871261597
    },
    {
      "epoch": 3.03497314453125e-05,
      "model_forward_time": 1.738901138305664,
      "step": 31824
    },
    {
      "epoch": 3.03497314453125e-05,
      "step": 31824,
      "training_step_time": 6.968624591827393
    },
    {
      "epoch": 3.0350685119628906e-05,
      "model_forward_time": 1.7290427684783936,
      "step": 31825
    },
    {
      "epoch": 3.0350685119628906e-05,
      "step": 31825,
      "training_step_time": 6.868085861206055
    },
    {
      "epoch": 3.0351638793945313e-05,
      "model_forward_time": 1.7438304424285889,
      "step": 31826
    },
    {
      "epoch": 3.0351638793945313e-05,
      "step": 31826,
      "training_step_time": 6.898612022399902
    },
    {
      "epoch": 3.035259246826172e-05,
      "model_forward_time": 1.7432384490966797,
      "step": 31827
    },
    {
      "epoch": 3.035259246826172e-05,
      "step": 31827,
      "training_step_time": 6.931344270706177
    },
    {
      "epoch": 3.0353546142578124e-05,
      "model_forward_time": 1.740908145904541,
      "step": 31828
    },
    {
      "epoch": 3.0353546142578124e-05,
      "step": 31828,
      "training_step_time": 8.069788694381714
    },
    {
      "epoch": 3.0354499816894532e-05,
      "model_forward_time": 1.7413370609283447,
      "step": 31829
    },
    {
      "epoch": 3.0354499816894532e-05,
      "step": 31829,
      "training_step_time": 8.748765707015991
    },
    {
      "epoch": 3.0355453491210936e-05,
      "grad_norm": 0.1068161353468895,
      "learning_rate": 8.1578537725065e-06,
      "loss": 0.0635,
      "step": 31830
    },
    {
      "epoch": 3.0355453491210936e-05,
      "model_forward_time": 2.910402774810791,
      "step": 31830
    },
    {
      "dynamics_loss_avg": 0.06781380828469992,
      "epoch": 3.0355453491210936e-05,
      "step": 31830
    },
    {
      "action_loss_avg": 0.0020323095784988256,
      "epoch": 3.0355453491210936e-05,
      "step": 31830
    },
    {
      "epoch": 3.0355453491210936e-05,
      "step": 31830,
      "training_step_time": 8.753044128417969
    },
    {
      "epoch": 3.0356407165527343e-05,
      "model_forward_time": 1.7513370513916016,
      "step": 31831
    },
    {
      "epoch": 3.0356407165527343e-05,
      "step": 31831,
      "training_step_time": 7.952234506607056
    },
    {
      "epoch": 3.035736083984375e-05,
      "model_forward_time": 1.7323126792907715,
      "step": 31832
    },
    {
      "epoch": 3.035736083984375e-05,
      "step": 31832,
      "training_step_time": 8.134255409240723
    },
    {
      "epoch": 3.0358314514160155e-05,
      "model_forward_time": 1.727102518081665,
      "step": 31833
    },
    {
      "epoch": 3.0358314514160155e-05,
      "step": 31833,
      "training_step_time": 7.445804595947266
    },
    {
      "epoch": 3.0359268188476562e-05,
      "model_forward_time": 1.7336499691009521,
      "step": 31834
    },
    {
      "epoch": 3.0359268188476562e-05,
      "step": 31834,
      "training_step_time": 6.973052263259888
    },
    {
      "epoch": 3.036022186279297e-05,
      "model_forward_time": 1.3703734874725342,
      "step": 31835
    },
    {
      "epoch": 3.036022186279297e-05,
      "step": 31835,
      "training_step_time": 7.007179021835327
    },
    {
      "epoch": 3.0361175537109374e-05,
      "model_forward_time": 1.7326247692108154,
      "step": 31836
    },
    {
      "epoch": 3.0361175537109374e-05,
      "step": 31836,
      "training_step_time": 7.020861625671387
    },
    {
      "epoch": 3.036212921142578e-05,
      "model_forward_time": 1.7375781536102295,
      "step": 31837
    },
    {
      "epoch": 3.036212921142578e-05,
      "step": 31837,
      "training_step_time": 8.110205888748169
    },
    {
      "epoch": 3.036308288574219e-05,
      "model_forward_time": 1.7336547374725342,
      "step": 31838
    },
    {
      "epoch": 3.036308288574219e-05,
      "step": 31838,
      "training_step_time": 7.898544788360596
    },
    {
      "epoch": 3.0364036560058592e-05,
      "model_forward_time": 2.8647408485412598,
      "step": 31839
    },
    {
      "epoch": 3.0364036560058592e-05,
      "step": 31839,
      "training_step_time": 9.591943264007568
    },
    {
      "epoch": 3.0364990234375e-05,
      "grad_norm": 0.19279071688652039,
      "learning_rate": 8.156571634542494e-06,
      "loss": 0.0631,
      "step": 31840
    },
    {
      "epoch": 3.0364990234375e-05,
      "model_forward_time": 1.3743443489074707,
      "step": 31840
    },
    {
      "dynamics_loss_avg": 0.06679476741701365,
      "epoch": 3.0364990234375e-05,
      "step": 31840
    },
    {
      "action_loss_avg": 0.0018142626708140598,
      "epoch": 3.0364990234375e-05,
      "step": 31840
    },
    {
      "epoch": 3.0364990234375e-05,
      "step": 31840,
      "training_step_time": 8.381755113601685
    },
    {
      "epoch": 3.0365943908691407e-05,
      "model_forward_time": 1.7276098728179932,
      "step": 31841
    },
    {
      "epoch": 3.0365943908691407e-05,
      "step": 31841,
      "training_step_time": 8.08634066581726
    },
    {
      "epoch": 3.036689758300781e-05,
      "model_forward_time": 1.7289986610412598,
      "step": 31842
    },
    {
      "epoch": 3.036689758300781e-05,
      "step": 31842,
      "training_step_time": 7.83055853843689
    },
    {
      "epoch": 3.036785125732422e-05,
      "model_forward_time": 1.7461585998535156,
      "step": 31843
    },
    {
      "epoch": 3.036785125732422e-05,
      "step": 31843,
      "training_step_time": 6.856039524078369
    },
    {
      "epoch": 3.0368804931640626e-05,
      "model_forward_time": 1.3659815788269043,
      "step": 31844
    },
    {
      "epoch": 3.0368804931640626e-05,
      "step": 31844,
      "training_step_time": 6.892535448074341
    },
    {
      "epoch": 3.036975860595703e-05,
      "model_forward_time": 1.7346041202545166,
      "step": 31845
    },
    {
      "epoch": 3.036975860595703e-05,
      "step": 31845,
      "training_step_time": 6.9044201374053955
    },
    {
      "epoch": 3.0370712280273438e-05,
      "model_forward_time": 1.735262155532837,
      "step": 31846
    },
    {
      "epoch": 3.0370712280273438e-05,
      "step": 31846,
      "training_step_time": 7.978633642196655
    },
    {
      "epoch": 3.0371665954589845e-05,
      "model_forward_time": 1.7427020072937012,
      "step": 31847
    },
    {
      "epoch": 3.0371665954589845e-05,
      "step": 31847,
      "training_step_time": 7.710326910018921
    },
    {
      "epoch": 3.037261962890625e-05,
      "model_forward_time": 1.7664320468902588,
      "step": 31848
    },
    {
      "epoch": 3.037261962890625e-05,
      "step": 31848,
      "training_step_time": 9.889682531356812
    },
    {
      "epoch": 3.0373573303222656e-05,
      "model_forward_time": 1.7394824028015137,
      "step": 31849
    },
    {
      "epoch": 3.0373573303222656e-05,
      "step": 31849,
      "training_step_time": 8.477072715759277
    },
    {
      "epoch": 3.0374526977539064e-05,
      "grad_norm": 0.13580840826034546,
      "learning_rate": 8.15528915138056e-06,
      "loss": 0.0622,
      "step": 31850
    },
    {
      "epoch": 3.0374526977539064e-05,
      "model_forward_time": 1.7378699779510498,
      "step": 31850
    },
    {
      "dynamics_loss_avg": 0.06153139211237431,
      "epoch": 3.0374526977539064e-05,
      "step": 31850
    },
    {
      "action_loss_avg": 0.0016654233128065242,
      "epoch": 3.0374526977539064e-05,
      "step": 31850
    },
    {
      "epoch": 3.0374526977539064e-05,
      "step": 31850,
      "training_step_time": 8.036546468734741
    },
    {
      "epoch": 3.0375480651855468e-05,
      "model_forward_time": 1.719012975692749,
      "step": 31851
    },
    {
      "epoch": 3.0375480651855468e-05,
      "step": 31851,
      "training_step_time": 7.765676736831665
    },
    {
      "epoch": 3.0376434326171875e-05,
      "model_forward_time": 1.7333340644836426,
      "step": 31852
    },
    {
      "epoch": 3.0376434326171875e-05,
      "step": 31852,
      "training_step_time": 6.873376369476318
    },
    {
      "epoch": 3.0377388000488283e-05,
      "model_forward_time": 1.7420880794525146,
      "step": 31853
    },
    {
      "epoch": 3.0377388000488283e-05,
      "step": 31853,
      "training_step_time": 6.877503871917725
    },
    {
      "epoch": 3.0378341674804687e-05,
      "model_forward_time": 1.7392661571502686,
      "step": 31854
    },
    {
      "epoch": 3.0378341674804687e-05,
      "step": 31854,
      "training_step_time": 6.879232168197632
    },
    {
      "epoch": 3.0379295349121094e-05,
      "model_forward_time": 1.7397408485412598,
      "step": 31855
    },
    {
      "epoch": 3.0379295349121094e-05,
      "step": 31855,
      "training_step_time": 7.946071147918701
    },
    {
      "epoch": 3.03802490234375e-05,
      "model_forward_time": 1.7374694347381592,
      "step": 31856
    },
    {
      "epoch": 3.03802490234375e-05,
      "step": 31856,
      "training_step_time": 7.777736186981201
    },
    {
      "epoch": 3.0381202697753906e-05,
      "model_forward_time": 1.7555196285247803,
      "step": 31857
    },
    {
      "epoch": 3.0381202697753906e-05,
      "step": 31857,
      "training_step_time": 10.112085819244385
    },
    {
      "epoch": 3.0382156372070313e-05,
      "model_forward_time": 2.92120361328125,
      "step": 31858
    },
    {
      "epoch": 3.0382156372070313e-05,
      "step": 31858,
      "training_step_time": 9.951075553894043
    },
    {
      "epoch": 3.038311004638672e-05,
      "model_forward_time": 1.7380034923553467,
      "step": 31859
    },
    {
      "epoch": 3.038311004638672e-05,
      "step": 31859,
      "training_step_time": 7.803506135940552
    },
    {
      "epoch": 3.0384063720703124e-05,
      "grad_norm": 0.1712210327386856,
      "learning_rate": 8.15400632316095e-06,
      "loss": 0.0585,
      "step": 31860
    },
    {
      "epoch": 3.0384063720703124e-05,
      "model_forward_time": 1.734553337097168,
      "step": 31860
    },
    {
      "dynamics_loss_avg": 0.05672370232641697,
      "epoch": 3.0384063720703124e-05,
      "step": 31860
    },
    {
      "action_loss_avg": 0.0021165388519875706,
      "epoch": 3.0384063720703124e-05,
      "step": 31860
    },
    {
      "epoch": 3.0384063720703124e-05,
      "step": 31860,
      "training_step_time": 8.076207876205444
    },
    {
      "epoch": 3.0385017395019532e-05,
      "model_forward_time": 1.7388176918029785,
      "step": 31861
    },
    {
      "epoch": 3.0385017395019532e-05,
      "step": 31861,
      "training_step_time": 7.397615671157837
    },
    {
      "epoch": 3.0385971069335936e-05,
      "model_forward_time": 1.7373251914978027,
      "step": 31862
    },
    {
      "epoch": 3.0385971069335936e-05,
      "step": 31862,
      "training_step_time": 6.952930212020874
    },
    {
      "epoch": 3.0386924743652343e-05,
      "model_forward_time": 1.7438855171203613,
      "step": 31863
    },
    {
      "epoch": 3.0386924743652343e-05,
      "step": 31863,
      "training_step_time": 6.9184534549713135
    },
    {
      "epoch": 3.038787841796875e-05,
      "model_forward_time": 1.7421557903289795,
      "step": 31864
    },
    {
      "epoch": 3.038787841796875e-05,
      "step": 31864,
      "training_step_time": 7.755958557128906
    },
    {
      "epoch": 3.0388832092285155e-05,
      "model_forward_time": 1.731379747390747,
      "step": 31865
    },
    {
      "epoch": 3.0388832092285155e-05,
      "step": 31865,
      "training_step_time": 8.247286319732666
    },
    {
      "epoch": 3.0389785766601562e-05,
      "model_forward_time": 1.7458345890045166,
      "step": 31866
    },
    {
      "epoch": 3.0389785766601562e-05,
      "step": 31866,
      "training_step_time": 8.503824472427368
    },
    {
      "epoch": 3.039073944091797e-05,
      "model_forward_time": 2.9652304649353027,
      "step": 31867
    },
    {
      "epoch": 3.039073944091797e-05,
      "step": 31867,
      "training_step_time": 9.368965864181519
    },
    {
      "epoch": 3.0391693115234374e-05,
      "model_forward_time": 1.7370061874389648,
      "step": 31868
    },
    {
      "epoch": 3.0391693115234374e-05,
      "step": 31868,
      "training_step_time": 7.863201379776001
    },
    {
      "epoch": 3.039264678955078e-05,
      "model_forward_time": 1.714761734008789,
      "step": 31869
    },
    {
      "epoch": 3.039264678955078e-05,
      "step": 31869,
      "training_step_time": 8.047597169876099
    },
    {
      "epoch": 3.039360046386719e-05,
      "grad_norm": 0.12634091079235077,
      "learning_rate": 8.152723150023948e-06,
      "loss": 0.0551,
      "step": 31870
    },
    {
      "epoch": 3.039360046386719e-05,
      "model_forward_time": 1.7197389602661133,
      "step": 31870
    },
    {
      "dynamics_loss_avg": 0.05557611025869846,
      "epoch": 3.039360046386719e-05,
      "step": 31870
    },
    {
      "action_loss_avg": 0.0019161696021910758,
      "epoch": 3.039360046386719e-05,
      "step": 31870
    },
    {
      "epoch": 3.039360046386719e-05,
      "step": 31870,
      "training_step_time": 7.402033567428589
    },
    {
      "epoch": 3.0394554138183592e-05,
      "model_forward_time": 1.7211105823516846,
      "step": 31871
    },
    {
      "epoch": 3.0394554138183592e-05,
      "step": 31871,
      "training_step_time": 6.879550457000732
    },
    {
      "epoch": 3.03955078125e-05,
      "model_forward_time": 1.7254135608673096,
      "step": 31872
    },
    {
      "epoch": 3.03955078125e-05,
      "step": 31872,
      "training_step_time": 6.858931064605713
    },
    {
      "epoch": 3.0396461486816407e-05,
      "model_forward_time": 1.7165300846099854,
      "step": 31873
    },
    {
      "epoch": 3.0396461486816407e-05,
      "step": 31873,
      "training_step_time": 7.7885167598724365
    },
    {
      "epoch": 3.039741516113281e-05,
      "model_forward_time": 1.7193608283996582,
      "step": 31874
    },
    {
      "epoch": 3.039741516113281e-05,
      "step": 31874,
      "training_step_time": 7.766399621963501
    },
    {
      "epoch": 3.039836883544922e-05,
      "model_forward_time": 1.7215898036956787,
      "step": 31875
    },
    {
      "epoch": 3.039836883544922e-05,
      "step": 31875,
      "training_step_time": 8.748565673828125
    },
    {
      "epoch": 3.0399322509765626e-05,
      "model_forward_time": 2.8953006267547607,
      "step": 31876
    },
    {
      "epoch": 3.0399322509765626e-05,
      "step": 31876,
      "training_step_time": 7.852415561676025
    },
    {
      "epoch": 3.040027618408203e-05,
      "model_forward_time": 1.7235605716705322,
      "step": 31877
    },
    {
      "epoch": 3.040027618408203e-05,
      "step": 31877,
      "training_step_time": 8.887583017349243
    },
    {
      "epoch": 3.0401229858398438e-05,
      "model_forward_time": 1.7178232669830322,
      "step": 31878
    },
    {
      "epoch": 3.0401229858398438e-05,
      "step": 31878,
      "training_step_time": 8.063806056976318
    },
    {
      "epoch": 3.0402183532714845e-05,
      "model_forward_time": 1.7198221683502197,
      "step": 31879
    },
    {
      "epoch": 3.0402183532714845e-05,
      "step": 31879,
      "training_step_time": 7.761543035507202
    },
    {
      "epoch": 3.040313720703125e-05,
      "grad_norm": 0.12653912603855133,
      "learning_rate": 8.151439632109885e-06,
      "loss": 0.0538,
      "step": 31880
    },
    {
      "epoch": 3.040313720703125e-05,
      "model_forward_time": 1.717162847518921,
      "step": 31880
    },
    {
      "dynamics_loss_avg": 0.05884892344474792,
      "epoch": 3.040313720703125e-05,
      "step": 31880
    },
    {
      "action_loss_avg": 0.0024621055024908857,
      "epoch": 3.040313720703125e-05,
      "step": 31880
    },
    {
      "epoch": 3.040313720703125e-05,
      "step": 31880,
      "training_step_time": 6.817126750946045
    },
    {
      "epoch": 3.0404090881347656e-05,
      "model_forward_time": 1.7106225490570068,
      "step": 31881
    },
    {
      "epoch": 3.0404090881347656e-05,
      "step": 31881,
      "training_step_time": 6.919792175292969
    },
    {
      "epoch": 3.0405044555664064e-05,
      "model_forward_time": 1.7181129455566406,
      "step": 31882
    },
    {
      "epoch": 3.0405044555664064e-05,
      "step": 31882,
      "training_step_time": 7.788443565368652
    },
    {
      "epoch": 3.0405998229980468e-05,
      "model_forward_time": 1.7233457565307617,
      "step": 31883
    },
    {
      "epoch": 3.0405998229980468e-05,
      "step": 31883,
      "training_step_time": 8.548770427703857
    },
    {
      "epoch": 3.0406951904296875e-05,
      "model_forward_time": 1.7203173637390137,
      "step": 31884
    },
    {
      "epoch": 3.0406951904296875e-05,
      "step": 31884,
      "training_step_time": 7.727949857711792
    },
    {
      "epoch": 3.0407905578613283e-05,
      "model_forward_time": 2.8194241523742676,
      "step": 31885
    },
    {
      "epoch": 3.0407905578613283e-05,
      "step": 31885,
      "training_step_time": 9.674293756484985
    },
    {
      "epoch": 3.0408859252929687e-05,
      "model_forward_time": 1.7390191555023193,
      "step": 31886
    },
    {
      "epoch": 3.0408859252929687e-05,
      "step": 31886,
      "training_step_time": 8.043308019638062
    },
    {
      "epoch": 3.0409812927246094e-05,
      "model_forward_time": 1.7201893329620361,
      "step": 31887
    },
    {
      "epoch": 3.0409812927246094e-05,
      "step": 31887,
      "training_step_time": 7.735913991928101
    },
    {
      "epoch": 3.04107666015625e-05,
      "model_forward_time": 1.7192811965942383,
      "step": 31888
    },
    {
      "epoch": 3.04107666015625e-05,
      "step": 31888,
      "training_step_time": 8.479912042617798
    },
    {
      "epoch": 3.0411720275878906e-05,
      "model_forward_time": 1.7194995880126953,
      "step": 31889
    },
    {
      "epoch": 3.0411720275878906e-05,
      "step": 31889,
      "training_step_time": 7.763180255889893
    },
    {
      "epoch": 3.0412673950195313e-05,
      "grad_norm": 0.1542298048734665,
      "learning_rate": 8.150155769559123e-06,
      "loss": 0.0538,
      "step": 31890
    },
    {
      "epoch": 3.0412673950195313e-05,
      "model_forward_time": 1.7201359272003174,
      "step": 31890
    },
    {
      "dynamics_loss_avg": 0.05279105454683304,
      "epoch": 3.0412673950195313e-05,
      "step": 31890
    },
    {
      "action_loss_avg": 0.0020420125685632227,
      "epoch": 3.0412673950195313e-05,
      "step": 31890
    },
    {
      "epoch": 3.0412673950195313e-05,
      "step": 31890,
      "training_step_time": 6.900378704071045
    },
    {
      "epoch": 3.0413627624511717e-05,
      "model_forward_time": 1.7265448570251465,
      "step": 31891
    },
    {
      "epoch": 3.0413627624511717e-05,
      "step": 31891,
      "training_step_time": 7.694126844406128
    },
    {
      "epoch": 3.0414581298828124e-05,
      "model_forward_time": 1.729008436203003,
      "step": 31892
    },
    {
      "epoch": 3.0414581298828124e-05,
      "step": 31892,
      "training_step_time": 8.70829725265503
    },
    {
      "epoch": 3.0415534973144532e-05,
      "model_forward_time": 1.7278759479522705,
      "step": 31893
    },
    {
      "epoch": 3.0415534973144532e-05,
      "step": 31893,
      "training_step_time": 7.763432502746582
    },
    {
      "epoch": 3.0416488647460936e-05,
      "model_forward_time": 2.8268861770629883,
      "step": 31894
    },
    {
      "epoch": 3.0416488647460936e-05,
      "step": 31894,
      "training_step_time": 7.800385236740112
    },
    {
      "epoch": 3.0417442321777343e-05,
      "model_forward_time": 1.747286319732666,
      "step": 31895
    },
    {
      "epoch": 3.0417442321777343e-05,
      "step": 31895,
      "training_step_time": 9.532262563705444
    },
    {
      "epoch": 3.041839599609375e-05,
      "model_forward_time": 1.7204840183258057,
      "step": 31896
    },
    {
      "epoch": 3.041839599609375e-05,
      "step": 31896,
      "training_step_time": 7.76790452003479
    },
    {
      "epoch": 3.0419349670410155e-05,
      "model_forward_time": 1.7305107116699219,
      "step": 31897
    },
    {
      "epoch": 3.0419349670410155e-05,
      "step": 31897,
      "training_step_time": 7.754123210906982
    },
    {
      "epoch": 3.0420303344726562e-05,
      "model_forward_time": 1.7227683067321777,
      "step": 31898
    },
    {
      "epoch": 3.0420303344726562e-05,
      "step": 31898,
      "training_step_time": 7.749171257019043
    },
    {
      "epoch": 3.042125701904297e-05,
      "model_forward_time": 1.7293410301208496,
      "step": 31899
    },
    {
      "epoch": 3.042125701904297e-05,
      "step": 31899,
      "training_step_time": 6.900575876235962
    },
    {
      "epoch": 3.0422210693359374e-05,
      "grad_norm": 0.11083459854125977,
      "learning_rate": 8.148871562512058e-06,
      "loss": 0.0563,
      "step": 31900
    },
    {
      "epoch": 3.0422210693359374e-05,
      "model_forward_time": 1.7179365158081055,
      "step": 31900
    },
    {
      "dynamics_loss_avg": 0.05327533911913633,
      "epoch": 3.0422210693359374e-05,
      "step": 31900
    },
    {
      "action_loss_avg": 0.001803650939837098,
      "epoch": 3.0422210693359374e-05,
      "step": 31900
    },
    {
      "epoch": 3.0422210693359374e-05,
      "step": 31900,
      "training_step_time": 6.779972553253174
    },
    {
      "epoch": 3.042316436767578e-05,
      "model_forward_time": 1.7271034717559814,
      "step": 31901
    },
    {
      "epoch": 3.042316436767578e-05,
      "step": 31901,
      "training_step_time": 7.871687173843384
    },
    {
      "epoch": 3.042411804199219e-05,
      "model_forward_time": 1.7227833271026611,
      "step": 31902
    },
    {
      "epoch": 3.042411804199219e-05,
      "step": 31902,
      "training_step_time": 8.65011739730835
    },
    {
      "epoch": 3.0425071716308593e-05,
      "model_forward_time": 1.7405858039855957,
      "step": 31903
    },
    {
      "epoch": 3.0425071716308593e-05,
      "step": 31903,
      "training_step_time": 10.527643918991089
    },
    {
      "epoch": 3.0426025390625e-05,
      "model_forward_time": 1.7555372714996338,
      "step": 31904
    },
    {
      "epoch": 3.0426025390625e-05,
      "step": 31904,
      "training_step_time": 9.707582235336304
    },
    {
      "epoch": 3.0426979064941407e-05,
      "model_forward_time": 1.731724500656128,
      "step": 31905
    },
    {
      "epoch": 3.0426979064941407e-05,
      "step": 31905,
      "training_step_time": 7.820067882537842
    },
    {
      "epoch": 3.042793273925781e-05,
      "model_forward_time": 1.7385590076446533,
      "step": 31906
    },
    {
      "epoch": 3.042793273925781e-05,
      "step": 31906,
      "training_step_time": 7.783994913101196
    },
    {
      "epoch": 3.042888641357422e-05,
      "model_forward_time": 1.7358810901641846,
      "step": 31907
    },
    {
      "epoch": 3.042888641357422e-05,
      "step": 31907,
      "training_step_time": 8.624877214431763
    },
    {
      "epoch": 3.0429840087890626e-05,
      "model_forward_time": 1.7391357421875,
      "step": 31908
    },
    {
      "epoch": 3.0429840087890626e-05,
      "step": 31908,
      "training_step_time": 6.974694490432739
    },
    {
      "epoch": 3.043079376220703e-05,
      "model_forward_time": 1.7376649379730225,
      "step": 31909
    },
    {
      "epoch": 3.043079376220703e-05,
      "step": 31909,
      "training_step_time": 6.9157555103302
    },
    {
      "epoch": 3.0431747436523438e-05,
      "grad_norm": 0.13403670489788055,
      "learning_rate": 8.147587011109136e-06,
      "loss": 0.0589,
      "step": 31910
    },
    {
      "epoch": 3.0431747436523438e-05,
      "model_forward_time": 1.7403161525726318,
      "step": 31910
    },
    {
      "dynamics_loss_avg": 0.05303122252225876,
      "epoch": 3.0431747436523438e-05,
      "step": 31910
    },
    {
      "action_loss_avg": 0.002389917819527909,
      "epoch": 3.0431747436523438e-05,
      "step": 31910
    },
    {
      "epoch": 3.0431747436523438e-05,
      "step": 31910,
      "training_step_time": 7.811196565628052
    },
    {
      "epoch": 3.0432701110839845e-05,
      "model_forward_time": 1.7376222610473633,
      "step": 31911
    },
    {
      "epoch": 3.0432701110839845e-05,
      "step": 31911,
      "training_step_time": 7.795381307601929
    },
    {
      "epoch": 3.043365478515625e-05,
      "model_forward_time": 1.7804222106933594,
      "step": 31912
    },
    {
      "epoch": 3.043365478515625e-05,
      "step": 31912,
      "training_step_time": 8.786476612091064
    },
    {
      "epoch": 3.0434608459472656e-05,
      "model_forward_time": 2.9735753536224365,
      "step": 31913
    },
    {
      "epoch": 3.0434608459472656e-05,
      "step": 31913,
      "training_step_time": 8.958919763565063
    },
    {
      "epoch": 3.0435562133789064e-05,
      "model_forward_time": 1.7341465950012207,
      "step": 31914
    },
    {
      "epoch": 3.0435562133789064e-05,
      "step": 31914,
      "training_step_time": 8.447218894958496
    },
    {
      "epoch": 3.0436515808105468e-05,
      "model_forward_time": 1.743199348449707,
      "step": 31915
    },
    {
      "epoch": 3.0436515808105468e-05,
      "step": 31915,
      "training_step_time": 7.757166385650635
    },
    {
      "epoch": 3.0437469482421875e-05,
      "model_forward_time": 1.7315614223480225,
      "step": 31916
    },
    {
      "epoch": 3.0437469482421875e-05,
      "step": 31916,
      "training_step_time": 7.81387996673584
    },
    {
      "epoch": 3.0438423156738283e-05,
      "model_forward_time": 1.7388267517089844,
      "step": 31917
    },
    {
      "epoch": 3.0438423156738283e-05,
      "step": 31917,
      "training_step_time": 7.432821750640869
    },
    {
      "epoch": 3.0439376831054687e-05,
      "model_forward_time": 1.7417216300964355,
      "step": 31918
    },
    {
      "epoch": 3.0439376831054687e-05,
      "step": 31918,
      "training_step_time": 6.869354248046875
    },
    {
      "epoch": 3.0440330505371094e-05,
      "model_forward_time": 1.7337207794189453,
      "step": 31919
    },
    {
      "epoch": 3.0440330505371094e-05,
      "step": 31919,
      "training_step_time": 7.86085057258606
    },
    {
      "epoch": 3.04412841796875e-05,
      "grad_norm": 0.1775524914264679,
      "learning_rate": 8.14630211549083e-06,
      "loss": 0.0608,
      "step": 31920
    },
    {
      "epoch": 3.04412841796875e-05,
      "model_forward_time": 1.7310307025909424,
      "step": 31920
    },
    {
      "dynamics_loss_avg": 0.08019187599420548,
      "epoch": 3.04412841796875e-05,
      "step": 31920
    },
    {
      "action_loss_avg": 0.0018922056362498552,
      "epoch": 3.04412841796875e-05,
      "step": 31920
    },
    {
      "epoch": 3.04412841796875e-05,
      "step": 31920,
      "training_step_time": 8.522729396820068
    },
    {
      "epoch": 3.0442237854003906e-05,
      "model_forward_time": 1.738806962966919,
      "step": 31921
    },
    {
      "epoch": 3.0442237854003906e-05,
      "step": 31921,
      "training_step_time": 8.404252767562866
    },
    {
      "epoch": 3.0443191528320313e-05,
      "model_forward_time": 2.9660706520080566,
      "step": 31922
    },
    {
      "epoch": 3.0443191528320313e-05,
      "step": 31922,
      "training_step_time": 9.097983837127686
    },
    {
      "epoch": 3.0444145202636717e-05,
      "model_forward_time": 1.7344732284545898,
      "step": 31923
    },
    {
      "epoch": 3.0444145202636717e-05,
      "step": 31923,
      "training_step_time": 8.416423559188843
    },
    {
      "epoch": 3.0445098876953125e-05,
      "model_forward_time": 1.7253241539001465,
      "step": 31924
    },
    {
      "epoch": 3.0445098876953125e-05,
      "step": 31924,
      "training_step_time": 7.697349786758423
    },
    {
      "epoch": 3.0446052551269532e-05,
      "model_forward_time": 1.7339434623718262,
      "step": 31925
    },
    {
      "epoch": 3.0446052551269532e-05,
      "step": 31925,
      "training_step_time": 7.792320728302002
    },
    {
      "epoch": 3.0447006225585936e-05,
      "model_forward_time": 1.0394117832183838,
      "step": 31926
    },
    {
      "epoch": 3.0447006225585936e-05,
      "step": 31926,
      "training_step_time": 7.383561611175537
    },
    {
      "epoch": 3.0447959899902343e-05,
      "model_forward_time": 1.7307918071746826,
      "step": 31927
    },
    {
      "epoch": 3.0447959899902343e-05,
      "step": 31927,
      "training_step_time": 6.951036691665649
    },
    {
      "epoch": 3.044891357421875e-05,
      "model_forward_time": 1.7366375923156738,
      "step": 31928
    },
    {
      "epoch": 3.044891357421875e-05,
      "step": 31928,
      "training_step_time": 8.363056898117065
    },
    {
      "epoch": 3.0449867248535155e-05,
      "model_forward_time": 1.7370994091033936,
      "step": 31929
    },
    {
      "epoch": 3.0449867248535155e-05,
      "step": 31929,
      "training_step_time": 7.831521272659302
    },
    {
      "epoch": 3.0450820922851562e-05,
      "grad_norm": 0.12120641022920609,
      "learning_rate": 8.145016875797655e-06,
      "loss": 0.0601,
      "step": 31930
    },
    {
      "epoch": 3.0450820922851562e-05,
      "model_forward_time": 1.732198715209961,
      "step": 31930
    },
    {
      "dynamics_loss_avg": 0.05040868632495403,
      "epoch": 3.0450820922851562e-05,
      "step": 31930
    },
    {
      "action_loss_avg": 0.0014551643340382724,
      "epoch": 3.0450820922851562e-05,
      "step": 31930
    },
    {
      "epoch": 3.0450820922851562e-05,
      "step": 31930,
      "training_step_time": 7.767216444015503
    },
    {
      "epoch": 3.045177459716797e-05,
      "model_forward_time": 2.8682174682617188,
      "step": 31931
    },
    {
      "epoch": 3.045177459716797e-05,
      "step": 31931,
      "training_step_time": 7.976214170455933
    },
    {
      "epoch": 3.0452728271484374e-05,
      "model_forward_time": 1.7410719394683838,
      "step": 31932
    },
    {
      "epoch": 3.0452728271484374e-05,
      "step": 31932,
      "training_step_time": 10.176009893417358
    },
    {
      "epoch": 3.045368194580078e-05,
      "model_forward_time": 1.7289750576019287,
      "step": 31933
    },
    {
      "epoch": 3.045368194580078e-05,
      "step": 31933,
      "training_step_time": 9.15711259841919
    },
    {
      "epoch": 3.045463562011719e-05,
      "model_forward_time": 1.733130931854248,
      "step": 31934
    },
    {
      "epoch": 3.045463562011719e-05,
      "step": 31934,
      "training_step_time": 7.762266159057617
    },
    {
      "epoch": 3.0455589294433593e-05,
      "model_forward_time": 1.7365145683288574,
      "step": 31935
    },
    {
      "epoch": 3.0455589294433593e-05,
      "step": 31935,
      "training_step_time": 7.4087889194488525
    },
    {
      "epoch": 3.045654296875e-05,
      "model_forward_time": 1.742361307144165,
      "step": 31936
    },
    {
      "epoch": 3.045654296875e-05,
      "step": 31936,
      "training_step_time": 6.9323320388793945
    },
    {
      "epoch": 3.0457496643066407e-05,
      "model_forward_time": 1.733137607574463,
      "step": 31937
    },
    {
      "epoch": 3.0457496643066407e-05,
      "step": 31937,
      "training_step_time": 7.784271478652954
    },
    {
      "epoch": 3.045845031738281e-05,
      "model_forward_time": 1.733412504196167,
      "step": 31938
    },
    {
      "epoch": 3.045845031738281e-05,
      "step": 31938,
      "training_step_time": 8.78857421875
    },
    {
      "epoch": 3.045940399169922e-05,
      "model_forward_time": 1.7318274974822998,
      "step": 31939
    },
    {
      "epoch": 3.045940399169922e-05,
      "step": 31939,
      "training_step_time": 8.386783838272095
    },
    {
      "epoch": 3.0460357666015626e-05,
      "grad_norm": 0.15068775415420532,
      "learning_rate": 8.143731292170164e-06,
      "loss": 0.0625,
      "step": 31940
    },
    {
      "epoch": 3.0460357666015626e-05,
      "model_forward_time": 2.8518524169921875,
      "step": 31940
    },
    {
      "dynamics_loss_avg": 0.04471756648272276,
      "epoch": 3.0460357666015626e-05,
      "step": 31940
    },
    {
      "action_loss_avg": 0.0014693583303596824,
      "epoch": 3.0460357666015626e-05,
      "step": 31940
    },
    {
      "epoch": 3.0460357666015626e-05,
      "step": 31940,
      "training_step_time": 9.609755992889404
    },
    {
      "epoch": 3.046131134033203e-05,
      "model_forward_time": 1.75455641746521,
      "step": 31941
    },
    {
      "epoch": 3.046131134033203e-05,
      "step": 31941,
      "training_step_time": 9.81983232498169
    },
    {
      "epoch": 3.0462265014648438e-05,
      "model_forward_time": 1.739241600036621,
      "step": 31942
    },
    {
      "epoch": 3.0462265014648438e-05,
      "step": 31942,
      "training_step_time": 7.849231004714966
    },
    {
      "epoch": 3.0463218688964845e-05,
      "model_forward_time": 1.7340645790100098,
      "step": 31943
    },
    {
      "epoch": 3.0463218688964845e-05,
      "step": 31943,
      "training_step_time": 7.782302379608154
    },
    {
      "epoch": 3.046417236328125e-05,
      "model_forward_time": 1.740922212600708,
      "step": 31944
    },
    {
      "epoch": 3.046417236328125e-05,
      "step": 31944,
      "training_step_time": 7.7606048583984375
    },
    {
      "epoch": 3.0465126037597657e-05,
      "model_forward_time": 1.736863136291504,
      "step": 31945
    },
    {
      "epoch": 3.0465126037597657e-05,
      "step": 31945,
      "training_step_time": 6.874060153961182
    },
    {
      "epoch": 3.0466079711914064e-05,
      "model_forward_time": 1.7166202068328857,
      "step": 31946
    },
    {
      "epoch": 3.0466079711914064e-05,
      "step": 31946,
      "training_step_time": 8.554665088653564
    },
    {
      "epoch": 3.0467033386230468e-05,
      "model_forward_time": 1.7255313396453857,
      "step": 31947
    },
    {
      "epoch": 3.0467033386230468e-05,
      "step": 31947,
      "training_step_time": 7.903764963150024
    },
    {
      "epoch": 3.0467987060546875e-05,
      "model_forward_time": 1.721172571182251,
      "step": 31948
    },
    {
      "epoch": 3.0467987060546875e-05,
      "step": 31948,
      "training_step_time": 9.349636554718018
    },
    {
      "epoch": 3.0468940734863283e-05,
      "model_forward_time": 1.7528958320617676,
      "step": 31949
    },
    {
      "epoch": 3.0468940734863283e-05,
      "step": 31949,
      "training_step_time": 8.224355697631836
    },
    {
      "epoch": 3.0469894409179687e-05,
      "grad_norm": 0.13094261288642883,
      "learning_rate": 8.142445364748945e-06,
      "loss": 0.0599,
      "step": 31950
    },
    {
      "epoch": 3.0469894409179687e-05,
      "model_forward_time": 2.9142887592315674,
      "step": 31950
    },
    {
      "dynamics_loss_avg": 0.060529534332454206,
      "epoch": 3.0469894409179687e-05,
      "step": 31950
    },
    {
      "action_loss_avg": 0.002436246071010828,
      "epoch": 3.0469894409179687e-05,
      "step": 31950
    },
    {
      "epoch": 3.0469894409179687e-05,
      "step": 31950,
      "training_step_time": 8.053956985473633
    },
    {
      "epoch": 3.0470848083496094e-05,
      "model_forward_time": 1.7227885723114014,
      "step": 31951
    },
    {
      "epoch": 3.0470848083496094e-05,
      "step": 31951,
      "training_step_time": 7.760523796081543
    },
    {
      "epoch": 3.04718017578125e-05,
      "model_forward_time": 1.7238764762878418,
      "step": 31952
    },
    {
      "epoch": 3.04718017578125e-05,
      "step": 31952,
      "training_step_time": 7.7114667892456055
    },
    {
      "epoch": 3.0472755432128906e-05,
      "model_forward_time": 1.7262675762176514,
      "step": 31953
    },
    {
      "epoch": 3.0472755432128906e-05,
      "step": 31953,
      "training_step_time": 7.776968240737915
    },
    {
      "epoch": 3.0473709106445313e-05,
      "model_forward_time": 1.7193279266357422,
      "step": 31954
    },
    {
      "epoch": 3.0473709106445313e-05,
      "step": 31954,
      "training_step_time": 7.380488157272339
    },
    {
      "epoch": 3.0474662780761717e-05,
      "model_forward_time": 1.723510503768921,
      "step": 31955
    },
    {
      "epoch": 3.0474662780761717e-05,
      "step": 31955,
      "training_step_time": 7.808399438858032
    },
    {
      "epoch": 3.0475616455078125e-05,
      "model_forward_time": 1.713902473449707,
      "step": 31956
    },
    {
      "epoch": 3.0475616455078125e-05,
      "step": 31956,
      "training_step_time": 7.775731086730957
    },
    {
      "epoch": 3.0476570129394532e-05,
      "model_forward_time": 1.720118522644043,
      "step": 31957
    },
    {
      "epoch": 3.0476570129394532e-05,
      "step": 31957,
      "training_step_time": 8.598930358886719
    },
    {
      "epoch": 3.0477523803710936e-05,
      "model_forward_time": 1.739253044128418,
      "step": 31958
    },
    {
      "epoch": 3.0477523803710936e-05,
      "step": 31958,
      "training_step_time": 7.721008777618408
    },
    {
      "epoch": 3.0478477478027343e-05,
      "model_forward_time": 2.8789844512939453,
      "step": 31959
    },
    {
      "epoch": 3.0478477478027343e-05,
      "step": 31959,
      "training_step_time": 8.810431957244873
    },
    {
      "epoch": 3.047943115234375e-05,
      "grad_norm": 0.1515256017446518,
      "learning_rate": 8.141159093674624e-06,
      "loss": 0.0622,
      "step": 31960
    },
    {
      "epoch": 3.047943115234375e-05,
      "model_forward_time": 1.7172768115997314,
      "step": 31960
    },
    {
      "dynamics_loss_avg": 0.06955435797572136,
      "epoch": 3.047943115234375e-05,
      "step": 31960
    },
    {
      "action_loss_avg": 0.0013397402479313314,
      "epoch": 3.047943115234375e-05,
      "step": 31960
    },
    {
      "epoch": 3.047943115234375e-05,
      "step": 31960,
      "training_step_time": 7.773030042648315
    },
    {
      "epoch": 3.0480384826660155e-05,
      "model_forward_time": 1.7179713249206543,
      "step": 31961
    },
    {
      "epoch": 3.0480384826660155e-05,
      "step": 31961,
      "training_step_time": 7.698507785797119
    },
    {
      "epoch": 3.0481338500976562e-05,
      "model_forward_time": 1.7205452919006348,
      "step": 31962
    },
    {
      "epoch": 3.0481338500976562e-05,
      "step": 31962,
      "training_step_time": 7.767359733581543
    },
    {
      "epoch": 3.048229217529297e-05,
      "model_forward_time": 1.7240996360778809,
      "step": 31963
    },
    {
      "epoch": 3.048229217529297e-05,
      "step": 31963,
      "training_step_time": 7.462552309036255
    },
    {
      "epoch": 3.0483245849609374e-05,
      "model_forward_time": 1.7263574600219727,
      "step": 31964
    },
    {
      "epoch": 3.0483245849609374e-05,
      "step": 31964,
      "training_step_time": 7.758950233459473
    },
    {
      "epoch": 3.048419952392578e-05,
      "model_forward_time": 1.720900297164917,
      "step": 31965
    },
    {
      "epoch": 3.048419952392578e-05,
      "step": 31965,
      "training_step_time": 8.711067199707031
    },
    {
      "epoch": 3.048515319824219e-05,
      "model_forward_time": 1.7246885299682617,
      "step": 31966
    },
    {
      "epoch": 3.048515319824219e-05,
      "step": 31966,
      "training_step_time": 8.005197286605835
    },
    {
      "epoch": 3.0486106872558593e-05,
      "model_forward_time": 1.723604679107666,
      "step": 31967
    },
    {
      "epoch": 3.0486106872558593e-05,
      "step": 31967,
      "training_step_time": 7.772469520568848
    },
    {
      "epoch": 3.0487060546875e-05,
      "model_forward_time": 2.927109718322754,
      "step": 31968
    },
    {
      "epoch": 3.0487060546875e-05,
      "step": 31968,
      "training_step_time": 8.862192392349243
    },
    {
      "epoch": 3.0488014221191407e-05,
      "model_forward_time": 1.7207696437835693,
      "step": 31969
    },
    {
      "epoch": 3.0488014221191407e-05,
      "step": 31969,
      "training_step_time": 8.322059631347656
    },
    {
      "epoch": 3.048896789550781e-05,
      "grad_norm": 0.16531671583652496,
      "learning_rate": 8.139872479087869e-06,
      "loss": 0.0619,
      "step": 31970
    },
    {
      "epoch": 3.048896789550781e-05,
      "model_forward_time": 1.720414161682129,
      "step": 31970
    },
    {
      "dynamics_loss_avg": 0.0418160354718566,
      "epoch": 3.048896789550781e-05,
      "step": 31970
    },
    {
      "action_loss_avg": 0.00154229105100967,
      "epoch": 3.048896789550781e-05,
      "step": 31970
    },
    {
      "epoch": 3.048896789550781e-05,
      "step": 31970,
      "training_step_time": 7.798751592636108
    },
    {
      "epoch": 3.048992156982422e-05,
      "model_forward_time": 1.7254667282104492,
      "step": 31971
    },
    {
      "epoch": 3.048992156982422e-05,
      "step": 31971,
      "training_step_time": 7.744386196136475
    },
    {
      "epoch": 3.0490875244140626e-05,
      "model_forward_time": 1.7242469787597656,
      "step": 31972
    },
    {
      "epoch": 3.0490875244140626e-05,
      "step": 31972,
      "training_step_time": 7.411408424377441
    },
    {
      "epoch": 3.049182891845703e-05,
      "model_forward_time": 1.7285833358764648,
      "step": 31973
    },
    {
      "epoch": 3.049182891845703e-05,
      "step": 31973,
      "training_step_time": 7.847920656204224
    },
    {
      "epoch": 3.0492782592773438e-05,
      "model_forward_time": 1.7265233993530273,
      "step": 31974
    },
    {
      "epoch": 3.0492782592773438e-05,
      "step": 31974,
      "training_step_time": 7.715393781661987
    },
    {
      "epoch": 3.0493736267089845e-05,
      "model_forward_time": 1.725550889968872,
      "step": 31975
    },
    {
      "epoch": 3.0493736267089845e-05,
      "step": 31975,
      "training_step_time": 7.924891471862793
    },
    {
      "epoch": 3.049468994140625e-05,
      "model_forward_time": 1.716094970703125,
      "step": 31976
    },
    {
      "epoch": 3.049468994140625e-05,
      "step": 31976,
      "training_step_time": 8.73640513420105
    },
    {
      "epoch": 3.0495643615722657e-05,
      "model_forward_time": 2.8266873359680176,
      "step": 31977
    },
    {
      "epoch": 3.0495643615722657e-05,
      "step": 31977,
      "training_step_time": 8.648859024047852
    },
    {
      "epoch": 3.0496597290039064e-05,
      "model_forward_time": 1.7242472171783447,
      "step": 31978
    },
    {
      "epoch": 3.0496597290039064e-05,
      "step": 31978,
      "training_step_time": 9.090997219085693
    },
    {
      "epoch": 3.0497550964355468e-05,
      "model_forward_time": 1.7250909805297852,
      "step": 31979
    },
    {
      "epoch": 3.0497550964355468e-05,
      "step": 31979,
      "training_step_time": 7.851053476333618
    },
    {
      "epoch": 3.0498504638671875e-05,
      "grad_norm": 0.1578291356563568,
      "learning_rate": 8.13858552112938e-06,
      "loss": 0.0629,
      "step": 31980
    },
    {
      "epoch": 3.0498504638671875e-05,
      "model_forward_time": 1.716186761856079,
      "step": 31980
    },
    {
      "dynamics_loss_avg": 0.06569228507578373,
      "epoch": 3.0498504638671875e-05,
      "step": 31980
    },
    {
      "action_loss_avg": 0.0022792497707996517,
      "epoch": 3.0498504638671875e-05,
      "step": 31980
    },
    {
      "epoch": 3.0498504638671875e-05,
      "step": 31980,
      "training_step_time": 7.712543964385986
    },
    {
      "epoch": 3.0499458312988283e-05,
      "model_forward_time": 1.722428798675537,
      "step": 31981
    },
    {
      "epoch": 3.0499458312988283e-05,
      "step": 31981,
      "training_step_time": 7.7376484870910645
    },
    {
      "epoch": 3.0500411987304687e-05,
      "model_forward_time": 1.7218654155731201,
      "step": 31982
    },
    {
      "epoch": 3.0500411987304687e-05,
      "step": 31982,
      "training_step_time": 7.872362375259399
    },
    {
      "epoch": 3.0501365661621094e-05,
      "model_forward_time": 1.7217166423797607,
      "step": 31983
    },
    {
      "epoch": 3.0501365661621094e-05,
      "step": 31983,
      "training_step_time": 7.685948610305786
    },
    {
      "epoch": 3.05023193359375e-05,
      "model_forward_time": 1.7219207286834717,
      "step": 31984
    },
    {
      "epoch": 3.05023193359375e-05,
      "step": 31984,
      "training_step_time": 8.00477933883667
    },
    {
      "epoch": 3.0503273010253906e-05,
      "model_forward_time": 1.717452049255371,
      "step": 31985
    },
    {
      "epoch": 3.0503273010253906e-05,
      "step": 31985,
      "training_step_time": 9.323199272155762
    },
    {
      "epoch": 3.0504226684570313e-05,
      "model_forward_time": 2.8206725120544434,
      "step": 31986
    },
    {
      "epoch": 3.0504226684570313e-05,
      "step": 31986,
      "training_step_time": 8.3382568359375
    },
    {
      "epoch": 3.0505180358886717e-05,
      "model_forward_time": 1.739748477935791,
      "step": 31987
    },
    {
      "epoch": 3.0505180358886717e-05,
      "step": 31987,
      "training_step_time": 8.854519844055176
    },
    {
      "epoch": 3.0506134033203125e-05,
      "model_forward_time": 1.7210466861724854,
      "step": 31988
    },
    {
      "epoch": 3.0506134033203125e-05,
      "step": 31988,
      "training_step_time": 7.792603254318237
    },
    {
      "epoch": 3.0507087707519532e-05,
      "model_forward_time": 1.7250213623046875,
      "step": 31989
    },
    {
      "epoch": 3.0507087707519532e-05,
      "step": 31989,
      "training_step_time": 8.56568193435669
    },
    {
      "epoch": 3.0508041381835936e-05,
      "grad_norm": 0.13679493963718414,
      "learning_rate": 8.137298219939896e-06,
      "loss": 0.0623,
      "step": 31990
    },
    {
      "epoch": 3.0508041381835936e-05,
      "model_forward_time": 1.7291970252990723,
      "step": 31990
    },
    {
      "dynamics_loss_avg": 0.06286735385656357,
      "epoch": 3.0508041381835936e-05,
      "step": 31990
    },
    {
      "action_loss_avg": 0.0020741239364724605,
      "epoch": 3.0508041381835936e-05,
      "step": 31990
    },
    {
      "epoch": 3.0508041381835936e-05,
      "step": 31990,
      "training_step_time": 7.718205690383911
    },
    {
      "epoch": 3.0508995056152343e-05,
      "model_forward_time": 1.7239713668823242,
      "step": 31991
    },
    {
      "epoch": 3.0508995056152343e-05,
      "step": 31991,
      "training_step_time": 7.894834518432617
    },
    {
      "epoch": 3.050994873046875e-05,
      "model_forward_time": 1.7282707691192627,
      "step": 31992
    },
    {
      "epoch": 3.050994873046875e-05,
      "step": 31992,
      "training_step_time": 7.660702705383301
    },
    {
      "epoch": 3.0510902404785155e-05,
      "model_forward_time": 1.7179250717163086,
      "step": 31993
    },
    {
      "epoch": 3.0510902404785155e-05,
      "step": 31993,
      "training_step_time": 8.863524198532104
    },
    {
      "epoch": 3.0511856079101562e-05,
      "model_forward_time": 1.7236347198486328,
      "step": 31994
    },
    {
      "epoch": 3.0511856079101562e-05,
      "step": 31994,
      "training_step_time": 9.087526798248291
    },
    {
      "epoch": 3.051280975341797e-05,
      "model_forward_time": 1.7490334510803223,
      "step": 31995
    },
    {
      "epoch": 3.051280975341797e-05,
      "step": 31995,
      "training_step_time": 8.40527868270874
    },
    {
      "epoch": 3.0513763427734374e-05,
      "model_forward_time": 1.7314956188201904,
      "step": 31996
    },
    {
      "epoch": 3.0513763427734374e-05,
      "step": 31996,
      "training_step_time": 8.997119665145874
    },
    {
      "epoch": 3.051471710205078e-05,
      "model_forward_time": 1.7444589138031006,
      "step": 31997
    },
    {
      "epoch": 3.051471710205078e-05,
      "step": 31997,
      "training_step_time": 10.395199537277222
    },
    {
      "epoch": 3.051567077636719e-05,
      "model_forward_time": 1.7740349769592285,
      "step": 31998
    },
    {
      "epoch": 3.051567077636719e-05,
      "step": 31998,
      "training_step_time": 8.210700511932373
    },
    {
      "epoch": 3.0516624450683593e-05,
      "model_forward_time": 1.7205824851989746,
      "step": 31999
    },
    {
      "epoch": 3.0516624450683593e-05,
      "step": 31999,
      "training_step_time": 7.51929497718811
    },
    {
      "epoch": 3.0517578125e-05,
      "grad_norm": 0.1269954890012741,
      "learning_rate": 8.136010575660197e-06,
      "loss": 0.0618,
      "step": 32000
    },
    {
      "epoch": 3.0517578125e-05,
      "model_forward_time": 1.7729289531707764,
      "step": 32000
    },
    {
      "dynamics_loss_avg": 0.04529317282140255,
      "epoch": 3.0517578125e-05,
      "step": 32000
    },
    {
      "action_loss_avg": 0.002239506272599101,
      "epoch": 3.0517578125e-05,
      "step": 32000
    },
    {
      "epoch": 3.0517578125e-05,
      "step": 32000,
      "training_step_time": 7.005316972732544
    },
    {
      "epoch": 3.0518531799316404e-05,
      "model_forward_time": 1.7490789890289307,
      "step": 32001
    },
    {
      "epoch": 3.0518531799316404e-05,
      "step": 32001,
      "training_step_time": 7.670384407043457
    },
    {
      "epoch": 3.0519485473632815e-05,
      "model_forward_time": 1.725745439529419,
      "step": 32002
    },
    {
      "epoch": 3.0519485473632815e-05,
      "step": 32002,
      "training_step_time": 8.657798528671265
    },
    {
      "epoch": 3.052043914794922e-05,
      "model_forward_time": 1.7482426166534424,
      "step": 32003
    },
    {
      "epoch": 3.052043914794922e-05,
      "step": 32003,
      "training_step_time": 8.834749698638916
    },
    {
      "epoch": 3.052139282226562e-05,
      "model_forward_time": 2.5635173320770264,
      "step": 32004
    },
    {
      "epoch": 3.052139282226562e-05,
      "step": 32004,
      "training_step_time": 10.55439043045044
    },
    {
      "epoch": 3.0522346496582034e-05,
      "model_forward_time": 1.7241449356079102,
      "step": 32005
    },
    {
      "epoch": 3.0522346496582034e-05,
      "step": 32005,
      "training_step_time": 9.518324851989746
    },
    {
      "epoch": 3.052330017089844e-05,
      "model_forward_time": 1.8021020889282227,
      "step": 32006
    },
    {
      "epoch": 3.052330017089844e-05,
      "step": 32006,
      "training_step_time": 10.425757646560669
    },
    {
      "epoch": 3.052425384521484e-05,
      "model_forward_time": 1.732208490371704,
      "step": 32007
    },
    {
      "epoch": 3.052425384521484e-05,
      "step": 32007,
      "training_step_time": 7.82694149017334
    },
    {
      "epoch": 3.052520751953125e-05,
      "model_forward_time": 1.7248961925506592,
      "step": 32008
    },
    {
      "epoch": 3.052520751953125e-05,
      "step": 32008,
      "training_step_time": 7.7779974937438965
    },
    {
      "epoch": 3.0526161193847657e-05,
      "model_forward_time": 1.7320318222045898,
      "step": 32009
    },
    {
      "epoch": 3.0526161193847657e-05,
      "step": 32009,
      "training_step_time": 7.909819841384888
    },
    {
      "epoch": 3.052711486816406e-05,
      "grad_norm": 0.20107194781303406,
      "learning_rate": 8.134722588431095e-06,
      "loss": 0.0678,
      "step": 32010
    },
    {
      "epoch": 3.052711486816406e-05,
      "model_forward_time": 1.7251863479614258,
      "step": 32010
    },
    {
      "dynamics_loss_avg": 0.05834823388140649,
      "epoch": 3.052711486816406e-05,
      "step": 32010
    },
    {
      "action_loss_avg": 0.0017082689388189464,
      "epoch": 3.052711486816406e-05,
      "step": 32010
    },
    {
      "epoch": 3.052711486816406e-05,
      "step": 32010,
      "training_step_time": 7.665369749069214
    },
    {
      "epoch": 3.052806854248047e-05,
      "model_forward_time": 1.7192857265472412,
      "step": 32011
    },
    {
      "epoch": 3.052806854248047e-05,
      "step": 32011,
      "training_step_time": 7.73399543762207
    },
    {
      "epoch": 3.0529022216796875e-05,
      "model_forward_time": 1.7828106880187988,
      "step": 32012
    },
    {
      "epoch": 3.0529022216796875e-05,
      "step": 32012,
      "training_step_time": 7.852513313293457
    },
    {
      "epoch": 3.052997589111328e-05,
      "model_forward_time": 1.7458720207214355,
      "step": 32013
    },
    {
      "epoch": 3.052997589111328e-05,
      "step": 32013,
      "training_step_time": 9.70997929573059
    },
    {
      "epoch": 3.053092956542969e-05,
      "model_forward_time": 2.9113385677337646,
      "step": 32014
    },
    {
      "epoch": 3.053092956542969e-05,
      "step": 32014,
      "training_step_time": 9.129564046859741
    },
    {
      "epoch": 3.0531883239746094e-05,
      "model_forward_time": 1.799107551574707,
      "step": 32015
    },
    {
      "epoch": 3.0531883239746094e-05,
      "step": 32015,
      "training_step_time": 9.81594967842102
    },
    {
      "epoch": 3.05328369140625e-05,
      "model_forward_time": 1.733670949935913,
      "step": 32016
    },
    {
      "epoch": 3.05328369140625e-05,
      "step": 32016,
      "training_step_time": 7.878880023956299
    },
    {
      "epoch": 3.053379058837891e-05,
      "model_forward_time": 1.7213077545166016,
      "step": 32017
    },
    {
      "epoch": 3.053379058837891e-05,
      "step": 32017,
      "training_step_time": 7.772962808609009
    },
    {
      "epoch": 3.053474426269531e-05,
      "model_forward_time": 1.7167654037475586,
      "step": 32018
    },
    {
      "epoch": 3.053474426269531e-05,
      "step": 32018,
      "training_step_time": 7.795130491256714
    },
    {
      "epoch": 3.053569793701172e-05,
      "model_forward_time": 1.7288601398468018,
      "step": 32019
    },
    {
      "epoch": 3.053569793701172e-05,
      "step": 32019,
      "training_step_time": 7.672043561935425
    },
    {
      "epoch": 3.053665161132813e-05,
      "grad_norm": 0.16352899372577667,
      "learning_rate": 8.133434258393444e-06,
      "loss": 0.0587,
      "step": 32020
    },
    {
      "epoch": 3.053665161132813e-05,
      "model_forward_time": 1.7352890968322754,
      "step": 32020
    },
    {
      "dynamics_loss_avg": 0.056044742465019226,
      "epoch": 3.053665161132813e-05,
      "step": 32020
    },
    {
      "action_loss_avg": 0.0038149505213368683,
      "epoch": 3.053665161132813e-05,
      "step": 32020
    },
    {
      "epoch": 3.053665161132813e-05,
      "step": 32020,
      "training_step_time": 7.598315954208374
    },
    {
      "epoch": 3.053760528564453e-05,
      "model_forward_time": 1.8205089569091797,
      "step": 32021
    },
    {
      "epoch": 3.053760528564453e-05,
      "step": 32021,
      "training_step_time": 7.722321510314941
    },
    {
      "epoch": 3.0538558959960936e-05,
      "model_forward_time": 1.7520012855529785,
      "step": 32022
    },
    {
      "epoch": 3.0538558959960936e-05,
      "step": 32022,
      "training_step_time": 9.473598718643188
    },
    {
      "epoch": 3.053951263427735e-05,
      "model_forward_time": 2.9143500328063965,
      "step": 32023
    },
    {
      "epoch": 3.053951263427735e-05,
      "step": 32023,
      "training_step_time": 9.22317099571228
    },
    {
      "epoch": 3.054046630859375e-05,
      "model_forward_time": 1.7545769214630127,
      "step": 32024
    },
    {
      "epoch": 3.054046630859375e-05,
      "step": 32024,
      "training_step_time": 9.728837251663208
    },
    {
      "epoch": 3.0541419982910155e-05,
      "model_forward_time": 1.8261778354644775,
      "step": 32025
    },
    {
      "epoch": 3.0541419982910155e-05,
      "step": 32025,
      "training_step_time": 8.230453729629517
    },
    {
      "epoch": 3.0542373657226566e-05,
      "model_forward_time": 1.7125887870788574,
      "step": 32026
    },
    {
      "epoch": 3.0542373657226566e-05,
      "step": 32026,
      "training_step_time": 7.6368088722229
    },
    {
      "epoch": 3.054332733154297e-05,
      "model_forward_time": 1.7317793369293213,
      "step": 32027
    },
    {
      "epoch": 3.054332733154297e-05,
      "step": 32027,
      "training_step_time": 7.589102506637573
    },
    {
      "epoch": 3.0544281005859374e-05,
      "model_forward_time": 1.7284057140350342,
      "step": 32028
    },
    {
      "epoch": 3.0544281005859374e-05,
      "step": 32028,
      "training_step_time": 7.482026100158691
    },
    {
      "epoch": 3.0545234680175785e-05,
      "model_forward_time": 1.7120692729949951,
      "step": 32029
    },
    {
      "epoch": 3.0545234680175785e-05,
      "step": 32029,
      "training_step_time": 7.493513107299805
    },
    {
      "epoch": 3.054618835449219e-05,
      "grad_norm": 0.1872965544462204,
      "learning_rate": 8.132145585688136e-06,
      "loss": 0.0596,
      "step": 32030
    },
    {
      "epoch": 3.054618835449219e-05,
      "model_forward_time": 1.7557957172393799,
      "step": 32030
    },
    {
      "dynamics_loss_avg": 0.05528336279094219,
      "epoch": 3.054618835449219e-05,
      "step": 32030
    },
    {
      "action_loss_avg": 0.0025309445220045744,
      "epoch": 3.054618835449219e-05,
      "step": 32030
    },
    {
      "epoch": 3.054618835449219e-05,
      "step": 32030,
      "training_step_time": 7.6529858112335205
    },
    {
      "epoch": 3.054714202880859e-05,
      "model_forward_time": 1.742321491241455,
      "step": 32031
    },
    {
      "epoch": 3.054714202880859e-05,
      "step": 32031,
      "training_step_time": 9.00276494026184
    },
    {
      "epoch": 3.0548095703125e-05,
      "model_forward_time": 2.8511745929718018,
      "step": 32032
    },
    {
      "epoch": 3.0548095703125e-05,
      "step": 32032,
      "training_step_time": 8.712632179260254
    },
    {
      "epoch": 3.054904937744141e-05,
      "model_forward_time": 1.7271881103515625,
      "step": 32033
    },
    {
      "epoch": 3.054904937744141e-05,
      "step": 32033,
      "training_step_time": 7.783106327056885
    },
    {
      "epoch": 3.055000305175781e-05,
      "model_forward_time": 1.8027656078338623,
      "step": 32034
    },
    {
      "epoch": 3.055000305175781e-05,
      "step": 32034,
      "training_step_time": 7.811874151229858
    },
    {
      "epoch": 3.0550956726074215e-05,
      "model_forward_time": 1.7238519191741943,
      "step": 32035
    },
    {
      "epoch": 3.0550956726074215e-05,
      "step": 32035,
      "training_step_time": 7.779272556304932
    },
    {
      "epoch": 3.0551910400390626e-05,
      "model_forward_time": 1.795137643814087,
      "step": 32036
    },
    {
      "epoch": 3.0551910400390626e-05,
      "step": 32036,
      "training_step_time": 7.53529953956604
    },
    {
      "epoch": 3.055286407470703e-05,
      "model_forward_time": 1.7504136562347412,
      "step": 32037
    },
    {
      "epoch": 3.055286407470703e-05,
      "step": 32037,
      "training_step_time": 7.678543567657471
    },
    {
      "epoch": 3.0553817749023434e-05,
      "model_forward_time": 1.7633254528045654,
      "step": 32038
    },
    {
      "epoch": 3.0553817749023434e-05,
      "step": 32038,
      "training_step_time": 8.467310190200806
    },
    {
      "epoch": 3.0554771423339845e-05,
      "model_forward_time": 1.7546064853668213,
      "step": 32039
    },
    {
      "epoch": 3.0554771423339845e-05,
      "step": 32039,
      "training_step_time": 7.538424730300903
    },
    {
      "epoch": 3.055572509765625e-05,
      "grad_norm": 0.17589612305164337,
      "learning_rate": 8.130856570456094e-06,
      "loss": 0.0603,
      "step": 32040
    },
    {
      "epoch": 3.055572509765625e-05,
      "model_forward_time": 1.8037891387939453,
      "step": 32040
    },
    {
      "dynamics_loss_avg": 0.04461393542587757,
      "epoch": 3.055572509765625e-05,
      "step": 32040
    },
    {
      "action_loss_avg": 0.0024866897176252676,
      "epoch": 3.055572509765625e-05,
      "step": 32040
    },
    {
      "epoch": 3.055572509765625e-05,
      "step": 32040,
      "training_step_time": 8.424557209014893
    },
    {
      "epoch": 3.055667877197265e-05,
      "model_forward_time": 2.8579936027526855,
      "step": 32041
    },
    {
      "epoch": 3.055667877197265e-05,
      "step": 32041,
      "training_step_time": 7.607513666152954
    },
    {
      "epoch": 3.0557632446289064e-05,
      "model_forward_time": 1.75693941116333,
      "step": 32042
    },
    {
      "epoch": 3.0557632446289064e-05,
      "step": 32042,
      "training_step_time": 8.584803104400635
    },
    {
      "epoch": 3.055858612060547e-05,
      "model_forward_time": 1.742673397064209,
      "step": 32043
    },
    {
      "epoch": 3.055858612060547e-05,
      "step": 32043,
      "training_step_time": 11.105373859405518
    },
    {
      "epoch": 3.055953979492187e-05,
      "model_forward_time": 1.8115110397338867,
      "step": 32044
    },
    {
      "epoch": 3.055953979492187e-05,
      "step": 32044,
      "training_step_time": 8.740188598632812
    },
    {
      "epoch": 3.056049346923828e-05,
      "model_forward_time": 1.7677593231201172,
      "step": 32045
    },
    {
      "epoch": 3.056049346923828e-05,
      "step": 32045,
      "training_step_time": 7.607910871505737
    },
    {
      "epoch": 3.056144714355469e-05,
      "model_forward_time": 1.757331132888794,
      "step": 32046
    },
    {
      "epoch": 3.056144714355469e-05,
      "step": 32046,
      "training_step_time": 7.815627574920654
    },
    {
      "epoch": 3.056240081787109e-05,
      "model_forward_time": 1.750807762145996,
      "step": 32047
    },
    {
      "epoch": 3.056240081787109e-05,
      "step": 32047,
      "training_step_time": 7.5903825759887695
    },
    {
      "epoch": 3.05633544921875e-05,
      "model_forward_time": 1.7591359615325928,
      "step": 32048
    },
    {
      "epoch": 3.05633544921875e-05,
      "step": 32048,
      "training_step_time": 7.71952223777771
    },
    {
      "epoch": 3.0564308166503906e-05,
      "model_forward_time": 1.762317419052124,
      "step": 32049
    },
    {
      "epoch": 3.0564308166503906e-05,
      "step": 32049,
      "training_step_time": 8.530844688415527
    },
    {
      "epoch": 3.056526184082031e-05,
      "grad_norm": 0.15085956454277039,
      "learning_rate": 8.129567212838282e-06,
      "loss": 0.0608,
      "step": 32050
    },
    {
      "epoch": 3.056526184082031e-05,
      "model_forward_time": 2.8364179134368896,
      "step": 32050
    },
    {
      "dynamics_loss_avg": 0.06375567633658648,
      "epoch": 3.056526184082031e-05,
      "step": 32050
    },
    {
      "action_loss_avg": 0.0029952038370538504,
      "epoch": 3.056526184082031e-05,
      "step": 32050
    },
    {
      "epoch": 3.056526184082031e-05,
      "step": 32050,
      "training_step_time": 8.968714714050293
    },
    {
      "epoch": 3.056621551513672e-05,
      "model_forward_time": 1.7514641284942627,
      "step": 32051
    },
    {
      "epoch": 3.056621551513672e-05,
      "step": 32051,
      "training_step_time": 8.726889371871948
    },
    {
      "epoch": 3.0567169189453125e-05,
      "model_forward_time": 1.7557399272918701,
      "step": 32052
    },
    {
      "epoch": 3.0567169189453125e-05,
      "step": 32052,
      "training_step_time": 8.526844024658203
    },
    {
      "epoch": 3.056812286376953e-05,
      "model_forward_time": 1.8062083721160889,
      "step": 32053
    },
    {
      "epoch": 3.056812286376953e-05,
      "step": 32053,
      "training_step_time": 7.760727405548096
    },
    {
      "epoch": 3.056907653808594e-05,
      "model_forward_time": 1.7516450881958008,
      "step": 32054
    },
    {
      "epoch": 3.056907653808594e-05,
      "step": 32054,
      "training_step_time": 7.520067453384399
    },
    {
      "epoch": 3.0570030212402343e-05,
      "model_forward_time": 1.7551250457763672,
      "step": 32055
    },
    {
      "epoch": 3.0570030212402343e-05,
      "step": 32055,
      "training_step_time": 7.628075122833252
    },
    {
      "epoch": 3.057098388671875e-05,
      "model_forward_time": 1.7629644870758057,
      "step": 32056
    },
    {
      "epoch": 3.057098388671875e-05,
      "step": 32056,
      "training_step_time": 7.380280256271362
    },
    {
      "epoch": 3.057193756103516e-05,
      "model_forward_time": 1.7590489387512207,
      "step": 32057
    },
    {
      "epoch": 3.057193756103516e-05,
      "step": 32057,
      "training_step_time": 7.564088821411133
    },
    {
      "epoch": 3.057289123535156e-05,
      "model_forward_time": 1.754809856414795,
      "step": 32058
    },
    {
      "epoch": 3.057289123535156e-05,
      "step": 32058,
      "training_step_time": 7.646495819091797
    },
    {
      "epoch": 3.0573844909667966e-05,
      "model_forward_time": 1.761042594909668,
      "step": 32059
    },
    {
      "epoch": 3.0573844909667966e-05,
      "step": 32059,
      "training_step_time": 7.705185890197754
    },
    {
      "epoch": 3.057479858398438e-05,
      "grad_norm": 0.16107717156410217,
      "learning_rate": 8.128277512975708e-06,
      "loss": 0.0628,
      "step": 32060
    },
    {
      "epoch": 3.057479858398438e-05,
      "model_forward_time": 2.959117889404297,
      "step": 32060
    },
    {
      "dynamics_loss_avg": 0.07468297742307187,
      "epoch": 3.057479858398438e-05,
      "step": 32060
    },
    {
      "action_loss_avg": 0.002641136257443577,
      "epoch": 3.057479858398438e-05,
      "step": 32060
    },
    {
      "epoch": 3.057479858398438e-05,
      "step": 32060,
      "training_step_time": 8.3534414768219
    },
    {
      "epoch": 3.057575225830078e-05,
      "model_forward_time": 1.7457382678985596,
      "step": 32061
    },
    {
      "epoch": 3.057575225830078e-05,
      "step": 32061,
      "training_step_time": 9.380367040634155
    },
    {
      "epoch": 3.0576705932617185e-05,
      "model_forward_time": 1.7777373790740967,
      "step": 32062
    },
    {
      "epoch": 3.0576705932617185e-05,
      "step": 32062,
      "training_step_time": 7.767102956771851
    },
    {
      "epoch": 3.0577659606933596e-05,
      "model_forward_time": 1.3637113571166992,
      "step": 32063
    },
    {
      "epoch": 3.0577659606933596e-05,
      "step": 32063,
      "training_step_time": 8.499817848205566
    },
    {
      "epoch": 3.057861328125e-05,
      "model_forward_time": 1.7408199310302734,
      "step": 32064
    },
    {
      "epoch": 3.057861328125e-05,
      "step": 32064,
      "training_step_time": 7.726167917251587
    },
    {
      "epoch": 3.0579566955566404e-05,
      "model_forward_time": 1.812772512435913,
      "step": 32065
    },
    {
      "epoch": 3.0579566955566404e-05,
      "step": 32065,
      "training_step_time": 7.501256704330444
    },
    {
      "epoch": 3.0580520629882815e-05,
      "model_forward_time": 1.7334129810333252,
      "step": 32066
    },
    {
      "epoch": 3.0580520629882815e-05,
      "step": 32066,
      "training_step_time": 7.624379634857178
    },
    {
      "epoch": 3.058147430419922e-05,
      "model_forward_time": 1.7363815307617188,
      "step": 32067
    },
    {
      "epoch": 3.058147430419922e-05,
      "step": 32067,
      "training_step_time": 7.642451286315918
    },
    {
      "epoch": 3.058242797851562e-05,
      "model_forward_time": 1.7291855812072754,
      "step": 32068
    },
    {
      "epoch": 3.058242797851562e-05,
      "step": 32068,
      "training_step_time": 8.587057113647461
    },
    {
      "epoch": 3.0583381652832034e-05,
      "model_forward_time": 2.9758059978485107,
      "step": 32069
    },
    {
      "epoch": 3.0583381652832034e-05,
      "step": 32069,
      "training_step_time": 8.872782468795776
    },
    {
      "epoch": 3.058433532714844e-05,
      "grad_norm": 0.1583779901266098,
      "learning_rate": 8.126987471009409e-06,
      "loss": 0.0662,
      "step": 32070
    },
    {
      "epoch": 3.058433532714844e-05,
      "model_forward_time": 1.7381584644317627,
      "step": 32070
    },
    {
      "dynamics_loss_avg": 0.04850525371730328,
      "epoch": 3.058433532714844e-05,
      "step": 32070
    },
    {
      "action_loss_avg": 0.002699224260868505,
      "epoch": 3.058433532714844e-05,
      "step": 32070
    },
    {
      "epoch": 3.058433532714844e-05,
      "step": 32070,
      "training_step_time": 7.733882665634155
    },
    {
      "epoch": 3.058528900146484e-05,
      "model_forward_time": 1.7490558624267578,
      "step": 32071
    },
    {
      "epoch": 3.058528900146484e-05,
      "step": 32071,
      "training_step_time": 9.762819051742554
    },
    {
      "epoch": 3.058624267578125e-05,
      "model_forward_time": 1.8230712413787842,
      "step": 32072
    },
    {
      "epoch": 3.058624267578125e-05,
      "step": 32072,
      "training_step_time": 7.654135465621948
    },
    {
      "epoch": 3.058719635009766e-05,
      "model_forward_time": 1.3948841094970703,
      "step": 32073
    },
    {
      "epoch": 3.058719635009766e-05,
      "step": 32073,
      "training_step_time": 7.653524160385132
    },
    {
      "epoch": 3.058815002441406e-05,
      "model_forward_time": 1.7436268329620361,
      "step": 32074
    },
    {
      "epoch": 3.058815002441406e-05,
      "step": 32074,
      "training_step_time": 7.4870288372039795
    },
    {
      "epoch": 3.058910369873047e-05,
      "model_forward_time": 1.7494785785675049,
      "step": 32075
    },
    {
      "epoch": 3.058910369873047e-05,
      "step": 32075,
      "training_step_time": 7.576011896133423
    },
    {
      "epoch": 3.0590057373046875e-05,
      "model_forward_time": 1.7414274215698242,
      "step": 32076
    },
    {
      "epoch": 3.0590057373046875e-05,
      "step": 32076,
      "training_step_time": 7.967174530029297
    },
    {
      "epoch": 3.059101104736328e-05,
      "model_forward_time": 1.7421026229858398,
      "step": 32077
    },
    {
      "epoch": 3.059101104736328e-05,
      "step": 32077,
      "training_step_time": 9.101909875869751
    },
    {
      "epoch": 3.059196472167969e-05,
      "model_forward_time": 2.8390963077545166,
      "step": 32078
    },
    {
      "epoch": 3.059196472167969e-05,
      "step": 32078,
      "training_step_time": 8.610951900482178
    },
    {
      "epoch": 3.0592918395996094e-05,
      "model_forward_time": 1.741001844406128,
      "step": 32079
    },
    {
      "epoch": 3.0592918395996094e-05,
      "step": 32079,
      "training_step_time": 8.879409074783325
    },
    {
      "epoch": 3.05938720703125e-05,
      "grad_norm": 0.20922191441059113,
      "learning_rate": 8.12569708708046e-06,
      "loss": 0.0631,
      "step": 32080
    },
    {
      "epoch": 3.05938720703125e-05,
      "model_forward_time": 1.7385814189910889,
      "step": 32080
    },
    {
      "dynamics_loss_avg": 0.05404284875839949,
      "epoch": 3.05938720703125e-05,
      "step": 32080
    },
    {
      "action_loss_avg": 0.0017420571413822473,
      "epoch": 3.05938720703125e-05,
      "step": 32080
    },
    {
      "epoch": 3.05938720703125e-05,
      "step": 32080,
      "training_step_time": 8.83892297744751
    },
    {
      "epoch": 3.059482574462891e-05,
      "model_forward_time": 1.7655017375946045,
      "step": 32081
    },
    {
      "epoch": 3.059482574462891e-05,
      "step": 32081,
      "training_step_time": 7.948880672454834
    },
    {
      "epoch": 3.059577941894531e-05,
      "model_forward_time": 1.7434394359588623,
      "step": 32082
    },
    {
      "epoch": 3.059577941894531e-05,
      "step": 32082,
      "training_step_time": 9.568575620651245
    },
    {
      "epoch": 3.059673309326172e-05,
      "model_forward_time": 1.7510035037994385,
      "step": 32083
    },
    {
      "epoch": 3.059673309326172e-05,
      "step": 32083,
      "training_step_time": 7.45701265335083
    },
    {
      "epoch": 3.059768676757813e-05,
      "model_forward_time": 1.7199170589447021,
      "step": 32084
    },
    {
      "epoch": 3.059768676757813e-05,
      "step": 32084,
      "training_step_time": 7.508570909500122
    },
    {
      "epoch": 3.059864044189453e-05,
      "model_forward_time": 1.7251253128051758,
      "step": 32085
    },
    {
      "epoch": 3.059864044189453e-05,
      "step": 32085,
      "training_step_time": 7.741107702255249
    },
    {
      "epoch": 3.0599594116210936e-05,
      "model_forward_time": 1.3575398921966553,
      "step": 32086
    },
    {
      "epoch": 3.0599594116210936e-05,
      "step": 32086,
      "training_step_time": 7.5214409828186035
    },
    {
      "epoch": 3.060054779052735e-05,
      "model_forward_time": 1.7527623176574707,
      "step": 32087
    },
    {
      "epoch": 3.060054779052735e-05,
      "step": 32087,
      "training_step_time": 9.27942967414856
    },
    {
      "epoch": 3.060150146484375e-05,
      "model_forward_time": 2.9091365337371826,
      "step": 32088
    },
    {
      "epoch": 3.060150146484375e-05,
      "step": 32088,
      "training_step_time": 7.790426015853882
    },
    {
      "epoch": 3.0602455139160155e-05,
      "model_forward_time": 1.7271060943603516,
      "step": 32089
    },
    {
      "epoch": 3.0602455139160155e-05,
      "step": 32089,
      "training_step_time": 8.319335222244263
    },
    {
      "epoch": 3.0603408813476566e-05,
      "grad_norm": 0.1701563447713852,
      "learning_rate": 8.124406361329977e-06,
      "loss": 0.0647,
      "step": 32090
    },
    {
      "epoch": 3.0603408813476566e-05,
      "model_forward_time": 1.7208290100097656,
      "step": 32090
    },
    {
      "dynamics_loss_avg": 0.08008629158139229,
      "epoch": 3.0603408813476566e-05,
      "step": 32090
    },
    {
      "action_loss_avg": 0.003001499257516116,
      "epoch": 3.0603408813476566e-05,
      "step": 32090
    },
    {
      "epoch": 3.0603408813476566e-05,
      "step": 32090,
      "training_step_time": 7.758167505264282
    },
    {
      "epoch": 3.060436248779297e-05,
      "model_forward_time": 1.7419633865356445,
      "step": 32091
    },
    {
      "epoch": 3.060436248779297e-05,
      "step": 32091,
      "training_step_time": 7.599523067474365
    },
    {
      "epoch": 3.0605316162109374e-05,
      "model_forward_time": 1.7255480289459229,
      "step": 32092
    },
    {
      "epoch": 3.0605316162109374e-05,
      "step": 32092,
      "training_step_time": 7.682037353515625
    },
    {
      "epoch": 3.0606269836425785e-05,
      "model_forward_time": 1.7321271896362305,
      "step": 32093
    },
    {
      "epoch": 3.0606269836425785e-05,
      "step": 32093,
      "training_step_time": 7.480267524719238
    },
    {
      "epoch": 3.060722351074219e-05,
      "model_forward_time": 1.7234632968902588,
      "step": 32094
    },
    {
      "epoch": 3.060722351074219e-05,
      "step": 32094,
      "training_step_time": 7.749274015426636
    },
    {
      "epoch": 3.060817718505859e-05,
      "model_forward_time": 1.7230298519134521,
      "step": 32095
    },
    {
      "epoch": 3.060817718505859e-05,
      "step": 32095,
      "training_step_time": 8.108064889907837
    },
    {
      "epoch": 3.0609130859375e-05,
      "model_forward_time": 1.7385613918304443,
      "step": 32096
    },
    {
      "epoch": 3.0609130859375e-05,
      "step": 32096,
      "training_step_time": 7.7618327140808105
    },
    {
      "epoch": 3.061008453369141e-05,
      "model_forward_time": 2.943450689315796,
      "step": 32097
    },
    {
      "epoch": 3.061008453369141e-05,
      "step": 32097,
      "training_step_time": 8.808194875717163
    },
    {
      "epoch": 3.061103820800781e-05,
      "model_forward_time": 1.7317936420440674,
      "step": 32098
    },
    {
      "epoch": 3.061103820800781e-05,
      "step": 32098,
      "training_step_time": 9.665202379226685
    },
    {
      "epoch": 3.0611991882324216e-05,
      "model_forward_time": 1.7400741577148438,
      "step": 32099
    },
    {
      "epoch": 3.0611991882324216e-05,
      "step": 32099,
      "training_step_time": 7.688625812530518
    },
    {
      "epoch": 3.0612945556640626e-05,
      "grad_norm": 0.11669985949993134,
      "learning_rate": 8.123115293899112e-06,
      "loss": 0.0641,
      "step": 32100
    },
    {
      "epoch": 3.0612945556640626e-05,
      "model_forward_time": 1.7169265747070312,
      "step": 32100
    },
    {
      "dynamics_loss_avg": 0.0478258446790278,
      "epoch": 3.0612945556640626e-05,
      "step": 32100
    },
    {
      "action_loss_avg": 0.0022680388879962265,
      "epoch": 3.0612945556640626e-05,
      "step": 32100
    },
    {
      "epoch": 3.0612945556640626e-05,
      "step": 32100,
      "training_step_time": 8.432308197021484
    },
    {
      "epoch": 3.061389923095703e-05,
      "model_forward_time": 1.7267906665802002,
      "step": 32101
    },
    {
      "epoch": 3.061389923095703e-05,
      "step": 32101,
      "training_step_time": 7.675117492675781
    },
    {
      "epoch": 3.0614852905273434e-05,
      "model_forward_time": 1.7329857349395752,
      "step": 32102
    },
    {
      "epoch": 3.0614852905273434e-05,
      "step": 32102,
      "training_step_time": 7.442739009857178
    },
    {
      "epoch": 3.0615806579589845e-05,
      "model_forward_time": 1.7244462966918945,
      "step": 32103
    },
    {
      "epoch": 3.0615806579589845e-05,
      "step": 32103,
      "training_step_time": 8.689655780792236
    },
    {
      "epoch": 3.061676025390625e-05,
      "model_forward_time": 1.726332187652588,
      "step": 32104
    },
    {
      "epoch": 3.061676025390625e-05,
      "step": 32104,
      "training_step_time": 7.785109996795654
    },
    {
      "epoch": 3.061771392822265e-05,
      "model_forward_time": 1.7369422912597656,
      "step": 32105
    },
    {
      "epoch": 3.061771392822265e-05,
      "step": 32105,
      "training_step_time": 8.592998504638672
    },
    {
      "epoch": 3.0618667602539064e-05,
      "model_forward_time": 2.8684897422790527,
      "step": 32106
    },
    {
      "epoch": 3.0618667602539064e-05,
      "step": 32106,
      "training_step_time": 8.748167037963867
    },
    {
      "epoch": 3.061962127685547e-05,
      "model_forward_time": 1.7304322719573975,
      "step": 32107
    },
    {
      "epoch": 3.061962127685547e-05,
      "step": 32107,
      "training_step_time": 10.135502338409424
    },
    {
      "epoch": 3.062057495117187e-05,
      "model_forward_time": 1.7208657264709473,
      "step": 32108
    },
    {
      "epoch": 3.062057495117187e-05,
      "step": 32108,
      "training_step_time": 8.76466703414917
    },
    {
      "epoch": 3.062152862548828e-05,
      "model_forward_time": 1.744074821472168,
      "step": 32109
    },
    {
      "epoch": 3.062152862548828e-05,
      "step": 32109,
      "training_step_time": 7.7261881828308105
    },
    {
      "epoch": 3.062248229980469e-05,
      "grad_norm": 0.108406662940979,
      "learning_rate": 8.121823884929054e-06,
      "loss": 0.0652,
      "step": 32110
    },
    {
      "epoch": 3.062248229980469e-05,
      "model_forward_time": 1.7121944427490234,
      "step": 32110
    },
    {
      "dynamics_loss_avg": 0.05380117781460285,
      "epoch": 3.062248229980469e-05,
      "step": 32110
    },
    {
      "action_loss_avg": 0.0029639769229106603,
      "epoch": 3.062248229980469e-05,
      "step": 32110
    },
    {
      "epoch": 3.062248229980469e-05,
      "step": 32110,
      "training_step_time": 7.754826307296753
    },
    {
      "epoch": 3.062343597412109e-05,
      "model_forward_time": 1.7373063564300537,
      "step": 32111
    },
    {
      "epoch": 3.062343597412109e-05,
      "step": 32111,
      "training_step_time": 7.512296438217163
    },
    {
      "epoch": 3.06243896484375e-05,
      "model_forward_time": 1.7225837707519531,
      "step": 32112
    },
    {
      "epoch": 3.06243896484375e-05,
      "step": 32112,
      "training_step_time": 8.273271322250366
    },
    {
      "epoch": 3.0625343322753906e-05,
      "model_forward_time": 1.7300677299499512,
      "step": 32113
    },
    {
      "epoch": 3.0625343322753906e-05,
      "step": 32113,
      "training_step_time": 7.851679086685181
    },
    {
      "epoch": 3.062629699707031e-05,
      "model_forward_time": 1.734236478805542,
      "step": 32114
    },
    {
      "epoch": 3.062629699707031e-05,
      "step": 32114,
      "training_step_time": 8.46891188621521
    },
    {
      "epoch": 3.062725067138672e-05,
      "model_forward_time": 2.829503059387207,
      "step": 32115
    },
    {
      "epoch": 3.062725067138672e-05,
      "step": 32115,
      "training_step_time": 8.687711477279663
    },
    {
      "epoch": 3.0628204345703125e-05,
      "model_forward_time": 1.7367961406707764,
      "step": 32116
    },
    {
      "epoch": 3.0628204345703125e-05,
      "step": 32116,
      "training_step_time": 8.965857982635498
    },
    {
      "epoch": 3.062915802001953e-05,
      "model_forward_time": 1.7332499027252197,
      "step": 32117
    },
    {
      "epoch": 3.062915802001953e-05,
      "step": 32117,
      "training_step_time": 7.762510776519775
    },
    {
      "epoch": 3.063011169433594e-05,
      "model_forward_time": 1.7546143531799316,
      "step": 32118
    },
    {
      "epoch": 3.063011169433594e-05,
      "step": 32118,
      "training_step_time": 7.707805156707764
    },
    {
      "epoch": 3.0631065368652344e-05,
      "model_forward_time": 1.7348012924194336,
      "step": 32119
    },
    {
      "epoch": 3.0631065368652344e-05,
      "step": 32119,
      "training_step_time": 7.69383430480957
    },
    {
      "epoch": 3.063201904296875e-05,
      "grad_norm": 0.1496153175830841,
      "learning_rate": 8.12053213456103e-06,
      "loss": 0.0609,
      "step": 32120
    },
    {
      "epoch": 3.063201904296875e-05,
      "model_forward_time": 1.7475512027740479,
      "step": 32120
    },
    {
      "dynamics_loss_avg": 0.07050656415522098,
      "epoch": 3.063201904296875e-05,
      "step": 32120
    },
    {
      "action_loss_avg": 0.0022302775701973586,
      "epoch": 3.063201904296875e-05,
      "step": 32120
    },
    {
      "epoch": 3.063201904296875e-05,
      "step": 32120,
      "training_step_time": 7.5110557079315186
    },
    {
      "epoch": 3.063297271728516e-05,
      "model_forward_time": 1.7428805828094482,
      "step": 32121
    },
    {
      "epoch": 3.063297271728516e-05,
      "step": 32121,
      "training_step_time": 7.664936065673828
    },
    {
      "epoch": 3.063392639160156e-05,
      "model_forward_time": 1.7372288703918457,
      "step": 32122
    },
    {
      "epoch": 3.063392639160156e-05,
      "step": 32122,
      "training_step_time": 7.859413385391235
    },
    {
      "epoch": 3.0634880065917966e-05,
      "model_forward_time": 1.7416012287139893,
      "step": 32123
    },
    {
      "epoch": 3.0634880065917966e-05,
      "step": 32123,
      "training_step_time": 7.788464307785034
    },
    {
      "epoch": 3.063583374023438e-05,
      "model_forward_time": 2.854196310043335,
      "step": 32124
    },
    {
      "epoch": 3.063583374023438e-05,
      "step": 32124,
      "training_step_time": 8.277895212173462
    },
    {
      "epoch": 3.063678741455078e-05,
      "model_forward_time": 1.741403579711914,
      "step": 32125
    },
    {
      "epoch": 3.063678741455078e-05,
      "step": 32125,
      "training_step_time": 8.308409929275513
    },
    {
      "epoch": 3.0637741088867185e-05,
      "model_forward_time": 1.7374916076660156,
      "step": 32126
    },
    {
      "epoch": 3.0637741088867185e-05,
      "step": 32126,
      "training_step_time": 8.916513204574585
    },
    {
      "epoch": 3.0638694763183596e-05,
      "model_forward_time": 1.7179830074310303,
      "step": 32127
    },
    {
      "epoch": 3.0638694763183596e-05,
      "step": 32127,
      "training_step_time": 8.493952512741089
    },
    {
      "epoch": 3.06396484375e-05,
      "model_forward_time": 1.7304470539093018,
      "step": 32128
    },
    {
      "epoch": 3.06396484375e-05,
      "step": 32128,
      "training_step_time": 7.649317264556885
    },
    {
      "epoch": 3.0640602111816404e-05,
      "model_forward_time": 1.733372688293457,
      "step": 32129
    },
    {
      "epoch": 3.0640602111816404e-05,
      "step": 32129,
      "training_step_time": 8.48068642616272
    },
    {
      "epoch": 3.0641555786132815e-05,
      "grad_norm": 0.11297358572483063,
      "learning_rate": 8.119240042936303e-06,
      "loss": 0.0643,
      "step": 32130
    },
    {
      "epoch": 3.0641555786132815e-05,
      "model_forward_time": 1.7269744873046875,
      "step": 32130
    },
    {
      "dynamics_loss_avg": 0.041924660466611384,
      "epoch": 3.0641555786132815e-05,
      "step": 32130
    },
    {
      "action_loss_avg": 0.0019253892824053764,
      "epoch": 3.0641555786132815e-05,
      "step": 32130
    },
    {
      "epoch": 3.0641555786132815e-05,
      "step": 32130,
      "training_step_time": 7.71967625617981
    },
    {
      "epoch": 3.064250946044922e-05,
      "model_forward_time": 1.7192411422729492,
      "step": 32131
    },
    {
      "epoch": 3.064250946044922e-05,
      "step": 32131,
      "training_step_time": 8.533313512802124
    },
    {
      "epoch": 3.064346313476562e-05,
      "model_forward_time": 1.7257025241851807,
      "step": 32132
    },
    {
      "epoch": 3.064346313476562e-05,
      "step": 32132,
      "training_step_time": 7.806171655654907
    },
    {
      "epoch": 3.0644416809082034e-05,
      "model_forward_time": 1.3766636848449707,
      "step": 32133
    },
    {
      "epoch": 3.0644416809082034e-05,
      "step": 32133,
      "training_step_time": 9.349037170410156
    },
    {
      "epoch": 3.064537048339844e-05,
      "model_forward_time": 2.880797863006592,
      "step": 32134
    },
    {
      "epoch": 3.064537048339844e-05,
      "step": 32134,
      "training_step_time": 8.813011884689331
    },
    {
      "epoch": 3.064632415771484e-05,
      "model_forward_time": 1.7321534156799316,
      "step": 32135
    },
    {
      "epoch": 3.064632415771484e-05,
      "step": 32135,
      "training_step_time": 8.310590744018555
    },
    {
      "epoch": 3.064727783203125e-05,
      "model_forward_time": 1.3877019882202148,
      "step": 32136
    },
    {
      "epoch": 3.064727783203125e-05,
      "step": 32136,
      "training_step_time": 8.499941349029541
    },
    {
      "epoch": 3.064823150634766e-05,
      "model_forward_time": 1.7631075382232666,
      "step": 32137
    },
    {
      "epoch": 3.064823150634766e-05,
      "step": 32137,
      "training_step_time": 7.673701286315918
    },
    {
      "epoch": 3.064918518066406e-05,
      "model_forward_time": 1.7509400844573975,
      "step": 32138
    },
    {
      "epoch": 3.064918518066406e-05,
      "step": 32138,
      "training_step_time": 7.83538293838501
    },
    {
      "epoch": 3.065013885498047e-05,
      "model_forward_time": 1.7535755634307861,
      "step": 32139
    },
    {
      "epoch": 3.065013885498047e-05,
      "step": 32139,
      "training_step_time": 8.451197862625122
    },
    {
      "epoch": 3.0651092529296876e-05,
      "grad_norm": 0.11194565892219543,
      "learning_rate": 8.117947610196175e-06,
      "loss": 0.0615,
      "step": 32140
    },
    {
      "epoch": 3.0651092529296876e-05,
      "model_forward_time": 1.745363473892212,
      "step": 32140
    },
    {
      "dynamics_loss_avg": 0.05550749124959111,
      "epoch": 3.0651092529296876e-05,
      "step": 32140
    },
    {
      "action_loss_avg": 0.0018625056138262153,
      "epoch": 3.0651092529296876e-05,
      "step": 32140
    },
    {
      "epoch": 3.0651092529296876e-05,
      "step": 32140,
      "training_step_time": 7.657453775405884
    },
    {
      "epoch": 3.065204620361328e-05,
      "model_forward_time": 1.7532894611358643,
      "step": 32141
    },
    {
      "epoch": 3.065204620361328e-05,
      "step": 32141,
      "training_step_time": 7.831512928009033
    },
    {
      "epoch": 3.065299987792969e-05,
      "model_forward_time": 1.7633404731750488,
      "step": 32142
    },
    {
      "epoch": 3.065299987792969e-05,
      "step": 32142,
      "training_step_time": 10.966142654418945
    },
    {
      "epoch": 3.0653953552246094e-05,
      "model_forward_time": 2.9052722454071045,
      "step": 32143
    },
    {
      "epoch": 3.0653953552246094e-05,
      "step": 32143,
      "training_step_time": 9.802469968795776
    },
    {
      "epoch": 3.06549072265625e-05,
      "model_forward_time": 1.7513377666473389,
      "step": 32144
    },
    {
      "epoch": 3.06549072265625e-05,
      "step": 32144,
      "training_step_time": 7.716607093811035
    },
    {
      "epoch": 3.065586090087891e-05,
      "model_forward_time": 1.749297857284546,
      "step": 32145
    },
    {
      "epoch": 3.065586090087891e-05,
      "step": 32145,
      "training_step_time": 7.830552101135254
    },
    {
      "epoch": 3.065681457519531e-05,
      "model_forward_time": 1.7628173828125,
      "step": 32146
    },
    {
      "epoch": 3.065681457519531e-05,
      "step": 32146,
      "training_step_time": 8.506557941436768
    },
    {
      "epoch": 3.065776824951172e-05,
      "model_forward_time": 1.7463316917419434,
      "step": 32147
    },
    {
      "epoch": 3.065776824951172e-05,
      "step": 32147,
      "training_step_time": 7.826002359390259
    },
    {
      "epoch": 3.065872192382813e-05,
      "model_forward_time": 1.7168855667114258,
      "step": 32148
    },
    {
      "epoch": 3.065872192382813e-05,
      "step": 32148,
      "training_step_time": 8.061696529388428
    },
    {
      "epoch": 3.065967559814453e-05,
      "model_forward_time": 1.7292380332946777,
      "step": 32149
    },
    {
      "epoch": 3.065967559814453e-05,
      "step": 32149,
      "training_step_time": 7.684618234634399
    },
    {
      "epoch": 3.0660629272460936e-05,
      "grad_norm": 0.1713307946920395,
      "learning_rate": 8.116654836481984e-06,
      "loss": 0.0646,
      "step": 32150
    },
    {
      "epoch": 3.0660629272460936e-05,
      "model_forward_time": 1.7296395301818848,
      "step": 32150
    },
    {
      "dynamics_loss_avg": 0.04829064626246691,
      "epoch": 3.0660629272460936e-05,
      "step": 32150
    },
    {
      "action_loss_avg": 0.0009668738814070821,
      "epoch": 3.0660629272460936e-05,
      "step": 32150
    },
    {
      "epoch": 3.0660629272460936e-05,
      "step": 32150,
      "training_step_time": 8.83025312423706
    },
    {
      "epoch": 3.066158294677735e-05,
      "model_forward_time": 1.741806983947754,
      "step": 32151
    },
    {
      "epoch": 3.066158294677735e-05,
      "step": 32151,
      "training_step_time": 8.60150694847107
    },
    {
      "epoch": 3.066253662109375e-05,
      "model_forward_time": 2.8638052940368652,
      "step": 32152
    },
    {
      "epoch": 3.066253662109375e-05,
      "step": 32152,
      "training_step_time": 8.322949647903442
    },
    {
      "epoch": 3.0663490295410155e-05,
      "model_forward_time": 1.727245807647705,
      "step": 32153
    },
    {
      "epoch": 3.0663490295410155e-05,
      "step": 32153,
      "training_step_time": 9.559971570968628
    },
    {
      "epoch": 3.0664443969726566e-05,
      "model_forward_time": 1.36958646774292,
      "step": 32154
    },
    {
      "epoch": 3.0664443969726566e-05,
      "step": 32154,
      "training_step_time": 7.799542188644409
    },
    {
      "epoch": 3.066539764404297e-05,
      "model_forward_time": 1.727085828781128,
      "step": 32155
    },
    {
      "epoch": 3.066539764404297e-05,
      "step": 32155,
      "training_step_time": 7.759708642959595
    },
    {
      "epoch": 3.0666351318359374e-05,
      "model_forward_time": 1.7277631759643555,
      "step": 32156
    },
    {
      "epoch": 3.0666351318359374e-05,
      "step": 32156,
      "training_step_time": 8.572057723999023
    },
    {
      "epoch": 3.066730499267578e-05,
      "model_forward_time": 1.7316229343414307,
      "step": 32157
    },
    {
      "epoch": 3.066730499267578e-05,
      "step": 32157,
      "training_step_time": 7.5213212966918945
    },
    {
      "epoch": 3.066825866699219e-05,
      "model_forward_time": 1.734576940536499,
      "step": 32158
    },
    {
      "epoch": 3.066825866699219e-05,
      "step": 32158,
      "training_step_time": 7.691347122192383
    },
    {
      "epoch": 3.066921234130859e-05,
      "model_forward_time": 1.726224660873413,
      "step": 32159
    },
    {
      "epoch": 3.066921234130859e-05,
      "step": 32159,
      "training_step_time": 7.741173267364502
    },
    {
      "epoch": 3.0670166015625e-05,
      "grad_norm": 0.18131019175052643,
      "learning_rate": 8.115361721935106e-06,
      "loss": 0.064,
      "step": 32160
    },
    {
      "epoch": 3.0670166015625e-05,
      "model_forward_time": 1.7172143459320068,
      "step": 32160
    },
    {
      "dynamics_loss_avg": 0.06859926674515009,
      "epoch": 3.0670166015625e-05,
      "step": 32160
    },
    {
      "action_loss_avg": 0.0033011754625476898,
      "epoch": 3.0670166015625e-05,
      "step": 32160
    },
    {
      "epoch": 3.0670166015625e-05,
      "step": 32160,
      "training_step_time": 8.539188861846924
    },
    {
      "epoch": 3.067111968994141e-05,
      "model_forward_time": 1.732133150100708,
      "step": 32161
    },
    {
      "epoch": 3.067111968994141e-05,
      "step": 32161,
      "training_step_time": 8.707553386688232
    },
    {
      "epoch": 3.067207336425781e-05,
      "model_forward_time": 2.8788461685180664,
      "step": 32162
    },
    {
      "epoch": 3.067207336425781e-05,
      "step": 32162,
      "training_step_time": 8.7761869430542
    },
    {
      "epoch": 3.0673027038574216e-05,
      "model_forward_time": 1.7333202362060547,
      "step": 32163
    },
    {
      "epoch": 3.0673027038574216e-05,
      "step": 32163,
      "training_step_time": 8.56243634223938
    },
    {
      "epoch": 3.0673980712890626e-05,
      "model_forward_time": 1.7380938529968262,
      "step": 32164
    },
    {
      "epoch": 3.0673980712890626e-05,
      "step": 32164,
      "training_step_time": 7.790362119674683
    },
    {
      "epoch": 3.067493438720703e-05,
      "model_forward_time": 1.7489292621612549,
      "step": 32165
    },
    {
      "epoch": 3.067493438720703e-05,
      "step": 32165,
      "training_step_time": 7.722978830337524
    },
    {
      "epoch": 3.0675888061523434e-05,
      "model_forward_time": 1.7264485359191895,
      "step": 32166
    },
    {
      "epoch": 3.0675888061523434e-05,
      "step": 32166,
      "training_step_time": 7.524293661117554
    },
    {
      "epoch": 3.0676841735839845e-05,
      "model_forward_time": 1.74161958694458,
      "step": 32167
    },
    {
      "epoch": 3.0676841735839845e-05,
      "step": 32167,
      "training_step_time": 7.638260364532471
    },
    {
      "epoch": 3.067779541015625e-05,
      "model_forward_time": 1.7284276485443115,
      "step": 32168
    },
    {
      "epoch": 3.067779541015625e-05,
      "step": 32168,
      "training_step_time": 8.340369462966919
    },
    {
      "epoch": 3.067874908447265e-05,
      "model_forward_time": 1.7408318519592285,
      "step": 32169
    },
    {
      "epoch": 3.067874908447265e-05,
      "step": 32169,
      "training_step_time": 8.512133836746216
    },
    {
      "epoch": 3.0679702758789064e-05,
      "grad_norm": 0.12506438791751862,
      "learning_rate": 8.114068266696953e-06,
      "loss": 0.0637,
      "step": 32170
    },
    {
      "epoch": 3.0679702758789064e-05,
      "model_forward_time": 1.7656724452972412,
      "step": 32170
    },
    {
      "dynamics_loss_avg": 0.07247337847948074,
      "epoch": 3.0679702758789064e-05,
      "step": 32170
    },
    {
      "action_loss_avg": 0.002380688255652785,
      "epoch": 3.0679702758789064e-05,
      "step": 32170
    },
    {
      "epoch": 3.0679702758789064e-05,
      "step": 32170,
      "training_step_time": 7.727622985839844
    },
    {
      "epoch": 3.068065643310547e-05,
      "model_forward_time": 2.9085886478424072,
      "step": 32171
    },
    {
      "epoch": 3.068065643310547e-05,
      "step": 32171,
      "training_step_time": 7.843196153640747
    },
    {
      "epoch": 3.068161010742187e-05,
      "model_forward_time": 1.7302825450897217,
      "step": 32172
    },
    {
      "epoch": 3.068161010742187e-05,
      "step": 32172,
      "training_step_time": 8.571237802505493
    },
    {
      "epoch": 3.068256378173828e-05,
      "model_forward_time": 1.7170860767364502,
      "step": 32173
    },
    {
      "epoch": 3.068256378173828e-05,
      "step": 32173,
      "training_step_time": 7.791114330291748
    },
    {
      "epoch": 3.068351745605469e-05,
      "model_forward_time": 1.747767448425293,
      "step": 32174
    },
    {
      "epoch": 3.068351745605469e-05,
      "step": 32174,
      "training_step_time": 8.61017656326294
    },
    {
      "epoch": 3.068447113037109e-05,
      "model_forward_time": 1.7242717742919922,
      "step": 32175
    },
    {
      "epoch": 3.068447113037109e-05,
      "step": 32175,
      "training_step_time": 7.539462089538574
    },
    {
      "epoch": 3.06854248046875e-05,
      "model_forward_time": 1.718409776687622,
      "step": 32176
    },
    {
      "epoch": 3.06854248046875e-05,
      "step": 32176,
      "training_step_time": 7.749222755432129
    },
    {
      "epoch": 3.0686378479003906e-05,
      "model_forward_time": 1.7351524829864502,
      "step": 32177
    },
    {
      "epoch": 3.0686378479003906e-05,
      "step": 32177,
      "training_step_time": 7.761783838272095
    },
    {
      "epoch": 3.068733215332031e-05,
      "model_forward_time": 1.734046459197998,
      "step": 32178
    },
    {
      "epoch": 3.068733215332031e-05,
      "step": 32178,
      "training_step_time": 7.7726147174835205
    },
    {
      "epoch": 3.068828582763672e-05,
      "model_forward_time": 1.7465555667877197,
      "step": 32179
    },
    {
      "epoch": 3.068828582763672e-05,
      "step": 32179,
      "training_step_time": 9.601584672927856
    },
    {
      "epoch": 3.0689239501953125e-05,
      "grad_norm": 0.09404583275318146,
      "learning_rate": 8.112774470908977e-06,
      "loss": 0.062,
      "step": 32180
    },
    {
      "epoch": 3.0689239501953125e-05,
      "model_forward_time": 2.8856582641601562,
      "step": 32180
    },
    {
      "dynamics_loss_avg": 0.05540055017918348,
      "epoch": 3.0689239501953125e-05,
      "step": 32180
    },
    {
      "action_loss_avg": 0.001909333057119511,
      "epoch": 3.0689239501953125e-05,
      "step": 32180
    },
    {
      "epoch": 3.0689239501953125e-05,
      "step": 32180,
      "training_step_time": 7.8073320388793945
    },
    {
      "epoch": 3.069019317626953e-05,
      "model_forward_time": 1.7369394302368164,
      "step": 32181
    },
    {
      "epoch": 3.069019317626953e-05,
      "step": 32181,
      "training_step_time": 8.21579384803772
    },
    {
      "epoch": 3.069114685058594e-05,
      "model_forward_time": 1.728496789932251,
      "step": 32182
    },
    {
      "epoch": 3.069114685058594e-05,
      "step": 32182,
      "training_step_time": 7.740747928619385
    },
    {
      "epoch": 3.0692100524902344e-05,
      "model_forward_time": 1.7426140308380127,
      "step": 32183
    },
    {
      "epoch": 3.0692100524902344e-05,
      "step": 32183,
      "training_step_time": 7.7630956172943115
    },
    {
      "epoch": 3.069305419921875e-05,
      "model_forward_time": 1.7380988597869873,
      "step": 32184
    },
    {
      "epoch": 3.069305419921875e-05,
      "step": 32184,
      "training_step_time": 7.750937223434448
    },
    {
      "epoch": 3.069400787353516e-05,
      "model_forward_time": 1.7247588634490967,
      "step": 32185
    },
    {
      "epoch": 3.069400787353516e-05,
      "step": 32185,
      "training_step_time": 8.410554885864258
    },
    {
      "epoch": 3.069496154785156e-05,
      "model_forward_time": 1.728468418121338,
      "step": 32186
    },
    {
      "epoch": 3.069496154785156e-05,
      "step": 32186,
      "training_step_time": 7.770313501358032
    },
    {
      "epoch": 3.0695915222167966e-05,
      "model_forward_time": 1.7242405414581299,
      "step": 32187
    },
    {
      "epoch": 3.0695915222167966e-05,
      "step": 32187,
      "training_step_time": 7.849056243896484
    },
    {
      "epoch": 3.069686889648438e-05,
      "model_forward_time": 1.7513816356658936,
      "step": 32188
    },
    {
      "epoch": 3.069686889648438e-05,
      "step": 32188,
      "training_step_time": 7.731584787368774
    },
    {
      "epoch": 3.069782257080078e-05,
      "model_forward_time": 2.4833438396453857,
      "step": 32189
    },
    {
      "epoch": 3.069782257080078e-05,
      "step": 32189,
      "training_step_time": 9.785176038742065
    },
    {
      "epoch": 3.0698776245117185e-05,
      "grad_norm": 0.20915307104587555,
      "learning_rate": 8.111480334712664e-06,
      "loss": 0.0608,
      "step": 32190
    },
    {
      "epoch": 3.0698776245117185e-05,
      "model_forward_time": 1.7215256690979004,
      "step": 32190
    },
    {
      "dynamics_loss_avg": 0.05616684500128031,
      "epoch": 3.0698776245117185e-05,
      "step": 32190
    },
    {
      "action_loss_avg": 0.002152124640997499,
      "epoch": 3.0698776245117185e-05,
      "step": 32190
    },
    {
      "epoch": 3.0698776245117185e-05,
      "step": 32190,
      "training_step_time": 10.275270938873291
    },
    {
      "epoch": 3.0699729919433596e-05,
      "model_forward_time": 1.73738694190979,
      "step": 32191
    },
    {
      "epoch": 3.0699729919433596e-05,
      "step": 32191,
      "training_step_time": 7.7928009033203125
    },
    {
      "epoch": 3.070068359375e-05,
      "model_forward_time": 1.7066903114318848,
      "step": 32192
    },
    {
      "epoch": 3.070068359375e-05,
      "step": 32192,
      "training_step_time": 7.736051559448242
    },
    {
      "epoch": 3.0701637268066404e-05,
      "model_forward_time": 1.739497423171997,
      "step": 32193
    },
    {
      "epoch": 3.0701637268066404e-05,
      "step": 32193,
      "training_step_time": 7.804998159408569
    },
    {
      "epoch": 3.0702590942382815e-05,
      "model_forward_time": 1.7361490726470947,
      "step": 32194
    },
    {
      "epoch": 3.0702590942382815e-05,
      "step": 32194,
      "training_step_time": 7.6414008140563965
    },
    {
      "epoch": 3.070354461669922e-05,
      "model_forward_time": 1.7155086994171143,
      "step": 32195
    },
    {
      "epoch": 3.070354461669922e-05,
      "step": 32195,
      "training_step_time": 7.664267301559448
    },
    {
      "epoch": 3.070449829101562e-05,
      "model_forward_time": 1.7337746620178223,
      "step": 32196
    },
    {
      "epoch": 3.070449829101562e-05,
      "step": 32196,
      "training_step_time": 7.838099956512451
    },
    {
      "epoch": 3.0705451965332034e-05,
      "model_forward_time": 1.7285194396972656,
      "step": 32197
    },
    {
      "epoch": 3.0705451965332034e-05,
      "step": 32197,
      "training_step_time": 7.704689979553223
    },
    {
      "epoch": 3.070640563964844e-05,
      "model_forward_time": 2.8166356086730957,
      "step": 32198
    },
    {
      "epoch": 3.070640563964844e-05,
      "step": 32198,
      "training_step_time": 8.575624227523804
    },
    {
      "epoch": 3.070735931396484e-05,
      "model_forward_time": 1.7258212566375732,
      "step": 32199
    },
    {
      "epoch": 3.070735931396484e-05,
      "step": 32199,
      "training_step_time": 7.739729166030884
    },
    {
      "epoch": 3.070831298828125e-05,
      "grad_norm": 0.16889768838882446,
      "learning_rate": 8.110185858249542e-06,
      "loss": 0.058,
      "step": 32200
    },
    {
      "epoch": 3.070831298828125e-05,
      "model_forward_time": 1.720336675643921,
      "step": 32200
    },
    {
      "dynamics_loss_avg": 0.054779023956507446,
      "epoch": 3.070831298828125e-05,
      "step": 32200
    },
    {
      "action_loss_avg": 0.002597746980609372,
      "epoch": 3.070831298828125e-05,
      "step": 32200
    },
    {
      "epoch": 3.070831298828125e-05,
      "step": 32200,
      "training_step_time": 8.761703252792358
    },
    {
      "epoch": 3.070926666259766e-05,
      "model_forward_time": 1.7298552989959717,
      "step": 32201
    },
    {
      "epoch": 3.070926666259766e-05,
      "step": 32201,
      "training_step_time": 7.865009307861328
    },
    {
      "epoch": 3.071022033691406e-05,
      "model_forward_time": 1.748438835144043,
      "step": 32202
    },
    {
      "epoch": 3.071022033691406e-05,
      "step": 32202,
      "training_step_time": 7.891539096832275
    },
    {
      "epoch": 3.071117401123047e-05,
      "model_forward_time": 1.7296111583709717,
      "step": 32203
    },
    {
      "epoch": 3.071117401123047e-05,
      "step": 32203,
      "training_step_time": 8.135757207870483
    },
    {
      "epoch": 3.0712127685546876e-05,
      "model_forward_time": 1.7214484214782715,
      "step": 32204
    },
    {
      "epoch": 3.0712127685546876e-05,
      "step": 32204,
      "training_step_time": 8.240756273269653
    },
    {
      "epoch": 3.071308135986328e-05,
      "model_forward_time": 1.7298319339752197,
      "step": 32205
    },
    {
      "epoch": 3.071308135986328e-05,
      "step": 32205,
      "training_step_time": 7.826423406600952
    },
    {
      "epoch": 3.071403503417969e-05,
      "model_forward_time": 1.7607707977294922,
      "step": 32206
    },
    {
      "epoch": 3.071403503417969e-05,
      "step": 32206,
      "training_step_time": 8.441458463668823
    },
    {
      "epoch": 3.0714988708496094e-05,
      "model_forward_time": 1.731560230255127,
      "step": 32207
    },
    {
      "epoch": 3.0714988708496094e-05,
      "step": 32207,
      "training_step_time": 9.706915378570557
    },
    {
      "epoch": 3.07159423828125e-05,
      "model_forward_time": 2.9357614517211914,
      "step": 32208
    },
    {
      "epoch": 3.07159423828125e-05,
      "step": 32208,
      "training_step_time": 7.818923711776733
    },
    {
      "epoch": 3.071689605712891e-05,
      "model_forward_time": 1.7256262302398682,
      "step": 32209
    },
    {
      "epoch": 3.071689605712891e-05,
      "step": 32209,
      "training_step_time": 7.690816164016724
    },
    {
      "epoch": 3.071784973144531e-05,
      "grad_norm": 0.13017423450946808,
      "learning_rate": 8.108891041661168e-06,
      "loss": 0.0645,
      "step": 32210
    },
    {
      "epoch": 3.071784973144531e-05,
      "model_forward_time": 1.7259190082550049,
      "step": 32210
    },
    {
      "dynamics_loss_avg": 0.062232038006186485,
      "epoch": 3.071784973144531e-05,
      "step": 32210
    },
    {
      "action_loss_avg": 0.002036918338853866,
      "epoch": 3.071784973144531e-05,
      "step": 32210
    },
    {
      "epoch": 3.071784973144531e-05,
      "step": 32210,
      "training_step_time": 7.868098020553589
    },
    {
      "epoch": 3.071880340576172e-05,
      "model_forward_time": 1.7698934078216553,
      "step": 32211
    },
    {
      "epoch": 3.071880340576172e-05,
      "step": 32211,
      "training_step_time": 7.83191990852356
    },
    {
      "epoch": 3.071975708007813e-05,
      "model_forward_time": 1.3742399215698242,
      "step": 32212
    },
    {
      "epoch": 3.071975708007813e-05,
      "step": 32212,
      "training_step_time": 7.568334341049194
    },
    {
      "epoch": 3.072071075439453e-05,
      "model_forward_time": 1.7450544834136963,
      "step": 32213
    },
    {
      "epoch": 3.072071075439453e-05,
      "step": 32213,
      "training_step_time": 7.754059791564941
    },
    {
      "epoch": 3.0721664428710936e-05,
      "model_forward_time": 1.3820183277130127,
      "step": 32214
    },
    {
      "epoch": 3.0721664428710936e-05,
      "step": 32214,
      "training_step_time": 7.701159477233887
    },
    {
      "epoch": 3.072261810302735e-05,
      "model_forward_time": 1.7666599750518799,
      "step": 32215
    },
    {
      "epoch": 3.072261810302735e-05,
      "step": 32215,
      "training_step_time": 9.238868713378906
    },
    {
      "epoch": 3.072357177734375e-05,
      "model_forward_time": 1.7378766536712646,
      "step": 32216
    },
    {
      "epoch": 3.072357177734375e-05,
      "step": 32216,
      "training_step_time": 7.77789831161499
    },
    {
      "epoch": 3.0724525451660155e-05,
      "model_forward_time": 2.8502376079559326,
      "step": 32217
    },
    {
      "epoch": 3.0724525451660155e-05,
      "step": 32217,
      "training_step_time": 7.882835149765015
    },
    {
      "epoch": 3.0725479125976566e-05,
      "model_forward_time": 1.745898962020874,
      "step": 32218
    },
    {
      "epoch": 3.0725479125976566e-05,
      "step": 32218,
      "training_step_time": 7.693264722824097
    },
    {
      "epoch": 3.072643280029297e-05,
      "model_forward_time": 1.7412004470825195,
      "step": 32219
    },
    {
      "epoch": 3.072643280029297e-05,
      "step": 32219,
      "training_step_time": 7.800455331802368
    },
    {
      "epoch": 3.0727386474609374e-05,
      "grad_norm": 0.15144559741020203,
      "learning_rate": 8.107595885089146e-06,
      "loss": 0.0598,
      "step": 32220
    },
    {
      "epoch": 3.0727386474609374e-05,
      "model_forward_time": 1.746711254119873,
      "step": 32220
    },
    {
      "dynamics_loss_avg": 0.05987624228000641,
      "epoch": 3.0727386474609374e-05,
      "step": 32220
    },
    {
      "action_loss_avg": 0.00217244642553851,
      "epoch": 3.0727386474609374e-05,
      "step": 32220
    },
    {
      "epoch": 3.0727386474609374e-05,
      "step": 32220,
      "training_step_time": 7.841891050338745
    },
    {
      "epoch": 3.072834014892578e-05,
      "model_forward_time": 1.7494475841522217,
      "step": 32221
    },
    {
      "epoch": 3.072834014892578e-05,
      "step": 32221,
      "training_step_time": 7.578370571136475
    },
    {
      "epoch": 3.072929382324219e-05,
      "model_forward_time": 1.7223129272460938,
      "step": 32222
    },
    {
      "epoch": 3.072929382324219e-05,
      "step": 32222,
      "training_step_time": 8.626346111297607
    },
    {
      "epoch": 3.073024749755859e-05,
      "model_forward_time": 1.7283952236175537,
      "step": 32223
    },
    {
      "epoch": 3.073024749755859e-05,
      "step": 32223,
      "training_step_time": 8.277302265167236
    },
    {
      "epoch": 3.0731201171875e-05,
      "model_forward_time": 1.7526640892028809,
      "step": 32224
    },
    {
      "epoch": 3.0731201171875e-05,
      "step": 32224,
      "training_step_time": 7.7473464012146
    },
    {
      "epoch": 3.073215484619141e-05,
      "model_forward_time": 1.7386763095855713,
      "step": 32225
    },
    {
      "epoch": 3.073215484619141e-05,
      "step": 32225,
      "training_step_time": 7.720559597015381
    },
    {
      "epoch": 3.073310852050781e-05,
      "model_forward_time": 2.8702709674835205,
      "step": 32226
    },
    {
      "epoch": 3.073310852050781e-05,
      "step": 32226,
      "training_step_time": 8.645094394683838
    },
    {
      "epoch": 3.0734062194824216e-05,
      "model_forward_time": 1.7236895561218262,
      "step": 32227
    },
    {
      "epoch": 3.0734062194824216e-05,
      "step": 32227,
      "training_step_time": 7.75551176071167
    },
    {
      "epoch": 3.0735015869140626e-05,
      "model_forward_time": 1.7337443828582764,
      "step": 32228
    },
    {
      "epoch": 3.0735015869140626e-05,
      "step": 32228,
      "training_step_time": 7.7932610511779785
    },
    {
      "epoch": 3.073596954345703e-05,
      "model_forward_time": 1.7201454639434814,
      "step": 32229
    },
    {
      "epoch": 3.073596954345703e-05,
      "step": 32229,
      "training_step_time": 7.720860958099365
    },
    {
      "epoch": 3.0736923217773435e-05,
      "grad_norm": 0.11885027587413788,
      "learning_rate": 8.106300388675107e-06,
      "loss": 0.0613,
      "step": 32230
    },
    {
      "epoch": 3.0736923217773435e-05,
      "model_forward_time": 1.730015516281128,
      "step": 32230
    },
    {
      "dynamics_loss_avg": 0.06163672897964716,
      "epoch": 3.0736923217773435e-05,
      "step": 32230
    },
    {
      "action_loss_avg": 0.0019652088143629954,
      "epoch": 3.0736923217773435e-05,
      "step": 32230
    },
    {
      "epoch": 3.0736923217773435e-05,
      "step": 32230,
      "training_step_time": 7.528218507766724
    },
    {
      "epoch": 3.0737876892089845e-05,
      "model_forward_time": 1.7227141857147217,
      "step": 32231
    },
    {
      "epoch": 3.0737876892089845e-05,
      "step": 32231,
      "training_step_time": 7.696876525878906
    },
    {
      "epoch": 3.073883056640625e-05,
      "model_forward_time": 1.7309472560882568,
      "step": 32232
    },
    {
      "epoch": 3.073883056640625e-05,
      "step": 32232,
      "training_step_time": 8.360202312469482
    },
    {
      "epoch": 3.073978424072265e-05,
      "model_forward_time": 1.750908613204956,
      "step": 32233
    },
    {
      "epoch": 3.073978424072265e-05,
      "step": 32233,
      "training_step_time": 8.77060866355896
    },
    {
      "epoch": 3.0740737915039064e-05,
      "model_forward_time": 1.7211532592773438,
      "step": 32234
    },
    {
      "epoch": 3.0740737915039064e-05,
      "step": 32234,
      "training_step_time": 8.270126342773438
    },
    {
      "epoch": 3.074169158935547e-05,
      "model_forward_time": 2.834343671798706,
      "step": 32235
    },
    {
      "epoch": 3.074169158935547e-05,
      "step": 32235,
      "training_step_time": 7.7943434715271
    },
    {
      "epoch": 3.074264526367187e-05,
      "model_forward_time": 1.7246522903442383,
      "step": 32236
    },
    {
      "epoch": 3.074264526367187e-05,
      "step": 32236,
      "training_step_time": 7.748316526412964
    },
    {
      "epoch": 3.074359893798828e-05,
      "model_forward_time": 1.7247378826141357,
      "step": 32237
    },
    {
      "epoch": 3.074359893798828e-05,
      "step": 32237,
      "training_step_time": 7.746808767318726
    },
    {
      "epoch": 3.074455261230469e-05,
      "model_forward_time": 1.7299017906188965,
      "step": 32238
    },
    {
      "epoch": 3.074455261230469e-05,
      "step": 32238,
      "training_step_time": 7.732935667037964
    },
    {
      "epoch": 3.074550628662109e-05,
      "model_forward_time": 1.7582051753997803,
      "step": 32239
    },
    {
      "epoch": 3.074550628662109e-05,
      "step": 32239,
      "training_step_time": 8.621039628982544
    },
    {
      "epoch": 3.07464599609375e-05,
      "grad_norm": 0.13588307797908783,
      "learning_rate": 8.10500455256073e-06,
      "loss": 0.06,
      "step": 32240
    },
    {
      "epoch": 3.07464599609375e-05,
      "model_forward_time": 1.722912311553955,
      "step": 32240
    },
    {
      "dynamics_loss_avg": 0.05997844487428665,
      "epoch": 3.07464599609375e-05,
      "step": 32240
    },
    {
      "action_loss_avg": 0.001970539099420421,
      "epoch": 3.07464599609375e-05,
      "step": 32240
    },
    {
      "epoch": 3.07464599609375e-05,
      "step": 32240,
      "training_step_time": 7.777810573577881
    },
    {
      "epoch": 3.0747413635253906e-05,
      "model_forward_time": 1.725583553314209,
      "step": 32241
    },
    {
      "epoch": 3.0747413635253906e-05,
      "step": 32241,
      "training_step_time": 7.713113069534302
    },
    {
      "epoch": 3.074836730957031e-05,
      "model_forward_time": 1.751469612121582,
      "step": 32242
    },
    {
      "epoch": 3.074836730957031e-05,
      "step": 32242,
      "training_step_time": 7.730134010314941
    },
    {
      "epoch": 3.074932098388672e-05,
      "model_forward_time": 1.7247092723846436,
      "step": 32243
    },
    {
      "epoch": 3.074932098388672e-05,
      "step": 32243,
      "training_step_time": 8.560635089874268
    },
    {
      "epoch": 3.0750274658203125e-05,
      "model_forward_time": 1.7213358879089355,
      "step": 32244
    },
    {
      "epoch": 3.0750274658203125e-05,
      "step": 32244,
      "training_step_time": 8.817831039428711
    },
    {
      "epoch": 3.075122833251953e-05,
      "model_forward_time": 1.7112078666687012,
      "step": 32245
    },
    {
      "epoch": 3.075122833251953e-05,
      "step": 32245,
      "training_step_time": 10.732997179031372
    },
    {
      "epoch": 3.075218200683594e-05,
      "model_forward_time": 1.728032112121582,
      "step": 32246
    },
    {
      "epoch": 3.075218200683594e-05,
      "step": 32246,
      "training_step_time": 7.735458135604858
    },
    {
      "epoch": 3.0753135681152344e-05,
      "model_forward_time": 1.7311267852783203,
      "step": 32247
    },
    {
      "epoch": 3.0753135681152344e-05,
      "step": 32247,
      "training_step_time": 7.7779552936553955
    },
    {
      "epoch": 3.075408935546875e-05,
      "model_forward_time": 1.7294890880584717,
      "step": 32248
    },
    {
      "epoch": 3.075408935546875e-05,
      "step": 32248,
      "training_step_time": 7.902661323547363
    },
    {
      "epoch": 3.075504302978516e-05,
      "model_forward_time": 1.7126491069793701,
      "step": 32249
    },
    {
      "epoch": 3.075504302978516e-05,
      "step": 32249,
      "training_step_time": 7.76540470123291
    },
    {
      "epoch": 3.075599670410156e-05,
      "grad_norm": 0.11356542259454727,
      "learning_rate": 8.103708376887725e-06,
      "loss": 0.0665,
      "step": 32250
    },
    {
      "epoch": 3.075599670410156e-05,
      "model_forward_time": 1.7479233741760254,
      "step": 32250
    },
    {
      "dynamics_loss_avg": 0.048456421494483946,
      "epoch": 3.075599670410156e-05,
      "step": 32250
    },
    {
      "action_loss_avg": 0.0012083050271030516,
      "epoch": 3.075599670410156e-05,
      "step": 32250
    },
    {
      "epoch": 3.075599670410156e-05,
      "step": 32250,
      "training_step_time": 7.711148023605347
    },
    {
      "epoch": 3.0756950378417967e-05,
      "model_forward_time": 1.7683320045471191,
      "step": 32251
    },
    {
      "epoch": 3.0756950378417967e-05,
      "step": 32251,
      "training_step_time": 8.268245935440063
    },
    {
      "epoch": 3.075790405273438e-05,
      "model_forward_time": 1.7365233898162842,
      "step": 32252
    },
    {
      "epoch": 3.075790405273438e-05,
      "step": 32252,
      "training_step_time": 8.390435218811035
    },
    {
      "epoch": 3.075885772705078e-05,
      "model_forward_time": 1.7444171905517578,
      "step": 32253
    },
    {
      "epoch": 3.075885772705078e-05,
      "step": 32253,
      "training_step_time": 8.710883855819702
    },
    {
      "epoch": 3.0759811401367185e-05,
      "model_forward_time": 2.8943333625793457,
      "step": 32254
    },
    {
      "epoch": 3.0759811401367185e-05,
      "step": 32254,
      "training_step_time": 8.347529649734497
    },
    {
      "epoch": 3.0760765075683596e-05,
      "model_forward_time": 1.7424345016479492,
      "step": 32255
    },
    {
      "epoch": 3.0760765075683596e-05,
      "step": 32255,
      "training_step_time": 8.333401918411255
    },
    {
      "epoch": 3.076171875e-05,
      "model_forward_time": 1.7398173809051514,
      "step": 32256
    },
    {
      "epoch": 3.076171875e-05,
      "step": 32256,
      "training_step_time": 7.801105260848999
    },
    {
      "epoch": 3.0762672424316404e-05,
      "model_forward_time": 1.746471881866455,
      "step": 32257
    },
    {
      "epoch": 3.0762672424316404e-05,
      "step": 32257,
      "training_step_time": 7.839270114898682
    },
    {
      "epoch": 3.0763626098632815e-05,
      "model_forward_time": 1.7586159706115723,
      "step": 32258
    },
    {
      "epoch": 3.0763626098632815e-05,
      "step": 32258,
      "training_step_time": 7.795369386672974
    },
    {
      "epoch": 3.076457977294922e-05,
      "model_forward_time": 1.7451865673065186,
      "step": 32259
    },
    {
      "epoch": 3.076457977294922e-05,
      "step": 32259,
      "training_step_time": 7.7854156494140625
    },
    {
      "epoch": 3.076553344726562e-05,
      "grad_norm": 0.13458667695522308,
      "learning_rate": 8.102411861797837e-06,
      "loss": 0.0639,
      "step": 32260
    },
    {
      "epoch": 3.076553344726562e-05,
      "model_forward_time": 1.7591700553894043,
      "step": 32260
    },
    {
      "dynamics_loss_avg": 0.04596123564988375,
      "epoch": 3.076553344726562e-05,
      "step": 32260
    },
    {
      "action_loss_avg": 0.0018410870747175069,
      "epoch": 3.076553344726562e-05,
      "step": 32260
    },
    {
      "epoch": 3.076553344726562e-05,
      "step": 32260,
      "training_step_time": 8.673888921737671
    },
    {
      "epoch": 3.0766487121582034e-05,
      "model_forward_time": 1.7426424026489258,
      "step": 32261
    },
    {
      "epoch": 3.0766487121582034e-05,
      "step": 32261,
      "training_step_time": 7.766161203384399
    },
    {
      "epoch": 3.076744079589844e-05,
      "model_forward_time": 1.7487635612487793,
      "step": 32262
    },
    {
      "epoch": 3.076744079589844e-05,
      "step": 32262,
      "training_step_time": 8.429765939712524
    },
    {
      "epoch": 3.076839447021484e-05,
      "model_forward_time": 2.868394136428833,
      "step": 32263
    },
    {
      "epoch": 3.076839447021484e-05,
      "step": 32263,
      "training_step_time": 7.823570489883423
    },
    {
      "epoch": 3.076934814453125e-05,
      "model_forward_time": 1.7169647216796875,
      "step": 32264
    },
    {
      "epoch": 3.076934814453125e-05,
      "step": 32264,
      "training_step_time": 7.789321422576904
    },
    {
      "epoch": 3.077030181884766e-05,
      "model_forward_time": 1.7313580513000488,
      "step": 32265
    },
    {
      "epoch": 3.077030181884766e-05,
      "step": 32265,
      "training_step_time": 9.322448253631592
    },
    {
      "epoch": 3.077125549316406e-05,
      "model_forward_time": 1.7370507717132568,
      "step": 32266
    },
    {
      "epoch": 3.077125549316406e-05,
      "step": 32266,
      "training_step_time": 7.763673543930054
    },
    {
      "epoch": 3.077220916748047e-05,
      "model_forward_time": 1.7539472579956055,
      "step": 32267
    },
    {
      "epoch": 3.077220916748047e-05,
      "step": 32267,
      "training_step_time": 7.78203272819519
    },
    {
      "epoch": 3.0773162841796876e-05,
      "model_forward_time": 1.735487699508667,
      "step": 32268
    },
    {
      "epoch": 3.0773162841796876e-05,
      "step": 32268,
      "training_step_time": 7.598407030105591
    },
    {
      "epoch": 3.077411651611328e-05,
      "model_forward_time": 1.7362263202667236,
      "step": 32269
    },
    {
      "epoch": 3.077411651611328e-05,
      "step": 32269,
      "training_step_time": 7.647851943969727
    },
    {
      "epoch": 3.077507019042969e-05,
      "grad_norm": 0.12909211218357086,
      "learning_rate": 8.101115007432852e-06,
      "loss": 0.0656,
      "step": 32270
    },
    {
      "epoch": 3.077507019042969e-05,
      "model_forward_time": 1.7700059413909912,
      "step": 32270
    },
    {
      "dynamics_loss_avg": 0.054861736670136454,
      "epoch": 3.077507019042969e-05,
      "step": 32270
    },
    {
      "action_loss_avg": 0.0017395659670000896,
      "epoch": 3.077507019042969e-05,
      "step": 32270
    },
    {
      "epoch": 3.077507019042969e-05,
      "step": 32270,
      "training_step_time": 7.787006616592407
    },
    {
      "epoch": 3.0776023864746095e-05,
      "model_forward_time": 1.7389781475067139,
      "step": 32271
    },
    {
      "epoch": 3.0776023864746095e-05,
      "step": 32271,
      "training_step_time": 8.141682863235474
    },
    {
      "epoch": 3.07769775390625e-05,
      "model_forward_time": 2.8681585788726807,
      "step": 32272
    },
    {
      "epoch": 3.07769775390625e-05,
      "step": 32272,
      "training_step_time": 7.791477918624878
    },
    {
      "epoch": 3.077793121337891e-05,
      "model_forward_time": 1.735659122467041,
      "step": 32273
    },
    {
      "epoch": 3.077793121337891e-05,
      "step": 32273,
      "training_step_time": 7.77317476272583
    },
    {
      "epoch": 3.077888488769531e-05,
      "model_forward_time": 1.7356362342834473,
      "step": 32274
    },
    {
      "epoch": 3.077888488769531e-05,
      "step": 32274,
      "training_step_time": 7.753656625747681
    },
    {
      "epoch": 3.077983856201172e-05,
      "model_forward_time": 1.735388994216919,
      "step": 32275
    },
    {
      "epoch": 3.077983856201172e-05,
      "step": 32275,
      "training_step_time": 8.541424989700317
    },
    {
      "epoch": 3.078079223632813e-05,
      "model_forward_time": 1.7570974826812744,
      "step": 32276
    },
    {
      "epoch": 3.078079223632813e-05,
      "step": 32276,
      "training_step_time": 7.834771394729614
    },
    {
      "epoch": 3.078174591064453e-05,
      "model_forward_time": 1.737865924835205,
      "step": 32277
    },
    {
      "epoch": 3.078174591064453e-05,
      "step": 32277,
      "training_step_time": 7.570128917694092
    },
    {
      "epoch": 3.0782699584960936e-05,
      "model_forward_time": 1.7343511581420898,
      "step": 32278
    },
    {
      "epoch": 3.0782699584960936e-05,
      "step": 32278,
      "training_step_time": 7.703965187072754
    },
    {
      "epoch": 3.078365325927735e-05,
      "model_forward_time": 1.8039064407348633,
      "step": 32279
    },
    {
      "epoch": 3.078365325927735e-05,
      "step": 32279,
      "training_step_time": 8.99212121963501
    },
    {
      "epoch": 3.078460693359375e-05,
      "grad_norm": 0.20922468602657318,
      "learning_rate": 8.09981781393459e-06,
      "loss": 0.066,
      "step": 32280
    },
    {
      "epoch": 3.078460693359375e-05,
      "model_forward_time": 1.7400422096252441,
      "step": 32280
    },
    {
      "dynamics_loss_avg": 0.05578841771930456,
      "epoch": 3.078460693359375e-05,
      "step": 32280
    },
    {
      "action_loss_avg": 0.0018484844506019727,
      "epoch": 3.078460693359375e-05,
      "step": 32280
    },
    {
      "epoch": 3.078460693359375e-05,
      "step": 32280,
      "training_step_time": 7.748266935348511
    },
    {
      "epoch": 3.0785560607910155e-05,
      "model_forward_time": 2.8551394939422607,
      "step": 32281
    },
    {
      "epoch": 3.0785560607910155e-05,
      "step": 32281,
      "training_step_time": 7.799072980880737
    },
    {
      "epoch": 3.078651428222656e-05,
      "model_forward_time": 1.7360820770263672,
      "step": 32282
    },
    {
      "epoch": 3.078651428222656e-05,
      "step": 32282,
      "training_step_time": 8.350155115127563
    },
    {
      "epoch": 3.078746795654297e-05,
      "model_forward_time": 1.734297513961792,
      "step": 32283
    },
    {
      "epoch": 3.078746795654297e-05,
      "step": 32283,
      "training_step_time": 9.604494571685791
    },
    {
      "epoch": 3.0788421630859374e-05,
      "model_forward_time": 1.7347886562347412,
      "step": 32284
    },
    {
      "epoch": 3.0788421630859374e-05,
      "step": 32284,
      "training_step_time": 7.677578449249268
    },
    {
      "epoch": 3.078937530517578e-05,
      "model_forward_time": 1.7622311115264893,
      "step": 32285
    },
    {
      "epoch": 3.078937530517578e-05,
      "step": 32285,
      "training_step_time": 7.898499011993408
    },
    {
      "epoch": 3.079032897949219e-05,
      "model_forward_time": 1.7291293144226074,
      "step": 32286
    },
    {
      "epoch": 3.079032897949219e-05,
      "step": 32286,
      "training_step_time": 7.586301565170288
    },
    {
      "epoch": 3.079128265380859e-05,
      "model_forward_time": 1.7318458557128906,
      "step": 32287
    },
    {
      "epoch": 3.079128265380859e-05,
      "step": 32287,
      "training_step_time": 7.756357431411743
    },
    {
      "epoch": 3.0792236328125e-05,
      "model_forward_time": 1.7616581916809082,
      "step": 32288
    },
    {
      "epoch": 3.0792236328125e-05,
      "step": 32288,
      "training_step_time": 9.046333074569702
    },
    {
      "epoch": 3.079319000244141e-05,
      "model_forward_time": 1.727893352508545,
      "step": 32289
    },
    {
      "epoch": 3.079319000244141e-05,
      "step": 32289,
      "training_step_time": 7.728801727294922
    },
    {
      "epoch": 3.079414367675781e-05,
      "grad_norm": 0.16680629551410675,
      "learning_rate": 8.098520281444915e-06,
      "loss": 0.0664,
      "step": 32290
    },
    {
      "epoch": 3.079414367675781e-05,
      "model_forward_time": 2.799258232116699,
      "step": 32290
    },
    {
      "dynamics_loss_avg": 0.05113007873296738,
      "epoch": 3.079414367675781e-05,
      "step": 32290
    },
    {
      "action_loss_avg": 0.0018849593849154188,
      "epoch": 3.079414367675781e-05,
      "step": 32290
    },
    {
      "epoch": 3.079414367675781e-05,
      "step": 32290,
      "training_step_time": 7.813415050506592
    },
    {
      "epoch": 3.0795097351074216e-05,
      "model_forward_time": 1.725344181060791,
      "step": 32291
    },
    {
      "epoch": 3.0795097351074216e-05,
      "step": 32291,
      "training_step_time": 7.7891106605529785
    },
    {
      "epoch": 3.0796051025390627e-05,
      "model_forward_time": 1.7340741157531738,
      "step": 32292
    },
    {
      "epoch": 3.0796051025390627e-05,
      "step": 32292,
      "training_step_time": 7.807780027389526
    },
    {
      "epoch": 3.079700469970703e-05,
      "model_forward_time": 1.3752851486206055,
      "step": 32293
    },
    {
      "epoch": 3.079700469970703e-05,
      "step": 32293,
      "training_step_time": 7.714560270309448
    },
    {
      "epoch": 3.0797958374023435e-05,
      "model_forward_time": 1.747002124786377,
      "step": 32294
    },
    {
      "epoch": 3.0797958374023435e-05,
      "step": 32294,
      "training_step_time": 8.673413038253784
    },
    {
      "epoch": 3.0798912048339845e-05,
      "model_forward_time": 1.7337868213653564,
      "step": 32295
    },
    {
      "epoch": 3.0798912048339845e-05,
      "step": 32295,
      "training_step_time": 7.6121931076049805
    },
    {
      "epoch": 3.079986572265625e-05,
      "model_forward_time": 1.7246520519256592,
      "step": 32296
    },
    {
      "epoch": 3.079986572265625e-05,
      "step": 32296,
      "training_step_time": 7.696815252304077
    },
    {
      "epoch": 3.0800819396972653e-05,
      "model_forward_time": 1.7405428886413574,
      "step": 32297
    },
    {
      "epoch": 3.0800819396972653e-05,
      "step": 32297,
      "training_step_time": 8.695688247680664
    },
    {
      "epoch": 3.0801773071289064e-05,
      "model_forward_time": 1.3594465255737305,
      "step": 32298
    },
    {
      "epoch": 3.0801773071289064e-05,
      "step": 32298,
      "training_step_time": 8.519580602645874
    },
    {
      "epoch": 3.080272674560547e-05,
      "model_forward_time": 1.7372076511383057,
      "step": 32299
    },
    {
      "epoch": 3.080272674560547e-05,
      "step": 32299,
      "training_step_time": 8.814832210540771
    },
    {
      "epoch": 3.080368041992187e-05,
      "grad_norm": 0.12592646479606628,
      "learning_rate": 8.09722241010572e-06,
      "loss": 0.0631,
      "step": 32300
    },
    {
      "epoch": 3.080368041992187e-05,
      "model_forward_time": 2.9511098861694336,
      "step": 32300
    },
    {
      "dynamics_loss_avg": 0.04909359198063612,
      "epoch": 3.080368041992187e-05,
      "step": 32300
    },
    {
      "action_loss_avg": 0.0015693890338297934,
      "epoch": 3.080368041992187e-05,
      "step": 32300
    },
    {
      "epoch": 3.080368041992187e-05,
      "step": 32300,
      "training_step_time": 7.8502562046051025
    },
    {
      "epoch": 3.080463409423828e-05,
      "model_forward_time": 1.7340779304504395,
      "step": 32301
    },
    {
      "epoch": 3.080463409423828e-05,
      "step": 32301,
      "training_step_time": 8.52629280090332
    },
    {
      "epoch": 3.080558776855469e-05,
      "model_forward_time": 1.7348361015319824,
      "step": 32302
    },
    {
      "epoch": 3.080558776855469e-05,
      "step": 32302,
      "training_step_time": 8.828750848770142
    },
    {
      "epoch": 3.080654144287109e-05,
      "model_forward_time": 1.732473611831665,
      "step": 32303
    },
    {
      "epoch": 3.080654144287109e-05,
      "step": 32303,
      "training_step_time": 7.806412696838379
    },
    {
      "epoch": 3.08074951171875e-05,
      "model_forward_time": 1.7335057258605957,
      "step": 32304
    },
    {
      "epoch": 3.08074951171875e-05,
      "step": 32304,
      "training_step_time": 7.619859218597412
    },
    {
      "epoch": 3.0808448791503906e-05,
      "model_forward_time": 1.7289705276489258,
      "step": 32305
    },
    {
      "epoch": 3.0808448791503906e-05,
      "step": 32305,
      "training_step_time": 7.615805387496948
    },
    {
      "epoch": 3.080940246582031e-05,
      "model_forward_time": 1.7355337142944336,
      "step": 32306
    },
    {
      "epoch": 3.080940246582031e-05,
      "step": 32306,
      "training_step_time": 7.744591236114502
    },
    {
      "epoch": 3.081035614013672e-05,
      "model_forward_time": 1.733374834060669,
      "step": 32307
    },
    {
      "epoch": 3.081035614013672e-05,
      "step": 32307,
      "training_step_time": 8.780386209487915
    },
    {
      "epoch": 3.0811309814453125e-05,
      "model_forward_time": 1.7274293899536133,
      "step": 32308
    },
    {
      "epoch": 3.0811309814453125e-05,
      "step": 32308,
      "training_step_time": 7.7390525341033936
    },
    {
      "epoch": 3.081226348876953e-05,
      "model_forward_time": 2.8224072456359863,
      "step": 32309
    },
    {
      "epoch": 3.081226348876953e-05,
      "step": 32309,
      "training_step_time": 9.686946630477905
    },
    {
      "epoch": 3.081321716308594e-05,
      "grad_norm": 0.1655907928943634,
      "learning_rate": 8.095924200058939e-06,
      "loss": 0.0662,
      "step": 32310
    },
    {
      "epoch": 3.081321716308594e-05,
      "model_forward_time": 1.7248663902282715,
      "step": 32310
    },
    {
      "dynamics_loss_avg": 0.06177432271651924,
      "epoch": 3.081321716308594e-05,
      "step": 32310
    },
    {
      "action_loss_avg": 0.003087868867442012,
      "epoch": 3.081321716308594e-05,
      "step": 32310
    },
    {
      "epoch": 3.081321716308594e-05,
      "step": 32310,
      "training_step_time": 9.158885717391968
    },
    {
      "epoch": 3.0814170837402344e-05,
      "model_forward_time": 1.7267320156097412,
      "step": 32311
    },
    {
      "epoch": 3.0814170837402344e-05,
      "step": 32311,
      "training_step_time": 7.69957709312439
    },
    {
      "epoch": 3.081512451171875e-05,
      "model_forward_time": 1.7317783832550049,
      "step": 32312
    },
    {
      "epoch": 3.081512451171875e-05,
      "step": 32312,
      "training_step_time": 7.709005355834961
    },
    {
      "epoch": 3.081607818603516e-05,
      "model_forward_time": 1.7510480880737305,
      "step": 32313
    },
    {
      "epoch": 3.081607818603516e-05,
      "step": 32313,
      "training_step_time": 7.585104942321777
    },
    {
      "epoch": 3.081703186035156e-05,
      "model_forward_time": 1.7354156970977783,
      "step": 32314
    },
    {
      "epoch": 3.081703186035156e-05,
      "step": 32314,
      "training_step_time": 8.013070583343506
    },
    {
      "epoch": 3.0817985534667967e-05,
      "model_forward_time": 1.3699495792388916,
      "step": 32315
    },
    {
      "epoch": 3.0817985534667967e-05,
      "step": 32315,
      "training_step_time": 7.720782518386841
    },
    {
      "epoch": 3.081893920898438e-05,
      "model_forward_time": 1.7518184185028076,
      "step": 32316
    },
    {
      "epoch": 3.081893920898438e-05,
      "step": 32316,
      "training_step_time": 7.797859191894531
    },
    {
      "epoch": 3.081989288330078e-05,
      "model_forward_time": 1.7304778099060059,
      "step": 32317
    },
    {
      "epoch": 3.081989288330078e-05,
      "step": 32317,
      "training_step_time": 9.051361083984375
    },
    {
      "epoch": 3.0820846557617185e-05,
      "model_forward_time": 2.8117270469665527,
      "step": 32318
    },
    {
      "epoch": 3.0820846557617185e-05,
      "step": 32318,
      "training_step_time": 7.809125661849976
    },
    {
      "epoch": 3.0821800231933596e-05,
      "model_forward_time": 1.7311432361602783,
      "step": 32319
    },
    {
      "epoch": 3.0821800231933596e-05,
      "step": 32319,
      "training_step_time": 9.306504487991333
    },
    {
      "epoch": 3.082275390625e-05,
      "grad_norm": 0.1905578225851059,
      "learning_rate": 8.094625651446542e-06,
      "loss": 0.063,
      "step": 32320
    },
    {
      "epoch": 3.082275390625e-05,
      "model_forward_time": 1.7301580905914307,
      "step": 32320
    },
    {
      "dynamics_loss_avg": 0.04691256228834391,
      "epoch": 3.082275390625e-05,
      "step": 32320
    },
    {
      "action_loss_avg": 0.002295343769947067,
      "epoch": 3.082275390625e-05,
      "step": 32320
    },
    {
      "epoch": 3.082275390625e-05,
      "step": 32320,
      "training_step_time": 7.803570032119751
    },
    {
      "epoch": 3.0823707580566404e-05,
      "model_forward_time": 1.73079514503479,
      "step": 32321
    },
    {
      "epoch": 3.0823707580566404e-05,
      "step": 32321,
      "training_step_time": 7.730669021606445
    },
    {
      "epoch": 3.0824661254882815e-05,
      "model_forward_time": 1.7630009651184082,
      "step": 32322
    },
    {
      "epoch": 3.0824661254882815e-05,
      "step": 32322,
      "training_step_time": 7.794611930847168
    },
    {
      "epoch": 3.082561492919922e-05,
      "model_forward_time": 1.7344334125518799,
      "step": 32323
    },
    {
      "epoch": 3.082561492919922e-05,
      "step": 32323,
      "training_step_time": 7.623408555984497
    },
    {
      "epoch": 3.082656860351562e-05,
      "model_forward_time": 1.7364563941955566,
      "step": 32324
    },
    {
      "epoch": 3.082656860351562e-05,
      "step": 32324,
      "training_step_time": 7.748204708099365
    },
    {
      "epoch": 3.0827522277832034e-05,
      "model_forward_time": 1.7527062892913818,
      "step": 32325
    },
    {
      "epoch": 3.0827522277832034e-05,
      "step": 32325,
      "training_step_time": 7.93343186378479
    },
    {
      "epoch": 3.082847595214844e-05,
      "model_forward_time": 1.7381651401519775,
      "step": 32326
    },
    {
      "epoch": 3.082847595214844e-05,
      "step": 32326,
      "training_step_time": 9.083169937133789
    },
    {
      "epoch": 3.082942962646484e-05,
      "model_forward_time": 1.7377820014953613,
      "step": 32327
    },
    {
      "epoch": 3.082942962646484e-05,
      "step": 32327,
      "training_step_time": 8.46105694770813
    },
    {
      "epoch": 3.083038330078125e-05,
      "model_forward_time": 2.882497549057007,
      "step": 32328
    },
    {
      "epoch": 3.083038330078125e-05,
      "step": 32328,
      "training_step_time": 7.81693172454834
    },
    {
      "epoch": 3.083133697509766e-05,
      "model_forward_time": 1.7322421073913574,
      "step": 32329
    },
    {
      "epoch": 3.083133697509766e-05,
      "step": 32329,
      "training_step_time": 7.8462607860565186
    },
    {
      "epoch": 3.083229064941406e-05,
      "grad_norm": 0.11997271329164505,
      "learning_rate": 8.093326764410537e-06,
      "loss": 0.0646,
      "step": 32330
    },
    {
      "epoch": 3.083229064941406e-05,
      "model_forward_time": 1.7338314056396484,
      "step": 32330
    },
    {
      "dynamics_loss_avg": 0.05593376476317644,
      "epoch": 3.083229064941406e-05,
      "step": 32330
    },
    {
      "action_loss_avg": 0.0021148985484614967,
      "epoch": 3.083229064941406e-05,
      "step": 32330
    },
    {
      "epoch": 3.083229064941406e-05,
      "step": 32330,
      "training_step_time": 7.740496397018433
    },
    {
      "epoch": 3.083324432373047e-05,
      "model_forward_time": 1.7480835914611816,
      "step": 32331
    },
    {
      "epoch": 3.083324432373047e-05,
      "step": 32331,
      "training_step_time": 7.90726375579834
    },
    {
      "epoch": 3.0834197998046876e-05,
      "model_forward_time": 1.7314600944519043,
      "step": 32332
    },
    {
      "epoch": 3.0834197998046876e-05,
      "step": 32332,
      "training_step_time": 7.593822002410889
    },
    {
      "epoch": 3.083515167236328e-05,
      "model_forward_time": 1.7362611293792725,
      "step": 32333
    },
    {
      "epoch": 3.083515167236328e-05,
      "step": 32333,
      "training_step_time": 7.651835918426514
    },
    {
      "epoch": 3.083610534667969e-05,
      "model_forward_time": 1.7852113246917725,
      "step": 32334
    },
    {
      "epoch": 3.083610534667969e-05,
      "step": 32334,
      "training_step_time": 8.64133882522583
    },
    {
      "epoch": 3.0837059020996095e-05,
      "model_forward_time": 1.7403590679168701,
      "step": 32335
    },
    {
      "epoch": 3.0837059020996095e-05,
      "step": 32335,
      "training_step_time": 7.591148376464844
    },
    {
      "epoch": 3.08380126953125e-05,
      "model_forward_time": 1.7294983863830566,
      "step": 32336
    },
    {
      "epoch": 3.08380126953125e-05,
      "step": 32336,
      "training_step_time": 8.867241382598877
    },
    {
      "epoch": 3.083896636962891e-05,
      "model_forward_time": 2.8673760890960693,
      "step": 32337
    },
    {
      "epoch": 3.083896636962891e-05,
      "step": 32337,
      "training_step_time": 8.520105600357056
    },
    {
      "epoch": 3.0839920043945313e-05,
      "model_forward_time": 1.7360341548919678,
      "step": 32338
    },
    {
      "epoch": 3.0839920043945313e-05,
      "step": 32338,
      "training_step_time": 8.272885084152222
    },
    {
      "epoch": 3.084087371826172e-05,
      "model_forward_time": 1.7339613437652588,
      "step": 32339
    },
    {
      "epoch": 3.084087371826172e-05,
      "step": 32339,
      "training_step_time": 8.77267837524414
    },
    {
      "epoch": 3.084182739257813e-05,
      "grad_norm": 0.16776928305625916,
      "learning_rate": 8.092027539092966e-06,
      "loss": 0.0619,
      "step": 32340
    },
    {
      "epoch": 3.084182739257813e-05,
      "model_forward_time": 1.7424867153167725,
      "step": 32340
    },
    {
      "dynamics_loss_avg": 0.04320004470646381,
      "epoch": 3.084182739257813e-05,
      "step": 32340
    },
    {
      "action_loss_avg": 0.0017038430203683674,
      "epoch": 3.084182739257813e-05,
      "step": 32340
    },
    {
      "epoch": 3.084182739257813e-05,
      "step": 32340,
      "training_step_time": 7.78168797492981
    },
    {
      "epoch": 3.084278106689453e-05,
      "model_forward_time": 1.7488160133361816,
      "step": 32341
    },
    {
      "epoch": 3.084278106689453e-05,
      "step": 32341,
      "training_step_time": 7.818131685256958
    },
    {
      "epoch": 3.0843734741210936e-05,
      "model_forward_time": 1.7364325523376465,
      "step": 32342
    },
    {
      "epoch": 3.0843734741210936e-05,
      "step": 32342,
      "training_step_time": 7.6165173053741455
    },
    {
      "epoch": 3.084468841552735e-05,
      "model_forward_time": 1.7413785457611084,
      "step": 32343
    },
    {
      "epoch": 3.084468841552735e-05,
      "step": 32343,
      "training_step_time": 9.27649736404419
    },
    {
      "epoch": 3.084564208984375e-05,
      "model_forward_time": 1.7293527126312256,
      "step": 32344
    },
    {
      "epoch": 3.084564208984375e-05,
      "step": 32344,
      "training_step_time": 9.444702625274658
    },
    {
      "epoch": 3.0846595764160155e-05,
      "model_forward_time": 1.7349815368652344,
      "step": 32345
    },
    {
      "epoch": 3.0846595764160155e-05,
      "step": 32345,
      "training_step_time": 8.418584823608398
    },
    {
      "epoch": 3.084754943847656e-05,
      "model_forward_time": 1.7391200065612793,
      "step": 32346
    },
    {
      "epoch": 3.084754943847656e-05,
      "step": 32346,
      "training_step_time": 8.716943264007568
    },
    {
      "epoch": 3.084850311279297e-05,
      "model_forward_time": 1.7391176223754883,
      "step": 32347
    },
    {
      "epoch": 3.084850311279297e-05,
      "step": 32347,
      "training_step_time": 6.915606498718262
    },
    {
      "epoch": 3.0849456787109374e-05,
      "model_forward_time": 1.7323603630065918,
      "step": 32348
    },
    {
      "epoch": 3.0849456787109374e-05,
      "step": 32348,
      "training_step_time": 6.901147127151489
    },
    {
      "epoch": 3.085041046142578e-05,
      "model_forward_time": 1.72945237159729,
      "step": 32349
    },
    {
      "epoch": 3.085041046142578e-05,
      "step": 32349,
      "training_step_time": 6.822779417037964
    },
    {
      "epoch": 3.085136413574219e-05,
      "grad_norm": 0.19824068248271942,
      "learning_rate": 8.090727975635914e-06,
      "loss": 0.0628,
      "step": 32350
    },
    {
      "epoch": 3.085136413574219e-05,
      "model_forward_time": 1.738816261291504,
      "step": 32350
    },
    {
      "dynamics_loss_avg": 0.04640283584594727,
      "epoch": 3.085136413574219e-05,
      "step": 32350
    },
    {
      "action_loss_avg": 0.001856818044325337,
      "epoch": 3.085136413574219e-05,
      "step": 32350
    },
    {
      "epoch": 3.085136413574219e-05,
      "step": 32350,
      "training_step_time": 6.891682147979736
    },
    {
      "epoch": 3.085231781005859e-05,
      "model_forward_time": 1.729445219039917,
      "step": 32351
    },
    {
      "epoch": 3.085231781005859e-05,
      "step": 32351,
      "training_step_time": 6.907432556152344
    },
    {
      "epoch": 3.0853271484375e-05,
      "model_forward_time": 1.7345154285430908,
      "step": 32352
    },
    {
      "epoch": 3.0853271484375e-05,
      "step": 32352,
      "training_step_time": 6.784512042999268
    },
    {
      "epoch": 3.085422515869141e-05,
      "model_forward_time": 1.734264612197876,
      "step": 32353
    },
    {
      "epoch": 3.085422515869141e-05,
      "step": 32353,
      "training_step_time": 6.74522590637207
    },
    {
      "epoch": 3.085517883300781e-05,
      "model_forward_time": 1.7098522186279297,
      "step": 32354
    },
    {
      "epoch": 3.085517883300781e-05,
      "step": 32354,
      "training_step_time": 6.829240798950195
    },
    {
      "epoch": 3.0856132507324216e-05,
      "model_forward_time": 2.8615410327911377,
      "step": 32355
    },
    {
      "epoch": 3.0856132507324216e-05,
      "step": 32355,
      "training_step_time": 11.104299783706665
    },
    {
      "epoch": 3.0857086181640627e-05,
      "model_forward_time": 1.7373802661895752,
      "step": 32356
    },
    {
      "epoch": 3.0857086181640627e-05,
      "step": 32356,
      "training_step_time": 7.715574026107788
    },
    {
      "epoch": 3.085803985595703e-05,
      "model_forward_time": 1.7357492446899414,
      "step": 32357
    },
    {
      "epoch": 3.085803985595703e-05,
      "step": 32357,
      "training_step_time": 6.8829665184021
    },
    {
      "epoch": 3.0858993530273435e-05,
      "model_forward_time": 1.7343809604644775,
      "step": 32358
    },
    {
      "epoch": 3.0858993530273435e-05,
      "step": 32358,
      "training_step_time": 6.8443028926849365
    },
    {
      "epoch": 3.0859947204589845e-05,
      "model_forward_time": 1.732182502746582,
      "step": 32359
    },
    {
      "epoch": 3.0859947204589845e-05,
      "step": 32359,
      "training_step_time": 6.716762065887451
    },
    {
      "epoch": 3.086090087890625e-05,
      "grad_norm": 0.1682397425174713,
      "learning_rate": 8.089428074181496e-06,
      "loss": 0.0611,
      "step": 32360
    },
    {
      "epoch": 3.086090087890625e-05,
      "model_forward_time": 1.7257895469665527,
      "step": 32360
    },
    {
      "dynamics_loss_avg": 0.05365704037249088,
      "epoch": 3.086090087890625e-05,
      "step": 32360
    },
    {
      "action_loss_avg": 0.0024126422475092115,
      "epoch": 3.086090087890625e-05,
      "step": 32360
    },
    {
      "epoch": 3.086090087890625e-05,
      "step": 32360,
      "training_step_time": 6.866987943649292
    },
    {
      "epoch": 3.0861854553222653e-05,
      "model_forward_time": 1.728693962097168,
      "step": 32361
    },
    {
      "epoch": 3.0861854553222653e-05,
      "step": 32361,
      "training_step_time": 6.8453919887542725
    },
    {
      "epoch": 3.0862808227539064e-05,
      "model_forward_time": 1.7159900665283203,
      "step": 32362
    },
    {
      "epoch": 3.0862808227539064e-05,
      "step": 32362,
      "training_step_time": 6.81827449798584
    },
    {
      "epoch": 3.086376190185547e-05,
      "model_forward_time": 1.7305786609649658,
      "step": 32363
    },
    {
      "epoch": 3.086376190185547e-05,
      "step": 32363,
      "training_step_time": 6.841261386871338
    },
    {
      "epoch": 3.086471557617187e-05,
      "model_forward_time": 2.8449113368988037,
      "step": 32364
    },
    {
      "epoch": 3.086471557617187e-05,
      "step": 32364,
      "training_step_time": 7.744920015335083
    },
    {
      "epoch": 3.086566925048828e-05,
      "model_forward_time": 1.7289068698883057,
      "step": 32365
    },
    {
      "epoch": 3.086566925048828e-05,
      "step": 32365,
      "training_step_time": 8.778548002243042
    },
    {
      "epoch": 3.086662292480469e-05,
      "model_forward_time": 1.7361910343170166,
      "step": 32366
    },
    {
      "epoch": 3.086662292480469e-05,
      "step": 32366,
      "training_step_time": 7.138949632644653
    },
    {
      "epoch": 3.086757659912109e-05,
      "model_forward_time": 1.7312543392181396,
      "step": 32367
    },
    {
      "epoch": 3.086757659912109e-05,
      "step": 32367,
      "training_step_time": 6.830467224121094
    },
    {
      "epoch": 3.08685302734375e-05,
      "model_forward_time": 1.7334179878234863,
      "step": 32368
    },
    {
      "epoch": 3.08685302734375e-05,
      "step": 32368,
      "training_step_time": 6.860650539398193
    },
    {
      "epoch": 3.0869483947753906e-05,
      "model_forward_time": 1.7191383838653564,
      "step": 32369
    },
    {
      "epoch": 3.0869483947753906e-05,
      "step": 32369,
      "training_step_time": 6.835096597671509
    },
    {
      "epoch": 3.087043762207031e-05,
      "grad_norm": 0.17499402165412903,
      "learning_rate": 8.08812783487187e-06,
      "loss": 0.0619,
      "step": 32370
    },
    {
      "epoch": 3.087043762207031e-05,
      "model_forward_time": 1.7334821224212646,
      "step": 32370
    },
    {
      "dynamics_loss_avg": 0.06305285142734647,
      "epoch": 3.087043762207031e-05,
      "step": 32370
    },
    {
      "action_loss_avg": 0.002214609694783576,
      "epoch": 3.087043762207031e-05,
      "step": 32370
    },
    {
      "epoch": 3.087043762207031e-05,
      "step": 32370,
      "training_step_time": 6.829395055770874
    },
    {
      "epoch": 3.087139129638672e-05,
      "model_forward_time": 1.366170883178711,
      "step": 32371
    },
    {
      "epoch": 3.087139129638672e-05,
      "step": 32371,
      "training_step_time": 6.88148045539856
    },
    {
      "epoch": 3.0872344970703125e-05,
      "model_forward_time": 1.7305192947387695,
      "step": 32372
    },
    {
      "epoch": 3.0872344970703125e-05,
      "step": 32372,
      "training_step_time": 6.84013819694519
    },
    {
      "epoch": 3.087329864501953e-05,
      "model_forward_time": 2.8546178340911865,
      "step": 32373
    },
    {
      "epoch": 3.087329864501953e-05,
      "step": 32373,
      "training_step_time": 7.771849632263184
    },
    {
      "epoch": 3.087425231933594e-05,
      "model_forward_time": 1.7388267517089844,
      "step": 32374
    },
    {
      "epoch": 3.087425231933594e-05,
      "step": 32374,
      "training_step_time": 9.13998007774353
    },
    {
      "epoch": 3.0875205993652344e-05,
      "model_forward_time": 1.7326767444610596,
      "step": 32375
    },
    {
      "epoch": 3.0875205993652344e-05,
      "step": 32375,
      "training_step_time": 7.707477569580078
    },
    {
      "epoch": 3.087615966796875e-05,
      "model_forward_time": 1.7396106719970703,
      "step": 32376
    },
    {
      "epoch": 3.087615966796875e-05,
      "step": 32376,
      "training_step_time": 6.856504678726196
    },
    {
      "epoch": 3.087711334228516e-05,
      "model_forward_time": 1.7381622791290283,
      "step": 32377
    },
    {
      "epoch": 3.087711334228516e-05,
      "step": 32377,
      "training_step_time": 6.825394153594971
    },
    {
      "epoch": 3.087806701660156e-05,
      "model_forward_time": 1.7404298782348633,
      "step": 32378
    },
    {
      "epoch": 3.087806701660156e-05,
      "step": 32378,
      "training_step_time": 6.919420003890991
    },
    {
      "epoch": 3.087902069091797e-05,
      "model_forward_time": 1.7229223251342773,
      "step": 32379
    },
    {
      "epoch": 3.087902069091797e-05,
      "step": 32379,
      "training_step_time": 6.8741466999053955
    },
    {
      "epoch": 3.087997436523438e-05,
      "grad_norm": 0.151412233710289,
      "learning_rate": 8.086827257849225e-06,
      "loss": 0.0581,
      "step": 32380
    },
    {
      "epoch": 3.087997436523438e-05,
      "model_forward_time": 1.7349698543548584,
      "step": 32380
    },
    {
      "dynamics_loss_avg": 0.06319599263370038,
      "epoch": 3.087997436523438e-05,
      "step": 32380
    },
    {
      "action_loss_avg": 0.0022609510459005834,
      "epoch": 3.087997436523438e-05,
      "step": 32380
    },
    {
      "epoch": 3.087997436523438e-05,
      "step": 32380,
      "training_step_time": 6.861843824386597
    },
    {
      "epoch": 3.088092803955078e-05,
      "model_forward_time": 1.7339098453521729,
      "step": 32381
    },
    {
      "epoch": 3.088092803955078e-05,
      "step": 32381,
      "training_step_time": 6.94199013710022
    },
    {
      "epoch": 3.0881881713867185e-05,
      "model_forward_time": 2.8322949409484863,
      "step": 32382
    },
    {
      "epoch": 3.0881881713867185e-05,
      "step": 32382,
      "training_step_time": 8.72899603843689
    },
    {
      "epoch": 3.0882835388183596e-05,
      "model_forward_time": 1.750101089477539,
      "step": 32383
    },
    {
      "epoch": 3.0882835388183596e-05,
      "step": 32383,
      "training_step_time": 9.409937381744385
    },
    {
      "epoch": 3.08837890625e-05,
      "model_forward_time": 1.7326359748840332,
      "step": 32384
    },
    {
      "epoch": 3.08837890625e-05,
      "step": 32384,
      "training_step_time": 7.64963436126709
    },
    {
      "epoch": 3.0884742736816404e-05,
      "model_forward_time": 1.7284395694732666,
      "step": 32385
    },
    {
      "epoch": 3.0884742736816404e-05,
      "step": 32385,
      "training_step_time": 6.77439284324646
    },
    {
      "epoch": 3.0885696411132815e-05,
      "model_forward_time": 1.735541820526123,
      "step": 32386
    },
    {
      "epoch": 3.0885696411132815e-05,
      "step": 32386,
      "training_step_time": 6.799027442932129
    },
    {
      "epoch": 3.088665008544922e-05,
      "model_forward_time": 1.7319445610046387,
      "step": 32387
    },
    {
      "epoch": 3.088665008544922e-05,
      "step": 32387,
      "training_step_time": 6.734770774841309
    },
    {
      "epoch": 3.088760375976562e-05,
      "model_forward_time": 1.7244110107421875,
      "step": 32388
    },
    {
      "epoch": 3.088760375976562e-05,
      "step": 32388,
      "training_step_time": 6.7879791259765625
    },
    {
      "epoch": 3.0888557434082034e-05,
      "model_forward_time": 1.73822021484375,
      "step": 32389
    },
    {
      "epoch": 3.0888557434082034e-05,
      "step": 32389,
      "training_step_time": 6.810168504714966
    },
    {
      "epoch": 3.088951110839844e-05,
      "grad_norm": 0.12701909244060516,
      "learning_rate": 8.085526343255796e-06,
      "loss": 0.0606,
      "step": 32390
    },
    {
      "epoch": 3.088951110839844e-05,
      "model_forward_time": 1.7329363822937012,
      "step": 32390
    },
    {
      "dynamics_loss_avg": 0.04359323726966977,
      "epoch": 3.088951110839844e-05,
      "step": 32390
    },
    {
      "action_loss_avg": 0.002130042077624239,
      "epoch": 3.088951110839844e-05,
      "step": 32390
    },
    {
      "epoch": 3.088951110839844e-05,
      "step": 32390,
      "training_step_time": 6.80199933052063
    },
    {
      "epoch": 3.089046478271484e-05,
      "model_forward_time": 1.7318909168243408,
      "step": 32391
    },
    {
      "epoch": 3.089046478271484e-05,
      "step": 32391,
      "training_step_time": 8.522111892700195
    },
    {
      "epoch": 3.089141845703125e-05,
      "model_forward_time": 2.8433609008789062,
      "step": 32392
    },
    {
      "epoch": 3.089141845703125e-05,
      "step": 32392,
      "training_step_time": 9.50032663345337
    },
    {
      "epoch": 3.089237213134766e-05,
      "model_forward_time": 1.7180516719818115,
      "step": 32393
    },
    {
      "epoch": 3.089237213134766e-05,
      "step": 32393,
      "training_step_time": 7.6906983852386475
    },
    {
      "epoch": 3.089332580566406e-05,
      "model_forward_time": 1.7287497520446777,
      "step": 32394
    },
    {
      "epoch": 3.089332580566406e-05,
      "step": 32394,
      "training_step_time": 7.3897247314453125
    },
    {
      "epoch": 3.089427947998047e-05,
      "model_forward_time": 1.748084306716919,
      "step": 32395
    },
    {
      "epoch": 3.089427947998047e-05,
      "step": 32395,
      "training_step_time": 6.79081654548645
    },
    {
      "epoch": 3.0895233154296876e-05,
      "model_forward_time": 1.7731478214263916,
      "step": 32396
    },
    {
      "epoch": 3.0895233154296876e-05,
      "step": 32396,
      "training_step_time": 6.819368362426758
    },
    {
      "epoch": 3.089618682861328e-05,
      "model_forward_time": 1.7119204998016357,
      "step": 32397
    },
    {
      "epoch": 3.089618682861328e-05,
      "step": 32397,
      "training_step_time": 6.8423988819122314
    },
    {
      "epoch": 3.089714050292969e-05,
      "model_forward_time": 1.729064702987671,
      "step": 32398
    },
    {
      "epoch": 3.089714050292969e-05,
      "step": 32398,
      "training_step_time": 6.855545520782471
    },
    {
      "epoch": 3.0898094177246095e-05,
      "model_forward_time": 1.719655990600586,
      "step": 32399
    },
    {
      "epoch": 3.0898094177246095e-05,
      "step": 32399,
      "training_step_time": 6.810809373855591
    },
    {
      "epoch": 3.08990478515625e-05,
      "grad_norm": 0.16729728877544403,
      "learning_rate": 8.084225091233842e-06,
      "loss": 0.0625,
      "step": 32400
    },
    {
      "epoch": 3.08990478515625e-05,
      "model_forward_time": 1.724663257598877,
      "step": 32400
    },
    {
      "dynamics_loss_avg": 0.04258708525449038,
      "epoch": 3.08990478515625e-05,
      "step": 32400
    },
    {
      "action_loss_avg": 0.0017855712445452808,
      "epoch": 3.08990478515625e-05,
      "step": 32400
    },
    {
      "epoch": 3.08990478515625e-05,
      "step": 32400,
      "training_step_time": 6.841817855834961
    },
    {
      "epoch": 3.090000152587891e-05,
      "model_forward_time": 2.2688045501708984,
      "step": 32401
    },
    {
      "epoch": 3.090000152587891e-05,
      "step": 32401,
      "training_step_time": 10.690151929855347
    },
    {
      "epoch": 3.0900955200195313e-05,
      "model_forward_time": 1.7349565029144287,
      "step": 32402
    },
    {
      "epoch": 3.0900955200195313e-05,
      "step": 32402,
      "training_step_time": 7.731945514678955
    },
    {
      "epoch": 3.090190887451172e-05,
      "model_forward_time": 1.7282097339630127,
      "step": 32403
    },
    {
      "epoch": 3.090190887451172e-05,
      "step": 32403,
      "training_step_time": 7.599242210388184
    },
    {
      "epoch": 3.090286254882813e-05,
      "model_forward_time": 1.7411863803863525,
      "step": 32404
    },
    {
      "epoch": 3.090286254882813e-05,
      "step": 32404,
      "training_step_time": 6.857452154159546
    },
    {
      "epoch": 3.090381622314453e-05,
      "model_forward_time": 1.7444465160369873,
      "step": 32405
    },
    {
      "epoch": 3.090381622314453e-05,
      "step": 32405,
      "training_step_time": 6.861011028289795
    },
    {
      "epoch": 3.0904769897460936e-05,
      "model_forward_time": 1.7534794807434082,
      "step": 32406
    },
    {
      "epoch": 3.0904769897460936e-05,
      "step": 32406,
      "training_step_time": 6.705871343612671
    },
    {
      "epoch": 3.090572357177734e-05,
      "model_forward_time": 1.7437098026275635,
      "step": 32407
    },
    {
      "epoch": 3.090572357177734e-05,
      "step": 32407,
      "training_step_time": 6.789696216583252
    },
    {
      "epoch": 3.090667724609375e-05,
      "model_forward_time": 1.7354233264923096,
      "step": 32408
    },
    {
      "epoch": 3.090667724609375e-05,
      "step": 32408,
      "training_step_time": 6.740984678268433
    },
    {
      "epoch": 3.0907630920410155e-05,
      "model_forward_time": 1.7150318622589111,
      "step": 32409
    },
    {
      "epoch": 3.0907630920410155e-05,
      "step": 32409,
      "training_step_time": 6.811940908432007
    },
    {
      "epoch": 3.090858459472656e-05,
      "grad_norm": 0.14443500339984894,
      "learning_rate": 8.082923501925668e-06,
      "loss": 0.0594,
      "step": 32410
    },
    {
      "epoch": 3.090858459472656e-05,
      "model_forward_time": 2.2744553089141846,
      "step": 32410
    },
    {
      "dynamics_loss_avg": 0.04777948744595051,
      "epoch": 3.090858459472656e-05,
      "step": 32410
    },
    {
      "action_loss_avg": 0.001728105713846162,
      "epoch": 3.090858459472656e-05,
      "step": 32410
    },
    {
      "epoch": 3.090858459472656e-05,
      "step": 32410,
      "training_step_time": 9.663071870803833
    },
    {
      "epoch": 3.090953826904297e-05,
      "model_forward_time": 1.730912685394287,
      "step": 32411
    },
    {
      "epoch": 3.090953826904297e-05,
      "step": 32411,
      "training_step_time": 8.44139313697815
    },
    {
      "epoch": 3.0910491943359374e-05,
      "model_forward_time": 1.7443289756774902,
      "step": 32412
    },
    {
      "epoch": 3.0910491943359374e-05,
      "step": 32412,
      "training_step_time": 7.530189752578735
    },
    {
      "epoch": 3.091144561767578e-05,
      "model_forward_time": 1.748854160308838,
      "step": 32413
    },
    {
      "epoch": 3.091144561767578e-05,
      "step": 32413,
      "training_step_time": 6.755829334259033
    },
    {
      "epoch": 3.091239929199219e-05,
      "model_forward_time": 1.713709831237793,
      "step": 32414
    },
    {
      "epoch": 3.091239929199219e-05,
      "step": 32414,
      "training_step_time": 6.866774559020996
    },
    {
      "epoch": 3.091335296630859e-05,
      "model_forward_time": 1.7240722179412842,
      "step": 32415
    },
    {
      "epoch": 3.091335296630859e-05,
      "step": 32415,
      "training_step_time": 6.8354411125183105
    },
    {
      "epoch": 3.0914306640625e-05,
      "model_forward_time": 1.7288646697998047,
      "step": 32416
    },
    {
      "epoch": 3.0914306640625e-05,
      "step": 32416,
      "training_step_time": 6.8243396282196045
    },
    {
      "epoch": 3.091526031494141e-05,
      "model_forward_time": 1.7242164611816406,
      "step": 32417
    },
    {
      "epoch": 3.091526031494141e-05,
      "step": 32417,
      "training_step_time": 6.864592552185059
    },
    {
      "epoch": 3.091621398925781e-05,
      "model_forward_time": 1.7365798950195312,
      "step": 32418
    },
    {
      "epoch": 3.091621398925781e-05,
      "step": 32418,
      "training_step_time": 6.8221330642700195
    },
    {
      "epoch": 3.0917167663574216e-05,
      "model_forward_time": 2.2556185722351074,
      "step": 32419
    },
    {
      "epoch": 3.0917167663574216e-05,
      "step": 32419,
      "training_step_time": 9.756199598312378
    },
    {
      "epoch": 3.091812133789063e-05,
      "grad_norm": 0.1878206878900528,
      "learning_rate": 8.081621575473617e-06,
      "loss": 0.0583,
      "step": 32420
    },
    {
      "epoch": 3.091812133789063e-05,
      "model_forward_time": 1.766688346862793,
      "step": 32420
    },
    {
      "dynamics_loss_avg": 0.053988619893789294,
      "epoch": 3.091812133789063e-05,
      "step": 32420
    },
    {
      "action_loss_avg": 0.002703273046063259,
      "epoch": 3.091812133789063e-05,
      "step": 32420
    },
    {
      "epoch": 3.091812133789063e-05,
      "step": 32420,
      "training_step_time": 8.782981872558594
    },
    {
      "epoch": 3.091907501220703e-05,
      "model_forward_time": 1.7469627857208252,
      "step": 32421
    },
    {
      "epoch": 3.091907501220703e-05,
      "step": 32421,
      "training_step_time": 7.691150426864624
    },
    {
      "epoch": 3.0920028686523435e-05,
      "model_forward_time": 1.7266488075256348,
      "step": 32422
    },
    {
      "epoch": 3.0920028686523435e-05,
      "step": 32422,
      "training_step_time": 6.840684652328491
    },
    {
      "epoch": 3.0920982360839845e-05,
      "model_forward_time": 1.740023136138916,
      "step": 32423
    },
    {
      "epoch": 3.0920982360839845e-05,
      "step": 32423,
      "training_step_time": 6.8072123527526855
    },
    {
      "epoch": 3.092193603515625e-05,
      "model_forward_time": 1.7349328994750977,
      "step": 32424
    },
    {
      "epoch": 3.092193603515625e-05,
      "step": 32424,
      "training_step_time": 6.774275064468384
    },
    {
      "epoch": 3.0922889709472654e-05,
      "model_forward_time": 1.7521514892578125,
      "step": 32425
    },
    {
      "epoch": 3.0922889709472654e-05,
      "step": 32425,
      "training_step_time": 6.842450857162476
    },
    {
      "epoch": 3.0923843383789064e-05,
      "model_forward_time": 1.7531132698059082,
      "step": 32426
    },
    {
      "epoch": 3.0923843383789064e-05,
      "step": 32426,
      "training_step_time": 6.7946858406066895
    },
    {
      "epoch": 3.092479705810547e-05,
      "model_forward_time": 1.7412431240081787,
      "step": 32427
    },
    {
      "epoch": 3.092479705810547e-05,
      "step": 32427,
      "training_step_time": 6.75364089012146
    },
    {
      "epoch": 3.092575073242187e-05,
      "model_forward_time": 2.252789258956909,
      "step": 32428
    },
    {
      "epoch": 3.092575073242187e-05,
      "step": 32428,
      "training_step_time": 7.580322742462158
    },
    {
      "epoch": 3.092670440673828e-05,
      "model_forward_time": 1.7777678966522217,
      "step": 32429
    },
    {
      "epoch": 3.092670440673828e-05,
      "step": 32429,
      "training_step_time": 10.301498174667358
    },
    {
      "epoch": 3.092765808105469e-05,
      "grad_norm": 0.10194377601146698,
      "learning_rate": 8.080319312020065e-06,
      "loss": 0.0596,
      "step": 32430
    },
    {
      "epoch": 3.092765808105469e-05,
      "model_forward_time": 1.746870756149292,
      "step": 32430
    },
    {
      "dynamics_loss_avg": 0.05124139841645956,
      "epoch": 3.092765808105469e-05,
      "step": 32430
    },
    {
      "action_loss_avg": 0.0025998383585829287,
      "epoch": 3.092765808105469e-05,
      "step": 32430
    },
    {
      "epoch": 3.092765808105469e-05,
      "step": 32430,
      "training_step_time": 7.670041084289551
    },
    {
      "epoch": 3.092861175537109e-05,
      "model_forward_time": 1.74055814743042,
      "step": 32431
    },
    {
      "epoch": 3.092861175537109e-05,
      "step": 32431,
      "training_step_time": 7.272869348526001
    },
    {
      "epoch": 3.09295654296875e-05,
      "model_forward_time": 1.7359540462493896,
      "step": 32432
    },
    {
      "epoch": 3.09295654296875e-05,
      "step": 32432,
      "training_step_time": 6.825013160705566
    },
    {
      "epoch": 3.0930519104003906e-05,
      "model_forward_time": 1.7379412651062012,
      "step": 32433
    },
    {
      "epoch": 3.0930519104003906e-05,
      "step": 32433,
      "training_step_time": 6.798704385757446
    },
    {
      "epoch": 3.093147277832031e-05,
      "model_forward_time": 1.7439806461334229,
      "step": 32434
    },
    {
      "epoch": 3.093147277832031e-05,
      "step": 32434,
      "training_step_time": 6.783637762069702
    },
    {
      "epoch": 3.093242645263672e-05,
      "model_forward_time": 1.7421867847442627,
      "step": 32435
    },
    {
      "epoch": 3.093242645263672e-05,
      "step": 32435,
      "training_step_time": 6.827479362487793
    },
    {
      "epoch": 3.0933380126953125e-05,
      "model_forward_time": 1.7395241260528564,
      "step": 32436
    },
    {
      "epoch": 3.0933380126953125e-05,
      "step": 32436,
      "training_step_time": 6.915099620819092
    },
    {
      "epoch": 3.093433380126953e-05,
      "model_forward_time": 2.282590389251709,
      "step": 32437
    },
    {
      "epoch": 3.093433380126953e-05,
      "step": 32437,
      "training_step_time": 9.510724067687988
    },
    {
      "epoch": 3.093528747558594e-05,
      "model_forward_time": 1.7782118320465088,
      "step": 32438
    },
    {
      "epoch": 3.093528747558594e-05,
      "step": 32438,
      "training_step_time": 9.374814510345459
    },
    {
      "epoch": 3.0936241149902344e-05,
      "model_forward_time": 1.735541820526123,
      "step": 32439
    },
    {
      "epoch": 3.0936241149902344e-05,
      "step": 32439,
      "training_step_time": 7.787147760391235
    },
    {
      "epoch": 3.093719482421875e-05,
      "grad_norm": 0.13931621611118317,
      "learning_rate": 8.079016711707422e-06,
      "loss": 0.0638,
      "step": 32440
    },
    {
      "epoch": 3.093719482421875e-05,
      "model_forward_time": 1.744706153869629,
      "step": 32440
    },
    {
      "dynamics_loss_avg": 0.048459132015705106,
      "epoch": 3.093719482421875e-05,
      "step": 32440
    },
    {
      "action_loss_avg": 0.0020018623967189343,
      "epoch": 3.093719482421875e-05,
      "step": 32440
    },
    {
      "epoch": 3.093719482421875e-05,
      "step": 32440,
      "training_step_time": 7.324956178665161
    },
    {
      "epoch": 3.093814849853516e-05,
      "model_forward_time": 1.737196445465088,
      "step": 32441
    },
    {
      "epoch": 3.093814849853516e-05,
      "step": 32441,
      "training_step_time": 6.80221700668335
    },
    {
      "epoch": 3.093910217285156e-05,
      "model_forward_time": 1.7382075786590576,
      "step": 32442
    },
    {
      "epoch": 3.093910217285156e-05,
      "step": 32442,
      "training_step_time": 6.872729063034058
    },
    {
      "epoch": 3.094005584716797e-05,
      "model_forward_time": 1.740718126296997,
      "step": 32443
    },
    {
      "epoch": 3.094005584716797e-05,
      "step": 32443,
      "training_step_time": 6.787353992462158
    },
    {
      "epoch": 3.094100952148438e-05,
      "model_forward_time": 1.7412259578704834,
      "step": 32444
    },
    {
      "epoch": 3.094100952148438e-05,
      "step": 32444,
      "training_step_time": 6.782537937164307
    },
    {
      "epoch": 3.094196319580078e-05,
      "model_forward_time": 1.7431671619415283,
      "step": 32445
    },
    {
      "epoch": 3.094196319580078e-05,
      "step": 32445,
      "training_step_time": 6.841583967208862
    },
    {
      "epoch": 3.0942916870117186e-05,
      "model_forward_time": 2.2908151149749756,
      "step": 32446
    },
    {
      "epoch": 3.0942916870117186e-05,
      "step": 32446,
      "training_step_time": 7.643263816833496
    },
    {
      "epoch": 3.0943870544433596e-05,
      "model_forward_time": 1.746922492980957,
      "step": 32447
    },
    {
      "epoch": 3.0943870544433596e-05,
      "step": 32447,
      "training_step_time": 10.892340183258057
    },
    {
      "epoch": 3.094482421875e-05,
      "model_forward_time": 1.7363040447235107,
      "step": 32448
    },
    {
      "epoch": 3.094482421875e-05,
      "step": 32448,
      "training_step_time": 8.914637804031372
    },
    {
      "epoch": 3.0945777893066404e-05,
      "model_forward_time": 1.7504587173461914,
      "step": 32449
    },
    {
      "epoch": 3.0945777893066404e-05,
      "step": 32449,
      "training_step_time": 7.617117881774902
    },
    {
      "epoch": 3.0946731567382815e-05,
      "grad_norm": 0.1265968382358551,
      "learning_rate": 8.07771377467814e-06,
      "loss": 0.0616,
      "step": 32450
    },
    {
      "epoch": 3.0946731567382815e-05,
      "model_forward_time": 1.737480878829956,
      "step": 32450
    },
    {
      "dynamics_loss_avg": 0.05710673639550805,
      "epoch": 3.0946731567382815e-05,
      "step": 32450
    },
    {
      "action_loss_avg": 0.0021251325437333436,
      "epoch": 3.0946731567382815e-05,
      "step": 32450
    },
    {
      "epoch": 3.0946731567382815e-05,
      "step": 32450,
      "training_step_time": 6.788837671279907
    },
    {
      "epoch": 3.094768524169922e-05,
      "model_forward_time": 1.753192663192749,
      "step": 32451
    },
    {
      "epoch": 3.094768524169922e-05,
      "step": 32451,
      "training_step_time": 6.884186744689941
    },
    {
      "epoch": 3.094863891601562e-05,
      "model_forward_time": 1.7515344619750977,
      "step": 32452
    },
    {
      "epoch": 3.094863891601562e-05,
      "step": 32452,
      "training_step_time": 6.852927923202515
    },
    {
      "epoch": 3.0949592590332034e-05,
      "model_forward_time": 1.7518157958984375,
      "step": 32453
    },
    {
      "epoch": 3.0949592590332034e-05,
      "step": 32453,
      "training_step_time": 6.819839715957642
    },
    {
      "epoch": 3.095054626464844e-05,
      "model_forward_time": 1.7322397232055664,
      "step": 32454
    },
    {
      "epoch": 3.095054626464844e-05,
      "step": 32454,
      "training_step_time": 6.912683010101318
    },
    {
      "epoch": 3.095149993896484e-05,
      "model_forward_time": 2.280261516571045,
      "step": 32455
    },
    {
      "epoch": 3.095149993896484e-05,
      "step": 32455,
      "training_step_time": 7.627612113952637
    },
    {
      "epoch": 3.095245361328125e-05,
      "model_forward_time": 1.7623801231384277,
      "step": 32456
    },
    {
      "epoch": 3.095245361328125e-05,
      "step": 32456,
      "training_step_time": 7.724110841751099
    },
    {
      "epoch": 3.095340728759766e-05,
      "model_forward_time": 1.7397544384002686,
      "step": 32457
    },
    {
      "epoch": 3.095340728759766e-05,
      "step": 32457,
      "training_step_time": 9.13457179069519
    },
    {
      "epoch": 3.095436096191406e-05,
      "model_forward_time": 1.733947515487671,
      "step": 32458
    },
    {
      "epoch": 3.095436096191406e-05,
      "step": 32458,
      "training_step_time": 7.688159227371216
    },
    {
      "epoch": 3.095531463623047e-05,
      "model_forward_time": 1.738405466079712,
      "step": 32459
    },
    {
      "epoch": 3.095531463623047e-05,
      "step": 32459,
      "training_step_time": 7.357177019119263
    },
    {
      "epoch": 3.0956268310546876e-05,
      "grad_norm": 0.15460816025733948,
      "learning_rate": 8.076410501074707e-06,
      "loss": 0.0586,
      "step": 32460
    },
    {
      "epoch": 3.0956268310546876e-05,
      "model_forward_time": 1.7553305625915527,
      "step": 32460
    },
    {
      "dynamics_loss_avg": 0.04403376746922731,
      "epoch": 3.0956268310546876e-05,
      "step": 32460
    },
    {
      "action_loss_avg": 0.0017947127809748054,
      "epoch": 3.0956268310546876e-05,
      "step": 32460
    },
    {
      "epoch": 3.0956268310546876e-05,
      "step": 32460,
      "training_step_time": 6.929524660110474
    },
    {
      "epoch": 3.095722198486328e-05,
      "model_forward_time": 1.7472400665283203,
      "step": 32461
    },
    {
      "epoch": 3.095722198486328e-05,
      "step": 32461,
      "training_step_time": 7.0035316944122314
    },
    {
      "epoch": 3.095817565917969e-05,
      "model_forward_time": 1.74184250831604,
      "step": 32462
    },
    {
      "epoch": 3.095817565917969e-05,
      "step": 32462,
      "training_step_time": 6.959822654724121
    },
    {
      "epoch": 3.0959129333496095e-05,
      "model_forward_time": 1.742811679840088,
      "step": 32463
    },
    {
      "epoch": 3.0959129333496095e-05,
      "step": 32463,
      "training_step_time": 6.912644386291504
    },
    {
      "epoch": 3.09600830078125e-05,
      "model_forward_time": 2.300896167755127,
      "step": 32464
    },
    {
      "epoch": 3.09600830078125e-05,
      "step": 32464,
      "training_step_time": 7.68601655960083
    },
    {
      "epoch": 3.096103668212891e-05,
      "model_forward_time": 1.745772361755371,
      "step": 32465
    },
    {
      "epoch": 3.096103668212891e-05,
      "step": 32465,
      "training_step_time": 7.766813278198242
    },
    {
      "epoch": 3.0961990356445314e-05,
      "model_forward_time": 1.7401397228240967,
      "step": 32466
    },
    {
      "epoch": 3.0961990356445314e-05,
      "step": 32466,
      "training_step_time": 9.326099395751953
    },
    {
      "epoch": 3.096294403076172e-05,
      "model_forward_time": 1.7512662410736084,
      "step": 32467
    },
    {
      "epoch": 3.096294403076172e-05,
      "step": 32467,
      "training_step_time": 7.747098445892334
    },
    {
      "epoch": 3.096389770507813e-05,
      "model_forward_time": 1.7406463623046875,
      "step": 32468
    },
    {
      "epoch": 3.096389770507813e-05,
      "step": 32468,
      "training_step_time": 7.624429702758789
    },
    {
      "epoch": 3.096485137939453e-05,
      "model_forward_time": 1.7300288677215576,
      "step": 32469
    },
    {
      "epoch": 3.096485137939453e-05,
      "step": 32469,
      "training_step_time": 7.17307710647583
    },
    {
      "epoch": 3.0965805053710936e-05,
      "grad_norm": 0.15028826892375946,
      "learning_rate": 8.075106891039648e-06,
      "loss": 0.0637,
      "step": 32470
    },
    {
      "epoch": 3.0965805053710936e-05,
      "model_forward_time": 1.7382957935333252,
      "step": 32470
    },
    {
      "dynamics_loss_avg": 0.05621466413140297,
      "epoch": 3.0965805053710936e-05,
      "step": 32470
    },
    {
      "action_loss_avg": 0.0016384036047384144,
      "epoch": 3.0965805053710936e-05,
      "step": 32470
    },
    {
      "epoch": 3.0965805053710936e-05,
      "step": 32470,
      "training_step_time": 6.790968894958496
    },
    {
      "epoch": 3.096675872802734e-05,
      "model_forward_time": 1.7309441566467285,
      "step": 32471
    },
    {
      "epoch": 3.096675872802734e-05,
      "step": 32471,
      "training_step_time": 6.901919603347778
    },
    {
      "epoch": 3.096771240234375e-05,
      "model_forward_time": 1.730088710784912,
      "step": 32472
    },
    {
      "epoch": 3.096771240234375e-05,
      "step": 32472,
      "training_step_time": 6.877284526824951
    },
    {
      "epoch": 3.0968666076660155e-05,
      "model_forward_time": 2.284060001373291,
      "step": 32473
    },
    {
      "epoch": 3.0968666076660155e-05,
      "step": 32473,
      "training_step_time": 7.694355249404907
    },
    {
      "epoch": 3.096961975097656e-05,
      "model_forward_time": 1.7285516262054443,
      "step": 32474
    },
    {
      "epoch": 3.096961975097656e-05,
      "step": 32474,
      "training_step_time": 8.303286075592041
    },
    {
      "epoch": 3.097057342529297e-05,
      "model_forward_time": 1.7679686546325684,
      "step": 32475
    },
    {
      "epoch": 3.097057342529297e-05,
      "step": 32475,
      "training_step_time": 9.095903158187866
    },
    {
      "epoch": 3.0971527099609374e-05,
      "model_forward_time": 1.7311592102050781,
      "step": 32476
    },
    {
      "epoch": 3.0971527099609374e-05,
      "step": 32476,
      "training_step_time": 7.6639556884765625
    },
    {
      "epoch": 3.097248077392578e-05,
      "model_forward_time": 1.7306110858917236,
      "step": 32477
    },
    {
      "epoch": 3.097248077392578e-05,
      "step": 32477,
      "training_step_time": 7.543910980224609
    },
    {
      "epoch": 3.097343444824219e-05,
      "model_forward_time": 1.7278087139129639,
      "step": 32478
    },
    {
      "epoch": 3.097343444824219e-05,
      "step": 32478,
      "training_step_time": 7.059234857559204
    },
    {
      "epoch": 3.097438812255859e-05,
      "model_forward_time": 1.7298576831817627,
      "step": 32479
    },
    {
      "epoch": 3.097438812255859e-05,
      "step": 32479,
      "training_step_time": 6.774045467376709
    },
    {
      "epoch": 3.0975341796875e-05,
      "grad_norm": 0.10983569175004959,
      "learning_rate": 8.073802944715524e-06,
      "loss": 0.0597,
      "step": 32480
    },
    {
      "epoch": 3.0975341796875e-05,
      "model_forward_time": 1.7281053066253662,
      "step": 32480
    },
    {
      "dynamics_loss_avg": 0.05201307814568281,
      "epoch": 3.0975341796875e-05,
      "step": 32480
    },
    {
      "action_loss_avg": 0.0014674504986032844,
      "epoch": 3.0975341796875e-05,
      "step": 32480
    },
    {
      "epoch": 3.0975341796875e-05,
      "step": 32480,
      "training_step_time": 6.772236108779907
    },
    {
      "epoch": 3.097629547119141e-05,
      "model_forward_time": 1.7236628532409668,
      "step": 32481
    },
    {
      "epoch": 3.097629547119141e-05,
      "step": 32481,
      "training_step_time": 6.781085729598999
    },
    {
      "epoch": 3.097724914550781e-05,
      "model_forward_time": 2.2678587436676025,
      "step": 32482
    },
    {
      "epoch": 3.097724914550781e-05,
      "step": 32482,
      "training_step_time": 7.624749183654785
    },
    {
      "epoch": 3.0978202819824216e-05,
      "model_forward_time": 1.3442718982696533,
      "step": 32483
    },
    {
      "epoch": 3.0978202819824216e-05,
      "step": 32483,
      "training_step_time": 8.772955179214478
    },
    {
      "epoch": 3.097915649414063e-05,
      "model_forward_time": 1.7671942710876465,
      "step": 32484
    },
    {
      "epoch": 3.097915649414063e-05,
      "step": 32484,
      "training_step_time": 10.947317123413086
    },
    {
      "epoch": 3.098011016845703e-05,
      "model_forward_time": 1.732607126235962,
      "step": 32485
    },
    {
      "epoch": 3.098011016845703e-05,
      "step": 32485,
      "training_step_time": 7.815284013748169
    },
    {
      "epoch": 3.0981063842773435e-05,
      "model_forward_time": 1.724782943725586,
      "step": 32486
    },
    {
      "epoch": 3.0981063842773435e-05,
      "step": 32486,
      "training_step_time": 7.805531740188599
    },
    {
      "epoch": 3.0982017517089846e-05,
      "model_forward_time": 1.7249844074249268,
      "step": 32487
    },
    {
      "epoch": 3.0982017517089846e-05,
      "step": 32487,
      "training_step_time": 7.256603479385376
    },
    {
      "epoch": 3.098297119140625e-05,
      "model_forward_time": 1.736377239227295,
      "step": 32488
    },
    {
      "epoch": 3.098297119140625e-05,
      "step": 32488,
      "training_step_time": 6.913907289505005
    },
    {
      "epoch": 3.0983924865722654e-05,
      "model_forward_time": 1.75942063331604,
      "step": 32489
    },
    {
      "epoch": 3.0983924865722654e-05,
      "step": 32489,
      "training_step_time": 6.919596910476685
    },
    {
      "epoch": 3.0984878540039064e-05,
      "grad_norm": 0.1603493094444275,
      "learning_rate": 8.07249866224493e-06,
      "loss": 0.0627,
      "step": 32490
    },
    {
      "epoch": 3.0984878540039064e-05,
      "model_forward_time": 1.7349834442138672,
      "step": 32490
    },
    {
      "dynamics_loss_avg": 0.043575504515320065,
      "epoch": 3.0984878540039064e-05,
      "step": 32490
    },
    {
      "action_loss_avg": 0.0013326840533409267,
      "epoch": 3.0984878540039064e-05,
      "step": 32490
    },
    {
      "epoch": 3.0984878540039064e-05,
      "step": 32490,
      "training_step_time": 6.925533771514893
    },
    {
      "epoch": 3.098583221435547e-05,
      "model_forward_time": 2.309706211090088,
      "step": 32491
    },
    {
      "epoch": 3.098583221435547e-05,
      "step": 32491,
      "training_step_time": 7.674902439117432
    },
    {
      "epoch": 3.098678588867187e-05,
      "model_forward_time": 1.7326409816741943,
      "step": 32492
    },
    {
      "epoch": 3.098678588867187e-05,
      "step": 32492,
      "training_step_time": 9.187830448150635
    },
    {
      "epoch": 3.098773956298828e-05,
      "model_forward_time": 1.7444241046905518,
      "step": 32493
    },
    {
      "epoch": 3.098773956298828e-05,
      "step": 32493,
      "training_step_time": 9.77873182296753
    },
    {
      "epoch": 3.098869323730469e-05,
      "model_forward_time": 1.022392749786377,
      "step": 32494
    },
    {
      "epoch": 3.098869323730469e-05,
      "step": 32494,
      "training_step_time": 9.712677955627441
    },
    {
      "epoch": 3.098964691162109e-05,
      "model_forward_time": 1.7275276184082031,
      "step": 32495
    },
    {
      "epoch": 3.098964691162109e-05,
      "step": 32495,
      "training_step_time": 7.7829930782318115
    },
    {
      "epoch": 3.09906005859375e-05,
      "model_forward_time": 1.7371118068695068,
      "step": 32496
    },
    {
      "epoch": 3.09906005859375e-05,
      "step": 32496,
      "training_step_time": 7.227331161499023
    },
    {
      "epoch": 3.0991554260253906e-05,
      "model_forward_time": 1.7285487651824951,
      "step": 32497
    },
    {
      "epoch": 3.0991554260253906e-05,
      "step": 32497,
      "training_step_time": 6.991041898727417
    },
    {
      "epoch": 3.099250793457031e-05,
      "model_forward_time": 1.7292718887329102,
      "step": 32498
    },
    {
      "epoch": 3.099250793457031e-05,
      "step": 32498,
      "training_step_time": 6.939053535461426
    },
    {
      "epoch": 3.099346160888672e-05,
      "model_forward_time": 1.7265689373016357,
      "step": 32499
    },
    {
      "epoch": 3.099346160888672e-05,
      "step": 32499,
      "training_step_time": 6.944651365280151
    },
    {
      "epoch": 3.0994415283203125e-05,
      "grad_norm": 0.1747080385684967,
      "learning_rate": 8.0711940437705e-06,
      "loss": 0.0615,
      "step": 32500
    },
    {
      "epoch": 3.0994415283203125e-05,
      "model_forward_time": 1.730262041091919,
      "step": 32500
    },
    {
      "dynamics_loss_avg": 0.06091527566313744,
      "epoch": 3.0994415283203125e-05,
      "step": 32500
    },
    {
      "action_loss_avg": 0.0017834454163676127,
      "epoch": 3.0994415283203125e-05,
      "step": 32500
    },
    {
      "epoch": 3.0994415283203125e-05,
      "step": 32500,
      "training_step_time": 6.947753429412842
    },
    {
      "epoch": 3.099536895751953e-05,
      "model_forward_time": 2.2697391510009766,
      "step": 32501
    },
    {
      "epoch": 3.099536895751953e-05,
      "step": 32501,
      "training_step_time": 8.819452285766602
    },
    {
      "epoch": 3.099632263183594e-05,
      "model_forward_time": 1.7208397388458252,
      "step": 32502
    },
    {
      "epoch": 3.099632263183594e-05,
      "step": 32502,
      "training_step_time": 8.844010829925537
    },
    {
      "epoch": 3.0997276306152344e-05,
      "model_forward_time": 1.727041482925415,
      "step": 32503
    },
    {
      "epoch": 3.0997276306152344e-05,
      "step": 32503,
      "training_step_time": 10.133636474609375
    },
    {
      "epoch": 3.099822998046875e-05,
      "model_forward_time": 1.729402780532837,
      "step": 32504
    },
    {
      "epoch": 3.099822998046875e-05,
      "step": 32504,
      "training_step_time": 7.639849662780762
    },
    {
      "epoch": 3.099918365478516e-05,
      "model_forward_time": 1.72981595993042,
      "step": 32505
    },
    {
      "epoch": 3.099918365478516e-05,
      "step": 32505,
      "training_step_time": 7.181201696395874
    },
    {
      "epoch": 3.100013732910156e-05,
      "model_forward_time": 1.7283813953399658,
      "step": 32506
    },
    {
      "epoch": 3.100013732910156e-05,
      "step": 32506,
      "training_step_time": 6.8053460121154785
    },
    {
      "epoch": 3.100109100341797e-05,
      "model_forward_time": 1.7294201850891113,
      "step": 32507
    },
    {
      "epoch": 3.100109100341797e-05,
      "step": 32507,
      "training_step_time": 6.783315181732178
    },
    {
      "epoch": 3.100204467773438e-05,
      "model_forward_time": 1.7352945804595947,
      "step": 32508
    },
    {
      "epoch": 3.100204467773438e-05,
      "step": 32508,
      "training_step_time": 6.763200998306274
    },
    {
      "epoch": 3.100299835205078e-05,
      "model_forward_time": 1.7364270687103271,
      "step": 32509
    },
    {
      "epoch": 3.100299835205078e-05,
      "step": 32509,
      "training_step_time": 6.773189306259155
    },
    {
      "epoch": 3.1003952026367186e-05,
      "grad_norm": 0.1702687293291092,
      "learning_rate": 8.069889089434911e-06,
      "loss": 0.0611,
      "step": 32510
    },
    {
      "epoch": 3.1003952026367186e-05,
      "model_forward_time": 2.2765004634857178,
      "step": 32510
    },
    {
      "dynamics_loss_avg": 0.05552710350602865,
      "epoch": 3.1003952026367186e-05,
      "step": 32510
    },
    {
      "action_loss_avg": 0.001663859054679051,
      "epoch": 3.1003952026367186e-05,
      "step": 32510
    },
    {
      "epoch": 3.1003952026367186e-05,
      "step": 32510,
      "training_step_time": 8.375048398971558
    },
    {
      "epoch": 3.1004905700683596e-05,
      "model_forward_time": 1.7188208103179932,
      "step": 32511
    },
    {
      "epoch": 3.1004905700683596e-05,
      "step": 32511,
      "training_step_time": 8.575866460800171
    },
    {
      "epoch": 3.1005859375e-05,
      "model_forward_time": 1.7631690502166748,
      "step": 32512
    },
    {
      "epoch": 3.1005859375e-05,
      "step": 32512,
      "training_step_time": 10.400649070739746
    },
    {
      "epoch": 3.1006813049316404e-05,
      "model_forward_time": 1.7345504760742188,
      "step": 32513
    },
    {
      "epoch": 3.1006813049316404e-05,
      "step": 32513,
      "training_step_time": 8.565556287765503
    },
    {
      "epoch": 3.1007766723632815e-05,
      "model_forward_time": 1.713021993637085,
      "step": 32514
    },
    {
      "epoch": 3.1007766723632815e-05,
      "step": 32514,
      "training_step_time": 7.532123804092407
    },
    {
      "epoch": 3.100872039794922e-05,
      "model_forward_time": 1.73069167137146,
      "step": 32515
    },
    {
      "epoch": 3.100872039794922e-05,
      "step": 32515,
      "training_step_time": 6.988192319869995
    },
    {
      "epoch": 3.100967407226562e-05,
      "model_forward_time": 1.7209599018096924,
      "step": 32516
    },
    {
      "epoch": 3.100967407226562e-05,
      "step": 32516,
      "training_step_time": 6.9621970653533936
    },
    {
      "epoch": 3.1010627746582034e-05,
      "model_forward_time": 1.7187275886535645,
      "step": 32517
    },
    {
      "epoch": 3.1010627746582034e-05,
      "step": 32517,
      "training_step_time": 6.795778512954712
    },
    {
      "epoch": 3.101158142089844e-05,
      "model_forward_time": 1.722783088684082,
      "step": 32518
    },
    {
      "epoch": 3.101158142089844e-05,
      "step": 32518,
      "training_step_time": 6.764536142349243
    },
    {
      "epoch": 3.101253509521484e-05,
      "model_forward_time": 2.2520041465759277,
      "step": 32519
    },
    {
      "epoch": 3.101253509521484e-05,
      "step": 32519,
      "training_step_time": 8.125810623168945
    },
    {
      "epoch": 3.101348876953125e-05,
      "grad_norm": 0.14653319120407104,
      "learning_rate": 8.068583799380862e-06,
      "loss": 0.0593,
      "step": 32520
    },
    {
      "epoch": 3.101348876953125e-05,
      "model_forward_time": 1.7180423736572266,
      "step": 32520
    },
    {
      "dynamics_loss_avg": 0.04944996200501919,
      "epoch": 3.101348876953125e-05,
      "step": 32520
    },
    {
      "action_loss_avg": 0.002019896556157619,
      "epoch": 3.101348876953125e-05,
      "step": 32520
    },
    {
      "epoch": 3.101348876953125e-05,
      "step": 32520,
      "training_step_time": 7.563433647155762
    },
    {
      "epoch": 3.101444244384766e-05,
      "model_forward_time": 1.7581183910369873,
      "step": 32521
    },
    {
      "epoch": 3.101444244384766e-05,
      "step": 32521,
      "training_step_time": 7.680773019790649
    },
    {
      "epoch": 3.101539611816406e-05,
      "model_forward_time": 1.7250452041625977,
      "step": 32522
    },
    {
      "epoch": 3.101539611816406e-05,
      "step": 32522,
      "training_step_time": 8.319701194763184
    },
    {
      "epoch": 3.101634979248047e-05,
      "model_forward_time": 1.7275257110595703,
      "step": 32523
    },
    {
      "epoch": 3.101634979248047e-05,
      "step": 32523,
      "training_step_time": 8.072056770324707
    },
    {
      "epoch": 3.1017303466796876e-05,
      "model_forward_time": 1.7172975540161133,
      "step": 32524
    },
    {
      "epoch": 3.1017303466796876e-05,
      "step": 32524,
      "training_step_time": 6.783644199371338
    },
    {
      "epoch": 3.101825714111328e-05,
      "model_forward_time": 1.7235746383666992,
      "step": 32525
    },
    {
      "epoch": 3.101825714111328e-05,
      "step": 32525,
      "training_step_time": 6.79713773727417
    },
    {
      "epoch": 3.101921081542969e-05,
      "model_forward_time": 1.7249739170074463,
      "step": 32526
    },
    {
      "epoch": 3.101921081542969e-05,
      "step": 32526,
      "training_step_time": 6.792006015777588
    },
    {
      "epoch": 3.1020164489746095e-05,
      "model_forward_time": 1.7232987880706787,
      "step": 32527
    },
    {
      "epoch": 3.1020164489746095e-05,
      "step": 32527,
      "training_step_time": 6.834608554840088
    },
    {
      "epoch": 3.10211181640625e-05,
      "model_forward_time": 2.2947230339050293,
      "step": 32528
    },
    {
      "epoch": 3.10211181640625e-05,
      "step": 32528,
      "training_step_time": 7.568700313568115
    },
    {
      "epoch": 3.102207183837891e-05,
      "model_forward_time": 1.714925765991211,
      "step": 32529
    },
    {
      "epoch": 3.102207183837891e-05,
      "step": 32529,
      "training_step_time": 10.184462547302246
    },
    {
      "epoch": 3.1023025512695314e-05,
      "grad_norm": 0.09394407272338867,
      "learning_rate": 8.067278173751106e-06,
      "loss": 0.0603,
      "step": 32530
    },
    {
      "epoch": 3.1023025512695314e-05,
      "model_forward_time": 1.7350530624389648,
      "step": 32530
    },
    {
      "dynamics_loss_avg": 0.051646208949387075,
      "epoch": 3.1023025512695314e-05,
      "step": 32530
    },
    {
      "action_loss_avg": 0.0016344731382559984,
      "epoch": 3.1023025512695314e-05,
      "step": 32530
    },
    {
      "epoch": 3.1023025512695314e-05,
      "step": 32530,
      "training_step_time": 9.70021367073059
    },
    {
      "epoch": 3.102397918701172e-05,
      "model_forward_time": 1.7298023700714111,
      "step": 32531
    },
    {
      "epoch": 3.102397918701172e-05,
      "step": 32531,
      "training_step_time": 9.87070631980896
    },
    {
      "epoch": 3.102493286132812e-05,
      "model_forward_time": 1.715601921081543,
      "step": 32532
    },
    {
      "epoch": 3.102493286132812e-05,
      "step": 32532,
      "training_step_time": 7.647002220153809
    },
    {
      "epoch": 3.102588653564453e-05,
      "model_forward_time": 1.7284772396087646,
      "step": 32533
    },
    {
      "epoch": 3.102588653564453e-05,
      "step": 32533,
      "training_step_time": 7.033681631088257
    },
    {
      "epoch": 3.1026840209960936e-05,
      "model_forward_time": 1.7120389938354492,
      "step": 32534
    },
    {
      "epoch": 3.1026840209960936e-05,
      "step": 32534,
      "training_step_time": 6.665085554122925
    },
    {
      "epoch": 3.102779388427734e-05,
      "model_forward_time": 1.7210118770599365,
      "step": 32535
    },
    {
      "epoch": 3.102779388427734e-05,
      "step": 32535,
      "training_step_time": 6.70228910446167
    },
    {
      "epoch": 3.102874755859375e-05,
      "model_forward_time": 1.7203564643859863,
      "step": 32536
    },
    {
      "epoch": 3.102874755859375e-05,
      "step": 32536,
      "training_step_time": 6.760023832321167
    },
    {
      "epoch": 3.1029701232910155e-05,
      "model_forward_time": 2.274425506591797,
      "step": 32537
    },
    {
      "epoch": 3.1029701232910155e-05,
      "step": 32537,
      "training_step_time": 7.528992414474487
    },
    {
      "epoch": 3.103065490722656e-05,
      "model_forward_time": 1.725032091140747,
      "step": 32538
    },
    {
      "epoch": 3.103065490722656e-05,
      "step": 32538,
      "training_step_time": 8.380342245101929
    },
    {
      "epoch": 3.103160858154297e-05,
      "model_forward_time": 1.720219373703003,
      "step": 32539
    },
    {
      "epoch": 3.103160858154297e-05,
      "step": 32539,
      "training_step_time": 10.296162605285645
    },
    {
      "epoch": 3.1032562255859374e-05,
      "grad_norm": 0.11298120021820068,
      "learning_rate": 8.065972212688418e-06,
      "loss": 0.0632,
      "step": 32540
    },
    {
      "epoch": 3.1032562255859374e-05,
      "model_forward_time": 1.7171037197113037,
      "step": 32540
    },
    {
      "dynamics_loss_avg": 0.04394607562571764,
      "epoch": 3.1032562255859374e-05,
      "step": 32540
    },
    {
      "action_loss_avg": 0.0019446573744062335,
      "epoch": 3.1032562255859374e-05,
      "step": 32540
    },
    {
      "epoch": 3.1032562255859374e-05,
      "step": 32540,
      "training_step_time": 10.108921527862549
    },
    {
      "epoch": 3.103351593017578e-05,
      "model_forward_time": 1.7267532348632812,
      "step": 32541
    },
    {
      "epoch": 3.103351593017578e-05,
      "step": 32541,
      "training_step_time": 7.829902410507202
    },
    {
      "epoch": 3.103446960449219e-05,
      "model_forward_time": 1.7249584197998047,
      "step": 32542
    },
    {
      "epoch": 3.103446960449219e-05,
      "step": 32542,
      "training_step_time": 7.552185535430908
    },
    {
      "epoch": 3.103542327880859e-05,
      "model_forward_time": 1.739283800125122,
      "step": 32543
    },
    {
      "epoch": 3.103542327880859e-05,
      "step": 32543,
      "training_step_time": 6.759602785110474
    },
    {
      "epoch": 3.1036376953125e-05,
      "model_forward_time": 1.7242910861968994,
      "step": 32544
    },
    {
      "epoch": 3.1036376953125e-05,
      "step": 32544,
      "training_step_time": 6.736066818237305
    },
    {
      "epoch": 3.103733062744141e-05,
      "model_forward_time": 1.7358014583587646,
      "step": 32545
    },
    {
      "epoch": 3.103733062744141e-05,
      "step": 32545,
      "training_step_time": 6.741483449935913
    },
    {
      "epoch": 3.103828430175781e-05,
      "model_forward_time": 1.7351582050323486,
      "step": 32546
    },
    {
      "epoch": 3.103828430175781e-05,
      "step": 32546,
      "training_step_time": 8.668130159378052
    },
    {
      "epoch": 3.1039237976074216e-05,
      "model_forward_time": 1.7224829196929932,
      "step": 32547
    },
    {
      "epoch": 3.1039237976074216e-05,
      "step": 32547,
      "training_step_time": 9.727967500686646
    },
    {
      "epoch": 3.104019165039063e-05,
      "model_forward_time": 1.733773946762085,
      "step": 32548
    },
    {
      "epoch": 3.104019165039063e-05,
      "step": 32548,
      "training_step_time": 7.685616731643677
    },
    {
      "epoch": 3.104114532470703e-05,
      "model_forward_time": 1.7653999328613281,
      "step": 32549
    },
    {
      "epoch": 3.104114532470703e-05,
      "step": 32549,
      "training_step_time": 8.635198831558228
    },
    {
      "epoch": 3.1042098999023435e-05,
      "grad_norm": 0.10942940413951874,
      "learning_rate": 8.06466591633562e-06,
      "loss": 0.0577,
      "step": 32550
    },
    {
      "epoch": 3.1042098999023435e-05,
      "model_forward_time": 1.7377362251281738,
      "step": 32550
    },
    {
      "dynamics_loss_avg": 0.056271781027317044,
      "epoch": 3.1042098999023435e-05,
      "step": 32550
    },
    {
      "action_loss_avg": 0.002312852971954271,
      "epoch": 3.1042098999023435e-05,
      "step": 32550
    },
    {
      "epoch": 3.1042098999023435e-05,
      "step": 32550,
      "training_step_time": 7.670036554336548
    },
    {
      "epoch": 3.1043052673339846e-05,
      "model_forward_time": 1.7193899154663086,
      "step": 32551
    },
    {
      "epoch": 3.1043052673339846e-05,
      "step": 32551,
      "training_step_time": 7.524585723876953
    },
    {
      "epoch": 3.104400634765625e-05,
      "model_forward_time": 1.7264189720153809,
      "step": 32552
    },
    {
      "epoch": 3.104400634765625e-05,
      "step": 32552,
      "training_step_time": 6.750071048736572
    },
    {
      "epoch": 3.1044960021972654e-05,
      "model_forward_time": 1.7302143573760986,
      "step": 32553
    },
    {
      "epoch": 3.1044960021972654e-05,
      "step": 32553,
      "training_step_time": 6.726301431655884
    },
    {
      "epoch": 3.1045913696289064e-05,
      "model_forward_time": 1.7315466403961182,
      "step": 32554
    },
    {
      "epoch": 3.1045913696289064e-05,
      "step": 32554,
      "training_step_time": 6.8203513622283936
    },
    {
      "epoch": 3.104686737060547e-05,
      "model_forward_time": 1.7269048690795898,
      "step": 32555
    },
    {
      "epoch": 3.104686737060547e-05,
      "step": 32555,
      "training_step_time": 7.519549608230591
    },
    {
      "epoch": 3.104782104492187e-05,
      "model_forward_time": 2.252474069595337,
      "step": 32556
    },
    {
      "epoch": 3.104782104492187e-05,
      "step": 32556,
      "training_step_time": 7.959525108337402
    },
    {
      "epoch": 3.104877471923828e-05,
      "model_forward_time": 1.7424228191375732,
      "step": 32557
    },
    {
      "epoch": 3.104877471923828e-05,
      "step": 32557,
      "training_step_time": 7.690237283706665
    },
    {
      "epoch": 3.104972839355469e-05,
      "model_forward_time": 1.7701756954193115,
      "step": 32558
    },
    {
      "epoch": 3.104972839355469e-05,
      "step": 32558,
      "training_step_time": 9.498706817626953
    },
    {
      "epoch": 3.105068206787109e-05,
      "model_forward_time": 1.7310459613800049,
      "step": 32559
    },
    {
      "epoch": 3.105068206787109e-05,
      "step": 32559,
      "training_step_time": 8.771257162094116
    },
    {
      "epoch": 3.10516357421875e-05,
      "grad_norm": 0.12213066965341568,
      "learning_rate": 8.063359284835563e-06,
      "loss": 0.0594,
      "step": 32560
    },
    {
      "epoch": 3.10516357421875e-05,
      "model_forward_time": 1.7260539531707764,
      "step": 32560
    },
    {
      "dynamics_loss_avg": 0.06941364146769047,
      "epoch": 3.10516357421875e-05,
      "step": 32560
    },
    {
      "action_loss_avg": 0.0029215991962701082,
      "epoch": 3.10516357421875e-05,
      "step": 32560
    },
    {
      "epoch": 3.10516357421875e-05,
      "step": 32560,
      "training_step_time": 7.590750217437744
    },
    {
      "epoch": 3.1052589416503906e-05,
      "model_forward_time": 1.7290444374084473,
      "step": 32561
    },
    {
      "epoch": 3.1052589416503906e-05,
      "step": 32561,
      "training_step_time": 7.024397134780884
    },
    {
      "epoch": 3.105354309082031e-05,
      "model_forward_time": 1.7356503009796143,
      "step": 32562
    },
    {
      "epoch": 3.105354309082031e-05,
      "step": 32562,
      "training_step_time": 6.777349233627319
    },
    {
      "epoch": 3.105449676513672e-05,
      "model_forward_time": 1.7329137325286865,
      "step": 32563
    },
    {
      "epoch": 3.105449676513672e-05,
      "step": 32563,
      "training_step_time": 6.840847969055176
    },
    {
      "epoch": 3.1055450439453125e-05,
      "model_forward_time": 1.7269995212554932,
      "step": 32564
    },
    {
      "epoch": 3.1055450439453125e-05,
      "step": 32564,
      "training_step_time": 7.014138221740723
    },
    {
      "epoch": 3.105640411376953e-05,
      "model_forward_time": 2.2571916580200195,
      "step": 32565
    },
    {
      "epoch": 3.105640411376953e-05,
      "step": 32565,
      "training_step_time": 7.941537141799927
    },
    {
      "epoch": 3.105735778808594e-05,
      "model_forward_time": 1.7332782745361328,
      "step": 32566
    },
    {
      "epoch": 3.105735778808594e-05,
      "step": 32566,
      "training_step_time": 7.680933713912964
    },
    {
      "epoch": 3.1058311462402344e-05,
      "model_forward_time": 1.7660019397735596,
      "step": 32567
    },
    {
      "epoch": 3.1058311462402344e-05,
      "step": 32567,
      "training_step_time": 9.627562999725342
    },
    {
      "epoch": 3.105926513671875e-05,
      "model_forward_time": 1.7286336421966553,
      "step": 32568
    },
    {
      "epoch": 3.105926513671875e-05,
      "step": 32568,
      "training_step_time": 8.211137056350708
    },
    {
      "epoch": 3.106021881103516e-05,
      "model_forward_time": 1.731278419494629,
      "step": 32569
    },
    {
      "epoch": 3.106021881103516e-05,
      "step": 32569,
      "training_step_time": 7.602123260498047
    },
    {
      "epoch": 3.106117248535156e-05,
      "grad_norm": 0.13324958086013794,
      "learning_rate": 8.062052318331142e-06,
      "loss": 0.06,
      "step": 32570
    },
    {
      "epoch": 3.106117248535156e-05,
      "model_forward_time": 1.718749761581421,
      "step": 32570
    },
    {
      "dynamics_loss_avg": 0.07020136453211308,
      "epoch": 3.106117248535156e-05,
      "step": 32570
    },
    {
      "action_loss_avg": 0.002720356121426448,
      "epoch": 3.106117248535156e-05,
      "step": 32570
    },
    {
      "epoch": 3.106117248535156e-05,
      "step": 32570,
      "training_step_time": 7.04678201675415
    },
    {
      "epoch": 3.106212615966797e-05,
      "model_forward_time": 1.7645461559295654,
      "step": 32571
    },
    {
      "epoch": 3.106212615966797e-05,
      "step": 32571,
      "training_step_time": 6.778256416320801
    },
    {
      "epoch": 3.106307983398438e-05,
      "model_forward_time": 1.3586773872375488,
      "step": 32572
    },
    {
      "epoch": 3.106307983398438e-05,
      "step": 32572,
      "training_step_time": 6.73866605758667
    },
    {
      "epoch": 3.106403350830078e-05,
      "model_forward_time": 1.7346653938293457,
      "step": 32573
    },
    {
      "epoch": 3.106403350830078e-05,
      "step": 32573,
      "training_step_time": 7.011060953140259
    },
    {
      "epoch": 3.1064987182617186e-05,
      "model_forward_time": 2.283917188644409,
      "step": 32574
    },
    {
      "epoch": 3.1064987182617186e-05,
      "step": 32574,
      "training_step_time": 8.698968648910522
    },
    {
      "epoch": 3.1065940856933596e-05,
      "model_forward_time": 1.7322437763214111,
      "step": 32575
    },
    {
      "epoch": 3.1065940856933596e-05,
      "step": 32575,
      "training_step_time": 9.736562490463257
    },
    {
      "epoch": 3.106689453125e-05,
      "model_forward_time": 1.733485221862793,
      "step": 32576
    },
    {
      "epoch": 3.106689453125e-05,
      "step": 32576,
      "training_step_time": 10.648996591567993
    },
    {
      "epoch": 3.1067848205566405e-05,
      "model_forward_time": 1.7267239093780518,
      "step": 32577
    },
    {
      "epoch": 3.1067848205566405e-05,
      "step": 32577,
      "training_step_time": 9.199904918670654
    },
    {
      "epoch": 3.1068801879882815e-05,
      "model_forward_time": 1.7373058795928955,
      "step": 32578
    },
    {
      "epoch": 3.1068801879882815e-05,
      "step": 32578,
      "training_step_time": 7.76514458656311
    },
    {
      "epoch": 3.106975555419922e-05,
      "model_forward_time": 1.737363576889038,
      "step": 32579
    },
    {
      "epoch": 3.106975555419922e-05,
      "step": 32579,
      "training_step_time": 7.458651781082153
    },
    {
      "epoch": 3.107070922851562e-05,
      "grad_norm": 0.11978163570165634,
      "learning_rate": 8.060745016965284e-06,
      "loss": 0.0607,
      "step": 32580
    },
    {
      "epoch": 3.107070922851562e-05,
      "model_forward_time": 1.7479186058044434,
      "step": 32580
    },
    {
      "dynamics_loss_avg": 0.05571720860898495,
      "epoch": 3.107070922851562e-05,
      "step": 32580
    },
    {
      "action_loss_avg": 0.0014620969246607273,
      "epoch": 3.107070922851562e-05,
      "step": 32580
    },
    {
      "epoch": 3.107070922851562e-05,
      "step": 32580,
      "training_step_time": 6.794570684432983
    },
    {
      "epoch": 3.1071662902832034e-05,
      "model_forward_time": 1.7288165092468262,
      "step": 32581
    },
    {
      "epoch": 3.1071662902832034e-05,
      "step": 32581,
      "training_step_time": 6.815186500549316
    },
    {
      "epoch": 3.107261657714844e-05,
      "model_forward_time": 1.738919973373413,
      "step": 32582
    },
    {
      "epoch": 3.107261657714844e-05,
      "step": 32582,
      "training_step_time": 7.053065776824951
    },
    {
      "epoch": 3.107357025146484e-05,
      "model_forward_time": 2.2887706756591797,
      "step": 32583
    },
    {
      "epoch": 3.107357025146484e-05,
      "step": 32583,
      "training_step_time": 7.635098457336426
    },
    {
      "epoch": 3.107452392578125e-05,
      "model_forward_time": 1.7292344570159912,
      "step": 32584
    },
    {
      "epoch": 3.107452392578125e-05,
      "step": 32584,
      "training_step_time": 7.922748327255249
    },
    {
      "epoch": 3.107547760009766e-05,
      "model_forward_time": 1.7321245670318604,
      "step": 32585
    },
    {
      "epoch": 3.107547760009766e-05,
      "step": 32585,
      "training_step_time": 8.768966674804688
    },
    {
      "epoch": 3.107643127441406e-05,
      "model_forward_time": 1.7724661827087402,
      "step": 32586
    },
    {
      "epoch": 3.107643127441406e-05,
      "step": 32586,
      "training_step_time": 8.935727834701538
    },
    {
      "epoch": 3.107738494873047e-05,
      "model_forward_time": 1.7359178066253662,
      "step": 32587
    },
    {
      "epoch": 3.107738494873047e-05,
      "step": 32587,
      "training_step_time": 8.801592111587524
    },
    {
      "epoch": 3.1078338623046876e-05,
      "model_forward_time": 1.7359695434570312,
      "step": 32588
    },
    {
      "epoch": 3.1078338623046876e-05,
      "step": 32588,
      "training_step_time": 7.546987533569336
    },
    {
      "epoch": 3.107929229736328e-05,
      "model_forward_time": 1.7324118614196777,
      "step": 32589
    },
    {
      "epoch": 3.107929229736328e-05,
      "step": 32589,
      "training_step_time": 7.142014503479004
    },
    {
      "epoch": 3.108024597167969e-05,
      "grad_norm": 0.11597036570310593,
      "learning_rate": 8.059437380880952e-06,
      "loss": 0.0593,
      "step": 32590
    },
    {
      "epoch": 3.108024597167969e-05,
      "model_forward_time": 1.728888750076294,
      "step": 32590
    },
    {
      "dynamics_loss_avg": 0.06124147810041904,
      "epoch": 3.108024597167969e-05,
      "step": 32590
    },
    {
      "action_loss_avg": 0.0026421589427627623,
      "epoch": 3.108024597167969e-05,
      "step": 32590
    },
    {
      "epoch": 3.108024597167969e-05,
      "step": 32590,
      "training_step_time": 6.786365032196045
    },
    {
      "epoch": 3.1081199645996095e-05,
      "model_forward_time": 1.7338638305664062,
      "step": 32591
    },
    {
      "epoch": 3.1081199645996095e-05,
      "step": 32591,
      "training_step_time": 7.0499138832092285
    },
    {
      "epoch": 3.10821533203125e-05,
      "model_forward_time": 2.302793025970459,
      "step": 32592
    },
    {
      "epoch": 3.10821533203125e-05,
      "step": 32592,
      "training_step_time": 8.470986366271973
    },
    {
      "epoch": 3.108310699462891e-05,
      "model_forward_time": 1.7327775955200195,
      "step": 32593
    },
    {
      "epoch": 3.108310699462891e-05,
      "step": 32593,
      "training_step_time": 8.771997690200806
    },
    {
      "epoch": 3.1084060668945314e-05,
      "model_forward_time": 1.7339575290679932,
      "step": 32594
    },
    {
      "epoch": 3.1084060668945314e-05,
      "step": 32594,
      "training_step_time": 7.973894357681274
    },
    {
      "epoch": 3.108501434326172e-05,
      "model_forward_time": 1.767240285873413,
      "step": 32595
    },
    {
      "epoch": 3.108501434326172e-05,
      "step": 32595,
      "training_step_time": 8.865977048873901
    },
    {
      "epoch": 3.108596801757812e-05,
      "model_forward_time": 1.7454102039337158,
      "step": 32596
    },
    {
      "epoch": 3.108596801757812e-05,
      "step": 32596,
      "training_step_time": 8.43307089805603
    },
    {
      "epoch": 3.108692169189453e-05,
      "model_forward_time": 1.735517978668213,
      "step": 32597
    },
    {
      "epoch": 3.108692169189453e-05,
      "step": 32597,
      "training_step_time": 7.561344385147095
    },
    {
      "epoch": 3.1087875366210937e-05,
      "model_forward_time": 1.733583927154541,
      "step": 32598
    },
    {
      "epoch": 3.1087875366210937e-05,
      "step": 32598,
      "training_step_time": 7.051332712173462
    },
    {
      "epoch": 3.108882904052734e-05,
      "model_forward_time": 1.7374992370605469,
      "step": 32599
    },
    {
      "epoch": 3.108882904052734e-05,
      "step": 32599,
      "training_step_time": 6.738734006881714
    },
    {
      "epoch": 3.108978271484375e-05,
      "grad_norm": 0.12350708246231079,
      "learning_rate": 8.058129410221147e-06,
      "loss": 0.0639,
      "step": 32600
    },
    {
      "epoch": 3.108978271484375e-05,
      "model_forward_time": 1.7308902740478516,
      "step": 32600
    },
    {
      "dynamics_loss_avg": 0.058236677385866645,
      "epoch": 3.108978271484375e-05,
      "step": 32600
    },
    {
      "action_loss_avg": 0.0017133675632067026,
      "epoch": 3.108978271484375e-05,
      "step": 32600
    },
    {
      "epoch": 3.108978271484375e-05,
      "step": 32600,
      "training_step_time": 7.026754140853882
    },
    {
      "epoch": 3.1090736389160155e-05,
      "model_forward_time": 2.2780864238739014,
      "step": 32601
    },
    {
      "epoch": 3.1090736389160155e-05,
      "step": 32601,
      "training_step_time": 8.4247727394104
    },
    {
      "epoch": 3.109169006347656e-05,
      "model_forward_time": 1.7399325370788574,
      "step": 32602
    },
    {
      "epoch": 3.109169006347656e-05,
      "step": 32602,
      "training_step_time": 8.308979272842407
    },
    {
      "epoch": 3.109264373779297e-05,
      "model_forward_time": 1.7211356163024902,
      "step": 32603
    },
    {
      "epoch": 3.109264373779297e-05,
      "step": 32603,
      "training_step_time": 8.536600828170776
    },
    {
      "epoch": 3.1093597412109374e-05,
      "model_forward_time": 1.7666077613830566,
      "step": 32604
    },
    {
      "epoch": 3.1093597412109374e-05,
      "step": 32604,
      "training_step_time": 8.827134609222412
    },
    {
      "epoch": 3.109455108642578e-05,
      "model_forward_time": 1.3534965515136719,
      "step": 32605
    },
    {
      "epoch": 3.109455108642578e-05,
      "step": 32605,
      "training_step_time": 8.80709981918335
    },
    {
      "epoch": 3.109550476074219e-05,
      "model_forward_time": 1.724998950958252,
      "step": 32606
    },
    {
      "epoch": 3.109550476074219e-05,
      "step": 32606,
      "training_step_time": 7.637740850448608
    },
    {
      "epoch": 3.109645843505859e-05,
      "model_forward_time": 1.7369966506958008,
      "step": 32607
    },
    {
      "epoch": 3.109645843505859e-05,
      "step": 32607,
      "training_step_time": 6.677505970001221
    },
    {
      "epoch": 3.1097412109375e-05,
      "model_forward_time": 1.727792501449585,
      "step": 32608
    },
    {
      "epoch": 3.1097412109375e-05,
      "step": 32608,
      "training_step_time": 6.76435923576355
    },
    {
      "epoch": 3.109836578369141e-05,
      "model_forward_time": 1.7374122142791748,
      "step": 32609
    },
    {
      "epoch": 3.109836578369141e-05,
      "step": 32609,
      "training_step_time": 6.992685079574585
    },
    {
      "epoch": 3.109931945800781e-05,
      "grad_norm": 0.10585217922925949,
      "learning_rate": 8.056821105128908e-06,
      "loss": 0.0646,
      "step": 32610
    },
    {
      "epoch": 3.109931945800781e-05,
      "model_forward_time": 1.726083755493164,
      "step": 32610
    },
    {
      "dynamics_loss_avg": 0.043127551674842834,
      "epoch": 3.109931945800781e-05,
      "step": 32610
    },
    {
      "action_loss_avg": 0.0019319858867675066,
      "epoch": 3.109931945800781e-05,
      "step": 32610
    },
    {
      "epoch": 3.109931945800781e-05,
      "step": 32610,
      "training_step_time": 7.519690752029419
    },
    {
      "epoch": 3.1100273132324216e-05,
      "model_forward_time": 2.245086669921875,
      "step": 32611
    },
    {
      "epoch": 3.1100273132324216e-05,
      "step": 32611,
      "training_step_time": 7.608947992324829
    },
    {
      "epoch": 3.110122680664063e-05,
      "model_forward_time": 1.709031343460083,
      "step": 32612
    },
    {
      "epoch": 3.110122680664063e-05,
      "step": 32612,
      "training_step_time": 8.13215970993042
    },
    {
      "epoch": 3.110218048095703e-05,
      "model_forward_time": 1.742182731628418,
      "step": 32613
    },
    {
      "epoch": 3.110218048095703e-05,
      "step": 32613,
      "training_step_time": 7.7997753620147705
    },
    {
      "epoch": 3.1103134155273435e-05,
      "model_forward_time": 1.7184770107269287,
      "step": 32614
    },
    {
      "epoch": 3.1103134155273435e-05,
      "step": 32614,
      "training_step_time": 7.68008828163147
    },
    {
      "epoch": 3.1104087829589846e-05,
      "model_forward_time": 1.7214436531066895,
      "step": 32615
    },
    {
      "epoch": 3.1104087829589846e-05,
      "step": 32615,
      "training_step_time": 7.5904152393341064
    },
    {
      "epoch": 3.110504150390625e-05,
      "model_forward_time": 1.7342681884765625,
      "step": 32616
    },
    {
      "epoch": 3.110504150390625e-05,
      "step": 32616,
      "training_step_time": 7.389366388320923
    },
    {
      "epoch": 3.1105995178222654e-05,
      "model_forward_time": 1.751692771911621,
      "step": 32617
    },
    {
      "epoch": 3.1105995178222654e-05,
      "step": 32617,
      "training_step_time": 6.650062084197998
    },
    {
      "epoch": 3.1106948852539065e-05,
      "model_forward_time": 1.7349603176116943,
      "step": 32618
    },
    {
      "epoch": 3.1106948852539065e-05,
      "step": 32618,
      "training_step_time": 6.989100217819214
    },
    {
      "epoch": 3.110790252685547e-05,
      "model_forward_time": 1.736100435256958,
      "step": 32619
    },
    {
      "epoch": 3.110790252685547e-05,
      "step": 32619,
      "training_step_time": 7.535011053085327
    },
    {
      "epoch": 3.110885620117187e-05,
      "grad_norm": 0.17692787945270538,
      "learning_rate": 8.05551246574731e-06,
      "loss": 0.0633,
      "step": 32620
    },
    {
      "epoch": 3.110885620117187e-05,
      "model_forward_time": 2.2795603275299072,
      "step": 32620
    },
    {
      "dynamics_loss_avg": 0.06572251692414284,
      "epoch": 3.110885620117187e-05,
      "step": 32620
    },
    {
      "action_loss_avg": 0.001943395147100091,
      "epoch": 3.110885620117187e-05,
      "step": 32620
    },
    {
      "epoch": 3.110885620117187e-05,
      "step": 32620,
      "training_step_time": 7.7057788372039795
    },
    {
      "epoch": 3.110980987548828e-05,
      "model_forward_time": 1.7320878505706787,
      "step": 32621
    },
    {
      "epoch": 3.110980987548828e-05,
      "step": 32621,
      "training_step_time": 8.759829998016357
    },
    {
      "epoch": 3.111076354980469e-05,
      "model_forward_time": 1.735724925994873,
      "step": 32622
    },
    {
      "epoch": 3.111076354980469e-05,
      "step": 32622,
      "training_step_time": 10.671455383300781
    },
    {
      "epoch": 3.111171722412109e-05,
      "model_forward_time": 1.7350378036499023,
      "step": 32623
    },
    {
      "epoch": 3.111171722412109e-05,
      "step": 32623,
      "training_step_time": 8.70504379272461
    },
    {
      "epoch": 3.11126708984375e-05,
      "model_forward_time": 1.7368881702423096,
      "step": 32624
    },
    {
      "epoch": 3.11126708984375e-05,
      "step": 32624,
      "training_step_time": 7.852572441101074
    },
    {
      "epoch": 3.1113624572753906e-05,
      "model_forward_time": 1.7374773025512695,
      "step": 32625
    },
    {
      "epoch": 3.1113624572753906e-05,
      "step": 32625,
      "training_step_time": 7.537335157394409
    },
    {
      "epoch": 3.111457824707031e-05,
      "model_forward_time": 1.707620620727539,
      "step": 32626
    },
    {
      "epoch": 3.111457824707031e-05,
      "step": 32626,
      "training_step_time": 6.766206502914429
    },
    {
      "epoch": 3.111553192138672e-05,
      "model_forward_time": 1.7161636352539062,
      "step": 32627
    },
    {
      "epoch": 3.111553192138672e-05,
      "step": 32627,
      "training_step_time": 6.949235677719116
    },
    {
      "epoch": 3.1116485595703125e-05,
      "model_forward_time": 1.7205955982208252,
      "step": 32628
    },
    {
      "epoch": 3.1116485595703125e-05,
      "step": 32628,
      "training_step_time": 7.551544189453125
    },
    {
      "epoch": 3.111743927001953e-05,
      "model_forward_time": 2.293673038482666,
      "step": 32629
    },
    {
      "epoch": 3.111743927001953e-05,
      "step": 32629,
      "training_step_time": 8.482688188552856
    },
    {
      "epoch": 3.111839294433594e-05,
      "grad_norm": 0.12951752543449402,
      "learning_rate": 8.05420349221946e-06,
      "loss": 0.0633,
      "step": 32630
    },
    {
      "epoch": 3.111839294433594e-05,
      "model_forward_time": 1.720942497253418,
      "step": 32630
    },
    {
      "dynamics_loss_avg": 0.04539485201239586,
      "epoch": 3.111839294433594e-05,
      "step": 32630
    },
    {
      "action_loss_avg": 0.001407660823315382,
      "epoch": 3.111839294433594e-05,
      "step": 32630
    },
    {
      "epoch": 3.111839294433594e-05,
      "step": 32630,
      "training_step_time": 8.587737321853638
    },
    {
      "epoch": 3.1119346618652344e-05,
      "model_forward_time": 1.7156956195831299,
      "step": 32631
    },
    {
      "epoch": 3.1119346618652344e-05,
      "step": 32631,
      "training_step_time": 8.337964296340942
    },
    {
      "epoch": 3.112030029296875e-05,
      "model_forward_time": 1.7561242580413818,
      "step": 32632
    },
    {
      "epoch": 3.112030029296875e-05,
      "step": 32632,
      "training_step_time": 10.117858648300171
    },
    {
      "epoch": 3.112125396728516e-05,
      "model_forward_time": 1.8186259269714355,
      "step": 32633
    },
    {
      "epoch": 3.112125396728516e-05,
      "step": 32633,
      "training_step_time": 8.372262716293335
    },
    {
      "epoch": 3.112220764160156e-05,
      "model_forward_time": 1.7701904773712158,
      "step": 32634
    },
    {
      "epoch": 3.112220764160156e-05,
      "step": 32634,
      "training_step_time": 7.571462392807007
    },
    {
      "epoch": 3.112316131591797e-05,
      "model_forward_time": 1.7366414070129395,
      "step": 32635
    },
    {
      "epoch": 3.112316131591797e-05,
      "step": 32635,
      "training_step_time": 7.014976739883423
    },
    {
      "epoch": 3.112411499023438e-05,
      "model_forward_time": 1.7311375141143799,
      "step": 32636
    },
    {
      "epoch": 3.112411499023438e-05,
      "step": 32636,
      "training_step_time": 6.971278429031372
    },
    {
      "epoch": 3.112506866455078e-05,
      "model_forward_time": 1.7256455421447754,
      "step": 32637
    },
    {
      "epoch": 3.112506866455078e-05,
      "step": 32637,
      "training_step_time": 7.509321212768555
    },
    {
      "epoch": 3.1126022338867186e-05,
      "model_forward_time": 2.2928316593170166,
      "step": 32638
    },
    {
      "epoch": 3.1126022338867186e-05,
      "step": 32638,
      "training_step_time": 7.633566379547119
    },
    {
      "epoch": 3.1126976013183597e-05,
      "model_forward_time": 1.7261197566986084,
      "step": 32639
    },
    {
      "epoch": 3.1126976013183597e-05,
      "step": 32639,
      "training_step_time": 8.44339656829834
    },
    {
      "epoch": 3.11279296875e-05,
      "grad_norm": 0.17587855458259583,
      "learning_rate": 8.05289418468851e-06,
      "loss": 0.0615,
      "step": 32640
    },
    {
      "epoch": 3.11279296875e-05,
      "model_forward_time": 1.724571943283081,
      "step": 32640
    },
    {
      "dynamics_loss_avg": 0.05084037538617849,
      "epoch": 3.11279296875e-05,
      "step": 32640
    },
    {
      "action_loss_avg": 0.001959090045420453,
      "epoch": 3.11279296875e-05,
      "step": 32640
    },
    {
      "epoch": 3.11279296875e-05,
      "step": 32640,
      "training_step_time": 8.483999490737915
    },
    {
      "epoch": 3.1128883361816405e-05,
      "model_forward_time": 1.3653466701507568,
      "step": 32641
    },
    {
      "epoch": 3.1128883361816405e-05,
      "step": 32641,
      "training_step_time": 8.511544466018677
    },
    {
      "epoch": 3.1129837036132815e-05,
      "model_forward_time": 1.790715217590332,
      "step": 32642
    },
    {
      "epoch": 3.1129837036132815e-05,
      "step": 32642,
      "training_step_time": 8.57348895072937
    },
    {
      "epoch": 3.113079071044922e-05,
      "model_forward_time": 1.7368512153625488,
      "step": 32643
    },
    {
      "epoch": 3.113079071044922e-05,
      "step": 32643,
      "training_step_time": 7.63883638381958
    },
    {
      "epoch": 3.1131744384765623e-05,
      "model_forward_time": 1.7288503646850586,
      "step": 32644
    },
    {
      "epoch": 3.1131744384765623e-05,
      "step": 32644,
      "training_step_time": 7.545130729675293
    },
    {
      "epoch": 3.1132698059082034e-05,
      "model_forward_time": 1.742084264755249,
      "step": 32645
    },
    {
      "epoch": 3.1132698059082034e-05,
      "step": 32645,
      "training_step_time": 7.146440029144287
    },
    {
      "epoch": 3.113365173339844e-05,
      "model_forward_time": 1.725217580795288,
      "step": 32646
    },
    {
      "epoch": 3.113365173339844e-05,
      "step": 32646,
      "training_step_time": 7.6805102825164795
    },
    {
      "epoch": 3.113460540771484e-05,
      "model_forward_time": 1.7336339950561523,
      "step": 32647
    },
    {
      "epoch": 3.113460540771484e-05,
      "step": 32647,
      "training_step_time": 7.712374210357666
    },
    {
      "epoch": 3.113555908203125e-05,
      "model_forward_time": 2.276989698410034,
      "step": 32648
    },
    {
      "epoch": 3.113555908203125e-05,
      "step": 32648,
      "training_step_time": 7.705129861831665
    },
    {
      "epoch": 3.113651275634766e-05,
      "model_forward_time": 1.7290740013122559,
      "step": 32649
    },
    {
      "epoch": 3.113651275634766e-05,
      "step": 32649,
      "training_step_time": 7.778711795806885
    },
    {
      "epoch": 3.113746643066406e-05,
      "grad_norm": 0.12122616171836853,
      "learning_rate": 8.051584543297641e-06,
      "loss": 0.0624,
      "step": 32650
    },
    {
      "epoch": 3.113746643066406e-05,
      "model_forward_time": 1.7275934219360352,
      "step": 32650
    },
    {
      "dynamics_loss_avg": 0.06404154393821955,
      "epoch": 3.113746643066406e-05,
      "step": 32650
    },
    {
      "action_loss_avg": 0.002504851561388932,
      "epoch": 3.113746643066406e-05,
      "step": 32650
    },
    {
      "epoch": 3.113746643066406e-05,
      "step": 32650,
      "training_step_time": 7.898295640945435
    },
    {
      "epoch": 3.113842010498047e-05,
      "model_forward_time": 1.7933216094970703,
      "step": 32651
    },
    {
      "epoch": 3.113842010498047e-05,
      "step": 32651,
      "training_step_time": 8.710355758666992
    },
    {
      "epoch": 3.1139373779296876e-05,
      "model_forward_time": 1.3655602931976318,
      "step": 32652
    },
    {
      "epoch": 3.1139373779296876e-05,
      "step": 32652,
      "training_step_time": 8.82501745223999
    },
    {
      "epoch": 3.114032745361328e-05,
      "model_forward_time": 1.353590488433838,
      "step": 32653
    },
    {
      "epoch": 3.114032745361328e-05,
      "step": 32653,
      "training_step_time": 7.513741970062256
    },
    {
      "epoch": 3.114128112792969e-05,
      "model_forward_time": 1.7352511882781982,
      "step": 32654
    },
    {
      "epoch": 3.114128112792969e-05,
      "step": 32654,
      "training_step_time": 7.160412073135376
    },
    {
      "epoch": 3.1142234802246095e-05,
      "model_forward_time": 1.7352328300476074,
      "step": 32655
    },
    {
      "epoch": 3.1142234802246095e-05,
      "step": 32655,
      "training_step_time": 7.6058759689331055
    },
    {
      "epoch": 3.11431884765625e-05,
      "model_forward_time": 1.7264654636383057,
      "step": 32656
    },
    {
      "epoch": 3.11431884765625e-05,
      "step": 32656,
      "training_step_time": 7.624358892440796
    },
    {
      "epoch": 3.11441421508789e-05,
      "model_forward_time": 2.2819931507110596,
      "step": 32657
    },
    {
      "epoch": 3.11441421508789e-05,
      "step": 32657,
      "training_step_time": 7.7163543701171875
    },
    {
      "epoch": 3.1145095825195314e-05,
      "model_forward_time": 1.7422420978546143,
      "step": 32658
    },
    {
      "epoch": 3.1145095825195314e-05,
      "step": 32658,
      "training_step_time": 8.58289361000061
    },
    {
      "epoch": 3.114604949951172e-05,
      "model_forward_time": 1.7348389625549316,
      "step": 32659
    },
    {
      "epoch": 3.114604949951172e-05,
      "step": 32659,
      "training_step_time": 8.71385145187378
    },
    {
      "epoch": 3.114700317382812e-05,
      "grad_norm": 0.14043718576431274,
      "learning_rate": 8.050274568190076e-06,
      "loss": 0.0567,
      "step": 32660
    },
    {
      "epoch": 3.114700317382812e-05,
      "model_forward_time": 1.717918872833252,
      "step": 32660
    },
    {
      "dynamics_loss_avg": 0.04734273264184594,
      "epoch": 3.114700317382812e-05,
      "step": 32660
    },
    {
      "action_loss_avg": 0.0015676593175157905,
      "epoch": 3.114700317382812e-05,
      "step": 32660
    },
    {
      "epoch": 3.114700317382812e-05,
      "step": 32660,
      "training_step_time": 7.868651866912842
    },
    {
      "epoch": 3.114795684814453e-05,
      "model_forward_time": 1.7124574184417725,
      "step": 32661
    },
    {
      "epoch": 3.114795684814453e-05,
      "step": 32661,
      "training_step_time": 8.332836151123047
    },
    {
      "epoch": 3.1148910522460937e-05,
      "model_forward_time": 1.7204022407531738,
      "step": 32662
    },
    {
      "epoch": 3.1148910522460937e-05,
      "step": 32662,
      "training_step_time": 7.427760601043701
    },
    {
      "epoch": 3.114986419677734e-05,
      "model_forward_time": 1.7256150245666504,
      "step": 32663
    },
    {
      "epoch": 3.114986419677734e-05,
      "step": 32663,
      "training_step_time": 6.9874427318573
    },
    {
      "epoch": 3.115081787109375e-05,
      "model_forward_time": 1.7158174514770508,
      "step": 32664
    },
    {
      "epoch": 3.115081787109375e-05,
      "step": 32664,
      "training_step_time": 7.283061265945435
    },
    {
      "epoch": 3.1151771545410155e-05,
      "model_forward_time": 1.7190136909484863,
      "step": 32665
    },
    {
      "epoch": 3.1151771545410155e-05,
      "step": 32665,
      "training_step_time": 7.697321891784668
    },
    {
      "epoch": 3.115272521972656e-05,
      "model_forward_time": 2.307356357574463,
      "step": 32666
    },
    {
      "epoch": 3.115272521972656e-05,
      "step": 32666,
      "training_step_time": 7.803622722625732
    },
    {
      "epoch": 3.115367889404297e-05,
      "model_forward_time": 1.7069766521453857,
      "step": 32667
    },
    {
      "epoch": 3.115367889404297e-05,
      "step": 32667,
      "training_step_time": 7.62412166595459
    },
    {
      "epoch": 3.1154632568359374e-05,
      "model_forward_time": 1.723759412765503,
      "step": 32668
    },
    {
      "epoch": 3.1154632568359374e-05,
      "step": 32668,
      "training_step_time": 8.43563175201416
    },
    {
      "epoch": 3.115558624267578e-05,
      "model_forward_time": 1.7597548961639404,
      "step": 32669
    },
    {
      "epoch": 3.115558624267578e-05,
      "step": 32669,
      "training_step_time": 8.819833040237427
    },
    {
      "epoch": 3.115653991699219e-05,
      "grad_norm": 0.14644041657447815,
      "learning_rate": 8.048964259509067e-06,
      "loss": 0.0588,
      "step": 32670
    },
    {
      "epoch": 3.115653991699219e-05,
      "model_forward_time": 1.7147908210754395,
      "step": 32670
    },
    {
      "dynamics_loss_avg": 0.06095930021256209,
      "epoch": 3.115653991699219e-05,
      "step": 32670
    },
    {
      "action_loss_avg": 0.0022460340289399026,
      "epoch": 3.115653991699219e-05,
      "step": 32670
    },
    {
      "epoch": 3.115653991699219e-05,
      "step": 32670,
      "training_step_time": 7.850707292556763
    },
    {
      "epoch": 3.115749359130859e-05,
      "model_forward_time": 1.7131319046020508,
      "step": 32671
    },
    {
      "epoch": 3.115749359130859e-05,
      "step": 32671,
      "training_step_time": 7.764698028564453
    },
    {
      "epoch": 3.1158447265625e-05,
      "model_forward_time": 1.7057874202728271,
      "step": 32672
    },
    {
      "epoch": 3.1158447265625e-05,
      "step": 32672,
      "training_step_time": 7.501992225646973
    },
    {
      "epoch": 3.115940093994141e-05,
      "model_forward_time": 1.3287429809570312,
      "step": 32673
    },
    {
      "epoch": 3.115940093994141e-05,
      "step": 32673,
      "training_step_time": 7.351678371429443
    },
    {
      "epoch": 3.116035461425781e-05,
      "model_forward_time": 1.713104248046875,
      "step": 32674
    },
    {
      "epoch": 3.116035461425781e-05,
      "step": 32674,
      "training_step_time": 7.741292715072632
    },
    {
      "epoch": 3.1161308288574216e-05,
      "model_forward_time": 2.2502293586730957,
      "step": 32675
    },
    {
      "epoch": 3.1161308288574216e-05,
      "step": 32675,
      "training_step_time": 8.276906728744507
    },
    {
      "epoch": 3.116226196289063e-05,
      "model_forward_time": 1.7217371463775635,
      "step": 32676
    },
    {
      "epoch": 3.116226196289063e-05,
      "step": 32676,
      "training_step_time": 8.598308801651001
    },
    {
      "epoch": 3.116321563720703e-05,
      "model_forward_time": 1.7238130569458008,
      "step": 32677
    },
    {
      "epoch": 3.116321563720703e-05,
      "step": 32677,
      "training_step_time": 9.13417649269104
    },
    {
      "epoch": 3.1164169311523435e-05,
      "model_forward_time": 1.7374682426452637,
      "step": 32678
    },
    {
      "epoch": 3.1164169311523435e-05,
      "step": 32678,
      "training_step_time": 8.705570697784424
    },
    {
      "epoch": 3.1165122985839846e-05,
      "model_forward_time": 1.7396814823150635,
      "step": 32679
    },
    {
      "epoch": 3.1165122985839846e-05,
      "step": 32679,
      "training_step_time": 8.626598119735718
    },
    {
      "epoch": 3.116607666015625e-05,
      "grad_norm": 0.12300917506217957,
      "learning_rate": 8.047653617397913e-06,
      "loss": 0.0582,
      "step": 32680
    },
    {
      "epoch": 3.116607666015625e-05,
      "model_forward_time": 1.7128477096557617,
      "step": 32680
    },
    {
      "dynamics_loss_avg": 0.06388058345764876,
      "epoch": 3.116607666015625e-05,
      "step": 32680
    },
    {
      "action_loss_avg": 0.0024944504111772403,
      "epoch": 3.116607666015625e-05,
      "step": 32680
    },
    {
      "epoch": 3.116607666015625e-05,
      "step": 32680,
      "training_step_time": 7.602350234985352
    },
    {
      "epoch": 3.1167030334472654e-05,
      "model_forward_time": 1.730125904083252,
      "step": 32681
    },
    {
      "epoch": 3.1167030334472654e-05,
      "step": 32681,
      "training_step_time": 7.490744113922119
    },
    {
      "epoch": 3.1167984008789065e-05,
      "model_forward_time": 1.705075979232788,
      "step": 32682
    },
    {
      "epoch": 3.1167984008789065e-05,
      "step": 32682,
      "training_step_time": 7.184420824050903
    },
    {
      "epoch": 3.116893768310547e-05,
      "model_forward_time": 1.710866928100586,
      "step": 32683
    },
    {
      "epoch": 3.116893768310547e-05,
      "step": 32683,
      "training_step_time": 7.649529457092285
    },
    {
      "epoch": 3.116989135742187e-05,
      "model_forward_time": 1.7028472423553467,
      "step": 32684
    },
    {
      "epoch": 3.116989135742187e-05,
      "step": 32684,
      "training_step_time": 7.807317495346069
    },
    {
      "epoch": 3.1170845031738283e-05,
      "model_forward_time": 2.2666845321655273,
      "step": 32685
    },
    {
      "epoch": 3.1170845031738283e-05,
      "step": 32685,
      "training_step_time": 8.183932781219482
    },
    {
      "epoch": 3.117179870605469e-05,
      "model_forward_time": 1.6983375549316406,
      "step": 32686
    },
    {
      "epoch": 3.117179870605469e-05,
      "step": 32686,
      "training_step_time": 7.78031063079834
    },
    {
      "epoch": 3.117275238037109e-05,
      "model_forward_time": 1.7407841682434082,
      "step": 32687
    },
    {
      "epoch": 3.117275238037109e-05,
      "step": 32687,
      "training_step_time": 8.543841123580933
    },
    {
      "epoch": 3.11737060546875e-05,
      "model_forward_time": 1.0039334297180176,
      "step": 32688
    },
    {
      "epoch": 3.11737060546875e-05,
      "step": 32688,
      "training_step_time": 8.56594443321228
    },
    {
      "epoch": 3.1174659729003906e-05,
      "model_forward_time": 1.71030592918396,
      "step": 32689
    },
    {
      "epoch": 3.1174659729003906e-05,
      "step": 32689,
      "training_step_time": 7.7476911544799805
    },
    {
      "epoch": 3.117561340332031e-05,
      "grad_norm": 0.16917085647583008,
      "learning_rate": 8.046342641999941e-06,
      "loss": 0.0624,
      "step": 32690
    },
    {
      "epoch": 3.117561340332031e-05,
      "model_forward_time": 1.7225134372711182,
      "step": 32690
    },
    {
      "dynamics_loss_avg": 0.07645645029842854,
      "epoch": 3.117561340332031e-05,
      "step": 32690
    },
    {
      "action_loss_avg": 0.003101109154522419,
      "epoch": 3.117561340332031e-05,
      "step": 32690
    },
    {
      "epoch": 3.117561340332031e-05,
      "step": 32690,
      "training_step_time": 7.470468044281006
    },
    {
      "epoch": 3.117656707763672e-05,
      "model_forward_time": 1.7255213260650635,
      "step": 32691
    },
    {
      "epoch": 3.117656707763672e-05,
      "step": 32691,
      "training_step_time": 7.265703916549683
    },
    {
      "epoch": 3.1177520751953125e-05,
      "model_forward_time": 1.722379446029663,
      "step": 32692
    },
    {
      "epoch": 3.1177520751953125e-05,
      "step": 32692,
      "training_step_time": 7.592739105224609
    },
    {
      "epoch": 3.117847442626953e-05,
      "model_forward_time": 1.7221019268035889,
      "step": 32693
    },
    {
      "epoch": 3.117847442626953e-05,
      "step": 32693,
      "training_step_time": 7.6282799243927
    },
    {
      "epoch": 3.117942810058594e-05,
      "model_forward_time": 2.2803406715393066,
      "step": 32694
    },
    {
      "epoch": 3.117942810058594e-05,
      "step": 32694,
      "training_step_time": 10.160582780838013
    },
    {
      "epoch": 3.1180381774902344e-05,
      "model_forward_time": 1.7274866104125977,
      "step": 32695
    },
    {
      "epoch": 3.1180381774902344e-05,
      "step": 32695,
      "training_step_time": 8.529747486114502
    },
    {
      "epoch": 3.118133544921875e-05,
      "model_forward_time": 1.7318480014801025,
      "step": 32696
    },
    {
      "epoch": 3.118133544921875e-05,
      "step": 32696,
      "training_step_time": 8.775198459625244
    },
    {
      "epoch": 3.118228912353516e-05,
      "model_forward_time": 1.731841802597046,
      "step": 32697
    },
    {
      "epoch": 3.118228912353516e-05,
      "step": 32697,
      "training_step_time": 8.416676759719849
    },
    {
      "epoch": 3.118324279785156e-05,
      "model_forward_time": 1.733508586883545,
      "step": 32698
    },
    {
      "epoch": 3.118324279785156e-05,
      "step": 32698,
      "training_step_time": 7.568948030471802
    },
    {
      "epoch": 3.118419647216797e-05,
      "model_forward_time": 1.7192249298095703,
      "step": 32699
    },
    {
      "epoch": 3.118419647216797e-05,
      "step": 32699,
      "training_step_time": 7.639088869094849
    },
    {
      "epoch": 3.118515014648438e-05,
      "grad_norm": 0.1415085643529892,
      "learning_rate": 8.045031333458518e-06,
      "loss": 0.0558,
      "step": 32700
    },
    {
      "epoch": 3.118515014648438e-05,
      "model_forward_time": 1.721405267715454,
      "step": 32700
    },
    {
      "dynamics_loss_avg": 0.051624369062483313,
      "epoch": 3.118515014648438e-05,
      "step": 32700
    },
    {
      "action_loss_avg": 0.0021494187007192523,
      "epoch": 3.118515014648438e-05,
      "step": 32700
    },
    {
      "epoch": 3.118515014648438e-05,
      "step": 32700,
      "training_step_time": 7.2231409549713135
    },
    {
      "epoch": 3.118610382080078e-05,
      "model_forward_time": 1.7255525588989258,
      "step": 32701
    },
    {
      "epoch": 3.118610382080078e-05,
      "step": 32701,
      "training_step_time": 8.483298540115356
    },
    {
      "epoch": 3.1187057495117186e-05,
      "model_forward_time": 1.7193660736083984,
      "step": 32702
    },
    {
      "epoch": 3.1187057495117186e-05,
      "step": 32702,
      "training_step_time": 7.538454055786133
    },
    {
      "epoch": 3.1188011169433597e-05,
      "model_forward_time": 2.278125762939453,
      "step": 32703
    },
    {
      "epoch": 3.1188011169433597e-05,
      "step": 32703,
      "training_step_time": 7.657506465911865
    },
    {
      "epoch": 3.118896484375e-05,
      "model_forward_time": 1.7158162593841553,
      "step": 32704
    },
    {
      "epoch": 3.118896484375e-05,
      "step": 32704,
      "training_step_time": 8.678982734680176
    },
    {
      "epoch": 3.1189918518066405e-05,
      "model_forward_time": 1.7258524894714355,
      "step": 32705
    },
    {
      "epoch": 3.1189918518066405e-05,
      "step": 32705,
      "training_step_time": 8.690437078475952
    },
    {
      "epoch": 3.1190872192382815e-05,
      "model_forward_time": 1.708860158920288,
      "step": 32706
    },
    {
      "epoch": 3.1190872192382815e-05,
      "step": 32706,
      "training_step_time": 7.754566669464111
    },
    {
      "epoch": 3.119182586669922e-05,
      "model_forward_time": 1.734178066253662,
      "step": 32707
    },
    {
      "epoch": 3.119182586669922e-05,
      "step": 32707,
      "training_step_time": 8.666527271270752
    },
    {
      "epoch": 3.1192779541015623e-05,
      "model_forward_time": 1.7150652408599854,
      "step": 32708
    },
    {
      "epoch": 3.1192779541015623e-05,
      "step": 32708,
      "training_step_time": 7.65720534324646
    },
    {
      "epoch": 3.1193733215332034e-05,
      "model_forward_time": 1.718437910079956,
      "step": 32709
    },
    {
      "epoch": 3.1193733215332034e-05,
      "step": 32709,
      "training_step_time": 7.466764450073242
    },
    {
      "epoch": 3.119468688964844e-05,
      "grad_norm": 0.1099279448390007,
      "learning_rate": 8.043719691917046e-06,
      "loss": 0.0596,
      "step": 32710
    },
    {
      "epoch": 3.119468688964844e-05,
      "model_forward_time": 1.7108168601989746,
      "step": 32710
    },
    {
      "dynamics_loss_avg": 0.048822729662060735,
      "epoch": 3.119468688964844e-05,
      "step": 32710
    },
    {
      "action_loss_avg": 0.002206352853681892,
      "epoch": 3.119468688964844e-05,
      "step": 32710
    },
    {
      "epoch": 3.119468688964844e-05,
      "step": 32710,
      "training_step_time": 7.531001091003418
    },
    {
      "epoch": 3.119564056396484e-05,
      "model_forward_time": 1.7254095077514648,
      "step": 32711
    },
    {
      "epoch": 3.119564056396484e-05,
      "step": 32711,
      "training_step_time": 7.694026231765747
    },
    {
      "epoch": 3.119659423828125e-05,
      "model_forward_time": 1.7187526226043701,
      "step": 32712
    },
    {
      "epoch": 3.119659423828125e-05,
      "step": 32712,
      "training_step_time": 7.725663423538208
    },
    {
      "epoch": 3.119754791259766e-05,
      "model_forward_time": 2.2657902240753174,
      "step": 32713
    },
    {
      "epoch": 3.119754791259766e-05,
      "step": 32713,
      "training_step_time": 7.619994878768921
    },
    {
      "epoch": 3.119850158691406e-05,
      "model_forward_time": 1.7323799133300781,
      "step": 32714
    },
    {
      "epoch": 3.119850158691406e-05,
      "step": 32714,
      "training_step_time": 8.54798173904419
    },
    {
      "epoch": 3.119945526123047e-05,
      "model_forward_time": 1.3361778259277344,
      "step": 32715
    },
    {
      "epoch": 3.119945526123047e-05,
      "step": 32715,
      "training_step_time": 8.79570484161377
    },
    {
      "epoch": 3.1200408935546876e-05,
      "model_forward_time": 1.7326385974884033,
      "step": 32716
    },
    {
      "epoch": 3.1200408935546876e-05,
      "step": 32716,
      "training_step_time": 7.7500879764556885
    },
    {
      "epoch": 3.120136260986328e-05,
      "model_forward_time": 1.7199478149414062,
      "step": 32717
    },
    {
      "epoch": 3.120136260986328e-05,
      "step": 32717,
      "training_step_time": 7.667180776596069
    },
    {
      "epoch": 3.120231628417969e-05,
      "model_forward_time": 1.7128410339355469,
      "step": 32718
    },
    {
      "epoch": 3.120231628417969e-05,
      "step": 32718,
      "training_step_time": 7.5038368701934814
    },
    {
      "epoch": 3.1203269958496095e-05,
      "model_forward_time": 1.7212111949920654,
      "step": 32719
    },
    {
      "epoch": 3.1203269958496095e-05,
      "step": 32719,
      "training_step_time": 7.560223817825317
    },
    {
      "epoch": 3.12042236328125e-05,
      "grad_norm": 0.16352114081382751,
      "learning_rate": 8.042407717518967e-06,
      "loss": 0.0596,
      "step": 32720
    },
    {
      "epoch": 3.12042236328125e-05,
      "model_forward_time": 1.7172210216522217,
      "step": 32720
    },
    {
      "dynamics_loss_avg": 0.05280768685042858,
      "epoch": 3.12042236328125e-05,
      "step": 32720
    },
    {
      "action_loss_avg": 0.002001926820958033,
      "epoch": 3.12042236328125e-05,
      "step": 32720
    },
    {
      "epoch": 3.12042236328125e-05,
      "step": 32720,
      "training_step_time": 9.340744972229004
    },
    {
      "epoch": 3.12051773071289e-05,
      "model_forward_time": 1.716435432434082,
      "step": 32721
    },
    {
      "epoch": 3.12051773071289e-05,
      "step": 32721,
      "training_step_time": 8.262486934661865
    },
    {
      "epoch": 3.1206130981445314e-05,
      "model_forward_time": 2.2755815982818604,
      "step": 32722
    },
    {
      "epoch": 3.1206130981445314e-05,
      "step": 32722,
      "training_step_time": 7.607827425003052
    },
    {
      "epoch": 3.120708465576172e-05,
      "model_forward_time": 1.7187447547912598,
      "step": 32723
    },
    {
      "epoch": 3.120708465576172e-05,
      "step": 32723,
      "training_step_time": 8.513325929641724
    },
    {
      "epoch": 3.120803833007812e-05,
      "model_forward_time": 1.7129480838775635,
      "step": 32724
    },
    {
      "epoch": 3.120803833007812e-05,
      "step": 32724,
      "training_step_time": 8.430943489074707
    },
    {
      "epoch": 3.120899200439453e-05,
      "model_forward_time": 1.7347664833068848,
      "step": 32725
    },
    {
      "epoch": 3.120899200439453e-05,
      "step": 32725,
      "training_step_time": 8.54229474067688
    },
    {
      "epoch": 3.120994567871094e-05,
      "model_forward_time": 1.7099616527557373,
      "step": 32726
    },
    {
      "epoch": 3.120994567871094e-05,
      "step": 32726,
      "training_step_time": 7.771935701370239
    },
    {
      "epoch": 3.121089935302734e-05,
      "model_forward_time": 1.712740421295166,
      "step": 32727
    },
    {
      "epoch": 3.121089935302734e-05,
      "step": 32727,
      "training_step_time": 7.596353530883789
    },
    {
      "epoch": 3.121185302734375e-05,
      "model_forward_time": 1.714045763015747,
      "step": 32728
    },
    {
      "epoch": 3.121185302734375e-05,
      "step": 32728,
      "training_step_time": 7.5558836460113525
    },
    {
      "epoch": 3.1212806701660155e-05,
      "model_forward_time": 1.3359670639038086,
      "step": 32729
    },
    {
      "epoch": 3.1212806701660155e-05,
      "step": 32729,
      "training_step_time": 8.328788757324219
    },
    {
      "epoch": 3.121376037597656e-05,
      "grad_norm": 0.16953475773334503,
      "learning_rate": 8.04109541040775e-06,
      "loss": 0.0599,
      "step": 32730
    },
    {
      "epoch": 3.121376037597656e-05,
      "model_forward_time": 1.709315299987793,
      "step": 32730
    },
    {
      "dynamics_loss_avg": 0.05218819249421358,
      "epoch": 3.121376037597656e-05,
      "step": 32730
    },
    {
      "action_loss_avg": 0.0014065661467611789,
      "epoch": 3.121376037597656e-05,
      "step": 32730
    },
    {
      "epoch": 3.121376037597656e-05,
      "step": 32730,
      "training_step_time": 7.789319276809692
    },
    {
      "epoch": 3.121471405029297e-05,
      "model_forward_time": 2.2745704650878906,
      "step": 32731
    },
    {
      "epoch": 3.121471405029297e-05,
      "step": 32731,
      "training_step_time": 7.638233184814453
    },
    {
      "epoch": 3.1215667724609374e-05,
      "model_forward_time": 1.7121286392211914,
      "step": 32732
    },
    {
      "epoch": 3.1215667724609374e-05,
      "step": 32732,
      "training_step_time": 9.815762281417847
    },
    {
      "epoch": 3.121662139892578e-05,
      "model_forward_time": 1.7151765823364258,
      "step": 32733
    },
    {
      "epoch": 3.121662139892578e-05,
      "step": 32733,
      "training_step_time": 8.42212700843811
    },
    {
      "epoch": 3.121757507324219e-05,
      "model_forward_time": 1.7204394340515137,
      "step": 32734
    },
    {
      "epoch": 3.121757507324219e-05,
      "step": 32734,
      "training_step_time": 8.932437419891357
    },
    {
      "epoch": 3.121852874755859e-05,
      "model_forward_time": 1.7181322574615479,
      "step": 32735
    },
    {
      "epoch": 3.121852874755859e-05,
      "step": 32735,
      "training_step_time": 7.756374835968018
    },
    {
      "epoch": 3.1219482421875e-05,
      "model_forward_time": 1.7158212661743164,
      "step": 32736
    },
    {
      "epoch": 3.1219482421875e-05,
      "step": 32736,
      "training_step_time": 7.721389293670654
    },
    {
      "epoch": 3.122043609619141e-05,
      "model_forward_time": 1.7189040184020996,
      "step": 32737
    },
    {
      "epoch": 3.122043609619141e-05,
      "step": 32737,
      "training_step_time": 7.589243650436401
    },
    {
      "epoch": 3.122138977050781e-05,
      "model_forward_time": 1.718989372253418,
      "step": 32738
    },
    {
      "epoch": 3.122138977050781e-05,
      "step": 32738,
      "training_step_time": 7.607890844345093
    },
    {
      "epoch": 3.1222343444824216e-05,
      "model_forward_time": 1.7126338481903076,
      "step": 32739
    },
    {
      "epoch": 3.1222343444824216e-05,
      "step": 32739,
      "training_step_time": 8.147547483444214
    },
    {
      "epoch": 3.122329711914063e-05,
      "grad_norm": 0.13843907415866852,
      "learning_rate": 8.039782770726913e-06,
      "loss": 0.0592,
      "step": 32740
    },
    {
      "epoch": 3.122329711914063e-05,
      "model_forward_time": 2.2927358150482178,
      "step": 32740
    },
    {
      "dynamics_loss_avg": 0.05036383531987667,
      "epoch": 3.122329711914063e-05,
      "step": 32740
    },
    {
      "action_loss_avg": 0.0020908573002088814,
      "epoch": 3.122329711914063e-05,
      "step": 32740
    },
    {
      "epoch": 3.122329711914063e-05,
      "step": 32740,
      "training_step_time": 8.336236715316772
    },
    {
      "epoch": 3.122425079345703e-05,
      "model_forward_time": 1.72086501121521,
      "step": 32741
    },
    {
      "epoch": 3.122425079345703e-05,
      "step": 32741,
      "training_step_time": 8.438222408294678
    },
    {
      "epoch": 3.1225204467773435e-05,
      "model_forward_time": 1.7286207675933838,
      "step": 32742
    },
    {
      "epoch": 3.1225204467773435e-05,
      "step": 32742,
      "training_step_time": 7.879093885421753
    },
    {
      "epoch": 3.1226158142089846e-05,
      "model_forward_time": 1.7175061702728271,
      "step": 32743
    },
    {
      "epoch": 3.1226158142089846e-05,
      "step": 32743,
      "training_step_time": 8.822056770324707
    },
    {
      "epoch": 3.122711181640625e-05,
      "model_forward_time": 1.7221546173095703,
      "step": 32744
    },
    {
      "epoch": 3.122711181640625e-05,
      "step": 32744,
      "training_step_time": 9.477554559707642
    },
    {
      "epoch": 3.1228065490722654e-05,
      "model_forward_time": 1.7119524478912354,
      "step": 32745
    },
    {
      "epoch": 3.1228065490722654e-05,
      "step": 32745,
      "training_step_time": 8.624318599700928
    },
    {
      "epoch": 3.1229019165039065e-05,
      "model_forward_time": 1.7131688594818115,
      "step": 32746
    },
    {
      "epoch": 3.1229019165039065e-05,
      "step": 32746,
      "training_step_time": 7.597998857498169
    },
    {
      "epoch": 3.122997283935547e-05,
      "model_forward_time": 1.7203190326690674,
      "step": 32747
    },
    {
      "epoch": 3.122997283935547e-05,
      "step": 32747,
      "training_step_time": 7.634239435195923
    },
    {
      "epoch": 3.123092651367187e-05,
      "model_forward_time": 1.7085118293762207,
      "step": 32748
    },
    {
      "epoch": 3.123092651367187e-05,
      "step": 32748,
      "training_step_time": 8.674606323242188
    },
    {
      "epoch": 3.1231880187988283e-05,
      "model_forward_time": 2.2609076499938965,
      "step": 32749
    },
    {
      "epoch": 3.1231880187988283e-05,
      "step": 32749,
      "training_step_time": 9.308389902114868
    },
    {
      "epoch": 3.123283386230469e-05,
      "grad_norm": 0.12137054651975632,
      "learning_rate": 8.038469798620005e-06,
      "loss": 0.0575,
      "step": 32750
    },
    {
      "epoch": 3.123283386230469e-05,
      "model_forward_time": 1.7146670818328857,
      "step": 32750
    },
    {
      "dynamics_loss_avg": 0.05235447622835636,
      "epoch": 3.123283386230469e-05,
      "step": 32750
    },
    {
      "action_loss_avg": 0.003298877566703595,
      "epoch": 3.123283386230469e-05,
      "step": 32750
    },
    {
      "epoch": 3.123283386230469e-05,
      "step": 32750,
      "training_step_time": 7.752951622009277
    },
    {
      "epoch": 3.123378753662109e-05,
      "model_forward_time": 1.7356817722320557,
      "step": 32751
    },
    {
      "epoch": 3.123378753662109e-05,
      "step": 32751,
      "training_step_time": 8.183563232421875
    },
    {
      "epoch": 3.12347412109375e-05,
      "model_forward_time": 1.7164161205291748,
      "step": 32752
    },
    {
      "epoch": 3.12347412109375e-05,
      "step": 32752,
      "training_step_time": 8.372116327285767
    },
    {
      "epoch": 3.1235694885253906e-05,
      "model_forward_time": 1.7242207527160645,
      "step": 32753
    },
    {
      "epoch": 3.1235694885253906e-05,
      "step": 32753,
      "training_step_time": 7.774888753890991
    },
    {
      "epoch": 3.123664855957031e-05,
      "model_forward_time": 1.7359042167663574,
      "step": 32754
    },
    {
      "epoch": 3.123664855957031e-05,
      "step": 32754,
      "training_step_time": 7.73758602142334
    },
    {
      "epoch": 3.123760223388672e-05,
      "model_forward_time": 1.7310621738433838,
      "step": 32755
    },
    {
      "epoch": 3.123760223388672e-05,
      "step": 32755,
      "training_step_time": 7.514883518218994
    },
    {
      "epoch": 3.1238555908203125e-05,
      "model_forward_time": 1.7269890308380127,
      "step": 32756
    },
    {
      "epoch": 3.1238555908203125e-05,
      "step": 32756,
      "training_step_time": 7.627183198928833
    },
    {
      "epoch": 3.123950958251953e-05,
      "model_forward_time": 1.7224006652832031,
      "step": 32757
    },
    {
      "epoch": 3.123950958251953e-05,
      "step": 32757,
      "training_step_time": 7.602913856506348
    },
    {
      "epoch": 3.124046325683594e-05,
      "model_forward_time": 1.7350187301635742,
      "step": 32758
    },
    {
      "epoch": 3.124046325683594e-05,
      "step": 32758,
      "training_step_time": 7.814850091934204
    },
    {
      "epoch": 3.1241416931152344e-05,
      "model_forward_time": 2.261215925216675,
      "step": 32759
    },
    {
      "epoch": 3.1241416931152344e-05,
      "step": 32759,
      "training_step_time": 7.552751064300537
    },
    {
      "epoch": 3.124237060546875e-05,
      "grad_norm": 0.12015645205974579,
      "learning_rate": 8.037156494230605e-06,
      "loss": 0.0578,
      "step": 32760
    },
    {
      "epoch": 3.124237060546875e-05,
      "model_forward_time": 1.7357845306396484,
      "step": 32760
    },
    {
      "dynamics_loss_avg": 0.061454433016479015,
      "epoch": 3.124237060546875e-05,
      "step": 32760
    },
    {
      "action_loss_avg": 0.002428562939167023,
      "epoch": 3.124237060546875e-05,
      "step": 32760
    },
    {
      "epoch": 3.124237060546875e-05,
      "step": 32760,
      "training_step_time": 8.595922946929932
    },
    {
      "epoch": 3.124332427978516e-05,
      "model_forward_time": 1.7302219867706299,
      "step": 32761
    },
    {
      "epoch": 3.124332427978516e-05,
      "step": 32761,
      "training_step_time": 7.82321572303772
    },
    {
      "epoch": 3.124427795410156e-05,
      "model_forward_time": 1.7343244552612305,
      "step": 32762
    },
    {
      "epoch": 3.124427795410156e-05,
      "step": 32762,
      "training_step_time": 8.204672813415527
    },
    {
      "epoch": 3.124523162841797e-05,
      "model_forward_time": 1.7265822887420654,
      "step": 32763
    },
    {
      "epoch": 3.124523162841797e-05,
      "step": 32763,
      "training_step_time": 7.636119365692139
    },
    {
      "epoch": 3.124618530273438e-05,
      "model_forward_time": 1.7330987453460693,
      "step": 32764
    },
    {
      "epoch": 3.124618530273438e-05,
      "step": 32764,
      "training_step_time": 7.602127313613892
    },
    {
      "epoch": 3.124713897705078e-05,
      "model_forward_time": 1.7194275856018066,
      "step": 32765
    },
    {
      "epoch": 3.124713897705078e-05,
      "step": 32765,
      "training_step_time": 6.656594514846802
    },
    {
      "epoch": 3.1248092651367186e-05,
      "model_forward_time": 1.7128314971923828,
      "step": 32766
    },
    {
      "epoch": 3.1248092651367186e-05,
      "step": 32766,
      "training_step_time": 7.730574369430542
    },
    {
      "epoch": 3.12490463256836e-05,
      "model_forward_time": 1.7245843410491943,
      "step": 32767
    },
    {
      "epoch": 3.12490463256836e-05,
      "step": 32767,
      "training_step_time": 7.760091304779053
    },
    {
      "epoch": 3.125e-05,
      "model_forward_time": 2.2873482704162598,
      "step": 32768
    },
    {
      "epoch": 3.125e-05,
      "step": 32768,
      "training_step_time": 8.62872862815857
    },
    {
      "epoch": 3.1250953674316405e-05,
      "model_forward_time": 1.7618110179901123,
      "step": 32769
    },
    {
      "epoch": 3.1250953674316405e-05,
      "step": 32769,
      "training_step_time": 9.23237133026123
    },
    {
      "epoch": 3.1251907348632815e-05,
      "grad_norm": 0.13157978653907776,
      "learning_rate": 8.035842857702339e-06,
      "loss": 0.0582,
      "step": 32770
    },
    {
      "epoch": 3.1251907348632815e-05,
      "model_forward_time": 1.7631309032440186,
      "step": 32770
    },
    {
      "dynamics_loss_avg": 0.05363933742046356,
      "epoch": 3.1251907348632815e-05,
      "step": 32770
    },
    {
      "action_loss_avg": 0.0018175444973167032,
      "epoch": 3.1251907348632815e-05,
      "step": 32770
    },
    {
      "epoch": 3.1251907348632815e-05,
      "step": 32770,
      "training_step_time": 10.317551136016846
    },
    {
      "epoch": 3.125286102294922e-05,
      "model_forward_time": 1.7514019012451172,
      "step": 32771
    },
    {
      "epoch": 3.125286102294922e-05,
      "step": 32771,
      "training_step_time": 8.714205741882324
    },
    {
      "epoch": 3.1253814697265624e-05,
      "model_forward_time": 1.7560093402862549,
      "step": 32772
    },
    {
      "epoch": 3.1253814697265624e-05,
      "step": 32772,
      "training_step_time": 7.64741063117981
    },
    {
      "epoch": 3.1254768371582034e-05,
      "model_forward_time": 1.7423319816589355,
      "step": 32773
    },
    {
      "epoch": 3.1254768371582034e-05,
      "step": 32773,
      "training_step_time": 7.689798593521118
    },
    {
      "epoch": 3.125572204589844e-05,
      "model_forward_time": 1.750429391860962,
      "step": 32774
    },
    {
      "epoch": 3.125572204589844e-05,
      "step": 32774,
      "training_step_time": 6.721123695373535
    },
    {
      "epoch": 3.125667572021484e-05,
      "model_forward_time": 1.8607027530670166,
      "step": 32775
    },
    {
      "epoch": 3.125667572021484e-05,
      "step": 32775,
      "training_step_time": 7.605940103530884
    },
    {
      "epoch": 3.125762939453125e-05,
      "model_forward_time": 1.7556085586547852,
      "step": 32776
    },
    {
      "epoch": 3.125762939453125e-05,
      "step": 32776,
      "training_step_time": 7.724283218383789
    },
    {
      "epoch": 3.125858306884766e-05,
      "model_forward_time": 2.8923721313476562,
      "step": 32777
    },
    {
      "epoch": 3.125858306884766e-05,
      "step": 32777,
      "training_step_time": 9.159122228622437
    },
    {
      "epoch": 3.125953674316406e-05,
      "model_forward_time": 1.7377984523773193,
      "step": 32778
    },
    {
      "epoch": 3.125953674316406e-05,
      "step": 32778,
      "training_step_time": 9.56308650970459
    },
    {
      "epoch": 3.126049041748047e-05,
      "model_forward_time": 1.7442207336425781,
      "step": 32779
    },
    {
      "epoch": 3.126049041748047e-05,
      "step": 32779,
      "training_step_time": 8.324844598770142
    },
    {
      "epoch": 3.1261444091796876e-05,
      "grad_norm": 0.13232165575027466,
      "learning_rate": 8.034528889178861e-06,
      "loss": 0.0563,
      "step": 32780
    },
    {
      "epoch": 3.1261444091796876e-05,
      "model_forward_time": 1.7547855377197266,
      "step": 32780
    },
    {
      "dynamics_loss_avg": 0.045010604336857794,
      "epoch": 3.1261444091796876e-05,
      "step": 32780
    },
    {
      "action_loss_avg": 0.001836946012917906,
      "epoch": 3.1261444091796876e-05,
      "step": 32780
    },
    {
      "epoch": 3.1261444091796876e-05,
      "step": 32780,
      "training_step_time": 8.79819393157959
    },
    {
      "epoch": 3.126239776611328e-05,
      "model_forward_time": 1.7516295909881592,
      "step": 32781
    },
    {
      "epoch": 3.126239776611328e-05,
      "step": 32781,
      "training_step_time": 7.660327434539795
    },
    {
      "epoch": 3.1263351440429684e-05,
      "model_forward_time": 1.76076340675354,
      "step": 32782
    },
    {
      "epoch": 3.1263351440429684e-05,
      "step": 32782,
      "training_step_time": 8.657947540283203
    },
    {
      "epoch": 3.1264305114746095e-05,
      "model_forward_time": 1.750373363494873,
      "step": 32783
    },
    {
      "epoch": 3.1264305114746095e-05,
      "step": 32783,
      "training_step_time": 6.694430828094482
    },
    {
      "epoch": 3.12652587890625e-05,
      "model_forward_time": 1.838322401046753,
      "step": 32784
    },
    {
      "epoch": 3.12652587890625e-05,
      "step": 32784,
      "training_step_time": 7.550511360168457
    },
    {
      "epoch": 3.12662124633789e-05,
      "model_forward_time": 1.759002923965454,
      "step": 32785
    },
    {
      "epoch": 3.12662124633789e-05,
      "step": 32785,
      "training_step_time": 7.6211934089660645
    },
    {
      "epoch": 3.1267166137695314e-05,
      "model_forward_time": 2.858698606491089,
      "step": 32786
    },
    {
      "epoch": 3.1267166137695314e-05,
      "step": 32786,
      "training_step_time": 8.609669923782349
    },
    {
      "epoch": 3.126811981201172e-05,
      "model_forward_time": 1.7516465187072754,
      "step": 32787
    },
    {
      "epoch": 3.126811981201172e-05,
      "step": 32787,
      "training_step_time": 8.483819484710693
    },
    {
      "epoch": 3.126907348632812e-05,
      "model_forward_time": 1.7762665748596191,
      "step": 32788
    },
    {
      "epoch": 3.126907348632812e-05,
      "step": 32788,
      "training_step_time": 7.639435291290283
    },
    {
      "epoch": 3.127002716064453e-05,
      "model_forward_time": 1.7498815059661865,
      "step": 32789
    },
    {
      "epoch": 3.127002716064453e-05,
      "step": 32789,
      "training_step_time": 7.8553307056427
    },
    {
      "epoch": 3.127098083496094e-05,
      "grad_norm": 0.1540694683790207,
      "learning_rate": 8.033214588803865e-06,
      "loss": 0.0602,
      "step": 32790
    },
    {
      "epoch": 3.127098083496094e-05,
      "model_forward_time": 1.7394919395446777,
      "step": 32790
    },
    {
      "dynamics_loss_avg": 0.060090622305870055,
      "epoch": 3.127098083496094e-05,
      "step": 32790
    },
    {
      "action_loss_avg": 0.0020924226351780816,
      "epoch": 3.127098083496094e-05,
      "step": 32790
    },
    {
      "epoch": 3.127098083496094e-05,
      "step": 32790,
      "training_step_time": 7.670117616653442
    },
    {
      "epoch": 3.127193450927734e-05,
      "model_forward_time": 1.7629272937774658,
      "step": 32791
    },
    {
      "epoch": 3.127193450927734e-05,
      "step": 32791,
      "training_step_time": 7.642867803573608
    },
    {
      "epoch": 3.127288818359375e-05,
      "model_forward_time": 1.7558259963989258,
      "step": 32792
    },
    {
      "epoch": 3.127288818359375e-05,
      "step": 32792,
      "training_step_time": 7.606968641281128
    },
    {
      "epoch": 3.1273841857910156e-05,
      "model_forward_time": 1.7994630336761475,
      "step": 32793
    },
    {
      "epoch": 3.1273841857910156e-05,
      "step": 32793,
      "training_step_time": 7.585979700088501
    },
    {
      "epoch": 3.127479553222656e-05,
      "model_forward_time": 1.747781753540039,
      "step": 32794
    },
    {
      "epoch": 3.127479553222656e-05,
      "step": 32794,
      "training_step_time": 7.646936655044556
    },
    {
      "epoch": 3.127574920654297e-05,
      "model_forward_time": 2.8715598583221436,
      "step": 32795
    },
    {
      "epoch": 3.127574920654297e-05,
      "step": 32795,
      "training_step_time": 8.357338190078735
    },
    {
      "epoch": 3.1276702880859374e-05,
      "model_forward_time": 1.760180950164795,
      "step": 32796
    },
    {
      "epoch": 3.1276702880859374e-05,
      "step": 32796,
      "training_step_time": 8.20751428604126
    },
    {
      "epoch": 3.127765655517578e-05,
      "model_forward_time": 1.7539262771606445,
      "step": 32797
    },
    {
      "epoch": 3.127765655517578e-05,
      "step": 32797,
      "training_step_time": 7.677952766418457
    },
    {
      "epoch": 3.127861022949219e-05,
      "model_forward_time": 1.748889684677124,
      "step": 32798
    },
    {
      "epoch": 3.127861022949219e-05,
      "step": 32798,
      "training_step_time": 7.702277898788452
    },
    {
      "epoch": 3.127956390380859e-05,
      "model_forward_time": 1.7571601867675781,
      "step": 32799
    },
    {
      "epoch": 3.127956390380859e-05,
      "step": 32799,
      "training_step_time": 9.918091535568237
    },
    {
      "epoch": 3.1280517578125e-05,
      "grad_norm": 0.1389998197555542,
      "learning_rate": 8.031899956721084e-06,
      "loss": 0.0614,
      "step": 32800
    },
    {
      "epoch": 3.1280517578125e-05,
      "model_forward_time": 1.7403366565704346,
      "step": 32800
    },
    {
      "dynamics_loss_avg": 0.05106842890381813,
      "epoch": 3.1280517578125e-05,
      "step": 32800
    },
    {
      "action_loss_avg": 0.001993791660061106,
      "epoch": 3.1280517578125e-05,
      "step": 32800
    },
    {
      "epoch": 3.1280517578125e-05,
      "step": 32800,
      "training_step_time": 7.541185617446899
    },
    {
      "epoch": 3.128147125244141e-05,
      "model_forward_time": 1.7647485733032227,
      "step": 32801
    },
    {
      "epoch": 3.128147125244141e-05,
      "step": 32801,
      "training_step_time": 7.619820594787598
    },
    {
      "epoch": 3.128242492675781e-05,
      "model_forward_time": 1.7732503414154053,
      "step": 32802
    },
    {
      "epoch": 3.128242492675781e-05,
      "step": 32802,
      "training_step_time": 8.01056981086731
    },
    {
      "epoch": 3.1283378601074216e-05,
      "model_forward_time": 1.7538366317749023,
      "step": 32803
    },
    {
      "epoch": 3.1283378601074216e-05,
      "step": 32803,
      "training_step_time": 8.460691690444946
    },
    {
      "epoch": 3.128433227539063e-05,
      "model_forward_time": 1.7526915073394775,
      "step": 32804
    },
    {
      "epoch": 3.128433227539063e-05,
      "step": 32804,
      "training_step_time": 8.13601565361023
    },
    {
      "epoch": 3.128528594970703e-05,
      "model_forward_time": 1.77042555809021,
      "step": 32805
    },
    {
      "epoch": 3.128528594970703e-05,
      "step": 32805,
      "training_step_time": 8.121799945831299
    },
    {
      "epoch": 3.1286239624023435e-05,
      "model_forward_time": 1.75724458694458,
      "step": 32806
    },
    {
      "epoch": 3.1286239624023435e-05,
      "step": 32806,
      "training_step_time": 7.59728741645813
    },
    {
      "epoch": 3.1287193298339846e-05,
      "model_forward_time": 1.7520685195922852,
      "step": 32807
    },
    {
      "epoch": 3.1287193298339846e-05,
      "step": 32807,
      "training_step_time": 7.51051139831543
    },
    {
      "epoch": 3.128814697265625e-05,
      "model_forward_time": 1.7602612972259521,
      "step": 32808
    },
    {
      "epoch": 3.128814697265625e-05,
      "step": 32808,
      "training_step_time": 8.260589599609375
    },
    {
      "epoch": 3.1289100646972654e-05,
      "model_forward_time": 1.740274429321289,
      "step": 32809
    },
    {
      "epoch": 3.1289100646972654e-05,
      "step": 32809,
      "training_step_time": 7.535651206970215
    },
    {
      "epoch": 3.1290054321289065e-05,
      "grad_norm": 0.14310477674007416,
      "learning_rate": 8.030584993074282e-06,
      "loss": 0.0614,
      "step": 32810
    },
    {
      "epoch": 3.1290054321289065e-05,
      "model_forward_time": 1.7394771575927734,
      "step": 32810
    },
    {
      "dynamics_loss_avg": 0.05091874003410339,
      "epoch": 3.1290054321289065e-05,
      "step": 32810
    },
    {
      "action_loss_avg": 0.0023672538809478283,
      "epoch": 3.1290054321289065e-05,
      "step": 32810
    },
    {
      "epoch": 3.1290054321289065e-05,
      "step": 32810,
      "training_step_time": 7.765539169311523
    },
    {
      "epoch": 3.129100799560547e-05,
      "model_forward_time": 1.7653961181640625,
      "step": 32811
    },
    {
      "epoch": 3.129100799560547e-05,
      "step": 32811,
      "training_step_time": 7.576310634613037
    },
    {
      "epoch": 3.129196166992187e-05,
      "model_forward_time": 1.861353874206543,
      "step": 32812
    },
    {
      "epoch": 3.129196166992187e-05,
      "step": 32812,
      "training_step_time": 7.636492013931274
    },
    {
      "epoch": 3.1292915344238284e-05,
      "model_forward_time": 1.7535450458526611,
      "step": 32813
    },
    {
      "epoch": 3.1292915344238284e-05,
      "step": 32813,
      "training_step_time": 7.695233106613159
    },
    {
      "epoch": 3.129386901855469e-05,
      "model_forward_time": 2.887362480163574,
      "step": 32814
    },
    {
      "epoch": 3.129386901855469e-05,
      "step": 32814,
      "training_step_time": 7.652217149734497
    },
    {
      "epoch": 3.129482269287109e-05,
      "model_forward_time": 1.764007329940796,
      "step": 32815
    },
    {
      "epoch": 3.129482269287109e-05,
      "step": 32815,
      "training_step_time": 8.208247423171997
    },
    {
      "epoch": 3.12957763671875e-05,
      "model_forward_time": 1.741699457168579,
      "step": 32816
    },
    {
      "epoch": 3.12957763671875e-05,
      "step": 32816,
      "training_step_time": 7.58961296081543
    },
    {
      "epoch": 3.1296730041503906e-05,
      "model_forward_time": 1.7537930011749268,
      "step": 32817
    },
    {
      "epoch": 3.1296730041503906e-05,
      "step": 32817,
      "training_step_time": 7.788469552993774
    },
    {
      "epoch": 3.129768371582031e-05,
      "model_forward_time": 1.7379546165466309,
      "step": 32818
    },
    {
      "epoch": 3.129768371582031e-05,
      "step": 32818,
      "training_step_time": 8.621675729751587
    },
    {
      "epoch": 3.129863739013672e-05,
      "model_forward_time": 1.7387073040008545,
      "step": 32819
    },
    {
      "epoch": 3.129863739013672e-05,
      "step": 32819,
      "training_step_time": 7.683210611343384
    },
    {
      "epoch": 3.1299591064453125e-05,
      "grad_norm": 0.12462508678436279,
      "learning_rate": 8.029269698007261e-06,
      "loss": 0.061,
      "step": 32820
    },
    {
      "epoch": 3.1299591064453125e-05,
      "model_forward_time": 1.9021365642547607,
      "step": 32820
    },
    {
      "dynamics_loss_avg": 0.043283377215266226,
      "epoch": 3.1299591064453125e-05,
      "step": 32820
    },
    {
      "action_loss_avg": 0.001931572612375021,
      "epoch": 3.1299591064453125e-05,
      "step": 32820
    },
    {
      "epoch": 3.1299591064453125e-05,
      "step": 32820,
      "training_step_time": 7.671100854873657
    },
    {
      "epoch": 3.130054473876953e-05,
      "model_forward_time": 1.9151623249053955,
      "step": 32821
    },
    {
      "epoch": 3.130054473876953e-05,
      "step": 32821,
      "training_step_time": 7.771432876586914
    },
    {
      "epoch": 3.130149841308594e-05,
      "model_forward_time": 1.9206416606903076,
      "step": 32822
    },
    {
      "epoch": 3.130149841308594e-05,
      "step": 32822,
      "training_step_time": 7.8134379386901855
    },
    {
      "epoch": 3.1302452087402344e-05,
      "model_forward_time": 2.972280979156494,
      "step": 32823
    },
    {
      "epoch": 3.1302452087402344e-05,
      "step": 32823,
      "training_step_time": 7.871335029602051
    },
    {
      "epoch": 3.130340576171875e-05,
      "model_forward_time": 1.9200875759124756,
      "step": 32824
    },
    {
      "epoch": 3.130340576171875e-05,
      "step": 32824,
      "training_step_time": 7.894285440444946
    },
    {
      "epoch": 3.130435943603516e-05,
      "model_forward_time": 1.928295612335205,
      "step": 32825
    },
    {
      "epoch": 3.130435943603516e-05,
      "step": 32825,
      "training_step_time": 7.715808629989624
    },
    {
      "epoch": 3.130531311035156e-05,
      "model_forward_time": 1.9229631423950195,
      "step": 32826
    },
    {
      "epoch": 3.130531311035156e-05,
      "step": 32826,
      "training_step_time": 8.294432163238525
    },
    {
      "epoch": 3.130626678466797e-05,
      "model_forward_time": 1.9122233390808105,
      "step": 32827
    },
    {
      "epoch": 3.130626678466797e-05,
      "step": 32827,
      "training_step_time": 7.862967014312744
    },
    {
      "epoch": 3.130722045898438e-05,
      "model_forward_time": 1.9297621250152588,
      "step": 32828
    },
    {
      "epoch": 3.130722045898438e-05,
      "step": 32828,
      "training_step_time": 8.62630844116211
    },
    {
      "epoch": 3.130817413330078e-05,
      "model_forward_time": 1.9110090732574463,
      "step": 32829
    },
    {
      "epoch": 3.130817413330078e-05,
      "step": 32829,
      "training_step_time": 7.58415412902832
    },
    {
      "epoch": 3.1309127807617186e-05,
      "grad_norm": 0.14613421261310577,
      "learning_rate": 8.027954071663859e-06,
      "loss": 0.0643,
      "step": 32830
    },
    {
      "epoch": 3.1309127807617186e-05,
      "model_forward_time": 1.9203014373779297,
      "step": 32830
    },
    {
      "dynamics_loss_avg": 0.06483152955770492,
      "epoch": 3.1309127807617186e-05,
      "step": 32830
    },
    {
      "action_loss_avg": 0.0017679815471637994,
      "epoch": 3.1309127807617186e-05,
      "step": 32830
    },
    {
      "epoch": 3.1309127807617186e-05,
      "step": 32830,
      "training_step_time": 7.6465537548065186
    },
    {
      "epoch": 3.13100814819336e-05,
      "model_forward_time": 1.9252145290374756,
      "step": 32831
    },
    {
      "epoch": 3.13100814819336e-05,
      "step": 32831,
      "training_step_time": 7.715564727783203
    },
    {
      "epoch": 3.131103515625e-05,
      "model_forward_time": 2.944875717163086,
      "step": 32832
    },
    {
      "epoch": 3.131103515625e-05,
      "step": 32832,
      "training_step_time": 7.668994665145874
    },
    {
      "epoch": 3.1311988830566405e-05,
      "model_forward_time": 1.923705816268921,
      "step": 32833
    },
    {
      "epoch": 3.1311988830566405e-05,
      "step": 32833,
      "training_step_time": 8.613549709320068
    },
    {
      "epoch": 3.1312942504882816e-05,
      "model_forward_time": 1.9085724353790283,
      "step": 32834
    },
    {
      "epoch": 3.1312942504882816e-05,
      "step": 32834,
      "training_step_time": 8.547437191009521
    },
    {
      "epoch": 3.131389617919922e-05,
      "model_forward_time": 1.8994455337524414,
      "step": 32835
    },
    {
      "epoch": 3.131389617919922e-05,
      "step": 32835,
      "training_step_time": 9.319753885269165
    },
    {
      "epoch": 3.1314849853515624e-05,
      "model_forward_time": 1.7509701251983643,
      "step": 32836
    },
    {
      "epoch": 3.1314849853515624e-05,
      "step": 32836,
      "training_step_time": 7.8103928565979
    },
    {
      "epoch": 3.1315803527832034e-05,
      "model_forward_time": 1.7404217720031738,
      "step": 32837
    },
    {
      "epoch": 3.1315803527832034e-05,
      "step": 32837,
      "training_step_time": 8.38883900642395
    },
    {
      "epoch": 3.131675720214844e-05,
      "model_forward_time": 1.746657133102417,
      "step": 32838
    },
    {
      "epoch": 3.131675720214844e-05,
      "step": 32838,
      "training_step_time": 7.66692042350769
    },
    {
      "epoch": 3.131771087646484e-05,
      "model_forward_time": 1.7740304470062256,
      "step": 32839
    },
    {
      "epoch": 3.131771087646484e-05,
      "step": 32839,
      "training_step_time": 7.682631969451904
    },
    {
      "epoch": 3.131866455078125e-05,
      "grad_norm": 0.15709291398525238,
      "learning_rate": 8.026638114187955e-06,
      "loss": 0.0629,
      "step": 32840
    },
    {
      "epoch": 3.131866455078125e-05,
      "model_forward_time": 1.751415491104126,
      "step": 32840
    },
    {
      "dynamics_loss_avg": 0.05827160701155663,
      "epoch": 3.131866455078125e-05,
      "step": 32840
    },
    {
      "action_loss_avg": 0.0014131104850093833,
      "epoch": 3.131866455078125e-05,
      "step": 32840
    },
    {
      "epoch": 3.131866455078125e-05,
      "step": 32840,
      "training_step_time": 7.6413185596466064
    },
    {
      "epoch": 3.131961822509766e-05,
      "model_forward_time": 2.9012858867645264,
      "step": 32841
    },
    {
      "epoch": 3.131961822509766e-05,
      "step": 32841,
      "training_step_time": 7.626236915588379
    },
    {
      "epoch": 3.132057189941406e-05,
      "model_forward_time": 1.745060920715332,
      "step": 32842
    },
    {
      "epoch": 3.132057189941406e-05,
      "step": 32842,
      "training_step_time": 7.576786279678345
    },
    {
      "epoch": 3.132152557373047e-05,
      "model_forward_time": 1.8737201690673828,
      "step": 32843
    },
    {
      "epoch": 3.132152557373047e-05,
      "step": 32843,
      "training_step_time": 7.740280628204346
    },
    {
      "epoch": 3.1322479248046876e-05,
      "model_forward_time": 1.748467206954956,
      "step": 32844
    },
    {
      "epoch": 3.1322479248046876e-05,
      "step": 32844,
      "training_step_time": 9.562622785568237
    },
    {
      "epoch": 3.132343292236328e-05,
      "model_forward_time": 1.7700245380401611,
      "step": 32845
    },
    {
      "epoch": 3.132343292236328e-05,
      "step": 32845,
      "training_step_time": 8.887157440185547
    },
    {
      "epoch": 3.1324386596679684e-05,
      "model_forward_time": 1.7516472339630127,
      "step": 32846
    },
    {
      "epoch": 3.1324386596679684e-05,
      "step": 32846,
      "training_step_time": 8.43782663345337
    },
    {
      "epoch": 3.1325340270996095e-05,
      "model_forward_time": 1.754875659942627,
      "step": 32847
    },
    {
      "epoch": 3.1325340270996095e-05,
      "step": 32847,
      "training_step_time": 7.604480266571045
    },
    {
      "epoch": 3.13262939453125e-05,
      "model_forward_time": 1.744931936264038,
      "step": 32848
    },
    {
      "epoch": 3.13262939453125e-05,
      "step": 32848,
      "training_step_time": 7.640030145645142
    },
    {
      "epoch": 3.13272476196289e-05,
      "model_forward_time": 1.7547624111175537,
      "step": 32849
    },
    {
      "epoch": 3.13272476196289e-05,
      "step": 32849,
      "training_step_time": 8.259965896606445
    },
    {
      "epoch": 3.1328201293945314e-05,
      "grad_norm": 0.1901588886976242,
      "learning_rate": 8.025321825723455e-06,
      "loss": 0.0622,
      "step": 32850
    },
    {
      "epoch": 3.1328201293945314e-05,
      "model_forward_time": 2.8874173164367676,
      "step": 32850
    },
    {
      "dynamics_loss_avg": 0.062249299883842465,
      "epoch": 3.1328201293945314e-05,
      "step": 32850
    },
    {
      "action_loss_avg": 0.0033317047680611724,
      "epoch": 3.1328201293945314e-05,
      "step": 32850
    },
    {
      "epoch": 3.1328201293945314e-05,
      "step": 32850,
      "training_step_time": 7.761096239089966
    },
    {
      "epoch": 3.132915496826172e-05,
      "model_forward_time": 1.7477211952209473,
      "step": 32851
    },
    {
      "epoch": 3.132915496826172e-05,
      "step": 32851,
      "training_step_time": 7.536155462265015
    },
    {
      "epoch": 3.133010864257812e-05,
      "model_forward_time": 1.7779603004455566,
      "step": 32852
    },
    {
      "epoch": 3.133010864257812e-05,
      "step": 32852,
      "training_step_time": 7.822721242904663
    },
    {
      "epoch": 3.133106231689453e-05,
      "model_forward_time": 1.7446184158325195,
      "step": 32853
    },
    {
      "epoch": 3.133106231689453e-05,
      "step": 32853,
      "training_step_time": 8.657655239105225
    },
    {
      "epoch": 3.133201599121094e-05,
      "model_forward_time": 1.7538001537322998,
      "step": 32854
    },
    {
      "epoch": 3.133201599121094e-05,
      "step": 32854,
      "training_step_time": 8.413995027542114
    },
    {
      "epoch": 3.133296966552734e-05,
      "model_forward_time": 1.3865211009979248,
      "step": 32855
    },
    {
      "epoch": 3.133296966552734e-05,
      "step": 32855,
      "training_step_time": 7.589285373687744
    },
    {
      "epoch": 3.133392333984375e-05,
      "model_forward_time": 1.7562870979309082,
      "step": 32856
    },
    {
      "epoch": 3.133392333984375e-05,
      "step": 32856,
      "training_step_time": 7.654745817184448
    },
    {
      "epoch": 3.1334877014160156e-05,
      "model_forward_time": 1.747316598892212,
      "step": 32857
    },
    {
      "epoch": 3.1334877014160156e-05,
      "step": 32857,
      "training_step_time": 7.625284194946289
    },
    {
      "epoch": 3.133583068847656e-05,
      "model_forward_time": 1.744464635848999,
      "step": 32858
    },
    {
      "epoch": 3.133583068847656e-05,
      "step": 32858,
      "training_step_time": 7.596920490264893
    },
    {
      "epoch": 3.133678436279297e-05,
      "model_forward_time": 2.926455497741699,
      "step": 32859
    },
    {
      "epoch": 3.133678436279297e-05,
      "step": 32859,
      "training_step_time": 8.587446928024292
    },
    {
      "epoch": 3.1337738037109374e-05,
      "grad_norm": 0.24972689151763916,
      "learning_rate": 8.02400520641431e-06,
      "loss": 0.0631,
      "step": 32860
    },
    {
      "epoch": 3.1337738037109374e-05,
      "model_forward_time": 1.7369318008422852,
      "step": 32860
    },
    {
      "dynamics_loss_avg": 0.05578323192894459,
      "epoch": 3.1337738037109374e-05,
      "step": 32860
    },
    {
      "action_loss_avg": 0.002011759567540139,
      "epoch": 3.1337738037109374e-05,
      "step": 32860
    },
    {
      "epoch": 3.1337738037109374e-05,
      "step": 32860,
      "training_step_time": 7.528428792953491
    },
    {
      "epoch": 3.133869171142578e-05,
      "model_forward_time": 1.7710118293762207,
      "step": 32861
    },
    {
      "epoch": 3.133869171142578e-05,
      "step": 32861,
      "training_step_time": 9.455418825149536
    },
    {
      "epoch": 3.133964538574219e-05,
      "model_forward_time": 1.8779361248016357,
      "step": 32862
    },
    {
      "epoch": 3.133964538574219e-05,
      "step": 32862,
      "training_step_time": 7.721790075302124
    },
    {
      "epoch": 3.134059906005859e-05,
      "model_forward_time": 1.7540245056152344,
      "step": 32863
    },
    {
      "epoch": 3.134059906005859e-05,
      "step": 32863,
      "training_step_time": 8.841026544570923
    },
    {
      "epoch": 3.1341552734375e-05,
      "model_forward_time": 1.7542657852172852,
      "step": 32864
    },
    {
      "epoch": 3.1341552734375e-05,
      "step": 32864,
      "training_step_time": 8.039516925811768
    },
    {
      "epoch": 3.134250640869141e-05,
      "model_forward_time": 1.7482247352600098,
      "step": 32865
    },
    {
      "epoch": 3.134250640869141e-05,
      "step": 32865,
      "training_step_time": 7.696643590927124
    },
    {
      "epoch": 3.134346008300781e-05,
      "model_forward_time": 1.7385218143463135,
      "step": 32866
    },
    {
      "epoch": 3.134346008300781e-05,
      "step": 32866,
      "training_step_time": 7.771466970443726
    },
    {
      "epoch": 3.1344413757324216e-05,
      "model_forward_time": 1.7454261779785156,
      "step": 32867
    },
    {
      "epoch": 3.1344413757324216e-05,
      "step": 32867,
      "training_step_time": 7.76273250579834
    },
    {
      "epoch": 3.134536743164063e-05,
      "model_forward_time": 2.8454668521881104,
      "step": 32868
    },
    {
      "epoch": 3.134536743164063e-05,
      "step": 32868,
      "training_step_time": 7.808579921722412
    },
    {
      "epoch": 3.134632110595703e-05,
      "model_forward_time": 1.7405047416687012,
      "step": 32869
    },
    {
      "epoch": 3.134632110595703e-05,
      "step": 32869,
      "training_step_time": 8.522679805755615
    },
    {
      "epoch": 3.1347274780273435e-05,
      "grad_norm": 0.14860796928405762,
      "learning_rate": 8.022688256404502e-06,
      "loss": 0.0628,
      "step": 32870
    },
    {
      "epoch": 3.1347274780273435e-05,
      "model_forward_time": 1.7364568710327148,
      "step": 32870
    },
    {
      "dynamics_loss_avg": 0.055065936781466006,
      "epoch": 3.1347274780273435e-05,
      "step": 32870
    },
    {
      "action_loss_avg": 0.001379495859146118,
      "epoch": 3.1347274780273435e-05,
      "step": 32870
    },
    {
      "epoch": 3.1347274780273435e-05,
      "step": 32870,
      "training_step_time": 9.304709672927856
    },
    {
      "epoch": 3.1348228454589846e-05,
      "model_forward_time": 1.8291099071502686,
      "step": 32871
    },
    {
      "epoch": 3.1348228454589846e-05,
      "step": 32871,
      "training_step_time": 8.453887224197388
    },
    {
      "epoch": 3.134918212890625e-05,
      "model_forward_time": 1.750542402267456,
      "step": 32872
    },
    {
      "epoch": 3.134918212890625e-05,
      "step": 32872,
      "training_step_time": 7.861397981643677
    },
    {
      "epoch": 3.1350135803222654e-05,
      "model_forward_time": 1.7614672183990479,
      "step": 32873
    },
    {
      "epoch": 3.1350135803222654e-05,
      "step": 32873,
      "training_step_time": 7.939902305603027
    },
    {
      "epoch": 3.1351089477539065e-05,
      "model_forward_time": 1.7408554553985596,
      "step": 32874
    },
    {
      "epoch": 3.1351089477539065e-05,
      "step": 32874,
      "training_step_time": 7.53098201751709
    },
    {
      "epoch": 3.135204315185547e-05,
      "model_forward_time": 1.7378349304199219,
      "step": 32875
    },
    {
      "epoch": 3.135204315185547e-05,
      "step": 32875,
      "training_step_time": 7.753518581390381
    },
    {
      "epoch": 3.135299682617187e-05,
      "model_forward_time": 1.746955156326294,
      "step": 32876
    },
    {
      "epoch": 3.135299682617187e-05,
      "step": 32876,
      "training_step_time": 7.773736953735352
    },
    {
      "epoch": 3.1353950500488284e-05,
      "model_forward_time": 2.894254446029663,
      "step": 32877
    },
    {
      "epoch": 3.1353950500488284e-05,
      "step": 32877,
      "training_step_time": 7.790745735168457
    },
    {
      "epoch": 3.135490417480469e-05,
      "model_forward_time": 1.7522363662719727,
      "step": 32878
    },
    {
      "epoch": 3.135490417480469e-05,
      "step": 32878,
      "training_step_time": 8.277178525924683
    },
    {
      "epoch": 3.135585784912109e-05,
      "model_forward_time": 1.750227928161621,
      "step": 32879
    },
    {
      "epoch": 3.135585784912109e-05,
      "step": 32879,
      "training_step_time": 7.744153022766113
    },
    {
      "epoch": 3.13568115234375e-05,
      "grad_norm": 0.1459158957004547,
      "learning_rate": 8.02137097583805e-06,
      "loss": 0.0597,
      "step": 32880
    },
    {
      "epoch": 3.13568115234375e-05,
      "model_forward_time": 1.795588731765747,
      "step": 32880
    },
    {
      "dynamics_loss_avg": 0.06394266858696937,
      "epoch": 3.13568115234375e-05,
      "step": 32880
    },
    {
      "action_loss_avg": 0.002394464862300083,
      "epoch": 3.13568115234375e-05,
      "step": 32880
    },
    {
      "epoch": 3.13568115234375e-05,
      "step": 32880,
      "training_step_time": 7.709836721420288
    },
    {
      "epoch": 3.1357765197753906e-05,
      "model_forward_time": 1.7563121318817139,
      "step": 32881
    },
    {
      "epoch": 3.1357765197753906e-05,
      "step": 32881,
      "training_step_time": 8.301151514053345
    },
    {
      "epoch": 3.135871887207031e-05,
      "model_forward_time": 1.7487964630126953,
      "step": 32882
    },
    {
      "epoch": 3.135871887207031e-05,
      "step": 32882,
      "training_step_time": 7.901979207992554
    },
    {
      "epoch": 3.135967254638672e-05,
      "model_forward_time": 1.7470767498016357,
      "step": 32883
    },
    {
      "epoch": 3.135967254638672e-05,
      "step": 32883,
      "training_step_time": 7.804181814193726
    },
    {
      "epoch": 3.1360626220703125e-05,
      "model_forward_time": 1.7330996990203857,
      "step": 32884
    },
    {
      "epoch": 3.1360626220703125e-05,
      "step": 32884,
      "training_step_time": 8.85195279121399
    },
    {
      "epoch": 3.136157989501953e-05,
      "model_forward_time": 1.7319440841674805,
      "step": 32885
    },
    {
      "epoch": 3.136157989501953e-05,
      "step": 32885,
      "training_step_time": 8.172163486480713
    },
    {
      "epoch": 3.136253356933594e-05,
      "model_forward_time": 2.8040356636047363,
      "step": 32886
    },
    {
      "epoch": 3.136253356933594e-05,
      "step": 32886,
      "training_step_time": 7.607257843017578
    },
    {
      "epoch": 3.1363487243652344e-05,
      "model_forward_time": 1.7214159965515137,
      "step": 32887
    },
    {
      "epoch": 3.1363487243652344e-05,
      "step": 32887,
      "training_step_time": 7.696197032928467
    },
    {
      "epoch": 3.136444091796875e-05,
      "model_forward_time": 1.7480411529541016,
      "step": 32888
    },
    {
      "epoch": 3.136444091796875e-05,
      "step": 32888,
      "training_step_time": 7.68825626373291
    },
    {
      "epoch": 3.136539459228516e-05,
      "model_forward_time": 1.7495415210723877,
      "step": 32889
    },
    {
      "epoch": 3.136539459228516e-05,
      "step": 32889,
      "training_step_time": 8.558359384536743
    },
    {
      "epoch": 3.136634826660156e-05,
      "grad_norm": 0.13208132982254028,
      "learning_rate": 8.020053364859011e-06,
      "loss": 0.0595,
      "step": 32890
    },
    {
      "epoch": 3.136634826660156e-05,
      "model_forward_time": 1.72963285446167,
      "step": 32890
    },
    {
      "dynamics_loss_avg": 0.03961887024343014,
      "epoch": 3.136634826660156e-05,
      "step": 32890
    },
    {
      "action_loss_avg": 0.0011420676775742323,
      "epoch": 3.136634826660156e-05,
      "step": 32890
    },
    {
      "epoch": 3.136634826660156e-05,
      "step": 32890,
      "training_step_time": 8.36673378944397
    },
    {
      "epoch": 3.136730194091797e-05,
      "model_forward_time": 1.034245491027832,
      "step": 32891
    },
    {
      "epoch": 3.136730194091797e-05,
      "step": 32891,
      "training_step_time": 7.754102468490601
    },
    {
      "epoch": 3.136825561523438e-05,
      "model_forward_time": 1.7365765571594238,
      "step": 32892
    },
    {
      "epoch": 3.136825561523438e-05,
      "step": 32892,
      "training_step_time": 7.668698072433472
    },
    {
      "epoch": 3.136920928955078e-05,
      "model_forward_time": 1.7353909015655518,
      "step": 32893
    },
    {
      "epoch": 3.136920928955078e-05,
      "step": 32893,
      "training_step_time": 7.664870023727417
    },
    {
      "epoch": 3.1370162963867186e-05,
      "model_forward_time": 1.7411670684814453,
      "step": 32894
    },
    {
      "epoch": 3.1370162963867186e-05,
      "step": 32894,
      "training_step_time": 8.090697288513184
    },
    {
      "epoch": 3.13711166381836e-05,
      "model_forward_time": 1.7434592247009277,
      "step": 32895
    },
    {
      "epoch": 3.13711166381836e-05,
      "step": 32895,
      "training_step_time": 7.411538124084473
    },
    {
      "epoch": 3.13720703125e-05,
      "model_forward_time": 2.8773036003112793,
      "step": 32896
    },
    {
      "epoch": 3.13720703125e-05,
      "step": 32896,
      "training_step_time": 9.47396731376648
    },
    {
      "epoch": 3.1373023986816405e-05,
      "model_forward_time": 1.733245849609375,
      "step": 32897
    },
    {
      "epoch": 3.1373023986816405e-05,
      "step": 32897,
      "training_step_time": 9.135583877563477
    },
    {
      "epoch": 3.1373977661132816e-05,
      "model_forward_time": 1.7481355667114258,
      "step": 32898
    },
    {
      "epoch": 3.1373977661132816e-05,
      "step": 32898,
      "training_step_time": 7.642515659332275
    },
    {
      "epoch": 3.137493133544922e-05,
      "model_forward_time": 1.8572592735290527,
      "step": 32899
    },
    {
      "epoch": 3.137493133544922e-05,
      "step": 32899,
      "training_step_time": 7.643052816390991
    },
    {
      "epoch": 3.1375885009765624e-05,
      "grad_norm": 0.14872033894062042,
      "learning_rate": 8.018735423611476e-06,
      "loss": 0.0637,
      "step": 32900
    },
    {
      "epoch": 3.1375885009765624e-05,
      "model_forward_time": 1.7203114032745361,
      "step": 32900
    },
    {
      "dynamics_loss_avg": 0.06679327376186847,
      "epoch": 3.1375885009765624e-05,
      "step": 32900
    },
    {
      "action_loss_avg": 0.0028964878350961954,
      "epoch": 3.1375885009765624e-05,
      "step": 32900
    },
    {
      "epoch": 3.1375885009765624e-05,
      "step": 32900,
      "training_step_time": 8.348671913146973
    },
    {
      "epoch": 3.1376838684082034e-05,
      "model_forward_time": 1.3699288368225098,
      "step": 32901
    },
    {
      "epoch": 3.1376838684082034e-05,
      "step": 32901,
      "training_step_time": 7.681872606277466
    },
    {
      "epoch": 3.137779235839844e-05,
      "model_forward_time": 1.742365837097168,
      "step": 32902
    },
    {
      "epoch": 3.137779235839844e-05,
      "step": 32902,
      "training_step_time": 7.6948723793029785
    },
    {
      "epoch": 3.137874603271484e-05,
      "model_forward_time": 1.7369260787963867,
      "step": 32903
    },
    {
      "epoch": 3.137874603271484e-05,
      "step": 32903,
      "training_step_time": 7.594717979431152
    },
    {
      "epoch": 3.137969970703125e-05,
      "model_forward_time": 1.7457215785980225,
      "step": 32904
    },
    {
      "epoch": 3.137969970703125e-05,
      "step": 32904,
      "training_step_time": 7.629163026809692
    },
    {
      "epoch": 3.138065338134766e-05,
      "model_forward_time": 2.8603367805480957,
      "step": 32905
    },
    {
      "epoch": 3.138065338134766e-05,
      "step": 32905,
      "training_step_time": 8.701205730438232
    },
    {
      "epoch": 3.138160705566406e-05,
      "model_forward_time": 1.7244513034820557,
      "step": 32906
    },
    {
      "epoch": 3.138160705566406e-05,
      "step": 32906,
      "training_step_time": 8.193644523620605
    },
    {
      "epoch": 3.1382560729980465e-05,
      "model_forward_time": 1.7216053009033203,
      "step": 32907
    },
    {
      "epoch": 3.1382560729980465e-05,
      "step": 32907,
      "training_step_time": 7.666266202926636
    },
    {
      "epoch": 3.1383514404296876e-05,
      "model_forward_time": 1.8581881523132324,
      "step": 32908
    },
    {
      "epoch": 3.1383514404296876e-05,
      "step": 32908,
      "training_step_time": 9.024099111557007
    },
    {
      "epoch": 3.138446807861328e-05,
      "model_forward_time": 1.7444605827331543,
      "step": 32909
    },
    {
      "epoch": 3.138446807861328e-05,
      "step": 32909,
      "training_step_time": 7.736131191253662
    },
    {
      "epoch": 3.1385421752929684e-05,
      "grad_norm": 0.13323694467544556,
      "learning_rate": 8.017417152239575e-06,
      "loss": 0.0575,
      "step": 32910
    },
    {
      "epoch": 3.1385421752929684e-05,
      "model_forward_time": 1.7280645370483398,
      "step": 32910
    },
    {
      "dynamics_loss_avg": 0.06529918918386102,
      "epoch": 3.1385421752929684e-05,
      "step": 32910
    },
    {
      "action_loss_avg": 0.004179295655922033,
      "epoch": 3.1385421752929684e-05,
      "step": 32910
    },
    {
      "epoch": 3.1385421752929684e-05,
      "step": 32910,
      "training_step_time": 7.657088994979858
    },
    {
      "epoch": 3.1386375427246095e-05,
      "model_forward_time": 1.7483820915222168,
      "step": 32911
    },
    {
      "epoch": 3.1386375427246095e-05,
      "step": 32911,
      "training_step_time": 8.00309705734253
    },
    {
      "epoch": 3.13873291015625e-05,
      "model_forward_time": 1.7351434230804443,
      "step": 32912
    },
    {
      "epoch": 3.13873291015625e-05,
      "step": 32912,
      "training_step_time": 7.721039772033691
    },
    {
      "epoch": 3.13882827758789e-05,
      "model_forward_time": 1.7450010776519775,
      "step": 32913
    },
    {
      "epoch": 3.13882827758789e-05,
      "step": 32913,
      "training_step_time": 8.171109199523926
    },
    {
      "epoch": 3.1389236450195314e-05,
      "model_forward_time": 1.7371563911437988,
      "step": 32914
    },
    {
      "epoch": 3.1389236450195314e-05,
      "step": 32914,
      "training_step_time": 9.343366384506226
    },
    {
      "epoch": 3.139019012451172e-05,
      "model_forward_time": 1.7564027309417725,
      "step": 32915
    },
    {
      "epoch": 3.139019012451172e-05,
      "step": 32915,
      "training_step_time": 7.693053722381592
    },
    {
      "epoch": 3.139114379882812e-05,
      "model_forward_time": 1.7567596435546875,
      "step": 32916
    },
    {
      "epoch": 3.139114379882812e-05,
      "step": 32916,
      "training_step_time": 7.763894557952881
    },
    {
      "epoch": 3.139209747314453e-05,
      "model_forward_time": 1.8170795440673828,
      "step": 32917
    },
    {
      "epoch": 3.139209747314453e-05,
      "step": 32917,
      "training_step_time": 7.625106334686279
    },
    {
      "epoch": 3.139305114746094e-05,
      "model_forward_time": 1.758774757385254,
      "step": 32918
    },
    {
      "epoch": 3.139305114746094e-05,
      "step": 32918,
      "training_step_time": 8.65098237991333
    },
    {
      "epoch": 3.139400482177734e-05,
      "model_forward_time": 1.7527952194213867,
      "step": 32919
    },
    {
      "epoch": 3.139400482177734e-05,
      "step": 32919,
      "training_step_time": 8.399073600769043
    },
    {
      "epoch": 3.139495849609375e-05,
      "grad_norm": 0.1298108696937561,
      "learning_rate": 8.01609855088747e-06,
      "loss": 0.058,
      "step": 32920
    },
    {
      "epoch": 3.139495849609375e-05,
      "model_forward_time": 1.7578020095825195,
      "step": 32920
    },
    {
      "dynamics_loss_avg": 0.05563136544078588,
      "epoch": 3.139495849609375e-05,
      "step": 32920
    },
    {
      "action_loss_avg": 0.0017107012216001749,
      "epoch": 3.139495849609375e-05,
      "step": 32920
    },
    {
      "epoch": 3.139495849609375e-05,
      "step": 32920,
      "training_step_time": 7.4653754234313965
    },
    {
      "epoch": 3.1395912170410156e-05,
      "model_forward_time": 1.7512717247009277,
      "step": 32921
    },
    {
      "epoch": 3.1395912170410156e-05,
      "step": 32921,
      "training_step_time": 7.755764484405518
    },
    {
      "epoch": 3.139686584472656e-05,
      "model_forward_time": 1.7519986629486084,
      "step": 32922
    },
    {
      "epoch": 3.139686584472656e-05,
      "step": 32922,
      "training_step_time": 7.702230215072632
    },
    {
      "epoch": 3.139781951904297e-05,
      "model_forward_time": 1.748358964920044,
      "step": 32923
    },
    {
      "epoch": 3.139781951904297e-05,
      "step": 32923,
      "training_step_time": 7.625482082366943
    },
    {
      "epoch": 3.1398773193359375e-05,
      "model_forward_time": 2.920098304748535,
      "step": 32924
    },
    {
      "epoch": 3.1398773193359375e-05,
      "step": 32924,
      "training_step_time": 7.7489173412323
    },
    {
      "epoch": 3.139972686767578e-05,
      "model_forward_time": 1.7566509246826172,
      "step": 32925
    },
    {
      "epoch": 3.139972686767578e-05,
      "step": 32925,
      "training_step_time": 8.877237796783447
    },
    {
      "epoch": 3.140068054199219e-05,
      "model_forward_time": 1.8051471710205078,
      "step": 32926
    },
    {
      "epoch": 3.140068054199219e-05,
      "step": 32926,
      "training_step_time": 7.6256914138793945
    },
    {
      "epoch": 3.140163421630859e-05,
      "model_forward_time": 1.7529969215393066,
      "step": 32927
    },
    {
      "epoch": 3.140163421630859e-05,
      "step": 32927,
      "training_step_time": 7.798682451248169
    },
    {
      "epoch": 3.1402587890625e-05,
      "model_forward_time": 1.7427384853363037,
      "step": 32928
    },
    {
      "epoch": 3.1402587890625e-05,
      "step": 32928,
      "training_step_time": 7.798604249954224
    },
    {
      "epoch": 3.140354156494141e-05,
      "model_forward_time": 1.7296142578125,
      "step": 32929
    },
    {
      "epoch": 3.140354156494141e-05,
      "step": 32929,
      "training_step_time": 8.55640959739685
    },
    {
      "epoch": 3.140449523925781e-05,
      "grad_norm": 0.13186879456043243,
      "learning_rate": 8.014779619699363e-06,
      "loss": 0.0591,
      "step": 32930
    },
    {
      "epoch": 3.140449523925781e-05,
      "model_forward_time": 1.718292236328125,
      "step": 32930
    },
    {
      "dynamics_loss_avg": 0.058943250123411416,
      "epoch": 3.140449523925781e-05,
      "step": 32930
    },
    {
      "action_loss_avg": 0.0024048728839261456,
      "epoch": 3.140449523925781e-05,
      "step": 32930
    },
    {
      "epoch": 3.140449523925781e-05,
      "step": 32930,
      "training_step_time": 7.768759250640869
    },
    {
      "epoch": 3.1405448913574216e-05,
      "model_forward_time": 1.7404084205627441,
      "step": 32931
    },
    {
      "epoch": 3.1405448913574216e-05,
      "step": 32931,
      "training_step_time": 7.590330123901367
    },
    {
      "epoch": 3.140640258789063e-05,
      "model_forward_time": 1.7318859100341797,
      "step": 32932
    },
    {
      "epoch": 3.140640258789063e-05,
      "step": 32932,
      "training_step_time": 8.677551031112671
    },
    {
      "epoch": 3.140735626220703e-05,
      "model_forward_time": 2.862006425857544,
      "step": 32933
    },
    {
      "epoch": 3.140735626220703e-05,
      "step": 32933,
      "training_step_time": 8.622587203979492
    },
    {
      "epoch": 3.1408309936523435e-05,
      "model_forward_time": 1.731433391571045,
      "step": 32934
    },
    {
      "epoch": 3.1408309936523435e-05,
      "step": 32934,
      "training_step_time": 8.764520645141602
    },
    {
      "epoch": 3.1409263610839846e-05,
      "model_forward_time": 1.744642972946167,
      "step": 32935
    },
    {
      "epoch": 3.1409263610839846e-05,
      "step": 32935,
      "training_step_time": 8.311277151107788
    },
    {
      "epoch": 3.141021728515625e-05,
      "model_forward_time": 1.8536572456359863,
      "step": 32936
    },
    {
      "epoch": 3.141021728515625e-05,
      "step": 32936,
      "training_step_time": 8.291605472564697
    },
    {
      "epoch": 3.1411170959472654e-05,
      "model_forward_time": 1.7344911098480225,
      "step": 32937
    },
    {
      "epoch": 3.1411170959472654e-05,
      "step": 32937,
      "training_step_time": 8.775381326675415
    },
    {
      "epoch": 3.1412124633789065e-05,
      "model_forward_time": 1.7374968528747559,
      "step": 32938
    },
    {
      "epoch": 3.1412124633789065e-05,
      "step": 32938,
      "training_step_time": 7.683910846710205
    },
    {
      "epoch": 3.141307830810547e-05,
      "model_forward_time": 1.7372660636901855,
      "step": 32939
    },
    {
      "epoch": 3.141307830810547e-05,
      "step": 32939,
      "training_step_time": 7.707595348358154
    },
    {
      "epoch": 3.141403198242187e-05,
      "grad_norm": 0.13104386627674103,
      "learning_rate": 8.01346035881949e-06,
      "loss": 0.055,
      "step": 32940
    },
    {
      "epoch": 3.141403198242187e-05,
      "model_forward_time": 1.7181775569915771,
      "step": 32940
    },
    {
      "dynamics_loss_avg": 0.04699694029986858,
      "epoch": 3.141403198242187e-05,
      "step": 32940
    },
    {
      "action_loss_avg": 0.0020191894262097778,
      "epoch": 3.141403198242187e-05,
      "step": 32940
    },
    {
      "epoch": 3.141403198242187e-05,
      "step": 32940,
      "training_step_time": 7.678785800933838
    },
    {
      "epoch": 3.1414985656738284e-05,
      "model_forward_time": 1.731424331665039,
      "step": 32941
    },
    {
      "epoch": 3.1414985656738284e-05,
      "step": 32941,
      "training_step_time": 8.195668697357178
    },
    {
      "epoch": 3.141593933105469e-05,
      "model_forward_time": 2.861232042312622,
      "step": 32942
    },
    {
      "epoch": 3.141593933105469e-05,
      "step": 32942,
      "training_step_time": 8.405731916427612
    },
    {
      "epoch": 3.141689300537109e-05,
      "model_forward_time": 1.7295856475830078,
      "step": 32943
    },
    {
      "epoch": 3.141689300537109e-05,
      "step": 32943,
      "training_step_time": 7.643048524856567
    },
    {
      "epoch": 3.14178466796875e-05,
      "model_forward_time": 1.7291693687438965,
      "step": 32944
    },
    {
      "epoch": 3.14178466796875e-05,
      "step": 32944,
      "training_step_time": 9.332533836364746
    },
    {
      "epoch": 3.1418800354003907e-05,
      "model_forward_time": 1.842247724533081,
      "step": 32945
    },
    {
      "epoch": 3.1418800354003907e-05,
      "step": 32945,
      "training_step_time": 8.297987222671509
    },
    {
      "epoch": 3.141975402832031e-05,
      "model_forward_time": 1.7263784408569336,
      "step": 32946
    },
    {
      "epoch": 3.141975402832031e-05,
      "step": 32946,
      "training_step_time": 8.766421794891357
    },
    {
      "epoch": 3.142070770263672e-05,
      "model_forward_time": 1.733213186264038,
      "step": 32947
    },
    {
      "epoch": 3.142070770263672e-05,
      "step": 32947,
      "training_step_time": 7.571898937225342
    },
    {
      "epoch": 3.1421661376953125e-05,
      "model_forward_time": 1.731590986251831,
      "step": 32948
    },
    {
      "epoch": 3.1421661376953125e-05,
      "step": 32948,
      "training_step_time": 7.6146345138549805
    },
    {
      "epoch": 3.142261505126953e-05,
      "model_forward_time": 1.7441215515136719,
      "step": 32949
    },
    {
      "epoch": 3.142261505126953e-05,
      "step": 32949,
      "training_step_time": 8.681060314178467
    },
    {
      "epoch": 3.142356872558594e-05,
      "grad_norm": 0.18192638456821442,
      "learning_rate": 8.01214076839212e-06,
      "loss": 0.0614,
      "step": 32950
    },
    {
      "epoch": 3.142356872558594e-05,
      "model_forward_time": 1.7245724201202393,
      "step": 32950
    },
    {
      "dynamics_loss_avg": 0.062214001640677453,
      "epoch": 3.142356872558594e-05,
      "step": 32950
    },
    {
      "action_loss_avg": 0.0026955025212373586,
      "epoch": 3.142356872558594e-05,
      "step": 32950
    },
    {
      "epoch": 3.142356872558594e-05,
      "step": 32950,
      "training_step_time": 7.754914283752441
    },
    {
      "epoch": 3.1424522399902344e-05,
      "model_forward_time": 2.8620359897613525,
      "step": 32951
    },
    {
      "epoch": 3.1424522399902344e-05,
      "step": 32951,
      "training_step_time": 8.65060019493103
    },
    {
      "epoch": 3.142547607421875e-05,
      "model_forward_time": 1.7286486625671387,
      "step": 32952
    },
    {
      "epoch": 3.142547607421875e-05,
      "step": 32952,
      "training_step_time": 7.678133726119995
    },
    {
      "epoch": 3.142642974853516e-05,
      "model_forward_time": 1.721660852432251,
      "step": 32953
    },
    {
      "epoch": 3.142642974853516e-05,
      "step": 32953,
      "training_step_time": 7.8159403800964355
    },
    {
      "epoch": 3.142738342285156e-05,
      "model_forward_time": 1.7912378311157227,
      "step": 32954
    },
    {
      "epoch": 3.142738342285156e-05,
      "step": 32954,
      "training_step_time": 7.594100475311279
    },
    {
      "epoch": 3.142833709716797e-05,
      "model_forward_time": 1.3611657619476318,
      "step": 32955
    },
    {
      "epoch": 3.142833709716797e-05,
      "step": 32955,
      "training_step_time": 8.902271270751953
    },
    {
      "epoch": 3.142929077148438e-05,
      "model_forward_time": 1.740837812423706,
      "step": 32956
    },
    {
      "epoch": 3.142929077148438e-05,
      "step": 32956,
      "training_step_time": 7.636181592941284
    },
    {
      "epoch": 3.143024444580078e-05,
      "model_forward_time": 1.73136305809021,
      "step": 32957
    },
    {
      "epoch": 3.143024444580078e-05,
      "step": 32957,
      "training_step_time": 8.337312698364258
    },
    {
      "epoch": 3.1431198120117186e-05,
      "model_forward_time": 1.7346575260162354,
      "step": 32958
    },
    {
      "epoch": 3.1431198120117186e-05,
      "step": 32958,
      "training_step_time": 7.71502685546875
    },
    {
      "epoch": 3.14321517944336e-05,
      "model_forward_time": 1.732234239578247,
      "step": 32959
    },
    {
      "epoch": 3.14321517944336e-05,
      "step": 32959,
      "training_step_time": 7.8876447677612305
    },
    {
      "epoch": 3.143310546875e-05,
      "grad_norm": 0.16110385954380035,
      "learning_rate": 8.010820848561565e-06,
      "loss": 0.0562,
      "step": 32960
    },
    {
      "epoch": 3.143310546875e-05,
      "model_forward_time": 2.7711129188537598,
      "step": 32960
    },
    {
      "dynamics_loss_avg": 0.06209536287933588,
      "epoch": 3.143310546875e-05,
      "step": 32960
    },
    {
      "action_loss_avg": 0.0023986744694411755,
      "epoch": 3.143310546875e-05,
      "step": 32960
    },
    {
      "epoch": 3.143310546875e-05,
      "step": 32960,
      "training_step_time": 8.41263198852539
    },
    {
      "epoch": 3.1434059143066405e-05,
      "model_forward_time": 1.7329325675964355,
      "step": 32961
    },
    {
      "epoch": 3.1434059143066405e-05,
      "step": 32961,
      "training_step_time": 7.69893741607666
    },
    {
      "epoch": 3.1435012817382816e-05,
      "model_forward_time": 1.7332875728607178,
      "step": 32962
    },
    {
      "epoch": 3.1435012817382816e-05,
      "step": 32962,
      "training_step_time": 8.306741714477539
    },
    {
      "epoch": 3.143596649169922e-05,
      "model_forward_time": 1.77223801612854,
      "step": 32963
    },
    {
      "epoch": 3.143596649169922e-05,
      "step": 32963,
      "training_step_time": 7.650972366333008
    },
    {
      "epoch": 3.1436920166015624e-05,
      "model_forward_time": 1.7415547370910645,
      "step": 32964
    },
    {
      "epoch": 3.1436920166015624e-05,
      "step": 32964,
      "training_step_time": 7.773216485977173
    },
    {
      "epoch": 3.1437873840332035e-05,
      "model_forward_time": 1.7307803630828857,
      "step": 32965
    },
    {
      "epoch": 3.1437873840332035e-05,
      "step": 32965,
      "training_step_time": 7.754382848739624
    },
    {
      "epoch": 3.143882751464844e-05,
      "model_forward_time": 1.7381386756896973,
      "step": 32966
    },
    {
      "epoch": 3.143882751464844e-05,
      "step": 32966,
      "training_step_time": 7.58466100692749
    },
    {
      "epoch": 3.143978118896484e-05,
      "model_forward_time": 1.7332494258880615,
      "step": 32967
    },
    {
      "epoch": 3.143978118896484e-05,
      "step": 32967,
      "training_step_time": 8.62521743774414
    },
    {
      "epoch": 3.144073486328125e-05,
      "model_forward_time": 1.726776123046875,
      "step": 32968
    },
    {
      "epoch": 3.144073486328125e-05,
      "step": 32968,
      "training_step_time": 7.6199846267700195
    },
    {
      "epoch": 3.144168853759766e-05,
      "model_forward_time": 1.7393040657043457,
      "step": 32969
    },
    {
      "epoch": 3.144168853759766e-05,
      "step": 32969,
      "training_step_time": 7.6137566566467285
    },
    {
      "epoch": 3.144264221191406e-05,
      "grad_norm": 0.08907652646303177,
      "learning_rate": 8.009500599472171e-06,
      "loss": 0.0571,
      "step": 32970
    },
    {
      "epoch": 3.144264221191406e-05,
      "model_forward_time": 2.9024033546447754,
      "step": 32970
    },
    {
      "dynamics_loss_avg": 0.05408310890197754,
      "epoch": 3.144264221191406e-05,
      "step": 32970
    },
    {
      "action_loss_avg": 0.003589421883225441,
      "epoch": 3.144264221191406e-05,
      "step": 32970
    },
    {
      "epoch": 3.144264221191406e-05,
      "step": 32970,
      "training_step_time": 8.183928728103638
    },
    {
      "epoch": 3.1443595886230465e-05,
      "model_forward_time": 1.7423288822174072,
      "step": 32971
    },
    {
      "epoch": 3.1443595886230465e-05,
      "step": 32971,
      "training_step_time": 8.10739278793335
    },
    {
      "epoch": 3.1444549560546876e-05,
      "model_forward_time": 1.743093729019165,
      "step": 32972
    },
    {
      "epoch": 3.1444549560546876e-05,
      "step": 32972,
      "training_step_time": 8.475533246994019
    },
    {
      "epoch": 3.144550323486328e-05,
      "model_forward_time": 1.7258918285369873,
      "step": 32973
    },
    {
      "epoch": 3.144550323486328e-05,
      "step": 32973,
      "training_step_time": 8.31919002532959
    },
    {
      "epoch": 3.1446456909179684e-05,
      "model_forward_time": 1.7378957271575928,
      "step": 32974
    },
    {
      "epoch": 3.1446456909179684e-05,
      "step": 32974,
      "training_step_time": 7.727649211883545
    },
    {
      "epoch": 3.1447410583496095e-05,
      "model_forward_time": 1.7369539737701416,
      "step": 32975
    },
    {
      "epoch": 3.1447410583496095e-05,
      "step": 32975,
      "training_step_time": 7.649700164794922
    },
    {
      "epoch": 3.14483642578125e-05,
      "model_forward_time": 1.729604721069336,
      "step": 32976
    },
    {
      "epoch": 3.14483642578125e-05,
      "step": 32976,
      "training_step_time": 7.660094738006592
    },
    {
      "epoch": 3.14493179321289e-05,
      "model_forward_time": 1.738948106765747,
      "step": 32977
    },
    {
      "epoch": 3.14493179321289e-05,
      "step": 32977,
      "training_step_time": 8.957025527954102
    },
    {
      "epoch": 3.1450271606445314e-05,
      "model_forward_time": 1.3592495918273926,
      "step": 32978
    },
    {
      "epoch": 3.1450271606445314e-05,
      "step": 32978,
      "training_step_time": 9.116544723510742
    },
    {
      "epoch": 3.145122528076172e-05,
      "model_forward_time": 2.9170079231262207,
      "step": 32979
    },
    {
      "epoch": 3.145122528076172e-05,
      "step": 32979,
      "training_step_time": 8.500184774398804
    },
    {
      "epoch": 3.145217895507812e-05,
      "grad_norm": 0.1436150074005127,
      "learning_rate": 8.008180021268314e-06,
      "loss": 0.0593,
      "step": 32980
    },
    {
      "epoch": 3.145217895507812e-05,
      "model_forward_time": 1.3886702060699463,
      "step": 32980
    },
    {
      "dynamics_loss_avg": 0.0403848709538579,
      "epoch": 3.145217895507812e-05,
      "step": 32980
    },
    {
      "action_loss_avg": 0.0013039639161434025,
      "epoch": 3.145217895507812e-05,
      "step": 32980
    },
    {
      "epoch": 3.145217895507812e-05,
      "step": 32980,
      "training_step_time": 7.687469720840454
    },
    {
      "epoch": 3.145313262939453e-05,
      "model_forward_time": 1.749074935913086,
      "step": 32981
    },
    {
      "epoch": 3.145313262939453e-05,
      "step": 32981,
      "training_step_time": 7.55975604057312
    },
    {
      "epoch": 3.145408630371094e-05,
      "model_forward_time": 1.8608903884887695,
      "step": 32982
    },
    {
      "epoch": 3.145408630371094e-05,
      "step": 32982,
      "training_step_time": 8.990536212921143
    },
    {
      "epoch": 3.145503997802734e-05,
      "model_forward_time": 1.7625670433044434,
      "step": 32983
    },
    {
      "epoch": 3.145503997802734e-05,
      "step": 32983,
      "training_step_time": 8.308110475540161
    },
    {
      "epoch": 3.145599365234375e-05,
      "model_forward_time": 1.750647783279419,
      "step": 32984
    },
    {
      "epoch": 3.145599365234375e-05,
      "step": 32984,
      "training_step_time": 7.771600246429443
    },
    {
      "epoch": 3.1456947326660156e-05,
      "model_forward_time": 1.75313138961792,
      "step": 32985
    },
    {
      "epoch": 3.1456947326660156e-05,
      "step": 32985,
      "training_step_time": 8.410747528076172
    },
    {
      "epoch": 3.145790100097656e-05,
      "model_forward_time": 1.7528259754180908,
      "step": 32986
    },
    {
      "epoch": 3.145790100097656e-05,
      "step": 32986,
      "training_step_time": 7.761855840682983
    },
    {
      "epoch": 3.145885467529297e-05,
      "model_forward_time": 1.7508819103240967,
      "step": 32987
    },
    {
      "epoch": 3.145885467529297e-05,
      "step": 32987,
      "training_step_time": 7.62652587890625
    },
    {
      "epoch": 3.1459808349609375e-05,
      "model_forward_time": 2.861245632171631,
      "step": 32988
    },
    {
      "epoch": 3.1459808349609375e-05,
      "step": 32988,
      "training_step_time": 7.683165550231934
    },
    {
      "epoch": 3.146076202392578e-05,
      "model_forward_time": 1.7510771751403809,
      "step": 32989
    },
    {
      "epoch": 3.146076202392578e-05,
      "step": 32989,
      "training_step_time": 7.5488972663879395
    },
    {
      "epoch": 3.146171569824219e-05,
      "grad_norm": 0.15949460864067078,
      "learning_rate": 8.006859114094414e-06,
      "loss": 0.0606,
      "step": 32990
    },
    {
      "epoch": 3.146171569824219e-05,
      "model_forward_time": 1.7475674152374268,
      "step": 32990
    },
    {
      "dynamics_loss_avg": 0.05945359170436859,
      "epoch": 3.146171569824219e-05,
      "step": 32990
    },
    {
      "action_loss_avg": 0.0022682877024635673,
      "epoch": 3.146171569824219e-05,
      "step": 32990
    },
    {
      "epoch": 3.146171569824219e-05,
      "step": 32990,
      "training_step_time": 7.75896692276001
    },
    {
      "epoch": 3.1462669372558593e-05,
      "model_forward_time": 1.8311913013458252,
      "step": 32991
    },
    {
      "epoch": 3.1462669372558593e-05,
      "step": 32991,
      "training_step_time": 9.527034044265747
    },
    {
      "epoch": 3.1463623046875e-05,
      "model_forward_time": 1.7482068538665771,
      "step": 32992
    },
    {
      "epoch": 3.1463623046875e-05,
      "step": 32992,
      "training_step_time": 8.780791759490967
    },
    {
      "epoch": 3.146457672119141e-05,
      "model_forward_time": 1.7543482780456543,
      "step": 32993
    },
    {
      "epoch": 3.146457672119141e-05,
      "step": 32993,
      "training_step_time": 7.874877214431763
    },
    {
      "epoch": 3.146553039550781e-05,
      "model_forward_time": 1.750159740447998,
      "step": 32994
    },
    {
      "epoch": 3.146553039550781e-05,
      "step": 32994,
      "training_step_time": 7.553505897521973
    },
    {
      "epoch": 3.1466484069824216e-05,
      "model_forward_time": 1.750962495803833,
      "step": 32995
    },
    {
      "epoch": 3.1466484069824216e-05,
      "step": 32995,
      "training_step_time": 7.687204837799072
    },
    {
      "epoch": 3.146743774414063e-05,
      "model_forward_time": 1.757758617401123,
      "step": 32996
    },
    {
      "epoch": 3.146743774414063e-05,
      "step": 32996,
      "training_step_time": 7.710283517837524
    },
    {
      "epoch": 3.146839141845703e-05,
      "model_forward_time": 1.7634024620056152,
      "step": 32997
    },
    {
      "epoch": 3.146839141845703e-05,
      "step": 32997,
      "training_step_time": 9.254200458526611
    },
    {
      "epoch": 3.1469345092773435e-05,
      "model_forward_time": 1.7904794216156006,
      "step": 32998
    },
    {
      "epoch": 3.1469345092773435e-05,
      "step": 32998,
      "training_step_time": 7.802767038345337
    },
    {
      "epoch": 3.1470298767089846e-05,
      "model_forward_time": 1.7749629020690918,
      "step": 32999
    },
    {
      "epoch": 3.1470298767089846e-05,
      "step": 32999,
      "training_step_time": 7.759052991867065
    },
    {
      "epoch": 3.147125244140625e-05,
      "grad_norm": 0.12203436344861984,
      "learning_rate": 8.00553787809492e-06,
      "loss": 0.0563,
      "step": 33000
    },
    {
      "epoch": 3.147125244140625e-05,
      "model_forward_time": 1.5019888877868652,
      "step": 33000
    },
    {
      "dynamics_loss_avg": 0.048268067091703414,
      "epoch": 3.147125244140625e-05,
      "step": 33000
    },
    {
      "action_loss_avg": 0.0027228617487708106,
      "epoch": 3.147125244140625e-05,
      "step": 33000
    },
    {
      "epoch": 3.147125244140625e-05,
      "step": 33000,
      "training_step_time": 8.687861919403076
    },
    {
      "epoch": 3.1472206115722654e-05,
      "model_forward_time": 1.7500667572021484,
      "step": 33001
    },
    {
      "epoch": 3.1472206115722654e-05,
      "step": 33001,
      "training_step_time": 9.652044296264648
    },
    {
      "epoch": 3.1473159790039065e-05,
      "model_forward_time": 1.7471814155578613,
      "step": 33002
    },
    {
      "epoch": 3.1473159790039065e-05,
      "step": 33002,
      "training_step_time": 7.7418975830078125
    },
    {
      "epoch": 3.147411346435547e-05,
      "model_forward_time": 1.7492971420288086,
      "step": 33003
    },
    {
      "epoch": 3.147411346435547e-05,
      "step": 33003,
      "training_step_time": 8.01166582107544
    },
    {
      "epoch": 3.147506713867187e-05,
      "model_forward_time": 1.7558069229125977,
      "step": 33004
    },
    {
      "epoch": 3.147506713867187e-05,
      "step": 33004,
      "training_step_time": 7.541481018066406
    },
    {
      "epoch": 3.1476020812988284e-05,
      "model_forward_time": 1.7508437633514404,
      "step": 33005
    },
    {
      "epoch": 3.1476020812988284e-05,
      "step": 33005,
      "training_step_time": 9.49755048751831
    },
    {
      "epoch": 3.147697448730469e-05,
      "model_forward_time": 3.297930955886841,
      "step": 33006
    },
    {
      "epoch": 3.147697448730469e-05,
      "step": 33006,
      "training_step_time": 10.245846271514893
    },
    {
      "epoch": 3.147792816162109e-05,
      "model_forward_time": 1.7510309219360352,
      "step": 33007
    },
    {
      "epoch": 3.147792816162109e-05,
      "step": 33007,
      "training_step_time": 10.205200910568237
    },
    {
      "epoch": 3.14788818359375e-05,
      "model_forward_time": 1.7520411014556885,
      "step": 33008
    },
    {
      "epoch": 3.14788818359375e-05,
      "step": 33008,
      "training_step_time": 10.180301189422607
    },
    {
      "epoch": 3.1479835510253907e-05,
      "model_forward_time": 1.7549543380737305,
      "step": 33009
    },
    {
      "epoch": 3.1479835510253907e-05,
      "step": 33009,
      "training_step_time": 8.598511457443237
    },
    {
      "epoch": 3.148078918457031e-05,
      "grad_norm": 0.11266015470027924,
      "learning_rate": 8.004216313414324e-06,
      "loss": 0.0604,
      "step": 33010
    },
    {
      "epoch": 3.148078918457031e-05,
      "model_forward_time": 1.7362160682678223,
      "step": 33010
    },
    {
      "dynamics_loss_avg": 0.04592077042907476,
      "epoch": 3.148078918457031e-05,
      "step": 33010
    },
    {
      "action_loss_avg": 0.0020376179338200016,
      "epoch": 3.148078918457031e-05,
      "step": 33010
    },
    {
      "epoch": 3.148078918457031e-05,
      "step": 33010,
      "training_step_time": 6.834052324295044
    },
    {
      "epoch": 3.148174285888672e-05,
      "model_forward_time": 1.7497148513793945,
      "step": 33011
    },
    {
      "epoch": 3.148174285888672e-05,
      "step": 33011,
      "training_step_time": 6.792060136795044
    },
    {
      "epoch": 3.1482696533203125e-05,
      "model_forward_time": 1.3924167156219482,
      "step": 33012
    },
    {
      "epoch": 3.1482696533203125e-05,
      "step": 33012,
      "training_step_time": 6.783279895782471
    },
    {
      "epoch": 3.148365020751953e-05,
      "model_forward_time": 1.7531356811523438,
      "step": 33013
    },
    {
      "epoch": 3.148365020751953e-05,
      "step": 33013,
      "training_step_time": 6.85296630859375
    },
    {
      "epoch": 3.148460388183594e-05,
      "model_forward_time": 1.758850336074829,
      "step": 33014
    },
    {
      "epoch": 3.148460388183594e-05,
      "step": 33014,
      "training_step_time": 6.8170716762542725
    },
    {
      "epoch": 3.1485557556152344e-05,
      "model_forward_time": 1.7431530952453613,
      "step": 33015
    },
    {
      "epoch": 3.1485557556152344e-05,
      "step": 33015,
      "training_step_time": 6.788137435913086
    },
    {
      "epoch": 3.148651123046875e-05,
      "model_forward_time": 1.751357078552246,
      "step": 33016
    },
    {
      "epoch": 3.148651123046875e-05,
      "step": 33016,
      "training_step_time": 6.768741130828857
    },
    {
      "epoch": 3.148746490478516e-05,
      "model_forward_time": 1.7462332248687744,
      "step": 33017
    },
    {
      "epoch": 3.148746490478516e-05,
      "step": 33017,
      "training_step_time": 6.774525165557861
    },
    {
      "epoch": 3.148841857910156e-05,
      "model_forward_time": 2.844813823699951,
      "step": 33018
    },
    {
      "epoch": 3.148841857910156e-05,
      "step": 33018,
      "training_step_time": 11.164200067520142
    },
    {
      "epoch": 3.148937225341797e-05,
      "model_forward_time": 1.7518203258514404,
      "step": 33019
    },
    {
      "epoch": 3.148937225341797e-05,
      "step": 33019,
      "training_step_time": 7.707828521728516
    },
    {
      "epoch": 3.149032592773438e-05,
      "grad_norm": 0.15340235829353333,
      "learning_rate": 8.00289442019715e-06,
      "loss": 0.0634,
      "step": 33020
    },
    {
      "epoch": 3.149032592773438e-05,
      "model_forward_time": 1.733414888381958,
      "step": 33020
    },
    {
      "dynamics_loss_avg": 0.059294730238616464,
      "epoch": 3.149032592773438e-05,
      "step": 33020
    },
    {
      "action_loss_avg": 0.0020650329533964396,
      "epoch": 3.149032592773438e-05,
      "step": 33020
    },
    {
      "epoch": 3.149032592773438e-05,
      "step": 33020,
      "training_step_time": 6.757312059402466
    },
    {
      "epoch": 3.149127960205078e-05,
      "model_forward_time": 1.7441973686218262,
      "step": 33021
    },
    {
      "epoch": 3.149127960205078e-05,
      "step": 33021,
      "training_step_time": 6.761774301528931
    },
    {
      "epoch": 3.1492233276367186e-05,
      "model_forward_time": 1.393540382385254,
      "step": 33022
    },
    {
      "epoch": 3.1492233276367186e-05,
      "step": 33022,
      "training_step_time": 6.75171685218811
    },
    {
      "epoch": 3.14931869506836e-05,
      "model_forward_time": 1.7500457763671875,
      "step": 33023
    },
    {
      "epoch": 3.14931869506836e-05,
      "step": 33023,
      "training_step_time": 6.826967477798462
    },
    {
      "epoch": 3.1494140625e-05,
      "model_forward_time": 1.749645471572876,
      "step": 33024
    },
    {
      "epoch": 3.1494140625e-05,
      "step": 33024,
      "training_step_time": 6.753984689712524
    },
    {
      "epoch": 3.1495094299316405e-05,
      "model_forward_time": 1.7495684623718262,
      "step": 33025
    },
    {
      "epoch": 3.1495094299316405e-05,
      "step": 33025,
      "training_step_time": 6.768404722213745
    },
    {
      "epoch": 3.1496047973632816e-05,
      "model_forward_time": 1.7455635070800781,
      "step": 33026
    },
    {
      "epoch": 3.1496047973632816e-05,
      "step": 33026,
      "training_step_time": 6.822961807250977
    },
    {
      "epoch": 3.149700164794922e-05,
      "model_forward_time": 2.842957019805908,
      "step": 33027
    },
    {
      "epoch": 3.149700164794922e-05,
      "step": 33027,
      "training_step_time": 8.46731185913086
    },
    {
      "epoch": 3.1497955322265624e-05,
      "model_forward_time": 1.3930494785308838,
      "step": 33028
    },
    {
      "epoch": 3.1497955322265624e-05,
      "step": 33028,
      "training_step_time": 7.934164762496948
    },
    {
      "epoch": 3.1498908996582035e-05,
      "model_forward_time": 1.7444584369659424,
      "step": 33029
    },
    {
      "epoch": 3.1498908996582035e-05,
      "step": 33029,
      "training_step_time": 6.809505462646484
    },
    {
      "epoch": 3.149986267089844e-05,
      "grad_norm": 0.14281760156154633,
      "learning_rate": 8.001572198587955e-06,
      "loss": 0.0605,
      "step": 33030
    },
    {
      "epoch": 3.149986267089844e-05,
      "model_forward_time": 1.7373971939086914,
      "step": 33030
    },
    {
      "dynamics_loss_avg": 0.05542207397520542,
      "epoch": 3.149986267089844e-05,
      "step": 33030
    },
    {
      "action_loss_avg": 0.0024826412787660956,
      "epoch": 3.149986267089844e-05,
      "step": 33030
    },
    {
      "epoch": 3.149986267089844e-05,
      "step": 33030,
      "training_step_time": 6.844323635101318
    },
    {
      "epoch": 3.150081634521484e-05,
      "model_forward_time": 1.0590252876281738,
      "step": 33031
    },
    {
      "epoch": 3.150081634521484e-05,
      "step": 33031,
      "training_step_time": 6.7985546588897705
    },
    {
      "epoch": 3.150177001953125e-05,
      "model_forward_time": 1.772568702697754,
      "step": 33032
    },
    {
      "epoch": 3.150177001953125e-05,
      "step": 33032,
      "training_step_time": 6.854128122329712
    },
    {
      "epoch": 3.150272369384766e-05,
      "model_forward_time": 1.7496006488800049,
      "step": 33033
    },
    {
      "epoch": 3.150272369384766e-05,
      "step": 33033,
      "training_step_time": 6.899848699569702
    },
    {
      "epoch": 3.150367736816406e-05,
      "model_forward_time": 1.7532086372375488,
      "step": 33034
    },
    {
      "epoch": 3.150367736816406e-05,
      "step": 33034,
      "training_step_time": 6.854009628295898
    },
    {
      "epoch": 3.1504631042480465e-05,
      "model_forward_time": 1.7377455234527588,
      "step": 33035
    },
    {
      "epoch": 3.1504631042480465e-05,
      "step": 33035,
      "training_step_time": 6.843006372451782
    },
    {
      "epoch": 3.1505584716796876e-05,
      "model_forward_time": 1.7966182231903076,
      "step": 33036
    },
    {
      "epoch": 3.1505584716796876e-05,
      "step": 33036,
      "training_step_time": 7.758191823959351
    },
    {
      "epoch": 3.150653839111328e-05,
      "model_forward_time": 2.915386199951172,
      "step": 33037
    },
    {
      "epoch": 3.150653839111328e-05,
      "step": 33037,
      "training_step_time": 8.960368156433105
    },
    {
      "epoch": 3.1507492065429684e-05,
      "model_forward_time": 1.7516953945159912,
      "step": 33038
    },
    {
      "epoch": 3.1507492065429684e-05,
      "step": 33038,
      "training_step_time": 7.546659708023071
    },
    {
      "epoch": 3.1508445739746095e-05,
      "model_forward_time": 1.7598962783813477,
      "step": 33039
    },
    {
      "epoch": 3.1508445739746095e-05,
      "step": 33039,
      "training_step_time": 6.820575952529907
    },
    {
      "epoch": 3.15093994140625e-05,
      "grad_norm": 0.13108298182487488,
      "learning_rate": 8.000249648731337e-06,
      "loss": 0.0623,
      "step": 33040
    },
    {
      "epoch": 3.15093994140625e-05,
      "model_forward_time": 1.749634027481079,
      "step": 33040
    },
    {
      "dynamics_loss_avg": 0.0570076173171401,
      "epoch": 3.15093994140625e-05,
      "step": 33040
    },
    {
      "action_loss_avg": 0.0034892115014372393,
      "epoch": 3.15093994140625e-05,
      "step": 33040
    },
    {
      "epoch": 3.15093994140625e-05,
      "step": 33040,
      "training_step_time": 6.826122760772705
    },
    {
      "epoch": 3.15103530883789e-05,
      "model_forward_time": 1.7514631748199463,
      "step": 33041
    },
    {
      "epoch": 3.15103530883789e-05,
      "step": 33041,
      "training_step_time": 6.780355215072632
    },
    {
      "epoch": 3.1511306762695314e-05,
      "model_forward_time": 1.753220796585083,
      "step": 33042
    },
    {
      "epoch": 3.1511306762695314e-05,
      "step": 33042,
      "training_step_time": 6.799952268600464
    },
    {
      "epoch": 3.151226043701172e-05,
      "model_forward_time": 1.759343147277832,
      "step": 33043
    },
    {
      "epoch": 3.151226043701172e-05,
      "step": 33043,
      "training_step_time": 6.870595693588257
    },
    {
      "epoch": 3.151321411132812e-05,
      "model_forward_time": 1.745840311050415,
      "step": 33044
    },
    {
      "epoch": 3.151321411132812e-05,
      "step": 33044,
      "training_step_time": 6.87475848197937
    },
    {
      "epoch": 3.151416778564453e-05,
      "model_forward_time": 1.7746148109436035,
      "step": 33045
    },
    {
      "epoch": 3.151416778564453e-05,
      "step": 33045,
      "training_step_time": 7.720067977905273
    },
    {
      "epoch": 3.151512145996094e-05,
      "model_forward_time": 2.8866546154022217,
      "step": 33046
    },
    {
      "epoch": 3.151512145996094e-05,
      "step": 33046,
      "training_step_time": 14.320966720581055
    },
    {
      "epoch": 3.151607513427734e-05,
      "model_forward_time": 1.7496991157531738,
      "step": 33047
    },
    {
      "epoch": 3.151607513427734e-05,
      "step": 33047,
      "training_step_time": 7.8564794063568115
    },
    {
      "epoch": 3.151702880859375e-05,
      "model_forward_time": 1.7612848281860352,
      "step": 33048
    },
    {
      "epoch": 3.151702880859375e-05,
      "step": 33048,
      "training_step_time": 6.866488218307495
    },
    {
      "epoch": 3.1517982482910156e-05,
      "model_forward_time": 1.7536025047302246,
      "step": 33049
    },
    {
      "epoch": 3.1517982482910156e-05,
      "step": 33049,
      "training_step_time": 6.821723937988281
    },
    {
      "epoch": 3.151893615722656e-05,
      "grad_norm": 0.19316934049129486,
      "learning_rate": 7.998926770771928e-06,
      "loss": 0.0596,
      "step": 33050
    },
    {
      "epoch": 3.151893615722656e-05,
      "model_forward_time": 1.7598490715026855,
      "step": 33050
    },
    {
      "dynamics_loss_avg": 0.04908069260418415,
      "epoch": 3.151893615722656e-05,
      "step": 33050
    },
    {
      "action_loss_avg": 0.002091898996150121,
      "epoch": 3.151893615722656e-05,
      "step": 33050
    },
    {
      "epoch": 3.151893615722656e-05,
      "step": 33050,
      "training_step_time": 6.923431158065796
    },
    {
      "epoch": 3.151988983154297e-05,
      "model_forward_time": 1.7468385696411133,
      "step": 33051
    },
    {
      "epoch": 3.151988983154297e-05,
      "step": 33051,
      "training_step_time": 6.842345476150513
    },
    {
      "epoch": 3.1520843505859375e-05,
      "model_forward_time": 1.7459156513214111,
      "step": 33052
    },
    {
      "epoch": 3.1520843505859375e-05,
      "step": 33052,
      "training_step_time": 6.714294195175171
    },
    {
      "epoch": 3.152179718017578e-05,
      "model_forward_time": 1.746727705001831,
      "step": 33053
    },
    {
      "epoch": 3.152179718017578e-05,
      "step": 33053,
      "training_step_time": 6.899957180023193
    },
    {
      "epoch": 3.152275085449219e-05,
      "model_forward_time": 1.754516363143921,
      "step": 33054
    },
    {
      "epoch": 3.152275085449219e-05,
      "step": 33054,
      "training_step_time": 7.751126766204834
    },
    {
      "epoch": 3.1523704528808593e-05,
      "model_forward_time": 2.860964775085449,
      "step": 33055
    },
    {
      "epoch": 3.1523704528808593e-05,
      "step": 33055,
      "training_step_time": 10.3490309715271
    },
    {
      "epoch": 3.1524658203125e-05,
      "model_forward_time": 1.7773056030273438,
      "step": 33056
    },
    {
      "epoch": 3.1524658203125e-05,
      "step": 33056,
      "training_step_time": 7.772610425949097
    },
    {
      "epoch": 3.152561187744141e-05,
      "model_forward_time": 1.7650585174560547,
      "step": 33057
    },
    {
      "epoch": 3.152561187744141e-05,
      "step": 33057,
      "training_step_time": 7.719764709472656
    },
    {
      "epoch": 3.152656555175781e-05,
      "model_forward_time": 1.7527313232421875,
      "step": 33058
    },
    {
      "epoch": 3.152656555175781e-05,
      "step": 33058,
      "training_step_time": 6.858522653579712
    },
    {
      "epoch": 3.1527519226074216e-05,
      "model_forward_time": 1.757582664489746,
      "step": 33059
    },
    {
      "epoch": 3.1527519226074216e-05,
      "step": 33059,
      "training_step_time": 6.818922996520996
    },
    {
      "epoch": 3.152847290039063e-05,
      "grad_norm": 0.13406719267368317,
      "learning_rate": 7.997603564854396e-06,
      "loss": 0.0597,
      "step": 33060
    },
    {
      "epoch": 3.152847290039063e-05,
      "model_forward_time": 1.7518632411956787,
      "step": 33060
    },
    {
      "dynamics_loss_avg": 0.04665474537760019,
      "epoch": 3.152847290039063e-05,
      "step": 33060
    },
    {
      "action_loss_avg": 0.0012325786985456944,
      "epoch": 3.152847290039063e-05,
      "step": 33060
    },
    {
      "epoch": 3.152847290039063e-05,
      "step": 33060,
      "training_step_time": 6.863966941833496
    },
    {
      "epoch": 3.152942657470703e-05,
      "model_forward_time": 1.7454261779785156,
      "step": 33061
    },
    {
      "epoch": 3.152942657470703e-05,
      "step": 33061,
      "training_step_time": 6.810842037200928
    },
    {
      "epoch": 3.1530380249023435e-05,
      "model_forward_time": 1.7465848922729492,
      "step": 33062
    },
    {
      "epoch": 3.1530380249023435e-05,
      "step": 33062,
      "training_step_time": 6.855842113494873
    },
    {
      "epoch": 3.1531333923339846e-05,
      "model_forward_time": 1.758366346359253,
      "step": 33063
    },
    {
      "epoch": 3.1531333923339846e-05,
      "step": 33063,
      "training_step_time": 7.59297251701355
    },
    {
      "epoch": 3.153228759765625e-05,
      "model_forward_time": 2.872591495513916,
      "step": 33064
    },
    {
      "epoch": 3.153228759765625e-05,
      "step": 33064,
      "training_step_time": 10.647136211395264
    },
    {
      "epoch": 3.1533241271972654e-05,
      "model_forward_time": 1.7467436790466309,
      "step": 33065
    },
    {
      "epoch": 3.1533241271972654e-05,
      "step": 33065,
      "training_step_time": 7.933787822723389
    },
    {
      "epoch": 3.1534194946289065e-05,
      "model_forward_time": 1.7527556419372559,
      "step": 33066
    },
    {
      "epoch": 3.1534194946289065e-05,
      "step": 33066,
      "training_step_time": 7.679271459579468
    },
    {
      "epoch": 3.153514862060547e-05,
      "model_forward_time": 1.751054048538208,
      "step": 33067
    },
    {
      "epoch": 3.153514862060547e-05,
      "step": 33067,
      "training_step_time": 6.860801696777344
    },
    {
      "epoch": 3.153610229492187e-05,
      "model_forward_time": 1.752267599105835,
      "step": 33068
    },
    {
      "epoch": 3.153610229492187e-05,
      "step": 33068,
      "training_step_time": 6.818074703216553
    },
    {
      "epoch": 3.1537055969238284e-05,
      "model_forward_time": 1.7516272068023682,
      "step": 33069
    },
    {
      "epoch": 3.1537055969238284e-05,
      "step": 33069,
      "training_step_time": 6.838234186172485
    },
    {
      "epoch": 3.153800964355469e-05,
      "grad_norm": 0.1496179699897766,
      "learning_rate": 7.996280031123448e-06,
      "loss": 0.0583,
      "step": 33070
    },
    {
      "epoch": 3.153800964355469e-05,
      "model_forward_time": 1.7359950542449951,
      "step": 33070
    },
    {
      "dynamics_loss_avg": 0.06214559692889452,
      "epoch": 3.153800964355469e-05,
      "step": 33070
    },
    {
      "action_loss_avg": 0.002228624693816528,
      "epoch": 3.153800964355469e-05,
      "step": 33070
    },
    {
      "epoch": 3.153800964355469e-05,
      "step": 33070,
      "training_step_time": 6.783077716827393
    },
    {
      "epoch": 3.153896331787109e-05,
      "model_forward_time": 1.7530958652496338,
      "step": 33071
    },
    {
      "epoch": 3.153896331787109e-05,
      "step": 33071,
      "training_step_time": 6.811744689941406
    },
    {
      "epoch": 3.15399169921875e-05,
      "model_forward_time": 1.7412092685699463,
      "step": 33072
    },
    {
      "epoch": 3.15399169921875e-05,
      "step": 33072,
      "training_step_time": 7.331405401229858
    },
    {
      "epoch": 3.154087066650391e-05,
      "model_forward_time": 1.8666753768920898,
      "step": 33073
    },
    {
      "epoch": 3.154087066650391e-05,
      "step": 33073,
      "training_step_time": 8.723054885864258
    },
    {
      "epoch": 3.154182434082031e-05,
      "model_forward_time": 1.7493386268615723,
      "step": 33074
    },
    {
      "epoch": 3.154182434082031e-05,
      "step": 33074,
      "training_step_time": 9.733302116394043
    },
    {
      "epoch": 3.154277801513672e-05,
      "model_forward_time": 1.7516624927520752,
      "step": 33075
    },
    {
      "epoch": 3.154277801513672e-05,
      "step": 33075,
      "training_step_time": 7.789464235305786
    },
    {
      "epoch": 3.1543731689453125e-05,
      "model_forward_time": 1.745563268661499,
      "step": 33076
    },
    {
      "epoch": 3.1543731689453125e-05,
      "step": 33076,
      "training_step_time": 6.744497299194336
    },
    {
      "epoch": 3.154468536376953e-05,
      "model_forward_time": 1.746922492980957,
      "step": 33077
    },
    {
      "epoch": 3.154468536376953e-05,
      "step": 33077,
      "training_step_time": 6.879010438919067
    },
    {
      "epoch": 3.154563903808594e-05,
      "model_forward_time": 1.7590117454528809,
      "step": 33078
    },
    {
      "epoch": 3.154563903808594e-05,
      "step": 33078,
      "training_step_time": 6.827307462692261
    },
    {
      "epoch": 3.1546592712402344e-05,
      "model_forward_time": 1.7611241340637207,
      "step": 33079
    },
    {
      "epoch": 3.1546592712402344e-05,
      "step": 33079,
      "training_step_time": 6.822541952133179
    },
    {
      "epoch": 3.154754638671875e-05,
      "grad_norm": 0.13374505937099457,
      "learning_rate": 7.994956169723817e-06,
      "loss": 0.0615,
      "step": 33080
    },
    {
      "epoch": 3.154754638671875e-05,
      "model_forward_time": 1.7435264587402344,
      "step": 33080
    },
    {
      "dynamics_loss_avg": 0.055064668878912926,
      "epoch": 3.154754638671875e-05,
      "step": 33080
    },
    {
      "action_loss_avg": 0.0028666175901889803,
      "epoch": 3.154754638671875e-05,
      "step": 33080
    },
    {
      "epoch": 3.154754638671875e-05,
      "step": 33080,
      "training_step_time": 6.827788829803467
    },
    {
      "epoch": 3.154850006103516e-05,
      "model_forward_time": 1.750783920288086,
      "step": 33081
    },
    {
      "epoch": 3.154850006103516e-05,
      "step": 33081,
      "training_step_time": 6.769723892211914
    },
    {
      "epoch": 3.154945373535156e-05,
      "model_forward_time": 1.8260445594787598,
      "step": 33082
    },
    {
      "epoch": 3.154945373535156e-05,
      "step": 33082,
      "training_step_time": 7.649186372756958
    },
    {
      "epoch": 3.155040740966797e-05,
      "model_forward_time": 2.9288079738616943,
      "step": 33083
    },
    {
      "epoch": 3.155040740966797e-05,
      "step": 33083,
      "training_step_time": 10.019373893737793
    },
    {
      "epoch": 3.155136108398438e-05,
      "model_forward_time": 1.7691559791564941,
      "step": 33084
    },
    {
      "epoch": 3.155136108398438e-05,
      "step": 33084,
      "training_step_time": 7.775177001953125
    },
    {
      "epoch": 3.155231475830078e-05,
      "model_forward_time": 1.7724392414093018,
      "step": 33085
    },
    {
      "epoch": 3.155231475830078e-05,
      "step": 33085,
      "training_step_time": 7.5917887687683105
    },
    {
      "epoch": 3.1553268432617186e-05,
      "model_forward_time": 1.7638561725616455,
      "step": 33086
    },
    {
      "epoch": 3.1553268432617186e-05,
      "step": 33086,
      "training_step_time": 6.753741502761841
    },
    {
      "epoch": 3.15542221069336e-05,
      "model_forward_time": 1.7582945823669434,
      "step": 33087
    },
    {
      "epoch": 3.15542221069336e-05,
      "step": 33087,
      "training_step_time": 6.726742744445801
    },
    {
      "epoch": 3.155517578125e-05,
      "model_forward_time": 1.759078025817871,
      "step": 33088
    },
    {
      "epoch": 3.155517578125e-05,
      "step": 33088,
      "training_step_time": 6.73195481300354
    },
    {
      "epoch": 3.1556129455566405e-05,
      "model_forward_time": 1.7570617198944092,
      "step": 33089
    },
    {
      "epoch": 3.1556129455566405e-05,
      "step": 33089,
      "training_step_time": 6.787081718444824
    },
    {
      "epoch": 3.1557083129882816e-05,
      "grad_norm": 0.10826018452644348,
      "learning_rate": 7.993631980800286e-06,
      "loss": 0.0591,
      "step": 33090
    },
    {
      "epoch": 3.1557083129882816e-05,
      "model_forward_time": 1.7649896144866943,
      "step": 33090
    },
    {
      "dynamics_loss_avg": 0.05215889783576131,
      "epoch": 3.1557083129882816e-05,
      "step": 33090
    },
    {
      "action_loss_avg": 0.003301660972647369,
      "epoch": 3.1557083129882816e-05,
      "step": 33090
    },
    {
      "epoch": 3.1557083129882816e-05,
      "step": 33090,
      "training_step_time": 6.708427667617798
    },
    {
      "epoch": 3.155803680419922e-05,
      "model_forward_time": 1.8291375637054443,
      "step": 33091
    },
    {
      "epoch": 3.155803680419922e-05,
      "step": 33091,
      "training_step_time": 7.596439838409424
    },
    {
      "epoch": 3.1558990478515624e-05,
      "model_forward_time": 2.916637420654297,
      "step": 33092
    },
    {
      "epoch": 3.1558990478515624e-05,
      "step": 33092,
      "training_step_time": 12.636780023574829
    },
    {
      "epoch": 3.1559944152832035e-05,
      "model_forward_time": 1.767385482788086,
      "step": 33093
    },
    {
      "epoch": 3.1559944152832035e-05,
      "step": 33093,
      "training_step_time": 8.284239768981934
    },
    {
      "epoch": 3.156089782714844e-05,
      "model_forward_time": 1.761927843093872,
      "step": 33094
    },
    {
      "epoch": 3.156089782714844e-05,
      "step": 33094,
      "training_step_time": 7.54140305519104
    },
    {
      "epoch": 3.156185150146484e-05,
      "model_forward_time": 1.7602362632751465,
      "step": 33095
    },
    {
      "epoch": 3.156185150146484e-05,
      "step": 33095,
      "training_step_time": 6.75035834312439
    },
    {
      "epoch": 3.156280517578125e-05,
      "model_forward_time": 1.7646329402923584,
      "step": 33096
    },
    {
      "epoch": 3.156280517578125e-05,
      "step": 33096,
      "training_step_time": 6.755302667617798
    },
    {
      "epoch": 3.156375885009766e-05,
      "model_forward_time": 1.7549002170562744,
      "step": 33097
    },
    {
      "epoch": 3.156375885009766e-05,
      "step": 33097,
      "training_step_time": 6.67846417427063
    },
    {
      "epoch": 3.156471252441406e-05,
      "model_forward_time": 1.7578785419464111,
      "step": 33098
    },
    {
      "epoch": 3.156471252441406e-05,
      "step": 33098,
      "training_step_time": 6.785490036010742
    },
    {
      "epoch": 3.1565666198730466e-05,
      "model_forward_time": 1.749079942703247,
      "step": 33099
    },
    {
      "epoch": 3.1565666198730466e-05,
      "step": 33099,
      "training_step_time": 6.70929479598999
    },
    {
      "epoch": 3.1566619873046876e-05,
      "grad_norm": 0.12198858708143234,
      "learning_rate": 7.99230746449766e-06,
      "loss": 0.0602,
      "step": 33100
    },
    {
      "epoch": 3.1566619873046876e-05,
      "model_forward_time": 1.7808294296264648,
      "step": 33100
    },
    {
      "dynamics_loss_avg": 0.06150230951607227,
      "epoch": 3.1566619873046876e-05,
      "step": 33100
    },
    {
      "action_loss_avg": 0.0021189925086218865,
      "epoch": 3.1566619873046876e-05,
      "step": 33100
    },
    {
      "epoch": 3.1566619873046876e-05,
      "step": 33100,
      "training_step_time": 7.57387900352478
    },
    {
      "epoch": 3.156757354736328e-05,
      "model_forward_time": 2.8720829486846924,
      "step": 33101
    },
    {
      "epoch": 3.156757354736328e-05,
      "step": 33101,
      "training_step_time": 10.269133806228638
    },
    {
      "epoch": 3.1568527221679684e-05,
      "model_forward_time": 1.75254225730896,
      "step": 33102
    },
    {
      "epoch": 3.1568527221679684e-05,
      "step": 33102,
      "training_step_time": 7.717133522033691
    },
    {
      "epoch": 3.1569480895996095e-05,
      "model_forward_time": 1.7524340152740479,
      "step": 33103
    },
    {
      "epoch": 3.1569480895996095e-05,
      "step": 33103,
      "training_step_time": 7.5719358921051025
    },
    {
      "epoch": 3.15704345703125e-05,
      "model_forward_time": 1.7515883445739746,
      "step": 33104
    },
    {
      "epoch": 3.15704345703125e-05,
      "step": 33104,
      "training_step_time": 7.5203704833984375
    },
    {
      "epoch": 3.15713882446289e-05,
      "model_forward_time": 1.7556846141815186,
      "step": 33105
    },
    {
      "epoch": 3.15713882446289e-05,
      "step": 33105,
      "training_step_time": 6.711533069610596
    },
    {
      "epoch": 3.1572341918945314e-05,
      "model_forward_time": 1.760204553604126,
      "step": 33106
    },
    {
      "epoch": 3.1572341918945314e-05,
      "step": 33106,
      "training_step_time": 6.7952563762664795
    },
    {
      "epoch": 3.157329559326172e-05,
      "model_forward_time": 1.7580671310424805,
      "step": 33107
    },
    {
      "epoch": 3.157329559326172e-05,
      "step": 33107,
      "training_step_time": 6.749065160751343
    },
    {
      "epoch": 3.157424926757812e-05,
      "model_forward_time": 1.7388718128204346,
      "step": 33108
    },
    {
      "epoch": 3.157424926757812e-05,
      "step": 33108,
      "training_step_time": 6.779431104660034
    },
    {
      "epoch": 3.157520294189453e-05,
      "model_forward_time": 1.7760143280029297,
      "step": 33109
    },
    {
      "epoch": 3.157520294189453e-05,
      "step": 33109,
      "training_step_time": 7.686282396316528
    },
    {
      "epoch": 3.157615661621094e-05,
      "grad_norm": 0.14473144710063934,
      "learning_rate": 7.990982620960788e-06,
      "loss": 0.059,
      "step": 33110
    },
    {
      "epoch": 3.157615661621094e-05,
      "model_forward_time": 2.9016225337982178,
      "step": 33110
    },
    {
      "dynamics_loss_avg": 0.06839058715850115,
      "epoch": 3.157615661621094e-05,
      "step": 33110
    },
    {
      "action_loss_avg": 0.0019672059308504684,
      "epoch": 3.157615661621094e-05,
      "step": 33110
    },
    {
      "epoch": 3.157615661621094e-05,
      "step": 33110,
      "training_step_time": 7.7655510902404785
    },
    {
      "epoch": 3.157711029052734e-05,
      "model_forward_time": 1.7344105243682861,
      "step": 33111
    },
    {
      "epoch": 3.157711029052734e-05,
      "step": 33111,
      "training_step_time": 9.514671802520752
    },
    {
      "epoch": 3.157806396484375e-05,
      "model_forward_time": 1.7296936511993408,
      "step": 33112
    },
    {
      "epoch": 3.157806396484375e-05,
      "step": 33112,
      "training_step_time": 7.669212341308594
    },
    {
      "epoch": 3.1579017639160156e-05,
      "model_forward_time": 1.7430081367492676,
      "step": 33113
    },
    {
      "epoch": 3.1579017639160156e-05,
      "step": 33113,
      "training_step_time": 7.582151174545288
    },
    {
      "epoch": 3.157997131347656e-05,
      "model_forward_time": 1.3682937622070312,
      "step": 33114
    },
    {
      "epoch": 3.157997131347656e-05,
      "step": 33114,
      "training_step_time": 6.810454845428467
    },
    {
      "epoch": 3.158092498779297e-05,
      "model_forward_time": 1.7470669746398926,
      "step": 33115
    },
    {
      "epoch": 3.158092498779297e-05,
      "step": 33115,
      "training_step_time": 6.705650091171265
    },
    {
      "epoch": 3.1581878662109375e-05,
      "model_forward_time": 1.7409687042236328,
      "step": 33116
    },
    {
      "epoch": 3.1581878662109375e-05,
      "step": 33116,
      "training_step_time": 6.746931076049805
    },
    {
      "epoch": 3.158283233642578e-05,
      "model_forward_time": 1.742060661315918,
      "step": 33117
    },
    {
      "epoch": 3.158283233642578e-05,
      "step": 33117,
      "training_step_time": 6.764037370681763
    },
    {
      "epoch": 3.158378601074219e-05,
      "model_forward_time": 1.7647359371185303,
      "step": 33118
    },
    {
      "epoch": 3.158378601074219e-05,
      "step": 33118,
      "training_step_time": 7.719051837921143
    },
    {
      "epoch": 3.1584739685058594e-05,
      "model_forward_time": 2.81796932220459,
      "step": 33119
    },
    {
      "epoch": 3.1584739685058594e-05,
      "step": 33119,
      "training_step_time": 7.59241509437561
    },
    {
      "epoch": 3.1585693359375e-05,
      "grad_norm": 0.1386270523071289,
      "learning_rate": 7.989657450334555e-06,
      "loss": 0.0595,
      "step": 33120
    },
    {
      "epoch": 3.1585693359375e-05,
      "model_forward_time": 1.752253770828247,
      "step": 33120
    },
    {
      "dynamics_loss_avg": 0.05765892565250397,
      "epoch": 3.1585693359375e-05,
      "step": 33120
    },
    {
      "action_loss_avg": 0.002537673490587622,
      "epoch": 3.1585693359375e-05,
      "step": 33120
    },
    {
      "epoch": 3.1585693359375e-05,
      "step": 33120,
      "training_step_time": 8.307790994644165
    },
    {
      "epoch": 3.158664703369141e-05,
      "model_forward_time": 1.7428667545318604,
      "step": 33121
    },
    {
      "epoch": 3.158664703369141e-05,
      "step": 33121,
      "training_step_time": 7.65147590637207
    },
    {
      "epoch": 3.158760070800781e-05,
      "model_forward_time": 1.737381935119629,
      "step": 33122
    },
    {
      "epoch": 3.158760070800781e-05,
      "step": 33122,
      "training_step_time": 7.596966505050659
    },
    {
      "epoch": 3.1588554382324216e-05,
      "model_forward_time": 1.7413415908813477,
      "step": 33123
    },
    {
      "epoch": 3.1588554382324216e-05,
      "step": 33123,
      "training_step_time": 6.776618957519531
    },
    {
      "epoch": 3.158950805664063e-05,
      "model_forward_time": 1.3706648349761963,
      "step": 33124
    },
    {
      "epoch": 3.158950805664063e-05,
      "step": 33124,
      "training_step_time": 6.771722316741943
    },
    {
      "epoch": 3.159046173095703e-05,
      "model_forward_time": 1.7314636707305908,
      "step": 33125
    },
    {
      "epoch": 3.159046173095703e-05,
      "step": 33125,
      "training_step_time": 6.868736743927002
    },
    {
      "epoch": 3.1591415405273435e-05,
      "model_forward_time": 1.7453460693359375,
      "step": 33126
    },
    {
      "epoch": 3.1591415405273435e-05,
      "step": 33126,
      "training_step_time": 6.762127876281738
    },
    {
      "epoch": 3.1592369079589846e-05,
      "model_forward_time": 1.4013047218322754,
      "step": 33127
    },
    {
      "epoch": 3.1592369079589846e-05,
      "step": 33127,
      "training_step_time": 7.690295457839966
    },
    {
      "epoch": 3.159332275390625e-05,
      "model_forward_time": 1.7656521797180176,
      "step": 33128
    },
    {
      "epoch": 3.159332275390625e-05,
      "step": 33128,
      "training_step_time": 8.655730962753296
    },
    {
      "epoch": 3.1594276428222654e-05,
      "model_forward_time": 2.87447190284729,
      "step": 33129
    },
    {
      "epoch": 3.1594276428222654e-05,
      "step": 33129,
      "training_step_time": 10.057239294052124
    },
    {
      "epoch": 3.1595230102539065e-05,
      "grad_norm": 0.15988165140151978,
      "learning_rate": 7.988331952763876e-06,
      "loss": 0.0538,
      "step": 33130
    },
    {
      "epoch": 3.1595230102539065e-05,
      "model_forward_time": 1.7248196601867676,
      "step": 33130
    },
    {
      "dynamics_loss_avg": 0.056883531250059606,
      "epoch": 3.1595230102539065e-05,
      "step": 33130
    },
    {
      "action_loss_avg": 0.0019513777690008283,
      "epoch": 3.1595230102539065e-05,
      "step": 33130
    },
    {
      "epoch": 3.1595230102539065e-05,
      "step": 33130,
      "training_step_time": 8.286829948425293
    },
    {
      "epoch": 3.159618377685547e-05,
      "model_forward_time": 1.374912977218628,
      "step": 33131
    },
    {
      "epoch": 3.159618377685547e-05,
      "step": 33131,
      "training_step_time": 7.58233380317688
    },
    {
      "epoch": 3.159713745117187e-05,
      "model_forward_time": 1.0278310775756836,
      "step": 33132
    },
    {
      "epoch": 3.159713745117187e-05,
      "step": 33132,
      "training_step_time": 6.79071831703186
    },
    {
      "epoch": 3.1598091125488284e-05,
      "model_forward_time": 1.7312045097351074,
      "step": 33133
    },
    {
      "epoch": 3.1598091125488284e-05,
      "step": 33133,
      "training_step_time": 6.83781361579895
    },
    {
      "epoch": 3.159904479980469e-05,
      "model_forward_time": 1.3707540035247803,
      "step": 33134
    },
    {
      "epoch": 3.159904479980469e-05,
      "step": 33134,
      "training_step_time": 6.891303062438965
    },
    {
      "epoch": 3.159999847412109e-05,
      "model_forward_time": 1.723134994506836,
      "step": 33135
    },
    {
      "epoch": 3.159999847412109e-05,
      "step": 33135,
      "training_step_time": 6.837002754211426
    },
    {
      "epoch": 3.16009521484375e-05,
      "model_forward_time": 1.7686481475830078,
      "step": 33136
    },
    {
      "epoch": 3.16009521484375e-05,
      "step": 33136,
      "training_step_time": 8.491518020629883
    },
    {
      "epoch": 3.160190582275391e-05,
      "model_forward_time": 1.7573485374450684,
      "step": 33137
    },
    {
      "epoch": 3.160190582275391e-05,
      "step": 33137,
      "training_step_time": 7.607757806777954
    },
    {
      "epoch": 3.160285949707031e-05,
      "model_forward_time": 1.7420990467071533,
      "step": 33138
    },
    {
      "epoch": 3.160285949707031e-05,
      "step": 33138,
      "training_step_time": 9.701820850372314
    },
    {
      "epoch": 3.160381317138672e-05,
      "model_forward_time": 2.948826313018799,
      "step": 33139
    },
    {
      "epoch": 3.160381317138672e-05,
      "step": 33139,
      "training_step_time": 7.789661645889282
    },
    {
      "epoch": 3.1604766845703126e-05,
      "grad_norm": 0.1589389443397522,
      "learning_rate": 7.98700612839371e-06,
      "loss": 0.0523,
      "step": 33140
    },
    {
      "epoch": 3.1604766845703126e-05,
      "model_forward_time": 1.7475526332855225,
      "step": 33140
    },
    {
      "dynamics_loss_avg": 0.05432748757302761,
      "epoch": 3.1604766845703126e-05,
      "step": 33140
    },
    {
      "action_loss_avg": 0.002292399341240525,
      "epoch": 3.1604766845703126e-05,
      "step": 33140
    },
    {
      "epoch": 3.1604766845703126e-05,
      "step": 33140,
      "training_step_time": 7.616074323654175
    },
    {
      "epoch": 3.160572052001953e-05,
      "model_forward_time": 1.742182731628418,
      "step": 33141
    },
    {
      "epoch": 3.160572052001953e-05,
      "step": 33141,
      "training_step_time": 7.614544868469238
    },
    {
      "epoch": 3.160667419433594e-05,
      "model_forward_time": 1.7464499473571777,
      "step": 33142
    },
    {
      "epoch": 3.160667419433594e-05,
      "step": 33142,
      "training_step_time": 6.769999265670776
    },
    {
      "epoch": 3.1607627868652344e-05,
      "model_forward_time": 1.738922119140625,
      "step": 33143
    },
    {
      "epoch": 3.1607627868652344e-05,
      "step": 33143,
      "training_step_time": 6.775316476821899
    },
    {
      "epoch": 3.160858154296875e-05,
      "model_forward_time": 1.7425565719604492,
      "step": 33144
    },
    {
      "epoch": 3.160858154296875e-05,
      "step": 33144,
      "training_step_time": 6.826615810394287
    },
    {
      "epoch": 3.160953521728516e-05,
      "model_forward_time": 1.753922462463379,
      "step": 33145
    },
    {
      "epoch": 3.160953521728516e-05,
      "step": 33145,
      "training_step_time": 7.450314044952393
    },
    {
      "epoch": 3.161048889160156e-05,
      "model_forward_time": 1.7564139366149902,
      "step": 33146
    },
    {
      "epoch": 3.161048889160156e-05,
      "step": 33146,
      "training_step_time": 8.573350667953491
    },
    {
      "epoch": 3.161144256591797e-05,
      "model_forward_time": 1.3810815811157227,
      "step": 33147
    },
    {
      "epoch": 3.161144256591797e-05,
      "step": 33147,
      "training_step_time": 9.650124073028564
    },
    {
      "epoch": 3.161239624023438e-05,
      "model_forward_time": 2.899587631225586,
      "step": 33148
    },
    {
      "epoch": 3.161239624023438e-05,
      "step": 33148,
      "training_step_time": 9.831621885299683
    },
    {
      "epoch": 3.161334991455078e-05,
      "model_forward_time": 1.748427152633667,
      "step": 33149
    },
    {
      "epoch": 3.161334991455078e-05,
      "step": 33149,
      "training_step_time": 7.73104453086853
    },
    {
      "epoch": 3.1614303588867186e-05,
      "grad_norm": 0.17530709505081177,
      "learning_rate": 7.985679977369042e-06,
      "loss": 0.0539,
      "step": 33150
    },
    {
      "epoch": 3.1614303588867186e-05,
      "model_forward_time": 1.7411212921142578,
      "step": 33150
    },
    {
      "dynamics_loss_avg": 0.04806678351014852,
      "epoch": 3.1614303588867186e-05,
      "step": 33150
    },
    {
      "action_loss_avg": 0.001349344232585281,
      "epoch": 3.1614303588867186e-05,
      "step": 33150
    },
    {
      "epoch": 3.1614303588867186e-05,
      "step": 33150,
      "training_step_time": 7.6657891273498535
    },
    {
      "epoch": 3.16152572631836e-05,
      "model_forward_time": 1.7409415245056152,
      "step": 33151
    },
    {
      "epoch": 3.16152572631836e-05,
      "step": 33151,
      "training_step_time": 6.772757291793823
    },
    {
      "epoch": 3.16162109375e-05,
      "model_forward_time": 1.742628574371338,
      "step": 33152
    },
    {
      "epoch": 3.16162109375e-05,
      "step": 33152,
      "training_step_time": 6.8346521854400635
    },
    {
      "epoch": 3.1617164611816405e-05,
      "model_forward_time": 1.7477335929870605,
      "step": 33153
    },
    {
      "epoch": 3.1617164611816405e-05,
      "step": 33153,
      "training_step_time": 6.801511526107788
    },
    {
      "epoch": 3.1618118286132816e-05,
      "model_forward_time": 1.7722351551055908,
      "step": 33154
    },
    {
      "epoch": 3.1618118286132816e-05,
      "step": 33154,
      "training_step_time": 7.494059085845947
    },
    {
      "epoch": 3.161907196044922e-05,
      "model_forward_time": 1.7521765232086182,
      "step": 33155
    },
    {
      "epoch": 3.161907196044922e-05,
      "step": 33155,
      "training_step_time": 7.933142423629761
    },
    {
      "epoch": 3.1620025634765624e-05,
      "model_forward_time": 1.7480604648590088,
      "step": 33156
    },
    {
      "epoch": 3.1620025634765624e-05,
      "step": 33156,
      "training_step_time": 9.695519924163818
    },
    {
      "epoch": 3.162097930908203e-05,
      "model_forward_time": 2.8626105785369873,
      "step": 33157
    },
    {
      "epoch": 3.162097930908203e-05,
      "step": 33157,
      "training_step_time": 10.440570831298828
    },
    {
      "epoch": 3.162193298339844e-05,
      "model_forward_time": 1.7465729713439941,
      "step": 33158
    },
    {
      "epoch": 3.162193298339844e-05,
      "step": 33158,
      "training_step_time": 7.869805574417114
    },
    {
      "epoch": 3.162288665771484e-05,
      "model_forward_time": 1.7493505477905273,
      "step": 33159
    },
    {
      "epoch": 3.162288665771484e-05,
      "step": 33159,
      "training_step_time": 7.794253826141357
    },
    {
      "epoch": 3.162384033203125e-05,
      "grad_norm": 0.19196279346942902,
      "learning_rate": 7.984353499834902e-06,
      "loss": 0.0542,
      "step": 33160
    },
    {
      "epoch": 3.162384033203125e-05,
      "model_forward_time": 1.7474570274353027,
      "step": 33160
    },
    {
      "dynamics_loss_avg": 0.05830317940562964,
      "epoch": 3.162384033203125e-05,
      "step": 33160
    },
    {
      "action_loss_avg": 0.0025855222658719866,
      "epoch": 3.162384033203125e-05,
      "step": 33160
    },
    {
      "epoch": 3.162384033203125e-05,
      "step": 33160,
      "training_step_time": 6.9836626052856445
    },
    {
      "epoch": 3.162479400634766e-05,
      "model_forward_time": 1.7485718727111816,
      "step": 33161
    },
    {
      "epoch": 3.162479400634766e-05,
      "step": 33161,
      "training_step_time": 6.985265731811523
    },
    {
      "epoch": 3.162574768066406e-05,
      "model_forward_time": 1.7395427227020264,
      "step": 33162
    },
    {
      "epoch": 3.162574768066406e-05,
      "step": 33162,
      "training_step_time": 7.035550832748413
    },
    {
      "epoch": 3.1626701354980466e-05,
      "model_forward_time": 1.7707624435424805,
      "step": 33163
    },
    {
      "epoch": 3.1626701354980466e-05,
      "step": 33163,
      "training_step_time": 7.689152717590332
    },
    {
      "epoch": 3.1627655029296876e-05,
      "model_forward_time": 1.7433726787567139,
      "step": 33164
    },
    {
      "epoch": 3.1627655029296876e-05,
      "step": 33164,
      "training_step_time": 8.717638731002808
    },
    {
      "epoch": 3.162860870361328e-05,
      "model_forward_time": 1.7702381610870361,
      "step": 33165
    },
    {
      "epoch": 3.162860870361328e-05,
      "step": 33165,
      "training_step_time": 9.289884567260742
    },
    {
      "epoch": 3.1629562377929684e-05,
      "model_forward_time": 1.7493412494659424,
      "step": 33166
    },
    {
      "epoch": 3.1629562377929684e-05,
      "step": 33166,
      "training_step_time": 10.172841310501099
    },
    {
      "epoch": 3.1630516052246095e-05,
      "model_forward_time": 1.749563217163086,
      "step": 33167
    },
    {
      "epoch": 3.1630516052246095e-05,
      "step": 33167,
      "training_step_time": 7.865243911743164
    },
    {
      "epoch": 3.16314697265625e-05,
      "model_forward_time": 1.7405650615692139,
      "step": 33168
    },
    {
      "epoch": 3.16314697265625e-05,
      "step": 33168,
      "training_step_time": 7.840455532073975
    },
    {
      "epoch": 3.16324234008789e-05,
      "model_forward_time": 1.753425121307373,
      "step": 33169
    },
    {
      "epoch": 3.16324234008789e-05,
      "step": 33169,
      "training_step_time": 6.873565673828125
    },
    {
      "epoch": 3.1633377075195314e-05,
      "grad_norm": 0.12544415891170502,
      "learning_rate": 7.983026695936351e-06,
      "loss": 0.0551,
      "step": 33170
    },
    {
      "epoch": 3.1633377075195314e-05,
      "model_forward_time": 1.7436401844024658,
      "step": 33170
    },
    {
      "dynamics_loss_avg": 0.057702760025858876,
      "epoch": 3.1633377075195314e-05,
      "step": 33170
    },
    {
      "action_loss_avg": 0.0038481831084936856,
      "epoch": 3.1633377075195314e-05,
      "step": 33170
    },
    {
      "epoch": 3.1633377075195314e-05,
      "step": 33170,
      "training_step_time": 6.920152902603149
    },
    {
      "epoch": 3.163433074951172e-05,
      "model_forward_time": 1.7453439235687256,
      "step": 33171
    },
    {
      "epoch": 3.163433074951172e-05,
      "step": 33171,
      "training_step_time": 6.864781379699707
    },
    {
      "epoch": 3.163528442382812e-05,
      "model_forward_time": 1.759078025817871,
      "step": 33172
    },
    {
      "epoch": 3.163528442382812e-05,
      "step": 33172,
      "training_step_time": 7.6100382804870605
    },
    {
      "epoch": 3.163623809814453e-05,
      "model_forward_time": 1.7376229763031006,
      "step": 33173
    },
    {
      "epoch": 3.163623809814453e-05,
      "step": 33173,
      "training_step_time": 7.87017822265625
    },
    {
      "epoch": 3.163719177246094e-05,
      "model_forward_time": 1.7654173374176025,
      "step": 33174
    },
    {
      "epoch": 3.163719177246094e-05,
      "step": 33174,
      "training_step_time": 7.733993291854858
    },
    {
      "epoch": 3.163814544677734e-05,
      "model_forward_time": 1.7494864463806152,
      "step": 33175
    },
    {
      "epoch": 3.163814544677734e-05,
      "step": 33175,
      "training_step_time": 9.144872188568115
    },
    {
      "epoch": 3.163909912109375e-05,
      "model_forward_time": 2.936774253845215,
      "step": 33176
    },
    {
      "epoch": 3.163909912109375e-05,
      "step": 33176,
      "training_step_time": 10.01109266281128
    },
    {
      "epoch": 3.1640052795410156e-05,
      "model_forward_time": 1.7488975524902344,
      "step": 33177
    },
    {
      "epoch": 3.1640052795410156e-05,
      "step": 33177,
      "training_step_time": 7.806002378463745
    },
    {
      "epoch": 3.164100646972656e-05,
      "model_forward_time": 1.744736909866333,
      "step": 33178
    },
    {
      "epoch": 3.164100646972656e-05,
      "step": 33178,
      "training_step_time": 6.912136793136597
    },
    {
      "epoch": 3.164196014404297e-05,
      "model_forward_time": 1.7486779689788818,
      "step": 33179
    },
    {
      "epoch": 3.164196014404297e-05,
      "step": 33179,
      "training_step_time": 6.865335702896118
    },
    {
      "epoch": 3.1642913818359375e-05,
      "grad_norm": 0.1784418374300003,
      "learning_rate": 7.981699565818486e-06,
      "loss": 0.0579,
      "step": 33180
    },
    {
      "epoch": 3.1642913818359375e-05,
      "model_forward_time": 1.7375810146331787,
      "step": 33180
    },
    {
      "dynamics_loss_avg": 0.051710499450564384,
      "epoch": 3.1642913818359375e-05,
      "step": 33180
    },
    {
      "action_loss_avg": 0.0019376147974980995,
      "epoch": 3.1642913818359375e-05,
      "step": 33180
    },
    {
      "epoch": 3.1642913818359375e-05,
      "step": 33180,
      "training_step_time": 6.7391676902771
    },
    {
      "epoch": 3.164386749267578e-05,
      "model_forward_time": 1.7477772235870361,
      "step": 33181
    },
    {
      "epoch": 3.164386749267578e-05,
      "step": 33181,
      "training_step_time": 7.414541482925415
    },
    {
      "epoch": 3.164482116699219e-05,
      "model_forward_time": 1.0525667667388916,
      "step": 33182
    },
    {
      "epoch": 3.164482116699219e-05,
      "step": 33182,
      "training_step_time": 8.124111652374268
    },
    {
      "epoch": 3.1645774841308594e-05,
      "model_forward_time": 1.751842975616455,
      "step": 33183
    },
    {
      "epoch": 3.1645774841308594e-05,
      "step": 33183,
      "training_step_time": 7.677237033843994
    },
    {
      "epoch": 3.1646728515625e-05,
      "model_forward_time": 1.7509679794311523,
      "step": 33184
    },
    {
      "epoch": 3.1646728515625e-05,
      "step": 33184,
      "training_step_time": 9.821983337402344
    },
    {
      "epoch": 3.164768218994141e-05,
      "model_forward_time": 2.921339273452759,
      "step": 33185
    },
    {
      "epoch": 3.164768218994141e-05,
      "step": 33185,
      "training_step_time": 7.816570997238159
    },
    {
      "epoch": 3.164863586425781e-05,
      "model_forward_time": 1.7446954250335693,
      "step": 33186
    },
    {
      "epoch": 3.164863586425781e-05,
      "step": 33186,
      "training_step_time": 8.167996644973755
    },
    {
      "epoch": 3.1649589538574216e-05,
      "model_forward_time": 1.7515416145324707,
      "step": 33187
    },
    {
      "epoch": 3.1649589538574216e-05,
      "step": 33187,
      "training_step_time": 7.577374458312988
    },
    {
      "epoch": 3.165054321289063e-05,
      "model_forward_time": 1.744211196899414,
      "step": 33188
    },
    {
      "epoch": 3.165054321289063e-05,
      "step": 33188,
      "training_step_time": 6.689260005950928
    },
    {
      "epoch": 3.165149688720703e-05,
      "model_forward_time": 1.748976230621338,
      "step": 33189
    },
    {
      "epoch": 3.165149688720703e-05,
      "step": 33189,
      "training_step_time": 6.681377410888672
    },
    {
      "epoch": 3.1652450561523435e-05,
      "grad_norm": 0.12073872238397598,
      "learning_rate": 7.980372109626437e-06,
      "loss": 0.0575,
      "step": 33190
    },
    {
      "epoch": 3.1652450561523435e-05,
      "model_forward_time": 1.7443807125091553,
      "step": 33190
    },
    {
      "dynamics_loss_avg": 0.04795744251459837,
      "epoch": 3.1652450561523435e-05,
      "step": 33190
    },
    {
      "action_loss_avg": 0.004204442317131907,
      "epoch": 3.1652450561523435e-05,
      "step": 33190
    },
    {
      "epoch": 3.1652450561523435e-05,
      "step": 33190,
      "training_step_time": 7.478846788406372
    },
    {
      "epoch": 3.1653404235839846e-05,
      "model_forward_time": 1.751758337020874,
      "step": 33191
    },
    {
      "epoch": 3.1653404235839846e-05,
      "step": 33191,
      "training_step_time": 7.471187591552734
    },
    {
      "epoch": 3.165435791015625e-05,
      "model_forward_time": 1.7397198677062988,
      "step": 33192
    },
    {
      "epoch": 3.165435791015625e-05,
      "step": 33192,
      "training_step_time": 8.292895793914795
    },
    {
      "epoch": 3.1655311584472654e-05,
      "model_forward_time": 1.768934965133667,
      "step": 33193
    },
    {
      "epoch": 3.1655311584472654e-05,
      "step": 33193,
      "training_step_time": 9.03441834449768
    },
    {
      "epoch": 3.1656265258789065e-05,
      "model_forward_time": 2.5311176776885986,
      "step": 33194
    },
    {
      "epoch": 3.1656265258789065e-05,
      "step": 33194,
      "training_step_time": 7.564054489135742
    },
    {
      "epoch": 3.165721893310547e-05,
      "model_forward_time": 1.7452402114868164,
      "step": 33195
    },
    {
      "epoch": 3.165721893310547e-05,
      "step": 33195,
      "training_step_time": 8.098090410232544
    },
    {
      "epoch": 3.165817260742187e-05,
      "model_forward_time": 1.741525411605835,
      "step": 33196
    },
    {
      "epoch": 3.165817260742187e-05,
      "step": 33196,
      "training_step_time": 7.567322492599487
    },
    {
      "epoch": 3.1659126281738284e-05,
      "model_forward_time": 1.7471776008605957,
      "step": 33197
    },
    {
      "epoch": 3.1659126281738284e-05,
      "step": 33197,
      "training_step_time": 6.691871643066406
    },
    {
      "epoch": 3.166007995605469e-05,
      "model_forward_time": 1.7418408393859863,
      "step": 33198
    },
    {
      "epoch": 3.166007995605469e-05,
      "step": 33198,
      "training_step_time": 6.71813178062439
    },
    {
      "epoch": 3.166103363037109e-05,
      "model_forward_time": 1.739623785018921,
      "step": 33199
    },
    {
      "epoch": 3.166103363037109e-05,
      "step": 33199,
      "training_step_time": 6.68331503868103
    },
    {
      "epoch": 3.16619873046875e-05,
      "grad_norm": 0.12656886875629425,
      "learning_rate": 7.979044327505377e-06,
      "loss": 0.0589,
      "step": 33200
    },
    {
      "epoch": 3.16619873046875e-05,
      "model_forward_time": 1.7724170684814453,
      "step": 33200
    },
    {
      "dynamics_loss_avg": 0.05357978660613298,
      "epoch": 3.16619873046875e-05,
      "step": 33200
    },
    {
      "action_loss_avg": 0.00258175433264114,
      "epoch": 3.16619873046875e-05,
      "step": 33200
    },
    {
      "epoch": 3.16619873046875e-05,
      "step": 33200,
      "training_step_time": 7.466843128204346
    },
    {
      "epoch": 3.166294097900391e-05,
      "model_forward_time": 1.7394957542419434,
      "step": 33201
    },
    {
      "epoch": 3.166294097900391e-05,
      "step": 33201,
      "training_step_time": 8.171836376190186
    },
    {
      "epoch": 3.166389465332031e-05,
      "model_forward_time": 1.7718830108642578,
      "step": 33202
    },
    {
      "epoch": 3.166389465332031e-05,
      "step": 33202,
      "training_step_time": 8.611664295196533
    },
    {
      "epoch": 3.166484832763672e-05,
      "model_forward_time": 2.863196611404419,
      "step": 33203
    },
    {
      "epoch": 3.166484832763672e-05,
      "step": 33203,
      "training_step_time": 8.539775371551514
    },
    {
      "epoch": 3.1665802001953126e-05,
      "model_forward_time": 1.7404756546020508,
      "step": 33204
    },
    {
      "epoch": 3.1665802001953126e-05,
      "step": 33204,
      "training_step_time": 7.627742767333984
    },
    {
      "epoch": 3.166675567626953e-05,
      "model_forward_time": 1.7413055896759033,
      "step": 33205
    },
    {
      "epoch": 3.166675567626953e-05,
      "step": 33205,
      "training_step_time": 7.600505590438843
    },
    {
      "epoch": 3.166770935058594e-05,
      "model_forward_time": 1.7459051609039307,
      "step": 33206
    },
    {
      "epoch": 3.166770935058594e-05,
      "step": 33206,
      "training_step_time": 6.694401979446411
    },
    {
      "epoch": 3.1668663024902344e-05,
      "model_forward_time": 1.7064290046691895,
      "step": 33207
    },
    {
      "epoch": 3.1668663024902344e-05,
      "step": 33207,
      "training_step_time": 6.704381942749023
    },
    {
      "epoch": 3.166961669921875e-05,
      "model_forward_time": 1.7187488079071045,
      "step": 33208
    },
    {
      "epoch": 3.166961669921875e-05,
      "step": 33208,
      "training_step_time": 6.63701319694519
    },
    {
      "epoch": 3.167057037353516e-05,
      "model_forward_time": 1.73333740234375,
      "step": 33209
    },
    {
      "epoch": 3.167057037353516e-05,
      "step": 33209,
      "training_step_time": 7.508546590805054
    },
    {
      "epoch": 3.167152404785156e-05,
      "grad_norm": 0.10795776546001434,
      "learning_rate": 7.977716219600507e-06,
      "loss": 0.0589,
      "step": 33210
    },
    {
      "epoch": 3.167152404785156e-05,
      "model_forward_time": 1.7216331958770752,
      "step": 33210
    },
    {
      "dynamics_loss_avg": 0.04499696884304285,
      "epoch": 3.167152404785156e-05,
      "step": 33210
    },
    {
      "action_loss_avg": 0.0018712850956944748,
      "epoch": 3.167152404785156e-05,
      "step": 33210
    },
    {
      "epoch": 3.167152404785156e-05,
      "step": 33210,
      "training_step_time": 7.679087162017822
    },
    {
      "epoch": 3.167247772216797e-05,
      "model_forward_time": 1.3653950691223145,
      "step": 33211
    },
    {
      "epoch": 3.167247772216797e-05,
      "step": 33211,
      "training_step_time": 8.121715068817139
    },
    {
      "epoch": 3.167343139648438e-05,
      "model_forward_time": 1.7280499935150146,
      "step": 33212
    },
    {
      "epoch": 3.167343139648438e-05,
      "step": 33212,
      "training_step_time": 9.039536952972412
    },
    {
      "epoch": 3.167438507080078e-05,
      "model_forward_time": 2.9508674144744873,
      "step": 33213
    },
    {
      "epoch": 3.167438507080078e-05,
      "step": 33213,
      "training_step_time": 7.688959836959839
    },
    {
      "epoch": 3.1675338745117186e-05,
      "model_forward_time": 1.7235262393951416,
      "step": 33214
    },
    {
      "epoch": 3.1675338745117186e-05,
      "step": 33214,
      "training_step_time": 7.560116529464722
    },
    {
      "epoch": 3.16762924194336e-05,
      "model_forward_time": 1.7317042350769043,
      "step": 33215
    },
    {
      "epoch": 3.16762924194336e-05,
      "step": 33215,
      "training_step_time": 6.7072529792785645
    },
    {
      "epoch": 3.167724609375e-05,
      "model_forward_time": 1.7239899635314941,
      "step": 33216
    },
    {
      "epoch": 3.167724609375e-05,
      "step": 33216,
      "training_step_time": 6.688730239868164
    },
    {
      "epoch": 3.1678199768066405e-05,
      "model_forward_time": 1.7233059406280518,
      "step": 33217
    },
    {
      "epoch": 3.1678199768066405e-05,
      "step": 33217,
      "training_step_time": 6.639475107192993
    },
    {
      "epoch": 3.1679153442382816e-05,
      "model_forward_time": 1.7504782676696777,
      "step": 33218
    },
    {
      "epoch": 3.1679153442382816e-05,
      "step": 33218,
      "training_step_time": 7.449743986129761
    },
    {
      "epoch": 3.168010711669922e-05,
      "model_forward_time": 1.7281436920166016,
      "step": 33219
    },
    {
      "epoch": 3.168010711669922e-05,
      "step": 33219,
      "training_step_time": 7.444097280502319
    },
    {
      "epoch": 3.1681060791015624e-05,
      "grad_norm": 0.11700037121772766,
      "learning_rate": 7.97638778605707e-06,
      "loss": 0.0589,
      "step": 33220
    },
    {
      "epoch": 3.1681060791015624e-05,
      "model_forward_time": 1.7152347564697266,
      "step": 33220
    },
    {
      "dynamics_loss_avg": 0.04508585575968027,
      "epoch": 3.1681060791015624e-05,
      "step": 33220
    },
    {
      "action_loss_avg": 0.0018145651061786338,
      "epoch": 3.1681060791015624e-05,
      "step": 33220
    },
    {
      "epoch": 3.1681060791015624e-05,
      "step": 33220,
      "training_step_time": 8.750804424285889
    },
    {
      "epoch": 3.168201446533203e-05,
      "model_forward_time": 1.727004051208496,
      "step": 33221
    },
    {
      "epoch": 3.168201446533203e-05,
      "step": 33221,
      "training_step_time": 8.77535629272461
    },
    {
      "epoch": 3.168296813964844e-05,
      "model_forward_time": 2.8464789390563965,
      "step": 33222
    },
    {
      "epoch": 3.168296813964844e-05,
      "step": 33222,
      "training_step_time": 8.876291751861572
    },
    {
      "epoch": 3.168392181396484e-05,
      "model_forward_time": 1.7254152297973633,
      "step": 33223
    },
    {
      "epoch": 3.168392181396484e-05,
      "step": 33223,
      "training_step_time": 7.693522691726685
    },
    {
      "epoch": 3.168487548828125e-05,
      "model_forward_time": 1.732004165649414,
      "step": 33224
    },
    {
      "epoch": 3.168487548828125e-05,
      "step": 33224,
      "training_step_time": 6.83597731590271
    },
    {
      "epoch": 3.168582916259766e-05,
      "model_forward_time": 1.727353811264038,
      "step": 33225
    },
    {
      "epoch": 3.168582916259766e-05,
      "step": 33225,
      "training_step_time": 6.8148839473724365
    },
    {
      "epoch": 3.168678283691406e-05,
      "model_forward_time": 1.7418427467346191,
      "step": 33226
    },
    {
      "epoch": 3.168678283691406e-05,
      "step": 33226,
      "training_step_time": 6.824740886688232
    },
    {
      "epoch": 3.1687736511230466e-05,
      "model_forward_time": 1.7570576667785645,
      "step": 33227
    },
    {
      "epoch": 3.1687736511230466e-05,
      "step": 33227,
      "training_step_time": 7.635727405548096
    },
    {
      "epoch": 3.1688690185546876e-05,
      "model_forward_time": 1.7484838962554932,
      "step": 33228
    },
    {
      "epoch": 3.1688690185546876e-05,
      "step": 33228,
      "training_step_time": 7.6611528396606445
    },
    {
      "epoch": 3.168964385986328e-05,
      "model_forward_time": 1.7439765930175781,
      "step": 33229
    },
    {
      "epoch": 3.168964385986328e-05,
      "step": 33229,
      "training_step_time": 7.782935380935669
    },
    {
      "epoch": 3.1690597534179685e-05,
      "grad_norm": 0.13060614466667175,
      "learning_rate": 7.975059027020338e-06,
      "loss": 0.0583,
      "step": 33230
    },
    {
      "epoch": 3.1690597534179685e-05,
      "model_forward_time": 1.7764675617218018,
      "step": 33230
    },
    {
      "dynamics_loss_avg": 0.04703648835420608,
      "epoch": 3.1690597534179685e-05,
      "step": 33230
    },
    {
      "action_loss_avg": 0.0015932861249893903,
      "epoch": 3.1690597534179685e-05,
      "step": 33230
    },
    {
      "epoch": 3.1690597534179685e-05,
      "step": 33230,
      "training_step_time": 7.937371253967285
    },
    {
      "epoch": 3.1691551208496095e-05,
      "model_forward_time": 2.903671979904175,
      "step": 33231
    },
    {
      "epoch": 3.1691551208496095e-05,
      "step": 33231,
      "training_step_time": 10.53653860092163
    },
    {
      "epoch": 3.16925048828125e-05,
      "model_forward_time": 1.7437961101531982,
      "step": 33232
    },
    {
      "epoch": 3.16925048828125e-05,
      "step": 33232,
      "training_step_time": 7.741502523422241
    },
    {
      "epoch": 3.16934585571289e-05,
      "model_forward_time": 1.742408275604248,
      "step": 33233
    },
    {
      "epoch": 3.16934585571289e-05,
      "step": 33233,
      "training_step_time": 7.742246389389038
    },
    {
      "epoch": 3.1694412231445314e-05,
      "model_forward_time": 1.7379732131958008,
      "step": 33234
    },
    {
      "epoch": 3.1694412231445314e-05,
      "step": 33234,
      "training_step_time": 6.843886137008667
    },
    {
      "epoch": 3.169536590576172e-05,
      "model_forward_time": 1.740121603012085,
      "step": 33235
    },
    {
      "epoch": 3.169536590576172e-05,
      "step": 33235,
      "training_step_time": 6.844040870666504
    },
    {
      "epoch": 3.169631958007812e-05,
      "model_forward_time": 1.7573778629302979,
      "step": 33236
    },
    {
      "epoch": 3.169631958007812e-05,
      "step": 33236,
      "training_step_time": 7.718608140945435
    },
    {
      "epoch": 3.169727325439453e-05,
      "model_forward_time": 1.7402241230010986,
      "step": 33237
    },
    {
      "epoch": 3.169727325439453e-05,
      "step": 33237,
      "training_step_time": 8.67308497428894
    },
    {
      "epoch": 3.169822692871094e-05,
      "model_forward_time": 1.7493720054626465,
      "step": 33238
    },
    {
      "epoch": 3.169822692871094e-05,
      "step": 33238,
      "training_step_time": 7.8899524211883545
    },
    {
      "epoch": 3.169918060302734e-05,
      "model_forward_time": 1.4096651077270508,
      "step": 33239
    },
    {
      "epoch": 3.169918060302734e-05,
      "step": 33239,
      "training_step_time": 9.331573724746704
    },
    {
      "epoch": 3.170013427734375e-05,
      "grad_norm": 0.19030030071735382,
      "learning_rate": 7.973729942635624e-06,
      "loss": 0.0554,
      "step": 33240
    },
    {
      "epoch": 3.170013427734375e-05,
      "model_forward_time": 2.8589928150177,
      "step": 33240
    },
    {
      "dynamics_loss_avg": 0.05086177084594965,
      "epoch": 3.170013427734375e-05,
      "step": 33240
    },
    {
      "action_loss_avg": 0.0028254874632693827,
      "epoch": 3.170013427734375e-05,
      "step": 33240
    },
    {
      "epoch": 3.170013427734375e-05,
      "step": 33240,
      "training_step_time": 9.992780685424805
    },
    {
      "epoch": 3.1701087951660156e-05,
      "model_forward_time": 1.7518606185913086,
      "step": 33241
    },
    {
      "epoch": 3.1701087951660156e-05,
      "step": 33241,
      "training_step_time": 8.649555444717407
    },
    {
      "epoch": 3.170204162597656e-05,
      "model_forward_time": 1.7478785514831543,
      "step": 33242
    },
    {
      "epoch": 3.170204162597656e-05,
      "step": 33242,
      "training_step_time": 7.56950044631958
    },
    {
      "epoch": 3.170299530029297e-05,
      "model_forward_time": 1.7451739311218262,
      "step": 33243
    },
    {
      "epoch": 3.170299530029297e-05,
      "step": 33243,
      "training_step_time": 7.616231679916382
    },
    {
      "epoch": 3.1703948974609375e-05,
      "model_forward_time": 1.7503304481506348,
      "step": 33244
    },
    {
      "epoch": 3.1703948974609375e-05,
      "step": 33244,
      "training_step_time": 6.912974119186401
    },
    {
      "epoch": 3.170490264892578e-05,
      "model_forward_time": 1.7399709224700928,
      "step": 33245
    },
    {
      "epoch": 3.170490264892578e-05,
      "step": 33245,
      "training_step_time": 7.7336585521698
    },
    {
      "epoch": 3.170585632324219e-05,
      "model_forward_time": 1.745394229888916,
      "step": 33246
    },
    {
      "epoch": 3.170585632324219e-05,
      "step": 33246,
      "training_step_time": 7.6272172927856445
    },
    {
      "epoch": 3.1706809997558594e-05,
      "model_forward_time": 1.7492730617523193,
      "step": 33247
    },
    {
      "epoch": 3.1706809997558594e-05,
      "step": 33247,
      "training_step_time": 8.421411991119385
    },
    {
      "epoch": 3.1707763671875e-05,
      "model_forward_time": 1.7603950500488281,
      "step": 33248
    },
    {
      "epoch": 3.1707763671875e-05,
      "step": 33248,
      "training_step_time": 8.363133192062378
    },
    {
      "epoch": 3.170871734619141e-05,
      "model_forward_time": 1.7522556781768799,
      "step": 33249
    },
    {
      "epoch": 3.170871734619141e-05,
      "step": 33249,
      "training_step_time": 7.965721130371094
    },
    {
      "epoch": 3.170967102050781e-05,
      "grad_norm": 0.13498882949352264,
      "learning_rate": 7.972400533048273e-06,
      "loss": 0.0579,
      "step": 33250
    },
    {
      "epoch": 3.170967102050781e-05,
      "model_forward_time": 2.90018892288208,
      "step": 33250
    },
    {
      "dynamics_loss_avg": 0.06214849632233381,
      "epoch": 3.170967102050781e-05,
      "step": 33250
    },
    {
      "action_loss_avg": 0.0029018049826845527,
      "epoch": 3.170967102050781e-05,
      "step": 33250
    },
    {
      "epoch": 3.170967102050781e-05,
      "step": 33250,
      "training_step_time": 7.708261489868164
    },
    {
      "epoch": 3.1710624694824217e-05,
      "model_forward_time": 1.743767499923706,
      "step": 33251
    },
    {
      "epoch": 3.1710624694824217e-05,
      "step": 33251,
      "training_step_time": 7.773620843887329
    },
    {
      "epoch": 3.171157836914063e-05,
      "model_forward_time": 1.7418837547302246,
      "step": 33252
    },
    {
      "epoch": 3.171157836914063e-05,
      "step": 33252,
      "training_step_time": 7.730484962463379
    },
    {
      "epoch": 3.171253204345703e-05,
      "model_forward_time": 1.745295524597168,
      "step": 33253
    },
    {
      "epoch": 3.171253204345703e-05,
      "step": 33253,
      "training_step_time": 6.888078451156616
    },
    {
      "epoch": 3.1713485717773435e-05,
      "model_forward_time": 1.7460663318634033,
      "step": 33254
    },
    {
      "epoch": 3.1713485717773435e-05,
      "step": 33254,
      "training_step_time": 7.663463354110718
    },
    {
      "epoch": 3.1714439392089846e-05,
      "model_forward_time": 1.7677032947540283,
      "step": 33255
    },
    {
      "epoch": 3.1714439392089846e-05,
      "step": 33255,
      "training_step_time": 7.684914827346802
    },
    {
      "epoch": 3.171539306640625e-05,
      "model_forward_time": 1.7450735569000244,
      "step": 33256
    },
    {
      "epoch": 3.171539306640625e-05,
      "step": 33256,
      "training_step_time": 8.456259965896606
    },
    {
      "epoch": 3.1716346740722654e-05,
      "model_forward_time": 1.7244834899902344,
      "step": 33257
    },
    {
      "epoch": 3.1716346740722654e-05,
      "step": 33257,
      "training_step_time": 8.609997272491455
    },
    {
      "epoch": 3.1717300415039065e-05,
      "model_forward_time": 1.745889663696289,
      "step": 33258
    },
    {
      "epoch": 3.1717300415039065e-05,
      "step": 33258,
      "training_step_time": 9.632421255111694
    },
    {
      "epoch": 3.171825408935547e-05,
      "model_forward_time": 2.9146926403045654,
      "step": 33259
    },
    {
      "epoch": 3.171825408935547e-05,
      "step": 33259,
      "training_step_time": 7.889124393463135
    },
    {
      "epoch": 3.171920776367187e-05,
      "grad_norm": 0.17641890048980713,
      "learning_rate": 7.971070798403671e-06,
      "loss": 0.0583,
      "step": 33260
    },
    {
      "epoch": 3.171920776367187e-05,
      "model_forward_time": 1.7161314487457275,
      "step": 33260
    },
    {
      "dynamics_loss_avg": 0.056856239587068556,
      "epoch": 3.171920776367187e-05,
      "step": 33260
    },
    {
      "action_loss_avg": 0.001963633461855352,
      "epoch": 3.171920776367187e-05,
      "step": 33260
    },
    {
      "epoch": 3.171920776367187e-05,
      "step": 33260,
      "training_step_time": 7.762787818908691
    },
    {
      "epoch": 3.1720161437988284e-05,
      "model_forward_time": 1.7156052589416504,
      "step": 33261
    },
    {
      "epoch": 3.1720161437988284e-05,
      "step": 33261,
      "training_step_time": 7.461416959762573
    },
    {
      "epoch": 3.172111511230469e-05,
      "model_forward_time": 1.7189116477966309,
      "step": 33262
    },
    {
      "epoch": 3.172111511230469e-05,
      "step": 33262,
      "training_step_time": 7.740908861160278
    },
    {
      "epoch": 3.172206878662109e-05,
      "model_forward_time": 1.7199463844299316,
      "step": 33263
    },
    {
      "epoch": 3.172206878662109e-05,
      "step": 33263,
      "training_step_time": 8.485719680786133
    },
    {
      "epoch": 3.17230224609375e-05,
      "model_forward_time": 1.7302603721618652,
      "step": 33264
    },
    {
      "epoch": 3.17230224609375e-05,
      "step": 33264,
      "training_step_time": 7.703874588012695
    },
    {
      "epoch": 3.172397613525391e-05,
      "model_forward_time": 1.721379280090332,
      "step": 33265
    },
    {
      "epoch": 3.172397613525391e-05,
      "step": 33265,
      "training_step_time": 8.313969135284424
    },
    {
      "epoch": 3.172492980957031e-05,
      "model_forward_time": 1.7251155376434326,
      "step": 33266
    },
    {
      "epoch": 3.172492980957031e-05,
      "step": 33266,
      "training_step_time": 10.721970081329346
    },
    {
      "epoch": 3.172588348388672e-05,
      "model_forward_time": 1.7514264583587646,
      "step": 33267
    },
    {
      "epoch": 3.172588348388672e-05,
      "step": 33267,
      "training_step_time": 9.273306846618652
    },
    {
      "epoch": 3.1726837158203126e-05,
      "model_forward_time": 2.867305040359497,
      "step": 33268
    },
    {
      "epoch": 3.1726837158203126e-05,
      "step": 33268,
      "training_step_time": 9.729297876358032
    },
    {
      "epoch": 3.172779083251953e-05,
      "model_forward_time": 1.7159836292266846,
      "step": 33269
    },
    {
      "epoch": 3.172779083251953e-05,
      "step": 33269,
      "training_step_time": 8.53384256362915
    },
    {
      "epoch": 3.172874450683594e-05,
      "grad_norm": 0.2057735025882721,
      "learning_rate": 7.969740738847231e-06,
      "loss": 0.0587,
      "step": 33270
    },
    {
      "epoch": 3.172874450683594e-05,
      "model_forward_time": 1.353949785232544,
      "step": 33270
    },
    {
      "dynamics_loss_avg": 0.051781839318573476,
      "epoch": 3.172874450683594e-05,
      "step": 33270
    },
    {
      "action_loss_avg": 0.001961055741412565,
      "epoch": 3.172874450683594e-05,
      "step": 33270
    },
    {
      "epoch": 3.172874450683594e-05,
      "step": 33270,
      "training_step_time": 7.640367269515991
    },
    {
      "epoch": 3.1729698181152345e-05,
      "model_forward_time": 1.7156355381011963,
      "step": 33271
    },
    {
      "epoch": 3.1729698181152345e-05,
      "step": 33271,
      "training_step_time": 7.753925323486328
    },
    {
      "epoch": 3.173065185546875e-05,
      "model_forward_time": 1.7217185497283936,
      "step": 33272
    },
    {
      "epoch": 3.173065185546875e-05,
      "step": 33272,
      "training_step_time": 7.576762914657593
    },
    {
      "epoch": 3.173160552978516e-05,
      "model_forward_time": 1.730421781539917,
      "step": 33273
    },
    {
      "epoch": 3.173160552978516e-05,
      "step": 33273,
      "training_step_time": 7.691784143447876
    },
    {
      "epoch": 3.173255920410156e-05,
      "model_forward_time": 1.7140378952026367,
      "step": 33274
    },
    {
      "epoch": 3.173255920410156e-05,
      "step": 33274,
      "training_step_time": 7.550206422805786
    },
    {
      "epoch": 3.173351287841797e-05,
      "model_forward_time": 1.7240147590637207,
      "step": 33275
    },
    {
      "epoch": 3.173351287841797e-05,
      "step": 33275,
      "training_step_time": 7.844112873077393
    },
    {
      "epoch": 3.173446655273438e-05,
      "model_forward_time": 1.3964602947235107,
      "step": 33276
    },
    {
      "epoch": 3.173446655273438e-05,
      "step": 33276,
      "training_step_time": 8.345875024795532
    },
    {
      "epoch": 3.173542022705078e-05,
      "model_forward_time": 2.7950947284698486,
      "step": 33277
    },
    {
      "epoch": 3.173542022705078e-05,
      "step": 33277,
      "training_step_time": 7.71006178855896
    },
    {
      "epoch": 3.1736373901367186e-05,
      "model_forward_time": 1.7107441425323486,
      "step": 33278
    },
    {
      "epoch": 3.1736373901367186e-05,
      "step": 33278,
      "training_step_time": 7.672818422317505
    },
    {
      "epoch": 3.17373275756836e-05,
      "model_forward_time": 1.0235393047332764,
      "step": 33279
    },
    {
      "epoch": 3.17373275756836e-05,
      "step": 33279,
      "training_step_time": 7.712514162063599
    },
    {
      "epoch": 3.173828125e-05,
      "grad_norm": 0.17985887825489044,
      "learning_rate": 7.96841035452441e-06,
      "loss": 0.0606,
      "step": 33280
    },
    {
      "epoch": 3.173828125e-05,
      "model_forward_time": 1.7197580337524414,
      "step": 33280
    },
    {
      "dynamics_loss_avg": 0.07098739156499505,
      "epoch": 3.173828125e-05,
      "step": 33280
    },
    {
      "action_loss_avg": 0.0024856337462551893,
      "epoch": 3.173828125e-05,
      "step": 33280
    },
    {
      "epoch": 3.173828125e-05,
      "step": 33280,
      "training_step_time": 7.70808219909668
    },
    {
      "epoch": 3.1739234924316405e-05,
      "model_forward_time": 1.726186990737915,
      "step": 33281
    },
    {
      "epoch": 3.1739234924316405e-05,
      "step": 33281,
      "training_step_time": 7.6265387535095215
    },
    {
      "epoch": 3.174018859863281e-05,
      "model_forward_time": 1.7152726650238037,
      "step": 33282
    },
    {
      "epoch": 3.174018859863281e-05,
      "step": 33282,
      "training_step_time": 8.502744913101196
    },
    {
      "epoch": 3.174114227294922e-05,
      "model_forward_time": 1.3500730991363525,
      "step": 33283
    },
    {
      "epoch": 3.174114227294922e-05,
      "step": 33283,
      "training_step_time": 7.583231449127197
    },
    {
      "epoch": 3.1742095947265624e-05,
      "model_forward_time": 1.7278592586517334,
      "step": 33284
    },
    {
      "epoch": 3.1742095947265624e-05,
      "step": 33284,
      "training_step_time": 7.718551158905029
    },
    {
      "epoch": 3.174304962158203e-05,
      "model_forward_time": 1.7416596412658691,
      "step": 33285
    },
    {
      "epoch": 3.174304962158203e-05,
      "step": 33285,
      "training_step_time": 8.841171741485596
    },
    {
      "epoch": 3.174400329589844e-05,
      "model_forward_time": 1.7268478870391846,
      "step": 33286
    },
    {
      "epoch": 3.174400329589844e-05,
      "step": 33286,
      "training_step_time": 10.278516292572021
    },
    {
      "epoch": 3.174495697021484e-05,
      "model_forward_time": 2.8725922107696533,
      "step": 33287
    },
    {
      "epoch": 3.174495697021484e-05,
      "step": 33287,
      "training_step_time": 7.830295085906982
    },
    {
      "epoch": 3.174591064453125e-05,
      "model_forward_time": 1.7220969200134277,
      "step": 33288
    },
    {
      "epoch": 3.174591064453125e-05,
      "step": 33288,
      "training_step_time": 7.782255411148071
    },
    {
      "epoch": 3.174686431884766e-05,
      "model_forward_time": 1.7276687622070312,
      "step": 33289
    },
    {
      "epoch": 3.174686431884766e-05,
      "step": 33289,
      "training_step_time": 7.738261461257935
    },
    {
      "epoch": 3.174781799316406e-05,
      "grad_norm": 0.14755050837993622,
      "learning_rate": 7.967079645580697e-06,
      "loss": 0.0621,
      "step": 33290
    },
    {
      "epoch": 3.174781799316406e-05,
      "model_forward_time": 1.7282238006591797,
      "step": 33290
    },
    {
      "dynamics_loss_avg": 0.037964732758700846,
      "epoch": 3.174781799316406e-05,
      "step": 33290
    },
    {
      "action_loss_avg": 0.0017185463337227702,
      "epoch": 3.174781799316406e-05,
      "step": 33290
    },
    {
      "epoch": 3.174781799316406e-05,
      "step": 33290,
      "training_step_time": 7.778105974197388
    },
    {
      "epoch": 3.1748771667480466e-05,
      "model_forward_time": 1.7215313911437988,
      "step": 33291
    },
    {
      "epoch": 3.1748771667480466e-05,
      "step": 33291,
      "training_step_time": 7.697335481643677
    },
    {
      "epoch": 3.1749725341796877e-05,
      "model_forward_time": 1.7388174533843994,
      "step": 33292
    },
    {
      "epoch": 3.1749725341796877e-05,
      "step": 33292,
      "training_step_time": 7.676249027252197
    },
    {
      "epoch": 3.175067901611328e-05,
      "model_forward_time": 1.7208044528961182,
      "step": 33293
    },
    {
      "epoch": 3.175067901611328e-05,
      "step": 33293,
      "training_step_time": 8.439027547836304
    },
    {
      "epoch": 3.1751632690429685e-05,
      "model_forward_time": 1.7633099555969238,
      "step": 33294
    },
    {
      "epoch": 3.1751632690429685e-05,
      "step": 33294,
      "training_step_time": 8.702334880828857
    },
    {
      "epoch": 3.1752586364746095e-05,
      "model_forward_time": 1.7236146926879883,
      "step": 33295
    },
    {
      "epoch": 3.1752586364746095e-05,
      "step": 33295,
      "training_step_time": 8.13900637626648
    },
    {
      "epoch": 3.17535400390625e-05,
      "model_forward_time": 2.8370070457458496,
      "step": 33296
    },
    {
      "epoch": 3.17535400390625e-05,
      "step": 33296,
      "training_step_time": 7.6644275188446045
    },
    {
      "epoch": 3.1754493713378903e-05,
      "model_forward_time": 1.722313404083252,
      "step": 33297
    },
    {
      "epoch": 3.1754493713378903e-05,
      "step": 33297,
      "training_step_time": 8.083505153656006
    },
    {
      "epoch": 3.1755447387695314e-05,
      "model_forward_time": 1.7329859733581543,
      "step": 33298
    },
    {
      "epoch": 3.1755447387695314e-05,
      "step": 33298,
      "training_step_time": 7.358055830001831
    },
    {
      "epoch": 3.175640106201172e-05,
      "model_forward_time": 1.7338762283325195,
      "step": 33299
    },
    {
      "epoch": 3.175640106201172e-05,
      "step": 33299,
      "training_step_time": 7.572579860687256
    },
    {
      "epoch": 3.175735473632812e-05,
      "grad_norm": 0.23522183299064636,
      "learning_rate": 7.965748612161613e-06,
      "loss": 0.0634,
      "step": 33300
    },
    {
      "epoch": 3.175735473632812e-05,
      "model_forward_time": 1.7202138900756836,
      "step": 33300
    },
    {
      "dynamics_loss_avg": 0.05753328669816256,
      "epoch": 3.175735473632812e-05,
      "step": 33300
    },
    {
      "action_loss_avg": 0.001805394483380951,
      "epoch": 3.175735473632812e-05,
      "step": 33300
    },
    {
      "epoch": 3.175735473632812e-05,
      "step": 33300,
      "training_step_time": 7.298047780990601
    },
    {
      "epoch": 3.175830841064453e-05,
      "model_forward_time": 1.7485640048980713,
      "step": 33301
    },
    {
      "epoch": 3.175830841064453e-05,
      "step": 33301,
      "training_step_time": 7.477643013000488
    },
    {
      "epoch": 3.175926208496094e-05,
      "model_forward_time": 1.7257752418518066,
      "step": 33302
    },
    {
      "epoch": 3.175926208496094e-05,
      "step": 33302,
      "training_step_time": 8.182760953903198
    },
    {
      "epoch": 3.176021575927734e-05,
      "model_forward_time": 1.7544031143188477,
      "step": 33303
    },
    {
      "epoch": 3.176021575927734e-05,
      "step": 33303,
      "training_step_time": 8.798040390014648
    },
    {
      "epoch": 3.176116943359375e-05,
      "model_forward_time": 1.7209892272949219,
      "step": 33304
    },
    {
      "epoch": 3.176116943359375e-05,
      "step": 33304,
      "training_step_time": 7.5835490226745605
    },
    {
      "epoch": 3.1762123107910156e-05,
      "model_forward_time": 1.715419054031372,
      "step": 33305
    },
    {
      "epoch": 3.1762123107910156e-05,
      "step": 33305,
      "training_step_time": 8.991395473480225
    },
    {
      "epoch": 3.176307678222656e-05,
      "model_forward_time": 1.7244572639465332,
      "step": 33306
    },
    {
      "epoch": 3.176307678222656e-05,
      "step": 33306,
      "training_step_time": 8.2988121509552
    },
    {
      "epoch": 3.176403045654297e-05,
      "model_forward_time": 1.7367348670959473,
      "step": 33307
    },
    {
      "epoch": 3.176403045654297e-05,
      "step": 33307,
      "training_step_time": 7.529007196426392
    },
    {
      "epoch": 3.1764984130859375e-05,
      "model_forward_time": 1.7225673198699951,
      "step": 33308
    },
    {
      "epoch": 3.1764984130859375e-05,
      "step": 33308,
      "training_step_time": 7.540799379348755
    },
    {
      "epoch": 3.176593780517578e-05,
      "model_forward_time": 1.7221488952636719,
      "step": 33309
    },
    {
      "epoch": 3.176593780517578e-05,
      "step": 33309,
      "training_step_time": 7.260756254196167
    },
    {
      "epoch": 3.176689147949219e-05,
      "grad_norm": 0.259594202041626,
      "learning_rate": 7.964417254412722e-06,
      "loss": 0.0631,
      "step": 33310
    },
    {
      "epoch": 3.176689147949219e-05,
      "model_forward_time": 1.7403345108032227,
      "step": 33310
    },
    {
      "dynamics_loss_avg": 0.061618156358599664,
      "epoch": 3.176689147949219e-05,
      "step": 33310
    },
    {
      "action_loss_avg": 0.00212556881015189,
      "epoch": 3.176689147949219e-05,
      "step": 33310
    },
    {
      "epoch": 3.176689147949219e-05,
      "step": 33310,
      "training_step_time": 8.350351095199585
    },
    {
      "epoch": 3.1767845153808594e-05,
      "model_forward_time": 1.7125463485717773,
      "step": 33311
    },
    {
      "epoch": 3.1767845153808594e-05,
      "step": 33311,
      "training_step_time": 8.14645767211914
    },
    {
      "epoch": 3.1768798828125e-05,
      "model_forward_time": 1.7745652198791504,
      "step": 33312
    },
    {
      "epoch": 3.1768798828125e-05,
      "step": 33312,
      "training_step_time": 8.802966117858887
    },
    {
      "epoch": 3.176975250244141e-05,
      "model_forward_time": 1.7305669784545898,
      "step": 33313
    },
    {
      "epoch": 3.176975250244141e-05,
      "step": 33313,
      "training_step_time": 7.708420991897583
    },
    {
      "epoch": 3.177070617675781e-05,
      "model_forward_time": 1.7206103801727295,
      "step": 33314
    },
    {
      "epoch": 3.177070617675781e-05,
      "step": 33314,
      "training_step_time": 8.828147411346436
    },
    {
      "epoch": 3.1771659851074217e-05,
      "model_forward_time": 2.9292783737182617,
      "step": 33315
    },
    {
      "epoch": 3.1771659851074217e-05,
      "step": 33315,
      "training_step_time": 7.654346942901611
    },
    {
      "epoch": 3.177261352539063e-05,
      "model_forward_time": 1.7345664501190186,
      "step": 33316
    },
    {
      "epoch": 3.177261352539063e-05,
      "step": 33316,
      "training_step_time": 7.506243467330933
    },
    {
      "epoch": 3.177356719970703e-05,
      "model_forward_time": 1.7371530532836914,
      "step": 33317
    },
    {
      "epoch": 3.177356719970703e-05,
      "step": 33317,
      "training_step_time": 7.54679536819458
    },
    {
      "epoch": 3.1774520874023435e-05,
      "model_forward_time": 1.7330763339996338,
      "step": 33318
    },
    {
      "epoch": 3.1774520874023435e-05,
      "step": 33318,
      "training_step_time": 7.212016820907593
    },
    {
      "epoch": 3.1775474548339846e-05,
      "model_forward_time": 1.7343077659606934,
      "step": 33319
    },
    {
      "epoch": 3.1775474548339846e-05,
      "step": 33319,
      "training_step_time": 7.537043333053589
    },
    {
      "epoch": 3.177642822265625e-05,
      "grad_norm": 0.18936282396316528,
      "learning_rate": 7.963085572479614e-06,
      "loss": 0.0622,
      "step": 33320
    },
    {
      "epoch": 3.177642822265625e-05,
      "model_forward_time": 1.7383029460906982,
      "step": 33320
    },
    {
      "dynamics_loss_avg": 0.0673422623425722,
      "epoch": 3.177642822265625e-05,
      "step": 33320
    },
    {
      "action_loss_avg": 0.0027400217251852153,
      "epoch": 3.177642822265625e-05,
      "step": 33320
    },
    {
      "epoch": 3.177642822265625e-05,
      "step": 33320,
      "training_step_time": 8.620663404464722
    },
    {
      "epoch": 3.1777381896972654e-05,
      "model_forward_time": 1.7606050968170166,
      "step": 33321
    },
    {
      "epoch": 3.1777381896972654e-05,
      "step": 33321,
      "training_step_time": 10.206711053848267
    },
    {
      "epoch": 3.1778335571289065e-05,
      "model_forward_time": 1.7312207221984863,
      "step": 33322
    },
    {
      "epoch": 3.1778335571289065e-05,
      "step": 33322,
      "training_step_time": 8.638887643814087
    },
    {
      "epoch": 3.177928924560547e-05,
      "model_forward_time": 1.73274827003479,
      "step": 33323
    },
    {
      "epoch": 3.177928924560547e-05,
      "step": 33323,
      "training_step_time": 9.320292711257935
    },
    {
      "epoch": 3.178024291992187e-05,
      "model_forward_time": 2.9335830211639404,
      "step": 33324
    },
    {
      "epoch": 3.178024291992187e-05,
      "step": 33324,
      "training_step_time": 7.672360897064209
    },
    {
      "epoch": 3.1781196594238284e-05,
      "model_forward_time": 1.7421679496765137,
      "step": 33325
    },
    {
      "epoch": 3.1781196594238284e-05,
      "step": 33325,
      "training_step_time": 7.528671979904175
    },
    {
      "epoch": 3.178215026855469e-05,
      "model_forward_time": 1.736269474029541,
      "step": 33326
    },
    {
      "epoch": 3.178215026855469e-05,
      "step": 33326,
      "training_step_time": 8.24661636352539
    },
    {
      "epoch": 3.178310394287109e-05,
      "model_forward_time": 1.7372827529907227,
      "step": 33327
    },
    {
      "epoch": 3.178310394287109e-05,
      "step": 33327,
      "training_step_time": 7.1193647384643555
    },
    {
      "epoch": 3.17840576171875e-05,
      "model_forward_time": 1.736588478088379,
      "step": 33328
    },
    {
      "epoch": 3.17840576171875e-05,
      "step": 33328,
      "training_step_time": 7.422873497009277
    },
    {
      "epoch": 3.178501129150391e-05,
      "model_forward_time": 1.7613105773925781,
      "step": 33329
    },
    {
      "epoch": 3.178501129150391e-05,
      "step": 33329,
      "training_step_time": 8.564576625823975
    },
    {
      "epoch": 3.178596496582031e-05,
      "grad_norm": 0.16379454731941223,
      "learning_rate": 7.961753566507924e-06,
      "loss": 0.0577,
      "step": 33330
    },
    {
      "epoch": 3.178596496582031e-05,
      "model_forward_time": 1.762662649154663,
      "step": 33330
    },
    {
      "dynamics_loss_avg": 0.05843926817178726,
      "epoch": 3.178596496582031e-05,
      "step": 33330
    },
    {
      "action_loss_avg": 0.00256534194923006,
      "epoch": 3.178596496582031e-05,
      "step": 33330
    },
    {
      "epoch": 3.178596496582031e-05,
      "step": 33330,
      "training_step_time": 8.308809280395508
    },
    {
      "epoch": 3.178691864013672e-05,
      "model_forward_time": 1.7342407703399658,
      "step": 33331
    },
    {
      "epoch": 3.178691864013672e-05,
      "step": 33331,
      "training_step_time": 9.174369812011719
    },
    {
      "epoch": 3.1787872314453126e-05,
      "model_forward_time": 1.7401270866394043,
      "step": 33332
    },
    {
      "epoch": 3.1787872314453126e-05,
      "step": 33332,
      "training_step_time": 8.60478162765503
    },
    {
      "epoch": 3.178882598876953e-05,
      "model_forward_time": 2.8970422744750977,
      "step": 33333
    },
    {
      "epoch": 3.178882598876953e-05,
      "step": 33333,
      "training_step_time": 7.943828821182251
    },
    {
      "epoch": 3.178977966308594e-05,
      "model_forward_time": 1.701948881149292,
      "step": 33334
    },
    {
      "epoch": 3.178977966308594e-05,
      "step": 33334,
      "training_step_time": 7.616408348083496
    },
    {
      "epoch": 3.1790733337402345e-05,
      "model_forward_time": 1.343933343887329,
      "step": 33335
    },
    {
      "epoch": 3.1790733337402345e-05,
      "step": 33335,
      "training_step_time": 8.241228818893433
    },
    {
      "epoch": 3.179168701171875e-05,
      "model_forward_time": 1.7138946056365967,
      "step": 33336
    },
    {
      "epoch": 3.179168701171875e-05,
      "step": 33336,
      "training_step_time": 7.738801956176758
    },
    {
      "epoch": 3.179264068603516e-05,
      "model_forward_time": 1.7385377883911133,
      "step": 33337
    },
    {
      "epoch": 3.179264068603516e-05,
      "step": 33337,
      "training_step_time": 7.529176712036133
    },
    {
      "epoch": 3.1793594360351563e-05,
      "model_forward_time": 1.3911306858062744,
      "step": 33338
    },
    {
      "epoch": 3.1793594360351563e-05,
      "step": 33338,
      "training_step_time": 7.6167473793029785
    },
    {
      "epoch": 3.179454803466797e-05,
      "model_forward_time": 1.7530877590179443,
      "step": 33339
    },
    {
      "epoch": 3.179454803466797e-05,
      "step": 33339,
      "training_step_time": 7.7330639362335205
    },
    {
      "epoch": 3.179550170898438e-05,
      "grad_norm": 0.19491498172283173,
      "learning_rate": 7.960421236643316e-06,
      "loss": 0.0623,
      "step": 33340
    },
    {
      "epoch": 3.179550170898438e-05,
      "model_forward_time": 1.7312450408935547,
      "step": 33340
    },
    {
      "dynamics_loss_avg": 0.052948057651519775,
      "epoch": 3.179550170898438e-05,
      "step": 33340
    },
    {
      "action_loss_avg": 0.002666214236523956,
      "epoch": 3.179550170898438e-05,
      "step": 33340
    },
    {
      "epoch": 3.179550170898438e-05,
      "step": 33340,
      "training_step_time": 8.405414819717407
    },
    {
      "epoch": 3.179645538330078e-05,
      "model_forward_time": 1.736342191696167,
      "step": 33341
    },
    {
      "epoch": 3.179645538330078e-05,
      "step": 33341,
      "training_step_time": 7.859827280044556
    },
    {
      "epoch": 3.1797409057617186e-05,
      "model_forward_time": 1.7416470050811768,
      "step": 33342
    },
    {
      "epoch": 3.1797409057617186e-05,
      "step": 33342,
      "training_step_time": 8.017145156860352
    },
    {
      "epoch": 3.17983627319336e-05,
      "model_forward_time": 1.7351484298706055,
      "step": 33343
    },
    {
      "epoch": 3.17983627319336e-05,
      "step": 33343,
      "training_step_time": 7.72163200378418
    },
    {
      "epoch": 3.179931640625e-05,
      "model_forward_time": 1.73655366897583,
      "step": 33344
    },
    {
      "epoch": 3.179931640625e-05,
      "step": 33344,
      "training_step_time": 7.655722379684448
    },
    {
      "epoch": 3.1800270080566405e-05,
      "model_forward_time": 1.7363064289093018,
      "step": 33345
    },
    {
      "epoch": 3.1800270080566405e-05,
      "step": 33345,
      "training_step_time": 7.723146438598633
    },
    {
      "epoch": 3.180122375488281e-05,
      "model_forward_time": 1.7344856262207031,
      "step": 33346
    },
    {
      "epoch": 3.180122375488281e-05,
      "step": 33346,
      "training_step_time": 7.321820497512817
    },
    {
      "epoch": 3.180217742919922e-05,
      "model_forward_time": 1.7594640254974365,
      "step": 33347
    },
    {
      "epoch": 3.180217742919922e-05,
      "step": 33347,
      "training_step_time": 7.834812164306641
    },
    {
      "epoch": 3.1803131103515624e-05,
      "model_forward_time": 1.7463061809539795,
      "step": 33348
    },
    {
      "epoch": 3.1803131103515624e-05,
      "step": 33348,
      "training_step_time": 9.527420282363892
    },
    {
      "epoch": 3.180408477783203e-05,
      "model_forward_time": 1.7324304580688477,
      "step": 33349
    },
    {
      "epoch": 3.180408477783203e-05,
      "step": 33349,
      "training_step_time": 7.897118330001831
    },
    {
      "epoch": 3.180503845214844e-05,
      "grad_norm": 0.15463677048683167,
      "learning_rate": 7.959088583031496e-06,
      "loss": 0.0616,
      "step": 33350
    },
    {
      "epoch": 3.180503845214844e-05,
      "model_forward_time": 1.7405967712402344,
      "step": 33350
    },
    {
      "dynamics_loss_avg": 0.054168807715177535,
      "epoch": 3.180503845214844e-05,
      "step": 33350
    },
    {
      "action_loss_avg": 0.0026994353451300413,
      "epoch": 3.180503845214844e-05,
      "step": 33350
    },
    {
      "epoch": 3.180503845214844e-05,
      "step": 33350,
      "training_step_time": 8.885488748550415
    },
    {
      "epoch": 3.180599212646484e-05,
      "model_forward_time": 1.7327263355255127,
      "step": 33351
    },
    {
      "epoch": 3.180599212646484e-05,
      "step": 33351,
      "training_step_time": 7.978010654449463
    },
    {
      "epoch": 3.180694580078125e-05,
      "model_forward_time": 2.954965114593506,
      "step": 33352
    },
    {
      "epoch": 3.180694580078125e-05,
      "step": 33352,
      "training_step_time": 8.713816165924072
    },
    {
      "epoch": 3.180789947509766e-05,
      "model_forward_time": 1.7265293598175049,
      "step": 33353
    },
    {
      "epoch": 3.180789947509766e-05,
      "step": 33353,
      "training_step_time": 7.772327423095703
    },
    {
      "epoch": 3.180885314941406e-05,
      "model_forward_time": 1.737961769104004,
      "step": 33354
    },
    {
      "epoch": 3.180885314941406e-05,
      "step": 33354,
      "training_step_time": 8.378966331481934
    },
    {
      "epoch": 3.1809806823730466e-05,
      "model_forward_time": 1.7354693412780762,
      "step": 33355
    },
    {
      "epoch": 3.1809806823730466e-05,
      "step": 33355,
      "training_step_time": 7.478281259536743
    },
    {
      "epoch": 3.1810760498046877e-05,
      "model_forward_time": 1.7836415767669678,
      "step": 33356
    },
    {
      "epoch": 3.1810760498046877e-05,
      "step": 33356,
      "training_step_time": 8.2890145778656
    },
    {
      "epoch": 3.181171417236328e-05,
      "model_forward_time": 1.7277588844299316,
      "step": 33357
    },
    {
      "epoch": 3.181171417236328e-05,
      "step": 33357,
      "training_step_time": 7.7241530418396
    },
    {
      "epoch": 3.1812667846679685e-05,
      "model_forward_time": 1.0295751094818115,
      "step": 33358
    },
    {
      "epoch": 3.1812667846679685e-05,
      "step": 33358,
      "training_step_time": 8.408925771713257
    },
    {
      "epoch": 3.1813621520996095e-05,
      "model_forward_time": 1.7352261543273926,
      "step": 33359
    },
    {
      "epoch": 3.1813621520996095e-05,
      "step": 33359,
      "training_step_time": 8.870043754577637
    },
    {
      "epoch": 3.18145751953125e-05,
      "grad_norm": 0.1649385541677475,
      "learning_rate": 7.957755605818193e-06,
      "loss": 0.0606,
      "step": 33360
    },
    {
      "epoch": 3.18145751953125e-05,
      "model_forward_time": 1.7384271621704102,
      "step": 33360
    },
    {
      "dynamics_loss_avg": 0.04739160379394889,
      "epoch": 3.18145751953125e-05,
      "step": 33360
    },
    {
      "action_loss_avg": 0.0016893140564206988,
      "epoch": 3.18145751953125e-05,
      "step": 33360
    },
    {
      "epoch": 3.18145751953125e-05,
      "step": 33360,
      "training_step_time": 8.693566083908081
    },
    {
      "epoch": 3.1815528869628903e-05,
      "model_forward_time": 2.894763946533203,
      "step": 33361
    },
    {
      "epoch": 3.1815528869628903e-05,
      "step": 33361,
      "training_step_time": 7.836831569671631
    },
    {
      "epoch": 3.1816482543945314e-05,
      "model_forward_time": 1.3682715892791748,
      "step": 33362
    },
    {
      "epoch": 3.1816482543945314e-05,
      "step": 33362,
      "training_step_time": 7.743652582168579
    },
    {
      "epoch": 3.181743621826172e-05,
      "model_forward_time": 1.7339808940887451,
      "step": 33363
    },
    {
      "epoch": 3.181743621826172e-05,
      "step": 33363,
      "training_step_time": 7.819526672363281
    },
    {
      "epoch": 3.181838989257812e-05,
      "model_forward_time": 1.7354907989501953,
      "step": 33364
    },
    {
      "epoch": 3.181838989257812e-05,
      "step": 33364,
      "training_step_time": 8.01215386390686
    },
    {
      "epoch": 3.181934356689453e-05,
      "model_forward_time": 1.741023302078247,
      "step": 33365
    },
    {
      "epoch": 3.181934356689453e-05,
      "step": 33365,
      "training_step_time": 8.535147905349731
    },
    {
      "epoch": 3.182029724121094e-05,
      "model_forward_time": 1.7374169826507568,
      "step": 33366
    },
    {
      "epoch": 3.182029724121094e-05,
      "step": 33366,
      "training_step_time": 7.637960910797119
    },
    {
      "epoch": 3.182125091552734e-05,
      "model_forward_time": 1.7459983825683594,
      "step": 33367
    },
    {
      "epoch": 3.182125091552734e-05,
      "step": 33367,
      "training_step_time": 7.813794374465942
    },
    {
      "epoch": 3.182220458984375e-05,
      "model_forward_time": 1.7339932918548584,
      "step": 33368
    },
    {
      "epoch": 3.182220458984375e-05,
      "step": 33368,
      "training_step_time": 7.849077224731445
    },
    {
      "epoch": 3.1823158264160156e-05,
      "model_forward_time": 1.7329020500183105,
      "step": 33369
    },
    {
      "epoch": 3.1823158264160156e-05,
      "step": 33369,
      "training_step_time": 9.853600978851318
    },
    {
      "epoch": 3.182411193847656e-05,
      "grad_norm": 0.14263935387134552,
      "learning_rate": 7.956422305149187e-06,
      "loss": 0.0586,
      "step": 33370
    },
    {
      "epoch": 3.182411193847656e-05,
      "model_forward_time": 1.737823247909546,
      "step": 33370
    },
    {
      "dynamics_loss_avg": 0.055139774642884734,
      "epoch": 3.182411193847656e-05,
      "step": 33370
    },
    {
      "action_loss_avg": 0.0019700558274053036,
      "epoch": 3.182411193847656e-05,
      "step": 33370
    },
    {
      "epoch": 3.182411193847656e-05,
      "step": 33370,
      "training_step_time": 8.27932095527649
    },
    {
      "epoch": 3.182506561279297e-05,
      "model_forward_time": 1.7382686138153076,
      "step": 33371
    },
    {
      "epoch": 3.182506561279297e-05,
      "step": 33371,
      "training_step_time": 8.261332750320435
    },
    {
      "epoch": 3.1826019287109375e-05,
      "model_forward_time": 1.6982922554016113,
      "step": 33372
    },
    {
      "epoch": 3.1826019287109375e-05,
      "step": 33372,
      "training_step_time": 7.674708604812622
    },
    {
      "epoch": 3.182697296142578e-05,
      "model_forward_time": 1.712691307067871,
      "step": 33373
    },
    {
      "epoch": 3.182697296142578e-05,
      "step": 33373,
      "training_step_time": 6.845965385437012
    },
    {
      "epoch": 3.182792663574219e-05,
      "model_forward_time": 1.7159180641174316,
      "step": 33374
    },
    {
      "epoch": 3.182792663574219e-05,
      "step": 33374,
      "training_step_time": 7.727080345153809
    },
    {
      "epoch": 3.1828880310058594e-05,
      "model_forward_time": 1.740849256515503,
      "step": 33375
    },
    {
      "epoch": 3.1828880310058594e-05,
      "step": 33375,
      "training_step_time": 7.670542240142822
    },
    {
      "epoch": 3.1829833984375e-05,
      "model_forward_time": 1.7351200580596924,
      "step": 33376
    },
    {
      "epoch": 3.1829833984375e-05,
      "step": 33376,
      "training_step_time": 7.791950941085815
    },
    {
      "epoch": 3.183078765869141e-05,
      "model_forward_time": 1.7162816524505615,
      "step": 33377
    },
    {
      "epoch": 3.183078765869141e-05,
      "step": 33377,
      "training_step_time": 8.668037176132202
    },
    {
      "epoch": 3.183174133300781e-05,
      "model_forward_time": 1.7246723175048828,
      "step": 33378
    },
    {
      "epoch": 3.183174133300781e-05,
      "step": 33378,
      "training_step_time": 8.869553327560425
    },
    {
      "epoch": 3.183269500732422e-05,
      "model_forward_time": 1.7146494388580322,
      "step": 33379
    },
    {
      "epoch": 3.183269500732422e-05,
      "step": 33379,
      "training_step_time": 8.388197422027588
    },
    {
      "epoch": 3.183364868164063e-05,
      "grad_norm": 0.12573957443237305,
      "learning_rate": 7.955088681170281e-06,
      "loss": 0.0632,
      "step": 33380
    },
    {
      "epoch": 3.183364868164063e-05,
      "model_forward_time": 1.7066535949707031,
      "step": 33380
    },
    {
      "dynamics_loss_avg": 0.0629125788807869,
      "epoch": 3.183364868164063e-05,
      "step": 33380
    },
    {
      "action_loss_avg": 0.0016854865185450762,
      "epoch": 3.183364868164063e-05,
      "step": 33380
    },
    {
      "epoch": 3.183364868164063e-05,
      "step": 33380,
      "training_step_time": 7.622562646865845
    },
    {
      "epoch": 3.183460235595703e-05,
      "model_forward_time": 1.714538335800171,
      "step": 33381
    },
    {
      "epoch": 3.183460235595703e-05,
      "step": 33381,
      "training_step_time": 7.590310096740723
    },
    {
      "epoch": 3.1835556030273435e-05,
      "model_forward_time": 1.7180423736572266,
      "step": 33382
    },
    {
      "epoch": 3.1835556030273435e-05,
      "step": 33382,
      "training_step_time": 7.249799966812134
    },
    {
      "epoch": 3.1836509704589846e-05,
      "model_forward_time": 1.7189571857452393,
      "step": 33383
    },
    {
      "epoch": 3.1836509704589846e-05,
      "step": 33383,
      "training_step_time": 7.557240962982178
    },
    {
      "epoch": 3.183746337890625e-05,
      "model_forward_time": 1.7320842742919922,
      "step": 33384
    },
    {
      "epoch": 3.183746337890625e-05,
      "step": 33384,
      "training_step_time": 7.424905300140381
    },
    {
      "epoch": 3.1838417053222654e-05,
      "model_forward_time": 1.730276107788086,
      "step": 33385
    },
    {
      "epoch": 3.1838417053222654e-05,
      "step": 33385,
      "training_step_time": 8.230478763580322
    },
    {
      "epoch": 3.1839370727539065e-05,
      "model_forward_time": 1.7170517444610596,
      "step": 33386
    },
    {
      "epoch": 3.1839370727539065e-05,
      "step": 33386,
      "training_step_time": 8.971870422363281
    },
    {
      "epoch": 3.184032440185547e-05,
      "model_forward_time": 1.709226369857788,
      "step": 33387
    },
    {
      "epoch": 3.184032440185547e-05,
      "step": 33387,
      "training_step_time": 8.207190752029419
    },
    {
      "epoch": 3.184127807617187e-05,
      "model_forward_time": 1.7063801288604736,
      "step": 33388
    },
    {
      "epoch": 3.184127807617187e-05,
      "step": 33388,
      "training_step_time": 9.54775619506836
    },
    {
      "epoch": 3.1842231750488284e-05,
      "model_forward_time": 1.7362358570098877,
      "step": 33389
    },
    {
      "epoch": 3.1842231750488284e-05,
      "step": 33389,
      "training_step_time": 7.2716686725616455
    },
    {
      "epoch": 3.184318542480469e-05,
      "grad_norm": 0.20817504823207855,
      "learning_rate": 7.95375473402732e-06,
      "loss": 0.0582,
      "step": 33390
    },
    {
      "epoch": 3.184318542480469e-05,
      "model_forward_time": 1.708449125289917,
      "step": 33390
    },
    {
      "dynamics_loss_avg": 0.04448874164372683,
      "epoch": 3.184318542480469e-05,
      "step": 33390
    },
    {
      "action_loss_avg": 0.0027946068439632654,
      "epoch": 3.184318542480469e-05,
      "step": 33390
    },
    {
      "epoch": 3.184318542480469e-05,
      "step": 33390,
      "training_step_time": 7.574857234954834
    },
    {
      "epoch": 3.184413909912109e-05,
      "model_forward_time": 1.7177953720092773,
      "step": 33391
    },
    {
      "epoch": 3.184413909912109e-05,
      "step": 33391,
      "training_step_time": 7.338839769363403
    },
    {
      "epoch": 3.18450927734375e-05,
      "model_forward_time": 1.717529058456421,
      "step": 33392
    },
    {
      "epoch": 3.18450927734375e-05,
      "step": 33392,
      "training_step_time": 7.3476457595825195
    },
    {
      "epoch": 3.184604644775391e-05,
      "model_forward_time": 1.7201628684997559,
      "step": 33393
    },
    {
      "epoch": 3.184604644775391e-05,
      "step": 33393,
      "training_step_time": 7.405536890029907
    },
    {
      "epoch": 3.184700012207031e-05,
      "model_forward_time": 1.7127878665924072,
      "step": 33394
    },
    {
      "epoch": 3.184700012207031e-05,
      "step": 33394,
      "training_step_time": 7.6378819942474365
    },
    {
      "epoch": 3.184795379638672e-05,
      "model_forward_time": 1.7073965072631836,
      "step": 33395
    },
    {
      "epoch": 3.184795379638672e-05,
      "step": 33395,
      "training_step_time": 7.627556562423706
    },
    {
      "epoch": 3.1848907470703126e-05,
      "model_forward_time": 1.7214760780334473,
      "step": 33396
    },
    {
      "epoch": 3.1848907470703126e-05,
      "step": 33396,
      "training_step_time": 8.53519582748413
    },
    {
      "epoch": 3.184986114501953e-05,
      "model_forward_time": 1.7123024463653564,
      "step": 33397
    },
    {
      "epoch": 3.184986114501953e-05,
      "step": 33397,
      "training_step_time": 8.418560981750488
    },
    {
      "epoch": 3.185081481933594e-05,
      "model_forward_time": 1.71488618850708,
      "step": 33398
    },
    {
      "epoch": 3.185081481933594e-05,
      "step": 33398,
      "training_step_time": 7.605970621109009
    },
    {
      "epoch": 3.1851768493652345e-05,
      "model_forward_time": 1.7127585411071777,
      "step": 33399
    },
    {
      "epoch": 3.1851768493652345e-05,
      "step": 33399,
      "training_step_time": 7.761162042617798
    },
    {
      "epoch": 3.185272216796875e-05,
      "grad_norm": 0.20071960985660553,
      "learning_rate": 7.952420463866182e-06,
      "loss": 0.0588,
      "step": 33400
    },
    {
      "epoch": 3.185272216796875e-05,
      "model_forward_time": 1.703559160232544,
      "step": 33400
    },
    {
      "dynamics_loss_avg": 0.0585562601685524,
      "epoch": 3.185272216796875e-05,
      "step": 33400
    },
    {
      "action_loss_avg": 0.0023605534515809268,
      "epoch": 3.185272216796875e-05,
      "step": 33400
    },
    {
      "epoch": 3.185272216796875e-05,
      "step": 33400,
      "training_step_time": 7.5513317584991455
    },
    {
      "epoch": 3.185367584228516e-05,
      "model_forward_time": 1.7101624011993408,
      "step": 33401
    },
    {
      "epoch": 3.185367584228516e-05,
      "step": 33401,
      "training_step_time": 8.059867858886719
    },
    {
      "epoch": 3.1854629516601563e-05,
      "model_forward_time": 1.702286720275879,
      "step": 33402
    },
    {
      "epoch": 3.1854629516601563e-05,
      "step": 33402,
      "training_step_time": 7.638132095336914
    },
    {
      "epoch": 3.185558319091797e-05,
      "model_forward_time": 1.7130887508392334,
      "step": 33403
    },
    {
      "epoch": 3.185558319091797e-05,
      "step": 33403,
      "training_step_time": 8.437485933303833
    },
    {
      "epoch": 3.185653686523438e-05,
      "model_forward_time": 1.733797550201416,
      "step": 33404
    },
    {
      "epoch": 3.185653686523438e-05,
      "step": 33404,
      "training_step_time": 7.948780059814453
    },
    {
      "epoch": 3.185749053955078e-05,
      "model_forward_time": 1.7167372703552246,
      "step": 33405
    },
    {
      "epoch": 3.185749053955078e-05,
      "step": 33405,
      "training_step_time": 9.046379089355469
    },
    {
      "epoch": 3.1858444213867186e-05,
      "model_forward_time": 1.7227489948272705,
      "step": 33406
    },
    {
      "epoch": 3.1858444213867186e-05,
      "step": 33406,
      "training_step_time": 8.682527303695679
    },
    {
      "epoch": 3.185939788818359e-05,
      "model_forward_time": 2.892956018447876,
      "step": 33407
    },
    {
      "epoch": 3.185939788818359e-05,
      "step": 33407,
      "training_step_time": 7.821106195449829
    },
    {
      "epoch": 3.18603515625e-05,
      "model_forward_time": 1.7177765369415283,
      "step": 33408
    },
    {
      "epoch": 3.18603515625e-05,
      "step": 33408,
      "training_step_time": 7.6874918937683105
    },
    {
      "epoch": 3.1861305236816405e-05,
      "model_forward_time": 1.7196571826934814,
      "step": 33409
    },
    {
      "epoch": 3.1861305236816405e-05,
      "step": 33409,
      "training_step_time": 7.526825428009033
    },
    {
      "epoch": 3.186225891113281e-05,
      "grad_norm": 0.14918936789035797,
      "learning_rate": 7.951085870832782e-06,
      "loss": 0.0628,
      "step": 33410
    },
    {
      "epoch": 3.186225891113281e-05,
      "model_forward_time": 1.7167909145355225,
      "step": 33410
    },
    {
      "dynamics_loss_avg": 0.06723485607653856,
      "epoch": 3.186225891113281e-05,
      "step": 33410
    },
    {
      "action_loss_avg": 0.002576318092178553,
      "epoch": 3.186225891113281e-05,
      "step": 33410
    },
    {
      "epoch": 3.186225891113281e-05,
      "step": 33410,
      "training_step_time": 7.425369739532471
    },
    {
      "epoch": 3.186321258544922e-05,
      "model_forward_time": 1.7210395336151123,
      "step": 33411
    },
    {
      "epoch": 3.186321258544922e-05,
      "step": 33411,
      "training_step_time": 7.678455829620361
    },
    {
      "epoch": 3.1864166259765624e-05,
      "model_forward_time": 1.7402901649475098,
      "step": 33412
    },
    {
      "epoch": 3.1864166259765624e-05,
      "step": 33412,
      "training_step_time": 8.420346021652222
    },
    {
      "epoch": 3.186511993408203e-05,
      "model_forward_time": 1.7265381813049316,
      "step": 33413
    },
    {
      "epoch": 3.186511993408203e-05,
      "step": 33413,
      "training_step_time": 8.54630732536316
    },
    {
      "epoch": 3.186607360839844e-05,
      "model_forward_time": 1.7199466228485107,
      "step": 33414
    },
    {
      "epoch": 3.186607360839844e-05,
      "step": 33414,
      "training_step_time": 9.136314153671265
    },
    {
      "epoch": 3.186702728271484e-05,
      "model_forward_time": 1.7173726558685303,
      "step": 33415
    },
    {
      "epoch": 3.186702728271484e-05,
      "step": 33415,
      "training_step_time": 8.55787968635559
    },
    {
      "epoch": 3.186798095703125e-05,
      "model_forward_time": 2.908738613128662,
      "step": 33416
    },
    {
      "epoch": 3.186798095703125e-05,
      "step": 33416,
      "training_step_time": 9.74380373954773
    },
    {
      "epoch": 3.186893463134766e-05,
      "model_forward_time": 1.7250494956970215,
      "step": 33417
    },
    {
      "epoch": 3.186893463134766e-05,
      "step": 33417,
      "training_step_time": 7.579561471939087
    },
    {
      "epoch": 3.186988830566406e-05,
      "model_forward_time": 1.7278406620025635,
      "step": 33418
    },
    {
      "epoch": 3.186988830566406e-05,
      "step": 33418,
      "training_step_time": 8.141152620315552
    },
    {
      "epoch": 3.1870841979980466e-05,
      "model_forward_time": 1.7189171314239502,
      "step": 33419
    },
    {
      "epoch": 3.1870841979980466e-05,
      "step": 33419,
      "training_step_time": 8.040969848632812
    },
    {
      "epoch": 3.187179565429688e-05,
      "grad_norm": 0.1757531762123108,
      "learning_rate": 7.949750955073068e-06,
      "loss": 0.0645,
      "step": 33420
    },
    {
      "epoch": 3.187179565429688e-05,
      "model_forward_time": 1.711897611618042,
      "step": 33420
    },
    {
      "dynamics_loss_avg": 0.05764628052711487,
      "epoch": 3.187179565429688e-05,
      "step": 33420
    },
    {
      "action_loss_avg": 0.002436498808674514,
      "epoch": 3.187179565429688e-05,
      "step": 33420
    },
    {
      "epoch": 3.187179565429688e-05,
      "step": 33420,
      "training_step_time": 7.551764965057373
    },
    {
      "epoch": 3.187274932861328e-05,
      "model_forward_time": 1.0051982402801514,
      "step": 33421
    },
    {
      "epoch": 3.187274932861328e-05,
      "step": 33421,
      "training_step_time": 7.443066358566284
    },
    {
      "epoch": 3.1873703002929685e-05,
      "model_forward_time": 1.7223646640777588,
      "step": 33422
    },
    {
      "epoch": 3.1873703002929685e-05,
      "step": 33422,
      "training_step_time": 8.752034187316895
    },
    {
      "epoch": 3.1874656677246095e-05,
      "model_forward_time": 1.7182936668395996,
      "step": 33423
    },
    {
      "epoch": 3.1874656677246095e-05,
      "step": 33423,
      "training_step_time": 9.466108560562134
    },
    {
      "epoch": 3.18756103515625e-05,
      "model_forward_time": 1.720374345779419,
      "step": 33424
    },
    {
      "epoch": 3.18756103515625e-05,
      "step": 33424,
      "training_step_time": 7.814649820327759
    },
    {
      "epoch": 3.1876564025878904e-05,
      "model_forward_time": 2.845435380935669,
      "step": 33425
    },
    {
      "epoch": 3.1876564025878904e-05,
      "step": 33425,
      "training_step_time": 8.754255294799805
    },
    {
      "epoch": 3.1877517700195314e-05,
      "model_forward_time": 1.7251036167144775,
      "step": 33426
    },
    {
      "epoch": 3.1877517700195314e-05,
      "step": 33426,
      "training_step_time": 8.208932161331177
    },
    {
      "epoch": 3.187847137451172e-05,
      "model_forward_time": 1.7208530902862549,
      "step": 33427
    },
    {
      "epoch": 3.187847137451172e-05,
      "step": 33427,
      "training_step_time": 7.27223801612854
    },
    {
      "epoch": 3.187942504882812e-05,
      "model_forward_time": 1.720716953277588,
      "step": 33428
    },
    {
      "epoch": 3.187942504882812e-05,
      "step": 33428,
      "training_step_time": 7.383232355117798
    },
    {
      "epoch": 3.188037872314453e-05,
      "model_forward_time": 1.722660779953003,
      "step": 33429
    },
    {
      "epoch": 3.188037872314453e-05,
      "step": 33429,
      "training_step_time": 7.419796466827393
    },
    {
      "epoch": 3.188133239746094e-05,
      "grad_norm": 0.17000003159046173,
      "learning_rate": 7.948415716733023e-06,
      "loss": 0.0598,
      "step": 33430
    },
    {
      "epoch": 3.188133239746094e-05,
      "model_forward_time": 1.736405372619629,
      "step": 33430
    },
    {
      "dynamics_loss_avg": 0.05340692810714245,
      "epoch": 3.188133239746094e-05,
      "step": 33430
    },
    {
      "action_loss_avg": 0.0021372241259086876,
      "epoch": 3.188133239746094e-05,
      "step": 33430
    },
    {
      "epoch": 3.188133239746094e-05,
      "step": 33430,
      "training_step_time": 8.135805606842041
    },
    {
      "epoch": 3.188228607177734e-05,
      "model_forward_time": 1.7201440334320068,
      "step": 33431
    },
    {
      "epoch": 3.188228607177734e-05,
      "step": 33431,
      "training_step_time": 8.42970895767212
    },
    {
      "epoch": 3.188323974609375e-05,
      "model_forward_time": 1.746814250946045,
      "step": 33432
    },
    {
      "epoch": 3.188323974609375e-05,
      "step": 33432,
      "training_step_time": 7.684237241744995
    },
    {
      "epoch": 3.1884193420410156e-05,
      "model_forward_time": 1.7243781089782715,
      "step": 33433
    },
    {
      "epoch": 3.1884193420410156e-05,
      "step": 33433,
      "training_step_time": 10.739394426345825
    },
    {
      "epoch": 3.188514709472656e-05,
      "model_forward_time": 1.7244293689727783,
      "step": 33434
    },
    {
      "epoch": 3.188514709472656e-05,
      "step": 33434,
      "training_step_time": 8.442821025848389
    },
    {
      "epoch": 3.188610076904297e-05,
      "model_forward_time": 1.7349252700805664,
      "step": 33435
    },
    {
      "epoch": 3.188610076904297e-05,
      "step": 33435,
      "training_step_time": 8.521998643875122
    },
    {
      "epoch": 3.1887054443359375e-05,
      "model_forward_time": 1.7198834419250488,
      "step": 33436
    },
    {
      "epoch": 3.1887054443359375e-05,
      "step": 33436,
      "training_step_time": 7.534746885299683
    },
    {
      "epoch": 3.188800811767578e-05,
      "model_forward_time": 1.7178394794464111,
      "step": 33437
    },
    {
      "epoch": 3.188800811767578e-05,
      "step": 33437,
      "training_step_time": 7.393738746643066
    },
    {
      "epoch": 3.188896179199219e-05,
      "model_forward_time": 1.7155213356018066,
      "step": 33438
    },
    {
      "epoch": 3.188896179199219e-05,
      "step": 33438,
      "training_step_time": 7.563977479934692
    },
    {
      "epoch": 3.1889915466308594e-05,
      "model_forward_time": 1.7415432929992676,
      "step": 33439
    },
    {
      "epoch": 3.1889915466308594e-05,
      "step": 33439,
      "training_step_time": 7.569982051849365
    },
    {
      "epoch": 3.1890869140625e-05,
      "grad_norm": 0.15741120278835297,
      "learning_rate": 7.947080155958668e-06,
      "loss": 0.0568,
      "step": 33440
    },
    {
      "epoch": 3.1890869140625e-05,
      "model_forward_time": 1.6997241973876953,
      "step": 33440
    },
    {
      "dynamics_loss_avg": 0.04251492638140917,
      "epoch": 3.1890869140625e-05,
      "step": 33440
    },
    {
      "action_loss_avg": 0.0022388215002138168,
      "epoch": 3.1890869140625e-05,
      "step": 33440
    },
    {
      "epoch": 3.1890869140625e-05,
      "step": 33440,
      "training_step_time": 7.7169318199157715
    },
    {
      "epoch": 3.189182281494141e-05,
      "model_forward_time": 1.7132785320281982,
      "step": 33441
    },
    {
      "epoch": 3.189182281494141e-05,
      "step": 33441,
      "training_step_time": 7.651602506637573
    },
    {
      "epoch": 3.189277648925781e-05,
      "model_forward_time": 1.7129161357879639,
      "step": 33442
    },
    {
      "epoch": 3.189277648925781e-05,
      "step": 33442,
      "training_step_time": 7.932190179824829
    },
    {
      "epoch": 3.189373016357422e-05,
      "model_forward_time": 1.7142717838287354,
      "step": 33443
    },
    {
      "epoch": 3.189373016357422e-05,
      "step": 33443,
      "training_step_time": 7.785718679428101
    },
    {
      "epoch": 3.189468383789063e-05,
      "model_forward_time": 2.9279897212982178,
      "step": 33444
    },
    {
      "epoch": 3.189468383789063e-05,
      "step": 33444,
      "training_step_time": 7.852354526519775
    },
    {
      "epoch": 3.189563751220703e-05,
      "model_forward_time": 1.7126579284667969,
      "step": 33445
    },
    {
      "epoch": 3.189563751220703e-05,
      "step": 33445,
      "training_step_time": 7.705726146697998
    },
    {
      "epoch": 3.1896591186523436e-05,
      "model_forward_time": 1.7135305404663086,
      "step": 33446
    },
    {
      "epoch": 3.1896591186523436e-05,
      "step": 33446,
      "training_step_time": 7.484266519546509
    },
    {
      "epoch": 3.1897544860839846e-05,
      "model_forward_time": 1.7104907035827637,
      "step": 33447
    },
    {
      "epoch": 3.1897544860839846e-05,
      "step": 33447,
      "training_step_time": 7.476313591003418
    },
    {
      "epoch": 3.189849853515625e-05,
      "model_forward_time": 1.7260079383850098,
      "step": 33448
    },
    {
      "epoch": 3.189849853515625e-05,
      "step": 33448,
      "training_step_time": 8.0413076877594
    },
    {
      "epoch": 3.1899452209472654e-05,
      "model_forward_time": 1.7129578590393066,
      "step": 33449
    },
    {
      "epoch": 3.1899452209472654e-05,
      "step": 33449,
      "training_step_time": 7.679544448852539
    },
    {
      "epoch": 3.1900405883789065e-05,
      "grad_norm": 0.13586637377738953,
      "learning_rate": 7.94574427289606e-06,
      "loss": 0.0599,
      "step": 33450
    },
    {
      "epoch": 3.1900405883789065e-05,
      "model_forward_time": 1.7067232131958008,
      "step": 33450
    },
    {
      "dynamics_loss_avg": 0.06403215825557709,
      "epoch": 3.1900405883789065e-05,
      "step": 33450
    },
    {
      "action_loss_avg": 0.0020918477093800902,
      "epoch": 3.1900405883789065e-05,
      "step": 33450
    },
    {
      "epoch": 3.1900405883789065e-05,
      "step": 33450,
      "training_step_time": 7.6718590259552
    },
    {
      "epoch": 3.190135955810547e-05,
      "model_forward_time": 1.7427959442138672,
      "step": 33451
    },
    {
      "epoch": 3.190135955810547e-05,
      "step": 33451,
      "training_step_time": 7.69339919090271
    },
    {
      "epoch": 3.190231323242187e-05,
      "model_forward_time": 1.7129883766174316,
      "step": 33452
    },
    {
      "epoch": 3.190231323242187e-05,
      "step": 33452,
      "training_step_time": 8.849746227264404
    },
    {
      "epoch": 3.1903266906738284e-05,
      "model_forward_time": 2.8866384029388428,
      "step": 33453
    },
    {
      "epoch": 3.1903266906738284e-05,
      "step": 33453,
      "training_step_time": 9.063849687576294
    },
    {
      "epoch": 3.190422058105469e-05,
      "model_forward_time": 1.7173361778259277,
      "step": 33454
    },
    {
      "epoch": 3.190422058105469e-05,
      "step": 33454,
      "training_step_time": 7.523223400115967
    },
    {
      "epoch": 3.190517425537109e-05,
      "model_forward_time": 1.7067391872406006,
      "step": 33455
    },
    {
      "epoch": 3.190517425537109e-05,
      "step": 33455,
      "training_step_time": 7.93154239654541
    },
    {
      "epoch": 3.19061279296875e-05,
      "model_forward_time": 1.7081263065338135,
      "step": 33456
    },
    {
      "epoch": 3.19061279296875e-05,
      "step": 33456,
      "training_step_time": 7.5003662109375
    },
    {
      "epoch": 3.190708160400391e-05,
      "model_forward_time": 1.7122435569763184,
      "step": 33457
    },
    {
      "epoch": 3.190708160400391e-05,
      "step": 33457,
      "training_step_time": 8.012418508529663
    },
    {
      "epoch": 3.190803527832031e-05,
      "model_forward_time": 1.7397966384887695,
      "step": 33458
    },
    {
      "epoch": 3.190803527832031e-05,
      "step": 33458,
      "training_step_time": 8.153663873672485
    },
    {
      "epoch": 3.190898895263672e-05,
      "model_forward_time": 1.7092208862304688,
      "step": 33459
    },
    {
      "epoch": 3.190898895263672e-05,
      "step": 33459,
      "training_step_time": 8.113698720932007
    },
    {
      "epoch": 3.1909942626953126e-05,
      "grad_norm": 0.17855823040008545,
      "learning_rate": 7.944408067691285e-06,
      "loss": 0.0579,
      "step": 33460
    },
    {
      "epoch": 3.1909942626953126e-05,
      "model_forward_time": 1.7331397533416748,
      "step": 33460
    },
    {
      "dynamics_loss_avg": 0.03901279382407665,
      "epoch": 3.1909942626953126e-05,
      "step": 33460
    },
    {
      "action_loss_avg": 0.0013481718022376298,
      "epoch": 3.1909942626953126e-05,
      "step": 33460
    },
    {
      "epoch": 3.1909942626953126e-05,
      "step": 33460,
      "training_step_time": 8.427996158599854
    },
    {
      "epoch": 3.191089630126953e-05,
      "model_forward_time": 1.7103233337402344,
      "step": 33461
    },
    {
      "epoch": 3.191089630126953e-05,
      "step": 33461,
      "training_step_time": 8.554765701293945
    },
    {
      "epoch": 3.191184997558594e-05,
      "model_forward_time": 2.8713743686676025,
      "step": 33462
    },
    {
      "epoch": 3.191184997558594e-05,
      "step": 33462,
      "training_step_time": 7.814354419708252
    },
    {
      "epoch": 3.1912803649902345e-05,
      "model_forward_time": 1.7141623497009277,
      "step": 33463
    },
    {
      "epoch": 3.1912803649902345e-05,
      "step": 33463,
      "training_step_time": 7.729143381118774
    },
    {
      "epoch": 3.191375732421875e-05,
      "model_forward_time": 1.7163190841674805,
      "step": 33464
    },
    {
      "epoch": 3.191375732421875e-05,
      "step": 33464,
      "training_step_time": 7.5775392055511475
    },
    {
      "epoch": 3.191471099853516e-05,
      "model_forward_time": 1.7130351066589355,
      "step": 33465
    },
    {
      "epoch": 3.191471099853516e-05,
      "step": 33465,
      "training_step_time": 7.6490678787231445
    },
    {
      "epoch": 3.1915664672851564e-05,
      "model_forward_time": 1.7084941864013672,
      "step": 33466
    },
    {
      "epoch": 3.1915664672851564e-05,
      "step": 33466,
      "training_step_time": 8.316410541534424
    },
    {
      "epoch": 3.191661834716797e-05,
      "model_forward_time": 1.7292718887329102,
      "step": 33467
    },
    {
      "epoch": 3.191661834716797e-05,
      "step": 33467,
      "training_step_time": 7.568938732147217
    },
    {
      "epoch": 3.191757202148438e-05,
      "model_forward_time": 1.717871904373169,
      "step": 33468
    },
    {
      "epoch": 3.191757202148438e-05,
      "step": 33468,
      "training_step_time": 7.675904035568237
    },
    {
      "epoch": 3.191852569580078e-05,
      "model_forward_time": 1.7190494537353516,
      "step": 33469
    },
    {
      "epoch": 3.191852569580078e-05,
      "step": 33469,
      "training_step_time": 8.655461549758911
    },
    {
      "epoch": 3.1919479370117186e-05,
      "grad_norm": 0.14223654568195343,
      "learning_rate": 7.943071540490473e-06,
      "loss": 0.0639,
      "step": 33470
    },
    {
      "epoch": 3.1919479370117186e-05,
      "model_forward_time": 1.7089667320251465,
      "step": 33470
    },
    {
      "dynamics_loss_avg": 0.04025614932179451,
      "epoch": 3.1919479370117186e-05,
      "step": 33470
    },
    {
      "action_loss_avg": 0.002232257992727682,
      "epoch": 3.1919479370117186e-05,
      "step": 33470
    },
    {
      "epoch": 3.1919479370117186e-05,
      "step": 33470,
      "training_step_time": 7.650673151016235
    },
    {
      "epoch": 3.192043304443359e-05,
      "model_forward_time": 2.8490843772888184,
      "step": 33471
    },
    {
      "epoch": 3.192043304443359e-05,
      "step": 33471,
      "training_step_time": 8.635060787200928
    },
    {
      "epoch": 3.192138671875e-05,
      "model_forward_time": 1.7143831253051758,
      "step": 33472
    },
    {
      "epoch": 3.192138671875e-05,
      "step": 33472,
      "training_step_time": 7.603721618652344
    },
    {
      "epoch": 3.1922340393066405e-05,
      "model_forward_time": 1.7122626304626465,
      "step": 33473
    },
    {
      "epoch": 3.1922340393066405e-05,
      "step": 33473,
      "training_step_time": 7.769219160079956
    },
    {
      "epoch": 3.192329406738281e-05,
      "model_forward_time": 1.7153129577636719,
      "step": 33474
    },
    {
      "epoch": 3.192329406738281e-05,
      "step": 33474,
      "training_step_time": 7.771387100219727
    },
    {
      "epoch": 3.192424774169922e-05,
      "model_forward_time": 1.7158091068267822,
      "step": 33475
    },
    {
      "epoch": 3.192424774169922e-05,
      "step": 33475,
      "training_step_time": 7.4678051471710205
    },
    {
      "epoch": 3.1925201416015624e-05,
      "model_forward_time": 1.7320597171783447,
      "step": 33476
    },
    {
      "epoch": 3.1925201416015624e-05,
      "step": 33476,
      "training_step_time": 8.490568161010742
    },
    {
      "epoch": 3.192615509033203e-05,
      "model_forward_time": 1.7077224254608154,
      "step": 33477
    },
    {
      "epoch": 3.192615509033203e-05,
      "step": 33477,
      "training_step_time": 8.5953049659729
    },
    {
      "epoch": 3.192710876464844e-05,
      "model_forward_time": 1.716545820236206,
      "step": 33478
    },
    {
      "epoch": 3.192710876464844e-05,
      "step": 33478,
      "training_step_time": 8.361034154891968
    },
    {
      "epoch": 3.192806243896484e-05,
      "model_forward_time": 1.7072055339813232,
      "step": 33479
    },
    {
      "epoch": 3.192806243896484e-05,
      "step": 33479,
      "training_step_time": 9.151624917984009
    },
    {
      "epoch": 3.192901611328125e-05,
      "grad_norm": 0.12328817695379257,
      "learning_rate": 7.941734691439781e-06,
      "loss": 0.0611,
      "step": 33480
    },
    {
      "epoch": 3.192901611328125e-05,
      "model_forward_time": 2.8036458492279053,
      "step": 33480
    },
    {
      "dynamics_loss_avg": 0.056076755933463576,
      "epoch": 3.192901611328125e-05,
      "step": 33480
    },
    {
      "action_loss_avg": 0.0025049986492376775,
      "epoch": 3.192901611328125e-05,
      "step": 33480
    },
    {
      "epoch": 3.192901611328125e-05,
      "step": 33480,
      "training_step_time": 8.948100566864014
    },
    {
      "epoch": 3.192996978759766e-05,
      "model_forward_time": 1.7129411697387695,
      "step": 33481
    },
    {
      "epoch": 3.192996978759766e-05,
      "step": 33481,
      "training_step_time": 8.4510977268219
    },
    {
      "epoch": 3.193092346191406e-05,
      "model_forward_time": 1.7170679569244385,
      "step": 33482
    },
    {
      "epoch": 3.193092346191406e-05,
      "step": 33482,
      "training_step_time": 7.753616571426392
    },
    {
      "epoch": 3.1931877136230466e-05,
      "model_forward_time": 1.7102770805358887,
      "step": 33483
    },
    {
      "epoch": 3.1931877136230466e-05,
      "step": 33483,
      "training_step_time": 7.490082502365112
    },
    {
      "epoch": 3.193283081054688e-05,
      "model_forward_time": 1.7149863243103027,
      "step": 33484
    },
    {
      "epoch": 3.193283081054688e-05,
      "step": 33484,
      "training_step_time": 7.488214015960693
    },
    {
      "epoch": 3.193378448486328e-05,
      "model_forward_time": 1.7231643199920654,
      "step": 33485
    },
    {
      "epoch": 3.193378448486328e-05,
      "step": 33485,
      "training_step_time": 7.740710735321045
    },
    {
      "epoch": 3.1934738159179685e-05,
      "model_forward_time": 1.3406894207000732,
      "step": 33486
    },
    {
      "epoch": 3.1934738159179685e-05,
      "step": 33486,
      "training_step_time": 7.798693895339966
    },
    {
      "epoch": 3.1935691833496096e-05,
      "model_forward_time": 1.7088823318481445,
      "step": 33487
    },
    {
      "epoch": 3.1935691833496096e-05,
      "step": 33487,
      "training_step_time": 7.9360527992248535
    },
    {
      "epoch": 3.19366455078125e-05,
      "model_forward_time": 1.7164232730865479,
      "step": 33488
    },
    {
      "epoch": 3.19366455078125e-05,
      "step": 33488,
      "training_step_time": 8.816195726394653
    },
    {
      "epoch": 3.1937599182128904e-05,
      "model_forward_time": 1.340977668762207,
      "step": 33489
    },
    {
      "epoch": 3.1937599182128904e-05,
      "step": 33489,
      "training_step_time": 8.757405996322632
    },
    {
      "epoch": 3.1938552856445314e-05,
      "grad_norm": 0.12363503873348236,
      "learning_rate": 7.940397520685406e-06,
      "loss": 0.0613,
      "step": 33490
    },
    {
      "epoch": 3.1938552856445314e-05,
      "model_forward_time": 2.855362892150879,
      "step": 33490
    },
    {
      "dynamics_loss_avg": 0.06630742345005274,
      "epoch": 3.1938552856445314e-05,
      "step": 33490
    },
    {
      "action_loss_avg": 0.0023945824068505317,
      "epoch": 3.1938552856445314e-05,
      "step": 33490
    },
    {
      "epoch": 3.1938552856445314e-05,
      "step": 33490,
      "training_step_time": 8.959190845489502
    },
    {
      "epoch": 3.193950653076172e-05,
      "model_forward_time": 1.7128944396972656,
      "step": 33491
    },
    {
      "epoch": 3.193950653076172e-05,
      "step": 33491,
      "training_step_time": 7.7589757442474365
    },
    {
      "epoch": 3.194046020507812e-05,
      "model_forward_time": 1.7067899703979492,
      "step": 33492
    },
    {
      "epoch": 3.194046020507812e-05,
      "step": 33492,
      "training_step_time": 8.138191938400269
    },
    {
      "epoch": 3.194141387939453e-05,
      "model_forward_time": 1.7036819458007812,
      "step": 33493
    },
    {
      "epoch": 3.194141387939453e-05,
      "step": 33493,
      "training_step_time": 7.581676244735718
    },
    {
      "epoch": 3.194236755371094e-05,
      "model_forward_time": 1.7054502964019775,
      "step": 33494
    },
    {
      "epoch": 3.194236755371094e-05,
      "step": 33494,
      "training_step_time": 8.412885427474976
    },
    {
      "epoch": 3.194332122802734e-05,
      "model_forward_time": 1.7080986499786377,
      "step": 33495
    },
    {
      "epoch": 3.194332122802734e-05,
      "step": 33495,
      "training_step_time": 7.573782444000244
    },
    {
      "epoch": 3.194427490234375e-05,
      "model_forward_time": 1.70241117477417,
      "step": 33496
    },
    {
      "epoch": 3.194427490234375e-05,
      "step": 33496,
      "training_step_time": 9.878331899642944
    },
    {
      "epoch": 3.1945228576660156e-05,
      "model_forward_time": 1.7308757305145264,
      "step": 33497
    },
    {
      "epoch": 3.1945228576660156e-05,
      "step": 33497,
      "training_step_time": 9.786820650100708
    },
    {
      "epoch": 3.194618225097656e-05,
      "model_forward_time": 0.9944498538970947,
      "step": 33498
    },
    {
      "epoch": 3.194618225097656e-05,
      "step": 33498,
      "training_step_time": 9.014426231384277
    },
    {
      "epoch": 3.194713592529297e-05,
      "model_forward_time": 2.7854342460632324,
      "step": 33499
    },
    {
      "epoch": 3.194713592529297e-05,
      "step": 33499,
      "training_step_time": 7.95737624168396
    },
    {
      "epoch": 3.1948089599609375e-05,
      "grad_norm": 0.16257236897945404,
      "learning_rate": 7.939060028373578e-06,
      "loss": 0.0639,
      "step": 33500
    },
    {
      "epoch": 3.1948089599609375e-05,
      "model_forward_time": 1.7165422439575195,
      "step": 33500
    },
    {
      "dynamics_loss_avg": 0.04950669966638088,
      "epoch": 3.1948089599609375e-05,
      "step": 33500
    },
    {
      "action_loss_avg": 0.0023341299791354688,
      "epoch": 3.1948089599609375e-05,
      "step": 33500
    },
    {
      "epoch": 3.1948089599609375e-05,
      "step": 33500,
      "training_step_time": 7.723708391189575
    },
    {
      "epoch": 3.194904327392578e-05,
      "model_forward_time": 1.7138042449951172,
      "step": 33501
    },
    {
      "epoch": 3.194904327392578e-05,
      "step": 33501,
      "training_step_time": 8.2148916721344
    },
    {
      "epoch": 3.194999694824219e-05,
      "model_forward_time": 1.705146074295044,
      "step": 33502
    },
    {
      "epoch": 3.194999694824219e-05,
      "step": 33502,
      "training_step_time": 7.548851490020752
    },
    {
      "epoch": 3.1950950622558594e-05,
      "model_forward_time": 1.7082850933074951,
      "step": 33503
    },
    {
      "epoch": 3.1950950622558594e-05,
      "step": 33503,
      "training_step_time": 7.557857275009155
    },
    {
      "epoch": 3.1951904296875e-05,
      "model_forward_time": 1.7308311462402344,
      "step": 33504
    },
    {
      "epoch": 3.1951904296875e-05,
      "step": 33504,
      "training_step_time": 7.6358537673950195
    },
    {
      "epoch": 3.195285797119141e-05,
      "model_forward_time": 1.7042558193206787,
      "step": 33505
    },
    {
      "epoch": 3.195285797119141e-05,
      "step": 33505,
      "training_step_time": 8.519351243972778
    },
    {
      "epoch": 3.195381164550781e-05,
      "model_forward_time": 1.7184317111968994,
      "step": 33506
    },
    {
      "epoch": 3.195381164550781e-05,
      "step": 33506,
      "training_step_time": 8.86948275566101
    },
    {
      "epoch": 3.195476531982422e-05,
      "model_forward_time": 0.9963092803955078,
      "step": 33507
    },
    {
      "epoch": 3.195476531982422e-05,
      "step": 33507,
      "training_step_time": 8.90520429611206
    },
    {
      "epoch": 3.195571899414063e-05,
      "model_forward_time": 1.7078149318695068,
      "step": 33508
    },
    {
      "epoch": 3.195571899414063e-05,
      "step": 33508,
      "training_step_time": 8.024733543395996
    },
    {
      "epoch": 3.195667266845703e-05,
      "model_forward_time": 2.873213529586792,
      "step": 33509
    },
    {
      "epoch": 3.195667266845703e-05,
      "step": 33509,
      "training_step_time": 7.775530099868774
    },
    {
      "epoch": 3.1957626342773436e-05,
      "grad_norm": 0.21868926286697388,
      "learning_rate": 7.937722214650565e-06,
      "loss": 0.0644,
      "step": 33510
    },
    {
      "epoch": 3.1957626342773436e-05,
      "model_forward_time": 1.7073283195495605,
      "step": 33510
    },
    {
      "dynamics_loss_avg": 0.06810741461813449,
      "epoch": 3.1957626342773436e-05,
      "step": 33510
    },
    {
      "action_loss_avg": 0.0032335558062186465,
      "epoch": 3.1957626342773436e-05,
      "step": 33510
    },
    {
      "epoch": 3.1957626342773436e-05,
      "step": 33510,
      "training_step_time": 7.705002784729004
    },
    {
      "epoch": 3.1958580017089846e-05,
      "model_forward_time": 1.7185418605804443,
      "step": 33511
    },
    {
      "epoch": 3.1958580017089846e-05,
      "step": 33511,
      "training_step_time": 7.486314535140991
    },
    {
      "epoch": 3.195953369140625e-05,
      "model_forward_time": 1.700021743774414,
      "step": 33512
    },
    {
      "epoch": 3.195953369140625e-05,
      "step": 33512,
      "training_step_time": 7.427928924560547
    },
    {
      "epoch": 3.1960487365722654e-05,
      "model_forward_time": 1.7319056987762451,
      "step": 33513
    },
    {
      "epoch": 3.1960487365722654e-05,
      "step": 33513,
      "training_step_time": 7.590282440185547
    },
    {
      "epoch": 3.1961441040039065e-05,
      "model_forward_time": 1.7089431285858154,
      "step": 33514
    },
    {
      "epoch": 3.1961441040039065e-05,
      "step": 33514,
      "training_step_time": 8.828323364257812
    },
    {
      "epoch": 3.196239471435547e-05,
      "model_forward_time": 1.7169735431671143,
      "step": 33515
    },
    {
      "epoch": 3.196239471435547e-05,
      "step": 33515,
      "training_step_time": 8.502321243286133
    },
    {
      "epoch": 3.196334838867187e-05,
      "model_forward_time": 1.710113763809204,
      "step": 33516
    },
    {
      "epoch": 3.196334838867187e-05,
      "step": 33516,
      "training_step_time": 8.950935125350952
    },
    {
      "epoch": 3.1964302062988284e-05,
      "model_forward_time": 1.7054469585418701,
      "step": 33517
    },
    {
      "epoch": 3.1964302062988284e-05,
      "step": 33517,
      "training_step_time": 9.585818529129028
    },
    {
      "epoch": 3.196525573730469e-05,
      "model_forward_time": 2.8096156120300293,
      "step": 33518
    },
    {
      "epoch": 3.196525573730469e-05,
      "step": 33518,
      "training_step_time": 7.841616630554199
    },
    {
      "epoch": 3.196620941162109e-05,
      "model_forward_time": 1.714195966720581,
      "step": 33519
    },
    {
      "epoch": 3.196620941162109e-05,
      "step": 33519,
      "training_step_time": 7.793790102005005
    },
    {
      "epoch": 3.19671630859375e-05,
      "grad_norm": 0.11452074348926544,
      "learning_rate": 7.936384079662666e-06,
      "loss": 0.0655,
      "step": 33520
    },
    {
      "epoch": 3.19671630859375e-05,
      "model_forward_time": 1.7084245681762695,
      "step": 33520
    },
    {
      "dynamics_loss_avg": 0.04513445757329464,
      "epoch": 3.19671630859375e-05,
      "step": 33520
    },
    {
      "action_loss_avg": 0.001865898523828946,
      "epoch": 3.19671630859375e-05,
      "step": 33520
    },
    {
      "epoch": 3.19671630859375e-05,
      "step": 33520,
      "training_step_time": 7.585831642150879
    },
    {
      "epoch": 3.196811676025391e-05,
      "model_forward_time": 1.70206880569458,
      "step": 33521
    },
    {
      "epoch": 3.196811676025391e-05,
      "step": 33521,
      "training_step_time": 7.4933178424835205
    },
    {
      "epoch": 3.196907043457031e-05,
      "model_forward_time": 1.724165439605713,
      "step": 33522
    },
    {
      "epoch": 3.196907043457031e-05,
      "step": 33522,
      "training_step_time": 8.321485042572021
    },
    {
      "epoch": 3.197002410888672e-05,
      "model_forward_time": 1.708855152130127,
      "step": 33523
    },
    {
      "epoch": 3.197002410888672e-05,
      "step": 33523,
      "training_step_time": 7.8368871212005615
    },
    {
      "epoch": 3.1970977783203126e-05,
      "model_forward_time": 1.7142417430877686,
      "step": 33524
    },
    {
      "epoch": 3.1970977783203126e-05,
      "step": 33524,
      "training_step_time": 7.878392457962036
    },
    {
      "epoch": 3.197193145751953e-05,
      "model_forward_time": 1.7092366218566895,
      "step": 33525
    },
    {
      "epoch": 3.197193145751953e-05,
      "step": 33525,
      "training_step_time": 8.928240537643433
    },
    {
      "epoch": 3.197288513183594e-05,
      "model_forward_time": 1.7095069885253906,
      "step": 33526
    },
    {
      "epoch": 3.197288513183594e-05,
      "step": 33526,
      "training_step_time": 8.001274824142456
    },
    {
      "epoch": 3.1973838806152345e-05,
      "model_forward_time": 2.818305492401123,
      "step": 33527
    },
    {
      "epoch": 3.1973838806152345e-05,
      "step": 33527,
      "training_step_time": 8.622472286224365
    },
    {
      "epoch": 3.197479248046875e-05,
      "model_forward_time": 1.7122316360473633,
      "step": 33528
    },
    {
      "epoch": 3.197479248046875e-05,
      "step": 33528,
      "training_step_time": 7.832762718200684
    },
    {
      "epoch": 3.197574615478516e-05,
      "model_forward_time": 1.713745355606079,
      "step": 33529
    },
    {
      "epoch": 3.197574615478516e-05,
      "step": 33529,
      "training_step_time": 7.541612863540649
    },
    {
      "epoch": 3.1976699829101564e-05,
      "grad_norm": 0.18991251289844513,
      "learning_rate": 7.93504562355622e-06,
      "loss": 0.0647,
      "step": 33530
    },
    {
      "epoch": 3.1976699829101564e-05,
      "model_forward_time": 1.7109436988830566,
      "step": 33530
    },
    {
      "dynamics_loss_avg": 0.07202614434063434,
      "epoch": 3.1976699829101564e-05,
      "step": 33530
    },
    {
      "action_loss_avg": 0.0022893122921232136,
      "epoch": 3.1976699829101564e-05,
      "step": 33530
    },
    {
      "epoch": 3.1976699829101564e-05,
      "step": 33530,
      "training_step_time": 7.556056022644043
    },
    {
      "epoch": 3.197765350341797e-05,
      "model_forward_time": 1.7150743007659912,
      "step": 33531
    },
    {
      "epoch": 3.197765350341797e-05,
      "step": 33531,
      "training_step_time": 7.615962028503418
    },
    {
      "epoch": 3.197860717773437e-05,
      "model_forward_time": 1.7126567363739014,
      "step": 33532
    },
    {
      "epoch": 3.197860717773437e-05,
      "step": 33532,
      "training_step_time": 7.776203870773315
    },
    {
      "epoch": 3.197956085205078e-05,
      "model_forward_time": 1.7116725444793701,
      "step": 33533
    },
    {
      "epoch": 3.197956085205078e-05,
      "step": 33533,
      "training_step_time": 8.40516996383667
    },
    {
      "epoch": 3.1980514526367186e-05,
      "model_forward_time": 1.7276132106781006,
      "step": 33534
    },
    {
      "epoch": 3.1980514526367186e-05,
      "step": 33534,
      "training_step_time": 8.525087594985962
    },
    {
      "epoch": 3.198146820068359e-05,
      "model_forward_time": 1.7114360332489014,
      "step": 33535
    },
    {
      "epoch": 3.198146820068359e-05,
      "step": 33535,
      "training_step_time": 7.920651197433472
    },
    {
      "epoch": 3.1982421875e-05,
      "model_forward_time": 1.712181568145752,
      "step": 33536
    },
    {
      "epoch": 3.1982421875e-05,
      "step": 33536,
      "training_step_time": 9.102794408798218
    },
    {
      "epoch": 3.1983375549316405e-05,
      "model_forward_time": 1.7099652290344238,
      "step": 33537
    },
    {
      "epoch": 3.1983375549316405e-05,
      "step": 33537,
      "training_step_time": 7.75377893447876
    },
    {
      "epoch": 3.198432922363281e-05,
      "model_forward_time": 1.7075464725494385,
      "step": 33538
    },
    {
      "epoch": 3.198432922363281e-05,
      "step": 33538,
      "training_step_time": 7.4081127643585205
    },
    {
      "epoch": 3.198528289794922e-05,
      "model_forward_time": 1.7127470970153809,
      "step": 33539
    },
    {
      "epoch": 3.198528289794922e-05,
      "step": 33539,
      "training_step_time": 7.5378577709198
    },
    {
      "epoch": 3.1986236572265624e-05,
      "grad_norm": 0.17377649247646332,
      "learning_rate": 7.933706846477598e-06,
      "loss": 0.0633,
      "step": 33540
    },
    {
      "epoch": 3.1986236572265624e-05,
      "model_forward_time": 1.3397212028503418,
      "step": 33540
    },
    {
      "dynamics_loss_avg": 0.06407939344644546,
      "epoch": 3.1986236572265624e-05,
      "step": 33540
    },
    {
      "action_loss_avg": 0.001981913176132366,
      "epoch": 3.1986236572265624e-05,
      "step": 33540
    },
    {
      "epoch": 3.1986236572265624e-05,
      "step": 33540,
      "training_step_time": 8.436775922775269
    },
    {
      "epoch": 3.198719024658203e-05,
      "model_forward_time": 1.729598045349121,
      "step": 33541
    },
    {
      "epoch": 3.198719024658203e-05,
      "step": 33541,
      "training_step_time": 7.702654600143433
    },
    {
      "epoch": 3.198814392089844e-05,
      "model_forward_time": 1.701892614364624,
      "step": 33542
    },
    {
      "epoch": 3.198814392089844e-05,
      "step": 33542,
      "training_step_time": 8.675645351409912
    },
    {
      "epoch": 3.198909759521484e-05,
      "model_forward_time": 1.7193467617034912,
      "step": 33543
    },
    {
      "epoch": 3.198909759521484e-05,
      "step": 33543,
      "training_step_time": 8.49646806716919
    },
    {
      "epoch": 3.199005126953125e-05,
      "model_forward_time": 1.7017548084259033,
      "step": 33544
    },
    {
      "epoch": 3.199005126953125e-05,
      "step": 33544,
      "training_step_time": 7.924273252487183
    },
    {
      "epoch": 3.199100494384766e-05,
      "model_forward_time": 1.7023975849151611,
      "step": 33545
    },
    {
      "epoch": 3.199100494384766e-05,
      "step": 33545,
      "training_step_time": 8.876382827758789
    },
    {
      "epoch": 3.199195861816406e-05,
      "model_forward_time": 2.8482542037963867,
      "step": 33546
    },
    {
      "epoch": 3.199195861816406e-05,
      "step": 33546,
      "training_step_time": 7.8709800243377686
    },
    {
      "epoch": 3.1992912292480466e-05,
      "model_forward_time": 1.712514877319336,
      "step": 33547
    },
    {
      "epoch": 3.1992912292480466e-05,
      "step": 33547,
      "training_step_time": 7.271165370941162
    },
    {
      "epoch": 3.199386596679688e-05,
      "model_forward_time": 1.7039918899536133,
      "step": 33548
    },
    {
      "epoch": 3.199386596679688e-05,
      "step": 33548,
      "training_step_time": 7.557707071304321
    },
    {
      "epoch": 3.199481964111328e-05,
      "model_forward_time": 1.7134172916412354,
      "step": 33549
    },
    {
      "epoch": 3.199481964111328e-05,
      "step": 33549,
      "training_step_time": 8.864823579788208
    },
    {
      "epoch": 3.1995773315429685e-05,
      "grad_norm": 0.14688514173030853,
      "learning_rate": 7.932367748573206e-06,
      "loss": 0.0621,
      "step": 33550
    },
    {
      "epoch": 3.1995773315429685e-05,
      "model_forward_time": 1.7391364574432373,
      "step": 33550
    },
    {
      "dynamics_loss_avg": 0.06773941684514284,
      "epoch": 3.1995773315429685e-05,
      "step": 33550
    },
    {
      "action_loss_avg": 0.0035331986029632388,
      "epoch": 3.1995773315429685e-05,
      "step": 33550
    },
    {
      "epoch": 3.1995773315429685e-05,
      "step": 33550,
      "training_step_time": 7.570014715194702
    },
    {
      "epoch": 3.1996726989746096e-05,
      "model_forward_time": 1.713045358657837,
      "step": 33551
    },
    {
      "epoch": 3.1996726989746096e-05,
      "step": 33551,
      "training_step_time": 7.937895059585571
    },
    {
      "epoch": 3.19976806640625e-05,
      "model_forward_time": 1.701890230178833,
      "step": 33552
    },
    {
      "epoch": 3.19976806640625e-05,
      "step": 33552,
      "training_step_time": 9.946072816848755
    },
    {
      "epoch": 3.1998634338378904e-05,
      "model_forward_time": 1.7063698768615723,
      "step": 33553
    },
    {
      "epoch": 3.1998634338378904e-05,
      "step": 33553,
      "training_step_time": 7.976821184158325
    },
    {
      "epoch": 3.1999588012695314e-05,
      "model_forward_time": 1.707047462463379,
      "step": 33554
    },
    {
      "epoch": 3.1999588012695314e-05,
      "step": 33554,
      "training_step_time": 8.958322286605835
    },
    {
      "epoch": 3.200054168701172e-05,
      "model_forward_time": 2.8455874919891357,
      "step": 33555
    },
    {
      "epoch": 3.200054168701172e-05,
      "step": 33555,
      "training_step_time": 7.919761419296265
    },
    {
      "epoch": 3.200149536132812e-05,
      "model_forward_time": 1.7043464183807373,
      "step": 33556
    },
    {
      "epoch": 3.200149536132812e-05,
      "step": 33556,
      "training_step_time": 7.852071762084961
    },
    {
      "epoch": 3.200244903564453e-05,
      "model_forward_time": 1.705261468887329,
      "step": 33557
    },
    {
      "epoch": 3.200244903564453e-05,
      "step": 33557,
      "training_step_time": 7.667718410491943
    },
    {
      "epoch": 3.200340270996094e-05,
      "model_forward_time": 1.7032127380371094,
      "step": 33558
    },
    {
      "epoch": 3.200340270996094e-05,
      "step": 33558,
      "training_step_time": 7.921278953552246
    },
    {
      "epoch": 3.200435638427734e-05,
      "model_forward_time": 1.7254092693328857,
      "step": 33559
    },
    {
      "epoch": 3.200435638427734e-05,
      "step": 33559,
      "training_step_time": 7.652672529220581
    },
    {
      "epoch": 3.200531005859375e-05,
      "grad_norm": 0.12783217430114746,
      "learning_rate": 7.931028329989485e-06,
      "loss": 0.0617,
      "step": 33560
    },
    {
      "epoch": 3.200531005859375e-05,
      "model_forward_time": 1.705125093460083,
      "step": 33560
    },
    {
      "dynamics_loss_avg": 0.04028096850961447,
      "epoch": 3.200531005859375e-05,
      "step": 33560
    },
    {
      "action_loss_avg": 0.0011104576056823134,
      "epoch": 3.200531005859375e-05,
      "step": 33560
    },
    {
      "epoch": 3.200531005859375e-05,
      "step": 33560,
      "training_step_time": 7.862604379653931
    },
    {
      "epoch": 3.2006263732910156e-05,
      "model_forward_time": 1.7093045711517334,
      "step": 33561
    },
    {
      "epoch": 3.2006263732910156e-05,
      "step": 33561,
      "training_step_time": 7.889700651168823
    },
    {
      "epoch": 3.200721740722656e-05,
      "model_forward_time": 1.7307636737823486,
      "step": 33562
    },
    {
      "epoch": 3.200721740722656e-05,
      "step": 33562,
      "training_step_time": 8.396012544631958
    },
    {
      "epoch": 3.200817108154297e-05,
      "model_forward_time": 1.7052099704742432,
      "step": 33563
    },
    {
      "epoch": 3.200817108154297e-05,
      "step": 33563,
      "training_step_time": 7.970132827758789
    },
    {
      "epoch": 3.2009124755859375e-05,
      "model_forward_time": 2.424738883972168,
      "step": 33564
    },
    {
      "epoch": 3.2009124755859375e-05,
      "step": 33564,
      "training_step_time": 8.846726655960083
    },
    {
      "epoch": 3.201007843017578e-05,
      "model_forward_time": 1.7095236778259277,
      "step": 33565
    },
    {
      "epoch": 3.201007843017578e-05,
      "step": 33565,
      "training_step_time": 7.802250862121582
    },
    {
      "epoch": 3.201103210449219e-05,
      "model_forward_time": 1.7079496383666992,
      "step": 33566
    },
    {
      "epoch": 3.201103210449219e-05,
      "step": 33566,
      "training_step_time": 7.575185775756836
    },
    {
      "epoch": 3.2011985778808594e-05,
      "model_forward_time": 1.7021677494049072,
      "step": 33567
    },
    {
      "epoch": 3.2011985778808594e-05,
      "step": 33567,
      "training_step_time": 7.432797908782959
    },
    {
      "epoch": 3.2012939453125e-05,
      "model_forward_time": 1.721489667892456,
      "step": 33568
    },
    {
      "epoch": 3.2012939453125e-05,
      "step": 33568,
      "training_step_time": 7.718876123428345
    },
    {
      "epoch": 3.201389312744141e-05,
      "model_forward_time": 1.7052867412567139,
      "step": 33569
    },
    {
      "epoch": 3.201389312744141e-05,
      "step": 33569,
      "training_step_time": 7.816314220428467
    },
    {
      "epoch": 3.201484680175781e-05,
      "grad_norm": 0.15054145455360413,
      "learning_rate": 7.929688590872913e-06,
      "loss": 0.0637,
      "step": 33570
    },
    {
      "epoch": 3.201484680175781e-05,
      "model_forward_time": 1.7075681686401367,
      "step": 33570
    },
    {
      "dynamics_loss_avg": 0.07210259158164263,
      "epoch": 3.201484680175781e-05,
      "step": 33570
    },
    {
      "action_loss_avg": 0.0020271695335395633,
      "epoch": 3.201484680175781e-05,
      "step": 33570
    },
    {
      "epoch": 3.201484680175781e-05,
      "step": 33570,
      "training_step_time": 7.853060483932495
    },
    {
      "epoch": 3.201580047607422e-05,
      "model_forward_time": 1.7294821739196777,
      "step": 33571
    },
    {
      "epoch": 3.201580047607422e-05,
      "step": 33571,
      "training_step_time": 10.457412481307983
    },
    {
      "epoch": 3.201675415039063e-05,
      "model_forward_time": 1.7078826427459717,
      "step": 33572
    },
    {
      "epoch": 3.201675415039063e-05,
      "step": 33572,
      "training_step_time": 7.911034345626831
    },
    {
      "epoch": 3.201770782470703e-05,
      "model_forward_time": 1.704192876815796,
      "step": 33573
    },
    {
      "epoch": 3.201770782470703e-05,
      "step": 33573,
      "training_step_time": 9.473600149154663
    },
    {
      "epoch": 3.2018661499023436e-05,
      "model_forward_time": 1.7076802253723145,
      "step": 33574
    },
    {
      "epoch": 3.2018661499023436e-05,
      "step": 33574,
      "training_step_time": 7.763920545578003
    },
    {
      "epoch": 3.2019615173339846e-05,
      "model_forward_time": 1.7047464847564697,
      "step": 33575
    },
    {
      "epoch": 3.2019615173339846e-05,
      "step": 33575,
      "training_step_time": 7.54021143913269
    },
    {
      "epoch": 3.202056884765625e-05,
      "model_forward_time": 1.7022819519042969,
      "step": 33576
    },
    {
      "epoch": 3.202056884765625e-05,
      "step": 33576,
      "training_step_time": 8.004534006118774
    },
    {
      "epoch": 3.2021522521972655e-05,
      "model_forward_time": 1.722229242324829,
      "step": 33577
    },
    {
      "epoch": 3.2021522521972655e-05,
      "step": 33577,
      "training_step_time": 7.6718409061431885
    },
    {
      "epoch": 3.2022476196289065e-05,
      "model_forward_time": 1.706338882446289,
      "step": 33578
    },
    {
      "epoch": 3.2022476196289065e-05,
      "step": 33578,
      "training_step_time": 7.569192886352539
    },
    {
      "epoch": 3.202342987060547e-05,
      "model_forward_time": 1.704787015914917,
      "step": 33579
    },
    {
      "epoch": 3.202342987060547e-05,
      "step": 33579,
      "training_step_time": 7.871623516082764
    },
    {
      "epoch": 3.202438354492187e-05,
      "grad_norm": 0.12814180552959442,
      "learning_rate": 7.928348531370003e-06,
      "loss": 0.0656,
      "step": 33580
    },
    {
      "epoch": 3.202438354492187e-05,
      "model_forward_time": 1.7383763790130615,
      "step": 33580
    },
    {
      "dynamics_loss_avg": 0.05714637888595462,
      "epoch": 3.202438354492187e-05,
      "step": 33580
    },
    {
      "action_loss_avg": 0.004013330355519429,
      "epoch": 3.202438354492187e-05,
      "step": 33580
    },
    {
      "epoch": 3.202438354492187e-05,
      "step": 33580,
      "training_step_time": 7.903352499008179
    },
    {
      "epoch": 3.2025337219238284e-05,
      "model_forward_time": 1.7088887691497803,
      "step": 33581
    },
    {
      "epoch": 3.2025337219238284e-05,
      "step": 33581,
      "training_step_time": 7.816395998001099
    },
    {
      "epoch": 3.202629089355469e-05,
      "model_forward_time": 1.7112481594085693,
      "step": 33582
    },
    {
      "epoch": 3.202629089355469e-05,
      "step": 33582,
      "training_step_time": 7.958191633224487
    },
    {
      "epoch": 3.202724456787109e-05,
      "model_forward_time": 2.896576404571533,
      "step": 33583
    },
    {
      "epoch": 3.202724456787109e-05,
      "step": 33583,
      "training_step_time": 8.372320413589478
    },
    {
      "epoch": 3.20281982421875e-05,
      "model_forward_time": 1.7092430591583252,
      "step": 33584
    },
    {
      "epoch": 3.20281982421875e-05,
      "step": 33584,
      "training_step_time": 8.201225757598877
    },
    {
      "epoch": 3.202915191650391e-05,
      "model_forward_time": 1.7149698734283447,
      "step": 33585
    },
    {
      "epoch": 3.202915191650391e-05,
      "step": 33585,
      "training_step_time": 7.384904146194458
    },
    {
      "epoch": 3.203010559082031e-05,
      "model_forward_time": 1.7156267166137695,
      "step": 33586
    },
    {
      "epoch": 3.203010559082031e-05,
      "step": 33586,
      "training_step_time": 8.498737812042236
    },
    {
      "epoch": 3.203105926513672e-05,
      "model_forward_time": 1.7082717418670654,
      "step": 33587
    },
    {
      "epoch": 3.203105926513672e-05,
      "step": 33587,
      "training_step_time": 7.654667139053345
    },
    {
      "epoch": 3.2032012939453126e-05,
      "model_forward_time": 1.7148163318634033,
      "step": 33588
    },
    {
      "epoch": 3.2032012939453126e-05,
      "step": 33588,
      "training_step_time": 8.603134632110596
    },
    {
      "epoch": 3.203296661376953e-05,
      "model_forward_time": 1.7360150814056396,
      "step": 33589
    },
    {
      "epoch": 3.203296661376953e-05,
      "step": 33589,
      "training_step_time": 9.548636436462402
    },
    {
      "epoch": 3.203392028808594e-05,
      "grad_norm": 0.14245939254760742,
      "learning_rate": 7.927008151627297e-06,
      "loss": 0.0649,
      "step": 33590
    },
    {
      "epoch": 3.203392028808594e-05,
      "model_forward_time": 1.6958036422729492,
      "step": 33590
    },
    {
      "dynamics_loss_avg": 0.05845926981419325,
      "epoch": 3.203392028808594e-05,
      "step": 33590
    },
    {
      "action_loss_avg": 0.0021262944093905388,
      "epoch": 3.203392028808594e-05,
      "step": 33590
    },
    {
      "epoch": 3.203392028808594e-05,
      "step": 33590,
      "training_step_time": 8.882115125656128
    },
    {
      "epoch": 3.2034873962402345e-05,
      "model_forward_time": 1.716956615447998,
      "step": 33591
    },
    {
      "epoch": 3.2034873962402345e-05,
      "step": 33591,
      "training_step_time": 8.704638004302979
    },
    {
      "epoch": 3.203582763671875e-05,
      "model_forward_time": 2.881136417388916,
      "step": 33592
    },
    {
      "epoch": 3.203582763671875e-05,
      "step": 33592,
      "training_step_time": 7.825850963592529
    },
    {
      "epoch": 3.203678131103516e-05,
      "model_forward_time": 1.7242236137390137,
      "step": 33593
    },
    {
      "epoch": 3.203678131103516e-05,
      "step": 33593,
      "training_step_time": 8.428635597229004
    },
    {
      "epoch": 3.2037734985351564e-05,
      "model_forward_time": 1.3506996631622314,
      "step": 33594
    },
    {
      "epoch": 3.2037734985351564e-05,
      "step": 33594,
      "training_step_time": 7.3810529708862305
    },
    {
      "epoch": 3.203868865966797e-05,
      "model_forward_time": 1.7469160556793213,
      "step": 33595
    },
    {
      "epoch": 3.203868865966797e-05,
      "step": 33595,
      "training_step_time": 7.647480487823486
    },
    {
      "epoch": 3.203964233398437e-05,
      "model_forward_time": 1.7208855152130127,
      "step": 33596
    },
    {
      "epoch": 3.203964233398437e-05,
      "step": 33596,
      "training_step_time": 9.07816743850708
    },
    {
      "epoch": 3.204059600830078e-05,
      "model_forward_time": 1.7154433727264404,
      "step": 33597
    },
    {
      "epoch": 3.204059600830078e-05,
      "step": 33597,
      "training_step_time": 8.357346773147583
    },
    {
      "epoch": 3.2041549682617187e-05,
      "model_forward_time": 1.7188076972961426,
      "step": 33598
    },
    {
      "epoch": 3.2041549682617187e-05,
      "step": 33598,
      "training_step_time": 8.727564573287964
    },
    {
      "epoch": 3.204250335693359e-05,
      "model_forward_time": 1.7126777172088623,
      "step": 33599
    },
    {
      "epoch": 3.204250335693359e-05,
      "step": 33599,
      "training_step_time": 9.727756977081299
    },
    {
      "epoch": 3.204345703125e-05,
      "grad_norm": 0.13675132393836975,
      "learning_rate": 7.925667451791383e-06,
      "loss": 0.0575,
      "step": 33600
    },
    {
      "epoch": 3.204345703125e-05,
      "model_forward_time": 1.7145164012908936,
      "step": 33600
    },
    {
      "dynamics_loss_avg": 0.051912596076726915,
      "epoch": 3.204345703125e-05,
      "step": 33600
    },
    {
      "action_loss_avg": 0.0027928762603551148,
      "epoch": 3.204345703125e-05,
      "step": 33600
    },
    {
      "epoch": 3.204345703125e-05,
      "step": 33600,
      "training_step_time": 9.546819925308228
    },
    {
      "epoch": 3.2044410705566405e-05,
      "model_forward_time": 2.8255491256713867,
      "step": 33601
    },
    {
      "epoch": 3.2044410705566405e-05,
      "step": 33601,
      "training_step_time": 9.434789657592773
    },
    {
      "epoch": 3.204536437988281e-05,
      "model_forward_time": 1.7188286781311035,
      "step": 33602
    },
    {
      "epoch": 3.204536437988281e-05,
      "step": 33602,
      "training_step_time": 7.709876537322998
    },
    {
      "epoch": 3.204631805419922e-05,
      "model_forward_time": 1.7194457054138184,
      "step": 33603
    },
    {
      "epoch": 3.204631805419922e-05,
      "step": 33603,
      "training_step_time": 7.584561109542847
    },
    {
      "epoch": 3.2047271728515624e-05,
      "model_forward_time": 1.7289862632751465,
      "step": 33604
    },
    {
      "epoch": 3.2047271728515624e-05,
      "step": 33604,
      "training_step_time": 7.5538036823272705
    },
    {
      "epoch": 3.204822540283203e-05,
      "model_forward_time": 1.7178423404693604,
      "step": 33605
    },
    {
      "epoch": 3.204822540283203e-05,
      "step": 33605,
      "training_step_time": 7.71044397354126
    },
    {
      "epoch": 3.204917907714844e-05,
      "model_forward_time": 1.7145321369171143,
      "step": 33606
    },
    {
      "epoch": 3.204917907714844e-05,
      "step": 33606,
      "training_step_time": 8.834392309188843
    },
    {
      "epoch": 3.205013275146484e-05,
      "model_forward_time": 1.7255382537841797,
      "step": 33607
    },
    {
      "epoch": 3.205013275146484e-05,
      "step": 33607,
      "training_step_time": 8.799235582351685
    },
    {
      "epoch": 3.205108642578125e-05,
      "model_forward_time": 1.7294535636901855,
      "step": 33608
    },
    {
      "epoch": 3.205108642578125e-05,
      "step": 33608,
      "training_step_time": 8.340346574783325
    },
    {
      "epoch": 3.205204010009766e-05,
      "model_forward_time": 1.718491554260254,
      "step": 33609
    },
    {
      "epoch": 3.205204010009766e-05,
      "step": 33609,
      "training_step_time": 8.028097867965698
    },
    {
      "epoch": 3.205299377441406e-05,
      "grad_norm": 0.16985481977462769,
      "learning_rate": 7.924326432008874e-06,
      "loss": 0.0632,
      "step": 33610
    },
    {
      "epoch": 3.205299377441406e-05,
      "model_forward_time": 2.792937755584717,
      "step": 33610
    },
    {
      "dynamics_loss_avg": 0.06060410011559725,
      "epoch": 3.205299377441406e-05,
      "step": 33610
    },
    {
      "action_loss_avg": 0.0015553800680208952,
      "epoch": 3.205299377441406e-05,
      "step": 33610
    },
    {
      "epoch": 3.205299377441406e-05,
      "step": 33610,
      "training_step_time": 7.844470024108887
    },
    {
      "epoch": 3.2053947448730466e-05,
      "model_forward_time": 1.7220039367675781,
      "step": 33611
    },
    {
      "epoch": 3.2053947448730466e-05,
      "step": 33611,
      "training_step_time": 7.7799975872039795
    },
    {
      "epoch": 3.205490112304688e-05,
      "model_forward_time": 1.723623275756836,
      "step": 33612
    },
    {
      "epoch": 3.205490112304688e-05,
      "step": 33612,
      "training_step_time": 7.693984031677246
    },
    {
      "epoch": 3.205585479736328e-05,
      "model_forward_time": 1.7178664207458496,
      "step": 33613
    },
    {
      "epoch": 3.205585479736328e-05,
      "step": 33613,
      "training_step_time": 7.590758562088013
    },
    {
      "epoch": 3.2056808471679685e-05,
      "model_forward_time": 1.7273650169372559,
      "step": 33614
    },
    {
      "epoch": 3.2056808471679685e-05,
      "step": 33614,
      "training_step_time": 8.118187427520752
    },
    {
      "epoch": 3.2057762145996096e-05,
      "model_forward_time": 1.715217113494873,
      "step": 33615
    },
    {
      "epoch": 3.2057762145996096e-05,
      "step": 33615,
      "training_step_time": 7.742274522781372
    },
    {
      "epoch": 3.20587158203125e-05,
      "model_forward_time": 1.727959394454956,
      "step": 33616
    },
    {
      "epoch": 3.20587158203125e-05,
      "step": 33616,
      "training_step_time": 8.875292778015137
    },
    {
      "epoch": 3.2059669494628904e-05,
      "model_forward_time": 1.7851197719573975,
      "step": 33617
    },
    {
      "epoch": 3.2059669494628904e-05,
      "step": 33617,
      "training_step_time": 8.882627964019775
    },
    {
      "epoch": 3.2060623168945315e-05,
      "model_forward_time": 1.8073561191558838,
      "step": 33618
    },
    {
      "epoch": 3.2060623168945315e-05,
      "step": 33618,
      "training_step_time": 8.988533735275269
    },
    {
      "epoch": 3.206157684326172e-05,
      "model_forward_time": 1.7411632537841797,
      "step": 33619
    },
    {
      "epoch": 3.206157684326172e-05,
      "step": 33619,
      "training_step_time": 10.871754169464111
    },
    {
      "epoch": 3.206253051757812e-05,
      "grad_norm": 0.15316063165664673,
      "learning_rate": 7.922985092426423e-06,
      "loss": 0.0621,
      "step": 33620
    },
    {
      "epoch": 3.206253051757812e-05,
      "model_forward_time": 2.9511663913726807,
      "step": 33620
    },
    {
      "dynamics_loss_avg": 0.04994434919208288,
      "epoch": 3.206253051757812e-05,
      "step": 33620
    },
    {
      "action_loss_avg": 0.0017435135639971122,
      "epoch": 3.206253051757812e-05,
      "step": 33620
    },
    {
      "epoch": 3.206253051757812e-05,
      "step": 33620,
      "training_step_time": 7.888664484024048
    },
    {
      "epoch": 3.206348419189453e-05,
      "model_forward_time": 1.743518352508545,
      "step": 33621
    },
    {
      "epoch": 3.206348419189453e-05,
      "step": 33621,
      "training_step_time": 8.208427429199219
    },
    {
      "epoch": 3.206443786621094e-05,
      "model_forward_time": 1.7401151657104492,
      "step": 33622
    },
    {
      "epoch": 3.206443786621094e-05,
      "step": 33622,
      "training_step_time": 8.25548768043518
    },
    {
      "epoch": 3.206539154052734e-05,
      "model_forward_time": 1.743257761001587,
      "step": 33623
    },
    {
      "epoch": 3.206539154052734e-05,
      "step": 33623,
      "training_step_time": 8.368545293807983
    },
    {
      "epoch": 3.206634521484375e-05,
      "model_forward_time": 1.7085671424865723,
      "step": 33624
    },
    {
      "epoch": 3.206634521484375e-05,
      "step": 33624,
      "training_step_time": 7.69587779045105
    },
    {
      "epoch": 3.2067298889160156e-05,
      "model_forward_time": 1.7454547882080078,
      "step": 33625
    },
    {
      "epoch": 3.2067298889160156e-05,
      "step": 33625,
      "training_step_time": 8.527109384536743
    },
    {
      "epoch": 3.206825256347656e-05,
      "model_forward_time": 1.722491979598999,
      "step": 33626
    },
    {
      "epoch": 3.206825256347656e-05,
      "step": 33626,
      "training_step_time": 9.067102193832397
    },
    {
      "epoch": 3.206920623779297e-05,
      "model_forward_time": 1.7986366748809814,
      "step": 33627
    },
    {
      "epoch": 3.206920623779297e-05,
      "step": 33627,
      "training_step_time": 8.876835584640503
    },
    {
      "epoch": 3.2070159912109375e-05,
      "model_forward_time": 1.7401137351989746,
      "step": 33628
    },
    {
      "epoch": 3.2070159912109375e-05,
      "step": 33628,
      "training_step_time": 8.381998062133789
    },
    {
      "epoch": 3.207111358642578e-05,
      "model_forward_time": 2.943819522857666,
      "step": 33629
    },
    {
      "epoch": 3.207111358642578e-05,
      "step": 33629,
      "training_step_time": 8.698702096939087
    },
    {
      "epoch": 3.207206726074219e-05,
      "grad_norm": 0.18512900173664093,
      "learning_rate": 7.921643433190718e-06,
      "loss": 0.0651,
      "step": 33630
    },
    {
      "epoch": 3.207206726074219e-05,
      "model_forward_time": 1.7103245258331299,
      "step": 33630
    },
    {
      "dynamics_loss_avg": 0.05892299562692642,
      "epoch": 3.207206726074219e-05,
      "step": 33630
    },
    {
      "action_loss_avg": 0.004377035911602434,
      "epoch": 3.207206726074219e-05,
      "step": 33630
    },
    {
      "epoch": 3.207206726074219e-05,
      "step": 33630,
      "training_step_time": 7.655923128128052
    },
    {
      "epoch": 3.2073020935058594e-05,
      "model_forward_time": 1.7239151000976562,
      "step": 33631
    },
    {
      "epoch": 3.2073020935058594e-05,
      "step": 33631,
      "training_step_time": 8.120285272598267
    },
    {
      "epoch": 3.2073974609375e-05,
      "model_forward_time": 1.7491569519042969,
      "step": 33632
    },
    {
      "epoch": 3.2073974609375e-05,
      "step": 33632,
      "training_step_time": 8.409266233444214
    },
    {
      "epoch": 3.207492828369141e-05,
      "model_forward_time": 1.7483980655670166,
      "step": 33633
    },
    {
      "epoch": 3.207492828369141e-05,
      "step": 33633,
      "training_step_time": 7.618519067764282
    },
    {
      "epoch": 3.207588195800781e-05,
      "model_forward_time": 1.7419893741607666,
      "step": 33634
    },
    {
      "epoch": 3.207588195800781e-05,
      "step": 33634,
      "training_step_time": 7.79569673538208
    },
    {
      "epoch": 3.207683563232422e-05,
      "model_forward_time": 1.735835075378418,
      "step": 33635
    },
    {
      "epoch": 3.207683563232422e-05,
      "step": 33635,
      "training_step_time": 8.301161289215088
    },
    {
      "epoch": 3.207778930664063e-05,
      "model_forward_time": 1.8054330348968506,
      "step": 33636
    },
    {
      "epoch": 3.207778930664063e-05,
      "step": 33636,
      "training_step_time": 7.894701957702637
    },
    {
      "epoch": 3.207874298095703e-05,
      "model_forward_time": 1.7304015159606934,
      "step": 33637
    },
    {
      "epoch": 3.207874298095703e-05,
      "step": 33637,
      "training_step_time": 8.007246494293213
    },
    {
      "epoch": 3.2079696655273436e-05,
      "model_forward_time": 2.8783085346221924,
      "step": 33638
    },
    {
      "epoch": 3.2079696655273436e-05,
      "step": 33638,
      "training_step_time": 7.859633922576904
    },
    {
      "epoch": 3.2080650329589847e-05,
      "model_forward_time": 1.7247583866119385,
      "step": 33639
    },
    {
      "epoch": 3.2080650329589847e-05,
      "step": 33639,
      "training_step_time": 7.673841953277588
    },
    {
      "epoch": 3.208160400390625e-05,
      "grad_norm": 0.150855153799057,
      "learning_rate": 7.920301454448477e-06,
      "loss": 0.0635,
      "step": 33640
    },
    {
      "epoch": 3.208160400390625e-05,
      "model_forward_time": 1.725156307220459,
      "step": 33640
    },
    {
      "dynamics_loss_avg": 0.057137949392199516,
      "epoch": 3.208160400390625e-05,
      "step": 33640
    },
    {
      "action_loss_avg": 0.002583865856286138,
      "epoch": 3.208160400390625e-05,
      "step": 33640
    },
    {
      "epoch": 3.208160400390625e-05,
      "step": 33640,
      "training_step_time": 7.619432210922241
    },
    {
      "epoch": 3.2082557678222655e-05,
      "model_forward_time": 1.7386746406555176,
      "step": 33641
    },
    {
      "epoch": 3.2082557678222655e-05,
      "step": 33641,
      "training_step_time": 7.670753479003906
    },
    {
      "epoch": 3.2083511352539065e-05,
      "model_forward_time": 1.7413568496704102,
      "step": 33642
    },
    {
      "epoch": 3.2083511352539065e-05,
      "step": 33642,
      "training_step_time": 7.625843524932861
    },
    {
      "epoch": 3.208446502685547e-05,
      "model_forward_time": 1.7467925548553467,
      "step": 33643
    },
    {
      "epoch": 3.208446502685547e-05,
      "step": 33643,
      "training_step_time": 7.848592758178711
    },
    {
      "epoch": 3.2085418701171873e-05,
      "model_forward_time": 1.7295019626617432,
      "step": 33644
    },
    {
      "epoch": 3.2085418701171873e-05,
      "step": 33644,
      "training_step_time": 8.333113670349121
    },
    {
      "epoch": 3.2086372375488284e-05,
      "model_forward_time": 1.7920174598693848,
      "step": 33645
    },
    {
      "epoch": 3.2086372375488284e-05,
      "step": 33645,
      "training_step_time": 8.756299495697021
    },
    {
      "epoch": 3.208732604980469e-05,
      "model_forward_time": 1.7382490634918213,
      "step": 33646
    },
    {
      "epoch": 3.208732604980469e-05,
      "step": 33646,
      "training_step_time": 7.908592224121094
    },
    {
      "epoch": 3.208827972412109e-05,
      "model_forward_time": 2.856560230255127,
      "step": 33647
    },
    {
      "epoch": 3.208827972412109e-05,
      "step": 33647,
      "training_step_time": 7.830786228179932
    },
    {
      "epoch": 3.20892333984375e-05,
      "model_forward_time": 1.7406821250915527,
      "step": 33648
    },
    {
      "epoch": 3.20892333984375e-05,
      "step": 33648,
      "training_step_time": 7.716795444488525
    },
    {
      "epoch": 3.209018707275391e-05,
      "model_forward_time": 1.7362751960754395,
      "step": 33649
    },
    {
      "epoch": 3.209018707275391e-05,
      "step": 33649,
      "training_step_time": 7.494787216186523
    },
    {
      "epoch": 3.209114074707031e-05,
      "grad_norm": 0.16843503713607788,
      "learning_rate": 7.918959156346462e-06,
      "loss": 0.0623,
      "step": 33650
    },
    {
      "epoch": 3.209114074707031e-05,
      "model_forward_time": 1.7157626152038574,
      "step": 33650
    },
    {
      "dynamics_loss_avg": 0.07279495410621166,
      "epoch": 3.209114074707031e-05,
      "step": 33650
    },
    {
      "action_loss_avg": 0.002807756781112403,
      "epoch": 3.209114074707031e-05,
      "step": 33650
    },
    {
      "epoch": 3.209114074707031e-05,
      "step": 33650,
      "training_step_time": 7.713249206542969
    },
    {
      "epoch": 3.209209442138672e-05,
      "model_forward_time": 1.7435317039489746,
      "step": 33651
    },
    {
      "epoch": 3.209209442138672e-05,
      "step": 33651,
      "training_step_time": 8.446570873260498
    },
    {
      "epoch": 3.2093048095703126e-05,
      "model_forward_time": 1.740100383758545,
      "step": 33652
    },
    {
      "epoch": 3.2093048095703126e-05,
      "step": 33652,
      "training_step_time": 7.837600946426392
    },
    {
      "epoch": 3.209400177001953e-05,
      "model_forward_time": 1.7388741970062256,
      "step": 33653
    },
    {
      "epoch": 3.209400177001953e-05,
      "step": 33653,
      "training_step_time": 7.8412628173828125
    },
    {
      "epoch": 3.209495544433594e-05,
      "model_forward_time": 1.7759201526641846,
      "step": 33654
    },
    {
      "epoch": 3.209495544433594e-05,
      "step": 33654,
      "training_step_time": 8.831905126571655
    },
    {
      "epoch": 3.2095909118652345e-05,
      "model_forward_time": 1.7342157363891602,
      "step": 33655
    },
    {
      "epoch": 3.2095909118652345e-05,
      "step": 33655,
      "training_step_time": 9.896361827850342
    },
    {
      "epoch": 3.209686279296875e-05,
      "model_forward_time": 2.8287224769592285,
      "step": 33656
    },
    {
      "epoch": 3.209686279296875e-05,
      "step": 33656,
      "training_step_time": 8.645822763442993
    },
    {
      "epoch": 3.209781646728515e-05,
      "model_forward_time": 1.7398409843444824,
      "step": 33657
    },
    {
      "epoch": 3.209781646728515e-05,
      "step": 33657,
      "training_step_time": 7.77729058265686
    },
    {
      "epoch": 3.2098770141601564e-05,
      "model_forward_time": 1.7521238327026367,
      "step": 33658
    },
    {
      "epoch": 3.2098770141601564e-05,
      "step": 33658,
      "training_step_time": 8.486798763275146
    },
    {
      "epoch": 3.209972381591797e-05,
      "model_forward_time": 1.7397730350494385,
      "step": 33659
    },
    {
      "epoch": 3.209972381591797e-05,
      "step": 33659,
      "training_step_time": 7.630087614059448
    },
    {
      "epoch": 3.210067749023437e-05,
      "grad_norm": 0.12338943034410477,
      "learning_rate": 7.917616539031462e-06,
      "loss": 0.0615,
      "step": 33660
    },
    {
      "epoch": 3.210067749023437e-05,
      "model_forward_time": 1.7581045627593994,
      "step": 33660
    },
    {
      "dynamics_loss_avg": 0.06130768693983555,
      "epoch": 3.210067749023437e-05,
      "step": 33660
    },
    {
      "action_loss_avg": 0.0037167688948102295,
      "epoch": 3.210067749023437e-05,
      "step": 33660
    },
    {
      "epoch": 3.210067749023437e-05,
      "step": 33660,
      "training_step_time": 8.60923957824707
    },
    {
      "epoch": 3.210163116455078e-05,
      "model_forward_time": 1.736260175704956,
      "step": 33661
    },
    {
      "epoch": 3.210163116455078e-05,
      "step": 33661,
      "training_step_time": 8.6904137134552
    },
    {
      "epoch": 3.2102584838867187e-05,
      "model_forward_time": 1.7223501205444336,
      "step": 33662
    },
    {
      "epoch": 3.2102584838867187e-05,
      "step": 33662,
      "training_step_time": 7.768946170806885
    },
    {
      "epoch": 3.210353851318359e-05,
      "model_forward_time": 1.7446398735046387,
      "step": 33663
    },
    {
      "epoch": 3.210353851318359e-05,
      "step": 33663,
      "training_step_time": 8.207321643829346
    },
    {
      "epoch": 3.21044921875e-05,
      "model_forward_time": 1.8025527000427246,
      "step": 33664
    },
    {
      "epoch": 3.21044921875e-05,
      "step": 33664,
      "training_step_time": 8.742194652557373
    },
    {
      "epoch": 3.2105445861816405e-05,
      "model_forward_time": 1.7565507888793945,
      "step": 33665
    },
    {
      "epoch": 3.2105445861816405e-05,
      "step": 33665,
      "training_step_time": 8.401029109954834
    },
    {
      "epoch": 3.210639953613281e-05,
      "model_forward_time": 1.7397723197937012,
      "step": 33666
    },
    {
      "epoch": 3.210639953613281e-05,
      "step": 33666,
      "training_step_time": 7.805893659591675
    },
    {
      "epoch": 3.210735321044922e-05,
      "model_forward_time": 1.7333238124847412,
      "step": 33667
    },
    {
      "epoch": 3.210735321044922e-05,
      "step": 33667,
      "training_step_time": 7.6466145515441895
    },
    {
      "epoch": 3.2108306884765624e-05,
      "model_forward_time": 1.7377564907073975,
      "step": 33668
    },
    {
      "epoch": 3.2108306884765624e-05,
      "step": 33668,
      "training_step_time": 7.553896427154541
    },
    {
      "epoch": 3.210926055908203e-05,
      "model_forward_time": 1.7431731224060059,
      "step": 33669
    },
    {
      "epoch": 3.210926055908203e-05,
      "step": 33669,
      "training_step_time": 8.394787788391113
    },
    {
      "epoch": 3.211021423339844e-05,
      "grad_norm": 0.14453834295272827,
      "learning_rate": 7.916273602650303e-06,
      "loss": 0.0626,
      "step": 33670
    },
    {
      "epoch": 3.211021423339844e-05,
      "model_forward_time": 1.7249274253845215,
      "step": 33670
    },
    {
      "dynamics_loss_avg": 0.05784263238310814,
      "epoch": 3.211021423339844e-05,
      "step": 33670
    },
    {
      "action_loss_avg": 0.001525832654442638,
      "epoch": 3.211021423339844e-05,
      "step": 33670
    },
    {
      "epoch": 3.211021423339844e-05,
      "step": 33670,
      "training_step_time": 7.809714317321777
    },
    {
      "epoch": 3.211116790771484e-05,
      "model_forward_time": 1.745335578918457,
      "step": 33671
    },
    {
      "epoch": 3.211116790771484e-05,
      "step": 33671,
      "training_step_time": 7.706328630447388
    },
    {
      "epoch": 3.211212158203125e-05,
      "model_forward_time": 1.7288494110107422,
      "step": 33672
    },
    {
      "epoch": 3.211212158203125e-05,
      "step": 33672,
      "training_step_time": 8.835381031036377
    },
    {
      "epoch": 3.211307525634766e-05,
      "model_forward_time": 1.8077399730682373,
      "step": 33673
    },
    {
      "epoch": 3.211307525634766e-05,
      "step": 33673,
      "training_step_time": 7.855625629425049
    },
    {
      "epoch": 3.211402893066406e-05,
      "model_forward_time": 1.730752944946289,
      "step": 33674
    },
    {
      "epoch": 3.211402893066406e-05,
      "step": 33674,
      "training_step_time": 7.902442932128906
    },
    {
      "epoch": 3.2114982604980466e-05,
      "model_forward_time": 2.898728847503662,
      "step": 33675
    },
    {
      "epoch": 3.2114982604980466e-05,
      "step": 33675,
      "training_step_time": 7.8668084144592285
    },
    {
      "epoch": 3.211593627929688e-05,
      "model_forward_time": 1.7481343746185303,
      "step": 33676
    },
    {
      "epoch": 3.211593627929688e-05,
      "step": 33676,
      "training_step_time": 7.662458419799805
    },
    {
      "epoch": 3.211688995361328e-05,
      "model_forward_time": 1.7410869598388672,
      "step": 33677
    },
    {
      "epoch": 3.211688995361328e-05,
      "step": 33677,
      "training_step_time": 7.3960466384887695
    },
    {
      "epoch": 3.2117843627929685e-05,
      "model_forward_time": 1.7383832931518555,
      "step": 33678
    },
    {
      "epoch": 3.2117843627929685e-05,
      "step": 33678,
      "training_step_time": 7.631491661071777
    },
    {
      "epoch": 3.2118797302246096e-05,
      "model_forward_time": 1.7548272609710693,
      "step": 33679
    },
    {
      "epoch": 3.2118797302246096e-05,
      "step": 33679,
      "training_step_time": 8.443825006484985
    },
    {
      "epoch": 3.21197509765625e-05,
      "grad_norm": 0.10513080656528473,
      "learning_rate": 7.914930347349848e-06,
      "loss": 0.06,
      "step": 33680
    },
    {
      "epoch": 3.21197509765625e-05,
      "model_forward_time": 1.7205464839935303,
      "step": 33680
    },
    {
      "dynamics_loss_avg": 0.0430188162252307,
      "epoch": 3.21197509765625e-05,
      "step": 33680
    },
    {
      "action_loss_avg": 0.0011137035093270243,
      "epoch": 3.21197509765625e-05,
      "step": 33680
    },
    {
      "epoch": 3.21197509765625e-05,
      "step": 33680,
      "training_step_time": 8.390892267227173
    },
    {
      "epoch": 3.2120704650878904e-05,
      "model_forward_time": 1.7309513092041016,
      "step": 33681
    },
    {
      "epoch": 3.2120704650878904e-05,
      "step": 33681,
      "training_step_time": 9.095438718795776
    },
    {
      "epoch": 3.2121658325195315e-05,
      "model_forward_time": 1.8117620944976807,
      "step": 33682
    },
    {
      "epoch": 3.2121658325195315e-05,
      "step": 33682,
      "training_step_time": 9.754295825958252
    },
    {
      "epoch": 3.212261199951172e-05,
      "model_forward_time": 1.7410285472869873,
      "step": 33683
    },
    {
      "epoch": 3.212261199951172e-05,
      "step": 33683,
      "training_step_time": 8.536539316177368
    },
    {
      "epoch": 3.212356567382812e-05,
      "model_forward_time": 2.868957757949829,
      "step": 33684
    },
    {
      "epoch": 3.212356567382812e-05,
      "step": 33684,
      "training_step_time": 8.391383171081543
    },
    {
      "epoch": 3.2124519348144533e-05,
      "model_forward_time": 1.731351375579834,
      "step": 33685
    },
    {
      "epoch": 3.2124519348144533e-05,
      "step": 33685,
      "training_step_time": 8.801641702651978
    },
    {
      "epoch": 3.212547302246094e-05,
      "model_forward_time": 1.7668285369873047,
      "step": 33686
    },
    {
      "epoch": 3.212547302246094e-05,
      "step": 33686,
      "training_step_time": 6.99543309211731
    },
    {
      "epoch": 3.212642669677734e-05,
      "model_forward_time": 1.740271806716919,
      "step": 33687
    },
    {
      "epoch": 3.212642669677734e-05,
      "step": 33687,
      "training_step_time": 7.058028221130371
    },
    {
      "epoch": 3.212738037109375e-05,
      "model_forward_time": 1.7503275871276855,
      "step": 33688
    },
    {
      "epoch": 3.212738037109375e-05,
      "step": 33688,
      "training_step_time": 7.013444185256958
    },
    {
      "epoch": 3.2128334045410156e-05,
      "model_forward_time": 1.706585168838501,
      "step": 33689
    },
    {
      "epoch": 3.2128334045410156e-05,
      "step": 33689,
      "training_step_time": 7.009442329406738
    },
    {
      "epoch": 3.212928771972656e-05,
      "grad_norm": 0.09771590679883957,
      "learning_rate": 7.913586773276993e-06,
      "loss": 0.0596,
      "step": 33690
    },
    {
      "epoch": 3.212928771972656e-05,
      "model_forward_time": 1.7192094326019287,
      "step": 33690
    },
    {
      "dynamics_loss_avg": 0.049889728613197806,
      "epoch": 3.212928771972656e-05,
      "step": 33690
    },
    {
      "action_loss_avg": 0.0017619282327359543,
      "epoch": 3.212928771972656e-05,
      "step": 33690
    },
    {
      "epoch": 3.212928771972656e-05,
      "step": 33690,
      "training_step_time": 7.025322437286377
    },
    {
      "epoch": 3.213024139404297e-05,
      "model_forward_time": 1.7622785568237305,
      "step": 33691
    },
    {
      "epoch": 3.213024139404297e-05,
      "step": 33691,
      "training_step_time": 7.076356887817383
    },
    {
      "epoch": 3.2131195068359375e-05,
      "model_forward_time": 1.7420940399169922,
      "step": 33692
    },
    {
      "epoch": 3.2131195068359375e-05,
      "step": 33692,
      "training_step_time": 7.012733697891235
    },
    {
      "epoch": 3.213214874267578e-05,
      "model_forward_time": 1.7493619918823242,
      "step": 33693
    },
    {
      "epoch": 3.213214874267578e-05,
      "step": 33693,
      "training_step_time": 8.423391342163086
    },
    {
      "epoch": 3.213310241699219e-05,
      "model_forward_time": 1.8116352558135986,
      "step": 33694
    },
    {
      "epoch": 3.213310241699219e-05,
      "step": 33694,
      "training_step_time": 10.54486894607544
    },
    {
      "epoch": 3.2134056091308594e-05,
      "model_forward_time": 1.728424072265625,
      "step": 33695
    },
    {
      "epoch": 3.2134056091308594e-05,
      "step": 33695,
      "training_step_time": 7.758792877197266
    },
    {
      "epoch": 3.2135009765625e-05,
      "model_forward_time": 1.7515177726745605,
      "step": 33696
    },
    {
      "epoch": 3.2135009765625e-05,
      "step": 33696,
      "training_step_time": 7.113551139831543
    },
    {
      "epoch": 3.213596343994141e-05,
      "model_forward_time": 1.7252211570739746,
      "step": 33697
    },
    {
      "epoch": 3.213596343994141e-05,
      "step": 33697,
      "training_step_time": 7.064645051956177
    },
    {
      "epoch": 3.213691711425781e-05,
      "model_forward_time": 1.736232042312622,
      "step": 33698
    },
    {
      "epoch": 3.213691711425781e-05,
      "step": 33698,
      "training_step_time": 7.021951913833618
    },
    {
      "epoch": 3.213787078857422e-05,
      "model_forward_time": 1.6976897716522217,
      "step": 33699
    },
    {
      "epoch": 3.213787078857422e-05,
      "step": 33699,
      "training_step_time": 6.987424612045288
    },
    {
      "epoch": 3.213882446289063e-05,
      "grad_norm": 0.24084918200969696,
      "learning_rate": 7.912242880578667e-06,
      "loss": 0.0556,
      "step": 33700
    },
    {
      "epoch": 3.213882446289063e-05,
      "model_forward_time": 1.7290067672729492,
      "step": 33700
    },
    {
      "dynamics_loss_avg": 0.05887434408068657,
      "epoch": 3.213882446289063e-05,
      "step": 33700
    },
    {
      "action_loss_avg": 0.002436176629271358,
      "epoch": 3.213882446289063e-05,
      "step": 33700
    },
    {
      "epoch": 3.213882446289063e-05,
      "step": 33700,
      "training_step_time": 6.93388295173645
    },
    {
      "epoch": 3.213977813720703e-05,
      "model_forward_time": 1.743347406387329,
      "step": 33701
    },
    {
      "epoch": 3.213977813720703e-05,
      "step": 33701,
      "training_step_time": 7.063603401184082
    },
    {
      "epoch": 3.2140731811523436e-05,
      "model_forward_time": 1.7486329078674316,
      "step": 33702
    },
    {
      "epoch": 3.2140731811523436e-05,
      "step": 33702,
      "training_step_time": 7.027151823043823
    },
    {
      "epoch": 3.2141685485839847e-05,
      "model_forward_time": 2.301506996154785,
      "step": 33703
    },
    {
      "epoch": 3.2141685485839847e-05,
      "step": 33703,
      "training_step_time": 8.734381675720215
    },
    {
      "epoch": 3.214263916015625e-05,
      "model_forward_time": 1.7432541847229004,
      "step": 33704
    },
    {
      "epoch": 3.214263916015625e-05,
      "step": 33704,
      "training_step_time": 7.875176668167114
    },
    {
      "epoch": 3.2143592834472655e-05,
      "model_forward_time": 1.7428138256072998,
      "step": 33705
    },
    {
      "epoch": 3.2143592834472655e-05,
      "step": 33705,
      "training_step_time": 7.056442737579346
    },
    {
      "epoch": 3.2144546508789065e-05,
      "model_forward_time": 1.7446646690368652,
      "step": 33706
    },
    {
      "epoch": 3.2144546508789065e-05,
      "step": 33706,
      "training_step_time": 7.048806428909302
    },
    {
      "epoch": 3.214550018310547e-05,
      "model_forward_time": 1.7513022422790527,
      "step": 33707
    },
    {
      "epoch": 3.214550018310547e-05,
      "step": 33707,
      "training_step_time": 7.082735300064087
    },
    {
      "epoch": 3.2146453857421873e-05,
      "model_forward_time": 1.7396440505981445,
      "step": 33708
    },
    {
      "epoch": 3.2146453857421873e-05,
      "step": 33708,
      "training_step_time": 7.029896259307861
    },
    {
      "epoch": 3.2147407531738284e-05,
      "model_forward_time": 1.7519090175628662,
      "step": 33709
    },
    {
      "epoch": 3.2147407531738284e-05,
      "step": 33709,
      "training_step_time": 7.078622817993164
    },
    {
      "epoch": 3.214836120605469e-05,
      "grad_norm": 0.11740460991859436,
      "learning_rate": 7.91089866940184e-06,
      "loss": 0.062,
      "step": 33710
    },
    {
      "epoch": 3.214836120605469e-05,
      "model_forward_time": 1.740717887878418,
      "step": 33710
    },
    {
      "dynamics_loss_avg": 0.054694805294275284,
      "epoch": 3.214836120605469e-05,
      "step": 33710
    },
    {
      "action_loss_avg": 0.0018182642059400677,
      "epoch": 3.214836120605469e-05,
      "step": 33710
    },
    {
      "epoch": 3.214836120605469e-05,
      "step": 33710,
      "training_step_time": 7.00305700302124
    },
    {
      "epoch": 3.214931488037109e-05,
      "model_forward_time": 1.7443628311157227,
      "step": 33711
    },
    {
      "epoch": 3.214931488037109e-05,
      "step": 33711,
      "training_step_time": 7.040506362915039
    },
    {
      "epoch": 3.21502685546875e-05,
      "model_forward_time": 2.2871148586273193,
      "step": 33712
    },
    {
      "epoch": 3.21502685546875e-05,
      "step": 33712,
      "training_step_time": 7.6430275440216064
    },
    {
      "epoch": 3.215122222900391e-05,
      "model_forward_time": 1.7194087505340576,
      "step": 33713
    },
    {
      "epoch": 3.215122222900391e-05,
      "step": 33713,
      "training_step_time": 7.865344524383545
    },
    {
      "epoch": 3.215217590332031e-05,
      "model_forward_time": 1.7392168045043945,
      "step": 33714
    },
    {
      "epoch": 3.215217590332031e-05,
      "step": 33714,
      "training_step_time": 8.059622764587402
    },
    {
      "epoch": 3.215312957763672e-05,
      "model_forward_time": 1.7183620929718018,
      "step": 33715
    },
    {
      "epoch": 3.215312957763672e-05,
      "step": 33715,
      "training_step_time": 7.025606870651245
    },
    {
      "epoch": 3.2154083251953126e-05,
      "model_forward_time": 1.71657395362854,
      "step": 33716
    },
    {
      "epoch": 3.2154083251953126e-05,
      "step": 33716,
      "training_step_time": 7.033820629119873
    },
    {
      "epoch": 3.215503692626953e-05,
      "model_forward_time": 1.7296159267425537,
      "step": 33717
    },
    {
      "epoch": 3.215503692626953e-05,
      "step": 33717,
      "training_step_time": 7.075331687927246
    },
    {
      "epoch": 3.215599060058594e-05,
      "model_forward_time": 1.751375675201416,
      "step": 33718
    },
    {
      "epoch": 3.215599060058594e-05,
      "step": 33718,
      "training_step_time": 7.0319201946258545
    },
    {
      "epoch": 3.2156944274902345e-05,
      "model_forward_time": 1.751617670059204,
      "step": 33719
    },
    {
      "epoch": 3.2156944274902345e-05,
      "step": 33719,
      "training_step_time": 7.061517238616943
    },
    {
      "epoch": 3.215789794921875e-05,
      "grad_norm": 0.12681059539318085,
      "learning_rate": 7.909554139893512e-06,
      "loss": 0.0607,
      "step": 33720
    },
    {
      "epoch": 3.215789794921875e-05,
      "model_forward_time": 1.741245985031128,
      "step": 33720
    },
    {
      "dynamics_loss_avg": 0.03939888505265117,
      "epoch": 3.215789794921875e-05,
      "step": 33720
    },
    {
      "action_loss_avg": 0.0013709710503462702,
      "epoch": 3.215789794921875e-05,
      "step": 33720
    },
    {
      "epoch": 3.215789794921875e-05,
      "step": 33720,
      "training_step_time": 7.005846261978149
    },
    {
      "epoch": 3.215885162353515e-05,
      "model_forward_time": 2.3313326835632324,
      "step": 33721
    },
    {
      "epoch": 3.215885162353515e-05,
      "step": 33721,
      "training_step_time": 7.585472583770752
    },
    {
      "epoch": 3.2159805297851564e-05,
      "model_forward_time": 1.7503201961517334,
      "step": 33722
    },
    {
      "epoch": 3.2159805297851564e-05,
      "step": 33722,
      "training_step_time": 10.175955057144165
    },
    {
      "epoch": 3.216075897216797e-05,
      "model_forward_time": 1.7389824390411377,
      "step": 33723
    },
    {
      "epoch": 3.216075897216797e-05,
      "step": 33723,
      "training_step_time": 7.892246246337891
    },
    {
      "epoch": 3.216171264648437e-05,
      "model_forward_time": 1.7401034832000732,
      "step": 33724
    },
    {
      "epoch": 3.216171264648437e-05,
      "step": 33724,
      "training_step_time": 7.064127206802368
    },
    {
      "epoch": 3.216266632080078e-05,
      "model_forward_time": 1.7455830574035645,
      "step": 33725
    },
    {
      "epoch": 3.216266632080078e-05,
      "step": 33725,
      "training_step_time": 7.054363012313843
    },
    {
      "epoch": 3.216361999511719e-05,
      "model_forward_time": 1.7539360523223877,
      "step": 33726
    },
    {
      "epoch": 3.216361999511719e-05,
      "step": 33726,
      "training_step_time": 7.078321695327759
    },
    {
      "epoch": 3.216457366943359e-05,
      "model_forward_time": 1.740218162536621,
      "step": 33727
    },
    {
      "epoch": 3.216457366943359e-05,
      "step": 33727,
      "training_step_time": 7.073085069656372
    },
    {
      "epoch": 3.216552734375e-05,
      "model_forward_time": 1.7430055141448975,
      "step": 33728
    },
    {
      "epoch": 3.216552734375e-05,
      "step": 33728,
      "training_step_time": 7.047326326370239
    },
    {
      "epoch": 3.2166481018066405e-05,
      "model_forward_time": 1.7504169940948486,
      "step": 33729
    },
    {
      "epoch": 3.2166481018066405e-05,
      "step": 33729,
      "training_step_time": 7.044626235961914
    },
    {
      "epoch": 3.216743469238281e-05,
      "grad_norm": 0.1822192370891571,
      "learning_rate": 7.908209292200715e-06,
      "loss": 0.0612,
      "step": 33730
    },
    {
      "epoch": 3.216743469238281e-05,
      "model_forward_time": 2.313488721847534,
      "step": 33730
    },
    {
      "dynamics_loss_avg": 0.06656883433461189,
      "epoch": 3.216743469238281e-05,
      "step": 33730
    },
    {
      "action_loss_avg": 0.002448596793692559,
      "epoch": 3.216743469238281e-05,
      "step": 33730
    },
    {
      "epoch": 3.216743469238281e-05,
      "step": 33730,
      "training_step_time": 7.556408166885376
    },
    {
      "epoch": 3.216838836669922e-05,
      "model_forward_time": 1.7457027435302734,
      "step": 33731
    },
    {
      "epoch": 3.216838836669922e-05,
      "step": 33731,
      "training_step_time": 9.221635341644287
    },
    {
      "epoch": 3.2169342041015624e-05,
      "model_forward_time": 1.757284164428711,
      "step": 33732
    },
    {
      "epoch": 3.2169342041015624e-05,
      "step": 33732,
      "training_step_time": 8.102226972579956
    },
    {
      "epoch": 3.217029571533203e-05,
      "model_forward_time": 1.7436771392822266,
      "step": 33733
    },
    {
      "epoch": 3.217029571533203e-05,
      "step": 33733,
      "training_step_time": 7.087797164916992
    },
    {
      "epoch": 3.217124938964844e-05,
      "model_forward_time": 1.759221076965332,
      "step": 33734
    },
    {
      "epoch": 3.217124938964844e-05,
      "step": 33734,
      "training_step_time": 7.048458099365234
    },
    {
      "epoch": 3.217220306396484e-05,
      "model_forward_time": 1.7429447174072266,
      "step": 33735
    },
    {
      "epoch": 3.217220306396484e-05,
      "step": 33735,
      "training_step_time": 7.026069641113281
    },
    {
      "epoch": 3.217315673828125e-05,
      "model_forward_time": 1.7339839935302734,
      "step": 33736
    },
    {
      "epoch": 3.217315673828125e-05,
      "step": 33736,
      "training_step_time": 7.063946962356567
    },
    {
      "epoch": 3.217411041259766e-05,
      "model_forward_time": 1.7516357898712158,
      "step": 33737
    },
    {
      "epoch": 3.217411041259766e-05,
      "step": 33737,
      "training_step_time": 7.059911251068115
    },
    {
      "epoch": 3.217506408691406e-05,
      "model_forward_time": 1.031421422958374,
      "step": 33738
    },
    {
      "epoch": 3.217506408691406e-05,
      "step": 33738,
      "training_step_time": 7.084154367446899
    },
    {
      "epoch": 3.2176017761230466e-05,
      "model_forward_time": 1.7406351566314697,
      "step": 33739
    },
    {
      "epoch": 3.2176017761230466e-05,
      "step": 33739,
      "training_step_time": 7.593734979629517
    },
    {
      "epoch": 3.217697143554688e-05,
      "grad_norm": 0.1586551070213318,
      "learning_rate": 7.906864126470524e-06,
      "loss": 0.0611,
      "step": 33740
    },
    {
      "epoch": 3.217697143554688e-05,
      "model_forward_time": 1.9120452404022217,
      "step": 33740
    },
    {
      "dynamics_loss_avg": 0.07396678738296032,
      "epoch": 3.217697143554688e-05,
      "step": 33740
    },
    {
      "action_loss_avg": 0.004140621027909219,
      "epoch": 3.217697143554688e-05,
      "step": 33740
    },
    {
      "epoch": 3.217697143554688e-05,
      "step": 33740,
      "training_step_time": 8.313376426696777
    },
    {
      "epoch": 3.217792510986328e-05,
      "model_forward_time": 1.7433929443359375,
      "step": 33741
    },
    {
      "epoch": 3.217792510986328e-05,
      "step": 33741,
      "training_step_time": 8.936603546142578
    },
    {
      "epoch": 3.2178878784179685e-05,
      "model_forward_time": 1.743297815322876,
      "step": 33742
    },
    {
      "epoch": 3.2178878784179685e-05,
      "step": 33742,
      "training_step_time": 7.33845853805542
    },
    {
      "epoch": 3.2179832458496096e-05,
      "model_forward_time": 1.7450027465820312,
      "step": 33743
    },
    {
      "epoch": 3.2179832458496096e-05,
      "step": 33743,
      "training_step_time": 7.054248809814453
    },
    {
      "epoch": 3.21807861328125e-05,
      "model_forward_time": 1.7340586185455322,
      "step": 33744
    },
    {
      "epoch": 3.21807861328125e-05,
      "step": 33744,
      "training_step_time": 7.06480073928833
    },
    {
      "epoch": 3.2181739807128904e-05,
      "model_forward_time": 1.7400124073028564,
      "step": 33745
    },
    {
      "epoch": 3.2181739807128904e-05,
      "step": 33745,
      "training_step_time": 7.075039625167847
    },
    {
      "epoch": 3.2182693481445315e-05,
      "model_forward_time": 1.7470088005065918,
      "step": 33746
    },
    {
      "epoch": 3.2182693481445315e-05,
      "step": 33746,
      "training_step_time": 7.030559778213501
    },
    {
      "epoch": 3.218364715576172e-05,
      "model_forward_time": 1.7506451606750488,
      "step": 33747
    },
    {
      "epoch": 3.218364715576172e-05,
      "step": 33747,
      "training_step_time": 7.0622429847717285
    },
    {
      "epoch": 3.218460083007812e-05,
      "model_forward_time": 1.7430756092071533,
      "step": 33748
    },
    {
      "epoch": 3.218460083007812e-05,
      "step": 33748,
      "training_step_time": 7.9309258460998535
    },
    {
      "epoch": 3.2185554504394533e-05,
      "model_forward_time": 2.393078565597534,
      "step": 33749
    },
    {
      "epoch": 3.2185554504394533e-05,
      "step": 33749,
      "training_step_time": 8.030571222305298
    },
    {
      "epoch": 3.218650817871094e-05,
      "grad_norm": 0.11757969111204147,
      "learning_rate": 7.905518642850041e-06,
      "loss": 0.0596,
      "step": 33750
    },
    {
      "epoch": 3.218650817871094e-05,
      "model_forward_time": 1.8401427268981934,
      "step": 33750
    },
    {
      "dynamics_loss_avg": 0.061311583314090964,
      "epoch": 3.218650817871094e-05,
      "step": 33750
    },
    {
      "action_loss_avg": 0.002549600671045482,
      "epoch": 3.218650817871094e-05,
      "step": 33750
    },
    {
      "epoch": 3.218650817871094e-05,
      "step": 33750,
      "training_step_time": 9.12976336479187
    },
    {
      "epoch": 3.218746185302734e-05,
      "model_forward_time": 1.8390967845916748,
      "step": 33751
    },
    {
      "epoch": 3.218746185302734e-05,
      "step": 33751,
      "training_step_time": 7.5924413204193115
    },
    {
      "epoch": 3.218841552734375e-05,
      "model_forward_time": 1.845139741897583,
      "step": 33752
    },
    {
      "epoch": 3.218841552734375e-05,
      "step": 33752,
      "training_step_time": 7.101977586746216
    },
    {
      "epoch": 3.2189369201660156e-05,
      "model_forward_time": 1.8412339687347412,
      "step": 33753
    },
    {
      "epoch": 3.2189369201660156e-05,
      "step": 33753,
      "training_step_time": 7.168946981430054
    },
    {
      "epoch": 3.219032287597656e-05,
      "model_forward_time": 1.8464441299438477,
      "step": 33754
    },
    {
      "epoch": 3.219032287597656e-05,
      "step": 33754,
      "training_step_time": 7.1442601680755615
    },
    {
      "epoch": 3.219127655029297e-05,
      "model_forward_time": 1.8301851749420166,
      "step": 33755
    },
    {
      "epoch": 3.219127655029297e-05,
      "step": 33755,
      "training_step_time": 7.120136260986328
    },
    {
      "epoch": 3.2192230224609375e-05,
      "model_forward_time": 1.815453290939331,
      "step": 33756
    },
    {
      "epoch": 3.2192230224609375e-05,
      "step": 33756,
      "training_step_time": 7.082442998886108
    },
    {
      "epoch": 3.219318389892578e-05,
      "model_forward_time": 1.0364031791687012,
      "step": 33757
    },
    {
      "epoch": 3.219318389892578e-05,
      "step": 33757,
      "training_step_time": 7.097911834716797
    },
    {
      "epoch": 3.219413757324219e-05,
      "model_forward_time": 1.7850432395935059,
      "step": 33758
    },
    {
      "epoch": 3.219413757324219e-05,
      "step": 33758,
      "training_step_time": 8.548619985580444
    },
    {
      "epoch": 3.2195091247558594e-05,
      "model_forward_time": 1.745800256729126,
      "step": 33759
    },
    {
      "epoch": 3.2195091247558594e-05,
      "step": 33759,
      "training_step_time": 8.750860214233398
    },
    {
      "epoch": 3.2196044921875e-05,
      "grad_norm": 0.20945531129837036,
      "learning_rate": 7.90417284148641e-06,
      "loss": 0.0635,
      "step": 33760
    },
    {
      "epoch": 3.2196044921875e-05,
      "model_forward_time": 1.73960542678833,
      "step": 33760
    },
    {
      "dynamics_loss_avg": 0.06275731641799212,
      "epoch": 3.2196044921875e-05,
      "step": 33760
    },
    {
      "action_loss_avg": 0.0013768706354312598,
      "epoch": 3.2196044921875e-05,
      "step": 33760
    },
    {
      "epoch": 3.2196044921875e-05,
      "step": 33760,
      "training_step_time": 8.169043064117432
    },
    {
      "epoch": 3.219699859619141e-05,
      "model_forward_time": 1.7402613162994385,
      "step": 33761
    },
    {
      "epoch": 3.219699859619141e-05,
      "step": 33761,
      "training_step_time": 7.116377115249634
    },
    {
      "epoch": 3.219795227050781e-05,
      "model_forward_time": 1.7630386352539062,
      "step": 33762
    },
    {
      "epoch": 3.219795227050781e-05,
      "step": 33762,
      "training_step_time": 7.147275447845459
    },
    {
      "epoch": 3.219890594482422e-05,
      "model_forward_time": 1.7332396507263184,
      "step": 33763
    },
    {
      "epoch": 3.219890594482422e-05,
      "step": 33763,
      "training_step_time": 7.154628038406372
    },
    {
      "epoch": 3.219985961914063e-05,
      "model_forward_time": 1.7420871257781982,
      "step": 33764
    },
    {
      "epoch": 3.219985961914063e-05,
      "step": 33764,
      "training_step_time": 7.160243034362793
    },
    {
      "epoch": 3.220081329345703e-05,
      "model_forward_time": 1.748042106628418,
      "step": 33765
    },
    {
      "epoch": 3.220081329345703e-05,
      "step": 33765,
      "training_step_time": 7.129323959350586
    },
    {
      "epoch": 3.2201766967773436e-05,
      "model_forward_time": 1.7266690731048584,
      "step": 33766
    },
    {
      "epoch": 3.2201766967773436e-05,
      "step": 33766,
      "training_step_time": 7.101215124130249
    },
    {
      "epoch": 3.220272064208985e-05,
      "model_forward_time": 1.7542634010314941,
      "step": 33767
    },
    {
      "epoch": 3.220272064208985e-05,
      "step": 33767,
      "training_step_time": 8.559011220932007
    },
    {
      "epoch": 3.220367431640625e-05,
      "model_forward_time": 2.302405595779419,
      "step": 33768
    },
    {
      "epoch": 3.220367431640625e-05,
      "step": 33768,
      "training_step_time": 8.388669967651367
    },
    {
      "epoch": 3.2204627990722655e-05,
      "model_forward_time": 1.7411186695098877,
      "step": 33769
    },
    {
      "epoch": 3.2204627990722655e-05,
      "step": 33769,
      "training_step_time": 8.150631666183472
    },
    {
      "epoch": 3.2205581665039065e-05,
      "grad_norm": 0.18039722740650177,
      "learning_rate": 7.902826722526801e-06,
      "loss": 0.0636,
      "step": 33770
    },
    {
      "epoch": 3.2205581665039065e-05,
      "model_forward_time": 1.724252462387085,
      "step": 33770
    },
    {
      "dynamics_loss_avg": 0.06760884579271079,
      "epoch": 3.2205581665039065e-05,
      "step": 33770
    },
    {
      "action_loss_avg": 0.0023954673641128467,
      "epoch": 3.2205581665039065e-05,
      "step": 33770
    },
    {
      "epoch": 3.2205581665039065e-05,
      "step": 33770,
      "training_step_time": 7.362563371658325
    },
    {
      "epoch": 3.220653533935547e-05,
      "model_forward_time": 1.7326164245605469,
      "step": 33771
    },
    {
      "epoch": 3.220653533935547e-05,
      "step": 33771,
      "training_step_time": 7.03336763381958
    },
    {
      "epoch": 3.2207489013671874e-05,
      "model_forward_time": 1.7349748611450195,
      "step": 33772
    },
    {
      "epoch": 3.2207489013671874e-05,
      "step": 33772,
      "training_step_time": 7.160849571228027
    },
    {
      "epoch": 3.2208442687988284e-05,
      "model_forward_time": 1.7288422584533691,
      "step": 33773
    },
    {
      "epoch": 3.2208442687988284e-05,
      "step": 33773,
      "training_step_time": 7.099454164505005
    },
    {
      "epoch": 3.220939636230469e-05,
      "model_forward_time": 1.7416040897369385,
      "step": 33774
    },
    {
      "epoch": 3.220939636230469e-05,
      "step": 33774,
      "training_step_time": 7.053361892700195
    },
    {
      "epoch": 3.221035003662109e-05,
      "model_forward_time": 1.7400569915771484,
      "step": 33775
    },
    {
      "epoch": 3.221035003662109e-05,
      "step": 33775,
      "training_step_time": 7.10785984992981
    },
    {
      "epoch": 3.22113037109375e-05,
      "model_forward_time": 1.7279753684997559,
      "step": 33776
    },
    {
      "epoch": 3.22113037109375e-05,
      "step": 33776,
      "training_step_time": 8.514851093292236
    },
    {
      "epoch": 3.221225738525391e-05,
      "model_forward_time": 2.2797985076904297,
      "step": 33777
    },
    {
      "epoch": 3.221225738525391e-05,
      "step": 33777,
      "training_step_time": 8.614115715026855
    },
    {
      "epoch": 3.221321105957031e-05,
      "model_forward_time": 1.7710061073303223,
      "step": 33778
    },
    {
      "epoch": 3.221321105957031e-05,
      "step": 33778,
      "training_step_time": 8.034595727920532
    },
    {
      "epoch": 3.221416473388672e-05,
      "model_forward_time": 1.7355363368988037,
      "step": 33779
    },
    {
      "epoch": 3.221416473388672e-05,
      "step": 33779,
      "training_step_time": 7.498699426651001
    },
    {
      "epoch": 3.2215118408203126e-05,
      "grad_norm": 0.12301371246576309,
      "learning_rate": 7.901480286118427e-06,
      "loss": 0.0626,
      "step": 33780
    },
    {
      "epoch": 3.2215118408203126e-05,
      "model_forward_time": 1.7349674701690674,
      "step": 33780
    },
    {
      "dynamics_loss_avg": 0.047603252436965705,
      "epoch": 3.2215118408203126e-05,
      "step": 33780
    },
    {
      "action_loss_avg": 0.0025855849700747056,
      "epoch": 3.2215118408203126e-05,
      "step": 33780
    },
    {
      "epoch": 3.2215118408203126e-05,
      "step": 33780,
      "training_step_time": 7.0043721199035645
    },
    {
      "epoch": 3.221607208251953e-05,
      "model_forward_time": 1.738055944442749,
      "step": 33781
    },
    {
      "epoch": 3.221607208251953e-05,
      "step": 33781,
      "training_step_time": 7.044961929321289
    },
    {
      "epoch": 3.2217025756835934e-05,
      "model_forward_time": 1.7090024948120117,
      "step": 33782
    },
    {
      "epoch": 3.2217025756835934e-05,
      "step": 33782,
      "training_step_time": 7.068312883377075
    },
    {
      "epoch": 3.2217979431152345e-05,
      "model_forward_time": 1.7215783596038818,
      "step": 33783
    },
    {
      "epoch": 3.2217979431152345e-05,
      "step": 33783,
      "training_step_time": 7.063666343688965
    },
    {
      "epoch": 3.221893310546875e-05,
      "model_forward_time": 1.7078096866607666,
      "step": 33784
    },
    {
      "epoch": 3.221893310546875e-05,
      "step": 33784,
      "training_step_time": 7.062635183334351
    },
    {
      "epoch": 3.221988677978515e-05,
      "model_forward_time": 1.710890293121338,
      "step": 33785
    },
    {
      "epoch": 3.221988677978515e-05,
      "step": 33785,
      "training_step_time": 7.724117040634155
    },
    {
      "epoch": 3.2220840454101564e-05,
      "model_forward_time": 2.256300449371338,
      "step": 33786
    },
    {
      "epoch": 3.2220840454101564e-05,
      "step": 33786,
      "training_step_time": 9.185563087463379
    },
    {
      "epoch": 3.222179412841797e-05,
      "model_forward_time": 1.7141342163085938,
      "step": 33787
    },
    {
      "epoch": 3.222179412841797e-05,
      "step": 33787,
      "training_step_time": 8.574557065963745
    },
    {
      "epoch": 3.222274780273437e-05,
      "model_forward_time": 1.716860294342041,
      "step": 33788
    },
    {
      "epoch": 3.222274780273437e-05,
      "step": 33788,
      "training_step_time": 7.673196077346802
    },
    {
      "epoch": 3.222370147705078e-05,
      "model_forward_time": 1.7092766761779785,
      "step": 33789
    },
    {
      "epoch": 3.222370147705078e-05,
      "step": 33789,
      "training_step_time": 7.024135589599609
    },
    {
      "epoch": 3.222465515136719e-05,
      "grad_norm": 0.12957844138145447,
      "learning_rate": 7.900133532408532e-06,
      "loss": 0.065,
      "step": 33790
    },
    {
      "epoch": 3.222465515136719e-05,
      "model_forward_time": 1.7149286270141602,
      "step": 33790
    },
    {
      "dynamics_loss_avg": 0.06713766548782588,
      "epoch": 3.222465515136719e-05,
      "step": 33790
    },
    {
      "action_loss_avg": 0.0029289271333254874,
      "epoch": 3.222465515136719e-05,
      "step": 33790
    },
    {
      "epoch": 3.222465515136719e-05,
      "step": 33790,
      "training_step_time": 7.006136655807495
    },
    {
      "epoch": 3.222560882568359e-05,
      "model_forward_time": 1.7309043407440186,
      "step": 33791
    },
    {
      "epoch": 3.222560882568359e-05,
      "step": 33791,
      "training_step_time": 7.0843665599823
    },
    {
      "epoch": 3.22265625e-05,
      "model_forward_time": 1.720080852508545,
      "step": 33792
    },
    {
      "epoch": 3.22265625e-05,
      "step": 33792,
      "training_step_time": 7.053896427154541
    },
    {
      "epoch": 3.2227516174316406e-05,
      "model_forward_time": 1.7150201797485352,
      "step": 33793
    },
    {
      "epoch": 3.2227516174316406e-05,
      "step": 33793,
      "training_step_time": 7.07858943939209
    },
    {
      "epoch": 3.222846984863281e-05,
      "model_forward_time": 1.6835708618164062,
      "step": 33794
    },
    {
      "epoch": 3.222846984863281e-05,
      "step": 33794,
      "training_step_time": 7.6039955615997314
    },
    {
      "epoch": 3.222942352294922e-05,
      "model_forward_time": 2.2721424102783203,
      "step": 33795
    },
    {
      "epoch": 3.222942352294922e-05,
      "step": 33795,
      "training_step_time": 10.162744283676147
    },
    {
      "epoch": 3.2230377197265624e-05,
      "model_forward_time": 1.7566065788269043,
      "step": 33796
    },
    {
      "epoch": 3.2230377197265624e-05,
      "step": 33796,
      "training_step_time": 8.652956008911133
    },
    {
      "epoch": 3.223133087158203e-05,
      "model_forward_time": 1.709458827972412,
      "step": 33797
    },
    {
      "epoch": 3.223133087158203e-05,
      "step": 33797,
      "training_step_time": 7.945075035095215
    },
    {
      "epoch": 3.223228454589844e-05,
      "model_forward_time": 1.7200665473937988,
      "step": 33798
    },
    {
      "epoch": 3.223228454589844e-05,
      "step": 33798,
      "training_step_time": 7.020153284072876
    },
    {
      "epoch": 3.223323822021484e-05,
      "model_forward_time": 1.710296869277954,
      "step": 33799
    },
    {
      "epoch": 3.223323822021484e-05,
      "step": 33799,
      "training_step_time": 7.024539470672607
    },
    {
      "epoch": 3.223419189453125e-05,
      "grad_norm": 0.15241673588752747,
      "learning_rate": 7.898786461544394e-06,
      "loss": 0.0634,
      "step": 33800
    },
    {
      "epoch": 3.223419189453125e-05,
      "model_forward_time": 1.719306230545044,
      "step": 33800
    },
    {
      "dynamics_loss_avg": 0.062265456281602384,
      "epoch": 3.223419189453125e-05,
      "step": 33800
    },
    {
      "action_loss_avg": 0.0017036118370015174,
      "epoch": 3.223419189453125e-05,
      "step": 33800
    },
    {
      "epoch": 3.223419189453125e-05,
      "step": 33800,
      "training_step_time": 7.0345985889434814
    },
    {
      "epoch": 3.223514556884766e-05,
      "model_forward_time": 1.7418220043182373,
      "step": 33801
    },
    {
      "epoch": 3.223514556884766e-05,
      "step": 33801,
      "training_step_time": 7.043472766876221
    },
    {
      "epoch": 3.223609924316406e-05,
      "model_forward_time": 1.7344565391540527,
      "step": 33802
    },
    {
      "epoch": 3.223609924316406e-05,
      "step": 33802,
      "training_step_time": 7.0496275424957275
    },
    {
      "epoch": 3.2237052917480466e-05,
      "model_forward_time": 1.73530912399292,
      "step": 33803
    },
    {
      "epoch": 3.2237052917480466e-05,
      "step": 33803,
      "training_step_time": 7.751335859298706
    },
    {
      "epoch": 3.223800659179688e-05,
      "model_forward_time": 2.295732259750366,
      "step": 33804
    },
    {
      "epoch": 3.223800659179688e-05,
      "step": 33804,
      "training_step_time": 9.623269081115723
    },
    {
      "epoch": 3.223896026611328e-05,
      "model_forward_time": 1.7756831645965576,
      "step": 33805
    },
    {
      "epoch": 3.223896026611328e-05,
      "step": 33805,
      "training_step_time": 9.661035776138306
    },
    {
      "epoch": 3.2239913940429685e-05,
      "model_forward_time": 1.7447264194488525,
      "step": 33806
    },
    {
      "epoch": 3.2239913940429685e-05,
      "step": 33806,
      "training_step_time": 7.9221391677856445
    },
    {
      "epoch": 3.2240867614746096e-05,
      "model_forward_time": 1.7392332553863525,
      "step": 33807
    },
    {
      "epoch": 3.2240867614746096e-05,
      "step": 33807,
      "training_step_time": 7.472825288772583
    },
    {
      "epoch": 3.22418212890625e-05,
      "model_forward_time": 1.7375609874725342,
      "step": 33808
    },
    {
      "epoch": 3.22418212890625e-05,
      "step": 33808,
      "training_step_time": 7.039732217788696
    },
    {
      "epoch": 3.2242774963378904e-05,
      "model_forward_time": 1.7387440204620361,
      "step": 33809
    },
    {
      "epoch": 3.2242774963378904e-05,
      "step": 33809,
      "training_step_time": 7.0688698291778564
    },
    {
      "epoch": 3.2243728637695315e-05,
      "grad_norm": 0.18390513956546783,
      "learning_rate": 7.897439073673325e-06,
      "loss": 0.0588,
      "step": 33810
    },
    {
      "epoch": 3.2243728637695315e-05,
      "model_forward_time": 1.7333378791809082,
      "step": 33810
    },
    {
      "dynamics_loss_avg": 0.04231377150863409,
      "epoch": 3.2243728637695315e-05,
      "step": 33810
    },
    {
      "action_loss_avg": 0.0017680212564300745,
      "epoch": 3.2243728637695315e-05,
      "step": 33810
    },
    {
      "epoch": 3.2243728637695315e-05,
      "step": 33810,
      "training_step_time": 7.0513598918914795
    },
    {
      "epoch": 3.224468231201172e-05,
      "model_forward_time": 1.730729103088379,
      "step": 33811
    },
    {
      "epoch": 3.224468231201172e-05,
      "step": 33811,
      "training_step_time": 6.94800066947937
    },
    {
      "epoch": 3.224563598632812e-05,
      "model_forward_time": 1.7261605262756348,
      "step": 33812
    },
    {
      "epoch": 3.224563598632812e-05,
      "step": 33812,
      "training_step_time": 7.691533327102661
    },
    {
      "epoch": 3.2246589660644534e-05,
      "model_forward_time": 2.262529134750366,
      "step": 33813
    },
    {
      "epoch": 3.2246589660644534e-05,
      "step": 33813,
      "training_step_time": 9.860430240631104
    },
    {
      "epoch": 3.224754333496094e-05,
      "model_forward_time": 1.7688829898834229,
      "step": 33814
    },
    {
      "epoch": 3.224754333496094e-05,
      "step": 33814,
      "training_step_time": 10.456998348236084
    },
    {
      "epoch": 3.224849700927734e-05,
      "model_forward_time": 1.73795747756958,
      "step": 33815
    },
    {
      "epoch": 3.224849700927734e-05,
      "step": 33815,
      "training_step_time": 8.796101093292236
    },
    {
      "epoch": 3.224945068359375e-05,
      "model_forward_time": 1.7374367713928223,
      "step": 33816
    },
    {
      "epoch": 3.224945068359375e-05,
      "step": 33816,
      "training_step_time": 7.746558427810669
    },
    {
      "epoch": 3.2250404357910156e-05,
      "model_forward_time": 1.7362773418426514,
      "step": 33817
    },
    {
      "epoch": 3.2250404357910156e-05,
      "step": 33817,
      "training_step_time": 7.091443061828613
    },
    {
      "epoch": 3.225135803222656e-05,
      "model_forward_time": 1.732168436050415,
      "step": 33818
    },
    {
      "epoch": 3.225135803222656e-05,
      "step": 33818,
      "training_step_time": 7.10613226890564
    },
    {
      "epoch": 3.225231170654297e-05,
      "model_forward_time": 1.742837905883789,
      "step": 33819
    },
    {
      "epoch": 3.225231170654297e-05,
      "step": 33819,
      "training_step_time": 7.089317560195923
    },
    {
      "epoch": 3.2253265380859375e-05,
      "grad_norm": 0.1295246034860611,
      "learning_rate": 7.896091368942677e-06,
      "loss": 0.0597,
      "step": 33820
    },
    {
      "epoch": 3.2253265380859375e-05,
      "model_forward_time": 1.7209503650665283,
      "step": 33820
    },
    {
      "dynamics_loss_avg": 0.05436754319816828,
      "epoch": 3.2253265380859375e-05,
      "step": 33820
    },
    {
      "action_loss_avg": 0.0027823868091218173,
      "epoch": 3.2253265380859375e-05,
      "step": 33820
    },
    {
      "epoch": 3.2253265380859375e-05,
      "step": 33820,
      "training_step_time": 7.079787969589233
    },
    {
      "epoch": 3.225421905517578e-05,
      "model_forward_time": 1.7339153289794922,
      "step": 33821
    },
    {
      "epoch": 3.225421905517578e-05,
      "step": 33821,
      "training_step_time": 7.739943265914917
    },
    {
      "epoch": 3.225517272949219e-05,
      "model_forward_time": 1.0261485576629639,
      "step": 33822
    },
    {
      "epoch": 3.225517272949219e-05,
      "step": 33822,
      "training_step_time": 10.293658018112183
    },
    {
      "epoch": 3.2256126403808594e-05,
      "model_forward_time": 2.288910150527954,
      "step": 33823
    },
    {
      "epoch": 3.2256126403808594e-05,
      "step": 33823,
      "training_step_time": 10.560625553131104
    },
    {
      "epoch": 3.2257080078125e-05,
      "model_forward_time": 1.3671433925628662,
      "step": 33824
    },
    {
      "epoch": 3.2257080078125e-05,
      "step": 33824,
      "training_step_time": 9.252066135406494
    },
    {
      "epoch": 3.225803375244141e-05,
      "model_forward_time": 1.7326481342315674,
      "step": 33825
    },
    {
      "epoch": 3.225803375244141e-05,
      "step": 33825,
      "training_step_time": 7.755008935928345
    },
    {
      "epoch": 3.225898742675781e-05,
      "model_forward_time": 1.7380714416503906,
      "step": 33826
    },
    {
      "epoch": 3.225898742675781e-05,
      "step": 33826,
      "training_step_time": 7.101561069488525
    },
    {
      "epoch": 3.225994110107422e-05,
      "model_forward_time": 1.7223742008209229,
      "step": 33827
    },
    {
      "epoch": 3.225994110107422e-05,
      "step": 33827,
      "training_step_time": 7.115065097808838
    },
    {
      "epoch": 3.226089477539063e-05,
      "model_forward_time": 1.7376213073730469,
      "step": 33828
    },
    {
      "epoch": 3.226089477539063e-05,
      "step": 33828,
      "training_step_time": 7.125793933868408
    },
    {
      "epoch": 3.226184844970703e-05,
      "model_forward_time": 1.0271754264831543,
      "step": 33829
    },
    {
      "epoch": 3.226184844970703e-05,
      "step": 33829,
      "training_step_time": 7.131014108657837
    },
    {
      "epoch": 3.2262802124023436e-05,
      "grad_norm": 0.09593082964420319,
      "learning_rate": 7.894743347499832e-06,
      "loss": 0.0611,
      "step": 33830
    },
    {
      "epoch": 3.2262802124023436e-05,
      "model_forward_time": 1.7366516590118408,
      "step": 33830
    },
    {
      "dynamics_loss_avg": 0.057028414122760294,
      "epoch": 3.2262802124023436e-05,
      "step": 33830
    },
    {
      "action_loss_avg": 0.002871253585908562,
      "epoch": 3.2262802124023436e-05,
      "step": 33830
    },
    {
      "epoch": 3.2262802124023436e-05,
      "step": 33830,
      "training_step_time": 7.735209703445435
    },
    {
      "epoch": 3.226375579833985e-05,
      "model_forward_time": 1.7297415733337402,
      "step": 33831
    },
    {
      "epoch": 3.226375579833985e-05,
      "step": 33831,
      "training_step_time": 8.50687050819397
    },
    {
      "epoch": 3.226470947265625e-05,
      "model_forward_time": 1.7886476516723633,
      "step": 33832
    },
    {
      "epoch": 3.226470947265625e-05,
      "step": 33832,
      "training_step_time": 9.942097663879395
    },
    {
      "epoch": 3.2265663146972655e-05,
      "model_forward_time": 2.293208599090576,
      "step": 33833
    },
    {
      "epoch": 3.2265663146972655e-05,
      "step": 33833,
      "training_step_time": 8.920968055725098
    },
    {
      "epoch": 3.2266616821289066e-05,
      "model_forward_time": 1.731489658355713,
      "step": 33834
    },
    {
      "epoch": 3.2266616821289066e-05,
      "step": 33834,
      "training_step_time": 7.890979290008545
    },
    {
      "epoch": 3.226757049560547e-05,
      "model_forward_time": 1.7348370552062988,
      "step": 33835
    },
    {
      "epoch": 3.226757049560547e-05,
      "step": 33835,
      "training_step_time": 7.084702491760254
    },
    {
      "epoch": 3.2268524169921874e-05,
      "model_forward_time": 1.730665922164917,
      "step": 33836
    },
    {
      "epoch": 3.2268524169921874e-05,
      "step": 33836,
      "training_step_time": 7.0642149448394775
    },
    {
      "epoch": 3.2269477844238284e-05,
      "model_forward_time": 1.7255768775939941,
      "step": 33837
    },
    {
      "epoch": 3.2269477844238284e-05,
      "step": 33837,
      "training_step_time": 7.007062911987305
    },
    {
      "epoch": 3.227043151855469e-05,
      "model_forward_time": 1.738347053527832,
      "step": 33838
    },
    {
      "epoch": 3.227043151855469e-05,
      "step": 33838,
      "training_step_time": 7.106482028961182
    },
    {
      "epoch": 3.227138519287109e-05,
      "model_forward_time": 1.8364222049713135,
      "step": 33839
    },
    {
      "epoch": 3.227138519287109e-05,
      "step": 33839,
      "training_step_time": 7.741288185119629
    },
    {
      "epoch": 3.22723388671875e-05,
      "grad_norm": 0.1943366527557373,
      "learning_rate": 7.893395009492205e-06,
      "loss": 0.0585,
      "step": 33840
    },
    {
      "epoch": 3.22723388671875e-05,
      "model_forward_time": 1.8355119228363037,
      "step": 33840
    },
    {
      "dynamics_loss_avg": 0.06722551882266999,
      "epoch": 3.22723388671875e-05,
      "step": 33840
    },
    {
      "action_loss_avg": 0.0028389366285409777,
      "epoch": 3.22723388671875e-05,
      "step": 33840
    },
    {
      "epoch": 3.22723388671875e-05,
      "step": 33840,
      "training_step_time": 8.463522911071777
    },
    {
      "epoch": 3.227329254150391e-05,
      "model_forward_time": 1.832235336303711,
      "step": 33841
    },
    {
      "epoch": 3.227329254150391e-05,
      "step": 33841,
      "training_step_time": 9.76916241645813
    },
    {
      "epoch": 3.227424621582031e-05,
      "model_forward_time": 2.398895502090454,
      "step": 33842
    },
    {
      "epoch": 3.227424621582031e-05,
      "step": 33842,
      "training_step_time": 8.629706621170044
    },
    {
      "epoch": 3.227519989013672e-05,
      "model_forward_time": 1.835150957107544,
      "step": 33843
    },
    {
      "epoch": 3.227519989013672e-05,
      "step": 33843,
      "training_step_time": 8.390320301055908
    },
    {
      "epoch": 3.2276153564453126e-05,
      "model_forward_time": 1.839057207107544,
      "step": 33844
    },
    {
      "epoch": 3.2276153564453126e-05,
      "step": 33844,
      "training_step_time": 7.532703638076782
    },
    {
      "epoch": 3.227710723876953e-05,
      "model_forward_time": 1.838052749633789,
      "step": 33845
    },
    {
      "epoch": 3.227710723876953e-05,
      "step": 33845,
      "training_step_time": 7.053262710571289
    },
    {
      "epoch": 3.2278060913085934e-05,
      "model_forward_time": 1.8408865928649902,
      "step": 33846
    },
    {
      "epoch": 3.2278060913085934e-05,
      "step": 33846,
      "training_step_time": 7.1433117389678955
    },
    {
      "epoch": 3.2279014587402345e-05,
      "model_forward_time": 1.838862419128418,
      "step": 33847
    },
    {
      "epoch": 3.2279014587402345e-05,
      "step": 33847,
      "training_step_time": 7.12169337272644
    },
    {
      "epoch": 3.227996826171875e-05,
      "model_forward_time": 1.8208043575286865,
      "step": 33848
    },
    {
      "epoch": 3.227996826171875e-05,
      "step": 33848,
      "training_step_time": 7.7930216789245605
    },
    {
      "epoch": 3.228092193603515e-05,
      "model_forward_time": 1.7356889247894287,
      "step": 33849
    },
    {
      "epoch": 3.228092193603515e-05,
      "step": 33849,
      "training_step_time": 7.713852643966675
    },
    {
      "epoch": 3.2281875610351564e-05,
      "grad_norm": 0.10630128532648087,
      "learning_rate": 7.892046355067249e-06,
      "loss": 0.0614,
      "step": 33850
    },
    {
      "epoch": 3.2281875610351564e-05,
      "model_forward_time": 1.3551392555236816,
      "step": 33850
    },
    {
      "dynamics_loss_avg": 0.06338754408061505,
      "epoch": 3.2281875610351564e-05,
      "step": 33850
    },
    {
      "action_loss_avg": 0.0017348248278722168,
      "epoch": 3.2281875610351564e-05,
      "step": 33850
    },
    {
      "epoch": 3.2281875610351564e-05,
      "step": 33850,
      "training_step_time": 10.638406753540039
    },
    {
      "epoch": 3.228282928466797e-05,
      "model_forward_time": 2.299448251724243,
      "step": 33851
    },
    {
      "epoch": 3.228282928466797e-05,
      "step": 33851,
      "training_step_time": 8.958308458328247
    },
    {
      "epoch": 3.228378295898437e-05,
      "model_forward_time": 1.7516846656799316,
      "step": 33852
    },
    {
      "epoch": 3.228378295898437e-05,
      "step": 33852,
      "training_step_time": 7.826432228088379
    },
    {
      "epoch": 3.228473663330078e-05,
      "model_forward_time": 1.7476577758789062,
      "step": 33853
    },
    {
      "epoch": 3.228473663330078e-05,
      "step": 33853,
      "training_step_time": 7.477910995483398
    },
    {
      "epoch": 3.228569030761719e-05,
      "model_forward_time": 1.7275097370147705,
      "step": 33854
    },
    {
      "epoch": 3.228569030761719e-05,
      "step": 33854,
      "training_step_time": 7.016758680343628
    },
    {
      "epoch": 3.228664398193359e-05,
      "model_forward_time": 1.7238352298736572,
      "step": 33855
    },
    {
      "epoch": 3.228664398193359e-05,
      "step": 33855,
      "training_step_time": 6.927484750747681
    },
    {
      "epoch": 3.228759765625e-05,
      "model_forward_time": 1.7209231853485107,
      "step": 33856
    },
    {
      "epoch": 3.228759765625e-05,
      "step": 33856,
      "training_step_time": 7.018615484237671
    },
    {
      "epoch": 3.2288551330566406e-05,
      "model_forward_time": 1.7250862121582031,
      "step": 33857
    },
    {
      "epoch": 3.2288551330566406e-05,
      "step": 33857,
      "training_step_time": 7.519282817840576
    },
    {
      "epoch": 3.228950500488281e-05,
      "model_forward_time": 1.3459575176239014,
      "step": 33858
    },
    {
      "epoch": 3.228950500488281e-05,
      "step": 33858,
      "training_step_time": 7.684979438781738
    },
    {
      "epoch": 3.229045867919922e-05,
      "model_forward_time": 1.7238531112670898,
      "step": 33859
    },
    {
      "epoch": 3.229045867919922e-05,
      "step": 33859,
      "training_step_time": 8.379518747329712
    },
    {
      "epoch": 3.2291412353515624e-05,
      "grad_norm": 0.1240648403763771,
      "learning_rate": 7.890697384372453e-06,
      "loss": 0.059,
      "step": 33860
    },
    {
      "epoch": 3.2291412353515624e-05,
      "model_forward_time": 1.4183688163757324,
      "step": 33860
    },
    {
      "dynamics_loss_avg": 0.0513448104262352,
      "epoch": 3.2291412353515624e-05,
      "step": 33860
    },
    {
      "action_loss_avg": 0.0019164105236995966,
      "epoch": 3.2291412353515624e-05,
      "step": 33860
    },
    {
      "epoch": 3.2291412353515624e-05,
      "step": 33860,
      "training_step_time": 8.31837010383606
    },
    {
      "epoch": 3.229236602783203e-05,
      "model_forward_time": 2.2694039344787598,
      "step": 33861
    },
    {
      "epoch": 3.229236602783203e-05,
      "step": 33861,
      "training_step_time": 8.350951194763184
    },
    {
      "epoch": 3.229331970214844e-05,
      "model_forward_time": 1.7268779277801514,
      "step": 33862
    },
    {
      "epoch": 3.229331970214844e-05,
      "step": 33862,
      "training_step_time": 7.508662223815918
    },
    {
      "epoch": 3.229427337646484e-05,
      "model_forward_time": 1.7266302108764648,
      "step": 33863
    },
    {
      "epoch": 3.229427337646484e-05,
      "step": 33863,
      "training_step_time": 7.084775686264038
    },
    {
      "epoch": 3.229522705078125e-05,
      "model_forward_time": 1.735703468322754,
      "step": 33864
    },
    {
      "epoch": 3.229522705078125e-05,
      "step": 33864,
      "training_step_time": 6.866699457168579
    },
    {
      "epoch": 3.229618072509766e-05,
      "model_forward_time": 1.7313151359558105,
      "step": 33865
    },
    {
      "epoch": 3.229618072509766e-05,
      "step": 33865,
      "training_step_time": 7.103205919265747
    },
    {
      "epoch": 3.229713439941406e-05,
      "model_forward_time": 1.7303106784820557,
      "step": 33866
    },
    {
      "epoch": 3.229713439941406e-05,
      "step": 33866,
      "training_step_time": 7.736680030822754
    },
    {
      "epoch": 3.2298088073730466e-05,
      "model_forward_time": 1.7310752868652344,
      "step": 33867
    },
    {
      "epoch": 3.2298088073730466e-05,
      "step": 33867,
      "training_step_time": 8.831539869308472
    },
    {
      "epoch": 3.229904174804688e-05,
      "model_forward_time": 1.7516138553619385,
      "step": 33868
    },
    {
      "epoch": 3.229904174804688e-05,
      "step": 33868,
      "training_step_time": 8.939380407333374
    },
    {
      "epoch": 3.229999542236328e-05,
      "model_forward_time": 1.763709306716919,
      "step": 33869
    },
    {
      "epoch": 3.229999542236328e-05,
      "step": 33869,
      "training_step_time": 9.733933925628662
    },
    {
      "epoch": 3.2300949096679685e-05,
      "grad_norm": 0.10684047639369965,
      "learning_rate": 7.889348097555336e-06,
      "loss": 0.0611,
      "step": 33870
    },
    {
      "epoch": 3.2300949096679685e-05,
      "model_forward_time": 2.2580442428588867,
      "step": 33870
    },
    {
      "dynamics_loss_avg": 0.04967353716492653,
      "epoch": 3.2300949096679685e-05,
      "step": 33870
    },
    {
      "action_loss_avg": 0.0013670703599927946,
      "epoch": 3.2300949096679685e-05,
      "step": 33870
    },
    {
      "epoch": 3.2300949096679685e-05,
      "step": 33870,
      "training_step_time": 8.556075811386108
    },
    {
      "epoch": 3.2301902770996096e-05,
      "model_forward_time": 1.735144853591919,
      "step": 33871
    },
    {
      "epoch": 3.2301902770996096e-05,
      "step": 33871,
      "training_step_time": 7.825559616088867
    },
    {
      "epoch": 3.23028564453125e-05,
      "model_forward_time": 1.7378389835357666,
      "step": 33872
    },
    {
      "epoch": 3.23028564453125e-05,
      "step": 33872,
      "training_step_time": 6.983428716659546
    },
    {
      "epoch": 3.2303810119628904e-05,
      "model_forward_time": 1.741386890411377,
      "step": 33873
    },
    {
      "epoch": 3.2303810119628904e-05,
      "step": 33873,
      "training_step_time": 7.039510488510132
    },
    {
      "epoch": 3.2304763793945315e-05,
      "model_forward_time": 1.7197251319885254,
      "step": 33874
    },
    {
      "epoch": 3.2304763793945315e-05,
      "step": 33874,
      "training_step_time": 7.012915372848511
    },
    {
      "epoch": 3.230571746826172e-05,
      "model_forward_time": 1.7354097366333008,
      "step": 33875
    },
    {
      "epoch": 3.230571746826172e-05,
      "step": 33875,
      "training_step_time": 7.731905698776245
    },
    {
      "epoch": 3.230667114257812e-05,
      "model_forward_time": 1.7250616550445557,
      "step": 33876
    },
    {
      "epoch": 3.230667114257812e-05,
      "step": 33876,
      "training_step_time": 7.803783416748047
    },
    {
      "epoch": 3.2307624816894534e-05,
      "model_forward_time": 1.713683843612671,
      "step": 33877
    },
    {
      "epoch": 3.2307624816894534e-05,
      "step": 33877,
      "training_step_time": 7.8473265171051025
    },
    {
      "epoch": 3.230857849121094e-05,
      "model_forward_time": 1.7453091144561768,
      "step": 33878
    },
    {
      "epoch": 3.230857849121094e-05,
      "step": 33878,
      "training_step_time": 9.913729667663574
    },
    {
      "epoch": 3.230953216552734e-05,
      "model_forward_time": 2.274385690689087,
      "step": 33879
    },
    {
      "epoch": 3.230953216552734e-05,
      "step": 33879,
      "training_step_time": 7.840159893035889
    },
    {
      "epoch": 3.231048583984375e-05,
      "grad_norm": 0.10810641944408417,
      "learning_rate": 7.887998494763457e-06,
      "loss": 0.062,
      "step": 33880
    },
    {
      "epoch": 3.231048583984375e-05,
      "model_forward_time": 1.7298824787139893,
      "step": 33880
    },
    {
      "dynamics_loss_avg": 0.05715153990313411,
      "epoch": 3.231048583984375e-05,
      "step": 33880
    },
    {
      "action_loss_avg": 0.002096745418384671,
      "epoch": 3.231048583984375e-05,
      "step": 33880
    },
    {
      "epoch": 3.231048583984375e-05,
      "step": 33880,
      "training_step_time": 8.033250093460083
    },
    {
      "epoch": 3.2311439514160156e-05,
      "model_forward_time": 1.7287092208862305,
      "step": 33881
    },
    {
      "epoch": 3.2311439514160156e-05,
      "step": 33881,
      "training_step_time": 7.199510335922241
    },
    {
      "epoch": 3.231239318847656e-05,
      "model_forward_time": 1.7242562770843506,
      "step": 33882
    },
    {
      "epoch": 3.231239318847656e-05,
      "step": 33882,
      "training_step_time": 7.085256338119507
    },
    {
      "epoch": 3.231334686279297e-05,
      "model_forward_time": 1.7250721454620361,
      "step": 33883
    },
    {
      "epoch": 3.231334686279297e-05,
      "step": 33883,
      "training_step_time": 7.040209770202637
    },
    {
      "epoch": 3.2314300537109375e-05,
      "model_forward_time": 1.723609447479248,
      "step": 33884
    },
    {
      "epoch": 3.2314300537109375e-05,
      "step": 33884,
      "training_step_time": 7.715080738067627
    },
    {
      "epoch": 3.231525421142578e-05,
      "model_forward_time": 1.7282025814056396,
      "step": 33885
    },
    {
      "epoch": 3.231525421142578e-05,
      "step": 33885,
      "training_step_time": 7.828623533248901
    },
    {
      "epoch": 3.231620788574219e-05,
      "model_forward_time": 1.723381519317627,
      "step": 33886
    },
    {
      "epoch": 3.231620788574219e-05,
      "step": 33886,
      "training_step_time": 8.914429187774658
    },
    {
      "epoch": 3.2317161560058594e-05,
      "model_forward_time": 1.731135368347168,
      "step": 33887
    },
    {
      "epoch": 3.2317161560058594e-05,
      "step": 33887,
      "training_step_time": 7.7871410846710205
    },
    {
      "epoch": 3.2318115234375e-05,
      "model_forward_time": 2.270883321762085,
      "step": 33888
    },
    {
      "epoch": 3.2318115234375e-05,
      "step": 33888,
      "training_step_time": 9.382863283157349
    },
    {
      "epoch": 3.231906890869141e-05,
      "model_forward_time": 1.728585958480835,
      "step": 33889
    },
    {
      "epoch": 3.231906890869141e-05,
      "step": 33889,
      "training_step_time": 8.062543630599976
    },
    {
      "epoch": 3.232002258300781e-05,
      "grad_norm": 0.16476011276245117,
      "learning_rate": 7.886648576144404e-06,
      "loss": 0.0604,
      "step": 33890
    },
    {
      "epoch": 3.232002258300781e-05,
      "model_forward_time": 1.7223155498504639,
      "step": 33890
    },
    {
      "dynamics_loss_avg": 0.058516382798552515,
      "epoch": 3.232002258300781e-05,
      "step": 33890
    },
    {
      "action_loss_avg": 0.0026030912646092474,
      "epoch": 3.232002258300781e-05,
      "step": 33890
    },
    {
      "epoch": 3.232002258300781e-05,
      "step": 33890,
      "training_step_time": 7.498762607574463
    },
    {
      "epoch": 3.232097625732422e-05,
      "model_forward_time": 1.726996898651123,
      "step": 33891
    },
    {
      "epoch": 3.232097625732422e-05,
      "step": 33891,
      "training_step_time": 7.046026706695557
    },
    {
      "epoch": 3.232192993164063e-05,
      "model_forward_time": 1.7473833560943604,
      "step": 33892
    },
    {
      "epoch": 3.232192993164063e-05,
      "step": 33892,
      "training_step_time": 7.027796030044556
    },
    {
      "epoch": 3.232288360595703e-05,
      "model_forward_time": 1.7318542003631592,
      "step": 33893
    },
    {
      "epoch": 3.232288360595703e-05,
      "step": 33893,
      "training_step_time": 7.713130950927734
    },
    {
      "epoch": 3.2323837280273436e-05,
      "model_forward_time": 1.7302074432373047,
      "step": 33894
    },
    {
      "epoch": 3.2323837280273436e-05,
      "step": 33894,
      "training_step_time": 7.764875173568726
    },
    {
      "epoch": 3.232479095458985e-05,
      "model_forward_time": 1.726407527923584,
      "step": 33895
    },
    {
      "epoch": 3.232479095458985e-05,
      "step": 33895,
      "training_step_time": 8.540337085723877
    },
    {
      "epoch": 3.232574462890625e-05,
      "model_forward_time": 1.7329001426696777,
      "step": 33896
    },
    {
      "epoch": 3.232574462890625e-05,
      "step": 33896,
      "training_step_time": 8.598160982131958
    },
    {
      "epoch": 3.2326698303222655e-05,
      "model_forward_time": 2.332103967666626,
      "step": 33897
    },
    {
      "epoch": 3.2326698303222655e-05,
      "step": 33897,
      "training_step_time": 10.065434455871582
    },
    {
      "epoch": 3.2327651977539066e-05,
      "model_forward_time": 1.73006272315979,
      "step": 33898
    },
    {
      "epoch": 3.2327651977539066e-05,
      "step": 33898,
      "training_step_time": 7.953518390655518
    },
    {
      "epoch": 3.232860565185547e-05,
      "model_forward_time": 1.7349636554718018,
      "step": 33899
    },
    {
      "epoch": 3.232860565185547e-05,
      "step": 33899,
      "training_step_time": 7.5801849365234375
    },
    {
      "epoch": 3.2329559326171874e-05,
      "grad_norm": 0.10144264996051788,
      "learning_rate": 7.885298341845803e-06,
      "loss": 0.0604,
      "step": 33900
    },
    {
      "epoch": 3.2329559326171874e-05,
      "model_forward_time": 1.728583812713623,
      "step": 33900
    },
    {
      "dynamics_loss_avg": 0.054878973960876466,
      "epoch": 3.2329559326171874e-05,
      "step": 33900
    },
    {
      "action_loss_avg": 0.002649339404888451,
      "epoch": 3.2329559326171874e-05,
      "step": 33900
    },
    {
      "epoch": 3.2329559326171874e-05,
      "step": 33900,
      "training_step_time": 6.994754314422607
    },
    {
      "epoch": 3.2330513000488284e-05,
      "model_forward_time": 1.717663288116455,
      "step": 33901
    },
    {
      "epoch": 3.2330513000488284e-05,
      "step": 33901,
      "training_step_time": 7.020864009857178
    },
    {
      "epoch": 3.233146667480469e-05,
      "model_forward_time": 1.7365696430206299,
      "step": 33902
    },
    {
      "epoch": 3.233146667480469e-05,
      "step": 33902,
      "training_step_time": 7.719517230987549
    },
    {
      "epoch": 3.233242034912109e-05,
      "model_forward_time": 1.7393238544464111,
      "step": 33903
    },
    {
      "epoch": 3.233242034912109e-05,
      "step": 33903,
      "training_step_time": 8.518478155136108
    },
    {
      "epoch": 3.23333740234375e-05,
      "model_forward_time": 1.7256865501403809,
      "step": 33904
    },
    {
      "epoch": 3.23333740234375e-05,
      "step": 33904,
      "training_step_time": 7.932510852813721
    },
    {
      "epoch": 3.233432769775391e-05,
      "model_forward_time": 1.7399585247039795,
      "step": 33905
    },
    {
      "epoch": 3.233432769775391e-05,
      "step": 33905,
      "training_step_time": 8.474998474121094
    },
    {
      "epoch": 3.233528137207031e-05,
      "model_forward_time": 2.309699773788452,
      "step": 33906
    },
    {
      "epoch": 3.233528137207031e-05,
      "step": 33906,
      "training_step_time": 8.662683010101318
    },
    {
      "epoch": 3.2336235046386715e-05,
      "model_forward_time": 1.7250192165374756,
      "step": 33907
    },
    {
      "epoch": 3.2336235046386715e-05,
      "step": 33907,
      "training_step_time": 8.581831455230713
    },
    {
      "epoch": 3.2337188720703126e-05,
      "model_forward_time": 1.7268304824829102,
      "step": 33908
    },
    {
      "epoch": 3.2337188720703126e-05,
      "step": 33908,
      "training_step_time": 7.829249382019043
    },
    {
      "epoch": 3.233814239501953e-05,
      "model_forward_time": 1.7202503681182861,
      "step": 33909
    },
    {
      "epoch": 3.233814239501953e-05,
      "step": 33909,
      "training_step_time": 7.578639984130859
    },
    {
      "epoch": 3.2339096069335934e-05,
      "grad_norm": 0.13162511587142944,
      "learning_rate": 7.883947792015312e-06,
      "loss": 0.0614,
      "step": 33910
    },
    {
      "epoch": 3.2339096069335934e-05,
      "model_forward_time": 1.7301018238067627,
      "step": 33910
    },
    {
      "dynamics_loss_avg": 0.049361109174788,
      "epoch": 3.2339096069335934e-05,
      "step": 33910
    },
    {
      "action_loss_avg": 0.0017267929099034518,
      "epoch": 3.2339096069335934e-05,
      "step": 33910
    },
    {
      "epoch": 3.2339096069335934e-05,
      "step": 33910,
      "training_step_time": 7.085769176483154
    },
    {
      "epoch": 3.2340049743652345e-05,
      "model_forward_time": 1.728360652923584,
      "step": 33911
    },
    {
      "epoch": 3.2340049743652345e-05,
      "step": 33911,
      "training_step_time": 7.484124422073364
    },
    {
      "epoch": 3.234100341796875e-05,
      "model_forward_time": 1.729158878326416,
      "step": 33912
    },
    {
      "epoch": 3.234100341796875e-05,
      "step": 33912,
      "training_step_time": 7.895475625991821
    },
    {
      "epoch": 3.234195709228515e-05,
      "model_forward_time": 1.718186378479004,
      "step": 33913
    },
    {
      "epoch": 3.234195709228515e-05,
      "step": 33913,
      "training_step_time": 7.741008758544922
    },
    {
      "epoch": 3.2342910766601564e-05,
      "model_forward_time": 1.7195336818695068,
      "step": 33914
    },
    {
      "epoch": 3.2342910766601564e-05,
      "step": 33914,
      "training_step_time": 7.8469438552856445
    },
    {
      "epoch": 3.234386444091797e-05,
      "model_forward_time": 2.2997851371765137,
      "step": 33915
    },
    {
      "epoch": 3.234386444091797e-05,
      "step": 33915,
      "training_step_time": 8.697188138961792
    },
    {
      "epoch": 3.234481811523437e-05,
      "model_forward_time": 1.7646443843841553,
      "step": 33916
    },
    {
      "epoch": 3.234481811523437e-05,
      "step": 33916,
      "training_step_time": 8.087096452713013
    },
    {
      "epoch": 3.234577178955078e-05,
      "model_forward_time": 1.7150928974151611,
      "step": 33917
    },
    {
      "epoch": 3.234577178955078e-05,
      "step": 33917,
      "training_step_time": 7.870265483856201
    },
    {
      "epoch": 3.234672546386719e-05,
      "model_forward_time": 1.7434587478637695,
      "step": 33918
    },
    {
      "epoch": 3.234672546386719e-05,
      "step": 33918,
      "training_step_time": 7.602993488311768
    },
    {
      "epoch": 3.234767913818359e-05,
      "model_forward_time": 1.7480030059814453,
      "step": 33919
    },
    {
      "epoch": 3.234767913818359e-05,
      "step": 33919,
      "training_step_time": 7.101165533065796
    },
    {
      "epoch": 3.23486328125e-05,
      "grad_norm": 0.1791854202747345,
      "learning_rate": 7.882596926800627e-06,
      "loss": 0.0597,
      "step": 33920
    },
    {
      "epoch": 3.23486328125e-05,
      "model_forward_time": 1.7273812294006348,
      "step": 33920
    },
    {
      "dynamics_loss_avg": 0.05264600161463022,
      "epoch": 3.23486328125e-05,
      "step": 33920
    },
    {
      "action_loss_avg": 0.0021053142350865527,
      "epoch": 3.23486328125e-05,
      "step": 33920
    },
    {
      "epoch": 3.23486328125e-05,
      "step": 33920,
      "training_step_time": 7.4897682666778564
    },
    {
      "epoch": 3.2349586486816406e-05,
      "model_forward_time": 1.7443079948425293,
      "step": 33921
    },
    {
      "epoch": 3.2349586486816406e-05,
      "step": 33921,
      "training_step_time": 7.913997411727905
    },
    {
      "epoch": 3.235054016113281e-05,
      "model_forward_time": 1.7350437641143799,
      "step": 33922
    },
    {
      "epoch": 3.235054016113281e-05,
      "step": 33922,
      "training_step_time": 7.61088752746582
    },
    {
      "epoch": 3.235149383544922e-05,
      "model_forward_time": 1.7418856620788574,
      "step": 33923
    },
    {
      "epoch": 3.235149383544922e-05,
      "step": 33923,
      "training_step_time": 9.320659399032593
    },
    {
      "epoch": 3.2352447509765625e-05,
      "model_forward_time": 2.275346040725708,
      "step": 33924
    },
    {
      "epoch": 3.2352447509765625e-05,
      "step": 33924,
      "training_step_time": 11.136576890945435
    },
    {
      "epoch": 3.235340118408203e-05,
      "model_forward_time": 1.742255449295044,
      "step": 33925
    },
    {
      "epoch": 3.235340118408203e-05,
      "step": 33925,
      "training_step_time": 7.847186803817749
    },
    {
      "epoch": 3.235435485839844e-05,
      "model_forward_time": 1.369330883026123,
      "step": 33926
    },
    {
      "epoch": 3.235435485839844e-05,
      "step": 33926,
      "training_step_time": 8.14635682106018
    },
    {
      "epoch": 3.235530853271484e-05,
      "model_forward_time": 1.7402050495147705,
      "step": 33927
    },
    {
      "epoch": 3.235530853271484e-05,
      "step": 33927,
      "training_step_time": 7.6828086376190186
    },
    {
      "epoch": 3.235626220703125e-05,
      "model_forward_time": 1.7477107048034668,
      "step": 33928
    },
    {
      "epoch": 3.235626220703125e-05,
      "step": 33928,
      "training_step_time": 7.117959499359131
    },
    {
      "epoch": 3.235721588134766e-05,
      "model_forward_time": 1.7462687492370605,
      "step": 33929
    },
    {
      "epoch": 3.235721588134766e-05,
      "step": 33929,
      "training_step_time": 7.518862724304199
    },
    {
      "epoch": 3.235816955566406e-05,
      "grad_norm": 0.17100580036640167,
      "learning_rate": 7.881245746349478e-06,
      "loss": 0.0556,
      "step": 33930
    },
    {
      "epoch": 3.235816955566406e-05,
      "model_forward_time": 1.7403223514556885,
      "step": 33930
    },
    {
      "dynamics_loss_avg": 0.049722098372876646,
      "epoch": 3.235816955566406e-05,
      "step": 33930
    },
    {
      "action_loss_avg": 0.0017089488974306732,
      "epoch": 3.235816955566406e-05,
      "step": 33930
    },
    {
      "epoch": 3.235816955566406e-05,
      "step": 33930,
      "training_step_time": 7.886317729949951
    },
    {
      "epoch": 3.2359123229980466e-05,
      "model_forward_time": 1.7134649753570557,
      "step": 33931
    },
    {
      "epoch": 3.2359123229980466e-05,
      "step": 33931,
      "training_step_time": 7.709091901779175
    },
    {
      "epoch": 3.236007690429688e-05,
      "model_forward_time": 1.73014235496521,
      "step": 33932
    },
    {
      "epoch": 3.236007690429688e-05,
      "step": 33932,
      "training_step_time": 9.35923433303833
    },
    {
      "epoch": 3.236103057861328e-05,
      "model_forward_time": 2.2757186889648438,
      "step": 33933
    },
    {
      "epoch": 3.236103057861328e-05,
      "step": 33933,
      "training_step_time": 9.758894205093384
    },
    {
      "epoch": 3.2361984252929685e-05,
      "model_forward_time": 1.732412338256836,
      "step": 33934
    },
    {
      "epoch": 3.2361984252929685e-05,
      "step": 33934,
      "training_step_time": 9.418835639953613
    },
    {
      "epoch": 3.2362937927246096e-05,
      "model_forward_time": 1.7265095710754395,
      "step": 33935
    },
    {
      "epoch": 3.2362937927246096e-05,
      "step": 33935,
      "training_step_time": 7.949105978012085
    },
    {
      "epoch": 3.23638916015625e-05,
      "model_forward_time": 1.7282302379608154,
      "step": 33936
    },
    {
      "epoch": 3.23638916015625e-05,
      "step": 33936,
      "training_step_time": 7.640022277832031
    },
    {
      "epoch": 3.2364845275878904e-05,
      "model_forward_time": 1.361457347869873,
      "step": 33937
    },
    {
      "epoch": 3.2364845275878904e-05,
      "step": 33937,
      "training_step_time": 7.042440176010132
    },
    {
      "epoch": 3.2365798950195315e-05,
      "model_forward_time": 1.7309894561767578,
      "step": 33938
    },
    {
      "epoch": 3.2365798950195315e-05,
      "step": 33938,
      "training_step_time": 7.4420692920684814
    },
    {
      "epoch": 3.236675262451172e-05,
      "model_forward_time": 1.7171282768249512,
      "step": 33939
    },
    {
      "epoch": 3.236675262451172e-05,
      "step": 33939,
      "training_step_time": 7.84512186050415
    },
    {
      "epoch": 3.236770629882812e-05,
      "grad_norm": 0.1184757873415947,
      "learning_rate": 7.879894250809623e-06,
      "loss": 0.0553,
      "step": 33940
    },
    {
      "epoch": 3.236770629882812e-05,
      "model_forward_time": 1.7128100395202637,
      "step": 33940
    },
    {
      "dynamics_loss_avg": 0.051116297394037245,
      "epoch": 3.236770629882812e-05,
      "step": 33940
    },
    {
      "action_loss_avg": 0.0022062715812353416,
      "epoch": 3.236770629882812e-05,
      "step": 33940
    },
    {
      "epoch": 3.236770629882812e-05,
      "step": 33940,
      "training_step_time": 7.598602056503296
    },
    {
      "epoch": 3.2368659973144534e-05,
      "model_forward_time": 1.0102438926696777,
      "step": 33941
    },
    {
      "epoch": 3.2368659973144534e-05,
      "step": 33941,
      "training_step_time": 8.490572452545166
    },
    {
      "epoch": 3.236961364746094e-05,
      "model_forward_time": 2.3113210201263428,
      "step": 33942
    },
    {
      "epoch": 3.236961364746094e-05,
      "step": 33942,
      "training_step_time": 7.824225187301636
    },
    {
      "epoch": 3.237056732177734e-05,
      "model_forward_time": 1.787708044052124,
      "step": 33943
    },
    {
      "epoch": 3.237056732177734e-05,
      "step": 33943,
      "training_step_time": 9.127242803573608
    },
    {
      "epoch": 3.237152099609375e-05,
      "model_forward_time": 1.7285289764404297,
      "step": 33944
    },
    {
      "epoch": 3.237152099609375e-05,
      "step": 33944,
      "training_step_time": 8.824237585067749
    },
    {
      "epoch": 3.2372474670410157e-05,
      "model_forward_time": 1.7344889640808105,
      "step": 33945
    },
    {
      "epoch": 3.2372474670410157e-05,
      "step": 33945,
      "training_step_time": 8.806201934814453
    },
    {
      "epoch": 3.237342834472656e-05,
      "model_forward_time": 1.7210991382598877,
      "step": 33946
    },
    {
      "epoch": 3.237342834472656e-05,
      "step": 33946,
      "training_step_time": 7.242083787918091
    },
    {
      "epoch": 3.237438201904297e-05,
      "model_forward_time": 1.738398551940918,
      "step": 33947
    },
    {
      "epoch": 3.237438201904297e-05,
      "step": 33947,
      "training_step_time": 7.452741861343384
    },
    {
      "epoch": 3.2375335693359375e-05,
      "model_forward_time": 1.738344430923462,
      "step": 33948
    },
    {
      "epoch": 3.2375335693359375e-05,
      "step": 33948,
      "training_step_time": 7.867464303970337
    },
    {
      "epoch": 3.237628936767578e-05,
      "model_forward_time": 1.7304773330688477,
      "step": 33949
    },
    {
      "epoch": 3.237628936767578e-05,
      "step": 33949,
      "training_step_time": 7.893313407897949
    },
    {
      "epoch": 3.237724304199219e-05,
      "grad_norm": 0.11200837790966034,
      "learning_rate": 7.878542440328865e-06,
      "loss": 0.062,
      "step": 33950
    },
    {
      "epoch": 3.237724304199219e-05,
      "model_forward_time": 1.7387397289276123,
      "step": 33950
    },
    {
      "dynamics_loss_avg": 0.06884048040956259,
      "epoch": 3.237724304199219e-05,
      "step": 33950
    },
    {
      "action_loss_avg": 0.001978410017909482,
      "epoch": 3.237724304199219e-05,
      "step": 33950
    },
    {
      "epoch": 3.237724304199219e-05,
      "step": 33950,
      "training_step_time": 8.634596586227417
    },
    {
      "epoch": 3.2378196716308594e-05,
      "model_forward_time": 2.2856836318969727,
      "step": 33951
    },
    {
      "epoch": 3.2378196716308594e-05,
      "step": 33951,
      "training_step_time": 8.750288486480713
    },
    {
      "epoch": 3.2379150390625e-05,
      "model_forward_time": 1.7806625366210938,
      "step": 33952
    },
    {
      "epoch": 3.2379150390625e-05,
      "step": 33952,
      "training_step_time": 7.922827243804932
    },
    {
      "epoch": 3.238010406494141e-05,
      "model_forward_time": 1.7322688102722168,
      "step": 33953
    },
    {
      "epoch": 3.238010406494141e-05,
      "step": 33953,
      "training_step_time": 9.09680986404419
    },
    {
      "epoch": 3.238105773925781e-05,
      "model_forward_time": 1.723513126373291,
      "step": 33954
    },
    {
      "epoch": 3.238105773925781e-05,
      "step": 33954,
      "training_step_time": 8.138408422470093
    },
    {
      "epoch": 3.238201141357422e-05,
      "model_forward_time": 1.7268047332763672,
      "step": 33955
    },
    {
      "epoch": 3.238201141357422e-05,
      "step": 33955,
      "training_step_time": 7.54762077331543
    },
    {
      "epoch": 3.238296508789063e-05,
      "model_forward_time": 1.7250759601593018,
      "step": 33956
    },
    {
      "epoch": 3.238296508789063e-05,
      "step": 33956,
      "training_step_time": 7.431630611419678
    },
    {
      "epoch": 3.238391876220703e-05,
      "model_forward_time": 1.70572829246521,
      "step": 33957
    },
    {
      "epoch": 3.238391876220703e-05,
      "step": 33957,
      "training_step_time": 7.837270975112915
    },
    {
      "epoch": 3.2384872436523436e-05,
      "model_forward_time": 1.7618913650512695,
      "step": 33958
    },
    {
      "epoch": 3.2384872436523436e-05,
      "step": 33958,
      "training_step_time": 7.807560920715332
    },
    {
      "epoch": 3.238582611083985e-05,
      "model_forward_time": 1.7428491115570068,
      "step": 33959
    },
    {
      "epoch": 3.238582611083985e-05,
      "step": 33959,
      "training_step_time": 7.542581081390381
    },
    {
      "epoch": 3.238677978515625e-05,
      "grad_norm": 0.17132534086704254,
      "learning_rate": 7.877190315055032e-06,
      "loss": 0.0588,
      "step": 33960
    },
    {
      "epoch": 3.238677978515625e-05,
      "model_forward_time": 1.7412188053131104,
      "step": 33960
    },
    {
      "dynamics_loss_avg": 0.05839694067835808,
      "epoch": 3.238677978515625e-05,
      "step": 33960
    },
    {
      "action_loss_avg": 0.0014004546450451017,
      "epoch": 3.238677978515625e-05,
      "step": 33960
    },
    {
      "epoch": 3.238677978515625e-05,
      "step": 33960,
      "training_step_time": 10.33931040763855
    },
    {
      "epoch": 3.2387733459472655e-05,
      "model_forward_time": 2.2985944747924805,
      "step": 33961
    },
    {
      "epoch": 3.2387733459472655e-05,
      "step": 33961,
      "training_step_time": 8.992504596710205
    },
    {
      "epoch": 3.2388687133789066e-05,
      "model_forward_time": 1.7435085773468018,
      "step": 33962
    },
    {
      "epoch": 3.2388687133789066e-05,
      "step": 33962,
      "training_step_time": 8.632347822189331
    },
    {
      "epoch": 3.238964080810547e-05,
      "model_forward_time": 1.7516465187072754,
      "step": 33963
    },
    {
      "epoch": 3.238964080810547e-05,
      "step": 33963,
      "training_step_time": 7.995677471160889
    },
    {
      "epoch": 3.2390594482421874e-05,
      "model_forward_time": 1.753354787826538,
      "step": 33964
    },
    {
      "epoch": 3.2390594482421874e-05,
      "step": 33964,
      "training_step_time": 7.727528810501099
    },
    {
      "epoch": 3.2391548156738285e-05,
      "model_forward_time": 1.7196130752563477,
      "step": 33965
    },
    {
      "epoch": 3.2391548156738285e-05,
      "step": 33965,
      "training_step_time": 7.43717360496521
    },
    {
      "epoch": 3.239250183105469e-05,
      "model_forward_time": 1.735727071762085,
      "step": 33966
    },
    {
      "epoch": 3.239250183105469e-05,
      "step": 33966,
      "training_step_time": 8.279395341873169
    },
    {
      "epoch": 3.239345550537109e-05,
      "model_forward_time": 1.726409912109375,
      "step": 33967
    },
    {
      "epoch": 3.239345550537109e-05,
      "step": 33967,
      "training_step_time": 7.79180121421814
    },
    {
      "epoch": 3.23944091796875e-05,
      "model_forward_time": 1.7264606952667236,
      "step": 33968
    },
    {
      "epoch": 3.23944091796875e-05,
      "step": 33968,
      "training_step_time": 7.46228289604187
    },
    {
      "epoch": 3.239536285400391e-05,
      "model_forward_time": 1.7401082515716553,
      "step": 33969
    },
    {
      "epoch": 3.239536285400391e-05,
      "step": 33969,
      "training_step_time": 10.269912481307983
    },
    {
      "epoch": 3.239631652832031e-05,
      "grad_norm": 0.1374233216047287,
      "learning_rate": 7.875837875135992e-06,
      "loss": 0.0601,
      "step": 33970
    },
    {
      "epoch": 3.239631652832031e-05,
      "model_forward_time": 2.294182062149048,
      "step": 33970
    },
    {
      "dynamics_loss_avg": 0.07571053337305785,
      "epoch": 3.239631652832031e-05,
      "step": 33970
    },
    {
      "action_loss_avg": 0.0035662311071064322,
      "epoch": 3.239631652832031e-05,
      "step": 33970
    },
    {
      "epoch": 3.239631652832031e-05,
      "step": 33970,
      "training_step_time": 8.892899513244629
    },
    {
      "epoch": 3.2397270202636715e-05,
      "model_forward_time": 1.7462389469146729,
      "step": 33971
    },
    {
      "epoch": 3.2397270202636715e-05,
      "step": 33971,
      "training_step_time": 7.800797700881958
    },
    {
      "epoch": 3.2398223876953126e-05,
      "model_forward_time": 1.755561351776123,
      "step": 33972
    },
    {
      "epoch": 3.2398223876953126e-05,
      "step": 33972,
      "training_step_time": 9.602552652359009
    },
    {
      "epoch": 3.239917755126953e-05,
      "model_forward_time": 1.7494137287139893,
      "step": 33973
    },
    {
      "epoch": 3.239917755126953e-05,
      "step": 33973,
      "training_step_time": 7.700139284133911
    },
    {
      "epoch": 3.2400131225585934e-05,
      "model_forward_time": 1.7465510368347168,
      "step": 33974
    },
    {
      "epoch": 3.2400131225585934e-05,
      "step": 33974,
      "training_step_time": 7.690044403076172
    },
    {
      "epoch": 3.2401084899902345e-05,
      "model_forward_time": 1.7249140739440918,
      "step": 33975
    },
    {
      "epoch": 3.2401084899902345e-05,
      "step": 33975,
      "training_step_time": 7.862411975860596
    },
    {
      "epoch": 3.240203857421875e-05,
      "model_forward_time": 1.7171125411987305,
      "step": 33976
    },
    {
      "epoch": 3.240203857421875e-05,
      "step": 33976,
      "training_step_time": 7.795267581939697
    },
    {
      "epoch": 3.240299224853515e-05,
      "model_forward_time": 1.7309277057647705,
      "step": 33977
    },
    {
      "epoch": 3.240299224853515e-05,
      "step": 33977,
      "training_step_time": 7.447731256484985
    },
    {
      "epoch": 3.2403945922851564e-05,
      "model_forward_time": 1.7266161441802979,
      "step": 33978
    },
    {
      "epoch": 3.2403945922851564e-05,
      "step": 33978,
      "training_step_time": 7.793216705322266
    },
    {
      "epoch": 3.240489959716797e-05,
      "model_forward_time": 2.2817656993865967,
      "step": 33979
    },
    {
      "epoch": 3.240489959716797e-05,
      "step": 33979,
      "training_step_time": 8.795856475830078
    },
    {
      "epoch": 3.240585327148437e-05,
      "grad_norm": 0.13052427768707275,
      "learning_rate": 7.874485120719647e-06,
      "loss": 0.0606,
      "step": 33980
    },
    {
      "epoch": 3.240585327148437e-05,
      "model_forward_time": 1.7318265438079834,
      "step": 33980
    },
    {
      "dynamics_loss_avg": 0.06043358212336898,
      "epoch": 3.240585327148437e-05,
      "step": 33980
    },
    {
      "action_loss_avg": 0.0019940045080147684,
      "epoch": 3.240585327148437e-05,
      "step": 33980
    },
    {
      "epoch": 3.240585327148437e-05,
      "step": 33980,
      "training_step_time": 7.855871677398682
    },
    {
      "epoch": 3.240680694580078e-05,
      "model_forward_time": 1.7345032691955566,
      "step": 33981
    },
    {
      "epoch": 3.240680694580078e-05,
      "step": 33981,
      "training_step_time": 8.810249090194702
    },
    {
      "epoch": 3.240776062011719e-05,
      "model_forward_time": 1.7240045070648193,
      "step": 33982
    },
    {
      "epoch": 3.240776062011719e-05,
      "step": 33982,
      "training_step_time": 8.015240907669067
    },
    {
      "epoch": 3.240871429443359e-05,
      "model_forward_time": 1.7218668460845947,
      "step": 33983
    },
    {
      "epoch": 3.240871429443359e-05,
      "step": 33983,
      "training_step_time": 7.703206300735474
    },
    {
      "epoch": 3.240966796875e-05,
      "model_forward_time": 1.7398009300231934,
      "step": 33984
    },
    {
      "epoch": 3.240966796875e-05,
      "step": 33984,
      "training_step_time": 7.814599514007568
    },
    {
      "epoch": 3.2410621643066406e-05,
      "model_forward_time": 1.776303768157959,
      "step": 33985
    },
    {
      "epoch": 3.2410621643066406e-05,
      "step": 33985,
      "training_step_time": 7.625175714492798
    },
    {
      "epoch": 3.241157531738281e-05,
      "model_forward_time": 1.744133710861206,
      "step": 33986
    },
    {
      "epoch": 3.241157531738281e-05,
      "step": 33986,
      "training_step_time": 7.796396017074585
    },
    {
      "epoch": 3.241252899169922e-05,
      "model_forward_time": 1.7361016273498535,
      "step": 33987
    },
    {
      "epoch": 3.241252899169922e-05,
      "step": 33987,
      "training_step_time": 8.599225521087646
    },
    {
      "epoch": 3.2413482666015625e-05,
      "model_forward_time": 2.3566460609436035,
      "step": 33988
    },
    {
      "epoch": 3.2413482666015625e-05,
      "step": 33988,
      "training_step_time": 9.091732501983643
    },
    {
      "epoch": 3.241443634033203e-05,
      "model_forward_time": 1.7387959957122803,
      "step": 33989
    },
    {
      "epoch": 3.241443634033203e-05,
      "step": 33989,
      "training_step_time": 9.44232988357544
    },
    {
      "epoch": 3.241539001464844e-05,
      "grad_norm": 0.1127745509147644,
      "learning_rate": 7.87313205195393e-06,
      "loss": 0.0569,
      "step": 33990
    },
    {
      "epoch": 3.241539001464844e-05,
      "model_forward_time": 1.7236628532409668,
      "step": 33990
    },
    {
      "dynamics_loss_avg": 0.04255216848105192,
      "epoch": 3.241539001464844e-05,
      "step": 33990
    },
    {
      "action_loss_avg": 0.00221481112530455,
      "epoch": 3.241539001464844e-05,
      "step": 33990
    },
    {
      "epoch": 3.241539001464844e-05,
      "step": 33990,
      "training_step_time": 7.838661193847656
    },
    {
      "epoch": 3.2416343688964843e-05,
      "model_forward_time": 1.725740909576416,
      "step": 33991
    },
    {
      "epoch": 3.2416343688964843e-05,
      "step": 33991,
      "training_step_time": 8.763604879379272
    },
    {
      "epoch": 3.241729736328125e-05,
      "model_forward_time": 1.7432465553283691,
      "step": 33992
    },
    {
      "epoch": 3.241729736328125e-05,
      "step": 33992,
      "training_step_time": 7.610108375549316
    },
    {
      "epoch": 3.241825103759766e-05,
      "model_forward_time": 1.3674027919769287,
      "step": 33993
    },
    {
      "epoch": 3.241825103759766e-05,
      "step": 33993,
      "training_step_time": 7.931711196899414
    },
    {
      "epoch": 3.241920471191406e-05,
      "model_forward_time": 1.7347660064697266,
      "step": 33994
    },
    {
      "epoch": 3.241920471191406e-05,
      "step": 33994,
      "training_step_time": 7.706679105758667
    },
    {
      "epoch": 3.2420158386230466e-05,
      "model_forward_time": 1.7444956302642822,
      "step": 33995
    },
    {
      "epoch": 3.2420158386230466e-05,
      "step": 33995,
      "training_step_time": 7.882893323898315
    },
    {
      "epoch": 3.242111206054688e-05,
      "model_forward_time": 1.733980417251587,
      "step": 33996
    },
    {
      "epoch": 3.242111206054688e-05,
      "step": 33996,
      "training_step_time": 7.814537286758423
    },
    {
      "epoch": 3.242206573486328e-05,
      "model_forward_time": 1.7582967281341553,
      "step": 33997
    },
    {
      "epoch": 3.242206573486328e-05,
      "step": 33997,
      "training_step_time": 10.200220584869385
    },
    {
      "epoch": 3.2423019409179685e-05,
      "model_forward_time": 1.7877418994903564,
      "step": 33998
    },
    {
      "epoch": 3.2423019409179685e-05,
      "step": 33998,
      "training_step_time": 9.050285577774048
    },
    {
      "epoch": 3.2423973083496096e-05,
      "model_forward_time": 1.7345044612884521,
      "step": 33999
    },
    {
      "epoch": 3.2423973083496096e-05,
      "step": 33999,
      "training_step_time": 8.334314107894897
    },
    {
      "epoch": 3.24249267578125e-05,
      "grad_norm": 0.12540894746780396,
      "learning_rate": 7.871778668986811e-06,
      "loss": 0.057,
      "step": 34000
    },
    {
      "epoch": 3.24249267578125e-05,
      "model_forward_time": 1.7422757148742676,
      "step": 34000
    },
    {
      "dynamics_loss_avg": 0.04701029621064663,
      "epoch": 3.24249267578125e-05,
      "step": 34000
    },
    {
      "action_loss_avg": 0.0013352581241633742,
      "epoch": 3.24249267578125e-05,
      "step": 34000
    },
    {
      "epoch": 3.24249267578125e-05,
      "step": 34000,
      "training_step_time": 7.158113956451416
    },
    {
      "epoch": 3.2425880432128904e-05,
      "model_forward_time": 1.727231502532959,
      "step": 34001
    },
    {
      "epoch": 3.2425880432128904e-05,
      "step": 34001,
      "training_step_time": 8.043171167373657
    },
    {
      "epoch": 3.2426834106445315e-05,
      "model_forward_time": 1.7296404838562012,
      "step": 34002
    },
    {
      "epoch": 3.2426834106445315e-05,
      "step": 34002,
      "training_step_time": 7.827430486679077
    },
    {
      "epoch": 3.242778778076172e-05,
      "model_forward_time": 1.7313508987426758,
      "step": 34003
    },
    {
      "epoch": 3.242778778076172e-05,
      "step": 34003,
      "training_step_time": 7.567063570022583
    },
    {
      "epoch": 3.242874145507812e-05,
      "model_forward_time": 1.3415539264678955,
      "step": 34004
    },
    {
      "epoch": 3.242874145507812e-05,
      "step": 34004,
      "training_step_time": 8.662092208862305
    },
    {
      "epoch": 3.2429695129394534e-05,
      "model_forward_time": 1.7315495014190674,
      "step": 34005
    },
    {
      "epoch": 3.2429695129394534e-05,
      "step": 34005,
      "training_step_time": 9.371772527694702
    },
    {
      "epoch": 3.243064880371094e-05,
      "model_forward_time": 2.7254905700683594,
      "step": 34006
    },
    {
      "epoch": 3.243064880371094e-05,
      "step": 34006,
      "training_step_time": 13.395976543426514
    },
    {
      "epoch": 3.243160247802734e-05,
      "model_forward_time": 1.7681574821472168,
      "step": 34007
    },
    {
      "epoch": 3.243160247802734e-05,
      "step": 34007,
      "training_step_time": 9.7187819480896
    },
    {
      "epoch": 3.243255615234375e-05,
      "model_forward_time": 1.712320327758789,
      "step": 34008
    },
    {
      "epoch": 3.243255615234375e-05,
      "step": 34008,
      "training_step_time": 10.311541318893433
    },
    {
      "epoch": 3.2433509826660157e-05,
      "model_forward_time": 1.7169604301452637,
      "step": 34009
    },
    {
      "epoch": 3.2433509826660157e-05,
      "step": 34009,
      "training_step_time": 7.906516075134277
    },
    {
      "epoch": 3.243446350097656e-05,
      "grad_norm": 0.11503800749778748,
      "learning_rate": 7.870424971966294e-06,
      "loss": 0.0588,
      "step": 34010
    },
    {
      "epoch": 3.243446350097656e-05,
      "model_forward_time": 1.7162704467773438,
      "step": 34010
    },
    {
      "dynamics_loss_avg": 0.055510681867599485,
      "epoch": 3.243446350097656e-05,
      "step": 34010
    },
    {
      "action_loss_avg": 0.0013798102590953931,
      "epoch": 3.243446350097656e-05,
      "step": 34010
    },
    {
      "epoch": 3.243446350097656e-05,
      "step": 34010,
      "training_step_time": 7.684950351715088
    },
    {
      "epoch": 3.243541717529297e-05,
      "model_forward_time": 1.7139265537261963,
      "step": 34011
    },
    {
      "epoch": 3.243541717529297e-05,
      "step": 34011,
      "training_step_time": 7.761366367340088
    },
    {
      "epoch": 3.2436370849609375e-05,
      "model_forward_time": 1.7196850776672363,
      "step": 34012
    },
    {
      "epoch": 3.2436370849609375e-05,
      "step": 34012,
      "training_step_time": 7.687067747116089
    },
    {
      "epoch": 3.243732452392578e-05,
      "model_forward_time": 1.7215676307678223,
      "step": 34013
    },
    {
      "epoch": 3.243732452392578e-05,
      "step": 34013,
      "training_step_time": 7.887099504470825
    },
    {
      "epoch": 3.243827819824219e-05,
      "model_forward_time": 1.7117831707000732,
      "step": 34014
    },
    {
      "epoch": 3.243827819824219e-05,
      "step": 34014,
      "training_step_time": 7.678718566894531
    },
    {
      "epoch": 3.2439231872558594e-05,
      "model_forward_time": 1.7148332595825195,
      "step": 34015
    },
    {
      "epoch": 3.2439231872558594e-05,
      "step": 34015,
      "training_step_time": 8.820308685302734
    },
    {
      "epoch": 3.2440185546875e-05,
      "model_forward_time": 2.313626766204834,
      "step": 34016
    },
    {
      "epoch": 3.2440185546875e-05,
      "step": 34016,
      "training_step_time": 9.24553656578064
    },
    {
      "epoch": 3.244113922119141e-05,
      "model_forward_time": 1.7103989124298096,
      "step": 34017
    },
    {
      "epoch": 3.244113922119141e-05,
      "step": 34017,
      "training_step_time": 8.4407479763031
    },
    {
      "epoch": 3.244209289550781e-05,
      "model_forward_time": 1.724107027053833,
      "step": 34018
    },
    {
      "epoch": 3.244209289550781e-05,
      "step": 34018,
      "training_step_time": 7.778823137283325
    },
    {
      "epoch": 3.244304656982422e-05,
      "model_forward_time": 1.72548508644104,
      "step": 34019
    },
    {
      "epoch": 3.244304656982422e-05,
      "step": 34019,
      "training_step_time": 8.067626953125
    },
    {
      "epoch": 3.244400024414063e-05,
      "grad_norm": 0.1161946952342987,
      "learning_rate": 7.86907096104042e-06,
      "loss": 0.0635,
      "step": 34020
    },
    {
      "epoch": 3.244400024414063e-05,
      "model_forward_time": 1.725513219833374,
      "step": 34020
    },
    {
      "dynamics_loss_avg": 0.0632435292005539,
      "epoch": 3.244400024414063e-05,
      "step": 34020
    },
    {
      "action_loss_avg": 0.0026656820147763936,
      "epoch": 3.244400024414063e-05,
      "step": 34020
    },
    {
      "epoch": 3.244400024414063e-05,
      "step": 34020,
      "training_step_time": 7.785127401351929
    },
    {
      "epoch": 3.244495391845703e-05,
      "model_forward_time": 1.731696605682373,
      "step": 34021
    },
    {
      "epoch": 3.244495391845703e-05,
      "step": 34021,
      "training_step_time": 7.6063244342803955
    },
    {
      "epoch": 3.2445907592773436e-05,
      "model_forward_time": 1.7276244163513184,
      "step": 34022
    },
    {
      "epoch": 3.2445907592773436e-05,
      "step": 34022,
      "training_step_time": 7.477764844894409
    },
    {
      "epoch": 3.244686126708985e-05,
      "model_forward_time": 1.728853702545166,
      "step": 34023
    },
    {
      "epoch": 3.244686126708985e-05,
      "step": 34023,
      "training_step_time": 7.67446494102478
    },
    {
      "epoch": 3.244781494140625e-05,
      "model_forward_time": 1.730515718460083,
      "step": 34024
    },
    {
      "epoch": 3.244781494140625e-05,
      "step": 34024,
      "training_step_time": 7.918639659881592
    },
    {
      "epoch": 3.2448768615722655e-05,
      "model_forward_time": 2.2622101306915283,
      "step": 34025
    },
    {
      "epoch": 3.2448768615722655e-05,
      "step": 34025,
      "training_step_time": 7.848304510116577
    },
    {
      "epoch": 3.2449722290039066e-05,
      "model_forward_time": 1.711029291152954,
      "step": 34026
    },
    {
      "epoch": 3.2449722290039066e-05,
      "step": 34026,
      "training_step_time": 8.377009868621826
    },
    {
      "epoch": 3.245067596435547e-05,
      "model_forward_time": 1.7056264877319336,
      "step": 34027
    },
    {
      "epoch": 3.245067596435547e-05,
      "step": 34027,
      "training_step_time": 8.761894226074219
    },
    {
      "epoch": 3.2451629638671874e-05,
      "model_forward_time": 1.7143292427062988,
      "step": 34028
    },
    {
      "epoch": 3.2451629638671874e-05,
      "step": 34028,
      "training_step_time": 8.602511167526245
    },
    {
      "epoch": 3.2452583312988285e-05,
      "model_forward_time": 1.7163207530975342,
      "step": 34029
    },
    {
      "epoch": 3.2452583312988285e-05,
      "step": 34029,
      "training_step_time": 8.723485946655273
    },
    {
      "epoch": 3.245353698730469e-05,
      "grad_norm": 0.20843464136123657,
      "learning_rate": 7.867716636357257e-06,
      "loss": 0.0625,
      "step": 34030
    },
    {
      "epoch": 3.245353698730469e-05,
      "model_forward_time": 1.7144725322723389,
      "step": 34030
    },
    {
      "dynamics_loss_avg": 0.05707617346197367,
      "epoch": 3.245353698730469e-05,
      "step": 34030
    },
    {
      "action_loss_avg": 0.0012725547654554247,
      "epoch": 3.245353698730469e-05,
      "step": 34030
    },
    {
      "epoch": 3.245353698730469e-05,
      "step": 34030,
      "training_step_time": 7.429126262664795
    },
    {
      "epoch": 3.245449066162109e-05,
      "model_forward_time": 1.7157814502716064,
      "step": 34031
    },
    {
      "epoch": 3.245449066162109e-05,
      "step": 34031,
      "training_step_time": 7.46094274520874
    },
    {
      "epoch": 3.24554443359375e-05,
      "model_forward_time": 1.7141003608703613,
      "step": 34032
    },
    {
      "epoch": 3.24554443359375e-05,
      "step": 34032,
      "training_step_time": 7.964991807937622
    },
    {
      "epoch": 3.245639801025391e-05,
      "model_forward_time": 1.7128443717956543,
      "step": 34033
    },
    {
      "epoch": 3.245639801025391e-05,
      "step": 34033,
      "training_step_time": 8.90262770652771
    },
    {
      "epoch": 3.245735168457031e-05,
      "model_forward_time": 2.259298086166382,
      "step": 34034
    },
    {
      "epoch": 3.245735168457031e-05,
      "step": 34034,
      "training_step_time": 8.911954879760742
    },
    {
      "epoch": 3.2458305358886715e-05,
      "model_forward_time": 1.7091546058654785,
      "step": 34035
    },
    {
      "epoch": 3.2458305358886715e-05,
      "step": 34035,
      "training_step_time": 8.42184066772461
    },
    {
      "epoch": 3.2459259033203126e-05,
      "model_forward_time": 1.7036314010620117,
      "step": 34036
    },
    {
      "epoch": 3.2459259033203126e-05,
      "step": 34036,
      "training_step_time": 7.787559986114502
    },
    {
      "epoch": 3.246021270751953e-05,
      "model_forward_time": 1.7081058025360107,
      "step": 34037
    },
    {
      "epoch": 3.246021270751953e-05,
      "step": 34037,
      "training_step_time": 8.918745517730713
    },
    {
      "epoch": 3.2461166381835934e-05,
      "model_forward_time": 1.710059404373169,
      "step": 34038
    },
    {
      "epoch": 3.2461166381835934e-05,
      "step": 34038,
      "training_step_time": 7.908811092376709
    },
    {
      "epoch": 3.2462120056152345e-05,
      "model_forward_time": 1.7071704864501953,
      "step": 34039
    },
    {
      "epoch": 3.2462120056152345e-05,
      "step": 34039,
      "training_step_time": 7.872172117233276
    },
    {
      "epoch": 3.246307373046875e-05,
      "grad_norm": 0.1318594515323639,
      "learning_rate": 7.866361998064915e-06,
      "loss": 0.0641,
      "step": 34040
    },
    {
      "epoch": 3.246307373046875e-05,
      "model_forward_time": 1.731870412826538,
      "step": 34040
    },
    {
      "dynamics_loss_avg": 0.05947867911309004,
      "epoch": 3.246307373046875e-05,
      "step": 34040
    },
    {
      "action_loss_avg": 0.0022066591191105545,
      "epoch": 3.246307373046875e-05,
      "step": 34040
    },
    {
      "epoch": 3.246307373046875e-05,
      "step": 34040,
      "training_step_time": 7.5002241134643555
    },
    {
      "epoch": 3.246402740478515e-05,
      "model_forward_time": 1.7077250480651855,
      "step": 34041
    },
    {
      "epoch": 3.246402740478515e-05,
      "step": 34041,
      "training_step_time": 7.929755926132202
    },
    {
      "epoch": 3.2464981079101564e-05,
      "model_forward_time": 1.7114458084106445,
      "step": 34042
    },
    {
      "epoch": 3.2464981079101564e-05,
      "step": 34042,
      "training_step_time": 9.969712495803833
    },
    {
      "epoch": 3.246593475341797e-05,
      "model_forward_time": 2.27059006690979,
      "step": 34043
    },
    {
      "epoch": 3.246593475341797e-05,
      "step": 34043,
      "training_step_time": 7.829115152359009
    },
    {
      "epoch": 3.246688842773437e-05,
      "model_forward_time": 1.7115650177001953,
      "step": 34044
    },
    {
      "epoch": 3.246688842773437e-05,
      "step": 34044,
      "training_step_time": 8.872722625732422
    },
    {
      "epoch": 3.246784210205078e-05,
      "model_forward_time": 1.7095534801483154,
      "step": 34045
    },
    {
      "epoch": 3.246784210205078e-05,
      "step": 34045,
      "training_step_time": 8.843589067459106
    },
    {
      "epoch": 3.246879577636719e-05,
      "model_forward_time": 1.714606761932373,
      "step": 34046
    },
    {
      "epoch": 3.246879577636719e-05,
      "step": 34046,
      "training_step_time": 8.07717490196228
    },
    {
      "epoch": 3.246974945068359e-05,
      "model_forward_time": 1.7088313102722168,
      "step": 34047
    },
    {
      "epoch": 3.246974945068359e-05,
      "step": 34047,
      "training_step_time": 7.8482420444488525
    },
    {
      "epoch": 3.2470703125e-05,
      "model_forward_time": 1.7145264148712158,
      "step": 34048
    },
    {
      "epoch": 3.2470703125e-05,
      "step": 34048,
      "training_step_time": 7.755829095840454
    },
    {
      "epoch": 3.2471656799316406e-05,
      "model_forward_time": 1.7314929962158203,
      "step": 34049
    },
    {
      "epoch": 3.2471656799316406e-05,
      "step": 34049,
      "training_step_time": 7.701322555541992
    },
    {
      "epoch": 3.247261047363281e-05,
      "grad_norm": 0.1514241248369217,
      "learning_rate": 7.865007046311534e-06,
      "loss": 0.0617,
      "step": 34050
    },
    {
      "epoch": 3.247261047363281e-05,
      "model_forward_time": 1.70589017868042,
      "step": 34050
    },
    {
      "dynamics_loss_avg": 0.06044425126165152,
      "epoch": 3.247261047363281e-05,
      "step": 34050
    },
    {
      "action_loss_avg": 0.001848635368514806,
      "epoch": 3.247261047363281e-05,
      "step": 34050
    },
    {
      "epoch": 3.247261047363281e-05,
      "step": 34050,
      "training_step_time": 8.17065715789795
    },
    {
      "epoch": 3.247356414794922e-05,
      "model_forward_time": 1.7147011756896973,
      "step": 34051
    },
    {
      "epoch": 3.247356414794922e-05,
      "step": 34051,
      "training_step_time": 9.093001127243042
    },
    {
      "epoch": 3.2474517822265625e-05,
      "model_forward_time": 2.2634997367858887,
      "step": 34052
    },
    {
      "epoch": 3.2474517822265625e-05,
      "step": 34052,
      "training_step_time": 8.780495405197144
    },
    {
      "epoch": 3.247547149658203e-05,
      "model_forward_time": 1.708012580871582,
      "step": 34053
    },
    {
      "epoch": 3.247547149658203e-05,
      "step": 34053,
      "training_step_time": 8.547308921813965
    },
    {
      "epoch": 3.247642517089844e-05,
      "model_forward_time": 1.7045831680297852,
      "step": 34054
    },
    {
      "epoch": 3.247642517089844e-05,
      "step": 34054,
      "training_step_time": 7.819812297821045
    },
    {
      "epoch": 3.2477378845214843e-05,
      "model_forward_time": 1.7132952213287354,
      "step": 34055
    },
    {
      "epoch": 3.2477378845214843e-05,
      "step": 34055,
      "training_step_time": 8.48805546760559
    },
    {
      "epoch": 3.247833251953125e-05,
      "model_forward_time": 1.712510585784912,
      "step": 34056
    },
    {
      "epoch": 3.247833251953125e-05,
      "step": 34056,
      "training_step_time": 7.672428607940674
    },
    {
      "epoch": 3.247928619384766e-05,
      "model_forward_time": 1.7069833278656006,
      "step": 34057
    },
    {
      "epoch": 3.247928619384766e-05,
      "step": 34057,
      "training_step_time": 7.675917387008667
    },
    {
      "epoch": 3.248023986816406e-05,
      "model_forward_time": 1.7373123168945312,
      "step": 34058
    },
    {
      "epoch": 3.248023986816406e-05,
      "step": 34058,
      "training_step_time": 7.65983772277832
    },
    {
      "epoch": 3.2481193542480466e-05,
      "model_forward_time": 1.7187378406524658,
      "step": 34059
    },
    {
      "epoch": 3.2481193542480466e-05,
      "step": 34059,
      "training_step_time": 7.565007448196411
    },
    {
      "epoch": 3.248214721679688e-05,
      "grad_norm": 0.12134090065956116,
      "learning_rate": 7.86365178124529e-06,
      "loss": 0.0606,
      "step": 34060
    },
    {
      "epoch": 3.248214721679688e-05,
      "model_forward_time": 1.7178220748901367,
      "step": 34060
    },
    {
      "dynamics_loss_avg": 0.05083796642720699,
      "epoch": 3.248214721679688e-05,
      "step": 34060
    },
    {
      "action_loss_avg": 0.001429424635716714,
      "epoch": 3.248214721679688e-05,
      "step": 34060
    },
    {
      "epoch": 3.248214721679688e-05,
      "step": 34060,
      "training_step_time": 7.731215476989746
    },
    {
      "epoch": 3.248310089111328e-05,
      "model_forward_time": 2.249450445175171,
      "step": 34061
    },
    {
      "epoch": 3.248310089111328e-05,
      "step": 34061,
      "training_step_time": 8.52137279510498
    },
    {
      "epoch": 3.2484054565429685e-05,
      "model_forward_time": 1.7285916805267334,
      "step": 34062
    },
    {
      "epoch": 3.2484054565429685e-05,
      "step": 34062,
      "training_step_time": 8.96671199798584
    },
    {
      "epoch": 3.2485008239746096e-05,
      "model_forward_time": 1.7030556201934814,
      "step": 34063
    },
    {
      "epoch": 3.2485008239746096e-05,
      "step": 34063,
      "training_step_time": 9.338151931762695
    },
    {
      "epoch": 3.24859619140625e-05,
      "model_forward_time": 1.6985342502593994,
      "step": 34064
    },
    {
      "epoch": 3.24859619140625e-05,
      "step": 34064,
      "training_step_time": 8.885002851486206
    },
    {
      "epoch": 3.2486915588378904e-05,
      "model_forward_time": 1.6980562210083008,
      "step": 34065
    },
    {
      "epoch": 3.2486915588378904e-05,
      "step": 34065,
      "training_step_time": 8.773424863815308
    },
    {
      "epoch": 3.2487869262695315e-05,
      "model_forward_time": 1.7034111022949219,
      "step": 34066
    },
    {
      "epoch": 3.2487869262695315e-05,
      "step": 34066,
      "training_step_time": 8.545479774475098
    },
    {
      "epoch": 3.248882293701172e-05,
      "model_forward_time": 1.7501108646392822,
      "step": 34067
    },
    {
      "epoch": 3.248882293701172e-05,
      "step": 34067,
      "training_step_time": 7.713494777679443
    },
    {
      "epoch": 3.248977661132812e-05,
      "model_forward_time": 1.7083873748779297,
      "step": 34068
    },
    {
      "epoch": 3.248977661132812e-05,
      "step": 34068,
      "training_step_time": 7.6457884311676025
    },
    {
      "epoch": 3.2490730285644534e-05,
      "model_forward_time": 1.707693338394165,
      "step": 34069
    },
    {
      "epoch": 3.2490730285644534e-05,
      "step": 34069,
      "training_step_time": 8.428990364074707
    },
    {
      "epoch": 3.249168395996094e-05,
      "grad_norm": 0.10538552701473236,
      "learning_rate": 7.862296203014394e-06,
      "loss": 0.061,
      "step": 34070
    },
    {
      "epoch": 3.249168395996094e-05,
      "model_forward_time": 2.2401771545410156,
      "step": 34070
    },
    {
      "dynamics_loss_avg": 0.046013441495597364,
      "epoch": 3.249168395996094e-05,
      "step": 34070
    },
    {
      "action_loss_avg": 0.0014897209650371224,
      "epoch": 3.249168395996094e-05,
      "step": 34070
    },
    {
      "epoch": 3.249168395996094e-05,
      "step": 34070,
      "training_step_time": 7.782025575637817
    },
    {
      "epoch": 3.249263763427734e-05,
      "model_forward_time": 1.71925950050354,
      "step": 34071
    },
    {
      "epoch": 3.249263763427734e-05,
      "step": 34071,
      "training_step_time": 11.446802854537964
    },
    {
      "epoch": 3.249359130859375e-05,
      "model_forward_time": 1.7009236812591553,
      "step": 34072
    },
    {
      "epoch": 3.249359130859375e-05,
      "step": 34072,
      "training_step_time": 7.860250473022461
    },
    {
      "epoch": 3.249454498291016e-05,
      "model_forward_time": 1.7073593139648438,
      "step": 34073
    },
    {
      "epoch": 3.249454498291016e-05,
      "step": 34073,
      "training_step_time": 7.859485626220703
    },
    {
      "epoch": 3.249549865722656e-05,
      "model_forward_time": 1.703418493270874,
      "step": 34074
    },
    {
      "epoch": 3.249549865722656e-05,
      "step": 34074,
      "training_step_time": 7.782623529434204
    },
    {
      "epoch": 3.249645233154297e-05,
      "model_forward_time": 1.7184545993804932,
      "step": 34075
    },
    {
      "epoch": 3.249645233154297e-05,
      "step": 34075,
      "training_step_time": 7.763632774353027
    },
    {
      "epoch": 3.2497406005859375e-05,
      "model_forward_time": 1.7197229862213135,
      "step": 34076
    },
    {
      "epoch": 3.2497406005859375e-05,
      "step": 34076,
      "training_step_time": 7.709217548370361
    },
    {
      "epoch": 3.249835968017578e-05,
      "model_forward_time": 1.7164502143859863,
      "step": 34077
    },
    {
      "epoch": 3.249835968017578e-05,
      "step": 34077,
      "training_step_time": 7.684089183807373
    },
    {
      "epoch": 3.249931335449219e-05,
      "model_forward_time": 1.7112517356872559,
      "step": 34078
    },
    {
      "epoch": 3.249931335449219e-05,
      "step": 34078,
      "training_step_time": 7.5524985790252686
    },
    {
      "epoch": 3.2500267028808594e-05,
      "model_forward_time": 2.2423667907714844,
      "step": 34079
    },
    {
      "epoch": 3.2500267028808594e-05,
      "step": 34079,
      "training_step_time": 8.41100025177002
    },
    {
      "epoch": 3.2501220703125e-05,
      "grad_norm": 0.1433064192533493,
      "learning_rate": 7.860940311767087e-06,
      "loss": 0.0598,
      "step": 34080
    },
    {
      "epoch": 3.2501220703125e-05,
      "model_forward_time": 1.7451279163360596,
      "step": 34080
    },
    {
      "dynamics_loss_avg": 0.05647645182907581,
      "epoch": 3.2501220703125e-05,
      "step": 34080
    },
    {
      "action_loss_avg": 0.0024910729203838854,
      "epoch": 3.2501220703125e-05,
      "step": 34080
    },
    {
      "epoch": 3.2501220703125e-05,
      "step": 34080,
      "training_step_time": 9.054022312164307
    },
    {
      "epoch": 3.250217437744141e-05,
      "model_forward_time": 1.7148704528808594,
      "step": 34081
    },
    {
      "epoch": 3.250217437744141e-05,
      "step": 34081,
      "training_step_time": 7.906055212020874
    },
    {
      "epoch": 3.250312805175781e-05,
      "model_forward_time": 1.7180202007293701,
      "step": 34082
    },
    {
      "epoch": 3.250312805175781e-05,
      "step": 34082,
      "training_step_time": 8.480459928512573
    },
    {
      "epoch": 3.250408172607422e-05,
      "model_forward_time": 1.7149755954742432,
      "step": 34083
    },
    {
      "epoch": 3.250408172607422e-05,
      "step": 34083,
      "training_step_time": 7.826717138290405
    },
    {
      "epoch": 3.250503540039063e-05,
      "model_forward_time": 1.7155120372772217,
      "step": 34084
    },
    {
      "epoch": 3.250503540039063e-05,
      "step": 34084,
      "training_step_time": 7.878817319869995
    },
    {
      "epoch": 3.250598907470703e-05,
      "model_forward_time": 1.7131116390228271,
      "step": 34085
    },
    {
      "epoch": 3.250598907470703e-05,
      "step": 34085,
      "training_step_time": 8.324687957763672
    },
    {
      "epoch": 3.2506942749023436e-05,
      "model_forward_time": 1.7151803970336914,
      "step": 34086
    },
    {
      "epoch": 3.2506942749023436e-05,
      "step": 34086,
      "training_step_time": 7.7488837242126465
    },
    {
      "epoch": 3.250789642333985e-05,
      "model_forward_time": 1.7174654006958008,
      "step": 34087
    },
    {
      "epoch": 3.250789642333985e-05,
      "step": 34087,
      "training_step_time": 7.635575532913208
    },
    {
      "epoch": 3.250885009765625e-05,
      "model_forward_time": 2.243441581726074,
      "step": 34088
    },
    {
      "epoch": 3.250885009765625e-05,
      "step": 34088,
      "training_step_time": 9.621709108352661
    },
    {
      "epoch": 3.2509803771972655e-05,
      "model_forward_time": 1.7564547061920166,
      "step": 34089
    },
    {
      "epoch": 3.2509803771972655e-05,
      "step": 34089,
      "training_step_time": 10.458846092224121
    },
    {
      "epoch": 3.2510757446289066e-05,
      "grad_norm": 0.11997106671333313,
      "learning_rate": 7.85958410765165e-06,
      "loss": 0.0579,
      "step": 34090
    },
    {
      "epoch": 3.2510757446289066e-05,
      "model_forward_time": 1.703366756439209,
      "step": 34090
    },
    {
      "dynamics_loss_avg": 0.043335477262735365,
      "epoch": 3.2510757446289066e-05,
      "step": 34090
    },
    {
      "action_loss_avg": 0.001409574193530716,
      "epoch": 3.2510757446289066e-05,
      "step": 34090
    },
    {
      "epoch": 3.2510757446289066e-05,
      "step": 34090,
      "training_step_time": 8.883996486663818
    },
    {
      "epoch": 3.251171112060547e-05,
      "model_forward_time": 1.7081785202026367,
      "step": 34091
    },
    {
      "epoch": 3.251171112060547e-05,
      "step": 34091,
      "training_step_time": 7.946310997009277
    },
    {
      "epoch": 3.2512664794921874e-05,
      "model_forward_time": 1.7150487899780273,
      "step": 34092
    },
    {
      "epoch": 3.2512664794921874e-05,
      "step": 34092,
      "training_step_time": 7.785490989685059
    },
    {
      "epoch": 3.2513618469238285e-05,
      "model_forward_time": 1.7213447093963623,
      "step": 34093
    },
    {
      "epoch": 3.2513618469238285e-05,
      "step": 34093,
      "training_step_time": 7.822614669799805
    },
    {
      "epoch": 3.251457214355469e-05,
      "model_forward_time": 1.7092900276184082,
      "step": 34094
    },
    {
      "epoch": 3.251457214355469e-05,
      "step": 34094,
      "training_step_time": 7.48735237121582
    },
    {
      "epoch": 3.251552581787109e-05,
      "model_forward_time": 1.7105650901794434,
      "step": 34095
    },
    {
      "epoch": 3.251552581787109e-05,
      "step": 34095,
      "training_step_time": 9.504129648208618
    },
    {
      "epoch": 3.25164794921875e-05,
      "model_forward_time": 1.7130184173583984,
      "step": 34096
    },
    {
      "epoch": 3.25164794921875e-05,
      "step": 34096,
      "training_step_time": 8.19767141342163
    },
    {
      "epoch": 3.251743316650391e-05,
      "model_forward_time": 2.2589893341064453,
      "step": 34097
    },
    {
      "epoch": 3.251743316650391e-05,
      "step": 34097,
      "training_step_time": 8.591551542282104
    },
    {
      "epoch": 3.251838684082031e-05,
      "model_forward_time": 1.7244701385498047,
      "step": 34098
    },
    {
      "epoch": 3.251838684082031e-05,
      "step": 34098,
      "training_step_time": 8.002930641174316
    },
    {
      "epoch": 3.2519340515136716e-05,
      "model_forward_time": 1.7154152393341064,
      "step": 34099
    },
    {
      "epoch": 3.2519340515136716e-05,
      "step": 34099,
      "training_step_time": 7.885470151901245
    },
    {
      "epoch": 3.2520294189453126e-05,
      "grad_norm": 0.18128928542137146,
      "learning_rate": 7.858227590816395e-06,
      "loss": 0.059,
      "step": 34100
    },
    {
      "epoch": 3.2520294189453126e-05,
      "model_forward_time": 1.704500436782837,
      "step": 34100
    },
    {
      "dynamics_loss_avg": 0.04396066982299089,
      "epoch": 3.2520294189453126e-05,
      "step": 34100
    },
    {
      "action_loss_avg": 0.0015325709886383266,
      "epoch": 3.2520294189453126e-05,
      "step": 34100
    },
    {
      "epoch": 3.2520294189453126e-05,
      "step": 34100,
      "training_step_time": 8.811821222305298
    },
    {
      "epoch": 3.252124786376953e-05,
      "model_forward_time": 1.7202210426330566,
      "step": 34101
    },
    {
      "epoch": 3.252124786376953e-05,
      "step": 34101,
      "training_step_time": 8.752586841583252
    },
    {
      "epoch": 3.2522201538085934e-05,
      "model_forward_time": 1.719512701034546,
      "step": 34102
    },
    {
      "epoch": 3.2522201538085934e-05,
      "step": 34102,
      "training_step_time": 7.809236764907837
    },
    {
      "epoch": 3.2523155212402345e-05,
      "model_forward_time": 1.716834306716919,
      "step": 34103
    },
    {
      "epoch": 3.2523155212402345e-05,
      "step": 34103,
      "training_step_time": 7.907569646835327
    },
    {
      "epoch": 3.252410888671875e-05,
      "model_forward_time": 1.718179702758789,
      "step": 34104
    },
    {
      "epoch": 3.252410888671875e-05,
      "step": 34104,
      "training_step_time": 8.43966794013977
    },
    {
      "epoch": 3.252506256103515e-05,
      "model_forward_time": 1.722336769104004,
      "step": 34105
    },
    {
      "epoch": 3.252506256103515e-05,
      "step": 34105,
      "training_step_time": 7.767889499664307
    },
    {
      "epoch": 3.2526016235351564e-05,
      "model_forward_time": 1.853200912475586,
      "step": 34106
    },
    {
      "epoch": 3.2526016235351564e-05,
      "step": 34106,
      "training_step_time": 7.843525171279907
    },
    {
      "epoch": 3.252696990966797e-05,
      "model_forward_time": 1.72322678565979,
      "step": 34107
    },
    {
      "epoch": 3.252696990966797e-05,
      "step": 34107,
      "training_step_time": 9.368076801300049
    },
    {
      "epoch": 3.252792358398437e-05,
      "model_forward_time": 1.7594006061553955,
      "step": 34108
    },
    {
      "epoch": 3.252792358398437e-05,
      "step": 34108,
      "training_step_time": 8.460452318191528
    },
    {
      "epoch": 3.252887725830078e-05,
      "model_forward_time": 1.715820074081421,
      "step": 34109
    },
    {
      "epoch": 3.252887725830078e-05,
      "step": 34109,
      "training_step_time": 7.999958276748657
    },
    {
      "epoch": 3.252983093261719e-05,
      "grad_norm": 0.13890807330608368,
      "learning_rate": 7.856870761409665e-06,
      "loss": 0.0589,
      "step": 34110
    },
    {
      "epoch": 3.252983093261719e-05,
      "model_forward_time": 1.7205390930175781,
      "step": 34110
    },
    {
      "dynamics_loss_avg": 0.06026427475735545,
      "epoch": 3.252983093261719e-05,
      "step": 34110
    },
    {
      "action_loss_avg": 0.002562880649929866,
      "epoch": 3.252983093261719e-05,
      "step": 34110
    },
    {
      "epoch": 3.252983093261719e-05,
      "step": 34110,
      "training_step_time": 7.784900426864624
    },
    {
      "epoch": 3.253078460693359e-05,
      "model_forward_time": 1.3419947624206543,
      "step": 34111
    },
    {
      "epoch": 3.253078460693359e-05,
      "step": 34111,
      "training_step_time": 7.881749629974365
    },
    {
      "epoch": 3.253173828125e-05,
      "model_forward_time": 1.7176876068115234,
      "step": 34112
    },
    {
      "epoch": 3.253173828125e-05,
      "step": 34112,
      "training_step_time": 7.69049596786499
    },
    {
      "epoch": 3.2532691955566406e-05,
      "model_forward_time": 1.7241230010986328,
      "step": 34113
    },
    {
      "epoch": 3.2532691955566406e-05,
      "step": 34113,
      "training_step_time": 8.321553468704224
    },
    {
      "epoch": 3.253364562988281e-05,
      "model_forward_time": 1.7284936904907227,
      "step": 34114
    },
    {
      "epoch": 3.253364562988281e-05,
      "step": 34114,
      "training_step_time": 8.712737083435059
    },
    {
      "epoch": 3.253459930419922e-05,
      "model_forward_time": 2.2498438358306885,
      "step": 34115
    },
    {
      "epoch": 3.253459930419922e-05,
      "step": 34115,
      "training_step_time": 7.891415596008301
    },
    {
      "epoch": 3.2535552978515625e-05,
      "model_forward_time": 1.716545581817627,
      "step": 34116
    },
    {
      "epoch": 3.2535552978515625e-05,
      "step": 34116,
      "training_step_time": 8.522749185562134
    },
    {
      "epoch": 3.253650665283203e-05,
      "model_forward_time": 1.3825440406799316,
      "step": 34117
    },
    {
      "epoch": 3.253650665283203e-05,
      "step": 34117,
      "training_step_time": 7.935585260391235
    },
    {
      "epoch": 3.253746032714844e-05,
      "model_forward_time": 1.7154858112335205,
      "step": 34118
    },
    {
      "epoch": 3.253746032714844e-05,
      "step": 34118,
      "training_step_time": 7.990824460983276
    },
    {
      "epoch": 3.2538414001464844e-05,
      "model_forward_time": 1.7195138931274414,
      "step": 34119
    },
    {
      "epoch": 3.2538414001464844e-05,
      "step": 34119,
      "training_step_time": 8.70808458328247
    },
    {
      "epoch": 3.253936767578125e-05,
      "grad_norm": 0.1430724561214447,
      "learning_rate": 7.855513619579846e-06,
      "loss": 0.0598,
      "step": 34120
    },
    {
      "epoch": 3.253936767578125e-05,
      "model_forward_time": 1.7305166721343994,
      "step": 34120
    },
    {
      "dynamics_loss_avg": 0.04473462402820587,
      "epoch": 3.253936767578125e-05,
      "step": 34120
    },
    {
      "action_loss_avg": 0.0018975818355102092,
      "epoch": 3.253936767578125e-05,
      "step": 34120
    },
    {
      "epoch": 3.253936767578125e-05,
      "step": 34120,
      "training_step_time": 7.888308048248291
    },
    {
      "epoch": 3.254032135009766e-05,
      "model_forward_time": 1.730884313583374,
      "step": 34121
    },
    {
      "epoch": 3.254032135009766e-05,
      "step": 34121,
      "training_step_time": 7.880633354187012
    },
    {
      "epoch": 3.254127502441406e-05,
      "model_forward_time": 1.7237169742584229,
      "step": 34122
    },
    {
      "epoch": 3.254127502441406e-05,
      "step": 34122,
      "training_step_time": 7.741283655166626
    },
    {
      "epoch": 3.2542228698730466e-05,
      "model_forward_time": 1.722055196762085,
      "step": 34123
    },
    {
      "epoch": 3.2542228698730466e-05,
      "step": 34123,
      "training_step_time": 7.77375864982605
    },
    {
      "epoch": 3.254318237304688e-05,
      "model_forward_time": 2.2861459255218506,
      "step": 34124
    },
    {
      "epoch": 3.254318237304688e-05,
      "step": 34124,
      "training_step_time": 8.386915683746338
    },
    {
      "epoch": 3.254413604736328e-05,
      "model_forward_time": 1.7313621044158936,
      "step": 34125
    },
    {
      "epoch": 3.254413604736328e-05,
      "step": 34125,
      "training_step_time": 9.834944009780884
    },
    {
      "epoch": 3.2545089721679685e-05,
      "model_forward_time": 1.7692351341247559,
      "step": 34126
    },
    {
      "epoch": 3.2545089721679685e-05,
      "step": 34126,
      "training_step_time": 9.272875308990479
    },
    {
      "epoch": 3.2546043395996096e-05,
      "model_forward_time": 1.737135410308838,
      "step": 34127
    },
    {
      "epoch": 3.2546043395996096e-05,
      "step": 34127,
      "training_step_time": 8.590383529663086
    },
    {
      "epoch": 3.25469970703125e-05,
      "model_forward_time": 1.7282662391662598,
      "step": 34128
    },
    {
      "epoch": 3.25469970703125e-05,
      "step": 34128,
      "training_step_time": 7.908305406570435
    },
    {
      "epoch": 3.2547950744628904e-05,
      "model_forward_time": 1.7146782875061035,
      "step": 34129
    },
    {
      "epoch": 3.2547950744628904e-05,
      "step": 34129,
      "training_step_time": 8.864858150482178
    },
    {
      "epoch": 3.2548904418945315e-05,
      "grad_norm": 0.175247922539711,
      "learning_rate": 7.85415616547535e-06,
      "loss": 0.0614,
      "step": 34130
    },
    {
      "epoch": 3.2548904418945315e-05,
      "model_forward_time": 1.7069075107574463,
      "step": 34130
    },
    {
      "dynamics_loss_avg": 0.0536218162626028,
      "epoch": 3.2548904418945315e-05,
      "step": 34130
    },
    {
      "action_loss_avg": 0.001329123432515189,
      "epoch": 3.2548904418945315e-05,
      "step": 34130
    },
    {
      "epoch": 3.2548904418945315e-05,
      "step": 34130,
      "training_step_time": 7.804072618484497
    },
    {
      "epoch": 3.254985809326172e-05,
      "model_forward_time": 1.7154278755187988,
      "step": 34131
    },
    {
      "epoch": 3.254985809326172e-05,
      "step": 34131,
      "training_step_time": 7.594523906707764
    },
    {
      "epoch": 3.255081176757812e-05,
      "model_forward_time": 1.715003252029419,
      "step": 34132
    },
    {
      "epoch": 3.255081176757812e-05,
      "step": 34132,
      "training_step_time": 7.851029396057129
    },
    {
      "epoch": 3.2551765441894534e-05,
      "model_forward_time": 2.2520711421966553,
      "step": 34133
    },
    {
      "epoch": 3.2551765441894534e-05,
      "step": 34133,
      "training_step_time": 7.685494899749756
    },
    {
      "epoch": 3.255271911621094e-05,
      "model_forward_time": 1.7185659408569336,
      "step": 34134
    },
    {
      "epoch": 3.255271911621094e-05,
      "step": 34134,
      "training_step_time": 7.924962759017944
    },
    {
      "epoch": 3.255367279052734e-05,
      "model_forward_time": 1.7268590927124023,
      "step": 34135
    },
    {
      "epoch": 3.255367279052734e-05,
      "step": 34135,
      "training_step_time": 8.39103388786316
    },
    {
      "epoch": 3.255462646484375e-05,
      "model_forward_time": 1.7141931056976318,
      "step": 34136
    },
    {
      "epoch": 3.255462646484375e-05,
      "step": 34136,
      "training_step_time": 8.757535696029663
    },
    {
      "epoch": 3.255558013916016e-05,
      "model_forward_time": 1.7107892036437988,
      "step": 34137
    },
    {
      "epoch": 3.255558013916016e-05,
      "step": 34137,
      "training_step_time": 7.840584993362427
    },
    {
      "epoch": 3.255653381347656e-05,
      "model_forward_time": 1.7099716663360596,
      "step": 34138
    },
    {
      "epoch": 3.255653381347656e-05,
      "step": 34138,
      "training_step_time": 7.832683801651001
    },
    {
      "epoch": 3.255748748779297e-05,
      "model_forward_time": 1.7143030166625977,
      "step": 34139
    },
    {
      "epoch": 3.255748748779297e-05,
      "step": 34139,
      "training_step_time": 7.789616584777832
    },
    {
      "epoch": 3.2558441162109376e-05,
      "grad_norm": 0.16445136070251465,
      "learning_rate": 7.852798399244628e-06,
      "loss": 0.0629,
      "step": 34140
    },
    {
      "epoch": 3.2558441162109376e-05,
      "model_forward_time": 1.7271325588226318,
      "step": 34140
    },
    {
      "dynamics_loss_avg": 0.05145283257588744,
      "epoch": 3.2558441162109376e-05,
      "step": 34140
    },
    {
      "action_loss_avg": 0.0021077464800328015,
      "epoch": 3.2558441162109376e-05,
      "step": 34140
    },
    {
      "epoch": 3.2558441162109376e-05,
      "step": 34140,
      "training_step_time": 7.7481043338775635
    },
    {
      "epoch": 3.255939483642578e-05,
      "model_forward_time": 1.7049181461334229,
      "step": 34141
    },
    {
      "epoch": 3.255939483642578e-05,
      "step": 34141,
      "training_step_time": 7.7516655921936035
    },
    {
      "epoch": 3.256034851074219e-05,
      "model_forward_time": 1.710571050643921,
      "step": 34142
    },
    {
      "epoch": 3.256034851074219e-05,
      "step": 34142,
      "training_step_time": 8.467961549758911
    },
    {
      "epoch": 3.2561302185058594e-05,
      "model_forward_time": 2.2748072147369385,
      "step": 34143
    },
    {
      "epoch": 3.2561302185058594e-05,
      "step": 34143,
      "training_step_time": 8.474541664123535
    },
    {
      "epoch": 3.2562255859375e-05,
      "model_forward_time": 1.7295911312103271,
      "step": 34144
    },
    {
      "epoch": 3.2562255859375e-05,
      "step": 34144,
      "training_step_time": 7.806219100952148
    },
    {
      "epoch": 3.256320953369141e-05,
      "model_forward_time": 1.7730770111083984,
      "step": 34145
    },
    {
      "epoch": 3.256320953369141e-05,
      "step": 34145,
      "training_step_time": 7.921552419662476
    },
    {
      "epoch": 3.256416320800781e-05,
      "model_forward_time": 1.7248592376708984,
      "step": 34146
    },
    {
      "epoch": 3.256416320800781e-05,
      "step": 34146,
      "training_step_time": 7.983926296234131
    },
    {
      "epoch": 3.256511688232422e-05,
      "model_forward_time": 1.725569248199463,
      "step": 34147
    },
    {
      "epoch": 3.256511688232422e-05,
      "step": 34147,
      "training_step_time": 8.395890474319458
    },
    {
      "epoch": 3.256607055664063e-05,
      "model_forward_time": 1.725292682647705,
      "step": 34148
    },
    {
      "epoch": 3.256607055664063e-05,
      "step": 34148,
      "training_step_time": 7.826135158538818
    },
    {
      "epoch": 3.256702423095703e-05,
      "model_forward_time": 1.7246615886688232,
      "step": 34149
    },
    {
      "epoch": 3.256702423095703e-05,
      "step": 34149,
      "training_step_time": 7.854980230331421
    },
    {
      "epoch": 3.2567977905273436e-05,
      "grad_norm": 0.1415991485118866,
      "learning_rate": 7.851440321036161e-06,
      "loss": 0.0639,
      "step": 34150
    },
    {
      "epoch": 3.2567977905273436e-05,
      "model_forward_time": 1.724454641342163,
      "step": 34150
    },
    {
      "dynamics_loss_avg": 0.05725310519337654,
      "epoch": 3.2567977905273436e-05,
      "step": 34150
    },
    {
      "action_loss_avg": 0.002300990663934499,
      "epoch": 3.2567977905273436e-05,
      "step": 34150
    },
    {
      "epoch": 3.2567977905273436e-05,
      "step": 34150,
      "training_step_time": 8.45429801940918
    },
    {
      "epoch": 3.256893157958985e-05,
      "model_forward_time": 1.7243072986602783,
      "step": 34151
    },
    {
      "epoch": 3.256893157958985e-05,
      "step": 34151,
      "training_step_time": 8.188735961914062
    },
    {
      "epoch": 3.256988525390625e-05,
      "model_forward_time": 2.2652170658111572,
      "step": 34152
    },
    {
      "epoch": 3.256988525390625e-05,
      "step": 34152,
      "training_step_time": 8.621763229370117
    },
    {
      "epoch": 3.2570838928222655e-05,
      "model_forward_time": 1.7254252433776855,
      "step": 34153
    },
    {
      "epoch": 3.2570838928222655e-05,
      "step": 34153,
      "training_step_time": 7.8454530239105225
    },
    {
      "epoch": 3.2571792602539066e-05,
      "model_forward_time": 1.7706475257873535,
      "step": 34154
    },
    {
      "epoch": 3.2571792602539066e-05,
      "step": 34154,
      "training_step_time": 8.456917762756348
    },
    {
      "epoch": 3.257274627685547e-05,
      "model_forward_time": 1.7240114212036133,
      "step": 34155
    },
    {
      "epoch": 3.257274627685547e-05,
      "step": 34155,
      "training_step_time": 9.36523962020874
    },
    {
      "epoch": 3.2573699951171874e-05,
      "model_forward_time": 1.7237517833709717,
      "step": 34156
    },
    {
      "epoch": 3.2573699951171874e-05,
      "step": 34156,
      "training_step_time": 7.904703855514526
    },
    {
      "epoch": 3.257465362548828e-05,
      "model_forward_time": 1.787158489227295,
      "step": 34157
    },
    {
      "epoch": 3.257465362548828e-05,
      "step": 34157,
      "training_step_time": 8.533546209335327
    },
    {
      "epoch": 3.257560729980469e-05,
      "model_forward_time": 1.7287993431091309,
      "step": 34158
    },
    {
      "epoch": 3.257560729980469e-05,
      "step": 34158,
      "training_step_time": 7.771288633346558
    },
    {
      "epoch": 3.257656097412109e-05,
      "model_forward_time": 1.7260663509368896,
      "step": 34159
    },
    {
      "epoch": 3.257656097412109e-05,
      "step": 34159,
      "training_step_time": 7.748591184616089
    },
    {
      "epoch": 3.25775146484375e-05,
      "grad_norm": 0.1789347231388092,
      "learning_rate": 7.850081930998467e-06,
      "loss": 0.06,
      "step": 34160
    },
    {
      "epoch": 3.25775146484375e-05,
      "model_forward_time": 1.7222778797149658,
      "step": 34160
    },
    {
      "dynamics_loss_avg": 0.0450777480378747,
      "epoch": 3.25775146484375e-05,
      "step": 34160
    },
    {
      "action_loss_avg": 0.0017309869290329516,
      "epoch": 3.25775146484375e-05,
      "step": 34160
    },
    {
      "epoch": 3.25775146484375e-05,
      "step": 34160,
      "training_step_time": 7.706492900848389
    },
    {
      "epoch": 3.257846832275391e-05,
      "model_forward_time": 2.263387680053711,
      "step": 34161
    },
    {
      "epoch": 3.257846832275391e-05,
      "step": 34161,
      "training_step_time": 7.91376805305481
    },
    {
      "epoch": 3.257942199707031e-05,
      "model_forward_time": 1.7203516960144043,
      "step": 34162
    },
    {
      "epoch": 3.257942199707031e-05,
      "step": 34162,
      "training_step_time": 8.574405193328857
    },
    {
      "epoch": 3.2580375671386716e-05,
      "model_forward_time": 1.737360954284668,
      "step": 34163
    },
    {
      "epoch": 3.2580375671386716e-05,
      "step": 34163,
      "training_step_time": 7.877791881561279
    },
    {
      "epoch": 3.2581329345703126e-05,
      "model_forward_time": 1.7668020725250244,
      "step": 34164
    },
    {
      "epoch": 3.2581329345703126e-05,
      "step": 34164,
      "training_step_time": 8.670714616775513
    },
    {
      "epoch": 3.258228302001953e-05,
      "model_forward_time": 1.7284760475158691,
      "step": 34165
    },
    {
      "epoch": 3.258228302001953e-05,
      "step": 34165,
      "training_step_time": 7.8449201583862305
    },
    {
      "epoch": 3.2583236694335934e-05,
      "model_forward_time": 1.7295665740966797,
      "step": 34166
    },
    {
      "epoch": 3.2583236694335934e-05,
      "step": 34166,
      "training_step_time": 7.826046466827393
    },
    {
      "epoch": 3.2584190368652345e-05,
      "model_forward_time": 1.7315504550933838,
      "step": 34167
    },
    {
      "epoch": 3.2584190368652345e-05,
      "step": 34167,
      "training_step_time": 7.815240383148193
    },
    {
      "epoch": 3.258514404296875e-05,
      "model_forward_time": 1.7197275161743164,
      "step": 34168
    },
    {
      "epoch": 3.258514404296875e-05,
      "step": 34168,
      "training_step_time": 7.727100372314453
    },
    {
      "epoch": 3.258609771728515e-05,
      "model_forward_time": 1.7250661849975586,
      "step": 34169
    },
    {
      "epoch": 3.258609771728515e-05,
      "step": 34169,
      "training_step_time": 7.704989433288574
    },
    {
      "epoch": 3.2587051391601564e-05,
      "grad_norm": 0.17344121634960175,
      "learning_rate": 7.8487232292801e-06,
      "loss": 0.0634,
      "step": 34170
    },
    {
      "epoch": 3.2587051391601564e-05,
      "model_forward_time": 2.290154457092285,
      "step": 34170
    },
    {
      "dynamics_loss_avg": 0.047368280962109564,
      "epoch": 3.2587051391601564e-05,
      "step": 34170
    },
    {
      "action_loss_avg": 0.0025238726957468316,
      "epoch": 3.2587051391601564e-05,
      "step": 34170
    },
    {
      "epoch": 3.2587051391601564e-05,
      "step": 34170,
      "training_step_time": 8.786371231079102
    },
    {
      "epoch": 3.258800506591797e-05,
      "model_forward_time": 1.7257657051086426,
      "step": 34171
    },
    {
      "epoch": 3.258800506591797e-05,
      "step": 34171,
      "training_step_time": 8.505244731903076
    },
    {
      "epoch": 3.258895874023437e-05,
      "model_forward_time": 1.721144676208496,
      "step": 34172
    },
    {
      "epoch": 3.258895874023437e-05,
      "step": 34172,
      "training_step_time": 8.006909132003784
    },
    {
      "epoch": 3.258991241455078e-05,
      "model_forward_time": 1.7691295146942139,
      "step": 34173
    },
    {
      "epoch": 3.258991241455078e-05,
      "step": 34173,
      "training_step_time": 11.309847593307495
    },
    {
      "epoch": 3.259086608886719e-05,
      "model_forward_time": 1.729935884475708,
      "step": 34174
    },
    {
      "epoch": 3.259086608886719e-05,
      "step": 34174,
      "training_step_time": 8.36784315109253
    },
    {
      "epoch": 3.259181976318359e-05,
      "model_forward_time": 1.7241806983947754,
      "step": 34175
    },
    {
      "epoch": 3.259181976318359e-05,
      "step": 34175,
      "training_step_time": 7.767340898513794
    },
    {
      "epoch": 3.25927734375e-05,
      "model_forward_time": 1.7259156703948975,
      "step": 34176
    },
    {
      "epoch": 3.25927734375e-05,
      "step": 34176,
      "training_step_time": 7.796753406524658
    },
    {
      "epoch": 3.2593727111816406e-05,
      "model_forward_time": 1.7257142066955566,
      "step": 34177
    },
    {
      "epoch": 3.2593727111816406e-05,
      "step": 34177,
      "training_step_time": 8.621673583984375
    },
    {
      "epoch": 3.259468078613281e-05,
      "model_forward_time": 1.7181086540222168,
      "step": 34178
    },
    {
      "epoch": 3.259468078613281e-05,
      "step": 34178,
      "training_step_time": 7.751189231872559
    },
    {
      "epoch": 3.259563446044922e-05,
      "model_forward_time": 2.2976009845733643,
      "step": 34179
    },
    {
      "epoch": 3.259563446044922e-05,
      "step": 34179,
      "training_step_time": 8.806851625442505
    },
    {
      "epoch": 3.2596588134765625e-05,
      "grad_norm": 0.18775172531604767,
      "learning_rate": 7.847364216029643e-06,
      "loss": 0.0627,
      "step": 34180
    },
    {
      "epoch": 3.2596588134765625e-05,
      "model_forward_time": 1.7619261741638184,
      "step": 34180
    },
    {
      "dynamics_loss_avg": 0.06379538774490356,
      "epoch": 3.2596588134765625e-05,
      "step": 34180
    },
    {
      "action_loss_avg": 0.0023375052958726885,
      "epoch": 3.2596588134765625e-05,
      "step": 34180
    },
    {
      "epoch": 3.2596588134765625e-05,
      "step": 34180,
      "training_step_time": 8.779392957687378
    },
    {
      "epoch": 3.259754180908203e-05,
      "model_forward_time": 1.727419376373291,
      "step": 34181
    },
    {
      "epoch": 3.259754180908203e-05,
      "step": 34181,
      "training_step_time": 7.892029285430908
    },
    {
      "epoch": 3.259849548339844e-05,
      "model_forward_time": 1.733320951461792,
      "step": 34182
    },
    {
      "epoch": 3.259849548339844e-05,
      "step": 34182,
      "training_step_time": 8.902082443237305
    },
    {
      "epoch": 3.2599449157714844e-05,
      "model_forward_time": 1.7355201244354248,
      "step": 34183
    },
    {
      "epoch": 3.2599449157714844e-05,
      "step": 34183,
      "training_step_time": 7.86178731918335
    },
    {
      "epoch": 3.260040283203125e-05,
      "model_forward_time": 1.7392692565917969,
      "step": 34184
    },
    {
      "epoch": 3.260040283203125e-05,
      "step": 34184,
      "training_step_time": 7.770429849624634
    },
    {
      "epoch": 3.260135650634766e-05,
      "model_forward_time": 1.7255394458770752,
      "step": 34185
    },
    {
      "epoch": 3.260135650634766e-05,
      "step": 34185,
      "training_step_time": 8.423247575759888
    },
    {
      "epoch": 3.260231018066406e-05,
      "model_forward_time": 1.731276035308838,
      "step": 34186
    },
    {
      "epoch": 3.260231018066406e-05,
      "step": 34186,
      "training_step_time": 7.807272434234619
    },
    {
      "epoch": 3.2603263854980466e-05,
      "model_forward_time": 1.7211923599243164,
      "step": 34187
    },
    {
      "epoch": 3.2603263854980466e-05,
      "step": 34187,
      "training_step_time": 7.722954273223877
    },
    {
      "epoch": 3.260421752929688e-05,
      "model_forward_time": 2.2797513008117676,
      "step": 34188
    },
    {
      "epoch": 3.260421752929688e-05,
      "step": 34188,
      "training_step_time": 7.7344584465026855
    },
    {
      "epoch": 3.260517120361328e-05,
      "model_forward_time": 1.7317023277282715,
      "step": 34189
    },
    {
      "epoch": 3.260517120361328e-05,
      "step": 34189,
      "training_step_time": 7.86617112159729
    },
    {
      "epoch": 3.2606124877929685e-05,
      "grad_norm": 0.1492418348789215,
      "learning_rate": 7.846004891395715e-06,
      "loss": 0.0627,
      "step": 34190
    },
    {
      "epoch": 3.2606124877929685e-05,
      "model_forward_time": 1.7264792919158936,
      "step": 34190
    },
    {
      "dynamics_loss_avg": 0.06397843174636364,
      "epoch": 3.2606124877929685e-05,
      "step": 34190
    },
    {
      "action_loss_avg": 0.002230606903322041,
      "epoch": 3.2606124877929685e-05,
      "step": 34190
    },
    {
      "epoch": 3.2606124877929685e-05,
      "step": 34190,
      "training_step_time": 7.83581805229187
    },
    {
      "epoch": 3.2607078552246096e-05,
      "model_forward_time": 1.7264280319213867,
      "step": 34191
    },
    {
      "epoch": 3.2607078552246096e-05,
      "step": 34191,
      "training_step_time": 9.689497470855713
    },
    {
      "epoch": 3.26080322265625e-05,
      "model_forward_time": 1.7381701469421387,
      "step": 34192
    },
    {
      "epoch": 3.26080322265625e-05,
      "step": 34192,
      "training_step_time": 9.169682264328003
    },
    {
      "epoch": 3.2608985900878904e-05,
      "model_forward_time": 1.719085693359375,
      "step": 34193
    },
    {
      "epoch": 3.2608985900878904e-05,
      "step": 34193,
      "training_step_time": 7.752835750579834
    },
    {
      "epoch": 3.2609939575195315e-05,
      "model_forward_time": 1.7135734558105469,
      "step": 34194
    },
    {
      "epoch": 3.2609939575195315e-05,
      "step": 34194,
      "training_step_time": 7.775489330291748
    },
    {
      "epoch": 3.261089324951172e-05,
      "model_forward_time": 1.7161827087402344,
      "step": 34195
    },
    {
      "epoch": 3.261089324951172e-05,
      "step": 34195,
      "training_step_time": 7.756536483764648
    },
    {
      "epoch": 3.261184692382812e-05,
      "model_forward_time": 1.7190570831298828,
      "step": 34196
    },
    {
      "epoch": 3.261184692382812e-05,
      "step": 34196,
      "training_step_time": 7.770395040512085
    },
    {
      "epoch": 3.2612800598144534e-05,
      "model_forward_time": 1.7130885124206543,
      "step": 34197
    },
    {
      "epoch": 3.2612800598144534e-05,
      "step": 34197,
      "training_step_time": 8.31976580619812
    },
    {
      "epoch": 3.261375427246094e-05,
      "model_forward_time": 1.712270975112915,
      "step": 34198
    },
    {
      "epoch": 3.261375427246094e-05,
      "step": 34198,
      "training_step_time": 7.91824197769165
    },
    {
      "epoch": 3.261470794677734e-05,
      "model_forward_time": 1.7072856426239014,
      "step": 34199
    },
    {
      "epoch": 3.261470794677734e-05,
      "step": 34199,
      "training_step_time": 7.854440212249756
    },
    {
      "epoch": 3.261566162109375e-05,
      "grad_norm": 0.1675938218832016,
      "learning_rate": 7.844645255526973e-06,
      "loss": 0.0636,
      "step": 34200
    },
    {
      "epoch": 3.261566162109375e-05,
      "model_forward_time": 1.708313226699829,
      "step": 34200
    },
    {
      "dynamics_loss_avg": 0.05548794995993376,
      "epoch": 3.261566162109375e-05,
      "step": 34200
    },
    {
      "action_loss_avg": 0.001959303754847497,
      "epoch": 3.261566162109375e-05,
      "step": 34200
    },
    {
      "epoch": 3.261566162109375e-05,
      "step": 34200,
      "training_step_time": 8.968271493911743
    },
    {
      "epoch": 3.261661529541016e-05,
      "model_forward_time": 1.7517399787902832,
      "step": 34201
    },
    {
      "epoch": 3.261661529541016e-05,
      "step": 34201,
      "training_step_time": 9.269617319107056
    },
    {
      "epoch": 3.261756896972656e-05,
      "model_forward_time": 1.7162408828735352,
      "step": 34202
    },
    {
      "epoch": 3.261756896972656e-05,
      "step": 34202,
      "training_step_time": 8.31616759300232
    },
    {
      "epoch": 3.261852264404297e-05,
      "model_forward_time": 1.721477746963501,
      "step": 34203
    },
    {
      "epoch": 3.261852264404297e-05,
      "step": 34203,
      "training_step_time": 8.332052230834961
    },
    {
      "epoch": 3.2619476318359376e-05,
      "model_forward_time": 1.7176878452301025,
      "step": 34204
    },
    {
      "epoch": 3.2619476318359376e-05,
      "step": 34204,
      "training_step_time": 8.801037788391113
    },
    {
      "epoch": 3.262042999267578e-05,
      "model_forward_time": 1.7163267135620117,
      "step": 34205
    },
    {
      "epoch": 3.262042999267578e-05,
      "step": 34205,
      "training_step_time": 7.730696201324463
    },
    {
      "epoch": 3.262138366699219e-05,
      "model_forward_time": 1.7182044982910156,
      "step": 34206
    },
    {
      "epoch": 3.262138366699219e-05,
      "step": 34206,
      "training_step_time": 7.775845766067505
    },
    {
      "epoch": 3.2622337341308594e-05,
      "model_forward_time": 2.2404942512512207,
      "step": 34207
    },
    {
      "epoch": 3.2622337341308594e-05,
      "step": 34207,
      "training_step_time": 7.865854501724243
    },
    {
      "epoch": 3.2623291015625e-05,
      "model_forward_time": 1.7190141677856445,
      "step": 34208
    },
    {
      "epoch": 3.2623291015625e-05,
      "step": 34208,
      "training_step_time": 7.876966953277588
    },
    {
      "epoch": 3.262424468994141e-05,
      "model_forward_time": 1.7147316932678223,
      "step": 34209
    },
    {
      "epoch": 3.262424468994141e-05,
      "step": 34209,
      "training_step_time": 8.336093425750732
    },
    {
      "epoch": 3.262519836425781e-05,
      "grad_norm": 0.08801725506782532,
      "learning_rate": 7.843285308572101e-06,
      "loss": 0.065,
      "step": 34210
    },
    {
      "epoch": 3.262519836425781e-05,
      "model_forward_time": 1.735224723815918,
      "step": 34210
    },
    {
      "dynamics_loss_avg": 0.06572137717157603,
      "epoch": 3.262519836425781e-05,
      "step": 34210
    },
    {
      "action_loss_avg": 0.001680102903628722,
      "epoch": 3.262519836425781e-05,
      "step": 34210
    },
    {
      "epoch": 3.262519836425781e-05,
      "step": 34210,
      "training_step_time": 8.667588710784912
    },
    {
      "epoch": 3.262615203857422e-05,
      "model_forward_time": 1.7157421112060547,
      "step": 34211
    },
    {
      "epoch": 3.262615203857422e-05,
      "step": 34211,
      "training_step_time": 7.805789232254028
    },
    {
      "epoch": 3.262710571289063e-05,
      "model_forward_time": 1.7125177383422852,
      "step": 34212
    },
    {
      "epoch": 3.262710571289063e-05,
      "step": 34212,
      "training_step_time": 9.248328447341919
    },
    {
      "epoch": 3.262805938720703e-05,
      "model_forward_time": 1.7097852230072021,
      "step": 34213
    },
    {
      "epoch": 3.262805938720703e-05,
      "step": 34213,
      "training_step_time": 7.7704784870147705
    },
    {
      "epoch": 3.2629013061523436e-05,
      "model_forward_time": 1.7160365581512451,
      "step": 34214
    },
    {
      "epoch": 3.2629013061523436e-05,
      "step": 34214,
      "training_step_time": 7.810361623764038
    },
    {
      "epoch": 3.262996673583985e-05,
      "model_forward_time": 1.7176563739776611,
      "step": 34215
    },
    {
      "epoch": 3.262996673583985e-05,
      "step": 34215,
      "training_step_time": 7.826347589492798
    },
    {
      "epoch": 3.263092041015625e-05,
      "model_forward_time": 2.2493503093719482,
      "step": 34216
    },
    {
      "epoch": 3.263092041015625e-05,
      "step": 34216,
      "training_step_time": 7.941715240478516
    },
    {
      "epoch": 3.2631874084472655e-05,
      "model_forward_time": 1.715416669845581,
      "step": 34217
    },
    {
      "epoch": 3.2631874084472655e-05,
      "step": 34217,
      "training_step_time": 7.897529125213623
    },
    {
      "epoch": 3.2632827758789066e-05,
      "model_forward_time": 1.7187399864196777,
      "step": 34218
    },
    {
      "epoch": 3.2632827758789066e-05,
      "step": 34218,
      "training_step_time": 8.912296295166016
    },
    {
      "epoch": 3.263378143310547e-05,
      "model_forward_time": 1.736302137374878,
      "step": 34219
    },
    {
      "epoch": 3.263378143310547e-05,
      "step": 34219,
      "training_step_time": 8.682652950286865
    },
    {
      "epoch": 3.2634735107421874e-05,
      "grad_norm": 0.12066835910081863,
      "learning_rate": 7.841925050679825e-06,
      "loss": 0.0643,
      "step": 34220
    },
    {
      "epoch": 3.2634735107421874e-05,
      "model_forward_time": 1.7117366790771484,
      "step": 34220
    },
    {
      "dynamics_loss_avg": 0.07233974765986204,
      "epoch": 3.2634735107421874e-05,
      "step": 34220
    },
    {
      "action_loss_avg": 0.002868758383556269,
      "epoch": 3.2634735107421874e-05,
      "step": 34220
    },
    {
      "epoch": 3.2634735107421874e-05,
      "step": 34220,
      "training_step_time": 9.491153001785278
    },
    {
      "epoch": 3.263568878173828e-05,
      "model_forward_time": 1.714721441268921,
      "step": 34221
    },
    {
      "epoch": 3.263568878173828e-05,
      "step": 34221,
      "training_step_time": 9.535571336746216
    },
    {
      "epoch": 3.263664245605469e-05,
      "model_forward_time": 1.709770917892456,
      "step": 34222
    },
    {
      "epoch": 3.263664245605469e-05,
      "step": 34222,
      "training_step_time": 7.7626166343688965
    },
    {
      "epoch": 3.263759613037109e-05,
      "model_forward_time": 1.7100803852081299,
      "step": 34223
    },
    {
      "epoch": 3.263759613037109e-05,
      "step": 34223,
      "training_step_time": 8.340807437896729
    },
    {
      "epoch": 3.26385498046875e-05,
      "model_forward_time": 1.7144289016723633,
      "step": 34224
    },
    {
      "epoch": 3.26385498046875e-05,
      "step": 34224,
      "training_step_time": 7.72310209274292
    },
    {
      "epoch": 3.263950347900391e-05,
      "model_forward_time": 2.2459185123443604,
      "step": 34225
    },
    {
      "epoch": 3.263950347900391e-05,
      "step": 34225,
      "training_step_time": 7.926898002624512
    },
    {
      "epoch": 3.264045715332031e-05,
      "model_forward_time": 1.709357500076294,
      "step": 34226
    },
    {
      "epoch": 3.264045715332031e-05,
      "step": 34226,
      "training_step_time": 8.54743480682373
    },
    {
      "epoch": 3.2641410827636716e-05,
      "model_forward_time": 1.691319227218628,
      "step": 34227
    },
    {
      "epoch": 3.2641410827636716e-05,
      "step": 34227,
      "training_step_time": 8.637511253356934
    },
    {
      "epoch": 3.2642364501953126e-05,
      "model_forward_time": 1.712965488433838,
      "step": 34228
    },
    {
      "epoch": 3.2642364501953126e-05,
      "step": 34228,
      "training_step_time": 8.776157855987549
    },
    {
      "epoch": 3.264331817626953e-05,
      "model_forward_time": 1.6950299739837646,
      "step": 34229
    },
    {
      "epoch": 3.264331817626953e-05,
      "step": 34229,
      "training_step_time": 8.632848501205444
    },
    {
      "epoch": 3.2644271850585935e-05,
      "grad_norm": 0.09339749813079834,
      "learning_rate": 7.840564481998895e-06,
      "loss": 0.0613,
      "step": 34230
    },
    {
      "epoch": 3.2644271850585935e-05,
      "model_forward_time": 1.7086076736450195,
      "step": 34230
    },
    {
      "dynamics_loss_avg": 0.05426320116966963,
      "epoch": 3.2644271850585935e-05,
      "step": 34230
    },
    {
      "action_loss_avg": 0.0015383002435555681,
      "epoch": 3.2644271850585935e-05,
      "step": 34230
    },
    {
      "epoch": 3.2644271850585935e-05,
      "step": 34230,
      "training_step_time": 7.6467297077178955
    },
    {
      "epoch": 3.2645225524902345e-05,
      "model_forward_time": 1.7254619598388672,
      "step": 34231
    },
    {
      "epoch": 3.2645225524902345e-05,
      "step": 34231,
      "training_step_time": 7.774648427963257
    },
    {
      "epoch": 3.264617919921875e-05,
      "model_forward_time": 1.7175302505493164,
      "step": 34232
    },
    {
      "epoch": 3.264617919921875e-05,
      "step": 34232,
      "training_step_time": 8.467331171035767
    },
    {
      "epoch": 3.264713287353515e-05,
      "model_forward_time": 1.71427321434021,
      "step": 34233
    },
    {
      "epoch": 3.264713287353515e-05,
      "step": 34233,
      "training_step_time": 7.841455459594727
    },
    {
      "epoch": 3.2648086547851564e-05,
      "model_forward_time": 2.2391202449798584,
      "step": 34234
    },
    {
      "epoch": 3.2648086547851564e-05,
      "step": 34234,
      "training_step_time": 7.645688533782959
    },
    {
      "epoch": 3.264904022216797e-05,
      "model_forward_time": 1.7178418636322021,
      "step": 34235
    },
    {
      "epoch": 3.264904022216797e-05,
      "step": 34235,
      "training_step_time": 8.467020988464355
    },
    {
      "epoch": 3.264999389648437e-05,
      "model_forward_time": 1.7050971984863281,
      "step": 34236
    },
    {
      "epoch": 3.264999389648437e-05,
      "step": 34236,
      "training_step_time": 8.998225927352905
    },
    {
      "epoch": 3.265094757080078e-05,
      "model_forward_time": 1.7067861557006836,
      "step": 34237
    },
    {
      "epoch": 3.265094757080078e-05,
      "step": 34237,
      "training_step_time": 7.915838003158569
    },
    {
      "epoch": 3.265190124511719e-05,
      "model_forward_time": 1.7437546253204346,
      "step": 34238
    },
    {
      "epoch": 3.265190124511719e-05,
      "step": 34238,
      "training_step_time": 8.657575130462646
    },
    {
      "epoch": 3.265285491943359e-05,
      "model_forward_time": 1.6879205703735352,
      "step": 34239
    },
    {
      "epoch": 3.265285491943359e-05,
      "step": 34239,
      "training_step_time": 7.747358560562134
    },
    {
      "epoch": 3.265380859375e-05,
      "grad_norm": 0.13176612555980682,
      "learning_rate": 7.839203602678105e-06,
      "loss": 0.0643,
      "step": 34240
    },
    {
      "epoch": 3.265380859375e-05,
      "model_forward_time": 1.698918104171753,
      "step": 34240
    },
    {
      "dynamics_loss_avg": 0.05618558526039123,
      "epoch": 3.265380859375e-05,
      "step": 34240
    },
    {
      "action_loss_avg": 0.0014946039445931092,
      "epoch": 3.265380859375e-05,
      "step": 34240
    },
    {
      "epoch": 3.265380859375e-05,
      "step": 34240,
      "training_step_time": 7.836014747619629
    },
    {
      "epoch": 3.2654762268066406e-05,
      "model_forward_time": 1.7189490795135498,
      "step": 34241
    },
    {
      "epoch": 3.2654762268066406e-05,
      "step": 34241,
      "training_step_time": 7.875981569290161
    },
    {
      "epoch": 3.265571594238281e-05,
      "model_forward_time": 1.703155755996704,
      "step": 34242
    },
    {
      "epoch": 3.265571594238281e-05,
      "step": 34242,
      "training_step_time": 7.8723649978637695
    },
    {
      "epoch": 3.265666961669922e-05,
      "model_forward_time": 2.2487456798553467,
      "step": 34243
    },
    {
      "epoch": 3.265666961669922e-05,
      "step": 34243,
      "training_step_time": 7.631479978561401
    },
    {
      "epoch": 3.2657623291015625e-05,
      "model_forward_time": 1.7160871028900146,
      "step": 34244
    },
    {
      "epoch": 3.2657623291015625e-05,
      "step": 34244,
      "training_step_time": 8.71231484413147
    },
    {
      "epoch": 3.265857696533203e-05,
      "model_forward_time": 1.7059588432312012,
      "step": 34245
    },
    {
      "epoch": 3.265857696533203e-05,
      "step": 34245,
      "training_step_time": 8.71399974822998
    },
    {
      "epoch": 3.265953063964844e-05,
      "model_forward_time": 1.712768793106079,
      "step": 34246
    },
    {
      "epoch": 3.265953063964844e-05,
      "step": 34246,
      "training_step_time": 8.60979437828064
    },
    {
      "epoch": 3.2660484313964844e-05,
      "model_forward_time": 1.7496213912963867,
      "step": 34247
    },
    {
      "epoch": 3.2660484313964844e-05,
      "step": 34247,
      "training_step_time": 8.541647672653198
    },
    {
      "epoch": 3.266143798828125e-05,
      "model_forward_time": 1.7006583213806152,
      "step": 34248
    },
    {
      "epoch": 3.266143798828125e-05,
      "step": 34248,
      "training_step_time": 7.833060264587402
    },
    {
      "epoch": 3.266239166259766e-05,
      "model_forward_time": 1.7004547119140625,
      "step": 34249
    },
    {
      "epoch": 3.266239166259766e-05,
      "step": 34249,
      "training_step_time": 8.554425477981567
    },
    {
      "epoch": 3.266334533691406e-05,
      "grad_norm": 0.12651069462299347,
      "learning_rate": 7.837842412866279e-06,
      "loss": 0.0635,
      "step": 34250
    },
    {
      "epoch": 3.266334533691406e-05,
      "model_forward_time": 1.7173893451690674,
      "step": 34250
    },
    {
      "dynamics_loss_avg": 0.05435951203107834,
      "epoch": 3.266334533691406e-05,
      "step": 34250
    },
    {
      "action_loss_avg": 0.0012817231996450572,
      "epoch": 3.266334533691406e-05,
      "step": 34250
    },
    {
      "epoch": 3.266334533691406e-05,
      "step": 34250,
      "training_step_time": 7.730088710784912
    },
    {
      "epoch": 3.2664299011230467e-05,
      "model_forward_time": 1.7056910991668701,
      "step": 34251
    },
    {
      "epoch": 3.2664299011230467e-05,
      "step": 34251,
      "training_step_time": 7.778717041015625
    },
    {
      "epoch": 3.266525268554688e-05,
      "model_forward_time": 2.244985342025757,
      "step": 34252
    },
    {
      "epoch": 3.266525268554688e-05,
      "step": 34252,
      "training_step_time": 7.3114001750946045
    },
    {
      "epoch": 3.266620635986328e-05,
      "model_forward_time": 1.7113368511199951,
      "step": 34253
    },
    {
      "epoch": 3.266620635986328e-05,
      "step": 34253,
      "training_step_time": 7.819242238998413
    },
    {
      "epoch": 3.2667160034179685e-05,
      "model_forward_time": 1.7126541137695312,
      "step": 34254
    },
    {
      "epoch": 3.2667160034179685e-05,
      "step": 34254,
      "training_step_time": 7.864482879638672
    },
    {
      "epoch": 3.2668113708496096e-05,
      "model_forward_time": 1.7071824073791504,
      "step": 34255
    },
    {
      "epoch": 3.2668113708496096e-05,
      "step": 34255,
      "training_step_time": 9.94297742843628
    },
    {
      "epoch": 3.26690673828125e-05,
      "model_forward_time": 1.7538070678710938,
      "step": 34256
    },
    {
      "epoch": 3.26690673828125e-05,
      "step": 34256,
      "training_step_time": 9.462376356124878
    },
    {
      "epoch": 3.2670021057128904e-05,
      "model_forward_time": 1.7018871307373047,
      "step": 34257
    },
    {
      "epoch": 3.2670021057128904e-05,
      "step": 34257,
      "training_step_time": 7.800656080245972
    },
    {
      "epoch": 3.2670974731445315e-05,
      "model_forward_time": 1.7183794975280762,
      "step": 34258
    },
    {
      "epoch": 3.2670974731445315e-05,
      "step": 34258,
      "training_step_time": 7.746245622634888
    },
    {
      "epoch": 3.267192840576172e-05,
      "model_forward_time": 1.3335607051849365,
      "step": 34259
    },
    {
      "epoch": 3.267192840576172e-05,
      "step": 34259,
      "training_step_time": 7.710058927536011
    },
    {
      "epoch": 3.267288208007812e-05,
      "grad_norm": 0.1295393407344818,
      "learning_rate": 7.836480912712272e-06,
      "loss": 0.0607,
      "step": 34260
    },
    {
      "epoch": 3.267288208007812e-05,
      "model_forward_time": 1.6969490051269531,
      "step": 34260
    },
    {
      "dynamics_loss_avg": 0.054471751861274245,
      "epoch": 3.267288208007812e-05,
      "step": 34260
    },
    {
      "action_loss_avg": 0.0014949439093470574,
      "epoch": 3.267288208007812e-05,
      "step": 34260
    },
    {
      "epoch": 3.267288208007812e-05,
      "step": 34260,
      "training_step_time": 8.903951406478882
    },
    {
      "epoch": 3.2673835754394534e-05,
      "model_forward_time": 2.2495830059051514,
      "step": 34261
    },
    {
      "epoch": 3.2673835754394534e-05,
      "step": 34261,
      "training_step_time": 7.359806060791016
    },
    {
      "epoch": 3.267478942871094e-05,
      "model_forward_time": 1.7034461498260498,
      "step": 34262
    },
    {
      "epoch": 3.267478942871094e-05,
      "step": 34262,
      "training_step_time": 7.876213073730469
    },
    {
      "epoch": 3.267574310302734e-05,
      "model_forward_time": 1.7110300064086914,
      "step": 34263
    },
    {
      "epoch": 3.267574310302734e-05,
      "step": 34263,
      "training_step_time": 8.684090852737427
    },
    {
      "epoch": 3.267669677734375e-05,
      "model_forward_time": 1.713578701019287,
      "step": 34264
    },
    {
      "epoch": 3.267669677734375e-05,
      "step": 34264,
      "training_step_time": 8.926843404769897
    },
    {
      "epoch": 3.267765045166016e-05,
      "model_forward_time": 1.7406575679779053,
      "step": 34265
    },
    {
      "epoch": 3.267765045166016e-05,
      "step": 34265,
      "training_step_time": 9.51711893081665
    },
    {
      "epoch": 3.267860412597656e-05,
      "model_forward_time": 1.7398626804351807,
      "step": 34266
    },
    {
      "epoch": 3.267860412597656e-05,
      "step": 34266,
      "training_step_time": 7.891923666000366
    },
    {
      "epoch": 3.267955780029297e-05,
      "model_forward_time": 1.7075459957122803,
      "step": 34267
    },
    {
      "epoch": 3.267955780029297e-05,
      "step": 34267,
      "training_step_time": 7.760670185089111
    },
    {
      "epoch": 3.2680511474609376e-05,
      "model_forward_time": 1.7095940113067627,
      "step": 34268
    },
    {
      "epoch": 3.2680511474609376e-05,
      "step": 34268,
      "training_step_time": 7.707304239273071
    },
    {
      "epoch": 3.268146514892578e-05,
      "model_forward_time": 1.706228256225586,
      "step": 34269
    },
    {
      "epoch": 3.268146514892578e-05,
      "step": 34269,
      "training_step_time": 8.589297533035278
    },
    {
      "epoch": 3.268241882324219e-05,
      "grad_norm": 0.1688980907201767,
      "learning_rate": 7.835119102364976e-06,
      "loss": 0.0622,
      "step": 34270
    },
    {
      "epoch": 3.268241882324219e-05,
      "model_forward_time": 2.257411241531372,
      "step": 34270
    },
    {
      "dynamics_loss_avg": 0.053800700977444646,
      "epoch": 3.268241882324219e-05,
      "step": 34270
    },
    {
      "action_loss_avg": 0.0015913829323835672,
      "epoch": 3.268241882324219e-05,
      "step": 34270
    },
    {
      "epoch": 3.268241882324219e-05,
      "step": 34270,
      "training_step_time": 7.443755388259888
    },
    {
      "epoch": 3.2683372497558595e-05,
      "model_forward_time": 1.705606460571289,
      "step": 34271
    },
    {
      "epoch": 3.2683372497558595e-05,
      "step": 34271,
      "training_step_time": 8.120348691940308
    },
    {
      "epoch": 3.2684326171875e-05,
      "model_forward_time": 1.7076170444488525,
      "step": 34272
    },
    {
      "epoch": 3.2684326171875e-05,
      "step": 34272,
      "training_step_time": 7.71424126625061
    },
    {
      "epoch": 3.268527984619141e-05,
      "model_forward_time": 1.7151713371276855,
      "step": 34273
    },
    {
      "epoch": 3.268527984619141e-05,
      "step": 34273,
      "training_step_time": 8.648306131362915
    },
    {
      "epoch": 3.268623352050781e-05,
      "model_forward_time": 1.7091236114501953,
      "step": 34274
    },
    {
      "epoch": 3.268623352050781e-05,
      "step": 34274,
      "training_step_time": 8.904778242111206
    },
    {
      "epoch": 3.268718719482422e-05,
      "model_forward_time": 1.7153522968292236,
      "step": 34275
    },
    {
      "epoch": 3.268718719482422e-05,
      "step": 34275,
      "training_step_time": 8.66239881515503
    },
    {
      "epoch": 3.268814086914063e-05,
      "model_forward_time": 1.7062184810638428,
      "step": 34276
    },
    {
      "epoch": 3.268814086914063e-05,
      "step": 34276,
      "training_step_time": 7.839181900024414
    },
    {
      "epoch": 3.268909454345703e-05,
      "model_forward_time": 1.7081003189086914,
      "step": 34277
    },
    {
      "epoch": 3.268909454345703e-05,
      "step": 34277,
      "training_step_time": 8.113256216049194
    },
    {
      "epoch": 3.2690048217773436e-05,
      "model_forward_time": 1.6819534301757812,
      "step": 34278
    },
    {
      "epoch": 3.2690048217773436e-05,
      "step": 34278,
      "training_step_time": 8.440907955169678
    },
    {
      "epoch": 3.269100189208985e-05,
      "model_forward_time": 2.274766683578491,
      "step": 34279
    },
    {
      "epoch": 3.269100189208985e-05,
      "step": 34279,
      "training_step_time": 8.146263599395752
    },
    {
      "epoch": 3.269195556640625e-05,
      "grad_norm": 0.16406159102916718,
      "learning_rate": 7.833756981973321e-06,
      "loss": 0.0578,
      "step": 34280
    },
    {
      "epoch": 3.269195556640625e-05,
      "model_forward_time": 1.706561803817749,
      "step": 34280
    },
    {
      "dynamics_loss_avg": 0.06021522507071495,
      "epoch": 3.269195556640625e-05,
      "step": 34280
    },
    {
      "action_loss_avg": 0.0024396963301114736,
      "epoch": 3.269195556640625e-05,
      "step": 34280
    },
    {
      "epoch": 3.269195556640625e-05,
      "step": 34280,
      "training_step_time": 9.309497117996216
    },
    {
      "epoch": 3.2692909240722655e-05,
      "model_forward_time": 1.7106099128723145,
      "step": 34281
    },
    {
      "epoch": 3.2692909240722655e-05,
      "step": 34281,
      "training_step_time": 9.61632251739502
    },
    {
      "epoch": 3.269386291503906e-05,
      "model_forward_time": 1.7037477493286133,
      "step": 34282
    },
    {
      "epoch": 3.269386291503906e-05,
      "step": 34282,
      "training_step_time": 7.958784103393555
    },
    {
      "epoch": 3.269481658935547e-05,
      "model_forward_time": 1.7084999084472656,
      "step": 34283
    },
    {
      "epoch": 3.269481658935547e-05,
      "step": 34283,
      "training_step_time": 9.904777765274048
    },
    {
      "epoch": 3.2695770263671874e-05,
      "model_forward_time": 1.704530954360962,
      "step": 34284
    },
    {
      "epoch": 3.2695770263671874e-05,
      "step": 34284,
      "training_step_time": 8.476564407348633
    },
    {
      "epoch": 3.269672393798828e-05,
      "model_forward_time": 1.7068383693695068,
      "step": 34285
    },
    {
      "epoch": 3.269672393798828e-05,
      "step": 34285,
      "training_step_time": 8.305301189422607
    },
    {
      "epoch": 3.269767761230469e-05,
      "model_forward_time": 1.7059569358825684,
      "step": 34286
    },
    {
      "epoch": 3.269767761230469e-05,
      "step": 34286,
      "training_step_time": 7.733453035354614
    },
    {
      "epoch": 3.269863128662109e-05,
      "model_forward_time": 1.7029991149902344,
      "step": 34287
    },
    {
      "epoch": 3.269863128662109e-05,
      "step": 34287,
      "training_step_time": 7.752217531204224
    },
    {
      "epoch": 3.26995849609375e-05,
      "model_forward_time": 2.242236614227295,
      "step": 34288
    },
    {
      "epoch": 3.26995849609375e-05,
      "step": 34288,
      "training_step_time": 7.797248601913452
    },
    {
      "epoch": 3.270053863525391e-05,
      "model_forward_time": 1.705007553100586,
      "step": 34289
    },
    {
      "epoch": 3.270053863525391e-05,
      "step": 34289,
      "training_step_time": 7.593312740325928
    },
    {
      "epoch": 3.270149230957031e-05,
      "grad_norm": 0.14985623955726624,
      "learning_rate": 7.83239455168626e-06,
      "loss": 0.0627,
      "step": 34290
    },
    {
      "epoch": 3.270149230957031e-05,
      "model_forward_time": 1.697659969329834,
      "step": 34290
    },
    {
      "dynamics_loss_avg": 0.05775772370398045,
      "epoch": 3.270149230957031e-05,
      "step": 34290
    },
    {
      "action_loss_avg": 0.0018307570484466852,
      "epoch": 3.270149230957031e-05,
      "step": 34290
    },
    {
      "epoch": 3.270149230957031e-05,
      "step": 34290,
      "training_step_time": 8.87190318107605
    },
    {
      "epoch": 3.2702445983886716e-05,
      "model_forward_time": 1.7168521881103516,
      "step": 34291
    },
    {
      "epoch": 3.2702445983886716e-05,
      "step": 34291,
      "training_step_time": 8.751275062561035
    },
    {
      "epoch": 3.2703399658203127e-05,
      "model_forward_time": 1.710622787475586,
      "step": 34292
    },
    {
      "epoch": 3.2703399658203127e-05,
      "step": 34292,
      "training_step_time": 8.944315195083618
    },
    {
      "epoch": 3.270435333251953e-05,
      "model_forward_time": 1.7468655109405518,
      "step": 34293
    },
    {
      "epoch": 3.270435333251953e-05,
      "step": 34293,
      "training_step_time": 8.49570631980896
    },
    {
      "epoch": 3.2705307006835935e-05,
      "model_forward_time": 1.6970176696777344,
      "step": 34294
    },
    {
      "epoch": 3.2705307006835935e-05,
      "step": 34294,
      "training_step_time": 8.37724781036377
    },
    {
      "epoch": 3.2706260681152345e-05,
      "model_forward_time": 1.7137489318847656,
      "step": 34295
    },
    {
      "epoch": 3.2706260681152345e-05,
      "step": 34295,
      "training_step_time": 7.692471265792847
    },
    {
      "epoch": 3.270721435546875e-05,
      "model_forward_time": 1.6994249820709229,
      "step": 34296
    },
    {
      "epoch": 3.270721435546875e-05,
      "step": 34296,
      "training_step_time": 7.72970175743103
    },
    {
      "epoch": 3.2708168029785153e-05,
      "model_forward_time": 1.7005174160003662,
      "step": 34297
    },
    {
      "epoch": 3.2708168029785153e-05,
      "step": 34297,
      "training_step_time": 7.895567178726196
    },
    {
      "epoch": 3.2709121704101564e-05,
      "model_forward_time": 2.2430455684661865,
      "step": 34298
    },
    {
      "epoch": 3.2709121704101564e-05,
      "step": 34298,
      "training_step_time": 8.231396675109863
    },
    {
      "epoch": 3.271007537841797e-05,
      "model_forward_time": 1.7109198570251465,
      "step": 34299
    },
    {
      "epoch": 3.271007537841797e-05,
      "step": 34299,
      "training_step_time": 7.817116737365723
    },
    {
      "epoch": 3.271102905273437e-05,
      "grad_norm": 0.1393256038427353,
      "learning_rate": 7.83103181165279e-06,
      "loss": 0.0576,
      "step": 34300
    },
    {
      "epoch": 3.271102905273437e-05,
      "model_forward_time": 1.6993603706359863,
      "step": 34300
    },
    {
      "dynamics_loss_avg": 0.07444031015038491,
      "epoch": 3.271102905273437e-05,
      "step": 34300
    },
    {
      "action_loss_avg": 0.0033162106643430888,
      "epoch": 3.271102905273437e-05,
      "step": 34300
    },
    {
      "epoch": 3.271102905273437e-05,
      "step": 34300,
      "training_step_time": 8.45246958732605
    },
    {
      "epoch": 3.271198272705078e-05,
      "model_forward_time": 1.7352778911590576,
      "step": 34301
    },
    {
      "epoch": 3.271198272705078e-05,
      "step": 34301,
      "training_step_time": 7.709233999252319
    },
    {
      "epoch": 3.271293640136719e-05,
      "model_forward_time": 1.7229743003845215,
      "step": 34302
    },
    {
      "epoch": 3.271293640136719e-05,
      "step": 34302,
      "training_step_time": 8.48641562461853
    },
    {
      "epoch": 3.271389007568359e-05,
      "model_forward_time": 1.7042076587677002,
      "step": 34303
    },
    {
      "epoch": 3.271389007568359e-05,
      "step": 34303,
      "training_step_time": 7.674067974090576
    },
    {
      "epoch": 3.271484375e-05,
      "model_forward_time": 1.7031326293945312,
      "step": 34304
    },
    {
      "epoch": 3.271484375e-05,
      "step": 34304,
      "training_step_time": 7.755005598068237
    },
    {
      "epoch": 3.2715797424316406e-05,
      "model_forward_time": 1.6957330703735352,
      "step": 34305
    },
    {
      "epoch": 3.2715797424316406e-05,
      "step": 34305,
      "training_step_time": 7.719768285751343
    },
    {
      "epoch": 3.271675109863281e-05,
      "model_forward_time": 1.6997230052947998,
      "step": 34306
    },
    {
      "epoch": 3.271675109863281e-05,
      "step": 34306,
      "training_step_time": 7.789055824279785
    },
    {
      "epoch": 3.271770477294922e-05,
      "model_forward_time": 2.235414981842041,
      "step": 34307
    },
    {
      "epoch": 3.271770477294922e-05,
      "step": 34307,
      "training_step_time": 7.831094741821289
    },
    {
      "epoch": 3.2718658447265625e-05,
      "model_forward_time": 1.6983182430267334,
      "step": 34308
    },
    {
      "epoch": 3.2718658447265625e-05,
      "step": 34308,
      "training_step_time": 7.738843679428101
    },
    {
      "epoch": 3.271961212158203e-05,
      "model_forward_time": 1.7059721946716309,
      "step": 34309
    },
    {
      "epoch": 3.271961212158203e-05,
      "step": 34309,
      "training_step_time": 7.716153383255005
    },
    {
      "epoch": 3.272056579589844e-05,
      "grad_norm": 0.12379442155361176,
      "learning_rate": 7.829668762021939e-06,
      "loss": 0.0615,
      "step": 34310
    },
    {
      "epoch": 3.272056579589844e-05,
      "model_forward_time": 1.695068359375,
      "step": 34310
    },
    {
      "dynamics_loss_avg": 0.061983727104961875,
      "epoch": 3.272056579589844e-05,
      "step": 34310
    },
    {
      "action_loss_avg": 0.002432605775538832,
      "epoch": 3.272056579589844e-05,
      "step": 34310
    },
    {
      "epoch": 3.272056579589844e-05,
      "step": 34310,
      "training_step_time": 8.29848051071167
    },
    {
      "epoch": 3.2721519470214844e-05,
      "model_forward_time": 1.7105762958526611,
      "step": 34311
    },
    {
      "epoch": 3.2721519470214844e-05,
      "step": 34311,
      "training_step_time": 8.791215181350708
    },
    {
      "epoch": 3.272247314453125e-05,
      "model_forward_time": 1.7015509605407715,
      "step": 34312
    },
    {
      "epoch": 3.272247314453125e-05,
      "step": 34312,
      "training_step_time": 7.689917325973511
    },
    {
      "epoch": 3.272342681884766e-05,
      "model_forward_time": 1.6974067687988281,
      "step": 34313
    },
    {
      "epoch": 3.272342681884766e-05,
      "step": 34313,
      "training_step_time": 7.748843193054199
    },
    {
      "epoch": 3.272438049316406e-05,
      "model_forward_time": 1.695232629776001,
      "step": 34314
    },
    {
      "epoch": 3.272438049316406e-05,
      "step": 34314,
      "training_step_time": 7.69046425819397
    },
    {
      "epoch": 3.2725334167480467e-05,
      "model_forward_time": 1.3400051593780518,
      "step": 34315
    },
    {
      "epoch": 3.2725334167480467e-05,
      "step": 34315,
      "training_step_time": 8.744703531265259
    },
    {
      "epoch": 3.272628784179688e-05,
      "model_forward_time": 2.240522861480713,
      "step": 34316
    },
    {
      "epoch": 3.272628784179688e-05,
      "step": 34316,
      "training_step_time": 7.295389413833618
    },
    {
      "epoch": 3.272724151611328e-05,
      "model_forward_time": 1.6984868049621582,
      "step": 34317
    },
    {
      "epoch": 3.272724151611328e-05,
      "step": 34317,
      "training_step_time": 7.677299976348877
    },
    {
      "epoch": 3.2728195190429685e-05,
      "model_forward_time": 1.7049384117126465,
      "step": 34318
    },
    {
      "epoch": 3.2728195190429685e-05,
      "step": 34318,
      "training_step_time": 8.391399145126343
    },
    {
      "epoch": 3.2729148864746096e-05,
      "model_forward_time": 1.703904151916504,
      "step": 34319
    },
    {
      "epoch": 3.2729148864746096e-05,
      "step": 34319,
      "training_step_time": 8.67668104171753
    },
    {
      "epoch": 3.27301025390625e-05,
      "grad_norm": 0.15191324055194855,
      "learning_rate": 7.828305402942765e-06,
      "loss": 0.0611,
      "step": 34320
    },
    {
      "epoch": 3.27301025390625e-05,
      "model_forward_time": 1.7081875801086426,
      "step": 34320
    },
    {
      "dynamics_loss_avg": 0.06326647978276015,
      "epoch": 3.27301025390625e-05,
      "step": 34320
    },
    {
      "action_loss_avg": 0.0024620565673103556,
      "epoch": 3.27301025390625e-05,
      "step": 34320
    },
    {
      "epoch": 3.27301025390625e-05,
      "step": 34320,
      "training_step_time": 7.7474634647369385
    },
    {
      "epoch": 3.2731056213378904e-05,
      "model_forward_time": 1.7359802722930908,
      "step": 34321
    },
    {
      "epoch": 3.2731056213378904e-05,
      "step": 34321,
      "training_step_time": 7.687652587890625
    },
    {
      "epoch": 3.2732009887695315e-05,
      "model_forward_time": 1.6979429721832275,
      "step": 34322
    },
    {
      "epoch": 3.2732009887695315e-05,
      "step": 34322,
      "training_step_time": 7.72294807434082
    },
    {
      "epoch": 3.273296356201172e-05,
      "model_forward_time": 1.7050254344940186,
      "step": 34323
    },
    {
      "epoch": 3.273296356201172e-05,
      "step": 34323,
      "training_step_time": 7.64798378944397
    },
    {
      "epoch": 3.273391723632812e-05,
      "model_forward_time": 1.7017686367034912,
      "step": 34324
    },
    {
      "epoch": 3.273391723632812e-05,
      "step": 34324,
      "training_step_time": 7.668863773345947
    },
    {
      "epoch": 3.2734870910644534e-05,
      "model_forward_time": 2.2309582233428955,
      "step": 34325
    },
    {
      "epoch": 3.2734870910644534e-05,
      "step": 34325,
      "training_step_time": 7.786462306976318
    },
    {
      "epoch": 3.273582458496094e-05,
      "model_forward_time": 1.7030575275421143,
      "step": 34326
    },
    {
      "epoch": 3.273582458496094e-05,
      "step": 34326,
      "training_step_time": 8.443861246109009
    },
    {
      "epoch": 3.273677825927734e-05,
      "model_forward_time": 1.7019224166870117,
      "step": 34327
    },
    {
      "epoch": 3.273677825927734e-05,
      "step": 34327,
      "training_step_time": 8.687567234039307
    },
    {
      "epoch": 3.273773193359375e-05,
      "model_forward_time": 1.7117891311645508,
      "step": 34328
    },
    {
      "epoch": 3.273773193359375e-05,
      "step": 34328,
      "training_step_time": 7.776759386062622
    },
    {
      "epoch": 3.273868560791016e-05,
      "model_forward_time": 1.704164743423462,
      "step": 34329
    },
    {
      "epoch": 3.273868560791016e-05,
      "step": 34329,
      "training_step_time": 8.293602705001831
    },
    {
      "epoch": 3.273963928222656e-05,
      "grad_norm": 0.18763341009616852,
      "learning_rate": 7.826941734564363e-06,
      "loss": 0.0613,
      "step": 34330
    },
    {
      "epoch": 3.273963928222656e-05,
      "model_forward_time": 1.7199723720550537,
      "step": 34330
    },
    {
      "dynamics_loss_avg": 0.060329965129494664,
      "epoch": 3.273963928222656e-05,
      "step": 34330
    },
    {
      "action_loss_avg": 0.0016335068503394723,
      "epoch": 3.273963928222656e-05,
      "step": 34330
    },
    {
      "epoch": 3.273963928222656e-05,
      "step": 34330,
      "training_step_time": 9.865171432495117
    },
    {
      "epoch": 3.274059295654297e-05,
      "model_forward_time": 1.7050118446350098,
      "step": 34331
    },
    {
      "epoch": 3.274059295654297e-05,
      "step": 34331,
      "training_step_time": 7.648193597793579
    },
    {
      "epoch": 3.2741546630859376e-05,
      "model_forward_time": 1.6958622932434082,
      "step": 34332
    },
    {
      "epoch": 3.2741546630859376e-05,
      "step": 34332,
      "training_step_time": 7.753468990325928
    },
    {
      "epoch": 3.274250030517578e-05,
      "model_forward_time": 1.687563419342041,
      "step": 34333
    },
    {
      "epoch": 3.274250030517578e-05,
      "step": 34333,
      "training_step_time": 7.64682149887085
    },
    {
      "epoch": 3.274345397949219e-05,
      "model_forward_time": 2.270906925201416,
      "step": 34334
    },
    {
      "epoch": 3.274345397949219e-05,
      "step": 34334,
      "training_step_time": 7.904523849487305
    },
    {
      "epoch": 3.2744407653808595e-05,
      "model_forward_time": 1.693549394607544,
      "step": 34335
    },
    {
      "epoch": 3.2744407653808595e-05,
      "step": 34335,
      "training_step_time": 8.037503957748413
    },
    {
      "epoch": 3.2745361328125e-05,
      "model_forward_time": 1.6916069984436035,
      "step": 34336
    },
    {
      "epoch": 3.2745361328125e-05,
      "step": 34336,
      "training_step_time": 6.990339040756226
    },
    {
      "epoch": 3.274631500244141e-05,
      "model_forward_time": 1.6918985843658447,
      "step": 34337
    },
    {
      "epoch": 3.274631500244141e-05,
      "step": 34337,
      "training_step_time": 6.987795829772949
    },
    {
      "epoch": 3.2747268676757813e-05,
      "model_forward_time": 1.6940443515777588,
      "step": 34338
    },
    {
      "epoch": 3.2747268676757813e-05,
      "step": 34338,
      "training_step_time": 7.032951354980469
    },
    {
      "epoch": 3.274822235107422e-05,
      "model_forward_time": 1.6945648193359375,
      "step": 34339
    },
    {
      "epoch": 3.274822235107422e-05,
      "step": 34339,
      "training_step_time": 7.025105714797974
    },
    {
      "epoch": 3.274917602539063e-05,
      "grad_norm": 0.17126567661762238,
      "learning_rate": 7.825577757035865e-06,
      "loss": 0.0626,
      "step": 34340
    },
    {
      "epoch": 3.274917602539063e-05,
      "model_forward_time": 1.6930208206176758,
      "step": 34340
    },
    {
      "dynamics_loss_avg": 0.0697068501263857,
      "epoch": 3.274917602539063e-05,
      "step": 34340
    },
    {
      "action_loss_avg": 0.0023201794014312328,
      "epoch": 3.274917602539063e-05,
      "step": 34340
    },
    {
      "epoch": 3.274917602539063e-05,
      "step": 34340,
      "training_step_time": 6.944844007492065
    },
    {
      "epoch": 3.275012969970703e-05,
      "model_forward_time": 1.693809986114502,
      "step": 34341
    },
    {
      "epoch": 3.275012969970703e-05,
      "step": 34341,
      "training_step_time": 7.017865896224976
    },
    {
      "epoch": 3.2751083374023436e-05,
      "model_forward_time": 1.6910133361816406,
      "step": 34342
    },
    {
      "epoch": 3.2751083374023436e-05,
      "step": 34342,
      "training_step_time": 6.93472957611084
    },
    {
      "epoch": 3.275203704833985e-05,
      "model_forward_time": 2.2606890201568604,
      "step": 34343
    },
    {
      "epoch": 3.275203704833985e-05,
      "step": 34343,
      "training_step_time": 9.127447366714478
    },
    {
      "epoch": 3.275299072265625e-05,
      "model_forward_time": 1.6971092224121094,
      "step": 34344
    },
    {
      "epoch": 3.275299072265625e-05,
      "step": 34344,
      "training_step_time": 7.902320861816406
    },
    {
      "epoch": 3.2753944396972655e-05,
      "model_forward_time": 1.6897714138031006,
      "step": 34345
    },
    {
      "epoch": 3.2753944396972655e-05,
      "step": 34345,
      "training_step_time": 7.87044358253479
    },
    {
      "epoch": 3.275489807128906e-05,
      "model_forward_time": 1.6968443393707275,
      "step": 34346
    },
    {
      "epoch": 3.275489807128906e-05,
      "step": 34346,
      "training_step_time": 7.069173812866211
    },
    {
      "epoch": 3.275585174560547e-05,
      "model_forward_time": 1.695080280303955,
      "step": 34347
    },
    {
      "epoch": 3.275585174560547e-05,
      "step": 34347,
      "training_step_time": 6.995592832565308
    },
    {
      "epoch": 3.2756805419921874e-05,
      "model_forward_time": 1.6938936710357666,
      "step": 34348
    },
    {
      "epoch": 3.2756805419921874e-05,
      "step": 34348,
      "training_step_time": 6.899014949798584
    },
    {
      "epoch": 3.275775909423828e-05,
      "model_forward_time": 1.6886951923370361,
      "step": 34349
    },
    {
      "epoch": 3.275775909423828e-05,
      "step": 34349,
      "training_step_time": 6.927630662918091
    },
    {
      "epoch": 3.275871276855469e-05,
      "grad_norm": 0.22977302968502045,
      "learning_rate": 7.82421347050643e-06,
      "loss": 0.0617,
      "step": 34350
    },
    {
      "epoch": 3.275871276855469e-05,
      "model_forward_time": 1.687896490097046,
      "step": 34350
    },
    {
      "dynamics_loss_avg": 0.05849821586161852,
      "epoch": 3.275871276855469e-05,
      "step": 34350
    },
    {
      "action_loss_avg": 0.0020479535567574205,
      "epoch": 3.275871276855469e-05,
      "step": 34350
    },
    {
      "epoch": 3.275871276855469e-05,
      "step": 34350,
      "training_step_time": 6.869874715805054
    },
    {
      "epoch": 3.275966644287109e-05,
      "model_forward_time": 1.6970407962799072,
      "step": 34351
    },
    {
      "epoch": 3.275966644287109e-05,
      "step": 34351,
      "training_step_time": 6.870587110519409
    },
    {
      "epoch": 3.27606201171875e-05,
      "model_forward_time": 2.2412936687469482,
      "step": 34352
    },
    {
      "epoch": 3.27606201171875e-05,
      "step": 34352,
      "training_step_time": 7.687031030654907
    },
    {
      "epoch": 3.276157379150391e-05,
      "model_forward_time": 1.7071280479431152,
      "step": 34353
    },
    {
      "epoch": 3.276157379150391e-05,
      "step": 34353,
      "training_step_time": 10.190555095672607
    },
    {
      "epoch": 3.276252746582031e-05,
      "model_forward_time": 1.6916401386260986,
      "step": 34354
    },
    {
      "epoch": 3.276252746582031e-05,
      "step": 34354,
      "training_step_time": 7.829857349395752
    },
    {
      "epoch": 3.2763481140136716e-05,
      "model_forward_time": 1.69405198097229,
      "step": 34355
    },
    {
      "epoch": 3.2763481140136716e-05,
      "step": 34355,
      "training_step_time": 6.863362073898315
    },
    {
      "epoch": 3.2764434814453127e-05,
      "model_forward_time": 1.6898667812347412,
      "step": 34356
    },
    {
      "epoch": 3.2764434814453127e-05,
      "step": 34356,
      "training_step_time": 6.878064393997192
    },
    {
      "epoch": 3.276538848876953e-05,
      "model_forward_time": 1.6929576396942139,
      "step": 34357
    },
    {
      "epoch": 3.276538848876953e-05,
      "step": 34357,
      "training_step_time": 6.894386291503906
    },
    {
      "epoch": 3.2766342163085935e-05,
      "model_forward_time": 1.6941990852355957,
      "step": 34358
    },
    {
      "epoch": 3.2766342163085935e-05,
      "step": 34358,
      "training_step_time": 6.880331754684448
    },
    {
      "epoch": 3.2767295837402345e-05,
      "model_forward_time": 1.6904854774475098,
      "step": 34359
    },
    {
      "epoch": 3.2767295837402345e-05,
      "step": 34359,
      "training_step_time": 6.896708011627197
    },
    {
      "epoch": 3.276824951171875e-05,
      "grad_norm": 0.19851529598236084,
      "learning_rate": 7.822848875125258e-06,
      "loss": 0.061,
      "step": 34360
    },
    {
      "epoch": 3.276824951171875e-05,
      "model_forward_time": 1.6936473846435547,
      "step": 34360
    },
    {
      "dynamics_loss_avg": 0.04292872678488493,
      "epoch": 3.276824951171875e-05,
      "step": 34360
    },
    {
      "action_loss_avg": 0.0018118885986041279,
      "epoch": 3.276824951171875e-05,
      "step": 34360
    },
    {
      "epoch": 3.276824951171875e-05,
      "step": 34360,
      "training_step_time": 6.881037473678589
    },
    {
      "epoch": 3.2769203186035153e-05,
      "model_forward_time": 0.9793922901153564,
      "step": 34361
    },
    {
      "epoch": 3.2769203186035153e-05,
      "step": 34361,
      "training_step_time": 7.889650344848633
    },
    {
      "epoch": 3.2770156860351564e-05,
      "model_forward_time": 2.253617286682129,
      "step": 34362
    },
    {
      "epoch": 3.2770156860351564e-05,
      "step": 34362,
      "training_step_time": 12.327389240264893
    },
    {
      "epoch": 3.277111053466797e-05,
      "model_forward_time": 1.692791223526001,
      "step": 34363
    },
    {
      "epoch": 3.277111053466797e-05,
      "step": 34363,
      "training_step_time": 7.692062616348267
    },
    {
      "epoch": 3.277206420898437e-05,
      "model_forward_time": 1.692284107208252,
      "step": 34364
    },
    {
      "epoch": 3.277206420898437e-05,
      "step": 34364,
      "training_step_time": 6.8153462409973145
    },
    {
      "epoch": 3.277301788330078e-05,
      "model_forward_time": 1.3227336406707764,
      "step": 34365
    },
    {
      "epoch": 3.277301788330078e-05,
      "step": 34365,
      "training_step_time": 6.893170118331909
    },
    {
      "epoch": 3.277397155761719e-05,
      "model_forward_time": 1.6931946277618408,
      "step": 34366
    },
    {
      "epoch": 3.277397155761719e-05,
      "step": 34366,
      "training_step_time": 6.925991535186768
    },
    {
      "epoch": 3.277492523193359e-05,
      "model_forward_time": 1.6957762241363525,
      "step": 34367
    },
    {
      "epoch": 3.277492523193359e-05,
      "step": 34367,
      "training_step_time": 6.867299556732178
    },
    {
      "epoch": 3.277587890625e-05,
      "model_forward_time": 1.698667049407959,
      "step": 34368
    },
    {
      "epoch": 3.277587890625e-05,
      "step": 34368,
      "training_step_time": 6.946069002151489
    },
    {
      "epoch": 3.2776832580566406e-05,
      "model_forward_time": 1.6963021755218506,
      "step": 34369
    },
    {
      "epoch": 3.2776832580566406e-05,
      "step": 34369,
      "training_step_time": 6.74602484703064
    },
    {
      "epoch": 3.277778625488281e-05,
      "grad_norm": 0.14218558371067047,
      "learning_rate": 7.821483971041577e-06,
      "loss": 0.0598,
      "step": 34370
    },
    {
      "epoch": 3.277778625488281e-05,
      "model_forward_time": 1.6911993026733398,
      "step": 34370
    },
    {
      "dynamics_loss_avg": 0.059885415248572826,
      "epoch": 3.277778625488281e-05,
      "step": 34370
    },
    {
      "action_loss_avg": 0.0023796424153260886,
      "epoch": 3.277778625488281e-05,
      "step": 34370
    },
    {
      "epoch": 3.277778625488281e-05,
      "step": 34370,
      "training_step_time": 6.826814413070679
    },
    {
      "epoch": 3.277873992919922e-05,
      "model_forward_time": 2.256113290786743,
      "step": 34371
    },
    {
      "epoch": 3.277873992919922e-05,
      "step": 34371,
      "training_step_time": 10.604827642440796
    },
    {
      "epoch": 3.2779693603515625e-05,
      "model_forward_time": 1.686593770980835,
      "step": 34372
    },
    {
      "epoch": 3.2779693603515625e-05,
      "step": 34372,
      "training_step_time": 8.820874214172363
    },
    {
      "epoch": 3.278064727783203e-05,
      "model_forward_time": 1.6963169574737549,
      "step": 34373
    },
    {
      "epoch": 3.278064727783203e-05,
      "step": 34373,
      "training_step_time": 6.9733617305755615
    },
    {
      "epoch": 3.278160095214844e-05,
      "model_forward_time": 1.6871325969696045,
      "step": 34374
    },
    {
      "epoch": 3.278160095214844e-05,
      "step": 34374,
      "training_step_time": 6.944913148880005
    },
    {
      "epoch": 3.2782554626464844e-05,
      "model_forward_time": 1.697805643081665,
      "step": 34375
    },
    {
      "epoch": 3.2782554626464844e-05,
      "step": 34375,
      "training_step_time": 6.979048490524292
    },
    {
      "epoch": 3.278350830078125e-05,
      "model_forward_time": 1.6888327598571777,
      "step": 34376
    },
    {
      "epoch": 3.278350830078125e-05,
      "step": 34376,
      "training_step_time": 6.992899179458618
    },
    {
      "epoch": 3.278446197509766e-05,
      "model_forward_time": 1.6899099349975586,
      "step": 34377
    },
    {
      "epoch": 3.278446197509766e-05,
      "step": 34377,
      "training_step_time": 6.984949588775635
    },
    {
      "epoch": 3.278541564941406e-05,
      "model_forward_time": 1.6974127292633057,
      "step": 34378
    },
    {
      "epoch": 3.278541564941406e-05,
      "step": 34378,
      "training_step_time": 7.017348527908325
    },
    {
      "epoch": 3.278636932373047e-05,
      "model_forward_time": 1.6936323642730713,
      "step": 34379
    },
    {
      "epoch": 3.278636932373047e-05,
      "step": 34379,
      "training_step_time": 7.023185729980469
    },
    {
      "epoch": 3.278732299804688e-05,
      "grad_norm": 0.10021819174289703,
      "learning_rate": 7.820118758404648e-06,
      "loss": 0.0588,
      "step": 34380
    },
    {
      "epoch": 3.278732299804688e-05,
      "model_forward_time": 2.239030361175537,
      "step": 34380
    },
    {
      "dynamics_loss_avg": 0.04053029790520668,
      "epoch": 3.278732299804688e-05,
      "step": 34380
    },
    {
      "action_loss_avg": 0.0011466454539913683,
      "epoch": 3.278732299804688e-05,
      "step": 34380
    },
    {
      "epoch": 3.278732299804688e-05,
      "step": 34380,
      "training_step_time": 9.681151628494263
    },
    {
      "epoch": 3.278827667236328e-05,
      "model_forward_time": 1.6960217952728271,
      "step": 34381
    },
    {
      "epoch": 3.278827667236328e-05,
      "step": 34381,
      "training_step_time": 8.381184816360474
    },
    {
      "epoch": 3.2789230346679685e-05,
      "model_forward_time": 1.6925723552703857,
      "step": 34382
    },
    {
      "epoch": 3.2789230346679685e-05,
      "step": 34382,
      "training_step_time": 7.777578592300415
    },
    {
      "epoch": 3.2790184020996096e-05,
      "model_forward_time": 1.694370985031128,
      "step": 34383
    },
    {
      "epoch": 3.2790184020996096e-05,
      "step": 34383,
      "training_step_time": 6.926575422286987
    },
    {
      "epoch": 3.27911376953125e-05,
      "model_forward_time": 1.703211784362793,
      "step": 34384
    },
    {
      "epoch": 3.27911376953125e-05,
      "step": 34384,
      "training_step_time": 6.9086525440216064
    },
    {
      "epoch": 3.2792091369628904e-05,
      "model_forward_time": 1.7032265663146973,
      "step": 34385
    },
    {
      "epoch": 3.2792091369628904e-05,
      "step": 34385,
      "training_step_time": 6.971997022628784
    },
    {
      "epoch": 3.2793045043945315e-05,
      "model_forward_time": 1.6965618133544922,
      "step": 34386
    },
    {
      "epoch": 3.2793045043945315e-05,
      "step": 34386,
      "training_step_time": 6.892248868942261
    },
    {
      "epoch": 3.279399871826172e-05,
      "model_forward_time": 1.6946465969085693,
      "step": 34387
    },
    {
      "epoch": 3.279399871826172e-05,
      "step": 34387,
      "training_step_time": 6.845829963684082
    },
    {
      "epoch": 3.279495239257812e-05,
      "model_forward_time": 1.69626784324646,
      "step": 34388
    },
    {
      "epoch": 3.279495239257812e-05,
      "step": 34388,
      "training_step_time": 6.924323797225952
    },
    {
      "epoch": 3.2795906066894534e-05,
      "model_forward_time": 1.706489086151123,
      "step": 34389
    },
    {
      "epoch": 3.2795906066894534e-05,
      "step": 34389,
      "training_step_time": 8.320844173431396
    },
    {
      "epoch": 3.279685974121094e-05,
      "grad_norm": 0.18428345024585724,
      "learning_rate": 7.818753237363777e-06,
      "loss": 0.0614,
      "step": 34390
    },
    {
      "epoch": 3.279685974121094e-05,
      "model_forward_time": 2.22556471824646,
      "step": 34390
    },
    {
      "dynamics_loss_avg": 0.0614573311060667,
      "epoch": 3.279685974121094e-05,
      "step": 34390
    },
    {
      "action_loss_avg": 0.0015196113032288848,
      "epoch": 3.279685974121094e-05,
      "step": 34390
    },
    {
      "epoch": 3.279685974121094e-05,
      "step": 34390,
      "training_step_time": 9.097231149673462
    },
    {
      "epoch": 3.279781341552734e-05,
      "model_forward_time": 1.7002997398376465,
      "step": 34391
    },
    {
      "epoch": 3.279781341552734e-05,
      "step": 34391,
      "training_step_time": 7.83670973777771
    },
    {
      "epoch": 3.279876708984375e-05,
      "model_forward_time": 1.7003953456878662,
      "step": 34392
    },
    {
      "epoch": 3.279876708984375e-05,
      "step": 34392,
      "training_step_time": 6.9067699909210205
    },
    {
      "epoch": 3.279972076416016e-05,
      "model_forward_time": 1.71272873878479,
      "step": 34393
    },
    {
      "epoch": 3.279972076416016e-05,
      "step": 34393,
      "training_step_time": 6.903947353363037
    },
    {
      "epoch": 3.280067443847656e-05,
      "model_forward_time": 1.3369312286376953,
      "step": 34394
    },
    {
      "epoch": 3.280067443847656e-05,
      "step": 34394,
      "training_step_time": 6.911387920379639
    },
    {
      "epoch": 3.280162811279297e-05,
      "model_forward_time": 1.710160732269287,
      "step": 34395
    },
    {
      "epoch": 3.280162811279297e-05,
      "step": 34395,
      "training_step_time": 6.926311731338501
    },
    {
      "epoch": 3.2802581787109376e-05,
      "model_forward_time": 1.7187607288360596,
      "step": 34396
    },
    {
      "epoch": 3.2802581787109376e-05,
      "step": 34396,
      "training_step_time": 6.887023448944092
    },
    {
      "epoch": 3.280353546142578e-05,
      "model_forward_time": 1.7139370441436768,
      "step": 34397
    },
    {
      "epoch": 3.280353546142578e-05,
      "step": 34397,
      "training_step_time": 6.927252292633057
    },
    {
      "epoch": 3.280448913574219e-05,
      "model_forward_time": 1.7288928031921387,
      "step": 34398
    },
    {
      "epoch": 3.280448913574219e-05,
      "step": 34398,
      "training_step_time": 7.65921688079834
    },
    {
      "epoch": 3.2805442810058595e-05,
      "model_forward_time": 2.2853803634643555,
      "step": 34399
    },
    {
      "epoch": 3.2805442810058595e-05,
      "step": 34399,
      "training_step_time": 8.57092833518982
    },
    {
      "epoch": 3.2806396484375e-05,
      "grad_norm": 0.13261009752750397,
      "learning_rate": 7.817387408068287e-06,
      "loss": 0.0619,
      "step": 34400
    },
    {
      "epoch": 3.2806396484375e-05,
      "model_forward_time": 1.7106492519378662,
      "step": 34400
    },
    {
      "dynamics_loss_avg": 0.07184604927897453,
      "epoch": 3.2806396484375e-05,
      "step": 34400
    },
    {
      "action_loss_avg": 0.0022099636902567,
      "epoch": 3.2806396484375e-05,
      "step": 34400
    },
    {
      "epoch": 3.2806396484375e-05,
      "step": 34400,
      "training_step_time": 8.07904577255249
    },
    {
      "epoch": 3.280735015869141e-05,
      "model_forward_time": 1.7136223316192627,
      "step": 34401
    },
    {
      "epoch": 3.280735015869141e-05,
      "step": 34401,
      "training_step_time": 6.900357961654663
    },
    {
      "epoch": 3.2808303833007813e-05,
      "model_forward_time": 1.7108983993530273,
      "step": 34402
    },
    {
      "epoch": 3.2808303833007813e-05,
      "step": 34402,
      "training_step_time": 6.889559507369995
    },
    {
      "epoch": 3.280925750732422e-05,
      "model_forward_time": 1.691967248916626,
      "step": 34403
    },
    {
      "epoch": 3.280925750732422e-05,
      "step": 34403,
      "training_step_time": 6.897094011306763
    },
    {
      "epoch": 3.281021118164063e-05,
      "model_forward_time": 1.6978516578674316,
      "step": 34404
    },
    {
      "epoch": 3.281021118164063e-05,
      "step": 34404,
      "training_step_time": 6.939610481262207
    },
    {
      "epoch": 3.281116485595703e-05,
      "model_forward_time": 1.6958742141723633,
      "step": 34405
    },
    {
      "epoch": 3.281116485595703e-05,
      "step": 34405,
      "training_step_time": 6.913989067077637
    },
    {
      "epoch": 3.2812118530273436e-05,
      "model_forward_time": 1.6959044933319092,
      "step": 34406
    },
    {
      "epoch": 3.2812118530273436e-05,
      "step": 34406,
      "training_step_time": 6.899412631988525
    },
    {
      "epoch": 3.281307220458984e-05,
      "model_forward_time": 1.6944141387939453,
      "step": 34407
    },
    {
      "epoch": 3.281307220458984e-05,
      "step": 34407,
      "training_step_time": 7.560242414474487
    },
    {
      "epoch": 3.281402587890625e-05,
      "model_forward_time": 2.2892680168151855,
      "step": 34408
    },
    {
      "epoch": 3.281402587890625e-05,
      "step": 34408,
      "training_step_time": 8.505429983139038
    },
    {
      "epoch": 3.2814979553222655e-05,
      "model_forward_time": 1.6919190883636475,
      "step": 34409
    },
    {
      "epoch": 3.2814979553222655e-05,
      "step": 34409,
      "training_step_time": 8.034452199935913
    },
    {
      "epoch": 3.281593322753906e-05,
      "grad_norm": 0.11612290889024734,
      "learning_rate": 7.816021270667544e-06,
      "loss": 0.0596,
      "step": 34410
    },
    {
      "epoch": 3.281593322753906e-05,
      "model_forward_time": 1.689626693725586,
      "step": 34410
    },
    {
      "dynamics_loss_avg": 0.051447713188827036,
      "epoch": 3.281593322753906e-05,
      "step": 34410
    },
    {
      "action_loss_avg": 0.0014857771864626556,
      "epoch": 3.281593322753906e-05,
      "step": 34410
    },
    {
      "epoch": 3.281593322753906e-05,
      "step": 34410,
      "training_step_time": 7.351989984512329
    },
    {
      "epoch": 3.281688690185547e-05,
      "model_forward_time": 1.698329210281372,
      "step": 34411
    },
    {
      "epoch": 3.281688690185547e-05,
      "step": 34411,
      "training_step_time": 6.882157325744629
    },
    {
      "epoch": 3.2817840576171874e-05,
      "model_forward_time": 1.6893203258514404,
      "step": 34412
    },
    {
      "epoch": 3.2817840576171874e-05,
      "step": 34412,
      "training_step_time": 6.926292181015015
    },
    {
      "epoch": 3.281879425048828e-05,
      "model_forward_time": 1.6947236061096191,
      "step": 34413
    },
    {
      "epoch": 3.281879425048828e-05,
      "step": 34413,
      "training_step_time": 6.877461671829224
    },
    {
      "epoch": 3.281974792480469e-05,
      "model_forward_time": 1.6978471279144287,
      "step": 34414
    },
    {
      "epoch": 3.281974792480469e-05,
      "step": 34414,
      "training_step_time": 6.871567010879517
    },
    {
      "epoch": 3.282070159912109e-05,
      "model_forward_time": 1.6994411945343018,
      "step": 34415
    },
    {
      "epoch": 3.282070159912109e-05,
      "step": 34415,
      "training_step_time": 6.850318670272827
    },
    {
      "epoch": 3.28216552734375e-05,
      "model_forward_time": 1.698958158493042,
      "step": 34416
    },
    {
      "epoch": 3.28216552734375e-05,
      "step": 34416,
      "training_step_time": 7.54181981086731
    },
    {
      "epoch": 3.282260894775391e-05,
      "model_forward_time": 2.2893588542938232,
      "step": 34417
    },
    {
      "epoch": 3.282260894775391e-05,
      "step": 34417,
      "training_step_time": 10.447237968444824
    },
    {
      "epoch": 3.282356262207031e-05,
      "model_forward_time": 1.7103183269500732,
      "step": 34418
    },
    {
      "epoch": 3.282356262207031e-05,
      "step": 34418,
      "training_step_time": 7.8838934898376465
    },
    {
      "epoch": 3.2824516296386716e-05,
      "model_forward_time": 1.707880973815918,
      "step": 34419
    },
    {
      "epoch": 3.2824516296386716e-05,
      "step": 34419,
      "training_step_time": 7.595580577850342
    },
    {
      "epoch": 3.282546997070313e-05,
      "grad_norm": 0.11841350793838501,
      "learning_rate": 7.81465482531095e-06,
      "loss": 0.0594,
      "step": 34420
    },
    {
      "epoch": 3.282546997070313e-05,
      "model_forward_time": 1.7017474174499512,
      "step": 34420
    },
    {
      "dynamics_loss_avg": 0.04061268549412489,
      "epoch": 3.282546997070313e-05,
      "step": 34420
    },
    {
      "action_loss_avg": 0.0018710193020524458,
      "epoch": 3.282546997070313e-05,
      "step": 34420
    },
    {
      "epoch": 3.282546997070313e-05,
      "step": 34420,
      "training_step_time": 6.937239646911621
    },
    {
      "epoch": 3.282642364501953e-05,
      "model_forward_time": 1.7116353511810303,
      "step": 34421
    },
    {
      "epoch": 3.282642364501953e-05,
      "step": 34421,
      "training_step_time": 6.971908807754517
    },
    {
      "epoch": 3.2827377319335935e-05,
      "model_forward_time": 1.7052395343780518,
      "step": 34422
    },
    {
      "epoch": 3.2827377319335935e-05,
      "step": 34422,
      "training_step_time": 6.992506265640259
    },
    {
      "epoch": 3.2828330993652345e-05,
      "model_forward_time": 1.7066211700439453,
      "step": 34423
    },
    {
      "epoch": 3.2828330993652345e-05,
      "step": 34423,
      "training_step_time": 6.932484865188599
    },
    {
      "epoch": 3.282928466796875e-05,
      "model_forward_time": 1.710883378982544,
      "step": 34424
    },
    {
      "epoch": 3.282928466796875e-05,
      "step": 34424,
      "training_step_time": 6.977620840072632
    },
    {
      "epoch": 3.2830238342285154e-05,
      "model_forward_time": 1.708650827407837,
      "step": 34425
    },
    {
      "epoch": 3.2830238342285154e-05,
      "step": 34425,
      "training_step_time": 7.506919860839844
    },
    {
      "epoch": 3.2831192016601564e-05,
      "model_forward_time": 2.3090057373046875,
      "step": 34426
    },
    {
      "epoch": 3.2831192016601564e-05,
      "step": 34426,
      "training_step_time": 8.182346820831299
    },
    {
      "epoch": 3.283214569091797e-05,
      "model_forward_time": 1.7019622325897217,
      "step": 34427
    },
    {
      "epoch": 3.283214569091797e-05,
      "step": 34427,
      "training_step_time": 9.988523721694946
    },
    {
      "epoch": 3.283309936523437e-05,
      "model_forward_time": 1.7074737548828125,
      "step": 34428
    },
    {
      "epoch": 3.283309936523437e-05,
      "step": 34428,
      "training_step_time": 7.653974533081055
    },
    {
      "epoch": 3.283405303955078e-05,
      "model_forward_time": 1.7100090980529785,
      "step": 34429
    },
    {
      "epoch": 3.283405303955078e-05,
      "step": 34429,
      "training_step_time": 6.995991468429565
    },
    {
      "epoch": 3.283500671386719e-05,
      "grad_norm": 0.1505926549434662,
      "learning_rate": 7.813288072147938e-06,
      "loss": 0.0605,
      "step": 34430
    },
    {
      "epoch": 3.283500671386719e-05,
      "model_forward_time": 1.7094838619232178,
      "step": 34430
    },
    {
      "dynamics_loss_avg": 0.04910583114251495,
      "epoch": 3.283500671386719e-05,
      "step": 34430
    },
    {
      "action_loss_avg": 0.001202004501828924,
      "epoch": 3.283500671386719e-05,
      "step": 34430
    },
    {
      "epoch": 3.283500671386719e-05,
      "step": 34430,
      "training_step_time": 6.904319763183594
    },
    {
      "epoch": 3.283596038818359e-05,
      "model_forward_time": 1.728987693786621,
      "step": 34431
    },
    {
      "epoch": 3.283596038818359e-05,
      "step": 34431,
      "training_step_time": 6.950991153717041
    },
    {
      "epoch": 3.28369140625e-05,
      "model_forward_time": 1.7327051162719727,
      "step": 34432
    },
    {
      "epoch": 3.28369140625e-05,
      "step": 34432,
      "training_step_time": 6.974479675292969
    },
    {
      "epoch": 3.2837867736816406e-05,
      "model_forward_time": 1.7232332229614258,
      "step": 34433
    },
    {
      "epoch": 3.2837867736816406e-05,
      "step": 34433,
      "training_step_time": 6.866888523101807
    },
    {
      "epoch": 3.283882141113281e-05,
      "model_forward_time": 1.7302539348602295,
      "step": 34434
    },
    {
      "epoch": 3.283882141113281e-05,
      "step": 34434,
      "training_step_time": 7.5040974617004395
    },
    {
      "epoch": 3.283977508544922e-05,
      "model_forward_time": 1.7410073280334473,
      "step": 34435
    },
    {
      "epoch": 3.283977508544922e-05,
      "step": 34435,
      "training_step_time": 9.215227842330933
    },
    {
      "epoch": 3.2840728759765625e-05,
      "model_forward_time": 1.7276873588562012,
      "step": 34436
    },
    {
      "epoch": 3.2840728759765625e-05,
      "step": 34436,
      "training_step_time": 9.555642127990723
    },
    {
      "epoch": 3.284168243408203e-05,
      "model_forward_time": 1.727135181427002,
      "step": 34437
    },
    {
      "epoch": 3.284168243408203e-05,
      "step": 34437,
      "training_step_time": 7.943761110305786
    },
    {
      "epoch": 3.284263610839844e-05,
      "model_forward_time": 1.7263686656951904,
      "step": 34438
    },
    {
      "epoch": 3.284263610839844e-05,
      "step": 34438,
      "training_step_time": 6.917497634887695
    },
    {
      "epoch": 3.2843589782714844e-05,
      "model_forward_time": 1.7265126705169678,
      "step": 34439
    },
    {
      "epoch": 3.2843589782714844e-05,
      "step": 34439,
      "training_step_time": 6.885145902633667
    },
    {
      "epoch": 3.284454345703125e-05,
      "grad_norm": 0.17482607066631317,
      "learning_rate": 7.811921011327973e-06,
      "loss": 0.0601,
      "step": 34440
    },
    {
      "epoch": 3.284454345703125e-05,
      "model_forward_time": 1.7282581329345703,
      "step": 34440
    },
    {
      "dynamics_loss_avg": 0.05633194651454687,
      "epoch": 3.284454345703125e-05,
      "step": 34440
    },
    {
      "action_loss_avg": 0.0025545852724462746,
      "epoch": 3.284454345703125e-05,
      "step": 34440
    },
    {
      "epoch": 3.284454345703125e-05,
      "step": 34440,
      "training_step_time": 6.883442640304565
    },
    {
      "epoch": 3.284549713134766e-05,
      "model_forward_time": 1.7317850589752197,
      "step": 34441
    },
    {
      "epoch": 3.284549713134766e-05,
      "step": 34441,
      "training_step_time": 6.934204816818237
    },
    {
      "epoch": 3.284645080566406e-05,
      "model_forward_time": 1.7308416366577148,
      "step": 34442
    },
    {
      "epoch": 3.284645080566406e-05,
      "step": 34442,
      "training_step_time": 6.8391265869140625
    },
    {
      "epoch": 3.284740447998047e-05,
      "model_forward_time": 1.7528514862060547,
      "step": 34443
    },
    {
      "epoch": 3.284740447998047e-05,
      "step": 34443,
      "training_step_time": 7.518476247787476
    },
    {
      "epoch": 3.284835815429688e-05,
      "model_forward_time": 1.7298808097839355,
      "step": 34444
    },
    {
      "epoch": 3.284835815429688e-05,
      "step": 34444,
      "training_step_time": 8.386375427246094
    },
    {
      "epoch": 3.284931182861328e-05,
      "model_forward_time": 2.3582777976989746,
      "step": 34445
    },
    {
      "epoch": 3.284931182861328e-05,
      "step": 34445,
      "training_step_time": 8.464420557022095
    },
    {
      "epoch": 3.2850265502929686e-05,
      "model_forward_time": 1.7323668003082275,
      "step": 34446
    },
    {
      "epoch": 3.2850265502929686e-05,
      "step": 34446,
      "training_step_time": 7.916445970535278
    },
    {
      "epoch": 3.2851219177246096e-05,
      "model_forward_time": 1.7235491275787354,
      "step": 34447
    },
    {
      "epoch": 3.2851219177246096e-05,
      "step": 34447,
      "training_step_time": 7.432280778884888
    },
    {
      "epoch": 3.28521728515625e-05,
      "model_forward_time": 1.7367074489593506,
      "step": 34448
    },
    {
      "epoch": 3.28521728515625e-05,
      "step": 34448,
      "training_step_time": 6.869723081588745
    },
    {
      "epoch": 3.2853126525878904e-05,
      "model_forward_time": 1.7292613983154297,
      "step": 34449
    },
    {
      "epoch": 3.2853126525878904e-05,
      "step": 34449,
      "training_step_time": 6.953754663467407
    },
    {
      "epoch": 3.2854080200195315e-05,
      "grad_norm": 0.14039145410060883,
      "learning_rate": 7.81055364300055e-06,
      "loss": 0.0597,
      "step": 34450
    },
    {
      "epoch": 3.2854080200195315e-05,
      "model_forward_time": 1.7251102924346924,
      "step": 34450
    },
    {
      "dynamics_loss_avg": 0.05661362055689097,
      "epoch": 3.2854080200195315e-05,
      "step": 34450
    },
    {
      "action_loss_avg": 0.0013696934940526262,
      "epoch": 3.2854080200195315e-05,
      "step": 34450
    },
    {
      "epoch": 3.2854080200195315e-05,
      "step": 34450,
      "training_step_time": 6.887003660202026
    },
    {
      "epoch": 3.285503387451172e-05,
      "model_forward_time": 1.7346580028533936,
      "step": 34451
    },
    {
      "epoch": 3.285503387451172e-05,
      "step": 34451,
      "training_step_time": 6.834439039230347
    },
    {
      "epoch": 3.285598754882812e-05,
      "model_forward_time": 1.722867727279663,
      "step": 34452
    },
    {
      "epoch": 3.285598754882812e-05,
      "step": 34452,
      "training_step_time": 8.413492918014526
    },
    {
      "epoch": 3.2856941223144534e-05,
      "model_forward_time": 1.72920560836792,
      "step": 34453
    },
    {
      "epoch": 3.2856941223144534e-05,
      "step": 34453,
      "training_step_time": 7.708890914916992
    },
    {
      "epoch": 3.285789489746094e-05,
      "model_forward_time": 1.9755163192749023,
      "step": 34454
    },
    {
      "epoch": 3.285789489746094e-05,
      "step": 34454,
      "training_step_time": 9.660622358322144
    },
    {
      "epoch": 3.285884857177734e-05,
      "model_forward_time": 1.7233188152313232,
      "step": 34455
    },
    {
      "epoch": 3.285884857177734e-05,
      "step": 34455,
      "training_step_time": 7.869809150695801
    },
    {
      "epoch": 3.285980224609375e-05,
      "model_forward_time": 1.7258460521697998,
      "step": 34456
    },
    {
      "epoch": 3.285980224609375e-05,
      "step": 34456,
      "training_step_time": 7.907527685165405
    },
    {
      "epoch": 3.286075592041016e-05,
      "model_forward_time": 1.7275094985961914,
      "step": 34457
    },
    {
      "epoch": 3.286075592041016e-05,
      "step": 34457,
      "training_step_time": 6.892932891845703
    },
    {
      "epoch": 3.286170959472656e-05,
      "model_forward_time": 1.7301359176635742,
      "step": 34458
    },
    {
      "epoch": 3.286170959472656e-05,
      "step": 34458,
      "training_step_time": 6.93042778968811
    },
    {
      "epoch": 3.286266326904297e-05,
      "model_forward_time": 1.729647159576416,
      "step": 34459
    },
    {
      "epoch": 3.286266326904297e-05,
      "step": 34459,
      "training_step_time": 6.913173675537109
    },
    {
      "epoch": 3.2863616943359376e-05,
      "grad_norm": 0.1277455985546112,
      "learning_rate": 7.809185967315207e-06,
      "loss": 0.0581,
      "step": 34460
    },
    {
      "epoch": 3.2863616943359376e-05,
      "model_forward_time": 1.7274672985076904,
      "step": 34460
    },
    {
      "dynamics_loss_avg": 0.05849731806665659,
      "epoch": 3.2863616943359376e-05,
      "step": 34460
    },
    {
      "action_loss_avg": 0.0022944161552004515,
      "epoch": 3.2863616943359376e-05,
      "step": 34460
    },
    {
      "epoch": 3.2863616943359376e-05,
      "step": 34460,
      "training_step_time": 6.856798410415649
    },
    {
      "epoch": 3.286457061767578e-05,
      "model_forward_time": 1.725503921508789,
      "step": 34461
    },
    {
      "epoch": 3.286457061767578e-05,
      "step": 34461,
      "training_step_time": 7.519683361053467
    },
    {
      "epoch": 3.286552429199219e-05,
      "model_forward_time": 1.724600076675415,
      "step": 34462
    },
    {
      "epoch": 3.286552429199219e-05,
      "step": 34462,
      "training_step_time": 7.673075914382935
    },
    {
      "epoch": 3.2866477966308595e-05,
      "model_forward_time": 2.309534788131714,
      "step": 34463
    },
    {
      "epoch": 3.2866477966308595e-05,
      "step": 34463,
      "training_step_time": 8.741360664367676
    },
    {
      "epoch": 3.2867431640625e-05,
      "model_forward_time": 1.7243168354034424,
      "step": 34464
    },
    {
      "epoch": 3.2867431640625e-05,
      "step": 34464,
      "training_step_time": 8.052781581878662
    },
    {
      "epoch": 3.286838531494141e-05,
      "model_forward_time": 1.7310070991516113,
      "step": 34465
    },
    {
      "epoch": 3.286838531494141e-05,
      "step": 34465,
      "training_step_time": 7.612941741943359
    },
    {
      "epoch": 3.2869338989257814e-05,
      "model_forward_time": 1.7284281253814697,
      "step": 34466
    },
    {
      "epoch": 3.2869338989257814e-05,
      "step": 34466,
      "training_step_time": 6.903080224990845
    },
    {
      "epoch": 3.287029266357422e-05,
      "model_forward_time": 1.721895456314087,
      "step": 34467
    },
    {
      "epoch": 3.287029266357422e-05,
      "step": 34467,
      "training_step_time": 6.856731176376343
    },
    {
      "epoch": 3.287124633789063e-05,
      "model_forward_time": 1.7267963886260986,
      "step": 34468
    },
    {
      "epoch": 3.287124633789063e-05,
      "step": 34468,
      "training_step_time": 6.833212375640869
    },
    {
      "epoch": 3.287220001220703e-05,
      "model_forward_time": 1.7342400550842285,
      "step": 34469
    },
    {
      "epoch": 3.287220001220703e-05,
      "step": 34469,
      "training_step_time": 6.87009334564209
    },
    {
      "epoch": 3.2873153686523436e-05,
      "grad_norm": 0.2227596640586853,
      "learning_rate": 7.80781798442151e-06,
      "loss": 0.0621,
      "step": 34470
    },
    {
      "epoch": 3.2873153686523436e-05,
      "model_forward_time": 1.7394952774047852,
      "step": 34470
    },
    {
      "dynamics_loss_avg": 0.06733889821916819,
      "epoch": 3.2873153686523436e-05,
      "step": 34470
    },
    {
      "action_loss_avg": 0.0016891539096832275,
      "epoch": 3.2873153686523436e-05,
      "step": 34470
    },
    {
      "epoch": 3.2873153686523436e-05,
      "step": 34470,
      "training_step_time": 7.544741630554199
    },
    {
      "epoch": 3.287410736083984e-05,
      "model_forward_time": 1.7292613983154297,
      "step": 34471
    },
    {
      "epoch": 3.287410736083984e-05,
      "step": 34471,
      "training_step_time": 7.4067277908325195
    },
    {
      "epoch": 3.287506103515625e-05,
      "model_forward_time": 2.3323142528533936,
      "step": 34472
    },
    {
      "epoch": 3.287506103515625e-05,
      "step": 34472,
      "training_step_time": 9.600566625595093
    },
    {
      "epoch": 3.2876014709472655e-05,
      "model_forward_time": 1.734816312789917,
      "step": 34473
    },
    {
      "epoch": 3.2876014709472655e-05,
      "step": 34473,
      "training_step_time": 9.480849027633667
    },
    {
      "epoch": 3.287696838378906e-05,
      "model_forward_time": 1.7253434658050537,
      "step": 34474
    },
    {
      "epoch": 3.287696838378906e-05,
      "step": 34474,
      "training_step_time": 8.357360124588013
    },
    {
      "epoch": 3.287792205810547e-05,
      "model_forward_time": 1.7301719188690186,
      "step": 34475
    },
    {
      "epoch": 3.287792205810547e-05,
      "step": 34475,
      "training_step_time": 6.928988456726074
    },
    {
      "epoch": 3.2878875732421874e-05,
      "model_forward_time": 1.7196192741394043,
      "step": 34476
    },
    {
      "epoch": 3.2878875732421874e-05,
      "step": 34476,
      "training_step_time": 6.881176471710205
    },
    {
      "epoch": 3.287982940673828e-05,
      "model_forward_time": 1.7318248748779297,
      "step": 34477
    },
    {
      "epoch": 3.287982940673828e-05,
      "step": 34477,
      "training_step_time": 6.892255544662476
    },
    {
      "epoch": 3.288078308105469e-05,
      "model_forward_time": 1.7268497943878174,
      "step": 34478
    },
    {
      "epoch": 3.288078308105469e-05,
      "step": 34478,
      "training_step_time": 6.909238576889038
    },
    {
      "epoch": 3.288173675537109e-05,
      "model_forward_time": 1.717651128768921,
      "step": 34479
    },
    {
      "epoch": 3.288173675537109e-05,
      "step": 34479,
      "training_step_time": 7.544244050979614
    },
    {
      "epoch": 3.28826904296875e-05,
      "grad_norm": 0.17603133618831635,
      "learning_rate": 7.806449694469062e-06,
      "loss": 0.0603,
      "step": 34480
    },
    {
      "epoch": 3.28826904296875e-05,
      "model_forward_time": 1.725569725036621,
      "step": 34480
    },
    {
      "dynamics_loss_avg": 0.04502685600891709,
      "epoch": 3.28826904296875e-05,
      "step": 34480
    },
    {
      "action_loss_avg": 0.0010863764211535454,
      "epoch": 3.28826904296875e-05,
      "step": 34480
    },
    {
      "epoch": 3.28826904296875e-05,
      "step": 34480,
      "training_step_time": 7.42336893081665
    },
    {
      "epoch": 3.288364410400391e-05,
      "model_forward_time": 1.8112168312072754,
      "step": 34481
    },
    {
      "epoch": 3.288364410400391e-05,
      "step": 34481,
      "training_step_time": 7.678762197494507
    },
    {
      "epoch": 3.288459777832031e-05,
      "model_forward_time": 2.2842626571655273,
      "step": 34482
    },
    {
      "epoch": 3.288459777832031e-05,
      "step": 34482,
      "training_step_time": 8.51595425605774
    },
    {
      "epoch": 3.2885551452636716e-05,
      "model_forward_time": 1.728971004486084,
      "step": 34483
    },
    {
      "epoch": 3.2885551452636716e-05,
      "step": 34483,
      "training_step_time": 7.9410247802734375
    },
    {
      "epoch": 3.288650512695313e-05,
      "model_forward_time": 1.7303273677825928,
      "step": 34484
    },
    {
      "epoch": 3.288650512695313e-05,
      "step": 34484,
      "training_step_time": 7.653344631195068
    },
    {
      "epoch": 3.288745880126953e-05,
      "model_forward_time": 1.7316970825195312,
      "step": 34485
    },
    {
      "epoch": 3.288745880126953e-05,
      "step": 34485,
      "training_step_time": 6.868023157119751
    },
    {
      "epoch": 3.2888412475585935e-05,
      "model_forward_time": 1.7319543361663818,
      "step": 34486
    },
    {
      "epoch": 3.2888412475585935e-05,
      "step": 34486,
      "training_step_time": 6.881489515304565
    },
    {
      "epoch": 3.2889366149902346e-05,
      "model_forward_time": 1.7261772155761719,
      "step": 34487
    },
    {
      "epoch": 3.2889366149902346e-05,
      "step": 34487,
      "training_step_time": 6.917144775390625
    },
    {
      "epoch": 3.289031982421875e-05,
      "model_forward_time": 1.7307558059692383,
      "step": 34488
    },
    {
      "epoch": 3.289031982421875e-05,
      "step": 34488,
      "training_step_time": 7.528674125671387
    },
    {
      "epoch": 3.2891273498535154e-05,
      "model_forward_time": 1.7274866104125977,
      "step": 34489
    },
    {
      "epoch": 3.2891273498535154e-05,
      "step": 34489,
      "training_step_time": 7.3605687618255615
    },
    {
      "epoch": 3.2892227172851564e-05,
      "grad_norm": 0.10896989703178406,
      "learning_rate": 7.805081097607493e-06,
      "loss": 0.0623,
      "step": 34490
    },
    {
      "epoch": 3.2892227172851564e-05,
      "model_forward_time": 1.8046596050262451,
      "step": 34490
    },
    {
      "dynamics_loss_avg": 0.06480913236737251,
      "epoch": 3.2892227172851564e-05,
      "step": 34490
    },
    {
      "action_loss_avg": 0.0027727557288017126,
      "epoch": 3.2892227172851564e-05,
      "step": 34490
    },
    {
      "epoch": 3.2892227172851564e-05,
      "step": 34490,
      "training_step_time": 8.627688884735107
    },
    {
      "epoch": 3.289318084716797e-05,
      "model_forward_time": 2.2705330848693848,
      "step": 34491
    },
    {
      "epoch": 3.289318084716797e-05,
      "step": 34491,
      "training_step_time": 9.236082553863525
    },
    {
      "epoch": 3.289413452148437e-05,
      "model_forward_time": 1.7263705730438232,
      "step": 34492
    },
    {
      "epoch": 3.289413452148437e-05,
      "step": 34492,
      "training_step_time": 8.568729877471924
    },
    {
      "epoch": 3.289508819580078e-05,
      "model_forward_time": 1.7248179912567139,
      "step": 34493
    },
    {
      "epoch": 3.289508819580078e-05,
      "step": 34493,
      "training_step_time": 7.741084098815918
    },
    {
      "epoch": 3.289604187011719e-05,
      "model_forward_time": 1.7308881282806396,
      "step": 34494
    },
    {
      "epoch": 3.289604187011719e-05,
      "step": 34494,
      "training_step_time": 6.885796070098877
    },
    {
      "epoch": 3.289699554443359e-05,
      "model_forward_time": 1.7197728157043457,
      "step": 34495
    },
    {
      "epoch": 3.289699554443359e-05,
      "step": 34495,
      "training_step_time": 6.882051229476929
    },
    {
      "epoch": 3.289794921875e-05,
      "model_forward_time": 1.727147102355957,
      "step": 34496
    },
    {
      "epoch": 3.289794921875e-05,
      "step": 34496,
      "training_step_time": 6.885511159896851
    },
    {
      "epoch": 3.2898902893066406e-05,
      "model_forward_time": 1.7261707782745361,
      "step": 34497
    },
    {
      "epoch": 3.2898902893066406e-05,
      "step": 34497,
      "training_step_time": 7.612632513046265
    },
    {
      "epoch": 3.289985656738281e-05,
      "model_forward_time": 1.71225905418396,
      "step": 34498
    },
    {
      "epoch": 3.289985656738281e-05,
      "step": 34498,
      "training_step_time": 6.892454147338867
    },
    {
      "epoch": 3.290081024169922e-05,
      "model_forward_time": 1.8158466815948486,
      "step": 34499
    },
    {
      "epoch": 3.290081024169922e-05,
      "step": 34499,
      "training_step_time": 10.177403211593628
    },
    {
      "epoch": 3.2901763916015625e-05,
      "grad_norm": 0.10601401329040527,
      "learning_rate": 7.803712193986473e-06,
      "loss": 0.0569,
      "step": 34500
    },
    {
      "epoch": 3.2901763916015625e-05,
      "model_forward_time": 2.2786448001861572,
      "step": 34500
    },
    {
      "dynamics_loss_avg": 0.05863274279981852,
      "epoch": 3.2901763916015625e-05,
      "step": 34500
    },
    {
      "action_loss_avg": 0.0021178209339268506,
      "epoch": 3.2901763916015625e-05,
      "step": 34500
    },
    {
      "epoch": 3.2901763916015625e-05,
      "step": 34500,
      "training_step_time": 10.416960716247559
    },
    {
      "epoch": 3.290271759033203e-05,
      "model_forward_time": 1.7409133911132812,
      "step": 34501
    },
    {
      "epoch": 3.290271759033203e-05,
      "step": 34501,
      "training_step_time": 9.581742286682129
    },
    {
      "epoch": 3.290367126464844e-05,
      "model_forward_time": 1.7361066341400146,
      "step": 34502
    },
    {
      "epoch": 3.290367126464844e-05,
      "step": 34502,
      "training_step_time": 7.776923894882202
    },
    {
      "epoch": 3.2904624938964844e-05,
      "model_forward_time": 1.7378811836242676,
      "step": 34503
    },
    {
      "epoch": 3.2904624938964844e-05,
      "step": 34503,
      "training_step_time": 6.8817524909973145
    },
    {
      "epoch": 3.290557861328125e-05,
      "model_forward_time": 1.732914924621582,
      "step": 34504
    },
    {
      "epoch": 3.290557861328125e-05,
      "step": 34504,
      "training_step_time": 6.811578989028931
    },
    {
      "epoch": 3.290653228759766e-05,
      "model_forward_time": 1.7186734676361084,
      "step": 34505
    },
    {
      "epoch": 3.290653228759766e-05,
      "step": 34505,
      "training_step_time": 6.866239309310913
    },
    {
      "epoch": 3.290748596191406e-05,
      "model_forward_time": 1.7427423000335693,
      "step": 34506
    },
    {
      "epoch": 3.290748596191406e-05,
      "step": 34506,
      "training_step_time": 7.5397584438323975
    },
    {
      "epoch": 3.290843963623047e-05,
      "model_forward_time": 1.7362279891967773,
      "step": 34507
    },
    {
      "epoch": 3.290843963623047e-05,
      "step": 34507,
      "training_step_time": 6.966912508010864
    },
    {
      "epoch": 3.290939331054688e-05,
      "model_forward_time": 1.821906566619873,
      "step": 34508
    },
    {
      "epoch": 3.290939331054688e-05,
      "step": 34508,
      "training_step_time": 8.693546056747437
    },
    {
      "epoch": 3.291034698486328e-05,
      "model_forward_time": 2.3115322589874268,
      "step": 34509
    },
    {
      "epoch": 3.291034698486328e-05,
      "step": 34509,
      "training_step_time": 9.237584829330444
    },
    {
      "epoch": 3.2911300659179686e-05,
      "grad_norm": 0.1277686059474945,
      "learning_rate": 7.802342983755701e-06,
      "loss": 0.0588,
      "step": 34510
    },
    {
      "epoch": 3.2911300659179686e-05,
      "model_forward_time": 1.7271456718444824,
      "step": 34510
    },
    {
      "dynamics_loss_avg": 0.04972093850374222,
      "epoch": 3.2911300659179686e-05,
      "step": 34510
    },
    {
      "action_loss_avg": 0.001455142704071477,
      "epoch": 3.2911300659179686e-05,
      "step": 34510
    },
    {
      "epoch": 3.2911300659179686e-05,
      "step": 34510,
      "training_step_time": 7.98740291595459
    },
    {
      "epoch": 3.2912254333496096e-05,
      "model_forward_time": 1.750880241394043,
      "step": 34511
    },
    {
      "epoch": 3.2912254333496096e-05,
      "step": 34511,
      "training_step_time": 7.772240161895752
    },
    {
      "epoch": 3.29132080078125e-05,
      "model_forward_time": 1.7303478717803955,
      "step": 34512
    },
    {
      "epoch": 3.29132080078125e-05,
      "step": 34512,
      "training_step_time": 7.667912721633911
    },
    {
      "epoch": 3.2914161682128904e-05,
      "model_forward_time": 1.7306468486785889,
      "step": 34513
    },
    {
      "epoch": 3.2914161682128904e-05,
      "step": 34513,
      "training_step_time": 6.867393970489502
    },
    {
      "epoch": 3.2915115356445315e-05,
      "model_forward_time": 1.73095703125,
      "step": 34514
    },
    {
      "epoch": 3.2915115356445315e-05,
      "step": 34514,
      "training_step_time": 6.8823401927948
    },
    {
      "epoch": 3.291606903076172e-05,
      "model_forward_time": 1.7378184795379639,
      "step": 34515
    },
    {
      "epoch": 3.291606903076172e-05,
      "step": 34515,
      "training_step_time": 7.64470648765564
    },
    {
      "epoch": 3.291702270507812e-05,
      "model_forward_time": 1.7276523113250732,
      "step": 34516
    },
    {
      "epoch": 3.291702270507812e-05,
      "step": 34516,
      "training_step_time": 6.90756893157959
    },
    {
      "epoch": 3.2917976379394534e-05,
      "model_forward_time": 1.7642936706542969,
      "step": 34517
    },
    {
      "epoch": 3.2917976379394534e-05,
      "step": 34517,
      "training_step_time": 8.34850788116455
    },
    {
      "epoch": 3.291893005371094e-05,
      "model_forward_time": 2.3012168407440186,
      "step": 34518
    },
    {
      "epoch": 3.291893005371094e-05,
      "step": 34518,
      "training_step_time": 8.877467632293701
    },
    {
      "epoch": 3.291988372802734e-05,
      "model_forward_time": 1.7320003509521484,
      "step": 34519
    },
    {
      "epoch": 3.291988372802734e-05,
      "step": 34519,
      "training_step_time": 8.609124660491943
    },
    {
      "epoch": 3.292083740234375e-05,
      "grad_norm": 0.12428004294633865,
      "learning_rate": 7.800973467064917e-06,
      "loss": 0.0581,
      "step": 34520
    },
    {
      "epoch": 3.292083740234375e-05,
      "model_forward_time": 1.7329485416412354,
      "step": 34520
    },
    {
      "dynamics_loss_avg": 0.05435154512524605,
      "epoch": 3.292083740234375e-05,
      "step": 34520
    },
    {
      "action_loss_avg": 0.0017952573194634168,
      "epoch": 3.292083740234375e-05,
      "step": 34520
    },
    {
      "epoch": 3.292083740234375e-05,
      "step": 34520,
      "training_step_time": 7.826563358306885
    },
    {
      "epoch": 3.292179107666016e-05,
      "model_forward_time": 1.709533452987671,
      "step": 34521
    },
    {
      "epoch": 3.292179107666016e-05,
      "step": 34521,
      "training_step_time": 7.7404866218566895
    },
    {
      "epoch": 3.292274475097656e-05,
      "model_forward_time": 1.714247226715088,
      "step": 34522
    },
    {
      "epoch": 3.292274475097656e-05,
      "step": 34522,
      "training_step_time": 6.804868698120117
    },
    {
      "epoch": 3.292369842529297e-05,
      "model_forward_time": 1.7184467315673828,
      "step": 34523
    },
    {
      "epoch": 3.292369842529297e-05,
      "step": 34523,
      "training_step_time": 6.80689811706543
    },
    {
      "epoch": 3.2924652099609376e-05,
      "model_forward_time": 1.7202296257019043,
      "step": 34524
    },
    {
      "epoch": 3.2924652099609376e-05,
      "step": 34524,
      "training_step_time": 7.535781145095825
    },
    {
      "epoch": 3.292560577392578e-05,
      "model_forward_time": 1.7230162620544434,
      "step": 34525
    },
    {
      "epoch": 3.292560577392578e-05,
      "step": 34525,
      "training_step_time": 6.865853548049927
    },
    {
      "epoch": 3.292655944824219e-05,
      "model_forward_time": 1.7508933544158936,
      "step": 34526
    },
    {
      "epoch": 3.292655944824219e-05,
      "step": 34526,
      "training_step_time": 7.717981338500977
    },
    {
      "epoch": 3.2927513122558595e-05,
      "model_forward_time": 1.7172458171844482,
      "step": 34527
    },
    {
      "epoch": 3.2927513122558595e-05,
      "step": 34527,
      "training_step_time": 8.326890707015991
    },
    {
      "epoch": 3.2928466796875e-05,
      "model_forward_time": 1.6894721984863281,
      "step": 34528
    },
    {
      "epoch": 3.2928466796875e-05,
      "step": 34528,
      "training_step_time": 9.671202898025513
    },
    {
      "epoch": 3.292942047119141e-05,
      "model_forward_time": 1.7184453010559082,
      "step": 34529
    },
    {
      "epoch": 3.292942047119141e-05,
      "step": 34529,
      "training_step_time": 8.769012451171875
    },
    {
      "epoch": 3.2930374145507814e-05,
      "grad_norm": 0.13838380575180054,
      "learning_rate": 7.799603644063884e-06,
      "loss": 0.0603,
      "step": 34530
    },
    {
      "epoch": 3.2930374145507814e-05,
      "model_forward_time": 1.7106168270111084,
      "step": 34530
    },
    {
      "dynamics_loss_avg": 0.04384624399244785,
      "epoch": 3.2930374145507814e-05,
      "step": 34530
    },
    {
      "action_loss_avg": 0.0018732005322817714,
      "epoch": 3.2930374145507814e-05,
      "step": 34530
    },
    {
      "epoch": 3.2930374145507814e-05,
      "step": 34530,
      "training_step_time": 7.641579866409302
    },
    {
      "epoch": 3.293132781982422e-05,
      "model_forward_time": 1.7123522758483887,
      "step": 34531
    },
    {
      "epoch": 3.293132781982422e-05,
      "step": 34531,
      "training_step_time": 6.859051942825317
    },
    {
      "epoch": 3.293228149414062e-05,
      "model_forward_time": 1.7177038192749023,
      "step": 34532
    },
    {
      "epoch": 3.293228149414062e-05,
      "step": 34532,
      "training_step_time": 6.82705545425415
    },
    {
      "epoch": 3.293323516845703e-05,
      "model_forward_time": 1.7216880321502686,
      "step": 34533
    },
    {
      "epoch": 3.293323516845703e-05,
      "step": 34533,
      "training_step_time": 7.573870658874512
    },
    {
      "epoch": 3.2934188842773436e-05,
      "model_forward_time": 1.7137067317962646,
      "step": 34534
    },
    {
      "epoch": 3.2934188842773436e-05,
      "step": 34534,
      "training_step_time": 6.814432382583618
    },
    {
      "epoch": 3.293514251708984e-05,
      "model_forward_time": 1.7328031063079834,
      "step": 34535
    },
    {
      "epoch": 3.293514251708984e-05,
      "step": 34535,
      "training_step_time": 9.366592168807983
    },
    {
      "epoch": 3.293609619140625e-05,
      "model_forward_time": 1.721353530883789,
      "step": 34536
    },
    {
      "epoch": 3.293609619140625e-05,
      "step": 34536,
      "training_step_time": 8.265332221984863
    },
    {
      "epoch": 3.2937049865722655e-05,
      "model_forward_time": 2.260162353515625,
      "step": 34537
    },
    {
      "epoch": 3.2937049865722655e-05,
      "step": 34537,
      "training_step_time": 7.8264923095703125
    },
    {
      "epoch": 3.293800354003906e-05,
      "model_forward_time": 1.713618278503418,
      "step": 34538
    },
    {
      "epoch": 3.293800354003906e-05,
      "step": 34538,
      "training_step_time": 8.965266942977905
    },
    {
      "epoch": 3.293895721435547e-05,
      "model_forward_time": 1.7154912948608398,
      "step": 34539
    },
    {
      "epoch": 3.293895721435547e-05,
      "step": 34539,
      "training_step_time": 7.747906446456909
    },
    {
      "epoch": 3.2939910888671874e-05,
      "grad_norm": 0.13694435358047485,
      "learning_rate": 7.798233514902406e-06,
      "loss": 0.0622,
      "step": 34540
    },
    {
      "epoch": 3.2939910888671874e-05,
      "model_forward_time": 1.716874599456787,
      "step": 34540
    },
    {
      "dynamics_loss_avg": 0.04592740423977375,
      "epoch": 3.2939910888671874e-05,
      "step": 34540
    },
    {
      "action_loss_avg": 0.0012635869439691304,
      "epoch": 3.2939910888671874e-05,
      "step": 34540
    },
    {
      "epoch": 3.2939910888671874e-05,
      "step": 34540,
      "training_step_time": 6.806016445159912
    },
    {
      "epoch": 3.294086456298828e-05,
      "model_forward_time": 1.72296142578125,
      "step": 34541
    },
    {
      "epoch": 3.294086456298828e-05,
      "step": 34541,
      "training_step_time": 6.795171737670898
    },
    {
      "epoch": 3.294181823730469e-05,
      "model_forward_time": 1.724811315536499,
      "step": 34542
    },
    {
      "epoch": 3.294181823730469e-05,
      "step": 34542,
      "training_step_time": 7.57241153717041
    },
    {
      "epoch": 3.294277191162109e-05,
      "model_forward_time": 1.7202062606811523,
      "step": 34543
    },
    {
      "epoch": 3.294277191162109e-05,
      "step": 34543,
      "training_step_time": 6.846556663513184
    },
    {
      "epoch": 3.29437255859375e-05,
      "model_forward_time": 1.716055154800415,
      "step": 34544
    },
    {
      "epoch": 3.29437255859375e-05,
      "step": 34544,
      "training_step_time": 8.214449644088745
    },
    {
      "epoch": 3.294467926025391e-05,
      "model_forward_time": 1.7214276790618896,
      "step": 34545
    },
    {
      "epoch": 3.294467926025391e-05,
      "step": 34545,
      "training_step_time": 8.735793590545654
    },
    {
      "epoch": 3.294563293457031e-05,
      "model_forward_time": 2.253758430480957,
      "step": 34546
    },
    {
      "epoch": 3.294563293457031e-05,
      "step": 34546,
      "training_step_time": 8.386086225509644
    },
    {
      "epoch": 3.2946586608886716e-05,
      "model_forward_time": 1.7125303745269775,
      "step": 34547
    },
    {
      "epoch": 3.2946586608886716e-05,
      "step": 34547,
      "training_step_time": 8.050399541854858
    },
    {
      "epoch": 3.294754028320313e-05,
      "model_forward_time": 1.7175045013427734,
      "step": 34548
    },
    {
      "epoch": 3.294754028320313e-05,
      "step": 34548,
      "training_step_time": 8.39007568359375
    },
    {
      "epoch": 3.294849395751953e-05,
      "model_forward_time": 1.715733289718628,
      "step": 34549
    },
    {
      "epoch": 3.294849395751953e-05,
      "step": 34549,
      "training_step_time": 6.856611728668213
    },
    {
      "epoch": 3.2949447631835935e-05,
      "grad_norm": 0.10258554667234421,
      "learning_rate": 7.796863079730317e-06,
      "loss": 0.0629,
      "step": 34550
    },
    {
      "epoch": 3.2949447631835935e-05,
      "model_forward_time": 1.7216687202453613,
      "step": 34550
    },
    {
      "dynamics_loss_avg": 0.05008375495672226,
      "epoch": 3.2949447631835935e-05,
      "step": 34550
    },
    {
      "action_loss_avg": 0.0015693007095251232,
      "epoch": 3.2949447631835935e-05,
      "step": 34550
    },
    {
      "epoch": 3.2949447631835935e-05,
      "step": 34550,
      "training_step_time": 6.826988458633423
    },
    {
      "epoch": 3.2950401306152346e-05,
      "model_forward_time": 1.7251653671264648,
      "step": 34551
    },
    {
      "epoch": 3.2950401306152346e-05,
      "step": 34551,
      "training_step_time": 7.426165342330933
    },
    {
      "epoch": 3.295135498046875e-05,
      "model_forward_time": 1.7340803146362305,
      "step": 34552
    },
    {
      "epoch": 3.295135498046875e-05,
      "step": 34552,
      "training_step_time": 6.892977714538574
    },
    {
      "epoch": 3.2952308654785154e-05,
      "model_forward_time": 1.7258670330047607,
      "step": 34553
    },
    {
      "epoch": 3.2952308654785154e-05,
      "step": 34553,
      "training_step_time": 7.168305158615112
    },
    {
      "epoch": 3.2953262329101564e-05,
      "model_forward_time": 1.7834372520446777,
      "step": 34554
    },
    {
      "epoch": 3.2953262329101564e-05,
      "step": 34554,
      "training_step_time": 7.734992742538452
    },
    {
      "epoch": 3.295421600341797e-05,
      "model_forward_time": 2.2934250831604004,
      "step": 34555
    },
    {
      "epoch": 3.295421600341797e-05,
      "step": 34555,
      "training_step_time": 8.678293704986572
    },
    {
      "epoch": 3.295516967773437e-05,
      "model_forward_time": 1.7325835227966309,
      "step": 34556
    },
    {
      "epoch": 3.295516967773437e-05,
      "step": 34556,
      "training_step_time": 9.789156436920166
    },
    {
      "epoch": 3.295612335205078e-05,
      "model_forward_time": 1.7266144752502441,
      "step": 34557
    },
    {
      "epoch": 3.295612335205078e-05,
      "step": 34557,
      "training_step_time": 7.837220191955566
    },
    {
      "epoch": 3.295707702636719e-05,
      "model_forward_time": 1.7301115989685059,
      "step": 34558
    },
    {
      "epoch": 3.295707702636719e-05,
      "step": 34558,
      "training_step_time": 7.638711929321289
    },
    {
      "epoch": 3.295803070068359e-05,
      "model_forward_time": 1.7309415340423584,
      "step": 34559
    },
    {
      "epoch": 3.295803070068359e-05,
      "step": 34559,
      "training_step_time": 6.829375505447388
    },
    {
      "epoch": 3.2958984375e-05,
      "grad_norm": 0.12692812085151672,
      "learning_rate": 7.795492338697488e-06,
      "loss": 0.0596,
      "step": 34560
    },
    {
      "epoch": 3.2958984375e-05,
      "model_forward_time": 1.7341532707214355,
      "step": 34560
    },
    {
      "dynamics_loss_avg": 0.050268758460879324,
      "epoch": 3.2958984375e-05,
      "step": 34560
    },
    {
      "action_loss_avg": 0.002472616476006806,
      "epoch": 3.2958984375e-05,
      "step": 34560
    },
    {
      "epoch": 3.2958984375e-05,
      "step": 34560,
      "training_step_time": 7.468275785446167
    },
    {
      "epoch": 3.2959938049316406e-05,
      "model_forward_time": 1.726393461227417,
      "step": 34561
    },
    {
      "epoch": 3.2959938049316406e-05,
      "step": 34561,
      "training_step_time": 6.900380611419678
    },
    {
      "epoch": 3.296089172363281e-05,
      "model_forward_time": 1.7292356491088867,
      "step": 34562
    },
    {
      "epoch": 3.296089172363281e-05,
      "step": 34562,
      "training_step_time": 7.647171258926392
    },
    {
      "epoch": 3.296184539794922e-05,
      "model_forward_time": 1.7174127101898193,
      "step": 34563
    },
    {
      "epoch": 3.296184539794922e-05,
      "step": 34563,
      "training_step_time": 7.73803973197937
    },
    {
      "epoch": 3.2962799072265625e-05,
      "model_forward_time": 2.271937847137451,
      "step": 34564
    },
    {
      "epoch": 3.2962799072265625e-05,
      "step": 34564,
      "training_step_time": 10.290166139602661
    },
    {
      "epoch": 3.296375274658203e-05,
      "model_forward_time": 1.7099816799163818,
      "step": 34565
    },
    {
      "epoch": 3.296375274658203e-05,
      "step": 34565,
      "training_step_time": 7.854856491088867
    },
    {
      "epoch": 3.296470642089844e-05,
      "model_forward_time": 1.7081856727600098,
      "step": 34566
    },
    {
      "epoch": 3.296470642089844e-05,
      "step": 34566,
      "training_step_time": 8.03366756439209
    },
    {
      "epoch": 3.2965660095214844e-05,
      "model_forward_time": 1.7093050479888916,
      "step": 34567
    },
    {
      "epoch": 3.2965660095214844e-05,
      "step": 34567,
      "training_step_time": 7.734071731567383
    },
    {
      "epoch": 3.296661376953125e-05,
      "model_forward_time": 1.7173950672149658,
      "step": 34568
    },
    {
      "epoch": 3.296661376953125e-05,
      "step": 34568,
      "training_step_time": 6.897199392318726
    },
    {
      "epoch": 3.296756744384766e-05,
      "model_forward_time": 1.713465690612793,
      "step": 34569
    },
    {
      "epoch": 3.296756744384766e-05,
      "step": 34569,
      "training_step_time": 7.559373140335083
    },
    {
      "epoch": 3.296852111816406e-05,
      "grad_norm": 0.17014028131961823,
      "learning_rate": 7.79412129195382e-06,
      "loss": 0.0624,
      "step": 34570
    },
    {
      "epoch": 3.296852111816406e-05,
      "model_forward_time": 1.7061073780059814,
      "step": 34570
    },
    {
      "dynamics_loss_avg": 0.05177620835602283,
      "epoch": 3.296852111816406e-05,
      "step": 34570
    },
    {
      "action_loss_avg": 0.0017715163005050272,
      "epoch": 3.296852111816406e-05,
      "step": 34570
    },
    {
      "epoch": 3.296852111816406e-05,
      "step": 34570,
      "training_step_time": 6.916229248046875
    },
    {
      "epoch": 3.296947479248047e-05,
      "model_forward_time": 1.7139134407043457,
      "step": 34571
    },
    {
      "epoch": 3.296947479248047e-05,
      "step": 34571,
      "training_step_time": 7.639135122299194
    },
    {
      "epoch": 3.297042846679688e-05,
      "model_forward_time": 1.7185180187225342,
      "step": 34572
    },
    {
      "epoch": 3.297042846679688e-05,
      "step": 34572,
      "training_step_time": 8.449283123016357
    },
    {
      "epoch": 3.297138214111328e-05,
      "model_forward_time": 2.2645909786224365,
      "step": 34573
    },
    {
      "epoch": 3.297138214111328e-05,
      "step": 34573,
      "training_step_time": 8.85147762298584
    },
    {
      "epoch": 3.2972335815429686e-05,
      "model_forward_time": 1.7092559337615967,
      "step": 34574
    },
    {
      "epoch": 3.2972335815429686e-05,
      "step": 34574,
      "training_step_time": 8.341984033584595
    },
    {
      "epoch": 3.2973289489746096e-05,
      "model_forward_time": 1.7105908393859863,
      "step": 34575
    },
    {
      "epoch": 3.2973289489746096e-05,
      "step": 34575,
      "training_step_time": 8.034096717834473
    },
    {
      "epoch": 3.29742431640625e-05,
      "model_forward_time": 1.7041561603546143,
      "step": 34576
    },
    {
      "epoch": 3.29742431640625e-05,
      "step": 34576,
      "training_step_time": 7.689251899719238
    },
    {
      "epoch": 3.2975196838378905e-05,
      "model_forward_time": 1.716238260269165,
      "step": 34577
    },
    {
      "epoch": 3.2975196838378905e-05,
      "step": 34577,
      "training_step_time": 7.568364381790161
    },
    {
      "epoch": 3.2976150512695315e-05,
      "model_forward_time": 1.7155306339263916,
      "step": 34578
    },
    {
      "epoch": 3.2976150512695315e-05,
      "step": 34578,
      "training_step_time": 7.572778701782227
    },
    {
      "epoch": 3.297710418701172e-05,
      "model_forward_time": 1.7116367816925049,
      "step": 34579
    },
    {
      "epoch": 3.297710418701172e-05,
      "step": 34579,
      "training_step_time": 6.809337854385376
    },
    {
      "epoch": 3.297805786132812e-05,
      "grad_norm": 0.16061051189899445,
      "learning_rate": 7.792749939649247e-06,
      "loss": 0.06,
      "step": 34580
    },
    {
      "epoch": 3.297805786132812e-05,
      "model_forward_time": 1.7060930728912354,
      "step": 34580
    },
    {
      "dynamics_loss_avg": 0.048456933535635474,
      "epoch": 3.297805786132812e-05,
      "step": 34580
    },
    {
      "action_loss_avg": 0.0019251278834417462,
      "epoch": 3.297805786132812e-05,
      "step": 34580
    },
    {
      "epoch": 3.297805786132812e-05,
      "step": 34580,
      "training_step_time": 7.5849997997283936
    },
    {
      "epoch": 3.2979011535644534e-05,
      "model_forward_time": 1.7094738483428955,
      "step": 34581
    },
    {
      "epoch": 3.2979011535644534e-05,
      "step": 34581,
      "training_step_time": 7.701606273651123
    },
    {
      "epoch": 3.297996520996094e-05,
      "model_forward_time": 1.7434709072113037,
      "step": 34582
    },
    {
      "epoch": 3.297996520996094e-05,
      "step": 34582,
      "training_step_time": 7.788802146911621
    },
    {
      "epoch": 3.298091888427734e-05,
      "model_forward_time": 2.273355722427368,
      "step": 34583
    },
    {
      "epoch": 3.298091888427734e-05,
      "step": 34583,
      "training_step_time": 10.85953402519226
    },
    {
      "epoch": 3.298187255859375e-05,
      "model_forward_time": 1.7098569869995117,
      "step": 34584
    },
    {
      "epoch": 3.298187255859375e-05,
      "step": 34584,
      "training_step_time": 8.972809553146362
    },
    {
      "epoch": 3.298282623291016e-05,
      "model_forward_time": 1.717270851135254,
      "step": 34585
    },
    {
      "epoch": 3.298282623291016e-05,
      "step": 34585,
      "training_step_time": 8.491658687591553
    },
    {
      "epoch": 3.298377990722656e-05,
      "model_forward_time": 1.7145452499389648,
      "step": 34586
    },
    {
      "epoch": 3.298377990722656e-05,
      "step": 34586,
      "training_step_time": 7.531198978424072
    },
    {
      "epoch": 3.298473358154297e-05,
      "model_forward_time": 1.710357666015625,
      "step": 34587
    },
    {
      "epoch": 3.298473358154297e-05,
      "step": 34587,
      "training_step_time": 7.49430513381958
    },
    {
      "epoch": 3.2985687255859376e-05,
      "model_forward_time": 1.7035987377166748,
      "step": 34588
    },
    {
      "epoch": 3.2985687255859376e-05,
      "step": 34588,
      "training_step_time": 6.78997278213501
    },
    {
      "epoch": 3.298664093017578e-05,
      "model_forward_time": 1.7204506397247314,
      "step": 34589
    },
    {
      "epoch": 3.298664093017578e-05,
      "step": 34589,
      "training_step_time": 7.581113338470459
    },
    {
      "epoch": 3.298759460449219e-05,
      "grad_norm": 0.12786176800727844,
      "learning_rate": 7.79137828193374e-06,
      "loss": 0.0651,
      "step": 34590
    },
    {
      "epoch": 3.298759460449219e-05,
      "model_forward_time": 1.7092692852020264,
      "step": 34590
    },
    {
      "dynamics_loss_avg": 0.06307200640439987,
      "epoch": 3.298759460449219e-05,
      "step": 34590
    },
    {
      "action_loss_avg": 0.002105516439769417,
      "epoch": 3.298759460449219e-05,
      "step": 34590
    },
    {
      "epoch": 3.298759460449219e-05,
      "step": 34590,
      "training_step_time": 7.597231864929199
    },
    {
      "epoch": 3.2988548278808595e-05,
      "model_forward_time": 1.7375726699829102,
      "step": 34591
    },
    {
      "epoch": 3.2988548278808595e-05,
      "step": 34591,
      "training_step_time": 9.359389305114746
    },
    {
      "epoch": 3.2989501953125e-05,
      "model_forward_time": 2.262197732925415,
      "step": 34592
    },
    {
      "epoch": 3.2989501953125e-05,
      "step": 34592,
      "training_step_time": 7.733090877532959
    },
    {
      "epoch": 3.299045562744141e-05,
      "model_forward_time": 1.7094366550445557,
      "step": 34593
    },
    {
      "epoch": 3.299045562744141e-05,
      "step": 34593,
      "training_step_time": 8.356722354888916
    },
    {
      "epoch": 3.2991409301757814e-05,
      "model_forward_time": 1.7174601554870605,
      "step": 34594
    },
    {
      "epoch": 3.2991409301757814e-05,
      "step": 34594,
      "training_step_time": 7.7831525802612305
    },
    {
      "epoch": 3.299236297607422e-05,
      "model_forward_time": 1.7198286056518555,
      "step": 34595
    },
    {
      "epoch": 3.299236297607422e-05,
      "step": 34595,
      "training_step_time": 8.289698123931885
    },
    {
      "epoch": 3.299331665039062e-05,
      "model_forward_time": 1.720832347869873,
      "step": 34596
    },
    {
      "epoch": 3.299331665039062e-05,
      "step": 34596,
      "training_step_time": 7.532919883728027
    },
    {
      "epoch": 3.299427032470703e-05,
      "model_forward_time": 1.7171316146850586,
      "step": 34597
    },
    {
      "epoch": 3.299427032470703e-05,
      "step": 34597,
      "training_step_time": 6.787400484085083
    },
    {
      "epoch": 3.2995223999023437e-05,
      "model_forward_time": 1.7171440124511719,
      "step": 34598
    },
    {
      "epoch": 3.2995223999023437e-05,
      "step": 34598,
      "training_step_time": 7.555413246154785
    },
    {
      "epoch": 3.299617767333984e-05,
      "model_forward_time": 1.7136430740356445,
      "step": 34599
    },
    {
      "epoch": 3.299617767333984e-05,
      "step": 34599,
      "training_step_time": 7.615882158279419
    },
    {
      "epoch": 3.299713134765625e-05,
      "grad_norm": 0.13505034148693085,
      "learning_rate": 7.7900063189573e-06,
      "loss": 0.0619,
      "step": 34600
    },
    {
      "epoch": 3.299713134765625e-05,
      "model_forward_time": 1.7350735664367676,
      "step": 34600
    },
    {
      "dynamics_loss_avg": 0.050350321829319,
      "epoch": 3.299713134765625e-05,
      "step": 34600
    },
    {
      "action_loss_avg": 0.0014952748198993504,
      "epoch": 3.299713134765625e-05,
      "step": 34600
    },
    {
      "epoch": 3.299713134765625e-05,
      "step": 34600,
      "training_step_time": 8.622483730316162
    },
    {
      "epoch": 3.2998085021972655e-05,
      "model_forward_time": 2.258408546447754,
      "step": 34601
    },
    {
      "epoch": 3.2998085021972655e-05,
      "step": 34601,
      "training_step_time": 7.74530291557312
    },
    {
      "epoch": 3.299903869628906e-05,
      "model_forward_time": 1.7106411457061768,
      "step": 34602
    },
    {
      "epoch": 3.299903869628906e-05,
      "step": 34602,
      "training_step_time": 8.279136180877686
    },
    {
      "epoch": 3.299999237060547e-05,
      "model_forward_time": 1.7166743278503418,
      "step": 34603
    },
    {
      "epoch": 3.299999237060547e-05,
      "step": 34603,
      "training_step_time": 8.892731428146362
    },
    {
      "epoch": 3.3000946044921874e-05,
      "model_forward_time": 1.7092881202697754,
      "step": 34604
    },
    {
      "epoch": 3.3000946044921874e-05,
      "step": 34604,
      "training_step_time": 7.635103940963745
    },
    {
      "epoch": 3.300189971923828e-05,
      "model_forward_time": 1.7156152725219727,
      "step": 34605
    },
    {
      "epoch": 3.300189971923828e-05,
      "step": 34605,
      "training_step_time": 7.516092300415039
    },
    {
      "epoch": 3.300285339355469e-05,
      "model_forward_time": 1.7081949710845947,
      "step": 34606
    },
    {
      "epoch": 3.300285339355469e-05,
      "step": 34606,
      "training_step_time": 6.793271780014038
    },
    {
      "epoch": 3.300380706787109e-05,
      "model_forward_time": 1.7134172916412354,
      "step": 34607
    },
    {
      "epoch": 3.300380706787109e-05,
      "step": 34607,
      "training_step_time": 7.723025798797607
    },
    {
      "epoch": 3.30047607421875e-05,
      "model_forward_time": 1.71639084815979,
      "step": 34608
    },
    {
      "epoch": 3.30047607421875e-05,
      "step": 34608,
      "training_step_time": 7.736071586608887
    },
    {
      "epoch": 3.300571441650391e-05,
      "model_forward_time": 1.733513593673706,
      "step": 34609
    },
    {
      "epoch": 3.300571441650391e-05,
      "step": 34609,
      "training_step_time": 7.837039232254028
    },
    {
      "epoch": 3.300666809082031e-05,
      "grad_norm": 0.14660291373729706,
      "learning_rate": 7.788634050869965e-06,
      "loss": 0.0591,
      "step": 34610
    },
    {
      "epoch": 3.300666809082031e-05,
      "model_forward_time": 2.2724854946136475,
      "step": 34610
    },
    {
      "dynamics_loss_avg": 0.05175413992255926,
      "epoch": 3.300666809082031e-05,
      "step": 34610
    },
    {
      "action_loss_avg": 0.001983969088178128,
      "epoch": 3.300666809082031e-05,
      "step": 34610
    },
    {
      "epoch": 3.300666809082031e-05,
      "step": 34610,
      "training_step_time": 9.747635126113892
    },
    {
      "epoch": 3.3007621765136716e-05,
      "model_forward_time": 1.7149653434753418,
      "step": 34611
    },
    {
      "epoch": 3.3007621765136716e-05,
      "step": 34611,
      "training_step_time": 9.615782976150513
    },
    {
      "epoch": 3.300857543945313e-05,
      "model_forward_time": 1.7193350791931152,
      "step": 34612
    },
    {
      "epoch": 3.300857543945313e-05,
      "step": 34612,
      "training_step_time": 7.7668845653533936
    },
    {
      "epoch": 3.300952911376953e-05,
      "model_forward_time": 1.7119991779327393,
      "step": 34613
    },
    {
      "epoch": 3.300952911376953e-05,
      "step": 34613,
      "training_step_time": 7.642730951309204
    },
    {
      "epoch": 3.3010482788085935e-05,
      "model_forward_time": 1.7212793827056885,
      "step": 34614
    },
    {
      "epoch": 3.3010482788085935e-05,
      "step": 34614,
      "training_step_time": 7.482041358947754
    },
    {
      "epoch": 3.3011436462402346e-05,
      "model_forward_time": 1.71812105178833,
      "step": 34615
    },
    {
      "epoch": 3.3011436462402346e-05,
      "step": 34615,
      "training_step_time": 6.696016788482666
    },
    {
      "epoch": 3.301239013671875e-05,
      "model_forward_time": 1.7250664234161377,
      "step": 34616
    },
    {
      "epoch": 3.301239013671875e-05,
      "step": 34616,
      "training_step_time": 7.508001089096069
    },
    {
      "epoch": 3.3013343811035154e-05,
      "model_forward_time": 1.7162938117980957,
      "step": 34617
    },
    {
      "epoch": 3.3013343811035154e-05,
      "step": 34617,
      "training_step_time": 7.630140781402588
    },
    {
      "epoch": 3.3014297485351565e-05,
      "model_forward_time": 1.7441418170928955,
      "step": 34618
    },
    {
      "epoch": 3.3014297485351565e-05,
      "step": 34618,
      "training_step_time": 9.241420030593872
    },
    {
      "epoch": 3.301525115966797e-05,
      "model_forward_time": 2.284733533859253,
      "step": 34619
    },
    {
      "epoch": 3.301525115966797e-05,
      "step": 34619,
      "training_step_time": 8.791934728622437
    },
    {
      "epoch": 3.301620483398437e-05,
      "grad_norm": 0.15163186192512512,
      "learning_rate": 7.787261477821803e-06,
      "loss": 0.0641,
      "step": 34620
    },
    {
      "epoch": 3.301620483398437e-05,
      "model_forward_time": 1.7183756828308105,
      "step": 34620
    },
    {
      "dynamics_loss_avg": 0.05706579405814409,
      "epoch": 3.301620483398437e-05,
      "step": 34620
    },
    {
      "action_loss_avg": 0.0018561495642643421,
      "epoch": 3.301620483398437e-05,
      "step": 34620
    },
    {
      "epoch": 3.301620483398437e-05,
      "step": 34620,
      "training_step_time": 7.690756320953369
    },
    {
      "epoch": 3.301715850830078e-05,
      "model_forward_time": 1.722900629043579,
      "step": 34621
    },
    {
      "epoch": 3.301715850830078e-05,
      "step": 34621,
      "training_step_time": 8.731738328933716
    },
    {
      "epoch": 3.301811218261719e-05,
      "model_forward_time": 1.7311184406280518,
      "step": 34622
    },
    {
      "epoch": 3.301811218261719e-05,
      "step": 34622,
      "training_step_time": 7.478479862213135
    },
    {
      "epoch": 3.301906585693359e-05,
      "model_forward_time": 1.716456651687622,
      "step": 34623
    },
    {
      "epoch": 3.301906585693359e-05,
      "step": 34623,
      "training_step_time": 7.3768486976623535
    },
    {
      "epoch": 3.302001953125e-05,
      "model_forward_time": 1.7310218811035156,
      "step": 34624
    },
    {
      "epoch": 3.302001953125e-05,
      "step": 34624,
      "training_step_time": 6.633198499679565
    },
    {
      "epoch": 3.3020973205566406e-05,
      "model_forward_time": 1.731109619140625,
      "step": 34625
    },
    {
      "epoch": 3.3020973205566406e-05,
      "step": 34625,
      "training_step_time": 7.4846107959747314
    },
    {
      "epoch": 3.302192687988281e-05,
      "model_forward_time": 1.7219579219818115,
      "step": 34626
    },
    {
      "epoch": 3.302192687988281e-05,
      "step": 34626,
      "training_step_time": 7.500578880310059
    },
    {
      "epoch": 3.302288055419922e-05,
      "model_forward_time": 1.7354109287261963,
      "step": 34627
    },
    {
      "epoch": 3.302288055419922e-05,
      "step": 34627,
      "training_step_time": 8.178968906402588
    },
    {
      "epoch": 3.3023834228515625e-05,
      "model_forward_time": 1.3563811779022217,
      "step": 34628
    },
    {
      "epoch": 3.3023834228515625e-05,
      "step": 34628,
      "training_step_time": 9.008477687835693
    },
    {
      "epoch": 3.302478790283203e-05,
      "model_forward_time": 2.2564268112182617,
      "step": 34629
    },
    {
      "epoch": 3.302478790283203e-05,
      "step": 34629,
      "training_step_time": 8.506810188293457
    },
    {
      "epoch": 3.302574157714844e-05,
      "grad_norm": 0.12752342224121094,
      "learning_rate": 7.785888599962916e-06,
      "loss": 0.0597,
      "step": 34630
    },
    {
      "epoch": 3.302574157714844e-05,
      "model_forward_time": 1.7195324897766113,
      "step": 34630
    },
    {
      "dynamics_loss_avg": 0.05691044246777892,
      "epoch": 3.302574157714844e-05,
      "step": 34630
    },
    {
      "action_loss_avg": 0.001977909848210402,
      "epoch": 3.302574157714844e-05,
      "step": 34630
    },
    {
      "epoch": 3.302574157714844e-05,
      "step": 34630,
      "training_step_time": 9.6383638381958
    },
    {
      "epoch": 3.3026695251464844e-05,
      "model_forward_time": 1.722196102142334,
      "step": 34631
    },
    {
      "epoch": 3.3026695251464844e-05,
      "step": 34631,
      "training_step_time": 8.582531452178955
    },
    {
      "epoch": 3.302764892578125e-05,
      "model_forward_time": 1.7210533618927002,
      "step": 34632
    },
    {
      "epoch": 3.302764892578125e-05,
      "step": 34632,
      "training_step_time": 8.431824684143066
    },
    {
      "epoch": 3.302860260009766e-05,
      "model_forward_time": 1.7234222888946533,
      "step": 34633
    },
    {
      "epoch": 3.302860260009766e-05,
      "step": 34633,
      "training_step_time": 7.12231183052063
    },
    {
      "epoch": 3.302955627441406e-05,
      "model_forward_time": 1.7231626510620117,
      "step": 34634
    },
    {
      "epoch": 3.302955627441406e-05,
      "step": 34634,
      "training_step_time": 7.656869649887085
    },
    {
      "epoch": 3.303050994873047e-05,
      "model_forward_time": 1.7171704769134521,
      "step": 34635
    },
    {
      "epoch": 3.303050994873047e-05,
      "step": 34635,
      "training_step_time": 7.404817342758179
    },
    {
      "epoch": 3.303146362304688e-05,
      "model_forward_time": 1.758371114730835,
      "step": 34636
    },
    {
      "epoch": 3.303146362304688e-05,
      "step": 34636,
      "training_step_time": 7.906459331512451
    },
    {
      "epoch": 3.303241729736328e-05,
      "model_forward_time": 1.7339887619018555,
      "step": 34637
    },
    {
      "epoch": 3.303241729736328e-05,
      "step": 34637,
      "training_step_time": 7.716470003128052
    },
    {
      "epoch": 3.3033370971679686e-05,
      "model_forward_time": 2.285609483718872,
      "step": 34638
    },
    {
      "epoch": 3.3033370971679686e-05,
      "step": 34638,
      "training_step_time": 9.805349826812744
    },
    {
      "epoch": 3.3034324645996097e-05,
      "model_forward_time": 1.7319538593292236,
      "step": 34639
    },
    {
      "epoch": 3.3034324645996097e-05,
      "step": 34639,
      "training_step_time": 7.7748353481292725
    },
    {
      "epoch": 3.30352783203125e-05,
      "grad_norm": 0.13537143170833588,
      "learning_rate": 7.78451541744344e-06,
      "loss": 0.0608,
      "step": 34640
    },
    {
      "epoch": 3.30352783203125e-05,
      "model_forward_time": 1.7259743213653564,
      "step": 34640
    },
    {
      "dynamics_loss_avg": 0.06388986315578223,
      "epoch": 3.30352783203125e-05,
      "step": 34640
    },
    {
      "action_loss_avg": 0.0022234788921196014,
      "epoch": 3.30352783203125e-05,
      "step": 34640
    },
    {
      "epoch": 3.30352783203125e-05,
      "step": 34640,
      "training_step_time": 7.862860202789307
    },
    {
      "epoch": 3.3036231994628905e-05,
      "model_forward_time": 1.7415947914123535,
      "step": 34641
    },
    {
      "epoch": 3.3036231994628905e-05,
      "step": 34641,
      "training_step_time": 7.707332611083984
    },
    {
      "epoch": 3.3037185668945315e-05,
      "model_forward_time": 1.7275805473327637,
      "step": 34642
    },
    {
      "epoch": 3.3037185668945315e-05,
      "step": 34642,
      "training_step_time": 7.480787515640259
    },
    {
      "epoch": 3.303813934326172e-05,
      "model_forward_time": 1.7238879203796387,
      "step": 34643
    },
    {
      "epoch": 3.303813934326172e-05,
      "step": 34643,
      "training_step_time": 7.5949156284332275
    },
    {
      "epoch": 3.3039093017578123e-05,
      "model_forward_time": 1.7312204837799072,
      "step": 34644
    },
    {
      "epoch": 3.3039093017578123e-05,
      "step": 34644,
      "training_step_time": 7.3784825801849365
    },
    {
      "epoch": 3.3040046691894534e-05,
      "model_forward_time": 1.7548859119415283,
      "step": 34645
    },
    {
      "epoch": 3.3040046691894534e-05,
      "step": 34645,
      "training_step_time": 7.879482984542847
    },
    {
      "epoch": 3.304100036621094e-05,
      "model_forward_time": 1.7361698150634766,
      "step": 34646
    },
    {
      "epoch": 3.304100036621094e-05,
      "step": 34646,
      "training_step_time": 7.688036918640137
    },
    {
      "epoch": 3.304195404052734e-05,
      "model_forward_time": 2.292476177215576,
      "step": 34647
    },
    {
      "epoch": 3.304195404052734e-05,
      "step": 34647,
      "training_step_time": 7.6881091594696045
    },
    {
      "epoch": 3.304290771484375e-05,
      "model_forward_time": 1.7279314994812012,
      "step": 34648
    },
    {
      "epoch": 3.304290771484375e-05,
      "step": 34648,
      "training_step_time": 7.79159140586853
    },
    {
      "epoch": 3.304386138916016e-05,
      "model_forward_time": 1.739180088043213,
      "step": 34649
    },
    {
      "epoch": 3.304386138916016e-05,
      "step": 34649,
      "training_step_time": 9.639352798461914
    },
    {
      "epoch": 3.304481506347656e-05,
      "grad_norm": 0.1506088823080063,
      "learning_rate": 7.783141930413545e-06,
      "loss": 0.0643,
      "step": 34650
    },
    {
      "epoch": 3.304481506347656e-05,
      "model_forward_time": 1.7371482849121094,
      "step": 34650
    },
    {
      "dynamics_loss_avg": 0.04413651470094919,
      "epoch": 3.304481506347656e-05,
      "step": 34650
    },
    {
      "action_loss_avg": 0.0013559734943555668,
      "epoch": 3.304481506347656e-05,
      "step": 34650
    },
    {
      "epoch": 3.304481506347656e-05,
      "step": 34650,
      "training_step_time": 7.578256368637085
    },
    {
      "epoch": 3.304576873779297e-05,
      "model_forward_time": 1.7133066654205322,
      "step": 34651
    },
    {
      "epoch": 3.304576873779297e-05,
      "step": 34651,
      "training_step_time": 7.434357643127441
    },
    {
      "epoch": 3.3046722412109376e-05,
      "model_forward_time": 1.723409652709961,
      "step": 34652
    },
    {
      "epoch": 3.3046722412109376e-05,
      "step": 34652,
      "training_step_time": 7.56370735168457
    },
    {
      "epoch": 3.304767608642578e-05,
      "model_forward_time": 1.7326455116271973,
      "step": 34653
    },
    {
      "epoch": 3.304767608642578e-05,
      "step": 34653,
      "training_step_time": 7.397641181945801
    },
    {
      "epoch": 3.304862976074219e-05,
      "model_forward_time": 1.750326156616211,
      "step": 34654
    },
    {
      "epoch": 3.304862976074219e-05,
      "step": 34654,
      "training_step_time": 7.815586090087891
    },
    {
      "epoch": 3.3049583435058595e-05,
      "model_forward_time": 1.7196297645568848,
      "step": 34655
    },
    {
      "epoch": 3.3049583435058595e-05,
      "step": 34655,
      "training_step_time": 8.657896041870117
    },
    {
      "epoch": 3.3050537109375e-05,
      "model_forward_time": 2.2920098304748535,
      "step": 34656
    },
    {
      "epoch": 3.3050537109375e-05,
      "step": 34656,
      "training_step_time": 8.740024328231812
    },
    {
      "epoch": 3.30514907836914e-05,
      "model_forward_time": 1.7209467887878418,
      "step": 34657
    },
    {
      "epoch": 3.30514907836914e-05,
      "step": 34657,
      "training_step_time": 7.7606799602508545
    },
    {
      "epoch": 3.3052444458007814e-05,
      "model_forward_time": 1.7199552059173584,
      "step": 34658
    },
    {
      "epoch": 3.3052444458007814e-05,
      "step": 34658,
      "training_step_time": 8.804460763931274
    },
    {
      "epoch": 3.305339813232422e-05,
      "model_forward_time": 1.7291970252990723,
      "step": 34659
    },
    {
      "epoch": 3.305339813232422e-05,
      "step": 34659,
      "training_step_time": 7.666274070739746
    },
    {
      "epoch": 3.305435180664062e-05,
      "grad_norm": 0.11697687208652496,
      "learning_rate": 7.781768139023431e-06,
      "loss": 0.0615,
      "step": 34660
    },
    {
      "epoch": 3.305435180664062e-05,
      "model_forward_time": 1.7186081409454346,
      "step": 34660
    },
    {
      "dynamics_loss_avg": 0.053157296217978,
      "epoch": 3.305435180664062e-05,
      "step": 34660
    },
    {
      "action_loss_avg": 0.001313231495441869,
      "epoch": 3.305435180664062e-05,
      "step": 34660
    },
    {
      "epoch": 3.305435180664062e-05,
      "step": 34660,
      "training_step_time": 8.0819730758667
    },
    {
      "epoch": 3.305530548095703e-05,
      "model_forward_time": 1.726128339767456,
      "step": 34661
    },
    {
      "epoch": 3.305530548095703e-05,
      "step": 34661,
      "training_step_time": 7.569526433944702
    },
    {
      "epoch": 3.3056259155273437e-05,
      "model_forward_time": 1.7240309715270996,
      "step": 34662
    },
    {
      "epoch": 3.3056259155273437e-05,
      "step": 34662,
      "training_step_time": 7.375086069107056
    },
    {
      "epoch": 3.305721282958984e-05,
      "model_forward_time": 1.7471363544464111,
      "step": 34663
    },
    {
      "epoch": 3.305721282958984e-05,
      "step": 34663,
      "training_step_time": 7.770114898681641
    },
    {
      "epoch": 3.305816650390625e-05,
      "model_forward_time": 1.3460612297058105,
      "step": 34664
    },
    {
      "epoch": 3.305816650390625e-05,
      "step": 34664,
      "training_step_time": 7.633326530456543
    },
    {
      "epoch": 3.3059120178222655e-05,
      "model_forward_time": 1.73884916305542,
      "step": 34665
    },
    {
      "epoch": 3.3059120178222655e-05,
      "step": 34665,
      "training_step_time": 10.641773223876953
    },
    {
      "epoch": 3.306007385253906e-05,
      "model_forward_time": 1.7367815971374512,
      "step": 34666
    },
    {
      "epoch": 3.306007385253906e-05,
      "step": 34666,
      "training_step_time": 9.338655471801758
    },
    {
      "epoch": 3.306102752685547e-05,
      "model_forward_time": 1.876079797744751,
      "step": 34667
    },
    {
      "epoch": 3.306102752685547e-05,
      "step": 34667,
      "training_step_time": 8.312172889709473
    },
    {
      "epoch": 3.3061981201171874e-05,
      "model_forward_time": 1.753152847290039,
      "step": 34668
    },
    {
      "epoch": 3.3061981201171874e-05,
      "step": 34668,
      "training_step_time": 8.386888027191162
    },
    {
      "epoch": 3.306293487548828e-05,
      "model_forward_time": 1.7391092777252197,
      "step": 34669
    },
    {
      "epoch": 3.306293487548828e-05,
      "step": 34669,
      "training_step_time": 7.479383945465088
    },
    {
      "epoch": 3.306388854980469e-05,
      "grad_norm": 0.16100835800170898,
      "learning_rate": 7.780394043423337e-06,
      "loss": 0.0604,
      "step": 34670
    },
    {
      "epoch": 3.306388854980469e-05,
      "model_forward_time": 1.7538280487060547,
      "step": 34670
    },
    {
      "dynamics_loss_avg": 0.059857673943042755,
      "epoch": 3.306388854980469e-05,
      "step": 34670
    },
    {
      "action_loss_avg": 0.0020955754444003105,
      "epoch": 3.306388854980469e-05,
      "step": 34670
    },
    {
      "epoch": 3.306388854980469e-05,
      "step": 34670,
      "training_step_time": 7.534571170806885
    },
    {
      "epoch": 3.306484222412109e-05,
      "model_forward_time": 1.7528226375579834,
      "step": 34671
    },
    {
      "epoch": 3.306484222412109e-05,
      "step": 34671,
      "training_step_time": 7.641803741455078
    },
    {
      "epoch": 3.30657958984375e-05,
      "model_forward_time": 1.7460434436798096,
      "step": 34672
    },
    {
      "epoch": 3.30657958984375e-05,
      "step": 34672,
      "training_step_time": 7.75849175453186
    },
    {
      "epoch": 3.306674957275391e-05,
      "model_forward_time": 1.7468607425689697,
      "step": 34673
    },
    {
      "epoch": 3.306674957275391e-05,
      "step": 34673,
      "training_step_time": 8.16278624534607
    },
    {
      "epoch": 3.306770324707031e-05,
      "model_forward_time": 1.6862235069274902,
      "step": 34674
    },
    {
      "epoch": 3.306770324707031e-05,
      "step": 34674,
      "training_step_time": 8.158175468444824
    },
    {
      "epoch": 3.3068656921386716e-05,
      "model_forward_time": 2.230329990386963,
      "step": 34675
    },
    {
      "epoch": 3.3068656921386716e-05,
      "step": 34675,
      "training_step_time": 7.739103078842163
    },
    {
      "epoch": 3.306961059570313e-05,
      "model_forward_time": 1.6932265758514404,
      "step": 34676
    },
    {
      "epoch": 3.306961059570313e-05,
      "step": 34676,
      "training_step_time": 8.842483758926392
    },
    {
      "epoch": 3.307056427001953e-05,
      "model_forward_time": 1.6978652477264404,
      "step": 34677
    },
    {
      "epoch": 3.307056427001953e-05,
      "step": 34677,
      "training_step_time": 8.484062433242798
    },
    {
      "epoch": 3.3071517944335935e-05,
      "model_forward_time": 1.6931672096252441,
      "step": 34678
    },
    {
      "epoch": 3.3071517944335935e-05,
      "step": 34678,
      "training_step_time": 7.781332015991211
    },
    {
      "epoch": 3.3072471618652346e-05,
      "model_forward_time": 1.704876184463501,
      "step": 34679
    },
    {
      "epoch": 3.3072471618652346e-05,
      "step": 34679,
      "training_step_time": 7.691293478012085
    },
    {
      "epoch": 3.307342529296875e-05,
      "grad_norm": 0.11853382736444473,
      "learning_rate": 7.77901964376353e-06,
      "loss": 0.0571,
      "step": 34680
    },
    {
      "epoch": 3.307342529296875e-05,
      "model_forward_time": 1.7054872512817383,
      "step": 34680
    },
    {
      "dynamics_loss_avg": 0.050641918741166594,
      "epoch": 3.307342529296875e-05,
      "step": 34680
    },
    {
      "action_loss_avg": 0.0014691665302962065,
      "epoch": 3.307342529296875e-05,
      "step": 34680
    },
    {
      "epoch": 3.307342529296875e-05,
      "step": 34680,
      "training_step_time": 6.7674243450164795
    },
    {
      "epoch": 3.3074378967285154e-05,
      "model_forward_time": 1.7243037223815918,
      "step": 34681
    },
    {
      "epoch": 3.3074378967285154e-05,
      "step": 34681,
      "training_step_time": 7.655514478683472
    },
    {
      "epoch": 3.3075332641601565e-05,
      "model_forward_time": 1.7026102542877197,
      "step": 34682
    },
    {
      "epoch": 3.3075332641601565e-05,
      "step": 34682,
      "training_step_time": 7.88060736656189
    },
    {
      "epoch": 3.307628631591797e-05,
      "model_forward_time": 1.7113947868347168,
      "step": 34683
    },
    {
      "epoch": 3.307628631591797e-05,
      "step": 34683,
      "training_step_time": 8.312992334365845
    },
    {
      "epoch": 3.307723999023437e-05,
      "model_forward_time": 2.2485647201538086,
      "step": 34684
    },
    {
      "epoch": 3.307723999023437e-05,
      "step": 34684,
      "training_step_time": 7.810011148452759
    },
    {
      "epoch": 3.3078193664550783e-05,
      "model_forward_time": 1.709787130355835,
      "step": 34685
    },
    {
      "epoch": 3.3078193664550783e-05,
      "step": 34685,
      "training_step_time": 8.368250608444214
    },
    {
      "epoch": 3.307914733886719e-05,
      "model_forward_time": 1.711507797241211,
      "step": 34686
    },
    {
      "epoch": 3.307914733886719e-05,
      "step": 34686,
      "training_step_time": 7.887232303619385
    },
    {
      "epoch": 3.308010101318359e-05,
      "model_forward_time": 1.6858789920806885,
      "step": 34687
    },
    {
      "epoch": 3.308010101318359e-05,
      "step": 34687,
      "training_step_time": 7.706756591796875
    },
    {
      "epoch": 3.30810546875e-05,
      "model_forward_time": 1.7031679153442383,
      "step": 34688
    },
    {
      "epoch": 3.30810546875e-05,
      "step": 34688,
      "training_step_time": 7.6365416049957275
    },
    {
      "epoch": 3.3082008361816406e-05,
      "model_forward_time": 1.7015559673309326,
      "step": 34689
    },
    {
      "epoch": 3.3082008361816406e-05,
      "step": 34689,
      "training_step_time": 6.8164145946502686
    },
    {
      "epoch": 3.308296203613281e-05,
      "grad_norm": 0.14899468421936035,
      "learning_rate": 7.777644940194315e-06,
      "loss": 0.0597,
      "step": 34690
    },
    {
      "epoch": 3.308296203613281e-05,
      "model_forward_time": 1.7170515060424805,
      "step": 34690
    },
    {
      "dynamics_loss_avg": 0.04429894369095564,
      "epoch": 3.308296203613281e-05,
      "step": 34690
    },
    {
      "action_loss_avg": 0.0011327612824970855,
      "epoch": 3.308296203613281e-05,
      "step": 34690
    },
    {
      "epoch": 3.308296203613281e-05,
      "step": 34690,
      "training_step_time": 8.396870136260986
    },
    {
      "epoch": 3.308391571044922e-05,
      "model_forward_time": 1.7004292011260986,
      "step": 34691
    },
    {
      "epoch": 3.308391571044922e-05,
      "step": 34691,
      "training_step_time": 8.6189603805542
    },
    {
      "epoch": 3.3084869384765625e-05,
      "model_forward_time": 1.69956374168396,
      "step": 34692
    },
    {
      "epoch": 3.3084869384765625e-05,
      "step": 34692,
      "training_step_time": 7.5900654792785645
    },
    {
      "epoch": 3.308582305908203e-05,
      "model_forward_time": 2.2346668243408203,
      "step": 34693
    },
    {
      "epoch": 3.308582305908203e-05,
      "step": 34693,
      "training_step_time": 10.27937364578247
    },
    {
      "epoch": 3.308677673339844e-05,
      "model_forward_time": 1.7040126323699951,
      "step": 34694
    },
    {
      "epoch": 3.308677673339844e-05,
      "step": 34694,
      "training_step_time": 8.649696826934814
    },
    {
      "epoch": 3.3087730407714844e-05,
      "model_forward_time": 1.694509506225586,
      "step": 34695
    },
    {
      "epoch": 3.3087730407714844e-05,
      "step": 34695,
      "training_step_time": 8.433768272399902
    },
    {
      "epoch": 3.308868408203125e-05,
      "model_forward_time": 1.6874842643737793,
      "step": 34696
    },
    {
      "epoch": 3.308868408203125e-05,
      "step": 34696,
      "training_step_time": 7.642390727996826
    },
    {
      "epoch": 3.308963775634766e-05,
      "model_forward_time": 1.7036283016204834,
      "step": 34697
    },
    {
      "epoch": 3.308963775634766e-05,
      "step": 34697,
      "training_step_time": 7.670007944107056
    },
    {
      "epoch": 3.309059143066406e-05,
      "model_forward_time": 1.3313639163970947,
      "step": 34698
    },
    {
      "epoch": 3.309059143066406e-05,
      "step": 34698,
      "training_step_time": 6.838615417480469
    },
    {
      "epoch": 3.309154510498047e-05,
      "model_forward_time": 1.7116568088531494,
      "step": 34699
    },
    {
      "epoch": 3.309154510498047e-05,
      "step": 34699,
      "training_step_time": 7.814366579055786
    },
    {
      "epoch": 3.309249877929688e-05,
      "grad_norm": 0.16975748538970947,
      "learning_rate": 7.776269932866024e-06,
      "loss": 0.0639,
      "step": 34700
    },
    {
      "epoch": 3.309249877929688e-05,
      "model_forward_time": 1.6996350288391113,
      "step": 34700
    },
    {
      "dynamics_loss_avg": 0.04404228907078504,
      "epoch": 3.309249877929688e-05,
      "step": 34700
    },
    {
      "action_loss_avg": 0.0018434933503158391,
      "epoch": 3.309249877929688e-05,
      "step": 34700
    },
    {
      "epoch": 3.309249877929688e-05,
      "step": 34700,
      "training_step_time": 7.901705741882324
    },
    {
      "epoch": 3.309345245361328e-05,
      "model_forward_time": 1.7009410858154297,
      "step": 34701
    },
    {
      "epoch": 3.309345245361328e-05,
      "step": 34701,
      "training_step_time": 9.467612743377686
    },
    {
      "epoch": 3.3094406127929686e-05,
      "model_forward_time": 2.236907958984375,
      "step": 34702
    },
    {
      "epoch": 3.3094406127929686e-05,
      "step": 34702,
      "training_step_time": 7.764211893081665
    },
    {
      "epoch": 3.3095359802246097e-05,
      "model_forward_time": 1.6995949745178223,
      "step": 34703
    },
    {
      "epoch": 3.3095359802246097e-05,
      "step": 34703,
      "training_step_time": 8.593750238418579
    },
    {
      "epoch": 3.30963134765625e-05,
      "model_forward_time": 1.326430082321167,
      "step": 34704
    },
    {
      "epoch": 3.30963134765625e-05,
      "step": 34704,
      "training_step_time": 9.852413654327393
    },
    {
      "epoch": 3.3097267150878905e-05,
      "model_forward_time": 1.7048122882843018,
      "step": 34705
    },
    {
      "epoch": 3.3097267150878905e-05,
      "step": 34705,
      "training_step_time": 8.608006954193115
    },
    {
      "epoch": 3.3098220825195315e-05,
      "model_forward_time": 1.7075631618499756,
      "step": 34706
    },
    {
      "epoch": 3.3098220825195315e-05,
      "step": 34706,
      "training_step_time": 7.689157724380493
    },
    {
      "epoch": 3.309917449951172e-05,
      "model_forward_time": 1.7044813632965088,
      "step": 34707
    },
    {
      "epoch": 3.309917449951172e-05,
      "step": 34707,
      "training_step_time": 7.209985733032227
    },
    {
      "epoch": 3.3100128173828123e-05,
      "model_forward_time": 1.3322882652282715,
      "step": 34708
    },
    {
      "epoch": 3.3100128173828123e-05,
      "step": 34708,
      "training_step_time": 7.800527095794678
    },
    {
      "epoch": 3.3101081848144534e-05,
      "model_forward_time": 1.6980676651000977,
      "step": 34709
    },
    {
      "epoch": 3.3101081848144534e-05,
      "step": 34709,
      "training_step_time": 7.957925319671631
    },
    {
      "epoch": 3.310203552246094e-05,
      "grad_norm": 0.19827352464199066,
      "learning_rate": 7.774894621929026e-06,
      "loss": 0.0635,
      "step": 34710
    },
    {
      "epoch": 3.310203552246094e-05,
      "model_forward_time": 1.7092723846435547,
      "step": 34710
    },
    {
      "dynamics_loss_avg": 0.05722754336893558,
      "epoch": 3.310203552246094e-05,
      "step": 34710
    },
    {
      "action_loss_avg": 0.0014443627616856248,
      "epoch": 3.310203552246094e-05,
      "step": 34710
    },
    {
      "epoch": 3.310203552246094e-05,
      "step": 34710,
      "training_step_time": 9.414282083511353
    },
    {
      "epoch": 3.310298919677734e-05,
      "model_forward_time": 1.7154741287231445,
      "step": 34711
    },
    {
      "epoch": 3.310298919677734e-05,
      "step": 34711,
      "training_step_time": 9.182531356811523
    },
    {
      "epoch": 3.310394287109375e-05,
      "model_forward_time": 2.224118947982788,
      "step": 34712
    },
    {
      "epoch": 3.310394287109375e-05,
      "step": 34712,
      "training_step_time": 9.303494215011597
    },
    {
      "epoch": 3.310489654541016e-05,
      "model_forward_time": 1.704942226409912,
      "step": 34713
    },
    {
      "epoch": 3.310489654541016e-05,
      "step": 34713,
      "training_step_time": 8.835130214691162
    },
    {
      "epoch": 3.310585021972656e-05,
      "model_forward_time": 1.7074851989746094,
      "step": 34714
    },
    {
      "epoch": 3.310585021972656e-05,
      "step": 34714,
      "training_step_time": 7.786463499069214
    },
    {
      "epoch": 3.310680389404297e-05,
      "model_forward_time": 1.7083113193511963,
      "step": 34715
    },
    {
      "epoch": 3.310680389404297e-05,
      "step": 34715,
      "training_step_time": 7.668569326400757
    },
    {
      "epoch": 3.3107757568359376e-05,
      "model_forward_time": 1.6957876682281494,
      "step": 34716
    },
    {
      "epoch": 3.3107757568359376e-05,
      "step": 34716,
      "training_step_time": 7.248445510864258
    },
    {
      "epoch": 3.310871124267578e-05,
      "model_forward_time": 1.6951189041137695,
      "step": 34717
    },
    {
      "epoch": 3.310871124267578e-05,
      "step": 34717,
      "training_step_time": 8.557145595550537
    },
    {
      "epoch": 3.310966491699219e-05,
      "model_forward_time": 1.6991543769836426,
      "step": 34718
    },
    {
      "epoch": 3.310966491699219e-05,
      "step": 34718,
      "training_step_time": 7.398189067840576
    },
    {
      "epoch": 3.3110618591308595e-05,
      "model_forward_time": 1.6936500072479248,
      "step": 34719
    },
    {
      "epoch": 3.3110618591308595e-05,
      "step": 34719,
      "training_step_time": 8.748716592788696
    },
    {
      "epoch": 3.3111572265625e-05,
      "grad_norm": 0.16763626039028168,
      "learning_rate": 7.773519007533726e-06,
      "loss": 0.0585,
      "step": 34720
    },
    {
      "epoch": 3.3111572265625e-05,
      "model_forward_time": 1.7207303047180176,
      "step": 34720
    },
    {
      "dynamics_loss_avg": 0.05633716955780983,
      "epoch": 3.3111572265625e-05,
      "step": 34720
    },
    {
      "action_loss_avg": 0.0025539688474964352,
      "epoch": 3.3111572265625e-05,
      "step": 34720
    },
    {
      "epoch": 3.3111572265625e-05,
      "step": 34720,
      "training_step_time": 7.835608243942261
    },
    {
      "epoch": 3.31125259399414e-05,
      "model_forward_time": 2.258460521697998,
      "step": 34721
    },
    {
      "epoch": 3.31125259399414e-05,
      "step": 34721,
      "training_step_time": 10.4574556350708
    },
    {
      "epoch": 3.3113479614257814e-05,
      "model_forward_time": 1.7534782886505127,
      "step": 34722
    },
    {
      "epoch": 3.3113479614257814e-05,
      "step": 34722,
      "training_step_time": 8.870723009109497
    },
    {
      "epoch": 3.311443328857422e-05,
      "model_forward_time": 1.8476862907409668,
      "step": 34723
    },
    {
      "epoch": 3.311443328857422e-05,
      "step": 34723,
      "training_step_time": 8.951727151870728
    },
    {
      "epoch": 3.311538696289062e-05,
      "model_forward_time": 1.7501592636108398,
      "step": 34724
    },
    {
      "epoch": 3.311538696289062e-05,
      "step": 34724,
      "training_step_time": 7.751771926879883
    },
    {
      "epoch": 3.311634063720703e-05,
      "model_forward_time": 1.7527081966400146,
      "step": 34725
    },
    {
      "epoch": 3.311634063720703e-05,
      "step": 34725,
      "training_step_time": 7.249274730682373
    },
    {
      "epoch": 3.311729431152344e-05,
      "model_forward_time": 1.7637033462524414,
      "step": 34726
    },
    {
      "epoch": 3.311729431152344e-05,
      "step": 34726,
      "training_step_time": 6.858478307723999
    },
    {
      "epoch": 3.311824798583984e-05,
      "model_forward_time": 1.7539913654327393,
      "step": 34727
    },
    {
      "epoch": 3.311824798583984e-05,
      "step": 34727,
      "training_step_time": 7.757482528686523
    },
    {
      "epoch": 3.311920166015625e-05,
      "model_forward_time": 1.7514100074768066,
      "step": 34728
    },
    {
      "epoch": 3.311920166015625e-05,
      "step": 34728,
      "training_step_time": 7.849902629852295
    },
    {
      "epoch": 3.3120155334472655e-05,
      "model_forward_time": 1.7492334842681885,
      "step": 34729
    },
    {
      "epoch": 3.3120155334472655e-05,
      "step": 34729,
      "training_step_time": 7.775039434432983
    },
    {
      "epoch": 3.312110900878906e-05,
      "grad_norm": 0.163043811917305,
      "learning_rate": 7.772143089830557e-06,
      "loss": 0.0606,
      "step": 34730
    },
    {
      "epoch": 3.312110900878906e-05,
      "model_forward_time": 2.936843156814575,
      "step": 34730
    },
    {
      "dynamics_loss_avg": 0.04514799527823925,
      "epoch": 3.312110900878906e-05,
      "step": 34730
    },
    {
      "action_loss_avg": 0.0015303739084629342,
      "epoch": 3.312110900878906e-05,
      "step": 34730
    },
    {
      "epoch": 3.312110900878906e-05,
      "step": 34730,
      "training_step_time": 7.928415536880493
    },
    {
      "epoch": 3.312206268310547e-05,
      "model_forward_time": 1.7587604522705078,
      "step": 34731
    },
    {
      "epoch": 3.312206268310547e-05,
      "step": 34731,
      "training_step_time": 8.799079656600952
    },
    {
      "epoch": 3.3123016357421874e-05,
      "model_forward_time": 1.8186042308807373,
      "step": 34732
    },
    {
      "epoch": 3.3123016357421874e-05,
      "step": 34732,
      "training_step_time": 7.919504642486572
    },
    {
      "epoch": 3.312397003173828e-05,
      "model_forward_time": 1.750112533569336,
      "step": 34733
    },
    {
      "epoch": 3.312397003173828e-05,
      "step": 34733,
      "training_step_time": 7.726674556732178
    },
    {
      "epoch": 3.312492370605469e-05,
      "model_forward_time": 1.7522971630096436,
      "step": 34734
    },
    {
      "epoch": 3.312492370605469e-05,
      "step": 34734,
      "training_step_time": 7.610406160354614
    },
    {
      "epoch": 3.312587738037109e-05,
      "model_forward_time": 1.7636184692382812,
      "step": 34735
    },
    {
      "epoch": 3.312587738037109e-05,
      "step": 34735,
      "training_step_time": 7.42065167427063
    },
    {
      "epoch": 3.31268310546875e-05,
      "model_forward_time": 1.7474277019500732,
      "step": 34736
    },
    {
      "epoch": 3.31268310546875e-05,
      "step": 34736,
      "training_step_time": 7.817466497421265
    },
    {
      "epoch": 3.312778472900391e-05,
      "model_forward_time": 1.7488048076629639,
      "step": 34737
    },
    {
      "epoch": 3.312778472900391e-05,
      "step": 34737,
      "training_step_time": 8.012049198150635
    },
    {
      "epoch": 3.312873840332031e-05,
      "model_forward_time": 1.7594914436340332,
      "step": 34738
    },
    {
      "epoch": 3.312873840332031e-05,
      "step": 34738,
      "training_step_time": 8.58168625831604
    },
    {
      "epoch": 3.3129692077636716e-05,
      "model_forward_time": 2.902355670928955,
      "step": 34739
    },
    {
      "epoch": 3.3129692077636716e-05,
      "step": 34739,
      "training_step_time": 8.593567609786987
    },
    {
      "epoch": 3.313064575195313e-05,
      "grad_norm": 0.17897997796535492,
      "learning_rate": 7.770766868969985e-06,
      "loss": 0.0554,
      "step": 34740
    },
    {
      "epoch": 3.313064575195313e-05,
      "model_forward_time": 1.764533281326294,
      "step": 34740
    },
    {
      "dynamics_loss_avg": 0.04142630305141211,
      "epoch": 3.313064575195313e-05,
      "step": 34740
    },
    {
      "action_loss_avg": 0.0011337646457832307,
      "epoch": 3.313064575195313e-05,
      "step": 34740
    },
    {
      "epoch": 3.313064575195313e-05,
      "step": 34740,
      "training_step_time": 8.496039390563965
    },
    {
      "epoch": 3.313159942626953e-05,
      "model_forward_time": 1.7864375114440918,
      "step": 34741
    },
    {
      "epoch": 3.313159942626953e-05,
      "step": 34741,
      "training_step_time": 7.924730062484741
    },
    {
      "epoch": 3.3132553100585935e-05,
      "model_forward_time": 1.7503812313079834,
      "step": 34742
    },
    {
      "epoch": 3.3132553100585935e-05,
      "step": 34742,
      "training_step_time": 7.769623756408691
    },
    {
      "epoch": 3.3133506774902346e-05,
      "model_forward_time": 1.7585973739624023,
      "step": 34743
    },
    {
      "epoch": 3.3133506774902346e-05,
      "step": 34743,
      "training_step_time": 7.728326797485352
    },
    {
      "epoch": 3.313446044921875e-05,
      "model_forward_time": 1.7490453720092773,
      "step": 34744
    },
    {
      "epoch": 3.313446044921875e-05,
      "step": 34744,
      "training_step_time": 7.491477966308594
    },
    {
      "epoch": 3.3135414123535154e-05,
      "model_forward_time": 1.7546234130859375,
      "step": 34745
    },
    {
      "epoch": 3.3135414123535154e-05,
      "step": 34745,
      "training_step_time": 7.761515140533447
    },
    {
      "epoch": 3.3136367797851565e-05,
      "model_forward_time": 1.7615280151367188,
      "step": 34746
    },
    {
      "epoch": 3.3136367797851565e-05,
      "step": 34746,
      "training_step_time": 7.916168928146362
    },
    {
      "epoch": 3.313732147216797e-05,
      "model_forward_time": 1.7336606979370117,
      "step": 34747
    },
    {
      "epoch": 3.313732147216797e-05,
      "step": 34747,
      "training_step_time": 7.8058648109436035
    },
    {
      "epoch": 3.313827514648437e-05,
      "model_forward_time": 2.8941915035247803,
      "step": 34748
    },
    {
      "epoch": 3.313827514648437e-05,
      "step": 34748,
      "training_step_time": 7.942030668258667
    },
    {
      "epoch": 3.3139228820800783e-05,
      "model_forward_time": 1.9398918151855469,
      "step": 34749
    },
    {
      "epoch": 3.3139228820800783e-05,
      "step": 34749,
      "training_step_time": 9.443219661712646
    },
    {
      "epoch": 3.314018249511719e-05,
      "grad_norm": 0.17241829633712769,
      "learning_rate": 7.769390345102517e-06,
      "loss": 0.0592,
      "step": 34750
    },
    {
      "epoch": 3.314018249511719e-05,
      "model_forward_time": 1.8990373611450195,
      "step": 34750
    },
    {
      "dynamics_loss_avg": 0.06036239424720406,
      "epoch": 3.314018249511719e-05,
      "step": 34750
    },
    {
      "action_loss_avg": 0.0025195430818712337,
      "epoch": 3.314018249511719e-05,
      "step": 34750
    },
    {
      "epoch": 3.314018249511719e-05,
      "step": 34750,
      "training_step_time": 9.712287902832031
    },
    {
      "epoch": 3.314113616943359e-05,
      "model_forward_time": 1.9411730766296387,
      "step": 34751
    },
    {
      "epoch": 3.314113616943359e-05,
      "step": 34751,
      "training_step_time": 7.9081549644470215
    },
    {
      "epoch": 3.314208984375e-05,
      "model_forward_time": 1.9326324462890625,
      "step": 34752
    },
    {
      "epoch": 3.314208984375e-05,
      "step": 34752,
      "training_step_time": 7.706185579299927
    },
    {
      "epoch": 3.3143043518066406e-05,
      "model_forward_time": 1.934185266494751,
      "step": 34753
    },
    {
      "epoch": 3.3143043518066406e-05,
      "step": 34753,
      "training_step_time": 7.612804889678955
    },
    {
      "epoch": 3.314399719238281e-05,
      "model_forward_time": 1.9161622524261475,
      "step": 34754
    },
    {
      "epoch": 3.314399719238281e-05,
      "step": 34754,
      "training_step_time": 7.720058441162109
    },
    {
      "epoch": 3.314495086669922e-05,
      "model_forward_time": 1.9259617328643799,
      "step": 34755
    },
    {
      "epoch": 3.314495086669922e-05,
      "step": 34755,
      "training_step_time": 7.919149398803711
    },
    {
      "epoch": 3.3145904541015625e-05,
      "model_forward_time": 1.9194118976593018,
      "step": 34756
    },
    {
      "epoch": 3.3145904541015625e-05,
      "step": 34756,
      "training_step_time": 7.764236927032471
    },
    {
      "epoch": 3.314685821533203e-05,
      "model_forward_time": 2.9967668056488037,
      "step": 34757
    },
    {
      "epoch": 3.314685821533203e-05,
      "step": 34757,
      "training_step_time": 7.881579160690308
    },
    {
      "epoch": 3.314781188964844e-05,
      "model_forward_time": 1.9054791927337646,
      "step": 34758
    },
    {
      "epoch": 3.314781188964844e-05,
      "step": 34758,
      "training_step_time": 8.74553894996643
    },
    {
      "epoch": 3.3148765563964844e-05,
      "model_forward_time": 1.918323278427124,
      "step": 34759
    },
    {
      "epoch": 3.3148765563964844e-05,
      "step": 34759,
      "training_step_time": 9.54268193244934
    },
    {
      "epoch": 3.314971923828125e-05,
      "grad_norm": 0.11302579194307327,
      "learning_rate": 7.768013518378683e-06,
      "loss": 0.0576,
      "step": 34760
    },
    {
      "epoch": 3.314971923828125e-05,
      "model_forward_time": 1.9356775283813477,
      "step": 34760
    },
    {
      "dynamics_loss_avg": 0.060519388597458604,
      "epoch": 3.314971923828125e-05,
      "step": 34760
    },
    {
      "action_loss_avg": 0.002430124225793406,
      "epoch": 3.314971923828125e-05,
      "step": 34760
    },
    {
      "epoch": 3.314971923828125e-05,
      "step": 34760,
      "training_step_time": 7.971869945526123
    },
    {
      "epoch": 3.315067291259766e-05,
      "model_forward_time": 1.9147067070007324,
      "step": 34761
    },
    {
      "epoch": 3.315067291259766e-05,
      "step": 34761,
      "training_step_time": 7.686739444732666
    },
    {
      "epoch": 3.315162658691406e-05,
      "model_forward_time": 1.9198734760284424,
      "step": 34762
    },
    {
      "epoch": 3.315162658691406e-05,
      "step": 34762,
      "training_step_time": 7.571877717971802
    },
    {
      "epoch": 3.315258026123047e-05,
      "model_forward_time": 1.9253883361816406,
      "step": 34763
    },
    {
      "epoch": 3.315258026123047e-05,
      "step": 34763,
      "training_step_time": 6.983189344406128
    },
    {
      "epoch": 3.315353393554688e-05,
      "model_forward_time": 1.9269211292266846,
      "step": 34764
    },
    {
      "epoch": 3.315353393554688e-05,
      "step": 34764,
      "training_step_time": 7.9300761222839355
    },
    {
      "epoch": 3.315448760986328e-05,
      "model_forward_time": 1.8882522583007812,
      "step": 34765
    },
    {
      "epoch": 3.315448760986328e-05,
      "step": 34765,
      "training_step_time": 9.515241622924805
    },
    {
      "epoch": 3.3155441284179686e-05,
      "model_forward_time": 1.9047632217407227,
      "step": 34766
    },
    {
      "epoch": 3.3155441284179686e-05,
      "step": 34766,
      "training_step_time": 7.7561354637146
    },
    {
      "epoch": 3.31563949584961e-05,
      "model_forward_time": 3.1061644554138184,
      "step": 34767
    },
    {
      "epoch": 3.31563949584961e-05,
      "step": 34767,
      "training_step_time": 8.75935435295105
    },
    {
      "epoch": 3.31573486328125e-05,
      "model_forward_time": 1.552534818649292,
      "step": 34768
    },
    {
      "epoch": 3.31573486328125e-05,
      "step": 34768,
      "training_step_time": 8.30645203590393
    },
    {
      "epoch": 3.3158302307128905e-05,
      "model_forward_time": 1.79006028175354,
      "step": 34769
    },
    {
      "epoch": 3.3158302307128905e-05,
      "step": 34769,
      "training_step_time": 8.497719526290894
    },
    {
      "epoch": 3.3159255981445315e-05,
      "grad_norm": 0.11508428305387497,
      "learning_rate": 7.766636388949054e-06,
      "loss": 0.0591,
      "step": 34770
    },
    {
      "epoch": 3.3159255981445315e-05,
      "model_forward_time": 1.7317779064178467,
      "step": 34770
    },
    {
      "dynamics_loss_avg": 0.044854399468749764,
      "epoch": 3.3159255981445315e-05,
      "step": 34770
    },
    {
      "action_loss_avg": 0.0014692204946186393,
      "epoch": 3.3159255981445315e-05,
      "step": 34770
    },
    {
      "epoch": 3.3159255981445315e-05,
      "step": 34770,
      "training_step_time": 8.363779783248901
    },
    {
      "epoch": 3.316020965576172e-05,
      "model_forward_time": 1.7312743663787842,
      "step": 34771
    },
    {
      "epoch": 3.316020965576172e-05,
      "step": 34771,
      "training_step_time": 7.862576007843018
    },
    {
      "epoch": 3.3161163330078124e-05,
      "model_forward_time": 1.7272288799285889,
      "step": 34772
    },
    {
      "epoch": 3.3161163330078124e-05,
      "step": 34772,
      "training_step_time": 6.844749212265015
    },
    {
      "epoch": 3.3162117004394534e-05,
      "model_forward_time": 1.7311131954193115,
      "step": 34773
    },
    {
      "epoch": 3.3162117004394534e-05,
      "step": 34773,
      "training_step_time": 8.855913400650024
    },
    {
      "epoch": 3.316307067871094e-05,
      "model_forward_time": 1.826373815536499,
      "step": 34774
    },
    {
      "epoch": 3.316307067871094e-05,
      "step": 34774,
      "training_step_time": 7.8154380321502686
    },
    {
      "epoch": 3.316402435302734e-05,
      "model_forward_time": 1.7255291938781738,
      "step": 34775
    },
    {
      "epoch": 3.316402435302734e-05,
      "step": 34775,
      "training_step_time": 7.759626388549805
    },
    {
      "epoch": 3.316497802734375e-05,
      "model_forward_time": 2.9665627479553223,
      "step": 34776
    },
    {
      "epoch": 3.316497802734375e-05,
      "step": 34776,
      "training_step_time": 9.041762590408325
    },
    {
      "epoch": 3.316593170166016e-05,
      "model_forward_time": 1.7359743118286133,
      "step": 34777
    },
    {
      "epoch": 3.316593170166016e-05,
      "step": 34777,
      "training_step_time": 7.750180959701538
    },
    {
      "epoch": 3.316688537597656e-05,
      "model_forward_time": 1.7462246417999268,
      "step": 34778
    },
    {
      "epoch": 3.316688537597656e-05,
      "step": 34778,
      "training_step_time": 8.46752142906189
    },
    {
      "epoch": 3.316783905029297e-05,
      "model_forward_time": 1.7274689674377441,
      "step": 34779
    },
    {
      "epoch": 3.316783905029297e-05,
      "step": 34779,
      "training_step_time": 8.89663052558899
    },
    {
      "epoch": 3.3168792724609376e-05,
      "grad_norm": 0.16930709779262543,
      "learning_rate": 7.765258956964229e-06,
      "loss": 0.0602,
      "step": 34780
    },
    {
      "epoch": 3.3168792724609376e-05,
      "model_forward_time": 1.7397451400756836,
      "step": 34780
    },
    {
      "dynamics_loss_avg": 0.03783089052885771,
      "epoch": 3.3168792724609376e-05,
      "step": 34780
    },
    {
      "action_loss_avg": 0.001317771035246551,
      "epoch": 3.3168792724609376e-05,
      "step": 34780
    },
    {
      "epoch": 3.3168792724609376e-05,
      "step": 34780,
      "training_step_time": 7.544490814208984
    },
    {
      "epoch": 3.316974639892578e-05,
      "model_forward_time": 1.7311201095581055,
      "step": 34781
    },
    {
      "epoch": 3.316974639892578e-05,
      "step": 34781,
      "training_step_time": 7.506690740585327
    },
    {
      "epoch": 3.3170700073242184e-05,
      "model_forward_time": 1.7272446155548096,
      "step": 34782
    },
    {
      "epoch": 3.3170700073242184e-05,
      "step": 34782,
      "training_step_time": 7.890567779541016
    },
    {
      "epoch": 3.3171653747558595e-05,
      "model_forward_time": 1.8053276538848877,
      "step": 34783
    },
    {
      "epoch": 3.3171653747558595e-05,
      "step": 34783,
      "training_step_time": 8.460423946380615
    },
    {
      "epoch": 3.3172607421875e-05,
      "model_forward_time": 1.7199428081512451,
      "step": 34784
    },
    {
      "epoch": 3.3172607421875e-05,
      "step": 34784,
      "training_step_time": 9.478221654891968
    },
    {
      "epoch": 3.31735610961914e-05,
      "model_forward_time": 2.8919641971588135,
      "step": 34785
    },
    {
      "epoch": 3.31735610961914e-05,
      "step": 34785,
      "training_step_time": 9.336318016052246
    },
    {
      "epoch": 3.3174514770507814e-05,
      "model_forward_time": 1.7339065074920654,
      "step": 34786
    },
    {
      "epoch": 3.3174514770507814e-05,
      "step": 34786,
      "training_step_time": 9.481710433959961
    },
    {
      "epoch": 3.317546844482422e-05,
      "model_forward_time": 1.7262670993804932,
      "step": 34787
    },
    {
      "epoch": 3.317546844482422e-05,
      "step": 34787,
      "training_step_time": 8.257599353790283
    },
    {
      "epoch": 3.317642211914062e-05,
      "model_forward_time": 1.7298698425292969,
      "step": 34788
    },
    {
      "epoch": 3.317642211914062e-05,
      "step": 34788,
      "training_step_time": 7.6367506980896
    },
    {
      "epoch": 3.317737579345703e-05,
      "model_forward_time": 1.7380516529083252,
      "step": 34789
    },
    {
      "epoch": 3.317737579345703e-05,
      "step": 34789,
      "training_step_time": 7.589500188827515
    },
    {
      "epoch": 3.317832946777344e-05,
      "grad_norm": 0.10478890687227249,
      "learning_rate": 7.76388122257484e-06,
      "loss": 0.0611,
      "step": 34790
    },
    {
      "epoch": 3.317832946777344e-05,
      "model_forward_time": 1.7335431575775146,
      "step": 34790
    },
    {
      "dynamics_loss_avg": 0.04720036871731281,
      "epoch": 3.317832946777344e-05,
      "step": 34790
    },
    {
      "action_loss_avg": 0.0011551653646165506,
      "epoch": 3.317832946777344e-05,
      "step": 34790
    },
    {
      "epoch": 3.317832946777344e-05,
      "step": 34790,
      "training_step_time": 7.60735297203064
    },
    {
      "epoch": 3.317928314208984e-05,
      "model_forward_time": 1.7227528095245361,
      "step": 34791
    },
    {
      "epoch": 3.317928314208984e-05,
      "step": 34791,
      "training_step_time": 7.8391900062561035
    },
    {
      "epoch": 3.318023681640625e-05,
      "model_forward_time": 1.758589506149292,
      "step": 34792
    },
    {
      "epoch": 3.318023681640625e-05,
      "step": 34792,
      "training_step_time": 7.752994775772095
    },
    {
      "epoch": 3.3181190490722656e-05,
      "model_forward_time": 1.7269141674041748,
      "step": 34793
    },
    {
      "epoch": 3.3181190490722656e-05,
      "step": 34793,
      "training_step_time": 9.446109056472778
    },
    {
      "epoch": 3.318214416503906e-05,
      "model_forward_time": 2.8624343872070312,
      "step": 34794
    },
    {
      "epoch": 3.318214416503906e-05,
      "step": 34794,
      "training_step_time": 8.791220664978027
    },
    {
      "epoch": 3.318309783935547e-05,
      "model_forward_time": 1.7339351177215576,
      "step": 34795
    },
    {
      "epoch": 3.318309783935547e-05,
      "step": 34795,
      "training_step_time": 8.743225574493408
    },
    {
      "epoch": 3.3184051513671874e-05,
      "model_forward_time": 1.7300035953521729,
      "step": 34796
    },
    {
      "epoch": 3.3184051513671874e-05,
      "step": 34796,
      "training_step_time": 8.74280571937561
    },
    {
      "epoch": 3.318500518798828e-05,
      "model_forward_time": 1.7169885635375977,
      "step": 34797
    },
    {
      "epoch": 3.318500518798828e-05,
      "step": 34797,
      "training_step_time": 7.966728687286377
    },
    {
      "epoch": 3.318595886230469e-05,
      "model_forward_time": 1.724301815032959,
      "step": 34798
    },
    {
      "epoch": 3.318595886230469e-05,
      "step": 34798,
      "training_step_time": 7.656756162643433
    },
    {
      "epoch": 3.318691253662109e-05,
      "model_forward_time": 1.763840675354004,
      "step": 34799
    },
    {
      "epoch": 3.318691253662109e-05,
      "step": 34799,
      "training_step_time": 7.552204847335815
    },
    {
      "epoch": 3.31878662109375e-05,
      "grad_norm": 0.2302374243736267,
      "learning_rate": 7.762503185931557e-06,
      "loss": 0.0603,
      "step": 34800
    },
    {
      "epoch": 3.31878662109375e-05,
      "model_forward_time": 1.713658332824707,
      "step": 34800
    },
    {
      "dynamics_loss_avg": 0.0570629796013236,
      "epoch": 3.31878662109375e-05,
      "step": 34800
    },
    {
      "action_loss_avg": 0.0017978876421693712,
      "epoch": 3.31878662109375e-05,
      "step": 34800
    },
    {
      "epoch": 3.31878662109375e-05,
      "step": 34800,
      "training_step_time": 7.774764537811279
    },
    {
      "epoch": 3.318881988525391e-05,
      "model_forward_time": 1.728010892868042,
      "step": 34801
    },
    {
      "epoch": 3.318881988525391e-05,
      "step": 34801,
      "training_step_time": 7.7396416664123535
    },
    {
      "epoch": 3.318977355957031e-05,
      "model_forward_time": 1.8509514331817627,
      "step": 34802
    },
    {
      "epoch": 3.318977355957031e-05,
      "step": 34802,
      "training_step_time": 8.278383731842041
    },
    {
      "epoch": 3.3190727233886716e-05,
      "model_forward_time": 2.8493194580078125,
      "step": 34803
    },
    {
      "epoch": 3.3190727233886716e-05,
      "step": 34803,
      "training_step_time": 8.560432195663452
    },
    {
      "epoch": 3.319168090820313e-05,
      "model_forward_time": 1.359405755996704,
      "step": 34804
    },
    {
      "epoch": 3.319168090820313e-05,
      "step": 34804,
      "training_step_time": 7.873995542526245
    },
    {
      "epoch": 3.319263458251953e-05,
      "model_forward_time": 1.731541633605957,
      "step": 34805
    },
    {
      "epoch": 3.319263458251953e-05,
      "step": 34805,
      "training_step_time": 9.276142358779907
    },
    {
      "epoch": 3.3193588256835935e-05,
      "model_forward_time": 1.7286715507507324,
      "step": 34806
    },
    {
      "epoch": 3.3193588256835935e-05,
      "step": 34806,
      "training_step_time": 7.946855545043945
    },
    {
      "epoch": 3.3194541931152346e-05,
      "model_forward_time": 1.7251818180084229,
      "step": 34807
    },
    {
      "epoch": 3.3194541931152346e-05,
      "step": 34807,
      "training_step_time": 7.721785306930542
    },
    {
      "epoch": 3.319549560546875e-05,
      "model_forward_time": 1.733020544052124,
      "step": 34808
    },
    {
      "epoch": 3.319549560546875e-05,
      "step": 34808,
      "training_step_time": 7.515733480453491
    },
    {
      "epoch": 3.3196449279785154e-05,
      "model_forward_time": 1.730499267578125,
      "step": 34809
    },
    {
      "epoch": 3.3196449279785154e-05,
      "step": 34809,
      "training_step_time": 7.812986373901367
    },
    {
      "epoch": 3.3197402954101565e-05,
      "grad_norm": 0.12074356526136398,
      "learning_rate": 7.76112484718508e-06,
      "loss": 0.0592,
      "step": 34810
    },
    {
      "epoch": 3.3197402954101565e-05,
      "model_forward_time": 1.7217028141021729,
      "step": 34810
    },
    {
      "dynamics_loss_avg": 0.052183459512889385,
      "epoch": 3.3197402954101565e-05,
      "step": 34810
    },
    {
      "action_loss_avg": 0.0020884456753265113,
      "epoch": 3.3197402954101565e-05,
      "step": 34810
    },
    {
      "epoch": 3.3197402954101565e-05,
      "step": 34810,
      "training_step_time": 7.706120252609253
    },
    {
      "epoch": 3.319835662841797e-05,
      "model_forward_time": 1.8424923419952393,
      "step": 34811
    },
    {
      "epoch": 3.319835662841797e-05,
      "step": 34811,
      "training_step_time": 9.611660480499268
    },
    {
      "epoch": 3.319931030273437e-05,
      "model_forward_time": 1.7236700057983398,
      "step": 34812
    },
    {
      "epoch": 3.319931030273437e-05,
      "step": 34812,
      "training_step_time": 9.247876644134521
    },
    {
      "epoch": 3.3200263977050784e-05,
      "model_forward_time": 2.827274799346924,
      "step": 34813
    },
    {
      "epoch": 3.3200263977050784e-05,
      "step": 34813,
      "training_step_time": 9.694886207580566
    },
    {
      "epoch": 3.320121765136719e-05,
      "model_forward_time": 1.7200531959533691,
      "step": 34814
    },
    {
      "epoch": 3.320121765136719e-05,
      "step": 34814,
      "training_step_time": 8.733969449996948
    },
    {
      "epoch": 3.320217132568359e-05,
      "model_forward_time": 1.7316069602966309,
      "step": 34815
    },
    {
      "epoch": 3.320217132568359e-05,
      "step": 34815,
      "training_step_time": 10.011976957321167
    },
    {
      "epoch": 3.3203125e-05,
      "model_forward_time": 1.7684190273284912,
      "step": 34816
    },
    {
      "epoch": 3.3203125e-05,
      "step": 34816,
      "training_step_time": 7.628431558609009
    },
    {
      "epoch": 3.3204078674316406e-05,
      "model_forward_time": 1.7240722179412842,
      "step": 34817
    },
    {
      "epoch": 3.3204078674316406e-05,
      "step": 34817,
      "training_step_time": 8.605262994766235
    },
    {
      "epoch": 3.320503234863281e-05,
      "model_forward_time": 1.7206690311431885,
      "step": 34818
    },
    {
      "epoch": 3.320503234863281e-05,
      "step": 34818,
      "training_step_time": 7.734659671783447
    },
    {
      "epoch": 3.320598602294922e-05,
      "model_forward_time": 1.7270534038543701,
      "step": 34819
    },
    {
      "epoch": 3.320598602294922e-05,
      "step": 34819,
      "training_step_time": 7.754456043243408
    },
    {
      "epoch": 3.3206939697265625e-05,
      "grad_norm": 0.1125088483095169,
      "learning_rate": 7.75974620648614e-06,
      "loss": 0.0606,
      "step": 34820
    },
    {
      "epoch": 3.3206939697265625e-05,
      "model_forward_time": 1.7775368690490723,
      "step": 34820
    },
    {
      "dynamics_loss_avg": 0.05142645034939051,
      "epoch": 3.3206939697265625e-05,
      "step": 34820
    },
    {
      "action_loss_avg": 0.0012535635440144687,
      "epoch": 3.3206939697265625e-05,
      "step": 34820
    },
    {
      "epoch": 3.3206939697265625e-05,
      "step": 34820,
      "training_step_time": 7.785515308380127
    },
    {
      "epoch": 3.320789337158203e-05,
      "model_forward_time": 1.7280867099761963,
      "step": 34821
    },
    {
      "epoch": 3.320789337158203e-05,
      "step": 34821,
      "training_step_time": 7.79315710067749
    },
    {
      "epoch": 3.320884704589844e-05,
      "model_forward_time": 2.8741002082824707,
      "step": 34822
    },
    {
      "epoch": 3.320884704589844e-05,
      "step": 34822,
      "training_step_time": 8.369550228118896
    },
    {
      "epoch": 3.3209800720214844e-05,
      "model_forward_time": 1.72471284866333,
      "step": 34823
    },
    {
      "epoch": 3.3209800720214844e-05,
      "step": 34823,
      "training_step_time": 7.776736497879028
    },
    {
      "epoch": 3.321075439453125e-05,
      "model_forward_time": 1.730243444442749,
      "step": 34824
    },
    {
      "epoch": 3.321075439453125e-05,
      "step": 34824,
      "training_step_time": 8.70129108428955
    },
    {
      "epoch": 3.321170806884766e-05,
      "model_forward_time": 1.728184461593628,
      "step": 34825
    },
    {
      "epoch": 3.321170806884766e-05,
      "step": 34825,
      "training_step_time": 8.625646591186523
    },
    {
      "epoch": 3.321266174316406e-05,
      "model_forward_time": 1.7264680862426758,
      "step": 34826
    },
    {
      "epoch": 3.321266174316406e-05,
      "step": 34826,
      "training_step_time": 7.597359657287598
    },
    {
      "epoch": 3.321361541748047e-05,
      "model_forward_time": 1.7253265380859375,
      "step": 34827
    },
    {
      "epoch": 3.321361541748047e-05,
      "step": 34827,
      "training_step_time": 7.799553871154785
    },
    {
      "epoch": 3.321456909179688e-05,
      "model_forward_time": 1.7262601852416992,
      "step": 34828
    },
    {
      "epoch": 3.321456909179688e-05,
      "step": 34828,
      "training_step_time": 7.695529937744141
    },
    {
      "epoch": 3.321552276611328e-05,
      "model_forward_time": 1.751978874206543,
      "step": 34829
    },
    {
      "epoch": 3.321552276611328e-05,
      "step": 34829,
      "training_step_time": 8.37410569190979
    },
    {
      "epoch": 3.3216476440429686e-05,
      "grad_norm": 0.12896455824375153,
      "learning_rate": 7.758367263985504e-06,
      "loss": 0.0617,
      "step": 34830
    },
    {
      "epoch": 3.3216476440429686e-05,
      "model_forward_time": 1.7266221046447754,
      "step": 34830
    },
    {
      "dynamics_loss_avg": 0.050871443934738636,
      "epoch": 3.3216476440429686e-05,
      "step": 34830
    },
    {
      "action_loss_avg": 0.0022286711260676383,
      "epoch": 3.3216476440429686e-05,
      "step": 34830
    },
    {
      "epoch": 3.3216476440429686e-05,
      "step": 34830,
      "training_step_time": 8.747549533843994
    },
    {
      "epoch": 3.32174301147461e-05,
      "model_forward_time": 2.879674196243286,
      "step": 34831
    },
    {
      "epoch": 3.32174301147461e-05,
      "step": 34831,
      "training_step_time": 9.434768438339233
    },
    {
      "epoch": 3.32183837890625e-05,
      "model_forward_time": 1.732234001159668,
      "step": 34832
    },
    {
      "epoch": 3.32183837890625e-05,
      "step": 34832,
      "training_step_time": 7.853867769241333
    },
    {
      "epoch": 3.3219337463378905e-05,
      "model_forward_time": 1.7241506576538086,
      "step": 34833
    },
    {
      "epoch": 3.3219337463378905e-05,
      "step": 34833,
      "training_step_time": 8.427067756652832
    },
    {
      "epoch": 3.3220291137695316e-05,
      "model_forward_time": 1.7267284393310547,
      "step": 34834
    },
    {
      "epoch": 3.3220291137695316e-05,
      "step": 34834,
      "training_step_time": 7.995990753173828
    },
    {
      "epoch": 3.322124481201172e-05,
      "model_forward_time": 1.7250175476074219,
      "step": 34835
    },
    {
      "epoch": 3.322124481201172e-05,
      "step": 34835,
      "training_step_time": 7.761544942855835
    },
    {
      "epoch": 3.3222198486328124e-05,
      "model_forward_time": 1.7171833515167236,
      "step": 34836
    },
    {
      "epoch": 3.3222198486328124e-05,
      "step": 34836,
      "training_step_time": 7.815985918045044
    },
    {
      "epoch": 3.3223152160644534e-05,
      "model_forward_time": 1.0240986347198486,
      "step": 34837
    },
    {
      "epoch": 3.3223152160644534e-05,
      "step": 34837,
      "training_step_time": 7.735883951187134
    },
    {
      "epoch": 3.322410583496094e-05,
      "model_forward_time": 1.7346551418304443,
      "step": 34838
    },
    {
      "epoch": 3.322410583496094e-05,
      "step": 34838,
      "training_step_time": 8.521520376205444
    },
    {
      "epoch": 3.322505950927734e-05,
      "model_forward_time": 1.7266969680786133,
      "step": 34839
    },
    {
      "epoch": 3.322505950927734e-05,
      "step": 34839,
      "training_step_time": 8.509064674377441
    },
    {
      "epoch": 3.322601318359375e-05,
      "grad_norm": 0.18077899515628815,
      "learning_rate": 7.75698801983397e-06,
      "loss": 0.062,
      "step": 34840
    },
    {
      "epoch": 3.322601318359375e-05,
      "model_forward_time": 2.8330178260803223,
      "step": 34840
    },
    {
      "dynamics_loss_avg": 0.0474690668284893,
      "epoch": 3.322601318359375e-05,
      "step": 34840
    },
    {
      "action_loss_avg": 0.00144967200467363,
      "epoch": 3.322601318359375e-05,
      "step": 34840
    },
    {
      "epoch": 3.322601318359375e-05,
      "step": 34840,
      "training_step_time": 9.134896993637085
    },
    {
      "epoch": 3.322696685791016e-05,
      "model_forward_time": 1.7186362743377686,
      "step": 34841
    },
    {
      "epoch": 3.322696685791016e-05,
      "step": 34841,
      "training_step_time": 8.874285697937012
    },
    {
      "epoch": 3.322792053222656e-05,
      "model_forward_time": 1.7229549884796143,
      "step": 34842
    },
    {
      "epoch": 3.322792053222656e-05,
      "step": 34842,
      "training_step_time": 9.186204671859741
    },
    {
      "epoch": 3.322887420654297e-05,
      "model_forward_time": 1.7269792556762695,
      "step": 34843
    },
    {
      "epoch": 3.322887420654297e-05,
      "step": 34843,
      "training_step_time": 8.591715097427368
    },
    {
      "epoch": 3.3229827880859376e-05,
      "model_forward_time": 1.7174789905548096,
      "step": 34844
    },
    {
      "epoch": 3.3229827880859376e-05,
      "step": 34844,
      "training_step_time": 7.773027420043945
    },
    {
      "epoch": 3.323078155517578e-05,
      "model_forward_time": 1.7211663722991943,
      "step": 34845
    },
    {
      "epoch": 3.323078155517578e-05,
      "step": 34845,
      "training_step_time": 8.383097648620605
    },
    {
      "epoch": 3.3231735229492184e-05,
      "model_forward_time": 1.725492238998413,
      "step": 34846
    },
    {
      "epoch": 3.3231735229492184e-05,
      "step": 34846,
      "training_step_time": 7.80995774269104
    },
    {
      "epoch": 3.3232688903808595e-05,
      "model_forward_time": 1.725102186203003,
      "step": 34847
    },
    {
      "epoch": 3.3232688903808595e-05,
      "step": 34847,
      "training_step_time": 7.788192987442017
    },
    {
      "epoch": 3.3233642578125e-05,
      "model_forward_time": 1.837864637374878,
      "step": 34848
    },
    {
      "epoch": 3.3233642578125e-05,
      "step": 34848,
      "training_step_time": 7.815531969070435
    },
    {
      "epoch": 3.32345962524414e-05,
      "model_forward_time": 1.7196779251098633,
      "step": 34849
    },
    {
      "epoch": 3.32345962524414e-05,
      "step": 34849,
      "training_step_time": 8.346530675888062
    },
    {
      "epoch": 3.3235549926757814e-05,
      "grad_norm": 0.0910135805606842,
      "learning_rate": 7.755608474182372e-06,
      "loss": 0.0588,
      "step": 34850
    },
    {
      "epoch": 3.3235549926757814e-05,
      "model_forward_time": 2.8524668216705322,
      "step": 34850
    },
    {
      "dynamics_loss_avg": 0.0630710480734706,
      "epoch": 3.3235549926757814e-05,
      "step": 34850
    },
    {
      "action_loss_avg": 0.0018075425934512169,
      "epoch": 3.3235549926757814e-05,
      "step": 34850
    },
    {
      "epoch": 3.3235549926757814e-05,
      "step": 34850,
      "training_step_time": 8.62847113609314
    },
    {
      "epoch": 3.323650360107422e-05,
      "model_forward_time": 1.7246334552764893,
      "step": 34851
    },
    {
      "epoch": 3.323650360107422e-05,
      "step": 34851,
      "training_step_time": 9.153488159179688
    },
    {
      "epoch": 3.323745727539062e-05,
      "model_forward_time": 1.7311947345733643,
      "step": 34852
    },
    {
      "epoch": 3.323745727539062e-05,
      "step": 34852,
      "training_step_time": 8.778754711151123
    },
    {
      "epoch": 3.323841094970703e-05,
      "model_forward_time": 1.7268784046173096,
      "step": 34853
    },
    {
      "epoch": 3.323841094970703e-05,
      "step": 34853,
      "training_step_time": 8.379372119903564
    },
    {
      "epoch": 3.323936462402344e-05,
      "model_forward_time": 1.72550630569458,
      "step": 34854
    },
    {
      "epoch": 3.323936462402344e-05,
      "step": 34854,
      "training_step_time": 8.713178873062134
    },
    {
      "epoch": 3.324031829833984e-05,
      "model_forward_time": 1.7264831066131592,
      "step": 34855
    },
    {
      "epoch": 3.324031829833984e-05,
      "step": 34855,
      "training_step_time": 7.806632995605469
    },
    {
      "epoch": 3.324127197265625e-05,
      "model_forward_time": 1.7191848754882812,
      "step": 34856
    },
    {
      "epoch": 3.324127197265625e-05,
      "step": 34856,
      "training_step_time": 7.775967836380005
    },
    {
      "epoch": 3.3242225646972656e-05,
      "model_forward_time": 1.8283214569091797,
      "step": 34857
    },
    {
      "epoch": 3.3242225646972656e-05,
      "step": 34857,
      "training_step_time": 8.841518640518188
    },
    {
      "epoch": 3.324317932128906e-05,
      "model_forward_time": 1.7260074615478516,
      "step": 34858
    },
    {
      "epoch": 3.324317932128906e-05,
      "step": 34858,
      "training_step_time": 7.743616342544556
    },
    {
      "epoch": 3.324413299560547e-05,
      "model_forward_time": 2.883838176727295,
      "step": 34859
    },
    {
      "epoch": 3.324413299560547e-05,
      "step": 34859,
      "training_step_time": 9.321596384048462
    },
    {
      "epoch": 3.3245086669921874e-05,
      "grad_norm": 0.11305321753025055,
      "learning_rate": 7.754228627181574e-06,
      "loss": 0.0601,
      "step": 34860
    },
    {
      "epoch": 3.3245086669921874e-05,
      "model_forward_time": 1.725024700164795,
      "step": 34860
    },
    {
      "dynamics_loss_avg": 0.04433555863797665,
      "epoch": 3.3245086669921874e-05,
      "step": 34860
    },
    {
      "action_loss_avg": 0.0015969833155395463,
      "epoch": 3.3245086669921874e-05,
      "step": 34860
    },
    {
      "epoch": 3.3245086669921874e-05,
      "step": 34860,
      "training_step_time": 7.784804344177246
    },
    {
      "epoch": 3.324604034423828e-05,
      "model_forward_time": 1.3582656383514404,
      "step": 34861
    },
    {
      "epoch": 3.324604034423828e-05,
      "step": 34861,
      "training_step_time": 7.831540107727051
    },
    {
      "epoch": 3.324699401855469e-05,
      "model_forward_time": 1.726194143295288,
      "step": 34862
    },
    {
      "epoch": 3.324699401855469e-05,
      "step": 34862,
      "training_step_time": 7.781313419342041
    },
    {
      "epoch": 3.324794769287109e-05,
      "model_forward_time": 1.726883888244629,
      "step": 34863
    },
    {
      "epoch": 3.324794769287109e-05,
      "step": 34863,
      "training_step_time": 7.689260005950928
    },
    {
      "epoch": 3.32489013671875e-05,
      "model_forward_time": 1.7356226444244385,
      "step": 34864
    },
    {
      "epoch": 3.32489013671875e-05,
      "step": 34864,
      "training_step_time": 7.721778392791748
    },
    {
      "epoch": 3.324985504150391e-05,
      "model_forward_time": 1.7238001823425293,
      "step": 34865
    },
    {
      "epoch": 3.324985504150391e-05,
      "step": 34865,
      "training_step_time": 7.716197967529297
    },
    {
      "epoch": 3.325080871582031e-05,
      "model_forward_time": 1.7949628829956055,
      "step": 34866
    },
    {
      "epoch": 3.325080871582031e-05,
      "step": 34866,
      "training_step_time": 8.189387083053589
    },
    {
      "epoch": 3.3251762390136716e-05,
      "model_forward_time": 1.751720666885376,
      "step": 34867
    },
    {
      "epoch": 3.3251762390136716e-05,
      "step": 34867,
      "training_step_time": 8.294058561325073
    },
    {
      "epoch": 3.325271606445313e-05,
      "model_forward_time": 2.8950557708740234,
      "step": 34868
    },
    {
      "epoch": 3.325271606445313e-05,
      "step": 34868,
      "training_step_time": 7.787129163742065
    },
    {
      "epoch": 3.325366973876953e-05,
      "model_forward_time": 1.7437398433685303,
      "step": 34869
    },
    {
      "epoch": 3.325366973876953e-05,
      "step": 34869,
      "training_step_time": 8.359894514083862
    },
    {
      "epoch": 3.3254623413085935e-05,
      "grad_norm": 0.1557864397764206,
      "learning_rate": 7.752848478982477e-06,
      "loss": 0.0583,
      "step": 34870
    },
    {
      "epoch": 3.3254623413085935e-05,
      "model_forward_time": 1.7367854118347168,
      "step": 34870
    },
    {
      "dynamics_loss_avg": 0.04989896789193153,
      "epoch": 3.3254623413085935e-05,
      "step": 34870
    },
    {
      "action_loss_avg": 0.004471443443617318,
      "epoch": 3.3254623413085935e-05,
      "step": 34870
    },
    {
      "epoch": 3.3254623413085935e-05,
      "step": 34870,
      "training_step_time": 8.791971683502197
    },
    {
      "epoch": 3.3255577087402346e-05,
      "model_forward_time": 1.7338755130767822,
      "step": 34871
    },
    {
      "epoch": 3.3255577087402346e-05,
      "step": 34871,
      "training_step_time": 7.666303396224976
    },
    {
      "epoch": 3.325653076171875e-05,
      "model_forward_time": 1.7474853992462158,
      "step": 34872
    },
    {
      "epoch": 3.325653076171875e-05,
      "step": 34872,
      "training_step_time": 7.67554235458374
    },
    {
      "epoch": 3.3257484436035154e-05,
      "model_forward_time": 1.7363383769989014,
      "step": 34873
    },
    {
      "epoch": 3.3257484436035154e-05,
      "step": 34873,
      "training_step_time": 8.250437259674072
    },
    {
      "epoch": 3.3258438110351565e-05,
      "model_forward_time": 1.7377614974975586,
      "step": 34874
    },
    {
      "epoch": 3.3258438110351565e-05,
      "step": 34874,
      "training_step_time": 7.727088451385498
    },
    {
      "epoch": 3.325939178466797e-05,
      "model_forward_time": 1.37471342086792,
      "step": 34875
    },
    {
      "epoch": 3.325939178466797e-05,
      "step": 34875,
      "training_step_time": 8.706526756286621
    },
    {
      "epoch": 3.326034545898437e-05,
      "model_forward_time": 1.7495837211608887,
      "step": 34876
    },
    {
      "epoch": 3.326034545898437e-05,
      "step": 34876,
      "training_step_time": 7.833791017532349
    },
    {
      "epoch": 3.3261299133300784e-05,
      "model_forward_time": 2.867675542831421,
      "step": 34877
    },
    {
      "epoch": 3.3261299133300784e-05,
      "step": 34877,
      "training_step_time": 10.040231227874756
    },
    {
      "epoch": 3.326225280761719e-05,
      "model_forward_time": 1.7389535903930664,
      "step": 34878
    },
    {
      "epoch": 3.326225280761719e-05,
      "step": 34878,
      "training_step_time": 7.759865760803223
    },
    {
      "epoch": 3.326320648193359e-05,
      "model_forward_time": 1.7441492080688477,
      "step": 34879
    },
    {
      "epoch": 3.326320648193359e-05,
      "step": 34879,
      "training_step_time": 9.848512172698975
    },
    {
      "epoch": 3.326416015625e-05,
      "grad_norm": 0.12574002146720886,
      "learning_rate": 7.751468029736007e-06,
      "loss": 0.0627,
      "step": 34880
    },
    {
      "epoch": 3.326416015625e-05,
      "model_forward_time": 1.741490364074707,
      "step": 34880
    },
    {
      "dynamics_loss_avg": 0.05700438842177391,
      "epoch": 3.326416015625e-05,
      "step": 34880
    },
    {
      "action_loss_avg": 0.002145209570880979,
      "epoch": 3.326416015625e-05,
      "step": 34880
    },
    {
      "epoch": 3.326416015625e-05,
      "step": 34880,
      "training_step_time": 9.474782943725586
    },
    {
      "epoch": 3.3265113830566406e-05,
      "model_forward_time": 1.7296488285064697,
      "step": 34881
    },
    {
      "epoch": 3.3265113830566406e-05,
      "step": 34881,
      "training_step_time": 7.561052322387695
    },
    {
      "epoch": 3.326606750488281e-05,
      "model_forward_time": 1.7417891025543213,
      "step": 34882
    },
    {
      "epoch": 3.326606750488281e-05,
      "step": 34882,
      "training_step_time": 7.593874931335449
    },
    {
      "epoch": 3.326702117919922e-05,
      "model_forward_time": 1.7467985153198242,
      "step": 34883
    },
    {
      "epoch": 3.326702117919922e-05,
      "step": 34883,
      "training_step_time": 7.746402978897095
    },
    {
      "epoch": 3.3267974853515625e-05,
      "model_forward_time": 1.7382347583770752,
      "step": 34884
    },
    {
      "epoch": 3.3267974853515625e-05,
      "step": 34884,
      "training_step_time": 7.571682691574097
    },
    {
      "epoch": 3.326892852783203e-05,
      "model_forward_time": 1.7392628192901611,
      "step": 34885
    },
    {
      "epoch": 3.326892852783203e-05,
      "step": 34885,
      "training_step_time": 8.486760377883911
    },
    {
      "epoch": 3.326988220214844e-05,
      "model_forward_time": 2.8340628147125244,
      "step": 34886
    },
    {
      "epoch": 3.326988220214844e-05,
      "step": 34886,
      "training_step_time": 8.223439693450928
    },
    {
      "epoch": 3.3270835876464844e-05,
      "model_forward_time": 1.7411024570465088,
      "step": 34887
    },
    {
      "epoch": 3.3270835876464844e-05,
      "step": 34887,
      "training_step_time": 7.737936735153198
    },
    {
      "epoch": 3.327178955078125e-05,
      "model_forward_time": 1.7315809726715088,
      "step": 34888
    },
    {
      "epoch": 3.327178955078125e-05,
      "step": 34888,
      "training_step_time": 7.630644798278809
    },
    {
      "epoch": 3.327274322509766e-05,
      "model_forward_time": 1.7483444213867188,
      "step": 34889
    },
    {
      "epoch": 3.327274322509766e-05,
      "step": 34889,
      "training_step_time": 9.357733011245728
    },
    {
      "epoch": 3.327369689941406e-05,
      "grad_norm": 0.15826603770256042,
      "learning_rate": 7.75008727959313e-06,
      "loss": 0.0615,
      "step": 34890
    },
    {
      "epoch": 3.327369689941406e-05,
      "model_forward_time": 1.7298893928527832,
      "step": 34890
    },
    {
      "dynamics_loss_avg": 0.05530408825725317,
      "epoch": 3.327369689941406e-05,
      "step": 34890
    },
    {
      "action_loss_avg": 0.0018411915749311446,
      "epoch": 3.327369689941406e-05,
      "step": 34890
    },
    {
      "epoch": 3.327369689941406e-05,
      "step": 34890,
      "training_step_time": 7.632130861282349
    },
    {
      "epoch": 3.327465057373047e-05,
      "model_forward_time": 1.7290377616882324,
      "step": 34891
    },
    {
      "epoch": 3.327465057373047e-05,
      "step": 34891,
      "training_step_time": 7.6575751304626465
    },
    {
      "epoch": 3.327560424804688e-05,
      "model_forward_time": 1.740436315536499,
      "step": 34892
    },
    {
      "epoch": 3.327560424804688e-05,
      "step": 34892,
      "training_step_time": 7.614317893981934
    },
    {
      "epoch": 3.327655792236328e-05,
      "model_forward_time": 1.7318329811096191,
      "step": 34893
    },
    {
      "epoch": 3.327655792236328e-05,
      "step": 34893,
      "training_step_time": 8.393089771270752
    },
    {
      "epoch": 3.3277511596679686e-05,
      "model_forward_time": 1.8322608470916748,
      "step": 34894
    },
    {
      "epoch": 3.3277511596679686e-05,
      "step": 34894,
      "training_step_time": 8.081047296524048
    },
    {
      "epoch": 3.32784652709961e-05,
      "model_forward_time": 1.7398993968963623,
      "step": 34895
    },
    {
      "epoch": 3.32784652709961e-05,
      "step": 34895,
      "training_step_time": 7.645750522613525
    },
    {
      "epoch": 3.32794189453125e-05,
      "model_forward_time": 2.922044038772583,
      "step": 34896
    },
    {
      "epoch": 3.32794189453125e-05,
      "step": 34896,
      "training_step_time": 9.147119760513306
    },
    {
      "epoch": 3.3280372619628905e-05,
      "model_forward_time": 1.7355763912200928,
      "step": 34897
    },
    {
      "epoch": 3.3280372619628905e-05,
      "step": 34897,
      "training_step_time": 7.542835712432861
    },
    {
      "epoch": 3.3281326293945316e-05,
      "model_forward_time": 1.7409775257110596,
      "step": 34898
    },
    {
      "epoch": 3.3281326293945316e-05,
      "step": 34898,
      "training_step_time": 7.571704149246216
    },
    {
      "epoch": 3.328227996826172e-05,
      "model_forward_time": 1.7367782592773438,
      "step": 34899
    },
    {
      "epoch": 3.328227996826172e-05,
      "step": 34899,
      "training_step_time": 8.579630851745605
    },
    {
      "epoch": 3.3283233642578124e-05,
      "grad_norm": 0.13629311323165894,
      "learning_rate": 7.748706228704843e-06,
      "loss": 0.0608,
      "step": 34900
    },
    {
      "epoch": 3.3283233642578124e-05,
      "model_forward_time": 1.732349157333374,
      "step": 34900
    },
    {
      "dynamics_loss_avg": 0.061364171002060174,
      "epoch": 3.3283233642578124e-05,
      "step": 34900
    },
    {
      "action_loss_avg": 0.00163712048670277,
      "epoch": 3.3283233642578124e-05,
      "step": 34900
    },
    {
      "epoch": 3.3283233642578124e-05,
      "step": 34900,
      "training_step_time": 7.639460563659668
    },
    {
      "epoch": 3.3284187316894534e-05,
      "model_forward_time": 1.734285593032837,
      "step": 34901
    },
    {
      "epoch": 3.3284187316894534e-05,
      "step": 34901,
      "training_step_time": 7.595586061477661
    },
    {
      "epoch": 3.328514099121094e-05,
      "model_forward_time": 1.7442960739135742,
      "step": 34902
    },
    {
      "epoch": 3.328514099121094e-05,
      "step": 34902,
      "training_step_time": 8.1840660572052
    },
    {
      "epoch": 3.328609466552734e-05,
      "model_forward_time": 1.83473539352417,
      "step": 34903
    },
    {
      "epoch": 3.328609466552734e-05,
      "step": 34903,
      "training_step_time": 9.298866987228394
    },
    {
      "epoch": 3.328704833984375e-05,
      "model_forward_time": 1.7287638187408447,
      "step": 34904
    },
    {
      "epoch": 3.328704833984375e-05,
      "step": 34904,
      "training_step_time": 7.6386377811431885
    },
    {
      "epoch": 3.328800201416016e-05,
      "model_forward_time": 2.938600540161133,
      "step": 34905
    },
    {
      "epoch": 3.328800201416016e-05,
      "step": 34905,
      "training_step_time": 8.64605975151062
    },
    {
      "epoch": 3.328895568847656e-05,
      "model_forward_time": 1.7383441925048828,
      "step": 34906
    },
    {
      "epoch": 3.328895568847656e-05,
      "step": 34906,
      "training_step_time": 8.235087633132935
    },
    {
      "epoch": 3.3289909362792965e-05,
      "model_forward_time": 1.7407875061035156,
      "step": 34907
    },
    {
      "epoch": 3.3289909362792965e-05,
      "step": 34907,
      "training_step_time": 7.63164758682251
    },
    {
      "epoch": 3.3290863037109376e-05,
      "model_forward_time": 1.7337279319763184,
      "step": 34908
    },
    {
      "epoch": 3.3290863037109376e-05,
      "step": 34908,
      "training_step_time": 8.678049564361572
    },
    {
      "epoch": 3.329181671142578e-05,
      "model_forward_time": 1.7405619621276855,
      "step": 34909
    },
    {
      "epoch": 3.329181671142578e-05,
      "step": 34909,
      "training_step_time": 7.59861946105957
    },
    {
      "epoch": 3.3292770385742184e-05,
      "grad_norm": 0.15472358465194702,
      "learning_rate": 7.747324877222176e-06,
      "loss": 0.0596,
      "step": 34910
    },
    {
      "epoch": 3.3292770385742184e-05,
      "model_forward_time": 1.7328364849090576,
      "step": 34910
    },
    {
      "dynamics_loss_avg": 0.06132583301514387,
      "epoch": 3.3292770385742184e-05,
      "step": 34910
    },
    {
      "action_loss_avg": 0.0019992382789496333,
      "epoch": 3.3292770385742184e-05,
      "step": 34910
    },
    {
      "epoch": 3.3292770385742184e-05,
      "step": 34910,
      "training_step_time": 7.596973180770874
    },
    {
      "epoch": 3.3293724060058595e-05,
      "model_forward_time": 1.7402169704437256,
      "step": 34911
    },
    {
      "epoch": 3.3293724060058595e-05,
      "step": 34911,
      "training_step_time": 7.352047443389893
    },
    {
      "epoch": 3.3294677734375e-05,
      "model_forward_time": 1.8118102550506592,
      "step": 34912
    },
    {
      "epoch": 3.3294677734375e-05,
      "step": 34912,
      "training_step_time": 8.596359491348267
    },
    {
      "epoch": 3.32956314086914e-05,
      "model_forward_time": 1.7462468147277832,
      "step": 34913
    },
    {
      "epoch": 3.32956314086914e-05,
      "step": 34913,
      "training_step_time": 9.15867567062378
    },
    {
      "epoch": 3.3296585083007814e-05,
      "model_forward_time": 2.891589879989624,
      "step": 34914
    },
    {
      "epoch": 3.3296585083007814e-05,
      "step": 34914,
      "training_step_time": 8.14273738861084
    },
    {
      "epoch": 3.329753875732422e-05,
      "model_forward_time": 1.7406041622161865,
      "step": 34915
    },
    {
      "epoch": 3.329753875732422e-05,
      "step": 34915,
      "training_step_time": 8.26731252670288
    },
    {
      "epoch": 3.329849243164062e-05,
      "model_forward_time": 1.7365846633911133,
      "step": 34916
    },
    {
      "epoch": 3.329849243164062e-05,
      "step": 34916,
      "training_step_time": 8.483192920684814
    },
    {
      "epoch": 3.329944610595703e-05,
      "model_forward_time": 1.739767074584961,
      "step": 34917
    },
    {
      "epoch": 3.329944610595703e-05,
      "step": 34917,
      "training_step_time": 7.667793035507202
    },
    {
      "epoch": 3.330039978027344e-05,
      "model_forward_time": 1.743384838104248,
      "step": 34918
    },
    {
      "epoch": 3.330039978027344e-05,
      "step": 34918,
      "training_step_time": 7.578115463256836
    },
    {
      "epoch": 3.330135345458984e-05,
      "model_forward_time": 1.7265264987945557,
      "step": 34919
    },
    {
      "epoch": 3.330135345458984e-05,
      "step": 34919,
      "training_step_time": 7.560714960098267
    },
    {
      "epoch": 3.330230712890625e-05,
      "grad_norm": 0.16885845363140106,
      "learning_rate": 7.745943225296189e-06,
      "loss": 0.0595,
      "step": 34920
    },
    {
      "epoch": 3.330230712890625e-05,
      "model_forward_time": 1.7351162433624268,
      "step": 34920
    },
    {
      "dynamics_loss_avg": 0.04463481549173594,
      "epoch": 3.330230712890625e-05,
      "step": 34920
    },
    {
      "action_loss_avg": 0.0013489224249497056,
      "epoch": 3.330230712890625e-05,
      "step": 34920
    },
    {
      "epoch": 3.330230712890625e-05,
      "step": 34920,
      "training_step_time": 7.379948616027832
    },
    {
      "epoch": 3.3303260803222656e-05,
      "model_forward_time": 1.7810044288635254,
      "step": 34921
    },
    {
      "epoch": 3.3303260803222656e-05,
      "step": 34921,
      "training_step_time": 7.519690990447998
    },
    {
      "epoch": 3.330421447753906e-05,
      "model_forward_time": 1.7442054748535156,
      "step": 34922
    },
    {
      "epoch": 3.330421447753906e-05,
      "step": 34922,
      "training_step_time": 7.563450574874878
    },
    {
      "epoch": 3.330516815185547e-05,
      "model_forward_time": 2.881131172180176,
      "step": 34923
    },
    {
      "epoch": 3.330516815185547e-05,
      "step": 34923,
      "training_step_time": 8.446491718292236
    },
    {
      "epoch": 3.3306121826171875e-05,
      "model_forward_time": 1.7431824207305908,
      "step": 34924
    },
    {
      "epoch": 3.3306121826171875e-05,
      "step": 34924,
      "training_step_time": 7.635240077972412
    },
    {
      "epoch": 3.330707550048828e-05,
      "model_forward_time": 1.741316556930542,
      "step": 34925
    },
    {
      "epoch": 3.330707550048828e-05,
      "step": 34925,
      "training_step_time": 7.6006386280059814
    },
    {
      "epoch": 3.330802917480469e-05,
      "model_forward_time": 1.742023229598999,
      "step": 34926
    },
    {
      "epoch": 3.330802917480469e-05,
      "step": 34926,
      "training_step_time": 8.530578851699829
    },
    {
      "epoch": 3.330898284912109e-05,
      "model_forward_time": 1.7443716526031494,
      "step": 34927
    },
    {
      "epoch": 3.330898284912109e-05,
      "step": 34927,
      "training_step_time": 9.350282907485962
    },
    {
      "epoch": 3.33099365234375e-05,
      "model_forward_time": 1.7363321781158447,
      "step": 34928
    },
    {
      "epoch": 3.33099365234375e-05,
      "step": 34928,
      "training_step_time": 7.5794677734375
    },
    {
      "epoch": 3.331089019775391e-05,
      "model_forward_time": 1.7414624691009521,
      "step": 34929
    },
    {
      "epoch": 3.331089019775391e-05,
      "step": 34929,
      "training_step_time": 7.422828912734985
    },
    {
      "epoch": 3.331184387207031e-05,
      "grad_norm": 0.12499791383743286,
      "learning_rate": 7.744561273077981e-06,
      "loss": 0.0578,
      "step": 34930
    },
    {
      "epoch": 3.331184387207031e-05,
      "model_forward_time": 1.7618892192840576,
      "step": 34930
    },
    {
      "dynamics_loss_avg": 0.051603032648563384,
      "epoch": 3.331184387207031e-05,
      "step": 34930
    },
    {
      "action_loss_avg": 0.0019387948414077982,
      "epoch": 3.331184387207031e-05,
      "step": 34930
    },
    {
      "epoch": 3.331184387207031e-05,
      "step": 34930,
      "training_step_time": 7.628419876098633
    },
    {
      "epoch": 3.3312797546386716e-05,
      "model_forward_time": 1.7358570098876953,
      "step": 34931
    },
    {
      "epoch": 3.3312797546386716e-05,
      "step": 34931,
      "training_step_time": 8.397158861160278
    },
    {
      "epoch": 3.331375122070313e-05,
      "model_forward_time": 2.875786542892456,
      "step": 34932
    },
    {
      "epoch": 3.331375122070313e-05,
      "step": 34932,
      "training_step_time": 7.610009431838989
    },
    {
      "epoch": 3.331470489501953e-05,
      "model_forward_time": 1.7286179065704346,
      "step": 34933
    },
    {
      "epoch": 3.331470489501953e-05,
      "step": 34933,
      "training_step_time": 9.11854600906372
    },
    {
      "epoch": 3.3315658569335935e-05,
      "model_forward_time": 1.7399439811706543,
      "step": 34934
    },
    {
      "epoch": 3.3315658569335935e-05,
      "step": 34934,
      "training_step_time": 7.604985475540161
    },
    {
      "epoch": 3.3316612243652346e-05,
      "model_forward_time": 1.7435944080352783,
      "step": 34935
    },
    {
      "epoch": 3.3316612243652346e-05,
      "step": 34935,
      "training_step_time": 8.532660007476807
    },
    {
      "epoch": 3.331756591796875e-05,
      "model_forward_time": 1.7367589473724365,
      "step": 34936
    },
    {
      "epoch": 3.331756591796875e-05,
      "step": 34936,
      "training_step_time": 7.697085380554199
    },
    {
      "epoch": 3.3318519592285154e-05,
      "model_forward_time": 1.7399568557739258,
      "step": 34937
    },
    {
      "epoch": 3.3318519592285154e-05,
      "step": 34937,
      "training_step_time": 7.71306848526001
    },
    {
      "epoch": 3.3319473266601565e-05,
      "model_forward_time": 1.7262823581695557,
      "step": 34938
    },
    {
      "epoch": 3.3319473266601565e-05,
      "step": 34938,
      "training_step_time": 7.379935026168823
    },
    {
      "epoch": 3.332042694091797e-05,
      "model_forward_time": 1.7312896251678467,
      "step": 34939
    },
    {
      "epoch": 3.332042694091797e-05,
      "step": 34939,
      "training_step_time": 8.422832012176514
    },
    {
      "epoch": 3.332138061523437e-05,
      "grad_norm": 0.25227949023246765,
      "learning_rate": 7.743179020718678e-06,
      "loss": 0.06,
      "step": 34940
    },
    {
      "epoch": 3.332138061523437e-05,
      "model_forward_time": 1.842841625213623,
      "step": 34940
    },
    {
      "dynamics_loss_avg": 0.044839567691087726,
      "epoch": 3.332138061523437e-05,
      "step": 34940
    },
    {
      "action_loss_avg": 0.0012368159135803581,
      "epoch": 3.332138061523437e-05,
      "step": 34940
    },
    {
      "epoch": 3.332138061523437e-05,
      "step": 34940,
      "training_step_time": 7.561587333679199
    },
    {
      "epoch": 3.3322334289550784e-05,
      "model_forward_time": 1.7325019836425781,
      "step": 34941
    },
    {
      "epoch": 3.3322334289550784e-05,
      "step": 34941,
      "training_step_time": 9.679413557052612
    },
    {
      "epoch": 3.332328796386719e-05,
      "model_forward_time": 1.7410690784454346,
      "step": 34942
    },
    {
      "epoch": 3.332328796386719e-05,
      "step": 34942,
      "training_step_time": 8.870140314102173
    },
    {
      "epoch": 3.332424163818359e-05,
      "model_forward_time": 1.737734079360962,
      "step": 34943
    },
    {
      "epoch": 3.332424163818359e-05,
      "step": 34943,
      "training_step_time": 8.686334133148193
    },
    {
      "epoch": 3.33251953125e-05,
      "model_forward_time": 1.7302169799804688,
      "step": 34944
    },
    {
      "epoch": 3.33251953125e-05,
      "step": 34944,
      "training_step_time": 7.461315870285034
    },
    {
      "epoch": 3.3326148986816407e-05,
      "model_forward_time": 1.3673243522644043,
      "step": 34945
    },
    {
      "epoch": 3.3326148986816407e-05,
      "step": 34945,
      "training_step_time": 8.406792163848877
    },
    {
      "epoch": 3.332710266113281e-05,
      "model_forward_time": 1.7470104694366455,
      "step": 34946
    },
    {
      "epoch": 3.332710266113281e-05,
      "step": 34946,
      "training_step_time": 7.586449861526489
    },
    {
      "epoch": 3.332805633544922e-05,
      "model_forward_time": 1.7203946113586426,
      "step": 34947
    },
    {
      "epoch": 3.332805633544922e-05,
      "step": 34947,
      "training_step_time": 8.315189599990845
    },
    {
      "epoch": 3.3329010009765625e-05,
      "model_forward_time": 1.737833023071289,
      "step": 34948
    },
    {
      "epoch": 3.3329010009765625e-05,
      "step": 34948,
      "training_step_time": 8.128873586654663
    },
    {
      "epoch": 3.332996368408203e-05,
      "model_forward_time": 1.852064609527588,
      "step": 34949
    },
    {
      "epoch": 3.332996368408203e-05,
      "step": 34949,
      "training_step_time": 8.773918867111206
    },
    {
      "epoch": 3.333091735839844e-05,
      "grad_norm": 0.16927555203437805,
      "learning_rate": 7.741796468369443e-06,
      "loss": 0.0594,
      "step": 34950
    },
    {
      "epoch": 3.333091735839844e-05,
      "model_forward_time": 1.7299771308898926,
      "step": 34950
    },
    {
      "dynamics_loss_avg": 0.05343153458088636,
      "epoch": 3.333091735839844e-05,
      "step": 34950
    },
    {
      "action_loss_avg": 0.0017905345594044775,
      "epoch": 3.333091735839844e-05,
      "step": 34950
    },
    {
      "epoch": 3.333091735839844e-05,
      "step": 34950,
      "training_step_time": 7.670513153076172
    },
    {
      "epoch": 3.3331871032714844e-05,
      "model_forward_time": 2.921942949295044,
      "step": 34951
    },
    {
      "epoch": 3.3331871032714844e-05,
      "step": 34951,
      "training_step_time": 8.231474876403809
    },
    {
      "epoch": 3.333282470703125e-05,
      "model_forward_time": 1.7359156608581543,
      "step": 34952
    },
    {
      "epoch": 3.333282470703125e-05,
      "step": 34952,
      "training_step_time": 7.686906337738037
    },
    {
      "epoch": 3.333377838134766e-05,
      "model_forward_time": 1.7846970558166504,
      "step": 34953
    },
    {
      "epoch": 3.333377838134766e-05,
      "step": 34953,
      "training_step_time": 7.762243986129761
    },
    {
      "epoch": 3.333473205566406e-05,
      "model_forward_time": 1.7318897247314453,
      "step": 34954
    },
    {
      "epoch": 3.333473205566406e-05,
      "step": 34954,
      "training_step_time": 7.72131609916687
    },
    {
      "epoch": 3.333568572998047e-05,
      "model_forward_time": 1.7447035312652588,
      "step": 34955
    },
    {
      "epoch": 3.333568572998047e-05,
      "step": 34955,
      "training_step_time": 7.634798049926758
    },
    {
      "epoch": 3.333663940429688e-05,
      "model_forward_time": 1.7327301502227783,
      "step": 34956
    },
    {
      "epoch": 3.333663940429688e-05,
      "step": 34956,
      "training_step_time": 8.154396533966064
    },
    {
      "epoch": 3.333759307861328e-05,
      "model_forward_time": 1.7392878532409668,
      "step": 34957
    },
    {
      "epoch": 3.333759307861328e-05,
      "step": 34957,
      "training_step_time": 8.648598432540894
    },
    {
      "epoch": 3.3338546752929686e-05,
      "model_forward_time": 1.7912209033966064,
      "step": 34958
    },
    {
      "epoch": 3.3338546752929686e-05,
      "step": 34958,
      "training_step_time": 7.5872578620910645
    },
    {
      "epoch": 3.33395004272461e-05,
      "model_forward_time": 1.7287423610687256,
      "step": 34959
    },
    {
      "epoch": 3.33395004272461e-05,
      "step": 34959,
      "training_step_time": 7.610175371170044
    },
    {
      "epoch": 3.33404541015625e-05,
      "grad_norm": 0.12594540417194366,
      "learning_rate": 7.740413616181467e-06,
      "loss": 0.062,
      "step": 34960
    },
    {
      "epoch": 3.33404541015625e-05,
      "model_forward_time": 2.8998281955718994,
      "step": 34960
    },
    {
      "dynamics_loss_avg": 0.06578864566981793,
      "epoch": 3.33404541015625e-05,
      "step": 34960
    },
    {
      "action_loss_avg": 0.001724658056627959,
      "epoch": 3.33404541015625e-05,
      "step": 34960
    },
    {
      "epoch": 3.33404541015625e-05,
      "step": 34960,
      "training_step_time": 7.776763200759888
    },
    {
      "epoch": 3.3341407775878905e-05,
      "model_forward_time": 1.775956392288208,
      "step": 34961
    },
    {
      "epoch": 3.3341407775878905e-05,
      "step": 34961,
      "training_step_time": 7.731132507324219
    },
    {
      "epoch": 3.3342361450195316e-05,
      "model_forward_time": 1.741391658782959,
      "step": 34962
    },
    {
      "epoch": 3.3342361450195316e-05,
      "step": 34962,
      "training_step_time": 8.382882595062256
    },
    {
      "epoch": 3.334331512451172e-05,
      "model_forward_time": 1.746535301208496,
      "step": 34963
    },
    {
      "epoch": 3.334331512451172e-05,
      "step": 34963,
      "training_step_time": 7.863188028335571
    },
    {
      "epoch": 3.3344268798828124e-05,
      "model_forward_time": 1.7398130893707275,
      "step": 34964
    },
    {
      "epoch": 3.3344268798828124e-05,
      "step": 34964,
      "training_step_time": 7.840478897094727
    },
    {
      "epoch": 3.3345222473144535e-05,
      "model_forward_time": 1.73874831199646,
      "step": 34965
    },
    {
      "epoch": 3.3345222473144535e-05,
      "step": 34965,
      "training_step_time": 8.449704647064209
    },
    {
      "epoch": 3.334617614746094e-05,
      "model_forward_time": 1.7533466815948486,
      "step": 34966
    },
    {
      "epoch": 3.334617614746094e-05,
      "step": 34966,
      "training_step_time": 7.768827199935913
    },
    {
      "epoch": 3.334712982177734e-05,
      "model_forward_time": 1.753833532333374,
      "step": 34967
    },
    {
      "epoch": 3.334712982177734e-05,
      "step": 34967,
      "training_step_time": 7.673905611038208
    },
    {
      "epoch": 3.334808349609375e-05,
      "model_forward_time": 1.742847204208374,
      "step": 34968
    },
    {
      "epoch": 3.334808349609375e-05,
      "step": 34968,
      "training_step_time": 9.204942226409912
    },
    {
      "epoch": 3.334903717041016e-05,
      "model_forward_time": 2.9267473220825195,
      "step": 34969
    },
    {
      "epoch": 3.334903717041016e-05,
      "step": 34969,
      "training_step_time": 9.493060827255249
    },
    {
      "epoch": 3.334999084472656e-05,
      "grad_norm": 0.1057468056678772,
      "learning_rate": 7.739030464305977e-06,
      "loss": 0.0627,
      "step": 34970
    },
    {
      "epoch": 3.334999084472656e-05,
      "model_forward_time": 1.7900397777557373,
      "step": 34970
    },
    {
      "dynamics_loss_avg": 0.06192300617694855,
      "epoch": 3.334999084472656e-05,
      "step": 34970
    },
    {
      "action_loss_avg": 0.0019670918118208647,
      "epoch": 3.334999084472656e-05,
      "step": 34970
    },
    {
      "epoch": 3.334999084472656e-05,
      "step": 34970,
      "training_step_time": 8.626431703567505
    },
    {
      "epoch": 3.3350944519042965e-05,
      "model_forward_time": 1.732285976409912,
      "step": 34971
    },
    {
      "epoch": 3.3350944519042965e-05,
      "step": 34971,
      "training_step_time": 7.7075581550598145
    },
    {
      "epoch": 3.3351898193359376e-05,
      "model_forward_time": 1.737048625946045,
      "step": 34972
    },
    {
      "epoch": 3.3351898193359376e-05,
      "step": 34972,
      "training_step_time": 7.670854568481445
    },
    {
      "epoch": 3.335285186767578e-05,
      "model_forward_time": 1.7391479015350342,
      "step": 34973
    },
    {
      "epoch": 3.335285186767578e-05,
      "step": 34973,
      "training_step_time": 7.728933334350586
    },
    {
      "epoch": 3.3353805541992184e-05,
      "model_forward_time": 1.7388317584991455,
      "step": 34974
    },
    {
      "epoch": 3.3353805541992184e-05,
      "step": 34974,
      "training_step_time": 7.693089485168457
    },
    {
      "epoch": 3.3354759216308595e-05,
      "model_forward_time": 1.7491321563720703,
      "step": 34975
    },
    {
      "epoch": 3.3354759216308595e-05,
      "step": 34975,
      "training_step_time": 7.614873170852661
    },
    {
      "epoch": 3.3355712890625e-05,
      "model_forward_time": 1.770833969116211,
      "step": 34976
    },
    {
      "epoch": 3.3355712890625e-05,
      "step": 34976,
      "training_step_time": 7.454277038574219
    },
    {
      "epoch": 3.33566665649414e-05,
      "model_forward_time": 1.8464770317077637,
      "step": 34977
    },
    {
      "epoch": 3.33566665649414e-05,
      "step": 34977,
      "training_step_time": 8.420542478561401
    },
    {
      "epoch": 3.3357620239257814e-05,
      "model_forward_time": 2.871155261993408,
      "step": 34978
    },
    {
      "epoch": 3.3357620239257814e-05,
      "step": 34978,
      "training_step_time": 7.697790861129761
    },
    {
      "epoch": 3.335857391357422e-05,
      "model_forward_time": 1.7408289909362793,
      "step": 34979
    },
    {
      "epoch": 3.335857391357422e-05,
      "step": 34979,
      "training_step_time": 8.149240970611572
    },
    {
      "epoch": 3.335952758789062e-05,
      "grad_norm": 0.17536556720733643,
      "learning_rate": 7.737647012894236e-06,
      "loss": 0.0635,
      "step": 34980
    },
    {
      "epoch": 3.335952758789062e-05,
      "model_forward_time": 1.7342638969421387,
      "step": 34980
    },
    {
      "dynamics_loss_avg": 0.05012425696477294,
      "epoch": 3.335952758789062e-05,
      "step": 34980
    },
    {
      "action_loss_avg": 0.001988595025613904,
      "epoch": 3.335952758789062e-05,
      "step": 34980
    },
    {
      "epoch": 3.335952758789062e-05,
      "step": 34980,
      "training_step_time": 8.332876205444336
    },
    {
      "epoch": 3.336048126220703e-05,
      "model_forward_time": 1.736387014389038,
      "step": 34981
    },
    {
      "epoch": 3.336048126220703e-05,
      "step": 34981,
      "training_step_time": 8.62610650062561
    },
    {
      "epoch": 3.336143493652344e-05,
      "model_forward_time": 1.749701976776123,
      "step": 34982
    },
    {
      "epoch": 3.336143493652344e-05,
      "step": 34982,
      "training_step_time": 8.565144300460815
    },
    {
      "epoch": 3.336238861083984e-05,
      "model_forward_time": 1.7329773902893066,
      "step": 34983
    },
    {
      "epoch": 3.336238861083984e-05,
      "step": 34983,
      "training_step_time": 7.680753469467163
    },
    {
      "epoch": 3.336334228515625e-05,
      "model_forward_time": 1.7322125434875488,
      "step": 34984
    },
    {
      "epoch": 3.336334228515625e-05,
      "step": 34984,
      "training_step_time": 7.5695812702178955
    },
    {
      "epoch": 3.3364295959472656e-05,
      "model_forward_time": 1.7390437126159668,
      "step": 34985
    },
    {
      "epoch": 3.3364295959472656e-05,
      "step": 34985,
      "training_step_time": 8.122238397598267
    },
    {
      "epoch": 3.336524963378906e-05,
      "model_forward_time": 1.8374605178833008,
      "step": 34986
    },
    {
      "epoch": 3.336524963378906e-05,
      "step": 34986,
      "training_step_time": 8.342239379882812
    },
    {
      "epoch": 3.336620330810547e-05,
      "model_forward_time": 2.8283190727233887,
      "step": 34987
    },
    {
      "epoch": 3.336620330810547e-05,
      "step": 34987,
      "training_step_time": 9.342774629592896
    },
    {
      "epoch": 3.3367156982421875e-05,
      "model_forward_time": 1.7579307556152344,
      "step": 34988
    },
    {
      "epoch": 3.3367156982421875e-05,
      "step": 34988,
      "training_step_time": 8.192830324172974
    },
    {
      "epoch": 3.336811065673828e-05,
      "model_forward_time": 1.7351322174072266,
      "step": 34989
    },
    {
      "epoch": 3.336811065673828e-05,
      "step": 34989,
      "training_step_time": 7.630603313446045
    },
    {
      "epoch": 3.336906433105469e-05,
      "grad_norm": 0.16629406809806824,
      "learning_rate": 7.736263262097531e-06,
      "loss": 0.0645,
      "step": 34990
    },
    {
      "epoch": 3.336906433105469e-05,
      "model_forward_time": 1.738326072692871,
      "step": 34990
    },
    {
      "dynamics_loss_avg": 0.05800513830035925,
      "epoch": 3.336906433105469e-05,
      "step": 34990
    },
    {
      "action_loss_avg": 0.0018799022247549147,
      "epoch": 3.336906433105469e-05,
      "step": 34990
    },
    {
      "epoch": 3.336906433105469e-05,
      "step": 34990,
      "training_step_time": 7.544169187545776
    },
    {
      "epoch": 3.3370018005371093e-05,
      "model_forward_time": 1.7232377529144287,
      "step": 34991
    },
    {
      "epoch": 3.3370018005371093e-05,
      "step": 34991,
      "training_step_time": 8.155239820480347
    },
    {
      "epoch": 3.33709716796875e-05,
      "model_forward_time": 1.7159795761108398,
      "step": 34992
    },
    {
      "epoch": 3.33709716796875e-05,
      "step": 34992,
      "training_step_time": 8.353155851364136
    },
    {
      "epoch": 3.337192535400391e-05,
      "model_forward_time": 1.728053092956543,
      "step": 34993
    },
    {
      "epoch": 3.337192535400391e-05,
      "step": 34993,
      "training_step_time": 7.525032043457031
    },
    {
      "epoch": 3.337287902832031e-05,
      "model_forward_time": 1.7270269393920898,
      "step": 34994
    },
    {
      "epoch": 3.337287902832031e-05,
      "step": 34994,
      "training_step_time": 7.5574095249176025
    },
    {
      "epoch": 3.3373832702636716e-05,
      "model_forward_time": 1.759819507598877,
      "step": 34995
    },
    {
      "epoch": 3.3373832702636716e-05,
      "step": 34995,
      "training_step_time": 10.378787517547607
    },
    {
      "epoch": 3.337478637695313e-05,
      "model_forward_time": 1.7325122356414795,
      "step": 34996
    },
    {
      "epoch": 3.337478637695313e-05,
      "step": 34996,
      "training_step_time": 8.620666742324829
    },
    {
      "epoch": 3.337574005126953e-05,
      "model_forward_time": 3.0281972885131836,
      "step": 34997
    },
    {
      "epoch": 3.337574005126953e-05,
      "step": 34997,
      "training_step_time": 7.7774927616119385
    },
    {
      "epoch": 3.3376693725585935e-05,
      "model_forward_time": 1.7299261093139648,
      "step": 34998
    },
    {
      "epoch": 3.3376693725585935e-05,
      "step": 34998,
      "training_step_time": 7.88722038269043
    },
    {
      "epoch": 3.3377647399902346e-05,
      "model_forward_time": 1.7359979152679443,
      "step": 34999
    },
    {
      "epoch": 3.3377647399902346e-05,
      "step": 34999,
      "training_step_time": 8.22864556312561
    },
    {
      "epoch": 3.337860107421875e-05,
      "grad_norm": 0.15849585831165314,
      "learning_rate": 7.734879212067192e-06,
      "loss": 0.0655,
      "step": 35000
    },
    {
      "epoch": 3.337860107421875e-05,
      "model_forward_time": 1.819990634918213,
      "step": 35000
    },
    {
      "dynamics_loss_avg": 0.043893154710531235,
      "epoch": 3.337860107421875e-05,
      "step": 35000
    },
    {
      "action_loss_avg": 0.0012941496912389993,
      "epoch": 3.337860107421875e-05,
      "step": 35000
    },
    {
      "epoch": 3.337860107421875e-05,
      "step": 35000,
      "training_step_time": 6.63091516494751
    },
    {
      "epoch": 3.3379554748535154e-05,
      "model_forward_time": 1.3700435161590576,
      "step": 35001
    },
    {
      "epoch": 3.3379554748535154e-05,
      "step": 35001,
      "training_step_time": 6.830216884613037
    },
    {
      "epoch": 3.3380508422851565e-05,
      "model_forward_time": 1.7555983066558838,
      "step": 35002
    },
    {
      "epoch": 3.3380508422851565e-05,
      "step": 35002,
      "training_step_time": 6.901140213012695
    },
    {
      "epoch": 3.338146209716797e-05,
      "model_forward_time": 1.7341396808624268,
      "step": 35003
    },
    {
      "epoch": 3.338146209716797e-05,
      "step": 35003,
      "training_step_time": 6.828304290771484
    },
    {
      "epoch": 3.338241577148437e-05,
      "model_forward_time": 1.7343885898590088,
      "step": 35004
    },
    {
      "epoch": 3.338241577148437e-05,
      "step": 35004,
      "training_step_time": 6.830296516418457
    },
    {
      "epoch": 3.3383369445800784e-05,
      "model_forward_time": 1.7466633319854736,
      "step": 35005
    },
    {
      "epoch": 3.3383369445800784e-05,
      "step": 35005,
      "training_step_time": 6.843918561935425
    },
    {
      "epoch": 3.338432312011719e-05,
      "model_forward_time": 1.7706489562988281,
      "step": 35006
    },
    {
      "epoch": 3.338432312011719e-05,
      "step": 35006,
      "training_step_time": 6.919743299484253
    },
    {
      "epoch": 3.338527679443359e-05,
      "model_forward_time": 1.7403204441070557,
      "step": 35007
    },
    {
      "epoch": 3.338527679443359e-05,
      "step": 35007,
      "training_step_time": 6.914873838424683
    },
    {
      "epoch": 3.338623046875e-05,
      "model_forward_time": 1.7690484523773193,
      "step": 35008
    },
    {
      "epoch": 3.338623046875e-05,
      "step": 35008,
      "training_step_time": 7.819046974182129
    },
    {
      "epoch": 3.3387184143066407e-05,
      "model_forward_time": 3.383120536804199,
      "step": 35009
    },
    {
      "epoch": 3.3387184143066407e-05,
      "step": 35009,
      "training_step_time": 8.240889549255371
    },
    {
      "epoch": 3.338813781738281e-05,
      "grad_norm": 0.174475759267807,
      "learning_rate": 7.733494862954573e-06,
      "loss": 0.0671,
      "step": 35010
    },
    {
      "epoch": 3.338813781738281e-05,
      "model_forward_time": 1.7289669513702393,
      "step": 35010
    },
    {
      "dynamics_loss_avg": 0.05206108205020428,
      "epoch": 3.338813781738281e-05,
      "step": 35010
    },
    {
      "action_loss_avg": 0.0028395958011969923,
      "epoch": 3.338813781738281e-05,
      "step": 35010
    },
    {
      "epoch": 3.338813781738281e-05,
      "step": 35010,
      "training_step_time": 6.847299098968506
    },
    {
      "epoch": 3.338909149169922e-05,
      "model_forward_time": 1.747112512588501,
      "step": 35011
    },
    {
      "epoch": 3.338909149169922e-05,
      "step": 35011,
      "training_step_time": 6.954625129699707
    },
    {
      "epoch": 3.3390045166015625e-05,
      "model_forward_time": 1.7401492595672607,
      "step": 35012
    },
    {
      "epoch": 3.3390045166015625e-05,
      "step": 35012,
      "training_step_time": 6.994373083114624
    },
    {
      "epoch": 3.339099884033203e-05,
      "model_forward_time": 1.7406210899353027,
      "step": 35013
    },
    {
      "epoch": 3.339099884033203e-05,
      "step": 35013,
      "training_step_time": 6.93826699256897
    },
    {
      "epoch": 3.339195251464844e-05,
      "model_forward_time": 1.7393314838409424,
      "step": 35014
    },
    {
      "epoch": 3.339195251464844e-05,
      "step": 35014,
      "training_step_time": 7.019097566604614
    },
    {
      "epoch": 3.3392906188964844e-05,
      "model_forward_time": 1.7302534580230713,
      "step": 35015
    },
    {
      "epoch": 3.3392906188964844e-05,
      "step": 35015,
      "training_step_time": 7.056126594543457
    },
    {
      "epoch": 3.339385986328125e-05,
      "model_forward_time": 1.7225792407989502,
      "step": 35016
    },
    {
      "epoch": 3.339385986328125e-05,
      "step": 35016,
      "training_step_time": 6.847376346588135
    },
    {
      "epoch": 3.339481353759766e-05,
      "model_forward_time": 1.7646546363830566,
      "step": 35017
    },
    {
      "epoch": 3.339481353759766e-05,
      "step": 35017,
      "training_step_time": 6.935884237289429
    },
    {
      "epoch": 3.339576721191406e-05,
      "model_forward_time": 3.00093412399292,
      "step": 35018
    },
    {
      "epoch": 3.339576721191406e-05,
      "step": 35018,
      "training_step_time": 9.04188084602356
    },
    {
      "epoch": 3.339672088623047e-05,
      "model_forward_time": 1.7402257919311523,
      "step": 35019
    },
    {
      "epoch": 3.339672088623047e-05,
      "step": 35019,
      "training_step_time": 7.496996164321899
    },
    {
      "epoch": 3.339767456054688e-05,
      "grad_norm": 0.11918427050113678,
      "learning_rate": 7.732110214911065e-06,
      "loss": 0.067,
      "step": 35020
    },
    {
      "epoch": 3.339767456054688e-05,
      "model_forward_time": 1.7676522731781006,
      "step": 35020
    },
    {
      "dynamics_loss_avg": 0.042383024282753465,
      "epoch": 3.339767456054688e-05,
      "step": 35020
    },
    {
      "action_loss_avg": 0.0011643458681646734,
      "epoch": 3.339767456054688e-05,
      "step": 35020
    },
    {
      "epoch": 3.339767456054688e-05,
      "step": 35020,
      "training_step_time": 6.889954328536987
    },
    {
      "epoch": 3.339862823486328e-05,
      "model_forward_time": 1.7312085628509521,
      "step": 35021
    },
    {
      "epoch": 3.339862823486328e-05,
      "step": 35021,
      "training_step_time": 6.924731969833374
    },
    {
      "epoch": 3.3399581909179686e-05,
      "model_forward_time": 1.7331457138061523,
      "step": 35022
    },
    {
      "epoch": 3.3399581909179686e-05,
      "step": 35022,
      "training_step_time": 6.906112194061279
    },
    {
      "epoch": 3.34005355834961e-05,
      "model_forward_time": 1.7439918518066406,
      "step": 35023
    },
    {
      "epoch": 3.34005355834961e-05,
      "step": 35023,
      "training_step_time": 6.834021806716919
    },
    {
      "epoch": 3.34014892578125e-05,
      "model_forward_time": 1.741638422012329,
      "step": 35024
    },
    {
      "epoch": 3.34014892578125e-05,
      "step": 35024,
      "training_step_time": 6.747610330581665
    },
    {
      "epoch": 3.3402442932128905e-05,
      "model_forward_time": 1.7592241764068604,
      "step": 35025
    },
    {
      "epoch": 3.3402442932128905e-05,
      "step": 35025,
      "training_step_time": 6.946691036224365
    },
    {
      "epoch": 3.3403396606445316e-05,
      "model_forward_time": 1.7312583923339844,
      "step": 35026
    },
    {
      "epoch": 3.3403396606445316e-05,
      "step": 35026,
      "training_step_time": 6.863020181655884
    },
    {
      "epoch": 3.340435028076172e-05,
      "model_forward_time": 2.877035140991211,
      "step": 35027
    },
    {
      "epoch": 3.340435028076172e-05,
      "step": 35027,
      "training_step_time": 7.865056276321411
    },
    {
      "epoch": 3.3405303955078124e-05,
      "model_forward_time": 1.8439898490905762,
      "step": 35028
    },
    {
      "epoch": 3.3405303955078124e-05,
      "step": 35028,
      "training_step_time": 7.837693214416504
    },
    {
      "epoch": 3.3406257629394535e-05,
      "model_forward_time": 1.7477195262908936,
      "step": 35029
    },
    {
      "epoch": 3.3406257629394535e-05,
      "step": 35029,
      "training_step_time": 6.89145302772522
    },
    {
      "epoch": 3.340721130371094e-05,
      "grad_norm": 0.1578550487756729,
      "learning_rate": 7.730725268088093e-06,
      "loss": 0.0641,
      "step": 35030
    },
    {
      "epoch": 3.340721130371094e-05,
      "model_forward_time": 1.7371013164520264,
      "step": 35030
    },
    {
      "dynamics_loss_avg": 0.06391545347869396,
      "epoch": 3.340721130371094e-05,
      "step": 35030
    },
    {
      "action_loss_avg": 0.002735256089363247,
      "epoch": 3.340721130371094e-05,
      "step": 35030
    },
    {
      "epoch": 3.340721130371094e-05,
      "step": 35030,
      "training_step_time": 6.807448148727417
    },
    {
      "epoch": 3.340816497802734e-05,
      "model_forward_time": 1.7414031028747559,
      "step": 35031
    },
    {
      "epoch": 3.340816497802734e-05,
      "step": 35031,
      "training_step_time": 6.903463840484619
    },
    {
      "epoch": 3.340911865234375e-05,
      "model_forward_time": 1.7504000663757324,
      "step": 35032
    },
    {
      "epoch": 3.340911865234375e-05,
      "step": 35032,
      "training_step_time": 6.847297430038452
    },
    {
      "epoch": 3.341007232666016e-05,
      "model_forward_time": 1.7357697486877441,
      "step": 35033
    },
    {
      "epoch": 3.341007232666016e-05,
      "step": 35033,
      "training_step_time": 6.849792718887329
    },
    {
      "epoch": 3.341102600097656e-05,
      "model_forward_time": 1.7443928718566895,
      "step": 35034
    },
    {
      "epoch": 3.341102600097656e-05,
      "step": 35034,
      "training_step_time": 6.860756874084473
    },
    {
      "epoch": 3.3411979675292965e-05,
      "model_forward_time": 1.7413923740386963,
      "step": 35035
    },
    {
      "epoch": 3.3411979675292965e-05,
      "step": 35035,
      "training_step_time": 6.908693790435791
    },
    {
      "epoch": 3.3412933349609376e-05,
      "model_forward_time": 2.864668369293213,
      "step": 35036
    },
    {
      "epoch": 3.3412933349609376e-05,
      "step": 35036,
      "training_step_time": 9.960293292999268
    },
    {
      "epoch": 3.341388702392578e-05,
      "model_forward_time": 1.8290526866912842,
      "step": 35037
    },
    {
      "epoch": 3.341388702392578e-05,
      "step": 35037,
      "training_step_time": 7.8669233322143555
    },
    {
      "epoch": 3.3414840698242184e-05,
      "model_forward_time": 1.7590851783752441,
      "step": 35038
    },
    {
      "epoch": 3.3414840698242184e-05,
      "step": 35038,
      "training_step_time": 6.850294589996338
    },
    {
      "epoch": 3.3415794372558595e-05,
      "model_forward_time": 1.7333800792694092,
      "step": 35039
    },
    {
      "epoch": 3.3415794372558595e-05,
      "step": 35039,
      "training_step_time": 6.750516891479492
    },
    {
      "epoch": 3.3416748046875e-05,
      "grad_norm": 0.19929544627666473,
      "learning_rate": 7.729340022637111e-06,
      "loss": 0.0631,
      "step": 35040
    },
    {
      "epoch": 3.3416748046875e-05,
      "model_forward_time": 1.0341413021087646,
      "step": 35040
    },
    {
      "dynamics_loss_avg": 0.059724262543022634,
      "epoch": 3.3416748046875e-05,
      "step": 35040
    },
    {
      "action_loss_avg": 0.0013348451524507255,
      "epoch": 3.3416748046875e-05,
      "step": 35040
    },
    {
      "epoch": 3.3416748046875e-05,
      "step": 35040,
      "training_step_time": 6.93886137008667
    },
    {
      "epoch": 3.34177017211914e-05,
      "model_forward_time": 1.7510097026824951,
      "step": 35041
    },
    {
      "epoch": 3.34177017211914e-05,
      "step": 35041,
      "training_step_time": 6.833903074264526
    },
    {
      "epoch": 3.3418655395507814e-05,
      "model_forward_time": 1.7465291023254395,
      "step": 35042
    },
    {
      "epoch": 3.3418655395507814e-05,
      "step": 35042,
      "training_step_time": 6.791046619415283
    },
    {
      "epoch": 3.341960906982422e-05,
      "model_forward_time": 1.7887961864471436,
      "step": 35043
    },
    {
      "epoch": 3.341960906982422e-05,
      "step": 35043,
      "training_step_time": 6.802208423614502
    },
    {
      "epoch": 3.342056274414062e-05,
      "model_forward_time": 1.7386975288391113,
      "step": 35044
    },
    {
      "epoch": 3.342056274414062e-05,
      "step": 35044,
      "training_step_time": 6.758782625198364
    },
    {
      "epoch": 3.342151641845703e-05,
      "model_forward_time": 1.7435462474822998,
      "step": 35045
    },
    {
      "epoch": 3.342151641845703e-05,
      "step": 35045,
      "training_step_time": 9.029121160507202
    },
    {
      "epoch": 3.342247009277344e-05,
      "model_forward_time": 2.9160733222961426,
      "step": 35046
    },
    {
      "epoch": 3.342247009277344e-05,
      "step": 35046,
      "training_step_time": 7.688462972640991
    },
    {
      "epoch": 3.342342376708984e-05,
      "model_forward_time": 1.7414534091949463,
      "step": 35047
    },
    {
      "epoch": 3.342342376708984e-05,
      "step": 35047,
      "training_step_time": 6.962266206741333
    },
    {
      "epoch": 3.342437744140625e-05,
      "model_forward_time": 1.7415821552276611,
      "step": 35048
    },
    {
      "epoch": 3.342437744140625e-05,
      "step": 35048,
      "training_step_time": 6.750213146209717
    },
    {
      "epoch": 3.3425331115722656e-05,
      "model_forward_time": 1.7403957843780518,
      "step": 35049
    },
    {
      "epoch": 3.3425331115722656e-05,
      "step": 35049,
      "training_step_time": 6.762493848800659
    },
    {
      "epoch": 3.342628479003906e-05,
      "grad_norm": 0.15365953743457794,
      "learning_rate": 7.727954478709607e-06,
      "loss": 0.0585,
      "step": 35050
    },
    {
      "epoch": 3.342628479003906e-05,
      "model_forward_time": 1.729797124862671,
      "step": 35050
    },
    {
      "dynamics_loss_avg": 0.06843727566301823,
      "epoch": 3.342628479003906e-05,
      "step": 35050
    },
    {
      "action_loss_avg": 0.002485188754508272,
      "epoch": 3.342628479003906e-05,
      "step": 35050
    },
    {
      "epoch": 3.342628479003906e-05,
      "step": 35050,
      "training_step_time": 6.77095365524292
    },
    {
      "epoch": 3.342723846435547e-05,
      "model_forward_time": 1.7428741455078125,
      "step": 35051
    },
    {
      "epoch": 3.342723846435547e-05,
      "step": 35051,
      "training_step_time": 6.768449306488037
    },
    {
      "epoch": 3.3428192138671875e-05,
      "model_forward_time": 1.7415339946746826,
      "step": 35052
    },
    {
      "epoch": 3.3428192138671875e-05,
      "step": 35052,
      "training_step_time": 6.728571176528931
    },
    {
      "epoch": 3.342914581298828e-05,
      "model_forward_time": 1.740283489227295,
      "step": 35053
    },
    {
      "epoch": 3.342914581298828e-05,
      "step": 35053,
      "training_step_time": 6.721094131469727
    },
    {
      "epoch": 3.343009948730469e-05,
      "model_forward_time": 1.7358767986297607,
      "step": 35054
    },
    {
      "epoch": 3.343009948730469e-05,
      "step": 35054,
      "training_step_time": 8.749112367630005
    },
    {
      "epoch": 3.3431053161621093e-05,
      "model_forward_time": 2.877822160720825,
      "step": 35055
    },
    {
      "epoch": 3.3431053161621093e-05,
      "step": 35055,
      "training_step_time": 7.7151947021484375
    },
    {
      "epoch": 3.34320068359375e-05,
      "model_forward_time": 1.8498005867004395,
      "step": 35056
    },
    {
      "epoch": 3.34320068359375e-05,
      "step": 35056,
      "training_step_time": 7.668684244155884
    },
    {
      "epoch": 3.343296051025391e-05,
      "model_forward_time": 1.7423079013824463,
      "step": 35057
    },
    {
      "epoch": 3.343296051025391e-05,
      "step": 35057,
      "training_step_time": 6.8427112102508545
    },
    {
      "epoch": 3.343391418457031e-05,
      "model_forward_time": 1.774216651916504,
      "step": 35058
    },
    {
      "epoch": 3.343391418457031e-05,
      "step": 35058,
      "training_step_time": 6.841146469116211
    },
    {
      "epoch": 3.3434867858886716e-05,
      "model_forward_time": 1.7278170585632324,
      "step": 35059
    },
    {
      "epoch": 3.3434867858886716e-05,
      "step": 35059,
      "training_step_time": 6.801935911178589
    },
    {
      "epoch": 3.343582153320313e-05,
      "grad_norm": 0.1378128081560135,
      "learning_rate": 7.726568636457103e-06,
      "loss": 0.0588,
      "step": 35060
    },
    {
      "epoch": 3.343582153320313e-05,
      "model_forward_time": 1.725846529006958,
      "step": 35060
    },
    {
      "dynamics_loss_avg": 0.05562175568193197,
      "epoch": 3.343582153320313e-05,
      "step": 35060
    },
    {
      "action_loss_avg": 0.002139129635179415,
      "epoch": 3.343582153320313e-05,
      "step": 35060
    },
    {
      "epoch": 3.343582153320313e-05,
      "step": 35060,
      "training_step_time": 6.826895475387573
    },
    {
      "epoch": 3.343677520751953e-05,
      "model_forward_time": 1.7484536170959473,
      "step": 35061
    },
    {
      "epoch": 3.343677520751953e-05,
      "step": 35061,
      "training_step_time": 6.877406358718872
    },
    {
      "epoch": 3.3437728881835935e-05,
      "model_forward_time": 1.7442269325256348,
      "step": 35062
    },
    {
      "epoch": 3.3437728881835935e-05,
      "step": 35062,
      "training_step_time": 6.837106466293335
    },
    {
      "epoch": 3.3438682556152346e-05,
      "model_forward_time": 1.7292604446411133,
      "step": 35063
    },
    {
      "epoch": 3.3438682556152346e-05,
      "step": 35063,
      "training_step_time": 8.466431617736816
    },
    {
      "epoch": 3.343963623046875e-05,
      "model_forward_time": 2.91974139213562,
      "step": 35064
    },
    {
      "epoch": 3.343963623046875e-05,
      "step": 35064,
      "training_step_time": 9.372306823730469
    },
    {
      "epoch": 3.3440589904785154e-05,
      "model_forward_time": 1.817903757095337,
      "step": 35065
    },
    {
      "epoch": 3.3440589904785154e-05,
      "step": 35065,
      "training_step_time": 7.64504075050354
    },
    {
      "epoch": 3.3441543579101565e-05,
      "model_forward_time": 1.735992431640625,
      "step": 35066
    },
    {
      "epoch": 3.3441543579101565e-05,
      "step": 35066,
      "training_step_time": 6.807492971420288
    },
    {
      "epoch": 3.344249725341797e-05,
      "model_forward_time": 1.733466386795044,
      "step": 35067
    },
    {
      "epoch": 3.344249725341797e-05,
      "step": 35067,
      "training_step_time": 6.774921178817749
    },
    {
      "epoch": 3.344345092773437e-05,
      "model_forward_time": 1.7726776599884033,
      "step": 35068
    },
    {
      "epoch": 3.344345092773437e-05,
      "step": 35068,
      "training_step_time": 6.774588108062744
    },
    {
      "epoch": 3.3444404602050784e-05,
      "model_forward_time": 1.7469444274902344,
      "step": 35069
    },
    {
      "epoch": 3.3444404602050784e-05,
      "step": 35069,
      "training_step_time": 6.820013046264648
    },
    {
      "epoch": 3.344535827636719e-05,
      "grad_norm": 0.09913915395736694,
      "learning_rate": 7.725182496031153e-06,
      "loss": 0.058,
      "step": 35070
    },
    {
      "epoch": 3.344535827636719e-05,
      "model_forward_time": 1.729137659072876,
      "step": 35070
    },
    {
      "dynamics_loss_avg": 0.06077446807175875,
      "epoch": 3.344535827636719e-05,
      "step": 35070
    },
    {
      "action_loss_avg": 0.001602692442247644,
      "epoch": 3.344535827636719e-05,
      "step": 35070
    },
    {
      "epoch": 3.344535827636719e-05,
      "step": 35070,
      "training_step_time": 6.772096633911133
    },
    {
      "epoch": 3.344631195068359e-05,
      "model_forward_time": 1.7255630493164062,
      "step": 35071
    },
    {
      "epoch": 3.344631195068359e-05,
      "step": 35071,
      "training_step_time": 6.7763612270355225
    },
    {
      "epoch": 3.3447265625e-05,
      "model_forward_time": 1.744201898574829,
      "step": 35072
    },
    {
      "epoch": 3.3447265625e-05,
      "step": 35072,
      "training_step_time": 7.6750242710113525
    },
    {
      "epoch": 3.344821929931641e-05,
      "model_forward_time": 2.942425012588501,
      "step": 35073
    },
    {
      "epoch": 3.344821929931641e-05,
      "step": 35073,
      "training_step_time": 9.60822606086731
    },
    {
      "epoch": 3.344917297363281e-05,
      "model_forward_time": 1.7589564323425293,
      "step": 35074
    },
    {
      "epoch": 3.344917297363281e-05,
      "step": 35074,
      "training_step_time": 7.689086198806763
    },
    {
      "epoch": 3.345012664794922e-05,
      "model_forward_time": 1.74703049659729,
      "step": 35075
    },
    {
      "epoch": 3.345012664794922e-05,
      "step": 35075,
      "training_step_time": 6.71734356880188
    },
    {
      "epoch": 3.3451080322265625e-05,
      "model_forward_time": 1.7442536354064941,
      "step": 35076
    },
    {
      "epoch": 3.3451080322265625e-05,
      "step": 35076,
      "training_step_time": 6.759716749191284
    },
    {
      "epoch": 3.345203399658203e-05,
      "model_forward_time": 1.7412610054016113,
      "step": 35077
    },
    {
      "epoch": 3.345203399658203e-05,
      "step": 35077,
      "training_step_time": 6.762479543685913
    },
    {
      "epoch": 3.345298767089844e-05,
      "model_forward_time": 1.7369985580444336,
      "step": 35078
    },
    {
      "epoch": 3.345298767089844e-05,
      "step": 35078,
      "training_step_time": 6.752233982086182
    },
    {
      "epoch": 3.3453941345214844e-05,
      "model_forward_time": 1.7339258193969727,
      "step": 35079
    },
    {
      "epoch": 3.3453941345214844e-05,
      "step": 35079,
      "training_step_time": 6.754031419754028
    },
    {
      "epoch": 3.345489501953125e-05,
      "grad_norm": 0.11912094801664352,
      "learning_rate": 7.723796057583342e-06,
      "loss": 0.0587,
      "step": 35080
    },
    {
      "epoch": 3.345489501953125e-05,
      "model_forward_time": 1.7358646392822266,
      "step": 35080
    },
    {
      "dynamics_loss_avg": 0.055421816743910314,
      "epoch": 3.345489501953125e-05,
      "step": 35080
    },
    {
      "action_loss_avg": 0.0024217696278356015,
      "epoch": 3.345489501953125e-05,
      "step": 35080
    },
    {
      "epoch": 3.345489501953125e-05,
      "step": 35080,
      "training_step_time": 6.73206353187561
    },
    {
      "epoch": 3.345584869384766e-05,
      "model_forward_time": 1.74407958984375,
      "step": 35081
    },
    {
      "epoch": 3.345584869384766e-05,
      "step": 35081,
      "training_step_time": 8.659198760986328
    },
    {
      "epoch": 3.345680236816406e-05,
      "model_forward_time": 2.908364772796631,
      "step": 35082
    },
    {
      "epoch": 3.345680236816406e-05,
      "step": 35082,
      "training_step_time": 9.98334813117981
    },
    {
      "epoch": 3.345775604248047e-05,
      "model_forward_time": 1.7321195602416992,
      "step": 35083
    },
    {
      "epoch": 3.345775604248047e-05,
      "step": 35083,
      "training_step_time": 8.379821062088013
    },
    {
      "epoch": 3.345870971679688e-05,
      "model_forward_time": 1.733942985534668,
      "step": 35084
    },
    {
      "epoch": 3.345870971679688e-05,
      "step": 35084,
      "training_step_time": 7.543661832809448
    },
    {
      "epoch": 3.345966339111328e-05,
      "model_forward_time": 1.0478086471557617,
      "step": 35085
    },
    {
      "epoch": 3.345966339111328e-05,
      "step": 35085,
      "training_step_time": 6.700608730316162
    },
    {
      "epoch": 3.3460617065429686e-05,
      "model_forward_time": 1.72639799118042,
      "step": 35086
    },
    {
      "epoch": 3.3460617065429686e-05,
      "step": 35086,
      "training_step_time": 6.719662666320801
    },
    {
      "epoch": 3.34615707397461e-05,
      "model_forward_time": 1.7362449169158936,
      "step": 35087
    },
    {
      "epoch": 3.34615707397461e-05,
      "step": 35087,
      "training_step_time": 6.738029479980469
    },
    {
      "epoch": 3.34625244140625e-05,
      "model_forward_time": 1.758246660232544,
      "step": 35088
    },
    {
      "epoch": 3.34625244140625e-05,
      "step": 35088,
      "training_step_time": 6.729786157608032
    },
    {
      "epoch": 3.3463478088378905e-05,
      "model_forward_time": 1.7367405891418457,
      "step": 35089
    },
    {
      "epoch": 3.3463478088378905e-05,
      "step": 35089,
      "training_step_time": 6.795844793319702
    },
    {
      "epoch": 3.3464431762695316e-05,
      "grad_norm": 0.1542402058839798,
      "learning_rate": 7.722409321265288e-06,
      "loss": 0.059,
      "step": 35090
    },
    {
      "epoch": 3.3464431762695316e-05,
      "model_forward_time": 1.7358601093292236,
      "step": 35090
    },
    {
      "dynamics_loss_avg": 0.07252789866179228,
      "epoch": 3.3464431762695316e-05,
      "step": 35090
    },
    {
      "action_loss_avg": 0.002559784281766042,
      "epoch": 3.3464431762695316e-05,
      "step": 35090
    },
    {
      "epoch": 3.3464431762695316e-05,
      "step": 35090,
      "training_step_time": 7.6355602741241455
    },
    {
      "epoch": 3.346538543701172e-05,
      "model_forward_time": 1.7357864379882812,
      "step": 35091
    },
    {
      "epoch": 3.346538543701172e-05,
      "step": 35091,
      "training_step_time": 11.325947284698486
    },
    {
      "epoch": 3.3466339111328124e-05,
      "model_forward_time": 1.7423219680786133,
      "step": 35092
    },
    {
      "epoch": 3.3466339111328124e-05,
      "step": 35092,
      "training_step_time": 7.63343358039856
    },
    {
      "epoch": 3.3467292785644535e-05,
      "model_forward_time": 1.8462703227996826,
      "step": 35093
    },
    {
      "epoch": 3.3467292785644535e-05,
      "step": 35093,
      "training_step_time": 7.586164712905884
    },
    {
      "epoch": 3.346824645996094e-05,
      "model_forward_time": 1.7340896129608154,
      "step": 35094
    },
    {
      "epoch": 3.346824645996094e-05,
      "step": 35094,
      "training_step_time": 6.687134265899658
    },
    {
      "epoch": 3.346920013427734e-05,
      "model_forward_time": 1.7308270931243896,
      "step": 35095
    },
    {
      "epoch": 3.346920013427734e-05,
      "step": 35095,
      "training_step_time": 6.731473207473755
    },
    {
      "epoch": 3.347015380859375e-05,
      "model_forward_time": 1.736618995666504,
      "step": 35096
    },
    {
      "epoch": 3.347015380859375e-05,
      "step": 35096,
      "training_step_time": 6.743653297424316
    },
    {
      "epoch": 3.347110748291016e-05,
      "model_forward_time": 1.7577567100524902,
      "step": 35097
    },
    {
      "epoch": 3.347110748291016e-05,
      "step": 35097,
      "training_step_time": 6.674058198928833
    },
    {
      "epoch": 3.347206115722656e-05,
      "model_forward_time": 1.737379550933838,
      "step": 35098
    },
    {
      "epoch": 3.347206115722656e-05,
      "step": 35098,
      "training_step_time": 6.700995206832886
    },
    {
      "epoch": 3.3473014831542966e-05,
      "model_forward_time": 1.7335560321807861,
      "step": 35099
    },
    {
      "epoch": 3.3473014831542966e-05,
      "step": 35099,
      "training_step_time": 7.524508476257324
    },
    {
      "epoch": 3.3473968505859376e-05,
      "grad_norm": 0.14578501880168915,
      "learning_rate": 7.721022287228645e-06,
      "loss": 0.0585,
      "step": 35100
    },
    {
      "epoch": 3.3473968505859376e-05,
      "model_forward_time": 1.7329449653625488,
      "step": 35100
    },
    {
      "dynamics_loss_avg": 0.05074133165180683,
      "epoch": 3.3473968505859376e-05,
      "step": 35100
    },
    {
      "action_loss_avg": 0.0015309734269976615,
      "epoch": 3.3473968505859376e-05,
      "step": 35100
    },
    {
      "epoch": 3.3473968505859376e-05,
      "step": 35100,
      "training_step_time": 9.58860445022583
    },
    {
      "epoch": 3.347492218017578e-05,
      "model_forward_time": 2.926151752471924,
      "step": 35101
    },
    {
      "epoch": 3.347492218017578e-05,
      "step": 35101,
      "training_step_time": 8.391359806060791
    },
    {
      "epoch": 3.3475875854492184e-05,
      "model_forward_time": 1.8250901699066162,
      "step": 35102
    },
    {
      "epoch": 3.3475875854492184e-05,
      "step": 35102,
      "training_step_time": 7.577438592910767
    },
    {
      "epoch": 3.3476829528808595e-05,
      "model_forward_time": 1.73868727684021,
      "step": 35103
    },
    {
      "epoch": 3.3476829528808595e-05,
      "step": 35103,
      "training_step_time": 6.679072856903076
    },
    {
      "epoch": 3.3477783203125e-05,
      "model_forward_time": 1.7325091361999512,
      "step": 35104
    },
    {
      "epoch": 3.3477783203125e-05,
      "step": 35104,
      "training_step_time": 6.672569036483765
    },
    {
      "epoch": 3.34787368774414e-05,
      "model_forward_time": 1.735971212387085,
      "step": 35105
    },
    {
      "epoch": 3.34787368774414e-05,
      "step": 35105,
      "training_step_time": 6.692364454269409
    },
    {
      "epoch": 3.3479690551757814e-05,
      "model_forward_time": 1.7319707870483398,
      "step": 35106
    },
    {
      "epoch": 3.3479690551757814e-05,
      "step": 35106,
      "training_step_time": 6.708562850952148
    },
    {
      "epoch": 3.348064422607422e-05,
      "model_forward_time": 1.7295503616333008,
      "step": 35107
    },
    {
      "epoch": 3.348064422607422e-05,
      "step": 35107,
      "training_step_time": 6.712553024291992
    },
    {
      "epoch": 3.348159790039062e-05,
      "model_forward_time": 1.7305052280426025,
      "step": 35108
    },
    {
      "epoch": 3.348159790039062e-05,
      "step": 35108,
      "training_step_time": 7.35898756980896
    },
    {
      "epoch": 3.348255157470703e-05,
      "model_forward_time": 1.7767884731292725,
      "step": 35109
    },
    {
      "epoch": 3.348255157470703e-05,
      "step": 35109,
      "training_step_time": 9.270435810089111
    },
    {
      "epoch": 3.348350524902344e-05,
      "grad_norm": 0.17419102787971497,
      "learning_rate": 7.719634955625095e-06,
      "loss": 0.0602,
      "step": 35110
    },
    {
      "epoch": 3.348350524902344e-05,
      "model_forward_time": 2.9251482486724854,
      "step": 35110
    },
    {
      "dynamics_loss_avg": 0.05655600633472204,
      "epoch": 3.348350524902344e-05,
      "step": 35110
    },
    {
      "action_loss_avg": 0.002366950240684673,
      "epoch": 3.348350524902344e-05,
      "step": 35110
    },
    {
      "epoch": 3.348350524902344e-05,
      "step": 35110,
      "training_step_time": 7.64845609664917
    },
    {
      "epoch": 3.348445892333984e-05,
      "model_forward_time": 1.7753698825836182,
      "step": 35111
    },
    {
      "epoch": 3.348445892333984e-05,
      "step": 35111,
      "training_step_time": 7.743170499801636
    },
    {
      "epoch": 3.348541259765625e-05,
      "model_forward_time": 1.3807222843170166,
      "step": 35112
    },
    {
      "epoch": 3.348541259765625e-05,
      "step": 35112,
      "training_step_time": 6.652030944824219
    },
    {
      "epoch": 3.3486366271972656e-05,
      "model_forward_time": 1.7727458477020264,
      "step": 35113
    },
    {
      "epoch": 3.3486366271972656e-05,
      "step": 35113,
      "training_step_time": 6.681645631790161
    },
    {
      "epoch": 3.348731994628906e-05,
      "model_forward_time": 1.7359976768493652,
      "step": 35114
    },
    {
      "epoch": 3.348731994628906e-05,
      "step": 35114,
      "training_step_time": 6.7122509479522705
    },
    {
      "epoch": 3.348827362060547e-05,
      "model_forward_time": 1.732743740081787,
      "step": 35115
    },
    {
      "epoch": 3.348827362060547e-05,
      "step": 35115,
      "training_step_time": 6.663577079772949
    },
    {
      "epoch": 3.3489227294921875e-05,
      "model_forward_time": 1.7332396507263184,
      "step": 35116
    },
    {
      "epoch": 3.3489227294921875e-05,
      "step": 35116,
      "training_step_time": 6.69388222694397
    },
    {
      "epoch": 3.349018096923828e-05,
      "model_forward_time": 1.7444114685058594,
      "step": 35117
    },
    {
      "epoch": 3.349018096923828e-05,
      "step": 35117,
      "training_step_time": 7.571733474731445
    },
    {
      "epoch": 3.349113464355469e-05,
      "model_forward_time": 1.7354662418365479,
      "step": 35118
    },
    {
      "epoch": 3.349113464355469e-05,
      "step": 35118,
      "training_step_time": 8.88221287727356
    },
    {
      "epoch": 3.3492088317871094e-05,
      "model_forward_time": 2.8819503784179688,
      "step": 35119
    },
    {
      "epoch": 3.3492088317871094e-05,
      "step": 35119,
      "training_step_time": 9.597470045089722
    },
    {
      "epoch": 3.34930419921875e-05,
      "grad_norm": 0.15712276101112366,
      "learning_rate": 7.718247326606357e-06,
      "loss": 0.0603,
      "step": 35120
    },
    {
      "epoch": 3.34930419921875e-05,
      "model_forward_time": 1.734609603881836,
      "step": 35120
    },
    {
      "dynamics_loss_avg": 0.051257360354065896,
      "epoch": 3.34930419921875e-05,
      "step": 35120
    },
    {
      "action_loss_avg": 0.0013410452957032248,
      "epoch": 3.34930419921875e-05,
      "step": 35120
    },
    {
      "epoch": 3.34930419921875e-05,
      "step": 35120,
      "training_step_time": 9.567619562149048
    },
    {
      "epoch": 3.349399566650391e-05,
      "model_forward_time": 1.847747802734375,
      "step": 35121
    },
    {
      "epoch": 3.349399566650391e-05,
      "step": 35121,
      "training_step_time": 7.404552221298218
    },
    {
      "epoch": 3.349494934082031e-05,
      "model_forward_time": 1.7281079292297363,
      "step": 35122
    },
    {
      "epoch": 3.349494934082031e-05,
      "step": 35122,
      "training_step_time": 6.708373546600342
    },
    {
      "epoch": 3.3495903015136716e-05,
      "model_forward_time": 1.7377514839172363,
      "step": 35123
    },
    {
      "epoch": 3.3495903015136716e-05,
      "step": 35123,
      "training_step_time": 6.71307110786438
    },
    {
      "epoch": 3.349685668945313e-05,
      "model_forward_time": 1.7437117099761963,
      "step": 35124
    },
    {
      "epoch": 3.349685668945313e-05,
      "step": 35124,
      "training_step_time": 6.715328693389893
    },
    {
      "epoch": 3.349781036376953e-05,
      "model_forward_time": 1.7445361614227295,
      "step": 35125
    },
    {
      "epoch": 3.349781036376953e-05,
      "step": 35125,
      "training_step_time": 6.686507940292358
    },
    {
      "epoch": 3.3498764038085935e-05,
      "model_forward_time": 1.7498221397399902,
      "step": 35126
    },
    {
      "epoch": 3.3498764038085935e-05,
      "step": 35126,
      "training_step_time": 7.664153575897217
    },
    {
      "epoch": 3.3499717712402346e-05,
      "model_forward_time": 1.7349164485931396,
      "step": 35127
    },
    {
      "epoch": 3.3499717712402346e-05,
      "step": 35127,
      "training_step_time": 9.295258283615112
    },
    {
      "epoch": 3.350067138671875e-05,
      "model_forward_time": 2.8734991550445557,
      "step": 35128
    },
    {
      "epoch": 3.350067138671875e-05,
      "step": 35128,
      "training_step_time": 8.197883605957031
    },
    {
      "epoch": 3.3501625061035154e-05,
      "model_forward_time": 1.7378652095794678,
      "step": 35129
    },
    {
      "epoch": 3.3501625061035154e-05,
      "step": 35129,
      "training_step_time": 7.640462875366211
    },
    {
      "epoch": 3.3502578735351565e-05,
      "grad_norm": 0.14632359147071838,
      "learning_rate": 7.716859400324176e-06,
      "loss": 0.0567,
      "step": 35130
    },
    {
      "epoch": 3.3502578735351565e-05,
      "model_forward_time": 1.8461880683898926,
      "step": 35130
    },
    {
      "dynamics_loss_avg": 0.057480336353182795,
      "epoch": 3.3502578735351565e-05,
      "step": 35130
    },
    {
      "action_loss_avg": 0.0020274971378967167,
      "epoch": 3.3502578735351565e-05,
      "step": 35130
    },
    {
      "epoch": 3.3502578735351565e-05,
      "step": 35130,
      "training_step_time": 7.621400833129883
    },
    {
      "epoch": 3.350353240966797e-05,
      "model_forward_time": 1.3883912563323975,
      "step": 35131
    },
    {
      "epoch": 3.350353240966797e-05,
      "step": 35131,
      "training_step_time": 6.649198532104492
    },
    {
      "epoch": 3.350448608398437e-05,
      "model_forward_time": 1.7333076000213623,
      "step": 35132
    },
    {
      "epoch": 3.350448608398437e-05,
      "step": 35132,
      "training_step_time": 6.6684911251068115
    },
    {
      "epoch": 3.3505439758300784e-05,
      "model_forward_time": 1.7380163669586182,
      "step": 35133
    },
    {
      "epoch": 3.3505439758300784e-05,
      "step": 35133,
      "training_step_time": 6.645514488220215
    },
    {
      "epoch": 3.350639343261719e-05,
      "model_forward_time": 1.7363135814666748,
      "step": 35134
    },
    {
      "epoch": 3.350639343261719e-05,
      "step": 35134,
      "training_step_time": 6.69295597076416
    },
    {
      "epoch": 3.350734710693359e-05,
      "model_forward_time": 1.7369182109832764,
      "step": 35135
    },
    {
      "epoch": 3.350734710693359e-05,
      "step": 35135,
      "training_step_time": 7.45267915725708
    },
    {
      "epoch": 3.350830078125e-05,
      "model_forward_time": 1.749253749847412,
      "step": 35136
    },
    {
      "epoch": 3.350830078125e-05,
      "step": 35136,
      "training_step_time": 8.466257333755493
    },
    {
      "epoch": 3.350925445556641e-05,
      "model_forward_time": 2.8256678581237793,
      "step": 35137
    },
    {
      "epoch": 3.350925445556641e-05,
      "step": 35137,
      "training_step_time": 7.793359994888306
    },
    {
      "epoch": 3.351020812988281e-05,
      "model_forward_time": 1.7322607040405273,
      "step": 35138
    },
    {
      "epoch": 3.351020812988281e-05,
      "step": 35138,
      "training_step_time": 7.75484037399292
    },
    {
      "epoch": 3.351116180419922e-05,
      "model_forward_time": 1.8527498245239258,
      "step": 35139
    },
    {
      "epoch": 3.351116180419922e-05,
      "step": 35139,
      "training_step_time": 7.6743104457855225
    },
    {
      "epoch": 3.3512115478515626e-05,
      "grad_norm": 0.13371361792087555,
      "learning_rate": 7.715471176930336e-06,
      "loss": 0.0578,
      "step": 35140
    },
    {
      "epoch": 3.3512115478515626e-05,
      "model_forward_time": 1.7362644672393799,
      "step": 35140
    },
    {
      "dynamics_loss_avg": 0.03763773366808891,
      "epoch": 3.3512115478515626e-05,
      "step": 35140
    },
    {
      "action_loss_avg": 0.0016247175459284335,
      "epoch": 3.3512115478515626e-05,
      "step": 35140
    },
    {
      "epoch": 3.3512115478515626e-05,
      "step": 35140,
      "training_step_time": 6.804789304733276
    },
    {
      "epoch": 3.351306915283203e-05,
      "model_forward_time": 1.7364635467529297,
      "step": 35141
    },
    {
      "epoch": 3.351306915283203e-05,
      "step": 35141,
      "training_step_time": 6.791518211364746
    },
    {
      "epoch": 3.351402282714844e-05,
      "model_forward_time": 1.733900785446167,
      "step": 35142
    },
    {
      "epoch": 3.351402282714844e-05,
      "step": 35142,
      "training_step_time": 6.845515966415405
    },
    {
      "epoch": 3.3514976501464844e-05,
      "model_forward_time": 1.746056079864502,
      "step": 35143
    },
    {
      "epoch": 3.3514976501464844e-05,
      "step": 35143,
      "training_step_time": 6.847689628601074
    },
    {
      "epoch": 3.351593017578125e-05,
      "model_forward_time": 1.7363014221191406,
      "step": 35144
    },
    {
      "epoch": 3.351593017578125e-05,
      "step": 35144,
      "training_step_time": 7.456041097640991
    },
    {
      "epoch": 3.351688385009766e-05,
      "model_forward_time": 1.730849027633667,
      "step": 35145
    },
    {
      "epoch": 3.351688385009766e-05,
      "step": 35145,
      "training_step_time": 7.611836910247803
    },
    {
      "epoch": 3.351783752441406e-05,
      "model_forward_time": 1.7239694595336914,
      "step": 35146
    },
    {
      "epoch": 3.351783752441406e-05,
      "step": 35146,
      "training_step_time": 8.600401163101196
    },
    {
      "epoch": 3.351879119873047e-05,
      "model_forward_time": 2.897998332977295,
      "step": 35147
    },
    {
      "epoch": 3.351879119873047e-05,
      "step": 35147,
      "training_step_time": 9.453385353088379
    },
    {
      "epoch": 3.351974487304688e-05,
      "model_forward_time": 1.8421766757965088,
      "step": 35148
    },
    {
      "epoch": 3.351974487304688e-05,
      "step": 35148,
      "training_step_time": 7.712171792984009
    },
    {
      "epoch": 3.352069854736328e-05,
      "model_forward_time": 1.7408208847045898,
      "step": 35149
    },
    {
      "epoch": 3.352069854736328e-05,
      "step": 35149,
      "training_step_time": 6.777614593505859
    },
    {
      "epoch": 3.3521652221679686e-05,
      "grad_norm": 0.10471449047327042,
      "learning_rate": 7.714082656576651e-06,
      "loss": 0.0574,
      "step": 35150
    },
    {
      "epoch": 3.3521652221679686e-05,
      "model_forward_time": 1.7342498302459717,
      "step": 35150
    },
    {
      "dynamics_loss_avg": 0.05493588708341122,
      "epoch": 3.3521652221679686e-05,
      "step": 35150
    },
    {
      "action_loss_avg": 0.0019428561092354358,
      "epoch": 3.3521652221679686e-05,
      "step": 35150
    },
    {
      "epoch": 3.3521652221679686e-05,
      "step": 35150,
      "training_step_time": 6.7826807498931885
    },
    {
      "epoch": 3.35226058959961e-05,
      "model_forward_time": 1.7308158874511719,
      "step": 35151
    },
    {
      "epoch": 3.35226058959961e-05,
      "step": 35151,
      "training_step_time": 6.778416633605957
    },
    {
      "epoch": 3.35235595703125e-05,
      "model_forward_time": 1.7299230098724365,
      "step": 35152
    },
    {
      "epoch": 3.35235595703125e-05,
      "step": 35152,
      "training_step_time": 6.793384552001953
    },
    {
      "epoch": 3.3524513244628905e-05,
      "model_forward_time": 1.7280566692352295,
      "step": 35153
    },
    {
      "epoch": 3.3524513244628905e-05,
      "step": 35153,
      "training_step_time": 7.267020225524902
    },
    {
      "epoch": 3.3525466918945316e-05,
      "model_forward_time": 1.7394342422485352,
      "step": 35154
    },
    {
      "epoch": 3.3525466918945316e-05,
      "step": 35154,
      "training_step_time": 8.382071733474731
    },
    {
      "epoch": 3.352642059326172e-05,
      "model_forward_time": 1.730658769607544,
      "step": 35155
    },
    {
      "epoch": 3.352642059326172e-05,
      "step": 35155,
      "training_step_time": 9.622426509857178
    },
    {
      "epoch": 3.3527374267578124e-05,
      "model_forward_time": 2.8847620487213135,
      "step": 35156
    },
    {
      "epoch": 3.3527374267578124e-05,
      "step": 35156,
      "training_step_time": 8.418054342269897
    },
    {
      "epoch": 3.352832794189453e-05,
      "model_forward_time": 1.4993610382080078,
      "step": 35157
    },
    {
      "epoch": 3.352832794189453e-05,
      "step": 35157,
      "training_step_time": 9.32499647140503
    },
    {
      "epoch": 3.352928161621094e-05,
      "model_forward_time": 1.7377126216888428,
      "step": 35158
    },
    {
      "epoch": 3.352928161621094e-05,
      "step": 35158,
      "training_step_time": 7.4528303146362305
    },
    {
      "epoch": 3.353023529052734e-05,
      "model_forward_time": 1.7296152114868164,
      "step": 35159
    },
    {
      "epoch": 3.353023529052734e-05,
      "step": 35159,
      "training_step_time": 6.783001661300659
    },
    {
      "epoch": 3.353118896484375e-05,
      "grad_norm": 0.10174131393432617,
      "learning_rate": 7.712693839414967e-06,
      "loss": 0.0603,
      "step": 35160
    },
    {
      "epoch": 3.353118896484375e-05,
      "model_forward_time": 1.7283425331115723,
      "step": 35160
    },
    {
      "dynamics_loss_avg": 0.041970652900636195,
      "epoch": 3.353118896484375e-05,
      "step": 35160
    },
    {
      "action_loss_avg": 0.0016720217128749937,
      "epoch": 3.353118896484375e-05,
      "step": 35160
    },
    {
      "epoch": 3.353118896484375e-05,
      "step": 35160,
      "training_step_time": 6.812758684158325
    },
    {
      "epoch": 3.353214263916016e-05,
      "model_forward_time": 1.776048183441162,
      "step": 35161
    },
    {
      "epoch": 3.353214263916016e-05,
      "step": 35161,
      "training_step_time": 6.768934488296509
    },
    {
      "epoch": 3.353309631347656e-05,
      "model_forward_time": 1.7400572299957275,
      "step": 35162
    },
    {
      "epoch": 3.353309631347656e-05,
      "step": 35162,
      "training_step_time": 7.554354190826416
    },
    {
      "epoch": 3.3534049987792966e-05,
      "model_forward_time": 1.7305631637573242,
      "step": 35163
    },
    {
      "epoch": 3.3534049987792966e-05,
      "step": 35163,
      "training_step_time": 7.492142200469971
    },
    {
      "epoch": 3.3535003662109376e-05,
      "model_forward_time": 1.7307400703430176,
      "step": 35164
    },
    {
      "epoch": 3.3535003662109376e-05,
      "step": 35164,
      "training_step_time": 8.212372064590454
    },
    {
      "epoch": 3.353595733642578e-05,
      "model_forward_time": 2.891648292541504,
      "step": 35165
    },
    {
      "epoch": 3.353595733642578e-05,
      "step": 35165,
      "training_step_time": 8.156932353973389
    },
    {
      "epoch": 3.3536911010742184e-05,
      "model_forward_time": 1.8179588317871094,
      "step": 35166
    },
    {
      "epoch": 3.3536911010742184e-05,
      "step": 35166,
      "training_step_time": 8.510972738265991
    },
    {
      "epoch": 3.3537864685058595e-05,
      "model_forward_time": 1.732093095779419,
      "step": 35167
    },
    {
      "epoch": 3.3537864685058595e-05,
      "step": 35167,
      "training_step_time": 7.510805130004883
    },
    {
      "epoch": 3.3538818359375e-05,
      "model_forward_time": 1.7395129203796387,
      "step": 35168
    },
    {
      "epoch": 3.3538818359375e-05,
      "step": 35168,
      "training_step_time": 6.818438529968262
    },
    {
      "epoch": 3.35397720336914e-05,
      "model_forward_time": 1.723165512084961,
      "step": 35169
    },
    {
      "epoch": 3.35397720336914e-05,
      "step": 35169,
      "training_step_time": 6.707420349121094
    },
    {
      "epoch": 3.3540725708007814e-05,
      "grad_norm": 0.1432424634695053,
      "learning_rate": 7.711304725597164e-06,
      "loss": 0.0584,
      "step": 35170
    },
    {
      "epoch": 3.3540725708007814e-05,
      "model_forward_time": 1.744034767150879,
      "step": 35170
    },
    {
      "dynamics_loss_avg": 0.05573858171701431,
      "epoch": 3.3540725708007814e-05,
      "step": 35170
    },
    {
      "action_loss_avg": 0.0025380624720128253,
      "epoch": 3.3540725708007814e-05,
      "step": 35170
    },
    {
      "epoch": 3.3540725708007814e-05,
      "step": 35170,
      "training_step_time": 6.708887338638306
    },
    {
      "epoch": 3.354167938232422e-05,
      "model_forward_time": 1.777040719985962,
      "step": 35171
    },
    {
      "epoch": 3.354167938232422e-05,
      "step": 35171,
      "training_step_time": 7.3060455322265625
    },
    {
      "epoch": 3.354263305664062e-05,
      "model_forward_time": 1.7333567142486572,
      "step": 35172
    },
    {
      "epoch": 3.354263305664062e-05,
      "step": 35172,
      "training_step_time": 7.628044843673706
    },
    {
      "epoch": 3.354358673095703e-05,
      "model_forward_time": 1.7396652698516846,
      "step": 35173
    },
    {
      "epoch": 3.354358673095703e-05,
      "step": 35173,
      "training_step_time": 9.531191349029541
    },
    {
      "epoch": 3.354454040527344e-05,
      "model_forward_time": 2.9092600345611572,
      "step": 35174
    },
    {
      "epoch": 3.354454040527344e-05,
      "step": 35174,
      "training_step_time": 10.531052827835083
    },
    {
      "epoch": 3.354549407958984e-05,
      "model_forward_time": 1.8346741199493408,
      "step": 35175
    },
    {
      "epoch": 3.354549407958984e-05,
      "step": 35175,
      "training_step_time": 7.78183126449585
    },
    {
      "epoch": 3.354644775390625e-05,
      "model_forward_time": 1.7463405132293701,
      "step": 35176
    },
    {
      "epoch": 3.354644775390625e-05,
      "step": 35176,
      "training_step_time": 8.400397777557373
    },
    {
      "epoch": 3.3547401428222656e-05,
      "model_forward_time": 1.7224833965301514,
      "step": 35177
    },
    {
      "epoch": 3.3547401428222656e-05,
      "step": 35177,
      "training_step_time": 6.980714559555054
    },
    {
      "epoch": 3.354835510253906e-05,
      "model_forward_time": 1.7363789081573486,
      "step": 35178
    },
    {
      "epoch": 3.354835510253906e-05,
      "step": 35178,
      "training_step_time": 6.87687873840332
    },
    {
      "epoch": 3.354930877685547e-05,
      "model_forward_time": 1.3740870952606201,
      "step": 35179
    },
    {
      "epoch": 3.354930877685547e-05,
      "step": 35179,
      "training_step_time": 6.913779258728027
    },
    {
      "epoch": 3.3550262451171875e-05,
      "grad_norm": 0.1412331759929657,
      "learning_rate": 7.709915315275151e-06,
      "loss": 0.0607,
      "step": 35180
    },
    {
      "epoch": 3.3550262451171875e-05,
      "model_forward_time": 1.7416191101074219,
      "step": 35180
    },
    {
      "dynamics_loss_avg": 0.047402587905526164,
      "epoch": 3.3550262451171875e-05,
      "step": 35180
    },
    {
      "action_loss_avg": 0.0015955919065163472,
      "epoch": 3.3550262451171875e-05,
      "step": 35180
    },
    {
      "epoch": 3.3550262451171875e-05,
      "step": 35180,
      "training_step_time": 7.346002101898193
    },
    {
      "epoch": 3.355121612548828e-05,
      "model_forward_time": 1.7331700325012207,
      "step": 35181
    },
    {
      "epoch": 3.355121612548828e-05,
      "step": 35181,
      "training_step_time": 7.72077202796936
    },
    {
      "epoch": 3.355216979980469e-05,
      "model_forward_time": 1.7271084785461426,
      "step": 35182
    },
    {
      "epoch": 3.355216979980469e-05,
      "step": 35182,
      "training_step_time": 8.649445295333862
    },
    {
      "epoch": 3.3553123474121094e-05,
      "model_forward_time": 1.7358481884002686,
      "step": 35183
    },
    {
      "epoch": 3.3553123474121094e-05,
      "step": 35183,
      "training_step_time": 10.22337007522583
    },
    {
      "epoch": 3.35540771484375e-05,
      "model_forward_time": 2.9384336471557617,
      "step": 35184
    },
    {
      "epoch": 3.35540771484375e-05,
      "step": 35184,
      "training_step_time": 9.680540323257446
    },
    {
      "epoch": 3.355503082275391e-05,
      "model_forward_time": 1.7423064708709717,
      "step": 35185
    },
    {
      "epoch": 3.355503082275391e-05,
      "step": 35185,
      "training_step_time": 7.827811241149902
    },
    {
      "epoch": 3.355598449707031e-05,
      "model_forward_time": 1.746321678161621,
      "step": 35186
    },
    {
      "epoch": 3.355598449707031e-05,
      "step": 35186,
      "training_step_time": 7.619413375854492
    },
    {
      "epoch": 3.3556938171386716e-05,
      "model_forward_time": 1.3959755897521973,
      "step": 35187
    },
    {
      "epoch": 3.3556938171386716e-05,
      "step": 35187,
      "training_step_time": 6.998414516448975
    },
    {
      "epoch": 3.355789184570313e-05,
      "model_forward_time": 1.738325595855713,
      "step": 35188
    },
    {
      "epoch": 3.355789184570313e-05,
      "step": 35188,
      "training_step_time": 6.906210660934448
    },
    {
      "epoch": 3.355884552001953e-05,
      "model_forward_time": 1.73014497756958,
      "step": 35189
    },
    {
      "epoch": 3.355884552001953e-05,
      "step": 35189,
      "training_step_time": 7.389665126800537
    },
    {
      "epoch": 3.3559799194335935e-05,
      "grad_norm": 0.1191108375787735,
      "learning_rate": 7.708525608600876e-06,
      "loss": 0.0634,
      "step": 35190
    },
    {
      "epoch": 3.3559799194335935e-05,
      "model_forward_time": 1.733656406402588,
      "step": 35190
    },
    {
      "dynamics_loss_avg": 0.02835158621892333,
      "epoch": 3.3559799194335935e-05,
      "step": 35190
    },
    {
      "action_loss_avg": 0.000993831249070354,
      "epoch": 3.3559799194335935e-05,
      "step": 35190
    },
    {
      "epoch": 3.3559799194335935e-05,
      "step": 35190,
      "training_step_time": 7.7866644859313965
    },
    {
      "epoch": 3.3560752868652346e-05,
      "model_forward_time": 1.7400071620941162,
      "step": 35191
    },
    {
      "epoch": 3.3560752868652346e-05,
      "step": 35191,
      "training_step_time": 7.9149463176727295
    },
    {
      "epoch": 3.356170654296875e-05,
      "model_forward_time": 1.7342250347137451,
      "step": 35192
    },
    {
      "epoch": 3.356170654296875e-05,
      "step": 35192,
      "training_step_time": 8.831715822219849
    },
    {
      "epoch": 3.3562660217285154e-05,
      "model_forward_time": 2.9417924880981445,
      "step": 35193
    },
    {
      "epoch": 3.3562660217285154e-05,
      "step": 35193,
      "training_step_time": 8.649577617645264
    },
    {
      "epoch": 3.3563613891601565e-05,
      "model_forward_time": 1.7266969680786133,
      "step": 35194
    },
    {
      "epoch": 3.3563613891601565e-05,
      "step": 35194,
      "training_step_time": 7.839237689971924
    },
    {
      "epoch": 3.356456756591797e-05,
      "model_forward_time": 1.7363300323486328,
      "step": 35195
    },
    {
      "epoch": 3.356456756591797e-05,
      "step": 35195,
      "training_step_time": 7.626025438308716
    },
    {
      "epoch": 3.356552124023437e-05,
      "model_forward_time": 1.737966537475586,
      "step": 35196
    },
    {
      "epoch": 3.356552124023437e-05,
      "step": 35196,
      "training_step_time": 6.886367082595825
    },
    {
      "epoch": 3.3566474914550784e-05,
      "model_forward_time": 1.7381396293640137,
      "step": 35197
    },
    {
      "epoch": 3.3566474914550784e-05,
      "step": 35197,
      "training_step_time": 6.923890113830566
    },
    {
      "epoch": 3.356742858886719e-05,
      "model_forward_time": 1.7385501861572266,
      "step": 35198
    },
    {
      "epoch": 3.356742858886719e-05,
      "step": 35198,
      "training_step_time": 7.432942152023315
    },
    {
      "epoch": 3.356838226318359e-05,
      "model_forward_time": 1.742372989654541,
      "step": 35199
    },
    {
      "epoch": 3.356838226318359e-05,
      "step": 35199,
      "training_step_time": 7.740102767944336
    },
    {
      "epoch": 3.35693359375e-05,
      "grad_norm": 0.17045056819915771,
      "learning_rate": 7.707135605726312e-06,
      "loss": 0.0634,
      "step": 35200
    },
    {
      "epoch": 3.35693359375e-05,
      "model_forward_time": 1.7677404880523682,
      "step": 35200
    },
    {
      "dynamics_loss_avg": 0.0644982310011983,
      "epoch": 3.35693359375e-05,
      "step": 35200
    },
    {
      "action_loss_avg": 0.002620599415968172,
      "epoch": 3.35693359375e-05,
      "step": 35200
    },
    {
      "epoch": 3.35693359375e-05,
      "step": 35200,
      "training_step_time": 8.980156660079956
    },
    {
      "epoch": 3.357028961181641e-05,
      "model_forward_time": 1.7305948734283447,
      "step": 35201
    },
    {
      "epoch": 3.357028961181641e-05,
      "step": 35201,
      "training_step_time": 9.926966428756714
    },
    {
      "epoch": 3.357124328613281e-05,
      "model_forward_time": 2.92824387550354,
      "step": 35202
    },
    {
      "epoch": 3.357124328613281e-05,
      "step": 35202,
      "training_step_time": 9.524770975112915
    },
    {
      "epoch": 3.357219696044922e-05,
      "model_forward_time": 1.7984437942504883,
      "step": 35203
    },
    {
      "epoch": 3.357219696044922e-05,
      "step": 35203,
      "training_step_time": 7.77829122543335
    },
    {
      "epoch": 3.3573150634765626e-05,
      "model_forward_time": 1.7282993793487549,
      "step": 35204
    },
    {
      "epoch": 3.3573150634765626e-05,
      "step": 35204,
      "training_step_time": 7.512732744216919
    },
    {
      "epoch": 3.357410430908203e-05,
      "model_forward_time": 1.731196641921997,
      "step": 35205
    },
    {
      "epoch": 3.357410430908203e-05,
      "step": 35205,
      "training_step_time": 6.856621503829956
    },
    {
      "epoch": 3.357505798339844e-05,
      "model_forward_time": 1.374845027923584,
      "step": 35206
    },
    {
      "epoch": 3.357505798339844e-05,
      "step": 35206,
      "training_step_time": 6.823966026306152
    },
    {
      "epoch": 3.3576011657714844e-05,
      "model_forward_time": 1.7397656440734863,
      "step": 35207
    },
    {
      "epoch": 3.3576011657714844e-05,
      "step": 35207,
      "training_step_time": 6.920134782791138
    },
    {
      "epoch": 3.357696533203125e-05,
      "model_forward_time": 1.7401847839355469,
      "step": 35208
    },
    {
      "epoch": 3.357696533203125e-05,
      "step": 35208,
      "training_step_time": 7.72962212562561
    },
    {
      "epoch": 3.357791900634766e-05,
      "model_forward_time": 1.7444615364074707,
      "step": 35209
    },
    {
      "epoch": 3.357791900634766e-05,
      "step": 35209,
      "training_step_time": 7.781568765640259
    },
    {
      "epoch": 3.357887268066406e-05,
      "grad_norm": 0.1337951272726059,
      "learning_rate": 7.705745306803466e-06,
      "loss": 0.0667,
      "step": 35210
    },
    {
      "epoch": 3.357887268066406e-05,
      "model_forward_time": 1.7322297096252441,
      "step": 35210
    },
    {
      "dynamics_loss_avg": 0.0598852826282382,
      "epoch": 3.357887268066406e-05,
      "step": 35210
    },
    {
      "action_loss_avg": 0.0019935346907004716,
      "epoch": 3.357887268066406e-05,
      "step": 35210
    },
    {
      "epoch": 3.357887268066406e-05,
      "step": 35210,
      "training_step_time": 8.90062689781189
    },
    {
      "epoch": 3.357982635498047e-05,
      "model_forward_time": 2.79468035697937,
      "step": 35211
    },
    {
      "epoch": 3.357982635498047e-05,
      "step": 35211,
      "training_step_time": 9.433289051055908
    },
    {
      "epoch": 3.358078002929688e-05,
      "model_forward_time": 1.842120885848999,
      "step": 35212
    },
    {
      "epoch": 3.358078002929688e-05,
      "step": 35212,
      "training_step_time": 7.818206787109375
    },
    {
      "epoch": 3.358173370361328e-05,
      "model_forward_time": 1.7437279224395752,
      "step": 35213
    },
    {
      "epoch": 3.358173370361328e-05,
      "step": 35213,
      "training_step_time": 7.790488243103027
    },
    {
      "epoch": 3.3582687377929686e-05,
      "model_forward_time": 1.7313246726989746,
      "step": 35214
    },
    {
      "epoch": 3.3582687377929686e-05,
      "step": 35214,
      "training_step_time": 6.896066188812256
    },
    {
      "epoch": 3.35836410522461e-05,
      "model_forward_time": 1.7390050888061523,
      "step": 35215
    },
    {
      "epoch": 3.35836410522461e-05,
      "step": 35215,
      "training_step_time": 6.890879392623901
    },
    {
      "epoch": 3.35845947265625e-05,
      "model_forward_time": 1.7534399032592773,
      "step": 35216
    },
    {
      "epoch": 3.35845947265625e-05,
      "step": 35216,
      "training_step_time": 6.938657522201538
    },
    {
      "epoch": 3.3585548400878905e-05,
      "model_forward_time": 1.7393484115600586,
      "step": 35217
    },
    {
      "epoch": 3.3585548400878905e-05,
      "step": 35217,
      "training_step_time": 7.424672842025757
    },
    {
      "epoch": 3.3586502075195316e-05,
      "model_forward_time": 1.7461042404174805,
      "step": 35218
    },
    {
      "epoch": 3.3586502075195316e-05,
      "step": 35218,
      "training_step_time": 7.841439723968506
    },
    {
      "epoch": 3.358745574951172e-05,
      "model_forward_time": 1.7300879955291748,
      "step": 35219
    },
    {
      "epoch": 3.358745574951172e-05,
      "step": 35219,
      "training_step_time": 8.961211681365967
    },
    {
      "epoch": 3.3588409423828124e-05,
      "grad_norm": 0.08929494768381119,
      "learning_rate": 7.704354711984384e-06,
      "loss": 0.0666,
      "step": 35220
    },
    {
      "epoch": 3.3588409423828124e-05,
      "model_forward_time": 1.7362861633300781,
      "step": 35220
    },
    {
      "dynamics_loss_avg": 0.04665887365117669,
      "epoch": 3.3588409423828124e-05,
      "step": 35220
    },
    {
      "action_loss_avg": 0.0016179931175429375,
      "epoch": 3.3588409423828124e-05,
      "step": 35220
    },
    {
      "epoch": 3.3588409423828124e-05,
      "step": 35220,
      "training_step_time": 9.183667421340942
    },
    {
      "epoch": 3.358936309814453e-05,
      "model_forward_time": 2.5984432697296143,
      "step": 35221
    },
    {
      "epoch": 3.358936309814453e-05,
      "step": 35221,
      "training_step_time": 9.418899297714233
    },
    {
      "epoch": 3.359031677246094e-05,
      "model_forward_time": 1.7375965118408203,
      "step": 35222
    },
    {
      "epoch": 3.359031677246094e-05,
      "step": 35222,
      "training_step_time": 7.726762294769287
    },
    {
      "epoch": 3.359127044677734e-05,
      "model_forward_time": 1.7342031002044678,
      "step": 35223
    },
    {
      "epoch": 3.359127044677734e-05,
      "step": 35223,
      "training_step_time": 7.581932067871094
    },
    {
      "epoch": 3.359222412109375e-05,
      "model_forward_time": 1.7464048862457275,
      "step": 35224
    },
    {
      "epoch": 3.359222412109375e-05,
      "step": 35224,
      "training_step_time": 6.882356643676758
    },
    {
      "epoch": 3.359317779541016e-05,
      "model_forward_time": 1.7306315898895264,
      "step": 35225
    },
    {
      "epoch": 3.359317779541016e-05,
      "step": 35225,
      "training_step_time": 6.873891830444336
    },
    {
      "epoch": 3.359413146972656e-05,
      "model_forward_time": 1.729485273361206,
      "step": 35226
    },
    {
      "epoch": 3.359413146972656e-05,
      "step": 35226,
      "training_step_time": 7.435047388076782
    },
    {
      "epoch": 3.3595085144042966e-05,
      "model_forward_time": 1.7358627319335938,
      "step": 35227
    },
    {
      "epoch": 3.3595085144042966e-05,
      "step": 35227,
      "training_step_time": 8.782948732376099
    },
    {
      "epoch": 3.3596038818359376e-05,
      "model_forward_time": 1.7391610145568848,
      "step": 35228
    },
    {
      "epoch": 3.3596038818359376e-05,
      "step": 35228,
      "training_step_time": 7.947243928909302
    },
    {
      "epoch": 3.359699249267578e-05,
      "model_forward_time": 1.735684871673584,
      "step": 35229
    },
    {
      "epoch": 3.359699249267578e-05,
      "step": 35229,
      "training_step_time": 8.541486024856567
    },
    {
      "epoch": 3.3597946166992185e-05,
      "grad_norm": 0.1442667543888092,
      "learning_rate": 7.702963821421135e-06,
      "loss": 0.0627,
      "step": 35230
    },
    {
      "epoch": 3.3597946166992185e-05,
      "model_forward_time": 2.8954498767852783,
      "step": 35230
    },
    {
      "dynamics_loss_avg": 0.051123237051069734,
      "epoch": 3.3597946166992185e-05,
      "step": 35230
    },
    {
      "action_loss_avg": 0.002254813595209271,
      "epoch": 3.3597946166992185e-05,
      "step": 35230
    },
    {
      "epoch": 3.3597946166992185e-05,
      "step": 35230,
      "training_step_time": 7.742517948150635
    },
    {
      "epoch": 3.3598899841308595e-05,
      "model_forward_time": 1.7373948097229004,
      "step": 35231
    },
    {
      "epoch": 3.3598899841308595e-05,
      "step": 35231,
      "training_step_time": 8.802244901657104
    },
    {
      "epoch": 3.3599853515625e-05,
      "model_forward_time": 1.7489280700683594,
      "step": 35232
    },
    {
      "epoch": 3.3599853515625e-05,
      "step": 35232,
      "training_step_time": 7.541346788406372
    },
    {
      "epoch": 3.36008071899414e-05,
      "model_forward_time": 1.7377941608428955,
      "step": 35233
    },
    {
      "epoch": 3.36008071899414e-05,
      "step": 35233,
      "training_step_time": 6.9240639209747314
    },
    {
      "epoch": 3.3601760864257814e-05,
      "model_forward_time": 1.747096300125122,
      "step": 35234
    },
    {
      "epoch": 3.3601760864257814e-05,
      "step": 35234,
      "training_step_time": 6.908027410507202
    },
    {
      "epoch": 3.360271453857422e-05,
      "model_forward_time": 1.7397942543029785,
      "step": 35235
    },
    {
      "epoch": 3.360271453857422e-05,
      "step": 35235,
      "training_step_time": 7.4002790451049805
    },
    {
      "epoch": 3.360366821289062e-05,
      "model_forward_time": 1.731523036956787,
      "step": 35236
    },
    {
      "epoch": 3.360366821289062e-05,
      "step": 35236,
      "training_step_time": 7.8030781745910645
    },
    {
      "epoch": 3.360462188720703e-05,
      "model_forward_time": 1.7485325336456299,
      "step": 35237
    },
    {
      "epoch": 3.360462188720703e-05,
      "step": 35237,
      "training_step_time": 9.601063013076782
    },
    {
      "epoch": 3.360557556152344e-05,
      "model_forward_time": 1.7769818305969238,
      "step": 35238
    },
    {
      "epoch": 3.360557556152344e-05,
      "step": 35238,
      "training_step_time": 9.927400350570679
    },
    {
      "epoch": 3.360652923583984e-05,
      "model_forward_time": 2.9389150142669678,
      "step": 35239
    },
    {
      "epoch": 3.360652923583984e-05,
      "step": 35239,
      "training_step_time": 8.80458688735962
    },
    {
      "epoch": 3.360748291015625e-05,
      "grad_norm": 0.18353213369846344,
      "learning_rate": 7.701572635265826e-06,
      "loss": 0.0637,
      "step": 35240
    },
    {
      "epoch": 3.360748291015625e-05,
      "model_forward_time": 1.7545981407165527,
      "step": 35240
    },
    {
      "dynamics_loss_avg": 0.05500399051234126,
      "epoch": 3.360748291015625e-05,
      "step": 35240
    },
    {
      "action_loss_avg": 0.0013855945930117742,
      "epoch": 3.360748291015625e-05,
      "step": 35240
    },
    {
      "epoch": 3.360748291015625e-05,
      "step": 35240,
      "training_step_time": 8.748160600662231
    },
    {
      "epoch": 3.3608436584472656e-05,
      "model_forward_time": 1.736027717590332,
      "step": 35241
    },
    {
      "epoch": 3.3608436584472656e-05,
      "step": 35241,
      "training_step_time": 7.79997706413269
    },
    {
      "epoch": 3.360939025878906e-05,
      "model_forward_time": 1.7363760471343994,
      "step": 35242
    },
    {
      "epoch": 3.360939025878906e-05,
      "step": 35242,
      "training_step_time": 6.803589582443237
    },
    {
      "epoch": 3.361034393310547e-05,
      "model_forward_time": 1.7374250888824463,
      "step": 35243
    },
    {
      "epoch": 3.361034393310547e-05,
      "step": 35243,
      "training_step_time": 6.8396079540252686
    },
    {
      "epoch": 3.3611297607421875e-05,
      "model_forward_time": 1.744978427886963,
      "step": 35244
    },
    {
      "epoch": 3.3611297607421875e-05,
      "step": 35244,
      "training_step_time": 7.362271785736084
    },
    {
      "epoch": 3.361225128173828e-05,
      "model_forward_time": 1.7285332679748535,
      "step": 35245
    },
    {
      "epoch": 3.361225128173828e-05,
      "step": 35245,
      "training_step_time": 7.728302717208862
    },
    {
      "epoch": 3.361320495605469e-05,
      "model_forward_time": 1.7730369567871094,
      "step": 35246
    },
    {
      "epoch": 3.361320495605469e-05,
      "step": 35246,
      "training_step_time": 8.161010026931763
    },
    {
      "epoch": 3.3614158630371094e-05,
      "model_forward_time": 1.7571651935577393,
      "step": 35247
    },
    {
      "epoch": 3.3614158630371094e-05,
      "step": 35247,
      "training_step_time": 8.736145973205566
    },
    {
      "epoch": 3.36151123046875e-05,
      "model_forward_time": 2.892998218536377,
      "step": 35248
    },
    {
      "epoch": 3.36151123046875e-05,
      "step": 35248,
      "training_step_time": 8.758797407150269
    },
    {
      "epoch": 3.361606597900391e-05,
      "model_forward_time": 1.8550119400024414,
      "step": 35249
    },
    {
      "epoch": 3.361606597900391e-05,
      "step": 35249,
      "training_step_time": 7.6553778648376465
    },
    {
      "epoch": 3.361701965332031e-05,
      "grad_norm": 0.12922807037830353,
      "learning_rate": 7.700181153670597e-06,
      "loss": 0.0625,
      "step": 35250
    },
    {
      "epoch": 3.361701965332031e-05,
      "model_forward_time": 1.7328155040740967,
      "step": 35250
    },
    {
      "dynamics_loss_avg": 0.05786523688584566,
      "epoch": 3.361701965332031e-05,
      "step": 35250
    },
    {
      "action_loss_avg": 0.001524941436946392,
      "epoch": 3.361701965332031e-05,
      "step": 35250
    },
    {
      "epoch": 3.361701965332031e-05,
      "step": 35250,
      "training_step_time": 7.852346181869507
    },
    {
      "epoch": 3.3617973327636717e-05,
      "model_forward_time": 1.7442116737365723,
      "step": 35251
    },
    {
      "epoch": 3.3617973327636717e-05,
      "step": 35251,
      "training_step_time": 7.140145540237427
    },
    {
      "epoch": 3.361892700195313e-05,
      "model_forward_time": 1.740384578704834,
      "step": 35252
    },
    {
      "epoch": 3.361892700195313e-05,
      "step": 35252,
      "training_step_time": 6.852931261062622
    },
    {
      "epoch": 3.361988067626953e-05,
      "model_forward_time": 1.73162841796875,
      "step": 35253
    },
    {
      "epoch": 3.361988067626953e-05,
      "step": 35253,
      "training_step_time": 6.877704858779907
    },
    {
      "epoch": 3.3620834350585935e-05,
      "model_forward_time": 1.7350928783416748,
      "step": 35254
    },
    {
      "epoch": 3.3620834350585935e-05,
      "step": 35254,
      "training_step_time": 7.702343463897705
    },
    {
      "epoch": 3.3621788024902346e-05,
      "model_forward_time": 1.7355175018310547,
      "step": 35255
    },
    {
      "epoch": 3.3621788024902346e-05,
      "step": 35255,
      "training_step_time": 8.438727140426636
    },
    {
      "epoch": 3.362274169921875e-05,
      "model_forward_time": 1.7302703857421875,
      "step": 35256
    },
    {
      "epoch": 3.362274169921875e-05,
      "step": 35256,
      "training_step_time": 9.229405879974365
    },
    {
      "epoch": 3.3623695373535154e-05,
      "model_forward_time": 1.7372817993164062,
      "step": 35257
    },
    {
      "epoch": 3.3623695373535154e-05,
      "step": 35257,
      "training_step_time": 10.148376226425171
    },
    {
      "epoch": 3.3624649047851565e-05,
      "model_forward_time": 2.9266083240509033,
      "step": 35258
    },
    {
      "epoch": 3.3624649047851565e-05,
      "step": 35258,
      "training_step_time": 8.310357332229614
    },
    {
      "epoch": 3.362560272216797e-05,
      "model_forward_time": 1.7424545288085938,
      "step": 35259
    },
    {
      "epoch": 3.362560272216797e-05,
      "step": 35259,
      "training_step_time": 7.74203896522522
    },
    {
      "epoch": 3.362655639648437e-05,
      "grad_norm": 0.15795834362506866,
      "learning_rate": 7.698789376787617e-06,
      "loss": 0.0641,
      "step": 35260
    },
    {
      "epoch": 3.362655639648437e-05,
      "model_forward_time": 1.3889679908752441,
      "step": 35260
    },
    {
      "dynamics_loss_avg": 0.060620241612195966,
      "epoch": 3.362655639648437e-05,
      "step": 35260
    },
    {
      "action_loss_avg": 0.0017332597344648093,
      "epoch": 3.362655639648437e-05,
      "step": 35260
    },
    {
      "epoch": 3.362655639648437e-05,
      "step": 35260,
      "training_step_time": 7.543130159378052
    },
    {
      "epoch": 3.3627510070800784e-05,
      "model_forward_time": 1.7337088584899902,
      "step": 35261
    },
    {
      "epoch": 3.3627510070800784e-05,
      "step": 35261,
      "training_step_time": 6.901318073272705
    },
    {
      "epoch": 3.362846374511719e-05,
      "model_forward_time": 1.7381222248077393,
      "step": 35262
    },
    {
      "epoch": 3.362846374511719e-05,
      "step": 35262,
      "training_step_time": 6.922373056411743
    },
    {
      "epoch": 3.362941741943359e-05,
      "model_forward_time": 1.7463531494140625,
      "step": 35263
    },
    {
      "epoch": 3.362941741943359e-05,
      "step": 35263,
      "training_step_time": 7.730430364608765
    },
    {
      "epoch": 3.363037109375e-05,
      "model_forward_time": 1.7333498001098633,
      "step": 35264
    },
    {
      "epoch": 3.363037109375e-05,
      "step": 35264,
      "training_step_time": 8.45281457901001
    },
    {
      "epoch": 3.363132476806641e-05,
      "model_forward_time": 1.7356901168823242,
      "step": 35265
    },
    {
      "epoch": 3.363132476806641e-05,
      "step": 35265,
      "training_step_time": 9.082401514053345
    },
    {
      "epoch": 3.363227844238281e-05,
      "model_forward_time": 1.7367298603057861,
      "step": 35266
    },
    {
      "epoch": 3.363227844238281e-05,
      "step": 35266,
      "training_step_time": 9.251633882522583
    },
    {
      "epoch": 3.363323211669922e-05,
      "model_forward_time": 2.993335485458374,
      "step": 35267
    },
    {
      "epoch": 3.363323211669922e-05,
      "step": 35267,
      "training_step_time": 7.863728761672974
    },
    {
      "epoch": 3.3634185791015626e-05,
      "model_forward_time": 1.7310843467712402,
      "step": 35268
    },
    {
      "epoch": 3.3634185791015626e-05,
      "step": 35268,
      "training_step_time": 7.8063459396362305
    },
    {
      "epoch": 3.363513946533203e-05,
      "model_forward_time": 1.7342312335968018,
      "step": 35269
    },
    {
      "epoch": 3.363513946533203e-05,
      "step": 35269,
      "training_step_time": 7.799244165420532
    },
    {
      "epoch": 3.363609313964844e-05,
      "grad_norm": 0.15740637481212616,
      "learning_rate": 7.697397304769085e-06,
      "loss": 0.0642,
      "step": 35270
    },
    {
      "epoch": 3.363609313964844e-05,
      "model_forward_time": 1.7434954643249512,
      "step": 35270
    },
    {
      "dynamics_loss_avg": 0.0660718135535717,
      "epoch": 3.363609313964844e-05,
      "step": 35270
    },
    {
      "action_loss_avg": 0.0024344700737856328,
      "epoch": 3.363609313964844e-05,
      "step": 35270
    },
    {
      "epoch": 3.363609313964844e-05,
      "step": 35270,
      "training_step_time": 6.895379304885864
    },
    {
      "epoch": 3.3637046813964845e-05,
      "model_forward_time": 1.7365806102752686,
      "step": 35271
    },
    {
      "epoch": 3.3637046813964845e-05,
      "step": 35271,
      "training_step_time": 6.940054178237915
    },
    {
      "epoch": 3.363800048828125e-05,
      "model_forward_time": 1.7376461029052734,
      "step": 35272
    },
    {
      "epoch": 3.363800048828125e-05,
      "step": 35272,
      "training_step_time": 7.628714561462402
    },
    {
      "epoch": 3.363895416259766e-05,
      "model_forward_time": 1.7313201427459717,
      "step": 35273
    },
    {
      "epoch": 3.363895416259766e-05,
      "step": 35273,
      "training_step_time": 7.797588586807251
    },
    {
      "epoch": 3.363990783691406e-05,
      "model_forward_time": 1.7469604015350342,
      "step": 35274
    },
    {
      "epoch": 3.363990783691406e-05,
      "step": 35274,
      "training_step_time": 9.72390365600586
    },
    {
      "epoch": 3.364086151123047e-05,
      "model_forward_time": 1.7263445854187012,
      "step": 35275
    },
    {
      "epoch": 3.364086151123047e-05,
      "step": 35275,
      "training_step_time": 9.437764644622803
    },
    {
      "epoch": 3.364181518554688e-05,
      "model_forward_time": 2.941403865814209,
      "step": 35276
    },
    {
      "epoch": 3.364181518554688e-05,
      "step": 35276,
      "training_step_time": 7.8090660572052
    },
    {
      "epoch": 3.364276885986328e-05,
      "model_forward_time": 1.737177848815918,
      "step": 35277
    },
    {
      "epoch": 3.364276885986328e-05,
      "step": 35277,
      "training_step_time": 7.729413747787476
    },
    {
      "epoch": 3.3643722534179686e-05,
      "model_forward_time": 1.384559154510498,
      "step": 35278
    },
    {
      "epoch": 3.3643722534179686e-05,
      "step": 35278,
      "training_step_time": 7.873844623565674
    },
    {
      "epoch": 3.36446762084961e-05,
      "model_forward_time": 1.746288776397705,
      "step": 35279
    },
    {
      "epoch": 3.36446762084961e-05,
      "step": 35279,
      "training_step_time": 6.833182334899902
    },
    {
      "epoch": 3.36456298828125e-05,
      "grad_norm": 0.1340598613023758,
      "learning_rate": 7.696004937767242e-06,
      "loss": 0.0615,
      "step": 35280
    },
    {
      "epoch": 3.36456298828125e-05,
      "model_forward_time": 1.7433128356933594,
      "step": 35280
    },
    {
      "dynamics_loss_avg": 0.05751869576051831,
      "epoch": 3.36456298828125e-05,
      "step": 35280
    },
    {
      "action_loss_avg": 0.002421309083001688,
      "epoch": 3.36456298828125e-05,
      "step": 35280
    },
    {
      "epoch": 3.36456298828125e-05,
      "step": 35280,
      "training_step_time": 6.939989328384399
    },
    {
      "epoch": 3.3646583557128905e-05,
      "model_forward_time": 1.7324168682098389,
      "step": 35281
    },
    {
      "epoch": 3.3646583557128905e-05,
      "step": 35281,
      "training_step_time": 7.40302586555481
    },
    {
      "epoch": 3.364753723144531e-05,
      "model_forward_time": 1.74586820602417,
      "step": 35282
    },
    {
      "epoch": 3.364753723144531e-05,
      "step": 35282,
      "training_step_time": 7.719310760498047
    },
    {
      "epoch": 3.364849090576172e-05,
      "model_forward_time": 1.3917522430419922,
      "step": 35283
    },
    {
      "epoch": 3.364849090576172e-05,
      "step": 35283,
      "training_step_time": 8.35265064239502
    },
    {
      "epoch": 3.3649444580078124e-05,
      "model_forward_time": 1.7334272861480713,
      "step": 35284
    },
    {
      "epoch": 3.3649444580078124e-05,
      "step": 35284,
      "training_step_time": 8.396965503692627
    },
    {
      "epoch": 3.365039825439453e-05,
      "model_forward_time": 2.853271722793579,
      "step": 35285
    },
    {
      "epoch": 3.365039825439453e-05,
      "step": 35285,
      "training_step_time": 7.7515575885772705
    },
    {
      "epoch": 3.365135192871094e-05,
      "model_forward_time": 1.8369576930999756,
      "step": 35286
    },
    {
      "epoch": 3.365135192871094e-05,
      "step": 35286,
      "training_step_time": 8.33691954612732
    },
    {
      "epoch": 3.365230560302734e-05,
      "model_forward_time": 1.7389943599700928,
      "step": 35287
    },
    {
      "epoch": 3.365230560302734e-05,
      "step": 35287,
      "training_step_time": 7.799229860305786
    },
    {
      "epoch": 3.365325927734375e-05,
      "model_forward_time": 1.3649787902832031,
      "step": 35288
    },
    {
      "epoch": 3.365325927734375e-05,
      "step": 35288,
      "training_step_time": 6.876126527786255
    },
    {
      "epoch": 3.365421295166016e-05,
      "model_forward_time": 1.7414631843566895,
      "step": 35289
    },
    {
      "epoch": 3.365421295166016e-05,
      "step": 35289,
      "training_step_time": 6.846601724624634
    },
    {
      "epoch": 3.365516662597656e-05,
      "grad_norm": 0.09991148114204407,
      "learning_rate": 7.694612275934352e-06,
      "loss": 0.0668,
      "step": 35290
    },
    {
      "epoch": 3.365516662597656e-05,
      "model_forward_time": 1.7171461582183838,
      "step": 35290
    },
    {
      "dynamics_loss_avg": 0.05989606622606516,
      "epoch": 3.365516662597656e-05,
      "step": 35290
    },
    {
      "action_loss_avg": 0.0011878204008098693,
      "epoch": 3.365516662597656e-05,
      "step": 35290
    },
    {
      "epoch": 3.365516662597656e-05,
      "step": 35290,
      "training_step_time": 7.543132781982422
    },
    {
      "epoch": 3.3656120300292966e-05,
      "model_forward_time": 1.7357618808746338,
      "step": 35291
    },
    {
      "epoch": 3.3656120300292966e-05,
      "step": 35291,
      "training_step_time": 7.67444109916687
    },
    {
      "epoch": 3.3657073974609377e-05,
      "model_forward_time": 1.0625100135803223,
      "step": 35292
    },
    {
      "epoch": 3.3657073974609377e-05,
      "step": 35292,
      "training_step_time": 7.6948466300964355
    },
    {
      "epoch": 3.365802764892578e-05,
      "model_forward_time": 1.7420680522918701,
      "step": 35293
    },
    {
      "epoch": 3.365802764892578e-05,
      "step": 35293,
      "training_step_time": 9.525848627090454
    },
    {
      "epoch": 3.3658981323242185e-05,
      "model_forward_time": 1.737114667892456,
      "step": 35294
    },
    {
      "epoch": 3.3658981323242185e-05,
      "step": 35294,
      "training_step_time": 8.98495101928711
    },
    {
      "epoch": 3.3659934997558595e-05,
      "model_forward_time": 2.87600040435791,
      "step": 35295
    },
    {
      "epoch": 3.3659934997558595e-05,
      "step": 35295,
      "training_step_time": 9.541211128234863
    },
    {
      "epoch": 3.3660888671875e-05,
      "model_forward_time": 1.731142282485962,
      "step": 35296
    },
    {
      "epoch": 3.3660888671875e-05,
      "step": 35296,
      "training_step_time": 7.817664861679077
    },
    {
      "epoch": 3.3661842346191403e-05,
      "model_forward_time": 1.7429718971252441,
      "step": 35297
    },
    {
      "epoch": 3.3661842346191403e-05,
      "step": 35297,
      "training_step_time": 7.778020858764648
    },
    {
      "epoch": 3.3662796020507814e-05,
      "model_forward_time": 1.740173101425171,
      "step": 35298
    },
    {
      "epoch": 3.3662796020507814e-05,
      "step": 35298,
      "training_step_time": 6.73115348815918
    },
    {
      "epoch": 3.366374969482422e-05,
      "model_forward_time": 1.7425942420959473,
      "step": 35299
    },
    {
      "epoch": 3.366374969482422e-05,
      "step": 35299,
      "training_step_time": 7.17177677154541
    },
    {
      "epoch": 3.366470336914062e-05,
      "grad_norm": 0.17374922335147858,
      "learning_rate": 7.693219319422714e-06,
      "loss": 0.061,
      "step": 35300
    },
    {
      "epoch": 3.366470336914062e-05,
      "model_forward_time": 1.3775951862335205,
      "step": 35300
    },
    {
      "dynamics_loss_avg": 0.06627885755151511,
      "epoch": 3.366470336914062e-05,
      "step": 35300
    },
    {
      "action_loss_avg": 0.001720195630332455,
      "epoch": 3.366470336914062e-05,
      "step": 35300
    },
    {
      "epoch": 3.366470336914062e-05,
      "step": 35300,
      "training_step_time": 7.531609773635864
    },
    {
      "epoch": 3.366565704345703e-05,
      "model_forward_time": 1.7373406887054443,
      "step": 35301
    },
    {
      "epoch": 3.366565704345703e-05,
      "step": 35301,
      "training_step_time": 9.90050196647644
    },
    {
      "epoch": 3.366661071777344e-05,
      "model_forward_time": 1.7326111793518066,
      "step": 35302
    },
    {
      "epoch": 3.366661071777344e-05,
      "step": 35302,
      "training_step_time": 7.7633795738220215
    },
    {
      "epoch": 3.366756439208984e-05,
      "model_forward_time": 1.7365643978118896,
      "step": 35303
    },
    {
      "epoch": 3.366756439208984e-05,
      "step": 35303,
      "training_step_time": 7.733392238616943
    },
    {
      "epoch": 3.366851806640625e-05,
      "model_forward_time": 2.861452102661133,
      "step": 35304
    },
    {
      "epoch": 3.366851806640625e-05,
      "step": 35304,
      "training_step_time": 7.677952766418457
    },
    {
      "epoch": 3.3669471740722656e-05,
      "model_forward_time": 1.728543996810913,
      "step": 35305
    },
    {
      "epoch": 3.3669471740722656e-05,
      "step": 35305,
      "training_step_time": 7.760051488876343
    },
    {
      "epoch": 3.367042541503906e-05,
      "model_forward_time": 1.7309043407440186,
      "step": 35306
    },
    {
      "epoch": 3.367042541503906e-05,
      "step": 35306,
      "training_step_time": 7.7828381061553955
    },
    {
      "epoch": 3.367137908935547e-05,
      "model_forward_time": 1.748931884765625,
      "step": 35307
    },
    {
      "epoch": 3.367137908935547e-05,
      "step": 35307,
      "training_step_time": 6.824082374572754
    },
    {
      "epoch": 3.3672332763671875e-05,
      "model_forward_time": 1.7486331462860107,
      "step": 35308
    },
    {
      "epoch": 3.3672332763671875e-05,
      "step": 35308,
      "training_step_time": 6.770236253738403
    },
    {
      "epoch": 3.367328643798828e-05,
      "model_forward_time": 1.7377448081970215,
      "step": 35309
    },
    {
      "epoch": 3.367328643798828e-05,
      "step": 35309,
      "training_step_time": 7.605956792831421
    },
    {
      "epoch": 3.367424011230469e-05,
      "grad_norm": 0.1534830927848816,
      "learning_rate": 7.691826068384662e-06,
      "loss": 0.0656,
      "step": 35310
    },
    {
      "epoch": 3.367424011230469e-05,
      "model_forward_time": 1.7348039150238037,
      "step": 35310
    },
    {
      "dynamics_loss_avg": 0.06616636086255312,
      "epoch": 3.367424011230469e-05,
      "step": 35310
    },
    {
      "action_loss_avg": 0.0019849939970299603,
      "epoch": 3.367424011230469e-05,
      "step": 35310
    },
    {
      "epoch": 3.367424011230469e-05,
      "step": 35310,
      "training_step_time": 8.253283500671387
    },
    {
      "epoch": 3.3675193786621094e-05,
      "model_forward_time": 1.7320120334625244,
      "step": 35311
    },
    {
      "epoch": 3.3675193786621094e-05,
      "step": 35311,
      "training_step_time": 8.516455888748169
    },
    {
      "epoch": 3.36761474609375e-05,
      "model_forward_time": 1.7389893531799316,
      "step": 35312
    },
    {
      "epoch": 3.36761474609375e-05,
      "step": 35312,
      "training_step_time": 8.86642837524414
    },
    {
      "epoch": 3.367710113525391e-05,
      "model_forward_time": 2.8663270473480225,
      "step": 35313
    },
    {
      "epoch": 3.367710113525391e-05,
      "step": 35313,
      "training_step_time": 7.8086206912994385
    },
    {
      "epoch": 3.367805480957031e-05,
      "model_forward_time": 1.73832106590271,
      "step": 35314
    },
    {
      "epoch": 3.367805480957031e-05,
      "step": 35314,
      "training_step_time": 8.445070266723633
    },
    {
      "epoch": 3.3679008483886717e-05,
      "model_forward_time": 1.7327542304992676,
      "step": 35315
    },
    {
      "epoch": 3.3679008483886717e-05,
      "step": 35315,
      "training_step_time": 8.425204753875732
    },
    {
      "epoch": 3.367996215820313e-05,
      "model_forward_time": 1.7333557605743408,
      "step": 35316
    },
    {
      "epoch": 3.367996215820313e-05,
      "step": 35316,
      "training_step_time": 6.891327857971191
    },
    {
      "epoch": 3.368091583251953e-05,
      "model_forward_time": 1.750063419342041,
      "step": 35317
    },
    {
      "epoch": 3.368091583251953e-05,
      "step": 35317,
      "training_step_time": 6.872742652893066
    },
    {
      "epoch": 3.3681869506835935e-05,
      "model_forward_time": 1.7361235618591309,
      "step": 35318
    },
    {
      "epoch": 3.3681869506835935e-05,
      "step": 35318,
      "training_step_time": 7.6476380825042725
    },
    {
      "epoch": 3.3682823181152346e-05,
      "model_forward_time": 1.748049020767212,
      "step": 35319
    },
    {
      "epoch": 3.3682823181152346e-05,
      "step": 35319,
      "training_step_time": 7.848696947097778
    },
    {
      "epoch": 3.368377685546875e-05,
      "grad_norm": 0.2899080812931061,
      "learning_rate": 7.690432522972557e-06,
      "loss": 0.0609,
      "step": 35320
    },
    {
      "epoch": 3.368377685546875e-05,
      "model_forward_time": 1.7312281131744385,
      "step": 35320
    },
    {
      "dynamics_loss_avg": 0.055500369891524313,
      "epoch": 3.368377685546875e-05,
      "step": 35320
    },
    {
      "action_loss_avg": 0.001911509461933747,
      "epoch": 3.368377685546875e-05,
      "step": 35320
    },
    {
      "epoch": 3.368377685546875e-05,
      "step": 35320,
      "training_step_time": 8.611066579818726
    },
    {
      "epoch": 3.3684730529785154e-05,
      "model_forward_time": 1.7352290153503418,
      "step": 35321
    },
    {
      "epoch": 3.3684730529785154e-05,
      "step": 35321,
      "training_step_time": 9.813666820526123
    },
    {
      "epoch": 3.3685684204101565e-05,
      "model_forward_time": 1.7406516075134277,
      "step": 35322
    },
    {
      "epoch": 3.3685684204101565e-05,
      "step": 35322,
      "training_step_time": 7.840512037277222
    },
    {
      "epoch": 3.368663787841797e-05,
      "model_forward_time": 2.98138689994812,
      "step": 35323
    },
    {
      "epoch": 3.368663787841797e-05,
      "step": 35323,
      "training_step_time": 8.836683750152588
    },
    {
      "epoch": 3.368759155273437e-05,
      "model_forward_time": 1.7384233474731445,
      "step": 35324
    },
    {
      "epoch": 3.368759155273437e-05,
      "step": 35324,
      "training_step_time": 8.21412706375122
    },
    {
      "epoch": 3.3688545227050784e-05,
      "model_forward_time": 1.3884034156799316,
      "step": 35325
    },
    {
      "epoch": 3.3688545227050784e-05,
      "step": 35325,
      "training_step_time": 7.165184736251831
    },
    {
      "epoch": 3.368949890136719e-05,
      "model_forward_time": 1.7496974468231201,
      "step": 35326
    },
    {
      "epoch": 3.368949890136719e-05,
      "step": 35326,
      "training_step_time": 6.808685302734375
    },
    {
      "epoch": 3.369045257568359e-05,
      "model_forward_time": 1.7250916957855225,
      "step": 35327
    },
    {
      "epoch": 3.369045257568359e-05,
      "step": 35327,
      "training_step_time": 7.381481170654297
    },
    {
      "epoch": 3.369140625e-05,
      "model_forward_time": 1.7302320003509521,
      "step": 35328
    },
    {
      "epoch": 3.369140625e-05,
      "step": 35328,
      "training_step_time": 8.332538366317749
    },
    {
      "epoch": 3.369235992431641e-05,
      "model_forward_time": 1.738774299621582,
      "step": 35329
    },
    {
      "epoch": 3.369235992431641e-05,
      "step": 35329,
      "training_step_time": 8.776970624923706
    },
    {
      "epoch": 3.369331359863281e-05,
      "grad_norm": 0.18334606289863586,
      "learning_rate": 7.689038683338796e-06,
      "loss": 0.0579,
      "step": 35330
    },
    {
      "epoch": 3.369331359863281e-05,
      "model_forward_time": 1.739126205444336,
      "step": 35330
    },
    {
      "dynamics_loss_avg": 0.05081454087048769,
      "epoch": 3.369331359863281e-05,
      "step": 35330
    },
    {
      "action_loss_avg": 0.0023189894971437754,
      "epoch": 3.369331359863281e-05,
      "step": 35330
    },
    {
      "epoch": 3.369331359863281e-05,
      "step": 35330,
      "training_step_time": 7.543362379074097
    },
    {
      "epoch": 3.369426727294922e-05,
      "model_forward_time": 1.73362398147583,
      "step": 35331
    },
    {
      "epoch": 3.369426727294922e-05,
      "step": 35331,
      "training_step_time": 8.763407230377197
    },
    {
      "epoch": 3.3695220947265626e-05,
      "model_forward_time": 2.858731985092163,
      "step": 35332
    },
    {
      "epoch": 3.3695220947265626e-05,
      "step": 35332,
      "training_step_time": 7.634979724884033
    },
    {
      "epoch": 3.369617462158203e-05,
      "model_forward_time": 1.7308154106140137,
      "step": 35333
    },
    {
      "epoch": 3.369617462158203e-05,
      "step": 35333,
      "training_step_time": 8.511452913284302
    },
    {
      "epoch": 3.369712829589844e-05,
      "model_forward_time": 1.7523212432861328,
      "step": 35334
    },
    {
      "epoch": 3.369712829589844e-05,
      "step": 35334,
      "training_step_time": 7.234712839126587
    },
    {
      "epoch": 3.3698081970214845e-05,
      "model_forward_time": 1.7524514198303223,
      "step": 35335
    },
    {
      "epoch": 3.3698081970214845e-05,
      "step": 35335,
      "training_step_time": 6.732544898986816
    },
    {
      "epoch": 3.369903564453125e-05,
      "model_forward_time": 1.7356510162353516,
      "step": 35336
    },
    {
      "epoch": 3.369903564453125e-05,
      "step": 35336,
      "training_step_time": 7.29419469833374
    },
    {
      "epoch": 3.369998931884766e-05,
      "model_forward_time": 1.7403578758239746,
      "step": 35337
    },
    {
      "epoch": 3.369998931884766e-05,
      "step": 35337,
      "training_step_time": 7.585272789001465
    },
    {
      "epoch": 3.3700942993164063e-05,
      "model_forward_time": 1.7442405223846436,
      "step": 35338
    },
    {
      "epoch": 3.3700942993164063e-05,
      "step": 35338,
      "training_step_time": 7.845746994018555
    },
    {
      "epoch": 3.370189666748047e-05,
      "model_forward_time": 1.7390387058258057,
      "step": 35339
    },
    {
      "epoch": 3.370189666748047e-05,
      "step": 35339,
      "training_step_time": 8.814072132110596
    },
    {
      "epoch": 3.370285034179688e-05,
      "grad_norm": 0.17723535001277924,
      "learning_rate": 7.687644549635808e-06,
      "loss": 0.0572,
      "step": 35340
    },
    {
      "epoch": 3.370285034179688e-05,
      "model_forward_time": 1.7360239028930664,
      "step": 35340
    },
    {
      "dynamics_loss_avg": 0.06375515852123499,
      "epoch": 3.370285034179688e-05,
      "step": 35340
    },
    {
      "action_loss_avg": 0.0024506734393071384,
      "epoch": 3.370285034179688e-05,
      "step": 35340
    },
    {
      "epoch": 3.370285034179688e-05,
      "step": 35340,
      "training_step_time": 8.506409645080566
    },
    {
      "epoch": 3.370380401611328e-05,
      "model_forward_time": 2.859574794769287,
      "step": 35341
    },
    {
      "epoch": 3.370380401611328e-05,
      "step": 35341,
      "training_step_time": 7.636311292648315
    },
    {
      "epoch": 3.3704757690429686e-05,
      "model_forward_time": 1.7344224452972412,
      "step": 35342
    },
    {
      "epoch": 3.3704757690429686e-05,
      "step": 35342,
      "training_step_time": 7.610031604766846
    },
    {
      "epoch": 3.37057113647461e-05,
      "model_forward_time": 1.3615195751190186,
      "step": 35343
    },
    {
      "epoch": 3.37057113647461e-05,
      "step": 35343,
      "training_step_time": 7.761022090911865
    },
    {
      "epoch": 3.37066650390625e-05,
      "model_forward_time": 1.763312578201294,
      "step": 35344
    },
    {
      "epoch": 3.37066650390625e-05,
      "step": 35344,
      "training_step_time": 6.780843734741211
    },
    {
      "epoch": 3.3707618713378905e-05,
      "model_forward_time": 1.736372470855713,
      "step": 35345
    },
    {
      "epoch": 3.3707618713378905e-05,
      "step": 35345,
      "training_step_time": 7.296778202056885
    },
    {
      "epoch": 3.370857238769531e-05,
      "model_forward_time": 1.735189437866211,
      "step": 35346
    },
    {
      "epoch": 3.370857238769531e-05,
      "step": 35346,
      "training_step_time": 8.053637027740479
    },
    {
      "epoch": 3.370952606201172e-05,
      "model_forward_time": 1.740159511566162,
      "step": 35347
    },
    {
      "epoch": 3.370952606201172e-05,
      "step": 35347,
      "training_step_time": 7.736879587173462
    },
    {
      "epoch": 3.3710479736328124e-05,
      "model_forward_time": 1.7323832511901855,
      "step": 35348
    },
    {
      "epoch": 3.3710479736328124e-05,
      "step": 35348,
      "training_step_time": 8.944994449615479
    },
    {
      "epoch": 3.371143341064453e-05,
      "model_forward_time": 1.7448959350585938,
      "step": 35349
    },
    {
      "epoch": 3.371143341064453e-05,
      "step": 35349,
      "training_step_time": 9.136987686157227
    },
    {
      "epoch": 3.371238708496094e-05,
      "grad_norm": 0.1459541916847229,
      "learning_rate": 7.686250122016053e-06,
      "loss": 0.0605,
      "step": 35350
    },
    {
      "epoch": 3.371238708496094e-05,
      "model_forward_time": 2.8654580116271973,
      "step": 35350
    },
    {
      "dynamics_loss_avg": 0.04529412724077701,
      "epoch": 3.371238708496094e-05,
      "step": 35350
    },
    {
      "action_loss_avg": 0.0009667685924796387,
      "epoch": 3.371238708496094e-05,
      "step": 35350
    },
    {
      "epoch": 3.371238708496094e-05,
      "step": 35350,
      "training_step_time": 9.76711368560791
    },
    {
      "epoch": 3.371334075927734e-05,
      "model_forward_time": 1.7321665287017822,
      "step": 35351
    },
    {
      "epoch": 3.371334075927734e-05,
      "step": 35351,
      "training_step_time": 7.649650812149048
    },
    {
      "epoch": 3.371429443359375e-05,
      "model_forward_time": 1.7340750694274902,
      "step": 35352
    },
    {
      "epoch": 3.371429443359375e-05,
      "step": 35352,
      "training_step_time": 7.738553524017334
    },
    {
      "epoch": 3.371524810791016e-05,
      "model_forward_time": 1.7848806381225586,
      "step": 35353
    },
    {
      "epoch": 3.371524810791016e-05,
      "step": 35353,
      "training_step_time": 7.2466630935668945
    },
    {
      "epoch": 3.371620178222656e-05,
      "model_forward_time": 1.7518596649169922,
      "step": 35354
    },
    {
      "epoch": 3.371620178222656e-05,
      "step": 35354,
      "training_step_time": 7.2351133823394775
    },
    {
      "epoch": 3.3717155456542966e-05,
      "model_forward_time": 1.7343158721923828,
      "step": 35355
    },
    {
      "epoch": 3.3717155456542966e-05,
      "step": 35355,
      "training_step_time": 8.628889560699463
    },
    {
      "epoch": 3.3718109130859377e-05,
      "model_forward_time": 1.7323338985443115,
      "step": 35356
    },
    {
      "epoch": 3.3718109130859377e-05,
      "step": 35356,
      "training_step_time": 8.451502561569214
    },
    {
      "epoch": 3.371906280517578e-05,
      "model_forward_time": 1.7608039379119873,
      "step": 35357
    },
    {
      "epoch": 3.371906280517578e-05,
      "step": 35357,
      "training_step_time": 8.608740091323853
    },
    {
      "epoch": 3.3720016479492185e-05,
      "model_forward_time": 1.7408316135406494,
      "step": 35358
    },
    {
      "epoch": 3.3720016479492185e-05,
      "step": 35358,
      "training_step_time": 8.576762914657593
    },
    {
      "epoch": 3.3720970153808595e-05,
      "model_forward_time": 2.8730149269104004,
      "step": 35359
    },
    {
      "epoch": 3.3720970153808595e-05,
      "step": 35359,
      "training_step_time": 8.37531304359436
    },
    {
      "epoch": 3.3721923828125e-05,
      "grad_norm": 0.12150872498750687,
      "learning_rate": 7.684855400632024e-06,
      "loss": 0.0648,
      "step": 35360
    },
    {
      "epoch": 3.3721923828125e-05,
      "model_forward_time": 1.7371063232421875,
      "step": 35360
    },
    {
      "dynamics_loss_avg": 0.05579789560288191,
      "epoch": 3.3721923828125e-05,
      "step": 35360
    },
    {
      "action_loss_avg": 0.001962824194924906,
      "epoch": 3.3721923828125e-05,
      "step": 35360
    },
    {
      "epoch": 3.3721923828125e-05,
      "step": 35360,
      "training_step_time": 7.720032453536987
    },
    {
      "epoch": 3.3722877502441403e-05,
      "model_forward_time": 1.7403721809387207,
      "step": 35361
    },
    {
      "epoch": 3.3722877502441403e-05,
      "step": 35361,
      "training_step_time": 7.752387762069702
    },
    {
      "epoch": 3.3723831176757814e-05,
      "model_forward_time": 1.7311220169067383,
      "step": 35362
    },
    {
      "epoch": 3.3723831176757814e-05,
      "step": 35362,
      "training_step_time": 7.574114084243774
    },
    {
      "epoch": 3.372478485107422e-05,
      "model_forward_time": 1.738332986831665,
      "step": 35363
    },
    {
      "epoch": 3.372478485107422e-05,
      "step": 35363,
      "training_step_time": 7.1803741455078125
    },
    {
      "epoch": 3.372573852539062e-05,
      "model_forward_time": 1.735905647277832,
      "step": 35364
    },
    {
      "epoch": 3.372573852539062e-05,
      "step": 35364,
      "training_step_time": 7.580498456954956
    },
    {
      "epoch": 3.372669219970703e-05,
      "model_forward_time": 1.7448368072509766,
      "step": 35365
    },
    {
      "epoch": 3.372669219970703e-05,
      "step": 35365,
      "training_step_time": 7.812837839126587
    },
    {
      "epoch": 3.372764587402344e-05,
      "model_forward_time": 1.7331547737121582,
      "step": 35366
    },
    {
      "epoch": 3.372764587402344e-05,
      "step": 35366,
      "training_step_time": 8.279829025268555
    },
    {
      "epoch": 3.372859954833984e-05,
      "model_forward_time": 1.7419319152832031,
      "step": 35367
    },
    {
      "epoch": 3.372859954833984e-05,
      "step": 35367,
      "training_step_time": 9.371048927307129
    },
    {
      "epoch": 3.372955322265625e-05,
      "model_forward_time": 1.7349529266357422,
      "step": 35368
    },
    {
      "epoch": 3.372955322265625e-05,
      "step": 35368,
      "training_step_time": 9.518763542175293
    },
    {
      "epoch": 3.3730506896972656e-05,
      "model_forward_time": 2.850292444229126,
      "step": 35369
    },
    {
      "epoch": 3.3730506896972656e-05,
      "step": 35369,
      "training_step_time": 7.631215810775757
    },
    {
      "epoch": 3.373146057128906e-05,
      "grad_norm": 0.13087844848632812,
      "learning_rate": 7.683460385636243e-06,
      "loss": 0.0603,
      "step": 35370
    },
    {
      "epoch": 3.373146057128906e-05,
      "model_forward_time": 1.7376677989959717,
      "step": 35370
    },
    {
      "dynamics_loss_avg": 0.06739353202283382,
      "epoch": 3.373146057128906e-05,
      "step": 35370
    },
    {
      "action_loss_avg": 0.0023928515147417784,
      "epoch": 3.373146057128906e-05,
      "step": 35370
    },
    {
      "epoch": 3.373146057128906e-05,
      "step": 35370,
      "training_step_time": 7.647433519363403
    },
    {
      "epoch": 3.373241424560547e-05,
      "model_forward_time": 1.737743854522705,
      "step": 35371
    },
    {
      "epoch": 3.373241424560547e-05,
      "step": 35371,
      "training_step_time": 7.727206468582153
    },
    {
      "epoch": 3.3733367919921875e-05,
      "model_forward_time": 1.7446644306182861,
      "step": 35372
    },
    {
      "epoch": 3.3733367919921875e-05,
      "step": 35372,
      "training_step_time": 7.086289882659912
    },
    {
      "epoch": 3.373432159423828e-05,
      "model_forward_time": 1.8377673625946045,
      "step": 35373
    },
    {
      "epoch": 3.373432159423828e-05,
      "step": 35373,
      "training_step_time": 7.345983028411865
    },
    {
      "epoch": 3.373527526855469e-05,
      "model_forward_time": 1.900646448135376,
      "step": 35374
    },
    {
      "epoch": 3.373527526855469e-05,
      "step": 35374,
      "training_step_time": 7.852262020111084
    },
    {
      "epoch": 3.3736228942871094e-05,
      "model_forward_time": 1.8514306545257568,
      "step": 35375
    },
    {
      "epoch": 3.3736228942871094e-05,
      "step": 35375,
      "training_step_time": 7.7016167640686035
    },
    {
      "epoch": 3.37371826171875e-05,
      "model_forward_time": 1.902313470840454,
      "step": 35376
    },
    {
      "epoch": 3.37371826171875e-05,
      "step": 35376,
      "training_step_time": 9.633714437484741
    },
    {
      "epoch": 3.373813629150391e-05,
      "model_forward_time": 1.8990981578826904,
      "step": 35377
    },
    {
      "epoch": 3.373813629150391e-05,
      "step": 35377,
      "training_step_time": 9.455793380737305
    },
    {
      "epoch": 3.373908996582031e-05,
      "model_forward_time": 2.9922988414764404,
      "step": 35378
    },
    {
      "epoch": 3.373908996582031e-05,
      "step": 35378,
      "training_step_time": 8.342378616333008
    },
    {
      "epoch": 3.374004364013672e-05,
      "model_forward_time": 1.7317802906036377,
      "step": 35379
    },
    {
      "epoch": 3.374004364013672e-05,
      "step": 35379,
      "training_step_time": 8.62474274635315
    },
    {
      "epoch": 3.374099731445313e-05,
      "grad_norm": 0.15012432634830475,
      "learning_rate": 7.68206507718127e-06,
      "loss": 0.0596,
      "step": 35380
    },
    {
      "epoch": 3.374099731445313e-05,
      "model_forward_time": 1.73116135597229,
      "step": 35380
    },
    {
      "dynamics_loss_avg": 0.06881837379187346,
      "epoch": 3.374099731445313e-05,
      "step": 35380
    },
    {
      "action_loss_avg": 0.0018212848633993417,
      "epoch": 3.374099731445313e-05,
      "step": 35380
    },
    {
      "epoch": 3.374099731445313e-05,
      "step": 35380,
      "training_step_time": 7.741734743118286
    },
    {
      "epoch": 3.374195098876953e-05,
      "model_forward_time": 1.723226547241211,
      "step": 35381
    },
    {
      "epoch": 3.374195098876953e-05,
      "step": 35381,
      "training_step_time": 7.535948276519775
    },
    {
      "epoch": 3.3742904663085935e-05,
      "model_forward_time": 1.7456612586975098,
      "step": 35382
    },
    {
      "epoch": 3.3742904663085935e-05,
      "step": 35382,
      "training_step_time": 7.373340129852295
    },
    {
      "epoch": 3.3743858337402346e-05,
      "model_forward_time": 1.8602445125579834,
      "step": 35383
    },
    {
      "epoch": 3.3743858337402346e-05,
      "step": 35383,
      "training_step_time": 7.896440505981445
    },
    {
      "epoch": 3.374481201171875e-05,
      "model_forward_time": 1.7304539680480957,
      "step": 35384
    },
    {
      "epoch": 3.374481201171875e-05,
      "step": 35384,
      "training_step_time": 8.496509313583374
    },
    {
      "epoch": 3.3745765686035154e-05,
      "model_forward_time": 1.7528438568115234,
      "step": 35385
    },
    {
      "epoch": 3.3745765686035154e-05,
      "step": 35385,
      "training_step_time": 7.7018723487854
    },
    {
      "epoch": 3.3746719360351565e-05,
      "model_forward_time": 1.7315573692321777,
      "step": 35386
    },
    {
      "epoch": 3.3746719360351565e-05,
      "step": 35386,
      "training_step_time": 8.55774974822998
    },
    {
      "epoch": 3.374767303466797e-05,
      "model_forward_time": 2.91149640083313,
      "step": 35387
    },
    {
      "epoch": 3.374767303466797e-05,
      "step": 35387,
      "training_step_time": 7.729482650756836
    },
    {
      "epoch": 3.374862670898437e-05,
      "model_forward_time": 1.742063283920288,
      "step": 35388
    },
    {
      "epoch": 3.374862670898437e-05,
      "step": 35388,
      "training_step_time": 7.667703151702881
    },
    {
      "epoch": 3.3749580383300784e-05,
      "model_forward_time": 1.7359850406646729,
      "step": 35389
    },
    {
      "epoch": 3.3749580383300784e-05,
      "step": 35389,
      "training_step_time": 7.667286396026611
    },
    {
      "epoch": 3.375053405761719e-05,
      "grad_norm": 0.2091720551252365,
      "learning_rate": 7.680669475419692e-06,
      "loss": 0.0595,
      "step": 35390
    },
    {
      "epoch": 3.375053405761719e-05,
      "model_forward_time": 1.7188429832458496,
      "step": 35390
    },
    {
      "dynamics_loss_avg": 0.061522997729480264,
      "epoch": 3.375053405761719e-05,
      "step": 35390
    },
    {
      "action_loss_avg": 0.002247061434900388,
      "epoch": 3.375053405761719e-05,
      "step": 35390
    },
    {
      "epoch": 3.375053405761719e-05,
      "step": 35390,
      "training_step_time": 7.535733461380005
    },
    {
      "epoch": 3.375148773193359e-05,
      "model_forward_time": 1.7440693378448486,
      "step": 35391
    },
    {
      "epoch": 3.375148773193359e-05,
      "step": 35391,
      "training_step_time": 7.275848388671875
    },
    {
      "epoch": 3.375244140625e-05,
      "model_forward_time": 1.8448660373687744,
      "step": 35392
    },
    {
      "epoch": 3.375244140625e-05,
      "step": 35392,
      "training_step_time": 7.793753385543823
    },
    {
      "epoch": 3.375339508056641e-05,
      "model_forward_time": 1.7296843528747559,
      "step": 35393
    },
    {
      "epoch": 3.375339508056641e-05,
      "step": 35393,
      "training_step_time": 7.605441570281982
    },
    {
      "epoch": 3.375434875488281e-05,
      "model_forward_time": 1.7653701305389404,
      "step": 35394
    },
    {
      "epoch": 3.375434875488281e-05,
      "step": 35394,
      "training_step_time": 8.301365375518799
    },
    {
      "epoch": 3.375530242919922e-05,
      "model_forward_time": 1.7343635559082031,
      "step": 35395
    },
    {
      "epoch": 3.375530242919922e-05,
      "step": 35395,
      "training_step_time": 8.68508791923523
    },
    {
      "epoch": 3.3756256103515626e-05,
      "model_forward_time": 2.947737216949463,
      "step": 35396
    },
    {
      "epoch": 3.3756256103515626e-05,
      "step": 35396,
      "training_step_time": 7.848360300064087
    },
    {
      "epoch": 3.375720977783203e-05,
      "model_forward_time": 1.7526874542236328,
      "step": 35397
    },
    {
      "epoch": 3.375720977783203e-05,
      "step": 35397,
      "training_step_time": 7.791054725646973
    },
    {
      "epoch": 3.375816345214844e-05,
      "model_forward_time": 1.7325944900512695,
      "step": 35398
    },
    {
      "epoch": 3.375816345214844e-05,
      "step": 35398,
      "training_step_time": 7.888534784317017
    },
    {
      "epoch": 3.3759117126464845e-05,
      "model_forward_time": 1.731285572052002,
      "step": 35399
    },
    {
      "epoch": 3.3759117126464845e-05,
      "step": 35399,
      "training_step_time": 7.704164266586304
    },
    {
      "epoch": 3.376007080078125e-05,
      "grad_norm": 0.24464258551597595,
      "learning_rate": 7.679273580504133e-06,
      "loss": 0.058,
      "step": 35400
    },
    {
      "epoch": 3.376007080078125e-05,
      "model_forward_time": 1.7196969985961914,
      "step": 35400
    },
    {
      "dynamics_loss_avg": 0.054039211478084326,
      "epoch": 3.376007080078125e-05,
      "step": 35400
    },
    {
      "action_loss_avg": 0.001902575881103985,
      "epoch": 3.376007080078125e-05,
      "step": 35400
    },
    {
      "epoch": 3.376007080078125e-05,
      "step": 35400,
      "training_step_time": 7.0023767948150635
    },
    {
      "epoch": 3.376102447509766e-05,
      "model_forward_time": 1.8495824337005615,
      "step": 35401
    },
    {
      "epoch": 3.376102447509766e-05,
      "step": 35401,
      "training_step_time": 8.45209527015686
    },
    {
      "epoch": 3.3761978149414063e-05,
      "model_forward_time": 1.3606789112091064,
      "step": 35402
    },
    {
      "epoch": 3.3761978149414063e-05,
      "step": 35402,
      "training_step_time": 8.191399574279785
    },
    {
      "epoch": 3.376293182373047e-05,
      "model_forward_time": 1.7596626281738281,
      "step": 35403
    },
    {
      "epoch": 3.376293182373047e-05,
      "step": 35403,
      "training_step_time": 9.37749981880188
    },
    {
      "epoch": 3.376388549804688e-05,
      "model_forward_time": 1.7392425537109375,
      "step": 35404
    },
    {
      "epoch": 3.376388549804688e-05,
      "step": 35404,
      "training_step_time": 9.861302614212036
    },
    {
      "epoch": 3.376483917236328e-05,
      "model_forward_time": 2.8109817504882812,
      "step": 35405
    },
    {
      "epoch": 3.376483917236328e-05,
      "step": 35405,
      "training_step_time": 8.612021684646606
    },
    {
      "epoch": 3.3765792846679686e-05,
      "model_forward_time": 1.7452774047851562,
      "step": 35406
    },
    {
      "epoch": 3.3765792846679686e-05,
      "step": 35406,
      "training_step_time": 8.754893779754639
    },
    {
      "epoch": 3.376674652099609e-05,
      "model_forward_time": 1.7342338562011719,
      "step": 35407
    },
    {
      "epoch": 3.376674652099609e-05,
      "step": 35407,
      "training_step_time": 7.739539861679077
    },
    {
      "epoch": 3.37677001953125e-05,
      "model_forward_time": 1.7333312034606934,
      "step": 35408
    },
    {
      "epoch": 3.37677001953125e-05,
      "step": 35408,
      "training_step_time": 7.798592805862427
    },
    {
      "epoch": 3.3768653869628905e-05,
      "model_forward_time": 1.7383170127868652,
      "step": 35409
    },
    {
      "epoch": 3.3768653869628905e-05,
      "step": 35409,
      "training_step_time": 7.412489414215088
    },
    {
      "epoch": 3.376960754394531e-05,
      "grad_norm": 0.1747157722711563,
      "learning_rate": 7.67787739258724e-06,
      "loss": 0.0591,
      "step": 35410
    },
    {
      "epoch": 3.376960754394531e-05,
      "model_forward_time": 1.8220973014831543,
      "step": 35410
    },
    {
      "dynamics_loss_avg": 0.057584467343986034,
      "epoch": 3.376960754394531e-05,
      "step": 35410
    },
    {
      "action_loss_avg": 0.0017641161277424545,
      "epoch": 3.376960754394531e-05,
      "step": 35410
    },
    {
      "epoch": 3.376960754394531e-05,
      "step": 35410,
      "training_step_time": 7.750656366348267
    },
    {
      "epoch": 3.377056121826172e-05,
      "model_forward_time": 1.7281928062438965,
      "step": 35411
    },
    {
      "epoch": 3.377056121826172e-05,
      "step": 35411,
      "training_step_time": 8.352158308029175
    },
    {
      "epoch": 3.3771514892578124e-05,
      "model_forward_time": 1.7366559505462646,
      "step": 35412
    },
    {
      "epoch": 3.3771514892578124e-05,
      "step": 35412,
      "training_step_time": 8.548149585723877
    },
    {
      "epoch": 3.377246856689453e-05,
      "model_forward_time": 1.352064609527588,
      "step": 35413
    },
    {
      "epoch": 3.377246856689453e-05,
      "step": 35413,
      "training_step_time": 7.765350341796875
    },
    {
      "epoch": 3.377342224121094e-05,
      "model_forward_time": 1.3754913806915283,
      "step": 35414
    },
    {
      "epoch": 3.377342224121094e-05,
      "step": 35414,
      "training_step_time": 7.746940612792969
    },
    {
      "epoch": 3.377437591552734e-05,
      "model_forward_time": 2.923322916030884,
      "step": 35415
    },
    {
      "epoch": 3.377437591552734e-05,
      "step": 35415,
      "training_step_time": 7.719792366027832
    },
    {
      "epoch": 3.377532958984375e-05,
      "model_forward_time": 1.731177806854248,
      "step": 35416
    },
    {
      "epoch": 3.377532958984375e-05,
      "step": 35416,
      "training_step_time": 7.6761698722839355
    },
    {
      "epoch": 3.377628326416016e-05,
      "model_forward_time": 1.7464187145233154,
      "step": 35417
    },
    {
      "epoch": 3.377628326416016e-05,
      "step": 35417,
      "training_step_time": 8.282501697540283
    },
    {
      "epoch": 3.377723693847656e-05,
      "model_forward_time": 1.7336647510528564,
      "step": 35418
    },
    {
      "epoch": 3.377723693847656e-05,
      "step": 35418,
      "training_step_time": 7.632367849349976
    },
    {
      "epoch": 3.3778190612792966e-05,
      "model_forward_time": 1.8078560829162598,
      "step": 35419
    },
    {
      "epoch": 3.3778190612792966e-05,
      "step": 35419,
      "training_step_time": 7.753230810165405
    },
    {
      "epoch": 3.377914428710938e-05,
      "grad_norm": 0.0997275561094284,
      "learning_rate": 7.676480911821705e-06,
      "loss": 0.0589,
      "step": 35420
    },
    {
      "epoch": 3.377914428710938e-05,
      "model_forward_time": 1.7380950450897217,
      "step": 35420
    },
    {
      "dynamics_loss_avg": 0.0556623138487339,
      "epoch": 3.377914428710938e-05,
      "step": 35420
    },
    {
      "action_loss_avg": 0.001683807943481952,
      "epoch": 3.377914428710938e-05,
      "step": 35420
    },
    {
      "epoch": 3.377914428710938e-05,
      "step": 35420,
      "training_step_time": 7.669020414352417
    },
    {
      "epoch": 3.378009796142578e-05,
      "model_forward_time": 1.7283494472503662,
      "step": 35421
    },
    {
      "epoch": 3.378009796142578e-05,
      "step": 35421,
      "training_step_time": 8.379682779312134
    },
    {
      "epoch": 3.3781051635742185e-05,
      "model_forward_time": 1.364499568939209,
      "step": 35422
    },
    {
      "epoch": 3.3781051635742185e-05,
      "step": 35422,
      "training_step_time": 7.674779176712036
    },
    {
      "epoch": 3.3782005310058595e-05,
      "model_forward_time": 1.7421975135803223,
      "step": 35423
    },
    {
      "epoch": 3.3782005310058595e-05,
      "step": 35423,
      "training_step_time": 7.621690273284912
    },
    {
      "epoch": 3.3782958984375e-05,
      "model_forward_time": 2.8719184398651123,
      "step": 35424
    },
    {
      "epoch": 3.3782958984375e-05,
      "step": 35424,
      "training_step_time": 7.778838396072388
    },
    {
      "epoch": 3.3783912658691404e-05,
      "model_forward_time": 1.7329437732696533,
      "step": 35425
    },
    {
      "epoch": 3.3783912658691404e-05,
      "step": 35425,
      "training_step_time": 7.574113607406616
    },
    {
      "epoch": 3.3784866333007814e-05,
      "model_forward_time": 1.748748540878296,
      "step": 35426
    },
    {
      "epoch": 3.3784866333007814e-05,
      "step": 35426,
      "training_step_time": 8.171910762786865
    },
    {
      "epoch": 3.378582000732422e-05,
      "model_forward_time": 1.7355499267578125,
      "step": 35427
    },
    {
      "epoch": 3.378582000732422e-05,
      "step": 35427,
      "training_step_time": 7.545131206512451
    },
    {
      "epoch": 3.378677368164062e-05,
      "model_forward_time": 1.7357561588287354,
      "step": 35428
    },
    {
      "epoch": 3.378677368164062e-05,
      "step": 35428,
      "training_step_time": 7.752322673797607
    },
    {
      "epoch": 3.378772735595703e-05,
      "model_forward_time": 1.870417594909668,
      "step": 35429
    },
    {
      "epoch": 3.378772735595703e-05,
      "step": 35429,
      "training_step_time": 7.645824193954468
    },
    {
      "epoch": 3.378868103027344e-05,
      "grad_norm": 0.13684412837028503,
      "learning_rate": 7.675084138360238e-06,
      "loss": 0.0592,
      "step": 35430
    },
    {
      "epoch": 3.378868103027344e-05,
      "model_forward_time": 1.7381112575531006,
      "step": 35430
    },
    {
      "dynamics_loss_avg": 0.06699814517050981,
      "epoch": 3.378868103027344e-05,
      "step": 35430
    },
    {
      "action_loss_avg": 0.0019745071796933187,
      "epoch": 3.378868103027344e-05,
      "step": 35430
    },
    {
      "epoch": 3.378868103027344e-05,
      "step": 35430,
      "training_step_time": 9.154950857162476
    },
    {
      "epoch": 3.378963470458984e-05,
      "model_forward_time": 1.7352917194366455,
      "step": 35431
    },
    {
      "epoch": 3.378963470458984e-05,
      "step": 35431,
      "training_step_time": 7.669494390487671
    },
    {
      "epoch": 3.379058837890625e-05,
      "model_forward_time": 1.720024585723877,
      "step": 35432
    },
    {
      "epoch": 3.379058837890625e-05,
      "step": 35432,
      "training_step_time": 7.607219219207764
    },
    {
      "epoch": 3.3791542053222656e-05,
      "model_forward_time": 2.822718858718872,
      "step": 35433
    },
    {
      "epoch": 3.3791542053222656e-05,
      "step": 35433,
      "training_step_time": 9.504827737808228
    },
    {
      "epoch": 3.379249572753906e-05,
      "model_forward_time": 1.7202677726745605,
      "step": 35434
    },
    {
      "epoch": 3.379249572753906e-05,
      "step": 35434,
      "training_step_time": 7.674996376037598
    },
    {
      "epoch": 3.379344940185547e-05,
      "model_forward_time": 1.7216217517852783,
      "step": 35435
    },
    {
      "epoch": 3.379344940185547e-05,
      "step": 35435,
      "training_step_time": 7.703089714050293
    },
    {
      "epoch": 3.3794403076171875e-05,
      "model_forward_time": 1.3536911010742188,
      "step": 35436
    },
    {
      "epoch": 3.3794403076171875e-05,
      "step": 35436,
      "training_step_time": 7.911924362182617
    },
    {
      "epoch": 3.379535675048828e-05,
      "model_forward_time": 1.7307355403900146,
      "step": 35437
    },
    {
      "epoch": 3.379535675048828e-05,
      "step": 35437,
      "training_step_time": 7.780220031738281
    },
    {
      "epoch": 3.379631042480469e-05,
      "model_forward_time": 1.7463948726654053,
      "step": 35438
    },
    {
      "epoch": 3.379631042480469e-05,
      "step": 35438,
      "training_step_time": 7.87165093421936
    },
    {
      "epoch": 3.3797264099121094e-05,
      "model_forward_time": 1.7232162952423096,
      "step": 35439
    },
    {
      "epoch": 3.3797264099121094e-05,
      "step": 35439,
      "training_step_time": 7.653208494186401
    },
    {
      "epoch": 3.37982177734375e-05,
      "grad_norm": 0.1500536948442459,
      "learning_rate": 7.673687072355594e-06,
      "loss": 0.0594,
      "step": 35440
    },
    {
      "epoch": 3.37982177734375e-05,
      "model_forward_time": 1.7182273864746094,
      "step": 35440
    },
    {
      "dynamics_loss_avg": 0.04750806521624327,
      "epoch": 3.37982177734375e-05,
      "step": 35440
    },
    {
      "action_loss_avg": 0.0017777064931578934,
      "epoch": 3.37982177734375e-05,
      "step": 35440
    },
    {
      "epoch": 3.37982177734375e-05,
      "step": 35440,
      "training_step_time": 8.572431564331055
    },
    {
      "epoch": 3.379917144775391e-05,
      "model_forward_time": 1.7188153266906738,
      "step": 35441
    },
    {
      "epoch": 3.379917144775391e-05,
      "step": 35441,
      "training_step_time": 8.933828592300415
    },
    {
      "epoch": 3.380012512207031e-05,
      "model_forward_time": 1.350212574005127,
      "step": 35442
    },
    {
      "epoch": 3.380012512207031e-05,
      "step": 35442,
      "training_step_time": 7.753211259841919
    },
    {
      "epoch": 3.380107879638672e-05,
      "model_forward_time": 2.8115854263305664,
      "step": 35443
    },
    {
      "epoch": 3.380107879638672e-05,
      "step": 35443,
      "training_step_time": 7.791408538818359
    },
    {
      "epoch": 3.380203247070313e-05,
      "model_forward_time": 1.718545913696289,
      "step": 35444
    },
    {
      "epoch": 3.380203247070313e-05,
      "step": 35444,
      "training_step_time": 7.698263645172119
    },
    {
      "epoch": 3.380298614501953e-05,
      "model_forward_time": 1.722668170928955,
      "step": 35445
    },
    {
      "epoch": 3.380298614501953e-05,
      "step": 35445,
      "training_step_time": 7.137693405151367
    },
    {
      "epoch": 3.3803939819335936e-05,
      "model_forward_time": 1.7208189964294434,
      "step": 35446
    },
    {
      "epoch": 3.3803939819335936e-05,
      "step": 35446,
      "training_step_time": 7.889750003814697
    },
    {
      "epoch": 3.3804893493652346e-05,
      "model_forward_time": 1.716627597808838,
      "step": 35447
    },
    {
      "epoch": 3.3804893493652346e-05,
      "step": 35447,
      "training_step_time": 7.798799753189087
    },
    {
      "epoch": 3.380584716796875e-05,
      "model_forward_time": 1.732764482498169,
      "step": 35448
    },
    {
      "epoch": 3.380584716796875e-05,
      "step": 35448,
      "training_step_time": 7.624536037445068
    },
    {
      "epoch": 3.3806800842285154e-05,
      "model_forward_time": 1.7206149101257324,
      "step": 35449
    },
    {
      "epoch": 3.3806800842285154e-05,
      "step": 35449,
      "training_step_time": 9.367833614349365
    },
    {
      "epoch": 3.3807754516601565e-05,
      "grad_norm": 0.10958795994520187,
      "learning_rate": 7.67228971396055e-06,
      "loss": 0.0592,
      "step": 35450
    },
    {
      "epoch": 3.3807754516601565e-05,
      "model_forward_time": 1.7244141101837158,
      "step": 35450
    },
    {
      "dynamics_loss_avg": 0.05360466223210096,
      "epoch": 3.3807754516601565e-05,
      "step": 35450
    },
    {
      "action_loss_avg": 0.0013484388910001144,
      "epoch": 3.3807754516601565e-05,
      "step": 35450
    },
    {
      "epoch": 3.3807754516601565e-05,
      "step": 35450,
      "training_step_time": 9.380723237991333
    },
    {
      "epoch": 3.380870819091797e-05,
      "model_forward_time": 1.729574203491211,
      "step": 35451
    },
    {
      "epoch": 3.380870819091797e-05,
      "step": 35451,
      "training_step_time": 8.411602973937988
    },
    {
      "epoch": 3.380966186523437e-05,
      "model_forward_time": 2.872385263442993,
      "step": 35452
    },
    {
      "epoch": 3.380966186523437e-05,
      "step": 35452,
      "training_step_time": 8.608823776245117
    },
    {
      "epoch": 3.3810615539550784e-05,
      "model_forward_time": 1.734459638595581,
      "step": 35453
    },
    {
      "epoch": 3.3810615539550784e-05,
      "step": 35453,
      "training_step_time": 7.574911594390869
    },
    {
      "epoch": 3.381156921386719e-05,
      "model_forward_time": 1.7237482070922852,
      "step": 35454
    },
    {
      "epoch": 3.381156921386719e-05,
      "step": 35454,
      "training_step_time": 7.673577785491943
    },
    {
      "epoch": 3.381252288818359e-05,
      "model_forward_time": 1.7427589893341064,
      "step": 35455
    },
    {
      "epoch": 3.381252288818359e-05,
      "step": 35455,
      "training_step_time": 7.858304977416992
    },
    {
      "epoch": 3.38134765625e-05,
      "model_forward_time": 1.729311227798462,
      "step": 35456
    },
    {
      "epoch": 3.38134765625e-05,
      "step": 35456,
      "training_step_time": 7.5797340869903564
    },
    {
      "epoch": 3.381443023681641e-05,
      "model_forward_time": 1.7293453216552734,
      "step": 35457
    },
    {
      "epoch": 3.381443023681641e-05,
      "step": 35457,
      "training_step_time": 8.34282922744751
    },
    {
      "epoch": 3.381538391113281e-05,
      "model_forward_time": 1.8703835010528564,
      "step": 35458
    },
    {
      "epoch": 3.381538391113281e-05,
      "step": 35458,
      "training_step_time": 7.615246534347534
    },
    {
      "epoch": 3.381633758544922e-05,
      "model_forward_time": 1.9198970794677734,
      "step": 35459
    },
    {
      "epoch": 3.381633758544922e-05,
      "step": 35459,
      "training_step_time": 8.62727403640747
    },
    {
      "epoch": 3.3817291259765626e-05,
      "grad_norm": 0.13228227198123932,
      "learning_rate": 7.670892063327921e-06,
      "loss": 0.057,
      "step": 35460
    },
    {
      "epoch": 3.3817291259765626e-05,
      "model_forward_time": 1.9117136001586914,
      "step": 35460
    },
    {
      "dynamics_loss_avg": 0.05391899645328522,
      "epoch": 3.3817291259765626e-05,
      "step": 35460
    },
    {
      "action_loss_avg": 0.0012539442221168428,
      "epoch": 3.3817291259765626e-05,
      "step": 35460
    },
    {
      "epoch": 3.3817291259765626e-05,
      "step": 35460,
      "training_step_time": 8.499692678451538
    },
    {
      "epoch": 3.381824493408203e-05,
      "model_forward_time": 2.9951698780059814,
      "step": 35461
    },
    {
      "epoch": 3.381824493408203e-05,
      "step": 35461,
      "training_step_time": 7.7758262157440186
    },
    {
      "epoch": 3.381919860839844e-05,
      "model_forward_time": 1.9098865985870361,
      "step": 35462
    },
    {
      "epoch": 3.381919860839844e-05,
      "step": 35462,
      "training_step_time": 7.640931844711304
    },
    {
      "epoch": 3.3820152282714845e-05,
      "model_forward_time": 1.924614667892456,
      "step": 35463
    },
    {
      "epoch": 3.3820152282714845e-05,
      "step": 35463,
      "training_step_time": 7.7344465255737305
    },
    {
      "epoch": 3.382110595703125e-05,
      "model_forward_time": 1.8984360694885254,
      "step": 35464
    },
    {
      "epoch": 3.382110595703125e-05,
      "step": 35464,
      "training_step_time": 7.8472254276275635
    },
    {
      "epoch": 3.382205963134766e-05,
      "model_forward_time": 1.9088959693908691,
      "step": 35465
    },
    {
      "epoch": 3.382205963134766e-05,
      "step": 35465,
      "training_step_time": 7.668281555175781
    },
    {
      "epoch": 3.3823013305664064e-05,
      "model_forward_time": 1.922084093093872,
      "step": 35466
    },
    {
      "epoch": 3.3823013305664064e-05,
      "step": 35466,
      "training_step_time": 8.623161792755127
    },
    {
      "epoch": 3.382396697998047e-05,
      "model_forward_time": 1.9063847064971924,
      "step": 35467
    },
    {
      "epoch": 3.382396697998047e-05,
      "step": 35467,
      "training_step_time": 8.376201629638672
    },
    {
      "epoch": 3.382492065429688e-05,
      "model_forward_time": 1.9135260581970215,
      "step": 35468
    },
    {
      "epoch": 3.382492065429688e-05,
      "step": 35468,
      "training_step_time": 8.89710783958435
    },
    {
      "epoch": 3.382587432861328e-05,
      "model_forward_time": 1.8887693881988525,
      "step": 35469
    },
    {
      "epoch": 3.382587432861328e-05,
      "step": 35469,
      "training_step_time": 9.5885169506073
    },
    {
      "epoch": 3.3826828002929686e-05,
      "grad_norm": 0.11992967128753662,
      "learning_rate": 7.669494120610551e-06,
      "loss": 0.0612,
      "step": 35470
    },
    {
      "epoch": 3.3826828002929686e-05,
      "model_forward_time": 1.909208059310913,
      "step": 35470
    },
    {
      "dynamics_loss_avg": 0.05225041061639786,
      "epoch": 3.3826828002929686e-05,
      "step": 35470
    },
    {
      "action_loss_avg": 0.0015414814057294278,
      "epoch": 3.3826828002929686e-05,
      "step": 35470
    },
    {
      "epoch": 3.3826828002929686e-05,
      "step": 35470,
      "training_step_time": 9.760305404663086
    },
    {
      "epoch": 3.382778167724609e-05,
      "model_forward_time": 1.8947150707244873,
      "step": 35471
    },
    {
      "epoch": 3.382778167724609e-05,
      "step": 35471,
      "training_step_time": 7.65422511100769
    },
    {
      "epoch": 3.38287353515625e-05,
      "model_forward_time": 1.9143681526184082,
      "step": 35472
    },
    {
      "epoch": 3.38287353515625e-05,
      "step": 35472,
      "training_step_time": 7.6977219581604
    },
    {
      "epoch": 3.3829689025878905e-05,
      "model_forward_time": 1.91532301902771,
      "step": 35473
    },
    {
      "epoch": 3.3829689025878905e-05,
      "step": 35473,
      "training_step_time": 8.719397068023682
    },
    {
      "epoch": 3.383064270019531e-05,
      "model_forward_time": 1.924448013305664,
      "step": 35474
    },
    {
      "epoch": 3.383064270019531e-05,
      "step": 35474,
      "training_step_time": 8.797182559967041
    },
    {
      "epoch": 3.383159637451172e-05,
      "model_forward_time": 1.9092903137207031,
      "step": 35475
    },
    {
      "epoch": 3.383159637451172e-05,
      "step": 35475,
      "training_step_time": 7.340924024581909
    },
    {
      "epoch": 3.3832550048828124e-05,
      "model_forward_time": 1.5109333992004395,
      "step": 35476
    },
    {
      "epoch": 3.3832550048828124e-05,
      "step": 35476,
      "training_step_time": 8.489968299865723
    },
    {
      "epoch": 3.383350372314453e-05,
      "model_forward_time": 1.8999724388122559,
      "step": 35477
    },
    {
      "epoch": 3.383350372314453e-05,
      "step": 35477,
      "training_step_time": 7.653326749801636
    },
    {
      "epoch": 3.383445739746094e-05,
      "model_forward_time": 1.893646478652954,
      "step": 35478
    },
    {
      "epoch": 3.383445739746094e-05,
      "step": 35478,
      "training_step_time": 8.757573366165161
    },
    {
      "epoch": 3.383541107177734e-05,
      "model_forward_time": 1.9283628463745117,
      "step": 35479
    },
    {
      "epoch": 3.383541107177734e-05,
      "step": 35479,
      "training_step_time": 7.629928827285767
    },
    {
      "epoch": 3.383636474609375e-05,
      "grad_norm": 0.11143358796834946,
      "learning_rate": 7.66809588596132e-06,
      "loss": 0.0653,
      "step": 35480
    },
    {
      "epoch": 3.383636474609375e-05,
      "model_forward_time": 3.026116371154785,
      "step": 35480
    },
    {
      "dynamics_loss_avg": 0.0642749972641468,
      "epoch": 3.383636474609375e-05,
      "step": 35480
    },
    {
      "action_loss_avg": 0.001492662396049127,
      "epoch": 3.383636474609375e-05,
      "step": 35480
    },
    {
      "epoch": 3.383636474609375e-05,
      "step": 35480,
      "training_step_time": 8.43541145324707
    },
    {
      "epoch": 3.383731842041016e-05,
      "model_forward_time": 1.8930675983428955,
      "step": 35481
    },
    {
      "epoch": 3.383731842041016e-05,
      "step": 35481,
      "training_step_time": 7.657583713531494
    },
    {
      "epoch": 3.383827209472656e-05,
      "model_forward_time": 1.5419583320617676,
      "step": 35482
    },
    {
      "epoch": 3.383827209472656e-05,
      "step": 35482,
      "training_step_time": 7.044583797454834
    },
    {
      "epoch": 3.3839225769042966e-05,
      "model_forward_time": 1.5184743404388428,
      "step": 35483
    },
    {
      "epoch": 3.3839225769042966e-05,
      "step": 35483,
      "training_step_time": 7.9229395389556885
    },
    {
      "epoch": 3.384017944335938e-05,
      "model_forward_time": 1.9257771968841553,
      "step": 35484
    },
    {
      "epoch": 3.384017944335938e-05,
      "step": 35484,
      "training_step_time": 7.683789014816284
    },
    {
      "epoch": 3.384113311767578e-05,
      "model_forward_time": 1.9144439697265625,
      "step": 35485
    },
    {
      "epoch": 3.384113311767578e-05,
      "step": 35485,
      "training_step_time": 8.70749044418335
    },
    {
      "epoch": 3.3842086791992185e-05,
      "model_forward_time": 1.8733372688293457,
      "step": 35486
    },
    {
      "epoch": 3.3842086791992185e-05,
      "step": 35486,
      "training_step_time": 8.26608157157898
    },
    {
      "epoch": 3.3843040466308596e-05,
      "model_forward_time": 1.881669282913208,
      "step": 35487
    },
    {
      "epoch": 3.3843040466308596e-05,
      "step": 35487,
      "training_step_time": 7.7905662059783936
    },
    {
      "epoch": 3.3843994140625e-05,
      "model_forward_time": 1.5484991073608398,
      "step": 35488
    },
    {
      "epoch": 3.3843994140625e-05,
      "step": 35488,
      "training_step_time": 8.591308355331421
    },
    {
      "epoch": 3.3844947814941404e-05,
      "model_forward_time": 3.0602025985717773,
      "step": 35489
    },
    {
      "epoch": 3.3844947814941404e-05,
      "step": 35489,
      "training_step_time": 8.576281785964966
    },
    {
      "epoch": 3.3845901489257814e-05,
      "grad_norm": 0.10673343390226364,
      "learning_rate": 7.666697359533134e-06,
      "loss": 0.0584,
      "step": 35490
    },
    {
      "epoch": 3.3845901489257814e-05,
      "model_forward_time": 1.8711795806884766,
      "step": 35490
    },
    {
      "dynamics_loss_avg": 0.051162687875330445,
      "epoch": 3.3845901489257814e-05,
      "step": 35490
    },
    {
      "action_loss_avg": 0.0016462214814964681,
      "epoch": 3.3845901489257814e-05,
      "step": 35490
    },
    {
      "epoch": 3.3845901489257814e-05,
      "step": 35490,
      "training_step_time": 8.342820882797241
    },
    {
      "epoch": 3.384685516357422e-05,
      "model_forward_time": 1.896726369857788,
      "step": 35491
    },
    {
      "epoch": 3.384685516357422e-05,
      "step": 35491,
      "training_step_time": 7.751535177230835
    },
    {
      "epoch": 3.384780883789062e-05,
      "model_forward_time": 1.9045836925506592,
      "step": 35492
    },
    {
      "epoch": 3.384780883789062e-05,
      "step": 35492,
      "training_step_time": 7.931913375854492
    },
    {
      "epoch": 3.384876251220703e-05,
      "model_forward_time": 1.8819191455841064,
      "step": 35493
    },
    {
      "epoch": 3.384876251220703e-05,
      "step": 35493,
      "training_step_time": 7.886884927749634
    },
    {
      "epoch": 3.384971618652344e-05,
      "model_forward_time": 1.8950207233428955,
      "step": 35494
    },
    {
      "epoch": 3.384971618652344e-05,
      "step": 35494,
      "training_step_time": 7.7426745891571045
    },
    {
      "epoch": 3.385066986083984e-05,
      "model_forward_time": 1.9203968048095703,
      "step": 35495
    },
    {
      "epoch": 3.385066986083984e-05,
      "step": 35495,
      "training_step_time": 8.39574646949768
    },
    {
      "epoch": 3.385162353515625e-05,
      "model_forward_time": 1.9098730087280273,
      "step": 35496
    },
    {
      "epoch": 3.385162353515625e-05,
      "step": 35496,
      "training_step_time": 7.773466110229492
    },
    {
      "epoch": 3.3852577209472656e-05,
      "model_forward_time": 1.523911952972412,
      "step": 35497
    },
    {
      "epoch": 3.3852577209472656e-05,
      "step": 35497,
      "training_step_time": 9.308745861053467
    },
    {
      "epoch": 3.385353088378906e-05,
      "model_forward_time": 1.8912522792816162,
      "step": 35498
    },
    {
      "epoch": 3.385353088378906e-05,
      "step": 35498,
      "training_step_time": 7.790684700012207
    },
    {
      "epoch": 3.385448455810547e-05,
      "model_forward_time": 2.533804178237915,
      "step": 35499
    },
    {
      "epoch": 3.385448455810547e-05,
      "step": 35499,
      "training_step_time": 7.696483612060547
    },
    {
      "epoch": 3.3855438232421875e-05,
      "grad_norm": 0.13001316785812378,
      "learning_rate": 7.665298541478931e-06,
      "loss": 0.0596,
      "step": 35500
    },
    {
      "epoch": 3.3855438232421875e-05,
      "model_forward_time": 1.9172954559326172,
      "step": 35500
    },
    {
      "dynamics_loss_avg": 0.04793879985809326,
      "epoch": 3.3855438232421875e-05,
      "step": 35500
    },
    {
      "action_loss_avg": 0.0017644824227318167,
      "epoch": 3.3855438232421875e-05,
      "step": 35500
    },
    {
      "epoch": 3.3855438232421875e-05,
      "step": 35500,
      "training_step_time": 7.760866165161133
    },
    {
      "epoch": 3.385639190673828e-05,
      "model_forward_time": 1.8971998691558838,
      "step": 35501
    },
    {
      "epoch": 3.385639190673828e-05,
      "step": 35501,
      "training_step_time": 8.038605213165283
    },
    {
      "epoch": 3.385734558105469e-05,
      "model_forward_time": 1.9338786602020264,
      "step": 35502
    },
    {
      "epoch": 3.385734558105469e-05,
      "step": 35502,
      "training_step_time": 7.817668199539185
    },
    {
      "epoch": 3.3858299255371094e-05,
      "model_forward_time": 1.9154531955718994,
      "step": 35503
    },
    {
      "epoch": 3.3858299255371094e-05,
      "step": 35503,
      "training_step_time": 8.927276372909546
    },
    {
      "epoch": 3.38592529296875e-05,
      "model_forward_time": 1.7341184616088867,
      "step": 35504
    },
    {
      "epoch": 3.38592529296875e-05,
      "step": 35504,
      "training_step_time": 7.739973783493042
    },
    {
      "epoch": 3.386020660400391e-05,
      "model_forward_time": 1.7414329051971436,
      "step": 35505
    },
    {
      "epoch": 3.386020660400391e-05,
      "step": 35505,
      "training_step_time": 7.739481687545776
    },
    {
      "epoch": 3.386116027832031e-05,
      "model_forward_time": 1.367025375366211,
      "step": 35506
    },
    {
      "epoch": 3.386116027832031e-05,
      "step": 35506,
      "training_step_time": 7.658729314804077
    },
    {
      "epoch": 3.386211395263672e-05,
      "model_forward_time": 1.7260944843292236,
      "step": 35507
    },
    {
      "epoch": 3.386211395263672e-05,
      "step": 35507,
      "training_step_time": 7.765000104904175
    },
    {
      "epoch": 3.386306762695313e-05,
      "model_forward_time": 2.8492844104766846,
      "step": 35508
    },
    {
      "epoch": 3.386306762695313e-05,
      "step": 35508,
      "training_step_time": 8.215850353240967
    },
    {
      "epoch": 3.386402130126953e-05,
      "model_forward_time": 1.7410237789154053,
      "step": 35509
    },
    {
      "epoch": 3.386402130126953e-05,
      "step": 35509,
      "training_step_time": 7.696506023406982
    },
    {
      "epoch": 3.3864974975585936e-05,
      "grad_norm": 0.12073315680027008,
      "learning_rate": 7.663899431951689e-06,
      "loss": 0.0591,
      "step": 35510
    },
    {
      "epoch": 3.3864974975585936e-05,
      "model_forward_time": 1.732816457748413,
      "step": 35510
    },
    {
      "dynamics_loss_avg": 0.039795775152742865,
      "epoch": 3.3864974975585936e-05,
      "step": 35510
    },
    {
      "action_loss_avg": 0.0017526464711409063,
      "epoch": 3.3864974975585936e-05,
      "step": 35510
    },
    {
      "epoch": 3.3864974975585936e-05,
      "step": 35510,
      "training_step_time": 8.952158212661743
    },
    {
      "epoch": 3.3865928649902346e-05,
      "model_forward_time": 1.788045883178711,
      "step": 35511
    },
    {
      "epoch": 3.3865928649902346e-05,
      "step": 35511,
      "training_step_time": 7.698671817779541
    },
    {
      "epoch": 3.386688232421875e-05,
      "model_forward_time": 1.7366883754730225,
      "step": 35512
    },
    {
      "epoch": 3.386688232421875e-05,
      "step": 35512,
      "training_step_time": 7.698746919631958
    },
    {
      "epoch": 3.3867835998535154e-05,
      "model_forward_time": 1.7423863410949707,
      "step": 35513
    },
    {
      "epoch": 3.3867835998535154e-05,
      "step": 35513,
      "training_step_time": 7.624999284744263
    },
    {
      "epoch": 3.3868789672851565e-05,
      "model_forward_time": 1.7414155006408691,
      "step": 35514
    },
    {
      "epoch": 3.3868789672851565e-05,
      "step": 35514,
      "training_step_time": 7.720837354660034
    },
    {
      "epoch": 3.386974334716797e-05,
      "model_forward_time": 1.7360565662384033,
      "step": 35515
    },
    {
      "epoch": 3.386974334716797e-05,
      "step": 35515,
      "training_step_time": 8.526273727416992
    },
    {
      "epoch": 3.387069702148437e-05,
      "model_forward_time": 1.7519958019256592,
      "step": 35516
    },
    {
      "epoch": 3.387069702148437e-05,
      "step": 35516,
      "training_step_time": 8.779102802276611
    },
    {
      "epoch": 3.3871650695800784e-05,
      "model_forward_time": 2.454409122467041,
      "step": 35517
    },
    {
      "epoch": 3.3871650695800784e-05,
      "step": 35517,
      "training_step_time": 8.50304651260376
    },
    {
      "epoch": 3.387260437011719e-05,
      "model_forward_time": 1.7400171756744385,
      "step": 35518
    },
    {
      "epoch": 3.387260437011719e-05,
      "step": 35518,
      "training_step_time": 7.598460674285889
    },
    {
      "epoch": 3.387355804443359e-05,
      "model_forward_time": 1.7340643405914307,
      "step": 35519
    },
    {
      "epoch": 3.387355804443359e-05,
      "step": 35519,
      "training_step_time": 7.866565704345703
    },
    {
      "epoch": 3.387451171875e-05,
      "grad_norm": 0.14336732029914856,
      "learning_rate": 7.66250003110441e-06,
      "loss": 0.061,
      "step": 35520
    },
    {
      "epoch": 3.387451171875e-05,
      "model_forward_time": 1.7412352561950684,
      "step": 35520
    },
    {
      "dynamics_loss_avg": 0.04226167257875204,
      "epoch": 3.387451171875e-05,
      "step": 35520
    },
    {
      "action_loss_avg": 0.0017960463417693972,
      "epoch": 3.387451171875e-05,
      "step": 35520
    },
    {
      "epoch": 3.387451171875e-05,
      "step": 35520,
      "training_step_time": 7.743988752365112
    },
    {
      "epoch": 3.387546539306641e-05,
      "model_forward_time": 1.7468783855438232,
      "step": 35521
    },
    {
      "epoch": 3.387546539306641e-05,
      "step": 35521,
      "training_step_time": 9.237747192382812
    },
    {
      "epoch": 3.387641906738281e-05,
      "model_forward_time": 1.7453184127807617,
      "step": 35522
    },
    {
      "epoch": 3.387641906738281e-05,
      "step": 35522,
      "training_step_time": 8.569674730300903
    },
    {
      "epoch": 3.387737274169922e-05,
      "model_forward_time": 1.753535270690918,
      "step": 35523
    },
    {
      "epoch": 3.387737274169922e-05,
      "step": 35523,
      "training_step_time": 7.811662197113037
    },
    {
      "epoch": 3.3878326416015626e-05,
      "model_forward_time": 1.7364239692687988,
      "step": 35524
    },
    {
      "epoch": 3.3878326416015626e-05,
      "step": 35524,
      "training_step_time": 7.737223148345947
    },
    {
      "epoch": 3.387928009033203e-05,
      "model_forward_time": 1.7368595600128174,
      "step": 35525
    },
    {
      "epoch": 3.387928009033203e-05,
      "step": 35525,
      "training_step_time": 9.182785034179688
    },
    {
      "epoch": 3.388023376464844e-05,
      "model_forward_time": 1.7300291061401367,
      "step": 35526
    },
    {
      "epoch": 3.388023376464844e-05,
      "step": 35526,
      "training_step_time": 7.512484788894653
    },
    {
      "epoch": 3.3881187438964845e-05,
      "model_forward_time": 2.936342239379883,
      "step": 35527
    },
    {
      "epoch": 3.3881187438964845e-05,
      "step": 35527,
      "training_step_time": 7.718453407287598
    },
    {
      "epoch": 3.388214111328125e-05,
      "model_forward_time": 1.7368803024291992,
      "step": 35528
    },
    {
      "epoch": 3.388214111328125e-05,
      "step": 35528,
      "training_step_time": 8.728755474090576
    },
    {
      "epoch": 3.388309478759766e-05,
      "model_forward_time": 1.7424015998840332,
      "step": 35529
    },
    {
      "epoch": 3.388309478759766e-05,
      "step": 35529,
      "training_step_time": 7.667069435119629
    },
    {
      "epoch": 3.3884048461914064e-05,
      "grad_norm": 0.1547202318906784,
      "learning_rate": 7.66110033909013e-06,
      "loss": 0.061,
      "step": 35530
    },
    {
      "epoch": 3.3884048461914064e-05,
      "model_forward_time": 1.858454942703247,
      "step": 35530
    },
    {
      "dynamics_loss_avg": 0.0695324694737792,
      "epoch": 3.3884048461914064e-05,
      "step": 35530
    },
    {
      "action_loss_avg": 0.0019404245540499687,
      "epoch": 3.3884048461914064e-05,
      "step": 35530
    },
    {
      "epoch": 3.3884048461914064e-05,
      "step": 35530,
      "training_step_time": 7.69055438041687
    },
    {
      "epoch": 3.388500213623047e-05,
      "model_forward_time": 1.7454297542572021,
      "step": 35531
    },
    {
      "epoch": 3.388500213623047e-05,
      "step": 35531,
      "training_step_time": 8.513532876968384
    },
    {
      "epoch": 3.388595581054687e-05,
      "model_forward_time": 1.5544791221618652,
      "step": 35532
    },
    {
      "epoch": 3.388595581054687e-05,
      "step": 35532,
      "training_step_time": 8.603519439697266
    },
    {
      "epoch": 3.388690948486328e-05,
      "model_forward_time": 1.916264533996582,
      "step": 35533
    },
    {
      "epoch": 3.388690948486328e-05,
      "step": 35533,
      "training_step_time": 8.547677040100098
    },
    {
      "epoch": 3.3887863159179686e-05,
      "model_forward_time": 1.9036545753479004,
      "step": 35534
    },
    {
      "epoch": 3.3887863159179686e-05,
      "step": 35534,
      "training_step_time": 8.922863245010376
    },
    {
      "epoch": 3.388881683349609e-05,
      "model_forward_time": 1.915325403213501,
      "step": 35535
    },
    {
      "epoch": 3.388881683349609e-05,
      "step": 35535,
      "training_step_time": 7.748982667922974
    },
    {
      "epoch": 3.38897705078125e-05,
      "model_forward_time": 3.0369760990142822,
      "step": 35536
    },
    {
      "epoch": 3.38897705078125e-05,
      "step": 35536,
      "training_step_time": 7.845482349395752
    },
    {
      "epoch": 3.3890724182128905e-05,
      "model_forward_time": 1.9096705913543701,
      "step": 35537
    },
    {
      "epoch": 3.3890724182128905e-05,
      "step": 35537,
      "training_step_time": 7.649609088897705
    },
    {
      "epoch": 3.389167785644531e-05,
      "model_forward_time": 1.934814214706421,
      "step": 35538
    },
    {
      "epoch": 3.389167785644531e-05,
      "step": 35538,
      "training_step_time": 7.855340957641602
    },
    {
      "epoch": 3.389263153076172e-05,
      "model_forward_time": 1.9053947925567627,
      "step": 35539
    },
    {
      "epoch": 3.389263153076172e-05,
      "step": 35539,
      "training_step_time": 7.807679176330566
    },
    {
      "epoch": 3.3893585205078124e-05,
      "grad_norm": 0.13373823463916779,
      "learning_rate": 7.65970035606192e-06,
      "loss": 0.0586,
      "step": 35540
    },
    {
      "epoch": 3.3893585205078124e-05,
      "model_forward_time": 1.903470754623413,
      "step": 35540
    },
    {
      "dynamics_loss_avg": 0.04513529110699892,
      "epoch": 3.3893585205078124e-05,
      "step": 35540
    },
    {
      "action_loss_avg": 0.0012026360782328993,
      "epoch": 3.3893585205078124e-05,
      "step": 35540
    },
    {
      "epoch": 3.3893585205078124e-05,
      "step": 35540,
      "training_step_time": 7.609162092208862
    },
    {
      "epoch": 3.389453887939453e-05,
      "model_forward_time": 1.9340925216674805,
      "step": 35541
    },
    {
      "epoch": 3.389453887939453e-05,
      "step": 35541,
      "training_step_time": 7.647711753845215
    },
    {
      "epoch": 3.389549255371094e-05,
      "model_forward_time": 1.8795523643493652,
      "step": 35542
    },
    {
      "epoch": 3.389549255371094e-05,
      "step": 35542,
      "training_step_time": 8.518248558044434
    },
    {
      "epoch": 3.389644622802734e-05,
      "model_forward_time": 1.922109842300415,
      "step": 35543
    },
    {
      "epoch": 3.389644622802734e-05,
      "step": 35543,
      "training_step_time": 8.371270895004272
    },
    {
      "epoch": 3.389739990234375e-05,
      "model_forward_time": 1.9259579181671143,
      "step": 35544
    },
    {
      "epoch": 3.389739990234375e-05,
      "step": 35544,
      "training_step_time": 7.759584903717041
    },
    {
      "epoch": 3.389835357666016e-05,
      "model_forward_time": 3.008756160736084,
      "step": 35545
    },
    {
      "epoch": 3.389835357666016e-05,
      "step": 35545,
      "training_step_time": 7.803873062133789
    },
    {
      "epoch": 3.389930725097656e-05,
      "model_forward_time": 1.8987627029418945,
      "step": 35546
    },
    {
      "epoch": 3.389930725097656e-05,
      "step": 35546,
      "training_step_time": 8.504731893539429
    },
    {
      "epoch": 3.3900260925292966e-05,
      "model_forward_time": 1.9006140232086182,
      "step": 35547
    },
    {
      "epoch": 3.3900260925292966e-05,
      "step": 35547,
      "training_step_time": 7.954450845718384
    },
    {
      "epoch": 3.390121459960938e-05,
      "model_forward_time": 1.9232861995697021,
      "step": 35548
    },
    {
      "epoch": 3.390121459960938e-05,
      "step": 35548,
      "training_step_time": 7.781810998916626
    },
    {
      "epoch": 3.390216827392578e-05,
      "model_forward_time": 1.8977789878845215,
      "step": 35549
    },
    {
      "epoch": 3.390216827392578e-05,
      "step": 35549,
      "training_step_time": 8.299288988113403
    },
    {
      "epoch": 3.3903121948242185e-05,
      "grad_norm": 0.11129788309335709,
      "learning_rate": 7.658300082172874e-06,
      "loss": 0.0577,
      "step": 35550
    },
    {
      "epoch": 3.3903121948242185e-05,
      "model_forward_time": 1.9091908931732178,
      "step": 35550
    },
    {
      "dynamics_loss_avg": 0.051219461299479005,
      "epoch": 3.3903121948242185e-05,
      "step": 35550
    },
    {
      "action_loss_avg": 0.0022713383223162964,
      "epoch": 3.3903121948242185e-05,
      "step": 35550
    },
    {
      "epoch": 3.3903121948242185e-05,
      "step": 35550,
      "training_step_time": 8.379161834716797
    },
    {
      "epoch": 3.3904075622558596e-05,
      "model_forward_time": 1.9098100662231445,
      "step": 35551
    },
    {
      "epoch": 3.3904075622558596e-05,
      "step": 35551,
      "training_step_time": 7.622772455215454
    },
    {
      "epoch": 3.3905029296875e-05,
      "model_forward_time": 1.909320592880249,
      "step": 35552
    },
    {
      "epoch": 3.3905029296875e-05,
      "step": 35552,
      "training_step_time": 8.484375715255737
    },
    {
      "epoch": 3.3905982971191404e-05,
      "model_forward_time": 1.5294132232666016,
      "step": 35553
    },
    {
      "epoch": 3.3905982971191404e-05,
      "step": 35553,
      "training_step_time": 7.788252353668213
    },
    {
      "epoch": 3.3906936645507814e-05,
      "model_forward_time": 2.968531847000122,
      "step": 35554
    },
    {
      "epoch": 3.3906936645507814e-05,
      "step": 35554,
      "training_step_time": 7.776810646057129
    },
    {
      "epoch": 3.390789031982422e-05,
      "model_forward_time": 1.9258530139923096,
      "step": 35555
    },
    {
      "epoch": 3.390789031982422e-05,
      "step": 35555,
      "training_step_time": 7.514976978302002
    },
    {
      "epoch": 3.390884399414062e-05,
      "model_forward_time": 1.8811488151550293,
      "step": 35556
    },
    {
      "epoch": 3.390884399414062e-05,
      "step": 35556,
      "training_step_time": 8.372542381286621
    },
    {
      "epoch": 3.390979766845703e-05,
      "model_forward_time": 1.9318938255310059,
      "step": 35557
    },
    {
      "epoch": 3.390979766845703e-05,
      "step": 35557,
      "training_step_time": 7.795222043991089
    },
    {
      "epoch": 3.391075134277344e-05,
      "model_forward_time": 1.930267333984375,
      "step": 35558
    },
    {
      "epoch": 3.391075134277344e-05,
      "step": 35558,
      "training_step_time": 8.188703536987305
    },
    {
      "epoch": 3.391170501708984e-05,
      "model_forward_time": 1.813185691833496,
      "step": 35559
    },
    {
      "epoch": 3.391170501708984e-05,
      "step": 35559,
      "training_step_time": 9.4943368434906
    },
    {
      "epoch": 3.391265869140625e-05,
      "grad_norm": 0.13636581599712372,
      "learning_rate": 7.65689951757613e-06,
      "loss": 0.0601,
      "step": 35560
    },
    {
      "epoch": 3.391265869140625e-05,
      "model_forward_time": 1.903754711151123,
      "step": 35560
    },
    {
      "dynamics_loss_avg": 0.06912983804941178,
      "epoch": 3.391265869140625e-05,
      "step": 35560
    },
    {
      "action_loss_avg": 0.00207582997973077,
      "epoch": 3.391265869140625e-05,
      "step": 35560
    },
    {
      "epoch": 3.391265869140625e-05,
      "step": 35560,
      "training_step_time": 8.376538515090942
    },
    {
      "epoch": 3.3913612365722656e-05,
      "model_forward_time": 1.5570635795593262,
      "step": 35561
    },
    {
      "epoch": 3.3913612365722656e-05,
      "step": 35561,
      "training_step_time": 9.710856199264526
    },
    {
      "epoch": 3.391456604003906e-05,
      "model_forward_time": 1.8973877429962158,
      "step": 35562
    },
    {
      "epoch": 3.391456604003906e-05,
      "step": 35562,
      "training_step_time": 7.824266195297241
    },
    {
      "epoch": 3.391551971435547e-05,
      "model_forward_time": 1.899784803390503,
      "step": 35563
    },
    {
      "epoch": 3.391551971435547e-05,
      "step": 35563,
      "training_step_time": 7.593103647232056
    },
    {
      "epoch": 3.3916473388671875e-05,
      "model_forward_time": 3.0035836696624756,
      "step": 35564
    },
    {
      "epoch": 3.3916473388671875e-05,
      "step": 35564,
      "training_step_time": 7.758149147033691
    },
    {
      "epoch": 3.391742706298828e-05,
      "model_forward_time": 1.9189224243164062,
      "step": 35565
    },
    {
      "epoch": 3.391742706298828e-05,
      "step": 35565,
      "training_step_time": 8.060468196868896
    },
    {
      "epoch": 3.391838073730469e-05,
      "model_forward_time": 1.920757532119751,
      "step": 35566
    },
    {
      "epoch": 3.391838073730469e-05,
      "step": 35566,
      "training_step_time": 8.548742532730103
    },
    {
      "epoch": 3.3919334411621094e-05,
      "model_forward_time": 1.9048357009887695,
      "step": 35567
    },
    {
      "epoch": 3.3919334411621094e-05,
      "step": 35567,
      "training_step_time": 7.964677810668945
    },
    {
      "epoch": 3.39202880859375e-05,
      "model_forward_time": 1.9211244583129883,
      "step": 35568
    },
    {
      "epoch": 3.39202880859375e-05,
      "step": 35568,
      "training_step_time": 7.788332939147949
    },
    {
      "epoch": 3.392124176025391e-05,
      "model_forward_time": 1.9148073196411133,
      "step": 35569
    },
    {
      "epoch": 3.392124176025391e-05,
      "step": 35569,
      "training_step_time": 7.575713634490967
    },
    {
      "epoch": 3.392219543457031e-05,
      "grad_norm": 0.09890427440404892,
      "learning_rate": 7.65549866242485e-06,
      "loss": 0.0582,
      "step": 35570
    },
    {
      "epoch": 3.392219543457031e-05,
      "model_forward_time": 1.9132158756256104,
      "step": 35570
    },
    {
      "dynamics_loss_avg": 0.0461682990193367,
      "epoch": 3.392219543457031e-05,
      "step": 35570
    },
    {
      "action_loss_avg": 0.0011981153744272887,
      "epoch": 3.392219543457031e-05,
      "step": 35570
    },
    {
      "epoch": 3.392219543457031e-05,
      "step": 35570,
      "training_step_time": 8.483400344848633
    },
    {
      "epoch": 3.392314910888672e-05,
      "model_forward_time": 1.900951862335205,
      "step": 35571
    },
    {
      "epoch": 3.392314910888672e-05,
      "step": 35571,
      "training_step_time": 8.616682052612305
    },
    {
      "epoch": 3.392410278320313e-05,
      "model_forward_time": 1.5634918212890625,
      "step": 35572
    },
    {
      "epoch": 3.392410278320313e-05,
      "step": 35572,
      "training_step_time": 8.435681104660034
    },
    {
      "epoch": 3.392505645751953e-05,
      "model_forward_time": 2.986480951309204,
      "step": 35573
    },
    {
      "epoch": 3.392505645751953e-05,
      "step": 35573,
      "training_step_time": 7.895890474319458
    },
    {
      "epoch": 3.3926010131835936e-05,
      "model_forward_time": 1.929610013961792,
      "step": 35574
    },
    {
      "epoch": 3.3926010131835936e-05,
      "step": 35574,
      "training_step_time": 7.980402946472168
    },
    {
      "epoch": 3.3926963806152346e-05,
      "model_forward_time": 1.9088995456695557,
      "step": 35575
    },
    {
      "epoch": 3.3926963806152346e-05,
      "step": 35575,
      "training_step_time": 7.734827995300293
    },
    {
      "epoch": 3.392791748046875e-05,
      "model_forward_time": 1.9301633834838867,
      "step": 35576
    },
    {
      "epoch": 3.392791748046875e-05,
      "step": 35576,
      "training_step_time": 7.7643444538116455
    },
    {
      "epoch": 3.3928871154785155e-05,
      "model_forward_time": 1.905745506286621,
      "step": 35577
    },
    {
      "epoch": 3.3928871154785155e-05,
      "step": 35577,
      "training_step_time": 8.86107087135315
    },
    {
      "epoch": 3.3929824829101565e-05,
      "model_forward_time": 1.924464464187622,
      "step": 35578
    },
    {
      "epoch": 3.3929824829101565e-05,
      "step": 35578,
      "training_step_time": 8.748539924621582
    },
    {
      "epoch": 3.393077850341797e-05,
      "model_forward_time": 1.9183917045593262,
      "step": 35579
    },
    {
      "epoch": 3.393077850341797e-05,
      "step": 35579,
      "training_step_time": 8.72262978553772
    },
    {
      "epoch": 3.393173217773437e-05,
      "grad_norm": 0.13025431334972382,
      "learning_rate": 7.654097516872227e-06,
      "loss": 0.0613,
      "step": 35580
    },
    {
      "epoch": 3.393173217773437e-05,
      "model_forward_time": 1.9171819686889648,
      "step": 35580
    },
    {
      "dynamics_loss_avg": 0.049383786879479884,
      "epoch": 3.393173217773437e-05,
      "step": 35580
    },
    {
      "action_loss_avg": 0.0020480374441831373,
      "epoch": 3.393173217773437e-05,
      "step": 35580
    },
    {
      "epoch": 3.393173217773437e-05,
      "step": 35580,
      "training_step_time": 9.946205854415894
    },
    {
      "epoch": 3.3932685852050784e-05,
      "model_forward_time": 1.925835371017456,
      "step": 35581
    },
    {
      "epoch": 3.3932685852050784e-05,
      "step": 35581,
      "training_step_time": 7.760954856872559
    },
    {
      "epoch": 3.393363952636719e-05,
      "model_forward_time": 2.995330572128296,
      "step": 35582
    },
    {
      "epoch": 3.393363952636719e-05,
      "step": 35582,
      "training_step_time": 9.33223843574524
    },
    {
      "epoch": 3.393459320068359e-05,
      "model_forward_time": 1.9269356727600098,
      "step": 35583
    },
    {
      "epoch": 3.393459320068359e-05,
      "step": 35583,
      "training_step_time": 9.030494451522827
    },
    {
      "epoch": 3.3935546875e-05,
      "model_forward_time": 1.9211132526397705,
      "step": 35584
    },
    {
      "epoch": 3.3935546875e-05,
      "step": 35584,
      "training_step_time": 7.835174798965454
    },
    {
      "epoch": 3.393650054931641e-05,
      "model_forward_time": 1.930680513381958,
      "step": 35585
    },
    {
      "epoch": 3.393650054931641e-05,
      "step": 35585,
      "training_step_time": 7.953329086303711
    },
    {
      "epoch": 3.393745422363281e-05,
      "model_forward_time": 1.9014880657196045,
      "step": 35586
    },
    {
      "epoch": 3.393745422363281e-05,
      "step": 35586,
      "training_step_time": 7.845495700836182
    },
    {
      "epoch": 3.393840789794922e-05,
      "model_forward_time": 1.9150910377502441,
      "step": 35587
    },
    {
      "epoch": 3.393840789794922e-05,
      "step": 35587,
      "training_step_time": 8.313030242919922
    },
    {
      "epoch": 3.3939361572265626e-05,
      "model_forward_time": 1.8825016021728516,
      "step": 35588
    },
    {
      "epoch": 3.3939361572265626e-05,
      "step": 35588,
      "training_step_time": 8.715225219726562
    },
    {
      "epoch": 3.394031524658203e-05,
      "model_forward_time": 1.9266915321350098,
      "step": 35589
    },
    {
      "epoch": 3.394031524658203e-05,
      "step": 35589,
      "training_step_time": 9.629492998123169
    },
    {
      "epoch": 3.394126892089844e-05,
      "grad_norm": 0.14841829240322113,
      "learning_rate": 7.652696081071491e-06,
      "loss": 0.0615,
      "step": 35590
    },
    {
      "epoch": 3.394126892089844e-05,
      "model_forward_time": 1.9181411266326904,
      "step": 35590
    },
    {
      "dynamics_loss_avg": 0.06237602736800909,
      "epoch": 3.394126892089844e-05,
      "step": 35590
    },
    {
      "action_loss_avg": 0.0019048291957005858,
      "epoch": 3.394126892089844e-05,
      "step": 35590
    },
    {
      "epoch": 3.394126892089844e-05,
      "step": 35590,
      "training_step_time": 7.9034504890441895
    },
    {
      "epoch": 3.3942222595214845e-05,
      "model_forward_time": 1.9135911464691162,
      "step": 35591
    },
    {
      "epoch": 3.3942222595214845e-05,
      "step": 35591,
      "training_step_time": 8.990443706512451
    },
    {
      "epoch": 3.394317626953125e-05,
      "model_forward_time": 1.550624132156372,
      "step": 35592
    },
    {
      "epoch": 3.394317626953125e-05,
      "step": 35592,
      "training_step_time": 7.823821783065796
    },
    {
      "epoch": 3.394412994384766e-05,
      "model_forward_time": 1.9289155006408691,
      "step": 35593
    },
    {
      "epoch": 3.394412994384766e-05,
      "step": 35593,
      "training_step_time": 9.267064809799194
    },
    {
      "epoch": 3.3945083618164064e-05,
      "model_forward_time": 1.9067418575286865,
      "step": 35594
    },
    {
      "epoch": 3.3945083618164064e-05,
      "step": 35594,
      "training_step_time": 10.037450075149536
    },
    {
      "epoch": 3.394603729248047e-05,
      "model_forward_time": 1.8906872272491455,
      "step": 35595
    },
    {
      "epoch": 3.394603729248047e-05,
      "step": 35595,
      "training_step_time": 7.82372260093689
    },
    {
      "epoch": 3.394699096679687e-05,
      "model_forward_time": 1.8937346935272217,
      "step": 35596
    },
    {
      "epoch": 3.394699096679687e-05,
      "step": 35596,
      "training_step_time": 7.83621072769165
    },
    {
      "epoch": 3.394794464111328e-05,
      "model_forward_time": 1.9228730201721191,
      "step": 35597
    },
    {
      "epoch": 3.394794464111328e-05,
      "step": 35597,
      "training_step_time": 7.925993919372559
    },
    {
      "epoch": 3.3948898315429687e-05,
      "model_forward_time": 1.906449556350708,
      "step": 35598
    },
    {
      "epoch": 3.3948898315429687e-05,
      "step": 35598,
      "training_step_time": 9.012256860733032
    },
    {
      "epoch": 3.394985198974609e-05,
      "model_forward_time": 1.9091334342956543,
      "step": 35599
    },
    {
      "epoch": 3.394985198974609e-05,
      "step": 35599,
      "training_step_time": 7.82400918006897
    },
    {
      "epoch": 3.39508056640625e-05,
      "grad_norm": 0.16024179756641388,
      "learning_rate": 7.6512943551759e-06,
      "loss": 0.0589,
      "step": 35600
    },
    {
      "epoch": 3.39508056640625e-05,
      "model_forward_time": 1.9126057624816895,
      "step": 35600
    },
    {
      "dynamics_loss_avg": 0.05869287494570017,
      "epoch": 3.39508056640625e-05,
      "step": 35600
    },
    {
      "action_loss_avg": 0.0017496694868896156,
      "epoch": 3.39508056640625e-05,
      "step": 35600
    },
    {
      "epoch": 3.39508056640625e-05,
      "step": 35600,
      "training_step_time": 7.748141050338745
    },
    {
      "epoch": 3.3951759338378905e-05,
      "model_forward_time": 2.9705841541290283,
      "step": 35601
    },
    {
      "epoch": 3.3951759338378905e-05,
      "step": 35601,
      "training_step_time": 8.896734237670898
    },
    {
      "epoch": 3.395271301269531e-05,
      "model_forward_time": 1.880525827407837,
      "step": 35602
    },
    {
      "epoch": 3.395271301269531e-05,
      "step": 35602,
      "training_step_time": 8.086085319519043
    },
    {
      "epoch": 3.395366668701172e-05,
      "model_forward_time": 1.930091381072998,
      "step": 35603
    },
    {
      "epoch": 3.395366668701172e-05,
      "step": 35603,
      "training_step_time": 8.94994592666626
    },
    {
      "epoch": 3.3954620361328124e-05,
      "model_forward_time": 1.922236680984497,
      "step": 35604
    },
    {
      "epoch": 3.3954620361328124e-05,
      "step": 35604,
      "training_step_time": 7.745625734329224
    },
    {
      "epoch": 3.395557403564453e-05,
      "model_forward_time": 1.9312171936035156,
      "step": 35605
    },
    {
      "epoch": 3.395557403564453e-05,
      "step": 35605,
      "training_step_time": 7.836670398712158
    },
    {
      "epoch": 3.395652770996094e-05,
      "model_forward_time": 1.9177570343017578,
      "step": 35606
    },
    {
      "epoch": 3.395652770996094e-05,
      "step": 35606,
      "training_step_time": 8.613712787628174
    },
    {
      "epoch": 3.395748138427734e-05,
      "model_forward_time": 1.5482902526855469,
      "step": 35607
    },
    {
      "epoch": 3.395748138427734e-05,
      "step": 35607,
      "training_step_time": 7.951709032058716
    },
    {
      "epoch": 3.395843505859375e-05,
      "model_forward_time": 1.5426793098449707,
      "step": 35608
    },
    {
      "epoch": 3.395843505859375e-05,
      "step": 35608,
      "training_step_time": 9.199292421340942
    },
    {
      "epoch": 3.395938873291016e-05,
      "model_forward_time": 1.9192922115325928,
      "step": 35609
    },
    {
      "epoch": 3.395938873291016e-05,
      "step": 35609,
      "training_step_time": 8.533185005187988
    },
    {
      "epoch": 3.396034240722656e-05,
      "grad_norm": 0.11756910383701324,
      "learning_rate": 7.649892339338743e-06,
      "loss": 0.0591,
      "step": 35610
    },
    {
      "epoch": 3.396034240722656e-05,
      "model_forward_time": 2.9897408485412598,
      "step": 35610
    },
    {
      "dynamics_loss_avg": 0.05040369108319283,
      "epoch": 3.396034240722656e-05,
      "step": 35610
    },
    {
      "action_loss_avg": 0.002426578081212938,
      "epoch": 3.396034240722656e-05,
      "step": 35610
    },
    {
      "epoch": 3.396034240722656e-05,
      "step": 35610,
      "training_step_time": 7.927129745483398
    },
    {
      "epoch": 3.3961296081542966e-05,
      "model_forward_time": 1.9102251529693604,
      "step": 35611
    },
    {
      "epoch": 3.3961296081542966e-05,
      "step": 35611,
      "training_step_time": 9.188584089279175
    },
    {
      "epoch": 3.396224975585938e-05,
      "model_forward_time": 1.5449872016906738,
      "step": 35612
    },
    {
      "epoch": 3.396224975585938e-05,
      "step": 35612,
      "training_step_time": 7.776962757110596
    },
    {
      "epoch": 3.396320343017578e-05,
      "model_forward_time": 1.9052586555480957,
      "step": 35613
    },
    {
      "epoch": 3.396320343017578e-05,
      "step": 35613,
      "training_step_time": 7.772923469543457
    },
    {
      "epoch": 3.3964157104492185e-05,
      "model_forward_time": 1.5379509925842285,
      "step": 35614
    },
    {
      "epoch": 3.3964157104492185e-05,
      "step": 35614,
      "training_step_time": 7.793958425521851
    },
    {
      "epoch": 3.3965110778808596e-05,
      "model_forward_time": 1.7448632717132568,
      "step": 35615
    },
    {
      "epoch": 3.3965110778808596e-05,
      "step": 35615,
      "training_step_time": 8.597476959228516
    },
    {
      "epoch": 3.3966064453125e-05,
      "model_forward_time": 1.7283270359039307,
      "step": 35616
    },
    {
      "epoch": 3.3966064453125e-05,
      "step": 35616,
      "training_step_time": 8.018198490142822
    },
    {
      "epoch": 3.3967018127441404e-05,
      "model_forward_time": 1.732166051864624,
      "step": 35617
    },
    {
      "epoch": 3.3967018127441404e-05,
      "step": 35617,
      "training_step_time": 7.984870910644531
    },
    {
      "epoch": 3.3967971801757815e-05,
      "model_forward_time": 1.7334089279174805,
      "step": 35618
    },
    {
      "epoch": 3.3967971801757815e-05,
      "step": 35618,
      "training_step_time": 7.749753952026367
    },
    {
      "epoch": 3.396892547607422e-05,
      "model_forward_time": 2.850048065185547,
      "step": 35619
    },
    {
      "epoch": 3.396892547607422e-05,
      "step": 35619,
      "training_step_time": 7.8181304931640625
    },
    {
      "epoch": 3.396987915039062e-05,
      "grad_norm": 0.19466018676757812,
      "learning_rate": 7.648490033713345e-06,
      "loss": 0.0583,
      "step": 35620
    },
    {
      "epoch": 3.396987915039062e-05,
      "model_forward_time": 1.7392845153808594,
      "step": 35620
    },
    {
      "dynamics_loss_avg": 0.05377272311598062,
      "epoch": 3.396987915039062e-05,
      "step": 35620
    },
    {
      "action_loss_avg": 0.002177922529517673,
      "epoch": 3.396987915039062e-05,
      "step": 35620
    },
    {
      "epoch": 3.396987915039062e-05,
      "step": 35620,
      "training_step_time": 8.143972396850586
    },
    {
      "epoch": 3.397083282470703e-05,
      "model_forward_time": 1.7284958362579346,
      "step": 35621
    },
    {
      "epoch": 3.397083282470703e-05,
      "step": 35621,
      "training_step_time": 7.608431339263916
    },
    {
      "epoch": 3.397178649902344e-05,
      "model_forward_time": 1.7756335735321045,
      "step": 35622
    },
    {
      "epoch": 3.397178649902344e-05,
      "step": 35622,
      "training_step_time": 8.0186607837677
    },
    {
      "epoch": 3.397274017333984e-05,
      "model_forward_time": 1.7391059398651123,
      "step": 35623
    },
    {
      "epoch": 3.397274017333984e-05,
      "step": 35623,
      "training_step_time": 7.871675729751587
    },
    {
      "epoch": 3.397369384765625e-05,
      "model_forward_time": 1.731581211090088,
      "step": 35624
    },
    {
      "epoch": 3.397369384765625e-05,
      "step": 35624,
      "training_step_time": 10.12611436843872
    },
    {
      "epoch": 3.3974647521972656e-05,
      "model_forward_time": 1.7386291027069092,
      "step": 35625
    },
    {
      "epoch": 3.3974647521972656e-05,
      "step": 35625,
      "training_step_time": 9.429117918014526
    },
    {
      "epoch": 3.397560119628906e-05,
      "model_forward_time": 1.7298815250396729,
      "step": 35626
    },
    {
      "epoch": 3.397560119628906e-05,
      "step": 35626,
      "training_step_time": 8.99446177482605
    },
    {
      "epoch": 3.397655487060547e-05,
      "model_forward_time": 1.7576367855072021,
      "step": 35627
    },
    {
      "epoch": 3.397655487060547e-05,
      "step": 35627,
      "training_step_time": 7.751637697219849
    },
    {
      "epoch": 3.3977508544921875e-05,
      "model_forward_time": 1.728475570678711,
      "step": 35628
    },
    {
      "epoch": 3.3977508544921875e-05,
      "step": 35628,
      "training_step_time": 9.354545831680298
    },
    {
      "epoch": 3.397846221923828e-05,
      "model_forward_time": 1.0296080112457275,
      "step": 35629
    },
    {
      "epoch": 3.397846221923828e-05,
      "step": 35629,
      "training_step_time": 8.040215969085693
    },
    {
      "epoch": 3.397941589355469e-05,
      "grad_norm": 0.11421411484479904,
      "learning_rate": 7.647087438453059e-06,
      "loss": 0.0573,
      "step": 35630
    },
    {
      "epoch": 3.397941589355469e-05,
      "model_forward_time": 1.7305002212524414,
      "step": 35630
    },
    {
      "dynamics_loss_avg": 0.04788648243993521,
      "epoch": 3.397941589355469e-05,
      "step": 35630
    },
    {
      "action_loss_avg": 0.0018235618364997209,
      "epoch": 3.397941589355469e-05,
      "step": 35630
    },
    {
      "epoch": 3.397941589355469e-05,
      "step": 35630,
      "training_step_time": 7.893648624420166
    },
    {
      "epoch": 3.3980369567871094e-05,
      "model_forward_time": 1.7599060535430908,
      "step": 35631
    },
    {
      "epoch": 3.3980369567871094e-05,
      "step": 35631,
      "training_step_time": 7.9828941822052
    },
    {
      "epoch": 3.39813232421875e-05,
      "model_forward_time": 1.8781726360321045,
      "step": 35632
    },
    {
      "epoch": 3.39813232421875e-05,
      "step": 35632,
      "training_step_time": 8.923092603683472
    },
    {
      "epoch": 3.398227691650391e-05,
      "model_forward_time": 1.7391631603240967,
      "step": 35633
    },
    {
      "epoch": 3.398227691650391e-05,
      "step": 35633,
      "training_step_time": 7.852223634719849
    },
    {
      "epoch": 3.398323059082031e-05,
      "model_forward_time": 1.7329704761505127,
      "step": 35634
    },
    {
      "epoch": 3.398323059082031e-05,
      "step": 35634,
      "training_step_time": 7.993693828582764
    },
    {
      "epoch": 3.398418426513672e-05,
      "model_forward_time": 1.7367784976959229,
      "step": 35635
    },
    {
      "epoch": 3.398418426513672e-05,
      "step": 35635,
      "training_step_time": 8.960436582565308
    },
    {
      "epoch": 3.398513793945313e-05,
      "model_forward_time": 1.7422268390655518,
      "step": 35636
    },
    {
      "epoch": 3.398513793945313e-05,
      "step": 35636,
      "training_step_time": 7.809509515762329
    },
    {
      "epoch": 3.398609161376953e-05,
      "model_forward_time": 1.7328014373779297,
      "step": 35637
    },
    {
      "epoch": 3.398609161376953e-05,
      "step": 35637,
      "training_step_time": 7.7453532218933105
    },
    {
      "epoch": 3.3987045288085936e-05,
      "model_forward_time": 2.8165123462677,
      "step": 35638
    },
    {
      "epoch": 3.3987045288085936e-05,
      "step": 35638,
      "training_step_time": 7.709985256195068
    },
    {
      "epoch": 3.3987998962402347e-05,
      "model_forward_time": 1.7432174682617188,
      "step": 35639
    },
    {
      "epoch": 3.3987998962402347e-05,
      "step": 35639,
      "training_step_time": 8.659844160079956
    },
    {
      "epoch": 3.398895263671875e-05,
      "grad_norm": 0.11873520165681839,
      "learning_rate": 7.64568455371127e-06,
      "loss": 0.0594,
      "step": 35640
    },
    {
      "epoch": 3.398895263671875e-05,
      "model_forward_time": 1.7256088256835938,
      "step": 35640
    },
    {
      "dynamics_loss_avg": 0.05297628957778215,
      "epoch": 3.398895263671875e-05,
      "step": 35640
    },
    {
      "action_loss_avg": 0.0019628595910035074,
      "epoch": 3.398895263671875e-05,
      "step": 35640
    },
    {
      "epoch": 3.398895263671875e-05,
      "step": 35640,
      "training_step_time": 9.711644887924194
    },
    {
      "epoch": 3.3989906311035155e-05,
      "model_forward_time": 1.8695225715637207,
      "step": 35641
    },
    {
      "epoch": 3.3989906311035155e-05,
      "step": 35641,
      "training_step_time": 7.882854461669922
    },
    {
      "epoch": 3.3990859985351565e-05,
      "model_forward_time": 1.7298002243041992,
      "step": 35642
    },
    {
      "epoch": 3.3990859985351565e-05,
      "step": 35642,
      "training_step_time": 9.614252090454102
    },
    {
      "epoch": 3.399181365966797e-05,
      "model_forward_time": 1.733884572982788,
      "step": 35643
    },
    {
      "epoch": 3.399181365966797e-05,
      "step": 35643,
      "training_step_time": 7.825151681900024
    },
    {
      "epoch": 3.3992767333984373e-05,
      "model_forward_time": 1.733872413635254,
      "step": 35644
    },
    {
      "epoch": 3.3992767333984373e-05,
      "step": 35644,
      "training_step_time": 9.267635345458984
    },
    {
      "epoch": 3.3993721008300784e-05,
      "model_forward_time": 1.7378010749816895,
      "step": 35645
    },
    {
      "epoch": 3.3993721008300784e-05,
      "step": 35645,
      "training_step_time": 8.70177149772644
    },
    {
      "epoch": 3.399467468261719e-05,
      "model_forward_time": 1.7290048599243164,
      "step": 35646
    },
    {
      "epoch": 3.399467468261719e-05,
      "step": 35646,
      "training_step_time": 7.725291967391968
    },
    {
      "epoch": 3.399562835693359e-05,
      "model_forward_time": 2.888819456100464,
      "step": 35647
    },
    {
      "epoch": 3.399562835693359e-05,
      "step": 35647,
      "training_step_time": 7.815104007720947
    },
    {
      "epoch": 3.399658203125e-05,
      "model_forward_time": 1.7299911975860596,
      "step": 35648
    },
    {
      "epoch": 3.399658203125e-05,
      "step": 35648,
      "training_step_time": 8.137303829193115
    },
    {
      "epoch": 3.399753570556641e-05,
      "model_forward_time": 1.7339158058166504,
      "step": 35649
    },
    {
      "epoch": 3.399753570556641e-05,
      "step": 35649,
      "training_step_time": 7.8314597606658936
    },
    {
      "epoch": 3.399848937988281e-05,
      "grad_norm": 0.16577006876468658,
      "learning_rate": 7.644281379641395e-06,
      "loss": 0.057,
      "step": 35650
    },
    {
      "epoch": 3.399848937988281e-05,
      "model_forward_time": 1.8519277572631836,
      "step": 35650
    },
    {
      "dynamics_loss_avg": 0.06564797386527062,
      "epoch": 3.399848937988281e-05,
      "step": 35650
    },
    {
      "action_loss_avg": 0.0024047335697105154,
      "epoch": 3.399848937988281e-05,
      "step": 35650
    },
    {
      "epoch": 3.399848937988281e-05,
      "step": 35650,
      "training_step_time": 7.96517014503479
    },
    {
      "epoch": 3.399944305419922e-05,
      "model_forward_time": 1.7370679378509521,
      "step": 35651
    },
    {
      "epoch": 3.399944305419922e-05,
      "step": 35651,
      "training_step_time": 7.722815036773682
    },
    {
      "epoch": 3.4000396728515626e-05,
      "model_forward_time": 1.7231168746948242,
      "step": 35652
    },
    {
      "epoch": 3.4000396728515626e-05,
      "step": 35652,
      "training_step_time": 7.7570390701293945
    },
    {
      "epoch": 3.400135040283203e-05,
      "model_forward_time": 1.7448394298553467,
      "step": 35653
    },
    {
      "epoch": 3.400135040283203e-05,
      "step": 35653,
      "training_step_time": 8.013943195343018
    },
    {
      "epoch": 3.400230407714844e-05,
      "model_forward_time": 1.3575663566589355,
      "step": 35654
    },
    {
      "epoch": 3.400230407714844e-05,
      "step": 35654,
      "training_step_time": 8.667197704315186
    },
    {
      "epoch": 3.4003257751464845e-05,
      "model_forward_time": 1.742307424545288,
      "step": 35655
    },
    {
      "epoch": 3.4003257751464845e-05,
      "step": 35655,
      "training_step_time": 7.7917914390563965
    },
    {
      "epoch": 3.400421142578125e-05,
      "model_forward_time": 1.743464708328247,
      "step": 35656
    },
    {
      "epoch": 3.400421142578125e-05,
      "step": 35656,
      "training_step_time": 8.967713832855225
    },
    {
      "epoch": 3.400516510009765e-05,
      "model_forward_time": 1.7364606857299805,
      "step": 35657
    },
    {
      "epoch": 3.400516510009765e-05,
      "step": 35657,
      "training_step_time": 9.011417865753174
    },
    {
      "epoch": 3.4006118774414064e-05,
      "model_forward_time": 1.7390427589416504,
      "step": 35658
    },
    {
      "epoch": 3.4006118774414064e-05,
      "step": 35658,
      "training_step_time": 8.390709638595581
    },
    {
      "epoch": 3.400707244873047e-05,
      "model_forward_time": 1.8263278007507324,
      "step": 35659
    },
    {
      "epoch": 3.400707244873047e-05,
      "step": 35659,
      "training_step_time": 8.01149606704712
    },
    {
      "epoch": 3.400802612304687e-05,
      "grad_norm": 0.13975298404693604,
      "learning_rate": 7.642877916396887e-06,
      "loss": 0.0587,
      "step": 35660
    },
    {
      "epoch": 3.400802612304687e-05,
      "model_forward_time": 1.7375829219818115,
      "step": 35660
    },
    {
      "dynamics_loss_avg": 0.055472924187779424,
      "epoch": 3.400802612304687e-05,
      "step": 35660
    },
    {
      "action_loss_avg": 0.001395339280134067,
      "epoch": 3.400802612304687e-05,
      "step": 35660
    },
    {
      "epoch": 3.400802612304687e-05,
      "step": 35660,
      "training_step_time": 7.890674352645874
    },
    {
      "epoch": 3.400897979736328e-05,
      "model_forward_time": 1.7180938720703125,
      "step": 35661
    },
    {
      "epoch": 3.400897979736328e-05,
      "step": 35661,
      "training_step_time": 7.810353755950928
    },
    {
      "epoch": 3.4009933471679687e-05,
      "model_forward_time": 1.7405922412872314,
      "step": 35662
    },
    {
      "epoch": 3.4009933471679687e-05,
      "step": 35662,
      "training_step_time": 7.905844688415527
    },
    {
      "epoch": 3.401088714599609e-05,
      "model_forward_time": 1.3709352016448975,
      "step": 35663
    },
    {
      "epoch": 3.401088714599609e-05,
      "step": 35663,
      "training_step_time": 9.84897232055664
    },
    {
      "epoch": 3.40118408203125e-05,
      "model_forward_time": 1.7398865222930908,
      "step": 35664
    },
    {
      "epoch": 3.40118408203125e-05,
      "step": 35664,
      "training_step_time": 7.804889678955078
    },
    {
      "epoch": 3.4012794494628905e-05,
      "model_forward_time": 1.7215473651885986,
      "step": 35665
    },
    {
      "epoch": 3.4012794494628905e-05,
      "step": 35665,
      "training_step_time": 7.779764413833618
    },
    {
      "epoch": 3.401374816894531e-05,
      "model_forward_time": 2.8302621841430664,
      "step": 35666
    },
    {
      "epoch": 3.401374816894531e-05,
      "step": 35666,
      "training_step_time": 7.90544867515564
    },
    {
      "epoch": 3.401470184326172e-05,
      "model_forward_time": 1.7473840713500977,
      "step": 35667
    },
    {
      "epoch": 3.401470184326172e-05,
      "step": 35667,
      "training_step_time": 9.195906162261963
    },
    {
      "epoch": 3.4015655517578124e-05,
      "model_forward_time": 1.7940549850463867,
      "step": 35668
    },
    {
      "epoch": 3.4015655517578124e-05,
      "step": 35668,
      "training_step_time": 8.874581098556519
    },
    {
      "epoch": 3.401660919189453e-05,
      "model_forward_time": 1.7494792938232422,
      "step": 35669
    },
    {
      "epoch": 3.401660919189453e-05,
      "step": 35669,
      "training_step_time": 10.268605709075928
    },
    {
      "epoch": 3.401756286621094e-05,
      "grad_norm": 0.16210287809371948,
      "learning_rate": 7.641474164131221e-06,
      "loss": 0.0574,
      "step": 35670
    },
    {
      "epoch": 3.401756286621094e-05,
      "model_forward_time": 1.7235119342803955,
      "step": 35670
    },
    {
      "dynamics_loss_avg": 0.04156098514795303,
      "epoch": 3.401756286621094e-05,
      "step": 35670
    },
    {
      "action_loss_avg": 0.0011802389577496796,
      "epoch": 3.401756286621094e-05,
      "step": 35670
    },
    {
      "epoch": 3.401756286621094e-05,
      "step": 35670,
      "training_step_time": 7.82811713218689
    },
    {
      "epoch": 3.401851654052734e-05,
      "model_forward_time": 1.745962381362915,
      "step": 35671
    },
    {
      "epoch": 3.401851654052734e-05,
      "step": 35671,
      "training_step_time": 7.983449220657349
    },
    {
      "epoch": 3.401947021484375e-05,
      "model_forward_time": 1.7331461906433105,
      "step": 35672
    },
    {
      "epoch": 3.401947021484375e-05,
      "step": 35672,
      "training_step_time": 9.61364197731018
    },
    {
      "epoch": 3.402042388916016e-05,
      "model_forward_time": 1.7429025173187256,
      "step": 35673
    },
    {
      "epoch": 3.402042388916016e-05,
      "step": 35673,
      "training_step_time": 9.553101778030396
    },
    {
      "epoch": 3.402137756347656e-05,
      "model_forward_time": 1.7246742248535156,
      "step": 35674
    },
    {
      "epoch": 3.402137756347656e-05,
      "step": 35674,
      "training_step_time": 6.964157819747925
    },
    {
      "epoch": 3.4022331237792966e-05,
      "model_forward_time": 1.739342212677002,
      "step": 35675
    },
    {
      "epoch": 3.4022331237792966e-05,
      "step": 35675,
      "training_step_time": 6.905196905136108
    },
    {
      "epoch": 3.402328491210938e-05,
      "model_forward_time": 1.3671252727508545,
      "step": 35676
    },
    {
      "epoch": 3.402328491210938e-05,
      "step": 35676,
      "training_step_time": 6.893732309341431
    },
    {
      "epoch": 3.402423858642578e-05,
      "model_forward_time": 1.733367681503296,
      "step": 35677
    },
    {
      "epoch": 3.402423858642578e-05,
      "step": 35677,
      "training_step_time": 7.0035271644592285
    },
    {
      "epoch": 3.4025192260742185e-05,
      "model_forward_time": 1.3605449199676514,
      "step": 35678
    },
    {
      "epoch": 3.4025192260742185e-05,
      "step": 35678,
      "training_step_time": 6.90519118309021
    },
    {
      "epoch": 3.4026145935058596e-05,
      "model_forward_time": 1.7249433994293213,
      "step": 35679
    },
    {
      "epoch": 3.4026145935058596e-05,
      "step": 35679,
      "training_step_time": 6.920058965682983
    },
    {
      "epoch": 3.4027099609375e-05,
      "grad_norm": 0.14164093136787415,
      "learning_rate": 7.640070122997913e-06,
      "loss": 0.0603,
      "step": 35680
    },
    {
      "epoch": 3.4027099609375e-05,
      "model_forward_time": 1.733597755432129,
      "step": 35680
    },
    {
      "dynamics_loss_avg": 0.05293548386543989,
      "epoch": 3.4027099609375e-05,
      "step": 35680
    },
    {
      "action_loss_avg": 0.0014817019226029515,
      "epoch": 3.4027099609375e-05,
      "step": 35680
    },
    {
      "epoch": 3.4027099609375e-05,
      "step": 35680,
      "training_step_time": 6.996820449829102
    },
    {
      "epoch": 3.4028053283691404e-05,
      "model_forward_time": 1.7404093742370605,
      "step": 35681
    },
    {
      "epoch": 3.4028053283691404e-05,
      "step": 35681,
      "training_step_time": 7.0095131397247314
    },
    {
      "epoch": 3.4029006958007815e-05,
      "model_forward_time": 2.8528804779052734,
      "step": 35682
    },
    {
      "epoch": 3.4029006958007815e-05,
      "step": 35682,
      "training_step_time": 7.822477340698242
    },
    {
      "epoch": 3.402996063232422e-05,
      "model_forward_time": 1.7386291027069092,
      "step": 35683
    },
    {
      "epoch": 3.402996063232422e-05,
      "step": 35683,
      "training_step_time": 9.06877613067627
    },
    {
      "epoch": 3.403091430664062e-05,
      "model_forward_time": 1.76346755027771,
      "step": 35684
    },
    {
      "epoch": 3.403091430664062e-05,
      "step": 35684,
      "training_step_time": 6.98906135559082
    },
    {
      "epoch": 3.4031867980957033e-05,
      "model_forward_time": 1.733914852142334,
      "step": 35685
    },
    {
      "epoch": 3.4031867980957033e-05,
      "step": 35685,
      "training_step_time": 7.039268732070923
    },
    {
      "epoch": 3.403282165527344e-05,
      "model_forward_time": 1.7267091274261475,
      "step": 35686
    },
    {
      "epoch": 3.403282165527344e-05,
      "step": 35686,
      "training_step_time": 6.980870485305786
    },
    {
      "epoch": 3.403377532958984e-05,
      "model_forward_time": 1.738111972808838,
      "step": 35687
    },
    {
      "epoch": 3.403377532958984e-05,
      "step": 35687,
      "training_step_time": 6.9946348667144775
    },
    {
      "epoch": 3.403472900390625e-05,
      "model_forward_time": 1.74747633934021,
      "step": 35688
    },
    {
      "epoch": 3.403472900390625e-05,
      "step": 35688,
      "training_step_time": 7.132845163345337
    },
    {
      "epoch": 3.4035682678222656e-05,
      "model_forward_time": 1.726273536682129,
      "step": 35689
    },
    {
      "epoch": 3.4035682678222656e-05,
      "step": 35689,
      "training_step_time": 7.031821012496948
    },
    {
      "epoch": 3.403663635253906e-05,
      "grad_norm": 0.14751578867435455,
      "learning_rate": 7.638665793150506e-06,
      "loss": 0.0579,
      "step": 35690
    },
    {
      "epoch": 3.403663635253906e-05,
      "model_forward_time": 1.721919059753418,
      "step": 35690
    },
    {
      "dynamics_loss_avg": 0.04677232801914215,
      "epoch": 3.403663635253906e-05,
      "step": 35690
    },
    {
      "action_loss_avg": 0.0017056266398867593,
      "epoch": 3.403663635253906e-05,
      "step": 35690
    },
    {
      "epoch": 3.403663635253906e-05,
      "step": 35690,
      "training_step_time": 7.024364233016968
    },
    {
      "epoch": 3.403759002685547e-05,
      "model_forward_time": 2.5015969276428223,
      "step": 35691
    },
    {
      "epoch": 3.403759002685547e-05,
      "step": 35691,
      "training_step_time": 7.7820518016815186
    },
    {
      "epoch": 3.4038543701171875e-05,
      "model_forward_time": 1.7362494468688965,
      "step": 35692
    },
    {
      "epoch": 3.4038543701171875e-05,
      "step": 35692,
      "training_step_time": 8.10006046295166
    },
    {
      "epoch": 3.403949737548828e-05,
      "model_forward_time": 1.7358367443084717,
      "step": 35693
    },
    {
      "epoch": 3.403949737548828e-05,
      "step": 35693,
      "training_step_time": 7.6811041831970215
    },
    {
      "epoch": 3.404045104980469e-05,
      "model_forward_time": 1.7207109928131104,
      "step": 35694
    },
    {
      "epoch": 3.404045104980469e-05,
      "step": 35694,
      "training_step_time": 6.743255615234375
    },
    {
      "epoch": 3.4041404724121094e-05,
      "model_forward_time": 1.7288947105407715,
      "step": 35695
    },
    {
      "epoch": 3.4041404724121094e-05,
      "step": 35695,
      "training_step_time": 6.8999342918396
    },
    {
      "epoch": 3.40423583984375e-05,
      "model_forward_time": 1.7476675510406494,
      "step": 35696
    },
    {
      "epoch": 3.40423583984375e-05,
      "step": 35696,
      "training_step_time": 6.989962577819824
    },
    {
      "epoch": 3.404331207275391e-05,
      "model_forward_time": 1.7143361568450928,
      "step": 35697
    },
    {
      "epoch": 3.404331207275391e-05,
      "step": 35697,
      "training_step_time": 6.8870227336883545
    },
    {
      "epoch": 3.404426574707031e-05,
      "model_forward_time": 1.7240946292877197,
      "step": 35698
    },
    {
      "epoch": 3.404426574707031e-05,
      "step": 35698,
      "training_step_time": 6.894624471664429
    },
    {
      "epoch": 3.404521942138672e-05,
      "model_forward_time": 1.7389545440673828,
      "step": 35699
    },
    {
      "epoch": 3.404521942138672e-05,
      "step": 35699,
      "training_step_time": 7.003313779830933
    },
    {
      "epoch": 3.404617309570313e-05,
      "grad_norm": 0.19203345477581024,
      "learning_rate": 7.637261174742574e-06,
      "loss": 0.0599,
      "step": 35700
    },
    {
      "epoch": 3.404617309570313e-05,
      "model_forward_time": 2.871652364730835,
      "step": 35700
    },
    {
      "dynamics_loss_avg": 0.04604558888822794,
      "epoch": 3.404617309570313e-05,
      "step": 35700
    },
    {
      "action_loss_avg": 0.001242031902074814,
      "epoch": 3.404617309570313e-05,
      "step": 35700
    },
    {
      "epoch": 3.404617309570313e-05,
      "step": 35700,
      "training_step_time": 7.8068389892578125
    },
    {
      "epoch": 3.404712677001953e-05,
      "model_forward_time": 1.7219302654266357,
      "step": 35701
    },
    {
      "epoch": 3.404712677001953e-05,
      "step": 35701,
      "training_step_time": 9.466857194900513
    },
    {
      "epoch": 3.4048080444335936e-05,
      "model_forward_time": 1.7139537334442139,
      "step": 35702
    },
    {
      "epoch": 3.4048080444335936e-05,
      "step": 35702,
      "training_step_time": 7.74159574508667
    },
    {
      "epoch": 3.4049034118652347e-05,
      "model_forward_time": 1.72328782081604,
      "step": 35703
    },
    {
      "epoch": 3.4049034118652347e-05,
      "step": 35703,
      "training_step_time": 6.991309642791748
    },
    {
      "epoch": 3.404998779296875e-05,
      "model_forward_time": 1.7234184741973877,
      "step": 35704
    },
    {
      "epoch": 3.404998779296875e-05,
      "step": 35704,
      "training_step_time": 6.87837815284729
    },
    {
      "epoch": 3.4050941467285155e-05,
      "model_forward_time": 1.7279624938964844,
      "step": 35705
    },
    {
      "epoch": 3.4050941467285155e-05,
      "step": 35705,
      "training_step_time": 6.934610605239868
    },
    {
      "epoch": 3.4051895141601565e-05,
      "model_forward_time": 1.7286622524261475,
      "step": 35706
    },
    {
      "epoch": 3.4051895141601565e-05,
      "step": 35706,
      "training_step_time": 6.932070970535278
    },
    {
      "epoch": 3.405284881591797e-05,
      "model_forward_time": 1.7177050113677979,
      "step": 35707
    },
    {
      "epoch": 3.405284881591797e-05,
      "step": 35707,
      "training_step_time": 7.0005316734313965
    },
    {
      "epoch": 3.4053802490234373e-05,
      "model_forward_time": 1.7231566905975342,
      "step": 35708
    },
    {
      "epoch": 3.4053802490234373e-05,
      "step": 35708,
      "training_step_time": 6.884736061096191
    },
    {
      "epoch": 3.4054756164550784e-05,
      "model_forward_time": 1.3854691982269287,
      "step": 35709
    },
    {
      "epoch": 3.4054756164550784e-05,
      "step": 35709,
      "training_step_time": 7.638834238052368
    },
    {
      "epoch": 3.405570983886719e-05,
      "grad_norm": 0.12424411624670029,
      "learning_rate": 7.635856267927726e-06,
      "loss": 0.0607,
      "step": 35710
    },
    {
      "epoch": 3.405570983886719e-05,
      "model_forward_time": 2.9437594413757324,
      "step": 35710
    },
    {
      "dynamics_loss_avg": 0.05024004951119423,
      "epoch": 3.405570983886719e-05,
      "step": 35710
    },
    {
      "action_loss_avg": 0.001991933659883216,
      "epoch": 3.405570983886719e-05,
      "step": 35710
    },
    {
      "epoch": 3.405570983886719e-05,
      "step": 35710,
      "training_step_time": 9.003791570663452
    },
    {
      "epoch": 3.405666351318359e-05,
      "model_forward_time": 1.7166552543640137,
      "step": 35711
    },
    {
      "epoch": 3.405666351318359e-05,
      "step": 35711,
      "training_step_time": 7.875188112258911
    },
    {
      "epoch": 3.40576171875e-05,
      "model_forward_time": 1.723719596862793,
      "step": 35712
    },
    {
      "epoch": 3.40576171875e-05,
      "step": 35712,
      "training_step_time": 7.681042909622192
    },
    {
      "epoch": 3.405857086181641e-05,
      "model_forward_time": 1.7217433452606201,
      "step": 35713
    },
    {
      "epoch": 3.405857086181641e-05,
      "step": 35713,
      "training_step_time": 6.940844297409058
    },
    {
      "epoch": 3.405952453613281e-05,
      "model_forward_time": 1.7194089889526367,
      "step": 35714
    },
    {
      "epoch": 3.405952453613281e-05,
      "step": 35714,
      "training_step_time": 6.874628067016602
    },
    {
      "epoch": 3.406047821044922e-05,
      "model_forward_time": 1.7249572277069092,
      "step": 35715
    },
    {
      "epoch": 3.406047821044922e-05,
      "step": 35715,
      "training_step_time": 6.998606204986572
    },
    {
      "epoch": 3.4061431884765626e-05,
      "model_forward_time": 1.7229182720184326,
      "step": 35716
    },
    {
      "epoch": 3.4061431884765626e-05,
      "step": 35716,
      "training_step_time": 6.94462513923645
    },
    {
      "epoch": 3.406238555908203e-05,
      "model_forward_time": 1.7295746803283691,
      "step": 35717
    },
    {
      "epoch": 3.406238555908203e-05,
      "step": 35717,
      "training_step_time": 6.830087423324585
    },
    {
      "epoch": 3.406333923339844e-05,
      "model_forward_time": 1.745532751083374,
      "step": 35718
    },
    {
      "epoch": 3.406333923339844e-05,
      "step": 35718,
      "training_step_time": 7.166793346405029
    },
    {
      "epoch": 3.4064292907714845e-05,
      "model_forward_time": 2.8525984287261963,
      "step": 35719
    },
    {
      "epoch": 3.4064292907714845e-05,
      "step": 35719,
      "training_step_time": 9.041407585144043
    },
    {
      "epoch": 3.406524658203125e-05,
      "grad_norm": 0.124961256980896,
      "learning_rate": 7.634451072859598e-06,
      "loss": 0.0626,
      "step": 35720
    },
    {
      "epoch": 3.406524658203125e-05,
      "model_forward_time": 1.7157752513885498,
      "step": 35720
    },
    {
      "dynamics_loss_avg": 0.05146757382899523,
      "epoch": 3.406524658203125e-05,
      "step": 35720
    },
    {
      "action_loss_avg": 0.0009224751469446346,
      "epoch": 3.406524658203125e-05,
      "step": 35720
    },
    {
      "epoch": 3.406524658203125e-05,
      "step": 35720,
      "training_step_time": 7.819770574569702
    },
    {
      "epoch": 3.406620025634765e-05,
      "model_forward_time": 1.7243177890777588,
      "step": 35721
    },
    {
      "epoch": 3.406620025634765e-05,
      "step": 35721,
      "training_step_time": 7.658462762832642
    },
    {
      "epoch": 3.4067153930664064e-05,
      "model_forward_time": 1.729076862335205,
      "step": 35722
    },
    {
      "epoch": 3.4067153930664064e-05,
      "step": 35722,
      "training_step_time": 6.88568639755249
    },
    {
      "epoch": 3.406810760498047e-05,
      "model_forward_time": 1.7245113849639893,
      "step": 35723
    },
    {
      "epoch": 3.406810760498047e-05,
      "step": 35723,
      "training_step_time": 6.967294454574585
    },
    {
      "epoch": 3.406906127929687e-05,
      "model_forward_time": 1.7233688831329346,
      "step": 35724
    },
    {
      "epoch": 3.406906127929687e-05,
      "step": 35724,
      "training_step_time": 6.866694211959839
    },
    {
      "epoch": 3.407001495361328e-05,
      "model_forward_time": 1.724271297454834,
      "step": 35725
    },
    {
      "epoch": 3.407001495361328e-05,
      "step": 35725,
      "training_step_time": 6.8780505657196045
    },
    {
      "epoch": 3.407096862792969e-05,
      "model_forward_time": 1.7207286357879639,
      "step": 35726
    },
    {
      "epoch": 3.407096862792969e-05,
      "step": 35726,
      "training_step_time": 6.911795139312744
    },
    {
      "epoch": 3.407192230224609e-05,
      "model_forward_time": 1.7444329261779785,
      "step": 35727
    },
    {
      "epoch": 3.407192230224609e-05,
      "step": 35727,
      "training_step_time": 7.183579683303833
    },
    {
      "epoch": 3.40728759765625e-05,
      "model_forward_time": 2.8496949672698975,
      "step": 35728
    },
    {
      "epoch": 3.40728759765625e-05,
      "step": 35728,
      "training_step_time": 8.74165415763855
    },
    {
      "epoch": 3.4073829650878905e-05,
      "model_forward_time": 1.7230052947998047,
      "step": 35729
    },
    {
      "epoch": 3.4073829650878905e-05,
      "step": 35729,
      "training_step_time": 8.38276982307434
    },
    {
      "epoch": 3.407478332519531e-05,
      "grad_norm": 0.14200779795646667,
      "learning_rate": 7.633045589691864e-06,
      "loss": 0.0627,
      "step": 35730
    },
    {
      "epoch": 3.407478332519531e-05,
      "model_forward_time": 1.7180321216583252,
      "step": 35730
    },
    {
      "dynamics_loss_avg": 0.05366682503372431,
      "epoch": 3.407478332519531e-05,
      "step": 35730
    },
    {
      "action_loss_avg": 0.001382766815368086,
      "epoch": 3.407478332519531e-05,
      "step": 35730
    },
    {
      "epoch": 3.407478332519531e-05,
      "step": 35730,
      "training_step_time": 7.528303861618042
    },
    {
      "epoch": 3.407573699951172e-05,
      "model_forward_time": 1.727957010269165,
      "step": 35731
    },
    {
      "epoch": 3.407573699951172e-05,
      "step": 35731,
      "training_step_time": 6.82631254196167
    },
    {
      "epoch": 3.4076690673828124e-05,
      "model_forward_time": 1.7251307964324951,
      "step": 35732
    },
    {
      "epoch": 3.4076690673828124e-05,
      "step": 35732,
      "training_step_time": 6.743125915527344
    },
    {
      "epoch": 3.407764434814453e-05,
      "model_forward_time": 1.7273695468902588,
      "step": 35733
    },
    {
      "epoch": 3.407764434814453e-05,
      "step": 35733,
      "training_step_time": 6.832112073898315
    },
    {
      "epoch": 3.407859802246094e-05,
      "model_forward_time": 1.7230336666107178,
      "step": 35734
    },
    {
      "epoch": 3.407859802246094e-05,
      "step": 35734,
      "training_step_time": 6.849215030670166
    },
    {
      "epoch": 3.407955169677734e-05,
      "model_forward_time": 1.7227599620819092,
      "step": 35735
    },
    {
      "epoch": 3.407955169677734e-05,
      "step": 35735,
      "training_step_time": 6.787097930908203
    },
    {
      "epoch": 3.408050537109375e-05,
      "model_forward_time": 1.7416372299194336,
      "step": 35736
    },
    {
      "epoch": 3.408050537109375e-05,
      "step": 35736,
      "training_step_time": 7.024462461471558
    },
    {
      "epoch": 3.408145904541016e-05,
      "model_forward_time": 2.904059648513794,
      "step": 35737
    },
    {
      "epoch": 3.408145904541016e-05,
      "step": 35737,
      "training_step_time": 8.497721672058105
    },
    {
      "epoch": 3.408241271972656e-05,
      "model_forward_time": 1.722466230392456,
      "step": 35738
    },
    {
      "epoch": 3.408241271972656e-05,
      "step": 35738,
      "training_step_time": 8.808959722518921
    },
    {
      "epoch": 3.4083366394042966e-05,
      "model_forward_time": 1.7150888442993164,
      "step": 35739
    },
    {
      "epoch": 3.4083366394042966e-05,
      "step": 35739,
      "training_step_time": 7.568646669387817
    },
    {
      "epoch": 3.408432006835938e-05,
      "grad_norm": 0.1411769688129425,
      "learning_rate": 7.63163981857822e-06,
      "loss": 0.0588,
      "step": 35740
    },
    {
      "epoch": 3.408432006835938e-05,
      "model_forward_time": 1.7174689769744873,
      "step": 35740
    },
    {
      "dynamics_loss_avg": 0.04481090772897005,
      "epoch": 3.408432006835938e-05,
      "step": 35740
    },
    {
      "action_loss_avg": 0.0023728057800326495,
      "epoch": 3.408432006835938e-05,
      "step": 35740
    },
    {
      "epoch": 3.408432006835938e-05,
      "step": 35740,
      "training_step_time": 6.798597097396851
    },
    {
      "epoch": 3.408527374267578e-05,
      "model_forward_time": 1.357238531112671,
      "step": 35741
    },
    {
      "epoch": 3.408527374267578e-05,
      "step": 35741,
      "training_step_time": 6.80851411819458
    },
    {
      "epoch": 3.4086227416992185e-05,
      "model_forward_time": 1.3512368202209473,
      "step": 35742
    },
    {
      "epoch": 3.4086227416992185e-05,
      "step": 35742,
      "training_step_time": 6.830256700515747
    },
    {
      "epoch": 3.4087181091308596e-05,
      "model_forward_time": 1.7088301181793213,
      "step": 35743
    },
    {
      "epoch": 3.4087181091308596e-05,
      "step": 35743,
      "training_step_time": 6.815653562545776
    },
    {
      "epoch": 3.4088134765625e-05,
      "model_forward_time": 1.7205989360809326,
      "step": 35744
    },
    {
      "epoch": 3.4088134765625e-05,
      "step": 35744,
      "training_step_time": 6.7377026081085205
    },
    {
      "epoch": 3.4089088439941404e-05,
      "model_forward_time": 1.749281406402588,
      "step": 35745
    },
    {
      "epoch": 3.4089088439941404e-05,
      "step": 35745,
      "training_step_time": 7.016997814178467
    },
    {
      "epoch": 3.4090042114257815e-05,
      "model_forward_time": 1.7339766025543213,
      "step": 35746
    },
    {
      "epoch": 3.4090042114257815e-05,
      "step": 35746,
      "training_step_time": 7.6652913093566895
    },
    {
      "epoch": 3.409099578857422e-05,
      "model_forward_time": 2.893200635910034,
      "step": 35747
    },
    {
      "epoch": 3.409099578857422e-05,
      "step": 35747,
      "training_step_time": 7.750014066696167
    },
    {
      "epoch": 3.409194946289062e-05,
      "model_forward_time": 1.7254559993743896,
      "step": 35748
    },
    {
      "epoch": 3.409194946289062e-05,
      "step": 35748,
      "training_step_time": 7.5906970500946045
    },
    {
      "epoch": 3.4092903137207033e-05,
      "model_forward_time": 1.731823444366455,
      "step": 35749
    },
    {
      "epoch": 3.4092903137207033e-05,
      "step": 35749,
      "training_step_time": 6.605838298797607
    },
    {
      "epoch": 3.409385681152344e-05,
      "grad_norm": 0.17271341383457184,
      "learning_rate": 7.630233759672402e-06,
      "loss": 0.0583,
      "step": 35750
    },
    {
      "epoch": 3.409385681152344e-05,
      "model_forward_time": 1.7381775379180908,
      "step": 35750
    },
    {
      "dynamics_loss_avg": 0.04468485154211521,
      "epoch": 3.409385681152344e-05,
      "step": 35750
    },
    {
      "action_loss_avg": 0.0015030220034532248,
      "epoch": 3.409385681152344e-05,
      "step": 35750
    },
    {
      "epoch": 3.409385681152344e-05,
      "step": 35750,
      "training_step_time": 6.764215469360352
    },
    {
      "epoch": 3.409481048583984e-05,
      "model_forward_time": 1.730384111404419,
      "step": 35751
    },
    {
      "epoch": 3.409481048583984e-05,
      "step": 35751,
      "training_step_time": 6.69863486289978
    },
    {
      "epoch": 3.409576416015625e-05,
      "model_forward_time": 1.7291045188903809,
      "step": 35752
    },
    {
      "epoch": 3.409576416015625e-05,
      "step": 35752,
      "training_step_time": 6.6726462841033936
    },
    {
      "epoch": 3.4096717834472656e-05,
      "model_forward_time": 1.728348970413208,
      "step": 35753
    },
    {
      "epoch": 3.4096717834472656e-05,
      "step": 35753,
      "training_step_time": 6.700439453125
    },
    {
      "epoch": 3.409767150878906e-05,
      "model_forward_time": 1.7560570240020752,
      "step": 35754
    },
    {
      "epoch": 3.409767150878906e-05,
      "step": 35754,
      "training_step_time": 7.020448207855225
    },
    {
      "epoch": 3.409862518310547e-05,
      "model_forward_time": 1.7271881103515625,
      "step": 35755
    },
    {
      "epoch": 3.409862518310547e-05,
      "step": 35755,
      "training_step_time": 7.596719741821289
    },
    {
      "epoch": 3.4099578857421875e-05,
      "model_forward_time": 2.9257771968841553,
      "step": 35756
    },
    {
      "epoch": 3.4099578857421875e-05,
      "step": 35756,
      "training_step_time": 11.923594951629639
    },
    {
      "epoch": 3.410053253173828e-05,
      "model_forward_time": 1.730257272720337,
      "step": 35757
    },
    {
      "epoch": 3.410053253173828e-05,
      "step": 35757,
      "training_step_time": 7.615810394287109
    },
    {
      "epoch": 3.410148620605469e-05,
      "model_forward_time": 1.3545246124267578,
      "step": 35758
    },
    {
      "epoch": 3.410148620605469e-05,
      "step": 35758,
      "training_step_time": 6.744217872619629
    },
    {
      "epoch": 3.4102439880371094e-05,
      "model_forward_time": 1.7289843559265137,
      "step": 35759
    },
    {
      "epoch": 3.4102439880371094e-05,
      "step": 35759,
      "training_step_time": 6.706709861755371
    },
    {
      "epoch": 3.41033935546875e-05,
      "grad_norm": 0.13594147562980652,
      "learning_rate": 7.628827413128177e-06,
      "loss": 0.0591,
      "step": 35760
    },
    {
      "epoch": 3.41033935546875e-05,
      "model_forward_time": 1.7288744449615479,
      "step": 35760
    },
    {
      "dynamics_loss_avg": 0.049428080394864085,
      "epoch": 3.41033935546875e-05,
      "step": 35760
    },
    {
      "action_loss_avg": 0.0026117124187294396,
      "epoch": 3.41033935546875e-05,
      "step": 35760
    },
    {
      "epoch": 3.41033935546875e-05,
      "step": 35760,
      "training_step_time": 6.618229866027832
    },
    {
      "epoch": 3.410434722900391e-05,
      "model_forward_time": 1.7279927730560303,
      "step": 35761
    },
    {
      "epoch": 3.410434722900391e-05,
      "step": 35761,
      "training_step_time": 6.734859466552734
    },
    {
      "epoch": 3.410530090332031e-05,
      "model_forward_time": 1.727689266204834,
      "step": 35762
    },
    {
      "epoch": 3.410530090332031e-05,
      "step": 35762,
      "training_step_time": 6.663092374801636
    },
    {
      "epoch": 3.410625457763672e-05,
      "model_forward_time": 1.7636897563934326,
      "step": 35763
    },
    {
      "epoch": 3.410625457763672e-05,
      "step": 35763,
      "training_step_time": 7.06255578994751
    },
    {
      "epoch": 3.410720825195313e-05,
      "model_forward_time": 1.7268648147583008,
      "step": 35764
    },
    {
      "epoch": 3.410720825195313e-05,
      "step": 35764,
      "training_step_time": 7.828696966171265
    },
    {
      "epoch": 3.410816192626953e-05,
      "model_forward_time": 2.942101001739502,
      "step": 35765
    },
    {
      "epoch": 3.410816192626953e-05,
      "step": 35765,
      "training_step_time": 10.753688097000122
    },
    {
      "epoch": 3.4109115600585936e-05,
      "model_forward_time": 1.7305562496185303,
      "step": 35766
    },
    {
      "epoch": 3.4109115600585936e-05,
      "step": 35766,
      "training_step_time": 9.618394613265991
    },
    {
      "epoch": 3.411006927490235e-05,
      "model_forward_time": 1.715022325515747,
      "step": 35767
    },
    {
      "epoch": 3.411006927490235e-05,
      "step": 35767,
      "training_step_time": 7.637414932250977
    },
    {
      "epoch": 3.411102294921875e-05,
      "model_forward_time": 1.7239949703216553,
      "step": 35768
    },
    {
      "epoch": 3.411102294921875e-05,
      "step": 35768,
      "training_step_time": 6.708269357681274
    },
    {
      "epoch": 3.4111976623535155e-05,
      "model_forward_time": 1.7177841663360596,
      "step": 35769
    },
    {
      "epoch": 3.4111976623535155e-05,
      "step": 35769,
      "training_step_time": 6.705773115158081
    },
    {
      "epoch": 3.4112930297851565e-05,
      "grad_norm": 0.1252283751964569,
      "learning_rate": 7.6274207790993384e-06,
      "loss": 0.0587,
      "step": 35770
    },
    {
      "epoch": 3.4112930297851565e-05,
      "model_forward_time": 1.7182021141052246,
      "step": 35770
    },
    {
      "dynamics_loss_avg": 0.06485636094585061,
      "epoch": 3.4112930297851565e-05,
      "step": 35770
    },
    {
      "action_loss_avg": 0.0015854135563131422,
      "epoch": 3.4112930297851565e-05,
      "step": 35770
    },
    {
      "epoch": 3.4112930297851565e-05,
      "step": 35770,
      "training_step_time": 6.757175445556641
    },
    {
      "epoch": 3.411388397216797e-05,
      "model_forward_time": 1.7248480319976807,
      "step": 35771
    },
    {
      "epoch": 3.411388397216797e-05,
      "step": 35771,
      "training_step_time": 6.745845556259155
    },
    {
      "epoch": 3.4114837646484374e-05,
      "model_forward_time": 1.718597650527954,
      "step": 35772
    },
    {
      "epoch": 3.4114837646484374e-05,
      "step": 35772,
      "training_step_time": 6.707868814468384
    },
    {
      "epoch": 3.4115791320800784e-05,
      "model_forward_time": 1.7405853271484375,
      "step": 35773
    },
    {
      "epoch": 3.4115791320800784e-05,
      "step": 35773,
      "training_step_time": 7.748369216918945
    },
    {
      "epoch": 3.411674499511719e-05,
      "model_forward_time": 2.586444854736328,
      "step": 35774
    },
    {
      "epoch": 3.411674499511719e-05,
      "step": 35774,
      "training_step_time": 8.691159725189209
    },
    {
      "epoch": 3.411769866943359e-05,
      "model_forward_time": 1.730224847793579,
      "step": 35775
    },
    {
      "epoch": 3.411769866943359e-05,
      "step": 35775,
      "training_step_time": 8.835780143737793
    },
    {
      "epoch": 3.411865234375e-05,
      "model_forward_time": 1.7298171520233154,
      "step": 35776
    },
    {
      "epoch": 3.411865234375e-05,
      "step": 35776,
      "training_step_time": 7.63375449180603
    },
    {
      "epoch": 3.411960601806641e-05,
      "model_forward_time": 1.7298192977905273,
      "step": 35777
    },
    {
      "epoch": 3.411960601806641e-05,
      "step": 35777,
      "training_step_time": 6.8308329582214355
    },
    {
      "epoch": 3.412055969238281e-05,
      "model_forward_time": 1.7305023670196533,
      "step": 35778
    },
    {
      "epoch": 3.412055969238281e-05,
      "step": 35778,
      "training_step_time": 6.877262115478516
    },
    {
      "epoch": 3.412151336669922e-05,
      "model_forward_time": 1.7228436470031738,
      "step": 35779
    },
    {
      "epoch": 3.412151336669922e-05,
      "step": 35779,
      "training_step_time": 6.8545918464660645
    },
    {
      "epoch": 3.4122467041015626e-05,
      "grad_norm": 0.1243082731962204,
      "learning_rate": 7.6260138577397114e-06,
      "loss": 0.0578,
      "step": 35780
    },
    {
      "epoch": 3.4122467041015626e-05,
      "model_forward_time": 1.7320561408996582,
      "step": 35780
    },
    {
      "dynamics_loss_avg": 0.048792633414268496,
      "epoch": 3.4122467041015626e-05,
      "step": 35780
    },
    {
      "action_loss_avg": 0.001665800009504892,
      "epoch": 3.4122467041015626e-05,
      "step": 35780
    },
    {
      "epoch": 3.4122467041015626e-05,
      "step": 35780,
      "training_step_time": 6.8578784465789795
    },
    {
      "epoch": 3.412342071533203e-05,
      "model_forward_time": 1.7309470176696777,
      "step": 35781
    },
    {
      "epoch": 3.412342071533203e-05,
      "step": 35781,
      "training_step_time": 6.758891820907593
    },
    {
      "epoch": 3.4124374389648434e-05,
      "model_forward_time": 1.7570111751556396,
      "step": 35782
    },
    {
      "epoch": 3.4124374389648434e-05,
      "step": 35782,
      "training_step_time": 7.860809087753296
    },
    {
      "epoch": 3.4125328063964845e-05,
      "model_forward_time": 2.880716323852539,
      "step": 35783
    },
    {
      "epoch": 3.4125328063964845e-05,
      "step": 35783,
      "training_step_time": 8.84099292755127
    },
    {
      "epoch": 3.412628173828125e-05,
      "model_forward_time": 1.7267568111419678,
      "step": 35784
    },
    {
      "epoch": 3.412628173828125e-05,
      "step": 35784,
      "training_step_time": 10.847803354263306
    },
    {
      "epoch": 3.412723541259765e-05,
      "model_forward_time": 1.7280704975128174,
      "step": 35785
    },
    {
      "epoch": 3.412723541259765e-05,
      "step": 35785,
      "training_step_time": 7.6974241733551025
    },
    {
      "epoch": 3.4128189086914064e-05,
      "model_forward_time": 1.7176463603973389,
      "step": 35786
    },
    {
      "epoch": 3.4128189086914064e-05,
      "step": 35786,
      "training_step_time": 6.90219783782959
    },
    {
      "epoch": 3.412914276123047e-05,
      "model_forward_time": 1.7226269245147705,
      "step": 35787
    },
    {
      "epoch": 3.412914276123047e-05,
      "step": 35787,
      "training_step_time": 6.917646884918213
    },
    {
      "epoch": 3.413009643554687e-05,
      "model_forward_time": 1.7239348888397217,
      "step": 35788
    },
    {
      "epoch": 3.413009643554687e-05,
      "step": 35788,
      "training_step_time": 6.8642659187316895
    },
    {
      "epoch": 3.413105010986328e-05,
      "model_forward_time": 1.7274329662322998,
      "step": 35789
    },
    {
      "epoch": 3.413105010986328e-05,
      "step": 35789,
      "training_step_time": 6.922279596328735
    },
    {
      "epoch": 3.413200378417969e-05,
      "grad_norm": 0.08613629639148712,
      "learning_rate": 7.624606649203159e-06,
      "loss": 0.0599,
      "step": 35790
    },
    {
      "epoch": 3.413200378417969e-05,
      "model_forward_time": 1.728062629699707,
      "step": 35790
    },
    {
      "dynamics_loss_avg": 0.06178948599845171,
      "epoch": 3.413200378417969e-05,
      "step": 35790
    },
    {
      "action_loss_avg": 0.001714057009667158,
      "epoch": 3.413200378417969e-05,
      "step": 35790
    },
    {
      "epoch": 3.413200378417969e-05,
      "step": 35790,
      "training_step_time": 6.926090478897095
    },
    {
      "epoch": 3.413295745849609e-05,
      "model_forward_time": 1.733402967453003,
      "step": 35791
    },
    {
      "epoch": 3.413295745849609e-05,
      "step": 35791,
      "training_step_time": 7.857478380203247
    },
    {
      "epoch": 3.41339111328125e-05,
      "model_forward_time": 2.81923508644104,
      "step": 35792
    },
    {
      "epoch": 3.41339111328125e-05,
      "step": 35792,
      "training_step_time": 7.836637496948242
    },
    {
      "epoch": 3.4134864807128906e-05,
      "model_forward_time": 1.7404727935791016,
      "step": 35793
    },
    {
      "epoch": 3.4134864807128906e-05,
      "step": 35793,
      "training_step_time": 9.452874183654785
    },
    {
      "epoch": 3.413581848144531e-05,
      "model_forward_time": 1.724320888519287,
      "step": 35794
    },
    {
      "epoch": 3.413581848144531e-05,
      "step": 35794,
      "training_step_time": 7.8520166873931885
    },
    {
      "epoch": 3.413677215576172e-05,
      "model_forward_time": 1.7258780002593994,
      "step": 35795
    },
    {
      "epoch": 3.413677215576172e-05,
      "step": 35795,
      "training_step_time": 7.599659442901611
    },
    {
      "epoch": 3.4137725830078124e-05,
      "model_forward_time": 1.72503662109375,
      "step": 35796
    },
    {
      "epoch": 3.4137725830078124e-05,
      "step": 35796,
      "training_step_time": 6.904433488845825
    },
    {
      "epoch": 3.413867950439453e-05,
      "model_forward_time": 1.7258327007293701,
      "step": 35797
    },
    {
      "epoch": 3.413867950439453e-05,
      "step": 35797,
      "training_step_time": 6.91812801361084
    },
    {
      "epoch": 3.413963317871094e-05,
      "model_forward_time": 1.7244179248809814,
      "step": 35798
    },
    {
      "epoch": 3.413963317871094e-05,
      "step": 35798,
      "training_step_time": 6.918734073638916
    },
    {
      "epoch": 3.414058685302734e-05,
      "model_forward_time": 1.7114531993865967,
      "step": 35799
    },
    {
      "epoch": 3.414058685302734e-05,
      "step": 35799,
      "training_step_time": 6.855634927749634
    },
    {
      "epoch": 3.414154052734375e-05,
      "grad_norm": 0.10627823323011398,
      "learning_rate": 7.623199153643569e-06,
      "loss": 0.0589,
      "step": 35800
    },
    {
      "epoch": 3.414154052734375e-05,
      "model_forward_time": 1.72706937789917,
      "step": 35800
    },
    {
      "dynamics_loss_avg": 0.05040918532758951,
      "epoch": 3.414154052734375e-05,
      "step": 35800
    },
    {
      "action_loss_avg": 0.0017186446784762666,
      "epoch": 3.414154052734375e-05,
      "step": 35800
    },
    {
      "epoch": 3.414154052734375e-05,
      "step": 35800,
      "training_step_time": 7.765766143798828
    },
    {
      "epoch": 3.414249420166016e-05,
      "model_forward_time": 1.7492883205413818,
      "step": 35801
    },
    {
      "epoch": 3.414249420166016e-05,
      "step": 35801,
      "training_step_time": 8.211547136306763
    },
    {
      "epoch": 3.414344787597656e-05,
      "model_forward_time": 2.9903900623321533,
      "step": 35802
    },
    {
      "epoch": 3.414344787597656e-05,
      "step": 35802,
      "training_step_time": 10.014057159423828
    },
    {
      "epoch": 3.4144401550292966e-05,
      "model_forward_time": 1.7295794486999512,
      "step": 35803
    },
    {
      "epoch": 3.4144401550292966e-05,
      "step": 35803,
      "training_step_time": 8.926025867462158
    },
    {
      "epoch": 3.414535522460938e-05,
      "model_forward_time": 1.7314839363098145,
      "step": 35804
    },
    {
      "epoch": 3.414535522460938e-05,
      "step": 35804,
      "training_step_time": 7.681511163711548
    },
    {
      "epoch": 3.414630889892578e-05,
      "model_forward_time": 1.7311432361602783,
      "step": 35805
    },
    {
      "epoch": 3.414630889892578e-05,
      "step": 35805,
      "training_step_time": 7.5656373500823975
    },
    {
      "epoch": 3.4147262573242185e-05,
      "model_forward_time": 1.7305831909179688,
      "step": 35806
    },
    {
      "epoch": 3.4147262573242185e-05,
      "step": 35806,
      "training_step_time": 6.898792743682861
    },
    {
      "epoch": 3.4148216247558596e-05,
      "model_forward_time": 1.7331507205963135,
      "step": 35807
    },
    {
      "epoch": 3.4148216247558596e-05,
      "step": 35807,
      "training_step_time": 6.892770051956177
    },
    {
      "epoch": 3.4149169921875e-05,
      "model_forward_time": 1.7224137783050537,
      "step": 35808
    },
    {
      "epoch": 3.4149169921875e-05,
      "step": 35808,
      "training_step_time": 6.813188552856445
    },
    {
      "epoch": 3.4150123596191404e-05,
      "model_forward_time": 1.7305688858032227,
      "step": 35809
    },
    {
      "epoch": 3.4150123596191404e-05,
      "step": 35809,
      "training_step_time": 7.888877630233765
    },
    {
      "epoch": 3.4151077270507815e-05,
      "grad_norm": 0.12366525828838348,
      "learning_rate": 7.621791371214863e-06,
      "loss": 0.0599,
      "step": 35810
    },
    {
      "epoch": 3.4151077270507815e-05,
      "model_forward_time": 1.744819164276123,
      "step": 35810
    },
    {
      "dynamics_loss_avg": 0.05433182567358017,
      "epoch": 3.4151077270507815e-05,
      "step": 35810
    },
    {
      "action_loss_avg": 0.0015166932018473745,
      "epoch": 3.4151077270507815e-05,
      "step": 35810
    },
    {
      "epoch": 3.4151077270507815e-05,
      "step": 35810,
      "training_step_time": 8.235580682754517
    },
    {
      "epoch": 3.415203094482422e-05,
      "model_forward_time": 2.995650053024292,
      "step": 35811
    },
    {
      "epoch": 3.415203094482422e-05,
      "step": 35811,
      "training_step_time": 10.034741640090942
    },
    {
      "epoch": 3.415298461914062e-05,
      "model_forward_time": 1.7019915580749512,
      "step": 35812
    },
    {
      "epoch": 3.415298461914062e-05,
      "step": 35812,
      "training_step_time": 8.627159833908081
    },
    {
      "epoch": 3.4153938293457034e-05,
      "model_forward_time": 1.3502044677734375,
      "step": 35813
    },
    {
      "epoch": 3.4153938293457034e-05,
      "step": 35813,
      "training_step_time": 7.695511102676392
    },
    {
      "epoch": 3.415489196777344e-05,
      "model_forward_time": 1.7137389183044434,
      "step": 35814
    },
    {
      "epoch": 3.415489196777344e-05,
      "step": 35814,
      "training_step_time": 7.289852142333984
    },
    {
      "epoch": 3.415584564208984e-05,
      "model_forward_time": 1.3514540195465088,
      "step": 35815
    },
    {
      "epoch": 3.415584564208984e-05,
      "step": 35815,
      "training_step_time": 6.9664306640625
    },
    {
      "epoch": 3.415679931640625e-05,
      "model_forward_time": 1.7181432247161865,
      "step": 35816
    },
    {
      "epoch": 3.415679931640625e-05,
      "step": 35816,
      "training_step_time": 6.970174312591553
    },
    {
      "epoch": 3.4157752990722656e-05,
      "model_forward_time": 1.7416284084320068,
      "step": 35817
    },
    {
      "epoch": 3.4157752990722656e-05,
      "step": 35817,
      "training_step_time": 6.97228741645813
    },
    {
      "epoch": 3.415870666503906e-05,
      "model_forward_time": 1.7363402843475342,
      "step": 35818
    },
    {
      "epoch": 3.415870666503906e-05,
      "step": 35818,
      "training_step_time": 7.791941404342651
    },
    {
      "epoch": 3.415966033935547e-05,
      "model_forward_time": 1.7722561359405518,
      "step": 35819
    },
    {
      "epoch": 3.415966033935547e-05,
      "step": 35819,
      "training_step_time": 8.288050413131714
    },
    {
      "epoch": 3.4160614013671875e-05,
      "grad_norm": 0.13122780621051788,
      "learning_rate": 7.620383302070995e-06,
      "loss": 0.0609,
      "step": 35820
    },
    {
      "epoch": 3.4160614013671875e-05,
      "model_forward_time": 1.9379444122314453,
      "step": 35820
    },
    {
      "dynamics_loss_avg": 0.03478263271972537,
      "epoch": 3.4160614013671875e-05,
      "step": 35820
    },
    {
      "action_loss_avg": 0.0013405891339061781,
      "epoch": 3.4160614013671875e-05,
      "step": 35820
    },
    {
      "epoch": 3.4160614013671875e-05,
      "step": 35820,
      "training_step_time": 8.661393165588379
    },
    {
      "epoch": 3.416156768798828e-05,
      "model_forward_time": 1.381669521331787,
      "step": 35821
    },
    {
      "epoch": 3.416156768798828e-05,
      "step": 35821,
      "training_step_time": 9.859641790390015
    },
    {
      "epoch": 3.416252136230469e-05,
      "model_forward_time": 1.7424283027648926,
      "step": 35822
    },
    {
      "epoch": 3.416252136230469e-05,
      "step": 35822,
      "training_step_time": 7.612661600112915
    },
    {
      "epoch": 3.4163475036621094e-05,
      "model_forward_time": 1.7465147972106934,
      "step": 35823
    },
    {
      "epoch": 3.4163475036621094e-05,
      "step": 35823,
      "training_step_time": 7.507611513137817
    },
    {
      "epoch": 3.41644287109375e-05,
      "model_forward_time": 1.7457478046417236,
      "step": 35824
    },
    {
      "epoch": 3.41644287109375e-05,
      "step": 35824,
      "training_step_time": 6.891473770141602
    },
    {
      "epoch": 3.416538238525391e-05,
      "model_forward_time": 1.7351157665252686,
      "step": 35825
    },
    {
      "epoch": 3.416538238525391e-05,
      "step": 35825,
      "training_step_time": 6.949066400527954
    },
    {
      "epoch": 3.416633605957031e-05,
      "model_forward_time": 1.7514557838439941,
      "step": 35826
    },
    {
      "epoch": 3.416633605957031e-05,
      "step": 35826,
      "training_step_time": 6.8828043937683105
    },
    {
      "epoch": 3.416728973388672e-05,
      "model_forward_time": 1.7589526176452637,
      "step": 35827
    },
    {
      "epoch": 3.416728973388672e-05,
      "step": 35827,
      "training_step_time": 7.842730760574341
    },
    {
      "epoch": 3.416824340820313e-05,
      "model_forward_time": 1.771723747253418,
      "step": 35828
    },
    {
      "epoch": 3.416824340820313e-05,
      "step": 35828,
      "training_step_time": 7.66219425201416
    },
    {
      "epoch": 3.416919708251953e-05,
      "model_forward_time": 1.7207119464874268,
      "step": 35829
    },
    {
      "epoch": 3.416919708251953e-05,
      "step": 35829,
      "training_step_time": 9.050977230072021
    },
    {
      "epoch": 3.4170150756835936e-05,
      "grad_norm": 0.10289677232503891,
      "learning_rate": 7.618974946365949e-06,
      "loss": 0.0594,
      "step": 35830
    },
    {
      "epoch": 3.4170150756835936e-05,
      "model_forward_time": 2.303600311279297,
      "step": 35830
    },
    {
      "dynamics_loss_avg": 0.0528742915019393,
      "epoch": 3.4170150756835936e-05,
      "step": 35830
    },
    {
      "action_loss_avg": 0.0012824651435948907,
      "epoch": 3.4170150756835936e-05,
      "step": 35830
    },
    {
      "epoch": 3.4170150756835936e-05,
      "step": 35830,
      "training_step_time": 8.608133554458618
    },
    {
      "epoch": 3.417110443115235e-05,
      "model_forward_time": 1.7425932884216309,
      "step": 35831
    },
    {
      "epoch": 3.417110443115235e-05,
      "step": 35831,
      "training_step_time": 7.708665132522583
    },
    {
      "epoch": 3.417205810546875e-05,
      "model_forward_time": 1.7336618900299072,
      "step": 35832
    },
    {
      "epoch": 3.417205810546875e-05,
      "step": 35832,
      "training_step_time": 7.317867279052734
    },
    {
      "epoch": 3.4173011779785155e-05,
      "model_forward_time": 1.7470338344573975,
      "step": 35833
    },
    {
      "epoch": 3.4173011779785155e-05,
      "step": 35833,
      "training_step_time": 6.795637369155884
    },
    {
      "epoch": 3.4173965454101566e-05,
      "model_forward_time": 1.739173173904419,
      "step": 35834
    },
    {
      "epoch": 3.4173965454101566e-05,
      "step": 35834,
      "training_step_time": 6.807306289672852
    },
    {
      "epoch": 3.417491912841797e-05,
      "model_forward_time": 1.7449188232421875,
      "step": 35835
    },
    {
      "epoch": 3.417491912841797e-05,
      "step": 35835,
      "training_step_time": 6.880820274353027
    },
    {
      "epoch": 3.4175872802734374e-05,
      "model_forward_time": 1.3837823867797852,
      "step": 35836
    },
    {
      "epoch": 3.4175872802734374e-05,
      "step": 35836,
      "training_step_time": 7.957327127456665
    },
    {
      "epoch": 3.4176826477050784e-05,
      "model_forward_time": 1.7461676597595215,
      "step": 35837
    },
    {
      "epoch": 3.4176826477050784e-05,
      "step": 35837,
      "training_step_time": 9.091628551483154
    },
    {
      "epoch": 3.417778015136719e-05,
      "model_forward_time": 1.7442433834075928,
      "step": 35838
    },
    {
      "epoch": 3.417778015136719e-05,
      "step": 35838,
      "training_step_time": 7.592152118682861
    },
    {
      "epoch": 3.417873382568359e-05,
      "model_forward_time": 2.2988667488098145,
      "step": 35839
    },
    {
      "epoch": 3.417873382568359e-05,
      "step": 35839,
      "training_step_time": 12.06986665725708
    },
    {
      "epoch": 3.41796875e-05,
      "grad_norm": 0.15966756641864777,
      "learning_rate": 7.617566304253738e-06,
      "loss": 0.0601,
      "step": 35840
    },
    {
      "epoch": 3.41796875e-05,
      "model_forward_time": 1.7540075778961182,
      "step": 35840
    },
    {
      "dynamics_loss_avg": 0.05827132854610682,
      "epoch": 3.41796875e-05,
      "step": 35840
    },
    {
      "action_loss_avg": 0.0022500129620311783,
      "epoch": 3.41796875e-05,
      "step": 35840
    },
    {
      "epoch": 3.41796875e-05,
      "step": 35840,
      "training_step_time": 7.776270389556885
    },
    {
      "epoch": 3.418064117431641e-05,
      "model_forward_time": 1.7655789852142334,
      "step": 35841
    },
    {
      "epoch": 3.418064117431641e-05,
      "step": 35841,
      "training_step_time": 7.457137107849121
    },
    {
      "epoch": 3.418159484863281e-05,
      "model_forward_time": 1.7540881633758545,
      "step": 35842
    },
    {
      "epoch": 3.418159484863281e-05,
      "step": 35842,
      "training_step_time": 7.139911413192749
    },
    {
      "epoch": 3.418254852294922e-05,
      "model_forward_time": 1.753413438796997,
      "step": 35843
    },
    {
      "epoch": 3.418254852294922e-05,
      "step": 35843,
      "training_step_time": 6.8124823570251465
    },
    {
      "epoch": 3.4183502197265626e-05,
      "model_forward_time": 1.759305477142334,
      "step": 35844
    },
    {
      "epoch": 3.4183502197265626e-05,
      "step": 35844,
      "training_step_time": 6.815905809402466
    },
    {
      "epoch": 3.418445587158203e-05,
      "model_forward_time": 1.7423789501190186,
      "step": 35845
    },
    {
      "epoch": 3.418445587158203e-05,
      "step": 35845,
      "training_step_time": 7.910956144332886
    },
    {
      "epoch": 3.4185409545898434e-05,
      "model_forward_time": 1.765634298324585,
      "step": 35846
    },
    {
      "epoch": 3.4185409545898434e-05,
      "step": 35846,
      "training_step_time": 8.591117143630981
    },
    {
      "epoch": 3.4186363220214845e-05,
      "model_forward_time": 1.7587487697601318,
      "step": 35847
    },
    {
      "epoch": 3.4186363220214845e-05,
      "step": 35847,
      "training_step_time": 9.368431806564331
    },
    {
      "epoch": 3.418731689453125e-05,
      "model_forward_time": 2.3223114013671875,
      "step": 35848
    },
    {
      "epoch": 3.418731689453125e-05,
      "step": 35848,
      "training_step_time": 9.355827808380127
    },
    {
      "epoch": 3.418827056884765e-05,
      "model_forward_time": 1.7560040950775146,
      "step": 35849
    },
    {
      "epoch": 3.418827056884765e-05,
      "step": 35849,
      "training_step_time": 8.778314590454102
    },
    {
      "epoch": 3.4189224243164064e-05,
      "grad_norm": 0.14786416292190552,
      "learning_rate": 7.616157375888416e-06,
      "loss": 0.0637,
      "step": 35850
    },
    {
      "epoch": 3.4189224243164064e-05,
      "model_forward_time": 1.7637968063354492,
      "step": 35850
    },
    {
      "dynamics_loss_avg": 0.07072452679276467,
      "epoch": 3.4189224243164064e-05,
      "step": 35850
    },
    {
      "action_loss_avg": 0.002055795432534069,
      "epoch": 3.4189224243164064e-05,
      "step": 35850
    },
    {
      "epoch": 3.4189224243164064e-05,
      "step": 35850,
      "training_step_time": 7.6242454051971436
    },
    {
      "epoch": 3.419017791748047e-05,
      "model_forward_time": 1.7680449485778809,
      "step": 35851
    },
    {
      "epoch": 3.419017791748047e-05,
      "step": 35851,
      "training_step_time": 7.56386137008667
    },
    {
      "epoch": 3.419113159179687e-05,
      "model_forward_time": 1.7585580348968506,
      "step": 35852
    },
    {
      "epoch": 3.419113159179687e-05,
      "step": 35852,
      "training_step_time": 6.841748476028442
    },
    {
      "epoch": 3.419208526611328e-05,
      "model_forward_time": 1.75156831741333,
      "step": 35853
    },
    {
      "epoch": 3.419208526611328e-05,
      "step": 35853,
      "training_step_time": 6.787210464477539
    },
    {
      "epoch": 3.419303894042969e-05,
      "model_forward_time": 1.7616915702819824,
      "step": 35854
    },
    {
      "epoch": 3.419303894042969e-05,
      "step": 35854,
      "training_step_time": 7.649883985519409
    },
    {
      "epoch": 3.419399261474609e-05,
      "model_forward_time": 1.7613189220428467,
      "step": 35855
    },
    {
      "epoch": 3.419399261474609e-05,
      "step": 35855,
      "training_step_time": 7.939093112945557
    },
    {
      "epoch": 3.41949462890625e-05,
      "model_forward_time": 1.7597622871398926,
      "step": 35856
    },
    {
      "epoch": 3.41949462890625e-05,
      "step": 35856,
      "training_step_time": 9.251786470413208
    },
    {
      "epoch": 3.4195899963378906e-05,
      "model_forward_time": 2.336263656616211,
      "step": 35857
    },
    {
      "epoch": 3.4195899963378906e-05,
      "step": 35857,
      "training_step_time": 9.436708927154541
    },
    {
      "epoch": 3.419685363769531e-05,
      "model_forward_time": 1.7565789222717285,
      "step": 35858
    },
    {
      "epoch": 3.419685363769531e-05,
      "step": 35858,
      "training_step_time": 7.815096855163574
    },
    {
      "epoch": 3.419780731201172e-05,
      "model_forward_time": 1.7620344161987305,
      "step": 35859
    },
    {
      "epoch": 3.419780731201172e-05,
      "step": 35859,
      "training_step_time": 7.563940525054932
    },
    {
      "epoch": 3.4198760986328124e-05,
      "grad_norm": 0.17824895679950714,
      "learning_rate": 7.6147481614240525e-06,
      "loss": 0.0594,
      "step": 35860
    },
    {
      "epoch": 3.4198760986328124e-05,
      "model_forward_time": 1.7620244026184082,
      "step": 35860
    },
    {
      "dynamics_loss_avg": 0.052748124301433566,
      "epoch": 3.4198760986328124e-05,
      "step": 35860
    },
    {
      "action_loss_avg": 0.0015208782802801578,
      "epoch": 3.4198760986328124e-05,
      "step": 35860
    },
    {
      "epoch": 3.4198760986328124e-05,
      "step": 35860,
      "training_step_time": 7.283215522766113
    },
    {
      "epoch": 3.419971466064453e-05,
      "model_forward_time": 1.7576353549957275,
      "step": 35861
    },
    {
      "epoch": 3.419971466064453e-05,
      "step": 35861,
      "training_step_time": 6.902880907058716
    },
    {
      "epoch": 3.420066833496094e-05,
      "model_forward_time": 1.7547757625579834,
      "step": 35862
    },
    {
      "epoch": 3.420066833496094e-05,
      "step": 35862,
      "training_step_time": 6.816266775131226
    },
    {
      "epoch": 3.420162200927734e-05,
      "model_forward_time": 1.7551071643829346,
      "step": 35863
    },
    {
      "epoch": 3.420162200927734e-05,
      "step": 35863,
      "training_step_time": 7.642345905303955
    },
    {
      "epoch": 3.420257568359375e-05,
      "model_forward_time": 1.764134168624878,
      "step": 35864
    },
    {
      "epoch": 3.420257568359375e-05,
      "step": 35864,
      "training_step_time": 7.950397253036499
    },
    {
      "epoch": 3.420352935791016e-05,
      "model_forward_time": 1.763465404510498,
      "step": 35865
    },
    {
      "epoch": 3.420352935791016e-05,
      "step": 35865,
      "training_step_time": 8.341692447662354
    },
    {
      "epoch": 3.420448303222656e-05,
      "model_forward_time": 1.7868528366088867,
      "step": 35866
    },
    {
      "epoch": 3.420448303222656e-05,
      "step": 35866,
      "training_step_time": 7.6861302852630615
    },
    {
      "epoch": 3.4205436706542966e-05,
      "model_forward_time": 2.2843916416168213,
      "step": 35867
    },
    {
      "epoch": 3.4205436706542966e-05,
      "step": 35867,
      "training_step_time": 8.413890361785889
    },
    {
      "epoch": 3.420639038085938e-05,
      "model_forward_time": 1.763418197631836,
      "step": 35868
    },
    {
      "epoch": 3.420639038085938e-05,
      "step": 35868,
      "training_step_time": 7.632025957107544
    },
    {
      "epoch": 3.420734405517578e-05,
      "model_forward_time": 1.7575411796569824,
      "step": 35869
    },
    {
      "epoch": 3.420734405517578e-05,
      "step": 35869,
      "training_step_time": 7.273137092590332
    },
    {
      "epoch": 3.4208297729492185e-05,
      "grad_norm": 0.16547398269176483,
      "learning_rate": 7.613338661014763e-06,
      "loss": 0.06,
      "step": 35870
    },
    {
      "epoch": 3.4208297729492185e-05,
      "model_forward_time": 1.7524082660675049,
      "step": 35870
    },
    {
      "dynamics_loss_avg": 0.06179269775748253,
      "epoch": 3.4208297729492185e-05,
      "step": 35870
    },
    {
      "action_loss_avg": 0.002022186847170815,
      "epoch": 3.4208297729492185e-05,
      "step": 35870
    },
    {
      "epoch": 3.4208297729492185e-05,
      "step": 35870,
      "training_step_time": 7.465043306350708
    },
    {
      "epoch": 3.4209251403808596e-05,
      "model_forward_time": 1.7512001991271973,
      "step": 35871
    },
    {
      "epoch": 3.4209251403808596e-05,
      "step": 35871,
      "training_step_time": 6.858752727508545
    },
    {
      "epoch": 3.4210205078125e-05,
      "model_forward_time": 1.7541816234588623,
      "step": 35872
    },
    {
      "epoch": 3.4210205078125e-05,
      "step": 35872,
      "training_step_time": 7.6170361042022705
    },
    {
      "epoch": 3.4211158752441404e-05,
      "model_forward_time": 1.7622323036193848,
      "step": 35873
    },
    {
      "epoch": 3.4211158752441404e-05,
      "step": 35873,
      "training_step_time": 8.604055643081665
    },
    {
      "epoch": 3.4212112426757815e-05,
      "model_forward_time": 1.7587974071502686,
      "step": 35874
    },
    {
      "epoch": 3.4212112426757815e-05,
      "step": 35874,
      "training_step_time": 7.603586912155151
    },
    {
      "epoch": 3.421306610107422e-05,
      "model_forward_time": 1.7520291805267334,
      "step": 35875
    },
    {
      "epoch": 3.421306610107422e-05,
      "step": 35875,
      "training_step_time": 8.915949583053589
    },
    {
      "epoch": 3.421401977539062e-05,
      "model_forward_time": 2.2902843952178955,
      "step": 35876
    },
    {
      "epoch": 3.421401977539062e-05,
      "step": 35876,
      "training_step_time": 7.98981499671936
    },
    {
      "epoch": 3.4214973449707034e-05,
      "model_forward_time": 1.7707552909851074,
      "step": 35877
    },
    {
      "epoch": 3.4214973449707034e-05,
      "step": 35877,
      "training_step_time": 8.309292554855347
    },
    {
      "epoch": 3.421592712402344e-05,
      "model_forward_time": 1.7564334869384766,
      "step": 35878
    },
    {
      "epoch": 3.421592712402344e-05,
      "step": 35878,
      "training_step_time": 7.321748733520508
    },
    {
      "epoch": 3.421688079833984e-05,
      "model_forward_time": 1.7553620338439941,
      "step": 35879
    },
    {
      "epoch": 3.421688079833984e-05,
      "step": 35879,
      "training_step_time": 7.5636820793151855
    },
    {
      "epoch": 3.421783447265625e-05,
      "grad_norm": 0.20277591049671173,
      "learning_rate": 7.611928874814686e-06,
      "loss": 0.0603,
      "step": 35880
    },
    {
      "epoch": 3.421783447265625e-05,
      "model_forward_time": 1.7554292678833008,
      "step": 35880
    },
    {
      "dynamics_loss_avg": 0.05646351594477892,
      "epoch": 3.421783447265625e-05,
      "step": 35880
    },
    {
      "action_loss_avg": 0.002633804292418063,
      "epoch": 3.421783447265625e-05,
      "step": 35880
    },
    {
      "epoch": 3.421783447265625e-05,
      "step": 35880,
      "training_step_time": 6.899734020233154
    },
    {
      "epoch": 3.4218788146972656e-05,
      "model_forward_time": 1.7708768844604492,
      "step": 35881
    },
    {
      "epoch": 3.4218788146972656e-05,
      "step": 35881,
      "training_step_time": 7.6040730476379395
    },
    {
      "epoch": 3.421974182128906e-05,
      "model_forward_time": 1.7572686672210693,
      "step": 35882
    },
    {
      "epoch": 3.421974182128906e-05,
      "step": 35882,
      "training_step_time": 7.953207731246948
    },
    {
      "epoch": 3.422069549560547e-05,
      "model_forward_time": 1.7555224895477295,
      "step": 35883
    },
    {
      "epoch": 3.422069549560547e-05,
      "step": 35883,
      "training_step_time": 7.7325499057769775
    },
    {
      "epoch": 3.4221649169921875e-05,
      "model_forward_time": 1.759448766708374,
      "step": 35884
    },
    {
      "epoch": 3.4221649169921875e-05,
      "step": 35884,
      "training_step_time": 7.8066442012786865
    },
    {
      "epoch": 3.422260284423828e-05,
      "model_forward_time": 2.3202810287475586,
      "step": 35885
    },
    {
      "epoch": 3.422260284423828e-05,
      "step": 35885,
      "training_step_time": 10.456301212310791
    },
    {
      "epoch": 3.422355651855469e-05,
      "model_forward_time": 1.7710044384002686,
      "step": 35886
    },
    {
      "epoch": 3.422355651855469e-05,
      "step": 35886,
      "training_step_time": 9.064931392669678
    },
    {
      "epoch": 3.4224510192871094e-05,
      "model_forward_time": 1.7556264400482178,
      "step": 35887
    },
    {
      "epoch": 3.4224510192871094e-05,
      "step": 35887,
      "training_step_time": 7.612721681594849
    },
    {
      "epoch": 3.42254638671875e-05,
      "model_forward_time": 1.7571020126342773,
      "step": 35888
    },
    {
      "epoch": 3.42254638671875e-05,
      "step": 35888,
      "training_step_time": 7.538681507110596
    },
    {
      "epoch": 3.422641754150391e-05,
      "model_forward_time": 1.7515196800231934,
      "step": 35889
    },
    {
      "epoch": 3.422641754150391e-05,
      "step": 35889,
      "training_step_time": 6.780277490615845
    },
    {
      "epoch": 3.422737121582031e-05,
      "grad_norm": 0.11476625502109528,
      "learning_rate": 7.610518802977995e-06,
      "loss": 0.0613,
      "step": 35890
    },
    {
      "epoch": 3.422737121582031e-05,
      "model_forward_time": 1.7668113708496094,
      "step": 35890
    },
    {
      "dynamics_loss_avg": 0.052103718370199205,
      "epoch": 3.422737121582031e-05,
      "step": 35890
    },
    {
      "action_loss_avg": 0.0014888380945194512,
      "epoch": 3.422737121582031e-05,
      "step": 35890
    },
    {
      "epoch": 3.422737121582031e-05,
      "step": 35890,
      "training_step_time": 6.810030937194824
    },
    {
      "epoch": 3.422832489013672e-05,
      "model_forward_time": 1.7519645690917969,
      "step": 35891
    },
    {
      "epoch": 3.422832489013672e-05,
      "step": 35891,
      "training_step_time": 7.851750612258911
    },
    {
      "epoch": 3.422927856445313e-05,
      "model_forward_time": 1.759828805923462,
      "step": 35892
    },
    {
      "epoch": 3.422927856445313e-05,
      "step": 35892,
      "training_step_time": 7.641317129135132
    },
    {
      "epoch": 3.423023223876953e-05,
      "model_forward_time": 1.7584867477416992,
      "step": 35893
    },
    {
      "epoch": 3.423023223876953e-05,
      "step": 35893,
      "training_step_time": 9.270043849945068
    },
    {
      "epoch": 3.4231185913085936e-05,
      "model_forward_time": 2.3411989212036133,
      "step": 35894
    },
    {
      "epoch": 3.4231185913085936e-05,
      "step": 35894,
      "training_step_time": 10.443661451339722
    },
    {
      "epoch": 3.423213958740235e-05,
      "model_forward_time": 1.7530951499938965,
      "step": 35895
    },
    {
      "epoch": 3.423213958740235e-05,
      "step": 35895,
      "training_step_time": 11.339504480361938
    },
    {
      "epoch": 3.423309326171875e-05,
      "model_forward_time": 1.7634615898132324,
      "step": 35896
    },
    {
      "epoch": 3.423309326171875e-05,
      "step": 35896,
      "training_step_time": 7.567861557006836
    },
    {
      "epoch": 3.4234046936035155e-05,
      "model_forward_time": 1.7795438766479492,
      "step": 35897
    },
    {
      "epoch": 3.4234046936035155e-05,
      "step": 35897,
      "training_step_time": 7.464087247848511
    },
    {
      "epoch": 3.4235000610351566e-05,
      "model_forward_time": 1.7549877166748047,
      "step": 35898
    },
    {
      "epoch": 3.4235000610351566e-05,
      "step": 35898,
      "training_step_time": 6.773707866668701
    },
    {
      "epoch": 3.423595428466797e-05,
      "model_forward_time": 1.7599904537200928,
      "step": 35899
    },
    {
      "epoch": 3.423595428466797e-05,
      "step": 35899,
      "training_step_time": 6.76200795173645
    },
    {
      "epoch": 3.4236907958984374e-05,
      "grad_norm": 0.09956617653369904,
      "learning_rate": 7.609108445658892e-06,
      "loss": 0.0635,
      "step": 35900
    },
    {
      "epoch": 3.4236907958984374e-05,
      "model_forward_time": 1.7524299621582031,
      "step": 35900
    },
    {
      "dynamics_loss_avg": 0.06127294078469277,
      "epoch": 3.4236907958984374e-05,
      "step": 35900
    },
    {
      "action_loss_avg": 0.0017654723429586737,
      "epoch": 3.4236907958984374e-05,
      "step": 35900
    },
    {
      "epoch": 3.4236907958984374e-05,
      "step": 35900,
      "training_step_time": 9.148638248443604
    },
    {
      "epoch": 3.4237861633300784e-05,
      "model_forward_time": 1.7522461414337158,
      "step": 35901
    },
    {
      "epoch": 3.4237861633300784e-05,
      "step": 35901,
      "training_step_time": 7.607000827789307
    },
    {
      "epoch": 3.423881530761719e-05,
      "model_forward_time": 1.7556936740875244,
      "step": 35902
    },
    {
      "epoch": 3.423881530761719e-05,
      "step": 35902,
      "training_step_time": 7.661761999130249
    },
    {
      "epoch": 3.423976898193359e-05,
      "model_forward_time": 2.3280539512634277,
      "step": 35903
    },
    {
      "epoch": 3.423976898193359e-05,
      "step": 35903,
      "training_step_time": 9.714284658432007
    },
    {
      "epoch": 3.424072265625e-05,
      "model_forward_time": 1.7572295665740967,
      "step": 35904
    },
    {
      "epoch": 3.424072265625e-05,
      "step": 35904,
      "training_step_time": 9.419935464859009
    },
    {
      "epoch": 3.424167633056641e-05,
      "model_forward_time": 1.749150037765503,
      "step": 35905
    },
    {
      "epoch": 3.424167633056641e-05,
      "step": 35905,
      "training_step_time": 7.624329090118408
    },
    {
      "epoch": 3.424263000488281e-05,
      "model_forward_time": 1.7513353824615479,
      "step": 35906
    },
    {
      "epoch": 3.424263000488281e-05,
      "step": 35906,
      "training_step_time": 7.533246278762817
    },
    {
      "epoch": 3.4243583679199215e-05,
      "model_forward_time": 1.3917255401611328,
      "step": 35907
    },
    {
      "epoch": 3.4243583679199215e-05,
      "step": 35907,
      "training_step_time": 6.826056241989136
    },
    {
      "epoch": 3.4244537353515626e-05,
      "model_forward_time": 1.7553331851959229,
      "step": 35908
    },
    {
      "epoch": 3.4244537353515626e-05,
      "step": 35908,
      "training_step_time": 6.7585320472717285
    },
    {
      "epoch": 3.424549102783203e-05,
      "model_forward_time": 1.757049322128296,
      "step": 35909
    },
    {
      "epoch": 3.424549102783203e-05,
      "step": 35909,
      "training_step_time": 7.582622051239014
    },
    {
      "epoch": 3.4246444702148434e-05,
      "grad_norm": 0.1428329348564148,
      "learning_rate": 7.607697803011613e-06,
      "loss": 0.06,
      "step": 35910
    },
    {
      "epoch": 3.4246444702148434e-05,
      "model_forward_time": 1.7497138977050781,
      "step": 35910
    },
    {
      "dynamics_loss_avg": 0.04624805320054293,
      "epoch": 3.4246444702148434e-05,
      "step": 35910
    },
    {
      "action_loss_avg": 0.0015959786251187325,
      "epoch": 3.4246444702148434e-05,
      "step": 35910
    },
    {
      "epoch": 3.4246444702148434e-05,
      "step": 35910,
      "training_step_time": 8.828951120376587
    },
    {
      "epoch": 3.4247398376464845e-05,
      "model_forward_time": 1.7591633796691895,
      "step": 35911
    },
    {
      "epoch": 3.4247398376464845e-05,
      "step": 35911,
      "training_step_time": 8.59363317489624
    },
    {
      "epoch": 3.424835205078125e-05,
      "model_forward_time": 1.762202501296997,
      "step": 35912
    },
    {
      "epoch": 3.424835205078125e-05,
      "step": 35912,
      "training_step_time": 8.874003171920776
    },
    {
      "epoch": 3.424930572509765e-05,
      "model_forward_time": 2.2914958000183105,
      "step": 35913
    },
    {
      "epoch": 3.424930572509765e-05,
      "step": 35913,
      "training_step_time": 8.672950983047485
    },
    {
      "epoch": 3.4250259399414064e-05,
      "model_forward_time": 1.7542035579681396,
      "step": 35914
    },
    {
      "epoch": 3.4250259399414064e-05,
      "step": 35914,
      "training_step_time": 9.441514730453491
    },
    {
      "epoch": 3.425121307373047e-05,
      "model_forward_time": 1.761821985244751,
      "step": 35915
    },
    {
      "epoch": 3.425121307373047e-05,
      "step": 35915,
      "training_step_time": 7.610564470291138
    },
    {
      "epoch": 3.425216674804687e-05,
      "model_forward_time": 1.7452812194824219,
      "step": 35916
    },
    {
      "epoch": 3.425216674804687e-05,
      "step": 35916,
      "training_step_time": 7.559982776641846
    },
    {
      "epoch": 3.425312042236328e-05,
      "model_forward_time": 1.7593576908111572,
      "step": 35917
    },
    {
      "epoch": 3.425312042236328e-05,
      "step": 35917,
      "training_step_time": 6.965856075286865
    },
    {
      "epoch": 3.425407409667969e-05,
      "model_forward_time": 1.7541241645812988,
      "step": 35918
    },
    {
      "epoch": 3.425407409667969e-05,
      "step": 35918,
      "training_step_time": 9.203395366668701
    },
    {
      "epoch": 3.425502777099609e-05,
      "model_forward_time": 1.7511568069458008,
      "step": 35919
    },
    {
      "epoch": 3.425502777099609e-05,
      "step": 35919,
      "training_step_time": 8.028545618057251
    },
    {
      "epoch": 3.42559814453125e-05,
      "grad_norm": 0.18555223941802979,
      "learning_rate": 7.606286875190422e-06,
      "loss": 0.0615,
      "step": 35920
    },
    {
      "epoch": 3.42559814453125e-05,
      "model_forward_time": 1.762648582458496,
      "step": 35920
    },
    {
      "dynamics_loss_avg": 0.04353519976139068,
      "epoch": 3.42559814453125e-05,
      "step": 35920
    },
    {
      "action_loss_avg": 0.0013343530881684273,
      "epoch": 3.42559814453125e-05,
      "step": 35920
    },
    {
      "epoch": 3.42559814453125e-05,
      "step": 35920,
      "training_step_time": 7.755129098892212
    },
    {
      "epoch": 3.4256935119628906e-05,
      "model_forward_time": 1.7475945949554443,
      "step": 35921
    },
    {
      "epoch": 3.4256935119628906e-05,
      "step": 35921,
      "training_step_time": 8.796319961547852
    },
    {
      "epoch": 3.425788879394531e-05,
      "model_forward_time": 2.3152928352355957,
      "step": 35922
    },
    {
      "epoch": 3.425788879394531e-05,
      "step": 35922,
      "training_step_time": 9.277437210083008
    },
    {
      "epoch": 3.425884246826172e-05,
      "model_forward_time": 1.7631616592407227,
      "step": 35923
    },
    {
      "epoch": 3.425884246826172e-05,
      "step": 35923,
      "training_step_time": 7.948225975036621
    },
    {
      "epoch": 3.4259796142578125e-05,
      "model_forward_time": 1.3968138694763184,
      "step": 35924
    },
    {
      "epoch": 3.4259796142578125e-05,
      "step": 35924,
      "training_step_time": 7.923033714294434
    },
    {
      "epoch": 3.426074981689453e-05,
      "model_forward_time": 1.7576899528503418,
      "step": 35925
    },
    {
      "epoch": 3.426074981689453e-05,
      "step": 35925,
      "training_step_time": 7.708737850189209
    },
    {
      "epoch": 3.426170349121094e-05,
      "model_forward_time": 1.7609386444091797,
      "step": 35926
    },
    {
      "epoch": 3.426170349121094e-05,
      "step": 35926,
      "training_step_time": 6.967923879623413
    },
    {
      "epoch": 3.426265716552734e-05,
      "model_forward_time": 1.061633825302124,
      "step": 35927
    },
    {
      "epoch": 3.426265716552734e-05,
      "step": 35927,
      "training_step_time": 7.743762254714966
    },
    {
      "epoch": 3.426361083984375e-05,
      "model_forward_time": 1.7554118633270264,
      "step": 35928
    },
    {
      "epoch": 3.426361083984375e-05,
      "step": 35928,
      "training_step_time": 8.216110706329346
    },
    {
      "epoch": 3.426456451416016e-05,
      "model_forward_time": 1.758643388748169,
      "step": 35929
    },
    {
      "epoch": 3.426456451416016e-05,
      "step": 35929,
      "training_step_time": 8.45599889755249
    },
    {
      "epoch": 3.426551818847656e-05,
      "grad_norm": 0.13632464408874512,
      "learning_rate": 7.604875662349618e-06,
      "loss": 0.0575,
      "step": 35930
    },
    {
      "epoch": 3.426551818847656e-05,
      "model_forward_time": 1.7509849071502686,
      "step": 35930
    },
    {
      "dynamics_loss_avg": 0.05536872651427984,
      "epoch": 3.426551818847656e-05,
      "step": 35930
    },
    {
      "action_loss_avg": 0.001886754756560549,
      "epoch": 3.426551818847656e-05,
      "step": 35930
    },
    {
      "epoch": 3.426551818847656e-05,
      "step": 35930,
      "training_step_time": 8.644109725952148
    },
    {
      "epoch": 3.4266471862792966e-05,
      "model_forward_time": 1.7991235256195068,
      "step": 35931
    },
    {
      "epoch": 3.4266471862792966e-05,
      "step": 35931,
      "training_step_time": 10.376292943954468
    },
    {
      "epoch": 3.426742553710938e-05,
      "model_forward_time": 1.752913236618042,
      "step": 35932
    },
    {
      "epoch": 3.426742553710938e-05,
      "step": 35932,
      "training_step_time": 8.95379114151001
    },
    {
      "epoch": 3.426837921142578e-05,
      "model_forward_time": 1.7518973350524902,
      "step": 35933
    },
    {
      "epoch": 3.426837921142578e-05,
      "step": 35933,
      "training_step_time": 7.84624171257019
    },
    {
      "epoch": 3.4269332885742185e-05,
      "model_forward_time": 1.7806763648986816,
      "step": 35934
    },
    {
      "epoch": 3.4269332885742185e-05,
      "step": 35934,
      "training_step_time": 7.536671876907349
    },
    {
      "epoch": 3.4270286560058596e-05,
      "model_forward_time": 1.7573778629302979,
      "step": 35935
    },
    {
      "epoch": 3.4270286560058596e-05,
      "step": 35935,
      "training_step_time": 6.9640281200408936
    },
    {
      "epoch": 3.4271240234375e-05,
      "model_forward_time": 1.7609484195709229,
      "step": 35936
    },
    {
      "epoch": 3.4271240234375e-05,
      "step": 35936,
      "training_step_time": 7.695625066757202
    },
    {
      "epoch": 3.4272193908691404e-05,
      "model_forward_time": 1.7696983814239502,
      "step": 35937
    },
    {
      "epoch": 3.4272193908691404e-05,
      "step": 35937,
      "training_step_time": 7.883847236633301
    },
    {
      "epoch": 3.4273147583007815e-05,
      "model_forward_time": 1.7645790576934814,
      "step": 35938
    },
    {
      "epoch": 3.4273147583007815e-05,
      "step": 35938,
      "training_step_time": 8.720718145370483
    },
    {
      "epoch": 3.427410125732422e-05,
      "model_forward_time": 1.7532320022583008,
      "step": 35939
    },
    {
      "epoch": 3.427410125732422e-05,
      "step": 35939,
      "training_step_time": 7.7957048416137695
    },
    {
      "epoch": 3.427505493164062e-05,
      "grad_norm": 0.139785498380661,
      "learning_rate": 7.603464164643526e-06,
      "loss": 0.0578,
      "step": 35940
    },
    {
      "epoch": 3.427505493164062e-05,
      "model_forward_time": 1.8073041439056396,
      "step": 35940
    },
    {
      "dynamics_loss_avg": 0.03400316787883639,
      "epoch": 3.427505493164062e-05,
      "step": 35940
    },
    {
      "action_loss_avg": 0.0012579143127368297,
      "epoch": 3.427505493164062e-05,
      "step": 35940
    },
    {
      "epoch": 3.427505493164062e-05,
      "step": 35940,
      "training_step_time": 10.02898907661438
    },
    {
      "epoch": 3.4276008605957034e-05,
      "model_forward_time": 1.7715845108032227,
      "step": 35941
    },
    {
      "epoch": 3.4276008605957034e-05,
      "step": 35941,
      "training_step_time": 8.445455074310303
    },
    {
      "epoch": 3.427696228027344e-05,
      "model_forward_time": 1.7629878520965576,
      "step": 35942
    },
    {
      "epoch": 3.427696228027344e-05,
      "step": 35942,
      "training_step_time": 7.689875364303589
    },
    {
      "epoch": 3.427791595458984e-05,
      "model_forward_time": 1.7589197158813477,
      "step": 35943
    },
    {
      "epoch": 3.427791595458984e-05,
      "step": 35943,
      "training_step_time": 7.817537069320679
    },
    {
      "epoch": 3.427886962890625e-05,
      "model_forward_time": 1.7552015781402588,
      "step": 35944
    },
    {
      "epoch": 3.427886962890625e-05,
      "step": 35944,
      "training_step_time": 7.586123943328857
    },
    {
      "epoch": 3.4279823303222657e-05,
      "model_forward_time": 1.756326675415039,
      "step": 35945
    },
    {
      "epoch": 3.4279823303222657e-05,
      "step": 35945,
      "training_step_time": 7.48981499671936
    },
    {
      "epoch": 3.428077697753906e-05,
      "model_forward_time": 1.754021167755127,
      "step": 35946
    },
    {
      "epoch": 3.428077697753906e-05,
      "step": 35946,
      "training_step_time": 8.065914392471313
    },
    {
      "epoch": 3.428173065185547e-05,
      "model_forward_time": 1.7568354606628418,
      "step": 35947
    },
    {
      "epoch": 3.428173065185547e-05,
      "step": 35947,
      "training_step_time": 8.44278335571289
    },
    {
      "epoch": 3.4282684326171875e-05,
      "model_forward_time": 1.7624168395996094,
      "step": 35948
    },
    {
      "epoch": 3.4282684326171875e-05,
      "step": 35948,
      "training_step_time": 8.6235933303833
    },
    {
      "epoch": 3.428363800048828e-05,
      "model_forward_time": 1.774141788482666,
      "step": 35949
    },
    {
      "epoch": 3.428363800048828e-05,
      "step": 35949,
      "training_step_time": 9.673173666000366
    },
    {
      "epoch": 3.428459167480469e-05,
      "grad_norm": 0.1342787742614746,
      "learning_rate": 7.602052382226508e-06,
      "loss": 0.0626,
      "step": 35950
    },
    {
      "epoch": 3.428459167480469e-05,
      "model_forward_time": 2.3028695583343506,
      "step": 35950
    },
    {
      "dynamics_loss_avg": 0.06569434665143489,
      "epoch": 3.428459167480469e-05,
      "step": 35950
    },
    {
      "action_loss_avg": 0.0020081235154066233,
      "epoch": 3.428459167480469e-05,
      "step": 35950
    },
    {
      "epoch": 3.428459167480469e-05,
      "step": 35950,
      "training_step_time": 7.960877180099487
    },
    {
      "epoch": 3.4285545349121094e-05,
      "model_forward_time": 1.7548558712005615,
      "step": 35951
    },
    {
      "epoch": 3.4285545349121094e-05,
      "step": 35951,
      "training_step_time": 7.843719482421875
    },
    {
      "epoch": 3.42864990234375e-05,
      "model_forward_time": 1.7610700130462646,
      "step": 35952
    },
    {
      "epoch": 3.42864990234375e-05,
      "step": 35952,
      "training_step_time": 7.798810958862305
    },
    {
      "epoch": 3.428745269775391e-05,
      "model_forward_time": 1.7577896118164062,
      "step": 35953
    },
    {
      "epoch": 3.428745269775391e-05,
      "step": 35953,
      "training_step_time": 7.641727685928345
    },
    {
      "epoch": 3.428840637207031e-05,
      "model_forward_time": 1.758068561553955,
      "step": 35954
    },
    {
      "epoch": 3.428840637207031e-05,
      "step": 35954,
      "training_step_time": 7.749498605728149
    },
    {
      "epoch": 3.428936004638672e-05,
      "model_forward_time": 1.7574632167816162,
      "step": 35955
    },
    {
      "epoch": 3.428936004638672e-05,
      "step": 35955,
      "training_step_time": 7.940388917922974
    },
    {
      "epoch": 3.429031372070313e-05,
      "model_forward_time": 1.7657208442687988,
      "step": 35956
    },
    {
      "epoch": 3.429031372070313e-05,
      "step": 35956,
      "training_step_time": 7.748617887496948
    },
    {
      "epoch": 3.429126739501953e-05,
      "model_forward_time": 1.7721829414367676,
      "step": 35957
    },
    {
      "epoch": 3.429126739501953e-05,
      "step": 35957,
      "training_step_time": 9.318336009979248
    },
    {
      "epoch": 3.4292221069335936e-05,
      "model_forward_time": 1.7456982135772705,
      "step": 35958
    },
    {
      "epoch": 3.4292221069335936e-05,
      "step": 35958,
      "training_step_time": 9.519325733184814
    },
    {
      "epoch": 3.429317474365235e-05,
      "model_forward_time": 2.2870495319366455,
      "step": 35959
    },
    {
      "epoch": 3.429317474365235e-05,
      "step": 35959,
      "training_step_time": 8.361505508422852
    },
    {
      "epoch": 3.429412841796875e-05,
      "grad_norm": 0.1392812430858612,
      "learning_rate": 7.600640315252954e-06,
      "loss": 0.0577,
      "step": 35960
    },
    {
      "epoch": 3.429412841796875e-05,
      "model_forward_time": 1.3896527290344238,
      "step": 35960
    },
    {
      "dynamics_loss_avg": 0.058831927552819255,
      "epoch": 3.429412841796875e-05,
      "step": 35960
    },
    {
      "action_loss_avg": 0.0020086014294065533,
      "epoch": 3.429412841796875e-05,
      "step": 35960
    },
    {
      "epoch": 3.429412841796875e-05,
      "step": 35960,
      "training_step_time": 7.547407150268555
    },
    {
      "epoch": 3.4295082092285155e-05,
      "model_forward_time": 1.7418723106384277,
      "step": 35961
    },
    {
      "epoch": 3.4295082092285155e-05,
      "step": 35961,
      "training_step_time": 8.574482440948486
    },
    {
      "epoch": 3.4296035766601566e-05,
      "model_forward_time": 1.7543466091156006,
      "step": 35962
    },
    {
      "epoch": 3.4296035766601566e-05,
      "step": 35962,
      "training_step_time": 7.092428922653198
    },
    {
      "epoch": 3.429698944091797e-05,
      "model_forward_time": 1.3930950164794922,
      "step": 35963
    },
    {
      "epoch": 3.429698944091797e-05,
      "step": 35963,
      "training_step_time": 7.481608152389526
    },
    {
      "epoch": 3.4297943115234374e-05,
      "model_forward_time": 1.7518036365509033,
      "step": 35964
    },
    {
      "epoch": 3.4297943115234374e-05,
      "step": 35964,
      "training_step_time": 8.407016277313232
    },
    {
      "epoch": 3.4298896789550785e-05,
      "model_forward_time": 1.7658939361572266,
      "step": 35965
    },
    {
      "epoch": 3.4298896789550785e-05,
      "step": 35965,
      "training_step_time": 8.835344314575195
    },
    {
      "epoch": 3.429985046386719e-05,
      "model_forward_time": 1.770287275314331,
      "step": 35966
    },
    {
      "epoch": 3.429985046386719e-05,
      "step": 35966,
      "training_step_time": 8.254766941070557
    },
    {
      "epoch": 3.430080413818359e-05,
      "model_forward_time": 1.745711326599121,
      "step": 35967
    },
    {
      "epoch": 3.430080413818359e-05,
      "step": 35967,
      "training_step_time": 9.120257377624512
    },
    {
      "epoch": 3.43017578125e-05,
      "model_forward_time": 2.258026599884033,
      "step": 35968
    },
    {
      "epoch": 3.43017578125e-05,
      "step": 35968,
      "training_step_time": 9.200085878372192
    },
    {
      "epoch": 3.430271148681641e-05,
      "model_forward_time": 1.7502305507659912,
      "step": 35969
    },
    {
      "epoch": 3.430271148681641e-05,
      "step": 35969,
      "training_step_time": 7.649617671966553
    },
    {
      "epoch": 3.430366516113281e-05,
      "grad_norm": 0.17317987978458405,
      "learning_rate": 7.599227963877285e-06,
      "loss": 0.0601,
      "step": 35970
    },
    {
      "epoch": 3.430366516113281e-05,
      "model_forward_time": 1.7284736633300781,
      "step": 35970
    },
    {
      "dynamics_loss_avg": 0.06859994344413281,
      "epoch": 3.430366516113281e-05,
      "step": 35970
    },
    {
      "action_loss_avg": 0.0026546785607933997,
      "epoch": 3.430366516113281e-05,
      "step": 35970
    },
    {
      "epoch": 3.430366516113281e-05,
      "step": 35970,
      "training_step_time": 7.542618751525879
    },
    {
      "epoch": 3.4304618835449215e-05,
      "model_forward_time": 1.7257423400878906,
      "step": 35971
    },
    {
      "epoch": 3.4304618835449215e-05,
      "step": 35971,
      "training_step_time": 7.575921058654785
    },
    {
      "epoch": 3.4305572509765626e-05,
      "model_forward_time": 1.7356419563293457,
      "step": 35972
    },
    {
      "epoch": 3.4305572509765626e-05,
      "step": 35972,
      "training_step_time": 7.523358583450317
    },
    {
      "epoch": 3.430652618408203e-05,
      "model_forward_time": 1.7404274940490723,
      "step": 35973
    },
    {
      "epoch": 3.430652618408203e-05,
      "step": 35973,
      "training_step_time": 7.448738098144531
    },
    {
      "epoch": 3.4307479858398434e-05,
      "model_forward_time": 1.7411901950836182,
      "step": 35974
    },
    {
      "epoch": 3.4307479858398434e-05,
      "step": 35974,
      "training_step_time": 7.870861053466797
    },
    {
      "epoch": 3.4308433532714845e-05,
      "model_forward_time": 1.7363147735595703,
      "step": 35975
    },
    {
      "epoch": 3.4308433532714845e-05,
      "step": 35975,
      "training_step_time": 7.597161054611206
    },
    {
      "epoch": 3.430938720703125e-05,
      "model_forward_time": 1.3603947162628174,
      "step": 35976
    },
    {
      "epoch": 3.430938720703125e-05,
      "step": 35976,
      "training_step_time": 9.365750551223755
    },
    {
      "epoch": 3.431034088134765e-05,
      "model_forward_time": 2.312274217605591,
      "step": 35977
    },
    {
      "epoch": 3.431034088134765e-05,
      "step": 35977,
      "training_step_time": 9.605437994003296
    },
    {
      "epoch": 3.4311294555664064e-05,
      "model_forward_time": 1.7334144115447998,
      "step": 35978
    },
    {
      "epoch": 3.4311294555664064e-05,
      "step": 35978,
      "training_step_time": 8.667701959609985
    },
    {
      "epoch": 3.431224822998047e-05,
      "model_forward_time": 1.7496898174285889,
      "step": 35979
    },
    {
      "epoch": 3.431224822998047e-05,
      "step": 35979,
      "training_step_time": 7.381315231323242
    },
    {
      "epoch": 3.431320190429687e-05,
      "grad_norm": 0.16659678518772125,
      "learning_rate": 7.597815328253955e-06,
      "loss": 0.0598,
      "step": 35980
    },
    {
      "epoch": 3.431320190429687e-05,
      "model_forward_time": 1.7667405605316162,
      "step": 35980
    },
    {
      "dynamics_loss_avg": 0.04766624253243208,
      "epoch": 3.431320190429687e-05,
      "step": 35980
    },
    {
      "action_loss_avg": 0.0015284947032341734,
      "epoch": 3.431320190429687e-05,
      "step": 35980
    },
    {
      "epoch": 3.431320190429687e-05,
      "step": 35980,
      "training_step_time": 7.618195533752441
    },
    {
      "epoch": 3.431415557861328e-05,
      "model_forward_time": 1.7495784759521484,
      "step": 35981
    },
    {
      "epoch": 3.431415557861328e-05,
      "step": 35981,
      "training_step_time": 7.575758695602417
    },
    {
      "epoch": 3.431510925292969e-05,
      "model_forward_time": 1.7470595836639404,
      "step": 35982
    },
    {
      "epoch": 3.431510925292969e-05,
      "step": 35982,
      "training_step_time": 7.413904428482056
    },
    {
      "epoch": 3.431606292724609e-05,
      "model_forward_time": 1.7513368129730225,
      "step": 35983
    },
    {
      "epoch": 3.431606292724609e-05,
      "step": 35983,
      "training_step_time": 8.257818460464478
    },
    {
      "epoch": 3.43170166015625e-05,
      "model_forward_time": 1.7568206787109375,
      "step": 35984
    },
    {
      "epoch": 3.43170166015625e-05,
      "step": 35984,
      "training_step_time": 7.564803123474121
    },
    {
      "epoch": 3.4317970275878906e-05,
      "model_forward_time": 1.7367186546325684,
      "step": 35985
    },
    {
      "epoch": 3.4317970275878906e-05,
      "step": 35985,
      "training_step_time": 7.559983491897583
    },
    {
      "epoch": 3.431892395019531e-05,
      "model_forward_time": 2.3243024349212646,
      "step": 35986
    },
    {
      "epoch": 3.431892395019531e-05,
      "step": 35986,
      "training_step_time": 7.75948429107666
    },
    {
      "epoch": 3.431987762451172e-05,
      "model_forward_time": 1.7477679252624512,
      "step": 35987
    },
    {
      "epoch": 3.431987762451172e-05,
      "step": 35987,
      "training_step_time": 8.560945987701416
    },
    {
      "epoch": 3.4320831298828125e-05,
      "model_forward_time": 1.7472960948944092,
      "step": 35988
    },
    {
      "epoch": 3.4320831298828125e-05,
      "step": 35988,
      "training_step_time": 8.809171199798584
    },
    {
      "epoch": 3.432178497314453e-05,
      "model_forward_time": 1.7479076385498047,
      "step": 35989
    },
    {
      "epoch": 3.432178497314453e-05,
      "step": 35989,
      "training_step_time": 8.601199388504028
    },
    {
      "epoch": 3.432273864746094e-05,
      "grad_norm": 0.16870860755443573,
      "learning_rate": 7.596402408537444e-06,
      "loss": 0.0587,
      "step": 35990
    },
    {
      "epoch": 3.432273864746094e-05,
      "model_forward_time": 1.749939203262329,
      "step": 35990
    },
    {
      "dynamics_loss_avg": 0.05520085208117962,
      "epoch": 3.432273864746094e-05,
      "step": 35990
    },
    {
      "action_loss_avg": 0.0015862020256463439,
      "epoch": 3.432273864746094e-05,
      "step": 35990
    },
    {
      "epoch": 3.432273864746094e-05,
      "step": 35990,
      "training_step_time": 7.357731580734253
    },
    {
      "epoch": 3.4323692321777343e-05,
      "model_forward_time": 1.7538883686065674,
      "step": 35991
    },
    {
      "epoch": 3.4323692321777343e-05,
      "step": 35991,
      "training_step_time": 7.460340976715088
    },
    {
      "epoch": 3.432464599609375e-05,
      "model_forward_time": 1.7381529808044434,
      "step": 35992
    },
    {
      "epoch": 3.432464599609375e-05,
      "step": 35992,
      "training_step_time": 7.781651735305786
    },
    {
      "epoch": 3.432559967041016e-05,
      "model_forward_time": 1.714017391204834,
      "step": 35993
    },
    {
      "epoch": 3.432559967041016e-05,
      "step": 35993,
      "training_step_time": 8.631767272949219
    },
    {
      "epoch": 3.432655334472656e-05,
      "model_forward_time": 1.726884365081787,
      "step": 35994
    },
    {
      "epoch": 3.432655334472656e-05,
      "step": 35994,
      "training_step_time": 8.42716908454895
    },
    {
      "epoch": 3.4327507019042966e-05,
      "model_forward_time": 1.784480333328247,
      "step": 35995
    },
    {
      "epoch": 3.4327507019042966e-05,
      "step": 35995,
      "training_step_time": 8.49224591255188
    },
    {
      "epoch": 3.432846069335938e-05,
      "model_forward_time": 2.301469564437866,
      "step": 35996
    },
    {
      "epoch": 3.432846069335938e-05,
      "step": 35996,
      "training_step_time": 8.49379825592041
    },
    {
      "epoch": 3.432941436767578e-05,
      "model_forward_time": 1.731959581375122,
      "step": 35997
    },
    {
      "epoch": 3.432941436767578e-05,
      "step": 35997,
      "training_step_time": 7.686450481414795
    },
    {
      "epoch": 3.4330368041992185e-05,
      "model_forward_time": 1.7315363883972168,
      "step": 35998
    },
    {
      "epoch": 3.4330368041992185e-05,
      "step": 35998,
      "training_step_time": 7.745069980621338
    },
    {
      "epoch": 3.4331321716308596e-05,
      "model_forward_time": 1.7448320388793945,
      "step": 35999
    },
    {
      "epoch": 3.4331321716308596e-05,
      "step": 35999,
      "training_step_time": 7.575894355773926
    },
    {
      "epoch": 3.4332275390625e-05,
      "grad_norm": 0.12262341380119324,
      "learning_rate": 7.594989204882271e-06,
      "loss": 0.0555,
      "step": 36000
    },
    {
      "epoch": 3.4332275390625e-05,
      "model_forward_time": 1.7541425228118896,
      "step": 36000
    },
    {
      "dynamics_loss_avg": 0.04251511143520474,
      "epoch": 3.4332275390625e-05,
      "step": 36000
    },
    {
      "action_loss_avg": 0.0013824680529069156,
      "epoch": 3.4332275390625e-05,
      "step": 36000
    },
    {
      "epoch": 3.4332275390625e-05,
      "step": 36000,
      "training_step_time": 8.233666181564331
    },
    {
      "epoch": 3.4333229064941404e-05,
      "model_forward_time": 1.7724919319152832,
      "step": 36001
    },
    {
      "epoch": 3.4333229064941404e-05,
      "step": 36001,
      "training_step_time": 7.52573037147522
    },
    {
      "epoch": 3.4334182739257815e-05,
      "model_forward_time": 1.7513115406036377,
      "step": 36002
    },
    {
      "epoch": 3.4334182739257815e-05,
      "step": 36002,
      "training_step_time": 8.14561128616333
    },
    {
      "epoch": 3.433513641357422e-05,
      "model_forward_time": 1.7895019054412842,
      "step": 36003
    },
    {
      "epoch": 3.433513641357422e-05,
      "step": 36003,
      "training_step_time": 9.627874851226807
    },
    {
      "epoch": 3.433609008789062e-05,
      "model_forward_time": 2.7466909885406494,
      "step": 36004
    },
    {
      "epoch": 3.433609008789062e-05,
      "step": 36004,
      "training_step_time": 10.647433757781982
    },
    {
      "epoch": 3.4337043762207034e-05,
      "model_forward_time": 1.746291160583496,
      "step": 36005
    },
    {
      "epoch": 3.4337043762207034e-05,
      "step": 36005,
      "training_step_time": 7.702980279922485
    },
    {
      "epoch": 3.433799743652344e-05,
      "model_forward_time": 1.7482290267944336,
      "step": 36006
    },
    {
      "epoch": 3.433799743652344e-05,
      "step": 36006,
      "training_step_time": 8.120196104049683
    },
    {
      "epoch": 3.433895111083984e-05,
      "model_forward_time": 1.7514948844909668,
      "step": 36007
    },
    {
      "epoch": 3.433895111083984e-05,
      "step": 36007,
      "training_step_time": 7.513320446014404
    },
    {
      "epoch": 3.433990478515625e-05,
      "model_forward_time": 1.7451262474060059,
      "step": 36008
    },
    {
      "epoch": 3.433990478515625e-05,
      "step": 36008,
      "training_step_time": 7.655313968658447
    },
    {
      "epoch": 3.4340858459472657e-05,
      "model_forward_time": 1.741934061050415,
      "step": 36009
    },
    {
      "epoch": 3.4340858459472657e-05,
      "step": 36009,
      "training_step_time": 7.516083478927612
    },
    {
      "epoch": 3.434181213378906e-05,
      "grad_norm": 0.16332392394542694,
      "learning_rate": 7.593575717442979e-06,
      "loss": 0.0594,
      "step": 36010
    },
    {
      "epoch": 3.434181213378906e-05,
      "model_forward_time": 1.743818759918213,
      "step": 36010
    },
    {
      "dynamics_loss_avg": 0.05941963009536266,
      "epoch": 3.434181213378906e-05,
      "step": 36010
    },
    {
      "action_loss_avg": 0.00225571789778769,
      "epoch": 3.434181213378906e-05,
      "step": 36010
    },
    {
      "epoch": 3.434181213378906e-05,
      "step": 36010,
      "training_step_time": 7.513104438781738
    },
    {
      "epoch": 3.434276580810547e-05,
      "model_forward_time": 1.7502803802490234,
      "step": 36011
    },
    {
      "epoch": 3.434276580810547e-05,
      "step": 36011,
      "training_step_time": 8.449113368988037
    },
    {
      "epoch": 3.4343719482421875e-05,
      "model_forward_time": 1.7974731922149658,
      "step": 36012
    },
    {
      "epoch": 3.4343719482421875e-05,
      "step": 36012,
      "training_step_time": 7.590475797653198
    },
    {
      "epoch": 3.434467315673828e-05,
      "model_forward_time": 2.2797048091888428,
      "step": 36013
    },
    {
      "epoch": 3.434467315673828e-05,
      "step": 36013,
      "training_step_time": 7.630379676818848
    },
    {
      "epoch": 3.434562683105469e-05,
      "model_forward_time": 1.7469820976257324,
      "step": 36014
    },
    {
      "epoch": 3.434562683105469e-05,
      "step": 36014,
      "training_step_time": 9.746359586715698
    },
    {
      "epoch": 3.4346580505371094e-05,
      "model_forward_time": 1.7391672134399414,
      "step": 36015
    },
    {
      "epoch": 3.4346580505371094e-05,
      "step": 36015,
      "training_step_time": 7.6623008251190186
    },
    {
      "epoch": 3.43475341796875e-05,
      "model_forward_time": 1.7527422904968262,
      "step": 36016
    },
    {
      "epoch": 3.43475341796875e-05,
      "step": 36016,
      "training_step_time": 7.492648124694824
    },
    {
      "epoch": 3.434848785400391e-05,
      "model_forward_time": 1.730252981185913,
      "step": 36017
    },
    {
      "epoch": 3.434848785400391e-05,
      "step": 36017,
      "training_step_time": 7.73228907585144
    },
    {
      "epoch": 3.434944152832031e-05,
      "model_forward_time": 1.7330257892608643,
      "step": 36018
    },
    {
      "epoch": 3.434944152832031e-05,
      "step": 36018,
      "training_step_time": 7.509713649749756
    },
    {
      "epoch": 3.435039520263672e-05,
      "model_forward_time": 1.7427890300750732,
      "step": 36019
    },
    {
      "epoch": 3.435039520263672e-05,
      "step": 36019,
      "training_step_time": 8.29430341720581
    },
    {
      "epoch": 3.435134887695313e-05,
      "grad_norm": 0.10756279528141022,
      "learning_rate": 7.592161946374147e-06,
      "loss": 0.0573,
      "step": 36020
    },
    {
      "epoch": 3.435134887695313e-05,
      "model_forward_time": 1.7234869003295898,
      "step": 36020
    },
    {
      "dynamics_loss_avg": 0.057882108725607395,
      "epoch": 3.435134887695313e-05,
      "step": 36020
    },
    {
      "action_loss_avg": 0.002154475843417458,
      "epoch": 3.435134887695313e-05,
      "step": 36020
    },
    {
      "epoch": 3.435134887695313e-05,
      "step": 36020,
      "training_step_time": 9.686860799789429
    },
    {
      "epoch": 3.435230255126953e-05,
      "model_forward_time": 1.7710323333740234,
      "step": 36021
    },
    {
      "epoch": 3.435230255126953e-05,
      "step": 36021,
      "training_step_time": 7.5754852294921875
    },
    {
      "epoch": 3.4353256225585936e-05,
      "model_forward_time": 1.7280795574188232,
      "step": 36022
    },
    {
      "epoch": 3.4353256225585936e-05,
      "step": 36022,
      "training_step_time": 8.262973070144653
    },
    {
      "epoch": 3.435420989990235e-05,
      "model_forward_time": 2.2788589000701904,
      "step": 36023
    },
    {
      "epoch": 3.435420989990235e-05,
      "step": 36023,
      "training_step_time": 8.349358797073364
    },
    {
      "epoch": 3.435516357421875e-05,
      "model_forward_time": 1.7345447540283203,
      "step": 36024
    },
    {
      "epoch": 3.435516357421875e-05,
      "step": 36024,
      "training_step_time": 9.402198314666748
    },
    {
      "epoch": 3.4356117248535155e-05,
      "model_forward_time": 1.745523452758789,
      "step": 36025
    },
    {
      "epoch": 3.4356117248535155e-05,
      "step": 36025,
      "training_step_time": 7.520327568054199
    },
    {
      "epoch": 3.4357070922851566e-05,
      "model_forward_time": 1.7310395240783691,
      "step": 36026
    },
    {
      "epoch": 3.4357070922851566e-05,
      "step": 36026,
      "training_step_time": 7.622911214828491
    },
    {
      "epoch": 3.435802459716797e-05,
      "model_forward_time": 1.7319815158843994,
      "step": 36027
    },
    {
      "epoch": 3.435802459716797e-05,
      "step": 36027,
      "training_step_time": 8.259602546691895
    },
    {
      "epoch": 3.4358978271484374e-05,
      "model_forward_time": 1.7346575260162354,
      "step": 36028
    },
    {
      "epoch": 3.4358978271484374e-05,
      "step": 36028,
      "training_step_time": 7.434794664382935
    },
    {
      "epoch": 3.4359931945800785e-05,
      "model_forward_time": 1.733428716659546,
      "step": 36029
    },
    {
      "epoch": 3.4359931945800785e-05,
      "step": 36029,
      "training_step_time": 7.620292663574219
    },
    {
      "epoch": 3.436088562011719e-05,
      "grad_norm": 0.11982791125774384,
      "learning_rate": 7.590747891830381e-06,
      "loss": 0.0606,
      "step": 36030
    },
    {
      "epoch": 3.436088562011719e-05,
      "model_forward_time": 1.7649915218353271,
      "step": 36030
    },
    {
      "dynamics_loss_avg": 0.05735105164349079,
      "epoch": 3.436088562011719e-05,
      "step": 36030
    },
    {
      "action_loss_avg": 0.001437400616123341,
      "epoch": 3.436088562011719e-05,
      "step": 36030
    },
    {
      "epoch": 3.436088562011719e-05,
      "step": 36030,
      "training_step_time": 7.527988910675049
    },
    {
      "epoch": 3.436183929443359e-05,
      "model_forward_time": 1.741858959197998,
      "step": 36031
    },
    {
      "epoch": 3.436183929443359e-05,
      "step": 36031,
      "training_step_time": 7.5784993171691895
    },
    {
      "epoch": 3.436279296875e-05,
      "model_forward_time": 2.2861533164978027,
      "step": 36032
    },
    {
      "epoch": 3.436279296875e-05,
      "step": 36032,
      "training_step_time": 8.628753662109375
    },
    {
      "epoch": 3.436374664306641e-05,
      "model_forward_time": 1.7435338497161865,
      "step": 36033
    },
    {
      "epoch": 3.436374664306641e-05,
      "step": 36033,
      "training_step_time": 8.676328182220459
    },
    {
      "epoch": 3.436470031738281e-05,
      "model_forward_time": 1.736565113067627,
      "step": 36034
    },
    {
      "epoch": 3.436470031738281e-05,
      "step": 36034,
      "training_step_time": 8.115164518356323
    },
    {
      "epoch": 3.4365653991699215e-05,
      "model_forward_time": 1.728518009185791,
      "step": 36035
    },
    {
      "epoch": 3.4365653991699215e-05,
      "step": 36035,
      "training_step_time": 8.581374168395996
    },
    {
      "epoch": 3.4366607666015626e-05,
      "model_forward_time": 1.728987693786621,
      "step": 36036
    },
    {
      "epoch": 3.4366607666015626e-05,
      "step": 36036,
      "training_step_time": 7.54268217086792
    },
    {
      "epoch": 3.436756134033203e-05,
      "model_forward_time": 1.7296438217163086,
      "step": 36037
    },
    {
      "epoch": 3.436756134033203e-05,
      "step": 36037,
      "training_step_time": 7.456286191940308
    },
    {
      "epoch": 3.4368515014648434e-05,
      "model_forward_time": 1.7303693294525146,
      "step": 36038
    },
    {
      "epoch": 3.4368515014648434e-05,
      "step": 36038,
      "training_step_time": 8.130486965179443
    },
    {
      "epoch": 3.4369468688964845e-05,
      "model_forward_time": 1.7717299461364746,
      "step": 36039
    },
    {
      "epoch": 3.4369468688964845e-05,
      "step": 36039,
      "training_step_time": 8.618657350540161
    },
    {
      "epoch": 3.437042236328125e-05,
      "grad_norm": 0.17900660634040833,
      "learning_rate": 7.58933355396632e-06,
      "loss": 0.0596,
      "step": 36040
    },
    {
      "epoch": 3.437042236328125e-05,
      "model_forward_time": 1.724996566772461,
      "step": 36040
    },
    {
      "dynamics_loss_avg": 0.055295685306191444,
      "epoch": 3.437042236328125e-05,
      "step": 36040
    },
    {
      "action_loss_avg": 0.0010717049342929385,
      "epoch": 3.437042236328125e-05,
      "step": 36040
    },
    {
      "epoch": 3.437042236328125e-05,
      "step": 36040,
      "training_step_time": 9.352832078933716
    },
    {
      "epoch": 3.437137603759765e-05,
      "model_forward_time": 2.2591631412506104,
      "step": 36041
    },
    {
      "epoch": 3.437137603759765e-05,
      "step": 36041,
      "training_step_time": 8.490744352340698
    },
    {
      "epoch": 3.4372329711914064e-05,
      "model_forward_time": 1.7306721210479736,
      "step": 36042
    },
    {
      "epoch": 3.4372329711914064e-05,
      "step": 36042,
      "training_step_time": 7.779465436935425
    },
    {
      "epoch": 3.437328338623047e-05,
      "model_forward_time": 1.7330312728881836,
      "step": 36043
    },
    {
      "epoch": 3.437328338623047e-05,
      "step": 36043,
      "training_step_time": 7.626667022705078
    },
    {
      "epoch": 3.437423706054687e-05,
      "model_forward_time": 1.7400169372558594,
      "step": 36044
    },
    {
      "epoch": 3.437423706054687e-05,
      "step": 36044,
      "training_step_time": 7.529866933822632
    },
    {
      "epoch": 3.437519073486328e-05,
      "model_forward_time": 1.7465040683746338,
      "step": 36045
    },
    {
      "epoch": 3.437519073486328e-05,
      "step": 36045,
      "training_step_time": 7.631590127944946
    },
    {
      "epoch": 3.437614440917969e-05,
      "model_forward_time": 1.7439887523651123,
      "step": 36046
    },
    {
      "epoch": 3.437614440917969e-05,
      "step": 36046,
      "training_step_time": 7.404008626937866
    },
    {
      "epoch": 3.437709808349609e-05,
      "model_forward_time": 1.7508862018585205,
      "step": 36047
    },
    {
      "epoch": 3.437709808349609e-05,
      "step": 36047,
      "training_step_time": 8.56003999710083
    },
    {
      "epoch": 3.43780517578125e-05,
      "model_forward_time": 1.773911476135254,
      "step": 36048
    },
    {
      "epoch": 3.43780517578125e-05,
      "step": 36048,
      "training_step_time": 7.53142786026001
    },
    {
      "epoch": 3.4379005432128906e-05,
      "model_forward_time": 1.738698959350586,
      "step": 36049
    },
    {
      "epoch": 3.4379005432128906e-05,
      "step": 36049,
      "training_step_time": 7.54862117767334
    },
    {
      "epoch": 3.437995910644531e-05,
      "grad_norm": 0.09322445094585419,
      "learning_rate": 7.587918932936637e-06,
      "loss": 0.0628,
      "step": 36050
    },
    {
      "epoch": 3.437995910644531e-05,
      "model_forward_time": 2.2492456436157227,
      "step": 36050
    },
    {
      "dynamics_loss_avg": 0.07621935084462166,
      "epoch": 3.437995910644531e-05,
      "step": 36050
    },
    {
      "action_loss_avg": 0.0024182899738661944,
      "epoch": 3.437995910644531e-05,
      "step": 36050
    },
    {
      "epoch": 3.437995910644531e-05,
      "step": 36050,
      "training_step_time": 9.293151617050171
    },
    {
      "epoch": 3.438091278076172e-05,
      "model_forward_time": 1.7278480529785156,
      "step": 36051
    },
    {
      "epoch": 3.438091278076172e-05,
      "step": 36051,
      "training_step_time": 8.575686693191528
    },
    {
      "epoch": 3.4381866455078125e-05,
      "model_forward_time": 1.7289083003997803,
      "step": 36052
    },
    {
      "epoch": 3.4381866455078125e-05,
      "step": 36052,
      "training_step_time": 8.63779616355896
    },
    {
      "epoch": 3.438282012939453e-05,
      "model_forward_time": 1.7356479167938232,
      "step": 36053
    },
    {
      "epoch": 3.438282012939453e-05,
      "step": 36053,
      "training_step_time": 8.431484937667847
    },
    {
      "epoch": 3.438377380371094e-05,
      "model_forward_time": 1.7407951354980469,
      "step": 36054
    },
    {
      "epoch": 3.438377380371094e-05,
      "step": 36054,
      "training_step_time": 8.751376390457153
    },
    {
      "epoch": 3.4384727478027343e-05,
      "model_forward_time": 1.733785629272461,
      "step": 36055
    },
    {
      "epoch": 3.4384727478027343e-05,
      "step": 36055,
      "training_step_time": 7.466219902038574
    },
    {
      "epoch": 3.438568115234375e-05,
      "model_forward_time": 1.7377700805664062,
      "step": 36056
    },
    {
      "epoch": 3.438568115234375e-05,
      "step": 36056,
      "training_step_time": 7.519971609115601
    },
    {
      "epoch": 3.438663482666016e-05,
      "model_forward_time": 1.7744832038879395,
      "step": 36057
    },
    {
      "epoch": 3.438663482666016e-05,
      "step": 36057,
      "training_step_time": 7.81773567199707
    },
    {
      "epoch": 3.438758850097656e-05,
      "model_forward_time": 1.7386412620544434,
      "step": 36058
    },
    {
      "epoch": 3.438758850097656e-05,
      "step": 36058,
      "training_step_time": 7.567982196807861
    },
    {
      "epoch": 3.4388542175292966e-05,
      "model_forward_time": 2.267666816711426,
      "step": 36059
    },
    {
      "epoch": 3.4388542175292966e-05,
      "step": 36059,
      "training_step_time": 7.543006658554077
    },
    {
      "epoch": 3.438949584960938e-05,
      "grad_norm": 0.1625676304101944,
      "learning_rate": 7.586504028896029e-06,
      "loss": 0.0627,
      "step": 36060
    },
    {
      "epoch": 3.438949584960938e-05,
      "model_forward_time": 1.7227041721343994,
      "step": 36060
    },
    {
      "dynamics_loss_avg": 0.05372857060283422,
      "epoch": 3.438949584960938e-05,
      "step": 36060
    },
    {
      "action_loss_avg": 0.0021119322045706213,
      "epoch": 3.438949584960938e-05,
      "step": 36060
    },
    {
      "epoch": 3.438949584960938e-05,
      "step": 36060,
      "training_step_time": 9.696999311447144
    },
    {
      "epoch": 3.439044952392578e-05,
      "model_forward_time": 1.737450361251831,
      "step": 36061
    },
    {
      "epoch": 3.439044952392578e-05,
      "step": 36061,
      "training_step_time": 8.376625776290894
    },
    {
      "epoch": 3.4391403198242185e-05,
      "model_forward_time": 1.7471258640289307,
      "step": 36062
    },
    {
      "epoch": 3.4391403198242185e-05,
      "step": 36062,
      "training_step_time": 7.5589423179626465
    },
    {
      "epoch": 3.4392356872558596e-05,
      "model_forward_time": 1.7446131706237793,
      "step": 36063
    },
    {
      "epoch": 3.4392356872558596e-05,
      "step": 36063,
      "training_step_time": 7.61989688873291
    },
    {
      "epoch": 3.4393310546875e-05,
      "model_forward_time": 1.733292579650879,
      "step": 36064
    },
    {
      "epoch": 3.4393310546875e-05,
      "step": 36064,
      "training_step_time": 7.3547043800354
    },
    {
      "epoch": 3.4394264221191404e-05,
      "model_forward_time": 1.74006986618042,
      "step": 36065
    },
    {
      "epoch": 3.4394264221191404e-05,
      "step": 36065,
      "training_step_time": 7.515271186828613
    },
    {
      "epoch": 3.4395217895507815e-05,
      "model_forward_time": 1.7418897151947021,
      "step": 36066
    },
    {
      "epoch": 3.4395217895507815e-05,
      "step": 36066,
      "training_step_time": 7.857847213745117
    },
    {
      "epoch": 3.439617156982422e-05,
      "model_forward_time": 1.738218069076538,
      "step": 36067
    },
    {
      "epoch": 3.439617156982422e-05,
      "step": 36067,
      "training_step_time": 7.468343496322632
    },
    {
      "epoch": 3.439712524414062e-05,
      "model_forward_time": 2.2891831398010254,
      "step": 36068
    },
    {
      "epoch": 3.439712524414062e-05,
      "step": 36068,
      "training_step_time": 8.296007871627808
    },
    {
      "epoch": 3.4398078918457034e-05,
      "model_forward_time": 1.7251067161560059,
      "step": 36069
    },
    {
      "epoch": 3.4398078918457034e-05,
      "step": 36069,
      "training_step_time": 9.374205350875854
    },
    {
      "epoch": 3.439903259277344e-05,
      "grad_norm": 0.15140487253665924,
      "learning_rate": 7.585088841999229e-06,
      "loss": 0.0603,
      "step": 36070
    },
    {
      "epoch": 3.439903259277344e-05,
      "model_forward_time": 1.736088752746582,
      "step": 36070
    },
    {
      "dynamics_loss_avg": 0.03646421330049634,
      "epoch": 3.439903259277344e-05,
      "step": 36070
    },
    {
      "action_loss_avg": 0.0011929124419111758,
      "epoch": 3.439903259277344e-05,
      "step": 36070
    },
    {
      "epoch": 3.439903259277344e-05,
      "step": 36070,
      "training_step_time": 8.734205484390259
    },
    {
      "epoch": 3.439998626708984e-05,
      "model_forward_time": 1.736469030380249,
      "step": 36071
    },
    {
      "epoch": 3.439998626708984e-05,
      "step": 36071,
      "training_step_time": 7.472462177276611
    },
    {
      "epoch": 3.440093994140625e-05,
      "model_forward_time": 1.7226037979125977,
      "step": 36072
    },
    {
      "epoch": 3.440093994140625e-05,
      "step": 36072,
      "training_step_time": 7.6933698654174805
    },
    {
      "epoch": 3.440189361572266e-05,
      "model_forward_time": 1.7314400672912598,
      "step": 36073
    },
    {
      "epoch": 3.440189361572266e-05,
      "step": 36073,
      "training_step_time": 7.537309646606445
    },
    {
      "epoch": 3.440284729003906e-05,
      "model_forward_time": 1.7485787868499756,
      "step": 36074
    },
    {
      "epoch": 3.440284729003906e-05,
      "step": 36074,
      "training_step_time": 7.4229185581207275
    },
    {
      "epoch": 3.440380096435547e-05,
      "model_forward_time": 1.7386891841888428,
      "step": 36075
    },
    {
      "epoch": 3.440380096435547e-05,
      "step": 36075,
      "training_step_time": 8.797271251678467
    },
    {
      "epoch": 3.4404754638671875e-05,
      "model_forward_time": 1.7258491516113281,
      "step": 36076
    },
    {
      "epoch": 3.4404754638671875e-05,
      "step": 36076,
      "training_step_time": 7.538144826889038
    },
    {
      "epoch": 3.440570831298828e-05,
      "model_forward_time": 2.909569025039673,
      "step": 36077
    },
    {
      "epoch": 3.440570831298828e-05,
      "step": 36077,
      "training_step_time": 7.616423606872559
    },
    {
      "epoch": 3.440666198730469e-05,
      "model_forward_time": 1.8153729438781738,
      "step": 36078
    },
    {
      "epoch": 3.440666198730469e-05,
      "step": 36078,
      "training_step_time": 10.98805046081543
    },
    {
      "epoch": 3.4407615661621094e-05,
      "model_forward_time": 1.7467138767242432,
      "step": 36079
    },
    {
      "epoch": 3.4407615661621094e-05,
      "step": 36079,
      "training_step_time": 8.945163249969482
    },
    {
      "epoch": 3.44085693359375e-05,
      "grad_norm": 0.17252132296562195,
      "learning_rate": 7.583673372400999e-06,
      "loss": 0.0627,
      "step": 36080
    },
    {
      "epoch": 3.44085693359375e-05,
      "model_forward_time": 1.7344682216644287,
      "step": 36080
    },
    {
      "dynamics_loss_avg": 0.05396686196327209,
      "epoch": 3.44085693359375e-05,
      "step": 36080
    },
    {
      "action_loss_avg": 0.002725814693258144,
      "epoch": 3.44085693359375e-05,
      "step": 36080
    },
    {
      "epoch": 3.44085693359375e-05,
      "step": 36080,
      "training_step_time": 7.860040903091431
    },
    {
      "epoch": 3.440952301025391e-05,
      "model_forward_time": 1.7434923648834229,
      "step": 36081
    },
    {
      "epoch": 3.440952301025391e-05,
      "step": 36081,
      "training_step_time": 7.872390508651733
    },
    {
      "epoch": 3.441047668457031e-05,
      "model_forward_time": 1.7372980117797852,
      "step": 36082
    },
    {
      "epoch": 3.441047668457031e-05,
      "step": 36082,
      "training_step_time": 7.776735544204712
    },
    {
      "epoch": 3.441143035888672e-05,
      "model_forward_time": 1.734043836593628,
      "step": 36083
    },
    {
      "epoch": 3.441143035888672e-05,
      "step": 36083,
      "training_step_time": 7.593777894973755
    },
    {
      "epoch": 3.441238403320313e-05,
      "model_forward_time": 1.7301990985870361,
      "step": 36084
    },
    {
      "epoch": 3.441238403320313e-05,
      "step": 36084,
      "training_step_time": 8.01107144355774
    },
    {
      "epoch": 3.441333770751953e-05,
      "model_forward_time": 1.7377235889434814,
      "step": 36085
    },
    {
      "epoch": 3.441333770751953e-05,
      "step": 36085,
      "training_step_time": 8.402673482894897
    },
    {
      "epoch": 3.4414291381835936e-05,
      "model_forward_time": 2.884842872619629,
      "step": 36086
    },
    {
      "epoch": 3.4414291381835936e-05,
      "step": 36086,
      "training_step_time": 8.539321899414062
    },
    {
      "epoch": 3.441524505615235e-05,
      "model_forward_time": 1.8047845363616943,
      "step": 36087
    },
    {
      "epoch": 3.441524505615235e-05,
      "step": 36087,
      "training_step_time": 9.70438003540039
    },
    {
      "epoch": 3.441619873046875e-05,
      "model_forward_time": 1.7277753353118896,
      "step": 36088
    },
    {
      "epoch": 3.441619873046875e-05,
      "step": 36088,
      "training_step_time": 7.938006401062012
    },
    {
      "epoch": 3.4417152404785155e-05,
      "model_forward_time": 1.7422912120819092,
      "step": 36089
    },
    {
      "epoch": 3.4417152404785155e-05,
      "step": 36089,
      "training_step_time": 7.78877854347229
    },
    {
      "epoch": 3.4418106079101566e-05,
      "grad_norm": 0.1311105638742447,
      "learning_rate": 7.582257620256134e-06,
      "loss": 0.0632,
      "step": 36090
    },
    {
      "epoch": 3.4418106079101566e-05,
      "model_forward_time": 1.7266814708709717,
      "step": 36090
    },
    {
      "dynamics_loss_avg": 0.05568382041528821,
      "epoch": 3.4418106079101566e-05,
      "step": 36090
    },
    {
      "action_loss_avg": 0.0015402056742459536,
      "epoch": 3.4418106079101566e-05,
      "step": 36090
    },
    {
      "epoch": 3.4418106079101566e-05,
      "step": 36090,
      "training_step_time": 7.7782182693481445
    },
    {
      "epoch": 3.441905975341797e-05,
      "model_forward_time": 1.738645076751709,
      "step": 36091
    },
    {
      "epoch": 3.441905975341797e-05,
      "step": 36091,
      "training_step_time": 7.581358432769775
    },
    {
      "epoch": 3.4420013427734374e-05,
      "model_forward_time": 1.7321891784667969,
      "step": 36092
    },
    {
      "epoch": 3.4420013427734374e-05,
      "step": 36092,
      "training_step_time": 8.577699422836304
    },
    {
      "epoch": 3.4420967102050785e-05,
      "model_forward_time": 1.7318978309631348,
      "step": 36093
    },
    {
      "epoch": 3.4420967102050785e-05,
      "step": 36093,
      "training_step_time": 7.917448282241821
    },
    {
      "epoch": 3.442192077636719e-05,
      "model_forward_time": 1.7332441806793213,
      "step": 36094
    },
    {
      "epoch": 3.442192077636719e-05,
      "step": 36094,
      "training_step_time": 7.731263875961304
    },
    {
      "epoch": 3.442287445068359e-05,
      "model_forward_time": 2.886007785797119,
      "step": 36095
    },
    {
      "epoch": 3.442287445068359e-05,
      "step": 36095,
      "training_step_time": 8.587923765182495
    },
    {
      "epoch": 3.4423828125e-05,
      "model_forward_time": 1.7885005474090576,
      "step": 36096
    },
    {
      "epoch": 3.4423828125e-05,
      "step": 36096,
      "training_step_time": 8.242254972457886
    },
    {
      "epoch": 3.442478179931641e-05,
      "model_forward_time": 1.7323989868164062,
      "step": 36097
    },
    {
      "epoch": 3.442478179931641e-05,
      "step": 36097,
      "training_step_time": 7.910393238067627
    },
    {
      "epoch": 3.442573547363281e-05,
      "model_forward_time": 1.7282471656799316,
      "step": 36098
    },
    {
      "epoch": 3.442573547363281e-05,
      "step": 36098,
      "training_step_time": 7.803826332092285
    },
    {
      "epoch": 3.4426689147949216e-05,
      "model_forward_time": 1.7292380332946777,
      "step": 36099
    },
    {
      "epoch": 3.4426689147949216e-05,
      "step": 36099,
      "training_step_time": 8.222395420074463
    },
    {
      "epoch": 3.4427642822265626e-05,
      "grad_norm": 0.1630464494228363,
      "learning_rate": 7.580841585719459e-06,
      "loss": 0.0599,
      "step": 36100
    },
    {
      "epoch": 3.4427642822265626e-05,
      "model_forward_time": 1.7502257823944092,
      "step": 36100
    },
    {
      "dynamics_loss_avg": 0.06005702838301659,
      "epoch": 3.4427642822265626e-05,
      "step": 36100
    },
    {
      "action_loss_avg": 0.0017231797683052719,
      "epoch": 3.4427642822265626e-05,
      "step": 36100
    },
    {
      "epoch": 3.4427642822265626e-05,
      "step": 36100,
      "training_step_time": 7.413218021392822
    },
    {
      "epoch": 3.442859649658203e-05,
      "model_forward_time": 1.7579002380371094,
      "step": 36101
    },
    {
      "epoch": 3.442859649658203e-05,
      "step": 36101,
      "training_step_time": 7.751725435256958
    },
    {
      "epoch": 3.4429550170898434e-05,
      "model_forward_time": 1.7293603420257568,
      "step": 36102
    },
    {
      "epoch": 3.4429550170898434e-05,
      "step": 36102,
      "training_step_time": 7.611120223999023
    },
    {
      "epoch": 3.4430503845214845e-05,
      "model_forward_time": 1.7345788478851318,
      "step": 36103
    },
    {
      "epoch": 3.4430503845214845e-05,
      "step": 36103,
      "training_step_time": 7.701097249984741
    },
    {
      "epoch": 3.443145751953125e-05,
      "model_forward_time": 2.925905466079712,
      "step": 36104
    },
    {
      "epoch": 3.443145751953125e-05,
      "step": 36104,
      "training_step_time": 8.844263553619385
    },
    {
      "epoch": 3.443241119384765e-05,
      "model_forward_time": 1.8141734600067139,
      "step": 36105
    },
    {
      "epoch": 3.443241119384765e-05,
      "step": 36105,
      "training_step_time": 9.110397100448608
    },
    {
      "epoch": 3.4433364868164064e-05,
      "model_forward_time": 1.7514162063598633,
      "step": 36106
    },
    {
      "epoch": 3.4433364868164064e-05,
      "step": 36106,
      "training_step_time": 9.274872541427612
    },
    {
      "epoch": 3.443431854248047e-05,
      "model_forward_time": 1.7733564376831055,
      "step": 36107
    },
    {
      "epoch": 3.443431854248047e-05,
      "step": 36107,
      "training_step_time": 7.795897483825684
    },
    {
      "epoch": 3.443527221679687e-05,
      "model_forward_time": 1.7513468265533447,
      "step": 36108
    },
    {
      "epoch": 3.443527221679687e-05,
      "step": 36108,
      "training_step_time": 7.2953245639801025
    },
    {
      "epoch": 3.443622589111328e-05,
      "model_forward_time": 1.7530994415283203,
      "step": 36109
    },
    {
      "epoch": 3.443622589111328e-05,
      "step": 36109,
      "training_step_time": 7.6694395542144775
    },
    {
      "epoch": 3.443717956542969e-05,
      "grad_norm": 0.15964706242084503,
      "learning_rate": 7.579425268945826e-06,
      "loss": 0.0572,
      "step": 36110
    },
    {
      "epoch": 3.443717956542969e-05,
      "model_forward_time": 1.7501349449157715,
      "step": 36110
    },
    {
      "dynamics_loss_avg": 0.062167814001441,
      "epoch": 3.443717956542969e-05,
      "step": 36110
    },
    {
      "action_loss_avg": 0.00154322445159778,
      "epoch": 3.443717956542969e-05,
      "step": 36110
    },
    {
      "epoch": 3.443717956542969e-05,
      "step": 36110,
      "training_step_time": 7.712217092514038
    },
    {
      "epoch": 3.443813323974609e-05,
      "model_forward_time": 1.7582244873046875,
      "step": 36111
    },
    {
      "epoch": 3.443813323974609e-05,
      "step": 36111,
      "training_step_time": 8.015820026397705
    },
    {
      "epoch": 3.44390869140625e-05,
      "model_forward_time": 1.7545957565307617,
      "step": 36112
    },
    {
      "epoch": 3.44390869140625e-05,
      "step": 36112,
      "training_step_time": 7.681295871734619
    },
    {
      "epoch": 3.4440040588378906e-05,
      "model_forward_time": 2.9439868927001953,
      "step": 36113
    },
    {
      "epoch": 3.4440040588378906e-05,
      "step": 36113,
      "training_step_time": 8.803809642791748
    },
    {
      "epoch": 3.444099426269531e-05,
      "model_forward_time": 1.8005337715148926,
      "step": 36114
    },
    {
      "epoch": 3.444099426269531e-05,
      "step": 36114,
      "training_step_time": 8.194590091705322
    },
    {
      "epoch": 3.444194793701172e-05,
      "model_forward_time": 1.747528314590454,
      "step": 36115
    },
    {
      "epoch": 3.444194793701172e-05,
      "step": 36115,
      "training_step_time": 8.294298887252808
    },
    {
      "epoch": 3.4442901611328125e-05,
      "model_forward_time": 1.7557783126831055,
      "step": 36116
    },
    {
      "epoch": 3.4442901611328125e-05,
      "step": 36116,
      "training_step_time": 8.43124794960022
    },
    {
      "epoch": 3.444385528564453e-05,
      "model_forward_time": 1.7347209453582764,
      "step": 36117
    },
    {
      "epoch": 3.444385528564453e-05,
      "step": 36117,
      "training_step_time": 8.36935830116272
    },
    {
      "epoch": 3.444480895996094e-05,
      "model_forward_time": 1.7612123489379883,
      "step": 36118
    },
    {
      "epoch": 3.444480895996094e-05,
      "step": 36118,
      "training_step_time": 7.7861552238464355
    },
    {
      "epoch": 3.4445762634277344e-05,
      "model_forward_time": 1.7677679061889648,
      "step": 36119
    },
    {
      "epoch": 3.4445762634277344e-05,
      "step": 36119,
      "training_step_time": 7.690404176712036
    },
    {
      "epoch": 3.444671630859375e-05,
      "grad_norm": 0.154489204287529,
      "learning_rate": 7.5780086700901264e-06,
      "loss": 0.0579,
      "step": 36120
    },
    {
      "epoch": 3.444671630859375e-05,
      "model_forward_time": 1.7419626712799072,
      "step": 36120
    },
    {
      "dynamics_loss_avg": 0.060752162151038645,
      "epoch": 3.444671630859375e-05,
      "step": 36120
    },
    {
      "action_loss_avg": 0.0019162472744937986,
      "epoch": 3.444671630859375e-05,
      "step": 36120
    },
    {
      "epoch": 3.444671630859375e-05,
      "step": 36120,
      "training_step_time": 8.115944385528564
    },
    {
      "epoch": 3.444766998291016e-05,
      "model_forward_time": 1.7436413764953613,
      "step": 36121
    },
    {
      "epoch": 3.444766998291016e-05,
      "step": 36121,
      "training_step_time": 8.395112991333008
    },
    {
      "epoch": 3.444862365722656e-05,
      "model_forward_time": 2.949704885482788,
      "step": 36122
    },
    {
      "epoch": 3.444862365722656e-05,
      "step": 36122,
      "training_step_time": 8.650060653686523
    },
    {
      "epoch": 3.4449577331542966e-05,
      "model_forward_time": 1.8092443943023682,
      "step": 36123
    },
    {
      "epoch": 3.4449577331542966e-05,
      "step": 36123,
      "training_step_time": 7.57440972328186
    },
    {
      "epoch": 3.445053100585938e-05,
      "model_forward_time": 1.7460551261901855,
      "step": 36124
    },
    {
      "epoch": 3.445053100585938e-05,
      "step": 36124,
      "training_step_time": 7.849853992462158
    },
    {
      "epoch": 3.445148468017578e-05,
      "model_forward_time": 1.7389767169952393,
      "step": 36125
    },
    {
      "epoch": 3.445148468017578e-05,
      "step": 36125,
      "training_step_time": 9.7451651096344
    },
    {
      "epoch": 3.4452438354492185e-05,
      "model_forward_time": 1.7423856258392334,
      "step": 36126
    },
    {
      "epoch": 3.4452438354492185e-05,
      "step": 36126,
      "training_step_time": 8.822991132736206
    },
    {
      "epoch": 3.4453392028808596e-05,
      "model_forward_time": 1.7443890571594238,
      "step": 36127
    },
    {
      "epoch": 3.4453392028808596e-05,
      "step": 36127,
      "training_step_time": 7.830360651016235
    },
    {
      "epoch": 3.4454345703125e-05,
      "model_forward_time": 1.736945390701294,
      "step": 36128
    },
    {
      "epoch": 3.4454345703125e-05,
      "step": 36128,
      "training_step_time": 7.6391282081604
    },
    {
      "epoch": 3.4455299377441404e-05,
      "model_forward_time": 1.748502492904663,
      "step": 36129
    },
    {
      "epoch": 3.4455299377441404e-05,
      "step": 36129,
      "training_step_time": 8.006191968917847
    },
    {
      "epoch": 3.4456253051757815e-05,
      "grad_norm": 0.10986775904893875,
      "learning_rate": 7.576591789307272e-06,
      "loss": 0.0539,
      "step": 36130
    },
    {
      "epoch": 3.4456253051757815e-05,
      "model_forward_time": 1.7374260425567627,
      "step": 36130
    },
    {
      "dynamics_loss_avg": 0.04810452032834291,
      "epoch": 3.4456253051757815e-05,
      "step": 36130
    },
    {
      "action_loss_avg": 0.002863867412088439,
      "epoch": 3.4456253051757815e-05,
      "step": 36130
    },
    {
      "epoch": 3.4456253051757815e-05,
      "step": 36130,
      "training_step_time": 7.550455570220947
    },
    {
      "epoch": 3.445720672607422e-05,
      "model_forward_time": 2.9329588413238525,
      "step": 36131
    },
    {
      "epoch": 3.445720672607422e-05,
      "step": 36131,
      "training_step_time": 8.49066972732544
    },
    {
      "epoch": 3.445816040039062e-05,
      "model_forward_time": 1.4586045742034912,
      "step": 36132
    },
    {
      "epoch": 3.445816040039062e-05,
      "step": 36132,
      "training_step_time": 7.611588954925537
    },
    {
      "epoch": 3.4459114074707034e-05,
      "model_forward_time": 1.750889778137207,
      "step": 36133
    },
    {
      "epoch": 3.4459114074707034e-05,
      "step": 36133,
      "training_step_time": 9.92995309829712
    },
    {
      "epoch": 3.446006774902344e-05,
      "model_forward_time": 1.752223253250122,
      "step": 36134
    },
    {
      "epoch": 3.446006774902344e-05,
      "step": 36134,
      "training_step_time": 7.954648971557617
    },
    {
      "epoch": 3.446102142333984e-05,
      "model_forward_time": 1.76377534866333,
      "step": 36135
    },
    {
      "epoch": 3.446102142333984e-05,
      "step": 36135,
      "training_step_time": 7.80250883102417
    },
    {
      "epoch": 3.446197509765625e-05,
      "model_forward_time": 1.754181146621704,
      "step": 36136
    },
    {
      "epoch": 3.446197509765625e-05,
      "step": 36136,
      "training_step_time": 7.807322978973389
    },
    {
      "epoch": 3.446292877197266e-05,
      "model_forward_time": 1.7399797439575195,
      "step": 36137
    },
    {
      "epoch": 3.446292877197266e-05,
      "step": 36137,
      "training_step_time": 8.770167589187622
    },
    {
      "epoch": 3.446388244628906e-05,
      "model_forward_time": 1.7474431991577148,
      "step": 36138
    },
    {
      "epoch": 3.446388244628906e-05,
      "step": 36138,
      "training_step_time": 8.071484565734863
    },
    {
      "epoch": 3.446483612060547e-05,
      "model_forward_time": 1.7588586807250977,
      "step": 36139
    },
    {
      "epoch": 3.446483612060547e-05,
      "step": 36139,
      "training_step_time": 7.699878692626953
    },
    {
      "epoch": 3.4465789794921876e-05,
      "grad_norm": 0.18922023475170135,
      "learning_rate": 7.575174626752216e-06,
      "loss": 0.0552,
      "step": 36140
    },
    {
      "epoch": 3.4465789794921876e-05,
      "model_forward_time": 2.9588446617126465,
      "step": 36140
    },
    {
      "dynamics_loss_avg": 0.05861394163221121,
      "epoch": 3.4465789794921876e-05,
      "step": 36140
    },
    {
      "action_loss_avg": 0.0017760430055204779,
      "epoch": 3.4465789794921876e-05,
      "step": 36140
    },
    {
      "epoch": 3.4465789794921876e-05,
      "step": 36140,
      "training_step_time": 7.846694469451904
    },
    {
      "epoch": 3.446674346923828e-05,
      "model_forward_time": 1.8158347606658936,
      "step": 36141
    },
    {
      "epoch": 3.446674346923828e-05,
      "step": 36141,
      "training_step_time": 9.065605401992798
    },
    {
      "epoch": 3.446769714355469e-05,
      "model_forward_time": 1.7896757125854492,
      "step": 36142
    },
    {
      "epoch": 3.446769714355469e-05,
      "step": 36142,
      "training_step_time": 8.394512414932251
    },
    {
      "epoch": 3.4468650817871094e-05,
      "model_forward_time": 1.7428956031799316,
      "step": 36143
    },
    {
      "epoch": 3.4468650817871094e-05,
      "step": 36143,
      "training_step_time": 8.857216596603394
    },
    {
      "epoch": 3.44696044921875e-05,
      "model_forward_time": 1.7564868927001953,
      "step": 36144
    },
    {
      "epoch": 3.44696044921875e-05,
      "step": 36144,
      "training_step_time": 7.807084083557129
    },
    {
      "epoch": 3.447055816650391e-05,
      "model_forward_time": 1.7539401054382324,
      "step": 36145
    },
    {
      "epoch": 3.447055816650391e-05,
      "step": 36145,
      "training_step_time": 9.214712619781494
    },
    {
      "epoch": 3.447151184082031e-05,
      "model_forward_time": 1.7402009963989258,
      "step": 36146
    },
    {
      "epoch": 3.447151184082031e-05,
      "step": 36146,
      "training_step_time": 7.589892625808716
    },
    {
      "epoch": 3.447246551513672e-05,
      "model_forward_time": 1.7420308589935303,
      "step": 36147
    },
    {
      "epoch": 3.447246551513672e-05,
      "step": 36147,
      "training_step_time": 7.665450572967529
    },
    {
      "epoch": 3.447341918945313e-05,
      "model_forward_time": 1.7381205558776855,
      "step": 36148
    },
    {
      "epoch": 3.447341918945313e-05,
      "step": 36148,
      "training_step_time": 8.035773754119873
    },
    {
      "epoch": 3.447437286376953e-05,
      "model_forward_time": 2.983272075653076,
      "step": 36149
    },
    {
      "epoch": 3.447437286376953e-05,
      "step": 36149,
      "training_step_time": 7.805275917053223
    },
    {
      "epoch": 3.4475326538085936e-05,
      "grad_norm": 0.22223927080631256,
      "learning_rate": 7.573757182579935e-06,
      "loss": 0.0552,
      "step": 36150
    },
    {
      "epoch": 3.4475326538085936e-05,
      "model_forward_time": 1.8121206760406494,
      "step": 36150
    },
    {
      "dynamics_loss_avg": 0.0508983738720417,
      "epoch": 3.4475326538085936e-05,
      "step": 36150
    },
    {
      "action_loss_avg": 0.0017553892714204267,
      "epoch": 3.4475326538085936e-05,
      "step": 36150
    },
    {
      "epoch": 3.4475326538085936e-05,
      "step": 36150,
      "training_step_time": 8.225149631500244
    },
    {
      "epoch": 3.447628021240235e-05,
      "model_forward_time": 1.75697922706604,
      "step": 36151
    },
    {
      "epoch": 3.447628021240235e-05,
      "step": 36151,
      "training_step_time": 10.14028024673462
    },
    {
      "epoch": 3.447723388671875e-05,
      "model_forward_time": 1.7437992095947266,
      "step": 36152
    },
    {
      "epoch": 3.447723388671875e-05,
      "step": 36152,
      "training_step_time": 8.84240436553955
    },
    {
      "epoch": 3.4478187561035155e-05,
      "model_forward_time": 1.736952781677246,
      "step": 36153
    },
    {
      "epoch": 3.4478187561035155e-05,
      "step": 36153,
      "training_step_time": 7.83380913734436
    },
    {
      "epoch": 3.4479141235351566e-05,
      "model_forward_time": 1.7330093383789062,
      "step": 36154
    },
    {
      "epoch": 3.4479141235351566e-05,
      "step": 36154,
      "training_step_time": 7.8071863651275635
    },
    {
      "epoch": 3.448009490966797e-05,
      "model_forward_time": 1.7410321235656738,
      "step": 36155
    },
    {
      "epoch": 3.448009490966797e-05,
      "step": 36155,
      "training_step_time": 7.728198289871216
    },
    {
      "epoch": 3.4481048583984374e-05,
      "model_forward_time": 1.7516438961029053,
      "step": 36156
    },
    {
      "epoch": 3.4481048583984374e-05,
      "step": 36156,
      "training_step_time": 7.69017767906189
    },
    {
      "epoch": 3.448200225830078e-05,
      "model_forward_time": 1.7838468551635742,
      "step": 36157
    },
    {
      "epoch": 3.448200225830078e-05,
      "step": 36157,
      "training_step_time": 9.024641036987305
    },
    {
      "epoch": 3.448295593261719e-05,
      "model_forward_time": 2.9187190532684326,
      "step": 36158
    },
    {
      "epoch": 3.448295593261719e-05,
      "step": 36158,
      "training_step_time": 8.756271839141846
    },
    {
      "epoch": 3.448390960693359e-05,
      "model_forward_time": 1.7976188659667969,
      "step": 36159
    },
    {
      "epoch": 3.448390960693359e-05,
      "step": 36159,
      "training_step_time": 8.519768953323364
    },
    {
      "epoch": 3.448486328125e-05,
      "grad_norm": 0.17319484055042267,
      "learning_rate": 7.572339456945435e-06,
      "loss": 0.0603,
      "step": 36160
    },
    {
      "epoch": 3.448486328125e-05,
      "model_forward_time": 1.7338545322418213,
      "step": 36160
    },
    {
      "dynamics_loss_avg": 0.06857037469744683,
      "epoch": 3.448486328125e-05,
      "step": 36160
    },
    {
      "action_loss_avg": 0.0026457534229848536,
      "epoch": 3.448486328125e-05,
      "step": 36160
    },
    {
      "epoch": 3.448486328125e-05,
      "step": 36160,
      "training_step_time": 8.767398834228516
    },
    {
      "epoch": 3.448581695556641e-05,
      "model_forward_time": 1.739915132522583,
      "step": 36161
    },
    {
      "epoch": 3.448581695556641e-05,
      "step": 36161,
      "training_step_time": 7.921114921569824
    },
    {
      "epoch": 3.448677062988281e-05,
      "model_forward_time": 1.7398531436920166,
      "step": 36162
    },
    {
      "epoch": 3.448677062988281e-05,
      "step": 36162,
      "training_step_time": 7.8210084438323975
    },
    {
      "epoch": 3.4487724304199216e-05,
      "model_forward_time": 1.7497808933258057,
      "step": 36163
    },
    {
      "epoch": 3.4487724304199216e-05,
      "step": 36163,
      "training_step_time": 7.787536144256592
    },
    {
      "epoch": 3.4488677978515626e-05,
      "model_forward_time": 1.7565498352050781,
      "step": 36164
    },
    {
      "epoch": 3.4488677978515626e-05,
      "step": 36164,
      "training_step_time": 7.759275913238525
    },
    {
      "epoch": 3.448963165283203e-05,
      "model_forward_time": 1.74747633934021,
      "step": 36165
    },
    {
      "epoch": 3.448963165283203e-05,
      "step": 36165,
      "training_step_time": 7.703650236129761
    },
    {
      "epoch": 3.4490585327148434e-05,
      "model_forward_time": 1.7369201183319092,
      "step": 36166
    },
    {
      "epoch": 3.4490585327148434e-05,
      "step": 36166,
      "training_step_time": 8.601178169250488
    },
    {
      "epoch": 3.4491539001464845e-05,
      "model_forward_time": 2.8550260066986084,
      "step": 36167
    },
    {
      "epoch": 3.4491539001464845e-05,
      "step": 36167,
      "training_step_time": 8.387837648391724
    },
    {
      "epoch": 3.449249267578125e-05,
      "model_forward_time": 1.7726731300354004,
      "step": 36168
    },
    {
      "epoch": 3.449249267578125e-05,
      "step": 36168,
      "training_step_time": 8.38828182220459
    },
    {
      "epoch": 3.449344635009765e-05,
      "model_forward_time": 1.746734380722046,
      "step": 36169
    },
    {
      "epoch": 3.449344635009765e-05,
      "step": 36169,
      "training_step_time": 7.641696214675903
    },
    {
      "epoch": 3.4494400024414064e-05,
      "grad_norm": 0.14594489336013794,
      "learning_rate": 7.570921450003762e-06,
      "loss": 0.0575,
      "step": 36170
    },
    {
      "epoch": 3.4494400024414064e-05,
      "model_forward_time": 1.748551607131958,
      "step": 36170
    },
    {
      "dynamics_loss_avg": 0.05045163678005338,
      "epoch": 3.4494400024414064e-05,
      "step": 36170
    },
    {
      "action_loss_avg": 0.0018241035286337137,
      "epoch": 3.4494400024414064e-05,
      "step": 36170
    },
    {
      "epoch": 3.4494400024414064e-05,
      "step": 36170,
      "training_step_time": 9.448237657546997
    },
    {
      "epoch": 3.449535369873047e-05,
      "model_forward_time": 1.7509078979492188,
      "step": 36171
    },
    {
      "epoch": 3.449535369873047e-05,
      "step": 36171,
      "training_step_time": 8.593661785125732
    },
    {
      "epoch": 3.449630737304687e-05,
      "model_forward_time": 1.7446465492248535,
      "step": 36172
    },
    {
      "epoch": 3.449630737304687e-05,
      "step": 36172,
      "training_step_time": 7.7940239906311035
    },
    {
      "epoch": 3.449726104736328e-05,
      "model_forward_time": 1.7568304538726807,
      "step": 36173
    },
    {
      "epoch": 3.449726104736328e-05,
      "step": 36173,
      "training_step_time": 7.802024841308594
    },
    {
      "epoch": 3.449821472167969e-05,
      "model_forward_time": 1.7499709129333496,
      "step": 36174
    },
    {
      "epoch": 3.449821472167969e-05,
      "step": 36174,
      "training_step_time": 7.576479911804199
    },
    {
      "epoch": 3.449916839599609e-05,
      "model_forward_time": 1.7447733879089355,
      "step": 36175
    },
    {
      "epoch": 3.449916839599609e-05,
      "step": 36175,
      "training_step_time": 8.611915826797485
    },
    {
      "epoch": 3.45001220703125e-05,
      "model_forward_time": 2.8286900520324707,
      "step": 36176
    },
    {
      "epoch": 3.45001220703125e-05,
      "step": 36176,
      "training_step_time": 8.087079286575317
    },
    {
      "epoch": 3.4501075744628906e-05,
      "model_forward_time": 1.7445542812347412,
      "step": 36177
    },
    {
      "epoch": 3.4501075744628906e-05,
      "step": 36177,
      "training_step_time": 7.754955530166626
    },
    {
      "epoch": 3.450202941894531e-05,
      "model_forward_time": 1.8002257347106934,
      "step": 36178
    },
    {
      "epoch": 3.450202941894531e-05,
      "step": 36178,
      "training_step_time": 7.728729724884033
    },
    {
      "epoch": 3.450298309326172e-05,
      "model_forward_time": 1.7579236030578613,
      "step": 36179
    },
    {
      "epoch": 3.450298309326172e-05,
      "step": 36179,
      "training_step_time": 7.816009283065796
    },
    {
      "epoch": 3.4503936767578125e-05,
      "grad_norm": 0.12998338043689728,
      "learning_rate": 7.569503161909984e-06,
      "loss": 0.0575,
      "step": 36180
    },
    {
      "epoch": 3.4503936767578125e-05,
      "model_forward_time": 1.7335751056671143,
      "step": 36180
    },
    {
      "dynamics_loss_avg": 0.04776879735291004,
      "epoch": 3.4503936767578125e-05,
      "step": 36180
    },
    {
      "action_loss_avg": 0.001714833575533703,
      "epoch": 3.4503936767578125e-05,
      "step": 36180
    },
    {
      "epoch": 3.4503936767578125e-05,
      "step": 36180,
      "training_step_time": 10.061859607696533
    },
    {
      "epoch": 3.450489044189453e-05,
      "model_forward_time": 1.7504947185516357,
      "step": 36181
    },
    {
      "epoch": 3.450489044189453e-05,
      "step": 36181,
      "training_step_time": 7.662540674209595
    },
    {
      "epoch": 3.450584411621094e-05,
      "model_forward_time": 1.748615026473999,
      "step": 36182
    },
    {
      "epoch": 3.450584411621094e-05,
      "step": 36182,
      "training_step_time": 7.7471630573272705
    },
    {
      "epoch": 3.4506797790527344e-05,
      "model_forward_time": 1.732234001159668,
      "step": 36183
    },
    {
      "epoch": 3.4506797790527344e-05,
      "step": 36183,
      "training_step_time": 7.696511745452881
    },
    {
      "epoch": 3.450775146484375e-05,
      "model_forward_time": 1.7492187023162842,
      "step": 36184
    },
    {
      "epoch": 3.450775146484375e-05,
      "step": 36184,
      "training_step_time": 8.667017221450806
    },
    {
      "epoch": 3.450870513916016e-05,
      "model_forward_time": 1.7512803077697754,
      "step": 36185
    },
    {
      "epoch": 3.450870513916016e-05,
      "step": 36185,
      "training_step_time": 8.029733419418335
    },
    {
      "epoch": 3.450965881347656e-05,
      "model_forward_time": 3.023634195327759,
      "step": 36186
    },
    {
      "epoch": 3.450965881347656e-05,
      "step": 36186,
      "training_step_time": 7.73740553855896
    },
    {
      "epoch": 3.4510612487792966e-05,
      "model_forward_time": 1.8075742721557617,
      "step": 36187
    },
    {
      "epoch": 3.4510612487792966e-05,
      "step": 36187,
      "training_step_time": 8.138599872589111
    },
    {
      "epoch": 3.451156616210938e-05,
      "model_forward_time": 1.7430996894836426,
      "step": 36188
    },
    {
      "epoch": 3.451156616210938e-05,
      "step": 36188,
      "training_step_time": 8.76587200164795
    },
    {
      "epoch": 3.451251983642578e-05,
      "model_forward_time": 1.7545819282531738,
      "step": 36189
    },
    {
      "epoch": 3.451251983642578e-05,
      "step": 36189,
      "training_step_time": 9.710963010787964
    },
    {
      "epoch": 3.4513473510742185e-05,
      "grad_norm": 0.12139569222927094,
      "learning_rate": 7.5680845928192024e-06,
      "loss": 0.0582,
      "step": 36190
    },
    {
      "epoch": 3.4513473510742185e-05,
      "model_forward_time": 1.7367196083068848,
      "step": 36190
    },
    {
      "dynamics_loss_avg": 0.05201098024845123,
      "epoch": 3.4513473510742185e-05,
      "step": 36190
    },
    {
      "action_loss_avg": 0.0031594995089108124,
      "epoch": 3.4513473510742185e-05,
      "step": 36190
    },
    {
      "epoch": 3.4513473510742185e-05,
      "step": 36190,
      "training_step_time": 7.809124946594238
    },
    {
      "epoch": 3.4514427185058596e-05,
      "model_forward_time": 1.7562639713287354,
      "step": 36191
    },
    {
      "epoch": 3.4514427185058596e-05,
      "step": 36191,
      "training_step_time": 7.630406141281128
    },
    {
      "epoch": 3.4515380859375e-05,
      "model_forward_time": 1.741025447845459,
      "step": 36192
    },
    {
      "epoch": 3.4515380859375e-05,
      "step": 36192,
      "training_step_time": 7.636380910873413
    },
    {
      "epoch": 3.4516334533691404e-05,
      "model_forward_time": 1.7480010986328125,
      "step": 36193
    },
    {
      "epoch": 3.4516334533691404e-05,
      "step": 36193,
      "training_step_time": 7.5435731410980225
    },
    {
      "epoch": 3.4517288208007815e-05,
      "model_forward_time": 1.7600769996643066,
      "step": 36194
    },
    {
      "epoch": 3.4517288208007815e-05,
      "step": 36194,
      "training_step_time": 7.898248672485352
    },
    {
      "epoch": 3.451824188232422e-05,
      "model_forward_time": 2.9270503520965576,
      "step": 36195
    },
    {
      "epoch": 3.451824188232422e-05,
      "step": 36195,
      "training_step_time": 7.602397680282593
    },
    {
      "epoch": 3.451919555664062e-05,
      "model_forward_time": 1.811568260192871,
      "step": 36196
    },
    {
      "epoch": 3.451919555664062e-05,
      "step": 36196,
      "training_step_time": 8.130840063095093
    },
    {
      "epoch": 3.4520149230957034e-05,
      "model_forward_time": 1.7482490539550781,
      "step": 36197
    },
    {
      "epoch": 3.4520149230957034e-05,
      "step": 36197,
      "training_step_time": 8.721310377120972
    },
    {
      "epoch": 3.452110290527344e-05,
      "model_forward_time": 1.7502057552337646,
      "step": 36198
    },
    {
      "epoch": 3.452110290527344e-05,
      "step": 36198,
      "training_step_time": 8.42377257347107
    },
    {
      "epoch": 3.452205657958984e-05,
      "model_forward_time": 1.7552642822265625,
      "step": 36199
    },
    {
      "epoch": 3.452205657958984e-05,
      "step": 36199,
      "training_step_time": 7.635459899902344
    },
    {
      "epoch": 3.452301025390625e-05,
      "grad_norm": 0.15160426497459412,
      "learning_rate": 7.566665742886552e-06,
      "loss": 0.0606,
      "step": 36200
    },
    {
      "epoch": 3.452301025390625e-05,
      "model_forward_time": 1.7356507778167725,
      "step": 36200
    },
    {
      "dynamics_loss_avg": 0.05476612001657486,
      "epoch": 3.452301025390625e-05,
      "step": 36200
    },
    {
      "action_loss_avg": 0.00190560431801714,
      "epoch": 3.452301025390625e-05,
      "step": 36200
    },
    {
      "epoch": 3.452301025390625e-05,
      "step": 36200,
      "training_step_time": 8.66984486579895
    },
    {
      "epoch": 3.452396392822266e-05,
      "model_forward_time": 1.749314546585083,
      "step": 36201
    },
    {
      "epoch": 3.452396392822266e-05,
      "step": 36201,
      "training_step_time": 7.682039976119995
    },
    {
      "epoch": 3.452491760253906e-05,
      "model_forward_time": 1.7581467628479004,
      "step": 36202
    },
    {
      "epoch": 3.452491760253906e-05,
      "step": 36202,
      "training_step_time": 7.250890016555786
    },
    {
      "epoch": 3.452587127685547e-05,
      "model_forward_time": 1.743598222732544,
      "step": 36203
    },
    {
      "epoch": 3.452587127685547e-05,
      "step": 36203,
      "training_step_time": 7.805899143218994
    },
    {
      "epoch": 3.4526824951171876e-05,
      "model_forward_time": 2.9117178916931152,
      "step": 36204
    },
    {
      "epoch": 3.4526824951171876e-05,
      "step": 36204,
      "training_step_time": 7.638750314712524
    },
    {
      "epoch": 3.452777862548828e-05,
      "model_forward_time": 1.801443338394165,
      "step": 36205
    },
    {
      "epoch": 3.452777862548828e-05,
      "step": 36205,
      "training_step_time": 7.564938068389893
    },
    {
      "epoch": 3.452873229980469e-05,
      "model_forward_time": 1.7528016567230225,
      "step": 36206
    },
    {
      "epoch": 3.452873229980469e-05,
      "step": 36206,
      "training_step_time": 7.852734327316284
    },
    {
      "epoch": 3.4529685974121094e-05,
      "model_forward_time": 1.7464854717254639,
      "step": 36207
    },
    {
      "epoch": 3.4529685974121094e-05,
      "step": 36207,
      "training_step_time": 9.384384393692017
    },
    {
      "epoch": 3.45306396484375e-05,
      "model_forward_time": 1.7486522197723389,
      "step": 36208
    },
    {
      "epoch": 3.45306396484375e-05,
      "step": 36208,
      "training_step_time": 7.657763242721558
    },
    {
      "epoch": 3.453159332275391e-05,
      "model_forward_time": 1.7484543323516846,
      "step": 36209
    },
    {
      "epoch": 3.453159332275391e-05,
      "step": 36209,
      "training_step_time": 8.185109376907349
    },
    {
      "epoch": 3.453254699707031e-05,
      "grad_norm": 0.14604417979717255,
      "learning_rate": 7.565246612267194e-06,
      "loss": 0.0559,
      "step": 36210
    },
    {
      "epoch": 3.453254699707031e-05,
      "model_forward_time": 1.7583255767822266,
      "step": 36210
    },
    {
      "dynamics_loss_avg": 0.0456170130521059,
      "epoch": 3.453254699707031e-05,
      "step": 36210
    },
    {
      "action_loss_avg": 0.0033334158710204066,
      "epoch": 3.453254699707031e-05,
      "step": 36210
    },
    {
      "epoch": 3.453254699707031e-05,
      "step": 36210,
      "training_step_time": 7.688882112503052
    },
    {
      "epoch": 3.453350067138672e-05,
      "model_forward_time": 1.7522897720336914,
      "step": 36211
    },
    {
      "epoch": 3.453350067138672e-05,
      "step": 36211,
      "training_step_time": 7.4212892055511475
    },
    {
      "epoch": 3.453445434570313e-05,
      "model_forward_time": 1.3800318241119385,
      "step": 36212
    },
    {
      "epoch": 3.453445434570313e-05,
      "step": 36212,
      "training_step_time": 7.455039024353027
    },
    {
      "epoch": 3.453540802001953e-05,
      "model_forward_time": 2.86763858795166,
      "step": 36213
    },
    {
      "epoch": 3.453540802001953e-05,
      "step": 36213,
      "training_step_time": 8.53193473815918
    },
    {
      "epoch": 3.4536361694335936e-05,
      "model_forward_time": 1.7487905025482178,
      "step": 36214
    },
    {
      "epoch": 3.4536361694335936e-05,
      "step": 36214,
      "training_step_time": 9.192113399505615
    },
    {
      "epoch": 3.453731536865235e-05,
      "model_forward_time": 1.7487404346466064,
      "step": 36215
    },
    {
      "epoch": 3.453731536865235e-05,
      "step": 36215,
      "training_step_time": 7.8073365688323975
    },
    {
      "epoch": 3.453826904296875e-05,
      "model_forward_time": 1.7487468719482422,
      "step": 36216
    },
    {
      "epoch": 3.453826904296875e-05,
      "step": 36216,
      "training_step_time": 7.620800018310547
    },
    {
      "epoch": 3.4539222717285155e-05,
      "model_forward_time": 1.746152400970459,
      "step": 36217
    },
    {
      "epoch": 3.4539222717285155e-05,
      "step": 36217,
      "training_step_time": 9.551836013793945
    },
    {
      "epoch": 3.4540176391601566e-05,
      "model_forward_time": 1.708031177520752,
      "step": 36218
    },
    {
      "epoch": 3.4540176391601566e-05,
      "step": 36218,
      "training_step_time": 7.654215574264526
    },
    {
      "epoch": 3.454113006591797e-05,
      "model_forward_time": 1.7319951057434082,
      "step": 36219
    },
    {
      "epoch": 3.454113006591797e-05,
      "step": 36219,
      "training_step_time": 7.588561058044434
    },
    {
      "epoch": 3.4542083740234374e-05,
      "grad_norm": 0.1272316873073578,
      "learning_rate": 7.563827201116325e-06,
      "loss": 0.0574,
      "step": 36220
    },
    {
      "epoch": 3.4542083740234374e-05,
      "model_forward_time": 1.7347824573516846,
      "step": 36220
    },
    {
      "dynamics_loss_avg": 0.04396098181605339,
      "epoch": 3.4542083740234374e-05,
      "step": 36220
    },
    {
      "action_loss_avg": 0.0011552707932423801,
      "epoch": 3.4542083740234374e-05,
      "step": 36220
    },
    {
      "epoch": 3.4542083740234374e-05,
      "step": 36220,
      "training_step_time": 7.0427305698394775
    },
    {
      "epoch": 3.454303741455078e-05,
      "model_forward_time": 1.747549295425415,
      "step": 36221
    },
    {
      "epoch": 3.454303741455078e-05,
      "step": 36221,
      "training_step_time": 7.593758821487427
    },
    {
      "epoch": 3.454399108886719e-05,
      "model_forward_time": 1.7264840602874756,
      "step": 36222
    },
    {
      "epoch": 3.454399108886719e-05,
      "step": 36222,
      "training_step_time": 9.046664476394653
    },
    {
      "epoch": 3.454494476318359e-05,
      "model_forward_time": 1.7342822551727295,
      "step": 36223
    },
    {
      "epoch": 3.454494476318359e-05,
      "step": 36223,
      "training_step_time": 7.688769102096558
    },
    {
      "epoch": 3.45458984375e-05,
      "model_forward_time": 1.7298758029937744,
      "step": 36224
    },
    {
      "epoch": 3.45458984375e-05,
      "step": 36224,
      "training_step_time": 7.558863639831543
    },
    {
      "epoch": 3.454685211181641e-05,
      "model_forward_time": 1.7409443855285645,
      "step": 36225
    },
    {
      "epoch": 3.454685211181641e-05,
      "step": 36225,
      "training_step_time": 7.720196485519409
    },
    {
      "epoch": 3.454780578613281e-05,
      "model_forward_time": 1.7341139316558838,
      "step": 36226
    },
    {
      "epoch": 3.454780578613281e-05,
      "step": 36226,
      "training_step_time": 8.5773446559906
    },
    {
      "epoch": 3.4548759460449216e-05,
      "model_forward_time": 1.7302279472351074,
      "step": 36227
    },
    {
      "epoch": 3.4548759460449216e-05,
      "step": 36227,
      "training_step_time": 8.70896029472351
    },
    {
      "epoch": 3.4549713134765626e-05,
      "model_forward_time": 1.7312228679656982,
      "step": 36228
    },
    {
      "epoch": 3.4549713134765626e-05,
      "step": 36228,
      "training_step_time": 8.55671215057373
    },
    {
      "epoch": 3.455066680908203e-05,
      "model_forward_time": 1.7302021980285645,
      "step": 36229
    },
    {
      "epoch": 3.455066680908203e-05,
      "step": 36229,
      "training_step_time": 7.577950954437256
    },
    {
      "epoch": 3.4551620483398435e-05,
      "grad_norm": 0.11534121632575989,
      "learning_rate": 7.562407509589165e-06,
      "loss": 0.0588,
      "step": 36230
    },
    {
      "epoch": 3.4551620483398435e-05,
      "model_forward_time": 1.7513020038604736,
      "step": 36230
    },
    {
      "dynamics_loss_avg": 0.07977881506085396,
      "epoch": 3.4551620483398435e-05,
      "step": 36230
    },
    {
      "action_loss_avg": 0.0028238698141649365,
      "epoch": 3.4551620483398435e-05,
      "step": 36230
    },
    {
      "epoch": 3.4551620483398435e-05,
      "step": 36230,
      "training_step_time": 8.656190156936646
    },
    {
      "epoch": 3.4552574157714845e-05,
      "model_forward_time": 1.7400033473968506,
      "step": 36231
    },
    {
      "epoch": 3.4552574157714845e-05,
      "step": 36231,
      "training_step_time": 8.59495759010315
    },
    {
      "epoch": 3.455352783203125e-05,
      "model_forward_time": 2.2534360885620117,
      "step": 36232
    },
    {
      "epoch": 3.455352783203125e-05,
      "step": 36232,
      "training_step_time": 8.093261241912842
    },
    {
      "epoch": 3.455448150634765e-05,
      "model_forward_time": 1.7322413921356201,
      "step": 36233
    },
    {
      "epoch": 3.455448150634765e-05,
      "step": 36233,
      "training_step_time": 7.609397888183594
    },
    {
      "epoch": 3.4555435180664064e-05,
      "model_forward_time": 1.7179651260375977,
      "step": 36234
    },
    {
      "epoch": 3.4555435180664064e-05,
      "step": 36234,
      "training_step_time": 8.448804140090942
    },
    {
      "epoch": 3.455638885498047e-05,
      "model_forward_time": 1.7279291152954102,
      "step": 36235
    },
    {
      "epoch": 3.455638885498047e-05,
      "step": 36235,
      "training_step_time": 8.250877141952515
    },
    {
      "epoch": 3.455734252929687e-05,
      "model_forward_time": 1.7379448413848877,
      "step": 36236
    },
    {
      "epoch": 3.455734252929687e-05,
      "step": 36236,
      "training_step_time": 8.625491857528687
    },
    {
      "epoch": 3.455829620361328e-05,
      "model_forward_time": 1.7313628196716309,
      "step": 36237
    },
    {
      "epoch": 3.455829620361328e-05,
      "step": 36237,
      "training_step_time": 8.30318832397461
    },
    {
      "epoch": 3.455924987792969e-05,
      "model_forward_time": 1.7311089038848877,
      "step": 36238
    },
    {
      "epoch": 3.455924987792969e-05,
      "step": 36238,
      "training_step_time": 7.727781295776367
    },
    {
      "epoch": 3.456020355224609e-05,
      "model_forward_time": 1.7478220462799072,
      "step": 36239
    },
    {
      "epoch": 3.456020355224609e-05,
      "step": 36239,
      "training_step_time": 7.820801258087158
    },
    {
      "epoch": 3.45611572265625e-05,
      "grad_norm": 0.10461775213479996,
      "learning_rate": 7.560987537840973e-06,
      "loss": 0.0568,
      "step": 36240
    },
    {
      "epoch": 3.45611572265625e-05,
      "model_forward_time": 1.7398695945739746,
      "step": 36240
    },
    {
      "dynamics_loss_avg": 0.05063171312212944,
      "epoch": 3.45611572265625e-05,
      "step": 36240
    },
    {
      "action_loss_avg": 0.0018725622736383229,
      "epoch": 3.45611572265625e-05,
      "step": 36240
    },
    {
      "epoch": 3.45611572265625e-05,
      "step": 36240,
      "training_step_time": 7.926898717880249
    },
    {
      "epoch": 3.4562110900878906e-05,
      "model_forward_time": 3.0160343647003174,
      "step": 36241
    },
    {
      "epoch": 3.4562110900878906e-05,
      "step": 36241,
      "training_step_time": 8.26294207572937
    },
    {
      "epoch": 3.456306457519531e-05,
      "model_forward_time": 1.7876777648925781,
      "step": 36242
    },
    {
      "epoch": 3.456306457519531e-05,
      "step": 36242,
      "training_step_time": 9.455461740493774
    },
    {
      "epoch": 3.456401824951172e-05,
      "model_forward_time": 1.7732813358306885,
      "step": 36243
    },
    {
      "epoch": 3.456401824951172e-05,
      "step": 36243,
      "training_step_time": 7.9118146896362305
    },
    {
      "epoch": 3.4564971923828125e-05,
      "model_forward_time": 1.7582311630249023,
      "step": 36244
    },
    {
      "epoch": 3.4564971923828125e-05,
      "step": 36244,
      "training_step_time": 8.184934616088867
    },
    {
      "epoch": 3.456592559814453e-05,
      "model_forward_time": 1.7540647983551025,
      "step": 36245
    },
    {
      "epoch": 3.456592559814453e-05,
      "step": 36245,
      "training_step_time": 7.611331939697266
    },
    {
      "epoch": 3.456687927246094e-05,
      "model_forward_time": 1.7716116905212402,
      "step": 36246
    },
    {
      "epoch": 3.456687927246094e-05,
      "step": 36246,
      "training_step_time": 7.6069183349609375
    },
    {
      "epoch": 3.4567832946777344e-05,
      "model_forward_time": 1.7496416568756104,
      "step": 36247
    },
    {
      "epoch": 3.4567832946777344e-05,
      "step": 36247,
      "training_step_time": 7.678276300430298
    },
    {
      "epoch": 3.456878662109375e-05,
      "model_forward_time": 1.756347894668579,
      "step": 36248
    },
    {
      "epoch": 3.456878662109375e-05,
      "step": 36248,
      "training_step_time": 8.336845874786377
    },
    {
      "epoch": 3.456974029541016e-05,
      "model_forward_time": 1.7649152278900146,
      "step": 36249
    },
    {
      "epoch": 3.456974029541016e-05,
      "step": 36249,
      "training_step_time": 7.682556629180908
    },
    {
      "epoch": 3.457069396972656e-05,
      "grad_norm": 0.11742865294218063,
      "learning_rate": 7.559567286027036e-06,
      "loss": 0.0593,
      "step": 36250
    },
    {
      "epoch": 3.457069396972656e-05,
      "model_forward_time": 2.9273505210876465,
      "step": 36250
    },
    {
      "dynamics_loss_avg": 0.04878387060016394,
      "epoch": 3.457069396972656e-05,
      "step": 36250
    },
    {
      "action_loss_avg": 0.0014891532220644877,
      "epoch": 3.457069396972656e-05,
      "step": 36250
    },
    {
      "epoch": 3.457069396972656e-05,
      "step": 36250,
      "training_step_time": 8.795313358306885
    },
    {
      "epoch": 3.4571647644042967e-05,
      "model_forward_time": 1.7624938488006592,
      "step": 36251
    },
    {
      "epoch": 3.4571647644042967e-05,
      "step": 36251,
      "training_step_time": 8.666488647460938
    },
    {
      "epoch": 3.457260131835938e-05,
      "model_forward_time": 1.761570930480957,
      "step": 36252
    },
    {
      "epoch": 3.457260131835938e-05,
      "step": 36252,
      "training_step_time": 9.2769296169281
    },
    {
      "epoch": 3.457355499267578e-05,
      "model_forward_time": 1.7488129138946533,
      "step": 36253
    },
    {
      "epoch": 3.457355499267578e-05,
      "step": 36253,
      "training_step_time": 8.727162599563599
    },
    {
      "epoch": 3.4574508666992185e-05,
      "model_forward_time": 1.7643301486968994,
      "step": 36254
    },
    {
      "epoch": 3.4574508666992185e-05,
      "step": 36254,
      "training_step_time": 7.723860740661621
    },
    {
      "epoch": 3.4575462341308596e-05,
      "model_forward_time": 1.7586424350738525,
      "step": 36255
    },
    {
      "epoch": 3.4575462341308596e-05,
      "step": 36255,
      "training_step_time": 7.569739818572998
    },
    {
      "epoch": 3.4576416015625e-05,
      "model_forward_time": 1.7658205032348633,
      "step": 36256
    },
    {
      "epoch": 3.4576416015625e-05,
      "step": 36256,
      "training_step_time": 7.686498165130615
    },
    {
      "epoch": 3.4577369689941404e-05,
      "model_forward_time": 1.761725664138794,
      "step": 36257
    },
    {
      "epoch": 3.4577369689941404e-05,
      "step": 36257,
      "training_step_time": 8.48988962173462
    },
    {
      "epoch": 3.4578323364257815e-05,
      "model_forward_time": 1.765798568725586,
      "step": 36258
    },
    {
      "epoch": 3.4578323364257815e-05,
      "step": 36258,
      "training_step_time": 7.749331712722778
    },
    {
      "epoch": 3.457927703857422e-05,
      "model_forward_time": 2.8937580585479736,
      "step": 36259
    },
    {
      "epoch": 3.457927703857422e-05,
      "step": 36259,
      "training_step_time": 8.416261196136475
    },
    {
      "epoch": 3.458023071289062e-05,
      "grad_norm": 0.10595806688070297,
      "learning_rate": 7.558146754302669e-06,
      "loss": 0.0553,
      "step": 36260
    },
    {
      "epoch": 3.458023071289062e-05,
      "model_forward_time": 1.7417786121368408,
      "step": 36260
    },
    {
      "dynamics_loss_avg": 0.05491217002272606,
      "epoch": 3.458023071289062e-05,
      "step": 36260
    },
    {
      "action_loss_avg": 0.0028956928174011408,
      "epoch": 3.458023071289062e-05,
      "step": 36260
    },
    {
      "epoch": 3.458023071289062e-05,
      "step": 36260,
      "training_step_time": 8.250515222549438
    },
    {
      "epoch": 3.4581184387207034e-05,
      "model_forward_time": 1.8215694427490234,
      "step": 36261
    },
    {
      "epoch": 3.4581184387207034e-05,
      "step": 36261,
      "training_step_time": 8.122797966003418
    },
    {
      "epoch": 3.458213806152344e-05,
      "model_forward_time": 1.7572722434997559,
      "step": 36262
    },
    {
      "epoch": 3.458213806152344e-05,
      "step": 36262,
      "training_step_time": 7.597561836242676
    },
    {
      "epoch": 3.458309173583984e-05,
      "model_forward_time": 1.7612009048461914,
      "step": 36263
    },
    {
      "epoch": 3.458309173583984e-05,
      "step": 36263,
      "training_step_time": 8.680673599243164
    },
    {
      "epoch": 3.458404541015625e-05,
      "model_forward_time": 1.7590880393981934,
      "step": 36264
    },
    {
      "epoch": 3.458404541015625e-05,
      "step": 36264,
      "training_step_time": 7.527691125869751
    },
    {
      "epoch": 3.458499908447266e-05,
      "model_forward_time": 1.7607696056365967,
      "step": 36265
    },
    {
      "epoch": 3.458499908447266e-05,
      "step": 36265,
      "training_step_time": 7.543705701828003
    },
    {
      "epoch": 3.458595275878906e-05,
      "model_forward_time": 1.7576203346252441,
      "step": 36266
    },
    {
      "epoch": 3.458595275878906e-05,
      "step": 36266,
      "training_step_time": 7.463716268539429
    },
    {
      "epoch": 3.458690643310547e-05,
      "model_forward_time": 1.7483081817626953,
      "step": 36267
    },
    {
      "epoch": 3.458690643310547e-05,
      "step": 36267,
      "training_step_time": 8.50739336013794
    },
    {
      "epoch": 3.4587860107421876e-05,
      "model_forward_time": 2.864729881286621,
      "step": 36268
    },
    {
      "epoch": 3.4587860107421876e-05,
      "step": 36268,
      "training_step_time": 8.802855491638184
    },
    {
      "epoch": 3.458881378173828e-05,
      "model_forward_time": 1.7470688819885254,
      "step": 36269
    },
    {
      "epoch": 3.458881378173828e-05,
      "step": 36269,
      "training_step_time": 7.617209434509277
    },
    {
      "epoch": 3.458976745605469e-05,
      "grad_norm": 0.12750695645809174,
      "learning_rate": 7.556725942823217e-06,
      "loss": 0.0578,
      "step": 36270
    },
    {
      "epoch": 3.458976745605469e-05,
      "model_forward_time": 1.8108820915222168,
      "step": 36270
    },
    {
      "dynamics_loss_avg": 0.047421162761747834,
      "epoch": 3.458976745605469e-05,
      "step": 36270
    },
    {
      "action_loss_avg": 0.0018791914597386494,
      "epoch": 3.458976745605469e-05,
      "step": 36270
    },
    {
      "epoch": 3.458976745605469e-05,
      "step": 36270,
      "training_step_time": 7.587582588195801
    },
    {
      "epoch": 3.4590721130371095e-05,
      "model_forward_time": 1.755187749862671,
      "step": 36271
    },
    {
      "epoch": 3.4590721130371095e-05,
      "step": 36271,
      "training_step_time": 8.590282201766968
    },
    {
      "epoch": 3.45916748046875e-05,
      "model_forward_time": 1.7525134086608887,
      "step": 36272
    },
    {
      "epoch": 3.45916748046875e-05,
      "step": 36272,
      "training_step_time": 7.756838798522949
    },
    {
      "epoch": 3.459262847900391e-05,
      "model_forward_time": 1.7609198093414307,
      "step": 36273
    },
    {
      "epoch": 3.459262847900391e-05,
      "step": 36273,
      "training_step_time": 8.363461971282959
    },
    {
      "epoch": 3.459358215332031e-05,
      "model_forward_time": 1.7614893913269043,
      "step": 36274
    },
    {
      "epoch": 3.459358215332031e-05,
      "step": 36274,
      "training_step_time": 8.755493402481079
    },
    {
      "epoch": 3.459453582763672e-05,
      "model_forward_time": 1.7577414512634277,
      "step": 36275
    },
    {
      "epoch": 3.459453582763672e-05,
      "step": 36275,
      "training_step_time": 8.291216611862183
    },
    {
      "epoch": 3.459548950195313e-05,
      "model_forward_time": 1.7545311450958252,
      "step": 36276
    },
    {
      "epoch": 3.459548950195313e-05,
      "step": 36276,
      "training_step_time": 7.694513559341431
    },
    {
      "epoch": 3.459644317626953e-05,
      "model_forward_time": 1.759871244430542,
      "step": 36277
    },
    {
      "epoch": 3.459644317626953e-05,
      "step": 36277,
      "training_step_time": 9.06639814376831
    },
    {
      "epoch": 3.4597396850585936e-05,
      "model_forward_time": 1.7395431995391846,
      "step": 36278
    },
    {
      "epoch": 3.4597396850585936e-05,
      "step": 36278,
      "training_step_time": 8.289308309555054
    },
    {
      "epoch": 3.459835052490235e-05,
      "model_forward_time": 1.821484088897705,
      "step": 36279
    },
    {
      "epoch": 3.459835052490235e-05,
      "step": 36279,
      "training_step_time": 8.854745149612427
    },
    {
      "epoch": 3.459930419921875e-05,
      "grad_norm": 0.12497443705797195,
      "learning_rate": 7.555304851744061e-06,
      "loss": 0.0557,
      "step": 36280
    },
    {
      "epoch": 3.459930419921875e-05,
      "model_forward_time": 1.7456309795379639,
      "step": 36280
    },
    {
      "dynamics_loss_avg": 0.047533458285033706,
      "epoch": 3.459930419921875e-05,
      "step": 36280
    },
    {
      "action_loss_avg": 0.0015778167318785564,
      "epoch": 3.459930419921875e-05,
      "step": 36280
    },
    {
      "epoch": 3.459930419921875e-05,
      "step": 36280,
      "training_step_time": 7.868493318557739
    },
    {
      "epoch": 3.4600257873535155e-05,
      "model_forward_time": 1.7681474685668945,
      "step": 36281
    },
    {
      "epoch": 3.4600257873535155e-05,
      "step": 36281,
      "training_step_time": 8.574237823486328
    },
    {
      "epoch": 3.460121154785156e-05,
      "model_forward_time": 1.7393362522125244,
      "step": 36282
    },
    {
      "epoch": 3.460121154785156e-05,
      "step": 36282,
      "training_step_time": 7.839404106140137
    },
    {
      "epoch": 3.460216522216797e-05,
      "model_forward_time": 1.7368419170379639,
      "step": 36283
    },
    {
      "epoch": 3.460216522216797e-05,
      "step": 36283,
      "training_step_time": 8.75724172592163
    },
    {
      "epoch": 3.4603118896484374e-05,
      "model_forward_time": 1.734133243560791,
      "step": 36284
    },
    {
      "epoch": 3.4603118896484374e-05,
      "step": 36284,
      "training_step_time": 7.740720272064209
    },
    {
      "epoch": 3.460407257080078e-05,
      "model_forward_time": 1.7489140033721924,
      "step": 36285
    },
    {
      "epoch": 3.460407257080078e-05,
      "step": 36285,
      "training_step_time": 8.755563974380493
    },
    {
      "epoch": 3.460502624511719e-05,
      "model_forward_time": 1.7329540252685547,
      "step": 36286
    },
    {
      "epoch": 3.460502624511719e-05,
      "step": 36286,
      "training_step_time": 8.080771684646606
    },
    {
      "epoch": 3.460597991943359e-05,
      "model_forward_time": 2.922959804534912,
      "step": 36287
    },
    {
      "epoch": 3.460597991943359e-05,
      "step": 36287,
      "training_step_time": 8.530397891998291
    },
    {
      "epoch": 3.460693359375e-05,
      "model_forward_time": 1.8107547760009766,
      "step": 36288
    },
    {
      "epoch": 3.460693359375e-05,
      "step": 36288,
      "training_step_time": 8.672664880752563
    },
    {
      "epoch": 3.460788726806641e-05,
      "model_forward_time": 1.737553596496582,
      "step": 36289
    },
    {
      "epoch": 3.460788726806641e-05,
      "step": 36289,
      "training_step_time": 9.41719388961792
    },
    {
      "epoch": 3.460884094238281e-05,
      "grad_norm": 0.1553582400083542,
      "learning_rate": 7.5538834812206075e-06,
      "loss": 0.056,
      "step": 36290
    },
    {
      "epoch": 3.460884094238281e-05,
      "model_forward_time": 1.7396085262298584,
      "step": 36290
    },
    {
      "dynamics_loss_avg": 0.07113340236246586,
      "epoch": 3.460884094238281e-05,
      "step": 36290
    },
    {
      "action_loss_avg": 0.0024080575560219584,
      "epoch": 3.460884094238281e-05,
      "step": 36290
    },
    {
      "epoch": 3.460884094238281e-05,
      "step": 36290,
      "training_step_time": 8.756009578704834
    },
    {
      "epoch": 3.4609794616699216e-05,
      "model_forward_time": 1.744551658630371,
      "step": 36291
    },
    {
      "epoch": 3.4609794616699216e-05,
      "step": 36291,
      "training_step_time": 7.727139472961426
    },
    {
      "epoch": 3.4610748291015627e-05,
      "model_forward_time": 1.7452032566070557,
      "step": 36292
    },
    {
      "epoch": 3.4610748291015627e-05,
      "step": 36292,
      "training_step_time": 7.525979995727539
    },
    {
      "epoch": 3.461170196533203e-05,
      "model_forward_time": 1.7308073043823242,
      "step": 36293
    },
    {
      "epoch": 3.461170196533203e-05,
      "step": 36293,
      "training_step_time": 8.36628794670105
    },
    {
      "epoch": 3.4612655639648435e-05,
      "model_forward_time": 1.742699146270752,
      "step": 36294
    },
    {
      "epoch": 3.4612655639648435e-05,
      "step": 36294,
      "training_step_time": 7.539183616638184
    },
    {
      "epoch": 3.4613609313964845e-05,
      "model_forward_time": 1.7460582256317139,
      "step": 36295
    },
    {
      "epoch": 3.4613609313964845e-05,
      "step": 36295,
      "training_step_time": 7.565660238265991
    },
    {
      "epoch": 3.461456298828125e-05,
      "model_forward_time": 2.9001619815826416,
      "step": 36296
    },
    {
      "epoch": 3.461456298828125e-05,
      "step": 36296,
      "training_step_time": 7.671854019165039
    },
    {
      "epoch": 3.4615516662597653e-05,
      "model_forward_time": 1.7995398044586182,
      "step": 36297
    },
    {
      "epoch": 3.4615516662597653e-05,
      "step": 36297,
      "training_step_time": 7.563438653945923
    },
    {
      "epoch": 3.4616470336914064e-05,
      "model_forward_time": 1.7421340942382812,
      "step": 36298
    },
    {
      "epoch": 3.4616470336914064e-05,
      "step": 36298,
      "training_step_time": 8.63969111442566
    },
    {
      "epoch": 3.461742401123047e-05,
      "model_forward_time": 1.7226452827453613,
      "step": 36299
    },
    {
      "epoch": 3.461742401123047e-05,
      "step": 36299,
      "training_step_time": 8.038856267929077
    },
    {
      "epoch": 3.461837768554687e-05,
      "grad_norm": 0.14488449692726135,
      "learning_rate": 7.552461831408297e-06,
      "loss": 0.0564,
      "step": 36300
    },
    {
      "epoch": 3.461837768554687e-05,
      "model_forward_time": 1.7528884410858154,
      "step": 36300
    },
    {
      "dynamics_loss_avg": 0.0531634408980608,
      "epoch": 3.461837768554687e-05,
      "step": 36300
    },
    {
      "action_loss_avg": 0.0020557386276777835,
      "epoch": 3.461837768554687e-05,
      "step": 36300
    },
    {
      "epoch": 3.461837768554687e-05,
      "step": 36300,
      "training_step_time": 8.216051578521729
    },
    {
      "epoch": 3.461933135986328e-05,
      "model_forward_time": 1.8049871921539307,
      "step": 36301
    },
    {
      "epoch": 3.461933135986328e-05,
      "step": 36301,
      "training_step_time": 7.507351875305176
    },
    {
      "epoch": 3.462028503417969e-05,
      "model_forward_time": 1.7527885437011719,
      "step": 36302
    },
    {
      "epoch": 3.462028503417969e-05,
      "step": 36302,
      "training_step_time": 7.581981658935547
    },
    {
      "epoch": 3.462123870849609e-05,
      "model_forward_time": 1.7758171558380127,
      "step": 36303
    },
    {
      "epoch": 3.462123870849609e-05,
      "step": 36303,
      "training_step_time": 7.504839181900024
    },
    {
      "epoch": 3.46221923828125e-05,
      "model_forward_time": 1.7567994594573975,
      "step": 36304
    },
    {
      "epoch": 3.46221923828125e-05,
      "step": 36304,
      "training_step_time": 7.5279412269592285
    },
    {
      "epoch": 3.4623146057128906e-05,
      "model_forward_time": 2.933093786239624,
      "step": 36305
    },
    {
      "epoch": 3.4623146057128906e-05,
      "step": 36305,
      "training_step_time": 9.558685302734375
    },
    {
      "epoch": 3.462409973144531e-05,
      "model_forward_time": 1.826150894165039,
      "step": 36306
    },
    {
      "epoch": 3.462409973144531e-05,
      "step": 36306,
      "training_step_time": 7.5655622482299805
    },
    {
      "epoch": 3.462505340576172e-05,
      "model_forward_time": 1.755800485610962,
      "step": 36307
    },
    {
      "epoch": 3.462505340576172e-05,
      "step": 36307,
      "training_step_time": 8.70350432395935
    },
    {
      "epoch": 3.4626007080078125e-05,
      "model_forward_time": 1.7493360042572021,
      "step": 36308
    },
    {
      "epoch": 3.4626007080078125e-05,
      "step": 36308,
      "training_step_time": 7.803056955337524
    },
    {
      "epoch": 3.462696075439453e-05,
      "model_forward_time": 1.761359691619873,
      "step": 36309
    },
    {
      "epoch": 3.462696075439453e-05,
      "step": 36309,
      "training_step_time": 7.717651605606079
    },
    {
      "epoch": 3.462791442871094e-05,
      "grad_norm": 0.11782209575176239,
      "learning_rate": 7.551039902462599e-06,
      "loss": 0.0569,
      "step": 36310
    },
    {
      "epoch": 3.462791442871094e-05,
      "model_forward_time": 1.7521519660949707,
      "step": 36310
    },
    {
      "dynamics_loss_avg": 0.05613781828433275,
      "epoch": 3.462791442871094e-05,
      "step": 36310
    },
    {
      "action_loss_avg": 0.0019940878584748132,
      "epoch": 3.462791442871094e-05,
      "step": 36310
    },
    {
      "epoch": 3.462791442871094e-05,
      "step": 36310,
      "training_step_time": 8.316831588745117
    },
    {
      "epoch": 3.4628868103027344e-05,
      "model_forward_time": 1.745553970336914,
      "step": 36311
    },
    {
      "epoch": 3.4628868103027344e-05,
      "step": 36311,
      "training_step_time": 9.57537317276001
    },
    {
      "epoch": 3.462982177734375e-05,
      "model_forward_time": 1.7461516857147217,
      "step": 36312
    },
    {
      "epoch": 3.462982177734375e-05,
      "step": 36312,
      "training_step_time": 7.48984169960022
    },
    {
      "epoch": 3.463077545166016e-05,
      "model_forward_time": 1.7519598007202148,
      "step": 36313
    },
    {
      "epoch": 3.463077545166016e-05,
      "step": 36313,
      "training_step_time": 7.61980414390564
    },
    {
      "epoch": 3.463172912597656e-05,
      "model_forward_time": 2.887699604034424,
      "step": 36314
    },
    {
      "epoch": 3.463172912597656e-05,
      "step": 36314,
      "training_step_time": 7.945275545120239
    },
    {
      "epoch": 3.4632682800292967e-05,
      "model_forward_time": 1.8039023876190186,
      "step": 36315
    },
    {
      "epoch": 3.4632682800292967e-05,
      "step": 36315,
      "training_step_time": 7.625291585922241
    },
    {
      "epoch": 3.463363647460938e-05,
      "model_forward_time": 1.7505087852478027,
      "step": 36316
    },
    {
      "epoch": 3.463363647460938e-05,
      "step": 36316,
      "training_step_time": 8.448307752609253
    },
    {
      "epoch": 3.463459014892578e-05,
      "model_forward_time": 1.802210807800293,
      "step": 36317
    },
    {
      "epoch": 3.463459014892578e-05,
      "step": 36317,
      "training_step_time": 7.731580972671509
    },
    {
      "epoch": 3.4635543823242185e-05,
      "model_forward_time": 1.744142770767212,
      "step": 36318
    },
    {
      "epoch": 3.4635543823242185e-05,
      "step": 36318,
      "training_step_time": 8.537683963775635
    },
    {
      "epoch": 3.4636497497558596e-05,
      "model_forward_time": 1.7473437786102295,
      "step": 36319
    },
    {
      "epoch": 3.4636497497558596e-05,
      "step": 36319,
      "training_step_time": 7.650312423706055
    },
    {
      "epoch": 3.4637451171875e-05,
      "grad_norm": 0.10638050734996796,
      "learning_rate": 7.549617694539011e-06,
      "loss": 0.0558,
      "step": 36320
    },
    {
      "epoch": 3.4637451171875e-05,
      "model_forward_time": 1.721942663192749,
      "step": 36320
    },
    {
      "dynamics_loss_avg": 0.05141300354152918,
      "epoch": 3.4637451171875e-05,
      "step": 36320
    },
    {
      "action_loss_avg": 0.002192928298609331,
      "epoch": 3.4637451171875e-05,
      "step": 36320
    },
    {
      "epoch": 3.4637451171875e-05,
      "step": 36320,
      "training_step_time": 7.67225193977356
    },
    {
      "epoch": 3.4638404846191404e-05,
      "model_forward_time": 1.7405037879943848,
      "step": 36321
    },
    {
      "epoch": 3.4638404846191404e-05,
      "step": 36321,
      "training_step_time": 7.57349705696106
    },
    {
      "epoch": 3.4639358520507815e-05,
      "model_forward_time": 1.749959945678711,
      "step": 36322
    },
    {
      "epoch": 3.4639358520507815e-05,
      "step": 36322,
      "training_step_time": 7.540700912475586
    },
    {
      "epoch": 3.464031219482422e-05,
      "model_forward_time": 2.8512628078460693,
      "step": 36323
    },
    {
      "epoch": 3.464031219482422e-05,
      "step": 36323,
      "training_step_time": 7.725161552429199
    },
    {
      "epoch": 3.464126586914062e-05,
      "model_forward_time": 1.7507762908935547,
      "step": 36324
    },
    {
      "epoch": 3.464126586914062e-05,
      "step": 36324,
      "training_step_time": 9.158995151519775
    },
    {
      "epoch": 3.4642219543457034e-05,
      "model_forward_time": 1.7563095092773438,
      "step": 36325
    },
    {
      "epoch": 3.4642219543457034e-05,
      "step": 36325,
      "training_step_time": 8.35464596748352
    },
    {
      "epoch": 3.464317321777344e-05,
      "model_forward_time": 1.7423696517944336,
      "step": 36326
    },
    {
      "epoch": 3.464317321777344e-05,
      "step": 36326,
      "training_step_time": 8.408523797988892
    },
    {
      "epoch": 3.464412689208984e-05,
      "model_forward_time": 1.7521004676818848,
      "step": 36327
    },
    {
      "epoch": 3.464412689208984e-05,
      "step": 36327,
      "training_step_time": 8.337335109710693
    },
    {
      "epoch": 3.464508056640625e-05,
      "model_forward_time": 1.3745412826538086,
      "step": 36328
    },
    {
      "epoch": 3.464508056640625e-05,
      "step": 36328,
      "training_step_time": 8.724857807159424
    },
    {
      "epoch": 3.464603424072266e-05,
      "model_forward_time": 1.748617172241211,
      "step": 36329
    },
    {
      "epoch": 3.464603424072266e-05,
      "step": 36329,
      "training_step_time": 9.491770505905151
    },
    {
      "epoch": 3.464698791503906e-05,
      "grad_norm": 0.11816898733377457,
      "learning_rate": 7.548195207793067e-06,
      "loss": 0.0549,
      "step": 36330
    },
    {
      "epoch": 3.464698791503906e-05,
      "model_forward_time": 1.7491064071655273,
      "step": 36330
    },
    {
      "dynamics_loss_avg": 0.06451500970870257,
      "epoch": 3.464698791503906e-05,
      "step": 36330
    },
    {
      "action_loss_avg": 0.002700054080924019,
      "epoch": 3.464698791503906e-05,
      "step": 36330
    },
    {
      "epoch": 3.464698791503906e-05,
      "step": 36330,
      "training_step_time": 6.784527778625488
    },
    {
      "epoch": 3.464794158935547e-05,
      "model_forward_time": 1.7484309673309326,
      "step": 36331
    },
    {
      "epoch": 3.464794158935547e-05,
      "step": 36331,
      "training_step_time": 6.831446886062622
    },
    {
      "epoch": 3.4648895263671876e-05,
      "model_forward_time": 1.7496800422668457,
      "step": 36332
    },
    {
      "epoch": 3.4648895263671876e-05,
      "step": 36332,
      "training_step_time": 6.752266883850098
    },
    {
      "epoch": 3.464984893798828e-05,
      "model_forward_time": 1.7498538494110107,
      "step": 36333
    },
    {
      "epoch": 3.464984893798828e-05,
      "step": 36333,
      "training_step_time": 6.729689598083496
    },
    {
      "epoch": 3.465080261230469e-05,
      "model_forward_time": 1.767601490020752,
      "step": 36334
    },
    {
      "epoch": 3.465080261230469e-05,
      "step": 36334,
      "training_step_time": 6.798925399780273
    },
    {
      "epoch": 3.4651756286621095e-05,
      "model_forward_time": 1.7549872398376465,
      "step": 36335
    },
    {
      "epoch": 3.4651756286621095e-05,
      "step": 36335,
      "training_step_time": 6.787473201751709
    },
    {
      "epoch": 3.46527099609375e-05,
      "model_forward_time": 1.7477407455444336,
      "step": 36336
    },
    {
      "epoch": 3.46527099609375e-05,
      "step": 36336,
      "training_step_time": 6.825178861618042
    },
    {
      "epoch": 3.465366363525391e-05,
      "model_forward_time": 1.7500457763671875,
      "step": 36337
    },
    {
      "epoch": 3.465366363525391e-05,
      "step": 36337,
      "training_step_time": 6.764232158660889
    },
    {
      "epoch": 3.4654617309570313e-05,
      "model_forward_time": 2.8645637035369873,
      "step": 36338
    },
    {
      "epoch": 3.4654617309570313e-05,
      "step": 36338,
      "training_step_time": 7.647429943084717
    },
    {
      "epoch": 3.465557098388672e-05,
      "model_forward_time": 1.764735221862793,
      "step": 36339
    },
    {
      "epoch": 3.465557098388672e-05,
      "step": 36339,
      "training_step_time": 7.8897864818573
    },
    {
      "epoch": 3.465652465820313e-05,
      "grad_norm": 0.12113536894321442,
      "learning_rate": 7.546772442380324e-06,
      "loss": 0.0552,
      "step": 36340
    },
    {
      "epoch": 3.465652465820313e-05,
      "model_forward_time": 1.7510154247283936,
      "step": 36340
    },
    {
      "dynamics_loss_avg": 0.04038810646161437,
      "epoch": 3.465652465820313e-05,
      "step": 36340
    },
    {
      "action_loss_avg": 0.0009492079931078479,
      "epoch": 3.465652465820313e-05,
      "step": 36340
    },
    {
      "epoch": 3.465652465820313e-05,
      "step": 36340,
      "training_step_time": 6.840099573135376
    },
    {
      "epoch": 3.465747833251953e-05,
      "model_forward_time": 1.748093605041504,
      "step": 36341
    },
    {
      "epoch": 3.465747833251953e-05,
      "step": 36341,
      "training_step_time": 6.83509373664856
    },
    {
      "epoch": 3.4658432006835936e-05,
      "model_forward_time": 1.7487540245056152,
      "step": 36342
    },
    {
      "epoch": 3.4658432006835936e-05,
      "step": 36342,
      "training_step_time": 6.7861151695251465
    },
    {
      "epoch": 3.465938568115235e-05,
      "model_forward_time": 1.7500853538513184,
      "step": 36343
    },
    {
      "epoch": 3.465938568115235e-05,
      "step": 36343,
      "training_step_time": 6.845741510391235
    },
    {
      "epoch": 3.466033935546875e-05,
      "model_forward_time": 1.7593872547149658,
      "step": 36344
    },
    {
      "epoch": 3.466033935546875e-05,
      "step": 36344,
      "training_step_time": 6.73432731628418
    },
    {
      "epoch": 3.4661293029785155e-05,
      "model_forward_time": 1.7496132850646973,
      "step": 36345
    },
    {
      "epoch": 3.4661293029785155e-05,
      "step": 36345,
      "training_step_time": 6.795368909835815
    },
    {
      "epoch": 3.466224670410156e-05,
      "model_forward_time": 1.7514078617095947,
      "step": 36346
    },
    {
      "epoch": 3.466224670410156e-05,
      "step": 36346,
      "training_step_time": 6.820397615432739
    },
    {
      "epoch": 3.466320037841797e-05,
      "model_forward_time": 2.863940954208374,
      "step": 36347
    },
    {
      "epoch": 3.466320037841797e-05,
      "step": 36347,
      "training_step_time": 7.632491111755371
    },
    {
      "epoch": 3.4664154052734374e-05,
      "model_forward_time": 1.7511036396026611,
      "step": 36348
    },
    {
      "epoch": 3.4664154052734374e-05,
      "step": 36348,
      "training_step_time": 9.221815586090088
    },
    {
      "epoch": 3.466510772705078e-05,
      "model_forward_time": 1.766798734664917,
      "step": 36349
    },
    {
      "epoch": 3.466510772705078e-05,
      "step": 36349,
      "training_step_time": 7.208870887756348
    },
    {
      "epoch": 3.466606140136719e-05,
      "grad_norm": 0.10190629214048386,
      "learning_rate": 7.5453493984563765e-06,
      "loss": 0.0571,
      "step": 36350
    },
    {
      "epoch": 3.466606140136719e-05,
      "model_forward_time": 1.754817247390747,
      "step": 36350
    },
    {
      "dynamics_loss_avg": 0.05561583377420902,
      "epoch": 3.466606140136719e-05,
      "step": 36350
    },
    {
      "action_loss_avg": 0.0018754740303847939,
      "epoch": 3.466606140136719e-05,
      "step": 36350
    },
    {
      "epoch": 3.466606140136719e-05,
      "step": 36350,
      "training_step_time": 6.7689878940582275
    },
    {
      "epoch": 3.466701507568359e-05,
      "model_forward_time": 1.7566277980804443,
      "step": 36351
    },
    {
      "epoch": 3.466701507568359e-05,
      "step": 36351,
      "training_step_time": 6.757928133010864
    },
    {
      "epoch": 3.466796875e-05,
      "model_forward_time": 1.768674612045288,
      "step": 36352
    },
    {
      "epoch": 3.466796875e-05,
      "step": 36352,
      "training_step_time": 6.704176187515259
    },
    {
      "epoch": 3.466892242431641e-05,
      "model_forward_time": 1.7577800750732422,
      "step": 36353
    },
    {
      "epoch": 3.466892242431641e-05,
      "step": 36353,
      "training_step_time": 6.713078022003174
    },
    {
      "epoch": 3.466987609863281e-05,
      "model_forward_time": 1.7497222423553467,
      "step": 36354
    },
    {
      "epoch": 3.466987609863281e-05,
      "step": 36354,
      "training_step_time": 6.7239158153533936
    },
    {
      "epoch": 3.4670829772949216e-05,
      "model_forward_time": 1.748685359954834,
      "step": 36355
    },
    {
      "epoch": 3.4670829772949216e-05,
      "step": 36355,
      "training_step_time": 6.771085023880005
    },
    {
      "epoch": 3.4671783447265627e-05,
      "model_forward_time": 2.8936941623687744,
      "step": 36356
    },
    {
      "epoch": 3.4671783447265627e-05,
      "step": 36356,
      "training_step_time": 7.631715297698975
    },
    {
      "epoch": 3.467273712158203e-05,
      "model_forward_time": 1.7531802654266357,
      "step": 36357
    },
    {
      "epoch": 3.467273712158203e-05,
      "step": 36357,
      "training_step_time": 10.2156982421875
    },
    {
      "epoch": 3.4673690795898435e-05,
      "model_forward_time": 1.7537462711334229,
      "step": 36358
    },
    {
      "epoch": 3.4673690795898435e-05,
      "step": 36358,
      "training_step_time": 7.554931879043579
    },
    {
      "epoch": 3.4674644470214845e-05,
      "model_forward_time": 1.7549800872802734,
      "step": 36359
    },
    {
      "epoch": 3.4674644470214845e-05,
      "step": 36359,
      "training_step_time": 6.725009918212891
    },
    {
      "epoch": 3.467559814453125e-05,
      "grad_norm": 0.1352664828300476,
      "learning_rate": 7.543926076176846e-06,
      "loss": 0.0565,
      "step": 36360
    },
    {
      "epoch": 3.467559814453125e-05,
      "model_forward_time": 1.7614328861236572,
      "step": 36360
    },
    {
      "dynamics_loss_avg": 0.06291905343532563,
      "epoch": 3.467559814453125e-05,
      "step": 36360
    },
    {
      "action_loss_avg": 0.001710225731949322,
      "epoch": 3.467559814453125e-05,
      "step": 36360
    },
    {
      "epoch": 3.467559814453125e-05,
      "step": 36360,
      "training_step_time": 6.748351812362671
    },
    {
      "epoch": 3.4676551818847653e-05,
      "model_forward_time": 1.749459981918335,
      "step": 36361
    },
    {
      "epoch": 3.4676551818847653e-05,
      "step": 36361,
      "training_step_time": 6.735258340835571
    },
    {
      "epoch": 3.4677505493164064e-05,
      "model_forward_time": 1.7500383853912354,
      "step": 36362
    },
    {
      "epoch": 3.4677505493164064e-05,
      "step": 36362,
      "training_step_time": 6.777015447616577
    },
    {
      "epoch": 3.467845916748047e-05,
      "model_forward_time": 1.7481622695922852,
      "step": 36363
    },
    {
      "epoch": 3.467845916748047e-05,
      "step": 36363,
      "training_step_time": 6.708954572677612
    },
    {
      "epoch": 3.467941284179687e-05,
      "model_forward_time": 1.7526304721832275,
      "step": 36364
    },
    {
      "epoch": 3.467941284179687e-05,
      "step": 36364,
      "training_step_time": 6.72801947593689
    },
    {
      "epoch": 3.468036651611328e-05,
      "model_forward_time": 2.9821574687957764,
      "step": 36365
    },
    {
      "epoch": 3.468036651611328e-05,
      "step": 36365,
      "training_step_time": 7.7574732303619385
    },
    {
      "epoch": 3.468132019042969e-05,
      "model_forward_time": 1.755957841873169,
      "step": 36366
    },
    {
      "epoch": 3.468132019042969e-05,
      "step": 36366,
      "training_step_time": 8.274563550949097
    },
    {
      "epoch": 3.468227386474609e-05,
      "model_forward_time": 1.7552368640899658,
      "step": 36367
    },
    {
      "epoch": 3.468227386474609e-05,
      "step": 36367,
      "training_step_time": 7.589164972305298
    },
    {
      "epoch": 3.46832275390625e-05,
      "model_forward_time": 1.7622888088226318,
      "step": 36368
    },
    {
      "epoch": 3.46832275390625e-05,
      "step": 36368,
      "training_step_time": 6.772530794143677
    },
    {
      "epoch": 3.4684181213378906e-05,
      "model_forward_time": 1.7558555603027344,
      "step": 36369
    },
    {
      "epoch": 3.4684181213378906e-05,
      "step": 36369,
      "training_step_time": 6.837309837341309
    },
    {
      "epoch": 3.468513488769531e-05,
      "grad_norm": 0.14590522646903992,
      "learning_rate": 7.542502475697385e-06,
      "loss": 0.0566,
      "step": 36370
    },
    {
      "epoch": 3.468513488769531e-05,
      "model_forward_time": 1.750612497329712,
      "step": 36370
    },
    {
      "dynamics_loss_avg": 0.05311394203454256,
      "epoch": 3.468513488769531e-05,
      "step": 36370
    },
    {
      "action_loss_avg": 0.0019412617955822497,
      "epoch": 3.468513488769531e-05,
      "step": 36370
    },
    {
      "epoch": 3.468513488769531e-05,
      "step": 36370,
      "training_step_time": 6.777260780334473
    },
    {
      "epoch": 3.468608856201172e-05,
      "model_forward_time": 1.7490613460540771,
      "step": 36371
    },
    {
      "epoch": 3.468608856201172e-05,
      "step": 36371,
      "training_step_time": 6.81916069984436
    },
    {
      "epoch": 3.4687042236328125e-05,
      "model_forward_time": 1.7541818618774414,
      "step": 36372
    },
    {
      "epoch": 3.4687042236328125e-05,
      "step": 36372,
      "training_step_time": 6.832482576370239
    },
    {
      "epoch": 3.468799591064453e-05,
      "model_forward_time": 1.7521288394927979,
      "step": 36373
    },
    {
      "epoch": 3.468799591064453e-05,
      "step": 36373,
      "training_step_time": 6.828640699386597
    },
    {
      "epoch": 3.468894958496094e-05,
      "model_forward_time": 2.9759106636047363,
      "step": 36374
    },
    {
      "epoch": 3.468894958496094e-05,
      "step": 36374,
      "training_step_time": 7.784501075744629
    },
    {
      "epoch": 3.4689903259277344e-05,
      "model_forward_time": 1.7482445240020752,
      "step": 36375
    },
    {
      "epoch": 3.4689903259277344e-05,
      "step": 36375,
      "training_step_time": 8.817055940628052
    },
    {
      "epoch": 3.469085693359375e-05,
      "model_forward_time": 1.7489819526672363,
      "step": 36376
    },
    {
      "epoch": 3.469085693359375e-05,
      "step": 36376,
      "training_step_time": 7.633182525634766
    },
    {
      "epoch": 3.469181060791016e-05,
      "model_forward_time": 1.7560327053070068,
      "step": 36377
    },
    {
      "epoch": 3.469181060791016e-05,
      "step": 36377,
      "training_step_time": 7.475708723068237
    },
    {
      "epoch": 3.469276428222656e-05,
      "model_forward_time": 1.751394510269165,
      "step": 36378
    },
    {
      "epoch": 3.469276428222656e-05,
      "step": 36378,
      "training_step_time": 6.6593005657196045
    },
    {
      "epoch": 3.469371795654297e-05,
      "model_forward_time": 1.754556655883789,
      "step": 36379
    },
    {
      "epoch": 3.469371795654297e-05,
      "step": 36379,
      "training_step_time": 6.749536037445068
    },
    {
      "epoch": 3.469467163085938e-05,
      "grad_norm": 0.12056107074022293,
      "learning_rate": 7.5410785971736755e-06,
      "loss": 0.0582,
      "step": 36380
    },
    {
      "epoch": 3.469467163085938e-05,
      "model_forward_time": 1.7484209537506104,
      "step": 36380
    },
    {
      "dynamics_loss_avg": 0.05595815945416689,
      "epoch": 3.469467163085938e-05,
      "step": 36380
    },
    {
      "action_loss_avg": 0.0027701194165274503,
      "epoch": 3.469467163085938e-05,
      "step": 36380
    },
    {
      "epoch": 3.469467163085938e-05,
      "step": 36380,
      "training_step_time": 6.803646564483643
    },
    {
      "epoch": 3.469562530517578e-05,
      "model_forward_time": 1.7594170570373535,
      "step": 36381
    },
    {
      "epoch": 3.469562530517578e-05,
      "step": 36381,
      "training_step_time": 6.774306058883667
    },
    {
      "epoch": 3.4696578979492185e-05,
      "model_forward_time": 1.7490968704223633,
      "step": 36382
    },
    {
      "epoch": 3.4696578979492185e-05,
      "step": 36382,
      "training_step_time": 6.785302400588989
    },
    {
      "epoch": 3.4697532653808596e-05,
      "model_forward_time": 2.9125216007232666,
      "step": 36383
    },
    {
      "epoch": 3.4697532653808596e-05,
      "step": 36383,
      "training_step_time": 7.661441087722778
    },
    {
      "epoch": 3.4698486328125e-05,
      "model_forward_time": 1.7467458248138428,
      "step": 36384
    },
    {
      "epoch": 3.4698486328125e-05,
      "step": 36384,
      "training_step_time": 9.712297439575195
    },
    {
      "epoch": 3.4699440002441404e-05,
      "model_forward_time": 1.7444498538970947,
      "step": 36385
    },
    {
      "epoch": 3.4699440002441404e-05,
      "step": 36385,
      "training_step_time": 7.7641167640686035
    },
    {
      "epoch": 3.4700393676757815e-05,
      "model_forward_time": 1.751739501953125,
      "step": 36386
    },
    {
      "epoch": 3.4700393676757815e-05,
      "step": 36386,
      "training_step_time": 7.617599725723267
    },
    {
      "epoch": 3.470134735107422e-05,
      "model_forward_time": 1.73362398147583,
      "step": 36387
    },
    {
      "epoch": 3.470134735107422e-05,
      "step": 36387,
      "training_step_time": 6.724894046783447
    },
    {
      "epoch": 3.470230102539062e-05,
      "model_forward_time": 1.7429845333099365,
      "step": 36388
    },
    {
      "epoch": 3.470230102539062e-05,
      "step": 36388,
      "training_step_time": 6.705971717834473
    },
    {
      "epoch": 3.4703254699707034e-05,
      "model_forward_time": 1.732893943786621,
      "step": 36389
    },
    {
      "epoch": 3.4703254699707034e-05,
      "step": 36389,
      "training_step_time": 6.7188122272491455
    },
    {
      "epoch": 3.470420837402344e-05,
      "grad_norm": 0.1032862588763237,
      "learning_rate": 7.53965444076143e-06,
      "loss": 0.0565,
      "step": 36390
    },
    {
      "epoch": 3.470420837402344e-05,
      "model_forward_time": 1.7261223793029785,
      "step": 36390
    },
    {
      "dynamics_loss_avg": 0.045315248612314465,
      "epoch": 3.470420837402344e-05,
      "step": 36390
    },
    {
      "action_loss_avg": 0.0015549207164440304,
      "epoch": 3.470420837402344e-05,
      "step": 36390
    },
    {
      "epoch": 3.470420837402344e-05,
      "step": 36390,
      "training_step_time": 6.707932472229004
    },
    {
      "epoch": 3.470516204833984e-05,
      "model_forward_time": 1.7462985515594482,
      "step": 36391
    },
    {
      "epoch": 3.470516204833984e-05,
      "step": 36391,
      "training_step_time": 6.696855068206787
    },
    {
      "epoch": 3.470611572265625e-05,
      "model_forward_time": 1.8030359745025635,
      "step": 36392
    },
    {
      "epoch": 3.470611572265625e-05,
      "step": 36392,
      "training_step_time": 7.1613993644714355
    },
    {
      "epoch": 3.470706939697266e-05,
      "model_forward_time": 2.946061372756958,
      "step": 36393
    },
    {
      "epoch": 3.470706939697266e-05,
      "step": 36393,
      "training_step_time": 8.28423261642456
    },
    {
      "epoch": 3.470802307128906e-05,
      "model_forward_time": 1.731435775756836,
      "step": 36394
    },
    {
      "epoch": 3.470802307128906e-05,
      "step": 36394,
      "training_step_time": 7.754700183868408
    },
    {
      "epoch": 3.470897674560547e-05,
      "model_forward_time": 1.7308368682861328,
      "step": 36395
    },
    {
      "epoch": 3.470897674560547e-05,
      "step": 36395,
      "training_step_time": 7.598189115524292
    },
    {
      "epoch": 3.4709930419921876e-05,
      "model_forward_time": 1.7284014225006104,
      "step": 36396
    },
    {
      "epoch": 3.4709930419921876e-05,
      "step": 36396,
      "training_step_time": 6.771149396896362
    },
    {
      "epoch": 3.471088409423828e-05,
      "model_forward_time": 1.7625441551208496,
      "step": 36397
    },
    {
      "epoch": 3.471088409423828e-05,
      "step": 36397,
      "training_step_time": 6.7635462284088135
    },
    {
      "epoch": 3.471183776855469e-05,
      "model_forward_time": 1.7415246963500977,
      "step": 36398
    },
    {
      "epoch": 3.471183776855469e-05,
      "step": 36398,
      "training_step_time": 6.755110502243042
    },
    {
      "epoch": 3.4712791442871095e-05,
      "model_forward_time": 1.7316880226135254,
      "step": 36399
    },
    {
      "epoch": 3.4712791442871095e-05,
      "step": 36399,
      "training_step_time": 6.748829364776611
    },
    {
      "epoch": 3.47137451171875e-05,
      "grad_norm": 0.12996530532836914,
      "learning_rate": 7.538230006616395e-06,
      "loss": 0.0583,
      "step": 36400
    },
    {
      "epoch": 3.47137451171875e-05,
      "model_forward_time": 1.721461534500122,
      "step": 36400
    },
    {
      "dynamics_loss_avg": 0.05952354315668344,
      "epoch": 3.47137451171875e-05,
      "step": 36400
    },
    {
      "action_loss_avg": 0.0016378847183659672,
      "epoch": 3.47137451171875e-05,
      "step": 36400
    },
    {
      "epoch": 3.47137451171875e-05,
      "step": 36400,
      "training_step_time": 6.762849807739258
    },
    {
      "epoch": 3.471469879150391e-05,
      "model_forward_time": 1.8032076358795166,
      "step": 36401
    },
    {
      "epoch": 3.471469879150391e-05,
      "step": 36401,
      "training_step_time": 7.255011081695557
    },
    {
      "epoch": 3.4715652465820313e-05,
      "model_forward_time": 2.9702179431915283,
      "step": 36402
    },
    {
      "epoch": 3.4715652465820313e-05,
      "step": 36402,
      "training_step_time": 8.33880877494812
    },
    {
      "epoch": 3.471660614013672e-05,
      "model_forward_time": 1.7297987937927246,
      "step": 36403
    },
    {
      "epoch": 3.471660614013672e-05,
      "step": 36403,
      "training_step_time": 7.856485366821289
    },
    {
      "epoch": 3.471755981445313e-05,
      "model_forward_time": 1.7372090816497803,
      "step": 36404
    },
    {
      "epoch": 3.471755981445313e-05,
      "step": 36404,
      "training_step_time": 8.13161039352417
    },
    {
      "epoch": 3.471851348876953e-05,
      "model_forward_time": 1.7420015335083008,
      "step": 36405
    },
    {
      "epoch": 3.471851348876953e-05,
      "step": 36405,
      "training_step_time": 6.719206809997559
    },
    {
      "epoch": 3.4719467163085936e-05,
      "model_forward_time": 1.7533068656921387,
      "step": 36406
    },
    {
      "epoch": 3.4719467163085936e-05,
      "step": 36406,
      "training_step_time": 6.713125944137573
    },
    {
      "epoch": 3.472042083740234e-05,
      "model_forward_time": 1.7318267822265625,
      "step": 36407
    },
    {
      "epoch": 3.472042083740234e-05,
      "step": 36407,
      "training_step_time": 6.737291574478149
    },
    {
      "epoch": 3.472137451171875e-05,
      "model_forward_time": 1.7439935207366943,
      "step": 36408
    },
    {
      "epoch": 3.472137451171875e-05,
      "step": 36408,
      "training_step_time": 6.725340366363525
    },
    {
      "epoch": 3.4722328186035155e-05,
      "model_forward_time": 1.7709379196166992,
      "step": 36409
    },
    {
      "epoch": 3.4722328186035155e-05,
      "step": 36409,
      "training_step_time": 6.661945343017578
    },
    {
      "epoch": 3.472328186035156e-05,
      "grad_norm": 0.15049387514591217,
      "learning_rate": 7.536805294894343e-06,
      "loss": 0.0573,
      "step": 36410
    },
    {
      "epoch": 3.472328186035156e-05,
      "model_forward_time": 1.8104496002197266,
      "step": 36410
    },
    {
      "dynamics_loss_avg": 0.047719692811369896,
      "epoch": 3.472328186035156e-05,
      "step": 36410
    },
    {
      "action_loss_avg": 0.001121285365661606,
      "epoch": 3.472328186035156e-05,
      "step": 36410
    },
    {
      "epoch": 3.472328186035156e-05,
      "step": 36410,
      "training_step_time": 7.224369287490845
    },
    {
      "epoch": 3.472423553466797e-05,
      "model_forward_time": 2.897897243499756,
      "step": 36411
    },
    {
      "epoch": 3.472423553466797e-05,
      "step": 36411,
      "training_step_time": 8.330203294754028
    },
    {
      "epoch": 3.4725189208984374e-05,
      "model_forward_time": 1.8349695205688477,
      "step": 36412
    },
    {
      "epoch": 3.4725189208984374e-05,
      "step": 36412,
      "training_step_time": 7.826465368270874
    },
    {
      "epoch": 3.472614288330078e-05,
      "model_forward_time": 1.7514686584472656,
      "step": 36413
    },
    {
      "epoch": 3.472614288330078e-05,
      "step": 36413,
      "training_step_time": 7.684130668640137
    },
    {
      "epoch": 3.472709655761719e-05,
      "model_forward_time": 1.734950304031372,
      "step": 36414
    },
    {
      "epoch": 3.472709655761719e-05,
      "step": 36414,
      "training_step_time": 7.160278797149658
    },
    {
      "epoch": 3.472805023193359e-05,
      "model_forward_time": 1.7292413711547852,
      "step": 36415
    },
    {
      "epoch": 3.472805023193359e-05,
      "step": 36415,
      "training_step_time": 6.681166648864746
    },
    {
      "epoch": 3.472900390625e-05,
      "model_forward_time": 1.7584185600280762,
      "step": 36416
    },
    {
      "epoch": 3.472900390625e-05,
      "step": 36416,
      "training_step_time": 6.719599962234497
    },
    {
      "epoch": 3.472995758056641e-05,
      "model_forward_time": 1.7532804012298584,
      "step": 36417
    },
    {
      "epoch": 3.472995758056641e-05,
      "step": 36417,
      "training_step_time": 6.742312431335449
    },
    {
      "epoch": 3.473091125488281e-05,
      "model_forward_time": 1.7620930671691895,
      "step": 36418
    },
    {
      "epoch": 3.473091125488281e-05,
      "step": 36418,
      "training_step_time": 6.7003493309021
    },
    {
      "epoch": 3.4731864929199216e-05,
      "model_forward_time": 1.7940855026245117,
      "step": 36419
    },
    {
      "epoch": 3.4731864929199216e-05,
      "step": 36419,
      "training_step_time": 7.18821907043457
    },
    {
      "epoch": 3.473281860351563e-05,
      "grad_norm": 0.12843045592308044,
      "learning_rate": 7.535380305751076e-06,
      "loss": 0.0578,
      "step": 36420
    },
    {
      "epoch": 3.473281860351563e-05,
      "model_forward_time": 2.8896241188049316,
      "step": 36420
    },
    {
      "dynamics_loss_avg": 0.05785158835351467,
      "epoch": 3.473281860351563e-05,
      "step": 36420
    },
    {
      "action_loss_avg": 0.0017653223418164997,
      "epoch": 3.473281860351563e-05,
      "step": 36420
    },
    {
      "epoch": 3.473281860351563e-05,
      "step": 36420,
      "training_step_time": 8.536897897720337
    },
    {
      "epoch": 3.473377227783203e-05,
      "model_forward_time": 1.7518882751464844,
      "step": 36421
    },
    {
      "epoch": 3.473377227783203e-05,
      "step": 36421,
      "training_step_time": 10.337828397750854
    },
    {
      "epoch": 3.4734725952148435e-05,
      "model_forward_time": 1.789670467376709,
      "step": 36422
    },
    {
      "epoch": 3.4734725952148435e-05,
      "step": 36422,
      "training_step_time": 7.873628616333008
    },
    {
      "epoch": 3.4735679626464845e-05,
      "model_forward_time": 1.7601003646850586,
      "step": 36423
    },
    {
      "epoch": 3.4735679626464845e-05,
      "step": 36423,
      "training_step_time": 7.195963144302368
    },
    {
      "epoch": 3.473663330078125e-05,
      "model_forward_time": 1.7735404968261719,
      "step": 36424
    },
    {
      "epoch": 3.473663330078125e-05,
      "step": 36424,
      "training_step_time": 6.712280988693237
    },
    {
      "epoch": 3.4737586975097654e-05,
      "model_forward_time": 1.7695887088775635,
      "step": 36425
    },
    {
      "epoch": 3.4737586975097654e-05,
      "step": 36425,
      "training_step_time": 6.794666528701782
    },
    {
      "epoch": 3.4738540649414064e-05,
      "model_forward_time": 1.758096694946289,
      "step": 36426
    },
    {
      "epoch": 3.4738540649414064e-05,
      "step": 36426,
      "training_step_time": 6.765660047531128
    },
    {
      "epoch": 3.473949432373047e-05,
      "model_forward_time": 1.761620283126831,
      "step": 36427
    },
    {
      "epoch": 3.473949432373047e-05,
      "step": 36427,
      "training_step_time": 6.719191551208496
    },
    {
      "epoch": 3.474044799804687e-05,
      "model_forward_time": 1.757758617401123,
      "step": 36428
    },
    {
      "epoch": 3.474044799804687e-05,
      "step": 36428,
      "training_step_time": 7.173462152481079
    },
    {
      "epoch": 3.474140167236328e-05,
      "model_forward_time": 2.920592784881592,
      "step": 36429
    },
    {
      "epoch": 3.474140167236328e-05,
      "step": 36429,
      "training_step_time": 8.5248441696167
    },
    {
      "epoch": 3.474235534667969e-05,
      "grad_norm": 0.12939508259296417,
      "learning_rate": 7.533955039342431e-06,
      "loss": 0.0608,
      "step": 36430
    },
    {
      "epoch": 3.474235534667969e-05,
      "model_forward_time": 1.7484207153320312,
      "step": 36430
    },
    {
      "dynamics_loss_avg": 0.0552816204726696,
      "epoch": 3.474235534667969e-05,
      "step": 36430
    },
    {
      "action_loss_avg": 0.0023582499736221505,
      "epoch": 3.474235534667969e-05,
      "step": 36430
    },
    {
      "epoch": 3.474235534667969e-05,
      "step": 36430,
      "training_step_time": 7.73278546333313
    },
    {
      "epoch": 3.474330902099609e-05,
      "model_forward_time": 1.7584407329559326,
      "step": 36431
    },
    {
      "epoch": 3.474330902099609e-05,
      "step": 36431,
      "training_step_time": 9.403379678726196
    },
    {
      "epoch": 3.47442626953125e-05,
      "model_forward_time": 1.7586266994476318,
      "step": 36432
    },
    {
      "epoch": 3.47442626953125e-05,
      "step": 36432,
      "training_step_time": 7.463890790939331
    },
    {
      "epoch": 3.4745216369628906e-05,
      "model_forward_time": 1.761216640472412,
      "step": 36433
    },
    {
      "epoch": 3.4745216369628906e-05,
      "step": 36433,
      "training_step_time": 6.702136516571045
    },
    {
      "epoch": 3.474617004394531e-05,
      "model_forward_time": 1.7586069107055664,
      "step": 36434
    },
    {
      "epoch": 3.474617004394531e-05,
      "step": 36434,
      "training_step_time": 6.741976022720337
    },
    {
      "epoch": 3.474712371826172e-05,
      "model_forward_time": 1.7554736137390137,
      "step": 36435
    },
    {
      "epoch": 3.474712371826172e-05,
      "step": 36435,
      "training_step_time": 6.724487543106079
    },
    {
      "epoch": 3.4748077392578125e-05,
      "model_forward_time": 1.7780940532684326,
      "step": 36436
    },
    {
      "epoch": 3.4748077392578125e-05,
      "step": 36436,
      "training_step_time": 6.725589036941528
    },
    {
      "epoch": 3.474903106689453e-05,
      "model_forward_time": 1.7601032257080078,
      "step": 36437
    },
    {
      "epoch": 3.474903106689453e-05,
      "step": 36437,
      "training_step_time": 6.770464181900024
    },
    {
      "epoch": 3.474998474121094e-05,
      "model_forward_time": 1.7998707294464111,
      "step": 36438
    },
    {
      "epoch": 3.474998474121094e-05,
      "step": 36438,
      "training_step_time": 8.634984493255615
    },
    {
      "epoch": 3.4750938415527344e-05,
      "model_forward_time": 2.94716477394104,
      "step": 36439
    },
    {
      "epoch": 3.4750938415527344e-05,
      "step": 36439,
      "training_step_time": 9.602546691894531
    },
    {
      "epoch": 3.475189208984375e-05,
      "grad_norm": 0.12489473819732666,
      "learning_rate": 7.532529495824274e-06,
      "loss": 0.0596,
      "step": 36440
    },
    {
      "epoch": 3.475189208984375e-05,
      "model_forward_time": 1.7535932064056396,
      "step": 36440
    },
    {
      "dynamics_loss_avg": 0.059774482529610394,
      "epoch": 3.475189208984375e-05,
      "step": 36440
    },
    {
      "action_loss_avg": 0.0015097565366886557,
      "epoch": 3.475189208984375e-05,
      "step": 36440
    },
    {
      "epoch": 3.475189208984375e-05,
      "step": 36440,
      "training_step_time": 7.791489124298096
    },
    {
      "epoch": 3.475284576416016e-05,
      "model_forward_time": 1.7472960948944092,
      "step": 36441
    },
    {
      "epoch": 3.475284576416016e-05,
      "step": 36441,
      "training_step_time": 7.932210683822632
    },
    {
      "epoch": 3.475379943847656e-05,
      "model_forward_time": 1.7760834693908691,
      "step": 36442
    },
    {
      "epoch": 3.475379943847656e-05,
      "step": 36442,
      "training_step_time": 6.814045429229736
    },
    {
      "epoch": 3.475475311279297e-05,
      "model_forward_time": 1.748467206954956,
      "step": 36443
    },
    {
      "epoch": 3.475475311279297e-05,
      "step": 36443,
      "training_step_time": 6.763540267944336
    },
    {
      "epoch": 3.475570678710938e-05,
      "model_forward_time": 1.7500450611114502,
      "step": 36444
    },
    {
      "epoch": 3.475570678710938e-05,
      "step": 36444,
      "training_step_time": 6.770189046859741
    },
    {
      "epoch": 3.475666046142578e-05,
      "model_forward_time": 1.7517218589782715,
      "step": 36445
    },
    {
      "epoch": 3.475666046142578e-05,
      "step": 36445,
      "training_step_time": 6.830331087112427
    },
    {
      "epoch": 3.4757614135742186e-05,
      "model_forward_time": 1.7521429061889648,
      "step": 36446
    },
    {
      "epoch": 3.4757614135742186e-05,
      "step": 36446,
      "training_step_time": 6.772234916687012
    },
    {
      "epoch": 3.4758567810058596e-05,
      "model_forward_time": 1.8226914405822754,
      "step": 36447
    },
    {
      "epoch": 3.4758567810058596e-05,
      "step": 36447,
      "training_step_time": 7.54568076133728
    },
    {
      "epoch": 3.4759521484375e-05,
      "model_forward_time": 2.9710819721221924,
      "step": 36448
    },
    {
      "epoch": 3.4759521484375e-05,
      "step": 36448,
      "training_step_time": 7.837320327758789
    },
    {
      "epoch": 3.4760475158691404e-05,
      "model_forward_time": 1.3806428909301758,
      "step": 36449
    },
    {
      "epoch": 3.4760475158691404e-05,
      "step": 36449,
      "training_step_time": 10.132166624069214
    },
    {
      "epoch": 3.4761428833007815e-05,
      "grad_norm": 0.0991595983505249,
      "learning_rate": 7.531103675352497e-06,
      "loss": 0.0617,
      "step": 36450
    },
    {
      "epoch": 3.4761428833007815e-05,
      "model_forward_time": 1.7489652633666992,
      "step": 36450
    },
    {
      "dynamics_loss_avg": 0.03885496202856302,
      "epoch": 3.4761428833007815e-05,
      "step": 36450
    },
    {
      "action_loss_avg": 0.0017785500720492564,
      "epoch": 3.4761428833007815e-05,
      "step": 36450
    },
    {
      "epoch": 3.4761428833007815e-05,
      "step": 36450,
      "training_step_time": 7.888689756393433
    },
    {
      "epoch": 3.476238250732422e-05,
      "model_forward_time": 1.746032953262329,
      "step": 36451
    },
    {
      "epoch": 3.476238250732422e-05,
      "step": 36451,
      "training_step_time": 6.733320951461792
    },
    {
      "epoch": 3.476333618164062e-05,
      "model_forward_time": 1.7512092590332031,
      "step": 36452
    },
    {
      "epoch": 3.476333618164062e-05,
      "step": 36452,
      "training_step_time": 6.761012315750122
    },
    {
      "epoch": 3.4764289855957034e-05,
      "model_forward_time": 1.7587385177612305,
      "step": 36453
    },
    {
      "epoch": 3.4764289855957034e-05,
      "step": 36453,
      "training_step_time": 6.836157321929932
    },
    {
      "epoch": 3.476524353027344e-05,
      "model_forward_time": 1.7482807636260986,
      "step": 36454
    },
    {
      "epoch": 3.476524353027344e-05,
      "step": 36454,
      "training_step_time": 6.691308498382568
    },
    {
      "epoch": 3.476619720458984e-05,
      "model_forward_time": 1.7516875267028809,
      "step": 36455
    },
    {
      "epoch": 3.476619720458984e-05,
      "step": 36455,
      "training_step_time": 6.709555387496948
    },
    {
      "epoch": 3.476715087890625e-05,
      "model_forward_time": 1.817032814025879,
      "step": 36456
    },
    {
      "epoch": 3.476715087890625e-05,
      "step": 36456,
      "training_step_time": 8.380645275115967
    },
    {
      "epoch": 3.476810455322266e-05,
      "model_forward_time": 2.8651740550994873,
      "step": 36457
    },
    {
      "epoch": 3.476810455322266e-05,
      "step": 36457,
      "training_step_time": 8.322748184204102
    },
    {
      "epoch": 3.476905822753906e-05,
      "model_forward_time": 1.7553377151489258,
      "step": 36458
    },
    {
      "epoch": 3.476905822753906e-05,
      "step": 36458,
      "training_step_time": 9.055168867111206
    },
    {
      "epoch": 3.477001190185547e-05,
      "model_forward_time": 1.763476848602295,
      "step": 36459
    },
    {
      "epoch": 3.477001190185547e-05,
      "step": 36459,
      "training_step_time": 8.223110914230347
    },
    {
      "epoch": 3.4770965576171876e-05,
      "grad_norm": 0.15653720498085022,
      "learning_rate": 7.5296775780830286e-06,
      "loss": 0.0581,
      "step": 36460
    },
    {
      "epoch": 3.4770965576171876e-05,
      "model_forward_time": 1.7344284057617188,
      "step": 36460
    },
    {
      "dynamics_loss_avg": 0.05158265437930822,
      "epoch": 3.4770965576171876e-05,
      "step": 36460
    },
    {
      "action_loss_avg": 0.0018597079091705383,
      "epoch": 3.4770965576171876e-05,
      "step": 36460
    },
    {
      "epoch": 3.4770965576171876e-05,
      "step": 36460,
      "training_step_time": 7.052285432815552
    },
    {
      "epoch": 3.477191925048828e-05,
      "model_forward_time": 1.7579047679901123,
      "step": 36461
    },
    {
      "epoch": 3.477191925048828e-05,
      "step": 36461,
      "training_step_time": 6.639808177947998
    },
    {
      "epoch": 3.477287292480469e-05,
      "model_forward_time": 1.7511439323425293,
      "step": 36462
    },
    {
      "epoch": 3.477287292480469e-05,
      "step": 36462,
      "training_step_time": 6.623164653778076
    },
    {
      "epoch": 3.4773826599121095e-05,
      "model_forward_time": 1.7471492290496826,
      "step": 36463
    },
    {
      "epoch": 3.4773826599121095e-05,
      "step": 36463,
      "training_step_time": 6.675424814224243
    },
    {
      "epoch": 3.47747802734375e-05,
      "model_forward_time": 1.7689998149871826,
      "step": 36464
    },
    {
      "epoch": 3.47747802734375e-05,
      "step": 36464,
      "training_step_time": 6.617248773574829
    },
    {
      "epoch": 3.477573394775391e-05,
      "model_forward_time": 1.7730040550231934,
      "step": 36465
    },
    {
      "epoch": 3.477573394775391e-05,
      "step": 36465,
      "training_step_time": 7.561643123626709
    },
    {
      "epoch": 3.4776687622070314e-05,
      "model_forward_time": 2.8797004222869873,
      "step": 36466
    },
    {
      "epoch": 3.4776687622070314e-05,
      "step": 36466,
      "training_step_time": 7.706451177597046
    },
    {
      "epoch": 3.477764129638672e-05,
      "model_forward_time": 1.756762981414795,
      "step": 36467
    },
    {
      "epoch": 3.477764129638672e-05,
      "step": 36467,
      "training_step_time": 7.717864274978638
    },
    {
      "epoch": 3.477859497070313e-05,
      "model_forward_time": 1.7603566646575928,
      "step": 36468
    },
    {
      "epoch": 3.477859497070313e-05,
      "step": 36468,
      "training_step_time": 10.210705518722534
    },
    {
      "epoch": 3.477954864501953e-05,
      "model_forward_time": 1.7515432834625244,
      "step": 36469
    },
    {
      "epoch": 3.477954864501953e-05,
      "step": 36469,
      "training_step_time": 7.529757022857666
    },
    {
      "epoch": 3.4780502319335936e-05,
      "grad_norm": 0.09938783198595047,
      "learning_rate": 7.528251204171824e-06,
      "loss": 0.058,
      "step": 36470
    },
    {
      "epoch": 3.4780502319335936e-05,
      "model_forward_time": 1.7432348728179932,
      "step": 36470
    },
    {
      "dynamics_loss_avg": 0.059497151896357535,
      "epoch": 3.4780502319335936e-05,
      "step": 36470
    },
    {
      "action_loss_avg": 0.0020829797664191573,
      "epoch": 3.4780502319335936e-05,
      "step": 36470
    },
    {
      "epoch": 3.4780502319335936e-05,
      "step": 36470,
      "training_step_time": 6.786767959594727
    },
    {
      "epoch": 3.478145599365234e-05,
      "model_forward_time": 1.7557346820831299,
      "step": 36471
    },
    {
      "epoch": 3.478145599365234e-05,
      "step": 36471,
      "training_step_time": 6.737428903579712
    },
    {
      "epoch": 3.478240966796875e-05,
      "model_forward_time": 1.7090187072753906,
      "step": 36472
    },
    {
      "epoch": 3.478240966796875e-05,
      "step": 36472,
      "training_step_time": 6.77783989906311
    },
    {
      "epoch": 3.4783363342285155e-05,
      "model_forward_time": 1.7352418899536133,
      "step": 36473
    },
    {
      "epoch": 3.4783363342285155e-05,
      "step": 36473,
      "training_step_time": 6.788396596908569
    },
    {
      "epoch": 3.478431701660156e-05,
      "model_forward_time": 1.3562018871307373,
      "step": 36474
    },
    {
      "epoch": 3.478431701660156e-05,
      "step": 36474,
      "training_step_time": 7.5491087436676025
    },
    {
      "epoch": 3.478527069091797e-05,
      "model_forward_time": 1.7297332286834717,
      "step": 36475
    },
    {
      "epoch": 3.478527069091797e-05,
      "step": 36475,
      "training_step_time": 7.638216495513916
    },
    {
      "epoch": 3.4786224365234374e-05,
      "model_forward_time": 2.941746711730957,
      "step": 36476
    },
    {
      "epoch": 3.4786224365234374e-05,
      "step": 36476,
      "training_step_time": 8.208710193634033
    },
    {
      "epoch": 3.478717803955078e-05,
      "model_forward_time": 1.751258134841919,
      "step": 36477
    },
    {
      "epoch": 3.478717803955078e-05,
      "step": 36477,
      "training_step_time": 11.185482025146484
    },
    {
      "epoch": 3.478813171386719e-05,
      "model_forward_time": 1.7502434253692627,
      "step": 36478
    },
    {
      "epoch": 3.478813171386719e-05,
      "step": 36478,
      "training_step_time": 7.5489280223846436
    },
    {
      "epoch": 3.478908538818359e-05,
      "model_forward_time": 1.7483422756195068,
      "step": 36479
    },
    {
      "epoch": 3.478908538818359e-05,
      "step": 36479,
      "training_step_time": 6.7800750732421875
    },
    {
      "epoch": 3.47900390625e-05,
      "grad_norm": 0.10688034445047379,
      "learning_rate": 7.5268245537748675e-06,
      "loss": 0.0581,
      "step": 36480
    },
    {
      "epoch": 3.47900390625e-05,
      "model_forward_time": 1.7390522956848145,
      "step": 36480
    },
    {
      "dynamics_loss_avg": 0.04787071514874697,
      "epoch": 3.47900390625e-05,
      "step": 36480
    },
    {
      "action_loss_avg": 0.001761350301967468,
      "epoch": 3.47900390625e-05,
      "step": 36480
    },
    {
      "epoch": 3.47900390625e-05,
      "step": 36480,
      "training_step_time": 6.776379823684692
    },
    {
      "epoch": 3.479099273681641e-05,
      "model_forward_time": 1.757775068283081,
      "step": 36481
    },
    {
      "epoch": 3.479099273681641e-05,
      "step": 36481,
      "training_step_time": 6.782538652420044
    },
    {
      "epoch": 3.479194641113281e-05,
      "model_forward_time": 1.7535960674285889,
      "step": 36482
    },
    {
      "epoch": 3.479194641113281e-05,
      "step": 36482,
      "training_step_time": 6.76314902305603
    },
    {
      "epoch": 3.4792900085449216e-05,
      "model_forward_time": 1.752600908279419,
      "step": 36483
    },
    {
      "epoch": 3.4792900085449216e-05,
      "step": 36483,
      "training_step_time": 7.631258726119995
    },
    {
      "epoch": 3.479385375976563e-05,
      "model_forward_time": 1.826662302017212,
      "step": 36484
    },
    {
      "epoch": 3.479385375976563e-05,
      "step": 36484,
      "training_step_time": 7.5554070472717285
    },
    {
      "epoch": 3.479480743408203e-05,
      "model_forward_time": 2.947150230407715,
      "step": 36485
    },
    {
      "epoch": 3.479480743408203e-05,
      "step": 36485,
      "training_step_time": 9.380699396133423
    },
    {
      "epoch": 3.4795761108398435e-05,
      "model_forward_time": 1.7524287700653076,
      "step": 36486
    },
    {
      "epoch": 3.4795761108398435e-05,
      "step": 36486,
      "training_step_time": 7.70508885383606
    },
    {
      "epoch": 3.4796714782714846e-05,
      "model_forward_time": 1.753568172454834,
      "step": 36487
    },
    {
      "epoch": 3.4796714782714846e-05,
      "step": 36487,
      "training_step_time": 7.57328462600708
    },
    {
      "epoch": 3.479766845703125e-05,
      "model_forward_time": 1.745452642440796,
      "step": 36488
    },
    {
      "epoch": 3.479766845703125e-05,
      "step": 36488,
      "training_step_time": 6.8707053661346436
    },
    {
      "epoch": 3.4798622131347654e-05,
      "model_forward_time": 1.7189545631408691,
      "step": 36489
    },
    {
      "epoch": 3.4798622131347654e-05,
      "step": 36489,
      "training_step_time": 6.904987573623657
    },
    {
      "epoch": 3.4799575805664064e-05,
      "grad_norm": 0.11654548346996307,
      "learning_rate": 7.525397627048179e-06,
      "loss": 0.0609,
      "step": 36490
    },
    {
      "epoch": 3.4799575805664064e-05,
      "model_forward_time": 1.7337474822998047,
      "step": 36490
    },
    {
      "dynamics_loss_avg": 0.061683606915175916,
      "epoch": 3.4799575805664064e-05,
      "step": 36490
    },
    {
      "action_loss_avg": 0.0014554160909028724,
      "epoch": 3.4799575805664064e-05,
      "step": 36490
    },
    {
      "epoch": 3.4799575805664064e-05,
      "step": 36490,
      "training_step_time": 6.8365912437438965
    },
    {
      "epoch": 3.480052947998047e-05,
      "model_forward_time": 1.7504777908325195,
      "step": 36491
    },
    {
      "epoch": 3.480052947998047e-05,
      "step": 36491,
      "training_step_time": 6.89173150062561
    },
    {
      "epoch": 3.480148315429687e-05,
      "model_forward_time": 1.7447757720947266,
      "step": 36492
    },
    {
      "epoch": 3.480148315429687e-05,
      "step": 36492,
      "training_step_time": 7.529607772827148
    },
    {
      "epoch": 3.480243682861328e-05,
      "model_forward_time": 1.8051583766937256,
      "step": 36493
    },
    {
      "epoch": 3.480243682861328e-05,
      "step": 36493,
      "training_step_time": 7.56735634803772
    },
    {
      "epoch": 3.480339050292969e-05,
      "model_forward_time": 2.9199717044830322,
      "step": 36494
    },
    {
      "epoch": 3.480339050292969e-05,
      "step": 36494,
      "training_step_time": 11.384538650512695
    },
    {
      "epoch": 3.480434417724609e-05,
      "model_forward_time": 1.739511251449585,
      "step": 36495
    },
    {
      "epoch": 3.480434417724609e-05,
      "step": 36495,
      "training_step_time": 9.713671445846558
    },
    {
      "epoch": 3.48052978515625e-05,
      "model_forward_time": 1.0387136936187744,
      "step": 36496
    },
    {
      "epoch": 3.48052978515625e-05,
      "step": 36496,
      "training_step_time": 8.717553853988647
    },
    {
      "epoch": 3.4806251525878906e-05,
      "model_forward_time": 1.7280807495117188,
      "step": 36497
    },
    {
      "epoch": 3.4806251525878906e-05,
      "step": 36497,
      "training_step_time": 7.5118348598480225
    },
    {
      "epoch": 3.480720520019531e-05,
      "model_forward_time": 1.745265007019043,
      "step": 36498
    },
    {
      "epoch": 3.480720520019531e-05,
      "step": 36498,
      "training_step_time": 6.719165802001953
    },
    {
      "epoch": 3.480815887451172e-05,
      "model_forward_time": 1.7463810443878174,
      "step": 36499
    },
    {
      "epoch": 3.480815887451172e-05,
      "step": 36499,
      "training_step_time": 6.727068662643433
    },
    {
      "epoch": 3.4809112548828125e-05,
      "grad_norm": 0.13404665887355804,
      "learning_rate": 7.523970424147802e-06,
      "loss": 0.0612,
      "step": 36500
    },
    {
      "epoch": 3.4809112548828125e-05,
      "model_forward_time": 1.7304904460906982,
      "step": 36500
    },
    {
      "dynamics_loss_avg": 0.05159959997981787,
      "epoch": 3.4809112548828125e-05,
      "step": 36500
    },
    {
      "action_loss_avg": 0.0016991722455713897,
      "epoch": 3.4809112548828125e-05,
      "step": 36500
    },
    {
      "epoch": 3.4809112548828125e-05,
      "step": 36500,
      "training_step_time": 6.800675392150879
    },
    {
      "epoch": 3.481006622314453e-05,
      "model_forward_time": 1.747192144393921,
      "step": 36501
    },
    {
      "epoch": 3.481006622314453e-05,
      "step": 36501,
      "training_step_time": 7.509095668792725
    },
    {
      "epoch": 3.481101989746094e-05,
      "model_forward_time": 1.8122830390930176,
      "step": 36502
    },
    {
      "epoch": 3.481101989746094e-05,
      "step": 36502,
      "training_step_time": 7.621798276901245
    },
    {
      "epoch": 3.4811973571777344e-05,
      "model_forward_time": 2.9345245361328125,
      "step": 36503
    },
    {
      "epoch": 3.4811973571777344e-05,
      "step": 36503,
      "training_step_time": 8.24630355834961
    },
    {
      "epoch": 3.481292724609375e-05,
      "model_forward_time": 1.7480065822601318,
      "step": 36504
    },
    {
      "epoch": 3.481292724609375e-05,
      "step": 36504,
      "training_step_time": 8.599130153656006
    },
    {
      "epoch": 3.481388092041016e-05,
      "model_forward_time": 1.7427334785461426,
      "step": 36505
    },
    {
      "epoch": 3.481388092041016e-05,
      "step": 36505,
      "training_step_time": 8.620266437530518
    },
    {
      "epoch": 3.481483459472656e-05,
      "model_forward_time": 1.7399580478668213,
      "step": 36506
    },
    {
      "epoch": 3.481483459472656e-05,
      "step": 36506,
      "training_step_time": 7.530580997467041
    },
    {
      "epoch": 3.481578826904297e-05,
      "model_forward_time": 1.7466423511505127,
      "step": 36507
    },
    {
      "epoch": 3.481578826904297e-05,
      "step": 36507,
      "training_step_time": 6.757086753845215
    },
    {
      "epoch": 3.481674194335938e-05,
      "model_forward_time": 1.747643232345581,
      "step": 36508
    },
    {
      "epoch": 3.481674194335938e-05,
      "step": 36508,
      "training_step_time": 6.798646450042725
    },
    {
      "epoch": 3.481769561767578e-05,
      "model_forward_time": 1.7298386096954346,
      "step": 36509
    },
    {
      "epoch": 3.481769561767578e-05,
      "step": 36509,
      "training_step_time": 6.874900579452515
    },
    {
      "epoch": 3.4818649291992186e-05,
      "grad_norm": 0.1406918168067932,
      "learning_rate": 7.5225429452298125e-06,
      "loss": 0.061,
      "step": 36510
    },
    {
      "epoch": 3.4818649291992186e-05,
      "model_forward_time": 1.736325740814209,
      "step": 36510
    },
    {
      "dynamics_loss_avg": 0.06102689430117607,
      "epoch": 3.4818649291992186e-05,
      "step": 36510
    },
    {
      "action_loss_avg": 0.0023501251125708222,
      "epoch": 3.4818649291992186e-05,
      "step": 36510
    },
    {
      "epoch": 3.4818649291992186e-05,
      "step": 36510,
      "training_step_time": 8.526078224182129
    },
    {
      "epoch": 3.4819602966308596e-05,
      "model_forward_time": 1.7965645790100098,
      "step": 36511
    },
    {
      "epoch": 3.4819602966308596e-05,
      "step": 36511,
      "training_step_time": 8.355057954788208
    },
    {
      "epoch": 3.4820556640625e-05,
      "model_forward_time": 2.86557674407959,
      "step": 36512
    },
    {
      "epoch": 3.4820556640625e-05,
      "step": 36512,
      "training_step_time": 8.548649072647095
    },
    {
      "epoch": 3.4821510314941404e-05,
      "model_forward_time": 1.7410542964935303,
      "step": 36513
    },
    {
      "epoch": 3.4821510314941404e-05,
      "step": 36513,
      "training_step_time": 9.930819511413574
    },
    {
      "epoch": 3.4822463989257815e-05,
      "model_forward_time": 1.7421019077301025,
      "step": 36514
    },
    {
      "epoch": 3.4822463989257815e-05,
      "step": 36514,
      "training_step_time": 10.039950132369995
    },
    {
      "epoch": 3.482341766357422e-05,
      "model_forward_time": 1.7500054836273193,
      "step": 36515
    },
    {
      "epoch": 3.482341766357422e-05,
      "step": 36515,
      "training_step_time": 7.557471513748169
    },
    {
      "epoch": 3.482437133789062e-05,
      "model_forward_time": 1.747756004333496,
      "step": 36516
    },
    {
      "epoch": 3.482437133789062e-05,
      "step": 36516,
      "training_step_time": 7.16113543510437
    },
    {
      "epoch": 3.4825325012207034e-05,
      "model_forward_time": 1.7343511581420898,
      "step": 36517
    },
    {
      "epoch": 3.4825325012207034e-05,
      "step": 36517,
      "training_step_time": 6.817526578903198
    },
    {
      "epoch": 3.482627868652344e-05,
      "model_forward_time": 1.7520780563354492,
      "step": 36518
    },
    {
      "epoch": 3.482627868652344e-05,
      "step": 36518,
      "training_step_time": 6.678725481033325
    },
    {
      "epoch": 3.482723236083984e-05,
      "model_forward_time": 1.741487979888916,
      "step": 36519
    },
    {
      "epoch": 3.482723236083984e-05,
      "step": 36519,
      "training_step_time": 7.462874174118042
    },
    {
      "epoch": 3.482818603515625e-05,
      "grad_norm": 0.15346407890319824,
      "learning_rate": 7.521115190450321e-06,
      "loss": 0.0585,
      "step": 36520
    },
    {
      "epoch": 3.482818603515625e-05,
      "model_forward_time": 1.7608864307403564,
      "step": 36520
    },
    {
      "dynamics_loss_avg": 0.05776120685040951,
      "epoch": 3.482818603515625e-05,
      "step": 36520
    },
    {
      "action_loss_avg": 0.001345025299815461,
      "epoch": 3.482818603515625e-05,
      "step": 36520
    },
    {
      "epoch": 3.482818603515625e-05,
      "step": 36520,
      "training_step_time": 7.548208951950073
    },
    {
      "epoch": 3.482913970947266e-05,
      "model_forward_time": 2.8647871017456055,
      "step": 36521
    },
    {
      "epoch": 3.482913970947266e-05,
      "step": 36521,
      "training_step_time": 8.60893177986145
    },
    {
      "epoch": 3.483009338378906e-05,
      "model_forward_time": 1.7583081722259521,
      "step": 36522
    },
    {
      "epoch": 3.483009338378906e-05,
      "step": 36522,
      "training_step_time": 10.301784038543701
    },
    {
      "epoch": 3.483104705810547e-05,
      "model_forward_time": 1.7406630516052246,
      "step": 36523
    },
    {
      "epoch": 3.483104705810547e-05,
      "step": 36523,
      "training_step_time": 8.35702633857727
    },
    {
      "epoch": 3.4832000732421876e-05,
      "model_forward_time": 1.7730510234832764,
      "step": 36524
    },
    {
      "epoch": 3.4832000732421876e-05,
      "step": 36524,
      "training_step_time": 7.609049081802368
    },
    {
      "epoch": 3.483295440673828e-05,
      "model_forward_time": 1.738553762435913,
      "step": 36525
    },
    {
      "epoch": 3.483295440673828e-05,
      "step": 36525,
      "training_step_time": 7.60313081741333
    },
    {
      "epoch": 3.483390808105469e-05,
      "model_forward_time": 1.7389373779296875,
      "step": 36526
    },
    {
      "epoch": 3.483390808105469e-05,
      "step": 36526,
      "training_step_time": 6.82616662979126
    },
    {
      "epoch": 3.4834861755371095e-05,
      "model_forward_time": 1.7540709972381592,
      "step": 36527
    },
    {
      "epoch": 3.4834861755371095e-05,
      "step": 36527,
      "training_step_time": 6.843871831893921
    },
    {
      "epoch": 3.48358154296875e-05,
      "model_forward_time": 1.7447919845581055,
      "step": 36528
    },
    {
      "epoch": 3.48358154296875e-05,
      "step": 36528,
      "training_step_time": 7.479277610778809
    },
    {
      "epoch": 3.483676910400391e-05,
      "model_forward_time": 1.7805562019348145,
      "step": 36529
    },
    {
      "epoch": 3.483676910400391e-05,
      "step": 36529,
      "training_step_time": 7.591667890548706
    },
    {
      "epoch": 3.4837722778320314e-05,
      "grad_norm": 0.08637559413909912,
      "learning_rate": 7.519687159965463e-06,
      "loss": 0.0609,
      "step": 36530
    },
    {
      "epoch": 3.4837722778320314e-05,
      "model_forward_time": 1.797912359237671,
      "step": 36530
    },
    {
      "dynamics_loss_avg": 0.059726111218333246,
      "epoch": 3.4837722778320314e-05,
      "step": 36530
    },
    {
      "action_loss_avg": 0.0009915878239553423,
      "epoch": 3.4837722778320314e-05,
      "step": 36530
    },
    {
      "epoch": 3.4837722778320314e-05,
      "step": 36530,
      "training_step_time": 8.64227294921875
    },
    {
      "epoch": 3.483867645263672e-05,
      "model_forward_time": 1.7441291809082031,
      "step": 36531
    },
    {
      "epoch": 3.483867645263672e-05,
      "step": 36531,
      "training_step_time": 9.461186408996582
    },
    {
      "epoch": 3.483963012695312e-05,
      "model_forward_time": 1.7467360496520996,
      "step": 36532
    },
    {
      "epoch": 3.483963012695312e-05,
      "step": 36532,
      "training_step_time": 7.8858232498168945
    },
    {
      "epoch": 3.484058380126953e-05,
      "model_forward_time": 1.736250877380371,
      "step": 36533
    },
    {
      "epoch": 3.484058380126953e-05,
      "step": 36533,
      "training_step_time": 7.860292673110962
    },
    {
      "epoch": 3.4841537475585936e-05,
      "model_forward_time": 1.7481999397277832,
      "step": 36534
    },
    {
      "epoch": 3.4841537475585936e-05,
      "step": 36534,
      "training_step_time": 7.739933013916016
    },
    {
      "epoch": 3.484249114990234e-05,
      "model_forward_time": 1.752694845199585,
      "step": 36535
    },
    {
      "epoch": 3.484249114990234e-05,
      "step": 36535,
      "training_step_time": 6.896556377410889
    },
    {
      "epoch": 3.484344482421875e-05,
      "model_forward_time": 1.7489750385284424,
      "step": 36536
    },
    {
      "epoch": 3.484344482421875e-05,
      "step": 36536,
      "training_step_time": 6.9181671142578125
    },
    {
      "epoch": 3.4844398498535155e-05,
      "model_forward_time": 1.7428059577941895,
      "step": 36537
    },
    {
      "epoch": 3.4844398498535155e-05,
      "step": 36537,
      "training_step_time": 7.582402944564819
    },
    {
      "epoch": 3.484535217285156e-05,
      "model_forward_time": 1.7409610748291016,
      "step": 36538
    },
    {
      "epoch": 3.484535217285156e-05,
      "step": 36538,
      "training_step_time": 7.825383186340332
    },
    {
      "epoch": 3.484630584716797e-05,
      "model_forward_time": 1.8142797946929932,
      "step": 36539
    },
    {
      "epoch": 3.484630584716797e-05,
      "step": 36539,
      "training_step_time": 8.60505986213684
    },
    {
      "epoch": 3.4847259521484374e-05,
      "grad_norm": 0.10042940080165863,
      "learning_rate": 7.5182588539314036e-06,
      "loss": 0.0594,
      "step": 36540
    },
    {
      "epoch": 3.4847259521484374e-05,
      "model_forward_time": 2.9378442764282227,
      "step": 36540
    },
    {
      "dynamics_loss_avg": 0.05506631135940552,
      "epoch": 3.4847259521484374e-05,
      "step": 36540
    },
    {
      "action_loss_avg": 0.0015162990253884344,
      "epoch": 3.4847259521484374e-05,
      "step": 36540
    },
    {
      "epoch": 3.4847259521484374e-05,
      "step": 36540,
      "training_step_time": 7.919072866439819
    },
    {
      "epoch": 3.484821319580078e-05,
      "model_forward_time": 1.741574764251709,
      "step": 36541
    },
    {
      "epoch": 3.484821319580078e-05,
      "step": 36541,
      "training_step_time": 9.514721393585205
    },
    {
      "epoch": 3.484916687011719e-05,
      "model_forward_time": 1.7459216117858887,
      "step": 36542
    },
    {
      "epoch": 3.484916687011719e-05,
      "step": 36542,
      "training_step_time": 8.307231426239014
    },
    {
      "epoch": 3.485012054443359e-05,
      "model_forward_time": 1.7460682392120361,
      "step": 36543
    },
    {
      "epoch": 3.485012054443359e-05,
      "step": 36543,
      "training_step_time": 7.530599117279053
    },
    {
      "epoch": 3.485107421875e-05,
      "model_forward_time": 1.7492895126342773,
      "step": 36544
    },
    {
      "epoch": 3.485107421875e-05,
      "step": 36544,
      "training_step_time": 6.779531478881836
    },
    {
      "epoch": 3.485202789306641e-05,
      "model_forward_time": 1.7398319244384766,
      "step": 36545
    },
    {
      "epoch": 3.485202789306641e-05,
      "step": 36545,
      "training_step_time": 6.7008748054504395
    },
    {
      "epoch": 3.485298156738281e-05,
      "model_forward_time": 1.7392072677612305,
      "step": 36546
    },
    {
      "epoch": 3.485298156738281e-05,
      "step": 36546,
      "training_step_time": 7.4443113803863525
    },
    {
      "epoch": 3.4853935241699216e-05,
      "model_forward_time": 1.7295382022857666,
      "step": 36547
    },
    {
      "epoch": 3.4853935241699216e-05,
      "step": 36547,
      "training_step_time": 7.491034507751465
    },
    {
      "epoch": 3.485488891601563e-05,
      "model_forward_time": 1.8021135330200195,
      "step": 36548
    },
    {
      "epoch": 3.485488891601563e-05,
      "step": 36548,
      "training_step_time": 8.379033088684082
    },
    {
      "epoch": 3.485584259033203e-05,
      "model_forward_time": 2.5642759799957275,
      "step": 36549
    },
    {
      "epoch": 3.485584259033203e-05,
      "step": 36549,
      "training_step_time": 8.649993896484375
    },
    {
      "epoch": 3.4856796264648435e-05,
      "grad_norm": 0.13550950586795807,
      "learning_rate": 7.516830272504343e-06,
      "loss": 0.0608,
      "step": 36550
    },
    {
      "epoch": 3.4856796264648435e-05,
      "model_forward_time": 1.750931739807129,
      "step": 36550
    },
    {
      "dynamics_loss_avg": 0.05422800499945879,
      "epoch": 3.4856796264648435e-05,
      "step": 36550
    },
    {
      "action_loss_avg": 0.001540741414646618,
      "epoch": 3.4856796264648435e-05,
      "step": 36550
    },
    {
      "epoch": 3.4856796264648435e-05,
      "step": 36550,
      "training_step_time": 7.683266878128052
    },
    {
      "epoch": 3.4857749938964846e-05,
      "model_forward_time": 1.7382922172546387,
      "step": 36551
    },
    {
      "epoch": 3.4857749938964846e-05,
      "step": 36551,
      "training_step_time": 7.722631454467773
    },
    {
      "epoch": 3.485870361328125e-05,
      "model_forward_time": 1.7386837005615234,
      "step": 36552
    },
    {
      "epoch": 3.485870361328125e-05,
      "step": 36552,
      "training_step_time": 8.01267123222351
    },
    {
      "epoch": 3.4859657287597654e-05,
      "model_forward_time": 1.7383010387420654,
      "step": 36553
    },
    {
      "epoch": 3.4859657287597654e-05,
      "step": 36553,
      "training_step_time": 6.888883113861084
    },
    {
      "epoch": 3.4860610961914064e-05,
      "model_forward_time": 1.7425777912139893,
      "step": 36554
    },
    {
      "epoch": 3.4860610961914064e-05,
      "step": 36554,
      "training_step_time": 6.777205944061279
    },
    {
      "epoch": 3.486156463623047e-05,
      "model_forward_time": 1.7382454872131348,
      "step": 36555
    },
    {
      "epoch": 3.486156463623047e-05,
      "step": 36555,
      "training_step_time": 7.489706993103027
    },
    {
      "epoch": 3.486251831054687e-05,
      "model_forward_time": 1.7339539527893066,
      "step": 36556
    },
    {
      "epoch": 3.486251831054687e-05,
      "step": 36556,
      "training_step_time": 7.663968086242676
    },
    {
      "epoch": 3.486347198486328e-05,
      "model_forward_time": 1.812185287475586,
      "step": 36557
    },
    {
      "epoch": 3.486347198486328e-05,
      "step": 36557,
      "training_step_time": 7.482479572296143
    },
    {
      "epoch": 3.486442565917969e-05,
      "model_forward_time": 2.8505239486694336,
      "step": 36558
    },
    {
      "epoch": 3.486442565917969e-05,
      "step": 36558,
      "training_step_time": 8.276105880737305
    },
    {
      "epoch": 3.486537933349609e-05,
      "model_forward_time": 1.7372090816497803,
      "step": 36559
    },
    {
      "epoch": 3.486537933349609e-05,
      "step": 36559,
      "training_step_time": 8.708955526351929
    },
    {
      "epoch": 3.48663330078125e-05,
      "grad_norm": 0.09479620307683945,
      "learning_rate": 7.515401415840509e-06,
      "loss": 0.0624,
      "step": 36560
    },
    {
      "epoch": 3.48663330078125e-05,
      "model_forward_time": 1.743755578994751,
      "step": 36560
    },
    {
      "dynamics_loss_avg": 0.05437421062961221,
      "epoch": 3.48663330078125e-05,
      "step": 36560
    },
    {
      "action_loss_avg": 0.0015148499922361224,
      "epoch": 3.48663330078125e-05,
      "step": 36560
    },
    {
      "epoch": 3.48663330078125e-05,
      "step": 36560,
      "training_step_time": 9.32201099395752
    },
    {
      "epoch": 3.4867286682128906e-05,
      "model_forward_time": 1.7400832176208496,
      "step": 36561
    },
    {
      "epoch": 3.4867286682128906e-05,
      "step": 36561,
      "training_step_time": 8.617826700210571
    },
    {
      "epoch": 3.486824035644531e-05,
      "model_forward_time": 1.75236177444458,
      "step": 36562
    },
    {
      "epoch": 3.486824035644531e-05,
      "step": 36562,
      "training_step_time": 6.811352491378784
    },
    {
      "epoch": 3.486919403076172e-05,
      "model_forward_time": 1.7342064380645752,
      "step": 36563
    },
    {
      "epoch": 3.486919403076172e-05,
      "step": 36563,
      "training_step_time": 6.6649839878082275
    },
    {
      "epoch": 3.4870147705078125e-05,
      "model_forward_time": 1.7703912258148193,
      "step": 36564
    },
    {
      "epoch": 3.4870147705078125e-05,
      "step": 36564,
      "training_step_time": 7.475759029388428
    },
    {
      "epoch": 3.487110137939453e-05,
      "model_forward_time": 1.7347774505615234,
      "step": 36565
    },
    {
      "epoch": 3.487110137939453e-05,
      "step": 36565,
      "training_step_time": 7.438395023345947
    },
    {
      "epoch": 3.487205505371094e-05,
      "model_forward_time": 1.8003692626953125,
      "step": 36566
    },
    {
      "epoch": 3.487205505371094e-05,
      "step": 36566,
      "training_step_time": 7.528361558914185
    },
    {
      "epoch": 3.4873008728027344e-05,
      "model_forward_time": 2.8248400688171387,
      "step": 36567
    },
    {
      "epoch": 3.4873008728027344e-05,
      "step": 36567,
      "training_step_time": 8.30303430557251
    },
    {
      "epoch": 3.487396240234375e-05,
      "model_forward_time": 1.7394838333129883,
      "step": 36568
    },
    {
      "epoch": 3.487396240234375e-05,
      "step": 36568,
      "training_step_time": 9.36362624168396
    },
    {
      "epoch": 3.487491607666016e-05,
      "model_forward_time": 1.7413771152496338,
      "step": 36569
    },
    {
      "epoch": 3.487491607666016e-05,
      "step": 36569,
      "training_step_time": 8.800203323364258
    },
    {
      "epoch": 3.487586975097656e-05,
      "grad_norm": 0.11463235318660736,
      "learning_rate": 7.513972284096156e-06,
      "loss": 0.0613,
      "step": 36570
    },
    {
      "epoch": 3.487586975097656e-05,
      "model_forward_time": 1.7492308616638184,
      "step": 36570
    },
    {
      "dynamics_loss_avg": 0.07998996954411268,
      "epoch": 3.487586975097656e-05,
      "step": 36570
    },
    {
      "action_loss_avg": 0.002561907387280371,
      "epoch": 3.487586975097656e-05,
      "step": 36570
    },
    {
      "epoch": 3.487586975097656e-05,
      "step": 36570,
      "training_step_time": 7.910457134246826
    },
    {
      "epoch": 3.487682342529297e-05,
      "model_forward_time": 1.7368831634521484,
      "step": 36571
    },
    {
      "epoch": 3.487682342529297e-05,
      "step": 36571,
      "training_step_time": 7.662973880767822
    },
    {
      "epoch": 3.487777709960938e-05,
      "model_forward_time": 1.7281320095062256,
      "step": 36572
    },
    {
      "epoch": 3.487777709960938e-05,
      "step": 36572,
      "training_step_time": 6.904526710510254
    },
    {
      "epoch": 3.487873077392578e-05,
      "model_forward_time": 1.7424230575561523,
      "step": 36573
    },
    {
      "epoch": 3.487873077392578e-05,
      "step": 36573,
      "training_step_time": 7.610130548477173
    },
    {
      "epoch": 3.4879684448242186e-05,
      "model_forward_time": 1.7351999282836914,
      "step": 36574
    },
    {
      "epoch": 3.4879684448242186e-05,
      "step": 36574,
      "training_step_time": 7.314836502075195
    },
    {
      "epoch": 3.4880638122558596e-05,
      "model_forward_time": 1.8018324375152588,
      "step": 36575
    },
    {
      "epoch": 3.4880638122558596e-05,
      "step": 36575,
      "training_step_time": 7.780931234359741
    },
    {
      "epoch": 3.4881591796875e-05,
      "model_forward_time": 1.7253224849700928,
      "step": 36576
    },
    {
      "epoch": 3.4881591796875e-05,
      "step": 36576,
      "training_step_time": 8.500184297561646
    },
    {
      "epoch": 3.4882545471191405e-05,
      "model_forward_time": 2.8751423358917236,
      "step": 36577
    },
    {
      "epoch": 3.4882545471191405e-05,
      "step": 36577,
      "training_step_time": 9.979640483856201
    },
    {
      "epoch": 3.4883499145507815e-05,
      "model_forward_time": 1.739591121673584,
      "step": 36578
    },
    {
      "epoch": 3.4883499145507815e-05,
      "step": 36578,
      "training_step_time": 7.736574411392212
    },
    {
      "epoch": 3.488445281982422e-05,
      "model_forward_time": 1.7515747547149658,
      "step": 36579
    },
    {
      "epoch": 3.488445281982422e-05,
      "step": 36579,
      "training_step_time": 8.24931287765503
    },
    {
      "epoch": 3.488540649414062e-05,
      "grad_norm": 0.12831871211528778,
      "learning_rate": 7.512542877427575e-06,
      "loss": 0.061,
      "step": 36580
    },
    {
      "epoch": 3.488540649414062e-05,
      "model_forward_time": 1.7292506694793701,
      "step": 36580
    },
    {
      "dynamics_loss_avg": 0.055151262879371644,
      "epoch": 3.488540649414062e-05,
      "step": 36580
    },
    {
      "action_loss_avg": 0.0025575485662557183,
      "epoch": 3.488540649414062e-05,
      "step": 36580
    },
    {
      "epoch": 3.488540649414062e-05,
      "step": 36580,
      "training_step_time": 7.570920467376709
    },
    {
      "epoch": 3.4886360168457034e-05,
      "model_forward_time": 1.7433784008026123,
      "step": 36581
    },
    {
      "epoch": 3.4886360168457034e-05,
      "step": 36581,
      "training_step_time": 6.758652210235596
    },
    {
      "epoch": 3.488731384277344e-05,
      "model_forward_time": 1.7521357536315918,
      "step": 36582
    },
    {
      "epoch": 3.488731384277344e-05,
      "step": 36582,
      "training_step_time": 7.4632322788238525
    },
    {
      "epoch": 3.488826751708984e-05,
      "model_forward_time": 1.7326068878173828,
      "step": 36583
    },
    {
      "epoch": 3.488826751708984e-05,
      "step": 36583,
      "training_step_time": 7.187052488327026
    },
    {
      "epoch": 3.488922119140625e-05,
      "model_forward_time": 1.7970740795135498,
      "step": 36584
    },
    {
      "epoch": 3.488922119140625e-05,
      "step": 36584,
      "training_step_time": 7.633019208908081
    },
    {
      "epoch": 3.489017486572266e-05,
      "model_forward_time": 1.7529964447021484,
      "step": 36585
    },
    {
      "epoch": 3.489017486572266e-05,
      "step": 36585,
      "training_step_time": 8.333592653274536
    },
    {
      "epoch": 3.489112854003906e-05,
      "model_forward_time": 2.9418654441833496,
      "step": 36586
    },
    {
      "epoch": 3.489112854003906e-05,
      "step": 36586,
      "training_step_time": 7.696020603179932
    },
    {
      "epoch": 3.489208221435547e-05,
      "model_forward_time": 1.7417280673980713,
      "step": 36587
    },
    {
      "epoch": 3.489208221435547e-05,
      "step": 36587,
      "training_step_time": 10.302088022232056
    },
    {
      "epoch": 3.4893035888671876e-05,
      "model_forward_time": 1.7757482528686523,
      "step": 36588
    },
    {
      "epoch": 3.4893035888671876e-05,
      "step": 36588,
      "training_step_time": 9.03506875038147
    },
    {
      "epoch": 3.489398956298828e-05,
      "model_forward_time": 1.7484779357910156,
      "step": 36589
    },
    {
      "epoch": 3.489398956298828e-05,
      "step": 36589,
      "training_step_time": 7.524662733078003
    },
    {
      "epoch": 3.489494323730469e-05,
      "grad_norm": 0.11852293461561203,
      "learning_rate": 7.511113195991081e-06,
      "loss": 0.059,
      "step": 36590
    },
    {
      "epoch": 3.489494323730469e-05,
      "model_forward_time": 1.726020336151123,
      "step": 36590
    },
    {
      "dynamics_loss_avg": 0.049995347298681735,
      "epoch": 3.489494323730469e-05,
      "step": 36590
    },
    {
      "action_loss_avg": 0.0018152037839172408,
      "epoch": 3.489494323730469e-05,
      "step": 36590
    },
    {
      "epoch": 3.489494323730469e-05,
      "step": 36590,
      "training_step_time": 6.759137392044067
    },
    {
      "epoch": 3.4895896911621095e-05,
      "model_forward_time": 1.7527704238891602,
      "step": 36591
    },
    {
      "epoch": 3.4895896911621095e-05,
      "step": 36591,
      "training_step_time": 7.516427040100098
    },
    {
      "epoch": 3.48968505859375e-05,
      "model_forward_time": 1.7443573474884033,
      "step": 36592
    },
    {
      "epoch": 3.48968505859375e-05,
      "step": 36592,
      "training_step_time": 7.243642568588257
    },
    {
      "epoch": 3.489780426025391e-05,
      "model_forward_time": 1.7591502666473389,
      "step": 36593
    },
    {
      "epoch": 3.489780426025391e-05,
      "step": 36593,
      "training_step_time": 7.652632713317871
    },
    {
      "epoch": 3.4898757934570314e-05,
      "model_forward_time": 1.81168532371521,
      "step": 36594
    },
    {
      "epoch": 3.4898757934570314e-05,
      "step": 36594,
      "training_step_time": 8.427086591720581
    },
    {
      "epoch": 3.489971160888672e-05,
      "model_forward_time": 2.9188921451568604,
      "step": 36595
    },
    {
      "epoch": 3.489971160888672e-05,
      "step": 36595,
      "training_step_time": 8.447964668273926
    },
    {
      "epoch": 3.490066528320312e-05,
      "model_forward_time": 1.7515854835510254,
      "step": 36596
    },
    {
      "epoch": 3.490066528320312e-05,
      "step": 36596,
      "training_step_time": 9.264469146728516
    },
    {
      "epoch": 3.490161895751953e-05,
      "model_forward_time": 1.7485816478729248,
      "step": 36597
    },
    {
      "epoch": 3.490161895751953e-05,
      "step": 36597,
      "training_step_time": 8.906935453414917
    },
    {
      "epoch": 3.4902572631835937e-05,
      "model_forward_time": 1.7485930919647217,
      "step": 36598
    },
    {
      "epoch": 3.4902572631835937e-05,
      "step": 36598,
      "training_step_time": 7.623365640640259
    },
    {
      "epoch": 3.490352630615234e-05,
      "model_forward_time": 1.7496106624603271,
      "step": 36599
    },
    {
      "epoch": 3.490352630615234e-05,
      "step": 36599,
      "training_step_time": 7.292086362838745
    },
    {
      "epoch": 3.490447998046875e-05,
      "grad_norm": 0.11911611258983612,
      "learning_rate": 7.509683239943024e-06,
      "loss": 0.0589,
      "step": 36600
    },
    {
      "epoch": 3.490447998046875e-05,
      "model_forward_time": 1.7839276790618896,
      "step": 36600
    },
    {
      "dynamics_loss_avg": 0.0653601847589016,
      "epoch": 3.490447998046875e-05,
      "step": 36600
    },
    {
      "action_loss_avg": 0.002882312471047044,
      "epoch": 3.490447998046875e-05,
      "step": 36600
    },
    {
      "epoch": 3.490447998046875e-05,
      "step": 36600,
      "training_step_time": 7.5383384227752686
    },
    {
      "epoch": 3.4905433654785155e-05,
      "model_forward_time": 1.7499797344207764,
      "step": 36601
    },
    {
      "epoch": 3.4905433654785155e-05,
      "step": 36601,
      "training_step_time": 8.110492944717407
    },
    {
      "epoch": 3.490638732910156e-05,
      "model_forward_time": 1.7463340759277344,
      "step": 36602
    },
    {
      "epoch": 3.490638732910156e-05,
      "step": 36602,
      "training_step_time": 8.463898420333862
    },
    {
      "epoch": 3.490734100341797e-05,
      "model_forward_time": 1.8204994201660156,
      "step": 36603
    },
    {
      "epoch": 3.490734100341797e-05,
      "step": 36603,
      "training_step_time": 8.471422672271729
    },
    {
      "epoch": 3.4908294677734374e-05,
      "model_forward_time": 2.861219644546509,
      "step": 36604
    },
    {
      "epoch": 3.4908294677734374e-05,
      "step": 36604,
      "training_step_time": 7.61164665222168
    },
    {
      "epoch": 3.490924835205078e-05,
      "model_forward_time": 1.7510700225830078,
      "step": 36605
    },
    {
      "epoch": 3.490924835205078e-05,
      "step": 36605,
      "training_step_time": 11.257352828979492
    },
    {
      "epoch": 3.491020202636719e-05,
      "model_forward_time": 1.7524559497833252,
      "step": 36606
    },
    {
      "epoch": 3.491020202636719e-05,
      "step": 36606,
      "training_step_time": 7.698410511016846
    },
    {
      "epoch": 3.491115570068359e-05,
      "model_forward_time": 1.7621746063232422,
      "step": 36607
    },
    {
      "epoch": 3.491115570068359e-05,
      "step": 36607,
      "training_step_time": 7.702310800552368
    },
    {
      "epoch": 3.4912109375e-05,
      "model_forward_time": 1.745624303817749,
      "step": 36608
    },
    {
      "epoch": 3.4912109375e-05,
      "step": 36608,
      "training_step_time": 7.49775505065918
    },
    {
      "epoch": 3.491306304931641e-05,
      "model_forward_time": 1.7524428367614746,
      "step": 36609
    },
    {
      "epoch": 3.491306304931641e-05,
      "step": 36609,
      "training_step_time": 7.464548587799072
    },
    {
      "epoch": 3.491401672363281e-05,
      "grad_norm": 0.22503189742565155,
      "learning_rate": 7.508253009439781e-06,
      "loss": 0.0573,
      "step": 36610
    },
    {
      "epoch": 3.491401672363281e-05,
      "model_forward_time": 1.7564294338226318,
      "step": 36610
    },
    {
      "dynamics_loss_avg": 0.06384049244225025,
      "epoch": 3.491401672363281e-05,
      "step": 36610
    },
    {
      "action_loss_avg": 0.0017047237837687136,
      "epoch": 3.491401672363281e-05,
      "step": 36610
    },
    {
      "epoch": 3.491401672363281e-05,
      "step": 36610,
      "training_step_time": 7.893566846847534
    },
    {
      "epoch": 3.4914970397949216e-05,
      "model_forward_time": 1.755843162536621,
      "step": 36611
    },
    {
      "epoch": 3.4914970397949216e-05,
      "step": 36611,
      "training_step_time": 7.278186321258545
    },
    {
      "epoch": 3.491592407226563e-05,
      "model_forward_time": 1.818974256515503,
      "step": 36612
    },
    {
      "epoch": 3.491592407226563e-05,
      "step": 36612,
      "training_step_time": 8.328214168548584
    },
    {
      "epoch": 3.491687774658203e-05,
      "model_forward_time": 2.8414599895477295,
      "step": 36613
    },
    {
      "epoch": 3.491687774658203e-05,
      "step": 36613,
      "training_step_time": 9.65009880065918
    },
    {
      "epoch": 3.4917831420898435e-05,
      "model_forward_time": 1.749143123626709,
      "step": 36614
    },
    {
      "epoch": 3.4917831420898435e-05,
      "step": 36614,
      "training_step_time": 7.685708284378052
    },
    {
      "epoch": 3.4918785095214846e-05,
      "model_forward_time": 1.7500762939453125,
      "step": 36615
    },
    {
      "epoch": 3.4918785095214846e-05,
      "step": 36615,
      "training_step_time": 8.215283632278442
    },
    {
      "epoch": 3.491973876953125e-05,
      "model_forward_time": 1.740211009979248,
      "step": 36616
    },
    {
      "epoch": 3.491973876953125e-05,
      "step": 36616,
      "training_step_time": 8.654531002044678
    },
    {
      "epoch": 3.4920692443847654e-05,
      "model_forward_time": 1.7509562969207764,
      "step": 36617
    },
    {
      "epoch": 3.4920692443847654e-05,
      "step": 36617,
      "training_step_time": 7.59410834312439
    },
    {
      "epoch": 3.4921646118164065e-05,
      "model_forward_time": 1.745370864868164,
      "step": 36618
    },
    {
      "epoch": 3.4921646118164065e-05,
      "step": 36618,
      "training_step_time": 7.4904890060424805
    },
    {
      "epoch": 3.492259979248047e-05,
      "model_forward_time": 1.7448251247406006,
      "step": 36619
    },
    {
      "epoch": 3.492259979248047e-05,
      "step": 36619,
      "training_step_time": 6.828221559524536
    },
    {
      "epoch": 3.492355346679687e-05,
      "grad_norm": 0.13074351847171783,
      "learning_rate": 7.506822504637761e-06,
      "loss": 0.0569,
      "step": 36620
    },
    {
      "epoch": 3.492355346679687e-05,
      "model_forward_time": 1.7470345497131348,
      "step": 36620
    },
    {
      "dynamics_loss_avg": 0.059712971281260255,
      "epoch": 3.492355346679687e-05,
      "step": 36620
    },
    {
      "action_loss_avg": 0.0022181630891282113,
      "epoch": 3.492355346679687e-05,
      "step": 36620
    },
    {
      "epoch": 3.492355346679687e-05,
      "step": 36620,
      "training_step_time": 7.25847864151001
    },
    {
      "epoch": 3.492450714111328e-05,
      "model_forward_time": 1.7740545272827148,
      "step": 36621
    },
    {
      "epoch": 3.492450714111328e-05,
      "step": 36621,
      "training_step_time": 7.687014579772949
    },
    {
      "epoch": 3.492546081542969e-05,
      "model_forward_time": 1.7489280700683594,
      "step": 36622
    },
    {
      "epoch": 3.492546081542969e-05,
      "step": 36622,
      "training_step_time": 9.733146667480469
    },
    {
      "epoch": 3.492641448974609e-05,
      "model_forward_time": 1.7511651515960693,
      "step": 36623
    },
    {
      "epoch": 3.492641448974609e-05,
      "step": 36623,
      "training_step_time": 7.745793104171753
    },
    {
      "epoch": 3.49273681640625e-05,
      "model_forward_time": 1.7701506614685059,
      "step": 36624
    },
    {
      "epoch": 3.49273681640625e-05,
      "step": 36624,
      "training_step_time": 7.844902753829956
    },
    {
      "epoch": 3.4928321838378906e-05,
      "model_forward_time": 1.750218391418457,
      "step": 36625
    },
    {
      "epoch": 3.4928321838378906e-05,
      "step": 36625,
      "training_step_time": 8.658392429351807
    },
    {
      "epoch": 3.492927551269531e-05,
      "model_forward_time": 1.7509605884552002,
      "step": 36626
    },
    {
      "epoch": 3.492927551269531e-05,
      "step": 36626,
      "training_step_time": 7.5273284912109375
    },
    {
      "epoch": 3.493022918701172e-05,
      "model_forward_time": 1.7481579780578613,
      "step": 36627
    },
    {
      "epoch": 3.493022918701172e-05,
      "step": 36627,
      "training_step_time": 7.340858459472656
    },
    {
      "epoch": 3.4931182861328125e-05,
      "model_forward_time": 1.7565786838531494,
      "step": 36628
    },
    {
      "epoch": 3.4931182861328125e-05,
      "step": 36628,
      "training_step_time": 7.474008798599243
    },
    {
      "epoch": 3.493213653564453e-05,
      "model_forward_time": 1.7191903591156006,
      "step": 36629
    },
    {
      "epoch": 3.493213653564453e-05,
      "step": 36629,
      "training_step_time": 7.219780206680298
    },
    {
      "epoch": 3.493309020996094e-05,
      "grad_norm": 0.13682889938354492,
      "learning_rate": 7.5053917256934e-06,
      "loss": 0.0598,
      "step": 36630
    },
    {
      "epoch": 3.493309020996094e-05,
      "model_forward_time": 1.7206389904022217,
      "step": 36630
    },
    {
      "dynamics_loss_avg": 0.05931776594370604,
      "epoch": 3.493309020996094e-05,
      "step": 36630
    },
    {
      "action_loss_avg": 0.001363744301488623,
      "epoch": 3.493309020996094e-05,
      "step": 36630
    },
    {
      "epoch": 3.493309020996094e-05,
      "step": 36630,
      "training_step_time": 8.671738624572754
    },
    {
      "epoch": 3.4934043884277344e-05,
      "model_forward_time": 1.8245329856872559,
      "step": 36631
    },
    {
      "epoch": 3.4934043884277344e-05,
      "step": 36631,
      "training_step_time": 7.5795910358428955
    },
    {
      "epoch": 3.493499755859375e-05,
      "model_forward_time": 2.942410707473755,
      "step": 36632
    },
    {
      "epoch": 3.493499755859375e-05,
      "step": 36632,
      "training_step_time": 8.24244737625122
    },
    {
      "epoch": 3.493595123291016e-05,
      "model_forward_time": 1.7435762882232666,
      "step": 36633
    },
    {
      "epoch": 3.493595123291016e-05,
      "step": 36633,
      "training_step_time": 8.281480550765991
    },
    {
      "epoch": 3.493690490722656e-05,
      "model_forward_time": 1.7486536502838135,
      "step": 36634
    },
    {
      "epoch": 3.493690490722656e-05,
      "step": 36634,
      "training_step_time": 7.920840263366699
    },
    {
      "epoch": 3.493785858154297e-05,
      "model_forward_time": 1.7545702457427979,
      "step": 36635
    },
    {
      "epoch": 3.493785858154297e-05,
      "step": 36635,
      "training_step_time": 7.824751615524292
    },
    {
      "epoch": 3.493881225585938e-05,
      "model_forward_time": 1.7454485893249512,
      "step": 36636
    },
    {
      "epoch": 3.493881225585938e-05,
      "step": 36636,
      "training_step_time": 7.550073146820068
    },
    {
      "epoch": 3.493976593017578e-05,
      "model_forward_time": 1.7900402545928955,
      "step": 36637
    },
    {
      "epoch": 3.493976593017578e-05,
      "step": 36637,
      "training_step_time": 7.630163669586182
    },
    {
      "epoch": 3.4940719604492186e-05,
      "model_forward_time": 1.7440218925476074,
      "step": 36638
    },
    {
      "epoch": 3.4940719604492186e-05,
      "step": 36638,
      "training_step_time": 7.391442537307739
    },
    {
      "epoch": 3.4941673278808597e-05,
      "model_forward_time": 1.746344804763794,
      "step": 36639
    },
    {
      "epoch": 3.4941673278808597e-05,
      "step": 36639,
      "training_step_time": 7.87422513961792
    },
    {
      "epoch": 3.4942626953125e-05,
      "grad_norm": 0.1534978747367859,
      "learning_rate": 7.503960672763167e-06,
      "loss": 0.0595,
      "step": 36640
    },
    {
      "epoch": 3.4942626953125e-05,
      "model_forward_time": 1.8131301403045654,
      "step": 36640
    },
    {
      "dynamics_loss_avg": 0.07056812793016434,
      "epoch": 3.4942626953125e-05,
      "step": 36640
    },
    {
      "action_loss_avg": 0.002756237256107852,
      "epoch": 3.4942626953125e-05,
      "step": 36640
    },
    {
      "epoch": 3.4942626953125e-05,
      "step": 36640,
      "training_step_time": 8.765840768814087
    },
    {
      "epoch": 3.4943580627441405e-05,
      "model_forward_time": 2.9317128658294678,
      "step": 36641
    },
    {
      "epoch": 3.4943580627441405e-05,
      "step": 36641,
      "training_step_time": 8.964174747467041
    },
    {
      "epoch": 3.4944534301757815e-05,
      "model_forward_time": 1.74784517288208,
      "step": 36642
    },
    {
      "epoch": 3.4944534301757815e-05,
      "step": 36642,
      "training_step_time": 8.455198764801025
    },
    {
      "epoch": 3.494548797607422e-05,
      "model_forward_time": 1.7440483570098877,
      "step": 36643
    },
    {
      "epoch": 3.494548797607422e-05,
      "step": 36643,
      "training_step_time": 8.745057106018066
    },
    {
      "epoch": 3.4946441650390623e-05,
      "model_forward_time": 1.7539036273956299,
      "step": 36644
    },
    {
      "epoch": 3.4946441650390623e-05,
      "step": 36644,
      "training_step_time": 7.658681631088257
    },
    {
      "epoch": 3.4947395324707034e-05,
      "model_forward_time": 1.7471942901611328,
      "step": 36645
    },
    {
      "epoch": 3.4947395324707034e-05,
      "step": 36645,
      "training_step_time": 7.5493996143341064
    },
    {
      "epoch": 3.494834899902344e-05,
      "model_forward_time": 1.744983196258545,
      "step": 36646
    },
    {
      "epoch": 3.494834899902344e-05,
      "step": 36646,
      "training_step_time": 7.433743000030518
    },
    {
      "epoch": 3.494930267333984e-05,
      "model_forward_time": 1.7432403564453125,
      "step": 36647
    },
    {
      "epoch": 3.494930267333984e-05,
      "step": 36647,
      "training_step_time": 7.178779602050781
    },
    {
      "epoch": 3.495025634765625e-05,
      "model_forward_time": 1.7520341873168945,
      "step": 36648
    },
    {
      "epoch": 3.495025634765625e-05,
      "step": 36648,
      "training_step_time": 8.052708625793457
    },
    {
      "epoch": 3.495121002197266e-05,
      "model_forward_time": 1.81663179397583,
      "step": 36649
    },
    {
      "epoch": 3.495121002197266e-05,
      "step": 36649,
      "training_step_time": 7.518116235733032
    },
    {
      "epoch": 3.495216369628906e-05,
      "grad_norm": 0.17771652340888977,
      "learning_rate": 7.502529346003559e-06,
      "loss": 0.0588,
      "step": 36650
    },
    {
      "epoch": 3.495216369628906e-05,
      "model_forward_time": 2.8961262702941895,
      "step": 36650
    },
    {
      "dynamics_loss_avg": 0.0543082932010293,
      "epoch": 3.495216369628906e-05,
      "step": 36650
    },
    {
      "action_loss_avg": 0.0012954349513165653,
      "epoch": 3.495216369628906e-05,
      "step": 36650
    },
    {
      "epoch": 3.495216369628906e-05,
      "step": 36650,
      "training_step_time": 7.594848155975342
    },
    {
      "epoch": 3.495311737060547e-05,
      "model_forward_time": 1.7445988655090332,
      "step": 36651
    },
    {
      "epoch": 3.495311737060547e-05,
      "step": 36651,
      "training_step_time": 8.420713186264038
    },
    {
      "epoch": 3.4954071044921876e-05,
      "model_forward_time": 1.7446260452270508,
      "step": 36652
    },
    {
      "epoch": 3.4954071044921876e-05,
      "step": 36652,
      "training_step_time": 8.596105337142944
    },
    {
      "epoch": 3.495502471923828e-05,
      "model_forward_time": 1.7491037845611572,
      "step": 36653
    },
    {
      "epoch": 3.495502471923828e-05,
      "step": 36653,
      "training_step_time": 7.724015951156616
    },
    {
      "epoch": 3.495597839355469e-05,
      "model_forward_time": 1.7355921268463135,
      "step": 36654
    },
    {
      "epoch": 3.495597839355469e-05,
      "step": 36654,
      "training_step_time": 7.519719839096069
    },
    {
      "epoch": 3.4956932067871095e-05,
      "model_forward_time": 1.7451457977294922,
      "step": 36655
    },
    {
      "epoch": 3.4956932067871095e-05,
      "step": 36655,
      "training_step_time": 7.444730520248413
    },
    {
      "epoch": 3.49578857421875e-05,
      "model_forward_time": 1.7421588897705078,
      "step": 36656
    },
    {
      "epoch": 3.49578857421875e-05,
      "step": 36656,
      "training_step_time": 7.23925518989563
    },
    {
      "epoch": 3.49588394165039e-05,
      "model_forward_time": 1.7417850494384766,
      "step": 36657
    },
    {
      "epoch": 3.49588394165039e-05,
      "step": 36657,
      "training_step_time": 7.552159309387207
    },
    {
      "epoch": 3.4959793090820314e-05,
      "model_forward_time": 1.792750597000122,
      "step": 36658
    },
    {
      "epoch": 3.4959793090820314e-05,
      "step": 36658,
      "training_step_time": 7.601156711578369
    },
    {
      "epoch": 3.496074676513672e-05,
      "model_forward_time": 2.8722991943359375,
      "step": 36659
    },
    {
      "epoch": 3.496074676513672e-05,
      "step": 36659,
      "training_step_time": 8.420920372009277
    },
    {
      "epoch": 3.496170043945312e-05,
      "grad_norm": 0.0979456976056099,
      "learning_rate": 7.501097745571107e-06,
      "loss": 0.0576,
      "step": 36660
    },
    {
      "epoch": 3.496170043945312e-05,
      "model_forward_time": 1.751075267791748,
      "step": 36660
    },
    {
      "dynamics_loss_avg": 0.05429815528914332,
      "epoch": 3.496170043945312e-05,
      "step": 36660
    },
    {
      "action_loss_avg": 0.0018898617359809577,
      "epoch": 3.496170043945312e-05,
      "step": 36660
    },
    {
      "epoch": 3.496170043945312e-05,
      "step": 36660,
      "training_step_time": 9.254145860671997
    },
    {
      "epoch": 3.496265411376953e-05,
      "model_forward_time": 1.7455499172210693,
      "step": 36661
    },
    {
      "epoch": 3.496265411376953e-05,
      "step": 36661,
      "training_step_time": 8.619094371795654
    },
    {
      "epoch": 3.4963607788085937e-05,
      "model_forward_time": 1.7651023864746094,
      "step": 36662
    },
    {
      "epoch": 3.4963607788085937e-05,
      "step": 36662,
      "training_step_time": 8.692306280136108
    },
    {
      "epoch": 3.496456146240234e-05,
      "model_forward_time": 1.7514276504516602,
      "step": 36663
    },
    {
      "epoch": 3.496456146240234e-05,
      "step": 36663,
      "training_step_time": 7.507834196090698
    },
    {
      "epoch": 3.496551513671875e-05,
      "model_forward_time": 1.7452154159545898,
      "step": 36664
    },
    {
      "epoch": 3.496551513671875e-05,
      "step": 36664,
      "training_step_time": 7.470164060592651
    },
    {
      "epoch": 3.4966468811035155e-05,
      "model_forward_time": 1.742751121520996,
      "step": 36665
    },
    {
      "epoch": 3.4966468811035155e-05,
      "step": 36665,
      "training_step_time": 7.257788896560669
    },
    {
      "epoch": 3.496742248535156e-05,
      "model_forward_time": 1.745922565460205,
      "step": 36666
    },
    {
      "epoch": 3.496742248535156e-05,
      "step": 36666,
      "training_step_time": 7.363646507263184
    },
    {
      "epoch": 3.496837615966797e-05,
      "model_forward_time": 1.756131887435913,
      "step": 36667
    },
    {
      "epoch": 3.496837615966797e-05,
      "step": 36667,
      "training_step_time": 8.618240594863892
    },
    {
      "epoch": 3.4969329833984374e-05,
      "model_forward_time": 2.8247647285461426,
      "step": 36668
    },
    {
      "epoch": 3.4969329833984374e-05,
      "step": 36668,
      "training_step_time": 7.643218278884888
    },
    {
      "epoch": 3.497028350830078e-05,
      "model_forward_time": 1.746283769607544,
      "step": 36669
    },
    {
      "epoch": 3.497028350830078e-05,
      "step": 36669,
      "training_step_time": 8.72095537185669
    },
    {
      "epoch": 3.497123718261719e-05,
      "grad_norm": 0.1121356412768364,
      "learning_rate": 7.499665871622362e-06,
      "loss": 0.0595,
      "step": 36670
    },
    {
      "epoch": 3.497123718261719e-05,
      "model_forward_time": 1.7520363330841064,
      "step": 36670
    },
    {
      "dynamics_loss_avg": 0.05236284425482154,
      "epoch": 3.497123718261719e-05,
      "step": 36670
    },
    {
      "action_loss_avg": 0.0015020627499325201,
      "epoch": 3.497123718261719e-05,
      "step": 36670
    },
    {
      "epoch": 3.497123718261719e-05,
      "step": 36670,
      "training_step_time": 8.438307285308838
    },
    {
      "epoch": 3.497219085693359e-05,
      "model_forward_time": 1.7526252269744873,
      "step": 36671
    },
    {
      "epoch": 3.497219085693359e-05,
      "step": 36671,
      "training_step_time": 9.172643184661865
    },
    {
      "epoch": 3.497314453125e-05,
      "model_forward_time": 1.751814842224121,
      "step": 36672
    },
    {
      "epoch": 3.497314453125e-05,
      "step": 36672,
      "training_step_time": 8.307170391082764
    },
    {
      "epoch": 3.497409820556641e-05,
      "model_forward_time": 1.7500848770141602,
      "step": 36673
    },
    {
      "epoch": 3.497409820556641e-05,
      "step": 36673,
      "training_step_time": 7.357065677642822
    },
    {
      "epoch": 3.497505187988281e-05,
      "model_forward_time": 1.7464351654052734,
      "step": 36674
    },
    {
      "epoch": 3.497505187988281e-05,
      "step": 36674,
      "training_step_time": 7.412964582443237
    },
    {
      "epoch": 3.4976005554199216e-05,
      "model_forward_time": 1.786935806274414,
      "step": 36675
    },
    {
      "epoch": 3.4976005554199216e-05,
      "step": 36675,
      "training_step_time": 7.274167537689209
    },
    {
      "epoch": 3.497695922851563e-05,
      "model_forward_time": 1.74802827835083,
      "step": 36676
    },
    {
      "epoch": 3.497695922851563e-05,
      "step": 36676,
      "training_step_time": 7.6003968715667725
    },
    {
      "epoch": 3.497791290283203e-05,
      "model_forward_time": 1.8059659004211426,
      "step": 36677
    },
    {
      "epoch": 3.497791290283203e-05,
      "step": 36677,
      "training_step_time": 8.055243015289307
    },
    {
      "epoch": 3.4978866577148435e-05,
      "model_forward_time": 2.9296340942382812,
      "step": 36678
    },
    {
      "epoch": 3.4978866577148435e-05,
      "step": 36678,
      "training_step_time": 10.375334739685059
    },
    {
      "epoch": 3.4979820251464846e-05,
      "model_forward_time": 1.7523765563964844,
      "step": 36679
    },
    {
      "epoch": 3.4979820251464846e-05,
      "step": 36679,
      "training_step_time": 9.028079509735107
    },
    {
      "epoch": 3.498077392578125e-05,
      "grad_norm": 0.12710535526275635,
      "learning_rate": 7.498233724313917e-06,
      "loss": 0.0587,
      "step": 36680
    },
    {
      "epoch": 3.498077392578125e-05,
      "model_forward_time": 1.7470920085906982,
      "step": 36680
    },
    {
      "dynamics_loss_avg": 0.057003376074135303,
      "epoch": 3.498077392578125e-05,
      "step": 36680
    },
    {
      "action_loss_avg": 0.0020214480813592673,
      "epoch": 3.498077392578125e-05,
      "step": 36680
    },
    {
      "epoch": 3.498077392578125e-05,
      "step": 36680,
      "training_step_time": 9.209598064422607
    },
    {
      "epoch": 3.4981727600097654e-05,
      "model_forward_time": 1.77622389793396,
      "step": 36681
    },
    {
      "epoch": 3.4981727600097654e-05,
      "step": 36681,
      "training_step_time": 7.573001146316528
    },
    {
      "epoch": 3.4982681274414065e-05,
      "model_forward_time": 1.7148103713989258,
      "step": 36682
    },
    {
      "epoch": 3.4982681274414065e-05,
      "step": 36682,
      "training_step_time": 7.292323589324951
    },
    {
      "epoch": 3.498363494873047e-05,
      "model_forward_time": 1.7435321807861328,
      "step": 36683
    },
    {
      "epoch": 3.498363494873047e-05,
      "step": 36683,
      "training_step_time": 7.430992603302002
    },
    {
      "epoch": 3.498458862304687e-05,
      "model_forward_time": 1.7342414855957031,
      "step": 36684
    },
    {
      "epoch": 3.498458862304687e-05,
      "step": 36684,
      "training_step_time": 7.285026550292969
    },
    {
      "epoch": 3.4985542297363283e-05,
      "model_forward_time": 1.7460589408874512,
      "step": 36685
    },
    {
      "epoch": 3.4985542297363283e-05,
      "step": 36685,
      "training_step_time": 8.638973474502563
    },
    {
      "epoch": 3.498649597167969e-05,
      "model_forward_time": 1.8115198612213135,
      "step": 36686
    },
    {
      "epoch": 3.498649597167969e-05,
      "step": 36686,
      "training_step_time": 7.605224132537842
    },
    {
      "epoch": 3.498744964599609e-05,
      "model_forward_time": 2.8959522247314453,
      "step": 36687
    },
    {
      "epoch": 3.498744964599609e-05,
      "step": 36687,
      "training_step_time": 8.747652530670166
    },
    {
      "epoch": 3.49884033203125e-05,
      "model_forward_time": 1.750295639038086,
      "step": 36688
    },
    {
      "epoch": 3.49884033203125e-05,
      "step": 36688,
      "training_step_time": 10.183987140655518
    },
    {
      "epoch": 3.4989356994628906e-05,
      "model_forward_time": 1.7447407245635986,
      "step": 36689
    },
    {
      "epoch": 3.4989356994628906e-05,
      "step": 36689,
      "training_step_time": 10.52496075630188
    },
    {
      "epoch": 3.499031066894531e-05,
      "grad_norm": 0.1705532968044281,
      "learning_rate": 7.496801303802388e-06,
      "loss": 0.0584,
      "step": 36690
    },
    {
      "epoch": 3.499031066894531e-05,
      "model_forward_time": 1.7451229095458984,
      "step": 36690
    },
    {
      "dynamics_loss_avg": 0.050466537661850455,
      "epoch": 3.499031066894531e-05,
      "step": 36690
    },
    {
      "action_loss_avg": 0.0015683101140893997,
      "epoch": 3.499031066894531e-05,
      "step": 36690
    },
    {
      "epoch": 3.499031066894531e-05,
      "step": 36690,
      "training_step_time": 8.557903528213501
    },
    {
      "epoch": 3.499126434326172e-05,
      "model_forward_time": 1.7440385818481445,
      "step": 36691
    },
    {
      "epoch": 3.499126434326172e-05,
      "step": 36691,
      "training_step_time": 7.503434419631958
    },
    {
      "epoch": 3.4992218017578125e-05,
      "model_forward_time": 1.742410659790039,
      "step": 36692
    },
    {
      "epoch": 3.4992218017578125e-05,
      "step": 36692,
      "training_step_time": 7.4694907665252686
    },
    {
      "epoch": 3.499317169189453e-05,
      "model_forward_time": 1.735473394393921,
      "step": 36693
    },
    {
      "epoch": 3.499317169189453e-05,
      "step": 36693,
      "training_step_time": 7.329212427139282
    },
    {
      "epoch": 3.499412536621094e-05,
      "model_forward_time": 1.7552127838134766,
      "step": 36694
    },
    {
      "epoch": 3.499412536621094e-05,
      "step": 36694,
      "training_step_time": 7.6752471923828125
    },
    {
      "epoch": 3.4995079040527344e-05,
      "model_forward_time": 1.7825648784637451,
      "step": 36695
    },
    {
      "epoch": 3.4995079040527344e-05,
      "step": 36695,
      "training_step_time": 7.572260856628418
    },
    {
      "epoch": 3.499603271484375e-05,
      "model_forward_time": 2.8823559284210205,
      "step": 36696
    },
    {
      "epoch": 3.499603271484375e-05,
      "step": 36696,
      "training_step_time": 7.593318223953247
    },
    {
      "epoch": 3.499698638916016e-05,
      "model_forward_time": 1.750448226928711,
      "step": 36697
    },
    {
      "epoch": 3.499698638916016e-05,
      "step": 36697,
      "training_step_time": 8.957483291625977
    },
    {
      "epoch": 3.499794006347656e-05,
      "model_forward_time": 1.7555758953094482,
      "step": 36698
    },
    {
      "epoch": 3.499794006347656e-05,
      "step": 36698,
      "training_step_time": 10.034253358840942
    },
    {
      "epoch": 3.499889373779297e-05,
      "model_forward_time": 1.7582671642303467,
      "step": 36699
    },
    {
      "epoch": 3.499889373779297e-05,
      "step": 36699,
      "training_step_time": 8.638799667358398
    },
    {
      "epoch": 3.499984741210938e-05,
      "grad_norm": 0.16806286573410034,
      "learning_rate": 7.495368610244422e-06,
      "loss": 0.0564,
      "step": 36700
    },
    {
      "epoch": 3.499984741210938e-05,
      "model_forward_time": 1.7497847080230713,
      "step": 36700
    },
    {
      "dynamics_loss_avg": 0.06510962191969157,
      "epoch": 3.499984741210938e-05,
      "step": 36700
    },
    {
      "action_loss_avg": 0.001908525815815665,
      "epoch": 3.499984741210938e-05,
      "step": 36700
    },
    {
      "epoch": 3.499984741210938e-05,
      "step": 36700,
      "training_step_time": 7.529228925704956
    },
    {
      "epoch": 3.500080108642578e-05,
      "model_forward_time": 1.7396297454833984,
      "step": 36701
    },
    {
      "epoch": 3.500080108642578e-05,
      "step": 36701,
      "training_step_time": 7.431828737258911
    },
    {
      "epoch": 3.5001754760742186e-05,
      "model_forward_time": 1.743851900100708,
      "step": 36702
    },
    {
      "epoch": 3.5001754760742186e-05,
      "step": 36702,
      "training_step_time": 7.315606594085693
    },
    {
      "epoch": 3.5002708435058597e-05,
      "model_forward_time": 1.742546796798706,
      "step": 36703
    },
    {
      "epoch": 3.5002708435058597e-05,
      "step": 36703,
      "training_step_time": 8.100696086883545
    },
    {
      "epoch": 3.5003662109375e-05,
      "model_forward_time": 1.752009630203247,
      "step": 36704
    },
    {
      "epoch": 3.5003662109375e-05,
      "step": 36704,
      "training_step_time": 8.338448762893677
    },
    {
      "epoch": 3.5004615783691405e-05,
      "model_forward_time": 2.901127576828003,
      "step": 36705
    },
    {
      "epoch": 3.5004615783691405e-05,
      "step": 36705,
      "training_step_time": 8.65573787689209
    },
    {
      "epoch": 3.5005569458007815e-05,
      "model_forward_time": 1.7490465641021729,
      "step": 36706
    },
    {
      "epoch": 3.5005569458007815e-05,
      "step": 36706,
      "training_step_time": 9.379544019699097
    },
    {
      "epoch": 3.500652313232422e-05,
      "model_forward_time": 1.7503674030303955,
      "step": 36707
    },
    {
      "epoch": 3.500652313232422e-05,
      "step": 36707,
      "training_step_time": 8.402663707733154
    },
    {
      "epoch": 3.5007476806640623e-05,
      "model_forward_time": 1.7534289360046387,
      "step": 36708
    },
    {
      "epoch": 3.5007476806640623e-05,
      "step": 36708,
      "training_step_time": 7.642474412918091
    },
    {
      "epoch": 3.5008430480957034e-05,
      "model_forward_time": 1.7541685104370117,
      "step": 36709
    },
    {
      "epoch": 3.5008430480957034e-05,
      "step": 36709,
      "training_step_time": 7.509514331817627
    },
    {
      "epoch": 3.500938415527344e-05,
      "grad_norm": 0.1331774741411209,
      "learning_rate": 7.493935643796697e-06,
      "loss": 0.0587,
      "step": 36710
    },
    {
      "epoch": 3.500938415527344e-05,
      "model_forward_time": 1.7551548480987549,
      "step": 36710
    },
    {
      "dynamics_loss_avg": 0.05751214511692524,
      "epoch": 3.500938415527344e-05,
      "step": 36710
    },
    {
      "action_loss_avg": 0.0016934455983573572,
      "epoch": 3.500938415527344e-05,
      "step": 36710
    },
    {
      "epoch": 3.500938415527344e-05,
      "step": 36710,
      "training_step_time": 7.398881912231445
    },
    {
      "epoch": 3.501033782958984e-05,
      "model_forward_time": 1.7511582374572754,
      "step": 36711
    },
    {
      "epoch": 3.501033782958984e-05,
      "step": 36711,
      "training_step_time": 7.2684009075164795
    },
    {
      "epoch": 3.501129150390625e-05,
      "model_forward_time": 1.7478115558624268,
      "step": 36712
    },
    {
      "epoch": 3.501129150390625e-05,
      "step": 36712,
      "training_step_time": 7.500732660293579
    },
    {
      "epoch": 3.501224517822266e-05,
      "model_forward_time": 1.7532029151916504,
      "step": 36713
    },
    {
      "epoch": 3.501224517822266e-05,
      "step": 36713,
      "training_step_time": 8.060332298278809
    },
    {
      "epoch": 3.501319885253906e-05,
      "model_forward_time": 2.9279472827911377,
      "step": 36714
    },
    {
      "epoch": 3.501319885253906e-05,
      "step": 36714,
      "training_step_time": 8.63833475112915
    },
    {
      "epoch": 3.501415252685547e-05,
      "model_forward_time": 1.7492053508758545,
      "step": 36715
    },
    {
      "epoch": 3.501415252685547e-05,
      "step": 36715,
      "training_step_time": 9.466972589492798
    },
    {
      "epoch": 3.5015106201171876e-05,
      "model_forward_time": 1.7562861442565918,
      "step": 36716
    },
    {
      "epoch": 3.5015106201171876e-05,
      "step": 36716,
      "training_step_time": 7.888732194900513
    },
    {
      "epoch": 3.501605987548828e-05,
      "model_forward_time": 1.752889633178711,
      "step": 36717
    },
    {
      "epoch": 3.501605987548828e-05,
      "step": 36717,
      "training_step_time": 7.639686822891235
    },
    {
      "epoch": 3.501701354980469e-05,
      "model_forward_time": 1.7553787231445312,
      "step": 36718
    },
    {
      "epoch": 3.501701354980469e-05,
      "step": 36718,
      "training_step_time": 7.506085157394409
    },
    {
      "epoch": 3.5017967224121095e-05,
      "model_forward_time": 1.757636547088623,
      "step": 36719
    },
    {
      "epoch": 3.5017967224121095e-05,
      "step": 36719,
      "training_step_time": 7.101219415664673
    },
    {
      "epoch": 3.50189208984375e-05,
      "grad_norm": 0.1394171118736267,
      "learning_rate": 7.49250240461592e-06,
      "loss": 0.0594,
      "step": 36720
    },
    {
      "epoch": 3.50189208984375e-05,
      "model_forward_time": 1.7517337799072266,
      "step": 36720
    },
    {
      "dynamics_loss_avg": 0.05840734364464879,
      "epoch": 3.50189208984375e-05,
      "step": 36720
    },
    {
      "action_loss_avg": 0.0020419913169462236,
      "epoch": 3.50189208984375e-05,
      "step": 36720
    },
    {
      "epoch": 3.50189208984375e-05,
      "step": 36720,
      "training_step_time": 7.40378475189209
    },
    {
      "epoch": 3.50198745727539e-05,
      "model_forward_time": 1.7488980293273926,
      "step": 36721
    },
    {
      "epoch": 3.50198745727539e-05,
      "step": 36721,
      "training_step_time": 7.484351873397827
    },
    {
      "epoch": 3.5020828247070314e-05,
      "model_forward_time": 1.7504444122314453,
      "step": 36722
    },
    {
      "epoch": 3.5020828247070314e-05,
      "step": 36722,
      "training_step_time": 7.584580421447754
    },
    {
      "epoch": 3.502178192138672e-05,
      "model_forward_time": 1.8227453231811523,
      "step": 36723
    },
    {
      "epoch": 3.502178192138672e-05,
      "step": 36723,
      "training_step_time": 8.66666841506958
    },
    {
      "epoch": 3.502273559570312e-05,
      "model_forward_time": 1.7484002113342285,
      "step": 36724
    },
    {
      "epoch": 3.502273559570312e-05,
      "step": 36724,
      "training_step_time": 8.50504732131958
    },
    {
      "epoch": 3.502368927001953e-05,
      "model_forward_time": 1.755955457687378,
      "step": 36725
    },
    {
      "epoch": 3.502368927001953e-05,
      "step": 36725,
      "training_step_time": 7.89129376411438
    },
    {
      "epoch": 3.502464294433594e-05,
      "model_forward_time": 1.759587049484253,
      "step": 36726
    },
    {
      "epoch": 3.502464294433594e-05,
      "step": 36726,
      "training_step_time": 8.689653158187866
    },
    {
      "epoch": 3.502559661865234e-05,
      "model_forward_time": 1.748492956161499,
      "step": 36727
    },
    {
      "epoch": 3.502559661865234e-05,
      "step": 36727,
      "training_step_time": 9.242854833602905
    },
    {
      "epoch": 3.502655029296875e-05,
      "model_forward_time": 1.7506906986236572,
      "step": 36728
    },
    {
      "epoch": 3.502655029296875e-05,
      "step": 36728,
      "training_step_time": 7.373207330703735
    },
    {
      "epoch": 3.5027503967285155e-05,
      "model_forward_time": 1.7572832107543945,
      "step": 36729
    },
    {
      "epoch": 3.5027503967285155e-05,
      "step": 36729,
      "training_step_time": 7.528848886489868
    },
    {
      "epoch": 3.502845764160156e-05,
      "grad_norm": 0.19118426740169525,
      "learning_rate": 7.4910688928588274e-06,
      "loss": 0.058,
      "step": 36730
    },
    {
      "epoch": 3.502845764160156e-05,
      "model_forward_time": 1.7451183795928955,
      "step": 36730
    },
    {
      "dynamics_loss_avg": 0.04770920854061842,
      "epoch": 3.502845764160156e-05,
      "step": 36730
    },
    {
      "action_loss_avg": 0.0019966877327533437,
      "epoch": 3.502845764160156e-05,
      "step": 36730
    },
    {
      "epoch": 3.502845764160156e-05,
      "step": 36730,
      "training_step_time": 7.670402526855469
    },
    {
      "epoch": 3.502941131591797e-05,
      "model_forward_time": 1.7550296783447266,
      "step": 36731
    },
    {
      "epoch": 3.502941131591797e-05,
      "step": 36731,
      "training_step_time": 7.624422311782837
    },
    {
      "epoch": 3.5030364990234374e-05,
      "model_forward_time": 1.8207275867462158,
      "step": 36732
    },
    {
      "epoch": 3.5030364990234374e-05,
      "step": 36732,
      "training_step_time": 8.32770848274231
    },
    {
      "epoch": 3.503131866455078e-05,
      "model_forward_time": 2.9048914909362793,
      "step": 36733
    },
    {
      "epoch": 3.503131866455078e-05,
      "step": 36733,
      "training_step_time": 7.831985712051392
    },
    {
      "epoch": 3.503227233886719e-05,
      "model_forward_time": 1.7554616928100586,
      "step": 36734
    },
    {
      "epoch": 3.503227233886719e-05,
      "step": 36734,
      "training_step_time": 8.708691120147705
    },
    {
      "epoch": 3.503322601318359e-05,
      "model_forward_time": 1.7547821998596191,
      "step": 36735
    },
    {
      "epoch": 3.503322601318359e-05,
      "step": 36735,
      "training_step_time": 8.358152627944946
    },
    {
      "epoch": 3.50341796875e-05,
      "model_forward_time": 1.7534770965576172,
      "step": 36736
    },
    {
      "epoch": 3.50341796875e-05,
      "step": 36736,
      "training_step_time": 7.712388515472412
    },
    {
      "epoch": 3.503513336181641e-05,
      "model_forward_time": 1.752760887145996,
      "step": 36737
    },
    {
      "epoch": 3.503513336181641e-05,
      "step": 36737,
      "training_step_time": 7.5714430809021
    },
    {
      "epoch": 3.503608703613281e-05,
      "model_forward_time": 1.7458033561706543,
      "step": 36738
    },
    {
      "epoch": 3.503608703613281e-05,
      "step": 36738,
      "training_step_time": 7.486896514892578
    },
    {
      "epoch": 3.5037040710449216e-05,
      "model_forward_time": 1.750708818435669,
      "step": 36739
    },
    {
      "epoch": 3.5037040710449216e-05,
      "step": 36739,
      "training_step_time": 7.630631923675537
    },
    {
      "epoch": 3.503799438476563e-05,
      "grad_norm": 0.1335170418024063,
      "learning_rate": 7.4896351086821835e-06,
      "loss": 0.0551,
      "step": 36740
    },
    {
      "epoch": 3.503799438476563e-05,
      "model_forward_time": 1.7564375400543213,
      "step": 36740
    },
    {
      "dynamics_loss_avg": 0.05302042383700609,
      "epoch": 3.503799438476563e-05,
      "step": 36740
    },
    {
      "action_loss_avg": 0.0021476861438713968,
      "epoch": 3.503799438476563e-05,
      "step": 36740
    },
    {
      "epoch": 3.503799438476563e-05,
      "step": 36740,
      "training_step_time": 7.632238149642944
    },
    {
      "epoch": 3.503894805908203e-05,
      "model_forward_time": 1.7925159931182861,
      "step": 36741
    },
    {
      "epoch": 3.503894805908203e-05,
      "step": 36741,
      "training_step_time": 7.650248765945435
    },
    {
      "epoch": 3.5039901733398435e-05,
      "model_forward_time": 2.936943769454956,
      "step": 36742
    },
    {
      "epoch": 3.5039901733398435e-05,
      "step": 36742,
      "training_step_time": 10.063027381896973
    },
    {
      "epoch": 3.5040855407714846e-05,
      "model_forward_time": 1.7580931186676025,
      "step": 36743
    },
    {
      "epoch": 3.5040855407714846e-05,
      "step": 36743,
      "training_step_time": 7.779639005661011
    },
    {
      "epoch": 3.504180908203125e-05,
      "model_forward_time": 1.7516968250274658,
      "step": 36744
    },
    {
      "epoch": 3.504180908203125e-05,
      "step": 36744,
      "training_step_time": 7.831876993179321
    },
    {
      "epoch": 3.5042762756347654e-05,
      "model_forward_time": 1.7544641494750977,
      "step": 36745
    },
    {
      "epoch": 3.5042762756347654e-05,
      "step": 36745,
      "training_step_time": 7.690599203109741
    },
    {
      "epoch": 3.5043716430664065e-05,
      "model_forward_time": 1.755053997039795,
      "step": 36746
    },
    {
      "epoch": 3.5043716430664065e-05,
      "step": 36746,
      "training_step_time": 7.638111114501953
    },
    {
      "epoch": 3.504467010498047e-05,
      "model_forward_time": 1.7513341903686523,
      "step": 36747
    },
    {
      "epoch": 3.504467010498047e-05,
      "step": 36747,
      "training_step_time": 7.5788328647613525
    },
    {
      "epoch": 3.504562377929687e-05,
      "model_forward_time": 1.752519130706787,
      "step": 36748
    },
    {
      "epoch": 3.504562377929687e-05,
      "step": 36748,
      "training_step_time": 7.615460157394409
    },
    {
      "epoch": 3.5046577453613283e-05,
      "model_forward_time": 1.7594521045684814,
      "step": 36749
    },
    {
      "epoch": 3.5046577453613283e-05,
      "step": 36749,
      "training_step_time": 7.626418590545654
    },
    {
      "epoch": 3.504753112792969e-05,
      "grad_norm": 0.17762328684329987,
      "learning_rate": 7.488201052242791e-06,
      "loss": 0.0587,
      "step": 36750
    },
    {
      "epoch": 3.504753112792969e-05,
      "model_forward_time": 1.764641284942627,
      "step": 36750
    },
    {
      "dynamics_loss_avg": 0.06111472146585584,
      "epoch": 3.504753112792969e-05,
      "step": 36750
    },
    {
      "action_loss_avg": 0.0022804064326919617,
      "epoch": 3.504753112792969e-05,
      "step": 36750
    },
    {
      "epoch": 3.504753112792969e-05,
      "step": 36750,
      "training_step_time": 7.596565008163452
    },
    {
      "epoch": 3.504848480224609e-05,
      "model_forward_time": 2.9155845642089844,
      "step": 36751
    },
    {
      "epoch": 3.504848480224609e-05,
      "step": 36751,
      "training_step_time": 9.624286413192749
    },
    {
      "epoch": 3.50494384765625e-05,
      "model_forward_time": 1.7544126510620117,
      "step": 36752
    },
    {
      "epoch": 3.50494384765625e-05,
      "step": 36752,
      "training_step_time": 8.676735162734985
    },
    {
      "epoch": 3.5050392150878906e-05,
      "model_forward_time": 1.7555193901062012,
      "step": 36753
    },
    {
      "epoch": 3.5050392150878906e-05,
      "step": 36753,
      "training_step_time": 8.884544134140015
    },
    {
      "epoch": 3.505134582519531e-05,
      "model_forward_time": 1.7496912479400635,
      "step": 36754
    },
    {
      "epoch": 3.505134582519531e-05,
      "step": 36754,
      "training_step_time": 7.777653932571411
    },
    {
      "epoch": 3.505229949951172e-05,
      "model_forward_time": 1.7528302669525146,
      "step": 36755
    },
    {
      "epoch": 3.505229949951172e-05,
      "step": 36755,
      "training_step_time": 7.582628011703491
    },
    {
      "epoch": 3.5053253173828125e-05,
      "model_forward_time": 1.7507834434509277,
      "step": 36756
    },
    {
      "epoch": 3.5053253173828125e-05,
      "step": 36756,
      "training_step_time": 7.530688047409058
    },
    {
      "epoch": 3.505420684814453e-05,
      "model_forward_time": 1.7606024742126465,
      "step": 36757
    },
    {
      "epoch": 3.505420684814453e-05,
      "step": 36757,
      "training_step_time": 8.281415939331055
    },
    {
      "epoch": 3.505516052246094e-05,
      "model_forward_time": 1.7624180316925049,
      "step": 36758
    },
    {
      "epoch": 3.505516052246094e-05,
      "step": 36758,
      "training_step_time": 8.623616695404053
    },
    {
      "epoch": 3.5056114196777344e-05,
      "model_forward_time": 1.751030445098877,
      "step": 36759
    },
    {
      "epoch": 3.5056114196777344e-05,
      "step": 36759,
      "training_step_time": 7.510038375854492
    },
    {
      "epoch": 3.505706787109375e-05,
      "grad_norm": 0.1394663155078888,
      "learning_rate": 7.4867667236974695e-06,
      "loss": 0.0547,
      "step": 36760
    },
    {
      "epoch": 3.505706787109375e-05,
      "model_forward_time": 2.8824570178985596,
      "step": 36760
    },
    {
      "dynamics_loss_avg": 0.05594191662967205,
      "epoch": 3.505706787109375e-05,
      "step": 36760
    },
    {
      "action_loss_avg": 0.0017888517759274691,
      "epoch": 3.505706787109375e-05,
      "step": 36760
    },
    {
      "epoch": 3.505706787109375e-05,
      "step": 36760,
      "training_step_time": 8.719287157058716
    },
    {
      "epoch": 3.505802154541016e-05,
      "model_forward_time": 1.7498161792755127,
      "step": 36761
    },
    {
      "epoch": 3.505802154541016e-05,
      "step": 36761,
      "training_step_time": 9.875296115875244
    },
    {
      "epoch": 3.505897521972656e-05,
      "model_forward_time": 1.753197193145752,
      "step": 36762
    },
    {
      "epoch": 3.505897521972656e-05,
      "step": 36762,
      "training_step_time": 7.865283966064453
    },
    {
      "epoch": 3.505992889404297e-05,
      "model_forward_time": 1.7469899654388428,
      "step": 36763
    },
    {
      "epoch": 3.505992889404297e-05,
      "step": 36763,
      "training_step_time": 7.720520496368408
    },
    {
      "epoch": 3.506088256835938e-05,
      "model_forward_time": 1.7490758895874023,
      "step": 36764
    },
    {
      "epoch": 3.506088256835938e-05,
      "step": 36764,
      "training_step_time": 8.664756536483765
    },
    {
      "epoch": 3.506183624267578e-05,
      "model_forward_time": 1.743769645690918,
      "step": 36765
    },
    {
      "epoch": 3.506183624267578e-05,
      "step": 36765,
      "training_step_time": 7.551192998886108
    },
    {
      "epoch": 3.5062789916992186e-05,
      "model_forward_time": 1.764211654663086,
      "step": 36766
    },
    {
      "epoch": 3.5062789916992186e-05,
      "step": 36766,
      "training_step_time": 8.40232229232788
    },
    {
      "epoch": 3.50637435913086e-05,
      "model_forward_time": 1.7869644165039062,
      "step": 36767
    },
    {
      "epoch": 3.50637435913086e-05,
      "step": 36767,
      "training_step_time": 7.597784519195557
    },
    {
      "epoch": 3.5064697265625e-05,
      "model_forward_time": 1.7510724067687988,
      "step": 36768
    },
    {
      "epoch": 3.5064697265625e-05,
      "step": 36768,
      "training_step_time": 7.6198084354400635
    },
    {
      "epoch": 3.5065650939941405e-05,
      "model_forward_time": 2.9527552127838135,
      "step": 36769
    },
    {
      "epoch": 3.5065650939941405e-05,
      "step": 36769,
      "training_step_time": 7.740011215209961
    },
    {
      "epoch": 3.5066604614257815e-05,
      "grad_norm": 0.15066903829574585,
      "learning_rate": 7.4853321232030786e-06,
      "loss": 0.0599,
      "step": 36770
    },
    {
      "epoch": 3.5066604614257815e-05,
      "model_forward_time": 1.7527220249176025,
      "step": 36770
    },
    {
      "dynamics_loss_avg": 0.04592310348525643,
      "epoch": 3.5066604614257815e-05,
      "step": 36770
    },
    {
      "action_loss_avg": 0.0015842083957977593,
      "epoch": 3.5066604614257815e-05,
      "step": 36770
    },
    {
      "epoch": 3.5066604614257815e-05,
      "step": 36770,
      "training_step_time": 7.76865029335022
    },
    {
      "epoch": 3.506755828857422e-05,
      "model_forward_time": 1.7444570064544678,
      "step": 36771
    },
    {
      "epoch": 3.506755828857422e-05,
      "step": 36771,
      "training_step_time": 7.689895868301392
    },
    {
      "epoch": 3.5068511962890624e-05,
      "model_forward_time": 1.9306731224060059,
      "step": 36772
    },
    {
      "epoch": 3.5068511962890624e-05,
      "step": 36772,
      "training_step_time": 8.361344337463379
    },
    {
      "epoch": 3.5069465637207034e-05,
      "model_forward_time": 1.5566182136535645,
      "step": 36773
    },
    {
      "epoch": 3.5069465637207034e-05,
      "step": 36773,
      "training_step_time": 8.62275505065918
    },
    {
      "epoch": 3.507041931152344e-05,
      "model_forward_time": 1.895310401916504,
      "step": 36774
    },
    {
      "epoch": 3.507041931152344e-05,
      "step": 36774,
      "training_step_time": 7.630919456481934
    },
    {
      "epoch": 3.507137298583984e-05,
      "model_forward_time": 1.9035112857818604,
      "step": 36775
    },
    {
      "epoch": 3.507137298583984e-05,
      "step": 36775,
      "training_step_time": 7.64482855796814
    },
    {
      "epoch": 3.507232666015625e-05,
      "model_forward_time": 1.9151990413665771,
      "step": 36776
    },
    {
      "epoch": 3.507232666015625e-05,
      "step": 36776,
      "training_step_time": 7.614721059799194
    },
    {
      "epoch": 3.507328033447266e-05,
      "model_forward_time": 1.9273905754089355,
      "step": 36777
    },
    {
      "epoch": 3.507328033447266e-05,
      "step": 36777,
      "training_step_time": 7.599006175994873
    },
    {
      "epoch": 3.507423400878906e-05,
      "model_forward_time": 3.0223371982574463,
      "step": 36778
    },
    {
      "epoch": 3.507423400878906e-05,
      "step": 36778,
      "training_step_time": 8.64383316040039
    },
    {
      "epoch": 3.507518768310547e-05,
      "model_forward_time": 1.9330689907073975,
      "step": 36779
    },
    {
      "epoch": 3.507518768310547e-05,
      "step": 36779,
      "training_step_time": 9.845542669296265
    },
    {
      "epoch": 3.5076141357421876e-05,
      "grad_norm": 0.12376134842634201,
      "learning_rate": 7.4838972509165045e-06,
      "loss": 0.0541,
      "step": 36780
    },
    {
      "epoch": 3.5076141357421876e-05,
      "model_forward_time": 1.9079833030700684,
      "step": 36780
    },
    {
      "dynamics_loss_avg": 0.061204462125897405,
      "epoch": 3.5076141357421876e-05,
      "step": 36780
    },
    {
      "action_loss_avg": 0.0019231782876886427,
      "epoch": 3.5076141357421876e-05,
      "step": 36780
    },
    {
      "epoch": 3.5076141357421876e-05,
      "step": 36780,
      "training_step_time": 7.6702001094818115
    },
    {
      "epoch": 3.507709503173828e-05,
      "model_forward_time": 1.9237825870513916,
      "step": 36781
    },
    {
      "epoch": 3.507709503173828e-05,
      "step": 36781,
      "training_step_time": 8.834988355636597
    },
    {
      "epoch": 3.5078048706054684e-05,
      "model_forward_time": 1.9370267391204834,
      "step": 36782
    },
    {
      "epoch": 3.5078048706054684e-05,
      "step": 36782,
      "training_step_time": 7.600181579589844
    },
    {
      "epoch": 3.5079002380371095e-05,
      "model_forward_time": 1.910280704498291,
      "step": 36783
    },
    {
      "epoch": 3.5079002380371095e-05,
      "step": 36783,
      "training_step_time": 7.634695053100586
    },
    {
      "epoch": 3.50799560546875e-05,
      "model_forward_time": 1.7565321922302246,
      "step": 36784
    },
    {
      "epoch": 3.50799560546875e-05,
      "step": 36784,
      "training_step_time": 7.597181081771851
    },
    {
      "epoch": 3.50809097290039e-05,
      "model_forward_time": 1.7448656558990479,
      "step": 36785
    },
    {
      "epoch": 3.50809097290039e-05,
      "step": 36785,
      "training_step_time": 7.627596616744995
    },
    {
      "epoch": 3.5081863403320314e-05,
      "model_forward_time": 1.7531421184539795,
      "step": 36786
    },
    {
      "epoch": 3.5081863403320314e-05,
      "step": 36786,
      "training_step_time": 7.587639331817627
    },
    {
      "epoch": 3.508281707763672e-05,
      "model_forward_time": 2.871429681777954,
      "step": 36787
    },
    {
      "epoch": 3.508281707763672e-05,
      "step": 36787,
      "training_step_time": 8.214082956314087
    },
    {
      "epoch": 3.508377075195312e-05,
      "model_forward_time": 1.7521719932556152,
      "step": 36788
    },
    {
      "epoch": 3.508377075195312e-05,
      "step": 36788,
      "training_step_time": 8.560629844665527
    },
    {
      "epoch": 3.508472442626953e-05,
      "model_forward_time": 1.7420589923858643,
      "step": 36789
    },
    {
      "epoch": 3.508472442626953e-05,
      "step": 36789,
      "training_step_time": 7.656090974807739
    },
    {
      "epoch": 3.508567810058594e-05,
      "grad_norm": 0.11644420027732849,
      "learning_rate": 7.48246210699466e-06,
      "loss": 0.0572,
      "step": 36790
    },
    {
      "epoch": 3.508567810058594e-05,
      "model_forward_time": 1.7450876235961914,
      "step": 36790
    },
    {
      "dynamics_loss_avg": 0.0623953178524971,
      "epoch": 3.508567810058594e-05,
      "step": 36790
    },
    {
      "action_loss_avg": 0.001846007673884742,
      "epoch": 3.508567810058594e-05,
      "step": 36790
    },
    {
      "epoch": 3.508567810058594e-05,
      "step": 36790,
      "training_step_time": 9.853943109512329
    },
    {
      "epoch": 3.508663177490234e-05,
      "model_forward_time": 1.7459776401519775,
      "step": 36791
    },
    {
      "epoch": 3.508663177490234e-05,
      "step": 36791,
      "training_step_time": 7.825634956359863
    },
    {
      "epoch": 3.508758544921875e-05,
      "model_forward_time": 1.748610258102417,
      "step": 36792
    },
    {
      "epoch": 3.508758544921875e-05,
      "step": 36792,
      "training_step_time": 7.623373985290527
    },
    {
      "epoch": 3.5088539123535156e-05,
      "model_forward_time": 1.753849983215332,
      "step": 36793
    },
    {
      "epoch": 3.5088539123535156e-05,
      "step": 36793,
      "training_step_time": 7.590588092803955
    },
    {
      "epoch": 3.508949279785156e-05,
      "model_forward_time": 1.7478110790252686,
      "step": 36794
    },
    {
      "epoch": 3.508949279785156e-05,
      "step": 36794,
      "training_step_time": 8.560206651687622
    },
    {
      "epoch": 3.509044647216797e-05,
      "model_forward_time": 1.7451000213623047,
      "step": 36795
    },
    {
      "epoch": 3.509044647216797e-05,
      "step": 36795,
      "training_step_time": 7.540624618530273
    },
    {
      "epoch": 3.5091400146484374e-05,
      "model_forward_time": 2.8869564533233643,
      "step": 36796
    },
    {
      "epoch": 3.5091400146484374e-05,
      "step": 36796,
      "training_step_time": 7.660090446472168
    },
    {
      "epoch": 3.509235382080078e-05,
      "model_forward_time": 1.8122146129608154,
      "step": 36797
    },
    {
      "epoch": 3.509235382080078e-05,
      "step": 36797,
      "training_step_time": 9.302198648452759
    },
    {
      "epoch": 3.509330749511719e-05,
      "model_forward_time": 1.7443337440490723,
      "step": 36798
    },
    {
      "epoch": 3.509330749511719e-05,
      "step": 36798,
      "training_step_time": 7.715556859970093
    },
    {
      "epoch": 3.509426116943359e-05,
      "model_forward_time": 1.7464113235473633,
      "step": 36799
    },
    {
      "epoch": 3.509426116943359e-05,
      "step": 36799,
      "training_step_time": 7.75256085395813
    },
    {
      "epoch": 3.509521484375e-05,
      "grad_norm": 0.11936287581920624,
      "learning_rate": 7.481026691594493e-06,
      "loss": 0.0582,
      "step": 36800
    },
    {
      "epoch": 3.509521484375e-05,
      "model_forward_time": 1.7516295909881592,
      "step": 36800
    },
    {
      "dynamics_loss_avg": 0.04535635793581605,
      "epoch": 3.509521484375e-05,
      "step": 36800
    },
    {
      "action_loss_avg": 0.0012744968815241009,
      "epoch": 3.509521484375e-05,
      "step": 36800
    },
    {
      "epoch": 3.509521484375e-05,
      "step": 36800,
      "training_step_time": 7.742088079452515
    },
    {
      "epoch": 3.509616851806641e-05,
      "model_forward_time": 1.750577449798584,
      "step": 36801
    },
    {
      "epoch": 3.509616851806641e-05,
      "step": 36801,
      "training_step_time": 8.425477027893066
    },
    {
      "epoch": 3.509712219238281e-05,
      "model_forward_time": 1.7485382556915283,
      "step": 36802
    },
    {
      "epoch": 3.509712219238281e-05,
      "step": 36802,
      "training_step_time": 7.529690265655518
    },
    {
      "epoch": 3.5098075866699216e-05,
      "model_forward_time": 1.75380539894104,
      "step": 36803
    },
    {
      "epoch": 3.5098075866699216e-05,
      "step": 36803,
      "training_step_time": 7.697511672973633
    },
    {
      "epoch": 3.509902954101563e-05,
      "model_forward_time": 1.7442512512207031,
      "step": 36804
    },
    {
      "epoch": 3.509902954101563e-05,
      "step": 36804,
      "training_step_time": 7.508028507232666
    },
    {
      "epoch": 3.509998321533203e-05,
      "model_forward_time": 2.8424339294433594,
      "step": 36805
    },
    {
      "epoch": 3.509998321533203e-05,
      "step": 36805,
      "training_step_time": 8.19181776046753
    },
    {
      "epoch": 3.5100936889648435e-05,
      "model_forward_time": 1.4464449882507324,
      "step": 36806
    },
    {
      "epoch": 3.5100936889648435e-05,
      "step": 36806,
      "training_step_time": 7.655808687210083
    },
    {
      "epoch": 3.5101890563964846e-05,
      "model_forward_time": 1.7511379718780518,
      "step": 36807
    },
    {
      "epoch": 3.5101890563964846e-05,
      "step": 36807,
      "training_step_time": 8.355465650558472
    },
    {
      "epoch": 3.510284423828125e-05,
      "model_forward_time": 1.7465693950653076,
      "step": 36808
    },
    {
      "epoch": 3.510284423828125e-05,
      "step": 36808,
      "training_step_time": 9.228909015655518
    },
    {
      "epoch": 3.5103797912597654e-05,
      "model_forward_time": 1.7463524341583252,
      "step": 36809
    },
    {
      "epoch": 3.5103797912597654e-05,
      "step": 36809,
      "training_step_time": 7.7125396728515625
    },
    {
      "epoch": 3.5104751586914065e-05,
      "grad_norm": 0.11826109886169434,
      "learning_rate": 7.479591004872976e-06,
      "loss": 0.0578,
      "step": 36810
    },
    {
      "epoch": 3.5104751586914065e-05,
      "model_forward_time": 1.7847239971160889,
      "step": 36810
    },
    {
      "dynamics_loss_avg": 0.05224880948662758,
      "epoch": 3.5104751586914065e-05,
      "step": 36810
    },
    {
      "action_loss_avg": 0.0018444184184772893,
      "epoch": 3.5104751586914065e-05,
      "step": 36810
    },
    {
      "epoch": 3.5104751586914065e-05,
      "step": 36810,
      "training_step_time": 8.206891298294067
    },
    {
      "epoch": 3.510570526123047e-05,
      "model_forward_time": 1.7502872943878174,
      "step": 36811
    },
    {
      "epoch": 3.510570526123047e-05,
      "step": 36811,
      "training_step_time": 8.267261266708374
    },
    {
      "epoch": 3.510665893554687e-05,
      "model_forward_time": 1.7453675270080566,
      "step": 36812
    },
    {
      "epoch": 3.510665893554687e-05,
      "step": 36812,
      "training_step_time": 7.629684686660767
    },
    {
      "epoch": 3.5107612609863284e-05,
      "model_forward_time": 1.739926815032959,
      "step": 36813
    },
    {
      "epoch": 3.5107612609863284e-05,
      "step": 36813,
      "training_step_time": 8.409221410751343
    },
    {
      "epoch": 3.510856628417969e-05,
      "model_forward_time": 1.7422845363616943,
      "step": 36814
    },
    {
      "epoch": 3.510856628417969e-05,
      "step": 36814,
      "training_step_time": 7.564007043838501
    },
    {
      "epoch": 3.510951995849609e-05,
      "model_forward_time": 2.926647901535034,
      "step": 36815
    },
    {
      "epoch": 3.510951995849609e-05,
      "step": 36815,
      "training_step_time": 8.196812629699707
    },
    {
      "epoch": 3.51104736328125e-05,
      "model_forward_time": 1.7452714443206787,
      "step": 36816
    },
    {
      "epoch": 3.51104736328125e-05,
      "step": 36816,
      "training_step_time": 7.735114336013794
    },
    {
      "epoch": 3.5111427307128906e-05,
      "model_forward_time": 1.7451674938201904,
      "step": 36817
    },
    {
      "epoch": 3.5111427307128906e-05,
      "step": 36817,
      "training_step_time": 8.636718273162842
    },
    {
      "epoch": 3.511238098144531e-05,
      "model_forward_time": 1.748861312866211,
      "step": 36818
    },
    {
      "epoch": 3.511238098144531e-05,
      "step": 36818,
      "training_step_time": 7.786686658859253
    },
    {
      "epoch": 3.511333465576172e-05,
      "model_forward_time": 1.7464399337768555,
      "step": 36819
    },
    {
      "epoch": 3.511333465576172e-05,
      "step": 36819,
      "training_step_time": 7.675609350204468
    },
    {
      "epoch": 3.5114288330078125e-05,
      "grad_norm": 0.1242249459028244,
      "learning_rate": 7.478155046987115e-06,
      "loss": 0.0636,
      "step": 36820
    },
    {
      "epoch": 3.5114288330078125e-05,
      "model_forward_time": 1.7491257190704346,
      "step": 36820
    },
    {
      "dynamics_loss_avg": 0.05836762972176075,
      "epoch": 3.5114288330078125e-05,
      "step": 36820
    },
    {
      "action_loss_avg": 0.002032780516310595,
      "epoch": 3.5114288330078125e-05,
      "step": 36820
    },
    {
      "epoch": 3.5114288330078125e-05,
      "step": 36820,
      "training_step_time": 7.593014478683472
    },
    {
      "epoch": 3.511524200439453e-05,
      "model_forward_time": 1.7446866035461426,
      "step": 36821
    },
    {
      "epoch": 3.511524200439453e-05,
      "step": 36821,
      "training_step_time": 7.6117706298828125
    },
    {
      "epoch": 3.511619567871094e-05,
      "model_forward_time": 1.7471797466278076,
      "step": 36822
    },
    {
      "epoch": 3.511619567871094e-05,
      "step": 36822,
      "training_step_time": 7.571759223937988
    },
    {
      "epoch": 3.5117149353027344e-05,
      "model_forward_time": 1.749692440032959,
      "step": 36823
    },
    {
      "epoch": 3.5117149353027344e-05,
      "step": 36823,
      "training_step_time": 9.646867752075195
    },
    {
      "epoch": 3.511810302734375e-05,
      "model_forward_time": 2.9291205406188965,
      "step": 36824
    },
    {
      "epoch": 3.511810302734375e-05,
      "step": 36824,
      "training_step_time": 7.702855587005615
    },
    {
      "epoch": 3.511905670166016e-05,
      "model_forward_time": 1.7506365776062012,
      "step": 36825
    },
    {
      "epoch": 3.511905670166016e-05,
      "step": 36825,
      "training_step_time": 9.019553422927856
    },
    {
      "epoch": 3.512001037597656e-05,
      "model_forward_time": 1.744645357131958,
      "step": 36826
    },
    {
      "epoch": 3.512001037597656e-05,
      "step": 36826,
      "training_step_time": 9.494343519210815
    },
    {
      "epoch": 3.512096405029297e-05,
      "model_forward_time": 1.7476658821105957,
      "step": 36827
    },
    {
      "epoch": 3.512096405029297e-05,
      "step": 36827,
      "training_step_time": 7.760893821716309
    },
    {
      "epoch": 3.512191772460938e-05,
      "model_forward_time": 1.7376575469970703,
      "step": 36828
    },
    {
      "epoch": 3.512191772460938e-05,
      "step": 36828,
      "training_step_time": 8.609193801879883
    },
    {
      "epoch": 3.512287139892578e-05,
      "model_forward_time": 1.745502233505249,
      "step": 36829
    },
    {
      "epoch": 3.512287139892578e-05,
      "step": 36829,
      "training_step_time": 7.639272212982178
    },
    {
      "epoch": 3.5123825073242186e-05,
      "grad_norm": 0.12079503387212753,
      "learning_rate": 7.476718818093945e-06,
      "loss": 0.0673,
      "step": 36830
    },
    {
      "epoch": 3.5123825073242186e-05,
      "model_forward_time": 1.3881609439849854,
      "step": 36830
    },
    {
      "dynamics_loss_avg": 0.056286286748945716,
      "epoch": 3.5123825073242186e-05,
      "step": 36830
    },
    {
      "action_loss_avg": 0.0020478854974498974,
      "epoch": 3.5123825073242186e-05,
      "step": 36830
    },
    {
      "epoch": 3.5123825073242186e-05,
      "step": 36830,
      "training_step_time": 8.150004386901855
    },
    {
      "epoch": 3.51247787475586e-05,
      "model_forward_time": 1.7514605522155762,
      "step": 36831
    },
    {
      "epoch": 3.51247787475586e-05,
      "step": 36831,
      "training_step_time": 7.596302032470703
    },
    {
      "epoch": 3.5125732421875e-05,
      "model_forward_time": 1.744842529296875,
      "step": 36832
    },
    {
      "epoch": 3.5125732421875e-05,
      "step": 36832,
      "training_step_time": 8.614328861236572
    },
    {
      "epoch": 3.5126686096191405e-05,
      "model_forward_time": 2.8797473907470703,
      "step": 36833
    },
    {
      "epoch": 3.5126686096191405e-05,
      "step": 36833,
      "training_step_time": 8.290637969970703
    },
    {
      "epoch": 3.5127639770507816e-05,
      "model_forward_time": 1.8247570991516113,
      "step": 36834
    },
    {
      "epoch": 3.5127639770507816e-05,
      "step": 36834,
      "training_step_time": 8.81184697151184
    },
    {
      "epoch": 3.512859344482422e-05,
      "model_forward_time": 1.7634491920471191,
      "step": 36835
    },
    {
      "epoch": 3.512859344482422e-05,
      "step": 36835,
      "training_step_time": 8.657746315002441
    },
    {
      "epoch": 3.5129547119140624e-05,
      "model_forward_time": 1.7467217445373535,
      "step": 36836
    },
    {
      "epoch": 3.5129547119140624e-05,
      "step": 36836,
      "training_step_time": 8.617793083190918
    },
    {
      "epoch": 3.5130500793457034e-05,
      "model_forward_time": 1.7403278350830078,
      "step": 36837
    },
    {
      "epoch": 3.5130500793457034e-05,
      "step": 36837,
      "training_step_time": 8.62953782081604
    },
    {
      "epoch": 3.513145446777344e-05,
      "model_forward_time": 1.7505686283111572,
      "step": 36838
    },
    {
      "epoch": 3.513145446777344e-05,
      "step": 36838,
      "training_step_time": 7.5795111656188965
    },
    {
      "epoch": 3.513240814208984e-05,
      "model_forward_time": 1.7446019649505615,
      "step": 36839
    },
    {
      "epoch": 3.513240814208984e-05,
      "step": 36839,
      "training_step_time": 7.476570129394531
    },
    {
      "epoch": 3.513336181640625e-05,
      "grad_norm": 0.15321174263954163,
      "learning_rate": 7.4752823183505275e-06,
      "loss": 0.0596,
      "step": 36840
    },
    {
      "epoch": 3.513336181640625e-05,
      "model_forward_time": 1.7294082641601562,
      "step": 36840
    },
    {
      "dynamics_loss_avg": 0.05181329194456339,
      "epoch": 3.513336181640625e-05,
      "step": 36840
    },
    {
      "action_loss_avg": 0.0022171560151036827,
      "epoch": 3.513336181640625e-05,
      "step": 36840
    },
    {
      "epoch": 3.513336181640625e-05,
      "step": 36840,
      "training_step_time": 8.459160804748535
    },
    {
      "epoch": 3.513431549072266e-05,
      "model_forward_time": 1.7454862594604492,
      "step": 36841
    },
    {
      "epoch": 3.513431549072266e-05,
      "step": 36841,
      "training_step_time": 7.522284984588623
    },
    {
      "epoch": 3.513526916503906e-05,
      "model_forward_time": 1.7452642917633057,
      "step": 36842
    },
    {
      "epoch": 3.513526916503906e-05,
      "step": 36842,
      "training_step_time": 8.649610042572021
    },
    {
      "epoch": 3.513622283935547e-05,
      "model_forward_time": 1.833752155303955,
      "step": 36843
    },
    {
      "epoch": 3.513622283935547e-05,
      "step": 36843,
      "training_step_time": 8.2732994556427
    },
    {
      "epoch": 3.5137176513671876e-05,
      "model_forward_time": 1.7518188953399658,
      "step": 36844
    },
    {
      "epoch": 3.5137176513671876e-05,
      "step": 36844,
      "training_step_time": 8.75840163230896
    },
    {
      "epoch": 3.513813018798828e-05,
      "model_forward_time": 1.7482564449310303,
      "step": 36845
    },
    {
      "epoch": 3.513813018798828e-05,
      "step": 36845,
      "training_step_time": 8.989432573318481
    },
    {
      "epoch": 3.5139083862304684e-05,
      "model_forward_time": 1.7486209869384766,
      "step": 36846
    },
    {
      "epoch": 3.5139083862304684e-05,
      "step": 36846,
      "training_step_time": 8.281151294708252
    },
    {
      "epoch": 3.5140037536621095e-05,
      "model_forward_time": 1.7447478771209717,
      "step": 36847
    },
    {
      "epoch": 3.5140037536621095e-05,
      "step": 36847,
      "training_step_time": 8.203075170516968
    },
    {
      "epoch": 3.51409912109375e-05,
      "model_forward_time": 1.7425203323364258,
      "step": 36848
    },
    {
      "epoch": 3.51409912109375e-05,
      "step": 36848,
      "training_step_time": 8.045114278793335
    },
    {
      "epoch": 3.51419448852539e-05,
      "model_forward_time": 1.7470519542694092,
      "step": 36849
    },
    {
      "epoch": 3.51419448852539e-05,
      "step": 36849,
      "training_step_time": 8.433019161224365
    },
    {
      "epoch": 3.5142898559570314e-05,
      "grad_norm": 0.16640499234199524,
      "learning_rate": 7.4738455479139575e-06,
      "loss": 0.062,
      "step": 36850
    },
    {
      "epoch": 3.5142898559570314e-05,
      "model_forward_time": 1.7545225620269775,
      "step": 36850
    },
    {
      "dynamics_loss_avg": 0.038822395540773866,
      "epoch": 3.5142898559570314e-05,
      "step": 36850
    },
    {
      "action_loss_avg": 0.0019677068921737373,
      "epoch": 3.5142898559570314e-05,
      "step": 36850
    },
    {
      "epoch": 3.5142898559570314e-05,
      "step": 36850,
      "training_step_time": 9.782247304916382
    },
    {
      "epoch": 3.514385223388672e-05,
      "model_forward_time": 1.7468492984771729,
      "step": 36851
    },
    {
      "epoch": 3.514385223388672e-05,
      "step": 36851,
      "training_step_time": 7.600636005401611
    },
    {
      "epoch": 3.514480590820312e-05,
      "model_forward_time": 2.9248671531677246,
      "step": 36852
    },
    {
      "epoch": 3.514480590820312e-05,
      "step": 36852,
      "training_step_time": 8.742648601531982
    },
    {
      "epoch": 3.514575958251953e-05,
      "model_forward_time": 1.7455847263336182,
      "step": 36853
    },
    {
      "epoch": 3.514575958251953e-05,
      "step": 36853,
      "training_step_time": 7.708287239074707
    },
    {
      "epoch": 3.514671325683594e-05,
      "model_forward_time": 1.7479186058044434,
      "step": 36854
    },
    {
      "epoch": 3.514671325683594e-05,
      "step": 36854,
      "training_step_time": 8.402011632919312
    },
    {
      "epoch": 3.514766693115234e-05,
      "model_forward_time": 1.7441725730895996,
      "step": 36855
    },
    {
      "epoch": 3.514766693115234e-05,
      "step": 36855,
      "training_step_time": 7.7192347049713135
    },
    {
      "epoch": 3.514862060546875e-05,
      "model_forward_time": 1.7807939052581787,
      "step": 36856
    },
    {
      "epoch": 3.514862060546875e-05,
      "step": 36856,
      "training_step_time": 7.602906703948975
    },
    {
      "epoch": 3.5149574279785156e-05,
      "model_forward_time": 1.7471799850463867,
      "step": 36857
    },
    {
      "epoch": 3.5149574279785156e-05,
      "step": 36857,
      "training_step_time": 7.535100221633911
    },
    {
      "epoch": 3.515052795410156e-05,
      "model_forward_time": 1.745638370513916,
      "step": 36858
    },
    {
      "epoch": 3.515052795410156e-05,
      "step": 36858,
      "training_step_time": 7.636734962463379
    },
    {
      "epoch": 3.515148162841797e-05,
      "model_forward_time": 1.74188232421875,
      "step": 36859
    },
    {
      "epoch": 3.515148162841797e-05,
      "step": 36859,
      "training_step_time": 7.56186580657959
    },
    {
      "epoch": 3.5152435302734374e-05,
      "grad_norm": 0.21367350220680237,
      "learning_rate": 7.472408506941358e-06,
      "loss": 0.0589,
      "step": 36860
    },
    {
      "epoch": 3.5152435302734374e-05,
      "model_forward_time": 1.7185461521148682,
      "step": 36860
    },
    {
      "dynamics_loss_avg": 0.05707818837836385,
      "epoch": 3.5152435302734374e-05,
      "step": 36860
    },
    {
      "action_loss_avg": 0.0021663123043254018,
      "epoch": 3.5152435302734374e-05,
      "step": 36860
    },
    {
      "epoch": 3.5152435302734374e-05,
      "step": 36860,
      "training_step_time": 8.387452840805054
    },
    {
      "epoch": 3.515338897705078e-05,
      "model_forward_time": 2.3120532035827637,
      "step": 36861
    },
    {
      "epoch": 3.515338897705078e-05,
      "step": 36861,
      "training_step_time": 8.50373363494873
    },
    {
      "epoch": 3.515434265136719e-05,
      "model_forward_time": 1.740633249282837,
      "step": 36862
    },
    {
      "epoch": 3.515434265136719e-05,
      "step": 36862,
      "training_step_time": 7.701219081878662
    },
    {
      "epoch": 3.515529632568359e-05,
      "model_forward_time": 1.7383973598480225,
      "step": 36863
    },
    {
      "epoch": 3.515529632568359e-05,
      "step": 36863,
      "training_step_time": 9.736284255981445
    },
    {
      "epoch": 3.515625e-05,
      "model_forward_time": 1.7487072944641113,
      "step": 36864
    },
    {
      "epoch": 3.515625e-05,
      "step": 36864,
      "training_step_time": 7.712727785110474
    },
    {
      "epoch": 3.515720367431641e-05,
      "model_forward_time": 1.7443230152130127,
      "step": 36865
    },
    {
      "epoch": 3.515720367431641e-05,
      "step": 36865,
      "training_step_time": 9.473587274551392
    },
    {
      "epoch": 3.515815734863281e-05,
      "model_forward_time": 1.7442021369934082,
      "step": 36866
    },
    {
      "epoch": 3.515815734863281e-05,
      "step": 36866,
      "training_step_time": 7.5602500438690186
    },
    {
      "epoch": 3.5159111022949216e-05,
      "model_forward_time": 1.7428359985351562,
      "step": 36867
    },
    {
      "epoch": 3.5159111022949216e-05,
      "step": 36867,
      "training_step_time": 7.633432626724243
    },
    {
      "epoch": 3.516006469726563e-05,
      "model_forward_time": 1.7428627014160156,
      "step": 36868
    },
    {
      "epoch": 3.516006469726563e-05,
      "step": 36868,
      "training_step_time": 7.625957250595093
    },
    {
      "epoch": 3.516101837158203e-05,
      "model_forward_time": 1.7465007305145264,
      "step": 36869
    },
    {
      "epoch": 3.516101837158203e-05,
      "step": 36869,
      "training_step_time": 7.605352163314819
    },
    {
      "epoch": 3.5161972045898435e-05,
      "grad_norm": 0.13607531785964966,
      "learning_rate": 7.470971195589881e-06,
      "loss": 0.0603,
      "step": 36870
    },
    {
      "epoch": 3.5161972045898435e-05,
      "model_forward_time": 2.8514482975006104,
      "step": 36870
    },
    {
      "dynamics_loss_avg": 0.06778054907917977,
      "epoch": 3.5161972045898435e-05,
      "step": 36870
    },
    {
      "action_loss_avg": 0.0020507485955022276,
      "epoch": 3.5161972045898435e-05,
      "step": 36870
    },
    {
      "epoch": 3.5161972045898435e-05,
      "step": 36870,
      "training_step_time": 7.6241865158081055
    },
    {
      "epoch": 3.5162925720214846e-05,
      "model_forward_time": 1.8310554027557373,
      "step": 36871
    },
    {
      "epoch": 3.5162925720214846e-05,
      "step": 36871,
      "training_step_time": 9.749365329742432
    },
    {
      "epoch": 3.516387939453125e-05,
      "model_forward_time": 1.752122163772583,
      "step": 36872
    },
    {
      "epoch": 3.516387939453125e-05,
      "step": 36872,
      "training_step_time": 8.345709800720215
    },
    {
      "epoch": 3.5164833068847654e-05,
      "model_forward_time": 1.7797727584838867,
      "step": 36873
    },
    {
      "epoch": 3.5164833068847654e-05,
      "step": 36873,
      "training_step_time": 9.066267013549805
    },
    {
      "epoch": 3.5165786743164065e-05,
      "model_forward_time": 1.7446010112762451,
      "step": 36874
    },
    {
      "epoch": 3.5165786743164065e-05,
      "step": 36874,
      "training_step_time": 7.72020149230957
    },
    {
      "epoch": 3.516674041748047e-05,
      "model_forward_time": 1.7369225025177002,
      "step": 36875
    },
    {
      "epoch": 3.516674041748047e-05,
      "step": 36875,
      "training_step_time": 7.564056158065796
    },
    {
      "epoch": 3.516769409179687e-05,
      "model_forward_time": 1.7510638236999512,
      "step": 36876
    },
    {
      "epoch": 3.516769409179687e-05,
      "step": 36876,
      "training_step_time": 7.566509008407593
    },
    {
      "epoch": 3.5168647766113284e-05,
      "model_forward_time": 1.7497713565826416,
      "step": 36877
    },
    {
      "epoch": 3.5168647766113284e-05,
      "step": 36877,
      "training_step_time": 8.604338645935059
    },
    {
      "epoch": 3.516960144042969e-05,
      "model_forward_time": 1.750821828842163,
      "step": 36878
    },
    {
      "epoch": 3.516960144042969e-05,
      "step": 36878,
      "training_step_time": 7.65796160697937
    },
    {
      "epoch": 3.517055511474609e-05,
      "model_forward_time": 2.8751962184906006,
      "step": 36879
    },
    {
      "epoch": 3.517055511474609e-05,
      "step": 36879,
      "training_step_time": 8.376277208328247
    },
    {
      "epoch": 3.51715087890625e-05,
      "grad_norm": 0.13666929304599762,
      "learning_rate": 7.46953361401671e-06,
      "loss": 0.0636,
      "step": 36880
    },
    {
      "epoch": 3.51715087890625e-05,
      "model_forward_time": 1.8044524192810059,
      "step": 36880
    },
    {
      "dynamics_loss_avg": 0.05809939801692963,
      "epoch": 3.51715087890625e-05,
      "step": 36880
    },
    {
      "action_loss_avg": 0.0016687703202478588,
      "epoch": 3.51715087890625e-05,
      "step": 36880
    },
    {
      "epoch": 3.51715087890625e-05,
      "step": 36880,
      "training_step_time": 7.811200141906738
    },
    {
      "epoch": 3.5172462463378906e-05,
      "model_forward_time": 1.747849941253662,
      "step": 36881
    },
    {
      "epoch": 3.5172462463378906e-05,
      "step": 36881,
      "training_step_time": 8.864959239959717
    },
    {
      "epoch": 3.517341613769531e-05,
      "model_forward_time": 1.7441682815551758,
      "step": 36882
    },
    {
      "epoch": 3.517341613769531e-05,
      "step": 36882,
      "training_step_time": 7.963187217712402
    },
    {
      "epoch": 3.517436981201172e-05,
      "model_forward_time": 1.744889497756958,
      "step": 36883
    },
    {
      "epoch": 3.517436981201172e-05,
      "step": 36883,
      "training_step_time": 7.71942925453186
    },
    {
      "epoch": 3.5175323486328125e-05,
      "model_forward_time": 1.7457435131072998,
      "step": 36884
    },
    {
      "epoch": 3.5175323486328125e-05,
      "step": 36884,
      "training_step_time": 7.672463417053223
    },
    {
      "epoch": 3.517627716064453e-05,
      "model_forward_time": 1.7459735870361328,
      "step": 36885
    },
    {
      "epoch": 3.517627716064453e-05,
      "step": 36885,
      "training_step_time": 7.587885618209839
    },
    {
      "epoch": 3.517723083496094e-05,
      "model_forward_time": 1.7385578155517578,
      "step": 36886
    },
    {
      "epoch": 3.517723083496094e-05,
      "step": 36886,
      "training_step_time": 7.643786907196045
    },
    {
      "epoch": 3.5178184509277344e-05,
      "model_forward_time": 1.7493336200714111,
      "step": 36887
    },
    {
      "epoch": 3.5178184509277344e-05,
      "step": 36887,
      "training_step_time": 7.555999755859375
    },
    {
      "epoch": 3.517913818359375e-05,
      "model_forward_time": 2.8519952297210693,
      "step": 36888
    },
    {
      "epoch": 3.517913818359375e-05,
      "step": 36888,
      "training_step_time": 8.768638372421265
    },
    {
      "epoch": 3.518009185791016e-05,
      "model_forward_time": 1.8049356937408447,
      "step": 36889
    },
    {
      "epoch": 3.518009185791016e-05,
      "step": 36889,
      "training_step_time": 7.784392356872559
    },
    {
      "epoch": 3.518104553222656e-05,
      "grad_norm": 0.12008114159107208,
      "learning_rate": 7.468095762379055e-06,
      "loss": 0.0638,
      "step": 36890
    },
    {
      "epoch": 3.518104553222656e-05,
      "model_forward_time": 1.7484180927276611,
      "step": 36890
    },
    {
      "dynamics_loss_avg": 0.07764324340969324,
      "epoch": 3.518104553222656e-05,
      "step": 36890
    },
    {
      "action_loss_avg": 0.002522213725023903,
      "epoch": 3.518104553222656e-05,
      "step": 36890
    },
    {
      "epoch": 3.518104553222656e-05,
      "step": 36890,
      "training_step_time": 7.689493417739868
    },
    {
      "epoch": 3.518199920654297e-05,
      "model_forward_time": 1.7449607849121094,
      "step": 36891
    },
    {
      "epoch": 3.518199920654297e-05,
      "step": 36891,
      "training_step_time": 8.48262643814087
    },
    {
      "epoch": 3.518295288085938e-05,
      "model_forward_time": 1.7506442070007324,
      "step": 36892
    },
    {
      "epoch": 3.518295288085938e-05,
      "step": 36892,
      "training_step_time": 7.719054698944092
    },
    {
      "epoch": 3.518390655517578e-05,
      "model_forward_time": 1.7505173683166504,
      "step": 36893
    },
    {
      "epoch": 3.518390655517578e-05,
      "step": 36893,
      "training_step_time": 8.65667176246643
    },
    {
      "epoch": 3.5184860229492186e-05,
      "model_forward_time": 1.7456302642822266,
      "step": 36894
    },
    {
      "epoch": 3.5184860229492186e-05,
      "step": 36894,
      "training_step_time": 8.19867992401123
    },
    {
      "epoch": 3.51858139038086e-05,
      "model_forward_time": 1.750281810760498,
      "step": 36895
    },
    {
      "epoch": 3.51858139038086e-05,
      "step": 36895,
      "training_step_time": 7.663279056549072
    },
    {
      "epoch": 3.5186767578125e-05,
      "model_forward_time": 1.7485246658325195,
      "step": 36896
    },
    {
      "epoch": 3.5186767578125e-05,
      "step": 36896,
      "training_step_time": 7.490516424179077
    },
    {
      "epoch": 3.5187721252441405e-05,
      "model_forward_time": 1.752950668334961,
      "step": 36897
    },
    {
      "epoch": 3.5187721252441405e-05,
      "step": 36897,
      "training_step_time": 8.680270433425903
    },
    {
      "epoch": 3.5188674926757816e-05,
      "model_forward_time": 1.769709825515747,
      "step": 36898
    },
    {
      "epoch": 3.5188674926757816e-05,
      "step": 36898,
      "training_step_time": 8.273751974105835
    },
    {
      "epoch": 3.518962860107422e-05,
      "model_forward_time": 1.7412071228027344,
      "step": 36899
    },
    {
      "epoch": 3.518962860107422e-05,
      "step": 36899,
      "training_step_time": 8.445344686508179
    },
    {
      "epoch": 3.5190582275390624e-05,
      "grad_norm": 0.10419727116823196,
      "learning_rate": 7.4666576408341586e-06,
      "loss": 0.0635,
      "step": 36900
    },
    {
      "epoch": 3.5190582275390624e-05,
      "model_forward_time": 1.7573633193969727,
      "step": 36900
    },
    {
      "dynamics_loss_avg": 0.07645074129104615,
      "epoch": 3.5190582275390624e-05,
      "step": 36900
    },
    {
      "action_loss_avg": 0.0025030273653101178,
      "epoch": 3.5190582275390624e-05,
      "step": 36900
    },
    {
      "epoch": 3.5190582275390624e-05,
      "step": 36900,
      "training_step_time": 7.9156999588012695
    },
    {
      "epoch": 3.5191535949707034e-05,
      "model_forward_time": 1.7344486713409424,
      "step": 36901
    },
    {
      "epoch": 3.5191535949707034e-05,
      "step": 36901,
      "training_step_time": 7.7381322383880615
    },
    {
      "epoch": 3.519248962402344e-05,
      "model_forward_time": 1.739426612854004,
      "step": 36902
    },
    {
      "epoch": 3.519248962402344e-05,
      "step": 36902,
      "training_step_time": 8.665320634841919
    },
    {
      "epoch": 3.519344329833984e-05,
      "model_forward_time": 1.7395517826080322,
      "step": 36903
    },
    {
      "epoch": 3.519344329833984e-05,
      "step": 36903,
      "training_step_time": 7.562662124633789
    },
    {
      "epoch": 3.519439697265625e-05,
      "model_forward_time": 1.7291443347930908,
      "step": 36904
    },
    {
      "epoch": 3.519439697265625e-05,
      "step": 36904,
      "training_step_time": 7.718555688858032
    },
    {
      "epoch": 3.519535064697266e-05,
      "model_forward_time": 1.7280983924865723,
      "step": 36905
    },
    {
      "epoch": 3.519535064697266e-05,
      "step": 36905,
      "training_step_time": 7.688654661178589
    },
    {
      "epoch": 3.519630432128906e-05,
      "model_forward_time": 1.7309279441833496,
      "step": 36906
    },
    {
      "epoch": 3.519630432128906e-05,
      "step": 36906,
      "training_step_time": 7.588742971420288
    },
    {
      "epoch": 3.5197257995605465e-05,
      "model_forward_time": 2.8475735187530518,
      "step": 36907
    },
    {
      "epoch": 3.5197257995605465e-05,
      "step": 36907,
      "training_step_time": 7.661571502685547
    },
    {
      "epoch": 3.5198211669921876e-05,
      "model_forward_time": 1.7930850982666016,
      "step": 36908
    },
    {
      "epoch": 3.5198211669921876e-05,
      "step": 36908,
      "training_step_time": 8.72836971282959
    },
    {
      "epoch": 3.519916534423828e-05,
      "model_forward_time": 1.759044885635376,
      "step": 36909
    },
    {
      "epoch": 3.519916534423828e-05,
      "step": 36909,
      "training_step_time": 7.858525037765503
    },
    {
      "epoch": 3.5200119018554684e-05,
      "grad_norm": 0.2032288759946823,
      "learning_rate": 7.465219249539291e-06,
      "loss": 0.0611,
      "step": 36910
    },
    {
      "epoch": 3.5200119018554684e-05,
      "model_forward_time": 1.71750807762146,
      "step": 36910
    },
    {
      "dynamics_loss_avg": 0.06081632897257805,
      "epoch": 3.5200119018554684e-05,
      "step": 36910
    },
    {
      "action_loss_avg": 0.0018947023956570775,
      "epoch": 3.5200119018554684e-05,
      "step": 36910
    },
    {
      "epoch": 3.5200119018554684e-05,
      "step": 36910,
      "training_step_time": 7.705235004425049
    },
    {
      "epoch": 3.5201072692871095e-05,
      "model_forward_time": 1.7330138683319092,
      "step": 36911
    },
    {
      "epoch": 3.5201072692871095e-05,
      "step": 36911,
      "training_step_time": 8.471528053283691
    },
    {
      "epoch": 3.52020263671875e-05,
      "model_forward_time": 1.7346971035003662,
      "step": 36912
    },
    {
      "epoch": 3.52020263671875e-05,
      "step": 36912,
      "training_step_time": 8.397860288619995
    },
    {
      "epoch": 3.52029800415039e-05,
      "model_forward_time": 1.3636610507965088,
      "step": 36913
    },
    {
      "epoch": 3.52029800415039e-05,
      "step": 36913,
      "training_step_time": 7.726459264755249
    },
    {
      "epoch": 3.5203933715820314e-05,
      "model_forward_time": 1.730475902557373,
      "step": 36914
    },
    {
      "epoch": 3.5203933715820314e-05,
      "step": 36914,
      "training_step_time": 7.622133255004883
    },
    {
      "epoch": 3.520488739013672e-05,
      "model_forward_time": 1.7388932704925537,
      "step": 36915
    },
    {
      "epoch": 3.520488739013672e-05,
      "step": 36915,
      "training_step_time": 8.455978870391846
    },
    {
      "epoch": 3.520584106445312e-05,
      "model_forward_time": 2.8257315158843994,
      "step": 36916
    },
    {
      "epoch": 3.520584106445312e-05,
      "step": 36916,
      "training_step_time": 8.657573223114014
    },
    {
      "epoch": 3.520679473876953e-05,
      "model_forward_time": 1.8063554763793945,
      "step": 36917
    },
    {
      "epoch": 3.520679473876953e-05,
      "step": 36917,
      "training_step_time": 7.740914821624756
    },
    {
      "epoch": 3.520774841308594e-05,
      "model_forward_time": 1.7345359325408936,
      "step": 36918
    },
    {
      "epoch": 3.520774841308594e-05,
      "step": 36918,
      "training_step_time": 7.912568092346191
    },
    {
      "epoch": 3.520870208740234e-05,
      "model_forward_time": 1.7245159149169922,
      "step": 36919
    },
    {
      "epoch": 3.520870208740234e-05,
      "step": 36919,
      "training_step_time": 9.779744148254395
    },
    {
      "epoch": 3.520965576171875e-05,
      "grad_norm": 0.15124858915805817,
      "learning_rate": 7.463780588651752e-06,
      "loss": 0.059,
      "step": 36920
    },
    {
      "epoch": 3.520965576171875e-05,
      "model_forward_time": 1.729907512664795,
      "step": 36920
    },
    {
      "dynamics_loss_avg": 0.05971948690712452,
      "epoch": 3.520965576171875e-05,
      "step": 36920
    },
    {
      "action_loss_avg": 0.001896556850988418,
      "epoch": 3.520965576171875e-05,
      "step": 36920
    },
    {
      "epoch": 3.520965576171875e-05,
      "step": 36920,
      "training_step_time": 8.198949813842773
    },
    {
      "epoch": 3.5210609436035156e-05,
      "model_forward_time": 1.3611557483673096,
      "step": 36921
    },
    {
      "epoch": 3.5210609436035156e-05,
      "step": 36921,
      "training_step_time": 7.579778671264648
    },
    {
      "epoch": 3.521156311035156e-05,
      "model_forward_time": 1.7373013496398926,
      "step": 36922
    },
    {
      "epoch": 3.521156311035156e-05,
      "step": 36922,
      "training_step_time": 7.7241926193237305
    },
    {
      "epoch": 3.521251678466797e-05,
      "model_forward_time": 1.7367174625396729,
      "step": 36923
    },
    {
      "epoch": 3.521251678466797e-05,
      "step": 36923,
      "training_step_time": 8.576773405075073
    },
    {
      "epoch": 3.5213470458984375e-05,
      "model_forward_time": 1.7443745136260986,
      "step": 36924
    },
    {
      "epoch": 3.5213470458984375e-05,
      "step": 36924,
      "training_step_time": 8.290081977844238
    },
    {
      "epoch": 3.521442413330078e-05,
      "model_forward_time": 2.8152809143066406,
      "step": 36925
    },
    {
      "epoch": 3.521442413330078e-05,
      "step": 36925,
      "training_step_time": 7.631291627883911
    },
    {
      "epoch": 3.521537780761719e-05,
      "model_forward_time": 1.810530185699463,
      "step": 36926
    },
    {
      "epoch": 3.521537780761719e-05,
      "step": 36926,
      "training_step_time": 8.743527889251709
    },
    {
      "epoch": 3.521633148193359e-05,
      "model_forward_time": 1.7453727722167969,
      "step": 36927
    },
    {
      "epoch": 3.521633148193359e-05,
      "step": 36927,
      "training_step_time": 8.42399787902832
    },
    {
      "epoch": 3.521728515625e-05,
      "model_forward_time": 1.7299144268035889,
      "step": 36928
    },
    {
      "epoch": 3.521728515625e-05,
      "step": 36928,
      "training_step_time": 8.16812515258789
    },
    {
      "epoch": 3.521823883056641e-05,
      "model_forward_time": 1.7415127754211426,
      "step": 36929
    },
    {
      "epoch": 3.521823883056641e-05,
      "step": 36929,
      "training_step_time": 8.564337015151978
    },
    {
      "epoch": 3.521919250488281e-05,
      "grad_norm": 0.12898024916648865,
      "learning_rate": 7.462341658328873e-06,
      "loss": 0.0609,
      "step": 36930
    },
    {
      "epoch": 3.521919250488281e-05,
      "model_forward_time": 1.72585129737854,
      "step": 36930
    },
    {
      "dynamics_loss_avg": 0.04872678089886904,
      "epoch": 3.521919250488281e-05,
      "step": 36930
    },
    {
      "action_loss_avg": 0.0017364879429806023,
      "epoch": 3.521919250488281e-05,
      "step": 36930
    },
    {
      "epoch": 3.521919250488281e-05,
      "step": 36930,
      "training_step_time": 7.592870473861694
    },
    {
      "epoch": 3.5220146179199216e-05,
      "model_forward_time": 1.7401797771453857,
      "step": 36931
    },
    {
      "epoch": 3.5220146179199216e-05,
      "step": 36931,
      "training_step_time": 8.332894563674927
    },
    {
      "epoch": 3.522109985351563e-05,
      "model_forward_time": 1.7440457344055176,
      "step": 36932
    },
    {
      "epoch": 3.522109985351563e-05,
      "step": 36932,
      "training_step_time": 9.396751165390015
    },
    {
      "epoch": 3.522205352783203e-05,
      "model_forward_time": 1.7304956912994385,
      "step": 36933
    },
    {
      "epoch": 3.522205352783203e-05,
      "step": 36933,
      "training_step_time": 8.306059122085571
    },
    {
      "epoch": 3.5223007202148435e-05,
      "model_forward_time": 2.804758310317993,
      "step": 36934
    },
    {
      "epoch": 3.5223007202148435e-05,
      "step": 36934,
      "training_step_time": 8.494494438171387
    },
    {
      "epoch": 3.5223960876464846e-05,
      "model_forward_time": 1.8037238121032715,
      "step": 36935
    },
    {
      "epoch": 3.5223960876464846e-05,
      "step": 36935,
      "training_step_time": 7.735769510269165
    },
    {
      "epoch": 3.522491455078125e-05,
      "model_forward_time": 1.7516326904296875,
      "step": 36936
    },
    {
      "epoch": 3.522491455078125e-05,
      "step": 36936,
      "training_step_time": 9.453543424606323
    },
    {
      "epoch": 3.5225868225097654e-05,
      "model_forward_time": 1.734863042831421,
      "step": 36937
    },
    {
      "epoch": 3.5225868225097654e-05,
      "step": 36937,
      "training_step_time": 8.228610754013062
    },
    {
      "epoch": 3.5226821899414065e-05,
      "model_forward_time": 1.7469797134399414,
      "step": 36938
    },
    {
      "epoch": 3.5226821899414065e-05,
      "step": 36938,
      "training_step_time": 7.809523582458496
    },
    {
      "epoch": 3.522777557373047e-05,
      "model_forward_time": 1.734363317489624,
      "step": 36939
    },
    {
      "epoch": 3.522777557373047e-05,
      "step": 36939,
      "training_step_time": 8.68497920036316
    },
    {
      "epoch": 3.522872924804687e-05,
      "grad_norm": 0.14718452095985413,
      "learning_rate": 7.460902458728012e-06,
      "loss": 0.0586,
      "step": 36940
    },
    {
      "epoch": 3.522872924804687e-05,
      "model_forward_time": 1.7193641662597656,
      "step": 36940
    },
    {
      "dynamics_loss_avg": 0.07014801539480686,
      "epoch": 3.522872924804687e-05,
      "step": 36940
    },
    {
      "action_loss_avg": 0.002848948130849749,
      "epoch": 3.522872924804687e-05,
      "step": 36940
    },
    {
      "epoch": 3.522872924804687e-05,
      "step": 36940,
      "training_step_time": 7.72457480430603
    },
    {
      "epoch": 3.5229682922363284e-05,
      "model_forward_time": 1.7373194694519043,
      "step": 36941
    },
    {
      "epoch": 3.5229682922363284e-05,
      "step": 36941,
      "training_step_time": 7.696850061416626
    },
    {
      "epoch": 3.523063659667969e-05,
      "model_forward_time": 1.7378466129302979,
      "step": 36942
    },
    {
      "epoch": 3.523063659667969e-05,
      "step": 36942,
      "training_step_time": 7.597445487976074
    },
    {
      "epoch": 3.523159027099609e-05,
      "model_forward_time": 2.809138298034668,
      "step": 36943
    },
    {
      "epoch": 3.523159027099609e-05,
      "step": 36943,
      "training_step_time": 7.668515682220459
    },
    {
      "epoch": 3.52325439453125e-05,
      "model_forward_time": 1.754533052444458,
      "step": 36944
    },
    {
      "epoch": 3.52325439453125e-05,
      "step": 36944,
      "training_step_time": 10.36759066581726
    },
    {
      "epoch": 3.5233497619628907e-05,
      "model_forward_time": 1.8018121719360352,
      "step": 36945
    },
    {
      "epoch": 3.5233497619628907e-05,
      "step": 36945,
      "training_step_time": 7.790204048156738
    },
    {
      "epoch": 3.523445129394531e-05,
      "model_forward_time": 1.7295277118682861,
      "step": 36946
    },
    {
      "epoch": 3.523445129394531e-05,
      "step": 36946,
      "training_step_time": 7.702576160430908
    },
    {
      "epoch": 3.523540496826172e-05,
      "model_forward_time": 1.7326204776763916,
      "step": 36947
    },
    {
      "epoch": 3.523540496826172e-05,
      "step": 36947,
      "training_step_time": 8.802196502685547
    },
    {
      "epoch": 3.5236358642578125e-05,
      "model_forward_time": 1.7381510734558105,
      "step": 36948
    },
    {
      "epoch": 3.5236358642578125e-05,
      "step": 36948,
      "training_step_time": 7.666743278503418
    },
    {
      "epoch": 3.523731231689453e-05,
      "model_forward_time": 1.7286784648895264,
      "step": 36949
    },
    {
      "epoch": 3.523731231689453e-05,
      "step": 36949,
      "training_step_time": 7.578718900680542
    },
    {
      "epoch": 3.523826599121094e-05,
      "grad_norm": 0.1804930567741394,
      "learning_rate": 7.459462990006559e-06,
      "loss": 0.0607,
      "step": 36950
    },
    {
      "epoch": 3.523826599121094e-05,
      "model_forward_time": 1.7270402908325195,
      "step": 36950
    },
    {
      "dynamics_loss_avg": 0.07171032987535,
      "epoch": 3.523826599121094e-05,
      "step": 36950
    },
    {
      "action_loss_avg": 0.0037748958566226065,
      "epoch": 3.523826599121094e-05,
      "step": 36950
    },
    {
      "epoch": 3.523826599121094e-05,
      "step": 36950,
      "training_step_time": 7.715461730957031
    },
    {
      "epoch": 3.5239219665527344e-05,
      "model_forward_time": 1.7321019172668457,
      "step": 36951
    },
    {
      "epoch": 3.5239219665527344e-05,
      "step": 36951,
      "training_step_time": 8.849357843399048
    },
    {
      "epoch": 3.524017333984375e-05,
      "model_forward_time": 1.7425916194915771,
      "step": 36952
    },
    {
      "epoch": 3.524017333984375e-05,
      "step": 36952,
      "training_step_time": 8.343003749847412
    },
    {
      "epoch": 3.524112701416016e-05,
      "model_forward_time": 2.8217804431915283,
      "step": 36953
    },
    {
      "epoch": 3.524112701416016e-05,
      "step": 36953,
      "training_step_time": 8.982764482498169
    },
    {
      "epoch": 3.524208068847656e-05,
      "model_forward_time": 1.7340717315673828,
      "step": 36954
    },
    {
      "epoch": 3.524208068847656e-05,
      "step": 36954,
      "training_step_time": 9.817878007888794
    },
    {
      "epoch": 3.524303436279297e-05,
      "model_forward_time": 1.7403254508972168,
      "step": 36955
    },
    {
      "epoch": 3.524303436279297e-05,
      "step": 36955,
      "training_step_time": 8.491299152374268
    },
    {
      "epoch": 3.524398803710938e-05,
      "model_forward_time": 1.734189748764038,
      "step": 36956
    },
    {
      "epoch": 3.524398803710938e-05,
      "step": 36956,
      "training_step_time": 8.863774538040161
    },
    {
      "epoch": 3.524494171142578e-05,
      "model_forward_time": 1.7359063625335693,
      "step": 36957
    },
    {
      "epoch": 3.524494171142578e-05,
      "step": 36957,
      "training_step_time": 7.7028045654296875
    },
    {
      "epoch": 3.5245895385742186e-05,
      "model_forward_time": 1.749269723892212,
      "step": 36958
    },
    {
      "epoch": 3.5245895385742186e-05,
      "step": 36958,
      "training_step_time": 7.63493800163269
    },
    {
      "epoch": 3.52468490600586e-05,
      "model_forward_time": 1.7323298454284668,
      "step": 36959
    },
    {
      "epoch": 3.52468490600586e-05,
      "step": 36959,
      "training_step_time": 8.738863229751587
    },
    {
      "epoch": 3.5247802734375e-05,
      "grad_norm": 0.16002792119979858,
      "learning_rate": 7.458023252321929e-06,
      "loss": 0.0613,
      "step": 36960
    },
    {
      "epoch": 3.5247802734375e-05,
      "model_forward_time": 1.7272045612335205,
      "step": 36960
    },
    {
      "dynamics_loss_avg": 0.045282449666410686,
      "epoch": 3.5247802734375e-05,
      "step": 36960
    },
    {
      "action_loss_avg": 0.0015059390745591373,
      "epoch": 3.5247802734375e-05,
      "step": 36960
    },
    {
      "epoch": 3.5247802734375e-05,
      "step": 36960,
      "training_step_time": 7.583927631378174
    },
    {
      "epoch": 3.5248756408691405e-05,
      "model_forward_time": 1.7424211502075195,
      "step": 36961
    },
    {
      "epoch": 3.5248756408691405e-05,
      "step": 36961,
      "training_step_time": 7.647857189178467
    },
    {
      "epoch": 3.5249710083007816e-05,
      "model_forward_time": 2.8710999488830566,
      "step": 36962
    },
    {
      "epoch": 3.5249710083007816e-05,
      "step": 36962,
      "training_step_time": 8.801663637161255
    },
    {
      "epoch": 3.525066375732422e-05,
      "model_forward_time": 1.8140864372253418,
      "step": 36963
    },
    {
      "epoch": 3.525066375732422e-05,
      "step": 36963,
      "training_step_time": 8.57033085823059
    },
    {
      "epoch": 3.5251617431640624e-05,
      "model_forward_time": 1.7419183254241943,
      "step": 36964
    },
    {
      "epoch": 3.5251617431640624e-05,
      "step": 36964,
      "training_step_time": 7.710052013397217
    },
    {
      "epoch": 3.5252571105957035e-05,
      "model_forward_time": 1.7633917331695557,
      "step": 36965
    },
    {
      "epoch": 3.5252571105957035e-05,
      "step": 36965,
      "training_step_time": 7.753210783004761
    },
    {
      "epoch": 3.525352478027344e-05,
      "model_forward_time": 1.7444417476654053,
      "step": 36966
    },
    {
      "epoch": 3.525352478027344e-05,
      "step": 36966,
      "training_step_time": 8.295898675918579
    },
    {
      "epoch": 3.525447845458984e-05,
      "model_forward_time": 1.3866450786590576,
      "step": 36967
    },
    {
      "epoch": 3.525447845458984e-05,
      "step": 36967,
      "training_step_time": 7.693116664886475
    },
    {
      "epoch": 3.525543212890625e-05,
      "model_forward_time": 1.7438387870788574,
      "step": 36968
    },
    {
      "epoch": 3.525543212890625e-05,
      "step": 36968,
      "training_step_time": 8.216300249099731
    },
    {
      "epoch": 3.525638580322266e-05,
      "model_forward_time": 1.750688076019287,
      "step": 36969
    },
    {
      "epoch": 3.525638580322266e-05,
      "step": 36969,
      "training_step_time": 7.638864755630493
    },
    {
      "epoch": 3.525733947753906e-05,
      "grad_norm": 0.15043355524539948,
      "learning_rate": 7.456583245831574e-06,
      "loss": 0.0599,
      "step": 36970
    },
    {
      "epoch": 3.525733947753906e-05,
      "model_forward_time": 1.7586450576782227,
      "step": 36970
    },
    {
      "dynamics_loss_avg": 0.06496774367988109,
      "epoch": 3.525733947753906e-05,
      "step": 36970
    },
    {
      "action_loss_avg": 0.0022609392821323127,
      "epoch": 3.525733947753906e-05,
      "step": 36970
    },
    {
      "epoch": 3.525733947753906e-05,
      "step": 36970,
      "training_step_time": 7.644705057144165
    },
    {
      "epoch": 3.5258293151855465e-05,
      "model_forward_time": 2.887676954269409,
      "step": 36971
    },
    {
      "epoch": 3.5258293151855465e-05,
      "step": 36971,
      "training_step_time": 8.477823257446289
    },
    {
      "epoch": 3.5259246826171876e-05,
      "model_forward_time": 1.823340654373169,
      "step": 36972
    },
    {
      "epoch": 3.5259246826171876e-05,
      "step": 36972,
      "training_step_time": 8.82706904411316
    },
    {
      "epoch": 3.526020050048828e-05,
      "model_forward_time": 1.746908187866211,
      "step": 36973
    },
    {
      "epoch": 3.526020050048828e-05,
      "step": 36973,
      "training_step_time": 7.683109760284424
    },
    {
      "epoch": 3.5261154174804684e-05,
      "model_forward_time": 1.7447562217712402,
      "step": 36974
    },
    {
      "epoch": 3.5261154174804684e-05,
      "step": 36974,
      "training_step_time": 8.305235385894775
    },
    {
      "epoch": 3.5262107849121095e-05,
      "model_forward_time": 1.7517471313476562,
      "step": 36975
    },
    {
      "epoch": 3.5262107849121095e-05,
      "step": 36975,
      "training_step_time": 8.768794536590576
    },
    {
      "epoch": 3.52630615234375e-05,
      "model_forward_time": 1.752687931060791,
      "step": 36976
    },
    {
      "epoch": 3.52630615234375e-05,
      "step": 36976,
      "training_step_time": 7.644397735595703
    },
    {
      "epoch": 3.52640151977539e-05,
      "model_forward_time": 1.7468442916870117,
      "step": 36977
    },
    {
      "epoch": 3.52640151977539e-05,
      "step": 36977,
      "training_step_time": 8.335174322128296
    },
    {
      "epoch": 3.5264968872070314e-05,
      "model_forward_time": 1.7489500045776367,
      "step": 36978
    },
    {
      "epoch": 3.5264968872070314e-05,
      "step": 36978,
      "training_step_time": 9.691295862197876
    },
    {
      "epoch": 3.526592254638672e-05,
      "model_forward_time": 1.7404062747955322,
      "step": 36979
    },
    {
      "epoch": 3.526592254638672e-05,
      "step": 36979,
      "training_step_time": 8.635503053665161
    },
    {
      "epoch": 3.526687622070312e-05,
      "grad_norm": 0.09565602988004684,
      "learning_rate": 7.455142970692968e-06,
      "loss": 0.0606,
      "step": 36980
    },
    {
      "epoch": 3.526687622070312e-05,
      "model_forward_time": 2.848329544067383,
      "step": 36980
    },
    {
      "dynamics_loss_avg": 0.05847950205206871,
      "epoch": 3.526687622070312e-05,
      "step": 36980
    },
    {
      "action_loss_avg": 0.001840176561381668,
      "epoch": 3.526687622070312e-05,
      "step": 36980
    },
    {
      "epoch": 3.526687622070312e-05,
      "step": 36980,
      "training_step_time": 8.599421262741089
    },
    {
      "epoch": 3.526782989501953e-05,
      "model_forward_time": 1.4091911315917969,
      "step": 36981
    },
    {
      "epoch": 3.526782989501953e-05,
      "step": 36981,
      "training_step_time": 8.764176368713379
    },
    {
      "epoch": 3.526878356933594e-05,
      "model_forward_time": 1.7509498596191406,
      "step": 36982
    },
    {
      "epoch": 3.526878356933594e-05,
      "step": 36982,
      "training_step_time": 9.015486478805542
    },
    {
      "epoch": 3.526973724365234e-05,
      "model_forward_time": 1.7538297176361084,
      "step": 36983
    },
    {
      "epoch": 3.526973724365234e-05,
      "step": 36983,
      "training_step_time": 7.6096251010894775
    },
    {
      "epoch": 3.527069091796875e-05,
      "model_forward_time": 1.7775371074676514,
      "step": 36984
    },
    {
      "epoch": 3.527069091796875e-05,
      "step": 36984,
      "training_step_time": 7.8051183223724365
    },
    {
      "epoch": 3.5271644592285156e-05,
      "model_forward_time": 1.7477688789367676,
      "step": 36985
    },
    {
      "epoch": 3.5271644592285156e-05,
      "step": 36985,
      "training_step_time": 9.29457139968872
    },
    {
      "epoch": 3.527259826660156e-05,
      "model_forward_time": 1.742539882659912,
      "step": 36986
    },
    {
      "epoch": 3.527259826660156e-05,
      "step": 36986,
      "training_step_time": 7.675579309463501
    },
    {
      "epoch": 3.527355194091797e-05,
      "model_forward_time": 1.7501249313354492,
      "step": 36987
    },
    {
      "epoch": 3.527355194091797e-05,
      "step": 36987,
      "training_step_time": 7.770956993103027
    },
    {
      "epoch": 3.5274505615234375e-05,
      "model_forward_time": 1.7552807331085205,
      "step": 36988
    },
    {
      "epoch": 3.5274505615234375e-05,
      "step": 36988,
      "training_step_time": 8.600075483322144
    },
    {
      "epoch": 3.527545928955078e-05,
      "model_forward_time": 1.7502541542053223,
      "step": 36989
    },
    {
      "epoch": 3.527545928955078e-05,
      "step": 36989,
      "training_step_time": 7.860485315322876
    },
    {
      "epoch": 3.527641296386719e-05,
      "grad_norm": 0.1154666393995285,
      "learning_rate": 7.453702427063616e-06,
      "loss": 0.0617,
      "step": 36990
    },
    {
      "epoch": 3.527641296386719e-05,
      "model_forward_time": 2.885335683822632,
      "step": 36990
    },
    {
      "dynamics_loss_avg": 0.05515060350298882,
      "epoch": 3.527641296386719e-05,
      "step": 36990
    },
    {
      "action_loss_avg": 0.0017419420182704926,
      "epoch": 3.527641296386719e-05,
      "step": 36990
    },
    {
      "epoch": 3.527641296386719e-05,
      "step": 36990,
      "training_step_time": 7.659066200256348
    },
    {
      "epoch": 3.5277366638183593e-05,
      "model_forward_time": 1.752321481704712,
      "step": 36991
    },
    {
      "epoch": 3.5277366638183593e-05,
      "step": 36991,
      "training_step_time": 8.724217414855957
    },
    {
      "epoch": 3.52783203125e-05,
      "model_forward_time": 1.7472288608551025,
      "step": 36992
    },
    {
      "epoch": 3.52783203125e-05,
      "step": 36992,
      "training_step_time": 8.688377618789673
    },
    {
      "epoch": 3.527927398681641e-05,
      "model_forward_time": 1.7445263862609863,
      "step": 36993
    },
    {
      "epoch": 3.527927398681641e-05,
      "step": 36993,
      "training_step_time": 9.566073417663574
    },
    {
      "epoch": 3.528022766113281e-05,
      "model_forward_time": 1.7474257946014404,
      "step": 36994
    },
    {
      "epoch": 3.528022766113281e-05,
      "step": 36994,
      "training_step_time": 7.6819007396698
    },
    {
      "epoch": 3.5281181335449216e-05,
      "model_forward_time": 1.748934030532837,
      "step": 36995
    },
    {
      "epoch": 3.5281181335449216e-05,
      "step": 36995,
      "training_step_time": 8.254356384277344
    },
    {
      "epoch": 3.528213500976563e-05,
      "model_forward_time": 1.7559809684753418,
      "step": 36996
    },
    {
      "epoch": 3.528213500976563e-05,
      "step": 36996,
      "training_step_time": 7.630074739456177
    },
    {
      "epoch": 3.528308868408203e-05,
      "model_forward_time": 1.7526516914367676,
      "step": 36997
    },
    {
      "epoch": 3.528308868408203e-05,
      "step": 36997,
      "training_step_time": 7.632694959640503
    },
    {
      "epoch": 3.5284042358398435e-05,
      "model_forward_time": 1.7916805744171143,
      "step": 36998
    },
    {
      "epoch": 3.5284042358398435e-05,
      "step": 36998,
      "training_step_time": 8.060964107513428
    },
    {
      "epoch": 3.5284996032714846e-05,
      "model_forward_time": 2.92358660697937,
      "step": 36999
    },
    {
      "epoch": 3.5284996032714846e-05,
      "step": 36999,
      "training_step_time": 9.449067831039429
    },
    {
      "epoch": 3.528594970703125e-05,
      "grad_norm": 0.11507736146450043,
      "learning_rate": 7.452261615101058e-06,
      "loss": 0.0574,
      "step": 37000
    },
    {
      "epoch": 3.528594970703125e-05,
      "model_forward_time": 1.7660713195800781,
      "step": 37000
    },
    {
      "dynamics_loss_avg": 0.0468374552205205,
      "epoch": 3.528594970703125e-05,
      "step": 37000
    },
    {
      "action_loss_avg": 0.00159458793932572,
      "epoch": 3.528594970703125e-05,
      "step": 37000
    },
    {
      "epoch": 3.528594970703125e-05,
      "step": 37000,
      "training_step_time": 6.595050811767578
    },
    {
      "epoch": 3.5286903381347654e-05,
      "model_forward_time": 1.7391445636749268,
      "step": 37001
    },
    {
      "epoch": 3.5286903381347654e-05,
      "step": 37001,
      "training_step_time": 6.660855293273926
    },
    {
      "epoch": 3.5287857055664065e-05,
      "model_forward_time": 1.758340835571289,
      "step": 37002
    },
    {
      "epoch": 3.5287857055664065e-05,
      "step": 37002,
      "training_step_time": 6.786221265792847
    },
    {
      "epoch": 3.528881072998047e-05,
      "model_forward_time": 1.7476959228515625,
      "step": 37003
    },
    {
      "epoch": 3.528881072998047e-05,
      "step": 37003,
      "training_step_time": 6.724318027496338
    },
    {
      "epoch": 3.528976440429687e-05,
      "model_forward_time": 1.7528021335601807,
      "step": 37004
    },
    {
      "epoch": 3.528976440429687e-05,
      "step": 37004,
      "training_step_time": 6.712344408035278
    },
    {
      "epoch": 3.5290718078613284e-05,
      "model_forward_time": 1.7479419708251953,
      "step": 37005
    },
    {
      "epoch": 3.5290718078613284e-05,
      "step": 37005,
      "training_step_time": 6.780009984970093
    },
    {
      "epoch": 3.529167175292969e-05,
      "model_forward_time": 1.745661735534668,
      "step": 37006
    },
    {
      "epoch": 3.529167175292969e-05,
      "step": 37006,
      "training_step_time": 6.689655303955078
    },
    {
      "epoch": 3.529262542724609e-05,
      "model_forward_time": 1.7522947788238525,
      "step": 37007
    },
    {
      "epoch": 3.529262542724609e-05,
      "step": 37007,
      "training_step_time": 6.70632004737854
    },
    {
      "epoch": 3.52935791015625e-05,
      "model_forward_time": 1.7317628860473633,
      "step": 37008
    },
    {
      "epoch": 3.52935791015625e-05,
      "step": 37008,
      "training_step_time": 8.536457061767578
    },
    {
      "epoch": 3.5294532775878907e-05,
      "model_forward_time": 3.3450353145599365,
      "step": 37009
    },
    {
      "epoch": 3.5294532775878907e-05,
      "step": 37009,
      "training_step_time": 8.156673669815063
    },
    {
      "epoch": 3.529548645019531e-05,
      "grad_norm": 0.1520364135503769,
      "learning_rate": 7.450820534962856e-06,
      "loss": 0.0577,
      "step": 37010
    },
    {
      "epoch": 3.529548645019531e-05,
      "model_forward_time": 1.7420103549957275,
      "step": 37010
    },
    {
      "dynamics_loss_avg": 0.04643580503761768,
      "epoch": 3.529548645019531e-05,
      "step": 37010
    },
    {
      "action_loss_avg": 0.0018532001238781958,
      "epoch": 3.529548645019531e-05,
      "step": 37010
    },
    {
      "epoch": 3.529548645019531e-05,
      "step": 37010,
      "training_step_time": 6.729294776916504
    },
    {
      "epoch": 3.529644012451172e-05,
      "model_forward_time": 1.7516076564788818,
      "step": 37011
    },
    {
      "epoch": 3.529644012451172e-05,
      "step": 37011,
      "training_step_time": 6.697146415710449
    },
    {
      "epoch": 3.5297393798828125e-05,
      "model_forward_time": 1.739976167678833,
      "step": 37012
    },
    {
      "epoch": 3.5297393798828125e-05,
      "step": 37012,
      "training_step_time": 6.7011542320251465
    },
    {
      "epoch": 3.529834747314453e-05,
      "model_forward_time": 1.7361071109771729,
      "step": 37013
    },
    {
      "epoch": 3.529834747314453e-05,
      "step": 37013,
      "training_step_time": 6.705984354019165
    },
    {
      "epoch": 3.529930114746094e-05,
      "model_forward_time": 1.748652696609497,
      "step": 37014
    },
    {
      "epoch": 3.529930114746094e-05,
      "step": 37014,
      "training_step_time": 6.692144393920898
    },
    {
      "epoch": 3.5300254821777344e-05,
      "model_forward_time": 1.3774595260620117,
      "step": 37015
    },
    {
      "epoch": 3.5300254821777344e-05,
      "step": 37015,
      "training_step_time": 6.696927070617676
    },
    {
      "epoch": 3.530120849609375e-05,
      "model_forward_time": 1.7417511940002441,
      "step": 37016
    },
    {
      "epoch": 3.530120849609375e-05,
      "step": 37016,
      "training_step_time": 6.785391330718994
    },
    {
      "epoch": 3.530216217041016e-05,
      "model_forward_time": 1.7472612857818604,
      "step": 37017
    },
    {
      "epoch": 3.530216217041016e-05,
      "step": 37017,
      "training_step_time": 6.705670118331909
    },
    {
      "epoch": 3.530311584472656e-05,
      "model_forward_time": 2.8563337326049805,
      "step": 37018
    },
    {
      "epoch": 3.530311584472656e-05,
      "step": 37018,
      "training_step_time": 7.7189624309539795
    },
    {
      "epoch": 3.530406951904297e-05,
      "model_forward_time": 1.7412481307983398,
      "step": 37019
    },
    {
      "epoch": 3.530406951904297e-05,
      "step": 37019,
      "training_step_time": 6.704747915267944
    },
    {
      "epoch": 3.530502319335938e-05,
      "grad_norm": 0.10303675383329391,
      "learning_rate": 7.449379186806605e-06,
      "loss": 0.0575,
      "step": 37020
    },
    {
      "epoch": 3.530502319335938e-05,
      "model_forward_time": 1.7347102165222168,
      "step": 37020
    },
    {
      "dynamics_loss_avg": 0.04505073633044958,
      "epoch": 3.530502319335938e-05,
      "step": 37020
    },
    {
      "action_loss_avg": 0.0029921800378360786,
      "epoch": 3.530502319335938e-05,
      "step": 37020
    },
    {
      "epoch": 3.530502319335938e-05,
      "step": 37020,
      "training_step_time": 6.704255104064941
    },
    {
      "epoch": 3.530597686767578e-05,
      "model_forward_time": 1.741344928741455,
      "step": 37021
    },
    {
      "epoch": 3.530597686767578e-05,
      "step": 37021,
      "training_step_time": 6.735661268234253
    },
    {
      "epoch": 3.5306930541992186e-05,
      "model_forward_time": 1.7628278732299805,
      "step": 37022
    },
    {
      "epoch": 3.5306930541992186e-05,
      "step": 37022,
      "training_step_time": 6.691546440124512
    },
    {
      "epoch": 3.53078842163086e-05,
      "model_forward_time": 1.7438251972198486,
      "step": 37023
    },
    {
      "epoch": 3.53078842163086e-05,
      "step": 37023,
      "training_step_time": 6.722447156906128
    },
    {
      "epoch": 3.5308837890625e-05,
      "model_forward_time": 1.7403783798217773,
      "step": 37024
    },
    {
      "epoch": 3.5308837890625e-05,
      "step": 37024,
      "training_step_time": 6.726550579071045
    },
    {
      "epoch": 3.5309791564941405e-05,
      "model_forward_time": 1.7481608390808105,
      "step": 37025
    },
    {
      "epoch": 3.5309791564941405e-05,
      "step": 37025,
      "training_step_time": 6.736532688140869
    },
    {
      "epoch": 3.5310745239257816e-05,
      "model_forward_time": 1.7426819801330566,
      "step": 37026
    },
    {
      "epoch": 3.5310745239257816e-05,
      "step": 37026,
      "training_step_time": 6.714451313018799
    },
    {
      "epoch": 3.531169891357422e-05,
      "model_forward_time": 2.8585877418518066,
      "step": 37027
    },
    {
      "epoch": 3.531169891357422e-05,
      "step": 37027,
      "training_step_time": 7.771019458770752
    },
    {
      "epoch": 3.5312652587890624e-05,
      "model_forward_time": 1.7410237789154053,
      "step": 37028
    },
    {
      "epoch": 3.5312652587890624e-05,
      "step": 37028,
      "training_step_time": 7.642723083496094
    },
    {
      "epoch": 3.5313606262207035e-05,
      "model_forward_time": 1.7503948211669922,
      "step": 37029
    },
    {
      "epoch": 3.5313606262207035e-05,
      "step": 37029,
      "training_step_time": 6.736507892608643
    },
    {
      "epoch": 3.531455993652344e-05,
      "grad_norm": 0.14212317764759064,
      "learning_rate": 7.447937570789929e-06,
      "loss": 0.0607,
      "step": 37030
    },
    {
      "epoch": 3.531455993652344e-05,
      "model_forward_time": 1.7280347347259521,
      "step": 37030
    },
    {
      "dynamics_loss_avg": 0.0617040166631341,
      "epoch": 3.531455993652344e-05,
      "step": 37030
    },
    {
      "action_loss_avg": 0.002396795351523906,
      "epoch": 3.531455993652344e-05,
      "step": 37030
    },
    {
      "epoch": 3.531455993652344e-05,
      "step": 37030,
      "training_step_time": 6.763745069503784
    },
    {
      "epoch": 3.531551361083984e-05,
      "model_forward_time": 1.726769208908081,
      "step": 37031
    },
    {
      "epoch": 3.531551361083984e-05,
      "step": 37031,
      "training_step_time": 6.762239456176758
    },
    {
      "epoch": 3.531646728515625e-05,
      "model_forward_time": 1.7370638847351074,
      "step": 37032
    },
    {
      "epoch": 3.531646728515625e-05,
      "step": 37032,
      "training_step_time": 6.7461090087890625
    },
    {
      "epoch": 3.531742095947266e-05,
      "model_forward_time": 1.7402732372283936,
      "step": 37033
    },
    {
      "epoch": 3.531742095947266e-05,
      "step": 37033,
      "training_step_time": 6.72089695930481
    },
    {
      "epoch": 3.531837463378906e-05,
      "model_forward_time": 1.7425754070281982,
      "step": 37034
    },
    {
      "epoch": 3.531837463378906e-05,
      "step": 37034,
      "training_step_time": 6.7654128074646
    },
    {
      "epoch": 3.5319328308105465e-05,
      "model_forward_time": 1.7485315799713135,
      "step": 37035
    },
    {
      "epoch": 3.5319328308105465e-05,
      "step": 37035,
      "training_step_time": 6.7422449588775635
    },
    {
      "epoch": 3.5320281982421876e-05,
      "model_forward_time": 2.8629310131073,
      "step": 37036
    },
    {
      "epoch": 3.5320281982421876e-05,
      "step": 37036,
      "training_step_time": 7.761491537094116
    },
    {
      "epoch": 3.532123565673828e-05,
      "model_forward_time": 1.7521822452545166,
      "step": 37037
    },
    {
      "epoch": 3.532123565673828e-05,
      "step": 37037,
      "training_step_time": 7.708775281906128
    },
    {
      "epoch": 3.5322189331054684e-05,
      "model_forward_time": 1.7461285591125488,
      "step": 37038
    },
    {
      "epoch": 3.5322189331054684e-05,
      "step": 37038,
      "training_step_time": 6.75096321105957
    },
    {
      "epoch": 3.5323143005371095e-05,
      "model_forward_time": 1.7503156661987305,
      "step": 37039
    },
    {
      "epoch": 3.5323143005371095e-05,
      "step": 37039,
      "training_step_time": 6.666286468505859
    },
    {
      "epoch": 3.53240966796875e-05,
      "grad_norm": 0.16948507726192474,
      "learning_rate": 7.446495687070481e-06,
      "loss": 0.0636,
      "step": 37040
    },
    {
      "epoch": 3.53240966796875e-05,
      "model_forward_time": 1.37980318069458,
      "step": 37040
    },
    {
      "dynamics_loss_avg": 0.05151315778493881,
      "epoch": 3.53240966796875e-05,
      "step": 37040
    },
    {
      "action_loss_avg": 0.0012041415786370635,
      "epoch": 3.53240966796875e-05,
      "step": 37040
    },
    {
      "epoch": 3.53240966796875e-05,
      "step": 37040,
      "training_step_time": 6.74322509765625
    },
    {
      "epoch": 3.53250503540039e-05,
      "model_forward_time": 1.7429468631744385,
      "step": 37041
    },
    {
      "epoch": 3.53250503540039e-05,
      "step": 37041,
      "training_step_time": 6.763257741928101
    },
    {
      "epoch": 3.5326004028320314e-05,
      "model_forward_time": 1.7444689273834229,
      "step": 37042
    },
    {
      "epoch": 3.5326004028320314e-05,
      "step": 37042,
      "training_step_time": 6.726771593093872
    },
    {
      "epoch": 3.532695770263672e-05,
      "model_forward_time": 1.7421350479125977,
      "step": 37043
    },
    {
      "epoch": 3.532695770263672e-05,
      "step": 37043,
      "training_step_time": 6.72632622718811
    },
    {
      "epoch": 3.532791137695312e-05,
      "model_forward_time": 1.924248218536377,
      "step": 37044
    },
    {
      "epoch": 3.532791137695312e-05,
      "step": 37044,
      "training_step_time": 6.813490152359009
    },
    {
      "epoch": 3.532886505126953e-05,
      "model_forward_time": 1.522557258605957,
      "step": 37045
    },
    {
      "epoch": 3.532886505126953e-05,
      "step": 37045,
      "training_step_time": 10.2535879611969
    },
    {
      "epoch": 3.532981872558594e-05,
      "model_forward_time": 1.9080493450164795,
      "step": 37046
    },
    {
      "epoch": 3.532981872558594e-05,
      "step": 37046,
      "training_step_time": 8.287603616714478
    },
    {
      "epoch": 3.533077239990234e-05,
      "model_forward_time": 1.9114952087402344,
      "step": 37047
    },
    {
      "epoch": 3.533077239990234e-05,
      "step": 37047,
      "training_step_time": 7.165517807006836
    },
    {
      "epoch": 3.533172607421875e-05,
      "model_forward_time": 1.884221076965332,
      "step": 37048
    },
    {
      "epoch": 3.533172607421875e-05,
      "step": 37048,
      "training_step_time": 6.76878023147583
    },
    {
      "epoch": 3.5332679748535156e-05,
      "model_forward_time": 1.9169936180114746,
      "step": 37049
    },
    {
      "epoch": 3.5332679748535156e-05,
      "step": 37049,
      "training_step_time": 6.751061201095581
    },
    {
      "epoch": 3.533363342285156e-05,
      "grad_norm": 0.12411144375801086,
      "learning_rate": 7.4450535358059415e-06,
      "loss": 0.0637,
      "step": 37050
    },
    {
      "epoch": 3.533363342285156e-05,
      "model_forward_time": 1.883751392364502,
      "step": 37050
    },
    {
      "dynamics_loss_avg": 0.054038528725504874,
      "epoch": 3.533363342285156e-05,
      "step": 37050
    },
    {
      "action_loss_avg": 0.0022663936542812735,
      "epoch": 3.533363342285156e-05,
      "step": 37050
    },
    {
      "epoch": 3.533363342285156e-05,
      "step": 37050,
      "training_step_time": 6.765331983566284
    },
    {
      "epoch": 3.533458709716797e-05,
      "model_forward_time": 1.914170742034912,
      "step": 37051
    },
    {
      "epoch": 3.533458709716797e-05,
      "step": 37051,
      "training_step_time": 6.7307798862457275
    },
    {
      "epoch": 3.5335540771484375e-05,
      "model_forward_time": 1.8778598308563232,
      "step": 37052
    },
    {
      "epoch": 3.5335540771484375e-05,
      "step": 37052,
      "training_step_time": 6.778892517089844
    },
    {
      "epoch": 3.533649444580078e-05,
      "model_forward_time": 1.877692461013794,
      "step": 37053
    },
    {
      "epoch": 3.533649444580078e-05,
      "step": 37053,
      "training_step_time": 6.791870832443237
    },
    {
      "epoch": 3.533744812011719e-05,
      "model_forward_time": 1.9053423404693604,
      "step": 37054
    },
    {
      "epoch": 3.533744812011719e-05,
      "step": 37054,
      "training_step_time": 11.185517072677612
    },
    {
      "epoch": 3.5338401794433593e-05,
      "model_forward_time": 2.96575665473938,
      "step": 37055
    },
    {
      "epoch": 3.5338401794433593e-05,
      "step": 37055,
      "training_step_time": 7.7406556606292725
    },
    {
      "epoch": 3.533935546875e-05,
      "model_forward_time": 1.8968427181243896,
      "step": 37056
    },
    {
      "epoch": 3.533935546875e-05,
      "step": 37056,
      "training_step_time": 7.185425758361816
    },
    {
      "epoch": 3.534030914306641e-05,
      "model_forward_time": 1.8820483684539795,
      "step": 37057
    },
    {
      "epoch": 3.534030914306641e-05,
      "step": 37057,
      "training_step_time": 6.755045652389526
    },
    {
      "epoch": 3.534126281738281e-05,
      "model_forward_time": 1.867358684539795,
      "step": 37058
    },
    {
      "epoch": 3.534126281738281e-05,
      "step": 37058,
      "training_step_time": 6.804291009902954
    },
    {
      "epoch": 3.5342216491699216e-05,
      "model_forward_time": 1.9015793800354004,
      "step": 37059
    },
    {
      "epoch": 3.5342216491699216e-05,
      "step": 37059,
      "training_step_time": 6.774292230606079
    },
    {
      "epoch": 3.534317016601563e-05,
      "grad_norm": 0.17972563207149506,
      "learning_rate": 7.443611117154023e-06,
      "loss": 0.0639,
      "step": 37060
    },
    {
      "epoch": 3.534317016601563e-05,
      "model_forward_time": 1.866154670715332,
      "step": 37060
    },
    {
      "dynamics_loss_avg": 0.05233765635639429,
      "epoch": 3.534317016601563e-05,
      "step": 37060
    },
    {
      "action_loss_avg": 0.0012892702958197332,
      "epoch": 3.534317016601563e-05,
      "step": 37060
    },
    {
      "epoch": 3.534317016601563e-05,
      "step": 37060,
      "training_step_time": 6.781342267990112
    },
    {
      "epoch": 3.534412384033203e-05,
      "model_forward_time": 1.8320386409759521,
      "step": 37061
    },
    {
      "epoch": 3.534412384033203e-05,
      "step": 37061,
      "training_step_time": 6.7480762004852295
    },
    {
      "epoch": 3.5345077514648435e-05,
      "model_forward_time": 1.74415922164917,
      "step": 37062
    },
    {
      "epoch": 3.5345077514648435e-05,
      "step": 37062,
      "training_step_time": 6.780512094497681
    },
    {
      "epoch": 3.5346031188964846e-05,
      "model_forward_time": 1.7448766231536865,
      "step": 37063
    },
    {
      "epoch": 3.5346031188964846e-05,
      "step": 37063,
      "training_step_time": 10.924815893173218
    },
    {
      "epoch": 3.534698486328125e-05,
      "model_forward_time": 2.9803783893585205,
      "step": 37064
    },
    {
      "epoch": 3.534698486328125e-05,
      "step": 37064,
      "training_step_time": 8.413728713989258
    },
    {
      "epoch": 3.5347938537597654e-05,
      "model_forward_time": 1.7498879432678223,
      "step": 37065
    },
    {
      "epoch": 3.5347938537597654e-05,
      "step": 37065,
      "training_step_time": 7.190638780593872
    },
    {
      "epoch": 3.5348892211914065e-05,
      "model_forward_time": 1.7515058517456055,
      "step": 37066
    },
    {
      "epoch": 3.5348892211914065e-05,
      "step": 37066,
      "training_step_time": 6.815603494644165
    },
    {
      "epoch": 3.534984588623047e-05,
      "model_forward_time": 1.7494535446166992,
      "step": 37067
    },
    {
      "epoch": 3.534984588623047e-05,
      "step": 37067,
      "training_step_time": 6.743863344192505
    },
    {
      "epoch": 3.535079956054687e-05,
      "model_forward_time": 1.7472431659698486,
      "step": 37068
    },
    {
      "epoch": 3.535079956054687e-05,
      "step": 37068,
      "training_step_time": 6.762040853500366
    },
    {
      "epoch": 3.5351753234863284e-05,
      "model_forward_time": 1.745783805847168,
      "step": 37069
    },
    {
      "epoch": 3.5351753234863284e-05,
      "step": 37069,
      "training_step_time": 6.772324800491333
    },
    {
      "epoch": 3.535270690917969e-05,
      "grad_norm": 0.16447700560092926,
      "learning_rate": 7.4421684312724685e-06,
      "loss": 0.0608,
      "step": 37070
    },
    {
      "epoch": 3.535270690917969e-05,
      "model_forward_time": 1.735231876373291,
      "step": 37070
    },
    {
      "dynamics_loss_avg": 0.056540252082049845,
      "epoch": 3.535270690917969e-05,
      "step": 37070
    },
    {
      "action_loss_avg": 0.0021434288239106534,
      "epoch": 3.535270690917969e-05,
      "step": 37070
    },
    {
      "epoch": 3.535270690917969e-05,
      "step": 37070,
      "training_step_time": 6.798982620239258
    },
    {
      "epoch": 3.535366058349609e-05,
      "model_forward_time": 1.746502161026001,
      "step": 37071
    },
    {
      "epoch": 3.535366058349609e-05,
      "step": 37071,
      "training_step_time": 6.754602432250977
    },
    {
      "epoch": 3.53546142578125e-05,
      "model_forward_time": 1.749361276626587,
      "step": 37072
    },
    {
      "epoch": 3.53546142578125e-05,
      "step": 37072,
      "training_step_time": 9.283194780349731
    },
    {
      "epoch": 3.535556793212891e-05,
      "model_forward_time": 2.9147818088531494,
      "step": 37073
    },
    {
      "epoch": 3.535556793212891e-05,
      "step": 37073,
      "training_step_time": 7.895689964294434
    },
    {
      "epoch": 3.535652160644531e-05,
      "model_forward_time": 1.747934341430664,
      "step": 37074
    },
    {
      "epoch": 3.535652160644531e-05,
      "step": 37074,
      "training_step_time": 7.727672576904297
    },
    {
      "epoch": 3.535747528076172e-05,
      "model_forward_time": 1.7451539039611816,
      "step": 37075
    },
    {
      "epoch": 3.535747528076172e-05,
      "step": 37075,
      "training_step_time": 6.764648675918579
    },
    {
      "epoch": 3.5358428955078125e-05,
      "model_forward_time": 1.7421462535858154,
      "step": 37076
    },
    {
      "epoch": 3.5358428955078125e-05,
      "step": 37076,
      "training_step_time": 6.7378199100494385
    },
    {
      "epoch": 3.535938262939453e-05,
      "model_forward_time": 1.739121913909912,
      "step": 37077
    },
    {
      "epoch": 3.535938262939453e-05,
      "step": 37077,
      "training_step_time": 6.756635427474976
    },
    {
      "epoch": 3.536033630371094e-05,
      "model_forward_time": 1.744795560836792,
      "step": 37078
    },
    {
      "epoch": 3.536033630371094e-05,
      "step": 37078,
      "training_step_time": 6.736778020858765
    },
    {
      "epoch": 3.5361289978027344e-05,
      "model_forward_time": 1.378232479095459,
      "step": 37079
    },
    {
      "epoch": 3.5361289978027344e-05,
      "step": 37079,
      "training_step_time": 6.776803970336914
    },
    {
      "epoch": 3.536224365234375e-05,
      "grad_norm": 0.14889337122440338,
      "learning_rate": 7.440725478319045e-06,
      "loss": 0.0648,
      "step": 37080
    },
    {
      "epoch": 3.536224365234375e-05,
      "model_forward_time": 1.7376432418823242,
      "step": 37080
    },
    {
      "dynamics_loss_avg": 0.05623632408678532,
      "epoch": 3.536224365234375e-05,
      "step": 37080
    },
    {
      "action_loss_avg": 0.001557672704802826,
      "epoch": 3.536224365234375e-05,
      "step": 37080
    },
    {
      "epoch": 3.536224365234375e-05,
      "step": 37080,
      "training_step_time": 6.8001868724823
    },
    {
      "epoch": 3.536319732666016e-05,
      "model_forward_time": 1.7282240390777588,
      "step": 37081
    },
    {
      "epoch": 3.536319732666016e-05,
      "step": 37081,
      "training_step_time": 7.592342138290405
    },
    {
      "epoch": 3.536415100097656e-05,
      "model_forward_time": 2.931826591491699,
      "step": 37082
    },
    {
      "epoch": 3.536415100097656e-05,
      "step": 37082,
      "training_step_time": 8.860764980316162
    },
    {
      "epoch": 3.536510467529297e-05,
      "model_forward_time": 1.7519714832305908,
      "step": 37083
    },
    {
      "epoch": 3.536510467529297e-05,
      "step": 37083,
      "training_step_time": 8.246148586273193
    },
    {
      "epoch": 3.536605834960938e-05,
      "model_forward_time": 1.7602753639221191,
      "step": 37084
    },
    {
      "epoch": 3.536605834960938e-05,
      "step": 37084,
      "training_step_time": 7.1957292556762695
    },
    {
      "epoch": 3.536701202392578e-05,
      "model_forward_time": 1.3858041763305664,
      "step": 37085
    },
    {
      "epoch": 3.536701202392578e-05,
      "step": 37085,
      "training_step_time": 6.896148920059204
    },
    {
      "epoch": 3.5367965698242186e-05,
      "model_forward_time": 1.7551136016845703,
      "step": 37086
    },
    {
      "epoch": 3.5367965698242186e-05,
      "step": 37086,
      "training_step_time": 6.819972276687622
    },
    {
      "epoch": 3.53689193725586e-05,
      "model_forward_time": 1.7464234828948975,
      "step": 37087
    },
    {
      "epoch": 3.53689193725586e-05,
      "step": 37087,
      "training_step_time": 6.7694971561431885
    },
    {
      "epoch": 3.5369873046875e-05,
      "model_forward_time": 1.7434723377227783,
      "step": 37088
    },
    {
      "epoch": 3.5369873046875e-05,
      "step": 37088,
      "training_step_time": 6.807083606719971
    },
    {
      "epoch": 3.5370826721191405e-05,
      "model_forward_time": 1.7561368942260742,
      "step": 37089
    },
    {
      "epoch": 3.5370826721191405e-05,
      "step": 37089,
      "training_step_time": 6.830461740493774
    },
    {
      "epoch": 3.5371780395507816e-05,
      "grad_norm": 0.16374807059764862,
      "learning_rate": 7.439282258451553e-06,
      "loss": 0.0662,
      "step": 37090
    },
    {
      "epoch": 3.5371780395507816e-05,
      "model_forward_time": 1.7290358543395996,
      "step": 37090
    },
    {
      "dynamics_loss_avg": 0.06152028664946556,
      "epoch": 3.5371780395507816e-05,
      "step": 37090
    },
    {
      "action_loss_avg": 0.0031291858991608023,
      "epoch": 3.5371780395507816e-05,
      "step": 37090
    },
    {
      "epoch": 3.5371780395507816e-05,
      "step": 37090,
      "training_step_time": 7.677990913391113
    },
    {
      "epoch": 3.537273406982422e-05,
      "model_forward_time": 1.8094244003295898,
      "step": 37091
    },
    {
      "epoch": 3.537273406982422e-05,
      "step": 37091,
      "training_step_time": 10.406363248825073
    },
    {
      "epoch": 3.5373687744140624e-05,
      "model_forward_time": 1.811084270477295,
      "step": 37092
    },
    {
      "epoch": 3.5373687744140624e-05,
      "step": 37092,
      "training_step_time": 7.616354942321777
    },
    {
      "epoch": 3.5374641418457035e-05,
      "model_forward_time": 1.7465331554412842,
      "step": 37093
    },
    {
      "epoch": 3.5374641418457035e-05,
      "step": 37093,
      "training_step_time": 7.706412076950073
    },
    {
      "epoch": 3.537559509277344e-05,
      "model_forward_time": 1.7455756664276123,
      "step": 37094
    },
    {
      "epoch": 3.537559509277344e-05,
      "step": 37094,
      "training_step_time": 6.792669057846069
    },
    {
      "epoch": 3.537654876708984e-05,
      "model_forward_time": 1.751910924911499,
      "step": 37095
    },
    {
      "epoch": 3.537654876708984e-05,
      "step": 37095,
      "training_step_time": 6.800824165344238
    },
    {
      "epoch": 3.537750244140625e-05,
      "model_forward_time": 1.7401094436645508,
      "step": 37096
    },
    {
      "epoch": 3.537750244140625e-05,
      "step": 37096,
      "training_step_time": 6.7844321727752686
    },
    {
      "epoch": 3.537845611572266e-05,
      "model_forward_time": 1.7449800968170166,
      "step": 37097
    },
    {
      "epoch": 3.537845611572266e-05,
      "step": 37097,
      "training_step_time": 6.767118453979492
    },
    {
      "epoch": 3.537940979003906e-05,
      "model_forward_time": 1.7480947971343994,
      "step": 37098
    },
    {
      "epoch": 3.537940979003906e-05,
      "step": 37098,
      "training_step_time": 6.7698352336883545
    },
    {
      "epoch": 3.5380363464355466e-05,
      "model_forward_time": 1.7427451610565186,
      "step": 37099
    },
    {
      "epoch": 3.5380363464355466e-05,
      "step": 37099,
      "training_step_time": 7.685141324996948
    },
    {
      "epoch": 3.5381317138671876e-05,
      "grad_norm": 0.15218211710453033,
      "learning_rate": 7.437838771827821e-06,
      "loss": 0.0582,
      "step": 37100
    },
    {
      "epoch": 3.5381317138671876e-05,
      "model_forward_time": 1.808748483657837,
      "step": 37100
    },
    {
      "dynamics_loss_avg": 0.0538804242387414,
      "epoch": 3.5381317138671876e-05,
      "step": 37100
    },
    {
      "action_loss_avg": 0.0034078734111972152,
      "epoch": 3.5381317138671876e-05,
      "step": 37100
    },
    {
      "epoch": 3.5381317138671876e-05,
      "step": 37100,
      "training_step_time": 9.491893768310547
    },
    {
      "epoch": 3.538227081298828e-05,
      "model_forward_time": 2.4771769046783447,
      "step": 37101
    },
    {
      "epoch": 3.538227081298828e-05,
      "step": 37101,
      "training_step_time": 7.7208569049835205
    },
    {
      "epoch": 3.5383224487304684e-05,
      "model_forward_time": 1.7515747547149658,
      "step": 37102
    },
    {
      "epoch": 3.5383224487304684e-05,
      "step": 37102,
      "training_step_time": 7.660028696060181
    },
    {
      "epoch": 3.5384178161621095e-05,
      "model_forward_time": 1.754995346069336,
      "step": 37103
    },
    {
      "epoch": 3.5384178161621095e-05,
      "step": 37103,
      "training_step_time": 6.793364763259888
    },
    {
      "epoch": 3.53851318359375e-05,
      "model_forward_time": 1.7465858459472656,
      "step": 37104
    },
    {
      "epoch": 3.53851318359375e-05,
      "step": 37104,
      "training_step_time": 6.775473117828369
    },
    {
      "epoch": 3.53860855102539e-05,
      "model_forward_time": 1.7475814819335938,
      "step": 37105
    },
    {
      "epoch": 3.53860855102539e-05,
      "step": 37105,
      "training_step_time": 6.726057767868042
    },
    {
      "epoch": 3.5387039184570314e-05,
      "model_forward_time": 1.7484359741210938,
      "step": 37106
    },
    {
      "epoch": 3.5387039184570314e-05,
      "step": 37106,
      "training_step_time": 6.700040578842163
    },
    {
      "epoch": 3.538799285888672e-05,
      "model_forward_time": 1.7495250701904297,
      "step": 37107
    },
    {
      "epoch": 3.538799285888672e-05,
      "step": 37107,
      "training_step_time": 6.699406147003174
    },
    {
      "epoch": 3.538894653320312e-05,
      "model_forward_time": 1.7446174621582031,
      "step": 37108
    },
    {
      "epoch": 3.538894653320312e-05,
      "step": 37108,
      "training_step_time": 7.326970100402832
    },
    {
      "epoch": 3.538990020751953e-05,
      "model_forward_time": 1.7857489585876465,
      "step": 37109
    },
    {
      "epoch": 3.538990020751953e-05,
      "step": 37109,
      "training_step_time": 10.769410848617554
    },
    {
      "epoch": 3.539085388183594e-05,
      "grad_norm": 0.15221327543258667,
      "learning_rate": 7.4363950186057065e-06,
      "loss": 0.0611,
      "step": 37110
    },
    {
      "epoch": 3.539085388183594e-05,
      "model_forward_time": 1.7459137439727783,
      "step": 37110
    },
    {
      "dynamics_loss_avg": 0.0726744769141078,
      "epoch": 3.539085388183594e-05,
      "step": 37110
    },
    {
      "action_loss_avg": 0.0031549369101412593,
      "epoch": 3.539085388183594e-05,
      "step": 37110
    },
    {
      "epoch": 3.539085388183594e-05,
      "step": 37110,
      "training_step_time": 8.579428911209106
    },
    {
      "epoch": 3.539180755615234e-05,
      "model_forward_time": 2.890120029449463,
      "step": 37111
    },
    {
      "epoch": 3.539180755615234e-05,
      "step": 37111,
      "training_step_time": 8.245169878005981
    },
    {
      "epoch": 3.539276123046875e-05,
      "model_forward_time": 1.7408514022827148,
      "step": 37112
    },
    {
      "epoch": 3.539276123046875e-05,
      "step": 37112,
      "training_step_time": 7.624970436096191
    },
    {
      "epoch": 3.5393714904785156e-05,
      "model_forward_time": 1.744913101196289,
      "step": 37113
    },
    {
      "epoch": 3.5393714904785156e-05,
      "step": 37113,
      "training_step_time": 6.7069313526153564
    },
    {
      "epoch": 3.539466857910156e-05,
      "model_forward_time": 1.7575650215148926,
      "step": 37114
    },
    {
      "epoch": 3.539466857910156e-05,
      "step": 37114,
      "training_step_time": 6.752109527587891
    },
    {
      "epoch": 3.539562225341797e-05,
      "model_forward_time": 1.7458512783050537,
      "step": 37115
    },
    {
      "epoch": 3.539562225341797e-05,
      "step": 37115,
      "training_step_time": 6.768712043762207
    },
    {
      "epoch": 3.5396575927734375e-05,
      "model_forward_time": 1.737248182296753,
      "step": 37116
    },
    {
      "epoch": 3.5396575927734375e-05,
      "step": 37116,
      "training_step_time": 6.763220310211182
    },
    {
      "epoch": 3.539752960205078e-05,
      "model_forward_time": 1.7485368251800537,
      "step": 37117
    },
    {
      "epoch": 3.539752960205078e-05,
      "step": 37117,
      "training_step_time": 7.309824705123901
    },
    {
      "epoch": 3.539848327636719e-05,
      "model_forward_time": 1.0510001182556152,
      "step": 37118
    },
    {
      "epoch": 3.539848327636719e-05,
      "step": 37118,
      "training_step_time": 8.225212812423706
    },
    {
      "epoch": 3.5399436950683594e-05,
      "model_forward_time": 1.7492890357971191,
      "step": 37119
    },
    {
      "epoch": 3.5399436950683594e-05,
      "step": 37119,
      "training_step_time": 7.822255849838257
    },
    {
      "epoch": 3.5400390625e-05,
      "grad_norm": 0.1129554808139801,
      "learning_rate": 7.434950998943094e-06,
      "loss": 0.0568,
      "step": 37120
    },
    {
      "epoch": 3.5400390625e-05,
      "model_forward_time": 2.9161176681518555,
      "step": 37120
    },
    {
      "dynamics_loss_avg": 0.05464597642421722,
      "epoch": 3.5400390625e-05,
      "step": 37120
    },
    {
      "action_loss_avg": 0.0016738116886699573,
      "epoch": 3.5400390625e-05,
      "step": 37120
    },
    {
      "epoch": 3.5400390625e-05,
      "step": 37120,
      "training_step_time": 7.767863512039185
    },
    {
      "epoch": 3.540134429931641e-05,
      "model_forward_time": 1.742814540863037,
      "step": 37121
    },
    {
      "epoch": 3.540134429931641e-05,
      "step": 37121,
      "training_step_time": 7.649733304977417
    },
    {
      "epoch": 3.540229797363281e-05,
      "model_forward_time": 1.749758243560791,
      "step": 37122
    },
    {
      "epoch": 3.540229797363281e-05,
      "step": 37122,
      "training_step_time": 6.6995580196380615
    },
    {
      "epoch": 3.5403251647949216e-05,
      "model_forward_time": 1.7474498748779297,
      "step": 37123
    },
    {
      "epoch": 3.5403251647949216e-05,
      "step": 37123,
      "training_step_time": 6.702022314071655
    },
    {
      "epoch": 3.540420532226563e-05,
      "model_forward_time": 1.74430513381958,
      "step": 37124
    },
    {
      "epoch": 3.540420532226563e-05,
      "step": 37124,
      "training_step_time": 6.726297616958618
    },
    {
      "epoch": 3.540515899658203e-05,
      "model_forward_time": 1.743638038635254,
      "step": 37125
    },
    {
      "epoch": 3.540515899658203e-05,
      "step": 37125,
      "training_step_time": 6.743412733078003
    },
    {
      "epoch": 3.5406112670898435e-05,
      "model_forward_time": 1.7459540367126465,
      "step": 37126
    },
    {
      "epoch": 3.5406112670898435e-05,
      "step": 37126,
      "training_step_time": 6.725109100341797
    },
    {
      "epoch": 3.5407066345214846e-05,
      "model_forward_time": 1.751358985900879,
      "step": 37127
    },
    {
      "epoch": 3.5407066345214846e-05,
      "step": 37127,
      "training_step_time": 8.682802677154541
    },
    {
      "epoch": 3.540802001953125e-05,
      "model_forward_time": 1.753601312637329,
      "step": 37128
    },
    {
      "epoch": 3.540802001953125e-05,
      "step": 37128,
      "training_step_time": 8.230881690979004
    },
    {
      "epoch": 3.5408973693847654e-05,
      "model_forward_time": 2.8850882053375244,
      "step": 37129
    },
    {
      "epoch": 3.5408973693847654e-05,
      "step": 37129,
      "training_step_time": 7.696935415267944
    },
    {
      "epoch": 3.5409927368164065e-05,
      "grad_norm": 0.11743351072072983,
      "learning_rate": 7.433506712997903e-06,
      "loss": 0.0607,
      "step": 37130
    },
    {
      "epoch": 3.5409927368164065e-05,
      "model_forward_time": 1.7433629035949707,
      "step": 37130
    },
    {
      "dynamics_loss_avg": 0.06526077259331942,
      "epoch": 3.5409927368164065e-05,
      "step": 37130
    },
    {
      "action_loss_avg": 0.002390025753993541,
      "epoch": 3.5409927368164065e-05,
      "step": 37130
    },
    {
      "epoch": 3.5409927368164065e-05,
      "step": 37130,
      "training_step_time": 7.5761330127716064
    },
    {
      "epoch": 3.541088104248047e-05,
      "model_forward_time": 1.7512691020965576,
      "step": 37131
    },
    {
      "epoch": 3.541088104248047e-05,
      "step": 37131,
      "training_step_time": 6.718987703323364
    },
    {
      "epoch": 3.541183471679687e-05,
      "model_forward_time": 1.7334537506103516,
      "step": 37132
    },
    {
      "epoch": 3.541183471679687e-05,
      "step": 37132,
      "training_step_time": 6.675179958343506
    },
    {
      "epoch": 3.5412788391113284e-05,
      "model_forward_time": 1.7478907108306885,
      "step": 37133
    },
    {
      "epoch": 3.5412788391113284e-05,
      "step": 37133,
      "training_step_time": 6.704503536224365
    },
    {
      "epoch": 3.541374206542969e-05,
      "model_forward_time": 1.7498950958251953,
      "step": 37134
    },
    {
      "epoch": 3.541374206542969e-05,
      "step": 37134,
      "training_step_time": 6.6999735832214355
    },
    {
      "epoch": 3.541469573974609e-05,
      "model_forward_time": 1.7325270175933838,
      "step": 37135
    },
    {
      "epoch": 3.541469573974609e-05,
      "step": 37135,
      "training_step_time": 6.768795490264893
    },
    {
      "epoch": 3.54156494140625e-05,
      "model_forward_time": 1.750009298324585,
      "step": 37136
    },
    {
      "epoch": 3.54156494140625e-05,
      "step": 37136,
      "training_step_time": 8.332858562469482
    },
    {
      "epoch": 3.541660308837891e-05,
      "model_forward_time": 1.0452797412872314,
      "step": 37137
    },
    {
      "epoch": 3.541660308837891e-05,
      "step": 37137,
      "training_step_time": 12.612650156021118
    },
    {
      "epoch": 3.541755676269531e-05,
      "model_forward_time": 1.7546632289886475,
      "step": 37138
    },
    {
      "epoch": 3.541755676269531e-05,
      "step": 37138,
      "training_step_time": 8.248997926712036
    },
    {
      "epoch": 3.541851043701172e-05,
      "model_forward_time": 2.9107978343963623,
      "step": 37139
    },
    {
      "epoch": 3.541851043701172e-05,
      "step": 37139,
      "training_step_time": 7.698483943939209
    },
    {
      "epoch": 3.5419464111328126e-05,
      "grad_norm": 0.14104774594306946,
      "learning_rate": 7.432062160928077e-06,
      "loss": 0.0661,
      "step": 37140
    },
    {
      "epoch": 3.5419464111328126e-05,
      "model_forward_time": 1.7437429428100586,
      "step": 37140
    },
    {
      "dynamics_loss_avg": 0.05399097446352243,
      "epoch": 3.5419464111328126e-05,
      "step": 37140
    },
    {
      "action_loss_avg": 0.002651012601563707,
      "epoch": 3.5419464111328126e-05,
      "step": 37140
    },
    {
      "epoch": 3.5419464111328126e-05,
      "step": 37140,
      "training_step_time": 7.089926242828369
    },
    {
      "epoch": 3.542041778564453e-05,
      "model_forward_time": 1.7443702220916748,
      "step": 37141
    },
    {
      "epoch": 3.542041778564453e-05,
      "step": 37141,
      "training_step_time": 6.673905849456787
    },
    {
      "epoch": 3.542137145996094e-05,
      "model_forward_time": 1.7477312088012695,
      "step": 37142
    },
    {
      "epoch": 3.542137145996094e-05,
      "step": 37142,
      "training_step_time": 6.706868410110474
    },
    {
      "epoch": 3.5422325134277344e-05,
      "model_forward_time": 1.7516934871673584,
      "step": 37143
    },
    {
      "epoch": 3.5422325134277344e-05,
      "step": 37143,
      "training_step_time": 6.709403991699219
    },
    {
      "epoch": 3.542327880859375e-05,
      "model_forward_time": 1.7413687705993652,
      "step": 37144
    },
    {
      "epoch": 3.542327880859375e-05,
      "step": 37144,
      "training_step_time": 6.71235466003418
    },
    {
      "epoch": 3.542423248291016e-05,
      "model_forward_time": 1.7478041648864746,
      "step": 37145
    },
    {
      "epoch": 3.542423248291016e-05,
      "step": 37145,
      "training_step_time": 8.246541500091553
    },
    {
      "epoch": 3.542518615722656e-05,
      "model_forward_time": 1.8049299716949463,
      "step": 37146
    },
    {
      "epoch": 3.542518615722656e-05,
      "step": 37146,
      "training_step_time": 9.63057017326355
    },
    {
      "epoch": 3.542613983154297e-05,
      "model_forward_time": 1.742020845413208,
      "step": 37147
    },
    {
      "epoch": 3.542613983154297e-05,
      "step": 37147,
      "training_step_time": 8.232675790786743
    },
    {
      "epoch": 3.542709350585938e-05,
      "model_forward_time": 2.9295644760131836,
      "step": 37148
    },
    {
      "epoch": 3.542709350585938e-05,
      "step": 37148,
      "training_step_time": 7.730175018310547
    },
    {
      "epoch": 3.542804718017578e-05,
      "model_forward_time": 1.7345690727233887,
      "step": 37149
    },
    {
      "epoch": 3.542804718017578e-05,
      "step": 37149,
      "training_step_time": 7.663037061691284
    },
    {
      "epoch": 3.5429000854492186e-05,
      "grad_norm": 0.11960840970277786,
      "learning_rate": 7.430617342891589e-06,
      "loss": 0.0583,
      "step": 37150
    },
    {
      "epoch": 3.5429000854492186e-05,
      "model_forward_time": 1.7429380416870117,
      "step": 37150
    },
    {
      "dynamics_loss_avg": 0.06068411283195019,
      "epoch": 3.5429000854492186e-05,
      "step": 37150
    },
    {
      "action_loss_avg": 0.003854745446005836,
      "epoch": 3.5429000854492186e-05,
      "step": 37150
    },
    {
      "epoch": 3.5429000854492186e-05,
      "step": 37150,
      "training_step_time": 6.7201690673828125
    },
    {
      "epoch": 3.54299545288086e-05,
      "model_forward_time": 1.0504076480865479,
      "step": 37151
    },
    {
      "epoch": 3.54299545288086e-05,
      "step": 37151,
      "training_step_time": 6.676739454269409
    },
    {
      "epoch": 3.5430908203125e-05,
      "model_forward_time": 1.7316412925720215,
      "step": 37152
    },
    {
      "epoch": 3.5430908203125e-05,
      "step": 37152,
      "training_step_time": 6.687274694442749
    },
    {
      "epoch": 3.5431861877441405e-05,
      "model_forward_time": 1.7498209476470947,
      "step": 37153
    },
    {
      "epoch": 3.5431861877441405e-05,
      "step": 37153,
      "training_step_time": 6.7127509117126465
    },
    {
      "epoch": 3.5432815551757816e-05,
      "model_forward_time": 1.0498638153076172,
      "step": 37154
    },
    {
      "epoch": 3.5432815551757816e-05,
      "step": 37154,
      "training_step_time": 8.405172109603882
    },
    {
      "epoch": 3.543376922607422e-05,
      "model_forward_time": 1.783202886581421,
      "step": 37155
    },
    {
      "epoch": 3.543376922607422e-05,
      "step": 37155,
      "training_step_time": 8.830795764923096
    },
    {
      "epoch": 3.5434722900390624e-05,
      "model_forward_time": 1.749953269958496,
      "step": 37156
    },
    {
      "epoch": 3.5434722900390624e-05,
      "step": 37156,
      "training_step_time": 9.425247192382812
    },
    {
      "epoch": 3.543567657470703e-05,
      "model_forward_time": 1.7493550777435303,
      "step": 37157
    },
    {
      "epoch": 3.543567657470703e-05,
      "step": 37157,
      "training_step_time": 7.688443899154663
    },
    {
      "epoch": 3.543663024902344e-05,
      "model_forward_time": 2.8890981674194336,
      "step": 37158
    },
    {
      "epoch": 3.543663024902344e-05,
      "step": 37158,
      "training_step_time": 7.719198942184448
    },
    {
      "epoch": 3.543758392333984e-05,
      "model_forward_time": 1.745934009552002,
      "step": 37159
    },
    {
      "epoch": 3.543758392333984e-05,
      "step": 37159,
      "training_step_time": 7.115957975387573
    },
    {
      "epoch": 3.543853759765625e-05,
      "grad_norm": 0.10925965756177902,
      "learning_rate": 7.429172259046442e-06,
      "loss": 0.0625,
      "step": 37160
    },
    {
      "epoch": 3.543853759765625e-05,
      "model_forward_time": 1.73964262008667,
      "step": 37160
    },
    {
      "dynamics_loss_avg": 0.04490624563768506,
      "epoch": 3.543853759765625e-05,
      "step": 37160
    },
    {
      "action_loss_avg": 0.0016156168989255094,
      "epoch": 3.543853759765625e-05,
      "step": 37160
    },
    {
      "epoch": 3.543853759765625e-05,
      "step": 37160,
      "training_step_time": 6.748974084854126
    },
    {
      "epoch": 3.543949127197266e-05,
      "model_forward_time": 1.7445058822631836,
      "step": 37161
    },
    {
      "epoch": 3.543949127197266e-05,
      "step": 37161,
      "training_step_time": 6.753449201583862
    },
    {
      "epoch": 3.544044494628906e-05,
      "model_forward_time": 1.7437939643859863,
      "step": 37162
    },
    {
      "epoch": 3.544044494628906e-05,
      "step": 37162,
      "training_step_time": 6.77807092666626
    },
    {
      "epoch": 3.5441398620605466e-05,
      "model_forward_time": 1.7450470924377441,
      "step": 37163
    },
    {
      "epoch": 3.5441398620605466e-05,
      "step": 37163,
      "training_step_time": 8.418397426605225
    },
    {
      "epoch": 3.5442352294921876e-05,
      "model_forward_time": 1.7480125427246094,
      "step": 37164
    },
    {
      "epoch": 3.5442352294921876e-05,
      "step": 37164,
      "training_step_time": 10.587591886520386
    },
    {
      "epoch": 3.544330596923828e-05,
      "model_forward_time": 1.745201826095581,
      "step": 37165
    },
    {
      "epoch": 3.544330596923828e-05,
      "step": 37165,
      "training_step_time": 8.551899433135986
    },
    {
      "epoch": 3.5444259643554684e-05,
      "model_forward_time": 1.7539186477661133,
      "step": 37166
    },
    {
      "epoch": 3.5444259643554684e-05,
      "step": 37166,
      "training_step_time": 8.736653089523315
    },
    {
      "epoch": 3.5445213317871095e-05,
      "model_forward_time": 2.9048454761505127,
      "step": 37167
    },
    {
      "epoch": 3.5445213317871095e-05,
      "step": 37167,
      "training_step_time": 7.702933311462402
    },
    {
      "epoch": 3.54461669921875e-05,
      "model_forward_time": 1.751452922821045,
      "step": 37168
    },
    {
      "epoch": 3.54461669921875e-05,
      "step": 37168,
      "training_step_time": 7.009960889816284
    },
    {
      "epoch": 3.54471206665039e-05,
      "model_forward_time": 1.747399091720581,
      "step": 37169
    },
    {
      "epoch": 3.54471206665039e-05,
      "step": 37169,
      "training_step_time": 6.6944739818573
    },
    {
      "epoch": 3.5448074340820314e-05,
      "grad_norm": 0.09031353145837784,
      "learning_rate": 7.42772690955067e-06,
      "loss": 0.0632,
      "step": 37170
    },
    {
      "epoch": 3.5448074340820314e-05,
      "model_forward_time": 1.7593984603881836,
      "step": 37170
    },
    {
      "dynamics_loss_avg": 0.068537232093513,
      "epoch": 3.5448074340820314e-05,
      "step": 37170
    },
    {
      "action_loss_avg": 0.002137629152275622,
      "epoch": 3.5448074340820314e-05,
      "step": 37170
    },
    {
      "epoch": 3.5448074340820314e-05,
      "step": 37170,
      "training_step_time": 6.760032653808594
    },
    {
      "epoch": 3.544902801513672e-05,
      "model_forward_time": 1.733936071395874,
      "step": 37171
    },
    {
      "epoch": 3.544902801513672e-05,
      "step": 37171,
      "training_step_time": 6.723770618438721
    },
    {
      "epoch": 3.544998168945312e-05,
      "model_forward_time": 1.7482268810272217,
      "step": 37172
    },
    {
      "epoch": 3.544998168945312e-05,
      "step": 37172,
      "training_step_time": 7.733867406845093
    },
    {
      "epoch": 3.545093536376953e-05,
      "model_forward_time": 1.050506591796875,
      "step": 37173
    },
    {
      "epoch": 3.545093536376953e-05,
      "step": 37173,
      "training_step_time": 8.799108743667603
    },
    {
      "epoch": 3.545188903808594e-05,
      "model_forward_time": 1.8202133178710938,
      "step": 37174
    },
    {
      "epoch": 3.545188903808594e-05,
      "step": 37174,
      "training_step_time": 8.848031520843506
    },
    {
      "epoch": 3.545284271240234e-05,
      "model_forward_time": 1.7460815906524658,
      "step": 37175
    },
    {
      "epoch": 3.545284271240234e-05,
      "step": 37175,
      "training_step_time": 8.754509925842285
    },
    {
      "epoch": 3.545379638671875e-05,
      "model_forward_time": 1.7393171787261963,
      "step": 37176
    },
    {
      "epoch": 3.545379638671875e-05,
      "step": 37176,
      "training_step_time": 8.64998459815979
    },
    {
      "epoch": 3.5454750061035156e-05,
      "model_forward_time": 1.7449355125427246,
      "step": 37177
    },
    {
      "epoch": 3.5454750061035156e-05,
      "step": 37177,
      "training_step_time": 7.610378742218018
    },
    {
      "epoch": 3.545570373535156e-05,
      "model_forward_time": 1.3785576820373535,
      "step": 37178
    },
    {
      "epoch": 3.545570373535156e-05,
      "step": 37178,
      "training_step_time": 6.8196563720703125
    },
    {
      "epoch": 3.545665740966797e-05,
      "model_forward_time": 1.746910572052002,
      "step": 37179
    },
    {
      "epoch": 3.545665740966797e-05,
      "step": 37179,
      "training_step_time": 6.779469966888428
    },
    {
      "epoch": 3.5457611083984375e-05,
      "grad_norm": 0.10801273584365845,
      "learning_rate": 7.426281294562332e-06,
      "loss": 0.0573,
      "step": 37180
    },
    {
      "epoch": 3.5457611083984375e-05,
      "model_forward_time": 1.7538056373596191,
      "step": 37180
    },
    {
      "dynamics_loss_avg": 0.05921022999100387,
      "epoch": 3.5457611083984375e-05,
      "step": 37180
    },
    {
      "action_loss_avg": 0.002268928047851659,
      "epoch": 3.5457611083984375e-05,
      "step": 37180
    },
    {
      "epoch": 3.5457611083984375e-05,
      "step": 37180,
      "training_step_time": 6.807650566101074
    },
    {
      "epoch": 3.545856475830078e-05,
      "model_forward_time": 1.7391152381896973,
      "step": 37181
    },
    {
      "epoch": 3.545856475830078e-05,
      "step": 37181,
      "training_step_time": 7.722281217575073
    },
    {
      "epoch": 3.545951843261719e-05,
      "model_forward_time": 1.7450284957885742,
      "step": 37182
    },
    {
      "epoch": 3.545951843261719e-05,
      "step": 37182,
      "training_step_time": 7.705116271972656
    },
    {
      "epoch": 3.5460472106933594e-05,
      "model_forward_time": 1.800398349761963,
      "step": 37183
    },
    {
      "epoch": 3.5460472106933594e-05,
      "step": 37183,
      "training_step_time": 9.287633180618286
    },
    {
      "epoch": 3.546142578125e-05,
      "model_forward_time": 1.7364206314086914,
      "step": 37184
    },
    {
      "epoch": 3.546142578125e-05,
      "step": 37184,
      "training_step_time": 7.740818023681641
    },
    {
      "epoch": 3.546237945556641e-05,
      "model_forward_time": 1.7386722564697266,
      "step": 37185
    },
    {
      "epoch": 3.546237945556641e-05,
      "step": 37185,
      "training_step_time": 7.606607675552368
    },
    {
      "epoch": 3.546333312988281e-05,
      "model_forward_time": 2.942150592803955,
      "step": 37186
    },
    {
      "epoch": 3.546333312988281e-05,
      "step": 37186,
      "training_step_time": 7.790088176727295
    },
    {
      "epoch": 3.5464286804199216e-05,
      "model_forward_time": 1.7343497276306152,
      "step": 37187
    },
    {
      "epoch": 3.5464286804199216e-05,
      "step": 37187,
      "training_step_time": 6.8485822677612305
    },
    {
      "epoch": 3.546524047851563e-05,
      "model_forward_time": 1.7390527725219727,
      "step": 37188
    },
    {
      "epoch": 3.546524047851563e-05,
      "step": 37188,
      "training_step_time": 6.777637004852295
    },
    {
      "epoch": 3.546619415283203e-05,
      "model_forward_time": 1.7362267971038818,
      "step": 37189
    },
    {
      "epoch": 3.546619415283203e-05,
      "step": 37189,
      "training_step_time": 6.7965028285980225
    },
    {
      "epoch": 3.5467147827148435e-05,
      "grad_norm": 0.12449737638235092,
      "learning_rate": 7.424835414239519e-06,
      "loss": 0.0572,
      "step": 37190
    },
    {
      "epoch": 3.5467147827148435e-05,
      "model_forward_time": 1.7278947830200195,
      "step": 37190
    },
    {
      "dynamics_loss_avg": 0.04739406481385231,
      "epoch": 3.5467147827148435e-05,
      "step": 37190
    },
    {
      "action_loss_avg": 0.001429503073450178,
      "epoch": 3.5467147827148435e-05,
      "step": 37190
    },
    {
      "epoch": 3.5467147827148435e-05,
      "step": 37190,
      "training_step_time": 7.660752058029175
    },
    {
      "epoch": 3.5468101501464846e-05,
      "model_forward_time": 1.7423126697540283,
      "step": 37191
    },
    {
      "epoch": 3.5468101501464846e-05,
      "step": 37191,
      "training_step_time": 9.203013181686401
    },
    {
      "epoch": 3.546905517578125e-05,
      "model_forward_time": 1.8048052787780762,
      "step": 37192
    },
    {
      "epoch": 3.546905517578125e-05,
      "step": 37192,
      "training_step_time": 8.542948484420776
    },
    {
      "epoch": 3.5470008850097654e-05,
      "model_forward_time": 1.7399475574493408,
      "step": 37193
    },
    {
      "epoch": 3.5470008850097654e-05,
      "step": 37193,
      "training_step_time": 8.512009859085083
    },
    {
      "epoch": 3.5470962524414065e-05,
      "model_forward_time": 1.7422497272491455,
      "step": 37194
    },
    {
      "epoch": 3.5470962524414065e-05,
      "step": 37194,
      "training_step_time": 7.706920385360718
    },
    {
      "epoch": 3.547191619873047e-05,
      "model_forward_time": 2.889789581298828,
      "step": 37195
    },
    {
      "epoch": 3.547191619873047e-05,
      "step": 37195,
      "training_step_time": 7.716718435287476
    },
    {
      "epoch": 3.547286987304687e-05,
      "model_forward_time": 1.7433850765228271,
      "step": 37196
    },
    {
      "epoch": 3.547286987304687e-05,
      "step": 37196,
      "training_step_time": 7.115092039108276
    },
    {
      "epoch": 3.5473823547363284e-05,
      "model_forward_time": 1.7461106777191162,
      "step": 37197
    },
    {
      "epoch": 3.5473823547363284e-05,
      "step": 37197,
      "training_step_time": 6.82904577255249
    },
    {
      "epoch": 3.547477722167969e-05,
      "model_forward_time": 1.7513704299926758,
      "step": 37198
    },
    {
      "epoch": 3.547477722167969e-05,
      "step": 37198,
      "training_step_time": 6.8501129150390625
    },
    {
      "epoch": 3.547573089599609e-05,
      "model_forward_time": 1.7268211841583252,
      "step": 37199
    },
    {
      "epoch": 3.547573089599609e-05,
      "step": 37199,
      "training_step_time": 7.631137371063232
    },
    {
      "epoch": 3.54766845703125e-05,
      "grad_norm": 0.12145081162452698,
      "learning_rate": 7.423389268740353e-06,
      "loss": 0.0592,
      "step": 37200
    },
    {
      "epoch": 3.54766845703125e-05,
      "model_forward_time": 1.7350518703460693,
      "step": 37200
    },
    {
      "dynamics_loss_avg": 0.058141623996198176,
      "epoch": 3.54766845703125e-05,
      "step": 37200
    },
    {
      "action_loss_avg": 0.0019190326042007656,
      "epoch": 3.54766845703125e-05,
      "step": 37200
    },
    {
      "epoch": 3.54766845703125e-05,
      "step": 37200,
      "training_step_time": 8.554619312286377
    },
    {
      "epoch": 3.547763824462891e-05,
      "model_forward_time": 1.7739849090576172,
      "step": 37201
    },
    {
      "epoch": 3.547763824462891e-05,
      "step": 37201,
      "training_step_time": 10.444334268569946
    },
    {
      "epoch": 3.547859191894531e-05,
      "model_forward_time": 1.7479889392852783,
      "step": 37202
    },
    {
      "epoch": 3.547859191894531e-05,
      "step": 37202,
      "training_step_time": 7.780155658721924
    },
    {
      "epoch": 3.547954559326172e-05,
      "model_forward_time": 1.744880199432373,
      "step": 37203
    },
    {
      "epoch": 3.547954559326172e-05,
      "step": 37203,
      "training_step_time": 7.940112113952637
    },
    {
      "epoch": 3.5480499267578126e-05,
      "model_forward_time": 2.8919830322265625,
      "step": 37204
    },
    {
      "epoch": 3.5480499267578126e-05,
      "step": 37204,
      "training_step_time": 7.73269510269165
    },
    {
      "epoch": 3.548145294189453e-05,
      "model_forward_time": 1.7392675876617432,
      "step": 37205
    },
    {
      "epoch": 3.548145294189453e-05,
      "step": 37205,
      "training_step_time": 7.103929042816162
    },
    {
      "epoch": 3.548240661621094e-05,
      "model_forward_time": 1.7457611560821533,
      "step": 37206
    },
    {
      "epoch": 3.548240661621094e-05,
      "step": 37206,
      "training_step_time": 6.825968980789185
    },
    {
      "epoch": 3.5483360290527344e-05,
      "model_forward_time": 1.727804183959961,
      "step": 37207
    },
    {
      "epoch": 3.5483360290527344e-05,
      "step": 37207,
      "training_step_time": 6.8487231731414795
    },
    {
      "epoch": 3.548431396484375e-05,
      "model_forward_time": 1.742269515991211,
      "step": 37208
    },
    {
      "epoch": 3.548431396484375e-05,
      "step": 37208,
      "training_step_time": 7.70049262046814
    },
    {
      "epoch": 3.548526763916016e-05,
      "model_forward_time": 1.7453436851501465,
      "step": 37209
    },
    {
      "epoch": 3.548526763916016e-05,
      "step": 37209,
      "training_step_time": 7.858755588531494
    },
    {
      "epoch": 3.548622131347656e-05,
      "grad_norm": 0.14301978051662445,
      "learning_rate": 7.421942858222979e-06,
      "loss": 0.0581,
      "step": 37210
    },
    {
      "epoch": 3.548622131347656e-05,
      "model_forward_time": 1.72119140625,
      "step": 37210
    },
    {
      "dynamics_loss_avg": 0.057657078839838505,
      "epoch": 3.548622131347656e-05,
      "step": 37210
    },
    {
      "action_loss_avg": 0.002037676773034036,
      "epoch": 3.548622131347656e-05,
      "step": 37210
    },
    {
      "epoch": 3.548622131347656e-05,
      "step": 37210,
      "training_step_time": 8.84218716621399
    },
    {
      "epoch": 3.548717498779297e-05,
      "model_forward_time": 1.7439754009246826,
      "step": 37211
    },
    {
      "epoch": 3.548717498779297e-05,
      "step": 37211,
      "training_step_time": 7.797541379928589
    },
    {
      "epoch": 3.548812866210938e-05,
      "model_forward_time": 1.7441565990447998,
      "step": 37212
    },
    {
      "epoch": 3.548812866210938e-05,
      "step": 37212,
      "training_step_time": 8.8300142288208
    },
    {
      "epoch": 3.548908233642578e-05,
      "model_forward_time": 2.8375327587127686,
      "step": 37213
    },
    {
      "epoch": 3.548908233642578e-05,
      "step": 37213,
      "training_step_time": 7.688526391983032
    },
    {
      "epoch": 3.5490036010742186e-05,
      "model_forward_time": 1.7418358325958252,
      "step": 37214
    },
    {
      "epoch": 3.5490036010742186e-05,
      "step": 37214,
      "training_step_time": 7.710308313369751
    },
    {
      "epoch": 3.54909896850586e-05,
      "model_forward_time": 1.727369785308838,
      "step": 37215
    },
    {
      "epoch": 3.54909896850586e-05,
      "step": 37215,
      "training_step_time": 6.867019414901733
    },
    {
      "epoch": 3.5491943359375e-05,
      "model_forward_time": 1.7462596893310547,
      "step": 37216
    },
    {
      "epoch": 3.5491943359375e-05,
      "step": 37216,
      "training_step_time": 6.905863285064697
    },
    {
      "epoch": 3.5492897033691405e-05,
      "model_forward_time": 1.7397582530975342,
      "step": 37217
    },
    {
      "epoch": 3.5492897033691405e-05,
      "step": 37217,
      "training_step_time": 7.774468660354614
    },
    {
      "epoch": 3.5493850708007816e-05,
      "model_forward_time": 1.7386665344238281,
      "step": 37218
    },
    {
      "epoch": 3.5493850708007816e-05,
      "step": 37218,
      "training_step_time": 7.796359539031982
    },
    {
      "epoch": 3.549480438232422e-05,
      "model_forward_time": 1.7392528057098389,
      "step": 37219
    },
    {
      "epoch": 3.549480438232422e-05,
      "step": 37219,
      "training_step_time": 8.696027994155884
    },
    {
      "epoch": 3.5495758056640624e-05,
      "grad_norm": 0.12275061011314392,
      "learning_rate": 7.4204961828455755e-06,
      "loss": 0.055,
      "step": 37220
    },
    {
      "epoch": 3.5495758056640624e-05,
      "model_forward_time": 1.8019585609436035,
      "step": 37220
    },
    {
      "dynamics_loss_avg": 0.055397506803274155,
      "epoch": 3.5495758056640624e-05,
      "step": 37220
    },
    {
      "action_loss_avg": 0.0034232567821163686,
      "epoch": 3.5495758056640624e-05,
      "step": 37220
    },
    {
      "epoch": 3.5495758056640624e-05,
      "step": 37220,
      "training_step_time": 8.792819023132324
    },
    {
      "epoch": 3.549671173095703e-05,
      "model_forward_time": 1.73618745803833,
      "step": 37221
    },
    {
      "epoch": 3.549671173095703e-05,
      "step": 37221,
      "training_step_time": 7.742455005645752
    },
    {
      "epoch": 3.549766540527344e-05,
      "model_forward_time": 1.7445473670959473,
      "step": 37222
    },
    {
      "epoch": 3.549766540527344e-05,
      "step": 37222,
      "training_step_time": 8.604901313781738
    },
    {
      "epoch": 3.549861907958984e-05,
      "model_forward_time": 1.7435932159423828,
      "step": 37223
    },
    {
      "epoch": 3.549861907958984e-05,
      "step": 37223,
      "training_step_time": 7.508312225341797
    },
    {
      "epoch": 3.549957275390625e-05,
      "model_forward_time": 1.7493839263916016,
      "step": 37224
    },
    {
      "epoch": 3.549957275390625e-05,
      "step": 37224,
      "training_step_time": 6.831161022186279
    },
    {
      "epoch": 3.550052642822266e-05,
      "model_forward_time": 1.7377285957336426,
      "step": 37225
    },
    {
      "epoch": 3.550052642822266e-05,
      "step": 37225,
      "training_step_time": 6.81941819190979
    },
    {
      "epoch": 3.550148010253906e-05,
      "model_forward_time": 1.733665943145752,
      "step": 37226
    },
    {
      "epoch": 3.550148010253906e-05,
      "step": 37226,
      "training_step_time": 7.628286361694336
    },
    {
      "epoch": 3.5502433776855466e-05,
      "model_forward_time": 1.7413859367370605,
      "step": 37227
    },
    {
      "epoch": 3.5502433776855466e-05,
      "step": 37227,
      "training_step_time": 8.047990560531616
    },
    {
      "epoch": 3.5503387451171876e-05,
      "model_forward_time": 1.7295258045196533,
      "step": 37228
    },
    {
      "epoch": 3.5503387451171876e-05,
      "step": 37228,
      "training_step_time": 8.280718326568604
    },
    {
      "epoch": 3.550434112548828e-05,
      "model_forward_time": 1.8112773895263672,
      "step": 37229
    },
    {
      "epoch": 3.550434112548828e-05,
      "step": 37229,
      "training_step_time": 8.745154857635498
    },
    {
      "epoch": 3.5505294799804685e-05,
      "grad_norm": 0.12808777391910553,
      "learning_rate": 7.419049242766348e-06,
      "loss": 0.0552,
      "step": 37230
    },
    {
      "epoch": 3.5505294799804685e-05,
      "model_forward_time": 1.7276766300201416,
      "step": 37230
    },
    {
      "dynamics_loss_avg": 0.07278447281569242,
      "epoch": 3.5505294799804685e-05,
      "step": 37230
    },
    {
      "action_loss_avg": 0.0020538991782814266,
      "epoch": 3.5505294799804685e-05,
      "step": 37230
    },
    {
      "epoch": 3.5505294799804685e-05,
      "step": 37230,
      "training_step_time": 7.780269384384155
    },
    {
      "epoch": 3.5506248474121095e-05,
      "model_forward_time": 1.7952461242675781,
      "step": 37231
    },
    {
      "epoch": 3.5506248474121095e-05,
      "step": 37231,
      "training_step_time": 8.428987503051758
    },
    {
      "epoch": 3.55072021484375e-05,
      "model_forward_time": 2.87713623046875,
      "step": 37232
    },
    {
      "epoch": 3.55072021484375e-05,
      "step": 37232,
      "training_step_time": 7.681808233261108
    },
    {
      "epoch": 3.55081558227539e-05,
      "model_forward_time": 1.7458817958831787,
      "step": 37233
    },
    {
      "epoch": 3.55081558227539e-05,
      "step": 37233,
      "training_step_time": 6.84776759147644
    },
    {
      "epoch": 3.5509109497070314e-05,
      "model_forward_time": 1.731776475906372,
      "step": 37234
    },
    {
      "epoch": 3.5509109497070314e-05,
      "step": 37234,
      "training_step_time": 6.859002590179443
    },
    {
      "epoch": 3.551006317138672e-05,
      "model_forward_time": 1.7508878707885742,
      "step": 37235
    },
    {
      "epoch": 3.551006317138672e-05,
      "step": 37235,
      "training_step_time": 7.6661365032196045
    },
    {
      "epoch": 3.551101684570312e-05,
      "model_forward_time": 1.7517046928405762,
      "step": 37236
    },
    {
      "epoch": 3.551101684570312e-05,
      "step": 37236,
      "training_step_time": 7.3168251514434814
    },
    {
      "epoch": 3.551197052001953e-05,
      "model_forward_time": 1.750042200088501,
      "step": 37237
    },
    {
      "epoch": 3.551197052001953e-05,
      "step": 37237,
      "training_step_time": 8.874203443527222
    },
    {
      "epoch": 3.551292419433594e-05,
      "model_forward_time": 1.7637205123901367,
      "step": 37238
    },
    {
      "epoch": 3.551292419433594e-05,
      "step": 37238,
      "training_step_time": 8.232638359069824
    },
    {
      "epoch": 3.551387786865234e-05,
      "model_forward_time": 1.7442805767059326,
      "step": 37239
    },
    {
      "epoch": 3.551387786865234e-05,
      "step": 37239,
      "training_step_time": 8.754857063293457
    },
    {
      "epoch": 3.551483154296875e-05,
      "grad_norm": 0.10371557623147964,
      "learning_rate": 7.417602038143533e-06,
      "loss": 0.0567,
      "step": 37240
    },
    {
      "epoch": 3.551483154296875e-05,
      "model_forward_time": 1.7370624542236328,
      "step": 37240
    },
    {
      "dynamics_loss_avg": 0.05411966945976019,
      "epoch": 3.551483154296875e-05,
      "step": 37240
    },
    {
      "action_loss_avg": 0.001395898543705698,
      "epoch": 3.551483154296875e-05,
      "step": 37240
    },
    {
      "epoch": 3.551483154296875e-05,
      "step": 37240,
      "training_step_time": 7.8031299114227295
    },
    {
      "epoch": 3.5515785217285156e-05,
      "model_forward_time": 2.9286577701568604,
      "step": 37241
    },
    {
      "epoch": 3.5515785217285156e-05,
      "step": 37241,
      "training_step_time": 7.748536109924316
    },
    {
      "epoch": 3.551673889160156e-05,
      "model_forward_time": 1.74713134765625,
      "step": 37242
    },
    {
      "epoch": 3.551673889160156e-05,
      "step": 37242,
      "training_step_time": 6.846662998199463
    },
    {
      "epoch": 3.551769256591797e-05,
      "model_forward_time": 1.744812250137329,
      "step": 37243
    },
    {
      "epoch": 3.551769256591797e-05,
      "step": 37243,
      "training_step_time": 6.851552963256836
    },
    {
      "epoch": 3.5518646240234375e-05,
      "model_forward_time": 1.7457704544067383,
      "step": 37244
    },
    {
      "epoch": 3.5518646240234375e-05,
      "step": 37244,
      "training_step_time": 7.664996862411499
    },
    {
      "epoch": 3.551959991455078e-05,
      "model_forward_time": 1.7485382556915283,
      "step": 37245
    },
    {
      "epoch": 3.551959991455078e-05,
      "step": 37245,
      "training_step_time": 7.429833650588989
    },
    {
      "epoch": 3.552055358886719e-05,
      "model_forward_time": 1.7483627796173096,
      "step": 37246
    },
    {
      "epoch": 3.552055358886719e-05,
      "step": 37246,
      "training_step_time": 7.711155652999878
    },
    {
      "epoch": 3.5521507263183594e-05,
      "model_forward_time": 1.7404067516326904,
      "step": 37247
    },
    {
      "epoch": 3.5521507263183594e-05,
      "step": 37247,
      "training_step_time": 8.663915872573853
    },
    {
      "epoch": 3.55224609375e-05,
      "model_forward_time": 1.8186028003692627,
      "step": 37248
    },
    {
      "epoch": 3.55224609375e-05,
      "step": 37248,
      "training_step_time": 9.426371574401855
    },
    {
      "epoch": 3.552341461181641e-05,
      "model_forward_time": 1.749840497970581,
      "step": 37249
    },
    {
      "epoch": 3.552341461181641e-05,
      "step": 37249,
      "training_step_time": 8.751991033554077
    },
    {
      "epoch": 3.552436828613281e-05,
      "grad_norm": 0.08443906158208847,
      "learning_rate": 7.416154569135393e-06,
      "loss": 0.0588,
      "step": 37250
    },
    {
      "epoch": 3.552436828613281e-05,
      "model_forward_time": 2.9011950492858887,
      "step": 37250
    },
    {
      "dynamics_loss_avg": 0.07545317895710468,
      "epoch": 3.552436828613281e-05,
      "step": 37250
    },
    {
      "action_loss_avg": 0.0024690595804713667,
      "epoch": 3.552436828613281e-05,
      "step": 37250
    },
    {
      "epoch": 3.552436828613281e-05,
      "step": 37250,
      "training_step_time": 7.752447843551636
    },
    {
      "epoch": 3.5525321960449217e-05,
      "model_forward_time": 1.7423632144927979,
      "step": 37251
    },
    {
      "epoch": 3.5525321960449217e-05,
      "step": 37251,
      "training_step_time": 7.663838624954224
    },
    {
      "epoch": 3.552627563476563e-05,
      "model_forward_time": 1.0568087100982666,
      "step": 37252
    },
    {
      "epoch": 3.552627563476563e-05,
      "step": 37252,
      "training_step_time": 6.898057460784912
    },
    {
      "epoch": 3.552722930908203e-05,
      "model_forward_time": 1.7524409294128418,
      "step": 37253
    },
    {
      "epoch": 3.552722930908203e-05,
      "step": 37253,
      "training_step_time": 6.8729822635650635
    },
    {
      "epoch": 3.5528182983398435e-05,
      "model_forward_time": 1.7424461841583252,
      "step": 37254
    },
    {
      "epoch": 3.5528182983398435e-05,
      "step": 37254,
      "training_step_time": 7.715785264968872
    },
    {
      "epoch": 3.5529136657714846e-05,
      "model_forward_time": 1.7419846057891846,
      "step": 37255
    },
    {
      "epoch": 3.5529136657714846e-05,
      "step": 37255,
      "training_step_time": 7.787862300872803
    },
    {
      "epoch": 3.553009033203125e-05,
      "model_forward_time": 1.744091272354126,
      "step": 37256
    },
    {
      "epoch": 3.553009033203125e-05,
      "step": 37256,
      "training_step_time": 10.48042893409729
    },
    {
      "epoch": 3.5531044006347654e-05,
      "model_forward_time": 1.4583182334899902,
      "step": 37257
    },
    {
      "epoch": 3.5531044006347654e-05,
      "step": 37257,
      "training_step_time": 10.488990306854248
    },
    {
      "epoch": 3.5531997680664065e-05,
      "model_forward_time": 1.7471153736114502,
      "step": 37258
    },
    {
      "epoch": 3.5531997680664065e-05,
      "step": 37258,
      "training_step_time": 7.763288736343384
    },
    {
      "epoch": 3.553295135498047e-05,
      "model_forward_time": 2.906489610671997,
      "step": 37259
    },
    {
      "epoch": 3.553295135498047e-05,
      "step": 37259,
      "training_step_time": 7.709954738616943
    },
    {
      "epoch": 3.553390502929687e-05,
      "grad_norm": 0.13787852227687836,
      "learning_rate": 7.414706835900223e-06,
      "loss": 0.0597,
      "step": 37260
    },
    {
      "epoch": 3.553390502929687e-05,
      "model_forward_time": 1.742546558380127,
      "step": 37260
    },
    {
      "dynamics_loss_avg": 0.06398998349905013,
      "epoch": 3.553390502929687e-05,
      "step": 37260
    },
    {
      "action_loss_avg": 0.0019990773522295057,
      "epoch": 3.553390502929687e-05,
      "step": 37260
    },
    {
      "epoch": 3.553390502929687e-05,
      "step": 37260,
      "training_step_time": 7.66875958442688
    },
    {
      "epoch": 3.5534858703613284e-05,
      "model_forward_time": 1.7395317554473877,
      "step": 37261
    },
    {
      "epoch": 3.5534858703613284e-05,
      "step": 37261,
      "training_step_time": 6.896515369415283
    },
    {
      "epoch": 3.553581237792969e-05,
      "model_forward_time": 1.7429394721984863,
      "step": 37262
    },
    {
      "epoch": 3.553581237792969e-05,
      "step": 37262,
      "training_step_time": 6.896366596221924
    },
    {
      "epoch": 3.553676605224609e-05,
      "model_forward_time": 1.7275316715240479,
      "step": 37263
    },
    {
      "epoch": 3.553676605224609e-05,
      "step": 37263,
      "training_step_time": 8.461957931518555
    },
    {
      "epoch": 3.55377197265625e-05,
      "model_forward_time": 1.745177984237671,
      "step": 37264
    },
    {
      "epoch": 3.55377197265625e-05,
      "step": 37264,
      "training_step_time": 7.7009971141815186
    },
    {
      "epoch": 3.553867340087891e-05,
      "model_forward_time": 1.7357261180877686,
      "step": 37265
    },
    {
      "epoch": 3.553867340087891e-05,
      "step": 37265,
      "training_step_time": 9.036747694015503
    },
    {
      "epoch": 3.553962707519531e-05,
      "model_forward_time": 1.8003811836242676,
      "step": 37266
    },
    {
      "epoch": 3.553962707519531e-05,
      "step": 37266,
      "training_step_time": 10.193977355957031
    },
    {
      "epoch": 3.554058074951172e-05,
      "model_forward_time": 1.7404329776763916,
      "step": 37267
    },
    {
      "epoch": 3.554058074951172e-05,
      "step": 37267,
      "training_step_time": 7.753632307052612
    },
    {
      "epoch": 3.5541534423828126e-05,
      "model_forward_time": 2.8372132778167725,
      "step": 37268
    },
    {
      "epoch": 3.5541534423828126e-05,
      "step": 37268,
      "training_step_time": 7.643425703048706
    },
    {
      "epoch": 3.554248809814453e-05,
      "model_forward_time": 1.7341890335083008,
      "step": 37269
    },
    {
      "epoch": 3.554248809814453e-05,
      "step": 37269,
      "training_step_time": 7.583624601364136
    },
    {
      "epoch": 3.554344177246094e-05,
      "grad_norm": 0.13052214682102203,
      "learning_rate": 7.413258838596344e-06,
      "loss": 0.0611,
      "step": 37270
    },
    {
      "epoch": 3.554344177246094e-05,
      "model_forward_time": 1.7260165214538574,
      "step": 37270
    },
    {
      "dynamics_loss_avg": 0.05129725895822048,
      "epoch": 3.554344177246094e-05,
      "step": 37270
    },
    {
      "action_loss_avg": 0.0016186847176868468,
      "epoch": 3.554344177246094e-05,
      "step": 37270
    },
    {
      "epoch": 3.554344177246094e-05,
      "step": 37270,
      "training_step_time": 6.774264812469482
    },
    {
      "epoch": 3.5544395446777345e-05,
      "model_forward_time": 1.7403244972229004,
      "step": 37271
    },
    {
      "epoch": 3.5544395446777345e-05,
      "step": 37271,
      "training_step_time": 6.8709423542022705
    },
    {
      "epoch": 3.554534912109375e-05,
      "model_forward_time": 1.7390618324279785,
      "step": 37272
    },
    {
      "epoch": 3.554534912109375e-05,
      "step": 37272,
      "training_step_time": 7.898705720901489
    },
    {
      "epoch": 3.554630279541016e-05,
      "model_forward_time": 1.7346303462982178,
      "step": 37273
    },
    {
      "epoch": 3.554630279541016e-05,
      "step": 37273,
      "training_step_time": 8.684970378875732
    },
    {
      "epoch": 3.554725646972656e-05,
      "model_forward_time": 1.7571163177490234,
      "step": 37274
    },
    {
      "epoch": 3.554725646972656e-05,
      "step": 37274,
      "training_step_time": 8.824559450149536
    },
    {
      "epoch": 3.554821014404297e-05,
      "model_forward_time": 1.7879624366760254,
      "step": 37275
    },
    {
      "epoch": 3.554821014404297e-05,
      "step": 37275,
      "training_step_time": 9.52187466621399
    },
    {
      "epoch": 3.554916381835938e-05,
      "model_forward_time": 1.7125182151794434,
      "step": 37276
    },
    {
      "epoch": 3.554916381835938e-05,
      "step": 37276,
      "training_step_time": 8.80210566520691
    },
    {
      "epoch": 3.555011749267578e-05,
      "model_forward_time": 1.73366117477417,
      "step": 37277
    },
    {
      "epoch": 3.555011749267578e-05,
      "step": 37277,
      "training_step_time": 7.87633752822876
    },
    {
      "epoch": 3.5551071166992186e-05,
      "model_forward_time": 2.8714401721954346,
      "step": 37278
    },
    {
      "epoch": 3.5551071166992186e-05,
      "step": 37278,
      "training_step_time": 7.7072813510894775
    },
    {
      "epoch": 3.55520248413086e-05,
      "model_forward_time": 1.7303924560546875,
      "step": 37279
    },
    {
      "epoch": 3.55520248413086e-05,
      "step": 37279,
      "training_step_time": 7.136225700378418
    },
    {
      "epoch": 3.5552978515625e-05,
      "grad_norm": 0.1490931659936905,
      "learning_rate": 7.4118105773821055e-06,
      "loss": 0.0608,
      "step": 37280
    },
    {
      "epoch": 3.5552978515625e-05,
      "model_forward_time": 1.7228996753692627,
      "step": 37280
    },
    {
      "dynamics_loss_avg": 0.06127760261297226,
      "epoch": 3.5552978515625e-05,
      "step": 37280
    },
    {
      "action_loss_avg": 0.0027720012702047827,
      "epoch": 3.5552978515625e-05,
      "step": 37280
    },
    {
      "epoch": 3.5552978515625e-05,
      "step": 37280,
      "training_step_time": 6.923059940338135
    },
    {
      "epoch": 3.5553932189941405e-05,
      "model_forward_time": 1.7088429927825928,
      "step": 37281
    },
    {
      "epoch": 3.5553932189941405e-05,
      "step": 37281,
      "training_step_time": 6.83386754989624
    },
    {
      "epoch": 3.555488586425781e-05,
      "model_forward_time": 1.7460601329803467,
      "step": 37282
    },
    {
      "epoch": 3.555488586425781e-05,
      "step": 37282,
      "training_step_time": 7.663716793060303
    },
    {
      "epoch": 3.555583953857422e-05,
      "model_forward_time": 1.7403693199157715,
      "step": 37283
    },
    {
      "epoch": 3.555583953857422e-05,
      "step": 37283,
      "training_step_time": 8.414525032043457
    },
    {
      "epoch": 3.5556793212890624e-05,
      "model_forward_time": 1.7441084384918213,
      "step": 37284
    },
    {
      "epoch": 3.5556793212890624e-05,
      "step": 37284,
      "training_step_time": 8.515330791473389
    },
    {
      "epoch": 3.555774688720703e-05,
      "model_forward_time": 1.7384638786315918,
      "step": 37285
    },
    {
      "epoch": 3.555774688720703e-05,
      "step": 37285,
      "training_step_time": 7.714922189712524
    },
    {
      "epoch": 3.555870056152344e-05,
      "model_forward_time": 1.7350614070892334,
      "step": 37286
    },
    {
      "epoch": 3.555870056152344e-05,
      "step": 37286,
      "training_step_time": 7.803531169891357
    },
    {
      "epoch": 3.555965423583984e-05,
      "model_forward_time": 2.919011116027832,
      "step": 37287
    },
    {
      "epoch": 3.555965423583984e-05,
      "step": 37287,
      "training_step_time": 7.7522313594818115
    },
    {
      "epoch": 3.556060791015625e-05,
      "model_forward_time": 1.745617389678955,
      "step": 37288
    },
    {
      "epoch": 3.556060791015625e-05,
      "step": 37288,
      "training_step_time": 7.588202238082886
    },
    {
      "epoch": 3.556156158447266e-05,
      "model_forward_time": 1.7330071926116943,
      "step": 37289
    },
    {
      "epoch": 3.556156158447266e-05,
      "step": 37289,
      "training_step_time": 6.828861474990845
    },
    {
      "epoch": 3.556251525878906e-05,
      "grad_norm": 0.14817999303340912,
      "learning_rate": 7.410362052415889e-06,
      "loss": 0.061,
      "step": 37290
    },
    {
      "epoch": 3.556251525878906e-05,
      "model_forward_time": 1.7200453281402588,
      "step": 37290
    },
    {
      "dynamics_loss_avg": 0.054548425041139124,
      "epoch": 3.556251525878906e-05,
      "step": 37290
    },
    {
      "action_loss_avg": 0.0029851255589164794,
      "epoch": 3.556251525878906e-05,
      "step": 37290
    },
    {
      "epoch": 3.556251525878906e-05,
      "step": 37290,
      "training_step_time": 6.886180877685547
    },
    {
      "epoch": 3.5563468933105466e-05,
      "model_forward_time": 1.372586727142334,
      "step": 37291
    },
    {
      "epoch": 3.5563468933105466e-05,
      "step": 37291,
      "training_step_time": 7.665334463119507
    },
    {
      "epoch": 3.5564422607421877e-05,
      "model_forward_time": 1.7344226837158203,
      "step": 37292
    },
    {
      "epoch": 3.5564422607421877e-05,
      "step": 37292,
      "training_step_time": 8.302862882614136
    },
    {
      "epoch": 3.556537628173828e-05,
      "model_forward_time": 1.7435719966888428,
      "step": 37293
    },
    {
      "epoch": 3.556537628173828e-05,
      "step": 37293,
      "training_step_time": 9.4765465259552
    },
    {
      "epoch": 3.5566329956054685e-05,
      "model_forward_time": 1.8142974376678467,
      "step": 37294
    },
    {
      "epoch": 3.5566329956054685e-05,
      "step": 37294,
      "training_step_time": 8.851995706558228
    },
    {
      "epoch": 3.5567283630371095e-05,
      "model_forward_time": 1.7299482822418213,
      "step": 37295
    },
    {
      "epoch": 3.5567283630371095e-05,
      "step": 37295,
      "training_step_time": 7.786790370941162
    },
    {
      "epoch": 3.55682373046875e-05,
      "model_forward_time": 2.9166901111602783,
      "step": 37296
    },
    {
      "epoch": 3.55682373046875e-05,
      "step": 37296,
      "training_step_time": 7.794204950332642
    },
    {
      "epoch": 3.5569190979003903e-05,
      "model_forward_time": 1.7507033348083496,
      "step": 37297
    },
    {
      "epoch": 3.5569190979003903e-05,
      "step": 37297,
      "training_step_time": 7.614195346832275
    },
    {
      "epoch": 3.5570144653320314e-05,
      "model_forward_time": 1.7423255443572998,
      "step": 37298
    },
    {
      "epoch": 3.5570144653320314e-05,
      "step": 37298,
      "training_step_time": 6.959834814071655
    },
    {
      "epoch": 3.557109832763672e-05,
      "model_forward_time": 1.7371182441711426,
      "step": 37299
    },
    {
      "epoch": 3.557109832763672e-05,
      "step": 37299,
      "training_step_time": 6.953113317489624
    },
    {
      "epoch": 3.557205200195312e-05,
      "grad_norm": 0.1914225071668625,
      "learning_rate": 7.408913263856102e-06,
      "loss": 0.0633,
      "step": 37300
    },
    {
      "epoch": 3.557205200195312e-05,
      "model_forward_time": 1.7282166481018066,
      "step": 37300
    },
    {
      "dynamics_loss_avg": 0.05459222141653299,
      "epoch": 3.557205200195312e-05,
      "step": 37300
    },
    {
      "action_loss_avg": 0.002672496336163022,
      "epoch": 3.557205200195312e-05,
      "step": 37300
    },
    {
      "epoch": 3.557205200195312e-05,
      "step": 37300,
      "training_step_time": 7.6747729778289795
    },
    {
      "epoch": 3.557300567626953e-05,
      "model_forward_time": 1.7528002262115479,
      "step": 37301
    },
    {
      "epoch": 3.557300567626953e-05,
      "step": 37301,
      "training_step_time": 7.7240118980407715
    },
    {
      "epoch": 3.557395935058594e-05,
      "model_forward_time": 1.7245020866394043,
      "step": 37302
    },
    {
      "epoch": 3.557395935058594e-05,
      "step": 37302,
      "training_step_time": 8.681931018829346
    },
    {
      "epoch": 3.557491302490234e-05,
      "model_forward_time": 1.7982308864593506,
      "step": 37303
    },
    {
      "epoch": 3.557491302490234e-05,
      "step": 37303,
      "training_step_time": 10.212260007858276
    },
    {
      "epoch": 3.557586669921875e-05,
      "model_forward_time": 1.7443220615386963,
      "step": 37304
    },
    {
      "epoch": 3.557586669921875e-05,
      "step": 37304,
      "training_step_time": 8.771506071090698
    },
    {
      "epoch": 3.5576820373535156e-05,
      "model_forward_time": 2.4824185371398926,
      "step": 37305
    },
    {
      "epoch": 3.5576820373535156e-05,
      "step": 37305,
      "training_step_time": 7.326799154281616
    },
    {
      "epoch": 3.557777404785156e-05,
      "model_forward_time": 1.7422583103179932,
      "step": 37306
    },
    {
      "epoch": 3.557777404785156e-05,
      "step": 37306,
      "training_step_time": 8.150091648101807
    },
    {
      "epoch": 3.557872772216797e-05,
      "model_forward_time": 1.793433666229248,
      "step": 37307
    },
    {
      "epoch": 3.557872772216797e-05,
      "step": 37307,
      "training_step_time": 6.917720079421997
    },
    {
      "epoch": 3.5579681396484375e-05,
      "model_forward_time": 1.740426778793335,
      "step": 37308
    },
    {
      "epoch": 3.5579681396484375e-05,
      "step": 37308,
      "training_step_time": 6.895293474197388
    },
    {
      "epoch": 3.558063507080078e-05,
      "model_forward_time": 1.740004062652588,
      "step": 37309
    },
    {
      "epoch": 3.558063507080078e-05,
      "step": 37309,
      "training_step_time": 7.685060739517212
    },
    {
      "epoch": 3.558158874511719e-05,
      "grad_norm": 0.12347828596830368,
      "learning_rate": 7.407464211861183e-06,
      "loss": 0.0643,
      "step": 37310
    },
    {
      "epoch": 3.558158874511719e-05,
      "model_forward_time": 1.7238130569458008,
      "step": 37310
    },
    {
      "dynamics_loss_avg": 0.050594020262360574,
      "epoch": 3.558158874511719e-05,
      "step": 37310
    },
    {
      "action_loss_avg": 0.0018546396458987147,
      "epoch": 3.558158874511719e-05,
      "step": 37310
    },
    {
      "epoch": 3.558158874511719e-05,
      "step": 37310,
      "training_step_time": 8.839120864868164
    },
    {
      "epoch": 3.5582542419433594e-05,
      "model_forward_time": 1.735008716583252,
      "step": 37311
    },
    {
      "epoch": 3.5582542419433594e-05,
      "step": 37311,
      "training_step_time": 8.911799907684326
    },
    {
      "epoch": 3.558349609375e-05,
      "model_forward_time": 1.7559361457824707,
      "step": 37312
    },
    {
      "epoch": 3.558349609375e-05,
      "step": 37312,
      "training_step_time": 8.21955418586731
    },
    {
      "epoch": 3.558444976806641e-05,
      "model_forward_time": 1.7336523532867432,
      "step": 37313
    },
    {
      "epoch": 3.558444976806641e-05,
      "step": 37313,
      "training_step_time": 7.757205009460449
    },
    {
      "epoch": 3.558540344238281e-05,
      "model_forward_time": 1.737994909286499,
      "step": 37314
    },
    {
      "epoch": 3.558540344238281e-05,
      "step": 37314,
      "training_step_time": 7.848046779632568
    },
    {
      "epoch": 3.5586357116699217e-05,
      "model_forward_time": 2.8965797424316406,
      "step": 37315
    },
    {
      "epoch": 3.5586357116699217e-05,
      "step": 37315,
      "training_step_time": 7.741151571273804
    },
    {
      "epoch": 3.558731079101563e-05,
      "model_forward_time": 1.7230207920074463,
      "step": 37316
    },
    {
      "epoch": 3.558731079101563e-05,
      "step": 37316,
      "training_step_time": 7.678409814834595
    },
    {
      "epoch": 3.558826446533203e-05,
      "model_forward_time": 1.7505276203155518,
      "step": 37317
    },
    {
      "epoch": 3.558826446533203e-05,
      "step": 37317,
      "training_step_time": 6.9115612506866455
    },
    {
      "epoch": 3.5589218139648435e-05,
      "model_forward_time": 1.7472987174987793,
      "step": 37318
    },
    {
      "epoch": 3.5589218139648435e-05,
      "step": 37318,
      "training_step_time": 7.720225811004639
    },
    {
      "epoch": 3.5590171813964846e-05,
      "model_forward_time": 1.7286384105682373,
      "step": 37319
    },
    {
      "epoch": 3.5590171813964846e-05,
      "step": 37319,
      "training_step_time": 7.826465606689453
    },
    {
      "epoch": 3.559112548828125e-05,
      "grad_norm": 0.1441042423248291,
      "learning_rate": 7.406014896589597e-06,
      "loss": 0.0611,
      "step": 37320
    },
    {
      "epoch": 3.559112548828125e-05,
      "model_forward_time": 1.7527239322662354,
      "step": 37320
    },
    {
      "dynamics_loss_avg": 0.0512022040784359,
      "epoch": 3.559112548828125e-05,
      "step": 37320
    },
    {
      "action_loss_avg": 0.0029501599055947736,
      "epoch": 3.559112548828125e-05,
      "step": 37320
    },
    {
      "epoch": 3.559112548828125e-05,
      "step": 37320,
      "training_step_time": 8.222200632095337
    },
    {
      "epoch": 3.5592079162597654e-05,
      "model_forward_time": 1.7506065368652344,
      "step": 37321
    },
    {
      "epoch": 3.5592079162597654e-05,
      "step": 37321,
      "training_step_time": 8.64790153503418
    },
    {
      "epoch": 3.5593032836914065e-05,
      "model_forward_time": 1.825132131576538,
      "step": 37322
    },
    {
      "epoch": 3.5593032836914065e-05,
      "step": 37322,
      "training_step_time": 8.49316120147705
    },
    {
      "epoch": 3.559398651123047e-05,
      "model_forward_time": 1.7506916522979736,
      "step": 37323
    },
    {
      "epoch": 3.559398651123047e-05,
      "step": 37323,
      "training_step_time": 7.809494733810425
    },
    {
      "epoch": 3.559494018554687e-05,
      "model_forward_time": 3.007535934448242,
      "step": 37324
    },
    {
      "epoch": 3.559494018554687e-05,
      "step": 37324,
      "training_step_time": 7.815277576446533
    },
    {
      "epoch": 3.5595893859863284e-05,
      "model_forward_time": 1.7668070793151855,
      "step": 37325
    },
    {
      "epoch": 3.5595893859863284e-05,
      "step": 37325,
      "training_step_time": 7.588131666183472
    },
    {
      "epoch": 3.559684753417969e-05,
      "model_forward_time": 1.750196933746338,
      "step": 37326
    },
    {
      "epoch": 3.559684753417969e-05,
      "step": 37326,
      "training_step_time": 6.770883560180664
    },
    {
      "epoch": 3.559780120849609e-05,
      "model_forward_time": 1.755596399307251,
      "step": 37327
    },
    {
      "epoch": 3.559780120849609e-05,
      "step": 37327,
      "training_step_time": 7.604456901550293
    },
    {
      "epoch": 3.55987548828125e-05,
      "model_forward_time": 1.7519755363464355,
      "step": 37328
    },
    {
      "epoch": 3.55987548828125e-05,
      "step": 37328,
      "training_step_time": 7.747230052947998
    },
    {
      "epoch": 3.559970855712891e-05,
      "model_forward_time": 1.7606067657470703,
      "step": 37329
    },
    {
      "epoch": 3.559970855712891e-05,
      "step": 37329,
      "training_step_time": 8.164683818817139
    },
    {
      "epoch": 3.560066223144531e-05,
      "grad_norm": 0.18634644150733948,
      "learning_rate": 7.404565318199839e-06,
      "loss": 0.0583,
      "step": 37330
    },
    {
      "epoch": 3.560066223144531e-05,
      "model_forward_time": 1.736938238143921,
      "step": 37330
    },
    {
      "dynamics_loss_avg": 0.05373321156948805,
      "epoch": 3.560066223144531e-05,
      "step": 37330
    },
    {
      "action_loss_avg": 0.0017554148944327608,
      "epoch": 3.560066223144531e-05,
      "step": 37330
    },
    {
      "epoch": 3.560066223144531e-05,
      "step": 37330,
      "training_step_time": 7.7731711864471436
    },
    {
      "epoch": 3.560161590576172e-05,
      "model_forward_time": 1.8282668590545654,
      "step": 37331
    },
    {
      "epoch": 3.560161590576172e-05,
      "step": 37331,
      "training_step_time": 7.719336271286011
    },
    {
      "epoch": 3.5602569580078126e-05,
      "model_forward_time": 1.775672197341919,
      "step": 37332
    },
    {
      "epoch": 3.5602569580078126e-05,
      "step": 37332,
      "training_step_time": 8.79247236251831
    },
    {
      "epoch": 3.560352325439453e-05,
      "model_forward_time": 2.9305307865142822,
      "step": 37333
    },
    {
      "epoch": 3.560352325439453e-05,
      "step": 37333,
      "training_step_time": 7.788856506347656
    },
    {
      "epoch": 3.560447692871094e-05,
      "model_forward_time": 1.7538869380950928,
      "step": 37334
    },
    {
      "epoch": 3.560447692871094e-05,
      "step": 37334,
      "training_step_time": 7.598642349243164
    },
    {
      "epoch": 3.5605430603027345e-05,
      "model_forward_time": 1.7390005588531494,
      "step": 37335
    },
    {
      "epoch": 3.5605430603027345e-05,
      "step": 37335,
      "training_step_time": 6.8337132930755615
    },
    {
      "epoch": 3.560638427734375e-05,
      "model_forward_time": 1.7398545742034912,
      "step": 37336
    },
    {
      "epoch": 3.560638427734375e-05,
      "step": 37336,
      "training_step_time": 7.986873388290405
    },
    {
      "epoch": 3.560733795166016e-05,
      "model_forward_time": 1.7477796077728271,
      "step": 37337
    },
    {
      "epoch": 3.560733795166016e-05,
      "step": 37337,
      "training_step_time": 8.560120105743408
    },
    {
      "epoch": 3.5608291625976563e-05,
      "model_forward_time": 1.7366116046905518,
      "step": 37338
    },
    {
      "epoch": 3.5608291625976563e-05,
      "step": 37338,
      "training_step_time": 7.707841873168945
    },
    {
      "epoch": 3.560924530029297e-05,
      "model_forward_time": 1.7389247417449951,
      "step": 37339
    },
    {
      "epoch": 3.560924530029297e-05,
      "step": 37339,
      "training_step_time": 7.6916704177856445
    },
    {
      "epoch": 3.561019897460938e-05,
      "grad_norm": 0.10973533987998962,
      "learning_rate": 7.403115476850433e-06,
      "loss": 0.0616,
      "step": 37340
    },
    {
      "epoch": 3.561019897460938e-05,
      "model_forward_time": 1.1258721351623535,
      "step": 37340
    },
    {
      "dynamics_loss_avg": 0.06885577328503131,
      "epoch": 3.561019897460938e-05,
      "step": 37340
    },
    {
      "action_loss_avg": 0.003165600704960525,
      "epoch": 3.561019897460938e-05,
      "step": 37340
    },
    {
      "epoch": 3.561019897460938e-05,
      "step": 37340,
      "training_step_time": 10.564373254776001
    },
    {
      "epoch": 3.561115264892578e-05,
      "model_forward_time": 1.7634570598602295,
      "step": 37341
    },
    {
      "epoch": 3.561115264892578e-05,
      "step": 37341,
      "training_step_time": 7.737402677536011
    },
    {
      "epoch": 3.5612106323242186e-05,
      "model_forward_time": 2.8422799110412598,
      "step": 37342
    },
    {
      "epoch": 3.5612106323242186e-05,
      "step": 37342,
      "training_step_time": 7.653874635696411
    },
    {
      "epoch": 3.56130599975586e-05,
      "model_forward_time": 1.7475099563598633,
      "step": 37343
    },
    {
      "epoch": 3.56130599975586e-05,
      "step": 37343,
      "training_step_time": 8.551572322845459
    },
    {
      "epoch": 3.5614013671875e-05,
      "model_forward_time": 1.7511827945709229,
      "step": 37344
    },
    {
      "epoch": 3.5614013671875e-05,
      "step": 37344,
      "training_step_time": 6.839285850524902
    },
    {
      "epoch": 3.5614967346191405e-05,
      "model_forward_time": 1.7434532642364502,
      "step": 37345
    },
    {
      "epoch": 3.5614967346191405e-05,
      "step": 37345,
      "training_step_time": 7.139914035797119
    },
    {
      "epoch": 3.561592102050781e-05,
      "model_forward_time": 1.7414309978485107,
      "step": 37346
    },
    {
      "epoch": 3.561592102050781e-05,
      "step": 37346,
      "training_step_time": 8.654819250106812
    },
    {
      "epoch": 3.561687469482422e-05,
      "model_forward_time": 1.7427952289581299,
      "step": 37347
    },
    {
      "epoch": 3.561687469482422e-05,
      "step": 37347,
      "training_step_time": 7.6992738246917725
    },
    {
      "epoch": 3.5617828369140624e-05,
      "model_forward_time": 1.7446074485778809,
      "step": 37348
    },
    {
      "epoch": 3.5617828369140624e-05,
      "step": 37348,
      "training_step_time": 8.28272271156311
    },
    {
      "epoch": 3.561878204345703e-05,
      "model_forward_time": 1.8153407573699951,
      "step": 37349
    },
    {
      "epoch": 3.561878204345703e-05,
      "step": 37349,
      "training_step_time": 8.298811912536621
    },
    {
      "epoch": 3.561973571777344e-05,
      "grad_norm": 0.12157069146633148,
      "learning_rate": 7.401665372699932e-06,
      "loss": 0.0619,
      "step": 37350
    },
    {
      "epoch": 3.561973571777344e-05,
      "model_forward_time": 1.7688636779785156,
      "step": 37350
    },
    {
      "dynamics_loss_avg": 0.07329637687653304,
      "epoch": 3.561973571777344e-05,
      "step": 37350
    },
    {
      "action_loss_avg": 0.003611182025633752,
      "epoch": 3.561973571777344e-05,
      "step": 37350
    },
    {
      "epoch": 3.561973571777344e-05,
      "step": 37350,
      "training_step_time": 8.287604808807373
    },
    {
      "epoch": 3.562068939208984e-05,
      "model_forward_time": 1.7478008270263672,
      "step": 37351
    },
    {
      "epoch": 3.562068939208984e-05,
      "step": 37351,
      "training_step_time": 7.419503450393677
    },
    {
      "epoch": 3.562164306640625e-05,
      "model_forward_time": 2.9288809299468994,
      "step": 37352
    },
    {
      "epoch": 3.562164306640625e-05,
      "step": 37352,
      "training_step_time": 7.736719608306885
    },
    {
      "epoch": 3.562259674072266e-05,
      "model_forward_time": 1.7617716789245605,
      "step": 37353
    },
    {
      "epoch": 3.562259674072266e-05,
      "step": 37353,
      "training_step_time": 7.5694169998168945
    },
    {
      "epoch": 3.562355041503906e-05,
      "model_forward_time": 1.7416963577270508,
      "step": 37354
    },
    {
      "epoch": 3.562355041503906e-05,
      "step": 37354,
      "training_step_time": 7.102862358093262
    },
    {
      "epoch": 3.5624504089355466e-05,
      "model_forward_time": 1.749091625213623,
      "step": 37355
    },
    {
      "epoch": 3.5624504089355466e-05,
      "step": 37355,
      "training_step_time": 7.614173650741577
    },
    {
      "epoch": 3.5625457763671877e-05,
      "model_forward_time": 1.699666976928711,
      "step": 37356
    },
    {
      "epoch": 3.5625457763671877e-05,
      "step": 37356,
      "training_step_time": 7.7042882442474365
    },
    {
      "epoch": 3.562641143798828e-05,
      "model_forward_time": 1.7389187812805176,
      "step": 37357
    },
    {
      "epoch": 3.562641143798828e-05,
      "step": 37357,
      "training_step_time": 10.261646270751953
    },
    {
      "epoch": 3.5627365112304685e-05,
      "model_forward_time": 1.0345747470855713,
      "step": 37358
    },
    {
      "epoch": 3.5627365112304685e-05,
      "step": 37358,
      "training_step_time": 9.398375749588013
    },
    {
      "epoch": 3.5628318786621095e-05,
      "model_forward_time": 1.7433507442474365,
      "step": 37359
    },
    {
      "epoch": 3.5628318786621095e-05,
      "step": 37359,
      "training_step_time": 8.320765256881714
    },
    {
      "epoch": 3.56292724609375e-05,
      "grad_norm": 0.1106734424829483,
      "learning_rate": 7.4002150059069165e-06,
      "loss": 0.0638,
      "step": 37360
    },
    {
      "epoch": 3.56292724609375e-05,
      "model_forward_time": 1.7434492111206055,
      "step": 37360
    },
    {
      "dynamics_loss_avg": 0.0720695959404111,
      "epoch": 3.56292724609375e-05,
      "step": 37360
    },
    {
      "action_loss_avg": 0.002678379142889753,
      "epoch": 3.56292724609375e-05,
      "step": 37360
    },
    {
      "epoch": 3.56292724609375e-05,
      "step": 37360,
      "training_step_time": 7.766395568847656
    },
    {
      "epoch": 3.5630226135253903e-05,
      "model_forward_time": 2.8809814453125,
      "step": 37361
    },
    {
      "epoch": 3.5630226135253903e-05,
      "step": 37361,
      "training_step_time": 7.6876749992370605
    },
    {
      "epoch": 3.5631179809570314e-05,
      "model_forward_time": 1.7311627864837646,
      "step": 37362
    },
    {
      "epoch": 3.5631179809570314e-05,
      "step": 37362,
      "training_step_time": 7.618126630783081
    },
    {
      "epoch": 3.563213348388672e-05,
      "model_forward_time": 1.736382007598877,
      "step": 37363
    },
    {
      "epoch": 3.563213348388672e-05,
      "step": 37363,
      "training_step_time": 6.8416712284088135
    },
    {
      "epoch": 3.563308715820312e-05,
      "model_forward_time": 1.7375688552856445,
      "step": 37364
    },
    {
      "epoch": 3.563308715820312e-05,
      "step": 37364,
      "training_step_time": 7.4422173500061035
    },
    {
      "epoch": 3.563404083251953e-05,
      "model_forward_time": 1.7304916381835938,
      "step": 37365
    },
    {
      "epoch": 3.563404083251953e-05,
      "step": 37365,
      "training_step_time": 7.76211404800415
    },
    {
      "epoch": 3.563499450683594e-05,
      "model_forward_time": 1.7318625450134277,
      "step": 37366
    },
    {
      "epoch": 3.563499450683594e-05,
      "step": 37366,
      "training_step_time": 8.673880338668823
    },
    {
      "epoch": 3.563594818115234e-05,
      "model_forward_time": 1.7844398021697998,
      "step": 37367
    },
    {
      "epoch": 3.563594818115234e-05,
      "step": 37367,
      "training_step_time": 9.274112224578857
    },
    {
      "epoch": 3.563690185546875e-05,
      "model_forward_time": 1.7396655082702637,
      "step": 37368
    },
    {
      "epoch": 3.563690185546875e-05,
      "step": 37368,
      "training_step_time": 8.345303058624268
    },
    {
      "epoch": 3.5637855529785156e-05,
      "model_forward_time": 1.7258753776550293,
      "step": 37369
    },
    {
      "epoch": 3.5637855529785156e-05,
      "step": 37369,
      "training_step_time": 7.686675071716309
    },
    {
      "epoch": 3.563880920410156e-05,
      "grad_norm": 0.14450222253799438,
      "learning_rate": 7.398764376629997e-06,
      "loss": 0.0595,
      "step": 37370
    },
    {
      "epoch": 3.563880920410156e-05,
      "model_forward_time": 2.851576805114746,
      "step": 37370
    },
    {
      "dynamics_loss_avg": 0.05587174072861671,
      "epoch": 3.563880920410156e-05,
      "step": 37370
    },
    {
      "action_loss_avg": 0.0017907056724652648,
      "epoch": 3.563880920410156e-05,
      "step": 37370
    },
    {
      "epoch": 3.563880920410156e-05,
      "step": 37370,
      "training_step_time": 7.6368091106414795
    },
    {
      "epoch": 3.563976287841797e-05,
      "model_forward_time": 1.72880220413208,
      "step": 37371
    },
    {
      "epoch": 3.563976287841797e-05,
      "step": 37371,
      "training_step_time": 7.583943128585815
    },
    {
      "epoch": 3.5640716552734375e-05,
      "model_forward_time": 1.742429494857788,
      "step": 37372
    },
    {
      "epoch": 3.5640716552734375e-05,
      "step": 37372,
      "training_step_time": 6.850389003753662
    },
    {
      "epoch": 3.564167022705078e-05,
      "model_forward_time": 1.7379753589630127,
      "step": 37373
    },
    {
      "epoch": 3.564167022705078e-05,
      "step": 37373,
      "training_step_time": 7.34667181968689
    },
    {
      "epoch": 3.564262390136719e-05,
      "model_forward_time": 1.7323577404022217,
      "step": 37374
    },
    {
      "epoch": 3.564262390136719e-05,
      "step": 37374,
      "training_step_time": 7.749886751174927
    },
    {
      "epoch": 3.5643577575683594e-05,
      "model_forward_time": 1.7299277782440186,
      "step": 37375
    },
    {
      "epoch": 3.5643577575683594e-05,
      "step": 37375,
      "training_step_time": 8.159296035766602
    },
    {
      "epoch": 3.564453125e-05,
      "model_forward_time": 1.7486240863800049,
      "step": 37376
    },
    {
      "epoch": 3.564453125e-05,
      "step": 37376,
      "training_step_time": 8.193857192993164
    },
    {
      "epoch": 3.564548492431641e-05,
      "model_forward_time": 1.0315093994140625,
      "step": 37377
    },
    {
      "epoch": 3.564548492431641e-05,
      "step": 37377,
      "training_step_time": 8.307620286941528
    },
    {
      "epoch": 3.564643859863281e-05,
      "model_forward_time": 1.7535572052001953,
      "step": 37378
    },
    {
      "epoch": 3.564643859863281e-05,
      "step": 37378,
      "training_step_time": 7.5507378578186035
    },
    {
      "epoch": 3.564739227294922e-05,
      "model_forward_time": 1.7226109504699707,
      "step": 37379
    },
    {
      "epoch": 3.564739227294922e-05,
      "step": 37379,
      "training_step_time": 7.116263389587402
    },
    {
      "epoch": 3.564834594726563e-05,
      "grad_norm": 0.12315261363983154,
      "learning_rate": 7.397313485027811e-06,
      "loss": 0.0615,
      "step": 37380
    },
    {
      "epoch": 3.564834594726563e-05,
      "model_forward_time": 2.1463541984558105,
      "step": 37380
    },
    {
      "dynamics_loss_avg": 0.057897796388715506,
      "epoch": 3.564834594726563e-05,
      "step": 37380
    },
    {
      "action_loss_avg": 0.0025077920232433825,
      "epoch": 3.564834594726563e-05,
      "step": 37380
    },
    {
      "epoch": 3.564834594726563e-05,
      "step": 37380,
      "training_step_time": 7.6034910678863525
    },
    {
      "epoch": 3.564929962158203e-05,
      "model_forward_time": 1.7547485828399658,
      "step": 37381
    },
    {
      "epoch": 3.564929962158203e-05,
      "step": 37381,
      "training_step_time": 6.831324815750122
    },
    {
      "epoch": 3.5650253295898435e-05,
      "model_forward_time": 1.7381582260131836,
      "step": 37382
    },
    {
      "epoch": 3.5650253295898435e-05,
      "step": 37382,
      "training_step_time": 7.336605072021484
    },
    {
      "epoch": 3.5651206970214846e-05,
      "model_forward_time": 1.7277987003326416,
      "step": 37383
    },
    {
      "epoch": 3.5651206970214846e-05,
      "step": 37383,
      "training_step_time": 7.6736204624176025
    },
    {
      "epoch": 3.565216064453125e-05,
      "model_forward_time": 1.3664045333862305,
      "step": 37384
    },
    {
      "epoch": 3.565216064453125e-05,
      "step": 37384,
      "training_step_time": 7.688065052032471
    },
    {
      "epoch": 3.5653114318847654e-05,
      "model_forward_time": 1.746633529663086,
      "step": 37385
    },
    {
      "epoch": 3.5653114318847654e-05,
      "step": 37385,
      "training_step_time": 9.975346565246582
    },
    {
      "epoch": 3.5654067993164065e-05,
      "model_forward_time": 1.8026249408721924,
      "step": 37386
    },
    {
      "epoch": 3.5654067993164065e-05,
      "step": 37386,
      "training_step_time": 9.579766273498535
    },
    {
      "epoch": 3.565502166748047e-05,
      "model_forward_time": 1.7332520484924316,
      "step": 37387
    },
    {
      "epoch": 3.565502166748047e-05,
      "step": 37387,
      "training_step_time": 7.625510215759277
    },
    {
      "epoch": 3.565597534179687e-05,
      "model_forward_time": 1.73185133934021,
      "step": 37388
    },
    {
      "epoch": 3.565597534179687e-05,
      "step": 37388,
      "training_step_time": 7.491188049316406
    },
    {
      "epoch": 3.5656929016113284e-05,
      "model_forward_time": 1.7456400394439697,
      "step": 37389
    },
    {
      "epoch": 3.5656929016113284e-05,
      "step": 37389,
      "training_step_time": 7.567712068557739
    },
    {
      "epoch": 3.565788269042969e-05,
      "grad_norm": 0.14731299877166748,
      "learning_rate": 7.395862331259027e-06,
      "loss": 0.0619,
      "step": 37390
    },
    {
      "epoch": 3.565788269042969e-05,
      "model_forward_time": 1.7209928035736084,
      "step": 37390
    },
    {
      "dynamics_loss_avg": 0.07132004145532847,
      "epoch": 3.565788269042969e-05,
      "step": 37390
    },
    {
      "action_loss_avg": 0.0032599308295175435,
      "epoch": 3.565788269042969e-05,
      "step": 37390
    },
    {
      "epoch": 3.565788269042969e-05,
      "step": 37390,
      "training_step_time": 7.6077516078948975
    },
    {
      "epoch": 3.565883636474609e-05,
      "model_forward_time": 1.7246887683868408,
      "step": 37391
    },
    {
      "epoch": 3.565883636474609e-05,
      "step": 37391,
      "training_step_time": 7.1925976276397705
    },
    {
      "epoch": 3.56597900390625e-05,
      "model_forward_time": 1.7407422065734863,
      "step": 37392
    },
    {
      "epoch": 3.56597900390625e-05,
      "step": 37392,
      "training_step_time": 7.726302862167358
    },
    {
      "epoch": 3.566074371337891e-05,
      "model_forward_time": 1.7394232749938965,
      "step": 37393
    },
    {
      "epoch": 3.566074371337891e-05,
      "step": 37393,
      "training_step_time": 8.70728063583374
    },
    {
      "epoch": 3.566169738769531e-05,
      "model_forward_time": 1.7420802116394043,
      "step": 37394
    },
    {
      "epoch": 3.566169738769531e-05,
      "step": 37394,
      "training_step_time": 9.428467750549316
    },
    {
      "epoch": 3.566265106201172e-05,
      "model_forward_time": 1.838355302810669,
      "step": 37395
    },
    {
      "epoch": 3.566265106201172e-05,
      "step": 37395,
      "training_step_time": 8.78278636932373
    },
    {
      "epoch": 3.5663604736328126e-05,
      "model_forward_time": 1.7443127632141113,
      "step": 37396
    },
    {
      "epoch": 3.5663604736328126e-05,
      "step": 37396,
      "training_step_time": 7.686187267303467
    },
    {
      "epoch": 3.566455841064453e-05,
      "model_forward_time": 1.7542521953582764,
      "step": 37397
    },
    {
      "epoch": 3.566455841064453e-05,
      "step": 37397,
      "training_step_time": 7.732733249664307
    },
    {
      "epoch": 3.566551208496094e-05,
      "model_forward_time": 1.7391526699066162,
      "step": 37398
    },
    {
      "epoch": 3.566551208496094e-05,
      "step": 37398,
      "training_step_time": 7.080267667770386
    },
    {
      "epoch": 3.5666465759277345e-05,
      "model_forward_time": 2.4990346431732178,
      "step": 37399
    },
    {
      "epoch": 3.5666465759277345e-05,
      "step": 37399,
      "training_step_time": 7.745051860809326
    },
    {
      "epoch": 3.566741943359375e-05,
      "grad_norm": 0.14854960143566132,
      "learning_rate": 7.394410915482342e-06,
      "loss": 0.0626,
      "step": 37400
    },
    {
      "epoch": 3.566741943359375e-05,
      "model_forward_time": 1.728818655014038,
      "step": 37400
    },
    {
      "dynamics_loss_avg": 0.06699328031390905,
      "epoch": 3.566741943359375e-05,
      "step": 37400
    },
    {
      "action_loss_avg": 0.0027134666132042186,
      "epoch": 3.566741943359375e-05,
      "step": 37400
    },
    {
      "epoch": 3.566741943359375e-05,
      "step": 37400,
      "training_step_time": 7.13457989692688
    },
    {
      "epoch": 3.566837310791016e-05,
      "model_forward_time": 1.7500038146972656,
      "step": 37401
    },
    {
      "epoch": 3.566837310791016e-05,
      "step": 37401,
      "training_step_time": 7.753608703613281
    },
    {
      "epoch": 3.5669326782226563e-05,
      "model_forward_time": 1.7162408828735352,
      "step": 37402
    },
    {
      "epoch": 3.5669326782226563e-05,
      "step": 37402,
      "training_step_time": 7.714601993560791
    },
    {
      "epoch": 3.567028045654297e-05,
      "model_forward_time": 1.7384984493255615,
      "step": 37403
    },
    {
      "epoch": 3.567028045654297e-05,
      "step": 37403,
      "training_step_time": 7.71572732925415
    },
    {
      "epoch": 3.567123413085938e-05,
      "model_forward_time": 1.799165964126587,
      "step": 37404
    },
    {
      "epoch": 3.567123413085938e-05,
      "step": 37404,
      "training_step_time": 7.767106056213379
    },
    {
      "epoch": 3.567218780517578e-05,
      "model_forward_time": 1.7391481399536133,
      "step": 37405
    },
    {
      "epoch": 3.567218780517578e-05,
      "step": 37405,
      "training_step_time": 7.773881912231445
    },
    {
      "epoch": 3.5673141479492186e-05,
      "model_forward_time": 1.7460176944732666,
      "step": 37406
    },
    {
      "epoch": 3.5673141479492186e-05,
      "step": 37406,
      "training_step_time": 7.772796392440796
    },
    {
      "epoch": 3.567409515380859e-05,
      "model_forward_time": 1.7505319118499756,
      "step": 37407
    },
    {
      "epoch": 3.567409515380859e-05,
      "step": 37407,
      "training_step_time": 7.352925062179565
    },
    {
      "epoch": 3.5675048828125e-05,
      "model_forward_time": 2.9095370769500732,
      "step": 37408
    },
    {
      "epoch": 3.5675048828125e-05,
      "step": 37408,
      "training_step_time": 7.862621307373047
    },
    {
      "epoch": 3.5676002502441405e-05,
      "model_forward_time": 1.7488734722137451,
      "step": 37409
    },
    {
      "epoch": 3.5676002502441405e-05,
      "step": 37409,
      "training_step_time": 7.644611597061157
    },
    {
      "epoch": 3.567695617675781e-05,
      "grad_norm": 0.1317359358072281,
      "learning_rate": 7.392959237856476e-06,
      "loss": 0.061,
      "step": 37410
    },
    {
      "epoch": 3.567695617675781e-05,
      "model_forward_time": 1.0489935874938965,
      "step": 37410
    },
    {
      "dynamics_loss_avg": 0.058824130333960056,
      "epoch": 3.567695617675781e-05,
      "step": 37410
    },
    {
      "action_loss_avg": 0.0019287691451609136,
      "epoch": 3.567695617675781e-05,
      "step": 37410
    },
    {
      "epoch": 3.567695617675781e-05,
      "step": 37410,
      "training_step_time": 7.654809236526489
    },
    {
      "epoch": 3.567790985107422e-05,
      "model_forward_time": 1.743553876876831,
      "step": 37411
    },
    {
      "epoch": 3.567790985107422e-05,
      "step": 37411,
      "training_step_time": 8.791526556015015
    },
    {
      "epoch": 3.5678863525390624e-05,
      "model_forward_time": 1.7495994567871094,
      "step": 37412
    },
    {
      "epoch": 3.5678863525390624e-05,
      "step": 37412,
      "training_step_time": 8.9649178981781
    },
    {
      "epoch": 3.567981719970703e-05,
      "model_forward_time": 1.7708990573883057,
      "step": 37413
    },
    {
      "epoch": 3.567981719970703e-05,
      "step": 37413,
      "training_step_time": 8.521304845809937
    },
    {
      "epoch": 3.568077087402344e-05,
      "model_forward_time": 1.745387315750122,
      "step": 37414
    },
    {
      "epoch": 3.568077087402344e-05,
      "step": 37414,
      "training_step_time": 9.275232315063477
    },
    {
      "epoch": 3.568172454833984e-05,
      "model_forward_time": 1.7464559078216553,
      "step": 37415
    },
    {
      "epoch": 3.568172454833984e-05,
      "step": 37415,
      "training_step_time": 7.657564163208008
    },
    {
      "epoch": 3.568267822265625e-05,
      "model_forward_time": 1.7154710292816162,
      "step": 37416
    },
    {
      "epoch": 3.568267822265625e-05,
      "step": 37416,
      "training_step_time": 7.722193479537964
    },
    {
      "epoch": 3.568363189697266e-05,
      "model_forward_time": 1.772268533706665,
      "step": 37417
    },
    {
      "epoch": 3.568363189697266e-05,
      "step": 37417,
      "training_step_time": 8.256054162979126
    },
    {
      "epoch": 3.568458557128906e-05,
      "model_forward_time": 1.7422728538513184,
      "step": 37418
    },
    {
      "epoch": 3.568458557128906e-05,
      "step": 37418,
      "training_step_time": 7.614002704620361
    },
    {
      "epoch": 3.5685539245605466e-05,
      "model_forward_time": 1.7407915592193604,
      "step": 37419
    },
    {
      "epoch": 3.5685539245605466e-05,
      "step": 37419,
      "training_step_time": 7.494100093841553
    },
    {
      "epoch": 3.568649291992188e-05,
      "grad_norm": 0.12326427549123764,
      "learning_rate": 7.391507298540188e-06,
      "loss": 0.0605,
      "step": 37420
    },
    {
      "epoch": 3.568649291992188e-05,
      "model_forward_time": 1.7281434535980225,
      "step": 37420
    },
    {
      "dynamics_loss_avg": 0.046415443904697894,
      "epoch": 3.568649291992188e-05,
      "step": 37420
    },
    {
      "action_loss_avg": 0.00176452724263072,
      "epoch": 3.568649291992188e-05,
      "step": 37420
    },
    {
      "epoch": 3.568649291992188e-05,
      "step": 37420,
      "training_step_time": 7.763284683227539
    },
    {
      "epoch": 3.568744659423828e-05,
      "model_forward_time": 1.7234539985656738,
      "step": 37421
    },
    {
      "epoch": 3.568744659423828e-05,
      "step": 37421,
      "training_step_time": 8.437004566192627
    },
    {
      "epoch": 3.5688400268554685e-05,
      "model_forward_time": 1.7340917587280273,
      "step": 37422
    },
    {
      "epoch": 3.5688400268554685e-05,
      "step": 37422,
      "training_step_time": 7.747427225112915
    },
    {
      "epoch": 3.5689353942871095e-05,
      "model_forward_time": 1.8105661869049072,
      "step": 37423
    },
    {
      "epoch": 3.5689353942871095e-05,
      "step": 37423,
      "training_step_time": 8.375437021255493
    },
    {
      "epoch": 3.56903076171875e-05,
      "model_forward_time": 1.725306749343872,
      "step": 37424
    },
    {
      "epoch": 3.56903076171875e-05,
      "step": 37424,
      "training_step_time": 7.696964263916016
    },
    {
      "epoch": 3.5691261291503904e-05,
      "model_forward_time": 1.7626409530639648,
      "step": 37425
    },
    {
      "epoch": 3.5691261291503904e-05,
      "step": 37425,
      "training_step_time": 7.702059507369995
    },
    {
      "epoch": 3.5692214965820314e-05,
      "model_forward_time": 1.7500972747802734,
      "step": 37426
    },
    {
      "epoch": 3.5692214965820314e-05,
      "step": 37426,
      "training_step_time": 7.788806438446045
    },
    {
      "epoch": 3.569316864013672e-05,
      "model_forward_time": 2.8591127395629883,
      "step": 37427
    },
    {
      "epoch": 3.569316864013672e-05,
      "step": 37427,
      "training_step_time": 7.669246673583984
    },
    {
      "epoch": 3.569412231445312e-05,
      "model_forward_time": 1.7536883354187012,
      "step": 37428
    },
    {
      "epoch": 3.569412231445312e-05,
      "step": 37428,
      "training_step_time": 7.4486753940582275
    },
    {
      "epoch": 3.569507598876953e-05,
      "model_forward_time": 1.7317368984222412,
      "step": 37429
    },
    {
      "epoch": 3.569507598876953e-05,
      "step": 37429,
      "training_step_time": 8.280313491821289
    },
    {
      "epoch": 3.569602966308594e-05,
      "grad_norm": 0.11611641198396683,
      "learning_rate": 7.390055097692257e-06,
      "loss": 0.0616,
      "step": 37430
    },
    {
      "epoch": 3.569602966308594e-05,
      "model_forward_time": 1.7227270603179932,
      "step": 37430
    },
    {
      "dynamics_loss_avg": 0.04049429595470429,
      "epoch": 3.569602966308594e-05,
      "step": 37430
    },
    {
      "action_loss_avg": 0.002045260011800565,
      "epoch": 3.569602966308594e-05,
      "step": 37430
    },
    {
      "epoch": 3.569602966308594e-05,
      "step": 37430,
      "training_step_time": 7.742918014526367
    },
    {
      "epoch": 3.569698333740234e-05,
      "model_forward_time": 1.7473669052124023,
      "step": 37431
    },
    {
      "epoch": 3.569698333740234e-05,
      "step": 37431,
      "training_step_time": 9.581146478652954
    },
    {
      "epoch": 3.569793701171875e-05,
      "model_forward_time": 1.7898037433624268,
      "step": 37432
    },
    {
      "epoch": 3.569793701171875e-05,
      "step": 37432,
      "training_step_time": 8.364952325820923
    },
    {
      "epoch": 3.5698890686035156e-05,
      "model_forward_time": 1.7516703605651855,
      "step": 37433
    },
    {
      "epoch": 3.5698890686035156e-05,
      "step": 37433,
      "training_step_time": 8.72361421585083
    },
    {
      "epoch": 3.569984436035156e-05,
      "model_forward_time": 1.7451188564300537,
      "step": 37434
    },
    {
      "epoch": 3.569984436035156e-05,
      "step": 37434,
      "training_step_time": 7.965662479400635
    },
    {
      "epoch": 3.570079803466797e-05,
      "model_forward_time": 1.7427427768707275,
      "step": 37435
    },
    {
      "epoch": 3.570079803466797e-05,
      "step": 37435,
      "training_step_time": 7.308082580566406
    },
    {
      "epoch": 3.5701751708984375e-05,
      "model_forward_time": 2.923597574234009,
      "step": 37436
    },
    {
      "epoch": 3.5701751708984375e-05,
      "step": 37436,
      "training_step_time": 7.765034198760986
    },
    {
      "epoch": 3.570270538330078e-05,
      "model_forward_time": 1.744887113571167,
      "step": 37437
    },
    {
      "epoch": 3.570270538330078e-05,
      "step": 37437,
      "training_step_time": 7.506643056869507
    },
    {
      "epoch": 3.570365905761719e-05,
      "model_forward_time": 1.7451469898223877,
      "step": 37438
    },
    {
      "epoch": 3.570365905761719e-05,
      "step": 37438,
      "training_step_time": 7.606531858444214
    },
    {
      "epoch": 3.5704612731933594e-05,
      "model_forward_time": 1.7459251880645752,
      "step": 37439
    },
    {
      "epoch": 3.5704612731933594e-05,
      "step": 37439,
      "training_step_time": 8.909599304199219
    },
    {
      "epoch": 3.570556640625e-05,
      "grad_norm": 0.13425759971141815,
      "learning_rate": 7.3886026354714935e-06,
      "loss": 0.0597,
      "step": 37440
    },
    {
      "epoch": 3.570556640625e-05,
      "model_forward_time": 1.729975938796997,
      "step": 37440
    },
    {
      "dynamics_loss_avg": 0.0664316613227129,
      "epoch": 3.570556640625e-05,
      "step": 37440
    },
    {
      "action_loss_avg": 0.0037860810582060365,
      "epoch": 3.570556640625e-05,
      "step": 37440
    },
    {
      "epoch": 3.570556640625e-05,
      "step": 37440,
      "training_step_time": 8.629359245300293
    },
    {
      "epoch": 3.570652008056641e-05,
      "model_forward_time": 1.7568037509918213,
      "step": 37441
    },
    {
      "epoch": 3.570652008056641e-05,
      "step": 37441,
      "training_step_time": 9.677878379821777
    },
    {
      "epoch": 3.570747375488281e-05,
      "model_forward_time": 1.7286067008972168,
      "step": 37442
    },
    {
      "epoch": 3.570747375488281e-05,
      "step": 37442,
      "training_step_time": 7.587207317352295
    },
    {
      "epoch": 3.570842742919922e-05,
      "model_forward_time": 1.736990213394165,
      "step": 37443
    },
    {
      "epoch": 3.570842742919922e-05,
      "step": 37443,
      "training_step_time": 7.32253360748291
    },
    {
      "epoch": 3.570938110351563e-05,
      "model_forward_time": 1.740211009979248,
      "step": 37444
    },
    {
      "epoch": 3.570938110351563e-05,
      "step": 37444,
      "training_step_time": 7.627131462097168
    },
    {
      "epoch": 3.571033477783203e-05,
      "model_forward_time": 2.8962526321411133,
      "step": 37445
    },
    {
      "epoch": 3.571033477783203e-05,
      "step": 37445,
      "training_step_time": 7.631288766860962
    },
    {
      "epoch": 3.5711288452148436e-05,
      "model_forward_time": 1.7461175918579102,
      "step": 37446
    },
    {
      "epoch": 3.5711288452148436e-05,
      "step": 37446,
      "training_step_time": 7.469541549682617
    },
    {
      "epoch": 3.5712242126464846e-05,
      "model_forward_time": 1.7464988231658936,
      "step": 37447
    },
    {
      "epoch": 3.5712242126464846e-05,
      "step": 37447,
      "training_step_time": 7.548214435577393
    },
    {
      "epoch": 3.571319580078125e-05,
      "model_forward_time": 1.738612413406372,
      "step": 37448
    },
    {
      "epoch": 3.571319580078125e-05,
      "step": 37448,
      "training_step_time": 7.597489833831787
    },
    {
      "epoch": 3.5714149475097654e-05,
      "model_forward_time": 1.7397997379302979,
      "step": 37449
    },
    {
      "epoch": 3.5714149475097654e-05,
      "step": 37449,
      "training_step_time": 7.691622257232666
    },
    {
      "epoch": 3.5715103149414065e-05,
      "grad_norm": 0.10800729691982269,
      "learning_rate": 7.387149912036737e-06,
      "loss": 0.0577,
      "step": 37450
    },
    {
      "epoch": 3.5715103149414065e-05,
      "model_forward_time": 1.7238695621490479,
      "step": 37450
    },
    {
      "dynamics_loss_avg": 0.05477457232773304,
      "epoch": 3.5715103149414065e-05,
      "step": 37450
    },
    {
      "action_loss_avg": 0.0020478793216170744,
      "epoch": 3.5715103149414065e-05,
      "step": 37450
    },
    {
      "epoch": 3.5715103149414065e-05,
      "step": 37450,
      "training_step_time": 10.182445764541626
    },
    {
      "epoch": 3.571605682373047e-05,
      "model_forward_time": 1.8025035858154297,
      "step": 37451
    },
    {
      "epoch": 3.571605682373047e-05,
      "step": 37451,
      "training_step_time": 10.161076068878174
    },
    {
      "epoch": 3.571701049804687e-05,
      "model_forward_time": 1.7411842346191406,
      "step": 37452
    },
    {
      "epoch": 3.571701049804687e-05,
      "step": 37452,
      "training_step_time": 7.363564491271973
    },
    {
      "epoch": 3.5717964172363284e-05,
      "model_forward_time": 1.522346019744873,
      "step": 37453
    },
    {
      "epoch": 3.5717964172363284e-05,
      "step": 37453,
      "training_step_time": 7.619351387023926
    },
    {
      "epoch": 3.571891784667969e-05,
      "model_forward_time": 3.0007200241088867,
      "step": 37454
    },
    {
      "epoch": 3.571891784667969e-05,
      "step": 37454,
      "training_step_time": 7.731492757797241
    },
    {
      "epoch": 3.571987152099609e-05,
      "model_forward_time": 1.870378017425537,
      "step": 37455
    },
    {
      "epoch": 3.571987152099609e-05,
      "step": 37455,
      "training_step_time": 8.326860427856445
    },
    {
      "epoch": 3.57208251953125e-05,
      "model_forward_time": 1.8850843906402588,
      "step": 37456
    },
    {
      "epoch": 3.57208251953125e-05,
      "step": 37456,
      "training_step_time": 8.366350889205933
    },
    {
      "epoch": 3.572177886962891e-05,
      "model_forward_time": 1.8933489322662354,
      "step": 37457
    },
    {
      "epoch": 3.572177886962891e-05,
      "step": 37457,
      "training_step_time": 7.675970554351807
    },
    {
      "epoch": 3.572273254394531e-05,
      "model_forward_time": 1.866804838180542,
      "step": 37458
    },
    {
      "epoch": 3.572273254394531e-05,
      "step": 37458,
      "training_step_time": 9.245843648910522
    },
    {
      "epoch": 3.572368621826172e-05,
      "model_forward_time": 1.9040229320526123,
      "step": 37459
    },
    {
      "epoch": 3.572368621826172e-05,
      "step": 37459,
      "training_step_time": 7.6981377601623535
    },
    {
      "epoch": 3.5724639892578126e-05,
      "grad_norm": 0.1891399770975113,
      "learning_rate": 7.3856969275468545e-06,
      "loss": 0.0627,
      "step": 37460
    },
    {
      "epoch": 3.5724639892578126e-05,
      "model_forward_time": 1.8965349197387695,
      "step": 37460
    },
    {
      "dynamics_loss_avg": 0.06485547646880149,
      "epoch": 3.5724639892578126e-05,
      "step": 37460
    },
    {
      "action_loss_avg": 0.0025684914784505964,
      "epoch": 3.5724639892578126e-05,
      "step": 37460
    },
    {
      "epoch": 3.5724639892578126e-05,
      "step": 37460,
      "training_step_time": 9.61806607246399
    },
    {
      "epoch": 3.572559356689453e-05,
      "model_forward_time": 1.8909993171691895,
      "step": 37461
    },
    {
      "epoch": 3.572559356689453e-05,
      "step": 37461,
      "training_step_time": 7.382177114486694
    },
    {
      "epoch": 3.572654724121094e-05,
      "model_forward_time": 1.8939743041992188,
      "step": 37462
    },
    {
      "epoch": 3.572654724121094e-05,
      "step": 37462,
      "training_step_time": 7.599510192871094
    },
    {
      "epoch": 3.5727500915527345e-05,
      "model_forward_time": 1.1619551181793213,
      "step": 37463
    },
    {
      "epoch": 3.5727500915527345e-05,
      "step": 37463,
      "training_step_time": 7.652300596237183
    },
    {
      "epoch": 3.572845458984375e-05,
      "model_forward_time": 2.9537031650543213,
      "step": 37464
    },
    {
      "epoch": 3.572845458984375e-05,
      "step": 37464,
      "training_step_time": 7.66990852355957
    },
    {
      "epoch": 3.572940826416016e-05,
      "model_forward_time": 1.8881430625915527,
      "step": 37465
    },
    {
      "epoch": 3.572940826416016e-05,
      "step": 37465,
      "training_step_time": 7.325727939605713
    },
    {
      "epoch": 3.5730361938476564e-05,
      "model_forward_time": 1.8587679862976074,
      "step": 37466
    },
    {
      "epoch": 3.5730361938476564e-05,
      "step": 37466,
      "training_step_time": 7.664611577987671
    },
    {
      "epoch": 3.573131561279297e-05,
      "model_forward_time": 1.8729588985443115,
      "step": 37467
    },
    {
      "epoch": 3.573131561279297e-05,
      "step": 37467,
      "training_step_time": 7.627250671386719
    },
    {
      "epoch": 3.573226928710938e-05,
      "model_forward_time": 1.8742947578430176,
      "step": 37468
    },
    {
      "epoch": 3.573226928710938e-05,
      "step": 37468,
      "training_step_time": 10.255452632904053
    },
    {
      "epoch": 3.573322296142578e-05,
      "model_forward_time": 1.5565159320831299,
      "step": 37469
    },
    {
      "epoch": 3.573322296142578e-05,
      "step": 37469,
      "training_step_time": 8.18790864944458
    },
    {
      "epoch": 3.5734176635742186e-05,
      "grad_norm": 0.12944188714027405,
      "learning_rate": 7.384243682160743e-06,
      "loss": 0.0596,
      "step": 37470
    },
    {
      "epoch": 3.5734176635742186e-05,
      "model_forward_time": 1.8792588710784912,
      "step": 37470
    },
    {
      "dynamics_loss_avg": 0.051225198246538636,
      "epoch": 3.5734176635742186e-05,
      "step": 37470
    },
    {
      "action_loss_avg": 0.0021458206698298453,
      "epoch": 3.5734176635742186e-05,
      "step": 37470
    },
    {
      "epoch": 3.5734176635742186e-05,
      "step": 37470,
      "training_step_time": 8.343296527862549
    },
    {
      "epoch": 3.573513031005859e-05,
      "model_forward_time": 1.9015271663665771,
      "step": 37471
    },
    {
      "epoch": 3.573513031005859e-05,
      "step": 37471,
      "training_step_time": 7.593871831893921
    },
    {
      "epoch": 3.5736083984375e-05,
      "model_forward_time": 1.8801462650299072,
      "step": 37472
    },
    {
      "epoch": 3.5736083984375e-05,
      "step": 37472,
      "training_step_time": 7.622828245162964
    },
    {
      "epoch": 3.5737037658691405e-05,
      "model_forward_time": 2.9590117931365967,
      "step": 37473
    },
    {
      "epoch": 3.5737037658691405e-05,
      "step": 37473,
      "training_step_time": 7.689068794250488
    },
    {
      "epoch": 3.573799133300781e-05,
      "model_forward_time": 1.9046084880828857,
      "step": 37474
    },
    {
      "epoch": 3.573799133300781e-05,
      "step": 37474,
      "training_step_time": 7.314526081085205
    },
    {
      "epoch": 3.573894500732422e-05,
      "model_forward_time": 1.8989546298980713,
      "step": 37475
    },
    {
      "epoch": 3.573894500732422e-05,
      "step": 37475,
      "training_step_time": 8.159029960632324
    },
    {
      "epoch": 3.5739898681640624e-05,
      "model_forward_time": 1.8595938682556152,
      "step": 37476
    },
    {
      "epoch": 3.5739898681640624e-05,
      "step": 37476,
      "training_step_time": 8.220836162567139
    },
    {
      "epoch": 3.574085235595703e-05,
      "model_forward_time": 1.8744559288024902,
      "step": 37477
    },
    {
      "epoch": 3.574085235595703e-05,
      "step": 37477,
      "training_step_time": 8.627318143844604
    },
    {
      "epoch": 3.574180603027344e-05,
      "model_forward_time": 1.845576524734497,
      "step": 37478
    },
    {
      "epoch": 3.574180603027344e-05,
      "step": 37478,
      "training_step_time": 9.17262578010559
    },
    {
      "epoch": 3.574275970458984e-05,
      "model_forward_time": 1.8787977695465088,
      "step": 37479
    },
    {
      "epoch": 3.574275970458984e-05,
      "step": 37479,
      "training_step_time": 7.6406848430633545
    },
    {
      "epoch": 3.574371337890625e-05,
      "grad_norm": 0.15182854235172272,
      "learning_rate": 7.3827901760373285e-06,
      "loss": 0.058,
      "step": 37480
    },
    {
      "epoch": 3.574371337890625e-05,
      "model_forward_time": 1.9062104225158691,
      "step": 37480
    },
    {
      "dynamics_loss_avg": 0.05120178498327732,
      "epoch": 3.574371337890625e-05,
      "step": 37480
    },
    {
      "action_loss_avg": 0.002126092615071684,
      "epoch": 3.574371337890625e-05,
      "step": 37480
    },
    {
      "epoch": 3.574371337890625e-05,
      "step": 37480,
      "training_step_time": 7.396477460861206
    },
    {
      "epoch": 3.574466705322266e-05,
      "model_forward_time": 1.78751802444458,
      "step": 37481
    },
    {
      "epoch": 3.574466705322266e-05,
      "step": 37481,
      "training_step_time": 8.64164924621582
    },
    {
      "epoch": 3.574562072753906e-05,
      "model_forward_time": 2.9293746948242188,
      "step": 37482
    },
    {
      "epoch": 3.574562072753906e-05,
      "step": 37482,
      "training_step_time": 7.661534309387207
    },
    {
      "epoch": 3.5746574401855466e-05,
      "model_forward_time": 1.7238855361938477,
      "step": 37483
    },
    {
      "epoch": 3.5746574401855466e-05,
      "step": 37483,
      "training_step_time": 7.466371059417725
    },
    {
      "epoch": 3.574752807617188e-05,
      "model_forward_time": 1.724214792251587,
      "step": 37484
    },
    {
      "epoch": 3.574752807617188e-05,
      "step": 37484,
      "training_step_time": 8.569880962371826
    },
    {
      "epoch": 3.574848175048828e-05,
      "model_forward_time": 1.7312490940093994,
      "step": 37485
    },
    {
      "epoch": 3.574848175048828e-05,
      "step": 37485,
      "training_step_time": 7.678507566452026
    },
    {
      "epoch": 3.5749435424804685e-05,
      "model_forward_time": 1.7716851234436035,
      "step": 37486
    },
    {
      "epoch": 3.5749435424804685e-05,
      "step": 37486,
      "training_step_time": 7.572997093200684
    },
    {
      "epoch": 3.5750389099121096e-05,
      "model_forward_time": 1.7247190475463867,
      "step": 37487
    },
    {
      "epoch": 3.5750389099121096e-05,
      "step": 37487,
      "training_step_time": 7.6259331703186035
    },
    {
      "epoch": 3.57513427734375e-05,
      "model_forward_time": 1.740039348602295,
      "step": 37488
    },
    {
      "epoch": 3.57513427734375e-05,
      "step": 37488,
      "training_step_time": 8.584716081619263
    },
    {
      "epoch": 3.5752296447753904e-05,
      "model_forward_time": 1.7266521453857422,
      "step": 37489
    },
    {
      "epoch": 3.5752296447753904e-05,
      "step": 37489,
      "training_step_time": 7.318433523178101
    },
    {
      "epoch": 3.5753250122070314e-05,
      "grad_norm": 0.13802947103977203,
      "learning_rate": 7.3813364093355625e-06,
      "loss": 0.0554,
      "step": 37490
    },
    {
      "epoch": 3.5753250122070314e-05,
      "model_forward_time": 1.7225027084350586,
      "step": 37490
    },
    {
      "dynamics_loss_avg": 0.057043037936091424,
      "epoch": 3.5753250122070314e-05,
      "step": 37490
    },
    {
      "action_loss_avg": 0.002720069827046245,
      "epoch": 3.5753250122070314e-05,
      "step": 37490
    },
    {
      "epoch": 3.5753250122070314e-05,
      "step": 37490,
      "training_step_time": 7.706737518310547
    },
    {
      "epoch": 3.575420379638672e-05,
      "model_forward_time": 2.850955009460449,
      "step": 37491
    },
    {
      "epoch": 3.575420379638672e-05,
      "step": 37491,
      "training_step_time": 7.61493992805481
    },
    {
      "epoch": 3.575515747070312e-05,
      "model_forward_time": 1.7342875003814697,
      "step": 37492
    },
    {
      "epoch": 3.575515747070312e-05,
      "step": 37492,
      "training_step_time": 7.541743516921997
    },
    {
      "epoch": 3.575611114501953e-05,
      "model_forward_time": 1.7232322692871094,
      "step": 37493
    },
    {
      "epoch": 3.575611114501953e-05,
      "step": 37493,
      "training_step_time": 7.625633716583252
    },
    {
      "epoch": 3.575706481933594e-05,
      "model_forward_time": 1.7317657470703125,
      "step": 37494
    },
    {
      "epoch": 3.575706481933594e-05,
      "step": 37494,
      "training_step_time": 8.859806776046753
    },
    {
      "epoch": 3.575801849365234e-05,
      "model_forward_time": 1.717102289199829,
      "step": 37495
    },
    {
      "epoch": 3.575801849365234e-05,
      "step": 37495,
      "training_step_time": 8.442627668380737
    },
    {
      "epoch": 3.575897216796875e-05,
      "model_forward_time": 1.7109754085540771,
      "step": 37496
    },
    {
      "epoch": 3.575897216796875e-05,
      "step": 37496,
      "training_step_time": 7.697760581970215
    },
    {
      "epoch": 3.5759925842285156e-05,
      "model_forward_time": 1.7271182537078857,
      "step": 37497
    },
    {
      "epoch": 3.5759925842285156e-05,
      "step": 37497,
      "training_step_time": 9.494133472442627
    },
    {
      "epoch": 3.576087951660156e-05,
      "model_forward_time": 1.7170162200927734,
      "step": 37498
    },
    {
      "epoch": 3.576087951660156e-05,
      "step": 37498,
      "training_step_time": 7.587761640548706
    },
    {
      "epoch": 3.576183319091797e-05,
      "model_forward_time": 1.7346928119659424,
      "step": 37499
    },
    {
      "epoch": 3.576183319091797e-05,
      "step": 37499,
      "training_step_time": 7.695052623748779
    },
    {
      "epoch": 3.5762786865234375e-05,
      "grad_norm": 0.12176480144262314,
      "learning_rate": 7.3798823822144265e-06,
      "loss": 0.0568,
      "step": 37500
    },
    {
      "epoch": 3.5762786865234375e-05,
      "model_forward_time": 2.8433704376220703,
      "step": 37500
    },
    {
      "dynamics_loss_avg": 0.05778504312038422,
      "epoch": 3.5762786865234375e-05,
      "step": 37500
    },
    {
      "action_loss_avg": 0.00250126343453303,
      "epoch": 3.5762786865234375e-05,
      "step": 37500
    },
    {
      "epoch": 3.5762786865234375e-05,
      "step": 37500,
      "training_step_time": 7.6863274574279785
    },
    {
      "epoch": 3.576374053955078e-05,
      "model_forward_time": 1.7239477634429932,
      "step": 37501
    },
    {
      "epoch": 3.576374053955078e-05,
      "step": 37501,
      "training_step_time": 7.593790292739868
    },
    {
      "epoch": 3.576469421386719e-05,
      "model_forward_time": 1.7355854511260986,
      "step": 37502
    },
    {
      "epoch": 3.576469421386719e-05,
      "step": 37502,
      "training_step_time": 7.611122369766235
    },
    {
      "epoch": 3.5765647888183594e-05,
      "model_forward_time": 1.71816086769104,
      "step": 37503
    },
    {
      "epoch": 3.5765647888183594e-05,
      "step": 37503,
      "training_step_time": 7.668302297592163
    },
    {
      "epoch": 3.57666015625e-05,
      "model_forward_time": 1.7257068157196045,
      "step": 37504
    },
    {
      "epoch": 3.57666015625e-05,
      "step": 37504,
      "training_step_time": 9.17458462715149
    },
    {
      "epoch": 3.576755523681641e-05,
      "model_forward_time": 1.725620985031128,
      "step": 37505
    },
    {
      "epoch": 3.576755523681641e-05,
      "step": 37505,
      "training_step_time": 9.628580331802368
    },
    {
      "epoch": 3.576850891113281e-05,
      "model_forward_time": 1.7111732959747314,
      "step": 37506
    },
    {
      "epoch": 3.576850891113281e-05,
      "step": 37506,
      "training_step_time": 9.180530309677124
    },
    {
      "epoch": 3.576946258544922e-05,
      "model_forward_time": 1.7078461647033691,
      "step": 37507
    },
    {
      "epoch": 3.576946258544922e-05,
      "step": 37507,
      "training_step_time": 8.243528366088867
    },
    {
      "epoch": 3.577041625976563e-05,
      "model_forward_time": 1.7069039344787598,
      "step": 37508
    },
    {
      "epoch": 3.577041625976563e-05,
      "step": 37508,
      "training_step_time": 7.6131556034088135
    },
    {
      "epoch": 3.577136993408203e-05,
      "model_forward_time": 1.7153058052062988,
      "step": 37509
    },
    {
      "epoch": 3.577136993408203e-05,
      "step": 37509,
      "training_step_time": 8.966288805007935
    },
    {
      "epoch": 3.5772323608398436e-05,
      "grad_norm": 0.13058942556381226,
      "learning_rate": 7.3784280948329315e-06,
      "loss": 0.0558,
      "step": 37510
    },
    {
      "epoch": 3.5772323608398436e-05,
      "model_forward_time": 1.7129285335540771,
      "step": 37510
    },
    {
      "dynamics_loss_avg": 0.057171502336859706,
      "epoch": 3.5772323608398436e-05,
      "step": 37510
    },
    {
      "action_loss_avg": 0.0019059692363953217,
      "epoch": 3.5772323608398436e-05,
      "step": 37510
    },
    {
      "epoch": 3.5772323608398436e-05,
      "step": 37510,
      "training_step_time": 7.590677261352539
    },
    {
      "epoch": 3.5773277282714846e-05,
      "model_forward_time": 1.7192449569702148,
      "step": 37511
    },
    {
      "epoch": 3.5773277282714846e-05,
      "step": 37511,
      "training_step_time": 7.7190611362457275
    },
    {
      "epoch": 3.577423095703125e-05,
      "model_forward_time": 1.7019898891448975,
      "step": 37512
    },
    {
      "epoch": 3.577423095703125e-05,
      "step": 37512,
      "training_step_time": 7.6916375160217285
    },
    {
      "epoch": 3.5775184631347654e-05,
      "model_forward_time": 1.7058658599853516,
      "step": 37513
    },
    {
      "epoch": 3.5775184631347654e-05,
      "step": 37513,
      "training_step_time": 8.450104475021362
    },
    {
      "epoch": 3.5776138305664065e-05,
      "model_forward_time": 1.3578503131866455,
      "step": 37514
    },
    {
      "epoch": 3.5776138305664065e-05,
      "step": 37514,
      "training_step_time": 9.355252504348755
    },
    {
      "epoch": 3.577709197998047e-05,
      "model_forward_time": 1.7239718437194824,
      "step": 37515
    },
    {
      "epoch": 3.577709197998047e-05,
      "step": 37515,
      "training_step_time": 9.888212442398071
    },
    {
      "epoch": 3.577804565429687e-05,
      "model_forward_time": 1.7079427242279053,
      "step": 37516
    },
    {
      "epoch": 3.577804565429687e-05,
      "step": 37516,
      "training_step_time": 7.6799938678741455
    },
    {
      "epoch": 3.5778999328613284e-05,
      "model_forward_time": 1.7109160423278809,
      "step": 37517
    },
    {
      "epoch": 3.5778999328613284e-05,
      "step": 37517,
      "training_step_time": 7.672801733016968
    },
    {
      "epoch": 3.577995300292969e-05,
      "model_forward_time": 1.7153081893920898,
      "step": 37518
    },
    {
      "epoch": 3.577995300292969e-05,
      "step": 37518,
      "training_step_time": 7.635464191436768
    },
    {
      "epoch": 3.578090667724609e-05,
      "model_forward_time": 2.909322500228882,
      "step": 37519
    },
    {
      "epoch": 3.578090667724609e-05,
      "step": 37519,
      "training_step_time": 7.695032358169556
    },
    {
      "epoch": 3.57818603515625e-05,
      "grad_norm": 0.10534868389368057,
      "learning_rate": 7.376973547350115e-06,
      "loss": 0.0563,
      "step": 37520
    },
    {
      "epoch": 3.57818603515625e-05,
      "model_forward_time": 1.70003080368042,
      "step": 37520
    },
    {
      "dynamics_loss_avg": 0.05569730959832668,
      "epoch": 3.57818603515625e-05,
      "step": 37520
    },
    {
      "action_loss_avg": 0.0017954580776859074,
      "epoch": 3.57818603515625e-05,
      "step": 37520
    },
    {
      "epoch": 3.57818603515625e-05,
      "step": 37520,
      "training_step_time": 7.625412225723267
    },
    {
      "epoch": 3.578281402587891e-05,
      "model_forward_time": 1.7124662399291992,
      "step": 37521
    },
    {
      "epoch": 3.578281402587891e-05,
      "step": 37521,
      "training_step_time": 8.423375606536865
    },
    {
      "epoch": 3.578376770019531e-05,
      "model_forward_time": 0.9881434440612793,
      "step": 37522
    },
    {
      "epoch": 3.578376770019531e-05,
      "step": 37522,
      "training_step_time": 7.732334136962891
    },
    {
      "epoch": 3.578472137451172e-05,
      "model_forward_time": 1.7209382057189941,
      "step": 37523
    },
    {
      "epoch": 3.578472137451172e-05,
      "step": 37523,
      "training_step_time": 9.815789461135864
    },
    {
      "epoch": 3.5785675048828126e-05,
      "model_forward_time": 1.712860107421875,
      "step": 37524
    },
    {
      "epoch": 3.5785675048828126e-05,
      "step": 37524,
      "training_step_time": 8.370614767074585
    },
    {
      "epoch": 3.578662872314453e-05,
      "model_forward_time": 1.7167859077453613,
      "step": 37525
    },
    {
      "epoch": 3.578662872314453e-05,
      "step": 37525,
      "training_step_time": 7.6796181201934814
    },
    {
      "epoch": 3.578758239746094e-05,
      "model_forward_time": 1.7294201850891113,
      "step": 37526
    },
    {
      "epoch": 3.578758239746094e-05,
      "step": 37526,
      "training_step_time": 7.588650465011597
    },
    {
      "epoch": 3.5788536071777345e-05,
      "model_forward_time": 1.7098240852355957,
      "step": 37527
    },
    {
      "epoch": 3.5788536071777345e-05,
      "step": 37527,
      "training_step_time": 7.697184324264526
    },
    {
      "epoch": 3.578948974609375e-05,
      "model_forward_time": 2.9183719158172607,
      "step": 37528
    },
    {
      "epoch": 3.578948974609375e-05,
      "step": 37528,
      "training_step_time": 7.755684852600098
    },
    {
      "epoch": 3.579044342041016e-05,
      "model_forward_time": 1.7274408340454102,
      "step": 37529
    },
    {
      "epoch": 3.579044342041016e-05,
      "step": 37529,
      "training_step_time": 7.701695442199707
    },
    {
      "epoch": 3.5791397094726564e-05,
      "grad_norm": 0.11310698091983795,
      "learning_rate": 7.375518739925046e-06,
      "loss": 0.0614,
      "step": 37530
    },
    {
      "epoch": 3.5791397094726564e-05,
      "model_forward_time": 1.7178716659545898,
      "step": 37530
    },
    {
      "dynamics_loss_avg": 0.07624592315405607,
      "epoch": 3.5791397094726564e-05,
      "step": 37530
    },
    {
      "action_loss_avg": 0.004904837149661034,
      "epoch": 3.5791397094726564e-05,
      "step": 37530
    },
    {
      "epoch": 3.5791397094726564e-05,
      "step": 37530,
      "training_step_time": 7.698236465454102
    },
    {
      "epoch": 3.579235076904297e-05,
      "model_forward_time": 1.713562250137329,
      "step": 37531
    },
    {
      "epoch": 3.579235076904297e-05,
      "step": 37531,
      "training_step_time": 8.846231460571289
    },
    {
      "epoch": 3.579330444335937e-05,
      "model_forward_time": 1.7158567905426025,
      "step": 37532
    },
    {
      "epoch": 3.579330444335937e-05,
      "step": 37532,
      "training_step_time": 7.6680872440338135
    },
    {
      "epoch": 3.579425811767578e-05,
      "model_forward_time": 1.7155849933624268,
      "step": 37533
    },
    {
      "epoch": 3.579425811767578e-05,
      "step": 37533,
      "training_step_time": 8.793415069580078
    },
    {
      "epoch": 3.5795211791992186e-05,
      "model_forward_time": 1.7394864559173584,
      "step": 37534
    },
    {
      "epoch": 3.5795211791992186e-05,
      "step": 37534,
      "training_step_time": 7.719780921936035
    },
    {
      "epoch": 3.579616546630859e-05,
      "model_forward_time": 1.712977647781372,
      "step": 37535
    },
    {
      "epoch": 3.579616546630859e-05,
      "step": 37535,
      "training_step_time": 7.633486747741699
    },
    {
      "epoch": 3.5797119140625e-05,
      "model_forward_time": 1.7235636711120605,
      "step": 37536
    },
    {
      "epoch": 3.5797119140625e-05,
      "step": 37536,
      "training_step_time": 7.839727878570557
    },
    {
      "epoch": 3.5798072814941405e-05,
      "model_forward_time": 2.8181936740875244,
      "step": 37537
    },
    {
      "epoch": 3.5798072814941405e-05,
      "step": 37537,
      "training_step_time": 8.537113189697266
    },
    {
      "epoch": 3.579902648925781e-05,
      "model_forward_time": 1.7367761135101318,
      "step": 37538
    },
    {
      "epoch": 3.579902648925781e-05,
      "step": 37538,
      "training_step_time": 7.764648199081421
    },
    {
      "epoch": 3.579998016357422e-05,
      "model_forward_time": 1.7369203567504883,
      "step": 37539
    },
    {
      "epoch": 3.579998016357422e-05,
      "step": 37539,
      "training_step_time": 8.333941221237183
    },
    {
      "epoch": 3.5800933837890624e-05,
      "grad_norm": 0.14042752981185913,
      "learning_rate": 7.374063672716818e-06,
      "loss": 0.0609,
      "step": 37540
    },
    {
      "epoch": 3.5800933837890624e-05,
      "model_forward_time": 1.7304537296295166,
      "step": 37540
    },
    {
      "dynamics_loss_avg": 0.05566566903144121,
      "epoch": 3.5800933837890624e-05,
      "step": 37540
    },
    {
      "action_loss_avg": 0.0022626523743383585,
      "epoch": 3.5800933837890624e-05,
      "step": 37540
    },
    {
      "epoch": 3.5800933837890624e-05,
      "step": 37540,
      "training_step_time": 8.733727216720581
    },
    {
      "epoch": 3.580188751220703e-05,
      "model_forward_time": 1.7158429622650146,
      "step": 37541
    },
    {
      "epoch": 3.580188751220703e-05,
      "step": 37541,
      "training_step_time": 8.352229118347168
    },
    {
      "epoch": 3.580284118652344e-05,
      "model_forward_time": 1.7471628189086914,
      "step": 37542
    },
    {
      "epoch": 3.580284118652344e-05,
      "step": 37542,
      "training_step_time": 7.72533917427063
    },
    {
      "epoch": 3.580379486083984e-05,
      "model_forward_time": 1.7396371364593506,
      "step": 37543
    },
    {
      "epoch": 3.580379486083984e-05,
      "step": 37543,
      "training_step_time": 7.652730941772461
    },
    {
      "epoch": 3.580474853515625e-05,
      "model_forward_time": 1.703885793685913,
      "step": 37544
    },
    {
      "epoch": 3.580474853515625e-05,
      "step": 37544,
      "training_step_time": 7.671600103378296
    },
    {
      "epoch": 3.580570220947266e-05,
      "model_forward_time": 1.745239019393921,
      "step": 37545
    },
    {
      "epoch": 3.580570220947266e-05,
      "step": 37545,
      "training_step_time": 7.748208522796631
    },
    {
      "epoch": 3.580665588378906e-05,
      "model_forward_time": 1.0225050449371338,
      "step": 37546
    },
    {
      "epoch": 3.580665588378906e-05,
      "step": 37546,
      "training_step_time": 7.708998203277588
    },
    {
      "epoch": 3.5807609558105466e-05,
      "model_forward_time": 2.8853600025177,
      "step": 37547
    },
    {
      "epoch": 3.5807609558105466e-05,
      "step": 37547,
      "training_step_time": 7.738200664520264
    },
    {
      "epoch": 3.580856323242188e-05,
      "model_forward_time": 1.7359745502471924,
      "step": 37548
    },
    {
      "epoch": 3.580856323242188e-05,
      "step": 37548,
      "training_step_time": 7.771306753158569
    },
    {
      "epoch": 3.580951690673828e-05,
      "model_forward_time": 1.72528076171875,
      "step": 37549
    },
    {
      "epoch": 3.580951690673828e-05,
      "step": 37549,
      "training_step_time": 7.739786386489868
    },
    {
      "epoch": 3.5810470581054685e-05,
      "grad_norm": 0.1436949074268341,
      "learning_rate": 7.372608345884558e-06,
      "loss": 0.0611,
      "step": 37550
    },
    {
      "epoch": 3.5810470581054685e-05,
      "model_forward_time": 1.7288997173309326,
      "step": 37550
    },
    {
      "dynamics_loss_avg": 0.055485906451940535,
      "epoch": 3.5810470581054685e-05,
      "step": 37550
    },
    {
      "action_loss_avg": 0.004411855948274024,
      "epoch": 3.5810470581054685e-05,
      "step": 37550
    },
    {
      "epoch": 3.5810470581054685e-05,
      "step": 37550,
      "training_step_time": 7.721553802490234
    },
    {
      "epoch": 3.5811424255371096e-05,
      "model_forward_time": 1.7554316520690918,
      "step": 37551
    },
    {
      "epoch": 3.5811424255371096e-05,
      "step": 37551,
      "training_step_time": 7.839905738830566
    },
    {
      "epoch": 3.58123779296875e-05,
      "model_forward_time": 1.7129316329956055,
      "step": 37552
    },
    {
      "epoch": 3.58123779296875e-05,
      "step": 37552,
      "training_step_time": 8.22288966178894
    },
    {
      "epoch": 3.5813331604003904e-05,
      "model_forward_time": 1.7308943271636963,
      "step": 37553
    },
    {
      "epoch": 3.5813331604003904e-05,
      "step": 37553,
      "training_step_time": 7.623734951019287
    },
    {
      "epoch": 3.5814285278320314e-05,
      "model_forward_time": 1.715620517730713,
      "step": 37554
    },
    {
      "epoch": 3.5814285278320314e-05,
      "step": 37554,
      "training_step_time": 7.631102561950684
    },
    {
      "epoch": 3.581523895263672e-05,
      "model_forward_time": 1.7146275043487549,
      "step": 37555
    },
    {
      "epoch": 3.581523895263672e-05,
      "step": 37555,
      "training_step_time": 7.656841039657593
    },
    {
      "epoch": 3.581619262695312e-05,
      "model_forward_time": 2.9090263843536377,
      "step": 37556
    },
    {
      "epoch": 3.581619262695312e-05,
      "step": 37556,
      "training_step_time": 7.698164701461792
    },
    {
      "epoch": 3.581714630126953e-05,
      "model_forward_time": 1.7146461009979248,
      "step": 37557
    },
    {
      "epoch": 3.581714630126953e-05,
      "step": 37557,
      "training_step_time": 7.66508936882019
    },
    {
      "epoch": 3.581809997558594e-05,
      "model_forward_time": 1.7274646759033203,
      "step": 37558
    },
    {
      "epoch": 3.581809997558594e-05,
      "step": 37558,
      "training_step_time": 7.693350076675415
    },
    {
      "epoch": 3.581905364990234e-05,
      "model_forward_time": 1.7239892482757568,
      "step": 37559
    },
    {
      "epoch": 3.581905364990234e-05,
      "step": 37559,
      "training_step_time": 7.712970733642578
    },
    {
      "epoch": 3.582000732421875e-05,
      "grad_norm": 0.12003994733095169,
      "learning_rate": 7.371152759587416e-06,
      "loss": 0.0607,
      "step": 37560
    },
    {
      "epoch": 3.582000732421875e-05,
      "model_forward_time": 1.724177360534668,
      "step": 37560
    },
    {
      "dynamics_loss_avg": 0.048897410929203036,
      "epoch": 3.582000732421875e-05,
      "step": 37560
    },
    {
      "action_loss_avg": 0.0024792305863229557,
      "epoch": 3.582000732421875e-05,
      "step": 37560
    },
    {
      "epoch": 3.582000732421875e-05,
      "step": 37560,
      "training_step_time": 8.887036323547363
    },
    {
      "epoch": 3.5820960998535156e-05,
      "model_forward_time": 1.723198413848877,
      "step": 37561
    },
    {
      "epoch": 3.5820960998535156e-05,
      "step": 37561,
      "training_step_time": 9.386176347732544
    },
    {
      "epoch": 3.582191467285156e-05,
      "model_forward_time": 1.7429990768432617,
      "step": 37562
    },
    {
      "epoch": 3.582191467285156e-05,
      "step": 37562,
      "training_step_time": 7.665732383728027
    },
    {
      "epoch": 3.582286834716797e-05,
      "model_forward_time": 1.7215580940246582,
      "step": 37563
    },
    {
      "epoch": 3.582286834716797e-05,
      "step": 37563,
      "training_step_time": 7.577598571777344
    },
    {
      "epoch": 3.5823822021484375e-05,
      "model_forward_time": 1.7245275974273682,
      "step": 37564
    },
    {
      "epoch": 3.5823822021484375e-05,
      "step": 37564,
      "training_step_time": 7.5972068309783936
    },
    {
      "epoch": 3.582477569580078e-05,
      "model_forward_time": 2.8218741416931152,
      "step": 37565
    },
    {
      "epoch": 3.582477569580078e-05,
      "step": 37565,
      "training_step_time": 8.295072793960571
    },
    {
      "epoch": 3.582572937011719e-05,
      "model_forward_time": 1.7177276611328125,
      "step": 37566
    },
    {
      "epoch": 3.582572937011719e-05,
      "step": 37566,
      "training_step_time": 8.413463354110718
    },
    {
      "epoch": 3.5826683044433594e-05,
      "model_forward_time": 1.7363665103912354,
      "step": 37567
    },
    {
      "epoch": 3.5826683044433594e-05,
      "step": 37567,
      "training_step_time": 8.650431156158447
    },
    {
      "epoch": 3.582763671875e-05,
      "model_forward_time": 1.7190978527069092,
      "step": 37568
    },
    {
      "epoch": 3.582763671875e-05,
      "step": 37568,
      "training_step_time": 7.7441534996032715
    },
    {
      "epoch": 3.582859039306641e-05,
      "model_forward_time": 1.7197062969207764,
      "step": 37569
    },
    {
      "epoch": 3.582859039306641e-05,
      "step": 37569,
      "training_step_time": 9.771154403686523
    },
    {
      "epoch": 3.582954406738281e-05,
      "grad_norm": 0.08865378797054291,
      "learning_rate": 7.369696913984572e-06,
      "loss": 0.0611,
      "step": 37570
    },
    {
      "epoch": 3.582954406738281e-05,
      "model_forward_time": 1.7313532829284668,
      "step": 37570
    },
    {
      "dynamics_loss_avg": 0.0823561497963965,
      "epoch": 3.582954406738281e-05,
      "step": 37570
    },
    {
      "action_loss_avg": 0.0040726637584157285,
      "epoch": 3.582954406738281e-05,
      "step": 37570
    },
    {
      "epoch": 3.582954406738281e-05,
      "step": 37570,
      "training_step_time": 8.59652590751648
    },
    {
      "epoch": 3.583049774169922e-05,
      "model_forward_time": 1.856576919555664,
      "step": 37571
    },
    {
      "epoch": 3.583049774169922e-05,
      "step": 37571,
      "training_step_time": 8.697830200195312
    },
    {
      "epoch": 3.583145141601563e-05,
      "model_forward_time": 1.7328660488128662,
      "step": 37572
    },
    {
      "epoch": 3.583145141601563e-05,
      "step": 37572,
      "training_step_time": 7.7148497104644775
    },
    {
      "epoch": 3.583240509033203e-05,
      "model_forward_time": 1.7376625537872314,
      "step": 37573
    },
    {
      "epoch": 3.583240509033203e-05,
      "step": 37573,
      "training_step_time": 7.638760089874268
    },
    {
      "epoch": 3.5833358764648436e-05,
      "model_forward_time": 1.7416462898254395,
      "step": 37574
    },
    {
      "epoch": 3.5833358764648436e-05,
      "step": 37574,
      "training_step_time": 9.060206413269043
    },
    {
      "epoch": 3.5834312438964846e-05,
      "model_forward_time": 1.729811668395996,
      "step": 37575
    },
    {
      "epoch": 3.5834312438964846e-05,
      "step": 37575,
      "training_step_time": 7.574816703796387
    },
    {
      "epoch": 3.583526611328125e-05,
      "model_forward_time": 1.7612967491149902,
      "step": 37576
    },
    {
      "epoch": 3.583526611328125e-05,
      "step": 37576,
      "training_step_time": 7.651480197906494
    },
    {
      "epoch": 3.5836219787597655e-05,
      "model_forward_time": 1.7353320121765137,
      "step": 37577
    },
    {
      "epoch": 3.5836219787597655e-05,
      "step": 37577,
      "training_step_time": 9.228866815567017
    },
    {
      "epoch": 3.5837173461914065e-05,
      "model_forward_time": 1.727952480316162,
      "step": 37578
    },
    {
      "epoch": 3.5837173461914065e-05,
      "step": 37578,
      "training_step_time": 8.493810415267944
    },
    {
      "epoch": 3.583812713623047e-05,
      "model_forward_time": 1.7534756660461426,
      "step": 37579
    },
    {
      "epoch": 3.583812713623047e-05,
      "step": 37579,
      "training_step_time": 8.67685580253601
    },
    {
      "epoch": 3.583908081054687e-05,
      "grad_norm": 0.13340593874454498,
      "learning_rate": 7.3682408092352364e-06,
      "loss": 0.0613,
      "step": 37580
    },
    {
      "epoch": 3.583908081054687e-05,
      "model_forward_time": 1.4221131801605225,
      "step": 37580
    },
    {
      "dynamics_loss_avg": 0.04423765502870083,
      "epoch": 3.583908081054687e-05,
      "step": 37580
    },
    {
      "action_loss_avg": 0.0013599204772617667,
      "epoch": 3.583908081054687e-05,
      "step": 37580
    },
    {
      "epoch": 3.583908081054687e-05,
      "step": 37580,
      "training_step_time": 7.67101526260376
    },
    {
      "epoch": 3.5840034484863284e-05,
      "model_forward_time": 1.7486324310302734,
      "step": 37581
    },
    {
      "epoch": 3.5840034484863284e-05,
      "step": 37581,
      "training_step_time": 7.68189549446106
    },
    {
      "epoch": 3.584098815917969e-05,
      "model_forward_time": 1.734830379486084,
      "step": 37582
    },
    {
      "epoch": 3.584098815917969e-05,
      "step": 37582,
      "training_step_time": 7.638290882110596
    },
    {
      "epoch": 3.584194183349609e-05,
      "model_forward_time": 1.7380399703979492,
      "step": 37583
    },
    {
      "epoch": 3.584194183349609e-05,
      "step": 37583,
      "training_step_time": 7.7784788608551025
    },
    {
      "epoch": 3.58428955078125e-05,
      "model_forward_time": 2.9100406169891357,
      "step": 37584
    },
    {
      "epoch": 3.58428955078125e-05,
      "step": 37584,
      "training_step_time": 7.711825132369995
    },
    {
      "epoch": 3.584384918212891e-05,
      "model_forward_time": 1.727710485458374,
      "step": 37585
    },
    {
      "epoch": 3.584384918212891e-05,
      "step": 37585,
      "training_step_time": 7.786791563034058
    },
    {
      "epoch": 3.584480285644531e-05,
      "model_forward_time": 1.741882085800171,
      "step": 37586
    },
    {
      "epoch": 3.584480285644531e-05,
      "step": 37586,
      "training_step_time": 7.619824171066284
    },
    {
      "epoch": 3.584575653076172e-05,
      "model_forward_time": 1.759202480316162,
      "step": 37587
    },
    {
      "epoch": 3.584575653076172e-05,
      "step": 37587,
      "training_step_time": 8.71301794052124
    },
    {
      "epoch": 3.5846710205078126e-05,
      "model_forward_time": 1.7433669567108154,
      "step": 37588
    },
    {
      "epoch": 3.5846710205078126e-05,
      "step": 37588,
      "training_step_time": 8.16677713394165
    },
    {
      "epoch": 3.584766387939453e-05,
      "model_forward_time": 1.7896723747253418,
      "step": 37589
    },
    {
      "epoch": 3.584766387939453e-05,
      "step": 37589,
      "training_step_time": 8.552757501602173
    },
    {
      "epoch": 3.584861755371094e-05,
      "grad_norm": 0.16614490747451782,
      "learning_rate": 7.366784445498648e-06,
      "loss": 0.0618,
      "step": 37590
    },
    {
      "epoch": 3.584861755371094e-05,
      "model_forward_time": 1.7380475997924805,
      "step": 37590
    },
    {
      "dynamics_loss_avg": 0.05557983350008726,
      "epoch": 3.584861755371094e-05,
      "step": 37590
    },
    {
      "action_loss_avg": 0.0018154853751184419,
      "epoch": 3.584861755371094e-05,
      "step": 37590
    },
    {
      "epoch": 3.584861755371094e-05,
      "step": 37590,
      "training_step_time": 7.677904367446899
    },
    {
      "epoch": 3.5849571228027345e-05,
      "model_forward_time": 1.7482349872589111,
      "step": 37591
    },
    {
      "epoch": 3.5849571228027345e-05,
      "step": 37591,
      "training_step_time": 7.64275860786438
    },
    {
      "epoch": 3.585052490234375e-05,
      "model_forward_time": 1.7464470863342285,
      "step": 37592
    },
    {
      "epoch": 3.585052490234375e-05,
      "step": 37592,
      "training_step_time": 7.776214122772217
    },
    {
      "epoch": 3.585147857666016e-05,
      "model_forward_time": 2.886461019515991,
      "step": 37593
    },
    {
      "epoch": 3.585147857666016e-05,
      "step": 37593,
      "training_step_time": 8.466890335083008
    },
    {
      "epoch": 3.5852432250976564e-05,
      "model_forward_time": 1.7429008483886719,
      "step": 37594
    },
    {
      "epoch": 3.5852432250976564e-05,
      "step": 37594,
      "training_step_time": 8.491117477416992
    },
    {
      "epoch": 3.585338592529297e-05,
      "model_forward_time": 1.755688190460205,
      "step": 37595
    },
    {
      "epoch": 3.585338592529297e-05,
      "step": 37595,
      "training_step_time": 7.733964204788208
    },
    {
      "epoch": 3.585433959960937e-05,
      "model_forward_time": 1.740617036819458,
      "step": 37596
    },
    {
      "epoch": 3.585433959960937e-05,
      "step": 37596,
      "training_step_time": 8.639878749847412
    },
    {
      "epoch": 3.585529327392578e-05,
      "model_forward_time": 1.7539970874786377,
      "step": 37597
    },
    {
      "epoch": 3.585529327392578e-05,
      "step": 37597,
      "training_step_time": 7.736956834793091
    },
    {
      "epoch": 3.5856246948242187e-05,
      "model_forward_time": 1.768768548965454,
      "step": 37598
    },
    {
      "epoch": 3.5856246948242187e-05,
      "step": 37598,
      "training_step_time": 9.380274057388306
    },
    {
      "epoch": 3.585720062255859e-05,
      "model_forward_time": 1.7663209438323975,
      "step": 37599
    },
    {
      "epoch": 3.585720062255859e-05,
      "step": 37599,
      "training_step_time": 7.678609371185303
    },
    {
      "epoch": 3.5858154296875e-05,
      "grad_norm": 0.1884305477142334,
      "learning_rate": 7.36532782293407e-06,
      "loss": 0.0636,
      "step": 37600
    },
    {
      "epoch": 3.5858154296875e-05,
      "model_forward_time": 1.7466318607330322,
      "step": 37600
    },
    {
      "dynamics_loss_avg": 0.05844593048095703,
      "epoch": 3.5858154296875e-05,
      "step": 37600
    },
    {
      "action_loss_avg": 0.002203640475636348,
      "epoch": 3.5858154296875e-05,
      "step": 37600
    },
    {
      "epoch": 3.5858154296875e-05,
      "step": 37600,
      "training_step_time": 7.676489591598511
    },
    {
      "epoch": 3.5859107971191405e-05,
      "model_forward_time": 1.7461066246032715,
      "step": 37601
    },
    {
      "epoch": 3.5859107971191405e-05,
      "step": 37601,
      "training_step_time": 7.756437540054321
    },
    {
      "epoch": 3.586006164550781e-05,
      "model_forward_time": 2.962045669555664,
      "step": 37602
    },
    {
      "epoch": 3.586006164550781e-05,
      "step": 37602,
      "training_step_time": 7.80817437171936
    },
    {
      "epoch": 3.586101531982422e-05,
      "model_forward_time": 1.7561335563659668,
      "step": 37603
    },
    {
      "epoch": 3.586101531982422e-05,
      "step": 37603,
      "training_step_time": 7.62117075920105
    },
    {
      "epoch": 3.5861968994140624e-05,
      "model_forward_time": 1.7375726699829102,
      "step": 37604
    },
    {
      "epoch": 3.5861968994140624e-05,
      "step": 37604,
      "training_step_time": 7.677220344543457
    },
    {
      "epoch": 3.586292266845703e-05,
      "model_forward_time": 1.7636265754699707,
      "step": 37605
    },
    {
      "epoch": 3.586292266845703e-05,
      "step": 37605,
      "training_step_time": 8.22220516204834
    },
    {
      "epoch": 3.586387634277344e-05,
      "model_forward_time": 1.7482614517211914,
      "step": 37606
    },
    {
      "epoch": 3.586387634277344e-05,
      "step": 37606,
      "training_step_time": 9.823670148849487
    },
    {
      "epoch": 3.586483001708984e-05,
      "model_forward_time": 1.8392016887664795,
      "step": 37607
    },
    {
      "epoch": 3.586483001708984e-05,
      "step": 37607,
      "training_step_time": 9.573669910430908
    },
    {
      "epoch": 3.586578369140625e-05,
      "model_forward_time": 1.839141845703125,
      "step": 37608
    },
    {
      "epoch": 3.586578369140625e-05,
      "step": 37608,
      "training_step_time": 7.670331239700317
    },
    {
      "epoch": 3.586673736572266e-05,
      "model_forward_time": 1.7398936748504639,
      "step": 37609
    },
    {
      "epoch": 3.586673736572266e-05,
      "step": 37609,
      "training_step_time": 7.747784376144409
    },
    {
      "epoch": 3.586769104003906e-05,
      "grad_norm": 0.10598456114530563,
      "learning_rate": 7.363870941700797e-06,
      "loss": 0.0603,
      "step": 37610
    },
    {
      "epoch": 3.586769104003906e-05,
      "model_forward_time": 1.7448227405548096,
      "step": 37610
    },
    {
      "dynamics_loss_avg": 0.05820580795407295,
      "epoch": 3.586769104003906e-05,
      "step": 37610
    },
    {
      "action_loss_avg": 0.002075045497622341,
      "epoch": 3.586769104003906e-05,
      "step": 37610
    },
    {
      "epoch": 3.586769104003906e-05,
      "step": 37610,
      "training_step_time": 8.606340169906616
    },
    {
      "epoch": 3.5868644714355466e-05,
      "model_forward_time": 2.129800796508789,
      "step": 37611
    },
    {
      "epoch": 3.5868644714355466e-05,
      "step": 37611,
      "training_step_time": 8.074905157089233
    },
    {
      "epoch": 3.586959838867188e-05,
      "model_forward_time": 1.7582621574401855,
      "step": 37612
    },
    {
      "epoch": 3.586959838867188e-05,
      "step": 37612,
      "training_step_time": 7.713414192199707
    },
    {
      "epoch": 3.587055206298828e-05,
      "model_forward_time": 1.7469346523284912,
      "step": 37613
    },
    {
      "epoch": 3.587055206298828e-05,
      "step": 37613,
      "training_step_time": 7.701180696487427
    },
    {
      "epoch": 3.5871505737304685e-05,
      "model_forward_time": 1.746819019317627,
      "step": 37614
    },
    {
      "epoch": 3.5871505737304685e-05,
      "step": 37614,
      "training_step_time": 8.71079134941101
    },
    {
      "epoch": 3.5872459411621096e-05,
      "model_forward_time": 1.7220828533172607,
      "step": 37615
    },
    {
      "epoch": 3.5872459411621096e-05,
      "step": 37615,
      "training_step_time": 7.721972465515137
    },
    {
      "epoch": 3.58734130859375e-05,
      "model_forward_time": 1.7536506652832031,
      "step": 37616
    },
    {
      "epoch": 3.58734130859375e-05,
      "step": 37616,
      "training_step_time": 7.696761131286621
    },
    {
      "epoch": 3.5874366760253904e-05,
      "model_forward_time": 1.8192744255065918,
      "step": 37617
    },
    {
      "epoch": 3.5874366760253904e-05,
      "step": 37617,
      "training_step_time": 7.730906009674072
    },
    {
      "epoch": 3.5875320434570315e-05,
      "model_forward_time": 1.7486193180084229,
      "step": 37618
    },
    {
      "epoch": 3.5875320434570315e-05,
      "step": 37618,
      "training_step_time": 7.712629556655884
    },
    {
      "epoch": 3.587627410888672e-05,
      "model_forward_time": 1.7561588287353516,
      "step": 37619
    },
    {
      "epoch": 3.587627410888672e-05,
      "step": 37619,
      "training_step_time": 7.622894048690796
    },
    {
      "epoch": 3.587722778320312e-05,
      "grad_norm": 0.13635018467903137,
      "learning_rate": 7.362413801958152e-06,
      "loss": 0.0583,
      "step": 37620
    },
    {
      "epoch": 3.587722778320312e-05,
      "model_forward_time": 2.947519540786743,
      "step": 37620
    },
    {
      "dynamics_loss_avg": 0.05357546992599964,
      "epoch": 3.587722778320312e-05,
      "step": 37620
    },
    {
      "action_loss_avg": 0.0014890450634993612,
      "epoch": 3.587722778320312e-05,
      "step": 37620
    },
    {
      "epoch": 3.587722778320312e-05,
      "step": 37620,
      "training_step_time": 7.84649395942688
    },
    {
      "epoch": 3.587818145751953e-05,
      "model_forward_time": 1.7396068572998047,
      "step": 37621
    },
    {
      "epoch": 3.587818145751953e-05,
      "step": 37621,
      "training_step_time": 8.814296245574951
    },
    {
      "epoch": 3.587913513183594e-05,
      "model_forward_time": 1.7432787418365479,
      "step": 37622
    },
    {
      "epoch": 3.587913513183594e-05,
      "step": 37622,
      "training_step_time": 7.748835802078247
    },
    {
      "epoch": 3.588008880615234e-05,
      "model_forward_time": 1.048032283782959,
      "step": 37623
    },
    {
      "epoch": 3.588008880615234e-05,
      "step": 37623,
      "training_step_time": 7.594867467880249
    },
    {
      "epoch": 3.588104248046875e-05,
      "model_forward_time": 1.7452492713928223,
      "step": 37624
    },
    {
      "epoch": 3.588104248046875e-05,
      "step": 37624,
      "training_step_time": 7.86591911315918
    },
    {
      "epoch": 3.5881996154785156e-05,
      "model_forward_time": 1.738149642944336,
      "step": 37625
    },
    {
      "epoch": 3.5881996154785156e-05,
      "step": 37625,
      "training_step_time": 7.735973596572876
    },
    {
      "epoch": 3.588294982910156e-05,
      "model_forward_time": 1.7866356372833252,
      "step": 37626
    },
    {
      "epoch": 3.588294982910156e-05,
      "step": 37626,
      "training_step_time": 7.762096881866455
    },
    {
      "epoch": 3.588390350341797e-05,
      "model_forward_time": 1.74391770362854,
      "step": 37627
    },
    {
      "epoch": 3.588390350341797e-05,
      "step": 37627,
      "training_step_time": 7.731762409210205
    },
    {
      "epoch": 3.5884857177734375e-05,
      "model_forward_time": 1.7441375255584717,
      "step": 37628
    },
    {
      "epoch": 3.5884857177734375e-05,
      "step": 37628,
      "training_step_time": 7.691344499588013
    },
    {
      "epoch": 3.588581085205078e-05,
      "model_forward_time": 1.7595188617706299,
      "step": 37629
    },
    {
      "epoch": 3.588581085205078e-05,
      "step": 37629,
      "training_step_time": 8.003203868865967
    },
    {
      "epoch": 3.588676452636719e-05,
      "grad_norm": 0.16031485795974731,
      "learning_rate": 7.360956403865487e-06,
      "loss": 0.0628,
      "step": 37630
    },
    {
      "epoch": 3.588676452636719e-05,
      "model_forward_time": 1.7442963123321533,
      "step": 37630
    },
    {
      "dynamics_loss_avg": 0.07573907636106014,
      "epoch": 3.588676452636719e-05,
      "step": 37630
    },
    {
      "action_loss_avg": 0.003385353193152696,
      "epoch": 3.588676452636719e-05,
      "step": 37630
    },
    {
      "epoch": 3.588676452636719e-05,
      "step": 37630,
      "training_step_time": 8.165104150772095
    },
    {
      "epoch": 3.5887718200683594e-05,
      "model_forward_time": 1.7439136505126953,
      "step": 37631
    },
    {
      "epoch": 3.5887718200683594e-05,
      "step": 37631,
      "training_step_time": 8.225178956985474
    },
    {
      "epoch": 3.5888671875e-05,
      "model_forward_time": 1.7596361637115479,
      "step": 37632
    },
    {
      "epoch": 3.5888671875e-05,
      "step": 37632,
      "training_step_time": 7.621685981750488
    },
    {
      "epoch": 3.588962554931641e-05,
      "model_forward_time": 1.7474627494812012,
      "step": 37633
    },
    {
      "epoch": 3.588962554931641e-05,
      "step": 37633,
      "training_step_time": 7.771581649780273
    },
    {
      "epoch": 3.589057922363281e-05,
      "model_forward_time": 1.7581846714019775,
      "step": 37634
    },
    {
      "epoch": 3.589057922363281e-05,
      "step": 37634,
      "training_step_time": 7.7023327350616455
    },
    {
      "epoch": 3.589153289794922e-05,
      "model_forward_time": 1.7407543659210205,
      "step": 37635
    },
    {
      "epoch": 3.589153289794922e-05,
      "step": 37635,
      "training_step_time": 9.102408170700073
    },
    {
      "epoch": 3.589248657226563e-05,
      "model_forward_time": 1.7475519180297852,
      "step": 37636
    },
    {
      "epoch": 3.589248657226563e-05,
      "step": 37636,
      "training_step_time": 8.272318363189697
    },
    {
      "epoch": 3.589344024658203e-05,
      "model_forward_time": 1.7215015888214111,
      "step": 37637
    },
    {
      "epoch": 3.589344024658203e-05,
      "step": 37637,
      "training_step_time": 8.544373750686646
    },
    {
      "epoch": 3.5894393920898436e-05,
      "model_forward_time": 1.7490875720977783,
      "step": 37638
    },
    {
      "epoch": 3.5894393920898436e-05,
      "step": 37638,
      "training_step_time": 7.773084402084351
    },
    {
      "epoch": 3.5895347595214847e-05,
      "model_forward_time": 2.951324939727783,
      "step": 37639
    },
    {
      "epoch": 3.5895347595214847e-05,
      "step": 37639,
      "training_step_time": 7.74714469909668
    },
    {
      "epoch": 3.589630126953125e-05,
      "grad_norm": 0.14687812328338623,
      "learning_rate": 7.359498747582178e-06,
      "loss": 0.0638,
      "step": 37640
    },
    {
      "epoch": 3.589630126953125e-05,
      "model_forward_time": 1.7493739128112793,
      "step": 37640
    },
    {
      "dynamics_loss_avg": 0.06046528276056051,
      "epoch": 3.589630126953125e-05,
      "step": 37640
    },
    {
      "action_loss_avg": 0.0030397210124647246,
      "epoch": 3.589630126953125e-05,
      "step": 37640
    },
    {
      "epoch": 3.589630126953125e-05,
      "step": 37640,
      "training_step_time": 7.628458738327026
    },
    {
      "epoch": 3.5897254943847655e-05,
      "model_forward_time": 1.741995096206665,
      "step": 37641
    },
    {
      "epoch": 3.5897254943847655e-05,
      "step": 37641,
      "training_step_time": 8.652317762374878
    },
    {
      "epoch": 3.5898208618164065e-05,
      "model_forward_time": 1.7538883686065674,
      "step": 37642
    },
    {
      "epoch": 3.5898208618164065e-05,
      "step": 37642,
      "training_step_time": 8.410282135009766
    },
    {
      "epoch": 3.589916229248047e-05,
      "model_forward_time": 1.751713514328003,
      "step": 37643
    },
    {
      "epoch": 3.589916229248047e-05,
      "step": 37643,
      "training_step_time": 8.686996936798096
    },
    {
      "epoch": 3.5900115966796873e-05,
      "model_forward_time": 1.7556965351104736,
      "step": 37644
    },
    {
      "epoch": 3.5900115966796873e-05,
      "step": 37644,
      "training_step_time": 8.64491581916809
    },
    {
      "epoch": 3.5901069641113284e-05,
      "model_forward_time": 1.8763477802276611,
      "step": 37645
    },
    {
      "epoch": 3.5901069641113284e-05,
      "step": 37645,
      "training_step_time": 8.522756338119507
    },
    {
      "epoch": 3.590202331542969e-05,
      "model_forward_time": 1.737147331237793,
      "step": 37646
    },
    {
      "epoch": 3.590202331542969e-05,
      "step": 37646,
      "training_step_time": 7.657903432846069
    },
    {
      "epoch": 3.590297698974609e-05,
      "model_forward_time": 1.7495853900909424,
      "step": 37647
    },
    {
      "epoch": 3.590297698974609e-05,
      "step": 37647,
      "training_step_time": 8.558149576187134
    },
    {
      "epoch": 3.59039306640625e-05,
      "model_forward_time": 2.9204883575439453,
      "step": 37648
    },
    {
      "epoch": 3.59039306640625e-05,
      "step": 37648,
      "training_step_time": 8.769970893859863
    },
    {
      "epoch": 3.590488433837891e-05,
      "model_forward_time": 1.7464091777801514,
      "step": 37649
    },
    {
      "epoch": 3.590488433837891e-05,
      "step": 37649,
      "training_step_time": 7.631522178649902
    },
    {
      "epoch": 3.590583801269531e-05,
      "grad_norm": 0.14414730668067932,
      "learning_rate": 7.358040833267634e-06,
      "loss": 0.0588,
      "step": 37650
    },
    {
      "epoch": 3.590583801269531e-05,
      "model_forward_time": 1.737330436706543,
      "step": 37650
    },
    {
      "dynamics_loss_avg": 0.06484834365546703,
      "epoch": 3.590583801269531e-05,
      "step": 37650
    },
    {
      "action_loss_avg": 0.0038015477010048928,
      "epoch": 3.590583801269531e-05,
      "step": 37650
    },
    {
      "epoch": 3.590583801269531e-05,
      "step": 37650,
      "training_step_time": 8.224449396133423
    },
    {
      "epoch": 3.590679168701172e-05,
      "model_forward_time": 1.7580170631408691,
      "step": 37651
    },
    {
      "epoch": 3.590679168701172e-05,
      "step": 37651,
      "training_step_time": 7.701534271240234
    },
    {
      "epoch": 3.5907745361328126e-05,
      "model_forward_time": 1.7416350841522217,
      "step": 37652
    },
    {
      "epoch": 3.5907745361328126e-05,
      "step": 37652,
      "training_step_time": 8.37795376777649
    },
    {
      "epoch": 3.590869903564453e-05,
      "model_forward_time": 1.728522777557373,
      "step": 37653
    },
    {
      "epoch": 3.590869903564453e-05,
      "step": 37653,
      "training_step_time": 7.646326780319214
    },
    {
      "epoch": 3.590965270996094e-05,
      "model_forward_time": 1.8390235900878906,
      "step": 37654
    },
    {
      "epoch": 3.590965270996094e-05,
      "step": 37654,
      "training_step_time": 7.712452173233032
    },
    {
      "epoch": 3.5910606384277345e-05,
      "model_forward_time": 1.3941724300384521,
      "step": 37655
    },
    {
      "epoch": 3.5910606384277345e-05,
      "step": 37655,
      "training_step_time": 7.602112293243408
    },
    {
      "epoch": 3.591156005859375e-05,
      "model_forward_time": 1.061079502105713,
      "step": 37656
    },
    {
      "epoch": 3.591156005859375e-05,
      "step": 37656,
      "training_step_time": 7.658352375030518
    },
    {
      "epoch": 3.591251373291015e-05,
      "model_forward_time": 2.8646931648254395,
      "step": 37657
    },
    {
      "epoch": 3.591251373291015e-05,
      "step": 37657,
      "training_step_time": 7.701740503311157
    },
    {
      "epoch": 3.5913467407226564e-05,
      "model_forward_time": 1.7577829360961914,
      "step": 37658
    },
    {
      "epoch": 3.5913467407226564e-05,
      "step": 37658,
      "training_step_time": 8.165769815444946
    },
    {
      "epoch": 3.591442108154297e-05,
      "model_forward_time": 1.750617504119873,
      "step": 37659
    },
    {
      "epoch": 3.591442108154297e-05,
      "step": 37659,
      "training_step_time": 7.724900960922241
    },
    {
      "epoch": 3.591537475585937e-05,
      "grad_norm": 0.1625562459230423,
      "learning_rate": 7.356582661081289e-06,
      "loss": 0.0597,
      "step": 37660
    },
    {
      "epoch": 3.591537475585937e-05,
      "model_forward_time": 1.7421023845672607,
      "step": 37660
    },
    {
      "dynamics_loss_avg": 0.06023891530930996,
      "epoch": 3.591537475585937e-05,
      "step": 37660
    },
    {
      "action_loss_avg": 0.00219056797795929,
      "epoch": 3.591537475585937e-05,
      "step": 37660
    },
    {
      "epoch": 3.591537475585937e-05,
      "step": 37660,
      "training_step_time": 8.29315185546875
    },
    {
      "epoch": 3.591632843017578e-05,
      "model_forward_time": 1.7550313472747803,
      "step": 37661
    },
    {
      "epoch": 3.591632843017578e-05,
      "step": 37661,
      "training_step_time": 8.685611724853516
    },
    {
      "epoch": 3.5917282104492187e-05,
      "model_forward_time": 1.7609570026397705,
      "step": 37662
    },
    {
      "epoch": 3.5917282104492187e-05,
      "step": 37662,
      "training_step_time": 7.645950794219971
    },
    {
      "epoch": 3.591823577880859e-05,
      "model_forward_time": 1.8331103324890137,
      "step": 37663
    },
    {
      "epoch": 3.591823577880859e-05,
      "step": 37663,
      "training_step_time": 8.555206775665283
    },
    {
      "epoch": 3.5919189453125e-05,
      "model_forward_time": 1.7500109672546387,
      "step": 37664
    },
    {
      "epoch": 3.5919189453125e-05,
      "step": 37664,
      "training_step_time": 8.542094469070435
    },
    {
      "epoch": 3.5920143127441405e-05,
      "model_forward_time": 1.7390854358673096,
      "step": 37665
    },
    {
      "epoch": 3.5920143127441405e-05,
      "step": 37665,
      "training_step_time": 7.690037488937378
    },
    {
      "epoch": 3.592109680175781e-05,
      "model_forward_time": 1.753901481628418,
      "step": 37666
    },
    {
      "epoch": 3.592109680175781e-05,
      "step": 37666,
      "training_step_time": 7.664693832397461
    },
    {
      "epoch": 3.592205047607422e-05,
      "model_forward_time": 3.0216009616851807,
      "step": 37667
    },
    {
      "epoch": 3.592205047607422e-05,
      "step": 37667,
      "training_step_time": 7.8499510288238525
    },
    {
      "epoch": 3.5923004150390624e-05,
      "model_forward_time": 1.7623965740203857,
      "step": 37668
    },
    {
      "epoch": 3.5923004150390624e-05,
      "step": 37668,
      "training_step_time": 8.17434811592102
    },
    {
      "epoch": 3.592395782470703e-05,
      "model_forward_time": 1.7477478981018066,
      "step": 37669
    },
    {
      "epoch": 3.592395782470703e-05,
      "step": 37669,
      "training_step_time": 7.814009428024292
    },
    {
      "epoch": 3.592491149902344e-05,
      "grad_norm": 0.17577716708183289,
      "learning_rate": 7.355124231182608e-06,
      "loss": 0.0629,
      "step": 37670
    },
    {
      "epoch": 3.592491149902344e-05,
      "model_forward_time": 1.7503876686096191,
      "step": 37670
    },
    {
      "dynamics_loss_avg": 0.07937292009592056,
      "epoch": 3.592491149902344e-05,
      "step": 37670
    },
    {
      "action_loss_avg": 0.002789615606889129,
      "epoch": 3.592491149902344e-05,
      "step": 37670
    },
    {
      "epoch": 3.592491149902344e-05,
      "step": 37670,
      "training_step_time": 7.776034832000732
    },
    {
      "epoch": 3.592586517333984e-05,
      "model_forward_time": 1.7590043544769287,
      "step": 37671
    },
    {
      "epoch": 3.592586517333984e-05,
      "step": 37671,
      "training_step_time": 8.787147045135498
    },
    {
      "epoch": 3.592681884765625e-05,
      "model_forward_time": 1.8067219257354736,
      "step": 37672
    },
    {
      "epoch": 3.592681884765625e-05,
      "step": 37672,
      "training_step_time": 9.280598163604736
    },
    {
      "epoch": 3.592777252197266e-05,
      "model_forward_time": 1.7458739280700684,
      "step": 37673
    },
    {
      "epoch": 3.592777252197266e-05,
      "step": 37673,
      "training_step_time": 7.616183519363403
    },
    {
      "epoch": 3.592872619628906e-05,
      "model_forward_time": 1.7440195083618164,
      "step": 37674
    },
    {
      "epoch": 3.592872619628906e-05,
      "step": 37674,
      "training_step_time": 8.658193588256836
    },
    {
      "epoch": 3.5929679870605466e-05,
      "model_forward_time": 1.7514326572418213,
      "step": 37675
    },
    {
      "epoch": 3.5929679870605466e-05,
      "step": 37675,
      "training_step_time": 7.701523542404175
    },
    {
      "epoch": 3.593063354492188e-05,
      "model_forward_time": 2.9751510620117188,
      "step": 37676
    },
    {
      "epoch": 3.593063354492188e-05,
      "step": 37676,
      "training_step_time": 8.795088052749634
    },
    {
      "epoch": 3.593158721923828e-05,
      "model_forward_time": 1.761124849319458,
      "step": 37677
    },
    {
      "epoch": 3.593158721923828e-05,
      "step": 37677,
      "training_step_time": 9.37353801727295
    },
    {
      "epoch": 3.5932540893554685e-05,
      "model_forward_time": 1.751441240310669,
      "step": 37678
    },
    {
      "epoch": 3.5932540893554685e-05,
      "step": 37678,
      "training_step_time": 6.807855606079102
    },
    {
      "epoch": 3.5933494567871096e-05,
      "model_forward_time": 1.7433805465698242,
      "step": 37679
    },
    {
      "epoch": 3.5933494567871096e-05,
      "step": 37679,
      "training_step_time": 6.768338203430176
    },
    {
      "epoch": 3.59344482421875e-05,
      "grad_norm": 0.1398201733827591,
      "learning_rate": 7.353665543731082e-06,
      "loss": 0.061,
      "step": 37680
    },
    {
      "epoch": 3.59344482421875e-05,
      "model_forward_time": 1.7486469745635986,
      "step": 37680
    },
    {
      "dynamics_loss_avg": 0.0557599195279181,
      "epoch": 3.59344482421875e-05,
      "step": 37680
    },
    {
      "action_loss_avg": 0.0017052665178198367,
      "epoch": 3.59344482421875e-05,
      "step": 37680
    },
    {
      "epoch": 3.59344482421875e-05,
      "step": 37680,
      "training_step_time": 6.757031679153442
    },
    {
      "epoch": 3.5935401916503904e-05,
      "model_forward_time": 1.751389980316162,
      "step": 37681
    },
    {
      "epoch": 3.5935401916503904e-05,
      "step": 37681,
      "training_step_time": 6.783440113067627
    },
    {
      "epoch": 3.5936355590820315e-05,
      "model_forward_time": 1.7443907260894775,
      "step": 37682
    },
    {
      "epoch": 3.5936355590820315e-05,
      "step": 37682,
      "training_step_time": 6.686551570892334
    },
    {
      "epoch": 3.593730926513672e-05,
      "model_forward_time": 1.7456648349761963,
      "step": 37683
    },
    {
      "epoch": 3.593730926513672e-05,
      "step": 37683,
      "training_step_time": 6.772222280502319
    },
    {
      "epoch": 3.593826293945312e-05,
      "model_forward_time": 1.7469382286071777,
      "step": 37684
    },
    {
      "epoch": 3.593826293945312e-05,
      "step": 37684,
      "training_step_time": 6.818225860595703
    },
    {
      "epoch": 3.5939216613769533e-05,
      "model_forward_time": 1.753450632095337,
      "step": 37685
    },
    {
      "epoch": 3.5939216613769533e-05,
      "step": 37685,
      "training_step_time": 6.7807817459106445
    },
    {
      "epoch": 3.594017028808594e-05,
      "model_forward_time": 2.926844835281372,
      "step": 37686
    },
    {
      "epoch": 3.594017028808594e-05,
      "step": 37686,
      "training_step_time": 7.7527756690979
    },
    {
      "epoch": 3.594112396240234e-05,
      "model_forward_time": 1.7540943622589111,
      "step": 37687
    },
    {
      "epoch": 3.594112396240234e-05,
      "step": 37687,
      "training_step_time": 7.677201747894287
    },
    {
      "epoch": 3.594207763671875e-05,
      "model_forward_time": 1.75315260887146,
      "step": 37688
    },
    {
      "epoch": 3.594207763671875e-05,
      "step": 37688,
      "training_step_time": 6.788727521896362
    },
    {
      "epoch": 3.5943031311035156e-05,
      "model_forward_time": 1.758439540863037,
      "step": 37689
    },
    {
      "epoch": 3.5943031311035156e-05,
      "step": 37689,
      "training_step_time": 6.794224500656128
    },
    {
      "epoch": 3.594398498535156e-05,
      "grad_norm": 0.16003380715847015,
      "learning_rate": 7.352206598886229e-06,
      "loss": 0.0625,
      "step": 37690
    },
    {
      "epoch": 3.594398498535156e-05,
      "model_forward_time": 1.7547307014465332,
      "step": 37690
    },
    {
      "dynamics_loss_avg": 0.060772957373410465,
      "epoch": 3.594398498535156e-05,
      "step": 37690
    },
    {
      "action_loss_avg": 0.003395039733732119,
      "epoch": 3.594398498535156e-05,
      "step": 37690
    },
    {
      "epoch": 3.594398498535156e-05,
      "step": 37690,
      "training_step_time": 6.813528776168823
    },
    {
      "epoch": 3.594493865966797e-05,
      "model_forward_time": 1.7568390369415283,
      "step": 37691
    },
    {
      "epoch": 3.594493865966797e-05,
      "step": 37691,
      "training_step_time": 6.8218159675598145
    },
    {
      "epoch": 3.5945892333984375e-05,
      "model_forward_time": 1.74477219581604,
      "step": 37692
    },
    {
      "epoch": 3.5945892333984375e-05,
      "step": 37692,
      "training_step_time": 6.750262260437012
    },
    {
      "epoch": 3.594684600830078e-05,
      "model_forward_time": 1.7550663948059082,
      "step": 37693
    },
    {
      "epoch": 3.594684600830078e-05,
      "step": 37693,
      "training_step_time": 6.743920564651489
    },
    {
      "epoch": 3.594779968261719e-05,
      "model_forward_time": 1.751509428024292,
      "step": 37694
    },
    {
      "epoch": 3.594779968261719e-05,
      "step": 37694,
      "training_step_time": 6.804843902587891
    },
    {
      "epoch": 3.5948753356933594e-05,
      "model_forward_time": 2.9112236499786377,
      "step": 37695
    },
    {
      "epoch": 3.5948753356933594e-05,
      "step": 37695,
      "training_step_time": 7.694992542266846
    },
    {
      "epoch": 3.594970703125e-05,
      "model_forward_time": 1.7593157291412354,
      "step": 37696
    },
    {
      "epoch": 3.594970703125e-05,
      "step": 37696,
      "training_step_time": 8.72679853439331
    },
    {
      "epoch": 3.595066070556641e-05,
      "model_forward_time": 1.7452716827392578,
      "step": 37697
    },
    {
      "epoch": 3.595066070556641e-05,
      "step": 37697,
      "training_step_time": 6.703766822814941
    },
    {
      "epoch": 3.595161437988281e-05,
      "model_forward_time": 1.744886875152588,
      "step": 37698
    },
    {
      "epoch": 3.595161437988281e-05,
      "step": 37698,
      "training_step_time": 6.695489168167114
    },
    {
      "epoch": 3.595256805419922e-05,
      "model_forward_time": 1.7503252029418945,
      "step": 37699
    },
    {
      "epoch": 3.595256805419922e-05,
      "step": 37699,
      "training_step_time": 6.708448886871338
    },
    {
      "epoch": 3.595352172851563e-05,
      "grad_norm": 0.19081728160381317,
      "learning_rate": 7.350747396807601e-06,
      "loss": 0.0659,
      "step": 37700
    },
    {
      "epoch": 3.595352172851563e-05,
      "model_forward_time": 1.7590656280517578,
      "step": 37700
    },
    {
      "dynamics_loss_avg": 0.07388873491436243,
      "epoch": 3.595352172851563e-05,
      "step": 37700
    },
    {
      "action_loss_avg": 0.004381057308637537,
      "epoch": 3.595352172851563e-05,
      "step": 37700
    },
    {
      "epoch": 3.595352172851563e-05,
      "step": 37700,
      "training_step_time": 6.755295753479004
    },
    {
      "epoch": 3.595447540283203e-05,
      "model_forward_time": 1.753899335861206,
      "step": 37701
    },
    {
      "epoch": 3.595447540283203e-05,
      "step": 37701,
      "training_step_time": 6.7192277908325195
    },
    {
      "epoch": 3.5955429077148436e-05,
      "model_forward_time": 1.7493295669555664,
      "step": 37702
    },
    {
      "epoch": 3.5955429077148436e-05,
      "step": 37702,
      "training_step_time": 6.697116374969482
    },
    {
      "epoch": 3.5956382751464847e-05,
      "model_forward_time": 1.7498273849487305,
      "step": 37703
    },
    {
      "epoch": 3.5956382751464847e-05,
      "step": 37703,
      "training_step_time": 6.723398208618164
    },
    {
      "epoch": 3.595733642578125e-05,
      "model_forward_time": 2.892859935760498,
      "step": 37704
    },
    {
      "epoch": 3.595733642578125e-05,
      "step": 37704,
      "training_step_time": 7.671969890594482
    },
    {
      "epoch": 3.5958290100097655e-05,
      "model_forward_time": 1.758495807647705,
      "step": 37705
    },
    {
      "epoch": 3.5958290100097655e-05,
      "step": 37705,
      "training_step_time": 9.28988003730774
    },
    {
      "epoch": 3.5959243774414065e-05,
      "model_forward_time": 1.38273286819458,
      "step": 37706
    },
    {
      "epoch": 3.5959243774414065e-05,
      "step": 37706,
      "training_step_time": 7.556190013885498
    },
    {
      "epoch": 3.596019744873047e-05,
      "model_forward_time": 1.7512483596801758,
      "step": 37707
    },
    {
      "epoch": 3.596019744873047e-05,
      "step": 37707,
      "training_step_time": 6.721417665481567
    },
    {
      "epoch": 3.5961151123046873e-05,
      "model_forward_time": 1.7563464641571045,
      "step": 37708
    },
    {
      "epoch": 3.5961151123046873e-05,
      "step": 37708,
      "training_step_time": 6.710217714309692
    },
    {
      "epoch": 3.5962104797363284e-05,
      "model_forward_time": 1.7820227146148682,
      "step": 37709
    },
    {
      "epoch": 3.5962104797363284e-05,
      "step": 37709,
      "training_step_time": 6.699100017547607
    },
    {
      "epoch": 3.596305847167969e-05,
      "grad_norm": 0.20302625000476837,
      "learning_rate": 7.349287937654772e-06,
      "loss": 0.0636,
      "step": 37710
    },
    {
      "epoch": 3.596305847167969e-05,
      "model_forward_time": 1.7558481693267822,
      "step": 37710
    },
    {
      "dynamics_loss_avg": 0.051861591450870034,
      "epoch": 3.596305847167969e-05,
      "step": 37710
    },
    {
      "action_loss_avg": 0.0013086983584798873,
      "epoch": 3.596305847167969e-05,
      "step": 37710
    },
    {
      "epoch": 3.596305847167969e-05,
      "step": 37710,
      "training_step_time": 6.753296136856079
    },
    {
      "epoch": 3.596401214599609e-05,
      "model_forward_time": 1.752678632736206,
      "step": 37711
    },
    {
      "epoch": 3.596401214599609e-05,
      "step": 37711,
      "training_step_time": 6.787384271621704
    },
    {
      "epoch": 3.59649658203125e-05,
      "model_forward_time": 1.7542834281921387,
      "step": 37712
    },
    {
      "epoch": 3.59649658203125e-05,
      "step": 37712,
      "training_step_time": 6.746838808059692
    },
    {
      "epoch": 3.596591949462891e-05,
      "model_forward_time": 2.930410385131836,
      "step": 37713
    },
    {
      "epoch": 3.596591949462891e-05,
      "step": 37713,
      "training_step_time": 7.72092080116272
    },
    {
      "epoch": 3.596687316894531e-05,
      "model_forward_time": 1.7576589584350586,
      "step": 37714
    },
    {
      "epoch": 3.596687316894531e-05,
      "step": 37714,
      "training_step_time": 9.988142251968384
    },
    {
      "epoch": 3.596782684326172e-05,
      "model_forward_time": 1.7426643371582031,
      "step": 37715
    },
    {
      "epoch": 3.596782684326172e-05,
      "step": 37715,
      "training_step_time": 7.647683620452881
    },
    {
      "epoch": 3.5968780517578126e-05,
      "model_forward_time": 1.7515075206756592,
      "step": 37716
    },
    {
      "epoch": 3.5968780517578126e-05,
      "step": 37716,
      "training_step_time": 6.80718469619751
    },
    {
      "epoch": 3.596973419189453e-05,
      "model_forward_time": 1.7619171142578125,
      "step": 37717
    },
    {
      "epoch": 3.596973419189453e-05,
      "step": 37717,
      "training_step_time": 6.7418739795684814
    },
    {
      "epoch": 3.597068786621094e-05,
      "model_forward_time": 1.7545316219329834,
      "step": 37718
    },
    {
      "epoch": 3.597068786621094e-05,
      "step": 37718,
      "training_step_time": 6.742101192474365
    },
    {
      "epoch": 3.5971641540527345e-05,
      "model_forward_time": 1.7863211631774902,
      "step": 37719
    },
    {
      "epoch": 3.5971641540527345e-05,
      "step": 37719,
      "training_step_time": 6.809598922729492
    },
    {
      "epoch": 3.597259521484375e-05,
      "grad_norm": 0.11864905804395676,
      "learning_rate": 7.347828221587345e-06,
      "loss": 0.063,
      "step": 37720
    },
    {
      "epoch": 3.597259521484375e-05,
      "model_forward_time": 1.7465953826904297,
      "step": 37720
    },
    {
      "dynamics_loss_avg": 0.05531351491808891,
      "epoch": 3.597259521484375e-05,
      "step": 37720
    },
    {
      "action_loss_avg": 0.0024389462661929427,
      "epoch": 3.597259521484375e-05,
      "step": 37720
    },
    {
      "epoch": 3.597259521484375e-05,
      "step": 37720,
      "training_step_time": 6.7749974727630615
    },
    {
      "epoch": 3.597354888916015e-05,
      "model_forward_time": 1.7543354034423828,
      "step": 37721
    },
    {
      "epoch": 3.597354888916015e-05,
      "step": 37721,
      "training_step_time": 6.750683784484863
    },
    {
      "epoch": 3.5974502563476564e-05,
      "model_forward_time": 1.752166509628296,
      "step": 37722
    },
    {
      "epoch": 3.5974502563476564e-05,
      "step": 37722,
      "training_step_time": 8.177828550338745
    },
    {
      "epoch": 3.597545623779297e-05,
      "model_forward_time": 1.8574049472808838,
      "step": 37723
    },
    {
      "epoch": 3.597545623779297e-05,
      "step": 37723,
      "training_step_time": 10.59540581703186
    },
    {
      "epoch": 3.597640991210937e-05,
      "model_forward_time": 1.7562057971954346,
      "step": 37724
    },
    {
      "epoch": 3.597640991210937e-05,
      "step": 37724,
      "training_step_time": 7.661117315292358
    },
    {
      "epoch": 3.597736358642578e-05,
      "model_forward_time": 1.7509486675262451,
      "step": 37725
    },
    {
      "epoch": 3.597736358642578e-05,
      "step": 37725,
      "training_step_time": 7.318302392959595
    },
    {
      "epoch": 3.597831726074219e-05,
      "model_forward_time": 1.7503662109375,
      "step": 37726
    },
    {
      "epoch": 3.597831726074219e-05,
      "step": 37726,
      "training_step_time": 6.775346279144287
    },
    {
      "epoch": 3.597927093505859e-05,
      "model_forward_time": 1.7572436332702637,
      "step": 37727
    },
    {
      "epoch": 3.597927093505859e-05,
      "step": 37727,
      "training_step_time": 6.765643358230591
    },
    {
      "epoch": 3.5980224609375e-05,
      "model_forward_time": 1.7385659217834473,
      "step": 37728
    },
    {
      "epoch": 3.5980224609375e-05,
      "step": 37728,
      "training_step_time": 6.673510313034058
    },
    {
      "epoch": 3.5981178283691405e-05,
      "model_forward_time": 1.7486896514892578,
      "step": 37729
    },
    {
      "epoch": 3.5981178283691405e-05,
      "step": 37729,
      "training_step_time": 6.738518714904785
    },
    {
      "epoch": 3.598213195800781e-05,
      "grad_norm": 0.1438482254743576,
      "learning_rate": 7.346368248764954e-06,
      "loss": 0.0585,
      "step": 37730
    },
    {
      "epoch": 3.598213195800781e-05,
      "model_forward_time": 1.7412848472595215,
      "step": 37730
    },
    {
      "dynamics_loss_avg": 0.06353397704660893,
      "epoch": 3.598213195800781e-05,
      "step": 37730
    },
    {
      "action_loss_avg": 0.0026758178253658118,
      "epoch": 3.598213195800781e-05,
      "step": 37730
    },
    {
      "epoch": 3.598213195800781e-05,
      "step": 37730,
      "training_step_time": 6.786576986312866
    },
    {
      "epoch": 3.598308563232422e-05,
      "model_forward_time": 1.7784347534179688,
      "step": 37731
    },
    {
      "epoch": 3.598308563232422e-05,
      "step": 37731,
      "training_step_time": 6.776980400085449
    },
    {
      "epoch": 3.5984039306640624e-05,
      "model_forward_time": 2.993154287338257,
      "step": 37732
    },
    {
      "epoch": 3.5984039306640624e-05,
      "step": 37732,
      "training_step_time": 11.79059386253357
    },
    {
      "epoch": 3.598499298095703e-05,
      "model_forward_time": 1.7470855712890625,
      "step": 37733
    },
    {
      "epoch": 3.598499298095703e-05,
      "step": 37733,
      "training_step_time": 7.716369867324829
    },
    {
      "epoch": 3.598594665527344e-05,
      "model_forward_time": 1.7510695457458496,
      "step": 37734
    },
    {
      "epoch": 3.598594665527344e-05,
      "step": 37734,
      "training_step_time": 7.27880859375
    },
    {
      "epoch": 3.598690032958984e-05,
      "model_forward_time": 1.7538294792175293,
      "step": 37735
    },
    {
      "epoch": 3.598690032958984e-05,
      "step": 37735,
      "training_step_time": 6.726619720458984
    },
    {
      "epoch": 3.598785400390625e-05,
      "model_forward_time": 1.7478876113891602,
      "step": 37736
    },
    {
      "epoch": 3.598785400390625e-05,
      "step": 37736,
      "training_step_time": 6.7530388832092285
    },
    {
      "epoch": 3.598880767822266e-05,
      "model_forward_time": 1.7458226680755615,
      "step": 37737
    },
    {
      "epoch": 3.598880767822266e-05,
      "step": 37737,
      "training_step_time": 6.807080984115601
    },
    {
      "epoch": 3.598976135253906e-05,
      "model_forward_time": 1.7520949840545654,
      "step": 37738
    },
    {
      "epoch": 3.598976135253906e-05,
      "step": 37738,
      "training_step_time": 6.81501841545105
    },
    {
      "epoch": 3.5990715026855466e-05,
      "model_forward_time": 1.7504448890686035,
      "step": 37739
    },
    {
      "epoch": 3.5990715026855466e-05,
      "step": 37739,
      "training_step_time": 6.77398419380188
    },
    {
      "epoch": 3.599166870117188e-05,
      "grad_norm": 0.11461282521486282,
      "learning_rate": 7.34490801934726e-06,
      "loss": 0.0637,
      "step": 37740
    },
    {
      "epoch": 3.599166870117188e-05,
      "model_forward_time": 1.7502522468566895,
      "step": 37740
    },
    {
      "dynamics_loss_avg": 0.049402137752622366,
      "epoch": 3.599166870117188e-05,
      "step": 37740
    },
    {
      "action_loss_avg": 0.001822371012531221,
      "epoch": 3.599166870117188e-05,
      "step": 37740
    },
    {
      "epoch": 3.599166870117188e-05,
      "step": 37740,
      "training_step_time": 6.744783401489258
    },
    {
      "epoch": 3.599262237548828e-05,
      "model_forward_time": 2.968492031097412,
      "step": 37741
    },
    {
      "epoch": 3.599262237548828e-05,
      "step": 37741,
      "training_step_time": 9.804491519927979
    },
    {
      "epoch": 3.5993576049804685e-05,
      "model_forward_time": 1.7561674118041992,
      "step": 37742
    },
    {
      "epoch": 3.5993576049804685e-05,
      "step": 37742,
      "training_step_time": 8.425700902938843
    },
    {
      "epoch": 3.5994529724121096e-05,
      "model_forward_time": 1.7478516101837158,
      "step": 37743
    },
    {
      "epoch": 3.5994529724121096e-05,
      "step": 37743,
      "training_step_time": 7.455640077590942
    },
    {
      "epoch": 3.59954833984375e-05,
      "model_forward_time": 1.7445573806762695,
      "step": 37744
    },
    {
      "epoch": 3.59954833984375e-05,
      "step": 37744,
      "training_step_time": 6.765531301498413
    },
    {
      "epoch": 3.5996437072753904e-05,
      "model_forward_time": 1.7580735683441162,
      "step": 37745
    },
    {
      "epoch": 3.5996437072753904e-05,
      "step": 37745,
      "training_step_time": 6.7778239250183105
    },
    {
      "epoch": 3.5997390747070315e-05,
      "model_forward_time": 1.7438468933105469,
      "step": 37746
    },
    {
      "epoch": 3.5997390747070315e-05,
      "step": 37746,
      "training_step_time": 6.826753616333008
    },
    {
      "epoch": 3.599834442138672e-05,
      "model_forward_time": 1.7277462482452393,
      "step": 37747
    },
    {
      "epoch": 3.599834442138672e-05,
      "step": 37747,
      "training_step_time": 6.770596027374268
    },
    {
      "epoch": 3.599929809570312e-05,
      "model_forward_time": 1.7509291172027588,
      "step": 37748
    },
    {
      "epoch": 3.599929809570312e-05,
      "step": 37748,
      "training_step_time": 6.718564510345459
    },
    {
      "epoch": 3.6000251770019533e-05,
      "model_forward_time": 1.7250642776489258,
      "step": 37749
    },
    {
      "epoch": 3.6000251770019533e-05,
      "step": 37749,
      "training_step_time": 6.765837669372559
    },
    {
      "epoch": 3.600120544433594e-05,
      "grad_norm": 0.20469948649406433,
      "learning_rate": 7.343447533493948e-06,
      "loss": 0.0605,
      "step": 37750
    },
    {
      "epoch": 3.600120544433594e-05,
      "model_forward_time": 2.971916675567627,
      "step": 37750
    },
    {
      "dynamics_loss_avg": 0.06265472061932087,
      "epoch": 3.600120544433594e-05,
      "step": 37750
    },
    {
      "action_loss_avg": 0.0018771488743368537,
      "epoch": 3.600120544433594e-05,
      "step": 37750
    },
    {
      "epoch": 3.600120544433594e-05,
      "step": 37750,
      "training_step_time": 8.803454875946045
    },
    {
      "epoch": 3.600215911865234e-05,
      "model_forward_time": 1.064283847808838,
      "step": 37751
    },
    {
      "epoch": 3.600215911865234e-05,
      "step": 37751,
      "training_step_time": 8.278948068618774
    },
    {
      "epoch": 3.600311279296875e-05,
      "model_forward_time": 1.74595046043396,
      "step": 37752
    },
    {
      "epoch": 3.600311279296875e-05,
      "step": 37752,
      "training_step_time": 7.686591863632202
    },
    {
      "epoch": 3.6004066467285156e-05,
      "model_forward_time": 1.7629284858703613,
      "step": 37753
    },
    {
      "epoch": 3.6004066467285156e-05,
      "step": 37753,
      "training_step_time": 6.828729152679443
    },
    {
      "epoch": 3.600502014160156e-05,
      "model_forward_time": 1.745330572128296,
      "step": 37754
    },
    {
      "epoch": 3.600502014160156e-05,
      "step": 37754,
      "training_step_time": 6.826671600341797
    },
    {
      "epoch": 3.600597381591797e-05,
      "model_forward_time": 1.75123929977417,
      "step": 37755
    },
    {
      "epoch": 3.600597381591797e-05,
      "step": 37755,
      "training_step_time": 6.793420791625977
    },
    {
      "epoch": 3.6006927490234375e-05,
      "model_forward_time": 1.7475361824035645,
      "step": 37756
    },
    {
      "epoch": 3.6006927490234375e-05,
      "step": 37756,
      "training_step_time": 6.7286810874938965
    },
    {
      "epoch": 3.600788116455078e-05,
      "model_forward_time": 1.765993356704712,
      "step": 37757
    },
    {
      "epoch": 3.600788116455078e-05,
      "step": 37757,
      "training_step_time": 6.778816223144531
    },
    {
      "epoch": 3.600883483886719e-05,
      "model_forward_time": 1.3870184421539307,
      "step": 37758
    },
    {
      "epoch": 3.600883483886719e-05,
      "step": 37758,
      "training_step_time": 6.847210168838501
    },
    {
      "epoch": 3.6009788513183594e-05,
      "model_forward_time": 1.7458033561706543,
      "step": 37759
    },
    {
      "epoch": 3.6009788513183594e-05,
      "step": 37759,
      "training_step_time": 9.306055784225464
    },
    {
      "epoch": 3.60107421875e-05,
      "grad_norm": 0.1434890180826187,
      "learning_rate": 7.341986791364739e-06,
      "loss": 0.0577,
      "step": 37760
    },
    {
      "epoch": 3.60107421875e-05,
      "model_forward_time": 2.88419246673584,
      "step": 37760
    },
    {
      "dynamics_loss_avg": 0.058242916595190763,
      "epoch": 3.60107421875e-05,
      "step": 37760
    },
    {
      "action_loss_avg": 0.0012675133359152824,
      "epoch": 3.60107421875e-05,
      "step": 37760
    },
    {
      "epoch": 3.60107421875e-05,
      "step": 37760,
      "training_step_time": 9.551608800888062
    },
    {
      "epoch": 3.601169586181641e-05,
      "model_forward_time": 1.0518889427185059,
      "step": 37761
    },
    {
      "epoch": 3.601169586181641e-05,
      "step": 37761,
      "training_step_time": 7.725860595703125
    },
    {
      "epoch": 3.601264953613281e-05,
      "model_forward_time": 1.7429871559143066,
      "step": 37762
    },
    {
      "epoch": 3.601264953613281e-05,
      "step": 37762,
      "training_step_time": 7.3706581592559814
    },
    {
      "epoch": 3.601360321044922e-05,
      "model_forward_time": 1.763984203338623,
      "step": 37763
    },
    {
      "epoch": 3.601360321044922e-05,
      "step": 37763,
      "training_step_time": 6.821754217147827
    },
    {
      "epoch": 3.601455688476563e-05,
      "model_forward_time": 1.057218313217163,
      "step": 37764
    },
    {
      "epoch": 3.601455688476563e-05,
      "step": 37764,
      "training_step_time": 6.813594579696655
    },
    {
      "epoch": 3.601551055908203e-05,
      "model_forward_time": 1.7506566047668457,
      "step": 37765
    },
    {
      "epoch": 3.601551055908203e-05,
      "step": 37765,
      "training_step_time": 6.8309102058410645
    },
    {
      "epoch": 3.6016464233398436e-05,
      "model_forward_time": 1.7529959678649902,
      "step": 37766
    },
    {
      "epoch": 3.6016464233398436e-05,
      "step": 37766,
      "training_step_time": 6.746681451797485
    },
    {
      "epoch": 3.601741790771485e-05,
      "model_forward_time": 1.7658255100250244,
      "step": 37767
    },
    {
      "epoch": 3.601741790771485e-05,
      "step": 37767,
      "training_step_time": 6.801333665847778
    },
    {
      "epoch": 3.601837158203125e-05,
      "model_forward_time": 1.7476706504821777,
      "step": 37768
    },
    {
      "epoch": 3.601837158203125e-05,
      "step": 37768,
      "training_step_time": 7.240842819213867
    },
    {
      "epoch": 3.6019325256347655e-05,
      "model_forward_time": 1.8113384246826172,
      "step": 37769
    },
    {
      "epoch": 3.6019325256347655e-05,
      "step": 37769,
      "training_step_time": 9.96339750289917
    },
    {
      "epoch": 3.6020278930664065e-05,
      "grad_norm": 0.08836305141448975,
      "learning_rate": 7.340525793119375e-06,
      "loss": 0.0586,
      "step": 37770
    },
    {
      "epoch": 3.6020278930664065e-05,
      "model_forward_time": 2.9596152305603027,
      "step": 37770
    },
    {
      "dynamics_loss_avg": 0.07383224256336689,
      "epoch": 3.6020278930664065e-05,
      "step": 37770
    },
    {
      "action_loss_avg": 0.0029448259621858595,
      "epoch": 3.6020278930664065e-05,
      "step": 37770
    },
    {
      "epoch": 3.6020278930664065e-05,
      "step": 37770,
      "training_step_time": 7.806390285491943
    },
    {
      "epoch": 3.602123260498047e-05,
      "model_forward_time": 1.768620491027832,
      "step": 37771
    },
    {
      "epoch": 3.602123260498047e-05,
      "step": 37771,
      "training_step_time": 7.649420261383057
    },
    {
      "epoch": 3.6022186279296874e-05,
      "model_forward_time": 1.7638027667999268,
      "step": 37772
    },
    {
      "epoch": 3.6022186279296874e-05,
      "step": 37772,
      "training_step_time": 6.815030813217163
    },
    {
      "epoch": 3.6023139953613284e-05,
      "model_forward_time": 1.7591407299041748,
      "step": 37773
    },
    {
      "epoch": 3.6023139953613284e-05,
      "step": 37773,
      "training_step_time": 6.7988879680633545
    },
    {
      "epoch": 3.602409362792969e-05,
      "model_forward_time": 1.7568981647491455,
      "step": 37774
    },
    {
      "epoch": 3.602409362792969e-05,
      "step": 37774,
      "training_step_time": 6.843499660491943
    },
    {
      "epoch": 3.602504730224609e-05,
      "model_forward_time": 1.7407474517822266,
      "step": 37775
    },
    {
      "epoch": 3.602504730224609e-05,
      "step": 37775,
      "training_step_time": 6.856662034988403
    },
    {
      "epoch": 3.60260009765625e-05,
      "model_forward_time": 1.7570044994354248,
      "step": 37776
    },
    {
      "epoch": 3.60260009765625e-05,
      "step": 37776,
      "training_step_time": 6.801254749298096
    },
    {
      "epoch": 3.602695465087891e-05,
      "model_forward_time": 1.7638797760009766,
      "step": 37777
    },
    {
      "epoch": 3.602695465087891e-05,
      "step": 37777,
      "training_step_time": 7.211190223693848
    },
    {
      "epoch": 3.602790832519531e-05,
      "model_forward_time": 1.7977750301361084,
      "step": 37778
    },
    {
      "epoch": 3.602790832519531e-05,
      "step": 37778,
      "training_step_time": 9.717084407806396
    },
    {
      "epoch": 3.602886199951172e-05,
      "model_forward_time": 2.9877710342407227,
      "step": 37779
    },
    {
      "epoch": 3.602886199951172e-05,
      "step": 37779,
      "training_step_time": 8.801762819290161
    },
    {
      "epoch": 3.6029815673828126e-05,
      "grad_norm": 0.1285858154296875,
      "learning_rate": 7.339064538917631e-06,
      "loss": 0.0573,
      "step": 37780
    },
    {
      "epoch": 3.6029815673828126e-05,
      "model_forward_time": 1.7604563236236572,
      "step": 37780
    },
    {
      "dynamics_loss_avg": 0.06681301780045032,
      "epoch": 3.6029815673828126e-05,
      "step": 37780
    },
    {
      "action_loss_avg": 0.001417661301093176,
      "epoch": 3.6029815673828126e-05,
      "step": 37780
    },
    {
      "epoch": 3.6029815673828126e-05,
      "step": 37780,
      "training_step_time": 7.6793622970581055
    },
    {
      "epoch": 3.603076934814453e-05,
      "model_forward_time": 1.734264850616455,
      "step": 37781
    },
    {
      "epoch": 3.603076934814453e-05,
      "step": 37781,
      "training_step_time": 7.288902521133423
    },
    {
      "epoch": 3.6031723022460934e-05,
      "model_forward_time": 1.7641615867614746,
      "step": 37782
    },
    {
      "epoch": 3.6031723022460934e-05,
      "step": 37782,
      "training_step_time": 6.829843759536743
    },
    {
      "epoch": 3.6032676696777345e-05,
      "model_forward_time": 1.7541680335998535,
      "step": 37783
    },
    {
      "epoch": 3.6032676696777345e-05,
      "step": 37783,
      "training_step_time": 6.804805517196655
    },
    {
      "epoch": 3.603363037109375e-05,
      "model_forward_time": 1.7524006366729736,
      "step": 37784
    },
    {
      "epoch": 3.603363037109375e-05,
      "step": 37784,
      "training_step_time": 6.817417621612549
    },
    {
      "epoch": 3.603458404541015e-05,
      "model_forward_time": 1.7492215633392334,
      "step": 37785
    },
    {
      "epoch": 3.603458404541015e-05,
      "step": 37785,
      "training_step_time": 6.81258749961853
    },
    {
      "epoch": 3.6035537719726564e-05,
      "model_forward_time": 1.7591557502746582,
      "step": 37786
    },
    {
      "epoch": 3.6035537719726564e-05,
      "step": 37786,
      "training_step_time": 7.243160963058472
    },
    {
      "epoch": 3.603649139404297e-05,
      "model_forward_time": 1.779308795928955,
      "step": 37787
    },
    {
      "epoch": 3.603649139404297e-05,
      "step": 37787,
      "training_step_time": 8.532465696334839
    },
    {
      "epoch": 3.603744506835937e-05,
      "model_forward_time": 2.931807279586792,
      "step": 37788
    },
    {
      "epoch": 3.603744506835937e-05,
      "step": 37788,
      "training_step_time": 8.397384881973267
    },
    {
      "epoch": 3.603839874267578e-05,
      "model_forward_time": 1.766648769378662,
      "step": 37789
    },
    {
      "epoch": 3.603839874267578e-05,
      "step": 37789,
      "training_step_time": 7.676942586898804
    },
    {
      "epoch": 3.603935241699219e-05,
      "grad_norm": 0.11866889148950577,
      "learning_rate": 7.337603028919302e-06,
      "loss": 0.0579,
      "step": 37790
    },
    {
      "epoch": 3.603935241699219e-05,
      "model_forward_time": 1.750535249710083,
      "step": 37790
    },
    {
      "dynamics_loss_avg": 0.05623219553381205,
      "epoch": 3.603935241699219e-05,
      "step": 37790
    },
    {
      "action_loss_avg": 0.002448248479049653,
      "epoch": 3.603935241699219e-05,
      "step": 37790
    },
    {
      "epoch": 3.603935241699219e-05,
      "step": 37790,
      "training_step_time": 7.330544471740723
    },
    {
      "epoch": 3.604030609130859e-05,
      "model_forward_time": 1.7630879878997803,
      "step": 37791
    },
    {
      "epoch": 3.604030609130859e-05,
      "step": 37791,
      "training_step_time": 6.9108569622039795
    },
    {
      "epoch": 3.6041259765625e-05,
      "model_forward_time": 1.7599778175354004,
      "step": 37792
    },
    {
      "epoch": 3.6041259765625e-05,
      "step": 37792,
      "training_step_time": 6.900949239730835
    },
    {
      "epoch": 3.6042213439941406e-05,
      "model_forward_time": 1.7704455852508545,
      "step": 37793
    },
    {
      "epoch": 3.6042213439941406e-05,
      "step": 37793,
      "training_step_time": 6.923841714859009
    },
    {
      "epoch": 3.604316711425781e-05,
      "model_forward_time": 1.7651920318603516,
      "step": 37794
    },
    {
      "epoch": 3.604316711425781e-05,
      "step": 37794,
      "training_step_time": 6.859530925750732
    },
    {
      "epoch": 3.604412078857422e-05,
      "model_forward_time": 1.757964849472046,
      "step": 37795
    },
    {
      "epoch": 3.604412078857422e-05,
      "step": 37795,
      "training_step_time": 7.251376628875732
    },
    {
      "epoch": 3.6045074462890624e-05,
      "model_forward_time": 1.754845142364502,
      "step": 37796
    },
    {
      "epoch": 3.6045074462890624e-05,
      "step": 37796,
      "training_step_time": 9.297817945480347
    },
    {
      "epoch": 3.604602813720703e-05,
      "model_forward_time": 2.9520280361175537,
      "step": 37797
    },
    {
      "epoch": 3.604602813720703e-05,
      "step": 37797,
      "training_step_time": 10.215266227722168
    },
    {
      "epoch": 3.604698181152344e-05,
      "model_forward_time": 1.7598583698272705,
      "step": 37798
    },
    {
      "epoch": 3.604698181152344e-05,
      "step": 37798,
      "training_step_time": 7.706872463226318
    },
    {
      "epoch": 3.604793548583984e-05,
      "model_forward_time": 1.7629270553588867,
      "step": 37799
    },
    {
      "epoch": 3.604793548583984e-05,
      "step": 37799,
      "training_step_time": 7.345881938934326
    },
    {
      "epoch": 3.604888916015625e-05,
      "grad_norm": 0.10021141916513443,
      "learning_rate": 7.336141263284225e-06,
      "loss": 0.0595,
      "step": 37800
    },
    {
      "epoch": 3.604888916015625e-05,
      "model_forward_time": 1.7385749816894531,
      "step": 37800
    },
    {
      "dynamics_loss_avg": 0.054148513823747635,
      "epoch": 3.604888916015625e-05,
      "step": 37800
    },
    {
      "action_loss_avg": 0.0018659194756764919,
      "epoch": 3.604888916015625e-05,
      "step": 37800
    },
    {
      "epoch": 3.604888916015625e-05,
      "step": 37800,
      "training_step_time": 6.820873260498047
    },
    {
      "epoch": 3.604984283447266e-05,
      "model_forward_time": 1.7621467113494873,
      "step": 37801
    },
    {
      "epoch": 3.604984283447266e-05,
      "step": 37801,
      "training_step_time": 6.796268701553345
    },
    {
      "epoch": 3.605079650878906e-05,
      "model_forward_time": 1.0367872714996338,
      "step": 37802
    },
    {
      "epoch": 3.605079650878906e-05,
      "step": 37802,
      "training_step_time": 6.8078906536102295
    },
    {
      "epoch": 3.6051750183105466e-05,
      "model_forward_time": 1.7577717304229736,
      "step": 37803
    },
    {
      "epoch": 3.6051750183105466e-05,
      "step": 37803,
      "training_step_time": 6.831885099411011
    },
    {
      "epoch": 3.605270385742188e-05,
      "model_forward_time": 1.0726964473724365,
      "step": 37804
    },
    {
      "epoch": 3.605270385742188e-05,
      "step": 37804,
      "training_step_time": 7.209364891052246
    },
    {
      "epoch": 3.605365753173828e-05,
      "model_forward_time": 1.760378360748291,
      "step": 37805
    },
    {
      "epoch": 3.605365753173828e-05,
      "step": 37805,
      "training_step_time": 7.58342432975769
    },
    {
      "epoch": 3.6054611206054685e-05,
      "model_forward_time": 1.8676111698150635,
      "step": 37806
    },
    {
      "epoch": 3.6054611206054685e-05,
      "step": 37806,
      "training_step_time": 10.418747186660767
    },
    {
      "epoch": 3.6055564880371096e-05,
      "model_forward_time": 2.9869701862335205,
      "step": 37807
    },
    {
      "epoch": 3.6055564880371096e-05,
      "step": 37807,
      "training_step_time": 8.832544088363647
    },
    {
      "epoch": 3.60565185546875e-05,
      "model_forward_time": 1.7553126811981201,
      "step": 37808
    },
    {
      "epoch": 3.60565185546875e-05,
      "step": 37808,
      "training_step_time": 7.700820446014404
    },
    {
      "epoch": 3.6057472229003904e-05,
      "model_forward_time": 1.763315200805664,
      "step": 37809
    },
    {
      "epoch": 3.6057472229003904e-05,
      "step": 37809,
      "training_step_time": 7.347597599029541
    },
    {
      "epoch": 3.6058425903320315e-05,
      "grad_norm": 0.12413037568330765,
      "learning_rate": 7.334679242172249e-06,
      "loss": 0.0605,
      "step": 37810
    },
    {
      "epoch": 3.6058425903320315e-05,
      "model_forward_time": 0.7326138019561768,
      "step": 37810
    },
    {
      "dynamics_loss_avg": 0.059566653519868853,
      "epoch": 3.6058425903320315e-05,
      "step": 37810
    },
    {
      "action_loss_avg": 0.0013376006128964946,
      "epoch": 3.6058425903320315e-05,
      "step": 37810
    },
    {
      "epoch": 3.6058425903320315e-05,
      "step": 37810,
      "training_step_time": 6.885487079620361
    },
    {
      "epoch": 3.605937957763672e-05,
      "model_forward_time": 1.7612433433532715,
      "step": 37811
    },
    {
      "epoch": 3.605937957763672e-05,
      "step": 37811,
      "training_step_time": 6.855389833450317
    },
    {
      "epoch": 3.606033325195312e-05,
      "model_forward_time": 1.748812198638916,
      "step": 37812
    },
    {
      "epoch": 3.606033325195312e-05,
      "step": 37812,
      "training_step_time": 6.8463146686553955
    },
    {
      "epoch": 3.6061286926269534e-05,
      "model_forward_time": 1.7369258403778076,
      "step": 37813
    },
    {
      "epoch": 3.6061286926269534e-05,
      "step": 37813,
      "training_step_time": 7.238572120666504
    },
    {
      "epoch": 3.606224060058594e-05,
      "model_forward_time": 1.3880488872528076,
      "step": 37814
    },
    {
      "epoch": 3.606224060058594e-05,
      "step": 37814,
      "training_step_time": 7.567369699478149
    },
    {
      "epoch": 3.606319427490234e-05,
      "model_forward_time": 1.7930564880371094,
      "step": 37815
    },
    {
      "epoch": 3.606319427490234e-05,
      "step": 37815,
      "training_step_time": 8.365151643753052
    },
    {
      "epoch": 3.606414794921875e-05,
      "model_forward_time": 1.7526817321777344,
      "step": 37816
    },
    {
      "epoch": 3.606414794921875e-05,
      "step": 37816,
      "training_step_time": 9.354006290435791
    },
    {
      "epoch": 3.6065101623535156e-05,
      "model_forward_time": 2.9288649559020996,
      "step": 37817
    },
    {
      "epoch": 3.6065101623535156e-05,
      "step": 37817,
      "training_step_time": 7.751325845718384
    },
    {
      "epoch": 3.606605529785156e-05,
      "model_forward_time": 1.7622790336608887,
      "step": 37818
    },
    {
      "epoch": 3.606605529785156e-05,
      "step": 37818,
      "training_step_time": 7.305467844009399
    },
    {
      "epoch": 3.606700897216797e-05,
      "model_forward_time": 1.7531955242156982,
      "step": 37819
    },
    {
      "epoch": 3.606700897216797e-05,
      "step": 37819,
      "training_step_time": 6.7798192501068115
    },
    {
      "epoch": 3.6067962646484375e-05,
      "grad_norm": 0.14809881150722504,
      "learning_rate": 7.33321696574326e-06,
      "loss": 0.0598,
      "step": 37820
    },
    {
      "epoch": 3.6067962646484375e-05,
      "model_forward_time": 1.7364327907562256,
      "step": 37820
    },
    {
      "dynamics_loss_avg": 0.05085794720798731,
      "epoch": 3.6067962646484375e-05,
      "step": 37820
    },
    {
      "action_loss_avg": 0.00144706642604433,
      "epoch": 3.6067962646484375e-05,
      "step": 37820
    },
    {
      "epoch": 3.6067962646484375e-05,
      "step": 37820,
      "training_step_time": 6.818292140960693
    },
    {
      "epoch": 3.606891632080078e-05,
      "model_forward_time": 1.7482564449310303,
      "step": 37821
    },
    {
      "epoch": 3.606891632080078e-05,
      "step": 37821,
      "training_step_time": 6.836797714233398
    },
    {
      "epoch": 3.606986999511719e-05,
      "model_forward_time": 1.7563364505767822,
      "step": 37822
    },
    {
      "epoch": 3.606986999511719e-05,
      "step": 37822,
      "training_step_time": 6.811227798461914
    },
    {
      "epoch": 3.6070823669433594e-05,
      "model_forward_time": 1.7550866603851318,
      "step": 37823
    },
    {
      "epoch": 3.6070823669433594e-05,
      "step": 37823,
      "training_step_time": 7.549455642700195
    },
    {
      "epoch": 3.607177734375e-05,
      "model_forward_time": 1.7631864547729492,
      "step": 37824
    },
    {
      "epoch": 3.607177734375e-05,
      "step": 37824,
      "training_step_time": 10.573734045028687
    },
    {
      "epoch": 3.607273101806641e-05,
      "model_forward_time": 1.7647805213928223,
      "step": 37825
    },
    {
      "epoch": 3.607273101806641e-05,
      "step": 37825,
      "training_step_time": 9.609066009521484
    },
    {
      "epoch": 3.607368469238281e-05,
      "model_forward_time": 2.9626564979553223,
      "step": 37826
    },
    {
      "epoch": 3.607368469238281e-05,
      "step": 37826,
      "training_step_time": 7.82062554359436
    },
    {
      "epoch": 3.607463836669922e-05,
      "model_forward_time": 1.7700271606445312,
      "step": 37827
    },
    {
      "epoch": 3.607463836669922e-05,
      "step": 37827,
      "training_step_time": 7.32883882522583
    },
    {
      "epoch": 3.607559204101563e-05,
      "model_forward_time": 1.749030351638794,
      "step": 37828
    },
    {
      "epoch": 3.607559204101563e-05,
      "step": 37828,
      "training_step_time": 6.737559795379639
    },
    {
      "epoch": 3.607654571533203e-05,
      "model_forward_time": 1.742546558380127,
      "step": 37829
    },
    {
      "epoch": 3.607654571533203e-05,
      "step": 37829,
      "training_step_time": 6.81043004989624
    },
    {
      "epoch": 3.6077499389648436e-05,
      "grad_norm": 0.17003194987773895,
      "learning_rate": 7.331754434157174e-06,
      "loss": 0.0612,
      "step": 37830
    },
    {
      "epoch": 3.6077499389648436e-05,
      "model_forward_time": 1.7267355918884277,
      "step": 37830
    },
    {
      "dynamics_loss_avg": 0.040597844030708076,
      "epoch": 3.6077499389648436e-05,
      "step": 37830
    },
    {
      "action_loss_avg": 0.0016800559998955578,
      "epoch": 3.6077499389648436e-05,
      "step": 37830
    },
    {
      "epoch": 3.6077499389648436e-05,
      "step": 37830,
      "training_step_time": 6.793269634246826
    },
    {
      "epoch": 3.607845306396485e-05,
      "model_forward_time": 1.7560780048370361,
      "step": 37831
    },
    {
      "epoch": 3.607845306396485e-05,
      "step": 37831,
      "training_step_time": 6.794311046600342
    },
    {
      "epoch": 3.607940673828125e-05,
      "model_forward_time": 1.0552070140838623,
      "step": 37832
    },
    {
      "epoch": 3.607940673828125e-05,
      "step": 37832,
      "training_step_time": 7.296898365020752
    },
    {
      "epoch": 3.6080360412597655e-05,
      "model_forward_time": 1.7597908973693848,
      "step": 37833
    },
    {
      "epoch": 3.6080360412597655e-05,
      "step": 37833,
      "training_step_time": 9.497506856918335
    },
    {
      "epoch": 3.6081314086914066e-05,
      "model_forward_time": 1.8700697422027588,
      "step": 37834
    },
    {
      "epoch": 3.6081314086914066e-05,
      "step": 37834,
      "training_step_time": 9.867923259735107
    },
    {
      "epoch": 3.608226776123047e-05,
      "model_forward_time": 2.889169692993164,
      "step": 37835
    },
    {
      "epoch": 3.608226776123047e-05,
      "step": 37835,
      "training_step_time": 7.745421886444092
    },
    {
      "epoch": 3.6083221435546874e-05,
      "model_forward_time": 1.7638401985168457,
      "step": 37836
    },
    {
      "epoch": 3.6083221435546874e-05,
      "step": 37836,
      "training_step_time": 7.674447536468506
    },
    {
      "epoch": 3.6084175109863284e-05,
      "model_forward_time": 1.7683708667755127,
      "step": 37837
    },
    {
      "epoch": 3.6084175109863284e-05,
      "step": 37837,
      "training_step_time": 6.819215536117554
    },
    {
      "epoch": 3.608512878417969e-05,
      "model_forward_time": 1.7618770599365234,
      "step": 37838
    },
    {
      "epoch": 3.608512878417969e-05,
      "step": 37838,
      "training_step_time": 6.810441493988037
    },
    {
      "epoch": 3.608608245849609e-05,
      "model_forward_time": 1.7552971839904785,
      "step": 37839
    },
    {
      "epoch": 3.608608245849609e-05,
      "step": 37839,
      "training_step_time": 6.780502796173096
    },
    {
      "epoch": 3.60870361328125e-05,
      "grad_norm": 0.13871939480304718,
      "learning_rate": 7.330291647573928e-06,
      "loss": 0.0642,
      "step": 37840
    },
    {
      "epoch": 3.60870361328125e-05,
      "model_forward_time": 1.7392337322235107,
      "step": 37840
    },
    {
      "dynamics_loss_avg": 0.06088586803525686,
      "epoch": 3.60870361328125e-05,
      "step": 37840
    },
    {
      "action_loss_avg": 0.0025579437147825955,
      "epoch": 3.60870361328125e-05,
      "step": 37840
    },
    {
      "epoch": 3.60870361328125e-05,
      "step": 37840,
      "training_step_time": 6.791784286499023
    },
    {
      "epoch": 3.608798980712891e-05,
      "model_forward_time": 1.7485957145690918,
      "step": 37841
    },
    {
      "epoch": 3.608798980712891e-05,
      "step": 37841,
      "training_step_time": 7.30346941947937
    },
    {
      "epoch": 3.608894348144531e-05,
      "model_forward_time": 1.7712697982788086,
      "step": 37842
    },
    {
      "epoch": 3.608894348144531e-05,
      "step": 37842,
      "training_step_time": 9.50106430053711
    },
    {
      "epoch": 3.608989715576172e-05,
      "model_forward_time": 1.8511407375335693,
      "step": 37843
    },
    {
      "epoch": 3.608989715576172e-05,
      "step": 37843,
      "training_step_time": 10.456789016723633
    },
    {
      "epoch": 3.6090850830078126e-05,
      "model_forward_time": 2.834019422531128,
      "step": 37844
    },
    {
      "epoch": 3.6090850830078126e-05,
      "step": 37844,
      "training_step_time": 7.625996351242065
    },
    {
      "epoch": 3.609180450439453e-05,
      "model_forward_time": 1.7608616352081299,
      "step": 37845
    },
    {
      "epoch": 3.609180450439453e-05,
      "step": 37845,
      "training_step_time": 7.6356940269470215
    },
    {
      "epoch": 3.6092758178710934e-05,
      "model_forward_time": 1.7706043720245361,
      "step": 37846
    },
    {
      "epoch": 3.6092758178710934e-05,
      "step": 37846,
      "training_step_time": 6.767560720443726
    },
    {
      "epoch": 3.6093711853027345e-05,
      "model_forward_time": 1.7595164775848389,
      "step": 37847
    },
    {
      "epoch": 3.6093711853027345e-05,
      "step": 37847,
      "training_step_time": 6.795960426330566
    },
    {
      "epoch": 3.609466552734375e-05,
      "model_forward_time": 1.761847734451294,
      "step": 37848
    },
    {
      "epoch": 3.609466552734375e-05,
      "step": 37848,
      "training_step_time": 6.782547473907471
    },
    {
      "epoch": 3.609561920166015e-05,
      "model_forward_time": 1.740511417388916,
      "step": 37849
    },
    {
      "epoch": 3.609561920166015e-05,
      "step": 37849,
      "training_step_time": 6.777843236923218
    },
    {
      "epoch": 3.6096572875976564e-05,
      "grad_norm": 0.14886312186717987,
      "learning_rate": 7.328828606153491e-06,
      "loss": 0.0605,
      "step": 37850
    },
    {
      "epoch": 3.6096572875976564e-05,
      "model_forward_time": 1.7402279376983643,
      "step": 37850
    },
    {
      "dynamics_loss_avg": 0.07369286045432091,
      "epoch": 3.6096572875976564e-05,
      "step": 37850
    },
    {
      "action_loss_avg": 0.005015680985525251,
      "epoch": 3.6096572875976564e-05,
      "step": 37850
    },
    {
      "epoch": 3.6096572875976564e-05,
      "step": 37850,
      "training_step_time": 7.281705379486084
    },
    {
      "epoch": 3.609752655029297e-05,
      "model_forward_time": 1.7616207599639893,
      "step": 37851
    },
    {
      "epoch": 3.609752655029297e-05,
      "step": 37851,
      "training_step_time": 8.194912433624268
    },
    {
      "epoch": 3.609848022460937e-05,
      "model_forward_time": 1.8303859233856201,
      "step": 37852
    },
    {
      "epoch": 3.609848022460937e-05,
      "step": 37852,
      "training_step_time": 9.814149379730225
    },
    {
      "epoch": 3.609943389892578e-05,
      "model_forward_time": 1.7636911869049072,
      "step": 37853
    },
    {
      "epoch": 3.609943389892578e-05,
      "step": 37853,
      "training_step_time": 9.248355865478516
    },
    {
      "epoch": 3.610038757324219e-05,
      "model_forward_time": 2.969822406768799,
      "step": 37854
    },
    {
      "epoch": 3.610038757324219e-05,
      "step": 37854,
      "training_step_time": 7.771264553070068
    },
    {
      "epoch": 3.610134124755859e-05,
      "model_forward_time": 1.7724127769470215,
      "step": 37855
    },
    {
      "epoch": 3.610134124755859e-05,
      "step": 37855,
      "training_step_time": 7.286965370178223
    },
    {
      "epoch": 3.6102294921875e-05,
      "model_forward_time": 1.7280726432800293,
      "step": 37856
    },
    {
      "epoch": 3.6102294921875e-05,
      "step": 37856,
      "training_step_time": 6.7363152503967285
    },
    {
      "epoch": 3.6103248596191406e-05,
      "model_forward_time": 1.752460241317749,
      "step": 37857
    },
    {
      "epoch": 3.6103248596191406e-05,
      "step": 37857,
      "training_step_time": 6.804854869842529
    },
    {
      "epoch": 3.610420227050781e-05,
      "model_forward_time": 1.7601497173309326,
      "step": 37858
    },
    {
      "epoch": 3.610420227050781e-05,
      "step": 37858,
      "training_step_time": 6.811180591583252
    },
    {
      "epoch": 3.610515594482422e-05,
      "model_forward_time": 1.7477164268493652,
      "step": 37859
    },
    {
      "epoch": 3.610515594482422e-05,
      "step": 37859,
      "training_step_time": 7.917535781860352
    },
    {
      "epoch": 3.6106109619140624e-05,
      "grad_norm": 0.16125808656215668,
      "learning_rate": 7.3273653100558585e-06,
      "loss": 0.0608,
      "step": 37860
    },
    {
      "epoch": 3.6106109619140624e-05,
      "model_forward_time": 1.7505369186401367,
      "step": 37860
    },
    {
      "dynamics_loss_avg": 0.05832052063196898,
      "epoch": 3.6106109619140624e-05,
      "step": 37860
    },
    {
      "action_loss_avg": 0.0037705016235122456,
      "epoch": 3.6106109619140624e-05,
      "step": 37860
    },
    {
      "epoch": 3.6106109619140624e-05,
      "step": 37860,
      "training_step_time": 8.617645740509033
    },
    {
      "epoch": 3.610706329345703e-05,
      "model_forward_time": 1.401257038116455,
      "step": 37861
    },
    {
      "epoch": 3.610706329345703e-05,
      "step": 37861,
      "training_step_time": 9.173209428787231
    },
    {
      "epoch": 3.610801696777344e-05,
      "model_forward_time": 1.7552993297576904,
      "step": 37862
    },
    {
      "epoch": 3.610801696777344e-05,
      "step": 37862,
      "training_step_time": 8.600076913833618
    },
    {
      "epoch": 3.610897064208984e-05,
      "model_forward_time": 2.916626214981079,
      "step": 37863
    },
    {
      "epoch": 3.610897064208984e-05,
      "step": 37863,
      "training_step_time": 7.7627272605896
    },
    {
      "epoch": 3.610992431640625e-05,
      "model_forward_time": 1.7545421123504639,
      "step": 37864
    },
    {
      "epoch": 3.610992431640625e-05,
      "step": 37864,
      "training_step_time": 7.364954233169556
    },
    {
      "epoch": 3.611087799072266e-05,
      "model_forward_time": 1.7362546920776367,
      "step": 37865
    },
    {
      "epoch": 3.611087799072266e-05,
      "step": 37865,
      "training_step_time": 6.790364980697632
    },
    {
      "epoch": 3.611183166503906e-05,
      "model_forward_time": 1.7546238899230957,
      "step": 37866
    },
    {
      "epoch": 3.611183166503906e-05,
      "step": 37866,
      "training_step_time": 6.781095504760742
    },
    {
      "epoch": 3.6112785339355466e-05,
      "model_forward_time": 1.7540245056152344,
      "step": 37867
    },
    {
      "epoch": 3.6112785339355466e-05,
      "step": 37867,
      "training_step_time": 6.824381351470947
    },
    {
      "epoch": 3.611373901367188e-05,
      "model_forward_time": 1.754408836364746,
      "step": 37868
    },
    {
      "epoch": 3.611373901367188e-05,
      "step": 37868,
      "training_step_time": 7.659013509750366
    },
    {
      "epoch": 3.611469268798828e-05,
      "model_forward_time": 1.7624330520629883,
      "step": 37869
    },
    {
      "epoch": 3.611469268798828e-05,
      "step": 37869,
      "training_step_time": 7.620739698410034
    },
    {
      "epoch": 3.6115646362304685e-05,
      "grad_norm": 0.26991739869117737,
      "learning_rate": 7.325901759441056e-06,
      "loss": 0.0583,
      "step": 37870
    },
    {
      "epoch": 3.6115646362304685e-05,
      "model_forward_time": 1.7732574939727783,
      "step": 37870
    },
    {
      "dynamics_loss_avg": 0.0362511339597404,
      "epoch": 3.6115646362304685e-05,
      "step": 37870
    },
    {
      "action_loss_avg": 0.0015100660210009664,
      "epoch": 3.6115646362304685e-05,
      "step": 37870
    },
    {
      "epoch": 3.6115646362304685e-05,
      "step": 37870,
      "training_step_time": 7.767767667770386
    },
    {
      "epoch": 3.6116600036621096e-05,
      "model_forward_time": 1.7614283561706543,
      "step": 37871
    },
    {
      "epoch": 3.6116600036621096e-05,
      "step": 37871,
      "training_step_time": 7.707821369171143
    },
    {
      "epoch": 3.61175537109375e-05,
      "model_forward_time": 2.912278413772583,
      "step": 37872
    },
    {
      "epoch": 3.61175537109375e-05,
      "step": 37872,
      "training_step_time": 7.735828638076782
    },
    {
      "epoch": 3.6118507385253904e-05,
      "model_forward_time": 1.7545604705810547,
      "step": 37873
    },
    {
      "epoch": 3.6118507385253904e-05,
      "step": 37873,
      "training_step_time": 7.63417911529541
    },
    {
      "epoch": 3.6119461059570315e-05,
      "model_forward_time": 1.7597384452819824,
      "step": 37874
    },
    {
      "epoch": 3.6119461059570315e-05,
      "step": 37874,
      "training_step_time": 6.79339075088501
    },
    {
      "epoch": 3.612041473388672e-05,
      "model_forward_time": 1.742520809173584,
      "step": 37875
    },
    {
      "epoch": 3.612041473388672e-05,
      "step": 37875,
      "training_step_time": 6.772133827209473
    },
    {
      "epoch": 3.612136840820312e-05,
      "model_forward_time": 1.769014596939087,
      "step": 37876
    },
    {
      "epoch": 3.612136840820312e-05,
      "step": 37876,
      "training_step_time": 6.791563034057617
    },
    {
      "epoch": 3.6122322082519534e-05,
      "model_forward_time": 1.0638413429260254,
      "step": 37877
    },
    {
      "epoch": 3.6122322082519534e-05,
      "step": 37877,
      "training_step_time": 7.144117832183838
    },
    {
      "epoch": 3.612327575683594e-05,
      "model_forward_time": 1.751028060913086,
      "step": 37878
    },
    {
      "epoch": 3.612327575683594e-05,
      "step": 37878,
      "training_step_time": 7.653887510299683
    },
    {
      "epoch": 3.612422943115234e-05,
      "model_forward_time": 1.753807544708252,
      "step": 37879
    },
    {
      "epoch": 3.612422943115234e-05,
      "step": 37879,
      "training_step_time": 9.960629224777222
    },
    {
      "epoch": 3.612518310546875e-05,
      "grad_norm": 0.1519501656293869,
      "learning_rate": 7.3244379544691325e-06,
      "loss": 0.059,
      "step": 37880
    },
    {
      "epoch": 3.612518310546875e-05,
      "model_forward_time": 1.8834104537963867,
      "step": 37880
    },
    {
      "dynamics_loss_avg": 0.053808934055268764,
      "epoch": 3.612518310546875e-05,
      "step": 37880
    },
    {
      "action_loss_avg": 0.0022740544169209897,
      "epoch": 3.612518310546875e-05,
      "step": 37880
    },
    {
      "epoch": 3.612518310546875e-05,
      "step": 37880,
      "training_step_time": 8.633976221084595
    },
    {
      "epoch": 3.6126136779785156e-05,
      "model_forward_time": 1.7450695037841797,
      "step": 37881
    },
    {
      "epoch": 3.6126136779785156e-05,
      "step": 37881,
      "training_step_time": 7.6530845165252686
    },
    {
      "epoch": 3.612709045410156e-05,
      "model_forward_time": 3.0127360820770264,
      "step": 37882
    },
    {
      "epoch": 3.612709045410156e-05,
      "step": 37882,
      "training_step_time": 7.800532579421997
    },
    {
      "epoch": 3.612804412841797e-05,
      "model_forward_time": 1.7548325061798096,
      "step": 37883
    },
    {
      "epoch": 3.612804412841797e-05,
      "step": 37883,
      "training_step_time": 7.269124984741211
    },
    {
      "epoch": 3.6128997802734375e-05,
      "model_forward_time": 1.7499580383300781,
      "step": 37884
    },
    {
      "epoch": 3.6128997802734375e-05,
      "step": 37884,
      "training_step_time": 6.775103807449341
    },
    {
      "epoch": 3.612995147705078e-05,
      "model_forward_time": 1.7504706382751465,
      "step": 37885
    },
    {
      "epoch": 3.612995147705078e-05,
      "step": 37885,
      "training_step_time": 6.799946546554565
    },
    {
      "epoch": 3.613090515136719e-05,
      "model_forward_time": 1.7569117546081543,
      "step": 37886
    },
    {
      "epoch": 3.613090515136719e-05,
      "step": 37886,
      "training_step_time": 7.137867450714111
    },
    {
      "epoch": 3.6131858825683594e-05,
      "model_forward_time": 1.7529089450836182,
      "step": 37887
    },
    {
      "epoch": 3.6131858825683594e-05,
      "step": 37887,
      "training_step_time": 8.149747133255005
    },
    {
      "epoch": 3.61328125e-05,
      "model_forward_time": 1.7546005249023438,
      "step": 37888
    },
    {
      "epoch": 3.61328125e-05,
      "step": 37888,
      "training_step_time": 9.080822229385376
    },
    {
      "epoch": 3.613376617431641e-05,
      "model_forward_time": 1.8504858016967773,
      "step": 37889
    },
    {
      "epoch": 3.613376617431641e-05,
      "step": 37889,
      "training_step_time": 8.596896409988403
    },
    {
      "epoch": 3.613471984863281e-05,
      "grad_norm": 0.16199032962322235,
      "learning_rate": 7.32297389530017e-06,
      "loss": 0.0594,
      "step": 37890
    },
    {
      "epoch": 3.613471984863281e-05,
      "model_forward_time": 1.7456731796264648,
      "step": 37890
    },
    {
      "dynamics_loss_avg": 0.05726368185132742,
      "epoch": 3.613471984863281e-05,
      "step": 37890
    },
    {
      "action_loss_avg": 0.00211018908303231,
      "epoch": 3.613471984863281e-05,
      "step": 37890
    },
    {
      "epoch": 3.613471984863281e-05,
      "step": 37890,
      "training_step_time": 7.824223041534424
    },
    {
      "epoch": 3.613567352294922e-05,
      "model_forward_time": 2.920224666595459,
      "step": 37891
    },
    {
      "epoch": 3.613567352294922e-05,
      "step": 37891,
      "training_step_time": 7.83193039894104
    },
    {
      "epoch": 3.613662719726563e-05,
      "model_forward_time": 1.751819133758545,
      "step": 37892
    },
    {
      "epoch": 3.613662719726563e-05,
      "step": 37892,
      "training_step_time": 7.282452821731567
    },
    {
      "epoch": 3.613758087158203e-05,
      "model_forward_time": 1.7560157775878906,
      "step": 37893
    },
    {
      "epoch": 3.613758087158203e-05,
      "step": 37893,
      "training_step_time": 6.848361015319824
    },
    {
      "epoch": 3.6138534545898436e-05,
      "model_forward_time": 1.395329236984253,
      "step": 37894
    },
    {
      "epoch": 3.6138534545898436e-05,
      "step": 37894,
      "training_step_time": 6.884443044662476
    },
    {
      "epoch": 3.613948822021485e-05,
      "model_forward_time": 1.7640929222106934,
      "step": 37895
    },
    {
      "epoch": 3.613948822021485e-05,
      "step": 37895,
      "training_step_time": 7.227683067321777
    },
    {
      "epoch": 3.614044189453125e-05,
      "model_forward_time": 1.7500832080841064,
      "step": 37896
    },
    {
      "epoch": 3.614044189453125e-05,
      "step": 37896,
      "training_step_time": 7.620800256729126
    },
    {
      "epoch": 3.6141395568847655e-05,
      "model_forward_time": 1.7522368431091309,
      "step": 37897
    },
    {
      "epoch": 3.6141395568847655e-05,
      "step": 37897,
      "training_step_time": 8.819695711135864
    },
    {
      "epoch": 3.6142349243164066e-05,
      "model_forward_time": 1.8054251670837402,
      "step": 37898
    },
    {
      "epoch": 3.6142349243164066e-05,
      "step": 37898,
      "training_step_time": 8.55019736289978
    },
    {
      "epoch": 3.614330291748047e-05,
      "model_forward_time": 1.7512283325195312,
      "step": 37899
    },
    {
      "epoch": 3.614330291748047e-05,
      "step": 37899,
      "training_step_time": 9.71360158920288
    },
    {
      "epoch": 3.6144256591796874e-05,
      "grad_norm": 0.1996423453092575,
      "learning_rate": 7.321509582094275e-06,
      "loss": 0.0602,
      "step": 37900
    },
    {
      "epoch": 3.6144256591796874e-05,
      "model_forward_time": 2.8840041160583496,
      "step": 37900
    },
    {
      "dynamics_loss_avg": 0.06659475518390537,
      "epoch": 3.6144256591796874e-05,
      "step": 37900
    },
    {
      "action_loss_avg": 0.0028710573926218784,
      "epoch": 3.6144256591796874e-05,
      "step": 37900
    },
    {
      "epoch": 3.6144256591796874e-05,
      "step": 37900,
      "training_step_time": 7.840352296829224
    },
    {
      "epoch": 3.6145210266113284e-05,
      "model_forward_time": 1.753938913345337,
      "step": 37901
    },
    {
      "epoch": 3.6145210266113284e-05,
      "step": 37901,
      "training_step_time": 7.45538330078125
    },
    {
      "epoch": 3.614616394042969e-05,
      "model_forward_time": 1.7610528469085693,
      "step": 37902
    },
    {
      "epoch": 3.614616394042969e-05,
      "step": 37902,
      "training_step_time": 6.947033882141113
    },
    {
      "epoch": 3.614711761474609e-05,
      "model_forward_time": 1.7632226943969727,
      "step": 37903
    },
    {
      "epoch": 3.614711761474609e-05,
      "step": 37903,
      "training_step_time": 6.932811737060547
    },
    {
      "epoch": 3.61480712890625e-05,
      "model_forward_time": 1.7527720928192139,
      "step": 37904
    },
    {
      "epoch": 3.61480712890625e-05,
      "step": 37904,
      "training_step_time": 6.885956525802612
    },
    {
      "epoch": 3.614902496337891e-05,
      "model_forward_time": 1.7604496479034424,
      "step": 37905
    },
    {
      "epoch": 3.614902496337891e-05,
      "step": 37905,
      "training_step_time": 7.5214502811431885
    },
    {
      "epoch": 3.614997863769531e-05,
      "model_forward_time": 1.7522010803222656,
      "step": 37906
    },
    {
      "epoch": 3.614997863769531e-05,
      "step": 37906,
      "training_step_time": 7.745938062667847
    },
    {
      "epoch": 3.6150932312011715e-05,
      "model_forward_time": 1.788747787475586,
      "step": 37907
    },
    {
      "epoch": 3.6150932312011715e-05,
      "step": 37907,
      "training_step_time": 9.333252429962158
    },
    {
      "epoch": 3.6151885986328126e-05,
      "model_forward_time": 1.7561166286468506,
      "step": 37908
    },
    {
      "epoch": 3.6151885986328126e-05,
      "step": 37908,
      "training_step_time": 9.401681423187256
    },
    {
      "epoch": 3.615283966064453e-05,
      "model_forward_time": 2.8578858375549316,
      "step": 37909
    },
    {
      "epoch": 3.615283966064453e-05,
      "step": 37909,
      "training_step_time": 8.675414085388184
    },
    {
      "epoch": 3.6153793334960934e-05,
      "grad_norm": 0.13179424405097961,
      "learning_rate": 7.320045015011582e-06,
      "loss": 0.0561,
      "step": 37910
    },
    {
      "epoch": 3.6153793334960934e-05,
      "model_forward_time": 1.7501673698425293,
      "step": 37910
    },
    {
      "dynamics_loss_avg": 0.04380818624049425,
      "epoch": 3.6153793334960934e-05,
      "step": 37910
    },
    {
      "action_loss_avg": 0.0016627795062959193,
      "epoch": 3.6153793334960934e-05,
      "step": 37910
    },
    {
      "epoch": 3.6153793334960934e-05,
      "step": 37910,
      "training_step_time": 7.326051950454712
    },
    {
      "epoch": 3.6154747009277345e-05,
      "model_forward_time": 1.7695982456207275,
      "step": 37911
    },
    {
      "epoch": 3.6154747009277345e-05,
      "step": 37911,
      "training_step_time": 6.761243104934692
    },
    {
      "epoch": 3.615570068359375e-05,
      "model_forward_time": 1.7272858619689941,
      "step": 37912
    },
    {
      "epoch": 3.615570068359375e-05,
      "step": 37912,
      "training_step_time": 6.710602521896362
    },
    {
      "epoch": 3.615665435791015e-05,
      "model_forward_time": 1.7548699378967285,
      "step": 37913
    },
    {
      "epoch": 3.615665435791015e-05,
      "step": 37913,
      "training_step_time": 7.125956773757935
    },
    {
      "epoch": 3.6157608032226564e-05,
      "model_forward_time": 1.7545392513275146,
      "step": 37914
    },
    {
      "epoch": 3.6157608032226564e-05,
      "step": 37914,
      "training_step_time": 7.270217180252075
    },
    {
      "epoch": 3.615856170654297e-05,
      "model_forward_time": 1.7565631866455078,
      "step": 37915
    },
    {
      "epoch": 3.615856170654297e-05,
      "step": 37915,
      "training_step_time": 7.563998699188232
    },
    {
      "epoch": 3.615951538085937e-05,
      "model_forward_time": 1.7614843845367432,
      "step": 37916
    },
    {
      "epoch": 3.615951538085937e-05,
      "step": 37916,
      "training_step_time": 9.663212537765503
    },
    {
      "epoch": 3.616046905517578e-05,
      "model_forward_time": 1.749091386795044,
      "step": 37917
    },
    {
      "epoch": 3.616046905517578e-05,
      "step": 37917,
      "training_step_time": 8.243337869644165
    },
    {
      "epoch": 3.616142272949219e-05,
      "model_forward_time": 1.7807254791259766,
      "step": 37918
    },
    {
      "epoch": 3.616142272949219e-05,
      "step": 37918,
      "training_step_time": 8.249525547027588
    },
    {
      "epoch": 3.616237640380859e-05,
      "model_forward_time": 2.950171947479248,
      "step": 37919
    },
    {
      "epoch": 3.616237640380859e-05,
      "step": 37919,
      "training_step_time": 7.652117967605591
    },
    {
      "epoch": 3.6163330078125e-05,
      "grad_norm": 0.16890056431293488,
      "learning_rate": 7.318580194212253e-06,
      "loss": 0.063,
      "step": 37920
    },
    {
      "epoch": 3.6163330078125e-05,
      "model_forward_time": 1.753326654434204,
      "step": 37920
    },
    {
      "dynamics_loss_avg": 0.05237124096602201,
      "epoch": 3.6163330078125e-05,
      "step": 37920
    },
    {
      "action_loss_avg": 0.0017102580852224492,
      "epoch": 3.6163330078125e-05,
      "step": 37920
    },
    {
      "epoch": 3.6163330078125e-05,
      "step": 37920,
      "training_step_time": 6.6956024169921875
    },
    {
      "epoch": 3.6164283752441406e-05,
      "model_forward_time": 1.389469861984253,
      "step": 37921
    },
    {
      "epoch": 3.6164283752441406e-05,
      "step": 37921,
      "training_step_time": 6.723079204559326
    },
    {
      "epoch": 3.616523742675781e-05,
      "model_forward_time": 1.7550430297851562,
      "step": 37922
    },
    {
      "epoch": 3.616523742675781e-05,
      "step": 37922,
      "training_step_time": 7.087610244750977
    },
    {
      "epoch": 3.616619110107422e-05,
      "model_forward_time": 1.3959629535675049,
      "step": 37923
    },
    {
      "epoch": 3.616619110107422e-05,
      "step": 37923,
      "training_step_time": 8.289191961288452
    },
    {
      "epoch": 3.6167144775390625e-05,
      "model_forward_time": 1.7509615421295166,
      "step": 37924
    },
    {
      "epoch": 3.6167144775390625e-05,
      "step": 37924,
      "training_step_time": 8.394665718078613
    },
    {
      "epoch": 3.616809844970703e-05,
      "model_forward_time": 1.7547006607055664,
      "step": 37925
    },
    {
      "epoch": 3.616809844970703e-05,
      "step": 37925,
      "training_step_time": 10.169577598571777
    },
    {
      "epoch": 3.616905212402344e-05,
      "model_forward_time": 1.8722317218780518,
      "step": 37926
    },
    {
      "epoch": 3.616905212402344e-05,
      "step": 37926,
      "training_step_time": 9.79473876953125
    },
    {
      "epoch": 3.617000579833984e-05,
      "model_forward_time": 1.742518424987793,
      "step": 37927
    },
    {
      "epoch": 3.617000579833984e-05,
      "step": 37927,
      "training_step_time": 7.553839445114136
    },
    {
      "epoch": 3.617095947265625e-05,
      "model_forward_time": 2.9227306842803955,
      "step": 37928
    },
    {
      "epoch": 3.617095947265625e-05,
      "step": 37928,
      "training_step_time": 8.176350116729736
    },
    {
      "epoch": 3.617191314697266e-05,
      "model_forward_time": 1.0513968467712402,
      "step": 37929
    },
    {
      "epoch": 3.617191314697266e-05,
      "step": 37929,
      "training_step_time": 7.188839912414551
    },
    {
      "epoch": 3.617286682128906e-05,
      "grad_norm": 0.15779809653759003,
      "learning_rate": 7.3171151198564826e-06,
      "loss": 0.0592,
      "step": 37930
    },
    {
      "epoch": 3.617286682128906e-05,
      "model_forward_time": 1.3772430419921875,
      "step": 37930
    },
    {
      "dynamics_loss_avg": 0.049955230206251144,
      "epoch": 3.617286682128906e-05,
      "step": 37930
    },
    {
      "action_loss_avg": 0.001601550160557963,
      "epoch": 3.617286682128906e-05,
      "step": 37930
    },
    {
      "epoch": 3.617286682128906e-05,
      "step": 37930,
      "training_step_time": 6.675306558609009
    },
    {
      "epoch": 3.6173820495605466e-05,
      "model_forward_time": 1.758554220199585,
      "step": 37931
    },
    {
      "epoch": 3.6173820495605466e-05,
      "step": 37931,
      "training_step_time": 7.058867931365967
    },
    {
      "epoch": 3.617477416992188e-05,
      "model_forward_time": 1.765901803970337,
      "step": 37932
    },
    {
      "epoch": 3.617477416992188e-05,
      "step": 37932,
      "training_step_time": 7.075612783432007
    },
    {
      "epoch": 3.617572784423828e-05,
      "model_forward_time": 1.7521216869354248,
      "step": 37933
    },
    {
      "epoch": 3.617572784423828e-05,
      "step": 37933,
      "training_step_time": 7.445062160491943
    },
    {
      "epoch": 3.6176681518554685e-05,
      "model_forward_time": 1.7562851905822754,
      "step": 37934
    },
    {
      "epoch": 3.6176681518554685e-05,
      "step": 37934,
      "training_step_time": 8.529779434204102
    },
    {
      "epoch": 3.6177635192871096e-05,
      "model_forward_time": 1.8471589088439941,
      "step": 37935
    },
    {
      "epoch": 3.6177635192871096e-05,
      "step": 37935,
      "training_step_time": 11.047751426696777
    },
    {
      "epoch": 3.61785888671875e-05,
      "model_forward_time": 1.758591651916504,
      "step": 37936
    },
    {
      "epoch": 3.61785888671875e-05,
      "step": 37936,
      "training_step_time": 8.780900001525879
    },
    {
      "epoch": 3.6179542541503904e-05,
      "model_forward_time": 1.7527198791503906,
      "step": 37937
    },
    {
      "epoch": 3.6179542541503904e-05,
      "step": 37937,
      "training_step_time": 8.27808690071106
    },
    {
      "epoch": 3.6180496215820315e-05,
      "model_forward_time": 1.748042345046997,
      "step": 37938
    },
    {
      "epoch": 3.6180496215820315e-05,
      "step": 37938,
      "training_step_time": 7.254020929336548
    },
    {
      "epoch": 3.618144989013672e-05,
      "model_forward_time": 1.7599308490753174,
      "step": 37939
    },
    {
      "epoch": 3.618144989013672e-05,
      "step": 37939,
      "training_step_time": 6.91985011100769
    },
    {
      "epoch": 3.618240356445312e-05,
      "grad_norm": 0.13559336960315704,
      "learning_rate": 7.315649792104486e-06,
      "loss": 0.0626,
      "step": 37940
    },
    {
      "epoch": 3.618240356445312e-05,
      "model_forward_time": 1.7796237468719482,
      "step": 37940
    },
    {
      "dynamics_loss_avg": 0.06777959056198597,
      "epoch": 3.618240356445312e-05,
      "step": 37940
    },
    {
      "action_loss_avg": 0.002818238316103816,
      "epoch": 3.618240356445312e-05,
      "step": 37940
    },
    {
      "epoch": 3.618240356445312e-05,
      "step": 37940,
      "training_step_time": 7.408881425857544
    },
    {
      "epoch": 3.6183357238769534e-05,
      "model_forward_time": 1.754361867904663,
      "step": 37941
    },
    {
      "epoch": 3.6183357238769534e-05,
      "step": 37941,
      "training_step_time": 7.244574308395386
    },
    {
      "epoch": 3.618431091308594e-05,
      "model_forward_time": 1.7517576217651367,
      "step": 37942
    },
    {
      "epoch": 3.618431091308594e-05,
      "step": 37942,
      "training_step_time": 7.751275300979614
    },
    {
      "epoch": 3.618526458740234e-05,
      "model_forward_time": 1.7610177993774414,
      "step": 37943
    },
    {
      "epoch": 3.618526458740234e-05,
      "step": 37943,
      "training_step_time": 8.647122144699097
    },
    {
      "epoch": 3.618621826171875e-05,
      "model_forward_time": 1.7916605472564697,
      "step": 37944
    },
    {
      "epoch": 3.618621826171875e-05,
      "step": 37944,
      "training_step_time": 10.413470268249512
    },
    {
      "epoch": 3.6187171936035157e-05,
      "model_forward_time": 1.7492907047271729,
      "step": 37945
    },
    {
      "epoch": 3.6187171936035157e-05,
      "step": 37945,
      "training_step_time": 7.876043319702148
    },
    {
      "epoch": 3.618812561035156e-05,
      "model_forward_time": 1.7561922073364258,
      "step": 37946
    },
    {
      "epoch": 3.618812561035156e-05,
      "step": 37946,
      "training_step_time": 8.23887300491333
    },
    {
      "epoch": 3.618907928466797e-05,
      "model_forward_time": 1.753697156906128,
      "step": 37947
    },
    {
      "epoch": 3.618907928466797e-05,
      "step": 37947,
      "training_step_time": 7.733453989028931
    },
    {
      "epoch": 3.6190032958984375e-05,
      "model_forward_time": 1.7619104385375977,
      "step": 37948
    },
    {
      "epoch": 3.6190032958984375e-05,
      "step": 37948,
      "training_step_time": 6.908486604690552
    },
    {
      "epoch": 3.619098663330078e-05,
      "model_forward_time": 1.7637500762939453,
      "step": 37949
    },
    {
      "epoch": 3.619098663330078e-05,
      "step": 37949,
      "training_step_time": 6.8649632930755615
    },
    {
      "epoch": 3.619194030761719e-05,
      "grad_norm": 0.11193495988845825,
      "learning_rate": 7.3141842111165075e-06,
      "loss": 0.0594,
      "step": 37950
    },
    {
      "epoch": 3.619194030761719e-05,
      "model_forward_time": 1.752150297164917,
      "step": 37950
    },
    {
      "dynamics_loss_avg": 0.056979861855506894,
      "epoch": 3.619194030761719e-05,
      "step": 37950
    },
    {
      "action_loss_avg": 0.0032506863411981613,
      "epoch": 3.619194030761719e-05,
      "step": 37950
    },
    {
      "epoch": 3.619194030761719e-05,
      "step": 37950,
      "training_step_time": 7.250489711761475
    },
    {
      "epoch": 3.6192893981933594e-05,
      "model_forward_time": 1.7921831607818604,
      "step": 37951
    },
    {
      "epoch": 3.6192893981933594e-05,
      "step": 37951,
      "training_step_time": 7.649837255477905
    },
    {
      "epoch": 3.619384765625e-05,
      "model_forward_time": 1.7498674392700195,
      "step": 37952
    },
    {
      "epoch": 3.619384765625e-05,
      "step": 37952,
      "training_step_time": 8.381184577941895
    },
    {
      "epoch": 3.619480133056641e-05,
      "model_forward_time": 1.7558197975158691,
      "step": 37953
    },
    {
      "epoch": 3.619480133056641e-05,
      "step": 37953,
      "training_step_time": 8.468850135803223
    },
    {
      "epoch": 3.619575500488281e-05,
      "model_forward_time": 1.7532720565795898,
      "step": 37954
    },
    {
      "epoch": 3.619575500488281e-05,
      "step": 37954,
      "training_step_time": 7.76294732093811
    },
    {
      "epoch": 3.619670867919922e-05,
      "model_forward_time": 1.7521302700042725,
      "step": 37955
    },
    {
      "epoch": 3.619670867919922e-05,
      "step": 37955,
      "training_step_time": 8.181311845779419
    },
    {
      "epoch": 3.619766235351563e-05,
      "model_forward_time": 1.7984569072723389,
      "step": 37956
    },
    {
      "epoch": 3.619766235351563e-05,
      "step": 37956,
      "training_step_time": 7.67531418800354
    },
    {
      "epoch": 3.619861602783203e-05,
      "model_forward_time": 1.737553596496582,
      "step": 37957
    },
    {
      "epoch": 3.619861602783203e-05,
      "step": 37957,
      "training_step_time": 6.916917324066162
    },
    {
      "epoch": 3.6199569702148436e-05,
      "model_forward_time": 1.7351140975952148,
      "step": 37958
    },
    {
      "epoch": 3.6199569702148436e-05,
      "step": 37958,
      "training_step_time": 6.896001577377319
    },
    {
      "epoch": 3.620052337646485e-05,
      "model_forward_time": 1.740724802017212,
      "step": 37959
    },
    {
      "epoch": 3.620052337646485e-05,
      "step": 37959,
      "training_step_time": 7.229357481002808
    },
    {
      "epoch": 3.620147705078125e-05,
      "grad_norm": 0.1200404167175293,
      "learning_rate": 7.312718377052824e-06,
      "loss": 0.0607,
      "step": 37960
    },
    {
      "epoch": 3.620147705078125e-05,
      "model_forward_time": 1.7489609718322754,
      "step": 37960
    },
    {
      "dynamics_loss_avg": 0.05926499255001545,
      "epoch": 3.620147705078125e-05,
      "step": 37960
    },
    {
      "action_loss_avg": 0.0018139186198823155,
      "epoch": 3.620147705078125e-05,
      "step": 37960
    },
    {
      "epoch": 3.620147705078125e-05,
      "step": 37960,
      "training_step_time": 7.732460021972656
    },
    {
      "epoch": 3.6202430725097655e-05,
      "model_forward_time": 1.7335965633392334,
      "step": 37961
    },
    {
      "epoch": 3.6202430725097655e-05,
      "step": 37961,
      "training_step_time": 8.642008781433105
    },
    {
      "epoch": 3.6203384399414066e-05,
      "model_forward_time": 1.7420434951782227,
      "step": 37962
    },
    {
      "epoch": 3.6203384399414066e-05,
      "step": 37962,
      "training_step_time": 9.852188110351562
    },
    {
      "epoch": 3.620433807373047e-05,
      "model_forward_time": 1.7503066062927246,
      "step": 37963
    },
    {
      "epoch": 3.620433807373047e-05,
      "step": 37963,
      "training_step_time": 7.791049480438232
    },
    {
      "epoch": 3.6205291748046874e-05,
      "model_forward_time": 1.747279405593872,
      "step": 37964
    },
    {
      "epoch": 3.6205291748046874e-05,
      "step": 37964,
      "training_step_time": 7.5089921951293945
    },
    {
      "epoch": 3.6206245422363285e-05,
      "model_forward_time": 2.544358730316162,
      "step": 37965
    },
    {
      "epoch": 3.6206245422363285e-05,
      "step": 37965,
      "training_step_time": 7.577839612960815
    },
    {
      "epoch": 3.620719909667969e-05,
      "model_forward_time": 1.7408146858215332,
      "step": 37966
    },
    {
      "epoch": 3.620719909667969e-05,
      "step": 37966,
      "training_step_time": 6.695770502090454
    },
    {
      "epoch": 3.620815277099609e-05,
      "model_forward_time": 1.7368090152740479,
      "step": 37967
    },
    {
      "epoch": 3.620815277099609e-05,
      "step": 37967,
      "training_step_time": 6.658713340759277
    },
    {
      "epoch": 3.62091064453125e-05,
      "model_forward_time": 1.741424560546875,
      "step": 37968
    },
    {
      "epoch": 3.62091064453125e-05,
      "step": 37968,
      "training_step_time": 7.072772264480591
    },
    {
      "epoch": 3.621006011962891e-05,
      "model_forward_time": 1.7447431087493896,
      "step": 37969
    },
    {
      "epoch": 3.621006011962891e-05,
      "step": 37969,
      "training_step_time": 8.195391416549683
    },
    {
      "epoch": 3.621101379394531e-05,
      "grad_norm": 0.15179644525051117,
      "learning_rate": 7.311252290073737e-06,
      "loss": 0.0583,
      "step": 37970
    },
    {
      "epoch": 3.621101379394531e-05,
      "model_forward_time": 1.7666325569152832,
      "step": 37970
    },
    {
      "dynamics_loss_avg": 0.05443586837500334,
      "epoch": 3.621101379394531e-05,
      "step": 37970
    },
    {
      "action_loss_avg": 0.002496183436596766,
      "epoch": 3.621101379394531e-05,
      "step": 37970
    },
    {
      "epoch": 3.621101379394531e-05,
      "step": 37970,
      "training_step_time": 7.638754367828369
    },
    {
      "epoch": 3.6211967468261715e-05,
      "model_forward_time": 1.7372901439666748,
      "step": 37971
    },
    {
      "epoch": 3.6211967468261715e-05,
      "step": 37971,
      "training_step_time": 8.226177453994751
    },
    {
      "epoch": 3.6212921142578126e-05,
      "model_forward_time": 1.7689063549041748,
      "step": 37972
    },
    {
      "epoch": 3.6212921142578126e-05,
      "step": 37972,
      "training_step_time": 7.659469127655029
    },
    {
      "epoch": 3.621387481689453e-05,
      "model_forward_time": 1.744539737701416,
      "step": 37973
    },
    {
      "epoch": 3.621387481689453e-05,
      "step": 37973,
      "training_step_time": 7.7129387855529785
    },
    {
      "epoch": 3.6214828491210934e-05,
      "model_forward_time": 2.912388801574707,
      "step": 37974
    },
    {
      "epoch": 3.6214828491210934e-05,
      "step": 37974,
      "training_step_time": 8.663869380950928
    },
    {
      "epoch": 3.6215782165527345e-05,
      "model_forward_time": 1.7474839687347412,
      "step": 37975
    },
    {
      "epoch": 3.6215782165527345e-05,
      "step": 37975,
      "training_step_time": 7.574373483657837
    },
    {
      "epoch": 3.621673583984375e-05,
      "model_forward_time": 1.7443361282348633,
      "step": 37976
    },
    {
      "epoch": 3.621673583984375e-05,
      "step": 37976,
      "training_step_time": 6.712519407272339
    },
    {
      "epoch": 3.621768951416015e-05,
      "model_forward_time": 1.7399983406066895,
      "step": 37977
    },
    {
      "epoch": 3.621768951416015e-05,
      "step": 37977,
      "training_step_time": 7.087158441543579
    },
    {
      "epoch": 3.6218643188476564e-05,
      "model_forward_time": 1.7448737621307373,
      "step": 37978
    },
    {
      "epoch": 3.6218643188476564e-05,
      "step": 37978,
      "training_step_time": 7.467671155929565
    },
    {
      "epoch": 3.621959686279297e-05,
      "model_forward_time": 1.7587122917175293,
      "step": 37979
    },
    {
      "epoch": 3.621959686279297e-05,
      "step": 37979,
      "training_step_time": 8.473295450210571
    },
    {
      "epoch": 3.622055053710937e-05,
      "grad_norm": 0.17924438416957855,
      "learning_rate": 7.309785950339573e-06,
      "loss": 0.0626,
      "step": 37980
    },
    {
      "epoch": 3.622055053710937e-05,
      "model_forward_time": 1.741015911102295,
      "step": 37980
    },
    {
      "dynamics_loss_avg": 0.058505570702254775,
      "epoch": 3.622055053710937e-05,
      "step": 37980
    },
    {
      "action_loss_avg": 0.0024376342917094007,
      "epoch": 3.622055053710937e-05,
      "step": 37980
    },
    {
      "epoch": 3.622055053710937e-05,
      "step": 37980,
      "training_step_time": 9.110770463943481
    },
    {
      "epoch": 3.622150421142578e-05,
      "model_forward_time": 1.7557566165924072,
      "step": 37981
    },
    {
      "epoch": 3.622150421142578e-05,
      "step": 37981,
      "training_step_time": 9.3617103099823
    },
    {
      "epoch": 3.622245788574219e-05,
      "model_forward_time": 1.7519183158874512,
      "step": 37982
    },
    {
      "epoch": 3.622245788574219e-05,
      "step": 37982,
      "training_step_time": 7.6577677726745605
    },
    {
      "epoch": 3.622341156005859e-05,
      "model_forward_time": 2.901275634765625,
      "step": 37983
    },
    {
      "epoch": 3.622341156005859e-05,
      "step": 37983,
      "training_step_time": 7.624074935913086
    },
    {
      "epoch": 3.6224365234375e-05,
      "model_forward_time": 1.7569618225097656,
      "step": 37984
    },
    {
      "epoch": 3.6224365234375e-05,
      "step": 37984,
      "training_step_time": 7.566601037979126
    },
    {
      "epoch": 3.6225318908691406e-05,
      "model_forward_time": 1.7615656852722168,
      "step": 37985
    },
    {
      "epoch": 3.6225318908691406e-05,
      "step": 37985,
      "training_step_time": 6.696511745452881
    },
    {
      "epoch": 3.622627258300781e-05,
      "model_forward_time": 1.7663307189941406,
      "step": 37986
    },
    {
      "epoch": 3.622627258300781e-05,
      "step": 37986,
      "training_step_time": 7.111263751983643
    },
    {
      "epoch": 3.622722625732422e-05,
      "model_forward_time": 1.7652418613433838,
      "step": 37987
    },
    {
      "epoch": 3.622722625732422e-05,
      "step": 37987,
      "training_step_time": 7.498678207397461
    },
    {
      "epoch": 3.6228179931640625e-05,
      "model_forward_time": 1.833740472793579,
      "step": 37988
    },
    {
      "epoch": 3.6228179931640625e-05,
      "step": 37988,
      "training_step_time": 8.384727478027344
    },
    {
      "epoch": 3.622913360595703e-05,
      "model_forward_time": 1.7603869438171387,
      "step": 37989
    },
    {
      "epoch": 3.622913360595703e-05,
      "step": 37989,
      "training_step_time": 8.099384784698486
    },
    {
      "epoch": 3.623008728027344e-05,
      "grad_norm": 0.18052460253238678,
      "learning_rate": 7.308319358010691e-06,
      "loss": 0.0608,
      "step": 37990
    },
    {
      "epoch": 3.623008728027344e-05,
      "model_forward_time": 1.7646355628967285,
      "step": 37990
    },
    {
      "dynamics_loss_avg": 0.0554956004023552,
      "epoch": 3.623008728027344e-05,
      "step": 37990
    },
    {
      "action_loss_avg": 0.002172655408503488,
      "epoch": 3.623008728027344e-05,
      "step": 37990
    },
    {
      "epoch": 3.623008728027344e-05,
      "step": 37990,
      "training_step_time": 8.184475660324097
    },
    {
      "epoch": 3.6231040954589843e-05,
      "model_forward_time": 1.7552783489227295,
      "step": 37991
    },
    {
      "epoch": 3.6231040954589843e-05,
      "step": 37991,
      "training_step_time": 7.6389079093933105
    },
    {
      "epoch": 3.623199462890625e-05,
      "model_forward_time": 2.9319188594818115,
      "step": 37992
    },
    {
      "epoch": 3.623199462890625e-05,
      "step": 37992,
      "training_step_time": 7.645226716995239
    },
    {
      "epoch": 3.623294830322266e-05,
      "model_forward_time": 1.7612125873565674,
      "step": 37993
    },
    {
      "epoch": 3.623294830322266e-05,
      "step": 37993,
      "training_step_time": 7.584993839263916
    },
    {
      "epoch": 3.623390197753906e-05,
      "model_forward_time": 1.7616722583770752,
      "step": 37994
    },
    {
      "epoch": 3.623390197753906e-05,
      "step": 37994,
      "training_step_time": 6.697711229324341
    },
    {
      "epoch": 3.6234855651855466e-05,
      "model_forward_time": 1.7509269714355469,
      "step": 37995
    },
    {
      "epoch": 3.6234855651855466e-05,
      "step": 37995,
      "training_step_time": 6.709009885787964
    },
    {
      "epoch": 3.623580932617188e-05,
      "model_forward_time": 1.7579271793365479,
      "step": 37996
    },
    {
      "epoch": 3.623580932617188e-05,
      "step": 37996,
      "training_step_time": 7.508349657058716
    },
    {
      "epoch": 3.623676300048828e-05,
      "model_forward_time": 1.8346920013427734,
      "step": 37997
    },
    {
      "epoch": 3.623676300048828e-05,
      "step": 37997,
      "training_step_time": 8.757889032363892
    },
    {
      "epoch": 3.6237716674804685e-05,
      "model_forward_time": 1.7711822986602783,
      "step": 37998
    },
    {
      "epoch": 3.6237716674804685e-05,
      "step": 37998,
      "training_step_time": 8.660592317581177
    },
    {
      "epoch": 3.6238670349121096e-05,
      "model_forward_time": 1.7497057914733887,
      "step": 37999
    },
    {
      "epoch": 3.6238670349121096e-05,
      "step": 37999,
      "training_step_time": 8.36103105545044
    },
    {
      "epoch": 3.62396240234375e-05,
      "grad_norm": 0.11400821059942245,
      "learning_rate": 7.3068525132474745e-06,
      "loss": 0.0602,
      "step": 38000
    },
    {
      "epoch": 3.62396240234375e-05,
      "model_forward_time": 1.722914218902588,
      "step": 38000
    },
    {
      "dynamics_loss_avg": 0.05098544657230377,
      "epoch": 3.62396240234375e-05,
      "step": 38000
    },
    {
      "action_loss_avg": 0.0019557921041268855,
      "epoch": 3.62396240234375e-05,
      "step": 38000
    },
    {
      "epoch": 3.62396240234375e-05,
      "step": 38000,
      "training_step_time": 8.196613311767578
    },
    {
      "epoch": 3.6240577697753904e-05,
      "model_forward_time": 2.7924976348876953,
      "step": 38001
    },
    {
      "epoch": 3.6240577697753904e-05,
      "step": 38001,
      "training_step_time": 7.573686599731445
    },
    {
      "epoch": 3.6241531372070315e-05,
      "model_forward_time": 1.709719181060791,
      "step": 38002
    },
    {
      "epoch": 3.6241531372070315e-05,
      "step": 38002,
      "training_step_time": 7.699367046356201
    },
    {
      "epoch": 3.624248504638672e-05,
      "model_forward_time": 1.7139554023742676,
      "step": 38003
    },
    {
      "epoch": 3.624248504638672e-05,
      "step": 38003,
      "training_step_time": 6.799761772155762
    },
    {
      "epoch": 3.624343872070312e-05,
      "model_forward_time": 1.6876709461212158,
      "step": 38004
    },
    {
      "epoch": 3.624343872070312e-05,
      "step": 38004,
      "training_step_time": 6.734521865844727
    },
    {
      "epoch": 3.6244392395019534e-05,
      "model_forward_time": 1.717958688735962,
      "step": 38005
    },
    {
      "epoch": 3.6244392395019534e-05,
      "step": 38005,
      "training_step_time": 8.66394591331482
    },
    {
      "epoch": 3.624534606933594e-05,
      "model_forward_time": 1.7063472270965576,
      "step": 38006
    },
    {
      "epoch": 3.624534606933594e-05,
      "step": 38006,
      "training_step_time": 9.811068058013916
    },
    {
      "epoch": 3.624629974365234e-05,
      "model_forward_time": 1.7010469436645508,
      "step": 38007
    },
    {
      "epoch": 3.624629974365234e-05,
      "step": 38007,
      "training_step_time": 9.737797498703003
    },
    {
      "epoch": 3.624725341796875e-05,
      "model_forward_time": 1.7238147258758545,
      "step": 38008
    },
    {
      "epoch": 3.624725341796875e-05,
      "step": 38008,
      "training_step_time": 13.325292587280273
    },
    {
      "epoch": 3.6248207092285157e-05,
      "model_forward_time": 1.7142927646636963,
      "step": 38009
    },
    {
      "epoch": 3.6248207092285157e-05,
      "step": 38009,
      "training_step_time": 9.808175563812256
    },
    {
      "epoch": 3.624916076660156e-05,
      "grad_norm": 0.12102491408586502,
      "learning_rate": 7.305385416210335e-06,
      "loss": 0.0619,
      "step": 38010
    },
    {
      "epoch": 3.624916076660156e-05,
      "model_forward_time": 2.2620062828063965,
      "step": 38010
    },
    {
      "dynamics_loss_avg": 0.06901194639503956,
      "epoch": 3.624916076660156e-05,
      "step": 38010
    },
    {
      "action_loss_avg": 0.0020896523608826103,
      "epoch": 3.624916076660156e-05,
      "step": 38010
    },
    {
      "epoch": 3.624916076660156e-05,
      "step": 38010,
      "training_step_time": 7.778094530105591
    },
    {
      "epoch": 3.625011444091797e-05,
      "model_forward_time": 0.997502326965332,
      "step": 38011
    },
    {
      "epoch": 3.625011444091797e-05,
      "step": 38011,
      "training_step_time": 7.803663492202759
    },
    {
      "epoch": 3.6251068115234375e-05,
      "model_forward_time": 1.6949143409729004,
      "step": 38012
    },
    {
      "epoch": 3.6251068115234375e-05,
      "step": 38012,
      "training_step_time": 6.87101936340332
    },
    {
      "epoch": 3.625202178955078e-05,
      "model_forward_time": 1.700756549835205,
      "step": 38013
    },
    {
      "epoch": 3.625202178955078e-05,
      "step": 38013,
      "training_step_time": 6.838726282119751
    },
    {
      "epoch": 3.625297546386719e-05,
      "model_forward_time": 1.7092232704162598,
      "step": 38014
    },
    {
      "epoch": 3.625297546386719e-05,
      "step": 38014,
      "training_step_time": 7.219203948974609
    },
    {
      "epoch": 3.6253929138183594e-05,
      "model_forward_time": 1.706850290298462,
      "step": 38015
    },
    {
      "epoch": 3.6253929138183594e-05,
      "step": 38015,
      "training_step_time": 8.378473997116089
    },
    {
      "epoch": 3.62548828125e-05,
      "model_forward_time": 1.6988115310668945,
      "step": 38016
    },
    {
      "epoch": 3.62548828125e-05,
      "step": 38016,
      "training_step_time": 8.42559027671814
    },
    {
      "epoch": 3.625583648681641e-05,
      "model_forward_time": 1.732417106628418,
      "step": 38017
    },
    {
      "epoch": 3.625583648681641e-05,
      "step": 38017,
      "training_step_time": 8.647526264190674
    },
    {
      "epoch": 3.625679016113281e-05,
      "model_forward_time": 1.7061643600463867,
      "step": 38018
    },
    {
      "epoch": 3.625679016113281e-05,
      "step": 38018,
      "training_step_time": 9.755887508392334
    },
    {
      "epoch": 3.625774383544922e-05,
      "model_forward_time": 1.6911211013793945,
      "step": 38019
    },
    {
      "epoch": 3.625774383544922e-05,
      "step": 38019,
      "training_step_time": 8.274298667907715
    },
    {
      "epoch": 3.625869750976563e-05,
      "grad_norm": 0.1308194398880005,
      "learning_rate": 7.303918067059712e-06,
      "loss": 0.0607,
      "step": 38020
    },
    {
      "epoch": 3.625869750976563e-05,
      "model_forward_time": 1.6991713047027588,
      "step": 38020
    },
    {
      "dynamics_loss_avg": 0.038272087834775446,
      "epoch": 3.625869750976563e-05,
      "step": 38020
    },
    {
      "action_loss_avg": 0.0014710488263517617,
      "epoch": 3.625869750976563e-05,
      "step": 38020
    },
    {
      "epoch": 3.625869750976563e-05,
      "step": 38020,
      "training_step_time": 7.636707305908203
    },
    {
      "epoch": 3.625965118408203e-05,
      "model_forward_time": 1.7171125411987305,
      "step": 38021
    },
    {
      "epoch": 3.625965118408203e-05,
      "step": 38021,
      "training_step_time": 7.687143802642822
    },
    {
      "epoch": 3.6260604858398436e-05,
      "model_forward_time": 1.711582899093628,
      "step": 38022
    },
    {
      "epoch": 3.6260604858398436e-05,
      "step": 38022,
      "training_step_time": 6.810549974441528
    },
    {
      "epoch": 3.626155853271485e-05,
      "model_forward_time": 1.7142722606658936,
      "step": 38023
    },
    {
      "epoch": 3.626155853271485e-05,
      "step": 38023,
      "training_step_time": 7.188699722290039
    },
    {
      "epoch": 3.626251220703125e-05,
      "model_forward_time": 1.7114086151123047,
      "step": 38024
    },
    {
      "epoch": 3.626251220703125e-05,
      "step": 38024,
      "training_step_time": 7.651611566543579
    },
    {
      "epoch": 3.6263465881347655e-05,
      "model_forward_time": 1.7138469219207764,
      "step": 38025
    },
    {
      "epoch": 3.6263465881347655e-05,
      "step": 38025,
      "training_step_time": 8.68852972984314
    },
    {
      "epoch": 3.6264419555664066e-05,
      "model_forward_time": 1.739804983139038,
      "step": 38026
    },
    {
      "epoch": 3.6264419555664066e-05,
      "step": 38026,
      "training_step_time": 8.691900253295898
    },
    {
      "epoch": 3.626537322998047e-05,
      "model_forward_time": 1.7199652194976807,
      "step": 38027
    },
    {
      "epoch": 3.626537322998047e-05,
      "step": 38027,
      "training_step_time": 7.855553388595581
    },
    {
      "epoch": 3.6266326904296874e-05,
      "model_forward_time": 1.7161870002746582,
      "step": 38028
    },
    {
      "epoch": 3.6266326904296874e-05,
      "step": 38028,
      "training_step_time": 7.647087097167969
    },
    {
      "epoch": 3.6267280578613285e-05,
      "model_forward_time": 2.253767251968384,
      "step": 38029
    },
    {
      "epoch": 3.6267280578613285e-05,
      "step": 38029,
      "training_step_time": 8.114936113357544
    },
    {
      "epoch": 3.626823425292969e-05,
      "grad_norm": 0.13915306329727173,
      "learning_rate": 7.302450465956075e-06,
      "loss": 0.0557,
      "step": 38030
    },
    {
      "epoch": 3.626823425292969e-05,
      "model_forward_time": 1.7086327075958252,
      "step": 38030
    },
    {
      "dynamics_loss_avg": 0.05336275231093168,
      "epoch": 3.626823425292969e-05,
      "step": 38030
    },
    {
      "action_loss_avg": 0.0018542095902375876,
      "epoch": 3.626823425292969e-05,
      "step": 38030
    },
    {
      "epoch": 3.626823425292969e-05,
      "step": 38030,
      "training_step_time": 7.602853059768677
    },
    {
      "epoch": 3.626918792724609e-05,
      "model_forward_time": 1.7111003398895264,
      "step": 38031
    },
    {
      "epoch": 3.626918792724609e-05,
      "step": 38031,
      "training_step_time": 6.665764093399048
    },
    {
      "epoch": 3.62701416015625e-05,
      "model_forward_time": 1.7155208587646484,
      "step": 38032
    },
    {
      "epoch": 3.62701416015625e-05,
      "step": 38032,
      "training_step_time": 7.275984287261963
    },
    {
      "epoch": 3.627109527587891e-05,
      "model_forward_time": 1.7118916511535645,
      "step": 38033
    },
    {
      "epoch": 3.627109527587891e-05,
      "step": 38033,
      "training_step_time": 7.5310235023498535
    },
    {
      "epoch": 3.627204895019531e-05,
      "model_forward_time": 1.7163372039794922,
      "step": 38034
    },
    {
      "epoch": 3.627204895019531e-05,
      "step": 38034,
      "training_step_time": 8.749511480331421
    },
    {
      "epoch": 3.6273002624511715e-05,
      "model_forward_time": 1.7384896278381348,
      "step": 38035
    },
    {
      "epoch": 3.6273002624511715e-05,
      "step": 38035,
      "training_step_time": 8.481342077255249
    },
    {
      "epoch": 3.6273956298828126e-05,
      "model_forward_time": 1.7202417850494385,
      "step": 38036
    },
    {
      "epoch": 3.6273956298828126e-05,
      "step": 38036,
      "training_step_time": 8.430152893066406
    },
    {
      "epoch": 3.627490997314453e-05,
      "model_forward_time": 1.704235315322876,
      "step": 38037
    },
    {
      "epoch": 3.627490997314453e-05,
      "step": 38037,
      "training_step_time": 9.355116367340088
    },
    {
      "epoch": 3.6275863647460934e-05,
      "model_forward_time": 2.2669546604156494,
      "step": 38038
    },
    {
      "epoch": 3.6275863647460934e-05,
      "step": 38038,
      "training_step_time": 7.571971893310547
    },
    {
      "epoch": 3.6276817321777345e-05,
      "model_forward_time": 1.7099857330322266,
      "step": 38039
    },
    {
      "epoch": 3.6276817321777345e-05,
      "step": 38039,
      "training_step_time": 7.5967583656311035
    },
    {
      "epoch": 3.627777099609375e-05,
      "grad_norm": 0.1286652386188507,
      "learning_rate": 7.3009826130599144e-06,
      "loss": 0.0622,
      "step": 38040
    },
    {
      "epoch": 3.627777099609375e-05,
      "model_forward_time": 1.702928066253662,
      "step": 38040
    },
    {
      "dynamics_loss_avg": 0.06312233712524176,
      "epoch": 3.627777099609375e-05,
      "step": 38040
    },
    {
      "action_loss_avg": 0.0017891180556034668,
      "epoch": 3.627777099609375e-05,
      "step": 38040
    },
    {
      "epoch": 3.627777099609375e-05,
      "step": 38040,
      "training_step_time": 6.692835807800293
    },
    {
      "epoch": 3.627872467041015e-05,
      "model_forward_time": 1.7074847221374512,
      "step": 38041
    },
    {
      "epoch": 3.627872467041015e-05,
      "step": 38041,
      "training_step_time": 6.710124969482422
    },
    {
      "epoch": 3.6279678344726564e-05,
      "model_forward_time": 1.7148330211639404,
      "step": 38042
    },
    {
      "epoch": 3.6279678344726564e-05,
      "step": 38042,
      "training_step_time": 8.086610317230225
    },
    {
      "epoch": 3.628063201904297e-05,
      "model_forward_time": 1.714590072631836,
      "step": 38043
    },
    {
      "epoch": 3.628063201904297e-05,
      "step": 38043,
      "training_step_time": 7.836615800857544
    },
    {
      "epoch": 3.628158569335937e-05,
      "model_forward_time": 1.7163984775543213,
      "step": 38044
    },
    {
      "epoch": 3.628158569335937e-05,
      "step": 38044,
      "training_step_time": 8.707502603530884
    },
    {
      "epoch": 3.628253936767578e-05,
      "model_forward_time": 1.7103171348571777,
      "step": 38045
    },
    {
      "epoch": 3.628253936767578e-05,
      "step": 38045,
      "training_step_time": 7.871442794799805
    },
    {
      "epoch": 3.628349304199219e-05,
      "model_forward_time": 1.7360124588012695,
      "step": 38046
    },
    {
      "epoch": 3.628349304199219e-05,
      "step": 38046,
      "training_step_time": 9.652974367141724
    },
    {
      "epoch": 3.628444671630859e-05,
      "model_forward_time": 2.258622884750366,
      "step": 38047
    },
    {
      "epoch": 3.628444671630859e-05,
      "step": 38047,
      "training_step_time": 7.7555670738220215
    },
    {
      "epoch": 3.6285400390625e-05,
      "model_forward_time": 1.7151734828948975,
      "step": 38048
    },
    {
      "epoch": 3.6285400390625e-05,
      "step": 38048,
      "training_step_time": 7.787789821624756
    },
    {
      "epoch": 3.6286354064941406e-05,
      "model_forward_time": 1.690234899520874,
      "step": 38049
    },
    {
      "epoch": 3.6286354064941406e-05,
      "step": 38049,
      "training_step_time": 7.386120796203613
    },
    {
      "epoch": 3.628730773925781e-05,
      "grad_norm": 0.13283629715442657,
      "learning_rate": 7.299514508531756e-06,
      "loss": 0.0612,
      "step": 38050
    },
    {
      "epoch": 3.628730773925781e-05,
      "model_forward_time": 1.7003247737884521,
      "step": 38050
    },
    {
      "dynamics_loss_avg": 0.04381913831457496,
      "epoch": 3.628730773925781e-05,
      "step": 38050
    },
    {
      "action_loss_avg": 0.001993925916031003,
      "epoch": 3.628730773925781e-05,
      "step": 38050
    },
    {
      "epoch": 3.628730773925781e-05,
      "step": 38050,
      "training_step_time": 6.760588645935059
    },
    {
      "epoch": 3.628826141357422e-05,
      "model_forward_time": 1.7078444957733154,
      "step": 38051
    },
    {
      "epoch": 3.628826141357422e-05,
      "step": 38051,
      "training_step_time": 7.6888267993927
    },
    {
      "epoch": 3.6289215087890625e-05,
      "model_forward_time": 1.7075893878936768,
      "step": 38052
    },
    {
      "epoch": 3.6289215087890625e-05,
      "step": 38052,
      "training_step_time": 7.953711986541748
    },
    {
      "epoch": 3.629016876220703e-05,
      "model_forward_time": 1.701866626739502,
      "step": 38053
    },
    {
      "epoch": 3.629016876220703e-05,
      "step": 38053,
      "training_step_time": 8.735614776611328
    },
    {
      "epoch": 3.629112243652344e-05,
      "model_forward_time": 1.7425150871276855,
      "step": 38054
    },
    {
      "epoch": 3.629112243652344e-05,
      "step": 38054,
      "training_step_time": 7.875530481338501
    },
    {
      "epoch": 3.6292076110839843e-05,
      "model_forward_time": 1.7201521396636963,
      "step": 38055
    },
    {
      "epoch": 3.6292076110839843e-05,
      "step": 38055,
      "training_step_time": 7.770474195480347
    },
    {
      "epoch": 3.629302978515625e-05,
      "model_forward_time": 2.255862236022949,
      "step": 38056
    },
    {
      "epoch": 3.629302978515625e-05,
      "step": 38056,
      "training_step_time": 7.764837265014648
    },
    {
      "epoch": 3.629398345947266e-05,
      "model_forward_time": 1.3296895027160645,
      "step": 38057
    },
    {
      "epoch": 3.629398345947266e-05,
      "step": 38057,
      "training_step_time": 7.682069301605225
    },
    {
      "epoch": 3.629493713378906e-05,
      "model_forward_time": 1.704575777053833,
      "step": 38058
    },
    {
      "epoch": 3.629493713378906e-05,
      "step": 38058,
      "training_step_time": 7.255420923233032
    },
    {
      "epoch": 3.6295890808105466e-05,
      "model_forward_time": 1.7125828266143799,
      "step": 38059
    },
    {
      "epoch": 3.6295890808105466e-05,
      "step": 38059,
      "training_step_time": 6.757324695587158
    },
    {
      "epoch": 3.629684448242188e-05,
      "grad_norm": 0.1485181599855423,
      "learning_rate": 7.298046152532149e-06,
      "loss": 0.0636,
      "step": 38060
    },
    {
      "epoch": 3.629684448242188e-05,
      "model_forward_time": 1.7036817073822021,
      "step": 38060
    },
    {
      "dynamics_loss_avg": 0.06535751800984144,
      "epoch": 3.629684448242188e-05,
      "step": 38060
    },
    {
      "action_loss_avg": 0.002430070866830647,
      "epoch": 3.629684448242188e-05,
      "step": 38060
    },
    {
      "epoch": 3.629684448242188e-05,
      "step": 38060,
      "training_step_time": 7.601383209228516
    },
    {
      "epoch": 3.629779815673828e-05,
      "model_forward_time": 1.7071378231048584,
      "step": 38061
    },
    {
      "epoch": 3.629779815673828e-05,
      "step": 38061,
      "training_step_time": 8.341848373413086
    },
    {
      "epoch": 3.6298751831054685e-05,
      "model_forward_time": 1.7018284797668457,
      "step": 38062
    },
    {
      "epoch": 3.6298751831054685e-05,
      "step": 38062,
      "training_step_time": 8.448033809661865
    },
    {
      "epoch": 3.6299705505371096e-05,
      "model_forward_time": 1.7291228771209717,
      "step": 38063
    },
    {
      "epoch": 3.6299705505371096e-05,
      "step": 38063,
      "training_step_time": 8.212007284164429
    },
    {
      "epoch": 3.63006591796875e-05,
      "model_forward_time": 1.7240509986877441,
      "step": 38064
    },
    {
      "epoch": 3.63006591796875e-05,
      "step": 38064,
      "training_step_time": 7.909526109695435
    },
    {
      "epoch": 3.6301612854003904e-05,
      "model_forward_time": 2.2375986576080322,
      "step": 38065
    },
    {
      "epoch": 3.6301612854003904e-05,
      "step": 38065,
      "training_step_time": 7.8663716316223145
    },
    {
      "epoch": 3.6302566528320315e-05,
      "model_forward_time": 1.7005736827850342,
      "step": 38066
    },
    {
      "epoch": 3.6302566528320315e-05,
      "step": 38066,
      "training_step_time": 7.752694368362427
    },
    {
      "epoch": 3.630352020263672e-05,
      "model_forward_time": 1.7063114643096924,
      "step": 38067
    },
    {
      "epoch": 3.630352020263672e-05,
      "step": 38067,
      "training_step_time": 7.7892372608184814
    },
    {
      "epoch": 3.630447387695312e-05,
      "model_forward_time": 0.9809858798980713,
      "step": 38068
    },
    {
      "epoch": 3.630447387695312e-05,
      "step": 38068,
      "training_step_time": 6.738865852355957
    },
    {
      "epoch": 3.6305427551269534e-05,
      "model_forward_time": 1.7131311893463135,
      "step": 38069
    },
    {
      "epoch": 3.6305427551269534e-05,
      "step": 38069,
      "training_step_time": 7.771870851516724
    },
    {
      "epoch": 3.630638122558594e-05,
      "grad_norm": 0.12035023421049118,
      "learning_rate": 7.2965775452216684e-06,
      "loss": 0.0619,
      "step": 38070
    },
    {
      "epoch": 3.630638122558594e-05,
      "model_forward_time": 1.7067539691925049,
      "step": 38070
    },
    {
      "dynamics_loss_avg": 0.046859874948859216,
      "epoch": 3.630638122558594e-05,
      "step": 38070
    },
    {
      "action_loss_avg": 0.0008137334458297119,
      "epoch": 3.630638122558594e-05,
      "step": 38070
    },
    {
      "epoch": 3.630638122558594e-05,
      "step": 38070,
      "training_step_time": 7.9045093059539795
    },
    {
      "epoch": 3.630733489990234e-05,
      "model_forward_time": 1.7055716514587402,
      "step": 38071
    },
    {
      "epoch": 3.630733489990234e-05,
      "step": 38071,
      "training_step_time": 8.685905694961548
    },
    {
      "epoch": 3.630828857421875e-05,
      "model_forward_time": 1.7320878505706787,
      "step": 38072
    },
    {
      "epoch": 3.630828857421875e-05,
      "step": 38072,
      "training_step_time": 8.872217655181885
    },
    {
      "epoch": 3.630924224853516e-05,
      "model_forward_time": 1.728395700454712,
      "step": 38073
    },
    {
      "epoch": 3.630924224853516e-05,
      "step": 38073,
      "training_step_time": 9.668087720870972
    },
    {
      "epoch": 3.631019592285156e-05,
      "model_forward_time": 1.7162048816680908,
      "step": 38074
    },
    {
      "epoch": 3.631019592285156e-05,
      "step": 38074,
      "training_step_time": 8.553263902664185
    },
    {
      "epoch": 3.631114959716797e-05,
      "model_forward_time": 2.297470808029175,
      "step": 38075
    },
    {
      "epoch": 3.631114959716797e-05,
      "step": 38075,
      "training_step_time": 7.821188449859619
    },
    {
      "epoch": 3.6312103271484375e-05,
      "model_forward_time": 1.716796636581421,
      "step": 38076
    },
    {
      "epoch": 3.6312103271484375e-05,
      "step": 38076,
      "training_step_time": 7.810001611709595
    },
    {
      "epoch": 3.631305694580078e-05,
      "model_forward_time": 1.720346450805664,
      "step": 38077
    },
    {
      "epoch": 3.631305694580078e-05,
      "step": 38077,
      "training_step_time": 7.579882860183716
    },
    {
      "epoch": 3.631401062011719e-05,
      "model_forward_time": 1.7200860977172852,
      "step": 38078
    },
    {
      "epoch": 3.631401062011719e-05,
      "step": 38078,
      "training_step_time": 7.368898868560791
    },
    {
      "epoch": 3.6314964294433594e-05,
      "model_forward_time": 1.7197659015655518,
      "step": 38079
    },
    {
      "epoch": 3.6314964294433594e-05,
      "step": 38079,
      "training_step_time": 7.7854228019714355
    },
    {
      "epoch": 3.631591796875e-05,
      "grad_norm": 0.10641089826822281,
      "learning_rate": 7.295108686760921e-06,
      "loss": 0.0623,
      "step": 38080
    },
    {
      "epoch": 3.631591796875e-05,
      "model_forward_time": 1.7161791324615479,
      "step": 38080
    },
    {
      "dynamics_loss_avg": 0.059356463886797425,
      "epoch": 3.631591796875e-05,
      "step": 38080
    },
    {
      "action_loss_avg": 0.002316492889076471,
      "epoch": 3.631591796875e-05,
      "step": 38080
    },
    {
      "epoch": 3.631591796875e-05,
      "step": 38080,
      "training_step_time": 8.446147918701172
    },
    {
      "epoch": 3.631687164306641e-05,
      "model_forward_time": 1.7260167598724365,
      "step": 38081
    },
    {
      "epoch": 3.631687164306641e-05,
      "step": 38081,
      "training_step_time": 7.62748122215271
    },
    {
      "epoch": 3.631782531738281e-05,
      "model_forward_time": 1.7210443019866943,
      "step": 38082
    },
    {
      "epoch": 3.631782531738281e-05,
      "step": 38082,
      "training_step_time": 8.462424278259277
    },
    {
      "epoch": 3.631877899169922e-05,
      "model_forward_time": 1.7232849597930908,
      "step": 38083
    },
    {
      "epoch": 3.631877899169922e-05,
      "step": 38083,
      "training_step_time": 8.369848251342773
    },
    {
      "epoch": 3.631973266601563e-05,
      "model_forward_time": 2.293224811553955,
      "step": 38084
    },
    {
      "epoch": 3.631973266601563e-05,
      "step": 38084,
      "training_step_time": 7.635443449020386
    },
    {
      "epoch": 3.632068634033203e-05,
      "model_forward_time": 1.717864990234375,
      "step": 38085
    },
    {
      "epoch": 3.632068634033203e-05,
      "step": 38085,
      "training_step_time": 7.649800539016724
    },
    {
      "epoch": 3.6321640014648436e-05,
      "model_forward_time": 1.7174863815307617,
      "step": 38086
    },
    {
      "epoch": 3.6321640014648436e-05,
      "step": 38086,
      "training_step_time": 7.429856300354004
    },
    {
      "epoch": 3.632259368896485e-05,
      "model_forward_time": 1.7234864234924316,
      "step": 38087
    },
    {
      "epoch": 3.632259368896485e-05,
      "step": 38087,
      "training_step_time": 7.222042560577393
    },
    {
      "epoch": 3.632354736328125e-05,
      "model_forward_time": 1.7214455604553223,
      "step": 38088
    },
    {
      "epoch": 3.632354736328125e-05,
      "step": 38088,
      "training_step_time": 8.4957275390625
    },
    {
      "epoch": 3.6324501037597655e-05,
      "model_forward_time": 1.7200851440429688,
      "step": 38089
    },
    {
      "epoch": 3.6324501037597655e-05,
      "step": 38089,
      "training_step_time": 7.605434417724609
    },
    {
      "epoch": 3.6325454711914066e-05,
      "grad_norm": 0.11193602532148361,
      "learning_rate": 7.293639577310538e-06,
      "loss": 0.064,
      "step": 38090
    },
    {
      "epoch": 3.6325454711914066e-05,
      "model_forward_time": 1.7104761600494385,
      "step": 38090
    },
    {
      "dynamics_loss_avg": 0.053788098879158495,
      "epoch": 3.6325454711914066e-05,
      "step": 38090
    },
    {
      "action_loss_avg": 0.001869569782866165,
      "epoch": 3.6325454711914066e-05,
      "step": 38090
    },
    {
      "epoch": 3.6325454711914066e-05,
      "step": 38090,
      "training_step_time": 9.268853187561035
    },
    {
      "epoch": 3.632640838623047e-05,
      "model_forward_time": 1.7362680435180664,
      "step": 38091
    },
    {
      "epoch": 3.632640838623047e-05,
      "step": 38091,
      "training_step_time": 7.747436285018921
    },
    {
      "epoch": 3.6327362060546874e-05,
      "model_forward_time": 1.7384583950042725,
      "step": 38092
    },
    {
      "epoch": 3.6327362060546874e-05,
      "step": 38092,
      "training_step_time": 7.809707880020142
    },
    {
      "epoch": 3.6328315734863285e-05,
      "model_forward_time": 1.7142279148101807,
      "step": 38093
    },
    {
      "epoch": 3.6328315734863285e-05,
      "step": 38093,
      "training_step_time": 8.19356369972229
    },
    {
      "epoch": 3.632926940917969e-05,
      "model_forward_time": 1.7182390689849854,
      "step": 38094
    },
    {
      "epoch": 3.632926940917969e-05,
      "step": 38094,
      "training_step_time": 7.621642827987671
    },
    {
      "epoch": 3.633022308349609e-05,
      "model_forward_time": 1.34063720703125,
      "step": 38095
    },
    {
      "epoch": 3.633022308349609e-05,
      "step": 38095,
      "training_step_time": 7.366565465927124
    },
    {
      "epoch": 3.63311767578125e-05,
      "model_forward_time": 1.7194273471832275,
      "step": 38096
    },
    {
      "epoch": 3.63311767578125e-05,
      "step": 38096,
      "training_step_time": 7.56053352355957
    },
    {
      "epoch": 3.633213043212891e-05,
      "model_forward_time": 1.7206017971038818,
      "step": 38097
    },
    {
      "epoch": 3.633213043212891e-05,
      "step": 38097,
      "training_step_time": 6.770884037017822
    },
    {
      "epoch": 3.633308410644531e-05,
      "model_forward_time": 1.7189538478851318,
      "step": 38098
    },
    {
      "epoch": 3.633308410644531e-05,
      "step": 38098,
      "training_step_time": 7.599210500717163
    },
    {
      "epoch": 3.6334037780761716e-05,
      "model_forward_time": 1.6959564685821533,
      "step": 38099
    },
    {
      "epoch": 3.6334037780761716e-05,
      "step": 38099,
      "training_step_time": 8.408257007598877
    },
    {
      "epoch": 3.6334991455078126e-05,
      "grad_norm": 0.16082239151000977,
      "learning_rate": 7.292170217031179e-06,
      "loss": 0.0658,
      "step": 38100
    },
    {
      "epoch": 3.6334991455078126e-05,
      "model_forward_time": 1.7278931140899658,
      "step": 38100
    },
    {
      "dynamics_loss_avg": 0.05956612452864647,
      "epoch": 3.6334991455078126e-05,
      "step": 38100
    },
    {
      "action_loss_avg": 0.001993980084080249,
      "epoch": 3.6334991455078126e-05,
      "step": 38100
    },
    {
      "epoch": 3.6334991455078126e-05,
      "step": 38100,
      "training_step_time": 9.046438217163086
    },
    {
      "epoch": 3.633594512939453e-05,
      "model_forward_time": 1.7398223876953125,
      "step": 38101
    },
    {
      "epoch": 3.633594512939453e-05,
      "step": 38101,
      "training_step_time": 7.8366477489471436
    },
    {
      "epoch": 3.6336898803710934e-05,
      "model_forward_time": 1.7233541011810303,
      "step": 38102
    },
    {
      "epoch": 3.6336898803710934e-05,
      "step": 38102,
      "training_step_time": 7.7482757568359375
    },
    {
      "epoch": 3.6337852478027345e-05,
      "model_forward_time": 2.2703490257263184,
      "step": 38103
    },
    {
      "epoch": 3.6337852478027345e-05,
      "step": 38103,
      "training_step_time": 7.6086084842681885
    },
    {
      "epoch": 3.633880615234375e-05,
      "model_forward_time": 1.7131402492523193,
      "step": 38104
    },
    {
      "epoch": 3.633880615234375e-05,
      "step": 38104,
      "training_step_time": 7.413325309753418
    },
    {
      "epoch": 3.633975982666015e-05,
      "model_forward_time": 1.717559814453125,
      "step": 38105
    },
    {
      "epoch": 3.633975982666015e-05,
      "step": 38105,
      "training_step_time": 7.228408336639404
    },
    {
      "epoch": 3.6340713500976564e-05,
      "model_forward_time": 1.7217230796813965,
      "step": 38106
    },
    {
      "epoch": 3.6340713500976564e-05,
      "step": 38106,
      "training_step_time": 7.625024318695068
    },
    {
      "epoch": 3.634166717529297e-05,
      "model_forward_time": 1.716047763824463,
      "step": 38107
    },
    {
      "epoch": 3.634166717529297e-05,
      "step": 38107,
      "training_step_time": 8.484760761260986
    },
    {
      "epoch": 3.634262084960937e-05,
      "model_forward_time": 1.7124872207641602,
      "step": 38108
    },
    {
      "epoch": 3.634262084960937e-05,
      "step": 38108,
      "training_step_time": 7.582405090332031
    },
    {
      "epoch": 3.634357452392578e-05,
      "model_forward_time": 1.7349164485931396,
      "step": 38109
    },
    {
      "epoch": 3.634357452392578e-05,
      "step": 38109,
      "training_step_time": 9.420087575912476
    },
    {
      "epoch": 3.634452819824219e-05,
      "grad_norm": 0.09654734283685684,
      "learning_rate": 7.290700606083532e-06,
      "loss": 0.063,
      "step": 38110
    },
    {
      "epoch": 3.634452819824219e-05,
      "model_forward_time": 1.7278633117675781,
      "step": 38110
    },
    {
      "dynamics_loss_avg": 0.049845258332788946,
      "epoch": 3.634452819824219e-05,
      "step": 38110
    },
    {
      "action_loss_avg": 0.0021142145851626992,
      "epoch": 3.634452819824219e-05,
      "step": 38110
    },
    {
      "epoch": 3.634452819824219e-05,
      "step": 38110,
      "training_step_time": 7.916569948196411
    },
    {
      "epoch": 3.634548187255859e-05,
      "model_forward_time": 1.7221546173095703,
      "step": 38111
    },
    {
      "epoch": 3.634548187255859e-05,
      "step": 38111,
      "training_step_time": 9.595538139343262
    },
    {
      "epoch": 3.6346435546875e-05,
      "model_forward_time": 2.280341863632202,
      "step": 38112
    },
    {
      "epoch": 3.6346435546875e-05,
      "step": 38112,
      "training_step_time": 8.22098708152771
    },
    {
      "epoch": 3.6347389221191406e-05,
      "model_forward_time": 1.724937915802002,
      "step": 38113
    },
    {
      "epoch": 3.6347389221191406e-05,
      "step": 38113,
      "training_step_time": 6.822669982910156
    },
    {
      "epoch": 3.634834289550781e-05,
      "model_forward_time": 1.7219703197479248,
      "step": 38114
    },
    {
      "epoch": 3.634834289550781e-05,
      "step": 38114,
      "training_step_time": 8.449869394302368
    },
    {
      "epoch": 3.634929656982422e-05,
      "model_forward_time": 1.7203762531280518,
      "step": 38115
    },
    {
      "epoch": 3.634929656982422e-05,
      "step": 38115,
      "training_step_time": 7.637968301773071
    },
    {
      "epoch": 3.6350250244140625e-05,
      "model_forward_time": 1.7202365398406982,
      "step": 38116
    },
    {
      "epoch": 3.6350250244140625e-05,
      "step": 38116,
      "training_step_time": 8.556803703308105
    },
    {
      "epoch": 3.635120391845703e-05,
      "model_forward_time": 1.717149257659912,
      "step": 38117
    },
    {
      "epoch": 3.635120391845703e-05,
      "step": 38117,
      "training_step_time": 8.661559820175171
    },
    {
      "epoch": 3.635215759277344e-05,
      "model_forward_time": 1.7310752868652344,
      "step": 38118
    },
    {
      "epoch": 3.635215759277344e-05,
      "step": 38118,
      "training_step_time": 10.099791765213013
    },
    {
      "epoch": 3.6353111267089844e-05,
      "model_forward_time": 1.7178328037261963,
      "step": 38119
    },
    {
      "epoch": 3.6353111267089844e-05,
      "step": 38119,
      "training_step_time": 8.391433954238892
    },
    {
      "epoch": 3.635406494140625e-05,
      "grad_norm": 0.11328853666782379,
      "learning_rate": 7.2892307446283106e-06,
      "loss": 0.0599,
      "step": 38120
    },
    {
      "epoch": 3.635406494140625e-05,
      "model_forward_time": 1.697432279586792,
      "step": 38120
    },
    {
      "dynamics_loss_avg": 0.04703241009265184,
      "epoch": 3.635406494140625e-05,
      "step": 38120
    },
    {
      "action_loss_avg": 0.001169996615499258,
      "epoch": 3.635406494140625e-05,
      "step": 38120
    },
    {
      "epoch": 3.635406494140625e-05,
      "step": 38120,
      "training_step_time": 7.7266552448272705
    },
    {
      "epoch": 3.635501861572266e-05,
      "model_forward_time": 2.2810778617858887,
      "step": 38121
    },
    {
      "epoch": 3.635501861572266e-05,
      "step": 38121,
      "training_step_time": 7.685889720916748
    },
    {
      "epoch": 3.635597229003906e-05,
      "model_forward_time": 1.3307712078094482,
      "step": 38122
    },
    {
      "epoch": 3.635597229003906e-05,
      "step": 38122,
      "training_step_time": 7.558738470077515
    },
    {
      "epoch": 3.6356925964355466e-05,
      "model_forward_time": 1.7084991931915283,
      "step": 38123
    },
    {
      "epoch": 3.6356925964355466e-05,
      "step": 38123,
      "training_step_time": 7.5568647384643555
    },
    {
      "epoch": 3.635787963867188e-05,
      "model_forward_time": 1.7245628833770752,
      "step": 38124
    },
    {
      "epoch": 3.635787963867188e-05,
      "step": 38124,
      "training_step_time": 7.678232192993164
    },
    {
      "epoch": 3.635883331298828e-05,
      "model_forward_time": 1.7191870212554932,
      "step": 38125
    },
    {
      "epoch": 3.635883331298828e-05,
      "step": 38125,
      "training_step_time": 8.3058340549469
    },
    {
      "epoch": 3.6359786987304685e-05,
      "model_forward_time": 1.7164967060089111,
      "step": 38126
    },
    {
      "epoch": 3.6359786987304685e-05,
      "step": 38126,
      "training_step_time": 8.209643840789795
    },
    {
      "epoch": 3.6360740661621096e-05,
      "model_forward_time": 1.7242846488952637,
      "step": 38127
    },
    {
      "epoch": 3.6360740661621096e-05,
      "step": 38127,
      "training_step_time": 9.293066024780273
    },
    {
      "epoch": 3.63616943359375e-05,
      "model_forward_time": 1.729806661605835,
      "step": 38128
    },
    {
      "epoch": 3.63616943359375e-05,
      "step": 38128,
      "training_step_time": 7.848106384277344
    },
    {
      "epoch": 3.6362648010253904e-05,
      "model_forward_time": 1.700514793395996,
      "step": 38129
    },
    {
      "epoch": 3.6362648010253904e-05,
      "step": 38129,
      "training_step_time": 8.654959201812744
    },
    {
      "epoch": 3.6363601684570315e-05,
      "grad_norm": 0.15585097670555115,
      "learning_rate": 7.287760632826256e-06,
      "loss": 0.0599,
      "step": 38130
    },
    {
      "epoch": 3.6363601684570315e-05,
      "model_forward_time": 1.7063124179840088,
      "step": 38130
    },
    {
      "dynamics_loss_avg": 0.05840526688843965,
      "epoch": 3.6363601684570315e-05,
      "step": 38130
    },
    {
      "action_loss_avg": 0.0026982179027982057,
      "epoch": 3.6363601684570315e-05,
      "step": 38130
    },
    {
      "epoch": 3.6363601684570315e-05,
      "step": 38130,
      "training_step_time": 8.184123754501343
    },
    {
      "epoch": 3.636455535888672e-05,
      "model_forward_time": 1.7144088745117188,
      "step": 38131
    },
    {
      "epoch": 3.636455535888672e-05,
      "step": 38131,
      "training_step_time": 7.693331718444824
    },
    {
      "epoch": 3.636550903320312e-05,
      "model_forward_time": 1.7067296504974365,
      "step": 38132
    },
    {
      "epoch": 3.636550903320312e-05,
      "step": 38132,
      "training_step_time": 8.279088258743286
    },
    {
      "epoch": 3.6366462707519534e-05,
      "model_forward_time": 1.7057342529296875,
      "step": 38133
    },
    {
      "epoch": 3.6366462707519534e-05,
      "step": 38133,
      "training_step_time": 7.544418096542358
    },
    {
      "epoch": 3.636741638183594e-05,
      "model_forward_time": 1.7259776592254639,
      "step": 38134
    },
    {
      "epoch": 3.636741638183594e-05,
      "step": 38134,
      "training_step_time": 7.656670570373535
    },
    {
      "epoch": 3.636837005615234e-05,
      "model_forward_time": 1.7178711891174316,
      "step": 38135
    },
    {
      "epoch": 3.636837005615234e-05,
      "step": 38135,
      "training_step_time": 7.663676977157593
    },
    {
      "epoch": 3.636932373046875e-05,
      "model_forward_time": 1.7189767360687256,
      "step": 38136
    },
    {
      "epoch": 3.636932373046875e-05,
      "step": 38136,
      "training_step_time": 9.528823852539062
    },
    {
      "epoch": 3.637027740478516e-05,
      "model_forward_time": 1.753298282623291,
      "step": 38137
    },
    {
      "epoch": 3.637027740478516e-05,
      "step": 38137,
      "training_step_time": 9.367520332336426
    },
    {
      "epoch": 3.637123107910156e-05,
      "model_forward_time": 1.7146477699279785,
      "step": 38138
    },
    {
      "epoch": 3.637123107910156e-05,
      "step": 38138,
      "training_step_time": 8.571104049682617
    },
    {
      "epoch": 3.637218475341797e-05,
      "model_forward_time": 1.7137219905853271,
      "step": 38139
    },
    {
      "epoch": 3.637218475341797e-05,
      "step": 38139,
      "training_step_time": 7.669670104980469
    },
    {
      "epoch": 3.6373138427734376e-05,
      "grad_norm": 0.13015231490135193,
      "learning_rate": 7.286290270838139e-06,
      "loss": 0.0613,
      "step": 38140
    },
    {
      "epoch": 3.6373138427734376e-05,
      "model_forward_time": 2.2533063888549805,
      "step": 38140
    },
    {
      "dynamics_loss_avg": 0.07137291394174099,
      "epoch": 3.6373138427734376e-05,
      "step": 38140
    },
    {
      "action_loss_avg": 0.0025809435726841913,
      "epoch": 3.6373138427734376e-05,
      "step": 38140
    },
    {
      "epoch": 3.6373138427734376e-05,
      "step": 38140,
      "training_step_time": 7.695910453796387
    },
    {
      "epoch": 3.637409210205078e-05,
      "model_forward_time": 1.7058768272399902,
      "step": 38141
    },
    {
      "epoch": 3.637409210205078e-05,
      "step": 38141,
      "training_step_time": 7.184002637863159
    },
    {
      "epoch": 3.637504577636719e-05,
      "model_forward_time": 1.7086801528930664,
      "step": 38142
    },
    {
      "epoch": 3.637504577636719e-05,
      "step": 38142,
      "training_step_time": 7.574193239212036
    },
    {
      "epoch": 3.6375999450683594e-05,
      "model_forward_time": 1.70774245262146,
      "step": 38143
    },
    {
      "epoch": 3.6375999450683594e-05,
      "step": 38143,
      "training_step_time": 7.64258074760437
    },
    {
      "epoch": 3.6376953125e-05,
      "model_forward_time": 1.7094554901123047,
      "step": 38144
    },
    {
      "epoch": 3.6376953125e-05,
      "step": 38144,
      "training_step_time": 8.198444366455078
    },
    {
      "epoch": 3.637790679931641e-05,
      "model_forward_time": 1.7080070972442627,
      "step": 38145
    },
    {
      "epoch": 3.637790679931641e-05,
      "step": 38145,
      "training_step_time": 8.190491676330566
    },
    {
      "epoch": 3.637886047363281e-05,
      "model_forward_time": 1.7360713481903076,
      "step": 38146
    },
    {
      "epoch": 3.637886047363281e-05,
      "step": 38146,
      "training_step_time": 9.189061164855957
    },
    {
      "epoch": 3.637981414794922e-05,
      "model_forward_time": 1.7304725646972656,
      "step": 38147
    },
    {
      "epoch": 3.637981414794922e-05,
      "step": 38147,
      "training_step_time": 8.927382707595825
    },
    {
      "epoch": 3.638076782226563e-05,
      "model_forward_time": 1.7089831829071045,
      "step": 38148
    },
    {
      "epoch": 3.638076782226563e-05,
      "step": 38148,
      "training_step_time": 7.690258979797363
    },
    {
      "epoch": 3.638172149658203e-05,
      "model_forward_time": 2.265456438064575,
      "step": 38149
    },
    {
      "epoch": 3.638172149658203e-05,
      "step": 38149,
      "training_step_time": 7.656542062759399
    },
    {
      "epoch": 3.6382675170898436e-05,
      "grad_norm": 0.18606969714164734,
      "learning_rate": 7.284819658824755e-06,
      "loss": 0.0572,
      "step": 38150
    },
    {
      "epoch": 3.6382675170898436e-05,
      "model_forward_time": 1.7122454643249512,
      "step": 38150
    },
    {
      "dynamics_loss_avg": 0.05941671822220087,
      "epoch": 3.6382675170898436e-05,
      "step": 38150
    },
    {
      "action_loss_avg": 0.0021179414354264734,
      "epoch": 3.6382675170898436e-05,
      "step": 38150
    },
    {
      "epoch": 3.6382675170898436e-05,
      "step": 38150,
      "training_step_time": 7.5653746128082275
    },
    {
      "epoch": 3.638362884521485e-05,
      "model_forward_time": 1.716392993927002,
      "step": 38151
    },
    {
      "epoch": 3.638362884521485e-05,
      "step": 38151,
      "training_step_time": 7.608507394790649
    },
    {
      "epoch": 3.638458251953125e-05,
      "model_forward_time": 1.704793930053711,
      "step": 38152
    },
    {
      "epoch": 3.638458251953125e-05,
      "step": 38152,
      "training_step_time": 7.618226051330566
    },
    {
      "epoch": 3.6385536193847655e-05,
      "model_forward_time": 1.7074875831604004,
      "step": 38153
    },
    {
      "epoch": 3.6385536193847655e-05,
      "step": 38153,
      "training_step_time": 7.692356586456299
    },
    {
      "epoch": 3.6386489868164066e-05,
      "model_forward_time": 1.712064266204834,
      "step": 38154
    },
    {
      "epoch": 3.6386489868164066e-05,
      "step": 38154,
      "training_step_time": 8.476608514785767
    },
    {
      "epoch": 3.638744354248047e-05,
      "model_forward_time": 1.7325565814971924,
      "step": 38155
    },
    {
      "epoch": 3.638744354248047e-05,
      "step": 38155,
      "training_step_time": 7.693052530288696
    },
    {
      "epoch": 3.6388397216796874e-05,
      "model_forward_time": 1.7284202575683594,
      "step": 38156
    },
    {
      "epoch": 3.6388397216796874e-05,
      "step": 38156,
      "training_step_time": 9.679512739181519
    },
    {
      "epoch": 3.638935089111328e-05,
      "model_forward_time": 1.7104923725128174,
      "step": 38157
    },
    {
      "epoch": 3.638935089111328e-05,
      "step": 38157,
      "training_step_time": 8.686991930007935
    },
    {
      "epoch": 3.639030456542969e-05,
      "model_forward_time": 2.267038106918335,
      "step": 38158
    },
    {
      "epoch": 3.639030456542969e-05,
      "step": 38158,
      "training_step_time": 8.544057607650757
    },
    {
      "epoch": 3.639125823974609e-05,
      "model_forward_time": 1.7119965553283691,
      "step": 38159
    },
    {
      "epoch": 3.639125823974609e-05,
      "step": 38159,
      "training_step_time": 6.7719995975494385
    },
    {
      "epoch": 3.63922119140625e-05,
      "grad_norm": 0.1378117948770523,
      "learning_rate": 7.283348796946928e-06,
      "loss": 0.0593,
      "step": 38160
    },
    {
      "epoch": 3.63922119140625e-05,
      "model_forward_time": 1.7079830169677734,
      "step": 38160
    },
    {
      "dynamics_loss_avg": 0.06456068828701973,
      "epoch": 3.63922119140625e-05,
      "step": 38160
    },
    {
      "action_loss_avg": 0.0017069512454327197,
      "epoch": 3.63922119140625e-05,
      "step": 38160
    },
    {
      "epoch": 3.63922119140625e-05,
      "step": 38160,
      "training_step_time": 7.524099111557007
    },
    {
      "epoch": 3.639316558837891e-05,
      "model_forward_time": 1.7117211818695068,
      "step": 38161
    },
    {
      "epoch": 3.639316558837891e-05,
      "step": 38161,
      "training_step_time": 7.532328128814697
    },
    {
      "epoch": 3.639411926269531e-05,
      "model_forward_time": 1.7162361145019531,
      "step": 38162
    },
    {
      "epoch": 3.639411926269531e-05,
      "step": 38162,
      "training_step_time": 8.525145530700684
    },
    {
      "epoch": 3.6395072937011716e-05,
      "model_forward_time": 1.7072267532348633,
      "step": 38163
    },
    {
      "epoch": 3.6395072937011716e-05,
      "step": 38163,
      "training_step_time": 8.701475143432617
    },
    {
      "epoch": 3.6396026611328126e-05,
      "model_forward_time": 1.7172026634216309,
      "step": 38164
    },
    {
      "epoch": 3.6396026611328126e-05,
      "step": 38164,
      "training_step_time": 8.238083124160767
    },
    {
      "epoch": 3.639698028564453e-05,
      "model_forward_time": 1.7297742366790771,
      "step": 38165
    },
    {
      "epoch": 3.639698028564453e-05,
      "step": 38165,
      "training_step_time": 7.84849739074707
    },
    {
      "epoch": 3.6397933959960934e-05,
      "model_forward_time": 1.7143492698669434,
      "step": 38166
    },
    {
      "epoch": 3.6397933959960934e-05,
      "step": 38166,
      "training_step_time": 9.304324865341187
    },
    {
      "epoch": 3.6398887634277345e-05,
      "model_forward_time": 2.2749955654144287,
      "step": 38167
    },
    {
      "epoch": 3.6398887634277345e-05,
      "step": 38167,
      "training_step_time": 7.596298933029175
    },
    {
      "epoch": 3.639984130859375e-05,
      "model_forward_time": 1.7079904079437256,
      "step": 38168
    },
    {
      "epoch": 3.639984130859375e-05,
      "step": 38168,
      "training_step_time": 7.57771897315979
    },
    {
      "epoch": 3.640079498291015e-05,
      "model_forward_time": 1.711817741394043,
      "step": 38169
    },
    {
      "epoch": 3.640079498291015e-05,
      "step": 38169,
      "training_step_time": 7.481661081314087
    },
    {
      "epoch": 3.6401748657226564e-05,
      "grad_norm": 0.21039894223213196,
      "learning_rate": 7.281877685365511e-06,
      "loss": 0.0584,
      "step": 38170
    },
    {
      "epoch": 3.6401748657226564e-05,
      "model_forward_time": 1.7066457271575928,
      "step": 38170
    },
    {
      "dynamics_loss_avg": 0.05155856143683195,
      "epoch": 3.6401748657226564e-05,
      "step": 38170
    },
    {
      "action_loss_avg": 0.0021887454262468965,
      "epoch": 3.6401748657226564e-05,
      "step": 38170
    },
    {
      "epoch": 3.6401748657226564e-05,
      "step": 38170,
      "training_step_time": 7.454970359802246
    },
    {
      "epoch": 3.640270233154297e-05,
      "model_forward_time": 1.7198636531829834,
      "step": 38171
    },
    {
      "epoch": 3.640270233154297e-05,
      "step": 38171,
      "training_step_time": 8.580365419387817
    },
    {
      "epoch": 3.640365600585937e-05,
      "model_forward_time": 1.7030870914459229,
      "step": 38172
    },
    {
      "epoch": 3.640365600585937e-05,
      "step": 38172,
      "training_step_time": 8.161213397979736
    },
    {
      "epoch": 3.640460968017578e-05,
      "model_forward_time": 1.7105326652526855,
      "step": 38173
    },
    {
      "epoch": 3.640460968017578e-05,
      "step": 38173,
      "training_step_time": 8.602445125579834
    },
    {
      "epoch": 3.640556335449219e-05,
      "model_forward_time": 1.7499449253082275,
      "step": 38174
    },
    {
      "epoch": 3.640556335449219e-05,
      "step": 38174,
      "training_step_time": 7.758031368255615
    },
    {
      "epoch": 3.640651702880859e-05,
      "model_forward_time": 1.708343267440796,
      "step": 38175
    },
    {
      "epoch": 3.640651702880859e-05,
      "step": 38175,
      "training_step_time": 8.535854816436768
    },
    {
      "epoch": 3.6407470703125e-05,
      "model_forward_time": 2.2421376705169678,
      "step": 38176
    },
    {
      "epoch": 3.6407470703125e-05,
      "step": 38176,
      "training_step_time": 9.051145315170288
    },
    {
      "epoch": 3.6408424377441406e-05,
      "model_forward_time": 1.712360143661499,
      "step": 38177
    },
    {
      "epoch": 3.6408424377441406e-05,
      "step": 38177,
      "training_step_time": 7.601478338241577
    },
    {
      "epoch": 3.640937805175781e-05,
      "model_forward_time": 1.7171311378479004,
      "step": 38178
    },
    {
      "epoch": 3.640937805175781e-05,
      "step": 38178,
      "training_step_time": 8.526576519012451
    },
    {
      "epoch": 3.641033172607422e-05,
      "model_forward_time": 1.7137694358825684,
      "step": 38179
    },
    {
      "epoch": 3.641033172607422e-05,
      "step": 38179,
      "training_step_time": 7.054911136627197
    },
    {
      "epoch": 3.6411285400390625e-05,
      "grad_norm": 0.1970720738172531,
      "learning_rate": 7.280406324241379e-06,
      "loss": 0.0635,
      "step": 38180
    },
    {
      "epoch": 3.6411285400390625e-05,
      "model_forward_time": 1.7024967670440674,
      "step": 38180
    },
    {
      "dynamics_loss_avg": 0.0658376470208168,
      "epoch": 3.6411285400390625e-05,
      "step": 38180
    },
    {
      "action_loss_avg": 0.0035964875307399778,
      "epoch": 3.6411285400390625e-05,
      "step": 38180
    },
    {
      "epoch": 3.6411285400390625e-05,
      "step": 38180,
      "training_step_time": 7.582326173782349
    },
    {
      "epoch": 3.641223907470703e-05,
      "model_forward_time": 1.7135162353515625,
      "step": 38181
    },
    {
      "epoch": 3.641223907470703e-05,
      "step": 38181,
      "training_step_time": 7.6189844608306885
    },
    {
      "epoch": 3.641319274902344e-05,
      "model_forward_time": 1.7160866260528564,
      "step": 38182
    },
    {
      "epoch": 3.641319274902344e-05,
      "step": 38182,
      "training_step_time": 7.590733289718628
    },
    {
      "epoch": 3.6414146423339844e-05,
      "model_forward_time": 1.697021722793579,
      "step": 38183
    },
    {
      "epoch": 3.6414146423339844e-05,
      "step": 38183,
      "training_step_time": 7.715144634246826
    },
    {
      "epoch": 3.641510009765625e-05,
      "model_forward_time": 1.7117819786071777,
      "step": 38184
    },
    {
      "epoch": 3.641510009765625e-05,
      "step": 38184,
      "training_step_time": 7.79933500289917
    },
    {
      "epoch": 3.641605377197266e-05,
      "model_forward_time": 1.7081189155578613,
      "step": 38185
    },
    {
      "epoch": 3.641605377197266e-05,
      "step": 38185,
      "training_step_time": 8.522160530090332
    },
    {
      "epoch": 3.641700744628906e-05,
      "model_forward_time": 2.2543487548828125,
      "step": 38186
    },
    {
      "epoch": 3.641700744628906e-05,
      "step": 38186,
      "training_step_time": 7.544319152832031
    },
    {
      "epoch": 3.6417961120605466e-05,
      "model_forward_time": 1.714690923690796,
      "step": 38187
    },
    {
      "epoch": 3.6417961120605466e-05,
      "step": 38187,
      "training_step_time": 7.597293376922607
    },
    {
      "epoch": 3.641891479492188e-05,
      "model_forward_time": 1.7131867408752441,
      "step": 38188
    },
    {
      "epoch": 3.641891479492188e-05,
      "step": 38188,
      "training_step_time": 7.543636083602905
    },
    {
      "epoch": 3.641986846923828e-05,
      "model_forward_time": 1.7127876281738281,
      "step": 38189
    },
    {
      "epoch": 3.641986846923828e-05,
      "step": 38189,
      "training_step_time": 7.620313882827759
    },
    {
      "epoch": 3.6420822143554685e-05,
      "grad_norm": 0.15297368168830872,
      "learning_rate": 7.278934713735441e-06,
      "loss": 0.0583,
      "step": 38190
    },
    {
      "epoch": 3.6420822143554685e-05,
      "model_forward_time": 1.7112736701965332,
      "step": 38190
    },
    {
      "dynamics_loss_avg": 0.0439019488170743,
      "epoch": 3.6420822143554685e-05,
      "step": 38190
    },
    {
      "action_loss_avg": 0.0013156387605704368,
      "epoch": 3.6420822143554685e-05,
      "step": 38190
    },
    {
      "epoch": 3.6420822143554685e-05,
      "step": 38190,
      "training_step_time": 7.644108772277832
    },
    {
      "epoch": 3.6421775817871096e-05,
      "model_forward_time": 1.7130317687988281,
      "step": 38191
    },
    {
      "epoch": 3.6421775817871096e-05,
      "step": 38191,
      "training_step_time": 8.598242282867432
    },
    {
      "epoch": 3.64227294921875e-05,
      "model_forward_time": 1.7064404487609863,
      "step": 38192
    },
    {
      "epoch": 3.64227294921875e-05,
      "step": 38192,
      "training_step_time": 7.68496298789978
    },
    {
      "epoch": 3.6423683166503904e-05,
      "model_forward_time": 1.7299418449401855,
      "step": 38193
    },
    {
      "epoch": 3.6423683166503904e-05,
      "step": 38193,
      "training_step_time": 8.71395230293274
    },
    {
      "epoch": 3.6424636840820315e-05,
      "model_forward_time": 1.715890645980835,
      "step": 38194
    },
    {
      "epoch": 3.6424636840820315e-05,
      "step": 38194,
      "training_step_time": 7.612456321716309
    },
    {
      "epoch": 3.642559051513672e-05,
      "model_forward_time": 2.2618775367736816,
      "step": 38195
    },
    {
      "epoch": 3.642559051513672e-05,
      "step": 38195,
      "training_step_time": 7.4908812046051025
    },
    {
      "epoch": 3.642654418945312e-05,
      "model_forward_time": 1.7136025428771973,
      "step": 38196
    },
    {
      "epoch": 3.642654418945312e-05,
      "step": 38196,
      "training_step_time": 7.611120223999023
    },
    {
      "epoch": 3.6427497863769534e-05,
      "model_forward_time": 1.7126107215881348,
      "step": 38197
    },
    {
      "epoch": 3.6427497863769534e-05,
      "step": 38197,
      "training_step_time": 7.544605255126953
    },
    {
      "epoch": 3.642845153808594e-05,
      "model_forward_time": 1.7123868465423584,
      "step": 38198
    },
    {
      "epoch": 3.642845153808594e-05,
      "step": 38198,
      "training_step_time": 7.559094667434692
    },
    {
      "epoch": 3.642940521240234e-05,
      "model_forward_time": 1.7169198989868164,
      "step": 38199
    },
    {
      "epoch": 3.642940521240234e-05,
      "step": 38199,
      "training_step_time": 7.603774547576904
    },
    {
      "epoch": 3.643035888671875e-05,
      "grad_norm": 0.14800629019737244,
      "learning_rate": 7.277462854008629e-06,
      "loss": 0.0591,
      "step": 38200
    },
    {
      "epoch": 3.643035888671875e-05,
      "model_forward_time": 1.7270472049713135,
      "step": 38200
    },
    {
      "dynamics_loss_avg": 0.052523167245090006,
      "epoch": 3.643035888671875e-05,
      "step": 38200
    },
    {
      "action_loss_avg": 0.0020178941311314704,
      "epoch": 3.643035888671875e-05,
      "step": 38200
    },
    {
      "epoch": 3.643035888671875e-05,
      "step": 38200,
      "training_step_time": 8.14474892616272
    },
    {
      "epoch": 3.643131256103516e-05,
      "model_forward_time": 1.7122535705566406,
      "step": 38201
    },
    {
      "epoch": 3.643131256103516e-05,
      "step": 38201,
      "training_step_time": 8.408969402313232
    },
    {
      "epoch": 3.643226623535156e-05,
      "model_forward_time": 1.727095365524292,
      "step": 38202
    },
    {
      "epoch": 3.643226623535156e-05,
      "step": 38202,
      "training_step_time": 8.527583360671997
    },
    {
      "epoch": 3.643321990966797e-05,
      "model_forward_time": 1.706906795501709,
      "step": 38203
    },
    {
      "epoch": 3.643321990966797e-05,
      "step": 38203,
      "training_step_time": 8.414824724197388
    },
    {
      "epoch": 3.6434173583984376e-05,
      "model_forward_time": 2.2735719680786133,
      "step": 38204
    },
    {
      "epoch": 3.6434173583984376e-05,
      "step": 38204,
      "training_step_time": 7.477871417999268
    },
    {
      "epoch": 3.643512725830078e-05,
      "model_forward_time": 1.712902545928955,
      "step": 38205
    },
    {
      "epoch": 3.643512725830078e-05,
      "step": 38205,
      "training_step_time": 7.6347973346710205
    },
    {
      "epoch": 3.643608093261719e-05,
      "model_forward_time": 1.7082695960998535,
      "step": 38206
    },
    {
      "epoch": 3.643608093261719e-05,
      "step": 38206,
      "training_step_time": 7.307308197021484
    },
    {
      "epoch": 3.6437034606933594e-05,
      "model_forward_time": 1.7063446044921875,
      "step": 38207
    },
    {
      "epoch": 3.6437034606933594e-05,
      "step": 38207,
      "training_step_time": 7.597686052322388
    },
    {
      "epoch": 3.643798828125e-05,
      "model_forward_time": 1.7114267349243164,
      "step": 38208
    },
    {
      "epoch": 3.643798828125e-05,
      "step": 38208,
      "training_step_time": 8.125299453735352
    },
    {
      "epoch": 3.643894195556641e-05,
      "model_forward_time": 1.7255916595458984,
      "step": 38209
    },
    {
      "epoch": 3.643894195556641e-05,
      "step": 38209,
      "training_step_time": 7.567280530929565
    },
    {
      "epoch": 3.643989562988281e-05,
      "grad_norm": 0.15359389781951904,
      "learning_rate": 7.275990745221903e-06,
      "loss": 0.0577,
      "step": 38210
    },
    {
      "epoch": 3.643989562988281e-05,
      "model_forward_time": 1.7064802646636963,
      "step": 38210
    },
    {
      "dynamics_loss_avg": 0.05400908701121807,
      "epoch": 3.643989562988281e-05,
      "step": 38210
    },
    {
      "action_loss_avg": 0.0012848346552345902,
      "epoch": 3.643989562988281e-05,
      "step": 38210
    },
    {
      "epoch": 3.643989562988281e-05,
      "step": 38210,
      "training_step_time": 9.210084199905396
    },
    {
      "epoch": 3.644084930419922e-05,
      "model_forward_time": 1.722228765487671,
      "step": 38211
    },
    {
      "epoch": 3.644084930419922e-05,
      "step": 38211,
      "training_step_time": 8.794044494628906
    },
    {
      "epoch": 3.644180297851563e-05,
      "model_forward_time": 1.7090282440185547,
      "step": 38212
    },
    {
      "epoch": 3.644180297851563e-05,
      "step": 38212,
      "training_step_time": 7.62783670425415
    },
    {
      "epoch": 3.644275665283203e-05,
      "model_forward_time": 2.2931087017059326,
      "step": 38213
    },
    {
      "epoch": 3.644275665283203e-05,
      "step": 38213,
      "training_step_time": 7.601181745529175
    },
    {
      "epoch": 3.6443710327148436e-05,
      "model_forward_time": 1.7099881172180176,
      "step": 38214
    },
    {
      "epoch": 3.6443710327148436e-05,
      "step": 38214,
      "training_step_time": 7.637863874435425
    },
    {
      "epoch": 3.644466400146485e-05,
      "model_forward_time": 1.696976661682129,
      "step": 38215
    },
    {
      "epoch": 3.644466400146485e-05,
      "step": 38215,
      "training_step_time": 7.523339033126831
    },
    {
      "epoch": 3.644561767578125e-05,
      "model_forward_time": 1.7156028747558594,
      "step": 38216
    },
    {
      "epoch": 3.644561767578125e-05,
      "step": 38216,
      "training_step_time": 7.583929538726807
    },
    {
      "epoch": 3.6446571350097655e-05,
      "model_forward_time": 1.7136287689208984,
      "step": 38217
    },
    {
      "epoch": 3.6446571350097655e-05,
      "step": 38217,
      "training_step_time": 7.452088356018066
    },
    {
      "epoch": 3.6447525024414066e-05,
      "model_forward_time": 1.7302124500274658,
      "step": 38218
    },
    {
      "epoch": 3.6447525024414066e-05,
      "step": 38218,
      "training_step_time": 7.662521600723267
    },
    {
      "epoch": 3.644847869873047e-05,
      "model_forward_time": 1.7112979888916016,
      "step": 38219
    },
    {
      "epoch": 3.644847869873047e-05,
      "step": 38219,
      "training_step_time": 7.682028532028198
    },
    {
      "epoch": 3.6449432373046874e-05,
      "grad_norm": 0.13669683039188385,
      "learning_rate": 7.274518387536253e-06,
      "loss": 0.0572,
      "step": 38220
    },
    {
      "epoch": 3.6449432373046874e-05,
      "model_forward_time": 1.7120964527130127,
      "step": 38220
    },
    {
      "dynamics_loss_avg": 0.0488943874835968,
      "epoch": 3.6449432373046874e-05,
      "step": 38220
    },
    {
      "action_loss_avg": 0.0015281877538654953,
      "epoch": 3.6449432373046874e-05,
      "step": 38220
    },
    {
      "epoch": 3.6449432373046874e-05,
      "step": 38220,
      "training_step_time": 8.333519220352173
    },
    {
      "epoch": 3.645038604736328e-05,
      "model_forward_time": 1.7100067138671875,
      "step": 38221
    },
    {
      "epoch": 3.645038604736328e-05,
      "step": 38221,
      "training_step_time": 8.939006328582764
    },
    {
      "epoch": 3.645133972167969e-05,
      "model_forward_time": 2.2600200176239014,
      "step": 38222
    },
    {
      "epoch": 3.645133972167969e-05,
      "step": 38222,
      "training_step_time": 10.306079149246216
    },
    {
      "epoch": 3.645229339599609e-05,
      "model_forward_time": 1.7095425128936768,
      "step": 38223
    },
    {
      "epoch": 3.645229339599609e-05,
      "step": 38223,
      "training_step_time": 7.672870397567749
    },
    {
      "epoch": 3.64532470703125e-05,
      "model_forward_time": 1.7135818004608154,
      "step": 38224
    },
    {
      "epoch": 3.64532470703125e-05,
      "step": 38224,
      "training_step_time": 7.5927605628967285
    },
    {
      "epoch": 3.645420074462891e-05,
      "model_forward_time": 1.6920289993286133,
      "step": 38225
    },
    {
      "epoch": 3.645420074462891e-05,
      "step": 38225,
      "training_step_time": 7.621831178665161
    },
    {
      "epoch": 3.645515441894531e-05,
      "model_forward_time": 1.7087929248809814,
      "step": 38226
    },
    {
      "epoch": 3.645515441894531e-05,
      "step": 38226,
      "training_step_time": 7.556377410888672
    },
    {
      "epoch": 3.6456108093261716e-05,
      "model_forward_time": 1.7078471183776855,
      "step": 38227
    },
    {
      "epoch": 3.6456108093261716e-05,
      "step": 38227,
      "training_step_time": 8.533033609390259
    },
    {
      "epoch": 3.6457061767578126e-05,
      "model_forward_time": 1.7051634788513184,
      "step": 38228
    },
    {
      "epoch": 3.6457061767578126e-05,
      "step": 38228,
      "training_step_time": 8.284151315689087
    },
    {
      "epoch": 3.645801544189453e-05,
      "model_forward_time": 1.7175962924957275,
      "step": 38229
    },
    {
      "epoch": 3.645801544189453e-05,
      "step": 38229,
      "training_step_time": 8.74453854560852
    },
    {
      "epoch": 3.6458969116210935e-05,
      "grad_norm": 0.15978363156318665,
      "learning_rate": 7.273045781112689e-06,
      "loss": 0.0567,
      "step": 38230
    },
    {
      "epoch": 3.6458969116210935e-05,
      "model_forward_time": 1.688549280166626,
      "step": 38230
    },
    {
      "dynamics_loss_avg": 0.04741539862006903,
      "epoch": 3.6458969116210935e-05,
      "step": 38230
    },
    {
      "action_loss_avg": 0.0018831211084034294,
      "epoch": 3.6458969116210935e-05,
      "step": 38230
    },
    {
      "epoch": 3.6458969116210935e-05,
      "step": 38230,
      "training_step_time": 8.534915447235107
    },
    {
      "epoch": 3.6459922790527345e-05,
      "model_forward_time": 1.7020385265350342,
      "step": 38231
    },
    {
      "epoch": 3.6459922790527345e-05,
      "step": 38231,
      "training_step_time": 7.648462295532227
    },
    {
      "epoch": 3.646087646484375e-05,
      "model_forward_time": 2.249756336212158,
      "step": 38232
    },
    {
      "epoch": 3.646087646484375e-05,
      "step": 38232,
      "training_step_time": 8.5121009349823
    },
    {
      "epoch": 3.646183013916015e-05,
      "model_forward_time": 1.7053203582763672,
      "step": 38233
    },
    {
      "epoch": 3.646183013916015e-05,
      "step": 38233,
      "training_step_time": 7.180575847625732
    },
    {
      "epoch": 3.6462783813476564e-05,
      "model_forward_time": 1.7032289505004883,
      "step": 38234
    },
    {
      "epoch": 3.6462783813476564e-05,
      "step": 38234,
      "training_step_time": 7.598190784454346
    },
    {
      "epoch": 3.646373748779297e-05,
      "model_forward_time": 1.702211856842041,
      "step": 38235
    },
    {
      "epoch": 3.646373748779297e-05,
      "step": 38235,
      "training_step_time": 8.405581712722778
    },
    {
      "epoch": 3.646469116210937e-05,
      "model_forward_time": 1.7059979438781738,
      "step": 38236
    },
    {
      "epoch": 3.646469116210937e-05,
      "step": 38236,
      "training_step_time": 8.89646053314209
    },
    {
      "epoch": 3.646564483642578e-05,
      "model_forward_time": 1.7172877788543701,
      "step": 38237
    },
    {
      "epoch": 3.646564483642578e-05,
      "step": 38237,
      "training_step_time": 8.461410522460938
    },
    {
      "epoch": 3.646659851074219e-05,
      "model_forward_time": 1.7239296436309814,
      "step": 38238
    },
    {
      "epoch": 3.646659851074219e-05,
      "step": 38238,
      "training_step_time": 8.975126504898071
    },
    {
      "epoch": 3.646755218505859e-05,
      "model_forward_time": 1.7049157619476318,
      "step": 38239
    },
    {
      "epoch": 3.646755218505859e-05,
      "step": 38239,
      "training_step_time": 9.61387038230896
    },
    {
      "epoch": 3.6468505859375e-05,
      "grad_norm": 0.1145688146352768,
      "learning_rate": 7.271572926112257e-06,
      "loss": 0.0545,
      "step": 38240
    },
    {
      "epoch": 3.6468505859375e-05,
      "model_forward_time": 1.70188307762146,
      "step": 38240
    },
    {
      "dynamics_loss_avg": 0.054502864740788935,
      "epoch": 3.6468505859375e-05,
      "step": 38240
    },
    {
      "action_loss_avg": 0.0017822501016780735,
      "epoch": 3.6468505859375e-05,
      "step": 38240
    },
    {
      "epoch": 3.6468505859375e-05,
      "step": 38240,
      "training_step_time": 7.5701868534088135
    },
    {
      "epoch": 3.6469459533691406e-05,
      "model_forward_time": 2.253291368484497,
      "step": 38241
    },
    {
      "epoch": 3.6469459533691406e-05,
      "step": 38241,
      "training_step_time": 8.318003416061401
    },
    {
      "epoch": 3.647041320800781e-05,
      "model_forward_time": 1.7025275230407715,
      "step": 38242
    },
    {
      "epoch": 3.647041320800781e-05,
      "step": 38242,
      "training_step_time": 7.5697808265686035
    },
    {
      "epoch": 3.647136688232422e-05,
      "model_forward_time": 1.7028553485870361,
      "step": 38243
    },
    {
      "epoch": 3.647136688232422e-05,
      "step": 38243,
      "training_step_time": 7.504976987838745
    },
    {
      "epoch": 3.6472320556640625e-05,
      "model_forward_time": 1.7059776782989502,
      "step": 38244
    },
    {
      "epoch": 3.6472320556640625e-05,
      "step": 38244,
      "training_step_time": 8.205572605133057
    },
    {
      "epoch": 3.647327423095703e-05,
      "model_forward_time": 1.7131555080413818,
      "step": 38245
    },
    {
      "epoch": 3.647327423095703e-05,
      "step": 38245,
      "training_step_time": 7.566080570220947
    },
    {
      "epoch": 3.647422790527344e-05,
      "model_forward_time": 1.7180187702178955,
      "step": 38246
    },
    {
      "epoch": 3.647422790527344e-05,
      "step": 38246,
      "training_step_time": 7.622411012649536
    },
    {
      "epoch": 3.6475181579589844e-05,
      "model_forward_time": 1.7300126552581787,
      "step": 38247
    },
    {
      "epoch": 3.6475181579589844e-05,
      "step": 38247,
      "training_step_time": 10.330100774765015
    },
    {
      "epoch": 3.647613525390625e-05,
      "model_forward_time": 0.9938597679138184,
      "step": 38248
    },
    {
      "epoch": 3.647613525390625e-05,
      "step": 38248,
      "training_step_time": 9.49356722831726
    },
    {
      "epoch": 3.647708892822266e-05,
      "model_forward_time": 1.7044923305511475,
      "step": 38249
    },
    {
      "epoch": 3.647708892822266e-05,
      "step": 38249,
      "training_step_time": 7.582509756088257
    },
    {
      "epoch": 3.647804260253906e-05,
      "grad_norm": 0.12145776301622391,
      "learning_rate": 7.270099822696024e-06,
      "loss": 0.0574,
      "step": 38250
    },
    {
      "epoch": 3.647804260253906e-05,
      "model_forward_time": 2.2731988430023193,
      "step": 38250
    },
    {
      "dynamics_loss_avg": 0.06002777386456728,
      "epoch": 3.647804260253906e-05,
      "step": 38250
    },
    {
      "action_loss_avg": 0.0015849544084630906,
      "epoch": 3.647804260253906e-05,
      "step": 38250
    },
    {
      "epoch": 3.647804260253906e-05,
      "step": 38250,
      "training_step_time": 7.581157684326172
    },
    {
      "epoch": 3.6478996276855467e-05,
      "model_forward_time": 1.714996576309204,
      "step": 38251
    },
    {
      "epoch": 3.6478996276855467e-05,
      "step": 38251,
      "training_step_time": 7.578778266906738
    },
    {
      "epoch": 3.647994995117188e-05,
      "model_forward_time": 1.708745002746582,
      "step": 38252
    },
    {
      "epoch": 3.647994995117188e-05,
      "step": 38252,
      "training_step_time": 8.593643426895142
    },
    {
      "epoch": 3.648090362548828e-05,
      "model_forward_time": 1.7007732391357422,
      "step": 38253
    },
    {
      "epoch": 3.648090362548828e-05,
      "step": 38253,
      "training_step_time": 7.9140119552612305
    },
    {
      "epoch": 3.6481857299804685e-05,
      "model_forward_time": 1.712388277053833,
      "step": 38254
    },
    {
      "epoch": 3.6481857299804685e-05,
      "step": 38254,
      "training_step_time": 8.601925134658813
    },
    {
      "epoch": 3.6482810974121096e-05,
      "model_forward_time": 1.7241833209991455,
      "step": 38255
    },
    {
      "epoch": 3.6482810974121096e-05,
      "step": 38255,
      "training_step_time": 8.591615438461304
    },
    {
      "epoch": 3.64837646484375e-05,
      "model_forward_time": 1.7294576168060303,
      "step": 38256
    },
    {
      "epoch": 3.64837646484375e-05,
      "step": 38256,
      "training_step_time": 8.38286280632019
    },
    {
      "epoch": 3.6484718322753904e-05,
      "model_forward_time": 1.708784580230713,
      "step": 38257
    },
    {
      "epoch": 3.6484718322753904e-05,
      "step": 38257,
      "training_step_time": 9.299829006195068
    },
    {
      "epoch": 3.6485671997070315e-05,
      "model_forward_time": 1.7035226821899414,
      "step": 38258
    },
    {
      "epoch": 3.6485671997070315e-05,
      "step": 38258,
      "training_step_time": 7.59431529045105
    },
    {
      "epoch": 3.648662567138672e-05,
      "model_forward_time": 2.259289264678955,
      "step": 38259
    },
    {
      "epoch": 3.648662567138672e-05,
      "step": 38259,
      "training_step_time": 8.676587581634521
    },
    {
      "epoch": 3.648757934570312e-05,
      "grad_norm": 0.1402369737625122,
      "learning_rate": 7.26862647102509e-06,
      "loss": 0.0582,
      "step": 38260
    },
    {
      "epoch": 3.648757934570312e-05,
      "model_forward_time": 1.70479416847229,
      "step": 38260
    },
    {
      "dynamics_loss_avg": 0.05271211341023445,
      "epoch": 3.648757934570312e-05,
      "step": 38260
    },
    {
      "action_loss_avg": 0.0012394776043947787,
      "epoch": 3.648757934570312e-05,
      "step": 38260
    },
    {
      "epoch": 3.648757934570312e-05,
      "step": 38260,
      "training_step_time": 7.526180982589722
    },
    {
      "epoch": 3.6488533020019534e-05,
      "model_forward_time": 1.7116246223449707,
      "step": 38261
    },
    {
      "epoch": 3.6488533020019534e-05,
      "step": 38261,
      "training_step_time": 7.549068450927734
    },
    {
      "epoch": 3.648948669433594e-05,
      "model_forward_time": 1.7027850151062012,
      "step": 38262
    },
    {
      "epoch": 3.648948669433594e-05,
      "step": 38262,
      "training_step_time": 7.609102487564087
    },
    {
      "epoch": 3.649044036865234e-05,
      "model_forward_time": 1.7071313858032227,
      "step": 38263
    },
    {
      "epoch": 3.649044036865234e-05,
      "step": 38263,
      "training_step_time": 7.629222393035889
    },
    {
      "epoch": 3.649139404296875e-05,
      "model_forward_time": 1.7063899040222168,
      "step": 38264
    },
    {
      "epoch": 3.649139404296875e-05,
      "step": 38264,
      "training_step_time": 8.774436473846436
    },
    {
      "epoch": 3.649234771728516e-05,
      "model_forward_time": 1.7687487602233887,
      "step": 38265
    },
    {
      "epoch": 3.649234771728516e-05,
      "step": 38265,
      "training_step_time": 8.717221736907959
    },
    {
      "epoch": 3.649330139160156e-05,
      "model_forward_time": 1.706967830657959,
      "step": 38266
    },
    {
      "epoch": 3.649330139160156e-05,
      "step": 38266,
      "training_step_time": 7.880936145782471
    },
    {
      "epoch": 3.649425506591797e-05,
      "model_forward_time": 1.709160566329956,
      "step": 38267
    },
    {
      "epoch": 3.649425506591797e-05,
      "step": 38267,
      "training_step_time": 7.5923309326171875
    },
    {
      "epoch": 3.6495208740234376e-05,
      "model_forward_time": 1.707143783569336,
      "step": 38268
    },
    {
      "epoch": 3.6495208740234376e-05,
      "step": 38268,
      "training_step_time": 9.05422592163086
    },
    {
      "epoch": 3.649616241455078e-05,
      "model_forward_time": 1.7131645679473877,
      "step": 38269
    },
    {
      "epoch": 3.649616241455078e-05,
      "step": 38269,
      "training_step_time": 7.6470746994018555
    },
    {
      "epoch": 3.649711608886719e-05,
      "grad_norm": 0.17008885741233826,
      "learning_rate": 7.267152871260573e-06,
      "loss": 0.0611,
      "step": 38270
    },
    {
      "epoch": 3.649711608886719e-05,
      "model_forward_time": 1.7090380191802979,
      "step": 38270
    },
    {
      "dynamics_loss_avg": 0.05836894512176514,
      "epoch": 3.649711608886719e-05,
      "step": 38270
    },
    {
      "action_loss_avg": 0.002919563715113327,
      "epoch": 3.649711608886719e-05,
      "step": 38270
    },
    {
      "epoch": 3.649711608886719e-05,
      "step": 38270,
      "training_step_time": 7.626778841018677
    },
    {
      "epoch": 3.6498069763183595e-05,
      "model_forward_time": 1.706995964050293,
      "step": 38271
    },
    {
      "epoch": 3.6498069763183595e-05,
      "step": 38271,
      "training_step_time": 7.616296291351318
    },
    {
      "epoch": 3.64990234375e-05,
      "model_forward_time": 1.7075915336608887,
      "step": 38272
    },
    {
      "epoch": 3.64990234375e-05,
      "step": 38272,
      "training_step_time": 8.349237203598022
    },
    {
      "epoch": 3.649997711181641e-05,
      "model_forward_time": 1.7168865203857422,
      "step": 38273
    },
    {
      "epoch": 3.649997711181641e-05,
      "step": 38273,
      "training_step_time": 7.712964057922363
    },
    {
      "epoch": 3.650093078613281e-05,
      "model_forward_time": 1.7542376518249512,
      "step": 38274
    },
    {
      "epoch": 3.650093078613281e-05,
      "step": 38274,
      "training_step_time": 8.702832460403442
    },
    {
      "epoch": 3.650188446044922e-05,
      "model_forward_time": 1.7076103687286377,
      "step": 38275
    },
    {
      "epoch": 3.650188446044922e-05,
      "step": 38275,
      "training_step_time": 7.753870010375977
    },
    {
      "epoch": 3.650283813476563e-05,
      "model_forward_time": 1.7154924869537354,
      "step": 38276
    },
    {
      "epoch": 3.650283813476563e-05,
      "step": 38276,
      "training_step_time": 7.553384780883789
    },
    {
      "epoch": 3.650379180908203e-05,
      "model_forward_time": 1.7087655067443848,
      "step": 38277
    },
    {
      "epoch": 3.650379180908203e-05,
      "step": 38277,
      "training_step_time": 7.612006425857544
    },
    {
      "epoch": 3.6504745483398436e-05,
      "model_forward_time": 2.260375738143921,
      "step": 38278
    },
    {
      "epoch": 3.6504745483398436e-05,
      "step": 38278,
      "training_step_time": 7.593952178955078
    },
    {
      "epoch": 3.650569915771485e-05,
      "model_forward_time": 1.7125024795532227,
      "step": 38279
    },
    {
      "epoch": 3.650569915771485e-05,
      "step": 38279,
      "training_step_time": 7.641977548599243
    },
    {
      "epoch": 3.650665283203125e-05,
      "grad_norm": 0.13108289241790771,
      "learning_rate": 7.265679023563629e-06,
      "loss": 0.0613,
      "step": 38280
    },
    {
      "epoch": 3.650665283203125e-05,
      "model_forward_time": 1.709813117980957,
      "step": 38280
    },
    {
      "dynamics_loss_avg": 0.062287920899689195,
      "epoch": 3.650665283203125e-05,
      "step": 38280
    },
    {
      "action_loss_avg": 0.002127229771576822,
      "epoch": 3.650665283203125e-05,
      "step": 38280
    },
    {
      "epoch": 3.650665283203125e-05,
      "step": 38280,
      "training_step_time": 7.529674768447876
    },
    {
      "epoch": 3.6507606506347655e-05,
      "model_forward_time": 1.713066816329956,
      "step": 38281
    },
    {
      "epoch": 3.6507606506347655e-05,
      "step": 38281,
      "training_step_time": 7.5541486740112305
    },
    {
      "epoch": 3.650856018066406e-05,
      "model_forward_time": 1.7111694812774658,
      "step": 38282
    },
    {
      "epoch": 3.650856018066406e-05,
      "step": 38282,
      "training_step_time": 8.954932928085327
    },
    {
      "epoch": 3.650951385498047e-05,
      "model_forward_time": 1.7559363842010498,
      "step": 38283
    },
    {
      "epoch": 3.650951385498047e-05,
      "step": 38283,
      "training_step_time": 9.672969102859497
    },
    {
      "epoch": 3.6510467529296874e-05,
      "model_forward_time": 1.7091388702392578,
      "step": 38284
    },
    {
      "epoch": 3.6510467529296874e-05,
      "step": 38284,
      "training_step_time": 8.366579532623291
    },
    {
      "epoch": 3.651142120361328e-05,
      "model_forward_time": 1.7109472751617432,
      "step": 38285
    },
    {
      "epoch": 3.651142120361328e-05,
      "step": 38285,
      "training_step_time": 8.660024642944336
    },
    {
      "epoch": 3.651237487792969e-05,
      "model_forward_time": 1.715503215789795,
      "step": 38286
    },
    {
      "epoch": 3.651237487792969e-05,
      "step": 38286,
      "training_step_time": 8.264341592788696
    },
    {
      "epoch": 3.651332855224609e-05,
      "model_forward_time": 2.259444236755371,
      "step": 38287
    },
    {
      "epoch": 3.651332855224609e-05,
      "step": 38287,
      "training_step_time": 7.640176296234131
    },
    {
      "epoch": 3.65142822265625e-05,
      "model_forward_time": 1.7117445468902588,
      "step": 38288
    },
    {
      "epoch": 3.65142822265625e-05,
      "step": 38288,
      "training_step_time": 7.652638673782349
    },
    {
      "epoch": 3.651523590087891e-05,
      "model_forward_time": 1.710433006286621,
      "step": 38289
    },
    {
      "epoch": 3.651523590087891e-05,
      "step": 38289,
      "training_step_time": 7.559075355529785
    },
    {
      "epoch": 3.651618957519531e-05,
      "grad_norm": 0.11077515780925751,
      "learning_rate": 7.264204928095433e-06,
      "loss": 0.0577,
      "step": 38290
    },
    {
      "epoch": 3.651618957519531e-05,
      "model_forward_time": 1.7109272480010986,
      "step": 38290
    },
    {
      "dynamics_loss_avg": 0.05295801777392626,
      "epoch": 3.651618957519531e-05,
      "step": 38290
    },
    {
      "action_loss_avg": 0.0021830725425388666,
      "epoch": 3.651618957519531e-05,
      "step": 38290
    },
    {
      "epoch": 3.651618957519531e-05,
      "step": 38290,
      "training_step_time": 8.282137632369995
    },
    {
      "epoch": 3.6517143249511716e-05,
      "model_forward_time": 1.7101683616638184,
      "step": 38291
    },
    {
      "epoch": 3.6517143249511716e-05,
      "step": 38291,
      "training_step_time": 7.555676460266113
    },
    {
      "epoch": 3.6518096923828127e-05,
      "model_forward_time": 1.7256309986114502,
      "step": 38292
    },
    {
      "epoch": 3.6518096923828127e-05,
      "step": 38292,
      "training_step_time": 9.587735176086426
    },
    {
      "epoch": 3.651905059814453e-05,
      "model_forward_time": 1.7079784870147705,
      "step": 38293
    },
    {
      "epoch": 3.651905059814453e-05,
      "step": 38293,
      "training_step_time": 8.344968557357788
    },
    {
      "epoch": 3.6520004272460935e-05,
      "model_forward_time": 1.709578514099121,
      "step": 38294
    },
    {
      "epoch": 3.6520004272460935e-05,
      "step": 38294,
      "training_step_time": 7.7196338176727295
    },
    {
      "epoch": 3.6520957946777345e-05,
      "model_forward_time": 1.7069103717803955,
      "step": 38295
    },
    {
      "epoch": 3.6520957946777345e-05,
      "step": 38295,
      "training_step_time": 7.5399370193481445
    },
    {
      "epoch": 3.652191162109375e-05,
      "model_forward_time": 2.229652166366577,
      "step": 38296
    },
    {
      "epoch": 3.652191162109375e-05,
      "step": 38296,
      "training_step_time": 7.6563942432403564
    },
    {
      "epoch": 3.6522865295410153e-05,
      "model_forward_time": 1.7048463821411133,
      "step": 38297
    },
    {
      "epoch": 3.6522865295410153e-05,
      "step": 38297,
      "training_step_time": 7.603528738021851
    },
    {
      "epoch": 3.6523818969726564e-05,
      "model_forward_time": 1.7045767307281494,
      "step": 38298
    },
    {
      "epoch": 3.6523818969726564e-05,
      "step": 38298,
      "training_step_time": 7.4978179931640625
    },
    {
      "epoch": 3.652477264404297e-05,
      "model_forward_time": 1.7020854949951172,
      "step": 38299
    },
    {
      "epoch": 3.652477264404297e-05,
      "step": 38299,
      "training_step_time": 7.717432498931885
    },
    {
      "epoch": 3.652572631835937e-05,
      "grad_norm": 0.0951514020562172,
      "learning_rate": 7.262730585017188e-06,
      "loss": 0.0567,
      "step": 38300
    },
    {
      "epoch": 3.652572631835937e-05,
      "model_forward_time": 1.7046520709991455,
      "step": 38300
    },
    {
      "dynamics_loss_avg": 0.03933692704886198,
      "epoch": 3.652572631835937e-05,
      "step": 38300
    },
    {
      "action_loss_avg": 0.0010862732044188306,
      "epoch": 3.652572631835937e-05,
      "step": 38300
    },
    {
      "epoch": 3.652572631835937e-05,
      "step": 38300,
      "training_step_time": 7.735966444015503
    },
    {
      "epoch": 3.652667999267578e-05,
      "model_forward_time": 1.706435203552246,
      "step": 38301
    },
    {
      "epoch": 3.652667999267578e-05,
      "step": 38301,
      "training_step_time": 9.273194789886475
    },
    {
      "epoch": 3.652763366699219e-05,
      "model_forward_time": 1.6893978118896484,
      "step": 38302
    },
    {
      "epoch": 3.652763366699219e-05,
      "step": 38302,
      "training_step_time": 8.074839353561401
    },
    {
      "epoch": 3.652858734130859e-05,
      "model_forward_time": 1.7089943885803223,
      "step": 38303
    },
    {
      "epoch": 3.652858734130859e-05,
      "step": 38303,
      "training_step_time": 9.027109861373901
    },
    {
      "epoch": 3.6529541015625e-05,
      "model_forward_time": 0.990645170211792,
      "step": 38304
    },
    {
      "epoch": 3.6529541015625e-05,
      "step": 38304,
      "training_step_time": 7.716480255126953
    },
    {
      "epoch": 3.6530494689941406e-05,
      "model_forward_time": 1.7076866626739502,
      "step": 38305
    },
    {
      "epoch": 3.6530494689941406e-05,
      "step": 38305,
      "training_step_time": 8.30914044380188
    },
    {
      "epoch": 3.653144836425781e-05,
      "model_forward_time": 1.7132000923156738,
      "step": 38306
    },
    {
      "epoch": 3.653144836425781e-05,
      "step": 38306,
      "training_step_time": 7.745697021484375
    },
    {
      "epoch": 3.653240203857422e-05,
      "model_forward_time": 1.7092792987823486,
      "step": 38307
    },
    {
      "epoch": 3.653240203857422e-05,
      "step": 38307,
      "training_step_time": 8.33909821510315
    },
    {
      "epoch": 3.6533355712890625e-05,
      "model_forward_time": 1.7108595371246338,
      "step": 38308
    },
    {
      "epoch": 3.6533355712890625e-05,
      "step": 38308,
      "training_step_time": 7.784496545791626
    },
    {
      "epoch": 3.653430938720703e-05,
      "model_forward_time": 1.708749771118164,
      "step": 38309
    },
    {
      "epoch": 3.653430938720703e-05,
      "step": 38309,
      "training_step_time": 7.79448390007019
    },
    {
      "epoch": 3.653526306152344e-05,
      "grad_norm": 0.10250313580036163,
      "learning_rate": 7.261255994490131e-06,
      "loss": 0.0607,
      "step": 38310
    },
    {
      "epoch": 3.653526306152344e-05,
      "model_forward_time": 1.71280837059021,
      "step": 38310
    },
    {
      "dynamics_loss_avg": 0.0638120548799634,
      "epoch": 3.653526306152344e-05,
      "step": 38310
    },
    {
      "action_loss_avg": 0.0015725322533398866,
      "epoch": 3.653526306152344e-05,
      "step": 38310
    },
    {
      "epoch": 3.653526306152344e-05,
      "step": 38310,
      "training_step_time": 7.806607246398926
    },
    {
      "epoch": 3.6536216735839844e-05,
      "model_forward_time": 1.762660026550293,
      "step": 38311
    },
    {
      "epoch": 3.6536216735839844e-05,
      "step": 38311,
      "training_step_time": 8.630080938339233
    },
    {
      "epoch": 3.653717041015625e-05,
      "model_forward_time": 1.7081875801086426,
      "step": 38312
    },
    {
      "epoch": 3.653717041015625e-05,
      "step": 38312,
      "training_step_time": 10.228246212005615
    },
    {
      "epoch": 3.653812408447266e-05,
      "model_forward_time": 1.7109475135803223,
      "step": 38313
    },
    {
      "epoch": 3.653812408447266e-05,
      "step": 38313,
      "training_step_time": 7.703570365905762
    },
    {
      "epoch": 3.653907775878906e-05,
      "model_forward_time": 1.7122292518615723,
      "step": 38314
    },
    {
      "epoch": 3.653907775878906e-05,
      "step": 38314,
      "training_step_time": 7.753528594970703
    },
    {
      "epoch": 3.6540031433105467e-05,
      "model_forward_time": 2.254566192626953,
      "step": 38315
    },
    {
      "epoch": 3.6540031433105467e-05,
      "step": 38315,
      "training_step_time": 7.706616401672363
    },
    {
      "epoch": 3.654098510742188e-05,
      "model_forward_time": 1.7088637351989746,
      "step": 38316
    },
    {
      "epoch": 3.654098510742188e-05,
      "step": 38316,
      "training_step_time": 7.592056751251221
    },
    {
      "epoch": 3.654193878173828e-05,
      "model_forward_time": 1.7124032974243164,
      "step": 38317
    },
    {
      "epoch": 3.654193878173828e-05,
      "step": 38317,
      "training_step_time": 8.473471403121948
    },
    {
      "epoch": 3.6542892456054685e-05,
      "model_forward_time": 1.7103190422058105,
      "step": 38318
    },
    {
      "epoch": 3.6542892456054685e-05,
      "step": 38318,
      "training_step_time": 7.7343831062316895
    },
    {
      "epoch": 3.6543846130371096e-05,
      "model_forward_time": 1.711198329925537,
      "step": 38319
    },
    {
      "epoch": 3.6543846130371096e-05,
      "step": 38319,
      "training_step_time": 8.790475606918335
    },
    {
      "epoch": 3.65447998046875e-05,
      "grad_norm": 0.13764400780200958,
      "learning_rate": 7.2597811566755174e-06,
      "loss": 0.0614,
      "step": 38320
    },
    {
      "epoch": 3.65447998046875e-05,
      "model_forward_time": 1.7302370071411133,
      "step": 38320
    },
    {
      "dynamics_loss_avg": 0.05694333650171757,
      "epoch": 3.65447998046875e-05,
      "step": 38320
    },
    {
      "action_loss_avg": 0.0017457217501942069,
      "epoch": 3.65447998046875e-05,
      "step": 38320
    },
    {
      "epoch": 3.65447998046875e-05,
      "step": 38320,
      "training_step_time": 8.034727096557617
    },
    {
      "epoch": 3.6545753479003904e-05,
      "model_forward_time": 1.710439920425415,
      "step": 38321
    },
    {
      "epoch": 3.6545753479003904e-05,
      "step": 38321,
      "training_step_time": 8.368130922317505
    },
    {
      "epoch": 3.6546707153320315e-05,
      "model_forward_time": 1.7098784446716309,
      "step": 38322
    },
    {
      "epoch": 3.6546707153320315e-05,
      "step": 38322,
      "training_step_time": 8.507153034210205
    },
    {
      "epoch": 3.654766082763672e-05,
      "model_forward_time": 1.710195779800415,
      "step": 38323
    },
    {
      "epoch": 3.654766082763672e-05,
      "step": 38323,
      "training_step_time": 8.08008861541748
    },
    {
      "epoch": 3.654861450195312e-05,
      "model_forward_time": 2.2449755668640137,
      "step": 38324
    },
    {
      "epoch": 3.654861450195312e-05,
      "step": 38324,
      "training_step_time": 7.676037073135376
    },
    {
      "epoch": 3.6549568176269534e-05,
      "model_forward_time": 1.7013685703277588,
      "step": 38325
    },
    {
      "epoch": 3.6549568176269534e-05,
      "step": 38325,
      "training_step_time": 7.587817192077637
    },
    {
      "epoch": 3.655052185058594e-05,
      "model_forward_time": 1.329364538192749,
      "step": 38326
    },
    {
      "epoch": 3.655052185058594e-05,
      "step": 38326,
      "training_step_time": 8.574583053588867
    },
    {
      "epoch": 3.655147552490234e-05,
      "model_forward_time": 1.7053189277648926,
      "step": 38327
    },
    {
      "epoch": 3.655147552490234e-05,
      "step": 38327,
      "training_step_time": 7.54533839225769
    },
    {
      "epoch": 3.655242919921875e-05,
      "model_forward_time": 1.7100896835327148,
      "step": 38328
    },
    {
      "epoch": 3.655242919921875e-05,
      "step": 38328,
      "training_step_time": 7.5386388301849365
    },
    {
      "epoch": 3.655338287353516e-05,
      "model_forward_time": 1.733403205871582,
      "step": 38329
    },
    {
      "epoch": 3.655338287353516e-05,
      "step": 38329,
      "training_step_time": 8.64979100227356
    },
    {
      "epoch": 3.655433654785156e-05,
      "grad_norm": 0.13406655192375183,
      "learning_rate": 7.2583060717346335e-06,
      "loss": 0.0551,
      "step": 38330
    },
    {
      "epoch": 3.655433654785156e-05,
      "model_forward_time": 1.6988394260406494,
      "step": 38330
    },
    {
      "dynamics_loss_avg": 0.053900498896837234,
      "epoch": 3.655433654785156e-05,
      "step": 38330
    },
    {
      "action_loss_avg": 0.0018005730380536988,
      "epoch": 3.655433654785156e-05,
      "step": 38330
    },
    {
      "epoch": 3.655433654785156e-05,
      "step": 38330,
      "training_step_time": 8.96273159980774
    },
    {
      "epoch": 3.655529022216797e-05,
      "model_forward_time": 1.6966781616210938,
      "step": 38331
    },
    {
      "epoch": 3.655529022216797e-05,
      "step": 38331,
      "training_step_time": 8.602978229522705
    },
    {
      "epoch": 3.6556243896484376e-05,
      "model_forward_time": 1.7057313919067383,
      "step": 38332
    },
    {
      "epoch": 3.6556243896484376e-05,
      "step": 38332,
      "training_step_time": 7.590545177459717
    },
    {
      "epoch": 3.655719757080078e-05,
      "model_forward_time": 1.8837966918945312,
      "step": 38333
    },
    {
      "epoch": 3.655719757080078e-05,
      "step": 38333,
      "training_step_time": 8.333786249160767
    },
    {
      "epoch": 3.655815124511719e-05,
      "model_forward_time": 1.716583013534546,
      "step": 38334
    },
    {
      "epoch": 3.655815124511719e-05,
      "step": 38334,
      "training_step_time": 7.579350709915161
    },
    {
      "epoch": 3.6559104919433595e-05,
      "model_forward_time": 1.7214105129241943,
      "step": 38335
    },
    {
      "epoch": 3.6559104919433595e-05,
      "step": 38335,
      "training_step_time": 8.738417625427246
    },
    {
      "epoch": 3.656005859375e-05,
      "model_forward_time": 1.7120335102081299,
      "step": 38336
    },
    {
      "epoch": 3.656005859375e-05,
      "step": 38336,
      "training_step_time": 7.533125877380371
    },
    {
      "epoch": 3.656101226806641e-05,
      "model_forward_time": 1.719139575958252,
      "step": 38337
    },
    {
      "epoch": 3.656101226806641e-05,
      "step": 38337,
      "training_step_time": 7.601893424987793
    },
    {
      "epoch": 3.6561965942382813e-05,
      "model_forward_time": 1.7256579399108887,
      "step": 38338
    },
    {
      "epoch": 3.6561965942382813e-05,
      "step": 38338,
      "training_step_time": 7.732027292251587
    },
    {
      "epoch": 3.656291961669922e-05,
      "model_forward_time": 1.7162752151489258,
      "step": 38339
    },
    {
      "epoch": 3.656291961669922e-05,
      "step": 38339,
      "training_step_time": 7.6062610149383545
    },
    {
      "epoch": 3.656387329101563e-05,
      "grad_norm": 0.11645183712244034,
      "learning_rate": 7.256830739828793e-06,
      "loss": 0.0616,
      "step": 38340
    },
    {
      "epoch": 3.656387329101563e-05,
      "model_forward_time": 1.709144115447998,
      "step": 38340
    },
    {
      "dynamics_loss_avg": 0.04873904436826706,
      "epoch": 3.656387329101563e-05,
      "step": 38340
    },
    {
      "action_loss_avg": 0.0024149313743691892,
      "epoch": 3.656387329101563e-05,
      "step": 38340
    },
    {
      "epoch": 3.656387329101563e-05,
      "step": 38340,
      "training_step_time": 9.389832973480225
    },
    {
      "epoch": 3.656482696533203e-05,
      "model_forward_time": 1.713294506072998,
      "step": 38341
    },
    {
      "epoch": 3.656482696533203e-05,
      "step": 38341,
      "training_step_time": 8.809392929077148
    },
    {
      "epoch": 3.6565780639648436e-05,
      "model_forward_time": 1.7116239070892334,
      "step": 38342
    },
    {
      "epoch": 3.6565780639648436e-05,
      "step": 38342,
      "training_step_time": 7.6744842529296875
    },
    {
      "epoch": 3.656673431396485e-05,
      "model_forward_time": 1.7094354629516602,
      "step": 38343
    },
    {
      "epoch": 3.656673431396485e-05,
      "step": 38343,
      "training_step_time": 8.777514934539795
    },
    {
      "epoch": 3.656768798828125e-05,
      "model_forward_time": 1.7101879119873047,
      "step": 38344
    },
    {
      "epoch": 3.656768798828125e-05,
      "step": 38344,
      "training_step_time": 9.298884630203247
    },
    {
      "epoch": 3.6568641662597655e-05,
      "model_forward_time": 1.7169175148010254,
      "step": 38345
    },
    {
      "epoch": 3.6568641662597655e-05,
      "step": 38345,
      "training_step_time": 6.791692495346069
    },
    {
      "epoch": 3.656959533691406e-05,
      "model_forward_time": 1.7158372402191162,
      "step": 38346
    },
    {
      "epoch": 3.656959533691406e-05,
      "step": 38346,
      "training_step_time": 6.799530267715454
    },
    {
      "epoch": 3.657054901123047e-05,
      "model_forward_time": 1.7188374996185303,
      "step": 38347
    },
    {
      "epoch": 3.657054901123047e-05,
      "step": 38347,
      "training_step_time": 6.799206972122192
    },
    {
      "epoch": 3.6571502685546874e-05,
      "model_forward_time": 1.718111515045166,
      "step": 38348
    },
    {
      "epoch": 3.6571502685546874e-05,
      "step": 38348,
      "training_step_time": 6.820502519607544
    },
    {
      "epoch": 3.657245635986328e-05,
      "model_forward_time": 1.7157385349273682,
      "step": 38349
    },
    {
      "epoch": 3.657245635986328e-05,
      "step": 38349,
      "training_step_time": 6.790929079055786
    },
    {
      "epoch": 3.657341003417969e-05,
      "grad_norm": 0.15288381278514862,
      "learning_rate": 7.255355161119336e-06,
      "loss": 0.0594,
      "step": 38350
    },
    {
      "epoch": 3.657341003417969e-05,
      "model_forward_time": 1.711505651473999,
      "step": 38350
    },
    {
      "dynamics_loss_avg": 0.059470074437558654,
      "epoch": 3.657341003417969e-05,
      "step": 38350
    },
    {
      "action_loss_avg": 0.0019055440498050302,
      "epoch": 3.657341003417969e-05,
      "step": 38350
    },
    {
      "epoch": 3.657341003417969e-05,
      "step": 38350,
      "training_step_time": 6.8154730796813965
    },
    {
      "epoch": 3.657436370849609e-05,
      "model_forward_time": 1.7110328674316406,
      "step": 38351
    },
    {
      "epoch": 3.657436370849609e-05,
      "step": 38351,
      "training_step_time": 6.765697479248047
    },
    {
      "epoch": 3.65753173828125e-05,
      "model_forward_time": 1.718322992324829,
      "step": 38352
    },
    {
      "epoch": 3.65753173828125e-05,
      "step": 38352,
      "training_step_time": 6.804890871047974
    },
    {
      "epoch": 3.657627105712891e-05,
      "model_forward_time": 2.2867324352264404,
      "step": 38353
    },
    {
      "epoch": 3.657627105712891e-05,
      "step": 38353,
      "training_step_time": 7.189287185668945
    },
    {
      "epoch": 3.657722473144531e-05,
      "model_forward_time": 1.7115933895111084,
      "step": 38354
    },
    {
      "epoch": 3.657722473144531e-05,
      "step": 38354,
      "training_step_time": 7.9198596477508545
    },
    {
      "epoch": 3.6578178405761716e-05,
      "model_forward_time": 1.7192816734313965,
      "step": 38355
    },
    {
      "epoch": 3.6578178405761716e-05,
      "step": 38355,
      "training_step_time": 6.72111701965332
    },
    {
      "epoch": 3.6579132080078127e-05,
      "model_forward_time": 1.7179956436157227,
      "step": 38356
    },
    {
      "epoch": 3.6579132080078127e-05,
      "step": 38356,
      "training_step_time": 6.756763935089111
    },
    {
      "epoch": 3.658008575439453e-05,
      "model_forward_time": 1.7142832279205322,
      "step": 38357
    },
    {
      "epoch": 3.658008575439453e-05,
      "step": 38357,
      "training_step_time": 6.736207008361816
    },
    {
      "epoch": 3.6581039428710935e-05,
      "model_forward_time": 1.7127459049224854,
      "step": 38358
    },
    {
      "epoch": 3.6581039428710935e-05,
      "step": 38358,
      "training_step_time": 6.698378562927246
    },
    {
      "epoch": 3.6581993103027345e-05,
      "model_forward_time": 1.719459056854248,
      "step": 38359
    },
    {
      "epoch": 3.6581993103027345e-05,
      "step": 38359,
      "training_step_time": 6.706718444824219
    },
    {
      "epoch": 3.658294677734375e-05,
      "grad_norm": 0.1295606791973114,
      "learning_rate": 7.253879335767628e-06,
      "loss": 0.0568,
      "step": 38360
    },
    {
      "epoch": 3.658294677734375e-05,
      "model_forward_time": 1.6977732181549072,
      "step": 38360
    },
    {
      "dynamics_loss_avg": 0.04905587080866099,
      "epoch": 3.658294677734375e-05,
      "step": 38360
    },
    {
      "action_loss_avg": 0.0015254324651323258,
      "epoch": 3.658294677734375e-05,
      "step": 38360
    },
    {
      "epoch": 3.658294677734375e-05,
      "step": 38360,
      "training_step_time": 6.74173378944397
    },
    {
      "epoch": 3.6583900451660153e-05,
      "model_forward_time": 1.7048821449279785,
      "step": 38361
    },
    {
      "epoch": 3.6583900451660153e-05,
      "step": 38361,
      "training_step_time": 6.742070436477661
    },
    {
      "epoch": 3.6584854125976564e-05,
      "model_forward_time": 2.267341375350952,
      "step": 38362
    },
    {
      "epoch": 3.6584854125976564e-05,
      "step": 38362,
      "training_step_time": 7.177762508392334
    },
    {
      "epoch": 3.658580780029297e-05,
      "model_forward_time": 1.6996991634368896,
      "step": 38363
    },
    {
      "epoch": 3.658580780029297e-05,
      "step": 38363,
      "training_step_time": 7.758555173873901
    },
    {
      "epoch": 3.658676147460937e-05,
      "model_forward_time": 1.7050437927246094,
      "step": 38364
    },
    {
      "epoch": 3.658676147460937e-05,
      "step": 38364,
      "training_step_time": 6.75850248336792
    },
    {
      "epoch": 3.658771514892578e-05,
      "model_forward_time": 1.7068514823913574,
      "step": 38365
    },
    {
      "epoch": 3.658771514892578e-05,
      "step": 38365,
      "training_step_time": 6.714767217636108
    },
    {
      "epoch": 3.658866882324219e-05,
      "model_forward_time": 1.7051825523376465,
      "step": 38366
    },
    {
      "epoch": 3.658866882324219e-05,
      "step": 38366,
      "training_step_time": 6.7331862449646
    },
    {
      "epoch": 3.658962249755859e-05,
      "model_forward_time": 1.7033181190490723,
      "step": 38367
    },
    {
      "epoch": 3.658962249755859e-05,
      "step": 38367,
      "training_step_time": 6.801107168197632
    },
    {
      "epoch": 3.6590576171875e-05,
      "model_forward_time": 1.6982004642486572,
      "step": 38368
    },
    {
      "epoch": 3.6590576171875e-05,
      "step": 38368,
      "training_step_time": 6.796299457550049
    },
    {
      "epoch": 3.6591529846191406e-05,
      "model_forward_time": 1.7064783573150635,
      "step": 38369
    },
    {
      "epoch": 3.6591529846191406e-05,
      "step": 38369,
      "training_step_time": 6.777951002120972
    },
    {
      "epoch": 3.659248352050781e-05,
      "grad_norm": 0.12586936354637146,
      "learning_rate": 7.2524032639350675e-06,
      "loss": 0.0613,
      "step": 38370
    },
    {
      "epoch": 3.659248352050781e-05,
      "model_forward_time": 1.6997714042663574,
      "step": 38370
    },
    {
      "dynamics_loss_avg": 0.05555595140904188,
      "epoch": 3.659248352050781e-05,
      "step": 38370
    },
    {
      "action_loss_avg": 0.001897542385268025,
      "epoch": 3.659248352050781e-05,
      "step": 38370
    },
    {
      "epoch": 3.659248352050781e-05,
      "step": 38370,
      "training_step_time": 6.764320373535156
    },
    {
      "epoch": 3.659343719482422e-05,
      "model_forward_time": 2.2433924674987793,
      "step": 38371
    },
    {
      "epoch": 3.659343719482422e-05,
      "step": 38371,
      "training_step_time": 7.194423198699951
    },
    {
      "epoch": 3.6594390869140625e-05,
      "model_forward_time": 1.7190780639648438,
      "step": 38372
    },
    {
      "epoch": 3.6594390869140625e-05,
      "step": 38372,
      "training_step_time": 8.741780519485474
    },
    {
      "epoch": 3.659534454345703e-05,
      "model_forward_time": 1.6970860958099365,
      "step": 38373
    },
    {
      "epoch": 3.659534454345703e-05,
      "step": 38373,
      "training_step_time": 7.4430060386657715
    },
    {
      "epoch": 3.659629821777344e-05,
      "model_forward_time": 1.702709674835205,
      "step": 38374
    },
    {
      "epoch": 3.659629821777344e-05,
      "step": 38374,
      "training_step_time": 6.754649639129639
    },
    {
      "epoch": 3.6597251892089844e-05,
      "model_forward_time": 1.7025530338287354,
      "step": 38375
    },
    {
      "epoch": 3.6597251892089844e-05,
      "step": 38375,
      "training_step_time": 6.765815496444702
    },
    {
      "epoch": 3.659820556640625e-05,
      "model_forward_time": 1.6968634128570557,
      "step": 38376
    },
    {
      "epoch": 3.659820556640625e-05,
      "step": 38376,
      "training_step_time": 6.741923570632935
    },
    {
      "epoch": 3.659915924072266e-05,
      "model_forward_time": 1.7048523426055908,
      "step": 38377
    },
    {
      "epoch": 3.659915924072266e-05,
      "step": 38377,
      "training_step_time": 6.8037109375
    },
    {
      "epoch": 3.660011291503906e-05,
      "model_forward_time": 1.7034029960632324,
      "step": 38378
    },
    {
      "epoch": 3.660011291503906e-05,
      "step": 38378,
      "training_step_time": 6.772702693939209
    },
    {
      "epoch": 3.660106658935547e-05,
      "model_forward_time": 1.6953890323638916,
      "step": 38379
    },
    {
      "epoch": 3.660106658935547e-05,
      "step": 38379,
      "training_step_time": 6.790802955627441
    },
    {
      "epoch": 3.660202026367188e-05,
      "grad_norm": 0.1297254115343094,
      "learning_rate": 7.250926945783071e-06,
      "loss": 0.057,
      "step": 38380
    },
    {
      "epoch": 3.660202026367188e-05,
      "model_forward_time": 2.2409002780914307,
      "step": 38380
    },
    {
      "dynamics_loss_avg": 0.04420018214732409,
      "epoch": 3.660202026367188e-05,
      "step": 38380
    },
    {
      "action_loss_avg": 0.0015913366398308427,
      "epoch": 3.660202026367188e-05,
      "step": 38380
    },
    {
      "epoch": 3.660202026367188e-05,
      "step": 38380,
      "training_step_time": 7.193472385406494
    },
    {
      "epoch": 3.660297393798828e-05,
      "model_forward_time": 1.7283248901367188,
      "step": 38381
    },
    {
      "epoch": 3.660297393798828e-05,
      "step": 38381,
      "training_step_time": 8.697458982467651
    },
    {
      "epoch": 3.6603927612304685e-05,
      "model_forward_time": 1.6959693431854248,
      "step": 38382
    },
    {
      "epoch": 3.6603927612304685e-05,
      "step": 38382,
      "training_step_time": 7.551209926605225
    },
    {
      "epoch": 3.6604881286621096e-05,
      "model_forward_time": 1.704214334487915,
      "step": 38383
    },
    {
      "epoch": 3.6604881286621096e-05,
      "step": 38383,
      "training_step_time": 6.71453332901001
    },
    {
      "epoch": 3.66058349609375e-05,
      "model_forward_time": 1.7054712772369385,
      "step": 38384
    },
    {
      "epoch": 3.66058349609375e-05,
      "step": 38384,
      "training_step_time": 6.692163944244385
    },
    {
      "epoch": 3.6606788635253904e-05,
      "model_forward_time": 1.6832537651062012,
      "step": 38385
    },
    {
      "epoch": 3.6606788635253904e-05,
      "step": 38385,
      "training_step_time": 6.7359819412231445
    },
    {
      "epoch": 3.6607742309570315e-05,
      "model_forward_time": 1.7051661014556885,
      "step": 38386
    },
    {
      "epoch": 3.6607742309570315e-05,
      "step": 38386,
      "training_step_time": 6.725651741027832
    },
    {
      "epoch": 3.660869598388672e-05,
      "model_forward_time": 1.7039144039154053,
      "step": 38387
    },
    {
      "epoch": 3.660869598388672e-05,
      "step": 38387,
      "training_step_time": 6.782065153121948
    },
    {
      "epoch": 3.660964965820312e-05,
      "model_forward_time": 1.6999847888946533,
      "step": 38388
    },
    {
      "epoch": 3.660964965820312e-05,
      "step": 38388,
      "training_step_time": 6.804646968841553
    },
    {
      "epoch": 3.6610603332519534e-05,
      "model_forward_time": 2.2583587169647217,
      "step": 38389
    },
    {
      "epoch": 3.6610603332519534e-05,
      "step": 38389,
      "training_step_time": 7.191484451293945
    },
    {
      "epoch": 3.661155700683594e-05,
      "grad_norm": 0.12458222359418869,
      "learning_rate": 7.2494503814730875e-06,
      "loss": 0.0576,
      "step": 38390
    },
    {
      "epoch": 3.661155700683594e-05,
      "model_forward_time": 1.7144854068756104,
      "step": 38390
    },
    {
      "dynamics_loss_avg": 0.05146640762686729,
      "epoch": 3.661155700683594e-05,
      "step": 38390
    },
    {
      "action_loss_avg": 0.0018821650242898612,
      "epoch": 3.661155700683594e-05,
      "step": 38390
    },
    {
      "epoch": 3.661155700683594e-05,
      "step": 38390,
      "training_step_time": 7.867514371871948
    },
    {
      "epoch": 3.661251068115234e-05,
      "model_forward_time": 1.7058138847351074,
      "step": 38391
    },
    {
      "epoch": 3.661251068115234e-05,
      "step": 38391,
      "training_step_time": 8.177016258239746
    },
    {
      "epoch": 3.661346435546875e-05,
      "model_forward_time": 1.711151123046875,
      "step": 38392
    },
    {
      "epoch": 3.661346435546875e-05,
      "step": 38392,
      "training_step_time": 6.735657215118408
    },
    {
      "epoch": 3.661441802978516e-05,
      "model_forward_time": 1.7023279666900635,
      "step": 38393
    },
    {
      "epoch": 3.661441802978516e-05,
      "step": 38393,
      "training_step_time": 6.774274826049805
    },
    {
      "epoch": 3.661537170410156e-05,
      "model_forward_time": 1.704737901687622,
      "step": 38394
    },
    {
      "epoch": 3.661537170410156e-05,
      "step": 38394,
      "training_step_time": 6.7266058921813965
    },
    {
      "epoch": 3.661632537841797e-05,
      "model_forward_time": 1.7092227935791016,
      "step": 38395
    },
    {
      "epoch": 3.661632537841797e-05,
      "step": 38395,
      "training_step_time": 6.73525857925415
    },
    {
      "epoch": 3.6617279052734376e-05,
      "model_forward_time": 1.701270341873169,
      "step": 38396
    },
    {
      "epoch": 3.6617279052734376e-05,
      "step": 38396,
      "training_step_time": 6.748869895935059
    },
    {
      "epoch": 3.661823272705078e-05,
      "model_forward_time": 1.7039172649383545,
      "step": 38397
    },
    {
      "epoch": 3.661823272705078e-05,
      "step": 38397,
      "training_step_time": 6.703741073608398
    },
    {
      "epoch": 3.661918640136719e-05,
      "model_forward_time": 2.2522454261779785,
      "step": 38398
    },
    {
      "epoch": 3.661918640136719e-05,
      "step": 38398,
      "training_step_time": 6.943472146987915
    },
    {
      "epoch": 3.6620140075683595e-05,
      "model_forward_time": 1.7240428924560547,
      "step": 38399
    },
    {
      "epoch": 3.6620140075683595e-05,
      "step": 38399,
      "training_step_time": 9.391815423965454
    },
    {
      "epoch": 3.662109375e-05,
      "grad_norm": 0.1271439641714096,
      "learning_rate": 7.2479735711665934e-06,
      "loss": 0.0589,
      "step": 38400
    },
    {
      "epoch": 3.662109375e-05,
      "model_forward_time": 1.7096996307373047,
      "step": 38400
    },
    {
      "dynamics_loss_avg": 0.06157274544239044,
      "epoch": 3.662109375e-05,
      "step": 38400
    },
    {
      "action_loss_avg": 0.0011846980778500438,
      "epoch": 3.662109375e-05,
      "step": 38400
    },
    {
      "epoch": 3.662109375e-05,
      "step": 38400,
      "training_step_time": 7.596988201141357
    },
    {
      "epoch": 3.662204742431641e-05,
      "model_forward_time": 1.716655969619751,
      "step": 38401
    },
    {
      "epoch": 3.662204742431641e-05,
      "step": 38401,
      "training_step_time": 7.553109645843506
    },
    {
      "epoch": 3.6623001098632813e-05,
      "model_forward_time": 1.7129480838775635,
      "step": 38402
    },
    {
      "epoch": 3.6623001098632813e-05,
      "step": 38402,
      "training_step_time": 6.859448432922363
    },
    {
      "epoch": 3.662395477294922e-05,
      "model_forward_time": 1.7176713943481445,
      "step": 38403
    },
    {
      "epoch": 3.662395477294922e-05,
      "step": 38403,
      "training_step_time": 6.831462144851685
    },
    {
      "epoch": 3.662490844726563e-05,
      "model_forward_time": 1.717599630355835,
      "step": 38404
    },
    {
      "epoch": 3.662490844726563e-05,
      "step": 38404,
      "training_step_time": 6.7392284870147705
    },
    {
      "epoch": 3.662586212158203e-05,
      "model_forward_time": 1.7170794010162354,
      "step": 38405
    },
    {
      "epoch": 3.662586212158203e-05,
      "step": 38405,
      "training_step_time": 6.709551572799683
    },
    {
      "epoch": 3.6626815795898436e-05,
      "model_forward_time": 1.716876745223999,
      "step": 38406
    },
    {
      "epoch": 3.6626815795898436e-05,
      "step": 38406,
      "training_step_time": 6.736878395080566
    },
    {
      "epoch": 3.662776947021484e-05,
      "model_forward_time": 2.275456190109253,
      "step": 38407
    },
    {
      "epoch": 3.662776947021484e-05,
      "step": 38407,
      "training_step_time": 6.9636735916137695
    },
    {
      "epoch": 3.662872314453125e-05,
      "model_forward_time": 1.72359037399292,
      "step": 38408
    },
    {
      "epoch": 3.662872314453125e-05,
      "step": 38408,
      "training_step_time": 8.783284187316895
    },
    {
      "epoch": 3.6629676818847655e-05,
      "model_forward_time": 1.7278404235839844,
      "step": 38409
    },
    {
      "epoch": 3.6629676818847655e-05,
      "step": 38409,
      "training_step_time": 7.581631660461426
    },
    {
      "epoch": 3.663063049316406e-05,
      "grad_norm": 0.14907582104206085,
      "learning_rate": 7.2464965150250884e-06,
      "loss": 0.0596,
      "step": 38410
    },
    {
      "epoch": 3.663063049316406e-05,
      "model_forward_time": 1.7113022804260254,
      "step": 38410
    },
    {
      "dynamics_loss_avg": 0.0576716797426343,
      "epoch": 3.663063049316406e-05,
      "step": 38410
    },
    {
      "action_loss_avg": 0.0015249391726683825,
      "epoch": 3.663063049316406e-05,
      "step": 38410
    },
    {
      "epoch": 3.663063049316406e-05,
      "step": 38410,
      "training_step_time": 8.052182674407959
    },
    {
      "epoch": 3.663158416748047e-05,
      "model_forward_time": 1.7085275650024414,
      "step": 38411
    },
    {
      "epoch": 3.663158416748047e-05,
      "step": 38411,
      "training_step_time": 6.788597822189331
    },
    {
      "epoch": 3.6632537841796874e-05,
      "model_forward_time": 1.7114660739898682,
      "step": 38412
    },
    {
      "epoch": 3.6632537841796874e-05,
      "step": 38412,
      "training_step_time": 6.765079498291016
    },
    {
      "epoch": 3.663349151611328e-05,
      "model_forward_time": 1.716662883758545,
      "step": 38413
    },
    {
      "epoch": 3.663349151611328e-05,
      "step": 38413,
      "training_step_time": 6.757739543914795
    },
    {
      "epoch": 3.663444519042969e-05,
      "model_forward_time": 1.7148945331573486,
      "step": 38414
    },
    {
      "epoch": 3.663444519042969e-05,
      "step": 38414,
      "training_step_time": 6.767854928970337
    },
    {
      "epoch": 3.663539886474609e-05,
      "model_forward_time": 1.7211697101593018,
      "step": 38415
    },
    {
      "epoch": 3.663539886474609e-05,
      "step": 38415,
      "training_step_time": 6.787777423858643
    },
    {
      "epoch": 3.66363525390625e-05,
      "model_forward_time": 2.2781126499176025,
      "step": 38416
    },
    {
      "epoch": 3.66363525390625e-05,
      "step": 38416,
      "training_step_time": 7.009326696395874
    },
    {
      "epoch": 3.663730621337891e-05,
      "model_forward_time": 1.7232928276062012,
      "step": 38417
    },
    {
      "epoch": 3.663730621337891e-05,
      "step": 38417,
      "training_step_time": 8.70993161201477
    },
    {
      "epoch": 3.663825988769531e-05,
      "model_forward_time": 1.3348026275634766,
      "step": 38418
    },
    {
      "epoch": 3.663825988769531e-05,
      "step": 38418,
      "training_step_time": 10.604547023773193
    },
    {
      "epoch": 3.6639213562011716e-05,
      "model_forward_time": 1.7165188789367676,
      "step": 38419
    },
    {
      "epoch": 3.6639213562011716e-05,
      "step": 38419,
      "training_step_time": 7.635347843170166
    },
    {
      "epoch": 3.664016723632813e-05,
      "grad_norm": 0.13896965980529785,
      "learning_rate": 7.245019213210102e-06,
      "loss": 0.0587,
      "step": 38420
    },
    {
      "epoch": 3.664016723632813e-05,
      "model_forward_time": 1.337599277496338,
      "step": 38420
    },
    {
      "dynamics_loss_avg": 0.05612047929316759,
      "epoch": 3.664016723632813e-05,
      "step": 38420
    },
    {
      "action_loss_avg": 0.0017774785112123937,
      "epoch": 3.664016723632813e-05,
      "step": 38420
    },
    {
      "epoch": 3.664016723632813e-05,
      "step": 38420,
      "training_step_time": 7.1151182651519775
    },
    {
      "epoch": 3.664112091064453e-05,
      "model_forward_time": 1.7128584384918213,
      "step": 38421
    },
    {
      "epoch": 3.664112091064453e-05,
      "step": 38421,
      "training_step_time": 6.765415191650391
    },
    {
      "epoch": 3.6642074584960935e-05,
      "model_forward_time": 1.712827205657959,
      "step": 38422
    },
    {
      "epoch": 3.6642074584960935e-05,
      "step": 38422,
      "training_step_time": 6.782891035079956
    },
    {
      "epoch": 3.6643028259277345e-05,
      "model_forward_time": 1.714078426361084,
      "step": 38423
    },
    {
      "epoch": 3.6643028259277345e-05,
      "step": 38423,
      "training_step_time": 6.833434581756592
    },
    {
      "epoch": 3.664398193359375e-05,
      "model_forward_time": 1.7146170139312744,
      "step": 38424
    },
    {
      "epoch": 3.664398193359375e-05,
      "step": 38424,
      "training_step_time": 6.764894723892212
    },
    {
      "epoch": 3.6644935607910154e-05,
      "model_forward_time": 1.7139277458190918,
      "step": 38425
    },
    {
      "epoch": 3.6644935607910154e-05,
      "step": 38425,
      "training_step_time": 6.795516490936279
    },
    {
      "epoch": 3.6645889282226564e-05,
      "model_forward_time": 2.2878501415252686,
      "step": 38426
    },
    {
      "epoch": 3.6645889282226564e-05,
      "step": 38426,
      "training_step_time": 7.553421974182129
    },
    {
      "epoch": 3.664684295654297e-05,
      "model_forward_time": 1.7274675369262695,
      "step": 38427
    },
    {
      "epoch": 3.664684295654297e-05,
      "step": 38427,
      "training_step_time": 10.05568790435791
    },
    {
      "epoch": 3.664779663085937e-05,
      "model_forward_time": 1.711930513381958,
      "step": 38428
    },
    {
      "epoch": 3.664779663085937e-05,
      "step": 38428,
      "training_step_time": 7.598461627960205
    },
    {
      "epoch": 3.664875030517578e-05,
      "model_forward_time": 1.7161617279052734,
      "step": 38429
    },
    {
      "epoch": 3.664875030517578e-05,
      "step": 38429,
      "training_step_time": 7.097743034362793
    },
    {
      "epoch": 3.664970397949219e-05,
      "grad_norm": 0.11595308035612106,
      "learning_rate": 7.24354166588319e-06,
      "loss": 0.0602,
      "step": 38430
    },
    {
      "epoch": 3.664970397949219e-05,
      "model_forward_time": 1.3325510025024414,
      "step": 38430
    },
    {
      "dynamics_loss_avg": 0.044492515921592715,
      "epoch": 3.664970397949219e-05,
      "step": 38430
    },
    {
      "action_loss_avg": 0.0014691227115690708,
      "epoch": 3.664970397949219e-05,
      "step": 38430
    },
    {
      "epoch": 3.664970397949219e-05,
      "step": 38430,
      "training_step_time": 6.798986434936523
    },
    {
      "epoch": 3.665065765380859e-05,
      "model_forward_time": 1.721996545791626,
      "step": 38431
    },
    {
      "epoch": 3.665065765380859e-05,
      "step": 38431,
      "training_step_time": 6.804893970489502
    },
    {
      "epoch": 3.6651611328125e-05,
      "model_forward_time": 1.7209956645965576,
      "step": 38432
    },
    {
      "epoch": 3.6651611328125e-05,
      "step": 38432,
      "training_step_time": 6.744826793670654
    },
    {
      "epoch": 3.6652565002441406e-05,
      "model_forward_time": 1.7164556980133057,
      "step": 38433
    },
    {
      "epoch": 3.6652565002441406e-05,
      "step": 38433,
      "training_step_time": 6.753742933273315
    },
    {
      "epoch": 3.665351867675781e-05,
      "model_forward_time": 1.7161533832550049,
      "step": 38434
    },
    {
      "epoch": 3.665351867675781e-05,
      "step": 38434,
      "training_step_time": 6.782929420471191
    },
    {
      "epoch": 3.665447235107422e-05,
      "model_forward_time": 2.295602560043335,
      "step": 38435
    },
    {
      "epoch": 3.665447235107422e-05,
      "step": 38435,
      "training_step_time": 8.141243934631348
    },
    {
      "epoch": 3.6655426025390625e-05,
      "model_forward_time": 1.7168567180633545,
      "step": 38436
    },
    {
      "epoch": 3.6655426025390625e-05,
      "step": 38436,
      "training_step_time": 10.088273763656616
    },
    {
      "epoch": 3.665637969970703e-05,
      "model_forward_time": 1.7128586769104004,
      "step": 38437
    },
    {
      "epoch": 3.665637969970703e-05,
      "step": 38437,
      "training_step_time": 8.375979900360107
    },
    {
      "epoch": 3.665733337402344e-05,
      "model_forward_time": 1.719757080078125,
      "step": 38438
    },
    {
      "epoch": 3.665733337402344e-05,
      "step": 38438,
      "training_step_time": 7.344669580459595
    },
    {
      "epoch": 3.6658287048339844e-05,
      "model_forward_time": 1.7103114128112793,
      "step": 38439
    },
    {
      "epoch": 3.6658287048339844e-05,
      "step": 38439,
      "training_step_time": 6.773787260055542
    },
    {
      "epoch": 3.665924072265625e-05,
      "grad_norm": 0.17437706887722015,
      "learning_rate": 7.242063873205936e-06,
      "loss": 0.0601,
      "step": 38440
    },
    {
      "epoch": 3.665924072265625e-05,
      "model_forward_time": 1.7086846828460693,
      "step": 38440
    },
    {
      "dynamics_loss_avg": 0.05727973384782672,
      "epoch": 3.665924072265625e-05,
      "step": 38440
    },
    {
      "action_loss_avg": 0.0012193602437037044,
      "epoch": 3.665924072265625e-05,
      "step": 38440
    },
    {
      "epoch": 3.665924072265625e-05,
      "step": 38440,
      "training_step_time": 6.7223060131073
    },
    {
      "epoch": 3.666019439697266e-05,
      "model_forward_time": 1.7148542404174805,
      "step": 38441
    },
    {
      "epoch": 3.666019439697266e-05,
      "step": 38441,
      "training_step_time": 6.736052751541138
    },
    {
      "epoch": 3.666114807128906e-05,
      "model_forward_time": 1.7081615924835205,
      "step": 38442
    },
    {
      "epoch": 3.666114807128906e-05,
      "step": 38442,
      "training_step_time": 6.847020387649536
    },
    {
      "epoch": 3.666210174560547e-05,
      "model_forward_time": 1.7171885967254639,
      "step": 38443
    },
    {
      "epoch": 3.666210174560547e-05,
      "step": 38443,
      "training_step_time": 6.743310928344727
    },
    {
      "epoch": 3.666305541992188e-05,
      "model_forward_time": 1.8772425651550293,
      "step": 38444
    },
    {
      "epoch": 3.666305541992188e-05,
      "step": 38444,
      "training_step_time": 8.740753650665283
    },
    {
      "epoch": 3.666400909423828e-05,
      "model_forward_time": 1.7219715118408203,
      "step": 38445
    },
    {
      "epoch": 3.666400909423828e-05,
      "step": 38445,
      "training_step_time": 9.48452639579773
    },
    {
      "epoch": 3.6664962768554686e-05,
      "model_forward_time": 1.712812900543213,
      "step": 38446
    },
    {
      "epoch": 3.6664962768554686e-05,
      "step": 38446,
      "training_step_time": 7.682414531707764
    },
    {
      "epoch": 3.6665916442871096e-05,
      "model_forward_time": 1.7114014625549316,
      "step": 38447
    },
    {
      "epoch": 3.6665916442871096e-05,
      "step": 38447,
      "training_step_time": 7.61273717880249
    },
    {
      "epoch": 3.66668701171875e-05,
      "model_forward_time": 1.7187836170196533,
      "step": 38448
    },
    {
      "epoch": 3.66668701171875e-05,
      "step": 38448,
      "training_step_time": 6.75156831741333
    },
    {
      "epoch": 3.6667823791503904e-05,
      "model_forward_time": 1.7125067710876465,
      "step": 38449
    },
    {
      "epoch": 3.6667823791503904e-05,
      "step": 38449,
      "training_step_time": 6.813364505767822
    },
    {
      "epoch": 3.6668777465820315e-05,
      "grad_norm": 0.11855665594339371,
      "learning_rate": 7.240585835339946e-06,
      "loss": 0.058,
      "step": 38450
    },
    {
      "epoch": 3.6668777465820315e-05,
      "model_forward_time": 1.7015702724456787,
      "step": 38450
    },
    {
      "dynamics_loss_avg": 0.07072577103972436,
      "epoch": 3.6668777465820315e-05,
      "step": 38450
    },
    {
      "action_loss_avg": 0.004005271883215755,
      "epoch": 3.6668777465820315e-05,
      "step": 38450
    },
    {
      "epoch": 3.6668777465820315e-05,
      "step": 38450,
      "training_step_time": 6.757726430892944
    },
    {
      "epoch": 3.666973114013672e-05,
      "model_forward_time": 1.7152321338653564,
      "step": 38451
    },
    {
      "epoch": 3.666973114013672e-05,
      "step": 38451,
      "training_step_time": 6.707515716552734
    },
    {
      "epoch": 3.667068481445312e-05,
      "model_forward_time": 1.7119779586791992,
      "step": 38452
    },
    {
      "epoch": 3.667068481445312e-05,
      "step": 38452,
      "training_step_time": 6.692641496658325
    },
    {
      "epoch": 3.6671638488769534e-05,
      "model_forward_time": 2.298851490020752,
      "step": 38453
    },
    {
      "epoch": 3.6671638488769534e-05,
      "step": 38453,
      "training_step_time": 7.694617986679077
    },
    {
      "epoch": 3.667259216308594e-05,
      "model_forward_time": 1.7195277214050293,
      "step": 38454
    },
    {
      "epoch": 3.667259216308594e-05,
      "step": 38454,
      "training_step_time": 9.703693866729736
    },
    {
      "epoch": 3.667354583740234e-05,
      "model_forward_time": 1.7375147342681885,
      "step": 38455
    },
    {
      "epoch": 3.667354583740234e-05,
      "step": 38455,
      "training_step_time": 8.829920530319214
    },
    {
      "epoch": 3.667449951171875e-05,
      "model_forward_time": 1.6919939517974854,
      "step": 38456
    },
    {
      "epoch": 3.667449951171875e-05,
      "step": 38456,
      "training_step_time": 7.533722400665283
    },
    {
      "epoch": 3.667545318603516e-05,
      "model_forward_time": 1.7031888961791992,
      "step": 38457
    },
    {
      "epoch": 3.667545318603516e-05,
      "step": 38457,
      "training_step_time": 6.764723777770996
    },
    {
      "epoch": 3.667640686035156e-05,
      "model_forward_time": 1.7026681900024414,
      "step": 38458
    },
    {
      "epoch": 3.667640686035156e-05,
      "step": 38458,
      "training_step_time": 6.76781153678894
    },
    {
      "epoch": 3.667736053466797e-05,
      "model_forward_time": 1.7070090770721436,
      "step": 38459
    },
    {
      "epoch": 3.667736053466797e-05,
      "step": 38459,
      "training_step_time": 6.781955003738403
    },
    {
      "epoch": 3.6678314208984376e-05,
      "grad_norm": 0.13049468398094177,
      "learning_rate": 7.239107552446858e-06,
      "loss": 0.0564,
      "step": 38460
    },
    {
      "epoch": 3.6678314208984376e-05,
      "model_forward_time": 1.693073034286499,
      "step": 38460
    },
    {
      "dynamics_loss_avg": 0.04676313921809196,
      "epoch": 3.6678314208984376e-05,
      "step": 38460
    },
    {
      "action_loss_avg": 0.0018113936806912535,
      "epoch": 3.6678314208984376e-05,
      "step": 38460
    },
    {
      "epoch": 3.6678314208984376e-05,
      "step": 38460,
      "training_step_time": 6.719909191131592
    },
    {
      "epoch": 3.667926788330078e-05,
      "model_forward_time": 1.7001423835754395,
      "step": 38461
    },
    {
      "epoch": 3.667926788330078e-05,
      "step": 38461,
      "training_step_time": 6.762215852737427
    },
    {
      "epoch": 3.668022155761719e-05,
      "model_forward_time": 0.9786927700042725,
      "step": 38462
    },
    {
      "epoch": 3.668022155761719e-05,
      "step": 38462,
      "training_step_time": 7.7113823890686035
    },
    {
      "epoch": 3.6681175231933595e-05,
      "model_forward_time": 2.2695443630218506,
      "step": 38463
    },
    {
      "epoch": 3.6681175231933595e-05,
      "step": 38463,
      "training_step_time": 8.618417024612427
    },
    {
      "epoch": 3.668212890625e-05,
      "model_forward_time": 1.722510576248169,
      "step": 38464
    },
    {
      "epoch": 3.668212890625e-05,
      "step": 38464,
      "training_step_time": 9.145832061767578
    },
    {
      "epoch": 3.668308258056641e-05,
      "model_forward_time": 1.7081506252288818,
      "step": 38465
    },
    {
      "epoch": 3.668308258056641e-05,
      "step": 38465,
      "training_step_time": 8.822479009628296
    },
    {
      "epoch": 3.6684036254882814e-05,
      "model_forward_time": 1.6994812488555908,
      "step": 38466
    },
    {
      "epoch": 3.6684036254882814e-05,
      "step": 38466,
      "training_step_time": 7.4569411277771
    },
    {
      "epoch": 3.668498992919922e-05,
      "model_forward_time": 1.7007637023925781,
      "step": 38467
    },
    {
      "epoch": 3.668498992919922e-05,
      "step": 38467,
      "training_step_time": 6.707613706588745
    },
    {
      "epoch": 3.668594360351563e-05,
      "model_forward_time": 1.7013413906097412,
      "step": 38468
    },
    {
      "epoch": 3.668594360351563e-05,
      "step": 38468,
      "training_step_time": 6.677698612213135
    },
    {
      "epoch": 3.668689727783203e-05,
      "model_forward_time": 1.7056074142456055,
      "step": 38469
    },
    {
      "epoch": 3.668689727783203e-05,
      "step": 38469,
      "training_step_time": 6.671111345291138
    },
    {
      "epoch": 3.6687850952148436e-05,
      "grad_norm": 0.1386934369802475,
      "learning_rate": 7.237629024688334e-06,
      "loss": 0.061,
      "step": 38470
    },
    {
      "epoch": 3.6687850952148436e-05,
      "model_forward_time": 1.6936500072479248,
      "step": 38470
    },
    {
      "dynamics_loss_avg": 0.04912149589508772,
      "epoch": 3.6687850952148436e-05,
      "step": 38470
    },
    {
      "action_loss_avg": 0.0026899459422565996,
      "epoch": 3.6687850952148436e-05,
      "step": 38470
    },
    {
      "epoch": 3.6687850952148436e-05,
      "step": 38470,
      "training_step_time": 6.691908121109009
    },
    {
      "epoch": 3.668880462646484e-05,
      "model_forward_time": 1.7015721797943115,
      "step": 38471
    },
    {
      "epoch": 3.668880462646484e-05,
      "step": 38471,
      "training_step_time": 7.733627796173096
    },
    {
      "epoch": 3.668975830078125e-05,
      "model_forward_time": 2.2633020877838135,
      "step": 38472
    },
    {
      "epoch": 3.668975830078125e-05,
      "step": 38472,
      "training_step_time": 9.498841762542725
    },
    {
      "epoch": 3.6690711975097655e-05,
      "model_forward_time": 1.7199575901031494,
      "step": 38473
    },
    {
      "epoch": 3.6690711975097655e-05,
      "step": 38473,
      "training_step_time": 8.649169445037842
    },
    {
      "epoch": 3.669166564941406e-05,
      "model_forward_time": 1.7023067474365234,
      "step": 38474
    },
    {
      "epoch": 3.669166564941406e-05,
      "step": 38474,
      "training_step_time": 8.67926549911499
    },
    {
      "epoch": 3.669261932373047e-05,
      "model_forward_time": 1.7050259113311768,
      "step": 38475
    },
    {
      "epoch": 3.669261932373047e-05,
      "step": 38475,
      "training_step_time": 7.167382478713989
    },
    {
      "epoch": 3.6693572998046874e-05,
      "model_forward_time": 1.7045719623565674,
      "step": 38476
    },
    {
      "epoch": 3.6693572998046874e-05,
      "step": 38476,
      "training_step_time": 6.76065993309021
    },
    {
      "epoch": 3.669452667236328e-05,
      "model_forward_time": 1.7035186290740967,
      "step": 38477
    },
    {
      "epoch": 3.669452667236328e-05,
      "step": 38477,
      "training_step_time": 6.723303318023682
    },
    {
      "epoch": 3.669548034667969e-05,
      "model_forward_time": 1.7055249214172363,
      "step": 38478
    },
    {
      "epoch": 3.669548034667969e-05,
      "step": 38478,
      "training_step_time": 6.690697431564331
    },
    {
      "epoch": 3.669643402099609e-05,
      "model_forward_time": 1.7075676918029785,
      "step": 38479
    },
    {
      "epoch": 3.669643402099609e-05,
      "step": 38479,
      "training_step_time": 6.791475057601929
    },
    {
      "epoch": 3.66973876953125e-05,
      "grad_norm": 0.11007028818130493,
      "learning_rate": 7.2361502522260644e-06,
      "loss": 0.0609,
      "step": 38480
    },
    {
      "epoch": 3.66973876953125e-05,
      "model_forward_time": 1.6990771293640137,
      "step": 38480
    },
    {
      "dynamics_loss_avg": 0.0573424156755209,
      "epoch": 3.66973876953125e-05,
      "step": 38480
    },
    {
      "action_loss_avg": 0.0017946935433428735,
      "epoch": 3.66973876953125e-05,
      "step": 38480
    },
    {
      "epoch": 3.66973876953125e-05,
      "step": 38480,
      "training_step_time": 7.649711847305298
    },
    {
      "epoch": 3.669834136962891e-05,
      "model_forward_time": 2.2698404788970947,
      "step": 38481
    },
    {
      "epoch": 3.669834136962891e-05,
      "step": 38481,
      "training_step_time": 8.392940521240234
    },
    {
      "epoch": 3.669929504394531e-05,
      "model_forward_time": 1.707000494003296,
      "step": 38482
    },
    {
      "epoch": 3.669929504394531e-05,
      "step": 38482,
      "training_step_time": 7.726980686187744
    },
    {
      "epoch": 3.6700248718261716e-05,
      "model_forward_time": 1.7007226943969727,
      "step": 38483
    },
    {
      "epoch": 3.6700248718261716e-05,
      "step": 38483,
      "training_step_time": 10.229417324066162
    },
    {
      "epoch": 3.670120239257813e-05,
      "model_forward_time": 1.6981956958770752,
      "step": 38484
    },
    {
      "epoch": 3.670120239257813e-05,
      "step": 38484,
      "training_step_time": 7.627924919128418
    },
    {
      "epoch": 3.670215606689453e-05,
      "model_forward_time": 1.7016124725341797,
      "step": 38485
    },
    {
      "epoch": 3.670215606689453e-05,
      "step": 38485,
      "training_step_time": 6.827217102050781
    },
    {
      "epoch": 3.6703109741210935e-05,
      "model_forward_time": 1.7031910419464111,
      "step": 38486
    },
    {
      "epoch": 3.6703109741210935e-05,
      "step": 38486,
      "training_step_time": 6.835009813308716
    },
    {
      "epoch": 3.6704063415527346e-05,
      "model_forward_time": 1.7120130062103271,
      "step": 38487
    },
    {
      "epoch": 3.6704063415527346e-05,
      "step": 38487,
      "training_step_time": 6.882261514663696
    },
    {
      "epoch": 3.670501708984375e-05,
      "model_forward_time": 1.7202510833740234,
      "step": 38488
    },
    {
      "epoch": 3.670501708984375e-05,
      "step": 38488,
      "training_step_time": 6.832834959030151
    },
    {
      "epoch": 3.6705970764160154e-05,
      "model_forward_time": 1.7216131687164307,
      "step": 38489
    },
    {
      "epoch": 3.6705970764160154e-05,
      "step": 38489,
      "training_step_time": 7.6216442584991455
    },
    {
      "epoch": 3.6706924438476564e-05,
      "grad_norm": 0.1493295580148697,
      "learning_rate": 7.234671235221765e-06,
      "loss": 0.0596,
      "step": 38490
    },
    {
      "epoch": 3.6706924438476564e-05,
      "model_forward_time": 2.2812118530273438,
      "step": 38490
    },
    {
      "dynamics_loss_avg": 0.053251917287707326,
      "epoch": 3.6706924438476564e-05,
      "step": 38490
    },
    {
      "action_loss_avg": 0.0019793423009105025,
      "epoch": 3.6706924438476564e-05,
      "step": 38490
    },
    {
      "epoch": 3.6706924438476564e-05,
      "step": 38490,
      "training_step_time": 7.715235710144043
    },
    {
      "epoch": 3.670787811279297e-05,
      "model_forward_time": 1.7209224700927734,
      "step": 38491
    },
    {
      "epoch": 3.670787811279297e-05,
      "step": 38491,
      "training_step_time": 9.807525873184204
    },
    {
      "epoch": 3.670883178710937e-05,
      "model_forward_time": 1.7406632900238037,
      "step": 38492
    },
    {
      "epoch": 3.670883178710937e-05,
      "step": 38492,
      "training_step_time": 8.578262567520142
    },
    {
      "epoch": 3.670978546142578e-05,
      "model_forward_time": 1.7210733890533447,
      "step": 38493
    },
    {
      "epoch": 3.670978546142578e-05,
      "step": 38493,
      "training_step_time": 7.730336427688599
    },
    {
      "epoch": 3.671073913574219e-05,
      "model_forward_time": 1.7172179222106934,
      "step": 38494
    },
    {
      "epoch": 3.671073913574219e-05,
      "step": 38494,
      "training_step_time": 6.787684679031372
    },
    {
      "epoch": 3.671169281005859e-05,
      "model_forward_time": 1.7162892818450928,
      "step": 38495
    },
    {
      "epoch": 3.671169281005859e-05,
      "step": 38495,
      "training_step_time": 6.769275903701782
    },
    {
      "epoch": 3.6712646484375e-05,
      "model_forward_time": 1.715780258178711,
      "step": 38496
    },
    {
      "epoch": 3.6712646484375e-05,
      "step": 38496,
      "training_step_time": 6.778192758560181
    },
    {
      "epoch": 3.6713600158691406e-05,
      "model_forward_time": 1.7214257717132568,
      "step": 38497
    },
    {
      "epoch": 3.6713600158691406e-05,
      "step": 38497,
      "training_step_time": 6.755035638809204
    },
    {
      "epoch": 3.671455383300781e-05,
      "model_forward_time": 1.727546215057373,
      "step": 38498
    },
    {
      "epoch": 3.671455383300781e-05,
      "step": 38498,
      "training_step_time": 7.572953701019287
    },
    {
      "epoch": 3.671550750732422e-05,
      "model_forward_time": 1.835404634475708,
      "step": 38499
    },
    {
      "epoch": 3.671550750732422e-05,
      "step": 38499,
      "training_step_time": 8.280163049697876
    },
    {
      "epoch": 3.6716461181640625e-05,
      "grad_norm": 0.10269252955913544,
      "learning_rate": 7.2331919738371795e-06,
      "loss": 0.0544,
      "step": 38500
    },
    {
      "epoch": 3.6716461181640625e-05,
      "model_forward_time": 1.7669613361358643,
      "step": 38500
    },
    {
      "dynamics_loss_avg": 0.06780118029564619,
      "epoch": 3.6716461181640625e-05,
      "step": 38500
    },
    {
      "action_loss_avg": 0.0018682563561014831,
      "epoch": 3.6716461181640625e-05,
      "step": 38500
    },
    {
      "epoch": 3.6716461181640625e-05,
      "step": 38500,
      "training_step_time": 8.199099779129028
    },
    {
      "epoch": 3.671741485595703e-05,
      "model_forward_time": 1.7388834953308105,
      "step": 38501
    },
    {
      "epoch": 3.671741485595703e-05,
      "step": 38501,
      "training_step_time": 8.327561140060425
    },
    {
      "epoch": 3.671836853027344e-05,
      "model_forward_time": 1.7551829814910889,
      "step": 38502
    },
    {
      "epoch": 3.671836853027344e-05,
      "step": 38502,
      "training_step_time": 7.60602331161499
    },
    {
      "epoch": 3.6719322204589844e-05,
      "model_forward_time": 1.750399112701416,
      "step": 38503
    },
    {
      "epoch": 3.6719322204589844e-05,
      "step": 38503,
      "training_step_time": 7.054912805557251
    },
    {
      "epoch": 3.672027587890625e-05,
      "model_forward_time": 1.7758543491363525,
      "step": 38504
    },
    {
      "epoch": 3.672027587890625e-05,
      "step": 38504,
      "training_step_time": 6.763986349105835
    },
    {
      "epoch": 3.672122955322266e-05,
      "model_forward_time": 1.7905921936035156,
      "step": 38505
    },
    {
      "epoch": 3.672122955322266e-05,
      "step": 38505,
      "training_step_time": 6.771989345550537
    },
    {
      "epoch": 3.672218322753906e-05,
      "model_forward_time": 1.7670600414276123,
      "step": 38506
    },
    {
      "epoch": 3.672218322753906e-05,
      "step": 38506,
      "training_step_time": 6.784836053848267
    },
    {
      "epoch": 3.672313690185547e-05,
      "model_forward_time": 1.760361909866333,
      "step": 38507
    },
    {
      "epoch": 3.672313690185547e-05,
      "step": 38507,
      "training_step_time": 7.54012131690979
    },
    {
      "epoch": 3.672409057617188e-05,
      "model_forward_time": 1.7787137031555176,
      "step": 38508
    },
    {
      "epoch": 3.672409057617188e-05,
      "step": 38508,
      "training_step_time": 7.6918745040893555
    },
    {
      "epoch": 3.672504425048828e-05,
      "model_forward_time": 2.917818784713745,
      "step": 38509
    },
    {
      "epoch": 3.672504425048828e-05,
      "step": 38509,
      "training_step_time": 9.498185157775879
    },
    {
      "epoch": 3.6725997924804686e-05,
      "grad_norm": 0.11211441457271576,
      "learning_rate": 7.231712468234078e-06,
      "loss": 0.0564,
      "step": 38510
    },
    {
      "epoch": 3.6725997924804686e-05,
      "model_forward_time": 1.7526919841766357,
      "step": 38510
    },
    {
      "dynamics_loss_avg": 0.04915748136118055,
      "epoch": 3.6725997924804686e-05,
      "step": 38510
    },
    {
      "action_loss_avg": 0.0019997836323454978,
      "epoch": 3.6725997924804686e-05,
      "step": 38510
    },
    {
      "epoch": 3.6725997924804686e-05,
      "step": 38510,
      "training_step_time": 8.727902173995972
    },
    {
      "epoch": 3.6726951599121096e-05,
      "model_forward_time": 1.7722771167755127,
      "step": 38511
    },
    {
      "epoch": 3.6726951599121096e-05,
      "step": 38511,
      "training_step_time": 7.595335245132446
    },
    {
      "epoch": 3.67279052734375e-05,
      "model_forward_time": 1.8234844207763672,
      "step": 38512
    },
    {
      "epoch": 3.67279052734375e-05,
      "step": 38512,
      "training_step_time": 7.107877731323242
    },
    {
      "epoch": 3.6728858947753904e-05,
      "model_forward_time": 1.7757527828216553,
      "step": 38513
    },
    {
      "epoch": 3.6728858947753904e-05,
      "step": 38513,
      "training_step_time": 6.728062391281128
    },
    {
      "epoch": 3.6729812622070315e-05,
      "model_forward_time": 1.7717695236206055,
      "step": 38514
    },
    {
      "epoch": 3.6729812622070315e-05,
      "step": 38514,
      "training_step_time": 6.726609230041504
    },
    {
      "epoch": 3.673076629638672e-05,
      "model_forward_time": 1.765758752822876,
      "step": 38515
    },
    {
      "epoch": 3.673076629638672e-05,
      "step": 38515,
      "training_step_time": 6.764927625656128
    },
    {
      "epoch": 3.673171997070312e-05,
      "model_forward_time": 1.7627506256103516,
      "step": 38516
    },
    {
      "epoch": 3.673171997070312e-05,
      "step": 38516,
      "training_step_time": 7.507971525192261
    },
    {
      "epoch": 3.6732673645019534e-05,
      "model_forward_time": 1.7674059867858887,
      "step": 38517
    },
    {
      "epoch": 3.6732673645019534e-05,
      "step": 38517,
      "training_step_time": 7.0831475257873535
    },
    {
      "epoch": 3.673362731933594e-05,
      "model_forward_time": 2.526021718978882,
      "step": 38518
    },
    {
      "epoch": 3.673362731933594e-05,
      "step": 38518,
      "training_step_time": 8.685309410095215
    },
    {
      "epoch": 3.673458099365234e-05,
      "model_forward_time": 1.7582104206085205,
      "step": 38519
    },
    {
      "epoch": 3.673458099365234e-05,
      "step": 38519,
      "training_step_time": 9.822989225387573
    },
    {
      "epoch": 3.673553466796875e-05,
      "grad_norm": 0.16118216514587402,
      "learning_rate": 7.230232718574255e-06,
      "loss": 0.0598,
      "step": 38520
    },
    {
      "epoch": 3.673553466796875e-05,
      "model_forward_time": 1.7656631469726562,
      "step": 38520
    },
    {
      "dynamics_loss_avg": 0.05530600324273109,
      "epoch": 3.673553466796875e-05,
      "step": 38520
    },
    {
      "action_loss_avg": 0.0016014852735679597,
      "epoch": 3.673553466796875e-05,
      "step": 38520
    },
    {
      "epoch": 3.673553466796875e-05,
      "step": 38520,
      "training_step_time": 7.573249816894531
    },
    {
      "epoch": 3.673648834228516e-05,
      "model_forward_time": 1.755309820175171,
      "step": 38521
    },
    {
      "epoch": 3.673648834228516e-05,
      "step": 38521,
      "training_step_time": 7.497467517852783
    },
    {
      "epoch": 3.673744201660156e-05,
      "model_forward_time": 1.7755346298217773,
      "step": 38522
    },
    {
      "epoch": 3.673744201660156e-05,
      "step": 38522,
      "training_step_time": 6.752471685409546
    },
    {
      "epoch": 3.673839569091797e-05,
      "model_forward_time": 1.7577571868896484,
      "step": 38523
    },
    {
      "epoch": 3.673839569091797e-05,
      "step": 38523,
      "training_step_time": 6.680577278137207
    },
    {
      "epoch": 3.6739349365234376e-05,
      "model_forward_time": 1.7657296657562256,
      "step": 38524
    },
    {
      "epoch": 3.6739349365234376e-05,
      "step": 38524,
      "training_step_time": 6.768959999084473
    },
    {
      "epoch": 3.674030303955078e-05,
      "model_forward_time": 1.756417989730835,
      "step": 38525
    },
    {
      "epoch": 3.674030303955078e-05,
      "step": 38525,
      "training_step_time": 7.524492263793945
    },
    {
      "epoch": 3.674125671386719e-05,
      "model_forward_time": 1.7746644020080566,
      "step": 38526
    },
    {
      "epoch": 3.674125671386719e-05,
      "step": 38526,
      "training_step_time": 7.065077781677246
    },
    {
      "epoch": 3.6742210388183595e-05,
      "model_forward_time": 2.8883473873138428,
      "step": 38527
    },
    {
      "epoch": 3.6742210388183595e-05,
      "step": 38527,
      "training_step_time": 8.782025337219238
    },
    {
      "epoch": 3.67431640625e-05,
      "model_forward_time": 1.7635462284088135,
      "step": 38528
    },
    {
      "epoch": 3.67431640625e-05,
      "step": 38528,
      "training_step_time": 7.693065166473389
    },
    {
      "epoch": 3.674411773681641e-05,
      "model_forward_time": 1.7609598636627197,
      "step": 38529
    },
    {
      "epoch": 3.674411773681641e-05,
      "step": 38529,
      "training_step_time": 8.314013004302979
    },
    {
      "epoch": 3.6745071411132814e-05,
      "grad_norm": 0.11859013885259628,
      "learning_rate": 7.228752725019536e-06,
      "loss": 0.062,
      "step": 38530
    },
    {
      "epoch": 3.6745071411132814e-05,
      "model_forward_time": 1.7640914916992188,
      "step": 38530
    },
    {
      "dynamics_loss_avg": 0.053144770301878454,
      "epoch": 3.6745071411132814e-05,
      "step": 38530
    },
    {
      "action_loss_avg": 0.0011335496121319011,
      "epoch": 3.6745071411132814e-05,
      "step": 38530
    },
    {
      "epoch": 3.6745071411132814e-05,
      "step": 38530,
      "training_step_time": 7.600371837615967
    },
    {
      "epoch": 3.674602508544922e-05,
      "model_forward_time": 1.7586016654968262,
      "step": 38531
    },
    {
      "epoch": 3.674602508544922e-05,
      "step": 38531,
      "training_step_time": 6.684205532073975
    },
    {
      "epoch": 3.674697875976562e-05,
      "model_forward_time": 1.759171962738037,
      "step": 38532
    },
    {
      "epoch": 3.674697875976562e-05,
      "step": 38532,
      "training_step_time": 6.701086759567261
    },
    {
      "epoch": 3.674793243408203e-05,
      "model_forward_time": 1.7656869888305664,
      "step": 38533
    },
    {
      "epoch": 3.674793243408203e-05,
      "step": 38533,
      "training_step_time": 6.807500839233398
    },
    {
      "epoch": 3.6748886108398436e-05,
      "model_forward_time": 1.7503507137298584,
      "step": 38534
    },
    {
      "epoch": 3.6748886108398436e-05,
      "step": 38534,
      "training_step_time": 7.538611173629761
    },
    {
      "epoch": 3.674983978271484e-05,
      "model_forward_time": 1.752044439315796,
      "step": 38535
    },
    {
      "epoch": 3.674983978271484e-05,
      "step": 38535,
      "training_step_time": 7.336273431777954
    },
    {
      "epoch": 3.675079345703125e-05,
      "model_forward_time": 2.9384381771087646,
      "step": 38536
    },
    {
      "epoch": 3.675079345703125e-05,
      "step": 38536,
      "training_step_time": 7.822660446166992
    },
    {
      "epoch": 3.6751747131347655e-05,
      "model_forward_time": 1.7500154972076416,
      "step": 38537
    },
    {
      "epoch": 3.6751747131347655e-05,
      "step": 38537,
      "training_step_time": 8.891632080078125
    },
    {
      "epoch": 3.675270080566406e-05,
      "model_forward_time": 1.758509874343872,
      "step": 38538
    },
    {
      "epoch": 3.675270080566406e-05,
      "step": 38538,
      "training_step_time": 10.589805126190186
    },
    {
      "epoch": 3.675365447998047e-05,
      "model_forward_time": 1.7588553428649902,
      "step": 38539
    },
    {
      "epoch": 3.675365447998047e-05,
      "step": 38539,
      "training_step_time": 7.77895975112915
    },
    {
      "epoch": 3.6754608154296874e-05,
      "grad_norm": 0.17552240192890167,
      "learning_rate": 7.227272487731769e-06,
      "loss": 0.0644,
      "step": 38540
    },
    {
      "epoch": 3.6754608154296874e-05,
      "model_forward_time": 1.7541978359222412,
      "step": 38540
    },
    {
      "dynamics_loss_avg": 0.05355711467564106,
      "epoch": 3.6754608154296874e-05,
      "step": 38540
    },
    {
      "action_loss_avg": 0.0012329673249041662,
      "epoch": 3.6754608154296874e-05,
      "step": 38540
    },
    {
      "epoch": 3.6754608154296874e-05,
      "step": 38540,
      "training_step_time": 6.977283716201782
    },
    {
      "epoch": 3.675556182861328e-05,
      "model_forward_time": 1.7571182250976562,
      "step": 38541
    },
    {
      "epoch": 3.675556182861328e-05,
      "step": 38541,
      "training_step_time": 6.996886253356934
    },
    {
      "epoch": 3.675651550292969e-05,
      "model_forward_time": 1.7681159973144531,
      "step": 38542
    },
    {
      "epoch": 3.675651550292969e-05,
      "step": 38542,
      "training_step_time": 7.007817983627319
    },
    {
      "epoch": 3.675746917724609e-05,
      "model_forward_time": 1.7634735107421875,
      "step": 38543
    },
    {
      "epoch": 3.675746917724609e-05,
      "step": 38543,
      "training_step_time": 7.628485918045044
    },
    {
      "epoch": 3.67584228515625e-05,
      "model_forward_time": 1.7666916847229004,
      "step": 38544
    },
    {
      "epoch": 3.67584228515625e-05,
      "step": 38544,
      "training_step_time": 7.700113534927368
    },
    {
      "epoch": 3.675937652587891e-05,
      "model_forward_time": 1.7561829090118408,
      "step": 38545
    },
    {
      "epoch": 3.675937652587891e-05,
      "step": 38545,
      "training_step_time": 9.349783658981323
    },
    {
      "epoch": 3.676033020019531e-05,
      "model_forward_time": 1.832512617111206,
      "step": 38546
    },
    {
      "epoch": 3.676033020019531e-05,
      "step": 38546,
      "training_step_time": 8.855810403823853
    },
    {
      "epoch": 3.6761283874511716e-05,
      "model_forward_time": 1.755734920501709,
      "step": 38547
    },
    {
      "epoch": 3.6761283874511716e-05,
      "step": 38547,
      "training_step_time": 9.667486190795898
    },
    {
      "epoch": 3.676223754882813e-05,
      "model_forward_time": 1.7664330005645752,
      "step": 38548
    },
    {
      "epoch": 3.676223754882813e-05,
      "step": 38548,
      "training_step_time": 7.7903289794921875
    },
    {
      "epoch": 3.676319122314453e-05,
      "model_forward_time": 1.414933443069458,
      "step": 38549
    },
    {
      "epoch": 3.676319122314453e-05,
      "step": 38549,
      "training_step_time": 7.62194561958313
    },
    {
      "epoch": 3.6764144897460935e-05,
      "grad_norm": 0.12793314456939697,
      "learning_rate": 7.225792006872831e-06,
      "loss": 0.0645,
      "step": 38550
    },
    {
      "epoch": 3.6764144897460935e-05,
      "model_forward_time": 1.7617440223693848,
      "step": 38550
    },
    {
      "dynamics_loss_avg": 0.06576838567852974,
      "epoch": 3.6764144897460935e-05,
      "step": 38550
    },
    {
      "action_loss_avg": 0.002424228738527745,
      "epoch": 3.6764144897460935e-05,
      "step": 38550
    },
    {
      "epoch": 3.6764144897460935e-05,
      "step": 38550,
      "training_step_time": 6.942700147628784
    },
    {
      "epoch": 3.6765098571777346e-05,
      "model_forward_time": 1.7714800834655762,
      "step": 38551
    },
    {
      "epoch": 3.6765098571777346e-05,
      "step": 38551,
      "training_step_time": 6.951736211776733
    },
    {
      "epoch": 3.676605224609375e-05,
      "model_forward_time": 1.7621753215789795,
      "step": 38552
    },
    {
      "epoch": 3.676605224609375e-05,
      "step": 38552,
      "training_step_time": 7.705764055252075
    },
    {
      "epoch": 3.6767005920410154e-05,
      "model_forward_time": 1.772244930267334,
      "step": 38553
    },
    {
      "epoch": 3.6767005920410154e-05,
      "step": 38553,
      "training_step_time": 7.648984432220459
    },
    {
      "epoch": 3.6767959594726564e-05,
      "model_forward_time": 1.4090244770050049,
      "step": 38554
    },
    {
      "epoch": 3.6767959594726564e-05,
      "step": 38554,
      "training_step_time": 7.7530107498168945
    },
    {
      "epoch": 3.676891326904297e-05,
      "model_forward_time": 2.8535566329956055,
      "step": 38555
    },
    {
      "epoch": 3.676891326904297e-05,
      "step": 38555,
      "training_step_time": 7.8276355266571045
    },
    {
      "epoch": 3.676986694335937e-05,
      "model_forward_time": 1.7693555355072021,
      "step": 38556
    },
    {
      "epoch": 3.676986694335937e-05,
      "step": 38556,
      "training_step_time": 7.850146770477295
    },
    {
      "epoch": 3.677082061767578e-05,
      "model_forward_time": 1.7751100063323975,
      "step": 38557
    },
    {
      "epoch": 3.677082061767578e-05,
      "step": 38557,
      "training_step_time": 7.805488348007202
    },
    {
      "epoch": 3.677177429199219e-05,
      "model_forward_time": 1.7610890865325928,
      "step": 38558
    },
    {
      "epoch": 3.677177429199219e-05,
      "step": 38558,
      "training_step_time": 7.64203143119812
    },
    {
      "epoch": 3.677272796630859e-05,
      "model_forward_time": 1.7638216018676758,
      "step": 38559
    },
    {
      "epoch": 3.677272796630859e-05,
      "step": 38559,
      "training_step_time": 6.924870014190674
    },
    {
      "epoch": 3.6773681640625e-05,
      "grad_norm": 0.18636023998260498,
      "learning_rate": 7.224311282604628e-06,
      "loss": 0.0623,
      "step": 38560
    },
    {
      "epoch": 3.6773681640625e-05,
      "model_forward_time": 1.728973388671875,
      "step": 38560
    },
    {
      "dynamics_loss_avg": 0.046758660953491925,
      "epoch": 3.6773681640625e-05,
      "step": 38560
    },
    {
      "action_loss_avg": 0.001594767376082018,
      "epoch": 3.6773681640625e-05,
      "step": 38560
    },
    {
      "epoch": 3.6773681640625e-05,
      "step": 38560,
      "training_step_time": 6.941570043563843
    },
    {
      "epoch": 3.6774635314941406e-05,
      "model_forward_time": 1.7522871494293213,
      "step": 38561
    },
    {
      "epoch": 3.6774635314941406e-05,
      "step": 38561,
      "training_step_time": 7.655484914779663
    },
    {
      "epoch": 3.677558898925781e-05,
      "model_forward_time": 1.7753829956054688,
      "step": 38562
    },
    {
      "epoch": 3.677558898925781e-05,
      "step": 38562,
      "training_step_time": 7.646342992782593
    },
    {
      "epoch": 3.677654266357422e-05,
      "model_forward_time": 1.7568864822387695,
      "step": 38563
    },
    {
      "epoch": 3.677654266357422e-05,
      "step": 38563,
      "training_step_time": 7.768473148345947
    },
    {
      "epoch": 3.6777496337890625e-05,
      "model_forward_time": 2.8826122283935547,
      "step": 38564
    },
    {
      "epoch": 3.6777496337890625e-05,
      "step": 38564,
      "training_step_time": 10.69299602508545
    },
    {
      "epoch": 3.677845001220703e-05,
      "model_forward_time": 1.7601566314697266,
      "step": 38565
    },
    {
      "epoch": 3.677845001220703e-05,
      "step": 38565,
      "training_step_time": 8.957377433776855
    },
    {
      "epoch": 3.677940368652344e-05,
      "model_forward_time": 1.753387689590454,
      "step": 38566
    },
    {
      "epoch": 3.677940368652344e-05,
      "step": 38566,
      "training_step_time": 9.653497695922852
    },
    {
      "epoch": 3.6780357360839844e-05,
      "model_forward_time": 1.7661254405975342,
      "step": 38567
    },
    {
      "epoch": 3.6780357360839844e-05,
      "step": 38567,
      "training_step_time": 7.716784954071045
    },
    {
      "epoch": 3.678131103515625e-05,
      "model_forward_time": 1.7653582096099854,
      "step": 38568
    },
    {
      "epoch": 3.678131103515625e-05,
      "step": 38568,
      "training_step_time": 6.965825319290161
    },
    {
      "epoch": 3.678226470947266e-05,
      "model_forward_time": 1.766833782196045,
      "step": 38569
    },
    {
      "epoch": 3.678226470947266e-05,
      "step": 38569,
      "training_step_time": 6.928041458129883
    },
    {
      "epoch": 3.678321838378906e-05,
      "grad_norm": 0.15291240811347961,
      "learning_rate": 7.2228303150890855e-06,
      "loss": 0.0629,
      "step": 38570
    },
    {
      "epoch": 3.678321838378906e-05,
      "model_forward_time": 1.7652010917663574,
      "step": 38570
    },
    {
      "dynamics_loss_avg": 0.04693040270358324,
      "epoch": 3.678321838378906e-05,
      "step": 38570
    },
    {
      "action_loss_avg": 0.0012507080449722708,
      "epoch": 3.678321838378906e-05,
      "step": 38570
    },
    {
      "epoch": 3.678321838378906e-05,
      "step": 38570,
      "training_step_time": 7.7001259326934814
    },
    {
      "epoch": 3.678417205810547e-05,
      "model_forward_time": 1.770508050918579,
      "step": 38571
    },
    {
      "epoch": 3.678417205810547e-05,
      "step": 38571,
      "training_step_time": 7.557716131210327
    },
    {
      "epoch": 3.678512573242188e-05,
      "model_forward_time": 1.7611775398254395,
      "step": 38572
    },
    {
      "epoch": 3.678512573242188e-05,
      "step": 38572,
      "training_step_time": 7.755184650421143
    },
    {
      "epoch": 3.678607940673828e-05,
      "model_forward_time": 2.8784968852996826,
      "step": 38573
    },
    {
      "epoch": 3.678607940673828e-05,
      "step": 38573,
      "training_step_time": 8.791610956192017
    },
    {
      "epoch": 3.6787033081054686e-05,
      "model_forward_time": 1.7830886840820312,
      "step": 38574
    },
    {
      "epoch": 3.6787033081054686e-05,
      "step": 38574,
      "training_step_time": 8.808151721954346
    },
    {
      "epoch": 3.6787986755371096e-05,
      "model_forward_time": 1.7641069889068604,
      "step": 38575
    },
    {
      "epoch": 3.6787986755371096e-05,
      "step": 38575,
      "training_step_time": 8.883041381835938
    },
    {
      "epoch": 3.67889404296875e-05,
      "model_forward_time": 1.7565102577209473,
      "step": 38576
    },
    {
      "epoch": 3.67889404296875e-05,
      "step": 38576,
      "training_step_time": 7.706954002380371
    },
    {
      "epoch": 3.6789894104003905e-05,
      "model_forward_time": 1.7846858501434326,
      "step": 38577
    },
    {
      "epoch": 3.6789894104003905e-05,
      "step": 38577,
      "training_step_time": 7.069737672805786
    },
    {
      "epoch": 3.6790847778320315e-05,
      "model_forward_time": 1.7621309757232666,
      "step": 38578
    },
    {
      "epoch": 3.6790847778320315e-05,
      "step": 38578,
      "training_step_time": 6.945263147354126
    },
    {
      "epoch": 3.679180145263672e-05,
      "model_forward_time": 1.7675364017486572,
      "step": 38579
    },
    {
      "epoch": 3.679180145263672e-05,
      "step": 38579,
      "training_step_time": 7.699434041976929
    },
    {
      "epoch": 3.679275512695312e-05,
      "grad_norm": 0.16220331192016602,
      "learning_rate": 7.221349104488161e-06,
      "loss": 0.0635,
      "step": 38580
    },
    {
      "epoch": 3.679275512695312e-05,
      "model_forward_time": 1.7635867595672607,
      "step": 38580
    },
    {
      "dynamics_loss_avg": 0.05424911193549633,
      "epoch": 3.679275512695312e-05,
      "step": 38580
    },
    {
      "action_loss_avg": 0.0017411114415153861,
      "epoch": 3.679275512695312e-05,
      "step": 38580
    },
    {
      "epoch": 3.679275512695312e-05,
      "step": 38580,
      "training_step_time": 8.474694967269897
    },
    {
      "epoch": 3.6793708801269534e-05,
      "model_forward_time": 1.7715864181518555,
      "step": 38581
    },
    {
      "epoch": 3.6793708801269534e-05,
      "step": 38581,
      "training_step_time": 7.791810035705566
    },
    {
      "epoch": 3.679466247558594e-05,
      "model_forward_time": 1.7456748485565186,
      "step": 38582
    },
    {
      "epoch": 3.679466247558594e-05,
      "step": 38582,
      "training_step_time": 8.939455270767212
    },
    {
      "epoch": 3.679561614990234e-05,
      "model_forward_time": 1.8343799114227295,
      "step": 38583
    },
    {
      "epoch": 3.679561614990234e-05,
      "step": 38583,
      "training_step_time": 7.8230390548706055
    },
    {
      "epoch": 3.679656982421875e-05,
      "model_forward_time": 1.7616190910339355,
      "step": 38584
    },
    {
      "epoch": 3.679656982421875e-05,
      "step": 38584,
      "training_step_time": 8.906380653381348
    },
    {
      "epoch": 3.679752349853516e-05,
      "model_forward_time": 1.7737009525299072,
      "step": 38585
    },
    {
      "epoch": 3.679752349853516e-05,
      "step": 38585,
      "training_step_time": 7.861532211303711
    },
    {
      "epoch": 3.679847717285156e-05,
      "model_forward_time": 1.7510967254638672,
      "step": 38586
    },
    {
      "epoch": 3.679847717285156e-05,
      "step": 38586,
      "training_step_time": 7.7044594287872314
    },
    {
      "epoch": 3.679943084716797e-05,
      "model_forward_time": 1.7619428634643555,
      "step": 38587
    },
    {
      "epoch": 3.679943084716797e-05,
      "step": 38587,
      "training_step_time": 7.071660757064819
    },
    {
      "epoch": 3.6800384521484376e-05,
      "model_forward_time": 1.7606031894683838,
      "step": 38588
    },
    {
      "epoch": 3.6800384521484376e-05,
      "step": 38588,
      "training_step_time": 7.766831874847412
    },
    {
      "epoch": 3.680133819580078e-05,
      "model_forward_time": 1.8205628395080566,
      "step": 38589
    },
    {
      "epoch": 3.680133819580078e-05,
      "step": 38589,
      "training_step_time": 7.621427774429321
    },
    {
      "epoch": 3.680229187011719e-05,
      "grad_norm": 0.14230740070343018,
      "learning_rate": 7.21986765096384e-06,
      "loss": 0.0614,
      "step": 38590
    },
    {
      "epoch": 3.680229187011719e-05,
      "model_forward_time": 1.7551586627960205,
      "step": 38590
    },
    {
      "dynamics_loss_avg": 0.05574340336024761,
      "epoch": 3.680229187011719e-05,
      "step": 38590
    },
    {
      "action_loss_avg": 0.0015781577909365295,
      "epoch": 3.680229187011719e-05,
      "step": 38590
    },
    {
      "epoch": 3.680229187011719e-05,
      "step": 38590,
      "training_step_time": 7.821202993392944
    },
    {
      "epoch": 3.6803245544433595e-05,
      "model_forward_time": 1.7688772678375244,
      "step": 38591
    },
    {
      "epoch": 3.6803245544433595e-05,
      "step": 38591,
      "training_step_time": 8.550756692886353
    },
    {
      "epoch": 3.680419921875e-05,
      "model_forward_time": 2.872039794921875,
      "step": 38592
    },
    {
      "epoch": 3.680419921875e-05,
      "step": 38592,
      "training_step_time": 7.8073813915252686
    },
    {
      "epoch": 3.680515289306641e-05,
      "model_forward_time": 1.7605762481689453,
      "step": 38593
    },
    {
      "epoch": 3.680515289306641e-05,
      "step": 38593,
      "training_step_time": 9.374094724655151
    },
    {
      "epoch": 3.6806106567382814e-05,
      "model_forward_time": 1.754370927810669,
      "step": 38594
    },
    {
      "epoch": 3.6806106567382814e-05,
      "step": 38594,
      "training_step_time": 9.56897759437561
    },
    {
      "epoch": 3.680706024169922e-05,
      "model_forward_time": 1.761972188949585,
      "step": 38595
    },
    {
      "epoch": 3.680706024169922e-05,
      "step": 38595,
      "training_step_time": 7.876451730728149
    },
    {
      "epoch": 3.680801391601562e-05,
      "model_forward_time": 1.7594995498657227,
      "step": 38596
    },
    {
      "epoch": 3.680801391601562e-05,
      "step": 38596,
      "training_step_time": 6.894044399261475
    },
    {
      "epoch": 3.680896759033203e-05,
      "model_forward_time": 1.7524943351745605,
      "step": 38597
    },
    {
      "epoch": 3.680896759033203e-05,
      "step": 38597,
      "training_step_time": 7.745255947113037
    },
    {
      "epoch": 3.6809921264648437e-05,
      "model_forward_time": 1.7689476013183594,
      "step": 38598
    },
    {
      "epoch": 3.6809921264648437e-05,
      "step": 38598,
      "training_step_time": 7.342797517776489
    },
    {
      "epoch": 3.681087493896484e-05,
      "model_forward_time": 1.743269920349121,
      "step": 38599
    },
    {
      "epoch": 3.681087493896484e-05,
      "step": 38599,
      "training_step_time": 7.660878896713257
    },
    {
      "epoch": 3.681182861328125e-05,
      "grad_norm": 0.11279815435409546,
      "learning_rate": 7.21838595467813e-06,
      "loss": 0.0569,
      "step": 38600
    },
    {
      "epoch": 3.681182861328125e-05,
      "model_forward_time": 1.7614045143127441,
      "step": 38600
    },
    {
      "dynamics_loss_avg": 0.038437603320926425,
      "epoch": 3.681182861328125e-05,
      "step": 38600
    },
    {
      "action_loss_avg": 0.0013400813593761995,
      "epoch": 3.681182861328125e-05,
      "step": 38600
    },
    {
      "epoch": 3.681182861328125e-05,
      "step": 38600,
      "training_step_time": 7.677427768707275
    },
    {
      "epoch": 3.6812782287597655e-05,
      "model_forward_time": 2.9418699741363525,
      "step": 38601
    },
    {
      "epoch": 3.6812782287597655e-05,
      "step": 38601,
      "training_step_time": 9.600113153457642
    },
    {
      "epoch": 3.681373596191406e-05,
      "model_forward_time": 1.7586069107055664,
      "step": 38602
    },
    {
      "epoch": 3.681373596191406e-05,
      "step": 38602,
      "training_step_time": 8.027493476867676
    },
    {
      "epoch": 3.681468963623047e-05,
      "model_forward_time": 1.7606406211853027,
      "step": 38603
    },
    {
      "epoch": 3.681468963623047e-05,
      "step": 38603,
      "training_step_time": 9.219880819320679
    },
    {
      "epoch": 3.6815643310546874e-05,
      "model_forward_time": 1.7460808753967285,
      "step": 38604
    },
    {
      "epoch": 3.6815643310546874e-05,
      "step": 38604,
      "training_step_time": 9.5649893283844
    },
    {
      "epoch": 3.681659698486328e-05,
      "model_forward_time": 1.7573068141937256,
      "step": 38605
    },
    {
      "epoch": 3.681659698486328e-05,
      "step": 38605,
      "training_step_time": 6.906291246414185
    },
    {
      "epoch": 3.681755065917969e-05,
      "model_forward_time": 1.7502996921539307,
      "step": 38606
    },
    {
      "epoch": 3.681755065917969e-05,
      "step": 38606,
      "training_step_time": 7.683562994003296
    },
    {
      "epoch": 3.681850433349609e-05,
      "model_forward_time": 1.769089698791504,
      "step": 38607
    },
    {
      "epoch": 3.681850433349609e-05,
      "step": 38607,
      "training_step_time": 7.3617775440216064
    },
    {
      "epoch": 3.68194580078125e-05,
      "model_forward_time": 1.826110601425171,
      "step": 38608
    },
    {
      "epoch": 3.68194580078125e-05,
      "step": 38608,
      "training_step_time": 7.79079270362854
    },
    {
      "epoch": 3.682041168212891e-05,
      "model_forward_time": 1.7755353450775146,
      "step": 38609
    },
    {
      "epoch": 3.682041168212891e-05,
      "step": 38609,
      "training_step_time": 7.703785419464111
    },
    {
      "epoch": 3.682136535644531e-05,
      "grad_norm": 0.11916553229093552,
      "learning_rate": 7.216904015793067e-06,
      "loss": 0.06,
      "step": 38610
    },
    {
      "epoch": 3.682136535644531e-05,
      "model_forward_time": 2.90242862701416,
      "step": 38610
    },
    {
      "dynamics_loss_avg": 0.05591280069202185,
      "epoch": 3.682136535644531e-05,
      "step": 38610
    },
    {
      "action_loss_avg": 0.0023077568970620633,
      "epoch": 3.682136535644531e-05,
      "step": 38610
    },
    {
      "epoch": 3.682136535644531e-05,
      "step": 38610,
      "training_step_time": 7.991674184799194
    },
    {
      "epoch": 3.6822319030761716e-05,
      "model_forward_time": 1.8322792053222656,
      "step": 38611
    },
    {
      "epoch": 3.6822319030761716e-05,
      "step": 38611,
      "training_step_time": 8.918139219284058
    },
    {
      "epoch": 3.682327270507813e-05,
      "model_forward_time": 1.7715113162994385,
      "step": 38612
    },
    {
      "epoch": 3.682327270507813e-05,
      "step": 38612,
      "training_step_time": 8.694718360900879
    },
    {
      "epoch": 3.682422637939453e-05,
      "model_forward_time": 1.7589774131774902,
      "step": 38613
    },
    {
      "epoch": 3.682422637939453e-05,
      "step": 38613,
      "training_step_time": 8.674269437789917
    },
    {
      "epoch": 3.6825180053710935e-05,
      "model_forward_time": 1.7511210441589355,
      "step": 38614
    },
    {
      "epoch": 3.6825180053710935e-05,
      "step": 38614,
      "training_step_time": 7.777921199798584
    },
    {
      "epoch": 3.6826133728027346e-05,
      "model_forward_time": 1.7575643062591553,
      "step": 38615
    },
    {
      "epoch": 3.6826133728027346e-05,
      "step": 38615,
      "training_step_time": 7.708467960357666
    },
    {
      "epoch": 3.682708740234375e-05,
      "model_forward_time": 1.7712891101837158,
      "step": 38616
    },
    {
      "epoch": 3.682708740234375e-05,
      "step": 38616,
      "training_step_time": 7.432555913925171
    },
    {
      "epoch": 3.6828041076660154e-05,
      "model_forward_time": 1.7668187618255615,
      "step": 38617
    },
    {
      "epoch": 3.6828041076660154e-05,
      "step": 38617,
      "training_step_time": 7.824559688568115
    },
    {
      "epoch": 3.6828994750976565e-05,
      "model_forward_time": 1.7655582427978516,
      "step": 38618
    },
    {
      "epoch": 3.6828994750976565e-05,
      "step": 38618,
      "training_step_time": 8.757730722427368
    },
    {
      "epoch": 3.682994842529297e-05,
      "model_forward_time": 2.883934259414673,
      "step": 38619
    },
    {
      "epoch": 3.682994842529297e-05,
      "step": 38619,
      "training_step_time": 8.433748245239258
    },
    {
      "epoch": 3.683090209960937e-05,
      "grad_norm": 0.1156611517071724,
      "learning_rate": 7.215421834470713e-06,
      "loss": 0.0548,
      "step": 38620
    },
    {
      "epoch": 3.683090209960937e-05,
      "model_forward_time": 1.8373637199401855,
      "step": 38620
    },
    {
      "dynamics_loss_avg": 0.05372356865555048,
      "epoch": 3.683090209960937e-05,
      "step": 38620
    },
    {
      "action_loss_avg": 0.0017264382651774213,
      "epoch": 3.683090209960937e-05,
      "step": 38620
    },
    {
      "epoch": 3.683090209960937e-05,
      "step": 38620,
      "training_step_time": 10.849371433258057
    },
    {
      "epoch": 3.683185577392578e-05,
      "model_forward_time": 1.770284652709961,
      "step": 38621
    },
    {
      "epoch": 3.683185577392578e-05,
      "step": 38621,
      "training_step_time": 7.863380432128906
    },
    {
      "epoch": 3.683280944824219e-05,
      "model_forward_time": 1.7772879600524902,
      "step": 38622
    },
    {
      "epoch": 3.683280944824219e-05,
      "step": 38622,
      "training_step_time": 7.795686721801758
    },
    {
      "epoch": 3.683376312255859e-05,
      "model_forward_time": 1.7598350048065186,
      "step": 38623
    },
    {
      "epoch": 3.683376312255859e-05,
      "step": 38623,
      "training_step_time": 7.5947089195251465
    },
    {
      "epoch": 3.6834716796875e-05,
      "model_forward_time": 1.7652029991149902,
      "step": 38624
    },
    {
      "epoch": 3.6834716796875e-05,
      "step": 38624,
      "training_step_time": 6.943048477172852
    },
    {
      "epoch": 3.6835670471191406e-05,
      "model_forward_time": 1.7627582550048828,
      "step": 38625
    },
    {
      "epoch": 3.6835670471191406e-05,
      "step": 38625,
      "training_step_time": 7.756934642791748
    },
    {
      "epoch": 3.683662414550781e-05,
      "model_forward_time": 1.760225772857666,
      "step": 38626
    },
    {
      "epoch": 3.683662414550781e-05,
      "step": 38626,
      "training_step_time": 7.72118878364563
    },
    {
      "epoch": 3.683757781982422e-05,
      "model_forward_time": 1.7587363719940186,
      "step": 38627
    },
    {
      "epoch": 3.683757781982422e-05,
      "step": 38627,
      "training_step_time": 7.610238313674927
    },
    {
      "epoch": 3.6838531494140625e-05,
      "model_forward_time": 2.8435351848602295,
      "step": 38628
    },
    {
      "epoch": 3.6838531494140625e-05,
      "step": 38628,
      "training_step_time": 7.861047744750977
    },
    {
      "epoch": 3.683948516845703e-05,
      "model_forward_time": 1.821314811706543,
      "step": 38629
    },
    {
      "epoch": 3.683948516845703e-05,
      "step": 38629,
      "training_step_time": 9.48469877243042
    },
    {
      "epoch": 3.684043884277344e-05,
      "grad_norm": 0.15004020929336548,
      "learning_rate": 7.213939410873159e-06,
      "loss": 0.0572,
      "step": 38630
    },
    {
      "epoch": 3.684043884277344e-05,
      "model_forward_time": 1.7663624286651611,
      "step": 38630
    },
    {
      "dynamics_loss_avg": 0.05533344056457281,
      "epoch": 3.684043884277344e-05,
      "step": 38630
    },
    {
      "action_loss_avg": 0.0017363616265356542,
      "epoch": 3.684043884277344e-05,
      "step": 38630
    },
    {
      "epoch": 3.684043884277344e-05,
      "step": 38630,
      "training_step_time": 8.705402374267578
    },
    {
      "epoch": 3.6841392517089844e-05,
      "model_forward_time": 1.7943975925445557,
      "step": 38631
    },
    {
      "epoch": 3.6841392517089844e-05,
      "step": 38631,
      "training_step_time": 7.755288600921631
    },
    {
      "epoch": 3.684234619140625e-05,
      "model_forward_time": 1.7482476234436035,
      "step": 38632
    },
    {
      "epoch": 3.684234619140625e-05,
      "step": 38632,
      "training_step_time": 7.67031717300415
    },
    {
      "epoch": 3.684329986572266e-05,
      "model_forward_time": 1.7622861862182617,
      "step": 38633
    },
    {
      "epoch": 3.684329986572266e-05,
      "step": 38633,
      "training_step_time": 6.939030647277832
    },
    {
      "epoch": 3.684425354003906e-05,
      "model_forward_time": 1.771796464920044,
      "step": 38634
    },
    {
      "epoch": 3.684425354003906e-05,
      "step": 38634,
      "training_step_time": 7.563433885574341
    },
    {
      "epoch": 3.684520721435547e-05,
      "model_forward_time": 1.7669048309326172,
      "step": 38635
    },
    {
      "epoch": 3.684520721435547e-05,
      "step": 38635,
      "training_step_time": 7.723494052886963
    },
    {
      "epoch": 3.684616088867188e-05,
      "model_forward_time": 1.76828932762146,
      "step": 38636
    },
    {
      "epoch": 3.684616088867188e-05,
      "step": 38636,
      "training_step_time": 7.619579553604126
    },
    {
      "epoch": 3.684711456298828e-05,
      "model_forward_time": 1.7597239017486572,
      "step": 38637
    },
    {
      "epoch": 3.684711456298828e-05,
      "step": 38637,
      "training_step_time": 9.667027473449707
    },
    {
      "epoch": 3.6848068237304686e-05,
      "model_forward_time": 2.9088032245635986,
      "step": 38638
    },
    {
      "epoch": 3.6848068237304686e-05,
      "step": 38638,
      "training_step_time": 9.795175313949585
    },
    {
      "epoch": 3.6849021911621097e-05,
      "model_forward_time": 1.7709376811981201,
      "step": 38639
    },
    {
      "epoch": 3.6849021911621097e-05,
      "step": 38639,
      "training_step_time": 7.921375751495361
    },
    {
      "epoch": 3.68499755859375e-05,
      "grad_norm": 0.11271549761295319,
      "learning_rate": 7.2124567451625195e-06,
      "loss": 0.0608,
      "step": 38640
    },
    {
      "epoch": 3.68499755859375e-05,
      "model_forward_time": 1.7666692733764648,
      "step": 38640
    },
    {
      "dynamics_loss_avg": 0.06000498943030834,
      "epoch": 3.68499755859375e-05,
      "step": 38640
    },
    {
      "action_loss_avg": 0.003037809970555827,
      "epoch": 3.68499755859375e-05,
      "step": 38640
    },
    {
      "epoch": 3.68499755859375e-05,
      "step": 38640,
      "training_step_time": 7.82724928855896
    },
    {
      "epoch": 3.6850929260253905e-05,
      "model_forward_time": 1.7661938667297363,
      "step": 38641
    },
    {
      "epoch": 3.6850929260253905e-05,
      "step": 38641,
      "training_step_time": 7.640629529953003
    },
    {
      "epoch": 3.6851882934570315e-05,
      "model_forward_time": 1.7621002197265625,
      "step": 38642
    },
    {
      "epoch": 3.6851882934570315e-05,
      "step": 38642,
      "training_step_time": 7.720911026000977
    },
    {
      "epoch": 3.685283660888672e-05,
      "model_forward_time": 1.765643835067749,
      "step": 38643
    },
    {
      "epoch": 3.685283660888672e-05,
      "step": 38643,
      "training_step_time": 7.751567125320435
    },
    {
      "epoch": 3.6853790283203123e-05,
      "model_forward_time": 1.7679290771484375,
      "step": 38644
    },
    {
      "epoch": 3.6853790283203123e-05,
      "step": 38644,
      "training_step_time": 7.775806665420532
    },
    {
      "epoch": 3.6854743957519534e-05,
      "model_forward_time": 1.7573051452636719,
      "step": 38645
    },
    {
      "epoch": 3.6854743957519534e-05,
      "step": 38645,
      "training_step_time": 7.681930065155029
    },
    {
      "epoch": 3.685569763183594e-05,
      "model_forward_time": 1.758589744567871,
      "step": 38646
    },
    {
      "epoch": 3.685569763183594e-05,
      "step": 38646,
      "training_step_time": 7.793874025344849
    },
    {
      "epoch": 3.685665130615234e-05,
      "model_forward_time": 2.90834903717041,
      "step": 38647
    },
    {
      "epoch": 3.685665130615234e-05,
      "step": 38647,
      "training_step_time": 9.994139909744263
    },
    {
      "epoch": 3.685760498046875e-05,
      "model_forward_time": 1.7617154121398926,
      "step": 38648
    },
    {
      "epoch": 3.685760498046875e-05,
      "step": 38648,
      "training_step_time": 8.503730058670044
    },
    {
      "epoch": 3.685855865478516e-05,
      "model_forward_time": 1.764512300491333,
      "step": 38649
    },
    {
      "epoch": 3.685855865478516e-05,
      "step": 38649,
      "training_step_time": 9.388330221176147
    },
    {
      "epoch": 3.685951232910156e-05,
      "grad_norm": 0.12701930105686188,
      "learning_rate": 7.210973837500936e-06,
      "loss": 0.0593,
      "step": 38650
    },
    {
      "epoch": 3.685951232910156e-05,
      "model_forward_time": 1.7699077129364014,
      "step": 38650
    },
    {
      "dynamics_loss_avg": 0.0568967517465353,
      "epoch": 3.685951232910156e-05,
      "step": 38650
    },
    {
      "action_loss_avg": 0.001712578916340135,
      "epoch": 3.685951232910156e-05,
      "step": 38650
    },
    {
      "epoch": 3.685951232910156e-05,
      "step": 38650,
      "training_step_time": 7.716090202331543
    },
    {
      "epoch": 3.686046600341797e-05,
      "model_forward_time": 1.7439792156219482,
      "step": 38651
    },
    {
      "epoch": 3.686046600341797e-05,
      "step": 38651,
      "training_step_time": 7.738865852355957
    },
    {
      "epoch": 3.6861419677734376e-05,
      "model_forward_time": 1.7547757625579834,
      "step": 38652
    },
    {
      "epoch": 3.6861419677734376e-05,
      "step": 38652,
      "training_step_time": 7.7138707637786865
    },
    {
      "epoch": 3.686237335205078e-05,
      "model_forward_time": 1.7479782104492188,
      "step": 38653
    },
    {
      "epoch": 3.686237335205078e-05,
      "step": 38653,
      "training_step_time": 7.772627830505371
    },
    {
      "epoch": 3.686332702636719e-05,
      "model_forward_time": 1.7749338150024414,
      "step": 38654
    },
    {
      "epoch": 3.686332702636719e-05,
      "step": 38654,
      "training_step_time": 7.643369436264038
    },
    {
      "epoch": 3.6864280700683595e-05,
      "model_forward_time": 1.7350623607635498,
      "step": 38655
    },
    {
      "epoch": 3.6864280700683595e-05,
      "step": 38655,
      "training_step_time": 8.419028520584106
    },
    {
      "epoch": 3.6865234375e-05,
      "model_forward_time": 2.304642915725708,
      "step": 38656
    },
    {
      "epoch": 3.6865234375e-05,
      "step": 38656,
      "training_step_time": 8.582849979400635
    },
    {
      "epoch": 3.68661880493164e-05,
      "model_forward_time": 1.7353451251983643,
      "step": 38657
    },
    {
      "epoch": 3.68661880493164e-05,
      "step": 38657,
      "training_step_time": 8.095722913742065
    },
    {
      "epoch": 3.6867141723632814e-05,
      "model_forward_time": 1.732407808303833,
      "step": 38658
    },
    {
      "epoch": 3.6867141723632814e-05,
      "step": 38658,
      "training_step_time": 8.850053787231445
    },
    {
      "epoch": 3.686809539794922e-05,
      "model_forward_time": 1.7248926162719727,
      "step": 38659
    },
    {
      "epoch": 3.686809539794922e-05,
      "step": 38659,
      "training_step_time": 7.803560972213745
    },
    {
      "epoch": 3.686904907226562e-05,
      "grad_norm": 0.15115924179553986,
      "learning_rate": 7.209490688050579e-06,
      "loss": 0.0594,
      "step": 38660
    },
    {
      "epoch": 3.686904907226562e-05,
      "model_forward_time": 1.7538299560546875,
      "step": 38660
    },
    {
      "dynamics_loss_avg": 0.057302871532738206,
      "epoch": 3.686904907226562e-05,
      "step": 38660
    },
    {
      "action_loss_avg": 0.0014130681287497283,
      "epoch": 3.686904907226562e-05,
      "step": 38660
    },
    {
      "epoch": 3.686904907226562e-05,
      "step": 38660,
      "training_step_time": 7.72669529914856
    },
    {
      "epoch": 3.687000274658203e-05,
      "model_forward_time": 1.7676477432250977,
      "step": 38661
    },
    {
      "epoch": 3.687000274658203e-05,
      "step": 38661,
      "training_step_time": 6.857842445373535
    },
    {
      "epoch": 3.6870956420898437e-05,
      "model_forward_time": 1.8177566528320312,
      "step": 38662
    },
    {
      "epoch": 3.6870956420898437e-05,
      "step": 38662,
      "training_step_time": 7.718772649765015
    },
    {
      "epoch": 3.687191009521484e-05,
      "model_forward_time": 1.7531342506408691,
      "step": 38663
    },
    {
      "epoch": 3.687191009521484e-05,
      "step": 38663,
      "training_step_time": 7.775501489639282
    },
    {
      "epoch": 3.687286376953125e-05,
      "model_forward_time": 1.7790162563323975,
      "step": 38664
    },
    {
      "epoch": 3.687286376953125e-05,
      "step": 38664,
      "training_step_time": 7.740178823471069
    },
    {
      "epoch": 3.6873817443847655e-05,
      "model_forward_time": 2.3499019145965576,
      "step": 38665
    },
    {
      "epoch": 3.6873817443847655e-05,
      "step": 38665,
      "training_step_time": 7.904537916183472
    },
    {
      "epoch": 3.687477111816406e-05,
      "model_forward_time": 1.7582364082336426,
      "step": 38666
    },
    {
      "epoch": 3.687477111816406e-05,
      "step": 38666,
      "training_step_time": 9.031289100646973
    },
    {
      "epoch": 3.687572479248047e-05,
      "model_forward_time": 1.762387752532959,
      "step": 38667
    },
    {
      "epoch": 3.687572479248047e-05,
      "step": 38667,
      "training_step_time": 8.88713788986206
    },
    {
      "epoch": 3.6876678466796874e-05,
      "model_forward_time": 1.7666876316070557,
      "step": 38668
    },
    {
      "epoch": 3.6876678466796874e-05,
      "step": 38668,
      "training_step_time": 7.7605836391448975
    },
    {
      "epoch": 3.687763214111328e-05,
      "model_forward_time": 1.7681853771209717,
      "step": 38669
    },
    {
      "epoch": 3.687763214111328e-05,
      "step": 38669,
      "training_step_time": 7.6562018394470215
    },
    {
      "epoch": 3.687858581542969e-05,
      "grad_norm": 0.11496075242757797,
      "learning_rate": 7.208007296973641e-06,
      "loss": 0.0553,
      "step": 38670
    },
    {
      "epoch": 3.687858581542969e-05,
      "model_forward_time": 1.759185552597046,
      "step": 38670
    },
    {
      "dynamics_loss_avg": 0.04211068488657475,
      "epoch": 3.687858581542969e-05,
      "step": 38670
    },
    {
      "action_loss_avg": 0.0014572915708413348,
      "epoch": 3.687858581542969e-05,
      "step": 38670
    },
    {
      "epoch": 3.687858581542969e-05,
      "step": 38670,
      "training_step_time": 7.0048840045928955
    },
    {
      "epoch": 3.687953948974609e-05,
      "model_forward_time": 1.8197448253631592,
      "step": 38671
    },
    {
      "epoch": 3.687953948974609e-05,
      "step": 38671,
      "training_step_time": 7.703242301940918
    },
    {
      "epoch": 3.68804931640625e-05,
      "model_forward_time": 1.7706224918365479,
      "step": 38672
    },
    {
      "epoch": 3.68804931640625e-05,
      "step": 38672,
      "training_step_time": 7.78132438659668
    },
    {
      "epoch": 3.688144683837891e-05,
      "model_forward_time": 1.7679648399353027,
      "step": 38673
    },
    {
      "epoch": 3.688144683837891e-05,
      "step": 38673,
      "training_step_time": 8.874955415725708
    },
    {
      "epoch": 3.688240051269531e-05,
      "model_forward_time": 2.351133346557617,
      "step": 38674
    },
    {
      "epoch": 3.688240051269531e-05,
      "step": 38674,
      "training_step_time": 8.823354482650757
    },
    {
      "epoch": 3.6883354187011716e-05,
      "model_forward_time": 1.752549171447754,
      "step": 38675
    },
    {
      "epoch": 3.6883354187011716e-05,
      "step": 38675,
      "training_step_time": 9.563261032104492
    },
    {
      "epoch": 3.688430786132813e-05,
      "model_forward_time": 1.7826168537139893,
      "step": 38676
    },
    {
      "epoch": 3.688430786132813e-05,
      "step": 38676,
      "training_step_time": 8.55455756187439
    },
    {
      "epoch": 3.688526153564453e-05,
      "model_forward_time": 1.762115478515625,
      "step": 38677
    },
    {
      "epoch": 3.688526153564453e-05,
      "step": 38677,
      "training_step_time": 8.756821632385254
    },
    {
      "epoch": 3.6886215209960935e-05,
      "model_forward_time": 1.7702901363372803,
      "step": 38678
    },
    {
      "epoch": 3.6886215209960935e-05,
      "step": 38678,
      "training_step_time": 8.61348009109497
    },
    {
      "epoch": 3.6887168884277346e-05,
      "model_forward_time": 1.7691841125488281,
      "step": 38679
    },
    {
      "epoch": 3.6887168884277346e-05,
      "step": 38679,
      "training_step_time": 7.727433443069458
    },
    {
      "epoch": 3.688812255859375e-05,
      "grad_norm": 0.12150002270936966,
      "learning_rate": 7.206523664432346e-06,
      "loss": 0.0568,
      "step": 38680
    },
    {
      "epoch": 3.688812255859375e-05,
      "model_forward_time": 1.812948226928711,
      "step": 38680
    },
    {
      "dynamics_loss_avg": 0.058133600559085605,
      "epoch": 3.688812255859375e-05,
      "step": 38680
    },
    {
      "action_loss_avg": 0.0020707458548713475,
      "epoch": 3.688812255859375e-05,
      "step": 38680
    },
    {
      "epoch": 3.688812255859375e-05,
      "step": 38680,
      "training_step_time": 7.738552570343018
    },
    {
      "epoch": 3.6889076232910154e-05,
      "model_forward_time": 1.7711560726165771,
      "step": 38681
    },
    {
      "epoch": 3.6889076232910154e-05,
      "step": 38681,
      "training_step_time": 7.851386070251465
    },
    {
      "epoch": 3.6890029907226565e-05,
      "model_forward_time": 1.7711522579193115,
      "step": 38682
    },
    {
      "epoch": 3.6890029907226565e-05,
      "step": 38682,
      "training_step_time": 7.882818698883057
    },
    {
      "epoch": 3.689098358154297e-05,
      "model_forward_time": 1.7753651142120361,
      "step": 38683
    },
    {
      "epoch": 3.689098358154297e-05,
      "step": 38683,
      "training_step_time": 9.355489253997803
    },
    {
      "epoch": 3.689193725585937e-05,
      "model_forward_time": 1.767826795578003,
      "step": 38684
    },
    {
      "epoch": 3.689193725585937e-05,
      "step": 38684,
      "training_step_time": 8.554556369781494
    },
    {
      "epoch": 3.6892890930175783e-05,
      "model_forward_time": 1.7624366283416748,
      "step": 38685
    },
    {
      "epoch": 3.6892890930175783e-05,
      "step": 38685,
      "training_step_time": 7.998414039611816
    },
    {
      "epoch": 3.689384460449219e-05,
      "model_forward_time": 1.7685399055480957,
      "step": 38686
    },
    {
      "epoch": 3.689384460449219e-05,
      "step": 38686,
      "training_step_time": 7.8562541007995605
    },
    {
      "epoch": 3.689479827880859e-05,
      "model_forward_time": 1.7837345600128174,
      "step": 38687
    },
    {
      "epoch": 3.689479827880859e-05,
      "step": 38687,
      "training_step_time": 7.810578107833862
    },
    {
      "epoch": 3.6895751953125e-05,
      "model_forward_time": 1.7669994831085205,
      "step": 38688
    },
    {
      "epoch": 3.6895751953125e-05,
      "step": 38688,
      "training_step_time": 7.617464303970337
    },
    {
      "epoch": 3.6896705627441406e-05,
      "model_forward_time": 1.8063349723815918,
      "step": 38689
    },
    {
      "epoch": 3.6896705627441406e-05,
      "step": 38689,
      "training_step_time": 7.735866069793701
    },
    {
      "epoch": 3.689765930175781e-05,
      "grad_norm": 0.17194882035255432,
      "learning_rate": 7.205039790588939e-06,
      "loss": 0.0598,
      "step": 38690
    },
    {
      "epoch": 3.689765930175781e-05,
      "model_forward_time": 1.7656831741333008,
      "step": 38690
    },
    {
      "dynamics_loss_avg": 0.05565239768475294,
      "epoch": 3.689765930175781e-05,
      "step": 38690
    },
    {
      "action_loss_avg": 0.002137709141243249,
      "epoch": 3.689765930175781e-05,
      "step": 38690
    },
    {
      "epoch": 3.689765930175781e-05,
      "step": 38690,
      "training_step_time": 7.699915170669556
    },
    {
      "epoch": 3.689861297607422e-05,
      "model_forward_time": 1.7819182872772217,
      "step": 38691
    },
    {
      "epoch": 3.689861297607422e-05,
      "step": 38691,
      "training_step_time": 7.8970396518707275
    },
    {
      "epoch": 3.6899566650390625e-05,
      "model_forward_time": 1.7645220756530762,
      "step": 38692
    },
    {
      "epoch": 3.6899566650390625e-05,
      "step": 38692,
      "training_step_time": 10.214496374130249
    },
    {
      "epoch": 3.690052032470703e-05,
      "model_forward_time": 1.7687501907348633,
      "step": 38693
    },
    {
      "epoch": 3.690052032470703e-05,
      "step": 38693,
      "training_step_time": 7.948106050491333
    },
    {
      "epoch": 3.690147399902344e-05,
      "model_forward_time": 1.7681756019592285,
      "step": 38694
    },
    {
      "epoch": 3.690147399902344e-05,
      "step": 38694,
      "training_step_time": 8.01435375213623
    },
    {
      "epoch": 3.6902427673339844e-05,
      "model_forward_time": 1.7493023872375488,
      "step": 38695
    },
    {
      "epoch": 3.6902427673339844e-05,
      "step": 38695,
      "training_step_time": 8.728010892868042
    },
    {
      "epoch": 3.690338134765625e-05,
      "model_forward_time": 1.7679121494293213,
      "step": 38696
    },
    {
      "epoch": 3.690338134765625e-05,
      "step": 38696,
      "training_step_time": 8.362865686416626
    },
    {
      "epoch": 3.690433502197266e-05,
      "model_forward_time": 1.7557504177093506,
      "step": 38697
    },
    {
      "epoch": 3.690433502197266e-05,
      "step": 38697,
      "training_step_time": 7.751270532608032
    },
    {
      "epoch": 3.690528869628906e-05,
      "model_forward_time": 1.798290729522705,
      "step": 38698
    },
    {
      "epoch": 3.690528869628906e-05,
      "step": 38698,
      "training_step_time": 7.645824432373047
    },
    {
      "epoch": 3.690624237060547e-05,
      "model_forward_time": 1.7726194858551025,
      "step": 38699
    },
    {
      "epoch": 3.690624237060547e-05,
      "step": 38699,
      "training_step_time": 7.569589614868164
    },
    {
      "epoch": 3.690719604492188e-05,
      "grad_norm": 0.18325428664684296,
      "learning_rate": 7.2035556756056965e-06,
      "loss": 0.0591,
      "step": 38700
    },
    {
      "epoch": 3.690719604492188e-05,
      "model_forward_time": 1.763580083847046,
      "step": 38700
    },
    {
      "dynamics_loss_avg": 0.057948078773915765,
      "epoch": 3.690719604492188e-05,
      "step": 38700
    },
    {
      "action_loss_avg": 0.001729344908380881,
      "epoch": 3.690719604492188e-05,
      "step": 38700
    },
    {
      "epoch": 3.690719604492188e-05,
      "step": 38700,
      "training_step_time": 7.8193910121917725
    },
    {
      "epoch": 3.690814971923828e-05,
      "model_forward_time": 1.7630012035369873,
      "step": 38701
    },
    {
      "epoch": 3.690814971923828e-05,
      "step": 38701,
      "training_step_time": 8.672410488128662
    },
    {
      "epoch": 3.6909103393554686e-05,
      "model_forward_time": 1.777005672454834,
      "step": 38702
    },
    {
      "epoch": 3.6909103393554686e-05,
      "step": 38702,
      "training_step_time": 8.407575130462646
    },
    {
      "epoch": 3.6910057067871097e-05,
      "model_forward_time": 1.7811505794525146,
      "step": 38703
    },
    {
      "epoch": 3.6910057067871097e-05,
      "step": 38703,
      "training_step_time": 8.834925651550293
    },
    {
      "epoch": 3.69110107421875e-05,
      "model_forward_time": 1.74167799949646,
      "step": 38704
    },
    {
      "epoch": 3.69110107421875e-05,
      "step": 38704,
      "training_step_time": 8.349730730056763
    },
    {
      "epoch": 3.6911964416503905e-05,
      "model_forward_time": 1.7721476554870605,
      "step": 38705
    },
    {
      "epoch": 3.6911964416503905e-05,
      "step": 38705,
      "training_step_time": 7.646917104721069
    },
    {
      "epoch": 3.6912918090820315e-05,
      "model_forward_time": 1.3833720684051514,
      "step": 38706
    },
    {
      "epoch": 3.6912918090820315e-05,
      "step": 38706,
      "training_step_time": 7.696512937545776
    },
    {
      "epoch": 3.691387176513672e-05,
      "model_forward_time": 1.7505121231079102,
      "step": 38707
    },
    {
      "epoch": 3.691387176513672e-05,
      "step": 38707,
      "training_step_time": 7.577491283416748
    },
    {
      "epoch": 3.6914825439453123e-05,
      "model_forward_time": 1.822575569152832,
      "step": 38708
    },
    {
      "epoch": 3.6914825439453123e-05,
      "step": 38708,
      "training_step_time": 7.671103477478027
    },
    {
      "epoch": 3.6915779113769534e-05,
      "model_forward_time": 1.75761079788208,
      "step": 38709
    },
    {
      "epoch": 3.6915779113769534e-05,
      "step": 38709,
      "training_step_time": 8.846683740615845
    },
    {
      "epoch": 3.691673278808594e-05,
      "grad_norm": 0.16741988062858582,
      "learning_rate": 7.202071319644918e-06,
      "loss": 0.0616,
      "step": 38710
    },
    {
      "epoch": 3.691673278808594e-05,
      "model_forward_time": 1.7627918720245361,
      "step": 38710
    },
    {
      "dynamics_loss_avg": 0.05557422153651714,
      "epoch": 3.691673278808594e-05,
      "step": 38710
    },
    {
      "action_loss_avg": 0.002247153577627614,
      "epoch": 3.691673278808594e-05,
      "step": 38710
    },
    {
      "epoch": 3.691673278808594e-05,
      "step": 38710,
      "training_step_time": 7.707486391067505
    },
    {
      "epoch": 3.691768646240234e-05,
      "model_forward_time": 2.343829393386841,
      "step": 38711
    },
    {
      "epoch": 3.691768646240234e-05,
      "step": 38711,
      "training_step_time": 7.830447673797607
    },
    {
      "epoch": 3.691864013671875e-05,
      "model_forward_time": 1.7582833766937256,
      "step": 38712
    },
    {
      "epoch": 3.691864013671875e-05,
      "step": 38712,
      "training_step_time": 7.879594564437866
    },
    {
      "epoch": 3.691959381103516e-05,
      "model_forward_time": 1.7688226699829102,
      "step": 38713
    },
    {
      "epoch": 3.691959381103516e-05,
      "step": 38713,
      "training_step_time": 8.474198579788208
    },
    {
      "epoch": 3.692054748535156e-05,
      "model_forward_time": 1.7613046169281006,
      "step": 38714
    },
    {
      "epoch": 3.692054748535156e-05,
      "step": 38714,
      "training_step_time": 7.796778202056885
    },
    {
      "epoch": 3.692150115966797e-05,
      "model_forward_time": 1.762120008468628,
      "step": 38715
    },
    {
      "epoch": 3.692150115966797e-05,
      "step": 38715,
      "training_step_time": 8.432528972625732
    },
    {
      "epoch": 3.6922454833984376e-05,
      "model_forward_time": 1.7753398418426514,
      "step": 38716
    },
    {
      "epoch": 3.6922454833984376e-05,
      "step": 38716,
      "training_step_time": 7.6597301959991455
    },
    {
      "epoch": 3.692340850830078e-05,
      "model_forward_time": 1.810685634613037,
      "step": 38717
    },
    {
      "epoch": 3.692340850830078e-05,
      "step": 38717,
      "training_step_time": 7.692270755767822
    },
    {
      "epoch": 3.692436218261719e-05,
      "model_forward_time": 1.7617661952972412,
      "step": 38718
    },
    {
      "epoch": 3.692436218261719e-05,
      "step": 38718,
      "training_step_time": 7.649945259094238
    },
    {
      "epoch": 3.6925315856933595e-05,
      "model_forward_time": 1.7695860862731934,
      "step": 38719
    },
    {
      "epoch": 3.6925315856933595e-05,
      "step": 38719,
      "training_step_time": 7.812993764877319
    },
    {
      "epoch": 3.692626953125e-05,
      "grad_norm": 0.1661807745695114,
      "learning_rate": 7.200586722868932e-06,
      "loss": 0.0646,
      "step": 38720
    },
    {
      "epoch": 3.692626953125e-05,
      "model_forward_time": 2.379784107208252,
      "step": 38720
    },
    {
      "dynamics_loss_avg": 0.04831673940643668,
      "epoch": 3.692626953125e-05,
      "step": 38720
    },
    {
      "action_loss_avg": 0.0017264094843994827,
      "epoch": 3.692626953125e-05,
      "step": 38720
    },
    {
      "epoch": 3.692626953125e-05,
      "step": 38720,
      "training_step_time": 7.754300117492676
    },
    {
      "epoch": 3.69272232055664e-05,
      "model_forward_time": 1.765528678894043,
      "step": 38721
    },
    {
      "epoch": 3.69272232055664e-05,
      "step": 38721,
      "training_step_time": 8.29560923576355
    },
    {
      "epoch": 3.6928176879882814e-05,
      "model_forward_time": 1.772171974182129,
      "step": 38722
    },
    {
      "epoch": 3.6928176879882814e-05,
      "step": 38722,
      "training_step_time": 9.215845108032227
    },
    {
      "epoch": 3.692913055419922e-05,
      "model_forward_time": 1.766892671585083,
      "step": 38723
    },
    {
      "epoch": 3.692913055419922e-05,
      "step": 38723,
      "training_step_time": 8.33859133720398
    },
    {
      "epoch": 3.693008422851562e-05,
      "model_forward_time": 1.7682318687438965,
      "step": 38724
    },
    {
      "epoch": 3.693008422851562e-05,
      "step": 38724,
      "training_step_time": 7.67961573600769
    },
    {
      "epoch": 3.693103790283203e-05,
      "model_forward_time": 1.767261266708374,
      "step": 38725
    },
    {
      "epoch": 3.693103790283203e-05,
      "step": 38725,
      "training_step_time": 7.551422595977783
    },
    {
      "epoch": 3.693199157714844e-05,
      "model_forward_time": 1.8232331275939941,
      "step": 38726
    },
    {
      "epoch": 3.693199157714844e-05,
      "step": 38726,
      "training_step_time": 7.524339437484741
    },
    {
      "epoch": 3.693294525146484e-05,
      "model_forward_time": 1.768674373626709,
      "step": 38727
    },
    {
      "epoch": 3.693294525146484e-05,
      "step": 38727,
      "training_step_time": 7.5514397621154785
    },
    {
      "epoch": 3.693389892578125e-05,
      "model_forward_time": 1.7167317867279053,
      "step": 38728
    },
    {
      "epoch": 3.693389892578125e-05,
      "step": 38728,
      "training_step_time": 7.725848913192749
    },
    {
      "epoch": 3.6934852600097655e-05,
      "model_forward_time": 2.2948317527770996,
      "step": 38729
    },
    {
      "epoch": 3.6934852600097655e-05,
      "step": 38729,
      "training_step_time": 8.398080825805664
    },
    {
      "epoch": 3.693580627441406e-05,
      "grad_norm": 0.14985334873199463,
      "learning_rate": 7.199101885440089e-06,
      "loss": 0.0651,
      "step": 38730
    },
    {
      "epoch": 3.693580627441406e-05,
      "model_forward_time": 1.7198309898376465,
      "step": 38730
    },
    {
      "dynamics_loss_avg": 0.0437429029494524,
      "epoch": 3.693580627441406e-05,
      "step": 38730
    },
    {
      "action_loss_avg": 0.0012919431610498577,
      "epoch": 3.693580627441406e-05,
      "step": 38730
    },
    {
      "epoch": 3.693580627441406e-05,
      "step": 38730,
      "training_step_time": 8.52649450302124
    },
    {
      "epoch": 3.693675994873047e-05,
      "model_forward_time": 1.7359588146209717,
      "step": 38731
    },
    {
      "epoch": 3.693675994873047e-05,
      "step": 38731,
      "training_step_time": 8.617543458938599
    },
    {
      "epoch": 3.6937713623046874e-05,
      "model_forward_time": 1.7377254962921143,
      "step": 38732
    },
    {
      "epoch": 3.6937713623046874e-05,
      "step": 38732,
      "training_step_time": 9.28034257888794
    },
    {
      "epoch": 3.693866729736328e-05,
      "model_forward_time": 1.725989818572998,
      "step": 38733
    },
    {
      "epoch": 3.693866729736328e-05,
      "step": 38733,
      "training_step_time": 7.814981698989868
    },
    {
      "epoch": 3.693962097167969e-05,
      "model_forward_time": 1.738086462020874,
      "step": 38734
    },
    {
      "epoch": 3.693962097167969e-05,
      "step": 38734,
      "training_step_time": 7.744961977005005
    },
    {
      "epoch": 3.694057464599609e-05,
      "model_forward_time": 1.7572839260101318,
      "step": 38735
    },
    {
      "epoch": 3.694057464599609e-05,
      "step": 38735,
      "training_step_time": 9.283393859863281
    },
    {
      "epoch": 3.69415283203125e-05,
      "model_forward_time": 1.7336480617523193,
      "step": 38736
    },
    {
      "epoch": 3.69415283203125e-05,
      "step": 38736,
      "training_step_time": 8.449177026748657
    },
    {
      "epoch": 3.694248199462891e-05,
      "model_forward_time": 1.7322847843170166,
      "step": 38737
    },
    {
      "epoch": 3.694248199462891e-05,
      "step": 38737,
      "training_step_time": 8.912375450134277
    },
    {
      "epoch": 3.694343566894531e-05,
      "model_forward_time": 2.2973289489746094,
      "step": 38738
    },
    {
      "epoch": 3.694343566894531e-05,
      "step": 38738,
      "training_step_time": 7.643213510513306
    },
    {
      "epoch": 3.6944389343261716e-05,
      "model_forward_time": 1.7339985370635986,
      "step": 38739
    },
    {
      "epoch": 3.6944389343261716e-05,
      "step": 38739,
      "training_step_time": 8.366864204406738
    },
    {
      "epoch": 3.694534301757813e-05,
      "grad_norm": 0.10558699816465378,
      "learning_rate": 7.197616807520771e-06,
      "loss": 0.0626,
      "step": 38740
    },
    {
      "epoch": 3.694534301757813e-05,
      "model_forward_time": 1.7224030494689941,
      "step": 38740
    },
    {
      "dynamics_loss_avg": 0.05755192171782255,
      "epoch": 3.694534301757813e-05,
      "step": 38740
    },
    {
      "action_loss_avg": 0.0011784153059124947,
      "epoch": 3.694534301757813e-05,
      "step": 38740
    },
    {
      "epoch": 3.694534301757813e-05,
      "step": 38740,
      "training_step_time": 8.56006669998169
    },
    {
      "epoch": 3.694629669189453e-05,
      "model_forward_time": 1.7236974239349365,
      "step": 38741
    },
    {
      "epoch": 3.694629669189453e-05,
      "step": 38741,
      "training_step_time": 7.69866418838501
    },
    {
      "epoch": 3.6947250366210935e-05,
      "model_forward_time": 1.735107421875,
      "step": 38742
    },
    {
      "epoch": 3.6947250366210935e-05,
      "step": 38742,
      "training_step_time": 7.6886515617370605
    },
    {
      "epoch": 3.6948204040527346e-05,
      "model_forward_time": 1.3557627201080322,
      "step": 38743
    },
    {
      "epoch": 3.6948204040527346e-05,
      "step": 38743,
      "training_step_time": 7.645668983459473
    },
    {
      "epoch": 3.694915771484375e-05,
      "model_forward_time": 1.7425494194030762,
      "step": 38744
    },
    {
      "epoch": 3.694915771484375e-05,
      "step": 38744,
      "training_step_time": 7.55796480178833
    },
    {
      "epoch": 3.6950111389160154e-05,
      "model_forward_time": 1.0089530944824219,
      "step": 38745
    },
    {
      "epoch": 3.6950111389160154e-05,
      "step": 38745,
      "training_step_time": 7.548194885253906
    },
    {
      "epoch": 3.6951065063476565e-05,
      "model_forward_time": 1.724412202835083,
      "step": 38746
    },
    {
      "epoch": 3.6951065063476565e-05,
      "step": 38746,
      "training_step_time": 8.630836009979248
    },
    {
      "epoch": 3.695201873779297e-05,
      "model_forward_time": 1.7220752239227295,
      "step": 38747
    },
    {
      "epoch": 3.695201873779297e-05,
      "step": 38747,
      "training_step_time": 7.499427556991577
    },
    {
      "epoch": 3.695297241210937e-05,
      "model_forward_time": 2.257035255432129,
      "step": 38748
    },
    {
      "epoch": 3.695297241210937e-05,
      "step": 38748,
      "training_step_time": 9.053157091140747
    },
    {
      "epoch": 3.6953926086425783e-05,
      "model_forward_time": 1.722632646560669,
      "step": 38749
    },
    {
      "epoch": 3.6953926086425783e-05,
      "step": 38749,
      "training_step_time": 9.052270650863647
    },
    {
      "epoch": 3.695487976074219e-05,
      "grad_norm": 0.15075992047786713,
      "learning_rate": 7.196131489273381e-06,
      "loss": 0.0624,
      "step": 38750
    },
    {
      "epoch": 3.695487976074219e-05,
      "model_forward_time": 1.7119684219360352,
      "step": 38750
    },
    {
      "dynamics_loss_avg": 0.05290074925869703,
      "epoch": 3.695487976074219e-05,
      "step": 38750
    },
    {
      "action_loss_avg": 0.0015038212906802074,
      "epoch": 3.695487976074219e-05,
      "step": 38750
    },
    {
      "epoch": 3.695487976074219e-05,
      "step": 38750,
      "training_step_time": 8.464929819107056
    },
    {
      "epoch": 3.695583343505859e-05,
      "model_forward_time": 1.7282326221466064,
      "step": 38751
    },
    {
      "epoch": 3.695583343505859e-05,
      "step": 38751,
      "training_step_time": 8.841155290603638
    },
    {
      "epoch": 3.6956787109375e-05,
      "model_forward_time": 1.7253620624542236,
      "step": 38752
    },
    {
      "epoch": 3.6956787109375e-05,
      "step": 38752,
      "training_step_time": 7.735847473144531
    },
    {
      "epoch": 3.6957740783691406e-05,
      "model_forward_time": 1.7344398498535156,
      "step": 38753
    },
    {
      "epoch": 3.6957740783691406e-05,
      "step": 38753,
      "training_step_time": 8.429402351379395
    },
    {
      "epoch": 3.695869445800781e-05,
      "model_forward_time": 1.77211332321167,
      "step": 38754
    },
    {
      "epoch": 3.695869445800781e-05,
      "step": 38754,
      "training_step_time": 7.758851766586304
    },
    {
      "epoch": 3.695964813232422e-05,
      "model_forward_time": 1.7410011291503906,
      "step": 38755
    },
    {
      "epoch": 3.695964813232422e-05,
      "step": 38755,
      "training_step_time": 8.866269588470459
    },
    {
      "epoch": 3.6960601806640625e-05,
      "model_forward_time": 1.728752613067627,
      "step": 38756
    },
    {
      "epoch": 3.6960601806640625e-05,
      "step": 38756,
      "training_step_time": 7.653828382492065
    },
    {
      "epoch": 3.696155548095703e-05,
      "model_forward_time": 2.2918806076049805,
      "step": 38757
    },
    {
      "epoch": 3.696155548095703e-05,
      "step": 38757,
      "training_step_time": 7.749741792678833
    },
    {
      "epoch": 3.696250915527344e-05,
      "model_forward_time": 1.7300405502319336,
      "step": 38758
    },
    {
      "epoch": 3.696250915527344e-05,
      "step": 38758,
      "training_step_time": 7.839802026748657
    },
    {
      "epoch": 3.6963462829589844e-05,
      "model_forward_time": 1.720912218093872,
      "step": 38759
    },
    {
      "epoch": 3.6963462829589844e-05,
      "step": 38759,
      "training_step_time": 7.88717246055603
    },
    {
      "epoch": 3.696441650390625e-05,
      "grad_norm": 0.13202089071273804,
      "learning_rate": 7.194645930860355e-06,
      "loss": 0.058,
      "step": 38760
    },
    {
      "epoch": 3.696441650390625e-05,
      "model_forward_time": 1.7309050559997559,
      "step": 38760
    },
    {
      "dynamics_loss_avg": 0.055501843243837355,
      "epoch": 3.696441650390625e-05,
      "step": 38760
    },
    {
      "action_loss_avg": 0.001301480340771377,
      "epoch": 3.696441650390625e-05,
      "step": 38760
    },
    {
      "epoch": 3.696441650390625e-05,
      "step": 38760,
      "training_step_time": 8.382883071899414
    },
    {
      "epoch": 3.696537017822266e-05,
      "model_forward_time": 1.7348129749298096,
      "step": 38761
    },
    {
      "epoch": 3.696537017822266e-05,
      "step": 38761,
      "training_step_time": 7.9630773067474365
    },
    {
      "epoch": 3.696632385253906e-05,
      "model_forward_time": 1.7455735206604004,
      "step": 38762
    },
    {
      "epoch": 3.696632385253906e-05,
      "step": 38762,
      "training_step_time": 7.796717643737793
    },
    {
      "epoch": 3.696727752685547e-05,
      "model_forward_time": 1.7807033061981201,
      "step": 38763
    },
    {
      "epoch": 3.696727752685547e-05,
      "step": 38763,
      "training_step_time": 7.760214567184448
    },
    {
      "epoch": 3.696823120117188e-05,
      "model_forward_time": 1.7322402000427246,
      "step": 38764
    },
    {
      "epoch": 3.696823120117188e-05,
      "step": 38764,
      "training_step_time": 7.933622360229492
    },
    {
      "epoch": 3.696918487548828e-05,
      "model_forward_time": 1.7388317584991455,
      "step": 38765
    },
    {
      "epoch": 3.696918487548828e-05,
      "step": 38765,
      "training_step_time": 7.825844049453735
    },
    {
      "epoch": 3.6970138549804686e-05,
      "model_forward_time": 2.2939951419830322,
      "step": 38766
    },
    {
      "epoch": 3.6970138549804686e-05,
      "step": 38766,
      "training_step_time": 8.457805633544922
    },
    {
      "epoch": 3.69710922241211e-05,
      "model_forward_time": 1.7231526374816895,
      "step": 38767
    },
    {
      "epoch": 3.69710922241211e-05,
      "step": 38767,
      "training_step_time": 9.771461725234985
    },
    {
      "epoch": 3.69720458984375e-05,
      "model_forward_time": 1.724412441253662,
      "step": 38768
    },
    {
      "epoch": 3.69720458984375e-05,
      "step": 38768,
      "training_step_time": 7.875431299209595
    },
    {
      "epoch": 3.6972999572753905e-05,
      "model_forward_time": 1.7284388542175293,
      "step": 38769
    },
    {
      "epoch": 3.6972999572753905e-05,
      "step": 38769,
      "training_step_time": 8.540258169174194
    },
    {
      "epoch": 3.6973953247070315e-05,
      "grad_norm": 0.16151507198810577,
      "learning_rate": 7.193160132444149e-06,
      "loss": 0.0573,
      "step": 38770
    },
    {
      "epoch": 3.6973953247070315e-05,
      "model_forward_time": 1.7315521240234375,
      "step": 38770
    },
    {
      "dynamics_loss_avg": 0.0479858634993434,
      "epoch": 3.6973953247070315e-05,
      "step": 38770
    },
    {
      "action_loss_avg": 0.002262954437173903,
      "epoch": 3.6973953247070315e-05,
      "step": 38770
    },
    {
      "epoch": 3.6973953247070315e-05,
      "step": 38770,
      "training_step_time": 7.689831256866455
    },
    {
      "epoch": 3.697490692138672e-05,
      "model_forward_time": 1.733534336090088,
      "step": 38771
    },
    {
      "epoch": 3.697490692138672e-05,
      "step": 38771,
      "training_step_time": 7.837077617645264
    },
    {
      "epoch": 3.6975860595703124e-05,
      "model_forward_time": 1.774449110031128,
      "step": 38772
    },
    {
      "epoch": 3.6975860595703124e-05,
      "step": 38772,
      "training_step_time": 8.30256986618042
    },
    {
      "epoch": 3.6976814270019534e-05,
      "model_forward_time": 1.7330987453460693,
      "step": 38773
    },
    {
      "epoch": 3.6976814270019534e-05,
      "step": 38773,
      "training_step_time": 7.8001708984375
    },
    {
      "epoch": 3.697776794433594e-05,
      "model_forward_time": 1.7321362495422363,
      "step": 38774
    },
    {
      "epoch": 3.697776794433594e-05,
      "step": 38774,
      "training_step_time": 7.744994163513184
    },
    {
      "epoch": 3.697872161865234e-05,
      "model_forward_time": 2.2907590866088867,
      "step": 38775
    },
    {
      "epoch": 3.697872161865234e-05,
      "step": 38775,
      "training_step_time": 8.585290670394897
    },
    {
      "epoch": 3.697967529296875e-05,
      "model_forward_time": 1.727161169052124,
      "step": 38776
    },
    {
      "epoch": 3.697967529296875e-05,
      "step": 38776,
      "training_step_time": 9.647490501403809
    },
    {
      "epoch": 3.698062896728516e-05,
      "model_forward_time": 1.7146360874176025,
      "step": 38777
    },
    {
      "epoch": 3.698062896728516e-05,
      "step": 38777,
      "training_step_time": 7.777562379837036
    },
    {
      "epoch": 3.698158264160156e-05,
      "model_forward_time": 1.7416620254516602,
      "step": 38778
    },
    {
      "epoch": 3.698158264160156e-05,
      "step": 38778,
      "training_step_time": 9.409782886505127
    },
    {
      "epoch": 3.698253631591797e-05,
      "model_forward_time": 1.7263145446777344,
      "step": 38779
    },
    {
      "epoch": 3.698253631591797e-05,
      "step": 38779,
      "training_step_time": 7.742088556289673
    },
    {
      "epoch": 3.6983489990234376e-05,
      "grad_norm": 0.21741177141666412,
      "learning_rate": 7.191674094187249e-06,
      "loss": 0.0612,
      "step": 38780
    },
    {
      "epoch": 3.6983489990234376e-05,
      "model_forward_time": 1.7257330417633057,
      "step": 38780
    },
    {
      "dynamics_loss_avg": 0.07312278598546981,
      "epoch": 3.6983489990234376e-05,
      "step": 38780
    },
    {
      "action_loss_avg": 0.0019318730221129954,
      "epoch": 3.6983489990234376e-05,
      "step": 38780
    },
    {
      "epoch": 3.6983489990234376e-05,
      "step": 38780,
      "training_step_time": 7.51870584487915
    },
    {
      "epoch": 3.698444366455078e-05,
      "model_forward_time": 1.7559537887573242,
      "step": 38781
    },
    {
      "epoch": 3.698444366455078e-05,
      "step": 38781,
      "training_step_time": 7.6396262645721436
    },
    {
      "epoch": 3.6985397338867184e-05,
      "model_forward_time": 1.7327218055725098,
      "step": 38782
    },
    {
      "epoch": 3.6985397338867184e-05,
      "step": 38782,
      "training_step_time": 7.567886829376221
    },
    {
      "epoch": 3.6986351013183595e-05,
      "model_forward_time": 1.7341434955596924,
      "step": 38783
    },
    {
      "epoch": 3.6986351013183595e-05,
      "step": 38783,
      "training_step_time": 8.515202283859253
    },
    {
      "epoch": 3.69873046875e-05,
      "model_forward_time": 1.7319424152374268,
      "step": 38784
    },
    {
      "epoch": 3.69873046875e-05,
      "step": 38784,
      "training_step_time": 8.055448055267334
    },
    {
      "epoch": 3.69882583618164e-05,
      "model_forward_time": 1.7409090995788574,
      "step": 38785
    },
    {
      "epoch": 3.69882583618164e-05,
      "step": 38785,
      "training_step_time": 8.917562007904053
    },
    {
      "epoch": 3.6989212036132814e-05,
      "model_forward_time": 1.727346420288086,
      "step": 38786
    },
    {
      "epoch": 3.6989212036132814e-05,
      "step": 38786,
      "training_step_time": 7.626660346984863
    },
    {
      "epoch": 3.699016571044922e-05,
      "model_forward_time": 1.7304461002349854,
      "step": 38787
    },
    {
      "epoch": 3.699016571044922e-05,
      "step": 38787,
      "training_step_time": 9.067872762680054
    },
    {
      "epoch": 3.699111938476562e-05,
      "model_forward_time": 1.83046293258667,
      "step": 38788
    },
    {
      "epoch": 3.699111938476562e-05,
      "step": 38788,
      "training_step_time": 8.622840642929077
    },
    {
      "epoch": 3.699207305908203e-05,
      "model_forward_time": 1.8388316631317139,
      "step": 38789
    },
    {
      "epoch": 3.699207305908203e-05,
      "step": 38789,
      "training_step_time": 7.811212062835693
    },
    {
      "epoch": 3.699302673339844e-05,
      "grad_norm": 0.13587914407253265,
      "learning_rate": 7.190187816252165e-06,
      "loss": 0.0624,
      "step": 38790
    },
    {
      "epoch": 3.699302673339844e-05,
      "model_forward_time": 1.8309142589569092,
      "step": 38790
    },
    {
      "dynamics_loss_avg": 0.05576045447960496,
      "epoch": 3.699302673339844e-05,
      "step": 38790
    },
    {
      "action_loss_avg": 0.0021144897560589015,
      "epoch": 3.699302673339844e-05,
      "step": 38790
    },
    {
      "epoch": 3.699302673339844e-05,
      "step": 38790,
      "training_step_time": 7.639910936355591
    },
    {
      "epoch": 3.699398040771484e-05,
      "model_forward_time": 1.8303883075714111,
      "step": 38791
    },
    {
      "epoch": 3.699398040771484e-05,
      "step": 38791,
      "training_step_time": 8.362751483917236
    },
    {
      "epoch": 3.699493408203125e-05,
      "model_forward_time": 1.8339910507202148,
      "step": 38792
    },
    {
      "epoch": 3.699493408203125e-05,
      "step": 38792,
      "training_step_time": 8.711978197097778
    },
    {
      "epoch": 3.6995887756347656e-05,
      "model_forward_time": 1.8324880599975586,
      "step": 38793
    },
    {
      "epoch": 3.6995887756347656e-05,
      "step": 38793,
      "training_step_time": 7.534616708755493
    },
    {
      "epoch": 3.699684143066406e-05,
      "model_forward_time": 2.3715872764587402,
      "step": 38794
    },
    {
      "epoch": 3.699684143066406e-05,
      "step": 38794,
      "training_step_time": 7.648684501647949
    },
    {
      "epoch": 3.699779510498047e-05,
      "model_forward_time": 1.8271045684814453,
      "step": 38795
    },
    {
      "epoch": 3.699779510498047e-05,
      "step": 38795,
      "training_step_time": 7.746055603027344
    },
    {
      "epoch": 3.6998748779296874e-05,
      "model_forward_time": 1.830380916595459,
      "step": 38796
    },
    {
      "epoch": 3.6998748779296874e-05,
      "step": 38796,
      "training_step_time": 9.490149021148682
    },
    {
      "epoch": 3.699970245361328e-05,
      "model_forward_time": 1.8175694942474365,
      "step": 38797
    },
    {
      "epoch": 3.699970245361328e-05,
      "step": 38797,
      "training_step_time": 7.630546808242798
    },
    {
      "epoch": 3.700065612792969e-05,
      "model_forward_time": 1.8279016017913818,
      "step": 38798
    },
    {
      "epoch": 3.700065612792969e-05,
      "step": 38798,
      "training_step_time": 8.603949308395386
    },
    {
      "epoch": 3.700160980224609e-05,
      "model_forward_time": 1.8351728916168213,
      "step": 38799
    },
    {
      "epoch": 3.700160980224609e-05,
      "step": 38799,
      "training_step_time": 7.544662952423096
    },
    {
      "epoch": 3.70025634765625e-05,
      "grad_norm": 0.11803635954856873,
      "learning_rate": 7.188701298801435e-06,
      "loss": 0.0589,
      "step": 38800
    },
    {
      "epoch": 3.70025634765625e-05,
      "model_forward_time": 1.809791088104248,
      "step": 38800
    },
    {
      "dynamics_loss_avg": 0.05294962842017412,
      "epoch": 3.70025634765625e-05,
      "step": 38800
    },
    {
      "action_loss_avg": 0.002590937918284908,
      "epoch": 3.70025634765625e-05,
      "step": 38800
    },
    {
      "epoch": 3.70025634765625e-05,
      "step": 38800,
      "training_step_time": 7.47209620475769
    },
    {
      "epoch": 3.700351715087891e-05,
      "model_forward_time": 1.7605292797088623,
      "step": 38801
    },
    {
      "epoch": 3.700351715087891e-05,
      "step": 38801,
      "training_step_time": 7.609244346618652
    },
    {
      "epoch": 3.700447082519531e-05,
      "model_forward_time": 1.7253012657165527,
      "step": 38802
    },
    {
      "epoch": 3.700447082519531e-05,
      "step": 38802,
      "training_step_time": 7.539653062820435
    },
    {
      "epoch": 3.7005424499511716e-05,
      "model_forward_time": 1.9053397178649902,
      "step": 38803
    },
    {
      "epoch": 3.7005424499511716e-05,
      "step": 38803,
      "training_step_time": 7.60770320892334
    },
    {
      "epoch": 3.700637817382813e-05,
      "model_forward_time": 1.7281701564788818,
      "step": 38804
    },
    {
      "epoch": 3.700637817382813e-05,
      "step": 38804,
      "training_step_time": 7.692504644393921
    },
    {
      "epoch": 3.700733184814453e-05,
      "model_forward_time": 1.7778124809265137,
      "step": 38805
    },
    {
      "epoch": 3.700733184814453e-05,
      "step": 38805,
      "training_step_time": 8.628366470336914
    },
    {
      "epoch": 3.7008285522460935e-05,
      "model_forward_time": 1.721562147140503,
      "step": 38806
    },
    {
      "epoch": 3.7008285522460935e-05,
      "step": 38806,
      "training_step_time": 7.440700531005859
    },
    {
      "epoch": 3.7009239196777346e-05,
      "model_forward_time": 1.718745231628418,
      "step": 38807
    },
    {
      "epoch": 3.7009239196777346e-05,
      "step": 38807,
      "training_step_time": 8.278751373291016
    },
    {
      "epoch": 3.701019287109375e-05,
      "model_forward_time": 1.7296712398529053,
      "step": 38808
    },
    {
      "epoch": 3.701019287109375e-05,
      "step": 38808,
      "training_step_time": 8.359202861785889
    },
    {
      "epoch": 3.7011146545410154e-05,
      "model_forward_time": 1.3526744842529297,
      "step": 38809
    },
    {
      "epoch": 3.7011146545410154e-05,
      "step": 38809,
      "training_step_time": 7.551742076873779
    },
    {
      "epoch": 3.7012100219726565e-05,
      "grad_norm": 0.12342048436403275,
      "learning_rate": 7.18721454199762e-06,
      "loss": 0.0581,
      "step": 38810
    },
    {
      "epoch": 3.7012100219726565e-05,
      "model_forward_time": 1.7211225032806396,
      "step": 38810
    },
    {
      "dynamics_loss_avg": 0.04469870934262872,
      "epoch": 3.7012100219726565e-05,
      "step": 38810
    },
    {
      "action_loss_avg": 0.0014734749856870621,
      "epoch": 3.7012100219726565e-05,
      "step": 38810
    },
    {
      "epoch": 3.7012100219726565e-05,
      "step": 38810,
      "training_step_time": 7.64043402671814
    },
    {
      "epoch": 3.701305389404297e-05,
      "model_forward_time": 1.7395122051239014,
      "step": 38811
    },
    {
      "epoch": 3.701305389404297e-05,
      "step": 38811,
      "training_step_time": 8.380842924118042
    },
    {
      "epoch": 3.701400756835937e-05,
      "model_forward_time": 2.2883288860321045,
      "step": 38812
    },
    {
      "epoch": 3.701400756835937e-05,
      "step": 38812,
      "training_step_time": 9.106520175933838
    },
    {
      "epoch": 3.7014961242675784e-05,
      "model_forward_time": 1.7286949157714844,
      "step": 38813
    },
    {
      "epoch": 3.7014961242675784e-05,
      "step": 38813,
      "training_step_time": 7.670748949050903
    },
    {
      "epoch": 3.701591491699219e-05,
      "model_forward_time": 1.7839324474334717,
      "step": 38814
    },
    {
      "epoch": 3.701591491699219e-05,
      "step": 38814,
      "training_step_time": 8.236921787261963
    },
    {
      "epoch": 3.701686859130859e-05,
      "model_forward_time": 1.7294676303863525,
      "step": 38815
    },
    {
      "epoch": 3.701686859130859e-05,
      "step": 38815,
      "training_step_time": 7.5164501667022705
    },
    {
      "epoch": 3.7017822265625e-05,
      "model_forward_time": 1.7295475006103516,
      "step": 38816
    },
    {
      "epoch": 3.7017822265625e-05,
      "step": 38816,
      "training_step_time": 8.323775291442871
    },
    {
      "epoch": 3.7018775939941406e-05,
      "model_forward_time": 1.726531744003296,
      "step": 38817
    },
    {
      "epoch": 3.7018775939941406e-05,
      "step": 38817,
      "training_step_time": 7.461794376373291
    },
    {
      "epoch": 3.701972961425781e-05,
      "model_forward_time": 1.7256367206573486,
      "step": 38818
    },
    {
      "epoch": 3.701972961425781e-05,
      "step": 38818,
      "training_step_time": 7.573989152908325
    },
    {
      "epoch": 3.702068328857422e-05,
      "model_forward_time": 1.7359771728515625,
      "step": 38819
    },
    {
      "epoch": 3.702068328857422e-05,
      "step": 38819,
      "training_step_time": 7.563215732574463
    },
    {
      "epoch": 3.7021636962890625e-05,
      "grad_norm": 0.13267932832241058,
      "learning_rate": 7.185727546003313e-06,
      "loss": 0.0602,
      "step": 38820
    },
    {
      "epoch": 3.7021636962890625e-05,
      "model_forward_time": 1.72463059425354,
      "step": 38820
    },
    {
      "dynamics_loss_avg": 0.04419892840087414,
      "epoch": 3.7021636962890625e-05,
      "step": 38820
    },
    {
      "action_loss_avg": 0.0008618960215244442,
      "epoch": 3.7021636962890625e-05,
      "step": 38820
    },
    {
      "epoch": 3.7021636962890625e-05,
      "step": 38820,
      "training_step_time": 7.547853946685791
    },
    {
      "epoch": 3.702259063720703e-05,
      "model_forward_time": 2.291185140609741,
      "step": 38821
    },
    {
      "epoch": 3.702259063720703e-05,
      "step": 38821,
      "training_step_time": 8.434757232666016
    },
    {
      "epoch": 3.702354431152344e-05,
      "model_forward_time": 1.7205567359924316,
      "step": 38822
    },
    {
      "epoch": 3.702354431152344e-05,
      "step": 38822,
      "training_step_time": 7.619236946105957
    },
    {
      "epoch": 3.7024497985839844e-05,
      "model_forward_time": 1.7772393226623535,
      "step": 38823
    },
    {
      "epoch": 3.7024497985839844e-05,
      "step": 38823,
      "training_step_time": 8.570178985595703
    },
    {
      "epoch": 3.702545166015625e-05,
      "model_forward_time": 1.7351202964782715,
      "step": 38824
    },
    {
      "epoch": 3.702545166015625e-05,
      "step": 38824,
      "training_step_time": 8.888585329055786
    },
    {
      "epoch": 3.702640533447266e-05,
      "model_forward_time": 1.7136108875274658,
      "step": 38825
    },
    {
      "epoch": 3.702640533447266e-05,
      "step": 38825,
      "training_step_time": 7.575336933135986
    },
    {
      "epoch": 3.702735900878906e-05,
      "model_forward_time": 1.7290558815002441,
      "step": 38826
    },
    {
      "epoch": 3.702735900878906e-05,
      "step": 38826,
      "training_step_time": 7.517840623855591
    },
    {
      "epoch": 3.702831268310547e-05,
      "model_forward_time": 1.3513195514678955,
      "step": 38827
    },
    {
      "epoch": 3.702831268310547e-05,
      "step": 38827,
      "training_step_time": 7.574722051620483
    },
    {
      "epoch": 3.702926635742188e-05,
      "model_forward_time": 1.727895975112915,
      "step": 38828
    },
    {
      "epoch": 3.702926635742188e-05,
      "step": 38828,
      "training_step_time": 7.589251279830933
    },
    {
      "epoch": 3.703022003173828e-05,
      "model_forward_time": 1.3582863807678223,
      "step": 38829
    },
    {
      "epoch": 3.703022003173828e-05,
      "step": 38829,
      "training_step_time": 8.568160772323608
    },
    {
      "epoch": 3.7031173706054686e-05,
      "grad_norm": 0.16866247355937958,
      "learning_rate": 7.184240310981126e-06,
      "loss": 0.0619,
      "step": 38830
    },
    {
      "epoch": 3.7031173706054686e-05,
      "model_forward_time": 2.256166696548462,
      "step": 38830
    },
    {
      "dynamics_loss_avg": 0.0709057793021202,
      "epoch": 3.7031173706054686e-05,
      "step": 38830
    },
    {
      "action_loss_avg": 0.0026124657480977474,
      "epoch": 3.7031173706054686e-05,
      "step": 38830
    },
    {
      "epoch": 3.7031173706054686e-05,
      "step": 38830,
      "training_step_time": 7.525728464126587
    },
    {
      "epoch": 3.70321273803711e-05,
      "model_forward_time": 1.7348642349243164,
      "step": 38831
    },
    {
      "epoch": 3.70321273803711e-05,
      "step": 38831,
      "training_step_time": 7.579638481140137
    },
    {
      "epoch": 3.70330810546875e-05,
      "model_forward_time": 1.789045810699463,
      "step": 38832
    },
    {
      "epoch": 3.70330810546875e-05,
      "step": 38832,
      "training_step_time": 8.247265815734863
    },
    {
      "epoch": 3.7034034729003905e-05,
      "model_forward_time": 1.7208750247955322,
      "step": 38833
    },
    {
      "epoch": 3.7034034729003905e-05,
      "step": 38833,
      "training_step_time": 8.398006916046143
    },
    {
      "epoch": 3.7034988403320316e-05,
      "model_forward_time": 1.7354342937469482,
      "step": 38834
    },
    {
      "epoch": 3.7034988403320316e-05,
      "step": 38834,
      "training_step_time": 8.64631462097168
    },
    {
      "epoch": 3.703594207763672e-05,
      "model_forward_time": 1.737748622894287,
      "step": 38835
    },
    {
      "epoch": 3.703594207763672e-05,
      "step": 38835,
      "training_step_time": 7.565252304077148
    },
    {
      "epoch": 3.7036895751953124e-05,
      "model_forward_time": 1.7301695346832275,
      "step": 38836
    },
    {
      "epoch": 3.7036895751953124e-05,
      "step": 38836,
      "training_step_time": 7.604134798049927
    },
    {
      "epoch": 3.7037849426269534e-05,
      "model_forward_time": 1.742499828338623,
      "step": 38837
    },
    {
      "epoch": 3.7037849426269534e-05,
      "step": 38837,
      "training_step_time": 7.667357921600342
    },
    {
      "epoch": 3.703880310058594e-05,
      "model_forward_time": 1.7480347156524658,
      "step": 38838
    },
    {
      "epoch": 3.703880310058594e-05,
      "step": 38838,
      "training_step_time": 7.597433805465698
    },
    {
      "epoch": 3.703975677490234e-05,
      "model_forward_time": 1.7368144989013672,
      "step": 38839
    },
    {
      "epoch": 3.703975677490234e-05,
      "step": 38839,
      "training_step_time": 7.552492141723633
    },
    {
      "epoch": 3.704071044921875e-05,
      "grad_norm": 0.21597056090831757,
      "learning_rate": 7.182752837093704e-06,
      "loss": 0.0602,
      "step": 38840
    },
    {
      "epoch": 3.704071044921875e-05,
      "model_forward_time": 2.295921802520752,
      "step": 38840
    },
    {
      "dynamics_loss_avg": 0.055800264701247215,
      "epoch": 3.704071044921875e-05,
      "step": 38840
    },
    {
      "action_loss_avg": 0.002051149000180885,
      "epoch": 3.704071044921875e-05,
      "step": 38840
    },
    {
      "epoch": 3.704071044921875e-05,
      "step": 38840,
      "training_step_time": 7.539125204086304
    },
    {
      "epoch": 3.704166412353516e-05,
      "model_forward_time": 1.7795300483703613,
      "step": 38841
    },
    {
      "epoch": 3.704166412353516e-05,
      "step": 38841,
      "training_step_time": 8.407887697219849
    },
    {
      "epoch": 3.704261779785156e-05,
      "model_forward_time": 1.7299447059631348,
      "step": 38842
    },
    {
      "epoch": 3.704261779785156e-05,
      "step": 38842,
      "training_step_time": 7.794573545455933
    },
    {
      "epoch": 3.704357147216797e-05,
      "model_forward_time": 1.7397668361663818,
      "step": 38843
    },
    {
      "epoch": 3.704357147216797e-05,
      "step": 38843,
      "training_step_time": 9.641515970230103
    },
    {
      "epoch": 3.7044525146484376e-05,
      "model_forward_time": 1.367490530014038,
      "step": 38844
    },
    {
      "epoch": 3.7044525146484376e-05,
      "step": 38844,
      "training_step_time": 7.564663410186768
    },
    {
      "epoch": 3.704547882080078e-05,
      "model_forward_time": 1.7319178581237793,
      "step": 38845
    },
    {
      "epoch": 3.704547882080078e-05,
      "step": 38845,
      "training_step_time": 7.6093504428863525
    },
    {
      "epoch": 3.7046432495117184e-05,
      "model_forward_time": 1.7382137775421143,
      "step": 38846
    },
    {
      "epoch": 3.7046432495117184e-05,
      "step": 38846,
      "training_step_time": 8.660909652709961
    },
    {
      "epoch": 3.7047386169433595e-05,
      "model_forward_time": 1.7330458164215088,
      "step": 38847
    },
    {
      "epoch": 3.7047386169433595e-05,
      "step": 38847,
      "training_step_time": 7.592998504638672
    },
    {
      "epoch": 3.704833984375e-05,
      "model_forward_time": 1.7390990257263184,
      "step": 38848
    },
    {
      "epoch": 3.704833984375e-05,
      "step": 38848,
      "training_step_time": 7.557358980178833
    },
    {
      "epoch": 3.70492935180664e-05,
      "model_forward_time": 2.319591522216797,
      "step": 38849
    },
    {
      "epoch": 3.70492935180664e-05,
      "step": 38849,
      "training_step_time": 7.613970041275024
    },
    {
      "epoch": 3.7050247192382814e-05,
      "grad_norm": 0.10257306694984436,
      "learning_rate": 7.181265124503712e-06,
      "loss": 0.0595,
      "step": 38850
    },
    {
      "epoch": 3.7050247192382814e-05,
      "model_forward_time": 1.4312705993652344,
      "step": 38850
    },
    {
      "dynamics_loss_avg": 0.06537494566291571,
      "epoch": 3.7050247192382814e-05,
      "step": 38850
    },
    {
      "action_loss_avg": 0.0018041776260361076,
      "epoch": 3.7050247192382814e-05,
      "step": 38850
    },
    {
      "epoch": 3.7050247192382814e-05,
      "step": 38850,
      "training_step_time": 8.644687175750732
    },
    {
      "epoch": 3.705120086669922e-05,
      "model_forward_time": 1.7395312786102295,
      "step": 38851
    },
    {
      "epoch": 3.705120086669922e-05,
      "step": 38851,
      "training_step_time": 8.740362167358398
    },
    {
      "epoch": 3.705215454101562e-05,
      "model_forward_time": 1.7570016384124756,
      "step": 38852
    },
    {
      "epoch": 3.705215454101562e-05,
      "step": 38852,
      "training_step_time": 7.785668134689331
    },
    {
      "epoch": 3.705310821533203e-05,
      "model_forward_time": 1.738288402557373,
      "step": 38853
    },
    {
      "epoch": 3.705310821533203e-05,
      "step": 38853,
      "training_step_time": 7.674417018890381
    },
    {
      "epoch": 3.705406188964844e-05,
      "model_forward_time": 1.7290623188018799,
      "step": 38854
    },
    {
      "epoch": 3.705406188964844e-05,
      "step": 38854,
      "training_step_time": 8.188365697860718
    },
    {
      "epoch": 3.705501556396484e-05,
      "model_forward_time": 1.7331528663635254,
      "step": 38855
    },
    {
      "epoch": 3.705501556396484e-05,
      "step": 38855,
      "training_step_time": 7.657847881317139
    },
    {
      "epoch": 3.705596923828125e-05,
      "model_forward_time": 1.741908073425293,
      "step": 38856
    },
    {
      "epoch": 3.705596923828125e-05,
      "step": 38856,
      "training_step_time": 7.582326889038086
    },
    {
      "epoch": 3.7056922912597656e-05,
      "model_forward_time": 1.7357051372528076,
      "step": 38857
    },
    {
      "epoch": 3.7056922912597656e-05,
      "step": 38857,
      "training_step_time": 7.646686792373657
    },
    {
      "epoch": 3.705787658691406e-05,
      "model_forward_time": 2.274435520172119,
      "step": 38858
    },
    {
      "epoch": 3.705787658691406e-05,
      "step": 38858,
      "training_step_time": 7.671924114227295
    },
    {
      "epoch": 3.705883026123047e-05,
      "model_forward_time": 1.7363231182098389,
      "step": 38859
    },
    {
      "epoch": 3.705883026123047e-05,
      "step": 38859,
      "training_step_time": 8.602627038955688
    },
    {
      "epoch": 3.7059783935546874e-05,
      "grad_norm": 0.15712951123714447,
      "learning_rate": 7.179777173373848e-06,
      "loss": 0.0612,
      "step": 38860
    },
    {
      "epoch": 3.7059783935546874e-05,
      "model_forward_time": 1.7183282375335693,
      "step": 38860
    },
    {
      "dynamics_loss_avg": 0.06517377849668264,
      "epoch": 3.7059783935546874e-05,
      "step": 38860
    },
    {
      "action_loss_avg": 0.0027007279335521162,
      "epoch": 3.7059783935546874e-05,
      "step": 38860
    },
    {
      "epoch": 3.7059783935546874e-05,
      "step": 38860,
      "training_step_time": 7.807624816894531
    },
    {
      "epoch": 3.706073760986328e-05,
      "model_forward_time": 1.7179596424102783,
      "step": 38861
    },
    {
      "epoch": 3.706073760986328e-05,
      "step": 38861,
      "training_step_time": 7.776591777801514
    },
    {
      "epoch": 3.706169128417969e-05,
      "model_forward_time": 1.7191946506500244,
      "step": 38862
    },
    {
      "epoch": 3.706169128417969e-05,
      "step": 38862,
      "training_step_time": 8.446051359176636
    },
    {
      "epoch": 3.706264495849609e-05,
      "model_forward_time": 1.733778476715088,
      "step": 38863
    },
    {
      "epoch": 3.706264495849609e-05,
      "step": 38863,
      "training_step_time": 7.562053918838501
    },
    {
      "epoch": 3.70635986328125e-05,
      "model_forward_time": 1.724635124206543,
      "step": 38864
    },
    {
      "epoch": 3.70635986328125e-05,
      "step": 38864,
      "training_step_time": 7.658669948577881
    },
    {
      "epoch": 3.706455230712891e-05,
      "model_forward_time": 1.7146382331848145,
      "step": 38865
    },
    {
      "epoch": 3.706455230712891e-05,
      "step": 38865,
      "training_step_time": 7.582468032836914
    },
    {
      "epoch": 3.706550598144531e-05,
      "model_forward_time": 1.7186806201934814,
      "step": 38866
    },
    {
      "epoch": 3.706550598144531e-05,
      "step": 38866,
      "training_step_time": 7.636504650115967
    },
    {
      "epoch": 3.7066459655761716e-05,
      "model_forward_time": 1.3523118495941162,
      "step": 38867
    },
    {
      "epoch": 3.7066459655761716e-05,
      "step": 38867,
      "training_step_time": 8.098081350326538
    },
    {
      "epoch": 3.706741333007813e-05,
      "model_forward_time": 2.2802603244781494,
      "step": 38868
    },
    {
      "epoch": 3.706741333007813e-05,
      "step": 38868,
      "training_step_time": 9.736555576324463
    },
    {
      "epoch": 3.706836700439453e-05,
      "model_forward_time": 1.7200477123260498,
      "step": 38869
    },
    {
      "epoch": 3.706836700439453e-05,
      "step": 38869,
      "training_step_time": 8.767547607421875
    },
    {
      "epoch": 3.7069320678710935e-05,
      "grad_norm": 0.12669065594673157,
      "learning_rate": 7.1782889838668255e-06,
      "loss": 0.0606,
      "step": 38870
    },
    {
      "epoch": 3.7069320678710935e-05,
      "model_forward_time": 1.7180523872375488,
      "step": 38870
    },
    {
      "dynamics_loss_avg": 0.056408021971583365,
      "epoch": 3.7069320678710935e-05,
      "step": 38870
    },
    {
      "action_loss_avg": 0.0013472884282236918,
      "epoch": 3.7069320678710935e-05,
      "step": 38870
    },
    {
      "epoch": 3.7069320678710935e-05,
      "step": 38870,
      "training_step_time": 8.720063209533691
    },
    {
      "epoch": 3.7070274353027346e-05,
      "model_forward_time": 1.724961280822754,
      "step": 38871
    },
    {
      "epoch": 3.7070274353027346e-05,
      "step": 38871,
      "training_step_time": 8.267714023590088
    },
    {
      "epoch": 3.707122802734375e-05,
      "model_forward_time": 1.7314894199371338,
      "step": 38872
    },
    {
      "epoch": 3.707122802734375e-05,
      "step": 38872,
      "training_step_time": 7.52818751335144
    },
    {
      "epoch": 3.7072181701660154e-05,
      "model_forward_time": 1.7200896739959717,
      "step": 38873
    },
    {
      "epoch": 3.7072181701660154e-05,
      "step": 38873,
      "training_step_time": 7.640964031219482
    },
    {
      "epoch": 3.7073135375976565e-05,
      "model_forward_time": 1.338139295578003,
      "step": 38874
    },
    {
      "epoch": 3.7073135375976565e-05,
      "step": 38874,
      "training_step_time": 7.605453252792358
    },
    {
      "epoch": 3.707408905029297e-05,
      "model_forward_time": 1.715212345123291,
      "step": 38875
    },
    {
      "epoch": 3.707408905029297e-05,
      "step": 38875,
      "training_step_time": 8.537478685379028
    },
    {
      "epoch": 3.707504272460937e-05,
      "model_forward_time": 1.719973087310791,
      "step": 38876
    },
    {
      "epoch": 3.707504272460937e-05,
      "step": 38876,
      "training_step_time": 8.4350106716156
    },
    {
      "epoch": 3.7075996398925784e-05,
      "model_forward_time": 2.302283525466919,
      "step": 38877
    },
    {
      "epoch": 3.7075996398925784e-05,
      "step": 38877,
      "training_step_time": 9.435816049575806
    },
    {
      "epoch": 3.707695007324219e-05,
      "model_forward_time": 1.7118291854858398,
      "step": 38878
    },
    {
      "epoch": 3.707695007324219e-05,
      "step": 38878,
      "training_step_time": 8.468950271606445
    },
    {
      "epoch": 3.707790374755859e-05,
      "model_forward_time": 1.7180798053741455,
      "step": 38879
    },
    {
      "epoch": 3.707790374755859e-05,
      "step": 38879,
      "training_step_time": 8.213864803314209
    },
    {
      "epoch": 3.7078857421875e-05,
      "grad_norm": 0.12790651619434357,
      "learning_rate": 7.176800556145398e-06,
      "loss": 0.0621,
      "step": 38880
    },
    {
      "epoch": 3.7078857421875e-05,
      "model_forward_time": 1.7181825637817383,
      "step": 38880
    },
    {
      "dynamics_loss_avg": 0.04636162407696247,
      "epoch": 3.7078857421875e-05,
      "step": 38880
    },
    {
      "action_loss_avg": 0.0014424763299757614,
      "epoch": 3.7078857421875e-05,
      "step": 38880
    },
    {
      "epoch": 3.7078857421875e-05,
      "step": 38880,
      "training_step_time": 7.775375604629517
    },
    {
      "epoch": 3.7079811096191406e-05,
      "model_forward_time": 1.729527235031128,
      "step": 38881
    },
    {
      "epoch": 3.7079811096191406e-05,
      "step": 38881,
      "training_step_time": 7.586816072463989
    },
    {
      "epoch": 3.708076477050781e-05,
      "model_forward_time": 1.7169675827026367,
      "step": 38882
    },
    {
      "epoch": 3.708076477050781e-05,
      "step": 38882,
      "training_step_time": 8.223309755325317
    },
    {
      "epoch": 3.708171844482422e-05,
      "model_forward_time": 1.719644546508789,
      "step": 38883
    },
    {
      "epoch": 3.708171844482422e-05,
      "step": 38883,
      "training_step_time": 8.727115631103516
    },
    {
      "epoch": 3.7082672119140625e-05,
      "model_forward_time": 1.7123620510101318,
      "step": 38884
    },
    {
      "epoch": 3.7082672119140625e-05,
      "step": 38884,
      "training_step_time": 7.52301287651062
    },
    {
      "epoch": 3.708362579345703e-05,
      "model_forward_time": 1.7183732986450195,
      "step": 38885
    },
    {
      "epoch": 3.708362579345703e-05,
      "step": 38885,
      "training_step_time": 7.495551824569702
    },
    {
      "epoch": 3.708457946777344e-05,
      "model_forward_time": 1.9085824489593506,
      "step": 38886
    },
    {
      "epoch": 3.708457946777344e-05,
      "step": 38886,
      "training_step_time": 9.612477779388428
    },
    {
      "epoch": 3.7085533142089844e-05,
      "model_forward_time": 1.734041452407837,
      "step": 38887
    },
    {
      "epoch": 3.7085533142089844e-05,
      "step": 38887,
      "training_step_time": 10.666356563568115
    },
    {
      "epoch": 3.708648681640625e-05,
      "model_forward_time": 1.719618797302246,
      "step": 38888
    },
    {
      "epoch": 3.708648681640625e-05,
      "step": 38888,
      "training_step_time": 8.365984678268433
    },
    {
      "epoch": 3.708744049072266e-05,
      "model_forward_time": 1.7215967178344727,
      "step": 38889
    },
    {
      "epoch": 3.708744049072266e-05,
      "step": 38889,
      "training_step_time": 9.23340630531311
    },
    {
      "epoch": 3.708839416503906e-05,
      "grad_norm": 0.11516164243221283,
      "learning_rate": 7.175311890372334e-06,
      "loss": 0.0592,
      "step": 38890
    },
    {
      "epoch": 3.708839416503906e-05,
      "model_forward_time": 1.732161045074463,
      "step": 38890
    },
    {
      "dynamics_loss_avg": 0.06768709346652031,
      "epoch": 3.708839416503906e-05,
      "step": 38890
    },
    {
      "action_loss_avg": 0.0025802460440900177,
      "epoch": 3.708839416503906e-05,
      "step": 38890
    },
    {
      "epoch": 3.708839416503906e-05,
      "step": 38890,
      "training_step_time": 7.5711541175842285
    },
    {
      "epoch": 3.708934783935547e-05,
      "model_forward_time": 1.720191478729248,
      "step": 38891
    },
    {
      "epoch": 3.708934783935547e-05,
      "step": 38891,
      "training_step_time": 7.53161883354187
    },
    {
      "epoch": 3.709030151367188e-05,
      "model_forward_time": 1.7262401580810547,
      "step": 38892
    },
    {
      "epoch": 3.709030151367188e-05,
      "step": 38892,
      "training_step_time": 7.6580891609191895
    },
    {
      "epoch": 3.709125518798828e-05,
      "model_forward_time": 1.7330009937286377,
      "step": 38893
    },
    {
      "epoch": 3.709125518798828e-05,
      "step": 38893,
      "training_step_time": 8.35646390914917
    },
    {
      "epoch": 3.7092208862304686e-05,
      "model_forward_time": 1.7359943389892578,
      "step": 38894
    },
    {
      "epoch": 3.7092208862304686e-05,
      "step": 38894,
      "training_step_time": 7.523251056671143
    },
    {
      "epoch": 3.70931625366211e-05,
      "model_forward_time": 1.7416646480560303,
      "step": 38895
    },
    {
      "epoch": 3.70931625366211e-05,
      "step": 38895,
      "training_step_time": 9.173713207244873
    },
    {
      "epoch": 3.70941162109375e-05,
      "model_forward_time": 1.7892639636993408,
      "step": 38896
    },
    {
      "epoch": 3.70941162109375e-05,
      "step": 38896,
      "training_step_time": 9.072977542877197
    },
    {
      "epoch": 3.7095069885253905e-05,
      "model_forward_time": 1.724255084991455,
      "step": 38897
    },
    {
      "epoch": 3.7095069885253905e-05,
      "step": 38897,
      "training_step_time": 8.349843740463257
    },
    {
      "epoch": 3.7096023559570316e-05,
      "model_forward_time": 1.7298405170440674,
      "step": 38898
    },
    {
      "epoch": 3.7096023559570316e-05,
      "step": 38898,
      "training_step_time": 8.79158878326416
    },
    {
      "epoch": 3.709697723388672e-05,
      "model_forward_time": 1.7256686687469482,
      "step": 38899
    },
    {
      "epoch": 3.709697723388672e-05,
      "step": 38899,
      "training_step_time": 7.717834949493408
    },
    {
      "epoch": 3.7097930908203124e-05,
      "grad_norm": 0.10412029922008514,
      "learning_rate": 7.17382298671043e-06,
      "loss": 0.0605,
      "step": 38900
    },
    {
      "epoch": 3.7097930908203124e-05,
      "model_forward_time": 1.7260169982910156,
      "step": 38900
    },
    {
      "dynamics_loss_avg": 0.04991293307393789,
      "epoch": 3.7097930908203124e-05,
      "step": 38900
    },
    {
      "action_loss_avg": 0.0019825826515443624,
      "epoch": 3.7097930908203124e-05,
      "step": 38900
    },
    {
      "epoch": 3.7097930908203124e-05,
      "step": 38900,
      "training_step_time": 7.573279857635498
    },
    {
      "epoch": 3.7098884582519534e-05,
      "model_forward_time": 1.7319583892822266,
      "step": 38901
    },
    {
      "epoch": 3.7098884582519534e-05,
      "step": 38901,
      "training_step_time": 7.544921398162842
    },
    {
      "epoch": 3.709983825683594e-05,
      "model_forward_time": 1.8515982627868652,
      "step": 38902
    },
    {
      "epoch": 3.709983825683594e-05,
      "step": 38902,
      "training_step_time": 7.64385986328125
    },
    {
      "epoch": 3.710079193115234e-05,
      "model_forward_time": 1.83935546875,
      "step": 38903
    },
    {
      "epoch": 3.710079193115234e-05,
      "step": 38903,
      "training_step_time": 7.650914907455444
    },
    {
      "epoch": 3.710174560546875e-05,
      "model_forward_time": 1.8476142883300781,
      "step": 38904
    },
    {
      "epoch": 3.710174560546875e-05,
      "step": 38904,
      "training_step_time": 7.651649713516235
    },
    {
      "epoch": 3.710269927978516e-05,
      "model_forward_time": 2.3943798542022705,
      "step": 38905
    },
    {
      "epoch": 3.710269927978516e-05,
      "step": 38905,
      "training_step_time": 7.717535734176636
    },
    {
      "epoch": 3.710365295410156e-05,
      "model_forward_time": 1.8350257873535156,
      "step": 38906
    },
    {
      "epoch": 3.710365295410156e-05,
      "step": 38906,
      "training_step_time": 7.760457277297974
    },
    {
      "epoch": 3.7104606628417965e-05,
      "model_forward_time": 1.836646318435669,
      "step": 38907
    },
    {
      "epoch": 3.7104606628417965e-05,
      "step": 38907,
      "training_step_time": 7.779163837432861
    },
    {
      "epoch": 3.7105560302734376e-05,
      "model_forward_time": 1.8447647094726562,
      "step": 38908
    },
    {
      "epoch": 3.7105560302734376e-05,
      "step": 38908,
      "training_step_time": 8.533518552780151
    },
    {
      "epoch": 3.710651397705078e-05,
      "model_forward_time": 1.8427894115447998,
      "step": 38909
    },
    {
      "epoch": 3.710651397705078e-05,
      "step": 38909,
      "training_step_time": 7.578172922134399
    },
    {
      "epoch": 3.7107467651367184e-05,
      "grad_norm": 0.12390376627445221,
      "learning_rate": 7.172333845322515e-06,
      "loss": 0.0598,
      "step": 38910
    },
    {
      "epoch": 3.7107467651367184e-05,
      "model_forward_time": 1.8379685878753662,
      "step": 38910
    },
    {
      "dynamics_loss_avg": 0.052763469889760016,
      "epoch": 3.7107467651367184e-05,
      "step": 38910
    },
    {
      "action_loss_avg": 0.0021440939279273154,
      "epoch": 3.7107467651367184e-05,
      "step": 38910
    },
    {
      "epoch": 3.7107467651367184e-05,
      "step": 38910,
      "training_step_time": 7.6082189083099365
    },
    {
      "epoch": 3.7108421325683595e-05,
      "model_forward_time": 1.833024501800537,
      "step": 38911
    },
    {
      "epoch": 3.7108421325683595e-05,
      "step": 38911,
      "training_step_time": 7.741200923919678
    },
    {
      "epoch": 3.7109375e-05,
      "model_forward_time": 1.8401744365692139,
      "step": 38912
    },
    {
      "epoch": 3.7109375e-05,
      "step": 38912,
      "training_step_time": 8.566060543060303
    },
    {
      "epoch": 3.71103286743164e-05,
      "model_forward_time": 1.1133785247802734,
      "step": 38913
    },
    {
      "epoch": 3.71103286743164e-05,
      "step": 38913,
      "training_step_time": 7.593675136566162
    },
    {
      "epoch": 3.7111282348632814e-05,
      "model_forward_time": 2.2718656063079834,
      "step": 38914
    },
    {
      "epoch": 3.7111282348632814e-05,
      "step": 38914,
      "training_step_time": 7.599697589874268
    },
    {
      "epoch": 3.711223602294922e-05,
      "model_forward_time": 1.743354320526123,
      "step": 38915
    },
    {
      "epoch": 3.711223602294922e-05,
      "step": 38915,
      "training_step_time": 7.721485137939453
    },
    {
      "epoch": 3.711318969726562e-05,
      "model_forward_time": 1.7388198375701904,
      "step": 38916
    },
    {
      "epoch": 3.711318969726562e-05,
      "step": 38916,
      "training_step_time": 8.458042621612549
    },
    {
      "epoch": 3.711414337158203e-05,
      "model_forward_time": 1.780303955078125,
      "step": 38917
    },
    {
      "epoch": 3.711414337158203e-05,
      "step": 38917,
      "training_step_time": 7.644536018371582
    },
    {
      "epoch": 3.711509704589844e-05,
      "model_forward_time": 1.7374978065490723,
      "step": 38918
    },
    {
      "epoch": 3.711509704589844e-05,
      "step": 38918,
      "training_step_time": 7.526608228683472
    },
    {
      "epoch": 3.711605072021484e-05,
      "model_forward_time": 1.7452235221862793,
      "step": 38919
    },
    {
      "epoch": 3.711605072021484e-05,
      "step": 38919,
      "training_step_time": 7.638878583908081
    },
    {
      "epoch": 3.711700439453125e-05,
      "grad_norm": 0.10885131359100342,
      "learning_rate": 7.170844466371436e-06,
      "loss": 0.0603,
      "step": 38920
    },
    {
      "epoch": 3.711700439453125e-05,
      "model_forward_time": 1.7368836402893066,
      "step": 38920
    },
    {
      "dynamics_loss_avg": 0.05872151665389538,
      "epoch": 3.711700439453125e-05,
      "step": 38920
    },
    {
      "action_loss_avg": 0.002552053064573556,
      "epoch": 3.711700439453125e-05,
      "step": 38920
    },
    {
      "epoch": 3.711700439453125e-05,
      "step": 38920,
      "training_step_time": 7.66801643371582
    },
    {
      "epoch": 3.7117958068847656e-05,
      "model_forward_time": 1.7380201816558838,
      "step": 38921
    },
    {
      "epoch": 3.7117958068847656e-05,
      "step": 38921,
      "training_step_time": 7.534261703491211
    },
    {
      "epoch": 3.711891174316406e-05,
      "model_forward_time": 1.7315325736999512,
      "step": 38922
    },
    {
      "epoch": 3.711891174316406e-05,
      "step": 38922,
      "training_step_time": 7.643798828125
    },
    {
      "epoch": 3.711986541748047e-05,
      "model_forward_time": 2.2893259525299072,
      "step": 38923
    },
    {
      "epoch": 3.711986541748047e-05,
      "step": 38923,
      "training_step_time": 8.290314197540283
    },
    {
      "epoch": 3.7120819091796875e-05,
      "model_forward_time": 1.737729549407959,
      "step": 38924
    },
    {
      "epoch": 3.7120819091796875e-05,
      "step": 38924,
      "training_step_time": 8.335668325424194
    },
    {
      "epoch": 3.712177276611328e-05,
      "model_forward_time": 1.7473044395446777,
      "step": 38925
    },
    {
      "epoch": 3.712177276611328e-05,
      "step": 38925,
      "training_step_time": 7.893314838409424
    },
    {
      "epoch": 3.712272644042969e-05,
      "model_forward_time": 1.7810192108154297,
      "step": 38926
    },
    {
      "epoch": 3.712272644042969e-05,
      "step": 38926,
      "training_step_time": 8.212978601455688
    },
    {
      "epoch": 3.712368011474609e-05,
      "model_forward_time": 1.7434568405151367,
      "step": 38927
    },
    {
      "epoch": 3.712368011474609e-05,
      "step": 38927,
      "training_step_time": 7.5990612506866455
    },
    {
      "epoch": 3.71246337890625e-05,
      "model_forward_time": 1.7434043884277344,
      "step": 38928
    },
    {
      "epoch": 3.71246337890625e-05,
      "step": 38928,
      "training_step_time": 7.6217875480651855
    },
    {
      "epoch": 3.712558746337891e-05,
      "model_forward_time": 1.7401607036590576,
      "step": 38929
    },
    {
      "epoch": 3.712558746337891e-05,
      "step": 38929,
      "training_step_time": 7.647548198699951
    },
    {
      "epoch": 3.712654113769531e-05,
      "grad_norm": 0.1165790930390358,
      "learning_rate": 7.169354850020069e-06,
      "loss": 0.0584,
      "step": 38930
    },
    {
      "epoch": 3.712654113769531e-05,
      "model_forward_time": 1.7387886047363281,
      "step": 38930
    },
    {
      "dynamics_loss_avg": 0.0532320536673069,
      "epoch": 3.712654113769531e-05,
      "step": 38930
    },
    {
      "action_loss_avg": 0.0025650912371929734,
      "epoch": 3.712654113769531e-05,
      "step": 38930
    },
    {
      "epoch": 3.712654113769531e-05,
      "step": 38930,
      "training_step_time": 7.400658130645752
    },
    {
      "epoch": 3.7127494812011716e-05,
      "model_forward_time": 1.7458915710449219,
      "step": 38931
    },
    {
      "epoch": 3.7127494812011716e-05,
      "step": 38931,
      "training_step_time": 7.5595057010650635
    },
    {
      "epoch": 3.712844848632813e-05,
      "model_forward_time": 1.741955041885376,
      "step": 38932
    },
    {
      "epoch": 3.712844848632813e-05,
      "step": 38932,
      "training_step_time": 7.5161943435668945
    },
    {
      "epoch": 3.712940216064453e-05,
      "model_forward_time": 2.2968575954437256,
      "step": 38933
    },
    {
      "epoch": 3.712940216064453e-05,
      "step": 38933,
      "training_step_time": 9.022778272628784
    },
    {
      "epoch": 3.7130355834960935e-05,
      "model_forward_time": 1.7408051490783691,
      "step": 38934
    },
    {
      "epoch": 3.7130355834960935e-05,
      "step": 38934,
      "training_step_time": 7.797485589981079
    },
    {
      "epoch": 3.7131309509277346e-05,
      "model_forward_time": 1.779904842376709,
      "step": 38935
    },
    {
      "epoch": 3.7131309509277346e-05,
      "step": 38935,
      "training_step_time": 7.476746082305908
    },
    {
      "epoch": 3.713226318359375e-05,
      "model_forward_time": 1.0284414291381836,
      "step": 38936
    },
    {
      "epoch": 3.713226318359375e-05,
      "step": 38936,
      "training_step_time": 7.638786792755127
    },
    {
      "epoch": 3.7133216857910154e-05,
      "model_forward_time": 1.7429521083831787,
      "step": 38937
    },
    {
      "epoch": 3.7133216857910154e-05,
      "step": 38937,
      "training_step_time": 7.621448278427124
    },
    {
      "epoch": 3.7134170532226565e-05,
      "model_forward_time": 1.738222599029541,
      "step": 38938
    },
    {
      "epoch": 3.7134170532226565e-05,
      "step": 38938,
      "training_step_time": 7.649764776229858
    },
    {
      "epoch": 3.713512420654297e-05,
      "model_forward_time": 1.7387666702270508,
      "step": 38939
    },
    {
      "epoch": 3.713512420654297e-05,
      "step": 38939,
      "training_step_time": 7.589486360549927
    },
    {
      "epoch": 3.713607788085937e-05,
      "grad_norm": 0.09187936782836914,
      "learning_rate": 7.1678649964313194e-06,
      "loss": 0.0575,
      "step": 38940
    },
    {
      "epoch": 3.713607788085937e-05,
      "model_forward_time": 1.7411763668060303,
      "step": 38940
    },
    {
      "dynamics_loss_avg": 0.04839982558041811,
      "epoch": 3.713607788085937e-05,
      "step": 38940
    },
    {
      "action_loss_avg": 0.0023262301459908486,
      "epoch": 3.713607788085937e-05,
      "step": 38940
    },
    {
      "epoch": 3.713607788085937e-05,
      "step": 38940,
      "training_step_time": 8.084260702133179
    },
    {
      "epoch": 3.7137031555175784e-05,
      "model_forward_time": 1.7365903854370117,
      "step": 38941
    },
    {
      "epoch": 3.7137031555175784e-05,
      "step": 38941,
      "training_step_time": 8.590415000915527
    },
    {
      "epoch": 3.713798522949219e-05,
      "model_forward_time": 2.3448545932769775,
      "step": 38942
    },
    {
      "epoch": 3.713798522949219e-05,
      "step": 38942,
      "training_step_time": 9.211849927902222
    },
    {
      "epoch": 3.713893890380859e-05,
      "model_forward_time": 1.729952335357666,
      "step": 38943
    },
    {
      "epoch": 3.713893890380859e-05,
      "step": 38943,
      "training_step_time": 8.703952312469482
    },
    {
      "epoch": 3.7139892578125e-05,
      "model_forward_time": 1.7810375690460205,
      "step": 38944
    },
    {
      "epoch": 3.7139892578125e-05,
      "step": 38944,
      "training_step_time": 8.464128971099854
    },
    {
      "epoch": 3.7140846252441407e-05,
      "model_forward_time": 1.7383532524108887,
      "step": 38945
    },
    {
      "epoch": 3.7140846252441407e-05,
      "step": 38945,
      "training_step_time": 7.6766557693481445
    },
    {
      "epoch": 3.714179992675781e-05,
      "model_forward_time": 1.737738847732544,
      "step": 38946
    },
    {
      "epoch": 3.714179992675781e-05,
      "step": 38946,
      "training_step_time": 8.592112064361572
    },
    {
      "epoch": 3.714275360107422e-05,
      "model_forward_time": 1.7366206645965576,
      "step": 38947
    },
    {
      "epoch": 3.714275360107422e-05,
      "step": 38947,
      "training_step_time": 7.564461708068848
    },
    {
      "epoch": 3.7143707275390625e-05,
      "model_forward_time": 1.7390451431274414,
      "step": 38948
    },
    {
      "epoch": 3.7143707275390625e-05,
      "step": 38948,
      "training_step_time": 7.5762858390808105
    },
    {
      "epoch": 3.714466094970703e-05,
      "model_forward_time": 1.7401807308197021,
      "step": 38949
    },
    {
      "epoch": 3.714466094970703e-05,
      "step": 38949,
      "training_step_time": 8.387181520462036
    },
    {
      "epoch": 3.714561462402344e-05,
      "grad_norm": 0.10527504980564117,
      "learning_rate": 7.166374905768112e-06,
      "loss": 0.0566,
      "step": 38950
    },
    {
      "epoch": 3.714561462402344e-05,
      "model_forward_time": 1.7343173027038574,
      "step": 38950
    },
    {
      "dynamics_loss_avg": 0.06443148218095303,
      "epoch": 3.714561462402344e-05,
      "step": 38950
    },
    {
      "action_loss_avg": 0.0012277728761546313,
      "epoch": 3.714561462402344e-05,
      "step": 38950
    },
    {
      "epoch": 3.714561462402344e-05,
      "step": 38950,
      "training_step_time": 7.4996984004974365
    },
    {
      "epoch": 3.7146568298339844e-05,
      "model_forward_time": 2.3126344680786133,
      "step": 38951
    },
    {
      "epoch": 3.7146568298339844e-05,
      "step": 38951,
      "training_step_time": 8.831321239471436
    },
    {
      "epoch": 3.714752197265625e-05,
      "model_forward_time": 1.7402544021606445,
      "step": 38952
    },
    {
      "epoch": 3.714752197265625e-05,
      "step": 38952,
      "training_step_time": 7.7437591552734375
    },
    {
      "epoch": 3.714847564697266e-05,
      "model_forward_time": 1.7811439037322998,
      "step": 38953
    },
    {
      "epoch": 3.714847564697266e-05,
      "step": 38953,
      "training_step_time": 7.670730829238892
    },
    {
      "epoch": 3.714942932128906e-05,
      "model_forward_time": 1.7416350841522217,
      "step": 38954
    },
    {
      "epoch": 3.714942932128906e-05,
      "step": 38954,
      "training_step_time": 7.622523307800293
    },
    {
      "epoch": 3.715038299560547e-05,
      "model_forward_time": 1.373826503753662,
      "step": 38955
    },
    {
      "epoch": 3.715038299560547e-05,
      "step": 38955,
      "training_step_time": 7.580881357192993
    },
    {
      "epoch": 3.715133666992188e-05,
      "model_forward_time": 1.73579740524292,
      "step": 38956
    },
    {
      "epoch": 3.715133666992188e-05,
      "step": 38956,
      "training_step_time": 7.583661317825317
    },
    {
      "epoch": 3.715229034423828e-05,
      "model_forward_time": 1.7455840110778809,
      "step": 38957
    },
    {
      "epoch": 3.715229034423828e-05,
      "step": 38957,
      "training_step_time": 7.484979629516602
    },
    {
      "epoch": 3.7153244018554686e-05,
      "model_forward_time": 1.3679254055023193,
      "step": 38958
    },
    {
      "epoch": 3.7153244018554686e-05,
      "step": 38958,
      "training_step_time": 7.625363349914551
    },
    {
      "epoch": 3.71541976928711e-05,
      "model_forward_time": 1.7404992580413818,
      "step": 38959
    },
    {
      "epoch": 3.71541976928711e-05,
      "step": 38959,
      "training_step_time": 8.45650863647461
    },
    {
      "epoch": 3.71551513671875e-05,
      "grad_norm": 0.13479726016521454,
      "learning_rate": 7.164884578193401e-06,
      "loss": 0.0571,
      "step": 38960
    },
    {
      "epoch": 3.71551513671875e-05,
      "model_forward_time": 1.7112467288970947,
      "step": 38960
    },
    {
      "dynamics_loss_avg": 0.0595141876488924,
      "epoch": 3.71551513671875e-05,
      "step": 38960
    },
    {
      "action_loss_avg": 0.0018363218317972497,
      "epoch": 3.71551513671875e-05,
      "step": 38960
    },
    {
      "epoch": 3.71551513671875e-05,
      "step": 38960,
      "training_step_time": 7.601002931594849
    },
    {
      "epoch": 3.7156105041503905e-05,
      "model_forward_time": 2.290505886077881,
      "step": 38961
    },
    {
      "epoch": 3.7156105041503905e-05,
      "step": 38961,
      "training_step_time": 7.778625965118408
    },
    {
      "epoch": 3.7157058715820316e-05,
      "model_forward_time": 1.767270803451538,
      "step": 38962
    },
    {
      "epoch": 3.7157058715820316e-05,
      "step": 38962,
      "training_step_time": 9.60793423652649
    },
    {
      "epoch": 3.715801239013672e-05,
      "model_forward_time": 1.7314131259918213,
      "step": 38963
    },
    {
      "epoch": 3.715801239013672e-05,
      "step": 38963,
      "training_step_time": 7.514078378677368
    },
    {
      "epoch": 3.7158966064453124e-05,
      "model_forward_time": 1.7264769077301025,
      "step": 38964
    },
    {
      "epoch": 3.7158966064453124e-05,
      "step": 38964,
      "training_step_time": 8.72801160812378
    },
    {
      "epoch": 3.7159919738769535e-05,
      "model_forward_time": 1.7286689281463623,
      "step": 38965
    },
    {
      "epoch": 3.7159919738769535e-05,
      "step": 38965,
      "training_step_time": 7.560368776321411
    },
    {
      "epoch": 3.716087341308594e-05,
      "model_forward_time": 1.7283072471618652,
      "step": 38966
    },
    {
      "epoch": 3.716087341308594e-05,
      "step": 38966,
      "training_step_time": 8.276111841201782
    },
    {
      "epoch": 3.716182708740234e-05,
      "model_forward_time": 1.7287683486938477,
      "step": 38967
    },
    {
      "epoch": 3.716182708740234e-05,
      "step": 38967,
      "training_step_time": 7.608972787857056
    },
    {
      "epoch": 3.716278076171875e-05,
      "model_forward_time": 1.7398126125335693,
      "step": 38968
    },
    {
      "epoch": 3.716278076171875e-05,
      "step": 38968,
      "training_step_time": 7.539583921432495
    },
    {
      "epoch": 3.716373443603516e-05,
      "model_forward_time": 1.7322742938995361,
      "step": 38969
    },
    {
      "epoch": 3.716373443603516e-05,
      "step": 38969,
      "training_step_time": 7.563622713088989
    },
    {
      "epoch": 3.716468811035156e-05,
      "grad_norm": 0.11920252442359924,
      "learning_rate": 7.163394013870169e-06,
      "loss": 0.0561,
      "step": 38970
    },
    {
      "epoch": 3.716468811035156e-05,
      "model_forward_time": 2.2754552364349365,
      "step": 38970
    },
    {
      "dynamics_loss_avg": 0.061438755318522456,
      "epoch": 3.716468811035156e-05,
      "step": 38970
    },
    {
      "action_loss_avg": 0.0021037703729234634,
      "epoch": 3.716468811035156e-05,
      "step": 38970
    },
    {
      "epoch": 3.716468811035156e-05,
      "step": 38970,
      "training_step_time": 8.673126697540283
    },
    {
      "epoch": 3.7165641784667965e-05,
      "model_forward_time": 1.75565505027771,
      "step": 38971
    },
    {
      "epoch": 3.7165641784667965e-05,
      "step": 38971,
      "training_step_time": 8.460569381713867
    },
    {
      "epoch": 3.7166595458984376e-05,
      "model_forward_time": 1.7338058948516846,
      "step": 38972
    },
    {
      "epoch": 3.7166595458984376e-05,
      "step": 38972,
      "training_step_time": 7.44231104850769
    },
    {
      "epoch": 3.716754913330078e-05,
      "model_forward_time": 1.7234892845153809,
      "step": 38973
    },
    {
      "epoch": 3.716754913330078e-05,
      "step": 38973,
      "training_step_time": 7.613331079483032
    },
    {
      "epoch": 3.7168502807617184e-05,
      "model_forward_time": 1.7379071712493896,
      "step": 38974
    },
    {
      "epoch": 3.7168502807617184e-05,
      "step": 38974,
      "training_step_time": 7.564424991607666
    },
    {
      "epoch": 3.7169456481933595e-05,
      "model_forward_time": 1.729166030883789,
      "step": 38975
    },
    {
      "epoch": 3.7169456481933595e-05,
      "step": 38975,
      "training_step_time": 7.677513837814331
    },
    {
      "epoch": 3.717041015625e-05,
      "model_forward_time": 1.734740972518921,
      "step": 38976
    },
    {
      "epoch": 3.717041015625e-05,
      "step": 38976,
      "training_step_time": 8.247672080993652
    },
    {
      "epoch": 3.71713638305664e-05,
      "model_forward_time": 1.7382230758666992,
      "step": 38977
    },
    {
      "epoch": 3.71713638305664e-05,
      "step": 38977,
      "training_step_time": 8.050116539001465
    },
    {
      "epoch": 3.7172317504882814e-05,
      "model_forward_time": 1.7345421314239502,
      "step": 38978
    },
    {
      "epoch": 3.7172317504882814e-05,
      "step": 38978,
      "training_step_time": 9.091508865356445
    },
    {
      "epoch": 3.717327117919922e-05,
      "model_forward_time": 2.3194944858551025,
      "step": 38979
    },
    {
      "epoch": 3.717327117919922e-05,
      "step": 38979,
      "training_step_time": 7.765547275543213
    },
    {
      "epoch": 3.717422485351562e-05,
      "grad_norm": 0.1371193379163742,
      "learning_rate": 7.161903212961418e-06,
      "loss": 0.0565,
      "step": 38980
    },
    {
      "epoch": 3.717422485351562e-05,
      "model_forward_time": 1.72670578956604,
      "step": 38980
    },
    {
      "dynamics_loss_avg": 0.06796987559646368,
      "epoch": 3.717422485351562e-05,
      "step": 38980
    },
    {
      "action_loss_avg": 0.001907395722810179,
      "epoch": 3.717422485351562e-05,
      "step": 38980
    },
    {
      "epoch": 3.717422485351562e-05,
      "step": 38980,
      "training_step_time": 7.804207801818848
    },
    {
      "epoch": 3.717517852783203e-05,
      "model_forward_time": 1.7274565696716309,
      "step": 38981
    },
    {
      "epoch": 3.717517852783203e-05,
      "step": 38981,
      "training_step_time": 7.527966499328613
    },
    {
      "epoch": 3.717613220214844e-05,
      "model_forward_time": 1.7394969463348389,
      "step": 38982
    },
    {
      "epoch": 3.717613220214844e-05,
      "step": 38982,
      "training_step_time": 7.715077877044678
    },
    {
      "epoch": 3.717708587646484e-05,
      "model_forward_time": 1.7279019355773926,
      "step": 38983
    },
    {
      "epoch": 3.717708587646484e-05,
      "step": 38983,
      "training_step_time": 7.68964958190918
    },
    {
      "epoch": 3.717803955078125e-05,
      "model_forward_time": 1.734823226928711,
      "step": 38984
    },
    {
      "epoch": 3.717803955078125e-05,
      "step": 38984,
      "training_step_time": 7.679515361785889
    },
    {
      "epoch": 3.7178993225097656e-05,
      "model_forward_time": 1.739354133605957,
      "step": 38985
    },
    {
      "epoch": 3.7178993225097656e-05,
      "step": 38985,
      "training_step_time": 9.1103355884552
    },
    {
      "epoch": 3.717994689941406e-05,
      "model_forward_time": 1.7269933223724365,
      "step": 38986
    },
    {
      "epoch": 3.717994689941406e-05,
      "step": 38986,
      "training_step_time": 7.574404954910278
    },
    {
      "epoch": 3.718090057373047e-05,
      "model_forward_time": 1.731722116470337,
      "step": 38987
    },
    {
      "epoch": 3.718090057373047e-05,
      "step": 38987,
      "training_step_time": 8.739826679229736
    },
    {
      "epoch": 3.7181854248046875e-05,
      "model_forward_time": 2.2947962284088135,
      "step": 38988
    },
    {
      "epoch": 3.7181854248046875e-05,
      "step": 38988,
      "training_step_time": 7.80329966545105
    },
    {
      "epoch": 3.718280792236328e-05,
      "model_forward_time": 1.7309370040893555,
      "step": 38989
    },
    {
      "epoch": 3.718280792236328e-05,
      "step": 38989,
      "training_step_time": 7.630909442901611
    },
    {
      "epoch": 3.718376159667969e-05,
      "grad_norm": 0.09610181301832199,
      "learning_rate": 7.160412175630184e-06,
      "loss": 0.0592,
      "step": 38990
    },
    {
      "epoch": 3.718376159667969e-05,
      "model_forward_time": 1.775801181793213,
      "step": 38990
    },
    {
      "dynamics_loss_avg": 0.04675563164055348,
      "epoch": 3.718376159667969e-05,
      "step": 38990
    },
    {
      "action_loss_avg": 0.000978383352048695,
      "epoch": 3.718376159667969e-05,
      "step": 38990
    },
    {
      "epoch": 3.718376159667969e-05,
      "step": 38990,
      "training_step_time": 9.420835018157959
    },
    {
      "epoch": 3.7184715270996093e-05,
      "model_forward_time": 1.728259801864624,
      "step": 38991
    },
    {
      "epoch": 3.7184715270996093e-05,
      "step": 38991,
      "training_step_time": 7.514523506164551
    },
    {
      "epoch": 3.71856689453125e-05,
      "model_forward_time": 1.7292370796203613,
      "step": 38992
    },
    {
      "epoch": 3.71856689453125e-05,
      "step": 38992,
      "training_step_time": 7.784214496612549
    },
    {
      "epoch": 3.718662261962891e-05,
      "model_forward_time": 1.7056403160095215,
      "step": 38993
    },
    {
      "epoch": 3.718662261962891e-05,
      "step": 38993,
      "training_step_time": 8.3784818649292
    },
    {
      "epoch": 3.718757629394531e-05,
      "model_forward_time": 1.7304027080535889,
      "step": 38994
    },
    {
      "epoch": 3.718757629394531e-05,
      "step": 38994,
      "training_step_time": 8.490746974945068
    },
    {
      "epoch": 3.7188529968261716e-05,
      "model_forward_time": 1.7325525283813477,
      "step": 38995
    },
    {
      "epoch": 3.7188529968261716e-05,
      "step": 38995,
      "training_step_time": 8.44148850440979
    },
    {
      "epoch": 3.718948364257813e-05,
      "model_forward_time": 1.7267992496490479,
      "step": 38996
    },
    {
      "epoch": 3.718948364257813e-05,
      "step": 38996,
      "training_step_time": 8.62756896018982
    },
    {
      "epoch": 3.719043731689453e-05,
      "model_forward_time": 1.742464303970337,
      "step": 38997
    },
    {
      "epoch": 3.719043731689453e-05,
      "step": 38997,
      "training_step_time": 8.381606340408325
    },
    {
      "epoch": 3.7191390991210935e-05,
      "model_forward_time": 1.7615909576416016,
      "step": 38998
    },
    {
      "epoch": 3.7191390991210935e-05,
      "step": 38998,
      "training_step_time": 8.156157732009888
    },
    {
      "epoch": 3.7192344665527346e-05,
      "model_forward_time": 1.7749114036560059,
      "step": 38999
    },
    {
      "epoch": 3.7192344665527346e-05,
      "step": 38999,
      "training_step_time": 8.481292963027954
    },
    {
      "epoch": 3.719329833984375e-05,
      "grad_norm": 0.12831392884254456,
      "learning_rate": 7.158920902039521e-06,
      "loss": 0.0587,
      "step": 39000
    },
    {
      "epoch": 3.719329833984375e-05,
      "model_forward_time": 1.753551959991455,
      "step": 39000
    },
    {
      "dynamics_loss_avg": 0.06172976810485124,
      "epoch": 3.719329833984375e-05,
      "step": 39000
    },
    {
      "action_loss_avg": 0.002073976263636723,
      "epoch": 3.719329833984375e-05,
      "step": 39000
    },
    {
      "epoch": 3.719329833984375e-05,
      "step": 39000,
      "training_step_time": 7.341411352157593
    },
    {
      "epoch": 3.7194252014160154e-05,
      "model_forward_time": 1.7253460884094238,
      "step": 39001
    },
    {
      "epoch": 3.7194252014160154e-05,
      "step": 39001,
      "training_step_time": 8.7405264377594
    },
    {
      "epoch": 3.7195205688476565e-05,
      "model_forward_time": 1.7322454452514648,
      "step": 39002
    },
    {
      "epoch": 3.7195205688476565e-05,
      "step": 39002,
      "training_step_time": 8.432408571243286
    },
    {
      "epoch": 3.719615936279297e-05,
      "model_forward_time": 1.7275750637054443,
      "step": 39003
    },
    {
      "epoch": 3.719615936279297e-05,
      "step": 39003,
      "training_step_time": 9.115337610244751
    },
    {
      "epoch": 3.719711303710937e-05,
      "model_forward_time": 1.7271485328674316,
      "step": 39004
    },
    {
      "epoch": 3.719711303710937e-05,
      "step": 39004,
      "training_step_time": 11.379744529724121
    },
    {
      "epoch": 3.7198066711425784e-05,
      "model_forward_time": 1.7273731231689453,
      "step": 39005
    },
    {
      "epoch": 3.7198066711425784e-05,
      "step": 39005,
      "training_step_time": 9.636936664581299
    },
    {
      "epoch": 3.719902038574219e-05,
      "model_forward_time": 2.755913496017456,
      "step": 39006
    },
    {
      "epoch": 3.719902038574219e-05,
      "step": 39006,
      "training_step_time": 10.643873929977417
    },
    {
      "epoch": 3.719997406005859e-05,
      "model_forward_time": 1.727705955505371,
      "step": 39007
    },
    {
      "epoch": 3.719997406005859e-05,
      "step": 39007,
      "training_step_time": 8.981474876403809
    },
    {
      "epoch": 3.7200927734375e-05,
      "model_forward_time": 1.7264182567596436,
      "step": 39008
    },
    {
      "epoch": 3.7200927734375e-05,
      "step": 39008,
      "training_step_time": 9.315709352493286
    },
    {
      "epoch": 3.7201881408691407e-05,
      "model_forward_time": 1.7317523956298828,
      "step": 39009
    },
    {
      "epoch": 3.7201881408691407e-05,
      "step": 39009,
      "training_step_time": 6.9229302406311035
    },
    {
      "epoch": 3.720283508300781e-05,
      "grad_norm": 0.1273299753665924,
      "learning_rate": 7.157429392352515e-06,
      "loss": 0.0577,
      "step": 39010
    },
    {
      "epoch": 3.720283508300781e-05,
      "model_forward_time": 1.7231123447418213,
      "step": 39010
    },
    {
      "dynamics_loss_avg": 0.048967549297958615,
      "epoch": 3.720283508300781e-05,
      "step": 39010
    },
    {
      "action_loss_avg": 0.0021248611563351007,
      "epoch": 3.720283508300781e-05,
      "step": 39010
    },
    {
      "epoch": 3.720283508300781e-05,
      "step": 39010,
      "training_step_time": 6.878874063491821
    },
    {
      "epoch": 3.720378875732422e-05,
      "model_forward_time": 1.7141458988189697,
      "step": 39011
    },
    {
      "epoch": 3.720378875732422e-05,
      "step": 39011,
      "training_step_time": 6.827914237976074
    },
    {
      "epoch": 3.7204742431640625e-05,
      "model_forward_time": 1.7221167087554932,
      "step": 39012
    },
    {
      "epoch": 3.7204742431640625e-05,
      "step": 39012,
      "training_step_time": 6.89633846282959
    },
    {
      "epoch": 3.720569610595703e-05,
      "model_forward_time": 1.7221102714538574,
      "step": 39013
    },
    {
      "epoch": 3.720569610595703e-05,
      "step": 39013,
      "training_step_time": 6.9159300327301025
    },
    {
      "epoch": 3.720664978027344e-05,
      "model_forward_time": 1.7255229949951172,
      "step": 39014
    },
    {
      "epoch": 3.720664978027344e-05,
      "step": 39014,
      "training_step_time": 6.8686888217926025
    },
    {
      "epoch": 3.7207603454589844e-05,
      "model_forward_time": 1.7381672859191895,
      "step": 39015
    },
    {
      "epoch": 3.7207603454589844e-05,
      "step": 39015,
      "training_step_time": 6.909240484237671
    },
    {
      "epoch": 3.720855712890625e-05,
      "model_forward_time": 1.7299494743347168,
      "step": 39016
    },
    {
      "epoch": 3.720855712890625e-05,
      "step": 39016,
      "training_step_time": 6.87005090713501
    },
    {
      "epoch": 3.720951080322266e-05,
      "model_forward_time": 2.2768871784210205,
      "step": 39017
    },
    {
      "epoch": 3.720951080322266e-05,
      "step": 39017,
      "training_step_time": 7.450923204421997
    },
    {
      "epoch": 3.721046447753906e-05,
      "model_forward_time": 1.7266805171966553,
      "step": 39018
    },
    {
      "epoch": 3.721046447753906e-05,
      "step": 39018,
      "training_step_time": 8.559887647628784
    },
    {
      "epoch": 3.721141815185547e-05,
      "model_forward_time": 1.719536542892456,
      "step": 39019
    },
    {
      "epoch": 3.721141815185547e-05,
      "step": 39019,
      "training_step_time": 6.894021034240723
    },
    {
      "epoch": 3.721237182617188e-05,
      "grad_norm": 0.1567494124174118,
      "learning_rate": 7.155937646732275e-06,
      "loss": 0.062,
      "step": 39020
    },
    {
      "epoch": 3.721237182617188e-05,
      "model_forward_time": 1.7138726711273193,
      "step": 39020
    },
    {
      "dynamics_loss_avg": 0.0683280486613512,
      "epoch": 3.721237182617188e-05,
      "step": 39020
    },
    {
      "action_loss_avg": 0.0028463851311244072,
      "epoch": 3.721237182617188e-05,
      "step": 39020
    },
    {
      "epoch": 3.721237182617188e-05,
      "step": 39020,
      "training_step_time": 6.78291654586792
    },
    {
      "epoch": 3.721332550048828e-05,
      "model_forward_time": 1.7224323749542236,
      "step": 39021
    },
    {
      "epoch": 3.721332550048828e-05,
      "step": 39021,
      "training_step_time": 6.800584316253662
    },
    {
      "epoch": 3.7214279174804686e-05,
      "model_forward_time": 1.7279860973358154,
      "step": 39022
    },
    {
      "epoch": 3.7214279174804686e-05,
      "step": 39022,
      "training_step_time": 6.879660367965698
    },
    {
      "epoch": 3.72152328491211e-05,
      "model_forward_time": 1.7277586460113525,
      "step": 39023
    },
    {
      "epoch": 3.72152328491211e-05,
      "step": 39023,
      "training_step_time": 6.814512014389038
    },
    {
      "epoch": 3.72161865234375e-05,
      "model_forward_time": 1.7379889488220215,
      "step": 39024
    },
    {
      "epoch": 3.72161865234375e-05,
      "step": 39024,
      "training_step_time": 6.864619255065918
    },
    {
      "epoch": 3.7217140197753905e-05,
      "model_forward_time": 1.7168562412261963,
      "step": 39025
    },
    {
      "epoch": 3.7217140197753905e-05,
      "step": 39025,
      "training_step_time": 6.86553168296814
    },
    {
      "epoch": 3.7218093872070316e-05,
      "model_forward_time": 2.289726734161377,
      "step": 39026
    },
    {
      "epoch": 3.7218093872070316e-05,
      "step": 39026,
      "training_step_time": 7.503732204437256
    },
    {
      "epoch": 3.721904754638672e-05,
      "model_forward_time": 1.7269444465637207,
      "step": 39027
    },
    {
      "epoch": 3.721904754638672e-05,
      "step": 39027,
      "training_step_time": 7.854632139205933
    },
    {
      "epoch": 3.7220001220703124e-05,
      "model_forward_time": 1.7336478233337402,
      "step": 39028
    },
    {
      "epoch": 3.7220001220703124e-05,
      "step": 39028,
      "training_step_time": 7.417139530181885
    },
    {
      "epoch": 3.7220954895019535e-05,
      "model_forward_time": 1.7265300750732422,
      "step": 39029
    },
    {
      "epoch": 3.7220954895019535e-05,
      "step": 39029,
      "training_step_time": 6.866464376449585
    },
    {
      "epoch": 3.722190856933594e-05,
      "grad_norm": 0.17858007550239563,
      "learning_rate": 7.154445665341933e-06,
      "loss": 0.0597,
      "step": 39030
    },
    {
      "epoch": 3.722190856933594e-05,
      "model_forward_time": 1.7303566932678223,
      "step": 39030
    },
    {
      "dynamics_loss_avg": 0.05801205188035965,
      "epoch": 3.722190856933594e-05,
      "step": 39030
    },
    {
      "action_loss_avg": 0.002052163815824315,
      "epoch": 3.722190856933594e-05,
      "step": 39030
    },
    {
      "epoch": 3.722190856933594e-05,
      "step": 39030,
      "training_step_time": 6.833885431289673
    },
    {
      "epoch": 3.722286224365234e-05,
      "model_forward_time": 1.7321877479553223,
      "step": 39031
    },
    {
      "epoch": 3.722286224365234e-05,
      "step": 39031,
      "training_step_time": 6.8619935512542725
    },
    {
      "epoch": 3.722381591796875e-05,
      "model_forward_time": 1.7219750881195068,
      "step": 39032
    },
    {
      "epoch": 3.722381591796875e-05,
      "step": 39032,
      "training_step_time": 6.875285387039185
    },
    {
      "epoch": 3.722476959228516e-05,
      "model_forward_time": 1.7414259910583496,
      "step": 39033
    },
    {
      "epoch": 3.722476959228516e-05,
      "step": 39033,
      "training_step_time": 6.879413604736328
    },
    {
      "epoch": 3.722572326660156e-05,
      "model_forward_time": 1.73103666305542,
      "step": 39034
    },
    {
      "epoch": 3.722572326660156e-05,
      "step": 39034,
      "training_step_time": 6.860458135604858
    },
    {
      "epoch": 3.7226676940917965e-05,
      "model_forward_time": 2.3051106929779053,
      "step": 39035
    },
    {
      "epoch": 3.7226676940917965e-05,
      "step": 39035,
      "training_step_time": 7.177502870559692
    },
    {
      "epoch": 3.7227630615234376e-05,
      "model_forward_time": 1.7306499481201172,
      "step": 39036
    },
    {
      "epoch": 3.7227630615234376e-05,
      "step": 39036,
      "training_step_time": 10.344213724136353
    },
    {
      "epoch": 3.722858428955078e-05,
      "model_forward_time": 1.7308814525604248,
      "step": 39037
    },
    {
      "epoch": 3.722858428955078e-05,
      "step": 39037,
      "training_step_time": 7.736203908920288
    },
    {
      "epoch": 3.7229537963867184e-05,
      "model_forward_time": 1.7276387214660645,
      "step": 39038
    },
    {
      "epoch": 3.7229537963867184e-05,
      "step": 39038,
      "training_step_time": 6.865656852722168
    },
    {
      "epoch": 3.7230491638183595e-05,
      "model_forward_time": 1.7463710308074951,
      "step": 39039
    },
    {
      "epoch": 3.7230491638183595e-05,
      "step": 39039,
      "training_step_time": 6.830934524536133
    },
    {
      "epoch": 3.72314453125e-05,
      "grad_norm": 0.15855315327644348,
      "learning_rate": 7.152953448344655e-06,
      "loss": 0.0606,
      "step": 39040
    },
    {
      "epoch": 3.72314453125e-05,
      "model_forward_time": 0.7082219123840332,
      "step": 39040
    },
    {
      "dynamics_loss_avg": 0.055264301411807534,
      "epoch": 3.72314453125e-05,
      "step": 39040
    },
    {
      "action_loss_avg": 0.0030078812735155223,
      "epoch": 3.72314453125e-05,
      "step": 39040
    },
    {
      "epoch": 3.72314453125e-05,
      "step": 39040,
      "training_step_time": 6.836461067199707
    },
    {
      "epoch": 3.72323989868164e-05,
      "model_forward_time": 1.7260997295379639,
      "step": 39041
    },
    {
      "epoch": 3.72323989868164e-05,
      "step": 39041,
      "training_step_time": 6.9063944816589355
    },
    {
      "epoch": 3.7233352661132814e-05,
      "model_forward_time": 1.7288844585418701,
      "step": 39042
    },
    {
      "epoch": 3.7233352661132814e-05,
      "step": 39042,
      "training_step_time": 6.883866786956787
    },
    {
      "epoch": 3.723430633544922e-05,
      "model_forward_time": 1.7341930866241455,
      "step": 39043
    },
    {
      "epoch": 3.723430633544922e-05,
      "step": 39043,
      "training_step_time": 6.866826295852661
    },
    {
      "epoch": 3.723526000976562e-05,
      "model_forward_time": 1.7640340328216553,
      "step": 39044
    },
    {
      "epoch": 3.723526000976562e-05,
      "step": 39044,
      "training_step_time": 7.181906700134277
    },
    {
      "epoch": 3.723621368408203e-05,
      "model_forward_time": 1.8899157047271729,
      "step": 39045
    },
    {
      "epoch": 3.723621368408203e-05,
      "step": 39045,
      "training_step_time": 9.551191568374634
    },
    {
      "epoch": 3.723716735839844e-05,
      "model_forward_time": 1.7253565788269043,
      "step": 39046
    },
    {
      "epoch": 3.723716735839844e-05,
      "step": 39046,
      "training_step_time": 7.7852866649627686
    },
    {
      "epoch": 3.723812103271484e-05,
      "model_forward_time": 1.7387995719909668,
      "step": 39047
    },
    {
      "epoch": 3.723812103271484e-05,
      "step": 39047,
      "training_step_time": 6.8716418743133545
    },
    {
      "epoch": 3.723907470703125e-05,
      "model_forward_time": 1.7155845165252686,
      "step": 39048
    },
    {
      "epoch": 3.723907470703125e-05,
      "step": 39048,
      "training_step_time": 6.845929861068726
    },
    {
      "epoch": 3.7240028381347656e-05,
      "model_forward_time": 1.7127041816711426,
      "step": 39049
    },
    {
      "epoch": 3.7240028381347656e-05,
      "step": 39049,
      "training_step_time": 6.842115163803101
    },
    {
      "epoch": 3.724098205566406e-05,
      "grad_norm": 0.15897998213768005,
      "learning_rate": 7.151460995903624e-06,
      "loss": 0.0578,
      "step": 39050
    },
    {
      "epoch": 3.724098205566406e-05,
      "model_forward_time": 1.7155280113220215,
      "step": 39050
    },
    {
      "dynamics_loss_avg": 0.06013678703457117,
      "epoch": 3.724098205566406e-05,
      "step": 39050
    },
    {
      "action_loss_avg": 0.0023725004051811994,
      "epoch": 3.724098205566406e-05,
      "step": 39050
    },
    {
      "epoch": 3.724098205566406e-05,
      "step": 39050,
      "training_step_time": 6.842491388320923
    },
    {
      "epoch": 3.724193572998047e-05,
      "model_forward_time": 1.705510139465332,
      "step": 39051
    },
    {
      "epoch": 3.724193572998047e-05,
      "step": 39051,
      "training_step_time": 6.822898626327515
    },
    {
      "epoch": 3.7242889404296875e-05,
      "model_forward_time": 1.7236311435699463,
      "step": 39052
    },
    {
      "epoch": 3.7242889404296875e-05,
      "step": 39052,
      "training_step_time": 6.867981195449829
    },
    {
      "epoch": 3.724384307861328e-05,
      "model_forward_time": 1.7467288970947266,
      "step": 39053
    },
    {
      "epoch": 3.724384307861328e-05,
      "step": 39053,
      "training_step_time": 7.176660776138306
    },
    {
      "epoch": 3.724479675292969e-05,
      "model_forward_time": 2.284858465194702,
      "step": 39054
    },
    {
      "epoch": 3.724479675292969e-05,
      "step": 39054,
      "training_step_time": 8.706308603286743
    },
    {
      "epoch": 3.7245750427246093e-05,
      "model_forward_time": 1.6874973773956299,
      "step": 39055
    },
    {
      "epoch": 3.7245750427246093e-05,
      "step": 39055,
      "training_step_time": 7.786096096038818
    },
    {
      "epoch": 3.72467041015625e-05,
      "model_forward_time": 1.6969046592712402,
      "step": 39056
    },
    {
      "epoch": 3.72467041015625e-05,
      "step": 39056,
      "training_step_time": 7.410170555114746
    },
    {
      "epoch": 3.724765777587891e-05,
      "model_forward_time": 1.7016246318817139,
      "step": 39057
    },
    {
      "epoch": 3.724765777587891e-05,
      "step": 39057,
      "training_step_time": 6.843156099319458
    },
    {
      "epoch": 3.724861145019531e-05,
      "model_forward_time": 1.7078723907470703,
      "step": 39058
    },
    {
      "epoch": 3.724861145019531e-05,
      "step": 39058,
      "training_step_time": 6.826189279556274
    },
    {
      "epoch": 3.7249565124511716e-05,
      "model_forward_time": 1.6992902755737305,
      "step": 39059
    },
    {
      "epoch": 3.7249565124511716e-05,
      "step": 39059,
      "training_step_time": 6.803918361663818
    },
    {
      "epoch": 3.725051879882813e-05,
      "grad_norm": 0.18698625266551971,
      "learning_rate": 7.149968308182053e-06,
      "loss": 0.0577,
      "step": 39060
    },
    {
      "epoch": 3.725051879882813e-05,
      "model_forward_time": 1.7036106586456299,
      "step": 39060
    },
    {
      "dynamics_loss_avg": 0.04106855969876051,
      "epoch": 3.725051879882813e-05,
      "step": 39060
    },
    {
      "action_loss_avg": 0.0013614289637189358,
      "epoch": 3.725051879882813e-05,
      "step": 39060
    },
    {
      "epoch": 3.725051879882813e-05,
      "step": 39060,
      "training_step_time": 6.796807289123535
    },
    {
      "epoch": 3.725147247314453e-05,
      "model_forward_time": 1.7117316722869873,
      "step": 39061
    },
    {
      "epoch": 3.725147247314453e-05,
      "step": 39061,
      "training_step_time": 6.803943634033203
    },
    {
      "epoch": 3.7252426147460935e-05,
      "model_forward_time": 1.7016422748565674,
      "step": 39062
    },
    {
      "epoch": 3.7252426147460935e-05,
      "step": 39062,
      "training_step_time": 6.841005563735962
    },
    {
      "epoch": 3.7253379821777346e-05,
      "model_forward_time": 2.29624342918396,
      "step": 39063
    },
    {
      "epoch": 3.7253379821777346e-05,
      "step": 39063,
      "training_step_time": 7.7010955810546875
    },
    {
      "epoch": 3.725433349609375e-05,
      "model_forward_time": 1.701589584350586,
      "step": 39064
    },
    {
      "epoch": 3.725433349609375e-05,
      "step": 39064,
      "training_step_time": 8.353121995925903
    },
    {
      "epoch": 3.7255287170410154e-05,
      "model_forward_time": 1.6897704601287842,
      "step": 39065
    },
    {
      "epoch": 3.7255287170410154e-05,
      "step": 39065,
      "training_step_time": 7.413660764694214
    },
    {
      "epoch": 3.7256240844726565e-05,
      "model_forward_time": 1.7030704021453857,
      "step": 39066
    },
    {
      "epoch": 3.7256240844726565e-05,
      "step": 39066,
      "training_step_time": 6.8460023403167725
    },
    {
      "epoch": 3.725719451904297e-05,
      "model_forward_time": 1.707906723022461,
      "step": 39067
    },
    {
      "epoch": 3.725719451904297e-05,
      "step": 39067,
      "training_step_time": 6.899324417114258
    },
    {
      "epoch": 3.725814819335937e-05,
      "model_forward_time": 1.697361707687378,
      "step": 39068
    },
    {
      "epoch": 3.725814819335937e-05,
      "step": 39068,
      "training_step_time": 6.844940185546875
    },
    {
      "epoch": 3.7259101867675784e-05,
      "model_forward_time": 1.704805612564087,
      "step": 39069
    },
    {
      "epoch": 3.7259101867675784e-05,
      "step": 39069,
      "training_step_time": 6.884329557418823
    },
    {
      "epoch": 3.726005554199219e-05,
      "grad_norm": 0.16923171281814575,
      "learning_rate": 7.14847538534318e-06,
      "loss": 0.0595,
      "step": 39070
    },
    {
      "epoch": 3.726005554199219e-05,
      "model_forward_time": 1.6991655826568604,
      "step": 39070
    },
    {
      "dynamics_loss_avg": 0.06556919030845165,
      "epoch": 3.726005554199219e-05,
      "step": 39070
    },
    {
      "action_loss_avg": 0.0017147799546364695,
      "epoch": 3.726005554199219e-05,
      "step": 39070
    },
    {
      "epoch": 3.726005554199219e-05,
      "step": 39070,
      "training_step_time": 6.821694850921631
    },
    {
      "epoch": 3.726100921630859e-05,
      "model_forward_time": 1.7021405696868896,
      "step": 39071
    },
    {
      "epoch": 3.726100921630859e-05,
      "step": 39071,
      "training_step_time": 6.808278322219849
    },
    {
      "epoch": 3.7261962890625e-05,
      "model_forward_time": 2.295079469680786,
      "step": 39072
    },
    {
      "epoch": 3.7261962890625e-05,
      "step": 39072,
      "training_step_time": 8.549087285995483
    },
    {
      "epoch": 3.726291656494141e-05,
      "model_forward_time": 1.7077224254608154,
      "step": 39073
    },
    {
      "epoch": 3.726291656494141e-05,
      "step": 39073,
      "training_step_time": 8.821139335632324
    },
    {
      "epoch": 3.726387023925781e-05,
      "model_forward_time": 1.7060537338256836,
      "step": 39074
    },
    {
      "epoch": 3.726387023925781e-05,
      "step": 39074,
      "training_step_time": 7.572134733200073
    },
    {
      "epoch": 3.726482391357422e-05,
      "model_forward_time": 1.701918601989746,
      "step": 39075
    },
    {
      "epoch": 3.726482391357422e-05,
      "step": 39075,
      "training_step_time": 6.913665294647217
    },
    {
      "epoch": 3.7265777587890625e-05,
      "model_forward_time": 1.7077040672302246,
      "step": 39076
    },
    {
      "epoch": 3.7265777587890625e-05,
      "step": 39076,
      "training_step_time": 6.805050611495972
    },
    {
      "epoch": 3.726673126220703e-05,
      "model_forward_time": 1.3284804821014404,
      "step": 39077
    },
    {
      "epoch": 3.726673126220703e-05,
      "step": 39077,
      "training_step_time": 6.825435161590576
    },
    {
      "epoch": 3.726768493652344e-05,
      "model_forward_time": 1.7039711475372314,
      "step": 39078
    },
    {
      "epoch": 3.726768493652344e-05,
      "step": 39078,
      "training_step_time": 6.892348051071167
    },
    {
      "epoch": 3.7268638610839844e-05,
      "model_forward_time": 1.7040820121765137,
      "step": 39079
    },
    {
      "epoch": 3.7268638610839844e-05,
      "step": 39079,
      "training_step_time": 6.837706565856934
    },
    {
      "epoch": 3.726959228515625e-05,
      "grad_norm": 0.13968870043754578,
      "learning_rate": 7.14698222755027e-06,
      "loss": 0.0598,
      "step": 39080
    },
    {
      "epoch": 3.726959228515625e-05,
      "model_forward_time": 1.6835310459136963,
      "step": 39080
    },
    {
      "dynamics_loss_avg": 0.048965674452483655,
      "epoch": 3.726959228515625e-05,
      "step": 39080
    },
    {
      "action_loss_avg": 0.0019144697522278874,
      "epoch": 3.726959228515625e-05,
      "step": 39080
    },
    {
      "epoch": 3.726959228515625e-05,
      "step": 39080,
      "training_step_time": 6.805952310562134
    },
    {
      "epoch": 3.727054595947266e-05,
      "model_forward_time": 1.7167232036590576,
      "step": 39081
    },
    {
      "epoch": 3.727054595947266e-05,
      "step": 39081,
      "training_step_time": 7.648093938827515
    },
    {
      "epoch": 3.727149963378906e-05,
      "model_forward_time": 2.237917900085449,
      "step": 39082
    },
    {
      "epoch": 3.727149963378906e-05,
      "step": 39082,
      "training_step_time": 9.979567289352417
    },
    {
      "epoch": 3.727245330810547e-05,
      "model_forward_time": 1.7025609016418457,
      "step": 39083
    },
    {
      "epoch": 3.727245330810547e-05,
      "step": 39083,
      "training_step_time": 7.63136887550354
    },
    {
      "epoch": 3.727340698242188e-05,
      "model_forward_time": 1.6995184421539307,
      "step": 39084
    },
    {
      "epoch": 3.727340698242188e-05,
      "step": 39084,
      "training_step_time": 6.831580638885498
    },
    {
      "epoch": 3.727436065673828e-05,
      "model_forward_time": 1.713907241821289,
      "step": 39085
    },
    {
      "epoch": 3.727436065673828e-05,
      "step": 39085,
      "training_step_time": 6.842893362045288
    },
    {
      "epoch": 3.7275314331054686e-05,
      "model_forward_time": 1.3407196998596191,
      "step": 39086
    },
    {
      "epoch": 3.7275314331054686e-05,
      "step": 39086,
      "training_step_time": 6.854173898696899
    },
    {
      "epoch": 3.72762680053711e-05,
      "model_forward_time": 1.7148408889770508,
      "step": 39087
    },
    {
      "epoch": 3.72762680053711e-05,
      "step": 39087,
      "training_step_time": 6.852811098098755
    },
    {
      "epoch": 3.72772216796875e-05,
      "model_forward_time": 1.719085454940796,
      "step": 39088
    },
    {
      "epoch": 3.72772216796875e-05,
      "step": 39088,
      "training_step_time": 6.818427562713623
    },
    {
      "epoch": 3.7278175354003905e-05,
      "model_forward_time": 1.7004573345184326,
      "step": 39089
    },
    {
      "epoch": 3.7278175354003905e-05,
      "step": 39089,
      "training_step_time": 6.810406684875488
    },
    {
      "epoch": 3.7279129028320316e-05,
      "grad_norm": 0.07911326736211777,
      "learning_rate": 7.14548883496661e-06,
      "loss": 0.0602,
      "step": 39090
    },
    {
      "epoch": 3.7279129028320316e-05,
      "model_forward_time": 1.3490586280822754,
      "step": 39090
    },
    {
      "dynamics_loss_avg": 0.0645388599485159,
      "epoch": 3.7279129028320316e-05,
      "step": 39090
    },
    {
      "action_loss_avg": 0.0015012370073236524,
      "epoch": 3.7279129028320316e-05,
      "step": 39090
    },
    {
      "epoch": 3.7279129028320316e-05,
      "step": 39090,
      "training_step_time": 7.59143590927124
    },
    {
      "epoch": 3.728008270263672e-05,
      "model_forward_time": 2.253603458404541,
      "step": 39091
    },
    {
      "epoch": 3.728008270263672e-05,
      "step": 39091,
      "training_step_time": 10.975887060165405
    },
    {
      "epoch": 3.7281036376953124e-05,
      "model_forward_time": 1.7045531272888184,
      "step": 39092
    },
    {
      "epoch": 3.7281036376953124e-05,
      "step": 39092,
      "training_step_time": 8.369166374206543
    },
    {
      "epoch": 3.7281990051269535e-05,
      "model_forward_time": 1.705913782119751,
      "step": 39093
    },
    {
      "epoch": 3.7281990051269535e-05,
      "step": 39093,
      "training_step_time": 7.479045152664185
    },
    {
      "epoch": 3.728294372558594e-05,
      "model_forward_time": 1.7102439403533936,
      "step": 39094
    },
    {
      "epoch": 3.728294372558594e-05,
      "step": 39094,
      "training_step_time": 6.813817501068115
    },
    {
      "epoch": 3.728389739990234e-05,
      "model_forward_time": 1.7238152027130127,
      "step": 39095
    },
    {
      "epoch": 3.728389739990234e-05,
      "step": 39095,
      "training_step_time": 6.799052715301514
    },
    {
      "epoch": 3.728485107421875e-05,
      "model_forward_time": 1.727041482925415,
      "step": 39096
    },
    {
      "epoch": 3.728485107421875e-05,
      "step": 39096,
      "training_step_time": 6.829582691192627
    },
    {
      "epoch": 3.728580474853516e-05,
      "model_forward_time": 1.7255988121032715,
      "step": 39097
    },
    {
      "epoch": 3.728580474853516e-05,
      "step": 39097,
      "training_step_time": 6.800841331481934
    },
    {
      "epoch": 3.728675842285156e-05,
      "model_forward_time": 1.7229583263397217,
      "step": 39098
    },
    {
      "epoch": 3.728675842285156e-05,
      "step": 39098,
      "training_step_time": 6.79740571975708
    },
    {
      "epoch": 3.7287712097167966e-05,
      "model_forward_time": 1.7217833995819092,
      "step": 39099
    },
    {
      "epoch": 3.7287712097167966e-05,
      "step": 39099,
      "training_step_time": 7.426148176193237
    },
    {
      "epoch": 3.7288665771484376e-05,
      "grad_norm": 0.11512716859579086,
      "learning_rate": 7.143995207755518e-06,
      "loss": 0.0586,
      "step": 39100
    },
    {
      "epoch": 3.7288665771484376e-05,
      "model_forward_time": 2.330662727355957,
      "step": 39100
    },
    {
      "dynamics_loss_avg": 0.06376397702842951,
      "epoch": 3.7288665771484376e-05,
      "step": 39100
    },
    {
      "action_loss_avg": 0.002218693250324577,
      "epoch": 3.7288665771484376e-05,
      "step": 39100
    },
    {
      "epoch": 3.7288665771484376e-05,
      "step": 39100,
      "training_step_time": 8.870382070541382
    },
    {
      "epoch": 3.728961944580078e-05,
      "model_forward_time": 1.726515769958496,
      "step": 39101
    },
    {
      "epoch": 3.728961944580078e-05,
      "step": 39101,
      "training_step_time": 9.286252975463867
    },
    {
      "epoch": 3.7290573120117184e-05,
      "model_forward_time": 1.7257299423217773,
      "step": 39102
    },
    {
      "epoch": 3.7290573120117184e-05,
      "step": 39102,
      "training_step_time": 7.408635854721069
    },
    {
      "epoch": 3.7291526794433595e-05,
      "model_forward_time": 1.7270002365112305,
      "step": 39103
    },
    {
      "epoch": 3.7291526794433595e-05,
      "step": 39103,
      "training_step_time": 6.808668851852417
    },
    {
      "epoch": 3.729248046875e-05,
      "model_forward_time": 1.7217004299163818,
      "step": 39104
    },
    {
      "epoch": 3.729248046875e-05,
      "step": 39104,
      "training_step_time": 6.789072036743164
    },
    {
      "epoch": 3.72934341430664e-05,
      "model_forward_time": 1.7329339981079102,
      "step": 39105
    },
    {
      "epoch": 3.72934341430664e-05,
      "step": 39105,
      "training_step_time": 6.796786785125732
    },
    {
      "epoch": 3.7294387817382814e-05,
      "model_forward_time": 1.723418951034546,
      "step": 39106
    },
    {
      "epoch": 3.7294387817382814e-05,
      "step": 39106,
      "training_step_time": 6.814599514007568
    },
    {
      "epoch": 3.729534149169922e-05,
      "model_forward_time": 1.7279307842254639,
      "step": 39107
    },
    {
      "epoch": 3.729534149169922e-05,
      "step": 39107,
      "training_step_time": 6.794321775436401
    },
    {
      "epoch": 3.729629516601562e-05,
      "model_forward_time": 1.7278783321380615,
      "step": 39108
    },
    {
      "epoch": 3.729629516601562e-05,
      "step": 39108,
      "training_step_time": 7.451614856719971
    },
    {
      "epoch": 3.729724884033203e-05,
      "model_forward_time": 2.3287739753723145,
      "step": 39109
    },
    {
      "epoch": 3.729724884033203e-05,
      "step": 39109,
      "training_step_time": 9.460537910461426
    },
    {
      "epoch": 3.729820251464844e-05,
      "grad_norm": 0.14390972256660461,
      "learning_rate": 7.142501346080333e-06,
      "loss": 0.0611,
      "step": 39110
    },
    {
      "epoch": 3.729820251464844e-05,
      "model_forward_time": 1.7197825908660889,
      "step": 39110
    },
    {
      "dynamics_loss_avg": 0.05688803382217884,
      "epoch": 3.729820251464844e-05,
      "step": 39110
    },
    {
      "action_loss_avg": 0.0027368435985408723,
      "epoch": 3.729820251464844e-05,
      "step": 39110
    },
    {
      "epoch": 3.729820251464844e-05,
      "step": 39110,
      "training_step_time": 8.481019973754883
    },
    {
      "epoch": 3.729915618896484e-05,
      "model_forward_time": 1.7310519218444824,
      "step": 39111
    },
    {
      "epoch": 3.729915618896484e-05,
      "step": 39111,
      "training_step_time": 7.417514324188232
    },
    {
      "epoch": 3.730010986328125e-05,
      "model_forward_time": 1.7072937488555908,
      "step": 39112
    },
    {
      "epoch": 3.730010986328125e-05,
      "step": 39112,
      "training_step_time": 6.787945747375488
    },
    {
      "epoch": 3.7301063537597656e-05,
      "model_forward_time": 1.724367618560791,
      "step": 39113
    },
    {
      "epoch": 3.7301063537597656e-05,
      "step": 39113,
      "training_step_time": 6.792416572570801
    },
    {
      "epoch": 3.730201721191406e-05,
      "model_forward_time": 1.7238245010375977,
      "step": 39114
    },
    {
      "epoch": 3.730201721191406e-05,
      "step": 39114,
      "training_step_time": 6.847360610961914
    },
    {
      "epoch": 3.730297088623047e-05,
      "model_forward_time": 1.726219892501831,
      "step": 39115
    },
    {
      "epoch": 3.730297088623047e-05,
      "step": 39115,
      "training_step_time": 6.814842939376831
    },
    {
      "epoch": 3.7303924560546875e-05,
      "model_forward_time": 1.7249603271484375,
      "step": 39116
    },
    {
      "epoch": 3.7303924560546875e-05,
      "step": 39116,
      "training_step_time": 6.918106555938721
    },
    {
      "epoch": 3.730487823486328e-05,
      "model_forward_time": 1.7303259372711182,
      "step": 39117
    },
    {
      "epoch": 3.730487823486328e-05,
      "step": 39117,
      "training_step_time": 7.4749767780303955
    },
    {
      "epoch": 3.730583190917969e-05,
      "model_forward_time": 1.7216713428497314,
      "step": 39118
    },
    {
      "epoch": 3.730583190917969e-05,
      "step": 39118,
      "training_step_time": 9.124471187591553
    },
    {
      "epoch": 3.7306785583496094e-05,
      "model_forward_time": 1.7220919132232666,
      "step": 39119
    },
    {
      "epoch": 3.7306785583496094e-05,
      "step": 39119,
      "training_step_time": 10.876615285873413
    },
    {
      "epoch": 3.73077392578125e-05,
      "grad_norm": 0.10380997508764267,
      "learning_rate": 7.141007250104421e-06,
      "loss": 0.0602,
      "step": 39120
    },
    {
      "epoch": 3.73077392578125e-05,
      "model_forward_time": 1.7202744483947754,
      "step": 39120
    },
    {
      "dynamics_loss_avg": 0.060241157189011574,
      "epoch": 3.73077392578125e-05,
      "step": 39120
    },
    {
      "action_loss_avg": 0.0022518944460898638,
      "epoch": 3.73077392578125e-05,
      "step": 39120
    },
    {
      "epoch": 3.73077392578125e-05,
      "step": 39120,
      "training_step_time": 7.625259876251221
    },
    {
      "epoch": 3.730869293212891e-05,
      "model_forward_time": 1.729278564453125,
      "step": 39121
    },
    {
      "epoch": 3.730869293212891e-05,
      "step": 39121,
      "training_step_time": 7.0530171394348145
    },
    {
      "epoch": 3.730964660644531e-05,
      "model_forward_time": 1.7260963916778564,
      "step": 39122
    },
    {
      "epoch": 3.730964660644531e-05,
      "step": 39122,
      "training_step_time": 6.788634538650513
    },
    {
      "epoch": 3.7310600280761716e-05,
      "model_forward_time": 1.812697172164917,
      "step": 39123
    },
    {
      "epoch": 3.7310600280761716e-05,
      "step": 39123,
      "training_step_time": 6.806923866271973
    },
    {
      "epoch": 3.731155395507813e-05,
      "model_forward_time": 1.8048129081726074,
      "step": 39124
    },
    {
      "epoch": 3.731155395507813e-05,
      "step": 39124,
      "training_step_time": 6.7925496101379395
    },
    {
      "epoch": 3.731250762939453e-05,
      "model_forward_time": 1.724289894104004,
      "step": 39125
    },
    {
      "epoch": 3.731250762939453e-05,
      "step": 39125,
      "training_step_time": 6.7983012199401855
    },
    {
      "epoch": 3.7313461303710935e-05,
      "model_forward_time": 1.7232091426849365,
      "step": 39126
    },
    {
      "epoch": 3.7313461303710935e-05,
      "step": 39126,
      "training_step_time": 6.8238420486450195
    },
    {
      "epoch": 3.7314414978027346e-05,
      "model_forward_time": 1.7260987758636475,
      "step": 39127
    },
    {
      "epoch": 3.7314414978027346e-05,
      "step": 39127,
      "training_step_time": 8.793090343475342
    },
    {
      "epoch": 3.731536865234375e-05,
      "model_forward_time": 2.327385425567627,
      "step": 39128
    },
    {
      "epoch": 3.731536865234375e-05,
      "step": 39128,
      "training_step_time": 10.178890466690063
    },
    {
      "epoch": 3.7316322326660154e-05,
      "model_forward_time": 1.72102952003479,
      "step": 39129
    },
    {
      "epoch": 3.7316322326660154e-05,
      "step": 39129,
      "training_step_time": 7.801754713058472
    },
    {
      "epoch": 3.7317276000976565e-05,
      "grad_norm": 0.13455688953399658,
      "learning_rate": 7.139512919991176e-06,
      "loss": 0.0626,
      "step": 39130
    },
    {
      "epoch": 3.7317276000976565e-05,
      "model_forward_time": 1.7209007740020752,
      "step": 39130
    },
    {
      "dynamics_loss_avg": 0.06408632136881351,
      "epoch": 3.7317276000976565e-05,
      "step": 39130
    },
    {
      "action_loss_avg": 0.002419666008790955,
      "epoch": 3.7317276000976565e-05,
      "step": 39130
    },
    {
      "epoch": 3.7317276000976565e-05,
      "step": 39130,
      "training_step_time": 7.414815902709961
    },
    {
      "epoch": 3.731822967529297e-05,
      "model_forward_time": 1.7265889644622803,
      "step": 39131
    },
    {
      "epoch": 3.731822967529297e-05,
      "step": 39131,
      "training_step_time": 6.836148738861084
    },
    {
      "epoch": 3.731918334960937e-05,
      "model_forward_time": 1.7274155616760254,
      "step": 39132
    },
    {
      "epoch": 3.731918334960937e-05,
      "step": 39132,
      "training_step_time": 6.825031995773315
    },
    {
      "epoch": 3.7320137023925784e-05,
      "model_forward_time": 1.7272450923919678,
      "step": 39133
    },
    {
      "epoch": 3.7320137023925784e-05,
      "step": 39133,
      "training_step_time": 6.824222564697266
    },
    {
      "epoch": 3.732109069824219e-05,
      "model_forward_time": 1.7202720642089844,
      "step": 39134
    },
    {
      "epoch": 3.732109069824219e-05,
      "step": 39134,
      "training_step_time": 6.831300735473633
    },
    {
      "epoch": 3.732204437255859e-05,
      "model_forward_time": 1.7209975719451904,
      "step": 39135
    },
    {
      "epoch": 3.732204437255859e-05,
      "step": 39135,
      "training_step_time": 6.844331502914429
    },
    {
      "epoch": 3.7322998046875e-05,
      "model_forward_time": 1.0119352340698242,
      "step": 39136
    },
    {
      "epoch": 3.7322998046875e-05,
      "step": 39136,
      "training_step_time": 7.691179513931274
    },
    {
      "epoch": 3.732395172119141e-05,
      "model_forward_time": 2.2845370769500732,
      "step": 39137
    },
    {
      "epoch": 3.732395172119141e-05,
      "step": 39137,
      "training_step_time": 9.096303701400757
    },
    {
      "epoch": 3.732490539550781e-05,
      "model_forward_time": 1.7148449420928955,
      "step": 39138
    },
    {
      "epoch": 3.732490539550781e-05,
      "step": 39138,
      "training_step_time": 8.71727705001831
    },
    {
      "epoch": 3.732585906982422e-05,
      "model_forward_time": 1.7309951782226562,
      "step": 39139
    },
    {
      "epoch": 3.732585906982422e-05,
      "step": 39139,
      "training_step_time": 7.478555679321289
    },
    {
      "epoch": 3.7326812744140626e-05,
      "grad_norm": 0.1026637926697731,
      "learning_rate": 7.138018355904015e-06,
      "loss": 0.0594,
      "step": 39140
    },
    {
      "epoch": 3.7326812744140626e-05,
      "model_forward_time": 1.7235453128814697,
      "step": 39140
    },
    {
      "dynamics_loss_avg": 0.06422309800982476,
      "epoch": 3.7326812744140626e-05,
      "step": 39140
    },
    {
      "action_loss_avg": 0.0018957449763547629,
      "epoch": 3.7326812744140626e-05,
      "step": 39140
    },
    {
      "epoch": 3.7326812744140626e-05,
      "step": 39140,
      "training_step_time": 6.817878484725952
    },
    {
      "epoch": 3.732776641845703e-05,
      "model_forward_time": 1.7305524349212646,
      "step": 39141
    },
    {
      "epoch": 3.732776641845703e-05,
      "step": 39141,
      "training_step_time": 6.800778865814209
    },
    {
      "epoch": 3.732872009277344e-05,
      "model_forward_time": 1.7245564460754395,
      "step": 39142
    },
    {
      "epoch": 3.732872009277344e-05,
      "step": 39142,
      "training_step_time": 6.771496295928955
    },
    {
      "epoch": 3.7329673767089844e-05,
      "model_forward_time": 1.7283778190612793,
      "step": 39143
    },
    {
      "epoch": 3.7329673767089844e-05,
      "step": 39143,
      "training_step_time": 6.8309056758880615
    },
    {
      "epoch": 3.733062744140625e-05,
      "model_forward_time": 1.7160654067993164,
      "step": 39144
    },
    {
      "epoch": 3.733062744140625e-05,
      "step": 39144,
      "training_step_time": 6.732842445373535
    },
    {
      "epoch": 3.733158111572266e-05,
      "model_forward_time": 1.7314403057098389,
      "step": 39145
    },
    {
      "epoch": 3.733158111572266e-05,
      "step": 39145,
      "training_step_time": 7.560662746429443
    },
    {
      "epoch": 3.733253479003906e-05,
      "model_forward_time": 2.3331966400146484,
      "step": 39146
    },
    {
      "epoch": 3.733253479003906e-05,
      "step": 39146,
      "training_step_time": 9.434722423553467
    },
    {
      "epoch": 3.733348846435547e-05,
      "model_forward_time": 1.7281162738800049,
      "step": 39147
    },
    {
      "epoch": 3.733348846435547e-05,
      "step": 39147,
      "training_step_time": 7.830926418304443
    },
    {
      "epoch": 3.733444213867188e-05,
      "model_forward_time": 1.7236802577972412,
      "step": 39148
    },
    {
      "epoch": 3.733444213867188e-05,
      "step": 39148,
      "training_step_time": 7.748534679412842
    },
    {
      "epoch": 3.733539581298828e-05,
      "model_forward_time": 1.7298858165740967,
      "step": 39149
    },
    {
      "epoch": 3.733539581298828e-05,
      "step": 39149,
      "training_step_time": 7.146965503692627
    },
    {
      "epoch": 3.7336349487304686e-05,
      "grad_norm": 0.13673166930675507,
      "learning_rate": 7.136523558006381e-06,
      "loss": 0.0592,
      "step": 39150
    },
    {
      "epoch": 3.7336349487304686e-05,
      "model_forward_time": 1.7351782321929932,
      "step": 39150
    },
    {
      "dynamics_loss_avg": 0.04942984227091074,
      "epoch": 3.7336349487304686e-05,
      "step": 39150
    },
    {
      "action_loss_avg": 0.002165766802500002,
      "epoch": 3.7336349487304686e-05,
      "step": 39150
    },
    {
      "epoch": 3.7336349487304686e-05,
      "step": 39150,
      "training_step_time": 6.868862867355347
    },
    {
      "epoch": 3.73373031616211e-05,
      "model_forward_time": 1.732572078704834,
      "step": 39151
    },
    {
      "epoch": 3.73373031616211e-05,
      "step": 39151,
      "training_step_time": 6.895514249801636
    },
    {
      "epoch": 3.73382568359375e-05,
      "model_forward_time": 1.7299275398254395,
      "step": 39152
    },
    {
      "epoch": 3.73382568359375e-05,
      "step": 39152,
      "training_step_time": 6.846339702606201
    },
    {
      "epoch": 3.7339210510253905e-05,
      "model_forward_time": 1.7313859462738037,
      "step": 39153
    },
    {
      "epoch": 3.7339210510253905e-05,
      "step": 39153,
      "training_step_time": 6.841287851333618
    },
    {
      "epoch": 3.7340164184570316e-05,
      "model_forward_time": 1.7304363250732422,
      "step": 39154
    },
    {
      "epoch": 3.7340164184570316e-05,
      "step": 39154,
      "training_step_time": 7.643145561218262
    },
    {
      "epoch": 3.734111785888672e-05,
      "model_forward_time": 2.296541929244995,
      "step": 39155
    },
    {
      "epoch": 3.734111785888672e-05,
      "step": 39155,
      "training_step_time": 11.125051498413086
    },
    {
      "epoch": 3.7342071533203124e-05,
      "model_forward_time": 1.7292449474334717,
      "step": 39156
    },
    {
      "epoch": 3.7342071533203124e-05,
      "step": 39156,
      "training_step_time": 8.439582586288452
    },
    {
      "epoch": 3.734302520751953e-05,
      "model_forward_time": 1.7284784317016602,
      "step": 39157
    },
    {
      "epoch": 3.734302520751953e-05,
      "step": 39157,
      "training_step_time": 7.679682970046997
    },
    {
      "epoch": 3.734397888183594e-05,
      "model_forward_time": 1.733621597290039,
      "step": 39158
    },
    {
      "epoch": 3.734397888183594e-05,
      "step": 39158,
      "training_step_time": 7.169928789138794
    },
    {
      "epoch": 3.734493255615234e-05,
      "model_forward_time": 1.7155344486236572,
      "step": 39159
    },
    {
      "epoch": 3.734493255615234e-05,
      "step": 39159,
      "training_step_time": 6.7471396923065186
    },
    {
      "epoch": 3.734588623046875e-05,
      "grad_norm": 0.14604215323925018,
      "learning_rate": 7.135028526461743e-06,
      "loss": 0.0574,
      "step": 39160
    },
    {
      "epoch": 3.734588623046875e-05,
      "model_forward_time": 1.7256293296813965,
      "step": 39160
    },
    {
      "dynamics_loss_avg": 0.04553789123892784,
      "epoch": 3.734588623046875e-05,
      "step": 39160
    },
    {
      "action_loss_avg": 0.0025123361090663822,
      "epoch": 3.734588623046875e-05,
      "step": 39160
    },
    {
      "epoch": 3.734588623046875e-05,
      "step": 39160,
      "training_step_time": 6.709170579910278
    },
    {
      "epoch": 3.734683990478516e-05,
      "model_forward_time": 1.7269079685211182,
      "step": 39161
    },
    {
      "epoch": 3.734683990478516e-05,
      "step": 39161,
      "training_step_time": 6.79575252532959
    },
    {
      "epoch": 3.734779357910156e-05,
      "model_forward_time": 1.729276418685913,
      "step": 39162
    },
    {
      "epoch": 3.734779357910156e-05,
      "step": 39162,
      "training_step_time": 6.735742092132568
    },
    {
      "epoch": 3.7348747253417966e-05,
      "model_forward_time": 1.7238712310791016,
      "step": 39163
    },
    {
      "epoch": 3.7348747253417966e-05,
      "step": 39163,
      "training_step_time": 7.528032541275024
    },
    {
      "epoch": 3.7349700927734376e-05,
      "model_forward_time": 1.737900733947754,
      "step": 39164
    },
    {
      "epoch": 3.7349700927734376e-05,
      "step": 39164,
      "training_step_time": 9.171353578567505
    },
    {
      "epoch": 3.735065460205078e-05,
      "model_forward_time": 2.2560505867004395,
      "step": 39165
    },
    {
      "epoch": 3.735065460205078e-05,
      "step": 39165,
      "training_step_time": 9.882007122039795
    },
    {
      "epoch": 3.7351608276367184e-05,
      "model_forward_time": 1.7376317977905273,
      "step": 39166
    },
    {
      "epoch": 3.7351608276367184e-05,
      "step": 39166,
      "training_step_time": 8.38020944595337
    },
    {
      "epoch": 3.7352561950683595e-05,
      "model_forward_time": 1.712083339691162,
      "step": 39167
    },
    {
      "epoch": 3.7352561950683595e-05,
      "step": 39167,
      "training_step_time": 7.087027311325073
    },
    {
      "epoch": 3.7353515625e-05,
      "model_forward_time": 1.733402967453003,
      "step": 39168
    },
    {
      "epoch": 3.7353515625e-05,
      "step": 39168,
      "training_step_time": 6.8392863273620605
    },
    {
      "epoch": 3.73544692993164e-05,
      "model_forward_time": 1.7215049266815186,
      "step": 39169
    },
    {
      "epoch": 3.73544692993164e-05,
      "step": 39169,
      "training_step_time": 6.780852556228638
    },
    {
      "epoch": 3.7355422973632814e-05,
      "grad_norm": 0.1298663467168808,
      "learning_rate": 7.133533261433595e-06,
      "loss": 0.0577,
      "step": 39170
    },
    {
      "epoch": 3.7355422973632814e-05,
      "model_forward_time": 1.7117738723754883,
      "step": 39170
    },
    {
      "dynamics_loss_avg": 0.05854398440569639,
      "epoch": 3.7355422973632814e-05,
      "step": 39170
    },
    {
      "action_loss_avg": 0.0022251089714700354,
      "epoch": 3.7355422973632814e-05,
      "step": 39170
    },
    {
      "epoch": 3.7355422973632814e-05,
      "step": 39170,
      "training_step_time": 6.7741734981536865
    },
    {
      "epoch": 3.735637664794922e-05,
      "model_forward_time": 1.7157690525054932,
      "step": 39171
    },
    {
      "epoch": 3.735637664794922e-05,
      "step": 39171,
      "training_step_time": 6.760901212692261
    },
    {
      "epoch": 3.735733032226562e-05,
      "model_forward_time": 1.7158334255218506,
      "step": 39172
    },
    {
      "epoch": 3.735733032226562e-05,
      "step": 39172,
      "training_step_time": 7.383714437484741
    },
    {
      "epoch": 3.735828399658203e-05,
      "model_forward_time": 1.7400386333465576,
      "step": 39173
    },
    {
      "epoch": 3.735828399658203e-05,
      "step": 39173,
      "training_step_time": 7.766883611679077
    },
    {
      "epoch": 3.735923767089844e-05,
      "model_forward_time": 2.94856333732605,
      "step": 39174
    },
    {
      "epoch": 3.735923767089844e-05,
      "step": 39174,
      "training_step_time": 10.665198564529419
    },
    {
      "epoch": 3.736019134521484e-05,
      "model_forward_time": 1.8411409854888916,
      "step": 39175
    },
    {
      "epoch": 3.736019134521484e-05,
      "step": 39175,
      "training_step_time": 7.692854881286621
    },
    {
      "epoch": 3.736114501953125e-05,
      "model_forward_time": 1.713944911956787,
      "step": 39176
    },
    {
      "epoch": 3.736114501953125e-05,
      "step": 39176,
      "training_step_time": 7.60047721862793
    },
    {
      "epoch": 3.7362098693847656e-05,
      "model_forward_time": 1.7141978740692139,
      "step": 39177
    },
    {
      "epoch": 3.7362098693847656e-05,
      "step": 39177,
      "training_step_time": 7.163462400436401
    },
    {
      "epoch": 3.736305236816406e-05,
      "model_forward_time": 1.721536636352539,
      "step": 39178
    },
    {
      "epoch": 3.736305236816406e-05,
      "step": 39178,
      "training_step_time": 6.7623395919799805
    },
    {
      "epoch": 3.736400604248047e-05,
      "model_forward_time": 1.7183496952056885,
      "step": 39179
    },
    {
      "epoch": 3.736400604248047e-05,
      "step": 39179,
      "training_step_time": 6.778044700622559
    },
    {
      "epoch": 3.7364959716796875e-05,
      "grad_norm": 0.14642398059368134,
      "learning_rate": 7.1320377630854575e-06,
      "loss": 0.0579,
      "step": 39180
    },
    {
      "epoch": 3.7364959716796875e-05,
      "model_forward_time": 1.706608772277832,
      "step": 39180
    },
    {
      "dynamics_loss_avg": 0.05128812249749899,
      "epoch": 3.7364959716796875e-05,
      "step": 39180
    },
    {
      "action_loss_avg": 0.0023798282200004905,
      "epoch": 3.7364959716796875e-05,
      "step": 39180
    },
    {
      "epoch": 3.7364959716796875e-05,
      "step": 39180,
      "training_step_time": 6.819973468780518
    },
    {
      "epoch": 3.736591339111328e-05,
      "model_forward_time": 1.7090368270874023,
      "step": 39181
    },
    {
      "epoch": 3.736591339111328e-05,
      "step": 39181,
      "training_step_time": 7.4536802768707275
    },
    {
      "epoch": 3.736686706542969e-05,
      "model_forward_time": 1.7379422187805176,
      "step": 39182
    },
    {
      "epoch": 3.736686706542969e-05,
      "step": 39182,
      "training_step_time": 9.020898818969727
    },
    {
      "epoch": 3.7367820739746094e-05,
      "model_forward_time": 2.888251543045044,
      "step": 39183
    },
    {
      "epoch": 3.7367820739746094e-05,
      "step": 39183,
      "training_step_time": 12.963846445083618
    },
    {
      "epoch": 3.73687744140625e-05,
      "model_forward_time": 1.726377010345459,
      "step": 39184
    },
    {
      "epoch": 3.73687744140625e-05,
      "step": 39184,
      "training_step_time": 10.283732414245605
    },
    {
      "epoch": 3.736972808837891e-05,
      "model_forward_time": 1.7119419574737549,
      "step": 39185
    },
    {
      "epoch": 3.736972808837891e-05,
      "step": 39185,
      "training_step_time": 7.59874153137207
    },
    {
      "epoch": 3.737068176269531e-05,
      "model_forward_time": 1.7118728160858154,
      "step": 39186
    },
    {
      "epoch": 3.737068176269531e-05,
      "step": 39186,
      "training_step_time": 7.2468039989471436
    },
    {
      "epoch": 3.7371635437011716e-05,
      "model_forward_time": 1.7135682106018066,
      "step": 39187
    },
    {
      "epoch": 3.7371635437011716e-05,
      "step": 39187,
      "training_step_time": 6.790186643600464
    },
    {
      "epoch": 3.737258911132813e-05,
      "model_forward_time": 1.7384190559387207,
      "step": 39188
    },
    {
      "epoch": 3.737258911132813e-05,
      "step": 39188,
      "training_step_time": 6.789995431900024
    },
    {
      "epoch": 3.737354278564453e-05,
      "model_forward_time": 1.7391536235809326,
      "step": 39189
    },
    {
      "epoch": 3.737354278564453e-05,
      "step": 39189,
      "training_step_time": 6.827565908432007
    },
    {
      "epoch": 3.7374496459960935e-05,
      "grad_norm": 0.5453968644142151,
      "learning_rate": 7.130542031580875e-06,
      "loss": 0.06,
      "step": 39190
    },
    {
      "epoch": 3.7374496459960935e-05,
      "model_forward_time": 1.7307770252227783,
      "step": 39190
    },
    {
      "dynamics_loss_avg": 0.05248787710443139,
      "epoch": 3.7374496459960935e-05,
      "step": 39190
    },
    {
      "action_loss_avg": 0.001841935218544677,
      "epoch": 3.7374496459960935e-05,
      "step": 39190
    },
    {
      "epoch": 3.7374496459960935e-05,
      "step": 39190,
      "training_step_time": 7.430718183517456
    },
    {
      "epoch": 3.7375450134277346e-05,
      "model_forward_time": 1.7156426906585693,
      "step": 39191
    },
    {
      "epoch": 3.7375450134277346e-05,
      "step": 39191,
      "training_step_time": 7.762102365493774
    },
    {
      "epoch": 3.737640380859375e-05,
      "model_forward_time": 2.3201372623443604,
      "step": 39192
    },
    {
      "epoch": 3.737640380859375e-05,
      "step": 39192,
      "training_step_time": 8.593271732330322
    },
    {
      "epoch": 3.7377357482910154e-05,
      "model_forward_time": 1.360741376876831,
      "step": 39193
    },
    {
      "epoch": 3.7377357482910154e-05,
      "step": 39193,
      "training_step_time": 9.87189507484436
    },
    {
      "epoch": 3.7378311157226565e-05,
      "model_forward_time": 1.7372357845306396,
      "step": 39194
    },
    {
      "epoch": 3.7378311157226565e-05,
      "step": 39194,
      "training_step_time": 7.7396275997161865
    },
    {
      "epoch": 3.737926483154297e-05,
      "model_forward_time": 1.727104663848877,
      "step": 39195
    },
    {
      "epoch": 3.737926483154297e-05,
      "step": 39195,
      "training_step_time": 7.4418275356292725
    },
    {
      "epoch": 3.738021850585937e-05,
      "model_forward_time": 1.7157807350158691,
      "step": 39196
    },
    {
      "epoch": 3.738021850585937e-05,
      "step": 39196,
      "training_step_time": 6.834619998931885
    },
    {
      "epoch": 3.7381172180175784e-05,
      "model_forward_time": 1.355555772781372,
      "step": 39197
    },
    {
      "epoch": 3.7381172180175784e-05,
      "step": 39197,
      "training_step_time": 6.764846324920654
    },
    {
      "epoch": 3.738212585449219e-05,
      "model_forward_time": 1.731881856918335,
      "step": 39198
    },
    {
      "epoch": 3.738212585449219e-05,
      "step": 39198,
      "training_step_time": 6.759925127029419
    },
    {
      "epoch": 3.738307952880859e-05,
      "model_forward_time": 1.7189760208129883,
      "step": 39199
    },
    {
      "epoch": 3.738307952880859e-05,
      "step": 39199,
      "training_step_time": 6.776476621627808
    },
    {
      "epoch": 3.7384033203125e-05,
      "grad_norm": 0.1124146580696106,
      "learning_rate": 7.12904606708342e-06,
      "loss": 0.0594,
      "step": 39200
    },
    {
      "epoch": 3.7384033203125e-05,
      "model_forward_time": 1.7224822044372559,
      "step": 39200
    },
    {
      "dynamics_loss_avg": 0.04682512488216162,
      "epoch": 3.7384033203125e-05,
      "step": 39200
    },
    {
      "action_loss_avg": 0.0015908766945358366,
      "epoch": 3.7384033203125e-05,
      "step": 39200
    },
    {
      "epoch": 3.7384033203125e-05,
      "step": 39200,
      "training_step_time": 7.6774866580963135
    },
    {
      "epoch": 3.738498687744141e-05,
      "model_forward_time": 1.7068159580230713,
      "step": 39201
    },
    {
      "epoch": 3.738498687744141e-05,
      "step": 39201,
      "training_step_time": 9.159326553344727
    },
    {
      "epoch": 3.738594055175781e-05,
      "model_forward_time": 1.7402021884918213,
      "step": 39202
    },
    {
      "epoch": 3.738594055175781e-05,
      "step": 39202,
      "training_step_time": 10.182210922241211
    },
    {
      "epoch": 3.738689422607422e-05,
      "model_forward_time": 1.7144103050231934,
      "step": 39203
    },
    {
      "epoch": 3.738689422607422e-05,
      "step": 39203,
      "training_step_time": 7.6485350131988525
    },
    {
      "epoch": 3.7387847900390626e-05,
      "model_forward_time": 1.719381332397461,
      "step": 39204
    },
    {
      "epoch": 3.7387847900390626e-05,
      "step": 39204,
      "training_step_time": 7.395654201507568
    },
    {
      "epoch": 3.738880157470703e-05,
      "model_forward_time": 1.7036373615264893,
      "step": 39205
    },
    {
      "epoch": 3.738880157470703e-05,
      "step": 39205,
      "training_step_time": 6.723123550415039
    },
    {
      "epoch": 3.738975524902344e-05,
      "model_forward_time": 1.7040066719055176,
      "step": 39206
    },
    {
      "epoch": 3.738975524902344e-05,
      "step": 39206,
      "training_step_time": 6.763847351074219
    },
    {
      "epoch": 3.7390708923339844e-05,
      "model_forward_time": 1.714050054550171,
      "step": 39207
    },
    {
      "epoch": 3.7390708923339844e-05,
      "step": 39207,
      "training_step_time": 6.776530981063843
    },
    {
      "epoch": 3.739166259765625e-05,
      "model_forward_time": 1.7213616371154785,
      "step": 39208
    },
    {
      "epoch": 3.739166259765625e-05,
      "step": 39208,
      "training_step_time": 6.760876417160034
    },
    {
      "epoch": 3.739261627197266e-05,
      "model_forward_time": 1.71309494972229,
      "step": 39209
    },
    {
      "epoch": 3.739261627197266e-05,
      "step": 39209,
      "training_step_time": 8.167877435684204
    },
    {
      "epoch": 3.739356994628906e-05,
      "grad_norm": 0.10923824459314346,
      "learning_rate": 7.127549869756687e-06,
      "loss": 0.0584,
      "step": 39210
    },
    {
      "epoch": 3.739356994628906e-05,
      "model_forward_time": 1.7022628784179688,
      "step": 39210
    },
    {
      "dynamics_loss_avg": 0.04495392888784409,
      "epoch": 3.739356994628906e-05,
      "step": 39210
    },
    {
      "action_loss_avg": 0.002063125226413831,
      "epoch": 3.739356994628906e-05,
      "step": 39210
    },
    {
      "epoch": 3.739356994628906e-05,
      "step": 39210,
      "training_step_time": 8.756155252456665
    },
    {
      "epoch": 3.739452362060547e-05,
      "model_forward_time": 2.895195722579956,
      "step": 39211
    },
    {
      "epoch": 3.739452362060547e-05,
      "step": 39211,
      "training_step_time": 9.84397578239441
    },
    {
      "epoch": 3.739547729492188e-05,
      "model_forward_time": 1.729485273361206,
      "step": 39212
    },
    {
      "epoch": 3.739547729492188e-05,
      "step": 39212,
      "training_step_time": 7.7307140827178955
    },
    {
      "epoch": 3.739643096923828e-05,
      "model_forward_time": 1.7172446250915527,
      "step": 39213
    },
    {
      "epoch": 3.739643096923828e-05,
      "step": 39213,
      "training_step_time": 7.632518291473389
    },
    {
      "epoch": 3.7397384643554686e-05,
      "model_forward_time": 1.7087457180023193,
      "step": 39214
    },
    {
      "epoch": 3.7397384643554686e-05,
      "step": 39214,
      "training_step_time": 7.058446407318115
    },
    {
      "epoch": 3.73983383178711e-05,
      "model_forward_time": 1.7048819065093994,
      "step": 39215
    },
    {
      "epoch": 3.73983383178711e-05,
      "step": 39215,
      "training_step_time": 6.801295757293701
    },
    {
      "epoch": 3.73992919921875e-05,
      "model_forward_time": 1.703824520111084,
      "step": 39216
    },
    {
      "epoch": 3.73992919921875e-05,
      "step": 39216,
      "training_step_time": 6.839641094207764
    },
    {
      "epoch": 3.7400245666503905e-05,
      "model_forward_time": 1.3272149562835693,
      "step": 39217
    },
    {
      "epoch": 3.7400245666503905e-05,
      "step": 39217,
      "training_step_time": 6.8299853801727295
    },
    {
      "epoch": 3.7401199340820316e-05,
      "model_forward_time": 1.7190816402435303,
      "step": 39218
    },
    {
      "epoch": 3.7401199340820316e-05,
      "step": 39218,
      "training_step_time": 7.621892213821411
    },
    {
      "epoch": 3.740215301513672e-05,
      "model_forward_time": 1.7355329990386963,
      "step": 39219
    },
    {
      "epoch": 3.740215301513672e-05,
      "step": 39219,
      "training_step_time": 8.742016553878784
    },
    {
      "epoch": 3.7403106689453124e-05,
      "grad_norm": 0.09902863204479218,
      "learning_rate": 7.1260534397643e-06,
      "loss": 0.061,
      "step": 39220
    },
    {
      "epoch": 3.7403106689453124e-05,
      "model_forward_time": 2.8414998054504395,
      "step": 39220
    },
    {
      "dynamics_loss_avg": 0.059451562166213986,
      "epoch": 3.7403106689453124e-05,
      "step": 39220
    },
    {
      "action_loss_avg": 0.0014924647402949631,
      "epoch": 3.7403106689453124e-05,
      "step": 39220
    },
    {
      "epoch": 3.7403106689453124e-05,
      "step": 39220,
      "training_step_time": 8.81994915008545
    },
    {
      "epoch": 3.740406036376953e-05,
      "model_forward_time": 1.7321786880493164,
      "step": 39221
    },
    {
      "epoch": 3.740406036376953e-05,
      "step": 39221,
      "training_step_time": 7.77578067779541
    },
    {
      "epoch": 3.740501403808594e-05,
      "model_forward_time": 1.7035856246948242,
      "step": 39222
    },
    {
      "epoch": 3.740501403808594e-05,
      "step": 39222,
      "training_step_time": 7.561194896697998
    },
    {
      "epoch": 3.740596771240234e-05,
      "model_forward_time": 1.712810754776001,
      "step": 39223
    },
    {
      "epoch": 3.740596771240234e-05,
      "step": 39223,
      "training_step_time": 7.44879674911499
    },
    {
      "epoch": 3.740692138671875e-05,
      "model_forward_time": 1.7003729343414307,
      "step": 39224
    },
    {
      "epoch": 3.740692138671875e-05,
      "step": 39224,
      "training_step_time": 6.7878570556640625
    },
    {
      "epoch": 3.740787506103516e-05,
      "model_forward_time": 1.7140486240386963,
      "step": 39225
    },
    {
      "epoch": 3.740787506103516e-05,
      "step": 39225,
      "training_step_time": 6.784878253936768
    },
    {
      "epoch": 3.740882873535156e-05,
      "model_forward_time": 1.7141361236572266,
      "step": 39226
    },
    {
      "epoch": 3.740882873535156e-05,
      "step": 39226,
      "training_step_time": 6.777572393417358
    },
    {
      "epoch": 3.7409782409667966e-05,
      "model_forward_time": 1.703770637512207,
      "step": 39227
    },
    {
      "epoch": 3.7409782409667966e-05,
      "step": 39227,
      "training_step_time": 7.5902228355407715
    },
    {
      "epoch": 3.7410736083984376e-05,
      "model_forward_time": 1.7348761558532715,
      "step": 39228
    },
    {
      "epoch": 3.7410736083984376e-05,
      "step": 39228,
      "training_step_time": 8.241185426712036
    },
    {
      "epoch": 3.741168975830078e-05,
      "model_forward_time": 2.813420534133911,
      "step": 39229
    },
    {
      "epoch": 3.741168975830078e-05,
      "step": 39229,
      "training_step_time": 8.88791561126709
    },
    {
      "epoch": 3.7412643432617185e-05,
      "grad_norm": 0.19401831924915314,
      "learning_rate": 7.124556777269904e-06,
      "loss": 0.0577,
      "step": 39230
    },
    {
      "epoch": 3.7412643432617185e-05,
      "model_forward_time": 1.725553035736084,
      "step": 39230
    },
    {
      "dynamics_loss_avg": 0.04848538236692548,
      "epoch": 3.7412643432617185e-05,
      "step": 39230
    },
    {
      "action_loss_avg": 0.0021441858960315587,
      "epoch": 3.7412643432617185e-05,
      "step": 39230
    },
    {
      "epoch": 3.7412643432617185e-05,
      "step": 39230,
      "training_step_time": 9.270403623580933
    },
    {
      "epoch": 3.7413597106933595e-05,
      "model_forward_time": 1.7052562236785889,
      "step": 39231
    },
    {
      "epoch": 3.7413597106933595e-05,
      "step": 39231,
      "training_step_time": 7.650107383728027
    },
    {
      "epoch": 3.741455078125e-05,
      "model_forward_time": 1.7166335582733154,
      "step": 39232
    },
    {
      "epoch": 3.741455078125e-05,
      "step": 39232,
      "training_step_time": 7.039492607116699
    },
    {
      "epoch": 3.74155044555664e-05,
      "model_forward_time": 1.7114496231079102,
      "step": 39233
    },
    {
      "epoch": 3.74155044555664e-05,
      "step": 39233,
      "training_step_time": 6.764187574386597
    },
    {
      "epoch": 3.7416458129882814e-05,
      "model_forward_time": 1.703726053237915,
      "step": 39234
    },
    {
      "epoch": 3.7416458129882814e-05,
      "step": 39234,
      "training_step_time": 6.779974460601807
    },
    {
      "epoch": 3.741741180419922e-05,
      "model_forward_time": 1.7039484977722168,
      "step": 39235
    },
    {
      "epoch": 3.741741180419922e-05,
      "step": 39235,
      "training_step_time": 6.748479843139648
    },
    {
      "epoch": 3.741836547851562e-05,
      "model_forward_time": 1.6999096870422363,
      "step": 39236
    },
    {
      "epoch": 3.741836547851562e-05,
      "step": 39236,
      "training_step_time": 7.513471603393555
    },
    {
      "epoch": 3.741931915283203e-05,
      "model_forward_time": 1.7388076782226562,
      "step": 39237
    },
    {
      "epoch": 3.741931915283203e-05,
      "step": 39237,
      "training_step_time": 7.6664719581604
    },
    {
      "epoch": 3.742027282714844e-05,
      "model_forward_time": 2.8420662879943848,
      "step": 39238
    },
    {
      "epoch": 3.742027282714844e-05,
      "step": 39238,
      "training_step_time": 8.382524251937866
    },
    {
      "epoch": 3.742122650146484e-05,
      "model_forward_time": 1.7016901969909668,
      "step": 39239
    },
    {
      "epoch": 3.742122650146484e-05,
      "step": 39239,
      "training_step_time": 8.214274883270264
    },
    {
      "epoch": 3.742218017578125e-05,
      "grad_norm": 0.14808347821235657,
      "learning_rate": 7.1230598824371744e-06,
      "loss": 0.0609,
      "step": 39240
    },
    {
      "epoch": 3.742218017578125e-05,
      "model_forward_time": 1.7121827602386475,
      "step": 39240
    },
    {
      "dynamics_loss_avg": 0.05806125532835722,
      "epoch": 3.742218017578125e-05,
      "step": 39240
    },
    {
      "action_loss_avg": 0.002064850233728066,
      "epoch": 3.742218017578125e-05,
      "step": 39240
    },
    {
      "epoch": 3.742218017578125e-05,
      "step": 39240,
      "training_step_time": 8.169152736663818
    },
    {
      "epoch": 3.7423133850097656e-05,
      "model_forward_time": 1.7142612934112549,
      "step": 39241
    },
    {
      "epoch": 3.7423133850097656e-05,
      "step": 39241,
      "training_step_time": 7.431265592575073
    },
    {
      "epoch": 3.742408752441406e-05,
      "model_forward_time": 1.7169132232666016,
      "step": 39242
    },
    {
      "epoch": 3.742408752441406e-05,
      "step": 39242,
      "training_step_time": 7.315577030181885
    },
    {
      "epoch": 3.742504119873047e-05,
      "model_forward_time": 1.7271690368652344,
      "step": 39243
    },
    {
      "epoch": 3.742504119873047e-05,
      "step": 39243,
      "training_step_time": 6.725671291351318
    },
    {
      "epoch": 3.7425994873046875e-05,
      "model_forward_time": 1.7073915004730225,
      "step": 39244
    },
    {
      "epoch": 3.7425994873046875e-05,
      "step": 39244,
      "training_step_time": 6.682115316390991
    },
    {
      "epoch": 3.742694854736328e-05,
      "model_forward_time": 1.711524486541748,
      "step": 39245
    },
    {
      "epoch": 3.742694854736328e-05,
      "step": 39245,
      "training_step_time": 8.148706912994385
    },
    {
      "epoch": 3.742790222167969e-05,
      "model_forward_time": 1.8158612251281738,
      "step": 39246
    },
    {
      "epoch": 3.742790222167969e-05,
      "step": 39246,
      "training_step_time": 7.660348653793335
    },
    {
      "epoch": 3.7428855895996094e-05,
      "model_forward_time": 1.7266170978546143,
      "step": 39247
    },
    {
      "epoch": 3.7428855895996094e-05,
      "step": 39247,
      "training_step_time": 8.943979978561401
    },
    {
      "epoch": 3.74298095703125e-05,
      "model_forward_time": 1.7189357280731201,
      "step": 39248
    },
    {
      "epoch": 3.74298095703125e-05,
      "step": 39248,
      "training_step_time": 8.49500298500061
    },
    {
      "epoch": 3.743076324462891e-05,
      "model_forward_time": 1.741715431213379,
      "step": 39249
    },
    {
      "epoch": 3.743076324462891e-05,
      "step": 39249,
      "training_step_time": 7.73717474937439
    },
    {
      "epoch": 3.743171691894531e-05,
      "grad_norm": 0.1256851851940155,
      "learning_rate": 7.121562755429808e-06,
      "loss": 0.0608,
      "step": 39250
    },
    {
      "epoch": 3.743171691894531e-05,
      "model_forward_time": 1.7136766910552979,
      "step": 39250
    },
    {
      "dynamics_loss_avg": 0.03893004283308983,
      "epoch": 3.743171691894531e-05,
      "step": 39250
    },
    {
      "action_loss_avg": 0.0016329062404111029,
      "epoch": 3.743171691894531e-05,
      "step": 39250
    },
    {
      "epoch": 3.743171691894531e-05,
      "step": 39250,
      "training_step_time": 7.450726509094238
    },
    {
      "epoch": 3.7432670593261717e-05,
      "model_forward_time": 1.7181639671325684,
      "step": 39251
    },
    {
      "epoch": 3.7432670593261717e-05,
      "step": 39251,
      "training_step_time": 7.556684494018555
    },
    {
      "epoch": 3.743362426757813e-05,
      "model_forward_time": 1.726747751235962,
      "step": 39252
    },
    {
      "epoch": 3.743362426757813e-05,
      "step": 39252,
      "training_step_time": 6.751192808151245
    },
    {
      "epoch": 3.743457794189453e-05,
      "model_forward_time": 1.7220475673675537,
      "step": 39253
    },
    {
      "epoch": 3.743457794189453e-05,
      "step": 39253,
      "training_step_time": 6.768400192260742
    },
    {
      "epoch": 3.7435531616210935e-05,
      "model_forward_time": 1.7192702293395996,
      "step": 39254
    },
    {
      "epoch": 3.7435531616210935e-05,
      "step": 39254,
      "training_step_time": 7.530928611755371
    },
    {
      "epoch": 3.7436485290527346e-05,
      "model_forward_time": 1.7442924976348877,
      "step": 39255
    },
    {
      "epoch": 3.7436485290527346e-05,
      "step": 39255,
      "training_step_time": 7.639042377471924
    },
    {
      "epoch": 3.743743896484375e-05,
      "model_forward_time": 1.7326946258544922,
      "step": 39256
    },
    {
      "epoch": 3.743743896484375e-05,
      "step": 39256,
      "training_step_time": 7.565504789352417
    },
    {
      "epoch": 3.7438392639160154e-05,
      "model_forward_time": 2.857795476913452,
      "step": 39257
    },
    {
      "epoch": 3.7438392639160154e-05,
      "step": 39257,
      "training_step_time": 11.364448308944702
    },
    {
      "epoch": 3.7439346313476565e-05,
      "model_forward_time": 1.7496471405029297,
      "step": 39258
    },
    {
      "epoch": 3.7439346313476565e-05,
      "step": 39258,
      "training_step_time": 8.114591598510742
    },
    {
      "epoch": 3.744029998779297e-05,
      "model_forward_time": 1.7236874103546143,
      "step": 39259
    },
    {
      "epoch": 3.744029998779297e-05,
      "step": 39259,
      "training_step_time": 7.306382656097412
    },
    {
      "epoch": 3.744125366210937e-05,
      "grad_norm": 0.17755620181560516,
      "learning_rate": 7.120065396411528e-06,
      "loss": 0.0607,
      "step": 39260
    },
    {
      "epoch": 3.744125366210937e-05,
      "model_forward_time": 1.7312402725219727,
      "step": 39260
    },
    {
      "dynamics_loss_avg": 0.05260603837668896,
      "epoch": 3.744125366210937e-05,
      "step": 39260
    },
    {
      "action_loss_avg": 0.0015190719510428607,
      "epoch": 3.744125366210937e-05,
      "step": 39260
    },
    {
      "epoch": 3.744125366210937e-05,
      "step": 39260,
      "training_step_time": 7.454484701156616
    },
    {
      "epoch": 3.7442207336425784e-05,
      "model_forward_time": 1.7314162254333496,
      "step": 39261
    },
    {
      "epoch": 3.7442207336425784e-05,
      "step": 39261,
      "training_step_time": 7.34985876083374
    },
    {
      "epoch": 3.744316101074219e-05,
      "model_forward_time": 1.7263059616088867,
      "step": 39262
    },
    {
      "epoch": 3.744316101074219e-05,
      "step": 39262,
      "training_step_time": 6.743067026138306
    },
    {
      "epoch": 3.744411468505859e-05,
      "model_forward_time": 1.7265629768371582,
      "step": 39263
    },
    {
      "epoch": 3.744411468505859e-05,
      "step": 39263,
      "training_step_time": 7.521365165710449
    },
    {
      "epoch": 3.7445068359375e-05,
      "model_forward_time": 1.750854730606079,
      "step": 39264
    },
    {
      "epoch": 3.7445068359375e-05,
      "step": 39264,
      "training_step_time": 7.5985283851623535
    },
    {
      "epoch": 3.744602203369141e-05,
      "model_forward_time": 1.7314658164978027,
      "step": 39265
    },
    {
      "epoch": 3.744602203369141e-05,
      "step": 39265,
      "training_step_time": 8.304308414459229
    },
    {
      "epoch": 3.744697570800781e-05,
      "model_forward_time": 2.912648916244507,
      "step": 39266
    },
    {
      "epoch": 3.744697570800781e-05,
      "step": 39266,
      "training_step_time": 10.815971374511719
    },
    {
      "epoch": 3.744792938232422e-05,
      "model_forward_time": 1.7246897220611572,
      "step": 39267
    },
    {
      "epoch": 3.744792938232422e-05,
      "step": 39267,
      "training_step_time": 8.632092952728271
    },
    {
      "epoch": 3.7448883056640626e-05,
      "model_forward_time": 1.7261784076690674,
      "step": 39268
    },
    {
      "epoch": 3.7448883056640626e-05,
      "step": 39268,
      "training_step_time": 7.605183124542236
    },
    {
      "epoch": 3.744983673095703e-05,
      "model_forward_time": 1.3582556247711182,
      "step": 39269
    },
    {
      "epoch": 3.744983673095703e-05,
      "step": 39269,
      "training_step_time": 7.463428258895874
    },
    {
      "epoch": 3.745079040527344e-05,
      "grad_norm": 0.13928620517253876,
      "learning_rate": 7.118567805546084e-06,
      "loss": 0.0648,
      "step": 39270
    },
    {
      "epoch": 3.745079040527344e-05,
      "model_forward_time": 1.7262938022613525,
      "step": 39270
    },
    {
      "dynamics_loss_avg": 0.06016529593616724,
      "epoch": 3.745079040527344e-05,
      "step": 39270
    },
    {
      "action_loss_avg": 0.0026156754436669873,
      "epoch": 3.745079040527344e-05,
      "step": 39270
    },
    {
      "epoch": 3.745079040527344e-05,
      "step": 39270,
      "training_step_time": 7.367286682128906
    },
    {
      "epoch": 3.7451744079589845e-05,
      "model_forward_time": 1.7245545387268066,
      "step": 39271
    },
    {
      "epoch": 3.7451744079589845e-05,
      "step": 39271,
      "training_step_time": 6.706698179244995
    },
    {
      "epoch": 3.745269775390625e-05,
      "model_forward_time": 1.7216863632202148,
      "step": 39272
    },
    {
      "epoch": 3.745269775390625e-05,
      "step": 39272,
      "training_step_time": 7.498782634735107
    },
    {
      "epoch": 3.745365142822266e-05,
      "model_forward_time": 1.7334833145141602,
      "step": 39273
    },
    {
      "epoch": 3.745365142822266e-05,
      "step": 39273,
      "training_step_time": 7.47619366645813
    },
    {
      "epoch": 3.745460510253906e-05,
      "model_forward_time": 1.3564460277557373,
      "step": 39274
    },
    {
      "epoch": 3.745460510253906e-05,
      "step": 39274,
      "training_step_time": 7.626461744308472
    },
    {
      "epoch": 3.745555877685547e-05,
      "model_forward_time": 2.882490396499634,
      "step": 39275
    },
    {
      "epoch": 3.745555877685547e-05,
      "step": 39275,
      "training_step_time": 10.430326461791992
    },
    {
      "epoch": 3.745651245117188e-05,
      "model_forward_time": 1.7147932052612305,
      "step": 39276
    },
    {
      "epoch": 3.745651245117188e-05,
      "step": 39276,
      "training_step_time": 8.528223037719727
    },
    {
      "epoch": 3.745746612548828e-05,
      "model_forward_time": 1.7544186115264893,
      "step": 39277
    },
    {
      "epoch": 3.745746612548828e-05,
      "step": 39277,
      "training_step_time": 7.595325231552124
    },
    {
      "epoch": 3.7458419799804686e-05,
      "model_forward_time": 1.7258963584899902,
      "step": 39278
    },
    {
      "epoch": 3.7458419799804686e-05,
      "step": 39278,
      "training_step_time": 7.42674446105957
    },
    {
      "epoch": 3.74593734741211e-05,
      "model_forward_time": 1.7272799015045166,
      "step": 39279
    },
    {
      "epoch": 3.74593734741211e-05,
      "step": 39279,
      "training_step_time": 7.354182958602905
    },
    {
      "epoch": 3.74603271484375e-05,
      "grad_norm": 0.15159454941749573,
      "learning_rate": 7.117069982997249e-06,
      "loss": 0.0625,
      "step": 39280
    },
    {
      "epoch": 3.74603271484375e-05,
      "model_forward_time": 1.7249188423156738,
      "step": 39280
    },
    {
      "dynamics_loss_avg": 0.05258681159466505,
      "epoch": 3.74603271484375e-05,
      "step": 39280
    },
    {
      "action_loss_avg": 0.002045887490385212,
      "epoch": 3.74603271484375e-05,
      "step": 39280
    },
    {
      "epoch": 3.74603271484375e-05,
      "step": 39280,
      "training_step_time": 6.7319841384887695
    },
    {
      "epoch": 3.7461280822753905e-05,
      "model_forward_time": 1.7357149124145508,
      "step": 39281
    },
    {
      "epoch": 3.7461280822753905e-05,
      "step": 39281,
      "training_step_time": 6.70971417427063
    },
    {
      "epoch": 3.746223449707031e-05,
      "model_forward_time": 1.73099946975708,
      "step": 39282
    },
    {
      "epoch": 3.746223449707031e-05,
      "step": 39282,
      "training_step_time": 7.470374345779419
    },
    {
      "epoch": 3.746318817138672e-05,
      "model_forward_time": 1.751861810684204,
      "step": 39283
    },
    {
      "epoch": 3.746318817138672e-05,
      "step": 39283,
      "training_step_time": 8.449818849563599
    },
    {
      "epoch": 3.7464141845703124e-05,
      "model_forward_time": 2.900620222091675,
      "step": 39284
    },
    {
      "epoch": 3.7464141845703124e-05,
      "step": 39284,
      "training_step_time": 9.537184000015259
    },
    {
      "epoch": 3.746509552001953e-05,
      "model_forward_time": 1.7220745086669922,
      "step": 39285
    },
    {
      "epoch": 3.746509552001953e-05,
      "step": 39285,
      "training_step_time": 8.686187505722046
    },
    {
      "epoch": 3.746604919433594e-05,
      "model_forward_time": 1.7487640380859375,
      "step": 39286
    },
    {
      "epoch": 3.746604919433594e-05,
      "step": 39286,
      "training_step_time": 7.610669136047363
    },
    {
      "epoch": 3.746700286865234e-05,
      "model_forward_time": 1.7258844375610352,
      "step": 39287
    },
    {
      "epoch": 3.746700286865234e-05,
      "step": 39287,
      "training_step_time": 7.382456541061401
    },
    {
      "epoch": 3.746795654296875e-05,
      "model_forward_time": 1.7239506244659424,
      "step": 39288
    },
    {
      "epoch": 3.746795654296875e-05,
      "step": 39288,
      "training_step_time": 6.706148386001587
    },
    {
      "epoch": 3.746891021728516e-05,
      "model_forward_time": 1.729344367980957,
      "step": 39289
    },
    {
      "epoch": 3.746891021728516e-05,
      "step": 39289,
      "training_step_time": 6.947094202041626
    },
    {
      "epoch": 3.746986389160156e-05,
      "grad_norm": 0.1316743940114975,
      "learning_rate": 7.115571928928825e-06,
      "loss": 0.0565,
      "step": 39290
    },
    {
      "epoch": 3.746986389160156e-05,
      "model_forward_time": 1.7337381839752197,
      "step": 39290
    },
    {
      "dynamics_loss_avg": 0.06760183768346906,
      "epoch": 3.746986389160156e-05,
      "step": 39290
    },
    {
      "action_loss_avg": 0.002669325901661068,
      "epoch": 3.746986389160156e-05,
      "step": 39290
    },
    {
      "epoch": 3.746986389160156e-05,
      "step": 39290,
      "training_step_time": 6.739843368530273
    },
    {
      "epoch": 3.7470817565917966e-05,
      "model_forward_time": 1.7357690334320068,
      "step": 39291
    },
    {
      "epoch": 3.7470817565917966e-05,
      "step": 39291,
      "training_step_time": 7.536834001541138
    },
    {
      "epoch": 3.7471771240234377e-05,
      "model_forward_time": 1.7570464611053467,
      "step": 39292
    },
    {
      "epoch": 3.7471771240234377e-05,
      "step": 39292,
      "training_step_time": 7.663023948669434
    },
    {
      "epoch": 3.747272491455078e-05,
      "model_forward_time": 2.8219518661499023,
      "step": 39293
    },
    {
      "epoch": 3.747272491455078e-05,
      "step": 39293,
      "training_step_time": 9.502315759658813
    },
    {
      "epoch": 3.7473678588867185e-05,
      "model_forward_time": 1.7299845218658447,
      "step": 39294
    },
    {
      "epoch": 3.7473678588867185e-05,
      "step": 39294,
      "training_step_time": 9.757255792617798
    },
    {
      "epoch": 3.7474632263183595e-05,
      "model_forward_time": 1.749070167541504,
      "step": 39295
    },
    {
      "epoch": 3.7474632263183595e-05,
      "step": 39295,
      "training_step_time": 7.696958065032959
    },
    {
      "epoch": 3.74755859375e-05,
      "model_forward_time": 1.7342026233673096,
      "step": 39296
    },
    {
      "epoch": 3.74755859375e-05,
      "step": 39296,
      "training_step_time": 7.481780767440796
    },
    {
      "epoch": 3.7476539611816403e-05,
      "model_forward_time": 1.7292239665985107,
      "step": 39297
    },
    {
      "epoch": 3.7476539611816403e-05,
      "step": 39297,
      "training_step_time": 6.716207981109619
    },
    {
      "epoch": 3.7477493286132814e-05,
      "model_forward_time": 1.715648889541626,
      "step": 39298
    },
    {
      "epoch": 3.7477493286132814e-05,
      "step": 39298,
      "training_step_time": 7.414379119873047
    },
    {
      "epoch": 3.747844696044922e-05,
      "model_forward_time": 1.72821044921875,
      "step": 39299
    },
    {
      "epoch": 3.747844696044922e-05,
      "step": 39299,
      "training_step_time": 6.75374436378479
    },
    {
      "epoch": 3.747940063476562e-05,
      "grad_norm": 0.1184142604470253,
      "learning_rate": 7.1140736435046355e-06,
      "loss": 0.0566,
      "step": 39300
    },
    {
      "epoch": 3.747940063476562e-05,
      "model_forward_time": 1.7275774478912354,
      "step": 39300
    },
    {
      "dynamics_loss_avg": 0.05872674640268087,
      "epoch": 3.747940063476562e-05,
      "step": 39300
    },
    {
      "action_loss_avg": 0.002117627806728706,
      "epoch": 3.747940063476562e-05,
      "step": 39300
    },
    {
      "epoch": 3.747940063476562e-05,
      "step": 39300,
      "training_step_time": 7.635258674621582
    },
    {
      "epoch": 3.748035430908203e-05,
      "model_forward_time": 1.7566180229187012,
      "step": 39301
    },
    {
      "epoch": 3.748035430908203e-05,
      "step": 39301,
      "training_step_time": 7.6251935958862305
    },
    {
      "epoch": 3.748130798339844e-05,
      "model_forward_time": 1.7200226783752441,
      "step": 39302
    },
    {
      "epoch": 3.748130798339844e-05,
      "step": 39302,
      "training_step_time": 9.118891716003418
    },
    {
      "epoch": 3.748226165771484e-05,
      "model_forward_time": 2.9329354763031006,
      "step": 39303
    },
    {
      "epoch": 3.748226165771484e-05,
      "step": 39303,
      "training_step_time": 8.640396118164062
    },
    {
      "epoch": 3.748321533203125e-05,
      "model_forward_time": 1.7309181690216064,
      "step": 39304
    },
    {
      "epoch": 3.748321533203125e-05,
      "step": 39304,
      "training_step_time": 7.724593162536621
    },
    {
      "epoch": 3.7484169006347656e-05,
      "model_forward_time": 1.7487270832061768,
      "step": 39305
    },
    {
      "epoch": 3.7484169006347656e-05,
      "step": 39305,
      "training_step_time": 7.4751105308532715
    },
    {
      "epoch": 3.748512268066406e-05,
      "model_forward_time": 1.7237358093261719,
      "step": 39306
    },
    {
      "epoch": 3.748512268066406e-05,
      "step": 39306,
      "training_step_time": 7.393724203109741
    },
    {
      "epoch": 3.748607635498047e-05,
      "model_forward_time": 1.7237610816955566,
      "step": 39307
    },
    {
      "epoch": 3.748607635498047e-05,
      "step": 39307,
      "training_step_time": 7.39846658706665
    },
    {
      "epoch": 3.7487030029296875e-05,
      "model_forward_time": 1.7317185401916504,
      "step": 39308
    },
    {
      "epoch": 3.7487030029296875e-05,
      "step": 39308,
      "training_step_time": 6.734732389450073
    },
    {
      "epoch": 3.748798370361328e-05,
      "model_forward_time": 1.7266442775726318,
      "step": 39309
    },
    {
      "epoch": 3.748798370361328e-05,
      "step": 39309,
      "training_step_time": 7.605956792831421
    },
    {
      "epoch": 3.748893737792969e-05,
      "grad_norm": 0.15214020013809204,
      "learning_rate": 7.11257512688853e-06,
      "loss": 0.0559,
      "step": 39310
    },
    {
      "epoch": 3.748893737792969e-05,
      "model_forward_time": 1.0182294845581055,
      "step": 39310
    },
    {
      "dynamics_loss_avg": 0.053535284101963045,
      "epoch": 3.748893737792969e-05,
      "step": 39310
    },
    {
      "action_loss_avg": 0.0017266588984057307,
      "epoch": 3.748893737792969e-05,
      "step": 39310
    },
    {
      "epoch": 3.748893737792969e-05,
      "step": 39310,
      "training_step_time": 7.530442953109741
    },
    {
      "epoch": 3.7489891052246094e-05,
      "model_forward_time": 1.696415662765503,
      "step": 39311
    },
    {
      "epoch": 3.7489891052246094e-05,
      "step": 39311,
      "training_step_time": 7.7074596881866455
    },
    {
      "epoch": 3.74908447265625e-05,
      "model_forward_time": 2.8940787315368652,
      "step": 39312
    },
    {
      "epoch": 3.74908447265625e-05,
      "step": 39312,
      "training_step_time": 9.217177152633667
    },
    {
      "epoch": 3.749179840087891e-05,
      "model_forward_time": 1.7200639247894287,
      "step": 39313
    },
    {
      "epoch": 3.749179840087891e-05,
      "step": 39313,
      "training_step_time": 7.6480443477630615
    },
    {
      "epoch": 3.749275207519531e-05,
      "model_forward_time": 1.7416200637817383,
      "step": 39314
    },
    {
      "epoch": 3.749275207519531e-05,
      "step": 39314,
      "training_step_time": 7.749520301818848
    },
    {
      "epoch": 3.7493705749511717e-05,
      "model_forward_time": 1.7216987609863281,
      "step": 39315
    },
    {
      "epoch": 3.7493705749511717e-05,
      "step": 39315,
      "training_step_time": 7.4635655879974365
    },
    {
      "epoch": 3.749465942382813e-05,
      "model_forward_time": 1.7190024852752686,
      "step": 39316
    },
    {
      "epoch": 3.749465942382813e-05,
      "step": 39316,
      "training_step_time": 7.693435192108154
    },
    {
      "epoch": 3.749561309814453e-05,
      "model_forward_time": 1.7162821292877197,
      "step": 39317
    },
    {
      "epoch": 3.749561309814453e-05,
      "step": 39317,
      "training_step_time": 6.876760482788086
    },
    {
      "epoch": 3.7496566772460935e-05,
      "model_forward_time": 1.7193748950958252,
      "step": 39318
    },
    {
      "epoch": 3.7496566772460935e-05,
      "step": 39318,
      "training_step_time": 7.701772928237915
    },
    {
      "epoch": 3.7497520446777346e-05,
      "model_forward_time": 1.7178096771240234,
      "step": 39319
    },
    {
      "epoch": 3.7497520446777346e-05,
      "step": 39319,
      "training_step_time": 8.060298681259155
    },
    {
      "epoch": 3.749847412109375e-05,
      "grad_norm": 0.14370602369308472,
      "learning_rate": 7.111076379244385e-06,
      "loss": 0.0584,
      "step": 39320
    },
    {
      "epoch": 3.749847412109375e-05,
      "model_forward_time": 1.7328732013702393,
      "step": 39320
    },
    {
      "dynamics_loss_avg": 0.05241603367030621,
      "epoch": 3.749847412109375e-05,
      "step": 39320
    },
    {
      "action_loss_avg": 0.0025528220052365214,
      "epoch": 3.749847412109375e-05,
      "step": 39320
    },
    {
      "epoch": 3.749847412109375e-05,
      "step": 39320,
      "training_step_time": 8.623579740524292
    },
    {
      "epoch": 3.7499427795410154e-05,
      "model_forward_time": 2.837043285369873,
      "step": 39321
    },
    {
      "epoch": 3.7499427795410154e-05,
      "step": 39321,
      "training_step_time": 8.84756350517273
    },
    {
      "epoch": 3.7500381469726565e-05,
      "model_forward_time": 1.730445146560669,
      "step": 39322
    },
    {
      "epoch": 3.7500381469726565e-05,
      "step": 39322,
      "training_step_time": 7.818618297576904
    },
    {
      "epoch": 3.750133514404297e-05,
      "model_forward_time": 1.7431731224060059,
      "step": 39323
    },
    {
      "epoch": 3.750133514404297e-05,
      "step": 39323,
      "training_step_time": 8.516030550003052
    },
    {
      "epoch": 3.750228881835937e-05,
      "model_forward_time": 1.7235968112945557,
      "step": 39324
    },
    {
      "epoch": 3.750228881835937e-05,
      "step": 39324,
      "training_step_time": 7.5126564502716064
    },
    {
      "epoch": 3.7503242492675784e-05,
      "model_forward_time": 1.7377395629882812,
      "step": 39325
    },
    {
      "epoch": 3.7503242492675784e-05,
      "step": 39325,
      "training_step_time": 6.786222696304321
    },
    {
      "epoch": 3.750419616699219e-05,
      "model_forward_time": 1.7315750122070312,
      "step": 39326
    },
    {
      "epoch": 3.750419616699219e-05,
      "step": 39326,
      "training_step_time": 7.396714687347412
    },
    {
      "epoch": 3.750514984130859e-05,
      "model_forward_time": 1.7329933643341064,
      "step": 39327
    },
    {
      "epoch": 3.750514984130859e-05,
      "step": 39327,
      "training_step_time": 7.213087320327759
    },
    {
      "epoch": 3.7506103515625e-05,
      "model_forward_time": 1.741119146347046,
      "step": 39328
    },
    {
      "epoch": 3.7506103515625e-05,
      "step": 39328,
      "training_step_time": 7.566641807556152
    },
    {
      "epoch": 3.750705718994141e-05,
      "model_forward_time": 1.7615134716033936,
      "step": 39329
    },
    {
      "epoch": 3.750705718994141e-05,
      "step": 39329,
      "training_step_time": 7.706230401992798
    },
    {
      "epoch": 3.750801086425781e-05,
      "grad_norm": 0.16475480794906616,
      "learning_rate": 7.109577400736101e-06,
      "loss": 0.0594,
      "step": 39330
    },
    {
      "epoch": 3.750801086425781e-05,
      "model_forward_time": 2.8020119667053223,
      "step": 39330
    },
    {
      "dynamics_loss_avg": 0.05898367743939161,
      "epoch": 3.750801086425781e-05,
      "step": 39330
    },
    {
      "action_loss_avg": 0.0026894734939560292,
      "epoch": 3.750801086425781e-05,
      "step": 39330
    },
    {
      "epoch": 3.750801086425781e-05,
      "step": 39330,
      "training_step_time": 9.721473217010498
    },
    {
      "epoch": 3.750896453857422e-05,
      "model_forward_time": 1.7418158054351807,
      "step": 39331
    },
    {
      "epoch": 3.750896453857422e-05,
      "step": 39331,
      "training_step_time": 8.709517478942871
    },
    {
      "epoch": 3.7509918212890626e-05,
      "model_forward_time": 1.7616605758666992,
      "step": 39332
    },
    {
      "epoch": 3.7509918212890626e-05,
      "step": 39332,
      "training_step_time": 9.311558246612549
    },
    {
      "epoch": 3.751087188720703e-05,
      "model_forward_time": 1.7292053699493408,
      "step": 39333
    },
    {
      "epoch": 3.751087188720703e-05,
      "step": 39333,
      "training_step_time": 7.525092124938965
    },
    {
      "epoch": 3.751182556152344e-05,
      "model_forward_time": 1.7297561168670654,
      "step": 39334
    },
    {
      "epoch": 3.751182556152344e-05,
      "step": 39334,
      "training_step_time": 7.071995735168457
    },
    {
      "epoch": 3.7512779235839845e-05,
      "model_forward_time": 1.7277977466583252,
      "step": 39335
    },
    {
      "epoch": 3.7512779235839845e-05,
      "step": 39335,
      "training_step_time": 7.450332403182983
    },
    {
      "epoch": 3.751373291015625e-05,
      "model_forward_time": 1.7257664203643799,
      "step": 39336
    },
    {
      "epoch": 3.751373291015625e-05,
      "step": 39336,
      "training_step_time": 7.184833526611328
    },
    {
      "epoch": 3.751468658447266e-05,
      "model_forward_time": 1.736602544784546,
      "step": 39337
    },
    {
      "epoch": 3.751468658447266e-05,
      "step": 39337,
      "training_step_time": 8.312524557113647
    },
    {
      "epoch": 3.7515640258789063e-05,
      "model_forward_time": 1.7463204860687256,
      "step": 39338
    },
    {
      "epoch": 3.7515640258789063e-05,
      "step": 39338,
      "training_step_time": 8.524650573730469
    },
    {
      "epoch": 3.751659393310547e-05,
      "model_forward_time": 1.7194061279296875,
      "step": 39339
    },
    {
      "epoch": 3.751659393310547e-05,
      "step": 39339,
      "training_step_time": 9.247309923171997
    },
    {
      "epoch": 3.751754760742188e-05,
      "grad_norm": 0.17042863368988037,
      "learning_rate": 7.108078191527606e-06,
      "loss": 0.0601,
      "step": 39340
    },
    {
      "epoch": 3.751754760742188e-05,
      "model_forward_time": 2.895418405532837,
      "step": 39340
    },
    {
      "dynamics_loss_avg": 0.0655718456953764,
      "epoch": 3.751754760742188e-05,
      "step": 39340
    },
    {
      "action_loss_avg": 0.002286815905245021,
      "epoch": 3.751754760742188e-05,
      "step": 39340
    },
    {
      "epoch": 3.751754760742188e-05,
      "step": 39340,
      "training_step_time": 8.422054529190063
    },
    {
      "epoch": 3.751850128173828e-05,
      "model_forward_time": 1.3474340438842773,
      "step": 39341
    },
    {
      "epoch": 3.751850128173828e-05,
      "step": 39341,
      "training_step_time": 8.658483743667603
    },
    {
      "epoch": 3.7519454956054686e-05,
      "model_forward_time": 1.7624125480651855,
      "step": 39342
    },
    {
      "epoch": 3.7519454956054686e-05,
      "step": 39342,
      "training_step_time": 7.48508882522583
    },
    {
      "epoch": 3.75204086303711e-05,
      "model_forward_time": 1.7375679016113281,
      "step": 39343
    },
    {
      "epoch": 3.75204086303711e-05,
      "step": 39343,
      "training_step_time": 7.038246393203735
    },
    {
      "epoch": 3.75213623046875e-05,
      "model_forward_time": 1.7392187118530273,
      "step": 39344
    },
    {
      "epoch": 3.75213623046875e-05,
      "step": 39344,
      "training_step_time": 7.3027184009552
    },
    {
      "epoch": 3.7522315979003905e-05,
      "model_forward_time": 1.737886905670166,
      "step": 39345
    },
    {
      "epoch": 3.7522315979003905e-05,
      "step": 39345,
      "training_step_time": 7.788615703582764
    },
    {
      "epoch": 3.752326965332031e-05,
      "model_forward_time": 1.7280116081237793,
      "step": 39346
    },
    {
      "epoch": 3.752326965332031e-05,
      "step": 39346,
      "training_step_time": 7.536524534225464
    },
    {
      "epoch": 3.752422332763672e-05,
      "model_forward_time": 1.7568714618682861,
      "step": 39347
    },
    {
      "epoch": 3.752422332763672e-05,
      "step": 39347,
      "training_step_time": 7.5369346141815186
    },
    {
      "epoch": 3.7525177001953124e-05,
      "model_forward_time": 1.73087477684021,
      "step": 39348
    },
    {
      "epoch": 3.7525177001953124e-05,
      "step": 39348,
      "training_step_time": 7.666515588760376
    },
    {
      "epoch": 3.752613067626953e-05,
      "model_forward_time": 2.866959571838379,
      "step": 39349
    },
    {
      "epoch": 3.752613067626953e-05,
      "step": 39349,
      "training_step_time": 7.659449100494385
    },
    {
      "epoch": 3.752708435058594e-05,
      "grad_norm": 0.15600930154323578,
      "learning_rate": 7.106578751782847e-06,
      "loss": 0.0546,
      "step": 39350
    },
    {
      "epoch": 3.752708435058594e-05,
      "model_forward_time": 1.7299535274505615,
      "step": 39350
    },
    {
      "dynamics_loss_avg": 0.05048751011490822,
      "epoch": 3.752708435058594e-05,
      "step": 39350
    },
    {
      "action_loss_avg": 0.00222418496850878,
      "epoch": 3.752708435058594e-05,
      "step": 39350
    },
    {
      "epoch": 3.752708435058594e-05,
      "step": 39350,
      "training_step_time": 9.08347773551941
    },
    {
      "epoch": 3.752803802490234e-05,
      "model_forward_time": 1.7618067264556885,
      "step": 39351
    },
    {
      "epoch": 3.752803802490234e-05,
      "step": 39351,
      "training_step_time": 7.474184036254883
    },
    {
      "epoch": 3.752899169921875e-05,
      "model_forward_time": 1.725597858428955,
      "step": 39352
    },
    {
      "epoch": 3.752899169921875e-05,
      "step": 39352,
      "training_step_time": 7.440257787704468
    },
    {
      "epoch": 3.752994537353516e-05,
      "model_forward_time": 1.724660873413086,
      "step": 39353
    },
    {
      "epoch": 3.752994537353516e-05,
      "step": 39353,
      "training_step_time": 7.331863641738892
    },
    {
      "epoch": 3.753089904785156e-05,
      "model_forward_time": 1.7308070659637451,
      "step": 39354
    },
    {
      "epoch": 3.753089904785156e-05,
      "step": 39354,
      "training_step_time": 6.697186708450317
    },
    {
      "epoch": 3.7531852722167966e-05,
      "model_forward_time": 1.7312495708465576,
      "step": 39355
    },
    {
      "epoch": 3.7531852722167966e-05,
      "step": 39355,
      "training_step_time": 8.365847110748291
    },
    {
      "epoch": 3.7532806396484377e-05,
      "model_forward_time": 1.0393946170806885,
      "step": 39356
    },
    {
      "epoch": 3.7532806396484377e-05,
      "step": 39356,
      "training_step_time": 7.669956684112549
    },
    {
      "epoch": 3.753376007080078e-05,
      "model_forward_time": 1.7236411571502686,
      "step": 39357
    },
    {
      "epoch": 3.753376007080078e-05,
      "step": 39357,
      "training_step_time": 9.535167217254639
    },
    {
      "epoch": 3.7534713745117185e-05,
      "model_forward_time": 2.7772727012634277,
      "step": 39358
    },
    {
      "epoch": 3.7534713745117185e-05,
      "step": 39358,
      "training_step_time": 8.455787420272827
    },
    {
      "epoch": 3.7535667419433595e-05,
      "model_forward_time": 1.7355575561523438,
      "step": 39359
    },
    {
      "epoch": 3.7535667419433595e-05,
      "step": 39359,
      "training_step_time": 7.666775226593018
    },
    {
      "epoch": 3.753662109375e-05,
      "grad_norm": 0.131126269698143,
      "learning_rate": 7.105079081665803e-06,
      "loss": 0.0626,
      "step": 39360
    },
    {
      "epoch": 3.753662109375e-05,
      "model_forward_time": 1.7501580715179443,
      "step": 39360
    },
    {
      "dynamics_loss_avg": 0.057234148308634755,
      "epoch": 3.753662109375e-05,
      "step": 39360
    },
    {
      "action_loss_avg": 0.00182658436242491,
      "epoch": 3.753662109375e-05,
      "step": 39360
    },
    {
      "epoch": 3.753662109375e-05,
      "step": 39360,
      "training_step_time": 7.707563877105713
    },
    {
      "epoch": 3.7537574768066403e-05,
      "model_forward_time": 1.727477788925171,
      "step": 39361
    },
    {
      "epoch": 3.7537574768066403e-05,
      "step": 39361,
      "training_step_time": 7.54031777381897
    },
    {
      "epoch": 3.7538528442382814e-05,
      "model_forward_time": 1.7500028610229492,
      "step": 39362
    },
    {
      "epoch": 3.7538528442382814e-05,
      "step": 39362,
      "training_step_time": 7.133129835128784
    },
    {
      "epoch": 3.753948211669922e-05,
      "model_forward_time": 1.7457001209259033,
      "step": 39363
    },
    {
      "epoch": 3.753948211669922e-05,
      "step": 39363,
      "training_step_time": 7.4341020584106445
    },
    {
      "epoch": 3.754043579101562e-05,
      "model_forward_time": 1.740687608718872,
      "step": 39364
    },
    {
      "epoch": 3.754043579101562e-05,
      "step": 39364,
      "training_step_time": 7.62126898765564
    },
    {
      "epoch": 3.754138946533203e-05,
      "model_forward_time": 1.731621503829956,
      "step": 39365
    },
    {
      "epoch": 3.754138946533203e-05,
      "step": 39365,
      "training_step_time": 7.761228799819946
    },
    {
      "epoch": 3.754234313964844e-05,
      "model_forward_time": 1.8243837356567383,
      "step": 39366
    },
    {
      "epoch": 3.754234313964844e-05,
      "step": 39366,
      "training_step_time": 8.551512956619263
    },
    {
      "epoch": 3.754329681396484e-05,
      "model_forward_time": 1.7402534484863281,
      "step": 39367
    },
    {
      "epoch": 3.754329681396484e-05,
      "step": 39367,
      "training_step_time": 8.511357307434082
    },
    {
      "epoch": 3.754425048828125e-05,
      "model_forward_time": 2.308145046234131,
      "step": 39368
    },
    {
      "epoch": 3.754425048828125e-05,
      "step": 39368,
      "training_step_time": 7.5601067543029785
    },
    {
      "epoch": 3.7545204162597656e-05,
      "model_forward_time": 1.7409157752990723,
      "step": 39369
    },
    {
      "epoch": 3.7545204162597656e-05,
      "step": 39369,
      "training_step_time": 7.643619060516357
    },
    {
      "epoch": 3.754615783691406e-05,
      "grad_norm": 0.1411808580160141,
      "learning_rate": 7.1035791813404766e-06,
      "loss": 0.0635,
      "step": 39370
    },
    {
      "epoch": 3.754615783691406e-05,
      "model_forward_time": 1.7503185272216797,
      "step": 39370
    },
    {
      "dynamics_loss_avg": 0.05078149326145649,
      "epoch": 3.754615783691406e-05,
      "step": 39370
    },
    {
      "action_loss_avg": 0.0013993157481309026,
      "epoch": 3.754615783691406e-05,
      "step": 39370
    },
    {
      "epoch": 3.754615783691406e-05,
      "step": 39370,
      "training_step_time": 7.380268096923828
    },
    {
      "epoch": 3.754711151123047e-05,
      "model_forward_time": 1.74741792678833,
      "step": 39371
    },
    {
      "epoch": 3.754711151123047e-05,
      "step": 39371,
      "training_step_time": 7.124955892562866
    },
    {
      "epoch": 3.7548065185546875e-05,
      "model_forward_time": 1.7497620582580566,
      "step": 39372
    },
    {
      "epoch": 3.7548065185546875e-05,
      "step": 39372,
      "training_step_time": 7.391442775726318
    },
    {
      "epoch": 3.754901885986328e-05,
      "model_forward_time": 1.7295947074890137,
      "step": 39373
    },
    {
      "epoch": 3.754901885986328e-05,
      "step": 39373,
      "training_step_time": 7.5991737842559814
    },
    {
      "epoch": 3.754997253417969e-05,
      "model_forward_time": 1.7189218997955322,
      "step": 39374
    },
    {
      "epoch": 3.754997253417969e-05,
      "step": 39374,
      "training_step_time": 7.579866886138916
    },
    {
      "epoch": 3.7550926208496094e-05,
      "model_forward_time": 1.8131909370422363,
      "step": 39375
    },
    {
      "epoch": 3.7550926208496094e-05,
      "step": 39375,
      "training_step_time": 8.261664628982544
    },
    {
      "epoch": 3.75518798828125e-05,
      "model_forward_time": 1.7299931049346924,
      "step": 39376
    },
    {
      "epoch": 3.75518798828125e-05,
      "step": 39376,
      "training_step_time": 7.8232645988464355
    },
    {
      "epoch": 3.755283355712891e-05,
      "model_forward_time": 2.282273530960083,
      "step": 39377
    },
    {
      "epoch": 3.755283355712891e-05,
      "step": 39377,
      "training_step_time": 7.708505630493164
    },
    {
      "epoch": 3.755378723144531e-05,
      "model_forward_time": 1.7359259128570557,
      "step": 39378
    },
    {
      "epoch": 3.755378723144531e-05,
      "step": 39378,
      "training_step_time": 7.668646812438965
    },
    {
      "epoch": 3.755474090576172e-05,
      "model_forward_time": 1.7282323837280273,
      "step": 39379
    },
    {
      "epoch": 3.755474090576172e-05,
      "step": 39379,
      "training_step_time": 7.573451995849609
    },
    {
      "epoch": 3.755569458007813e-05,
      "grad_norm": 0.13761547207832336,
      "learning_rate": 7.102079050970894e-06,
      "loss": 0.0555,
      "step": 39380
    },
    {
      "epoch": 3.755569458007813e-05,
      "model_forward_time": 1.7284448146820068,
      "step": 39380
    },
    {
      "dynamics_loss_avg": 0.058667780831456184,
      "epoch": 3.755569458007813e-05,
      "step": 39380
    },
    {
      "action_loss_avg": 0.002081442647613585,
      "epoch": 3.755569458007813e-05,
      "step": 39380
    },
    {
      "epoch": 3.755569458007813e-05,
      "step": 39380,
      "training_step_time": 6.9866862297058105
    },
    {
      "epoch": 3.755664825439453e-05,
      "model_forward_time": 1.7303924560546875,
      "step": 39381
    },
    {
      "epoch": 3.755664825439453e-05,
      "step": 39381,
      "training_step_time": 6.799143075942993
    },
    {
      "epoch": 3.7557601928710935e-05,
      "model_forward_time": 1.7390766143798828,
      "step": 39382
    },
    {
      "epoch": 3.7557601928710935e-05,
      "step": 39382,
      "training_step_time": 7.522600412368774
    },
    {
      "epoch": 3.7558555603027346e-05,
      "model_forward_time": 1.72896409034729,
      "step": 39383
    },
    {
      "epoch": 3.7558555603027346e-05,
      "step": 39383,
      "training_step_time": 7.566102504730225
    },
    {
      "epoch": 3.755950927734375e-05,
      "model_forward_time": 1.8123998641967773,
      "step": 39384
    },
    {
      "epoch": 3.755950927734375e-05,
      "step": 39384,
      "training_step_time": 8.349517107009888
    },
    {
      "epoch": 3.7560462951660154e-05,
      "model_forward_time": 1.7350423336029053,
      "step": 39385
    },
    {
      "epoch": 3.7560462951660154e-05,
      "step": 39385,
      "training_step_time": 8.82814359664917
    },
    {
      "epoch": 3.7561416625976565e-05,
      "model_forward_time": 2.2843332290649414,
      "step": 39386
    },
    {
      "epoch": 3.7561416625976565e-05,
      "step": 39386,
      "training_step_time": 8.563679933547974
    },
    {
      "epoch": 3.756237030029297e-05,
      "model_forward_time": 1.7386741638183594,
      "step": 39387
    },
    {
      "epoch": 3.756237030029297e-05,
      "step": 39387,
      "training_step_time": 7.61183500289917
    },
    {
      "epoch": 3.756332397460937e-05,
      "model_forward_time": 1.7308149337768555,
      "step": 39388
    },
    {
      "epoch": 3.756332397460937e-05,
      "step": 39388,
      "training_step_time": 7.6503753662109375
    },
    {
      "epoch": 3.7564277648925784e-05,
      "model_forward_time": 1.7345209121704102,
      "step": 39389
    },
    {
      "epoch": 3.7564277648925784e-05,
      "step": 39389,
      "training_step_time": 7.496694564819336
    },
    {
      "epoch": 3.756523132324219e-05,
      "grad_norm": 0.16100101172924042,
      "learning_rate": 7.100578690721105e-06,
      "loss": 0.0531,
      "step": 39390
    },
    {
      "epoch": 3.756523132324219e-05,
      "model_forward_time": 1.7237331867218018,
      "step": 39390
    },
    {
      "dynamics_loss_avg": 0.05379117839038372,
      "epoch": 3.756523132324219e-05,
      "step": 39390
    },
    {
      "action_loss_avg": 0.002306771487928927,
      "epoch": 3.756523132324219e-05,
      "step": 39390
    },
    {
      "epoch": 3.756523132324219e-05,
      "step": 39390,
      "training_step_time": 7.069241046905518
    },
    {
      "epoch": 3.756618499755859e-05,
      "model_forward_time": 1.7342286109924316,
      "step": 39391
    },
    {
      "epoch": 3.756618499755859e-05,
      "step": 39391,
      "training_step_time": 7.379892826080322
    },
    {
      "epoch": 3.7567138671875e-05,
      "model_forward_time": 1.7315342426300049,
      "step": 39392
    },
    {
      "epoch": 3.7567138671875e-05,
      "step": 39392,
      "training_step_time": 8.633564710617065
    },
    {
      "epoch": 3.756809234619141e-05,
      "model_forward_time": 1.8123466968536377,
      "step": 39393
    },
    {
      "epoch": 3.756809234619141e-05,
      "step": 39393,
      "training_step_time": 7.694925546646118
    },
    {
      "epoch": 3.756904602050781e-05,
      "model_forward_time": 1.7314379215240479,
      "step": 39394
    },
    {
      "epoch": 3.756904602050781e-05,
      "step": 39394,
      "training_step_time": 8.551299571990967
    },
    {
      "epoch": 3.756999969482422e-05,
      "model_forward_time": 2.3023200035095215,
      "step": 39395
    },
    {
      "epoch": 3.756999969482422e-05,
      "step": 39395,
      "training_step_time": 9.210917949676514
    },
    {
      "epoch": 3.7570953369140626e-05,
      "model_forward_time": 1.7262380123138428,
      "step": 39396
    },
    {
      "epoch": 3.7570953369140626e-05,
      "step": 39396,
      "training_step_time": 7.641409635543823
    },
    {
      "epoch": 3.757190704345703e-05,
      "model_forward_time": 1.7363545894622803,
      "step": 39397
    },
    {
      "epoch": 3.757190704345703e-05,
      "step": 39397,
      "training_step_time": 8.424801588058472
    },
    {
      "epoch": 3.757286071777344e-05,
      "model_forward_time": 1.3585412502288818,
      "step": 39398
    },
    {
      "epoch": 3.757286071777344e-05,
      "step": 39398,
      "training_step_time": 7.462855577468872
    },
    {
      "epoch": 3.7573814392089845e-05,
      "model_forward_time": 1.7300877571105957,
      "step": 39399
    },
    {
      "epoch": 3.7573814392089845e-05,
      "step": 39399,
      "training_step_time": 7.114028692245483
    },
    {
      "epoch": 3.757476806640625e-05,
      "grad_norm": 0.18456247448921204,
      "learning_rate": 7.09907810075519e-06,
      "loss": 0.0536,
      "step": 39400
    },
    {
      "epoch": 3.757476806640625e-05,
      "model_forward_time": 1.7295894622802734,
      "step": 39400
    },
    {
      "dynamics_loss_avg": 0.04951240755617618,
      "epoch": 3.757476806640625e-05,
      "step": 39400
    },
    {
      "action_loss_avg": 0.0024327205843292176,
      "epoch": 3.757476806640625e-05,
      "step": 39400
    },
    {
      "epoch": 3.757476806640625e-05,
      "step": 39400,
      "training_step_time": 7.348618745803833
    },
    {
      "epoch": 3.757572174072266e-05,
      "model_forward_time": 1.7745280265808105,
      "step": 39401
    },
    {
      "epoch": 3.757572174072266e-05,
      "step": 39401,
      "training_step_time": 7.580138921737671
    },
    {
      "epoch": 3.7576675415039063e-05,
      "model_forward_time": 1.7787995338439941,
      "step": 39402
    },
    {
      "epoch": 3.7576675415039063e-05,
      "step": 39402,
      "training_step_time": 7.6673572063446045
    },
    {
      "epoch": 3.757762908935547e-05,
      "model_forward_time": 1.7377698421478271,
      "step": 39403
    },
    {
      "epoch": 3.757762908935547e-05,
      "step": 39403,
      "training_step_time": 7.782090663909912
    },
    {
      "epoch": 3.757858276367188e-05,
      "model_forward_time": 1.748203992843628,
      "step": 39404
    },
    {
      "epoch": 3.757858276367188e-05,
      "step": 39404,
      "training_step_time": 9.054312229156494
    },
    {
      "epoch": 3.757953643798828e-05,
      "model_forward_time": 1.7373743057250977,
      "step": 39405
    },
    {
      "epoch": 3.757953643798828e-05,
      "step": 39405,
      "training_step_time": 8.182382106781006
    },
    {
      "epoch": 3.7580490112304686e-05,
      "model_forward_time": 1.7392678260803223,
      "step": 39406
    },
    {
      "epoch": 3.7580490112304686e-05,
      "step": 39406,
      "training_step_time": 7.674870491027832
    },
    {
      "epoch": 3.758144378662109e-05,
      "model_forward_time": 1.7364153861999512,
      "step": 39407
    },
    {
      "epoch": 3.758144378662109e-05,
      "step": 39407,
      "training_step_time": 7.4333672523498535
    },
    {
      "epoch": 3.75823974609375e-05,
      "model_forward_time": 1.738112211227417,
      "step": 39408
    },
    {
      "epoch": 3.75823974609375e-05,
      "step": 39408,
      "training_step_time": 6.775212526321411
    },
    {
      "epoch": 3.7583351135253905e-05,
      "model_forward_time": 1.7479796409606934,
      "step": 39409
    },
    {
      "epoch": 3.7583351135253905e-05,
      "step": 39409,
      "training_step_time": 7.337639093399048
    },
    {
      "epoch": 3.758430480957031e-05,
      "grad_norm": 0.18042902648448944,
      "learning_rate": 7.097577281237249e-06,
      "loss": 0.0619,
      "step": 39410
    },
    {
      "epoch": 3.758430480957031e-05,
      "model_forward_time": 1.7312674522399902,
      "step": 39410
    },
    {
      "dynamics_loss_avg": 0.07043023407459259,
      "epoch": 3.758430480957031e-05,
      "step": 39410
    },
    {
      "action_loss_avg": 0.0018849285756004975,
      "epoch": 3.758430480957031e-05,
      "step": 39410
    },
    {
      "epoch": 3.758430480957031e-05,
      "step": 39410,
      "training_step_time": 7.595404624938965
    },
    {
      "epoch": 3.758525848388672e-05,
      "model_forward_time": 1.7560720443725586,
      "step": 39411
    },
    {
      "epoch": 3.758525848388672e-05,
      "step": 39411,
      "training_step_time": 7.608061075210571
    },
    {
      "epoch": 3.7586212158203124e-05,
      "model_forward_time": 1.7337546348571777,
      "step": 39412
    },
    {
      "epoch": 3.7586212158203124e-05,
      "step": 39412,
      "training_step_time": 8.697513580322266
    },
    {
      "epoch": 3.758716583251953e-05,
      "model_forward_time": 1.7357409000396729,
      "step": 39413
    },
    {
      "epoch": 3.758716583251953e-05,
      "step": 39413,
      "training_step_time": 8.993791341781616
    },
    {
      "epoch": 3.758811950683594e-05,
      "model_forward_time": 2.279568910598755,
      "step": 39414
    },
    {
      "epoch": 3.758811950683594e-05,
      "step": 39414,
      "training_step_time": 7.608079671859741
    },
    {
      "epoch": 3.758907318115234e-05,
      "model_forward_time": 1.741753101348877,
      "step": 39415
    },
    {
      "epoch": 3.758907318115234e-05,
      "step": 39415,
      "training_step_time": 9.311272621154785
    },
    {
      "epoch": 3.759002685546875e-05,
      "model_forward_time": 1.7284150123596191,
      "step": 39416
    },
    {
      "epoch": 3.759002685546875e-05,
      "step": 39416,
      "training_step_time": 7.4583423137664795
    },
    {
      "epoch": 3.759098052978516e-05,
      "model_forward_time": 1.7301278114318848,
      "step": 39417
    },
    {
      "epoch": 3.759098052978516e-05,
      "step": 39417,
      "training_step_time": 7.371840715408325
    },
    {
      "epoch": 3.759193420410156e-05,
      "model_forward_time": 1.734095573425293,
      "step": 39418
    },
    {
      "epoch": 3.759193420410156e-05,
      "step": 39418,
      "training_step_time": 7.299838066101074
    },
    {
      "epoch": 3.7592887878417966e-05,
      "model_forward_time": 1.7266936302185059,
      "step": 39419
    },
    {
      "epoch": 3.7592887878417966e-05,
      "step": 39419,
      "training_step_time": 7.6250319480896
    },
    {
      "epoch": 3.759384155273438e-05,
      "grad_norm": 0.16458328068256378,
      "learning_rate": 7.096076232331412e-06,
      "loss": 0.0597,
      "step": 39420
    },
    {
      "epoch": 3.759384155273438e-05,
      "model_forward_time": 1.7334587574005127,
      "step": 39420
    },
    {
      "dynamics_loss_avg": 0.035167915653437375,
      "epoch": 3.759384155273438e-05,
      "step": 39420
    },
    {
      "action_loss_avg": 0.0008948267379309982,
      "epoch": 3.759384155273438e-05,
      "step": 39420
    },
    {
      "epoch": 3.759384155273438e-05,
      "step": 39420,
      "training_step_time": 8.248904705047607
    },
    {
      "epoch": 3.759479522705078e-05,
      "model_forward_time": 1.7409801483154297,
      "step": 39421
    },
    {
      "epoch": 3.759479522705078e-05,
      "step": 39421,
      "training_step_time": 7.823978900909424
    },
    {
      "epoch": 3.7595748901367185e-05,
      "model_forward_time": 1.7318456172943115,
      "step": 39422
    },
    {
      "epoch": 3.7595748901367185e-05,
      "step": 39422,
      "training_step_time": 7.629224538803101
    },
    {
      "epoch": 3.7596702575683595e-05,
      "model_forward_time": 2.2798688411712646,
      "step": 39423
    },
    {
      "epoch": 3.7596702575683595e-05,
      "step": 39423,
      "training_step_time": 8.129857540130615
    },
    {
      "epoch": 3.759765625e-05,
      "model_forward_time": 1.7309930324554443,
      "step": 39424
    },
    {
      "epoch": 3.759765625e-05,
      "step": 39424,
      "training_step_time": 7.6532323360443115
    },
    {
      "epoch": 3.7598609924316404e-05,
      "model_forward_time": 1.7272570133209229,
      "step": 39425
    },
    {
      "epoch": 3.7598609924316404e-05,
      "step": 39425,
      "training_step_time": 7.571823596954346
    },
    {
      "epoch": 3.7599563598632814e-05,
      "model_forward_time": 1.7324917316436768,
      "step": 39426
    },
    {
      "epoch": 3.7599563598632814e-05,
      "step": 39426,
      "training_step_time": 7.422394275665283
    },
    {
      "epoch": 3.760051727294922e-05,
      "model_forward_time": 1.735480546951294,
      "step": 39427
    },
    {
      "epoch": 3.760051727294922e-05,
      "step": 39427,
      "training_step_time": 7.338904619216919
    },
    {
      "epoch": 3.760147094726562e-05,
      "model_forward_time": 1.733208417892456,
      "step": 39428
    },
    {
      "epoch": 3.760147094726562e-05,
      "step": 39428,
      "training_step_time": 7.403340101242065
    },
    {
      "epoch": 3.760242462158203e-05,
      "model_forward_time": 1.725008487701416,
      "step": 39429
    },
    {
      "epoch": 3.760242462158203e-05,
      "step": 39429,
      "training_step_time": 7.67763352394104
    },
    {
      "epoch": 3.760337829589844e-05,
      "grad_norm": 0.16613495349884033,
      "learning_rate": 7.094574954201829e-06,
      "loss": 0.0596,
      "step": 39430
    },
    {
      "epoch": 3.760337829589844e-05,
      "model_forward_time": 1.792123794555664,
      "step": 39430
    },
    {
      "dynamics_loss_avg": 0.06808103639632464,
      "epoch": 3.760337829589844e-05,
      "step": 39430
    },
    {
      "action_loss_avg": 0.003032732830615714,
      "epoch": 3.760337829589844e-05,
      "step": 39430
    },
    {
      "epoch": 3.760337829589844e-05,
      "step": 39430,
      "training_step_time": 8.747327089309692
    },
    {
      "epoch": 3.760433197021484e-05,
      "model_forward_time": 1.7370312213897705,
      "step": 39431
    },
    {
      "epoch": 3.760433197021484e-05,
      "step": 39431,
      "training_step_time": 8.241157293319702
    },
    {
      "epoch": 3.760528564453125e-05,
      "model_forward_time": 2.2768707275390625,
      "step": 39432
    },
    {
      "epoch": 3.760528564453125e-05,
      "step": 39432,
      "training_step_time": 9.761935710906982
    },
    {
      "epoch": 3.7606239318847656e-05,
      "model_forward_time": 1.7331693172454834,
      "step": 39433
    },
    {
      "epoch": 3.7606239318847656e-05,
      "step": 39433,
      "training_step_time": 8.743602275848389
    },
    {
      "epoch": 3.760719299316406e-05,
      "model_forward_time": 1.7283916473388672,
      "step": 39434
    },
    {
      "epoch": 3.760719299316406e-05,
      "step": 39434,
      "training_step_time": 7.734493017196655
    },
    {
      "epoch": 3.760814666748047e-05,
      "model_forward_time": 1.735856533050537,
      "step": 39435
    },
    {
      "epoch": 3.760814666748047e-05,
      "step": 39435,
      "training_step_time": 7.530454874038696
    },
    {
      "epoch": 3.7609100341796875e-05,
      "model_forward_time": 1.7258145809173584,
      "step": 39436
    },
    {
      "epoch": 3.7609100341796875e-05,
      "step": 39436,
      "training_step_time": 7.414437532424927
    },
    {
      "epoch": 3.761005401611328e-05,
      "model_forward_time": 1.736098289489746,
      "step": 39437
    },
    {
      "epoch": 3.761005401611328e-05,
      "step": 39437,
      "training_step_time": 7.443648099899292
    },
    {
      "epoch": 3.761100769042969e-05,
      "model_forward_time": 1.736445426940918,
      "step": 39438
    },
    {
      "epoch": 3.761100769042969e-05,
      "step": 39438,
      "training_step_time": 7.798787355422974
    },
    {
      "epoch": 3.7611961364746094e-05,
      "model_forward_time": 1.783499002456665,
      "step": 39439
    },
    {
      "epoch": 3.7611961364746094e-05,
      "step": 39439,
      "training_step_time": 8.482344388961792
    },
    {
      "epoch": 3.76129150390625e-05,
      "grad_norm": 0.1288089007139206,
      "learning_rate": 7.093073447012676e-06,
      "loss": 0.0608,
      "step": 39440
    },
    {
      "epoch": 3.76129150390625e-05,
      "model_forward_time": 1.7307682037353516,
      "step": 39440
    },
    {
      "dynamics_loss_avg": 0.05901108887046576,
      "epoch": 3.76129150390625e-05,
      "step": 39440
    },
    {
      "action_loss_avg": 0.0015969788306392728,
      "epoch": 3.76129150390625e-05,
      "step": 39440
    },
    {
      "epoch": 3.76129150390625e-05,
      "step": 39440,
      "training_step_time": 9.518873691558838
    },
    {
      "epoch": 3.761386871337891e-05,
      "model_forward_time": 2.295466423034668,
      "step": 39441
    },
    {
      "epoch": 3.761386871337891e-05,
      "step": 39441,
      "training_step_time": 8.708939790725708
    },
    {
      "epoch": 3.761482238769531e-05,
      "model_forward_time": 1.7371294498443604,
      "step": 39442
    },
    {
      "epoch": 3.761482238769531e-05,
      "step": 39442,
      "training_step_time": 7.7478485107421875
    },
    {
      "epoch": 3.761577606201172e-05,
      "model_forward_time": 1.7342183589935303,
      "step": 39443
    },
    {
      "epoch": 3.761577606201172e-05,
      "step": 39443,
      "training_step_time": 7.727273225784302
    },
    {
      "epoch": 3.761672973632813e-05,
      "model_forward_time": 1.7351715564727783,
      "step": 39444
    },
    {
      "epoch": 3.761672973632813e-05,
      "step": 39444,
      "training_step_time": 7.523444175720215
    },
    {
      "epoch": 3.761768341064453e-05,
      "model_forward_time": 1.7362637519836426,
      "step": 39445
    },
    {
      "epoch": 3.761768341064453e-05,
      "step": 39445,
      "training_step_time": 7.223066568374634
    },
    {
      "epoch": 3.7618637084960936e-05,
      "model_forward_time": 1.7319045066833496,
      "step": 39446
    },
    {
      "epoch": 3.7618637084960936e-05,
      "step": 39446,
      "training_step_time": 7.237658500671387
    },
    {
      "epoch": 3.7619590759277346e-05,
      "model_forward_time": 1.7267394065856934,
      "step": 39447
    },
    {
      "epoch": 3.7619590759277346e-05,
      "step": 39447,
      "training_step_time": 8.766335010528564
    },
    {
      "epoch": 3.762054443359375e-05,
      "model_forward_time": 1.7603766918182373,
      "step": 39448
    },
    {
      "epoch": 3.762054443359375e-05,
      "step": 39448,
      "training_step_time": 7.8708717823028564
    },
    {
      "epoch": 3.7621498107910154e-05,
      "model_forward_time": 1.7329518795013428,
      "step": 39449
    },
    {
      "epoch": 3.7621498107910154e-05,
      "step": 39449,
      "training_step_time": 8.762269020080566
    },
    {
      "epoch": 3.7622451782226565e-05,
      "grad_norm": 0.19684724509716034,
      "learning_rate": 7.09157171092816e-06,
      "loss": 0.0583,
      "step": 39450
    },
    {
      "epoch": 3.7622451782226565e-05,
      "model_forward_time": 2.298705816268921,
      "step": 39450
    },
    {
      "dynamics_loss_avg": 0.04816636238247156,
      "epoch": 3.7622451782226565e-05,
      "step": 39450
    },
    {
      "action_loss_avg": 0.0019021057058125734,
      "epoch": 3.7622451782226565e-05,
      "step": 39450
    },
    {
      "epoch": 3.7622451782226565e-05,
      "step": 39450,
      "training_step_time": 8.551183223724365
    },
    {
      "epoch": 3.762340545654297e-05,
      "model_forward_time": 1.7349166870117188,
      "step": 39451
    },
    {
      "epoch": 3.762340545654297e-05,
      "step": 39451,
      "training_step_time": 7.512766122817993
    },
    {
      "epoch": 3.762435913085937e-05,
      "model_forward_time": 1.7285385131835938,
      "step": 39452
    },
    {
      "epoch": 3.762435913085937e-05,
      "step": 39452,
      "training_step_time": 8.659987926483154
    },
    {
      "epoch": 3.7625312805175784e-05,
      "model_forward_time": 1.7338640689849854,
      "step": 39453
    },
    {
      "epoch": 3.7625312805175784e-05,
      "step": 39453,
      "training_step_time": 7.361887693405151
    },
    {
      "epoch": 3.762626647949219e-05,
      "model_forward_time": 1.7353100776672363,
      "step": 39454
    },
    {
      "epoch": 3.762626647949219e-05,
      "step": 39454,
      "training_step_time": 7.398513555526733
    },
    {
      "epoch": 3.762722015380859e-05,
      "model_forward_time": 1.7275371551513672,
      "step": 39455
    },
    {
      "epoch": 3.762722015380859e-05,
      "step": 39455,
      "training_step_time": 7.276520013809204
    },
    {
      "epoch": 3.7628173828125e-05,
      "model_forward_time": 1.730381727218628,
      "step": 39456
    },
    {
      "epoch": 3.7628173828125e-05,
      "step": 39456,
      "training_step_time": 7.692082166671753
    },
    {
      "epoch": 3.762912750244141e-05,
      "model_forward_time": 1.7458181381225586,
      "step": 39457
    },
    {
      "epoch": 3.762912750244141e-05,
      "step": 39457,
      "training_step_time": 7.586880207061768
    },
    {
      "epoch": 3.763008117675781e-05,
      "model_forward_time": 1.7273774147033691,
      "step": 39458
    },
    {
      "epoch": 3.763008117675781e-05,
      "step": 39458,
      "training_step_time": 8.836743593215942
    },
    {
      "epoch": 3.763103485107422e-05,
      "model_forward_time": 1.7341556549072266,
      "step": 39459
    },
    {
      "epoch": 3.763103485107422e-05,
      "step": 39459,
      "training_step_time": 9.006701231002808
    },
    {
      "epoch": 3.7631988525390626e-05,
      "grad_norm": 0.12692123651504517,
      "learning_rate": 7.090069746112505e-06,
      "loss": 0.0595,
      "step": 39460
    },
    {
      "epoch": 3.7631988525390626e-05,
      "model_forward_time": 1.7283177375793457,
      "step": 39460
    },
    {
      "dynamics_loss_avg": 0.05479609966278076,
      "epoch": 3.7631988525390626e-05,
      "step": 39460
    },
    {
      "action_loss_avg": 0.0016660199558828025,
      "epoch": 3.7631988525390626e-05,
      "step": 39460
    },
    {
      "epoch": 3.7631988525390626e-05,
      "step": 39460,
      "training_step_time": 7.709113359451294
    },
    {
      "epoch": 3.763294219970703e-05,
      "model_forward_time": 1.7311866283416748,
      "step": 39461
    },
    {
      "epoch": 3.763294219970703e-05,
      "step": 39461,
      "training_step_time": 7.722973346710205
    },
    {
      "epoch": 3.763389587402344e-05,
      "model_forward_time": 1.7359113693237305,
      "step": 39462
    },
    {
      "epoch": 3.763389587402344e-05,
      "step": 39462,
      "training_step_time": 8.340811252593994
    },
    {
      "epoch": 3.7634849548339845e-05,
      "model_forward_time": 1.7302970886230469,
      "step": 39463
    },
    {
      "epoch": 3.7634849548339845e-05,
      "step": 39463,
      "training_step_time": 7.5069053173065186
    },
    {
      "epoch": 3.763580322265625e-05,
      "model_forward_time": 1.3562214374542236,
      "step": 39464
    },
    {
      "epoch": 3.763580322265625e-05,
      "step": 39464,
      "training_step_time": 7.365312099456787
    },
    {
      "epoch": 3.763675689697266e-05,
      "model_forward_time": 1.73563551902771,
      "step": 39465
    },
    {
      "epoch": 3.763675689697266e-05,
      "step": 39465,
      "training_step_time": 7.744031667709351
    },
    {
      "epoch": 3.7637710571289064e-05,
      "model_forward_time": 1.735314130783081,
      "step": 39466
    },
    {
      "epoch": 3.7637710571289064e-05,
      "step": 39466,
      "training_step_time": 7.6486663818359375
    },
    {
      "epoch": 3.763866424560547e-05,
      "model_forward_time": 1.8089141845703125,
      "step": 39467
    },
    {
      "epoch": 3.763866424560547e-05,
      "step": 39467,
      "training_step_time": 8.902290105819702
    },
    {
      "epoch": 3.763961791992188e-05,
      "model_forward_time": 1.7344908714294434,
      "step": 39468
    },
    {
      "epoch": 3.763961791992188e-05,
      "step": 39468,
      "training_step_time": 8.556097269058228
    },
    {
      "epoch": 3.764057159423828e-05,
      "model_forward_time": 2.289224147796631,
      "step": 39469
    },
    {
      "epoch": 3.764057159423828e-05,
      "step": 39469,
      "training_step_time": 7.674000978469849
    },
    {
      "epoch": 3.7641525268554686e-05,
      "grad_norm": 0.13887928426265717,
      "learning_rate": 7.088567552729966e-06,
      "loss": 0.059,
      "step": 39470
    },
    {
      "epoch": 3.7641525268554686e-05,
      "model_forward_time": 1.728041172027588,
      "step": 39470
    },
    {
      "dynamics_loss_avg": 0.04746045339852571,
      "epoch": 3.7641525268554686e-05,
      "step": 39470
    },
    {
      "action_loss_avg": 0.001515142514836043,
      "epoch": 3.7641525268554686e-05,
      "step": 39470
    },
    {
      "epoch": 3.7641525268554686e-05,
      "step": 39470,
      "training_step_time": 7.718398809432983
    },
    {
      "epoch": 3.764247894287109e-05,
      "model_forward_time": 1.7317478656768799,
      "step": 39471
    },
    {
      "epoch": 3.764247894287109e-05,
      "step": 39471,
      "training_step_time": 7.694047927856445
    },
    {
      "epoch": 3.76434326171875e-05,
      "model_forward_time": 1.7300348281860352,
      "step": 39472
    },
    {
      "epoch": 3.76434326171875e-05,
      "step": 39472,
      "training_step_time": 7.537559509277344
    },
    {
      "epoch": 3.7644386291503905e-05,
      "model_forward_time": 1.7530536651611328,
      "step": 39473
    },
    {
      "epoch": 3.7644386291503905e-05,
      "step": 39473,
      "training_step_time": 7.348031520843506
    },
    {
      "epoch": 3.764533996582031e-05,
      "model_forward_time": 1.7497613430023193,
      "step": 39474
    },
    {
      "epoch": 3.764533996582031e-05,
      "step": 39474,
      "training_step_time": 7.544164419174194
    },
    {
      "epoch": 3.764629364013672e-05,
      "model_forward_time": 1.7445995807647705,
      "step": 39475
    },
    {
      "epoch": 3.764629364013672e-05,
      "step": 39475,
      "training_step_time": 7.7606422901153564
    },
    {
      "epoch": 3.7647247314453124e-05,
      "model_forward_time": 1.8359620571136475,
      "step": 39476
    },
    {
      "epoch": 3.7647247314453124e-05,
      "step": 39476,
      "training_step_time": 8.8075532913208
    },
    {
      "epoch": 3.764820098876953e-05,
      "model_forward_time": 1.7200286388397217,
      "step": 39477
    },
    {
      "epoch": 3.764820098876953e-05,
      "step": 39477,
      "training_step_time": 8.514756917953491
    },
    {
      "epoch": 3.764915466308594e-05,
      "model_forward_time": 2.285121440887451,
      "step": 39478
    },
    {
      "epoch": 3.764915466308594e-05,
      "step": 39478,
      "training_step_time": 9.23151421546936
    },
    {
      "epoch": 3.765010833740234e-05,
      "model_forward_time": 1.7232460975646973,
      "step": 39479
    },
    {
      "epoch": 3.765010833740234e-05,
      "step": 39479,
      "training_step_time": 7.619852781295776
    },
    {
      "epoch": 3.765106201171875e-05,
      "grad_norm": 0.11398029327392578,
      "learning_rate": 7.087065130944818e-06,
      "loss": 0.0575,
      "step": 39480
    },
    {
      "epoch": 3.765106201171875e-05,
      "model_forward_time": 1.7243611812591553,
      "step": 39480
    },
    {
      "dynamics_loss_avg": 0.037157684378325936,
      "epoch": 3.765106201171875e-05,
      "step": 39480
    },
    {
      "action_loss_avg": 0.0015196355059742927,
      "epoch": 3.765106201171875e-05,
      "step": 39480
    },
    {
      "epoch": 3.765106201171875e-05,
      "step": 39480,
      "training_step_time": 7.708122730255127
    },
    {
      "epoch": 3.765201568603516e-05,
      "model_forward_time": 1.7302930355072021,
      "step": 39481
    },
    {
      "epoch": 3.765201568603516e-05,
      "step": 39481,
      "training_step_time": 7.503462076187134
    },
    {
      "epoch": 3.765296936035156e-05,
      "model_forward_time": 1.736182451248169,
      "step": 39482
    },
    {
      "epoch": 3.765296936035156e-05,
      "step": 39482,
      "training_step_time": 7.154799461364746
    },
    {
      "epoch": 3.7653923034667966e-05,
      "model_forward_time": 1.727970838546753,
      "step": 39483
    },
    {
      "epoch": 3.7653923034667966e-05,
      "step": 39483,
      "training_step_time": 7.450449705123901
    },
    {
      "epoch": 3.765487670898438e-05,
      "model_forward_time": 1.7079799175262451,
      "step": 39484
    },
    {
      "epoch": 3.765487670898438e-05,
      "step": 39484,
      "training_step_time": 7.747296333312988
    },
    {
      "epoch": 3.765583038330078e-05,
      "model_forward_time": 1.7266738414764404,
      "step": 39485
    },
    {
      "epoch": 3.765583038330078e-05,
      "step": 39485,
      "training_step_time": 7.760299444198608
    },
    {
      "epoch": 3.7656784057617185e-05,
      "model_forward_time": 1.7113006114959717,
      "step": 39486
    },
    {
      "epoch": 3.7656784057617185e-05,
      "step": 39486,
      "training_step_time": 9.92256784439087
    },
    {
      "epoch": 3.7657737731933596e-05,
      "model_forward_time": 2.295450210571289,
      "step": 39487
    },
    {
      "epoch": 3.7657737731933596e-05,
      "step": 39487,
      "training_step_time": 7.702661037445068
    },
    {
      "epoch": 3.765869140625e-05,
      "model_forward_time": 1.7102675437927246,
      "step": 39488
    },
    {
      "epoch": 3.765869140625e-05,
      "step": 39488,
      "training_step_time": 8.149707794189453
    },
    {
      "epoch": 3.7659645080566404e-05,
      "model_forward_time": 1.7151288986206055,
      "step": 39489
    },
    {
      "epoch": 3.7659645080566404e-05,
      "step": 39489,
      "training_step_time": 7.738470554351807
    },
    {
      "epoch": 3.7660598754882814e-05,
      "grad_norm": 0.10623041540384293,
      "learning_rate": 7.085562480921367e-06,
      "loss": 0.06,
      "step": 39490
    },
    {
      "epoch": 3.7660598754882814e-05,
      "model_forward_time": 1.7274584770202637,
      "step": 39490
    },
    {
      "dynamics_loss_avg": 0.06416828110814095,
      "epoch": 3.7660598754882814e-05,
      "step": 39490
    },
    {
      "action_loss_avg": 0.0015384305821498855,
      "epoch": 3.7660598754882814e-05,
      "step": 39490
    },
    {
      "epoch": 3.7660598754882814e-05,
      "step": 39490,
      "training_step_time": 7.593591213226318
    },
    {
      "epoch": 3.766155242919922e-05,
      "model_forward_time": 1.7097845077514648,
      "step": 39491
    },
    {
      "epoch": 3.766155242919922e-05,
      "step": 39491,
      "training_step_time": 7.1429972648620605
    },
    {
      "epoch": 3.766250610351562e-05,
      "model_forward_time": 1.7131032943725586,
      "step": 39492
    },
    {
      "epoch": 3.766250610351562e-05,
      "step": 39492,
      "training_step_time": 7.439474582672119
    },
    {
      "epoch": 3.766345977783203e-05,
      "model_forward_time": 1.7094225883483887,
      "step": 39493
    },
    {
      "epoch": 3.766345977783203e-05,
      "step": 39493,
      "training_step_time": 7.583691120147705
    },
    {
      "epoch": 3.766441345214844e-05,
      "model_forward_time": 1.7329201698303223,
      "step": 39494
    },
    {
      "epoch": 3.766441345214844e-05,
      "step": 39494,
      "training_step_time": 7.654063940048218
    },
    {
      "epoch": 3.766536712646484e-05,
      "model_forward_time": 1.7183918952941895,
      "step": 39495
    },
    {
      "epoch": 3.766536712646484e-05,
      "step": 39495,
      "training_step_time": 8.920297384262085
    },
    {
      "epoch": 3.766632080078125e-05,
      "model_forward_time": 2.253457546234131,
      "step": 39496
    },
    {
      "epoch": 3.766632080078125e-05,
      "step": 39496,
      "training_step_time": 8.737104415893555
    },
    {
      "epoch": 3.7667274475097656e-05,
      "model_forward_time": 1.7146399021148682,
      "step": 39497
    },
    {
      "epoch": 3.7667274475097656e-05,
      "step": 39497,
      "training_step_time": 7.598124742507935
    },
    {
      "epoch": 3.766822814941406e-05,
      "model_forward_time": 1.7164509296417236,
      "step": 39498
    },
    {
      "epoch": 3.766822814941406e-05,
      "step": 39498,
      "training_step_time": 9.535539865493774
    },
    {
      "epoch": 3.766918182373047e-05,
      "model_forward_time": 1.73506498336792,
      "step": 39499
    },
    {
      "epoch": 3.766918182373047e-05,
      "step": 39499,
      "training_step_time": 8.37979507446289
    },
    {
      "epoch": 3.7670135498046875e-05,
      "grad_norm": 0.1256454735994339,
      "learning_rate": 7.084059602823937e-06,
      "loss": 0.0566,
      "step": 39500
    },
    {
      "epoch": 3.7670135498046875e-05,
      "model_forward_time": 1.720423698425293,
      "step": 39500
    },
    {
      "dynamics_loss_avg": 0.050749943032860755,
      "epoch": 3.7670135498046875e-05,
      "step": 39500
    },
    {
      "action_loss_avg": 0.002042080007959157,
      "epoch": 3.7670135498046875e-05,
      "step": 39500
    },
    {
      "epoch": 3.7670135498046875e-05,
      "step": 39500,
      "training_step_time": 7.4251978397369385
    },
    {
      "epoch": 3.767108917236328e-05,
      "model_forward_time": 1.7108540534973145,
      "step": 39501
    },
    {
      "epoch": 3.767108917236328e-05,
      "step": 39501,
      "training_step_time": 8.225772142410278
    },
    {
      "epoch": 3.767204284667969e-05,
      "model_forward_time": 1.7040183544158936,
      "step": 39502
    },
    {
      "epoch": 3.767204284667969e-05,
      "step": 39502,
      "training_step_time": 7.664215326309204
    },
    {
      "epoch": 3.7672996520996094e-05,
      "model_forward_time": 1.7273871898651123,
      "step": 39503
    },
    {
      "epoch": 3.7672996520996094e-05,
      "step": 39503,
      "training_step_time": 7.619556665420532
    },
    {
      "epoch": 3.76739501953125e-05,
      "model_forward_time": 1.7143440246582031,
      "step": 39504
    },
    {
      "epoch": 3.76739501953125e-05,
      "step": 39504,
      "training_step_time": 7.853178262710571
    },
    {
      "epoch": 3.767490386962891e-05,
      "model_forward_time": 1.706385850906372,
      "step": 39505
    },
    {
      "epoch": 3.767490386962891e-05,
      "step": 39505,
      "training_step_time": 9.689208984375
    },
    {
      "epoch": 3.767585754394531e-05,
      "model_forward_time": 2.255854368209839,
      "step": 39506
    },
    {
      "epoch": 3.767585754394531e-05,
      "step": 39506,
      "training_step_time": 7.5395753383636475
    },
    {
      "epoch": 3.767681121826172e-05,
      "model_forward_time": 1.7122893333435059,
      "step": 39507
    },
    {
      "epoch": 3.767681121826172e-05,
      "step": 39507,
      "training_step_time": 7.6919105052948
    },
    {
      "epoch": 3.767776489257813e-05,
      "model_forward_time": 1.7230782508850098,
      "step": 39508
    },
    {
      "epoch": 3.767776489257813e-05,
      "step": 39508,
      "training_step_time": 7.683334112167358
    },
    {
      "epoch": 3.767871856689453e-05,
      "model_forward_time": 1.7109298706054688,
      "step": 39509
    },
    {
      "epoch": 3.767871856689453e-05,
      "step": 39509,
      "training_step_time": 7.492567777633667
    },
    {
      "epoch": 3.7679672241210936e-05,
      "grad_norm": 0.12687081098556519,
      "learning_rate": 7.082556496816882e-06,
      "loss": 0.0587,
      "step": 39510
    },
    {
      "epoch": 3.7679672241210936e-05,
      "model_forward_time": 1.710017442703247,
      "step": 39510
    },
    {
      "dynamics_loss_avg": 0.0650283433496952,
      "epoch": 3.7679672241210936e-05,
      "step": 39510
    },
    {
      "action_loss_avg": 0.0023808965328498744,
      "epoch": 3.7679672241210936e-05,
      "step": 39510
    },
    {
      "epoch": 3.7679672241210936e-05,
      "step": 39510,
      "training_step_time": 7.468868017196655
    },
    {
      "epoch": 3.7680625915527346e-05,
      "model_forward_time": 1.7096045017242432,
      "step": 39511
    },
    {
      "epoch": 3.7680625915527346e-05,
      "step": 39511,
      "training_step_time": 7.75007963180542
    },
    {
      "epoch": 3.768157958984375e-05,
      "model_forward_time": 1.7529799938201904,
      "step": 39512
    },
    {
      "epoch": 3.768157958984375e-05,
      "step": 39512,
      "training_step_time": 7.672250270843506
    },
    {
      "epoch": 3.7682533264160154e-05,
      "model_forward_time": 1.709078073501587,
      "step": 39513
    },
    {
      "epoch": 3.7682533264160154e-05,
      "step": 39513,
      "training_step_time": 7.890530824661255
    },
    {
      "epoch": 3.7683486938476565e-05,
      "model_forward_time": 1.7097463607788086,
      "step": 39514
    },
    {
      "epoch": 3.7683486938476565e-05,
      "step": 39514,
      "training_step_time": 8.266668558120728
    },
    {
      "epoch": 3.768444061279297e-05,
      "model_forward_time": 2.271064519882202,
      "step": 39515
    },
    {
      "epoch": 3.768444061279297e-05,
      "step": 39515,
      "training_step_time": 11.236690282821655
    },
    {
      "epoch": 3.768539428710937e-05,
      "model_forward_time": 1.712632656097412,
      "step": 39516
    },
    {
      "epoch": 3.768539428710937e-05,
      "step": 39516,
      "training_step_time": 8.180786609649658
    },
    {
      "epoch": 3.7686347961425784e-05,
      "model_forward_time": 1.713407039642334,
      "step": 39517
    },
    {
      "epoch": 3.7686347961425784e-05,
      "step": 39517,
      "training_step_time": 7.7278151512146
    },
    {
      "epoch": 3.768730163574219e-05,
      "model_forward_time": 1.732494592666626,
      "step": 39518
    },
    {
      "epoch": 3.768730163574219e-05,
      "step": 39518,
      "training_step_time": 7.4580488204956055
    },
    {
      "epoch": 3.768825531005859e-05,
      "model_forward_time": 1.71396803855896,
      "step": 39519
    },
    {
      "epoch": 3.768825531005859e-05,
      "step": 39519,
      "training_step_time": 8.299505710601807
    },
    {
      "epoch": 3.7689208984375e-05,
      "grad_norm": 0.1063641682267189,
      "learning_rate": 7.0810531630645815e-06,
      "loss": 0.0591,
      "step": 39520
    },
    {
      "epoch": 3.7689208984375e-05,
      "model_forward_time": 1.70526123046875,
      "step": 39520
    },
    {
      "dynamics_loss_avg": 0.05160161778330803,
      "epoch": 3.7689208984375e-05,
      "step": 39520
    },
    {
      "action_loss_avg": 0.0018042082199826836,
      "epoch": 3.7689208984375e-05,
      "step": 39520
    },
    {
      "epoch": 3.7689208984375e-05,
      "step": 39520,
      "training_step_time": 7.373934507369995
    },
    {
      "epoch": 3.769016265869141e-05,
      "model_forward_time": 1.7193593978881836,
      "step": 39521
    },
    {
      "epoch": 3.769016265869141e-05,
      "step": 39521,
      "training_step_time": 7.778067588806152
    },
    {
      "epoch": 3.769111633300781e-05,
      "model_forward_time": 1.7161526679992676,
      "step": 39522
    },
    {
      "epoch": 3.769111633300781e-05,
      "step": 39522,
      "training_step_time": 8.693202257156372
    },
    {
      "epoch": 3.769207000732422e-05,
      "model_forward_time": 1.7123363018035889,
      "step": 39523
    },
    {
      "epoch": 3.769207000732422e-05,
      "step": 39523,
      "training_step_time": 7.738492727279663
    },
    {
      "epoch": 3.7693023681640626e-05,
      "model_forward_time": 2.2560434341430664,
      "step": 39524
    },
    {
      "epoch": 3.7693023681640626e-05,
      "step": 39524,
      "training_step_time": 9.578054904937744
    },
    {
      "epoch": 3.769397735595703e-05,
      "model_forward_time": 1.715275764465332,
      "step": 39525
    },
    {
      "epoch": 3.769397735595703e-05,
      "step": 39525,
      "training_step_time": 7.652873516082764
    },
    {
      "epoch": 3.769493103027344e-05,
      "model_forward_time": 1.7085180282592773,
      "step": 39526
    },
    {
      "epoch": 3.769493103027344e-05,
      "step": 39526,
      "training_step_time": 7.756548166275024
    },
    {
      "epoch": 3.7695884704589845e-05,
      "model_forward_time": 1.733095407485962,
      "step": 39527
    },
    {
      "epoch": 3.7695884704589845e-05,
      "step": 39527,
      "training_step_time": 7.43317985534668
    },
    {
      "epoch": 3.769683837890625e-05,
      "model_forward_time": 1.709826946258545,
      "step": 39528
    },
    {
      "epoch": 3.769683837890625e-05,
      "step": 39528,
      "training_step_time": 7.472666263580322
    },
    {
      "epoch": 3.769779205322266e-05,
      "model_forward_time": 1.716888666152954,
      "step": 39529
    },
    {
      "epoch": 3.769779205322266e-05,
      "step": 39529,
      "training_step_time": 7.404801368713379
    },
    {
      "epoch": 3.7698745727539064e-05,
      "grad_norm": 0.13216370344161987,
      "learning_rate": 7.0795496017314345e-06,
      "loss": 0.0591,
      "step": 39530
    },
    {
      "epoch": 3.7698745727539064e-05,
      "model_forward_time": 1.7228569984436035,
      "step": 39530
    },
    {
      "dynamics_loss_avg": 0.05370068289339543,
      "epoch": 3.7698745727539064e-05,
      "step": 39530
    },
    {
      "action_loss_avg": 0.0022800158942118285,
      "epoch": 3.7698745727539064e-05,
      "step": 39530
    },
    {
      "epoch": 3.7698745727539064e-05,
      "step": 39530,
      "training_step_time": 7.83448052406311
    },
    {
      "epoch": 3.769969940185547e-05,
      "model_forward_time": 1.7073352336883545,
      "step": 39531
    },
    {
      "epoch": 3.769969940185547e-05,
      "step": 39531,
      "training_step_time": 8.909424304962158
    },
    {
      "epoch": 3.770065307617187e-05,
      "model_forward_time": 1.7053875923156738,
      "step": 39532
    },
    {
      "epoch": 3.770065307617187e-05,
      "step": 39532,
      "training_step_time": 8.844402074813843
    },
    {
      "epoch": 3.770160675048828e-05,
      "model_forward_time": 2.253221273422241,
      "step": 39533
    },
    {
      "epoch": 3.770160675048828e-05,
      "step": 39533,
      "training_step_time": 7.765239953994751
    },
    {
      "epoch": 3.7702560424804686e-05,
      "model_forward_time": 1.333042860031128,
      "step": 39534
    },
    {
      "epoch": 3.7702560424804686e-05,
      "step": 39534,
      "training_step_time": 8.801938772201538
    },
    {
      "epoch": 3.770351409912109e-05,
      "model_forward_time": 1.7174291610717773,
      "step": 39535
    },
    {
      "epoch": 3.770351409912109e-05,
      "step": 39535,
      "training_step_time": 7.752553939819336
    },
    {
      "epoch": 3.77044677734375e-05,
      "model_forward_time": 1.718822717666626,
      "step": 39536
    },
    {
      "epoch": 3.77044677734375e-05,
      "step": 39536,
      "training_step_time": 7.618621349334717
    },
    {
      "epoch": 3.7705421447753905e-05,
      "model_forward_time": 1.7075893878936768,
      "step": 39537
    },
    {
      "epoch": 3.7705421447753905e-05,
      "step": 39537,
      "training_step_time": 7.457888126373291
    },
    {
      "epoch": 3.770637512207031e-05,
      "model_forward_time": 1.7120754718780518,
      "step": 39538
    },
    {
      "epoch": 3.770637512207031e-05,
      "step": 39538,
      "training_step_time": 7.362582206726074
    },
    {
      "epoch": 3.770732879638672e-05,
      "model_forward_time": 1.7269370555877686,
      "step": 39539
    },
    {
      "epoch": 3.770732879638672e-05,
      "step": 39539,
      "training_step_time": 7.733775615692139
    },
    {
      "epoch": 3.7708282470703124e-05,
      "grad_norm": 0.1417747437953949,
      "learning_rate": 7.078045812981871e-06,
      "loss": 0.062,
      "step": 39540
    },
    {
      "epoch": 3.7708282470703124e-05,
      "model_forward_time": 1.7020151615142822,
      "step": 39540
    },
    {
      "dynamics_loss_avg": 0.04812448043376207,
      "epoch": 3.7708282470703124e-05,
      "step": 39540
    },
    {
      "action_loss_avg": 0.0016417098930105567,
      "epoch": 3.7708282470703124e-05,
      "step": 39540
    },
    {
      "epoch": 3.7708282470703124e-05,
      "step": 39540,
      "training_step_time": 7.334062576293945
    },
    {
      "epoch": 3.770923614501953e-05,
      "model_forward_time": 1.3310918807983398,
      "step": 39541
    },
    {
      "epoch": 3.770923614501953e-05,
      "step": 39541,
      "training_step_time": 7.759618520736694
    },
    {
      "epoch": 3.771018981933594e-05,
      "model_forward_time": 1.717198133468628,
      "step": 39542
    },
    {
      "epoch": 3.771018981933594e-05,
      "step": 39542,
      "training_step_time": 8.20116138458252
    },
    {
      "epoch": 3.771114349365234e-05,
      "model_forward_time": 1.7244338989257812,
      "step": 39543
    },
    {
      "epoch": 3.771114349365234e-05,
      "step": 39543,
      "training_step_time": 7.5737879276275635
    },
    {
      "epoch": 3.771209716796875e-05,
      "model_forward_time": 1.707188606262207,
      "step": 39544
    },
    {
      "epoch": 3.771209716796875e-05,
      "step": 39544,
      "training_step_time": 8.091680526733398
    },
    {
      "epoch": 3.771305084228516e-05,
      "model_forward_time": 1.7079710960388184,
      "step": 39545
    },
    {
      "epoch": 3.771305084228516e-05,
      "step": 39545,
      "training_step_time": 7.636423110961914
    },
    {
      "epoch": 3.771400451660156e-05,
      "model_forward_time": 1.7072374820709229,
      "step": 39546
    },
    {
      "epoch": 3.771400451660156e-05,
      "step": 39546,
      "training_step_time": 7.502302885055542
    },
    {
      "epoch": 3.7714958190917966e-05,
      "model_forward_time": 1.7058522701263428,
      "step": 39547
    },
    {
      "epoch": 3.7714958190917966e-05,
      "step": 39547,
      "training_step_time": 7.404801607131958
    },
    {
      "epoch": 3.771591186523438e-05,
      "model_forward_time": 1.725243091583252,
      "step": 39548
    },
    {
      "epoch": 3.771591186523438e-05,
      "step": 39548,
      "training_step_time": 7.699525356292725
    },
    {
      "epoch": 3.771686553955078e-05,
      "model_forward_time": 1.711378574371338,
      "step": 39549
    },
    {
      "epoch": 3.771686553955078e-05,
      "step": 39549,
      "training_step_time": 7.274189233779907
    },
    {
      "epoch": 3.7717819213867185e-05,
      "grad_norm": 0.14969897270202637,
      "learning_rate": 7.076541796980341e-06,
      "loss": 0.0604,
      "step": 39550
    },
    {
      "epoch": 3.7717819213867185e-05,
      "model_forward_time": 1.7045528888702393,
      "step": 39550
    },
    {
      "dynamics_loss_avg": 0.05268200747668743,
      "epoch": 3.7717819213867185e-05,
      "step": 39550
    },
    {
      "action_loss_avg": 0.00200530425645411,
      "epoch": 3.7717819213867185e-05,
      "step": 39550
    },
    {
      "epoch": 3.7717819213867185e-05,
      "step": 39550,
      "training_step_time": 7.791294097900391
    },
    {
      "epoch": 3.7718772888183596e-05,
      "model_forward_time": 1.7133934497833252,
      "step": 39551
    },
    {
      "epoch": 3.7718772888183596e-05,
      "step": 39551,
      "training_step_time": 7.726605415344238
    },
    {
      "epoch": 3.77197265625e-05,
      "model_forward_time": 2.2526793479919434,
      "step": 39552
    },
    {
      "epoch": 3.77197265625e-05,
      "step": 39552,
      "training_step_time": 8.350280523300171
    },
    {
      "epoch": 3.7720680236816404e-05,
      "model_forward_time": 1.710728406906128,
      "step": 39553
    },
    {
      "epoch": 3.7720680236816404e-05,
      "step": 39553,
      "training_step_time": 7.590971231460571
    },
    {
      "epoch": 3.7721633911132814e-05,
      "model_forward_time": 1.7106530666351318,
      "step": 39554
    },
    {
      "epoch": 3.7721633911132814e-05,
      "step": 39554,
      "training_step_time": 7.631675481796265
    },
    {
      "epoch": 3.772258758544922e-05,
      "model_forward_time": 1.7128753662109375,
      "step": 39555
    },
    {
      "epoch": 3.772258758544922e-05,
      "step": 39555,
      "training_step_time": 8.464853286743164
    },
    {
      "epoch": 3.772354125976562e-05,
      "model_forward_time": 1.7220706939697266,
      "step": 39556
    },
    {
      "epoch": 3.772354125976562e-05,
      "step": 39556,
      "training_step_time": 7.38201117515564
    },
    {
      "epoch": 3.772449493408203e-05,
      "model_forward_time": 1.7319109439849854,
      "step": 39557
    },
    {
      "epoch": 3.772449493408203e-05,
      "step": 39557,
      "training_step_time": 8.15277647972107
    },
    {
      "epoch": 3.772544860839844e-05,
      "model_forward_time": 1.7125229835510254,
      "step": 39558
    },
    {
      "epoch": 3.772544860839844e-05,
      "step": 39558,
      "training_step_time": 7.236045598983765
    },
    {
      "epoch": 3.772640228271484e-05,
      "model_forward_time": 1.7118644714355469,
      "step": 39559
    },
    {
      "epoch": 3.772640228271484e-05,
      "step": 39559,
      "training_step_time": 7.788638591766357
    },
    {
      "epoch": 3.772735595703125e-05,
      "grad_norm": 0.15702401101589203,
      "learning_rate": 7.075037553891322e-06,
      "loss": 0.0638,
      "step": 39560
    },
    {
      "epoch": 3.772735595703125e-05,
      "model_forward_time": 1.7079336643218994,
      "step": 39560
    },
    {
      "dynamics_loss_avg": 0.06102814357727766,
      "epoch": 3.772735595703125e-05,
      "step": 39560
    },
    {
      "action_loss_avg": 0.002599285339238122,
      "epoch": 3.772735595703125e-05,
      "step": 39560
    },
    {
      "epoch": 3.772735595703125e-05,
      "step": 39560,
      "training_step_time": 8.955626487731934
    },
    {
      "epoch": 3.7728309631347656e-05,
      "model_forward_time": 2.26303768157959,
      "step": 39561
    },
    {
      "epoch": 3.7728309631347656e-05,
      "step": 39561,
      "training_step_time": 8.668307065963745
    },
    {
      "epoch": 3.772926330566406e-05,
      "model_forward_time": 1.7035884857177734,
      "step": 39562
    },
    {
      "epoch": 3.772926330566406e-05,
      "step": 39562,
      "training_step_time": 8.244692325592041
    },
    {
      "epoch": 3.773021697998047e-05,
      "model_forward_time": 1.715918779373169,
      "step": 39563
    },
    {
      "epoch": 3.773021697998047e-05,
      "step": 39563,
      "training_step_time": 7.6658995151519775
    },
    {
      "epoch": 3.7731170654296875e-05,
      "model_forward_time": 1.709888219833374,
      "step": 39564
    },
    {
      "epoch": 3.7731170654296875e-05,
      "step": 39564,
      "training_step_time": 7.720942974090576
    },
    {
      "epoch": 3.773212432861328e-05,
      "model_forward_time": 1.7259478569030762,
      "step": 39565
    },
    {
      "epoch": 3.773212432861328e-05,
      "step": 39565,
      "training_step_time": 7.22095513343811
    },
    {
      "epoch": 3.773307800292969e-05,
      "model_forward_time": 1.7493987083435059,
      "step": 39566
    },
    {
      "epoch": 3.773307800292969e-05,
      "step": 39566,
      "training_step_time": 8.424843311309814
    },
    {
      "epoch": 3.7734031677246094e-05,
      "model_forward_time": 1.7063992023468018,
      "step": 39567
    },
    {
      "epoch": 3.7734031677246094e-05,
      "step": 39567,
      "training_step_time": 7.635666608810425
    },
    {
      "epoch": 3.77349853515625e-05,
      "model_forward_time": 1.7113456726074219,
      "step": 39568
    },
    {
      "epoch": 3.77349853515625e-05,
      "step": 39568,
      "training_step_time": 8.56142282485962
    },
    {
      "epoch": 3.773593902587891e-05,
      "model_forward_time": 1.7158515453338623,
      "step": 39569
    },
    {
      "epoch": 3.773593902587891e-05,
      "step": 39569,
      "training_step_time": 8.675001621246338
    },
    {
      "epoch": 3.773689270019531e-05,
      "grad_norm": 0.11366717517375946,
      "learning_rate": 7.073533083879315e-06,
      "loss": 0.0663,
      "step": 39570
    },
    {
      "epoch": 3.773689270019531e-05,
      "model_forward_time": 2.271658182144165,
      "step": 39570
    },
    {
      "dynamics_loss_avg": 0.06057245954871178,
      "epoch": 3.773689270019531e-05,
      "step": 39570
    },
    {
      "action_loss_avg": 0.0022268256114330144,
      "epoch": 3.773689270019531e-05,
      "step": 39570
    },
    {
      "epoch": 3.773689270019531e-05,
      "step": 39570,
      "training_step_time": 8.301196575164795
    },
    {
      "epoch": 3.773784637451172e-05,
      "model_forward_time": 1.7145118713378906,
      "step": 39571
    },
    {
      "epoch": 3.773784637451172e-05,
      "step": 39571,
      "training_step_time": 8.921505212783813
    },
    {
      "epoch": 3.773880004882813e-05,
      "model_forward_time": 1.7118806838989258,
      "step": 39572
    },
    {
      "epoch": 3.773880004882813e-05,
      "step": 39572,
      "training_step_time": 7.715835094451904
    },
    {
      "epoch": 3.773975372314453e-05,
      "model_forward_time": 1.713629961013794,
      "step": 39573
    },
    {
      "epoch": 3.773975372314453e-05,
      "step": 39573,
      "training_step_time": 8.778178453445435
    },
    {
      "epoch": 3.7740707397460936e-05,
      "model_forward_time": 1.720444679260254,
      "step": 39574
    },
    {
      "epoch": 3.7740707397460936e-05,
      "step": 39574,
      "training_step_time": 7.512078762054443
    },
    {
      "epoch": 3.7741661071777346e-05,
      "model_forward_time": 1.7331461906433105,
      "step": 39575
    },
    {
      "epoch": 3.7741661071777346e-05,
      "step": 39575,
      "training_step_time": 7.591431617736816
    },
    {
      "epoch": 3.774261474609375e-05,
      "model_forward_time": 1.7043259143829346,
      "step": 39576
    },
    {
      "epoch": 3.774261474609375e-05,
      "step": 39576,
      "training_step_time": 7.6395652294158936
    },
    {
      "epoch": 3.7743568420410155e-05,
      "model_forward_time": 1.7108755111694336,
      "step": 39577
    },
    {
      "epoch": 3.7743568420410155e-05,
      "step": 39577,
      "training_step_time": 8.455588579177856
    },
    {
      "epoch": 3.7744522094726565e-05,
      "model_forward_time": 1.7138614654541016,
      "step": 39578
    },
    {
      "epoch": 3.7744522094726565e-05,
      "step": 39578,
      "training_step_time": 7.617099046707153
    },
    {
      "epoch": 3.774547576904297e-05,
      "model_forward_time": 2.2934277057647705,
      "step": 39579
    },
    {
      "epoch": 3.774547576904297e-05,
      "step": 39579,
      "training_step_time": 8.614857912063599
    },
    {
      "epoch": 3.774642944335937e-05,
      "grad_norm": 0.09539984166622162,
      "learning_rate": 7.072028387108849e-06,
      "loss": 0.0636,
      "step": 39580
    },
    {
      "epoch": 3.774642944335937e-05,
      "model_forward_time": 1.7136847972869873,
      "step": 39580
    },
    {
      "dynamics_loss_avg": 0.0629333758726716,
      "epoch": 3.774642944335937e-05,
      "step": 39580
    },
    {
      "action_loss_avg": 0.001571501069702208,
      "epoch": 3.774642944335937e-05,
      "step": 39580
    },
    {
      "epoch": 3.774642944335937e-05,
      "step": 39580,
      "training_step_time": 8.382333040237427
    },
    {
      "epoch": 3.7747383117675784e-05,
      "model_forward_time": 1.7109367847442627,
      "step": 39581
    },
    {
      "epoch": 3.7747383117675784e-05,
      "step": 39581,
      "training_step_time": 7.69292140007019
    },
    {
      "epoch": 3.774833679199219e-05,
      "model_forward_time": 1.71181321144104,
      "step": 39582
    },
    {
      "epoch": 3.774833679199219e-05,
      "step": 39582,
      "training_step_time": 7.756448745727539
    },
    {
      "epoch": 3.774929046630859e-05,
      "model_forward_time": 1.7202563285827637,
      "step": 39583
    },
    {
      "epoch": 3.774929046630859e-05,
      "step": 39583,
      "training_step_time": 8.182901859283447
    },
    {
      "epoch": 3.7750244140625e-05,
      "model_forward_time": 1.7477328777313232,
      "step": 39584
    },
    {
      "epoch": 3.7750244140625e-05,
      "step": 39584,
      "training_step_time": 7.578387260437012
    },
    {
      "epoch": 3.775119781494141e-05,
      "model_forward_time": 1.7132635116577148,
      "step": 39585
    },
    {
      "epoch": 3.775119781494141e-05,
      "step": 39585,
      "training_step_time": 7.667686700820923
    },
    {
      "epoch": 3.775215148925781e-05,
      "model_forward_time": 1.7146153450012207,
      "step": 39586
    },
    {
      "epoch": 3.775215148925781e-05,
      "step": 39586,
      "training_step_time": 7.194686412811279
    },
    {
      "epoch": 3.775310516357422e-05,
      "model_forward_time": 1.7352924346923828,
      "step": 39587
    },
    {
      "epoch": 3.775310516357422e-05,
      "step": 39587,
      "training_step_time": 8.197717666625977
    },
    {
      "epoch": 3.7754058837890626e-05,
      "model_forward_time": 2.2908847332000732,
      "step": 39588
    },
    {
      "epoch": 3.7754058837890626e-05,
      "step": 39588,
      "training_step_time": 7.628881454467773
    },
    {
      "epoch": 3.775501251220703e-05,
      "model_forward_time": 1.7374544143676758,
      "step": 39589
    },
    {
      "epoch": 3.775501251220703e-05,
      "step": 39589,
      "training_step_time": 7.577227354049683
    },
    {
      "epoch": 3.775596618652344e-05,
      "grad_norm": 0.1447005420923233,
      "learning_rate": 7.070523463744473e-06,
      "loss": 0.0622,
      "step": 39590
    },
    {
      "epoch": 3.775596618652344e-05,
      "model_forward_time": 1.728273868560791,
      "step": 39590
    },
    {
      "dynamics_loss_avg": 0.05234059225767851,
      "epoch": 3.775596618652344e-05,
      "step": 39590
    },
    {
      "action_loss_avg": 0.0016122625092975795,
      "epoch": 3.775596618652344e-05,
      "step": 39590
    },
    {
      "epoch": 3.775596618652344e-05,
      "step": 39590,
      "training_step_time": 7.475695848464966
    },
    {
      "epoch": 3.7756919860839845e-05,
      "model_forward_time": 1.7332792282104492,
      "step": 39591
    },
    {
      "epoch": 3.7756919860839845e-05,
      "step": 39591,
      "training_step_time": 7.625802278518677
    },
    {
      "epoch": 3.775787353515625e-05,
      "model_forward_time": 1.7251989841461182,
      "step": 39592
    },
    {
      "epoch": 3.775787353515625e-05,
      "step": 39592,
      "training_step_time": 7.569045543670654
    },
    {
      "epoch": 3.775882720947266e-05,
      "model_forward_time": 1.7926111221313477,
      "step": 39593
    },
    {
      "epoch": 3.775882720947266e-05,
      "step": 39593,
      "training_step_time": 7.540326833724976
    },
    {
      "epoch": 3.7759780883789064e-05,
      "model_forward_time": 1.7149271965026855,
      "step": 39594
    },
    {
      "epoch": 3.7759780883789064e-05,
      "step": 39594,
      "training_step_time": 7.656419038772583
    },
    {
      "epoch": 3.776073455810547e-05,
      "model_forward_time": 1.7176198959350586,
      "step": 39595
    },
    {
      "epoch": 3.776073455810547e-05,
      "step": 39595,
      "training_step_time": 7.202846527099609
    },
    {
      "epoch": 3.776168823242187e-05,
      "model_forward_time": 1.7194421291351318,
      "step": 39596
    },
    {
      "epoch": 3.776168823242187e-05,
      "step": 39596,
      "training_step_time": 9.373749732971191
    },
    {
      "epoch": 3.776264190673828e-05,
      "model_forward_time": 1.7168610095977783,
      "step": 39597
    },
    {
      "epoch": 3.776264190673828e-05,
      "step": 39597,
      "training_step_time": 10.14670467376709
    },
    {
      "epoch": 3.7763595581054687e-05,
      "model_forward_time": 1.7168848514556885,
      "step": 39598
    },
    {
      "epoch": 3.7763595581054687e-05,
      "step": 39598,
      "training_step_time": 8.624738454818726
    },
    {
      "epoch": 3.776454925537109e-05,
      "model_forward_time": 1.722245454788208,
      "step": 39599
    },
    {
      "epoch": 3.776454925537109e-05,
      "step": 39599,
      "training_step_time": 7.546171426773071
    },
    {
      "epoch": 3.77655029296875e-05,
      "grad_norm": 0.12510131299495697,
      "learning_rate": 7.0690183139507625e-06,
      "loss": 0.0583,
      "step": 39600
    },
    {
      "epoch": 3.77655029296875e-05,
      "model_forward_time": 1.7193236351013184,
      "step": 39600
    },
    {
      "dynamics_loss_avg": 0.0682673178613186,
      "epoch": 3.77655029296875e-05,
      "step": 39600
    },
    {
      "action_loss_avg": 0.001754646172048524,
      "epoch": 3.77655029296875e-05,
      "step": 39600
    },
    {
      "epoch": 3.77655029296875e-05,
      "step": 39600,
      "training_step_time": 7.73176121711731
    },
    {
      "epoch": 3.7766456604003905e-05,
      "model_forward_time": 1.3472394943237305,
      "step": 39601
    },
    {
      "epoch": 3.7766456604003905e-05,
      "step": 39601,
      "training_step_time": 7.615647315979004
    },
    {
      "epoch": 3.776741027832031e-05,
      "model_forward_time": 1.7715067863464355,
      "step": 39602
    },
    {
      "epoch": 3.776741027832031e-05,
      "step": 39602,
      "training_step_time": 7.645040512084961
    },
    {
      "epoch": 3.776836395263672e-05,
      "model_forward_time": 1.71040940284729,
      "step": 39603
    },
    {
      "epoch": 3.776836395263672e-05,
      "step": 39603,
      "training_step_time": 7.638716697692871
    },
    {
      "epoch": 3.7769317626953124e-05,
      "model_forward_time": 1.7154512405395508,
      "step": 39604
    },
    {
      "epoch": 3.7769317626953124e-05,
      "step": 39604,
      "training_step_time": 7.8260276317596436
    },
    {
      "epoch": 3.777027130126953e-05,
      "model_forward_time": 1.6978070735931396,
      "step": 39605
    },
    {
      "epoch": 3.777027130126953e-05,
      "step": 39605,
      "training_step_time": 7.64345645904541
    },
    {
      "epoch": 3.777122497558594e-05,
      "model_forward_time": 1.7112829685211182,
      "step": 39606
    },
    {
      "epoch": 3.777122497558594e-05,
      "step": 39606,
      "training_step_time": 8.527919054031372
    },
    {
      "epoch": 3.777217864990234e-05,
      "model_forward_time": 2.2581124305725098,
      "step": 39607
    },
    {
      "epoch": 3.777217864990234e-05,
      "step": 39607,
      "training_step_time": 10.084484338760376
    },
    {
      "epoch": 3.777313232421875e-05,
      "model_forward_time": 1.7173550128936768,
      "step": 39608
    },
    {
      "epoch": 3.777313232421875e-05,
      "step": 39608,
      "training_step_time": 8.348937511444092
    },
    {
      "epoch": 3.777408599853516e-05,
      "model_forward_time": 1.7149629592895508,
      "step": 39609
    },
    {
      "epoch": 3.777408599853516e-05,
      "step": 39609,
      "training_step_time": 7.665342330932617
    },
    {
      "epoch": 3.777503967285156e-05,
      "grad_norm": 0.10490766167640686,
      "learning_rate": 7.0675129378923224e-06,
      "loss": 0.0554,
      "step": 39610
    },
    {
      "epoch": 3.777503967285156e-05,
      "model_forward_time": 1.7067608833312988,
      "step": 39610
    },
    {
      "dynamics_loss_avg": 0.04995247721672058,
      "epoch": 3.777503967285156e-05,
      "step": 39610
    },
    {
      "action_loss_avg": 0.0018788496905472129,
      "epoch": 3.777503967285156e-05,
      "step": 39610
    },
    {
      "epoch": 3.777503967285156e-05,
      "step": 39610,
      "training_step_time": 7.522868871688843
    },
    {
      "epoch": 3.7775993347167966e-05,
      "model_forward_time": 1.7431044578552246,
      "step": 39611
    },
    {
      "epoch": 3.7775993347167966e-05,
      "step": 39611,
      "training_step_time": 8.434834957122803
    },
    {
      "epoch": 3.777694702148438e-05,
      "model_forward_time": 1.7100679874420166,
      "step": 39612
    },
    {
      "epoch": 3.777694702148438e-05,
      "step": 39612,
      "training_step_time": 8.211539268493652
    },
    {
      "epoch": 3.777790069580078e-05,
      "model_forward_time": 1.7064075469970703,
      "step": 39613
    },
    {
      "epoch": 3.777790069580078e-05,
      "step": 39613,
      "training_step_time": 9.40153193473816
    },
    {
      "epoch": 3.7778854370117185e-05,
      "model_forward_time": 1.713360071182251,
      "step": 39614
    },
    {
      "epoch": 3.7778854370117185e-05,
      "step": 39614,
      "training_step_time": 9.222598552703857
    },
    {
      "epoch": 3.7779808044433596e-05,
      "model_forward_time": 1.7103655338287354,
      "step": 39615
    },
    {
      "epoch": 3.7779808044433596e-05,
      "step": 39615,
      "training_step_time": 8.662587642669678
    },
    {
      "epoch": 3.778076171875e-05,
      "model_forward_time": 2.2688186168670654,
      "step": 39616
    },
    {
      "epoch": 3.778076171875e-05,
      "step": 39616,
      "training_step_time": 7.655886173248291
    },
    {
      "epoch": 3.7781715393066404e-05,
      "model_forward_time": 1.7158591747283936,
      "step": 39617
    },
    {
      "epoch": 3.7781715393066404e-05,
      "step": 39617,
      "training_step_time": 9.819833755493164
    },
    {
      "epoch": 3.7782669067382815e-05,
      "model_forward_time": 1.7126047611236572,
      "step": 39618
    },
    {
      "epoch": 3.7782669067382815e-05,
      "step": 39618,
      "training_step_time": 7.606167554855347
    },
    {
      "epoch": 3.778362274169922e-05,
      "model_forward_time": 1.7099289894104004,
      "step": 39619
    },
    {
      "epoch": 3.778362274169922e-05,
      "step": 39619,
      "training_step_time": 7.5987184047698975
    },
    {
      "epoch": 3.778457641601562e-05,
      "grad_norm": 0.10008437186479568,
      "learning_rate": 7.066007335733775e-06,
      "loss": 0.0568,
      "step": 39620
    },
    {
      "epoch": 3.778457641601562e-05,
      "model_forward_time": 1.7222771644592285,
      "step": 39620
    },
    {
      "dynamics_loss_avg": 0.053587361425161364,
      "epoch": 3.778457641601562e-05,
      "step": 39620
    },
    {
      "action_loss_avg": 0.0020337109453976154,
      "epoch": 3.778457641601562e-05,
      "step": 39620
    },
    {
      "epoch": 3.778457641601562e-05,
      "step": 39620,
      "training_step_time": 8.23385500907898
    },
    {
      "epoch": 3.778553009033203e-05,
      "model_forward_time": 1.7127103805541992,
      "step": 39621
    },
    {
      "epoch": 3.778553009033203e-05,
      "step": 39621,
      "training_step_time": 7.53871488571167
    },
    {
      "epoch": 3.778648376464844e-05,
      "model_forward_time": 1.3364107608795166,
      "step": 39622
    },
    {
      "epoch": 3.778648376464844e-05,
      "step": 39622,
      "training_step_time": 8.513413906097412
    },
    {
      "epoch": 3.778743743896484e-05,
      "model_forward_time": 1.7185916900634766,
      "step": 39623
    },
    {
      "epoch": 3.778743743896484e-05,
      "step": 39623,
      "training_step_time": 7.686928749084473
    },
    {
      "epoch": 3.778839111328125e-05,
      "model_forward_time": 1.716414451599121,
      "step": 39624
    },
    {
      "epoch": 3.778839111328125e-05,
      "step": 39624,
      "training_step_time": 8.581972599029541
    },
    {
      "epoch": 3.7789344787597656e-05,
      "model_forward_time": 2.2934389114379883,
      "step": 39625
    },
    {
      "epoch": 3.7789344787597656e-05,
      "step": 39625,
      "training_step_time": 8.401685953140259
    },
    {
      "epoch": 3.779029846191406e-05,
      "model_forward_time": 1.7261593341827393,
      "step": 39626
    },
    {
      "epoch": 3.779029846191406e-05,
      "step": 39626,
      "training_step_time": 8.325040578842163
    },
    {
      "epoch": 3.779125213623047e-05,
      "model_forward_time": 1.715447187423706,
      "step": 39627
    },
    {
      "epoch": 3.779125213623047e-05,
      "step": 39627,
      "training_step_time": 7.422304153442383
    },
    {
      "epoch": 3.7792205810546875e-05,
      "model_forward_time": 1.7025599479675293,
      "step": 39628
    },
    {
      "epoch": 3.7792205810546875e-05,
      "step": 39628,
      "training_step_time": 8.732966661453247
    },
    {
      "epoch": 3.779315948486328e-05,
      "model_forward_time": 1.7138340473175049,
      "step": 39629
    },
    {
      "epoch": 3.779315948486328e-05,
      "step": 39629,
      "training_step_time": 7.537003755569458
    },
    {
      "epoch": 3.779411315917969e-05,
      "grad_norm": 0.10166393220424652,
      "learning_rate": 7.064501507639772e-06,
      "loss": 0.0543,
      "step": 39630
    },
    {
      "epoch": 3.779411315917969e-05,
      "model_forward_time": 1.7021067142486572,
      "step": 39630
    },
    {
      "dynamics_loss_avg": 0.07190762031823397,
      "epoch": 3.779411315917969e-05,
      "step": 39630
    },
    {
      "action_loss_avg": 0.0020785140804946423,
      "epoch": 3.779411315917969e-05,
      "step": 39630
    },
    {
      "epoch": 3.779411315917969e-05,
      "step": 39630,
      "training_step_time": 7.573081970214844
    },
    {
      "epoch": 3.7795066833496094e-05,
      "model_forward_time": 1.6961193084716797,
      "step": 39631
    },
    {
      "epoch": 3.7795066833496094e-05,
      "step": 39631,
      "training_step_time": 7.601307392120361
    },
    {
      "epoch": 3.77960205078125e-05,
      "model_forward_time": 1.7075607776641846,
      "step": 39632
    },
    {
      "epoch": 3.77960205078125e-05,
      "step": 39632,
      "training_step_time": 8.618332624435425
    },
    {
      "epoch": 3.779697418212891e-05,
      "model_forward_time": 1.7051334381103516,
      "step": 39633
    },
    {
      "epoch": 3.779697418212891e-05,
      "step": 39633,
      "training_step_time": 8.460824251174927
    },
    {
      "epoch": 3.779792785644531e-05,
      "model_forward_time": 2.2544262409210205,
      "step": 39634
    },
    {
      "epoch": 3.779792785644531e-05,
      "step": 39634,
      "training_step_time": 8.336535692214966
    },
    {
      "epoch": 3.779888153076172e-05,
      "model_forward_time": 1.711289644241333,
      "step": 39635
    },
    {
      "epoch": 3.779888153076172e-05,
      "step": 39635,
      "training_step_time": 7.652356147766113
    },
    {
      "epoch": 3.779983520507813e-05,
      "model_forward_time": 1.7117600440979004,
      "step": 39636
    },
    {
      "epoch": 3.779983520507813e-05,
      "step": 39636,
      "training_step_time": 7.470423460006714
    },
    {
      "epoch": 3.780078887939453e-05,
      "model_forward_time": 1.7129175662994385,
      "step": 39637
    },
    {
      "epoch": 3.780078887939453e-05,
      "step": 39637,
      "training_step_time": 7.655772924423218
    },
    {
      "epoch": 3.7801742553710936e-05,
      "model_forward_time": 1.7334346771240234,
      "step": 39638
    },
    {
      "epoch": 3.7801742553710936e-05,
      "step": 39638,
      "training_step_time": 8.356430530548096
    },
    {
      "epoch": 3.7802696228027347e-05,
      "model_forward_time": 1.7124664783477783,
      "step": 39639
    },
    {
      "epoch": 3.7802696228027347e-05,
      "step": 39639,
      "training_step_time": 7.61717963218689
    },
    {
      "epoch": 3.780364990234375e-05,
      "grad_norm": 0.09603720158338547,
      "learning_rate": 7.062995453774988e-06,
      "loss": 0.0563,
      "step": 39640
    },
    {
      "epoch": 3.780364990234375e-05,
      "model_forward_time": 1.7114906311035156,
      "step": 39640
    },
    {
      "dynamics_loss_avg": 0.05776324924081564,
      "epoch": 3.780364990234375e-05,
      "step": 39640
    },
    {
      "action_loss_avg": 0.0018799503217451275,
      "epoch": 3.780364990234375e-05,
      "step": 39640
    },
    {
      "epoch": 3.780364990234375e-05,
      "step": 39640,
      "training_step_time": 7.660874128341675
    },
    {
      "epoch": 3.7804603576660155e-05,
      "model_forward_time": 1.7170500755310059,
      "step": 39641
    },
    {
      "epoch": 3.7804603576660155e-05,
      "step": 39641,
      "training_step_time": 7.652245998382568
    },
    {
      "epoch": 3.7805557250976565e-05,
      "model_forward_time": 1.7125012874603271,
      "step": 39642
    },
    {
      "epoch": 3.7805557250976565e-05,
      "step": 39642,
      "training_step_time": 7.654119968414307
    },
    {
      "epoch": 3.780651092529297e-05,
      "model_forward_time": 1.7079954147338867,
      "step": 39643
    },
    {
      "epoch": 3.780651092529297e-05,
      "step": 39643,
      "training_step_time": 8.680172681808472
    },
    {
      "epoch": 3.7807464599609373e-05,
      "model_forward_time": 2.259697675704956,
      "step": 39644
    },
    {
      "epoch": 3.7807464599609373e-05,
      "step": 39644,
      "training_step_time": 8.642837762832642
    },
    {
      "epoch": 3.7808418273925784e-05,
      "model_forward_time": 1.7038660049438477,
      "step": 39645
    },
    {
      "epoch": 3.7808418273925784e-05,
      "step": 39645,
      "training_step_time": 7.4766623973846436
    },
    {
      "epoch": 3.780937194824219e-05,
      "model_forward_time": 1.701613426208496,
      "step": 39646
    },
    {
      "epoch": 3.780937194824219e-05,
      "step": 39646,
      "training_step_time": 7.612191915512085
    },
    {
      "epoch": 3.781032562255859e-05,
      "model_forward_time": 1.712357521057129,
      "step": 39647
    },
    {
      "epoch": 3.781032562255859e-05,
      "step": 39647,
      "training_step_time": 9.113172769546509
    },
    {
      "epoch": 3.7811279296875e-05,
      "model_forward_time": 1.7091867923736572,
      "step": 39648
    },
    {
      "epoch": 3.7811279296875e-05,
      "step": 39648,
      "training_step_time": 7.582705020904541
    },
    {
      "epoch": 3.781223297119141e-05,
      "model_forward_time": 1.7109689712524414,
      "step": 39649
    },
    {
      "epoch": 3.781223297119141e-05,
      "step": 39649,
      "training_step_time": 7.612493991851807
    },
    {
      "epoch": 3.781318664550781e-05,
      "grad_norm": 0.09633246809244156,
      "learning_rate": 7.061489174304121e-06,
      "loss": 0.0551,
      "step": 39650
    },
    {
      "epoch": 3.781318664550781e-05,
      "model_forward_time": 1.7080984115600586,
      "step": 39650
    },
    {
      "dynamics_loss_avg": 0.06281359512358904,
      "epoch": 3.781318664550781e-05,
      "step": 39650
    },
    {
      "action_loss_avg": 0.0019985974358860403,
      "epoch": 3.781318664550781e-05,
      "step": 39650
    },
    {
      "epoch": 3.781318664550781e-05,
      "step": 39650,
      "training_step_time": 7.510939836502075
    },
    {
      "epoch": 3.781414031982422e-05,
      "model_forward_time": 1.7067890167236328,
      "step": 39651
    },
    {
      "epoch": 3.781414031982422e-05,
      "step": 39651,
      "training_step_time": 8.647512197494507
    },
    {
      "epoch": 3.7815093994140626e-05,
      "model_forward_time": 1.714451551437378,
      "step": 39652
    },
    {
      "epoch": 3.7815093994140626e-05,
      "step": 39652,
      "training_step_time": 7.5999298095703125
    },
    {
      "epoch": 3.781604766845703e-05,
      "model_forward_time": 2.2776880264282227,
      "step": 39653
    },
    {
      "epoch": 3.781604766845703e-05,
      "step": 39653,
      "training_step_time": 8.606107234954834
    },
    {
      "epoch": 3.781700134277344e-05,
      "model_forward_time": 1.7080676555633545,
      "step": 39654
    },
    {
      "epoch": 3.781700134277344e-05,
      "step": 39654,
      "training_step_time": 8.392431259155273
    },
    {
      "epoch": 3.7817955017089845e-05,
      "model_forward_time": 1.711418628692627,
      "step": 39655
    },
    {
      "epoch": 3.7817955017089845e-05,
      "step": 39655,
      "training_step_time": 7.6084465980529785
    },
    {
      "epoch": 3.781890869140625e-05,
      "model_forward_time": 1.708855390548706,
      "step": 39656
    },
    {
      "epoch": 3.781890869140625e-05,
      "step": 39656,
      "training_step_time": 7.535334587097168
    },
    {
      "epoch": 3.781986236572265e-05,
      "model_forward_time": 1.7034924030303955,
      "step": 39657
    },
    {
      "epoch": 3.781986236572265e-05,
      "step": 39657,
      "training_step_time": 7.479823112487793
    },
    {
      "epoch": 3.7820816040039064e-05,
      "model_forward_time": 1.7163701057434082,
      "step": 39658
    },
    {
      "epoch": 3.7820816040039064e-05,
      "step": 39658,
      "training_step_time": 8.435840129852295
    },
    {
      "epoch": 3.782176971435547e-05,
      "model_forward_time": 1.7184123992919922,
      "step": 39659
    },
    {
      "epoch": 3.782176971435547e-05,
      "step": 39659,
      "training_step_time": 7.540111303329468
    },
    {
      "epoch": 3.782272338867187e-05,
      "grad_norm": 0.09822701662778854,
      "learning_rate": 7.059982669391901e-06,
      "loss": 0.0572,
      "step": 39660
    },
    {
      "epoch": 3.782272338867187e-05,
      "model_forward_time": 1.700873613357544,
      "step": 39660
    },
    {
      "dynamics_loss_avg": 0.0641310852020979,
      "epoch": 3.782272338867187e-05,
      "step": 39660
    },
    {
      "action_loss_avg": 0.0018605695222504437,
      "epoch": 3.782272338867187e-05,
      "step": 39660
    },
    {
      "epoch": 3.782272338867187e-05,
      "step": 39660,
      "training_step_time": 7.706131458282471
    },
    {
      "epoch": 3.782367706298828e-05,
      "model_forward_time": 1.7237143516540527,
      "step": 39661
    },
    {
      "epoch": 3.782367706298828e-05,
      "step": 39661,
      "training_step_time": 7.66921854019165
    },
    {
      "epoch": 3.7824630737304687e-05,
      "model_forward_time": 2.9138028621673584,
      "step": 39662
    },
    {
      "epoch": 3.7824630737304687e-05,
      "step": 39662,
      "training_step_time": 8.465965986251831
    },
    {
      "epoch": 3.782558441162109e-05,
      "model_forward_time": 1.7435734272003174,
      "step": 39663
    },
    {
      "epoch": 3.782558441162109e-05,
      "step": 39663,
      "training_step_time": 8.404263734817505
    },
    {
      "epoch": 3.78265380859375e-05,
      "model_forward_time": 1.7533502578735352,
      "step": 39664
    },
    {
      "epoch": 3.78265380859375e-05,
      "step": 39664,
      "training_step_time": 8.160895347595215
    },
    {
      "epoch": 3.7827491760253905e-05,
      "model_forward_time": 1.809678077697754,
      "step": 39665
    },
    {
      "epoch": 3.7827491760253905e-05,
      "step": 39665,
      "training_step_time": 7.565265893936157
    },
    {
      "epoch": 3.782844543457031e-05,
      "model_forward_time": 1.7340011596679688,
      "step": 39666
    },
    {
      "epoch": 3.782844543457031e-05,
      "step": 39666,
      "training_step_time": 7.5629661083221436
    },
    {
      "epoch": 3.782939910888672e-05,
      "model_forward_time": 1.7465476989746094,
      "step": 39667
    },
    {
      "epoch": 3.782939910888672e-05,
      "step": 39667,
      "training_step_time": 7.56290078163147
    },
    {
      "epoch": 3.7830352783203124e-05,
      "model_forward_time": 1.368053674697876,
      "step": 39668
    },
    {
      "epoch": 3.7830352783203124e-05,
      "step": 39668,
      "training_step_time": 7.517171382904053
    },
    {
      "epoch": 3.783130645751953e-05,
      "model_forward_time": 1.7381484508514404,
      "step": 39669
    },
    {
      "epoch": 3.783130645751953e-05,
      "step": 39669,
      "training_step_time": 7.7241880893707275
    },
    {
      "epoch": 3.783226013183594e-05,
      "grad_norm": 0.1125105693936348,
      "learning_rate": 7.05847593920307e-06,
      "loss": 0.0594,
      "step": 39670
    },
    {
      "epoch": 3.783226013183594e-05,
      "model_forward_time": 1.734245777130127,
      "step": 39670
    },
    {
      "dynamics_loss_avg": 0.05806831140071154,
      "epoch": 3.783226013183594e-05,
      "step": 39670
    },
    {
      "action_loss_avg": 0.0012628317024791613,
      "epoch": 3.783226013183594e-05,
      "step": 39670
    },
    {
      "epoch": 3.783226013183594e-05,
      "step": 39670,
      "training_step_time": 7.647553443908691
    },
    {
      "epoch": 3.783321380615234e-05,
      "model_forward_time": 2.8887290954589844,
      "step": 39671
    },
    {
      "epoch": 3.783321380615234e-05,
      "step": 39671,
      "training_step_time": 7.661720275878906
    },
    {
      "epoch": 3.783416748046875e-05,
      "model_forward_time": 1.7353394031524658,
      "step": 39672
    },
    {
      "epoch": 3.783416748046875e-05,
      "step": 39672,
      "training_step_time": 7.593647003173828
    },
    {
      "epoch": 3.783512115478516e-05,
      "model_forward_time": 1.7317297458648682,
      "step": 39673
    },
    {
      "epoch": 3.783512115478516e-05,
      "step": 39673,
      "training_step_time": 8.152949571609497
    },
    {
      "epoch": 3.783607482910156e-05,
      "model_forward_time": 1.785548448562622,
      "step": 39674
    },
    {
      "epoch": 3.783607482910156e-05,
      "step": 39674,
      "training_step_time": 7.587496995925903
    },
    {
      "epoch": 3.7837028503417966e-05,
      "model_forward_time": 1.738391399383545,
      "step": 39675
    },
    {
      "epoch": 3.7837028503417966e-05,
      "step": 39675,
      "training_step_time": 7.537819862365723
    },
    {
      "epoch": 3.783798217773438e-05,
      "model_forward_time": 1.734992265701294,
      "step": 39676
    },
    {
      "epoch": 3.783798217773438e-05,
      "step": 39676,
      "training_step_time": 7.65482759475708
    },
    {
      "epoch": 3.783893585205078e-05,
      "model_forward_time": 1.7372932434082031,
      "step": 39677
    },
    {
      "epoch": 3.783893585205078e-05,
      "step": 39677,
      "training_step_time": 7.6197829246521
    },
    {
      "epoch": 3.7839889526367185e-05,
      "model_forward_time": 1.7362089157104492,
      "step": 39678
    },
    {
      "epoch": 3.7839889526367185e-05,
      "step": 39678,
      "training_step_time": 8.385671377182007
    },
    {
      "epoch": 3.7840843200683596e-05,
      "model_forward_time": 1.7447569370269775,
      "step": 39679
    },
    {
      "epoch": 3.7840843200683596e-05,
      "step": 39679,
      "training_step_time": 9.278721809387207
    },
    {
      "epoch": 3.7841796875e-05,
      "grad_norm": 0.09300386160612106,
      "learning_rate": 7.056968983902406e-06,
      "loss": 0.0559,
      "step": 39680
    },
    {
      "epoch": 3.7841796875e-05,
      "model_forward_time": 2.8450350761413574,
      "step": 39680
    },
    {
      "dynamics_loss_avg": 0.06485046185553074,
      "epoch": 3.7841796875e-05,
      "step": 39680
    },
    {
      "action_loss_avg": 0.0034532721794676034,
      "epoch": 3.7841796875e-05,
      "step": 39680
    },
    {
      "epoch": 3.7841796875e-05,
      "step": 39680,
      "training_step_time": 7.684967756271362
    },
    {
      "epoch": 3.7842750549316404e-05,
      "model_forward_time": 1.7250497341156006,
      "step": 39681
    },
    {
      "epoch": 3.7842750549316404e-05,
      "step": 39681,
      "training_step_time": 8.155656814575195
    },
    {
      "epoch": 3.7843704223632815e-05,
      "model_forward_time": 1.710597276687622,
      "step": 39682
    },
    {
      "epoch": 3.7843704223632815e-05,
      "step": 39682,
      "training_step_time": 7.319810628890991
    },
    {
      "epoch": 3.784465789794922e-05,
      "model_forward_time": 1.7429099082946777,
      "step": 39683
    },
    {
      "epoch": 3.784465789794922e-05,
      "step": 39683,
      "training_step_time": 7.565121412277222
    },
    {
      "epoch": 3.784561157226562e-05,
      "model_forward_time": 1.730257511138916,
      "step": 39684
    },
    {
      "epoch": 3.784561157226562e-05,
      "step": 39684,
      "training_step_time": 8.622347354888916
    },
    {
      "epoch": 3.7846565246582033e-05,
      "model_forward_time": 1.7330224514007568,
      "step": 39685
    },
    {
      "epoch": 3.7846565246582033e-05,
      "step": 39685,
      "training_step_time": 7.589324474334717
    },
    {
      "epoch": 3.784751892089844e-05,
      "model_forward_time": 1.7202320098876953,
      "step": 39686
    },
    {
      "epoch": 3.784751892089844e-05,
      "step": 39686,
      "training_step_time": 7.560530662536621
    },
    {
      "epoch": 3.784847259521484e-05,
      "model_forward_time": 1.7325360774993896,
      "step": 39687
    },
    {
      "epoch": 3.784847259521484e-05,
      "step": 39687,
      "training_step_time": 8.299267530441284
    },
    {
      "epoch": 3.784942626953125e-05,
      "model_forward_time": 1.7119157314300537,
      "step": 39688
    },
    {
      "epoch": 3.784942626953125e-05,
      "step": 39688,
      "training_step_time": 7.600736141204834
    },
    {
      "epoch": 3.7850379943847656e-05,
      "model_forward_time": 2.8233861923217773,
      "step": 39689
    },
    {
      "epoch": 3.7850379943847656e-05,
      "step": 39689,
      "training_step_time": 9.498788595199585
    },
    {
      "epoch": 3.785133361816406e-05,
      "grad_norm": 0.14041878283023834,
      "learning_rate": 7.0554618036547066e-06,
      "loss": 0.0569,
      "step": 39690
    },
    {
      "epoch": 3.785133361816406e-05,
      "model_forward_time": 1.721358299255371,
      "step": 39690
    },
    {
      "dynamics_loss_avg": 0.04579576775431633,
      "epoch": 3.785133361816406e-05,
      "step": 39690
    },
    {
      "action_loss_avg": 0.0013345099956495686,
      "epoch": 3.785133361816406e-05,
      "step": 39690
    },
    {
      "epoch": 3.785133361816406e-05,
      "step": 39690,
      "training_step_time": 8.306442260742188
    },
    {
      "epoch": 3.785228729248047e-05,
      "model_forward_time": 1.7231872081756592,
      "step": 39691
    },
    {
      "epoch": 3.785228729248047e-05,
      "step": 39691,
      "training_step_time": 7.59578275680542
    },
    {
      "epoch": 3.7853240966796875e-05,
      "model_forward_time": 1.733273983001709,
      "step": 39692
    },
    {
      "epoch": 3.7853240966796875e-05,
      "step": 39692,
      "training_step_time": 7.552655935287476
    },
    {
      "epoch": 3.785419464111328e-05,
      "model_forward_time": 1.432180643081665,
      "step": 39693
    },
    {
      "epoch": 3.785419464111328e-05,
      "step": 39693,
      "training_step_time": 7.593799591064453
    },
    {
      "epoch": 3.785514831542969e-05,
      "model_forward_time": 1.7472681999206543,
      "step": 39694
    },
    {
      "epoch": 3.785514831542969e-05,
      "step": 39694,
      "training_step_time": 7.543610334396362
    },
    {
      "epoch": 3.7856101989746094e-05,
      "model_forward_time": 1.7195672988891602,
      "step": 39695
    },
    {
      "epoch": 3.7856101989746094e-05,
      "step": 39695,
      "training_step_time": 7.560254096984863
    },
    {
      "epoch": 3.78570556640625e-05,
      "model_forward_time": 1.7320327758789062,
      "step": 39696
    },
    {
      "epoch": 3.78570556640625e-05,
      "step": 39696,
      "training_step_time": 7.536466598510742
    },
    {
      "epoch": 3.785800933837891e-05,
      "model_forward_time": 1.7318625450134277,
      "step": 39697
    },
    {
      "epoch": 3.785800933837891e-05,
      "step": 39697,
      "training_step_time": 9.378312110900879
    },
    {
      "epoch": 3.785896301269531e-05,
      "model_forward_time": 1.7366523742675781,
      "step": 39698
    },
    {
      "epoch": 3.785896301269531e-05,
      "step": 39698,
      "training_step_time": 10.297529935836792
    },
    {
      "epoch": 3.785991668701172e-05,
      "model_forward_time": 1.753666877746582,
      "step": 39699
    },
    {
      "epoch": 3.785991668701172e-05,
      "step": 39699,
      "training_step_time": 9.035000562667847
    },
    {
      "epoch": 3.786087036132813e-05,
      "grad_norm": 0.15448717772960663,
      "learning_rate": 7.053954398624795e-06,
      "loss": 0.0571,
      "step": 39700
    },
    {
      "epoch": 3.786087036132813e-05,
      "model_forward_time": 1.7167303562164307,
      "step": 39700
    },
    {
      "dynamics_loss_avg": 0.047851186618208884,
      "epoch": 3.786087036132813e-05,
      "step": 39700
    },
    {
      "action_loss_avg": 0.0016351699188817293,
      "epoch": 3.786087036132813e-05,
      "step": 39700
    },
    {
      "epoch": 3.786087036132813e-05,
      "step": 39700,
      "training_step_time": 6.7212975025177
    },
    {
      "epoch": 3.786182403564453e-05,
      "model_forward_time": 1.7288017272949219,
      "step": 39701
    },
    {
      "epoch": 3.786182403564453e-05,
      "step": 39701,
      "training_step_time": 6.7747907638549805
    },
    {
      "epoch": 3.7862777709960936e-05,
      "model_forward_time": 1.7819066047668457,
      "step": 39702
    },
    {
      "epoch": 3.7862777709960936e-05,
      "step": 39702,
      "training_step_time": 6.750662803649902
    },
    {
      "epoch": 3.7863731384277347e-05,
      "model_forward_time": 1.7256104946136475,
      "step": 39703
    },
    {
      "epoch": 3.7863731384277347e-05,
      "step": 39703,
      "training_step_time": 6.692109107971191
    },
    {
      "epoch": 3.786468505859375e-05,
      "model_forward_time": 1.7485942840576172,
      "step": 39704
    },
    {
      "epoch": 3.786468505859375e-05,
      "step": 39704,
      "training_step_time": 6.742315053939819
    },
    {
      "epoch": 3.7865638732910155e-05,
      "model_forward_time": 1.7341327667236328,
      "step": 39705
    },
    {
      "epoch": 3.7865638732910155e-05,
      "step": 39705,
      "training_step_time": 6.732590913772583
    },
    {
      "epoch": 3.7866592407226565e-05,
      "model_forward_time": 1.7286767959594727,
      "step": 39706
    },
    {
      "epoch": 3.7866592407226565e-05,
      "step": 39706,
      "training_step_time": 6.71985125541687
    },
    {
      "epoch": 3.786754608154297e-05,
      "model_forward_time": 1.728679895401001,
      "step": 39707
    },
    {
      "epoch": 3.786754608154297e-05,
      "step": 39707,
      "training_step_time": 6.718843936920166
    },
    {
      "epoch": 3.7868499755859373e-05,
      "model_forward_time": 2.9176087379455566,
      "step": 39708
    },
    {
      "epoch": 3.7868499755859373e-05,
      "step": 39708,
      "training_step_time": 7.60467267036438
    },
    {
      "epoch": 3.7869453430175784e-05,
      "model_forward_time": 1.7516355514526367,
      "step": 39709
    },
    {
      "epoch": 3.7869453430175784e-05,
      "step": 39709,
      "training_step_time": 7.63506817817688
    },
    {
      "epoch": 3.787040710449219e-05,
      "grad_norm": 0.09105772525072098,
      "learning_rate": 7.052446768977517e-06,
      "loss": 0.0581,
      "step": 39710
    },
    {
      "epoch": 3.787040710449219e-05,
      "model_forward_time": 1.7242627143859863,
      "step": 39710
    },
    {
      "dynamics_loss_avg": 0.049964950792491435,
      "epoch": 3.787040710449219e-05,
      "step": 39710
    },
    {
      "action_loss_avg": 0.0015291200747014956,
      "epoch": 3.787040710449219e-05,
      "step": 39710
    },
    {
      "epoch": 3.787040710449219e-05,
      "step": 39710,
      "training_step_time": 6.7407684326171875
    },
    {
      "epoch": 3.787136077880859e-05,
      "model_forward_time": 1.734403133392334,
      "step": 39711
    },
    {
      "epoch": 3.787136077880859e-05,
      "step": 39711,
      "training_step_time": 6.738507270812988
    },
    {
      "epoch": 3.7872314453125e-05,
      "model_forward_time": 1.7282671928405762,
      "step": 39712
    },
    {
      "epoch": 3.7872314453125e-05,
      "step": 39712,
      "training_step_time": 6.752155542373657
    },
    {
      "epoch": 3.787326812744141e-05,
      "model_forward_time": 1.0198733806610107,
      "step": 39713
    },
    {
      "epoch": 3.787326812744141e-05,
      "step": 39713,
      "training_step_time": 6.743124485015869
    },
    {
      "epoch": 3.787422180175781e-05,
      "model_forward_time": 1.7194581031799316,
      "step": 39714
    },
    {
      "epoch": 3.787422180175781e-05,
      "step": 39714,
      "training_step_time": 6.730570077896118
    },
    {
      "epoch": 3.787517547607422e-05,
      "model_forward_time": 1.7386467456817627,
      "step": 39715
    },
    {
      "epoch": 3.787517547607422e-05,
      "step": 39715,
      "training_step_time": 6.746660232543945
    },
    {
      "epoch": 3.7876129150390626e-05,
      "model_forward_time": 1.730398178100586,
      "step": 39716
    },
    {
      "epoch": 3.7876129150390626e-05,
      "step": 39716,
      "training_step_time": 6.670183181762695
    },
    {
      "epoch": 3.787708282470703e-05,
      "model_forward_time": 2.530081272125244,
      "step": 39717
    },
    {
      "epoch": 3.787708282470703e-05,
      "step": 39717,
      "training_step_time": 7.248857021331787
    },
    {
      "epoch": 3.787803649902344e-05,
      "model_forward_time": 1.7114737033843994,
      "step": 39718
    },
    {
      "epoch": 3.787803649902344e-05,
      "step": 39718,
      "training_step_time": 7.748314142227173
    },
    {
      "epoch": 3.7878990173339845e-05,
      "model_forward_time": 1.7414424419403076,
      "step": 39719
    },
    {
      "epoch": 3.7878990173339845e-05,
      "step": 39719,
      "training_step_time": 6.723450183868408
    },
    {
      "epoch": 3.787994384765625e-05,
      "grad_norm": 0.1394733488559723,
      "learning_rate": 7.050938914877748e-06,
      "loss": 0.0594,
      "step": 39720
    },
    {
      "epoch": 3.787994384765625e-05,
      "model_forward_time": 1.719531774520874,
      "step": 39720
    },
    {
      "dynamics_loss_avg": 0.05743702631443739,
      "epoch": 3.787994384765625e-05,
      "step": 39720
    },
    {
      "action_loss_avg": 0.00271676960401237,
      "epoch": 3.787994384765625e-05,
      "step": 39720
    },
    {
      "epoch": 3.787994384765625e-05,
      "step": 39720,
      "training_step_time": 6.783473253250122
    },
    {
      "epoch": 3.788089752197265e-05,
      "model_forward_time": 1.71714186668396,
      "step": 39721
    },
    {
      "epoch": 3.788089752197265e-05,
      "step": 39721,
      "training_step_time": 6.672891855239868
    },
    {
      "epoch": 3.7881851196289064e-05,
      "model_forward_time": 1.7318391799926758,
      "step": 39722
    },
    {
      "epoch": 3.7881851196289064e-05,
      "step": 39722,
      "training_step_time": 6.736104965209961
    },
    {
      "epoch": 3.788280487060547e-05,
      "model_forward_time": 1.725581407546997,
      "step": 39723
    },
    {
      "epoch": 3.788280487060547e-05,
      "step": 39723,
      "training_step_time": 6.723090887069702
    },
    {
      "epoch": 3.788375854492187e-05,
      "model_forward_time": 1.723738670349121,
      "step": 39724
    },
    {
      "epoch": 3.788375854492187e-05,
      "step": 39724,
      "training_step_time": 6.844195604324341
    },
    {
      "epoch": 3.788471221923828e-05,
      "model_forward_time": 1.7256126403808594,
      "step": 39725
    },
    {
      "epoch": 3.788471221923828e-05,
      "step": 39725,
      "training_step_time": 6.693503141403198
    },
    {
      "epoch": 3.788566589355469e-05,
      "model_forward_time": 2.870971918106079,
      "step": 39726
    },
    {
      "epoch": 3.788566589355469e-05,
      "step": 39726,
      "training_step_time": 7.596473693847656
    },
    {
      "epoch": 3.788661956787109e-05,
      "model_forward_time": 1.7327840328216553,
      "step": 39727
    },
    {
      "epoch": 3.788661956787109e-05,
      "step": 39727,
      "training_step_time": 9.141041278839111
    },
    {
      "epoch": 3.78875732421875e-05,
      "model_forward_time": 1.736994743347168,
      "step": 39728
    },
    {
      "epoch": 3.78875732421875e-05,
      "step": 39728,
      "training_step_time": 7.214051723480225
    },
    {
      "epoch": 3.7888526916503905e-05,
      "model_forward_time": 1.7378928661346436,
      "step": 39729
    },
    {
      "epoch": 3.7888526916503905e-05,
      "step": 39729,
      "training_step_time": 6.747629642486572
    },
    {
      "epoch": 3.788948059082031e-05,
      "grad_norm": 0.10617152601480484,
      "learning_rate": 7.0494308364903815e-06,
      "loss": 0.0573,
      "step": 39730
    },
    {
      "epoch": 3.788948059082031e-05,
      "model_forward_time": 1.7078773975372314,
      "step": 39730
    },
    {
      "dynamics_loss_avg": 0.0512737900018692,
      "epoch": 3.788948059082031e-05,
      "step": 39730
    },
    {
      "action_loss_avg": 0.0013810270320391282,
      "epoch": 3.788948059082031e-05,
      "step": 39730
    },
    {
      "epoch": 3.788948059082031e-05,
      "step": 39730,
      "training_step_time": 6.752588748931885
    },
    {
      "epoch": 3.789043426513672e-05,
      "model_forward_time": 1.7318906784057617,
      "step": 39731
    },
    {
      "epoch": 3.789043426513672e-05,
      "step": 39731,
      "training_step_time": 6.758365154266357
    },
    {
      "epoch": 3.7891387939453124e-05,
      "model_forward_time": 1.7339036464691162,
      "step": 39732
    },
    {
      "epoch": 3.7891387939453124e-05,
      "step": 39732,
      "training_step_time": 6.758049726486206
    },
    {
      "epoch": 3.789234161376953e-05,
      "model_forward_time": 1.7345235347747803,
      "step": 39733
    },
    {
      "epoch": 3.789234161376953e-05,
      "step": 39733,
      "training_step_time": 6.742594480514526
    },
    {
      "epoch": 3.789329528808594e-05,
      "model_forward_time": 1.732757806777954,
      "step": 39734
    },
    {
      "epoch": 3.789329528808594e-05,
      "step": 39734,
      "training_step_time": 6.712846279144287
    },
    {
      "epoch": 3.789424896240234e-05,
      "model_forward_time": 2.804377794265747,
      "step": 39735
    },
    {
      "epoch": 3.789424896240234e-05,
      "step": 39735,
      "training_step_time": 7.547026872634888
    },
    {
      "epoch": 3.789520263671875e-05,
      "model_forward_time": 1.8055248260498047,
      "step": 39736
    },
    {
      "epoch": 3.789520263671875e-05,
      "step": 39736,
      "training_step_time": 8.489009857177734
    },
    {
      "epoch": 3.789615631103516e-05,
      "model_forward_time": 1.7286934852600098,
      "step": 39737
    },
    {
      "epoch": 3.789615631103516e-05,
      "step": 39737,
      "training_step_time": 8.52522897720337
    },
    {
      "epoch": 3.789710998535156e-05,
      "model_forward_time": 1.7343976497650146,
      "step": 39738
    },
    {
      "epoch": 3.789710998535156e-05,
      "step": 39738,
      "training_step_time": 6.776569843292236
    },
    {
      "epoch": 3.7898063659667966e-05,
      "model_forward_time": 1.7261707782745361,
      "step": 39739
    },
    {
      "epoch": 3.7898063659667966e-05,
      "step": 39739,
      "training_step_time": 6.767282962799072
    },
    {
      "epoch": 3.789901733398438e-05,
      "grad_norm": 0.10693078488111496,
      "learning_rate": 7.047922533980342e-06,
      "loss": 0.0605,
      "step": 39740
    },
    {
      "epoch": 3.789901733398438e-05,
      "model_forward_time": 1.717233419418335,
      "step": 39740
    },
    {
      "dynamics_loss_avg": 0.04639501944184303,
      "epoch": 3.789901733398438e-05,
      "step": 39740
    },
    {
      "action_loss_avg": 0.002194168069399893,
      "epoch": 3.789901733398438e-05,
      "step": 39740
    },
    {
      "epoch": 3.789901733398438e-05,
      "step": 39740,
      "training_step_time": 6.787990093231201
    },
    {
      "epoch": 3.789997100830078e-05,
      "model_forward_time": 1.7320830821990967,
      "step": 39741
    },
    {
      "epoch": 3.789997100830078e-05,
      "step": 39741,
      "training_step_time": 6.739238977432251
    },
    {
      "epoch": 3.7900924682617185e-05,
      "model_forward_time": 1.7419536113739014,
      "step": 39742
    },
    {
      "epoch": 3.7900924682617185e-05,
      "step": 39742,
      "training_step_time": 6.712177276611328
    },
    {
      "epoch": 3.7901878356933596e-05,
      "model_forward_time": 1.7404038906097412,
      "step": 39743
    },
    {
      "epoch": 3.7901878356933596e-05,
      "step": 39743,
      "training_step_time": 6.666008234024048
    },
    {
      "epoch": 3.790283203125e-05,
      "model_forward_time": 1.730642318725586,
      "step": 39744
    },
    {
      "epoch": 3.790283203125e-05,
      "step": 39744,
      "training_step_time": 6.696826934814453
    },
    {
      "epoch": 3.7903785705566404e-05,
      "model_forward_time": 2.8409628868103027,
      "step": 39745
    },
    {
      "epoch": 3.7903785705566404e-05,
      "step": 39745,
      "training_step_time": 7.55935001373291
    },
    {
      "epoch": 3.7904739379882815e-05,
      "model_forward_time": 1.733912706375122,
      "step": 39746
    },
    {
      "epoch": 3.7904739379882815e-05,
      "step": 39746,
      "training_step_time": 7.651911497116089
    },
    {
      "epoch": 3.790569305419922e-05,
      "model_forward_time": 1.74491548538208,
      "step": 39747
    },
    {
      "epoch": 3.790569305419922e-05,
      "step": 39747,
      "training_step_time": 7.181613922119141
    },
    {
      "epoch": 3.790664672851562e-05,
      "model_forward_time": 1.7367141246795654,
      "step": 39748
    },
    {
      "epoch": 3.790664672851562e-05,
      "step": 39748,
      "training_step_time": 6.71533203125
    },
    {
      "epoch": 3.7907600402832033e-05,
      "model_forward_time": 1.737091302871704,
      "step": 39749
    },
    {
      "epoch": 3.7907600402832033e-05,
      "step": 39749,
      "training_step_time": 6.726498365402222
    },
    {
      "epoch": 3.790855407714844e-05,
      "grad_norm": 0.09033642709255219,
      "learning_rate": 7.046414007512572e-06,
      "loss": 0.058,
      "step": 39750
    },
    {
      "epoch": 3.790855407714844e-05,
      "model_forward_time": 1.7154417037963867,
      "step": 39750
    },
    {
      "dynamics_loss_avg": 0.058557382132858035,
      "epoch": 3.790855407714844e-05,
      "step": 39750
    },
    {
      "action_loss_avg": 0.0024336880014743656,
      "epoch": 3.790855407714844e-05,
      "step": 39750
    },
    {
      "epoch": 3.790855407714844e-05,
      "step": 39750,
      "training_step_time": 6.762655258178711
    },
    {
      "epoch": 3.790950775146484e-05,
      "model_forward_time": 1.7314565181732178,
      "step": 39751
    },
    {
      "epoch": 3.790950775146484e-05,
      "step": 39751,
      "training_step_time": 6.689734220504761
    },
    {
      "epoch": 3.791046142578125e-05,
      "model_forward_time": 1.7374277114868164,
      "step": 39752
    },
    {
      "epoch": 3.791046142578125e-05,
      "step": 39752,
      "training_step_time": 6.750797748565674
    },
    {
      "epoch": 3.7911415100097656e-05,
      "model_forward_time": 1.7542986869812012,
      "step": 39753
    },
    {
      "epoch": 3.7911415100097656e-05,
      "step": 39753,
      "training_step_time": 6.707685232162476
    },
    {
      "epoch": 3.791236877441406e-05,
      "model_forward_time": 2.8611834049224854,
      "step": 39754
    },
    {
      "epoch": 3.791236877441406e-05,
      "step": 39754,
      "training_step_time": 7.562019348144531
    },
    {
      "epoch": 3.791332244873047e-05,
      "model_forward_time": 1.7262983322143555,
      "step": 39755
    },
    {
      "epoch": 3.791332244873047e-05,
      "step": 39755,
      "training_step_time": 7.678238868713379
    },
    {
      "epoch": 3.7914276123046875e-05,
      "model_forward_time": 1.725956916809082,
      "step": 39756
    },
    {
      "epoch": 3.7914276123046875e-05,
      "step": 39756,
      "training_step_time": 7.531426191329956
    },
    {
      "epoch": 3.791522979736328e-05,
      "model_forward_time": 1.741173505783081,
      "step": 39757
    },
    {
      "epoch": 3.791522979736328e-05,
      "step": 39757,
      "training_step_time": 6.742735385894775
    },
    {
      "epoch": 3.791618347167969e-05,
      "model_forward_time": 1.7198233604431152,
      "step": 39758
    },
    {
      "epoch": 3.791618347167969e-05,
      "step": 39758,
      "training_step_time": 6.6887195110321045
    },
    {
      "epoch": 3.7917137145996094e-05,
      "model_forward_time": 1.7361581325531006,
      "step": 39759
    },
    {
      "epoch": 3.7917137145996094e-05,
      "step": 39759,
      "training_step_time": 6.73514461517334
    },
    {
      "epoch": 3.79180908203125e-05,
      "grad_norm": 0.13566221296787262,
      "learning_rate": 7.044905257252042e-06,
      "loss": 0.0625,
      "step": 39760
    },
    {
      "epoch": 3.79180908203125e-05,
      "model_forward_time": 1.698974370956421,
      "step": 39760
    },
    {
      "dynamics_loss_avg": 0.059208940342068674,
      "epoch": 3.79180908203125e-05,
      "step": 39760
    },
    {
      "action_loss_avg": 0.0017823370173573494,
      "epoch": 3.79180908203125e-05,
      "step": 39760
    },
    {
      "epoch": 3.79180908203125e-05,
      "step": 39760,
      "training_step_time": 6.7992165088653564
    },
    {
      "epoch": 3.791904449462891e-05,
      "model_forward_time": 1.7249958515167236,
      "step": 39761
    },
    {
      "epoch": 3.791904449462891e-05,
      "step": 39761,
      "training_step_time": 6.705106019973755
    },
    {
      "epoch": 3.791999816894531e-05,
      "model_forward_time": 1.73110032081604,
      "step": 39762
    },
    {
      "epoch": 3.791999816894531e-05,
      "step": 39762,
      "training_step_time": 6.696519136428833
    },
    {
      "epoch": 3.792095184326172e-05,
      "model_forward_time": 2.854069948196411,
      "step": 39763
    },
    {
      "epoch": 3.792095184326172e-05,
      "step": 39763,
      "training_step_time": 7.546481132507324
    },
    {
      "epoch": 3.792190551757813e-05,
      "model_forward_time": 1.725235939025879,
      "step": 39764
    },
    {
      "epoch": 3.792190551757813e-05,
      "step": 39764,
      "training_step_time": 8.638367414474487
    },
    {
      "epoch": 3.792285919189453e-05,
      "model_forward_time": 1.714996337890625,
      "step": 39765
    },
    {
      "epoch": 3.792285919189453e-05,
      "step": 39765,
      "training_step_time": 8.429992198944092
    },
    {
      "epoch": 3.7923812866210936e-05,
      "model_forward_time": 1.7495975494384766,
      "step": 39766
    },
    {
      "epoch": 3.7923812866210936e-05,
      "step": 39766,
      "training_step_time": 6.745344877243042
    },
    {
      "epoch": 3.792476654052735e-05,
      "model_forward_time": 1.7256255149841309,
      "step": 39767
    },
    {
      "epoch": 3.792476654052735e-05,
      "step": 39767,
      "training_step_time": 6.7084338665008545
    },
    {
      "epoch": 3.792572021484375e-05,
      "model_forward_time": 1.7207794189453125,
      "step": 39768
    },
    {
      "epoch": 3.792572021484375e-05,
      "step": 39768,
      "training_step_time": 6.742361307144165
    },
    {
      "epoch": 3.7926673889160155e-05,
      "model_forward_time": 1.7235121726989746,
      "step": 39769
    },
    {
      "epoch": 3.7926673889160155e-05,
      "step": 39769,
      "training_step_time": 6.675396680831909
    },
    {
      "epoch": 3.7927627563476565e-05,
      "grad_norm": 0.08956734836101532,
      "learning_rate": 7.043396283363749e-06,
      "loss": 0.0629,
      "step": 39770
    },
    {
      "epoch": 3.7927627563476565e-05,
      "model_forward_time": 1.7036712169647217,
      "step": 39770
    },
    {
      "dynamics_loss_avg": 0.0517429968342185,
      "epoch": 3.7927627563476565e-05,
      "step": 39770
    },
    {
      "action_loss_avg": 0.002363700699061155,
      "epoch": 3.7927627563476565e-05,
      "step": 39770
    },
    {
      "epoch": 3.7927627563476565e-05,
      "step": 39770,
      "training_step_time": 6.751190662384033
    },
    {
      "epoch": 3.792858123779297e-05,
      "model_forward_time": 1.730431079864502,
      "step": 39771
    },
    {
      "epoch": 3.792858123779297e-05,
      "step": 39771,
      "training_step_time": 6.77355694770813
    },
    {
      "epoch": 3.7929534912109374e-05,
      "model_forward_time": 2.916322708129883,
      "step": 39772
    },
    {
      "epoch": 3.7929534912109374e-05,
      "step": 39772,
      "training_step_time": 8.695650815963745
    },
    {
      "epoch": 3.7930488586425784e-05,
      "model_forward_time": 1.7225258350372314,
      "step": 39773
    },
    {
      "epoch": 3.7930488586425784e-05,
      "step": 39773,
      "training_step_time": 8.549641370773315
    },
    {
      "epoch": 3.793144226074219e-05,
      "model_forward_time": 1.7119858264923096,
      "step": 39774
    },
    {
      "epoch": 3.793144226074219e-05,
      "step": 39774,
      "training_step_time": 7.68765664100647
    },
    {
      "epoch": 3.793239593505859e-05,
      "model_forward_time": 1.723031759262085,
      "step": 39775
    },
    {
      "epoch": 3.793239593505859e-05,
      "step": 39775,
      "training_step_time": 7.240894556045532
    },
    {
      "epoch": 3.7933349609375e-05,
      "model_forward_time": 1.7309634685516357,
      "step": 39776
    },
    {
      "epoch": 3.7933349609375e-05,
      "step": 39776,
      "training_step_time": 6.792981386184692
    },
    {
      "epoch": 3.793430328369141e-05,
      "model_forward_time": 1.7245023250579834,
      "step": 39777
    },
    {
      "epoch": 3.793430328369141e-05,
      "step": 39777,
      "training_step_time": 6.734370470046997
    },
    {
      "epoch": 3.793525695800781e-05,
      "model_forward_time": 1.728713035583496,
      "step": 39778
    },
    {
      "epoch": 3.793525695800781e-05,
      "step": 39778,
      "training_step_time": 6.707085371017456
    },
    {
      "epoch": 3.793621063232422e-05,
      "model_forward_time": 1.7282612323760986,
      "step": 39779
    },
    {
      "epoch": 3.793621063232422e-05,
      "step": 39779,
      "training_step_time": 6.734156131744385
    },
    {
      "epoch": 3.7937164306640626e-05,
      "grad_norm": 0.10217030346393585,
      "learning_rate": 7.041887086012711e-06,
      "loss": 0.0613,
      "step": 39780
    },
    {
      "epoch": 3.7937164306640626e-05,
      "model_forward_time": 1.7091214656829834,
      "step": 39780
    },
    {
      "dynamics_loss_avg": 0.05572723224759102,
      "epoch": 3.7937164306640626e-05,
      "step": 39780
    },
    {
      "action_loss_avg": 0.002245115756522864,
      "epoch": 3.7937164306640626e-05,
      "step": 39780
    },
    {
      "epoch": 3.7937164306640626e-05,
      "step": 39780,
      "training_step_time": 6.809024810791016
    },
    {
      "epoch": 3.793811798095703e-05,
      "model_forward_time": 2.8088326454162598,
      "step": 39781
    },
    {
      "epoch": 3.793811798095703e-05,
      "step": 39781,
      "training_step_time": 7.553321838378906
    },
    {
      "epoch": 3.7939071655273434e-05,
      "model_forward_time": 1.7932395935058594,
      "step": 39782
    },
    {
      "epoch": 3.7939071655273434e-05,
      "step": 39782,
      "training_step_time": 11.258034229278564
    },
    {
      "epoch": 3.7940025329589845e-05,
      "model_forward_time": 1.7225558757781982,
      "step": 39783
    },
    {
      "epoch": 3.7940025329589845e-05,
      "step": 39783,
      "training_step_time": 7.7229485511779785
    },
    {
      "epoch": 3.794097900390625e-05,
      "model_forward_time": 1.731128454208374,
      "step": 39784
    },
    {
      "epoch": 3.794097900390625e-05,
      "step": 39784,
      "training_step_time": 7.2872045040130615
    },
    {
      "epoch": 3.794193267822265e-05,
      "model_forward_time": 1.727905035018921,
      "step": 39785
    },
    {
      "epoch": 3.794193267822265e-05,
      "step": 39785,
      "training_step_time": 6.770430564880371
    },
    {
      "epoch": 3.7942886352539064e-05,
      "model_forward_time": 1.720747709274292,
      "step": 39786
    },
    {
      "epoch": 3.7942886352539064e-05,
      "step": 39786,
      "training_step_time": 6.797784805297852
    },
    {
      "epoch": 3.794384002685547e-05,
      "model_forward_time": 1.7329001426696777,
      "step": 39787
    },
    {
      "epoch": 3.794384002685547e-05,
      "step": 39787,
      "training_step_time": 6.8154296875
    },
    {
      "epoch": 3.794479370117187e-05,
      "model_forward_time": 1.7246968746185303,
      "step": 39788
    },
    {
      "epoch": 3.794479370117187e-05,
      "step": 39788,
      "training_step_time": 6.794150114059448
    },
    {
      "epoch": 3.794574737548828e-05,
      "model_forward_time": 1.7500345706939697,
      "step": 39789
    },
    {
      "epoch": 3.794574737548828e-05,
      "step": 39789,
      "training_step_time": 6.790072202682495
    },
    {
      "epoch": 3.794670104980469e-05,
      "grad_norm": 0.10300572961568832,
      "learning_rate": 7.0403776653639685e-06,
      "loss": 0.0601,
      "step": 39790
    },
    {
      "epoch": 3.794670104980469e-05,
      "model_forward_time": 1.7191689014434814,
      "step": 39790
    },
    {
      "dynamics_loss_avg": 0.06746217422187328,
      "epoch": 3.794670104980469e-05,
      "step": 39790
    },
    {
      "action_loss_avg": 0.0021141109406016766,
      "epoch": 3.794670104980469e-05,
      "step": 39790
    },
    {
      "epoch": 3.794670104980469e-05,
      "step": 39790,
      "training_step_time": 7.291022777557373
    },
    {
      "epoch": 3.794765472412109e-05,
      "model_forward_time": 1.7931079864501953,
      "step": 39791
    },
    {
      "epoch": 3.794765472412109e-05,
      "step": 39791,
      "training_step_time": 10.199997186660767
    },
    {
      "epoch": 3.79486083984375e-05,
      "model_forward_time": 1.7258632183074951,
      "step": 39792
    },
    {
      "epoch": 3.79486083984375e-05,
      "step": 39792,
      "training_step_time": 8.68848729133606
    },
    {
      "epoch": 3.7949562072753906e-05,
      "model_forward_time": 1.7202811241149902,
      "step": 39793
    },
    {
      "epoch": 3.7949562072753906e-05,
      "step": 39793,
      "training_step_time": 7.498259544372559
    },
    {
      "epoch": 3.795051574707031e-05,
      "model_forward_time": 1.7276666164398193,
      "step": 39794
    },
    {
      "epoch": 3.795051574707031e-05,
      "step": 39794,
      "training_step_time": 6.77899432182312
    },
    {
      "epoch": 3.795146942138672e-05,
      "model_forward_time": 1.7353436946868896,
      "step": 39795
    },
    {
      "epoch": 3.795146942138672e-05,
      "step": 39795,
      "training_step_time": 6.809301376342773
    },
    {
      "epoch": 3.7952423095703124e-05,
      "model_forward_time": 1.7298047542572021,
      "step": 39796
    },
    {
      "epoch": 3.7952423095703124e-05,
      "step": 39796,
      "training_step_time": 6.736947774887085
    },
    {
      "epoch": 3.795337677001953e-05,
      "model_forward_time": 1.7239303588867188,
      "step": 39797
    },
    {
      "epoch": 3.795337677001953e-05,
      "step": 39797,
      "training_step_time": 6.773985385894775
    },
    {
      "epoch": 3.795433044433594e-05,
      "model_forward_time": 1.7209017276763916,
      "step": 39798
    },
    {
      "epoch": 3.795433044433594e-05,
      "step": 39798,
      "training_step_time": 6.723567962646484
    },
    {
      "epoch": 3.795528411865234e-05,
      "model_forward_time": 1.7218410968780518,
      "step": 39799
    },
    {
      "epoch": 3.795528411865234e-05,
      "step": 39799,
      "training_step_time": 6.793405771255493
    },
    {
      "epoch": 3.795623779296875e-05,
      "grad_norm": 0.13058917224407196,
      "learning_rate": 7.038868021582595e-06,
      "loss": 0.0625,
      "step": 39800
    },
    {
      "epoch": 3.795623779296875e-05,
      "model_forward_time": 2.9377810955047607,
      "step": 39800
    },
    {
      "dynamics_loss_avg": 0.07752511268481613,
      "epoch": 3.795623779296875e-05,
      "step": 39800
    },
    {
      "action_loss_avg": 0.001834003342082724,
      "epoch": 3.795623779296875e-05,
      "step": 39800
    },
    {
      "epoch": 3.795623779296875e-05,
      "step": 39800,
      "training_step_time": 7.725212335586548
    },
    {
      "epoch": 3.795719146728516e-05,
      "model_forward_time": 1.7245492935180664,
      "step": 39801
    },
    {
      "epoch": 3.795719146728516e-05,
      "step": 39801,
      "training_step_time": 8.962326765060425
    },
    {
      "epoch": 3.795814514160156e-05,
      "model_forward_time": 1.7263665199279785,
      "step": 39802
    },
    {
      "epoch": 3.795814514160156e-05,
      "step": 39802,
      "training_step_time": 7.564520359039307
    },
    {
      "epoch": 3.7959098815917966e-05,
      "model_forward_time": 1.7308921813964844,
      "step": 39803
    },
    {
      "epoch": 3.7959098815917966e-05,
      "step": 39803,
      "training_step_time": 7.212480306625366
    },
    {
      "epoch": 3.796005249023438e-05,
      "model_forward_time": 1.7287793159484863,
      "step": 39804
    },
    {
      "epoch": 3.796005249023438e-05,
      "step": 39804,
      "training_step_time": 6.755462408065796
    },
    {
      "epoch": 3.796100616455078e-05,
      "model_forward_time": 1.7268378734588623,
      "step": 39805
    },
    {
      "epoch": 3.796100616455078e-05,
      "step": 39805,
      "training_step_time": 6.746908903121948
    },
    {
      "epoch": 3.7961959838867185e-05,
      "model_forward_time": 1.0186359882354736,
      "step": 39806
    },
    {
      "epoch": 3.7961959838867185e-05,
      "step": 39806,
      "training_step_time": 6.799095153808594
    },
    {
      "epoch": 3.7962913513183596e-05,
      "model_forward_time": 1.714177131652832,
      "step": 39807
    },
    {
      "epoch": 3.7962913513183596e-05,
      "step": 39807,
      "training_step_time": 6.811271667480469
    },
    {
      "epoch": 3.79638671875e-05,
      "model_forward_time": 1.7359509468078613,
      "step": 39808
    },
    {
      "epoch": 3.79638671875e-05,
      "step": 39808,
      "training_step_time": 6.754359245300293
    },
    {
      "epoch": 3.7964820861816404e-05,
      "model_forward_time": 2.1942551136016846,
      "step": 39809
    },
    {
      "epoch": 3.7964820861816404e-05,
      "step": 39809,
      "training_step_time": 7.6059699058532715
    },
    {
      "epoch": 3.7965774536132815e-05,
      "grad_norm": 0.1838400959968567,
      "learning_rate": 7.037358154833679e-06,
      "loss": 0.0632,
      "step": 39810
    },
    {
      "epoch": 3.7965774536132815e-05,
      "model_forward_time": 1.6999554634094238,
      "step": 39810
    },
    {
      "dynamics_loss_avg": 0.04661261308938265,
      "epoch": 3.7965774536132815e-05,
      "step": 39810
    },
    {
      "action_loss_avg": 0.0015596476674545555,
      "epoch": 3.7965774536132815e-05,
      "step": 39810
    },
    {
      "epoch": 3.7965774536132815e-05,
      "step": 39810,
      "training_step_time": 9.787842273712158
    },
    {
      "epoch": 3.796672821044922e-05,
      "model_forward_time": 1.7257914543151855,
      "step": 39811
    },
    {
      "epoch": 3.796672821044922e-05,
      "step": 39811,
      "training_step_time": 7.604531288146973
    },
    {
      "epoch": 3.796768188476562e-05,
      "model_forward_time": 1.7177481651306152,
      "step": 39812
    },
    {
      "epoch": 3.796768188476562e-05,
      "step": 39812,
      "training_step_time": 7.197914123535156
    },
    {
      "epoch": 3.7968635559082034e-05,
      "model_forward_time": 1.7196526527404785,
      "step": 39813
    },
    {
      "epoch": 3.7968635559082034e-05,
      "step": 39813,
      "training_step_time": 6.8230016231536865
    },
    {
      "epoch": 3.796958923339844e-05,
      "model_forward_time": 1.7182698249816895,
      "step": 39814
    },
    {
      "epoch": 3.796958923339844e-05,
      "step": 39814,
      "training_step_time": 6.833666801452637
    },
    {
      "epoch": 3.797054290771484e-05,
      "model_forward_time": 1.7308127880096436,
      "step": 39815
    },
    {
      "epoch": 3.797054290771484e-05,
      "step": 39815,
      "training_step_time": 6.781733274459839
    },
    {
      "epoch": 3.797149658203125e-05,
      "model_forward_time": 1.7276363372802734,
      "step": 39816
    },
    {
      "epoch": 3.797149658203125e-05,
      "step": 39816,
      "training_step_time": 6.837952136993408
    },
    {
      "epoch": 3.7972450256347656e-05,
      "model_forward_time": 1.7175266742706299,
      "step": 39817
    },
    {
      "epoch": 3.7972450256347656e-05,
      "step": 39817,
      "training_step_time": 6.779642105102539
    },
    {
      "epoch": 3.797340393066406e-05,
      "model_forward_time": 1.733896255493164,
      "step": 39818
    },
    {
      "epoch": 3.797340393066406e-05,
      "step": 39818,
      "training_step_time": 8.126142263412476
    },
    {
      "epoch": 3.797435760498047e-05,
      "model_forward_time": 1.726914644241333,
      "step": 39819
    },
    {
      "epoch": 3.797435760498047e-05,
      "step": 39819,
      "training_step_time": 11.053293228149414
    },
    {
      "epoch": 3.7975311279296875e-05,
      "grad_norm": 0.13791905343532562,
      "learning_rate": 7.035848065282339e-06,
      "loss": 0.059,
      "step": 39820
    },
    {
      "epoch": 3.7975311279296875e-05,
      "model_forward_time": 1.7174580097198486,
      "step": 39820
    },
    {
      "dynamics_loss_avg": 0.060840258933603765,
      "epoch": 3.7975311279296875e-05,
      "step": 39820
    },
    {
      "action_loss_avg": 0.002166521066101268,
      "epoch": 3.7975311279296875e-05,
      "step": 39820
    },
    {
      "epoch": 3.7975311279296875e-05,
      "step": 39820,
      "training_step_time": 7.806907892227173
    },
    {
      "epoch": 3.797626495361328e-05,
      "model_forward_time": 1.725334644317627,
      "step": 39821
    },
    {
      "epoch": 3.797626495361328e-05,
      "step": 39821,
      "training_step_time": 7.258338928222656
    },
    {
      "epoch": 3.797721862792969e-05,
      "model_forward_time": 1.34893798828125,
      "step": 39822
    },
    {
      "epoch": 3.797721862792969e-05,
      "step": 39822,
      "training_step_time": 7.190953731536865
    },
    {
      "epoch": 3.7978172302246094e-05,
      "model_forward_time": 1.7230322360992432,
      "step": 39823
    },
    {
      "epoch": 3.7978172302246094e-05,
      "step": 39823,
      "training_step_time": 6.838925123214722
    },
    {
      "epoch": 3.79791259765625e-05,
      "model_forward_time": 1.7226336002349854,
      "step": 39824
    },
    {
      "epoch": 3.79791259765625e-05,
      "step": 39824,
      "training_step_time": 6.769679546356201
    },
    {
      "epoch": 3.798007965087891e-05,
      "model_forward_time": 1.7262229919433594,
      "step": 39825
    },
    {
      "epoch": 3.798007965087891e-05,
      "step": 39825,
      "training_step_time": 6.791013717651367
    },
    {
      "epoch": 3.798103332519531e-05,
      "model_forward_time": 1.0168447494506836,
      "step": 39826
    },
    {
      "epoch": 3.798103332519531e-05,
      "step": 39826,
      "training_step_time": 6.729449987411499
    },
    {
      "epoch": 3.798198699951172e-05,
      "model_forward_time": 1.3496322631835938,
      "step": 39827
    },
    {
      "epoch": 3.798198699951172e-05,
      "step": 39827,
      "training_step_time": 7.333736419677734
    },
    {
      "epoch": 3.798294067382813e-05,
      "model_forward_time": 2.9106554985046387,
      "step": 39828
    },
    {
      "epoch": 3.798294067382813e-05,
      "step": 39828,
      "training_step_time": 8.492964029312134
    },
    {
      "epoch": 3.798389434814453e-05,
      "model_forward_time": 1.7306740283966064,
      "step": 39829
    },
    {
      "epoch": 3.798389434814453e-05,
      "step": 39829,
      "training_step_time": 8.654571056365967
    },
    {
      "epoch": 3.7984848022460936e-05,
      "grad_norm": 0.14865578711032867,
      "learning_rate": 7.034337753093714e-06,
      "loss": 0.0583,
      "step": 39830
    },
    {
      "epoch": 3.7984848022460936e-05,
      "model_forward_time": 1.7058727741241455,
      "step": 39830
    },
    {
      "dynamics_loss_avg": 0.05912130922079086,
      "epoch": 3.7984848022460936e-05,
      "step": 39830
    },
    {
      "action_loss_avg": 0.003458691691048443,
      "epoch": 3.7984848022460936e-05,
      "step": 39830
    },
    {
      "epoch": 3.7984848022460936e-05,
      "step": 39830,
      "training_step_time": 7.427505254745483
    },
    {
      "epoch": 3.798580169677735e-05,
      "model_forward_time": 1.7152490615844727,
      "step": 39831
    },
    {
      "epoch": 3.798580169677735e-05,
      "step": 39831,
      "training_step_time": 7.153355360031128
    },
    {
      "epoch": 3.798675537109375e-05,
      "model_forward_time": 1.7199795246124268,
      "step": 39832
    },
    {
      "epoch": 3.798675537109375e-05,
      "step": 39832,
      "training_step_time": 6.7380945682525635
    },
    {
      "epoch": 3.7987709045410155e-05,
      "model_forward_time": 1.7426121234893799,
      "step": 39833
    },
    {
      "epoch": 3.7987709045410155e-05,
      "step": 39833,
      "training_step_time": 6.749650001525879
    },
    {
      "epoch": 3.7988662719726566e-05,
      "model_forward_time": 1.7190186977386475,
      "step": 39834
    },
    {
      "epoch": 3.7988662719726566e-05,
      "step": 39834,
      "training_step_time": 6.767743349075317
    },
    {
      "epoch": 3.798961639404297e-05,
      "model_forward_time": 1.781397819519043,
      "step": 39835
    },
    {
      "epoch": 3.798961639404297e-05,
      "step": 39835,
      "training_step_time": 6.711044549942017
    },
    {
      "epoch": 3.7990570068359374e-05,
      "model_forward_time": 1.715789556503296,
      "step": 39836
    },
    {
      "epoch": 3.7990570068359374e-05,
      "step": 39836,
      "training_step_time": 7.326772689819336
    },
    {
      "epoch": 3.7991523742675784e-05,
      "model_forward_time": 2.8822576999664307,
      "step": 39837
    },
    {
      "epoch": 3.7991523742675784e-05,
      "step": 39837,
      "training_step_time": 11.085355758666992
    },
    {
      "epoch": 3.799247741699219e-05,
      "model_forward_time": 1.7313752174377441,
      "step": 39838
    },
    {
      "epoch": 3.799247741699219e-05,
      "step": 39838,
      "training_step_time": 10.524771451950073
    },
    {
      "epoch": 3.799343109130859e-05,
      "model_forward_time": 1.738853931427002,
      "step": 39839
    },
    {
      "epoch": 3.799343109130859e-05,
      "step": 39839,
      "training_step_time": 7.536344051361084
    },
    {
      "epoch": 3.7994384765625e-05,
      "grad_norm": 0.16702423989772797,
      "learning_rate": 7.032827218432971e-06,
      "loss": 0.0578,
      "step": 39840
    },
    {
      "epoch": 3.7994384765625e-05,
      "model_forward_time": 1.816892385482788,
      "step": 39840
    },
    {
      "dynamics_loss_avg": 0.06671685837209225,
      "epoch": 3.7994384765625e-05,
      "step": 39840
    },
    {
      "action_loss_avg": 0.001733371411683038,
      "epoch": 3.7994384765625e-05,
      "step": 39840
    },
    {
      "epoch": 3.7994384765625e-05,
      "step": 39840,
      "training_step_time": 7.209827184677124
    },
    {
      "epoch": 3.799533843994141e-05,
      "model_forward_time": 1.7339797019958496,
      "step": 39841
    },
    {
      "epoch": 3.799533843994141e-05,
      "step": 39841,
      "training_step_time": 6.723284482955933
    },
    {
      "epoch": 3.799629211425781e-05,
      "model_forward_time": 1.7340514659881592,
      "step": 39842
    },
    {
      "epoch": 3.799629211425781e-05,
      "step": 39842,
      "training_step_time": 6.734609365463257
    },
    {
      "epoch": 3.799724578857422e-05,
      "model_forward_time": 1.7355659008026123,
      "step": 39843
    },
    {
      "epoch": 3.799724578857422e-05,
      "step": 39843,
      "training_step_time": 6.7100584506988525
    },
    {
      "epoch": 3.7998199462890626e-05,
      "model_forward_time": 1.741295576095581,
      "step": 39844
    },
    {
      "epoch": 3.7998199462890626e-05,
      "step": 39844,
      "training_step_time": 6.697195768356323
    },
    {
      "epoch": 3.799915313720703e-05,
      "model_forward_time": 1.7347400188446045,
      "step": 39845
    },
    {
      "epoch": 3.799915313720703e-05,
      "step": 39845,
      "training_step_time": 7.363510370254517
    },
    {
      "epoch": 3.8000106811523434e-05,
      "model_forward_time": 1.7451176643371582,
      "step": 39846
    },
    {
      "epoch": 3.8000106811523434e-05,
      "step": 39846,
      "training_step_time": 9.889904260635376
    },
    {
      "epoch": 3.8001060485839845e-05,
      "model_forward_time": 2.874321222305298,
      "step": 39847
    },
    {
      "epoch": 3.8001060485839845e-05,
      "step": 39847,
      "training_step_time": 8.185821294784546
    },
    {
      "epoch": 3.800201416015625e-05,
      "model_forward_time": 1.7410457134246826,
      "step": 39848
    },
    {
      "epoch": 3.800201416015625e-05,
      "step": 39848,
      "training_step_time": 8.376168966293335
    },
    {
      "epoch": 3.800296783447265e-05,
      "model_forward_time": 1.7351758480072021,
      "step": 39849
    },
    {
      "epoch": 3.800296783447265e-05,
      "step": 39849,
      "training_step_time": 7.209893703460693
    },
    {
      "epoch": 3.8003921508789064e-05,
      "grad_norm": 0.1849452406167984,
      "learning_rate": 7.031316461465302e-06,
      "loss": 0.0562,
      "step": 39850
    },
    {
      "epoch": 3.8003921508789064e-05,
      "model_forward_time": 1.748258352279663,
      "step": 39850
    },
    {
      "dynamics_loss_avg": 0.04410859998315573,
      "epoch": 3.8003921508789064e-05,
      "step": 39850
    },
    {
      "action_loss_avg": 0.0010575607360806317,
      "epoch": 3.8003921508789064e-05,
      "step": 39850
    },
    {
      "epoch": 3.8003921508789064e-05,
      "step": 39850,
      "training_step_time": 6.729123115539551
    },
    {
      "epoch": 3.800487518310547e-05,
      "model_forward_time": 1.7476372718811035,
      "step": 39851
    },
    {
      "epoch": 3.800487518310547e-05,
      "step": 39851,
      "training_step_time": 6.686144590377808
    },
    {
      "epoch": 3.800582885742187e-05,
      "model_forward_time": 1.732248306274414,
      "step": 39852
    },
    {
      "epoch": 3.800582885742187e-05,
      "step": 39852,
      "training_step_time": 6.710298299789429
    },
    {
      "epoch": 3.800678253173828e-05,
      "model_forward_time": 1.7396559715270996,
      "step": 39853
    },
    {
      "epoch": 3.800678253173828e-05,
      "step": 39853,
      "training_step_time": 6.708596229553223
    },
    {
      "epoch": 3.800773620605469e-05,
      "model_forward_time": 1.746415138244629,
      "step": 39854
    },
    {
      "epoch": 3.800773620605469e-05,
      "step": 39854,
      "training_step_time": 7.259296417236328
    },
    {
      "epoch": 3.800868988037109e-05,
      "model_forward_time": 1.7408833503723145,
      "step": 39855
    },
    {
      "epoch": 3.800868988037109e-05,
      "step": 39855,
      "training_step_time": 7.670283079147339
    },
    {
      "epoch": 3.80096435546875e-05,
      "model_forward_time": 2.859776496887207,
      "step": 39856
    },
    {
      "epoch": 3.80096435546875e-05,
      "step": 39856,
      "training_step_time": 8.399955749511719
    },
    {
      "epoch": 3.8010597229003906e-05,
      "model_forward_time": 1.7277369499206543,
      "step": 39857
    },
    {
      "epoch": 3.8010597229003906e-05,
      "step": 39857,
      "training_step_time": 8.430150508880615
    },
    {
      "epoch": 3.801155090332031e-05,
      "model_forward_time": 1.7298810482025146,
      "step": 39858
    },
    {
      "epoch": 3.801155090332031e-05,
      "step": 39858,
      "training_step_time": 7.458149194717407
    },
    {
      "epoch": 3.801250457763672e-05,
      "model_forward_time": 1.0377001762390137,
      "step": 39859
    },
    {
      "epoch": 3.801250457763672e-05,
      "step": 39859,
      "training_step_time": 6.794381618499756
    },
    {
      "epoch": 3.8013458251953124e-05,
      "grad_norm": 0.1653207540512085,
      "learning_rate": 7.029805482355916e-06,
      "loss": 0.0626,
      "step": 39860
    },
    {
      "epoch": 3.8013458251953124e-05,
      "model_forward_time": 1.7205212116241455,
      "step": 39860
    },
    {
      "dynamics_loss_avg": 0.060313375666737556,
      "epoch": 3.8013458251953124e-05,
      "step": 39860
    },
    {
      "action_loss_avg": 0.0018473750678822398,
      "epoch": 3.8013458251953124e-05,
      "step": 39860
    },
    {
      "epoch": 3.8013458251953124e-05,
      "step": 39860,
      "training_step_time": 6.812396764755249
    },
    {
      "epoch": 3.801441192626953e-05,
      "model_forward_time": 1.7371392250061035,
      "step": 39861
    },
    {
      "epoch": 3.801441192626953e-05,
      "step": 39861,
      "training_step_time": 6.785851240158081
    },
    {
      "epoch": 3.801536560058594e-05,
      "model_forward_time": 1.733689785003662,
      "step": 39862
    },
    {
      "epoch": 3.801536560058594e-05,
      "step": 39862,
      "training_step_time": 6.776365041732788
    },
    {
      "epoch": 3.801631927490234e-05,
      "model_forward_time": 1.7328314781188965,
      "step": 39863
    },
    {
      "epoch": 3.801631927490234e-05,
      "step": 39863,
      "training_step_time": 7.296459197998047
    },
    {
      "epoch": 3.801727294921875e-05,
      "model_forward_time": 1.7403862476348877,
      "step": 39864
    },
    {
      "epoch": 3.801727294921875e-05,
      "step": 39864,
      "training_step_time": 7.640712022781372
    },
    {
      "epoch": 3.801822662353516e-05,
      "model_forward_time": 2.902123212814331,
      "step": 39865
    },
    {
      "epoch": 3.801822662353516e-05,
      "step": 39865,
      "training_step_time": 9.93819808959961
    },
    {
      "epoch": 3.801918029785156e-05,
      "model_forward_time": 1.7357919216156006,
      "step": 39866
    },
    {
      "epoch": 3.801918029785156e-05,
      "step": 39866,
      "training_step_time": 8.981380939483643
    },
    {
      "epoch": 3.8020133972167966e-05,
      "model_forward_time": 1.7223777770996094,
      "step": 39867
    },
    {
      "epoch": 3.8020133972167966e-05,
      "step": 39867,
      "training_step_time": 8.476568698883057
    },
    {
      "epoch": 3.802108764648438e-05,
      "model_forward_time": 1.7267847061157227,
      "step": 39868
    },
    {
      "epoch": 3.802108764648438e-05,
      "step": 39868,
      "training_step_time": 7.295927047729492
    },
    {
      "epoch": 3.802204132080078e-05,
      "model_forward_time": 1.7387127876281738,
      "step": 39869
    },
    {
      "epoch": 3.802204132080078e-05,
      "step": 39869,
      "training_step_time": 6.845770835876465
    },
    {
      "epoch": 3.8022994995117185e-05,
      "grad_norm": 0.23806720972061157,
      "learning_rate": 7.028294281270056e-06,
      "loss": 0.0554,
      "step": 39870
    },
    {
      "epoch": 3.8022994995117185e-05,
      "model_forward_time": 1.7354941368103027,
      "step": 39870
    },
    {
      "dynamics_loss_avg": 0.045158947352319956,
      "epoch": 3.8022994995117185e-05,
      "step": 39870
    },
    {
      "action_loss_avg": 0.0018497542128898203,
      "epoch": 3.8022994995117185e-05,
      "step": 39870
    },
    {
      "epoch": 3.8022994995117185e-05,
      "step": 39870,
      "training_step_time": 6.885495185852051
    },
    {
      "epoch": 3.8023948669433596e-05,
      "model_forward_time": 1.7336716651916504,
      "step": 39871
    },
    {
      "epoch": 3.8023948669433596e-05,
      "step": 39871,
      "training_step_time": 6.843561887741089
    },
    {
      "epoch": 3.802490234375e-05,
      "model_forward_time": 1.7374486923217773,
      "step": 39872
    },
    {
      "epoch": 3.802490234375e-05,
      "step": 39872,
      "training_step_time": 7.406849384307861
    },
    {
      "epoch": 3.8025856018066404e-05,
      "model_forward_time": 1.7305963039398193,
      "step": 39873
    },
    {
      "epoch": 3.8025856018066404e-05,
      "step": 39873,
      "training_step_time": 7.5765790939331055
    },
    {
      "epoch": 3.8026809692382815e-05,
      "model_forward_time": 2.926314115524292,
      "step": 39874
    },
    {
      "epoch": 3.8026809692382815e-05,
      "step": 39874,
      "training_step_time": 8.713330030441284
    },
    {
      "epoch": 3.802776336669922e-05,
      "model_forward_time": 1.740126609802246,
      "step": 39875
    },
    {
      "epoch": 3.802776336669922e-05,
      "step": 39875,
      "training_step_time": 9.037540197372437
    },
    {
      "epoch": 3.802871704101562e-05,
      "model_forward_time": 1.7381260395050049,
      "step": 39876
    },
    {
      "epoch": 3.802871704101562e-05,
      "step": 39876,
      "training_step_time": 7.737762212753296
    },
    {
      "epoch": 3.8029670715332034e-05,
      "model_forward_time": 1.7374656200408936,
      "step": 39877
    },
    {
      "epoch": 3.8029670715332034e-05,
      "step": 39877,
      "training_step_time": 7.466045618057251
    },
    {
      "epoch": 3.803062438964844e-05,
      "model_forward_time": 1.7396624088287354,
      "step": 39878
    },
    {
      "epoch": 3.803062438964844e-05,
      "step": 39878,
      "training_step_time": 6.916516304016113
    },
    {
      "epoch": 3.803157806396484e-05,
      "model_forward_time": 1.7280614376068115,
      "step": 39879
    },
    {
      "epoch": 3.803157806396484e-05,
      "step": 39879,
      "training_step_time": 6.8574254512786865
    },
    {
      "epoch": 3.803253173828125e-05,
      "grad_norm": 0.1107032299041748,
      "learning_rate": 7.026782858372982e-06,
      "loss": 0.0578,
      "step": 39880
    },
    {
      "epoch": 3.803253173828125e-05,
      "model_forward_time": 1.7493305206298828,
      "step": 39880
    },
    {
      "dynamics_loss_avg": 0.07084873858839273,
      "epoch": 3.803253173828125e-05,
      "step": 39880
    },
    {
      "action_loss_avg": 0.0019273413985501976,
      "epoch": 3.803253173828125e-05,
      "step": 39880
    },
    {
      "epoch": 3.803253173828125e-05,
      "step": 39880,
      "training_step_time": 6.8737382888793945
    },
    {
      "epoch": 3.8033485412597656e-05,
      "model_forward_time": 1.7450854778289795,
      "step": 39881
    },
    {
      "epoch": 3.8033485412597656e-05,
      "step": 39881,
      "training_step_time": 7.435251235961914
    },
    {
      "epoch": 3.803443908691406e-05,
      "model_forward_time": 1.7621917724609375,
      "step": 39882
    },
    {
      "epoch": 3.803443908691406e-05,
      "step": 39882,
      "training_step_time": 7.568317890167236
    },
    {
      "epoch": 3.803539276123047e-05,
      "model_forward_time": 2.855245351791382,
      "step": 39883
    },
    {
      "epoch": 3.803539276123047e-05,
      "step": 39883,
      "training_step_time": 9.329195499420166
    },
    {
      "epoch": 3.8036346435546875e-05,
      "model_forward_time": 1.7628552913665771,
      "step": 39884
    },
    {
      "epoch": 3.8036346435546875e-05,
      "step": 39884,
      "training_step_time": 10.758196115493774
    },
    {
      "epoch": 3.803730010986328e-05,
      "model_forward_time": 1.7419312000274658,
      "step": 39885
    },
    {
      "epoch": 3.803730010986328e-05,
      "step": 39885,
      "training_step_time": 7.686571836471558
    },
    {
      "epoch": 3.803825378417969e-05,
      "model_forward_time": 1.7363579273223877,
      "step": 39886
    },
    {
      "epoch": 3.803825378417969e-05,
      "step": 39886,
      "training_step_time": 7.485916614532471
    },
    {
      "epoch": 3.8039207458496094e-05,
      "model_forward_time": 1.7442255020141602,
      "step": 39887
    },
    {
      "epoch": 3.8039207458496094e-05,
      "step": 39887,
      "training_step_time": 6.890706300735474
    },
    {
      "epoch": 3.80401611328125e-05,
      "model_forward_time": 1.7573137283325195,
      "step": 39888
    },
    {
      "epoch": 3.80401611328125e-05,
      "step": 39888,
      "training_step_time": 6.824064016342163
    },
    {
      "epoch": 3.804111480712891e-05,
      "model_forward_time": 1.745035171508789,
      "step": 39889
    },
    {
      "epoch": 3.804111480712891e-05,
      "step": 39889,
      "training_step_time": 6.828378438949585
    },
    {
      "epoch": 3.804206848144531e-05,
      "grad_norm": 0.16399921476840973,
      "learning_rate": 7.025271213829981e-06,
      "loss": 0.0661,
      "step": 39890
    },
    {
      "epoch": 3.804206848144531e-05,
      "model_forward_time": 1.754701852798462,
      "step": 39890
    },
    {
      "dynamics_loss_avg": 0.07232805155217648,
      "epoch": 3.804206848144531e-05,
      "step": 39890
    },
    {
      "action_loss_avg": 0.0028529925213661043,
      "epoch": 3.804206848144531e-05,
      "step": 39890
    },
    {
      "epoch": 3.804206848144531e-05,
      "step": 39890,
      "training_step_time": 7.374478101730347
    },
    {
      "epoch": 3.804302215576172e-05,
      "model_forward_time": 1.7450881004333496,
      "step": 39891
    },
    {
      "epoch": 3.804302215576172e-05,
      "step": 39891,
      "training_step_time": 8.258185386657715
    },
    {
      "epoch": 3.804397583007813e-05,
      "model_forward_time": 1.7659087181091309,
      "step": 39892
    },
    {
      "epoch": 3.804397583007813e-05,
      "step": 39892,
      "training_step_time": 7.643930912017822
    },
    {
      "epoch": 3.804492950439453e-05,
      "model_forward_time": 3.0160601139068604,
      "step": 39893
    },
    {
      "epoch": 3.804492950439453e-05,
      "step": 39893,
      "training_step_time": 8.873871564865112
    },
    {
      "epoch": 3.8045883178710936e-05,
      "model_forward_time": 1.7624003887176514,
      "step": 39894
    },
    {
      "epoch": 3.8045883178710936e-05,
      "step": 39894,
      "training_step_time": 7.8824241161346436
    },
    {
      "epoch": 3.804683685302735e-05,
      "model_forward_time": 1.760831594467163,
      "step": 39895
    },
    {
      "epoch": 3.804683685302735e-05,
      "step": 39895,
      "training_step_time": 7.491974830627441
    },
    {
      "epoch": 3.804779052734375e-05,
      "model_forward_time": 1.7642793655395508,
      "step": 39896
    },
    {
      "epoch": 3.804779052734375e-05,
      "step": 39896,
      "training_step_time": 7.238029718399048
    },
    {
      "epoch": 3.8048744201660155e-05,
      "model_forward_time": 1.749903678894043,
      "step": 39897
    },
    {
      "epoch": 3.8048744201660155e-05,
      "step": 39897,
      "training_step_time": 6.812021970748901
    },
    {
      "epoch": 3.8049697875976566e-05,
      "model_forward_time": 1.748706340789795,
      "step": 39898
    },
    {
      "epoch": 3.8049697875976566e-05,
      "step": 39898,
      "training_step_time": 6.8382391929626465
    },
    {
      "epoch": 3.805065155029297e-05,
      "model_forward_time": 1.7509949207305908,
      "step": 39899
    },
    {
      "epoch": 3.805065155029297e-05,
      "step": 39899,
      "training_step_time": 7.3742640018463135
    },
    {
      "epoch": 3.8051605224609374e-05,
      "grad_norm": 0.15786759555339813,
      "learning_rate": 7.023759347806367e-06,
      "loss": 0.0607,
      "step": 39900
    },
    {
      "epoch": 3.8051605224609374e-05,
      "model_forward_time": 1.765134572982788,
      "step": 39900
    },
    {
      "dynamics_loss_avg": 0.06464545615017414,
      "epoch": 3.8051605224609374e-05,
      "step": 39900
    },
    {
      "action_loss_avg": 0.003030729410238564,
      "epoch": 3.8051605224609374e-05,
      "step": 39900
    },
    {
      "epoch": 3.8051605224609374e-05,
      "step": 39900,
      "training_step_time": 8.226386070251465
    },
    {
      "epoch": 3.8052558898925784e-05,
      "model_forward_time": 1.7575864791870117,
      "step": 39901
    },
    {
      "epoch": 3.8052558898925784e-05,
      "step": 39901,
      "training_step_time": 8.50850796699524
    },
    {
      "epoch": 3.805351257324219e-05,
      "model_forward_time": 2.899392604827881,
      "step": 39902
    },
    {
      "epoch": 3.805351257324219e-05,
      "step": 39902,
      "training_step_time": 9.741192817687988
    },
    {
      "epoch": 3.805446624755859e-05,
      "model_forward_time": 1.7647628784179688,
      "step": 39903
    },
    {
      "epoch": 3.805446624755859e-05,
      "step": 39903,
      "training_step_time": 7.817726135253906
    },
    {
      "epoch": 3.8055419921875e-05,
      "model_forward_time": 1.7491073608398438,
      "step": 39904
    },
    {
      "epoch": 3.8055419921875e-05,
      "step": 39904,
      "training_step_time": 7.563314437866211
    },
    {
      "epoch": 3.805637359619141e-05,
      "model_forward_time": 1.7520875930786133,
      "step": 39905
    },
    {
      "epoch": 3.805637359619141e-05,
      "step": 39905,
      "training_step_time": 7.455870628356934
    },
    {
      "epoch": 3.805732727050781e-05,
      "model_forward_time": 1.7554850578308105,
      "step": 39906
    },
    {
      "epoch": 3.805732727050781e-05,
      "step": 39906,
      "training_step_time": 6.774305582046509
    },
    {
      "epoch": 3.8058280944824215e-05,
      "model_forward_time": 1.7586264610290527,
      "step": 39907
    },
    {
      "epoch": 3.8058280944824215e-05,
      "step": 39907,
      "training_step_time": 6.811164379119873
    },
    {
      "epoch": 3.8059234619140626e-05,
      "model_forward_time": 1.7382290363311768,
      "step": 39908
    },
    {
      "epoch": 3.8059234619140626e-05,
      "step": 39908,
      "training_step_time": 7.366085767745972
    },
    {
      "epoch": 3.806018829345703e-05,
      "model_forward_time": 1.7800686359405518,
      "step": 39909
    },
    {
      "epoch": 3.806018829345703e-05,
      "step": 39909,
      "training_step_time": 7.106164455413818
    },
    {
      "epoch": 3.8061141967773434e-05,
      "grad_norm": 0.13066934049129486,
      "learning_rate": 7.022247260467473e-06,
      "loss": 0.0625,
      "step": 39910
    },
    {
      "epoch": 3.8061141967773434e-05,
      "model_forward_time": 1.7524828910827637,
      "step": 39910
    },
    {
      "dynamics_loss_avg": 0.04902185946702957,
      "epoch": 3.8061141967773434e-05,
      "step": 39910
    },
    {
      "action_loss_avg": 0.001723472512094304,
      "epoch": 3.8061141967773434e-05,
      "step": 39910
    },
    {
      "epoch": 3.8061141967773434e-05,
      "step": 39910,
      "training_step_time": 7.73152232170105
    },
    {
      "epoch": 3.8062095642089845e-05,
      "model_forward_time": 2.882194757461548,
      "step": 39911
    },
    {
      "epoch": 3.8062095642089845e-05,
      "step": 39911,
      "training_step_time": 9.45242977142334
    },
    {
      "epoch": 3.806304931640625e-05,
      "model_forward_time": 1.765026569366455,
      "step": 39912
    },
    {
      "epoch": 3.806304931640625e-05,
      "step": 39912,
      "training_step_time": 9.304215908050537
    },
    {
      "epoch": 3.806400299072265e-05,
      "model_forward_time": 1.7598822116851807,
      "step": 39913
    },
    {
      "epoch": 3.806400299072265e-05,
      "step": 39913,
      "training_step_time": 7.659041404724121
    },
    {
      "epoch": 3.8064956665039064e-05,
      "model_forward_time": 1.7483923435211182,
      "step": 39914
    },
    {
      "epoch": 3.8064956665039064e-05,
      "step": 39914,
      "training_step_time": 7.4666547775268555
    },
    {
      "epoch": 3.806591033935547e-05,
      "model_forward_time": 1.7523350715637207,
      "step": 39915
    },
    {
      "epoch": 3.806591033935547e-05,
      "step": 39915,
      "training_step_time": 6.813392877578735
    },
    {
      "epoch": 3.806686401367187e-05,
      "model_forward_time": 1.7603566646575928,
      "step": 39916
    },
    {
      "epoch": 3.806686401367187e-05,
      "step": 39916,
      "training_step_time": 6.794821500778198
    },
    {
      "epoch": 3.806781768798828e-05,
      "model_forward_time": 1.7575793266296387,
      "step": 39917
    },
    {
      "epoch": 3.806781768798828e-05,
      "step": 39917,
      "training_step_time": 7.39451265335083
    },
    {
      "epoch": 3.806877136230469e-05,
      "model_forward_time": 1.7611098289489746,
      "step": 39918
    },
    {
      "epoch": 3.806877136230469e-05,
      "step": 39918,
      "training_step_time": 7.057096242904663
    },
    {
      "epoch": 3.806972503662109e-05,
      "model_forward_time": 1.7879011631011963,
      "step": 39919
    },
    {
      "epoch": 3.806972503662109e-05,
      "step": 39919,
      "training_step_time": 8.605308771133423
    },
    {
      "epoch": 3.80706787109375e-05,
      "grad_norm": 0.14576289057731628,
      "learning_rate": 7.020734951978658e-06,
      "loss": 0.0615,
      "step": 39920
    },
    {
      "epoch": 3.80706787109375e-05,
      "model_forward_time": 2.883227825164795,
      "step": 39920
    },
    {
      "dynamics_loss_avg": 0.055426903814077375,
      "epoch": 3.80706787109375e-05,
      "step": 39920
    },
    {
      "action_loss_avg": 0.0017882265121443198,
      "epoch": 3.80706787109375e-05,
      "step": 39920
    },
    {
      "epoch": 3.80706787109375e-05,
      "step": 39920,
      "training_step_time": 8.763659715652466
    },
    {
      "epoch": 3.8071632385253906e-05,
      "model_forward_time": 1.6876633167266846,
      "step": 39921
    },
    {
      "epoch": 3.8071632385253906e-05,
      "step": 39921,
      "training_step_time": 9.50743842124939
    },
    {
      "epoch": 3.807258605957031e-05,
      "model_forward_time": 1.691967487335205,
      "step": 39922
    },
    {
      "epoch": 3.807258605957031e-05,
      "step": 39922,
      "training_step_time": 7.681017875671387
    },
    {
      "epoch": 3.807353973388672e-05,
      "model_forward_time": 1.689178228378296,
      "step": 39923
    },
    {
      "epoch": 3.807353973388672e-05,
      "step": 39923,
      "training_step_time": 7.600973129272461
    },
    {
      "epoch": 3.8074493408203125e-05,
      "model_forward_time": 1.7142505645751953,
      "step": 39924
    },
    {
      "epoch": 3.8074493408203125e-05,
      "step": 39924,
      "training_step_time": 7.364772081375122
    },
    {
      "epoch": 3.807544708251953e-05,
      "model_forward_time": 1.688236951828003,
      "step": 39925
    },
    {
      "epoch": 3.807544708251953e-05,
      "step": 39925,
      "training_step_time": 6.830358028411865
    },
    {
      "epoch": 3.807640075683594e-05,
      "model_forward_time": 1.6889522075653076,
      "step": 39926
    },
    {
      "epoch": 3.807640075683594e-05,
      "step": 39926,
      "training_step_time": 7.411468029022217
    },
    {
      "epoch": 3.807735443115234e-05,
      "model_forward_time": 1.691495418548584,
      "step": 39927
    },
    {
      "epoch": 3.807735443115234e-05,
      "step": 39927,
      "training_step_time": 6.859772682189941
    },
    {
      "epoch": 3.807830810546875e-05,
      "model_forward_time": 1.3165717124938965,
      "step": 39928
    },
    {
      "epoch": 3.807830810546875e-05,
      "step": 39928,
      "training_step_time": 7.6275341510772705
    },
    {
      "epoch": 3.807926177978516e-05,
      "model_forward_time": 1.6977524757385254,
      "step": 39929
    },
    {
      "epoch": 3.807926177978516e-05,
      "step": 39929,
      "training_step_time": 9.936567068099976
    },
    {
      "epoch": 3.808021545410156e-05,
      "grad_norm": 0.12960481643676758,
      "learning_rate": 7.0192224225053075e-06,
      "loss": 0.061,
      "step": 39930
    },
    {
      "epoch": 3.808021545410156e-05,
      "model_forward_time": 1.6897239685058594,
      "step": 39930
    },
    {
      "dynamics_loss_avg": 0.06421523094177246,
      "epoch": 3.808021545410156e-05,
      "step": 39930
    },
    {
      "action_loss_avg": 0.003020511387148872,
      "epoch": 3.808021545410156e-05,
      "step": 39930
    },
    {
      "epoch": 3.808021545410156e-05,
      "step": 39930,
      "training_step_time": 8.952987432479858
    },
    {
      "epoch": 3.8081169128417966e-05,
      "model_forward_time": 1.682671308517456,
      "step": 39931
    },
    {
      "epoch": 3.8081169128417966e-05,
      "step": 39931,
      "training_step_time": 7.695350408554077
    },
    {
      "epoch": 3.808212280273438e-05,
      "model_forward_time": 1.6900098323822021,
      "step": 39932
    },
    {
      "epoch": 3.808212280273438e-05,
      "step": 39932,
      "training_step_time": 7.6274755001068115
    },
    {
      "epoch": 3.808307647705078e-05,
      "model_forward_time": 1.7080605030059814,
      "step": 39933
    },
    {
      "epoch": 3.808307647705078e-05,
      "step": 39933,
      "training_step_time": 7.488318681716919
    },
    {
      "epoch": 3.8084030151367185e-05,
      "model_forward_time": 1.6861629486083984,
      "step": 39934
    },
    {
      "epoch": 3.8084030151367185e-05,
      "step": 39934,
      "training_step_time": 6.892522573471069
    },
    {
      "epoch": 3.8084983825683596e-05,
      "model_forward_time": 1.6899325847625732,
      "step": 39935
    },
    {
      "epoch": 3.8084983825683596e-05,
      "step": 39935,
      "training_step_time": 7.2991156578063965
    },
    {
      "epoch": 3.80859375e-05,
      "model_forward_time": 1.6925511360168457,
      "step": 39936
    },
    {
      "epoch": 3.80859375e-05,
      "step": 39936,
      "training_step_time": 6.923272609710693
    },
    {
      "epoch": 3.8086891174316404e-05,
      "model_forward_time": 1.6925923824310303,
      "step": 39937
    },
    {
      "epoch": 3.8086891174316404e-05,
      "step": 39937,
      "training_step_time": 7.718594551086426
    },
    {
      "epoch": 3.8087844848632815e-05,
      "model_forward_time": 1.6895945072174072,
      "step": 39938
    },
    {
      "epoch": 3.8087844848632815e-05,
      "step": 39938,
      "training_step_time": 8.37588381767273
    },
    {
      "epoch": 3.808879852294922e-05,
      "model_forward_time": 2.2300236225128174,
      "step": 39939
    },
    {
      "epoch": 3.808879852294922e-05,
      "step": 39939,
      "training_step_time": 8.906125545501709
    },
    {
      "epoch": 3.808975219726562e-05,
      "grad_norm": 0.14657533168792725,
      "learning_rate": 7.017709672212828e-06,
      "loss": 0.0596,
      "step": 39940
    },
    {
      "epoch": 3.808975219726562e-05,
      "model_forward_time": 1.6850590705871582,
      "step": 39940
    },
    {
      "dynamics_loss_avg": 0.03968543261289596,
      "epoch": 3.808975219726562e-05,
      "step": 39940
    },
    {
      "action_loss_avg": 0.0017490249127149581,
      "epoch": 3.808975219726562e-05,
      "step": 39940
    },
    {
      "epoch": 3.808975219726562e-05,
      "step": 39940,
      "training_step_time": 7.766592741012573
    },
    {
      "epoch": 3.8090705871582034e-05,
      "model_forward_time": 1.688408374786377,
      "step": 39941
    },
    {
      "epoch": 3.8090705871582034e-05,
      "step": 39941,
      "training_step_time": 7.557273864746094
    },
    {
      "epoch": 3.809165954589844e-05,
      "model_forward_time": 1.6981439590454102,
      "step": 39942
    },
    {
      "epoch": 3.809165954589844e-05,
      "step": 39942,
      "training_step_time": 7.410325050354004
    },
    {
      "epoch": 3.809261322021484e-05,
      "model_forward_time": 1.693678617477417,
      "step": 39943
    },
    {
      "epoch": 3.809261322021484e-05,
      "step": 39943,
      "training_step_time": 6.864583492279053
    },
    {
      "epoch": 3.809356689453125e-05,
      "model_forward_time": 1.6951801776885986,
      "step": 39944
    },
    {
      "epoch": 3.809356689453125e-05,
      "step": 39944,
      "training_step_time": 6.866017580032349
    },
    {
      "epoch": 3.8094520568847657e-05,
      "model_forward_time": 1.6977336406707764,
      "step": 39945
    },
    {
      "epoch": 3.8094520568847657e-05,
      "step": 39945,
      "training_step_time": 7.352474212646484
    },
    {
      "epoch": 3.809547424316406e-05,
      "model_forward_time": 1.6922619342803955,
      "step": 39946
    },
    {
      "epoch": 3.809547424316406e-05,
      "step": 39946,
      "training_step_time": 7.619189977645874
    },
    {
      "epoch": 3.809642791748047e-05,
      "model_forward_time": 1.6869802474975586,
      "step": 39947
    },
    {
      "epoch": 3.809642791748047e-05,
      "step": 39947,
      "training_step_time": 8.211187839508057
    },
    {
      "epoch": 3.8097381591796875e-05,
      "model_forward_time": 2.846585750579834,
      "step": 39948
    },
    {
      "epoch": 3.8097381591796875e-05,
      "step": 39948,
      "training_step_time": 10.320450782775879
    },
    {
      "epoch": 3.809833526611328e-05,
      "model_forward_time": 1.7645478248596191,
      "step": 39949
    },
    {
      "epoch": 3.809833526611328e-05,
      "step": 39949,
      "training_step_time": 9.037467241287231
    },
    {
      "epoch": 3.809928894042969e-05,
      "grad_norm": 0.0946073830127716,
      "learning_rate": 7.016196701266653e-06,
      "loss": 0.0587,
      "step": 39950
    },
    {
      "epoch": 3.809928894042969e-05,
      "model_forward_time": 1.756596565246582,
      "step": 39950
    },
    {
      "dynamics_loss_avg": 0.04513092525303364,
      "epoch": 3.809928894042969e-05,
      "step": 39950
    },
    {
      "action_loss_avg": 0.0012378274230286479,
      "epoch": 3.809928894042969e-05,
      "step": 39950
    },
    {
      "epoch": 3.809928894042969e-05,
      "step": 39950,
      "training_step_time": 7.527097225189209
    },
    {
      "epoch": 3.8100242614746094e-05,
      "model_forward_time": 1.7557883262634277,
      "step": 39951
    },
    {
      "epoch": 3.8100242614746094e-05,
      "step": 39951,
      "training_step_time": 7.564605474472046
    },
    {
      "epoch": 3.81011962890625e-05,
      "model_forward_time": 1.7763574123382568,
      "step": 39952
    },
    {
      "epoch": 3.81011962890625e-05,
      "step": 39952,
      "training_step_time": 7.3272881507873535
    },
    {
      "epoch": 3.810214996337891e-05,
      "model_forward_time": 1.7668194770812988,
      "step": 39953
    },
    {
      "epoch": 3.810214996337891e-05,
      "step": 39953,
      "training_step_time": 6.80717921257019
    },
    {
      "epoch": 3.810310363769531e-05,
      "model_forward_time": 1.7610714435577393,
      "step": 39954
    },
    {
      "epoch": 3.810310363769531e-05,
      "step": 39954,
      "training_step_time": 7.380553960800171
    },
    {
      "epoch": 3.810405731201172e-05,
      "model_forward_time": 1.732618808746338,
      "step": 39955
    },
    {
      "epoch": 3.810405731201172e-05,
      "step": 39955,
      "training_step_time": 8.352868556976318
    },
    {
      "epoch": 3.810501098632813e-05,
      "model_forward_time": 1.7507684230804443,
      "step": 39956
    },
    {
      "epoch": 3.810501098632813e-05,
      "step": 39956,
      "training_step_time": 7.709517955780029
    },
    {
      "epoch": 3.810596466064453e-05,
      "model_forward_time": 2.8498575687408447,
      "step": 39957
    },
    {
      "epoch": 3.810596466064453e-05,
      "step": 39957,
      "training_step_time": 8.65203857421875
    },
    {
      "epoch": 3.8106918334960936e-05,
      "model_forward_time": 1.7606236934661865,
      "step": 39958
    },
    {
      "epoch": 3.8106918334960936e-05,
      "step": 39958,
      "training_step_time": 8.491996049880981
    },
    {
      "epoch": 3.810787200927735e-05,
      "model_forward_time": 1.7434892654418945,
      "step": 39959
    },
    {
      "epoch": 3.810787200927735e-05,
      "step": 39959,
      "training_step_time": 7.7127110958099365
    },
    {
      "epoch": 3.810882568359375e-05,
      "grad_norm": 0.11931680142879486,
      "learning_rate": 7.014683509832237e-06,
      "loss": 0.0576,
      "step": 39960
    },
    {
      "epoch": 3.810882568359375e-05,
      "model_forward_time": 1.7469720840454102,
      "step": 39960
    },
    {
      "dynamics_loss_avg": 0.06685759332031012,
      "epoch": 3.810882568359375e-05,
      "step": 39960
    },
    {
      "action_loss_avg": 0.0016512925300048663,
      "epoch": 3.810882568359375e-05,
      "step": 39960
    },
    {
      "epoch": 3.810882568359375e-05,
      "step": 39960,
      "training_step_time": 7.59568977355957
    },
    {
      "epoch": 3.8109779357910155e-05,
      "model_forward_time": 1.7938625812530518,
      "step": 39961
    },
    {
      "epoch": 3.8109779357910155e-05,
      "step": 39961,
      "training_step_time": 7.4044294357299805
    },
    {
      "epoch": 3.8110733032226566e-05,
      "model_forward_time": 1.7470958232879639,
      "step": 39962
    },
    {
      "epoch": 3.8110733032226566e-05,
      "step": 39962,
      "training_step_time": 6.787184238433838
    },
    {
      "epoch": 3.811168670654297e-05,
      "model_forward_time": 1.7680153846740723,
      "step": 39963
    },
    {
      "epoch": 3.811168670654297e-05,
      "step": 39963,
      "training_step_time": 7.377965688705444
    },
    {
      "epoch": 3.8112640380859374e-05,
      "model_forward_time": 1.768078327178955,
      "step": 39964
    },
    {
      "epoch": 3.8112640380859374e-05,
      "step": 39964,
      "training_step_time": 7.333224296569824
    },
    {
      "epoch": 3.8113594055175785e-05,
      "model_forward_time": 1.7860102653503418,
      "step": 39965
    },
    {
      "epoch": 3.8113594055175785e-05,
      "step": 39965,
      "training_step_time": 7.6325085163116455
    },
    {
      "epoch": 3.811454772949219e-05,
      "model_forward_time": 2.960663318634033,
      "step": 39966
    },
    {
      "epoch": 3.811454772949219e-05,
      "step": 39966,
      "training_step_time": 9.149287700653076
    },
    {
      "epoch": 3.811550140380859e-05,
      "model_forward_time": 1.7771620750427246,
      "step": 39967
    },
    {
      "epoch": 3.811550140380859e-05,
      "step": 39967,
      "training_step_time": 8.949394941329956
    },
    {
      "epoch": 3.8116455078125e-05,
      "model_forward_time": 1.7648262977600098,
      "step": 39968
    },
    {
      "epoch": 3.8116455078125e-05,
      "step": 39968,
      "training_step_time": 7.638754367828369
    },
    {
      "epoch": 3.811740875244141e-05,
      "model_forward_time": 1.7686808109283447,
      "step": 39969
    },
    {
      "epoch": 3.811740875244141e-05,
      "step": 39969,
      "training_step_time": 7.389442682266235
    },
    {
      "epoch": 3.811836242675781e-05,
      "grad_norm": 0.1144680306315422,
      "learning_rate": 7.013170098075063e-06,
      "loss": 0.0581,
      "step": 39970
    },
    {
      "epoch": 3.811836242675781e-05,
      "model_forward_time": 1.791254997253418,
      "step": 39970
    },
    {
      "dynamics_loss_avg": 0.0431157436221838,
      "epoch": 3.811836242675781e-05,
      "step": 39970
    },
    {
      "action_loss_avg": 0.0016294836241286248,
      "epoch": 3.811836242675781e-05,
      "step": 39970
    },
    {
      "epoch": 3.811836242675781e-05,
      "step": 39970,
      "training_step_time": 7.336359739303589
    },
    {
      "epoch": 3.8119316101074215e-05,
      "model_forward_time": 1.8051540851593018,
      "step": 39971
    },
    {
      "epoch": 3.8119316101074215e-05,
      "step": 39971,
      "training_step_time": 6.7447004318237305
    },
    {
      "epoch": 3.8120269775390626e-05,
      "model_forward_time": 1.7601242065429688,
      "step": 39972
    },
    {
      "epoch": 3.8120269775390626e-05,
      "step": 39972,
      "training_step_time": 7.267086029052734
    },
    {
      "epoch": 3.812122344970703e-05,
      "model_forward_time": 1.771533727645874,
      "step": 39973
    },
    {
      "epoch": 3.812122344970703e-05,
      "step": 39973,
      "training_step_time": 7.267935514450073
    },
    {
      "epoch": 3.8122177124023434e-05,
      "model_forward_time": 1.7634570598602295,
      "step": 39974
    },
    {
      "epoch": 3.8122177124023434e-05,
      "step": 39974,
      "training_step_time": 7.561201572418213
    },
    {
      "epoch": 3.8123130798339845e-05,
      "model_forward_time": 2.9028851985931396,
      "step": 39975
    },
    {
      "epoch": 3.8123130798339845e-05,
      "step": 39975,
      "training_step_time": 9.103368759155273
    },
    {
      "epoch": 3.812408447265625e-05,
      "model_forward_time": 1.8362360000610352,
      "step": 39976
    },
    {
      "epoch": 3.812408447265625e-05,
      "step": 39976,
      "training_step_time": 8.050847291946411
    },
    {
      "epoch": 3.812503814697265e-05,
      "model_forward_time": 1.7625539302825928,
      "step": 39977
    },
    {
      "epoch": 3.812503814697265e-05,
      "step": 39977,
      "training_step_time": 7.760057210922241
    },
    {
      "epoch": 3.8125991821289064e-05,
      "model_forward_time": 1.7649149894714355,
      "step": 39978
    },
    {
      "epoch": 3.8125991821289064e-05,
      "step": 39978,
      "training_step_time": 7.445802211761475
    },
    {
      "epoch": 3.812694549560547e-05,
      "model_forward_time": 1.7517890930175781,
      "step": 39979
    },
    {
      "epoch": 3.812694549560547e-05,
      "step": 39979,
      "training_step_time": 7.350794315338135
    },
    {
      "epoch": 3.812789916992187e-05,
      "grad_norm": 0.1066998690366745,
      "learning_rate": 7.011656466160633e-06,
      "loss": 0.0542,
      "step": 39980
    },
    {
      "epoch": 3.812789916992187e-05,
      "model_forward_time": 1.7705333232879639,
      "step": 39980
    },
    {
      "dynamics_loss_avg": 0.054412329010665415,
      "epoch": 3.812789916992187e-05,
      "step": 39980
    },
    {
      "action_loss_avg": 0.0024824523861752823,
      "epoch": 3.812789916992187e-05,
      "step": 39980
    },
    {
      "epoch": 3.812789916992187e-05,
      "step": 39980,
      "training_step_time": 6.758645534515381
    },
    {
      "epoch": 3.812885284423828e-05,
      "model_forward_time": 1.7771427631378174,
      "step": 39981
    },
    {
      "epoch": 3.812885284423828e-05,
      "step": 39981,
      "training_step_time": 6.796187400817871
    },
    {
      "epoch": 3.812980651855469e-05,
      "model_forward_time": 1.768019437789917,
      "step": 39982
    },
    {
      "epoch": 3.812980651855469e-05,
      "step": 39982,
      "training_step_time": 7.259541988372803
    },
    {
      "epoch": 3.813076019287109e-05,
      "model_forward_time": 1.7529513835906982,
      "step": 39983
    },
    {
      "epoch": 3.813076019287109e-05,
      "step": 39983,
      "training_step_time": 7.660967588424683
    },
    {
      "epoch": 3.81317138671875e-05,
      "model_forward_time": 1.7719051837921143,
      "step": 39984
    },
    {
      "epoch": 3.81317138671875e-05,
      "step": 39984,
      "training_step_time": 8.14470362663269
    },
    {
      "epoch": 3.8132667541503906e-05,
      "model_forward_time": 1.4739575386047363,
      "step": 39985
    },
    {
      "epoch": 3.8132667541503906e-05,
      "step": 39985,
      "training_step_time": 9.033684253692627
    },
    {
      "epoch": 3.813362121582031e-05,
      "model_forward_time": 1.750575065612793,
      "step": 39986
    },
    {
      "epoch": 3.813362121582031e-05,
      "step": 39986,
      "training_step_time": 7.673112154006958
    },
    {
      "epoch": 3.813457489013672e-05,
      "model_forward_time": 1.768972635269165,
      "step": 39987
    },
    {
      "epoch": 3.813457489013672e-05,
      "step": 39987,
      "training_step_time": 7.509057998657227
    },
    {
      "epoch": 3.8135528564453125e-05,
      "model_forward_time": 1.756690263748169,
      "step": 39988
    },
    {
      "epoch": 3.8135528564453125e-05,
      "step": 39988,
      "training_step_time": 7.537785053253174
    },
    {
      "epoch": 3.813648223876953e-05,
      "model_forward_time": 1.7681279182434082,
      "step": 39989
    },
    {
      "epoch": 3.813648223876953e-05,
      "step": 39989,
      "training_step_time": 6.778243541717529
    },
    {
      "epoch": 3.813743591308594e-05,
      "grad_norm": 0.13632702827453613,
      "learning_rate": 7.010142614254476e-06,
      "loss": 0.0568,
      "step": 39990
    },
    {
      "epoch": 3.813743591308594e-05,
      "model_forward_time": 1.7607102394104004,
      "step": 39990
    },
    {
      "dynamics_loss_avg": 0.044225200451910494,
      "epoch": 3.813743591308594e-05,
      "step": 39990
    },
    {
      "action_loss_avg": 0.0017775277723558248,
      "epoch": 3.813743591308594e-05,
      "step": 39990
    },
    {
      "epoch": 3.813743591308594e-05,
      "step": 39990,
      "training_step_time": 6.840887069702148
    },
    {
      "epoch": 3.8138389587402343e-05,
      "model_forward_time": 1.7574782371520996,
      "step": 39991
    },
    {
      "epoch": 3.8138389587402343e-05,
      "step": 39991,
      "training_step_time": 7.264442682266235
    },
    {
      "epoch": 3.813934326171875e-05,
      "model_forward_time": 1.4129416942596436,
      "step": 39992
    },
    {
      "epoch": 3.813934326171875e-05,
      "step": 39992,
      "training_step_time": 8.129131317138672
    },
    {
      "epoch": 3.814029693603516e-05,
      "model_forward_time": 1.4080214500427246,
      "step": 39993
    },
    {
      "epoch": 3.814029693603516e-05,
      "step": 39993,
      "training_step_time": 7.591398477554321
    },
    {
      "epoch": 3.814125061035156e-05,
      "model_forward_time": 2.9457688331604004,
      "step": 39994
    },
    {
      "epoch": 3.814125061035156e-05,
      "step": 39994,
      "training_step_time": 8.800816059112549
    },
    {
      "epoch": 3.8142204284667966e-05,
      "model_forward_time": 1.7634539604187012,
      "step": 39995
    },
    {
      "epoch": 3.8142204284667966e-05,
      "step": 39995,
      "training_step_time": 8.29360055923462
    },
    {
      "epoch": 3.814315795898438e-05,
      "model_forward_time": 1.771125316619873,
      "step": 39996
    },
    {
      "epoch": 3.814315795898438e-05,
      "step": 39996,
      "training_step_time": 7.5303425788879395
    },
    {
      "epoch": 3.814411163330078e-05,
      "model_forward_time": 1.7871229648590088,
      "step": 39997
    },
    {
      "epoch": 3.814411163330078e-05,
      "step": 39997,
      "training_step_time": 7.587386608123779
    },
    {
      "epoch": 3.8145065307617185e-05,
      "model_forward_time": 1.7985615730285645,
      "step": 39998
    },
    {
      "epoch": 3.8145065307617185e-05,
      "step": 39998,
      "training_step_time": 7.455546140670776
    },
    {
      "epoch": 3.8146018981933596e-05,
      "model_forward_time": 1.7937781810760498,
      "step": 39999
    },
    {
      "epoch": 3.8146018981933596e-05,
      "step": 39999,
      "training_step_time": 6.871437311172485
    },
    {
      "epoch": 3.814697265625e-05,
      "grad_norm": 0.12759344279766083,
      "learning_rate": 7.008628542522146e-06,
      "loss": 0.0576,
      "step": 40000
    },
    {
      "epoch": 3.814697265625e-05,
      "model_forward_time": 1.7796642780303955,
      "step": 40000
    },
    {
      "dynamics_loss_avg": 0.05207154452800751,
      "epoch": 3.814697265625e-05,
      "step": 40000
    },
    {
      "action_loss_avg": 0.001615812853560783,
      "epoch": 3.814697265625e-05,
      "step": 40000
    },
    {
      "epoch": 3.814697265625e-05,
      "step": 40000,
      "training_step_time": 7.498897314071655
    },
    {
      "epoch": 3.8147926330566404e-05,
      "model_forward_time": 1.7555160522460938,
      "step": 40001
    },
    {
      "epoch": 3.8147926330566404e-05,
      "step": 40001,
      "training_step_time": 7.861031770706177
    },
    {
      "epoch": 3.8148880004882815e-05,
      "model_forward_time": 1.767650842666626,
      "step": 40002
    },
    {
      "epoch": 3.8148880004882815e-05,
      "step": 40002,
      "training_step_time": 8.41990613937378
    },
    {
      "epoch": 3.814983367919922e-05,
      "model_forward_time": 3.405264139175415,
      "step": 40003
    },
    {
      "epoch": 3.814983367919922e-05,
      "step": 40003,
      "training_step_time": 12.845359802246094
    },
    {
      "epoch": 3.815078735351562e-05,
      "model_forward_time": 1.7570605278015137,
      "step": 40004
    },
    {
      "epoch": 3.815078735351562e-05,
      "step": 40004,
      "training_step_time": 8.673028945922852
    },
    {
      "epoch": 3.8151741027832034e-05,
      "model_forward_time": 1.7960641384124756,
      "step": 40005
    },
    {
      "epoch": 3.8151741027832034e-05,
      "step": 40005,
      "training_step_time": 8.146634578704834
    },
    {
      "epoch": 3.815269470214844e-05,
      "model_forward_time": 1.7444641590118408,
      "step": 40006
    },
    {
      "epoch": 3.815269470214844e-05,
      "step": 40006,
      "training_step_time": 7.513581991195679
    },
    {
      "epoch": 3.815364837646484e-05,
      "model_forward_time": 1.7606933116912842,
      "step": 40007
    },
    {
      "epoch": 3.815364837646484e-05,
      "step": 40007,
      "training_step_time": 7.329736232757568
    },
    {
      "epoch": 3.815460205078125e-05,
      "model_forward_time": 1.7577519416809082,
      "step": 40008
    },
    {
      "epoch": 3.815460205078125e-05,
      "step": 40008,
      "training_step_time": 6.737669467926025
    },
    {
      "epoch": 3.8155555725097657e-05,
      "model_forward_time": 1.7597036361694336,
      "step": 40009
    },
    {
      "epoch": 3.8155555725097657e-05,
      "step": 40009,
      "training_step_time": 8.52168893814087
    },
    {
      "epoch": 3.815650939941406e-05,
      "grad_norm": 0.12140990048646927,
      "learning_rate": 7.007114251129219e-06,
      "loss": 0.0554,
      "step": 40010
    },
    {
      "epoch": 3.815650939941406e-05,
      "model_forward_time": 1.7475671768188477,
      "step": 40010
    },
    {
      "dynamics_loss_avg": 0.06551170628517866,
      "epoch": 3.815650939941406e-05,
      "step": 40010
    },
    {
      "action_loss_avg": 0.0018245488521642984,
      "epoch": 3.815650939941406e-05,
      "step": 40010
    },
    {
      "epoch": 3.815650939941406e-05,
      "step": 40010,
      "training_step_time": 7.6880271434783936
    },
    {
      "epoch": 3.815746307373047e-05,
      "model_forward_time": 1.7518577575683594,
      "step": 40011
    },
    {
      "epoch": 3.815746307373047e-05,
      "step": 40011,
      "training_step_time": 9.159027338027954
    },
    {
      "epoch": 3.8158416748046875e-05,
      "model_forward_time": 2.298504114151001,
      "step": 40012
    },
    {
      "epoch": 3.8158416748046875e-05,
      "step": 40012,
      "training_step_time": 9.770509958267212
    },
    {
      "epoch": 3.815937042236328e-05,
      "model_forward_time": 1.7626280784606934,
      "step": 40013
    },
    {
      "epoch": 3.815937042236328e-05,
      "step": 40013,
      "training_step_time": 8.21420669555664
    },
    {
      "epoch": 3.816032409667969e-05,
      "model_forward_time": 1.7527527809143066,
      "step": 40014
    },
    {
      "epoch": 3.816032409667969e-05,
      "step": 40014,
      "training_step_time": 8.36401081085205
    },
    {
      "epoch": 3.8161277770996094e-05,
      "model_forward_time": 1.7742862701416016,
      "step": 40015
    },
    {
      "epoch": 3.8161277770996094e-05,
      "step": 40015,
      "training_step_time": 7.561439514160156
    },
    {
      "epoch": 3.81622314453125e-05,
      "model_forward_time": 1.767329454421997,
      "step": 40016
    },
    {
      "epoch": 3.81622314453125e-05,
      "step": 40016,
      "training_step_time": 7.340147495269775
    },
    {
      "epoch": 3.816318511962891e-05,
      "model_forward_time": 1.7740063667297363,
      "step": 40017
    },
    {
      "epoch": 3.816318511962891e-05,
      "step": 40017,
      "training_step_time": 6.713970184326172
    },
    {
      "epoch": 3.816413879394531e-05,
      "model_forward_time": 1.77604341506958,
      "step": 40018
    },
    {
      "epoch": 3.816413879394531e-05,
      "step": 40018,
      "training_step_time": 8.382959842681885
    },
    {
      "epoch": 3.816509246826172e-05,
      "model_forward_time": 1.75189208984375,
      "step": 40019
    },
    {
      "epoch": 3.816509246826172e-05,
      "step": 40019,
      "training_step_time": 7.595247983932495
    },
    {
      "epoch": 3.816604614257813e-05,
      "grad_norm": 0.0997433289885521,
      "learning_rate": 7.005599740241293e-06,
      "loss": 0.0581,
      "step": 40020
    },
    {
      "epoch": 3.816604614257813e-05,
      "model_forward_time": 1.7542648315429688,
      "step": 40020
    },
    {
      "dynamics_loss_avg": 0.07183880563825369,
      "epoch": 3.816604614257813e-05,
      "step": 40020
    },
    {
      "action_loss_avg": 0.003006168274441734,
      "epoch": 3.816604614257813e-05,
      "step": 40020
    },
    {
      "epoch": 3.816604614257813e-05,
      "step": 40020,
      "training_step_time": 7.555177688598633
    },
    {
      "epoch": 3.816699981689453e-05,
      "model_forward_time": 2.8988938331604004,
      "step": 40021
    },
    {
      "epoch": 3.816699981689453e-05,
      "step": 40021,
      "training_step_time": 7.70729398727417
    },
    {
      "epoch": 3.8167953491210936e-05,
      "model_forward_time": 1.7720997333526611,
      "step": 40022
    },
    {
      "epoch": 3.8167953491210936e-05,
      "step": 40022,
      "training_step_time": 8.724695682525635
    },
    {
      "epoch": 3.816890716552735e-05,
      "model_forward_time": 1.7638440132141113,
      "step": 40023
    },
    {
      "epoch": 3.816890716552735e-05,
      "step": 40023,
      "training_step_time": 8.510266065597534
    },
    {
      "epoch": 3.816986083984375e-05,
      "model_forward_time": 1.7606937885284424,
      "step": 40024
    },
    {
      "epoch": 3.816986083984375e-05,
      "step": 40024,
      "training_step_time": 7.498741149902344
    },
    {
      "epoch": 3.8170814514160155e-05,
      "model_forward_time": 1.7852106094360352,
      "step": 40025
    },
    {
      "epoch": 3.8170814514160155e-05,
      "step": 40025,
      "training_step_time": 7.35932469367981
    },
    {
      "epoch": 3.8171768188476566e-05,
      "model_forward_time": 1.7470941543579102,
      "step": 40026
    },
    {
      "epoch": 3.8171768188476566e-05,
      "step": 40026,
      "training_step_time": 6.737240314483643
    },
    {
      "epoch": 3.817272186279297e-05,
      "model_forward_time": 1.7622153759002686,
      "step": 40027
    },
    {
      "epoch": 3.817272186279297e-05,
      "step": 40027,
      "training_step_time": 7.569540500640869
    },
    {
      "epoch": 3.8173675537109374e-05,
      "model_forward_time": 1.7549912929534912,
      "step": 40028
    },
    {
      "epoch": 3.8173675537109374e-05,
      "step": 40028,
      "training_step_time": 7.548521280288696
    },
    {
      "epoch": 3.8174629211425785e-05,
      "model_forward_time": 1.7767188549041748,
      "step": 40029
    },
    {
      "epoch": 3.8174629211425785e-05,
      "step": 40029,
      "training_step_time": 8.206014156341553
    },
    {
      "epoch": 3.817558288574219e-05,
      "grad_norm": 0.12026958167552948,
      "learning_rate": 7.004085010023996e-06,
      "loss": 0.0563,
      "step": 40030
    },
    {
      "epoch": 3.817558288574219e-05,
      "model_forward_time": 1.7485542297363281,
      "step": 40030
    },
    {
      "dynamics_loss_avg": 0.04288069847971201,
      "epoch": 3.817558288574219e-05,
      "step": 40030
    },
    {
      "action_loss_avg": 0.0012979389022802934,
      "epoch": 3.817558288574219e-05,
      "step": 40030
    },
    {
      "epoch": 3.817558288574219e-05,
      "step": 40030,
      "training_step_time": 9.18591856956482
    },
    {
      "epoch": 3.817653656005859e-05,
      "model_forward_time": 1.8099987506866455,
      "step": 40031
    },
    {
      "epoch": 3.817653656005859e-05,
      "step": 40031,
      "training_step_time": 8.71541428565979
    },
    {
      "epoch": 3.8177490234375e-05,
      "model_forward_time": 1.7610533237457275,
      "step": 40032
    },
    {
      "epoch": 3.8177490234375e-05,
      "step": 40032,
      "training_step_time": 9.193883419036865
    },
    {
      "epoch": 3.817844390869141e-05,
      "model_forward_time": 1.4062178134918213,
      "step": 40033
    },
    {
      "epoch": 3.817844390869141e-05,
      "step": 40033,
      "training_step_time": 7.7798590660095215
    },
    {
      "epoch": 3.817939758300781e-05,
      "model_forward_time": 1.7426953315734863,
      "step": 40034
    },
    {
      "epoch": 3.817939758300781e-05,
      "step": 40034,
      "training_step_time": 7.635776042938232
    },
    {
      "epoch": 3.8180351257324215e-05,
      "model_forward_time": 1.761718988418579,
      "step": 40035
    },
    {
      "epoch": 3.8180351257324215e-05,
      "step": 40035,
      "training_step_time": 7.370714902877808
    },
    {
      "epoch": 3.8181304931640626e-05,
      "model_forward_time": 1.7914836406707764,
      "step": 40036
    },
    {
      "epoch": 3.8181304931640626e-05,
      "step": 40036,
      "training_step_time": 7.6189351081848145
    },
    {
      "epoch": 3.818225860595703e-05,
      "model_forward_time": 1.7636029720306396,
      "step": 40037
    },
    {
      "epoch": 3.818225860595703e-05,
      "step": 40037,
      "training_step_time": 7.639125108718872
    },
    {
      "epoch": 3.8183212280273434e-05,
      "model_forward_time": 1.765359878540039,
      "step": 40038
    },
    {
      "epoch": 3.8183212280273434e-05,
      "step": 40038,
      "training_step_time": 8.027120113372803
    },
    {
      "epoch": 3.8184165954589845e-05,
      "model_forward_time": 1.772994041442871,
      "step": 40039
    },
    {
      "epoch": 3.8184165954589845e-05,
      "step": 40039,
      "training_step_time": 8.884566068649292
    },
    {
      "epoch": 3.818511962890625e-05,
      "grad_norm": 0.09699050337076187,
      "learning_rate": 7.002570060642976e-06,
      "loss": 0.058,
      "step": 40040
    },
    {
      "epoch": 3.818511962890625e-05,
      "model_forward_time": 2.8808634281158447,
      "step": 40040
    },
    {
      "dynamics_loss_avg": 0.0668860947713256,
      "epoch": 3.818511962890625e-05,
      "step": 40040
    },
    {
      "action_loss_avg": 0.0021792388404719533,
      "epoch": 3.818511962890625e-05,
      "step": 40040
    },
    {
      "epoch": 3.818511962890625e-05,
      "step": 40040,
      "training_step_time": 9.96810531616211
    },
    {
      "epoch": 3.818607330322265e-05,
      "model_forward_time": 1.7558197975158691,
      "step": 40041
    },
    {
      "epoch": 3.818607330322265e-05,
      "step": 40041,
      "training_step_time": 8.762816667556763
    },
    {
      "epoch": 3.8187026977539064e-05,
      "model_forward_time": 1.7540311813354492,
      "step": 40042
    },
    {
      "epoch": 3.8187026977539064e-05,
      "step": 40042,
      "training_step_time": 7.609031677246094
    },
    {
      "epoch": 3.818798065185547e-05,
      "model_forward_time": 1.7478389739990234,
      "step": 40043
    },
    {
      "epoch": 3.818798065185547e-05,
      "step": 40043,
      "training_step_time": 7.4915196895599365
    },
    {
      "epoch": 3.818893432617187e-05,
      "model_forward_time": 1.7582359313964844,
      "step": 40044
    },
    {
      "epoch": 3.818893432617187e-05,
      "step": 40044,
      "training_step_time": 7.247716426849365
    },
    {
      "epoch": 3.818988800048828e-05,
      "model_forward_time": 1.7600152492523193,
      "step": 40045
    },
    {
      "epoch": 3.818988800048828e-05,
      "step": 40045,
      "training_step_time": 7.281855344772339
    },
    {
      "epoch": 3.819084167480469e-05,
      "model_forward_time": 1.7539865970611572,
      "step": 40046
    },
    {
      "epoch": 3.819084167480469e-05,
      "step": 40046,
      "training_step_time": 8.547570943832397
    },
    {
      "epoch": 3.819179534912109e-05,
      "model_forward_time": 1.7654399871826172,
      "step": 40047
    },
    {
      "epoch": 3.819179534912109e-05,
      "step": 40047,
      "training_step_time": 7.294746398925781
    },
    {
      "epoch": 3.81927490234375e-05,
      "model_forward_time": 1.7489802837371826,
      "step": 40048
    },
    {
      "epoch": 3.81927490234375e-05,
      "step": 40048,
      "training_step_time": 9.0053870677948
    },
    {
      "epoch": 3.8193702697753906e-05,
      "model_forward_time": 2.8607046604156494,
      "step": 40049
    },
    {
      "epoch": 3.8193702697753906e-05,
      "step": 40049,
      "training_step_time": 8.51083493232727
    },
    {
      "epoch": 3.819465637207031e-05,
      "grad_norm": 0.19752497971057892,
      "learning_rate": 7.001054892263903e-06,
      "loss": 0.0591,
      "step": 40050
    },
    {
      "epoch": 3.819465637207031e-05,
      "model_forward_time": 1.7587463855743408,
      "step": 40050
    },
    {
      "dynamics_loss_avg": 0.0446666719391942,
      "epoch": 3.819465637207031e-05,
      "step": 40050
    },
    {
      "action_loss_avg": 0.00167173154768534,
      "epoch": 3.819465637207031e-05,
      "step": 40050
    },
    {
      "epoch": 3.819465637207031e-05,
      "step": 40050,
      "training_step_time": 9.035003900527954
    },
    {
      "epoch": 3.819561004638672e-05,
      "model_forward_time": 1.7493782043457031,
      "step": 40051
    },
    {
      "epoch": 3.819561004638672e-05,
      "step": 40051,
      "training_step_time": 7.565821170806885
    },
    {
      "epoch": 3.8196563720703125e-05,
      "model_forward_time": 1.7397003173828125,
      "step": 40052
    },
    {
      "epoch": 3.8196563720703125e-05,
      "step": 40052,
      "training_step_time": 7.5120909214019775
    },
    {
      "epoch": 3.819751739501953e-05,
      "model_forward_time": 1.7660975456237793,
      "step": 40053
    },
    {
      "epoch": 3.819751739501953e-05,
      "step": 40053,
      "training_step_time": 7.246314287185669
    },
    {
      "epoch": 3.819847106933594e-05,
      "model_forward_time": 1.7785580158233643,
      "step": 40054
    },
    {
      "epoch": 3.819847106933594e-05,
      "step": 40054,
      "training_step_time": 6.7676613330841064
    },
    {
      "epoch": 3.8199424743652343e-05,
      "model_forward_time": 1.77767014503479,
      "step": 40055
    },
    {
      "epoch": 3.8199424743652343e-05,
      "step": 40055,
      "training_step_time": 7.65023946762085
    },
    {
      "epoch": 3.820037841796875e-05,
      "model_forward_time": 1.757683515548706,
      "step": 40056
    },
    {
      "epoch": 3.820037841796875e-05,
      "step": 40056,
      "training_step_time": 7.56015157699585
    },
    {
      "epoch": 3.820133209228516e-05,
      "model_forward_time": 1.7465200424194336,
      "step": 40057
    },
    {
      "epoch": 3.820133209228516e-05,
      "step": 40057,
      "training_step_time": 9.296535730361938
    },
    {
      "epoch": 3.820228576660156e-05,
      "model_forward_time": 2.904351234436035,
      "step": 40058
    },
    {
      "epoch": 3.820228576660156e-05,
      "step": 40058,
      "training_step_time": 9.6781485080719
    },
    {
      "epoch": 3.8203239440917966e-05,
      "model_forward_time": 1.7694718837738037,
      "step": 40059
    },
    {
      "epoch": 3.8203239440917966e-05,
      "step": 40059,
      "training_step_time": 9.361030578613281
    },
    {
      "epoch": 3.820419311523438e-05,
      "grad_norm": 0.15707609057426453,
      "learning_rate": 6.999539505052477e-06,
      "loss": 0.0578,
      "step": 40060
    },
    {
      "epoch": 3.820419311523438e-05,
      "model_forward_time": 1.7629516124725342,
      "step": 40060
    },
    {
      "dynamics_loss_avg": 0.05186712332069874,
      "epoch": 3.820419311523438e-05,
      "step": 40060
    },
    {
      "action_loss_avg": 0.0016714659694116562,
      "epoch": 3.820419311523438e-05,
      "step": 40060
    },
    {
      "epoch": 3.820419311523438e-05,
      "step": 40060,
      "training_step_time": 9.144315242767334
    },
    {
      "epoch": 3.820514678955078e-05,
      "model_forward_time": 1.7520153522491455,
      "step": 40061
    },
    {
      "epoch": 3.820514678955078e-05,
      "step": 40061,
      "training_step_time": 7.726436614990234
    },
    {
      "epoch": 3.8206100463867185e-05,
      "model_forward_time": 1.7613856792449951,
      "step": 40062
    },
    {
      "epoch": 3.8206100463867185e-05,
      "step": 40062,
      "training_step_time": 7.305810213088989
    },
    {
      "epoch": 3.8207054138183596e-05,
      "model_forward_time": 1.3953773975372314,
      "step": 40063
    },
    {
      "epoch": 3.8207054138183596e-05,
      "step": 40063,
      "training_step_time": 6.842525243759155
    },
    {
      "epoch": 3.82080078125e-05,
      "model_forward_time": 1.753359079360962,
      "step": 40064
    },
    {
      "epoch": 3.82080078125e-05,
      "step": 40064,
      "training_step_time": 7.6182026863098145
    },
    {
      "epoch": 3.8208961486816404e-05,
      "model_forward_time": 1.7587594985961914,
      "step": 40065
    },
    {
      "epoch": 3.8208961486816404e-05,
      "step": 40065,
      "training_step_time": 9.728851318359375
    },
    {
      "epoch": 3.8209915161132815e-05,
      "model_forward_time": 1.7504925727844238,
      "step": 40066
    },
    {
      "epoch": 3.8209915161132815e-05,
      "step": 40066,
      "training_step_time": 8.313074827194214
    },
    {
      "epoch": 3.821086883544922e-05,
      "model_forward_time": 2.870729923248291,
      "step": 40067
    },
    {
      "epoch": 3.821086883544922e-05,
      "step": 40067,
      "training_step_time": 9.60171914100647
    },
    {
      "epoch": 3.821182250976562e-05,
      "model_forward_time": 1.762101173400879,
      "step": 40068
    },
    {
      "epoch": 3.821182250976562e-05,
      "step": 40068,
      "training_step_time": 8.380563497543335
    },
    {
      "epoch": 3.8212776184082034e-05,
      "model_forward_time": 1.7510380744934082,
      "step": 40069
    },
    {
      "epoch": 3.8212776184082034e-05,
      "step": 40069,
      "training_step_time": 8.446370363235474
    },
    {
      "epoch": 3.821372985839844e-05,
      "grad_norm": 0.08433374017477036,
      "learning_rate": 6.998023899174418e-06,
      "loss": 0.0618,
      "step": 40070
    },
    {
      "epoch": 3.821372985839844e-05,
      "model_forward_time": 1.7575972080230713,
      "step": 40070
    },
    {
      "dynamics_loss_avg": 0.05442626196891069,
      "epoch": 3.821372985839844e-05,
      "step": 40070
    },
    {
      "action_loss_avg": 0.0015881230880040675,
      "epoch": 3.821372985839844e-05,
      "step": 40070
    },
    {
      "epoch": 3.821372985839844e-05,
      "step": 40070,
      "training_step_time": 7.707903623580933
    },
    {
      "epoch": 3.821468353271484e-05,
      "model_forward_time": 1.7618918418884277,
      "step": 40071
    },
    {
      "epoch": 3.821468353271484e-05,
      "step": 40071,
      "training_step_time": 7.274704217910767
    },
    {
      "epoch": 3.821563720703125e-05,
      "model_forward_time": 1.7631113529205322,
      "step": 40072
    },
    {
      "epoch": 3.821563720703125e-05,
      "step": 40072,
      "training_step_time": 6.8047871589660645
    },
    {
      "epoch": 3.821659088134766e-05,
      "model_forward_time": 1.7683210372924805,
      "step": 40073
    },
    {
      "epoch": 3.821659088134766e-05,
      "step": 40073,
      "training_step_time": 7.603468418121338
    },
    {
      "epoch": 3.821754455566406e-05,
      "model_forward_time": 1.7642722129821777,
      "step": 40074
    },
    {
      "epoch": 3.821754455566406e-05,
      "step": 40074,
      "training_step_time": 7.757063388824463
    },
    {
      "epoch": 3.821849822998047e-05,
      "model_forward_time": 1.759833574295044,
      "step": 40075
    },
    {
      "epoch": 3.821849822998047e-05,
      "step": 40075,
      "training_step_time": 8.369585752487183
    },
    {
      "epoch": 3.8219451904296875e-05,
      "model_forward_time": 1.768491268157959,
      "step": 40076
    },
    {
      "epoch": 3.8219451904296875e-05,
      "step": 40076,
      "training_step_time": 10.78274154663086
    },
    {
      "epoch": 3.822040557861328e-05,
      "model_forward_time": 2.901459217071533,
      "step": 40077
    },
    {
      "epoch": 3.822040557861328e-05,
      "step": 40077,
      "training_step_time": 9.627864122390747
    },
    {
      "epoch": 3.822135925292969e-05,
      "model_forward_time": 1.7806613445281982,
      "step": 40078
    },
    {
      "epoch": 3.822135925292969e-05,
      "step": 40078,
      "training_step_time": 7.739694595336914
    },
    {
      "epoch": 3.8222312927246094e-05,
      "model_forward_time": 1.7865278720855713,
      "step": 40079
    },
    {
      "epoch": 3.8222312927246094e-05,
      "step": 40079,
      "training_step_time": 7.688100576400757
    },
    {
      "epoch": 3.82232666015625e-05,
      "grad_norm": 0.11645407229661942,
      "learning_rate": 6.996508074795467e-06,
      "loss": 0.0628,
      "step": 40080
    },
    {
      "epoch": 3.82232666015625e-05,
      "model_forward_time": 1.759687900543213,
      "step": 40080
    },
    {
      "dynamics_loss_avg": 0.05936150886118412,
      "epoch": 3.82232666015625e-05,
      "step": 40080
    },
    {
      "action_loss_avg": 0.0023257009277585896,
      "epoch": 3.82232666015625e-05,
      "step": 40080
    },
    {
      "epoch": 3.82232666015625e-05,
      "step": 40080,
      "training_step_time": 7.281494140625
    },
    {
      "epoch": 3.822422027587891e-05,
      "model_forward_time": 1.770801305770874,
      "step": 40081
    },
    {
      "epoch": 3.822422027587891e-05,
      "step": 40081,
      "training_step_time": 6.873565912246704
    },
    {
      "epoch": 3.822517395019531e-05,
      "model_forward_time": 1.7777400016784668,
      "step": 40082
    },
    {
      "epoch": 3.822517395019531e-05,
      "step": 40082,
      "training_step_time": 7.59544825553894
    },
    {
      "epoch": 3.822612762451172e-05,
      "model_forward_time": 1.7779102325439453,
      "step": 40083
    },
    {
      "epoch": 3.822612762451172e-05,
      "step": 40083,
      "training_step_time": 7.758224964141846
    },
    {
      "epoch": 3.822708129882813e-05,
      "model_forward_time": 1.7780213356018066,
      "step": 40084
    },
    {
      "epoch": 3.822708129882813e-05,
      "step": 40084,
      "training_step_time": 7.934897184371948
    },
    {
      "epoch": 3.822803497314453e-05,
      "model_forward_time": 1.7640371322631836,
      "step": 40085
    },
    {
      "epoch": 3.822803497314453e-05,
      "step": 40085,
      "training_step_time": 8.520635604858398
    },
    {
      "epoch": 3.8228988647460936e-05,
      "model_forward_time": 2.8811848163604736,
      "step": 40086
    },
    {
      "epoch": 3.8228988647460936e-05,
      "step": 40086,
      "training_step_time": 9.281128168106079
    },
    {
      "epoch": 3.822994232177735e-05,
      "model_forward_time": 1.7754108905792236,
      "step": 40087
    },
    {
      "epoch": 3.822994232177735e-05,
      "step": 40087,
      "training_step_time": 9.485026836395264
    },
    {
      "epoch": 3.823089599609375e-05,
      "model_forward_time": 1.7423183917999268,
      "step": 40088
    },
    {
      "epoch": 3.823089599609375e-05,
      "step": 40088,
      "training_step_time": 8.494747638702393
    },
    {
      "epoch": 3.8231849670410155e-05,
      "model_forward_time": 1.7711834907531738,
      "step": 40089
    },
    {
      "epoch": 3.8231849670410155e-05,
      "step": 40089,
      "training_step_time": 7.1864519119262695
    },
    {
      "epoch": 3.8232803344726566e-05,
      "grad_norm": 0.09960456192493439,
      "learning_rate": 6.9949920320813955e-06,
      "loss": 0.0628,
      "step": 40090
    },
    {
      "epoch": 3.8232803344726566e-05,
      "model_forward_time": 1.7554256916046143,
      "step": 40090
    },
    {
      "dynamics_loss_avg": 0.06319529823958873,
      "epoch": 3.8232803344726566e-05,
      "step": 40090
    },
    {
      "action_loss_avg": 0.0021050289273262025,
      "epoch": 3.8232803344726566e-05,
      "step": 40090
    },
    {
      "epoch": 3.8232803344726566e-05,
      "step": 40090,
      "training_step_time": 6.781784772872925
    },
    {
      "epoch": 3.823375701904297e-05,
      "model_forward_time": 1.7551758289337158,
      "step": 40091
    },
    {
      "epoch": 3.823375701904297e-05,
      "step": 40091,
      "training_step_time": 7.632262468338013
    },
    {
      "epoch": 3.8234710693359374e-05,
      "model_forward_time": 1.7620508670806885,
      "step": 40092
    },
    {
      "epoch": 3.8234710693359374e-05,
      "step": 40092,
      "training_step_time": 8.582571744918823
    },
    {
      "epoch": 3.8235664367675785e-05,
      "model_forward_time": 1.765094518661499,
      "step": 40093
    },
    {
      "epoch": 3.8235664367675785e-05,
      "step": 40093,
      "training_step_time": 7.342820405960083
    },
    {
      "epoch": 3.823661804199219e-05,
      "model_forward_time": 1.759758472442627,
      "step": 40094
    },
    {
      "epoch": 3.823661804199219e-05,
      "step": 40094,
      "training_step_time": 7.5871007442474365
    },
    {
      "epoch": 3.823757171630859e-05,
      "model_forward_time": 2.9121854305267334,
      "step": 40095
    },
    {
      "epoch": 3.823757171630859e-05,
      "step": 40095,
      "training_step_time": 9.665224313735962
    },
    {
      "epoch": 3.8238525390625e-05,
      "model_forward_time": 1.7529935836791992,
      "step": 40096
    },
    {
      "epoch": 3.8238525390625e-05,
      "step": 40096,
      "training_step_time": 8.657904863357544
    },
    {
      "epoch": 3.823947906494141e-05,
      "model_forward_time": 1.7566213607788086,
      "step": 40097
    },
    {
      "epoch": 3.823947906494141e-05,
      "step": 40097,
      "training_step_time": 7.779088020324707
    },
    {
      "epoch": 3.824043273925781e-05,
      "model_forward_time": 1.750673770904541,
      "step": 40098
    },
    {
      "epoch": 3.824043273925781e-05,
      "step": 40098,
      "training_step_time": 7.663310527801514
    },
    {
      "epoch": 3.8241386413574216e-05,
      "model_forward_time": 1.7646327018737793,
      "step": 40099
    },
    {
      "epoch": 3.8241386413574216e-05,
      "step": 40099,
      "training_step_time": 7.317834854125977
    },
    {
      "epoch": 3.8242340087890626e-05,
      "grad_norm": 0.12201914936304092,
      "learning_rate": 6.993475771197995e-06,
      "loss": 0.0619,
      "step": 40100
    },
    {
      "epoch": 3.8242340087890626e-05,
      "model_forward_time": 1.7587275505065918,
      "step": 40100
    },
    {
      "dynamics_loss_avg": 0.04816620275378227,
      "epoch": 3.8242340087890626e-05,
      "step": 40100
    },
    {
      "action_loss_avg": 0.0021156000264454635,
      "epoch": 3.8242340087890626e-05,
      "step": 40100
    },
    {
      "epoch": 3.8242340087890626e-05,
      "step": 40100,
      "training_step_time": 7.6135993003845215
    },
    {
      "epoch": 3.824329376220703e-05,
      "model_forward_time": 1.7438561916351318,
      "step": 40101
    },
    {
      "epoch": 3.824329376220703e-05,
      "step": 40101,
      "training_step_time": 7.423543214797974
    },
    {
      "epoch": 3.8244247436523434e-05,
      "model_forward_time": 1.7609949111938477,
      "step": 40102
    },
    {
      "epoch": 3.8244247436523434e-05,
      "step": 40102,
      "training_step_time": 7.8103718757629395
    },
    {
      "epoch": 3.8245201110839845e-05,
      "model_forward_time": 1.0688021183013916,
      "step": 40103
    },
    {
      "epoch": 3.8245201110839845e-05,
      "step": 40103,
      "training_step_time": 8.395416259765625
    },
    {
      "epoch": 3.824615478515625e-05,
      "model_forward_time": 1.792905330657959,
      "step": 40104
    },
    {
      "epoch": 3.824615478515625e-05,
      "step": 40104,
      "training_step_time": 8.289758682250977
    },
    {
      "epoch": 3.824710845947265e-05,
      "model_forward_time": 1.757838487625122,
      "step": 40105
    },
    {
      "epoch": 3.824710845947265e-05,
      "step": 40105,
      "training_step_time": 9.751049757003784
    },
    {
      "epoch": 3.8248062133789064e-05,
      "model_forward_time": 1.7577993869781494,
      "step": 40106
    },
    {
      "epoch": 3.8248062133789064e-05,
      "step": 40106,
      "training_step_time": 7.689811706542969
    },
    {
      "epoch": 3.824901580810547e-05,
      "model_forward_time": 1.7594218254089355,
      "step": 40107
    },
    {
      "epoch": 3.824901580810547e-05,
      "step": 40107,
      "training_step_time": 8.361103534698486
    },
    {
      "epoch": 3.824996948242187e-05,
      "model_forward_time": 1.7650160789489746,
      "step": 40108
    },
    {
      "epoch": 3.824996948242187e-05,
      "step": 40108,
      "training_step_time": 7.484959363937378
    },
    {
      "epoch": 3.825092315673828e-05,
      "model_forward_time": 1.7581000328063965,
      "step": 40109
    },
    {
      "epoch": 3.825092315673828e-05,
      "step": 40109,
      "training_step_time": 7.628711223602295
    },
    {
      "epoch": 3.825187683105469e-05,
      "grad_norm": 0.12618473172187805,
      "learning_rate": 6.991959292311082e-06,
      "loss": 0.0603,
      "step": 40110
    },
    {
      "epoch": 3.825187683105469e-05,
      "model_forward_time": 1.7571918964385986,
      "step": 40110
    },
    {
      "dynamics_loss_avg": 0.0611165015026927,
      "epoch": 3.825187683105469e-05,
      "step": 40110
    },
    {
      "action_loss_avg": 0.003277822176460177,
      "epoch": 3.825187683105469e-05,
      "step": 40110
    },
    {
      "epoch": 3.825187683105469e-05,
      "step": 40110,
      "training_step_time": 7.565831661224365
    },
    {
      "epoch": 3.825283050537109e-05,
      "model_forward_time": 1.7522170543670654,
      "step": 40111
    },
    {
      "epoch": 3.825283050537109e-05,
      "step": 40111,
      "training_step_time": 7.607641220092773
    },
    {
      "epoch": 3.82537841796875e-05,
      "model_forward_time": 1.755321979522705,
      "step": 40112
    },
    {
      "epoch": 3.82537841796875e-05,
      "step": 40112,
      "training_step_time": 8.144416332244873
    },
    {
      "epoch": 3.8254737854003906e-05,
      "model_forward_time": 1.7779691219329834,
      "step": 40113
    },
    {
      "epoch": 3.8254737854003906e-05,
      "step": 40113,
      "training_step_time": 7.806347608566284
    },
    {
      "epoch": 3.825569152832031e-05,
      "model_forward_time": 2.515211343765259,
      "step": 40114
    },
    {
      "epoch": 3.825569152832031e-05,
      "step": 40114,
      "training_step_time": 8.280637502670288
    },
    {
      "epoch": 3.825664520263672e-05,
      "model_forward_time": 1.758730411529541,
      "step": 40115
    },
    {
      "epoch": 3.825664520263672e-05,
      "step": 40115,
      "training_step_time": 9.34938669204712
    },
    {
      "epoch": 3.8257598876953125e-05,
      "model_forward_time": 1.7646560668945312,
      "step": 40116
    },
    {
      "epoch": 3.8257598876953125e-05,
      "step": 40116,
      "training_step_time": 7.630387544631958
    },
    {
      "epoch": 3.825855255126953e-05,
      "model_forward_time": 1.7574207782745361,
      "step": 40117
    },
    {
      "epoch": 3.825855255126953e-05,
      "step": 40117,
      "training_step_time": 7.385270833969116
    },
    {
      "epoch": 3.825950622558594e-05,
      "model_forward_time": 1.7637951374053955,
      "step": 40118
    },
    {
      "epoch": 3.825950622558594e-05,
      "step": 40118,
      "training_step_time": 7.6993513107299805
    },
    {
      "epoch": 3.8260459899902344e-05,
      "model_forward_time": 1.7541382312774658,
      "step": 40119
    },
    {
      "epoch": 3.8260459899902344e-05,
      "step": 40119,
      "training_step_time": 7.573235750198364
    },
    {
      "epoch": 3.826141357421875e-05,
      "grad_norm": 0.18090815842151642,
      "learning_rate": 6.990442595586495e-06,
      "loss": 0.0613,
      "step": 40120
    },
    {
      "epoch": 3.826141357421875e-05,
      "model_forward_time": 1.751777172088623,
      "step": 40120
    },
    {
      "dynamics_loss_avg": 0.05754067208617926,
      "epoch": 3.826141357421875e-05,
      "step": 40120
    },
    {
      "action_loss_avg": 0.0020583455421729012,
      "epoch": 3.826141357421875e-05,
      "step": 40120
    },
    {
      "epoch": 3.826141357421875e-05,
      "step": 40120,
      "training_step_time": 7.629296541213989
    },
    {
      "epoch": 3.826236724853516e-05,
      "model_forward_time": 1.7551419734954834,
      "step": 40121
    },
    {
      "epoch": 3.826236724853516e-05,
      "step": 40121,
      "training_step_time": 7.5684874057769775
    },
    {
      "epoch": 3.826332092285156e-05,
      "model_forward_time": 1.7895784378051758,
      "step": 40122
    },
    {
      "epoch": 3.826332092285156e-05,
      "step": 40122,
      "training_step_time": 9.445223569869995
    },
    {
      "epoch": 3.8264274597167966e-05,
      "model_forward_time": 2.8822267055511475,
      "step": 40123
    },
    {
      "epoch": 3.8264274597167966e-05,
      "step": 40123,
      "training_step_time": 7.809933185577393
    },
    {
      "epoch": 3.826522827148438e-05,
      "model_forward_time": 1.7619774341583252,
      "step": 40124
    },
    {
      "epoch": 3.826522827148438e-05,
      "step": 40124,
      "training_step_time": 8.639456987380981
    },
    {
      "epoch": 3.826618194580078e-05,
      "model_forward_time": 1.7565104961395264,
      "step": 40125
    },
    {
      "epoch": 3.826618194580078e-05,
      "step": 40125,
      "training_step_time": 8.258858919143677
    },
    {
      "epoch": 3.8267135620117185e-05,
      "model_forward_time": 1.7542426586151123,
      "step": 40126
    },
    {
      "epoch": 3.8267135620117185e-05,
      "step": 40126,
      "training_step_time": 7.480952978134155
    },
    {
      "epoch": 3.8268089294433596e-05,
      "model_forward_time": 1.7567689418792725,
      "step": 40127
    },
    {
      "epoch": 3.8268089294433596e-05,
      "step": 40127,
      "training_step_time": 7.3002824783325195
    },
    {
      "epoch": 3.826904296875e-05,
      "model_forward_time": 1.7597572803497314,
      "step": 40128
    },
    {
      "epoch": 3.826904296875e-05,
      "step": 40128,
      "training_step_time": 7.435733795166016
    },
    {
      "epoch": 3.8269996643066404e-05,
      "model_forward_time": 1.7606441974639893,
      "step": 40129
    },
    {
      "epoch": 3.8269996643066404e-05,
      "step": 40129,
      "training_step_time": 7.906826496124268
    },
    {
      "epoch": 3.8270950317382815e-05,
      "grad_norm": 0.19992896914482117,
      "learning_rate": 6.988925681190098e-06,
      "loss": 0.0619,
      "step": 40130
    },
    {
      "epoch": 3.8270950317382815e-05,
      "model_forward_time": 1.7636005878448486,
      "step": 40130
    },
    {
      "dynamics_loss_avg": 0.05618564747273922,
      "epoch": 3.8270950317382815e-05,
      "step": 40130
    },
    {
      "action_loss_avg": 0.0024360655050259085,
      "epoch": 3.8270950317382815e-05,
      "step": 40130
    },
    {
      "epoch": 3.8270950317382815e-05,
      "step": 40130,
      "training_step_time": 7.679007530212402
    },
    {
      "epoch": 3.827190399169922e-05,
      "model_forward_time": 1.760345458984375,
      "step": 40131
    },
    {
      "epoch": 3.827190399169922e-05,
      "step": 40131,
      "training_step_time": 7.863550662994385
    },
    {
      "epoch": 3.827285766601562e-05,
      "model_forward_time": 2.8797202110290527,
      "step": 40132
    },
    {
      "epoch": 3.827285766601562e-05,
      "step": 40132,
      "training_step_time": 8.462993621826172
    },
    {
      "epoch": 3.8273811340332034e-05,
      "model_forward_time": 1.7603495121002197,
      "step": 40133
    },
    {
      "epoch": 3.8273811340332034e-05,
      "step": 40133,
      "training_step_time": 9.537806510925293
    },
    {
      "epoch": 3.827476501464844e-05,
      "model_forward_time": 1.7526612281799316,
      "step": 40134
    },
    {
      "epoch": 3.827476501464844e-05,
      "step": 40134,
      "training_step_time": 7.736123323440552
    },
    {
      "epoch": 3.827571868896484e-05,
      "model_forward_time": 1.759885311126709,
      "step": 40135
    },
    {
      "epoch": 3.827571868896484e-05,
      "step": 40135,
      "training_step_time": 7.447294235229492
    },
    {
      "epoch": 3.827667236328125e-05,
      "model_forward_time": 1.761277675628662,
      "step": 40136
    },
    {
      "epoch": 3.827667236328125e-05,
      "step": 40136,
      "training_step_time": 7.317229986190796
    },
    {
      "epoch": 3.827762603759766e-05,
      "model_forward_time": 1.7592496871948242,
      "step": 40137
    },
    {
      "epoch": 3.827762603759766e-05,
      "step": 40137,
      "training_step_time": 7.127390384674072
    },
    {
      "epoch": 3.827857971191406e-05,
      "model_forward_time": 1.7591681480407715,
      "step": 40138
    },
    {
      "epoch": 3.827857971191406e-05,
      "step": 40138,
      "training_step_time": 7.452028512954712
    },
    {
      "epoch": 3.827953338623047e-05,
      "model_forward_time": 1.7642285823822021,
      "step": 40139
    },
    {
      "epoch": 3.827953338623047e-05,
      "step": 40139,
      "training_step_time": 7.729156494140625
    },
    {
      "epoch": 3.8280487060546876e-05,
      "grad_norm": 0.17116676270961761,
      "learning_rate": 6.987408549287778e-06,
      "loss": 0.0613,
      "step": 40140
    },
    {
      "epoch": 3.8280487060546876e-05,
      "model_forward_time": 1.7522077560424805,
      "step": 40140
    },
    {
      "dynamics_loss_avg": 0.05860412549227476,
      "epoch": 3.8280487060546876e-05,
      "step": 40140
    },
    {
      "action_loss_avg": 0.001998139556962997,
      "epoch": 3.8280487060546876e-05,
      "step": 40140
    },
    {
      "epoch": 3.8280487060546876e-05,
      "step": 40140,
      "training_step_time": 8.458518028259277
    },
    {
      "epoch": 3.828144073486328e-05,
      "model_forward_time": 2.8545479774475098,
      "step": 40141
    },
    {
      "epoch": 3.828144073486328e-05,
      "step": 40141,
      "training_step_time": 8.505974292755127
    },
    {
      "epoch": 3.828239440917969e-05,
      "model_forward_time": 1.7609477043151855,
      "step": 40142
    },
    {
      "epoch": 3.828239440917969e-05,
      "step": 40142,
      "training_step_time": 8.381772756576538
    },
    {
      "epoch": 3.8283348083496094e-05,
      "model_forward_time": 1.7581686973571777,
      "step": 40143
    },
    {
      "epoch": 3.8283348083496094e-05,
      "step": 40143,
      "training_step_time": 9.233271837234497
    },
    {
      "epoch": 3.82843017578125e-05,
      "model_forward_time": 1.7596654891967773,
      "step": 40144
    },
    {
      "epoch": 3.82843017578125e-05,
      "step": 40144,
      "training_step_time": 7.965179920196533
    },
    {
      "epoch": 3.828525543212891e-05,
      "model_forward_time": 1.755422830581665,
      "step": 40145
    },
    {
      "epoch": 3.828525543212891e-05,
      "step": 40145,
      "training_step_time": 6.8934125900268555
    },
    {
      "epoch": 3.828620910644531e-05,
      "model_forward_time": 1.7550957202911377,
      "step": 40146
    },
    {
      "epoch": 3.828620910644531e-05,
      "step": 40146,
      "training_step_time": 7.085845947265625
    },
    {
      "epoch": 3.828716278076172e-05,
      "model_forward_time": 1.7579097747802734,
      "step": 40147
    },
    {
      "epoch": 3.828716278076172e-05,
      "step": 40147,
      "training_step_time": 7.974097013473511
    },
    {
      "epoch": 3.828811645507813e-05,
      "model_forward_time": 1.7535812854766846,
      "step": 40148
    },
    {
      "epoch": 3.828811645507813e-05,
      "step": 40148,
      "training_step_time": 8.391071319580078
    },
    {
      "epoch": 3.828907012939453e-05,
      "model_forward_time": 1.763892650604248,
      "step": 40149
    },
    {
      "epoch": 3.828907012939453e-05,
      "step": 40149,
      "training_step_time": 7.794022083282471
    },
    {
      "epoch": 3.8290023803710936e-05,
      "grad_norm": 0.1204734742641449,
      "learning_rate": 6.985891200045449e-06,
      "loss": 0.0581,
      "step": 40150
    },
    {
      "epoch": 3.8290023803710936e-05,
      "model_forward_time": 2.895595073699951,
      "step": 40150
    },
    {
      "dynamics_loss_avg": 0.04980709189549089,
      "epoch": 3.8290023803710936e-05,
      "step": 40150
    },
    {
      "action_loss_avg": 0.0020539890974760056,
      "epoch": 3.8290023803710936e-05,
      "step": 40150
    },
    {
      "epoch": 3.8290023803710936e-05,
      "step": 40150,
      "training_step_time": 9.10661268234253
    },
    {
      "epoch": 3.829097747802735e-05,
      "model_forward_time": 1.7672348022460938,
      "step": 40151
    },
    {
      "epoch": 3.829097747802735e-05,
      "step": 40151,
      "training_step_time": 9.380343914031982
    },
    {
      "epoch": 3.829193115234375e-05,
      "model_forward_time": 1.7663443088531494,
      "step": 40152
    },
    {
      "epoch": 3.829193115234375e-05,
      "step": 40152,
      "training_step_time": 9.036867141723633
    },
    {
      "epoch": 3.8292884826660155e-05,
      "model_forward_time": 1.7605104446411133,
      "step": 40153
    },
    {
      "epoch": 3.8292884826660155e-05,
      "step": 40153,
      "training_step_time": 7.644670009613037
    },
    {
      "epoch": 3.8293838500976566e-05,
      "model_forward_time": 1.7659120559692383,
      "step": 40154
    },
    {
      "epoch": 3.8293838500976566e-05,
      "step": 40154,
      "training_step_time": 8.096430778503418
    },
    {
      "epoch": 3.829479217529297e-05,
      "model_forward_time": 1.7737791538238525,
      "step": 40155
    },
    {
      "epoch": 3.829479217529297e-05,
      "step": 40155,
      "training_step_time": 7.240354061126709
    },
    {
      "epoch": 3.8295745849609374e-05,
      "model_forward_time": 1.756986141204834,
      "step": 40156
    },
    {
      "epoch": 3.8295745849609374e-05,
      "step": 40156,
      "training_step_time": 7.416675090789795
    },
    {
      "epoch": 3.829669952392578e-05,
      "model_forward_time": 1.7618513107299805,
      "step": 40157
    },
    {
      "epoch": 3.829669952392578e-05,
      "step": 40157,
      "training_step_time": 7.681441783905029
    },
    {
      "epoch": 3.829765319824219e-05,
      "model_forward_time": 1.76645827293396,
      "step": 40158
    },
    {
      "epoch": 3.829765319824219e-05,
      "step": 40158,
      "training_step_time": 8.867786645889282
    },
    {
      "epoch": 3.829860687255859e-05,
      "model_forward_time": 1.7654368877410889,
      "step": 40159
    },
    {
      "epoch": 3.829860687255859e-05,
      "step": 40159,
      "training_step_time": 9.346243858337402
    },
    {
      "epoch": 3.8299560546875e-05,
      "grad_norm": 0.19028975069522858,
      "learning_rate": 6.984373633629045e-06,
      "loss": 0.0582,
      "step": 40160
    },
    {
      "epoch": 3.8299560546875e-05,
      "model_forward_time": 1.77030611038208,
      "step": 40160
    },
    {
      "dynamics_loss_avg": 0.05148938223719597,
      "epoch": 3.8299560546875e-05,
      "step": 40160
    },
    {
      "action_loss_avg": 0.002726231957785785,
      "epoch": 3.8299560546875e-05,
      "step": 40160
    },
    {
      "epoch": 3.8299560546875e-05,
      "step": 40160,
      "training_step_time": 7.792606830596924
    },
    {
      "epoch": 3.830051422119141e-05,
      "model_forward_time": 1.7593748569488525,
      "step": 40161
    },
    {
      "epoch": 3.830051422119141e-05,
      "step": 40161,
      "training_step_time": 7.67892861366272
    },
    {
      "epoch": 3.830146789550781e-05,
      "model_forward_time": 1.7509398460388184,
      "step": 40162
    },
    {
      "epoch": 3.830146789550781e-05,
      "step": 40162,
      "training_step_time": 7.655617952346802
    },
    {
      "epoch": 3.8302421569824216e-05,
      "model_forward_time": 1.7608888149261475,
      "step": 40163
    },
    {
      "epoch": 3.8302421569824216e-05,
      "step": 40163,
      "training_step_time": 7.420710563659668
    },
    {
      "epoch": 3.8303375244140626e-05,
      "model_forward_time": 1.750382900238037,
      "step": 40164
    },
    {
      "epoch": 3.8303375244140626e-05,
      "step": 40164,
      "training_step_time": 7.305366516113281
    },
    {
      "epoch": 3.830432891845703e-05,
      "model_forward_time": 1.7724111080169678,
      "step": 40165
    },
    {
      "epoch": 3.830432891845703e-05,
      "step": 40165,
      "training_step_time": 7.403727769851685
    },
    {
      "epoch": 3.8305282592773434e-05,
      "model_forward_time": 1.7562875747680664,
      "step": 40166
    },
    {
      "epoch": 3.8305282592773434e-05,
      "step": 40166,
      "training_step_time": 8.211683511734009
    },
    {
      "epoch": 3.8306236267089845e-05,
      "model_forward_time": 1.7686409950256348,
      "step": 40167
    },
    {
      "epoch": 3.8306236267089845e-05,
      "step": 40167,
      "training_step_time": 7.744935512542725
    },
    {
      "epoch": 3.830718994140625e-05,
      "model_forward_time": 1.7564196586608887,
      "step": 40168
    },
    {
      "epoch": 3.830718994140625e-05,
      "step": 40168,
      "training_step_time": 7.87338924407959
    },
    {
      "epoch": 3.830814361572265e-05,
      "model_forward_time": 2.888727903366089,
      "step": 40169
    },
    {
      "epoch": 3.830814361572265e-05,
      "step": 40169,
      "training_step_time": 7.831184387207031
    },
    {
      "epoch": 3.8309097290039064e-05,
      "grad_norm": 0.1660185605287552,
      "learning_rate": 6.982855850204521e-06,
      "loss": 0.0599,
      "step": 40170
    },
    {
      "epoch": 3.8309097290039064e-05,
      "model_forward_time": 1.758995532989502,
      "step": 40170
    },
    {
      "dynamics_loss_avg": 0.0619828999042511,
      "epoch": 3.8309097290039064e-05,
      "step": 40170
    },
    {
      "action_loss_avg": 0.0016572858381550758,
      "epoch": 3.8309097290039064e-05,
      "step": 40170
    },
    {
      "epoch": 3.8309097290039064e-05,
      "step": 40170,
      "training_step_time": 9.450562238693237
    },
    {
      "epoch": 3.831005096435547e-05,
      "model_forward_time": 1.7620570659637451,
      "step": 40171
    },
    {
      "epoch": 3.831005096435547e-05,
      "step": 40171,
      "training_step_time": 7.98555588722229
    },
    {
      "epoch": 3.831100463867187e-05,
      "model_forward_time": 1.7532718181610107,
      "step": 40172
    },
    {
      "epoch": 3.831100463867187e-05,
      "step": 40172,
      "training_step_time": 7.556335926055908
    },
    {
      "epoch": 3.831195831298828e-05,
      "model_forward_time": 1.759443998336792,
      "step": 40173
    },
    {
      "epoch": 3.831195831298828e-05,
      "step": 40173,
      "training_step_time": 7.362803936004639
    },
    {
      "epoch": 3.831291198730469e-05,
      "model_forward_time": 1.7549359798431396,
      "step": 40174
    },
    {
      "epoch": 3.831291198730469e-05,
      "step": 40174,
      "training_step_time": 7.197948694229126
    },
    {
      "epoch": 3.831386566162109e-05,
      "model_forward_time": 1.760157823562622,
      "step": 40175
    },
    {
      "epoch": 3.831386566162109e-05,
      "step": 40175,
      "training_step_time": 7.696812152862549
    },
    {
      "epoch": 3.83148193359375e-05,
      "model_forward_time": 1.7625923156738281,
      "step": 40176
    },
    {
      "epoch": 3.83148193359375e-05,
      "step": 40176,
      "training_step_time": 8.652251482009888
    },
    {
      "epoch": 3.8315773010253906e-05,
      "model_forward_time": 1.77004075050354,
      "step": 40177
    },
    {
      "epoch": 3.8315773010253906e-05,
      "step": 40177,
      "training_step_time": 8.618252277374268
    },
    {
      "epoch": 3.831672668457031e-05,
      "model_forward_time": 2.8821117877960205,
      "step": 40178
    },
    {
      "epoch": 3.831672668457031e-05,
      "step": 40178,
      "training_step_time": 10.200348138809204
    },
    {
      "epoch": 3.831768035888672e-05,
      "model_forward_time": 1.748741626739502,
      "step": 40179
    },
    {
      "epoch": 3.831768035888672e-05,
      "step": 40179,
      "training_step_time": 9.208971500396729
    },
    {
      "epoch": 3.8318634033203125e-05,
      "grad_norm": 0.2043442577123642,
      "learning_rate": 6.981337849937864e-06,
      "loss": 0.0578,
      "step": 40180
    },
    {
      "epoch": 3.8318634033203125e-05,
      "model_forward_time": 1.7601139545440674,
      "step": 40180
    },
    {
      "dynamics_loss_avg": 0.06978037357330322,
      "epoch": 3.8318634033203125e-05,
      "step": 40180
    },
    {
      "action_loss_avg": 0.0020697713654953985,
      "epoch": 3.8318634033203125e-05,
      "step": 40180
    },
    {
      "epoch": 3.8318634033203125e-05,
      "step": 40180,
      "training_step_time": 7.849206209182739
    },
    {
      "epoch": 3.831958770751953e-05,
      "model_forward_time": 1.7661747932434082,
      "step": 40181
    },
    {
      "epoch": 3.831958770751953e-05,
      "step": 40181,
      "training_step_time": 7.500914573669434
    },
    {
      "epoch": 3.832054138183594e-05,
      "model_forward_time": 1.7532975673675537,
      "step": 40182
    },
    {
      "epoch": 3.832054138183594e-05,
      "step": 40182,
      "training_step_time": 8.145672798156738
    },
    {
      "epoch": 3.8321495056152344e-05,
      "model_forward_time": 1.7672927379608154,
      "step": 40183
    },
    {
      "epoch": 3.8321495056152344e-05,
      "step": 40183,
      "training_step_time": 7.165791273117065
    },
    {
      "epoch": 3.832244873046875e-05,
      "model_forward_time": 1.7566649913787842,
      "step": 40184
    },
    {
      "epoch": 3.832244873046875e-05,
      "step": 40184,
      "training_step_time": 7.478978157043457
    },
    {
      "epoch": 3.832340240478516e-05,
      "model_forward_time": 1.7579128742218018,
      "step": 40185
    },
    {
      "epoch": 3.832340240478516e-05,
      "step": 40185,
      "training_step_time": 8.435057640075684
    },
    {
      "epoch": 3.832435607910156e-05,
      "model_forward_time": 1.7611143589019775,
      "step": 40186
    },
    {
      "epoch": 3.832435607910156e-05,
      "step": 40186,
      "training_step_time": 7.875708818435669
    },
    {
      "epoch": 3.8325309753417966e-05,
      "model_forward_time": 2.8879566192626953,
      "step": 40187
    },
    {
      "epoch": 3.8325309753417966e-05,
      "step": 40187,
      "training_step_time": 7.826744318008423
    },
    {
      "epoch": 3.832626342773438e-05,
      "model_forward_time": 1.7611238956451416,
      "step": 40188
    },
    {
      "epoch": 3.832626342773438e-05,
      "step": 40188,
      "training_step_time": 7.878162622451782
    },
    {
      "epoch": 3.832721710205078e-05,
      "model_forward_time": 1.7554137706756592,
      "step": 40189
    },
    {
      "epoch": 3.832721710205078e-05,
      "step": 40189,
      "training_step_time": 8.03035020828247
    },
    {
      "epoch": 3.8328170776367185e-05,
      "grad_norm": 0.10740221291780472,
      "learning_rate": 6.979819632995077e-06,
      "loss": 0.0594,
      "step": 40190
    },
    {
      "epoch": 3.8328170776367185e-05,
      "model_forward_time": 1.7641279697418213,
      "step": 40190
    },
    {
      "dynamics_loss_avg": 0.06161389499902725,
      "epoch": 3.8328170776367185e-05,
      "step": 40190
    },
    {
      "action_loss_avg": 0.0023986682121176273,
      "epoch": 3.8328170776367185e-05,
      "step": 40190
    },
    {
      "epoch": 3.8328170776367185e-05,
      "step": 40190,
      "training_step_time": 7.554001331329346
    },
    {
      "epoch": 3.8329124450683596e-05,
      "model_forward_time": 1.759092092514038,
      "step": 40191
    },
    {
      "epoch": 3.8329124450683596e-05,
      "step": 40191,
      "training_step_time": 7.109093427658081
    },
    {
      "epoch": 3.8330078125e-05,
      "model_forward_time": 1.7576723098754883,
      "step": 40192
    },
    {
      "epoch": 3.8330078125e-05,
      "step": 40192,
      "training_step_time": 7.278188228607178
    },
    {
      "epoch": 3.8331031799316404e-05,
      "model_forward_time": 1.7537527084350586,
      "step": 40193
    },
    {
      "epoch": 3.8331031799316404e-05,
      "step": 40193,
      "training_step_time": 8.265500545501709
    },
    {
      "epoch": 3.8331985473632815e-05,
      "model_forward_time": 1.7282166481018066,
      "step": 40194
    },
    {
      "epoch": 3.8331985473632815e-05,
      "step": 40194,
      "training_step_time": 7.645359039306641
    },
    {
      "epoch": 3.833293914794922e-05,
      "model_forward_time": 1.7406525611877441,
      "step": 40195
    },
    {
      "epoch": 3.833293914794922e-05,
      "step": 40195,
      "training_step_time": 9.661394596099854
    },
    {
      "epoch": 3.833389282226562e-05,
      "model_forward_time": 2.9322845935821533,
      "step": 40196
    },
    {
      "epoch": 3.833389282226562e-05,
      "step": 40196,
      "training_step_time": 8.74517822265625
    },
    {
      "epoch": 3.8334846496582034e-05,
      "model_forward_time": 1.7427635192871094,
      "step": 40197
    },
    {
      "epoch": 3.8334846496582034e-05,
      "step": 40197,
      "training_step_time": 10.420232772827148
    },
    {
      "epoch": 3.833580017089844e-05,
      "model_forward_time": 1.7495827674865723,
      "step": 40198
    },
    {
      "epoch": 3.833580017089844e-05,
      "step": 40198,
      "training_step_time": 8.595332860946655
    },
    {
      "epoch": 3.833675384521484e-05,
      "model_forward_time": 1.7448580265045166,
      "step": 40199
    },
    {
      "epoch": 3.833675384521484e-05,
      "step": 40199,
      "training_step_time": 7.553611516952515
    },
    {
      "epoch": 3.833770751953125e-05,
      "grad_norm": 0.14821110665798187,
      "learning_rate": 6.978301199542194e-06,
      "loss": 0.0571,
      "step": 40200
    },
    {
      "epoch": 3.833770751953125e-05,
      "model_forward_time": 1.7556746006011963,
      "step": 40200
    },
    {
      "dynamics_loss_avg": 0.05620357375591993,
      "epoch": 3.833770751953125e-05,
      "step": 40200
    },
    {
      "action_loss_avg": 0.0014979973697336391,
      "epoch": 3.833770751953125e-05,
      "step": 40200
    },
    {
      "epoch": 3.833770751953125e-05,
      "step": 40200,
      "training_step_time": 7.556954383850098
    },
    {
      "epoch": 3.833866119384766e-05,
      "model_forward_time": 1.7462434768676758,
      "step": 40201
    },
    {
      "epoch": 3.833866119384766e-05,
      "step": 40201,
      "training_step_time": 7.296697616577148
    },
    {
      "epoch": 3.833961486816406e-05,
      "model_forward_time": 1.7505605220794678,
      "step": 40202
    },
    {
      "epoch": 3.833961486816406e-05,
      "step": 40202,
      "training_step_time": 7.517627239227295
    },
    {
      "epoch": 3.834056854248047e-05,
      "model_forward_time": 1.7561519145965576,
      "step": 40203
    },
    {
      "epoch": 3.834056854248047e-05,
      "step": 40203,
      "training_step_time": 8.704035520553589
    },
    {
      "epoch": 3.8341522216796876e-05,
      "model_forward_time": 1.7449421882629395,
      "step": 40204
    },
    {
      "epoch": 3.8341522216796876e-05,
      "step": 40204,
      "training_step_time": 8.663936614990234
    },
    {
      "epoch": 3.834247589111328e-05,
      "model_forward_time": 2.8880043029785156,
      "step": 40205
    },
    {
      "epoch": 3.834247589111328e-05,
      "step": 40205,
      "training_step_time": 7.856418609619141
    },
    {
      "epoch": 3.834342956542969e-05,
      "model_forward_time": 1.747880458831787,
      "step": 40206
    },
    {
      "epoch": 3.834342956542969e-05,
      "step": 40206,
      "training_step_time": 9.395589828491211
    },
    {
      "epoch": 3.8344383239746094e-05,
      "model_forward_time": 1.7563958168029785,
      "step": 40207
    },
    {
      "epoch": 3.8344383239746094e-05,
      "step": 40207,
      "training_step_time": 7.948322534561157
    },
    {
      "epoch": 3.83453369140625e-05,
      "model_forward_time": 1.7744340896606445,
      "step": 40208
    },
    {
      "epoch": 3.83453369140625e-05,
      "step": 40208,
      "training_step_time": 7.728180646896362
    },
    {
      "epoch": 3.834629058837891e-05,
      "model_forward_time": 1.7444508075714111,
      "step": 40209
    },
    {
      "epoch": 3.834629058837891e-05,
      "step": 40209,
      "training_step_time": 7.5136942863464355
    },
    {
      "epoch": 3.834724426269531e-05,
      "grad_norm": 0.14029566943645477,
      "learning_rate": 6.976782549745264e-06,
      "loss": 0.0599,
      "step": 40210
    },
    {
      "epoch": 3.834724426269531e-05,
      "model_forward_time": 1.7573881149291992,
      "step": 40210
    },
    {
      "dynamics_loss_avg": 0.04694947451353073,
      "epoch": 3.834724426269531e-05,
      "step": 40210
    },
    {
      "action_loss_avg": 0.001495602075010538,
      "epoch": 3.834724426269531e-05,
      "step": 40210
    },
    {
      "epoch": 3.834724426269531e-05,
      "step": 40210,
      "training_step_time": 7.338639974594116
    },
    {
      "epoch": 3.834819793701172e-05,
      "model_forward_time": 1.7554097175598145,
      "step": 40211
    },
    {
      "epoch": 3.834819793701172e-05,
      "step": 40211,
      "training_step_time": 7.434584379196167
    },
    {
      "epoch": 3.834915161132813e-05,
      "model_forward_time": 1.749433994293213,
      "step": 40212
    },
    {
      "epoch": 3.834915161132813e-05,
      "step": 40212,
      "training_step_time": 7.6771769523620605
    },
    {
      "epoch": 3.835010528564453e-05,
      "model_forward_time": 1.7568581104278564,
      "step": 40213
    },
    {
      "epoch": 3.835010528564453e-05,
      "step": 40213,
      "training_step_time": 7.775857210159302
    },
    {
      "epoch": 3.8351058959960936e-05,
      "model_forward_time": 1.7639822959899902,
      "step": 40214
    },
    {
      "epoch": 3.8351058959960936e-05,
      "step": 40214,
      "training_step_time": 8.371289730072021
    },
    {
      "epoch": 3.835201263427735e-05,
      "model_forward_time": 1.8154101371765137,
      "step": 40215
    },
    {
      "epoch": 3.835201263427735e-05,
      "step": 40215,
      "training_step_time": 8.813003540039062
    },
    {
      "epoch": 3.835296630859375e-05,
      "model_forward_time": 1.7460649013519287,
      "step": 40216
    },
    {
      "epoch": 3.835296630859375e-05,
      "step": 40216,
      "training_step_time": 7.879374027252197
    },
    {
      "epoch": 3.8353919982910155e-05,
      "model_forward_time": 1.7587411403656006,
      "step": 40217
    },
    {
      "epoch": 3.8353919982910155e-05,
      "step": 40217,
      "training_step_time": 7.985654592514038
    },
    {
      "epoch": 3.8354873657226566e-05,
      "model_forward_time": 1.7484431266784668,
      "step": 40218
    },
    {
      "epoch": 3.8354873657226566e-05,
      "step": 40218,
      "training_step_time": 7.560735464096069
    },
    {
      "epoch": 3.835582733154297e-05,
      "model_forward_time": 1.7435972690582275,
      "step": 40219
    },
    {
      "epoch": 3.835582733154297e-05,
      "step": 40219,
      "training_step_time": 7.320084810256958
    },
    {
      "epoch": 3.8356781005859374e-05,
      "grad_norm": 0.15958067774772644,
      "learning_rate": 6.975263683770368e-06,
      "loss": 0.0574,
      "step": 40220
    },
    {
      "epoch": 3.8356781005859374e-05,
      "model_forward_time": 1.7553009986877441,
      "step": 40220
    },
    {
      "dynamics_loss_avg": 0.06580039206892252,
      "epoch": 3.8356781005859374e-05,
      "step": 40220
    },
    {
      "action_loss_avg": 0.0027547596138902007,
      "epoch": 3.8356781005859374e-05,
      "step": 40220
    },
    {
      "epoch": 3.8356781005859374e-05,
      "step": 40220,
      "training_step_time": 7.367884635925293
    },
    {
      "epoch": 3.835773468017578e-05,
      "model_forward_time": 1.7471246719360352,
      "step": 40221
    },
    {
      "epoch": 3.835773468017578e-05,
      "step": 40221,
      "training_step_time": 7.641891956329346
    },
    {
      "epoch": 3.835868835449219e-05,
      "model_forward_time": 1.748966932296753,
      "step": 40222
    },
    {
      "epoch": 3.835868835449219e-05,
      "step": 40222,
      "training_step_time": 8.704042673110962
    },
    {
      "epoch": 3.835964202880859e-05,
      "model_forward_time": 1.7419939041137695,
      "step": 40223
    },
    {
      "epoch": 3.835964202880859e-05,
      "step": 40223,
      "training_step_time": 7.738042593002319
    },
    {
      "epoch": 3.8360595703125e-05,
      "model_forward_time": 2.94962477684021,
      "step": 40224
    },
    {
      "epoch": 3.8360595703125e-05,
      "step": 40224,
      "training_step_time": 7.933375120162964
    },
    {
      "epoch": 3.836154937744141e-05,
      "model_forward_time": 1.739978551864624,
      "step": 40225
    },
    {
      "epoch": 3.836154937744141e-05,
      "step": 40225,
      "training_step_time": 8.582895517349243
    },
    {
      "epoch": 3.836250305175781e-05,
      "model_forward_time": 1.7507405281066895,
      "step": 40226
    },
    {
      "epoch": 3.836250305175781e-05,
      "step": 40226,
      "training_step_time": 7.98728609085083
    },
    {
      "epoch": 3.8363456726074216e-05,
      "model_forward_time": 1.745492696762085,
      "step": 40227
    },
    {
      "epoch": 3.8363456726074216e-05,
      "step": 40227,
      "training_step_time": 7.422841548919678
    },
    {
      "epoch": 3.8364410400390626e-05,
      "model_forward_time": 1.742349624633789,
      "step": 40228
    },
    {
      "epoch": 3.8364410400390626e-05,
      "step": 40228,
      "training_step_time": 7.4581615924835205
    },
    {
      "epoch": 3.836536407470703e-05,
      "model_forward_time": 1.7548425197601318,
      "step": 40229
    },
    {
      "epoch": 3.836536407470703e-05,
      "step": 40229,
      "training_step_time": 6.861294746398926
    },
    {
      "epoch": 3.8366317749023435e-05,
      "grad_norm": 0.21276338398456573,
      "learning_rate": 6.973744601783604e-06,
      "loss": 0.0565,
      "step": 40230
    },
    {
      "epoch": 3.8366317749023435e-05,
      "model_forward_time": 1.7660133838653564,
      "step": 40230
    },
    {
      "dynamics_loss_avg": 0.05701068937778473,
      "epoch": 3.8366317749023435e-05,
      "step": 40230
    },
    {
      "action_loss_avg": 0.001979479541478213,
      "epoch": 3.8366317749023435e-05,
      "step": 40230
    },
    {
      "epoch": 3.8366317749023435e-05,
      "step": 40230,
      "training_step_time": 7.9560675621032715
    },
    {
      "epoch": 3.8367271423339845e-05,
      "model_forward_time": 1.7638928890228271,
      "step": 40231
    },
    {
      "epoch": 3.8367271423339845e-05,
      "step": 40231,
      "training_step_time": 8.679337501525879
    },
    {
      "epoch": 3.836822509765625e-05,
      "model_forward_time": 1.7437918186187744,
      "step": 40232
    },
    {
      "epoch": 3.836822509765625e-05,
      "step": 40232,
      "training_step_time": 8.48489236831665
    },
    {
      "epoch": 3.836917877197265e-05,
      "model_forward_time": 2.989089250564575,
      "step": 40233
    },
    {
      "epoch": 3.836917877197265e-05,
      "step": 40233,
      "training_step_time": 9.881877422332764
    },
    {
      "epoch": 3.8370132446289064e-05,
      "model_forward_time": 1.7551426887512207,
      "step": 40234
    },
    {
      "epoch": 3.8370132446289064e-05,
      "step": 40234,
      "training_step_time": 7.8986430168151855
    },
    {
      "epoch": 3.837108612060547e-05,
      "model_forward_time": 1.7434370517730713,
      "step": 40235
    },
    {
      "epoch": 3.837108612060547e-05,
      "step": 40235,
      "training_step_time": 9.208436250686646
    },
    {
      "epoch": 3.837203979492187e-05,
      "model_forward_time": 1.738649606704712,
      "step": 40236
    },
    {
      "epoch": 3.837203979492187e-05,
      "step": 40236,
      "training_step_time": 7.475294828414917
    },
    {
      "epoch": 3.837299346923828e-05,
      "model_forward_time": 1.7558841705322266,
      "step": 40237
    },
    {
      "epoch": 3.837299346923828e-05,
      "step": 40237,
      "training_step_time": 7.445733547210693
    },
    {
      "epoch": 3.837394714355469e-05,
      "model_forward_time": 1.7468359470367432,
      "step": 40238
    },
    {
      "epoch": 3.837394714355469e-05,
      "step": 40238,
      "training_step_time": 7.350437879562378
    },
    {
      "epoch": 3.837490081787109e-05,
      "model_forward_time": 1.7458419799804688,
      "step": 40239
    },
    {
      "epoch": 3.837490081787109e-05,
      "step": 40239,
      "training_step_time": 8.744887828826904
    },
    {
      "epoch": 3.83758544921875e-05,
      "grad_norm": 0.1742868274450302,
      "learning_rate": 6.972225303951098e-06,
      "loss": 0.0576,
      "step": 40240
    },
    {
      "epoch": 3.83758544921875e-05,
      "model_forward_time": 1.758317470550537,
      "step": 40240
    },
    {
      "dynamics_loss_avg": 0.05983988400548697,
      "epoch": 3.83758544921875e-05,
      "step": 40240
    },
    {
      "action_loss_avg": 0.001837763679213822,
      "epoch": 3.83758544921875e-05,
      "step": 40240
    },
    {
      "epoch": 3.83758544921875e-05,
      "step": 40240,
      "training_step_time": 7.905882120132446
    },
    {
      "epoch": 3.8376808166503906e-05,
      "model_forward_time": 1.751060962677002,
      "step": 40241
    },
    {
      "epoch": 3.8376808166503906e-05,
      "step": 40241,
      "training_step_time": 7.738214731216431
    },
    {
      "epoch": 3.837776184082031e-05,
      "model_forward_time": 2.944424629211426,
      "step": 40242
    },
    {
      "epoch": 3.837776184082031e-05,
      "step": 40242,
      "training_step_time": 8.580343961715698
    },
    {
      "epoch": 3.837871551513672e-05,
      "model_forward_time": 1.7511036396026611,
      "step": 40243
    },
    {
      "epoch": 3.837871551513672e-05,
      "step": 40243,
      "training_step_time": 8.412081956863403
    },
    {
      "epoch": 3.8379669189453125e-05,
      "model_forward_time": 1.7620718479156494,
      "step": 40244
    },
    {
      "epoch": 3.8379669189453125e-05,
      "step": 40244,
      "training_step_time": 9.53920292854309
    },
    {
      "epoch": 3.838062286376953e-05,
      "model_forward_time": 1.7537519931793213,
      "step": 40245
    },
    {
      "epoch": 3.838062286376953e-05,
      "step": 40245,
      "training_step_time": 7.595729827880859
    },
    {
      "epoch": 3.838157653808594e-05,
      "model_forward_time": 1.747577428817749,
      "step": 40246
    },
    {
      "epoch": 3.838157653808594e-05,
      "step": 40246,
      "training_step_time": 7.111487865447998
    },
    {
      "epoch": 3.8382530212402344e-05,
      "model_forward_time": 1.7499661445617676,
      "step": 40247
    },
    {
      "epoch": 3.8382530212402344e-05,
      "step": 40247,
      "training_step_time": 7.360592603683472
    },
    {
      "epoch": 3.838348388671875e-05,
      "model_forward_time": 1.756746530532837,
      "step": 40248
    },
    {
      "epoch": 3.838348388671875e-05,
      "step": 40248,
      "training_step_time": 7.312609434127808
    },
    {
      "epoch": 3.838443756103516e-05,
      "model_forward_time": 1.7547450065612793,
      "step": 40249
    },
    {
      "epoch": 3.838443756103516e-05,
      "step": 40249,
      "training_step_time": 7.733258247375488
    },
    {
      "epoch": 3.838539123535156e-05,
      "grad_norm": 0.11985399574041367,
      "learning_rate": 6.970705790438999e-06,
      "loss": 0.0576,
      "step": 40250
    },
    {
      "epoch": 3.838539123535156e-05,
      "model_forward_time": 1.7448616027832031,
      "step": 40250
    },
    {
      "dynamics_loss_avg": 0.046659634448587894,
      "epoch": 3.838539123535156e-05,
      "step": 40250
    },
    {
      "action_loss_avg": 0.0017638498859014363,
      "epoch": 3.838539123535156e-05,
      "step": 40250
    },
    {
      "epoch": 3.838539123535156e-05,
      "step": 40250,
      "training_step_time": 7.672942399978638
    },
    {
      "epoch": 3.8386344909667967e-05,
      "model_forward_time": 2.883592367172241,
      "step": 40251
    },
    {
      "epoch": 3.8386344909667967e-05,
      "step": 40251,
      "training_step_time": 7.735280752182007
    },
    {
      "epoch": 3.838729858398438e-05,
      "model_forward_time": 1.8234493732452393,
      "step": 40252
    },
    {
      "epoch": 3.838729858398438e-05,
      "step": 40252,
      "training_step_time": 8.53678297996521
    },
    {
      "epoch": 3.838825225830078e-05,
      "model_forward_time": 1.7635784149169922,
      "step": 40253
    },
    {
      "epoch": 3.838825225830078e-05,
      "step": 40253,
      "training_step_time": 7.751699209213257
    },
    {
      "epoch": 3.8389205932617185e-05,
      "model_forward_time": 1.7419803142547607,
      "step": 40254
    },
    {
      "epoch": 3.8389205932617185e-05,
      "step": 40254,
      "training_step_time": 7.582411527633667
    },
    {
      "epoch": 3.8390159606933596e-05,
      "model_forward_time": 1.7472810745239258,
      "step": 40255
    },
    {
      "epoch": 3.8390159606933596e-05,
      "step": 40255,
      "training_step_time": 7.217876434326172
    },
    {
      "epoch": 3.839111328125e-05,
      "model_forward_time": 1.749680995941162,
      "step": 40256
    },
    {
      "epoch": 3.839111328125e-05,
      "step": 40256,
      "training_step_time": 7.360042572021484
    },
    {
      "epoch": 3.8392066955566404e-05,
      "model_forward_time": 1.7370147705078125,
      "step": 40257
    },
    {
      "epoch": 3.8392066955566404e-05,
      "step": 40257,
      "training_step_time": 7.331629276275635
    },
    {
      "epoch": 3.8393020629882815e-05,
      "model_forward_time": 1.7483294010162354,
      "step": 40258
    },
    {
      "epoch": 3.8393020629882815e-05,
      "step": 40258,
      "training_step_time": 7.7791666984558105
    },
    {
      "epoch": 3.839397430419922e-05,
      "model_forward_time": 1.7437944412231445,
      "step": 40259
    },
    {
      "epoch": 3.839397430419922e-05,
      "step": 40259,
      "training_step_time": 8.566135883331299
    },
    {
      "epoch": 3.839492797851562e-05,
      "grad_norm": 0.09092698246240616,
      "learning_rate": 6.969186061413478e-06,
      "loss": 0.0596,
      "step": 40260
    },
    {
      "epoch": 3.839492797851562e-05,
      "model_forward_time": 2.870847225189209,
      "step": 40260
    },
    {
      "dynamics_loss_avg": 0.05377120841294527,
      "epoch": 3.839492797851562e-05,
      "step": 40260
    },
    {
      "action_loss_avg": 0.0017791416961699725,
      "epoch": 3.839492797851562e-05,
      "step": 40260
    },
    {
      "epoch": 3.839492797851562e-05,
      "step": 40260,
      "training_step_time": 9.711091995239258
    },
    {
      "epoch": 3.8395881652832034e-05,
      "model_forward_time": 1.819105863571167,
      "step": 40261
    },
    {
      "epoch": 3.8395881652832034e-05,
      "step": 40261,
      "training_step_time": 9.204261541366577
    },
    {
      "epoch": 3.839683532714844e-05,
      "model_forward_time": 1.7408719062805176,
      "step": 40262
    },
    {
      "epoch": 3.839683532714844e-05,
      "step": 40262,
      "training_step_time": 7.782369613647461
    },
    {
      "epoch": 3.839778900146484e-05,
      "model_forward_time": 1.7529258728027344,
      "step": 40263
    },
    {
      "epoch": 3.839778900146484e-05,
      "step": 40263,
      "training_step_time": 7.896466493606567
    },
    {
      "epoch": 3.839874267578125e-05,
      "model_forward_time": 1.7378442287445068,
      "step": 40264
    },
    {
      "epoch": 3.839874267578125e-05,
      "step": 40264,
      "training_step_time": 7.195344924926758
    },
    {
      "epoch": 3.839969635009766e-05,
      "model_forward_time": 1.7514209747314453,
      "step": 40265
    },
    {
      "epoch": 3.839969635009766e-05,
      "step": 40265,
      "training_step_time": 7.378051519393921
    },
    {
      "epoch": 3.840065002441406e-05,
      "model_forward_time": 1.7419166564941406,
      "step": 40266
    },
    {
      "epoch": 3.840065002441406e-05,
      "step": 40266,
      "training_step_time": 7.260186672210693
    },
    {
      "epoch": 3.840160369873047e-05,
      "model_forward_time": 1.7768397331237793,
      "step": 40267
    },
    {
      "epoch": 3.840160369873047e-05,
      "step": 40267,
      "training_step_time": 7.691248893737793
    },
    {
      "epoch": 3.8402557373046876e-05,
      "model_forward_time": 1.744248867034912,
      "step": 40268
    },
    {
      "epoch": 3.8402557373046876e-05,
      "step": 40268,
      "training_step_time": 8.20254397392273
    },
    {
      "epoch": 3.840351104736328e-05,
      "model_forward_time": 2.814452886581421,
      "step": 40269
    },
    {
      "epoch": 3.840351104736328e-05,
      "step": 40269,
      "training_step_time": 8.714084148406982
    },
    {
      "epoch": 3.840446472167969e-05,
      "grad_norm": 0.1077764704823494,
      "learning_rate": 6.967666117040728e-06,
      "loss": 0.0595,
      "step": 40270
    },
    {
      "epoch": 3.840446472167969e-05,
      "model_forward_time": 1.808072805404663,
      "step": 40270
    },
    {
      "dynamics_loss_avg": 0.05243310090154409,
      "epoch": 3.840446472167969e-05,
      "step": 40270
    },
    {
      "action_loss_avg": 0.0013631823443574832,
      "epoch": 3.840446472167969e-05,
      "step": 40270
    },
    {
      "epoch": 3.840446472167969e-05,
      "step": 40270,
      "training_step_time": 7.710144281387329
    },
    {
      "epoch": 3.8405418395996095e-05,
      "model_forward_time": 1.730154275894165,
      "step": 40271
    },
    {
      "epoch": 3.8405418395996095e-05,
      "step": 40271,
      "training_step_time": 8.226454973220825
    },
    {
      "epoch": 3.84063720703125e-05,
      "model_forward_time": 1.6957342624664307,
      "step": 40272
    },
    {
      "epoch": 3.84063720703125e-05,
      "step": 40272,
      "training_step_time": 8.482239484786987
    },
    {
      "epoch": 3.840732574462891e-05,
      "model_forward_time": 1.6914443969726562,
      "step": 40273
    },
    {
      "epoch": 3.840732574462891e-05,
      "step": 40273,
      "training_step_time": 7.3856658935546875
    },
    {
      "epoch": 3.840827941894531e-05,
      "model_forward_time": 1.6993379592895508,
      "step": 40274
    },
    {
      "epoch": 3.840827941894531e-05,
      "step": 40274,
      "training_step_time": 7.305865526199341
    },
    {
      "epoch": 3.840923309326172e-05,
      "model_forward_time": 1.6926839351654053,
      "step": 40275
    },
    {
      "epoch": 3.840923309326172e-05,
      "step": 40275,
      "training_step_time": 7.294683218002319
    },
    {
      "epoch": 3.841018676757813e-05,
      "model_forward_time": 1.6831943988800049,
      "step": 40276
    },
    {
      "epoch": 3.841018676757813e-05,
      "step": 40276,
      "training_step_time": 7.530070781707764
    },
    {
      "epoch": 3.841114044189453e-05,
      "model_forward_time": 1.6938345432281494,
      "step": 40277
    },
    {
      "epoch": 3.841114044189453e-05,
      "step": 40277,
      "training_step_time": 7.626514434814453
    },
    {
      "epoch": 3.8412094116210936e-05,
      "model_forward_time": 1.7021243572235107,
      "step": 40278
    },
    {
      "epoch": 3.8412094116210936e-05,
      "step": 40278,
      "training_step_time": 9.190813302993774
    },
    {
      "epoch": 3.841304779052735e-05,
      "model_forward_time": 2.230262517929077,
      "step": 40279
    },
    {
      "epoch": 3.841304779052735e-05,
      "step": 40279,
      "training_step_time": 7.7934911251068115
    },
    {
      "epoch": 3.841400146484375e-05,
      "grad_norm": 0.11272174119949341,
      "learning_rate": 6.966145957486972e-06,
      "loss": 0.058,
      "step": 40280
    },
    {
      "epoch": 3.841400146484375e-05,
      "model_forward_time": 1.6928367614746094,
      "step": 40280
    },
    {
      "dynamics_loss_avg": 0.05340620633214712,
      "epoch": 3.841400146484375e-05,
      "step": 40280
    },
    {
      "action_loss_avg": 0.0014691360236611217,
      "epoch": 3.841400146484375e-05,
      "step": 40280
    },
    {
      "epoch": 3.841400146484375e-05,
      "step": 40280,
      "training_step_time": 10.167969465255737
    },
    {
      "epoch": 3.8414955139160155e-05,
      "model_forward_time": 1.6949982643127441,
      "step": 40281
    },
    {
      "epoch": 3.8414955139160155e-05,
      "step": 40281,
      "training_step_time": 8.009873867034912
    },
    {
      "epoch": 3.841590881347656e-05,
      "model_forward_time": 1.7009027004241943,
      "step": 40282
    },
    {
      "epoch": 3.841590881347656e-05,
      "step": 40282,
      "training_step_time": 8.117470026016235
    },
    {
      "epoch": 3.841686248779297e-05,
      "model_forward_time": 1.698045253753662,
      "step": 40283
    },
    {
      "epoch": 3.841686248779297e-05,
      "step": 40283,
      "training_step_time": 7.047225713729858
    },
    {
      "epoch": 3.8417816162109374e-05,
      "model_forward_time": 1.6945674419403076,
      "step": 40284
    },
    {
      "epoch": 3.8417816162109374e-05,
      "step": 40284,
      "training_step_time": 7.333112001419067
    },
    {
      "epoch": 3.841876983642578e-05,
      "model_forward_time": 1.7041704654693604,
      "step": 40285
    },
    {
      "epoch": 3.841876983642578e-05,
      "step": 40285,
      "training_step_time": 8.131167888641357
    },
    {
      "epoch": 3.841972351074219e-05,
      "model_forward_time": 1.6976540088653564,
      "step": 40286
    },
    {
      "epoch": 3.841972351074219e-05,
      "step": 40286,
      "training_step_time": 8.500799179077148
    },
    {
      "epoch": 3.842067718505859e-05,
      "model_forward_time": 1.705634355545044,
      "step": 40287
    },
    {
      "epoch": 3.842067718505859e-05,
      "step": 40287,
      "training_step_time": 9.672303199768066
    },
    {
      "epoch": 3.8421630859375e-05,
      "model_forward_time": 2.26167631149292,
      "step": 40288
    },
    {
      "epoch": 3.8421630859375e-05,
      "step": 40288,
      "training_step_time": 8.581849813461304
    },
    {
      "epoch": 3.842258453369141e-05,
      "model_forward_time": 1.701244831085205,
      "step": 40289
    },
    {
      "epoch": 3.842258453369141e-05,
      "step": 40289,
      "training_step_time": 8.624115943908691
    },
    {
      "epoch": 3.842353820800781e-05,
      "grad_norm": 0.13488133251667023,
      "learning_rate": 6.964625582918448e-06,
      "loss": 0.0602,
      "step": 40290
    },
    {
      "epoch": 3.842353820800781e-05,
      "model_forward_time": 1.7213916778564453,
      "step": 40290
    },
    {
      "dynamics_loss_avg": 0.06707625351846218,
      "epoch": 3.842353820800781e-05,
      "step": 40290
    },
    {
      "action_loss_avg": 0.002304277796065435,
      "epoch": 3.842353820800781e-05,
      "step": 40290
    },
    {
      "epoch": 3.842353820800781e-05,
      "step": 40290,
      "training_step_time": 8.39337968826294
    },
    {
      "epoch": 3.8424491882324216e-05,
      "model_forward_time": 1.7042820453643799,
      "step": 40291
    },
    {
      "epoch": 3.8424491882324216e-05,
      "step": 40291,
      "training_step_time": 7.517735958099365
    },
    {
      "epoch": 3.8425445556640627e-05,
      "model_forward_time": 1.7057061195373535,
      "step": 40292
    },
    {
      "epoch": 3.8425445556640627e-05,
      "step": 40292,
      "training_step_time": 7.208873271942139
    },
    {
      "epoch": 3.842639923095703e-05,
      "model_forward_time": 1.7060351371765137,
      "step": 40293
    },
    {
      "epoch": 3.842639923095703e-05,
      "step": 40293,
      "training_step_time": 7.388777256011963
    },
    {
      "epoch": 3.8427352905273435e-05,
      "model_forward_time": 1.7024140357971191,
      "step": 40294
    },
    {
      "epoch": 3.8427352905273435e-05,
      "step": 40294,
      "training_step_time": 8.241034269332886
    },
    {
      "epoch": 3.8428306579589845e-05,
      "model_forward_time": 1.704988718032837,
      "step": 40295
    },
    {
      "epoch": 3.8428306579589845e-05,
      "step": 40295,
      "training_step_time": 8.74820613861084
    },
    {
      "epoch": 3.842926025390625e-05,
      "model_forward_time": 1.7002239227294922,
      "step": 40296
    },
    {
      "epoch": 3.842926025390625e-05,
      "step": 40296,
      "training_step_time": 7.733031511306763
    },
    {
      "epoch": 3.8430213928222653e-05,
      "model_forward_time": 2.271872043609619,
      "step": 40297
    },
    {
      "epoch": 3.8430213928222653e-05,
      "step": 40297,
      "training_step_time": 8.426070928573608
    },
    {
      "epoch": 3.8431167602539064e-05,
      "model_forward_time": 1.8042263984680176,
      "step": 40298
    },
    {
      "epoch": 3.8431167602539064e-05,
      "step": 40298,
      "training_step_time": 8.802776336669922
    },
    {
      "epoch": 3.843212127685547e-05,
      "model_forward_time": 1.7545666694641113,
      "step": 40299
    },
    {
      "epoch": 3.843212127685547e-05,
      "step": 40299,
      "training_step_time": 7.730311870574951
    },
    {
      "epoch": 3.843307495117187e-05,
      "grad_norm": 0.10464973747730255,
      "learning_rate": 6.963104993501425e-06,
      "loss": 0.0577,
      "step": 40300
    },
    {
      "epoch": 3.843307495117187e-05,
      "model_forward_time": 1.7576956748962402,
      "step": 40300
    },
    {
      "dynamics_loss_avg": 0.051258789747953414,
      "epoch": 3.843307495117187e-05,
      "step": 40300
    },
    {
      "action_loss_avg": 0.0020660101872636005,
      "epoch": 3.843307495117187e-05,
      "step": 40300
    },
    {
      "epoch": 3.843307495117187e-05,
      "step": 40300,
      "training_step_time": 8.234790086746216
    },
    {
      "epoch": 3.843402862548828e-05,
      "model_forward_time": 1.7556192874908447,
      "step": 40301
    },
    {
      "epoch": 3.843402862548828e-05,
      "step": 40301,
      "training_step_time": 7.561505079269409
    },
    {
      "epoch": 3.843498229980469e-05,
      "model_forward_time": 1.7507290840148926,
      "step": 40302
    },
    {
      "epoch": 3.843498229980469e-05,
      "step": 40302,
      "training_step_time": 7.597136974334717
    },
    {
      "epoch": 3.843593597412109e-05,
      "model_forward_time": 1.746518611907959,
      "step": 40303
    },
    {
      "epoch": 3.843593597412109e-05,
      "step": 40303,
      "training_step_time": 7.597599029541016
    },
    {
      "epoch": 3.84368896484375e-05,
      "model_forward_time": 1.755652904510498,
      "step": 40304
    },
    {
      "epoch": 3.84368896484375e-05,
      "step": 40304,
      "training_step_time": 8.61947774887085
    },
    {
      "epoch": 3.8437843322753906e-05,
      "model_forward_time": 1.7496201992034912,
      "step": 40305
    },
    {
      "epoch": 3.8437843322753906e-05,
      "step": 40305,
      "training_step_time": 7.640634059906006
    },
    {
      "epoch": 3.843879699707031e-05,
      "model_forward_time": 2.87412691116333,
      "step": 40306
    },
    {
      "epoch": 3.843879699707031e-05,
      "step": 40306,
      "training_step_time": 9.495914459228516
    },
    {
      "epoch": 3.843975067138672e-05,
      "model_forward_time": 1.8030633926391602,
      "step": 40307
    },
    {
      "epoch": 3.843975067138672e-05,
      "step": 40307,
      "training_step_time": 8.732588529586792
    },
    {
      "epoch": 3.8440704345703125e-05,
      "model_forward_time": 1.7480871677398682,
      "step": 40308
    },
    {
      "epoch": 3.8440704345703125e-05,
      "step": 40308,
      "training_step_time": 8.419678449630737
    },
    {
      "epoch": 3.844165802001953e-05,
      "model_forward_time": 1.753150224685669,
      "step": 40309
    },
    {
      "epoch": 3.844165802001953e-05,
      "step": 40309,
      "training_step_time": 7.967039108276367
    },
    {
      "epoch": 3.844261169433594e-05,
      "grad_norm": 0.11655334383249283,
      "learning_rate": 6.961584189402192e-06,
      "loss": 0.0549,
      "step": 40310
    },
    {
      "epoch": 3.844261169433594e-05,
      "model_forward_time": 1.765343189239502,
      "step": 40310
    },
    {
      "dynamics_loss_avg": 0.05622175727039576,
      "epoch": 3.844261169433594e-05,
      "step": 40310
    },
    {
      "action_loss_avg": 0.0014741182269062847,
      "epoch": 3.844261169433594e-05,
      "step": 40310
    },
    {
      "epoch": 3.844261169433594e-05,
      "step": 40310,
      "training_step_time": 7.463218688964844
    },
    {
      "epoch": 3.8443565368652344e-05,
      "model_forward_time": 1.7468926906585693,
      "step": 40311
    },
    {
      "epoch": 3.8443565368652344e-05,
      "step": 40311,
      "training_step_time": 6.97524881362915
    },
    {
      "epoch": 3.844451904296875e-05,
      "model_forward_time": 1.7560501098632812,
      "step": 40312
    },
    {
      "epoch": 3.844451904296875e-05,
      "step": 40312,
      "training_step_time": 8.094829797744751
    },
    {
      "epoch": 3.844547271728516e-05,
      "model_forward_time": 1.7556543350219727,
      "step": 40313
    },
    {
      "epoch": 3.844547271728516e-05,
      "step": 40313,
      "training_step_time": 8.368278503417969
    },
    {
      "epoch": 3.844642639160156e-05,
      "model_forward_time": 1.749518632888794,
      "step": 40314
    },
    {
      "epoch": 3.844642639160156e-05,
      "step": 40314,
      "training_step_time": 7.644083738327026
    },
    {
      "epoch": 3.8447380065917967e-05,
      "model_forward_time": 2.8448171615600586,
      "step": 40315
    },
    {
      "epoch": 3.8447380065917967e-05,
      "step": 40315,
      "training_step_time": 7.661396503448486
    },
    {
      "epoch": 3.844833374023438e-05,
      "model_forward_time": 1.7750790119171143,
      "step": 40316
    },
    {
      "epoch": 3.844833374023438e-05,
      "step": 40316,
      "training_step_time": 8.371453762054443
    },
    {
      "epoch": 3.844928741455078e-05,
      "model_forward_time": 1.7511370182037354,
      "step": 40317
    },
    {
      "epoch": 3.844928741455078e-05,
      "step": 40317,
      "training_step_time": 9.324647188186646
    },
    {
      "epoch": 3.8450241088867185e-05,
      "model_forward_time": 1.753702163696289,
      "step": 40318
    },
    {
      "epoch": 3.8450241088867185e-05,
      "step": 40318,
      "training_step_time": 7.887843132019043
    },
    {
      "epoch": 3.8451194763183596e-05,
      "model_forward_time": 1.7491748332977295,
      "step": 40319
    },
    {
      "epoch": 3.8451194763183596e-05,
      "step": 40319,
      "training_step_time": 7.3270769119262695
    },
    {
      "epoch": 3.84521484375e-05,
      "grad_norm": 0.12368335574865341,
      "learning_rate": 6.96006317078706e-06,
      "loss": 0.0547,
      "step": 40320
    },
    {
      "epoch": 3.84521484375e-05,
      "model_forward_time": 1.7604389190673828,
      "step": 40320
    },
    {
      "dynamics_loss_avg": 0.06011594831943512,
      "epoch": 3.84521484375e-05,
      "step": 40320
    },
    {
      "action_loss_avg": 0.0022043122939066962,
      "epoch": 3.84521484375e-05,
      "step": 40320
    },
    {
      "epoch": 3.84521484375e-05,
      "step": 40320,
      "training_step_time": 7.54662299156189
    },
    {
      "epoch": 3.8453102111816404e-05,
      "model_forward_time": 1.7512938976287842,
      "step": 40321
    },
    {
      "epoch": 3.8453102111816404e-05,
      "step": 40321,
      "training_step_time": 7.628937721252441
    },
    {
      "epoch": 3.8454055786132815e-05,
      "model_forward_time": 1.7682723999023438,
      "step": 40322
    },
    {
      "epoch": 3.8454055786132815e-05,
      "step": 40322,
      "training_step_time": 8.862356901168823
    },
    {
      "epoch": 3.845500946044922e-05,
      "model_forward_time": 1.76951003074646,
      "step": 40323
    },
    {
      "epoch": 3.845500946044922e-05,
      "step": 40323,
      "training_step_time": 8.76785159111023
    },
    {
      "epoch": 3.845596313476562e-05,
      "model_forward_time": 1.7505218982696533,
      "step": 40324
    },
    {
      "epoch": 3.845596313476562e-05,
      "step": 40324,
      "training_step_time": 9.00784707069397
    },
    {
      "epoch": 3.8456916809082034e-05,
      "model_forward_time": 2.892819404602051,
      "step": 40325
    },
    {
      "epoch": 3.8456916809082034e-05,
      "step": 40325,
      "training_step_time": 8.85586667060852
    },
    {
      "epoch": 3.845787048339844e-05,
      "model_forward_time": 1.745908260345459,
      "step": 40326
    },
    {
      "epoch": 3.845787048339844e-05,
      "step": 40326,
      "training_step_time": 7.583357334136963
    },
    {
      "epoch": 3.845882415771484e-05,
      "model_forward_time": 1.7294237613677979,
      "step": 40327
    },
    {
      "epoch": 3.845882415771484e-05,
      "step": 40327,
      "training_step_time": 8.21487283706665
    },
    {
      "epoch": 3.845977783203125e-05,
      "model_forward_time": 1.735795021057129,
      "step": 40328
    },
    {
      "epoch": 3.845977783203125e-05,
      "step": 40328,
      "training_step_time": 7.798438787460327
    },
    {
      "epoch": 3.846073150634766e-05,
      "model_forward_time": 1.7467143535614014,
      "step": 40329
    },
    {
      "epoch": 3.846073150634766e-05,
      "step": 40329,
      "training_step_time": 7.185741662979126
    },
    {
      "epoch": 3.846168518066406e-05,
      "grad_norm": 0.11329049617052078,
      "learning_rate": 6.958541937822368e-06,
      "loss": 0.0572,
      "step": 40330
    },
    {
      "epoch": 3.846168518066406e-05,
      "model_forward_time": 1.7299911975860596,
      "step": 40330
    },
    {
      "dynamics_loss_avg": 0.05791681408882141,
      "epoch": 3.846168518066406e-05,
      "step": 40330
    },
    {
      "action_loss_avg": 0.00246669506886974,
      "epoch": 3.846168518066406e-05,
      "step": 40330
    },
    {
      "epoch": 3.846168518066406e-05,
      "step": 40330,
      "training_step_time": 7.568749904632568
    },
    {
      "epoch": 3.846263885498047e-05,
      "model_forward_time": 1.7346162796020508,
      "step": 40331
    },
    {
      "epoch": 3.846263885498047e-05,
      "step": 40331,
      "training_step_time": 7.410496950149536
    },
    {
      "epoch": 3.8463592529296876e-05,
      "model_forward_time": 1.7396535873413086,
      "step": 40332
    },
    {
      "epoch": 3.8463592529296876e-05,
      "step": 40332,
      "training_step_time": 7.632380723953247
    },
    {
      "epoch": 3.846454620361328e-05,
      "model_forward_time": 1.7462022304534912,
      "step": 40333
    },
    {
      "epoch": 3.846454620361328e-05,
      "step": 40333,
      "training_step_time": 7.587906837463379
    },
    {
      "epoch": 3.846549987792969e-05,
      "model_forward_time": 2.9241273403167725,
      "step": 40334
    },
    {
      "epoch": 3.846549987792969e-05,
      "step": 40334,
      "training_step_time": 9.940574169158936
    },
    {
      "epoch": 3.8466453552246095e-05,
      "model_forward_time": 1.8126158714294434,
      "step": 40335
    },
    {
      "epoch": 3.8466453552246095e-05,
      "step": 40335,
      "training_step_time": 8.982062816619873
    },
    {
      "epoch": 3.84674072265625e-05,
      "model_forward_time": 1.739640474319458,
      "step": 40336
    },
    {
      "epoch": 3.84674072265625e-05,
      "step": 40336,
      "training_step_time": 8.529246807098389
    },
    {
      "epoch": 3.846836090087891e-05,
      "model_forward_time": 1.7462966442108154,
      "step": 40337
    },
    {
      "epoch": 3.846836090087891e-05,
      "step": 40337,
      "training_step_time": 7.849213361740112
    },
    {
      "epoch": 3.8469314575195313e-05,
      "model_forward_time": 1.7384517192840576,
      "step": 40338
    },
    {
      "epoch": 3.8469314575195313e-05,
      "step": 40338,
      "training_step_time": 9.052475452423096
    },
    {
      "epoch": 3.847026824951172e-05,
      "model_forward_time": 1.721923589706421,
      "step": 40339
    },
    {
      "epoch": 3.847026824951172e-05,
      "step": 40339,
      "training_step_time": 9.051238059997559
    },
    {
      "epoch": 3.847122192382813e-05,
      "grad_norm": 0.11054208874702454,
      "learning_rate": 6.957020490674473e-06,
      "loss": 0.0563,
      "step": 40340
    },
    {
      "epoch": 3.847122192382813e-05,
      "model_forward_time": 1.7405555248260498,
      "step": 40340
    },
    {
      "dynamics_loss_avg": 0.0588201068341732,
      "epoch": 3.847122192382813e-05,
      "step": 40340
    },
    {
      "action_loss_avg": 0.002177109743934125,
      "epoch": 3.847122192382813e-05,
      "step": 40340
    },
    {
      "epoch": 3.847122192382813e-05,
      "step": 40340,
      "training_step_time": 7.401546239852905
    },
    {
      "epoch": 3.847217559814453e-05,
      "model_forward_time": 0.735992431640625,
      "step": 40341
    },
    {
      "epoch": 3.847217559814453e-05,
      "step": 40341,
      "training_step_time": 7.644126653671265
    },
    {
      "epoch": 3.8473129272460936e-05,
      "model_forward_time": 1.7391455173492432,
      "step": 40342
    },
    {
      "epoch": 3.8473129272460936e-05,
      "step": 40342,
      "training_step_time": 7.630975723266602
    },
    {
      "epoch": 3.847408294677735e-05,
      "model_forward_time": 2.8130173683166504,
      "step": 40343
    },
    {
      "epoch": 3.847408294677735e-05,
      "step": 40343,
      "training_step_time": 7.668768644332886
    },
    {
      "epoch": 3.847503662109375e-05,
      "model_forward_time": 1.8034887313842773,
      "step": 40344
    },
    {
      "epoch": 3.847503662109375e-05,
      "step": 40344,
      "training_step_time": 8.450941324234009
    },
    {
      "epoch": 3.8475990295410155e-05,
      "model_forward_time": 1.735985279083252,
      "step": 40345
    },
    {
      "epoch": 3.8475990295410155e-05,
      "step": 40345,
      "training_step_time": 7.6461708545684814
    },
    {
      "epoch": 3.847694396972656e-05,
      "model_forward_time": 1.732133150100708,
      "step": 40346
    },
    {
      "epoch": 3.847694396972656e-05,
      "step": 40346,
      "training_step_time": 7.82117772102356
    },
    {
      "epoch": 3.847789764404297e-05,
      "model_forward_time": 1.738346815109253,
      "step": 40347
    },
    {
      "epoch": 3.847789764404297e-05,
      "step": 40347,
      "training_step_time": 7.2373762130737305
    },
    {
      "epoch": 3.8478851318359374e-05,
      "model_forward_time": 1.7312204837799072,
      "step": 40348
    },
    {
      "epoch": 3.8478851318359374e-05,
      "step": 40348,
      "training_step_time": 8.372703790664673
    },
    {
      "epoch": 3.847980499267578e-05,
      "model_forward_time": 1.73228120803833,
      "step": 40349
    },
    {
      "epoch": 3.847980499267578e-05,
      "step": 40349,
      "training_step_time": 7.433655738830566
    },
    {
      "epoch": 3.848075866699219e-05,
      "grad_norm": 0.11434610188007355,
      "learning_rate": 6.95549882950976e-06,
      "loss": 0.0558,
      "step": 40350
    },
    {
      "epoch": 3.848075866699219e-05,
      "model_forward_time": 1.732097864151001,
      "step": 40350
    },
    {
      "dynamics_loss_avg": 0.0682106539607048,
      "epoch": 3.848075866699219e-05,
      "step": 40350
    },
    {
      "action_loss_avg": 0.004129493911750615,
      "epoch": 3.848075866699219e-05,
      "step": 40350
    },
    {
      "epoch": 3.848075866699219e-05,
      "step": 40350,
      "training_step_time": 8.141841888427734
    },
    {
      "epoch": 3.848171234130859e-05,
      "model_forward_time": 1.7453300952911377,
      "step": 40351
    },
    {
      "epoch": 3.848171234130859e-05,
      "step": 40351,
      "training_step_time": 7.53137731552124
    },
    {
      "epoch": 3.8482666015625e-05,
      "model_forward_time": 1.744905948638916,
      "step": 40352
    },
    {
      "epoch": 3.8482666015625e-05,
      "step": 40352,
      "training_step_time": 9.520344972610474
    },
    {
      "epoch": 3.848361968994141e-05,
      "model_forward_time": 2.9212889671325684,
      "step": 40353
    },
    {
      "epoch": 3.848361968994141e-05,
      "step": 40353,
      "training_step_time": 8.704292058944702
    },
    {
      "epoch": 3.848457336425781e-05,
      "model_forward_time": 1.7278928756713867,
      "step": 40354
    },
    {
      "epoch": 3.848457336425781e-05,
      "step": 40354,
      "training_step_time": 7.620392799377441
    },
    {
      "epoch": 3.8485527038574216e-05,
      "model_forward_time": 1.7373652458190918,
      "step": 40355
    },
    {
      "epoch": 3.8485527038574216e-05,
      "step": 40355,
      "training_step_time": 8.297454118728638
    },
    {
      "epoch": 3.8486480712890627e-05,
      "model_forward_time": 1.7660651206970215,
      "step": 40356
    },
    {
      "epoch": 3.8486480712890627e-05,
      "step": 40356,
      "training_step_time": 8.647124767303467
    },
    {
      "epoch": 3.848743438720703e-05,
      "model_forward_time": 1.7417843341827393,
      "step": 40357
    },
    {
      "epoch": 3.848743438720703e-05,
      "step": 40357,
      "training_step_time": 7.536197662353516
    },
    {
      "epoch": 3.8488388061523435e-05,
      "model_forward_time": 1.7416346073150635,
      "step": 40358
    },
    {
      "epoch": 3.8488388061523435e-05,
      "step": 40358,
      "training_step_time": 7.495769023895264
    },
    {
      "epoch": 3.8489341735839845e-05,
      "model_forward_time": 1.7321062088012695,
      "step": 40359
    },
    {
      "epoch": 3.8489341735839845e-05,
      "step": 40359,
      "training_step_time": 7.663430690765381
    },
    {
      "epoch": 3.849029541015625e-05,
      "grad_norm": 0.0944281667470932,
      "learning_rate": 6.953976954494636e-06,
      "loss": 0.0559,
      "step": 40360
    },
    {
      "epoch": 3.849029541015625e-05,
      "model_forward_time": 1.720155954360962,
      "step": 40360
    },
    {
      "dynamics_loss_avg": 0.05716062802821398,
      "epoch": 3.849029541015625e-05,
      "step": 40360
    },
    {
      "action_loss_avg": 0.002414647155092098,
      "epoch": 3.849029541015625e-05,
      "step": 40360
    },
    {
      "epoch": 3.849029541015625e-05,
      "step": 40360,
      "training_step_time": 8.473966121673584
    },
    {
      "epoch": 3.8491249084472653e-05,
      "model_forward_time": 1.7570667266845703,
      "step": 40361
    },
    {
      "epoch": 3.8491249084472653e-05,
      "step": 40361,
      "training_step_time": 7.79000186920166
    },
    {
      "epoch": 3.8492202758789064e-05,
      "model_forward_time": 2.8971128463745117,
      "step": 40362
    },
    {
      "epoch": 3.8492202758789064e-05,
      "step": 40362,
      "training_step_time": 8.995569229125977
    },
    {
      "epoch": 3.849315643310547e-05,
      "model_forward_time": 1.7364368438720703,
      "step": 40363
    },
    {
      "epoch": 3.849315643310547e-05,
      "step": 40363,
      "training_step_time": 8.976907730102539
    },
    {
      "epoch": 3.849411010742187e-05,
      "model_forward_time": 1.7450706958770752,
      "step": 40364
    },
    {
      "epoch": 3.849411010742187e-05,
      "step": 40364,
      "training_step_time": 8.120860576629639
    },
    {
      "epoch": 3.849506378173828e-05,
      "model_forward_time": 1.733492374420166,
      "step": 40365
    },
    {
      "epoch": 3.849506378173828e-05,
      "step": 40365,
      "training_step_time": 8.868247032165527
    },
    {
      "epoch": 3.849601745605469e-05,
      "model_forward_time": 1.7379224300384521,
      "step": 40366
    },
    {
      "epoch": 3.849601745605469e-05,
      "step": 40366,
      "training_step_time": 7.513366937637329
    },
    {
      "epoch": 3.849697113037109e-05,
      "model_forward_time": 1.7321505546569824,
      "step": 40367
    },
    {
      "epoch": 3.849697113037109e-05,
      "step": 40367,
      "training_step_time": 7.98497462272644
    },
    {
      "epoch": 3.84979248046875e-05,
      "model_forward_time": 1.7427756786346436,
      "step": 40368
    },
    {
      "epoch": 3.84979248046875e-05,
      "step": 40368,
      "training_step_time": 8.608755111694336
    },
    {
      "epoch": 3.8498878479003906e-05,
      "model_forward_time": 1.7428879737854004,
      "step": 40369
    },
    {
      "epoch": 3.8498878479003906e-05,
      "step": 40369,
      "training_step_time": 8.550303220748901
    },
    {
      "epoch": 3.849983215332031e-05,
      "grad_norm": 0.16452045738697052,
      "learning_rate": 6.952454865795528e-06,
      "loss": 0.0566,
      "step": 40370
    },
    {
      "epoch": 3.849983215332031e-05,
      "model_forward_time": 1.750692367553711,
      "step": 40370
    },
    {
      "dynamics_loss_avg": 0.06185827273875475,
      "epoch": 3.849983215332031e-05,
      "step": 40370
    },
    {
      "action_loss_avg": 0.0019581049273256212,
      "epoch": 3.849983215332031e-05,
      "step": 40370
    },
    {
      "epoch": 3.849983215332031e-05,
      "step": 40370,
      "training_step_time": 6.733008146286011
    },
    {
      "epoch": 3.850078582763672e-05,
      "model_forward_time": 1.7333815097808838,
      "step": 40371
    },
    {
      "epoch": 3.850078582763672e-05,
      "step": 40371,
      "training_step_time": 6.73058009147644
    },
    {
      "epoch": 3.8501739501953125e-05,
      "model_forward_time": 1.7422566413879395,
      "step": 40372
    },
    {
      "epoch": 3.8501739501953125e-05,
      "step": 40372,
      "training_step_time": 6.681325674057007
    },
    {
      "epoch": 3.850269317626953e-05,
      "model_forward_time": 1.7455151081085205,
      "step": 40373
    },
    {
      "epoch": 3.850269317626953e-05,
      "step": 40373,
      "training_step_time": 6.698173761367798
    },
    {
      "epoch": 3.850364685058594e-05,
      "model_forward_time": 1.7408816814422607,
      "step": 40374
    },
    {
      "epoch": 3.850364685058594e-05,
      "step": 40374,
      "training_step_time": 6.699635028839111
    },
    {
      "epoch": 3.8504600524902344e-05,
      "model_forward_time": 1.7421824932098389,
      "step": 40375
    },
    {
      "epoch": 3.8504600524902344e-05,
      "step": 40375,
      "training_step_time": 6.671316385269165
    },
    {
      "epoch": 3.850555419921875e-05,
      "model_forward_time": 1.741628885269165,
      "step": 40376
    },
    {
      "epoch": 3.850555419921875e-05,
      "step": 40376,
      "training_step_time": 6.713103532791138
    },
    {
      "epoch": 3.850650787353516e-05,
      "model_forward_time": 1.7364528179168701,
      "step": 40377
    },
    {
      "epoch": 3.850650787353516e-05,
      "step": 40377,
      "training_step_time": 6.72223973274231
    },
    {
      "epoch": 3.850746154785156e-05,
      "model_forward_time": 2.8569893836975098,
      "step": 40378
    },
    {
      "epoch": 3.850746154785156e-05,
      "step": 40378,
      "training_step_time": 7.549911737442017
    },
    {
      "epoch": 3.850841522216797e-05,
      "model_forward_time": 1.7409319877624512,
      "step": 40379
    },
    {
      "epoch": 3.850841522216797e-05,
      "step": 40379,
      "training_step_time": 7.763463735580444
    },
    {
      "epoch": 3.850936889648438e-05,
      "grad_norm": 0.12316352128982544,
      "learning_rate": 6.950932563578892e-06,
      "loss": 0.0558,
      "step": 40380
    },
    {
      "epoch": 3.850936889648438e-05,
      "model_forward_time": 1.7320067882537842,
      "step": 40380
    },
    {
      "dynamics_loss_avg": 0.038654005900025365,
      "epoch": 3.850936889648438e-05,
      "step": 40380
    },
    {
      "action_loss_avg": 0.0012057608139002696,
      "epoch": 3.850936889648438e-05,
      "step": 40380
    },
    {
      "epoch": 3.850936889648438e-05,
      "step": 40380,
      "training_step_time": 6.736192464828491
    },
    {
      "epoch": 3.851032257080078e-05,
      "model_forward_time": 1.7418060302734375,
      "step": 40381
    },
    {
      "epoch": 3.851032257080078e-05,
      "step": 40381,
      "training_step_time": 6.75752854347229
    },
    {
      "epoch": 3.8511276245117185e-05,
      "model_forward_time": 1.7412364482879639,
      "step": 40382
    },
    {
      "epoch": 3.8511276245117185e-05,
      "step": 40382,
      "training_step_time": 6.760804176330566
    },
    {
      "epoch": 3.8512229919433596e-05,
      "model_forward_time": 1.3902013301849365,
      "step": 40383
    },
    {
      "epoch": 3.8512229919433596e-05,
      "step": 40383,
      "training_step_time": 6.749178886413574
    },
    {
      "epoch": 3.851318359375e-05,
      "model_forward_time": 1.7494614124298096,
      "step": 40384
    },
    {
      "epoch": 3.851318359375e-05,
      "step": 40384,
      "training_step_time": 6.758203983306885
    },
    {
      "epoch": 3.8514137268066404e-05,
      "model_forward_time": 1.7491991519927979,
      "step": 40385
    },
    {
      "epoch": 3.8514137268066404e-05,
      "step": 40385,
      "training_step_time": 6.726460218429565
    },
    {
      "epoch": 3.8515090942382815e-05,
      "model_forward_time": 1.7338175773620605,
      "step": 40386
    },
    {
      "epoch": 3.8515090942382815e-05,
      "step": 40386,
      "training_step_time": 6.775038957595825
    },
    {
      "epoch": 3.851604461669922e-05,
      "model_forward_time": 2.8606038093566895,
      "step": 40387
    },
    {
      "epoch": 3.851604461669922e-05,
      "step": 40387,
      "training_step_time": 7.5975141525268555
    },
    {
      "epoch": 3.851699829101562e-05,
      "model_forward_time": 1.3838672637939453,
      "step": 40388
    },
    {
      "epoch": 3.851699829101562e-05,
      "step": 40388,
      "training_step_time": 8.711840629577637
    },
    {
      "epoch": 3.8517951965332034e-05,
      "model_forward_time": 1.7528667449951172,
      "step": 40389
    },
    {
      "epoch": 3.8517951965332034e-05,
      "step": 40389,
      "training_step_time": 7.603333950042725
    },
    {
      "epoch": 3.851890563964844e-05,
      "grad_norm": 0.15207010507583618,
      "learning_rate": 6.949410048011206e-06,
      "loss": 0.0546,
      "step": 40390
    },
    {
      "epoch": 3.851890563964844e-05,
      "model_forward_time": 1.7577624320983887,
      "step": 40390
    },
    {
      "dynamics_loss_avg": 0.04530996512621641,
      "epoch": 3.851890563964844e-05,
      "step": 40390
    },
    {
      "action_loss_avg": 0.0016170557908480987,
      "epoch": 3.851890563964844e-05,
      "step": 40390
    },
    {
      "epoch": 3.851890563964844e-05,
      "step": 40390,
      "training_step_time": 6.731364011764526
    },
    {
      "epoch": 3.851985931396484e-05,
      "model_forward_time": 1.739729642868042,
      "step": 40391
    },
    {
      "epoch": 3.851985931396484e-05,
      "step": 40391,
      "training_step_time": 6.697296142578125
    },
    {
      "epoch": 3.852081298828125e-05,
      "model_forward_time": 1.7409167289733887,
      "step": 40392
    },
    {
      "epoch": 3.852081298828125e-05,
      "step": 40392,
      "training_step_time": 6.699221134185791
    },
    {
      "epoch": 3.852176666259766e-05,
      "model_forward_time": 1.7490122318267822,
      "step": 40393
    },
    {
      "epoch": 3.852176666259766e-05,
      "step": 40393,
      "training_step_time": 6.742985963821411
    },
    {
      "epoch": 3.852272033691406e-05,
      "model_forward_time": 1.7388784885406494,
      "step": 40394
    },
    {
      "epoch": 3.852272033691406e-05,
      "step": 40394,
      "training_step_time": 6.716188192367554
    },
    {
      "epoch": 3.852367401123047e-05,
      "model_forward_time": 1.718414306640625,
      "step": 40395
    },
    {
      "epoch": 3.852367401123047e-05,
      "step": 40395,
      "training_step_time": 6.632707834243774
    },
    {
      "epoch": 3.8524627685546876e-05,
      "model_forward_time": 2.8754427433013916,
      "step": 40396
    },
    {
      "epoch": 3.8524627685546876e-05,
      "step": 40396,
      "training_step_time": 7.58827805519104
    },
    {
      "epoch": 3.852558135986328e-05,
      "model_forward_time": 1.73891282081604,
      "step": 40397
    },
    {
      "epoch": 3.852558135986328e-05,
      "step": 40397,
      "training_step_time": 9.612804889678955
    },
    {
      "epoch": 3.852653503417969e-05,
      "model_forward_time": 1.750317096710205,
      "step": 40398
    },
    {
      "epoch": 3.852653503417969e-05,
      "step": 40398,
      "training_step_time": 7.659541606903076
    },
    {
      "epoch": 3.8527488708496095e-05,
      "model_forward_time": 1.7406535148620605,
      "step": 40399
    },
    {
      "epoch": 3.8527488708496095e-05,
      "step": 40399,
      "training_step_time": 6.705766201019287
    },
    {
      "epoch": 3.85284423828125e-05,
      "grad_norm": 0.0961202010512352,
      "learning_rate": 6.947887319258965e-06,
      "loss": 0.0577,
      "step": 40400
    },
    {
      "epoch": 3.85284423828125e-05,
      "model_forward_time": 1.7463953495025635,
      "step": 40400
    },
    {
      "dynamics_loss_avg": 0.06929052574560046,
      "epoch": 3.85284423828125e-05,
      "step": 40400
    },
    {
      "action_loss_avg": 0.0026727554912213238,
      "epoch": 3.85284423828125e-05,
      "step": 40400
    },
    {
      "epoch": 3.85284423828125e-05,
      "step": 40400,
      "training_step_time": 6.68934178352356
    },
    {
      "epoch": 3.852939605712891e-05,
      "model_forward_time": 1.7335972785949707,
      "step": 40401
    },
    {
      "epoch": 3.852939605712891e-05,
      "step": 40401,
      "training_step_time": 6.676321744918823
    },
    {
      "epoch": 3.8530349731445313e-05,
      "model_forward_time": 1.7375357151031494,
      "step": 40402
    },
    {
      "epoch": 3.8530349731445313e-05,
      "step": 40402,
      "training_step_time": 6.732126235961914
    },
    {
      "epoch": 3.853130340576172e-05,
      "model_forward_time": 1.7447230815887451,
      "step": 40403
    },
    {
      "epoch": 3.853130340576172e-05,
      "step": 40403,
      "training_step_time": 6.649392127990723
    },
    {
      "epoch": 3.853225708007813e-05,
      "model_forward_time": 1.7467453479766846,
      "step": 40404
    },
    {
      "epoch": 3.853225708007813e-05,
      "step": 40404,
      "training_step_time": 6.705029487609863
    },
    {
      "epoch": 3.853321075439453e-05,
      "model_forward_time": 2.867154836654663,
      "step": 40405
    },
    {
      "epoch": 3.853321075439453e-05,
      "step": 40405,
      "training_step_time": 7.571483135223389
    },
    {
      "epoch": 3.8534164428710936e-05,
      "model_forward_time": 1.739795446395874,
      "step": 40406
    },
    {
      "epoch": 3.8534164428710936e-05,
      "step": 40406,
      "training_step_time": 12.818217754364014
    },
    {
      "epoch": 3.853511810302734e-05,
      "model_forward_time": 1.7459745407104492,
      "step": 40407
    },
    {
      "epoch": 3.853511810302734e-05,
      "step": 40407,
      "training_step_time": 7.658342123031616
    },
    {
      "epoch": 3.853607177734375e-05,
      "model_forward_time": 1.7460792064666748,
      "step": 40408
    },
    {
      "epoch": 3.853607177734375e-05,
      "step": 40408,
      "training_step_time": 6.6417036056518555
    },
    {
      "epoch": 3.8537025451660155e-05,
      "model_forward_time": 1.744598150253296,
      "step": 40409
    },
    {
      "epoch": 3.8537025451660155e-05,
      "step": 40409,
      "training_step_time": 6.649390697479248
    },
    {
      "epoch": 3.853797912597656e-05,
      "grad_norm": 0.12777261435985565,
      "learning_rate": 6.9463643774886965e-06,
      "loss": 0.0532,
      "step": 40410
    },
    {
      "epoch": 3.853797912597656e-05,
      "model_forward_time": 1.745908498764038,
      "step": 40410
    },
    {
      "dynamics_loss_avg": 0.05760386791080237,
      "epoch": 3.853797912597656e-05,
      "step": 40410
    },
    {
      "action_loss_avg": 0.0024272233131341636,
      "epoch": 3.853797912597656e-05,
      "step": 40410
    },
    {
      "epoch": 3.853797912597656e-05,
      "step": 40410,
      "training_step_time": 6.68891716003418
    },
    {
      "epoch": 3.853893280029297e-05,
      "model_forward_time": 1.7436516284942627,
      "step": 40411
    },
    {
      "epoch": 3.853893280029297e-05,
      "step": 40411,
      "training_step_time": 6.6661083698272705
    },
    {
      "epoch": 3.8539886474609374e-05,
      "model_forward_time": 1.7423028945922852,
      "step": 40412
    },
    {
      "epoch": 3.8539886474609374e-05,
      "step": 40412,
      "training_step_time": 6.686628818511963
    },
    {
      "epoch": 3.854084014892578e-05,
      "model_forward_time": 1.7395305633544922,
      "step": 40413
    },
    {
      "epoch": 3.854084014892578e-05,
      "step": 40413,
      "training_step_time": 6.660146951675415
    },
    {
      "epoch": 3.854179382324219e-05,
      "model_forward_time": 2.8358237743377686,
      "step": 40414
    },
    {
      "epoch": 3.854179382324219e-05,
      "step": 40414,
      "training_step_time": 7.553997278213501
    },
    {
      "epoch": 3.854274749755859e-05,
      "model_forward_time": 1.7428293228149414,
      "step": 40415
    },
    {
      "epoch": 3.854274749755859e-05,
      "step": 40415,
      "training_step_time": 8.423373460769653
    },
    {
      "epoch": 3.8543701171875e-05,
      "model_forward_time": 1.7497327327728271,
      "step": 40416
    },
    {
      "epoch": 3.8543701171875e-05,
      "step": 40416,
      "training_step_time": 7.721115350723267
    },
    {
      "epoch": 3.854465484619141e-05,
      "model_forward_time": 1.744246244430542,
      "step": 40417
    },
    {
      "epoch": 3.854465484619141e-05,
      "step": 40417,
      "training_step_time": 6.624869346618652
    },
    {
      "epoch": 3.854560852050781e-05,
      "model_forward_time": 1.73521089553833,
      "step": 40418
    },
    {
      "epoch": 3.854560852050781e-05,
      "step": 40418,
      "training_step_time": 6.651973247528076
    },
    {
      "epoch": 3.8546562194824216e-05,
      "model_forward_time": 1.7380239963531494,
      "step": 40419
    },
    {
      "epoch": 3.8546562194824216e-05,
      "step": 40419,
      "training_step_time": 6.626059055328369
    },
    {
      "epoch": 3.854751586914063e-05,
      "grad_norm": 0.1495128870010376,
      "learning_rate": 6.944841222866947e-06,
      "loss": 0.0598,
      "step": 40420
    },
    {
      "epoch": 3.854751586914063e-05,
      "model_forward_time": 1.7309446334838867,
      "step": 40420
    },
    {
      "dynamics_loss_avg": 0.0385827736929059,
      "epoch": 3.854751586914063e-05,
      "step": 40420
    },
    {
      "action_loss_avg": 0.00109370457357727,
      "epoch": 3.854751586914063e-05,
      "step": 40420
    },
    {
      "epoch": 3.854751586914063e-05,
      "step": 40420,
      "training_step_time": 6.639694690704346
    },
    {
      "epoch": 3.854846954345703e-05,
      "model_forward_time": 1.7400925159454346,
      "step": 40421
    },
    {
      "epoch": 3.854846954345703e-05,
      "step": 40421,
      "training_step_time": 6.665552616119385
    },
    {
      "epoch": 3.8549423217773435e-05,
      "model_forward_time": 1.7437641620635986,
      "step": 40422
    },
    {
      "epoch": 3.8549423217773435e-05,
      "step": 40422,
      "training_step_time": 6.666287422180176
    },
    {
      "epoch": 3.8550376892089845e-05,
      "model_forward_time": 1.7427442073822021,
      "step": 40423
    },
    {
      "epoch": 3.8550376892089845e-05,
      "step": 40423,
      "training_step_time": 7.548191785812378
    },
    {
      "epoch": 3.855133056640625e-05,
      "model_forward_time": 2.886622190475464,
      "step": 40424
    },
    {
      "epoch": 3.855133056640625e-05,
      "step": 40424,
      "training_step_time": 7.634816646575928
    },
    {
      "epoch": 3.8552284240722654e-05,
      "model_forward_time": 1.7442915439605713,
      "step": 40425
    },
    {
      "epoch": 3.8552284240722654e-05,
      "step": 40425,
      "training_step_time": 7.719482421875
    },
    {
      "epoch": 3.8553237915039064e-05,
      "model_forward_time": 1.7417302131652832,
      "step": 40426
    },
    {
      "epoch": 3.8553237915039064e-05,
      "step": 40426,
      "training_step_time": 7.6917572021484375
    },
    {
      "epoch": 3.855419158935547e-05,
      "model_forward_time": 1.7496790885925293,
      "step": 40427
    },
    {
      "epoch": 3.855419158935547e-05,
      "step": 40427,
      "training_step_time": 6.670190334320068
    },
    {
      "epoch": 3.855514526367187e-05,
      "model_forward_time": 1.7424538135528564,
      "step": 40428
    },
    {
      "epoch": 3.855514526367187e-05,
      "step": 40428,
      "training_step_time": 6.637358903884888
    },
    {
      "epoch": 3.855609893798828e-05,
      "model_forward_time": 1.7369754314422607,
      "step": 40429
    },
    {
      "epoch": 3.855609893798828e-05,
      "step": 40429,
      "training_step_time": 6.7016520500183105
    },
    {
      "epoch": 3.855705261230469e-05,
      "grad_norm": 0.12242158502340317,
      "learning_rate": 6.9433178555602845e-06,
      "loss": 0.0605,
      "step": 40430
    },
    {
      "epoch": 3.855705261230469e-05,
      "model_forward_time": 1.7583224773406982,
      "step": 40430
    },
    {
      "dynamics_loss_avg": 0.04798587318509817,
      "epoch": 3.855705261230469e-05,
      "step": 40430
    },
    {
      "action_loss_avg": 0.0016811475448776036,
      "epoch": 3.855705261230469e-05,
      "step": 40430
    },
    {
      "epoch": 3.855705261230469e-05,
      "step": 40430,
      "training_step_time": 6.706094026565552
    },
    {
      "epoch": 3.855800628662109e-05,
      "model_forward_time": 1.7495918273925781,
      "step": 40431
    },
    {
      "epoch": 3.855800628662109e-05,
      "step": 40431,
      "training_step_time": 6.706201076507568
    },
    {
      "epoch": 3.85589599609375e-05,
      "model_forward_time": 1.7430081367492676,
      "step": 40432
    },
    {
      "epoch": 3.85589599609375e-05,
      "step": 40432,
      "training_step_time": 6.680540323257446
    },
    {
      "epoch": 3.8559913635253906e-05,
      "model_forward_time": 2.911583423614502,
      "step": 40433
    },
    {
      "epoch": 3.8559913635253906e-05,
      "step": 40433,
      "training_step_time": 7.6789867877960205
    },
    {
      "epoch": 3.856086730957031e-05,
      "model_forward_time": 1.7491161823272705,
      "step": 40434
    },
    {
      "epoch": 3.856086730957031e-05,
      "step": 40434,
      "training_step_time": 7.82269287109375
    },
    {
      "epoch": 3.856182098388672e-05,
      "model_forward_time": 1.7375214099884033,
      "step": 40435
    },
    {
      "epoch": 3.856182098388672e-05,
      "step": 40435,
      "training_step_time": 7.683521509170532
    },
    {
      "epoch": 3.8562774658203125e-05,
      "model_forward_time": 1.7453229427337646,
      "step": 40436
    },
    {
      "epoch": 3.8562774658203125e-05,
      "step": 40436,
      "training_step_time": 6.721441984176636
    },
    {
      "epoch": 3.856372833251953e-05,
      "model_forward_time": 1.7414796352386475,
      "step": 40437
    },
    {
      "epoch": 3.856372833251953e-05,
      "step": 40437,
      "training_step_time": 6.708375930786133
    },
    {
      "epoch": 3.856468200683594e-05,
      "model_forward_time": 1.7445545196533203,
      "step": 40438
    },
    {
      "epoch": 3.856468200683594e-05,
      "step": 40438,
      "training_step_time": 6.717276096343994
    },
    {
      "epoch": 3.8565635681152344e-05,
      "model_forward_time": 1.7392442226409912,
      "step": 40439
    },
    {
      "epoch": 3.8565635681152344e-05,
      "step": 40439,
      "training_step_time": 6.707046270370483
    },
    {
      "epoch": 3.856658935546875e-05,
      "grad_norm": 0.11043417453765869,
      "learning_rate": 6.941794275735303e-06,
      "loss": 0.0553,
      "step": 40440
    },
    {
      "epoch": 3.856658935546875e-05,
      "model_forward_time": 1.7607407569885254,
      "step": 40440
    },
    {
      "dynamics_loss_avg": 0.049351486191153524,
      "epoch": 3.856658935546875e-05,
      "step": 40440
    },
    {
      "action_loss_avg": 0.001437384681776166,
      "epoch": 3.856658935546875e-05,
      "step": 40440
    },
    {
      "epoch": 3.856658935546875e-05,
      "step": 40440,
      "training_step_time": 6.706816673278809
    },
    {
      "epoch": 3.856754302978516e-05,
      "model_forward_time": 1.7430975437164307,
      "step": 40441
    },
    {
      "epoch": 3.856754302978516e-05,
      "step": 40441,
      "training_step_time": 6.71280312538147
    },
    {
      "epoch": 3.856849670410156e-05,
      "model_forward_time": 2.9014296531677246,
      "step": 40442
    },
    {
      "epoch": 3.856849670410156e-05,
      "step": 40442,
      "training_step_time": 7.6665098667144775
    },
    {
      "epoch": 3.856945037841797e-05,
      "model_forward_time": 1.7402911186218262,
      "step": 40443
    },
    {
      "epoch": 3.856945037841797e-05,
      "step": 40443,
      "training_step_time": 10.850760221481323
    },
    {
      "epoch": 3.857040405273438e-05,
      "model_forward_time": 1.742952585220337,
      "step": 40444
    },
    {
      "epoch": 3.857040405273438e-05,
      "step": 40444,
      "training_step_time": 7.663799047470093
    },
    {
      "epoch": 3.857135772705078e-05,
      "model_forward_time": 1.7383830547332764,
      "step": 40445
    },
    {
      "epoch": 3.857135772705078e-05,
      "step": 40445,
      "training_step_time": 6.749892711639404
    },
    {
      "epoch": 3.8572311401367186e-05,
      "model_forward_time": 1.741182804107666,
      "step": 40446
    },
    {
      "epoch": 3.8572311401367186e-05,
      "step": 40446,
      "training_step_time": 6.704124927520752
    },
    {
      "epoch": 3.8573265075683596e-05,
      "model_forward_time": 1.738809585571289,
      "step": 40447
    },
    {
      "epoch": 3.8573265075683596e-05,
      "step": 40447,
      "training_step_time": 6.71397852897644
    },
    {
      "epoch": 3.857421875e-05,
      "model_forward_time": 1.7444918155670166,
      "step": 40448
    },
    {
      "epoch": 3.857421875e-05,
      "step": 40448,
      "training_step_time": 6.706133604049683
    },
    {
      "epoch": 3.8575172424316404e-05,
      "model_forward_time": 1.7377650737762451,
      "step": 40449
    },
    {
      "epoch": 3.8575172424316404e-05,
      "step": 40449,
      "training_step_time": 6.71490740776062
    },
    {
      "epoch": 3.8576126098632815e-05,
      "grad_norm": 0.12389323115348816,
      "learning_rate": 6.94027048355862e-06,
      "loss": 0.0615,
      "step": 40450
    },
    {
      "epoch": 3.8576126098632815e-05,
      "model_forward_time": 1.7306106090545654,
      "step": 40450
    },
    {
      "dynamics_loss_avg": 0.06371707934886217,
      "epoch": 3.8576126098632815e-05,
      "step": 40450
    },
    {
      "action_loss_avg": 0.003759796533267945,
      "epoch": 3.8576126098632815e-05,
      "step": 40450
    },
    {
      "epoch": 3.8576126098632815e-05,
      "step": 40450,
      "training_step_time": 6.679815292358398
    },
    {
      "epoch": 3.857707977294922e-05,
      "model_forward_time": 2.890636682510376,
      "step": 40451
    },
    {
      "epoch": 3.857707977294922e-05,
      "step": 40451,
      "training_step_time": 7.666441440582275
    },
    {
      "epoch": 3.857803344726562e-05,
      "model_forward_time": 1.7277121543884277,
      "step": 40452
    },
    {
      "epoch": 3.857803344726562e-05,
      "step": 40452,
      "training_step_time": 10.08743953704834
    },
    {
      "epoch": 3.8578987121582034e-05,
      "model_forward_time": 1.7473511695861816,
      "step": 40453
    },
    {
      "epoch": 3.8578987121582034e-05,
      "step": 40453,
      "training_step_time": 8.588301420211792
    },
    {
      "epoch": 3.857994079589844e-05,
      "model_forward_time": 1.751451015472412,
      "step": 40454
    },
    {
      "epoch": 3.857994079589844e-05,
      "step": 40454,
      "training_step_time": 7.565026521682739
    },
    {
      "epoch": 3.858089447021484e-05,
      "model_forward_time": 1.7478411197662354,
      "step": 40455
    },
    {
      "epoch": 3.858089447021484e-05,
      "step": 40455,
      "training_step_time": 6.730987548828125
    },
    {
      "epoch": 3.858184814453125e-05,
      "model_forward_time": 1.7480945587158203,
      "step": 40456
    },
    {
      "epoch": 3.858184814453125e-05,
      "step": 40456,
      "training_step_time": 6.734870910644531
    },
    {
      "epoch": 3.858280181884766e-05,
      "model_forward_time": 1.7489140033721924,
      "step": 40457
    },
    {
      "epoch": 3.858280181884766e-05,
      "step": 40457,
      "training_step_time": 6.747221231460571
    },
    {
      "epoch": 3.858375549316406e-05,
      "model_forward_time": 1.748985767364502,
      "step": 40458
    },
    {
      "epoch": 3.858375549316406e-05,
      "step": 40458,
      "training_step_time": 6.694078683853149
    },
    {
      "epoch": 3.858470916748047e-05,
      "model_forward_time": 1.7571067810058594,
      "step": 40459
    },
    {
      "epoch": 3.858470916748047e-05,
      "step": 40459,
      "training_step_time": 6.705971002578735
    },
    {
      "epoch": 3.8585662841796876e-05,
      "grad_norm": 0.11311215907335281,
      "learning_rate": 6.9387464791968715e-06,
      "loss": 0.0588,
      "step": 40460
    },
    {
      "epoch": 3.8585662841796876e-05,
      "model_forward_time": 2.9054486751556396,
      "step": 40460
    },
    {
      "dynamics_loss_avg": 0.05168574377894401,
      "epoch": 3.8585662841796876e-05,
      "step": 40460
    },
    {
      "action_loss_avg": 0.0014853363420115784,
      "epoch": 3.8585662841796876e-05,
      "step": 40460
    },
    {
      "epoch": 3.8585662841796876e-05,
      "step": 40460,
      "training_step_time": 7.621587038040161
    },
    {
      "epoch": 3.858661651611328e-05,
      "model_forward_time": 1.75614333152771,
      "step": 40461
    },
    {
      "epoch": 3.858661651611328e-05,
      "step": 40461,
      "training_step_time": 9.847604990005493
    },
    {
      "epoch": 3.858757019042969e-05,
      "model_forward_time": 1.7444682121276855,
      "step": 40462
    },
    {
      "epoch": 3.858757019042969e-05,
      "step": 40462,
      "training_step_time": 10.367738246917725
    },
    {
      "epoch": 3.8588523864746095e-05,
      "model_forward_time": 1.7511930465698242,
      "step": 40463
    },
    {
      "epoch": 3.8588523864746095e-05,
      "step": 40463,
      "training_step_time": 7.608883619308472
    },
    {
      "epoch": 3.85894775390625e-05,
      "model_forward_time": 1.7507226467132568,
      "step": 40464
    },
    {
      "epoch": 3.85894775390625e-05,
      "step": 40464,
      "training_step_time": 6.74793267250061
    },
    {
      "epoch": 3.859043121337891e-05,
      "model_forward_time": 1.7506005764007568,
      "step": 40465
    },
    {
      "epoch": 3.859043121337891e-05,
      "step": 40465,
      "training_step_time": 6.746801137924194
    },
    {
      "epoch": 3.8591384887695314e-05,
      "model_forward_time": 1.7535381317138672,
      "step": 40466
    },
    {
      "epoch": 3.8591384887695314e-05,
      "step": 40466,
      "training_step_time": 6.737971305847168
    },
    {
      "epoch": 3.859233856201172e-05,
      "model_forward_time": 1.7435500621795654,
      "step": 40467
    },
    {
      "epoch": 3.859233856201172e-05,
      "step": 40467,
      "training_step_time": 6.741931438446045
    },
    {
      "epoch": 3.859329223632813e-05,
      "model_forward_time": 1.7450206279754639,
      "step": 40468
    },
    {
      "epoch": 3.859329223632813e-05,
      "step": 40468,
      "training_step_time": 6.731513977050781
    },
    {
      "epoch": 3.859424591064453e-05,
      "model_forward_time": 1.7825520038604736,
      "step": 40469
    },
    {
      "epoch": 3.859424591064453e-05,
      "step": 40469,
      "training_step_time": 7.327001094818115
    },
    {
      "epoch": 3.8595199584960936e-05,
      "grad_norm": 0.09357977658510208,
      "learning_rate": 6.937222262816725e-06,
      "loss": 0.0613,
      "step": 40470
    },
    {
      "epoch": 3.8595199584960936e-05,
      "model_forward_time": 2.8672096729278564,
      "step": 40470
    },
    {
      "dynamics_loss_avg": 0.057687551341950895,
      "epoch": 3.8595199584960936e-05,
      "step": 40470
    },
    {
      "action_loss_avg": 0.0017723098571877927,
      "epoch": 3.8595199584960936e-05,
      "step": 40470
    },
    {
      "epoch": 3.8595199584960936e-05,
      "step": 40470,
      "training_step_time": 10.696234941482544
    },
    {
      "epoch": 3.859615325927734e-05,
      "model_forward_time": 1.7432243824005127,
      "step": 40471
    },
    {
      "epoch": 3.859615325927734e-05,
      "step": 40471,
      "training_step_time": 10.827133417129517
    },
    {
      "epoch": 3.859710693359375e-05,
      "model_forward_time": 1.7448062896728516,
      "step": 40472
    },
    {
      "epoch": 3.859710693359375e-05,
      "step": 40472,
      "training_step_time": 7.629016637802124
    },
    {
      "epoch": 3.8598060607910155e-05,
      "model_forward_time": 1.744189977645874,
      "step": 40473
    },
    {
      "epoch": 3.8598060607910155e-05,
      "step": 40473,
      "training_step_time": 6.659396648406982
    },
    {
      "epoch": 3.859901428222656e-05,
      "model_forward_time": 1.7462267875671387,
      "step": 40474
    },
    {
      "epoch": 3.859901428222656e-05,
      "step": 40474,
      "training_step_time": 6.670900344848633
    },
    {
      "epoch": 3.859996795654297e-05,
      "model_forward_time": 1.750659465789795,
      "step": 40475
    },
    {
      "epoch": 3.859996795654297e-05,
      "step": 40475,
      "training_step_time": 6.709181308746338
    },
    {
      "epoch": 3.8600921630859374e-05,
      "model_forward_time": 1.7502496242523193,
      "step": 40476
    },
    {
      "epoch": 3.8600921630859374e-05,
      "step": 40476,
      "training_step_time": 6.718291521072388
    },
    {
      "epoch": 3.860187530517578e-05,
      "model_forward_time": 1.7544817924499512,
      "step": 40477
    },
    {
      "epoch": 3.860187530517578e-05,
      "step": 40477,
      "training_step_time": 6.72402286529541
    },
    {
      "epoch": 3.860282897949219e-05,
      "model_forward_time": 1.7543485164642334,
      "step": 40478
    },
    {
      "epoch": 3.860282897949219e-05,
      "step": 40478,
      "training_step_time": 7.150658845901489
    },
    {
      "epoch": 3.860378265380859e-05,
      "model_forward_time": 2.8922312259674072,
      "step": 40479
    },
    {
      "epoch": 3.860378265380859e-05,
      "step": 40479,
      "training_step_time": 8.662957668304443
    },
    {
      "epoch": 3.8604736328125e-05,
      "grad_norm": 0.1110154464840889,
      "learning_rate": 6.935697834584864e-06,
      "loss": 0.0602,
      "step": 40480
    },
    {
      "epoch": 3.8604736328125e-05,
      "model_forward_time": 1.7489867210388184,
      "step": 40480
    },
    {
      "dynamics_loss_avg": 0.05879234541207552,
      "epoch": 3.8604736328125e-05,
      "step": 40480
    },
    {
      "action_loss_avg": 0.001458338863449171,
      "epoch": 3.8604736328125e-05,
      "step": 40480
    },
    {
      "epoch": 3.8604736328125e-05,
      "step": 40480,
      "training_step_time": 8.5331130027771
    },
    {
      "epoch": 3.860569000244141e-05,
      "model_forward_time": 1.7434263229370117,
      "step": 40481
    },
    {
      "epoch": 3.860569000244141e-05,
      "step": 40481,
      "training_step_time": 7.718012094497681
    },
    {
      "epoch": 3.860664367675781e-05,
      "model_forward_time": 1.7382550239562988,
      "step": 40482
    },
    {
      "epoch": 3.860664367675781e-05,
      "step": 40482,
      "training_step_time": 7.559304237365723
    },
    {
      "epoch": 3.8607597351074216e-05,
      "model_forward_time": 1.7794883251190186,
      "step": 40483
    },
    {
      "epoch": 3.8607597351074216e-05,
      "step": 40483,
      "training_step_time": 6.743741512298584
    },
    {
      "epoch": 3.860855102539063e-05,
      "model_forward_time": 1.757002592086792,
      "step": 40484
    },
    {
      "epoch": 3.860855102539063e-05,
      "step": 40484,
      "training_step_time": 6.7481536865234375
    },
    {
      "epoch": 3.860950469970703e-05,
      "model_forward_time": 1.7432396411895752,
      "step": 40485
    },
    {
      "epoch": 3.860950469970703e-05,
      "step": 40485,
      "training_step_time": 6.773868083953857
    },
    {
      "epoch": 3.8610458374023435e-05,
      "model_forward_time": 1.744126558303833,
      "step": 40486
    },
    {
      "epoch": 3.8610458374023435e-05,
      "step": 40486,
      "training_step_time": 6.7378504276275635
    },
    {
      "epoch": 3.8611412048339846e-05,
      "model_forward_time": 1.7454097270965576,
      "step": 40487
    },
    {
      "epoch": 3.8611412048339846e-05,
      "step": 40487,
      "training_step_time": 6.764135360717773
    },
    {
      "epoch": 3.861236572265625e-05,
      "model_forward_time": 2.907834053039551,
      "step": 40488
    },
    {
      "epoch": 3.861236572265625e-05,
      "step": 40488,
      "training_step_time": 8.31338882446289
    },
    {
      "epoch": 3.8613319396972654e-05,
      "model_forward_time": 1.7521405220031738,
      "step": 40489
    },
    {
      "epoch": 3.8613319396972654e-05,
      "step": 40489,
      "training_step_time": 11.522435903549194
    },
    {
      "epoch": 3.8614273071289064e-05,
      "grad_norm": 0.14894266426563263,
      "learning_rate": 6.934173194667999e-06,
      "loss": 0.0653,
      "step": 40490
    },
    {
      "epoch": 3.8614273071289064e-05,
      "model_forward_time": 1.7516834735870361,
      "step": 40490
    },
    {
      "dynamics_loss_avg": 0.0518732188269496,
      "epoch": 3.8614273071289064e-05,
      "step": 40490
    },
    {
      "action_loss_avg": 0.0011188671138370409,
      "epoch": 3.8614273071289064e-05,
      "step": 40490
    },
    {
      "epoch": 3.8614273071289064e-05,
      "step": 40490,
      "training_step_time": 8.790841102600098
    },
    {
      "epoch": 3.861522674560547e-05,
      "model_forward_time": 1.7503681182861328,
      "step": 40491
    },
    {
      "epoch": 3.861522674560547e-05,
      "step": 40491,
      "training_step_time": 7.592642545700073
    },
    {
      "epoch": 3.861618041992187e-05,
      "model_forward_time": 1.7484502792358398,
      "step": 40492
    },
    {
      "epoch": 3.861618041992187e-05,
      "step": 40492,
      "training_step_time": 6.730778217315674
    },
    {
      "epoch": 3.861713409423828e-05,
      "model_forward_time": 1.7569177150726318,
      "step": 40493
    },
    {
      "epoch": 3.861713409423828e-05,
      "step": 40493,
      "training_step_time": 6.749456167221069
    },
    {
      "epoch": 3.861808776855469e-05,
      "model_forward_time": 1.7561829090118408,
      "step": 40494
    },
    {
      "epoch": 3.861808776855469e-05,
      "step": 40494,
      "training_step_time": 6.744962930679321
    },
    {
      "epoch": 3.861904144287109e-05,
      "model_forward_time": 1.7513604164123535,
      "step": 40495
    },
    {
      "epoch": 3.861904144287109e-05,
      "step": 40495,
      "training_step_time": 6.7892069816589355
    },
    {
      "epoch": 3.86199951171875e-05,
      "model_forward_time": 1.7548210620880127,
      "step": 40496
    },
    {
      "epoch": 3.86199951171875e-05,
      "step": 40496,
      "training_step_time": 7.0890302658081055
    },
    {
      "epoch": 3.8620948791503906e-05,
      "model_forward_time": 2.8972976207733154,
      "step": 40497
    },
    {
      "epoch": 3.8620948791503906e-05,
      "step": 40497,
      "training_step_time": 9.461417436599731
    },
    {
      "epoch": 3.862190246582031e-05,
      "model_forward_time": 1.7487328052520752,
      "step": 40498
    },
    {
      "epoch": 3.862190246582031e-05,
      "step": 40498,
      "training_step_time": 8.810312032699585
    },
    {
      "epoch": 3.862285614013672e-05,
      "model_forward_time": 1.789862871170044,
      "step": 40499
    },
    {
      "epoch": 3.862285614013672e-05,
      "step": 40499,
      "training_step_time": 7.735426902770996
    },
    {
      "epoch": 3.8623809814453125e-05,
      "grad_norm": 0.13320450484752655,
      "learning_rate": 6.93264834323286e-06,
      "loss": 0.0652,
      "step": 40500
    },
    {
      "epoch": 3.8623809814453125e-05,
      "model_forward_time": 1.7581019401550293,
      "step": 40500
    },
    {
      "dynamics_loss_avg": 0.050301789119839665,
      "epoch": 3.8623809814453125e-05,
      "step": 40500
    },
    {
      "action_loss_avg": 0.002338592379237525,
      "epoch": 3.8623809814453125e-05,
      "step": 40500
    },
    {
      "epoch": 3.8623809814453125e-05,
      "step": 40500,
      "training_step_time": 7.570632219314575
    },
    {
      "epoch": 3.862476348876953e-05,
      "model_forward_time": 1.7490012645721436,
      "step": 40501
    },
    {
      "epoch": 3.862476348876953e-05,
      "step": 40501,
      "training_step_time": 6.714972972869873
    },
    {
      "epoch": 3.862571716308594e-05,
      "model_forward_time": 1.753871202468872,
      "step": 40502
    },
    {
      "epoch": 3.862571716308594e-05,
      "step": 40502,
      "training_step_time": 6.719520807266235
    },
    {
      "epoch": 3.8626670837402344e-05,
      "model_forward_time": 1.752150058746338,
      "step": 40503
    },
    {
      "epoch": 3.8626670837402344e-05,
      "step": 40503,
      "training_step_time": 6.708967924118042
    },
    {
      "epoch": 3.862762451171875e-05,
      "model_forward_time": 1.753563404083252,
      "step": 40504
    },
    {
      "epoch": 3.862762451171875e-05,
      "step": 40504,
      "training_step_time": 6.687432765960693
    },
    {
      "epoch": 3.862857818603516e-05,
      "model_forward_time": 1.749908208847046,
      "step": 40505
    },
    {
      "epoch": 3.862857818603516e-05,
      "step": 40505,
      "training_step_time": 7.081238508224487
    },
    {
      "epoch": 3.862953186035156e-05,
      "model_forward_time": 2.8912084102630615,
      "step": 40506
    },
    {
      "epoch": 3.862953186035156e-05,
      "step": 40506,
      "training_step_time": 7.646939992904663
    },
    {
      "epoch": 3.863048553466797e-05,
      "model_forward_time": 1.7634456157684326,
      "step": 40507
    },
    {
      "epoch": 3.863048553466797e-05,
      "step": 40507,
      "training_step_time": 7.77679181098938
    },
    {
      "epoch": 3.863143920898438e-05,
      "model_forward_time": 1.7434570789337158,
      "step": 40508
    },
    {
      "epoch": 3.863143920898438e-05,
      "step": 40508,
      "training_step_time": 8.53296446800232
    },
    {
      "epoch": 3.863239288330078e-05,
      "model_forward_time": 1.7455980777740479,
      "step": 40509
    },
    {
      "epoch": 3.863239288330078e-05,
      "step": 40509,
      "training_step_time": 8.485318899154663
    },
    {
      "epoch": 3.8633346557617186e-05,
      "grad_norm": 0.1330685317516327,
      "learning_rate": 6.931123280446206e-06,
      "loss": 0.0623,
      "step": 40510
    },
    {
      "epoch": 3.8633346557617186e-05,
      "model_forward_time": 1.7701985836029053,
      "step": 40510
    },
    {
      "dynamics_loss_avg": 0.05877172537147999,
      "epoch": 3.8633346557617186e-05,
      "step": 40510
    },
    {
      "action_loss_avg": 0.0020632363797631116,
      "epoch": 3.8633346557617186e-05,
      "step": 40510
    },
    {
      "epoch": 3.8633346557617186e-05,
      "step": 40510,
      "training_step_time": 6.785017013549805
    },
    {
      "epoch": 3.8634300231933596e-05,
      "model_forward_time": 1.7542896270751953,
      "step": 40511
    },
    {
      "epoch": 3.8634300231933596e-05,
      "step": 40511,
      "training_step_time": 6.7623207569122314
    },
    {
      "epoch": 3.863525390625e-05,
      "model_forward_time": 1.7513625621795654,
      "step": 40512
    },
    {
      "epoch": 3.863525390625e-05,
      "step": 40512,
      "training_step_time": 6.746166229248047
    },
    {
      "epoch": 3.8636207580566404e-05,
      "model_forward_time": 1.753300428390503,
      "step": 40513
    },
    {
      "epoch": 3.8636207580566404e-05,
      "step": 40513,
      "training_step_time": 6.765763282775879
    },
    {
      "epoch": 3.8637161254882815e-05,
      "model_forward_time": 1.745553731918335,
      "step": 40514
    },
    {
      "epoch": 3.8637161254882815e-05,
      "step": 40514,
      "training_step_time": 7.1586949825286865
    },
    {
      "epoch": 3.863811492919922e-05,
      "model_forward_time": 1.7564547061920166,
      "step": 40515
    },
    {
      "epoch": 3.863811492919922e-05,
      "step": 40515,
      "training_step_time": 9.6546950340271
    },
    {
      "epoch": 3.863906860351562e-05,
      "model_forward_time": 1.755122423171997,
      "step": 40516
    },
    {
      "epoch": 3.863906860351562e-05,
      "step": 40516,
      "training_step_time": 7.762825012207031
    },
    {
      "epoch": 3.8640022277832034e-05,
      "model_forward_time": 1.7657079696655273,
      "step": 40517
    },
    {
      "epoch": 3.8640022277832034e-05,
      "step": 40517,
      "training_step_time": 8.647186517715454
    },
    {
      "epoch": 3.864097595214844e-05,
      "model_forward_time": 1.7509338855743408,
      "step": 40518
    },
    {
      "epoch": 3.864097595214844e-05,
      "step": 40518,
      "training_step_time": 8.656020641326904
    },
    {
      "epoch": 3.864192962646484e-05,
      "model_forward_time": 1.7609281539916992,
      "step": 40519
    },
    {
      "epoch": 3.864192962646484e-05,
      "step": 40519,
      "training_step_time": 7.524613857269287
    },
    {
      "epoch": 3.864288330078125e-05,
      "grad_norm": 0.13038507103919983,
      "learning_rate": 6.929598006474811e-06,
      "loss": 0.0653,
      "step": 40520
    },
    {
      "epoch": 3.864288330078125e-05,
      "model_forward_time": 1.7218165397644043,
      "step": 40520
    },
    {
      "dynamics_loss_avg": 0.06416032090783119,
      "epoch": 3.864288330078125e-05,
      "step": 40520
    },
    {
      "action_loss_avg": 0.0017835350474342704,
      "epoch": 3.864288330078125e-05,
      "step": 40520
    },
    {
      "epoch": 3.864288330078125e-05,
      "step": 40520,
      "training_step_time": 6.797544479370117
    },
    {
      "epoch": 3.864383697509766e-05,
      "model_forward_time": 1.746258020401001,
      "step": 40521
    },
    {
      "epoch": 3.864383697509766e-05,
      "step": 40521,
      "training_step_time": 6.729743480682373
    },
    {
      "epoch": 3.864479064941406e-05,
      "model_forward_time": 1.3824725151062012,
      "step": 40522
    },
    {
      "epoch": 3.864479064941406e-05,
      "step": 40522,
      "training_step_time": 6.7549889087677
    },
    {
      "epoch": 3.864574432373047e-05,
      "model_forward_time": 1.7578952312469482,
      "step": 40523
    },
    {
      "epoch": 3.864574432373047e-05,
      "step": 40523,
      "training_step_time": 6.7523581981658936
    },
    {
      "epoch": 3.8646697998046876e-05,
      "model_forward_time": 1.7586698532104492,
      "step": 40524
    },
    {
      "epoch": 3.8646697998046876e-05,
      "step": 40524,
      "training_step_time": 7.561492443084717
    },
    {
      "epoch": 3.864765167236328e-05,
      "model_forward_time": 2.8596746921539307,
      "step": 40525
    },
    {
      "epoch": 3.864765167236328e-05,
      "step": 40525,
      "training_step_time": 8.388482809066772
    },
    {
      "epoch": 3.864860534667969e-05,
      "model_forward_time": 1.7304792404174805,
      "step": 40526
    },
    {
      "epoch": 3.864860534667969e-05,
      "step": 40526,
      "training_step_time": 9.315098524093628
    },
    {
      "epoch": 3.8649559020996095e-05,
      "model_forward_time": 1.7385430335998535,
      "step": 40527
    },
    {
      "epoch": 3.8649559020996095e-05,
      "step": 40527,
      "training_step_time": 8.232000589370728
    },
    {
      "epoch": 3.86505126953125e-05,
      "model_forward_time": 1.743966817855835,
      "step": 40528
    },
    {
      "epoch": 3.86505126953125e-05,
      "step": 40528,
      "training_step_time": 7.456509113311768
    },
    {
      "epoch": 3.865146636962891e-05,
      "model_forward_time": 1.74076509475708,
      "step": 40529
    },
    {
      "epoch": 3.865146636962891e-05,
      "step": 40529,
      "training_step_time": 6.735138177871704
    },
    {
      "epoch": 3.8652420043945314e-05,
      "grad_norm": 0.1550241857767105,
      "learning_rate": 6.928072521485481e-06,
      "loss": 0.065,
      "step": 40530
    },
    {
      "epoch": 3.8652420043945314e-05,
      "model_forward_time": 1.7285082340240479,
      "step": 40530
    },
    {
      "dynamics_loss_avg": 0.07020199876278639,
      "epoch": 3.8652420043945314e-05,
      "step": 40530
    },
    {
      "action_loss_avg": 0.0028358789044432343,
      "epoch": 3.8652420043945314e-05,
      "step": 40530
    },
    {
      "epoch": 3.8652420043945314e-05,
      "step": 40530,
      "training_step_time": 6.768593788146973
    },
    {
      "epoch": 3.865337371826172e-05,
      "model_forward_time": 1.756056308746338,
      "step": 40531
    },
    {
      "epoch": 3.865337371826172e-05,
      "step": 40531,
      "training_step_time": 6.778894424438477
    },
    {
      "epoch": 3.865432739257812e-05,
      "model_forward_time": 1.7511043548583984,
      "step": 40532
    },
    {
      "epoch": 3.865432739257812e-05,
      "step": 40532,
      "training_step_time": 6.7487523555755615
    },
    {
      "epoch": 3.865528106689453e-05,
      "model_forward_time": 1.7530672550201416,
      "step": 40533
    },
    {
      "epoch": 3.865528106689453e-05,
      "step": 40533,
      "training_step_time": 7.517186164855957
    },
    {
      "epoch": 3.8656234741210936e-05,
      "model_forward_time": 2.918083906173706,
      "step": 40534
    },
    {
      "epoch": 3.8656234741210936e-05,
      "step": 40534,
      "training_step_time": 9.970641851425171
    },
    {
      "epoch": 3.865718841552734e-05,
      "model_forward_time": 1.753523826599121,
      "step": 40535
    },
    {
      "epoch": 3.865718841552734e-05,
      "step": 40535,
      "training_step_time": 8.722699403762817
    },
    {
      "epoch": 3.865814208984375e-05,
      "model_forward_time": 1.7551069259643555,
      "step": 40536
    },
    {
      "epoch": 3.865814208984375e-05,
      "step": 40536,
      "training_step_time": 7.662403345108032
    },
    {
      "epoch": 3.8659095764160155e-05,
      "model_forward_time": 1.7456021308898926,
      "step": 40537
    },
    {
      "epoch": 3.8659095764160155e-05,
      "step": 40537,
      "training_step_time": 7.4830005168914795
    },
    {
      "epoch": 3.866004943847656e-05,
      "model_forward_time": 1.7745325565338135,
      "step": 40538
    },
    {
      "epoch": 3.866004943847656e-05,
      "step": 40538,
      "training_step_time": 7.542587757110596
    },
    {
      "epoch": 3.866100311279297e-05,
      "model_forward_time": 1.7588427066802979,
      "step": 40539
    },
    {
      "epoch": 3.866100311279297e-05,
      "step": 40539,
      "training_step_time": 6.718653917312622
    },
    {
      "epoch": 3.8661956787109374e-05,
      "grad_norm": 0.10770061612129211,
      "learning_rate": 6.926546825645037e-06,
      "loss": 0.0607,
      "step": 40540
    },
    {
      "epoch": 3.8661956787109374e-05,
      "model_forward_time": 1.7418386936187744,
      "step": 40540
    },
    {
      "dynamics_loss_avg": 0.06025469414889813,
      "epoch": 3.8661956787109374e-05,
      "step": 40540
    },
    {
      "action_loss_avg": 0.0018445821770001203,
      "epoch": 3.8661956787109374e-05,
      "step": 40540
    },
    {
      "epoch": 3.8661956787109374e-05,
      "step": 40540,
      "training_step_time": 6.795592308044434
    },
    {
      "epoch": 3.866291046142578e-05,
      "model_forward_time": 1.747987985610962,
      "step": 40541
    },
    {
      "epoch": 3.866291046142578e-05,
      "step": 40541,
      "training_step_time": 6.747394561767578
    },
    {
      "epoch": 3.866386413574219e-05,
      "model_forward_time": 1.7538611888885498,
      "step": 40542
    },
    {
      "epoch": 3.866386413574219e-05,
      "step": 40542,
      "training_step_time": 7.52254581451416
    },
    {
      "epoch": 3.866481781005859e-05,
      "model_forward_time": 2.9103968143463135,
      "step": 40543
    },
    {
      "epoch": 3.866481781005859e-05,
      "step": 40543,
      "training_step_time": 9.301012992858887
    },
    {
      "epoch": 3.8665771484375e-05,
      "model_forward_time": 1.743851661682129,
      "step": 40544
    },
    {
      "epoch": 3.8665771484375e-05,
      "step": 40544,
      "training_step_time": 8.564620971679688
    },
    {
      "epoch": 3.866672515869141e-05,
      "model_forward_time": 1.752117395401001,
      "step": 40545
    },
    {
      "epoch": 3.866672515869141e-05,
      "step": 40545,
      "training_step_time": 7.62605881690979
    },
    {
      "epoch": 3.866767883300781e-05,
      "model_forward_time": 1.7562000751495361,
      "step": 40546
    },
    {
      "epoch": 3.866767883300781e-05,
      "step": 40546,
      "training_step_time": 8.44840955734253
    },
    {
      "epoch": 3.8668632507324216e-05,
      "model_forward_time": 1.7479188442230225,
      "step": 40547
    },
    {
      "epoch": 3.8668632507324216e-05,
      "step": 40547,
      "training_step_time": 7.483383655548096
    },
    {
      "epoch": 3.866958618164063e-05,
      "model_forward_time": 1.7478532791137695,
      "step": 40548
    },
    {
      "epoch": 3.866958618164063e-05,
      "step": 40548,
      "training_step_time": 6.7034385204315186
    },
    {
      "epoch": 3.867053985595703e-05,
      "model_forward_time": 1.7615900039672852,
      "step": 40549
    },
    {
      "epoch": 3.867053985595703e-05,
      "step": 40549,
      "training_step_time": 6.728811025619507
    },
    {
      "epoch": 3.8671493530273435e-05,
      "grad_norm": 0.16767731308937073,
      "learning_rate": 6.925020919120331e-06,
      "loss": 0.0611,
      "step": 40550
    },
    {
      "epoch": 3.8671493530273435e-05,
      "model_forward_time": 1.7505991458892822,
      "step": 40550
    },
    {
      "dynamics_loss_avg": 0.054717206954956056,
      "epoch": 3.8671493530273435e-05,
      "step": 40550
    },
    {
      "action_loss_avg": 0.0018188559683039785,
      "epoch": 3.8671493530273435e-05,
      "step": 40550
    },
    {
      "epoch": 3.8671493530273435e-05,
      "step": 40550,
      "training_step_time": 6.74661111831665
    },
    {
      "epoch": 3.8672447204589846e-05,
      "model_forward_time": 1.7464008331298828,
      "step": 40551
    },
    {
      "epoch": 3.8672447204589846e-05,
      "step": 40551,
      "training_step_time": 8.291662216186523
    },
    {
      "epoch": 3.867340087890625e-05,
      "model_forward_time": 2.8286993503570557,
      "step": 40552
    },
    {
      "epoch": 3.867340087890625e-05,
      "step": 40552,
      "training_step_time": 8.94638705253601
    },
    {
      "epoch": 3.8674354553222654e-05,
      "model_forward_time": 1.7474150657653809,
      "step": 40553
    },
    {
      "epoch": 3.8674354553222654e-05,
      "step": 40553,
      "training_step_time": 7.772948980331421
    },
    {
      "epoch": 3.8675308227539064e-05,
      "model_forward_time": 1.7763078212738037,
      "step": 40554
    },
    {
      "epoch": 3.8675308227539064e-05,
      "step": 40554,
      "training_step_time": 9.072474241256714
    },
    {
      "epoch": 3.867626190185547e-05,
      "model_forward_time": 1.7278921604156494,
      "step": 40555
    },
    {
      "epoch": 3.867626190185547e-05,
      "step": 40555,
      "training_step_time": 7.527191638946533
    },
    {
      "epoch": 3.867721557617187e-05,
      "model_forward_time": 1.7383167743682861,
      "step": 40556
    },
    {
      "epoch": 3.867721557617187e-05,
      "step": 40556,
      "training_step_time": 7.512251853942871
    },
    {
      "epoch": 3.867816925048828e-05,
      "model_forward_time": 1.7339012622833252,
      "step": 40557
    },
    {
      "epoch": 3.867816925048828e-05,
      "step": 40557,
      "training_step_time": 6.692455053329468
    },
    {
      "epoch": 3.867912292480469e-05,
      "model_forward_time": 1.7411315441131592,
      "step": 40558
    },
    {
      "epoch": 3.867912292480469e-05,
      "step": 40558,
      "training_step_time": 6.7022058963775635
    },
    {
      "epoch": 3.868007659912109e-05,
      "model_forward_time": 1.7435328960418701,
      "step": 40559
    },
    {
      "epoch": 3.868007659912109e-05,
      "step": 40559,
      "training_step_time": 6.6654016971588135
    },
    {
      "epoch": 3.86810302734375e-05,
      "grad_norm": 0.10031425952911377,
      "learning_rate": 6.923494802078229e-06,
      "loss": 0.058,
      "step": 40560
    },
    {
      "epoch": 3.86810302734375e-05,
      "model_forward_time": 1.7495412826538086,
      "step": 40560
    },
    {
      "dynamics_loss_avg": 0.046319519821554425,
      "epoch": 3.86810302734375e-05,
      "step": 40560
    },
    {
      "action_loss_avg": 0.00170752348494716,
      "epoch": 3.86810302734375e-05,
      "step": 40560
    },
    {
      "epoch": 3.86810302734375e-05,
      "step": 40560,
      "training_step_time": 7.2600767612457275
    },
    {
      "epoch": 3.8681983947753906e-05,
      "model_forward_time": 1.7351317405700684,
      "step": 40561
    },
    {
      "epoch": 3.8681983947753906e-05,
      "step": 40561,
      "training_step_time": 8.545848608016968
    },
    {
      "epoch": 3.868293762207031e-05,
      "model_forward_time": 2.8490726947784424,
      "step": 40562
    },
    {
      "epoch": 3.868293762207031e-05,
      "step": 40562,
      "training_step_time": 8.703048467636108
    },
    {
      "epoch": 3.868389129638672e-05,
      "model_forward_time": 1.744123935699463,
      "step": 40563
    },
    {
      "epoch": 3.868389129638672e-05,
      "step": 40563,
      "training_step_time": 9.717245101928711
    },
    {
      "epoch": 3.8684844970703125e-05,
      "model_forward_time": 1.7653563022613525,
      "step": 40564
    },
    {
      "epoch": 3.8684844970703125e-05,
      "step": 40564,
      "training_step_time": 7.5584657192230225
    },
    {
      "epoch": 3.868579864501953e-05,
      "model_forward_time": 1.7470619678497314,
      "step": 40565
    },
    {
      "epoch": 3.868579864501953e-05,
      "step": 40565,
      "training_step_time": 7.534879684448242
    },
    {
      "epoch": 3.868675231933594e-05,
      "model_forward_time": 1.7511978149414062,
      "step": 40566
    },
    {
      "epoch": 3.868675231933594e-05,
      "step": 40566,
      "training_step_time": 6.6996331214904785
    },
    {
      "epoch": 3.8687705993652344e-05,
      "model_forward_time": 1.3978321552276611,
      "step": 40567
    },
    {
      "epoch": 3.8687705993652344e-05,
      "step": 40567,
      "training_step_time": 6.751648187637329
    },
    {
      "epoch": 3.868865966796875e-05,
      "model_forward_time": 1.7459402084350586,
      "step": 40568
    },
    {
      "epoch": 3.868865966796875e-05,
      "step": 40568,
      "training_step_time": 6.710745334625244
    },
    {
      "epoch": 3.868961334228516e-05,
      "model_forward_time": 1.7500720024108887,
      "step": 40569
    },
    {
      "epoch": 3.868961334228516e-05,
      "step": 40569,
      "training_step_time": 7.2707624435424805
    },
    {
      "epoch": 3.869056701660156e-05,
      "grad_norm": 0.13716591894626617,
      "learning_rate": 6.921968474685629e-06,
      "loss": 0.0624,
      "step": 40570
    },
    {
      "epoch": 3.869056701660156e-05,
      "model_forward_time": 1.7492167949676514,
      "step": 40570
    },
    {
      "dynamics_loss_avg": 0.07078353054821492,
      "epoch": 3.869056701660156e-05,
      "step": 40570
    },
    {
      "action_loss_avg": 0.0024136039661243557,
      "epoch": 3.869056701660156e-05,
      "step": 40570
    },
    {
      "epoch": 3.869056701660156e-05,
      "step": 40570,
      "training_step_time": 7.580637216567993
    },
    {
      "epoch": 3.869152069091797e-05,
      "model_forward_time": 2.8847029209136963,
      "step": 40571
    },
    {
      "epoch": 3.869152069091797e-05,
      "step": 40571,
      "training_step_time": 8.307196855545044
    },
    {
      "epoch": 3.869247436523438e-05,
      "model_forward_time": 1.7324998378753662,
      "step": 40572
    },
    {
      "epoch": 3.869247436523438e-05,
      "step": 40572,
      "training_step_time": 8.828072309494019
    },
    {
      "epoch": 3.869342803955078e-05,
      "model_forward_time": 1.7427220344543457,
      "step": 40573
    },
    {
      "epoch": 3.869342803955078e-05,
      "step": 40573,
      "training_step_time": 7.573359489440918
    },
    {
      "epoch": 3.8694381713867186e-05,
      "model_forward_time": 1.7398452758789062,
      "step": 40574
    },
    {
      "epoch": 3.8694381713867186e-05,
      "step": 40574,
      "training_step_time": 8.383312702178955
    },
    {
      "epoch": 3.8695335388183596e-05,
      "model_forward_time": 1.7386229038238525,
      "step": 40575
    },
    {
      "epoch": 3.8695335388183596e-05,
      "step": 40575,
      "training_step_time": 7.522413492202759
    },
    {
      "epoch": 3.86962890625e-05,
      "model_forward_time": 1.7358150482177734,
      "step": 40576
    },
    {
      "epoch": 3.86962890625e-05,
      "step": 40576,
      "training_step_time": 6.751103162765503
    },
    {
      "epoch": 3.8697242736816405e-05,
      "model_forward_time": 1.7397446632385254,
      "step": 40577
    },
    {
      "epoch": 3.8697242736816405e-05,
      "step": 40577,
      "training_step_time": 6.681154489517212
    },
    {
      "epoch": 3.8698196411132815e-05,
      "model_forward_time": 1.7386012077331543,
      "step": 40578
    },
    {
      "epoch": 3.8698196411132815e-05,
      "step": 40578,
      "training_step_time": 7.2499120235443115
    },
    {
      "epoch": 3.869915008544922e-05,
      "model_forward_time": 1.7300994396209717,
      "step": 40579
    },
    {
      "epoch": 3.869915008544922e-05,
      "step": 40579,
      "training_step_time": 7.50960898399353
    },
    {
      "epoch": 3.870010375976562e-05,
      "grad_norm": 0.09261083602905273,
      "learning_rate": 6.920441937109446e-06,
      "loss": 0.0602,
      "step": 40580
    },
    {
      "epoch": 3.870010375976562e-05,
      "model_forward_time": 2.8821072578430176,
      "step": 40580
    },
    {
      "dynamics_loss_avg": 0.05768288541585207,
      "epoch": 3.870010375976562e-05,
      "step": 40580
    },
    {
      "action_loss_avg": 0.0021348423557356,
      "epoch": 3.870010375976562e-05,
      "step": 40580
    },
    {
      "epoch": 3.870010375976562e-05,
      "step": 40580,
      "training_step_time": 8.335736274719238
    },
    {
      "epoch": 3.8701057434082034e-05,
      "model_forward_time": 1.7721104621887207,
      "step": 40581
    },
    {
      "epoch": 3.8701057434082034e-05,
      "step": 40581,
      "training_step_time": 7.734761953353882
    },
    {
      "epoch": 3.870201110839844e-05,
      "model_forward_time": 1.750983476638794,
      "step": 40582
    },
    {
      "epoch": 3.870201110839844e-05,
      "step": 40582,
      "training_step_time": 7.557002544403076
    },
    {
      "epoch": 3.870296478271484e-05,
      "model_forward_time": 1.7660908699035645,
      "step": 40583
    },
    {
      "epoch": 3.870296478271484e-05,
      "step": 40583,
      "training_step_time": 7.431544542312622
    },
    {
      "epoch": 3.870391845703125e-05,
      "model_forward_time": 1.7459120750427246,
      "step": 40584
    },
    {
      "epoch": 3.870391845703125e-05,
      "step": 40584,
      "training_step_time": 7.4511494636535645
    },
    {
      "epoch": 3.870487213134766e-05,
      "model_forward_time": 1.7493836879730225,
      "step": 40585
    },
    {
      "epoch": 3.870487213134766e-05,
      "step": 40585,
      "training_step_time": 6.706789493560791
    },
    {
      "epoch": 3.870582580566406e-05,
      "model_forward_time": 1.7490513324737549,
      "step": 40586
    },
    {
      "epoch": 3.870582580566406e-05,
      "step": 40586,
      "training_step_time": 6.7095232009887695
    },
    {
      "epoch": 3.870677947998047e-05,
      "model_forward_time": 1.749251127243042,
      "step": 40587
    },
    {
      "epoch": 3.870677947998047e-05,
      "step": 40587,
      "training_step_time": 7.224301815032959
    },
    {
      "epoch": 3.8707733154296876e-05,
      "model_forward_time": 1.7605202198028564,
      "step": 40588
    },
    {
      "epoch": 3.8707733154296876e-05,
      "step": 40588,
      "training_step_time": 7.503739356994629
    },
    {
      "epoch": 3.870868682861328e-05,
      "model_forward_time": 2.945880651473999,
      "step": 40589
    },
    {
      "epoch": 3.870868682861328e-05,
      "step": 40589,
      "training_step_time": 8.62648320198059
    },
    {
      "epoch": 3.870964050292969e-05,
      "grad_norm": 0.10123394429683685,
      "learning_rate": 6.91891518951662e-06,
      "loss": 0.0556,
      "step": 40590
    },
    {
      "epoch": 3.870964050292969e-05,
      "model_forward_time": 1.750581979751587,
      "step": 40590
    },
    {
      "dynamics_loss_avg": 0.047439427487552165,
      "epoch": 3.870964050292969e-05,
      "step": 40590
    },
    {
      "action_loss_avg": 0.0017123342637205496,
      "epoch": 3.870964050292969e-05,
      "step": 40590
    },
    {
      "epoch": 3.870964050292969e-05,
      "step": 40590,
      "training_step_time": 8.438320398330688
    },
    {
      "epoch": 3.8710594177246095e-05,
      "model_forward_time": 1.7484235763549805,
      "step": 40591
    },
    {
      "epoch": 3.8710594177246095e-05,
      "step": 40591,
      "training_step_time": 8.436394929885864
    },
    {
      "epoch": 3.87115478515625e-05,
      "model_forward_time": 1.7483727931976318,
      "step": 40592
    },
    {
      "epoch": 3.87115478515625e-05,
      "step": 40592,
      "training_step_time": 8.616170167922974
    },
    {
      "epoch": 3.871250152587891e-05,
      "model_forward_time": 1.7524256706237793,
      "step": 40593
    },
    {
      "epoch": 3.871250152587891e-05,
      "step": 40593,
      "training_step_time": 7.55891752243042
    },
    {
      "epoch": 3.8713455200195314e-05,
      "model_forward_time": 1.7643182277679443,
      "step": 40594
    },
    {
      "epoch": 3.8713455200195314e-05,
      "step": 40594,
      "training_step_time": 7.718376636505127
    },
    {
      "epoch": 3.871440887451172e-05,
      "model_forward_time": 1.7501590251922607,
      "step": 40595
    },
    {
      "epoch": 3.871440887451172e-05,
      "step": 40595,
      "training_step_time": 6.9573729038238525
    },
    {
      "epoch": 3.871536254882812e-05,
      "model_forward_time": 1.74580717086792,
      "step": 40596
    },
    {
      "epoch": 3.871536254882812e-05,
      "step": 40596,
      "training_step_time": 7.4611005783081055
    },
    {
      "epoch": 3.871631622314453e-05,
      "model_forward_time": 1.7473583221435547,
      "step": 40597
    },
    {
      "epoch": 3.871631622314453e-05,
      "step": 40597,
      "training_step_time": 7.739755392074585
    },
    {
      "epoch": 3.8717269897460937e-05,
      "model_forward_time": 2.8858776092529297,
      "step": 40598
    },
    {
      "epoch": 3.8717269897460937e-05,
      "step": 40598,
      "training_step_time": 7.932988882064819
    },
    {
      "epoch": 3.871822357177734e-05,
      "model_forward_time": 1.8078627586364746,
      "step": 40599
    },
    {
      "epoch": 3.871822357177734e-05,
      "step": 40599,
      "training_step_time": 10.607010841369629
    },
    {
      "epoch": 3.871917724609375e-05,
      "grad_norm": 0.11958632618188858,
      "learning_rate": 6.917388232074113e-06,
      "loss": 0.0596,
      "step": 40600
    },
    {
      "epoch": 3.871917724609375e-05,
      "model_forward_time": 1.7525455951690674,
      "step": 40600
    },
    {
      "dynamics_loss_avg": 0.08147646393626928,
      "epoch": 3.871917724609375e-05,
      "step": 40600
    },
    {
      "action_loss_avg": 0.0021892830496653914,
      "epoch": 3.871917724609375e-05,
      "step": 40600
    },
    {
      "epoch": 3.871917724609375e-05,
      "step": 40600,
      "training_step_time": 9.497902631759644
    },
    {
      "epoch": 3.8720130920410155e-05,
      "model_forward_time": 1.4029500484466553,
      "step": 40601
    },
    {
      "epoch": 3.8720130920410155e-05,
      "step": 40601,
      "training_step_time": 8.518702983856201
    },
    {
      "epoch": 3.872108459472656e-05,
      "model_forward_time": 1.7437307834625244,
      "step": 40602
    },
    {
      "epoch": 3.872108459472656e-05,
      "step": 40602,
      "training_step_time": 7.707389831542969
    },
    {
      "epoch": 3.872203826904297e-05,
      "model_forward_time": 1.745309829711914,
      "step": 40603
    },
    {
      "epoch": 3.872203826904297e-05,
      "step": 40603,
      "training_step_time": 7.610913515090942
    },
    {
      "epoch": 3.8722991943359374e-05,
      "model_forward_time": 1.7472357749938965,
      "step": 40604
    },
    {
      "epoch": 3.8722991943359374e-05,
      "step": 40604,
      "training_step_time": 6.798130989074707
    },
    {
      "epoch": 3.872394561767578e-05,
      "model_forward_time": 1.7613129615783691,
      "step": 40605
    },
    {
      "epoch": 3.872394561767578e-05,
      "step": 40605,
      "training_step_time": 7.340758562088013
    },
    {
      "epoch": 3.872489929199219e-05,
      "model_forward_time": 1.7512261867523193,
      "step": 40606
    },
    {
      "epoch": 3.872489929199219e-05,
      "step": 40606,
      "training_step_time": 7.603294610977173
    },
    {
      "epoch": 3.872585296630859e-05,
      "model_forward_time": 1.761857509613037,
      "step": 40607
    },
    {
      "epoch": 3.872585296630859e-05,
      "step": 40607,
      "training_step_time": 8.714851140975952
    },
    {
      "epoch": 3.8726806640625e-05,
      "model_forward_time": 2.864697217941284,
      "step": 40608
    },
    {
      "epoch": 3.8726806640625e-05,
      "step": 40608,
      "training_step_time": 11.600250482559204
    },
    {
      "epoch": 3.872776031494141e-05,
      "model_forward_time": 1.7582430839538574,
      "step": 40609
    },
    {
      "epoch": 3.872776031494141e-05,
      "step": 40609,
      "training_step_time": 8.904035568237305
    },
    {
      "epoch": 3.872871398925781e-05,
      "grad_norm": 0.16736887395381927,
      "learning_rate": 6.915861064948914e-06,
      "loss": 0.0617,
      "step": 40610
    },
    {
      "epoch": 3.872871398925781e-05,
      "model_forward_time": 1.7455313205718994,
      "step": 40610
    },
    {
      "dynamics_loss_avg": 0.05243449714034796,
      "epoch": 3.872871398925781e-05,
      "step": 40610
    },
    {
      "action_loss_avg": 0.0020394357678014787,
      "epoch": 3.872871398925781e-05,
      "step": 40610
    },
    {
      "epoch": 3.872871398925781e-05,
      "step": 40610,
      "training_step_time": 7.629475116729736
    },
    {
      "epoch": 3.8729667663574216e-05,
      "model_forward_time": 1.746380090713501,
      "step": 40611
    },
    {
      "epoch": 3.8729667663574216e-05,
      "step": 40611,
      "training_step_time": 7.764434576034546
    },
    {
      "epoch": 3.873062133789063e-05,
      "model_forward_time": 1.7542989253997803,
      "step": 40612
    },
    {
      "epoch": 3.873062133789063e-05,
      "step": 40612,
      "training_step_time": 7.574179172515869
    },
    {
      "epoch": 3.873157501220703e-05,
      "model_forward_time": 1.74810791015625,
      "step": 40613
    },
    {
      "epoch": 3.873157501220703e-05,
      "step": 40613,
      "training_step_time": 6.850215673446655
    },
    {
      "epoch": 3.8732528686523435e-05,
      "model_forward_time": 1.7353174686431885,
      "step": 40614
    },
    {
      "epoch": 3.8732528686523435e-05,
      "step": 40614,
      "training_step_time": 7.36971640586853
    },
    {
      "epoch": 3.8733482360839846e-05,
      "model_forward_time": 1.7557148933410645,
      "step": 40615
    },
    {
      "epoch": 3.8733482360839846e-05,
      "step": 40615,
      "training_step_time": 7.664153814315796
    },
    {
      "epoch": 3.873443603515625e-05,
      "model_forward_time": 1.744605302810669,
      "step": 40616
    },
    {
      "epoch": 3.873443603515625e-05,
      "step": 40616,
      "training_step_time": 7.764435052871704
    },
    {
      "epoch": 3.8735389709472654e-05,
      "model_forward_time": 2.8761422634124756,
      "step": 40617
    },
    {
      "epoch": 3.8735389709472654e-05,
      "step": 40617,
      "training_step_time": 9.86315107345581
    },
    {
      "epoch": 3.8736343383789065e-05,
      "model_forward_time": 1.749434232711792,
      "step": 40618
    },
    {
      "epoch": 3.8736343383789065e-05,
      "step": 40618,
      "training_step_time": 8.147517681121826
    },
    {
      "epoch": 3.873729705810547e-05,
      "model_forward_time": 1.7714598178863525,
      "step": 40619
    },
    {
      "epoch": 3.873729705810547e-05,
      "step": 40619,
      "training_step_time": 9.155766725540161
    },
    {
      "epoch": 3.873825073242187e-05,
      "grad_norm": 0.16357751190662384,
      "learning_rate": 6.9143336883080255e-06,
      "loss": 0.06,
      "step": 40620
    },
    {
      "epoch": 3.873825073242187e-05,
      "model_forward_time": 1.7487156391143799,
      "step": 40620
    },
    {
      "dynamics_loss_avg": 0.0626531658694148,
      "epoch": 3.873825073242187e-05,
      "step": 40620
    },
    {
      "action_loss_avg": 0.002482870750827715,
      "epoch": 3.873825073242187e-05,
      "step": 40620
    },
    {
      "epoch": 3.873825073242187e-05,
      "step": 40620,
      "training_step_time": 7.785843849182129
    },
    {
      "epoch": 3.873920440673828e-05,
      "model_forward_time": 1.3856844902038574,
      "step": 40621
    },
    {
      "epoch": 3.873920440673828e-05,
      "step": 40621,
      "training_step_time": 6.861823558807373
    },
    {
      "epoch": 3.874015808105469e-05,
      "model_forward_time": 1.7499849796295166,
      "step": 40622
    },
    {
      "epoch": 3.874015808105469e-05,
      "step": 40622,
      "training_step_time": 7.70288610458374
    },
    {
      "epoch": 3.874111175537109e-05,
      "model_forward_time": 1.741919994354248,
      "step": 40623
    },
    {
      "epoch": 3.874111175537109e-05,
      "step": 40623,
      "training_step_time": 7.153699636459351
    },
    {
      "epoch": 3.87420654296875e-05,
      "model_forward_time": 1.75022554397583,
      "step": 40624
    },
    {
      "epoch": 3.87420654296875e-05,
      "step": 40624,
      "training_step_time": 7.647457838058472
    },
    {
      "epoch": 3.8743019104003906e-05,
      "model_forward_time": 1.74774169921875,
      "step": 40625
    },
    {
      "epoch": 3.8743019104003906e-05,
      "step": 40625,
      "training_step_time": 7.76193642616272
    },
    {
      "epoch": 3.874397277832031e-05,
      "model_forward_time": 2.8894991874694824,
      "step": 40626
    },
    {
      "epoch": 3.874397277832031e-05,
      "step": 40626,
      "training_step_time": 10.035353183746338
    },
    {
      "epoch": 3.874492645263672e-05,
      "model_forward_time": 1.7458086013793945,
      "step": 40627
    },
    {
      "epoch": 3.874492645263672e-05,
      "step": 40627,
      "training_step_time": 9.468823432922363
    },
    {
      "epoch": 3.8745880126953125e-05,
      "model_forward_time": 1.7420599460601807,
      "step": 40628
    },
    {
      "epoch": 3.8745880126953125e-05,
      "step": 40628,
      "training_step_time": 9.69004225730896
    },
    {
      "epoch": 3.874683380126953e-05,
      "model_forward_time": 1.7405824661254883,
      "step": 40629
    },
    {
      "epoch": 3.874683380126953e-05,
      "step": 40629,
      "training_step_time": 7.756422758102417
    },
    {
      "epoch": 3.874778747558594e-05,
      "grad_norm": 0.1683250069618225,
      "learning_rate": 6.9128061023184865e-06,
      "loss": 0.062,
      "step": 40630
    },
    {
      "epoch": 3.874778747558594e-05,
      "model_forward_time": 1.7544002532958984,
      "step": 40630
    },
    {
      "dynamics_loss_avg": 0.05034680534154177,
      "epoch": 3.874778747558594e-05,
      "step": 40630
    },
    {
      "action_loss_avg": 0.002357449586270377,
      "epoch": 3.874778747558594e-05,
      "step": 40630
    },
    {
      "epoch": 3.874778747558594e-05,
      "step": 40630,
      "training_step_time": 7.747546195983887
    },
    {
      "epoch": 3.8748741149902344e-05,
      "model_forward_time": 1.746495246887207,
      "step": 40631
    },
    {
      "epoch": 3.8748741149902344e-05,
      "step": 40631,
      "training_step_time": 7.361998796463013
    },
    {
      "epoch": 3.874969482421875e-05,
      "model_forward_time": 1.747556209564209,
      "step": 40632
    },
    {
      "epoch": 3.874969482421875e-05,
      "step": 40632,
      "training_step_time": 7.21227240562439
    },
    {
      "epoch": 3.875064849853516e-05,
      "model_forward_time": 1.7527825832366943,
      "step": 40633
    },
    {
      "epoch": 3.875064849853516e-05,
      "step": 40633,
      "training_step_time": 7.761277198791504
    },
    {
      "epoch": 3.875160217285156e-05,
      "model_forward_time": 1.7456891536712646,
      "step": 40634
    },
    {
      "epoch": 3.875160217285156e-05,
      "step": 40634,
      "training_step_time": 7.773356676101685
    },
    {
      "epoch": 3.875255584716797e-05,
      "model_forward_time": 2.838596820831299,
      "step": 40635
    },
    {
      "epoch": 3.875255584716797e-05,
      "step": 40635,
      "training_step_time": 8.670462608337402
    },
    {
      "epoch": 3.875350952148438e-05,
      "model_forward_time": 1.8039450645446777,
      "step": 40636
    },
    {
      "epoch": 3.875350952148438e-05,
      "step": 40636,
      "training_step_time": 7.832009792327881
    },
    {
      "epoch": 3.875446319580078e-05,
      "model_forward_time": 1.7508361339569092,
      "step": 40637
    },
    {
      "epoch": 3.875446319580078e-05,
      "step": 40637,
      "training_step_time": 9.459812879562378
    },
    {
      "epoch": 3.8755416870117186e-05,
      "model_forward_time": 1.7438395023345947,
      "step": 40638
    },
    {
      "epoch": 3.8755416870117186e-05,
      "step": 40638,
      "training_step_time": 7.720863342285156
    },
    {
      "epoch": 3.8756370544433597e-05,
      "model_forward_time": 1.757305383682251,
      "step": 40639
    },
    {
      "epoch": 3.8756370544433597e-05,
      "step": 40639,
      "training_step_time": 7.647260665893555
    },
    {
      "epoch": 3.875732421875e-05,
      "grad_norm": 0.15517348051071167,
      "learning_rate": 6.911278307147347e-06,
      "loss": 0.0569,
      "step": 40640
    },
    {
      "epoch": 3.875732421875e-05,
      "model_forward_time": 1.7453396320343018,
      "step": 40640
    },
    {
      "dynamics_loss_avg": 0.0661678371950984,
      "epoch": 3.875732421875e-05,
      "step": 40640
    },
    {
      "action_loss_avg": 0.004051192733459175,
      "epoch": 3.875732421875e-05,
      "step": 40640
    },
    {
      "epoch": 3.875732421875e-05,
      "step": 40640,
      "training_step_time": 7.74448299407959
    },
    {
      "epoch": 3.8758277893066405e-05,
      "model_forward_time": 1.7492895126342773,
      "step": 40641
    },
    {
      "epoch": 3.8758277893066405e-05,
      "step": 40641,
      "training_step_time": 7.182989597320557
    },
    {
      "epoch": 3.8759231567382815e-05,
      "model_forward_time": 1.7611238956451416,
      "step": 40642
    },
    {
      "epoch": 3.8759231567382815e-05,
      "step": 40642,
      "training_step_time": 7.725255250930786
    },
    {
      "epoch": 3.876018524169922e-05,
      "model_forward_time": 1.746103286743164,
      "step": 40643
    },
    {
      "epoch": 3.876018524169922e-05,
      "step": 40643,
      "training_step_time": 8.346077919006348
    },
    {
      "epoch": 3.8761138916015623e-05,
      "model_forward_time": 1.7419257164001465,
      "step": 40644
    },
    {
      "epoch": 3.8761138916015623e-05,
      "step": 40644,
      "training_step_time": 8.410536050796509
    },
    {
      "epoch": 3.8762092590332034e-05,
      "model_forward_time": 1.8166487216949463,
      "step": 40645
    },
    {
      "epoch": 3.8762092590332034e-05,
      "step": 40645,
      "training_step_time": 10.347158670425415
    },
    {
      "epoch": 3.876304626464844e-05,
      "model_forward_time": 1.735550880432129,
      "step": 40646
    },
    {
      "epoch": 3.876304626464844e-05,
      "step": 40646,
      "training_step_time": 7.910478591918945
    },
    {
      "epoch": 3.876399993896484e-05,
      "model_forward_time": 1.751485824584961,
      "step": 40647
    },
    {
      "epoch": 3.876399993896484e-05,
      "step": 40647,
      "training_step_time": 7.763721942901611
    },
    {
      "epoch": 3.876495361328125e-05,
      "model_forward_time": 1.775719404220581,
      "step": 40648
    },
    {
      "epoch": 3.876495361328125e-05,
      "step": 40648,
      "training_step_time": 8.629741191864014
    },
    {
      "epoch": 3.876590728759766e-05,
      "model_forward_time": 1.742396593093872,
      "step": 40649
    },
    {
      "epoch": 3.876590728759766e-05,
      "step": 40649,
      "training_step_time": 7.844547510147095
    },
    {
      "epoch": 3.876686096191406e-05,
      "grad_norm": 0.10467741638422012,
      "learning_rate": 6.909750302961684e-06,
      "loss": 0.0588,
      "step": 40650
    },
    {
      "epoch": 3.876686096191406e-05,
      "model_forward_time": 1.751319408416748,
      "step": 40650
    },
    {
      "dynamics_loss_avg": 0.04703802634030581,
      "epoch": 3.876686096191406e-05,
      "step": 40650
    },
    {
      "action_loss_avg": 0.00230439932201989,
      "epoch": 3.876686096191406e-05,
      "step": 40650
    },
    {
      "epoch": 3.876686096191406e-05,
      "step": 40650,
      "training_step_time": 7.18916654586792
    },
    {
      "epoch": 3.876781463623047e-05,
      "model_forward_time": 1.750605583190918,
      "step": 40651
    },
    {
      "epoch": 3.876781463623047e-05,
      "step": 40651,
      "training_step_time": 7.551549434661865
    },
    {
      "epoch": 3.8768768310546876e-05,
      "model_forward_time": 1.7408833503723145,
      "step": 40652
    },
    {
      "epoch": 3.8768768310546876e-05,
      "step": 40652,
      "training_step_time": 7.647874593734741
    },
    {
      "epoch": 3.876972198486328e-05,
      "model_forward_time": 1.7432630062103271,
      "step": 40653
    },
    {
      "epoch": 3.876972198486328e-05,
      "step": 40653,
      "training_step_time": 8.56886076927185
    },
    {
      "epoch": 3.877067565917969e-05,
      "model_forward_time": 2.9501144886016846,
      "step": 40654
    },
    {
      "epoch": 3.877067565917969e-05,
      "step": 40654,
      "training_step_time": 9.764394283294678
    },
    {
      "epoch": 3.8771629333496095e-05,
      "model_forward_time": 1.7394144535064697,
      "step": 40655
    },
    {
      "epoch": 3.8771629333496095e-05,
      "step": 40655,
      "training_step_time": 7.9753737449646
    },
    {
      "epoch": 3.87725830078125e-05,
      "model_forward_time": 1.7413084506988525,
      "step": 40656
    },
    {
      "epoch": 3.87725830078125e-05,
      "step": 40656,
      "training_step_time": 8.233136653900146
    },
    {
      "epoch": 3.87735366821289e-05,
      "model_forward_time": 1.7406914234161377,
      "step": 40657
    },
    {
      "epoch": 3.87735366821289e-05,
      "step": 40657,
      "training_step_time": 7.728887557983398
    },
    {
      "epoch": 3.8774490356445314e-05,
      "model_forward_time": 1.739058494567871,
      "step": 40658
    },
    {
      "epoch": 3.8774490356445314e-05,
      "step": 40658,
      "training_step_time": 7.2756125926971436
    },
    {
      "epoch": 3.877544403076172e-05,
      "model_forward_time": 1.7448976039886475,
      "step": 40659
    },
    {
      "epoch": 3.877544403076172e-05,
      "step": 40659,
      "training_step_time": 7.591523885726929
    },
    {
      "epoch": 3.877639770507812e-05,
      "grad_norm": 0.09727250039577484,
      "learning_rate": 6.9082220899286e-06,
      "loss": 0.0579,
      "step": 40660
    },
    {
      "epoch": 3.877639770507812e-05,
      "model_forward_time": 1.744225263595581,
      "step": 40660
    },
    {
      "dynamics_loss_avg": 0.05858972705900669,
      "epoch": 3.877639770507812e-05,
      "step": 40660
    },
    {
      "action_loss_avg": 0.0024518780235666783,
      "epoch": 3.877639770507812e-05,
      "step": 40660
    },
    {
      "epoch": 3.877639770507812e-05,
      "step": 40660,
      "training_step_time": 7.592262506484985
    },
    {
      "epoch": 3.877735137939453e-05,
      "model_forward_time": 1.75123929977417,
      "step": 40661
    },
    {
      "epoch": 3.877735137939453e-05,
      "step": 40661,
      "training_step_time": 7.706036806106567
    },
    {
      "epoch": 3.8778305053710937e-05,
      "model_forward_time": 1.7466819286346436,
      "step": 40662
    },
    {
      "epoch": 3.8778305053710937e-05,
      "step": 40662,
      "training_step_time": 7.660022974014282
    },
    {
      "epoch": 3.877925872802734e-05,
      "model_forward_time": 2.915602922439575,
      "step": 40663
    },
    {
      "epoch": 3.877925872802734e-05,
      "step": 40663,
      "training_step_time": 8.695748805999756
    },
    {
      "epoch": 3.878021240234375e-05,
      "model_forward_time": 1.7442042827606201,
      "step": 40664
    },
    {
      "epoch": 3.878021240234375e-05,
      "step": 40664,
      "training_step_time": 9.38458776473999
    },
    {
      "epoch": 3.8781166076660155e-05,
      "model_forward_time": 1.7540085315704346,
      "step": 40665
    },
    {
      "epoch": 3.8781166076660155e-05,
      "step": 40665,
      "training_step_time": 7.837784051895142
    },
    {
      "epoch": 3.878211975097656e-05,
      "model_forward_time": 1.7689814567565918,
      "step": 40666
    },
    {
      "epoch": 3.878211975097656e-05,
      "step": 40666,
      "training_step_time": 7.949148416519165
    },
    {
      "epoch": 3.878307342529297e-05,
      "model_forward_time": 1.7391278743743896,
      "step": 40667
    },
    {
      "epoch": 3.878307342529297e-05,
      "step": 40667,
      "training_step_time": 7.685433864593506
    },
    {
      "epoch": 3.8784027099609374e-05,
      "model_forward_time": 1.7485990524291992,
      "step": 40668
    },
    {
      "epoch": 3.8784027099609374e-05,
      "step": 40668,
      "training_step_time": 7.751210689544678
    },
    {
      "epoch": 3.878498077392578e-05,
      "model_forward_time": 1.742771863937378,
      "step": 40669
    },
    {
      "epoch": 3.878498077392578e-05,
      "step": 40669,
      "training_step_time": 7.783427953720093
    },
    {
      "epoch": 3.878593444824219e-05,
      "grad_norm": 0.13322368264198303,
      "learning_rate": 6.906693668215219e-06,
      "loss": 0.0595,
      "step": 40670
    },
    {
      "epoch": 3.878593444824219e-05,
      "model_forward_time": 1.74462890625,
      "step": 40670
    },
    {
      "dynamics_loss_avg": 0.06367907049134373,
      "epoch": 3.878593444824219e-05,
      "step": 40670
    },
    {
      "action_loss_avg": 0.002982837360468693,
      "epoch": 3.878593444824219e-05,
      "step": 40670
    },
    {
      "epoch": 3.878593444824219e-05,
      "step": 40670,
      "training_step_time": 7.768995761871338
    },
    {
      "epoch": 3.878688812255859e-05,
      "model_forward_time": 1.7455189228057861,
      "step": 40671
    },
    {
      "epoch": 3.878688812255859e-05,
      "step": 40671,
      "training_step_time": 7.715914249420166
    },
    {
      "epoch": 3.8787841796875e-05,
      "model_forward_time": 2.897446870803833,
      "step": 40672
    },
    {
      "epoch": 3.8787841796875e-05,
      "step": 40672,
      "training_step_time": 9.103869199752808
    },
    {
      "epoch": 3.878879547119141e-05,
      "model_forward_time": 1.7464849948883057,
      "step": 40673
    },
    {
      "epoch": 3.878879547119141e-05,
      "step": 40673,
      "training_step_time": 9.484530448913574
    },
    {
      "epoch": 3.878974914550781e-05,
      "model_forward_time": 1.7437357902526855,
      "step": 40674
    },
    {
      "epoch": 3.878974914550781e-05,
      "step": 40674,
      "training_step_time": 7.70863676071167
    },
    {
      "epoch": 3.8790702819824216e-05,
      "model_forward_time": 1.7378361225128174,
      "step": 40675
    },
    {
      "epoch": 3.8790702819824216e-05,
      "step": 40675,
      "training_step_time": 7.726742744445801
    },
    {
      "epoch": 3.879165649414063e-05,
      "model_forward_time": 1.7426695823669434,
      "step": 40676
    },
    {
      "epoch": 3.879165649414063e-05,
      "step": 40676,
      "training_step_time": 7.611145973205566
    },
    {
      "epoch": 3.879261016845703e-05,
      "model_forward_time": 1.753173589706421,
      "step": 40677
    },
    {
      "epoch": 3.879261016845703e-05,
      "step": 40677,
      "training_step_time": 7.673711538314819
    },
    {
      "epoch": 3.8793563842773435e-05,
      "model_forward_time": 1.761540174484253,
      "step": 40678
    },
    {
      "epoch": 3.8793563842773435e-05,
      "step": 40678,
      "training_step_time": 7.628211736679077
    },
    {
      "epoch": 3.8794517517089846e-05,
      "model_forward_time": 1.7604990005493164,
      "step": 40679
    },
    {
      "epoch": 3.8794517517089846e-05,
      "step": 40679,
      "training_step_time": 7.652513027191162
    },
    {
      "epoch": 3.879547119140625e-05,
      "grad_norm": 0.17826375365257263,
      "learning_rate": 6.905165037988684e-06,
      "loss": 0.06,
      "step": 40680
    },
    {
      "epoch": 3.879547119140625e-05,
      "model_forward_time": 1.7443137168884277,
      "step": 40680
    },
    {
      "dynamics_loss_avg": 0.057422390952706334,
      "epoch": 3.879547119140625e-05,
      "step": 40680
    },
    {
      "action_loss_avg": 0.0019123712729196996,
      "epoch": 3.879547119140625e-05,
      "step": 40680
    },
    {
      "epoch": 3.879547119140625e-05,
      "step": 40680,
      "training_step_time": 7.83483099937439
    },
    {
      "epoch": 3.8796424865722654e-05,
      "model_forward_time": 2.8594188690185547,
      "step": 40681
    },
    {
      "epoch": 3.8796424865722654e-05,
      "step": 40681,
      "training_step_time": 8.771774530410767
    },
    {
      "epoch": 3.8797378540039065e-05,
      "model_forward_time": 1.8136742115020752,
      "step": 40682
    },
    {
      "epoch": 3.8797378540039065e-05,
      "step": 40682,
      "training_step_time": 8.763733386993408
    },
    {
      "epoch": 3.879833221435547e-05,
      "model_forward_time": 1.7404708862304688,
      "step": 40683
    },
    {
      "epoch": 3.879833221435547e-05,
      "step": 40683,
      "training_step_time": 10.80491852760315
    },
    {
      "epoch": 3.879928588867187e-05,
      "model_forward_time": 1.7476325035095215,
      "step": 40684
    },
    {
      "epoch": 3.879928588867187e-05,
      "step": 40684,
      "training_step_time": 8.066034078598022
    },
    {
      "epoch": 3.8800239562988283e-05,
      "model_forward_time": 1.7504451274871826,
      "step": 40685
    },
    {
      "epoch": 3.8800239562988283e-05,
      "step": 40685,
      "training_step_time": 7.7406933307647705
    },
    {
      "epoch": 3.880119323730469e-05,
      "model_forward_time": 1.7514915466308594,
      "step": 40686
    },
    {
      "epoch": 3.880119323730469e-05,
      "step": 40686,
      "training_step_time": 7.507327318191528
    },
    {
      "epoch": 3.880214691162109e-05,
      "model_forward_time": 1.7398576736450195,
      "step": 40687
    },
    {
      "epoch": 3.880214691162109e-05,
      "step": 40687,
      "training_step_time": 7.620573043823242
    },
    {
      "epoch": 3.88031005859375e-05,
      "model_forward_time": 1.743941307067871,
      "step": 40688
    },
    {
      "epoch": 3.88031005859375e-05,
      "step": 40688,
      "training_step_time": 7.51151704788208
    },
    {
      "epoch": 3.8804054260253906e-05,
      "model_forward_time": 1.7416157722473145,
      "step": 40689
    },
    {
      "epoch": 3.8804054260253906e-05,
      "step": 40689,
      "training_step_time": 7.608270168304443
    },
    {
      "epoch": 3.880500793457031e-05,
      "grad_norm": 0.1661265641450882,
      "learning_rate": 6.903636199416164e-06,
      "loss": 0.0573,
      "step": 40690
    },
    {
      "epoch": 3.880500793457031e-05,
      "model_forward_time": 2.851889133453369,
      "step": 40690
    },
    {
      "dynamics_loss_avg": 0.05570848286151886,
      "epoch": 3.880500793457031e-05,
      "step": 40690
    },
    {
      "action_loss_avg": 0.0018917334964498878,
      "epoch": 3.880500793457031e-05,
      "step": 40690
    },
    {
      "epoch": 3.880500793457031e-05,
      "step": 40690,
      "training_step_time": 7.60944676399231
    },
    {
      "epoch": 3.880596160888672e-05,
      "model_forward_time": 1.1454603672027588,
      "step": 40691
    },
    {
      "epoch": 3.880596160888672e-05,
      "step": 40691,
      "training_step_time": 9.326968908309937
    },
    {
      "epoch": 3.8806915283203125e-05,
      "model_forward_time": 1.7440025806427002,
      "step": 40692
    },
    {
      "epoch": 3.8806915283203125e-05,
      "step": 40692,
      "training_step_time": 8.560085773468018
    },
    {
      "epoch": 3.880786895751953e-05,
      "model_forward_time": 1.748232126235962,
      "step": 40693
    },
    {
      "epoch": 3.880786895751953e-05,
      "step": 40693,
      "training_step_time": 7.553524494171143
    },
    {
      "epoch": 3.880882263183594e-05,
      "model_forward_time": 1.7515437602996826,
      "step": 40694
    },
    {
      "epoch": 3.880882263183594e-05,
      "step": 40694,
      "training_step_time": 7.67966103553772
    },
    {
      "epoch": 3.8809776306152344e-05,
      "model_forward_time": 1.7440969944000244,
      "step": 40695
    },
    {
      "epoch": 3.8809776306152344e-05,
      "step": 40695,
      "training_step_time": 7.346004247665405
    },
    {
      "epoch": 3.881072998046875e-05,
      "model_forward_time": 1.7498681545257568,
      "step": 40696
    },
    {
      "epoch": 3.881072998046875e-05,
      "step": 40696,
      "training_step_time": 7.499432325363159
    },
    {
      "epoch": 3.881168365478516e-05,
      "model_forward_time": 1.744640588760376,
      "step": 40697
    },
    {
      "epoch": 3.881168365478516e-05,
      "step": 40697,
      "training_step_time": 7.52391505241394
    },
    {
      "epoch": 3.881263732910156e-05,
      "model_forward_time": 1.7660086154937744,
      "step": 40698
    },
    {
      "epoch": 3.881263732910156e-05,
      "step": 40698,
      "training_step_time": 7.60216212272644
    },
    {
      "epoch": 3.881359100341797e-05,
      "model_forward_time": 1.7462637424468994,
      "step": 40699
    },
    {
      "epoch": 3.881359100341797e-05,
      "step": 40699,
      "training_step_time": 8.559700965881348
    },
    {
      "epoch": 3.881454467773438e-05,
      "grad_norm": 0.3043407201766968,
      "learning_rate": 6.902107152664852e-06,
      "loss": 0.058,
      "step": 40700
    },
    {
      "epoch": 3.881454467773438e-05,
      "model_forward_time": 2.9431493282318115,
      "step": 40700
    },
    {
      "dynamics_loss_avg": 0.06249753572046757,
      "epoch": 3.881454467773438e-05,
      "step": 40700
    },
    {
      "action_loss_avg": 0.002104284689994529,
      "epoch": 3.881454467773438e-05,
      "step": 40700
    },
    {
      "epoch": 3.881454467773438e-05,
      "step": 40700,
      "training_step_time": 8.546959161758423
    },
    {
      "epoch": 3.881549835205078e-05,
      "model_forward_time": 1.7408161163330078,
      "step": 40701
    },
    {
      "epoch": 3.881549835205078e-05,
      "step": 40701,
      "training_step_time": 8.763114929199219
    },
    {
      "epoch": 3.8816452026367186e-05,
      "model_forward_time": 1.7542259693145752,
      "step": 40702
    },
    {
      "epoch": 3.8816452026367186e-05,
      "step": 40702,
      "training_step_time": 8.391421794891357
    },
    {
      "epoch": 3.8817405700683597e-05,
      "model_forward_time": 1.7397823333740234,
      "step": 40703
    },
    {
      "epoch": 3.8817405700683597e-05,
      "step": 40703,
      "training_step_time": 8.396369218826294
    },
    {
      "epoch": 3.8818359375e-05,
      "model_forward_time": 1.7407300472259521,
      "step": 40704
    },
    {
      "epoch": 3.8818359375e-05,
      "step": 40704,
      "training_step_time": 7.6863439083099365
    },
    {
      "epoch": 3.8819313049316405e-05,
      "model_forward_time": 1.740095853805542,
      "step": 40705
    },
    {
      "epoch": 3.8819313049316405e-05,
      "step": 40705,
      "training_step_time": 7.261216402053833
    },
    {
      "epoch": 3.8820266723632815e-05,
      "model_forward_time": 1.7381541728973389,
      "step": 40706
    },
    {
      "epoch": 3.8820266723632815e-05,
      "step": 40706,
      "training_step_time": 7.702138423919678
    },
    {
      "epoch": 3.882122039794922e-05,
      "model_forward_time": 1.7488157749176025,
      "step": 40707
    },
    {
      "epoch": 3.882122039794922e-05,
      "step": 40707,
      "training_step_time": 7.823147773742676
    },
    {
      "epoch": 3.8822174072265623e-05,
      "model_forward_time": 1.7517683506011963,
      "step": 40708
    },
    {
      "epoch": 3.8822174072265623e-05,
      "step": 40708,
      "training_step_time": 7.532629013061523
    },
    {
      "epoch": 3.8823127746582034e-05,
      "model_forward_time": 2.9229931831359863,
      "step": 40709
    },
    {
      "epoch": 3.8823127746582034e-05,
      "step": 40709,
      "training_step_time": 7.673499822616577
    },
    {
      "epoch": 3.882408142089844e-05,
      "grad_norm": 0.16421431303024292,
      "learning_rate": 6.900577897901961e-06,
      "loss": 0.0569,
      "step": 40710
    },
    {
      "epoch": 3.882408142089844e-05,
      "model_forward_time": 1.748220682144165,
      "step": 40710
    },
    {
      "dynamics_loss_avg": 0.07779010850936174,
      "epoch": 3.882408142089844e-05,
      "step": 40710
    },
    {
      "action_loss_avg": 0.002687557606259361,
      "epoch": 3.882408142089844e-05,
      "step": 40710
    },
    {
      "epoch": 3.882408142089844e-05,
      "step": 40710,
      "training_step_time": 8.159622430801392
    },
    {
      "epoch": 3.882503509521484e-05,
      "model_forward_time": 1.7548930644989014,
      "step": 40711
    },
    {
      "epoch": 3.882503509521484e-05,
      "step": 40711,
      "training_step_time": 8.764454126358032
    },
    {
      "epoch": 3.882598876953125e-05,
      "model_forward_time": 1.7374694347381592,
      "step": 40712
    },
    {
      "epoch": 3.882598876953125e-05,
      "step": 40712,
      "training_step_time": 8.83546757698059
    },
    {
      "epoch": 3.882694244384766e-05,
      "model_forward_time": 1.7417125701904297,
      "step": 40713
    },
    {
      "epoch": 3.882694244384766e-05,
      "step": 40713,
      "training_step_time": 7.626247406005859
    },
    {
      "epoch": 3.882789611816406e-05,
      "model_forward_time": 1.752223014831543,
      "step": 40714
    },
    {
      "epoch": 3.882789611816406e-05,
      "step": 40714,
      "training_step_time": 7.459375858306885
    },
    {
      "epoch": 3.882884979248047e-05,
      "model_forward_time": 1.749598503112793,
      "step": 40715
    },
    {
      "epoch": 3.882884979248047e-05,
      "step": 40715,
      "training_step_time": 7.72806191444397
    },
    {
      "epoch": 3.8829803466796876e-05,
      "model_forward_time": 1.756408929824829,
      "step": 40716
    },
    {
      "epoch": 3.8829803466796876e-05,
      "step": 40716,
      "training_step_time": 7.751891851425171
    },
    {
      "epoch": 3.883075714111328e-05,
      "model_forward_time": 1.7475917339324951,
      "step": 40717
    },
    {
      "epoch": 3.883075714111328e-05,
      "step": 40717,
      "training_step_time": 7.712928533554077
    },
    {
      "epoch": 3.883171081542969e-05,
      "model_forward_time": 2.89735746383667,
      "step": 40718
    },
    {
      "epoch": 3.883171081542969e-05,
      "step": 40718,
      "training_step_time": 9.32379674911499
    },
    {
      "epoch": 3.8832664489746095e-05,
      "model_forward_time": 1.7444121837615967,
      "step": 40719
    },
    {
      "epoch": 3.8832664489746095e-05,
      "step": 40719,
      "training_step_time": 8.886383771896362
    },
    {
      "epoch": 3.88336181640625e-05,
      "grad_norm": 0.15322177112102509,
      "learning_rate": 6.8990484352947285e-06,
      "loss": 0.056,
      "step": 40720
    },
    {
      "epoch": 3.88336181640625e-05,
      "model_forward_time": 1.7479732036590576,
      "step": 40720
    },
    {
      "dynamics_loss_avg": 0.06875318847596645,
      "epoch": 3.88336181640625e-05,
      "step": 40720
    },
    {
      "action_loss_avg": 0.002066780495806597,
      "epoch": 3.88336181640625e-05,
      "step": 40720
    },
    {
      "epoch": 3.88336181640625e-05,
      "step": 40720,
      "training_step_time": 8.800621032714844
    },
    {
      "epoch": 3.88345718383789e-05,
      "model_forward_time": 1.746427059173584,
      "step": 40721
    },
    {
      "epoch": 3.88345718383789e-05,
      "step": 40721,
      "training_step_time": 7.611734628677368
    },
    {
      "epoch": 3.8835525512695314e-05,
      "model_forward_time": 1.7452929019927979,
      "step": 40722
    },
    {
      "epoch": 3.8835525512695314e-05,
      "step": 40722,
      "training_step_time": 7.777434825897217
    },
    {
      "epoch": 3.883647918701172e-05,
      "model_forward_time": 1.7563469409942627,
      "step": 40723
    },
    {
      "epoch": 3.883647918701172e-05,
      "step": 40723,
      "training_step_time": 7.4142351150512695
    },
    {
      "epoch": 3.883743286132812e-05,
      "model_forward_time": 1.7541542053222656,
      "step": 40724
    },
    {
      "epoch": 3.883743286132812e-05,
      "step": 40724,
      "training_step_time": 9.168207883834839
    },
    {
      "epoch": 3.883838653564453e-05,
      "model_forward_time": 1.774237871170044,
      "step": 40725
    },
    {
      "epoch": 3.883838653564453e-05,
      "step": 40725,
      "training_step_time": 7.640488147735596
    },
    {
      "epoch": 3.883934020996094e-05,
      "model_forward_time": 1.7651340961456299,
      "step": 40726
    },
    {
      "epoch": 3.883934020996094e-05,
      "step": 40726,
      "training_step_time": 7.616497755050659
    },
    {
      "epoch": 3.884029388427734e-05,
      "model_forward_time": 1.8308720588684082,
      "step": 40727
    },
    {
      "epoch": 3.884029388427734e-05,
      "step": 40727,
      "training_step_time": 7.595110654830933
    },
    {
      "epoch": 3.884124755859375e-05,
      "model_forward_time": 2.954122304916382,
      "step": 40728
    },
    {
      "epoch": 3.884124755859375e-05,
      "step": 40728,
      "training_step_time": 8.63508129119873
    },
    {
      "epoch": 3.8842201232910155e-05,
      "model_forward_time": 1.7705888748168945,
      "step": 40729
    },
    {
      "epoch": 3.8842201232910155e-05,
      "step": 40729,
      "training_step_time": 8.596338033676147
    },
    {
      "epoch": 3.884315490722656e-05,
      "grad_norm": 0.12864716351032257,
      "learning_rate": 6.897518765010415e-06,
      "loss": 0.0582,
      "step": 40730
    },
    {
      "epoch": 3.884315490722656e-05,
      "model_forward_time": 1.768303632736206,
      "step": 40730
    },
    {
      "dynamics_loss_avg": 0.037468787655234335,
      "epoch": 3.884315490722656e-05,
      "step": 40730
    },
    {
      "action_loss_avg": 0.0015447314945049584,
      "epoch": 3.884315490722656e-05,
      "step": 40730
    },
    {
      "epoch": 3.884315490722656e-05,
      "step": 40730,
      "training_step_time": 7.557226896286011
    },
    {
      "epoch": 3.884410858154297e-05,
      "model_forward_time": 1.7710208892822266,
      "step": 40731
    },
    {
      "epoch": 3.884410858154297e-05,
      "step": 40731,
      "training_step_time": 7.600188970565796
    },
    {
      "epoch": 3.8845062255859374e-05,
      "model_forward_time": 1.7599561214447021,
      "step": 40732
    },
    {
      "epoch": 3.8845062255859374e-05,
      "step": 40732,
      "training_step_time": 7.329115629196167
    },
    {
      "epoch": 3.884601593017578e-05,
      "model_forward_time": 1.4001281261444092,
      "step": 40733
    },
    {
      "epoch": 3.884601593017578e-05,
      "step": 40733,
      "training_step_time": 7.5859620571136475
    },
    {
      "epoch": 3.884696960449219e-05,
      "model_forward_time": 1.7667484283447266,
      "step": 40734
    },
    {
      "epoch": 3.884696960449219e-05,
      "step": 40734,
      "training_step_time": 8.611801147460938
    },
    {
      "epoch": 3.884792327880859e-05,
      "model_forward_time": 1.767503261566162,
      "step": 40735
    },
    {
      "epoch": 3.884792327880859e-05,
      "step": 40735,
      "training_step_time": 7.6624908447265625
    },
    {
      "epoch": 3.8848876953125e-05,
      "model_forward_time": 1.8228943347930908,
      "step": 40736
    },
    {
      "epoch": 3.8848876953125e-05,
      "step": 40736,
      "training_step_time": 9.323815107345581
    },
    {
      "epoch": 3.884983062744141e-05,
      "model_forward_time": 2.884481191635132,
      "step": 40737
    },
    {
      "epoch": 3.884983062744141e-05,
      "step": 40737,
      "training_step_time": 8.278358459472656
    },
    {
      "epoch": 3.885078430175781e-05,
      "model_forward_time": 1.7621901035308838,
      "step": 40738
    },
    {
      "epoch": 3.885078430175781e-05,
      "step": 40738,
      "training_step_time": 7.828415870666504
    },
    {
      "epoch": 3.8851737976074216e-05,
      "model_forward_time": 1.765357255935669,
      "step": 40739
    },
    {
      "epoch": 3.8851737976074216e-05,
      "step": 40739,
      "training_step_time": 7.5847344398498535
    },
    {
      "epoch": 3.885269165039063e-05,
      "grad_norm": 0.1405850052833557,
      "learning_rate": 6.895988887216303e-06,
      "loss": 0.0566,
      "step": 40740
    },
    {
      "epoch": 3.885269165039063e-05,
      "model_forward_time": 1.755063533782959,
      "step": 40740
    },
    {
      "dynamics_loss_avg": 0.0645140515640378,
      "epoch": 3.885269165039063e-05,
      "step": 40740
    },
    {
      "action_loss_avg": 0.002283137338235974,
      "epoch": 3.885269165039063e-05,
      "step": 40740
    },
    {
      "epoch": 3.885269165039063e-05,
      "step": 40740,
      "training_step_time": 7.607683181762695
    },
    {
      "epoch": 3.885364532470703e-05,
      "model_forward_time": 1.7491722106933594,
      "step": 40741
    },
    {
      "epoch": 3.885364532470703e-05,
      "step": 40741,
      "training_step_time": 7.369425535202026
    },
    {
      "epoch": 3.8854598999023435e-05,
      "model_forward_time": 1.7719552516937256,
      "step": 40742
    },
    {
      "epoch": 3.8854598999023435e-05,
      "step": 40742,
      "training_step_time": 7.583372116088867
    },
    {
      "epoch": 3.8855552673339846e-05,
      "model_forward_time": 1.7479121685028076,
      "step": 40743
    },
    {
      "epoch": 3.8855552673339846e-05,
      "step": 40743,
      "training_step_time": 8.133100032806396
    },
    {
      "epoch": 3.885650634765625e-05,
      "model_forward_time": 1.751513957977295,
      "step": 40744
    },
    {
      "epoch": 3.885650634765625e-05,
      "step": 40744,
      "training_step_time": 7.626054525375366
    },
    {
      "epoch": 3.8857460021972654e-05,
      "model_forward_time": 1.8183386325836182,
      "step": 40745
    },
    {
      "epoch": 3.8857460021972654e-05,
      "step": 40745,
      "training_step_time": 7.579655408859253
    },
    {
      "epoch": 3.8858413696289065e-05,
      "model_forward_time": 2.908524990081787,
      "step": 40746
    },
    {
      "epoch": 3.8858413696289065e-05,
      "step": 40746,
      "training_step_time": 9.210467338562012
    },
    {
      "epoch": 3.885936737060547e-05,
      "model_forward_time": 1.7478599548339844,
      "step": 40747
    },
    {
      "epoch": 3.885936737060547e-05,
      "step": 40747,
      "training_step_time": 7.951213836669922
    },
    {
      "epoch": 3.886032104492187e-05,
      "model_forward_time": 1.752915859222412,
      "step": 40748
    },
    {
      "epoch": 3.886032104492187e-05,
      "step": 40748,
      "training_step_time": 7.602965354919434
    },
    {
      "epoch": 3.8861274719238283e-05,
      "model_forward_time": 1.7569429874420166,
      "step": 40749
    },
    {
      "epoch": 3.8861274719238283e-05,
      "step": 40749,
      "training_step_time": 7.5020036697387695
    },
    {
      "epoch": 3.886222839355469e-05,
      "grad_norm": 0.19866827130317688,
      "learning_rate": 6.894458802079694e-06,
      "loss": 0.0592,
      "step": 40750
    },
    {
      "epoch": 3.886222839355469e-05,
      "model_forward_time": 1.753359317779541,
      "step": 40750
    },
    {
      "dynamics_loss_avg": 0.057343796268105506,
      "epoch": 3.886222839355469e-05,
      "step": 40750
    },
    {
      "action_loss_avg": 0.0018804221937898547,
      "epoch": 3.886222839355469e-05,
      "step": 40750
    },
    {
      "epoch": 3.886222839355469e-05,
      "step": 40750,
      "training_step_time": 7.67165207862854
    },
    {
      "epoch": 3.886318206787109e-05,
      "model_forward_time": 1.7548584938049316,
      "step": 40751
    },
    {
      "epoch": 3.886318206787109e-05,
      "step": 40751,
      "training_step_time": 7.505398750305176
    },
    {
      "epoch": 3.88641357421875e-05,
      "model_forward_time": 1.745859146118164,
      "step": 40752
    },
    {
      "epoch": 3.88641357421875e-05,
      "step": 40752,
      "training_step_time": 7.492122650146484
    },
    {
      "epoch": 3.8865089416503906e-05,
      "model_forward_time": 1.744441032409668,
      "step": 40753
    },
    {
      "epoch": 3.8865089416503906e-05,
      "step": 40753,
      "training_step_time": 7.582828044891357
    },
    {
      "epoch": 3.886604309082031e-05,
      "model_forward_time": 1.81644606590271,
      "step": 40754
    },
    {
      "epoch": 3.886604309082031e-05,
      "step": 40754,
      "training_step_time": 8.401134490966797
    },
    {
      "epoch": 3.886699676513672e-05,
      "model_forward_time": 2.8664755821228027,
      "step": 40755
    },
    {
      "epoch": 3.886699676513672e-05,
      "step": 40755,
      "training_step_time": 8.257218837738037
    },
    {
      "epoch": 3.8867950439453125e-05,
      "model_forward_time": 1.7724101543426514,
      "step": 40756
    },
    {
      "epoch": 3.8867950439453125e-05,
      "step": 40756,
      "training_step_time": 8.36382269859314
    },
    {
      "epoch": 3.886890411376953e-05,
      "model_forward_time": 1.7704215049743652,
      "step": 40757
    },
    {
      "epoch": 3.886890411376953e-05,
      "step": 40757,
      "training_step_time": 8.67989706993103
    },
    {
      "epoch": 3.886985778808594e-05,
      "model_forward_time": 1.766685962677002,
      "step": 40758
    },
    {
      "epoch": 3.886985778808594e-05,
      "step": 40758,
      "training_step_time": 8.270896673202515
    },
    {
      "epoch": 3.8870811462402344e-05,
      "model_forward_time": 1.7420642375946045,
      "step": 40759
    },
    {
      "epoch": 3.8870811462402344e-05,
      "step": 40759,
      "training_step_time": 8.653356790542603
    },
    {
      "epoch": 3.887176513671875e-05,
      "grad_norm": 0.132035031914711,
      "learning_rate": 6.892928509767922e-06,
      "loss": 0.0625,
      "step": 40760
    },
    {
      "epoch": 3.887176513671875e-05,
      "model_forward_time": 1.753173589706421,
      "step": 40760
    },
    {
      "dynamics_loss_avg": 0.05397526528686285,
      "epoch": 3.887176513671875e-05,
      "step": 40760
    },
    {
      "action_loss_avg": 0.002544940268853679,
      "epoch": 3.887176513671875e-05,
      "step": 40760
    },
    {
      "epoch": 3.887176513671875e-05,
      "step": 40760,
      "training_step_time": 9.312763452529907
    },
    {
      "epoch": 3.887271881103516e-05,
      "model_forward_time": 1.7644264698028564,
      "step": 40761
    },
    {
      "epoch": 3.887271881103516e-05,
      "step": 40761,
      "training_step_time": 7.489984512329102
    },
    {
      "epoch": 3.887367248535156e-05,
      "model_forward_time": 1.7591335773468018,
      "step": 40762
    },
    {
      "epoch": 3.887367248535156e-05,
      "step": 40762,
      "training_step_time": 7.623323202133179
    },
    {
      "epoch": 3.887462615966797e-05,
      "model_forward_time": 1.809849500656128,
      "step": 40763
    },
    {
      "epoch": 3.887462615966797e-05,
      "step": 40763,
      "training_step_time": 8.236025333404541
    },
    {
      "epoch": 3.887557983398438e-05,
      "model_forward_time": 1.7376272678375244,
      "step": 40764
    },
    {
      "epoch": 3.887557983398438e-05,
      "step": 40764,
      "training_step_time": 8.697256803512573
    },
    {
      "epoch": 3.887653350830078e-05,
      "model_forward_time": 1.7598938941955566,
      "step": 40765
    },
    {
      "epoch": 3.887653350830078e-05,
      "step": 40765,
      "training_step_time": 8.461225748062134
    },
    {
      "epoch": 3.8877487182617186e-05,
      "model_forward_time": 1.7769899368286133,
      "step": 40766
    },
    {
      "epoch": 3.8877487182617186e-05,
      "step": 40766,
      "training_step_time": 7.915501356124878
    },
    {
      "epoch": 3.88784408569336e-05,
      "model_forward_time": 1.7563104629516602,
      "step": 40767
    },
    {
      "epoch": 3.88784408569336e-05,
      "step": 40767,
      "training_step_time": 7.997151136398315
    },
    {
      "epoch": 3.887939453125e-05,
      "model_forward_time": 1.7554008960723877,
      "step": 40768
    },
    {
      "epoch": 3.887939453125e-05,
      "step": 40768,
      "training_step_time": 7.622250318527222
    },
    {
      "epoch": 3.8880348205566405e-05,
      "model_forward_time": 1.7578611373901367,
      "step": 40769
    },
    {
      "epoch": 3.8880348205566405e-05,
      "step": 40769,
      "training_step_time": 7.359591245651245
    },
    {
      "epoch": 3.8881301879882815e-05,
      "grad_norm": 0.12163247913122177,
      "learning_rate": 6.891398010448333e-06,
      "loss": 0.0622,
      "step": 40770
    },
    {
      "epoch": 3.8881301879882815e-05,
      "model_forward_time": 1.7473385334014893,
      "step": 40770
    },
    {
      "dynamics_loss_avg": 0.057810566201806066,
      "epoch": 3.8881301879882815e-05,
      "step": 40770
    },
    {
      "action_loss_avg": 0.002988029518746771,
      "epoch": 3.8881301879882815e-05,
      "step": 40770
    },
    {
      "epoch": 3.8881301879882815e-05,
      "step": 40770,
      "training_step_time": 7.599122524261475
    },
    {
      "epoch": 3.888225555419922e-05,
      "model_forward_time": 1.7500455379486084,
      "step": 40771
    },
    {
      "epoch": 3.888225555419922e-05,
      "step": 40771,
      "training_step_time": 7.671937465667725
    },
    {
      "epoch": 3.8883209228515624e-05,
      "model_forward_time": 1.8276197910308838,
      "step": 40772
    },
    {
      "epoch": 3.8883209228515624e-05,
      "step": 40772,
      "training_step_time": 7.169172525405884
    },
    {
      "epoch": 3.8884162902832034e-05,
      "model_forward_time": 1.7534575462341309,
      "step": 40773
    },
    {
      "epoch": 3.8884162902832034e-05,
      "step": 40773,
      "training_step_time": 7.6788246631622314
    },
    {
      "epoch": 3.888511657714844e-05,
      "model_forward_time": 2.8922946453094482,
      "step": 40774
    },
    {
      "epoch": 3.888511657714844e-05,
      "step": 40774,
      "training_step_time": 8.475572109222412
    },
    {
      "epoch": 3.888607025146484e-05,
      "model_forward_time": 1.7540440559387207,
      "step": 40775
    },
    {
      "epoch": 3.888607025146484e-05,
      "step": 40775,
      "training_step_time": 8.740601539611816
    },
    {
      "epoch": 3.888702392578125e-05,
      "model_forward_time": 1.7399535179138184,
      "step": 40776
    },
    {
      "epoch": 3.888702392578125e-05,
      "step": 40776,
      "training_step_time": 7.533391952514648
    },
    {
      "epoch": 3.888797760009766e-05,
      "model_forward_time": 1.7598986625671387,
      "step": 40777
    },
    {
      "epoch": 3.888797760009766e-05,
      "step": 40777,
      "training_step_time": 8.000656604766846
    },
    {
      "epoch": 3.888893127441406e-05,
      "model_forward_time": 1.7574787139892578,
      "step": 40778
    },
    {
      "epoch": 3.888893127441406e-05,
      "step": 40778,
      "training_step_time": 7.673658847808838
    },
    {
      "epoch": 3.888988494873047e-05,
      "model_forward_time": 1.7406580448150635,
      "step": 40779
    },
    {
      "epoch": 3.888988494873047e-05,
      "step": 40779,
      "training_step_time": 7.553577899932861
    },
    {
      "epoch": 3.8890838623046876e-05,
      "grad_norm": 0.14053203165531158,
      "learning_rate": 6.8898673042883e-06,
      "loss": 0.0639,
      "step": 40780
    },
    {
      "epoch": 3.8890838623046876e-05,
      "model_forward_time": 1.7558410167694092,
      "step": 40780
    },
    {
      "dynamics_loss_avg": 0.04953835960477591,
      "epoch": 3.8890838623046876e-05,
      "step": 40780
    },
    {
      "action_loss_avg": 0.00213756441953592,
      "epoch": 3.8890838623046876e-05,
      "step": 40780
    },
    {
      "epoch": 3.8890838623046876e-05,
      "step": 40780,
      "training_step_time": 8.830619812011719
    },
    {
      "epoch": 3.889179229736328e-05,
      "model_forward_time": 1.801344871520996,
      "step": 40781
    },
    {
      "epoch": 3.889179229736328e-05,
      "step": 40781,
      "training_step_time": 7.4906325340271
    },
    {
      "epoch": 3.8892745971679684e-05,
      "model_forward_time": 1.7541978359222412,
      "step": 40782
    },
    {
      "epoch": 3.8892745971679684e-05,
      "step": 40782,
      "training_step_time": 8.753128051757812
    },
    {
      "epoch": 3.8893699645996095e-05,
      "model_forward_time": 2.995094060897827,
      "step": 40783
    },
    {
      "epoch": 3.8893699645996095e-05,
      "step": 40783,
      "training_step_time": 7.774025201797485
    },
    {
      "epoch": 3.88946533203125e-05,
      "model_forward_time": 1.7529399394989014,
      "step": 40784
    },
    {
      "epoch": 3.88946533203125e-05,
      "step": 40784,
      "training_step_time": 9.550944805145264
    },
    {
      "epoch": 3.88956069946289e-05,
      "model_forward_time": 1.757324457168579,
      "step": 40785
    },
    {
      "epoch": 3.88956069946289e-05,
      "step": 40785,
      "training_step_time": 8.761788606643677
    },
    {
      "epoch": 3.8896560668945314e-05,
      "model_forward_time": 1.755467414855957,
      "step": 40786
    },
    {
      "epoch": 3.8896560668945314e-05,
      "step": 40786,
      "training_step_time": 7.432638645172119
    },
    {
      "epoch": 3.889751434326172e-05,
      "model_forward_time": 1.7745962142944336,
      "step": 40787
    },
    {
      "epoch": 3.889751434326172e-05,
      "step": 40787,
      "training_step_time": 8.458512306213379
    },
    {
      "epoch": 3.889846801757812e-05,
      "model_forward_time": 1.763028621673584,
      "step": 40788
    },
    {
      "epoch": 3.889846801757812e-05,
      "step": 40788,
      "training_step_time": 7.551743030548096
    },
    {
      "epoch": 3.889942169189453e-05,
      "model_forward_time": 1.7457075119018555,
      "step": 40789
    },
    {
      "epoch": 3.889942169189453e-05,
      "step": 40789,
      "training_step_time": 7.612126350402832
    },
    {
      "epoch": 3.890037536621094e-05,
      "grad_norm": 0.13330481946468353,
      "learning_rate": 6.888336391455221e-06,
      "loss": 0.0576,
      "step": 40790
    },
    {
      "epoch": 3.890037536621094e-05,
      "model_forward_time": 1.7745304107666016,
      "step": 40790
    },
    {
      "dynamics_loss_avg": 0.059349835570901635,
      "epoch": 3.890037536621094e-05,
      "step": 40790
    },
    {
      "action_loss_avg": 0.0015585388930048794,
      "epoch": 3.890037536621094e-05,
      "step": 40790
    },
    {
      "epoch": 3.890037536621094e-05,
      "step": 40790,
      "training_step_time": 8.37908935546875
    },
    {
      "epoch": 3.890132904052734e-05,
      "model_forward_time": 1.409146785736084,
      "step": 40791
    },
    {
      "epoch": 3.890132904052734e-05,
      "step": 40791,
      "training_step_time": 7.696778059005737
    },
    {
      "epoch": 3.890228271484375e-05,
      "model_forward_time": 2.9234848022460938,
      "step": 40792
    },
    {
      "epoch": 3.890228271484375e-05,
      "step": 40792,
      "training_step_time": 8.432282447814941
    },
    {
      "epoch": 3.8903236389160156e-05,
      "model_forward_time": 1.744680643081665,
      "step": 40793
    },
    {
      "epoch": 3.8903236389160156e-05,
      "step": 40793,
      "training_step_time": 9.137290477752686
    },
    {
      "epoch": 3.890419006347656e-05,
      "model_forward_time": 1.7515454292297363,
      "step": 40794
    },
    {
      "epoch": 3.890419006347656e-05,
      "step": 40794,
      "training_step_time": 7.953733682632446
    },
    {
      "epoch": 3.890514373779297e-05,
      "model_forward_time": 1.403127908706665,
      "step": 40795
    },
    {
      "epoch": 3.890514373779297e-05,
      "step": 40795,
      "training_step_time": 7.526587009429932
    },
    {
      "epoch": 3.8906097412109374e-05,
      "model_forward_time": 1.7478525638580322,
      "step": 40796
    },
    {
      "epoch": 3.8906097412109374e-05,
      "step": 40796,
      "training_step_time": 7.676082372665405
    },
    {
      "epoch": 3.890705108642578e-05,
      "model_forward_time": 1.7620518207550049,
      "step": 40797
    },
    {
      "epoch": 3.890705108642578e-05,
      "step": 40797,
      "training_step_time": 7.47602391242981
    },
    {
      "epoch": 3.890800476074219e-05,
      "model_forward_time": 1.7699782848358154,
      "step": 40798
    },
    {
      "epoch": 3.890800476074219e-05,
      "step": 40798,
      "training_step_time": 7.543740749359131
    },
    {
      "epoch": 3.890895843505859e-05,
      "model_forward_time": 1.740541696548462,
      "step": 40799
    },
    {
      "epoch": 3.890895843505859e-05,
      "step": 40799,
      "training_step_time": 7.469866752624512
    },
    {
      "epoch": 3.8909912109375e-05,
      "grad_norm": 0.12617804110050201,
      "learning_rate": 6.886805272116513e-06,
      "loss": 0.0585,
      "step": 40800
    },
    {
      "epoch": 3.8909912109375e-05,
      "model_forward_time": 1.8179848194122314,
      "step": 40800
    },
    {
      "dynamics_loss_avg": 0.0560006495565176,
      "epoch": 3.8909912109375e-05,
      "step": 40800
    },
    {
      "action_loss_avg": 0.0020834906492382286,
      "epoch": 3.8909912109375e-05,
      "step": 40800
    },
    {
      "epoch": 3.8909912109375e-05,
      "step": 40800,
      "training_step_time": 7.6418445110321045
    },
    {
      "epoch": 3.891086578369141e-05,
      "model_forward_time": 2.847151041030884,
      "step": 40801
    },
    {
      "epoch": 3.891086578369141e-05,
      "step": 40801,
      "training_step_time": 7.868183374404907
    },
    {
      "epoch": 3.891181945800781e-05,
      "model_forward_time": 1.7500133514404297,
      "step": 40802
    },
    {
      "epoch": 3.891181945800781e-05,
      "step": 40802,
      "training_step_time": 7.783658742904663
    },
    {
      "epoch": 3.8912773132324216e-05,
      "model_forward_time": 1.7523269653320312,
      "step": 40803
    },
    {
      "epoch": 3.8912773132324216e-05,
      "step": 40803,
      "training_step_time": 9.669366121292114
    },
    {
      "epoch": 3.891372680664063e-05,
      "model_forward_time": 1.750351905822754,
      "step": 40804
    },
    {
      "epoch": 3.891372680664063e-05,
      "step": 40804,
      "training_step_time": 7.494088888168335
    },
    {
      "epoch": 3.891468048095703e-05,
      "model_forward_time": 1.761918067932129,
      "step": 40805
    },
    {
      "epoch": 3.891468048095703e-05,
      "step": 40805,
      "training_step_time": 8.5655357837677
    },
    {
      "epoch": 3.8915634155273435e-05,
      "model_forward_time": 1.7595655918121338,
      "step": 40806
    },
    {
      "epoch": 3.8915634155273435e-05,
      "step": 40806,
      "training_step_time": 7.166162490844727
    },
    {
      "epoch": 3.8916587829589846e-05,
      "model_forward_time": 1.7522151470184326,
      "step": 40807
    },
    {
      "epoch": 3.8916587829589846e-05,
      "step": 40807,
      "training_step_time": 7.613038778305054
    },
    {
      "epoch": 3.891754150390625e-05,
      "model_forward_time": 1.7665436267852783,
      "step": 40808
    },
    {
      "epoch": 3.891754150390625e-05,
      "step": 40808,
      "training_step_time": 7.656862497329712
    },
    {
      "epoch": 3.8918495178222654e-05,
      "model_forward_time": 1.8177175521850586,
      "step": 40809
    },
    {
      "epoch": 3.8918495178222654e-05,
      "step": 40809,
      "training_step_time": 8.253156900405884
    },
    {
      "epoch": 3.8919448852539065e-05,
      "grad_norm": 0.13278451561927795,
      "learning_rate": 6.885273946439617e-06,
      "loss": 0.0582,
      "step": 40810
    },
    {
      "epoch": 3.8919448852539065e-05,
      "model_forward_time": 1.7548742294311523,
      "step": 40810
    },
    {
      "dynamics_loss_avg": 0.04531121253967285,
      "epoch": 3.8919448852539065e-05,
      "step": 40810
    },
    {
      "action_loss_avg": 0.0014129665330983698,
      "epoch": 3.8919448852539065e-05,
      "step": 40810
    },
    {
      "epoch": 3.8919448852539065e-05,
      "step": 40810,
      "training_step_time": 8.43116283416748
    },
    {
      "epoch": 3.892040252685547e-05,
      "model_forward_time": 2.8931267261505127,
      "step": 40811
    },
    {
      "epoch": 3.892040252685547e-05,
      "step": 40811,
      "training_step_time": 7.861128807067871
    },
    {
      "epoch": 3.892135620117187e-05,
      "model_forward_time": 1.7626152038574219,
      "step": 40812
    },
    {
      "epoch": 3.892135620117187e-05,
      "step": 40812,
      "training_step_time": 7.636179685592651
    },
    {
      "epoch": 3.8922309875488284e-05,
      "model_forward_time": 1.7624733448028564,
      "step": 40813
    },
    {
      "epoch": 3.8922309875488284e-05,
      "step": 40813,
      "training_step_time": 7.616450309753418
    },
    {
      "epoch": 3.892326354980469e-05,
      "model_forward_time": 1.7516400814056396,
      "step": 40814
    },
    {
      "epoch": 3.892326354980469e-05,
      "step": 40814,
      "training_step_time": 8.140580892562866
    },
    {
      "epoch": 3.892421722412109e-05,
      "model_forward_time": 1.7488186359405518,
      "step": 40815
    },
    {
      "epoch": 3.892421722412109e-05,
      "step": 40815,
      "training_step_time": 7.305530548095703
    },
    {
      "epoch": 3.89251708984375e-05,
      "model_forward_time": 1.7638325691223145,
      "step": 40816
    },
    {
      "epoch": 3.89251708984375e-05,
      "step": 40816,
      "training_step_time": 7.680971622467041
    },
    {
      "epoch": 3.8926124572753906e-05,
      "model_forward_time": 1.7534465789794922,
      "step": 40817
    },
    {
      "epoch": 3.8926124572753906e-05,
      "step": 40817,
      "training_step_time": 8.13489055633545
    },
    {
      "epoch": 3.892707824707031e-05,
      "model_forward_time": 1.8178496360778809,
      "step": 40818
    },
    {
      "epoch": 3.892707824707031e-05,
      "step": 40818,
      "training_step_time": 7.7422802448272705
    },
    {
      "epoch": 3.892803192138672e-05,
      "model_forward_time": 1.0780034065246582,
      "step": 40819
    },
    {
      "epoch": 3.892803192138672e-05,
      "step": 40819,
      "training_step_time": 7.50458836555481
    },
    {
      "epoch": 3.8928985595703125e-05,
      "grad_norm": 0.099428690969944,
      "learning_rate": 6.883742414591999e-06,
      "loss": 0.0592,
      "step": 40820
    },
    {
      "epoch": 3.8928985595703125e-05,
      "model_forward_time": 2.8918492794036865,
      "step": 40820
    },
    {
      "dynamics_loss_avg": 0.04851291161030531,
      "epoch": 3.8928985595703125e-05,
      "step": 40820
    },
    {
      "action_loss_avg": 0.0015539507672656327,
      "epoch": 3.8928985595703125e-05,
      "step": 40820
    },
    {
      "epoch": 3.8928985595703125e-05,
      "step": 40820,
      "training_step_time": 7.883158206939697
    },
    {
      "epoch": 3.892993927001953e-05,
      "model_forward_time": 1.7539632320404053,
      "step": 40821
    },
    {
      "epoch": 3.892993927001953e-05,
      "step": 40821,
      "training_step_time": 8.493451833724976
    },
    {
      "epoch": 3.893089294433594e-05,
      "model_forward_time": 0.7458715438842773,
      "step": 40822
    },
    {
      "epoch": 3.893089294433594e-05,
      "step": 40822,
      "training_step_time": 7.866617441177368
    },
    {
      "epoch": 3.8931846618652344e-05,
      "model_forward_time": 1.7439537048339844,
      "step": 40823
    },
    {
      "epoch": 3.8931846618652344e-05,
      "step": 40823,
      "training_step_time": 8.423228025436401
    },
    {
      "epoch": 3.893280029296875e-05,
      "model_forward_time": 1.7566921710968018,
      "step": 40824
    },
    {
      "epoch": 3.893280029296875e-05,
      "step": 40824,
      "training_step_time": 7.6589226722717285
    },
    {
      "epoch": 3.893375396728516e-05,
      "model_forward_time": 1.7692689895629883,
      "step": 40825
    },
    {
      "epoch": 3.893375396728516e-05,
      "step": 40825,
      "training_step_time": 7.543413877487183
    },
    {
      "epoch": 3.893470764160156e-05,
      "model_forward_time": 1.760439157485962,
      "step": 40826
    },
    {
      "epoch": 3.893470764160156e-05,
      "step": 40826,
      "training_step_time": 8.13689398765564
    },
    {
      "epoch": 3.893566131591797e-05,
      "model_forward_time": 1.8119449615478516,
      "step": 40827
    },
    {
      "epoch": 3.893566131591797e-05,
      "step": 40827,
      "training_step_time": 7.554933786392212
    },
    {
      "epoch": 3.893661499023438e-05,
      "model_forward_time": 1.7695283889770508,
      "step": 40828
    },
    {
      "epoch": 3.893661499023438e-05,
      "step": 40828,
      "training_step_time": 7.488934755325317
    },
    {
      "epoch": 3.893756866455078e-05,
      "model_forward_time": 1.7439212799072266,
      "step": 40829
    },
    {
      "epoch": 3.893756866455078e-05,
      "step": 40829,
      "training_step_time": 8.378849029541016
    },
    {
      "epoch": 3.8938522338867186e-05,
      "grad_norm": 0.11363628506660461,
      "learning_rate": 6.88221067674114e-06,
      "loss": 0.0569,
      "step": 40830
    },
    {
      "epoch": 3.8938522338867186e-05,
      "model_forward_time": 2.9044759273529053,
      "step": 40830
    },
    {
      "dynamics_loss_avg": 0.04870340153574944,
      "epoch": 3.8938522338867186e-05,
      "step": 40830
    },
    {
      "action_loss_avg": 0.0017640078876866028,
      "epoch": 3.8938522338867186e-05,
      "step": 40830
    },
    {
      "epoch": 3.8938522338867186e-05,
      "step": 40830,
      "training_step_time": 8.499862909317017
    },
    {
      "epoch": 3.89394760131836e-05,
      "model_forward_time": 1.7719874382019043,
      "step": 40831
    },
    {
      "epoch": 3.89394760131836e-05,
      "step": 40831,
      "training_step_time": 9.314228296279907
    },
    {
      "epoch": 3.89404296875e-05,
      "model_forward_time": 1.7599437236785889,
      "step": 40832
    },
    {
      "epoch": 3.89404296875e-05,
      "step": 40832,
      "training_step_time": 7.6288604736328125
    },
    {
      "epoch": 3.8941383361816405e-05,
      "model_forward_time": 1.763756275177002,
      "step": 40833
    },
    {
      "epoch": 3.8941383361816405e-05,
      "step": 40833,
      "training_step_time": 7.742833137512207
    },
    {
      "epoch": 3.8942337036132816e-05,
      "model_forward_time": 1.7677366733551025,
      "step": 40834
    },
    {
      "epoch": 3.8942337036132816e-05,
      "step": 40834,
      "training_step_time": 7.546641111373901
    },
    {
      "epoch": 3.894329071044922e-05,
      "model_forward_time": 1.757594347000122,
      "step": 40835
    },
    {
      "epoch": 3.894329071044922e-05,
      "step": 40835,
      "training_step_time": 7.61990213394165
    },
    {
      "epoch": 3.8944244384765624e-05,
      "model_forward_time": 1.7949693202972412,
      "step": 40836
    },
    {
      "epoch": 3.8944244384765624e-05,
      "step": 40836,
      "training_step_time": 7.660065174102783
    },
    {
      "epoch": 3.8945198059082034e-05,
      "model_forward_time": 1.7674899101257324,
      "step": 40837
    },
    {
      "epoch": 3.8945198059082034e-05,
      "step": 40837,
      "training_step_time": 7.718993663787842
    },
    {
      "epoch": 3.894615173339844e-05,
      "model_forward_time": 1.7617897987365723,
      "step": 40838
    },
    {
      "epoch": 3.894615173339844e-05,
      "step": 40838,
      "training_step_time": 8.319674491882324
    },
    {
      "epoch": 3.894710540771484e-05,
      "model_forward_time": 1.77083158493042,
      "step": 40839
    },
    {
      "epoch": 3.894710540771484e-05,
      "step": 40839,
      "training_step_time": 7.69976282119751
    },
    {
      "epoch": 3.894805908203125e-05,
      "grad_norm": 0.13462910056114197,
      "learning_rate": 6.880678733054551e-06,
      "loss": 0.0562,
      "step": 40840
    },
    {
      "epoch": 3.894805908203125e-05,
      "model_forward_time": 1.756272792816162,
      "step": 40840
    },
    {
      "dynamics_loss_avg": 0.06026469822973013,
      "epoch": 3.894805908203125e-05,
      "step": 40840
    },
    {
      "action_loss_avg": 0.001849153297371231,
      "epoch": 3.894805908203125e-05,
      "step": 40840
    },
    {
      "epoch": 3.894805908203125e-05,
      "step": 40840,
      "training_step_time": 7.956114768981934
    },
    {
      "epoch": 3.894901275634766e-05,
      "model_forward_time": 1.772230625152588,
      "step": 40841
    },
    {
      "epoch": 3.894901275634766e-05,
      "step": 40841,
      "training_step_time": 8.070096731185913
    },
    {
      "epoch": 3.894996643066406e-05,
      "model_forward_time": 1.0762782096862793,
      "step": 40842
    },
    {
      "epoch": 3.894996643066406e-05,
      "step": 40842,
      "training_step_time": 9.1580650806427
    },
    {
      "epoch": 3.895092010498047e-05,
      "model_forward_time": 1.7614479064941406,
      "step": 40843
    },
    {
      "epoch": 3.895092010498047e-05,
      "step": 40843,
      "training_step_time": 7.535518169403076
    },
    {
      "epoch": 3.8951873779296876e-05,
      "model_forward_time": 1.763892412185669,
      "step": 40844
    },
    {
      "epoch": 3.8951873779296876e-05,
      "step": 40844,
      "training_step_time": 7.394074440002441
    },
    {
      "epoch": 3.895282745361328e-05,
      "model_forward_time": 1.7792701721191406,
      "step": 40845
    },
    {
      "epoch": 3.895282745361328e-05,
      "step": 40845,
      "training_step_time": 9.172128200531006
    },
    {
      "epoch": 3.8953781127929684e-05,
      "model_forward_time": 1.7655940055847168,
      "step": 40846
    },
    {
      "epoch": 3.8953781127929684e-05,
      "step": 40846,
      "training_step_time": 8.239766120910645
    },
    {
      "epoch": 3.8954734802246095e-05,
      "model_forward_time": 1.7702577114105225,
      "step": 40847
    },
    {
      "epoch": 3.8954734802246095e-05,
      "step": 40847,
      "training_step_time": 8.617915391921997
    },
    {
      "epoch": 3.89556884765625e-05,
      "model_forward_time": 2.9162962436676025,
      "step": 40848
    },
    {
      "epoch": 3.89556884765625e-05,
      "step": 40848,
      "training_step_time": 8.594192743301392
    },
    {
      "epoch": 3.89566421508789e-05,
      "model_forward_time": 1.759652853012085,
      "step": 40849
    },
    {
      "epoch": 3.89566421508789e-05,
      "step": 40849,
      "training_step_time": 8.507445096969604
    },
    {
      "epoch": 3.8957595825195314e-05,
      "grad_norm": 0.09143353253602982,
      "learning_rate": 6.879146583699765e-06,
      "loss": 0.0607,
      "step": 40850
    },
    {
      "epoch": 3.8957595825195314e-05,
      "model_forward_time": 1.76558256149292,
      "step": 40850
    },
    {
      "dynamics_loss_avg": 0.0780234606936574,
      "epoch": 3.8957595825195314e-05,
      "step": 40850
    },
    {
      "action_loss_avg": 0.0018141123000532389,
      "epoch": 3.8957595825195314e-05,
      "step": 40850
    },
    {
      "epoch": 3.8957595825195314e-05,
      "step": 40850,
      "training_step_time": 7.601818084716797
    },
    {
      "epoch": 3.895854949951172e-05,
      "model_forward_time": 1.7646217346191406,
      "step": 40851
    },
    {
      "epoch": 3.895854949951172e-05,
      "step": 40851,
      "training_step_time": 7.477407455444336
    },
    {
      "epoch": 3.895950317382812e-05,
      "model_forward_time": 1.7479488849639893,
      "step": 40852
    },
    {
      "epoch": 3.895950317382812e-05,
      "step": 40852,
      "training_step_time": 8.719010591506958
    },
    {
      "epoch": 3.896045684814453e-05,
      "model_forward_time": 1.7635879516601562,
      "step": 40853
    },
    {
      "epoch": 3.896045684814453e-05,
      "step": 40853,
      "training_step_time": 7.370337247848511
    },
    {
      "epoch": 3.896141052246094e-05,
      "model_forward_time": 1.7605211734771729,
      "step": 40854
    },
    {
      "epoch": 3.896141052246094e-05,
      "step": 40854,
      "training_step_time": 7.617663145065308
    },
    {
      "epoch": 3.896236419677734e-05,
      "model_forward_time": 1.8234164714813232,
      "step": 40855
    },
    {
      "epoch": 3.896236419677734e-05,
      "step": 40855,
      "training_step_time": 7.626982688903809
    },
    {
      "epoch": 3.896331787109375e-05,
      "model_forward_time": 1.7620794773101807,
      "step": 40856
    },
    {
      "epoch": 3.896331787109375e-05,
      "step": 40856,
      "training_step_time": 8.613329887390137
    },
    {
      "epoch": 3.8964271545410156e-05,
      "model_forward_time": 2.8864598274230957,
      "step": 40857
    },
    {
      "epoch": 3.8964271545410156e-05,
      "step": 40857,
      "training_step_time": 7.85417914390564
    },
    {
      "epoch": 3.896522521972656e-05,
      "model_forward_time": 1.7608695030212402,
      "step": 40858
    },
    {
      "epoch": 3.896522521972656e-05,
      "step": 40858,
      "training_step_time": 8.465020895004272
    },
    {
      "epoch": 3.896617889404297e-05,
      "model_forward_time": 1.7698049545288086,
      "step": 40859
    },
    {
      "epoch": 3.896617889404297e-05,
      "step": 40859,
      "training_step_time": 7.6459715366363525
    },
    {
      "epoch": 3.8967132568359374e-05,
      "grad_norm": 0.12095660716295242,
      "learning_rate": 6.877614228844334e-06,
      "loss": 0.0605,
      "step": 40860
    },
    {
      "epoch": 3.8967132568359374e-05,
      "model_forward_time": 1.7711503505706787,
      "step": 40860
    },
    {
      "dynamics_loss_avg": 0.060235757008194925,
      "epoch": 3.8967132568359374e-05,
      "step": 40860
    },
    {
      "action_loss_avg": 0.001968867500545457,
      "epoch": 3.8967132568359374e-05,
      "step": 40860
    },
    {
      "epoch": 3.8967132568359374e-05,
      "step": 40860,
      "training_step_time": 8.748055696487427
    },
    {
      "epoch": 3.896808624267578e-05,
      "model_forward_time": 1.7713546752929688,
      "step": 40861
    },
    {
      "epoch": 3.896808624267578e-05,
      "step": 40861,
      "training_step_time": 7.779603004455566
    },
    {
      "epoch": 3.896903991699219e-05,
      "model_forward_time": 1.766404151916504,
      "step": 40862
    },
    {
      "epoch": 3.896903991699219e-05,
      "step": 40862,
      "training_step_time": 7.597548246383667
    },
    {
      "epoch": 3.896999359130859e-05,
      "model_forward_time": 1.7688508033752441,
      "step": 40863
    },
    {
      "epoch": 3.896999359130859e-05,
      "step": 40863,
      "training_step_time": 8.25451946258545
    },
    {
      "epoch": 3.8970947265625e-05,
      "model_forward_time": 1.8296267986297607,
      "step": 40864
    },
    {
      "epoch": 3.8970947265625e-05,
      "step": 40864,
      "training_step_time": 8.527109622955322
    },
    {
      "epoch": 3.897190093994141e-05,
      "model_forward_time": 1.762486457824707,
      "step": 40865
    },
    {
      "epoch": 3.897190093994141e-05,
      "step": 40865,
      "training_step_time": 7.503494024276733
    },
    {
      "epoch": 3.897285461425781e-05,
      "model_forward_time": 2.8661067485809326,
      "step": 40866
    },
    {
      "epoch": 3.897285461425781e-05,
      "step": 40866,
      "training_step_time": 7.812479019165039
    },
    {
      "epoch": 3.8973808288574216e-05,
      "model_forward_time": 1.7671611309051514,
      "step": 40867
    },
    {
      "epoch": 3.8973808288574216e-05,
      "step": 40867,
      "training_step_time": 7.62174391746521
    },
    {
      "epoch": 3.897476196289063e-05,
      "model_forward_time": 1.7589778900146484,
      "step": 40868
    },
    {
      "epoch": 3.897476196289063e-05,
      "step": 40868,
      "training_step_time": 7.544959783554077
    },
    {
      "epoch": 3.897571563720703e-05,
      "model_forward_time": 1.7707643508911133,
      "step": 40869
    },
    {
      "epoch": 3.897571563720703e-05,
      "step": 40869,
      "training_step_time": 7.590863943099976
    },
    {
      "epoch": 3.8976669311523435e-05,
      "grad_norm": 0.16239389777183533,
      "learning_rate": 6.8760816686558315e-06,
      "loss": 0.0558,
      "step": 40870
    },
    {
      "epoch": 3.8976669311523435e-05,
      "model_forward_time": 1.7658157348632812,
      "step": 40870
    },
    {
      "dynamics_loss_avg": 0.0559053048491478,
      "epoch": 3.8976669311523435e-05,
      "step": 40870
    },
    {
      "action_loss_avg": 0.001042952697025612,
      "epoch": 3.8976669311523435e-05,
      "step": 40870
    },
    {
      "epoch": 3.8976669311523435e-05,
      "step": 40870,
      "training_step_time": 7.6887288093566895
    },
    {
      "epoch": 3.8977622985839846e-05,
      "model_forward_time": 1.7706668376922607,
      "step": 40871
    },
    {
      "epoch": 3.8977622985839846e-05,
      "step": 40871,
      "training_step_time": 7.568679571151733
    },
    {
      "epoch": 3.897857666015625e-05,
      "model_forward_time": 1.7593047618865967,
      "step": 40872
    },
    {
      "epoch": 3.897857666015625e-05,
      "step": 40872,
      "training_step_time": 8.265608549118042
    },
    {
      "epoch": 3.8979530334472654e-05,
      "model_forward_time": 1.8296763896942139,
      "step": 40873
    },
    {
      "epoch": 3.8979530334472654e-05,
      "step": 40873,
      "training_step_time": 7.624972581863403
    },
    {
      "epoch": 3.8980484008789065e-05,
      "model_forward_time": 1.7658698558807373,
      "step": 40874
    },
    {
      "epoch": 3.8980484008789065e-05,
      "step": 40874,
      "training_step_time": 7.601551055908203
    },
    {
      "epoch": 3.898143768310547e-05,
      "model_forward_time": 1.755544900894165,
      "step": 40875
    },
    {
      "epoch": 3.898143768310547e-05,
      "step": 40875,
      "training_step_time": 8.97016191482544
    },
    {
      "epoch": 3.898239135742187e-05,
      "model_forward_time": 1.7642343044281006,
      "step": 40876
    },
    {
      "epoch": 3.898239135742187e-05,
      "step": 40876,
      "training_step_time": 8.732274770736694
    },
    {
      "epoch": 3.8983345031738284e-05,
      "model_forward_time": 1.756575584411621,
      "step": 40877
    },
    {
      "epoch": 3.8983345031738284e-05,
      "step": 40877,
      "training_step_time": 8.635139226913452
    },
    {
      "epoch": 3.898429870605469e-05,
      "model_forward_time": 1.7537782192230225,
      "step": 40878
    },
    {
      "epoch": 3.898429870605469e-05,
      "step": 40878,
      "training_step_time": 7.67347526550293
    },
    {
      "epoch": 3.898525238037109e-05,
      "model_forward_time": 1.7574126720428467,
      "step": 40879
    },
    {
      "epoch": 3.898525238037109e-05,
      "step": 40879,
      "training_step_time": 7.575703382492065
    },
    {
      "epoch": 3.89862060546875e-05,
      "grad_norm": 0.14008860290050507,
      "learning_rate": 6.8745489033018606e-06,
      "loss": 0.0576,
      "step": 40880
    },
    {
      "epoch": 3.89862060546875e-05,
      "model_forward_time": 1.7581830024719238,
      "step": 40880
    },
    {
      "dynamics_loss_avg": 0.05712627023458481,
      "epoch": 3.89862060546875e-05,
      "step": 40880
    },
    {
      "action_loss_avg": 0.0018003983306698502,
      "epoch": 3.89862060546875e-05,
      "step": 40880
    },
    {
      "epoch": 3.89862060546875e-05,
      "step": 40880,
      "training_step_time": 7.5945656299591064
    },
    {
      "epoch": 3.8987159729003906e-05,
      "model_forward_time": 1.7600297927856445,
      "step": 40881
    },
    {
      "epoch": 3.8987159729003906e-05,
      "step": 40881,
      "training_step_time": 7.663265943527222
    },
    {
      "epoch": 3.898811340332031e-05,
      "model_forward_time": 1.8083574771881104,
      "step": 40882
    },
    {
      "epoch": 3.898811340332031e-05,
      "step": 40882,
      "training_step_time": 7.619517087936401
    },
    {
      "epoch": 3.898906707763672e-05,
      "model_forward_time": 1.7642173767089844,
      "step": 40883
    },
    {
      "epoch": 3.898906707763672e-05,
      "step": 40883,
      "training_step_time": 7.605390310287476
    },
    {
      "epoch": 3.8990020751953125e-05,
      "model_forward_time": 1.7782819271087646,
      "step": 40884
    },
    {
      "epoch": 3.8990020751953125e-05,
      "step": 40884,
      "training_step_time": 9.860052585601807
    },
    {
      "epoch": 3.899097442626953e-05,
      "model_forward_time": 2.88543701171875,
      "step": 40885
    },
    {
      "epoch": 3.899097442626953e-05,
      "step": 40885,
      "training_step_time": 8.637024641036987
    },
    {
      "epoch": 3.899192810058594e-05,
      "model_forward_time": 1.757016658782959,
      "step": 40886
    },
    {
      "epoch": 3.899192810058594e-05,
      "step": 40886,
      "training_step_time": 9.126461267471313
    },
    {
      "epoch": 3.8992881774902344e-05,
      "model_forward_time": 1.7548694610595703,
      "step": 40887
    },
    {
      "epoch": 3.8992881774902344e-05,
      "step": 40887,
      "training_step_time": 8.356355428695679
    },
    {
      "epoch": 3.899383544921875e-05,
      "model_forward_time": 1.7590463161468506,
      "step": 40888
    },
    {
      "epoch": 3.899383544921875e-05,
      "step": 40888,
      "training_step_time": 7.367901563644409
    },
    {
      "epoch": 3.899478912353516e-05,
      "model_forward_time": 1.7640228271484375,
      "step": 40889
    },
    {
      "epoch": 3.899478912353516e-05,
      "step": 40889,
      "training_step_time": 7.641372442245483
    },
    {
      "epoch": 3.899574279785156e-05,
      "grad_norm": 0.14520291984081268,
      "learning_rate": 6.873015932950039e-06,
      "loss": 0.0562,
      "step": 40890
    },
    {
      "epoch": 3.899574279785156e-05,
      "model_forward_time": 1.7640769481658936,
      "step": 40890
    },
    {
      "dynamics_loss_avg": 0.05270489528775215,
      "epoch": 3.899574279785156e-05,
      "step": 40890
    },
    {
      "action_loss_avg": 0.0023422600352205336,
      "epoch": 3.899574279785156e-05,
      "step": 40890
    },
    {
      "epoch": 3.899574279785156e-05,
      "step": 40890,
      "training_step_time": 8.219279766082764
    },
    {
      "epoch": 3.899669647216797e-05,
      "model_forward_time": 1.8236281871795654,
      "step": 40891
    },
    {
      "epoch": 3.899669647216797e-05,
      "step": 40891,
      "training_step_time": 7.668545246124268
    },
    {
      "epoch": 3.899765014648438e-05,
      "model_forward_time": 1.7745659351348877,
      "step": 40892
    },
    {
      "epoch": 3.899765014648438e-05,
      "step": 40892,
      "training_step_time": 7.564549684524536
    },
    {
      "epoch": 3.899860382080078e-05,
      "model_forward_time": 1.7555804252624512,
      "step": 40893
    },
    {
      "epoch": 3.899860382080078e-05,
      "step": 40893,
      "training_step_time": 8.630177974700928
    },
    {
      "epoch": 3.8999557495117186e-05,
      "model_forward_time": 2.918490171432495,
      "step": 40894
    },
    {
      "epoch": 3.8999557495117186e-05,
      "step": 40894,
      "training_step_time": 8.67768669128418
    },
    {
      "epoch": 3.90005111694336e-05,
      "model_forward_time": 1.7582600116729736,
      "step": 40895
    },
    {
      "epoch": 3.90005111694336e-05,
      "step": 40895,
      "training_step_time": 8.377840518951416
    },
    {
      "epoch": 3.900146484375e-05,
      "model_forward_time": 1.7561500072479248,
      "step": 40896
    },
    {
      "epoch": 3.900146484375e-05,
      "step": 40896,
      "training_step_time": 7.674117803573608
    },
    {
      "epoch": 3.9002418518066405e-05,
      "model_forward_time": 1.7816035747528076,
      "step": 40897
    },
    {
      "epoch": 3.9002418518066405e-05,
      "step": 40897,
      "training_step_time": 7.629502773284912
    },
    {
      "epoch": 3.9003372192382816e-05,
      "model_forward_time": 1.7386541366577148,
      "step": 40898
    },
    {
      "epoch": 3.9003372192382816e-05,
      "step": 40898,
      "training_step_time": 7.686804533004761
    },
    {
      "epoch": 3.900432586669922e-05,
      "model_forward_time": 1.740788221359253,
      "step": 40899
    },
    {
      "epoch": 3.900432586669922e-05,
      "step": 40899,
      "training_step_time": 7.562770843505859
    },
    {
      "epoch": 3.9005279541015624e-05,
      "grad_norm": 0.1967639923095703,
      "learning_rate": 6.871482757768012e-06,
      "loss": 0.0606,
      "step": 40900
    },
    {
      "epoch": 3.9005279541015624e-05,
      "model_forward_time": 1.8081722259521484,
      "step": 40900
    },
    {
      "dynamics_loss_avg": 0.06643375605344773,
      "epoch": 3.9005279541015624e-05,
      "step": 40900
    },
    {
      "action_loss_avg": 0.0014419620914850385,
      "epoch": 3.9005279541015624e-05,
      "step": 40900
    },
    {
      "epoch": 3.9005279541015624e-05,
      "step": 40900,
      "training_step_time": 8.262712717056274
    },
    {
      "epoch": 3.9006233215332034e-05,
      "model_forward_time": 1.7709050178527832,
      "step": 40901
    },
    {
      "epoch": 3.9006233215332034e-05,
      "step": 40901,
      "training_step_time": 7.737544536590576
    },
    {
      "epoch": 3.900718688964844e-05,
      "model_forward_time": 1.7515087127685547,
      "step": 40902
    },
    {
      "epoch": 3.900718688964844e-05,
      "step": 40902,
      "training_step_time": 8.274167776107788
    },
    {
      "epoch": 3.900814056396484e-05,
      "model_forward_time": 2.8775546550750732,
      "step": 40903
    },
    {
      "epoch": 3.900814056396484e-05,
      "step": 40903,
      "training_step_time": 7.905740022659302
    },
    {
      "epoch": 3.900909423828125e-05,
      "model_forward_time": 1.7507522106170654,
      "step": 40904
    },
    {
      "epoch": 3.900909423828125e-05,
      "step": 40904,
      "training_step_time": 7.893494129180908
    },
    {
      "epoch": 3.901004791259766e-05,
      "model_forward_time": 1.7502849102020264,
      "step": 40905
    },
    {
      "epoch": 3.901004791259766e-05,
      "step": 40905,
      "training_step_time": 7.641690254211426
    },
    {
      "epoch": 3.901100158691406e-05,
      "model_forward_time": 1.7408709526062012,
      "step": 40906
    },
    {
      "epoch": 3.901100158691406e-05,
      "step": 40906,
      "training_step_time": 8.62100100517273
    },
    {
      "epoch": 3.9011955261230465e-05,
      "model_forward_time": 1.7406401634216309,
      "step": 40907
    },
    {
      "epoch": 3.9011955261230465e-05,
      "step": 40907,
      "training_step_time": 7.603338956832886
    },
    {
      "epoch": 3.9012908935546876e-05,
      "model_forward_time": 1.7550392150878906,
      "step": 40908
    },
    {
      "epoch": 3.9012908935546876e-05,
      "step": 40908,
      "training_step_time": 7.601847171783447
    },
    {
      "epoch": 3.901386260986328e-05,
      "model_forward_time": 1.7570865154266357,
      "step": 40909
    },
    {
      "epoch": 3.901386260986328e-05,
      "step": 40909,
      "training_step_time": 7.6332848072052
    },
    {
      "epoch": 3.9014816284179684e-05,
      "grad_norm": 0.11985448002815247,
      "learning_rate": 6.869949377923443e-06,
      "loss": 0.0547,
      "step": 40910
    },
    {
      "epoch": 3.9014816284179684e-05,
      "model_forward_time": 1.7462654113769531,
      "step": 40910
    },
    {
      "dynamics_loss_avg": 0.04709627330303192,
      "epoch": 3.9014816284179684e-05,
      "step": 40910
    },
    {
      "action_loss_avg": 0.0019442118820734323,
      "epoch": 3.9014816284179684e-05,
      "step": 40910
    },
    {
      "epoch": 3.9014816284179684e-05,
      "step": 40910,
      "training_step_time": 7.75145697593689
    },
    {
      "epoch": 3.9015769958496095e-05,
      "model_forward_time": 1.735950231552124,
      "step": 40911
    },
    {
      "epoch": 3.9015769958496095e-05,
      "step": 40911,
      "training_step_time": 7.638781785964966
    },
    {
      "epoch": 3.90167236328125e-05,
      "model_forward_time": 2.8835432529449463,
      "step": 40912
    },
    {
      "epoch": 3.90167236328125e-05,
      "step": 40912,
      "training_step_time": 7.676805734634399
    },
    {
      "epoch": 3.90176773071289e-05,
      "model_forward_time": 1.7556335926055908,
      "step": 40913
    },
    {
      "epoch": 3.90176773071289e-05,
      "step": 40913,
      "training_step_time": 8.453611135482788
    },
    {
      "epoch": 3.9018630981445314e-05,
      "model_forward_time": 1.7437448501586914,
      "step": 40914
    },
    {
      "epoch": 3.9018630981445314e-05,
      "step": 40914,
      "training_step_time": 7.650225877761841
    },
    {
      "epoch": 3.901958465576172e-05,
      "model_forward_time": 1.7472155094146729,
      "step": 40915
    },
    {
      "epoch": 3.901958465576172e-05,
      "step": 40915,
      "training_step_time": 8.13367772102356
    },
    {
      "epoch": 3.902053833007812e-05,
      "model_forward_time": 1.7450981140136719,
      "step": 40916
    },
    {
      "epoch": 3.902053833007812e-05,
      "step": 40916,
      "training_step_time": 7.570286273956299
    },
    {
      "epoch": 3.902149200439453e-05,
      "model_forward_time": 1.7370352745056152,
      "step": 40917
    },
    {
      "epoch": 3.902149200439453e-05,
      "step": 40917,
      "training_step_time": 7.739717960357666
    },
    {
      "epoch": 3.902244567871094e-05,
      "model_forward_time": 1.7478892803192139,
      "step": 40918
    },
    {
      "epoch": 3.902244567871094e-05,
      "step": 40918,
      "training_step_time": 7.624669313430786
    },
    {
      "epoch": 3.902339935302734e-05,
      "model_forward_time": 1.456202745437622,
      "step": 40919
    },
    {
      "epoch": 3.902339935302734e-05,
      "step": 40919,
      "training_step_time": 7.7275285720825195
    },
    {
      "epoch": 3.902435302734375e-05,
      "grad_norm": 0.10645225644111633,
      "learning_rate": 6.868415793584023e-06,
      "loss": 0.0584,
      "step": 40920
    },
    {
      "epoch": 3.902435302734375e-05,
      "model_forward_time": 1.7435908317565918,
      "step": 40920
    },
    {
      "dynamics_loss_avg": 0.07328684646636248,
      "epoch": 3.902435302734375e-05,
      "step": 40920
    },
    {
      "action_loss_avg": 0.0016351936908904463,
      "epoch": 3.902435302734375e-05,
      "step": 40920
    },
    {
      "epoch": 3.902435302734375e-05,
      "step": 40920,
      "training_step_time": 8.291297197341919
    },
    {
      "epoch": 3.9025306701660156e-05,
      "model_forward_time": 2.896925926208496,
      "step": 40921
    },
    {
      "epoch": 3.9025306701660156e-05,
      "step": 40921,
      "training_step_time": 7.937328815460205
    },
    {
      "epoch": 3.902626037597656e-05,
      "model_forward_time": 1.7366983890533447,
      "step": 40922
    },
    {
      "epoch": 3.902626037597656e-05,
      "step": 40922,
      "training_step_time": 8.641279220581055
    },
    {
      "epoch": 3.902721405029297e-05,
      "model_forward_time": 1.739037036895752,
      "step": 40923
    },
    {
      "epoch": 3.902721405029297e-05,
      "step": 40923,
      "training_step_time": 9.038830518722534
    },
    {
      "epoch": 3.9028167724609375e-05,
      "model_forward_time": 1.7552824020385742,
      "step": 40924
    },
    {
      "epoch": 3.9028167724609375e-05,
      "step": 40924,
      "training_step_time": 8.679445505142212
    },
    {
      "epoch": 3.902912139892578e-05,
      "model_forward_time": 1.7507801055908203,
      "step": 40925
    },
    {
      "epoch": 3.902912139892578e-05,
      "step": 40925,
      "training_step_time": 8.697426080703735
    },
    {
      "epoch": 3.903007507324219e-05,
      "model_forward_time": 1.7494964599609375,
      "step": 40926
    },
    {
      "epoch": 3.903007507324219e-05,
      "step": 40926,
      "training_step_time": 7.719810485839844
    },
    {
      "epoch": 3.903102874755859e-05,
      "model_forward_time": 1.7384800910949707,
      "step": 40927
    },
    {
      "epoch": 3.903102874755859e-05,
      "step": 40927,
      "training_step_time": 7.596719741821289
    },
    {
      "epoch": 3.9031982421875e-05,
      "model_forward_time": 1.831223726272583,
      "step": 40928
    },
    {
      "epoch": 3.9031982421875e-05,
      "step": 40928,
      "training_step_time": 7.689427137374878
    },
    {
      "epoch": 3.903293609619141e-05,
      "model_forward_time": 1.7419154644012451,
      "step": 40929
    },
    {
      "epoch": 3.903293609619141e-05,
      "step": 40929,
      "training_step_time": 7.545606851577759
    },
    {
      "epoch": 3.903388977050781e-05,
      "grad_norm": 0.0983816534280777,
      "learning_rate": 6.86688200491746e-06,
      "loss": 0.0532,
      "step": 40930
    },
    {
      "epoch": 3.903388977050781e-05,
      "model_forward_time": 2.887497663497925,
      "step": 40930
    },
    {
      "dynamics_loss_avg": 0.03967921286821365,
      "epoch": 3.903388977050781e-05,
      "step": 40930
    },
    {
      "action_loss_avg": 0.0011332965455949306,
      "epoch": 3.903388977050781e-05,
      "step": 40930
    },
    {
      "epoch": 3.903388977050781e-05,
      "step": 40930,
      "training_step_time": 7.761245012283325
    },
    {
      "epoch": 3.9034843444824216e-05,
      "model_forward_time": 1.7486305236816406,
      "step": 40931
    },
    {
      "epoch": 3.9034843444824216e-05,
      "step": 40931,
      "training_step_time": 7.935248851776123
    },
    {
      "epoch": 3.903579711914063e-05,
      "model_forward_time": 1.7424793243408203,
      "step": 40932
    },
    {
      "epoch": 3.903579711914063e-05,
      "step": 40932,
      "training_step_time": 8.656081199645996
    },
    {
      "epoch": 3.903675079345703e-05,
      "model_forward_time": 1.7369720935821533,
      "step": 40933
    },
    {
      "epoch": 3.903675079345703e-05,
      "step": 40933,
      "training_step_time": 7.64828896522522
    },
    {
      "epoch": 3.9037704467773435e-05,
      "model_forward_time": 1.7641873359680176,
      "step": 40934
    },
    {
      "epoch": 3.9037704467773435e-05,
      "step": 40934,
      "training_step_time": 8.508521556854248
    },
    {
      "epoch": 3.9038658142089846e-05,
      "model_forward_time": 1.7493538856506348,
      "step": 40935
    },
    {
      "epoch": 3.9038658142089846e-05,
      "step": 40935,
      "training_step_time": 8.31746768951416
    },
    {
      "epoch": 3.903961181640625e-05,
      "model_forward_time": 1.7428059577941895,
      "step": 40936
    },
    {
      "epoch": 3.903961181640625e-05,
      "step": 40936,
      "training_step_time": 7.565490484237671
    },
    {
      "epoch": 3.9040565490722654e-05,
      "model_forward_time": 1.814046859741211,
      "step": 40937
    },
    {
      "epoch": 3.9040565490722654e-05,
      "step": 40937,
      "training_step_time": 7.75366997718811
    },
    {
      "epoch": 3.9041519165039065e-05,
      "model_forward_time": 1.749054193496704,
      "step": 40938
    },
    {
      "epoch": 3.9041519165039065e-05,
      "step": 40938,
      "training_step_time": 8.456952810287476
    },
    {
      "epoch": 3.904247283935547e-05,
      "model_forward_time": 2.850482940673828,
      "step": 40939
    },
    {
      "epoch": 3.904247283935547e-05,
      "step": 40939,
      "training_step_time": 8.947472095489502
    },
    {
      "epoch": 3.904342651367187e-05,
      "grad_norm": 0.10618340969085693,
      "learning_rate": 6.8653480120914874e-06,
      "loss": 0.057,
      "step": 40940
    },
    {
      "epoch": 3.904342651367187e-05,
      "model_forward_time": 1.7495381832122803,
      "step": 40940
    },
    {
      "dynamics_loss_avg": 0.05705342572182417,
      "epoch": 3.904342651367187e-05,
      "step": 40940
    },
    {
      "action_loss_avg": 0.0013806961360387504,
      "epoch": 3.904342651367187e-05,
      "step": 40940
    },
    {
      "epoch": 3.904342651367187e-05,
      "step": 40940,
      "training_step_time": 7.921697616577148
    },
    {
      "epoch": 3.9044380187988284e-05,
      "model_forward_time": 1.7382512092590332,
      "step": 40941
    },
    {
      "epoch": 3.9044380187988284e-05,
      "step": 40941,
      "training_step_time": 7.603428363800049
    },
    {
      "epoch": 3.904533386230469e-05,
      "model_forward_time": 1.7436037063598633,
      "step": 40942
    },
    {
      "epoch": 3.904533386230469e-05,
      "step": 40942,
      "training_step_time": 8.62446928024292
    },
    {
      "epoch": 3.904628753662109e-05,
      "model_forward_time": 1.744544267654419,
      "step": 40943
    },
    {
      "epoch": 3.904628753662109e-05,
      "step": 40943,
      "training_step_time": 9.442637920379639
    },
    {
      "epoch": 3.90472412109375e-05,
      "model_forward_time": 1.7397558689117432,
      "step": 40944
    },
    {
      "epoch": 3.90472412109375e-05,
      "step": 40944,
      "training_step_time": 7.777425289154053
    },
    {
      "epoch": 3.9048194885253907e-05,
      "model_forward_time": 1.7619614601135254,
      "step": 40945
    },
    {
      "epoch": 3.9048194885253907e-05,
      "step": 40945,
      "training_step_time": 7.603153944015503
    },
    {
      "epoch": 3.904914855957031e-05,
      "model_forward_time": 1.8089358806610107,
      "step": 40946
    },
    {
      "epoch": 3.904914855957031e-05,
      "step": 40946,
      "training_step_time": 7.722529649734497
    },
    {
      "epoch": 3.905010223388672e-05,
      "model_forward_time": 1.7364726066589355,
      "step": 40947
    },
    {
      "epoch": 3.905010223388672e-05,
      "step": 40947,
      "training_step_time": 7.645374536514282
    },
    {
      "epoch": 3.9051055908203125e-05,
      "model_forward_time": 2.83902907371521,
      "step": 40948
    },
    {
      "epoch": 3.9051055908203125e-05,
      "step": 40948,
      "training_step_time": 7.657840013504028
    },
    {
      "epoch": 3.905200958251953e-05,
      "model_forward_time": 1.7536356449127197,
      "step": 40949
    },
    {
      "epoch": 3.905200958251953e-05,
      "step": 40949,
      "training_step_time": 9.035459518432617
    },
    {
      "epoch": 3.905296325683594e-05,
      "grad_norm": 0.10957931727170944,
      "learning_rate": 6.863813815273861e-06,
      "loss": 0.0548,
      "step": 40950
    },
    {
      "epoch": 3.905296325683594e-05,
      "model_forward_time": 1.749955654144287,
      "step": 40950
    },
    {
      "dynamics_loss_avg": 0.05359509885311127,
      "epoch": 3.905296325683594e-05,
      "step": 40950
    },
    {
      "action_loss_avg": 0.0011650295462459325,
      "epoch": 3.905296325683594e-05,
      "step": 40950
    },
    {
      "epoch": 3.905296325683594e-05,
      "step": 40950,
      "training_step_time": 7.5634496212005615
    },
    {
      "epoch": 3.9053916931152344e-05,
      "model_forward_time": 1.743025779724121,
      "step": 40951
    },
    {
      "epoch": 3.9053916931152344e-05,
      "step": 40951,
      "training_step_time": 7.610530376434326
    },
    {
      "epoch": 3.905487060546875e-05,
      "model_forward_time": 1.7359881401062012,
      "step": 40952
    },
    {
      "epoch": 3.905487060546875e-05,
      "step": 40952,
      "training_step_time": 7.660563707351685
    },
    {
      "epoch": 3.905582427978516e-05,
      "model_forward_time": 1.7479023933410645,
      "step": 40953
    },
    {
      "epoch": 3.905582427978516e-05,
      "step": 40953,
      "training_step_time": 9.47294569015503
    },
    {
      "epoch": 3.905677795410156e-05,
      "model_forward_time": 1.782766342163086,
      "step": 40954
    },
    {
      "epoch": 3.905677795410156e-05,
      "step": 40954,
      "training_step_time": 7.668107748031616
    },
    {
      "epoch": 3.905773162841797e-05,
      "model_forward_time": 1.7593662738800049,
      "step": 40955
    },
    {
      "epoch": 3.905773162841797e-05,
      "step": 40955,
      "training_step_time": 7.671269178390503
    },
    {
      "epoch": 3.905868530273438e-05,
      "model_forward_time": 1.7423033714294434,
      "step": 40956
    },
    {
      "epoch": 3.905868530273438e-05,
      "step": 40956,
      "training_step_time": 8.784139394760132
    },
    {
      "epoch": 3.905963897705078e-05,
      "model_forward_time": 1.749763011932373,
      "step": 40957
    },
    {
      "epoch": 3.905963897705078e-05,
      "step": 40957,
      "training_step_time": 8.748250007629395
    },
    {
      "epoch": 3.9060592651367186e-05,
      "model_forward_time": 2.907773017883301,
      "step": 40958
    },
    {
      "epoch": 3.9060592651367186e-05,
      "step": 40958,
      "training_step_time": 9.114144325256348
    },
    {
      "epoch": 3.90615463256836e-05,
      "model_forward_time": 1.756155252456665,
      "step": 40959
    },
    {
      "epoch": 3.90615463256836e-05,
      "step": 40959,
      "training_step_time": 7.6620588302612305
    },
    {
      "epoch": 3.90625e-05,
      "grad_norm": 0.09133794158697128,
      "learning_rate": 6.862279414632359e-06,
      "loss": 0.0553,
      "step": 40960
    },
    {
      "epoch": 3.90625e-05,
      "model_forward_time": 1.7499878406524658,
      "step": 40960
    },
    {
      "dynamics_loss_avg": 0.06779462695121766,
      "epoch": 3.90625e-05,
      "step": 40960
    },
    {
      "action_loss_avg": 0.0024932429252658038,
      "epoch": 3.90625e-05,
      "step": 40960
    },
    {
      "epoch": 3.90625e-05,
      "step": 40960,
      "training_step_time": 8.376243114471436
    },
    {
      "epoch": 3.9063453674316405e-05,
      "model_forward_time": 1.7439424991607666,
      "step": 40961
    },
    {
      "epoch": 3.9063453674316405e-05,
      "step": 40961,
      "training_step_time": 7.880172967910767
    },
    {
      "epoch": 3.9064407348632816e-05,
      "model_forward_time": 1.7406840324401855,
      "step": 40962
    },
    {
      "epoch": 3.9064407348632816e-05,
      "step": 40962,
      "training_step_time": 7.65576696395874
    },
    {
      "epoch": 3.906536102294922e-05,
      "model_forward_time": 1.747171401977539,
      "step": 40963
    },
    {
      "epoch": 3.906536102294922e-05,
      "step": 40963,
      "training_step_time": 7.6363701820373535
    },
    {
      "epoch": 3.9066314697265624e-05,
      "model_forward_time": 1.7450456619262695,
      "step": 40964
    },
    {
      "epoch": 3.9066314697265624e-05,
      "step": 40964,
      "training_step_time": 7.69061279296875
    },
    {
      "epoch": 3.9067268371582035e-05,
      "model_forward_time": 1.8189282417297363,
      "step": 40965
    },
    {
      "epoch": 3.9067268371582035e-05,
      "step": 40965,
      "training_step_time": 7.689613103866577
    },
    {
      "epoch": 3.906822204589844e-05,
      "model_forward_time": 1.389939785003662,
      "step": 40966
    },
    {
      "epoch": 3.906822204589844e-05,
      "step": 40966,
      "training_step_time": 7.521014928817749
    },
    {
      "epoch": 3.906917572021484e-05,
      "model_forward_time": 2.8913474082946777,
      "step": 40967
    },
    {
      "epoch": 3.906917572021484e-05,
      "step": 40967,
      "training_step_time": 8.013724088668823
    },
    {
      "epoch": 3.907012939453125e-05,
      "model_forward_time": 1.7542929649353027,
      "step": 40968
    },
    {
      "epoch": 3.907012939453125e-05,
      "step": 40968,
      "training_step_time": 10.052245140075684
    },
    {
      "epoch": 3.907108306884766e-05,
      "model_forward_time": 1.7529053688049316,
      "step": 40969
    },
    {
      "epoch": 3.907108306884766e-05,
      "step": 40969,
      "training_step_time": 7.639556646347046
    },
    {
      "epoch": 3.907203674316406e-05,
      "grad_norm": 0.1392601728439331,
      "learning_rate": 6.860744810334779e-06,
      "loss": 0.0563,
      "step": 40970
    },
    {
      "epoch": 3.907203674316406e-05,
      "model_forward_time": 1.747523546218872,
      "step": 40970
    },
    {
      "dynamics_loss_avg": 0.049343375861644743,
      "epoch": 3.907203674316406e-05,
      "step": 40970
    },
    {
      "action_loss_avg": 0.0023231060709804296,
      "epoch": 3.907203674316406e-05,
      "step": 40970
    },
    {
      "epoch": 3.907203674316406e-05,
      "step": 40970,
      "training_step_time": 7.655507802963257
    },
    {
      "epoch": 3.9072990417480465e-05,
      "model_forward_time": 1.7453327178955078,
      "step": 40971
    },
    {
      "epoch": 3.9072990417480465e-05,
      "step": 40971,
      "training_step_time": 7.67679762840271
    },
    {
      "epoch": 3.9073944091796876e-05,
      "model_forward_time": 1.7426114082336426,
      "step": 40972
    },
    {
      "epoch": 3.9073944091796876e-05,
      "step": 40972,
      "training_step_time": 7.677669525146484
    },
    {
      "epoch": 3.907489776611328e-05,
      "model_forward_time": 1.7353284358978271,
      "step": 40973
    },
    {
      "epoch": 3.907489776611328e-05,
      "step": 40973,
      "training_step_time": 8.414625883102417
    },
    {
      "epoch": 3.9075851440429684e-05,
      "model_forward_time": 1.830423355102539,
      "step": 40974
    },
    {
      "epoch": 3.9075851440429684e-05,
      "step": 40974,
      "training_step_time": 7.707782983779907
    },
    {
      "epoch": 3.9076805114746095e-05,
      "model_forward_time": 1.7778732776641846,
      "step": 40975
    },
    {
      "epoch": 3.9076805114746095e-05,
      "step": 40975,
      "training_step_time": 8.155606985092163
    },
    {
      "epoch": 3.90777587890625e-05,
      "model_forward_time": 2.888732671737671,
      "step": 40976
    },
    {
      "epoch": 3.90777587890625e-05,
      "step": 40976,
      "training_step_time": 9.213663816452026
    },
    {
      "epoch": 3.90787124633789e-05,
      "model_forward_time": 1.7735590934753418,
      "step": 40977
    },
    {
      "epoch": 3.90787124633789e-05,
      "step": 40977,
      "training_step_time": 7.793299436569214
    },
    {
      "epoch": 3.9079666137695314e-05,
      "model_forward_time": 1.770930528640747,
      "step": 40978
    },
    {
      "epoch": 3.9079666137695314e-05,
      "step": 40978,
      "training_step_time": 7.499497175216675
    },
    {
      "epoch": 3.908061981201172e-05,
      "model_forward_time": 1.7733185291290283,
      "step": 40979
    },
    {
      "epoch": 3.908061981201172e-05,
      "step": 40979,
      "training_step_time": 7.591161251068115
    },
    {
      "epoch": 3.908157348632812e-05,
      "grad_norm": 0.08956363797187805,
      "learning_rate": 6.859210002548944e-06,
      "loss": 0.0554,
      "step": 40980
    },
    {
      "epoch": 3.908157348632812e-05,
      "model_forward_time": 1.7590677738189697,
      "step": 40980
    },
    {
      "dynamics_loss_avg": 0.04866644199937582,
      "epoch": 3.908157348632812e-05,
      "step": 40980
    },
    {
      "action_loss_avg": 0.0019093121693003923,
      "epoch": 3.908157348632812e-05,
      "step": 40980
    },
    {
      "epoch": 3.908157348632812e-05,
      "step": 40980,
      "training_step_time": 7.643768072128296
    },
    {
      "epoch": 3.908252716064453e-05,
      "model_forward_time": 1.768324851989746,
      "step": 40981
    },
    {
      "epoch": 3.908252716064453e-05,
      "step": 40981,
      "training_step_time": 8.745202779769897
    },
    {
      "epoch": 3.908348083496094e-05,
      "model_forward_time": 1.7633459568023682,
      "step": 40982
    },
    {
      "epoch": 3.908348083496094e-05,
      "step": 40982,
      "training_step_time": 9.105025053024292
    },
    {
      "epoch": 3.908443450927734e-05,
      "model_forward_time": 1.8277008533477783,
      "step": 40983
    },
    {
      "epoch": 3.908443450927734e-05,
      "step": 40983,
      "training_step_time": 7.6478071212768555
    },
    {
      "epoch": 3.908538818359375e-05,
      "model_forward_time": 1.7594256401062012,
      "step": 40984
    },
    {
      "epoch": 3.908538818359375e-05,
      "step": 40984,
      "training_step_time": 8.184305191040039
    },
    {
      "epoch": 3.9086341857910156e-05,
      "model_forward_time": 2.85298228263855,
      "step": 40985
    },
    {
      "epoch": 3.9086341857910156e-05,
      "step": 40985,
      "training_step_time": 7.907002925872803
    },
    {
      "epoch": 3.908729553222656e-05,
      "model_forward_time": 1.7642009258270264,
      "step": 40986
    },
    {
      "epoch": 3.908729553222656e-05,
      "step": 40986,
      "training_step_time": 9.584516048431396
    },
    {
      "epoch": 3.908824920654297e-05,
      "model_forward_time": 1.7627685070037842,
      "step": 40987
    },
    {
      "epoch": 3.908824920654297e-05,
      "step": 40987,
      "training_step_time": 9.029291868209839
    },
    {
      "epoch": 3.9089202880859375e-05,
      "model_forward_time": 1.7636067867279053,
      "step": 40988
    },
    {
      "epoch": 3.9089202880859375e-05,
      "step": 40988,
      "training_step_time": 7.70078182220459
    },
    {
      "epoch": 3.909015655517578e-05,
      "model_forward_time": 1.7644362449645996,
      "step": 40989
    },
    {
      "epoch": 3.909015655517578e-05,
      "step": 40989,
      "training_step_time": 9.183916807174683
    },
    {
      "epoch": 3.909111022949219e-05,
      "grad_norm": 0.08626911789178848,
      "learning_rate": 6.8576749914427e-06,
      "loss": 0.0583,
      "step": 40990
    },
    {
      "epoch": 3.909111022949219e-05,
      "model_forward_time": 1.7537777423858643,
      "step": 40990
    },
    {
      "dynamics_loss_avg": 0.0664367252960801,
      "epoch": 3.909111022949219e-05,
      "step": 40990
    },
    {
      "action_loss_avg": 0.002098787191789597,
      "epoch": 3.909111022949219e-05,
      "step": 40990
    },
    {
      "epoch": 3.909111022949219e-05,
      "step": 40990,
      "training_step_time": 7.869157791137695
    },
    {
      "epoch": 3.9092063903808593e-05,
      "model_forward_time": 1.7651300430297852,
      "step": 40991
    },
    {
      "epoch": 3.9092063903808593e-05,
      "step": 40991,
      "training_step_time": 7.785031080245972
    },
    {
      "epoch": 3.9093017578125e-05,
      "model_forward_time": 1.4827656745910645,
      "step": 40992
    },
    {
      "epoch": 3.9093017578125e-05,
      "step": 40992,
      "training_step_time": 8.646044969558716
    },
    {
      "epoch": 3.909397125244141e-05,
      "model_forward_time": 1.7542755603790283,
      "step": 40993
    },
    {
      "epoch": 3.909397125244141e-05,
      "step": 40993,
      "training_step_time": 7.686627626419067
    },
    {
      "epoch": 3.909492492675781e-05,
      "model_forward_time": 1.7717225551605225,
      "step": 40994
    },
    {
      "epoch": 3.909492492675781e-05,
      "step": 40994,
      "training_step_time": 7.9223103523254395
    },
    {
      "epoch": 3.9095878601074216e-05,
      "model_forward_time": 2.928112030029297,
      "step": 40995
    },
    {
      "epoch": 3.9095878601074216e-05,
      "step": 40995,
      "training_step_time": 8.408643007278442
    },
    {
      "epoch": 3.909683227539063e-05,
      "model_forward_time": 1.7668368816375732,
      "step": 40996
    },
    {
      "epoch": 3.909683227539063e-05,
      "step": 40996,
      "training_step_time": 7.601522922515869
    },
    {
      "epoch": 3.909778594970703e-05,
      "model_forward_time": 1.7757394313812256,
      "step": 40997
    },
    {
      "epoch": 3.909778594970703e-05,
      "step": 40997,
      "training_step_time": 8.9630126953125
    },
    {
      "epoch": 3.9098739624023435e-05,
      "model_forward_time": 1.809602975845337,
      "step": 40998
    },
    {
      "epoch": 3.9098739624023435e-05,
      "step": 40998,
      "training_step_time": 8.543469429016113
    },
    {
      "epoch": 3.9099693298339846e-05,
      "model_forward_time": 1.7686750888824463,
      "step": 40999
    },
    {
      "epoch": 3.9099693298339846e-05,
      "step": 40999,
      "training_step_time": 7.804225206375122
    },
    {
      "epoch": 3.910064697265625e-05,
      "grad_norm": 0.13448025286197662,
      "learning_rate": 6.85613977718391e-06,
      "loss": 0.0609,
      "step": 41000
    },
    {
      "epoch": 3.910064697265625e-05,
      "model_forward_time": 1.7796740531921387,
      "step": 41000
    },
    {
      "dynamics_loss_avg": 0.049381925351917745,
      "epoch": 3.910064697265625e-05,
      "step": 41000
    },
    {
      "action_loss_avg": 0.0016648984208586625,
      "epoch": 3.910064697265625e-05,
      "step": 41000
    },
    {
      "epoch": 3.910064697265625e-05,
      "step": 41000,
      "training_step_time": 6.838027715682983
    },
    {
      "epoch": 3.9101600646972654e-05,
      "model_forward_time": 1.8185126781463623,
      "step": 41001
    },
    {
      "epoch": 3.9101600646972654e-05,
      "step": 41001,
      "training_step_time": 7.687247037887573
    },
    {
      "epoch": 3.9102554321289065e-05,
      "model_forward_time": 1.773505449295044,
      "step": 41002
    },
    {
      "epoch": 3.9102554321289065e-05,
      "step": 41002,
      "training_step_time": 8.709161520004272
    },
    {
      "epoch": 3.910350799560547e-05,
      "model_forward_time": 3.025306463241577,
      "step": 41003
    },
    {
      "epoch": 3.910350799560547e-05,
      "step": 41003,
      "training_step_time": 10.969833135604858
    },
    {
      "epoch": 3.910446166992187e-05,
      "model_forward_time": 1.7526464462280273,
      "step": 41004
    },
    {
      "epoch": 3.910446166992187e-05,
      "step": 41004,
      "training_step_time": 7.835358619689941
    },
    {
      "epoch": 3.9105415344238284e-05,
      "model_forward_time": 1.770855188369751,
      "step": 41005
    },
    {
      "epoch": 3.9105415344238284e-05,
      "step": 41005,
      "training_step_time": 8.441445112228394
    },
    {
      "epoch": 3.910636901855469e-05,
      "model_forward_time": 1.7863667011260986,
      "step": 41006
    },
    {
      "epoch": 3.910636901855469e-05,
      "step": 41006,
      "training_step_time": 8.878143548965454
    },
    {
      "epoch": 3.910732269287109e-05,
      "model_forward_time": 1.7571489810943604,
      "step": 41007
    },
    {
      "epoch": 3.910732269287109e-05,
      "step": 41007,
      "training_step_time": 8.643388509750366
    },
    {
      "epoch": 3.91082763671875e-05,
      "model_forward_time": 1.790234088897705,
      "step": 41008
    },
    {
      "epoch": 3.91082763671875e-05,
      "step": 41008,
      "training_step_time": 8.705211400985718
    },
    {
      "epoch": 3.9109230041503907e-05,
      "model_forward_time": 1.7629575729370117,
      "step": 41009
    },
    {
      "epoch": 3.9109230041503907e-05,
      "step": 41009,
      "training_step_time": 7.687252759933472
    },
    {
      "epoch": 3.911018371582031e-05,
      "grad_norm": 0.11237212270498276,
      "learning_rate": 6.854604359940465e-06,
      "loss": 0.0618,
      "step": 41010
    },
    {
      "epoch": 3.911018371582031e-05,
      "model_forward_time": 1.793872356414795,
      "step": 41010
    },
    {
      "dynamics_loss_avg": 0.06497268956154585,
      "epoch": 3.911018371582031e-05,
      "step": 41010
    },
    {
      "action_loss_avg": 0.003676145477220416,
      "epoch": 3.911018371582031e-05,
      "step": 41010
    },
    {
      "epoch": 3.911018371582031e-05,
      "step": 41010,
      "training_step_time": 7.698864936828613
    },
    {
      "epoch": 3.911113739013672e-05,
      "model_forward_time": 1.7617967128753662,
      "step": 41011
    },
    {
      "epoch": 3.911113739013672e-05,
      "step": 41011,
      "training_step_time": 7.722337961196899
    },
    {
      "epoch": 3.9112091064453125e-05,
      "model_forward_time": 2.855259895324707,
      "step": 41012
    },
    {
      "epoch": 3.9112091064453125e-05,
      "step": 41012,
      "training_step_time": 8.960945129394531
    },
    {
      "epoch": 3.911304473876953e-05,
      "model_forward_time": 1.7588438987731934,
      "step": 41013
    },
    {
      "epoch": 3.911304473876953e-05,
      "step": 41013,
      "training_step_time": 7.664432764053345
    },
    {
      "epoch": 3.911399841308594e-05,
      "model_forward_time": 1.7656283378601074,
      "step": 41014
    },
    {
      "epoch": 3.911399841308594e-05,
      "step": 41014,
      "training_step_time": 8.472785711288452
    },
    {
      "epoch": 3.9114952087402344e-05,
      "model_forward_time": 1.7717397212982178,
      "step": 41015
    },
    {
      "epoch": 3.9114952087402344e-05,
      "step": 41015,
      "training_step_time": 8.658778667449951
    },
    {
      "epoch": 3.911590576171875e-05,
      "model_forward_time": 1.7690315246582031,
      "step": 41016
    },
    {
      "epoch": 3.911590576171875e-05,
      "step": 41016,
      "training_step_time": 7.6252195835113525
    },
    {
      "epoch": 3.911685943603516e-05,
      "model_forward_time": 1.769843339920044,
      "step": 41017
    },
    {
      "epoch": 3.911685943603516e-05,
      "step": 41017,
      "training_step_time": 7.6771862506866455
    },
    {
      "epoch": 3.911781311035156e-05,
      "model_forward_time": 1.7540888786315918,
      "step": 41018
    },
    {
      "epoch": 3.911781311035156e-05,
      "step": 41018,
      "training_step_time": 7.644052028656006
    },
    {
      "epoch": 3.911876678466797e-05,
      "model_forward_time": 1.7643826007843018,
      "step": 41019
    },
    {
      "epoch": 3.911876678466797e-05,
      "step": 41019,
      "training_step_time": 8.178861141204834
    },
    {
      "epoch": 3.911972045898438e-05,
      "grad_norm": 0.12526527047157288,
      "learning_rate": 6.853068739880276e-06,
      "loss": 0.0618,
      "step": 41020
    },
    {
      "epoch": 3.911972045898438e-05,
      "model_forward_time": 1.4027917385101318,
      "step": 41020
    },
    {
      "dynamics_loss_avg": 0.0627388957887888,
      "epoch": 3.911972045898438e-05,
      "step": 41020
    },
    {
      "action_loss_avg": 0.0024299691198393703,
      "epoch": 3.911972045898438e-05,
      "step": 41020
    },
    {
      "epoch": 3.911972045898438e-05,
      "step": 41020,
      "training_step_time": 8.629898309707642
    },
    {
      "epoch": 3.912067413330078e-05,
      "model_forward_time": 1.768453598022461,
      "step": 41021
    },
    {
      "epoch": 3.912067413330078e-05,
      "step": 41021,
      "training_step_time": 7.897787809371948
    },
    {
      "epoch": 3.9121627807617186e-05,
      "model_forward_time": 2.9356589317321777,
      "step": 41022
    },
    {
      "epoch": 3.9121627807617186e-05,
      "step": 41022,
      "training_step_time": 7.980940341949463
    },
    {
      "epoch": 3.91225814819336e-05,
      "model_forward_time": 1.753391981124878,
      "step": 41023
    },
    {
      "epoch": 3.91225814819336e-05,
      "step": 41023,
      "training_step_time": 8.394468545913696
    },
    {
      "epoch": 3.912353515625e-05,
      "model_forward_time": 1.7706491947174072,
      "step": 41024
    },
    {
      "epoch": 3.912353515625e-05,
      "step": 41024,
      "training_step_time": 8.511403322219849
    },
    {
      "epoch": 3.9124488830566405e-05,
      "model_forward_time": 1.7623240947723389,
      "step": 41025
    },
    {
      "epoch": 3.9124488830566405e-05,
      "step": 41025,
      "training_step_time": 7.625077724456787
    },
    {
      "epoch": 3.9125442504882816e-05,
      "model_forward_time": 1.765641689300537,
      "step": 41026
    },
    {
      "epoch": 3.9125442504882816e-05,
      "step": 41026,
      "training_step_time": 7.612697601318359
    },
    {
      "epoch": 3.912639617919922e-05,
      "model_forward_time": 1.7705252170562744,
      "step": 41027
    },
    {
      "epoch": 3.912639617919922e-05,
      "step": 41027,
      "training_step_time": 7.797106981277466
    },
    {
      "epoch": 3.9127349853515624e-05,
      "model_forward_time": 1.7547996044158936,
      "step": 41028
    },
    {
      "epoch": 3.9127349853515624e-05,
      "step": 41028,
      "training_step_time": 7.703487157821655
    },
    {
      "epoch": 3.9128303527832035e-05,
      "model_forward_time": 1.847374439239502,
      "step": 41029
    },
    {
      "epoch": 3.9128303527832035e-05,
      "step": 41029,
      "training_step_time": 7.758902549743652
    },
    {
      "epoch": 3.912925720214844e-05,
      "grad_norm": 0.1234356090426445,
      "learning_rate": 6.851532917171276e-06,
      "loss": 0.0604,
      "step": 41030
    },
    {
      "epoch": 3.912925720214844e-05,
      "model_forward_time": 1.7651593685150146,
      "step": 41030
    },
    {
      "dynamics_loss_avg": 0.07368458732962609,
      "epoch": 3.912925720214844e-05,
      "step": 41030
    },
    {
      "action_loss_avg": 0.004324140920653008,
      "epoch": 3.912925720214844e-05,
      "step": 41030
    },
    {
      "epoch": 3.912925720214844e-05,
      "step": 41030,
      "training_step_time": 8.307969808578491
    },
    {
      "epoch": 3.913021087646484e-05,
      "model_forward_time": 2.924703598022461,
      "step": 41031
    },
    {
      "epoch": 3.913021087646484e-05,
      "step": 41031,
      "training_step_time": 8.645136594772339
    },
    {
      "epoch": 3.913116455078125e-05,
      "model_forward_time": 1.7737464904785156,
      "step": 41032
    },
    {
      "epoch": 3.913116455078125e-05,
      "step": 41032,
      "training_step_time": 7.766325235366821
    },
    {
      "epoch": 3.913211822509766e-05,
      "model_forward_time": 1.7716655731201172,
      "step": 41033
    },
    {
      "epoch": 3.913211822509766e-05,
      "step": 41033,
      "training_step_time": 7.669848203659058
    },
    {
      "epoch": 3.913307189941406e-05,
      "model_forward_time": 2.985543727874756,
      "step": 41034
    },
    {
      "epoch": 3.913307189941406e-05,
      "step": 41034,
      "training_step_time": 7.829325914382935
    },
    {
      "epoch": 3.9134025573730465e-05,
      "model_forward_time": 1.761723279953003,
      "step": 41035
    },
    {
      "epoch": 3.9134025573730465e-05,
      "step": 41035,
      "training_step_time": 8.555582761764526
    },
    {
      "epoch": 3.9134979248046876e-05,
      "model_forward_time": 1.7652997970581055,
      "step": 41036
    },
    {
      "epoch": 3.9134979248046876e-05,
      "step": 41036,
      "training_step_time": 6.861354112625122
    },
    {
      "epoch": 3.913593292236328e-05,
      "model_forward_time": 1.375779390335083,
      "step": 41037
    },
    {
      "epoch": 3.913593292236328e-05,
      "step": 41037,
      "training_step_time": 6.7537689208984375
    },
    {
      "epoch": 3.9136886596679684e-05,
      "model_forward_time": 1.7466599941253662,
      "step": 41038
    },
    {
      "epoch": 3.9136886596679684e-05,
      "step": 41038,
      "training_step_time": 6.819427251815796
    },
    {
      "epoch": 3.9137840270996095e-05,
      "model_forward_time": 1.7604444026947021,
      "step": 41039
    },
    {
      "epoch": 3.9137840270996095e-05,
      "step": 41039,
      "training_step_time": 6.81166410446167
    },
    {
      "epoch": 3.91387939453125e-05,
      "grad_norm": 0.1735054850578308,
      "learning_rate": 6.8499968919814184e-06,
      "loss": 0.0644,
      "step": 41040
    },
    {
      "epoch": 3.91387939453125e-05,
      "model_forward_time": 1.7472498416900635,
      "step": 41040
    },
    {
      "dynamics_loss_avg": 0.05724930372089147,
      "epoch": 3.91387939453125e-05,
      "step": 41040
    },
    {
      "action_loss_avg": 0.003201581751636695,
      "epoch": 3.91387939453125e-05,
      "step": 41040
    },
    {
      "epoch": 3.91387939453125e-05,
      "step": 41040,
      "training_step_time": 6.891657590866089
    },
    {
      "epoch": 3.91397476196289e-05,
      "model_forward_time": 1.7418627738952637,
      "step": 41041
    },
    {
      "epoch": 3.91397476196289e-05,
      "step": 41041,
      "training_step_time": 6.811624050140381
    },
    {
      "epoch": 3.9140701293945314e-05,
      "model_forward_time": 1.7518861293792725,
      "step": 41042
    },
    {
      "epoch": 3.9140701293945314e-05,
      "step": 41042,
      "training_step_time": 6.810712099075317
    },
    {
      "epoch": 3.914165496826172e-05,
      "model_forward_time": 2.5089237689971924,
      "step": 41043
    },
    {
      "epoch": 3.914165496826172e-05,
      "step": 41043,
      "training_step_time": 10.51191520690918
    },
    {
      "epoch": 3.914260864257812e-05,
      "model_forward_time": 1.7542846202850342,
      "step": 41044
    },
    {
      "epoch": 3.914260864257812e-05,
      "step": 41044,
      "training_step_time": 7.8538432121276855
    },
    {
      "epoch": 3.914356231689453e-05,
      "model_forward_time": 1.7618257999420166,
      "step": 41045
    },
    {
      "epoch": 3.914356231689453e-05,
      "step": 41045,
      "training_step_time": 6.769806385040283
    },
    {
      "epoch": 3.914451599121094e-05,
      "model_forward_time": 1.7629742622375488,
      "step": 41046
    },
    {
      "epoch": 3.914451599121094e-05,
      "step": 41046,
      "training_step_time": 6.786124229431152
    },
    {
      "epoch": 3.914546966552734e-05,
      "model_forward_time": 1.7396330833435059,
      "step": 41047
    },
    {
      "epoch": 3.914546966552734e-05,
      "step": 41047,
      "training_step_time": 6.828313827514648
    },
    {
      "epoch": 3.914642333984375e-05,
      "model_forward_time": 1.7564613819122314,
      "step": 41048
    },
    {
      "epoch": 3.914642333984375e-05,
      "step": 41048,
      "training_step_time": 6.7633726596832275
    },
    {
      "epoch": 3.9147377014160156e-05,
      "model_forward_time": 1.7642133235931396,
      "step": 41049
    },
    {
      "epoch": 3.9147377014160156e-05,
      "step": 41049,
      "training_step_time": 6.739370822906494
    },
    {
      "epoch": 3.914833068847656e-05,
      "grad_norm": 0.13624899089336395,
      "learning_rate": 6.848460664478685e-06,
      "loss": 0.0617,
      "step": 41050
    },
    {
      "epoch": 3.914833068847656e-05,
      "model_forward_time": 1.7801613807678223,
      "step": 41050
    },
    {
      "dynamics_loss_avg": 0.04755161572247744,
      "epoch": 3.914833068847656e-05,
      "step": 41050
    },
    {
      "action_loss_avg": 0.002464153617620468,
      "epoch": 3.914833068847656e-05,
      "step": 41050
    },
    {
      "epoch": 3.914833068847656e-05,
      "step": 41050,
      "training_step_time": 6.692394733428955
    },
    {
      "epoch": 3.914928436279297e-05,
      "model_forward_time": 1.76596999168396,
      "step": 41051
    },
    {
      "epoch": 3.914928436279297e-05,
      "step": 41051,
      "training_step_time": 6.736531972885132
    },
    {
      "epoch": 3.9150238037109375e-05,
      "model_forward_time": 2.893266201019287,
      "step": 41052
    },
    {
      "epoch": 3.9150238037109375e-05,
      "step": 41052,
      "training_step_time": 7.6044111251831055
    },
    {
      "epoch": 3.915119171142578e-05,
      "model_forward_time": 1.748567819595337,
      "step": 41053
    },
    {
      "epoch": 3.915119171142578e-05,
      "step": 41053,
      "training_step_time": 8.614766597747803
    },
    {
      "epoch": 3.915214538574219e-05,
      "model_forward_time": 1.756302833557129,
      "step": 41054
    },
    {
      "epoch": 3.915214538574219e-05,
      "step": 41054,
      "training_step_time": 7.195412874221802
    },
    {
      "epoch": 3.9153099060058593e-05,
      "model_forward_time": 1.764331340789795,
      "step": 41055
    },
    {
      "epoch": 3.9153099060058593e-05,
      "step": 41055,
      "training_step_time": 6.752394676208496
    },
    {
      "epoch": 3.9154052734375e-05,
      "model_forward_time": 1.7747256755828857,
      "step": 41056
    },
    {
      "epoch": 3.9154052734375e-05,
      "step": 41056,
      "training_step_time": 6.750231027603149
    },
    {
      "epoch": 3.915500640869141e-05,
      "model_forward_time": 1.7727479934692383,
      "step": 41057
    },
    {
      "epoch": 3.915500640869141e-05,
      "step": 41057,
      "training_step_time": 6.756347417831421
    },
    {
      "epoch": 3.915596008300781e-05,
      "model_forward_time": 1.7682721614837646,
      "step": 41058
    },
    {
      "epoch": 3.915596008300781e-05,
      "step": 41058,
      "training_step_time": 6.766859769821167
    },
    {
      "epoch": 3.9156913757324216e-05,
      "model_forward_time": 1.758237361907959,
      "step": 41059
    },
    {
      "epoch": 3.9156913757324216e-05,
      "step": 41059,
      "training_step_time": 6.758045196533203
    },
    {
      "epoch": 3.915786743164063e-05,
      "grad_norm": 0.1502600610256195,
      "learning_rate": 6.846924234831069e-06,
      "loss": 0.0604,
      "step": 41060
    },
    {
      "epoch": 3.915786743164063e-05,
      "model_forward_time": 1.7633771896362305,
      "step": 41060
    },
    {
      "dynamics_loss_avg": 0.05209682434797287,
      "epoch": 3.915786743164063e-05,
      "step": 41060
    },
    {
      "action_loss_avg": 0.0026555864489637314,
      "epoch": 3.915786743164063e-05,
      "step": 41060
    },
    {
      "epoch": 3.915786743164063e-05,
      "step": 41060,
      "training_step_time": 6.81846022605896
    },
    {
      "epoch": 3.915882110595703e-05,
      "model_forward_time": 1.7603952884674072,
      "step": 41061
    },
    {
      "epoch": 3.915882110595703e-05,
      "step": 41061,
      "training_step_time": 7.1492486000061035
    },
    {
      "epoch": 3.9159774780273435e-05,
      "model_forward_time": 2.8748104572296143,
      "step": 41062
    },
    {
      "epoch": 3.9159774780273435e-05,
      "step": 41062,
      "training_step_time": 9.512930154800415
    },
    {
      "epoch": 3.9160728454589846e-05,
      "model_forward_time": 1.7537012100219727,
      "step": 41063
    },
    {
      "epoch": 3.9160728454589846e-05,
      "step": 41063,
      "training_step_time": 7.653066873550415
    },
    {
      "epoch": 3.916168212890625e-05,
      "model_forward_time": 1.757096767425537,
      "step": 41064
    },
    {
      "epoch": 3.916168212890625e-05,
      "step": 41064,
      "training_step_time": 6.802164077758789
    },
    {
      "epoch": 3.9162635803222654e-05,
      "model_forward_time": 1.7653734683990479,
      "step": 41065
    },
    {
      "epoch": 3.9162635803222654e-05,
      "step": 41065,
      "training_step_time": 6.815715074539185
    },
    {
      "epoch": 3.9163589477539065e-05,
      "model_forward_time": 1.765876054763794,
      "step": 41066
    },
    {
      "epoch": 3.9163589477539065e-05,
      "step": 41066,
      "training_step_time": 6.776352882385254
    },
    {
      "epoch": 3.916454315185547e-05,
      "model_forward_time": 1.76472806930542,
      "step": 41067
    },
    {
      "epoch": 3.916454315185547e-05,
      "step": 41067,
      "training_step_time": 6.778085708618164
    },
    {
      "epoch": 3.916549682617187e-05,
      "model_forward_time": 1.7611491680145264,
      "step": 41068
    },
    {
      "epoch": 3.916549682617187e-05,
      "step": 41068,
      "training_step_time": 6.815623998641968
    },
    {
      "epoch": 3.9166450500488284e-05,
      "model_forward_time": 1.7596125602722168,
      "step": 41069
    },
    {
      "epoch": 3.9166450500488284e-05,
      "step": 41069,
      "training_step_time": 6.810994625091553
    },
    {
      "epoch": 3.916740417480469e-05,
      "grad_norm": 0.12436375766992569,
      "learning_rate": 6.845387603206597e-06,
      "loss": 0.0568,
      "step": 41070
    },
    {
      "epoch": 3.916740417480469e-05,
      "model_forward_time": 1.7595722675323486,
      "step": 41070
    },
    {
      "dynamics_loss_avg": 0.06491180043667555,
      "epoch": 3.916740417480469e-05,
      "step": 41070
    },
    {
      "action_loss_avg": 0.0036424234043806792,
      "epoch": 3.916740417480469e-05,
      "step": 41070
    },
    {
      "epoch": 3.916740417480469e-05,
      "step": 41070,
      "training_step_time": 7.283947229385376
    },
    {
      "epoch": 3.916835784912109e-05,
      "model_forward_time": 2.8892695903778076,
      "step": 41071
    },
    {
      "epoch": 3.916835784912109e-05,
      "step": 41071,
      "training_step_time": 10.029199600219727
    },
    {
      "epoch": 3.91693115234375e-05,
      "model_forward_time": 1.7553818225860596,
      "step": 41072
    },
    {
      "epoch": 3.91693115234375e-05,
      "step": 41072,
      "training_step_time": 7.703116416931152
    },
    {
      "epoch": 3.917026519775391e-05,
      "model_forward_time": 1.7710223197937012,
      "step": 41073
    },
    {
      "epoch": 3.917026519775391e-05,
      "step": 41073,
      "training_step_time": 6.793971538543701
    },
    {
      "epoch": 3.917121887207031e-05,
      "model_forward_time": 1.747976303100586,
      "step": 41074
    },
    {
      "epoch": 3.917121887207031e-05,
      "step": 41074,
      "training_step_time": 6.742018699645996
    },
    {
      "epoch": 3.917217254638672e-05,
      "model_forward_time": 1.7619132995605469,
      "step": 41075
    },
    {
      "epoch": 3.917217254638672e-05,
      "step": 41075,
      "training_step_time": 6.758999347686768
    },
    {
      "epoch": 3.9173126220703125e-05,
      "model_forward_time": 1.7623674869537354,
      "step": 41076
    },
    {
      "epoch": 3.9173126220703125e-05,
      "step": 41076,
      "training_step_time": 6.796957969665527
    },
    {
      "epoch": 3.917407989501953e-05,
      "model_forward_time": 1.7560782432556152,
      "step": 41077
    },
    {
      "epoch": 3.917407989501953e-05,
      "step": 41077,
      "training_step_time": 6.795191049575806
    },
    {
      "epoch": 3.917503356933594e-05,
      "model_forward_time": 1.758561134338379,
      "step": 41078
    },
    {
      "epoch": 3.917503356933594e-05,
      "step": 41078,
      "training_step_time": 6.780318021774292
    },
    {
      "epoch": 3.9175987243652344e-05,
      "model_forward_time": 1.418372631072998,
      "step": 41079
    },
    {
      "epoch": 3.9175987243652344e-05,
      "step": 41079,
      "training_step_time": 6.818182468414307
    },
    {
      "epoch": 3.917694091796875e-05,
      "grad_norm": 0.14283764362335205,
      "learning_rate": 6.8438507697733126e-06,
      "loss": 0.0584,
      "step": 41080
    },
    {
      "epoch": 3.917694091796875e-05,
      "model_forward_time": 2.9145002365112305,
      "step": 41080
    },
    {
      "dynamics_loss_avg": 0.07037303205579519,
      "epoch": 3.917694091796875e-05,
      "step": 41080
    },
    {
      "action_loss_avg": 0.005290296021848917,
      "epoch": 3.917694091796875e-05,
      "step": 41080
    },
    {
      "epoch": 3.917694091796875e-05,
      "step": 41080,
      "training_step_time": 9.02377724647522
    },
    {
      "epoch": 3.917789459228516e-05,
      "model_forward_time": 1.761331558227539,
      "step": 41081
    },
    {
      "epoch": 3.917789459228516e-05,
      "step": 41081,
      "training_step_time": 8.668890476226807
    },
    {
      "epoch": 3.917884826660156e-05,
      "model_forward_time": 1.766181230545044,
      "step": 41082
    },
    {
      "epoch": 3.917884826660156e-05,
      "step": 41082,
      "training_step_time": 7.413026571273804
    },
    {
      "epoch": 3.917980194091797e-05,
      "model_forward_time": 1.7590725421905518,
      "step": 41083
    },
    {
      "epoch": 3.917980194091797e-05,
      "step": 41083,
      "training_step_time": 6.773683786392212
    },
    {
      "epoch": 3.918075561523438e-05,
      "model_forward_time": 1.7548549175262451,
      "step": 41084
    },
    {
      "epoch": 3.918075561523438e-05,
      "step": 41084,
      "training_step_time": 6.742523908615112
    },
    {
      "epoch": 3.918170928955078e-05,
      "model_forward_time": 1.764521598815918,
      "step": 41085
    },
    {
      "epoch": 3.918170928955078e-05,
      "step": 41085,
      "training_step_time": 6.780372619628906
    },
    {
      "epoch": 3.9182662963867186e-05,
      "model_forward_time": 1.7703642845153809,
      "step": 41086
    },
    {
      "epoch": 3.9182662963867186e-05,
      "step": 41086,
      "training_step_time": 6.810548543930054
    },
    {
      "epoch": 3.91836166381836e-05,
      "model_forward_time": 1.7627794742584229,
      "step": 41087
    },
    {
      "epoch": 3.91836166381836e-05,
      "step": 41087,
      "training_step_time": 6.822512865066528
    },
    {
      "epoch": 3.91845703125e-05,
      "model_forward_time": 1.7647385597229004,
      "step": 41088
    },
    {
      "epoch": 3.91845703125e-05,
      "step": 41088,
      "training_step_time": 6.80463719367981
    },
    {
      "epoch": 3.9185523986816405e-05,
      "model_forward_time": 2.9249513149261475,
      "step": 41089
    },
    {
      "epoch": 3.9185523986816405e-05,
      "step": 41089,
      "training_step_time": 11.115833282470703
    },
    {
      "epoch": 3.9186477661132816e-05,
      "grad_norm": 0.1676337718963623,
      "learning_rate": 6.842313734699279e-06,
      "loss": 0.0612,
      "step": 41090
    },
    {
      "epoch": 3.9186477661132816e-05,
      "model_forward_time": 1.7682957649230957,
      "step": 41090
    },
    {
      "dynamics_loss_avg": 0.06421465575695037,
      "epoch": 3.9186477661132816e-05,
      "step": 41090
    },
    {
      "action_loss_avg": 0.0037827147520147265,
      "epoch": 3.9186477661132816e-05,
      "step": 41090
    },
    {
      "epoch": 3.9186477661132816e-05,
      "step": 41090,
      "training_step_time": 7.92428994178772
    },
    {
      "epoch": 3.918743133544922e-05,
      "model_forward_time": 1.764777421951294,
      "step": 41091
    },
    {
      "epoch": 3.918743133544922e-05,
      "step": 41091,
      "training_step_time": 7.408866882324219
    },
    {
      "epoch": 3.9188385009765624e-05,
      "model_forward_time": 1.7573292255401611,
      "step": 41092
    },
    {
      "epoch": 3.9188385009765624e-05,
      "step": 41092,
      "training_step_time": 6.7994701862335205
    },
    {
      "epoch": 3.9189338684082035e-05,
      "model_forward_time": 1.765542984008789,
      "step": 41093
    },
    {
      "epoch": 3.9189338684082035e-05,
      "step": 41093,
      "training_step_time": 6.77710223197937
    },
    {
      "epoch": 3.919029235839844e-05,
      "model_forward_time": 1.7688407897949219,
      "step": 41094
    },
    {
      "epoch": 3.919029235839844e-05,
      "step": 41094,
      "training_step_time": 6.781038284301758
    },
    {
      "epoch": 3.919124603271484e-05,
      "model_forward_time": 1.7723820209503174,
      "step": 41095
    },
    {
      "epoch": 3.919124603271484e-05,
      "step": 41095,
      "training_step_time": 6.787148952484131
    },
    {
      "epoch": 3.919219970703125e-05,
      "model_forward_time": 1.7646844387054443,
      "step": 41096
    },
    {
      "epoch": 3.919219970703125e-05,
      "step": 41096,
      "training_step_time": 6.766438007354736
    },
    {
      "epoch": 3.919315338134766e-05,
      "model_forward_time": 1.7592742443084717,
      "step": 41097
    },
    {
      "epoch": 3.919315338134766e-05,
      "step": 41097,
      "training_step_time": 6.7818543910980225
    },
    {
      "epoch": 3.919410705566406e-05,
      "model_forward_time": 1.811896562576294,
      "step": 41098
    },
    {
      "epoch": 3.919410705566406e-05,
      "step": 41098,
      "training_step_time": 10.101498365402222
    },
    {
      "epoch": 3.9195060729980466e-05,
      "model_forward_time": 1.7544240951538086,
      "step": 41099
    },
    {
      "epoch": 3.9195060729980466e-05,
      "step": 41099,
      "training_step_time": 9.858701705932617
    },
    {
      "epoch": 3.9196014404296876e-05,
      "grad_norm": 0.17656153440475464,
      "learning_rate": 6.8407764981525835e-06,
      "loss": 0.062,
      "step": 41100
    },
    {
      "epoch": 3.9196014404296876e-05,
      "model_forward_time": 1.7654998302459717,
      "step": 41100
    },
    {
      "dynamics_loss_avg": 0.05700651984661818,
      "epoch": 3.9196014404296876e-05,
      "step": 41100
    },
    {
      "action_loss_avg": 0.00295153844053857,
      "epoch": 3.9196014404296876e-05,
      "step": 41100
    },
    {
      "epoch": 3.9196014404296876e-05,
      "step": 41100,
      "training_step_time": 7.662896394729614
    },
    {
      "epoch": 3.919696807861328e-05,
      "model_forward_time": 1.7606515884399414,
      "step": 41101
    },
    {
      "epoch": 3.919696807861328e-05,
      "step": 41101,
      "training_step_time": 6.7922492027282715
    },
    {
      "epoch": 3.9197921752929684e-05,
      "model_forward_time": 1.756415843963623,
      "step": 41102
    },
    {
      "epoch": 3.9197921752929684e-05,
      "step": 41102,
      "training_step_time": 6.667776823043823
    },
    {
      "epoch": 3.9198875427246095e-05,
      "model_forward_time": 1.7519025802612305,
      "step": 41103
    },
    {
      "epoch": 3.9198875427246095e-05,
      "step": 41103,
      "training_step_time": 6.747970342636108
    },
    {
      "epoch": 3.91998291015625e-05,
      "model_forward_time": 1.7628405094146729,
      "step": 41104
    },
    {
      "epoch": 3.91998291015625e-05,
      "step": 41104,
      "training_step_time": 6.747343301773071
    },
    {
      "epoch": 3.92007827758789e-05,
      "model_forward_time": 1.406811237335205,
      "step": 41105
    },
    {
      "epoch": 3.92007827758789e-05,
      "step": 41105,
      "training_step_time": 6.793541431427002
    },
    {
      "epoch": 3.9201736450195314e-05,
      "model_forward_time": 1.748556137084961,
      "step": 41106
    },
    {
      "epoch": 3.9201736450195314e-05,
      "step": 41106,
      "training_step_time": 6.826179265975952
    },
    {
      "epoch": 3.920269012451172e-05,
      "model_forward_time": 1.7805976867675781,
      "step": 41107
    },
    {
      "epoch": 3.920269012451172e-05,
      "step": 41107,
      "training_step_time": 8.519000053405762
    },
    {
      "epoch": 3.920364379882812e-05,
      "model_forward_time": 2.9452831745147705,
      "step": 41108
    },
    {
      "epoch": 3.920364379882812e-05,
      "step": 41108,
      "training_step_time": 9.834446668624878
    },
    {
      "epoch": 3.920459747314453e-05,
      "model_forward_time": 1.7637741565704346,
      "step": 41109
    },
    {
      "epoch": 3.920459747314453e-05,
      "step": 41109,
      "training_step_time": 7.687732934951782
    },
    {
      "epoch": 3.920555114746094e-05,
      "grad_norm": 0.16337676346302032,
      "learning_rate": 6.83923906030134e-06,
      "loss": 0.0611,
      "step": 41110
    },
    {
      "epoch": 3.920555114746094e-05,
      "model_forward_time": 1.7643613815307617,
      "step": 41110
    },
    {
      "dynamics_loss_avg": 0.07399968653917313,
      "epoch": 3.920555114746094e-05,
      "step": 41110
    },
    {
      "action_loss_avg": 0.002789310924708843,
      "epoch": 3.920555114746094e-05,
      "step": 41110
    },
    {
      "epoch": 3.920555114746094e-05,
      "step": 41110,
      "training_step_time": 6.7896459102630615
    },
    {
      "epoch": 3.920650482177734e-05,
      "model_forward_time": 1.7573237419128418,
      "step": 41111
    },
    {
      "epoch": 3.920650482177734e-05,
      "step": 41111,
      "training_step_time": 6.757976293563843
    },
    {
      "epoch": 3.920745849609375e-05,
      "model_forward_time": 1.7694616317749023,
      "step": 41112
    },
    {
      "epoch": 3.920745849609375e-05,
      "step": 41112,
      "training_step_time": 6.764047622680664
    },
    {
      "epoch": 3.9208412170410156e-05,
      "model_forward_time": 1.7672557830810547,
      "step": 41113
    },
    {
      "epoch": 3.9208412170410156e-05,
      "step": 41113,
      "training_step_time": 6.75701904296875
    },
    {
      "epoch": 3.920936584472656e-05,
      "model_forward_time": 1.7621066570281982,
      "step": 41114
    },
    {
      "epoch": 3.920936584472656e-05,
      "step": 41114,
      "training_step_time": 6.804442405700684
    },
    {
      "epoch": 3.921031951904297e-05,
      "model_forward_time": 1.7644963264465332,
      "step": 41115
    },
    {
      "epoch": 3.921031951904297e-05,
      "step": 41115,
      "training_step_time": 6.798896551132202
    },
    {
      "epoch": 3.9211273193359375e-05,
      "model_forward_time": 1.7680995464324951,
      "step": 41116
    },
    {
      "epoch": 3.9211273193359375e-05,
      "step": 41116,
      "training_step_time": 7.629830598831177
    },
    {
      "epoch": 3.921222686767578e-05,
      "model_forward_time": 2.9301271438598633,
      "step": 41117
    },
    {
      "epoch": 3.921222686767578e-05,
      "step": 41117,
      "training_step_time": 10.589318752288818
    },
    {
      "epoch": 3.921318054199219e-05,
      "model_forward_time": 1.7680153846740723,
      "step": 41118
    },
    {
      "epoch": 3.921318054199219e-05,
      "step": 41118,
      "training_step_time": 7.702837705612183
    },
    {
      "epoch": 3.9214134216308594e-05,
      "model_forward_time": 1.7622859477996826,
      "step": 41119
    },
    {
      "epoch": 3.9214134216308594e-05,
      "step": 41119,
      "training_step_time": 6.991811275482178
    },
    {
      "epoch": 3.9215087890625e-05,
      "grad_norm": 0.14076554775238037,
      "learning_rate": 6.837701421313677e-06,
      "loss": 0.0588,
      "step": 41120
    },
    {
      "epoch": 3.9215087890625e-05,
      "model_forward_time": 1.7621607780456543,
      "step": 41120
    },
    {
      "dynamics_loss_avg": 0.07994761969894171,
      "epoch": 3.9215087890625e-05,
      "step": 41120
    },
    {
      "action_loss_avg": 0.0036692472160211763,
      "epoch": 3.9215087890625e-05,
      "step": 41120
    },
    {
      "epoch": 3.9215087890625e-05,
      "step": 41120,
      "training_step_time": 6.793436765670776
    },
    {
      "epoch": 3.921604156494141e-05,
      "model_forward_time": 1.747133493423462,
      "step": 41121
    },
    {
      "epoch": 3.921604156494141e-05,
      "step": 41121,
      "training_step_time": 6.7286460399627686
    },
    {
      "epoch": 3.921699523925781e-05,
      "model_forward_time": 1.7785751819610596,
      "step": 41122
    },
    {
      "epoch": 3.921699523925781e-05,
      "step": 41122,
      "training_step_time": 6.764369249343872
    },
    {
      "epoch": 3.9217948913574216e-05,
      "model_forward_time": 1.3975863456726074,
      "step": 41123
    },
    {
      "epoch": 3.9217948913574216e-05,
      "step": 41123,
      "training_step_time": 6.777016878128052
    },
    {
      "epoch": 3.921890258789063e-05,
      "model_forward_time": 1.7694628238677979,
      "step": 41124
    },
    {
      "epoch": 3.921890258789063e-05,
      "step": 41124,
      "training_step_time": 6.740110158920288
    },
    {
      "epoch": 3.921985626220703e-05,
      "model_forward_time": 1.7747597694396973,
      "step": 41125
    },
    {
      "epoch": 3.921985626220703e-05,
      "step": 41125,
      "training_step_time": 7.519198179244995
    },
    {
      "epoch": 3.9220809936523435e-05,
      "model_forward_time": 2.9830322265625,
      "step": 41126
    },
    {
      "epoch": 3.9220809936523435e-05,
      "step": 41126,
      "training_step_time": 9.551390647888184
    },
    {
      "epoch": 3.9221763610839846e-05,
      "model_forward_time": 1.756028652191162,
      "step": 41127
    },
    {
      "epoch": 3.9221763610839846e-05,
      "step": 41127,
      "training_step_time": 7.8323962688446045
    },
    {
      "epoch": 3.922271728515625e-05,
      "model_forward_time": 1.765850305557251,
      "step": 41128
    },
    {
      "epoch": 3.922271728515625e-05,
      "step": 41128,
      "training_step_time": 7.591683864593506
    },
    {
      "epoch": 3.9223670959472654e-05,
      "model_forward_time": 1.7715394496917725,
      "step": 41129
    },
    {
      "epoch": 3.9223670959472654e-05,
      "step": 41129,
      "training_step_time": 6.7530999183654785
    },
    {
      "epoch": 3.9224624633789065e-05,
      "grad_norm": 0.1277678906917572,
      "learning_rate": 6.836163581357751e-06,
      "loss": 0.0627,
      "step": 41130
    },
    {
      "epoch": 3.9224624633789065e-05,
      "model_forward_time": 1.7607040405273438,
      "step": 41130
    },
    {
      "dynamics_loss_avg": 0.061051379144191745,
      "epoch": 3.9224624633789065e-05,
      "step": 41130
    },
    {
      "action_loss_avg": 0.002005169645417482,
      "epoch": 3.9224624633789065e-05,
      "step": 41130
    },
    {
      "epoch": 3.9224624633789065e-05,
      "step": 41130,
      "training_step_time": 6.771357297897339
    },
    {
      "epoch": 3.922557830810547e-05,
      "model_forward_time": 1.7636020183563232,
      "step": 41131
    },
    {
      "epoch": 3.922557830810547e-05,
      "step": 41131,
      "training_step_time": 6.805511951446533
    },
    {
      "epoch": 3.922653198242187e-05,
      "model_forward_time": 1.7616064548492432,
      "step": 41132
    },
    {
      "epoch": 3.922653198242187e-05,
      "step": 41132,
      "training_step_time": 6.743871212005615
    },
    {
      "epoch": 3.9227485656738284e-05,
      "model_forward_time": 1.752202033996582,
      "step": 41133
    },
    {
      "epoch": 3.9227485656738284e-05,
      "step": 41133,
      "training_step_time": 6.783293724060059
    },
    {
      "epoch": 3.922843933105469e-05,
      "model_forward_time": 1.7629001140594482,
      "step": 41134
    },
    {
      "epoch": 3.922843933105469e-05,
      "step": 41134,
      "training_step_time": 7.539872407913208
    },
    {
      "epoch": 3.922939300537109e-05,
      "model_forward_time": 1.4650776386260986,
      "step": 41135
    },
    {
      "epoch": 3.922939300537109e-05,
      "step": 41135,
      "training_step_time": 9.322424411773682
    },
    {
      "epoch": 3.92303466796875e-05,
      "model_forward_time": 2.9543488025665283,
      "step": 41136
    },
    {
      "epoch": 3.92303466796875e-05,
      "step": 41136,
      "training_step_time": 9.793057680130005
    },
    {
      "epoch": 3.923130035400391e-05,
      "model_forward_time": 1.765866756439209,
      "step": 41137
    },
    {
      "epoch": 3.923130035400391e-05,
      "step": 41137,
      "training_step_time": 7.705711364746094
    },
    {
      "epoch": 3.923225402832031e-05,
      "model_forward_time": 1.7518863677978516,
      "step": 41138
    },
    {
      "epoch": 3.923225402832031e-05,
      "step": 41138,
      "training_step_time": 6.731885671615601
    },
    {
      "epoch": 3.923320770263672e-05,
      "model_forward_time": 1.7519967555999756,
      "step": 41139
    },
    {
      "epoch": 3.923320770263672e-05,
      "step": 41139,
      "training_step_time": 6.793242931365967
    },
    {
      "epoch": 3.9234161376953126e-05,
      "grad_norm": 0.13864241540431976,
      "learning_rate": 6.834625540601734e-06,
      "loss": 0.0588,
      "step": 41140
    },
    {
      "epoch": 3.9234161376953126e-05,
      "model_forward_time": 1.7616372108459473,
      "step": 41140
    },
    {
      "dynamics_loss_avg": 0.06106856279075146,
      "epoch": 3.9234161376953126e-05,
      "step": 41140
    },
    {
      "action_loss_avg": 0.0024176430597435683,
      "epoch": 3.9234161376953126e-05,
      "step": 41140
    },
    {
      "epoch": 3.9234161376953126e-05,
      "step": 41140,
      "training_step_time": 6.769547939300537
    },
    {
      "epoch": 3.923511505126953e-05,
      "model_forward_time": 1.7568650245666504,
      "step": 41141
    },
    {
      "epoch": 3.923511505126953e-05,
      "step": 41141,
      "training_step_time": 6.785824298858643
    },
    {
      "epoch": 3.923606872558594e-05,
      "model_forward_time": 1.766566276550293,
      "step": 41142
    },
    {
      "epoch": 3.923606872558594e-05,
      "step": 41142,
      "training_step_time": 6.775931358337402
    },
    {
      "epoch": 3.9237022399902344e-05,
      "model_forward_time": 1.3955419063568115,
      "step": 41143
    },
    {
      "epoch": 3.9237022399902344e-05,
      "step": 41143,
      "training_step_time": 8.462958097457886
    },
    {
      "epoch": 3.923797607421875e-05,
      "model_forward_time": 1.8146870136260986,
      "step": 41144
    },
    {
      "epoch": 3.923797607421875e-05,
      "step": 41144,
      "training_step_time": 10.218283891677856
    },
    {
      "epoch": 3.923892974853516e-05,
      "model_forward_time": 2.9277141094207764,
      "step": 41145
    },
    {
      "epoch": 3.923892974853516e-05,
      "step": 41145,
      "training_step_time": 9.35468578338623
    },
    {
      "epoch": 3.923988342285156e-05,
      "model_forward_time": 1.7664895057678223,
      "step": 41146
    },
    {
      "epoch": 3.923988342285156e-05,
      "step": 41146,
      "training_step_time": 7.678797483444214
    },
    {
      "epoch": 3.924083709716797e-05,
      "model_forward_time": 1.7704639434814453,
      "step": 41147
    },
    {
      "epoch": 3.924083709716797e-05,
      "step": 41147,
      "training_step_time": 6.75010085105896
    },
    {
      "epoch": 3.924179077148438e-05,
      "model_forward_time": 1.758866310119629,
      "step": 41148
    },
    {
      "epoch": 3.924179077148438e-05,
      "step": 41148,
      "training_step_time": 6.766249179840088
    },
    {
      "epoch": 3.924274444580078e-05,
      "model_forward_time": 1.748652458190918,
      "step": 41149
    },
    {
      "epoch": 3.924274444580078e-05,
      "step": 41149,
      "training_step_time": 6.762131214141846
    },
    {
      "epoch": 3.9243698120117186e-05,
      "grad_norm": 0.10059385746717453,
      "learning_rate": 6.833087299213829e-06,
      "loss": 0.0577,
      "step": 41150
    },
    {
      "epoch": 3.9243698120117186e-05,
      "model_forward_time": 1.7533433437347412,
      "step": 41150
    },
    {
      "dynamics_loss_avg": 0.07028688723221421,
      "epoch": 3.9243698120117186e-05,
      "step": 41150
    },
    {
      "action_loss_avg": 0.0034507480449974537,
      "epoch": 3.9243698120117186e-05,
      "step": 41150
    },
    {
      "epoch": 3.9243698120117186e-05,
      "step": 41150,
      "training_step_time": 6.771500825881958
    },
    {
      "epoch": 3.92446517944336e-05,
      "model_forward_time": 1.7689175605773926,
      "step": 41151
    },
    {
      "epoch": 3.92446517944336e-05,
      "step": 41151,
      "training_step_time": 6.701562881469727
    },
    {
      "epoch": 3.924560546875e-05,
      "model_forward_time": 1.7649693489074707,
      "step": 41152
    },
    {
      "epoch": 3.924560546875e-05,
      "step": 41152,
      "training_step_time": 7.502567768096924
    },
    {
      "epoch": 3.9246559143066405e-05,
      "model_forward_time": 1.7760546207427979,
      "step": 41153
    },
    {
      "epoch": 3.9246559143066405e-05,
      "step": 41153,
      "training_step_time": 8.24821949005127
    },
    {
      "epoch": 3.9247512817382816e-05,
      "model_forward_time": 2.9117536544799805,
      "step": 41154
    },
    {
      "epoch": 3.9247512817382816e-05,
      "step": 41154,
      "training_step_time": 8.498610019683838
    },
    {
      "epoch": 3.924846649169922e-05,
      "model_forward_time": 1.7690691947937012,
      "step": 41155
    },
    {
      "epoch": 3.924846649169922e-05,
      "step": 41155,
      "training_step_time": 8.578649997711182
    },
    {
      "epoch": 3.9249420166015624e-05,
      "model_forward_time": 1.7396876811981201,
      "step": 41156
    },
    {
      "epoch": 3.9249420166015624e-05,
      "step": 41156,
      "training_step_time": 7.800801038742065
    },
    {
      "epoch": 3.925037384033203e-05,
      "model_forward_time": 1.742591381072998,
      "step": 41157
    },
    {
      "epoch": 3.925037384033203e-05,
      "step": 41157,
      "training_step_time": 6.973056077957153
    },
    {
      "epoch": 3.925132751464844e-05,
      "model_forward_time": 1.4027090072631836,
      "step": 41158
    },
    {
      "epoch": 3.925132751464844e-05,
      "step": 41158,
      "training_step_time": 6.933374881744385
    },
    {
      "epoch": 3.925228118896484e-05,
      "model_forward_time": 1.7582345008850098,
      "step": 41159
    },
    {
      "epoch": 3.925228118896484e-05,
      "step": 41159,
      "training_step_time": 6.915444850921631
    },
    {
      "epoch": 3.925323486328125e-05,
      "grad_norm": 0.10689213126897812,
      "learning_rate": 6.83154885736225e-06,
      "loss": 0.0552,
      "step": 41160
    },
    {
      "epoch": 3.925323486328125e-05,
      "model_forward_time": 1.757335901260376,
      "step": 41160
    },
    {
      "dynamics_loss_avg": 0.05961034949868917,
      "epoch": 3.925323486328125e-05,
      "step": 41160
    },
    {
      "action_loss_avg": 0.0020321539603173734,
      "epoch": 3.925323486328125e-05,
      "step": 41160
    },
    {
      "epoch": 3.925323486328125e-05,
      "step": 41160,
      "training_step_time": 7.000654458999634
    },
    {
      "epoch": 3.925418853759766e-05,
      "model_forward_time": 1.757890224456787,
      "step": 41161
    },
    {
      "epoch": 3.925418853759766e-05,
      "step": 41161,
      "training_step_time": 7.710666179656982
    },
    {
      "epoch": 3.925514221191406e-05,
      "model_forward_time": 1.7508864402770996,
      "step": 41162
    },
    {
      "epoch": 3.925514221191406e-05,
      "step": 41162,
      "training_step_time": 8.343983888626099
    },
    {
      "epoch": 3.9256095886230466e-05,
      "model_forward_time": 2.844132423400879,
      "step": 41163
    },
    {
      "epoch": 3.9256095886230466e-05,
      "step": 41163,
      "training_step_time": 9.803607940673828
    },
    {
      "epoch": 3.9257049560546876e-05,
      "model_forward_time": 1.7538466453552246,
      "step": 41164
    },
    {
      "epoch": 3.9257049560546876e-05,
      "step": 41164,
      "training_step_time": 7.840741157531738
    },
    {
      "epoch": 3.925800323486328e-05,
      "model_forward_time": 1.7669713497161865,
      "step": 41165
    },
    {
      "epoch": 3.925800323486328e-05,
      "step": 41165,
      "training_step_time": 7.946310043334961
    },
    {
      "epoch": 3.9258956909179684e-05,
      "model_forward_time": 1.7576103210449219,
      "step": 41166
    },
    {
      "epoch": 3.9258956909179684e-05,
      "step": 41166,
      "training_step_time": 6.97541880607605
    },
    {
      "epoch": 3.9259910583496095e-05,
      "model_forward_time": 1.7666079998016357,
      "step": 41167
    },
    {
      "epoch": 3.9259910583496095e-05,
      "step": 41167,
      "training_step_time": 6.978707551956177
    },
    {
      "epoch": 3.92608642578125e-05,
      "model_forward_time": 1.7538039684295654,
      "step": 41168
    },
    {
      "epoch": 3.92608642578125e-05,
      "step": 41168,
      "training_step_time": 6.991293907165527
    },
    {
      "epoch": 3.92618179321289e-05,
      "model_forward_time": 1.763746738433838,
      "step": 41169
    },
    {
      "epoch": 3.92618179321289e-05,
      "step": 41169,
      "training_step_time": 6.94981575012207
    },
    {
      "epoch": 3.9262771606445314e-05,
      "grad_norm": 0.12672528624534607,
      "learning_rate": 6.830010215215244e-06,
      "loss": 0.0611,
      "step": 41170
    },
    {
      "epoch": 3.9262771606445314e-05,
      "model_forward_time": 1.752577304840088,
      "step": 41170
    },
    {
      "dynamics_loss_avg": 0.07009613178670407,
      "epoch": 3.9262771606445314e-05,
      "step": 41170
    },
    {
      "action_loss_avg": 0.0029155433992855253,
      "epoch": 3.9262771606445314e-05,
      "step": 41170
    },
    {
      "epoch": 3.9262771606445314e-05,
      "step": 41170,
      "training_step_time": 7.706708669662476
    },
    {
      "epoch": 3.926372528076172e-05,
      "model_forward_time": 1.7536680698394775,
      "step": 41171
    },
    {
      "epoch": 3.926372528076172e-05,
      "step": 41171,
      "training_step_time": 7.843511581420898
    },
    {
      "epoch": 3.926467895507812e-05,
      "model_forward_time": 2.8925089836120605,
      "step": 41172
    },
    {
      "epoch": 3.926467895507812e-05,
      "step": 41172,
      "training_step_time": 10.443328380584717
    },
    {
      "epoch": 3.926563262939453e-05,
      "model_forward_time": 1.7528913021087646,
      "step": 41173
    },
    {
      "epoch": 3.926563262939453e-05,
      "step": 41173,
      "training_step_time": 8.560976028442383
    },
    {
      "epoch": 3.926658630371094e-05,
      "model_forward_time": 1.741410732269287,
      "step": 41174
    },
    {
      "epoch": 3.926658630371094e-05,
      "step": 41174,
      "training_step_time": 7.844680070877075
    },
    {
      "epoch": 3.926753997802734e-05,
      "model_forward_time": 1.751575231552124,
      "step": 41175
    },
    {
      "epoch": 3.926753997802734e-05,
      "step": 41175,
      "training_step_time": 6.969557523727417
    },
    {
      "epoch": 3.926849365234375e-05,
      "model_forward_time": 1.7576687335968018,
      "step": 41176
    },
    {
      "epoch": 3.926849365234375e-05,
      "step": 41176,
      "training_step_time": 6.879921913146973
    },
    {
      "epoch": 3.9269447326660156e-05,
      "model_forward_time": 1.7537717819213867,
      "step": 41177
    },
    {
      "epoch": 3.9269447326660156e-05,
      "step": 41177,
      "training_step_time": 6.918578386306763
    },
    {
      "epoch": 3.927040100097656e-05,
      "model_forward_time": 1.7584264278411865,
      "step": 41178
    },
    {
      "epoch": 3.927040100097656e-05,
      "step": 41178,
      "training_step_time": 6.912832736968994
    },
    {
      "epoch": 3.927135467529297e-05,
      "model_forward_time": 1.7835547924041748,
      "step": 41179
    },
    {
      "epoch": 3.927135467529297e-05,
      "step": 41179,
      "training_step_time": 8.650930166244507
    },
    {
      "epoch": 3.9272308349609375e-05,
      "grad_norm": 0.17590098083019257,
      "learning_rate": 6.828471372941072e-06,
      "loss": 0.06,
      "step": 41180
    },
    {
      "epoch": 3.9272308349609375e-05,
      "model_forward_time": 1.7449166774749756,
      "step": 41180
    },
    {
      "dynamics_loss_avg": 0.059669270552694796,
      "epoch": 3.9272308349609375e-05,
      "step": 41180
    },
    {
      "action_loss_avg": 0.004591804230585694,
      "epoch": 3.9272308349609375e-05,
      "step": 41180
    },
    {
      "epoch": 3.9272308349609375e-05,
      "step": 41180,
      "training_step_time": 8.914562225341797
    },
    {
      "epoch": 3.927326202392578e-05,
      "model_forward_time": 1.818185567855835,
      "step": 41181
    },
    {
      "epoch": 3.927326202392578e-05,
      "step": 41181,
      "training_step_time": 9.447253942489624
    },
    {
      "epoch": 3.927421569824219e-05,
      "model_forward_time": 2.886723518371582,
      "step": 41182
    },
    {
      "epoch": 3.927421569824219e-05,
      "step": 41182,
      "training_step_time": 8.734763860702515
    },
    {
      "epoch": 3.9275169372558594e-05,
      "model_forward_time": 1.754011869430542,
      "step": 41183
    },
    {
      "epoch": 3.9275169372558594e-05,
      "step": 41183,
      "training_step_time": 8.484971523284912
    },
    {
      "epoch": 3.9276123046875e-05,
      "model_forward_time": 1.7479844093322754,
      "step": 41184
    },
    {
      "epoch": 3.9276123046875e-05,
      "step": 41184,
      "training_step_time": 7.828472137451172
    },
    {
      "epoch": 3.927707672119141e-05,
      "model_forward_time": 1.7802481651306152,
      "step": 41185
    },
    {
      "epoch": 3.927707672119141e-05,
      "step": 41185,
      "training_step_time": 6.99634861946106
    },
    {
      "epoch": 3.927803039550781e-05,
      "model_forward_time": 1.753340244293213,
      "step": 41186
    },
    {
      "epoch": 3.927803039550781e-05,
      "step": 41186,
      "training_step_time": 6.980019807815552
    },
    {
      "epoch": 3.9278984069824216e-05,
      "model_forward_time": 1.7430269718170166,
      "step": 41187
    },
    {
      "epoch": 3.9278984069824216e-05,
      "step": 41187,
      "training_step_time": 6.9101033210754395
    },
    {
      "epoch": 3.927993774414063e-05,
      "model_forward_time": 1.7496047019958496,
      "step": 41188
    },
    {
      "epoch": 3.927993774414063e-05,
      "step": 41188,
      "training_step_time": 7.7303853034973145
    },
    {
      "epoch": 3.928089141845703e-05,
      "model_forward_time": 1.7490947246551514,
      "step": 41189
    },
    {
      "epoch": 3.928089141845703e-05,
      "step": 41189,
      "training_step_time": 7.678420305252075
    },
    {
      "epoch": 3.9281845092773435e-05,
      "grad_norm": 0.16147063672542572,
      "learning_rate": 6.826932330708018e-06,
      "loss": 0.0618,
      "step": 41190
    },
    {
      "epoch": 3.9281845092773435e-05,
      "model_forward_time": 1.8033504486083984,
      "step": 41190
    },
    {
      "dynamics_loss_avg": 0.06904012206941842,
      "epoch": 3.9281845092773435e-05,
      "step": 41190
    },
    {
      "action_loss_avg": 0.0023530076345195995,
      "epoch": 3.9281845092773435e-05,
      "step": 41190
    },
    {
      "epoch": 3.9281845092773435e-05,
      "step": 41190,
      "training_step_time": 8.898264408111572
    },
    {
      "epoch": 3.9282798767089846e-05,
      "model_forward_time": 2.923813581466675,
      "step": 41191
    },
    {
      "epoch": 3.9282798767089846e-05,
      "step": 41191,
      "training_step_time": 8.517143726348877
    },
    {
      "epoch": 3.928375244140625e-05,
      "model_forward_time": 1.7549922466278076,
      "step": 41192
    },
    {
      "epoch": 3.928375244140625e-05,
      "step": 41192,
      "training_step_time": 9.384159088134766
    },
    {
      "epoch": 3.9284706115722654e-05,
      "model_forward_time": 1.7529699802398682,
      "step": 41193
    },
    {
      "epoch": 3.9284706115722654e-05,
      "step": 41193,
      "training_step_time": 7.695814371109009
    },
    {
      "epoch": 3.9285659790039065e-05,
      "model_forward_time": 1.7436256408691406,
      "step": 41194
    },
    {
      "epoch": 3.9285659790039065e-05,
      "step": 41194,
      "training_step_time": 6.987157583236694
    },
    {
      "epoch": 3.928661346435547e-05,
      "model_forward_time": 1.7456130981445312,
      "step": 41195
    },
    {
      "epoch": 3.928661346435547e-05,
      "step": 41195,
      "training_step_time": 6.96158242225647
    },
    {
      "epoch": 3.928756713867187e-05,
      "model_forward_time": 1.754652500152588,
      "step": 41196
    },
    {
      "epoch": 3.928756713867187e-05,
      "step": 41196,
      "training_step_time": 6.9005043506622314
    },
    {
      "epoch": 3.9288520812988284e-05,
      "model_forward_time": 1.758540391921997,
      "step": 41197
    },
    {
      "epoch": 3.9288520812988284e-05,
      "step": 41197,
      "training_step_time": 7.561467885971069
    },
    {
      "epoch": 3.928947448730469e-05,
      "model_forward_time": 1.7471890449523926,
      "step": 41198
    },
    {
      "epoch": 3.928947448730469e-05,
      "step": 41198,
      "training_step_time": 7.77186393737793
    },
    {
      "epoch": 3.929042816162109e-05,
      "model_forward_time": 1.7818729877471924,
      "step": 41199
    },
    {
      "epoch": 3.929042816162109e-05,
      "step": 41199,
      "training_step_time": 8.922100067138672
    },
    {
      "epoch": 3.92913818359375e-05,
      "grad_norm": 0.17104117572307587,
      "learning_rate": 6.8253930886843935e-06,
      "loss": 0.0574,
      "step": 41200
    },
    {
      "epoch": 3.92913818359375e-05,
      "model_forward_time": 2.8753342628479004,
      "step": 41200
    },
    {
      "dynamics_loss_avg": 0.0698806680738926,
      "epoch": 3.92913818359375e-05,
      "step": 41200
    },
    {
      "action_loss_avg": 0.002579773572506383,
      "epoch": 3.92913818359375e-05,
      "step": 41200
    },
    {
      "epoch": 3.92913818359375e-05,
      "step": 41200,
      "training_step_time": 8.874886512756348
    },
    {
      "epoch": 3.929233551025391e-05,
      "model_forward_time": 1.7515923976898193,
      "step": 41201
    },
    {
      "epoch": 3.929233551025391e-05,
      "step": 41201,
      "training_step_time": 7.9452714920043945
    },
    {
      "epoch": 3.929328918457031e-05,
      "model_forward_time": 1.7730507850646973,
      "step": 41202
    },
    {
      "epoch": 3.929328918457031e-05,
      "step": 41202,
      "training_step_time": 7.6853086948394775
    },
    {
      "epoch": 3.929424285888672e-05,
      "model_forward_time": 1.7518208026885986,
      "step": 41203
    },
    {
      "epoch": 3.929424285888672e-05,
      "step": 41203,
      "training_step_time": 6.973743915557861
    },
    {
      "epoch": 3.9295196533203126e-05,
      "model_forward_time": 1.7648603916168213,
      "step": 41204
    },
    {
      "epoch": 3.9295196533203126e-05,
      "step": 41204,
      "training_step_time": 6.966796159744263
    },
    {
      "epoch": 3.929615020751953e-05,
      "model_forward_time": 1.7555222511291504,
      "step": 41205
    },
    {
      "epoch": 3.929615020751953e-05,
      "step": 41205,
      "training_step_time": 6.988096237182617
    },
    {
      "epoch": 3.929710388183594e-05,
      "model_forward_time": 1.7630650997161865,
      "step": 41206
    },
    {
      "epoch": 3.929710388183594e-05,
      "step": 41206,
      "training_step_time": 7.286078214645386
    },
    {
      "epoch": 3.9298057556152344e-05,
      "model_forward_time": 1.7680842876434326,
      "step": 41207
    },
    {
      "epoch": 3.9298057556152344e-05,
      "step": 41207,
      "training_step_time": 8.54309892654419
    },
    {
      "epoch": 3.929901123046875e-05,
      "model_forward_time": 1.3979742527008057,
      "step": 41208
    },
    {
      "epoch": 3.929901123046875e-05,
      "step": 41208,
      "training_step_time": 8.147831201553345
    },
    {
      "epoch": 3.929996490478516e-05,
      "model_forward_time": 2.889209508895874,
      "step": 41209
    },
    {
      "epoch": 3.929996490478516e-05,
      "step": 41209,
      "training_step_time": 9.644800424575806
    },
    {
      "epoch": 3.930091857910156e-05,
      "grad_norm": 0.14466971158981323,
      "learning_rate": 6.823853647038524e-06,
      "loss": 0.0599,
      "step": 41210
    },
    {
      "epoch": 3.930091857910156e-05,
      "model_forward_time": 1.7540221214294434,
      "step": 41210
    },
    {
      "dynamics_loss_avg": 0.07044481802731753,
      "epoch": 3.930091857910156e-05,
      "step": 41210
    },
    {
      "action_loss_avg": 0.0035495653515681624,
      "epoch": 3.930091857910156e-05,
      "step": 41210
    },
    {
      "epoch": 3.930091857910156e-05,
      "step": 41210,
      "training_step_time": 7.903720140457153
    },
    {
      "epoch": 3.930187225341797e-05,
      "model_forward_time": 1.7448110580444336,
      "step": 41211
    },
    {
      "epoch": 3.930187225341797e-05,
      "step": 41211,
      "training_step_time": 7.778396368026733
    },
    {
      "epoch": 3.930282592773438e-05,
      "model_forward_time": 1.756786584854126,
      "step": 41212
    },
    {
      "epoch": 3.930282592773438e-05,
      "step": 41212,
      "training_step_time": 6.956221342086792
    },
    {
      "epoch": 3.930377960205078e-05,
      "model_forward_time": 1.7617735862731934,
      "step": 41213
    },
    {
      "epoch": 3.930377960205078e-05,
      "step": 41213,
      "training_step_time": 6.899648666381836
    },
    {
      "epoch": 3.9304733276367186e-05,
      "model_forward_time": 1.768773078918457,
      "step": 41214
    },
    {
      "epoch": 3.9304733276367186e-05,
      "step": 41214,
      "training_step_time": 6.911595582962036
    },
    {
      "epoch": 3.93056869506836e-05,
      "model_forward_time": 1.7556815147399902,
      "step": 41215
    },
    {
      "epoch": 3.93056869506836e-05,
      "step": 41215,
      "training_step_time": 7.217648029327393
    },
    {
      "epoch": 3.9306640625e-05,
      "model_forward_time": 1.7579967975616455,
      "step": 41216
    },
    {
      "epoch": 3.9306640625e-05,
      "step": 41216,
      "training_step_time": 7.721272945404053
    },
    {
      "epoch": 3.9307594299316405e-05,
      "model_forward_time": 1.7536873817443848,
      "step": 41217
    },
    {
      "epoch": 3.9307594299316405e-05,
      "step": 41217,
      "training_step_time": 7.8308751583099365
    },
    {
      "epoch": 3.9308547973632816e-05,
      "model_forward_time": 1.8253004550933838,
      "step": 41218
    },
    {
      "epoch": 3.9308547973632816e-05,
      "step": 41218,
      "training_step_time": 8.536489009857178
    },
    {
      "epoch": 3.930950164794922e-05,
      "model_forward_time": 2.938108205795288,
      "step": 41219
    },
    {
      "epoch": 3.930950164794922e-05,
      "step": 41219,
      "training_step_time": 9.405033349990845
    },
    {
      "epoch": 3.9310455322265624e-05,
      "grad_norm": 0.11473092436790466,
      "learning_rate": 6.822314005938761e-06,
      "loss": 0.0633,
      "step": 41220
    },
    {
      "epoch": 3.9310455322265624e-05,
      "model_forward_time": 1.7633368968963623,
      "step": 41220
    },
    {
      "dynamics_loss_avg": 0.06293600518256426,
      "epoch": 3.9310455322265624e-05,
      "step": 41220
    },
    {
      "action_loss_avg": 0.0022746083093807103,
      "epoch": 3.9310455322265624e-05,
      "step": 41220
    },
    {
      "epoch": 3.9310455322265624e-05,
      "step": 41220,
      "training_step_time": 7.894213676452637
    },
    {
      "epoch": 3.931140899658203e-05,
      "model_forward_time": 1.7617640495300293,
      "step": 41221
    },
    {
      "epoch": 3.931140899658203e-05,
      "step": 41221,
      "training_step_time": 7.790789604187012
    },
    {
      "epoch": 3.931236267089844e-05,
      "model_forward_time": 1.7621097564697266,
      "step": 41222
    },
    {
      "epoch": 3.931236267089844e-05,
      "step": 41222,
      "training_step_time": 6.896308898925781
    },
    {
      "epoch": 3.931331634521484e-05,
      "model_forward_time": 1.755868911743164,
      "step": 41223
    },
    {
      "epoch": 3.931331634521484e-05,
      "step": 41223,
      "training_step_time": 6.914827823638916
    },
    {
      "epoch": 3.931427001953125e-05,
      "model_forward_time": 1.7628962993621826,
      "step": 41224
    },
    {
      "epoch": 3.931427001953125e-05,
      "step": 41224,
      "training_step_time": 6.9194416999816895
    },
    {
      "epoch": 3.931522369384766e-05,
      "model_forward_time": 1.7557342052459717,
      "step": 41225
    },
    {
      "epoch": 3.931522369384766e-05,
      "step": 41225,
      "training_step_time": 7.774578094482422
    },
    {
      "epoch": 3.931617736816406e-05,
      "model_forward_time": 1.7497711181640625,
      "step": 41226
    },
    {
      "epoch": 3.931617736816406e-05,
      "step": 41226,
      "training_step_time": 8.333449602127075
    },
    {
      "epoch": 3.9317131042480466e-05,
      "model_forward_time": 1.8310885429382324,
      "step": 41227
    },
    {
      "epoch": 3.9317131042480466e-05,
      "step": 41227,
      "training_step_time": 10.814360618591309
    },
    {
      "epoch": 3.9318084716796876e-05,
      "model_forward_time": 2.955594062805176,
      "step": 41228
    },
    {
      "epoch": 3.9318084716796876e-05,
      "step": 41228,
      "training_step_time": 7.902238607406616
    },
    {
      "epoch": 3.931903839111328e-05,
      "model_forward_time": 1.766667127609253,
      "step": 41229
    },
    {
      "epoch": 3.931903839111328e-05,
      "step": 41229,
      "training_step_time": 8.958748579025269
    },
    {
      "epoch": 3.9319992065429685e-05,
      "grad_norm": 0.1553988754749298,
      "learning_rate": 6.820774165553479e-06,
      "loss": 0.0615,
      "step": 41230
    },
    {
      "epoch": 3.9319992065429685e-05,
      "model_forward_time": 1.7616138458251953,
      "step": 41230
    },
    {
      "dynamics_loss_avg": 0.044039137475192544,
      "epoch": 3.9319992065429685e-05,
      "step": 41230
    },
    {
      "action_loss_avg": 0.001427802475518547,
      "epoch": 3.9319992065429685e-05,
      "step": 41230
    },
    {
      "epoch": 3.9319992065429685e-05,
      "step": 41230,
      "training_step_time": 7.478739976882935
    },
    {
      "epoch": 3.9320945739746095e-05,
      "model_forward_time": 1.7521123886108398,
      "step": 41231
    },
    {
      "epoch": 3.9320945739746095e-05,
      "step": 41231,
      "training_step_time": 6.951949596405029
    },
    {
      "epoch": 3.93218994140625e-05,
      "model_forward_time": 1.7431869506835938,
      "step": 41232
    },
    {
      "epoch": 3.93218994140625e-05,
      "step": 41232,
      "training_step_time": 6.887784242630005
    },
    {
      "epoch": 3.93228530883789e-05,
      "model_forward_time": 1.7440252304077148,
      "step": 41233
    },
    {
      "epoch": 3.93228530883789e-05,
      "step": 41233,
      "training_step_time": 6.925402641296387
    },
    {
      "epoch": 3.9323806762695314e-05,
      "model_forward_time": 1.748507022857666,
      "step": 41234
    },
    {
      "epoch": 3.9323806762695314e-05,
      "step": 41234,
      "training_step_time": 7.23778772354126
    },
    {
      "epoch": 3.932476043701172e-05,
      "model_forward_time": 1.7476370334625244,
      "step": 41235
    },
    {
      "epoch": 3.932476043701172e-05,
      "step": 41235,
      "training_step_time": 9.015043497085571
    },
    {
      "epoch": 3.932571411132812e-05,
      "model_forward_time": 1.8091411590576172,
      "step": 41236
    },
    {
      "epoch": 3.932571411132812e-05,
      "step": 41236,
      "training_step_time": 7.875146389007568
    },
    {
      "epoch": 3.932666778564453e-05,
      "model_forward_time": 2.915308713912964,
      "step": 41237
    },
    {
      "epoch": 3.932666778564453e-05,
      "step": 41237,
      "training_step_time": 7.788433790206909
    },
    {
      "epoch": 3.932762145996094e-05,
      "model_forward_time": 1.7429041862487793,
      "step": 41238
    },
    {
      "epoch": 3.932762145996094e-05,
      "step": 41238,
      "training_step_time": 7.718961477279663
    },
    {
      "epoch": 3.932857513427734e-05,
      "model_forward_time": 1.746142864227295,
      "step": 41239
    },
    {
      "epoch": 3.932857513427734e-05,
      "step": 41239,
      "training_step_time": 7.821751832962036
    },
    {
      "epoch": 3.932952880859375e-05,
      "grad_norm": 0.16323687136173248,
      "learning_rate": 6.819234126051072e-06,
      "loss": 0.0575,
      "step": 41240
    },
    {
      "epoch": 3.932952880859375e-05,
      "model_forward_time": 1.7538113594055176,
      "step": 41240
    },
    {
      "dynamics_loss_avg": 0.062199671380221845,
      "epoch": 3.932952880859375e-05,
      "step": 41240
    },
    {
      "action_loss_avg": 0.0037932782841380687,
      "epoch": 3.932952880859375e-05,
      "step": 41240
    },
    {
      "epoch": 3.932952880859375e-05,
      "step": 41240,
      "training_step_time": 7.025620937347412
    },
    {
      "epoch": 3.9330482482910156e-05,
      "model_forward_time": 1.7361021041870117,
      "step": 41241
    },
    {
      "epoch": 3.9330482482910156e-05,
      "step": 41241,
      "training_step_time": 6.9620068073272705
    },
    {
      "epoch": 3.933143615722656e-05,
      "model_forward_time": 1.7353856563568115,
      "step": 41242
    },
    {
      "epoch": 3.933143615722656e-05,
      "step": 41242,
      "training_step_time": 6.962807655334473
    },
    {
      "epoch": 3.933238983154297e-05,
      "model_forward_time": 1.7349035739898682,
      "step": 41243
    },
    {
      "epoch": 3.933238983154297e-05,
      "step": 41243,
      "training_step_time": 7.352035760879517
    },
    {
      "epoch": 3.9333343505859375e-05,
      "model_forward_time": 1.3684320449829102,
      "step": 41244
    },
    {
      "epoch": 3.9333343505859375e-05,
      "step": 41244,
      "training_step_time": 8.924187898635864
    },
    {
      "epoch": 3.933429718017578e-05,
      "model_forward_time": 1.7566759586334229,
      "step": 41245
    },
    {
      "epoch": 3.933429718017578e-05,
      "step": 41245,
      "training_step_time": 8.897136211395264
    },
    {
      "epoch": 3.933525085449219e-05,
      "model_forward_time": 2.842177152633667,
      "step": 41246
    },
    {
      "epoch": 3.933525085449219e-05,
      "step": 41246,
      "training_step_time": 9.257586002349854
    },
    {
      "epoch": 3.9336204528808594e-05,
      "model_forward_time": 1.738011121749878,
      "step": 41247
    },
    {
      "epoch": 3.9336204528808594e-05,
      "step": 41247,
      "training_step_time": 8.407310009002686
    },
    {
      "epoch": 3.9337158203125e-05,
      "model_forward_time": 1.7334444522857666,
      "step": 41248
    },
    {
      "epoch": 3.9337158203125e-05,
      "step": 41248,
      "training_step_time": 7.613900184631348
    },
    {
      "epoch": 3.933811187744141e-05,
      "model_forward_time": 1.7326431274414062,
      "step": 41249
    },
    {
      "epoch": 3.933811187744141e-05,
      "step": 41249,
      "training_step_time": 7.032991170883179
    },
    {
      "epoch": 3.933906555175781e-05,
      "grad_norm": 0.16223368048667908,
      "learning_rate": 6.817693887599956e-06,
      "loss": 0.0594,
      "step": 41250
    },
    {
      "epoch": 3.933906555175781e-05,
      "model_forward_time": 1.7357494831085205,
      "step": 41250
    },
    {
      "dynamics_loss_avg": 0.06969247665256262,
      "epoch": 3.933906555175781e-05,
      "step": 41250
    },
    {
      "action_loss_avg": 0.002775592065881938,
      "epoch": 3.933906555175781e-05,
      "step": 41250
    },
    {
      "epoch": 3.933906555175781e-05,
      "step": 41250,
      "training_step_time": 6.733521223068237
    },
    {
      "epoch": 3.9340019226074217e-05,
      "model_forward_time": 1.7436275482177734,
      "step": 41251
    },
    {
      "epoch": 3.9340019226074217e-05,
      "step": 41251,
      "training_step_time": 6.67414116859436
    },
    {
      "epoch": 3.934097290039063e-05,
      "model_forward_time": 1.7359583377838135,
      "step": 41252
    },
    {
      "epoch": 3.934097290039063e-05,
      "step": 41252,
      "training_step_time": 7.025840759277344
    },
    {
      "epoch": 3.934192657470703e-05,
      "model_forward_time": 1.741523265838623,
      "step": 41253
    },
    {
      "epoch": 3.934192657470703e-05,
      "step": 41253,
      "training_step_time": 7.491726636886597
    },
    {
      "epoch": 3.9342880249023435e-05,
      "model_forward_time": 1.7360539436340332,
      "step": 41254
    },
    {
      "epoch": 3.9342880249023435e-05,
      "step": 41254,
      "training_step_time": 8.645453691482544
    },
    {
      "epoch": 3.9343833923339846e-05,
      "model_forward_time": 1.7561290264129639,
      "step": 41255
    },
    {
      "epoch": 3.9343833923339846e-05,
      "step": 41255,
      "training_step_time": 9.86426329612732
    },
    {
      "epoch": 3.934478759765625e-05,
      "model_forward_time": 1.7358510494232178,
      "step": 41256
    },
    {
      "epoch": 3.934478759765625e-05,
      "step": 41256,
      "training_step_time": 8.273681163787842
    },
    {
      "epoch": 3.9345741271972654e-05,
      "model_forward_time": 1.7314732074737549,
      "step": 41257
    },
    {
      "epoch": 3.9345741271972654e-05,
      "step": 41257,
      "training_step_time": 8.680764436721802
    },
    {
      "epoch": 3.9346694946289065e-05,
      "model_forward_time": 1.730565071105957,
      "step": 41258
    },
    {
      "epoch": 3.9346694946289065e-05,
      "step": 41258,
      "training_step_time": 7.056546449661255
    },
    {
      "epoch": 3.934764862060547e-05,
      "model_forward_time": 1.7284965515136719,
      "step": 41259
    },
    {
      "epoch": 3.934764862060547e-05,
      "step": 41259,
      "training_step_time": 6.802020788192749
    },
    {
      "epoch": 3.934860229492187e-05,
      "grad_norm": 0.11608771234750748,
      "learning_rate": 6.816153450368569e-06,
      "loss": 0.0576,
      "step": 41260
    },
    {
      "epoch": 3.934860229492187e-05,
      "model_forward_time": 1.7341132164001465,
      "step": 41260
    },
    {
      "dynamics_loss_avg": 0.07044341899454594,
      "epoch": 3.934860229492187e-05,
      "step": 41260
    },
    {
      "action_loss_avg": 0.0017792226397432386,
      "epoch": 3.934860229492187e-05,
      "step": 41260
    },
    {
      "epoch": 3.934860229492187e-05,
      "step": 41260,
      "training_step_time": 6.775387287139893
    },
    {
      "epoch": 3.9349555969238284e-05,
      "model_forward_time": 1.7354283332824707,
      "step": 41261
    },
    {
      "epoch": 3.9349555969238284e-05,
      "step": 41261,
      "training_step_time": 7.0919764041900635
    },
    {
      "epoch": 3.935050964355469e-05,
      "model_forward_time": 1.7246744632720947,
      "step": 41262
    },
    {
      "epoch": 3.935050964355469e-05,
      "step": 41262,
      "training_step_time": 7.536253452301025
    },
    {
      "epoch": 3.935146331787109e-05,
      "model_forward_time": 1.731156826019287,
      "step": 41263
    },
    {
      "epoch": 3.935146331787109e-05,
      "step": 41263,
      "training_step_time": 8.629862308502197
    },
    {
      "epoch": 3.93524169921875e-05,
      "model_forward_time": 1.8215477466583252,
      "step": 41264
    },
    {
      "epoch": 3.93524169921875e-05,
      "step": 41264,
      "training_step_time": 9.35024094581604
    },
    {
      "epoch": 3.935337066650391e-05,
      "model_forward_time": 2.9092113971710205,
      "step": 41265
    },
    {
      "epoch": 3.935337066650391e-05,
      "step": 41265,
      "training_step_time": 8.688759565353394
    },
    {
      "epoch": 3.935432434082031e-05,
      "model_forward_time": 1.7350101470947266,
      "step": 41266
    },
    {
      "epoch": 3.935432434082031e-05,
      "step": 41266,
      "training_step_time": 8.378937005996704
    },
    {
      "epoch": 3.935527801513672e-05,
      "model_forward_time": 1.7402434349060059,
      "step": 41267
    },
    {
      "epoch": 3.935527801513672e-05,
      "step": 41267,
      "training_step_time": 7.594743967056274
    },
    {
      "epoch": 3.9356231689453126e-05,
      "model_forward_time": 1.7346701622009277,
      "step": 41268
    },
    {
      "epoch": 3.9356231689453126e-05,
      "step": 41268,
      "training_step_time": 6.7308995723724365
    },
    {
      "epoch": 3.935718536376953e-05,
      "model_forward_time": 1.7443499565124512,
      "step": 41269
    },
    {
      "epoch": 3.935718536376953e-05,
      "step": 41269,
      "training_step_time": 6.750983238220215
    },
    {
      "epoch": 3.935813903808594e-05,
      "grad_norm": 0.12103862315416336,
      "learning_rate": 6.814612814525372e-06,
      "loss": 0.0576,
      "step": 41270
    },
    {
      "epoch": 3.935813903808594e-05,
      "model_forward_time": 1.7511348724365234,
      "step": 41270
    },
    {
      "dynamics_loss_avg": 0.05112654650583863,
      "epoch": 3.935813903808594e-05,
      "step": 41270
    },
    {
      "action_loss_avg": 0.0017126470847870223,
      "epoch": 3.935813903808594e-05,
      "step": 41270
    },
    {
      "epoch": 3.935813903808594e-05,
      "step": 41270,
      "training_step_time": 7.31682276725769
    },
    {
      "epoch": 3.9359092712402345e-05,
      "model_forward_time": 1.7467386722564697,
      "step": 41271
    },
    {
      "epoch": 3.9359092712402345e-05,
      "step": 41271,
      "training_step_time": 7.575017690658569
    },
    {
      "epoch": 3.936004638671875e-05,
      "model_forward_time": 1.752413034439087,
      "step": 41272
    },
    {
      "epoch": 3.936004638671875e-05,
      "step": 41272,
      "training_step_time": 7.722987174987793
    },
    {
      "epoch": 3.936100006103516e-05,
      "model_forward_time": 1.8082327842712402,
      "step": 41273
    },
    {
      "epoch": 3.936100006103516e-05,
      "step": 41273,
      "training_step_time": 9.56187629699707
    },
    {
      "epoch": 3.936195373535156e-05,
      "model_forward_time": 2.917905330657959,
      "step": 41274
    },
    {
      "epoch": 3.936195373535156e-05,
      "step": 41274,
      "training_step_time": 8.343279123306274
    },
    {
      "epoch": 3.936290740966797e-05,
      "model_forward_time": 1.7398550510406494,
      "step": 41275
    },
    {
      "epoch": 3.936290740966797e-05,
      "step": 41275,
      "training_step_time": 7.691913366317749
    },
    {
      "epoch": 3.936386108398438e-05,
      "model_forward_time": 1.7361621856689453,
      "step": 41276
    },
    {
      "epoch": 3.936386108398438e-05,
      "step": 41276,
      "training_step_time": 7.736119031906128
    },
    {
      "epoch": 3.936481475830078e-05,
      "model_forward_time": 1.750586748123169,
      "step": 41277
    },
    {
      "epoch": 3.936481475830078e-05,
      "step": 41277,
      "training_step_time": 6.773370265960693
    },
    {
      "epoch": 3.9365768432617186e-05,
      "model_forward_time": 1.3819286823272705,
      "step": 41278
    },
    {
      "epoch": 3.9365768432617186e-05,
      "step": 41278,
      "training_step_time": 6.7199928760528564
    },
    {
      "epoch": 3.93667221069336e-05,
      "model_forward_time": 1.7497389316558838,
      "step": 41279
    },
    {
      "epoch": 3.93667221069336e-05,
      "step": 41279,
      "training_step_time": 6.6680684089660645
    },
    {
      "epoch": 3.936767578125e-05,
      "grad_norm": 0.1380445957183838,
      "learning_rate": 6.813071980238846e-06,
      "loss": 0.0605,
      "step": 41280
    },
    {
      "epoch": 3.936767578125e-05,
      "model_forward_time": 1.7472758293151855,
      "step": 41280
    },
    {
      "dynamics_loss_avg": 0.0631424318999052,
      "epoch": 3.936767578125e-05,
      "step": 41280
    },
    {
      "action_loss_avg": 0.001948353269835934,
      "epoch": 3.936767578125e-05,
      "step": 41280
    },
    {
      "epoch": 3.936767578125e-05,
      "step": 41280,
      "training_step_time": 7.366425275802612
    },
    {
      "epoch": 3.9368629455566405e-05,
      "model_forward_time": 1.7440240383148193,
      "step": 41281
    },
    {
      "epoch": 3.9368629455566405e-05,
      "step": 41281,
      "training_step_time": 7.576218366622925
    },
    {
      "epoch": 3.936958312988281e-05,
      "model_forward_time": 1.8014540672302246,
      "step": 41282
    },
    {
      "epoch": 3.936958312988281e-05,
      "step": 41282,
      "training_step_time": 8.500266075134277
    },
    {
      "epoch": 3.937053680419922e-05,
      "model_forward_time": 2.858320713043213,
      "step": 41283
    },
    {
      "epoch": 3.937053680419922e-05,
      "step": 41283,
      "training_step_time": 9.41010046005249
    },
    {
      "epoch": 3.9371490478515624e-05,
      "model_forward_time": 1.7394912242889404,
      "step": 41284
    },
    {
      "epoch": 3.9371490478515624e-05,
      "step": 41284,
      "training_step_time": 7.718841552734375
    },
    {
      "epoch": 3.937244415283203e-05,
      "model_forward_time": 1.742199182510376,
      "step": 41285
    },
    {
      "epoch": 3.937244415283203e-05,
      "step": 41285,
      "training_step_time": 7.849183082580566
    },
    {
      "epoch": 3.937339782714844e-05,
      "model_forward_time": 1.745635986328125,
      "step": 41286
    },
    {
      "epoch": 3.937339782714844e-05,
      "step": 41286,
      "training_step_time": 7.200307369232178
    },
    {
      "epoch": 3.937435150146484e-05,
      "model_forward_time": 1.740513801574707,
      "step": 41287
    },
    {
      "epoch": 3.937435150146484e-05,
      "step": 41287,
      "training_step_time": 6.718426704406738
    },
    {
      "epoch": 3.937530517578125e-05,
      "model_forward_time": 1.7449018955230713,
      "step": 41288
    },
    {
      "epoch": 3.937530517578125e-05,
      "step": 41288,
      "training_step_time": 6.8966286182403564
    },
    {
      "epoch": 3.937625885009766e-05,
      "model_forward_time": 1.7381305694580078,
      "step": 41289
    },
    {
      "epoch": 3.937625885009766e-05,
      "step": 41289,
      "training_step_time": 7.522858619689941
    },
    {
      "epoch": 3.937721252441406e-05,
      "grad_norm": 0.13600146770477295,
      "learning_rate": 6.811530947677491e-06,
      "loss": 0.0589,
      "step": 41290
    },
    {
      "epoch": 3.937721252441406e-05,
      "model_forward_time": 1.754753828048706,
      "step": 41290
    },
    {
      "dynamics_loss_avg": 0.060957851633429525,
      "epoch": 3.937721252441406e-05,
      "step": 41290
    },
    {
      "action_loss_avg": 0.002364747691899538,
      "epoch": 3.937721252441406e-05,
      "step": 41290
    },
    {
      "epoch": 3.937721252441406e-05,
      "step": 41290,
      "training_step_time": 7.792478799819946
    },
    {
      "epoch": 3.9378166198730466e-05,
      "model_forward_time": 1.8057174682617188,
      "step": 41291
    },
    {
      "epoch": 3.9378166198730466e-05,
      "step": 41291,
      "training_step_time": 8.211943864822388
    },
    {
      "epoch": 3.9379119873046877e-05,
      "model_forward_time": 2.8616230487823486,
      "step": 41292
    },
    {
      "epoch": 3.9379119873046877e-05,
      "step": 41292,
      "training_step_time": 8.380422115325928
    },
    {
      "epoch": 3.938007354736328e-05,
      "model_forward_time": 1.7456324100494385,
      "step": 41293
    },
    {
      "epoch": 3.938007354736328e-05,
      "step": 41293,
      "training_step_time": 7.642443656921387
    },
    {
      "epoch": 3.9381027221679685e-05,
      "model_forward_time": 1.7385780811309814,
      "step": 41294
    },
    {
      "epoch": 3.9381027221679685e-05,
      "step": 41294,
      "training_step_time": 7.670156002044678
    },
    {
      "epoch": 3.9381980895996095e-05,
      "model_forward_time": 1.7582573890686035,
      "step": 41295
    },
    {
      "epoch": 3.9381980895996095e-05,
      "step": 41295,
      "training_step_time": 7.300904273986816
    },
    {
      "epoch": 3.93829345703125e-05,
      "model_forward_time": 1.7261254787445068,
      "step": 41296
    },
    {
      "epoch": 3.93829345703125e-05,
      "step": 41296,
      "training_step_time": 6.792688608169556
    },
    {
      "epoch": 3.9383888244628903e-05,
      "model_forward_time": 1.7392873764038086,
      "step": 41297
    },
    {
      "epoch": 3.9383888244628903e-05,
      "step": 41297,
      "training_step_time": 6.795873641967773
    },
    {
      "epoch": 3.9384841918945314e-05,
      "model_forward_time": 1.7354419231414795,
      "step": 41298
    },
    {
      "epoch": 3.9384841918945314e-05,
      "step": 41298,
      "training_step_time": 7.309750080108643
    },
    {
      "epoch": 3.938579559326172e-05,
      "model_forward_time": 1.7320361137390137,
      "step": 41299
    },
    {
      "epoch": 3.938579559326172e-05,
      "step": 41299,
      "training_step_time": 8.47033429145813
    },
    {
      "epoch": 3.938674926757812e-05,
      "grad_norm": 0.13241581618785858,
      "learning_rate": 6.809989717009838e-06,
      "loss": 0.0589,
      "step": 41300
    },
    {
      "epoch": 3.938674926757812e-05,
      "model_forward_time": 1.800142765045166,
      "step": 41300
    },
    {
      "dynamics_loss_avg": 0.06967778913676739,
      "epoch": 3.938674926757812e-05,
      "step": 41300
    },
    {
      "action_loss_avg": 0.0051021943683736025,
      "epoch": 3.938674926757812e-05,
      "step": 41300
    },
    {
      "epoch": 3.938674926757812e-05,
      "step": 41300,
      "training_step_time": 8.542824506759644
    },
    {
      "epoch": 3.938770294189453e-05,
      "model_forward_time": 1.7368860244750977,
      "step": 41301
    },
    {
      "epoch": 3.938770294189453e-05,
      "step": 41301,
      "training_step_time": 9.306976079940796
    },
    {
      "epoch": 3.938865661621094e-05,
      "model_forward_time": 1.7389616966247559,
      "step": 41302
    },
    {
      "epoch": 3.938865661621094e-05,
      "step": 41302,
      "training_step_time": 8.13402795791626
    },
    {
      "epoch": 3.938961029052734e-05,
      "model_forward_time": 1.7447400093078613,
      "step": 41303
    },
    {
      "epoch": 3.938961029052734e-05,
      "step": 41303,
      "training_step_time": 7.8380677700042725
    },
    {
      "epoch": 3.939056396484375e-05,
      "model_forward_time": 1.75567626953125,
      "step": 41304
    },
    {
      "epoch": 3.939056396484375e-05,
      "step": 41304,
      "training_step_time": 7.677831649780273
    },
    {
      "epoch": 3.9391517639160156e-05,
      "model_forward_time": 1.733311653137207,
      "step": 41305
    },
    {
      "epoch": 3.9391517639160156e-05,
      "step": 41305,
      "training_step_time": 7.0072338581085205
    },
    {
      "epoch": 3.939247131347656e-05,
      "model_forward_time": 1.7510993480682373,
      "step": 41306
    },
    {
      "epoch": 3.939247131347656e-05,
      "step": 41306,
      "training_step_time": 6.981873273849487
    },
    {
      "epoch": 3.939342498779297e-05,
      "model_forward_time": 1.7434275150299072,
      "step": 41307
    },
    {
      "epoch": 3.939342498779297e-05,
      "step": 41307,
      "training_step_time": 7.554387331008911
    },
    {
      "epoch": 3.9394378662109375e-05,
      "model_forward_time": 1.7376396656036377,
      "step": 41308
    },
    {
      "epoch": 3.9394378662109375e-05,
      "step": 41308,
      "training_step_time": 7.576592445373535
    },
    {
      "epoch": 3.939533233642578e-05,
      "model_forward_time": 1.814605474472046,
      "step": 41309
    },
    {
      "epoch": 3.939533233642578e-05,
      "step": 41309,
      "training_step_time": 8.742254257202148
    },
    {
      "epoch": 3.939628601074219e-05,
      "grad_norm": 0.16754773259162903,
      "learning_rate": 6.808448288404431e-06,
      "loss": 0.0603,
      "step": 41310
    },
    {
      "epoch": 3.939628601074219e-05,
      "model_forward_time": 1.7424063682556152,
      "step": 41310
    },
    {
      "dynamics_loss_avg": 0.0598229655995965,
      "epoch": 3.939628601074219e-05,
      "step": 41310
    },
    {
      "action_loss_avg": 0.0017245866591110825,
      "epoch": 3.939628601074219e-05,
      "step": 41310
    },
    {
      "epoch": 3.939628601074219e-05,
      "step": 41310,
      "training_step_time": 9.10451364517212
    },
    {
      "epoch": 3.9397239685058594e-05,
      "model_forward_time": 2.911374092102051,
      "step": 41311
    },
    {
      "epoch": 3.9397239685058594e-05,
      "step": 41311,
      "training_step_time": 9.341831684112549
    },
    {
      "epoch": 3.9398193359375e-05,
      "model_forward_time": 1.7425673007965088,
      "step": 41312
    },
    {
      "epoch": 3.9398193359375e-05,
      "step": 41312,
      "training_step_time": 8.689249038696289
    },
    {
      "epoch": 3.939914703369141e-05,
      "model_forward_time": 1.7454688549041748,
      "step": 41313
    },
    {
      "epoch": 3.939914703369141e-05,
      "step": 41313,
      "training_step_time": 7.5991833209991455
    },
    {
      "epoch": 3.940010070800781e-05,
      "model_forward_time": 1.7345452308654785,
      "step": 41314
    },
    {
      "epoch": 3.940010070800781e-05,
      "step": 41314,
      "training_step_time": 6.999463081359863
    },
    {
      "epoch": 3.9401054382324217e-05,
      "model_forward_time": 1.7386367321014404,
      "step": 41315
    },
    {
      "epoch": 3.9401054382324217e-05,
      "step": 41315,
      "training_step_time": 6.814074516296387
    },
    {
      "epoch": 3.940200805664063e-05,
      "model_forward_time": 1.7344651222229004,
      "step": 41316
    },
    {
      "epoch": 3.940200805664063e-05,
      "step": 41316,
      "training_step_time": 7.324015855789185
    },
    {
      "epoch": 3.940296173095703e-05,
      "model_forward_time": 1.7329909801483154,
      "step": 41317
    },
    {
      "epoch": 3.940296173095703e-05,
      "step": 41317,
      "training_step_time": 7.489246368408203
    },
    {
      "epoch": 3.9403915405273435e-05,
      "model_forward_time": 1.8088152408599854,
      "step": 41318
    },
    {
      "epoch": 3.9403915405273435e-05,
      "step": 41318,
      "training_step_time": 7.521070718765259
    },
    {
      "epoch": 3.9404869079589846e-05,
      "model_forward_time": 1.737962245941162,
      "step": 41319
    },
    {
      "epoch": 3.9404869079589846e-05,
      "step": 41319,
      "training_step_time": 10.496535301208496
    },
    {
      "epoch": 3.940582275390625e-05,
      "grad_norm": 0.1797417849302292,
      "learning_rate": 6.806906662029837e-06,
      "loss": 0.0611,
      "step": 41320
    },
    {
      "epoch": 3.940582275390625e-05,
      "model_forward_time": 2.8873603343963623,
      "step": 41320
    },
    {
      "dynamics_loss_avg": 0.07123259976506233,
      "epoch": 3.940582275390625e-05,
      "step": 41320
    },
    {
      "action_loss_avg": 0.004341118736192584,
      "epoch": 3.940582275390625e-05,
      "step": 41320
    },
    {
      "epoch": 3.940582275390625e-05,
      "step": 41320,
      "training_step_time": 7.9478819370269775
    },
    {
      "epoch": 3.9406776428222654e-05,
      "model_forward_time": 1.7534501552581787,
      "step": 41321
    },
    {
      "epoch": 3.9406776428222654e-05,
      "step": 41321,
      "training_step_time": 8.74967074394226
    },
    {
      "epoch": 3.9407730102539065e-05,
      "model_forward_time": 1.7541923522949219,
      "step": 41322
    },
    {
      "epoch": 3.9407730102539065e-05,
      "step": 41322,
      "training_step_time": 8.207286834716797
    },
    {
      "epoch": 3.940868377685547e-05,
      "model_forward_time": 1.732370138168335,
      "step": 41323
    },
    {
      "epoch": 3.940868377685547e-05,
      "step": 41323,
      "training_step_time": 7.133360147476196
    },
    {
      "epoch": 3.940963745117187e-05,
      "model_forward_time": 1.7256381511688232,
      "step": 41324
    },
    {
      "epoch": 3.940963745117187e-05,
      "step": 41324,
      "training_step_time": 6.902278184890747
    },
    {
      "epoch": 3.9410591125488284e-05,
      "model_forward_time": 1.7415783405303955,
      "step": 41325
    },
    {
      "epoch": 3.9410591125488284e-05,
      "step": 41325,
      "training_step_time": 7.361464977264404
    },
    {
      "epoch": 3.941154479980469e-05,
      "model_forward_time": 1.0559966564178467,
      "step": 41326
    },
    {
      "epoch": 3.941154479980469e-05,
      "step": 41326,
      "training_step_time": 7.656665325164795
    },
    {
      "epoch": 3.941249847412109e-05,
      "model_forward_time": 1.8158435821533203,
      "step": 41327
    },
    {
      "epoch": 3.941249847412109e-05,
      "step": 41327,
      "training_step_time": 7.791903018951416
    },
    {
      "epoch": 3.94134521484375e-05,
      "model_forward_time": 1.7585372924804688,
      "step": 41328
    },
    {
      "epoch": 3.94134521484375e-05,
      "step": 41328,
      "training_step_time": 8.718696355819702
    },
    {
      "epoch": 3.941440582275391e-05,
      "model_forward_time": 2.904554843902588,
      "step": 41329
    },
    {
      "epoch": 3.941440582275391e-05,
      "step": 41329,
      "training_step_time": 7.916744232177734
    },
    {
      "epoch": 3.941535949707031e-05,
      "grad_norm": 0.16808933019638062,
      "learning_rate": 6.805364838054648e-06,
      "loss": 0.0621,
      "step": 41330
    },
    {
      "epoch": 3.941535949707031e-05,
      "model_forward_time": 1.7379889488220215,
      "step": 41330
    },
    {
      "dynamics_loss_avg": 0.056335237622261045,
      "epoch": 3.941535949707031e-05,
      "step": 41330
    },
    {
      "action_loss_avg": 0.003161321784136817,
      "epoch": 3.941535949707031e-05,
      "step": 41330
    },
    {
      "epoch": 3.941535949707031e-05,
      "step": 41330,
      "training_step_time": 7.737791299819946
    },
    {
      "epoch": 3.941631317138672e-05,
      "model_forward_time": 1.7521169185638428,
      "step": 41331
    },
    {
      "epoch": 3.941631317138672e-05,
      "step": 41331,
      "training_step_time": 8.556860446929932
    },
    {
      "epoch": 3.9417266845703126e-05,
      "model_forward_time": 1.747481107711792,
      "step": 41332
    },
    {
      "epoch": 3.9417266845703126e-05,
      "step": 41332,
      "training_step_time": 7.612465143203735
    },
    {
      "epoch": 3.941822052001953e-05,
      "model_forward_time": 1.7498743534088135,
      "step": 41333
    },
    {
      "epoch": 3.941822052001953e-05,
      "step": 41333,
      "training_step_time": 6.928128719329834
    },
    {
      "epoch": 3.941917419433594e-05,
      "model_forward_time": 1.7495017051696777,
      "step": 41334
    },
    {
      "epoch": 3.941917419433594e-05,
      "step": 41334,
      "training_step_time": 7.4283459186553955
    },
    {
      "epoch": 3.9420127868652345e-05,
      "model_forward_time": 1.7504169940948486,
      "step": 41335
    },
    {
      "epoch": 3.9420127868652345e-05,
      "step": 41335,
      "training_step_time": 7.685835599899292
    },
    {
      "epoch": 3.942108154296875e-05,
      "model_forward_time": 1.8118617534637451,
      "step": 41336
    },
    {
      "epoch": 3.942108154296875e-05,
      "step": 41336,
      "training_step_time": 8.573402404785156
    },
    {
      "epoch": 3.942203521728516e-05,
      "model_forward_time": 1.7400991916656494,
      "step": 41337
    },
    {
      "epoch": 3.942203521728516e-05,
      "step": 41337,
      "training_step_time": 8.569827556610107
    },
    {
      "epoch": 3.9422988891601563e-05,
      "model_forward_time": 2.9310595989227295,
      "step": 41338
    },
    {
      "epoch": 3.9422988891601563e-05,
      "step": 41338,
      "training_step_time": 8.768563270568848
    },
    {
      "epoch": 3.942394256591797e-05,
      "model_forward_time": 1.3796837329864502,
      "step": 41339
    },
    {
      "epoch": 3.942394256591797e-05,
      "step": 41339,
      "training_step_time": 8.007619857788086
    },
    {
      "epoch": 3.942489624023438e-05,
      "grad_norm": 0.13644452393054962,
      "learning_rate": 6.803822816647474e-06,
      "loss": 0.0582,
      "step": 41340
    },
    {
      "epoch": 3.942489624023438e-05,
      "model_forward_time": 1.738044023513794,
      "step": 41340
    },
    {
      "dynamics_loss_avg": 0.06110903937369585,
      "epoch": 3.942489624023438e-05,
      "step": 41340
    },
    {
      "action_loss_avg": 0.0032040643563959748,
      "epoch": 3.942489624023438e-05,
      "step": 41340
    },
    {
      "epoch": 3.942489624023438e-05,
      "step": 41340,
      "training_step_time": 7.851299524307251
    },
    {
      "epoch": 3.942584991455078e-05,
      "model_forward_time": 1.7539870738983154,
      "step": 41341
    },
    {
      "epoch": 3.942584991455078e-05,
      "step": 41341,
      "training_step_time": 7.599286079406738
    },
    {
      "epoch": 3.9426803588867186e-05,
      "model_forward_time": 1.7478663921356201,
      "step": 41342
    },
    {
      "epoch": 3.9426803588867186e-05,
      "step": 41342,
      "training_step_time": 6.937973499298096
    },
    {
      "epoch": 3.94277572631836e-05,
      "model_forward_time": 1.7458491325378418,
      "step": 41343
    },
    {
      "epoch": 3.94277572631836e-05,
      "step": 41343,
      "training_step_time": 6.988378286361694
    },
    {
      "epoch": 3.94287109375e-05,
      "model_forward_time": 1.7432470321655273,
      "step": 41344
    },
    {
      "epoch": 3.94287109375e-05,
      "step": 41344,
      "training_step_time": 7.547382593154907
    },
    {
      "epoch": 3.9429664611816405e-05,
      "model_forward_time": 1.8106684684753418,
      "step": 41345
    },
    {
      "epoch": 3.9429664611816405e-05,
      "step": 41345,
      "training_step_time": 7.853564500808716
    },
    {
      "epoch": 3.943061828613281e-05,
      "model_forward_time": 1.745476245880127,
      "step": 41346
    },
    {
      "epoch": 3.943061828613281e-05,
      "step": 41346,
      "training_step_time": 8.672891855239868
    },
    {
      "epoch": 3.943157196044922e-05,
      "model_forward_time": 1.7362627983093262,
      "step": 41347
    },
    {
      "epoch": 3.943157196044922e-05,
      "step": 41347,
      "training_step_time": 8.9779953956604
    },
    {
      "epoch": 3.9432525634765624e-05,
      "model_forward_time": 1.7499654293060303,
      "step": 41348
    },
    {
      "epoch": 3.9432525634765624e-05,
      "step": 41348,
      "training_step_time": 7.874811172485352
    },
    {
      "epoch": 3.943347930908203e-05,
      "model_forward_time": 1.742326259613037,
      "step": 41349
    },
    {
      "epoch": 3.943347930908203e-05,
      "step": 41349,
      "training_step_time": 7.762211322784424
    },
    {
      "epoch": 3.943443298339844e-05,
      "grad_norm": 0.14599034190177917,
      "learning_rate": 6.802280597976949e-06,
      "loss": 0.0596,
      "step": 41350
    },
    {
      "epoch": 3.943443298339844e-05,
      "model_forward_time": 1.7468135356903076,
      "step": 41350
    },
    {
      "dynamics_loss_avg": 0.06593454517424106,
      "epoch": 3.943443298339844e-05,
      "step": 41350
    },
    {
      "action_loss_avg": 0.0025913504301570355,
      "epoch": 3.943443298339844e-05,
      "step": 41350
    },
    {
      "epoch": 3.943443298339844e-05,
      "step": 41350,
      "training_step_time": 7.774043321609497
    },
    {
      "epoch": 3.943538665771484e-05,
      "model_forward_time": 1.737635612487793,
      "step": 41351
    },
    {
      "epoch": 3.943538665771484e-05,
      "step": 41351,
      "training_step_time": 7.17376446723938
    },
    {
      "epoch": 3.943634033203125e-05,
      "model_forward_time": 1.7414324283599854,
      "step": 41352
    },
    {
      "epoch": 3.943634033203125e-05,
      "step": 41352,
      "training_step_time": 6.97002387046814
    },
    {
      "epoch": 3.943729400634766e-05,
      "model_forward_time": 1.7366061210632324,
      "step": 41353
    },
    {
      "epoch": 3.943729400634766e-05,
      "step": 41353,
      "training_step_time": 8.168975114822388
    },
    {
      "epoch": 3.943824768066406e-05,
      "model_forward_time": 1.8165788650512695,
      "step": 41354
    },
    {
      "epoch": 3.943824768066406e-05,
      "step": 41354,
      "training_step_time": 7.87507176399231
    },
    {
      "epoch": 3.9439201354980466e-05,
      "model_forward_time": 1.7366948127746582,
      "step": 41355
    },
    {
      "epoch": 3.9439201354980466e-05,
      "step": 41355,
      "training_step_time": 8.66734504699707
    },
    {
      "epoch": 3.9440155029296877e-05,
      "model_forward_time": 1.743762493133545,
      "step": 41356
    },
    {
      "epoch": 3.9440155029296877e-05,
      "step": 41356,
      "training_step_time": 7.774955987930298
    },
    {
      "epoch": 3.944110870361328e-05,
      "model_forward_time": 2.9161229133605957,
      "step": 41357
    },
    {
      "epoch": 3.944110870361328e-05,
      "step": 41357,
      "training_step_time": 8.762834787368774
    },
    {
      "epoch": 3.9442062377929685e-05,
      "model_forward_time": 1.7380781173706055,
      "step": 41358
    },
    {
      "epoch": 3.9442062377929685e-05,
      "step": 41358,
      "training_step_time": 8.772373676300049
    },
    {
      "epoch": 3.9443016052246095e-05,
      "model_forward_time": 1.740168809890747,
      "step": 41359
    },
    {
      "epoch": 3.9443016052246095e-05,
      "step": 41359,
      "training_step_time": 7.693533658981323
    },
    {
      "epoch": 3.94439697265625e-05,
      "grad_norm": 0.1832110583782196,
      "learning_rate": 6.800738182211728e-06,
      "loss": 0.0595,
      "step": 41360
    },
    {
      "epoch": 3.94439697265625e-05,
      "model_forward_time": 1.7537565231323242,
      "step": 41360
    },
    {
      "dynamics_loss_avg": 0.06832891590893268,
      "epoch": 3.94439697265625e-05,
      "step": 41360
    },
    {
      "action_loss_avg": 0.004215080972062424,
      "epoch": 3.94439697265625e-05,
      "step": 41360
    },
    {
      "epoch": 3.94439697265625e-05,
      "step": 41360,
      "training_step_time": 7.167875289916992
    },
    {
      "epoch": 3.9444923400878903e-05,
      "model_forward_time": 1.7414436340332031,
      "step": 41361
    },
    {
      "epoch": 3.9444923400878903e-05,
      "step": 41361,
      "training_step_time": 6.917675733566284
    },
    {
      "epoch": 3.9445877075195314e-05,
      "model_forward_time": 1.7472679615020752,
      "step": 41362
    },
    {
      "epoch": 3.9445877075195314e-05,
      "step": 41362,
      "training_step_time": 7.538527727127075
    },
    {
      "epoch": 3.944683074951172e-05,
      "model_forward_time": 1.813028335571289,
      "step": 41363
    },
    {
      "epoch": 3.944683074951172e-05,
      "step": 41363,
      "training_step_time": 7.694225311279297
    },
    {
      "epoch": 3.944778442382812e-05,
      "model_forward_time": 1.7461519241333008,
      "step": 41364
    },
    {
      "epoch": 3.944778442382812e-05,
      "step": 41364,
      "training_step_time": 7.599909782409668
    },
    {
      "epoch": 3.944873809814453e-05,
      "model_forward_time": 1.7443320751190186,
      "step": 41365
    },
    {
      "epoch": 3.944873809814453e-05,
      "step": 41365,
      "training_step_time": 9.346247911453247
    },
    {
      "epoch": 3.944969177246094e-05,
      "model_forward_time": 2.933112621307373,
      "step": 41366
    },
    {
      "epoch": 3.944969177246094e-05,
      "step": 41366,
      "training_step_time": 8.973422288894653
    },
    {
      "epoch": 3.945064544677734e-05,
      "model_forward_time": 1.737215518951416,
      "step": 41367
    },
    {
      "epoch": 3.945064544677734e-05,
      "step": 41367,
      "training_step_time": 8.703798055648804
    },
    {
      "epoch": 3.945159912109375e-05,
      "model_forward_time": 1.7368416786193848,
      "step": 41368
    },
    {
      "epoch": 3.945159912109375e-05,
      "step": 41368,
      "training_step_time": 7.626931190490723
    },
    {
      "epoch": 3.9452552795410156e-05,
      "model_forward_time": 1.77296781539917,
      "step": 41369
    },
    {
      "epoch": 3.9452552795410156e-05,
      "step": 41369,
      "training_step_time": 7.81643009185791
    },
    {
      "epoch": 3.945350646972656e-05,
      "grad_norm": 0.08253559470176697,
      "learning_rate": 6.799195569520484e-06,
      "loss": 0.0655,
      "step": 41370
    },
    {
      "epoch": 3.945350646972656e-05,
      "model_forward_time": 1.75042724609375,
      "step": 41370
    },
    {
      "dynamics_loss_avg": 0.05727513376623392,
      "epoch": 3.945350646972656e-05,
      "step": 41370
    },
    {
      "action_loss_avg": 0.002181779281818308,
      "epoch": 3.945350646972656e-05,
      "step": 41370
    },
    {
      "epoch": 3.945350646972656e-05,
      "step": 41370,
      "training_step_time": 6.91465950012207
    },
    {
      "epoch": 3.945446014404297e-05,
      "model_forward_time": 1.7431244850158691,
      "step": 41371
    },
    {
      "epoch": 3.945446014404297e-05,
      "step": 41371,
      "training_step_time": 7.521136045455933
    },
    {
      "epoch": 3.9455413818359375e-05,
      "model_forward_time": 1.8153324127197266,
      "step": 41372
    },
    {
      "epoch": 3.9455413818359375e-05,
      "step": 41372,
      "training_step_time": 8.201841831207275
    },
    {
      "epoch": 3.945636749267578e-05,
      "model_forward_time": 1.7372653484344482,
      "step": 41373
    },
    {
      "epoch": 3.945636749267578e-05,
      "step": 41373,
      "training_step_time": 8.220918655395508
    },
    {
      "epoch": 3.945732116699219e-05,
      "model_forward_time": 1.7322039604187012,
      "step": 41374
    },
    {
      "epoch": 3.945732116699219e-05,
      "step": 41374,
      "training_step_time": 8.275160074234009
    },
    {
      "epoch": 3.9458274841308594e-05,
      "model_forward_time": 2.885464668273926,
      "step": 41375
    },
    {
      "epoch": 3.9458274841308594e-05,
      "step": 41375,
      "training_step_time": 8.522024869918823
    },
    {
      "epoch": 3.9459228515625e-05,
      "model_forward_time": 1.7463080883026123,
      "step": 41376
    },
    {
      "epoch": 3.9459228515625e-05,
      "step": 41376,
      "training_step_time": 7.654135227203369
    },
    {
      "epoch": 3.946018218994141e-05,
      "model_forward_time": 1.741729497909546,
      "step": 41377
    },
    {
      "epoch": 3.946018218994141e-05,
      "step": 41377,
      "training_step_time": 7.777672052383423
    },
    {
      "epoch": 3.946113586425781e-05,
      "model_forward_time": 1.7416064739227295,
      "step": 41378
    },
    {
      "epoch": 3.946113586425781e-05,
      "step": 41378,
      "training_step_time": 7.634429693222046
    },
    {
      "epoch": 3.946208953857422e-05,
      "model_forward_time": 1.7425813674926758,
      "step": 41379
    },
    {
      "epoch": 3.946208953857422e-05,
      "step": 41379,
      "training_step_time": 6.915616989135742
    },
    {
      "epoch": 3.946304321289063e-05,
      "grad_norm": 0.16136117279529572,
      "learning_rate": 6.7976527600719215e-06,
      "loss": 0.0588,
      "step": 41380
    },
    {
      "epoch": 3.946304321289063e-05,
      "model_forward_time": 1.7538187503814697,
      "step": 41380
    },
    {
      "dynamics_loss_avg": 0.048831230774521825,
      "epoch": 3.946304321289063e-05,
      "step": 41380
    },
    {
      "action_loss_avg": 0.002661610738141462,
      "epoch": 3.946304321289063e-05,
      "step": 41380
    },
    {
      "epoch": 3.946304321289063e-05,
      "step": 41380,
      "training_step_time": 7.46468710899353
    },
    {
      "epoch": 3.946399688720703e-05,
      "model_forward_time": 1.7984979152679443,
      "step": 41381
    },
    {
      "epoch": 3.946399688720703e-05,
      "step": 41381,
      "training_step_time": 7.631212472915649
    },
    {
      "epoch": 3.9464950561523435e-05,
      "model_forward_time": 1.7440028190612793,
      "step": 41382
    },
    {
      "epoch": 3.9464950561523435e-05,
      "step": 41382,
      "training_step_time": 7.713835954666138
    },
    {
      "epoch": 3.9465904235839846e-05,
      "model_forward_time": 1.7381691932678223,
      "step": 41383
    },
    {
      "epoch": 3.9465904235839846e-05,
      "step": 41383,
      "training_step_time": 9.806540489196777
    },
    {
      "epoch": 3.946685791015625e-05,
      "model_forward_time": 2.875119686126709,
      "step": 41384
    },
    {
      "epoch": 3.946685791015625e-05,
      "step": 41384,
      "training_step_time": 8.706860303878784
    },
    {
      "epoch": 3.9467811584472654e-05,
      "model_forward_time": 1.7429633140563965,
      "step": 41385
    },
    {
      "epoch": 3.9467811584472654e-05,
      "step": 41385,
      "training_step_time": 7.934324026107788
    },
    {
      "epoch": 3.9468765258789065e-05,
      "model_forward_time": 1.7466771602630615,
      "step": 41386
    },
    {
      "epoch": 3.9468765258789065e-05,
      "step": 41386,
      "training_step_time": 7.766624212265015
    },
    {
      "epoch": 3.946971893310547e-05,
      "model_forward_time": 1.7429215908050537,
      "step": 41387
    },
    {
      "epoch": 3.946971893310547e-05,
      "step": 41387,
      "training_step_time": 7.563356399536133
    },
    {
      "epoch": 3.947067260742187e-05,
      "model_forward_time": 1.7396342754364014,
      "step": 41388
    },
    {
      "epoch": 3.947067260742187e-05,
      "step": 41388,
      "training_step_time": 7.76015043258667
    },
    {
      "epoch": 3.9471626281738284e-05,
      "model_forward_time": 1.750633955001831,
      "step": 41389
    },
    {
      "epoch": 3.9471626281738284e-05,
      "step": 41389,
      "training_step_time": 7.734617710113525
    },
    {
      "epoch": 3.947257995605469e-05,
      "grad_norm": 0.10958345234394073,
      "learning_rate": 6.796109754034753e-06,
      "loss": 0.0621,
      "step": 41390
    },
    {
      "epoch": 3.947257995605469e-05,
      "model_forward_time": 1.7749464511871338,
      "step": 41390
    },
    {
      "dynamics_loss_avg": 0.07022789642214775,
      "epoch": 3.947257995605469e-05,
      "step": 41390
    },
    {
      "action_loss_avg": 0.0031244993326254187,
      "epoch": 3.947257995605469e-05,
      "step": 41390
    },
    {
      "epoch": 3.947257995605469e-05,
      "step": 41390,
      "training_step_time": 7.674578428268433
    },
    {
      "epoch": 3.947353363037109e-05,
      "model_forward_time": 1.744189739227295,
      "step": 41391
    },
    {
      "epoch": 3.947353363037109e-05,
      "step": 41391,
      "training_step_time": 7.809060573577881
    },
    {
      "epoch": 3.94744873046875e-05,
      "model_forward_time": 1.7261810302734375,
      "step": 41392
    },
    {
      "epoch": 3.94744873046875e-05,
      "step": 41392,
      "training_step_time": 9.522019147872925
    },
    {
      "epoch": 3.947544097900391e-05,
      "model_forward_time": 1.714545726776123,
      "step": 41393
    },
    {
      "epoch": 3.947544097900391e-05,
      "step": 41393,
      "training_step_time": 8.3321852684021
    },
    {
      "epoch": 3.947639465332031e-05,
      "model_forward_time": 1.7345902919769287,
      "step": 41394
    },
    {
      "epoch": 3.947639465332031e-05,
      "step": 41394,
      "training_step_time": 9.34820008277893
    },
    {
      "epoch": 3.947734832763672e-05,
      "model_forward_time": 1.7217471599578857,
      "step": 41395
    },
    {
      "epoch": 3.947734832763672e-05,
      "step": 41395,
      "training_step_time": 8.480901956558228
    },
    {
      "epoch": 3.9478302001953126e-05,
      "model_forward_time": 1.7224187850952148,
      "step": 41396
    },
    {
      "epoch": 3.9478302001953126e-05,
      "step": 41396,
      "training_step_time": 7.620164155960083
    },
    {
      "epoch": 3.947925567626953e-05,
      "model_forward_time": 1.7167165279388428,
      "step": 41397
    },
    {
      "epoch": 3.947925567626953e-05,
      "step": 41397,
      "training_step_time": 7.716578722000122
    },
    {
      "epoch": 3.948020935058594e-05,
      "model_forward_time": 1.7048254013061523,
      "step": 41398
    },
    {
      "epoch": 3.948020935058594e-05,
      "step": 41398,
      "training_step_time": 6.920921087265015
    },
    {
      "epoch": 3.9481163024902345e-05,
      "model_forward_time": 1.711153507232666,
      "step": 41399
    },
    {
      "epoch": 3.9481163024902345e-05,
      "step": 41399,
      "training_step_time": 8.172721862792969
    },
    {
      "epoch": 3.948211669921875e-05,
      "grad_norm": 0.08034276217222214,
      "learning_rate": 6.7945665515777235e-06,
      "loss": 0.0598,
      "step": 41400
    },
    {
      "epoch": 3.948211669921875e-05,
      "model_forward_time": 1.7130210399627686,
      "step": 41400
    },
    {
      "dynamics_loss_avg": 0.06503470074385405,
      "epoch": 3.948211669921875e-05,
      "step": 41400
    },
    {
      "action_loss_avg": 0.0038293771096505223,
      "epoch": 3.948211669921875e-05,
      "step": 41400
    },
    {
      "epoch": 3.948211669921875e-05,
      "step": 41400,
      "training_step_time": 8.597687721252441
    },
    {
      "epoch": 3.948307037353516e-05,
      "model_forward_time": 1.7118861675262451,
      "step": 41401
    },
    {
      "epoch": 3.948307037353516e-05,
      "step": 41401,
      "training_step_time": 7.695642709732056
    },
    {
      "epoch": 3.9484024047851563e-05,
      "model_forward_time": 1.714871883392334,
      "step": 41402
    },
    {
      "epoch": 3.9484024047851563e-05,
      "step": 41402,
      "training_step_time": 8.627249717712402
    },
    {
      "epoch": 3.948497772216797e-05,
      "model_forward_time": 2.28432559967041,
      "step": 41403
    },
    {
      "epoch": 3.948497772216797e-05,
      "step": 41403,
      "training_step_time": 9.539406299591064
    },
    {
      "epoch": 3.948593139648438e-05,
      "model_forward_time": 1.7139134407043457,
      "step": 41404
    },
    {
      "epoch": 3.948593139648438e-05,
      "step": 41404,
      "training_step_time": 8.79721474647522
    },
    {
      "epoch": 3.948688507080078e-05,
      "model_forward_time": 1.7117581367492676,
      "step": 41405
    },
    {
      "epoch": 3.948688507080078e-05,
      "step": 41405,
      "training_step_time": 7.7000885009765625
    },
    {
      "epoch": 3.9487838745117186e-05,
      "model_forward_time": 1.706010341644287,
      "step": 41406
    },
    {
      "epoch": 3.9487838745117186e-05,
      "step": 41406,
      "training_step_time": 7.66571831703186
    },
    {
      "epoch": 3.948879241943359e-05,
      "model_forward_time": 1.707719087600708,
      "step": 41407
    },
    {
      "epoch": 3.948879241943359e-05,
      "step": 41407,
      "training_step_time": 7.568533420562744
    },
    {
      "epoch": 3.948974609375e-05,
      "model_forward_time": 1.7113924026489258,
      "step": 41408
    },
    {
      "epoch": 3.948974609375e-05,
      "step": 41408,
      "training_step_time": 7.735527753829956
    },
    {
      "epoch": 3.9490699768066405e-05,
      "model_forward_time": 1.70247483253479,
      "step": 41409
    },
    {
      "epoch": 3.9490699768066405e-05,
      "step": 41409,
      "training_step_time": 7.754340410232544
    },
    {
      "epoch": 3.949165344238281e-05,
      "grad_norm": 0.11867925524711609,
      "learning_rate": 6.793023152869595e-06,
      "loss": 0.0594,
      "step": 41410
    },
    {
      "epoch": 3.949165344238281e-05,
      "model_forward_time": 1.7063872814178467,
      "step": 41410
    },
    {
      "dynamics_loss_avg": 0.05761288600042462,
      "epoch": 3.949165344238281e-05,
      "step": 41410
    },
    {
      "action_loss_avg": 0.0021277747029671444,
      "epoch": 3.949165344238281e-05,
      "step": 41410
    },
    {
      "epoch": 3.949165344238281e-05,
      "step": 41410,
      "training_step_time": 7.749572277069092
    },
    {
      "epoch": 3.949260711669922e-05,
      "model_forward_time": 1.7227933406829834,
      "step": 41411
    },
    {
      "epoch": 3.949260711669922e-05,
      "step": 41411,
      "training_step_time": 9.800536155700684
    },
    {
      "epoch": 3.9493560791015624e-05,
      "model_forward_time": 2.305901050567627,
      "step": 41412
    },
    {
      "epoch": 3.9493560791015624e-05,
      "step": 41412,
      "training_step_time": 8.285649061203003
    },
    {
      "epoch": 3.949451446533203e-05,
      "model_forward_time": 1.708418369293213,
      "step": 41413
    },
    {
      "epoch": 3.949451446533203e-05,
      "step": 41413,
      "training_step_time": 7.942453861236572
    },
    {
      "epoch": 3.949546813964844e-05,
      "model_forward_time": 1.711270809173584,
      "step": 41414
    },
    {
      "epoch": 3.949546813964844e-05,
      "step": 41414,
      "training_step_time": 7.761652231216431
    },
    {
      "epoch": 3.949642181396484e-05,
      "model_forward_time": 1.7127816677093506,
      "step": 41415
    },
    {
      "epoch": 3.949642181396484e-05,
      "step": 41415,
      "training_step_time": 7.128824710845947
    },
    {
      "epoch": 3.949737548828125e-05,
      "model_forward_time": 1.7076869010925293,
      "step": 41416
    },
    {
      "epoch": 3.949737548828125e-05,
      "step": 41416,
      "training_step_time": 7.736070156097412
    },
    {
      "epoch": 3.949832916259766e-05,
      "model_forward_time": 1.7100834846496582,
      "step": 41417
    },
    {
      "epoch": 3.949832916259766e-05,
      "step": 41417,
      "training_step_time": 7.690524578094482
    },
    {
      "epoch": 3.949928283691406e-05,
      "model_forward_time": 1.7054431438446045,
      "step": 41418
    },
    {
      "epoch": 3.949928283691406e-05,
      "step": 41418,
      "training_step_time": 7.671037912368774
    },
    {
      "epoch": 3.9500236511230466e-05,
      "model_forward_time": 1.7029979228973389,
      "step": 41419
    },
    {
      "epoch": 3.9500236511230466e-05,
      "step": 41419,
      "training_step_time": 7.766796827316284
    },
    {
      "epoch": 3.950119018554688e-05,
      "grad_norm": 0.13877500593662262,
      "learning_rate": 6.791479558079148e-06,
      "loss": 0.0613,
      "step": 41420
    },
    {
      "epoch": 3.950119018554688e-05,
      "model_forward_time": 1.7154755592346191,
      "step": 41420
    },
    {
      "dynamics_loss_avg": 0.06329270079731941,
      "epoch": 3.950119018554688e-05,
      "step": 41420
    },
    {
      "action_loss_avg": 0.0025316632236354053,
      "epoch": 3.950119018554688e-05,
      "step": 41420
    },
    {
      "epoch": 3.950119018554688e-05,
      "step": 41420,
      "training_step_time": 8.4639413356781
    },
    {
      "epoch": 3.950214385986328e-05,
      "model_forward_time": 2.321251153945923,
      "step": 41421
    },
    {
      "epoch": 3.950214385986328e-05,
      "step": 41421,
      "training_step_time": 8.636542081832886
    },
    {
      "epoch": 3.9503097534179685e-05,
      "model_forward_time": 1.7137017250061035,
      "step": 41422
    },
    {
      "epoch": 3.9503097534179685e-05,
      "step": 41422,
      "training_step_time": 8.820293188095093
    },
    {
      "epoch": 3.9504051208496095e-05,
      "model_forward_time": 1.714217185974121,
      "step": 41423
    },
    {
      "epoch": 3.9504051208496095e-05,
      "step": 41423,
      "training_step_time": 7.7525694370269775
    },
    {
      "epoch": 3.95050048828125e-05,
      "model_forward_time": 1.704268455505371,
      "step": 41424
    },
    {
      "epoch": 3.95050048828125e-05,
      "step": 41424,
      "training_step_time": 7.690863609313965
    },
    {
      "epoch": 3.9505958557128904e-05,
      "model_forward_time": 1.7084121704101562,
      "step": 41425
    },
    {
      "epoch": 3.9505958557128904e-05,
      "step": 41425,
      "training_step_time": 7.610694885253906
    },
    {
      "epoch": 3.9506912231445314e-05,
      "model_forward_time": 1.7071037292480469,
      "step": 41426
    },
    {
      "epoch": 3.9506912231445314e-05,
      "step": 41426,
      "training_step_time": 7.628434419631958
    },
    {
      "epoch": 3.950786590576172e-05,
      "model_forward_time": 1.7041115760803223,
      "step": 41427
    },
    {
      "epoch": 3.950786590576172e-05,
      "step": 41427,
      "training_step_time": 7.606268405914307
    },
    {
      "epoch": 3.950881958007812e-05,
      "model_forward_time": 1.7086007595062256,
      "step": 41428
    },
    {
      "epoch": 3.950881958007812e-05,
      "step": 41428,
      "training_step_time": 7.732242584228516
    },
    {
      "epoch": 3.950977325439453e-05,
      "model_forward_time": 1.7243471145629883,
      "step": 41429
    },
    {
      "epoch": 3.950977325439453e-05,
      "step": 41429,
      "training_step_time": 8.699998378753662
    },
    {
      "epoch": 3.951072692871094e-05,
      "grad_norm": 0.1122746542096138,
      "learning_rate": 6.789935767375192e-06,
      "loss": 0.0579,
      "step": 41430
    },
    {
      "epoch": 3.951072692871094e-05,
      "model_forward_time": 1.8774139881134033,
      "step": 41430
    },
    {
      "dynamics_loss_avg": 0.05272438302636147,
      "epoch": 3.951072692871094e-05,
      "step": 41430
    },
    {
      "action_loss_avg": 0.002734483272070065,
      "epoch": 3.951072692871094e-05,
      "step": 41430
    },
    {
      "epoch": 3.951072692871094e-05,
      "step": 41430,
      "training_step_time": 7.754556894302368
    },
    {
      "epoch": 3.951168060302734e-05,
      "model_forward_time": 1.7345554828643799,
      "step": 41431
    },
    {
      "epoch": 3.951168060302734e-05,
      "step": 41431,
      "training_step_time": 8.458378314971924
    },
    {
      "epoch": 3.951263427734375e-05,
      "model_forward_time": 1.3449819087982178,
      "step": 41432
    },
    {
      "epoch": 3.951263427734375e-05,
      "step": 41432,
      "training_step_time": 7.771414518356323
    },
    {
      "epoch": 3.9513587951660156e-05,
      "model_forward_time": 1.683354139328003,
      "step": 41433
    },
    {
      "epoch": 3.9513587951660156e-05,
      "step": 41433,
      "training_step_time": 7.698233604431152
    },
    {
      "epoch": 3.951454162597656e-05,
      "model_forward_time": 1.7072665691375732,
      "step": 41434
    },
    {
      "epoch": 3.951454162597656e-05,
      "step": 41434,
      "training_step_time": 7.694459438323975
    },
    {
      "epoch": 3.951549530029297e-05,
      "model_forward_time": 1.3401830196380615,
      "step": 41435
    },
    {
      "epoch": 3.951549530029297e-05,
      "step": 41435,
      "training_step_time": 7.642648696899414
    },
    {
      "epoch": 3.9516448974609375e-05,
      "model_forward_time": 1.704092264175415,
      "step": 41436
    },
    {
      "epoch": 3.9516448974609375e-05,
      "step": 41436,
      "training_step_time": 8.333574056625366
    },
    {
      "epoch": 3.951740264892578e-05,
      "model_forward_time": 1.703749656677246,
      "step": 41437
    },
    {
      "epoch": 3.951740264892578e-05,
      "step": 41437,
      "training_step_time": 7.759194374084473
    },
    {
      "epoch": 3.951835632324219e-05,
      "model_forward_time": 1.720139741897583,
      "step": 41438
    },
    {
      "epoch": 3.951835632324219e-05,
      "step": 41438,
      "training_step_time": 7.7347447872161865
    },
    {
      "epoch": 3.9519309997558594e-05,
      "model_forward_time": 1.7080707550048828,
      "step": 41439
    },
    {
      "epoch": 3.9519309997558594e-05,
      "step": 41439,
      "training_step_time": 7.7917869091033936
    },
    {
      "epoch": 3.9520263671875e-05,
      "grad_norm": 0.11473669111728668,
      "learning_rate": 6.788391780926551e-06,
      "loss": 0.0626,
      "step": 41440
    },
    {
      "epoch": 3.9520263671875e-05,
      "model_forward_time": 2.2416985034942627,
      "step": 41440
    },
    {
      "dynamics_loss_avg": 0.07791846431791782,
      "epoch": 3.9520263671875e-05,
      "step": 41440
    },
    {
      "action_loss_avg": 0.0026810476905666293,
      "epoch": 3.9520263671875e-05,
      "step": 41440
    },
    {
      "epoch": 3.9520263671875e-05,
      "step": 41440,
      "training_step_time": 10.330795764923096
    },
    {
      "epoch": 3.952121734619141e-05,
      "model_forward_time": 1.7078487873077393,
      "step": 41441
    },
    {
      "epoch": 3.952121734619141e-05,
      "step": 41441,
      "training_step_time": 9.196012020111084
    },
    {
      "epoch": 3.952217102050781e-05,
      "model_forward_time": 1.7064049243927002,
      "step": 41442
    },
    {
      "epoch": 3.952217102050781e-05,
      "step": 41442,
      "training_step_time": 8.164588212966919
    },
    {
      "epoch": 3.952312469482422e-05,
      "model_forward_time": 1.7109582424163818,
      "step": 41443
    },
    {
      "epoch": 3.952312469482422e-05,
      "step": 41443,
      "training_step_time": 8.309341430664062
    },
    {
      "epoch": 3.952407836914063e-05,
      "model_forward_time": 1.7110099792480469,
      "step": 41444
    },
    {
      "epoch": 3.952407836914063e-05,
      "step": 41444,
      "training_step_time": 8.592470169067383
    },
    {
      "epoch": 3.952503204345703e-05,
      "model_forward_time": 1.7035512924194336,
      "step": 41445
    },
    {
      "epoch": 3.952503204345703e-05,
      "step": 41445,
      "training_step_time": 7.569432497024536
    },
    {
      "epoch": 3.9525985717773436e-05,
      "model_forward_time": 1.704052448272705,
      "step": 41446
    },
    {
      "epoch": 3.9525985717773436e-05,
      "step": 41446,
      "training_step_time": 7.663226127624512
    },
    {
      "epoch": 3.9526939392089846e-05,
      "model_forward_time": 1.7191073894500732,
      "step": 41447
    },
    {
      "epoch": 3.9526939392089846e-05,
      "step": 41447,
      "training_step_time": 7.704074382781982
    },
    {
      "epoch": 3.952789306640625e-05,
      "model_forward_time": 1.7032816410064697,
      "step": 41448
    },
    {
      "epoch": 3.952789306640625e-05,
      "step": 41448,
      "training_step_time": 8.782734870910645
    },
    {
      "epoch": 3.9528846740722654e-05,
      "model_forward_time": 2.259220600128174,
      "step": 41449
    },
    {
      "epoch": 3.9528846740722654e-05,
      "step": 41449,
      "training_step_time": 7.94199275970459
    },
    {
      "epoch": 3.9529800415039065e-05,
      "grad_norm": 0.11788901686668396,
      "learning_rate": 6.786847598902072e-06,
      "loss": 0.0552,
      "step": 41450
    },
    {
      "epoch": 3.9529800415039065e-05,
      "model_forward_time": 1.713667631149292,
      "step": 41450
    },
    {
      "dynamics_loss_avg": 0.07312066778540612,
      "epoch": 3.9529800415039065e-05,
      "step": 41450
    },
    {
      "action_loss_avg": 0.004350801400141791,
      "epoch": 3.9529800415039065e-05,
      "step": 41450
    },
    {
      "epoch": 3.9529800415039065e-05,
      "step": 41450,
      "training_step_time": 7.741609811782837
    },
    {
      "epoch": 3.953075408935547e-05,
      "model_forward_time": 1.707028865814209,
      "step": 41451
    },
    {
      "epoch": 3.953075408935547e-05,
      "step": 41451,
      "training_step_time": 7.818412780761719
    },
    {
      "epoch": 3.953170776367187e-05,
      "model_forward_time": 1.7106661796569824,
      "step": 41452
    },
    {
      "epoch": 3.953170776367187e-05,
      "step": 41452,
      "training_step_time": 7.676623106002808
    },
    {
      "epoch": 3.9532661437988284e-05,
      "model_forward_time": 1.708191156387329,
      "step": 41453
    },
    {
      "epoch": 3.9532661437988284e-05,
      "step": 41453,
      "training_step_time": 7.728656053543091
    },
    {
      "epoch": 3.953361511230469e-05,
      "model_forward_time": 1.6898372173309326,
      "step": 41454
    },
    {
      "epoch": 3.953361511230469e-05,
      "step": 41454,
      "training_step_time": 7.632229804992676
    },
    {
      "epoch": 3.953456878662109e-05,
      "model_forward_time": 1.705446481704712,
      "step": 41455
    },
    {
      "epoch": 3.953456878662109e-05,
      "step": 41455,
      "training_step_time": 8.533623933792114
    },
    {
      "epoch": 3.95355224609375e-05,
      "model_forward_time": 1.7160887718200684,
      "step": 41456
    },
    {
      "epoch": 3.95355224609375e-05,
      "step": 41456,
      "training_step_time": 8.455914974212646
    },
    {
      "epoch": 3.953647613525391e-05,
      "model_forward_time": 1.7127041816711426,
      "step": 41457
    },
    {
      "epoch": 3.953647613525391e-05,
      "step": 41457,
      "training_step_time": 10.362452507019043
    },
    {
      "epoch": 3.953742980957031e-05,
      "model_forward_time": 2.272096872329712,
      "step": 41458
    },
    {
      "epoch": 3.953742980957031e-05,
      "step": 41458,
      "training_step_time": 8.778325080871582
    },
    {
      "epoch": 3.953838348388672e-05,
      "model_forward_time": 1.7086389064788818,
      "step": 41459
    },
    {
      "epoch": 3.953838348388672e-05,
      "step": 41459,
      "training_step_time": 8.270166397094727
    },
    {
      "epoch": 3.9539337158203126e-05,
      "grad_norm": 0.12764115631580353,
      "learning_rate": 6.785303221470628e-06,
      "loss": 0.0581,
      "step": 41460
    },
    {
      "epoch": 3.9539337158203126e-05,
      "model_forward_time": 1.6941587924957275,
      "step": 41460
    },
    {
      "dynamics_loss_avg": 0.07198313772678375,
      "epoch": 3.9539337158203126e-05,
      "step": 41460
    },
    {
      "action_loss_avg": 0.0022535017313202845,
      "epoch": 3.9539337158203126e-05,
      "step": 41460
    },
    {
      "epoch": 3.9539337158203126e-05,
      "step": 41460,
      "training_step_time": 7.695579290390015
    },
    {
      "epoch": 3.954029083251953e-05,
      "model_forward_time": 1.6968402862548828,
      "step": 41461
    },
    {
      "epoch": 3.954029083251953e-05,
      "step": 41461,
      "training_step_time": 7.689659595489502
    },
    {
      "epoch": 3.954124450683594e-05,
      "model_forward_time": 1.7008781433105469,
      "step": 41462
    },
    {
      "epoch": 3.954124450683594e-05,
      "step": 41462,
      "training_step_time": 7.716468572616577
    },
    {
      "epoch": 3.9542198181152345e-05,
      "model_forward_time": 1.6959078311920166,
      "step": 41463
    },
    {
      "epoch": 3.9542198181152345e-05,
      "step": 41463,
      "training_step_time": 7.643220901489258
    },
    {
      "epoch": 3.954315185546875e-05,
      "model_forward_time": 1.6995892524719238,
      "step": 41464
    },
    {
      "epoch": 3.954315185546875e-05,
      "step": 41464,
      "training_step_time": 7.648097276687622
    },
    {
      "epoch": 3.954410552978516e-05,
      "model_forward_time": 1.7199463844299316,
      "step": 41465
    },
    {
      "epoch": 3.954410552978516e-05,
      "step": 41465,
      "training_step_time": 8.746140003204346
    },
    {
      "epoch": 3.9545059204101564e-05,
      "model_forward_time": 1.700016736984253,
      "step": 41466
    },
    {
      "epoch": 3.9545059204101564e-05,
      "step": 41466,
      "training_step_time": 7.682646036148071
    },
    {
      "epoch": 3.954601287841797e-05,
      "model_forward_time": 2.2508926391601562,
      "step": 41467
    },
    {
      "epoch": 3.954601287841797e-05,
      "step": 41467,
      "training_step_time": 9.234236001968384
    },
    {
      "epoch": 3.954696655273438e-05,
      "model_forward_time": 1.7229466438293457,
      "step": 41468
    },
    {
      "epoch": 3.954696655273438e-05,
      "step": 41468,
      "training_step_time": 7.672905206680298
    },
    {
      "epoch": 3.954792022705078e-05,
      "model_forward_time": 1.7073771953582764,
      "step": 41469
    },
    {
      "epoch": 3.954792022705078e-05,
      "step": 41469,
      "training_step_time": 8.466200351715088
    },
    {
      "epoch": 3.9548873901367186e-05,
      "grad_norm": 0.11905790120363235,
      "learning_rate": 6.783758648801108e-06,
      "loss": 0.0599,
      "step": 41470
    },
    {
      "epoch": 3.9548873901367186e-05,
      "model_forward_time": 1.7004435062408447,
      "step": 41470
    },
    {
      "dynamics_loss_avg": 0.07339792549610138,
      "epoch": 3.9548873901367186e-05,
      "step": 41470
    },
    {
      "action_loss_avg": 0.0023044969770126046,
      "epoch": 3.9548873901367186e-05,
      "step": 41470
    },
    {
      "epoch": 3.9548873901367186e-05,
      "step": 41470,
      "training_step_time": 7.528107166290283
    },
    {
      "epoch": 3.954982757568359e-05,
      "model_forward_time": 1.6996111869812012,
      "step": 41471
    },
    {
      "epoch": 3.954982757568359e-05,
      "step": 41471,
      "training_step_time": 7.429839372634888
    },
    {
      "epoch": 3.955078125e-05,
      "model_forward_time": 1.6882731914520264,
      "step": 41472
    },
    {
      "epoch": 3.955078125e-05,
      "step": 41472,
      "training_step_time": 7.518763303756714
    },
    {
      "epoch": 3.9551734924316405e-05,
      "model_forward_time": 1.6998133659362793,
      "step": 41473
    },
    {
      "epoch": 3.9551734924316405e-05,
      "step": 41473,
      "training_step_time": 7.435841083526611
    },
    {
      "epoch": 3.955268859863281e-05,
      "model_forward_time": 1.7083337306976318,
      "step": 41474
    },
    {
      "epoch": 3.955268859863281e-05,
      "step": 41474,
      "training_step_time": 7.516988277435303
    },
    {
      "epoch": 3.955364227294922e-05,
      "model_forward_time": 1.6986122131347656,
      "step": 41475
    },
    {
      "epoch": 3.955364227294922e-05,
      "step": 41475,
      "training_step_time": 9.386218547821045
    },
    {
      "epoch": 3.9554595947265624e-05,
      "model_forward_time": 1.696497917175293,
      "step": 41476
    },
    {
      "epoch": 3.9554595947265624e-05,
      "step": 41476,
      "training_step_time": 8.483211040496826
    },
    {
      "epoch": 3.955554962158203e-05,
      "model_forward_time": 2.267075538635254,
      "step": 41477
    },
    {
      "epoch": 3.955554962158203e-05,
      "step": 41477,
      "training_step_time": 7.879559516906738
    },
    {
      "epoch": 3.955650329589844e-05,
      "model_forward_time": 1.7027359008789062,
      "step": 41478
    },
    {
      "epoch": 3.955650329589844e-05,
      "step": 41478,
      "training_step_time": 7.656620740890503
    },
    {
      "epoch": 3.955745697021484e-05,
      "model_forward_time": 1.6943233013153076,
      "step": 41479
    },
    {
      "epoch": 3.955745697021484e-05,
      "step": 41479,
      "training_step_time": 7.341344356536865
    },
    {
      "epoch": 3.955841064453125e-05,
      "grad_norm": 0.11804535984992981,
      "learning_rate": 6.782213881062423e-06,
      "loss": 0.0584,
      "step": 41480
    },
    {
      "epoch": 3.955841064453125e-05,
      "model_forward_time": 1.6976604461669922,
      "step": 41480
    },
    {
      "dynamics_loss_avg": 0.07214361373335124,
      "epoch": 3.955841064453125e-05,
      "step": 41480
    },
    {
      "action_loss_avg": 0.004023513098945841,
      "epoch": 3.955841064453125e-05,
      "step": 41480
    },
    {
      "epoch": 3.955841064453125e-05,
      "step": 41480,
      "training_step_time": 7.53794002532959
    },
    {
      "epoch": 3.955936431884766e-05,
      "model_forward_time": 1.6996405124664307,
      "step": 41481
    },
    {
      "epoch": 3.955936431884766e-05,
      "step": 41481,
      "training_step_time": 7.556698322296143
    },
    {
      "epoch": 3.956031799316406e-05,
      "model_forward_time": 1.6999144554138184,
      "step": 41482
    },
    {
      "epoch": 3.956031799316406e-05,
      "step": 41482,
      "training_step_time": 8.369645118713379
    },
    {
      "epoch": 3.9561271667480466e-05,
      "model_forward_time": 1.712773084640503,
      "step": 41483
    },
    {
      "epoch": 3.9561271667480466e-05,
      "step": 41483,
      "training_step_time": 8.155569076538086
    },
    {
      "epoch": 3.956222534179688e-05,
      "model_forward_time": 1.7049543857574463,
      "step": 41484
    },
    {
      "epoch": 3.956222534179688e-05,
      "step": 41484,
      "training_step_time": 8.642715215682983
    },
    {
      "epoch": 3.956317901611328e-05,
      "model_forward_time": 1.705383062362671,
      "step": 41485
    },
    {
      "epoch": 3.956317901611328e-05,
      "step": 41485,
      "training_step_time": 7.7691473960876465
    },
    {
      "epoch": 3.9564132690429685e-05,
      "model_forward_time": 2.2539453506469727,
      "step": 41486
    },
    {
      "epoch": 3.9564132690429685e-05,
      "step": 41486,
      "training_step_time": 8.533660411834717
    },
    {
      "epoch": 3.9565086364746096e-05,
      "model_forward_time": 1.6924800872802734,
      "step": 41487
    },
    {
      "epoch": 3.9565086364746096e-05,
      "step": 41487,
      "training_step_time": 9.160298585891724
    },
    {
      "epoch": 3.95660400390625e-05,
      "model_forward_time": 0.9780499935150146,
      "step": 41488
    },
    {
      "epoch": 3.95660400390625e-05,
      "step": 41488,
      "training_step_time": 7.46068000793457
    },
    {
      "epoch": 3.9566993713378904e-05,
      "model_forward_time": 1.693054437637329,
      "step": 41489
    },
    {
      "epoch": 3.9566993713378904e-05,
      "step": 41489,
      "training_step_time": 7.621758460998535
    },
    {
      "epoch": 3.9567947387695314e-05,
      "grad_norm": 0.15039277076721191,
      "learning_rate": 6.780668918423507e-06,
      "loss": 0.0552,
      "step": 41490
    },
    {
      "epoch": 3.9567947387695314e-05,
      "model_forward_time": 1.7003521919250488,
      "step": 41490
    },
    {
      "dynamics_loss_avg": 0.05722842272371054,
      "epoch": 3.9567947387695314e-05,
      "step": 41490
    },
    {
      "action_loss_avg": 0.003565611562225968,
      "epoch": 3.9567947387695314e-05,
      "step": 41490
    },
    {
      "epoch": 3.9567947387695314e-05,
      "step": 41490,
      "training_step_time": 7.571569919586182
    },
    {
      "epoch": 3.956890106201172e-05,
      "model_forward_time": 1.7028241157531738,
      "step": 41491
    },
    {
      "epoch": 3.956890106201172e-05,
      "step": 41491,
      "training_step_time": 8.177120208740234
    },
    {
      "epoch": 3.956985473632812e-05,
      "model_forward_time": 1.6951828002929688,
      "step": 41492
    },
    {
      "epoch": 3.956985473632812e-05,
      "step": 41492,
      "training_step_time": 7.644862174987793
    },
    {
      "epoch": 3.957080841064453e-05,
      "model_forward_time": 1.6946601867675781,
      "step": 41493
    },
    {
      "epoch": 3.957080841064453e-05,
      "step": 41493,
      "training_step_time": 8.11384654045105
    },
    {
      "epoch": 3.957176208496094e-05,
      "model_forward_time": 1.698026180267334,
      "step": 41494
    },
    {
      "epoch": 3.957176208496094e-05,
      "step": 41494,
      "training_step_time": 8.721077680587769
    },
    {
      "epoch": 3.957271575927734e-05,
      "model_forward_time": 2.245539903640747,
      "step": 41495
    },
    {
      "epoch": 3.957271575927734e-05,
      "step": 41495,
      "training_step_time": 9.519022941589355
    },
    {
      "epoch": 3.957366943359375e-05,
      "model_forward_time": 1.7202391624450684,
      "step": 41496
    },
    {
      "epoch": 3.957366943359375e-05,
      "step": 41496,
      "training_step_time": 8.28160572052002
    },
    {
      "epoch": 3.9574623107910156e-05,
      "model_forward_time": 1.698000192642212,
      "step": 41497
    },
    {
      "epoch": 3.9574623107910156e-05,
      "step": 41497,
      "training_step_time": 7.6696388721466064
    },
    {
      "epoch": 3.957557678222656e-05,
      "model_forward_time": 1.7024528980255127,
      "step": 41498
    },
    {
      "epoch": 3.957557678222656e-05,
      "step": 41498,
      "training_step_time": 7.432095289230347
    },
    {
      "epoch": 3.957653045654297e-05,
      "model_forward_time": 1.7158939838409424,
      "step": 41499
    },
    {
      "epoch": 3.957653045654297e-05,
      "step": 41499,
      "training_step_time": 8.185309410095215
    },
    {
      "epoch": 3.9577484130859375e-05,
      "grad_norm": 0.1305978000164032,
      "learning_rate": 6.779123761053317e-06,
      "loss": 0.0547,
      "step": 41500
    },
    {
      "epoch": 3.9577484130859375e-05,
      "model_forward_time": 1.697512149810791,
      "step": 41500
    },
    {
      "dynamics_loss_avg": 0.0652005672454834,
      "epoch": 3.9577484130859375e-05,
      "step": 41500
    },
    {
      "action_loss_avg": 0.002421644615242258,
      "epoch": 3.9577484130859375e-05,
      "step": 41500
    },
    {
      "epoch": 3.9577484130859375e-05,
      "step": 41500,
      "training_step_time": 7.468924045562744
    },
    {
      "epoch": 3.957843780517578e-05,
      "model_forward_time": 1.6984469890594482,
      "step": 41501
    },
    {
      "epoch": 3.957843780517578e-05,
      "step": 41501,
      "training_step_time": 9.437619924545288
    },
    {
      "epoch": 3.957939147949219e-05,
      "model_forward_time": 1.6889731884002686,
      "step": 41502
    },
    {
      "epoch": 3.957939147949219e-05,
      "step": 41502,
      "training_step_time": 8.184049606323242
    },
    {
      "epoch": 3.9580345153808594e-05,
      "model_forward_time": 1.6963560581207275,
      "step": 41503
    },
    {
      "epoch": 3.9580345153808594e-05,
      "step": 41503,
      "training_step_time": 8.290462017059326
    },
    {
      "epoch": 3.9581298828125e-05,
      "model_forward_time": 2.23410964012146,
      "step": 41504
    },
    {
      "epoch": 3.9581298828125e-05,
      "step": 41504,
      "training_step_time": 8.865595579147339
    },
    {
      "epoch": 3.958225250244141e-05,
      "model_forward_time": 1.730767011642456,
      "step": 41505
    },
    {
      "epoch": 3.958225250244141e-05,
      "step": 41505,
      "training_step_time": 7.6989734172821045
    },
    {
      "epoch": 3.958320617675781e-05,
      "model_forward_time": 1.7377853393554688,
      "step": 41506
    },
    {
      "epoch": 3.958320617675781e-05,
      "step": 41506,
      "training_step_time": 7.612955093383789
    },
    {
      "epoch": 3.958415985107422e-05,
      "model_forward_time": 1.7336962223052979,
      "step": 41507
    },
    {
      "epoch": 3.958415985107422e-05,
      "step": 41507,
      "training_step_time": 7.613286018371582
    },
    {
      "epoch": 3.958511352539063e-05,
      "model_forward_time": 1.7370100021362305,
      "step": 41508
    },
    {
      "epoch": 3.958511352539063e-05,
      "step": 41508,
      "training_step_time": 7.552245616912842
    },
    {
      "epoch": 3.958606719970703e-05,
      "model_forward_time": 1.743595838546753,
      "step": 41509
    },
    {
      "epoch": 3.958606719970703e-05,
      "step": 41509,
      "training_step_time": 7.495469570159912
    },
    {
      "epoch": 3.9587020874023436e-05,
      "grad_norm": 0.16985337436199188,
      "learning_rate": 6.777578409120826e-06,
      "loss": 0.0565,
      "step": 41510
    },
    {
      "epoch": 3.9587020874023436e-05,
      "model_forward_time": 1.8195929527282715,
      "step": 41510
    },
    {
      "dynamics_loss_avg": 0.06055403556674719,
      "epoch": 3.9587020874023436e-05,
      "step": 41510
    },
    {
      "action_loss_avg": 0.0022477604448795318,
      "epoch": 3.9587020874023436e-05,
      "step": 41510
    },
    {
      "epoch": 3.9587020874023436e-05,
      "step": 41510,
      "training_step_time": 7.5144171714782715
    },
    {
      "epoch": 3.9587974548339846e-05,
      "model_forward_time": 1.7400078773498535,
      "step": 41511
    },
    {
      "epoch": 3.9587974548339846e-05,
      "step": 41511,
      "training_step_time": 7.567646741867065
    },
    {
      "epoch": 3.958892822265625e-05,
      "model_forward_time": 1.737626552581787,
      "step": 41512
    },
    {
      "epoch": 3.958892822265625e-05,
      "step": 41512,
      "training_step_time": 8.650611400604248
    },
    {
      "epoch": 3.9589881896972654e-05,
      "model_forward_time": 1.7357614040374756,
      "step": 41513
    },
    {
      "epoch": 3.9589881896972654e-05,
      "step": 41513,
      "training_step_time": 9.622460842132568
    },
    {
      "epoch": 3.9590835571289065e-05,
      "model_forward_time": 2.877415895462036,
      "step": 41514
    },
    {
      "epoch": 3.9590835571289065e-05,
      "step": 41514,
      "training_step_time": 7.678730249404907
    },
    {
      "epoch": 3.959178924560547e-05,
      "model_forward_time": 1.7340023517608643,
      "step": 41515
    },
    {
      "epoch": 3.959178924560547e-05,
      "step": 41515,
      "training_step_time": 8.417956352233887
    },
    {
      "epoch": 3.959274291992187e-05,
      "model_forward_time": 1.7325506210327148,
      "step": 41516
    },
    {
      "epoch": 3.959274291992187e-05,
      "step": 41516,
      "training_step_time": 7.723083257675171
    },
    {
      "epoch": 3.9593696594238284e-05,
      "model_forward_time": 1.7400355339050293,
      "step": 41517
    },
    {
      "epoch": 3.9593696594238284e-05,
      "step": 41517,
      "training_step_time": 7.5799572467803955
    },
    {
      "epoch": 3.959465026855469e-05,
      "model_forward_time": 1.732572317123413,
      "step": 41518
    },
    {
      "epoch": 3.959465026855469e-05,
      "step": 41518,
      "training_step_time": 7.523052453994751
    },
    {
      "epoch": 3.959560394287109e-05,
      "model_forward_time": 1.7849249839782715,
      "step": 41519
    },
    {
      "epoch": 3.959560394287109e-05,
      "step": 41519,
      "training_step_time": 8.260404586791992
    },
    {
      "epoch": 3.95965576171875e-05,
      "grad_norm": 0.10005535185337067,
      "learning_rate": 6.776032862795034e-06,
      "loss": 0.0561,
      "step": 41520
    },
    {
      "epoch": 3.95965576171875e-05,
      "model_forward_time": 1.7426793575286865,
      "step": 41520
    },
    {
      "dynamics_loss_avg": 0.057501439936459064,
      "epoch": 3.95965576171875e-05,
      "step": 41520
    },
    {
      "action_loss_avg": 0.0038506134616909547,
      "epoch": 3.95965576171875e-05,
      "step": 41520
    },
    {
      "epoch": 3.95965576171875e-05,
      "step": 41520,
      "training_step_time": 8.317315578460693
    },
    {
      "epoch": 3.959751129150391e-05,
      "model_forward_time": 1.737229824066162,
      "step": 41521
    },
    {
      "epoch": 3.959751129150391e-05,
      "step": 41521,
      "training_step_time": 7.7244062423706055
    },
    {
      "epoch": 3.959846496582031e-05,
      "model_forward_time": 1.7343521118164062,
      "step": 41522
    },
    {
      "epoch": 3.959846496582031e-05,
      "step": 41522,
      "training_step_time": 8.310386180877686
    },
    {
      "epoch": 3.959941864013672e-05,
      "model_forward_time": 2.887544631958008,
      "step": 41523
    },
    {
      "epoch": 3.959941864013672e-05,
      "step": 41523,
      "training_step_time": 8.767215728759766
    },
    {
      "epoch": 3.9600372314453126e-05,
      "model_forward_time": 1.738184928894043,
      "step": 41524
    },
    {
      "epoch": 3.9600372314453126e-05,
      "step": 41524,
      "training_step_time": 9.109762907028198
    },
    {
      "epoch": 3.960132598876953e-05,
      "model_forward_time": 1.7766039371490479,
      "step": 41525
    },
    {
      "epoch": 3.960132598876953e-05,
      "step": 41525,
      "training_step_time": 7.564407825469971
    },
    {
      "epoch": 3.960227966308594e-05,
      "model_forward_time": 1.732328176498413,
      "step": 41526
    },
    {
      "epoch": 3.960227966308594e-05,
      "step": 41526,
      "training_step_time": 7.748696804046631
    },
    {
      "epoch": 3.9603233337402345e-05,
      "model_forward_time": 1.7441470623016357,
      "step": 41527
    },
    {
      "epoch": 3.9603233337402345e-05,
      "step": 41527,
      "training_step_time": 7.57951807975769
    },
    {
      "epoch": 3.960418701171875e-05,
      "model_forward_time": 1.7420353889465332,
      "step": 41528
    },
    {
      "epoch": 3.960418701171875e-05,
      "step": 41528,
      "training_step_time": 8.369904518127441
    },
    {
      "epoch": 3.960514068603516e-05,
      "model_forward_time": 1.9085612297058105,
      "step": 41529
    },
    {
      "epoch": 3.960514068603516e-05,
      "step": 41529,
      "training_step_time": 8.560413599014282
    },
    {
      "epoch": 3.9606094360351564e-05,
      "grad_norm": 0.18616902828216553,
      "learning_rate": 6.7744871222449575e-06,
      "loss": 0.0546,
      "step": 41530
    },
    {
      "epoch": 3.9606094360351564e-05,
      "model_forward_time": 1.8860864639282227,
      "step": 41530
    },
    {
      "dynamics_loss_avg": 0.07207915987819433,
      "epoch": 3.9606094360351564e-05,
      "step": 41530
    },
    {
      "action_loss_avg": 0.002672033879207447,
      "epoch": 3.9606094360351564e-05,
      "step": 41530
    },
    {
      "epoch": 3.9606094360351564e-05,
      "step": 41530,
      "training_step_time": 7.611693859100342
    },
    {
      "epoch": 3.960704803466797e-05,
      "model_forward_time": 1.7664237022399902,
      "step": 41531
    },
    {
      "epoch": 3.960704803466797e-05,
      "step": 41531,
      "training_step_time": 8.478368759155273
    },
    {
      "epoch": 3.960800170898437e-05,
      "model_forward_time": 2.882126808166504,
      "step": 41532
    },
    {
      "epoch": 3.960800170898437e-05,
      "step": 41532,
      "training_step_time": 8.562556505203247
    },
    {
      "epoch": 3.960895538330078e-05,
      "model_forward_time": 1.8471717834472656,
      "step": 41533
    },
    {
      "epoch": 3.960895538330078e-05,
      "step": 41533,
      "training_step_time": 7.621484279632568
    },
    {
      "epoch": 3.9609909057617186e-05,
      "model_forward_time": 1.7419300079345703,
      "step": 41534
    },
    {
      "epoch": 3.9609909057617186e-05,
      "step": 41534,
      "training_step_time": 7.666016101837158
    },
    {
      "epoch": 3.961086273193359e-05,
      "model_forward_time": 1.739396572113037,
      "step": 41535
    },
    {
      "epoch": 3.961086273193359e-05,
      "step": 41535,
      "training_step_time": 7.54165244102478
    },
    {
      "epoch": 3.961181640625e-05,
      "model_forward_time": 1.7211756706237793,
      "step": 41536
    },
    {
      "epoch": 3.961181640625e-05,
      "step": 41536,
      "training_step_time": 7.575603485107422
    },
    {
      "epoch": 3.9612770080566405e-05,
      "model_forward_time": 1.7269411087036133,
      "step": 41537
    },
    {
      "epoch": 3.9612770080566405e-05,
      "step": 41537,
      "training_step_time": 7.2628984451293945
    },
    {
      "epoch": 3.961372375488281e-05,
      "model_forward_time": 1.7345359325408936,
      "step": 41538
    },
    {
      "epoch": 3.961372375488281e-05,
      "step": 41538,
      "training_step_time": 8.079885244369507
    },
    {
      "epoch": 3.961467742919922e-05,
      "model_forward_time": 1.712618112564087,
      "step": 41539
    },
    {
      "epoch": 3.961467742919922e-05,
      "step": 41539,
      "training_step_time": 7.720681667327881
    },
    {
      "epoch": 3.9615631103515624e-05,
      "grad_norm": 0.13117995858192444,
      "learning_rate": 6.7729411876396365e-06,
      "loss": 0.0565,
      "step": 41540
    },
    {
      "epoch": 3.9615631103515624e-05,
      "model_forward_time": 1.7212765216827393,
      "step": 41540
    },
    {
      "dynamics_loss_avg": 0.048948577418923375,
      "epoch": 3.9615631103515624e-05,
      "step": 41540
    },
    {
      "action_loss_avg": 0.0021172413747990505,
      "epoch": 3.9615631103515624e-05,
      "step": 41540
    },
    {
      "epoch": 3.9615631103515624e-05,
      "step": 41540,
      "training_step_time": 7.815988540649414
    },
    {
      "epoch": 3.961658477783203e-05,
      "model_forward_time": 2.8568148612976074,
      "step": 41541
    },
    {
      "epoch": 3.961658477783203e-05,
      "step": 41541,
      "training_step_time": 9.494703531265259
    },
    {
      "epoch": 3.961753845214844e-05,
      "model_forward_time": 1.828260898590088,
      "step": 41542
    },
    {
      "epoch": 3.961753845214844e-05,
      "step": 41542,
      "training_step_time": 7.81860876083374
    },
    {
      "epoch": 3.961849212646484e-05,
      "model_forward_time": 1.7143266201019287,
      "step": 41543
    },
    {
      "epoch": 3.961849212646484e-05,
      "step": 41543,
      "training_step_time": 7.774712324142456
    },
    {
      "epoch": 3.961944580078125e-05,
      "model_forward_time": 1.7198476791381836,
      "step": 41544
    },
    {
      "epoch": 3.961944580078125e-05,
      "step": 41544,
      "training_step_time": 7.641785144805908
    },
    {
      "epoch": 3.962039947509766e-05,
      "model_forward_time": 1.7420597076416016,
      "step": 41545
    },
    {
      "epoch": 3.962039947509766e-05,
      "step": 41545,
      "training_step_time": 7.700597524642944
    },
    {
      "epoch": 3.962135314941406e-05,
      "model_forward_time": 1.7275059223175049,
      "step": 41546
    },
    {
      "epoch": 3.962135314941406e-05,
      "step": 41546,
      "training_step_time": 8.182881355285645
    },
    {
      "epoch": 3.9622306823730466e-05,
      "model_forward_time": 1.7252333164215088,
      "step": 41547
    },
    {
      "epoch": 3.9622306823730466e-05,
      "step": 41547,
      "training_step_time": 7.81987452507019
    },
    {
      "epoch": 3.962326049804688e-05,
      "model_forward_time": 1.725616455078125,
      "step": 41548
    },
    {
      "epoch": 3.962326049804688e-05,
      "step": 41548,
      "training_step_time": 8.2186758518219
    },
    {
      "epoch": 3.962421417236328e-05,
      "model_forward_time": 1.7301812171936035,
      "step": 41549
    },
    {
      "epoch": 3.962421417236328e-05,
      "step": 41549,
      "training_step_time": 8.538965702056885
    },
    {
      "epoch": 3.9625167846679685e-05,
      "grad_norm": 0.09778107702732086,
      "learning_rate": 6.771395059148133e-06,
      "loss": 0.0587,
      "step": 41550
    },
    {
      "epoch": 3.9625167846679685e-05,
      "model_forward_time": 2.8341124057769775,
      "step": 41550
    },
    {
      "dynamics_loss_avg": 0.05429288819432258,
      "epoch": 3.9625167846679685e-05,
      "step": 41550
    },
    {
      "action_loss_avg": 0.0035747194022405894,
      "epoch": 3.9625167846679685e-05,
      "step": 41550
    },
    {
      "epoch": 3.9625167846679685e-05,
      "step": 41550,
      "training_step_time": 8.472463607788086
    },
    {
      "epoch": 3.9626121520996096e-05,
      "model_forward_time": 1.8026978969573975,
      "step": 41551
    },
    {
      "epoch": 3.9626121520996096e-05,
      "step": 41551,
      "training_step_time": 8.653818607330322
    },
    {
      "epoch": 3.96270751953125e-05,
      "model_forward_time": 1.7193810939788818,
      "step": 41552
    },
    {
      "epoch": 3.96270751953125e-05,
      "step": 41552,
      "training_step_time": 7.7867372035980225
    },
    {
      "epoch": 3.9628028869628904e-05,
      "model_forward_time": 1.7251129150390625,
      "step": 41553
    },
    {
      "epoch": 3.9628028869628904e-05,
      "step": 41553,
      "training_step_time": 7.771615982055664
    },
    {
      "epoch": 3.9628982543945314e-05,
      "model_forward_time": 1.7342770099639893,
      "step": 41554
    },
    {
      "epoch": 3.9628982543945314e-05,
      "step": 41554,
      "training_step_time": 7.839936256408691
    },
    {
      "epoch": 3.962993621826172e-05,
      "model_forward_time": 1.7349979877471924,
      "step": 41555
    },
    {
      "epoch": 3.962993621826172e-05,
      "step": 41555,
      "training_step_time": 7.358764171600342
    },
    {
      "epoch": 3.963088989257812e-05,
      "model_forward_time": 1.7259225845336914,
      "step": 41556
    },
    {
      "epoch": 3.963088989257812e-05,
      "step": 41556,
      "training_step_time": 7.7098917961120605
    },
    {
      "epoch": 3.963184356689453e-05,
      "model_forward_time": 1.744025468826294,
      "step": 41557
    },
    {
      "epoch": 3.963184356689453e-05,
      "step": 41557,
      "training_step_time": 9.261674642562866
    },
    {
      "epoch": 3.963279724121094e-05,
      "model_forward_time": 1.731917142868042,
      "step": 41558
    },
    {
      "epoch": 3.963279724121094e-05,
      "step": 41558,
      "training_step_time": 9.410964250564575
    },
    {
      "epoch": 3.963375091552734e-05,
      "model_forward_time": 1.7271201610565186,
      "step": 41559
    },
    {
      "epoch": 3.963375091552734e-05,
      "step": 41559,
      "training_step_time": 8.69796371459961
    },
    {
      "epoch": 3.963470458984375e-05,
      "grad_norm": 0.11323238909244537,
      "learning_rate": 6.76984873693953e-06,
      "loss": 0.0618,
      "step": 41560
    },
    {
      "epoch": 3.963470458984375e-05,
      "model_forward_time": 2.9564709663391113,
      "step": 41560
    },
    {
      "dynamics_loss_avg": 0.04911141097545624,
      "epoch": 3.963470458984375e-05,
      "step": 41560
    },
    {
      "action_loss_avg": 0.002255777036771178,
      "epoch": 3.963470458984375e-05,
      "step": 41560
    },
    {
      "epoch": 3.963470458984375e-05,
      "step": 41560,
      "training_step_time": 8.593257665634155
    },
    {
      "epoch": 3.9635658264160156e-05,
      "model_forward_time": 1.726471185684204,
      "step": 41561
    },
    {
      "epoch": 3.9635658264160156e-05,
      "step": 41561,
      "training_step_time": 8.831325769424438
    },
    {
      "epoch": 3.963661193847656e-05,
      "model_forward_time": 1.7459592819213867,
      "step": 41562
    },
    {
      "epoch": 3.963661193847656e-05,
      "step": 41562,
      "training_step_time": 7.729966878890991
    },
    {
      "epoch": 3.963756561279297e-05,
      "model_forward_time": 1.7424287796020508,
      "step": 41563
    },
    {
      "epoch": 3.963756561279297e-05,
      "step": 41563,
      "training_step_time": 7.770428895950317
    },
    {
      "epoch": 3.9638519287109375e-05,
      "model_forward_time": 1.718583106994629,
      "step": 41564
    },
    {
      "epoch": 3.9638519287109375e-05,
      "step": 41564,
      "training_step_time": 7.531454086303711
    },
    {
      "epoch": 3.963947296142578e-05,
      "model_forward_time": 1.7225704193115234,
      "step": 41565
    },
    {
      "epoch": 3.963947296142578e-05,
      "step": 41565,
      "training_step_time": 7.695856094360352
    },
    {
      "epoch": 3.964042663574219e-05,
      "model_forward_time": 1.7337214946746826,
      "step": 41566
    },
    {
      "epoch": 3.964042663574219e-05,
      "step": 41566,
      "training_step_time": 7.718355178833008
    },
    {
      "epoch": 3.9641380310058594e-05,
      "model_forward_time": 1.7220916748046875,
      "step": 41567
    },
    {
      "epoch": 3.9641380310058594e-05,
      "step": 41567,
      "training_step_time": 7.835766315460205
    },
    {
      "epoch": 3.9642333984375e-05,
      "model_forward_time": 1.719040870666504,
      "step": 41568
    },
    {
      "epoch": 3.9642333984375e-05,
      "step": 41568,
      "training_step_time": 9.289439916610718
    },
    {
      "epoch": 3.964328765869141e-05,
      "model_forward_time": 2.9348275661468506,
      "step": 41569
    },
    {
      "epoch": 3.964328765869141e-05,
      "step": 41569,
      "training_step_time": 9.616907835006714
    },
    {
      "epoch": 3.964424133300781e-05,
      "grad_norm": 0.15925347805023193,
      "learning_rate": 6.768302221182931e-06,
      "loss": 0.0615,
      "step": 41570
    },
    {
      "epoch": 3.964424133300781e-05,
      "model_forward_time": 1.7329044342041016,
      "step": 41570
    },
    {
      "dynamics_loss_avg": 0.08287958800792694,
      "epoch": 3.964424133300781e-05,
      "step": 41570
    },
    {
      "action_loss_avg": 0.0038844541530124845,
      "epoch": 3.964424133300781e-05,
      "step": 41570
    },
    {
      "epoch": 3.964424133300781e-05,
      "step": 41570,
      "training_step_time": 7.843114852905273
    },
    {
      "epoch": 3.964519500732422e-05,
      "model_forward_time": 1.7224836349487305,
      "step": 41571
    },
    {
      "epoch": 3.964519500732422e-05,
      "step": 41571,
      "training_step_time": 7.719209909439087
    },
    {
      "epoch": 3.964614868164063e-05,
      "model_forward_time": 1.7228617668151855,
      "step": 41572
    },
    {
      "epoch": 3.964614868164063e-05,
      "step": 41572,
      "training_step_time": 7.72234320640564
    },
    {
      "epoch": 3.964710235595703e-05,
      "model_forward_time": 1.7364213466644287,
      "step": 41573
    },
    {
      "epoch": 3.964710235595703e-05,
      "step": 41573,
      "training_step_time": 7.629096269607544
    },
    {
      "epoch": 3.9648056030273436e-05,
      "model_forward_time": 1.722545862197876,
      "step": 41574
    },
    {
      "epoch": 3.9648056030273436e-05,
      "step": 41574,
      "training_step_time": 8.989940404891968
    },
    {
      "epoch": 3.9649009704589846e-05,
      "model_forward_time": 1.7306053638458252,
      "step": 41575
    },
    {
      "epoch": 3.9649009704589846e-05,
      "step": 41575,
      "training_step_time": 8.54378056526184
    },
    {
      "epoch": 3.964996337890625e-05,
      "model_forward_time": 1.74212646484375,
      "step": 41576
    },
    {
      "epoch": 3.964996337890625e-05,
      "step": 41576,
      "training_step_time": 8.625724077224731
    },
    {
      "epoch": 3.9650917053222655e-05,
      "model_forward_time": 1.723625659942627,
      "step": 41577
    },
    {
      "epoch": 3.9650917053222655e-05,
      "step": 41577,
      "training_step_time": 7.743464946746826
    },
    {
      "epoch": 3.9651870727539065e-05,
      "model_forward_time": 2.904432535171509,
      "step": 41578
    },
    {
      "epoch": 3.9651870727539065e-05,
      "step": 41578,
      "training_step_time": 7.942960023880005
    },
    {
      "epoch": 3.965282440185547e-05,
      "model_forward_time": 1.829683780670166,
      "step": 41579
    },
    {
      "epoch": 3.965282440185547e-05,
      "step": 41579,
      "training_step_time": 8.377363443374634
    },
    {
      "epoch": 3.965377807617187e-05,
      "grad_norm": 0.1484968513250351,
      "learning_rate": 6.766755512047458e-06,
      "loss": 0.0609,
      "step": 41580
    },
    {
      "epoch": 3.965377807617187e-05,
      "model_forward_time": 1.722977876663208,
      "step": 41580
    },
    {
      "dynamics_loss_avg": 0.051884543150663376,
      "epoch": 3.965377807617187e-05,
      "step": 41580
    },
    {
      "action_loss_avg": 0.0019043363281525671,
      "epoch": 3.965377807617187e-05,
      "step": 41580
    },
    {
      "epoch": 3.965377807617187e-05,
      "step": 41580,
      "training_step_time": 7.654915809631348
    },
    {
      "epoch": 3.9654731750488284e-05,
      "model_forward_time": 1.7374651432037354,
      "step": 41581
    },
    {
      "epoch": 3.9654731750488284e-05,
      "step": 41581,
      "training_step_time": 8.575883626937866
    },
    {
      "epoch": 3.965568542480469e-05,
      "model_forward_time": 1.723252534866333,
      "step": 41582
    },
    {
      "epoch": 3.965568542480469e-05,
      "step": 41582,
      "training_step_time": 7.684807062149048
    },
    {
      "epoch": 3.965663909912109e-05,
      "model_forward_time": 1.7409961223602295,
      "step": 41583
    },
    {
      "epoch": 3.965663909912109e-05,
      "step": 41583,
      "training_step_time": 8.294381618499756
    },
    {
      "epoch": 3.96575927734375e-05,
      "model_forward_time": 1.721724271774292,
      "step": 41584
    },
    {
      "epoch": 3.96575927734375e-05,
      "step": 41584,
      "training_step_time": 7.591382026672363
    },
    {
      "epoch": 3.965854644775391e-05,
      "model_forward_time": 1.7352323532104492,
      "step": 41585
    },
    {
      "epoch": 3.965854644775391e-05,
      "step": 41585,
      "training_step_time": 8.858474969863892
    },
    {
      "epoch": 3.965950012207031e-05,
      "model_forward_time": 1.734344720840454,
      "step": 41586
    },
    {
      "epoch": 3.965950012207031e-05,
      "step": 41586,
      "training_step_time": 8.805968284606934
    },
    {
      "epoch": 3.966045379638672e-05,
      "model_forward_time": 2.871821165084839,
      "step": 41587
    },
    {
      "epoch": 3.966045379638672e-05,
      "step": 41587,
      "training_step_time": 8.396837711334229
    },
    {
      "epoch": 3.9661407470703126e-05,
      "model_forward_time": 1.838390588760376,
      "step": 41588
    },
    {
      "epoch": 3.9661407470703126e-05,
      "step": 41588,
      "training_step_time": 8.949604749679565
    },
    {
      "epoch": 3.966236114501953e-05,
      "model_forward_time": 1.7297837734222412,
      "step": 41589
    },
    {
      "epoch": 3.966236114501953e-05,
      "step": 41589,
      "training_step_time": 7.721566915512085
    },
    {
      "epoch": 3.966331481933594e-05,
      "grad_norm": 0.13057954609394073,
      "learning_rate": 6.765208609702258e-06,
      "loss": 0.0625,
      "step": 41590
    },
    {
      "epoch": 3.966331481933594e-05,
      "model_forward_time": 1.7358863353729248,
      "step": 41590
    },
    {
      "dynamics_loss_avg": 0.07361022159457206,
      "epoch": 3.966331481933594e-05,
      "step": 41590
    },
    {
      "action_loss_avg": 0.003383710817433894,
      "epoch": 3.966331481933594e-05,
      "step": 41590
    },
    {
      "epoch": 3.966331481933594e-05,
      "step": 41590,
      "training_step_time": 7.578898668289185
    },
    {
      "epoch": 3.9664268493652345e-05,
      "model_forward_time": 1.7257745265960693,
      "step": 41591
    },
    {
      "epoch": 3.9664268493652345e-05,
      "step": 41591,
      "training_step_time": 7.725308895111084
    },
    {
      "epoch": 3.966522216796875e-05,
      "model_forward_time": 1.7255604267120361,
      "step": 41592
    },
    {
      "epoch": 3.966522216796875e-05,
      "step": 41592,
      "training_step_time": 8.180755376815796
    },
    {
      "epoch": 3.966617584228516e-05,
      "model_forward_time": 1.735827922821045,
      "step": 41593
    },
    {
      "epoch": 3.966617584228516e-05,
      "step": 41593,
      "training_step_time": 7.5668017864227295
    },
    {
      "epoch": 3.9667129516601564e-05,
      "model_forward_time": 1.727525234222412,
      "step": 41594
    },
    {
      "epoch": 3.9667129516601564e-05,
      "step": 41594,
      "training_step_time": 7.689420461654663
    },
    {
      "epoch": 3.966808319091797e-05,
      "model_forward_time": 1.7262442111968994,
      "step": 41595
    },
    {
      "epoch": 3.966808319091797e-05,
      "step": 41595,
      "training_step_time": 8.782238006591797
    },
    {
      "epoch": 3.966903686523437e-05,
      "model_forward_time": 2.852509021759033,
      "step": 41596
    },
    {
      "epoch": 3.966903686523437e-05,
      "step": 41596,
      "training_step_time": 7.766637325286865
    },
    {
      "epoch": 3.966999053955078e-05,
      "model_forward_time": 1.7943828105926514,
      "step": 41597
    },
    {
      "epoch": 3.966999053955078e-05,
      "step": 41597,
      "training_step_time": 7.9703919887542725
    },
    {
      "epoch": 3.9670944213867187e-05,
      "model_forward_time": 1.72922945022583,
      "step": 41598
    },
    {
      "epoch": 3.9670944213867187e-05,
      "step": 41598,
      "training_step_time": 7.727111577987671
    },
    {
      "epoch": 3.967189788818359e-05,
      "model_forward_time": 1.730586290359497,
      "step": 41599
    },
    {
      "epoch": 3.967189788818359e-05,
      "step": 41599,
      "training_step_time": 7.517927646636963
    },
    {
      "epoch": 3.96728515625e-05,
      "grad_norm": 0.15372847020626068,
      "learning_rate": 6.7636615143165e-06,
      "loss": 0.0574,
      "step": 41600
    },
    {
      "epoch": 3.96728515625e-05,
      "model_forward_time": 1.72169828414917,
      "step": 41600
    },
    {
      "dynamics_loss_avg": 0.06019307225942612,
      "epoch": 3.96728515625e-05,
      "step": 41600
    },
    {
      "action_loss_avg": 0.001669186056824401,
      "epoch": 3.96728515625e-05,
      "step": 41600
    },
    {
      "epoch": 3.96728515625e-05,
      "step": 41600,
      "training_step_time": 7.761127233505249
    },
    {
      "epoch": 3.9673805236816405e-05,
      "model_forward_time": 1.7275547981262207,
      "step": 41601
    },
    {
      "epoch": 3.9673805236816405e-05,
      "step": 41601,
      "training_step_time": 7.575651168823242
    },
    {
      "epoch": 3.967475891113281e-05,
      "model_forward_time": 1.7349884510040283,
      "step": 41602
    },
    {
      "epoch": 3.967475891113281e-05,
      "step": 41602,
      "training_step_time": 8.078950881958008
    },
    {
      "epoch": 3.967571258544922e-05,
      "model_forward_time": 1.7305493354797363,
      "step": 41603
    },
    {
      "epoch": 3.967571258544922e-05,
      "step": 41603,
      "training_step_time": 8.453674554824829
    },
    {
      "epoch": 3.9676666259765624e-05,
      "model_forward_time": 1.7383430004119873,
      "step": 41604
    },
    {
      "epoch": 3.9676666259765624e-05,
      "step": 41604,
      "training_step_time": 7.678024530410767
    },
    {
      "epoch": 3.967761993408203e-05,
      "model_forward_time": 1.7250092029571533,
      "step": 41605
    },
    {
      "epoch": 3.967761993408203e-05,
      "step": 41605,
      "training_step_time": 8.238160848617554
    },
    {
      "epoch": 3.967857360839844e-05,
      "model_forward_time": 1.7711646556854248,
      "step": 41606
    },
    {
      "epoch": 3.967857360839844e-05,
      "step": 41606,
      "training_step_time": 8.9454505443573
    },
    {
      "epoch": 3.967952728271484e-05,
      "model_forward_time": 1.7183277606964111,
      "step": 41607
    },
    {
      "epoch": 3.967952728271484e-05,
      "step": 41607,
      "training_step_time": 8.51349687576294
    },
    {
      "epoch": 3.968048095703125e-05,
      "model_forward_time": 1.7183785438537598,
      "step": 41608
    },
    {
      "epoch": 3.968048095703125e-05,
      "step": 41608,
      "training_step_time": 7.481002569198608
    },
    {
      "epoch": 3.968143463134766e-05,
      "model_forward_time": 1.7301087379455566,
      "step": 41609
    },
    {
      "epoch": 3.968143463134766e-05,
      "step": 41609,
      "training_step_time": 7.764073848724365
    },
    {
      "epoch": 3.968238830566406e-05,
      "grad_norm": 0.16577333211898804,
      "learning_rate": 6.76211422605937e-06,
      "loss": 0.0619,
      "step": 41610
    },
    {
      "epoch": 3.968238830566406e-05,
      "model_forward_time": 1.7245306968688965,
      "step": 41610
    },
    {
      "dynamics_loss_avg": 0.05820973310619593,
      "epoch": 3.968238830566406e-05,
      "step": 41610
    },
    {
      "action_loss_avg": 0.0025106939661782234,
      "epoch": 3.968238830566406e-05,
      "step": 41610
    },
    {
      "epoch": 3.968238830566406e-05,
      "step": 41610,
      "training_step_time": 8.386519193649292
    },
    {
      "epoch": 3.9683341979980466e-05,
      "model_forward_time": 1.7236990928649902,
      "step": 41611
    },
    {
      "epoch": 3.9683341979980466e-05,
      "step": 41611,
      "training_step_time": 8.081418991088867
    },
    {
      "epoch": 3.968429565429688e-05,
      "model_forward_time": 1.7245841026306152,
      "step": 41612
    },
    {
      "epoch": 3.968429565429688e-05,
      "step": 41612,
      "training_step_time": 8.43994665145874
    },
    {
      "epoch": 3.968524932861328e-05,
      "model_forward_time": 1.7204689979553223,
      "step": 41613
    },
    {
      "epoch": 3.968524932861328e-05,
      "step": 41613,
      "training_step_time": 8.895589590072632
    },
    {
      "epoch": 3.9686203002929685e-05,
      "model_forward_time": 1.7281768321990967,
      "step": 41614
    },
    {
      "epoch": 3.9686203002929685e-05,
      "step": 41614,
      "training_step_time": 7.817683935165405
    },
    {
      "epoch": 3.9687156677246096e-05,
      "model_forward_time": 2.9031059741973877,
      "step": 41615
    },
    {
      "epoch": 3.9687156677246096e-05,
      "step": 41615,
      "training_step_time": 8.90510630607605
    },
    {
      "epoch": 3.96881103515625e-05,
      "model_forward_time": 1.8416149616241455,
      "step": 41616
    },
    {
      "epoch": 3.96881103515625e-05,
      "step": 41616,
      "training_step_time": 7.740605115890503
    },
    {
      "epoch": 3.9689064025878904e-05,
      "model_forward_time": 1.7178027629852295,
      "step": 41617
    },
    {
      "epoch": 3.9689064025878904e-05,
      "step": 41617,
      "training_step_time": 7.805163621902466
    },
    {
      "epoch": 3.9690017700195315e-05,
      "model_forward_time": 1.7183837890625,
      "step": 41618
    },
    {
      "epoch": 3.9690017700195315e-05,
      "step": 41618,
      "training_step_time": 8.823607683181763
    },
    {
      "epoch": 3.969097137451172e-05,
      "model_forward_time": 1.729651927947998,
      "step": 41619
    },
    {
      "epoch": 3.969097137451172e-05,
      "step": 41619,
      "training_step_time": 7.757026433944702
    },
    {
      "epoch": 3.969192504882812e-05,
      "grad_norm": 0.142945796251297,
      "learning_rate": 6.760566745100077e-06,
      "loss": 0.0558,
      "step": 41620
    },
    {
      "epoch": 3.969192504882812e-05,
      "model_forward_time": 1.7242999076843262,
      "step": 41620
    },
    {
      "dynamics_loss_avg": 0.06095174252986908,
      "epoch": 3.969192504882812e-05,
      "step": 41620
    },
    {
      "action_loss_avg": 0.005499591422267258,
      "epoch": 3.969192504882812e-05,
      "step": 41620
    },
    {
      "epoch": 3.969192504882812e-05,
      "step": 41620,
      "training_step_time": 8.501366376876831
    },
    {
      "epoch": 3.969287872314453e-05,
      "model_forward_time": 1.7291033267974854,
      "step": 41621
    },
    {
      "epoch": 3.969287872314453e-05,
      "step": 41621,
      "training_step_time": 7.6774351596832275
    },
    {
      "epoch": 3.969383239746094e-05,
      "model_forward_time": 1.7273280620574951,
      "step": 41622
    },
    {
      "epoch": 3.969383239746094e-05,
      "step": 41622,
      "training_step_time": 7.828577280044556
    },
    {
      "epoch": 3.969478607177734e-05,
      "model_forward_time": 1.7231366634368896,
      "step": 41623
    },
    {
      "epoch": 3.969478607177734e-05,
      "step": 41623,
      "training_step_time": 9.490496397018433
    },
    {
      "epoch": 3.969573974609375e-05,
      "model_forward_time": 2.9156885147094727,
      "step": 41624
    },
    {
      "epoch": 3.969573974609375e-05,
      "step": 41624,
      "training_step_time": 8.786302328109741
    },
    {
      "epoch": 3.9696693420410156e-05,
      "model_forward_time": 1.8292667865753174,
      "step": 41625
    },
    {
      "epoch": 3.9696693420410156e-05,
      "step": 41625,
      "training_step_time": 9.530446529388428
    },
    {
      "epoch": 3.969764709472656e-05,
      "model_forward_time": 1.720341444015503,
      "step": 41626
    },
    {
      "epoch": 3.969764709472656e-05,
      "step": 41626,
      "training_step_time": 8.695107221603394
    },
    {
      "epoch": 3.969860076904297e-05,
      "model_forward_time": 1.7186236381530762,
      "step": 41627
    },
    {
      "epoch": 3.969860076904297e-05,
      "step": 41627,
      "training_step_time": 7.631122350692749
    },
    {
      "epoch": 3.9699554443359375e-05,
      "model_forward_time": 1.7261807918548584,
      "step": 41628
    },
    {
      "epoch": 3.9699554443359375e-05,
      "step": 41628,
      "training_step_time": 7.795253753662109
    },
    {
      "epoch": 3.970050811767578e-05,
      "model_forward_time": 1.7167243957519531,
      "step": 41629
    },
    {
      "epoch": 3.970050811767578e-05,
      "step": 41629,
      "training_step_time": 7.542832612991333
    },
    {
      "epoch": 3.970146179199219e-05,
      "grad_norm": 0.07934778928756714,
      "learning_rate": 6.759019071607851e-06,
      "loss": 0.0582,
      "step": 41630
    },
    {
      "epoch": 3.970146179199219e-05,
      "model_forward_time": 1.7177057266235352,
      "step": 41630
    },
    {
      "dynamics_loss_avg": 0.06334971264004707,
      "epoch": 3.970146179199219e-05,
      "step": 41630
    },
    {
      "action_loss_avg": 0.0020574329362716528,
      "epoch": 3.970146179199219e-05,
      "step": 41630
    },
    {
      "epoch": 3.970146179199219e-05,
      "step": 41630,
      "training_step_time": 7.594235420227051
    },
    {
      "epoch": 3.9702415466308594e-05,
      "model_forward_time": 1.3676769733428955,
      "step": 41631
    },
    {
      "epoch": 3.9702415466308594e-05,
      "step": 41631,
      "training_step_time": 7.761892080307007
    },
    {
      "epoch": 3.9703369140625e-05,
      "model_forward_time": 1.7228903770446777,
      "step": 41632
    },
    {
      "epoch": 3.9703369140625e-05,
      "step": 41632,
      "training_step_time": 10.389726877212524
    },
    {
      "epoch": 3.970432281494141e-05,
      "model_forward_time": 2.870046377182007,
      "step": 41633
    },
    {
      "epoch": 3.970432281494141e-05,
      "step": 41633,
      "training_step_time": 8.781412363052368
    },
    {
      "epoch": 3.970527648925781e-05,
      "model_forward_time": 1.8277385234832764,
      "step": 41634
    },
    {
      "epoch": 3.970527648925781e-05,
      "step": 41634,
      "training_step_time": 8.030041933059692
    },
    {
      "epoch": 3.970623016357422e-05,
      "model_forward_time": 1.7340350151062012,
      "step": 41635
    },
    {
      "epoch": 3.970623016357422e-05,
      "step": 41635,
      "training_step_time": 7.771137714385986
    },
    {
      "epoch": 3.970718383789063e-05,
      "model_forward_time": 1.7195408344268799,
      "step": 41636
    },
    {
      "epoch": 3.970718383789063e-05,
      "step": 41636,
      "training_step_time": 7.567500352859497
    },
    {
      "epoch": 3.970813751220703e-05,
      "model_forward_time": 1.7255825996398926,
      "step": 41637
    },
    {
      "epoch": 3.970813751220703e-05,
      "step": 41637,
      "training_step_time": 7.691441774368286
    },
    {
      "epoch": 3.9709091186523436e-05,
      "model_forward_time": 1.7289175987243652,
      "step": 41638
    },
    {
      "epoch": 3.9709091186523436e-05,
      "step": 41638,
      "training_step_time": 7.66938042640686
    },
    {
      "epoch": 3.9710044860839847e-05,
      "model_forward_time": 1.7216825485229492,
      "step": 41639
    },
    {
      "epoch": 3.9710044860839847e-05,
      "step": 41639,
      "training_step_time": 7.596285343170166
    },
    {
      "epoch": 3.971099853515625e-05,
      "grad_norm": 0.12171072512865067,
      "learning_rate": 6.757471205751944e-06,
      "loss": 0.0617,
      "step": 41640
    },
    {
      "epoch": 3.971099853515625e-05,
      "model_forward_time": 1.7179889678955078,
      "step": 41640
    },
    {
      "dynamics_loss_avg": 0.05208919160068035,
      "epoch": 3.971099853515625e-05,
      "step": 41640
    },
    {
      "action_loss_avg": 0.0010259101225528866,
      "epoch": 3.971099853515625e-05,
      "step": 41640
    },
    {
      "epoch": 3.971099853515625e-05,
      "step": 41640,
      "training_step_time": 7.734410762786865
    },
    {
      "epoch": 3.9711952209472655e-05,
      "model_forward_time": 1.7369499206542969,
      "step": 41641
    },
    {
      "epoch": 3.9711952209472655e-05,
      "step": 41641,
      "training_step_time": 8.347724437713623
    },
    {
      "epoch": 3.9712905883789065e-05,
      "model_forward_time": 2.83821702003479,
      "step": 41642
    },
    {
      "epoch": 3.9712905883789065e-05,
      "step": 41642,
      "training_step_time": 8.247137784957886
    },
    {
      "epoch": 3.971385955810547e-05,
      "model_forward_time": 1.8290972709655762,
      "step": 41643
    },
    {
      "epoch": 3.971385955810547e-05,
      "step": 41643,
      "training_step_time": 8.778507947921753
    },
    {
      "epoch": 3.9714813232421873e-05,
      "model_forward_time": 1.728071928024292,
      "step": 41644
    },
    {
      "epoch": 3.9714813232421873e-05,
      "step": 41644,
      "training_step_time": 8.939810276031494
    },
    {
      "epoch": 3.9715766906738284e-05,
      "model_forward_time": 1.7203891277313232,
      "step": 41645
    },
    {
      "epoch": 3.9715766906738284e-05,
      "step": 41645,
      "training_step_time": 7.586831331253052
    },
    {
      "epoch": 3.971672058105469e-05,
      "model_forward_time": 1.7500548362731934,
      "step": 41646
    },
    {
      "epoch": 3.971672058105469e-05,
      "step": 41646,
      "training_step_time": 7.671219348907471
    },
    {
      "epoch": 3.971767425537109e-05,
      "model_forward_time": 1.7180819511413574,
      "step": 41647
    },
    {
      "epoch": 3.971767425537109e-05,
      "step": 41647,
      "training_step_time": 8.496445178985596
    },
    {
      "epoch": 3.97186279296875e-05,
      "model_forward_time": 1.734239101409912,
      "step": 41648
    },
    {
      "epoch": 3.97186279296875e-05,
      "step": 41648,
      "training_step_time": 8.588493347167969
    },
    {
      "epoch": 3.971958160400391e-05,
      "model_forward_time": 1.7154972553253174,
      "step": 41649
    },
    {
      "epoch": 3.971958160400391e-05,
      "step": 41649,
      "training_step_time": 8.352294921875
    },
    {
      "epoch": 3.972053527832031e-05,
      "grad_norm": 0.1435653120279312,
      "learning_rate": 6.755923147701631e-06,
      "loss": 0.0636,
      "step": 41650
    },
    {
      "epoch": 3.972053527832031e-05,
      "model_forward_time": 1.7256391048431396,
      "step": 41650
    },
    {
      "dynamics_loss_avg": 0.07232625037431717,
      "epoch": 3.972053527832031e-05,
      "step": 41650
    },
    {
      "action_loss_avg": 0.002715619560331106,
      "epoch": 3.972053527832031e-05,
      "step": 41650
    },
    {
      "epoch": 3.972053527832031e-05,
      "step": 41650,
      "training_step_time": 8.76115870475769
    },
    {
      "epoch": 3.972148895263672e-05,
      "model_forward_time": 1.7240204811096191,
      "step": 41651
    },
    {
      "epoch": 3.972148895263672e-05,
      "step": 41651,
      "training_step_time": 10.304551839828491
    },
    {
      "epoch": 3.9722442626953126e-05,
      "model_forward_time": 1.84456205368042,
      "step": 41652
    },
    {
      "epoch": 3.9722442626953126e-05,
      "step": 41652,
      "training_step_time": 7.959330081939697
    },
    {
      "epoch": 3.972339630126953e-05,
      "model_forward_time": 1.718445062637329,
      "step": 41653
    },
    {
      "epoch": 3.972339630126953e-05,
      "step": 41653,
      "training_step_time": 8.71555233001709
    },
    {
      "epoch": 3.972434997558594e-05,
      "model_forward_time": 1.7292447090148926,
      "step": 41654
    },
    {
      "epoch": 3.972434997558594e-05,
      "step": 41654,
      "training_step_time": 7.573218822479248
    },
    {
      "epoch": 3.9725303649902345e-05,
      "model_forward_time": 1.722179651260376,
      "step": 41655
    },
    {
      "epoch": 3.9725303649902345e-05,
      "step": 41655,
      "training_step_time": 7.660635471343994
    },
    {
      "epoch": 3.972625732421875e-05,
      "model_forward_time": 1.7230346202850342,
      "step": 41656
    },
    {
      "epoch": 3.972625732421875e-05,
      "step": 41656,
      "training_step_time": 7.324538230895996
    },
    {
      "epoch": 3.972721099853515e-05,
      "model_forward_time": 1.7384259700775146,
      "step": 41657
    },
    {
      "epoch": 3.972721099853515e-05,
      "step": 41657,
      "training_step_time": 8.675141096115112
    },
    {
      "epoch": 3.9728164672851564e-05,
      "model_forward_time": 1.725245475769043,
      "step": 41658
    },
    {
      "epoch": 3.9728164672851564e-05,
      "step": 41658,
      "training_step_time": 7.518221139907837
    },
    {
      "epoch": 3.972911834716797e-05,
      "model_forward_time": 1.7206780910491943,
      "step": 41659
    },
    {
      "epoch": 3.972911834716797e-05,
      "step": 41659,
      "training_step_time": 7.819812536239624
    },
    {
      "epoch": 3.973007202148437e-05,
      "grad_norm": 0.1263718605041504,
      "learning_rate": 6.7543748976262e-06,
      "loss": 0.0656,
      "step": 41660
    },
    {
      "epoch": 3.973007202148437e-05,
      "model_forward_time": 1.7320168018341064,
      "step": 41660
    },
    {
      "dynamics_loss_avg": 0.06573774944990873,
      "epoch": 3.973007202148437e-05,
      "step": 41660
    },
    {
      "action_loss_avg": 0.0019441079610260203,
      "epoch": 3.973007202148437e-05,
      "step": 41660
    },
    {
      "epoch": 3.973007202148437e-05,
      "step": 41660,
      "training_step_time": 10.42661428451538
    },
    {
      "epoch": 3.973102569580078e-05,
      "model_forward_time": 1.8382294178009033,
      "step": 41661
    },
    {
      "epoch": 3.973102569580078e-05,
      "step": 41661,
      "training_step_time": 9.01823377609253
    },
    {
      "epoch": 3.9731979370117187e-05,
      "model_forward_time": 1.72282075881958,
      "step": 41662
    },
    {
      "epoch": 3.9731979370117187e-05,
      "step": 41662,
      "training_step_time": 7.837002754211426
    },
    {
      "epoch": 3.973293304443359e-05,
      "model_forward_time": 1.7223670482635498,
      "step": 41663
    },
    {
      "epoch": 3.973293304443359e-05,
      "step": 41663,
      "training_step_time": 7.751049757003784
    },
    {
      "epoch": 3.973388671875e-05,
      "model_forward_time": 1.7394301891326904,
      "step": 41664
    },
    {
      "epoch": 3.973388671875e-05,
      "step": 41664,
      "training_step_time": 7.701981544494629
    },
    {
      "epoch": 3.9734840393066405e-05,
      "model_forward_time": 1.7220466136932373,
      "step": 41665
    },
    {
      "epoch": 3.9734840393066405e-05,
      "step": 41665,
      "training_step_time": 8.265640020370483
    },
    {
      "epoch": 3.973579406738281e-05,
      "model_forward_time": 1.715113639831543,
      "step": 41666
    },
    {
      "epoch": 3.973579406738281e-05,
      "step": 41666,
      "training_step_time": 7.712674379348755
    },
    {
      "epoch": 3.973674774169922e-05,
      "model_forward_time": 1.734673261642456,
      "step": 41667
    },
    {
      "epoch": 3.973674774169922e-05,
      "step": 41667,
      "training_step_time": 7.506171226501465
    },
    {
      "epoch": 3.9737701416015624e-05,
      "model_forward_time": 1.7341208457946777,
      "step": 41668
    },
    {
      "epoch": 3.9737701416015624e-05,
      "step": 41668,
      "training_step_time": 8.685565710067749
    },
    {
      "epoch": 3.973865509033203e-05,
      "model_forward_time": 1.724134922027588,
      "step": 41669
    },
    {
      "epoch": 3.973865509033203e-05,
      "step": 41669,
      "training_step_time": 8.782192945480347
    },
    {
      "epoch": 3.973960876464844e-05,
      "grad_norm": 0.12234925478696823,
      "learning_rate": 6.752826455694968e-06,
      "loss": 0.0607,
      "step": 41670
    },
    {
      "epoch": 3.973960876464844e-05,
      "model_forward_time": 1.828183889389038,
      "step": 41670
    },
    {
      "dynamics_loss_avg": 0.0650996008887887,
      "epoch": 3.973960876464844e-05,
      "step": 41670
    },
    {
      "action_loss_avg": 0.002601478388532996,
      "epoch": 3.973960876464844e-05,
      "step": 41670
    },
    {
      "epoch": 3.973960876464844e-05,
      "step": 41670,
      "training_step_time": 8.592406511306763
    },
    {
      "epoch": 3.974056243896484e-05,
      "model_forward_time": 1.741014003753662,
      "step": 41671
    },
    {
      "epoch": 3.974056243896484e-05,
      "step": 41671,
      "training_step_time": 9.130037069320679
    },
    {
      "epoch": 3.974151611328125e-05,
      "model_forward_time": 1.733288049697876,
      "step": 41672
    },
    {
      "epoch": 3.974151611328125e-05,
      "step": 41672,
      "training_step_time": 7.797589302062988
    },
    {
      "epoch": 3.974246978759766e-05,
      "model_forward_time": 1.7287757396697998,
      "step": 41673
    },
    {
      "epoch": 3.974246978759766e-05,
      "step": 41673,
      "training_step_time": 7.720097064971924
    },
    {
      "epoch": 3.974342346191406e-05,
      "model_forward_time": 1.7327001094818115,
      "step": 41674
    },
    {
      "epoch": 3.974342346191406e-05,
      "step": 41674,
      "training_step_time": 7.783633232116699
    },
    {
      "epoch": 3.9744377136230466e-05,
      "model_forward_time": 1.7113680839538574,
      "step": 41675
    },
    {
      "epoch": 3.9744377136230466e-05,
      "step": 41675,
      "training_step_time": 7.4984636306762695
    },
    {
      "epoch": 3.974533081054688e-05,
      "model_forward_time": 1.7094230651855469,
      "step": 41676
    },
    {
      "epoch": 3.974533081054688e-05,
      "step": 41676,
      "training_step_time": 7.537358045578003
    },
    {
      "epoch": 3.974628448486328e-05,
      "model_forward_time": 1.731905221939087,
      "step": 41677
    },
    {
      "epoch": 3.974628448486328e-05,
      "step": 41677,
      "training_step_time": 9.185853719711304
    },
    {
      "epoch": 3.9747238159179685e-05,
      "model_forward_time": 1.717926025390625,
      "step": 41678
    },
    {
      "epoch": 3.9747238159179685e-05,
      "step": 41678,
      "training_step_time": 7.661268711090088
    },
    {
      "epoch": 3.9748191833496096e-05,
      "model_forward_time": 2.834165334701538,
      "step": 41679
    },
    {
      "epoch": 3.9748191833496096e-05,
      "step": 41679,
      "training_step_time": 7.856671333312988
    },
    {
      "epoch": 3.97491455078125e-05,
      "grad_norm": 0.1311236470937729,
      "learning_rate": 6.751277822077271e-06,
      "loss": 0.0618,
      "step": 41680
    },
    {
      "epoch": 3.97491455078125e-05,
      "model_forward_time": 1.7197394371032715,
      "step": 41680
    },
    {
      "dynamics_loss_avg": 0.056215477827936414,
      "epoch": 3.97491455078125e-05,
      "step": 41680
    },
    {
      "action_loss_avg": 0.002131746234954335,
      "epoch": 3.97491455078125e-05,
      "step": 41680
    },
    {
      "epoch": 3.97491455078125e-05,
      "step": 41680,
      "training_step_time": 8.892413854598999
    },
    {
      "epoch": 3.9750099182128904e-05,
      "model_forward_time": 1.7183620929718018,
      "step": 41681
    },
    {
      "epoch": 3.9750099182128904e-05,
      "step": 41681,
      "training_step_time": 7.754190444946289
    },
    {
      "epoch": 3.9751052856445315e-05,
      "model_forward_time": 1.7241404056549072,
      "step": 41682
    },
    {
      "epoch": 3.9751052856445315e-05,
      "step": 41682,
      "training_step_time": 7.7738165855407715
    },
    {
      "epoch": 3.975200653076172e-05,
      "model_forward_time": 1.7113642692565918,
      "step": 41683
    },
    {
      "epoch": 3.975200653076172e-05,
      "step": 41683,
      "training_step_time": 8.532575368881226
    },
    {
      "epoch": 3.975296020507812e-05,
      "model_forward_time": 1.7176728248596191,
      "step": 41684
    },
    {
      "epoch": 3.975296020507812e-05,
      "step": 41684,
      "training_step_time": 7.503127098083496
    },
    {
      "epoch": 3.9753913879394533e-05,
      "model_forward_time": 1.7239699363708496,
      "step": 41685
    },
    {
      "epoch": 3.9753913879394533e-05,
      "step": 41685,
      "training_step_time": 9.305979251861572
    },
    {
      "epoch": 3.975486755371094e-05,
      "model_forward_time": 1.718566656112671,
      "step": 41686
    },
    {
      "epoch": 3.975486755371094e-05,
      "step": 41686,
      "training_step_time": 8.325500249862671
    },
    {
      "epoch": 3.975582122802734e-05,
      "model_forward_time": 1.7256183624267578,
      "step": 41687
    },
    {
      "epoch": 3.975582122802734e-05,
      "step": 41687,
      "training_step_time": 8.478654146194458
    },
    {
      "epoch": 3.975677490234375e-05,
      "model_forward_time": 2.8144187927246094,
      "step": 41688
    },
    {
      "epoch": 3.975677490234375e-05,
      "step": 41688,
      "training_step_time": 7.827812194824219
    },
    {
      "epoch": 3.9757728576660156e-05,
      "model_forward_time": 1.7230372428894043,
      "step": 41689
    },
    {
      "epoch": 3.9757728576660156e-05,
      "step": 41689,
      "training_step_time": 7.854033946990967
    },
    {
      "epoch": 3.975868225097656e-05,
      "grad_norm": 0.1134345531463623,
      "learning_rate": 6.749728996942465e-06,
      "loss": 0.0597,
      "step": 41690
    },
    {
      "epoch": 3.975868225097656e-05,
      "model_forward_time": 1.0114457607269287,
      "step": 41690
    },
    {
      "dynamics_loss_avg": 0.057942182570695874,
      "epoch": 3.975868225097656e-05,
      "step": 41690
    },
    {
      "action_loss_avg": 0.0015869308524997904,
      "epoch": 3.975868225097656e-05,
      "step": 41690
    },
    {
      "epoch": 3.975868225097656e-05,
      "step": 41690,
      "training_step_time": 8.421808242797852
    },
    {
      "epoch": 3.975963592529297e-05,
      "model_forward_time": 1.7175791263580322,
      "step": 41691
    },
    {
      "epoch": 3.975963592529297e-05,
      "step": 41691,
      "training_step_time": 8.61201548576355
    },
    {
      "epoch": 3.9760589599609375e-05,
      "model_forward_time": 1.7165427207946777,
      "step": 41692
    },
    {
      "epoch": 3.9760589599609375e-05,
      "step": 41692,
      "training_step_time": 8.516711950302124
    },
    {
      "epoch": 3.976154327392578e-05,
      "model_forward_time": 1.718644380569458,
      "step": 41693
    },
    {
      "epoch": 3.976154327392578e-05,
      "step": 41693,
      "training_step_time": 7.430587291717529
    },
    {
      "epoch": 3.976249694824219e-05,
      "model_forward_time": 1.7206223011016846,
      "step": 41694
    },
    {
      "epoch": 3.976249694824219e-05,
      "step": 41694,
      "training_step_time": 8.118140459060669
    },
    {
      "epoch": 3.9763450622558594e-05,
      "model_forward_time": 1.7158188819885254,
      "step": 41695
    },
    {
      "epoch": 3.9763450622558594e-05,
      "step": 41695,
      "training_step_time": 8.550449132919312
    },
    {
      "epoch": 3.9764404296875e-05,
      "model_forward_time": 1.7080943584442139,
      "step": 41696
    },
    {
      "epoch": 3.9764404296875e-05,
      "step": 41696,
      "training_step_time": 7.668076038360596
    },
    {
      "epoch": 3.976535797119141e-05,
      "model_forward_time": 2.860138177871704,
      "step": 41697
    },
    {
      "epoch": 3.976535797119141e-05,
      "step": 41697,
      "training_step_time": 7.806106805801392
    },
    {
      "epoch": 3.976631164550781e-05,
      "model_forward_time": 1.7141058444976807,
      "step": 41698
    },
    {
      "epoch": 3.976631164550781e-05,
      "step": 41698,
      "training_step_time": 7.784365653991699
    },
    {
      "epoch": 3.976726531982422e-05,
      "model_forward_time": 1.7157812118530273,
      "step": 41699
    },
    {
      "epoch": 3.976726531982422e-05,
      "step": 41699,
      "training_step_time": 7.905239105224609
    },
    {
      "epoch": 3.976821899414063e-05,
      "grad_norm": 0.12583567202091217,
      "learning_rate": 6.748179980459924e-06,
      "loss": 0.0587,
      "step": 41700
    },
    {
      "epoch": 3.976821899414063e-05,
      "model_forward_time": 1.718740701675415,
      "step": 41700
    },
    {
      "dynamics_loss_avg": 0.062262572906911375,
      "epoch": 3.976821899414063e-05,
      "step": 41700
    },
    {
      "action_loss_avg": 0.0038272384845186025,
      "epoch": 3.976821899414063e-05,
      "step": 41700
    },
    {
      "epoch": 3.976821899414063e-05,
      "step": 41700,
      "training_step_time": 8.22607135772705
    },
    {
      "epoch": 3.976917266845703e-05,
      "model_forward_time": 1.7072322368621826,
      "step": 41701
    },
    {
      "epoch": 3.976917266845703e-05,
      "step": 41701,
      "training_step_time": 8.406175136566162
    },
    {
      "epoch": 3.9770126342773436e-05,
      "model_forward_time": 1.7179737091064453,
      "step": 41702
    },
    {
      "epoch": 3.9770126342773436e-05,
      "step": 41702,
      "training_step_time": 7.319589376449585
    },
    {
      "epoch": 3.9771080017089847e-05,
      "model_forward_time": 1.741201639175415,
      "step": 41703
    },
    {
      "epoch": 3.9771080017089847e-05,
      "step": 41703,
      "training_step_time": 7.689178705215454
    },
    {
      "epoch": 3.977203369140625e-05,
      "model_forward_time": 1.709404706954956,
      "step": 41704
    },
    {
      "epoch": 3.977203369140625e-05,
      "step": 41704,
      "training_step_time": 8.165021657943726
    },
    {
      "epoch": 3.9772987365722655e-05,
      "model_forward_time": 1.753014326095581,
      "step": 41705
    },
    {
      "epoch": 3.9772987365722655e-05,
      "step": 41705,
      "training_step_time": 7.622463703155518
    },
    {
      "epoch": 3.9773941040039065e-05,
      "model_forward_time": 2.905371904373169,
      "step": 41706
    },
    {
      "epoch": 3.9773941040039065e-05,
      "step": 41706,
      "training_step_time": 7.623336553573608
    },
    {
      "epoch": 3.977489471435547e-05,
      "model_forward_time": 1.7147669792175293,
      "step": 41707
    },
    {
      "epoch": 3.977489471435547e-05,
      "step": 41707,
      "training_step_time": 8.612908363342285
    },
    {
      "epoch": 3.9775848388671873e-05,
      "model_forward_time": 1.7325005531311035,
      "step": 41708
    },
    {
      "epoch": 3.9775848388671873e-05,
      "step": 41708,
      "training_step_time": 7.903567790985107
    },
    {
      "epoch": 3.9776802062988284e-05,
      "model_forward_time": 1.7232000827789307,
      "step": 41709
    },
    {
      "epoch": 3.9776802062988284e-05,
      "step": 41709,
      "training_step_time": 7.612574815750122
    },
    {
      "epoch": 3.977775573730469e-05,
      "grad_norm": 0.3794170916080475,
      "learning_rate": 6.746630772799052e-06,
      "loss": 0.0599,
      "step": 41710
    },
    {
      "epoch": 3.977775573730469e-05,
      "model_forward_time": 1.7143347263336182,
      "step": 41710
    },
    {
      "dynamics_loss_avg": 0.05056934971362352,
      "epoch": 3.977775573730469e-05,
      "step": 41710
    },
    {
      "action_loss_avg": 0.002967750700190663,
      "epoch": 3.977775573730469e-05,
      "step": 41710
    },
    {
      "epoch": 3.977775573730469e-05,
      "step": 41710,
      "training_step_time": 7.634156703948975
    },
    {
      "epoch": 3.977870941162109e-05,
      "model_forward_time": 1.723799467086792,
      "step": 41711
    },
    {
      "epoch": 3.977870941162109e-05,
      "step": 41711,
      "training_step_time": 8.398744821548462
    },
    {
      "epoch": 3.97796630859375e-05,
      "model_forward_time": 1.7208147048950195,
      "step": 41712
    },
    {
      "epoch": 3.97796630859375e-05,
      "step": 41712,
      "training_step_time": 7.59749960899353
    },
    {
      "epoch": 3.978061676025391e-05,
      "model_forward_time": 1.7146646976470947,
      "step": 41713
    },
    {
      "epoch": 3.978061676025391e-05,
      "step": 41713,
      "training_step_time": 8.127197742462158
    },
    {
      "epoch": 3.978157043457031e-05,
      "model_forward_time": 1.7243390083312988,
      "step": 41714
    },
    {
      "epoch": 3.978157043457031e-05,
      "step": 41714,
      "training_step_time": 10.762501955032349
    },
    {
      "epoch": 3.978252410888672e-05,
      "model_forward_time": 1.7197346687316895,
      "step": 41715
    },
    {
      "epoch": 3.978252410888672e-05,
      "step": 41715,
      "training_step_time": 8.624810457229614
    },
    {
      "epoch": 3.9783477783203126e-05,
      "model_forward_time": 1.7049791812896729,
      "step": 41716
    },
    {
      "epoch": 3.9783477783203126e-05,
      "step": 41716,
      "training_step_time": 6.753269195556641
    },
    {
      "epoch": 3.978443145751953e-05,
      "model_forward_time": 1.726189136505127,
      "step": 41717
    },
    {
      "epoch": 3.978443145751953e-05,
      "step": 41717,
      "training_step_time": 6.680308103561401
    },
    {
      "epoch": 3.978538513183594e-05,
      "model_forward_time": 1.3511481285095215,
      "step": 41718
    },
    {
      "epoch": 3.978538513183594e-05,
      "step": 41718,
      "training_step_time": 6.630166292190552
    },
    {
      "epoch": 3.9786338806152345e-05,
      "model_forward_time": 1.7170991897583008,
      "step": 41719
    },
    {
      "epoch": 3.9786338806152345e-05,
      "step": 41719,
      "training_step_time": 6.7334582805633545
    },
    {
      "epoch": 3.978729248046875e-05,
      "grad_norm": 0.11817611753940582,
      "learning_rate": 6.745081374129263e-06,
      "loss": 0.0606,
      "step": 41720
    },
    {
      "epoch": 3.978729248046875e-05,
      "model_forward_time": 1.7126870155334473,
      "step": 41720
    },
    {
      "dynamics_loss_avg": 0.07715853098779916,
      "epoch": 3.978729248046875e-05,
      "step": 41720
    },
    {
      "action_loss_avg": 0.002416411298327148,
      "epoch": 3.978729248046875e-05,
      "step": 41720
    },
    {
      "epoch": 3.978729248046875e-05,
      "step": 41720,
      "training_step_time": 6.75579833984375
    },
    {
      "epoch": 3.978824615478515e-05,
      "model_forward_time": 1.7088332176208496,
      "step": 41721
    },
    {
      "epoch": 3.978824615478515e-05,
      "step": 41721,
      "training_step_time": 6.674018144607544
    },
    {
      "epoch": 3.9789199829101564e-05,
      "model_forward_time": 1.7064712047576904,
      "step": 41722
    },
    {
      "epoch": 3.9789199829101564e-05,
      "step": 41722,
      "training_step_time": 6.736369371414185
    },
    {
      "epoch": 3.979015350341797e-05,
      "model_forward_time": 1.706023931503296,
      "step": 41723
    },
    {
      "epoch": 3.979015350341797e-05,
      "step": 41723,
      "training_step_time": 7.6223320960998535
    },
    {
      "epoch": 3.979110717773437e-05,
      "model_forward_time": 2.87968111038208,
      "step": 41724
    },
    {
      "epoch": 3.979110717773437e-05,
      "step": 41724,
      "training_step_time": 9.715953588485718
    },
    {
      "epoch": 3.979206085205078e-05,
      "model_forward_time": 1.7059190273284912,
      "step": 41725
    },
    {
      "epoch": 3.979206085205078e-05,
      "step": 41725,
      "training_step_time": 7.8052589893341064
    },
    {
      "epoch": 3.979301452636719e-05,
      "model_forward_time": 1.706390619277954,
      "step": 41726
    },
    {
      "epoch": 3.979301452636719e-05,
      "step": 41726,
      "training_step_time": 6.76675009727478
    },
    {
      "epoch": 3.979396820068359e-05,
      "model_forward_time": 1.7158257961273193,
      "step": 41727
    },
    {
      "epoch": 3.979396820068359e-05,
      "step": 41727,
      "training_step_time": 6.73989725112915
    },
    {
      "epoch": 3.9794921875e-05,
      "model_forward_time": 1.712003231048584,
      "step": 41728
    },
    {
      "epoch": 3.9794921875e-05,
      "step": 41728,
      "training_step_time": 6.762727975845337
    },
    {
      "epoch": 3.9795875549316405e-05,
      "model_forward_time": 1.7098474502563477,
      "step": 41729
    },
    {
      "epoch": 3.9795875549316405e-05,
      "step": 41729,
      "training_step_time": 6.669450521469116
    },
    {
      "epoch": 3.979682922363281e-05,
      "grad_norm": 0.13775283098220825,
      "learning_rate": 6.7435317846199985e-06,
      "loss": 0.0599,
      "step": 41730
    },
    {
      "epoch": 3.979682922363281e-05,
      "model_forward_time": 1.7091708183288574,
      "step": 41730
    },
    {
      "dynamics_loss_avg": 0.06004734262824059,
      "epoch": 3.979682922363281e-05,
      "step": 41730
    },
    {
      "action_loss_avg": 0.0025880688976030798,
      "epoch": 3.979682922363281e-05,
      "step": 41730
    },
    {
      "epoch": 3.979682922363281e-05,
      "step": 41730,
      "training_step_time": 6.7298078536987305
    },
    {
      "epoch": 3.979778289794922e-05,
      "model_forward_time": 1.714353322982788,
      "step": 41731
    },
    {
      "epoch": 3.979778289794922e-05,
      "step": 41731,
      "training_step_time": 6.741694211959839
    },
    {
      "epoch": 3.9798736572265624e-05,
      "model_forward_time": 1.7121670246124268,
      "step": 41732
    },
    {
      "epoch": 3.9798736572265624e-05,
      "step": 41732,
      "training_step_time": 6.702401638031006
    },
    {
      "epoch": 3.979969024658203e-05,
      "model_forward_time": 2.8376383781433105,
      "step": 41733
    },
    {
      "epoch": 3.979969024658203e-05,
      "step": 41733,
      "training_step_time": 7.581432104110718
    },
    {
      "epoch": 3.980064392089844e-05,
      "model_forward_time": 1.7083346843719482,
      "step": 41734
    },
    {
      "epoch": 3.980064392089844e-05,
      "step": 41734,
      "training_step_time": 7.777116298675537
    },
    {
      "epoch": 3.980159759521484e-05,
      "model_forward_time": 1.7059648036956787,
      "step": 41735
    },
    {
      "epoch": 3.980159759521484e-05,
      "step": 41735,
      "training_step_time": 6.675273180007935
    },
    {
      "epoch": 3.980255126953125e-05,
      "model_forward_time": 1.7118942737579346,
      "step": 41736
    },
    {
      "epoch": 3.980255126953125e-05,
      "step": 41736,
      "training_step_time": 6.744815826416016
    },
    {
      "epoch": 3.980350494384766e-05,
      "model_forward_time": 1.7062397003173828,
      "step": 41737
    },
    {
      "epoch": 3.980350494384766e-05,
      "step": 41737,
      "training_step_time": 6.722833633422852
    },
    {
      "epoch": 3.980445861816406e-05,
      "model_forward_time": 1.717618703842163,
      "step": 41738
    },
    {
      "epoch": 3.980445861816406e-05,
      "step": 41738,
      "training_step_time": 6.73903751373291
    },
    {
      "epoch": 3.9805412292480466e-05,
      "model_forward_time": 1.6971909999847412,
      "step": 41739
    },
    {
      "epoch": 3.9805412292480466e-05,
      "step": 41739,
      "training_step_time": 6.629278182983398
    },
    {
      "epoch": 3.980636596679688e-05,
      "grad_norm": 0.12519647181034088,
      "learning_rate": 6.741982004440719e-06,
      "loss": 0.0629,
      "step": 41740
    },
    {
      "epoch": 3.980636596679688e-05,
      "model_forward_time": 1.707137107849121,
      "step": 41740
    },
    {
      "dynamics_loss_avg": 0.05939827263355255,
      "epoch": 3.980636596679688e-05,
      "step": 41740
    },
    {
      "action_loss_avg": 0.0032774297724245116,
      "epoch": 3.980636596679688e-05,
      "step": 41740
    },
    {
      "epoch": 3.980636596679688e-05,
      "step": 41740,
      "training_step_time": 6.7410969734191895
    },
    {
      "epoch": 3.980731964111328e-05,
      "model_forward_time": 1.7092797756195068,
      "step": 41741
    },
    {
      "epoch": 3.980731964111328e-05,
      "step": 41741,
      "training_step_time": 6.775969505310059
    },
    {
      "epoch": 3.9808273315429685e-05,
      "model_forward_time": 2.829310178756714,
      "step": 41742
    },
    {
      "epoch": 3.9808273315429685e-05,
      "step": 41742,
      "training_step_time": 8.586713075637817
    },
    {
      "epoch": 3.9809226989746096e-05,
      "model_forward_time": 1.715646505355835,
      "step": 41743
    },
    {
      "epoch": 3.9809226989746096e-05,
      "step": 41743,
      "training_step_time": 8.849305391311646
    },
    {
      "epoch": 3.98101806640625e-05,
      "model_forward_time": 1.7014641761779785,
      "step": 41744
    },
    {
      "epoch": 3.98101806640625e-05,
      "step": 41744,
      "training_step_time": 7.613520383834839
    },
    {
      "epoch": 3.9811134338378904e-05,
      "model_forward_time": 1.7205877304077148,
      "step": 41745
    },
    {
      "epoch": 3.9811134338378904e-05,
      "step": 41745,
      "training_step_time": 6.754743814468384
    },
    {
      "epoch": 3.9812088012695315e-05,
      "model_forward_time": 1.709596872329712,
      "step": 41746
    },
    {
      "epoch": 3.9812088012695315e-05,
      "step": 41746,
      "training_step_time": 6.734087944030762
    },
    {
      "epoch": 3.981304168701172e-05,
      "model_forward_time": 1.7110893726348877,
      "step": 41747
    },
    {
      "epoch": 3.981304168701172e-05,
      "step": 41747,
      "training_step_time": 6.737768650054932
    },
    {
      "epoch": 3.981399536132812e-05,
      "model_forward_time": 1.3472299575805664,
      "step": 41748
    },
    {
      "epoch": 3.981399536132812e-05,
      "step": 41748,
      "training_step_time": 6.698720216751099
    },
    {
      "epoch": 3.9814949035644533e-05,
      "model_forward_time": 1.7071702480316162,
      "step": 41749
    },
    {
      "epoch": 3.9814949035644533e-05,
      "step": 41749,
      "training_step_time": 6.69965934753418
    },
    {
      "epoch": 3.981590270996094e-05,
      "grad_norm": 0.09862009435892105,
      "learning_rate": 6.740432033760908e-06,
      "loss": 0.0607,
      "step": 41750
    },
    {
      "epoch": 3.981590270996094e-05,
      "model_forward_time": 1.727940320968628,
      "step": 41750
    },
    {
      "dynamics_loss_avg": 0.05163883697241545,
      "epoch": 3.981590270996094e-05,
      "step": 41750
    },
    {
      "action_loss_avg": 0.002389604347990826,
      "epoch": 3.981590270996094e-05,
      "step": 41750
    },
    {
      "epoch": 3.981590270996094e-05,
      "step": 41750,
      "training_step_time": 6.740108489990234
    },
    {
      "epoch": 3.981685638427734e-05,
      "model_forward_time": 2.8450324535369873,
      "step": 41751
    },
    {
      "epoch": 3.981685638427734e-05,
      "step": 41751,
      "training_step_time": 7.580890655517578
    },
    {
      "epoch": 3.981781005859375e-05,
      "model_forward_time": 1.7240781784057617,
      "step": 41752
    },
    {
      "epoch": 3.981781005859375e-05,
      "step": 41752,
      "training_step_time": 9.240381002426147
    },
    {
      "epoch": 3.9818763732910156e-05,
      "model_forward_time": 1.7122015953063965,
      "step": 41753
    },
    {
      "epoch": 3.9818763732910156e-05,
      "step": 41753,
      "training_step_time": 7.703570365905762
    },
    {
      "epoch": 3.981971740722656e-05,
      "model_forward_time": 1.7078866958618164,
      "step": 41754
    },
    {
      "epoch": 3.981971740722656e-05,
      "step": 41754,
      "training_step_time": 6.69193172454834
    },
    {
      "epoch": 3.982067108154297e-05,
      "model_forward_time": 1.7091622352600098,
      "step": 41755
    },
    {
      "epoch": 3.982067108154297e-05,
      "step": 41755,
      "training_step_time": 6.721728563308716
    },
    {
      "epoch": 3.9821624755859375e-05,
      "model_forward_time": 1.7721905708312988,
      "step": 41756
    },
    {
      "epoch": 3.9821624755859375e-05,
      "step": 41756,
      "training_step_time": 6.714048147201538
    },
    {
      "epoch": 3.982257843017578e-05,
      "model_forward_time": 1.733292579650879,
      "step": 41757
    },
    {
      "epoch": 3.982257843017578e-05,
      "step": 41757,
      "training_step_time": 6.6964709758758545
    },
    {
      "epoch": 3.982353210449219e-05,
      "model_forward_time": 1.704310655593872,
      "step": 41758
    },
    {
      "epoch": 3.982353210449219e-05,
      "step": 41758,
      "training_step_time": 6.704414367675781
    },
    {
      "epoch": 3.9824485778808594e-05,
      "model_forward_time": 1.7195591926574707,
      "step": 41759
    },
    {
      "epoch": 3.9824485778808594e-05,
      "step": 41759,
      "training_step_time": 6.686770915985107
    },
    {
      "epoch": 3.9825439453125e-05,
      "grad_norm": 0.11492813378572464,
      "learning_rate": 6.738881872750066e-06,
      "loss": 0.0599,
      "step": 41760
    },
    {
      "epoch": 3.9825439453125e-05,
      "model_forward_time": 1.7925570011138916,
      "step": 41760
    },
    {
      "dynamics_loss_avg": 0.0605043089017272,
      "epoch": 3.9825439453125e-05,
      "step": 41760
    },
    {
      "action_loss_avg": 0.002738259092438966,
      "epoch": 3.9825439453125e-05,
      "step": 41760
    },
    {
      "epoch": 3.9825439453125e-05,
      "step": 41760,
      "training_step_time": 8.153140544891357
    },
    {
      "epoch": 3.982639312744141e-05,
      "model_forward_time": 1.822572946548462,
      "step": 41761
    },
    {
      "epoch": 3.982639312744141e-05,
      "step": 41761,
      "training_step_time": 9.281481266021729
    },
    {
      "epoch": 3.982734680175781e-05,
      "model_forward_time": 1.7386693954467773,
      "step": 41762
    },
    {
      "epoch": 3.982734680175781e-05,
      "step": 41762,
      "training_step_time": 7.757827520370483
    },
    {
      "epoch": 3.982830047607422e-05,
      "model_forward_time": 1.698194980621338,
      "step": 41763
    },
    {
      "epoch": 3.982830047607422e-05,
      "step": 41763,
      "training_step_time": 6.700938940048218
    },
    {
      "epoch": 3.982925415039063e-05,
      "model_forward_time": 1.7078804969787598,
      "step": 41764
    },
    {
      "epoch": 3.982925415039063e-05,
      "step": 41764,
      "training_step_time": 6.785391330718994
    },
    {
      "epoch": 3.983020782470703e-05,
      "model_forward_time": 1.7062456607818604,
      "step": 41765
    },
    {
      "epoch": 3.983020782470703e-05,
      "step": 41765,
      "training_step_time": 6.699224472045898
    },
    {
      "epoch": 3.9831161499023436e-05,
      "model_forward_time": 1.7045443058013916,
      "step": 41766
    },
    {
      "epoch": 3.9831161499023436e-05,
      "step": 41766,
      "training_step_time": 6.6992027759552
    },
    {
      "epoch": 3.983211517333985e-05,
      "model_forward_time": 1.727341890335083,
      "step": 41767
    },
    {
      "epoch": 3.983211517333985e-05,
      "step": 41767,
      "training_step_time": 6.69921350479126
    },
    {
      "epoch": 3.983306884765625e-05,
      "model_forward_time": 1.7433199882507324,
      "step": 41768
    },
    {
      "epoch": 3.983306884765625e-05,
      "step": 41768,
      "training_step_time": 6.715224742889404
    },
    {
      "epoch": 3.9834022521972655e-05,
      "model_forward_time": 1.7453765869140625,
      "step": 41769
    },
    {
      "epoch": 3.9834022521972655e-05,
      "step": 41769,
      "training_step_time": 6.970044851303101
    },
    {
      "epoch": 3.9834976196289065e-05,
      "grad_norm": 0.11430627107620239,
      "learning_rate": 6.737331521577716e-06,
      "loss": 0.0594,
      "step": 41770
    },
    {
      "epoch": 3.9834976196289065e-05,
      "model_forward_time": 2.3199057579040527,
      "step": 41770
    },
    {
      "dynamics_loss_avg": 0.07111185006797313,
      "epoch": 3.9834976196289065e-05,
      "step": 41770
    },
    {
      "action_loss_avg": 0.0037212215160252527,
      "epoch": 3.9834976196289065e-05,
      "step": 41770
    },
    {
      "epoch": 3.9834976196289065e-05,
      "step": 41770,
      "training_step_time": 9.019083499908447
    },
    {
      "epoch": 3.983592987060547e-05,
      "model_forward_time": 1.7327098846435547,
      "step": 41771
    },
    {
      "epoch": 3.983592987060547e-05,
      "step": 41771,
      "training_step_time": 9.197680473327637
    },
    {
      "epoch": 3.9836883544921874e-05,
      "model_forward_time": 1.7410907745361328,
      "step": 41772
    },
    {
      "epoch": 3.9836883544921874e-05,
      "step": 41772,
      "training_step_time": 6.733124732971191
    },
    {
      "epoch": 3.9837837219238284e-05,
      "model_forward_time": 1.7517147064208984,
      "step": 41773
    },
    {
      "epoch": 3.9837837219238284e-05,
      "step": 41773,
      "training_step_time": 6.743317604064941
    },
    {
      "epoch": 3.983879089355469e-05,
      "model_forward_time": 1.7315375804901123,
      "step": 41774
    },
    {
      "epoch": 3.983879089355469e-05,
      "step": 41774,
      "training_step_time": 6.693004608154297
    },
    {
      "epoch": 3.983974456787109e-05,
      "model_forward_time": 1.7467737197875977,
      "step": 41775
    },
    {
      "epoch": 3.983974456787109e-05,
      "step": 41775,
      "training_step_time": 6.7009077072143555
    },
    {
      "epoch": 3.98406982421875e-05,
      "model_forward_time": 1.7393617630004883,
      "step": 41776
    },
    {
      "epoch": 3.98406982421875e-05,
      "step": 41776,
      "training_step_time": 6.732252359390259
    },
    {
      "epoch": 3.984165191650391e-05,
      "model_forward_time": 1.7340877056121826,
      "step": 41777
    },
    {
      "epoch": 3.984165191650391e-05,
      "step": 41777,
      "training_step_time": 6.659593820571899
    },
    {
      "epoch": 3.984260559082031e-05,
      "model_forward_time": 1.741528034210205,
      "step": 41778
    },
    {
      "epoch": 3.984260559082031e-05,
      "step": 41778,
      "training_step_time": 7.251720428466797
    },
    {
      "epoch": 3.984355926513672e-05,
      "model_forward_time": 2.3333232402801514,
      "step": 41779
    },
    {
      "epoch": 3.984355926513672e-05,
      "step": 41779,
      "training_step_time": 10.44481086730957
    },
    {
      "epoch": 3.9844512939453126e-05,
      "grad_norm": 0.09927304089069366,
      "learning_rate": 6.735780980413404e-06,
      "loss": 0.0629,
      "step": 41780
    },
    {
      "epoch": 3.9844512939453126e-05,
      "model_forward_time": 1.7204704284667969,
      "step": 41780
    },
    {
      "dynamics_loss_avg": 0.0594686470925808,
      "epoch": 3.9844512939453126e-05,
      "step": 41780
    },
    {
      "action_loss_avg": 0.0037121098954230546,
      "epoch": 3.9844512939453126e-05,
      "step": 41780
    },
    {
      "epoch": 3.9844512939453126e-05,
      "step": 41780,
      "training_step_time": 8.810766696929932
    },
    {
      "epoch": 3.984546661376953e-05,
      "model_forward_time": 1.7403724193572998,
      "step": 41781
    },
    {
      "epoch": 3.984546661376953e-05,
      "step": 41781,
      "training_step_time": 7.844080924987793
    },
    {
      "epoch": 3.9846420288085934e-05,
      "model_forward_time": 1.3772039413452148,
      "step": 41782
    },
    {
      "epoch": 3.9846420288085934e-05,
      "step": 41782,
      "training_step_time": 6.664593458175659
    },
    {
      "epoch": 3.9847373962402345e-05,
      "model_forward_time": 1.7445101737976074,
      "step": 41783
    },
    {
      "epoch": 3.9847373962402345e-05,
      "step": 41783,
      "training_step_time": 6.750516653060913
    },
    {
      "epoch": 3.984832763671875e-05,
      "model_forward_time": 1.7413394451141357,
      "step": 41784
    },
    {
      "epoch": 3.984832763671875e-05,
      "step": 41784,
      "training_step_time": 6.71471643447876
    },
    {
      "epoch": 3.984928131103515e-05,
      "model_forward_time": 1.7376301288604736,
      "step": 41785
    },
    {
      "epoch": 3.984928131103515e-05,
      "step": 41785,
      "training_step_time": 6.706149339675903
    },
    {
      "epoch": 3.9850234985351564e-05,
      "model_forward_time": 1.7347283363342285,
      "step": 41786
    },
    {
      "epoch": 3.9850234985351564e-05,
      "step": 41786,
      "training_step_time": 6.690548658370972
    },
    {
      "epoch": 3.985118865966797e-05,
      "model_forward_time": 1.732475996017456,
      "step": 41787
    },
    {
      "epoch": 3.985118865966797e-05,
      "step": 41787,
      "training_step_time": 6.663983106613159
    },
    {
      "epoch": 3.985214233398437e-05,
      "model_forward_time": 2.349421262741089,
      "step": 41788
    },
    {
      "epoch": 3.985214233398437e-05,
      "step": 41788,
      "training_step_time": 8.73317813873291
    },
    {
      "epoch": 3.985309600830078e-05,
      "model_forward_time": 1.7256948947906494,
      "step": 41789
    },
    {
      "epoch": 3.985309600830078e-05,
      "step": 41789,
      "training_step_time": 8.341795921325684
    },
    {
      "epoch": 3.985404968261719e-05,
      "grad_norm": 0.13287150859832764,
      "learning_rate": 6.734230249426692e-06,
      "loss": 0.0622,
      "step": 41790
    },
    {
      "epoch": 3.985404968261719e-05,
      "model_forward_time": 1.7497940063476562,
      "step": 41790
    },
    {
      "dynamics_loss_avg": 0.05976336970925331,
      "epoch": 3.985404968261719e-05,
      "step": 41790
    },
    {
      "action_loss_avg": 0.00372602806892246,
      "epoch": 3.985404968261719e-05,
      "step": 41790
    },
    {
      "epoch": 3.985404968261719e-05,
      "step": 41790,
      "training_step_time": 7.733505964279175
    },
    {
      "epoch": 3.985500335693359e-05,
      "model_forward_time": 1.7348988056182861,
      "step": 41791
    },
    {
      "epoch": 3.985500335693359e-05,
      "step": 41791,
      "training_step_time": 6.663480281829834
    },
    {
      "epoch": 3.985595703125e-05,
      "model_forward_time": 1.7406365871429443,
      "step": 41792
    },
    {
      "epoch": 3.985595703125e-05,
      "step": 41792,
      "training_step_time": 6.713508605957031
    },
    {
      "epoch": 3.9856910705566406e-05,
      "model_forward_time": 1.3762619495391846,
      "step": 41793
    },
    {
      "epoch": 3.9856910705566406e-05,
      "step": 41793,
      "training_step_time": 6.67170524597168
    },
    {
      "epoch": 3.985786437988281e-05,
      "model_forward_time": 1.737379789352417,
      "step": 41794
    },
    {
      "epoch": 3.985786437988281e-05,
      "step": 41794,
      "training_step_time": 6.716222047805786
    },
    {
      "epoch": 3.985881805419922e-05,
      "model_forward_time": 1.7332394123077393,
      "step": 41795
    },
    {
      "epoch": 3.985881805419922e-05,
      "step": 41795,
      "training_step_time": 6.690657138824463
    },
    {
      "epoch": 3.9859771728515624e-05,
      "model_forward_time": 1.7380695343017578,
      "step": 41796
    },
    {
      "epoch": 3.9859771728515624e-05,
      "step": 41796,
      "training_step_time": 6.661194562911987
    },
    {
      "epoch": 3.986072540283203e-05,
      "model_forward_time": 2.257735013961792,
      "step": 41797
    },
    {
      "epoch": 3.986072540283203e-05,
      "step": 41797,
      "training_step_time": 7.564615488052368
    },
    {
      "epoch": 3.986167907714844e-05,
      "model_forward_time": 1.774827003479004,
      "step": 41798
    },
    {
      "epoch": 3.986167907714844e-05,
      "step": 41798,
      "training_step_time": 9.478967905044556
    },
    {
      "epoch": 3.986263275146484e-05,
      "model_forward_time": 1.749990463256836,
      "step": 41799
    },
    {
      "epoch": 3.986263275146484e-05,
      "step": 41799,
      "training_step_time": 7.728851795196533
    },
    {
      "epoch": 3.986358642578125e-05,
      "grad_norm": 0.11471238732337952,
      "learning_rate": 6.732679328787168e-06,
      "loss": 0.063,
      "step": 41800
    },
    {
      "epoch": 3.986358642578125e-05,
      "model_forward_time": 1.7319457530975342,
      "step": 41800
    },
    {
      "dynamics_loss_avg": 0.07932634763419628,
      "epoch": 3.986358642578125e-05,
      "step": 41800
    },
    {
      "action_loss_avg": 0.003392354975221679,
      "epoch": 3.986358642578125e-05,
      "step": 41800
    },
    {
      "epoch": 3.986358642578125e-05,
      "step": 41800,
      "training_step_time": 6.686062335968018
    },
    {
      "epoch": 3.986454010009766e-05,
      "model_forward_time": 1.7279655933380127,
      "step": 41801
    },
    {
      "epoch": 3.986454010009766e-05,
      "step": 41801,
      "training_step_time": 6.720055341720581
    },
    {
      "epoch": 3.986549377441406e-05,
      "model_forward_time": 1.73319411277771,
      "step": 41802
    },
    {
      "epoch": 3.986549377441406e-05,
      "step": 41802,
      "training_step_time": 6.691609144210815
    },
    {
      "epoch": 3.9866447448730466e-05,
      "model_forward_time": 1.7329888343811035,
      "step": 41803
    },
    {
      "epoch": 3.9866447448730466e-05,
      "step": 41803,
      "training_step_time": 6.6616575717926025
    },
    {
      "epoch": 3.986740112304688e-05,
      "model_forward_time": 1.74643874168396,
      "step": 41804
    },
    {
      "epoch": 3.986740112304688e-05,
      "step": 41804,
      "training_step_time": 6.658475875854492
    },
    {
      "epoch": 3.986835479736328e-05,
      "model_forward_time": 1.7400193214416504,
      "step": 41805
    },
    {
      "epoch": 3.986835479736328e-05,
      "step": 41805,
      "training_step_time": 6.666980743408203
    },
    {
      "epoch": 3.9869308471679685e-05,
      "model_forward_time": 1.7365548610687256,
      "step": 41806
    },
    {
      "epoch": 3.9869308471679685e-05,
      "step": 41806,
      "training_step_time": 7.657571792602539
    },
    {
      "epoch": 3.9870262145996096e-05,
      "model_forward_time": 2.3133411407470703,
      "step": 41807
    },
    {
      "epoch": 3.9870262145996096e-05,
      "step": 41807,
      "training_step_time": 9.558613300323486
    },
    {
      "epoch": 3.98712158203125e-05,
      "model_forward_time": 1.7478351593017578,
      "step": 41808
    },
    {
      "epoch": 3.98712158203125e-05,
      "step": 41808,
      "training_step_time": 8.792111158370972
    },
    {
      "epoch": 3.9872169494628904e-05,
      "model_forward_time": 1.7404940128326416,
      "step": 41809
    },
    {
      "epoch": 3.9872169494628904e-05,
      "step": 41809,
      "training_step_time": 7.4760401248931885
    },
    {
      "epoch": 3.9873123168945315e-05,
      "grad_norm": 0.10491039603948593,
      "learning_rate": 6.7311282186644375e-06,
      "loss": 0.06,
      "step": 41810
    },
    {
      "epoch": 3.9873123168945315e-05,
      "model_forward_time": 1.7360124588012695,
      "step": 41810
    },
    {
      "dynamics_loss_avg": 0.05773635748773813,
      "epoch": 3.9873123168945315e-05,
      "step": 41810
    },
    {
      "action_loss_avg": 0.002233648399123922,
      "epoch": 3.9873123168945315e-05,
      "step": 41810
    },
    {
      "epoch": 3.9873123168945315e-05,
      "step": 41810,
      "training_step_time": 6.670112371444702
    },
    {
      "epoch": 3.987407684326172e-05,
      "model_forward_time": 1.7464573383331299,
      "step": 41811
    },
    {
      "epoch": 3.987407684326172e-05,
      "step": 41811,
      "training_step_time": 6.699224948883057
    },
    {
      "epoch": 3.987503051757812e-05,
      "model_forward_time": 1.7436063289642334,
      "step": 41812
    },
    {
      "epoch": 3.987503051757812e-05,
      "step": 41812,
      "training_step_time": 6.662167549133301
    },
    {
      "epoch": 3.9875984191894534e-05,
      "model_forward_time": 1.7306768894195557,
      "step": 41813
    },
    {
      "epoch": 3.9875984191894534e-05,
      "step": 41813,
      "training_step_time": 6.681092262268066
    },
    {
      "epoch": 3.987693786621094e-05,
      "model_forward_time": 1.7394957542419434,
      "step": 41814
    },
    {
      "epoch": 3.987693786621094e-05,
      "step": 41814,
      "training_step_time": 6.667710065841675
    },
    {
      "epoch": 3.987789154052734e-05,
      "model_forward_time": 1.734666347503662,
      "step": 41815
    },
    {
      "epoch": 3.987789154052734e-05,
      "step": 41815,
      "training_step_time": 7.612945318222046
    },
    {
      "epoch": 3.987884521484375e-05,
      "model_forward_time": 2.3277926445007324,
      "step": 41816
    },
    {
      "epoch": 3.987884521484375e-05,
      "step": 41816,
      "training_step_time": 10.997613430023193
    },
    {
      "epoch": 3.9879798889160156e-05,
      "model_forward_time": 1.712388277053833,
      "step": 41817
    },
    {
      "epoch": 3.9879798889160156e-05,
      "step": 41817,
      "training_step_time": 7.652575254440308
    },
    {
      "epoch": 3.988075256347656e-05,
      "model_forward_time": 1.7371625900268555,
      "step": 41818
    },
    {
      "epoch": 3.988075256347656e-05,
      "step": 41818,
      "training_step_time": 7.65036416053772
    },
    {
      "epoch": 3.988170623779297e-05,
      "model_forward_time": 1.7410802841186523,
      "step": 41819
    },
    {
      "epoch": 3.988170623779297e-05,
      "step": 41819,
      "training_step_time": 6.7600202560424805
    },
    {
      "epoch": 3.9882659912109375e-05,
      "grad_norm": 0.14675648510456085,
      "learning_rate": 6.729576919228129e-06,
      "loss": 0.0566,
      "step": 41820
    },
    {
      "epoch": 3.9882659912109375e-05,
      "model_forward_time": 1.7269649505615234,
      "step": 41820
    },
    {
      "dynamics_loss_avg": 0.0663112074136734,
      "epoch": 3.9882659912109375e-05,
      "step": 41820
    },
    {
      "action_loss_avg": 0.0027151286543812604,
      "epoch": 3.9882659912109375e-05,
      "step": 41820
    },
    {
      "epoch": 3.9882659912109375e-05,
      "step": 41820,
      "training_step_time": 6.661173105239868
    },
    {
      "epoch": 3.988361358642578e-05,
      "model_forward_time": 1.7351503372192383,
      "step": 41821
    },
    {
      "epoch": 3.988361358642578e-05,
      "step": 41821,
      "training_step_time": 6.6724982261657715
    },
    {
      "epoch": 3.988456726074219e-05,
      "model_forward_time": 1.7437067031860352,
      "step": 41822
    },
    {
      "epoch": 3.988456726074219e-05,
      "step": 41822,
      "training_step_time": 6.691842555999756
    },
    {
      "epoch": 3.9885520935058594e-05,
      "model_forward_time": 1.7413685321807861,
      "step": 41823
    },
    {
      "epoch": 3.9885520935058594e-05,
      "step": 41823,
      "training_step_time": 6.65851354598999
    },
    {
      "epoch": 3.9886474609375e-05,
      "model_forward_time": 1.7324182987213135,
      "step": 41824
    },
    {
      "epoch": 3.9886474609375e-05,
      "step": 41824,
      "training_step_time": 7.643291234970093
    },
    {
      "epoch": 3.988742828369141e-05,
      "model_forward_time": 2.333700656890869,
      "step": 41825
    },
    {
      "epoch": 3.988742828369141e-05,
      "step": 41825,
      "training_step_time": 8.1503164768219
    },
    {
      "epoch": 3.988838195800781e-05,
      "model_forward_time": 1.7435297966003418,
      "step": 41826
    },
    {
      "epoch": 3.988838195800781e-05,
      "step": 41826,
      "training_step_time": 8.218856573104858
    },
    {
      "epoch": 3.988933563232422e-05,
      "model_forward_time": 1.7724401950836182,
      "step": 41827
    },
    {
      "epoch": 3.988933563232422e-05,
      "step": 41827,
      "training_step_time": 7.778804779052734
    },
    {
      "epoch": 3.989028930664063e-05,
      "model_forward_time": 1.776247501373291,
      "step": 41828
    },
    {
      "epoch": 3.989028930664063e-05,
      "step": 41828,
      "training_step_time": 6.680941581726074
    },
    {
      "epoch": 3.989124298095703e-05,
      "model_forward_time": 1.7785139083862305,
      "step": 41829
    },
    {
      "epoch": 3.989124298095703e-05,
      "step": 41829,
      "training_step_time": 6.697451829910278
    },
    {
      "epoch": 3.9892196655273436e-05,
      "grad_norm": 0.14720961451530457,
      "learning_rate": 6.728025430647888e-06,
      "loss": 0.0607,
      "step": 41830
    },
    {
      "epoch": 3.9892196655273436e-05,
      "model_forward_time": 1.7805211544036865,
      "step": 41830
    },
    {
      "dynamics_loss_avg": 0.06950516607612371,
      "epoch": 3.9892196655273436e-05,
      "step": 41830
    },
    {
      "action_loss_avg": 0.0037928136705886574,
      "epoch": 3.9892196655273436e-05,
      "step": 41830
    },
    {
      "epoch": 3.9892196655273436e-05,
      "step": 41830,
      "training_step_time": 6.685656309127808
    },
    {
      "epoch": 3.989315032958985e-05,
      "model_forward_time": 1.778803825378418,
      "step": 41831
    },
    {
      "epoch": 3.989315032958985e-05,
      "step": 41831,
      "training_step_time": 6.68007230758667
    },
    {
      "epoch": 3.989410400390625e-05,
      "model_forward_time": 1.7790484428405762,
      "step": 41832
    },
    {
      "epoch": 3.989410400390625e-05,
      "step": 41832,
      "training_step_time": 6.7191197872161865
    },
    {
      "epoch": 3.9895057678222655e-05,
      "model_forward_time": 1.7729167938232422,
      "step": 41833
    },
    {
      "epoch": 3.9895057678222655e-05,
      "step": 41833,
      "training_step_time": 7.548078298568726
    },
    {
      "epoch": 3.9896011352539066e-05,
      "model_forward_time": 2.373650074005127,
      "step": 41834
    },
    {
      "epoch": 3.9896011352539066e-05,
      "step": 41834,
      "training_step_time": 9.470223188400269
    },
    {
      "epoch": 3.989696502685547e-05,
      "model_forward_time": 1.776386022567749,
      "step": 41835
    },
    {
      "epoch": 3.989696502685547e-05,
      "step": 41835,
      "training_step_time": 10.119508266448975
    },
    {
      "epoch": 3.9897918701171874e-05,
      "model_forward_time": 1.7699155807495117,
      "step": 41836
    },
    {
      "epoch": 3.9897918701171874e-05,
      "step": 41836,
      "training_step_time": 7.717581033706665
    },
    {
      "epoch": 3.9898872375488284e-05,
      "model_forward_time": 1.7694413661956787,
      "step": 41837
    },
    {
      "epoch": 3.9898872375488284e-05,
      "step": 41837,
      "training_step_time": 6.661937952041626
    },
    {
      "epoch": 3.989982604980469e-05,
      "model_forward_time": 1.7836296558380127,
      "step": 41838
    },
    {
      "epoch": 3.989982604980469e-05,
      "step": 41838,
      "training_step_time": 6.6493353843688965
    },
    {
      "epoch": 3.990077972412109e-05,
      "model_forward_time": 1.7804508209228516,
      "step": 41839
    },
    {
      "epoch": 3.990077972412109e-05,
      "step": 41839,
      "training_step_time": 6.681121826171875
    },
    {
      "epoch": 3.99017333984375e-05,
      "grad_norm": 0.14116515219211578,
      "learning_rate": 6.7264737530933836e-06,
      "loss": 0.0603,
      "step": 41840
    },
    {
      "epoch": 3.99017333984375e-05,
      "model_forward_time": 1.7925739288330078,
      "step": 41840
    },
    {
      "dynamics_loss_avg": 0.07096925992518663,
      "epoch": 3.99017333984375e-05,
      "step": 41840
    },
    {
      "action_loss_avg": 0.004594807152170688,
      "epoch": 3.99017333984375e-05,
      "step": 41840
    },
    {
      "epoch": 3.99017333984375e-05,
      "step": 41840,
      "training_step_time": 6.640084743499756
    },
    {
      "epoch": 3.990268707275391e-05,
      "model_forward_time": 1.7761104106903076,
      "step": 41841
    },
    {
      "epoch": 3.990268707275391e-05,
      "step": 41841,
      "training_step_time": 6.694456100463867
    },
    {
      "epoch": 3.990364074707031e-05,
      "model_forward_time": 1.7706987857818604,
      "step": 41842
    },
    {
      "epoch": 3.990364074707031e-05,
      "step": 41842,
      "training_step_time": 7.658361434936523
    },
    {
      "epoch": 3.990459442138672e-05,
      "model_forward_time": 2.3313117027282715,
      "step": 41843
    },
    {
      "epoch": 3.990459442138672e-05,
      "step": 41843,
      "training_step_time": 9.421574592590332
    },
    {
      "epoch": 3.9905548095703126e-05,
      "model_forward_time": 1.7768101692199707,
      "step": 41844
    },
    {
      "epoch": 3.9905548095703126e-05,
      "step": 41844,
      "training_step_time": 8.318300485610962
    },
    {
      "epoch": 3.990650177001953e-05,
      "model_forward_time": 1.7540497779846191,
      "step": 41845
    },
    {
      "epoch": 3.990650177001953e-05,
      "step": 41845,
      "training_step_time": 7.72812557220459
    },
    {
      "epoch": 3.9907455444335934e-05,
      "model_forward_time": 1.7719225883483887,
      "step": 41846
    },
    {
      "epoch": 3.9907455444335934e-05,
      "step": 41846,
      "training_step_time": 7.217646360397339
    },
    {
      "epoch": 3.9908409118652345e-05,
      "model_forward_time": 1.7854833602905273,
      "step": 41847
    },
    {
      "epoch": 3.9908409118652345e-05,
      "step": 41847,
      "training_step_time": 6.67646050453186
    },
    {
      "epoch": 3.990936279296875e-05,
      "model_forward_time": 1.7940130233764648,
      "step": 41848
    },
    {
      "epoch": 3.990936279296875e-05,
      "step": 41848,
      "training_step_time": 6.671990156173706
    },
    {
      "epoch": 3.991031646728515e-05,
      "model_forward_time": 1.739325761795044,
      "step": 41849
    },
    {
      "epoch": 3.991031646728515e-05,
      "step": 41849,
      "training_step_time": 6.673798084259033
    },
    {
      "epoch": 3.9911270141601564e-05,
      "grad_norm": 0.0917746052145958,
      "learning_rate": 6.724921886734306e-06,
      "loss": 0.0575,
      "step": 41850
    },
    {
      "epoch": 3.9911270141601564e-05,
      "model_forward_time": 1.7492649555206299,
      "step": 41850
    },
    {
      "dynamics_loss_avg": 0.06300434796139598,
      "epoch": 3.9911270141601564e-05,
      "step": 41850
    },
    {
      "action_loss_avg": 0.001982286118436605,
      "epoch": 3.9911270141601564e-05,
      "step": 41850
    },
    {
      "epoch": 3.9911270141601564e-05,
      "step": 41850,
      "training_step_time": 6.730740070343018
    },
    {
      "epoch": 3.991222381591797e-05,
      "model_forward_time": 1.7478790283203125,
      "step": 41851
    },
    {
      "epoch": 3.991222381591797e-05,
      "step": 41851,
      "training_step_time": 7.614485740661621
    },
    {
      "epoch": 3.991317749023437e-05,
      "model_forward_time": 1.7383074760437012,
      "step": 41852
    },
    {
      "epoch": 3.991317749023437e-05,
      "step": 41852,
      "training_step_time": 7.491595029830933
    },
    {
      "epoch": 3.991413116455078e-05,
      "model_forward_time": 2.3412656784057617,
      "step": 41853
    },
    {
      "epoch": 3.991413116455078e-05,
      "step": 41853,
      "training_step_time": 10.00228238105774
    },
    {
      "epoch": 3.991508483886719e-05,
      "model_forward_time": 1.719822883605957,
      "step": 41854
    },
    {
      "epoch": 3.991508483886719e-05,
      "step": 41854,
      "training_step_time": 8.29858112335205
    },
    {
      "epoch": 3.991603851318359e-05,
      "model_forward_time": 1.7437591552734375,
      "step": 41855
    },
    {
      "epoch": 3.991603851318359e-05,
      "step": 41855,
      "training_step_time": 7.447633743286133
    },
    {
      "epoch": 3.99169921875e-05,
      "model_forward_time": 1.7372639179229736,
      "step": 41856
    },
    {
      "epoch": 3.99169921875e-05,
      "step": 41856,
      "training_step_time": 6.678917646408081
    },
    {
      "epoch": 3.9917945861816406e-05,
      "model_forward_time": 1.7380893230438232,
      "step": 41857
    },
    {
      "epoch": 3.9917945861816406e-05,
      "step": 41857,
      "training_step_time": 6.654947996139526
    },
    {
      "epoch": 3.991889953613281e-05,
      "model_forward_time": 1.7361187934875488,
      "step": 41858
    },
    {
      "epoch": 3.991889953613281e-05,
      "step": 41858,
      "training_step_time": 6.690201759338379
    },
    {
      "epoch": 3.991985321044922e-05,
      "model_forward_time": 1.7341034412384033,
      "step": 41859
    },
    {
      "epoch": 3.991985321044922e-05,
      "step": 41859,
      "training_step_time": 6.669329643249512
    },
    {
      "epoch": 3.9920806884765624e-05,
      "grad_norm": 0.09653200209140778,
      "learning_rate": 6.723369831740363e-06,
      "loss": 0.0595,
      "step": 41860
    },
    {
      "epoch": 3.9920806884765624e-05,
      "model_forward_time": 1.7364470958709717,
      "step": 41860
    },
    {
      "dynamics_loss_avg": 0.08743481673300266,
      "epoch": 3.9920806884765624e-05,
      "step": 41860
    },
    {
      "action_loss_avg": 0.004141789019922726,
      "epoch": 3.9920806884765624e-05,
      "step": 41860
    },
    {
      "epoch": 3.9920806884765624e-05,
      "step": 41860,
      "training_step_time": 7.607062816619873
    },
    {
      "epoch": 3.992176055908203e-05,
      "model_forward_time": 1.724536418914795,
      "step": 41861
    },
    {
      "epoch": 3.992176055908203e-05,
      "step": 41861,
      "training_step_time": 7.54816460609436
    },
    {
      "epoch": 3.992271423339844e-05,
      "model_forward_time": 2.3513402938842773,
      "step": 41862
    },
    {
      "epoch": 3.992271423339844e-05,
      "step": 41862,
      "training_step_time": 9.07235836982727
    },
    {
      "epoch": 3.992366790771484e-05,
      "model_forward_time": 1.7350454330444336,
      "step": 41863
    },
    {
      "epoch": 3.992366790771484e-05,
      "step": 41863,
      "training_step_time": 7.651151657104492
    },
    {
      "epoch": 3.992462158203125e-05,
      "model_forward_time": 1.74591064453125,
      "step": 41864
    },
    {
      "epoch": 3.992462158203125e-05,
      "step": 41864,
      "training_step_time": 7.7207255363464355
    },
    {
      "epoch": 3.992557525634766e-05,
      "model_forward_time": 1.7413134574890137,
      "step": 41865
    },
    {
      "epoch": 3.992557525634766e-05,
      "step": 41865,
      "training_step_time": 6.635209560394287
    },
    {
      "epoch": 3.992652893066406e-05,
      "model_forward_time": 1.7444605827331543,
      "step": 41866
    },
    {
      "epoch": 3.992652893066406e-05,
      "step": 41866,
      "training_step_time": 6.678375244140625
    },
    {
      "epoch": 3.9927482604980466e-05,
      "model_forward_time": 1.7416648864746094,
      "step": 41867
    },
    {
      "epoch": 3.9927482604980466e-05,
      "step": 41867,
      "training_step_time": 6.675035715103149
    },
    {
      "epoch": 3.992843627929688e-05,
      "model_forward_time": 1.7368190288543701,
      "step": 41868
    },
    {
      "epoch": 3.992843627929688e-05,
      "step": 41868,
      "training_step_time": 6.699636936187744
    },
    {
      "epoch": 3.992938995361328e-05,
      "model_forward_time": 1.7375149726867676,
      "step": 41869
    },
    {
      "epoch": 3.992938995361328e-05,
      "step": 41869,
      "training_step_time": 7.663858890533447
    },
    {
      "epoch": 3.9930343627929685e-05,
      "grad_norm": 0.09919126331806183,
      "learning_rate": 6.721817588281287e-06,
      "loss": 0.06,
      "step": 41870
    },
    {
      "epoch": 3.9930343627929685e-05,
      "model_forward_time": 1.736882209777832,
      "step": 41870
    },
    {
      "dynamics_loss_avg": 0.05623770095407963,
      "epoch": 3.9930343627929685e-05,
      "step": 41870
    },
    {
      "action_loss_avg": 0.002051501924870536,
      "epoch": 3.9930343627929685e-05,
      "step": 41870
    },
    {
      "epoch": 3.9930343627929685e-05,
      "step": 41870,
      "training_step_time": 7.502562999725342
    },
    {
      "epoch": 3.9931297302246096e-05,
      "model_forward_time": 2.336150884628296,
      "step": 41871
    },
    {
      "epoch": 3.9931297302246096e-05,
      "step": 41871,
      "training_step_time": 9.062838554382324
    },
    {
      "epoch": 3.99322509765625e-05,
      "model_forward_time": 1.734591007232666,
      "step": 41872
    },
    {
      "epoch": 3.99322509765625e-05,
      "step": 41872,
      "training_step_time": 7.7026755809783936
    },
    {
      "epoch": 3.9933204650878904e-05,
      "model_forward_time": 1.733069658279419,
      "step": 41873
    },
    {
      "epoch": 3.9933204650878904e-05,
      "step": 41873,
      "training_step_time": 8.256198167800903
    },
    {
      "epoch": 3.9934158325195315e-05,
      "model_forward_time": 1.735396385192871,
      "step": 41874
    },
    {
      "epoch": 3.9934158325195315e-05,
      "step": 41874,
      "training_step_time": 7.2282140254974365
    },
    {
      "epoch": 3.993511199951172e-05,
      "model_forward_time": 1.740910530090332,
      "step": 41875
    },
    {
      "epoch": 3.993511199951172e-05,
      "step": 41875,
      "training_step_time": 6.65058159828186
    },
    {
      "epoch": 3.993606567382812e-05,
      "model_forward_time": 1.7335307598114014,
      "step": 41876
    },
    {
      "epoch": 3.993606567382812e-05,
      "step": 41876,
      "training_step_time": 6.679107189178467
    },
    {
      "epoch": 3.9937019348144534e-05,
      "model_forward_time": 1.7418153285980225,
      "step": 41877
    },
    {
      "epoch": 3.9937019348144534e-05,
      "step": 41877,
      "training_step_time": 6.670624256134033
    },
    {
      "epoch": 3.993797302246094e-05,
      "model_forward_time": 1.7401392459869385,
      "step": 41878
    },
    {
      "epoch": 3.993797302246094e-05,
      "step": 41878,
      "training_step_time": 7.57005500793457
    },
    {
      "epoch": 3.993892669677734e-05,
      "model_forward_time": 1.7307090759277344,
      "step": 41879
    },
    {
      "epoch": 3.993892669677734e-05,
      "step": 41879,
      "training_step_time": 7.479143381118774
    },
    {
      "epoch": 3.993988037109375e-05,
      "grad_norm": 0.13072824478149414,
      "learning_rate": 6.720265156526828e-06,
      "loss": 0.0605,
      "step": 41880
    },
    {
      "epoch": 3.993988037109375e-05,
      "model_forward_time": 2.3459415435791016,
      "step": 41880
    },
    {
      "dynamics_loss_avg": 0.060995038598775864,
      "epoch": 3.993988037109375e-05,
      "step": 41880
    },
    {
      "action_loss_avg": 0.0031440298684174196,
      "epoch": 3.993988037109375e-05,
      "step": 41880
    },
    {
      "epoch": 3.993988037109375e-05,
      "step": 41880,
      "training_step_time": 9.576380014419556
    },
    {
      "epoch": 3.9940834045410156e-05,
      "model_forward_time": 1.7379863262176514,
      "step": 41881
    },
    {
      "epoch": 3.9940834045410156e-05,
      "step": 41881,
      "training_step_time": 9.442593574523926
    },
    {
      "epoch": 3.994178771972656e-05,
      "model_forward_time": 1.733060598373413,
      "step": 41882
    },
    {
      "epoch": 3.994178771972656e-05,
      "step": 41882,
      "training_step_time": 7.7188310623168945
    },
    {
      "epoch": 3.994274139404297e-05,
      "model_forward_time": 1.7369120121002197,
      "step": 41883
    },
    {
      "epoch": 3.994274139404297e-05,
      "step": 41883,
      "training_step_time": 7.577944278717041
    },
    {
      "epoch": 3.9943695068359375e-05,
      "model_forward_time": 1.7432427406311035,
      "step": 41884
    },
    {
      "epoch": 3.9943695068359375e-05,
      "step": 41884,
      "training_step_time": 6.672212839126587
    },
    {
      "epoch": 3.994464874267578e-05,
      "model_forward_time": 1.774385690689087,
      "step": 41885
    },
    {
      "epoch": 3.994464874267578e-05,
      "step": 41885,
      "training_step_time": 6.7063257694244385
    },
    {
      "epoch": 3.994560241699219e-05,
      "model_forward_time": 1.7327723503112793,
      "step": 41886
    },
    {
      "epoch": 3.994560241699219e-05,
      "step": 41886,
      "training_step_time": 6.674280166625977
    },
    {
      "epoch": 3.9946556091308594e-05,
      "model_forward_time": 1.7376959323883057,
      "step": 41887
    },
    {
      "epoch": 3.9946556091308594e-05,
      "step": 41887,
      "training_step_time": 7.6351118087768555
    },
    {
      "epoch": 3.9947509765625e-05,
      "model_forward_time": 1.732611894607544,
      "step": 41888
    },
    {
      "epoch": 3.9947509765625e-05,
      "step": 41888,
      "training_step_time": 7.459764719009399
    },
    {
      "epoch": 3.994846343994141e-05,
      "model_forward_time": 1.0378360748291016,
      "step": 41889
    },
    {
      "epoch": 3.994846343994141e-05,
      "step": 41889,
      "training_step_time": 8.714544534683228
    },
    {
      "epoch": 3.994941711425781e-05,
      "grad_norm": 0.16124528646469116,
      "learning_rate": 6.718712536646759e-06,
      "loss": 0.0592,
      "step": 41890
    },
    {
      "epoch": 3.994941711425781e-05,
      "model_forward_time": 2.3396809101104736,
      "step": 41890
    },
    {
      "dynamics_loss_avg": 0.04595325812697411,
      "epoch": 3.994941711425781e-05,
      "step": 41890
    },
    {
      "action_loss_avg": 0.001970066747162491,
      "epoch": 3.994941711425781e-05,
      "step": 41890
    },
    {
      "epoch": 3.994941711425781e-05,
      "step": 41890,
      "training_step_time": 7.749717712402344
    },
    {
      "epoch": 3.995037078857422e-05,
      "model_forward_time": 1.7276010513305664,
      "step": 41891
    },
    {
      "epoch": 3.995037078857422e-05,
      "step": 41891,
      "training_step_time": 8.357216119766235
    },
    {
      "epoch": 3.995132446289063e-05,
      "model_forward_time": 1.7360146045684814,
      "step": 41892
    },
    {
      "epoch": 3.995132446289063e-05,
      "step": 41892,
      "training_step_time": 7.645803928375244
    },
    {
      "epoch": 3.995227813720703e-05,
      "model_forward_time": 1.7424874305725098,
      "step": 41893
    },
    {
      "epoch": 3.995227813720703e-05,
      "step": 41893,
      "training_step_time": 6.675930023193359
    },
    {
      "epoch": 3.9953231811523436e-05,
      "model_forward_time": 1.7410387992858887,
      "step": 41894
    },
    {
      "epoch": 3.9953231811523436e-05,
      "step": 41894,
      "training_step_time": 6.801392316818237
    },
    {
      "epoch": 3.995418548583985e-05,
      "model_forward_time": 1.7385828495025635,
      "step": 41895
    },
    {
      "epoch": 3.995418548583985e-05,
      "step": 41895,
      "training_step_time": 6.730241298675537
    },
    {
      "epoch": 3.995513916015625e-05,
      "model_forward_time": 1.747694969177246,
      "step": 41896
    },
    {
      "epoch": 3.995513916015625e-05,
      "step": 41896,
      "training_step_time": 7.638847351074219
    },
    {
      "epoch": 3.9956092834472655e-05,
      "model_forward_time": 1.7319724559783936,
      "step": 41897
    },
    {
      "epoch": 3.9956092834472655e-05,
      "step": 41897,
      "training_step_time": 7.455819845199585
    },
    {
      "epoch": 3.9957046508789066e-05,
      "model_forward_time": 1.7349023818969727,
      "step": 41898
    },
    {
      "epoch": 3.9957046508789066e-05,
      "step": 41898,
      "training_step_time": 7.603609323501587
    },
    {
      "epoch": 3.995800018310547e-05,
      "model_forward_time": 2.379960775375366,
      "step": 41899
    },
    {
      "epoch": 3.995800018310547e-05,
      "step": 41899,
      "training_step_time": 7.685820817947388
    },
    {
      "epoch": 3.9958953857421874e-05,
      "grad_norm": 0.13861334323883057,
      "learning_rate": 6.71715972881087e-06,
      "loss": 0.0588,
      "step": 41900
    },
    {
      "epoch": 3.9958953857421874e-05,
      "model_forward_time": 1.747927188873291,
      "step": 41900
    },
    {
      "dynamics_loss_avg": 0.05219317972660065,
      "epoch": 3.9958953857421874e-05,
      "step": 41900
    },
    {
      "action_loss_avg": 0.0033414774166885763,
      "epoch": 3.9958953857421874e-05,
      "step": 41900
    },
    {
      "epoch": 3.9958953857421874e-05,
      "step": 41900,
      "training_step_time": 7.72737455368042
    },
    {
      "epoch": 3.9959907531738284e-05,
      "model_forward_time": 1.7376070022583008,
      "step": 41901
    },
    {
      "epoch": 3.9959907531738284e-05,
      "step": 41901,
      "training_step_time": 7.788095951080322
    },
    {
      "epoch": 3.996086120605469e-05,
      "model_forward_time": 1.749291181564331,
      "step": 41902
    },
    {
      "epoch": 3.996086120605469e-05,
      "step": 41902,
      "training_step_time": 6.741464853286743
    },
    {
      "epoch": 3.996181488037109e-05,
      "model_forward_time": 1.748354196548462,
      "step": 41903
    },
    {
      "epoch": 3.996181488037109e-05,
      "step": 41903,
      "training_step_time": 6.7624900341033936
    },
    {
      "epoch": 3.99627685546875e-05,
      "model_forward_time": 1.7554876804351807,
      "step": 41904
    },
    {
      "epoch": 3.99627685546875e-05,
      "step": 41904,
      "training_step_time": 6.713931083679199
    },
    {
      "epoch": 3.996372222900391e-05,
      "model_forward_time": 1.750581979751587,
      "step": 41905
    },
    {
      "epoch": 3.996372222900391e-05,
      "step": 41905,
      "training_step_time": 7.638649940490723
    },
    {
      "epoch": 3.996467590332031e-05,
      "model_forward_time": 1.742361068725586,
      "step": 41906
    },
    {
      "epoch": 3.996467590332031e-05,
      "step": 41906,
      "training_step_time": 7.4324870109558105
    },
    {
      "epoch": 3.9965629577636715e-05,
      "model_forward_time": 1.7473862171173096,
      "step": 41907
    },
    {
      "epoch": 3.9965629577636715e-05,
      "step": 41907,
      "training_step_time": 7.617990970611572
    },
    {
      "epoch": 3.9966583251953126e-05,
      "model_forward_time": 2.383812189102173,
      "step": 41908
    },
    {
      "epoch": 3.9966583251953126e-05,
      "step": 41908,
      "training_step_time": 9.319488525390625
    },
    {
      "epoch": 3.996753692626953e-05,
      "model_forward_time": 1.7489547729492188,
      "step": 41909
    },
    {
      "epoch": 3.996753692626953e-05,
      "step": 41909,
      "training_step_time": 8.596534729003906
    },
    {
      "epoch": 3.9968490600585934e-05,
      "grad_norm": 0.14086666703224182,
      "learning_rate": 6.715606733188975e-06,
      "loss": 0.057,
      "step": 41910
    },
    {
      "epoch": 3.9968490600585934e-05,
      "model_forward_time": 1.7458412647247314,
      "step": 41910
    },
    {
      "dynamics_loss_avg": 0.053848297148942945,
      "epoch": 3.9968490600585934e-05,
      "step": 41910
    },
    {
      "action_loss_avg": 0.0030492182791931554,
      "epoch": 3.9968490600585934e-05,
      "step": 41910
    },
    {
      "epoch": 3.9968490600585934e-05,
      "step": 41910,
      "training_step_time": 8.327529191970825
    },
    {
      "epoch": 3.9969444274902345e-05,
      "model_forward_time": 1.753861904144287,
      "step": 41911
    },
    {
      "epoch": 3.9969444274902345e-05,
      "step": 41911,
      "training_step_time": 7.299820184707642
    },
    {
      "epoch": 3.997039794921875e-05,
      "model_forward_time": 1.7555816173553467,
      "step": 41912
    },
    {
      "epoch": 3.997039794921875e-05,
      "step": 41912,
      "training_step_time": 6.816470623016357
    },
    {
      "epoch": 3.997135162353515e-05,
      "model_forward_time": 1.7550764083862305,
      "step": 41913
    },
    {
      "epoch": 3.997135162353515e-05,
      "step": 41913,
      "training_step_time": 6.768144845962524
    },
    {
      "epoch": 3.9972305297851564e-05,
      "model_forward_time": 1.745234727859497,
      "step": 41914
    },
    {
      "epoch": 3.9972305297851564e-05,
      "step": 41914,
      "training_step_time": 7.677269697189331
    },
    {
      "epoch": 3.997325897216797e-05,
      "model_forward_time": 1.743072271347046,
      "step": 41915
    },
    {
      "epoch": 3.997325897216797e-05,
      "step": 41915,
      "training_step_time": 8.341653823852539
    },
    {
      "epoch": 3.997421264648437e-05,
      "model_forward_time": 1.3925590515136719,
      "step": 41916
    },
    {
      "epoch": 3.997421264648437e-05,
      "step": 41916,
      "training_step_time": 7.629696607589722
    },
    {
      "epoch": 3.997516632080078e-05,
      "model_forward_time": 2.3544163703918457,
      "step": 41917
    },
    {
      "epoch": 3.997516632080078e-05,
      "step": 41917,
      "training_step_time": 7.734119176864624
    },
    {
      "epoch": 3.997611999511719e-05,
      "model_forward_time": 1.7477965354919434,
      "step": 41918
    },
    {
      "epoch": 3.997611999511719e-05,
      "step": 41918,
      "training_step_time": 8.815821647644043
    },
    {
      "epoch": 3.997707366943359e-05,
      "model_forward_time": 1.753230333328247,
      "step": 41919
    },
    {
      "epoch": 3.997707366943359e-05,
      "step": 41919,
      "training_step_time": 7.81328821182251
    },
    {
      "epoch": 3.997802734375e-05,
      "grad_norm": 0.19867947697639465,
      "learning_rate": 6.714053549950909e-06,
      "loss": 0.0629,
      "step": 41920
    },
    {
      "epoch": 3.997802734375e-05,
      "model_forward_time": 1.7531702518463135,
      "step": 41920
    },
    {
      "dynamics_loss_avg": 0.07123216111212968,
      "epoch": 3.997802734375e-05,
      "step": 41920
    },
    {
      "action_loss_avg": 0.0026660182571504264,
      "epoch": 3.997802734375e-05,
      "step": 41920
    },
    {
      "epoch": 3.997802734375e-05,
      "step": 41920,
      "training_step_time": 7.621894359588623
    },
    {
      "epoch": 3.9978981018066406e-05,
      "model_forward_time": 1.7475714683532715,
      "step": 41921
    },
    {
      "epoch": 3.9978981018066406e-05,
      "step": 41921,
      "training_step_time": 6.7976014614105225
    },
    {
      "epoch": 3.997993469238281e-05,
      "model_forward_time": 1.7231214046478271,
      "step": 41922
    },
    {
      "epoch": 3.997993469238281e-05,
      "step": 41922,
      "training_step_time": 6.748733758926392
    },
    {
      "epoch": 3.998088836669922e-05,
      "model_forward_time": 1.7377629280090332,
      "step": 41923
    },
    {
      "epoch": 3.998088836669922e-05,
      "step": 41923,
      "training_step_time": 7.699226140975952
    },
    {
      "epoch": 3.9981842041015625e-05,
      "model_forward_time": 1.7358951568603516,
      "step": 41924
    },
    {
      "epoch": 3.9981842041015625e-05,
      "step": 41924,
      "training_step_time": 7.563166856765747
    },
    {
      "epoch": 3.998279571533203e-05,
      "model_forward_time": 1.7344515323638916,
      "step": 41925
    },
    {
      "epoch": 3.998279571533203e-05,
      "step": 41925,
      "training_step_time": 7.637718439102173
    },
    {
      "epoch": 3.998374938964844e-05,
      "model_forward_time": 1.743469476699829,
      "step": 41926
    },
    {
      "epoch": 3.998374938964844e-05,
      "step": 41926,
      "training_step_time": 9.936483144760132
    },
    {
      "epoch": 3.998470306396484e-05,
      "model_forward_time": 1.7879037857055664,
      "step": 41927
    },
    {
      "epoch": 3.998470306396484e-05,
      "step": 41927,
      "training_step_time": 7.812114238739014
    },
    {
      "epoch": 3.998565673828125e-05,
      "model_forward_time": 1.740354061126709,
      "step": 41928
    },
    {
      "epoch": 3.998565673828125e-05,
      "step": 41928,
      "training_step_time": 8.677489995956421
    },
    {
      "epoch": 3.998661041259766e-05,
      "model_forward_time": 1.7386260032653809,
      "step": 41929
    },
    {
      "epoch": 3.998661041259766e-05,
      "step": 41929,
      "training_step_time": 7.710683584213257
    },
    {
      "epoch": 3.998756408691406e-05,
      "grad_norm": 0.1349629908800125,
      "learning_rate": 6.71250017926652e-06,
      "loss": 0.0627,
      "step": 41930
    },
    {
      "epoch": 3.998756408691406e-05,
      "model_forward_time": 1.734006643295288,
      "step": 41930
    },
    {
      "dynamics_loss_avg": 0.052863025292754176,
      "epoch": 3.998756408691406e-05,
      "step": 41930
    },
    {
      "action_loss_avg": 0.001987546164309606,
      "epoch": 3.998756408691406e-05,
      "step": 41930
    },
    {
      "epoch": 3.998756408691406e-05,
      "step": 41930,
      "training_step_time": 6.832765102386475
    },
    {
      "epoch": 3.9988517761230466e-05,
      "model_forward_time": 1.7424848079681396,
      "step": 41931
    },
    {
      "epoch": 3.9988517761230466e-05,
      "step": 41931,
      "training_step_time": 6.814992666244507
    },
    {
      "epoch": 3.998947143554688e-05,
      "model_forward_time": 1.7369489669799805,
      "step": 41932
    },
    {
      "epoch": 3.998947143554688e-05,
      "step": 41932,
      "training_step_time": 7.698483943939209
    },
    {
      "epoch": 3.999042510986328e-05,
      "model_forward_time": 1.7412960529327393,
      "step": 41933
    },
    {
      "epoch": 3.999042510986328e-05,
      "step": 41933,
      "training_step_time": 7.463580369949341
    },
    {
      "epoch": 3.9991378784179685e-05,
      "model_forward_time": 1.723120927810669,
      "step": 41934
    },
    {
      "epoch": 3.9991378784179685e-05,
      "step": 41934,
      "training_step_time": 8.165296792984009
    },
    {
      "epoch": 3.9992332458496096e-05,
      "model_forward_time": 1.7413175106048584,
      "step": 41935
    },
    {
      "epoch": 3.9992332458496096e-05,
      "step": 41935,
      "training_step_time": 8.82294511795044
    },
    {
      "epoch": 3.99932861328125e-05,
      "model_forward_time": 2.343928575515747,
      "step": 41936
    },
    {
      "epoch": 3.99932861328125e-05,
      "step": 41936,
      "training_step_time": 8.70983099937439
    },
    {
      "epoch": 3.9994239807128904e-05,
      "model_forward_time": 1.7420885562896729,
      "step": 41937
    },
    {
      "epoch": 3.9994239807128904e-05,
      "step": 41937,
      "training_step_time": 7.6637513637542725
    },
    {
      "epoch": 3.9995193481445315e-05,
      "model_forward_time": 1.7402794361114502,
      "step": 41938
    },
    {
      "epoch": 3.9995193481445315e-05,
      "step": 41938,
      "training_step_time": 7.7245025634765625
    },
    {
      "epoch": 3.999614715576172e-05,
      "model_forward_time": 1.7462913990020752,
      "step": 41939
    },
    {
      "epoch": 3.999614715576172e-05,
      "step": 41939,
      "training_step_time": 6.6764442920684814
    },
    {
      "epoch": 3.999710083007812e-05,
      "grad_norm": 0.17873607575893402,
      "learning_rate": 6.710946621305689e-06,
      "loss": 0.0582,
      "step": 41940
    },
    {
      "epoch": 3.999710083007812e-05,
      "model_forward_time": 1.7418720722198486,
      "step": 41940
    },
    {
      "dynamics_loss_avg": 0.07020961083471774,
      "epoch": 3.999710083007812e-05,
      "step": 41940
    },
    {
      "action_loss_avg": 0.002981063409242779,
      "epoch": 3.999710083007812e-05,
      "step": 41940
    },
    {
      "epoch": 3.999710083007812e-05,
      "step": 41940,
      "training_step_time": 6.673799753189087
    },
    {
      "epoch": 3.9998054504394534e-05,
      "model_forward_time": 1.7478220462799072,
      "step": 41941
    },
    {
      "epoch": 3.9998054504394534e-05,
      "step": 41941,
      "training_step_time": 7.5774431228637695
    },
    {
      "epoch": 3.999900817871094e-05,
      "model_forward_time": 1.747490644454956,
      "step": 41942
    },
    {
      "epoch": 3.999900817871094e-05,
      "step": 41942,
      "training_step_time": 8.197190761566162
    },
    {
      "epoch": 3.999996185302734e-05,
      "model_forward_time": 1.7300519943237305,
      "step": 41943
    },
    {
      "epoch": 3.999996185302734e-05,
      "step": 41943,
      "training_step_time": 7.409439563751221
    },
    {
      "epoch": 4.000091552734375e-05,
      "model_forward_time": 1.749119758605957,
      "step": 41944
    },
    {
      "epoch": 4.000091552734375e-05,
      "step": 41944,
      "training_step_time": 8.409796953201294
    },
    {
      "epoch": 4.0001869201660157e-05,
      "model_forward_time": 2.3594415187835693,
      "step": 41945
    },
    {
      "epoch": 4.0001869201660157e-05,
      "step": 41945,
      "training_step_time": 8.64647889137268
    },
    {
      "epoch": 4.000282287597656e-05,
      "model_forward_time": 1.752162218093872,
      "step": 41946
    },
    {
      "epoch": 4.000282287597656e-05,
      "step": 41946,
      "training_step_time": 8.699388027191162
    },
    {
      "epoch": 4.000377655029297e-05,
      "model_forward_time": 1.7497475147247314,
      "step": 41947
    },
    {
      "epoch": 4.000377655029297e-05,
      "step": 41947,
      "training_step_time": 8.358561754226685
    },
    {
      "epoch": 4.0004730224609375e-05,
      "model_forward_time": 1.7437846660614014,
      "step": 41948
    },
    {
      "epoch": 4.0004730224609375e-05,
      "step": 41948,
      "training_step_time": 7.147813558578491
    },
    {
      "epoch": 4.000568389892578e-05,
      "model_forward_time": 1.7568423748016357,
      "step": 41949
    },
    {
      "epoch": 4.000568389892578e-05,
      "step": 41949,
      "training_step_time": 6.735506296157837
    },
    {
      "epoch": 4.000663757324219e-05,
      "grad_norm": 0.12266290187835693,
      "learning_rate": 6.709392876238307e-06,
      "loss": 0.0598,
      "step": 41950
    },
    {
      "epoch": 4.000663757324219e-05,
      "model_forward_time": 1.735823631286621,
      "step": 41950
    },
    {
      "dynamics_loss_avg": 0.06578166652470827,
      "epoch": 4.000663757324219e-05,
      "step": 41950
    },
    {
      "action_loss_avg": 0.0024526119406800715,
      "epoch": 4.000663757324219e-05,
      "step": 41950
    },
    {
      "epoch": 4.000663757324219e-05,
      "step": 41950,
      "training_step_time": 7.541937828063965
    },
    {
      "epoch": 4.0007591247558594e-05,
      "model_forward_time": 1.7478060722351074,
      "step": 41951
    },
    {
      "epoch": 4.0007591247558594e-05,
      "step": 41951,
      "training_step_time": 7.8789942264556885
    },
    {
      "epoch": 4.0008544921875e-05,
      "model_forward_time": 1.746741533279419,
      "step": 41952
    },
    {
      "epoch": 4.0008544921875e-05,
      "step": 41952,
      "training_step_time": 7.438549518585205
    },
    {
      "epoch": 4.000949859619141e-05,
      "model_forward_time": 1.7522335052490234,
      "step": 41953
    },
    {
      "epoch": 4.000949859619141e-05,
      "step": 41953,
      "training_step_time": 8.269112348556519
    },
    {
      "epoch": 4.001045227050781e-05,
      "model_forward_time": 2.347033739089966,
      "step": 41954
    },
    {
      "epoch": 4.001045227050781e-05,
      "step": 41954,
      "training_step_time": 7.710386037826538
    },
    {
      "epoch": 4.001140594482422e-05,
      "model_forward_time": 1.7373206615447998,
      "step": 41955
    },
    {
      "epoch": 4.001140594482422e-05,
      "step": 41955,
      "training_step_time": 9.4005708694458
    },
    {
      "epoch": 4.001235961914063e-05,
      "model_forward_time": 1.7340216636657715,
      "step": 41956
    },
    {
      "epoch": 4.001235961914063e-05,
      "step": 41956,
      "training_step_time": 7.9587202072143555
    },
    {
      "epoch": 4.001331329345703e-05,
      "model_forward_time": 1.743896722793579,
      "step": 41957
    },
    {
      "epoch": 4.001331329345703e-05,
      "step": 41957,
      "training_step_time": 7.768284320831299
    },
    {
      "epoch": 4.0014266967773436e-05,
      "model_forward_time": 1.7515292167663574,
      "step": 41958
    },
    {
      "epoch": 4.0014266967773436e-05,
      "step": 41958,
      "training_step_time": 6.907107830047607
    },
    {
      "epoch": 4.001522064208985e-05,
      "model_forward_time": 1.754920244216919,
      "step": 41959
    },
    {
      "epoch": 4.001522064208985e-05,
      "step": 41959,
      "training_step_time": 7.751418828964233
    },
    {
      "epoch": 4.001617431640625e-05,
      "grad_norm": 0.16911578178405762,
      "learning_rate": 6.7078389442342895e-06,
      "loss": 0.0556,
      "step": 41960
    },
    {
      "epoch": 4.001617431640625e-05,
      "model_forward_time": 1.7769966125488281,
      "step": 41960
    },
    {
      "dynamics_loss_avg": 0.05490949563682079,
      "epoch": 4.001617431640625e-05,
      "step": 41960
    },
    {
      "action_loss_avg": 0.0023408285065670497,
      "epoch": 4.001617431640625e-05,
      "step": 41960
    },
    {
      "epoch": 4.001617431640625e-05,
      "step": 41960,
      "training_step_time": 6.8848936557769775
    },
    {
      "epoch": 4.0017127990722655e-05,
      "model_forward_time": 1.7500100135803223,
      "step": 41961
    },
    {
      "epoch": 4.0017127990722655e-05,
      "step": 41961,
      "training_step_time": 7.600110292434692
    },
    {
      "epoch": 4.0018081665039066e-05,
      "model_forward_time": 1.7484185695648193,
      "step": 41962
    },
    {
      "epoch": 4.0018081665039066e-05,
      "step": 41962,
      "training_step_time": 7.707324028015137
    },
    {
      "epoch": 4.001903533935547e-05,
      "model_forward_time": 2.291868209838867,
      "step": 41963
    },
    {
      "epoch": 4.001903533935547e-05,
      "step": 41963,
      "training_step_time": 8.537333726882935
    },
    {
      "epoch": 4.0019989013671874e-05,
      "model_forward_time": 1.7849326133728027,
      "step": 41964
    },
    {
      "epoch": 4.0019989013671874e-05,
      "step": 41964,
      "training_step_time": 8.53043508529663
    },
    {
      "epoch": 4.0020942687988285e-05,
      "model_forward_time": 1.7366530895233154,
      "step": 41965
    },
    {
      "epoch": 4.0020942687988285e-05,
      "step": 41965,
      "training_step_time": 7.763136625289917
    },
    {
      "epoch": 4.002189636230469e-05,
      "model_forward_time": 1.3754079341888428,
      "step": 41966
    },
    {
      "epoch": 4.002189636230469e-05,
      "step": 41966,
      "training_step_time": 7.578453063964844
    },
    {
      "epoch": 4.002285003662109e-05,
      "model_forward_time": 1.7841720581054688,
      "step": 41967
    },
    {
      "epoch": 4.002285003662109e-05,
      "step": 41967,
      "training_step_time": 6.703904151916504
    },
    {
      "epoch": 4.00238037109375e-05,
      "model_forward_time": 1.3794734477996826,
      "step": 41968
    },
    {
      "epoch": 4.00238037109375e-05,
      "step": 41968,
      "training_step_time": 7.487295627593994
    },
    {
      "epoch": 4.002475738525391e-05,
      "model_forward_time": 1.7505440711975098,
      "step": 41969
    },
    {
      "epoch": 4.002475738525391e-05,
      "step": 41969,
      "training_step_time": 6.74211311340332
    },
    {
      "epoch": 4.002571105957031e-05,
      "grad_norm": 0.1428571343421936,
      "learning_rate": 6.7062848254635735e-06,
      "loss": 0.0599,
      "step": 41970
    },
    {
      "epoch": 4.002571105957031e-05,
      "model_forward_time": 1.7211039066314697,
      "step": 41970
    },
    {
      "dynamics_loss_avg": 0.06546340677887201,
      "epoch": 4.002571105957031e-05,
      "step": 41970
    },
    {
      "action_loss_avg": 0.004269606526941061,
      "epoch": 4.002571105957031e-05,
      "step": 41970
    },
    {
      "epoch": 4.002571105957031e-05,
      "step": 41970,
      "training_step_time": 7.382582664489746
    },
    {
      "epoch": 4.0026664733886715e-05,
      "model_forward_time": 1.7498302459716797,
      "step": 41971
    },
    {
      "epoch": 4.0026664733886715e-05,
      "step": 41971,
      "training_step_time": 7.460639238357544
    },
    {
      "epoch": 4.0027618408203126e-05,
      "model_forward_time": 1.7431056499481201,
      "step": 41972
    },
    {
      "epoch": 4.0027618408203126e-05,
      "step": 41972,
      "training_step_time": 9.120795726776123
    },
    {
      "epoch": 4.002857208251953e-05,
      "model_forward_time": 1.7887959480285645,
      "step": 41973
    },
    {
      "epoch": 4.002857208251953e-05,
      "step": 41973,
      "training_step_time": 9.345688819885254
    },
    {
      "epoch": 4.0029525756835934e-05,
      "model_forward_time": 1.7384557723999023,
      "step": 41974
    },
    {
      "epoch": 4.0029525756835934e-05,
      "step": 41974,
      "training_step_time": 8.737565040588379
    },
    {
      "epoch": 4.0030479431152345e-05,
      "model_forward_time": 1.7451720237731934,
      "step": 41975
    },
    {
      "epoch": 4.0030479431152345e-05,
      "step": 41975,
      "training_step_time": 7.793565511703491
    },
    {
      "epoch": 4.003143310546875e-05,
      "model_forward_time": 1.7489783763885498,
      "step": 41976
    },
    {
      "epoch": 4.003143310546875e-05,
      "step": 41976,
      "training_step_time": 6.7052319049835205
    },
    {
      "epoch": 4.003238677978515e-05,
      "model_forward_time": 1.748319149017334,
      "step": 41977
    },
    {
      "epoch": 4.003238677978515e-05,
      "step": 41977,
      "training_step_time": 7.101377725601196
    },
    {
      "epoch": 4.0033340454101564e-05,
      "model_forward_time": 1.7489311695098877,
      "step": 41978
    },
    {
      "epoch": 4.0033340454101564e-05,
      "step": 41978,
      "training_step_time": 7.650153160095215
    },
    {
      "epoch": 4.003429412841797e-05,
      "model_forward_time": 1.7432935237884521,
      "step": 41979
    },
    {
      "epoch": 4.003429412841797e-05,
      "step": 41979,
      "training_step_time": 8.122529029846191
    },
    {
      "epoch": 4.003524780273437e-05,
      "grad_norm": 0.10420737415552139,
      "learning_rate": 6.704730520096115e-06,
      "loss": 0.0582,
      "step": 41980
    },
    {
      "epoch": 4.003524780273437e-05,
      "model_forward_time": 1.73856520652771,
      "step": 41980
    },
    {
      "dynamics_loss_avg": 0.07402782365679741,
      "epoch": 4.003524780273437e-05,
      "step": 41980
    },
    {
      "action_loss_avg": 0.0028212137287482618,
      "epoch": 4.003524780273437e-05,
      "step": 41980
    },
    {
      "epoch": 4.003524780273437e-05,
      "step": 41980,
      "training_step_time": 7.499884128570557
    },
    {
      "epoch": 4.003620147705078e-05,
      "model_forward_time": 1.749976396560669,
      "step": 41981
    },
    {
      "epoch": 4.003620147705078e-05,
      "step": 41981,
      "training_step_time": 9.503867149353027
    },
    {
      "epoch": 4.003715515136719e-05,
      "model_forward_time": 1.965287208557129,
      "step": 41982
    },
    {
      "epoch": 4.003715515136719e-05,
      "step": 41982,
      "training_step_time": 9.417043447494507
    },
    {
      "epoch": 4.003810882568359e-05,
      "model_forward_time": 1.7516937255859375,
      "step": 41983
    },
    {
      "epoch": 4.003810882568359e-05,
      "step": 41983,
      "training_step_time": 9.301248788833618
    },
    {
      "epoch": 4.00390625e-05,
      "model_forward_time": 1.749842882156372,
      "step": 41984
    },
    {
      "epoch": 4.00390625e-05,
      "step": 41984,
      "training_step_time": 7.768165349960327
    },
    {
      "epoch": 4.0040016174316406e-05,
      "model_forward_time": 1.7382466793060303,
      "step": 41985
    },
    {
      "epoch": 4.0040016174316406e-05,
      "step": 41985,
      "training_step_time": 7.598758935928345
    },
    {
      "epoch": 4.004096984863281e-05,
      "model_forward_time": 1.7469799518585205,
      "step": 41986
    },
    {
      "epoch": 4.004096984863281e-05,
      "step": 41986,
      "training_step_time": 7.310218334197998
    },
    {
      "epoch": 4.004192352294922e-05,
      "model_forward_time": 1.7490272521972656,
      "step": 41987
    },
    {
      "epoch": 4.004192352294922e-05,
      "step": 41987,
      "training_step_time": 7.631929874420166
    },
    {
      "epoch": 4.0042877197265625e-05,
      "model_forward_time": 1.7616424560546875,
      "step": 41988
    },
    {
      "epoch": 4.0042877197265625e-05,
      "step": 41988,
      "training_step_time": 7.402878999710083
    },
    {
      "epoch": 4.004383087158203e-05,
      "model_forward_time": 1.7448081970214844,
      "step": 41989
    },
    {
      "epoch": 4.004383087158203e-05,
      "step": 41989,
      "training_step_time": 7.444655179977417
    },
    {
      "epoch": 4.004478454589844e-05,
      "grad_norm": 0.13972939550876617,
      "learning_rate": 6.703176028301888e-06,
      "loss": 0.0588,
      "step": 41990
    },
    {
      "epoch": 4.004478454589844e-05,
      "model_forward_time": 1.744131326675415,
      "step": 41990
    },
    {
      "dynamics_loss_avg": 0.05953162871301174,
      "epoch": 4.004478454589844e-05,
      "step": 41990
    },
    {
      "action_loss_avg": 0.00251924098120071,
      "epoch": 4.004478454589844e-05,
      "step": 41990
    },
    {
      "epoch": 4.004478454589844e-05,
      "step": 41990,
      "training_step_time": 7.566602468490601
    },
    {
      "epoch": 4.0045738220214843e-05,
      "model_forward_time": 2.304492473602295,
      "step": 41991
    },
    {
      "epoch": 4.0045738220214843e-05,
      "step": 41991,
      "training_step_time": 8.348201751708984
    },
    {
      "epoch": 4.004669189453125e-05,
      "model_forward_time": 1.7588694095611572,
      "step": 41992
    },
    {
      "epoch": 4.004669189453125e-05,
      "step": 41992,
      "training_step_time": 7.638567209243774
    },
    {
      "epoch": 4.004764556884766e-05,
      "model_forward_time": 1.7441353797912598,
      "step": 41993
    },
    {
      "epoch": 4.004764556884766e-05,
      "step": 41993,
      "training_step_time": 7.85094952583313
    },
    {
      "epoch": 4.004859924316406e-05,
      "model_forward_time": 1.759016513824463,
      "step": 41994
    },
    {
      "epoch": 4.004859924316406e-05,
      "step": 41994,
      "training_step_time": 7.227279901504517
    },
    {
      "epoch": 4.0049552917480466e-05,
      "model_forward_time": 1.753530502319336,
      "step": 41995
    },
    {
      "epoch": 4.0049552917480466e-05,
      "step": 41995,
      "training_step_time": 6.775277853012085
    },
    {
      "epoch": 4.005050659179688e-05,
      "model_forward_time": 1.7247164249420166,
      "step": 41996
    },
    {
      "epoch": 4.005050659179688e-05,
      "step": 41996,
      "training_step_time": 7.611027002334595
    },
    {
      "epoch": 4.005146026611328e-05,
      "model_forward_time": 1.7506041526794434,
      "step": 41997
    },
    {
      "epoch": 4.005146026611328e-05,
      "step": 41997,
      "training_step_time": 8.134029626846313
    },
    {
      "epoch": 4.0052413940429685e-05,
      "model_forward_time": 1.74589204788208,
      "step": 41998
    },
    {
      "epoch": 4.0052413940429685e-05,
      "step": 41998,
      "training_step_time": 7.80367636680603
    },
    {
      "epoch": 4.0053367614746096e-05,
      "model_forward_time": 1.743889570236206,
      "step": 41999
    },
    {
      "epoch": 4.0053367614746096e-05,
      "step": 41999,
      "training_step_time": 9.357601642608643
    },
    {
      "epoch": 4.00543212890625e-05,
      "grad_norm": 0.0933937057852745,
      "learning_rate": 6.701621350250893e-06,
      "loss": 0.0583,
      "step": 42000
    },
    {
      "epoch": 4.00543212890625e-05,
      "model_forward_time": 1.7561473846435547,
      "step": 42000
    },
    {
      "dynamics_loss_avg": 0.0708342108875513,
      "epoch": 4.00543212890625e-05,
      "step": 42000
    },
    {
      "action_loss_avg": 0.0031449405068997294,
      "epoch": 4.00543212890625e-05,
      "step": 42000
    },
    {
      "epoch": 4.00543212890625e-05,
      "step": 42000,
      "training_step_time": 10.806694269180298
    },
    {
      "epoch": 4.0055274963378904e-05,
      "model_forward_time": 1.7983956336975098,
      "step": 42001
    },
    {
      "epoch": 4.0055274963378904e-05,
      "step": 42001,
      "training_step_time": 9.275654077529907
    },
    {
      "epoch": 4.0056228637695315e-05,
      "model_forward_time": 1.7501885890960693,
      "step": 42002
    },
    {
      "epoch": 4.0056228637695315e-05,
      "step": 42002,
      "training_step_time": 8.58188271522522
    },
    {
      "epoch": 4.005718231201172e-05,
      "model_forward_time": 1.750220775604248,
      "step": 42003
    },
    {
      "epoch": 4.005718231201172e-05,
      "step": 42003,
      "training_step_time": 7.676773309707642
    },
    {
      "epoch": 4.005813598632812e-05,
      "model_forward_time": 1.7544498443603516,
      "step": 42004
    },
    {
      "epoch": 4.005813598632812e-05,
      "step": 42004,
      "training_step_time": 8.600612878799438
    },
    {
      "epoch": 4.0059089660644534e-05,
      "model_forward_time": 1.7506260871887207,
      "step": 42005
    },
    {
      "epoch": 4.0059089660644534e-05,
      "step": 42005,
      "training_step_time": 7.174840688705444
    },
    {
      "epoch": 4.006004333496094e-05,
      "model_forward_time": 1.7514307498931885,
      "step": 42006
    },
    {
      "epoch": 4.006004333496094e-05,
      "step": 42006,
      "training_step_time": 8.762022733688354
    },
    {
      "epoch": 4.006099700927734e-05,
      "model_forward_time": 1.7451610565185547,
      "step": 42007
    },
    {
      "epoch": 4.006099700927734e-05,
      "step": 42007,
      "training_step_time": 8.227884531021118
    },
    {
      "epoch": 4.006195068359375e-05,
      "model_forward_time": 1.750213623046875,
      "step": 42008
    },
    {
      "epoch": 4.006195068359375e-05,
      "step": 42008,
      "training_step_time": 9.533606052398682
    },
    {
      "epoch": 4.0062904357910157e-05,
      "model_forward_time": 2.7703771591186523,
      "step": 42009
    },
    {
      "epoch": 4.0062904357910157e-05,
      "step": 42009,
      "training_step_time": 10.4852294921875
    },
    {
      "epoch": 4.006385803222656e-05,
      "grad_norm": 0.11381462216377258,
      "learning_rate": 6.700066486113145e-06,
      "loss": 0.0558,
      "step": 42010
    },
    {
      "epoch": 4.006385803222656e-05,
      "model_forward_time": 1.7896203994750977,
      "step": 42010
    },
    {
      "dynamics_loss_avg": 0.051696193777024745,
      "epoch": 4.006385803222656e-05,
      "step": 42010
    },
    {
      "action_loss_avg": 0.00147801170242019,
      "epoch": 4.006385803222656e-05,
      "step": 42010
    },
    {
      "epoch": 4.006385803222656e-05,
      "step": 42010,
      "training_step_time": 8.574503898620605
    },
    {
      "epoch": 4.006481170654297e-05,
      "model_forward_time": 1.744786262512207,
      "step": 42011
    },
    {
      "epoch": 4.006481170654297e-05,
      "step": 42011,
      "training_step_time": 9.67713451385498
    },
    {
      "epoch": 4.0065765380859375e-05,
      "model_forward_time": 1.7534289360046387,
      "step": 42012
    },
    {
      "epoch": 4.0065765380859375e-05,
      "step": 42012,
      "training_step_time": 7.6838531494140625
    },
    {
      "epoch": 4.006671905517578e-05,
      "model_forward_time": 1.7504322528839111,
      "step": 42013
    },
    {
      "epoch": 4.006671905517578e-05,
      "step": 42013,
      "training_step_time": 7.590223789215088
    },
    {
      "epoch": 4.006767272949219e-05,
      "model_forward_time": 1.755601406097412,
      "step": 42014
    },
    {
      "epoch": 4.006767272949219e-05,
      "step": 42014,
      "training_step_time": 7.740986347198486
    },
    {
      "epoch": 4.0068626403808594e-05,
      "model_forward_time": 1.7509300708770752,
      "step": 42015
    },
    {
      "epoch": 4.0068626403808594e-05,
      "step": 42015,
      "training_step_time": 6.824198246002197
    },
    {
      "epoch": 4.0069580078125e-05,
      "model_forward_time": 1.7416212558746338,
      "step": 42016
    },
    {
      "epoch": 4.0069580078125e-05,
      "step": 42016,
      "training_step_time": 8.086272716522217
    },
    {
      "epoch": 4.007053375244141e-05,
      "model_forward_time": 1.746636152267456,
      "step": 42017
    },
    {
      "epoch": 4.007053375244141e-05,
      "step": 42017,
      "training_step_time": 7.750669002532959
    },
    {
      "epoch": 4.007148742675781e-05,
      "model_forward_time": 2.3034603595733643,
      "step": 42018
    },
    {
      "epoch": 4.007148742675781e-05,
      "step": 42018,
      "training_step_time": 8.227240800857544
    },
    {
      "epoch": 4.007244110107422e-05,
      "model_forward_time": 1.4329488277435303,
      "step": 42019
    },
    {
      "epoch": 4.007244110107422e-05,
      "step": 42019,
      "training_step_time": 9.810355186462402
    },
    {
      "epoch": 4.007339477539063e-05,
      "grad_norm": 0.10423866659402847,
      "learning_rate": 6.6985114360586815e-06,
      "loss": 0.062,
      "step": 42020
    },
    {
      "epoch": 4.007339477539063e-05,
      "model_forward_time": 1.7433083057403564,
      "step": 42020
    },
    {
      "dynamics_loss_avg": 0.05178838837891817,
      "epoch": 4.007339477539063e-05,
      "step": 42020
    },
    {
      "action_loss_avg": 0.0014581353403627872,
      "epoch": 4.007339477539063e-05,
      "step": 42020
    },
    {
      "epoch": 4.007339477539063e-05,
      "step": 42020,
      "training_step_time": 9.171400547027588
    },
    {
      "epoch": 4.007434844970703e-05,
      "model_forward_time": 1.7424404621124268,
      "step": 42021
    },
    {
      "epoch": 4.007434844970703e-05,
      "step": 42021,
      "training_step_time": 7.7918782234191895
    },
    {
      "epoch": 4.0075302124023436e-05,
      "model_forward_time": 1.7256665229797363,
      "step": 42022
    },
    {
      "epoch": 4.0075302124023436e-05,
      "step": 42022,
      "training_step_time": 7.596763372421265
    },
    {
      "epoch": 4.007625579833985e-05,
      "model_forward_time": 1.7430529594421387,
      "step": 42023
    },
    {
      "epoch": 4.007625579833985e-05,
      "step": 42023,
      "training_step_time": 7.648318529129028
    },
    {
      "epoch": 4.007720947265625e-05,
      "model_forward_time": 1.737342119216919,
      "step": 42024
    },
    {
      "epoch": 4.007720947265625e-05,
      "step": 42024,
      "training_step_time": 6.780221700668335
    },
    {
      "epoch": 4.0078163146972655e-05,
      "model_forward_time": 1.7347824573516846,
      "step": 42025
    },
    {
      "epoch": 4.0078163146972655e-05,
      "step": 42025,
      "training_step_time": 7.488316535949707
    },
    {
      "epoch": 4.0079116821289066e-05,
      "model_forward_time": 1.731858491897583,
      "step": 42026
    },
    {
      "epoch": 4.0079116821289066e-05,
      "step": 42026,
      "training_step_time": 7.5892653465271
    },
    {
      "epoch": 4.008007049560547e-05,
      "model_forward_time": 2.3133091926574707,
      "step": 42027
    },
    {
      "epoch": 4.008007049560547e-05,
      "step": 42027,
      "training_step_time": 7.707411527633667
    },
    {
      "epoch": 4.0081024169921874e-05,
      "model_forward_time": 1.789381504058838,
      "step": 42028
    },
    {
      "epoch": 4.0081024169921874e-05,
      "step": 42028,
      "training_step_time": 7.762428522109985
    },
    {
      "epoch": 4.0081977844238285e-05,
      "model_forward_time": 1.7499361038208008,
      "step": 42029
    },
    {
      "epoch": 4.0081977844238285e-05,
      "step": 42029,
      "training_step_time": 8.537233352661133
    },
    {
      "epoch": 4.008293151855469e-05,
      "grad_norm": 0.11217562854290009,
      "learning_rate": 6.696956200257561e-06,
      "loss": 0.0597,
      "step": 42030
    },
    {
      "epoch": 4.008293151855469e-05,
      "model_forward_time": 1.734877586364746,
      "step": 42030
    },
    {
      "dynamics_loss_avg": 0.06585937701165676,
      "epoch": 4.008293151855469e-05,
      "step": 42030
    },
    {
      "action_loss_avg": 0.0019490218721330166,
      "epoch": 4.008293151855469e-05,
      "step": 42030
    },
    {
      "epoch": 4.008293151855469e-05,
      "step": 42030,
      "training_step_time": 7.895257949829102
    },
    {
      "epoch": 4.008388519287109e-05,
      "model_forward_time": 1.7485687732696533,
      "step": 42031
    },
    {
      "epoch": 4.008388519287109e-05,
      "step": 42031,
      "training_step_time": 7.635016918182373
    },
    {
      "epoch": 4.00848388671875e-05,
      "model_forward_time": 1.7511651515960693,
      "step": 42032
    },
    {
      "epoch": 4.00848388671875e-05,
      "step": 42032,
      "training_step_time": 7.590303182601929
    },
    {
      "epoch": 4.008579254150391e-05,
      "model_forward_time": 1.745542287826538,
      "step": 42033
    },
    {
      "epoch": 4.008579254150391e-05,
      "step": 42033,
      "training_step_time": 6.7544519901275635
    },
    {
      "epoch": 4.008674621582031e-05,
      "model_forward_time": 1.7473978996276855,
      "step": 42034
    },
    {
      "epoch": 4.008674621582031e-05,
      "step": 42034,
      "training_step_time": 7.380616664886475
    },
    {
      "epoch": 4.0087699890136715e-05,
      "model_forward_time": 1.7593274116516113,
      "step": 42035
    },
    {
      "epoch": 4.0087699890136715e-05,
      "step": 42035,
      "training_step_time": 8.387898921966553
    },
    {
      "epoch": 4.0088653564453126e-05,
      "model_forward_time": 2.318898916244507,
      "step": 42036
    },
    {
      "epoch": 4.0088653564453126e-05,
      "step": 42036,
      "training_step_time": 7.6809587478637695
    },
    {
      "epoch": 4.008960723876953e-05,
      "model_forward_time": 1.7971255779266357,
      "step": 42037
    },
    {
      "epoch": 4.008960723876953e-05,
      "step": 42037,
      "training_step_time": 10.129770040512085
    },
    {
      "epoch": 4.0090560913085934e-05,
      "model_forward_time": 1.7525360584259033,
      "step": 42038
    },
    {
      "epoch": 4.0090560913085934e-05,
      "step": 42038,
      "training_step_time": 9.506053686141968
    },
    {
      "epoch": 4.0091514587402345e-05,
      "model_forward_time": 1.743394374847412,
      "step": 42039
    },
    {
      "epoch": 4.0091514587402345e-05,
      "step": 42039,
      "training_step_time": 7.887755870819092
    },
    {
      "epoch": 4.009246826171875e-05,
      "grad_norm": 0.11732944846153259,
      "learning_rate": 6.695400778879863e-06,
      "loss": 0.0599,
      "step": 42040
    },
    {
      "epoch": 4.009246826171875e-05,
      "model_forward_time": 1.7431907653808594,
      "step": 42040
    },
    {
      "dynamics_loss_avg": 0.05526859816163778,
      "epoch": 4.009246826171875e-05,
      "step": 42040
    },
    {
      "action_loss_avg": 0.0029822550830431284,
      "epoch": 4.009246826171875e-05,
      "step": 42040
    },
    {
      "epoch": 4.009246826171875e-05,
      "step": 42040,
      "training_step_time": 7.711404323577881
    },
    {
      "epoch": 4.009342193603515e-05,
      "model_forward_time": 1.7491509914398193,
      "step": 42041
    },
    {
      "epoch": 4.009342193603515e-05,
      "step": 42041,
      "training_step_time": 6.806498289108276
    },
    {
      "epoch": 4.0094375610351564e-05,
      "model_forward_time": 1.7182462215423584,
      "step": 42042
    },
    {
      "epoch": 4.0094375610351564e-05,
      "step": 42042,
      "training_step_time": 7.656603813171387
    },
    {
      "epoch": 4.009532928466797e-05,
      "model_forward_time": 1.7144417762756348,
      "step": 42043
    },
    {
      "epoch": 4.009532928466797e-05,
      "step": 42043,
      "training_step_time": 7.585295915603638
    },
    {
      "epoch": 4.009628295898437e-05,
      "model_forward_time": 1.7394328117370605,
      "step": 42044
    },
    {
      "epoch": 4.009628295898437e-05,
      "step": 42044,
      "training_step_time": 8.35388445854187
    },
    {
      "epoch": 4.009723663330078e-05,
      "model_forward_time": 1.7217559814453125,
      "step": 42045
    },
    {
      "epoch": 4.009723663330078e-05,
      "step": 42045,
      "training_step_time": 8.704854249954224
    },
    {
      "epoch": 4.009819030761719e-05,
      "model_forward_time": 1.7119934558868408,
      "step": 42046
    },
    {
      "epoch": 4.009819030761719e-05,
      "step": 42046,
      "training_step_time": 9.536962747573853
    },
    {
      "epoch": 4.009914398193359e-05,
      "model_forward_time": 1.7043750286102295,
      "step": 42047
    },
    {
      "epoch": 4.009914398193359e-05,
      "step": 42047,
      "training_step_time": 8.663911581039429
    },
    {
      "epoch": 4.010009765625e-05,
      "model_forward_time": 1.720914363861084,
      "step": 42048
    },
    {
      "epoch": 4.010009765625e-05,
      "step": 42048,
      "training_step_time": 8.089274644851685
    },
    {
      "epoch": 4.0101051330566406e-05,
      "model_forward_time": 1.7093074321746826,
      "step": 42049
    },
    {
      "epoch": 4.0101051330566406e-05,
      "step": 42049,
      "training_step_time": 7.778079271316528
    },
    {
      "epoch": 4.010200500488281e-05,
      "grad_norm": 0.1293751746416092,
      "learning_rate": 6.693845172095682e-06,
      "loss": 0.0591,
      "step": 42050
    },
    {
      "epoch": 4.010200500488281e-05,
      "model_forward_time": 1.7217981815338135,
      "step": 42050
    },
    {
      "dynamics_loss_avg": 0.050329642137512565,
      "epoch": 4.010200500488281e-05,
      "step": 42050
    },
    {
      "action_loss_avg": 0.0028884704457595944,
      "epoch": 4.010200500488281e-05,
      "step": 42050
    },
    {
      "epoch": 4.010200500488281e-05,
      "step": 42050,
      "training_step_time": 7.628749847412109
    },
    {
      "epoch": 4.010295867919922e-05,
      "model_forward_time": 1.3398642539978027,
      "step": 42051
    },
    {
      "epoch": 4.010295867919922e-05,
      "step": 42051,
      "training_step_time": 7.807172060012817
    },
    {
      "epoch": 4.0103912353515625e-05,
      "model_forward_time": 1.7095224857330322,
      "step": 42052
    },
    {
      "epoch": 4.0103912353515625e-05,
      "step": 42052,
      "training_step_time": 7.655970096588135
    },
    {
      "epoch": 4.010486602783203e-05,
      "model_forward_time": 1.75789475440979,
      "step": 42053
    },
    {
      "epoch": 4.010486602783203e-05,
      "step": 42053,
      "training_step_time": 8.463168621063232
    },
    {
      "epoch": 4.010581970214844e-05,
      "model_forward_time": 1.7382159233093262,
      "step": 42054
    },
    {
      "epoch": 4.010581970214844e-05,
      "step": 42054,
      "training_step_time": 8.713071823120117
    },
    {
      "epoch": 4.0106773376464843e-05,
      "model_forward_time": 2.9011478424072266,
      "step": 42055
    },
    {
      "epoch": 4.0106773376464843e-05,
      "step": 42055,
      "training_step_time": 7.731234788894653
    },
    {
      "epoch": 4.010772705078125e-05,
      "model_forward_time": 1.7199721336364746,
      "step": 42056
    },
    {
      "epoch": 4.010772705078125e-05,
      "step": 42056,
      "training_step_time": 8.964837551116943
    },
    {
      "epoch": 4.010868072509766e-05,
      "model_forward_time": 1.723907709121704,
      "step": 42057
    },
    {
      "epoch": 4.010868072509766e-05,
      "step": 42057,
      "training_step_time": 7.985234260559082
    },
    {
      "epoch": 4.010963439941406e-05,
      "model_forward_time": 1.7234644889831543,
      "step": 42058
    },
    {
      "epoch": 4.010963439941406e-05,
      "step": 42058,
      "training_step_time": 7.8161656856536865
    },
    {
      "epoch": 4.0110588073730466e-05,
      "model_forward_time": 1.7362427711486816,
      "step": 42059
    },
    {
      "epoch": 4.0110588073730466e-05,
      "step": 42059,
      "training_step_time": 7.642786264419556
    },
    {
      "epoch": 4.011154174804688e-05,
      "grad_norm": 0.157992884516716,
      "learning_rate": 6.692289380075142e-06,
      "loss": 0.0604,
      "step": 42060
    },
    {
      "epoch": 4.011154174804688e-05,
      "model_forward_time": 1.7164099216461182,
      "step": 42060
    },
    {
      "dynamics_loss_avg": 0.05327416118234396,
      "epoch": 4.011154174804688e-05,
      "step": 42060
    },
    {
      "action_loss_avg": 0.001587883903994225,
      "epoch": 4.011154174804688e-05,
      "step": 42060
    },
    {
      "epoch": 4.011154174804688e-05,
      "step": 42060,
      "training_step_time": 7.751834869384766
    },
    {
      "epoch": 4.011249542236328e-05,
      "model_forward_time": 1.72239089012146,
      "step": 42061
    },
    {
      "epoch": 4.011249542236328e-05,
      "step": 42061,
      "training_step_time": 7.570915222167969
    },
    {
      "epoch": 4.0113449096679685e-05,
      "model_forward_time": 1.7360756397247314,
      "step": 42062
    },
    {
      "epoch": 4.0113449096679685e-05,
      "step": 42062,
      "training_step_time": 7.31489372253418
    },
    {
      "epoch": 4.0114402770996096e-05,
      "model_forward_time": 1.7190580368041992,
      "step": 42063
    },
    {
      "epoch": 4.0114402770996096e-05,
      "step": 42063,
      "training_step_time": 7.746160984039307
    },
    {
      "epoch": 4.01153564453125e-05,
      "model_forward_time": 2.8648810386657715,
      "step": 42064
    },
    {
      "epoch": 4.01153564453125e-05,
      "step": 42064,
      "training_step_time": 8.153528451919556
    },
    {
      "epoch": 4.0116310119628904e-05,
      "model_forward_time": 1.038175106048584,
      "step": 42065
    },
    {
      "epoch": 4.0116310119628904e-05,
      "step": 42065,
      "training_step_time": 8.709277153015137
    },
    {
      "epoch": 4.0117263793945315e-05,
      "model_forward_time": 1.729048252105713,
      "step": 42066
    },
    {
      "epoch": 4.0117263793945315e-05,
      "step": 42066,
      "training_step_time": 7.906174659729004
    },
    {
      "epoch": 4.011821746826172e-05,
      "model_forward_time": 1.7371301651000977,
      "step": 42067
    },
    {
      "epoch": 4.011821746826172e-05,
      "step": 42067,
      "training_step_time": 7.7557148933410645
    },
    {
      "epoch": 4.011917114257812e-05,
      "model_forward_time": 1.72420334815979,
      "step": 42068
    },
    {
      "epoch": 4.011917114257812e-05,
      "step": 42068,
      "training_step_time": 7.559728622436523
    },
    {
      "epoch": 4.0120124816894534e-05,
      "model_forward_time": 1.727550983428955,
      "step": 42069
    },
    {
      "epoch": 4.0120124816894534e-05,
      "step": 42069,
      "training_step_time": 7.607965707778931
    },
    {
      "epoch": 4.012107849121094e-05,
      "grad_norm": 0.1780509501695633,
      "learning_rate": 6.69073340298838e-06,
      "loss": 0.061,
      "step": 42070
    },
    {
      "epoch": 4.012107849121094e-05,
      "model_forward_time": 1.7606196403503418,
      "step": 42070
    },
    {
      "dynamics_loss_avg": 0.07470749095082282,
      "epoch": 4.012107849121094e-05,
      "step": 42070
    },
    {
      "action_loss_avg": 0.0022483288776129484,
      "epoch": 4.012107849121094e-05,
      "step": 42070
    },
    {
      "epoch": 4.012107849121094e-05,
      "step": 42070,
      "training_step_time": 7.2222607135772705
    },
    {
      "epoch": 4.012203216552734e-05,
      "model_forward_time": 1.748905897140503,
      "step": 42071
    },
    {
      "epoch": 4.012203216552734e-05,
      "step": 42071,
      "training_step_time": 7.518582820892334
    },
    {
      "epoch": 4.012298583984375e-05,
      "model_forward_time": 1.8175792694091797,
      "step": 42072
    },
    {
      "epoch": 4.012298583984375e-05,
      "step": 42072,
      "training_step_time": 9.026420593261719
    },
    {
      "epoch": 4.012393951416016e-05,
      "model_forward_time": 1.7508819103240967,
      "step": 42073
    },
    {
      "epoch": 4.012393951416016e-05,
      "step": 42073,
      "training_step_time": 8.782009363174438
    },
    {
      "epoch": 4.012489318847656e-05,
      "model_forward_time": 1.7461206912994385,
      "step": 42074
    },
    {
      "epoch": 4.012489318847656e-05,
      "step": 42074,
      "training_step_time": 8.23503303527832
    },
    {
      "epoch": 4.012584686279297e-05,
      "model_forward_time": 1.744950771331787,
      "step": 42075
    },
    {
      "epoch": 4.012584686279297e-05,
      "step": 42075,
      "training_step_time": 8.597399234771729
    },
    {
      "epoch": 4.0126800537109375e-05,
      "model_forward_time": 1.730492353439331,
      "step": 42076
    },
    {
      "epoch": 4.0126800537109375e-05,
      "step": 42076,
      "training_step_time": 8.424593448638916
    },
    {
      "epoch": 4.012775421142578e-05,
      "model_forward_time": 1.7431411743164062,
      "step": 42077
    },
    {
      "epoch": 4.012775421142578e-05,
      "step": 42077,
      "training_step_time": 7.686375379562378
    },
    {
      "epoch": 4.012870788574219e-05,
      "model_forward_time": 1.744948387145996,
      "step": 42078
    },
    {
      "epoch": 4.012870788574219e-05,
      "step": 42078,
      "training_step_time": 8.185288667678833
    },
    {
      "epoch": 4.0129661560058594e-05,
      "model_forward_time": 1.7460062503814697,
      "step": 42079
    },
    {
      "epoch": 4.0129661560058594e-05,
      "step": 42079,
      "training_step_time": 6.833444595336914
    },
    {
      "epoch": 4.0130615234375e-05,
      "grad_norm": 0.11828628927469254,
      "learning_rate": 6.689177241005553e-06,
      "loss": 0.0612,
      "step": 42080
    },
    {
      "epoch": 4.0130615234375e-05,
      "model_forward_time": 1.7431550025939941,
      "step": 42080
    },
    {
      "dynamics_loss_avg": 0.06240733824670315,
      "epoch": 4.0130615234375e-05,
      "step": 42080
    },
    {
      "action_loss_avg": 0.0024995424086228014,
      "epoch": 4.0130615234375e-05,
      "step": 42080
    },
    {
      "epoch": 4.0130615234375e-05,
      "step": 42080,
      "training_step_time": 8.037832975387573
    },
    {
      "epoch": 4.013156890869141e-05,
      "model_forward_time": 1.8175184726715088,
      "step": 42081
    },
    {
      "epoch": 4.013156890869141e-05,
      "step": 42081,
      "training_step_time": 8.235992193222046
    },
    {
      "epoch": 4.013252258300781e-05,
      "model_forward_time": 1.7457311153411865,
      "step": 42082
    },
    {
      "epoch": 4.013252258300781e-05,
      "step": 42082,
      "training_step_time": 7.627839803695679
    },
    {
      "epoch": 4.013347625732422e-05,
      "model_forward_time": 2.917652130126953,
      "step": 42083
    },
    {
      "epoch": 4.013347625732422e-05,
      "step": 42083,
      "training_step_time": 10.06252145767212
    },
    {
      "epoch": 4.013442993164063e-05,
      "model_forward_time": 1.7472419738769531,
      "step": 42084
    },
    {
      "epoch": 4.013442993164063e-05,
      "step": 42084,
      "training_step_time": 8.6241295337677
    },
    {
      "epoch": 4.013538360595703e-05,
      "model_forward_time": 1.7540333271026611,
      "step": 42085
    },
    {
      "epoch": 4.013538360595703e-05,
      "step": 42085,
      "training_step_time": 7.944709777832031
    },
    {
      "epoch": 4.0136337280273436e-05,
      "model_forward_time": 1.7642912864685059,
      "step": 42086
    },
    {
      "epoch": 4.0136337280273436e-05,
      "step": 42086,
      "training_step_time": 7.643989562988281
    },
    {
      "epoch": 4.013729095458985e-05,
      "model_forward_time": 1.7533018589019775,
      "step": 42087
    },
    {
      "epoch": 4.013729095458985e-05,
      "step": 42087,
      "training_step_time": 7.17061710357666
    },
    {
      "epoch": 4.013824462890625e-05,
      "model_forward_time": 1.7547070980072021,
      "step": 42088
    },
    {
      "epoch": 4.013824462890625e-05,
      "step": 42088,
      "training_step_time": 7.712084054946899
    },
    {
      "epoch": 4.0139198303222655e-05,
      "model_forward_time": 1.7485146522521973,
      "step": 42089
    },
    {
      "epoch": 4.0139198303222655e-05,
      "step": 42089,
      "training_step_time": 7.519536256790161
    },
    {
      "epoch": 4.0140151977539066e-05,
      "grad_norm": 0.11642203480005264,
      "learning_rate": 6.687620894296844e-06,
      "loss": 0.0588,
      "step": 42090
    },
    {
      "epoch": 4.0140151977539066e-05,
      "model_forward_time": 1.7814326286315918,
      "step": 42090
    },
    {
      "dynamics_loss_avg": 0.07784974165260791,
      "epoch": 4.0140151977539066e-05,
      "step": 42090
    },
    {
      "action_loss_avg": 0.003600286855362356,
      "epoch": 4.0140151977539066e-05,
      "step": 42090
    },
    {
      "epoch": 4.0140151977539066e-05,
      "step": 42090,
      "training_step_time": 8.31493067741394
    },
    {
      "epoch": 4.014110565185547e-05,
      "model_forward_time": 1.8090617656707764,
      "step": 42091
    },
    {
      "epoch": 4.014110565185547e-05,
      "step": 42091,
      "training_step_time": 9.223576068878174
    },
    {
      "epoch": 4.0142059326171874e-05,
      "model_forward_time": 2.8943569660186768,
      "step": 42092
    },
    {
      "epoch": 4.0142059326171874e-05,
      "step": 42092,
      "training_step_time": 8.852033138275146
    },
    {
      "epoch": 4.0143013000488285e-05,
      "model_forward_time": 1.7477359771728516,
      "step": 42093
    },
    {
      "epoch": 4.0143013000488285e-05,
      "step": 42093,
      "training_step_time": 7.794797420501709
    },
    {
      "epoch": 4.014396667480469e-05,
      "model_forward_time": 1.7387728691101074,
      "step": 42094
    },
    {
      "epoch": 4.014396667480469e-05,
      "step": 42094,
      "training_step_time": 7.852831602096558
    },
    {
      "epoch": 4.014492034912109e-05,
      "model_forward_time": 1.7463045120239258,
      "step": 42095
    },
    {
      "epoch": 4.014492034912109e-05,
      "step": 42095,
      "training_step_time": 7.725027084350586
    },
    {
      "epoch": 4.01458740234375e-05,
      "model_forward_time": 1.3873436450958252,
      "step": 42096
    },
    {
      "epoch": 4.01458740234375e-05,
      "step": 42096,
      "training_step_time": 7.21124529838562
    },
    {
      "epoch": 4.014682769775391e-05,
      "model_forward_time": 1.7443790435791016,
      "step": 42097
    },
    {
      "epoch": 4.014682769775391e-05,
      "step": 42097,
      "training_step_time": 7.704777717590332
    },
    {
      "epoch": 4.014778137207031e-05,
      "model_forward_time": 1.7378604412078857,
      "step": 42098
    },
    {
      "epoch": 4.014778137207031e-05,
      "step": 42098,
      "training_step_time": 7.48178768157959
    },
    {
      "epoch": 4.0148735046386716e-05,
      "model_forward_time": 1.7568128108978271,
      "step": 42099
    },
    {
      "epoch": 4.0148735046386716e-05,
      "step": 42099,
      "training_step_time": 8.951025485992432
    },
    {
      "epoch": 4.0149688720703126e-05,
      "grad_norm": 0.1453031301498413,
      "learning_rate": 6.686064363032451e-06,
      "loss": 0.0626,
      "step": 42100
    },
    {
      "epoch": 4.0149688720703126e-05,
      "model_forward_time": 1.8113632202148438,
      "step": 42100
    },
    {
      "dynamics_loss_avg": 0.0676138086244464,
      "epoch": 4.0149688720703126e-05,
      "step": 42100
    },
    {
      "action_loss_avg": 0.003947207960300148,
      "epoch": 4.0149688720703126e-05,
      "step": 42100
    },
    {
      "epoch": 4.0149688720703126e-05,
      "step": 42100,
      "training_step_time": 7.621393918991089
    },
    {
      "epoch": 4.015064239501953e-05,
      "model_forward_time": 2.9094338417053223,
      "step": 42101
    },
    {
      "epoch": 4.015064239501953e-05,
      "step": 42101,
      "training_step_time": 8.249282836914062
    },
    {
      "epoch": 4.0151596069335934e-05,
      "model_forward_time": 1.7474172115325928,
      "step": 42102
    },
    {
      "epoch": 4.0151596069335934e-05,
      "step": 42102,
      "training_step_time": 7.64425253868103
    },
    {
      "epoch": 4.0152549743652345e-05,
      "model_forward_time": 1.7461650371551514,
      "step": 42103
    },
    {
      "epoch": 4.0152549743652345e-05,
      "step": 42103,
      "training_step_time": 9.308963537216187
    },
    {
      "epoch": 4.015350341796875e-05,
      "model_forward_time": 1.7495014667510986,
      "step": 42104
    },
    {
      "epoch": 4.015350341796875e-05,
      "step": 42104,
      "training_step_time": 7.729685544967651
    },
    {
      "epoch": 4.015445709228515e-05,
      "model_forward_time": 1.7446849346160889,
      "step": 42105
    },
    {
      "epoch": 4.015445709228515e-05,
      "step": 42105,
      "training_step_time": 8.68735146522522
    },
    {
      "epoch": 4.0155410766601564e-05,
      "model_forward_time": 1.753563642501831,
      "step": 42106
    },
    {
      "epoch": 4.0155410766601564e-05,
      "step": 42106,
      "training_step_time": 7.6302125453948975
    },
    {
      "epoch": 4.015636444091797e-05,
      "model_forward_time": 1.7700364589691162,
      "step": 42107
    },
    {
      "epoch": 4.015636444091797e-05,
      "step": 42107,
      "training_step_time": 7.451484680175781
    },
    {
      "epoch": 4.015731811523437e-05,
      "model_forward_time": 1.742542028427124,
      "step": 42108
    },
    {
      "epoch": 4.015731811523437e-05,
      "step": 42108,
      "training_step_time": 7.481765985488892
    },
    {
      "epoch": 4.015827178955078e-05,
      "model_forward_time": 1.815340518951416,
      "step": 42109
    },
    {
      "epoch": 4.015827178955078e-05,
      "step": 42109,
      "training_step_time": 8.350761413574219
    },
    {
      "epoch": 4.015922546386719e-05,
      "grad_norm": 0.13356171548366547,
      "learning_rate": 6.684507647382595e-06,
      "loss": 0.0589,
      "step": 42110
    },
    {
      "epoch": 4.015922546386719e-05,
      "model_forward_time": 2.895662784576416,
      "step": 42110
    },
    {
      "dynamics_loss_avg": 0.057623945735394956,
      "epoch": 4.015922546386719e-05,
      "step": 42110
    },
    {
      "action_loss_avg": 0.0020117551903240383,
      "epoch": 4.015922546386719e-05,
      "step": 42110
    },
    {
      "epoch": 4.015922546386719e-05,
      "step": 42110,
      "training_step_time": 8.324665546417236
    },
    {
      "epoch": 4.016017913818359e-05,
      "model_forward_time": 1.7398669719696045,
      "step": 42111
    },
    {
      "epoch": 4.016017913818359e-05,
      "step": 42111,
      "training_step_time": 7.743953227996826
    },
    {
      "epoch": 4.01611328125e-05,
      "model_forward_time": 1.7491941452026367,
      "step": 42112
    },
    {
      "epoch": 4.01611328125e-05,
      "step": 42112,
      "training_step_time": 8.75226879119873
    },
    {
      "epoch": 4.0162086486816406e-05,
      "model_forward_time": 1.7285826206207275,
      "step": 42113
    },
    {
      "epoch": 4.0162086486816406e-05,
      "step": 42113,
      "training_step_time": 7.868114948272705
    },
    {
      "epoch": 4.016304016113281e-05,
      "model_forward_time": 1.7489986419677734,
      "step": 42114
    },
    {
      "epoch": 4.016304016113281e-05,
      "step": 42114,
      "training_step_time": 7.64034104347229
    },
    {
      "epoch": 4.016399383544922e-05,
      "model_forward_time": 1.748220443725586,
      "step": 42115
    },
    {
      "epoch": 4.016399383544922e-05,
      "step": 42115,
      "training_step_time": 7.6389055252075195
    },
    {
      "epoch": 4.0164947509765625e-05,
      "model_forward_time": 1.7423937320709229,
      "step": 42116
    },
    {
      "epoch": 4.0164947509765625e-05,
      "step": 42116,
      "training_step_time": 8.101641178131104
    },
    {
      "epoch": 4.016590118408203e-05,
      "model_forward_time": 1.7359614372253418,
      "step": 42117
    },
    {
      "epoch": 4.016590118408203e-05,
      "step": 42117,
      "training_step_time": 7.515776872634888
    },
    {
      "epoch": 4.016685485839844e-05,
      "model_forward_time": 1.8033602237701416,
      "step": 42118
    },
    {
      "epoch": 4.016685485839844e-05,
      "step": 42118,
      "training_step_time": 7.688641786575317
    },
    {
      "epoch": 4.0167808532714844e-05,
      "model_forward_time": 2.87172269821167,
      "step": 42119
    },
    {
      "epoch": 4.0167808532714844e-05,
      "step": 42119,
      "training_step_time": 8.270944118499756
    },
    {
      "epoch": 4.016876220703125e-05,
      "grad_norm": 0.12062480300664902,
      "learning_rate": 6.682950747517514e-06,
      "loss": 0.065,
      "step": 42120
    },
    {
      "epoch": 4.016876220703125e-05,
      "model_forward_time": 1.7366487979888916,
      "step": 42120
    },
    {
      "dynamics_loss_avg": 0.05548827480524778,
      "epoch": 4.016876220703125e-05,
      "step": 42120
    },
    {
      "action_loss_avg": 0.0018697688938118517,
      "epoch": 4.016876220703125e-05,
      "step": 42120
    },
    {
      "epoch": 4.016876220703125e-05,
      "step": 42120,
      "training_step_time": 8.621212482452393
    },
    {
      "epoch": 4.016971588134766e-05,
      "model_forward_time": 1.7306172847747803,
      "step": 42121
    },
    {
      "epoch": 4.016971588134766e-05,
      "step": 42121,
      "training_step_time": 7.683416128158569
    },
    {
      "epoch": 4.017066955566406e-05,
      "model_forward_time": 1.7357828617095947,
      "step": 42122
    },
    {
      "epoch": 4.017066955566406e-05,
      "step": 42122,
      "training_step_time": 7.966515779495239
    },
    {
      "epoch": 4.0171623229980466e-05,
      "model_forward_time": 1.7472538948059082,
      "step": 42123
    },
    {
      "epoch": 4.0171623229980466e-05,
      "step": 42123,
      "training_step_time": 7.676805734634399
    },
    {
      "epoch": 4.017257690429688e-05,
      "model_forward_time": 1.7429792881011963,
      "step": 42124
    },
    {
      "epoch": 4.017257690429688e-05,
      "step": 42124,
      "training_step_time": 7.1867899894714355
    },
    {
      "epoch": 4.017353057861328e-05,
      "model_forward_time": 1.727529764175415,
      "step": 42125
    },
    {
      "epoch": 4.017353057861328e-05,
      "step": 42125,
      "training_step_time": 7.715500116348267
    },
    {
      "epoch": 4.0174484252929685e-05,
      "model_forward_time": 1.729804277420044,
      "step": 42126
    },
    {
      "epoch": 4.0174484252929685e-05,
      "step": 42126,
      "training_step_time": 8.359599828720093
    },
    {
      "epoch": 4.0175437927246096e-05,
      "model_forward_time": 1.7195241451263428,
      "step": 42127
    },
    {
      "epoch": 4.0175437927246096e-05,
      "step": 42127,
      "training_step_time": 7.627344608306885
    },
    {
      "epoch": 4.01763916015625e-05,
      "model_forward_time": 2.811612129211426,
      "step": 42128
    },
    {
      "epoch": 4.01763916015625e-05,
      "step": 42128,
      "training_step_time": 8.587732553482056
    },
    {
      "epoch": 4.0177345275878904e-05,
      "model_forward_time": 1.7318239212036133,
      "step": 42129
    },
    {
      "epoch": 4.0177345275878904e-05,
      "step": 42129,
      "training_step_time": 8.67470669746399
    },
    {
      "epoch": 4.0178298950195315e-05,
      "grad_norm": 0.16051162779331207,
      "learning_rate": 6.6813936636074705e-06,
      "loss": 0.0647,
      "step": 42130
    },
    {
      "epoch": 4.0178298950195315e-05,
      "model_forward_time": 1.7283666133880615,
      "step": 42130
    },
    {
      "dynamics_loss_avg": 0.0715677946805954,
      "epoch": 4.0178298950195315e-05,
      "step": 42130
    },
    {
      "action_loss_avg": 0.0024553256516810507,
      "epoch": 4.0178298950195315e-05,
      "step": 42130
    },
    {
      "epoch": 4.0178298950195315e-05,
      "step": 42130,
      "training_step_time": 9.604143857955933
    },
    {
      "epoch": 4.017925262451172e-05,
      "model_forward_time": 1.748091697692871,
      "step": 42131
    },
    {
      "epoch": 4.017925262451172e-05,
      "step": 42131,
      "training_step_time": 8.493698120117188
    },
    {
      "epoch": 4.018020629882812e-05,
      "model_forward_time": 1.7233242988586426,
      "step": 42132
    },
    {
      "epoch": 4.018020629882812e-05,
      "step": 42132,
      "training_step_time": 7.710569858551025
    },
    {
      "epoch": 4.0181159973144534e-05,
      "model_forward_time": 1.7288868427276611,
      "step": 42133
    },
    {
      "epoch": 4.0181159973144534e-05,
      "step": 42133,
      "training_step_time": 7.728314399719238
    },
    {
      "epoch": 4.018211364746094e-05,
      "model_forward_time": 1.717543125152588,
      "step": 42134
    },
    {
      "epoch": 4.018211364746094e-05,
      "step": 42134,
      "training_step_time": 7.683830976486206
    },
    {
      "epoch": 4.018306732177734e-05,
      "model_forward_time": 1.719153642654419,
      "step": 42135
    },
    {
      "epoch": 4.018306732177734e-05,
      "step": 42135,
      "training_step_time": 7.535721778869629
    },
    {
      "epoch": 4.018402099609375e-05,
      "model_forward_time": 1.733234167098999,
      "step": 42136
    },
    {
      "epoch": 4.018402099609375e-05,
      "step": 42136,
      "training_step_time": 7.6457836627960205
    },
    {
      "epoch": 4.018497467041016e-05,
      "model_forward_time": 1.7457668781280518,
      "step": 42137
    },
    {
      "epoch": 4.018497467041016e-05,
      "step": 42137,
      "training_step_time": 8.568928956985474
    },
    {
      "epoch": 4.018592834472656e-05,
      "model_forward_time": 2.927952289581299,
      "step": 42138
    },
    {
      "epoch": 4.018592834472656e-05,
      "step": 42138,
      "training_step_time": 7.88128399848938
    },
    {
      "epoch": 4.018688201904297e-05,
      "model_forward_time": 1.7309720516204834,
      "step": 42139
    },
    {
      "epoch": 4.018688201904297e-05,
      "step": 42139,
      "training_step_time": 8.982191324234009
    },
    {
      "epoch": 4.0187835693359376e-05,
      "grad_norm": 0.1355743110179901,
      "learning_rate": 6.679836395822744e-06,
      "loss": 0.0605,
      "step": 42140
    },
    {
      "epoch": 4.0187835693359376e-05,
      "model_forward_time": 1.7324888706207275,
      "step": 42140
    },
    {
      "dynamics_loss_avg": 0.05565152037888765,
      "epoch": 4.0187835693359376e-05,
      "step": 42140
    },
    {
      "action_loss_avg": 0.001972768345149234,
      "epoch": 4.0187835693359376e-05,
      "step": 42140
    },
    {
      "epoch": 4.0187835693359376e-05,
      "step": 42140,
      "training_step_time": 8.011365175247192
    },
    {
      "epoch": 4.018878936767578e-05,
      "model_forward_time": 1.7343714237213135,
      "step": 42141
    },
    {
      "epoch": 4.018878936767578e-05,
      "step": 42141,
      "training_step_time": 7.733205795288086
    },
    {
      "epoch": 4.018974304199219e-05,
      "model_forward_time": 1.7202398777008057,
      "step": 42142
    },
    {
      "epoch": 4.018974304199219e-05,
      "step": 42142,
      "training_step_time": 7.707178115844727
    },
    {
      "epoch": 4.0190696716308594e-05,
      "model_forward_time": 1.7260403633117676,
      "step": 42143
    },
    {
      "epoch": 4.0190696716308594e-05,
      "step": 42143,
      "training_step_time": 7.563824415206909
    },
    {
      "epoch": 4.0191650390625e-05,
      "model_forward_time": 1.7328836917877197,
      "step": 42144
    },
    {
      "epoch": 4.0191650390625e-05,
      "step": 42144,
      "training_step_time": 8.075361013412476
    },
    {
      "epoch": 4.019260406494141e-05,
      "model_forward_time": 1.7277607917785645,
      "step": 42145
    },
    {
      "epoch": 4.019260406494141e-05,
      "step": 42145,
      "training_step_time": 7.5596678256988525
    },
    {
      "epoch": 4.019355773925781e-05,
      "model_forward_time": 1.743673324584961,
      "step": 42146
    },
    {
      "epoch": 4.019355773925781e-05,
      "step": 42146,
      "training_step_time": 9.403361797332764
    },
    {
      "epoch": 4.019451141357422e-05,
      "model_forward_time": 2.8918838500976562,
      "step": 42147
    },
    {
      "epoch": 4.019451141357422e-05,
      "step": 42147,
      "training_step_time": 7.632756233215332
    },
    {
      "epoch": 4.019546508789063e-05,
      "model_forward_time": 1.7211954593658447,
      "step": 42148
    },
    {
      "epoch": 4.019546508789063e-05,
      "step": 42148,
      "training_step_time": 7.774971008300781
    },
    {
      "epoch": 4.019641876220703e-05,
      "model_forward_time": 1.7337300777435303,
      "step": 42149
    },
    {
      "epoch": 4.019641876220703e-05,
      "step": 42149,
      "training_step_time": 8.436217069625854
    },
    {
      "epoch": 4.0197372436523436e-05,
      "grad_norm": 0.12044448405504227,
      "learning_rate": 6.6782789443336335e-06,
      "loss": 0.0611,
      "step": 42150
    },
    {
      "epoch": 4.0197372436523436e-05,
      "model_forward_time": 1.723294973373413,
      "step": 42150
    },
    {
      "dynamics_loss_avg": 0.05904531152918935,
      "epoch": 4.0197372436523436e-05,
      "step": 42150
    },
    {
      "action_loss_avg": 0.0033383707399480046,
      "epoch": 4.0197372436523436e-05,
      "step": 42150
    },
    {
      "epoch": 4.0197372436523436e-05,
      "step": 42150,
      "training_step_time": 7.683434247970581
    },
    {
      "epoch": 4.019832611083985e-05,
      "model_forward_time": 1.7216386795043945,
      "step": 42151
    },
    {
      "epoch": 4.019832611083985e-05,
      "step": 42151,
      "training_step_time": 7.61008620262146
    },
    {
      "epoch": 4.019927978515625e-05,
      "model_forward_time": 1.7336208820343018,
      "step": 42152
    },
    {
      "epoch": 4.019927978515625e-05,
      "step": 42152,
      "training_step_time": 7.580453634262085
    },
    {
      "epoch": 4.0200233459472655e-05,
      "model_forward_time": 1.7327406406402588,
      "step": 42153
    },
    {
      "epoch": 4.0200233459472655e-05,
      "step": 42153,
      "training_step_time": 7.47512412071228
    },
    {
      "epoch": 4.0201187133789066e-05,
      "model_forward_time": 1.7264063358306885,
      "step": 42154
    },
    {
      "epoch": 4.0201187133789066e-05,
      "step": 42154,
      "training_step_time": 7.550169467926025
    },
    {
      "epoch": 4.020214080810547e-05,
      "model_forward_time": 1.7566416263580322,
      "step": 42155
    },
    {
      "epoch": 4.020214080810547e-05,
      "step": 42155,
      "training_step_time": 8.632716417312622
    },
    {
      "epoch": 4.0203094482421874e-05,
      "model_forward_time": 2.903972864151001,
      "step": 42156
    },
    {
      "epoch": 4.0203094482421874e-05,
      "step": 42156,
      "training_step_time": 7.68831992149353
    },
    {
      "epoch": 4.020404815673828e-05,
      "model_forward_time": 1.723691701889038,
      "step": 42157
    },
    {
      "epoch": 4.020404815673828e-05,
      "step": 42157,
      "training_step_time": 7.769613027572632
    },
    {
      "epoch": 4.020500183105469e-05,
      "model_forward_time": 1.7363276481628418,
      "step": 42158
    },
    {
      "epoch": 4.020500183105469e-05,
      "step": 42158,
      "training_step_time": 8.66607666015625
    },
    {
      "epoch": 4.020595550537109e-05,
      "model_forward_time": 1.7352094650268555,
      "step": 42159
    },
    {
      "epoch": 4.020595550537109e-05,
      "step": 42159,
      "training_step_time": 7.862385034561157
    },
    {
      "epoch": 4.02069091796875e-05,
      "grad_norm": 0.14692017436027527,
      "learning_rate": 6.676721309310462e-06,
      "loss": 0.063,
      "step": 42160
    },
    {
      "epoch": 4.02069091796875e-05,
      "model_forward_time": 1.7316725254058838,
      "step": 42160
    },
    {
      "dynamics_loss_avg": 0.05863828733563423,
      "epoch": 4.02069091796875e-05,
      "step": 42160
    },
    {
      "action_loss_avg": 0.002301870982046239,
      "epoch": 4.02069091796875e-05,
      "step": 42160
    },
    {
      "epoch": 4.02069091796875e-05,
      "step": 42160,
      "training_step_time": 7.694158554077148
    },
    {
      "epoch": 4.020786285400391e-05,
      "model_forward_time": 1.7299296855926514,
      "step": 42161
    },
    {
      "epoch": 4.020786285400391e-05,
      "step": 42161,
      "training_step_time": 8.191940069198608
    },
    {
      "epoch": 4.020881652832031e-05,
      "model_forward_time": 1.7345821857452393,
      "step": 42162
    },
    {
      "epoch": 4.020881652832031e-05,
      "step": 42162,
      "training_step_time": 7.414083480834961
    },
    {
      "epoch": 4.0209770202636716e-05,
      "model_forward_time": 1.7283198833465576,
      "step": 42163
    },
    {
      "epoch": 4.0209770202636716e-05,
      "step": 42163,
      "training_step_time": 8.389774799346924
    },
    {
      "epoch": 4.0210723876953126e-05,
      "model_forward_time": 1.7272562980651855,
      "step": 42164
    },
    {
      "epoch": 4.0210723876953126e-05,
      "step": 42164,
      "training_step_time": 8.689392328262329
    },
    {
      "epoch": 4.021167755126953e-05,
      "model_forward_time": 2.8717963695526123,
      "step": 42165
    },
    {
      "epoch": 4.021167755126953e-05,
      "step": 42165,
      "training_step_time": 8.656038522720337
    },
    {
      "epoch": 4.0212631225585934e-05,
      "model_forward_time": 1.7239742279052734,
      "step": 42166
    },
    {
      "epoch": 4.0212631225585934e-05,
      "step": 42166,
      "training_step_time": 8.61053729057312
    },
    {
      "epoch": 4.0213584899902345e-05,
      "model_forward_time": 1.7231109142303467,
      "step": 42167
    },
    {
      "epoch": 4.0213584899902345e-05,
      "step": 42167,
      "training_step_time": 8.488171100616455
    },
    {
      "epoch": 4.021453857421875e-05,
      "model_forward_time": 1.7424440383911133,
      "step": 42168
    },
    {
      "epoch": 4.021453857421875e-05,
      "step": 42168,
      "training_step_time": 8.443072319030762
    },
    {
      "epoch": 4.021549224853515e-05,
      "model_forward_time": 1.719193458557129,
      "step": 42169
    },
    {
      "epoch": 4.021549224853515e-05,
      "step": 42169,
      "training_step_time": 7.722618579864502
    },
    {
      "epoch": 4.0216445922851564e-05,
      "grad_norm": 0.14424817264080048,
      "learning_rate": 6.6751634909235695e-06,
      "loss": 0.0623,
      "step": 42170
    },
    {
      "epoch": 4.0216445922851564e-05,
      "model_forward_time": 1.7287471294403076,
      "step": 42170
    },
    {
      "dynamics_loss_avg": 0.07242225743830204,
      "epoch": 4.0216445922851564e-05,
      "step": 42170
    },
    {
      "action_loss_avg": 0.003360535454703495,
      "epoch": 4.0216445922851564e-05,
      "step": 42170
    },
    {
      "epoch": 4.0216445922851564e-05,
      "step": 42170,
      "training_step_time": 7.676835298538208
    },
    {
      "epoch": 4.021739959716797e-05,
      "model_forward_time": 1.732889175415039,
      "step": 42171
    },
    {
      "epoch": 4.021739959716797e-05,
      "step": 42171,
      "training_step_time": 7.737634181976318
    },
    {
      "epoch": 4.021835327148437e-05,
      "model_forward_time": 1.7280783653259277,
      "step": 42172
    },
    {
      "epoch": 4.021835327148437e-05,
      "step": 42172,
      "training_step_time": 8.386730909347534
    },
    {
      "epoch": 4.021930694580078e-05,
      "model_forward_time": 1.733924150466919,
      "step": 42173
    },
    {
      "epoch": 4.021930694580078e-05,
      "step": 42173,
      "training_step_time": 8.288393020629883
    },
    {
      "epoch": 4.022026062011719e-05,
      "model_forward_time": 2.852001428604126,
      "step": 42174
    },
    {
      "epoch": 4.022026062011719e-05,
      "step": 42174,
      "training_step_time": 7.696115016937256
    },
    {
      "epoch": 4.022121429443359e-05,
      "model_forward_time": 1.7237133979797363,
      "step": 42175
    },
    {
      "epoch": 4.022121429443359e-05,
      "step": 42175,
      "training_step_time": 9.508323669433594
    },
    {
      "epoch": 4.022216796875e-05,
      "model_forward_time": 1.7215116024017334,
      "step": 42176
    },
    {
      "epoch": 4.022216796875e-05,
      "step": 42176,
      "training_step_time": 8.469677686691284
    },
    {
      "epoch": 4.0223121643066406e-05,
      "model_forward_time": 1.7478139400482178,
      "step": 42177
    },
    {
      "epoch": 4.0223121643066406e-05,
      "step": 42177,
      "training_step_time": 9.369477272033691
    },
    {
      "epoch": 4.022407531738281e-05,
      "model_forward_time": 1.7406630516052246,
      "step": 42178
    },
    {
      "epoch": 4.022407531738281e-05,
      "step": 42178,
      "training_step_time": 7.688634395599365
    },
    {
      "epoch": 4.022502899169922e-05,
      "model_forward_time": 1.7307963371276855,
      "step": 42179
    },
    {
      "epoch": 4.022502899169922e-05,
      "step": 42179,
      "training_step_time": 7.678970098495483
    },
    {
      "epoch": 4.0225982666015625e-05,
      "grad_norm": 0.1232469379901886,
      "learning_rate": 6.673605489343316e-06,
      "loss": 0.0633,
      "step": 42180
    },
    {
      "epoch": 4.0225982666015625e-05,
      "model_forward_time": 1.7228338718414307,
      "step": 42180
    },
    {
      "dynamics_loss_avg": 0.05045284554362297,
      "epoch": 4.0225982666015625e-05,
      "step": 42180
    },
    {
      "action_loss_avg": 0.0028840085928095506,
      "epoch": 4.0225982666015625e-05,
      "step": 42180
    },
    {
      "epoch": 4.0225982666015625e-05,
      "step": 42180,
      "training_step_time": 7.728591442108154
    },
    {
      "epoch": 4.022693634033203e-05,
      "model_forward_time": 1.7281804084777832,
      "step": 42181
    },
    {
      "epoch": 4.022693634033203e-05,
      "step": 42181,
      "training_step_time": 8.176615715026855
    },
    {
      "epoch": 4.022789001464844e-05,
      "model_forward_time": 1.717491626739502,
      "step": 42182
    },
    {
      "epoch": 4.022789001464844e-05,
      "step": 42182,
      "training_step_time": 8.277280569076538
    },
    {
      "epoch": 4.0228843688964844e-05,
      "model_forward_time": 1.745091199874878,
      "step": 42183
    },
    {
      "epoch": 4.0228843688964844e-05,
      "step": 42183,
      "training_step_time": 9.595637083053589
    },
    {
      "epoch": 4.022979736328125e-05,
      "model_forward_time": 1.7381095886230469,
      "step": 42184
    },
    {
      "epoch": 4.022979736328125e-05,
      "step": 42184,
      "training_step_time": 9.222396612167358
    },
    {
      "epoch": 4.023075103759766e-05,
      "model_forward_time": 1.7365221977233887,
      "step": 42185
    },
    {
      "epoch": 4.023075103759766e-05,
      "step": 42185,
      "training_step_time": 7.748916387557983
    },
    {
      "epoch": 4.023170471191406e-05,
      "model_forward_time": 1.7408087253570557,
      "step": 42186
    },
    {
      "epoch": 4.023170471191406e-05,
      "step": 42186,
      "training_step_time": 9.451242685317993
    },
    {
      "epoch": 4.0232658386230466e-05,
      "model_forward_time": 1.7267911434173584,
      "step": 42187
    },
    {
      "epoch": 4.0232658386230466e-05,
      "step": 42187,
      "training_step_time": 7.630436182022095
    },
    {
      "epoch": 4.023361206054688e-05,
      "model_forward_time": 1.7252774238586426,
      "step": 42188
    },
    {
      "epoch": 4.023361206054688e-05,
      "step": 42188,
      "training_step_time": 7.633324384689331
    },
    {
      "epoch": 4.023456573486328e-05,
      "model_forward_time": 1.7310967445373535,
      "step": 42189
    },
    {
      "epoch": 4.023456573486328e-05,
      "step": 42189,
      "training_step_time": 7.749822378158569
    },
    {
      "epoch": 4.0235519409179685e-05,
      "grad_norm": 0.12847106158733368,
      "learning_rate": 6.672047304740081e-06,
      "loss": 0.0625,
      "step": 42190
    },
    {
      "epoch": 4.0235519409179685e-05,
      "model_forward_time": 1.7107713222503662,
      "step": 42190
    },
    {
      "dynamics_loss_avg": 0.059575132094323635,
      "epoch": 4.0235519409179685e-05,
      "step": 42190
    },
    {
      "action_loss_avg": 0.001830526109552011,
      "epoch": 4.0235519409179685e-05,
      "step": 42190
    },
    {
      "epoch": 4.0235519409179685e-05,
      "step": 42190,
      "training_step_time": 8.399853944778442
    },
    {
      "epoch": 4.0236473083496096e-05,
      "model_forward_time": 1.732020616531372,
      "step": 42191
    },
    {
      "epoch": 4.0236473083496096e-05,
      "step": 42191,
      "training_step_time": 7.699145317077637
    },
    {
      "epoch": 4.02374267578125e-05,
      "model_forward_time": 1.7479429244995117,
      "step": 42192
    },
    {
      "epoch": 4.02374267578125e-05,
      "step": 42192,
      "training_step_time": 7.618754625320435
    },
    {
      "epoch": 4.0238380432128904e-05,
      "model_forward_time": 2.8834280967712402,
      "step": 42193
    },
    {
      "epoch": 4.0238380432128904e-05,
      "step": 42193,
      "training_step_time": 8.897332429885864
    },
    {
      "epoch": 4.0239334106445315e-05,
      "model_forward_time": 1.7341601848602295,
      "step": 42194
    },
    {
      "epoch": 4.0239334106445315e-05,
      "step": 42194,
      "training_step_time": 8.873402118682861
    },
    {
      "epoch": 4.024028778076172e-05,
      "model_forward_time": 1.714684247970581,
      "step": 42195
    },
    {
      "epoch": 4.024028778076172e-05,
      "step": 42195,
      "training_step_time": 8.64302921295166
    },
    {
      "epoch": 4.024124145507812e-05,
      "model_forward_time": 1.7333734035491943,
      "step": 42196
    },
    {
      "epoch": 4.024124145507812e-05,
      "step": 42196,
      "training_step_time": 7.779938697814941
    },
    {
      "epoch": 4.0242195129394534e-05,
      "model_forward_time": 1.727529525756836,
      "step": 42197
    },
    {
      "epoch": 4.0242195129394534e-05,
      "step": 42197,
      "training_step_time": 7.631541967391968
    },
    {
      "epoch": 4.024314880371094e-05,
      "model_forward_time": 1.7333521842956543,
      "step": 42198
    },
    {
      "epoch": 4.024314880371094e-05,
      "step": 42198,
      "training_step_time": 7.643483400344849
    },
    {
      "epoch": 4.024410247802734e-05,
      "model_forward_time": 1.72214937210083,
      "step": 42199
    },
    {
      "epoch": 4.024410247802734e-05,
      "step": 42199,
      "training_step_time": 7.4393150806427
    },
    {
      "epoch": 4.024505615234375e-05,
      "grad_norm": 0.13267101347446442,
      "learning_rate": 6.670488937284268e-06,
      "loss": 0.0578,
      "step": 42200
    },
    {
      "epoch": 4.024505615234375e-05,
      "model_forward_time": 1.731029987335205,
      "step": 42200
    },
    {
      "dynamics_loss_avg": 0.05416299309581518,
      "epoch": 4.024505615234375e-05,
      "step": 42200
    },
    {
      "action_loss_avg": 0.0019054421922191978,
      "epoch": 4.024505615234375e-05,
      "step": 42200
    },
    {
      "epoch": 4.024505615234375e-05,
      "step": 42200,
      "training_step_time": 7.62317156791687
    },
    {
      "epoch": 4.024600982666016e-05,
      "model_forward_time": 1.731121301651001,
      "step": 42201
    },
    {
      "epoch": 4.024600982666016e-05,
      "step": 42201,
      "training_step_time": 7.541446208953857
    },
    {
      "epoch": 4.024696350097656e-05,
      "model_forward_time": 2.9211065769195557,
      "step": 42202
    },
    {
      "epoch": 4.024696350097656e-05,
      "step": 42202,
      "training_step_time": 9.354703903198242
    },
    {
      "epoch": 4.024791717529297e-05,
      "model_forward_time": 1.718127965927124,
      "step": 42203
    },
    {
      "epoch": 4.024791717529297e-05,
      "step": 42203,
      "training_step_time": 7.710226774215698
    },
    {
      "epoch": 4.0248870849609376e-05,
      "model_forward_time": 1.7266261577606201,
      "step": 42204
    },
    {
      "epoch": 4.0248870849609376e-05,
      "step": 42204,
      "training_step_time": 9.459977149963379
    },
    {
      "epoch": 4.024982452392578e-05,
      "model_forward_time": 1.7426419258117676,
      "step": 42205
    },
    {
      "epoch": 4.024982452392578e-05,
      "step": 42205,
      "training_step_time": 8.637224912643433
    },
    {
      "epoch": 4.025077819824219e-05,
      "model_forward_time": 1.7314574718475342,
      "step": 42206
    },
    {
      "epoch": 4.025077819824219e-05,
      "step": 42206,
      "training_step_time": 7.51978325843811
    },
    {
      "epoch": 4.0251731872558594e-05,
      "model_forward_time": 1.729111909866333,
      "step": 42207
    },
    {
      "epoch": 4.0251731872558594e-05,
      "step": 42207,
      "training_step_time": 7.663980484008789
    },
    {
      "epoch": 4.0252685546875e-05,
      "model_forward_time": 1.7164700031280518,
      "step": 42208
    },
    {
      "epoch": 4.0252685546875e-05,
      "step": 42208,
      "training_step_time": 7.420260429382324
    },
    {
      "epoch": 4.025363922119141e-05,
      "model_forward_time": 1.7324512004852295,
      "step": 42209
    },
    {
      "epoch": 4.025363922119141e-05,
      "step": 42209,
      "training_step_time": 7.687526702880859
    },
    {
      "epoch": 4.025459289550781e-05,
      "grad_norm": 0.13683892786502838,
      "learning_rate": 6.668930387146295e-06,
      "loss": 0.06,
      "step": 42210
    },
    {
      "epoch": 4.025459289550781e-05,
      "model_forward_time": 1.778630018234253,
      "step": 42210
    },
    {
      "dynamics_loss_avg": 0.0739943671040237,
      "epoch": 4.025459289550781e-05,
      "step": 42210
    },
    {
      "action_loss_avg": 0.0017885267327073962,
      "epoch": 4.025459289550781e-05,
      "step": 42210
    },
    {
      "epoch": 4.025459289550781e-05,
      "step": 42210,
      "training_step_time": 7.559086799621582
    },
    {
      "epoch": 4.025554656982422e-05,
      "model_forward_time": 2.888972043991089,
      "step": 42211
    },
    {
      "epoch": 4.025554656982422e-05,
      "step": 42211,
      "training_step_time": 7.643883943557739
    },
    {
      "epoch": 4.025650024414063e-05,
      "model_forward_time": 1.724482774734497,
      "step": 42212
    },
    {
      "epoch": 4.025650024414063e-05,
      "step": 42212,
      "training_step_time": 10.444074630737305
    },
    {
      "epoch": 4.025745391845703e-05,
      "model_forward_time": 1.727637529373169,
      "step": 42213
    },
    {
      "epoch": 4.025745391845703e-05,
      "step": 42213,
      "training_step_time": 9.54150938987732
    },
    {
      "epoch": 4.0258407592773436e-05,
      "model_forward_time": 1.8387041091918945,
      "step": 42214
    },
    {
      "epoch": 4.0258407592773436e-05,
      "step": 42214,
      "training_step_time": 8.715134859085083
    },
    {
      "epoch": 4.025936126708985e-05,
      "model_forward_time": 1.7524464130401611,
      "step": 42215
    },
    {
      "epoch": 4.025936126708985e-05,
      "step": 42215,
      "training_step_time": 7.678854703903198
    },
    {
      "epoch": 4.026031494140625e-05,
      "model_forward_time": 1.732163906097412,
      "step": 42216
    },
    {
      "epoch": 4.026031494140625e-05,
      "step": 42216,
      "training_step_time": 7.767160892486572
    },
    {
      "epoch": 4.0261268615722655e-05,
      "model_forward_time": 1.7426676750183105,
      "step": 42217
    },
    {
      "epoch": 4.0261268615722655e-05,
      "step": 42217,
      "training_step_time": 7.740020275115967
    },
    {
      "epoch": 4.0262222290039066e-05,
      "model_forward_time": 1.7270021438598633,
      "step": 42218
    },
    {
      "epoch": 4.0262222290039066e-05,
      "step": 42218,
      "training_step_time": 8.483054161071777
    },
    {
      "epoch": 4.026317596435547e-05,
      "model_forward_time": 1.746246099472046,
      "step": 42219
    },
    {
      "epoch": 4.026317596435547e-05,
      "step": 42219,
      "training_step_time": 7.676482200622559
    },
    {
      "epoch": 4.0264129638671874e-05,
      "grad_norm": 0.09682824462652206,
      "learning_rate": 6.667371654496606e-06,
      "loss": 0.0581,
      "step": 42220
    },
    {
      "epoch": 4.0264129638671874e-05,
      "model_forward_time": 2.8811001777648926,
      "step": 42220
    },
    {
      "dynamics_loss_avg": 0.05195126086473465,
      "epoch": 4.0264129638671874e-05,
      "step": 42220
    },
    {
      "action_loss_avg": 0.0025151130103040487,
      "epoch": 4.0264129638671874e-05,
      "step": 42220
    },
    {
      "epoch": 4.0264129638671874e-05,
      "step": 42220,
      "training_step_time": 9.33065414428711
    },
    {
      "epoch": 4.026508331298828e-05,
      "model_forward_time": 1.735764503479004,
      "step": 42221
    },
    {
      "epoch": 4.026508331298828e-05,
      "step": 42221,
      "training_step_time": 8.782500743865967
    },
    {
      "epoch": 4.026603698730469e-05,
      "model_forward_time": 1.7229530811309814,
      "step": 42222
    },
    {
      "epoch": 4.026603698730469e-05,
      "step": 42222,
      "training_step_time": 7.718800783157349
    },
    {
      "epoch": 4.026699066162109e-05,
      "model_forward_time": 1.7687909603118896,
      "step": 42223
    },
    {
      "epoch": 4.026699066162109e-05,
      "step": 42223,
      "training_step_time": 8.393308639526367
    },
    {
      "epoch": 4.02679443359375e-05,
      "model_forward_time": 1.7638864517211914,
      "step": 42224
    },
    {
      "epoch": 4.02679443359375e-05,
      "step": 42224,
      "training_step_time": 7.529511213302612
    },
    {
      "epoch": 4.026889801025391e-05,
      "model_forward_time": 1.7565319538116455,
      "step": 42225
    },
    {
      "epoch": 4.026889801025391e-05,
      "step": 42225,
      "training_step_time": 7.587072372436523
    },
    {
      "epoch": 4.026985168457031e-05,
      "model_forward_time": 1.730656623840332,
      "step": 42226
    },
    {
      "epoch": 4.026985168457031e-05,
      "step": 42226,
      "training_step_time": 7.688252925872803
    },
    {
      "epoch": 4.0270805358886716e-05,
      "model_forward_time": 1.7354726791381836,
      "step": 42227
    },
    {
      "epoch": 4.0270805358886716e-05,
      "step": 42227,
      "training_step_time": 7.589978456497192
    },
    {
      "epoch": 4.0271759033203126e-05,
      "model_forward_time": 1.7339389324188232,
      "step": 42228
    },
    {
      "epoch": 4.0271759033203126e-05,
      "step": 42228,
      "training_step_time": 8.55262804031372
    },
    {
      "epoch": 4.027271270751953e-05,
      "model_forward_time": 2.9180376529693604,
      "step": 42229
    },
    {
      "epoch": 4.027271270751953e-05,
      "step": 42229,
      "training_step_time": 7.631995677947998
    },
    {
      "epoch": 4.0273666381835935e-05,
      "grad_norm": 0.09286630153656006,
      "learning_rate": 6.6658127395056595e-06,
      "loss": 0.0539,
      "step": 42230
    },
    {
      "epoch": 4.0273666381835935e-05,
      "model_forward_time": 1.7355542182922363,
      "step": 42230
    },
    {
      "dynamics_loss_avg": 0.061791686341166496,
      "epoch": 4.0273666381835935e-05,
      "step": 42230
    },
    {
      "action_loss_avg": 0.0022563475649803877,
      "epoch": 4.0273666381835935e-05,
      "step": 42230
    },
    {
      "epoch": 4.0273666381835935e-05,
      "step": 42230,
      "training_step_time": 7.530798435211182
    },
    {
      "epoch": 4.0274620056152345e-05,
      "model_forward_time": 1.7561469078063965,
      "step": 42231
    },
    {
      "epoch": 4.0274620056152345e-05,
      "step": 42231,
      "training_step_time": 7.643503427505493
    },
    {
      "epoch": 4.027557373046875e-05,
      "model_forward_time": 1.7271502017974854,
      "step": 42232
    },
    {
      "epoch": 4.027557373046875e-05,
      "step": 42232,
      "training_step_time": 7.797082424163818
    },
    {
      "epoch": 4.027652740478515e-05,
      "model_forward_time": 1.846184253692627,
      "step": 42233
    },
    {
      "epoch": 4.027652740478515e-05,
      "step": 42233,
      "training_step_time": 8.19127106666565
    },
    {
      "epoch": 4.0277481079101564e-05,
      "model_forward_time": 1.7533454895019531,
      "step": 42234
    },
    {
      "epoch": 4.0277481079101564e-05,
      "step": 42234,
      "training_step_time": 7.843773365020752
    },
    {
      "epoch": 4.027843475341797e-05,
      "model_forward_time": 1.727583408355713,
      "step": 42235
    },
    {
      "epoch": 4.027843475341797e-05,
      "step": 42235,
      "training_step_time": 7.649057626724243
    },
    {
      "epoch": 4.027938842773437e-05,
      "model_forward_time": 1.7368888854980469,
      "step": 42236
    },
    {
      "epoch": 4.027938842773437e-05,
      "step": 42236,
      "training_step_time": 7.566016674041748
    },
    {
      "epoch": 4.028034210205078e-05,
      "model_forward_time": 1.7452313899993896,
      "step": 42237
    },
    {
      "epoch": 4.028034210205078e-05,
      "step": 42237,
      "training_step_time": 7.56291389465332
    },
    {
      "epoch": 4.028129577636719e-05,
      "model_forward_time": 2.8865442276000977,
      "step": 42238
    },
    {
      "epoch": 4.028129577636719e-05,
      "step": 42238,
      "training_step_time": 8.215092897415161
    },
    {
      "epoch": 4.028224945068359e-05,
      "model_forward_time": 1.7530088424682617,
      "step": 42239
    },
    {
      "epoch": 4.028224945068359e-05,
      "step": 42239,
      "training_step_time": 8.550849914550781
    },
    {
      "epoch": 4.0283203125e-05,
      "grad_norm": 0.08825954049825668,
      "learning_rate": 6.664253642343935e-06,
      "loss": 0.0562,
      "step": 42240
    },
    {
      "epoch": 4.0283203125e-05,
      "model_forward_time": 1.7311670780181885,
      "step": 42240
    },
    {
      "dynamics_loss_avg": 0.052247377950698136,
      "epoch": 4.0283203125e-05,
      "step": 42240
    },
    {
      "action_loss_avg": 0.002108241035602987,
      "epoch": 4.0283203125e-05,
      "step": 42240
    },
    {
      "epoch": 4.0283203125e-05,
      "step": 42240,
      "training_step_time": 8.266571521759033
    },
    {
      "epoch": 4.0284156799316406e-05,
      "model_forward_time": 1.7334249019622803,
      "step": 42241
    },
    {
      "epoch": 4.0284156799316406e-05,
      "step": 42241,
      "training_step_time": 8.736010551452637
    },
    {
      "epoch": 4.028511047363281e-05,
      "model_forward_time": 1.8391847610473633,
      "step": 42242
    },
    {
      "epoch": 4.028511047363281e-05,
      "step": 42242,
      "training_step_time": 7.615007400512695
    },
    {
      "epoch": 4.028606414794922e-05,
      "model_forward_time": 1.732607126235962,
      "step": 42243
    },
    {
      "epoch": 4.028606414794922e-05,
      "step": 42243,
      "training_step_time": 7.48015284538269
    },
    {
      "epoch": 4.0287017822265625e-05,
      "model_forward_time": 1.7471156120300293,
      "step": 42244
    },
    {
      "epoch": 4.0287017822265625e-05,
      "step": 42244,
      "training_step_time": 8.730773210525513
    },
    {
      "epoch": 4.028797149658203e-05,
      "model_forward_time": 1.7563598155975342,
      "step": 42245
    },
    {
      "epoch": 4.028797149658203e-05,
      "step": 42245,
      "training_step_time": 7.563307046890259
    },
    {
      "epoch": 4.028892517089844e-05,
      "model_forward_time": 1.7417020797729492,
      "step": 42246
    },
    {
      "epoch": 4.028892517089844e-05,
      "step": 42246,
      "training_step_time": 9.535608291625977
    },
    {
      "epoch": 4.0289878845214844e-05,
      "model_forward_time": 2.8707468509674072,
      "step": 42247
    },
    {
      "epoch": 4.0289878845214844e-05,
      "step": 42247,
      "training_step_time": 7.575177431106567
    },
    {
      "epoch": 4.029083251953125e-05,
      "model_forward_time": 1.745380163192749,
      "step": 42248
    },
    {
      "epoch": 4.029083251953125e-05,
      "step": 42248,
      "training_step_time": 7.584391832351685
    },
    {
      "epoch": 4.029178619384766e-05,
      "model_forward_time": 1.7439072132110596,
      "step": 42249
    },
    {
      "epoch": 4.029178619384766e-05,
      "step": 42249,
      "training_step_time": 7.7382965087890625
    },
    {
      "epoch": 4.029273986816406e-05,
      "grad_norm": 0.10793135315179825,
      "learning_rate": 6.662694363181935e-06,
      "loss": 0.0561,
      "step": 42250
    },
    {
      "epoch": 4.029273986816406e-05,
      "model_forward_time": 1.7333323955535889,
      "step": 42250
    },
    {
      "dynamics_loss_avg": 0.06398148499429226,
      "epoch": 4.029273986816406e-05,
      "step": 42250
    },
    {
      "action_loss_avg": 0.0027632509852992372,
      "epoch": 4.029273986816406e-05,
      "step": 42250
    },
    {
      "epoch": 4.029273986816406e-05,
      "step": 42250,
      "training_step_time": 7.62900710105896
    },
    {
      "epoch": 4.0293693542480467e-05,
      "model_forward_time": 1.829397201538086,
      "step": 42251
    },
    {
      "epoch": 4.0293693542480467e-05,
      "step": 42251,
      "training_step_time": 8.985363960266113
    },
    {
      "epoch": 4.029464721679688e-05,
      "model_forward_time": 1.7292232513427734,
      "step": 42252
    },
    {
      "epoch": 4.029464721679688e-05,
      "step": 42252,
      "training_step_time": 7.562246561050415
    },
    {
      "epoch": 4.029560089111328e-05,
      "model_forward_time": 1.746915578842163,
      "step": 42253
    },
    {
      "epoch": 4.029560089111328e-05,
      "step": 42253,
      "training_step_time": 7.6467673778533936
    },
    {
      "epoch": 4.0296554565429685e-05,
      "model_forward_time": 1.7560691833496094,
      "step": 42254
    },
    {
      "epoch": 4.0296554565429685e-05,
      "step": 42254,
      "training_step_time": 7.77493143081665
    },
    {
      "epoch": 4.0297508239746096e-05,
      "model_forward_time": 1.7225241661071777,
      "step": 42255
    },
    {
      "epoch": 4.0297508239746096e-05,
      "step": 42255,
      "training_step_time": 8.124523639678955
    },
    {
      "epoch": 4.02984619140625e-05,
      "model_forward_time": 2.880557060241699,
      "step": 42256
    },
    {
      "epoch": 4.02984619140625e-05,
      "step": 42256,
      "training_step_time": 7.607306003570557
    },
    {
      "epoch": 4.0299415588378904e-05,
      "model_forward_time": 1.751147747039795,
      "step": 42257
    },
    {
      "epoch": 4.0299415588378904e-05,
      "step": 42257,
      "training_step_time": 9.619237422943115
    },
    {
      "epoch": 4.0300369262695315e-05,
      "model_forward_time": 1.73353910446167,
      "step": 42258
    },
    {
      "epoch": 4.0300369262695315e-05,
      "step": 42258,
      "training_step_time": 7.705469369888306
    },
    {
      "epoch": 4.030132293701172e-05,
      "model_forward_time": 1.7514657974243164,
      "step": 42259
    },
    {
      "epoch": 4.030132293701172e-05,
      "step": 42259,
      "training_step_time": 8.54112720489502
    },
    {
      "epoch": 4.030227661132812e-05,
      "grad_norm": 0.106565922498703,
      "learning_rate": 6.6611349021901795e-06,
      "loss": 0.0571,
      "step": 42260
    },
    {
      "epoch": 4.030227661132812e-05,
      "model_forward_time": 1.7826004028320312,
      "step": 42260
    },
    {
      "dynamics_loss_avg": 0.05999317467212677,
      "epoch": 4.030227661132812e-05,
      "step": 42260
    },
    {
      "action_loss_avg": 0.001345900003798306,
      "epoch": 4.030227661132812e-05,
      "step": 42260
    },
    {
      "epoch": 4.030227661132812e-05,
      "step": 42260,
      "training_step_time": 7.62155818939209
    },
    {
      "epoch": 4.0303230285644534e-05,
      "model_forward_time": 1.7362027168273926,
      "step": 42261
    },
    {
      "epoch": 4.0303230285644534e-05,
      "step": 42261,
      "training_step_time": 7.720952033996582
    },
    {
      "epoch": 4.030418395996094e-05,
      "model_forward_time": 1.750375747680664,
      "step": 42262
    },
    {
      "epoch": 4.030418395996094e-05,
      "step": 42262,
      "training_step_time": 7.604355812072754
    },
    {
      "epoch": 4.030513763427734e-05,
      "model_forward_time": 1.7358314990997314,
      "step": 42263
    },
    {
      "epoch": 4.030513763427734e-05,
      "step": 42263,
      "training_step_time": 7.762685060501099
    },
    {
      "epoch": 4.030609130859375e-05,
      "model_forward_time": 1.7379589080810547,
      "step": 42264
    },
    {
      "epoch": 4.030609130859375e-05,
      "step": 42264,
      "training_step_time": 7.643980264663696
    },
    {
      "epoch": 4.030704498291016e-05,
      "model_forward_time": 2.8925836086273193,
      "step": 42265
    },
    {
      "epoch": 4.030704498291016e-05,
      "step": 42265,
      "training_step_time": 7.611075401306152
    },
    {
      "epoch": 4.030799865722656e-05,
      "model_forward_time": 1.7297110557556152,
      "step": 42266
    },
    {
      "epoch": 4.030799865722656e-05,
      "step": 42266,
      "training_step_time": 8.28337574005127
    },
    {
      "epoch": 4.030895233154297e-05,
      "model_forward_time": 1.7375257015228271,
      "step": 42267
    },
    {
      "epoch": 4.030895233154297e-05,
      "step": 42267,
      "training_step_time": 8.520140886306763
    },
    {
      "epoch": 4.0309906005859376e-05,
      "model_forward_time": 1.7594575881958008,
      "step": 42268
    },
    {
      "epoch": 4.0309906005859376e-05,
      "step": 42268,
      "training_step_time": 8.828133344650269
    },
    {
      "epoch": 4.031085968017578e-05,
      "model_forward_time": 1.7497520446777344,
      "step": 42269
    },
    {
      "epoch": 4.031085968017578e-05,
      "step": 42269,
      "training_step_time": 8.633681297302246
    },
    {
      "epoch": 4.031181335449219e-05,
      "grad_norm": 0.15039505064487457,
      "learning_rate": 6.659575259539208e-06,
      "loss": 0.0592,
      "step": 42270
    },
    {
      "epoch": 4.031181335449219e-05,
      "model_forward_time": 1.7286207675933838,
      "step": 42270
    },
    {
      "dynamics_loss_avg": 0.07370623275637626,
      "epoch": 4.031181335449219e-05,
      "step": 42270
    },
    {
      "action_loss_avg": 0.0017319546604994684,
      "epoch": 4.031181335449219e-05,
      "step": 42270
    },
    {
      "epoch": 4.031181335449219e-05,
      "step": 42270,
      "training_step_time": 7.610384225845337
    },
    {
      "epoch": 4.0312767028808595e-05,
      "model_forward_time": 1.7085258960723877,
      "step": 42271
    },
    {
      "epoch": 4.0312767028808595e-05,
      "step": 42271,
      "training_step_time": 7.044049978256226
    },
    {
      "epoch": 4.0313720703125e-05,
      "model_forward_time": 1.7361531257629395,
      "step": 42272
    },
    {
      "epoch": 4.0313720703125e-05,
      "step": 42272,
      "training_step_time": 7.663283586502075
    },
    {
      "epoch": 4.031467437744141e-05,
      "model_forward_time": 1.742565393447876,
      "step": 42273
    },
    {
      "epoch": 4.031467437744141e-05,
      "step": 42273,
      "training_step_time": 7.552260637283325
    },
    {
      "epoch": 4.031562805175781e-05,
      "model_forward_time": 2.873652219772339,
      "step": 42274
    },
    {
      "epoch": 4.031562805175781e-05,
      "step": 42274,
      "training_step_time": 7.594958066940308
    },
    {
      "epoch": 4.031658172607422e-05,
      "model_forward_time": 1.7463746070861816,
      "step": 42275
    },
    {
      "epoch": 4.031658172607422e-05,
      "step": 42275,
      "training_step_time": 9.482953786849976
    },
    {
      "epoch": 4.031753540039063e-05,
      "model_forward_time": 1.732764720916748,
      "step": 42276
    },
    {
      "epoch": 4.031753540039063e-05,
      "step": 42276,
      "training_step_time": 9.144816637039185
    },
    {
      "epoch": 4.031848907470703e-05,
      "model_forward_time": 1.7341949939727783,
      "step": 42277
    },
    {
      "epoch": 4.031848907470703e-05,
      "step": 42277,
      "training_step_time": 7.633325099945068
    },
    {
      "epoch": 4.0319442749023436e-05,
      "model_forward_time": 1.7503111362457275,
      "step": 42278
    },
    {
      "epoch": 4.0319442749023436e-05,
      "step": 42278,
      "training_step_time": 8.234829902648926
    },
    {
      "epoch": 4.032039642333985e-05,
      "model_forward_time": 1.844947099685669,
      "step": 42279
    },
    {
      "epoch": 4.032039642333985e-05,
      "step": 42279,
      "training_step_time": 7.566791772842407
    },
    {
      "epoch": 4.032135009765625e-05,
      "grad_norm": 0.11060649901628494,
      "learning_rate": 6.658015435399581e-06,
      "loss": 0.0567,
      "step": 42280
    },
    {
      "epoch": 4.032135009765625e-05,
      "model_forward_time": 1.7227051258087158,
      "step": 42280
    },
    {
      "dynamics_loss_avg": 0.05416307942941785,
      "epoch": 4.032135009765625e-05,
      "step": 42280
    },
    {
      "action_loss_avg": 0.001977872379939072,
      "epoch": 4.032135009765625e-05,
      "step": 42280
    },
    {
      "epoch": 4.032135009765625e-05,
      "step": 42280,
      "training_step_time": 7.621456861495972
    },
    {
      "epoch": 4.0322303771972655e-05,
      "model_forward_time": 1.7431764602661133,
      "step": 42281
    },
    {
      "epoch": 4.0322303771972655e-05,
      "step": 42281,
      "training_step_time": 7.704004764556885
    },
    {
      "epoch": 4.032325744628906e-05,
      "model_forward_time": 1.728689193725586,
      "step": 42282
    },
    {
      "epoch": 4.032325744628906e-05,
      "step": 42282,
      "training_step_time": 7.529808521270752
    },
    {
      "epoch": 4.032421112060547e-05,
      "model_forward_time": 2.881571054458618,
      "step": 42283
    },
    {
      "epoch": 4.032421112060547e-05,
      "step": 42283,
      "training_step_time": 7.58734393119812
    },
    {
      "epoch": 4.0325164794921874e-05,
      "model_forward_time": 1.7480189800262451,
      "step": 42284
    },
    {
      "epoch": 4.0325164794921874e-05,
      "step": 42284,
      "training_step_time": 8.557125568389893
    },
    {
      "epoch": 4.032611846923828e-05,
      "model_forward_time": 1.7361528873443604,
      "step": 42285
    },
    {
      "epoch": 4.032611846923828e-05,
      "step": 42285,
      "training_step_time": 7.609059572219849
    },
    {
      "epoch": 4.032707214355469e-05,
      "model_forward_time": 1.7344751358032227,
      "step": 42286
    },
    {
      "epoch": 4.032707214355469e-05,
      "step": 42286,
      "training_step_time": 7.651257276535034
    },
    {
      "epoch": 4.032802581787109e-05,
      "model_forward_time": 1.7377326488494873,
      "step": 42287
    },
    {
      "epoch": 4.032802581787109e-05,
      "step": 42287,
      "training_step_time": 8.27965259552002
    },
    {
      "epoch": 4.03289794921875e-05,
      "model_forward_time": 1.8476202487945557,
      "step": 42288
    },
    {
      "epoch": 4.03289794921875e-05,
      "step": 42288,
      "training_step_time": 7.469274520874023
    },
    {
      "epoch": 4.032993316650391e-05,
      "model_forward_time": 1.7389614582061768,
      "step": 42289
    },
    {
      "epoch": 4.032993316650391e-05,
      "step": 42289,
      "training_step_time": 7.533250331878662
    },
    {
      "epoch": 4.033088684082031e-05,
      "grad_norm": 0.09476611018180847,
      "learning_rate": 6.65645542994188e-06,
      "loss": 0.0542,
      "step": 42290
    },
    {
      "epoch": 4.033088684082031e-05,
      "model_forward_time": 1.7323122024536133,
      "step": 42290
    },
    {
      "dynamics_loss_avg": 0.04278845451772213,
      "epoch": 4.033088684082031e-05,
      "step": 42290
    },
    {
      "action_loss_avg": 0.0016992968274280428,
      "epoch": 4.033088684082031e-05,
      "step": 42290
    },
    {
      "epoch": 4.033088684082031e-05,
      "step": 42290,
      "training_step_time": 7.6102235317230225
    },
    {
      "epoch": 4.0331840515136716e-05,
      "model_forward_time": 1.7296488285064697,
      "step": 42291
    },
    {
      "epoch": 4.0331840515136716e-05,
      "step": 42291,
      "training_step_time": 7.568122863769531
    },
    {
      "epoch": 4.0332794189453127e-05,
      "model_forward_time": 2.871967077255249,
      "step": 42292
    },
    {
      "epoch": 4.0332794189453127e-05,
      "step": 42292,
      "training_step_time": 8.299726009368896
    },
    {
      "epoch": 4.033374786376953e-05,
      "model_forward_time": 1.7428791522979736,
      "step": 42293
    },
    {
      "epoch": 4.033374786376953e-05,
      "step": 42293,
      "training_step_time": 7.583153963088989
    },
    {
      "epoch": 4.0334701538085935e-05,
      "model_forward_time": 1.7319836616516113,
      "step": 42294
    },
    {
      "epoch": 4.0334701538085935e-05,
      "step": 42294,
      "training_step_time": 9.164905071258545
    },
    {
      "epoch": 4.0335655212402345e-05,
      "model_forward_time": 1.7459118366241455,
      "step": 42295
    },
    {
      "epoch": 4.0335655212402345e-05,
      "step": 42295,
      "training_step_time": 7.757561922073364
    },
    {
      "epoch": 4.033660888671875e-05,
      "model_forward_time": 1.7347378730773926,
      "step": 42296
    },
    {
      "epoch": 4.033660888671875e-05,
      "step": 42296,
      "training_step_time": 8.622430562973022
    },
    {
      "epoch": 4.0337562561035153e-05,
      "model_forward_time": 1.831892728805542,
      "step": 42297
    },
    {
      "epoch": 4.0337562561035153e-05,
      "step": 42297,
      "training_step_time": 8.603931188583374
    },
    {
      "epoch": 4.0338516235351564e-05,
      "model_forward_time": 1.7284023761749268,
      "step": 42298
    },
    {
      "epoch": 4.0338516235351564e-05,
      "step": 42298,
      "training_step_time": 7.651270866394043
    },
    {
      "epoch": 4.033946990966797e-05,
      "model_forward_time": 1.7274305820465088,
      "step": 42299
    },
    {
      "epoch": 4.033946990966797e-05,
      "step": 42299,
      "training_step_time": 7.621001720428467
    },
    {
      "epoch": 4.034042358398437e-05,
      "grad_norm": 0.11845067143440247,
      "learning_rate": 6.6548952433367034e-06,
      "loss": 0.0619,
      "step": 42300
    },
    {
      "epoch": 4.034042358398437e-05,
      "model_forward_time": 1.7271947860717773,
      "step": 42300
    },
    {
      "dynamics_loss_avg": 0.05509896102594212,
      "epoch": 4.034042358398437e-05,
      "step": 42300
    },
    {
      "action_loss_avg": 0.001525333005702123,
      "epoch": 4.034042358398437e-05,
      "step": 42300
    },
    {
      "epoch": 4.034042358398437e-05,
      "step": 42300,
      "training_step_time": 7.767728090286255
    },
    {
      "epoch": 4.034137725830078e-05,
      "model_forward_time": 2.880549430847168,
      "step": 42301
    },
    {
      "epoch": 4.034137725830078e-05,
      "step": 42301,
      "training_step_time": 7.654759407043457
    },
    {
      "epoch": 4.034233093261719e-05,
      "model_forward_time": 1.7373006343841553,
      "step": 42302
    },
    {
      "epoch": 4.034233093261719e-05,
      "step": 42302,
      "training_step_time": 7.590193271636963
    },
    {
      "epoch": 4.034328460693359e-05,
      "model_forward_time": 1.730830192565918,
      "step": 42303
    },
    {
      "epoch": 4.034328460693359e-05,
      "step": 42303,
      "training_step_time": 8.202085494995117
    },
    {
      "epoch": 4.034423828125e-05,
      "model_forward_time": 1.7293052673339844,
      "step": 42304
    },
    {
      "epoch": 4.034423828125e-05,
      "step": 42304,
      "training_step_time": 7.730224847793579
    },
    {
      "epoch": 4.0345191955566406e-05,
      "model_forward_time": 1.7425782680511475,
      "step": 42305
    },
    {
      "epoch": 4.0345191955566406e-05,
      "step": 42305,
      "training_step_time": 7.787405252456665
    },
    {
      "epoch": 4.034614562988281e-05,
      "model_forward_time": 1.8468518257141113,
      "step": 42306
    },
    {
      "epoch": 4.034614562988281e-05,
      "step": 42306,
      "training_step_time": 7.576590538024902
    },
    {
      "epoch": 4.034709930419922e-05,
      "model_forward_time": 1.7396063804626465,
      "step": 42307
    },
    {
      "epoch": 4.034709930419922e-05,
      "step": 42307,
      "training_step_time": 7.4534947872161865
    },
    {
      "epoch": 4.0348052978515625e-05,
      "model_forward_time": 1.7272651195526123,
      "step": 42308
    },
    {
      "epoch": 4.0348052978515625e-05,
      "step": 42308,
      "training_step_time": 8.176748991012573
    },
    {
      "epoch": 4.034900665283203e-05,
      "model_forward_time": 1.7434074878692627,
      "step": 42309
    },
    {
      "epoch": 4.034900665283203e-05,
      "step": 42309,
      "training_step_time": 7.684921026229858
    },
    {
      "epoch": 4.034996032714844e-05,
      "grad_norm": 0.11722251772880554,
      "learning_rate": 6.65333487575467e-06,
      "loss": 0.0606,
      "step": 42310
    },
    {
      "epoch": 4.034996032714844e-05,
      "model_forward_time": 2.8722739219665527,
      "step": 42310
    },
    {
      "dynamics_loss_avg": 0.06361878793686629,
      "epoch": 4.034996032714844e-05,
      "step": 42310
    },
    {
      "action_loss_avg": 0.0022727286646841093,
      "epoch": 4.034996032714844e-05,
      "step": 42310
    },
    {
      "epoch": 4.034996032714844e-05,
      "step": 42310,
      "training_step_time": 8.28000283241272
    },
    {
      "epoch": 4.0350914001464844e-05,
      "model_forward_time": 1.748436450958252,
      "step": 42311
    },
    {
      "epoch": 4.0350914001464844e-05,
      "step": 42311,
      "training_step_time": 7.533920526504517
    },
    {
      "epoch": 4.035186767578125e-05,
      "model_forward_time": 1.740307092666626,
      "step": 42312
    },
    {
      "epoch": 4.035186767578125e-05,
      "step": 42312,
      "training_step_time": 8.249058246612549
    },
    {
      "epoch": 4.035282135009766e-05,
      "model_forward_time": 1.7345681190490723,
      "step": 42313
    },
    {
      "epoch": 4.035282135009766e-05,
      "step": 42313,
      "training_step_time": 9.36798357963562
    },
    {
      "epoch": 4.035377502441406e-05,
      "model_forward_time": 1.736570119857788,
      "step": 42314
    },
    {
      "epoch": 4.035377502441406e-05,
      "step": 42314,
      "training_step_time": 9.283375263214111
    },
    {
      "epoch": 4.0354728698730467e-05,
      "model_forward_time": 1.847254991531372,
      "step": 42315
    },
    {
      "epoch": 4.0354728698730467e-05,
      "step": 42315,
      "training_step_time": 7.723365545272827
    },
    {
      "epoch": 4.035568237304688e-05,
      "model_forward_time": 1.7666473388671875,
      "step": 42316
    },
    {
      "epoch": 4.035568237304688e-05,
      "step": 42316,
      "training_step_time": 8.5837082862854
    },
    {
      "epoch": 4.035663604736328e-05,
      "model_forward_time": 1.7488150596618652,
      "step": 42317
    },
    {
      "epoch": 4.035663604736328e-05,
      "step": 42317,
      "training_step_time": 7.6895716190338135
    },
    {
      "epoch": 4.0357589721679685e-05,
      "model_forward_time": 1.7373597621917725,
      "step": 42318
    },
    {
      "epoch": 4.0357589721679685e-05,
      "step": 42318,
      "training_step_time": 7.742958307266235
    },
    {
      "epoch": 4.0358543395996096e-05,
      "model_forward_time": 2.904794692993164,
      "step": 42319
    },
    {
      "epoch": 4.0358543395996096e-05,
      "step": 42319,
      "training_step_time": 8.555193662643433
    },
    {
      "epoch": 4.03594970703125e-05,
      "grad_norm": 0.11095578223466873,
      "learning_rate": 6.6517743273664215e-06,
      "loss": 0.0573,
      "step": 42320
    },
    {
      "epoch": 4.03594970703125e-05,
      "model_forward_time": 1.7258353233337402,
      "step": 42320
    },
    {
      "dynamics_loss_avg": 0.06435630489140749,
      "epoch": 4.03594970703125e-05,
      "step": 42320
    },
    {
      "action_loss_avg": 0.0030324121296871455,
      "epoch": 4.03594970703125e-05,
      "step": 42320
    },
    {
      "epoch": 4.03594970703125e-05,
      "step": 42320,
      "training_step_time": 8.401509284973145
    },
    {
      "epoch": 4.0360450744628904e-05,
      "model_forward_time": 1.753568172454834,
      "step": 42321
    },
    {
      "epoch": 4.0360450744628904e-05,
      "step": 42321,
      "training_step_time": 7.610721826553345
    },
    {
      "epoch": 4.0361404418945315e-05,
      "model_forward_time": 1.7302196025848389,
      "step": 42322
    },
    {
      "epoch": 4.0361404418945315e-05,
      "step": 42322,
      "training_step_time": 9.1904878616333
    },
    {
      "epoch": 4.036235809326172e-05,
      "model_forward_time": 1.740311861038208,
      "step": 42323
    },
    {
      "epoch": 4.036235809326172e-05,
      "step": 42323,
      "training_step_time": 8.702593088150024
    },
    {
      "epoch": 4.036331176757812e-05,
      "model_forward_time": 1.810107707977295,
      "step": 42324
    },
    {
      "epoch": 4.036331176757812e-05,
      "step": 42324,
      "training_step_time": 7.76899266242981
    },
    {
      "epoch": 4.0364265441894534e-05,
      "model_forward_time": 1.756545066833496,
      "step": 42325
    },
    {
      "epoch": 4.0364265441894534e-05,
      "step": 42325,
      "training_step_time": 7.564351797103882
    },
    {
      "epoch": 4.036521911621094e-05,
      "model_forward_time": 1.81886625289917,
      "step": 42326
    },
    {
      "epoch": 4.036521911621094e-05,
      "step": 42326,
      "training_step_time": 7.652585744857788
    },
    {
      "epoch": 4.036617279052734e-05,
      "model_forward_time": 1.7489533424377441,
      "step": 42327
    },
    {
      "epoch": 4.036617279052734e-05,
      "step": 42327,
      "training_step_time": 7.665081024169922
    },
    {
      "epoch": 4.036712646484375e-05,
      "model_forward_time": 2.8835134506225586,
      "step": 42328
    },
    {
      "epoch": 4.036712646484375e-05,
      "step": 42328,
      "training_step_time": 7.742035627365112
    },
    {
      "epoch": 4.036808013916016e-05,
      "model_forward_time": 1.7498698234558105,
      "step": 42329
    },
    {
      "epoch": 4.036808013916016e-05,
      "step": 42329,
      "training_step_time": 7.647204637527466
    },
    {
      "epoch": 4.036903381347656e-05,
      "grad_norm": 0.12417962402105331,
      "learning_rate": 6.6502135983426145e-06,
      "loss": 0.0607,
      "step": 42330
    },
    {
      "epoch": 4.036903381347656e-05,
      "model_forward_time": 1.7354860305786133,
      "step": 42330
    },
    {
      "dynamics_loss_avg": 0.08035751730203629,
      "epoch": 4.036903381347656e-05,
      "step": 42330
    },
    {
      "action_loss_avg": 0.002362561470363289,
      "epoch": 4.036903381347656e-05,
      "step": 42330
    },
    {
      "epoch": 4.036903381347656e-05,
      "step": 42330,
      "training_step_time": 8.25235390663147
    },
    {
      "epoch": 4.036998748779297e-05,
      "model_forward_time": 1.75238037109375,
      "step": 42331
    },
    {
      "epoch": 4.036998748779297e-05,
      "step": 42331,
      "training_step_time": 8.600045204162598
    },
    {
      "epoch": 4.0370941162109376e-05,
      "model_forward_time": 1.7347514629364014,
      "step": 42332
    },
    {
      "epoch": 4.0370941162109376e-05,
      "step": 42332,
      "training_step_time": 9.34200406074524
    },
    {
      "epoch": 4.037189483642578e-05,
      "model_forward_time": 1.7800629138946533,
      "step": 42333
    },
    {
      "epoch": 4.037189483642578e-05,
      "step": 42333,
      "training_step_time": 9.547807455062866
    },
    {
      "epoch": 4.037284851074219e-05,
      "model_forward_time": 1.3725340366363525,
      "step": 42334
    },
    {
      "epoch": 4.037284851074219e-05,
      "step": 42334,
      "training_step_time": 9.142630338668823
    },
    {
      "epoch": 4.0373802185058595e-05,
      "model_forward_time": 1.751704454421997,
      "step": 42335
    },
    {
      "epoch": 4.0373802185058595e-05,
      "step": 42335,
      "training_step_time": 7.692719221115112
    },
    {
      "epoch": 4.0374755859375e-05,
      "model_forward_time": 1.7456564903259277,
      "step": 42336
    },
    {
      "epoch": 4.0374755859375e-05,
      "step": 42336,
      "training_step_time": 7.664856433868408
    },
    {
      "epoch": 4.037570953369141e-05,
      "model_forward_time": 2.8703198432922363,
      "step": 42337
    },
    {
      "epoch": 4.037570953369141e-05,
      "step": 42337,
      "training_step_time": 7.711181640625
    },
    {
      "epoch": 4.0376663208007813e-05,
      "model_forward_time": 1.7591640949249268,
      "step": 42338
    },
    {
      "epoch": 4.0376663208007813e-05,
      "step": 42338,
      "training_step_time": 8.651985883712769
    },
    {
      "epoch": 4.037761688232422e-05,
      "model_forward_time": 1.7552947998046875,
      "step": 42339
    },
    {
      "epoch": 4.037761688232422e-05,
      "step": 42339,
      "training_step_time": 7.57101035118103
    },
    {
      "epoch": 4.037857055664063e-05,
      "grad_norm": 0.1468948870897293,
      "learning_rate": 6.64865268885393e-06,
      "loss": 0.0583,
      "step": 42340
    },
    {
      "epoch": 4.037857055664063e-05,
      "model_forward_time": 1.7228145599365234,
      "step": 42340
    },
    {
      "dynamics_loss_avg": 0.04785608537495136,
      "epoch": 4.037857055664063e-05,
      "step": 42340
    },
    {
      "action_loss_avg": 0.002700057926995214,
      "epoch": 4.037857055664063e-05,
      "step": 42340
    },
    {
      "epoch": 4.037857055664063e-05,
      "step": 42340,
      "training_step_time": 7.625520706176758
    },
    {
      "epoch": 4.037952423095703e-05,
      "model_forward_time": 1.7752485275268555,
      "step": 42341
    },
    {
      "epoch": 4.037952423095703e-05,
      "step": 42341,
      "training_step_time": 7.698462724685669
    },
    {
      "epoch": 4.0380477905273436e-05,
      "model_forward_time": 1.7383930683135986,
      "step": 42342
    },
    {
      "epoch": 4.0380477905273436e-05,
      "step": 42342,
      "training_step_time": 7.718623161315918
    },
    {
      "epoch": 4.038143157958985e-05,
      "model_forward_time": 1.7359259128570557,
      "step": 42343
    },
    {
      "epoch": 4.038143157958985e-05,
      "step": 42343,
      "training_step_time": 7.554027795791626
    },
    {
      "epoch": 4.038238525390625e-05,
      "model_forward_time": 1.747318983078003,
      "step": 42344
    },
    {
      "epoch": 4.038238525390625e-05,
      "step": 42344,
      "training_step_time": 7.807803630828857
    },
    {
      "epoch": 4.0383338928222655e-05,
      "model_forward_time": 1.7397916316986084,
      "step": 42345
    },
    {
      "epoch": 4.0383338928222655e-05,
      "step": 42345,
      "training_step_time": 7.735664129257202
    },
    {
      "epoch": 4.038429260253906e-05,
      "model_forward_time": 2.9151928424835205,
      "step": 42346
    },
    {
      "epoch": 4.038429260253906e-05,
      "step": 42346,
      "training_step_time": 7.760005950927734
    },
    {
      "epoch": 4.038524627685547e-05,
      "model_forward_time": 1.7465322017669678,
      "step": 42347
    },
    {
      "epoch": 4.038524627685547e-05,
      "step": 42347,
      "training_step_time": 8.365188837051392
    },
    {
      "epoch": 4.0386199951171874e-05,
      "model_forward_time": 1.73073148727417,
      "step": 42348
    },
    {
      "epoch": 4.0386199951171874e-05,
      "step": 42348,
      "training_step_time": 7.582880735397339
    },
    {
      "epoch": 4.038715362548828e-05,
      "model_forward_time": 1.7428221702575684,
      "step": 42349
    },
    {
      "epoch": 4.038715362548828e-05,
      "step": 42349,
      "training_step_time": 7.577218532562256
    },
    {
      "epoch": 4.038810729980469e-05,
      "grad_norm": 0.19052645564079285,
      "learning_rate": 6.647091599071066e-06,
      "loss": 0.0553,
      "step": 42350
    },
    {
      "epoch": 4.038810729980469e-05,
      "model_forward_time": 1.737424612045288,
      "step": 42350
    },
    {
      "dynamics_loss_avg": 0.04056761059910059,
      "epoch": 4.038810729980469e-05,
      "step": 42350
    },
    {
      "action_loss_avg": 0.001510787132428959,
      "epoch": 4.038810729980469e-05,
      "step": 42350
    },
    {
      "epoch": 4.038810729980469e-05,
      "step": 42350,
      "training_step_time": 7.735141277313232
    },
    {
      "epoch": 4.038906097412109e-05,
      "model_forward_time": 1.716231346130371,
      "step": 42351
    },
    {
      "epoch": 4.038906097412109e-05,
      "step": 42351,
      "training_step_time": 8.250683784484863
    },
    {
      "epoch": 4.03900146484375e-05,
      "model_forward_time": 1.835744857788086,
      "step": 42352
    },
    {
      "epoch": 4.03900146484375e-05,
      "step": 42352,
      "training_step_time": 8.201645135879517
    },
    {
      "epoch": 4.039096832275391e-05,
      "model_forward_time": 1.752572774887085,
      "step": 42353
    },
    {
      "epoch": 4.039096832275391e-05,
      "step": 42353,
      "training_step_time": 7.603342533111572
    },
    {
      "epoch": 4.039192199707031e-05,
      "model_forward_time": 1.3631775379180908,
      "step": 42354
    },
    {
      "epoch": 4.039192199707031e-05,
      "step": 42354,
      "training_step_time": 7.719717264175415
    },
    {
      "epoch": 4.0392875671386716e-05,
      "model_forward_time": 2.8415029048919678,
      "step": 42355
    },
    {
      "epoch": 4.0392875671386716e-05,
      "step": 42355,
      "training_step_time": 7.764365196228027
    },
    {
      "epoch": 4.0393829345703127e-05,
      "model_forward_time": 1.3898377418518066,
      "step": 42356
    },
    {
      "epoch": 4.0393829345703127e-05,
      "step": 42356,
      "training_step_time": 8.402338981628418
    },
    {
      "epoch": 4.039478302001953e-05,
      "model_forward_time": 1.7248659133911133,
      "step": 42357
    },
    {
      "epoch": 4.039478302001953e-05,
      "step": 42357,
      "training_step_time": 7.605807304382324
    },
    {
      "epoch": 4.0395736694335935e-05,
      "model_forward_time": 1.7291884422302246,
      "step": 42358
    },
    {
      "epoch": 4.0395736694335935e-05,
      "step": 42358,
      "training_step_time": 8.5849609375
    },
    {
      "epoch": 4.0396690368652345e-05,
      "model_forward_time": 1.7160241603851318,
      "step": 42359
    },
    {
      "epoch": 4.0396690368652345e-05,
      "step": 42359,
      "training_step_time": 8.200830936431885
    },
    {
      "epoch": 4.039764404296875e-05,
      "grad_norm": 0.14618425071239471,
      "learning_rate": 6.645530329164742e-06,
      "loss": 0.0562,
      "step": 42360
    },
    {
      "epoch": 4.039764404296875e-05,
      "model_forward_time": 1.7381923198699951,
      "step": 42360
    },
    {
      "dynamics_loss_avg": 0.05681963283568621,
      "epoch": 4.039764404296875e-05,
      "step": 42360
    },
    {
      "action_loss_avg": 0.0023214024142362176,
      "epoch": 4.039764404296875e-05,
      "step": 42360
    },
    {
      "epoch": 4.039764404296875e-05,
      "step": 42360,
      "training_step_time": 7.805832624435425
    },
    {
      "epoch": 4.0398597717285153e-05,
      "model_forward_time": 1.8504517078399658,
      "step": 42361
    },
    {
      "epoch": 4.0398597717285153e-05,
      "step": 42361,
      "training_step_time": 8.606153726577759
    },
    {
      "epoch": 4.0399551391601564e-05,
      "model_forward_time": 1.7299203872680664,
      "step": 42362
    },
    {
      "epoch": 4.0399551391601564e-05,
      "step": 42362,
      "training_step_time": 7.5689966678619385
    },
    {
      "epoch": 4.040050506591797e-05,
      "model_forward_time": 1.7341704368591309,
      "step": 42363
    },
    {
      "epoch": 4.040050506591797e-05,
      "step": 42363,
      "training_step_time": 8.682582139968872
    },
    {
      "epoch": 4.040145874023437e-05,
      "model_forward_time": 1.730358362197876,
      "step": 42364
    },
    {
      "epoch": 4.040145874023437e-05,
      "step": 42364,
      "training_step_time": 7.65660834312439
    },
    {
      "epoch": 4.040241241455078e-05,
      "model_forward_time": 1.7297508716583252,
      "step": 42365
    },
    {
      "epoch": 4.040241241455078e-05,
      "step": 42365,
      "training_step_time": 8.270460367202759
    },
    {
      "epoch": 4.040336608886719e-05,
      "model_forward_time": 1.7495088577270508,
      "step": 42366
    },
    {
      "epoch": 4.040336608886719e-05,
      "step": 42366,
      "training_step_time": 8.42514157295227
    },
    {
      "epoch": 4.040431976318359e-05,
      "model_forward_time": 1.7260465621948242,
      "step": 42367
    },
    {
      "epoch": 4.040431976318359e-05,
      "step": 42367,
      "training_step_time": 8.584309816360474
    },
    {
      "epoch": 4.04052734375e-05,
      "model_forward_time": 1.7253673076629639,
      "step": 42368
    },
    {
      "epoch": 4.04052734375e-05,
      "step": 42368,
      "training_step_time": 8.27380084991455
    },
    {
      "epoch": 4.0406227111816406e-05,
      "model_forward_time": 1.733647346496582,
      "step": 42369
    },
    {
      "epoch": 4.0406227111816406e-05,
      "step": 42369,
      "training_step_time": 7.770292282104492
    },
    {
      "epoch": 4.040718078613281e-05,
      "grad_norm": 0.12755317986011505,
      "learning_rate": 6.643968879305695e-06,
      "loss": 0.0597,
      "step": 42370
    },
    {
      "epoch": 4.040718078613281e-05,
      "model_forward_time": 1.807131052017212,
      "step": 42370
    },
    {
      "dynamics_loss_avg": 0.06684370562434197,
      "epoch": 4.040718078613281e-05,
      "step": 42370
    },
    {
      "action_loss_avg": 0.0030430396262090655,
      "epoch": 4.040718078613281e-05,
      "step": 42370
    },
    {
      "epoch": 4.040718078613281e-05,
      "step": 42370,
      "training_step_time": 7.750683069229126
    },
    {
      "epoch": 4.040813446044922e-05,
      "model_forward_time": 1.7409353256225586,
      "step": 42371
    },
    {
      "epoch": 4.040813446044922e-05,
      "step": 42371,
      "training_step_time": 7.484875917434692
    },
    {
      "epoch": 4.0409088134765625e-05,
      "model_forward_time": 1.7894353866577148,
      "step": 42372
    },
    {
      "epoch": 4.0409088134765625e-05,
      "step": 42372,
      "training_step_time": 7.637187480926514
    },
    {
      "epoch": 4.041004180908203e-05,
      "model_forward_time": 1.7431080341339111,
      "step": 42373
    },
    {
      "epoch": 4.041004180908203e-05,
      "step": 42373,
      "training_step_time": 7.685225963592529
    },
    {
      "epoch": 4.041099548339844e-05,
      "model_forward_time": 2.94071364402771,
      "step": 42374
    },
    {
      "epoch": 4.041099548339844e-05,
      "step": 42374,
      "training_step_time": 7.741817235946655
    },
    {
      "epoch": 4.0411949157714844e-05,
      "model_forward_time": 1.7259571552276611,
      "step": 42375
    },
    {
      "epoch": 4.0411949157714844e-05,
      "step": 42375,
      "training_step_time": 7.6716718673706055
    },
    {
      "epoch": 4.041290283203125e-05,
      "model_forward_time": 1.7285394668579102,
      "step": 42376
    },
    {
      "epoch": 4.041290283203125e-05,
      "step": 42376,
      "training_step_time": 7.5673134326934814
    },
    {
      "epoch": 4.041385650634766e-05,
      "model_forward_time": 1.7257435321807861,
      "step": 42377
    },
    {
      "epoch": 4.041385650634766e-05,
      "step": 42377,
      "training_step_time": 7.780328989028931
    },
    {
      "epoch": 4.041481018066406e-05,
      "model_forward_time": 1.7384190559387207,
      "step": 42378
    },
    {
      "epoch": 4.041481018066406e-05,
      "step": 42378,
      "training_step_time": 7.726468086242676
    },
    {
      "epoch": 4.041576385498047e-05,
      "model_forward_time": 1.7193384170532227,
      "step": 42379
    },
    {
      "epoch": 4.041576385498047e-05,
      "step": 42379,
      "training_step_time": 9.281787395477295
    },
    {
      "epoch": 4.041671752929688e-05,
      "grad_norm": 0.12852410972118378,
      "learning_rate": 6.642407249664683e-06,
      "loss": 0.0624,
      "step": 42380
    },
    {
      "epoch": 4.041671752929688e-05,
      "model_forward_time": 1.829702377319336,
      "step": 42380
    },
    {
      "dynamics_loss_avg": 0.05439197923988104,
      "epoch": 4.041671752929688e-05,
      "step": 42380
    },
    {
      "action_loss_avg": 0.0019096051633823662,
      "epoch": 4.041671752929688e-05,
      "step": 42380
    },
    {
      "epoch": 4.041671752929688e-05,
      "step": 42380,
      "training_step_time": 7.57296347618103
    },
    {
      "epoch": 4.041767120361328e-05,
      "model_forward_time": 1.7296159267425537,
      "step": 42381
    },
    {
      "epoch": 4.041767120361328e-05,
      "step": 42381,
      "training_step_time": 8.193546056747437
    },
    {
      "epoch": 4.0418624877929685e-05,
      "model_forward_time": 1.7255914211273193,
      "step": 42382
    },
    {
      "epoch": 4.0418624877929685e-05,
      "step": 42382,
      "training_step_time": 8.535271644592285
    },
    {
      "epoch": 4.0419578552246096e-05,
      "model_forward_time": 1.7257413864135742,
      "step": 42383
    },
    {
      "epoch": 4.0419578552246096e-05,
      "step": 42383,
      "training_step_time": 9.20286250114441
    },
    {
      "epoch": 4.04205322265625e-05,
      "model_forward_time": 1.7570075988769531,
      "step": 42384
    },
    {
      "epoch": 4.04205322265625e-05,
      "step": 42384,
      "training_step_time": 6.791155576705933
    },
    {
      "epoch": 4.0421485900878904e-05,
      "model_forward_time": 1.7667505741119385,
      "step": 42385
    },
    {
      "epoch": 4.0421485900878904e-05,
      "step": 42385,
      "training_step_time": 6.801959037780762
    },
    {
      "epoch": 4.0422439575195315e-05,
      "model_forward_time": 1.7294249534606934,
      "step": 42386
    },
    {
      "epoch": 4.0422439575195315e-05,
      "step": 42386,
      "training_step_time": 6.79129695892334
    },
    {
      "epoch": 4.042339324951172e-05,
      "model_forward_time": 1.734269142150879,
      "step": 42387
    },
    {
      "epoch": 4.042339324951172e-05,
      "step": 42387,
      "training_step_time": 6.764629364013672
    },
    {
      "epoch": 4.042434692382812e-05,
      "model_forward_time": 1.7200567722320557,
      "step": 42388
    },
    {
      "epoch": 4.042434692382812e-05,
      "step": 42388,
      "training_step_time": 6.7706780433654785
    },
    {
      "epoch": 4.0425300598144534e-05,
      "model_forward_time": 1.7304356098175049,
      "step": 42389
    },
    {
      "epoch": 4.0425300598144534e-05,
      "step": 42389,
      "training_step_time": 6.783785104751587
    },
    {
      "epoch": 4.042625427246094e-05,
      "grad_norm": 0.18174391984939575,
      "learning_rate": 6.6408454404124835e-06,
      "loss": 0.0591,
      "step": 42390
    },
    {
      "epoch": 4.042625427246094e-05,
      "model_forward_time": 1.724029541015625,
      "step": 42390
    },
    {
      "dynamics_loss_avg": 0.05034144632518291,
      "epoch": 4.042625427246094e-05,
      "step": 42390
    },
    {
      "action_loss_avg": 0.0029061333101708443,
      "epoch": 4.042625427246094e-05,
      "step": 42390
    },
    {
      "epoch": 4.042625427246094e-05,
      "step": 42390,
      "training_step_time": 6.820510625839233
    },
    {
      "epoch": 4.042720794677734e-05,
      "model_forward_time": 1.7327940464019775,
      "step": 42391
    },
    {
      "epoch": 4.042720794677734e-05,
      "step": 42391,
      "training_step_time": 6.845997095108032
    },
    {
      "epoch": 4.042816162109375e-05,
      "model_forward_time": 2.8094773292541504,
      "step": 42392
    },
    {
      "epoch": 4.042816162109375e-05,
      "step": 42392,
      "training_step_time": 7.594800233840942
    },
    {
      "epoch": 4.042911529541016e-05,
      "model_forward_time": 1.7317073345184326,
      "step": 42393
    },
    {
      "epoch": 4.042911529541016e-05,
      "step": 42393,
      "training_step_time": 7.930793046951294
    },
    {
      "epoch": 4.043006896972656e-05,
      "model_forward_time": 1.7156314849853516,
      "step": 42394
    },
    {
      "epoch": 4.043006896972656e-05,
      "step": 42394,
      "training_step_time": 6.715566396713257
    },
    {
      "epoch": 4.043102264404297e-05,
      "model_forward_time": 1.7368683815002441,
      "step": 42395
    },
    {
      "epoch": 4.043102264404297e-05,
      "step": 42395,
      "training_step_time": 6.788030385971069
    },
    {
      "epoch": 4.0431976318359376e-05,
      "model_forward_time": 1.7470641136169434,
      "step": 42396
    },
    {
      "epoch": 4.0431976318359376e-05,
      "step": 42396,
      "training_step_time": 6.794602155685425
    },
    {
      "epoch": 4.043292999267578e-05,
      "model_forward_time": 1.734541654586792,
      "step": 42397
    },
    {
      "epoch": 4.043292999267578e-05,
      "step": 42397,
      "training_step_time": 6.754645347595215
    },
    {
      "epoch": 4.043388366699219e-05,
      "model_forward_time": 1.7529222965240479,
      "step": 42398
    },
    {
      "epoch": 4.043388366699219e-05,
      "step": 42398,
      "training_step_time": 6.764967679977417
    },
    {
      "epoch": 4.0434837341308595e-05,
      "model_forward_time": 1.7535960674285889,
      "step": 42399
    },
    {
      "epoch": 4.0434837341308595e-05,
      "step": 42399,
      "training_step_time": 6.764353275299072
    },
    {
      "epoch": 4.0435791015625e-05,
      "grad_norm": 0.14481522142887115,
      "learning_rate": 6.639283451719894e-06,
      "loss": 0.0616,
      "step": 42400
    },
    {
      "epoch": 4.0435791015625e-05,
      "model_forward_time": 1.7250864505767822,
      "step": 42400
    },
    {
      "dynamics_loss_avg": 0.06000853441655636,
      "epoch": 4.0435791015625e-05,
      "step": 42400
    },
    {
      "action_loss_avg": 0.0028406801633536817,
      "epoch": 4.0435791015625e-05,
      "step": 42400
    },
    {
      "epoch": 4.0435791015625e-05,
      "step": 42400,
      "training_step_time": 6.84067440032959
    },
    {
      "epoch": 4.043674468994141e-05,
      "model_forward_time": 2.8736369609832764,
      "step": 42401
    },
    {
      "epoch": 4.043674468994141e-05,
      "step": 42401,
      "training_step_time": 7.646334409713745
    },
    {
      "epoch": 4.0437698364257813e-05,
      "model_forward_time": 1.7372443675994873,
      "step": 42402
    },
    {
      "epoch": 4.0437698364257813e-05,
      "step": 42402,
      "training_step_time": 7.925260543823242
    },
    {
      "epoch": 4.043865203857422e-05,
      "model_forward_time": 1.7173981666564941,
      "step": 42403
    },
    {
      "epoch": 4.043865203857422e-05,
      "step": 42403,
      "training_step_time": 6.787008047103882
    },
    {
      "epoch": 4.043960571289063e-05,
      "model_forward_time": 1.720749855041504,
      "step": 42404
    },
    {
      "epoch": 4.043960571289063e-05,
      "step": 42404,
      "training_step_time": 6.778919219970703
    },
    {
      "epoch": 4.044055938720703e-05,
      "model_forward_time": 1.7292888164520264,
      "step": 42405
    },
    {
      "epoch": 4.044055938720703e-05,
      "step": 42405,
      "training_step_time": 6.777861595153809
    },
    {
      "epoch": 4.0441513061523436e-05,
      "model_forward_time": 1.7220144271850586,
      "step": 42406
    },
    {
      "epoch": 4.0441513061523436e-05,
      "step": 42406,
      "training_step_time": 6.8165552616119385
    },
    {
      "epoch": 4.044246673583984e-05,
      "model_forward_time": 1.7170436382293701,
      "step": 42407
    },
    {
      "epoch": 4.044246673583984e-05,
      "step": 42407,
      "training_step_time": 6.761080265045166
    },
    {
      "epoch": 4.044342041015625e-05,
      "model_forward_time": 1.7233717441558838,
      "step": 42408
    },
    {
      "epoch": 4.044342041015625e-05,
      "step": 42408,
      "training_step_time": 6.815978765487671
    },
    {
      "epoch": 4.0444374084472655e-05,
      "model_forward_time": 1.7743830680847168,
      "step": 42409
    },
    {
      "epoch": 4.0444374084472655e-05,
      "step": 42409,
      "training_step_time": 6.792469024658203
    },
    {
      "epoch": 4.044532775878906e-05,
      "grad_norm": 0.13021720945835114,
      "learning_rate": 6.63772128375773e-06,
      "loss": 0.0615,
      "step": 42410
    },
    {
      "epoch": 4.044532775878906e-05,
      "model_forward_time": 2.8792564868927,
      "step": 42410
    },
    {
      "dynamics_loss_avg": 0.05987446773797274,
      "epoch": 4.044532775878906e-05,
      "step": 42410
    },
    {
      "action_loss_avg": 0.0016082330606877805,
      "epoch": 4.044532775878906e-05,
      "step": 42410
    },
    {
      "epoch": 4.044532775878906e-05,
      "step": 42410,
      "training_step_time": 7.6799726486206055
    },
    {
      "epoch": 4.044628143310547e-05,
      "model_forward_time": 1.721482276916504,
      "step": 42411
    },
    {
      "epoch": 4.044628143310547e-05,
      "step": 42411,
      "training_step_time": 8.422720432281494
    },
    {
      "epoch": 4.0447235107421874e-05,
      "model_forward_time": 1.7277019023895264,
      "step": 42412
    },
    {
      "epoch": 4.0447235107421874e-05,
      "step": 42412,
      "training_step_time": 7.700082778930664
    },
    {
      "epoch": 4.044818878173828e-05,
      "model_forward_time": 1.7552690505981445,
      "step": 42413
    },
    {
      "epoch": 4.044818878173828e-05,
      "step": 42413,
      "training_step_time": 6.796407461166382
    },
    {
      "epoch": 4.044914245605469e-05,
      "model_forward_time": 1.7194538116455078,
      "step": 42414
    },
    {
      "epoch": 4.044914245605469e-05,
      "step": 42414,
      "training_step_time": 6.788778781890869
    },
    {
      "epoch": 4.045009613037109e-05,
      "model_forward_time": 1.7213528156280518,
      "step": 42415
    },
    {
      "epoch": 4.045009613037109e-05,
      "step": 42415,
      "training_step_time": 6.808568000793457
    },
    {
      "epoch": 4.04510498046875e-05,
      "model_forward_time": 1.724980354309082,
      "step": 42416
    },
    {
      "epoch": 4.04510498046875e-05,
      "step": 42416,
      "training_step_time": 6.830177307128906
    },
    {
      "epoch": 4.045200347900391e-05,
      "model_forward_time": 1.7280471324920654,
      "step": 42417
    },
    {
      "epoch": 4.045200347900391e-05,
      "step": 42417,
      "training_step_time": 6.799863338470459
    },
    {
      "epoch": 4.045295715332031e-05,
      "model_forward_time": 1.7330832481384277,
      "step": 42418
    },
    {
      "epoch": 4.045295715332031e-05,
      "step": 42418,
      "training_step_time": 6.785439491271973
    },
    {
      "epoch": 4.0453910827636716e-05,
      "model_forward_time": 2.877178907394409,
      "step": 42419
    },
    {
      "epoch": 4.0453910827636716e-05,
      "step": 42419,
      "training_step_time": 7.681473731994629
    },
    {
      "epoch": 4.045486450195313e-05,
      "grad_norm": 0.12926355004310608,
      "learning_rate": 6.636158936696828e-06,
      "loss": 0.0604,
      "step": 42420
    },
    {
      "epoch": 4.045486450195313e-05,
      "model_forward_time": 1.729987621307373,
      "step": 42420
    },
    {
      "dynamics_loss_avg": 0.06670853458344936,
      "epoch": 4.045486450195313e-05,
      "step": 42420
    },
    {
      "action_loss_avg": 0.003465682765818201,
      "epoch": 4.045486450195313e-05,
      "step": 42420
    },
    {
      "epoch": 4.045486450195313e-05,
      "step": 42420,
      "training_step_time": 9.406540632247925
    },
    {
      "epoch": 4.045581817626953e-05,
      "model_forward_time": 1.7192070484161377,
      "step": 42421
    },
    {
      "epoch": 4.045581817626953e-05,
      "step": 42421,
      "training_step_time": 7.7277326583862305
    },
    {
      "epoch": 4.0456771850585935e-05,
      "model_forward_time": 1.7327053546905518,
      "step": 42422
    },
    {
      "epoch": 4.0456771850585935e-05,
      "step": 42422,
      "training_step_time": 6.781500577926636
    },
    {
      "epoch": 4.0457725524902345e-05,
      "model_forward_time": 1.7233226299285889,
      "step": 42423
    },
    {
      "epoch": 4.0457725524902345e-05,
      "step": 42423,
      "training_step_time": 6.841036081314087
    },
    {
      "epoch": 4.045867919921875e-05,
      "model_forward_time": 1.7370686531066895,
      "step": 42424
    },
    {
      "epoch": 4.045867919921875e-05,
      "step": 42424,
      "training_step_time": 6.836700916290283
    },
    {
      "epoch": 4.0459632873535154e-05,
      "model_forward_time": 1.7231104373931885,
      "step": 42425
    },
    {
      "epoch": 4.0459632873535154e-05,
      "step": 42425,
      "training_step_time": 6.855584144592285
    },
    {
      "epoch": 4.0460586547851564e-05,
      "model_forward_time": 1.7271218299865723,
      "step": 42426
    },
    {
      "epoch": 4.0460586547851564e-05,
      "step": 42426,
      "training_step_time": 6.84920859336853
    },
    {
      "epoch": 4.046154022216797e-05,
      "model_forward_time": 1.7324583530426025,
      "step": 42427
    },
    {
      "epoch": 4.046154022216797e-05,
      "step": 42427,
      "training_step_time": 6.8178184032440186
    },
    {
      "epoch": 4.046249389648437e-05,
      "model_forward_time": 2.8738582134246826,
      "step": 42428
    },
    {
      "epoch": 4.046249389648437e-05,
      "step": 42428,
      "training_step_time": 7.686561346054077
    },
    {
      "epoch": 4.046344757080078e-05,
      "model_forward_time": 1.7348263263702393,
      "step": 42429
    },
    {
      "epoch": 4.046344757080078e-05,
      "step": 42429,
      "training_step_time": 8.336297035217285
    },
    {
      "epoch": 4.046440124511719e-05,
      "grad_norm": 0.11262691020965576,
      "learning_rate": 6.634596410708047e-06,
      "loss": 0.058,
      "step": 42430
    },
    {
      "epoch": 4.046440124511719e-05,
      "model_forward_time": 1.7180206775665283,
      "step": 42430
    },
    {
      "dynamics_loss_avg": 0.05063141509890556,
      "epoch": 4.046440124511719e-05,
      "step": 42430
    },
    {
      "action_loss_avg": 0.00171769775042776,
      "epoch": 4.046440124511719e-05,
      "step": 42430
    },
    {
      "epoch": 4.046440124511719e-05,
      "step": 42430,
      "training_step_time": 8.277389287948608
    },
    {
      "epoch": 4.046535491943359e-05,
      "model_forward_time": 1.7138426303863525,
      "step": 42431
    },
    {
      "epoch": 4.046535491943359e-05,
      "step": 42431,
      "training_step_time": 6.745849132537842
    },
    {
      "epoch": 4.046630859375e-05,
      "model_forward_time": 1.739330530166626,
      "step": 42432
    },
    {
      "epoch": 4.046630859375e-05,
      "step": 42432,
      "training_step_time": 6.767605543136597
    },
    {
      "epoch": 4.0467262268066406e-05,
      "model_forward_time": 1.7143640518188477,
      "step": 42433
    },
    {
      "epoch": 4.0467262268066406e-05,
      "step": 42433,
      "training_step_time": 6.755229473114014
    },
    {
      "epoch": 4.046821594238281e-05,
      "model_forward_time": 1.7344975471496582,
      "step": 42434
    },
    {
      "epoch": 4.046821594238281e-05,
      "step": 42434,
      "training_step_time": 6.760258197784424
    },
    {
      "epoch": 4.046916961669922e-05,
      "model_forward_time": 1.7250056266784668,
      "step": 42435
    },
    {
      "epoch": 4.046916961669922e-05,
      "step": 42435,
      "training_step_time": 6.741393089294434
    },
    {
      "epoch": 4.0470123291015625e-05,
      "model_forward_time": 1.7232563495635986,
      "step": 42436
    },
    {
      "epoch": 4.0470123291015625e-05,
      "step": 42436,
      "training_step_time": 6.772697448730469
    },
    {
      "epoch": 4.047107696533203e-05,
      "model_forward_time": 2.7990119457244873,
      "step": 42437
    },
    {
      "epoch": 4.047107696533203e-05,
      "step": 42437,
      "training_step_time": 7.642026901245117
    },
    {
      "epoch": 4.047203063964844e-05,
      "model_forward_time": 1.8420791625976562,
      "step": 42438
    },
    {
      "epoch": 4.047203063964844e-05,
      "step": 42438,
      "training_step_time": 7.606319427490234
    },
    {
      "epoch": 4.0472984313964844e-05,
      "model_forward_time": 1.724644422531128,
      "step": 42439
    },
    {
      "epoch": 4.0472984313964844e-05,
      "step": 42439,
      "training_step_time": 7.832201719284058
    },
    {
      "epoch": 4.047393798828125e-05,
      "grad_norm": 0.1409427672624588,
      "learning_rate": 6.633033705962257e-06,
      "loss": 0.0583,
      "step": 42440
    },
    {
      "epoch": 4.047393798828125e-05,
      "model_forward_time": 1.7304863929748535,
      "step": 42440
    },
    {
      "dynamics_loss_avg": 0.06270577628165483,
      "epoch": 4.047393798828125e-05,
      "step": 42440
    },
    {
      "action_loss_avg": 0.002707764841034077,
      "epoch": 4.047393798828125e-05,
      "step": 42440
    },
    {
      "epoch": 4.047393798828125e-05,
      "step": 42440,
      "training_step_time": 7.604162931442261
    },
    {
      "epoch": 4.047489166259766e-05,
      "model_forward_time": 1.7290701866149902,
      "step": 42441
    },
    {
      "epoch": 4.047489166259766e-05,
      "step": 42441,
      "training_step_time": 6.80493426322937
    },
    {
      "epoch": 4.047584533691406e-05,
      "model_forward_time": 1.7252085208892822,
      "step": 42442
    },
    {
      "epoch": 4.047584533691406e-05,
      "step": 42442,
      "training_step_time": 6.803555727005005
    },
    {
      "epoch": 4.047679901123047e-05,
      "model_forward_time": 1.7241899967193604,
      "step": 42443
    },
    {
      "epoch": 4.047679901123047e-05,
      "step": 42443,
      "training_step_time": 6.8153862953186035
    },
    {
      "epoch": 4.047775268554688e-05,
      "model_forward_time": 1.7421603202819824,
      "step": 42444
    },
    {
      "epoch": 4.047775268554688e-05,
      "step": 42444,
      "training_step_time": 6.812189102172852
    },
    {
      "epoch": 4.047870635986328e-05,
      "model_forward_time": 1.7537024021148682,
      "step": 42445
    },
    {
      "epoch": 4.047870635986328e-05,
      "step": 42445,
      "training_step_time": 6.786553144454956
    },
    {
      "epoch": 4.0479660034179686e-05,
      "model_forward_time": 1.7406301498413086,
      "step": 42446
    },
    {
      "epoch": 4.0479660034179686e-05,
      "step": 42446,
      "training_step_time": 7.632853269577026
    },
    {
      "epoch": 4.0480613708496096e-05,
      "model_forward_time": 2.8614304065704346,
      "step": 42447
    },
    {
      "epoch": 4.0480613708496096e-05,
      "step": 42447,
      "training_step_time": 7.607376337051392
    },
    {
      "epoch": 4.04815673828125e-05,
      "model_forward_time": 1.7366230487823486,
      "step": 42448
    },
    {
      "epoch": 4.04815673828125e-05,
      "step": 42448,
      "training_step_time": 8.710370779037476
    },
    {
      "epoch": 4.0482521057128904e-05,
      "model_forward_time": 1.7379469871520996,
      "step": 42449
    },
    {
      "epoch": 4.0482521057128904e-05,
      "step": 42449,
      "training_step_time": 7.755304574966431
    },
    {
      "epoch": 4.0483474731445315e-05,
      "grad_norm": 0.103478342294693,
      "learning_rate": 6.6314708226303596e-06,
      "loss": 0.0601,
      "step": 42450
    },
    {
      "epoch": 4.0483474731445315e-05,
      "model_forward_time": 1.7283170223236084,
      "step": 42450
    },
    {
      "dynamics_loss_avg": 0.08251983821392059,
      "epoch": 4.0483474731445315e-05,
      "step": 42450
    },
    {
      "action_loss_avg": 0.0029418781865388154,
      "epoch": 4.0483474731445315e-05,
      "step": 42450
    },
    {
      "epoch": 4.0483474731445315e-05,
      "step": 42450,
      "training_step_time": 6.837883234024048
    },
    {
      "epoch": 4.048442840576172e-05,
      "model_forward_time": 1.7431514263153076,
      "step": 42451
    },
    {
      "epoch": 4.048442840576172e-05,
      "step": 42451,
      "training_step_time": 6.872925281524658
    },
    {
      "epoch": 4.048538208007812e-05,
      "model_forward_time": 1.7417542934417725,
      "step": 42452
    },
    {
      "epoch": 4.048538208007812e-05,
      "step": 42452,
      "training_step_time": 6.8583784103393555
    },
    {
      "epoch": 4.0486335754394534e-05,
      "model_forward_time": 1.7276453971862793,
      "step": 42453
    },
    {
      "epoch": 4.0486335754394534e-05,
      "step": 42453,
      "training_step_time": 6.852434873580933
    },
    {
      "epoch": 4.048728942871094e-05,
      "model_forward_time": 1.733391523361206,
      "step": 42454
    },
    {
      "epoch": 4.048728942871094e-05,
      "step": 42454,
      "training_step_time": 6.840203046798706
    },
    {
      "epoch": 4.048824310302734e-05,
      "model_forward_time": 1.7290740013122559,
      "step": 42455
    },
    {
      "epoch": 4.048824310302734e-05,
      "step": 42455,
      "training_step_time": 6.8644139766693115
    },
    {
      "epoch": 4.048919677734375e-05,
      "model_forward_time": 2.869053602218628,
      "step": 42456
    },
    {
      "epoch": 4.048919677734375e-05,
      "step": 42456,
      "training_step_time": 7.71677303314209
    },
    {
      "epoch": 4.049015045166016e-05,
      "model_forward_time": 1.7290921211242676,
      "step": 42457
    },
    {
      "epoch": 4.049015045166016e-05,
      "step": 42457,
      "training_step_time": 11.518034219741821
    },
    {
      "epoch": 4.049110412597656e-05,
      "model_forward_time": 1.7391128540039062,
      "step": 42458
    },
    {
      "epoch": 4.049110412597656e-05,
      "step": 42458,
      "training_step_time": 7.797361135482788
    },
    {
      "epoch": 4.049205780029297e-05,
      "model_forward_time": 1.7222630977630615,
      "step": 42459
    },
    {
      "epoch": 4.049205780029297e-05,
      "step": 42459,
      "training_step_time": 7.628838062286377
    },
    {
      "epoch": 4.0493011474609376e-05,
      "grad_norm": 0.10807391256093979,
      "learning_rate": 6.629907760883264e-06,
      "loss": 0.0588,
      "step": 42460
    },
    {
      "epoch": 4.0493011474609376e-05,
      "model_forward_time": 1.7154617309570312,
      "step": 42460
    },
    {
      "dynamics_loss_avg": 0.05765469502657652,
      "epoch": 4.0493011474609376e-05,
      "step": 42460
    },
    {
      "action_loss_avg": 0.0035438151273410766,
      "epoch": 4.0493011474609376e-05,
      "step": 42460
    },
    {
      "epoch": 4.0493011474609376e-05,
      "step": 42460,
      "training_step_time": 6.852072477340698
    },
    {
      "epoch": 4.049396514892578e-05,
      "model_forward_time": 1.7408087253570557,
      "step": 42461
    },
    {
      "epoch": 4.049396514892578e-05,
      "step": 42461,
      "training_step_time": 6.89400315284729
    },
    {
      "epoch": 4.049491882324219e-05,
      "model_forward_time": 1.7359704971313477,
      "step": 42462
    },
    {
      "epoch": 4.049491882324219e-05,
      "step": 42462,
      "training_step_time": 6.904585599899292
    },
    {
      "epoch": 4.0495872497558595e-05,
      "model_forward_time": 1.7538328170776367,
      "step": 42463
    },
    {
      "epoch": 4.0495872497558595e-05,
      "step": 42463,
      "training_step_time": 6.871243715286255
    },
    {
      "epoch": 4.0496826171875e-05,
      "model_forward_time": 1.7292370796203613,
      "step": 42464
    },
    {
      "epoch": 4.0496826171875e-05,
      "step": 42464,
      "training_step_time": 6.751485347747803
    },
    {
      "epoch": 4.049777984619141e-05,
      "model_forward_time": 2.8671987056732178,
      "step": 42465
    },
    {
      "epoch": 4.049777984619141e-05,
      "step": 42465,
      "training_step_time": 7.716994524002075
    },
    {
      "epoch": 4.0498733520507814e-05,
      "model_forward_time": 1.733170509338379,
      "step": 42466
    },
    {
      "epoch": 4.0498733520507814e-05,
      "step": 42466,
      "training_step_time": 11.784898281097412
    },
    {
      "epoch": 4.049968719482422e-05,
      "model_forward_time": 1.7272741794586182,
      "step": 42467
    },
    {
      "epoch": 4.049968719482422e-05,
      "step": 42467,
      "training_step_time": 8.828455448150635
    },
    {
      "epoch": 4.050064086914063e-05,
      "model_forward_time": 1.7073450088500977,
      "step": 42468
    },
    {
      "epoch": 4.050064086914063e-05,
      "step": 42468,
      "training_step_time": 7.5617876052856445
    },
    {
      "epoch": 4.050159454345703e-05,
      "model_forward_time": 1.7176532745361328,
      "step": 42469
    },
    {
      "epoch": 4.050159454345703e-05,
      "step": 42469,
      "training_step_time": 6.929450035095215
    },
    {
      "epoch": 4.0502548217773436e-05,
      "grad_norm": 0.1628045290708542,
      "learning_rate": 6.628344520891905e-06,
      "loss": 0.0591,
      "step": 42470
    },
    {
      "epoch": 4.0502548217773436e-05,
      "model_forward_time": 1.725196123123169,
      "step": 42470
    },
    {
      "dynamics_loss_avg": 0.06583398897200823,
      "epoch": 4.0502548217773436e-05,
      "step": 42470
    },
    {
      "action_loss_avg": 0.0023549616918899117,
      "epoch": 4.0502548217773436e-05,
      "step": 42470
    },
    {
      "epoch": 4.0502548217773436e-05,
      "step": 42470,
      "training_step_time": 6.912787199020386
    },
    {
      "epoch": 4.050350189208984e-05,
      "model_forward_time": 1.749453067779541,
      "step": 42471
    },
    {
      "epoch": 4.050350189208984e-05,
      "step": 42471,
      "training_step_time": 6.891430854797363
    },
    {
      "epoch": 4.050445556640625e-05,
      "model_forward_time": 1.7306334972381592,
      "step": 42472
    },
    {
      "epoch": 4.050445556640625e-05,
      "step": 42472,
      "training_step_time": 6.880771160125732
    },
    {
      "epoch": 4.0505409240722655e-05,
      "model_forward_time": 1.7165963649749756,
      "step": 42473
    },
    {
      "epoch": 4.0505409240722655e-05,
      "step": 42473,
      "training_step_time": 6.858733177185059
    },
    {
      "epoch": 4.050636291503906e-05,
      "model_forward_time": 2.9032249450683594,
      "step": 42474
    },
    {
      "epoch": 4.050636291503906e-05,
      "step": 42474,
      "training_step_time": 7.743459701538086
    },
    {
      "epoch": 4.050731658935547e-05,
      "model_forward_time": 1.7307755947113037,
      "step": 42475
    },
    {
      "epoch": 4.050731658935547e-05,
      "step": 42475,
      "training_step_time": 10.90504503250122
    },
    {
      "epoch": 4.0508270263671874e-05,
      "model_forward_time": 1.7167901992797852,
      "step": 42476
    },
    {
      "epoch": 4.0508270263671874e-05,
      "step": 42476,
      "training_step_time": 8.892303466796875
    },
    {
      "epoch": 4.050922393798828e-05,
      "model_forward_time": 1.7153112888336182,
      "step": 42477
    },
    {
      "epoch": 4.050922393798828e-05,
      "step": 42477,
      "training_step_time": 8.29167628288269
    },
    {
      "epoch": 4.051017761230469e-05,
      "model_forward_time": 1.7272179126739502,
      "step": 42478
    },
    {
      "epoch": 4.051017761230469e-05,
      "step": 42478,
      "training_step_time": 6.89143705368042
    },
    {
      "epoch": 4.051113128662109e-05,
      "model_forward_time": 1.7428030967712402,
      "step": 42479
    },
    {
      "epoch": 4.051113128662109e-05,
      "step": 42479,
      "training_step_time": 6.897507429122925
    },
    {
      "epoch": 4.05120849609375e-05,
      "grad_norm": 0.17045673727989197,
      "learning_rate": 6.6267811028272396e-06,
      "loss": 0.0598,
      "step": 42480
    },
    {
      "epoch": 4.05120849609375e-05,
      "model_forward_time": 1.7310516834259033,
      "step": 42480
    },
    {
      "dynamics_loss_avg": 0.05042970273643732,
      "epoch": 4.05120849609375e-05,
      "step": 42480
    },
    {
      "action_loss_avg": 0.001978990429779515,
      "epoch": 4.05120849609375e-05,
      "step": 42480
    },
    {
      "epoch": 4.05120849609375e-05,
      "step": 42480,
      "training_step_time": 6.885099172592163
    },
    {
      "epoch": 4.051303863525391e-05,
      "model_forward_time": 1.7229559421539307,
      "step": 42481
    },
    {
      "epoch": 4.051303863525391e-05,
      "step": 42481,
      "training_step_time": 6.8845930099487305
    },
    {
      "epoch": 4.051399230957031e-05,
      "model_forward_time": 1.7452678680419922,
      "step": 42482
    },
    {
      "epoch": 4.051399230957031e-05,
      "step": 42482,
      "training_step_time": 6.851871967315674
    },
    {
      "epoch": 4.0514945983886716e-05,
      "model_forward_time": 2.840965986251831,
      "step": 42483
    },
    {
      "epoch": 4.0514945983886716e-05,
      "step": 42483,
      "training_step_time": 7.683523654937744
    },
    {
      "epoch": 4.051589965820313e-05,
      "model_forward_time": 1.8467330932617188,
      "step": 42484
    },
    {
      "epoch": 4.051589965820313e-05,
      "step": 42484,
      "training_step_time": 8.687542200088501
    },
    {
      "epoch": 4.051685333251953e-05,
      "model_forward_time": 1.7403032779693604,
      "step": 42485
    },
    {
      "epoch": 4.051685333251953e-05,
      "step": 42485,
      "training_step_time": 9.174629211425781
    },
    {
      "epoch": 4.0517807006835935e-05,
      "model_forward_time": 1.7349998950958252,
      "step": 42486
    },
    {
      "epoch": 4.0517807006835935e-05,
      "step": 42486,
      "training_step_time": 7.7805421352386475
    },
    {
      "epoch": 4.0518760681152346e-05,
      "model_forward_time": 1.711670160293579,
      "step": 42487
    },
    {
      "epoch": 4.0518760681152346e-05,
      "step": 42487,
      "training_step_time": 7.6231443881988525
    },
    {
      "epoch": 4.051971435546875e-05,
      "model_forward_time": 1.7287685871124268,
      "step": 42488
    },
    {
      "epoch": 4.051971435546875e-05,
      "step": 42488,
      "training_step_time": 6.8866353034973145
    },
    {
      "epoch": 4.0520668029785154e-05,
      "model_forward_time": 1.7202332019805908,
      "step": 42489
    },
    {
      "epoch": 4.0520668029785154e-05,
      "step": 42489,
      "training_step_time": 6.891229629516602
    },
    {
      "epoch": 4.0521621704101564e-05,
      "grad_norm": 0.19614873826503754,
      "learning_rate": 6.625217506860238e-06,
      "loss": 0.0573,
      "step": 42490
    },
    {
      "epoch": 4.0521621704101564e-05,
      "model_forward_time": 1.7201249599456787,
      "step": 42490
    },
    {
      "dynamics_loss_avg": 0.06612700521945954,
      "epoch": 4.0521621704101564e-05,
      "step": 42490
    },
    {
      "action_loss_avg": 0.002145048134843819,
      "epoch": 4.0521621704101564e-05,
      "step": 42490
    },
    {
      "epoch": 4.0521621704101564e-05,
      "step": 42490,
      "training_step_time": 6.874913692474365
    },
    {
      "epoch": 4.052257537841797e-05,
      "model_forward_time": 1.716371774673462,
      "step": 42491
    },
    {
      "epoch": 4.052257537841797e-05,
      "step": 42491,
      "training_step_time": 6.887239217758179
    },
    {
      "epoch": 4.052352905273437e-05,
      "model_forward_time": 1.7278835773468018,
      "step": 42492
    },
    {
      "epoch": 4.052352905273437e-05,
      "step": 42492,
      "training_step_time": 7.762979745864868
    },
    {
      "epoch": 4.052448272705078e-05,
      "model_forward_time": 2.872643232345581,
      "step": 42493
    },
    {
      "epoch": 4.052448272705078e-05,
      "step": 42493,
      "training_step_time": 8.66264271736145
    },
    {
      "epoch": 4.052543640136719e-05,
      "model_forward_time": 1.7358343601226807,
      "step": 42494
    },
    {
      "epoch": 4.052543640136719e-05,
      "step": 42494,
      "training_step_time": 11.151490449905396
    },
    {
      "epoch": 4.052639007568359e-05,
      "model_forward_time": 1.7172050476074219,
      "step": 42495
    },
    {
      "epoch": 4.052639007568359e-05,
      "step": 42495,
      "training_step_time": 7.894233703613281
    },
    {
      "epoch": 4.052734375e-05,
      "model_forward_time": 1.7128915786743164,
      "step": 42496
    },
    {
      "epoch": 4.052734375e-05,
      "step": 42496,
      "training_step_time": 7.665621519088745
    },
    {
      "epoch": 4.0528297424316406e-05,
      "model_forward_time": 1.7097387313842773,
      "step": 42497
    },
    {
      "epoch": 4.0528297424316406e-05,
      "step": 42497,
      "training_step_time": 6.855849742889404
    },
    {
      "epoch": 4.052925109863281e-05,
      "model_forward_time": 1.7212414741516113,
      "step": 42498
    },
    {
      "epoch": 4.052925109863281e-05,
      "step": 42498,
      "training_step_time": 6.840568542480469
    },
    {
      "epoch": 4.053020477294922e-05,
      "model_forward_time": 1.7284924983978271,
      "step": 42499
    },
    {
      "epoch": 4.053020477294922e-05,
      "step": 42499,
      "training_step_time": 6.858960866928101
    },
    {
      "epoch": 4.0531158447265625e-05,
      "grad_norm": 0.19992463290691376,
      "learning_rate": 6.623653733161892e-06,
      "loss": 0.0591,
      "step": 42500
    },
    {
      "epoch": 4.0531158447265625e-05,
      "model_forward_time": 1.7198009490966797,
      "step": 42500
    },
    {
      "dynamics_loss_avg": 0.05399573994800448,
      "epoch": 4.0531158447265625e-05,
      "step": 42500
    },
    {
      "action_loss_avg": 0.00272572657559067,
      "epoch": 4.0531158447265625e-05,
      "step": 42500
    },
    {
      "epoch": 4.0531158447265625e-05,
      "step": 42500,
      "training_step_time": 6.873764991760254
    },
    {
      "epoch": 4.053211212158203e-05,
      "model_forward_time": 1.7157940864562988,
      "step": 42501
    },
    {
      "epoch": 4.053211212158203e-05,
      "step": 42501,
      "training_step_time": 7.692898750305176
    },
    {
      "epoch": 4.053306579589844e-05,
      "model_forward_time": 2.932777166366577,
      "step": 42502
    },
    {
      "epoch": 4.053306579589844e-05,
      "step": 42502,
      "training_step_time": 7.740789413452148
    },
    {
      "epoch": 4.0534019470214844e-05,
      "model_forward_time": 1.7195756435394287,
      "step": 42503
    },
    {
      "epoch": 4.0534019470214844e-05,
      "step": 42503,
      "training_step_time": 10.85067868232727
    },
    {
      "epoch": 4.053497314453125e-05,
      "model_forward_time": 1.747781753540039,
      "step": 42504
    },
    {
      "epoch": 4.053497314453125e-05,
      "step": 42504,
      "training_step_time": 8.35784363746643
    },
    {
      "epoch": 4.053592681884766e-05,
      "model_forward_time": 1.7368080615997314,
      "step": 42505
    },
    {
      "epoch": 4.053592681884766e-05,
      "step": 42505,
      "training_step_time": 7.633877754211426
    },
    {
      "epoch": 4.053688049316406e-05,
      "model_forward_time": 1.723750114440918,
      "step": 42506
    },
    {
      "epoch": 4.053688049316406e-05,
      "step": 42506,
      "training_step_time": 6.872705459594727
    },
    {
      "epoch": 4.053783416748047e-05,
      "model_forward_time": 1.748528242111206,
      "step": 42507
    },
    {
      "epoch": 4.053783416748047e-05,
      "step": 42507,
      "training_step_time": 6.863531112670898
    },
    {
      "epoch": 4.053878784179688e-05,
      "model_forward_time": 1.7331557273864746,
      "step": 42508
    },
    {
      "epoch": 4.053878784179688e-05,
      "step": 42508,
      "training_step_time": 6.880536079406738
    },
    {
      "epoch": 4.053974151611328e-05,
      "model_forward_time": 1.7495551109313965,
      "step": 42509
    },
    {
      "epoch": 4.053974151611328e-05,
      "step": 42509,
      "training_step_time": 6.8304407596588135
    },
    {
      "epoch": 4.0540695190429686e-05,
      "grad_norm": 0.12297257035970688,
      "learning_rate": 6.622089781903217e-06,
      "loss": 0.0571,
      "step": 42510
    },
    {
      "epoch": 4.0540695190429686e-05,
      "model_forward_time": 1.7361500263214111,
      "step": 42510
    },
    {
      "dynamics_loss_avg": 0.04709078166633844,
      "epoch": 4.0540695190429686e-05,
      "step": 42510
    },
    {
      "action_loss_avg": 0.0014116871810983867,
      "epoch": 4.0540695190429686e-05,
      "step": 42510
    },
    {
      "epoch": 4.0540695190429686e-05,
      "step": 42510,
      "training_step_time": 7.672267198562622
    },
    {
      "epoch": 4.0541648864746096e-05,
      "model_forward_time": 2.9152791500091553,
      "step": 42511
    },
    {
      "epoch": 4.0541648864746096e-05,
      "step": 42511,
      "training_step_time": 7.918173789978027
    },
    {
      "epoch": 4.05426025390625e-05,
      "model_forward_time": 1.7292845249176025,
      "step": 42512
    },
    {
      "epoch": 4.05426025390625e-05,
      "step": 42512,
      "training_step_time": 8.988723516464233
    },
    {
      "epoch": 4.0543556213378904e-05,
      "model_forward_time": 1.75999116897583,
      "step": 42513
    },
    {
      "epoch": 4.0543556213378904e-05,
      "step": 42513,
      "training_step_time": 8.699522495269775
    },
    {
      "epoch": 4.0544509887695315e-05,
      "model_forward_time": 1.7371623516082764,
      "step": 42514
    },
    {
      "epoch": 4.0544509887695315e-05,
      "step": 42514,
      "training_step_time": 7.631964206695557
    },
    {
      "epoch": 4.054546356201172e-05,
      "model_forward_time": 1.7417433261871338,
      "step": 42515
    },
    {
      "epoch": 4.054546356201172e-05,
      "step": 42515,
      "training_step_time": 7.205838441848755
    },
    {
      "epoch": 4.054641723632812e-05,
      "model_forward_time": 1.7480278015136719,
      "step": 42516
    },
    {
      "epoch": 4.054641723632812e-05,
      "step": 42516,
      "training_step_time": 6.876362085342407
    },
    {
      "epoch": 4.0547370910644534e-05,
      "model_forward_time": 1.7452075481414795,
      "step": 42517
    },
    {
      "epoch": 4.0547370910644534e-05,
      "step": 42517,
      "training_step_time": 6.8830602169036865
    },
    {
      "epoch": 4.054832458496094e-05,
      "model_forward_time": 1.7430329322814941,
      "step": 42518
    },
    {
      "epoch": 4.054832458496094e-05,
      "step": 42518,
      "training_step_time": 6.860762596130371
    },
    {
      "epoch": 4.054927825927734e-05,
      "model_forward_time": 1.7468650341033936,
      "step": 42519
    },
    {
      "epoch": 4.054927825927734e-05,
      "step": 42519,
      "training_step_time": 7.636894226074219
    },
    {
      "epoch": 4.055023193359375e-05,
      "grad_norm": 0.14849284291267395,
      "learning_rate": 6.620525653255241e-06,
      "loss": 0.0583,
      "step": 42520
    },
    {
      "epoch": 4.055023193359375e-05,
      "model_forward_time": 2.8538451194763184,
      "step": 42520
    },
    {
      "dynamics_loss_avg": 0.05469194352626801,
      "epoch": 4.055023193359375e-05,
      "step": 42520
    },
    {
      "action_loss_avg": 0.0027702286432031544,
      "epoch": 4.055023193359375e-05,
      "step": 42520
    },
    {
      "epoch": 4.055023193359375e-05,
      "step": 42520,
      "training_step_time": 8.894141674041748
    },
    {
      "epoch": 4.055118560791016e-05,
      "model_forward_time": 1.8419792652130127,
      "step": 42521
    },
    {
      "epoch": 4.055118560791016e-05,
      "step": 42521,
      "training_step_time": 8.531248569488525
    },
    {
      "epoch": 4.055213928222656e-05,
      "model_forward_time": 1.748654842376709,
      "step": 42522
    },
    {
      "epoch": 4.055213928222656e-05,
      "step": 42522,
      "training_step_time": 7.852834939956665
    },
    {
      "epoch": 4.055309295654297e-05,
      "model_forward_time": 1.749779224395752,
      "step": 42523
    },
    {
      "epoch": 4.055309295654297e-05,
      "step": 42523,
      "training_step_time": 7.715407848358154
    },
    {
      "epoch": 4.0554046630859376e-05,
      "model_forward_time": 1.7304751873016357,
      "step": 42524
    },
    {
      "epoch": 4.0554046630859376e-05,
      "step": 42524,
      "training_step_time": 7.205807447433472
    },
    {
      "epoch": 4.055500030517578e-05,
      "model_forward_time": 1.7348976135253906,
      "step": 42525
    },
    {
      "epoch": 4.055500030517578e-05,
      "step": 42525,
      "training_step_time": 6.874409437179565
    },
    {
      "epoch": 4.055595397949219e-05,
      "model_forward_time": 1.7375407218933105,
      "step": 42526
    },
    {
      "epoch": 4.055595397949219e-05,
      "step": 42526,
      "training_step_time": 6.786287546157837
    },
    {
      "epoch": 4.0556907653808595e-05,
      "model_forward_time": 1.7454478740692139,
      "step": 42527
    },
    {
      "epoch": 4.0556907653808595e-05,
      "step": 42527,
      "training_step_time": 6.815831422805786
    },
    {
      "epoch": 4.0557861328125e-05,
      "model_forward_time": 1.7289576530456543,
      "step": 42528
    },
    {
      "epoch": 4.0557861328125e-05,
      "step": 42528,
      "training_step_time": 8.280961990356445
    },
    {
      "epoch": 4.055881500244141e-05,
      "model_forward_time": 2.846137523651123,
      "step": 42529
    },
    {
      "epoch": 4.055881500244141e-05,
      "step": 42529,
      "training_step_time": 7.848094940185547
    },
    {
      "epoch": 4.0559768676757814e-05,
      "grad_norm": 0.16140802204608917,
      "learning_rate": 6.618961347389016e-06,
      "loss": 0.0556,
      "step": 42530
    },
    {
      "epoch": 4.0559768676757814e-05,
      "model_forward_time": 1.8169035911560059,
      "step": 42530
    },
    {
      "dynamics_loss_avg": 0.0410735297948122,
      "epoch": 4.0559768676757814e-05,
      "step": 42530
    },
    {
      "action_loss_avg": 0.0014058914908673613,
      "epoch": 4.0559768676757814e-05,
      "step": 42530
    },
    {
      "epoch": 4.0559768676757814e-05,
      "step": 42530,
      "training_step_time": 9.279054641723633
    },
    {
      "epoch": 4.056072235107422e-05,
      "model_forward_time": 1.7358670234680176,
      "step": 42531
    },
    {
      "epoch": 4.056072235107422e-05,
      "step": 42531,
      "training_step_time": 9.801906824111938
    },
    {
      "epoch": 4.056167602539062e-05,
      "model_forward_time": 1.7332737445831299,
      "step": 42532
    },
    {
      "epoch": 4.056167602539062e-05,
      "step": 42532,
      "training_step_time": 8.91538143157959
    },
    {
      "epoch": 4.056262969970703e-05,
      "model_forward_time": 1.7339766025543213,
      "step": 42533
    },
    {
      "epoch": 4.056262969970703e-05,
      "step": 42533,
      "training_step_time": 7.291937828063965
    },
    {
      "epoch": 4.0563583374023436e-05,
      "model_forward_time": 1.7478444576263428,
      "step": 42534
    },
    {
      "epoch": 4.0563583374023436e-05,
      "step": 42534,
      "training_step_time": 6.894378185272217
    },
    {
      "epoch": 4.056453704833984e-05,
      "model_forward_time": 1.7417738437652588,
      "step": 42535
    },
    {
      "epoch": 4.056453704833984e-05,
      "step": 42535,
      "training_step_time": 6.869035720825195
    },
    {
      "epoch": 4.056549072265625e-05,
      "model_forward_time": 1.743114948272705,
      "step": 42536
    },
    {
      "epoch": 4.056549072265625e-05,
      "step": 42536,
      "training_step_time": 6.865846872329712
    },
    {
      "epoch": 4.0566444396972655e-05,
      "model_forward_time": 1.731133222579956,
      "step": 42537
    },
    {
      "epoch": 4.0566444396972655e-05,
      "step": 42537,
      "training_step_time": 7.643123149871826
    },
    {
      "epoch": 4.056739807128906e-05,
      "model_forward_time": 1.726060152053833,
      "step": 42538
    },
    {
      "epoch": 4.056739807128906e-05,
      "step": 42538,
      "training_step_time": 8.37264347076416
    },
    {
      "epoch": 4.056835174560547e-05,
      "model_forward_time": 1.779407262802124,
      "step": 42539
    },
    {
      "epoch": 4.056835174560547e-05,
      "step": 42539,
      "training_step_time": 7.704190731048584
    },
    {
      "epoch": 4.0569305419921874e-05,
      "grad_norm": 0.15295645594596863,
      "learning_rate": 6.617396864475612e-06,
      "loss": 0.0586,
      "step": 42540
    },
    {
      "epoch": 4.0569305419921874e-05,
      "model_forward_time": 1.7183890342712402,
      "step": 42540
    },
    {
      "dynamics_loss_avg": 0.05193137358874082,
      "epoch": 4.0569305419921874e-05,
      "step": 42540
    },
    {
      "action_loss_avg": 0.0014982501685153692,
      "epoch": 4.0569305419921874e-05,
      "step": 42540
    },
    {
      "epoch": 4.0569305419921874e-05,
      "step": 42540,
      "training_step_time": 8.37480616569519
    },
    {
      "epoch": 4.057025909423828e-05,
      "model_forward_time": 1.7229516506195068,
      "step": 42541
    },
    {
      "epoch": 4.057025909423828e-05,
      "step": 42541,
      "training_step_time": 7.852010488510132
    },
    {
      "epoch": 4.057121276855469e-05,
      "model_forward_time": 1.7315905094146729,
      "step": 42542
    },
    {
      "epoch": 4.057121276855469e-05,
      "step": 42542,
      "training_step_time": 7.741326332092285
    },
    {
      "epoch": 4.057216644287109e-05,
      "model_forward_time": 1.7357735633850098,
      "step": 42543
    },
    {
      "epoch": 4.057216644287109e-05,
      "step": 42543,
      "training_step_time": 6.841922998428345
    },
    {
      "epoch": 4.05731201171875e-05,
      "model_forward_time": 1.7397468090057373,
      "step": 42544
    },
    {
      "epoch": 4.05731201171875e-05,
      "step": 42544,
      "training_step_time": 6.863347053527832
    },
    {
      "epoch": 4.057407379150391e-05,
      "model_forward_time": 1.7288949489593506,
      "step": 42545
    },
    {
      "epoch": 4.057407379150391e-05,
      "step": 42545,
      "training_step_time": 6.868969917297363
    },
    {
      "epoch": 4.057502746582031e-05,
      "model_forward_time": 1.7309751510620117,
      "step": 42546
    },
    {
      "epoch": 4.057502746582031e-05,
      "step": 42546,
      "training_step_time": 7.65714955329895
    },
    {
      "epoch": 4.0575981140136716e-05,
      "model_forward_time": 1.7398991584777832,
      "step": 42547
    },
    {
      "epoch": 4.0575981140136716e-05,
      "step": 42547,
      "training_step_time": 7.825196981430054
    },
    {
      "epoch": 4.057693481445313e-05,
      "model_forward_time": 2.9744045734405518,
      "step": 42548
    },
    {
      "epoch": 4.057693481445313e-05,
      "step": 42548,
      "training_step_time": 8.860705137252808
    },
    {
      "epoch": 4.057788848876953e-05,
      "model_forward_time": 1.7217631340026855,
      "step": 42549
    },
    {
      "epoch": 4.057788848876953e-05,
      "step": 42549,
      "training_step_time": 9.627916097640991
    },
    {
      "epoch": 4.0578842163085935e-05,
      "grad_norm": 0.18004067242145538,
      "learning_rate": 6.61583220468612e-06,
      "loss": 0.0586,
      "step": 42550
    },
    {
      "epoch": 4.0578842163085935e-05,
      "model_forward_time": 1.747087001800537,
      "step": 42550
    },
    {
      "dynamics_loss_avg": 0.06919027287513017,
      "epoch": 4.0578842163085935e-05,
      "step": 42550
    },
    {
      "action_loss_avg": 0.0023977197357453405,
      "epoch": 4.0578842163085935e-05,
      "step": 42550
    },
    {
      "epoch": 4.0578842163085935e-05,
      "step": 42550,
      "training_step_time": 9.9103364944458
    },
    {
      "epoch": 4.0579795837402346e-05,
      "model_forward_time": 1.7229433059692383,
      "step": 42551
    },
    {
      "epoch": 4.0579795837402346e-05,
      "step": 42551,
      "training_step_time": 7.7268500328063965
    },
    {
      "epoch": 4.058074951171875e-05,
      "model_forward_time": 1.7198669910430908,
      "step": 42552
    },
    {
      "epoch": 4.058074951171875e-05,
      "step": 42552,
      "training_step_time": 6.8494884967803955
    },
    {
      "epoch": 4.0581703186035154e-05,
      "model_forward_time": 1.7335197925567627,
      "step": 42553
    },
    {
      "epoch": 4.0581703186035154e-05,
      "step": 42553,
      "training_step_time": 6.885210990905762
    },
    {
      "epoch": 4.0582656860351564e-05,
      "model_forward_time": 1.72530198097229,
      "step": 42554
    },
    {
      "epoch": 4.0582656860351564e-05,
      "step": 42554,
      "training_step_time": 6.893301725387573
    },
    {
      "epoch": 4.058361053466797e-05,
      "model_forward_time": 1.7259769439697266,
      "step": 42555
    },
    {
      "epoch": 4.058361053466797e-05,
      "step": 42555,
      "training_step_time": 7.666090250015259
    },
    {
      "epoch": 4.058456420898437e-05,
      "model_forward_time": 1.7107059955596924,
      "step": 42556
    },
    {
      "epoch": 4.058456420898437e-05,
      "step": 42556,
      "training_step_time": 7.826079845428467
    },
    {
      "epoch": 4.058551788330078e-05,
      "model_forward_time": 2.9493143558502197,
      "step": 42557
    },
    {
      "epoch": 4.058551788330078e-05,
      "step": 42557,
      "training_step_time": 7.8221564292907715
    },
    {
      "epoch": 4.058647155761719e-05,
      "model_forward_time": 1.741600513458252,
      "step": 42558
    },
    {
      "epoch": 4.058647155761719e-05,
      "step": 42558,
      "training_step_time": 10.083744764328003
    },
    {
      "epoch": 4.058742523193359e-05,
      "model_forward_time": 1.751885175704956,
      "step": 42559
    },
    {
      "epoch": 4.058742523193359e-05,
      "step": 42559,
      "training_step_time": 9.400033235549927
    },
    {
      "epoch": 4.058837890625e-05,
      "grad_norm": 0.16733050346374512,
      "learning_rate": 6.614267368191646e-06,
      "loss": 0.0598,
      "step": 42560
    },
    {
      "epoch": 4.058837890625e-05,
      "model_forward_time": 1.7198960781097412,
      "step": 42560
    },
    {
      "dynamics_loss_avg": 0.050891502387821676,
      "epoch": 4.058837890625e-05,
      "step": 42560
    },
    {
      "action_loss_avg": 0.0018210092501249164,
      "epoch": 4.058837890625e-05,
      "step": 42560
    },
    {
      "epoch": 4.058837890625e-05,
      "step": 42560,
      "training_step_time": 8.001870393753052
    },
    {
      "epoch": 4.0589332580566406e-05,
      "model_forward_time": 1.7282254695892334,
      "step": 42561
    },
    {
      "epoch": 4.0589332580566406e-05,
      "step": 42561,
      "training_step_time": 7.317600250244141
    },
    {
      "epoch": 4.059028625488281e-05,
      "model_forward_time": 1.721562385559082,
      "step": 42562
    },
    {
      "epoch": 4.059028625488281e-05,
      "step": 42562,
      "training_step_time": 6.9037251472473145
    },
    {
      "epoch": 4.059123992919922e-05,
      "model_forward_time": 1.743384838104248,
      "step": 42563
    },
    {
      "epoch": 4.059123992919922e-05,
      "step": 42563,
      "training_step_time": 6.904131650924683
    },
    {
      "epoch": 4.0592193603515625e-05,
      "model_forward_time": 1.7285349369049072,
      "step": 42564
    },
    {
      "epoch": 4.0592193603515625e-05,
      "step": 42564,
      "training_step_time": 7.7408177852630615
    },
    {
      "epoch": 4.059314727783203e-05,
      "model_forward_time": 1.7212371826171875,
      "step": 42565
    },
    {
      "epoch": 4.059314727783203e-05,
      "step": 42565,
      "training_step_time": 7.858099460601807
    },
    {
      "epoch": 4.059410095214844e-05,
      "model_forward_time": 2.8215785026550293,
      "step": 42566
    },
    {
      "epoch": 4.059410095214844e-05,
      "step": 42566,
      "training_step_time": 7.857943773269653
    },
    {
      "epoch": 4.0595054626464844e-05,
      "model_forward_time": 1.820826530456543,
      "step": 42567
    },
    {
      "epoch": 4.0595054626464844e-05,
      "step": 42567,
      "training_step_time": 7.755519390106201
    },
    {
      "epoch": 4.059600830078125e-05,
      "model_forward_time": 1.7225432395935059,
      "step": 42568
    },
    {
      "epoch": 4.059600830078125e-05,
      "step": 42568,
      "training_step_time": 7.818953037261963
    },
    {
      "epoch": 4.059696197509766e-05,
      "model_forward_time": 1.7413105964660645,
      "step": 42569
    },
    {
      "epoch": 4.059696197509766e-05,
      "step": 42569,
      "training_step_time": 7.914246559143066
    },
    {
      "epoch": 4.059791564941406e-05,
      "grad_norm": 0.13286912441253662,
      "learning_rate": 6.612702355163321e-06,
      "loss": 0.0619,
      "step": 42570
    },
    {
      "epoch": 4.059791564941406e-05,
      "model_forward_time": 1.7199316024780273,
      "step": 42570
    },
    {
      "dynamics_loss_avg": 0.06148896124213934,
      "epoch": 4.059791564941406e-05,
      "step": 42570
    },
    {
      "action_loss_avg": 0.0018835927417967468,
      "epoch": 4.059791564941406e-05,
      "step": 42570
    },
    {
      "epoch": 4.059791564941406e-05,
      "step": 42570,
      "training_step_time": 7.735389709472656
    },
    {
      "epoch": 4.059886932373047e-05,
      "model_forward_time": 1.7343487739562988,
      "step": 42571
    },
    {
      "epoch": 4.059886932373047e-05,
      "step": 42571,
      "training_step_time": 6.827509880065918
    },
    {
      "epoch": 4.059982299804688e-05,
      "model_forward_time": 1.731574296951294,
      "step": 42572
    },
    {
      "epoch": 4.059982299804688e-05,
      "step": 42572,
      "training_step_time": 6.900846719741821
    },
    {
      "epoch": 4.060077667236328e-05,
      "model_forward_time": 1.750248670578003,
      "step": 42573
    },
    {
      "epoch": 4.060077667236328e-05,
      "step": 42573,
      "training_step_time": 7.69150447845459
    },
    {
      "epoch": 4.0601730346679686e-05,
      "model_forward_time": 1.7284090518951416,
      "step": 42574
    },
    {
      "epoch": 4.0601730346679686e-05,
      "step": 42574,
      "training_step_time": 7.884559392929077
    },
    {
      "epoch": 4.0602684020996096e-05,
      "model_forward_time": 2.8453047275543213,
      "step": 42575
    },
    {
      "epoch": 4.0602684020996096e-05,
      "step": 42575,
      "training_step_time": 8.573601484298706
    },
    {
      "epoch": 4.06036376953125e-05,
      "model_forward_time": 1.7944271564483643,
      "step": 42576
    },
    {
      "epoch": 4.06036376953125e-05,
      "step": 42576,
      "training_step_time": 8.9563627243042
    },
    {
      "epoch": 4.0604591369628905e-05,
      "model_forward_time": 1.7239623069763184,
      "step": 42577
    },
    {
      "epoch": 4.0604591369628905e-05,
      "step": 42577,
      "training_step_time": 8.940540313720703
    },
    {
      "epoch": 4.0605545043945315e-05,
      "model_forward_time": 1.7245635986328125,
      "step": 42578
    },
    {
      "epoch": 4.0605545043945315e-05,
      "step": 42578,
      "training_step_time": 8.53151273727417
    },
    {
      "epoch": 4.060649871826172e-05,
      "model_forward_time": 1.733506202697754,
      "step": 42579
    },
    {
      "epoch": 4.060649871826172e-05,
      "step": 42579,
      "training_step_time": 7.859276294708252
    },
    {
      "epoch": 4.060745239257812e-05,
      "grad_norm": 0.11641339212656021,
      "learning_rate": 6.6111371657722875e-06,
      "loss": 0.0593,
      "step": 42580
    },
    {
      "epoch": 4.060745239257812e-05,
      "model_forward_time": 1.7251369953155518,
      "step": 42580
    },
    {
      "dynamics_loss_avg": 0.05406962363049388,
      "epoch": 4.060745239257812e-05,
      "step": 42580
    },
    {
      "action_loss_avg": 0.0019738444534596056,
      "epoch": 4.060745239257812e-05,
      "step": 42580
    },
    {
      "epoch": 4.060745239257812e-05,
      "step": 42580,
      "training_step_time": 6.857424020767212
    },
    {
      "epoch": 4.0608406066894534e-05,
      "model_forward_time": 1.7174737453460693,
      "step": 42581
    },
    {
      "epoch": 4.0608406066894534e-05,
      "step": 42581,
      "training_step_time": 6.90323281288147
    },
    {
      "epoch": 4.060935974121094e-05,
      "model_forward_time": 1.743682861328125,
      "step": 42582
    },
    {
      "epoch": 4.060935974121094e-05,
      "step": 42582,
      "training_step_time": 7.650291204452515
    },
    {
      "epoch": 4.061031341552734e-05,
      "model_forward_time": 1.367556095123291,
      "step": 42583
    },
    {
      "epoch": 4.061031341552734e-05,
      "step": 42583,
      "training_step_time": 7.88972806930542
    },
    {
      "epoch": 4.061126708984375e-05,
      "model_forward_time": 1.7329552173614502,
      "step": 42584
    },
    {
      "epoch": 4.061126708984375e-05,
      "step": 42584,
      "training_step_time": 7.803317546844482
    },
    {
      "epoch": 4.061222076416016e-05,
      "model_forward_time": 2.969430923461914,
      "step": 42585
    },
    {
      "epoch": 4.061222076416016e-05,
      "step": 42585,
      "training_step_time": 8.793883562088013
    },
    {
      "epoch": 4.061317443847656e-05,
      "model_forward_time": 1.7217297554016113,
      "step": 42586
    },
    {
      "epoch": 4.061317443847656e-05,
      "step": 42586,
      "training_step_time": 9.62783432006836
    },
    {
      "epoch": 4.061412811279297e-05,
      "model_forward_time": 1.7185895442962646,
      "step": 42587
    },
    {
      "epoch": 4.061412811279297e-05,
      "step": 42587,
      "training_step_time": 8.409570932388306
    },
    {
      "epoch": 4.0615081787109376e-05,
      "model_forward_time": 1.725996732711792,
      "step": 42588
    },
    {
      "epoch": 4.0615081787109376e-05,
      "step": 42588,
      "training_step_time": 7.931801080703735
    },
    {
      "epoch": 4.061603546142578e-05,
      "model_forward_time": 1.732072353363037,
      "step": 42589
    },
    {
      "epoch": 4.061603546142578e-05,
      "step": 42589,
      "training_step_time": 7.268556118011475
    },
    {
      "epoch": 4.061698913574219e-05,
      "grad_norm": 0.126459538936615,
      "learning_rate": 6.6095718001897205e-06,
      "loss": 0.0591,
      "step": 42590
    },
    {
      "epoch": 4.061698913574219e-05,
      "model_forward_time": 1.7278249263763428,
      "step": 42590
    },
    {
      "dynamics_loss_avg": 0.05617174431681633,
      "epoch": 4.061698913574219e-05,
      "step": 42590
    },
    {
      "action_loss_avg": 0.0039744560315739365,
      "epoch": 4.061698913574219e-05,
      "step": 42590
    },
    {
      "epoch": 4.061698913574219e-05,
      "step": 42590,
      "training_step_time": 6.819299697875977
    },
    {
      "epoch": 4.0617942810058595e-05,
      "model_forward_time": 1.7365312576293945,
      "step": 42591
    },
    {
      "epoch": 4.0617942810058595e-05,
      "step": 42591,
      "training_step_time": 6.814119815826416
    },
    {
      "epoch": 4.0618896484375e-05,
      "model_forward_time": 1.746246099472046,
      "step": 42592
    },
    {
      "epoch": 4.0618896484375e-05,
      "step": 42592,
      "training_step_time": 7.697769641876221
    },
    {
      "epoch": 4.061985015869141e-05,
      "model_forward_time": 1.7278916835784912,
      "step": 42593
    },
    {
      "epoch": 4.061985015869141e-05,
      "step": 42593,
      "training_step_time": 7.768411874771118
    },
    {
      "epoch": 4.0620803833007814e-05,
      "model_forward_time": 2.9597973823547363,
      "step": 42594
    },
    {
      "epoch": 4.0620803833007814e-05,
      "step": 42594,
      "training_step_time": 7.804055452346802
    },
    {
      "epoch": 4.062175750732422e-05,
      "model_forward_time": 1.7259283065795898,
      "step": 42595
    },
    {
      "epoch": 4.062175750732422e-05,
      "step": 42595,
      "training_step_time": 10.186388969421387
    },
    {
      "epoch": 4.062271118164062e-05,
      "model_forward_time": 1.7250828742980957,
      "step": 42596
    },
    {
      "epoch": 4.062271118164062e-05,
      "step": 42596,
      "training_step_time": 7.794581174850464
    },
    {
      "epoch": 4.062366485595703e-05,
      "model_forward_time": 1.7302196025848389,
      "step": 42597
    },
    {
      "epoch": 4.062366485595703e-05,
      "step": 42597,
      "training_step_time": 7.754183292388916
    },
    {
      "epoch": 4.0624618530273437e-05,
      "model_forward_time": 1.7492446899414062,
      "step": 42598
    },
    {
      "epoch": 4.0624618530273437e-05,
      "step": 42598,
      "training_step_time": 7.311867713928223
    },
    {
      "epoch": 4.062557220458984e-05,
      "model_forward_time": 1.7450697422027588,
      "step": 42599
    },
    {
      "epoch": 4.062557220458984e-05,
      "step": 42599,
      "training_step_time": 6.776276350021362
    },
    {
      "epoch": 4.062652587890625e-05,
      "grad_norm": 0.12774260342121124,
      "learning_rate": 6.608006258586798e-06,
      "loss": 0.0606,
      "step": 42600
    },
    {
      "epoch": 4.062652587890625e-05,
      "model_forward_time": 1.7241394519805908,
      "step": 42600
    },
    {
      "dynamics_loss_avg": 0.056384147889912126,
      "epoch": 4.062652587890625e-05,
      "step": 42600
    },
    {
      "action_loss_avg": 0.0019111690169665962,
      "epoch": 4.062652587890625e-05,
      "step": 42600
    },
    {
      "epoch": 4.062652587890625e-05,
      "step": 42600,
      "training_step_time": 6.821691274642944
    },
    {
      "epoch": 4.0627479553222655e-05,
      "model_forward_time": 1.7448456287384033,
      "step": 42601
    },
    {
      "epoch": 4.0627479553222655e-05,
      "step": 42601,
      "training_step_time": 7.781951904296875
    },
    {
      "epoch": 4.062843322753906e-05,
      "model_forward_time": 1.7144081592559814,
      "step": 42602
    },
    {
      "epoch": 4.062843322753906e-05,
      "step": 42602,
      "training_step_time": 7.786734580993652
    },
    {
      "epoch": 4.062938690185547e-05,
      "model_forward_time": 2.8712897300720215,
      "step": 42603
    },
    {
      "epoch": 4.062938690185547e-05,
      "step": 42603,
      "training_step_time": 7.892126083374023
    },
    {
      "epoch": 4.0630340576171874e-05,
      "model_forward_time": 1.8401589393615723,
      "step": 42604
    },
    {
      "epoch": 4.0630340576171874e-05,
      "step": 42604,
      "training_step_time": 9.850404739379883
    },
    {
      "epoch": 4.063129425048828e-05,
      "model_forward_time": 1.737767219543457,
      "step": 42605
    },
    {
      "epoch": 4.063129425048828e-05,
      "step": 42605,
      "training_step_time": 7.872344255447388
    },
    {
      "epoch": 4.063224792480469e-05,
      "model_forward_time": 1.7461662292480469,
      "step": 42606
    },
    {
      "epoch": 4.063224792480469e-05,
      "step": 42606,
      "training_step_time": 7.795614957809448
    },
    {
      "epoch": 4.063320159912109e-05,
      "model_forward_time": 1.7403054237365723,
      "step": 42607
    },
    {
      "epoch": 4.063320159912109e-05,
      "step": 42607,
      "training_step_time": 7.3394341468811035
    },
    {
      "epoch": 4.06341552734375e-05,
      "model_forward_time": 1.7384610176086426,
      "step": 42608
    },
    {
      "epoch": 4.06341552734375e-05,
      "step": 42608,
      "training_step_time": 6.837354421615601
    },
    {
      "epoch": 4.063510894775391e-05,
      "model_forward_time": 1.7424373626708984,
      "step": 42609
    },
    {
      "epoch": 4.063510894775391e-05,
      "step": 42609,
      "training_step_time": 6.801001310348511
    },
    {
      "epoch": 4.063606262207031e-05,
      "grad_norm": 0.08948762714862823,
      "learning_rate": 6.606440541134731e-06,
      "loss": 0.0584,
      "step": 42610
    },
    {
      "epoch": 4.063606262207031e-05,
      "model_forward_time": 1.7252960205078125,
      "step": 42610
    },
    {
      "dynamics_loss_avg": 0.042927947267889975,
      "epoch": 4.063606262207031e-05,
      "step": 42610
    },
    {
      "action_loss_avg": 0.0017226589960046112,
      "epoch": 4.063606262207031e-05,
      "step": 42610
    },
    {
      "epoch": 4.063606262207031e-05,
      "step": 42610,
      "training_step_time": 7.7762181758880615
    },
    {
      "epoch": 4.0637016296386716e-05,
      "model_forward_time": 1.7374107837677002,
      "step": 42611
    },
    {
      "epoch": 4.0637016296386716e-05,
      "step": 42611,
      "training_step_time": 7.748384475708008
    },
    {
      "epoch": 4.063796997070313e-05,
      "model_forward_time": 2.8362038135528564,
      "step": 42612
    },
    {
      "epoch": 4.063796997070313e-05,
      "step": 42612,
      "training_step_time": 8.646047353744507
    },
    {
      "epoch": 4.063892364501953e-05,
      "model_forward_time": 1.8357787132263184,
      "step": 42613
    },
    {
      "epoch": 4.063892364501953e-05,
      "step": 42613,
      "training_step_time": 9.497873783111572
    },
    {
      "epoch": 4.0639877319335935e-05,
      "model_forward_time": 1.738227128982544,
      "step": 42614
    },
    {
      "epoch": 4.0639877319335935e-05,
      "step": 42614,
      "training_step_time": 8.475722074508667
    },
    {
      "epoch": 4.0640830993652346e-05,
      "model_forward_time": 1.369865894317627,
      "step": 42615
    },
    {
      "epoch": 4.0640830993652346e-05,
      "step": 42615,
      "training_step_time": 7.787895441055298
    },
    {
      "epoch": 4.064178466796875e-05,
      "model_forward_time": 1.7469089031219482,
      "step": 42616
    },
    {
      "epoch": 4.064178466796875e-05,
      "step": 42616,
      "training_step_time": 7.876785516738892
    },
    {
      "epoch": 4.0642738342285154e-05,
      "model_forward_time": 1.369945764541626,
      "step": 42617
    },
    {
      "epoch": 4.0642738342285154e-05,
      "step": 42617,
      "training_step_time": 7.238056421279907
    },
    {
      "epoch": 4.0643692016601565e-05,
      "model_forward_time": 1.7298493385314941,
      "step": 42618
    },
    {
      "epoch": 4.0643692016601565e-05,
      "step": 42618,
      "training_step_time": 6.873917579650879
    },
    {
      "epoch": 4.064464569091797e-05,
      "model_forward_time": 1.7359466552734375,
      "step": 42619
    },
    {
      "epoch": 4.064464569091797e-05,
      "step": 42619,
      "training_step_time": 7.6939778327941895
    },
    {
      "epoch": 4.064559936523437e-05,
      "grad_norm": 0.13661149144172668,
      "learning_rate": 6.60487464800474e-06,
      "loss": 0.0582,
      "step": 42620
    },
    {
      "epoch": 4.064559936523437e-05,
      "model_forward_time": 1.746401309967041,
      "step": 42620
    },
    {
      "dynamics_loss_avg": 0.07010700162500143,
      "epoch": 4.064559936523437e-05,
      "step": 42620
    },
    {
      "action_loss_avg": 0.002538483645184897,
      "epoch": 4.064559936523437e-05,
      "step": 42620
    },
    {
      "epoch": 4.064559936523437e-05,
      "step": 42620,
      "training_step_time": 7.689291000366211
    },
    {
      "epoch": 4.064655303955078e-05,
      "model_forward_time": 1.7229125499725342,
      "step": 42621
    },
    {
      "epoch": 4.064655303955078e-05,
      "step": 42621,
      "training_step_time": 8.281106233596802
    },
    {
      "epoch": 4.064750671386719e-05,
      "model_forward_time": 1.818481683731079,
      "step": 42622
    },
    {
      "epoch": 4.064750671386719e-05,
      "step": 42622,
      "training_step_time": 8.323421955108643
    },
    {
      "epoch": 4.064846038818359e-05,
      "model_forward_time": 1.73280930519104,
      "step": 42623
    },
    {
      "epoch": 4.064846038818359e-05,
      "step": 42623,
      "training_step_time": 10.008317470550537
    },
    {
      "epoch": 4.06494140625e-05,
      "model_forward_time": 1.0375006198883057,
      "step": 42624
    },
    {
      "epoch": 4.06494140625e-05,
      "step": 42624,
      "training_step_time": 8.922173261642456
    },
    {
      "epoch": 4.0650367736816406e-05,
      "model_forward_time": 1.7167649269104004,
      "step": 42625
    },
    {
      "epoch": 4.0650367736816406e-05,
      "step": 42625,
      "training_step_time": 7.877541542053223
    },
    {
      "epoch": 4.065132141113281e-05,
      "model_forward_time": 1.7238802909851074,
      "step": 42626
    },
    {
      "epoch": 4.065132141113281e-05,
      "step": 42626,
      "training_step_time": 7.3511576652526855
    },
    {
      "epoch": 4.065227508544922e-05,
      "model_forward_time": 1.742527961730957,
      "step": 42627
    },
    {
      "epoch": 4.065227508544922e-05,
      "step": 42627,
      "training_step_time": 6.825754404067993
    },
    {
      "epoch": 4.0653228759765625e-05,
      "model_forward_time": 1.7363955974578857,
      "step": 42628
    },
    {
      "epoch": 4.0653228759765625e-05,
      "step": 42628,
      "training_step_time": 7.7769386768341064
    },
    {
      "epoch": 4.065418243408203e-05,
      "model_forward_time": 1.7189695835113525,
      "step": 42629
    },
    {
      "epoch": 4.065418243408203e-05,
      "step": 42629,
      "training_step_time": 7.837784767150879
    },
    {
      "epoch": 4.065513610839844e-05,
      "grad_norm": 0.13087336719036102,
      "learning_rate": 6.60330857936807e-06,
      "loss": 0.0588,
      "step": 42630
    },
    {
      "epoch": 4.065513610839844e-05,
      "model_forward_time": 1.7214689254760742,
      "step": 42630
    },
    {
      "dynamics_loss_avg": 0.06394372936338186,
      "epoch": 4.065513610839844e-05,
      "step": 42630
    },
    {
      "action_loss_avg": 0.001917594822589308,
      "epoch": 4.065513610839844e-05,
      "step": 42630
    },
    {
      "epoch": 4.065513610839844e-05,
      "step": 42630,
      "training_step_time": 7.778961896896362
    },
    {
      "epoch": 4.0656089782714844e-05,
      "model_forward_time": 2.8368735313415527,
      "step": 42631
    },
    {
      "epoch": 4.0656089782714844e-05,
      "step": 42631,
      "training_step_time": 10.09952163696289
    },
    {
      "epoch": 4.065704345703125e-05,
      "model_forward_time": 1.7075486183166504,
      "step": 42632
    },
    {
      "epoch": 4.065704345703125e-05,
      "step": 42632,
      "training_step_time": 8.87984585762024
    },
    {
      "epoch": 4.065799713134766e-05,
      "model_forward_time": 1.365886926651001,
      "step": 42633
    },
    {
      "epoch": 4.065799713134766e-05,
      "step": 42633,
      "training_step_time": 7.857177257537842
    },
    {
      "epoch": 4.065895080566406e-05,
      "model_forward_time": 1.7394757270812988,
      "step": 42634
    },
    {
      "epoch": 4.065895080566406e-05,
      "step": 42634,
      "training_step_time": 7.785169363021851
    },
    {
      "epoch": 4.065990447998047e-05,
      "model_forward_time": 1.7461035251617432,
      "step": 42635
    },
    {
      "epoch": 4.065990447998047e-05,
      "step": 42635,
      "training_step_time": 7.307240962982178
    },
    {
      "epoch": 4.066085815429688e-05,
      "model_forward_time": 1.7551629543304443,
      "step": 42636
    },
    {
      "epoch": 4.066085815429688e-05,
      "step": 42636,
      "training_step_time": 6.77999472618103
    },
    {
      "epoch": 4.066181182861328e-05,
      "model_forward_time": 1.731036901473999,
      "step": 42637
    },
    {
      "epoch": 4.066181182861328e-05,
      "step": 42637,
      "training_step_time": 7.721769332885742
    },
    {
      "epoch": 4.0662765502929686e-05,
      "model_forward_time": 1.7332360744476318,
      "step": 42638
    },
    {
      "epoch": 4.0662765502929686e-05,
      "step": 42638,
      "training_step_time": 7.770552158355713
    },
    {
      "epoch": 4.0663719177246097e-05,
      "model_forward_time": 1.7444498538970947,
      "step": 42639
    },
    {
      "epoch": 4.0663719177246097e-05,
      "step": 42639,
      "training_step_time": 7.758637189865112
    },
    {
      "epoch": 4.06646728515625e-05,
      "grad_norm": 0.7453880906105042,
      "learning_rate": 6.601742335395987e-06,
      "loss": 0.0597,
      "step": 42640
    },
    {
      "epoch": 4.06646728515625e-05,
      "model_forward_time": 2.2708160877227783,
      "step": 42640
    },
    {
      "dynamics_loss_avg": 0.05122315604239702,
      "epoch": 4.06646728515625e-05,
      "step": 42640
    },
    {
      "action_loss_avg": 0.002994688565377146,
      "epoch": 4.06646728515625e-05,
      "step": 42640
    },
    {
      "epoch": 4.06646728515625e-05,
      "step": 42640,
      "training_step_time": 9.30351209640503
    },
    {
      "epoch": 4.0665626525878905e-05,
      "model_forward_time": 1.7639989852905273,
      "step": 42641
    },
    {
      "epoch": 4.0665626525878905e-05,
      "step": 42641,
      "training_step_time": 10.46523380279541
    },
    {
      "epoch": 4.0666580200195315e-05,
      "model_forward_time": 1.7354485988616943,
      "step": 42642
    },
    {
      "epoch": 4.0666580200195315e-05,
      "step": 42642,
      "training_step_time": 8.602815389633179
    },
    {
      "epoch": 4.066753387451172e-05,
      "model_forward_time": 1.7420482635498047,
      "step": 42643
    },
    {
      "epoch": 4.066753387451172e-05,
      "step": 42643,
      "training_step_time": 8.292277097702026
    },
    {
      "epoch": 4.0668487548828123e-05,
      "model_forward_time": 1.7287297248840332,
      "step": 42644
    },
    {
      "epoch": 4.0668487548828123e-05,
      "step": 42644,
      "training_step_time": 7.692073583602905
    },
    {
      "epoch": 4.0669441223144534e-05,
      "model_forward_time": 1.7348542213439941,
      "step": 42645
    },
    {
      "epoch": 4.0669441223144534e-05,
      "step": 42645,
      "training_step_time": 6.828907012939453
    },
    {
      "epoch": 4.067039489746094e-05,
      "model_forward_time": 1.7286174297332764,
      "step": 42646
    },
    {
      "epoch": 4.067039489746094e-05,
      "step": 42646,
      "training_step_time": 7.721966505050659
    },
    {
      "epoch": 4.067134857177734e-05,
      "model_forward_time": 1.735290288925171,
      "step": 42647
    },
    {
      "epoch": 4.067134857177734e-05,
      "step": 42647,
      "training_step_time": 8.725437879562378
    },
    {
      "epoch": 4.067230224609375e-05,
      "model_forward_time": 1.7281959056854248,
      "step": 42648
    },
    {
      "epoch": 4.067230224609375e-05,
      "step": 42648,
      "training_step_time": 7.649496555328369
    },
    {
      "epoch": 4.067325592041016e-05,
      "model_forward_time": 2.2576470375061035,
      "step": 42649
    },
    {
      "epoch": 4.067325592041016e-05,
      "step": 42649,
      "training_step_time": 7.732543468475342
    },
    {
      "epoch": 4.067420959472656e-05,
      "grad_norm": 0.13906428217887878,
      "learning_rate": 6.600175916259769e-06,
      "loss": 0.059,
      "step": 42650
    },
    {
      "epoch": 4.067420959472656e-05,
      "model_forward_time": 1.7386465072631836,
      "step": 42650
    },
    {
      "dynamics_loss_avg": 0.06173628270626068,
      "epoch": 4.067420959472656e-05,
      "step": 42650
    },
    {
      "action_loss_avg": 0.0017903732950799167,
      "epoch": 4.067420959472656e-05,
      "step": 42650
    },
    {
      "epoch": 4.067420959472656e-05,
      "step": 42650,
      "training_step_time": 9.128623962402344
    },
    {
      "epoch": 4.067516326904297e-05,
      "model_forward_time": 1.7682890892028809,
      "step": 42651
    },
    {
      "epoch": 4.067516326904297e-05,
      "step": 42651,
      "training_step_time": 8.384271144866943
    },
    {
      "epoch": 4.0676116943359376e-05,
      "model_forward_time": 1.731666088104248,
      "step": 42652
    },
    {
      "epoch": 4.0676116943359376e-05,
      "step": 42652,
      "training_step_time": 7.750371694564819
    },
    {
      "epoch": 4.067707061767578e-05,
      "model_forward_time": 1.734436273574829,
      "step": 42653
    },
    {
      "epoch": 4.067707061767578e-05,
      "step": 42653,
      "training_step_time": 7.730373859405518
    },
    {
      "epoch": 4.067802429199219e-05,
      "model_forward_time": 1.7275707721710205,
      "step": 42654
    },
    {
      "epoch": 4.067802429199219e-05,
      "step": 42654,
      "training_step_time": 7.196243524551392
    },
    {
      "epoch": 4.0678977966308595e-05,
      "model_forward_time": 1.7091662883758545,
      "step": 42655
    },
    {
      "epoch": 4.0678977966308595e-05,
      "step": 42655,
      "training_step_time": 6.83314061164856
    },
    {
      "epoch": 4.0679931640625e-05,
      "model_forward_time": 1.7228732109069824,
      "step": 42656
    },
    {
      "epoch": 4.0679931640625e-05,
      "step": 42656,
      "training_step_time": 7.827800512313843
    },
    {
      "epoch": 4.06808853149414e-05,
      "model_forward_time": 1.7219412326812744,
      "step": 42657
    },
    {
      "epoch": 4.06808853149414e-05,
      "step": 42657,
      "training_step_time": 8.169673681259155
    },
    {
      "epoch": 4.0681838989257814e-05,
      "model_forward_time": 1.7169971466064453,
      "step": 42658
    },
    {
      "epoch": 4.0681838989257814e-05,
      "step": 42658,
      "training_step_time": 7.673156261444092
    },
    {
      "epoch": 4.068279266357422e-05,
      "model_forward_time": 2.271644115447998,
      "step": 42659
    },
    {
      "epoch": 4.068279266357422e-05,
      "step": 42659,
      "training_step_time": 8.85413122177124
    },
    {
      "epoch": 4.068374633789062e-05,
      "grad_norm": 0.15881913900375366,
      "learning_rate": 6.598609322130718e-06,
      "loss": 0.0592,
      "step": 42660
    },
    {
      "epoch": 4.068374633789062e-05,
      "model_forward_time": 1.767209529876709,
      "step": 42660
    },
    {
      "dynamics_loss_avg": 0.07626562304794789,
      "epoch": 4.068374633789062e-05,
      "step": 42660
    },
    {
      "action_loss_avg": 0.00243344638729468,
      "epoch": 4.068374633789062e-05,
      "step": 42660
    },
    {
      "epoch": 4.068374633789062e-05,
      "step": 42660,
      "training_step_time": 9.556682586669922
    },
    {
      "epoch": 4.068470001220703e-05,
      "model_forward_time": 1.7286345958709717,
      "step": 42661
    },
    {
      "epoch": 4.068470001220703e-05,
      "step": 42661,
      "training_step_time": 8.50062084197998
    },
    {
      "epoch": 4.0685653686523437e-05,
      "model_forward_time": 1.7199928760528564,
      "step": 42662
    },
    {
      "epoch": 4.0685653686523437e-05,
      "step": 42662,
      "training_step_time": 7.775323867797852
    },
    {
      "epoch": 4.068660736083984e-05,
      "model_forward_time": 1.7263953685760498,
      "step": 42663
    },
    {
      "epoch": 4.068660736083984e-05,
      "step": 42663,
      "training_step_time": 7.23551082611084
    },
    {
      "epoch": 4.068756103515625e-05,
      "model_forward_time": 1.7263309955596924,
      "step": 42664
    },
    {
      "epoch": 4.068756103515625e-05,
      "step": 42664,
      "training_step_time": 6.800099849700928
    },
    {
      "epoch": 4.0688514709472655e-05,
      "model_forward_time": 1.7229561805725098,
      "step": 42665
    },
    {
      "epoch": 4.0688514709472655e-05,
      "step": 42665,
      "training_step_time": 7.839796304702759
    },
    {
      "epoch": 4.068946838378906e-05,
      "model_forward_time": 1.7222025394439697,
      "step": 42666
    },
    {
      "epoch": 4.068946838378906e-05,
      "step": 42666,
      "training_step_time": 8.451062202453613
    },
    {
      "epoch": 4.069042205810547e-05,
      "model_forward_time": 1.72953200340271,
      "step": 42667
    },
    {
      "epoch": 4.069042205810547e-05,
      "step": 42667,
      "training_step_time": 8.577737092971802
    },
    {
      "epoch": 4.0691375732421874e-05,
      "model_forward_time": 2.272367238998413,
      "step": 42668
    },
    {
      "epoch": 4.0691375732421874e-05,
      "step": 42668,
      "training_step_time": 7.7864134311676025
    },
    {
      "epoch": 4.069232940673828e-05,
      "model_forward_time": 1.7690191268920898,
      "step": 42669
    },
    {
      "epoch": 4.069232940673828e-05,
      "step": 42669,
      "training_step_time": 8.852332830429077
    },
    {
      "epoch": 4.069328308105469e-05,
      "grad_norm": 0.13030798733234406,
      "learning_rate": 6.597042553180155e-06,
      "loss": 0.0574,
      "step": 42670
    },
    {
      "epoch": 4.069328308105469e-05,
      "model_forward_time": 1.719512701034546,
      "step": 42670
    },
    {
      "dynamics_loss_avg": 0.06985391769558191,
      "epoch": 4.069328308105469e-05,
      "step": 42670
    },
    {
      "action_loss_avg": 0.0029668744507944213,
      "epoch": 4.069328308105469e-05,
      "step": 42670
    },
    {
      "epoch": 4.069328308105469e-05,
      "step": 42670,
      "training_step_time": 8.900322198867798
    },
    {
      "epoch": 4.069423675537109e-05,
      "model_forward_time": 1.7278883457183838,
      "step": 42671
    },
    {
      "epoch": 4.069423675537109e-05,
      "step": 42671,
      "training_step_time": 7.734660625457764
    },
    {
      "epoch": 4.06951904296875e-05,
      "model_forward_time": 1.7205121517181396,
      "step": 42672
    },
    {
      "epoch": 4.06951904296875e-05,
      "step": 42672,
      "training_step_time": 7.304222106933594
    },
    {
      "epoch": 4.069614410400391e-05,
      "model_forward_time": 1.7239799499511719,
      "step": 42673
    },
    {
      "epoch": 4.069614410400391e-05,
      "step": 42673,
      "training_step_time": 6.804243087768555
    },
    {
      "epoch": 4.069709777832031e-05,
      "model_forward_time": 1.720393419265747,
      "step": 42674
    },
    {
      "epoch": 4.069709777832031e-05,
      "step": 42674,
      "training_step_time": 7.781033754348755
    },
    {
      "epoch": 4.0698051452636716e-05,
      "model_forward_time": 1.7231180667877197,
      "step": 42675
    },
    {
      "epoch": 4.0698051452636716e-05,
      "step": 42675,
      "training_step_time": 7.631862640380859
    },
    {
      "epoch": 4.069900512695313e-05,
      "model_forward_time": 1.722125768661499,
      "step": 42676
    },
    {
      "epoch": 4.069900512695313e-05,
      "step": 42676,
      "training_step_time": 7.709075927734375
    },
    {
      "epoch": 4.069995880126953e-05,
      "model_forward_time": 2.2868309020996094,
      "step": 42677
    },
    {
      "epoch": 4.069995880126953e-05,
      "step": 42677,
      "training_step_time": 8.468282699584961
    },
    {
      "epoch": 4.0700912475585935e-05,
      "model_forward_time": 1.770655870437622,
      "step": 42678
    },
    {
      "epoch": 4.0700912475585935e-05,
      "step": 42678,
      "training_step_time": 7.78994607925415
    },
    {
      "epoch": 4.0701866149902346e-05,
      "model_forward_time": 1.7291631698608398,
      "step": 42679
    },
    {
      "epoch": 4.0701866149902346e-05,
      "step": 42679,
      "training_step_time": 7.7776899337768555
    },
    {
      "epoch": 4.070281982421875e-05,
      "grad_norm": 0.1441698670387268,
      "learning_rate": 6.59547560957942e-06,
      "loss": 0.0557,
      "step": 42680
    },
    {
      "epoch": 4.070281982421875e-05,
      "model_forward_time": 1.7280259132385254,
      "step": 42680
    },
    {
      "dynamics_loss_avg": 0.06080363132059574,
      "epoch": 4.070281982421875e-05,
      "step": 42680
    },
    {
      "action_loss_avg": 0.0019415828806813805,
      "epoch": 4.070281982421875e-05,
      "step": 42680
    },
    {
      "epoch": 4.070281982421875e-05,
      "step": 42680,
      "training_step_time": 7.7762534618377686
    },
    {
      "epoch": 4.0703773498535154e-05,
      "model_forward_time": 1.726635456085205,
      "step": 42681
    },
    {
      "epoch": 4.0703773498535154e-05,
      "step": 42681,
      "training_step_time": 7.699563264846802
    },
    {
      "epoch": 4.0704727172851565e-05,
      "model_forward_time": 1.7288837432861328,
      "step": 42682
    },
    {
      "epoch": 4.0704727172851565e-05,
      "step": 42682,
      "training_step_time": 7.229977130889893
    },
    {
      "epoch": 4.070568084716797e-05,
      "model_forward_time": 1.3573722839355469,
      "step": 42683
    },
    {
      "epoch": 4.070568084716797e-05,
      "step": 42683,
      "training_step_time": 7.723667860031128
    },
    {
      "epoch": 4.070663452148437e-05,
      "model_forward_time": 1.7604570388793945,
      "step": 42684
    },
    {
      "epoch": 4.070663452148437e-05,
      "step": 42684,
      "training_step_time": 7.609686374664307
    },
    {
      "epoch": 4.0707588195800783e-05,
      "model_forward_time": 1.7244822978973389,
      "step": 42685
    },
    {
      "epoch": 4.0707588195800783e-05,
      "step": 42685,
      "training_step_time": 7.630244731903076
    },
    {
      "epoch": 4.070854187011719e-05,
      "model_forward_time": 2.28082537651062,
      "step": 42686
    },
    {
      "epoch": 4.070854187011719e-05,
      "step": 42686,
      "training_step_time": 8.603271007537842
    },
    {
      "epoch": 4.070949554443359e-05,
      "model_forward_time": 1.762768030166626,
      "step": 42687
    },
    {
      "epoch": 4.070949554443359e-05,
      "step": 42687,
      "training_step_time": 10.110887050628662
    },
    {
      "epoch": 4.071044921875e-05,
      "model_forward_time": 1.7267341613769531,
      "step": 42688
    },
    {
      "epoch": 4.071044921875e-05,
      "step": 42688,
      "training_step_time": 7.81709623336792
    },
    {
      "epoch": 4.0711402893066406e-05,
      "model_forward_time": 1.72861909866333,
      "step": 42689
    },
    {
      "epoch": 4.0711402893066406e-05,
      "step": 42689,
      "training_step_time": 7.807338714599609
    },
    {
      "epoch": 4.071235656738281e-05,
      "grad_norm": 0.08361376076936722,
      "learning_rate": 6.593908491499871e-06,
      "loss": 0.0597,
      "step": 42690
    },
    {
      "epoch": 4.071235656738281e-05,
      "model_forward_time": 1.7259502410888672,
      "step": 42690
    },
    {
      "dynamics_loss_avg": 0.05478836838155985,
      "epoch": 4.071235656738281e-05,
      "step": 42690
    },
    {
      "action_loss_avg": 0.0015312350064050407,
      "epoch": 4.071235656738281e-05,
      "step": 42690
    },
    {
      "epoch": 4.071235656738281e-05,
      "step": 42690,
      "training_step_time": 7.876096487045288
    },
    {
      "epoch": 4.071331024169922e-05,
      "model_forward_time": 1.7350742816925049,
      "step": 42691
    },
    {
      "epoch": 4.071331024169922e-05,
      "step": 42691,
      "training_step_time": 6.902175664901733
    },
    {
      "epoch": 4.0714263916015625e-05,
      "model_forward_time": 1.734257698059082,
      "step": 42692
    },
    {
      "epoch": 4.0714263916015625e-05,
      "step": 42692,
      "training_step_time": 8.206194400787354
    },
    {
      "epoch": 4.071521759033203e-05,
      "model_forward_time": 1.724151611328125,
      "step": 42693
    },
    {
      "epoch": 4.071521759033203e-05,
      "step": 42693,
      "training_step_time": 7.805079698562622
    },
    {
      "epoch": 4.071617126464844e-05,
      "model_forward_time": 1.7269961833953857,
      "step": 42694
    },
    {
      "epoch": 4.071617126464844e-05,
      "step": 42694,
      "training_step_time": 7.671034336090088
    },
    {
      "epoch": 4.0717124938964844e-05,
      "model_forward_time": 1.7361335754394531,
      "step": 42695
    },
    {
      "epoch": 4.0717124938964844e-05,
      "step": 42695,
      "training_step_time": 8.321641206741333
    },
    {
      "epoch": 4.071807861328125e-05,
      "model_forward_time": 1.7502593994140625,
      "step": 42696
    },
    {
      "epoch": 4.071807861328125e-05,
      "step": 42696,
      "training_step_time": 8.465662717819214
    },
    {
      "epoch": 4.071903228759766e-05,
      "model_forward_time": 1.7343626022338867,
      "step": 42697
    },
    {
      "epoch": 4.071903228759766e-05,
      "step": 42697,
      "training_step_time": 9.540169954299927
    },
    {
      "epoch": 4.071998596191406e-05,
      "model_forward_time": 1.729813575744629,
      "step": 42698
    },
    {
      "epoch": 4.071998596191406e-05,
      "step": 42698,
      "training_step_time": 8.48482894897461
    },
    {
      "epoch": 4.072093963623047e-05,
      "model_forward_time": 1.727464199066162,
      "step": 42699
    },
    {
      "epoch": 4.072093963623047e-05,
      "step": 42699,
      "training_step_time": 7.831974744796753
    },
    {
      "epoch": 4.072189331054688e-05,
      "grad_norm": 0.1052749902009964,
      "learning_rate": 6.592341199112886e-06,
      "loss": 0.0558,
      "step": 42700
    },
    {
      "epoch": 4.072189331054688e-05,
      "model_forward_time": 1.7073593139648438,
      "step": 42700
    },
    {
      "dynamics_loss_avg": 0.044701299723237754,
      "epoch": 4.072189331054688e-05,
      "step": 42700
    },
    {
      "action_loss_avg": 0.0017858110135421158,
      "epoch": 4.072189331054688e-05,
      "step": 42700
    },
    {
      "epoch": 4.072189331054688e-05,
      "step": 42700,
      "training_step_time": 7.321813106536865
    },
    {
      "epoch": 4.072284698486328e-05,
      "model_forward_time": 1.7204668521881104,
      "step": 42701
    },
    {
      "epoch": 4.072284698486328e-05,
      "step": 42701,
      "training_step_time": 7.713698148727417
    },
    {
      "epoch": 4.0723800659179686e-05,
      "model_forward_time": 1.7250425815582275,
      "step": 42702
    },
    {
      "epoch": 4.0723800659179686e-05,
      "step": 42702,
      "training_step_time": 7.8616228103637695
    },
    {
      "epoch": 4.0724754333496097e-05,
      "model_forward_time": 1.7225852012634277,
      "step": 42703
    },
    {
      "epoch": 4.0724754333496097e-05,
      "step": 42703,
      "training_step_time": 7.2138683795928955
    },
    {
      "epoch": 4.07257080078125e-05,
      "model_forward_time": 1.7181830406188965,
      "step": 42704
    },
    {
      "epoch": 4.07257080078125e-05,
      "step": 42704,
      "training_step_time": 7.807532787322998
    },
    {
      "epoch": 4.0726661682128905e-05,
      "model_forward_time": 2.2997913360595703,
      "step": 42705
    },
    {
      "epoch": 4.0726661682128905e-05,
      "step": 42705,
      "training_step_time": 9.752735614776611
    },
    {
      "epoch": 4.0727615356445315e-05,
      "model_forward_time": 1.7214655876159668,
      "step": 42706
    },
    {
      "epoch": 4.0727615356445315e-05,
      "step": 42706,
      "training_step_time": 8.602068662643433
    },
    {
      "epoch": 4.072856903076172e-05,
      "model_forward_time": 1.7192823886871338,
      "step": 42707
    },
    {
      "epoch": 4.072856903076172e-05,
      "step": 42707,
      "training_step_time": 7.975418567657471
    },
    {
      "epoch": 4.0729522705078123e-05,
      "model_forward_time": 1.7230501174926758,
      "step": 42708
    },
    {
      "epoch": 4.0729522705078123e-05,
      "step": 42708,
      "training_step_time": 8.3182373046875
    },
    {
      "epoch": 4.0730476379394534e-05,
      "model_forward_time": 1.7259557247161865,
      "step": 42709
    },
    {
      "epoch": 4.0730476379394534e-05,
      "step": 42709,
      "training_step_time": 7.246920585632324
    },
    {
      "epoch": 4.073143005371094e-05,
      "grad_norm": 0.11221137642860413,
      "learning_rate": 6.5907737325898615e-06,
      "loss": 0.0581,
      "step": 42710
    },
    {
      "epoch": 4.073143005371094e-05,
      "model_forward_time": 1.7175860404968262,
      "step": 42710
    },
    {
      "dynamics_loss_avg": 0.057679282315075395,
      "epoch": 4.073143005371094e-05,
      "step": 42710
    },
    {
      "action_loss_avg": 0.002744219417218119,
      "epoch": 4.073143005371094e-05,
      "step": 42710
    },
    {
      "epoch": 4.073143005371094e-05,
      "step": 42710,
      "training_step_time": 7.550489902496338
    },
    {
      "epoch": 4.073238372802734e-05,
      "model_forward_time": 1.7231769561767578,
      "step": 42711
    },
    {
      "epoch": 4.073238372802734e-05,
      "step": 42711,
      "training_step_time": 7.76101016998291
    },
    {
      "epoch": 4.073333740234375e-05,
      "model_forward_time": 1.7153277397155762,
      "step": 42712
    },
    {
      "epoch": 4.073333740234375e-05,
      "step": 42712,
      "training_step_time": 8.221781492233276
    },
    {
      "epoch": 4.073429107666016e-05,
      "model_forward_time": 1.7291953563690186,
      "step": 42713
    },
    {
      "epoch": 4.073429107666016e-05,
      "step": 42713,
      "training_step_time": 7.641573429107666
    },
    {
      "epoch": 4.073524475097656e-05,
      "model_forward_time": 2.2761237621307373,
      "step": 42714
    },
    {
      "epoch": 4.073524475097656e-05,
      "step": 42714,
      "training_step_time": 8.609972715377808
    },
    {
      "epoch": 4.073619842529297e-05,
      "model_forward_time": 1.7734370231628418,
      "step": 42715
    },
    {
      "epoch": 4.073619842529297e-05,
      "step": 42715,
      "training_step_time": 7.774040222167969
    },
    {
      "epoch": 4.0737152099609376e-05,
      "model_forward_time": 1.7479133605957031,
      "step": 42716
    },
    {
      "epoch": 4.0737152099609376e-05,
      "step": 42716,
      "training_step_time": 8.762594223022461
    },
    {
      "epoch": 4.073810577392578e-05,
      "model_forward_time": 1.7203238010406494,
      "step": 42717
    },
    {
      "epoch": 4.073810577392578e-05,
      "step": 42717,
      "training_step_time": 8.306818723678589
    },
    {
      "epoch": 4.073905944824219e-05,
      "model_forward_time": 1.728374719619751,
      "step": 42718
    },
    {
      "epoch": 4.073905944824219e-05,
      "step": 42718,
      "training_step_time": 7.225499153137207
    },
    {
      "epoch": 4.0740013122558595e-05,
      "model_forward_time": 1.7306420803070068,
      "step": 42719
    },
    {
      "epoch": 4.0740013122558595e-05,
      "step": 42719,
      "training_step_time": 7.524101495742798
    },
    {
      "epoch": 4.0740966796875e-05,
      "grad_norm": 0.10667191445827484,
      "learning_rate": 6.589206092102212e-06,
      "loss": 0.0573,
      "step": 42720
    },
    {
      "epoch": 4.0740966796875e-05,
      "model_forward_time": 1.7589523792266846,
      "step": 42720
    },
    {
      "dynamics_loss_avg": 0.07502630352973938,
      "epoch": 4.0740966796875e-05,
      "step": 42720
    },
    {
      "action_loss_avg": 0.0026976998429745435,
      "epoch": 4.0740966796875e-05,
      "step": 42720
    },
    {
      "epoch": 4.0740966796875e-05,
      "step": 42720,
      "training_step_time": 7.799447298049927
    },
    {
      "epoch": 4.07419204711914e-05,
      "model_forward_time": 1.7314887046813965,
      "step": 42721
    },
    {
      "epoch": 4.07419204711914e-05,
      "step": 42721,
      "training_step_time": 7.563802003860474
    },
    {
      "epoch": 4.0742874145507814e-05,
      "model_forward_time": 1.733839511871338,
      "step": 42722
    },
    {
      "epoch": 4.0742874145507814e-05,
      "step": 42722,
      "training_step_time": 8.760581731796265
    },
    {
      "epoch": 4.074382781982422e-05,
      "model_forward_time": 2.3031814098358154,
      "step": 42723
    },
    {
      "epoch": 4.074382781982422e-05,
      "step": 42723,
      "training_step_time": 8.59773850440979
    },
    {
      "epoch": 4.074478149414062e-05,
      "model_forward_time": 1.7669966220855713,
      "step": 42724
    },
    {
      "epoch": 4.074478149414062e-05,
      "step": 42724,
      "training_step_time": 9.007818698883057
    },
    {
      "epoch": 4.074573516845703e-05,
      "model_forward_time": 1.7478604316711426,
      "step": 42725
    },
    {
      "epoch": 4.074573516845703e-05,
      "step": 42725,
      "training_step_time": 7.702540159225464
    },
    {
      "epoch": 4.074668884277344e-05,
      "model_forward_time": 1.7443623542785645,
      "step": 42726
    },
    {
      "epoch": 4.074668884277344e-05,
      "step": 42726,
      "training_step_time": 8.661733150482178
    },
    {
      "epoch": 4.074764251708984e-05,
      "model_forward_time": 1.7405049800872803,
      "step": 42727
    },
    {
      "epoch": 4.074764251708984e-05,
      "step": 42727,
      "training_step_time": 7.673171758651733
    },
    {
      "epoch": 4.074859619140625e-05,
      "model_forward_time": 1.7432990074157715,
      "step": 42728
    },
    {
      "epoch": 4.074859619140625e-05,
      "step": 42728,
      "training_step_time": 7.577169418334961
    },
    {
      "epoch": 4.0749549865722655e-05,
      "model_forward_time": 1.7447960376739502,
      "step": 42729
    },
    {
      "epoch": 4.0749549865722655e-05,
      "step": 42729,
      "training_step_time": 7.746190547943115
    },
    {
      "epoch": 4.075050354003906e-05,
      "grad_norm": 0.10457818955183029,
      "learning_rate": 6.5876382778213756e-06,
      "loss": 0.0568,
      "step": 42730
    },
    {
      "epoch": 4.075050354003906e-05,
      "model_forward_time": 1.7343080043792725,
      "step": 42730
    },
    {
      "dynamics_loss_avg": 0.03967035096138716,
      "epoch": 4.075050354003906e-05,
      "step": 42730
    },
    {
      "action_loss_avg": 0.0019106341642327607,
      "epoch": 4.075050354003906e-05,
      "step": 42730
    },
    {
      "epoch": 4.075050354003906e-05,
      "step": 42730,
      "training_step_time": 7.650160312652588
    },
    {
      "epoch": 4.075145721435547e-05,
      "model_forward_time": 1.7432277202606201,
      "step": 42731
    },
    {
      "epoch": 4.075145721435547e-05,
      "step": 42731,
      "training_step_time": 7.7337646484375
    },
    {
      "epoch": 4.0752410888671874e-05,
      "model_forward_time": 2.317591905593872,
      "step": 42732
    },
    {
      "epoch": 4.0752410888671874e-05,
      "step": 42732,
      "training_step_time": 7.713646650314331
    },
    {
      "epoch": 4.075336456298828e-05,
      "model_forward_time": 1.7560863494873047,
      "step": 42733
    },
    {
      "epoch": 4.075336456298828e-05,
      "step": 42733,
      "training_step_time": 8.315958738327026
    },
    {
      "epoch": 4.075431823730469e-05,
      "model_forward_time": 1.7473747730255127,
      "step": 42734
    },
    {
      "epoch": 4.075431823730469e-05,
      "step": 42734,
      "training_step_time": 9.278628826141357
    },
    {
      "epoch": 4.075527191162109e-05,
      "model_forward_time": 1.7437615394592285,
      "step": 42735
    },
    {
      "epoch": 4.075527191162109e-05,
      "step": 42735,
      "training_step_time": 7.82900071144104
    },
    {
      "epoch": 4.07562255859375e-05,
      "model_forward_time": 1.7456185817718506,
      "step": 42736
    },
    {
      "epoch": 4.07562255859375e-05,
      "step": 42736,
      "training_step_time": 7.79041600227356
    },
    {
      "epoch": 4.075717926025391e-05,
      "model_forward_time": 1.7405760288238525,
      "step": 42737
    },
    {
      "epoch": 4.075717926025391e-05,
      "step": 42737,
      "training_step_time": 7.6178059577941895
    },
    {
      "epoch": 4.075813293457031e-05,
      "model_forward_time": 1.7443947792053223,
      "step": 42738
    },
    {
      "epoch": 4.075813293457031e-05,
      "step": 42738,
      "training_step_time": 7.8445823192596436
    },
    {
      "epoch": 4.0759086608886716e-05,
      "model_forward_time": 1.7434558868408203,
      "step": 42739
    },
    {
      "epoch": 4.0759086608886716e-05,
      "step": 42739,
      "training_step_time": 7.6822826862335205
    },
    {
      "epoch": 4.076004028320313e-05,
      "grad_norm": 0.09667184948921204,
      "learning_rate": 6.586070289918803e-06,
      "loss": 0.0559,
      "step": 42740
    },
    {
      "epoch": 4.076004028320313e-05,
      "model_forward_time": 1.7384629249572754,
      "step": 42740
    },
    {
      "dynamics_loss_avg": 0.07527118641883135,
      "epoch": 4.076004028320313e-05,
      "step": 42740
    },
    {
      "action_loss_avg": 0.0025861938076559452,
      "epoch": 4.076004028320313e-05,
      "step": 42740
    },
    {
      "epoch": 4.076004028320313e-05,
      "step": 42740,
      "training_step_time": 7.525899887084961
    },
    {
      "epoch": 4.076099395751953e-05,
      "model_forward_time": 2.2628602981567383,
      "step": 42741
    },
    {
      "epoch": 4.076099395751953e-05,
      "step": 42741,
      "training_step_time": 8.241298198699951
    },
    {
      "epoch": 4.0761947631835935e-05,
      "model_forward_time": 1.7425105571746826,
      "step": 42742
    },
    {
      "epoch": 4.0761947631835935e-05,
      "step": 42742,
      "training_step_time": 8.894851207733154
    },
    {
      "epoch": 4.0762901306152346e-05,
      "model_forward_time": 1.7772588729858398,
      "step": 42743
    },
    {
      "epoch": 4.0762901306152346e-05,
      "step": 42743,
      "training_step_time": 8.760963201522827
    },
    {
      "epoch": 4.076385498046875e-05,
      "model_forward_time": 1.7392611503601074,
      "step": 42744
    },
    {
      "epoch": 4.076385498046875e-05,
      "step": 42744,
      "training_step_time": 8.77957034111023
    },
    {
      "epoch": 4.0764808654785154e-05,
      "model_forward_time": 1.7449743747711182,
      "step": 42745
    },
    {
      "epoch": 4.0764808654785154e-05,
      "step": 42745,
      "training_step_time": 7.791278839111328
    },
    {
      "epoch": 4.0765762329101565e-05,
      "model_forward_time": 1.746715784072876,
      "step": 42746
    },
    {
      "epoch": 4.0765762329101565e-05,
      "step": 42746,
      "training_step_time": 7.708315134048462
    },
    {
      "epoch": 4.076671600341797e-05,
      "model_forward_time": 1.374483585357666,
      "step": 42747
    },
    {
      "epoch": 4.076671600341797e-05,
      "step": 42747,
      "training_step_time": 7.683986186981201
    },
    {
      "epoch": 4.076766967773437e-05,
      "model_forward_time": 1.7499687671661377,
      "step": 42748
    },
    {
      "epoch": 4.076766967773437e-05,
      "step": 42748,
      "training_step_time": 7.7572407722473145
    },
    {
      "epoch": 4.0768623352050783e-05,
      "model_forward_time": 1.7499959468841553,
      "step": 42749
    },
    {
      "epoch": 4.0768623352050783e-05,
      "step": 42749,
      "training_step_time": 7.648857593536377
    },
    {
      "epoch": 4.076957702636719e-05,
      "grad_norm": 0.10976498574018478,
      "learning_rate": 6.584502128565968e-06,
      "loss": 0.0597,
      "step": 42750
    },
    {
      "epoch": 4.076957702636719e-05,
      "model_forward_time": 1.7473478317260742,
      "step": 42750
    },
    {
      "dynamics_loss_avg": 0.05937628913670778,
      "epoch": 4.076957702636719e-05,
      "step": 42750
    },
    {
      "action_loss_avg": 0.0021291475626640023,
      "epoch": 4.076957702636719e-05,
      "step": 42750
    },
    {
      "epoch": 4.076957702636719e-05,
      "step": 42750,
      "training_step_time": 7.680485010147095
    },
    {
      "epoch": 4.077053070068359e-05,
      "model_forward_time": 2.2985172271728516,
      "step": 42751
    },
    {
      "epoch": 4.077053070068359e-05,
      "step": 42751,
      "training_step_time": 8.689881801605225
    },
    {
      "epoch": 4.0771484375e-05,
      "model_forward_time": 1.8010029792785645,
      "step": 42752
    },
    {
      "epoch": 4.0771484375e-05,
      "step": 42752,
      "training_step_time": 8.344859600067139
    },
    {
      "epoch": 4.0772438049316406e-05,
      "model_forward_time": 1.7498211860656738,
      "step": 42753
    },
    {
      "epoch": 4.0772438049316406e-05,
      "step": 42753,
      "training_step_time": 7.841953277587891
    },
    {
      "epoch": 4.077339172363281e-05,
      "model_forward_time": 1.7475264072418213,
      "step": 42754
    },
    {
      "epoch": 4.077339172363281e-05,
      "step": 42754,
      "training_step_time": 7.910750865936279
    },
    {
      "epoch": 4.077434539794922e-05,
      "model_forward_time": 1.750645637512207,
      "step": 42755
    },
    {
      "epoch": 4.077434539794922e-05,
      "step": 42755,
      "training_step_time": 7.761256456375122
    },
    {
      "epoch": 4.0775299072265625e-05,
      "model_forward_time": 1.7517337799072266,
      "step": 42756
    },
    {
      "epoch": 4.0775299072265625e-05,
      "step": 42756,
      "training_step_time": 7.346499681472778
    },
    {
      "epoch": 4.077625274658203e-05,
      "model_forward_time": 1.750903606414795,
      "step": 42757
    },
    {
      "epoch": 4.077625274658203e-05,
      "step": 42757,
      "training_step_time": 7.808971881866455
    },
    {
      "epoch": 4.077720642089844e-05,
      "model_forward_time": 1.7476990222930908,
      "step": 42758
    },
    {
      "epoch": 4.077720642089844e-05,
      "step": 42758,
      "training_step_time": 7.657475233078003
    },
    {
      "epoch": 4.0778160095214844e-05,
      "model_forward_time": 1.7451984882354736,
      "step": 42759
    },
    {
      "epoch": 4.0778160095214844e-05,
      "step": 42759,
      "training_step_time": 7.717891454696655
    },
    {
      "epoch": 4.077911376953125e-05,
      "grad_norm": 0.10858640819787979,
      "learning_rate": 6.582933793934363e-06,
      "loss": 0.0599,
      "step": 42760
    },
    {
      "epoch": 4.077911376953125e-05,
      "model_forward_time": 2.2993357181549072,
      "step": 42760
    },
    {
      "dynamics_loss_avg": 0.07753934264183045,
      "epoch": 4.077911376953125e-05,
      "step": 42760
    },
    {
      "action_loss_avg": 0.0017932111746631562,
      "epoch": 4.077911376953125e-05,
      "step": 42760
    },
    {
      "epoch": 4.077911376953125e-05,
      "step": 42760,
      "training_step_time": 9.62357783317566
    },
    {
      "epoch": 4.078006744384766e-05,
      "model_forward_time": 1.78627610206604,
      "step": 42761
    },
    {
      "epoch": 4.078006744384766e-05,
      "step": 42761,
      "training_step_time": 10.147514581680298
    },
    {
      "epoch": 4.078102111816406e-05,
      "model_forward_time": 1.7629778385162354,
      "step": 42762
    },
    {
      "epoch": 4.078102111816406e-05,
      "step": 42762,
      "training_step_time": 7.854932546615601
    },
    {
      "epoch": 4.078197479248047e-05,
      "model_forward_time": 1.7515296936035156,
      "step": 42763
    },
    {
      "epoch": 4.078197479248047e-05,
      "step": 42763,
      "training_step_time": 7.916699647903442
    },
    {
      "epoch": 4.078292846679688e-05,
      "model_forward_time": 1.748291015625,
      "step": 42764
    },
    {
      "epoch": 4.078292846679688e-05,
      "step": 42764,
      "training_step_time": 8.64899492263794
    },
    {
      "epoch": 4.078388214111328e-05,
      "model_forward_time": 1.7509329319000244,
      "step": 42765
    },
    {
      "epoch": 4.078388214111328e-05,
      "step": 42765,
      "training_step_time": 7.376672267913818
    },
    {
      "epoch": 4.0784835815429686e-05,
      "model_forward_time": 1.7455716133117676,
      "step": 42766
    },
    {
      "epoch": 4.0784835815429686e-05,
      "step": 42766,
      "training_step_time": 7.914209842681885
    },
    {
      "epoch": 4.07857894897461e-05,
      "model_forward_time": 1.7498106956481934,
      "step": 42767
    },
    {
      "epoch": 4.07857894897461e-05,
      "step": 42767,
      "training_step_time": 7.681875467300415
    },
    {
      "epoch": 4.07867431640625e-05,
      "model_forward_time": 1.748053789138794,
      "step": 42768
    },
    {
      "epoch": 4.07867431640625e-05,
      "step": 42768,
      "training_step_time": 7.8015265464782715
    },
    {
      "epoch": 4.0787696838378905e-05,
      "model_forward_time": 2.3159067630767822,
      "step": 42769
    },
    {
      "epoch": 4.0787696838378905e-05,
      "step": 42769,
      "training_step_time": 8.351573944091797
    },
    {
      "epoch": 4.0788650512695315e-05,
      "grad_norm": 0.12351599335670471,
      "learning_rate": 6.581365286195498e-06,
      "loss": 0.0586,
      "step": 42770
    },
    {
      "epoch": 4.0788650512695315e-05,
      "model_forward_time": 1.7701787948608398,
      "step": 42770
    },
    {
      "dynamics_loss_avg": 0.059539202600717545,
      "epoch": 4.0788650512695315e-05,
      "step": 42770
    },
    {
      "action_loss_avg": 0.0014507025713101029,
      "epoch": 4.0788650512695315e-05,
      "step": 42770
    },
    {
      "epoch": 4.0788650512695315e-05,
      "step": 42770,
      "training_step_time": 9.244550466537476
    },
    {
      "epoch": 4.078960418701172e-05,
      "model_forward_time": 1.7461779117584229,
      "step": 42771
    },
    {
      "epoch": 4.078960418701172e-05,
      "step": 42771,
      "training_step_time": 7.928565979003906
    },
    {
      "epoch": 4.0790557861328124e-05,
      "model_forward_time": 1.75563645362854,
      "step": 42772
    },
    {
      "epoch": 4.0790557861328124e-05,
      "step": 42772,
      "training_step_time": 7.871097087860107
    },
    {
      "epoch": 4.0791511535644534e-05,
      "model_forward_time": 1.7420716285705566,
      "step": 42773
    },
    {
      "epoch": 4.0791511535644534e-05,
      "step": 42773,
      "training_step_time": 7.764749050140381
    },
    {
      "epoch": 4.079246520996094e-05,
      "model_forward_time": 1.7477121353149414,
      "step": 42774
    },
    {
      "epoch": 4.079246520996094e-05,
      "step": 42774,
      "training_step_time": 8.28655457496643
    },
    {
      "epoch": 4.079341888427734e-05,
      "model_forward_time": 1.7482833862304688,
      "step": 42775
    },
    {
      "epoch": 4.079341888427734e-05,
      "step": 42775,
      "training_step_time": 7.840113162994385
    },
    {
      "epoch": 4.079437255859375e-05,
      "model_forward_time": 1.7528102397918701,
      "step": 42776
    },
    {
      "epoch": 4.079437255859375e-05,
      "step": 42776,
      "training_step_time": 7.7468554973602295
    },
    {
      "epoch": 4.079532623291016e-05,
      "model_forward_time": 1.7572250366210938,
      "step": 42777
    },
    {
      "epoch": 4.079532623291016e-05,
      "step": 42777,
      "training_step_time": 9.673470258712769
    },
    {
      "epoch": 4.079627990722656e-05,
      "model_forward_time": 2.2880806922912598,
      "step": 42778
    },
    {
      "epoch": 4.079627990722656e-05,
      "step": 42778,
      "training_step_time": 8.722352027893066
    },
    {
      "epoch": 4.079723358154297e-05,
      "model_forward_time": 1.7380435466766357,
      "step": 42779
    },
    {
      "epoch": 4.079723358154297e-05,
      "step": 42779,
      "training_step_time": 8.895910501480103
    },
    {
      "epoch": 4.0798187255859376e-05,
      "grad_norm": 0.123959019780159,
      "learning_rate": 6.579796605520904e-06,
      "loss": 0.06,
      "step": 42780
    },
    {
      "epoch": 4.0798187255859376e-05,
      "model_forward_time": 1.7498209476470947,
      "step": 42780
    },
    {
      "dynamics_loss_avg": 0.05597179215401411,
      "epoch": 4.0798187255859376e-05,
      "step": 42780
    },
    {
      "action_loss_avg": 0.002451074108830653,
      "epoch": 4.0798187255859376e-05,
      "step": 42780
    },
    {
      "epoch": 4.0798187255859376e-05,
      "step": 42780,
      "training_step_time": 9.454724073410034
    },
    {
      "epoch": 4.079914093017578e-05,
      "model_forward_time": 1.745344638824463,
      "step": 42781
    },
    {
      "epoch": 4.079914093017578e-05,
      "step": 42781,
      "training_step_time": 7.949977874755859
    },
    {
      "epoch": 4.0800094604492184e-05,
      "model_forward_time": 1.7487058639526367,
      "step": 42782
    },
    {
      "epoch": 4.0800094604492184e-05,
      "step": 42782,
      "training_step_time": 7.715504169464111
    },
    {
      "epoch": 4.0801048278808595e-05,
      "model_forward_time": 1.7486414909362793,
      "step": 42783
    },
    {
      "epoch": 4.0801048278808595e-05,
      "step": 42783,
      "training_step_time": 7.741365432739258
    },
    {
      "epoch": 4.0802001953125e-05,
      "model_forward_time": 1.751807689666748,
      "step": 42784
    },
    {
      "epoch": 4.0802001953125e-05,
      "step": 42784,
      "training_step_time": 7.887862205505371
    },
    {
      "epoch": 4.08029556274414e-05,
      "model_forward_time": 1.7513127326965332,
      "step": 42785
    },
    {
      "epoch": 4.08029556274414e-05,
      "step": 42785,
      "training_step_time": 7.72368597984314
    },
    {
      "epoch": 4.0803909301757814e-05,
      "model_forward_time": 1.7487578392028809,
      "step": 42786
    },
    {
      "epoch": 4.0803909301757814e-05,
      "step": 42786,
      "training_step_time": 8.395215034484863
    },
    {
      "epoch": 4.080486297607422e-05,
      "model_forward_time": 1.7519805431365967,
      "step": 42787
    },
    {
      "epoch": 4.080486297607422e-05,
      "step": 42787,
      "training_step_time": 8.285959005355835
    },
    {
      "epoch": 4.080581665039062e-05,
      "model_forward_time": 1.7420127391815186,
      "step": 42788
    },
    {
      "epoch": 4.080581665039062e-05,
      "step": 42788,
      "training_step_time": 8.719502687454224
    },
    {
      "epoch": 4.080677032470703e-05,
      "model_forward_time": 1.7925007343292236,
      "step": 42789
    },
    {
      "epoch": 4.080677032470703e-05,
      "step": 42789,
      "training_step_time": 8.03587293624878
    },
    {
      "epoch": 4.080772399902344e-05,
      "grad_norm": 0.1389610320329666,
      "learning_rate": 6.578227752082127e-06,
      "loss": 0.0567,
      "step": 42790
    },
    {
      "epoch": 4.080772399902344e-05,
      "model_forward_time": 1.7337555885314941,
      "step": 42790
    },
    {
      "dynamics_loss_avg": 0.05284802541136742,
      "epoch": 4.080772399902344e-05,
      "step": 42790
    },
    {
      "action_loss_avg": 0.0021278087864629925,
      "epoch": 4.080772399902344e-05,
      "step": 42790
    },
    {
      "epoch": 4.080772399902344e-05,
      "step": 42790,
      "training_step_time": 7.965463638305664
    },
    {
      "epoch": 4.080867767333984e-05,
      "model_forward_time": 1.7394845485687256,
      "step": 42791
    },
    {
      "epoch": 4.080867767333984e-05,
      "step": 42791,
      "training_step_time": 7.887733697891235
    },
    {
      "epoch": 4.080963134765625e-05,
      "model_forward_time": 1.7341182231903076,
      "step": 42792
    },
    {
      "epoch": 4.080963134765625e-05,
      "step": 42792,
      "training_step_time": 7.863528251647949
    },
    {
      "epoch": 4.0810585021972656e-05,
      "model_forward_time": 1.739159107208252,
      "step": 42793
    },
    {
      "epoch": 4.0810585021972656e-05,
      "step": 42793,
      "training_step_time": 8.414427757263184
    },
    {
      "epoch": 4.081153869628906e-05,
      "model_forward_time": 1.3872017860412598,
      "step": 42794
    },
    {
      "epoch": 4.081153869628906e-05,
      "step": 42794,
      "training_step_time": 7.73887825012207
    },
    {
      "epoch": 4.081249237060547e-05,
      "model_forward_time": 1.7358369827270508,
      "step": 42795
    },
    {
      "epoch": 4.081249237060547e-05,
      "step": 42795,
      "training_step_time": 7.618032217025757
    },
    {
      "epoch": 4.0813446044921874e-05,
      "model_forward_time": 1.7219583988189697,
      "step": 42796
    },
    {
      "epoch": 4.0813446044921874e-05,
      "step": 42796,
      "training_step_time": 8.39473843574524
    },
    {
      "epoch": 4.081439971923828e-05,
      "model_forward_time": 2.294271230697632,
      "step": 42797
    },
    {
      "epoch": 4.081439971923828e-05,
      "step": 42797,
      "training_step_time": 7.948945999145508
    },
    {
      "epoch": 4.081535339355469e-05,
      "model_forward_time": 1.7691407203674316,
      "step": 42798
    },
    {
      "epoch": 4.081535339355469e-05,
      "step": 42798,
      "training_step_time": 7.992248773574829
    },
    {
      "epoch": 4.081630706787109e-05,
      "model_forward_time": 1.7467901706695557,
      "step": 42799
    },
    {
      "epoch": 4.081630706787109e-05,
      "step": 42799,
      "training_step_time": 8.643087148666382
    },
    {
      "epoch": 4.08172607421875e-05,
      "grad_norm": 0.10947204381227493,
      "learning_rate": 6.576658726050735e-06,
      "loss": 0.0586,
      "step": 42800
    },
    {
      "epoch": 4.08172607421875e-05,
      "model_forward_time": 1.7484385967254639,
      "step": 42800
    },
    {
      "dynamics_loss_avg": 0.05764107378199697,
      "epoch": 4.08172607421875e-05,
      "step": 42800
    },
    {
      "action_loss_avg": 0.0020123371243244035,
      "epoch": 4.08172607421875e-05,
      "step": 42800
    },
    {
      "epoch": 4.08172607421875e-05,
      "step": 42800,
      "training_step_time": 8.686562538146973
    },
    {
      "epoch": 4.081821441650391e-05,
      "model_forward_time": 1.750530481338501,
      "step": 42801
    },
    {
      "epoch": 4.081821441650391e-05,
      "step": 42801,
      "training_step_time": 8.09857177734375
    },
    {
      "epoch": 4.081916809082031e-05,
      "model_forward_time": 1.74570631980896,
      "step": 42802
    },
    {
      "epoch": 4.081916809082031e-05,
      "step": 42802,
      "training_step_time": 7.3101277351379395
    },
    {
      "epoch": 4.0820121765136716e-05,
      "model_forward_time": 1.7499175071716309,
      "step": 42803
    },
    {
      "epoch": 4.0820121765136716e-05,
      "step": 42803,
      "training_step_time": 7.796962022781372
    },
    {
      "epoch": 4.082107543945313e-05,
      "model_forward_time": 1.7319278717041016,
      "step": 42804
    },
    {
      "epoch": 4.082107543945313e-05,
      "step": 42804,
      "training_step_time": 7.599693298339844
    },
    {
      "epoch": 4.082202911376953e-05,
      "model_forward_time": 1.7439355850219727,
      "step": 42805
    },
    {
      "epoch": 4.082202911376953e-05,
      "step": 42805,
      "training_step_time": 7.807848691940308
    },
    {
      "epoch": 4.0822982788085935e-05,
      "model_forward_time": 2.2896201610565186,
      "step": 42806
    },
    {
      "epoch": 4.0822982788085935e-05,
      "step": 42806,
      "training_step_time": 8.385759830474854
    },
    {
      "epoch": 4.0823936462402346e-05,
      "model_forward_time": 1.7919492721557617,
      "step": 42807
    },
    {
      "epoch": 4.0823936462402346e-05,
      "step": 42807,
      "training_step_time": 8.387949228286743
    },
    {
      "epoch": 4.082489013671875e-05,
      "model_forward_time": 1.7446751594543457,
      "step": 42808
    },
    {
      "epoch": 4.082489013671875e-05,
      "step": 42808,
      "training_step_time": 7.856341123580933
    },
    {
      "epoch": 4.0825843811035154e-05,
      "model_forward_time": 1.7408134937286377,
      "step": 42809
    },
    {
      "epoch": 4.0825843811035154e-05,
      "step": 42809,
      "training_step_time": 7.760560750961304
    },
    {
      "epoch": 4.0826797485351565e-05,
      "grad_norm": 0.12879404425621033,
      "learning_rate": 6.575089527598316e-06,
      "loss": 0.0612,
      "step": 42810
    },
    {
      "epoch": 4.0826797485351565e-05,
      "model_forward_time": 1.731835126876831,
      "step": 42810
    },
    {
      "dynamics_loss_avg": 0.05618702881038189,
      "epoch": 4.0826797485351565e-05,
      "step": 42810
    },
    {
      "action_loss_avg": 0.002214532741345465,
      "epoch": 4.0826797485351565e-05,
      "step": 42810
    },
    {
      "epoch": 4.0826797485351565e-05,
      "step": 42810,
      "training_step_time": 7.511030435562134
    },
    {
      "epoch": 4.082775115966797e-05,
      "model_forward_time": 1.7417538166046143,
      "step": 42811
    },
    {
      "epoch": 4.082775115966797e-05,
      "step": 42811,
      "training_step_time": 7.749403476715088
    },
    {
      "epoch": 4.082870483398437e-05,
      "model_forward_time": 1.7379488945007324,
      "step": 42812
    },
    {
      "epoch": 4.082870483398437e-05,
      "step": 42812,
      "training_step_time": 7.920299291610718
    },
    {
      "epoch": 4.0829658508300784e-05,
      "model_forward_time": 1.7478690147399902,
      "step": 42813
    },
    {
      "epoch": 4.0829658508300784e-05,
      "step": 42813,
      "training_step_time": 8.526313781738281
    },
    {
      "epoch": 4.083061218261719e-05,
      "model_forward_time": 1.7437644004821777,
      "step": 42814
    },
    {
      "epoch": 4.083061218261719e-05,
      "step": 42814,
      "training_step_time": 8.278193235397339
    },
    {
      "epoch": 4.083156585693359e-05,
      "model_forward_time": 2.3078861236572266,
      "step": 42815
    },
    {
      "epoch": 4.083156585693359e-05,
      "step": 42815,
      "training_step_time": 9.902141571044922
    },
    {
      "epoch": 4.083251953125e-05,
      "model_forward_time": 1.7665348052978516,
      "step": 42816
    },
    {
      "epoch": 4.083251953125e-05,
      "step": 42816,
      "training_step_time": 9.688154697418213
    },
    {
      "epoch": 4.0833473205566406e-05,
      "model_forward_time": 1.7624216079711914,
      "step": 42817
    },
    {
      "epoch": 4.0833473205566406e-05,
      "step": 42817,
      "training_step_time": 7.773634195327759
    },
    {
      "epoch": 4.083442687988281e-05,
      "model_forward_time": 1.7366628646850586,
      "step": 42818
    },
    {
      "epoch": 4.083442687988281e-05,
      "step": 42818,
      "training_step_time": 8.542588472366333
    },
    {
      "epoch": 4.083538055419922e-05,
      "model_forward_time": 1.7585084438323975,
      "step": 42819
    },
    {
      "epoch": 4.083538055419922e-05,
      "step": 42819,
      "training_step_time": 7.772218227386475
    },
    {
      "epoch": 4.0836334228515625e-05,
      "grad_norm": 0.1442718803882599,
      "learning_rate": 6.573520156896476e-06,
      "loss": 0.0626,
      "step": 42820
    },
    {
      "epoch": 4.0836334228515625e-05,
      "model_forward_time": 1.7322280406951904,
      "step": 42820
    },
    {
      "dynamics_loss_avg": 0.06520549580454826,
      "epoch": 4.0836334228515625e-05,
      "step": 42820
    },
    {
      "action_loss_avg": 0.002212503500049934,
      "epoch": 4.0836334228515625e-05,
      "step": 42820
    },
    {
      "epoch": 4.0836334228515625e-05,
      "step": 42820,
      "training_step_time": 7.713206768035889
    },
    {
      "epoch": 4.083728790283203e-05,
      "model_forward_time": 1.7430305480957031,
      "step": 42821
    },
    {
      "epoch": 4.083728790283203e-05,
      "step": 42821,
      "training_step_time": 7.803618431091309
    },
    {
      "epoch": 4.083824157714844e-05,
      "model_forward_time": 1.7464940547943115,
      "step": 42822
    },
    {
      "epoch": 4.083824157714844e-05,
      "step": 42822,
      "training_step_time": 7.173794507980347
    },
    {
      "epoch": 4.0839195251464844e-05,
      "model_forward_time": 1.7413575649261475,
      "step": 42823
    },
    {
      "epoch": 4.0839195251464844e-05,
      "step": 42823,
      "training_step_time": 7.746150255203247
    },
    {
      "epoch": 4.084014892578125e-05,
      "model_forward_time": 2.314039945602417,
      "step": 42824
    },
    {
      "epoch": 4.084014892578125e-05,
      "step": 42824,
      "training_step_time": 7.723447322845459
    },
    {
      "epoch": 4.084110260009766e-05,
      "model_forward_time": 1.7425220012664795,
      "step": 42825
    },
    {
      "epoch": 4.084110260009766e-05,
      "step": 42825,
      "training_step_time": 8.878561019897461
    },
    {
      "epoch": 4.084205627441406e-05,
      "model_forward_time": 1.7894234657287598,
      "step": 42826
    },
    {
      "epoch": 4.084205627441406e-05,
      "step": 42826,
      "training_step_time": 7.752102613449097
    },
    {
      "epoch": 4.084300994873047e-05,
      "model_forward_time": 1.739149808883667,
      "step": 42827
    },
    {
      "epoch": 4.084300994873047e-05,
      "step": 42827,
      "training_step_time": 7.885018587112427
    },
    {
      "epoch": 4.084396362304688e-05,
      "model_forward_time": 1.7304410934448242,
      "step": 42828
    },
    {
      "epoch": 4.084396362304688e-05,
      "step": 42828,
      "training_step_time": 8.765194654464722
    },
    {
      "epoch": 4.084491729736328e-05,
      "model_forward_time": 1.7420375347137451,
      "step": 42829
    },
    {
      "epoch": 4.084491729736328e-05,
      "step": 42829,
      "training_step_time": 7.667498588562012
    },
    {
      "epoch": 4.0845870971679686e-05,
      "grad_norm": 0.14462634921073914,
      "learning_rate": 6.571950614116835e-06,
      "loss": 0.0608,
      "step": 42830
    },
    {
      "epoch": 4.0845870971679686e-05,
      "model_forward_time": 1.7296881675720215,
      "step": 42830
    },
    {
      "dynamics_loss_avg": 0.06686555817723275,
      "epoch": 4.0845870971679686e-05,
      "step": 42830
    },
    {
      "action_loss_avg": 0.0021962244180031123,
      "epoch": 4.0845870971679686e-05,
      "step": 42830
    },
    {
      "epoch": 4.0845870971679686e-05,
      "step": 42830,
      "training_step_time": 7.720704555511475
    },
    {
      "epoch": 4.08468246459961e-05,
      "model_forward_time": 1.741224765777588,
      "step": 42831
    },
    {
      "epoch": 4.08468246459961e-05,
      "step": 42831,
      "training_step_time": 7.190880537033081
    },
    {
      "epoch": 4.08477783203125e-05,
      "model_forward_time": 1.7307171821594238,
      "step": 42832
    },
    {
      "epoch": 4.08477783203125e-05,
      "step": 42832,
      "training_step_time": 7.677994012832642
    },
    {
      "epoch": 4.0848731994628905e-05,
      "model_forward_time": 1.7408833503723145,
      "step": 42833
    },
    {
      "epoch": 4.0848731994628905e-05,
      "step": 42833,
      "training_step_time": 8.259110689163208
    },
    {
      "epoch": 4.0849685668945316e-05,
      "model_forward_time": 1.760164737701416,
      "step": 42834
    },
    {
      "epoch": 4.0849685668945316e-05,
      "step": 42834,
      "training_step_time": 9.491621971130371
    },
    {
      "epoch": 4.085063934326172e-05,
      "model_forward_time": 1.7419748306274414,
      "step": 42835
    },
    {
      "epoch": 4.085063934326172e-05,
      "step": 42835,
      "training_step_time": 9.064281702041626
    },
    {
      "epoch": 4.0851593017578124e-05,
      "model_forward_time": 1.7399578094482422,
      "step": 42836
    },
    {
      "epoch": 4.0851593017578124e-05,
      "step": 42836,
      "training_step_time": 8.748197078704834
    },
    {
      "epoch": 4.0852546691894534e-05,
      "model_forward_time": 1.7446489334106445,
      "step": 42837
    },
    {
      "epoch": 4.0852546691894534e-05,
      "step": 42837,
      "training_step_time": 7.606492519378662
    },
    {
      "epoch": 4.085350036621094e-05,
      "model_forward_time": 1.742687463760376,
      "step": 42838
    },
    {
      "epoch": 4.085350036621094e-05,
      "step": 42838,
      "training_step_time": 7.656945466995239
    },
    {
      "epoch": 4.085445404052734e-05,
      "model_forward_time": 1.7430152893066406,
      "step": 42839
    },
    {
      "epoch": 4.085445404052734e-05,
      "step": 42839,
      "training_step_time": 8.10246467590332
    },
    {
      "epoch": 4.085540771484375e-05,
      "grad_norm": 0.1326771378517151,
      "learning_rate": 6.570380899431039e-06,
      "loss": 0.0606,
      "step": 42840
    },
    {
      "epoch": 4.085540771484375e-05,
      "model_forward_time": 1.7352294921875,
      "step": 42840
    },
    {
      "dynamics_loss_avg": 0.06077769286930561,
      "epoch": 4.085540771484375e-05,
      "step": 42840
    },
    {
      "action_loss_avg": 0.0032074549701064823,
      "epoch": 4.085540771484375e-05,
      "step": 42840
    },
    {
      "epoch": 4.085540771484375e-05,
      "step": 42840,
      "training_step_time": 7.601769924163818
    },
    {
      "epoch": 4.085636138916016e-05,
      "model_forward_time": 1.7428393363952637,
      "step": 42841
    },
    {
      "epoch": 4.085636138916016e-05,
      "step": 42841,
      "training_step_time": 7.484337568283081
    },
    {
      "epoch": 4.085731506347656e-05,
      "model_forward_time": 1.7390310764312744,
      "step": 42842
    },
    {
      "epoch": 4.085731506347656e-05,
      "step": 42842,
      "training_step_time": 8.496151685714722
    },
    {
      "epoch": 4.085826873779297e-05,
      "model_forward_time": 2.286619186401367,
      "step": 42843
    },
    {
      "epoch": 4.085826873779297e-05,
      "step": 42843,
      "training_step_time": 9.447425842285156
    },
    {
      "epoch": 4.0859222412109376e-05,
      "model_forward_time": 1.738572120666504,
      "step": 42844
    },
    {
      "epoch": 4.0859222412109376e-05,
      "step": 42844,
      "training_step_time": 8.341841697692871
    },
    {
      "epoch": 4.086017608642578e-05,
      "model_forward_time": 1.789987325668335,
      "step": 42845
    },
    {
      "epoch": 4.086017608642578e-05,
      "step": 42845,
      "training_step_time": 7.655318975448608
    },
    {
      "epoch": 4.0861129760742184e-05,
      "model_forward_time": 1.7440946102142334,
      "step": 42846
    },
    {
      "epoch": 4.0861129760742184e-05,
      "step": 42846,
      "training_step_time": 7.630244731903076
    },
    {
      "epoch": 4.0862083435058595e-05,
      "model_forward_time": 1.7439918518066406,
      "step": 42847
    },
    {
      "epoch": 4.0862083435058595e-05,
      "step": 42847,
      "training_step_time": 7.653429985046387
    },
    {
      "epoch": 4.0863037109375e-05,
      "model_forward_time": 1.7582058906555176,
      "step": 42848
    },
    {
      "epoch": 4.0863037109375e-05,
      "step": 42848,
      "training_step_time": 7.6558849811553955
    },
    {
      "epoch": 4.08639907836914e-05,
      "model_forward_time": 1.7542154788970947,
      "step": 42849
    },
    {
      "epoch": 4.08639907836914e-05,
      "step": 42849,
      "training_step_time": 7.750413179397583
    },
    {
      "epoch": 4.0864944458007814e-05,
      "grad_norm": 0.13063576817512512,
      "learning_rate": 6.568811013010749e-06,
      "loss": 0.0598,
      "step": 42850
    },
    {
      "epoch": 4.0864944458007814e-05,
      "model_forward_time": 1.7427699565887451,
      "step": 42850
    },
    {
      "dynamics_loss_avg": 0.05148239191621542,
      "epoch": 4.0864944458007814e-05,
      "step": 42850
    },
    {
      "action_loss_avg": 0.0018074114501359872,
      "epoch": 4.0864944458007814e-05,
      "step": 42850
    },
    {
      "epoch": 4.0864944458007814e-05,
      "step": 42850,
      "training_step_time": 7.476682424545288
    },
    {
      "epoch": 4.086589813232422e-05,
      "model_forward_time": 1.7536985874176025,
      "step": 42851
    },
    {
      "epoch": 4.086589813232422e-05,
      "step": 42851,
      "training_step_time": 9.108473777770996
    },
    {
      "epoch": 4.086685180664062e-05,
      "model_forward_time": 2.304185390472412,
      "step": 42852
    },
    {
      "epoch": 4.086685180664062e-05,
      "step": 42852,
      "training_step_time": 8.358365297317505
    },
    {
      "epoch": 4.086780548095703e-05,
      "model_forward_time": 1.7482020854949951,
      "step": 42853
    },
    {
      "epoch": 4.086780548095703e-05,
      "step": 42853,
      "training_step_time": 9.2174551486969
    },
    {
      "epoch": 4.086875915527344e-05,
      "model_forward_time": 1.7841041088104248,
      "step": 42854
    },
    {
      "epoch": 4.086875915527344e-05,
      "step": 42854,
      "training_step_time": 8.376656770706177
    },
    {
      "epoch": 4.086971282958984e-05,
      "model_forward_time": 1.7585780620574951,
      "step": 42855
    },
    {
      "epoch": 4.086971282958984e-05,
      "step": 42855,
      "training_step_time": 7.845242977142334
    },
    {
      "epoch": 4.087066650390625e-05,
      "model_forward_time": 1.7548613548278809,
      "step": 42856
    },
    {
      "epoch": 4.087066650390625e-05,
      "step": 42856,
      "training_step_time": 7.746587753295898
    },
    {
      "epoch": 4.0871620178222656e-05,
      "model_forward_time": 1.749394178390503,
      "step": 42857
    },
    {
      "epoch": 4.0871620178222656e-05,
      "step": 42857,
      "training_step_time": 7.746920108795166
    },
    {
      "epoch": 4.087257385253906e-05,
      "model_forward_time": 1.761756420135498,
      "step": 42858
    },
    {
      "epoch": 4.087257385253906e-05,
      "step": 42858,
      "training_step_time": 7.8457300662994385
    },
    {
      "epoch": 4.087352752685547e-05,
      "model_forward_time": 1.7500672340393066,
      "step": 42859
    },
    {
      "epoch": 4.087352752685547e-05,
      "step": 42859,
      "training_step_time": 8.379244565963745
    },
    {
      "epoch": 4.0874481201171874e-05,
      "grad_norm": 0.11131580173969269,
      "learning_rate": 6.567240955027646e-06,
      "loss": 0.0604,
      "step": 42860
    },
    {
      "epoch": 4.0874481201171874e-05,
      "model_forward_time": 1.7459032535552979,
      "step": 42860
    },
    {
      "dynamics_loss_avg": 0.0691434433683753,
      "epoch": 4.0874481201171874e-05,
      "step": 42860
    },
    {
      "action_loss_avg": 0.0037620505318045616,
      "epoch": 4.0874481201171874e-05,
      "step": 42860
    },
    {
      "epoch": 4.0874481201171874e-05,
      "step": 42860,
      "training_step_time": 8.357120752334595
    },
    {
      "epoch": 4.087543487548828e-05,
      "model_forward_time": 2.297379493713379,
      "step": 42861
    },
    {
      "epoch": 4.087543487548828e-05,
      "step": 42861,
      "training_step_time": 8.58265233039856
    },
    {
      "epoch": 4.087638854980469e-05,
      "model_forward_time": 1.7507960796356201,
      "step": 42862
    },
    {
      "epoch": 4.087638854980469e-05,
      "step": 42862,
      "training_step_time": 8.551868915557861
    },
    {
      "epoch": 4.087734222412109e-05,
      "model_forward_time": 1.7907567024230957,
      "step": 42863
    },
    {
      "epoch": 4.087734222412109e-05,
      "step": 42863,
      "training_step_time": 8.28036093711853
    },
    {
      "epoch": 4.08782958984375e-05,
      "model_forward_time": 1.7451567649841309,
      "step": 42864
    },
    {
      "epoch": 4.08782958984375e-05,
      "step": 42864,
      "training_step_time": 7.8697028160095215
    },
    {
      "epoch": 4.087924957275391e-05,
      "model_forward_time": 1.7570388317108154,
      "step": 42865
    },
    {
      "epoch": 4.087924957275391e-05,
      "step": 42865,
      "training_step_time": 8.731264591217041
    },
    {
      "epoch": 4.088020324707031e-05,
      "model_forward_time": 1.7507305145263672,
      "step": 42866
    },
    {
      "epoch": 4.088020324707031e-05,
      "step": 42866,
      "training_step_time": 7.787198305130005
    },
    {
      "epoch": 4.0881156921386716e-05,
      "model_forward_time": 1.7431983947753906,
      "step": 42867
    },
    {
      "epoch": 4.0881156921386716e-05,
      "step": 42867,
      "training_step_time": 8.366729974746704
    },
    {
      "epoch": 4.088211059570313e-05,
      "model_forward_time": 1.7487616539001465,
      "step": 42868
    },
    {
      "epoch": 4.088211059570313e-05,
      "step": 42868,
      "training_step_time": 6.943331480026245
    },
    {
      "epoch": 4.088306427001953e-05,
      "model_forward_time": 1.749046802520752,
      "step": 42869
    },
    {
      "epoch": 4.088306427001953e-05,
      "step": 42869,
      "training_step_time": 8.347469568252563
    },
    {
      "epoch": 4.0884017944335935e-05,
      "grad_norm": 0.15344388782978058,
      "learning_rate": 6.565670725653428e-06,
      "loss": 0.057,
      "step": 42870
    },
    {
      "epoch": 4.0884017944335935e-05,
      "model_forward_time": 2.3096694946289062,
      "step": 42870
    },
    {
      "dynamics_loss_avg": 0.04128639716655016,
      "epoch": 4.0884017944335935e-05,
      "step": 42870
    },
    {
      "action_loss_avg": 0.001467223817599006,
      "epoch": 4.0884017944335935e-05,
      "step": 42870
    },
    {
      "epoch": 4.0884017944335935e-05,
      "step": 42870,
      "training_step_time": 8.721700668334961
    },
    {
      "epoch": 4.0884971618652346e-05,
      "model_forward_time": 1.7429101467132568,
      "step": 42871
    },
    {
      "epoch": 4.0884971618652346e-05,
      "step": 42871,
      "training_step_time": 9.621189832687378
    },
    {
      "epoch": 4.088592529296875e-05,
      "model_forward_time": 1.7550716400146484,
      "step": 42872
    },
    {
      "epoch": 4.088592529296875e-05,
      "step": 42872,
      "training_step_time": 7.896435737609863
    },
    {
      "epoch": 4.0886878967285154e-05,
      "model_forward_time": 1.7858326435089111,
      "step": 42873
    },
    {
      "epoch": 4.0886878967285154e-05,
      "step": 42873,
      "training_step_time": 7.944735527038574
    },
    {
      "epoch": 4.0887832641601565e-05,
      "model_forward_time": 1.7460639476776123,
      "step": 42874
    },
    {
      "epoch": 4.0887832641601565e-05,
      "step": 42874,
      "training_step_time": 8.495140075683594
    },
    {
      "epoch": 4.088878631591797e-05,
      "model_forward_time": 1.7407710552215576,
      "step": 42875
    },
    {
      "epoch": 4.088878631591797e-05,
      "step": 42875,
      "training_step_time": 8.817120552062988
    },
    {
      "epoch": 4.088973999023437e-05,
      "model_forward_time": 1.7384247779846191,
      "step": 42876
    },
    {
      "epoch": 4.088973999023437e-05,
      "step": 42876,
      "training_step_time": 7.794100522994995
    },
    {
      "epoch": 4.0890693664550784e-05,
      "model_forward_time": 1.737668752670288,
      "step": 42877
    },
    {
      "epoch": 4.0890693664550784e-05,
      "step": 42877,
      "training_step_time": 7.302002191543579
    },
    {
      "epoch": 4.089164733886719e-05,
      "model_forward_time": 1.738194227218628,
      "step": 42878
    },
    {
      "epoch": 4.089164733886719e-05,
      "step": 42878,
      "training_step_time": 9.840798139572144
    },
    {
      "epoch": 4.089260101318359e-05,
      "model_forward_time": 2.264474630355835,
      "step": 42879
    },
    {
      "epoch": 4.089260101318359e-05,
      "step": 42879,
      "training_step_time": 7.852507591247559
    },
    {
      "epoch": 4.08935546875e-05,
      "grad_norm": 0.12751397490501404,
      "learning_rate": 6.564100325059812e-06,
      "loss": 0.058,
      "step": 42880
    },
    {
      "epoch": 4.08935546875e-05,
      "model_forward_time": 1.7106361389160156,
      "step": 42880
    },
    {
      "dynamics_loss_avg": 0.06481729559600354,
      "epoch": 4.08935546875e-05,
      "step": 42880
    },
    {
      "action_loss_avg": 0.0015074065115186386,
      "epoch": 4.08935546875e-05,
      "step": 42880
    },
    {
      "epoch": 4.08935546875e-05,
      "step": 42880,
      "training_step_time": 7.89581298828125
    },
    {
      "epoch": 4.0894508361816406e-05,
      "model_forward_time": 1.7079176902770996,
      "step": 42881
    },
    {
      "epoch": 4.0894508361816406e-05,
      "step": 42881,
      "training_step_time": 8.466013669967651
    },
    {
      "epoch": 4.089546203613281e-05,
      "model_forward_time": 1.7717692852020264,
      "step": 42882
    },
    {
      "epoch": 4.089546203613281e-05,
      "step": 42882,
      "training_step_time": 7.991445064544678
    },
    {
      "epoch": 4.089641571044922e-05,
      "model_forward_time": 1.7033607959747314,
      "step": 42883
    },
    {
      "epoch": 4.089641571044922e-05,
      "step": 42883,
      "training_step_time": 8.896567344665527
    },
    {
      "epoch": 4.0897369384765625e-05,
      "model_forward_time": 1.728363275527954,
      "step": 42884
    },
    {
      "epoch": 4.0897369384765625e-05,
      "step": 42884,
      "training_step_time": 8.493027687072754
    },
    {
      "epoch": 4.089832305908203e-05,
      "model_forward_time": 1.7280349731445312,
      "step": 42885
    },
    {
      "epoch": 4.089832305908203e-05,
      "step": 42885,
      "training_step_time": 8.564881801605225
    },
    {
      "epoch": 4.089927673339844e-05,
      "model_forward_time": 1.727816104888916,
      "step": 42886
    },
    {
      "epoch": 4.089927673339844e-05,
      "step": 42886,
      "training_step_time": 7.787816286087036
    },
    {
      "epoch": 4.0900230407714844e-05,
      "model_forward_time": 1.7319624423980713,
      "step": 42887
    },
    {
      "epoch": 4.0900230407714844e-05,
      "step": 42887,
      "training_step_time": 7.833558797836304
    },
    {
      "epoch": 4.090118408203125e-05,
      "model_forward_time": 1.7326593399047852,
      "step": 42888
    },
    {
      "epoch": 4.090118408203125e-05,
      "step": 42888,
      "training_step_time": 8.119343996047974
    },
    {
      "epoch": 4.090213775634766e-05,
      "model_forward_time": 1.7236201763153076,
      "step": 42889
    },
    {
      "epoch": 4.090213775634766e-05,
      "step": 42889,
      "training_step_time": 7.707374334335327
    },
    {
      "epoch": 4.090309143066406e-05,
      "grad_norm": 0.09945815801620483,
      "learning_rate": 6.5625297534185386e-06,
      "loss": 0.0573,
      "step": 42890
    },
    {
      "epoch": 4.090309143066406e-05,
      "model_forward_time": 1.7155005931854248,
      "step": 42890
    },
    {
      "dynamics_loss_avg": 0.06867719292640687,
      "epoch": 4.090309143066406e-05,
      "step": 42890
    },
    {
      "action_loss_avg": 0.002049652679124847,
      "epoch": 4.090309143066406e-05,
      "step": 42890
    },
    {
      "epoch": 4.090309143066406e-05,
      "step": 42890,
      "training_step_time": 8.257567644119263
    },
    {
      "epoch": 4.090404510498047e-05,
      "model_forward_time": 1.7708005905151367,
      "step": 42891
    },
    {
      "epoch": 4.090404510498047e-05,
      "step": 42891,
      "training_step_time": 7.752685308456421
    },
    {
      "epoch": 4.090499877929688e-05,
      "model_forward_time": 1.724149465560913,
      "step": 42892
    },
    {
      "epoch": 4.090499877929688e-05,
      "step": 42892,
      "training_step_time": 9.207695960998535
    },
    {
      "epoch": 4.090595245361328e-05,
      "model_forward_time": 1.7335989475250244,
      "step": 42893
    },
    {
      "epoch": 4.090595245361328e-05,
      "step": 42893,
      "training_step_time": 8.265314102172852
    },
    {
      "epoch": 4.0906906127929686e-05,
      "model_forward_time": 1.7118632793426514,
      "step": 42894
    },
    {
      "epoch": 4.0906906127929686e-05,
      "step": 42894,
      "training_step_time": 7.654057741165161
    },
    {
      "epoch": 4.09078598022461e-05,
      "model_forward_time": 1.7225332260131836,
      "step": 42895
    },
    {
      "epoch": 4.09078598022461e-05,
      "step": 42895,
      "training_step_time": 7.693413734436035
    },
    {
      "epoch": 4.09088134765625e-05,
      "model_forward_time": 1.3453497886657715,
      "step": 42896
    },
    {
      "epoch": 4.09088134765625e-05,
      "step": 42896,
      "training_step_time": 7.756559133529663
    },
    {
      "epoch": 4.0909767150878905e-05,
      "model_forward_time": 1.000368356704712,
      "step": 42897
    },
    {
      "epoch": 4.0909767150878905e-05,
      "step": 42897,
      "training_step_time": 7.63489294052124
    },
    {
      "epoch": 4.0910720825195316e-05,
      "model_forward_time": 2.2702393531799316,
      "step": 42898
    },
    {
      "epoch": 4.0910720825195316e-05,
      "step": 42898,
      "training_step_time": 9.638522624969482
    },
    {
      "epoch": 4.091167449951172e-05,
      "model_forward_time": 1.7236230373382568,
      "step": 42899
    },
    {
      "epoch": 4.091167449951172e-05,
      "step": 42899,
      "training_step_time": 9.220956325531006
    },
    {
      "epoch": 4.0912628173828124e-05,
      "grad_norm": 0.1863565742969513,
      "learning_rate": 6.56095901090136e-06,
      "loss": 0.0547,
      "step": 42900
    },
    {
      "epoch": 4.0912628173828124e-05,
      "model_forward_time": 1.7584054470062256,
      "step": 42900
    },
    {
      "dynamics_loss_avg": 0.05383736351504922,
      "epoch": 4.0912628173828124e-05,
      "step": 42900
    },
    {
      "action_loss_avg": 0.0023026981973089276,
      "epoch": 4.0912628173828124e-05,
      "step": 42900
    },
    {
      "epoch": 4.0912628173828124e-05,
      "step": 42900,
      "training_step_time": 8.501420736312866
    },
    {
      "epoch": 4.0913581848144534e-05,
      "model_forward_time": 1.729158878326416,
      "step": 42901
    },
    {
      "epoch": 4.0913581848144534e-05,
      "step": 42901,
      "training_step_time": 8.512815952301025
    },
    {
      "epoch": 4.091453552246094e-05,
      "model_forward_time": 1.7284862995147705,
      "step": 42902
    },
    {
      "epoch": 4.091453552246094e-05,
      "step": 42902,
      "training_step_time": 7.684539794921875
    },
    {
      "epoch": 4.091548919677734e-05,
      "model_forward_time": 1.7341375350952148,
      "step": 42903
    },
    {
      "epoch": 4.091548919677734e-05,
      "step": 42903,
      "training_step_time": 7.681366443634033
    },
    {
      "epoch": 4.091644287109375e-05,
      "model_forward_time": 1.734302282333374,
      "step": 42904
    },
    {
      "epoch": 4.091644287109375e-05,
      "step": 42904,
      "training_step_time": 7.753129243850708
    },
    {
      "epoch": 4.091739654541016e-05,
      "model_forward_time": 1.7149152755737305,
      "step": 42905
    },
    {
      "epoch": 4.091739654541016e-05,
      "step": 42905,
      "training_step_time": 7.693168878555298
    },
    {
      "epoch": 4.091835021972656e-05,
      "model_forward_time": 1.788379430770874,
      "step": 42906
    },
    {
      "epoch": 4.091835021972656e-05,
      "step": 42906,
      "training_step_time": 8.401590585708618
    },
    {
      "epoch": 4.0919303894042965e-05,
      "model_forward_time": 2.341240167617798,
      "step": 42907
    },
    {
      "epoch": 4.0919303894042965e-05,
      "step": 42907,
      "training_step_time": 7.7623631954193115
    },
    {
      "epoch": 4.0920257568359376e-05,
      "model_forward_time": 1.8273158073425293,
      "step": 42908
    },
    {
      "epoch": 4.0920257568359376e-05,
      "step": 42908,
      "training_step_time": 9.823819160461426
    },
    {
      "epoch": 4.092121124267578e-05,
      "model_forward_time": 1.8284797668457031,
      "step": 42909
    },
    {
      "epoch": 4.092121124267578e-05,
      "step": 42909,
      "training_step_time": 8.424579381942749
    },
    {
      "epoch": 4.0922164916992184e-05,
      "grad_norm": 0.13296261429786682,
      "learning_rate": 6.5593880976800495e-06,
      "loss": 0.0564,
      "step": 42910
    },
    {
      "epoch": 4.0922164916992184e-05,
      "model_forward_time": 1.8451480865478516,
      "step": 42910
    },
    {
      "dynamics_loss_avg": 0.06405097097158433,
      "epoch": 4.0922164916992184e-05,
      "step": 42910
    },
    {
      "action_loss_avg": 0.0022461185697466137,
      "epoch": 4.0922164916992184e-05,
      "step": 42910
    },
    {
      "epoch": 4.0922164916992184e-05,
      "step": 42910,
      "training_step_time": 7.632386922836304
    },
    {
      "epoch": 4.0923118591308595e-05,
      "model_forward_time": 1.8358492851257324,
      "step": 42911
    },
    {
      "epoch": 4.0923118591308595e-05,
      "step": 42911,
      "training_step_time": 7.631085634231567
    },
    {
      "epoch": 4.0924072265625e-05,
      "model_forward_time": 1.8373169898986816,
      "step": 42912
    },
    {
      "epoch": 4.0924072265625e-05,
      "step": 42912,
      "training_step_time": 7.598342180252075
    },
    {
      "epoch": 4.09250259399414e-05,
      "model_forward_time": 1.4351460933685303,
      "step": 42913
    },
    {
      "epoch": 4.09250259399414e-05,
      "step": 42913,
      "training_step_time": 7.703745126724243
    },
    {
      "epoch": 4.0925979614257814e-05,
      "model_forward_time": 1.82468843460083,
      "step": 42914
    },
    {
      "epoch": 4.0925979614257814e-05,
      "step": 42914,
      "training_step_time": 7.726091623306274
    },
    {
      "epoch": 4.092693328857422e-05,
      "model_forward_time": 1.8292920589447021,
      "step": 42915
    },
    {
      "epoch": 4.092693328857422e-05,
      "step": 42915,
      "training_step_time": 7.58843994140625
    },
    {
      "epoch": 4.092788696289062e-05,
      "model_forward_time": 1.8245000839233398,
      "step": 42916
    },
    {
      "epoch": 4.092788696289062e-05,
      "step": 42916,
      "training_step_time": 9.503494024276733
    },
    {
      "epoch": 4.092884063720703e-05,
      "model_forward_time": 1.8114068508148193,
      "step": 42917
    },
    {
      "epoch": 4.092884063720703e-05,
      "step": 42917,
      "training_step_time": 10.24394702911377
    },
    {
      "epoch": 4.092979431152344e-05,
      "model_forward_time": 1.8271028995513916,
      "step": 42918
    },
    {
      "epoch": 4.092979431152344e-05,
      "step": 42918,
      "training_step_time": 9.38744854927063
    },
    {
      "epoch": 4.093074798583984e-05,
      "model_forward_time": 1.8259217739105225,
      "step": 42919
    },
    {
      "epoch": 4.093074798583984e-05,
      "step": 42919,
      "training_step_time": 8.453557252883911
    },
    {
      "epoch": 4.093170166015625e-05,
      "grad_norm": 0.13812629878520966,
      "learning_rate": 6.5578170139264015e-06,
      "loss": 0.0583,
      "step": 42920
    },
    {
      "epoch": 4.093170166015625e-05,
      "model_forward_time": 1.8252544403076172,
      "step": 42920
    },
    {
      "dynamics_loss_avg": 0.07537336461246014,
      "epoch": 4.093170166015625e-05,
      "step": 42920
    },
    {
      "action_loss_avg": 0.0030717013927642257,
      "epoch": 4.093170166015625e-05,
      "step": 42920
    },
    {
      "epoch": 4.093170166015625e-05,
      "step": 42920,
      "training_step_time": 7.718871831893921
    },
    {
      "epoch": 4.0932655334472656e-05,
      "model_forward_time": 1.8210484981536865,
      "step": 42921
    },
    {
      "epoch": 4.0932655334472656e-05,
      "step": 42921,
      "training_step_time": 7.675376892089844
    },
    {
      "epoch": 4.093360900878906e-05,
      "model_forward_time": 1.8300538063049316,
      "step": 42922
    },
    {
      "epoch": 4.093360900878906e-05,
      "step": 42922,
      "training_step_time": 8.35250997543335
    },
    {
      "epoch": 4.093456268310547e-05,
      "model_forward_time": 1.8278634548187256,
      "step": 42923
    },
    {
      "epoch": 4.093456268310547e-05,
      "step": 42923,
      "training_step_time": 7.2162697315216064
    },
    {
      "epoch": 4.0935516357421875e-05,
      "model_forward_time": 1.8253870010375977,
      "step": 42924
    },
    {
      "epoch": 4.0935516357421875e-05,
      "step": 42924,
      "training_step_time": 8.469626665115356
    },
    {
      "epoch": 4.093647003173828e-05,
      "model_forward_time": 1.8266475200653076,
      "step": 42925
    },
    {
      "epoch": 4.093647003173828e-05,
      "step": 42925,
      "training_step_time": 7.720838308334351
    },
    {
      "epoch": 4.093742370605469e-05,
      "model_forward_time": 2.3579800128936768,
      "step": 42926
    },
    {
      "epoch": 4.093742370605469e-05,
      "step": 42926,
      "training_step_time": 8.597533464431763
    },
    {
      "epoch": 4.093837738037109e-05,
      "model_forward_time": 1.8406527042388916,
      "step": 42927
    },
    {
      "epoch": 4.093837738037109e-05,
      "step": 42927,
      "training_step_time": 8.496955394744873
    },
    {
      "epoch": 4.09393310546875e-05,
      "model_forward_time": 1.8469016551971436,
      "step": 42928
    },
    {
      "epoch": 4.09393310546875e-05,
      "step": 42928,
      "training_step_time": 7.842945098876953
    },
    {
      "epoch": 4.094028472900391e-05,
      "model_forward_time": 1.8396356105804443,
      "step": 42929
    },
    {
      "epoch": 4.094028472900391e-05,
      "step": 42929,
      "training_step_time": 7.705726385116577
    },
    {
      "epoch": 4.094123840332031e-05,
      "grad_norm": 0.12790247797966003,
      "learning_rate": 6.556245759812227e-06,
      "loss": 0.0621,
      "step": 42930
    },
    {
      "epoch": 4.094123840332031e-05,
      "model_forward_time": 1.8383269309997559,
      "step": 42930
    },
    {
      "dynamics_loss_avg": 0.06371803116053343,
      "epoch": 4.094123840332031e-05,
      "step": 42930
    },
    {
      "action_loss_avg": 0.0025471152388490737,
      "epoch": 4.094123840332031e-05,
      "step": 42930
    },
    {
      "epoch": 4.094123840332031e-05,
      "step": 42930,
      "training_step_time": 8.618724584579468
    },
    {
      "epoch": 4.0942192077636716e-05,
      "model_forward_time": 1.8349220752716064,
      "step": 42931
    },
    {
      "epoch": 4.0942192077636716e-05,
      "step": 42931,
      "training_step_time": 7.703453302383423
    },
    {
      "epoch": 4.094314575195313e-05,
      "model_forward_time": 1.8496673107147217,
      "step": 42932
    },
    {
      "epoch": 4.094314575195313e-05,
      "step": 42932,
      "training_step_time": 7.490811824798584
    },
    {
      "epoch": 4.094409942626953e-05,
      "model_forward_time": 1.8347384929656982,
      "step": 42933
    },
    {
      "epoch": 4.094409942626953e-05,
      "step": 42933,
      "training_step_time": 7.723650693893433
    },
    {
      "epoch": 4.0945053100585935e-05,
      "model_forward_time": 1.847823143005371,
      "step": 42934
    },
    {
      "epoch": 4.0945053100585935e-05,
      "step": 42934,
      "training_step_time": 8.528459072113037
    },
    {
      "epoch": 4.0946006774902346e-05,
      "model_forward_time": 2.386518955230713,
      "step": 42935
    },
    {
      "epoch": 4.0946006774902346e-05,
      "step": 42935,
      "training_step_time": 7.687984466552734
    },
    {
      "epoch": 4.094696044921875e-05,
      "model_forward_time": 1.8282361030578613,
      "step": 42936
    },
    {
      "epoch": 4.094696044921875e-05,
      "step": 42936,
      "training_step_time": 7.776175022125244
    },
    {
      "epoch": 4.0947914123535154e-05,
      "model_forward_time": 1.8466641902923584,
      "step": 42937
    },
    {
      "epoch": 4.0947914123535154e-05,
      "step": 42937,
      "training_step_time": 7.769052028656006
    },
    {
      "epoch": 4.0948867797851565e-05,
      "model_forward_time": 1.7997472286224365,
      "step": 42938
    },
    {
      "epoch": 4.0948867797851565e-05,
      "step": 42938,
      "training_step_time": 8.260259628295898
    },
    {
      "epoch": 4.094982147216797e-05,
      "model_forward_time": 1.8222379684448242,
      "step": 42939
    },
    {
      "epoch": 4.094982147216797e-05,
      "step": 42939,
      "training_step_time": 8.438540935516357
    },
    {
      "epoch": 4.095077514648437e-05,
      "grad_norm": 0.10251598805189133,
      "learning_rate": 6.554674335509358e-06,
      "loss": 0.0583,
      "step": 42940
    },
    {
      "epoch": 4.095077514648437e-05,
      "model_forward_time": 1.8302180767059326,
      "step": 42940
    },
    {
      "dynamics_loss_avg": 0.07295786067843438,
      "epoch": 4.095077514648437e-05,
      "step": 42940
    },
    {
      "action_loss_avg": 0.003034800209570676,
      "epoch": 4.095077514648437e-05,
      "step": 42940
    },
    {
      "epoch": 4.095077514648437e-05,
      "step": 42940,
      "training_step_time": 7.66321587562561
    },
    {
      "epoch": 4.0951728820800784e-05,
      "model_forward_time": 1.8225867748260498,
      "step": 42941
    },
    {
      "epoch": 4.0951728820800784e-05,
      "step": 42941,
      "training_step_time": 7.594688415527344
    },
    {
      "epoch": 4.095268249511719e-05,
      "model_forward_time": 1.8299269676208496,
      "step": 42942
    },
    {
      "epoch": 4.095268249511719e-05,
      "step": 42942,
      "training_step_time": 8.617235660552979
    },
    {
      "epoch": 4.095363616943359e-05,
      "model_forward_time": 1.8301947116851807,
      "step": 42943
    },
    {
      "epoch": 4.095363616943359e-05,
      "step": 42943,
      "training_step_time": 8.777098894119263
    },
    {
      "epoch": 4.095458984375e-05,
      "model_forward_time": 2.3406529426574707,
      "step": 42944
    },
    {
      "epoch": 4.095458984375e-05,
      "step": 42944,
      "training_step_time": 8.64951467514038
    },
    {
      "epoch": 4.0955543518066407e-05,
      "model_forward_time": 1.8382034301757812,
      "step": 42945
    },
    {
      "epoch": 4.0955543518066407e-05,
      "step": 42945,
      "training_step_time": 8.641306161880493
    },
    {
      "epoch": 4.095649719238281e-05,
      "model_forward_time": 1.8353300094604492,
      "step": 42946
    },
    {
      "epoch": 4.095649719238281e-05,
      "step": 42946,
      "training_step_time": 9.177159547805786
    },
    {
      "epoch": 4.095745086669922e-05,
      "model_forward_time": 1.814164400100708,
      "step": 42947
    },
    {
      "epoch": 4.095745086669922e-05,
      "step": 42947,
      "training_step_time": 7.647444009780884
    },
    {
      "epoch": 4.0958404541015625e-05,
      "model_forward_time": 1.82676100730896,
      "step": 42948
    },
    {
      "epoch": 4.0958404541015625e-05,
      "step": 42948,
      "training_step_time": 7.723280668258667
    },
    {
      "epoch": 4.095935821533203e-05,
      "model_forward_time": 1.8289158344268799,
      "step": 42949
    },
    {
      "epoch": 4.095935821533203e-05,
      "step": 42949,
      "training_step_time": 7.67455530166626
    },
    {
      "epoch": 4.096031188964844e-05,
      "grad_norm": 0.14209836721420288,
      "learning_rate": 6.5531027411896385e-06,
      "loss": 0.0562,
      "step": 42950
    },
    {
      "epoch": 4.096031188964844e-05,
      "model_forward_time": 1.8263967037200928,
      "step": 42950
    },
    {
      "dynamics_loss_avg": 0.05318892300128937,
      "epoch": 4.096031188964844e-05,
      "step": 42950
    },
    {
      "action_loss_avg": 0.002421937062172219,
      "epoch": 4.096031188964844e-05,
      "step": 42950
    },
    {
      "epoch": 4.096031188964844e-05,
      "step": 42950,
      "training_step_time": 8.2319655418396
    },
    {
      "epoch": 4.0961265563964844e-05,
      "model_forward_time": 1.8173022270202637,
      "step": 42951
    },
    {
      "epoch": 4.0961265563964844e-05,
      "step": 42951,
      "training_step_time": 8.319480657577515
    },
    {
      "epoch": 4.096221923828125e-05,
      "model_forward_time": 1.8289153575897217,
      "step": 42952
    },
    {
      "epoch": 4.096221923828125e-05,
      "step": 42952,
      "training_step_time": 8.247319459915161
    },
    {
      "epoch": 4.096317291259766e-05,
      "model_forward_time": 2.327007532119751,
      "step": 42953
    },
    {
      "epoch": 4.096317291259766e-05,
      "step": 42953,
      "training_step_time": 8.187128067016602
    },
    {
      "epoch": 4.096412658691406e-05,
      "model_forward_time": 1.8232810497283936,
      "step": 42954
    },
    {
      "epoch": 4.096412658691406e-05,
      "step": 42954,
      "training_step_time": 9.09522008895874
    },
    {
      "epoch": 4.096508026123047e-05,
      "model_forward_time": 1.8331959247589111,
      "step": 42955
    },
    {
      "epoch": 4.096508026123047e-05,
      "step": 42955,
      "training_step_time": 7.805365562438965
    },
    {
      "epoch": 4.096603393554688e-05,
      "model_forward_time": 1.8244225978851318,
      "step": 42956
    },
    {
      "epoch": 4.096603393554688e-05,
      "step": 42956,
      "training_step_time": 8.837926864624023
    },
    {
      "epoch": 4.096698760986328e-05,
      "model_forward_time": 1.81260347366333,
      "step": 42957
    },
    {
      "epoch": 4.096698760986328e-05,
      "step": 42957,
      "training_step_time": 7.626077651977539
    },
    {
      "epoch": 4.0967941284179686e-05,
      "model_forward_time": 1.846322774887085,
      "step": 42958
    },
    {
      "epoch": 4.0967941284179686e-05,
      "step": 42958,
      "training_step_time": 7.649046421051025
    },
    {
      "epoch": 4.09688949584961e-05,
      "model_forward_time": 1.8388822078704834,
      "step": 42959
    },
    {
      "epoch": 4.09688949584961e-05,
      "step": 42959,
      "training_step_time": 7.701993703842163
    },
    {
      "epoch": 4.09698486328125e-05,
      "grad_norm": 0.24196521937847137,
      "learning_rate": 6.55153097702494e-06,
      "loss": 0.0545,
      "step": 42960
    },
    {
      "epoch": 4.09698486328125e-05,
      "model_forward_time": 1.850898027420044,
      "step": 42960
    },
    {
      "dynamics_loss_avg": 0.06679667774587869,
      "epoch": 4.09698486328125e-05,
      "step": 42960
    },
    {
      "action_loss_avg": 0.0020641196839278563,
      "epoch": 4.09698486328125e-05,
      "step": 42960
    },
    {
      "epoch": 4.09698486328125e-05,
      "step": 42960,
      "training_step_time": 7.660972356796265
    },
    {
      "epoch": 4.0970802307128905e-05,
      "model_forward_time": 1.8471345901489258,
      "step": 42961
    },
    {
      "epoch": 4.0970802307128905e-05,
      "step": 42961,
      "training_step_time": 7.733317136764526
    },
    {
      "epoch": 4.0971755981445316e-05,
      "model_forward_time": 2.4225053787231445,
      "step": 42962
    },
    {
      "epoch": 4.0971755981445316e-05,
      "step": 42962,
      "training_step_time": 8.227989912033081
    },
    {
      "epoch": 4.097270965576172e-05,
      "model_forward_time": 1.8333451747894287,
      "step": 42963
    },
    {
      "epoch": 4.097270965576172e-05,
      "step": 42963,
      "training_step_time": 9.40984058380127
    },
    {
      "epoch": 4.0973663330078124e-05,
      "model_forward_time": 1.836259365081787,
      "step": 42964
    },
    {
      "epoch": 4.0973663330078124e-05,
      "step": 42964,
      "training_step_time": 7.75649881362915
    },
    {
      "epoch": 4.0974617004394535e-05,
      "model_forward_time": 1.8331241607666016,
      "step": 42965
    },
    {
      "epoch": 4.0974617004394535e-05,
      "step": 42965,
      "training_step_time": 9.335347175598145
    },
    {
      "epoch": 4.097557067871094e-05,
      "model_forward_time": 1.829425573348999,
      "step": 42966
    },
    {
      "epoch": 4.097557067871094e-05,
      "step": 42966,
      "training_step_time": 8.755870342254639
    },
    {
      "epoch": 4.097652435302734e-05,
      "model_forward_time": 1.8263733386993408,
      "step": 42967
    },
    {
      "epoch": 4.097652435302734e-05,
      "step": 42967,
      "training_step_time": 8.647542715072632
    },
    {
      "epoch": 4.097747802734375e-05,
      "model_forward_time": 1.837500810623169,
      "step": 42968
    },
    {
      "epoch": 4.097747802734375e-05,
      "step": 42968,
      "training_step_time": 7.808884382247925
    },
    {
      "epoch": 4.097843170166016e-05,
      "model_forward_time": 1.8328537940979004,
      "step": 42969
    },
    {
      "epoch": 4.097843170166016e-05,
      "step": 42969,
      "training_step_time": 7.716219902038574
    },
    {
      "epoch": 4.097938537597656e-05,
      "grad_norm": 0.1760004162788391,
      "learning_rate": 6.549959043187147e-06,
      "loss": 0.0574,
      "step": 42970
    },
    {
      "epoch": 4.097938537597656e-05,
      "model_forward_time": 1.829479455947876,
      "step": 42970
    },
    {
      "dynamics_loss_avg": 0.06149625405669212,
      "epoch": 4.097938537597656e-05,
      "step": 42970
    },
    {
      "action_loss_avg": 0.0018451403651852162,
      "epoch": 4.097938537597656e-05,
      "step": 42970
    },
    {
      "epoch": 4.097938537597656e-05,
      "step": 42970,
      "training_step_time": 7.6505126953125
    },
    {
      "epoch": 4.0980339050292965e-05,
      "model_forward_time": 1.8309605121612549,
      "step": 42971
    },
    {
      "epoch": 4.0980339050292965e-05,
      "step": 42971,
      "training_step_time": 8.731087923049927
    },
    {
      "epoch": 4.0981292724609376e-05,
      "model_forward_time": 1.8318254947662354,
      "step": 42972
    },
    {
      "epoch": 4.0981292724609376e-05,
      "step": 42972,
      "training_step_time": 7.729025363922119
    },
    {
      "epoch": 4.098224639892578e-05,
      "model_forward_time": 1.8378260135650635,
      "step": 42973
    },
    {
      "epoch": 4.098224639892578e-05,
      "step": 42973,
      "training_step_time": 8.49479341506958
    },
    {
      "epoch": 4.0983200073242184e-05,
      "model_forward_time": 1.8362929821014404,
      "step": 42974
    },
    {
      "epoch": 4.0983200073242184e-05,
      "step": 42974,
      "training_step_time": 7.8017661571502686
    },
    {
      "epoch": 4.0984153747558595e-05,
      "model_forward_time": 1.8345441818237305,
      "step": 42975
    },
    {
      "epoch": 4.0984153747558595e-05,
      "step": 42975,
      "training_step_time": 7.760790586471558
    },
    {
      "epoch": 4.0985107421875e-05,
      "model_forward_time": 1.8297226428985596,
      "step": 42976
    },
    {
      "epoch": 4.0985107421875e-05,
      "step": 42976,
      "training_step_time": 7.541731595993042
    },
    {
      "epoch": 4.09860610961914e-05,
      "model_forward_time": 1.8474195003509521,
      "step": 42977
    },
    {
      "epoch": 4.09860610961914e-05,
      "step": 42977,
      "training_step_time": 7.8025963306427
    },
    {
      "epoch": 4.0987014770507814e-05,
      "model_forward_time": 1.8413515090942383,
      "step": 42978
    },
    {
      "epoch": 4.0987014770507814e-05,
      "step": 42978,
      "training_step_time": 8.246172666549683
    },
    {
      "epoch": 4.098796844482422e-05,
      "model_forward_time": 1.8330585956573486,
      "step": 42979
    },
    {
      "epoch": 4.098796844482422e-05,
      "step": 42979,
      "training_step_time": 7.570643424987793
    },
    {
      "epoch": 4.098892211914062e-05,
      "grad_norm": 0.10448130965232849,
      "learning_rate": 6.54838693984816e-06,
      "loss": 0.0564,
      "step": 42980
    },
    {
      "epoch": 4.098892211914062e-05,
      "model_forward_time": 1.8293044567108154,
      "step": 42980
    },
    {
      "dynamics_loss_avg": 0.042474212031811474,
      "epoch": 4.098892211914062e-05,
      "step": 42980
    },
    {
      "action_loss_avg": 0.001553411068744026,
      "epoch": 4.098892211914062e-05,
      "step": 42980
    },
    {
      "epoch": 4.098892211914062e-05,
      "step": 42980,
      "training_step_time": 7.781925439834595
    },
    {
      "epoch": 4.098987579345703e-05,
      "model_forward_time": 2.388476848602295,
      "step": 42981
    },
    {
      "epoch": 4.098987579345703e-05,
      "step": 42981,
      "training_step_time": 8.88683271408081
    },
    {
      "epoch": 4.099082946777344e-05,
      "model_forward_time": 1.8382375240325928,
      "step": 42982
    },
    {
      "epoch": 4.099082946777344e-05,
      "step": 42982,
      "training_step_time": 8.304844856262207
    },
    {
      "epoch": 4.099178314208984e-05,
      "model_forward_time": 1.844505786895752,
      "step": 42983
    },
    {
      "epoch": 4.099178314208984e-05,
      "step": 42983,
      "training_step_time": 7.7933690547943115
    },
    {
      "epoch": 4.099273681640625e-05,
      "model_forward_time": 1.8283607959747314,
      "step": 42984
    },
    {
      "epoch": 4.099273681640625e-05,
      "step": 42984,
      "training_step_time": 7.7192606925964355
    },
    {
      "epoch": 4.0993690490722656e-05,
      "model_forward_time": 1.8319299221038818,
      "step": 42985
    },
    {
      "epoch": 4.0993690490722656e-05,
      "step": 42985,
      "training_step_time": 7.5522990226745605
    },
    {
      "epoch": 4.099464416503906e-05,
      "model_forward_time": 1.8312158584594727,
      "step": 42986
    },
    {
      "epoch": 4.099464416503906e-05,
      "step": 42986,
      "training_step_time": 7.697717666625977
    },
    {
      "epoch": 4.099559783935547e-05,
      "model_forward_time": 1.827632188796997,
      "step": 42987
    },
    {
      "epoch": 4.099559783935547e-05,
      "step": 42987,
      "training_step_time": 7.752713203430176
    },
    {
      "epoch": 4.0996551513671875e-05,
      "model_forward_time": 1.8352460861206055,
      "step": 42988
    },
    {
      "epoch": 4.0996551513671875e-05,
      "step": 42988,
      "training_step_time": 7.714142799377441
    },
    {
      "epoch": 4.099750518798828e-05,
      "model_forward_time": 1.829434871673584,
      "step": 42989
    },
    {
      "epoch": 4.099750518798828e-05,
      "step": 42989,
      "training_step_time": 7.745153188705444
    },
    {
      "epoch": 4.099845886230469e-05,
      "grad_norm": 0.1061679795384407,
      "learning_rate": 6.546814667179909e-06,
      "loss": 0.0562,
      "step": 42990
    },
    {
      "epoch": 4.099845886230469e-05,
      "model_forward_time": 2.358851671218872,
      "step": 42990
    },
    {
      "dynamics_loss_avg": 0.04710626881569624,
      "epoch": 4.099845886230469e-05,
      "step": 42990
    },
    {
      "action_loss_avg": 0.0019118491210974753,
      "epoch": 4.099845886230469e-05,
      "step": 42990
    },
    {
      "epoch": 4.099845886230469e-05,
      "step": 42990,
      "training_step_time": 9.590012073516846
    },
    {
      "epoch": 4.0999412536621093e-05,
      "model_forward_time": 1.8411786556243896,
      "step": 42991
    },
    {
      "epoch": 4.0999412536621093e-05,
      "step": 42991,
      "training_step_time": 7.683063745498657
    },
    {
      "epoch": 4.10003662109375e-05,
      "model_forward_time": 1.8465113639831543,
      "step": 42992
    },
    {
      "epoch": 4.10003662109375e-05,
      "step": 42992,
      "training_step_time": 9.25960397720337
    },
    {
      "epoch": 4.100131988525391e-05,
      "model_forward_time": 1.8229074478149414,
      "step": 42993
    },
    {
      "epoch": 4.100131988525391e-05,
      "step": 42993,
      "training_step_time": 7.68697190284729
    },
    {
      "epoch": 4.100227355957031e-05,
      "model_forward_time": 1.8367300033569336,
      "step": 42994
    },
    {
      "epoch": 4.100227355957031e-05,
      "step": 42994,
      "training_step_time": 7.664591550827026
    },
    {
      "epoch": 4.1003227233886716e-05,
      "model_forward_time": 1.8429045677185059,
      "step": 42995
    },
    {
      "epoch": 4.1003227233886716e-05,
      "step": 42995,
      "training_step_time": 8.159245252609253
    },
    {
      "epoch": 4.100418090820313e-05,
      "model_forward_time": 1.8389780521392822,
      "step": 42996
    },
    {
      "epoch": 4.100418090820313e-05,
      "step": 42996,
      "training_step_time": 9.386296272277832
    },
    {
      "epoch": 4.100513458251953e-05,
      "model_forward_time": 1.8431270122528076,
      "step": 42997
    },
    {
      "epoch": 4.100513458251953e-05,
      "step": 42997,
      "training_step_time": 8.672555685043335
    },
    {
      "epoch": 4.1006088256835935e-05,
      "model_forward_time": 1.84163498878479,
      "step": 42998
    },
    {
      "epoch": 4.1006088256835935e-05,
      "step": 42998,
      "training_step_time": 8.813716411590576
    },
    {
      "epoch": 4.1007041931152346e-05,
      "model_forward_time": 2.3666906356811523,
      "step": 42999
    },
    {
      "epoch": 4.1007041931152346e-05,
      "step": 42999,
      "training_step_time": 8.31071949005127
    },
    {
      "epoch": 4.100799560546875e-05,
      "grad_norm": 0.14436347782611847,
      "learning_rate": 6.545242225354329e-06,
      "loss": 0.0568,
      "step": 43000
    },
    {
      "epoch": 4.100799560546875e-05,
      "model_forward_time": 1.7734596729278564,
      "step": 43000
    },
    {
      "dynamics_loss_avg": 0.0525006540119648,
      "epoch": 4.100799560546875e-05,
      "step": 43000
    },
    {
      "action_loss_avg": 0.0018223434337414802,
      "epoch": 4.100799560546875e-05,
      "step": 43000
    },
    {
      "epoch": 4.100799560546875e-05,
      "step": 43000,
      "training_step_time": 10.245773315429688
    },
    {
      "epoch": 4.1008949279785154e-05,
      "model_forward_time": 1.7444567680358887,
      "step": 43001
    },
    {
      "epoch": 4.1008949279785154e-05,
      "step": 43001,
      "training_step_time": 10.080465078353882
    },
    {
      "epoch": 4.1009902954101565e-05,
      "model_forward_time": 1.7430109977722168,
      "step": 43002
    },
    {
      "epoch": 4.1009902954101565e-05,
      "step": 43002,
      "training_step_time": 8.532750368118286
    },
    {
      "epoch": 4.101085662841797e-05,
      "model_forward_time": 1.7552330493927002,
      "step": 43003
    },
    {
      "epoch": 4.101085662841797e-05,
      "step": 43003,
      "training_step_time": 8.173344373703003
    },
    {
      "epoch": 4.101181030273437e-05,
      "model_forward_time": 1.747546672821045,
      "step": 43004
    },
    {
      "epoch": 4.101181030273437e-05,
      "step": 43004,
      "training_step_time": 7.582693576812744
    },
    {
      "epoch": 4.1012763977050784e-05,
      "model_forward_time": 1.752293586730957,
      "step": 43005
    },
    {
      "epoch": 4.1012763977050784e-05,
      "step": 43005,
      "training_step_time": 8.393815994262695
    },
    {
      "epoch": 4.101371765136719e-05,
      "model_forward_time": 1.7529628276824951,
      "step": 43006
    },
    {
      "epoch": 4.101371765136719e-05,
      "step": 43006,
      "training_step_time": 8.463077068328857
    },
    {
      "epoch": 4.101467132568359e-05,
      "model_forward_time": 1.7130258083343506,
      "step": 43007
    },
    {
      "epoch": 4.101467132568359e-05,
      "step": 43007,
      "training_step_time": 8.44762897491455
    },
    {
      "epoch": 4.1015625e-05,
      "model_forward_time": 2.835597276687622,
      "step": 43008
    },
    {
      "epoch": 4.1015625e-05,
      "step": 43008,
      "training_step_time": 10.542725801467896
    },
    {
      "epoch": 4.1016578674316407e-05,
      "model_forward_time": 1.7209346294403076,
      "step": 43009
    },
    {
      "epoch": 4.1016578674316407e-05,
      "step": 43009,
      "training_step_time": 7.579749345779419
    },
    {
      "epoch": 4.101753234863281e-05,
      "grad_norm": 0.1596921980381012,
      "learning_rate": 6.543669614543383e-06,
      "loss": 0.0581,
      "step": 43010
    },
    {
      "epoch": 4.101753234863281e-05,
      "model_forward_time": 1.7239248752593994,
      "step": 43010
    },
    {
      "dynamics_loss_avg": 0.06728204414248466,
      "epoch": 4.101753234863281e-05,
      "step": 43010
    },
    {
      "action_loss_avg": 0.0029414641408948227,
      "epoch": 4.101753234863281e-05,
      "step": 43010
    },
    {
      "epoch": 4.101753234863281e-05,
      "step": 43010,
      "training_step_time": 7.719608545303345
    },
    {
      "epoch": 4.101848602294922e-05,
      "model_forward_time": 1.7155237197875977,
      "step": 43011
    },
    {
      "epoch": 4.101848602294922e-05,
      "step": 43011,
      "training_step_time": 7.749502182006836
    },
    {
      "epoch": 4.1019439697265625e-05,
      "model_forward_time": 1.72271728515625,
      "step": 43012
    },
    {
      "epoch": 4.1019439697265625e-05,
      "step": 43012,
      "training_step_time": 8.363639116287231
    },
    {
      "epoch": 4.102039337158203e-05,
      "model_forward_time": 1.7366759777069092,
      "step": 43013
    },
    {
      "epoch": 4.102039337158203e-05,
      "step": 43013,
      "training_step_time": 7.670183420181274
    },
    {
      "epoch": 4.102134704589844e-05,
      "model_forward_time": 1.7220501899719238,
      "step": 43014
    },
    {
      "epoch": 4.102134704589844e-05,
      "step": 43014,
      "training_step_time": 7.763543605804443
    },
    {
      "epoch": 4.1022300720214844e-05,
      "model_forward_time": 1.7488892078399658,
      "step": 43015
    },
    {
      "epoch": 4.1022300720214844e-05,
      "step": 43015,
      "training_step_time": 7.7298924922943115
    },
    {
      "epoch": 4.102325439453125e-05,
      "model_forward_time": 1.7447404861450195,
      "step": 43016
    },
    {
      "epoch": 4.102325439453125e-05,
      "step": 43016,
      "training_step_time": 7.654099464416504
    },
    {
      "epoch": 4.102420806884766e-05,
      "model_forward_time": 2.300426959991455,
      "step": 43017
    },
    {
      "epoch": 4.102420806884766e-05,
      "step": 43017,
      "training_step_time": 7.731248140335083
    },
    {
      "epoch": 4.102516174316406e-05,
      "model_forward_time": 1.8061978816986084,
      "step": 43018
    },
    {
      "epoch": 4.102516174316406e-05,
      "step": 43018,
      "training_step_time": 7.698362350463867
    },
    {
      "epoch": 4.102611541748047e-05,
      "model_forward_time": 1.7448351383209229,
      "step": 43019
    },
    {
      "epoch": 4.102611541748047e-05,
      "step": 43019,
      "training_step_time": 9.03712773323059
    },
    {
      "epoch": 4.102706909179688e-05,
      "grad_norm": 0.1192636713385582,
      "learning_rate": 6.542096834919049e-06,
      "loss": 0.0559,
      "step": 43020
    },
    {
      "epoch": 4.102706909179688e-05,
      "model_forward_time": 1.7313060760498047,
      "step": 43020
    },
    {
      "dynamics_loss_avg": 0.051810268126428125,
      "epoch": 4.102706909179688e-05,
      "step": 43020
    },
    {
      "action_loss_avg": 0.0008812324027530849,
      "epoch": 4.102706909179688e-05,
      "step": 43020
    },
    {
      "epoch": 4.102706909179688e-05,
      "step": 43020,
      "training_step_time": 7.671369791030884
    },
    {
      "epoch": 4.102802276611328e-05,
      "model_forward_time": 1.7499058246612549,
      "step": 43021
    },
    {
      "epoch": 4.102802276611328e-05,
      "step": 43021,
      "training_step_time": 7.641328573226929
    },
    {
      "epoch": 4.1028976440429686e-05,
      "model_forward_time": 1.73801589012146,
      "step": 43022
    },
    {
      "epoch": 4.1028976440429686e-05,
      "step": 43022,
      "training_step_time": 7.621393203735352
    },
    {
      "epoch": 4.10299301147461e-05,
      "model_forward_time": 1.7420322895050049,
      "step": 43023
    },
    {
      "epoch": 4.10299301147461e-05,
      "step": 43023,
      "training_step_time": 7.6796934604644775
    },
    {
      "epoch": 4.10308837890625e-05,
      "model_forward_time": 1.7539002895355225,
      "step": 43024
    },
    {
      "epoch": 4.10308837890625e-05,
      "step": 43024,
      "training_step_time": 7.6600258350372314
    },
    {
      "epoch": 4.1031837463378905e-05,
      "model_forward_time": 1.7419147491455078,
      "step": 43025
    },
    {
      "epoch": 4.1031837463378905e-05,
      "step": 43025,
      "training_step_time": 7.540642499923706
    },
    {
      "epoch": 4.1032791137695316e-05,
      "model_forward_time": 2.2975926399230957,
      "step": 43026
    },
    {
      "epoch": 4.1032791137695316e-05,
      "step": 43026,
      "training_step_time": 8.301786184310913
    },
    {
      "epoch": 4.103374481201172e-05,
      "model_forward_time": 1.7953119277954102,
      "step": 43027
    },
    {
      "epoch": 4.103374481201172e-05,
      "step": 43027,
      "training_step_time": 7.709489583969116
    },
    {
      "epoch": 4.1034698486328124e-05,
      "model_forward_time": 1.740614414215088,
      "step": 43028
    },
    {
      "epoch": 4.1034698486328124e-05,
      "step": 43028,
      "training_step_time": 7.71714973449707
    },
    {
      "epoch": 4.1035652160644535e-05,
      "model_forward_time": 1.757450819015503,
      "step": 43029
    },
    {
      "epoch": 4.1035652160644535e-05,
      "step": 43029,
      "training_step_time": 7.728130102157593
    },
    {
      "epoch": 4.103660583496094e-05,
      "grad_norm": 0.11426801979541779,
      "learning_rate": 6.540523886653325e-06,
      "loss": 0.0582,
      "step": 43030
    },
    {
      "epoch": 4.103660583496094e-05,
      "model_forward_time": 1.733468770980835,
      "step": 43030
    },
    {
      "dynamics_loss_avg": 0.052160870470106604,
      "epoch": 4.103660583496094e-05,
      "step": 43030
    },
    {
      "action_loss_avg": 0.001737578841857612,
      "epoch": 4.103660583496094e-05,
      "step": 43030
    },
    {
      "epoch": 4.103660583496094e-05,
      "step": 43030,
      "training_step_time": 7.697016954421997
    },
    {
      "epoch": 4.103755950927734e-05,
      "model_forward_time": 1.7415831089019775,
      "step": 43031
    },
    {
      "epoch": 4.103755950927734e-05,
      "step": 43031,
      "training_step_time": 7.716133117675781
    },
    {
      "epoch": 4.103851318359375e-05,
      "model_forward_time": 1.7371406555175781,
      "step": 43032
    },
    {
      "epoch": 4.103851318359375e-05,
      "step": 43032,
      "training_step_time": 7.790481090545654
    },
    {
      "epoch": 4.103946685791016e-05,
      "model_forward_time": 1.7415738105773926,
      "step": 43033
    },
    {
      "epoch": 4.103946685791016e-05,
      "step": 43033,
      "training_step_time": 8.355238914489746
    },
    {
      "epoch": 4.104042053222656e-05,
      "model_forward_time": 1.7354059219360352,
      "step": 43034
    },
    {
      "epoch": 4.104042053222656e-05,
      "step": 43034,
      "training_step_time": 8.668320655822754
    },
    {
      "epoch": 4.1041374206542965e-05,
      "model_forward_time": 2.321326971054077,
      "step": 43035
    },
    {
      "epoch": 4.1041374206542965e-05,
      "step": 43035,
      "training_step_time": 8.89010763168335
    },
    {
      "epoch": 4.1042327880859376e-05,
      "model_forward_time": 1.743856430053711,
      "step": 43036
    },
    {
      "epoch": 4.1042327880859376e-05,
      "step": 43036,
      "training_step_time": 7.778721332550049
    },
    {
      "epoch": 4.104328155517578e-05,
      "model_forward_time": 1.7409250736236572,
      "step": 43037
    },
    {
      "epoch": 4.104328155517578e-05,
      "step": 43037,
      "training_step_time": 7.82742977142334
    },
    {
      "epoch": 4.1044235229492184e-05,
      "model_forward_time": 1.756716251373291,
      "step": 43038
    },
    {
      "epoch": 4.1044235229492184e-05,
      "step": 43038,
      "training_step_time": 7.824173927307129
    },
    {
      "epoch": 4.1045188903808595e-05,
      "model_forward_time": 1.7353198528289795,
      "step": 43039
    },
    {
      "epoch": 4.1045188903808595e-05,
      "step": 43039,
      "training_step_time": 9.119087219238281
    },
    {
      "epoch": 4.1046142578125e-05,
      "grad_norm": 0.11875506490468979,
      "learning_rate": 6.538950769918223e-06,
      "loss": 0.0579,
      "step": 43040
    },
    {
      "epoch": 4.1046142578125e-05,
      "model_forward_time": 1.7326090335845947,
      "step": 43040
    },
    {
      "dynamics_loss_avg": 0.043833835422992705,
      "epoch": 4.1046142578125e-05,
      "step": 43040
    },
    {
      "action_loss_avg": 0.0015031859249575064,
      "epoch": 4.1046142578125e-05,
      "step": 43040
    },
    {
      "epoch": 4.1046142578125e-05,
      "step": 43040,
      "training_step_time": 7.616166591644287
    },
    {
      "epoch": 4.10470962524414e-05,
      "model_forward_time": 1.7434546947479248,
      "step": 43041
    },
    {
      "epoch": 4.10470962524414e-05,
      "step": 43041,
      "training_step_time": 8.730254888534546
    },
    {
      "epoch": 4.1048049926757814e-05,
      "model_forward_time": 1.7484345436096191,
      "step": 43042
    },
    {
      "epoch": 4.1048049926757814e-05,
      "step": 43042,
      "training_step_time": 9.573451519012451
    },
    {
      "epoch": 4.104900360107422e-05,
      "model_forward_time": 1.7397522926330566,
      "step": 43043
    },
    {
      "epoch": 4.104900360107422e-05,
      "step": 43043,
      "training_step_time": 6.822785139083862
    },
    {
      "epoch": 4.104995727539062e-05,
      "model_forward_time": 1.7381913661956787,
      "step": 43044
    },
    {
      "epoch": 4.104995727539062e-05,
      "step": 43044,
      "training_step_time": 6.799768686294556
    },
    {
      "epoch": 4.105091094970703e-05,
      "model_forward_time": 1.7465994358062744,
      "step": 43045
    },
    {
      "epoch": 4.105091094970703e-05,
      "step": 43045,
      "training_step_time": 6.735572338104248
    },
    {
      "epoch": 4.105186462402344e-05,
      "model_forward_time": 1.744149923324585,
      "step": 43046
    },
    {
      "epoch": 4.105186462402344e-05,
      "step": 43046,
      "training_step_time": 6.810121059417725
    },
    {
      "epoch": 4.105281829833984e-05,
      "model_forward_time": 1.742445707321167,
      "step": 43047
    },
    {
      "epoch": 4.105281829833984e-05,
      "step": 43047,
      "training_step_time": 6.8424763679504395
    },
    {
      "epoch": 4.105377197265625e-05,
      "model_forward_time": 1.745427131652832,
      "step": 43048
    },
    {
      "epoch": 4.105377197265625e-05,
      "step": 43048,
      "training_step_time": 6.834856033325195
    },
    {
      "epoch": 4.1054725646972656e-05,
      "model_forward_time": 1.740992546081543,
      "step": 43049
    },
    {
      "epoch": 4.1054725646972656e-05,
      "step": 43049,
      "training_step_time": 6.816544055938721
    },
    {
      "epoch": 4.105567932128906e-05,
      "grad_norm": 0.1310308873653412,
      "learning_rate": 6.53737748488578e-06,
      "loss": 0.0577,
      "step": 43050
    },
    {
      "epoch": 4.105567932128906e-05,
      "model_forward_time": 1.729942798614502,
      "step": 43050
    },
    {
      "dynamics_loss_avg": 0.05252207815647125,
      "epoch": 4.105567932128906e-05,
      "step": 43050
    },
    {
      "action_loss_avg": 0.002844573464244604,
      "epoch": 4.105567932128906e-05,
      "step": 43050
    },
    {
      "epoch": 4.105567932128906e-05,
      "step": 43050,
      "training_step_time": 6.772848844528198
    },
    {
      "epoch": 4.105663299560547e-05,
      "model_forward_time": 2.3096764087677,
      "step": 43051
    },
    {
      "epoch": 4.105663299560547e-05,
      "step": 43051,
      "training_step_time": 7.737744092941284
    },
    {
      "epoch": 4.1057586669921875e-05,
      "model_forward_time": 1.7393238544464111,
      "step": 43052
    },
    {
      "epoch": 4.1057586669921875e-05,
      "step": 43052,
      "training_step_time": 7.755712270736694
    },
    {
      "epoch": 4.105854034423828e-05,
      "model_forward_time": 1.3605787754058838,
      "step": 43053
    },
    {
      "epoch": 4.105854034423828e-05,
      "step": 43053,
      "training_step_time": 6.813894510269165
    },
    {
      "epoch": 4.105949401855469e-05,
      "model_forward_time": 1.7483735084533691,
      "step": 43054
    },
    {
      "epoch": 4.105949401855469e-05,
      "step": 43054,
      "training_step_time": 6.787543296813965
    },
    {
      "epoch": 4.1060447692871093e-05,
      "model_forward_time": 1.7563145160675049,
      "step": 43055
    },
    {
      "epoch": 4.1060447692871093e-05,
      "step": 43055,
      "training_step_time": 6.872625350952148
    },
    {
      "epoch": 4.10614013671875e-05,
      "model_forward_time": 1.7261080741882324,
      "step": 43056
    },
    {
      "epoch": 4.10614013671875e-05,
      "step": 43056,
      "training_step_time": 6.842097759246826
    },
    {
      "epoch": 4.106235504150391e-05,
      "model_forward_time": 1.7279119491577148,
      "step": 43057
    },
    {
      "epoch": 4.106235504150391e-05,
      "step": 43057,
      "training_step_time": 6.739381313323975
    },
    {
      "epoch": 4.106330871582031e-05,
      "model_forward_time": 1.721601963043213,
      "step": 43058
    },
    {
      "epoch": 4.106330871582031e-05,
      "step": 43058,
      "training_step_time": 6.821988344192505
    },
    {
      "epoch": 4.1064262390136716e-05,
      "model_forward_time": 1.731137990951538,
      "step": 43059
    },
    {
      "epoch": 4.1064262390136716e-05,
      "step": 43059,
      "training_step_time": 6.7936131954193115
    },
    {
      "epoch": 4.106521606445313e-05,
      "grad_norm": 0.09202296286821365,
      "learning_rate": 6.535804031728044e-06,
      "loss": 0.0597,
      "step": 43060
    },
    {
      "epoch": 4.106521606445313e-05,
      "model_forward_time": 2.8403513431549072,
      "step": 43060
    },
    {
      "dynamics_loss_avg": 0.05476532969623804,
      "epoch": 4.106521606445313e-05,
      "step": 43060
    },
    {
      "action_loss_avg": 0.0027435520139988514,
      "epoch": 4.106521606445313e-05,
      "step": 43060
    },
    {
      "epoch": 4.106521606445313e-05,
      "step": 43060,
      "training_step_time": 7.707451581954956
    },
    {
      "epoch": 4.106616973876953e-05,
      "model_forward_time": 1.7411010265350342,
      "step": 43061
    },
    {
      "epoch": 4.106616973876953e-05,
      "step": 43061,
      "training_step_time": 7.716040849685669
    },
    {
      "epoch": 4.1067123413085935e-05,
      "model_forward_time": 1.7434179782867432,
      "step": 43062
    },
    {
      "epoch": 4.1067123413085935e-05,
      "step": 43062,
      "training_step_time": 7.214536428451538
    },
    {
      "epoch": 4.1068077087402346e-05,
      "model_forward_time": 1.746354341506958,
      "step": 43063
    },
    {
      "epoch": 4.1068077087402346e-05,
      "step": 43063,
      "training_step_time": 6.740392446517944
    },
    {
      "epoch": 4.106903076171875e-05,
      "model_forward_time": 1.7398855686187744,
      "step": 43064
    },
    {
      "epoch": 4.106903076171875e-05,
      "step": 43064,
      "training_step_time": 6.759625673294067
    },
    {
      "epoch": 4.1069984436035154e-05,
      "model_forward_time": 1.7477033138275146,
      "step": 43065
    },
    {
      "epoch": 4.1069984436035154e-05,
      "step": 43065,
      "training_step_time": 6.712250471115112
    },
    {
      "epoch": 4.1070938110351565e-05,
      "model_forward_time": 1.7498388290405273,
      "step": 43066
    },
    {
      "epoch": 4.1070938110351565e-05,
      "step": 43066,
      "training_step_time": 6.760669231414795
    },
    {
      "epoch": 4.107189178466797e-05,
      "model_forward_time": 1.7373151779174805,
      "step": 43067
    },
    {
      "epoch": 4.107189178466797e-05,
      "step": 43067,
      "training_step_time": 6.695189952850342
    },
    {
      "epoch": 4.107284545898437e-05,
      "model_forward_time": 1.7336409091949463,
      "step": 43068
    },
    {
      "epoch": 4.107284545898437e-05,
      "step": 43068,
      "training_step_time": 6.814544439315796
    },
    {
      "epoch": 4.1073799133300784e-05,
      "model_forward_time": 2.8627521991729736,
      "step": 43069
    },
    {
      "epoch": 4.1073799133300784e-05,
      "step": 43069,
      "training_step_time": 7.7233734130859375
    },
    {
      "epoch": 4.107475280761719e-05,
      "grad_norm": 0.15801967680454254,
      "learning_rate": 6.5342304106170915e-06,
      "loss": 0.0625,
      "step": 43070
    },
    {
      "epoch": 4.107475280761719e-05,
      "model_forward_time": 1.760890245437622,
      "step": 43070
    },
    {
      "dynamics_loss_avg": 0.050284956209361555,
      "epoch": 4.107475280761719e-05,
      "step": 43070
    },
    {
      "action_loss_avg": 0.002579727239208296,
      "epoch": 4.107475280761719e-05,
      "step": 43070
    },
    {
      "epoch": 4.107475280761719e-05,
      "step": 43070,
      "training_step_time": 8.60559868812561
    },
    {
      "epoch": 4.107570648193359e-05,
      "model_forward_time": 1.7367854118347168,
      "step": 43071
    },
    {
      "epoch": 4.107570648193359e-05,
      "step": 43071,
      "training_step_time": 7.2215352058410645
    },
    {
      "epoch": 4.107666015625e-05,
      "model_forward_time": 1.7441585063934326,
      "step": 43072
    },
    {
      "epoch": 4.107666015625e-05,
      "step": 43072,
      "training_step_time": 6.7827489376068115
    },
    {
      "epoch": 4.107761383056641e-05,
      "model_forward_time": 1.7395586967468262,
      "step": 43073
    },
    {
      "epoch": 4.107761383056641e-05,
      "step": 43073,
      "training_step_time": 6.737816333770752
    },
    {
      "epoch": 4.107856750488281e-05,
      "model_forward_time": 1.738316297531128,
      "step": 43074
    },
    {
      "epoch": 4.107856750488281e-05,
      "step": 43074,
      "training_step_time": 6.7600791454315186
    },
    {
      "epoch": 4.107952117919922e-05,
      "model_forward_time": 1.7392377853393555,
      "step": 43075
    },
    {
      "epoch": 4.107952117919922e-05,
      "step": 43075,
      "training_step_time": 6.779526710510254
    },
    {
      "epoch": 4.1080474853515625e-05,
      "model_forward_time": 1.908121109008789,
      "step": 43076
    },
    {
      "epoch": 4.1080474853515625e-05,
      "step": 43076,
      "training_step_time": 6.755816698074341
    },
    {
      "epoch": 4.108142852783203e-05,
      "model_forward_time": 1.894054889678955,
      "step": 43077
    },
    {
      "epoch": 4.108142852783203e-05,
      "step": 43077,
      "training_step_time": 6.750182390213013
    },
    {
      "epoch": 4.108238220214844e-05,
      "model_forward_time": 3.0258753299713135,
      "step": 43078
    },
    {
      "epoch": 4.108238220214844e-05,
      "step": 43078,
      "training_step_time": 8.36230206489563
    },
    {
      "epoch": 4.1083335876464844e-05,
      "model_forward_time": 1.8804748058319092,
      "step": 43079
    },
    {
      "epoch": 4.1083335876464844e-05,
      "step": 43079,
      "training_step_time": 9.082014799118042
    },
    {
      "epoch": 4.108428955078125e-05,
      "grad_norm": 0.23667699098587036,
      "learning_rate": 6.532656621725006e-06,
      "loss": 0.0594,
      "step": 43080
    },
    {
      "epoch": 4.108428955078125e-05,
      "model_forward_time": 1.8958489894866943,
      "step": 43080
    },
    {
      "dynamics_loss_avg": 0.0611043207347393,
      "epoch": 4.108428955078125e-05,
      "step": 43080
    },
    {
      "action_loss_avg": 0.00120696333469823,
      "epoch": 4.108428955078125e-05,
      "step": 43080
    },
    {
      "epoch": 4.108428955078125e-05,
      "step": 43080,
      "training_step_time": 7.670689582824707
    },
    {
      "epoch": 4.108524322509766e-05,
      "model_forward_time": 1.8911254405975342,
      "step": 43081
    },
    {
      "epoch": 4.108524322509766e-05,
      "step": 43081,
      "training_step_time": 6.81596565246582
    },
    {
      "epoch": 4.108619689941406e-05,
      "model_forward_time": 1.8748819828033447,
      "step": 43082
    },
    {
      "epoch": 4.108619689941406e-05,
      "step": 43082,
      "training_step_time": 6.782944202423096
    },
    {
      "epoch": 4.108715057373047e-05,
      "model_forward_time": 1.8900489807128906,
      "step": 43083
    },
    {
      "epoch": 4.108715057373047e-05,
      "step": 43083,
      "training_step_time": 6.7936413288116455
    },
    {
      "epoch": 4.108810424804688e-05,
      "model_forward_time": 1.9039905071258545,
      "step": 43084
    },
    {
      "epoch": 4.108810424804688e-05,
      "step": 43084,
      "training_step_time": 6.805534362792969
    },
    {
      "epoch": 4.108905792236328e-05,
      "model_forward_time": 1.8882522583007812,
      "step": 43085
    },
    {
      "epoch": 4.108905792236328e-05,
      "step": 43085,
      "training_step_time": 6.769100904464722
    },
    {
      "epoch": 4.1090011596679686e-05,
      "model_forward_time": 1.9096558094024658,
      "step": 43086
    },
    {
      "epoch": 4.1090011596679686e-05,
      "step": 43086,
      "training_step_time": 6.8527138233184814
    },
    {
      "epoch": 4.10909652709961e-05,
      "model_forward_time": 1.913020133972168,
      "step": 43087
    },
    {
      "epoch": 4.10909652709961e-05,
      "step": 43087,
      "training_step_time": 7.295717477798462
    },
    {
      "epoch": 4.10919189453125e-05,
      "model_forward_time": 1.8525123596191406,
      "step": 43088
    },
    {
      "epoch": 4.10919189453125e-05,
      "step": 43088,
      "training_step_time": 7.7136008739471436
    },
    {
      "epoch": 4.1092872619628905e-05,
      "model_forward_time": 1.8865809440612793,
      "step": 43089
    },
    {
      "epoch": 4.1092872619628905e-05,
      "step": 43089,
      "training_step_time": 7.659526109695435
    },
    {
      "epoch": 4.1093826293945316e-05,
      "grad_norm": 0.13873104751110077,
      "learning_rate": 6.531082665223898e-06,
      "loss": 0.0625,
      "step": 43090
    },
    {
      "epoch": 4.1093826293945316e-05,
      "model_forward_time": 1.91475510597229,
      "step": 43090
    },
    {
      "dynamics_loss_avg": 0.06579385623335839,
      "epoch": 4.1093826293945316e-05,
      "step": 43090
    },
    {
      "action_loss_avg": 0.002105428322101943,
      "epoch": 4.1093826293945316e-05,
      "step": 43090
    },
    {
      "epoch": 4.1093826293945316e-05,
      "step": 43090,
      "training_step_time": 7.199249982833862
    },
    {
      "epoch": 4.109477996826172e-05,
      "model_forward_time": 1.89420485496521,
      "step": 43091
    },
    {
      "epoch": 4.109477996826172e-05,
      "step": 43091,
      "training_step_time": 6.720210313796997
    },
    {
      "epoch": 4.1095733642578124e-05,
      "model_forward_time": 1.89756178855896,
      "step": 43092
    },
    {
      "epoch": 4.1095733642578124e-05,
      "step": 43092,
      "training_step_time": 6.752454042434692
    },
    {
      "epoch": 4.1096687316894535e-05,
      "model_forward_time": 1.879525899887085,
      "step": 43093
    },
    {
      "epoch": 4.1096687316894535e-05,
      "step": 43093,
      "training_step_time": 6.790509223937988
    },
    {
      "epoch": 4.109764099121094e-05,
      "model_forward_time": 1.9086644649505615,
      "step": 43094
    },
    {
      "epoch": 4.109764099121094e-05,
      "step": 43094,
      "training_step_time": 6.737932443618774
    },
    {
      "epoch": 4.109859466552734e-05,
      "model_forward_time": 1.8966617584228516,
      "step": 43095
    },
    {
      "epoch": 4.109859466552734e-05,
      "step": 43095,
      "training_step_time": 6.756628036499023
    },
    {
      "epoch": 4.109954833984375e-05,
      "model_forward_time": 1.8968157768249512,
      "step": 43096
    },
    {
      "epoch": 4.109954833984375e-05,
      "step": 43096,
      "training_step_time": 6.8432066440582275
    },
    {
      "epoch": 4.110050201416016e-05,
      "model_forward_time": 3.0059385299682617,
      "step": 43097
    },
    {
      "epoch": 4.110050201416016e-05,
      "step": 43097,
      "training_step_time": 8.382938623428345
    },
    {
      "epoch": 4.110145568847656e-05,
      "model_forward_time": 1.9089171886444092,
      "step": 43098
    },
    {
      "epoch": 4.110145568847656e-05,
      "step": 43098,
      "training_step_time": 7.70732855796814
    },
    {
      "epoch": 4.1102409362792966e-05,
      "model_forward_time": 1.8999159336090088,
      "step": 43099
    },
    {
      "epoch": 4.1102409362792966e-05,
      "step": 43099,
      "training_step_time": 7.357031345367432
    },
    {
      "epoch": 4.1103363037109376e-05,
      "grad_norm": 0.10206408053636551,
      "learning_rate": 6.529508541285889e-06,
      "loss": 0.0563,
      "step": 43100
    },
    {
      "epoch": 4.1103363037109376e-05,
      "model_forward_time": 1.9044442176818848,
      "step": 43100
    },
    {
      "dynamics_loss_avg": 0.060037936363369224,
      "epoch": 4.1103363037109376e-05,
      "step": 43100
    },
    {
      "action_loss_avg": 0.0014802582969423383,
      "epoch": 4.1103363037109376e-05,
      "step": 43100
    },
    {
      "epoch": 4.1103363037109376e-05,
      "step": 43100,
      "training_step_time": 6.769241094589233
    },
    {
      "epoch": 4.110431671142578e-05,
      "model_forward_time": 1.8873212337493896,
      "step": 43101
    },
    {
      "epoch": 4.110431671142578e-05,
      "step": 43101,
      "training_step_time": 6.74861216545105
    },
    {
      "epoch": 4.1105270385742184e-05,
      "model_forward_time": 1.8810243606567383,
      "step": 43102
    },
    {
      "epoch": 4.1105270385742184e-05,
      "step": 43102,
      "training_step_time": 6.7922937870025635
    },
    {
      "epoch": 4.1106224060058595e-05,
      "model_forward_time": 1.8996756076812744,
      "step": 43103
    },
    {
      "epoch": 4.1106224060058595e-05,
      "step": 43103,
      "training_step_time": 6.773436069488525
    },
    {
      "epoch": 4.1107177734375e-05,
      "model_forward_time": 1.8896830081939697,
      "step": 43104
    },
    {
      "epoch": 4.1107177734375e-05,
      "step": 43104,
      "training_step_time": 6.760426759719849
    },
    {
      "epoch": 4.11081314086914e-05,
      "model_forward_time": 1.8970386981964111,
      "step": 43105
    },
    {
      "epoch": 4.11081314086914e-05,
      "step": 43105,
      "training_step_time": 6.808594703674316
    },
    {
      "epoch": 4.1109085083007814e-05,
      "model_forward_time": 3.019850254058838,
      "step": 43106
    },
    {
      "epoch": 4.1109085083007814e-05,
      "step": 43106,
      "training_step_time": 8.33211064338684
    },
    {
      "epoch": 4.111003875732422e-05,
      "model_forward_time": 1.8847601413726807,
      "step": 43107
    },
    {
      "epoch": 4.111003875732422e-05,
      "step": 43107,
      "training_step_time": 8.377399444580078
    },
    {
      "epoch": 4.111099243164062e-05,
      "model_forward_time": 1.9065766334533691,
      "step": 43108
    },
    {
      "epoch": 4.111099243164062e-05,
      "step": 43108,
      "training_step_time": 7.5667054653167725
    },
    {
      "epoch": 4.111194610595703e-05,
      "model_forward_time": 1.904242992401123,
      "step": 43109
    },
    {
      "epoch": 4.111194610595703e-05,
      "step": 43109,
      "training_step_time": 6.827122211456299
    },
    {
      "epoch": 4.111289978027344e-05,
      "grad_norm": 0.12280550599098206,
      "learning_rate": 6.527934250083129e-06,
      "loss": 0.0578,
      "step": 43110
    },
    {
      "epoch": 4.111289978027344e-05,
      "model_forward_time": 1.9011766910552979,
      "step": 43110
    },
    {
      "dynamics_loss_avg": 0.05136373173445463,
      "epoch": 4.111289978027344e-05,
      "step": 43110
    },
    {
      "action_loss_avg": 0.0019460448966128752,
      "epoch": 4.111289978027344e-05,
      "step": 43110
    },
    {
      "epoch": 4.111289978027344e-05,
      "step": 43110,
      "training_step_time": 6.889414310455322
    },
    {
      "epoch": 4.111385345458984e-05,
      "model_forward_time": 1.8813297748565674,
      "step": 43111
    },
    {
      "epoch": 4.111385345458984e-05,
      "step": 43111,
      "training_step_time": 6.899120092391968
    },
    {
      "epoch": 4.111480712890625e-05,
      "model_forward_time": 1.9241774082183838,
      "step": 43112
    },
    {
      "epoch": 4.111480712890625e-05,
      "step": 43112,
      "training_step_time": 6.8569769859313965
    },
    {
      "epoch": 4.1115760803222656e-05,
      "model_forward_time": 1.7757632732391357,
      "step": 43113
    },
    {
      "epoch": 4.1115760803222656e-05,
      "step": 43113,
      "training_step_time": 6.909052610397339
    },
    {
      "epoch": 4.111671447753906e-05,
      "model_forward_time": 1.7276082038879395,
      "step": 43114
    },
    {
      "epoch": 4.111671447753906e-05,
      "step": 43114,
      "training_step_time": 6.9002463817596436
    },
    {
      "epoch": 4.111766815185547e-05,
      "model_forward_time": 2.8816773891448975,
      "step": 43115
    },
    {
      "epoch": 4.111766815185547e-05,
      "step": 43115,
      "training_step_time": 8.653345346450806
    },
    {
      "epoch": 4.1118621826171875e-05,
      "model_forward_time": 1.8134069442749023,
      "step": 43116
    },
    {
      "epoch": 4.1118621826171875e-05,
      "step": 43116,
      "training_step_time": 9.40299391746521
    },
    {
      "epoch": 4.111957550048828e-05,
      "model_forward_time": 1.7225639820098877,
      "step": 43117
    },
    {
      "epoch": 4.111957550048828e-05,
      "step": 43117,
      "training_step_time": 7.740961790084839
    },
    {
      "epoch": 4.112052917480469e-05,
      "model_forward_time": 1.7490551471710205,
      "step": 43118
    },
    {
      "epoch": 4.112052917480469e-05,
      "step": 43118,
      "training_step_time": 7.570882558822632
    },
    {
      "epoch": 4.1121482849121094e-05,
      "model_forward_time": 1.7254133224487305,
      "step": 43119
    },
    {
      "epoch": 4.1121482849121094e-05,
      "step": 43119,
      "training_step_time": 6.851701736450195
    },
    {
      "epoch": 4.11224365234375e-05,
      "grad_norm": 0.10809322446584702,
      "learning_rate": 6.526359791787776e-06,
      "loss": 0.0544,
      "step": 43120
    },
    {
      "epoch": 4.11224365234375e-05,
      "model_forward_time": 1.7275564670562744,
      "step": 43120
    },
    {
      "dynamics_loss_avg": 0.047927274741232394,
      "epoch": 4.11224365234375e-05,
      "step": 43120
    },
    {
      "action_loss_avg": 0.001066438452107832,
      "epoch": 4.11224365234375e-05,
      "step": 43120
    },
    {
      "epoch": 4.11224365234375e-05,
      "step": 43120,
      "training_step_time": 6.832351207733154
    },
    {
      "epoch": 4.112339019775391e-05,
      "model_forward_time": 1.7311394214630127,
      "step": 43121
    },
    {
      "epoch": 4.112339019775391e-05,
      "step": 43121,
      "training_step_time": 6.968228101730347
    },
    {
      "epoch": 4.112434387207031e-05,
      "model_forward_time": 1.7298097610473633,
      "step": 43122
    },
    {
      "epoch": 4.112434387207031e-05,
      "step": 43122,
      "training_step_time": 6.935638904571533
    },
    {
      "epoch": 4.1125297546386716e-05,
      "model_forward_time": 1.7317354679107666,
      "step": 43123
    },
    {
      "epoch": 4.1125297546386716e-05,
      "step": 43123,
      "training_step_time": 6.915047883987427
    },
    {
      "epoch": 4.112625122070313e-05,
      "model_forward_time": 2.9074513912200928,
      "step": 43124
    },
    {
      "epoch": 4.112625122070313e-05,
      "step": 43124,
      "training_step_time": 7.802004098892212
    },
    {
      "epoch": 4.112720489501953e-05,
      "model_forward_time": 1.8039631843566895,
      "step": 43125
    },
    {
      "epoch": 4.112720489501953e-05,
      "step": 43125,
      "training_step_time": 9.184064388275146
    },
    {
      "epoch": 4.1128158569335935e-05,
      "model_forward_time": 1.7323780059814453,
      "step": 43126
    },
    {
      "epoch": 4.1128158569335935e-05,
      "step": 43126,
      "training_step_time": 7.8248467445373535
    },
    {
      "epoch": 4.1129112243652346e-05,
      "model_forward_time": 1.727893352508545,
      "step": 43127
    },
    {
      "epoch": 4.1129112243652346e-05,
      "step": 43127,
      "training_step_time": 7.159418106079102
    },
    {
      "epoch": 4.113006591796875e-05,
      "model_forward_time": 1.7319309711456299,
      "step": 43128
    },
    {
      "epoch": 4.113006591796875e-05,
      "step": 43128,
      "training_step_time": 7.406500816345215
    },
    {
      "epoch": 4.1131019592285154e-05,
      "model_forward_time": 1.7320506572723389,
      "step": 43129
    },
    {
      "epoch": 4.1131019592285154e-05,
      "step": 43129,
      "training_step_time": 6.9166929721832275
    },
    {
      "epoch": 4.1131973266601565e-05,
      "grad_norm": 0.22909462451934814,
      "learning_rate": 6.524785166572009e-06,
      "loss": 0.0572,
      "step": 43130
    },
    {
      "epoch": 4.1131973266601565e-05,
      "model_forward_time": 1.7262022495269775,
      "step": 43130
    },
    {
      "dynamics_loss_avg": 0.05324495071545243,
      "epoch": 4.1131973266601565e-05,
      "step": 43130
    },
    {
      "action_loss_avg": 0.0018033172265859321,
      "epoch": 4.1131973266601565e-05,
      "step": 43130
    },
    {
      "epoch": 4.1131973266601565e-05,
      "step": 43130,
      "training_step_time": 6.890455484390259
    },
    {
      "epoch": 4.113292694091797e-05,
      "model_forward_time": 1.740532398223877,
      "step": 43131
    },
    {
      "epoch": 4.113292694091797e-05,
      "step": 43131,
      "training_step_time": 6.883538007736206
    },
    {
      "epoch": 4.113388061523437e-05,
      "model_forward_time": 1.7350459098815918,
      "step": 43132
    },
    {
      "epoch": 4.113388061523437e-05,
      "step": 43132,
      "training_step_time": 6.860260009765625
    },
    {
      "epoch": 4.1134834289550784e-05,
      "model_forward_time": 2.8989298343658447,
      "step": 43133
    },
    {
      "epoch": 4.1134834289550784e-05,
      "step": 43133,
      "training_step_time": 7.588545799255371
    },
    {
      "epoch": 4.113578796386719e-05,
      "model_forward_time": 1.8007652759552002,
      "step": 43134
    },
    {
      "epoch": 4.113578796386719e-05,
      "step": 43134,
      "training_step_time": 10.14620566368103
    },
    {
      "epoch": 4.113674163818359e-05,
      "model_forward_time": 1.7259869575500488,
      "step": 43135
    },
    {
      "epoch": 4.113674163818359e-05,
      "step": 43135,
      "training_step_time": 7.6113481521606445
    },
    {
      "epoch": 4.11376953125e-05,
      "model_forward_time": 1.7291321754455566,
      "step": 43136
    },
    {
      "epoch": 4.11376953125e-05,
      "step": 43136,
      "training_step_time": 7.5210559368133545
    },
    {
      "epoch": 4.113864898681641e-05,
      "model_forward_time": 1.746516227722168,
      "step": 43137
    },
    {
      "epoch": 4.113864898681641e-05,
      "step": 43137,
      "training_step_time": 7.217866659164429
    },
    {
      "epoch": 4.113960266113281e-05,
      "model_forward_time": 1.739013671875,
      "step": 43138
    },
    {
      "epoch": 4.113960266113281e-05,
      "step": 43138,
      "training_step_time": 6.712872743606567
    },
    {
      "epoch": 4.114055633544922e-05,
      "model_forward_time": 1.7377901077270508,
      "step": 43139
    },
    {
      "epoch": 4.114055633544922e-05,
      "step": 43139,
      "training_step_time": 6.741169214248657
    },
    {
      "epoch": 4.1141510009765626e-05,
      "grad_norm": 0.1448555290699005,
      "learning_rate": 6.52321037460803e-06,
      "loss": 0.0577,
      "step": 43140
    },
    {
      "epoch": 4.1141510009765626e-05,
      "model_forward_time": 1.3719253540039062,
      "step": 43140
    },
    {
      "dynamics_loss_avg": 0.054383463226258755,
      "epoch": 4.1141510009765626e-05,
      "step": 43140
    },
    {
      "action_loss_avg": 0.0018994918558746577,
      "epoch": 4.1141510009765626e-05,
      "step": 43140
    },
    {
      "epoch": 4.1141510009765626e-05,
      "step": 43140,
      "training_step_time": 6.773283004760742
    },
    {
      "epoch": 4.114246368408203e-05,
      "model_forward_time": 1.739879846572876,
      "step": 43141
    },
    {
      "epoch": 4.114246368408203e-05,
      "step": 43141,
      "training_step_time": 6.798986196517944
    },
    {
      "epoch": 4.114341735839844e-05,
      "model_forward_time": 2.909572124481201,
      "step": 43142
    },
    {
      "epoch": 4.114341735839844e-05,
      "step": 43142,
      "training_step_time": 7.653441667556763
    },
    {
      "epoch": 4.1144371032714844e-05,
      "model_forward_time": 1.7863214015960693,
      "step": 43143
    },
    {
      "epoch": 4.1144371032714844e-05,
      "step": 43143,
      "training_step_time": 9.786421298980713
    },
    {
      "epoch": 4.114532470703125e-05,
      "model_forward_time": 1.7475647926330566,
      "step": 43144
    },
    {
      "epoch": 4.114532470703125e-05,
      "step": 43144,
      "training_step_time": 8.73625898361206
    },
    {
      "epoch": 4.114627838134766e-05,
      "model_forward_time": 1.7311089038848877,
      "step": 43145
    },
    {
      "epoch": 4.114627838134766e-05,
      "step": 43145,
      "training_step_time": 7.608764171600342
    },
    {
      "epoch": 4.114723205566406e-05,
      "model_forward_time": 1.7370471954345703,
      "step": 43146
    },
    {
      "epoch": 4.114723205566406e-05,
      "step": 43146,
      "training_step_time": 7.345426321029663
    },
    {
      "epoch": 4.114818572998047e-05,
      "model_forward_time": 1.7330639362335205,
      "step": 43147
    },
    {
      "epoch": 4.114818572998047e-05,
      "step": 43147,
      "training_step_time": 6.809308290481567
    },
    {
      "epoch": 4.114913940429688e-05,
      "model_forward_time": 1.7399520874023438,
      "step": 43148
    },
    {
      "epoch": 4.114913940429688e-05,
      "step": 43148,
      "training_step_time": 6.794095277786255
    },
    {
      "epoch": 4.115009307861328e-05,
      "model_forward_time": 1.7346370220184326,
      "step": 43149
    },
    {
      "epoch": 4.115009307861328e-05,
      "step": 43149,
      "training_step_time": 6.776899337768555
    },
    {
      "epoch": 4.1151046752929686e-05,
      "grad_norm": 0.11405974626541138,
      "learning_rate": 6.521635416068054e-06,
      "loss": 0.0552,
      "step": 43150
    },
    {
      "epoch": 4.1151046752929686e-05,
      "model_forward_time": 1.742546558380127,
      "step": 43150
    },
    {
      "dynamics_loss_avg": 0.06214708816260099,
      "epoch": 4.1151046752929686e-05,
      "step": 43150
    },
    {
      "action_loss_avg": 0.0017986277875024826,
      "epoch": 4.1151046752929686e-05,
      "step": 43150
    },
    {
      "epoch": 4.1151046752929686e-05,
      "step": 43150,
      "training_step_time": 6.734246730804443
    },
    {
      "epoch": 4.11520004272461e-05,
      "model_forward_time": 2.8911471366882324,
      "step": 43151
    },
    {
      "epoch": 4.11520004272461e-05,
      "step": 43151,
      "training_step_time": 7.642164468765259
    },
    {
      "epoch": 4.11529541015625e-05,
      "model_forward_time": 1.74788498878479,
      "step": 43152
    },
    {
      "epoch": 4.11529541015625e-05,
      "step": 43152,
      "training_step_time": 7.639158487319946
    },
    {
      "epoch": 4.1153907775878905e-05,
      "model_forward_time": 1.7361328601837158,
      "step": 43153
    },
    {
      "epoch": 4.1153907775878905e-05,
      "step": 43153,
      "training_step_time": 9.716276168823242
    },
    {
      "epoch": 4.1154861450195316e-05,
      "model_forward_time": 1.7392947673797607,
      "step": 43154
    },
    {
      "epoch": 4.1154861450195316e-05,
      "step": 43154,
      "training_step_time": 8.298390626907349
    },
    {
      "epoch": 4.115581512451172e-05,
      "model_forward_time": 1.7352254390716553,
      "step": 43155
    },
    {
      "epoch": 4.115581512451172e-05,
      "step": 43155,
      "training_step_time": 7.210411787033081
    },
    {
      "epoch": 4.1156768798828124e-05,
      "model_forward_time": 1.7349753379821777,
      "step": 43156
    },
    {
      "epoch": 4.1156768798828124e-05,
      "step": 43156,
      "training_step_time": 6.6408936977386475
    },
    {
      "epoch": 4.115772247314453e-05,
      "model_forward_time": 1.735316276550293,
      "step": 43157
    },
    {
      "epoch": 4.115772247314453e-05,
      "step": 43157,
      "training_step_time": 6.755208969116211
    },
    {
      "epoch": 4.115867614746094e-05,
      "model_forward_time": 1.7345035076141357,
      "step": 43158
    },
    {
      "epoch": 4.115867614746094e-05,
      "step": 43158,
      "training_step_time": 6.7696309089660645
    },
    {
      "epoch": 4.115962982177734e-05,
      "model_forward_time": 1.7279858589172363,
      "step": 43159
    },
    {
      "epoch": 4.115962982177734e-05,
      "step": 43159,
      "training_step_time": 6.743340253829956
    },
    {
      "epoch": 4.116058349609375e-05,
      "grad_norm": 0.140250563621521,
      "learning_rate": 6.520060291124317e-06,
      "loss": 0.0578,
      "step": 43160
    },
    {
      "epoch": 4.116058349609375e-05,
      "model_forward_time": 2.8339426517486572,
      "step": 43160
    },
    {
      "dynamics_loss_avg": 0.05885110795497894,
      "epoch": 4.116058349609375e-05,
      "step": 43160
    },
    {
      "action_loss_avg": 0.0011575830343645067,
      "epoch": 4.116058349609375e-05,
      "step": 43160
    },
    {
      "epoch": 4.116058349609375e-05,
      "step": 43160,
      "training_step_time": 7.583930492401123
    },
    {
      "epoch": 4.116153717041016e-05,
      "model_forward_time": 1.7369985580444336,
      "step": 43161
    },
    {
      "epoch": 4.116153717041016e-05,
      "step": 43161,
      "training_step_time": 10.910077810287476
    },
    {
      "epoch": 4.116249084472656e-05,
      "model_forward_time": 1.806919813156128,
      "step": 43162
    },
    {
      "epoch": 4.116249084472656e-05,
      "step": 43162,
      "training_step_time": 9.83931040763855
    },
    {
      "epoch": 4.1163444519042966e-05,
      "model_forward_time": 1.746875286102295,
      "step": 43163
    },
    {
      "epoch": 4.1163444519042966e-05,
      "step": 43163,
      "training_step_time": 8.617221355438232
    },
    {
      "epoch": 4.1164398193359376e-05,
      "model_forward_time": 1.740692138671875,
      "step": 43164
    },
    {
      "epoch": 4.1164398193359376e-05,
      "step": 43164,
      "training_step_time": 7.401067495346069
    },
    {
      "epoch": 4.116535186767578e-05,
      "model_forward_time": 1.7382726669311523,
      "step": 43165
    },
    {
      "epoch": 4.116535186767578e-05,
      "step": 43165,
      "training_step_time": 6.718564510345459
    },
    {
      "epoch": 4.1166305541992184e-05,
      "model_forward_time": 1.7417895793914795,
      "step": 43166
    },
    {
      "epoch": 4.1166305541992184e-05,
      "step": 43166,
      "training_step_time": 6.710943698883057
    },
    {
      "epoch": 4.1167259216308595e-05,
      "model_forward_time": 1.737428903579712,
      "step": 43167
    },
    {
      "epoch": 4.1167259216308595e-05,
      "step": 43167,
      "training_step_time": 6.732097387313843
    },
    {
      "epoch": 4.1168212890625e-05,
      "model_forward_time": 1.7361476421356201,
      "step": 43168
    },
    {
      "epoch": 4.1168212890625e-05,
      "step": 43168,
      "training_step_time": 6.775116443634033
    },
    {
      "epoch": 4.11691665649414e-05,
      "model_forward_time": 2.831537961959839,
      "step": 43169
    },
    {
      "epoch": 4.11691665649414e-05,
      "step": 43169,
      "training_step_time": 7.569332122802734
    },
    {
      "epoch": 4.1170120239257814e-05,
      "grad_norm": 0.12101579457521439,
      "learning_rate": 6.5184849999490714e-06,
      "loss": 0.0588,
      "step": 43170
    },
    {
      "epoch": 4.1170120239257814e-05,
      "model_forward_time": 1.7380406856536865,
      "step": 43170
    },
    {
      "dynamics_loss_avg": 0.0510047790594399,
      "epoch": 4.1170120239257814e-05,
      "step": 43170
    },
    {
      "action_loss_avg": 0.0013572767056757584,
      "epoch": 4.1170120239257814e-05,
      "step": 43170
    },
    {
      "epoch": 4.1170120239257814e-05,
      "step": 43170,
      "training_step_time": 7.728565692901611
    },
    {
      "epoch": 4.117107391357422e-05,
      "model_forward_time": 1.7914540767669678,
      "step": 43171
    },
    {
      "epoch": 4.117107391357422e-05,
      "step": 43171,
      "training_step_time": 10.103043794631958
    },
    {
      "epoch": 4.117202758789062e-05,
      "model_forward_time": 1.731426477432251,
      "step": 43172
    },
    {
      "epoch": 4.117202758789062e-05,
      "step": 43172,
      "training_step_time": 7.619854927062988
    },
    {
      "epoch": 4.117298126220703e-05,
      "model_forward_time": 1.7402658462524414,
      "step": 43173
    },
    {
      "epoch": 4.117298126220703e-05,
      "step": 43173,
      "training_step_time": 7.348057270050049
    },
    {
      "epoch": 4.117393493652344e-05,
      "model_forward_time": 1.7424864768981934,
      "step": 43174
    },
    {
      "epoch": 4.117393493652344e-05,
      "step": 43174,
      "training_step_time": 7.290937185287476
    },
    {
      "epoch": 4.117488861083984e-05,
      "model_forward_time": 1.7399930953979492,
      "step": 43175
    },
    {
      "epoch": 4.117488861083984e-05,
      "step": 43175,
      "training_step_time": 6.753467559814453
    },
    {
      "epoch": 4.117584228515625e-05,
      "model_forward_time": 1.7418699264526367,
      "step": 43176
    },
    {
      "epoch": 4.117584228515625e-05,
      "step": 43176,
      "training_step_time": 6.7311851978302
    },
    {
      "epoch": 4.1176795959472656e-05,
      "model_forward_time": 1.7381393909454346,
      "step": 43177
    },
    {
      "epoch": 4.1176795959472656e-05,
      "step": 43177,
      "training_step_time": 6.704458236694336
    },
    {
      "epoch": 4.117774963378906e-05,
      "model_forward_time": 1.7320411205291748,
      "step": 43178
    },
    {
      "epoch": 4.117774963378906e-05,
      "step": 43178,
      "training_step_time": 7.530280113220215
    },
    {
      "epoch": 4.117870330810547e-05,
      "model_forward_time": 2.899803638458252,
      "step": 43179
    },
    {
      "epoch": 4.117870330810547e-05,
      "step": 43179,
      "training_step_time": 7.594786167144775
    },
    {
      "epoch": 4.1179656982421875e-05,
      "grad_norm": 0.1325417011976242,
      "learning_rate": 6.516909542714591e-06,
      "loss": 0.0573,
      "step": 43180
    },
    {
      "epoch": 4.1179656982421875e-05,
      "model_forward_time": 1.7992303371429443,
      "step": 43180
    },
    {
      "dynamics_loss_avg": 0.04789006561040878,
      "epoch": 4.1179656982421875e-05,
      "step": 43180
    },
    {
      "action_loss_avg": 0.002062192518496886,
      "epoch": 4.1179656982421875e-05,
      "step": 43180
    },
    {
      "epoch": 4.1179656982421875e-05,
      "step": 43180,
      "training_step_time": 7.6693480014801025
    },
    {
      "epoch": 4.118061065673828e-05,
      "model_forward_time": 1.7335562705993652,
      "step": 43181
    },
    {
      "epoch": 4.118061065673828e-05,
      "step": 43181,
      "training_step_time": 8.27358865737915
    },
    {
      "epoch": 4.118156433105469e-05,
      "model_forward_time": 1.7351446151733398,
      "step": 43182
    },
    {
      "epoch": 4.118156433105469e-05,
      "step": 43182,
      "training_step_time": 7.375203371047974
    },
    {
      "epoch": 4.1182518005371094e-05,
      "model_forward_time": 1.7329962253570557,
      "step": 43183
    },
    {
      "epoch": 4.1182518005371094e-05,
      "step": 43183,
      "training_step_time": 7.238336086273193
    },
    {
      "epoch": 4.11834716796875e-05,
      "model_forward_time": 1.732971429824829,
      "step": 43184
    },
    {
      "epoch": 4.11834716796875e-05,
      "step": 43184,
      "training_step_time": 6.712021112442017
    },
    {
      "epoch": 4.118442535400391e-05,
      "model_forward_time": 1.7348074913024902,
      "step": 43185
    },
    {
      "epoch": 4.118442535400391e-05,
      "step": 43185,
      "training_step_time": 6.720350742340088
    },
    {
      "epoch": 4.118537902832031e-05,
      "model_forward_time": 1.7368502616882324,
      "step": 43186
    },
    {
      "epoch": 4.118537902832031e-05,
      "step": 43186,
      "training_step_time": 6.731273651123047
    },
    {
      "epoch": 4.1186332702636716e-05,
      "model_forward_time": 1.7387893199920654,
      "step": 43187
    },
    {
      "epoch": 4.1186332702636716e-05,
      "step": 43187,
      "training_step_time": 7.526954174041748
    },
    {
      "epoch": 4.118728637695313e-05,
      "model_forward_time": 2.932858943939209,
      "step": 43188
    },
    {
      "epoch": 4.118728637695313e-05,
      "step": 43188,
      "training_step_time": 8.607353687286377
    },
    {
      "epoch": 4.118824005126953e-05,
      "model_forward_time": 1.8023791313171387,
      "step": 43189
    },
    {
      "epoch": 4.118824005126953e-05,
      "step": 43189,
      "training_step_time": 7.614574670791626
    },
    {
      "epoch": 4.1189193725585935e-05,
      "grad_norm": 0.09264887869358063,
      "learning_rate": 6.515333919593162e-06,
      "loss": 0.0608,
      "step": 43190
    },
    {
      "epoch": 4.1189193725585935e-05,
      "model_forward_time": 1.737783670425415,
      "step": 43190
    },
    {
      "dynamics_loss_avg": 0.06099089626222849,
      "epoch": 4.1189193725585935e-05,
      "step": 43190
    },
    {
      "action_loss_avg": 0.0020870512758847325,
      "epoch": 4.1189193725585935e-05,
      "step": 43190
    },
    {
      "epoch": 4.1189193725585935e-05,
      "step": 43190,
      "training_step_time": 8.724726676940918
    },
    {
      "epoch": 4.1190147399902346e-05,
      "model_forward_time": 1.736788034439087,
      "step": 43191
    },
    {
      "epoch": 4.1190147399902346e-05,
      "step": 43191,
      "training_step_time": 7.4332780838012695
    },
    {
      "epoch": 4.119110107421875e-05,
      "model_forward_time": 1.738074779510498,
      "step": 43192
    },
    {
      "epoch": 4.119110107421875e-05,
      "step": 43192,
      "training_step_time": 7.166362047195435
    },
    {
      "epoch": 4.1192054748535154e-05,
      "model_forward_time": 1.7362990379333496,
      "step": 43193
    },
    {
      "epoch": 4.1192054748535154e-05,
      "step": 43193,
      "training_step_time": 6.723670959472656
    },
    {
      "epoch": 4.1193008422851565e-05,
      "model_forward_time": 1.7351553440093994,
      "step": 43194
    },
    {
      "epoch": 4.1193008422851565e-05,
      "step": 43194,
      "training_step_time": 6.753262281417847
    },
    {
      "epoch": 4.119396209716797e-05,
      "model_forward_time": 1.7305412292480469,
      "step": 43195
    },
    {
      "epoch": 4.119396209716797e-05,
      "step": 43195,
      "training_step_time": 6.777386903762817
    },
    {
      "epoch": 4.119491577148437e-05,
      "model_forward_time": 1.7600560188293457,
      "step": 43196
    },
    {
      "epoch": 4.119491577148437e-05,
      "step": 43196,
      "training_step_time": 7.610973834991455
    },
    {
      "epoch": 4.1195869445800784e-05,
      "model_forward_time": 2.892828941345215,
      "step": 43197
    },
    {
      "epoch": 4.1195869445800784e-05,
      "step": 43197,
      "training_step_time": 7.642217397689819
    },
    {
      "epoch": 4.119682312011719e-05,
      "model_forward_time": 1.7562525272369385,
      "step": 43198
    },
    {
      "epoch": 4.119682312011719e-05,
      "step": 43198,
      "training_step_time": 8.239269733428955
    },
    {
      "epoch": 4.119777679443359e-05,
      "model_forward_time": 1.7271811962127686,
      "step": 43199
    },
    {
      "epoch": 4.119777679443359e-05,
      "step": 43199,
      "training_step_time": 7.646805763244629
    },
    {
      "epoch": 4.119873046875e-05,
      "grad_norm": 0.13123929500579834,
      "learning_rate": 6.513758130757094e-06,
      "loss": 0.0584,
      "step": 43200
    },
    {
      "epoch": 4.119873046875e-05,
      "model_forward_time": 1.7273004055023193,
      "step": 43200
    },
    {
      "dynamics_loss_avg": 0.05012097284197807,
      "epoch": 4.119873046875e-05,
      "step": 43200
    },
    {
      "action_loss_avg": 0.0013703038217499851,
      "epoch": 4.119873046875e-05,
      "step": 43200
    },
    {
      "epoch": 4.119873046875e-05,
      "step": 43200,
      "training_step_time": 7.606683969497681
    },
    {
      "epoch": 4.119968414306641e-05,
      "model_forward_time": 1.72623610496521,
      "step": 43201
    },
    {
      "epoch": 4.119968414306641e-05,
      "step": 43201,
      "training_step_time": 7.252673387527466
    },
    {
      "epoch": 4.120063781738281e-05,
      "model_forward_time": 1.714123249053955,
      "step": 43202
    },
    {
      "epoch": 4.120063781738281e-05,
      "step": 43202,
      "training_step_time": 7.22664213180542
    },
    {
      "epoch": 4.120159149169922e-05,
      "model_forward_time": 1.7253243923187256,
      "step": 43203
    },
    {
      "epoch": 4.120159149169922e-05,
      "step": 43203,
      "training_step_time": 6.763643503189087
    },
    {
      "epoch": 4.1202545166015626e-05,
      "model_forward_time": 1.7279822826385498,
      "step": 43204
    },
    {
      "epoch": 4.1202545166015626e-05,
      "step": 43204,
      "training_step_time": 6.693079471588135
    },
    {
      "epoch": 4.120349884033203e-05,
      "model_forward_time": 1.726867914199829,
      "step": 43205
    },
    {
      "epoch": 4.120349884033203e-05,
      "step": 43205,
      "training_step_time": 7.573235511779785
    },
    {
      "epoch": 4.120445251464844e-05,
      "model_forward_time": 2.8811473846435547,
      "step": 43206
    },
    {
      "epoch": 4.120445251464844e-05,
      "step": 43206,
      "training_step_time": 7.581512689590454
    },
    {
      "epoch": 4.1205406188964844e-05,
      "model_forward_time": 1.7516746520996094,
      "step": 43207
    },
    {
      "epoch": 4.1205406188964844e-05,
      "step": 43207,
      "training_step_time": 8.67579984664917
    },
    {
      "epoch": 4.120635986328125e-05,
      "model_forward_time": 1.743032693862915,
      "step": 43208
    },
    {
      "epoch": 4.120635986328125e-05,
      "step": 43208,
      "training_step_time": 9.337008714675903
    },
    {
      "epoch": 4.120731353759766e-05,
      "model_forward_time": 1.726973056793213,
      "step": 43209
    },
    {
      "epoch": 4.120731353759766e-05,
      "step": 43209,
      "training_step_time": 7.6229469776153564
    },
    {
      "epoch": 4.120826721191406e-05,
      "grad_norm": 0.1301552653312683,
      "learning_rate": 6.512182176378713e-06,
      "loss": 0.0608,
      "step": 43210
    },
    {
      "epoch": 4.120826721191406e-05,
      "model_forward_time": 1.7206790447235107,
      "step": 43210
    },
    {
      "dynamics_loss_avg": 0.05033695213496685,
      "epoch": 4.120826721191406e-05,
      "step": 43210
    },
    {
      "action_loss_avg": 0.0009567797242198139,
      "epoch": 4.120826721191406e-05,
      "step": 43210
    },
    {
      "epoch": 4.120826721191406e-05,
      "step": 43210,
      "training_step_time": 7.238147497177124
    },
    {
      "epoch": 4.120922088623047e-05,
      "model_forward_time": 1.730454921722412,
      "step": 43211
    },
    {
      "epoch": 4.120922088623047e-05,
      "step": 43211,
      "training_step_time": 7.212599992752075
    },
    {
      "epoch": 4.121017456054688e-05,
      "model_forward_time": 1.734375,
      "step": 43212
    },
    {
      "epoch": 4.121017456054688e-05,
      "step": 43212,
      "training_step_time": 6.750364780426025
    },
    {
      "epoch": 4.121112823486328e-05,
      "model_forward_time": 1.7275545597076416,
      "step": 43213
    },
    {
      "epoch": 4.121112823486328e-05,
      "step": 43213,
      "training_step_time": 6.747528076171875
    },
    {
      "epoch": 4.1212081909179686e-05,
      "model_forward_time": 1.7274932861328125,
      "step": 43214
    },
    {
      "epoch": 4.1212081909179686e-05,
      "step": 43214,
      "training_step_time": 7.542025566101074
    },
    {
      "epoch": 4.12130355834961e-05,
      "model_forward_time": 2.8579177856445312,
      "step": 43215
    },
    {
      "epoch": 4.12130355834961e-05,
      "step": 43215,
      "training_step_time": 9.393655061721802
    },
    {
      "epoch": 4.12139892578125e-05,
      "model_forward_time": 1.7501685619354248,
      "step": 43216
    },
    {
      "epoch": 4.12139892578125e-05,
      "step": 43216,
      "training_step_time": 8.590442895889282
    },
    {
      "epoch": 4.1214942932128905e-05,
      "model_forward_time": 1.7542595863342285,
      "step": 43217
    },
    {
      "epoch": 4.1214942932128905e-05,
      "step": 43217,
      "training_step_time": 8.327213764190674
    },
    {
      "epoch": 4.1215896606445316e-05,
      "model_forward_time": 1.7005622386932373,
      "step": 43218
    },
    {
      "epoch": 4.1215896606445316e-05,
      "step": 43218,
      "training_step_time": 8.16599726676941
    },
    {
      "epoch": 4.121685028076172e-05,
      "model_forward_time": 1.3459336757659912,
      "step": 43219
    },
    {
      "epoch": 4.121685028076172e-05,
      "step": 43219,
      "training_step_time": 7.549568176269531
    },
    {
      "epoch": 4.1217803955078124e-05,
      "grad_norm": 0.10880263149738312,
      "learning_rate": 6.510606056630363e-06,
      "loss": 0.062,
      "step": 43220
    },
    {
      "epoch": 4.1217803955078124e-05,
      "model_forward_time": 1.7175068855285645,
      "step": 43220
    },
    {
      "dynamics_loss_avg": 0.04827180234715343,
      "epoch": 4.1217803955078124e-05,
      "step": 43220
    },
    {
      "action_loss_avg": 0.0012390120653435587,
      "epoch": 4.1217803955078124e-05,
      "step": 43220
    },
    {
      "epoch": 4.1217803955078124e-05,
      "step": 43220,
      "training_step_time": 7.247079610824585
    },
    {
      "epoch": 4.121875762939453e-05,
      "model_forward_time": 1.720571517944336,
      "step": 43221
    },
    {
      "epoch": 4.121875762939453e-05,
      "step": 43221,
      "training_step_time": 6.7906653881073
    },
    {
      "epoch": 4.121971130371094e-05,
      "model_forward_time": 1.7194344997406006,
      "step": 43222
    },
    {
      "epoch": 4.121971130371094e-05,
      "step": 43222,
      "training_step_time": 6.734204530715942
    },
    {
      "epoch": 4.122066497802734e-05,
      "model_forward_time": 1.716813564300537,
      "step": 43223
    },
    {
      "epoch": 4.122066497802734e-05,
      "step": 43223,
      "training_step_time": 7.7431230545043945
    },
    {
      "epoch": 4.122161865234375e-05,
      "model_forward_time": 1.7222161293029785,
      "step": 43224
    },
    {
      "epoch": 4.122161865234375e-05,
      "step": 43224,
      "training_step_time": 7.873295545578003
    },
    {
      "epoch": 4.122257232666016e-05,
      "model_forward_time": 2.822686195373535,
      "step": 43225
    },
    {
      "epoch": 4.122257232666016e-05,
      "step": 43225,
      "training_step_time": 8.54678750038147
    },
    {
      "epoch": 4.122352600097656e-05,
      "model_forward_time": 1.7320399284362793,
      "step": 43226
    },
    {
      "epoch": 4.122352600097656e-05,
      "step": 43226,
      "training_step_time": 9.07340407371521
    },
    {
      "epoch": 4.1224479675292966e-05,
      "model_forward_time": 1.7148990631103516,
      "step": 43227
    },
    {
      "epoch": 4.1224479675292966e-05,
      "step": 43227,
      "training_step_time": 7.5873637199401855
    },
    {
      "epoch": 4.1225433349609376e-05,
      "model_forward_time": 1.7173829078674316,
      "step": 43228
    },
    {
      "epoch": 4.1225433349609376e-05,
      "step": 43228,
      "training_step_time": 7.608457326889038
    },
    {
      "epoch": 4.122638702392578e-05,
      "model_forward_time": 1.6972198486328125,
      "step": 43229
    },
    {
      "epoch": 4.122638702392578e-05,
      "step": 43229,
      "training_step_time": 7.244679689407349
    },
    {
      "epoch": 4.1227340698242185e-05,
      "grad_norm": 0.12842775881290436,
      "learning_rate": 6.5090297716844055e-06,
      "loss": 0.0654,
      "step": 43230
    },
    {
      "epoch": 4.1227340698242185e-05,
      "model_forward_time": 1.7079863548278809,
      "step": 43230
    },
    {
      "dynamics_loss_avg": 0.058362698461860416,
      "epoch": 4.1227340698242185e-05,
      "step": 43230
    },
    {
      "action_loss_avg": 0.0028728156903525814,
      "epoch": 4.1227340698242185e-05,
      "step": 43230
    },
    {
      "epoch": 4.1227340698242185e-05,
      "step": 43230,
      "training_step_time": 6.763565540313721
    },
    {
      "epoch": 4.1228294372558595e-05,
      "model_forward_time": 1.7035267353057861,
      "step": 43231
    },
    {
      "epoch": 4.1228294372558595e-05,
      "step": 43231,
      "training_step_time": 6.758174419403076
    },
    {
      "epoch": 4.1229248046875e-05,
      "model_forward_time": 1.7109038829803467,
      "step": 43232
    },
    {
      "epoch": 4.1229248046875e-05,
      "step": 43232,
      "training_step_time": 7.88912558555603
    },
    {
      "epoch": 4.12302017211914e-05,
      "model_forward_time": 1.7125840187072754,
      "step": 43233
    },
    {
      "epoch": 4.12302017211914e-05,
      "step": 43233,
      "training_step_time": 7.341451644897461
    },
    {
      "epoch": 4.1231155395507814e-05,
      "model_forward_time": 2.81350040435791,
      "step": 43234
    },
    {
      "epoch": 4.1231155395507814e-05,
      "step": 43234,
      "training_step_time": 7.540679693222046
    },
    {
      "epoch": 4.123210906982422e-05,
      "model_forward_time": 1.726332664489746,
      "step": 43235
    },
    {
      "epoch": 4.123210906982422e-05,
      "step": 43235,
      "training_step_time": 10.294986486434937
    },
    {
      "epoch": 4.123306274414062e-05,
      "model_forward_time": 1.7116787433624268,
      "step": 43236
    },
    {
      "epoch": 4.123306274414062e-05,
      "step": 43236,
      "training_step_time": 8.709481000900269
    },
    {
      "epoch": 4.123401641845703e-05,
      "model_forward_time": 1.712336778640747,
      "step": 43237
    },
    {
      "epoch": 4.123401641845703e-05,
      "step": 43237,
      "training_step_time": 8.37000846862793
    },
    {
      "epoch": 4.123497009277344e-05,
      "model_forward_time": 1.704378366470337,
      "step": 43238
    },
    {
      "epoch": 4.123497009277344e-05,
      "step": 43238,
      "training_step_time": 7.001845359802246
    },
    {
      "epoch": 4.123592376708984e-05,
      "model_forward_time": 1.7106585502624512,
      "step": 43239
    },
    {
      "epoch": 4.123592376708984e-05,
      "step": 43239,
      "training_step_time": 7.149660348892212
    },
    {
      "epoch": 4.123687744140625e-05,
      "grad_norm": 0.1650562435388565,
      "learning_rate": 6.507453321713221e-06,
      "loss": 0.0638,
      "step": 43240
    },
    {
      "epoch": 4.123687744140625e-05,
      "model_forward_time": 1.7061767578125,
      "step": 43240
    },
    {
      "dynamics_loss_avg": 0.05952252224087715,
      "epoch": 4.123687744140625e-05,
      "step": 43240
    },
    {
      "action_loss_avg": 0.0015470426646061242,
      "epoch": 4.123687744140625e-05,
      "step": 43240
    },
    {
      "epoch": 4.123687744140625e-05,
      "step": 43240,
      "training_step_time": 6.721445083618164
    },
    {
      "epoch": 4.1237831115722656e-05,
      "model_forward_time": 1.7218031883239746,
      "step": 43241
    },
    {
      "epoch": 4.1237831115722656e-05,
      "step": 43241,
      "training_step_time": 7.870725393295288
    },
    {
      "epoch": 4.123878479003906e-05,
      "model_forward_time": 1.7168240547180176,
      "step": 43242
    },
    {
      "epoch": 4.123878479003906e-05,
      "step": 43242,
      "training_step_time": 7.358126878738403
    },
    {
      "epoch": 4.123973846435547e-05,
      "model_forward_time": 2.834641456604004,
      "step": 43243
    },
    {
      "epoch": 4.123973846435547e-05,
      "step": 43243,
      "training_step_time": 8.322160005569458
    },
    {
      "epoch": 4.1240692138671875e-05,
      "model_forward_time": 1.7128846645355225,
      "step": 43244
    },
    {
      "epoch": 4.1240692138671875e-05,
      "step": 43244,
      "training_step_time": 8.270088195800781
    },
    {
      "epoch": 4.124164581298828e-05,
      "model_forward_time": 1.7247123718261719,
      "step": 43245
    },
    {
      "epoch": 4.124164581298828e-05,
      "step": 43245,
      "training_step_time": 8.657326459884644
    },
    {
      "epoch": 4.124259948730469e-05,
      "model_forward_time": 1.7217862606048584,
      "step": 43246
    },
    {
      "epoch": 4.124259948730469e-05,
      "step": 43246,
      "training_step_time": 7.5854339599609375
    },
    {
      "epoch": 4.1243553161621094e-05,
      "model_forward_time": 1.7174112796783447,
      "step": 43247
    },
    {
      "epoch": 4.1243553161621094e-05,
      "step": 43247,
      "training_step_time": 7.4023425579071045
    },
    {
      "epoch": 4.12445068359375e-05,
      "model_forward_time": 1.7141525745391846,
      "step": 43248
    },
    {
      "epoch": 4.12445068359375e-05,
      "step": 43248,
      "training_step_time": 7.219875335693359
    },
    {
      "epoch": 4.124546051025391e-05,
      "model_forward_time": 1.7139897346496582,
      "step": 43249
    },
    {
      "epoch": 4.124546051025391e-05,
      "step": 43249,
      "training_step_time": 6.748935699462891
    },
    {
      "epoch": 4.124641418457031e-05,
      "grad_norm": 0.11222130060195923,
      "learning_rate": 6.505876706889207e-06,
      "loss": 0.0604,
      "step": 43250
    },
    {
      "epoch": 4.124641418457031e-05,
      "model_forward_time": 1.7155160903930664,
      "step": 43250
    },
    {
      "dynamics_loss_avg": 0.053326488845050334,
      "epoch": 4.124641418457031e-05,
      "step": 43250
    },
    {
      "action_loss_avg": 0.0017107886291341857,
      "epoch": 4.124641418457031e-05,
      "step": 43250
    },
    {
      "epoch": 4.124641418457031e-05,
      "step": 43250,
      "training_step_time": 7.831333637237549
    },
    {
      "epoch": 4.1247367858886717e-05,
      "model_forward_time": 1.7134757041931152,
      "step": 43251
    },
    {
      "epoch": 4.1247367858886717e-05,
      "step": 43251,
      "training_step_time": 8.833054065704346
    },
    {
      "epoch": 4.124832153320313e-05,
      "model_forward_time": 2.824098825454712,
      "step": 43252
    },
    {
      "epoch": 4.124832153320313e-05,
      "step": 43252,
      "training_step_time": 7.607072353363037
    },
    {
      "epoch": 4.124927520751953e-05,
      "model_forward_time": 1.713653564453125,
      "step": 43253
    },
    {
      "epoch": 4.124927520751953e-05,
      "step": 43253,
      "training_step_time": 7.668330430984497
    },
    {
      "epoch": 4.1250228881835935e-05,
      "model_forward_time": 1.7331516742706299,
      "step": 43254
    },
    {
      "epoch": 4.1250228881835935e-05,
      "step": 43254,
      "training_step_time": 7.724873304367065
    },
    {
      "epoch": 4.1251182556152346e-05,
      "model_forward_time": 1.711566686630249,
      "step": 43255
    },
    {
      "epoch": 4.1251182556152346e-05,
      "step": 43255,
      "training_step_time": 8.128072261810303
    },
    {
      "epoch": 4.125213623046875e-05,
      "model_forward_time": 1.7106664180755615,
      "step": 43256
    },
    {
      "epoch": 4.125213623046875e-05,
      "step": 43256,
      "training_step_time": 7.4840123653411865
    },
    {
      "epoch": 4.1253089904785154e-05,
      "model_forward_time": 1.7042129039764404,
      "step": 43257
    },
    {
      "epoch": 4.1253089904785154e-05,
      "step": 43257,
      "training_step_time": 7.124175310134888
    },
    {
      "epoch": 4.1254043579101565e-05,
      "model_forward_time": 1.7074775695800781,
      "step": 43258
    },
    {
      "epoch": 4.1254043579101565e-05,
      "step": 43258,
      "training_step_time": 6.720562696456909
    },
    {
      "epoch": 4.125499725341797e-05,
      "model_forward_time": 1.7097506523132324,
      "step": 43259
    },
    {
      "epoch": 4.125499725341797e-05,
      "step": 43259,
      "training_step_time": 7.587711572647095
    },
    {
      "epoch": 4.125595092773437e-05,
      "grad_norm": 0.10679057240486145,
      "learning_rate": 6.504299927384781e-06,
      "loss": 0.0629,
      "step": 43260
    },
    {
      "epoch": 4.125595092773437e-05,
      "model_forward_time": 1.7171978950500488,
      "step": 43260
    },
    {
      "dynamics_loss_avg": 0.044570327177643775,
      "epoch": 4.125595092773437e-05,
      "step": 43260
    },
    {
      "action_loss_avg": 0.001555143401492387,
      "epoch": 4.125595092773437e-05,
      "step": 43260
    },
    {
      "epoch": 4.125595092773437e-05,
      "step": 43260,
      "training_step_time": 7.857242584228516
    },
    {
      "epoch": 4.1256904602050784e-05,
      "model_forward_time": 2.8177449703216553,
      "step": 43261
    },
    {
      "epoch": 4.1256904602050784e-05,
      "step": 43261,
      "training_step_time": 7.546883583068848
    },
    {
      "epoch": 4.125785827636719e-05,
      "model_forward_time": 1.7099637985229492,
      "step": 43262
    },
    {
      "epoch": 4.125785827636719e-05,
      "step": 43262,
      "training_step_time": 7.582512378692627
    },
    {
      "epoch": 4.125881195068359e-05,
      "model_forward_time": 1.7314815521240234,
      "step": 43263
    },
    {
      "epoch": 4.125881195068359e-05,
      "step": 43263,
      "training_step_time": 8.266886711120605
    },
    {
      "epoch": 4.1259765625e-05,
      "model_forward_time": 1.719322919845581,
      "step": 43264
    },
    {
      "epoch": 4.1259765625e-05,
      "step": 43264,
      "training_step_time": 8.169017314910889
    },
    {
      "epoch": 4.126071929931641e-05,
      "model_forward_time": 1.7215323448181152,
      "step": 43265
    },
    {
      "epoch": 4.126071929931641e-05,
      "step": 43265,
      "training_step_time": 7.606963872909546
    },
    {
      "epoch": 4.126167297363281e-05,
      "model_forward_time": 1.7159020900726318,
      "step": 43266
    },
    {
      "epoch": 4.126167297363281e-05,
      "step": 43266,
      "training_step_time": 7.046861886978149
    },
    {
      "epoch": 4.126262664794922e-05,
      "model_forward_time": 1.717599868774414,
      "step": 43267
    },
    {
      "epoch": 4.126262664794922e-05,
      "step": 43267,
      "training_step_time": 7.115697145462036
    },
    {
      "epoch": 4.1263580322265626e-05,
      "model_forward_time": 1.7101101875305176,
      "step": 43268
    },
    {
      "epoch": 4.1263580322265626e-05,
      "step": 43268,
      "training_step_time": 7.523404836654663
    },
    {
      "epoch": 4.126453399658203e-05,
      "model_forward_time": 1.7176308631896973,
      "step": 43269
    },
    {
      "epoch": 4.126453399658203e-05,
      "step": 43269,
      "training_step_time": 7.886146306991577
    },
    {
      "epoch": 4.126548767089844e-05,
      "grad_norm": 0.12239093333482742,
      "learning_rate": 6.502722983372377e-06,
      "loss": 0.0633,
      "step": 43270
    },
    {
      "epoch": 4.126548767089844e-05,
      "model_forward_time": 1.7470858097076416,
      "step": 43270
    },
    {
      "dynamics_loss_avg": 0.07411793731153012,
      "epoch": 4.126548767089844e-05,
      "step": 43270
    },
    {
      "action_loss_avg": 0.0017966742860153317,
      "epoch": 4.126548767089844e-05,
      "step": 43270
    },
    {
      "epoch": 4.126548767089844e-05,
      "step": 43270,
      "training_step_time": 7.430162191390991
    },
    {
      "epoch": 4.1266441345214845e-05,
      "model_forward_time": 2.8403801918029785,
      "step": 43271
    },
    {
      "epoch": 4.1266441345214845e-05,
      "step": 43271,
      "training_step_time": 8.140928745269775
    },
    {
      "epoch": 4.126739501953125e-05,
      "model_forward_time": 1.7105555534362793,
      "step": 43272
    },
    {
      "epoch": 4.126739501953125e-05,
      "step": 43272,
      "training_step_time": 8.632935762405396
    },
    {
      "epoch": 4.126834869384766e-05,
      "model_forward_time": 1.7255418300628662,
      "step": 43273
    },
    {
      "epoch": 4.126834869384766e-05,
      "step": 43273,
      "training_step_time": 8.883511066436768
    },
    {
      "epoch": 4.126930236816406e-05,
      "model_forward_time": 1.722057819366455,
      "step": 43274
    },
    {
      "epoch": 4.126930236816406e-05,
      "step": 43274,
      "training_step_time": 7.51591944694519
    },
    {
      "epoch": 4.127025604248047e-05,
      "model_forward_time": 1.7127962112426758,
      "step": 43275
    },
    {
      "epoch": 4.127025604248047e-05,
      "step": 43275,
      "training_step_time": 7.481189966201782
    },
    {
      "epoch": 4.127120971679688e-05,
      "model_forward_time": 1.717489242553711,
      "step": 43276
    },
    {
      "epoch": 4.127120971679688e-05,
      "step": 43276,
      "training_step_time": 7.089754581451416
    },
    {
      "epoch": 4.127216339111328e-05,
      "model_forward_time": 1.7159602642059326,
      "step": 43277
    },
    {
      "epoch": 4.127216339111328e-05,
      "step": 43277,
      "training_step_time": 7.079648733139038
    },
    {
      "epoch": 4.1273117065429686e-05,
      "model_forward_time": 1.715264081954956,
      "step": 43278
    },
    {
      "epoch": 4.1273117065429686e-05,
      "step": 43278,
      "training_step_time": 7.744993686676025
    },
    {
      "epoch": 4.12740707397461e-05,
      "model_forward_time": 1.7342853546142578,
      "step": 43279
    },
    {
      "epoch": 4.12740707397461e-05,
      "step": 43279,
      "training_step_time": 7.376672267913818
    },
    {
      "epoch": 4.12750244140625e-05,
      "grad_norm": 0.13544286787509918,
      "learning_rate": 6.501145875024446e-06,
      "loss": 0.0609,
      "step": 43280
    },
    {
      "epoch": 4.12750244140625e-05,
      "model_forward_time": 2.892150402069092,
      "step": 43280
    },
    {
      "dynamics_loss_avg": 0.07203913293778896,
      "epoch": 4.12750244140625e-05,
      "step": 43280
    },
    {
      "action_loss_avg": 0.0019308599672513082,
      "epoch": 4.12750244140625e-05,
      "step": 43280
    },
    {
      "epoch": 4.12750244140625e-05,
      "step": 43280,
      "training_step_time": 8.635215282440186
    },
    {
      "epoch": 4.1275978088378905e-05,
      "model_forward_time": 1.7111148834228516,
      "step": 43281
    },
    {
      "epoch": 4.1275978088378905e-05,
      "step": 43281,
      "training_step_time": 11.284550189971924
    },
    {
      "epoch": 4.127693176269531e-05,
      "model_forward_time": 1.7268731594085693,
      "step": 43282
    },
    {
      "epoch": 4.127693176269531e-05,
      "step": 43282,
      "training_step_time": 10.193279266357422
    },
    {
      "epoch": 4.127788543701172e-05,
      "model_forward_time": 1.714583396911621,
      "step": 43283
    },
    {
      "epoch": 4.127788543701172e-05,
      "step": 43283,
      "training_step_time": 9.167113065719604
    },
    {
      "epoch": 4.1278839111328124e-05,
      "model_forward_time": 1.7130045890808105,
      "step": 43284
    },
    {
      "epoch": 4.1278839111328124e-05,
      "step": 43284,
      "training_step_time": 7.705989599227905
    },
    {
      "epoch": 4.127979278564453e-05,
      "model_forward_time": 1.715043544769287,
      "step": 43285
    },
    {
      "epoch": 4.127979278564453e-05,
      "step": 43285,
      "training_step_time": 7.305067539215088
    },
    {
      "epoch": 4.128074645996094e-05,
      "model_forward_time": 1.7195756435394287,
      "step": 43286
    },
    {
      "epoch": 4.128074645996094e-05,
      "step": 43286,
      "training_step_time": 7.739147186279297
    },
    {
      "epoch": 4.128170013427734e-05,
      "model_forward_time": 1.7127082347869873,
      "step": 43287
    },
    {
      "epoch": 4.128170013427734e-05,
      "step": 43287,
      "training_step_time": 6.89054536819458
    },
    {
      "epoch": 4.128265380859375e-05,
      "model_forward_time": 1.7389962673187256,
      "step": 43288
    },
    {
      "epoch": 4.128265380859375e-05,
      "step": 43288,
      "training_step_time": 7.909906625747681
    },
    {
      "epoch": 4.128360748291016e-05,
      "model_forward_time": 2.8595941066741943,
      "step": 43289
    },
    {
      "epoch": 4.128360748291016e-05,
      "step": 43289,
      "training_step_time": 8.607941150665283
    },
    {
      "epoch": 4.128456115722656e-05,
      "grad_norm": 0.17918939888477325,
      "learning_rate": 6.499568602513459e-06,
      "loss": 0.0626,
      "step": 43290
    },
    {
      "epoch": 4.128456115722656e-05,
      "model_forward_time": 1.7193782329559326,
      "step": 43290
    },
    {
      "dynamics_loss_avg": 0.06579595804214478,
      "epoch": 4.128456115722656e-05,
      "step": 43290
    },
    {
      "action_loss_avg": 0.002500563330249861,
      "epoch": 4.128456115722656e-05,
      "step": 43290
    },
    {
      "epoch": 4.128456115722656e-05,
      "step": 43290,
      "training_step_time": 9.706583738327026
    },
    {
      "epoch": 4.1285514831542966e-05,
      "model_forward_time": 1.7286436557769775,
      "step": 43291
    },
    {
      "epoch": 4.1285514831542966e-05,
      "step": 43291,
      "training_step_time": 9.176891565322876
    },
    {
      "epoch": 4.1286468505859377e-05,
      "model_forward_time": 1.712937355041504,
      "step": 43292
    },
    {
      "epoch": 4.1286468505859377e-05,
      "step": 43292,
      "training_step_time": 8.940720796585083
    },
    {
      "epoch": 4.128742218017578e-05,
      "model_forward_time": 1.7104706764221191,
      "step": 43293
    },
    {
      "epoch": 4.128742218017578e-05,
      "step": 43293,
      "training_step_time": 7.663428783416748
    },
    {
      "epoch": 4.1288375854492185e-05,
      "model_forward_time": 1.723738193511963,
      "step": 43294
    },
    {
      "epoch": 4.1288375854492185e-05,
      "step": 43294,
      "training_step_time": 7.34559440612793
    },
    {
      "epoch": 4.1289329528808595e-05,
      "model_forward_time": 1.7196035385131836,
      "step": 43295
    },
    {
      "epoch": 4.1289329528808595e-05,
      "step": 43295,
      "training_step_time": 7.742326021194458
    },
    {
      "epoch": 4.1290283203125e-05,
      "model_forward_time": 1.723991870880127,
      "step": 43296
    },
    {
      "epoch": 4.1290283203125e-05,
      "step": 43296,
      "training_step_time": 6.688595294952393
    },
    {
      "epoch": 4.1291236877441403e-05,
      "model_forward_time": 1.749204158782959,
      "step": 43297
    },
    {
      "epoch": 4.1291236877441403e-05,
      "step": 43297,
      "training_step_time": 7.817633867263794
    },
    {
      "epoch": 4.1292190551757814e-05,
      "model_forward_time": 2.8081471920013428,
      "step": 43298
    },
    {
      "epoch": 4.1292190551757814e-05,
      "step": 43298,
      "training_step_time": 9.256060123443604
    },
    {
      "epoch": 4.129314422607422e-05,
      "model_forward_time": 1.7200753688812256,
      "step": 43299
    },
    {
      "epoch": 4.129314422607422e-05,
      "step": 43299,
      "training_step_time": 8.651052474975586
    },
    {
      "epoch": 4.129409790039062e-05,
      "grad_norm": 0.1125783771276474,
      "learning_rate": 6.497991166011904e-06,
      "loss": 0.0608,
      "step": 43300
    },
    {
      "epoch": 4.129409790039062e-05,
      "model_forward_time": 1.7186484336853027,
      "step": 43300
    },
    {
      "dynamics_loss_avg": 0.06461192555725574,
      "epoch": 4.129409790039062e-05,
      "step": 43300
    },
    {
      "action_loss_avg": 0.0019261978333815933,
      "epoch": 4.129409790039062e-05,
      "step": 43300
    },
    {
      "epoch": 4.129409790039062e-05,
      "step": 43300,
      "training_step_time": 8.981888771057129
    },
    {
      "epoch": 4.129505157470703e-05,
      "model_forward_time": 1.714893102645874,
      "step": 43301
    },
    {
      "epoch": 4.129505157470703e-05,
      "step": 43301,
      "training_step_time": 10.452024221420288
    },
    {
      "epoch": 4.129600524902344e-05,
      "model_forward_time": 1.722780466079712,
      "step": 43302
    },
    {
      "epoch": 4.129600524902344e-05,
      "step": 43302,
      "training_step_time": 7.583732604980469
    },
    {
      "epoch": 4.129695892333984e-05,
      "model_forward_time": 1.7262392044067383,
      "step": 43303
    },
    {
      "epoch": 4.129695892333984e-05,
      "step": 43303,
      "training_step_time": 7.752787351608276
    },
    {
      "epoch": 4.129791259765625e-05,
      "model_forward_time": 1.7230308055877686,
      "step": 43304
    },
    {
      "epoch": 4.129791259765625e-05,
      "step": 43304,
      "training_step_time": 7.7107861042022705
    },
    {
      "epoch": 4.1298866271972656e-05,
      "model_forward_time": 1.7156596183776855,
      "step": 43305
    },
    {
      "epoch": 4.1298866271972656e-05,
      "step": 43305,
      "training_step_time": 6.785080909729004
    },
    {
      "epoch": 4.129981994628906e-05,
      "model_forward_time": 1.7333884239196777,
      "step": 43306
    },
    {
      "epoch": 4.129981994628906e-05,
      "step": 43306,
      "training_step_time": 7.8518595695495605
    },
    {
      "epoch": 4.130077362060547e-05,
      "model_forward_time": 2.82835054397583,
      "step": 43307
    },
    {
      "epoch": 4.130077362060547e-05,
      "step": 43307,
      "training_step_time": 7.725831031799316
    },
    {
      "epoch": 4.1301727294921875e-05,
      "model_forward_time": 1.7181119918823242,
      "step": 43308
    },
    {
      "epoch": 4.1301727294921875e-05,
      "step": 43308,
      "training_step_time": 7.827793598175049
    },
    {
      "epoch": 4.130268096923828e-05,
      "model_forward_time": 1.7208940982818604,
      "step": 43309
    },
    {
      "epoch": 4.130268096923828e-05,
      "step": 43309,
      "training_step_time": 7.85136079788208
    },
    {
      "epoch": 4.130363464355469e-05,
      "grad_norm": 0.11478112637996674,
      "learning_rate": 6.496413565692289e-06,
      "loss": 0.062,
      "step": 43310
    },
    {
      "epoch": 4.130363464355469e-05,
      "model_forward_time": 1.730154275894165,
      "step": 43310
    },
    {
      "dynamics_loss_avg": 0.05767394686117768,
      "epoch": 4.130363464355469e-05,
      "step": 43310
    },
    {
      "action_loss_avg": 0.0010276601038640364,
      "epoch": 4.130363464355469e-05,
      "step": 43310
    },
    {
      "epoch": 4.130363464355469e-05,
      "step": 43310,
      "training_step_time": 7.889582872390747
    },
    {
      "epoch": 4.1304588317871094e-05,
      "model_forward_time": 1.721768856048584,
      "step": 43311
    },
    {
      "epoch": 4.1304588317871094e-05,
      "step": 43311,
      "training_step_time": 7.727580785751343
    },
    {
      "epoch": 4.13055419921875e-05,
      "model_forward_time": 1.7001116275787354,
      "step": 43312
    },
    {
      "epoch": 4.13055419921875e-05,
      "step": 43312,
      "training_step_time": 7.725524187088013
    },
    {
      "epoch": 4.130649566650391e-05,
      "model_forward_time": 1.705780029296875,
      "step": 43313
    },
    {
      "epoch": 4.130649566650391e-05,
      "step": 43313,
      "training_step_time": 7.7209601402282715
    },
    {
      "epoch": 4.130744934082031e-05,
      "model_forward_time": 1.7034220695495605,
      "step": 43314
    },
    {
      "epoch": 4.130744934082031e-05,
      "step": 43314,
      "training_step_time": 6.828670263290405
    },
    {
      "epoch": 4.1308403015136717e-05,
      "model_forward_time": 1.7250170707702637,
      "step": 43315
    },
    {
      "epoch": 4.1308403015136717e-05,
      "step": 43315,
      "training_step_time": 7.797668218612671
    },
    {
      "epoch": 4.130935668945313e-05,
      "model_forward_time": 1.7035534381866455,
      "step": 43316
    },
    {
      "epoch": 4.130935668945313e-05,
      "step": 43316,
      "training_step_time": 9.368866920471191
    },
    {
      "epoch": 4.131031036376953e-05,
      "model_forward_time": 1.6998531818389893,
      "step": 43317
    },
    {
      "epoch": 4.131031036376953e-05,
      "step": 43317,
      "training_step_time": 7.699519634246826
    },
    {
      "epoch": 4.1311264038085935e-05,
      "model_forward_time": 1.7058053016662598,
      "step": 43318
    },
    {
      "epoch": 4.1311264038085935e-05,
      "step": 43318,
      "training_step_time": 9.952080249786377
    },
    {
      "epoch": 4.1312217712402346e-05,
      "model_forward_time": 1.7219362258911133,
      "step": 43319
    },
    {
      "epoch": 4.1312217712402346e-05,
      "step": 43319,
      "training_step_time": 7.845302104949951
    },
    {
      "epoch": 4.131317138671875e-05,
      "grad_norm": 0.1041184812784195,
      "learning_rate": 6.494835801727135e-06,
      "loss": 0.0627,
      "step": 43320
    },
    {
      "epoch": 4.131317138671875e-05,
      "model_forward_time": 1.7115561962127686,
      "step": 43320
    },
    {
      "dynamics_loss_avg": 0.08447764068841934,
      "epoch": 4.131317138671875e-05,
      "step": 43320
    },
    {
      "action_loss_avg": 0.003307199172559194,
      "epoch": 4.131317138671875e-05,
      "step": 43320
    },
    {
      "epoch": 4.131317138671875e-05,
      "step": 43320,
      "training_step_time": 7.724947929382324
    },
    {
      "epoch": 4.1314125061035154e-05,
      "model_forward_time": 1.702082872390747,
      "step": 43321
    },
    {
      "epoch": 4.1314125061035154e-05,
      "step": 43321,
      "training_step_time": 7.388825178146362
    },
    {
      "epoch": 4.1315078735351565e-05,
      "model_forward_time": 1.7173666954040527,
      "step": 43322
    },
    {
      "epoch": 4.1315078735351565e-05,
      "step": 43322,
      "training_step_time": 7.733797788619995
    },
    {
      "epoch": 4.131603240966797e-05,
      "model_forward_time": 1.724215030670166,
      "step": 43323
    },
    {
      "epoch": 4.131603240966797e-05,
      "step": 43323,
      "training_step_time": 7.7906577587127686
    },
    {
      "epoch": 4.131698608398437e-05,
      "model_forward_time": 1.7458367347717285,
      "step": 43324
    },
    {
      "epoch": 4.131698608398437e-05,
      "step": 43324,
      "training_step_time": 7.67718505859375
    },
    {
      "epoch": 4.1317939758300784e-05,
      "model_forward_time": 1.7160229682922363,
      "step": 43325
    },
    {
      "epoch": 4.1317939758300784e-05,
      "step": 43325,
      "training_step_time": 9.205235481262207
    },
    {
      "epoch": 4.131889343261719e-05,
      "model_forward_time": 2.826770782470703,
      "step": 43326
    },
    {
      "epoch": 4.131889343261719e-05,
      "step": 43326,
      "training_step_time": 8.5936279296875
    },
    {
      "epoch": 4.131984710693359e-05,
      "model_forward_time": 1.7181329727172852,
      "step": 43327
    },
    {
      "epoch": 4.131984710693359e-05,
      "step": 43327,
      "training_step_time": 9.632026195526123
    },
    {
      "epoch": 4.132080078125e-05,
      "model_forward_time": 1.7283835411071777,
      "step": 43328
    },
    {
      "epoch": 4.132080078125e-05,
      "step": 43328,
      "training_step_time": 7.904515266418457
    },
    {
      "epoch": 4.132175445556641e-05,
      "model_forward_time": 1.7138128280639648,
      "step": 43329
    },
    {
      "epoch": 4.132175445556641e-05,
      "step": 43329,
      "training_step_time": 7.7715537548065186
    },
    {
      "epoch": 4.132270812988281e-05,
      "grad_norm": 0.13679003715515137,
      "learning_rate": 6.493257874288985e-06,
      "loss": 0.0561,
      "step": 43330
    },
    {
      "epoch": 4.132270812988281e-05,
      "model_forward_time": 1.7188634872436523,
      "step": 43330
    },
    {
      "dynamics_loss_avg": 0.05654034186154604,
      "epoch": 4.132270812988281e-05,
      "step": 43330
    },
    {
      "action_loss_avg": 0.0015800106368260459,
      "epoch": 4.132270812988281e-05,
      "step": 43330
    },
    {
      "epoch": 4.132270812988281e-05,
      "step": 43330,
      "training_step_time": 7.476125478744507
    },
    {
      "epoch": 4.132366180419922e-05,
      "model_forward_time": 1.7164220809936523,
      "step": 43331
    },
    {
      "epoch": 4.132366180419922e-05,
      "step": 43331,
      "training_step_time": 7.688815593719482
    },
    {
      "epoch": 4.1324615478515626e-05,
      "model_forward_time": 1.729694128036499,
      "step": 43332
    },
    {
      "epoch": 4.1324615478515626e-05,
      "step": 43332,
      "training_step_time": 7.779412269592285
    },
    {
      "epoch": 4.132556915283203e-05,
      "model_forward_time": 1.7373573780059814,
      "step": 43333
    },
    {
      "epoch": 4.132556915283203e-05,
      "step": 43333,
      "training_step_time": 7.473302602767944
    },
    {
      "epoch": 4.132652282714844e-05,
      "model_forward_time": 1.7092690467834473,
      "step": 43334
    },
    {
      "epoch": 4.132652282714844e-05,
      "step": 43334,
      "training_step_time": 7.822131633758545
    },
    {
      "epoch": 4.1327476501464845e-05,
      "model_forward_time": 2.9323863983154297,
      "step": 43335
    },
    {
      "epoch": 4.1327476501464845e-05,
      "step": 43335,
      "training_step_time": 8.688182592391968
    },
    {
      "epoch": 4.132843017578125e-05,
      "model_forward_time": 1.7279202938079834,
      "step": 43336
    },
    {
      "epoch": 4.132843017578125e-05,
      "step": 43336,
      "training_step_time": 8.22282338142395
    },
    {
      "epoch": 4.132938385009766e-05,
      "model_forward_time": 1.7248506546020508,
      "step": 43337
    },
    {
      "epoch": 4.132938385009766e-05,
      "step": 43337,
      "training_step_time": 8.733350038528442
    },
    {
      "epoch": 4.1330337524414063e-05,
      "model_forward_time": 1.7076184749603271,
      "step": 43338
    },
    {
      "epoch": 4.1330337524414063e-05,
      "step": 43338,
      "training_step_time": 7.726385831832886
    },
    {
      "epoch": 4.133129119873047e-05,
      "model_forward_time": 1.708859920501709,
      "step": 43339
    },
    {
      "epoch": 4.133129119873047e-05,
      "step": 43339,
      "training_step_time": 7.473598480224609
    },
    {
      "epoch": 4.133224487304688e-05,
      "grad_norm": 0.1110430657863617,
      "learning_rate": 6.4916797835503995e-06,
      "loss": 0.0595,
      "step": 43340
    },
    {
      "epoch": 4.133224487304688e-05,
      "model_forward_time": 1.7051618099212646,
      "step": 43340
    },
    {
      "dynamics_loss_avg": 0.044369492679834366,
      "epoch": 4.133224487304688e-05,
      "step": 43340
    },
    {
      "action_loss_avg": 0.001515110206673853,
      "epoch": 4.133224487304688e-05,
      "step": 43340
    },
    {
      "epoch": 4.133224487304688e-05,
      "step": 43340,
      "training_step_time": 7.73027777671814
    },
    {
      "epoch": 4.133319854736328e-05,
      "model_forward_time": 1.7080028057098389,
      "step": 43341
    },
    {
      "epoch": 4.133319854736328e-05,
      "step": 43341,
      "training_step_time": 7.732486009597778
    },
    {
      "epoch": 4.1334152221679686e-05,
      "model_forward_time": 1.7092766761779785,
      "step": 43342
    },
    {
      "epoch": 4.1334152221679686e-05,
      "step": 43342,
      "training_step_time": 6.811784982681274
    },
    {
      "epoch": 4.13351058959961e-05,
      "model_forward_time": 1.734832763671875,
      "step": 43343
    },
    {
      "epoch": 4.13351058959961e-05,
      "step": 43343,
      "training_step_time": 7.754566669464111
    },
    {
      "epoch": 4.13360595703125e-05,
      "model_forward_time": 2.8911776542663574,
      "step": 43344
    },
    {
      "epoch": 4.13360595703125e-05,
      "step": 43344,
      "training_step_time": 7.829249382019043
    },
    {
      "epoch": 4.1337013244628905e-05,
      "model_forward_time": 1.7093055248260498,
      "step": 43345
    },
    {
      "epoch": 4.1337013244628905e-05,
      "step": 43345,
      "training_step_time": 9.687986612319946
    },
    {
      "epoch": 4.133796691894531e-05,
      "model_forward_time": 1.7164978981018066,
      "step": 43346
    },
    {
      "epoch": 4.133796691894531e-05,
      "step": 43346,
      "training_step_time": 8.517774105072021
    },
    {
      "epoch": 4.133892059326172e-05,
      "model_forward_time": 1.7105402946472168,
      "step": 43347
    },
    {
      "epoch": 4.133892059326172e-05,
      "step": 43347,
      "training_step_time": 8.38834524154663
    },
    {
      "epoch": 4.1339874267578124e-05,
      "model_forward_time": 1.7102549076080322,
      "step": 43348
    },
    {
      "epoch": 4.1339874267578124e-05,
      "step": 43348,
      "training_step_time": 7.719484806060791
    },
    {
      "epoch": 4.134082794189453e-05,
      "model_forward_time": 1.7054991722106934,
      "step": 43349
    },
    {
      "epoch": 4.134082794189453e-05,
      "step": 43349,
      "training_step_time": 7.746598958969116
    },
    {
      "epoch": 4.134178161621094e-05,
      "grad_norm": 0.14749903976917267,
      "learning_rate": 6.490101529683956e-06,
      "loss": 0.0591,
      "step": 43350
    },
    {
      "epoch": 4.134178161621094e-05,
      "model_forward_time": 1.7026801109313965,
      "step": 43350
    },
    {
      "dynamics_loss_avg": 0.06401830613613128,
      "epoch": 4.134178161621094e-05,
      "step": 43350
    },
    {
      "action_loss_avg": 0.001667343234294094,
      "epoch": 4.134178161621094e-05,
      "step": 43350
    },
    {
      "epoch": 4.134178161621094e-05,
      "step": 43350,
      "training_step_time": 7.8316521644592285
    },
    {
      "epoch": 4.134273529052734e-05,
      "model_forward_time": 1.7123908996582031,
      "step": 43351
    },
    {
      "epoch": 4.134273529052734e-05,
      "step": 43351,
      "training_step_time": 6.86440372467041
    },
    {
      "epoch": 4.134368896484375e-05,
      "model_forward_time": 1.7263636589050293,
      "step": 43352
    },
    {
      "epoch": 4.134368896484375e-05,
      "step": 43352,
      "training_step_time": 7.787946939468384
    },
    {
      "epoch": 4.134464263916016e-05,
      "model_forward_time": 2.9043774604797363,
      "step": 43353
    },
    {
      "epoch": 4.134464263916016e-05,
      "step": 43353,
      "training_step_time": 7.840084791183472
    },
    {
      "epoch": 4.134559631347656e-05,
      "model_forward_time": 1.709717035293579,
      "step": 43354
    },
    {
      "epoch": 4.134559631347656e-05,
      "step": 43354,
      "training_step_time": 7.703719615936279
    },
    {
      "epoch": 4.1346549987792966e-05,
      "model_forward_time": 1.705275535583496,
      "step": 43355
    },
    {
      "epoch": 4.1346549987792966e-05,
      "step": 43355,
      "training_step_time": 9.75885820388794
    },
    {
      "epoch": 4.1347503662109377e-05,
      "model_forward_time": 1.727827548980713,
      "step": 43356
    },
    {
      "epoch": 4.1347503662109377e-05,
      "step": 43356,
      "training_step_time": 7.885401248931885
    },
    {
      "epoch": 4.134845733642578e-05,
      "model_forward_time": 1.7120563983917236,
      "step": 43357
    },
    {
      "epoch": 4.134845733642578e-05,
      "step": 43357,
      "training_step_time": 7.760278224945068
    },
    {
      "epoch": 4.1349411010742185e-05,
      "model_forward_time": 1.7036683559417725,
      "step": 43358
    },
    {
      "epoch": 4.1349411010742185e-05,
      "step": 43358,
      "training_step_time": 7.594339847564697
    },
    {
      "epoch": 4.1350364685058595e-05,
      "model_forward_time": 1.7084367275238037,
      "step": 43359
    },
    {
      "epoch": 4.1350364685058595e-05,
      "step": 43359,
      "training_step_time": 7.649925231933594
    },
    {
      "epoch": 4.1351318359375e-05,
      "grad_norm": 0.11704700440168381,
      "learning_rate": 6.488523112862248e-06,
      "loss": 0.0589,
      "step": 43360
    },
    {
      "epoch": 4.1351318359375e-05,
      "model_forward_time": 1.7098748683929443,
      "step": 43360
    },
    {
      "dynamics_loss_avg": 0.05224700048565865,
      "epoch": 4.1351318359375e-05,
      "step": 43360
    },
    {
      "action_loss_avg": 0.0016122326691402123,
      "epoch": 4.1351318359375e-05,
      "step": 43360
    },
    {
      "epoch": 4.1351318359375e-05,
      "step": 43360,
      "training_step_time": 6.746403455734253
    },
    {
      "epoch": 4.1352272033691403e-05,
      "model_forward_time": 1.7149608135223389,
      "step": 43361
    },
    {
      "epoch": 4.1352272033691403e-05,
      "step": 43361,
      "training_step_time": 8.3882315158844
    },
    {
      "epoch": 4.1353225708007814e-05,
      "model_forward_time": 2.9017810821533203,
      "step": 43362
    },
    {
      "epoch": 4.1353225708007814e-05,
      "step": 43362,
      "training_step_time": 7.768025159835815
    },
    {
      "epoch": 4.135417938232422e-05,
      "model_forward_time": 1.7115354537963867,
      "step": 43363
    },
    {
      "epoch": 4.135417938232422e-05,
      "step": 43363,
      "training_step_time": 8.462777614593506
    },
    {
      "epoch": 4.135513305664062e-05,
      "model_forward_time": 1.7065649032592773,
      "step": 43364
    },
    {
      "epoch": 4.135513305664062e-05,
      "step": 43364,
      "training_step_time": 9.166063785552979
    },
    {
      "epoch": 4.135608673095703e-05,
      "model_forward_time": 1.72536039352417,
      "step": 43365
    },
    {
      "epoch": 4.135608673095703e-05,
      "step": 43365,
      "training_step_time": 8.747352361679077
    },
    {
      "epoch": 4.135704040527344e-05,
      "model_forward_time": 1.7000856399536133,
      "step": 43366
    },
    {
      "epoch": 4.135704040527344e-05,
      "step": 43366,
      "training_step_time": 8.318844318389893
    },
    {
      "epoch": 4.135799407958984e-05,
      "model_forward_time": 1.7086756229400635,
      "step": 43367
    },
    {
      "epoch": 4.135799407958984e-05,
      "step": 43367,
      "training_step_time": 7.659923553466797
    },
    {
      "epoch": 4.135894775390625e-05,
      "model_forward_time": 1.7062625885009766,
      "step": 43368
    },
    {
      "epoch": 4.135894775390625e-05,
      "step": 43368,
      "training_step_time": 7.361106634140015
    },
    {
      "epoch": 4.1359901428222656e-05,
      "model_forward_time": 1.7097573280334473,
      "step": 43369
    },
    {
      "epoch": 4.1359901428222656e-05,
      "step": 43369,
      "training_step_time": 7.735363006591797
    },
    {
      "epoch": 4.136085510253906e-05,
      "grad_norm": 0.08454179763793945,
      "learning_rate": 6.486944533257892e-06,
      "loss": 0.0588,
      "step": 43370
    },
    {
      "epoch": 4.136085510253906e-05,
      "model_forward_time": 1.7015013694763184,
      "step": 43370
    },
    {
      "dynamics_loss_avg": 0.06940452121198178,
      "epoch": 4.136085510253906e-05,
      "step": 43370
    },
    {
      "action_loss_avg": 0.0018072092731017619,
      "epoch": 4.136085510253906e-05,
      "step": 43370
    },
    {
      "epoch": 4.136085510253906e-05,
      "step": 43370,
      "training_step_time": 8.878209114074707
    },
    {
      "epoch": 4.136180877685547e-05,
      "model_forward_time": 2.84812331199646,
      "step": 43371
    },
    {
      "epoch": 4.136180877685547e-05,
      "step": 43371,
      "training_step_time": 7.771432399749756
    },
    {
      "epoch": 4.1362762451171875e-05,
      "model_forward_time": 1.717815637588501,
      "step": 43372
    },
    {
      "epoch": 4.1362762451171875e-05,
      "step": 43372,
      "training_step_time": 8.351568698883057
    },
    {
      "epoch": 4.136371612548828e-05,
      "model_forward_time": 1.716799020767212,
      "step": 43373
    },
    {
      "epoch": 4.136371612548828e-05,
      "step": 43373,
      "training_step_time": 9.947923421859741
    },
    {
      "epoch": 4.136466979980469e-05,
      "model_forward_time": 1.7239603996276855,
      "step": 43374
    },
    {
      "epoch": 4.136466979980469e-05,
      "step": 43374,
      "training_step_time": 8.774990797042847
    },
    {
      "epoch": 4.1365623474121094e-05,
      "model_forward_time": 1.7088747024536133,
      "step": 43375
    },
    {
      "epoch": 4.1365623474121094e-05,
      "step": 43375,
      "training_step_time": 7.718424558639526
    },
    {
      "epoch": 4.13665771484375e-05,
      "model_forward_time": 1.707075595855713,
      "step": 43376
    },
    {
      "epoch": 4.13665771484375e-05,
      "step": 43376,
      "training_step_time": 7.698450088500977
    },
    {
      "epoch": 4.136753082275391e-05,
      "model_forward_time": 1.7081098556518555,
      "step": 43377
    },
    {
      "epoch": 4.136753082275391e-05,
      "step": 43377,
      "training_step_time": 7.732548952102661
    },
    {
      "epoch": 4.136848449707031e-05,
      "model_forward_time": 1.7083165645599365,
      "step": 43378
    },
    {
      "epoch": 4.136848449707031e-05,
      "step": 43378,
      "training_step_time": 7.7272865772247314
    },
    {
      "epoch": 4.136943817138672e-05,
      "model_forward_time": 1.7070305347442627,
      "step": 43379
    },
    {
      "epoch": 4.136943817138672e-05,
      "step": 43379,
      "training_step_time": 7.759904384613037
    },
    {
      "epoch": 4.137039184570313e-05,
      "grad_norm": 0.11087768524885178,
      "learning_rate": 6.485365791043518e-06,
      "loss": 0.0577,
      "step": 43380
    },
    {
      "epoch": 4.137039184570313e-05,
      "model_forward_time": 2.8626725673675537,
      "step": 43380
    },
    {
      "dynamics_loss_avg": 0.060131194349378345,
      "epoch": 4.137039184570313e-05,
      "step": 43380
    },
    {
      "action_loss_avg": 0.001617502755834721,
      "epoch": 4.137039184570313e-05,
      "step": 43380
    },
    {
      "epoch": 4.137039184570313e-05,
      "step": 43380,
      "training_step_time": 7.789805173873901
    },
    {
      "epoch": 4.137134552001953e-05,
      "model_forward_time": 1.7023718357086182,
      "step": 43381
    },
    {
      "epoch": 4.137134552001953e-05,
      "step": 43381,
      "training_step_time": 9.35390043258667
    },
    {
      "epoch": 4.1372299194335935e-05,
      "model_forward_time": 1.7069041728973389,
      "step": 43382
    },
    {
      "epoch": 4.1372299194335935e-05,
      "step": 43382,
      "training_step_time": 8.441739320755005
    },
    {
      "epoch": 4.1373252868652346e-05,
      "model_forward_time": 1.7151553630828857,
      "step": 43383
    },
    {
      "epoch": 4.1373252868652346e-05,
      "step": 43383,
      "training_step_time": 10.371617555618286
    },
    {
      "epoch": 4.137420654296875e-05,
      "model_forward_time": 1.6981053352355957,
      "step": 43384
    },
    {
      "epoch": 4.137420654296875e-05,
      "step": 43384,
      "training_step_time": 7.907052516937256
    },
    {
      "epoch": 4.1375160217285154e-05,
      "model_forward_time": 1.6908211708068848,
      "step": 43385
    },
    {
      "epoch": 4.1375160217285154e-05,
      "step": 43385,
      "training_step_time": 7.754760026931763
    },
    {
      "epoch": 4.1376113891601565e-05,
      "model_forward_time": 1.7068655490875244,
      "step": 43386
    },
    {
      "epoch": 4.1376113891601565e-05,
      "step": 43386,
      "training_step_time": 8.375485181808472
    },
    {
      "epoch": 4.137706756591797e-05,
      "model_forward_time": 1.7088801860809326,
      "step": 43387
    },
    {
      "epoch": 4.137706756591797e-05,
      "step": 43387,
      "training_step_time": 7.786470890045166
    },
    {
      "epoch": 4.137802124023437e-05,
      "model_forward_time": 1.7052621841430664,
      "step": 43388
    },
    {
      "epoch": 4.137802124023437e-05,
      "step": 43388,
      "training_step_time": 8.293062925338745
    },
    {
      "epoch": 4.1378974914550784e-05,
      "model_forward_time": 2.9225149154663086,
      "step": 43389
    },
    {
      "epoch": 4.1378974914550784e-05,
      "step": 43389,
      "training_step_time": 8.574392795562744
    },
    {
      "epoch": 4.137992858886719e-05,
      "grad_norm": 0.12216861546039581,
      "learning_rate": 6.483786886391772e-06,
      "loss": 0.0597,
      "step": 43390
    },
    {
      "epoch": 4.137992858886719e-05,
      "model_forward_time": 1.7130520343780518,
      "step": 43390
    },
    {
      "dynamics_loss_avg": 0.05750309582799673,
      "epoch": 4.137992858886719e-05,
      "step": 43390
    },
    {
      "action_loss_avg": 0.0014698360348120332,
      "epoch": 4.137992858886719e-05,
      "step": 43390
    },
    {
      "epoch": 4.137992858886719e-05,
      "step": 43390,
      "training_step_time": 9.606051445007324
    },
    {
      "epoch": 4.138088226318359e-05,
      "model_forward_time": 1.7074155807495117,
      "step": 43391
    },
    {
      "epoch": 4.138088226318359e-05,
      "step": 43391,
      "training_step_time": 9.165819644927979
    },
    {
      "epoch": 4.13818359375e-05,
      "model_forward_time": 1.7088396549224854,
      "step": 43392
    },
    {
      "epoch": 4.13818359375e-05,
      "step": 43392,
      "training_step_time": 8.63776183128357
    },
    {
      "epoch": 4.138278961181641e-05,
      "model_forward_time": 1.7238943576812744,
      "step": 43393
    },
    {
      "epoch": 4.138278961181641e-05,
      "step": 43393,
      "training_step_time": 8.68007493019104
    },
    {
      "epoch": 4.138374328613281e-05,
      "model_forward_time": 1.701603889465332,
      "step": 43394
    },
    {
      "epoch": 4.138374328613281e-05,
      "step": 43394,
      "training_step_time": 7.831743240356445
    },
    {
      "epoch": 4.138469696044922e-05,
      "model_forward_time": 1.7069611549377441,
      "step": 43395
    },
    {
      "epoch": 4.138469696044922e-05,
      "step": 43395,
      "training_step_time": 7.684112787246704
    },
    {
      "epoch": 4.1385650634765626e-05,
      "model_forward_time": 1.7105138301849365,
      "step": 43396
    },
    {
      "epoch": 4.1385650634765626e-05,
      "step": 43396,
      "training_step_time": 7.760698556900024
    },
    {
      "epoch": 4.138660430908203e-05,
      "model_forward_time": 1.708057165145874,
      "step": 43397
    },
    {
      "epoch": 4.138660430908203e-05,
      "step": 43397,
      "training_step_time": 7.266354560852051
    },
    {
      "epoch": 4.138755798339844e-05,
      "model_forward_time": 2.9351532459259033,
      "step": 43398
    },
    {
      "epoch": 4.138755798339844e-05,
      "step": 43398,
      "training_step_time": 7.850828170776367
    },
    {
      "epoch": 4.1388511657714845e-05,
      "model_forward_time": 1.7099556922912598,
      "step": 43399
    },
    {
      "epoch": 4.1388511657714845e-05,
      "step": 43399,
      "training_step_time": 7.812278985977173
    },
    {
      "epoch": 4.138946533203125e-05,
      "grad_norm": 0.15470652282238007,
      "learning_rate": 6.482207819475324e-06,
      "loss": 0.0614,
      "step": 43400
    },
    {
      "epoch": 4.138946533203125e-05,
      "model_forward_time": 1.713606834411621,
      "step": 43400
    },
    {
      "dynamics_loss_avg": 0.056464449502527714,
      "epoch": 4.138946533203125e-05,
      "step": 43400
    },
    {
      "action_loss_avg": 0.0016412592405686155,
      "epoch": 4.138946533203125e-05,
      "step": 43400
    },
    {
      "epoch": 4.138946533203125e-05,
      "step": 43400,
      "training_step_time": 8.6666738986969
    },
    {
      "epoch": 4.139041900634766e-05,
      "model_forward_time": 1.7056314945220947,
      "step": 43401
    },
    {
      "epoch": 4.139041900634766e-05,
      "step": 43401,
      "training_step_time": 7.796402215957642
    },
    {
      "epoch": 4.1391372680664063e-05,
      "model_forward_time": 1.7308909893035889,
      "step": 43402
    },
    {
      "epoch": 4.1391372680664063e-05,
      "step": 43402,
      "training_step_time": 9.965082168579102
    },
    {
      "epoch": 4.139232635498047e-05,
      "model_forward_time": 1.7044978141784668,
      "step": 43403
    },
    {
      "epoch": 4.139232635498047e-05,
      "step": 43403,
      "training_step_time": 7.67449164390564
    },
    {
      "epoch": 4.139328002929688e-05,
      "model_forward_time": 1.7118010520935059,
      "step": 43404
    },
    {
      "epoch": 4.139328002929688e-05,
      "step": 43404,
      "training_step_time": 7.609473466873169
    },
    {
      "epoch": 4.139423370361328e-05,
      "model_forward_time": 1.7051587104797363,
      "step": 43405
    },
    {
      "epoch": 4.139423370361328e-05,
      "step": 43405,
      "training_step_time": 7.712794542312622
    },
    {
      "epoch": 4.1395187377929686e-05,
      "model_forward_time": 1.7041573524475098,
      "step": 43406
    },
    {
      "epoch": 4.1395187377929686e-05,
      "step": 43406,
      "training_step_time": 7.223439931869507
    },
    {
      "epoch": 4.139614105224609e-05,
      "model_forward_time": 2.8693370819091797,
      "step": 43407
    },
    {
      "epoch": 4.139614105224609e-05,
      "step": 43407,
      "training_step_time": 8.598408937454224
    },
    {
      "epoch": 4.13970947265625e-05,
      "model_forward_time": 1.7135026454925537,
      "step": 43408
    },
    {
      "epoch": 4.13970947265625e-05,
      "step": 43408,
      "training_step_time": 8.22406816482544
    },
    {
      "epoch": 4.1398048400878905e-05,
      "model_forward_time": 1.7066004276275635,
      "step": 43409
    },
    {
      "epoch": 4.1398048400878905e-05,
      "step": 43409,
      "training_step_time": 7.798454761505127
    },
    {
      "epoch": 4.139900207519531e-05,
      "grad_norm": 0.10889308899641037,
      "learning_rate": 6.4806285904668575e-06,
      "loss": 0.0602,
      "step": 43410
    },
    {
      "epoch": 4.139900207519531e-05,
      "model_forward_time": 1.711761713027954,
      "step": 43410
    },
    {
      "dynamics_loss_avg": 0.05958686377853155,
      "epoch": 4.139900207519531e-05,
      "step": 43410
    },
    {
      "action_loss_avg": 0.001593240664806217,
      "epoch": 4.139900207519531e-05,
      "step": 43410
    },
    {
      "epoch": 4.139900207519531e-05,
      "step": 43410,
      "training_step_time": 7.814405679702759
    },
    {
      "epoch": 4.139995574951172e-05,
      "model_forward_time": 1.7325077056884766,
      "step": 43411
    },
    {
      "epoch": 4.139995574951172e-05,
      "step": 43411,
      "training_step_time": 7.8227691650390625
    },
    {
      "epoch": 4.1400909423828124e-05,
      "model_forward_time": 1.7063605785369873,
      "step": 43412
    },
    {
      "epoch": 4.1400909423828124e-05,
      "step": 43412,
      "training_step_time": 7.968163251876831
    },
    {
      "epoch": 4.140186309814453e-05,
      "model_forward_time": 1.7067899703979492,
      "step": 43413
    },
    {
      "epoch": 4.140186309814453e-05,
      "step": 43413,
      "training_step_time": 7.5980565547943115
    },
    {
      "epoch": 4.140281677246094e-05,
      "model_forward_time": 1.7142751216888428,
      "step": 43414
    },
    {
      "epoch": 4.140281677246094e-05,
      "step": 43414,
      "training_step_time": 7.776484489440918
    },
    {
      "epoch": 4.140377044677734e-05,
      "model_forward_time": 1.704890489578247,
      "step": 43415
    },
    {
      "epoch": 4.140377044677734e-05,
      "step": 43415,
      "training_step_time": 7.245824575424194
    },
    {
      "epoch": 4.140472412109375e-05,
      "model_forward_time": 2.902919054031372,
      "step": 43416
    },
    {
      "epoch": 4.140472412109375e-05,
      "step": 43416,
      "training_step_time": 7.866258144378662
    },
    {
      "epoch": 4.140567779541016e-05,
      "model_forward_time": 1.7300598621368408,
      "step": 43417
    },
    {
      "epoch": 4.140567779541016e-05,
      "step": 43417,
      "training_step_time": 7.507035970687866
    },
    {
      "epoch": 4.140663146972656e-05,
      "model_forward_time": 1.6984639167785645,
      "step": 43418
    },
    {
      "epoch": 4.140663146972656e-05,
      "step": 43418,
      "training_step_time": 7.753492593765259
    },
    {
      "epoch": 4.1407585144042966e-05,
      "model_forward_time": 1.7108087539672852,
      "step": 43419
    },
    {
      "epoch": 4.1407585144042966e-05,
      "step": 43419,
      "training_step_time": 8.70923662185669
    },
    {
      "epoch": 4.140853881835938e-05,
      "grad_norm": 0.12327830493450165,
      "learning_rate": 6.479049199539073e-06,
      "loss": 0.0601,
      "step": 43420
    },
    {
      "epoch": 4.140853881835938e-05,
      "model_forward_time": 1.3303911685943604,
      "step": 43420
    },
    {
      "dynamics_loss_avg": 0.06836753729730845,
      "epoch": 4.140853881835938e-05,
      "step": 43420
    },
    {
      "action_loss_avg": 0.0013057900039711967,
      "epoch": 4.140853881835938e-05,
      "step": 43420
    },
    {
      "epoch": 4.140853881835938e-05,
      "step": 43420,
      "training_step_time": 9.059856414794922
    },
    {
      "epoch": 4.140949249267578e-05,
      "model_forward_time": 1.7115814685821533,
      "step": 43421
    },
    {
      "epoch": 4.140949249267578e-05,
      "step": 43421,
      "training_step_time": 8.256254434585571
    },
    {
      "epoch": 4.1410446166992185e-05,
      "model_forward_time": 1.7038590908050537,
      "step": 43422
    },
    {
      "epoch": 4.1410446166992185e-05,
      "step": 43422,
      "training_step_time": 7.8795881271362305
    },
    {
      "epoch": 4.1411399841308595e-05,
      "model_forward_time": 1.713313102722168,
      "step": 43423
    },
    {
      "epoch": 4.1411399841308595e-05,
      "step": 43423,
      "training_step_time": 7.759196758270264
    },
    {
      "epoch": 4.1412353515625e-05,
      "model_forward_time": 1.7052559852600098,
      "step": 43424
    },
    {
      "epoch": 4.1412353515625e-05,
      "step": 43424,
      "training_step_time": 7.437525272369385
    },
    {
      "epoch": 4.1413307189941404e-05,
      "model_forward_time": 2.83133602142334,
      "step": 43425
    },
    {
      "epoch": 4.1413307189941404e-05,
      "step": 43425,
      "training_step_time": 7.744466066360474
    },
    {
      "epoch": 4.1414260864257814e-05,
      "model_forward_time": 1.7321932315826416,
      "step": 43426
    },
    {
      "epoch": 4.1414260864257814e-05,
      "step": 43426,
      "training_step_time": 7.476835489273071
    },
    {
      "epoch": 4.141521453857422e-05,
      "model_forward_time": 1.7053077220916748,
      "step": 43427
    },
    {
      "epoch": 4.141521453857422e-05,
      "step": 43427,
      "training_step_time": 8.596760272979736
    },
    {
      "epoch": 4.141616821289062e-05,
      "model_forward_time": 1.7120180130004883,
      "step": 43428
    },
    {
      "epoch": 4.141616821289062e-05,
      "step": 43428,
      "training_step_time": 9.366248607635498
    },
    {
      "epoch": 4.141712188720703e-05,
      "model_forward_time": 1.7084667682647705,
      "step": 43429
    },
    {
      "epoch": 4.141712188720703e-05,
      "step": 43429,
      "training_step_time": 9.026182413101196
    },
    {
      "epoch": 4.141807556152344e-05,
      "grad_norm": 0.11765152961015701,
      "learning_rate": 6.477469646864693e-06,
      "loss": 0.0619,
      "step": 43430
    },
    {
      "epoch": 4.141807556152344e-05,
      "model_forward_time": 1.7852375507354736,
      "step": 43430
    },
    {
      "dynamics_loss_avg": 0.05512427557259798,
      "epoch": 4.141807556152344e-05,
      "step": 43430
    },
    {
      "action_loss_avg": 0.001212128798943013,
      "epoch": 4.141807556152344e-05,
      "step": 43430
    },
    {
      "epoch": 4.141807556152344e-05,
      "step": 43430,
      "training_step_time": 8.382992267608643
    },
    {
      "epoch": 4.141902923583984e-05,
      "model_forward_time": 1.7308323383331299,
      "step": 43431
    },
    {
      "epoch": 4.141902923583984e-05,
      "step": 43431,
      "training_step_time": 7.6640095710754395
    },
    {
      "epoch": 4.141998291015625e-05,
      "model_forward_time": 1.7241103649139404,
      "step": 43432
    },
    {
      "epoch": 4.141998291015625e-05,
      "step": 43432,
      "training_step_time": 7.756201982498169
    },
    {
      "epoch": 4.1420936584472656e-05,
      "model_forward_time": 1.7258846759796143,
      "step": 43433
    },
    {
      "epoch": 4.1420936584472656e-05,
      "step": 43433,
      "training_step_time": 7.6076672077178955
    },
    {
      "epoch": 4.142189025878906e-05,
      "model_forward_time": 2.8656768798828125,
      "step": 43434
    },
    {
      "epoch": 4.142189025878906e-05,
      "step": 43434,
      "training_step_time": 7.754513263702393
    },
    {
      "epoch": 4.142284393310547e-05,
      "model_forward_time": 1.3502106666564941,
      "step": 43435
    },
    {
      "epoch": 4.142284393310547e-05,
      "step": 43435,
      "training_step_time": 7.745369911193848
    },
    {
      "epoch": 4.1423797607421875e-05,
      "model_forward_time": 1.7179644107818604,
      "step": 43436
    },
    {
      "epoch": 4.1423797607421875e-05,
      "step": 43436,
      "training_step_time": 7.827970504760742
    },
    {
      "epoch": 4.142475128173828e-05,
      "model_forward_time": 1.720731258392334,
      "step": 43437
    },
    {
      "epoch": 4.142475128173828e-05,
      "step": 43437,
      "training_step_time": 8.895353078842163
    },
    {
      "epoch": 4.142570495605469e-05,
      "model_forward_time": 1.717390537261963,
      "step": 43438
    },
    {
      "epoch": 4.142570495605469e-05,
      "step": 43438,
      "training_step_time": 8.4129159450531
    },
    {
      "epoch": 4.1426658630371094e-05,
      "model_forward_time": 1.772463083267212,
      "step": 43439
    },
    {
      "epoch": 4.1426658630371094e-05,
      "step": 43439,
      "training_step_time": 9.41421627998352
    },
    {
      "epoch": 4.14276123046875e-05,
      "grad_norm": 0.13844846189022064,
      "learning_rate": 6.475889932616455e-06,
      "loss": 0.0578,
      "step": 43440
    },
    {
      "epoch": 4.14276123046875e-05,
      "model_forward_time": 1.7091147899627686,
      "step": 43440
    },
    {
      "dynamics_loss_avg": 0.034149206010624766,
      "epoch": 4.14276123046875e-05,
      "step": 43440
    },
    {
      "action_loss_avg": 0.0012741918733809143,
      "epoch": 4.14276123046875e-05,
      "step": 43440
    },
    {
      "epoch": 4.14276123046875e-05,
      "step": 43440,
      "training_step_time": 8.629087448120117
    },
    {
      "epoch": 4.142856597900391e-05,
      "model_forward_time": 1.7187108993530273,
      "step": 43441
    },
    {
      "epoch": 4.142856597900391e-05,
      "step": 43441,
      "training_step_time": 7.691839694976807
    },
    {
      "epoch": 4.142951965332031e-05,
      "model_forward_time": 1.7191352844238281,
      "step": 43442
    },
    {
      "epoch": 4.142951965332031e-05,
      "step": 43442,
      "training_step_time": 7.725068807601929
    },
    {
      "epoch": 4.143047332763672e-05,
      "model_forward_time": 2.8082449436187744,
      "step": 43443
    },
    {
      "epoch": 4.143047332763672e-05,
      "step": 43443,
      "training_step_time": 7.714838266372681
    },
    {
      "epoch": 4.143142700195313e-05,
      "model_forward_time": 1.7159943580627441,
      "step": 43444
    },
    {
      "epoch": 4.143142700195313e-05,
      "step": 43444,
      "training_step_time": 7.763535022735596
    },
    {
      "epoch": 4.143238067626953e-05,
      "model_forward_time": 1.7131602764129639,
      "step": 43445
    },
    {
      "epoch": 4.143238067626953e-05,
      "step": 43445,
      "training_step_time": 9.926869869232178
    },
    {
      "epoch": 4.1433334350585936e-05,
      "model_forward_time": 1.7167706489562988,
      "step": 43446
    },
    {
      "epoch": 4.1433334350585936e-05,
      "step": 43446,
      "training_step_time": 8.507448196411133
    },
    {
      "epoch": 4.1434288024902346e-05,
      "model_forward_time": 1.7244510650634766,
      "step": 43447
    },
    {
      "epoch": 4.1434288024902346e-05,
      "step": 43447,
      "training_step_time": 8.739151000976562
    },
    {
      "epoch": 4.143524169921875e-05,
      "model_forward_time": 1.7910103797912598,
      "step": 43448
    },
    {
      "epoch": 4.143524169921875e-05,
      "step": 43448,
      "training_step_time": 7.907835960388184
    },
    {
      "epoch": 4.1436195373535154e-05,
      "model_forward_time": 1.7134082317352295,
      "step": 43449
    },
    {
      "epoch": 4.1436195373535154e-05,
      "step": 43449,
      "training_step_time": 7.844659090042114
    },
    {
      "epoch": 4.1437149047851565e-05,
      "grad_norm": 0.12914294004440308,
      "learning_rate": 6.474310056967111e-06,
      "loss": 0.0601,
      "step": 43450
    },
    {
      "epoch": 4.1437149047851565e-05,
      "model_forward_time": 1.7090866565704346,
      "step": 43450
    },
    {
      "dynamics_loss_avg": 0.05617305245250463,
      "epoch": 4.1437149047851565e-05,
      "step": 43450
    },
    {
      "action_loss_avg": 0.0014652392914285884,
      "epoch": 4.1437149047851565e-05,
      "step": 43450
    },
    {
      "epoch": 4.1437149047851565e-05,
      "step": 43450,
      "training_step_time": 8.165499687194824
    },
    {
      "epoch": 4.143810272216797e-05,
      "model_forward_time": 1.7162339687347412,
      "step": 43451
    },
    {
      "epoch": 4.143810272216797e-05,
      "step": 43451,
      "training_step_time": 7.867014408111572
    },
    {
      "epoch": 4.143905639648437e-05,
      "model_forward_time": 1.7193150520324707,
      "step": 43452
    },
    {
      "epoch": 4.143905639648437e-05,
      "step": 43452,
      "training_step_time": 7.13901424407959
    },
    {
      "epoch": 4.1440010070800784e-05,
      "model_forward_time": 2.887725353240967,
      "step": 43453
    },
    {
      "epoch": 4.1440010070800784e-05,
      "step": 43453,
      "training_step_time": 7.7823145389556885
    },
    {
      "epoch": 4.144096374511719e-05,
      "model_forward_time": 1.715118169784546,
      "step": 43454
    },
    {
      "epoch": 4.144096374511719e-05,
      "step": 43454,
      "training_step_time": 7.7963550090789795
    },
    {
      "epoch": 4.144191741943359e-05,
      "model_forward_time": 1.3522775173187256,
      "step": 43455
    },
    {
      "epoch": 4.144191741943359e-05,
      "step": 43455,
      "training_step_time": 8.522752523422241
    },
    {
      "epoch": 4.144287109375e-05,
      "model_forward_time": 1.7213735580444336,
      "step": 43456
    },
    {
      "epoch": 4.144287109375e-05,
      "step": 43456,
      "training_step_time": 9.328411102294922
    },
    {
      "epoch": 4.144382476806641e-05,
      "model_forward_time": 1.7592134475708008,
      "step": 43457
    },
    {
      "epoch": 4.144382476806641e-05,
      "step": 43457,
      "training_step_time": 8.749404668807983
    },
    {
      "epoch": 4.144477844238281e-05,
      "model_forward_time": 1.7245280742645264,
      "step": 43458
    },
    {
      "epoch": 4.144477844238281e-05,
      "step": 43458,
      "training_step_time": 7.73986029624939
    },
    {
      "epoch": 4.144573211669922e-05,
      "model_forward_time": 1.7177624702453613,
      "step": 43459
    },
    {
      "epoch": 4.144573211669922e-05,
      "step": 43459,
      "training_step_time": 7.979142427444458
    },
    {
      "epoch": 4.1446685791015626e-05,
      "grad_norm": 0.08555678278207779,
      "learning_rate": 6.472730020089437e-06,
      "loss": 0.0623,
      "step": 43460
    },
    {
      "epoch": 4.1446685791015626e-05,
      "model_forward_time": 1.7113606929779053,
      "step": 43460
    },
    {
      "dynamics_loss_avg": 0.06646929532289506,
      "epoch": 4.1446685791015626e-05,
      "step": 43460
    },
    {
      "action_loss_avg": 0.0025933571858331563,
      "epoch": 4.1446685791015626e-05,
      "step": 43460
    },
    {
      "epoch": 4.1446685791015626e-05,
      "step": 43460,
      "training_step_time": 7.780452013015747
    },
    {
      "epoch": 4.144763946533203e-05,
      "model_forward_time": 1.715573787689209,
      "step": 43461
    },
    {
      "epoch": 4.144763946533203e-05,
      "step": 43461,
      "training_step_time": 7.521939754486084
    },
    {
      "epoch": 4.144859313964844e-05,
      "model_forward_time": 2.900320291519165,
      "step": 43462
    },
    {
      "epoch": 4.144859313964844e-05,
      "step": 43462,
      "training_step_time": 8.899196147918701
    },
    {
      "epoch": 4.1449546813964845e-05,
      "model_forward_time": 1.7180066108703613,
      "step": 43463
    },
    {
      "epoch": 4.1449546813964845e-05,
      "step": 43463,
      "training_step_time": 8.78827977180481
    },
    {
      "epoch": 4.145050048828125e-05,
      "model_forward_time": 1.7179882526397705,
      "step": 43464
    },
    {
      "epoch": 4.145050048828125e-05,
      "step": 43464,
      "training_step_time": 8.333255052566528
    },
    {
      "epoch": 4.145145416259766e-05,
      "model_forward_time": 1.716639757156372,
      "step": 43465
    },
    {
      "epoch": 4.145145416259766e-05,
      "step": 43465,
      "training_step_time": 8.400048017501831
    },
    {
      "epoch": 4.1452407836914064e-05,
      "model_forward_time": 1.7267448902130127,
      "step": 43466
    },
    {
      "epoch": 4.1452407836914064e-05,
      "step": 43466,
      "training_step_time": 8.397514820098877
    },
    {
      "epoch": 4.145336151123047e-05,
      "model_forward_time": 1.721836805343628,
      "step": 43467
    },
    {
      "epoch": 4.145336151123047e-05,
      "step": 43467,
      "training_step_time": 8.601050615310669
    },
    {
      "epoch": 4.145431518554688e-05,
      "model_forward_time": 1.7176220417022705,
      "step": 43468
    },
    {
      "epoch": 4.145431518554688e-05,
      "step": 43468,
      "training_step_time": 7.815974473953247
    },
    {
      "epoch": 4.145526885986328e-05,
      "model_forward_time": 1.717085838317871,
      "step": 43469
    },
    {
      "epoch": 4.145526885986328e-05,
      "step": 43469,
      "training_step_time": 8.809808015823364
    },
    {
      "epoch": 4.1456222534179686e-05,
      "grad_norm": 0.12405521422624588,
      "learning_rate": 6.471149822156222e-06,
      "loss": 0.0612,
      "step": 43470
    },
    {
      "epoch": 4.1456222534179686e-05,
      "model_forward_time": 1.7450568675994873,
      "step": 43470
    },
    {
      "dynamics_loss_avg": 0.06526124011725187,
      "epoch": 4.1456222534179686e-05,
      "step": 43470
    },
    {
      "action_loss_avg": 0.002499508982873522,
      "epoch": 4.1456222534179686e-05,
      "step": 43470
    },
    {
      "epoch": 4.1456222534179686e-05,
      "step": 43470,
      "training_step_time": 7.27888560295105
    },
    {
      "epoch": 4.145717620849609e-05,
      "model_forward_time": 2.8741605281829834,
      "step": 43471
    },
    {
      "epoch": 4.145717620849609e-05,
      "step": 43471,
      "training_step_time": 7.835005044937134
    },
    {
      "epoch": 4.14581298828125e-05,
      "model_forward_time": 1.717245101928711,
      "step": 43472
    },
    {
      "epoch": 4.14581298828125e-05,
      "step": 43472,
      "training_step_time": 7.743297576904297
    },
    {
      "epoch": 4.1459083557128905e-05,
      "model_forward_time": 1.7197306156158447,
      "step": 43473
    },
    {
      "epoch": 4.1459083557128905e-05,
      "step": 43473,
      "training_step_time": 8.223081827163696
    },
    {
      "epoch": 4.146003723144531e-05,
      "model_forward_time": 1.7120780944824219,
      "step": 43474
    },
    {
      "epoch": 4.146003723144531e-05,
      "step": 43474,
      "training_step_time": 7.811457395553589
    },
    {
      "epoch": 4.146099090576172e-05,
      "model_forward_time": 1.7190828323364258,
      "step": 43475
    },
    {
      "epoch": 4.146099090576172e-05,
      "step": 43475,
      "training_step_time": 9.074199438095093
    },
    {
      "epoch": 4.1461944580078124e-05,
      "model_forward_time": 1.7848122119903564,
      "step": 43476
    },
    {
      "epoch": 4.1461944580078124e-05,
      "step": 43476,
      "training_step_time": 7.784045457839966
    },
    {
      "epoch": 4.146289825439453e-05,
      "model_forward_time": 1.7119500637054443,
      "step": 43477
    },
    {
      "epoch": 4.146289825439453e-05,
      "step": 43477,
      "training_step_time": 8.4946768283844
    },
    {
      "epoch": 4.146385192871094e-05,
      "model_forward_time": 1.7189667224884033,
      "step": 43478
    },
    {
      "epoch": 4.146385192871094e-05,
      "step": 43478,
      "training_step_time": 7.716723442077637
    },
    {
      "epoch": 4.146480560302734e-05,
      "model_forward_time": 1.719529151916504,
      "step": 43479
    },
    {
      "epoch": 4.146480560302734e-05,
      "step": 43479,
      "training_step_time": 7.687470197677612
    },
    {
      "epoch": 4.146575927734375e-05,
      "grad_norm": 0.12450986355543137,
      "learning_rate": 6.469569463340275e-06,
      "loss": 0.0575,
      "step": 43480
    },
    {
      "epoch": 4.146575927734375e-05,
      "model_forward_time": 2.855712413787842,
      "step": 43480
    },
    {
      "dynamics_loss_avg": 0.044141468778252604,
      "epoch": 4.146575927734375e-05,
      "step": 43480
    },
    {
      "action_loss_avg": 0.0019354263611603528,
      "epoch": 4.146575927734375e-05,
      "step": 43480
    },
    {
      "epoch": 4.146575927734375e-05,
      "step": 43480,
      "training_step_time": 7.768357515335083
    },
    {
      "epoch": 4.146671295166016e-05,
      "model_forward_time": 1.7243807315826416,
      "step": 43481
    },
    {
      "epoch": 4.146671295166016e-05,
      "step": 43481,
      "training_step_time": 8.619294166564941
    },
    {
      "epoch": 4.146766662597656e-05,
      "model_forward_time": 1.7206757068634033,
      "step": 43482
    },
    {
      "epoch": 4.146766662597656e-05,
      "step": 43482,
      "training_step_time": 9.303795099258423
    },
    {
      "epoch": 4.1468620300292966e-05,
      "model_forward_time": 1.7282824516296387,
      "step": 43483
    },
    {
      "epoch": 4.1468620300292966e-05,
      "step": 43483,
      "training_step_time": 8.323984146118164
    },
    {
      "epoch": 4.146957397460938e-05,
      "model_forward_time": 1.7213771343231201,
      "step": 43484
    },
    {
      "epoch": 4.146957397460938e-05,
      "step": 43484,
      "training_step_time": 7.81340479850769
    },
    {
      "epoch": 4.147052764892578e-05,
      "model_forward_time": 1.7883214950561523,
      "step": 43485
    },
    {
      "epoch": 4.147052764892578e-05,
      "step": 43485,
      "training_step_time": 8.444935321807861
    },
    {
      "epoch": 4.1471481323242185e-05,
      "model_forward_time": 1.7214000225067139,
      "step": 43486
    },
    {
      "epoch": 4.1471481323242185e-05,
      "step": 43486,
      "training_step_time": 8.304190635681152
    },
    {
      "epoch": 4.1472434997558596e-05,
      "model_forward_time": 1.7232279777526855,
      "step": 43487
    },
    {
      "epoch": 4.1472434997558596e-05,
      "step": 43487,
      "training_step_time": 7.930082082748413
    },
    {
      "epoch": 4.1473388671875e-05,
      "model_forward_time": 1.727750301361084,
      "step": 43488
    },
    {
      "epoch": 4.1473388671875e-05,
      "step": 43488,
      "training_step_time": 7.743327379226685
    },
    {
      "epoch": 4.1474342346191404e-05,
      "model_forward_time": 2.43080735206604,
      "step": 43489
    },
    {
      "epoch": 4.1474342346191404e-05,
      "step": 43489,
      "training_step_time": 7.7339112758636475
    },
    {
      "epoch": 4.1475296020507814e-05,
      "grad_norm": 0.13578897714614868,
      "learning_rate": 6.467988943814418e-06,
      "loss": 0.0562,
      "step": 43490
    },
    {
      "epoch": 4.1475296020507814e-05,
      "model_forward_time": 1.7107458114624023,
      "step": 43490
    },
    {
      "dynamics_loss_avg": 0.06472502201795578,
      "epoch": 4.1475296020507814e-05,
      "step": 43490
    },
    {
      "action_loss_avg": 0.0015654929156880826,
      "epoch": 4.1475296020507814e-05,
      "step": 43490
    },
    {
      "epoch": 4.1475296020507814e-05,
      "step": 43490,
      "training_step_time": 7.72338342666626
    },
    {
      "epoch": 4.147624969482422e-05,
      "model_forward_time": 1.7154150009155273,
      "step": 43491
    },
    {
      "epoch": 4.147624969482422e-05,
      "step": 43491,
      "training_step_time": 7.650412559509277
    },
    {
      "epoch": 4.147720336914062e-05,
      "model_forward_time": 1.7239410877227783,
      "step": 43492
    },
    {
      "epoch": 4.147720336914062e-05,
      "step": 43492,
      "training_step_time": 7.667054176330566
    },
    {
      "epoch": 4.147815704345703e-05,
      "model_forward_time": 1.7261090278625488,
      "step": 43493
    },
    {
      "epoch": 4.147815704345703e-05,
      "step": 43493,
      "training_step_time": 8.737589359283447
    },
    {
      "epoch": 4.147911071777344e-05,
      "model_forward_time": 1.767657995223999,
      "step": 43494
    },
    {
      "epoch": 4.147911071777344e-05,
      "step": 43494,
      "training_step_time": 7.803181171417236
    },
    {
      "epoch": 4.148006439208984e-05,
      "model_forward_time": 1.7480549812316895,
      "step": 43495
    },
    {
      "epoch": 4.148006439208984e-05,
      "step": 43495,
      "training_step_time": 7.641510963439941
    },
    {
      "epoch": 4.148101806640625e-05,
      "model_forward_time": 1.7227239608764648,
      "step": 43496
    },
    {
      "epoch": 4.148101806640625e-05,
      "step": 43496,
      "training_step_time": 7.8574182987213135
    },
    {
      "epoch": 4.1481971740722656e-05,
      "model_forward_time": 1.719088077545166,
      "step": 43497
    },
    {
      "epoch": 4.1481971740722656e-05,
      "step": 43497,
      "training_step_time": 7.644037961959839
    },
    {
      "epoch": 4.148292541503906e-05,
      "model_forward_time": 1.7206947803497314,
      "step": 43498
    },
    {
      "epoch": 4.148292541503906e-05,
      "step": 43498,
      "training_step_time": 7.425797462463379
    },
    {
      "epoch": 4.148387908935547e-05,
      "model_forward_time": 2.8448479175567627,
      "step": 43499
    },
    {
      "epoch": 4.148387908935547e-05,
      "step": 43499,
      "training_step_time": 7.71081018447876
    },
    {
      "epoch": 4.1484832763671875e-05,
      "grad_norm": 0.14949502050876617,
      "learning_rate": 6.4664082637515e-06,
      "loss": 0.0612,
      "step": 43500
    },
    {
      "epoch": 4.1484832763671875e-05,
      "model_forward_time": 1.7104039192199707,
      "step": 43500
    },
    {
      "dynamics_loss_avg": 0.06825286988168955,
      "epoch": 4.1484832763671875e-05,
      "step": 43500
    },
    {
      "action_loss_avg": 0.0020964021037798374,
      "epoch": 4.1484832763671875e-05,
      "step": 43500
    },
    {
      "epoch": 4.1484832763671875e-05,
      "step": 43500,
      "training_step_time": 7.600658416748047
    },
    {
      "epoch": 4.148578643798828e-05,
      "model_forward_time": 1.7245192527770996,
      "step": 43501
    },
    {
      "epoch": 4.148578643798828e-05,
      "step": 43501,
      "training_step_time": 8.393167734146118
    },
    {
      "epoch": 4.148674011230469e-05,
      "model_forward_time": 1.7208881378173828,
      "step": 43502
    },
    {
      "epoch": 4.148674011230469e-05,
      "step": 43502,
      "training_step_time": 7.695743799209595
    },
    {
      "epoch": 4.1487693786621094e-05,
      "model_forward_time": 1.7222504615783691,
      "step": 43503
    },
    {
      "epoch": 4.1487693786621094e-05,
      "step": 43503,
      "training_step_time": 8.51836609840393
    },
    {
      "epoch": 4.14886474609375e-05,
      "model_forward_time": 1.7246427536010742,
      "step": 43504
    },
    {
      "epoch": 4.14886474609375e-05,
      "step": 43504,
      "training_step_time": 9.646559476852417
    },
    {
      "epoch": 4.148960113525391e-05,
      "model_forward_time": 1.7206995487213135,
      "step": 43505
    },
    {
      "epoch": 4.148960113525391e-05,
      "step": 43505,
      "training_step_time": 7.754008531570435
    },
    {
      "epoch": 4.149055480957031e-05,
      "model_forward_time": 1.7190308570861816,
      "step": 43506
    },
    {
      "epoch": 4.149055480957031e-05,
      "step": 43506,
      "training_step_time": 7.65482759475708
    },
    {
      "epoch": 4.149150848388672e-05,
      "model_forward_time": 1.7231059074401855,
      "step": 43507
    },
    {
      "epoch": 4.149150848388672e-05,
      "step": 43507,
      "training_step_time": 7.326478004455566
    },
    {
      "epoch": 4.149246215820313e-05,
      "model_forward_time": 2.9120404720306396,
      "step": 43508
    },
    {
      "epoch": 4.149246215820313e-05,
      "step": 43508,
      "training_step_time": 7.725200653076172
    },
    {
      "epoch": 4.149341583251953e-05,
      "model_forward_time": 1.7180306911468506,
      "step": 43509
    },
    {
      "epoch": 4.149341583251953e-05,
      "step": 43509,
      "training_step_time": 7.534093618392944
    },
    {
      "epoch": 4.1494369506835936e-05,
      "grad_norm": 0.14309215545654297,
      "learning_rate": 6.464827423324378e-06,
      "loss": 0.0563,
      "step": 43510
    },
    {
      "epoch": 4.1494369506835936e-05,
      "model_forward_time": 1.7271475791931152,
      "step": 43510
    },
    {
      "dynamics_loss_avg": 0.06672108173370361,
      "epoch": 4.1494369506835936e-05,
      "step": 43510
    },
    {
      "action_loss_avg": 0.001875609024136793,
      "epoch": 4.1494369506835936e-05,
      "step": 43510
    },
    {
      "epoch": 4.1494369506835936e-05,
      "step": 43510,
      "training_step_time": 9.137624502182007
    },
    {
      "epoch": 4.1495323181152346e-05,
      "model_forward_time": 1.7223870754241943,
      "step": 43511
    },
    {
      "epoch": 4.1495323181152346e-05,
      "step": 43511,
      "training_step_time": 8.635499954223633
    },
    {
      "epoch": 4.149627685546875e-05,
      "model_forward_time": 1.7491800785064697,
      "step": 43512
    },
    {
      "epoch": 4.149627685546875e-05,
      "step": 43512,
      "training_step_time": 8.783187866210938
    },
    {
      "epoch": 4.1497230529785154e-05,
      "model_forward_time": 1.7781858444213867,
      "step": 43513
    },
    {
      "epoch": 4.1497230529785154e-05,
      "step": 43513,
      "training_step_time": 8.919559717178345
    },
    {
      "epoch": 4.1498184204101565e-05,
      "model_forward_time": 1.7241604328155518,
      "step": 43514
    },
    {
      "epoch": 4.1498184204101565e-05,
      "step": 43514,
      "training_step_time": 7.835278749465942
    },
    {
      "epoch": 4.149913787841797e-05,
      "model_forward_time": 1.7217848300933838,
      "step": 43515
    },
    {
      "epoch": 4.149913787841797e-05,
      "step": 43515,
      "training_step_time": 7.577438831329346
    },
    {
      "epoch": 4.150009155273437e-05,
      "model_forward_time": 1.7218940258026123,
      "step": 43516
    },
    {
      "epoch": 4.150009155273437e-05,
      "step": 43516,
      "training_step_time": 7.515158176422119
    },
    {
      "epoch": 4.1501045227050784e-05,
      "model_forward_time": 2.891535520553589,
      "step": 43517
    },
    {
      "epoch": 4.1501045227050784e-05,
      "step": 43517,
      "training_step_time": 8.391605854034424
    },
    {
      "epoch": 4.150199890136719e-05,
      "model_forward_time": 1.722259283065796,
      "step": 43518
    },
    {
      "epoch": 4.150199890136719e-05,
      "step": 43518,
      "training_step_time": 7.47028112411499
    },
    {
      "epoch": 4.150295257568359e-05,
      "model_forward_time": 1.7271313667297363,
      "step": 43519
    },
    {
      "epoch": 4.150295257568359e-05,
      "step": 43519,
      "training_step_time": 8.1301429271698
    },
    {
      "epoch": 4.150390625e-05,
      "grad_norm": 0.0994766429066658,
      "learning_rate": 6.463246422705931e-06,
      "loss": 0.0623,
      "step": 43520
    },
    {
      "epoch": 4.150390625e-05,
      "model_forward_time": 1.7363314628601074,
      "step": 43520
    },
    {
      "dynamics_loss_avg": 0.06158953122794628,
      "epoch": 4.150390625e-05,
      "step": 43520
    },
    {
      "action_loss_avg": 0.0014567517617251724,
      "epoch": 4.150390625e-05,
      "step": 43520
    },
    {
      "epoch": 4.150390625e-05,
      "step": 43520,
      "training_step_time": 8.24799633026123
    },
    {
      "epoch": 4.150485992431641e-05,
      "model_forward_time": 1.72068452835083,
      "step": 43521
    },
    {
      "epoch": 4.150485992431641e-05,
      "step": 43521,
      "training_step_time": 8.417175769805908
    },
    {
      "epoch": 4.150581359863281e-05,
      "model_forward_time": 1.7489464282989502,
      "step": 43522
    },
    {
      "epoch": 4.150581359863281e-05,
      "step": 43522,
      "training_step_time": 7.674844741821289
    },
    {
      "epoch": 4.150676727294922e-05,
      "model_forward_time": 1.7205238342285156,
      "step": 43523
    },
    {
      "epoch": 4.150676727294922e-05,
      "step": 43523,
      "training_step_time": 7.611849069595337
    },
    {
      "epoch": 4.1507720947265626e-05,
      "model_forward_time": 1.7225165367126465,
      "step": 43524
    },
    {
      "epoch": 4.1507720947265626e-05,
      "step": 43524,
      "training_step_time": 7.7983551025390625
    },
    {
      "epoch": 4.150867462158203e-05,
      "model_forward_time": 1.7245845794677734,
      "step": 43525
    },
    {
      "epoch": 4.150867462158203e-05,
      "step": 43525,
      "training_step_time": 7.591328859329224
    },
    {
      "epoch": 4.150962829589844e-05,
      "model_forward_time": 2.853976011276245,
      "step": 43526
    },
    {
      "epoch": 4.150962829589844e-05,
      "step": 43526,
      "training_step_time": 7.655493259429932
    },
    {
      "epoch": 4.1510581970214845e-05,
      "model_forward_time": 1.723043441772461,
      "step": 43527
    },
    {
      "epoch": 4.1510581970214845e-05,
      "step": 43527,
      "training_step_time": 7.745172739028931
    },
    {
      "epoch": 4.151153564453125e-05,
      "model_forward_time": 1.722029685974121,
      "step": 43528
    },
    {
      "epoch": 4.151153564453125e-05,
      "step": 43528,
      "training_step_time": 8.346747398376465
    },
    {
      "epoch": 4.151248931884766e-05,
      "model_forward_time": 1.7218191623687744,
      "step": 43529
    },
    {
      "epoch": 4.151248931884766e-05,
      "step": 43529,
      "training_step_time": 8.364910125732422
    },
    {
      "epoch": 4.1513442993164064e-05,
      "grad_norm": 0.10369299352169037,
      "learning_rate": 6.461665262069056e-06,
      "loss": 0.0592,
      "step": 43530
    },
    {
      "epoch": 4.1513442993164064e-05,
      "model_forward_time": 1.7211058139801025,
      "step": 43530
    },
    {
      "dynamics_loss_avg": 0.057868863455951214,
      "epoch": 4.1513442993164064e-05,
      "step": 43530
    },
    {
      "action_loss_avg": 0.0023606130620464683,
      "epoch": 4.1513442993164064e-05,
      "step": 43530
    },
    {
      "epoch": 4.1513442993164064e-05,
      "step": 43530,
      "training_step_time": 8.272683382034302
    },
    {
      "epoch": 4.151439666748047e-05,
      "model_forward_time": 1.7230565547943115,
      "step": 43531
    },
    {
      "epoch": 4.151439666748047e-05,
      "step": 43531,
      "training_step_time": 8.521459817886353
    },
    {
      "epoch": 4.151535034179687e-05,
      "model_forward_time": 1.7919504642486572,
      "step": 43532
    },
    {
      "epoch": 4.151535034179687e-05,
      "step": 43532,
      "training_step_time": 7.677454471588135
    },
    {
      "epoch": 4.151630401611328e-05,
      "model_forward_time": 1.719886302947998,
      "step": 43533
    },
    {
      "epoch": 4.151630401611328e-05,
      "step": 43533,
      "training_step_time": 8.35300087928772
    },
    {
      "epoch": 4.1517257690429686e-05,
      "model_forward_time": 1.726151943206787,
      "step": 43534
    },
    {
      "epoch": 4.1517257690429686e-05,
      "step": 43534,
      "training_step_time": 7.190852642059326
    },
    {
      "epoch": 4.151821136474609e-05,
      "model_forward_time": 2.8199939727783203,
      "step": 43535
    },
    {
      "epoch": 4.151821136474609e-05,
      "step": 43535,
      "training_step_time": 7.614145517349243
    },
    {
      "epoch": 4.15191650390625e-05,
      "model_forward_time": 1.7285993099212646,
      "step": 43536
    },
    {
      "epoch": 4.15191650390625e-05,
      "step": 43536,
      "training_step_time": 8.51166582107544
    },
    {
      "epoch": 4.1520118713378905e-05,
      "model_forward_time": 1.7265007495880127,
      "step": 43537
    },
    {
      "epoch": 4.1520118713378905e-05,
      "step": 43537,
      "training_step_time": 8.686587572097778
    },
    {
      "epoch": 4.152107238769531e-05,
      "model_forward_time": 1.705399990081787,
      "step": 43538
    },
    {
      "epoch": 4.152107238769531e-05,
      "step": 43538,
      "training_step_time": 8.27071762084961
    },
    {
      "epoch": 4.152202606201172e-05,
      "model_forward_time": 1.7184298038482666,
      "step": 43539
    },
    {
      "epoch": 4.152202606201172e-05,
      "step": 43539,
      "training_step_time": 8.50101375579834
    },
    {
      "epoch": 4.1522979736328124e-05,
      "grad_norm": 0.1291125863790512,
      "learning_rate": 6.4600839415866655e-06,
      "loss": 0.0571,
      "step": 43540
    },
    {
      "epoch": 4.1522979736328124e-05,
      "model_forward_time": 1.70501708984375,
      "step": 43540
    },
    {
      "dynamics_loss_avg": 0.044285989087075,
      "epoch": 4.1522979736328124e-05,
      "step": 43540
    },
    {
      "action_loss_avg": 0.0014993687742389738,
      "epoch": 4.1522979736328124e-05,
      "step": 43540
    },
    {
      "epoch": 4.1522979736328124e-05,
      "step": 43540,
      "training_step_time": 9.337700128555298
    },
    {
      "epoch": 4.152393341064453e-05,
      "model_forward_time": 1.4303889274597168,
      "step": 43541
    },
    {
      "epoch": 4.152393341064453e-05,
      "step": 43541,
      "training_step_time": 8.433294534683228
    },
    {
      "epoch": 4.152488708496094e-05,
      "model_forward_time": 1.6892540454864502,
      "step": 43542
    },
    {
      "epoch": 4.152488708496094e-05,
      "step": 43542,
      "training_step_time": 8.609598398208618
    },
    {
      "epoch": 4.152584075927734e-05,
      "model_forward_time": 1.7081530094146729,
      "step": 43543
    },
    {
      "epoch": 4.152584075927734e-05,
      "step": 43543,
      "training_step_time": 8.535691499710083
    },
    {
      "epoch": 4.152679443359375e-05,
      "model_forward_time": 1.6939656734466553,
      "step": 43544
    },
    {
      "epoch": 4.152679443359375e-05,
      "step": 43544,
      "training_step_time": 7.380186319351196
    },
    {
      "epoch": 4.152774810791016e-05,
      "model_forward_time": 2.8698630332946777,
      "step": 43545
    },
    {
      "epoch": 4.152774810791016e-05,
      "step": 43545,
      "training_step_time": 7.629330635070801
    },
    {
      "epoch": 4.152870178222656e-05,
      "model_forward_time": 1.6887471675872803,
      "step": 43546
    },
    {
      "epoch": 4.152870178222656e-05,
      "step": 43546,
      "training_step_time": 7.502442121505737
    },
    {
      "epoch": 4.1529655456542966e-05,
      "model_forward_time": 1.6904356479644775,
      "step": 43547
    },
    {
      "epoch": 4.1529655456542966e-05,
      "step": 43547,
      "training_step_time": 8.2100350856781
    },
    {
      "epoch": 4.153060913085938e-05,
      "model_forward_time": 1.688176155090332,
      "step": 43548
    },
    {
      "epoch": 4.153060913085938e-05,
      "step": 43548,
      "training_step_time": 8.413437604904175
    },
    {
      "epoch": 4.153156280517578e-05,
      "model_forward_time": 1.6910748481750488,
      "step": 43549
    },
    {
      "epoch": 4.153156280517578e-05,
      "step": 43549,
      "training_step_time": 7.841193437576294
    },
    {
      "epoch": 4.1532516479492185e-05,
      "grad_norm": 0.10095714777708054,
      "learning_rate": 6.458502461431689e-06,
      "loss": 0.0593,
      "step": 43550
    },
    {
      "epoch": 4.1532516479492185e-05,
      "model_forward_time": 1.7092170715332031,
      "step": 43550
    },
    {
      "dynamics_loss_avg": 0.04900128887966275,
      "epoch": 4.1532516479492185e-05,
      "step": 43550
    },
    {
      "action_loss_avg": 0.0013735173706663772,
      "epoch": 4.1532516479492185e-05,
      "step": 43550
    },
    {
      "epoch": 4.1532516479492185e-05,
      "step": 43550,
      "training_step_time": 7.725514888763428
    },
    {
      "epoch": 4.1533470153808596e-05,
      "model_forward_time": 1.6914143562316895,
      "step": 43551
    },
    {
      "epoch": 4.1533470153808596e-05,
      "step": 43551,
      "training_step_time": 7.94750189781189
    },
    {
      "epoch": 4.1534423828125e-05,
      "model_forward_time": 1.7145707607269287,
      "step": 43552
    },
    {
      "epoch": 4.1534423828125e-05,
      "step": 43552,
      "training_step_time": 7.6863343715667725
    },
    {
      "epoch": 4.1535377502441404e-05,
      "model_forward_time": 1.6899011135101318,
      "step": 43553
    },
    {
      "epoch": 4.1535377502441404e-05,
      "step": 43553,
      "training_step_time": 8.454030275344849
    },
    {
      "epoch": 4.1536331176757814e-05,
      "model_forward_time": 2.8819940090179443,
      "step": 43554
    },
    {
      "epoch": 4.1536331176757814e-05,
      "step": 43554,
      "training_step_time": 7.6849424839019775
    },
    {
      "epoch": 4.153728485107422e-05,
      "model_forward_time": 1.6888363361358643,
      "step": 43555
    },
    {
      "epoch": 4.153728485107422e-05,
      "step": 43555,
      "training_step_time": 7.497718095779419
    },
    {
      "epoch": 4.153823852539062e-05,
      "model_forward_time": 1.6948297023773193,
      "step": 43556
    },
    {
      "epoch": 4.153823852539062e-05,
      "step": 43556,
      "training_step_time": 7.615259885787964
    },
    {
      "epoch": 4.153919219970703e-05,
      "model_forward_time": 1.6968555450439453,
      "step": 43557
    },
    {
      "epoch": 4.153919219970703e-05,
      "step": 43557,
      "training_step_time": 9.287569284439087
    },
    {
      "epoch": 4.154014587402344e-05,
      "model_forward_time": 1.6929054260253906,
      "step": 43558
    },
    {
      "epoch": 4.154014587402344e-05,
      "step": 43558,
      "training_step_time": 8.91970682144165
    },
    {
      "epoch": 4.154109954833984e-05,
      "model_forward_time": 1.719515323638916,
      "step": 43559
    },
    {
      "epoch": 4.154109954833984e-05,
      "step": 43559,
      "training_step_time": 9.49914288520813
    },
    {
      "epoch": 4.154205322265625e-05,
      "grad_norm": 0.12171031534671783,
      "learning_rate": 6.456920821777077e-06,
      "loss": 0.0591,
      "step": 43560
    },
    {
      "epoch": 4.154205322265625e-05,
      "model_forward_time": 1.685852289199829,
      "step": 43560
    },
    {
      "dynamics_loss_avg": 0.04660782068967819,
      "epoch": 4.154205322265625e-05,
      "step": 43560
    },
    {
      "action_loss_avg": 0.0018896008608862759,
      "epoch": 4.154205322265625e-05,
      "step": 43560
    },
    {
      "epoch": 4.154205322265625e-05,
      "step": 43560,
      "training_step_time": 8.344056606292725
    },
    {
      "epoch": 4.1543006896972656e-05,
      "model_forward_time": 1.3143842220306396,
      "step": 43561
    },
    {
      "epoch": 4.1543006896972656e-05,
      "step": 43561,
      "training_step_time": 9.71479868888855
    },
    {
      "epoch": 4.154396057128906e-05,
      "model_forward_time": 1.7022950649261475,
      "step": 43562
    },
    {
      "epoch": 4.154396057128906e-05,
      "step": 43562,
      "training_step_time": 7.65166711807251
    },
    {
      "epoch": 4.154491424560547e-05,
      "model_forward_time": 2.8203489780426025,
      "step": 43563
    },
    {
      "epoch": 4.154491424560547e-05,
      "step": 43563,
      "training_step_time": 7.924534559249878
    },
    {
      "epoch": 4.1545867919921875e-05,
      "model_forward_time": 1.6887071132659912,
      "step": 43564
    },
    {
      "epoch": 4.1545867919921875e-05,
      "step": 43564,
      "training_step_time": 7.689465761184692
    },
    {
      "epoch": 4.154682159423828e-05,
      "model_forward_time": 1.6983535289764404,
      "step": 43565
    },
    {
      "epoch": 4.154682159423828e-05,
      "step": 43565,
      "training_step_time": 8.358100652694702
    },
    {
      "epoch": 4.154777526855469e-05,
      "model_forward_time": 1.6952934265136719,
      "step": 43566
    },
    {
      "epoch": 4.154777526855469e-05,
      "step": 43566,
      "training_step_time": 7.8516294956207275
    },
    {
      "epoch": 4.1548728942871094e-05,
      "model_forward_time": 1.693937063217163,
      "step": 43567
    },
    {
      "epoch": 4.1548728942871094e-05,
      "step": 43567,
      "training_step_time": 7.866218566894531
    },
    {
      "epoch": 4.15496826171875e-05,
      "model_forward_time": 1.7091500759124756,
      "step": 43568
    },
    {
      "epoch": 4.15496826171875e-05,
      "step": 43568,
      "training_step_time": 9.002012014389038
    },
    {
      "epoch": 4.155063629150391e-05,
      "model_forward_time": 1.6968200206756592,
      "step": 43569
    },
    {
      "epoch": 4.155063629150391e-05,
      "step": 43569,
      "training_step_time": 7.613588571548462
    },
    {
      "epoch": 4.155158996582031e-05,
      "grad_norm": 0.16725251078605652,
      "learning_rate": 6.455339022795796e-06,
      "loss": 0.0597,
      "step": 43570
    },
    {
      "epoch": 4.155158996582031e-05,
      "model_forward_time": 1.6905434131622314,
      "step": 43570
    },
    {
      "dynamics_loss_avg": 0.06802408285439014,
      "epoch": 4.155158996582031e-05,
      "step": 43570
    },
    {
      "action_loss_avg": 0.0021374605712480843,
      "epoch": 4.155158996582031e-05,
      "step": 43570
    },
    {
      "epoch": 4.155158996582031e-05,
      "step": 43570,
      "training_step_time": 8.281217813491821
    },
    {
      "epoch": 4.155254364013672e-05,
      "model_forward_time": 1.6967108249664307,
      "step": 43571
    },
    {
      "epoch": 4.155254364013672e-05,
      "step": 43571,
      "training_step_time": 7.518638849258423
    },
    {
      "epoch": 4.155349731445313e-05,
      "model_forward_time": 2.801846504211426,
      "step": 43572
    },
    {
      "epoch": 4.155349731445313e-05,
      "step": 43572,
      "training_step_time": 8.344184875488281
    },
    {
      "epoch": 4.155445098876953e-05,
      "model_forward_time": 1.6877200603485107,
      "step": 43573
    },
    {
      "epoch": 4.155445098876953e-05,
      "step": 43573,
      "training_step_time": 7.6942219734191895
    },
    {
      "epoch": 4.1555404663085936e-05,
      "model_forward_time": 1.695368766784668,
      "step": 43574
    },
    {
      "epoch": 4.1555404663085936e-05,
      "step": 43574,
      "training_step_time": 7.56622576713562
    },
    {
      "epoch": 4.1556358337402346e-05,
      "model_forward_time": 1.6930749416351318,
      "step": 43575
    },
    {
      "epoch": 4.1556358337402346e-05,
      "step": 43575,
      "training_step_time": 8.183896541595459
    },
    {
      "epoch": 4.155731201171875e-05,
      "model_forward_time": 1.6983704566955566,
      "step": 43576
    },
    {
      "epoch": 4.155731201171875e-05,
      "step": 43576,
      "training_step_time": 8.504348278045654
    },
    {
      "epoch": 4.1558265686035155e-05,
      "model_forward_time": 1.6977014541625977,
      "step": 43577
    },
    {
      "epoch": 4.1558265686035155e-05,
      "step": 43577,
      "training_step_time": 7.6988420486450195
    },
    {
      "epoch": 4.1559219360351565e-05,
      "model_forward_time": 1.6997950077056885,
      "step": 43578
    },
    {
      "epoch": 4.1559219360351565e-05,
      "step": 43578,
      "training_step_time": 8.616517543792725
    },
    {
      "epoch": 4.156017303466797e-05,
      "model_forward_time": 1.684274673461914,
      "step": 43579
    },
    {
      "epoch": 4.156017303466797e-05,
      "step": 43579,
      "training_step_time": 8.9580659866333
    },
    {
      "epoch": 4.156112670898437e-05,
      "grad_norm": 0.12103207409381866,
      "learning_rate": 6.453757064660825e-06,
      "loss": 0.0593,
      "step": 43580
    },
    {
      "epoch": 4.156112670898437e-05,
      "model_forward_time": 1.7162187099456787,
      "step": 43580
    },
    {
      "dynamics_loss_avg": 0.05613500140607357,
      "epoch": 4.156112670898437e-05,
      "step": 43580
    },
    {
      "action_loss_avg": 0.0018549688713392243,
      "epoch": 4.156112670898437e-05,
      "step": 43580
    },
    {
      "epoch": 4.156112670898437e-05,
      "step": 43580,
      "training_step_time": 7.742968797683716
    },
    {
      "epoch": 4.1562080383300784e-05,
      "model_forward_time": 1.3142378330230713,
      "step": 43581
    },
    {
      "epoch": 4.1562080383300784e-05,
      "step": 43581,
      "training_step_time": 7.564866542816162
    },
    {
      "epoch": 4.156303405761719e-05,
      "model_forward_time": 2.838200807571411,
      "step": 43582
    },
    {
      "epoch": 4.156303405761719e-05,
      "step": 43582,
      "training_step_time": 8.645070314407349
    },
    {
      "epoch": 4.156398773193359e-05,
      "model_forward_time": 1.6917941570281982,
      "step": 43583
    },
    {
      "epoch": 4.156398773193359e-05,
      "step": 43583,
      "training_step_time": 8.433775424957275
    },
    {
      "epoch": 4.156494140625e-05,
      "model_forward_time": 1.6895885467529297,
      "step": 43584
    },
    {
      "epoch": 4.156494140625e-05,
      "step": 43584,
      "training_step_time": 8.47030758857727
    },
    {
      "epoch": 4.156589508056641e-05,
      "model_forward_time": 1.6925265789031982,
      "step": 43585
    },
    {
      "epoch": 4.156589508056641e-05,
      "step": 43585,
      "training_step_time": 8.219110488891602
    },
    {
      "epoch": 4.156684875488281e-05,
      "model_forward_time": 1.6926326751708984,
      "step": 43586
    },
    {
      "epoch": 4.156684875488281e-05,
      "step": 43586,
      "training_step_time": 7.862112760543823
    },
    {
      "epoch": 4.156780242919922e-05,
      "model_forward_time": 1.7126357555389404,
      "step": 43587
    },
    {
      "epoch": 4.156780242919922e-05,
      "step": 43587,
      "training_step_time": 8.504003763198853
    },
    {
      "epoch": 4.1568756103515626e-05,
      "model_forward_time": 1.6971254348754883,
      "step": 43588
    },
    {
      "epoch": 4.1568756103515626e-05,
      "step": 43588,
      "training_step_time": 7.833176851272583
    },
    {
      "epoch": 4.156970977783203e-05,
      "model_forward_time": 1.720346212387085,
      "step": 43589
    },
    {
      "epoch": 4.156970977783203e-05,
      "step": 43589,
      "training_step_time": 7.576170921325684
    },
    {
      "epoch": 4.157066345214844e-05,
      "grad_norm": 0.09160564094781876,
      "learning_rate": 6.452174947545169e-06,
      "loss": 0.0589,
      "step": 43590
    },
    {
      "epoch": 4.157066345214844e-05,
      "model_forward_time": 1.692131757736206,
      "step": 43590
    },
    {
      "dynamics_loss_avg": 0.05923644751310349,
      "epoch": 4.157066345214844e-05,
      "step": 43590
    },
    {
      "action_loss_avg": 0.0020294629561249168,
      "epoch": 4.157066345214844e-05,
      "step": 43590
    },
    {
      "epoch": 4.157066345214844e-05,
      "step": 43590,
      "training_step_time": 8.994815349578857
    },
    {
      "epoch": 4.1571617126464845e-05,
      "model_forward_time": 2.8021883964538574,
      "step": 43591
    },
    {
      "epoch": 4.1571617126464845e-05,
      "step": 43591,
      "training_step_time": 8.507348775863647
    },
    {
      "epoch": 4.157257080078125e-05,
      "model_forward_time": 1.6948535442352295,
      "step": 43592
    },
    {
      "epoch": 4.157257080078125e-05,
      "step": 43592,
      "training_step_time": 7.603230953216553
    },
    {
      "epoch": 4.157352447509766e-05,
      "model_forward_time": 1.6886682510375977,
      "step": 43593
    },
    {
      "epoch": 4.157352447509766e-05,
      "step": 43593,
      "training_step_time": 9.360856771469116
    },
    {
      "epoch": 4.1574478149414064e-05,
      "model_forward_time": 1.6930391788482666,
      "step": 43594
    },
    {
      "epoch": 4.1574478149414064e-05,
      "step": 43594,
      "training_step_time": 8.730624914169312
    },
    {
      "epoch": 4.157543182373047e-05,
      "model_forward_time": 1.69795823097229,
      "step": 43595
    },
    {
      "epoch": 4.157543182373047e-05,
      "step": 43595,
      "training_step_time": 7.847102165222168
    },
    {
      "epoch": 4.157638549804687e-05,
      "model_forward_time": 1.71002197265625,
      "step": 43596
    },
    {
      "epoch": 4.157638549804687e-05,
      "step": 43596,
      "training_step_time": 8.391858100891113
    },
    {
      "epoch": 4.157733917236328e-05,
      "model_forward_time": 1.6971540451049805,
      "step": 43597
    },
    {
      "epoch": 4.157733917236328e-05,
      "step": 43597,
      "training_step_time": 7.641105651855469
    },
    {
      "epoch": 4.1578292846679687e-05,
      "model_forward_time": 1.7007074356079102,
      "step": 43598
    },
    {
      "epoch": 4.1578292846679687e-05,
      "step": 43598,
      "training_step_time": 7.49776554107666
    },
    {
      "epoch": 4.157924652099609e-05,
      "model_forward_time": 1.6936581134796143,
      "step": 43599
    },
    {
      "epoch": 4.157924652099609e-05,
      "step": 43599,
      "training_step_time": 7.254878759384155
    },
    {
      "epoch": 4.15802001953125e-05,
      "grad_norm": 0.09732600301504135,
      "learning_rate": 6.450592671621842e-06,
      "loss": 0.0609,
      "step": 43600
    },
    {
      "epoch": 4.15802001953125e-05,
      "model_forward_time": 2.824241876602173,
      "step": 43600
    },
    {
      "dynamics_loss_avg": 0.07179644890129566,
      "epoch": 4.15802001953125e-05,
      "step": 43600
    },
    {
      "action_loss_avg": 0.0020610751060303302,
      "epoch": 4.15802001953125e-05,
      "step": 43600
    },
    {
      "epoch": 4.15802001953125e-05,
      "step": 43600,
      "training_step_time": 7.741949558258057
    },
    {
      "epoch": 4.1581153869628905e-05,
      "model_forward_time": 1.6909527778625488,
      "step": 43601
    },
    {
      "epoch": 4.1581153869628905e-05,
      "step": 43601,
      "training_step_time": 8.530059337615967
    },
    {
      "epoch": 4.158210754394531e-05,
      "model_forward_time": 1.6976008415222168,
      "step": 43602
    },
    {
      "epoch": 4.158210754394531e-05,
      "step": 43602,
      "training_step_time": 7.635532379150391
    },
    {
      "epoch": 4.158306121826172e-05,
      "model_forward_time": 1.6919102668762207,
      "step": 43603
    },
    {
      "epoch": 4.158306121826172e-05,
      "step": 43603,
      "training_step_time": 8.352152585983276
    },
    {
      "epoch": 4.1584014892578124e-05,
      "model_forward_time": 1.6938917636871338,
      "step": 43604
    },
    {
      "epoch": 4.1584014892578124e-05,
      "step": 43604,
      "training_step_time": 7.8227808475494385
    },
    {
      "epoch": 4.158496856689453e-05,
      "model_forward_time": 1.7068562507629395,
      "step": 43605
    },
    {
      "epoch": 4.158496856689453e-05,
      "step": 43605,
      "training_step_time": 7.795401573181152
    },
    {
      "epoch": 4.158592224121094e-05,
      "model_forward_time": 1.697052240371704,
      "step": 43606
    },
    {
      "epoch": 4.158592224121094e-05,
      "step": 43606,
      "training_step_time": 7.605737924575806
    },
    {
      "epoch": 4.158687591552734e-05,
      "model_forward_time": 1.6920690536499023,
      "step": 43607
    },
    {
      "epoch": 4.158687591552734e-05,
      "step": 43607,
      "training_step_time": 7.751115560531616
    },
    {
      "epoch": 4.158782958984375e-05,
      "model_forward_time": 1.7187714576721191,
      "step": 43608
    },
    {
      "epoch": 4.158782958984375e-05,
      "step": 43608,
      "training_step_time": 7.558687925338745
    },
    {
      "epoch": 4.158878326416016e-05,
      "model_forward_time": 2.8236594200134277,
      "step": 43609
    },
    {
      "epoch": 4.158878326416016e-05,
      "step": 43609,
      "training_step_time": 7.673765182495117
    },
    {
      "epoch": 4.158973693847656e-05,
      "grad_norm": 0.10455221682786942,
      "learning_rate": 6.4490102370638826e-06,
      "loss": 0.0595,
      "step": 43610
    },
    {
      "epoch": 4.158973693847656e-05,
      "model_forward_time": 1.6932635307312012,
      "step": 43610
    },
    {
      "dynamics_loss_avg": 0.06652294881641865,
      "epoch": 4.158973693847656e-05,
      "step": 43610
    },
    {
      "action_loss_avg": 0.0018765060667647049,
      "epoch": 4.158973693847656e-05,
      "step": 43610
    },
    {
      "epoch": 4.158973693847656e-05,
      "step": 43610,
      "training_step_time": 7.603366136550903
    },
    {
      "epoch": 4.1590690612792966e-05,
      "model_forward_time": 1.6950676441192627,
      "step": 43611
    },
    {
      "epoch": 4.1590690612792966e-05,
      "step": 43611,
      "training_step_time": 7.971978425979614
    },
    {
      "epoch": 4.159164428710938e-05,
      "model_forward_time": 1.6908485889434814,
      "step": 43612
    },
    {
      "epoch": 4.159164428710938e-05,
      "step": 43612,
      "training_step_time": 7.63150429725647
    },
    {
      "epoch": 4.159259796142578e-05,
      "model_forward_time": 1.6936686038970947,
      "step": 43613
    },
    {
      "epoch": 4.159259796142578e-05,
      "step": 43613,
      "training_step_time": 8.248644351959229
    },
    {
      "epoch": 4.1593551635742185e-05,
      "model_forward_time": 1.6940367221832275,
      "step": 43614
    },
    {
      "epoch": 4.1593551635742185e-05,
      "step": 43614,
      "training_step_time": 8.184536695480347
    },
    {
      "epoch": 4.1594505310058596e-05,
      "model_forward_time": 1.695908784866333,
      "step": 43615
    },
    {
      "epoch": 4.1594505310058596e-05,
      "step": 43615,
      "training_step_time": 7.600210905075073
    },
    {
      "epoch": 4.1595458984375e-05,
      "model_forward_time": 1.6984062194824219,
      "step": 43616
    },
    {
      "epoch": 4.1595458984375e-05,
      "step": 43616,
      "training_step_time": 7.806962966918945
    },
    {
      "epoch": 4.1596412658691404e-05,
      "model_forward_time": 1.7085509300231934,
      "step": 43617
    },
    {
      "epoch": 4.1596412658691404e-05,
      "step": 43617,
      "training_step_time": 8.42066216468811
    },
    {
      "epoch": 4.1597366333007815e-05,
      "model_forward_time": 1.6907057762145996,
      "step": 43618
    },
    {
      "epoch": 4.1597366333007815e-05,
      "step": 43618,
      "training_step_time": 8.835786581039429
    },
    {
      "epoch": 4.159832000732422e-05,
      "model_forward_time": 1.6950342655181885,
      "step": 43619
    },
    {
      "epoch": 4.159832000732422e-05,
      "step": 43619,
      "training_step_time": 7.6339194774627686
    },
    {
      "epoch": 4.159927368164062e-05,
      "grad_norm": 0.09836184233427048,
      "learning_rate": 6.447427644044343e-06,
      "loss": 0.0577,
      "step": 43620
    },
    {
      "epoch": 4.159927368164062e-05,
      "model_forward_time": 1.6907837390899658,
      "step": 43620
    },
    {
      "dynamics_loss_avg": 0.06376980021595954,
      "epoch": 4.159927368164062e-05,
      "step": 43620
    },
    {
      "action_loss_avg": 0.0014452926930971443,
      "epoch": 4.159927368164062e-05,
      "step": 43620
    },
    {
      "epoch": 4.159927368164062e-05,
      "step": 43620,
      "training_step_time": 7.562204122543335
    },
    {
      "epoch": 4.160022735595703e-05,
      "model_forward_time": 1.693157434463501,
      "step": 43621
    },
    {
      "epoch": 4.160022735595703e-05,
      "step": 43621,
      "training_step_time": 9.56972885131836
    },
    {
      "epoch": 4.160118103027344e-05,
      "model_forward_time": 1.6964340209960938,
      "step": 43622
    },
    {
      "epoch": 4.160118103027344e-05,
      "step": 43622,
      "training_step_time": 8.822691440582275
    },
    {
      "epoch": 4.160213470458984e-05,
      "model_forward_time": 1.6894927024841309,
      "step": 43623
    },
    {
      "epoch": 4.160213470458984e-05,
      "step": 43623,
      "training_step_time": 9.104885339736938
    },
    {
      "epoch": 4.160308837890625e-05,
      "model_forward_time": 1.7182369232177734,
      "step": 43624
    },
    {
      "epoch": 4.160308837890625e-05,
      "step": 43624,
      "training_step_time": 7.609212398529053
    },
    {
      "epoch": 4.1604042053222656e-05,
      "model_forward_time": 1.6977028846740723,
      "step": 43625
    },
    {
      "epoch": 4.1604042053222656e-05,
      "step": 43625,
      "training_step_time": 7.664063930511475
    },
    {
      "epoch": 4.160499572753906e-05,
      "model_forward_time": 1.68867826461792,
      "step": 43626
    },
    {
      "epoch": 4.160499572753906e-05,
      "step": 43626,
      "training_step_time": 7.527954339981079
    },
    {
      "epoch": 4.160594940185547e-05,
      "model_forward_time": 1.695741891860962,
      "step": 43627
    },
    {
      "epoch": 4.160594940185547e-05,
      "step": 43627,
      "training_step_time": 8.116756200790405
    },
    {
      "epoch": 4.1606903076171875e-05,
      "model_forward_time": 2.9010367393493652,
      "step": 43628
    },
    {
      "epoch": 4.1606903076171875e-05,
      "step": 43628,
      "training_step_time": 7.6677939891815186
    },
    {
      "epoch": 4.160785675048828e-05,
      "model_forward_time": 1.6939632892608643,
      "step": 43629
    },
    {
      "epoch": 4.160785675048828e-05,
      "step": 43629,
      "training_step_time": 7.617282867431641
    },
    {
      "epoch": 4.160881042480469e-05,
      "grad_norm": 0.09406621009111404,
      "learning_rate": 6.445844892736292e-06,
      "loss": 0.0599,
      "step": 43630
    },
    {
      "epoch": 4.160881042480469e-05,
      "model_forward_time": 1.6904489994049072,
      "step": 43630
    },
    {
      "dynamics_loss_avg": 0.06401042304933072,
      "epoch": 4.160881042480469e-05,
      "step": 43630
    },
    {
      "action_loss_avg": 0.0017319602105999366,
      "epoch": 4.160881042480469e-05,
      "step": 43630
    },
    {
      "epoch": 4.160881042480469e-05,
      "step": 43630,
      "training_step_time": 7.631361484527588
    },
    {
      "epoch": 4.1609764099121094e-05,
      "model_forward_time": 1.6926913261413574,
      "step": 43631
    },
    {
      "epoch": 4.1609764099121094e-05,
      "step": 43631,
      "training_step_time": 7.652217626571655
    },
    {
      "epoch": 4.16107177734375e-05,
      "model_forward_time": 1.6973273754119873,
      "step": 43632
    },
    {
      "epoch": 4.16107177734375e-05,
      "step": 43632,
      "training_step_time": 7.789583921432495
    },
    {
      "epoch": 4.161167144775391e-05,
      "model_forward_time": 1.7250423431396484,
      "step": 43633
    },
    {
      "epoch": 4.161167144775391e-05,
      "step": 43633,
      "training_step_time": 8.6718168258667
    },
    {
      "epoch": 4.161262512207031e-05,
      "model_forward_time": 1.7006607055664062,
      "step": 43634
    },
    {
      "epoch": 4.161262512207031e-05,
      "step": 43634,
      "training_step_time": 7.8246283531188965
    },
    {
      "epoch": 4.161357879638672e-05,
      "model_forward_time": 1.7000501155853271,
      "step": 43635
    },
    {
      "epoch": 4.161357879638672e-05,
      "step": 43635,
      "training_step_time": 8.003443717956543
    },
    {
      "epoch": 4.161453247070313e-05,
      "model_forward_time": 1.7243988513946533,
      "step": 43636
    },
    {
      "epoch": 4.161453247070313e-05,
      "step": 43636,
      "training_step_time": 7.681758403778076
    },
    {
      "epoch": 4.161548614501953e-05,
      "model_forward_time": 2.885134696960449,
      "step": 43637
    },
    {
      "epoch": 4.161548614501953e-05,
      "step": 43637,
      "training_step_time": 7.7900471687316895
    },
    {
      "epoch": 4.1616439819335936e-05,
      "model_forward_time": 1.6966736316680908,
      "step": 43638
    },
    {
      "epoch": 4.1616439819335936e-05,
      "step": 43638,
      "training_step_time": 7.701445817947388
    },
    {
      "epoch": 4.1617393493652347e-05,
      "model_forward_time": 1.6962971687316895,
      "step": 43639
    },
    {
      "epoch": 4.1617393493652347e-05,
      "step": 43639,
      "training_step_time": 7.678845643997192
    },
    {
      "epoch": 4.161834716796875e-05,
      "grad_norm": 0.10546296834945679,
      "learning_rate": 6.444261983312817e-06,
      "loss": 0.0561,
      "step": 43640
    },
    {
      "epoch": 4.161834716796875e-05,
      "model_forward_time": 1.7077503204345703,
      "step": 43640
    },
    {
      "dynamics_loss_avg": 0.07592189488932491,
      "epoch": 4.161834716796875e-05,
      "step": 43640
    },
    {
      "action_loss_avg": 0.0020095049228984864,
      "epoch": 4.161834716796875e-05,
      "step": 43640
    },
    {
      "epoch": 4.161834716796875e-05,
      "step": 43640,
      "training_step_time": 8.40640139579773
    },
    {
      "epoch": 4.1619300842285155e-05,
      "model_forward_time": 1.6881818771362305,
      "step": 43641
    },
    {
      "epoch": 4.1619300842285155e-05,
      "step": 43641,
      "training_step_time": 8.857093811035156
    },
    {
      "epoch": 4.1620254516601565e-05,
      "model_forward_time": 1.7174351215362549,
      "step": 43642
    },
    {
      "epoch": 4.1620254516601565e-05,
      "step": 43642,
      "training_step_time": 8.65962290763855
    },
    {
      "epoch": 4.162120819091797e-05,
      "model_forward_time": 1.7006268501281738,
      "step": 43643
    },
    {
      "epoch": 4.162120819091797e-05,
      "step": 43643,
      "training_step_time": 7.655655384063721
    },
    {
      "epoch": 4.1622161865234373e-05,
      "model_forward_time": 1.7025768756866455,
      "step": 43644
    },
    {
      "epoch": 4.1622161865234373e-05,
      "step": 43644,
      "training_step_time": 7.771180868148804
    },
    {
      "epoch": 4.1623115539550784e-05,
      "model_forward_time": 1.7189412117004395,
      "step": 43645
    },
    {
      "epoch": 4.1623115539550784e-05,
      "step": 43645,
      "training_step_time": 8.609936952590942
    },
    {
      "epoch": 4.162406921386719e-05,
      "model_forward_time": 2.8519577980041504,
      "step": 43646
    },
    {
      "epoch": 4.162406921386719e-05,
      "step": 43646,
      "training_step_time": 7.611126661300659
    },
    {
      "epoch": 4.162502288818359e-05,
      "model_forward_time": 1.6828820705413818,
      "step": 43647
    },
    {
      "epoch": 4.162502288818359e-05,
      "step": 43647,
      "training_step_time": 7.57377552986145
    },
    {
      "epoch": 4.16259765625e-05,
      "model_forward_time": 1.6898386478424072,
      "step": 43648
    },
    {
      "epoch": 4.16259765625e-05,
      "step": 43648,
      "training_step_time": 7.545764207839966
    },
    {
      "epoch": 4.162693023681641e-05,
      "model_forward_time": 1.6893043518066406,
      "step": 43649
    },
    {
      "epoch": 4.162693023681641e-05,
      "step": 43649,
      "training_step_time": 9.215855121612549
    },
    {
      "epoch": 4.162788391113281e-05,
      "grad_norm": 0.09789275377988815,
      "learning_rate": 6.442678915947023e-06,
      "loss": 0.0547,
      "step": 43650
    },
    {
      "epoch": 4.162788391113281e-05,
      "model_forward_time": 1.6893303394317627,
      "step": 43650
    },
    {
      "dynamics_loss_avg": 0.047396695055067536,
      "epoch": 4.162788391113281e-05,
      "step": 43650
    },
    {
      "action_loss_avg": 0.001512975772493519,
      "epoch": 4.162788391113281e-05,
      "step": 43650
    },
    {
      "epoch": 4.162788391113281e-05,
      "step": 43650,
      "training_step_time": 8.386432409286499
    },
    {
      "epoch": 4.162883758544922e-05,
      "model_forward_time": 1.7120695114135742,
      "step": 43651
    },
    {
      "epoch": 4.162883758544922e-05,
      "step": 43651,
      "training_step_time": 7.634823322296143
    },
    {
      "epoch": 4.1629791259765626e-05,
      "model_forward_time": 1.6969411373138428,
      "step": 43652
    },
    {
      "epoch": 4.1629791259765626e-05,
      "step": 43652,
      "training_step_time": 8.651993036270142
    },
    {
      "epoch": 4.163074493408203e-05,
      "model_forward_time": 1.6927821636199951,
      "step": 43653
    },
    {
      "epoch": 4.163074493408203e-05,
      "step": 43653,
      "training_step_time": 7.842959403991699
    },
    {
      "epoch": 4.163169860839844e-05,
      "model_forward_time": 1.7048828601837158,
      "step": 43654
    },
    {
      "epoch": 4.163169860839844e-05,
      "step": 43654,
      "training_step_time": 7.558640241622925
    },
    {
      "epoch": 4.1632652282714845e-05,
      "model_forward_time": 2.832017183303833,
      "step": 43655
    },
    {
      "epoch": 4.1632652282714845e-05,
      "step": 43655,
      "training_step_time": 7.592047691345215
    },
    {
      "epoch": 4.163360595703125e-05,
      "model_forward_time": 1.687950849533081,
      "step": 43656
    },
    {
      "epoch": 4.163360595703125e-05,
      "step": 43656,
      "training_step_time": 7.516763687133789
    },
    {
      "epoch": 4.163455963134765e-05,
      "model_forward_time": 1.6952810287475586,
      "step": 43657
    },
    {
      "epoch": 4.163455963134765e-05,
      "step": 43657,
      "training_step_time": 8.239768505096436
    },
    {
      "epoch": 4.1635513305664064e-05,
      "model_forward_time": 1.6888291835784912,
      "step": 43658
    },
    {
      "epoch": 4.1635513305664064e-05,
      "step": 43658,
      "training_step_time": 7.6270647048950195
    },
    {
      "epoch": 4.163646697998047e-05,
      "model_forward_time": 1.6929354667663574,
      "step": 43659
    },
    {
      "epoch": 4.163646697998047e-05,
      "step": 43659,
      "training_step_time": 9.38437819480896
    },
    {
      "epoch": 4.163742065429687e-05,
      "grad_norm": 0.11333911865949631,
      "learning_rate": 6.441095690812031e-06,
      "loss": 0.0585,
      "step": 43660
    },
    {
      "epoch": 4.163742065429687e-05,
      "model_forward_time": 1.7076992988586426,
      "step": 43660
    },
    {
      "dynamics_loss_avg": 0.0791452307254076,
      "epoch": 4.163742065429687e-05,
      "step": 43660
    },
    {
      "action_loss_avg": 0.0020739259489346296,
      "epoch": 4.163742065429687e-05,
      "step": 43660
    },
    {
      "epoch": 4.163742065429687e-05,
      "step": 43660,
      "training_step_time": 8.55168628692627
    },
    {
      "epoch": 4.163837432861328e-05,
      "model_forward_time": 1.696077823638916,
      "step": 43661
    },
    {
      "epoch": 4.163837432861328e-05,
      "step": 43661,
      "training_step_time": 7.646222829818726
    },
    {
      "epoch": 4.1639328002929687e-05,
      "model_forward_time": 1.6917510032653809,
      "step": 43662
    },
    {
      "epoch": 4.1639328002929687e-05,
      "step": 43662,
      "training_step_time": 7.763586521148682
    },
    {
      "epoch": 4.164028167724609e-05,
      "model_forward_time": 1.7030656337738037,
      "step": 43663
    },
    {
      "epoch": 4.164028167724609e-05,
      "step": 43663,
      "training_step_time": 7.6390159130096436
    },
    {
      "epoch": 4.16412353515625e-05,
      "model_forward_time": 2.7739365100860596,
      "step": 43664
    },
    {
      "epoch": 4.16412353515625e-05,
      "step": 43664,
      "training_step_time": 7.6448915004730225
    },
    {
      "epoch": 4.1642189025878905e-05,
      "model_forward_time": 1.68947172164917,
      "step": 43665
    },
    {
      "epoch": 4.1642189025878905e-05,
      "step": 43665,
      "training_step_time": 7.743946075439453
    },
    {
      "epoch": 4.164314270019531e-05,
      "model_forward_time": 1.691910982131958,
      "step": 43666
    },
    {
      "epoch": 4.164314270019531e-05,
      "step": 43666,
      "training_step_time": 7.683650732040405
    },
    {
      "epoch": 4.164409637451172e-05,
      "model_forward_time": 1.69315505027771,
      "step": 43667
    },
    {
      "epoch": 4.164409637451172e-05,
      "step": 43667,
      "training_step_time": 8.350937128067017
    },
    {
      "epoch": 4.1645050048828124e-05,
      "model_forward_time": 1.7058026790618896,
      "step": 43668
    },
    {
      "epoch": 4.1645050048828124e-05,
      "step": 43668,
      "training_step_time": 9.190048456192017
    },
    {
      "epoch": 4.164600372314453e-05,
      "model_forward_time": 1.6915838718414307,
      "step": 43669
    },
    {
      "epoch": 4.164600372314453e-05,
      "step": 43669,
      "training_step_time": 8.686776638031006
    },
    {
      "epoch": 4.164695739746094e-05,
      "grad_norm": 0.12068114429712296,
      "learning_rate": 6.4395123080809815e-06,
      "loss": 0.056,
      "step": 43670
    },
    {
      "epoch": 4.164695739746094e-05,
      "model_forward_time": 1.6876051425933838,
      "step": 43670
    },
    {
      "dynamics_loss_avg": 0.05586917102336884,
      "epoch": 4.164695739746094e-05,
      "step": 43670
    },
    {
      "action_loss_avg": 0.0016424891335191205,
      "epoch": 4.164695739746094e-05,
      "step": 43670
    },
    {
      "epoch": 4.164695739746094e-05,
      "step": 43670,
      "training_step_time": 7.751878261566162
    },
    {
      "epoch": 4.164791107177734e-05,
      "model_forward_time": 1.6942510604858398,
      "step": 43671
    },
    {
      "epoch": 4.164791107177734e-05,
      "step": 43671,
      "training_step_time": 7.697390079498291
    },
    {
      "epoch": 4.164886474609375e-05,
      "model_forward_time": 1.6867377758026123,
      "step": 43672
    },
    {
      "epoch": 4.164886474609375e-05,
      "step": 43672,
      "training_step_time": 8.488966226577759
    },
    {
      "epoch": 4.164981842041016e-05,
      "model_forward_time": 1.6908056735992432,
      "step": 43673
    },
    {
      "epoch": 4.164981842041016e-05,
      "step": 43673,
      "training_step_time": 9.184458255767822
    },
    {
      "epoch": 4.165077209472656e-05,
      "model_forward_time": 1.6942641735076904,
      "step": 43674
    },
    {
      "epoch": 4.165077209472656e-05,
      "step": 43674,
      "training_step_time": 9.151427507400513
    },
    {
      "epoch": 4.1651725769042966e-05,
      "model_forward_time": 1.6946148872375488,
      "step": 43675
    },
    {
      "epoch": 4.1651725769042966e-05,
      "step": 43675,
      "training_step_time": 7.644427299499512
    },
    {
      "epoch": 4.165267944335938e-05,
      "model_forward_time": 1.6892199516296387,
      "step": 43676
    },
    {
      "epoch": 4.165267944335938e-05,
      "step": 43676,
      "training_step_time": 7.789870738983154
    },
    {
      "epoch": 4.165363311767578e-05,
      "model_forward_time": 1.692387342453003,
      "step": 43677
    },
    {
      "epoch": 4.165363311767578e-05,
      "step": 43677,
      "training_step_time": 8.636164665222168
    },
    {
      "epoch": 4.1654586791992185e-05,
      "model_forward_time": 1.6948027610778809,
      "step": 43678
    },
    {
      "epoch": 4.1654586791992185e-05,
      "step": 43678,
      "training_step_time": 8.77560019493103
    },
    {
      "epoch": 4.1655540466308596e-05,
      "model_forward_time": 1.7249059677124023,
      "step": 43679
    },
    {
      "epoch": 4.1655540466308596e-05,
      "step": 43679,
      "training_step_time": 8.405949592590332
    },
    {
      "epoch": 4.1656494140625e-05,
      "grad_norm": 0.1113971695303917,
      "learning_rate": 6.437928767927031e-06,
      "loss": 0.0573,
      "step": 43680
    },
    {
      "epoch": 4.1656494140625e-05,
      "model_forward_time": 1.689232587814331,
      "step": 43680
    },
    {
      "dynamics_loss_avg": 0.04062701351940632,
      "epoch": 4.1656494140625e-05,
      "step": 43680
    },
    {
      "action_loss_avg": 0.0012831988919060678,
      "epoch": 4.1656494140625e-05,
      "step": 43680
    },
    {
      "epoch": 4.1656494140625e-05,
      "step": 43680,
      "training_step_time": 7.791077136993408
    },
    {
      "epoch": 4.1657447814941404e-05,
      "model_forward_time": 1.7053840160369873,
      "step": 43681
    },
    {
      "epoch": 4.1657447814941404e-05,
      "step": 43681,
      "training_step_time": 8.039156198501587
    },
    {
      "epoch": 4.1658401489257815e-05,
      "model_forward_time": 1.7278180122375488,
      "step": 43682
    },
    {
      "epoch": 4.1658401489257815e-05,
      "step": 43682,
      "training_step_time": 7.7718024253845215
    },
    {
      "epoch": 4.165935516357422e-05,
      "model_forward_time": 2.845947504043579,
      "step": 43683
    },
    {
      "epoch": 4.165935516357422e-05,
      "step": 43683,
      "training_step_time": 7.8601391315460205
    },
    {
      "epoch": 4.166030883789062e-05,
      "model_forward_time": 1.705476999282837,
      "step": 43684
    },
    {
      "epoch": 4.166030883789062e-05,
      "step": 43684,
      "training_step_time": 9.152417421340942
    },
    {
      "epoch": 4.1661262512207033e-05,
      "model_forward_time": 1.6916475296020508,
      "step": 43685
    },
    {
      "epoch": 4.1661262512207033e-05,
      "step": 43685,
      "training_step_time": 8.855775594711304
    },
    {
      "epoch": 4.166221618652344e-05,
      "model_forward_time": 1.696897268295288,
      "step": 43686
    },
    {
      "epoch": 4.166221618652344e-05,
      "step": 43686,
      "training_step_time": 8.501258373260498
    },
    {
      "epoch": 4.166316986083984e-05,
      "model_forward_time": 1.6970760822296143,
      "step": 43687
    },
    {
      "epoch": 4.166316986083984e-05,
      "step": 43687,
      "training_step_time": 8.656817197799683
    },
    {
      "epoch": 4.166412353515625e-05,
      "model_forward_time": 1.7261390686035156,
      "step": 43688
    },
    {
      "epoch": 4.166412353515625e-05,
      "step": 43688,
      "training_step_time": 7.820549726486206
    },
    {
      "epoch": 4.1665077209472656e-05,
      "model_forward_time": 1.7018425464630127,
      "step": 43689
    },
    {
      "epoch": 4.1665077209472656e-05,
      "step": 43689,
      "training_step_time": 7.711927175521851
    },
    {
      "epoch": 4.166603088378906e-05,
      "grad_norm": 0.13395564258098602,
      "learning_rate": 6.436345070523351e-06,
      "loss": 0.0558,
      "step": 43690
    },
    {
      "epoch": 4.166603088378906e-05,
      "model_forward_time": 1.7047581672668457,
      "step": 43690
    },
    {
      "dynamics_loss_avg": 0.05556758493185043,
      "epoch": 4.166603088378906e-05,
      "step": 43690
    },
    {
      "action_loss_avg": 0.0025438719312660396,
      "epoch": 4.166603088378906e-05,
      "step": 43690
    },
    {
      "epoch": 4.166603088378906e-05,
      "step": 43690,
      "training_step_time": 7.7873735427856445
    },
    {
      "epoch": 4.166698455810547e-05,
      "model_forward_time": 1.717646837234497,
      "step": 43691
    },
    {
      "epoch": 4.166698455810547e-05,
      "step": 43691,
      "training_step_time": 8.689450740814209
    },
    {
      "epoch": 4.1667938232421875e-05,
      "model_forward_time": 2.814300537109375,
      "step": 43692
    },
    {
      "epoch": 4.1667938232421875e-05,
      "step": 43692,
      "training_step_time": 7.890270233154297
    },
    {
      "epoch": 4.166889190673828e-05,
      "model_forward_time": 1.6983165740966797,
      "step": 43693
    },
    {
      "epoch": 4.166889190673828e-05,
      "step": 43693,
      "training_step_time": 7.541539907455444
    },
    {
      "epoch": 4.166984558105469e-05,
      "model_forward_time": 1.6997942924499512,
      "step": 43694
    },
    {
      "epoch": 4.166984558105469e-05,
      "step": 43694,
      "training_step_time": 7.6870996952056885
    },
    {
      "epoch": 4.1670799255371094e-05,
      "model_forward_time": 1.6935384273529053,
      "step": 43695
    },
    {
      "epoch": 4.1670799255371094e-05,
      "step": 43695,
      "training_step_time": 7.634417772293091
    },
    {
      "epoch": 4.16717529296875e-05,
      "model_forward_time": 1.7056903839111328,
      "step": 43696
    },
    {
      "epoch": 4.16717529296875e-05,
      "step": 43696,
      "training_step_time": 7.704502820968628
    },
    {
      "epoch": 4.167270660400391e-05,
      "model_forward_time": 1.7225563526153564,
      "step": 43697
    },
    {
      "epoch": 4.167270660400391e-05,
      "step": 43697,
      "training_step_time": 7.6229636669158936
    },
    {
      "epoch": 4.167366027832031e-05,
      "model_forward_time": 1.701589584350586,
      "step": 43698
    },
    {
      "epoch": 4.167366027832031e-05,
      "step": 43698,
      "training_step_time": 7.671680212020874
    },
    {
      "epoch": 4.167461395263672e-05,
      "model_forward_time": 1.7002577781677246,
      "step": 43699
    },
    {
      "epoch": 4.167461395263672e-05,
      "step": 43699,
      "training_step_time": 8.181750535964966
    },
    {
      "epoch": 4.167556762695313e-05,
      "grad_norm": 0.18798968195915222,
      "learning_rate": 6.434761216043135e-06,
      "loss": 0.0555,
      "step": 43700
    },
    {
      "epoch": 4.167556762695313e-05,
      "model_forward_time": 1.7015161514282227,
      "step": 43700
    },
    {
      "dynamics_loss_avg": 0.04768920298665762,
      "epoch": 4.167556762695313e-05,
      "step": 43700
    },
    {
      "action_loss_avg": 0.0012457617733161897,
      "epoch": 4.167556762695313e-05,
      "step": 43700
    },
    {
      "epoch": 4.167556762695313e-05,
      "step": 43700,
      "training_step_time": 8.431707382202148
    },
    {
      "epoch": 4.167652130126953e-05,
      "model_forward_time": 2.822331666946411,
      "step": 43701
    },
    {
      "epoch": 4.167652130126953e-05,
      "step": 43701,
      "training_step_time": 7.612756967544556
    },
    {
      "epoch": 4.1677474975585936e-05,
      "model_forward_time": 1.69866943359375,
      "step": 43702
    },
    {
      "epoch": 4.1677474975585936e-05,
      "step": 43702,
      "training_step_time": 7.600952386856079
    },
    {
      "epoch": 4.1678428649902347e-05,
      "model_forward_time": 1.697829246520996,
      "step": 43703
    },
    {
      "epoch": 4.1678428649902347e-05,
      "step": 43703,
      "training_step_time": 7.581691265106201
    },
    {
      "epoch": 4.167938232421875e-05,
      "model_forward_time": 1.6976912021636963,
      "step": 43704
    },
    {
      "epoch": 4.167938232421875e-05,
      "step": 43704,
      "training_step_time": 7.584627389907837
    },
    {
      "epoch": 4.1680335998535155e-05,
      "model_forward_time": 1.6969237327575684,
      "step": 43705
    },
    {
      "epoch": 4.1680335998535155e-05,
      "step": 43705,
      "training_step_time": 7.678694248199463
    },
    {
      "epoch": 4.1681289672851565e-05,
      "model_forward_time": 1.724391222000122,
      "step": 43706
    },
    {
      "epoch": 4.1681289672851565e-05,
      "step": 43706,
      "training_step_time": 7.592038869857788
    },
    {
      "epoch": 4.168224334716797e-05,
      "model_forward_time": 1.700242519378662,
      "step": 43707
    },
    {
      "epoch": 4.168224334716797e-05,
      "step": 43707,
      "training_step_time": 9.033523321151733
    },
    {
      "epoch": 4.1683197021484373e-05,
      "model_forward_time": 1.6962265968322754,
      "step": 43708
    },
    {
      "epoch": 4.1683197021484373e-05,
      "step": 43708,
      "training_step_time": 7.446906805038452
    },
    {
      "epoch": 4.1684150695800784e-05,
      "model_forward_time": 1.6961419582366943,
      "step": 43709
    },
    {
      "epoch": 4.1684150695800784e-05,
      "step": 43709,
      "training_step_time": 7.846715688705444
    },
    {
      "epoch": 4.168510437011719e-05,
      "grad_norm": 0.11261703073978424,
      "learning_rate": 6.433177204659588e-06,
      "loss": 0.0577,
      "step": 43710
    },
    {
      "epoch": 4.168510437011719e-05,
      "model_forward_time": 2.824862480163574,
      "step": 43710
    },
    {
      "dynamics_loss_avg": 0.04594055265188217,
      "epoch": 4.168510437011719e-05,
      "step": 43710
    },
    {
      "action_loss_avg": 0.0014739171878318303,
      "epoch": 4.168510437011719e-05,
      "step": 43710
    },
    {
      "epoch": 4.168510437011719e-05,
      "step": 43710,
      "training_step_time": 7.8514745235443115
    },
    {
      "epoch": 4.168605804443359e-05,
      "model_forward_time": 1.7001550197601318,
      "step": 43711
    },
    {
      "epoch": 4.168605804443359e-05,
      "step": 43711,
      "training_step_time": 7.496518611907959
    },
    {
      "epoch": 4.168701171875e-05,
      "model_forward_time": 1.7001965045928955,
      "step": 43712
    },
    {
      "epoch": 4.168701171875e-05,
      "step": 43712,
      "training_step_time": 7.428934335708618
    },
    {
      "epoch": 4.168796539306641e-05,
      "model_forward_time": 1.7315571308135986,
      "step": 43713
    },
    {
      "epoch": 4.168796539306641e-05,
      "step": 43713,
      "training_step_time": 7.566404342651367
    },
    {
      "epoch": 4.168891906738281e-05,
      "model_forward_time": 1.700700283050537,
      "step": 43714
    },
    {
      "epoch": 4.168891906738281e-05,
      "step": 43714,
      "training_step_time": 8.160923480987549
    },
    {
      "epoch": 4.168987274169922e-05,
      "model_forward_time": 1.7215924263000488,
      "step": 43715
    },
    {
      "epoch": 4.168987274169922e-05,
      "step": 43715,
      "training_step_time": 7.5524656772613525
    },
    {
      "epoch": 4.1690826416015626e-05,
      "model_forward_time": 1.7335801124572754,
      "step": 43716
    },
    {
      "epoch": 4.1690826416015626e-05,
      "step": 43716,
      "training_step_time": 7.627155303955078
    },
    {
      "epoch": 4.169178009033203e-05,
      "model_forward_time": 1.7272412776947021,
      "step": 43717
    },
    {
      "epoch": 4.169178009033203e-05,
      "step": 43717,
      "training_step_time": 7.590622901916504
    },
    {
      "epoch": 4.169273376464844e-05,
      "model_forward_time": 1.729611873626709,
      "step": 43718
    },
    {
      "epoch": 4.169273376464844e-05,
      "step": 43718,
      "training_step_time": 7.877847194671631
    },
    {
      "epoch": 4.1693687438964845e-05,
      "model_forward_time": 2.289212703704834,
      "step": 43719
    },
    {
      "epoch": 4.1693687438964845e-05,
      "step": 43719,
      "training_step_time": 7.804021120071411
    },
    {
      "epoch": 4.169464111328125e-05,
      "grad_norm": 0.09746135026216507,
      "learning_rate": 6.431593036545937e-06,
      "loss": 0.0605,
      "step": 43720
    },
    {
      "epoch": 4.169464111328125e-05,
      "model_forward_time": 1.7531707286834717,
      "step": 43720
    },
    {
      "dynamics_loss_avg": 0.042053766548633575,
      "epoch": 4.169464111328125e-05,
      "step": 43720
    },
    {
      "action_loss_avg": 0.0011281907325610518,
      "epoch": 4.169464111328125e-05,
      "step": 43720
    },
    {
      "epoch": 4.169464111328125e-05,
      "step": 43720,
      "training_step_time": 8.503174543380737
    },
    {
      "epoch": 4.169559478759765e-05,
      "model_forward_time": 1.7286956310272217,
      "step": 43721
    },
    {
      "epoch": 4.169559478759765e-05,
      "step": 43721,
      "training_step_time": 7.592985153198242
    },
    {
      "epoch": 4.1696548461914064e-05,
      "model_forward_time": 1.7348475456237793,
      "step": 43722
    },
    {
      "epoch": 4.1696548461914064e-05,
      "step": 43722,
      "training_step_time": 8.410730123519897
    },
    {
      "epoch": 4.169750213623047e-05,
      "model_forward_time": 1.732128620147705,
      "step": 43723
    },
    {
      "epoch": 4.169750213623047e-05,
      "step": 43723,
      "training_step_time": 9.809492349624634
    },
    {
      "epoch": 4.169845581054687e-05,
      "model_forward_time": 1.7198350429534912,
      "step": 43724
    },
    {
      "epoch": 4.169845581054687e-05,
      "step": 43724,
      "training_step_time": 9.327239274978638
    },
    {
      "epoch": 4.169940948486328e-05,
      "model_forward_time": 1.7329535484313965,
      "step": 43725
    },
    {
      "epoch": 4.169940948486328e-05,
      "step": 43725,
      "training_step_time": 6.8366029262542725
    },
    {
      "epoch": 4.170036315917969e-05,
      "model_forward_time": 1.730072021484375,
      "step": 43726
    },
    {
      "epoch": 4.170036315917969e-05,
      "step": 43726,
      "training_step_time": 6.846526622772217
    },
    {
      "epoch": 4.170131683349609e-05,
      "model_forward_time": 1.7235801219940186,
      "step": 43727
    },
    {
      "epoch": 4.170131683349609e-05,
      "step": 43727,
      "training_step_time": 6.874818325042725
    },
    {
      "epoch": 4.17022705078125e-05,
      "model_forward_time": 1.7263998985290527,
      "step": 43728
    },
    {
      "epoch": 4.17022705078125e-05,
      "step": 43728,
      "training_step_time": 6.7871973514556885
    },
    {
      "epoch": 4.1703224182128905e-05,
      "model_forward_time": 1.7874581813812256,
      "step": 43729
    },
    {
      "epoch": 4.1703224182128905e-05,
      "step": 43729,
      "training_step_time": 6.852487325668335
    },
    {
      "epoch": 4.170417785644531e-05,
      "grad_norm": 0.10256543010473251,
      "learning_rate": 6.4300087118754266e-06,
      "loss": 0.0612,
      "step": 43730
    },
    {
      "epoch": 4.170417785644531e-05,
      "model_forward_time": 1.815307855606079,
      "step": 43730
    },
    {
      "dynamics_loss_avg": 0.06212140507996082,
      "epoch": 4.170417785644531e-05,
      "step": 43730
    },
    {
      "action_loss_avg": 0.0024452709301840516,
      "epoch": 4.170417785644531e-05,
      "step": 43730
    },
    {
      "epoch": 4.170417785644531e-05,
      "step": 43730,
      "training_step_time": 6.840758323669434
    },
    {
      "epoch": 4.170513153076172e-05,
      "model_forward_time": 1.8158495426177979,
      "step": 43731
    },
    {
      "epoch": 4.170513153076172e-05,
      "step": 43731,
      "training_step_time": 6.841416835784912
    },
    {
      "epoch": 4.1706085205078124e-05,
      "model_forward_time": 1.8150296211242676,
      "step": 43732
    },
    {
      "epoch": 4.1706085205078124e-05,
      "step": 43732,
      "training_step_time": 6.844188213348389
    },
    {
      "epoch": 4.170703887939453e-05,
      "model_forward_time": 2.2983927726745605,
      "step": 43733
    },
    {
      "epoch": 4.170703887939453e-05,
      "step": 43733,
      "training_step_time": 7.64962363243103
    },
    {
      "epoch": 4.170799255371094e-05,
      "model_forward_time": 1.8055732250213623,
      "step": 43734
    },
    {
      "epoch": 4.170799255371094e-05,
      "step": 43734,
      "training_step_time": 7.902042388916016
    },
    {
      "epoch": 4.170894622802734e-05,
      "model_forward_time": 1.81074857711792,
      "step": 43735
    },
    {
      "epoch": 4.170894622802734e-05,
      "step": 43735,
      "training_step_time": 6.818115234375
    },
    {
      "epoch": 4.170989990234375e-05,
      "model_forward_time": 1.8094313144683838,
      "step": 43736
    },
    {
      "epoch": 4.170989990234375e-05,
      "step": 43736,
      "training_step_time": 6.869096040725708
    },
    {
      "epoch": 4.171085357666016e-05,
      "model_forward_time": 1.8091380596160889,
      "step": 43737
    },
    {
      "epoch": 4.171085357666016e-05,
      "step": 43737,
      "training_step_time": 6.962333679199219
    },
    {
      "epoch": 4.171180725097656e-05,
      "model_forward_time": 1.8120999336242676,
      "step": 43738
    },
    {
      "epoch": 4.171180725097656e-05,
      "step": 43738,
      "training_step_time": 6.985419511795044
    },
    {
      "epoch": 4.1712760925292966e-05,
      "model_forward_time": 1.8105554580688477,
      "step": 43739
    },
    {
      "epoch": 4.1712760925292966e-05,
      "step": 43739,
      "training_step_time": 6.9030845165252686
    },
    {
      "epoch": 4.171371459960938e-05,
      "grad_norm": 0.11267149448394775,
      "learning_rate": 6.4284242308213096e-06,
      "loss": 0.0594,
      "step": 43740
    },
    {
      "epoch": 4.171371459960938e-05,
      "model_forward_time": 1.8115904331207275,
      "step": 43740
    },
    {
      "dynamics_loss_avg": 0.07201535943895579,
      "epoch": 4.171371459960938e-05,
      "step": 43740
    },
    {
      "action_loss_avg": 0.0017867775459308177,
      "epoch": 4.171371459960938e-05,
      "step": 43740
    },
    {
      "epoch": 4.171371459960938e-05,
      "step": 43740,
      "training_step_time": 6.840576887130737
    },
    {
      "epoch": 4.171466827392578e-05,
      "model_forward_time": 1.8101599216461182,
      "step": 43741
    },
    {
      "epoch": 4.171466827392578e-05,
      "step": 43741,
      "training_step_time": 6.819596529006958
    },
    {
      "epoch": 4.1715621948242185e-05,
      "model_forward_time": 2.347438335418701,
      "step": 43742
    },
    {
      "epoch": 4.1715621948242185e-05,
      "step": 43742,
      "training_step_time": 7.683689832687378
    },
    {
      "epoch": 4.1716575622558596e-05,
      "model_forward_time": 1.8112635612487793,
      "step": 43743
    },
    {
      "epoch": 4.1716575622558596e-05,
      "step": 43743,
      "training_step_time": 9.69320797920227
    },
    {
      "epoch": 4.1717529296875e-05,
      "model_forward_time": 1.8137173652648926,
      "step": 43744
    },
    {
      "epoch": 4.1717529296875e-05,
      "step": 43744,
      "training_step_time": 7.7404444217681885
    },
    {
      "epoch": 4.1718482971191404e-05,
      "model_forward_time": 1.8079407215118408,
      "step": 43745
    },
    {
      "epoch": 4.1718482971191404e-05,
      "step": 43745,
      "training_step_time": 6.822399139404297
    },
    {
      "epoch": 4.1719436645507815e-05,
      "model_forward_time": 1.8172385692596436,
      "step": 43746
    },
    {
      "epoch": 4.1719436645507815e-05,
      "step": 43746,
      "training_step_time": 6.848179578781128
    },
    {
      "epoch": 4.172039031982422e-05,
      "model_forward_time": 1.806908130645752,
      "step": 43747
    },
    {
      "epoch": 4.172039031982422e-05,
      "step": 43747,
      "training_step_time": 6.873429298400879
    },
    {
      "epoch": 4.172134399414062e-05,
      "model_forward_time": 1.80430269241333,
      "step": 43748
    },
    {
      "epoch": 4.172134399414062e-05,
      "step": 43748,
      "training_step_time": 6.866512060165405
    },
    {
      "epoch": 4.1722297668457033e-05,
      "model_forward_time": 1.8061814308166504,
      "step": 43749
    },
    {
      "epoch": 4.1722297668457033e-05,
      "step": 43749,
      "training_step_time": 6.87678337097168
    },
    {
      "epoch": 4.172325134277344e-05,
      "grad_norm": 0.12985336780548096,
      "learning_rate": 6.426839593556868e-06,
      "loss": 0.0635,
      "step": 43750
    },
    {
      "epoch": 4.172325134277344e-05,
      "model_forward_time": 1.7874248027801514,
      "step": 43750
    },
    {
      "dynamics_loss_avg": 0.049827251583337784,
      "epoch": 4.172325134277344e-05,
      "step": 43750
    },
    {
      "action_loss_avg": 0.001241447072243318,
      "epoch": 4.172325134277344e-05,
      "step": 43750
    },
    {
      "epoch": 4.172325134277344e-05,
      "step": 43750,
      "training_step_time": 6.922371864318848
    },
    {
      "epoch": 4.172420501708984e-05,
      "model_forward_time": 2.347799062728882,
      "step": 43751
    },
    {
      "epoch": 4.172420501708984e-05,
      "step": 43751,
      "training_step_time": 7.591671943664551
    },
    {
      "epoch": 4.172515869140625e-05,
      "model_forward_time": 1.810772180557251,
      "step": 43752
    },
    {
      "epoch": 4.172515869140625e-05,
      "step": 43752,
      "training_step_time": 10.516434669494629
    },
    {
      "epoch": 4.1726112365722656e-05,
      "model_forward_time": 1.801008939743042,
      "step": 43753
    },
    {
      "epoch": 4.1726112365722656e-05,
      "step": 43753,
      "training_step_time": 7.948363780975342
    },
    {
      "epoch": 4.172706604003906e-05,
      "model_forward_time": 1.7883758544921875,
      "step": 43754
    },
    {
      "epoch": 4.172706604003906e-05,
      "step": 43754,
      "training_step_time": 6.887600421905518
    },
    {
      "epoch": 4.172801971435547e-05,
      "model_forward_time": 1.8022243976593018,
      "step": 43755
    },
    {
      "epoch": 4.172801971435547e-05,
      "step": 43755,
      "training_step_time": 6.842328786849976
    },
    {
      "epoch": 4.1728973388671875e-05,
      "model_forward_time": 1.8049566745758057,
      "step": 43756
    },
    {
      "epoch": 4.1728973388671875e-05,
      "step": 43756,
      "training_step_time": 6.84546422958374
    },
    {
      "epoch": 4.172992706298828e-05,
      "model_forward_time": 1.8052330017089844,
      "step": 43757
    },
    {
      "epoch": 4.172992706298828e-05,
      "step": 43757,
      "training_step_time": 6.8510401248931885
    },
    {
      "epoch": 4.173088073730469e-05,
      "model_forward_time": 1.437790870666504,
      "step": 43758
    },
    {
      "epoch": 4.173088073730469e-05,
      "step": 43758,
      "training_step_time": 6.872195482254028
    },
    {
      "epoch": 4.1731834411621094e-05,
      "model_forward_time": 1.8082993030548096,
      "step": 43759
    },
    {
      "epoch": 4.1731834411621094e-05,
      "step": 43759,
      "training_step_time": 6.88126540184021
    },
    {
      "epoch": 4.17327880859375e-05,
      "grad_norm": 0.14452989399433136,
      "learning_rate": 6.425254800255393e-06,
      "loss": 0.0613,
      "step": 43760
    },
    {
      "epoch": 4.17327880859375e-05,
      "model_forward_time": 2.3662524223327637,
      "step": 43760
    },
    {
      "dynamics_loss_avg": 0.055041706375777724,
      "epoch": 4.17327880859375e-05,
      "step": 43760
    },
    {
      "action_loss_avg": 0.0017779989808332174,
      "epoch": 4.17327880859375e-05,
      "step": 43760
    },
    {
      "epoch": 4.17327880859375e-05,
      "step": 43760,
      "training_step_time": 7.64257550239563
    },
    {
      "epoch": 4.173374176025391e-05,
      "model_forward_time": 1.8051137924194336,
      "step": 43761
    },
    {
      "epoch": 4.173374176025391e-05,
      "step": 43761,
      "training_step_time": 10.982162952423096
    },
    {
      "epoch": 4.173469543457031e-05,
      "model_forward_time": 1.8065342903137207,
      "step": 43762
    },
    {
      "epoch": 4.173469543457031e-05,
      "step": 43762,
      "training_step_time": 7.867745876312256
    },
    {
      "epoch": 4.173564910888672e-05,
      "model_forward_time": 1.8029170036315918,
      "step": 43763
    },
    {
      "epoch": 4.173564910888672e-05,
      "step": 43763,
      "training_step_time": 6.816395282745361
    },
    {
      "epoch": 4.173660278320313e-05,
      "model_forward_time": 1.8097543716430664,
      "step": 43764
    },
    {
      "epoch": 4.173660278320313e-05,
      "step": 43764,
      "training_step_time": 6.809243202209473
    },
    {
      "epoch": 4.173755645751953e-05,
      "model_forward_time": 1.8145246505737305,
      "step": 43765
    },
    {
      "epoch": 4.173755645751953e-05,
      "step": 43765,
      "training_step_time": 6.827643156051636
    },
    {
      "epoch": 4.1738510131835936e-05,
      "model_forward_time": 1.8039426803588867,
      "step": 43766
    },
    {
      "epoch": 4.1738510131835936e-05,
      "step": 43766,
      "training_step_time": 6.823394298553467
    },
    {
      "epoch": 4.173946380615235e-05,
      "model_forward_time": 1.7991769313812256,
      "step": 43767
    },
    {
      "epoch": 4.173946380615235e-05,
      "step": 43767,
      "training_step_time": 6.8286497592926025
    },
    {
      "epoch": 4.174041748046875e-05,
      "model_forward_time": 1.8104102611541748,
      "step": 43768
    },
    {
      "epoch": 4.174041748046875e-05,
      "step": 43768,
      "training_step_time": 6.843183279037476
    },
    {
      "epoch": 4.1741371154785155e-05,
      "model_forward_time": 2.3562774658203125,
      "step": 43769
    },
    {
      "epoch": 4.1741371154785155e-05,
      "step": 43769,
      "training_step_time": 7.624184608459473
    },
    {
      "epoch": 4.1742324829101565e-05,
      "grad_norm": 0.13944770395755768,
      "learning_rate": 6.423669851090197e-06,
      "loss": 0.0639,
      "step": 43770
    },
    {
      "epoch": 4.1742324829101565e-05,
      "model_forward_time": 1.8050270080566406,
      "step": 43770
    },
    {
      "dynamics_loss_avg": 0.06139011941850185,
      "epoch": 4.1742324829101565e-05,
      "step": 43770
    },
    {
      "action_loss_avg": 0.0011619756958680227,
      "epoch": 4.1742324829101565e-05,
      "step": 43770
    },
    {
      "epoch": 4.1742324829101565e-05,
      "step": 43770,
      "training_step_time": 9.412049293518066
    },
    {
      "epoch": 4.174327850341797e-05,
      "model_forward_time": 1.8107926845550537,
      "step": 43771
    },
    {
      "epoch": 4.174327850341797e-05,
      "step": 43771,
      "training_step_time": 7.996896028518677
    },
    {
      "epoch": 4.1744232177734374e-05,
      "model_forward_time": 1.8121755123138428,
      "step": 43772
    },
    {
      "epoch": 4.1744232177734374e-05,
      "step": 43772,
      "training_step_time": 6.844479322433472
    },
    {
      "epoch": 4.1745185852050784e-05,
      "model_forward_time": 1.8013217449188232,
      "step": 43773
    },
    {
      "epoch": 4.1745185852050784e-05,
      "step": 43773,
      "training_step_time": 6.837065935134888
    },
    {
      "epoch": 4.174613952636719e-05,
      "model_forward_time": 1.8105125427246094,
      "step": 43774
    },
    {
      "epoch": 4.174613952636719e-05,
      "step": 43774,
      "training_step_time": 6.908732652664185
    },
    {
      "epoch": 4.174709320068359e-05,
      "model_forward_time": 1.8057498931884766,
      "step": 43775
    },
    {
      "epoch": 4.174709320068359e-05,
      "step": 43775,
      "training_step_time": 6.852726459503174
    },
    {
      "epoch": 4.1748046875e-05,
      "model_forward_time": 1.8043792247772217,
      "step": 43776
    },
    {
      "epoch": 4.1748046875e-05,
      "step": 43776,
      "training_step_time": 6.836658239364624
    },
    {
      "epoch": 4.174900054931641e-05,
      "model_forward_time": 1.8064379692077637,
      "step": 43777
    },
    {
      "epoch": 4.174900054931641e-05,
      "step": 43777,
      "training_step_time": 6.845189809799194
    },
    {
      "epoch": 4.174995422363281e-05,
      "model_forward_time": 1.8066978454589844,
      "step": 43778
    },
    {
      "epoch": 4.174995422363281e-05,
      "step": 43778,
      "training_step_time": 7.628103017807007
    },
    {
      "epoch": 4.175090789794922e-05,
      "model_forward_time": 2.3104703426361084,
      "step": 43779
    },
    {
      "epoch": 4.175090789794922e-05,
      "step": 43779,
      "training_step_time": 7.625844955444336
    },
    {
      "epoch": 4.1751861572265626e-05,
      "grad_norm": 0.1420973837375641,
      "learning_rate": 6.422084746234605e-06,
      "loss": 0.0633,
      "step": 43780
    },
    {
      "epoch": 4.1751861572265626e-05,
      "model_forward_time": 1.8036508560180664,
      "step": 43780
    },
    {
      "dynamics_loss_avg": 0.061576754599809644,
      "epoch": 4.1751861572265626e-05,
      "step": 43780
    },
    {
      "action_loss_avg": 0.002088232370442711,
      "epoch": 4.1751861572265626e-05,
      "step": 43780
    },
    {
      "epoch": 4.1751861572265626e-05,
      "step": 43780,
      "training_step_time": 8.749552011489868
    },
    {
      "epoch": 4.175281524658203e-05,
      "model_forward_time": 1.8053066730499268,
      "step": 43781
    },
    {
      "epoch": 4.175281524658203e-05,
      "step": 43781,
      "training_step_time": 7.484155893325806
    },
    {
      "epoch": 4.1753768920898434e-05,
      "model_forward_time": 1.806685209274292,
      "step": 43782
    },
    {
      "epoch": 4.1753768920898434e-05,
      "step": 43782,
      "training_step_time": 6.9132397174835205
    },
    {
      "epoch": 4.1754722595214845e-05,
      "model_forward_time": 1.8031446933746338,
      "step": 43783
    },
    {
      "epoch": 4.1754722595214845e-05,
      "step": 43783,
      "training_step_time": 6.847876787185669
    },
    {
      "epoch": 4.175567626953125e-05,
      "model_forward_time": 1.8045470714569092,
      "step": 43784
    },
    {
      "epoch": 4.175567626953125e-05,
      "step": 43784,
      "training_step_time": 6.865098237991333
    },
    {
      "epoch": 4.175662994384765e-05,
      "model_forward_time": 1.8060846328735352,
      "step": 43785
    },
    {
      "epoch": 4.175662994384765e-05,
      "step": 43785,
      "training_step_time": 6.834431886672974
    },
    {
      "epoch": 4.1757583618164064e-05,
      "model_forward_time": 1.8025777339935303,
      "step": 43786
    },
    {
      "epoch": 4.1757583618164064e-05,
      "step": 43786,
      "training_step_time": 6.818074464797974
    },
    {
      "epoch": 4.175853729248047e-05,
      "model_forward_time": 1.8079307079315186,
      "step": 43787
    },
    {
      "epoch": 4.175853729248047e-05,
      "step": 43787,
      "training_step_time": 7.589786767959595
    },
    {
      "epoch": 4.175949096679687e-05,
      "model_forward_time": 2.306645393371582,
      "step": 43788
    },
    {
      "epoch": 4.175949096679687e-05,
      "step": 43788,
      "training_step_time": 7.573951959609985
    },
    {
      "epoch": 4.176044464111328e-05,
      "model_forward_time": 1.8110637664794922,
      "step": 43789
    },
    {
      "epoch": 4.176044464111328e-05,
      "step": 43789,
      "training_step_time": 7.820672273635864
    },
    {
      "epoch": 4.176139831542969e-05,
      "grad_norm": 0.15837207436561584,
      "learning_rate": 6.420499485861965e-06,
      "loss": 0.0588,
      "step": 43790
    },
    {
      "epoch": 4.176139831542969e-05,
      "model_forward_time": 1.8111140727996826,
      "step": 43790
    },
    {
      "dynamics_loss_avg": 0.05680384337902069,
      "epoch": 4.176139831542969e-05,
      "step": 43790
    },
    {
      "action_loss_avg": 0.0012443847372196615,
      "epoch": 4.176139831542969e-05,
      "step": 43790
    },
    {
      "epoch": 4.176139831542969e-05,
      "step": 43790,
      "training_step_time": 7.963557720184326
    },
    {
      "epoch": 4.176235198974609e-05,
      "model_forward_time": 1.7903003692626953,
      "step": 43791
    },
    {
      "epoch": 4.176235198974609e-05,
      "step": 43791,
      "training_step_time": 6.895519018173218
    },
    {
      "epoch": 4.17633056640625e-05,
      "model_forward_time": 1.8173835277557373,
      "step": 43792
    },
    {
      "epoch": 4.17633056640625e-05,
      "step": 43792,
      "training_step_time": 6.881873369216919
    },
    {
      "epoch": 4.1764259338378906e-05,
      "model_forward_time": 1.8057379722595215,
      "step": 43793
    },
    {
      "epoch": 4.1764259338378906e-05,
      "step": 43793,
      "training_step_time": 6.922899484634399
    },
    {
      "epoch": 4.176521301269531e-05,
      "model_forward_time": 1.80393648147583,
      "step": 43794
    },
    {
      "epoch": 4.176521301269531e-05,
      "step": 43794,
      "training_step_time": 6.710797309875488
    },
    {
      "epoch": 4.176616668701172e-05,
      "model_forward_time": 1.810347080230713,
      "step": 43795
    },
    {
      "epoch": 4.176616668701172e-05,
      "step": 43795,
      "training_step_time": 6.860036134719849
    },
    {
      "epoch": 4.1767120361328124e-05,
      "model_forward_time": 1.8005869388580322,
      "step": 43796
    },
    {
      "epoch": 4.1767120361328124e-05,
      "step": 43796,
      "training_step_time": 7.133787155151367
    },
    {
      "epoch": 4.176807403564453e-05,
      "model_forward_time": 2.3747687339782715,
      "step": 43797
    },
    {
      "epoch": 4.176807403564453e-05,
      "step": 43797,
      "training_step_time": 7.7306365966796875
    },
    {
      "epoch": 4.176902770996094e-05,
      "model_forward_time": 1.8117928504943848,
      "step": 43798
    },
    {
      "epoch": 4.176902770996094e-05,
      "step": 43798,
      "training_step_time": 8.637272834777832
    },
    {
      "epoch": 4.176998138427734e-05,
      "model_forward_time": 1.8070604801177979,
      "step": 43799
    },
    {
      "epoch": 4.176998138427734e-05,
      "step": 43799,
      "training_step_time": 7.962717294692993
    },
    {
      "epoch": 4.177093505859375e-05,
      "grad_norm": 0.10511837899684906,
      "learning_rate": 6.418914070145636e-06,
      "loss": 0.0614,
      "step": 43800
    },
    {
      "epoch": 4.177093505859375e-05,
      "model_forward_time": 1.813706874847412,
      "step": 43800
    },
    {
      "dynamics_loss_avg": 0.05241143284365535,
      "epoch": 4.177093505859375e-05,
      "step": 43800
    },
    {
      "action_loss_avg": 0.0013910571695305407,
      "epoch": 4.177093505859375e-05,
      "step": 43800
    },
    {
      "epoch": 4.177093505859375e-05,
      "step": 43800,
      "training_step_time": 6.862690210342407
    },
    {
      "epoch": 4.177188873291016e-05,
      "model_forward_time": 1.8033523559570312,
      "step": 43801
    },
    {
      "epoch": 4.177188873291016e-05,
      "step": 43801,
      "training_step_time": 6.906373500823975
    },
    {
      "epoch": 4.177284240722656e-05,
      "model_forward_time": 1.8041999340057373,
      "step": 43802
    },
    {
      "epoch": 4.177284240722656e-05,
      "step": 43802,
      "training_step_time": 6.938002347946167
    },
    {
      "epoch": 4.1773796081542966e-05,
      "model_forward_time": 1.8072466850280762,
      "step": 43803
    },
    {
      "epoch": 4.1773796081542966e-05,
      "step": 43803,
      "training_step_time": 6.828562259674072
    },
    {
      "epoch": 4.177474975585938e-05,
      "model_forward_time": 1.8171894550323486,
      "step": 43804
    },
    {
      "epoch": 4.177474975585938e-05,
      "step": 43804,
      "training_step_time": 6.864853620529175
    },
    {
      "epoch": 4.177570343017578e-05,
      "model_forward_time": 1.8116228580474854,
      "step": 43805
    },
    {
      "epoch": 4.177570343017578e-05,
      "step": 43805,
      "training_step_time": 7.147819519042969
    },
    {
      "epoch": 4.1776657104492185e-05,
      "model_forward_time": 2.357954263687134,
      "step": 43806
    },
    {
      "epoch": 4.1776657104492185e-05,
      "step": 43806,
      "training_step_time": 7.701575994491577
    },
    {
      "epoch": 4.1777610778808596e-05,
      "model_forward_time": 1.8063311576843262,
      "step": 43807
    },
    {
      "epoch": 4.1777610778808596e-05,
      "step": 43807,
      "training_step_time": 10.029058694839478
    },
    {
      "epoch": 4.1778564453125e-05,
      "model_forward_time": 1.8035478591918945,
      "step": 43808
    },
    {
      "epoch": 4.1778564453125e-05,
      "step": 43808,
      "training_step_time": 8.531453132629395
    },
    {
      "epoch": 4.1779518127441404e-05,
      "model_forward_time": 1.808892011642456,
      "step": 43809
    },
    {
      "epoch": 4.1779518127441404e-05,
      "step": 43809,
      "training_step_time": 6.859572649002075
    },
    {
      "epoch": 4.1780471801757815e-05,
      "grad_norm": 0.08870341628789902,
      "learning_rate": 6.417328499258997e-06,
      "loss": 0.0589,
      "step": 43810
    },
    {
      "epoch": 4.1780471801757815e-05,
      "model_forward_time": 1.8072378635406494,
      "step": 43810
    },
    {
      "dynamics_loss_avg": 0.05694365091621876,
      "epoch": 4.1780471801757815e-05,
      "step": 43810
    },
    {
      "action_loss_avg": 0.0018860171170672403,
      "epoch": 4.1780471801757815e-05,
      "step": 43810
    },
    {
      "epoch": 4.1780471801757815e-05,
      "step": 43810,
      "training_step_time": 6.848456621170044
    },
    {
      "epoch": 4.178142547607422e-05,
      "model_forward_time": 1.7911098003387451,
      "step": 43811
    },
    {
      "epoch": 4.178142547607422e-05,
      "step": 43811,
      "training_step_time": 6.8317437171936035
    },
    {
      "epoch": 4.178237915039062e-05,
      "model_forward_time": 1.8066656589508057,
      "step": 43812
    },
    {
      "epoch": 4.178237915039062e-05,
      "step": 43812,
      "training_step_time": 6.8934409618377686
    },
    {
      "epoch": 4.1783332824707034e-05,
      "model_forward_time": 1.804370641708374,
      "step": 43813
    },
    {
      "epoch": 4.1783332824707034e-05,
      "step": 43813,
      "training_step_time": 6.82781982421875
    },
    {
      "epoch": 4.178428649902344e-05,
      "model_forward_time": 1.806406021118164,
      "step": 43814
    },
    {
      "epoch": 4.178428649902344e-05,
      "step": 43814,
      "training_step_time": 7.116311550140381
    },
    {
      "epoch": 4.178524017333984e-05,
      "model_forward_time": 2.358222007751465,
      "step": 43815
    },
    {
      "epoch": 4.178524017333984e-05,
      "step": 43815,
      "training_step_time": 7.654983043670654
    },
    {
      "epoch": 4.178619384765625e-05,
      "model_forward_time": 1.8078601360321045,
      "step": 43816
    },
    {
      "epoch": 4.178619384765625e-05,
      "step": 43816,
      "training_step_time": 9.516860008239746
    },
    {
      "epoch": 4.1787147521972656e-05,
      "model_forward_time": 1.811410665512085,
      "step": 43817
    },
    {
      "epoch": 4.1787147521972656e-05,
      "step": 43817,
      "training_step_time": 8.388346195220947
    },
    {
      "epoch": 4.178810119628906e-05,
      "model_forward_time": 1.8042190074920654,
      "step": 43818
    },
    {
      "epoch": 4.178810119628906e-05,
      "step": 43818,
      "training_step_time": 7.493051767349243
    },
    {
      "epoch": 4.178905487060547e-05,
      "model_forward_time": 1.7928926944732666,
      "step": 43819
    },
    {
      "epoch": 4.178905487060547e-05,
      "step": 43819,
      "training_step_time": 6.799334287643433
    },
    {
      "epoch": 4.1790008544921875e-05,
      "grad_norm": 0.13043944537639618,
      "learning_rate": 6.415742773375446e-06,
      "loss": 0.0576,
      "step": 43820
    },
    {
      "epoch": 4.1790008544921875e-05,
      "model_forward_time": 1.8141140937805176,
      "step": 43820
    },
    {
      "dynamics_loss_avg": 0.0727249862626195,
      "epoch": 4.1790008544921875e-05,
      "step": 43820
    },
    {
      "action_loss_avg": 0.0015525330323725938,
      "epoch": 4.1790008544921875e-05,
      "step": 43820
    },
    {
      "epoch": 4.1790008544921875e-05,
      "step": 43820,
      "training_step_time": 6.799799203872681
    },
    {
      "epoch": 4.179096221923828e-05,
      "model_forward_time": 1.8077187538146973,
      "step": 43821
    },
    {
      "epoch": 4.179096221923828e-05,
      "step": 43821,
      "training_step_time": 6.839324474334717
    },
    {
      "epoch": 4.179191589355469e-05,
      "model_forward_time": 1.8113453388214111,
      "step": 43822
    },
    {
      "epoch": 4.179191589355469e-05,
      "step": 43822,
      "training_step_time": 6.81795597076416
    },
    {
      "epoch": 4.1792869567871094e-05,
      "model_forward_time": 1.811729907989502,
      "step": 43823
    },
    {
      "epoch": 4.1792869567871094e-05,
      "step": 43823,
      "training_step_time": 7.118092775344849
    },
    {
      "epoch": 4.17938232421875e-05,
      "model_forward_time": 1.8181300163269043,
      "step": 43824
    },
    {
      "epoch": 4.17938232421875e-05,
      "step": 43824,
      "training_step_time": 8.082399368286133
    },
    {
      "epoch": 4.179477691650391e-05,
      "model_forward_time": 1.7907514572143555,
      "step": 43825
    },
    {
      "epoch": 4.179477691650391e-05,
      "step": 43825,
      "training_step_time": 7.715940475463867
    },
    {
      "epoch": 4.179573059082031e-05,
      "model_forward_time": 1.8093929290771484,
      "step": 43826
    },
    {
      "epoch": 4.179573059082031e-05,
      "step": 43826,
      "training_step_time": 8.890905618667603
    },
    {
      "epoch": 4.179668426513672e-05,
      "model_forward_time": 1.8034467697143555,
      "step": 43827
    },
    {
      "epoch": 4.179668426513672e-05,
      "step": 43827,
      "training_step_time": 8.629882335662842
    },
    {
      "epoch": 4.179763793945313e-05,
      "model_forward_time": 1.8098456859588623,
      "step": 43828
    },
    {
      "epoch": 4.179763793945313e-05,
      "step": 43828,
      "training_step_time": 7.673370361328125
    },
    {
      "epoch": 4.179859161376953e-05,
      "model_forward_time": 1.8107490539550781,
      "step": 43829
    },
    {
      "epoch": 4.179859161376953e-05,
      "step": 43829,
      "training_step_time": 6.928462982177734
    },
    {
      "epoch": 4.1799545288085936e-05,
      "grad_norm": 0.12015816569328308,
      "learning_rate": 6.414156892668393e-06,
      "loss": 0.0568,
      "step": 43830
    },
    {
      "epoch": 4.1799545288085936e-05,
      "model_forward_time": 1.8145601749420166,
      "step": 43830
    },
    {
      "dynamics_loss_avg": 0.061509347148239614,
      "epoch": 4.1799545288085936e-05,
      "step": 43830
    },
    {
      "action_loss_avg": 0.0018138131476007403,
      "epoch": 4.1799545288085936e-05,
      "step": 43830
    },
    {
      "epoch": 4.1799545288085936e-05,
      "step": 43830,
      "training_step_time": 6.796626091003418
    },
    {
      "epoch": 4.180049896240235e-05,
      "model_forward_time": 1.809734582901001,
      "step": 43831
    },
    {
      "epoch": 4.180049896240235e-05,
      "step": 43831,
      "training_step_time": 6.862001180648804
    },
    {
      "epoch": 4.180145263671875e-05,
      "model_forward_time": 1.787470817565918,
      "step": 43832
    },
    {
      "epoch": 4.180145263671875e-05,
      "step": 43832,
      "training_step_time": 7.003197908401489
    },
    {
      "epoch": 4.1802406311035155e-05,
      "model_forward_time": 1.8056368827819824,
      "step": 43833
    },
    {
      "epoch": 4.1802406311035155e-05,
      "step": 43833,
      "training_step_time": 7.345754384994507
    },
    {
      "epoch": 4.1803359985351566e-05,
      "model_forward_time": 2.344093084335327,
      "step": 43834
    },
    {
      "epoch": 4.1803359985351566e-05,
      "step": 43834,
      "training_step_time": 10.217434883117676
    },
    {
      "epoch": 4.180431365966797e-05,
      "model_forward_time": 1.8113789558410645,
      "step": 43835
    },
    {
      "epoch": 4.180431365966797e-05,
      "step": 43835,
      "training_step_time": 8.875749349594116
    },
    {
      "epoch": 4.1805267333984374e-05,
      "model_forward_time": 1.8179118633270264,
      "step": 43836
    },
    {
      "epoch": 4.1805267333984374e-05,
      "step": 43836,
      "training_step_time": 7.780667543411255
    },
    {
      "epoch": 4.1806221008300784e-05,
      "model_forward_time": 1.8124616146087646,
      "step": 43837
    },
    {
      "epoch": 4.1806221008300784e-05,
      "step": 43837,
      "training_step_time": 7.703255891799927
    },
    {
      "epoch": 4.180717468261719e-05,
      "model_forward_time": 1.799989938735962,
      "step": 43838
    },
    {
      "epoch": 4.180717468261719e-05,
      "step": 43838,
      "training_step_time": 6.851207256317139
    },
    {
      "epoch": 4.180812835693359e-05,
      "model_forward_time": 1.8122923374176025,
      "step": 43839
    },
    {
      "epoch": 4.180812835693359e-05,
      "step": 43839,
      "training_step_time": 6.837286472320557
    },
    {
      "epoch": 4.180908203125e-05,
      "grad_norm": 0.11290530115365982,
      "learning_rate": 6.4125708573112675e-06,
      "loss": 0.0549,
      "step": 43840
    },
    {
      "epoch": 4.180908203125e-05,
      "model_forward_time": 1.8056893348693848,
      "step": 43840
    },
    {
      "dynamics_loss_avg": 0.051265642419457434,
      "epoch": 4.180908203125e-05,
      "step": 43840
    },
    {
      "action_loss_avg": 0.0011175541236298158,
      "epoch": 4.180908203125e-05,
      "step": 43840
    },
    {
      "epoch": 4.180908203125e-05,
      "step": 43840,
      "training_step_time": 6.808078765869141
    },
    {
      "epoch": 4.181003570556641e-05,
      "model_forward_time": 1.8097960948944092,
      "step": 43841
    },
    {
      "epoch": 4.181003570556641e-05,
      "step": 43841,
      "training_step_time": 6.861681699752808
    },
    {
      "epoch": 4.181098937988281e-05,
      "model_forward_time": 1.7998290061950684,
      "step": 43842
    },
    {
      "epoch": 4.181098937988281e-05,
      "step": 43842,
      "training_step_time": 7.074105739593506
    },
    {
      "epoch": 4.181194305419922e-05,
      "model_forward_time": 2.342345952987671,
      "step": 43843
    },
    {
      "epoch": 4.181194305419922e-05,
      "step": 43843,
      "training_step_time": 8.666586875915527
    },
    {
      "epoch": 4.1812896728515626e-05,
      "model_forward_time": 1.8086857795715332,
      "step": 43844
    },
    {
      "epoch": 4.1812896728515626e-05,
      "step": 43844,
      "training_step_time": 11.626667976379395
    },
    {
      "epoch": 4.181385040283203e-05,
      "model_forward_time": 1.8109102249145508,
      "step": 43845
    },
    {
      "epoch": 4.181385040283203e-05,
      "step": 43845,
      "training_step_time": 9.052009582519531
    },
    {
      "epoch": 4.1814804077148434e-05,
      "model_forward_time": 1.8071250915527344,
      "step": 43846
    },
    {
      "epoch": 4.1814804077148434e-05,
      "step": 43846,
      "training_step_time": 7.642759799957275
    },
    {
      "epoch": 4.1815757751464845e-05,
      "model_forward_time": 1.8229780197143555,
      "step": 43847
    },
    {
      "epoch": 4.1815757751464845e-05,
      "step": 43847,
      "training_step_time": 6.797907114028931
    },
    {
      "epoch": 4.181671142578125e-05,
      "model_forward_time": 1.8122129440307617,
      "step": 43848
    },
    {
      "epoch": 4.181671142578125e-05,
      "step": 43848,
      "training_step_time": 6.871882677078247
    },
    {
      "epoch": 4.181766510009765e-05,
      "model_forward_time": 1.811347246170044,
      "step": 43849
    },
    {
      "epoch": 4.181766510009765e-05,
      "step": 43849,
      "training_step_time": 6.885993003845215
    },
    {
      "epoch": 4.1818618774414064e-05,
      "grad_norm": 0.11624489724636078,
      "learning_rate": 6.410984667477518e-06,
      "loss": 0.0581,
      "step": 43850
    },
    {
      "epoch": 4.1818618774414064e-05,
      "model_forward_time": 1.8045637607574463,
      "step": 43850
    },
    {
      "dynamics_loss_avg": 0.06850853133946658,
      "epoch": 4.1818618774414064e-05,
      "step": 43850
    },
    {
      "action_loss_avg": 0.001781109147123061,
      "epoch": 4.1818618774414064e-05,
      "step": 43850
    },
    {
      "epoch": 4.1818618774414064e-05,
      "step": 43850,
      "training_step_time": 6.862454175949097
    },
    {
      "epoch": 4.181957244873047e-05,
      "model_forward_time": 1.8058414459228516,
      "step": 43851
    },
    {
      "epoch": 4.181957244873047e-05,
      "step": 43851,
      "training_step_time": 6.995962858200073
    },
    {
      "epoch": 4.182052612304687e-05,
      "model_forward_time": 2.348374605178833,
      "step": 43852
    },
    {
      "epoch": 4.182052612304687e-05,
      "step": 43852,
      "training_step_time": 7.714689493179321
    },
    {
      "epoch": 4.182147979736328e-05,
      "model_forward_time": 1.8093068599700928,
      "step": 43853
    },
    {
      "epoch": 4.182147979736328e-05,
      "step": 43853,
      "training_step_time": 10.635087013244629
    },
    {
      "epoch": 4.182243347167969e-05,
      "model_forward_time": 1.8157637119293213,
      "step": 43854
    },
    {
      "epoch": 4.182243347167969e-05,
      "step": 43854,
      "training_step_time": 8.022550106048584
    },
    {
      "epoch": 4.182338714599609e-05,
      "model_forward_time": 1.8059723377227783,
      "step": 43855
    },
    {
      "epoch": 4.182338714599609e-05,
      "step": 43855,
      "training_step_time": 8.575331687927246
    },
    {
      "epoch": 4.18243408203125e-05,
      "model_forward_time": 1.8105313777923584,
      "step": 43856
    },
    {
      "epoch": 4.18243408203125e-05,
      "step": 43856,
      "training_step_time": 6.8709187507629395
    },
    {
      "epoch": 4.1825294494628906e-05,
      "model_forward_time": 1.8161699771881104,
      "step": 43857
    },
    {
      "epoch": 4.1825294494628906e-05,
      "step": 43857,
      "training_step_time": 6.8006510734558105
    },
    {
      "epoch": 4.182624816894531e-05,
      "model_forward_time": 1.8027269840240479,
      "step": 43858
    },
    {
      "epoch": 4.182624816894531e-05,
      "step": 43858,
      "training_step_time": 6.855792760848999
    },
    {
      "epoch": 4.182720184326172e-05,
      "model_forward_time": 1.815464735031128,
      "step": 43859
    },
    {
      "epoch": 4.182720184326172e-05,
      "step": 43859,
      "training_step_time": 6.839871883392334
    },
    {
      "epoch": 4.1828155517578124e-05,
      "grad_norm": 0.14244019985198975,
      "learning_rate": 6.409398323340606e-06,
      "loss": 0.0558,
      "step": 43860
    },
    {
      "epoch": 4.1828155517578124e-05,
      "model_forward_time": 1.8105370998382568,
      "step": 43860
    },
    {
      "dynamics_loss_avg": 0.05941489618271589,
      "epoch": 4.1828155517578124e-05,
      "step": 43860
    },
    {
      "action_loss_avg": 0.001996123747085221,
      "epoch": 4.1828155517578124e-05,
      "step": 43860
    },
    {
      "epoch": 4.1828155517578124e-05,
      "step": 43860,
      "training_step_time": 7.1452319622039795
    },
    {
      "epoch": 4.182910919189453e-05,
      "model_forward_time": 2.347440242767334,
      "step": 43861
    },
    {
      "epoch": 4.182910919189453e-05,
      "step": 43861,
      "training_step_time": 7.742149829864502
    },
    {
      "epoch": 4.183006286621094e-05,
      "model_forward_time": 1.8157968521118164,
      "step": 43862
    },
    {
      "epoch": 4.183006286621094e-05,
      "step": 43862,
      "training_step_time": 9.127593755722046
    },
    {
      "epoch": 4.183101654052734e-05,
      "model_forward_time": 1.807809591293335,
      "step": 43863
    },
    {
      "epoch": 4.183101654052734e-05,
      "step": 43863,
      "training_step_time": 8.662938356399536
    },
    {
      "epoch": 4.183197021484375e-05,
      "model_forward_time": 1.8110740184783936,
      "step": 43864
    },
    {
      "epoch": 4.183197021484375e-05,
      "step": 43864,
      "training_step_time": 8.299709558486938
    },
    {
      "epoch": 4.183292388916016e-05,
      "model_forward_time": 1.8164770603179932,
      "step": 43865
    },
    {
      "epoch": 4.183292388916016e-05,
      "step": 43865,
      "training_step_time": 7.471266269683838
    },
    {
      "epoch": 4.183387756347656e-05,
      "model_forward_time": 1.8162944316864014,
      "step": 43866
    },
    {
      "epoch": 4.183387756347656e-05,
      "step": 43866,
      "training_step_time": 6.813345193862915
    },
    {
      "epoch": 4.1834831237792966e-05,
      "model_forward_time": 1.8173961639404297,
      "step": 43867
    },
    {
      "epoch": 4.1834831237792966e-05,
      "step": 43867,
      "training_step_time": 6.8170928955078125
    },
    {
      "epoch": 4.183578491210938e-05,
      "model_forward_time": 1.815356969833374,
      "step": 43868
    },
    {
      "epoch": 4.183578491210938e-05,
      "step": 43868,
      "training_step_time": 6.790725469589233
    },
    {
      "epoch": 4.183673858642578e-05,
      "model_forward_time": 1.8140716552734375,
      "step": 43869
    },
    {
      "epoch": 4.183673858642578e-05,
      "step": 43869,
      "training_step_time": 7.344938039779663
    },
    {
      "epoch": 4.1837692260742185e-05,
      "grad_norm": 0.15240754187107086,
      "learning_rate": 6.407811825074013e-06,
      "loss": 0.0574,
      "step": 43870
    },
    {
      "epoch": 4.1837692260742185e-05,
      "model_forward_time": 2.3515207767486572,
      "step": 43870
    },
    {
      "dynamics_loss_avg": 0.06266644112765789,
      "epoch": 4.1837692260742185e-05,
      "step": 43870
    },
    {
      "action_loss_avg": 0.0015226105868350714,
      "epoch": 4.1837692260742185e-05,
      "step": 43870
    },
    {
      "epoch": 4.1837692260742185e-05,
      "step": 43870,
      "training_step_time": 7.744389295578003
    },
    {
      "epoch": 4.1838645935058596e-05,
      "model_forward_time": 1.8267755508422852,
      "step": 43871
    },
    {
      "epoch": 4.1838645935058596e-05,
      "step": 43871,
      "training_step_time": 9.090509176254272
    },
    {
      "epoch": 4.1839599609375e-05,
      "model_forward_time": 1.8071885108947754,
      "step": 43872
    },
    {
      "epoch": 4.1839599609375e-05,
      "step": 43872,
      "training_step_time": 8.774237155914307
    },
    {
      "epoch": 4.1840553283691404e-05,
      "model_forward_time": 1.812230110168457,
      "step": 43873
    },
    {
      "epoch": 4.1840553283691404e-05,
      "step": 43873,
      "training_step_time": 8.324468851089478
    },
    {
      "epoch": 4.1841506958007815e-05,
      "model_forward_time": 1.8080358505249023,
      "step": 43874
    },
    {
      "epoch": 4.1841506958007815e-05,
      "step": 43874,
      "training_step_time": 7.644745826721191
    },
    {
      "epoch": 4.184246063232422e-05,
      "model_forward_time": 1.8111765384674072,
      "step": 43875
    },
    {
      "epoch": 4.184246063232422e-05,
      "step": 43875,
      "training_step_time": 6.810934066772461
    },
    {
      "epoch": 4.184341430664062e-05,
      "model_forward_time": 1.8188903331756592,
      "step": 43876
    },
    {
      "epoch": 4.184341430664062e-05,
      "step": 43876,
      "training_step_time": 6.802219867706299
    },
    {
      "epoch": 4.1844367980957034e-05,
      "model_forward_time": 1.8108015060424805,
      "step": 43877
    },
    {
      "epoch": 4.1844367980957034e-05,
      "step": 43877,
      "training_step_time": 6.7777931690216064
    },
    {
      "epoch": 4.184532165527344e-05,
      "model_forward_time": 1.8151524066925049,
      "step": 43878
    },
    {
      "epoch": 4.184532165527344e-05,
      "step": 43878,
      "training_step_time": 6.826600074768066
    },
    {
      "epoch": 4.184627532958984e-05,
      "model_forward_time": 2.366098642349243,
      "step": 43879
    },
    {
      "epoch": 4.184627532958984e-05,
      "step": 43879,
      "training_step_time": 7.705825090408325
    },
    {
      "epoch": 4.184722900390625e-05,
      "grad_norm": 0.1464948207139969,
      "learning_rate": 6.406225172851235e-06,
      "loss": 0.0544,
      "step": 43880
    },
    {
      "epoch": 4.184722900390625e-05,
      "model_forward_time": 1.8258872032165527,
      "step": 43880
    },
    {
      "dynamics_loss_avg": 0.057011676207184794,
      "epoch": 4.184722900390625e-05,
      "step": 43880
    },
    {
      "action_loss_avg": 0.0013088860054267571,
      "epoch": 4.184722900390625e-05,
      "step": 43880
    },
    {
      "epoch": 4.184722900390625e-05,
      "step": 43880,
      "training_step_time": 8.72936224937439
    },
    {
      "epoch": 4.1848182678222656e-05,
      "model_forward_time": 1.817152976989746,
      "step": 43881
    },
    {
      "epoch": 4.1848182678222656e-05,
      "step": 43881,
      "training_step_time": 7.968273639678955
    },
    {
      "epoch": 4.184913635253906e-05,
      "model_forward_time": 1.8218340873718262,
      "step": 43882
    },
    {
      "epoch": 4.184913635253906e-05,
      "step": 43882,
      "training_step_time": 7.744608163833618
    },
    {
      "epoch": 4.185009002685547e-05,
      "model_forward_time": 1.823781967163086,
      "step": 43883
    },
    {
      "epoch": 4.185009002685547e-05,
      "step": 43883,
      "training_step_time": 7.6062586307525635
    },
    {
      "epoch": 4.1851043701171875e-05,
      "model_forward_time": 1.811286449432373,
      "step": 43884
    },
    {
      "epoch": 4.1851043701171875e-05,
      "step": 43884,
      "training_step_time": 6.783048391342163
    },
    {
      "epoch": 4.185199737548828e-05,
      "model_forward_time": 1.8243088722229004,
      "step": 43885
    },
    {
      "epoch": 4.185199737548828e-05,
      "step": 43885,
      "training_step_time": 6.823519229888916
    },
    {
      "epoch": 4.185295104980469e-05,
      "model_forward_time": 1.8261997699737549,
      "step": 43886
    },
    {
      "epoch": 4.185295104980469e-05,
      "step": 43886,
      "training_step_time": 6.779616594314575
    },
    {
      "epoch": 4.1853904724121094e-05,
      "model_forward_time": 1.8296267986297607,
      "step": 43887
    },
    {
      "epoch": 4.1853904724121094e-05,
      "step": 43887,
      "training_step_time": 6.856780529022217
    },
    {
      "epoch": 4.18548583984375e-05,
      "model_forward_time": 2.3636465072631836,
      "step": 43888
    },
    {
      "epoch": 4.18548583984375e-05,
      "step": 43888,
      "training_step_time": 7.738618612289429
    },
    {
      "epoch": 4.185581207275391e-05,
      "model_forward_time": 1.8292124271392822,
      "step": 43889
    },
    {
      "epoch": 4.185581207275391e-05,
      "step": 43889,
      "training_step_time": 8.493465423583984
    },
    {
      "epoch": 4.185676574707031e-05,
      "grad_norm": 0.13769477605819702,
      "learning_rate": 6.404638366845785e-06,
      "loss": 0.0582,
      "step": 43890
    },
    {
      "epoch": 4.185676574707031e-05,
      "model_forward_time": 1.8154280185699463,
      "step": 43890
    },
    {
      "dynamics_loss_avg": 0.05866391807794571,
      "epoch": 4.185676574707031e-05,
      "step": 43890
    },
    {
      "action_loss_avg": 0.0015190939651802183,
      "epoch": 4.185676574707031e-05,
      "step": 43890
    },
    {
      "epoch": 4.185676574707031e-05,
      "step": 43890,
      "training_step_time": 9.104426622390747
    },
    {
      "epoch": 4.185771942138672e-05,
      "model_forward_time": 1.8285386562347412,
      "step": 43891
    },
    {
      "epoch": 4.185771942138672e-05,
      "step": 43891,
      "training_step_time": 7.730693817138672
    },
    {
      "epoch": 4.185867309570313e-05,
      "model_forward_time": 1.8255829811096191,
      "step": 43892
    },
    {
      "epoch": 4.185867309570313e-05,
      "step": 43892,
      "training_step_time": 7.606312274932861
    },
    {
      "epoch": 4.185962677001953e-05,
      "model_forward_time": 1.822890043258667,
      "step": 43893
    },
    {
      "epoch": 4.185962677001953e-05,
      "step": 43893,
      "training_step_time": 7.468700408935547
    },
    {
      "epoch": 4.1860580444335936e-05,
      "model_forward_time": 1.8235406875610352,
      "step": 43894
    },
    {
      "epoch": 4.1860580444335936e-05,
      "step": 43894,
      "training_step_time": 6.8838841915130615
    },
    {
      "epoch": 4.186153411865235e-05,
      "model_forward_time": 1.8208565711975098,
      "step": 43895
    },
    {
      "epoch": 4.186153411865235e-05,
      "step": 43895,
      "training_step_time": 6.810498237609863
    },
    {
      "epoch": 4.186248779296875e-05,
      "model_forward_time": 1.816667079925537,
      "step": 43896
    },
    {
      "epoch": 4.186248779296875e-05,
      "step": 43896,
      "training_step_time": 6.827319145202637
    },
    {
      "epoch": 4.1863441467285155e-05,
      "model_forward_time": 2.361272096633911,
      "step": 43897
    },
    {
      "epoch": 4.1863441467285155e-05,
      "step": 43897,
      "training_step_time": 7.6663734912872314
    },
    {
      "epoch": 4.1864395141601566e-05,
      "model_forward_time": 1.8278303146362305,
      "step": 43898
    },
    {
      "epoch": 4.1864395141601566e-05,
      "step": 43898,
      "training_step_time": 8.737308502197266
    },
    {
      "epoch": 4.186534881591797e-05,
      "model_forward_time": 1.7907867431640625,
      "step": 43899
    },
    {
      "epoch": 4.186534881591797e-05,
      "step": 43899,
      "training_step_time": 10.086423873901367
    },
    {
      "epoch": 4.1866302490234374e-05,
      "grad_norm": 0.14482171833515167,
      "learning_rate": 6.403051407231197e-06,
      "loss": 0.058,
      "step": 43900
    },
    {
      "epoch": 4.1866302490234374e-05,
      "model_forward_time": 1.8207240104675293,
      "step": 43900
    },
    {
      "dynamics_loss_avg": 0.0659878296777606,
      "epoch": 4.1866302490234374e-05,
      "step": 43900
    },
    {
      "action_loss_avg": 0.00241135231917724,
      "epoch": 4.1866302490234374e-05,
      "step": 43900
    },
    {
      "epoch": 4.1866302490234374e-05,
      "step": 43900,
      "training_step_time": 8.409297943115234
    },
    {
      "epoch": 4.1867256164550784e-05,
      "model_forward_time": 1.453681230545044,
      "step": 43901
    },
    {
      "epoch": 4.1867256164550784e-05,
      "step": 43901,
      "training_step_time": 7.728499889373779
    },
    {
      "epoch": 4.186820983886719e-05,
      "model_forward_time": 1.451014518737793,
      "step": 43902
    },
    {
      "epoch": 4.186820983886719e-05,
      "step": 43902,
      "training_step_time": 7.59510064125061
    },
    {
      "epoch": 4.186916351318359e-05,
      "model_forward_time": 1.8210668563842773,
      "step": 43903
    },
    {
      "epoch": 4.186916351318359e-05,
      "step": 43903,
      "training_step_time": 6.822998523712158
    },
    {
      "epoch": 4.18701171875e-05,
      "model_forward_time": 1.8152577877044678,
      "step": 43904
    },
    {
      "epoch": 4.18701171875e-05,
      "step": 43904,
      "training_step_time": 6.907845735549927
    },
    {
      "epoch": 4.187107086181641e-05,
      "model_forward_time": 1.8300151824951172,
      "step": 43905
    },
    {
      "epoch": 4.187107086181641e-05,
      "step": 43905,
      "training_step_time": 6.85321044921875
    },
    {
      "epoch": 4.187202453613281e-05,
      "model_forward_time": 2.3481733798980713,
      "step": 43906
    },
    {
      "epoch": 4.187202453613281e-05,
      "step": 43906,
      "training_step_time": 7.1559741497039795
    },
    {
      "epoch": 4.1872978210449215e-05,
      "model_forward_time": 1.8252146244049072,
      "step": 43907
    },
    {
      "epoch": 4.1872978210449215e-05,
      "step": 43907,
      "training_step_time": 8.306271314620972
    },
    {
      "epoch": 4.1873931884765626e-05,
      "model_forward_time": 1.8105933666229248,
      "step": 43908
    },
    {
      "epoch": 4.1873931884765626e-05,
      "step": 43908,
      "training_step_time": 10.43822431564331
    },
    {
      "epoch": 4.187488555908203e-05,
      "model_forward_time": 1.8134651184082031,
      "step": 43909
    },
    {
      "epoch": 4.187488555908203e-05,
      "step": 43909,
      "training_step_time": 7.766231298446655
    },
    {
      "epoch": 4.1875839233398434e-05,
      "grad_norm": 0.12174984812736511,
      "learning_rate": 6.401464294181016e-06,
      "loss": 0.0573,
      "step": 43910
    },
    {
      "epoch": 4.1875839233398434e-05,
      "model_forward_time": 1.8172602653503418,
      "step": 43910
    },
    {
      "dynamics_loss_avg": 0.05268083475530148,
      "epoch": 4.1875839233398434e-05,
      "step": 43910
    },
    {
      "action_loss_avg": 0.0010765330167487263,
      "epoch": 4.1875839233398434e-05,
      "step": 43910
    },
    {
      "epoch": 4.1875839233398434e-05,
      "step": 43910,
      "training_step_time": 7.765373229980469
    },
    {
      "epoch": 4.1876792907714845e-05,
      "model_forward_time": 1.451685905456543,
      "step": 43911
    },
    {
      "epoch": 4.1876792907714845e-05,
      "step": 43911,
      "training_step_time": 7.518906116485596
    },
    {
      "epoch": 4.187774658203125e-05,
      "model_forward_time": 1.8114333152770996,
      "step": 43912
    },
    {
      "epoch": 4.187774658203125e-05,
      "step": 43912,
      "training_step_time": 6.864900827407837
    },
    {
      "epoch": 4.187870025634765e-05,
      "model_forward_time": 1.8089120388031006,
      "step": 43913
    },
    {
      "epoch": 4.187870025634765e-05,
      "step": 43913,
      "training_step_time": 6.826451539993286
    },
    {
      "epoch": 4.1879653930664064e-05,
      "model_forward_time": 1.445784091949463,
      "step": 43914
    },
    {
      "epoch": 4.1879653930664064e-05,
      "step": 43914,
      "training_step_time": 6.798959493637085
    },
    {
      "epoch": 4.188060760498047e-05,
      "model_forward_time": 2.33136248588562,
      "step": 43915
    },
    {
      "epoch": 4.188060760498047e-05,
      "step": 43915,
      "training_step_time": 7.1616857051849365
    },
    {
      "epoch": 4.188156127929687e-05,
      "model_forward_time": 1.823000431060791,
      "step": 43916
    },
    {
      "epoch": 4.188156127929687e-05,
      "step": 43916,
      "training_step_time": 9.600987434387207
    },
    {
      "epoch": 4.188251495361328e-05,
      "model_forward_time": 1.8228044509887695,
      "step": 43917
    },
    {
      "epoch": 4.188251495361328e-05,
      "step": 43917,
      "training_step_time": 8.72442364692688
    },
    {
      "epoch": 4.188346862792969e-05,
      "model_forward_time": 1.8166131973266602,
      "step": 43918
    },
    {
      "epoch": 4.188346862792969e-05,
      "step": 43918,
      "training_step_time": 9.442105531692505
    },
    {
      "epoch": 4.188442230224609e-05,
      "model_forward_time": 1.8127703666687012,
      "step": 43919
    },
    {
      "epoch": 4.188442230224609e-05,
      "step": 43919,
      "training_step_time": 7.728718996047974
    },
    {
      "epoch": 4.18853759765625e-05,
      "grad_norm": 0.08574273437261581,
      "learning_rate": 6.399877027868807e-06,
      "loss": 0.0622,
      "step": 43920
    },
    {
      "epoch": 4.18853759765625e-05,
      "model_forward_time": 1.811682939529419,
      "step": 43920
    },
    {
      "dynamics_loss_avg": 0.046222638338804245,
      "epoch": 4.18853759765625e-05,
      "step": 43920
    },
    {
      "action_loss_avg": 0.0011849540416733362,
      "epoch": 4.18853759765625e-05,
      "step": 43920
    },
    {
      "epoch": 4.18853759765625e-05,
      "step": 43920,
      "training_step_time": 7.5546159744262695
    },
    {
      "epoch": 4.1886329650878906e-05,
      "model_forward_time": 1.8212800025939941,
      "step": 43921
    },
    {
      "epoch": 4.1886329650878906e-05,
      "step": 43921,
      "training_step_time": 6.827193260192871
    },
    {
      "epoch": 4.188728332519531e-05,
      "model_forward_time": 1.8136465549468994,
      "step": 43922
    },
    {
      "epoch": 4.188728332519531e-05,
      "step": 43922,
      "training_step_time": 6.837070465087891
    },
    {
      "epoch": 4.188823699951172e-05,
      "model_forward_time": 1.8197076320648193,
      "step": 43923
    },
    {
      "epoch": 4.188823699951172e-05,
      "step": 43923,
      "training_step_time": 6.839806795120239
    },
    {
      "epoch": 4.1889190673828125e-05,
      "model_forward_time": 1.791555643081665,
      "step": 43924
    },
    {
      "epoch": 4.1889190673828125e-05,
      "step": 43924,
      "training_step_time": 6.814427852630615
    },
    {
      "epoch": 4.189014434814453e-05,
      "model_forward_time": 2.3397603034973145,
      "step": 43925
    },
    {
      "epoch": 4.189014434814453e-05,
      "step": 43925,
      "training_step_time": 7.384910821914673
    },
    {
      "epoch": 4.189109802246094e-05,
      "model_forward_time": 1.8070523738861084,
      "step": 43926
    },
    {
      "epoch": 4.189109802246094e-05,
      "step": 43926,
      "training_step_time": 8.311239004135132
    },
    {
      "epoch": 4.189205169677734e-05,
      "model_forward_time": 1.8071703910827637,
      "step": 43927
    },
    {
      "epoch": 4.189205169677734e-05,
      "step": 43927,
      "training_step_time": 8.844288110733032
    },
    {
      "epoch": 4.189300537109375e-05,
      "model_forward_time": 1.8129401206970215,
      "step": 43928
    },
    {
      "epoch": 4.189300537109375e-05,
      "step": 43928,
      "training_step_time": 8.3572998046875
    },
    {
      "epoch": 4.189395904541016e-05,
      "model_forward_time": 1.8072092533111572,
      "step": 43929
    },
    {
      "epoch": 4.189395904541016e-05,
      "step": 43929,
      "training_step_time": 8.205689191818237
    },
    {
      "epoch": 4.189491271972656e-05,
      "grad_norm": 0.1634116917848587,
      "learning_rate": 6.39828960846815e-06,
      "loss": 0.0593,
      "step": 43930
    },
    {
      "epoch": 4.189491271972656e-05,
      "model_forward_time": 1.8131728172302246,
      "step": 43930
    },
    {
      "dynamics_loss_avg": 0.06166843269020319,
      "epoch": 4.189491271972656e-05,
      "step": 43930
    },
    {
      "action_loss_avg": 0.0010561691888142377,
      "epoch": 4.189491271972656e-05,
      "step": 43930
    },
    {
      "epoch": 4.189491271972656e-05,
      "step": 43930,
      "training_step_time": 6.890228509902954
    },
    {
      "epoch": 4.1895866394042966e-05,
      "model_forward_time": 1.8121519088745117,
      "step": 43931
    },
    {
      "epoch": 4.1895866394042966e-05,
      "step": 43931,
      "training_step_time": 6.861260175704956
    },
    {
      "epoch": 4.189682006835938e-05,
      "model_forward_time": 1.8106112480163574,
      "step": 43932
    },
    {
      "epoch": 4.189682006835938e-05,
      "step": 43932,
      "training_step_time": 6.876272201538086
    },
    {
      "epoch": 4.189777374267578e-05,
      "model_forward_time": 1.821244478225708,
      "step": 43933
    },
    {
      "epoch": 4.189777374267578e-05,
      "step": 43933,
      "training_step_time": 6.863698244094849
    },
    {
      "epoch": 4.1898727416992185e-05,
      "model_forward_time": 2.3286750316619873,
      "step": 43934
    },
    {
      "epoch": 4.1898727416992185e-05,
      "step": 43934,
      "training_step_time": 8.393137216567993
    },
    {
      "epoch": 4.1899681091308596e-05,
      "model_forward_time": 1.8043928146362305,
      "step": 43935
    },
    {
      "epoch": 4.1899681091308596e-05,
      "step": 43935,
      "training_step_time": 7.861355543136597
    },
    {
      "epoch": 4.1900634765625e-05,
      "model_forward_time": 1.8160085678100586,
      "step": 43936
    },
    {
      "epoch": 4.1900634765625e-05,
      "step": 43936,
      "training_step_time": 9.590886354446411
    },
    {
      "epoch": 4.1901588439941404e-05,
      "model_forward_time": 1.811288595199585,
      "step": 43937
    },
    {
      "epoch": 4.1901588439941404e-05,
      "step": 43937,
      "training_step_time": 9.066619157791138
    },
    {
      "epoch": 4.1902542114257815e-05,
      "model_forward_time": 1.8096976280212402,
      "step": 43938
    },
    {
      "epoch": 4.1902542114257815e-05,
      "step": 43938,
      "training_step_time": 8.674055099487305
    },
    {
      "epoch": 4.190349578857422e-05,
      "model_forward_time": 1.82481050491333,
      "step": 43939
    },
    {
      "epoch": 4.190349578857422e-05,
      "step": 43939,
      "training_step_time": 7.449315547943115
    },
    {
      "epoch": 4.190444946289062e-05,
      "grad_norm": 0.12986518442630768,
      "learning_rate": 6.396702036152644e-06,
      "loss": 0.0552,
      "step": 43940
    },
    {
      "epoch": 4.190444946289062e-05,
      "model_forward_time": 1.8203315734863281,
      "step": 43940
    },
    {
      "dynamics_loss_avg": 0.04553779065608978,
      "epoch": 4.190444946289062e-05,
      "step": 43940
    },
    {
      "action_loss_avg": 0.0018035476125078276,
      "epoch": 4.190444946289062e-05,
      "step": 43940
    },
    {
      "epoch": 4.190444946289062e-05,
      "step": 43940,
      "training_step_time": 6.800687074661255
    },
    {
      "epoch": 4.1905403137207034e-05,
      "model_forward_time": 1.8101096153259277,
      "step": 43941
    },
    {
      "epoch": 4.1905403137207034e-05,
      "step": 43941,
      "training_step_time": 6.900805711746216
    },
    {
      "epoch": 4.190635681152344e-05,
      "model_forward_time": 1.8183026313781738,
      "step": 43942
    },
    {
      "epoch": 4.190635681152344e-05,
      "step": 43942,
      "training_step_time": 6.682965517044067
    },
    {
      "epoch": 4.190731048583984e-05,
      "model_forward_time": 2.338500499725342,
      "step": 43943
    },
    {
      "epoch": 4.190731048583984e-05,
      "step": 43943,
      "training_step_time": 8.078620433807373
    },
    {
      "epoch": 4.190826416015625e-05,
      "model_forward_time": 1.8152623176574707,
      "step": 43944
    },
    {
      "epoch": 4.190826416015625e-05,
      "step": 43944,
      "training_step_time": 8.417433738708496
    },
    {
      "epoch": 4.1909217834472657e-05,
      "model_forward_time": 1.811164379119873,
      "step": 43945
    },
    {
      "epoch": 4.1909217834472657e-05,
      "step": 43945,
      "training_step_time": 7.730532169342041
    },
    {
      "epoch": 4.191017150878906e-05,
      "model_forward_time": 1.8088276386260986,
      "step": 43946
    },
    {
      "epoch": 4.191017150878906e-05,
      "step": 43946,
      "training_step_time": 9.288221836090088
    },
    {
      "epoch": 4.191112518310547e-05,
      "model_forward_time": 1.814807415008545,
      "step": 43947
    },
    {
      "epoch": 4.191112518310547e-05,
      "step": 43947,
      "training_step_time": 9.458404779434204
    },
    {
      "epoch": 4.1912078857421875e-05,
      "model_forward_time": 1.804354190826416,
      "step": 43948
    },
    {
      "epoch": 4.1912078857421875e-05,
      "step": 43948,
      "training_step_time": 7.450477838516235
    },
    {
      "epoch": 4.191303253173828e-05,
      "model_forward_time": 1.7983667850494385,
      "step": 43949
    },
    {
      "epoch": 4.191303253173828e-05,
      "step": 43949,
      "training_step_time": 6.744020938873291
    },
    {
      "epoch": 4.191398620605469e-05,
      "grad_norm": 0.11664306372404099,
      "learning_rate": 6.395114311095904e-06,
      "loss": 0.0591,
      "step": 43950
    },
    {
      "epoch": 4.191398620605469e-05,
      "model_forward_time": 1.8054466247558594,
      "step": 43950
    },
    {
      "dynamics_loss_avg": 0.057834482938051227,
      "epoch": 4.191398620605469e-05,
      "step": 43950
    },
    {
      "action_loss_avg": 0.0014596221037209034,
      "epoch": 4.191398620605469e-05,
      "step": 43950
    },
    {
      "epoch": 4.191398620605469e-05,
      "step": 43950,
      "training_step_time": 6.7132298946380615
    },
    {
      "epoch": 4.1914939880371094e-05,
      "model_forward_time": 1.804778814315796,
      "step": 43951
    },
    {
      "epoch": 4.1914939880371094e-05,
      "step": 43951,
      "training_step_time": 6.715811491012573
    },
    {
      "epoch": 4.19158935546875e-05,
      "model_forward_time": 2.3445565700531006,
      "step": 43952
    },
    {
      "epoch": 4.19158935546875e-05,
      "step": 43952,
      "training_step_time": 7.058152198791504
    },
    {
      "epoch": 4.191684722900391e-05,
      "model_forward_time": 1.8096981048583984,
      "step": 43953
    },
    {
      "epoch": 4.191684722900391e-05,
      "step": 43953,
      "training_step_time": 9.312823057174683
    },
    {
      "epoch": 4.191780090332031e-05,
      "model_forward_time": 1.810988187789917,
      "step": 43954
    },
    {
      "epoch": 4.191780090332031e-05,
      "step": 43954,
      "training_step_time": 9.115500926971436
    },
    {
      "epoch": 4.191875457763672e-05,
      "model_forward_time": 1.8051245212554932,
      "step": 43955
    },
    {
      "epoch": 4.191875457763672e-05,
      "step": 43955,
      "training_step_time": 7.654316663742065
    },
    {
      "epoch": 4.191970825195313e-05,
      "model_forward_time": 1.807910442352295,
      "step": 43956
    },
    {
      "epoch": 4.191970825195313e-05,
      "step": 43956,
      "training_step_time": 7.713428258895874
    },
    {
      "epoch": 4.192066192626953e-05,
      "model_forward_time": 1.8114371299743652,
      "step": 43957
    },
    {
      "epoch": 4.192066192626953e-05,
      "step": 43957,
      "training_step_time": 7.447343826293945
    },
    {
      "epoch": 4.1921615600585936e-05,
      "model_forward_time": 1.8050050735473633,
      "step": 43958
    },
    {
      "epoch": 4.1921615600585936e-05,
      "step": 43958,
      "training_step_time": 6.8327906131744385
    },
    {
      "epoch": 4.192256927490235e-05,
      "model_forward_time": 1.8079705238342285,
      "step": 43959
    },
    {
      "epoch": 4.192256927490235e-05,
      "step": 43959,
      "training_step_time": 6.770854711532593
    },
    {
      "epoch": 4.192352294921875e-05,
      "grad_norm": 0.16314451396465302,
      "learning_rate": 6.39352643347156e-06,
      "loss": 0.0555,
      "step": 43960
    },
    {
      "epoch": 4.192352294921875e-05,
      "model_forward_time": 1.804262399673462,
      "step": 43960
    },
    {
      "dynamics_loss_avg": 0.03709839526563883,
      "epoch": 4.192352294921875e-05,
      "step": 43960
    },
    {
      "action_loss_avg": 0.0007093639738741331,
      "epoch": 4.192352294921875e-05,
      "step": 43960
    },
    {
      "epoch": 4.192352294921875e-05,
      "step": 43960,
      "training_step_time": 6.828567266464233
    },
    {
      "epoch": 4.1924476623535155e-05,
      "model_forward_time": 2.336444139480591,
      "step": 43961
    },
    {
      "epoch": 4.1924476623535155e-05,
      "step": 43961,
      "training_step_time": 7.160013914108276
    },
    {
      "epoch": 4.1925430297851566e-05,
      "model_forward_time": 1.7897765636444092,
      "step": 43962
    },
    {
      "epoch": 4.1925430297851566e-05,
      "step": 43962,
      "training_step_time": 9.33634877204895
    },
    {
      "epoch": 4.192638397216797e-05,
      "model_forward_time": 1.8119051456451416,
      "step": 43963
    },
    {
      "epoch": 4.192638397216797e-05,
      "step": 43963,
      "training_step_time": 10.181031942367554
    },
    {
      "epoch": 4.1927337646484374e-05,
      "model_forward_time": 1.8073625564575195,
      "step": 43964
    },
    {
      "epoch": 4.1927337646484374e-05,
      "step": 43964,
      "training_step_time": 9.982361555099487
    },
    {
      "epoch": 4.1928291320800785e-05,
      "model_forward_time": 1.812711477279663,
      "step": 43965
    },
    {
      "epoch": 4.1928291320800785e-05,
      "step": 43965,
      "training_step_time": 7.856423616409302
    },
    {
      "epoch": 4.192924499511719e-05,
      "model_forward_time": 1.819878339767456,
      "step": 43966
    },
    {
      "epoch": 4.192924499511719e-05,
      "step": 43966,
      "training_step_time": 7.592267990112305
    },
    {
      "epoch": 4.193019866943359e-05,
      "model_forward_time": 1.8185079097747803,
      "step": 43967
    },
    {
      "epoch": 4.193019866943359e-05,
      "step": 43967,
      "training_step_time": 7.556414604187012
    },
    {
      "epoch": 4.193115234375e-05,
      "model_forward_time": 1.8167624473571777,
      "step": 43968
    },
    {
      "epoch": 4.193115234375e-05,
      "step": 43968,
      "training_step_time": 6.94546103477478
    },
    {
      "epoch": 4.193210601806641e-05,
      "model_forward_time": 1.8167810440063477,
      "step": 43969
    },
    {
      "epoch": 4.193210601806641e-05,
      "step": 43969,
      "training_step_time": 6.859436988830566
    },
    {
      "epoch": 4.193305969238281e-05,
      "grad_norm": 0.1456633061170578,
      "learning_rate": 6.391938403453262e-06,
      "loss": 0.0609,
      "step": 43970
    },
    {
      "epoch": 4.193305969238281e-05,
      "model_forward_time": 1.8194241523742676,
      "step": 43970
    },
    {
      "dynamics_loss_avg": 0.05885912124067545,
      "epoch": 4.193305969238281e-05,
      "step": 43970
    },
    {
      "action_loss_avg": 0.0017064841464161872,
      "epoch": 4.193305969238281e-05,
      "step": 43970
    },
    {
      "epoch": 4.193305969238281e-05,
      "step": 43970,
      "training_step_time": 7.189621448516846
    },
    {
      "epoch": 4.1934013366699215e-05,
      "model_forward_time": 2.352762460708618,
      "step": 43971
    },
    {
      "epoch": 4.1934013366699215e-05,
      "step": 43971,
      "training_step_time": 8.896734952926636
    },
    {
      "epoch": 4.1934967041015626e-05,
      "model_forward_time": 1.827446699142456,
      "step": 43972
    },
    {
      "epoch": 4.1934967041015626e-05,
      "step": 43972,
      "training_step_time": 8.561243534088135
    },
    {
      "epoch": 4.193592071533203e-05,
      "model_forward_time": 1.8137352466583252,
      "step": 43973
    },
    {
      "epoch": 4.193592071533203e-05,
      "step": 43973,
      "training_step_time": 9.005879640579224
    },
    {
      "epoch": 4.1936874389648434e-05,
      "model_forward_time": 1.8121984004974365,
      "step": 43974
    },
    {
      "epoch": 4.1936874389648434e-05,
      "step": 43974,
      "training_step_time": 8.290608406066895
    },
    {
      "epoch": 4.1937828063964845e-05,
      "model_forward_time": 1.8231074810028076,
      "step": 43975
    },
    {
      "epoch": 4.1937828063964845e-05,
      "step": 43975,
      "training_step_time": 7.810639142990112
    },
    {
      "epoch": 4.193878173828125e-05,
      "model_forward_time": 1.810575008392334,
      "step": 43976
    },
    {
      "epoch": 4.193878173828125e-05,
      "step": 43976,
      "training_step_time": 7.568644046783447
    },
    {
      "epoch": 4.193973541259765e-05,
      "model_forward_time": 1.8251466751098633,
      "step": 43977
    },
    {
      "epoch": 4.193973541259765e-05,
      "step": 43977,
      "training_step_time": 6.8893208503723145
    },
    {
      "epoch": 4.1940689086914064e-05,
      "model_forward_time": 1.8220891952514648,
      "step": 43978
    },
    {
      "epoch": 4.1940689086914064e-05,
      "step": 43978,
      "training_step_time": 6.83213210105896
    },
    {
      "epoch": 4.194164276123047e-05,
      "model_forward_time": 1.8245458602905273,
      "step": 43979
    },
    {
      "epoch": 4.194164276123047e-05,
      "step": 43979,
      "training_step_time": 7.136624097824097
    },
    {
      "epoch": 4.194259643554687e-05,
      "grad_norm": 0.1062539666891098,
      "learning_rate": 6.390350221214671e-06,
      "loss": 0.059,
      "step": 43980
    },
    {
      "epoch": 4.194259643554687e-05,
      "model_forward_time": 2.3432600498199463,
      "step": 43980
    },
    {
      "dynamics_loss_avg": 0.05516864284873009,
      "epoch": 4.194259643554687e-05,
      "step": 43980
    },
    {
      "action_loss_avg": 0.0012564076838316395,
      "epoch": 4.194259643554687e-05,
      "step": 43980
    },
    {
      "epoch": 4.194259643554687e-05,
      "step": 43980,
      "training_step_time": 7.849818706512451
    },
    {
      "epoch": 4.194355010986328e-05,
      "model_forward_time": 1.8212182521820068,
      "step": 43981
    },
    {
      "epoch": 4.194355010986328e-05,
      "step": 43981,
      "training_step_time": 8.949826717376709
    },
    {
      "epoch": 4.194450378417969e-05,
      "model_forward_time": 1.8217620849609375,
      "step": 43982
    },
    {
      "epoch": 4.194450378417969e-05,
      "step": 43982,
      "training_step_time": 8.741181373596191
    },
    {
      "epoch": 4.194545745849609e-05,
      "model_forward_time": 1.822411060333252,
      "step": 43983
    },
    {
      "epoch": 4.194545745849609e-05,
      "step": 43983,
      "training_step_time": 7.734662771224976
    },
    {
      "epoch": 4.19464111328125e-05,
      "model_forward_time": 1.8234610557556152,
      "step": 43984
    },
    {
      "epoch": 4.19464111328125e-05,
      "step": 43984,
      "training_step_time": 8.578807592391968
    },
    {
      "epoch": 4.1947364807128906e-05,
      "model_forward_time": 1.820666790008545,
      "step": 43985
    },
    {
      "epoch": 4.1947364807128906e-05,
      "step": 43985,
      "training_step_time": 7.627486228942871
    },
    {
      "epoch": 4.194831848144531e-05,
      "model_forward_time": 1.8234169483184814,
      "step": 43986
    },
    {
      "epoch": 4.194831848144531e-05,
      "step": 43986,
      "training_step_time": 6.919934272766113
    },
    {
      "epoch": 4.194927215576172e-05,
      "model_forward_time": 1.8186192512512207,
      "step": 43987
    },
    {
      "epoch": 4.194927215576172e-05,
      "step": 43987,
      "training_step_time": 6.91661810874939
    },
    {
      "epoch": 4.1950225830078125e-05,
      "model_forward_time": 1.8178579807281494,
      "step": 43988
    },
    {
      "epoch": 4.1950225830078125e-05,
      "step": 43988,
      "training_step_time": 7.11960506439209
    },
    {
      "epoch": 4.195117950439453e-05,
      "model_forward_time": 2.3329520225524902,
      "step": 43989
    },
    {
      "epoch": 4.195117950439453e-05,
      "step": 43989,
      "training_step_time": 7.894644021987915
    },
    {
      "epoch": 4.195213317871094e-05,
      "grad_norm": 0.13038504123687744,
      "learning_rate": 6.38876188692947e-06,
      "loss": 0.0608,
      "step": 43990
    },
    {
      "epoch": 4.195213317871094e-05,
      "model_forward_time": 1.799006462097168,
      "step": 43990
    },
    {
      "dynamics_loss_avg": 0.05990949422121048,
      "epoch": 4.195213317871094e-05,
      "step": 43990
    },
    {
      "action_loss_avg": 0.0015248638170305639,
      "epoch": 4.195213317871094e-05,
      "step": 43990
    },
    {
      "epoch": 4.195213317871094e-05,
      "step": 43990,
      "training_step_time": 8.891887426376343
    },
    {
      "epoch": 4.1953086853027343e-05,
      "model_forward_time": 1.8170697689056396,
      "step": 43991
    },
    {
      "epoch": 4.1953086853027343e-05,
      "step": 43991,
      "training_step_time": 8.652246475219727
    },
    {
      "epoch": 4.195404052734375e-05,
      "model_forward_time": 1.816983938217163,
      "step": 43992
    },
    {
      "epoch": 4.195404052734375e-05,
      "step": 43992,
      "training_step_time": 7.829163551330566
    },
    {
      "epoch": 4.195499420166016e-05,
      "model_forward_time": 1.8217754364013672,
      "step": 43993
    },
    {
      "epoch": 4.195499420166016e-05,
      "step": 43993,
      "training_step_time": 7.734737396240234
    },
    {
      "epoch": 4.195594787597656e-05,
      "model_forward_time": 1.8095884323120117,
      "step": 43994
    },
    {
      "epoch": 4.195594787597656e-05,
      "step": 43994,
      "training_step_time": 8.134842872619629
    },
    {
      "epoch": 4.1956901550292966e-05,
      "model_forward_time": 1.8180391788482666,
      "step": 43995
    },
    {
      "epoch": 4.1956901550292966e-05,
      "step": 43995,
      "training_step_time": 7.533819675445557
    },
    {
      "epoch": 4.195785522460938e-05,
      "model_forward_time": 1.811150312423706,
      "step": 43996
    },
    {
      "epoch": 4.195785522460938e-05,
      "step": 43996,
      "training_step_time": 6.906184196472168
    },
    {
      "epoch": 4.195880889892578e-05,
      "model_forward_time": 1.8089351654052734,
      "step": 43997
    },
    {
      "epoch": 4.195880889892578e-05,
      "step": 43997,
      "training_step_time": 7.140010595321655
    },
    {
      "epoch": 4.1959762573242185e-05,
      "model_forward_time": 2.352445125579834,
      "step": 43998
    },
    {
      "epoch": 4.1959762573242185e-05,
      "step": 43998,
      "training_step_time": 9.094247102737427
    },
    {
      "epoch": 4.1960716247558596e-05,
      "model_forward_time": 1.7958433628082275,
      "step": 43999
    },
    {
      "epoch": 4.1960716247558596e-05,
      "step": 43999,
      "training_step_time": 7.739361047744751
    },
    {
      "epoch": 4.1961669921875e-05,
      "grad_norm": 0.15723618865013123,
      "learning_rate": 6.387173400771358e-06,
      "loss": 0.0569,
      "step": 44000
    },
    {
      "epoch": 4.1961669921875e-05,
      "model_forward_time": 1.7531187534332275,
      "step": 44000
    },
    {
      "dynamics_loss_avg": 0.06829477995634078,
      "epoch": 4.1961669921875e-05,
      "step": 44000
    },
    {
      "action_loss_avg": 0.0015778229077113791,
      "epoch": 4.1961669921875e-05,
      "step": 44000
    },
    {
      "epoch": 4.1961669921875e-05,
      "step": 44000,
      "training_step_time": 9.913079977035522
    },
    {
      "epoch": 4.1962623596191404e-05,
      "model_forward_time": 1.7209196090698242,
      "step": 44001
    },
    {
      "epoch": 4.1962623596191404e-05,
      "step": 44001,
      "training_step_time": 9.987597227096558
    },
    {
      "epoch": 4.1963577270507815e-05,
      "model_forward_time": 1.0319468975067139,
      "step": 44002
    },
    {
      "epoch": 4.1963577270507815e-05,
      "step": 44002,
      "training_step_time": 8.622417449951172
    },
    {
      "epoch": 4.196453094482422e-05,
      "model_forward_time": 1.7284557819366455,
      "step": 44003
    },
    {
      "epoch": 4.196453094482422e-05,
      "step": 44003,
      "training_step_time": 7.43639349937439
    },
    {
      "epoch": 4.196548461914062e-05,
      "model_forward_time": 1.7293283939361572,
      "step": 44004
    },
    {
      "epoch": 4.196548461914062e-05,
      "step": 44004,
      "training_step_time": 7.252615213394165
    },
    {
      "epoch": 4.1966438293457034e-05,
      "model_forward_time": 1.730560541152954,
      "step": 44005
    },
    {
      "epoch": 4.1966438293457034e-05,
      "step": 44005,
      "training_step_time": 7.328381538391113
    },
    {
      "epoch": 4.196739196777344e-05,
      "model_forward_time": 1.729416847229004,
      "step": 44006
    },
    {
      "epoch": 4.196739196777344e-05,
      "step": 44006,
      "training_step_time": 7.189247131347656
    },
    {
      "epoch": 4.196834564208984e-05,
      "model_forward_time": 2.7642569541931152,
      "step": 44007
    },
    {
      "epoch": 4.196834564208984e-05,
      "step": 44007,
      "training_step_time": 8.742073774337769
    },
    {
      "epoch": 4.196929931640625e-05,
      "model_forward_time": 1.729471206665039,
      "step": 44008
    },
    {
      "epoch": 4.196929931640625e-05,
      "step": 44008,
      "training_step_time": 9.625538349151611
    },
    {
      "epoch": 4.1970252990722657e-05,
      "model_forward_time": 1.7522125244140625,
      "step": 44009
    },
    {
      "epoch": 4.1970252990722657e-05,
      "step": 44009,
      "training_step_time": 10.42367696762085
    },
    {
      "epoch": 4.197120666503906e-05,
      "grad_norm": 0.11187385022640228,
      "learning_rate": 6.385584762914047e-06,
      "loss": 0.0573,
      "step": 44010
    },
    {
      "epoch": 4.197120666503906e-05,
      "model_forward_time": 1.7184593677520752,
      "step": 44010
    },
    {
      "dynamics_loss_avg": 0.0633521681651473,
      "epoch": 4.197120666503906e-05,
      "step": 44010
    },
    {
      "action_loss_avg": 0.002648017491446808,
      "epoch": 4.197120666503906e-05,
      "step": 44010
    },
    {
      "epoch": 4.197120666503906e-05,
      "step": 44010,
      "training_step_time": 8.684809923171997
    },
    {
      "epoch": 4.197216033935547e-05,
      "model_forward_time": 1.7224912643432617,
      "step": 44011
    },
    {
      "epoch": 4.197216033935547e-05,
      "step": 44011,
      "training_step_time": 8.314514636993408
    },
    {
      "epoch": 4.1973114013671875e-05,
      "model_forward_time": 1.72054123878479,
      "step": 44012
    },
    {
      "epoch": 4.1973114013671875e-05,
      "step": 44012,
      "training_step_time": 7.462260961532593
    },
    {
      "epoch": 4.197406768798828e-05,
      "model_forward_time": 1.7191555500030518,
      "step": 44013
    },
    {
      "epoch": 4.197406768798828e-05,
      "step": 44013,
      "training_step_time": 7.517672061920166
    },
    {
      "epoch": 4.197502136230469e-05,
      "model_forward_time": 1.7224040031433105,
      "step": 44014
    },
    {
      "epoch": 4.197502136230469e-05,
      "step": 44014,
      "training_step_time": 6.796101331710815
    },
    {
      "epoch": 4.1975975036621094e-05,
      "model_forward_time": 1.7089881896972656,
      "step": 44015
    },
    {
      "epoch": 4.1975975036621094e-05,
      "step": 44015,
      "training_step_time": 7.158925533294678
    },
    {
      "epoch": 4.19769287109375e-05,
      "model_forward_time": 2.2814791202545166,
      "step": 44016
    },
    {
      "epoch": 4.19769287109375e-05,
      "step": 44016,
      "training_step_time": 7.800001859664917
    },
    {
      "epoch": 4.197788238525391e-05,
      "model_forward_time": 1.7102153301239014,
      "step": 44017
    },
    {
      "epoch": 4.197788238525391e-05,
      "step": 44017,
      "training_step_time": 7.774145841598511
    },
    {
      "epoch": 4.197883605957031e-05,
      "model_forward_time": 1.7155816555023193,
      "step": 44018
    },
    {
      "epoch": 4.197883605957031e-05,
      "step": 44018,
      "training_step_time": 10.5901460647583
    },
    {
      "epoch": 4.197978973388672e-05,
      "model_forward_time": 1.714127779006958,
      "step": 44019
    },
    {
      "epoch": 4.197978973388672e-05,
      "step": 44019,
      "training_step_time": 7.743870735168457
    },
    {
      "epoch": 4.198074340820313e-05,
      "grad_norm": 0.14646857976913452,
      "learning_rate": 6.383995973531271e-06,
      "loss": 0.058,
      "step": 44020
    },
    {
      "epoch": 4.198074340820313e-05,
      "model_forward_time": 1.7170250415802002,
      "step": 44020
    },
    {
      "dynamics_loss_avg": 0.0512159263715148,
      "epoch": 4.198074340820313e-05,
      "step": 44020
    },
    {
      "action_loss_avg": 0.001546081656124443,
      "epoch": 4.198074340820313e-05,
      "step": 44020
    },
    {
      "epoch": 4.198074340820313e-05,
      "step": 44020,
      "training_step_time": 8.850940704345703
    },
    {
      "epoch": 4.198169708251953e-05,
      "model_forward_time": 1.7072691917419434,
      "step": 44021
    },
    {
      "epoch": 4.198169708251953e-05,
      "step": 44021,
      "training_step_time": 7.7873523235321045
    },
    {
      "epoch": 4.1982650756835936e-05,
      "model_forward_time": 1.7087576389312744,
      "step": 44022
    },
    {
      "epoch": 4.1982650756835936e-05,
      "step": 44022,
      "training_step_time": 7.590701580047607
    },
    {
      "epoch": 4.198360443115235e-05,
      "model_forward_time": 1.7089219093322754,
      "step": 44023
    },
    {
      "epoch": 4.198360443115235e-05,
      "step": 44023,
      "training_step_time": 6.792271375656128
    },
    {
      "epoch": 4.198455810546875e-05,
      "model_forward_time": 1.7087576389312744,
      "step": 44024
    },
    {
      "epoch": 4.198455810546875e-05,
      "step": 44024,
      "training_step_time": 7.213668584823608
    },
    {
      "epoch": 4.1985511779785155e-05,
      "model_forward_time": 1.723695993423462,
      "step": 44025
    },
    {
      "epoch": 4.1985511779785155e-05,
      "step": 44025,
      "training_step_time": 8.201359272003174
    },
    {
      "epoch": 4.1986465454101566e-05,
      "model_forward_time": 1.7048091888427734,
      "step": 44026
    },
    {
      "epoch": 4.1986465454101566e-05,
      "step": 44026,
      "training_step_time": 7.810070276260376
    },
    {
      "epoch": 4.198741912841797e-05,
      "model_forward_time": 1.707550048828125,
      "step": 44027
    },
    {
      "epoch": 4.198741912841797e-05,
      "step": 44027,
      "training_step_time": 8.679290294647217
    },
    {
      "epoch": 4.1988372802734374e-05,
      "model_forward_time": 1.749351978302002,
      "step": 44028
    },
    {
      "epoch": 4.1988372802734374e-05,
      "step": 44028,
      "training_step_time": 8.5676589012146
    },
    {
      "epoch": 4.1989326477050785e-05,
      "model_forward_time": 1.7093253135681152,
      "step": 44029
    },
    {
      "epoch": 4.1989326477050785e-05,
      "step": 44029,
      "training_step_time": 7.934905290603638
    },
    {
      "epoch": 4.199028015136719e-05,
      "grad_norm": 0.10330747067928314,
      "learning_rate": 6.3824070327967744e-06,
      "loss": 0.059,
      "step": 44030
    },
    {
      "epoch": 4.199028015136719e-05,
      "model_forward_time": 1.7205660343170166,
      "step": 44030
    },
    {
      "dynamics_loss_avg": 0.06189906094223261,
      "epoch": 4.199028015136719e-05,
      "step": 44030
    },
    {
      "action_loss_avg": 0.0021805531956488266,
      "epoch": 4.199028015136719e-05,
      "step": 44030
    },
    {
      "epoch": 4.199028015136719e-05,
      "step": 44030,
      "training_step_time": 7.854055643081665
    },
    {
      "epoch": 4.199123382568359e-05,
      "model_forward_time": 1.7033390998840332,
      "step": 44031
    },
    {
      "epoch": 4.199123382568359e-05,
      "step": 44031,
      "training_step_time": 7.537961483001709
    },
    {
      "epoch": 4.19921875e-05,
      "model_forward_time": 1.7119243144989014,
      "step": 44032
    },
    {
      "epoch": 4.19921875e-05,
      "step": 44032,
      "training_step_time": 6.860890865325928
    },
    {
      "epoch": 4.199314117431641e-05,
      "model_forward_time": 1.7072298526763916,
      "step": 44033
    },
    {
      "epoch": 4.199314117431641e-05,
      "step": 44033,
      "training_step_time": 7.173675060272217
    },
    {
      "epoch": 4.199409484863281e-05,
      "model_forward_time": 1.711416244506836,
      "step": 44034
    },
    {
      "epoch": 4.199409484863281e-05,
      "step": 44034,
      "training_step_time": 7.690566539764404
    },
    {
      "epoch": 4.1995048522949215e-05,
      "model_forward_time": 1.545729637145996,
      "step": 44035
    },
    {
      "epoch": 4.1995048522949215e-05,
      "step": 44035,
      "training_step_time": 7.747058391571045
    },
    {
      "epoch": 4.1996002197265626e-05,
      "model_forward_time": 1.708498239517212,
      "step": 44036
    },
    {
      "epoch": 4.1996002197265626e-05,
      "step": 44036,
      "training_step_time": 8.77060842514038
    },
    {
      "epoch": 4.199695587158203e-05,
      "model_forward_time": 1.7381000518798828,
      "step": 44037
    },
    {
      "epoch": 4.199695587158203e-05,
      "step": 44037,
      "training_step_time": 7.783975601196289
    },
    {
      "epoch": 4.1997909545898434e-05,
      "model_forward_time": 1.7174184322357178,
      "step": 44038
    },
    {
      "epoch": 4.1997909545898434e-05,
      "step": 44038,
      "training_step_time": 7.898097991943359
    },
    {
      "epoch": 4.1998863220214845e-05,
      "model_forward_time": 1.730616807937622,
      "step": 44039
    },
    {
      "epoch": 4.1998863220214845e-05,
      "step": 44039,
      "training_step_time": 7.817853927612305
    },
    {
      "epoch": 4.199981689453125e-05,
      "grad_norm": 0.10640984028577805,
      "learning_rate": 6.3808179408843255e-06,
      "loss": 0.0583,
      "step": 44040
    },
    {
      "epoch": 4.199981689453125e-05,
      "model_forward_time": 1.7211096286773682,
      "step": 44040
    },
    {
      "dynamics_loss_avg": 0.06241401676088572,
      "epoch": 4.199981689453125e-05,
      "step": 44040
    },
    {
      "action_loss_avg": 0.002104388293810189,
      "epoch": 4.199981689453125e-05,
      "step": 44040
    },
    {
      "epoch": 4.199981689453125e-05,
      "step": 44040,
      "training_step_time": 7.524707555770874
    },
    {
      "epoch": 4.200077056884765e-05,
      "model_forward_time": 1.7208070755004883,
      "step": 44041
    },
    {
      "epoch": 4.200077056884765e-05,
      "step": 44041,
      "training_step_time": 6.81938910484314
    },
    {
      "epoch": 4.2001724243164064e-05,
      "model_forward_time": 1.7201817035675049,
      "step": 44042
    },
    {
      "epoch": 4.2001724243164064e-05,
      "step": 44042,
      "training_step_time": 7.224123954772949
    },
    {
      "epoch": 4.200267791748047e-05,
      "model_forward_time": 1.7236809730529785,
      "step": 44043
    },
    {
      "epoch": 4.200267791748047e-05,
      "step": 44043,
      "training_step_time": 7.3696582317352295
    },
    {
      "epoch": 4.200363159179687e-05,
      "model_forward_time": 2.272968292236328,
      "step": 44044
    },
    {
      "epoch": 4.200363159179687e-05,
      "step": 44044,
      "training_step_time": 8.409855842590332
    },
    {
      "epoch": 4.200458526611328e-05,
      "model_forward_time": 1.7159531116485596,
      "step": 44045
    },
    {
      "epoch": 4.200458526611328e-05,
      "step": 44045,
      "training_step_time": 9.16533613204956
    },
    {
      "epoch": 4.200553894042969e-05,
      "model_forward_time": 1.7264368534088135,
      "step": 44046
    },
    {
      "epoch": 4.200553894042969e-05,
      "step": 44046,
      "training_step_time": 10.078558206558228
    },
    {
      "epoch": 4.200649261474609e-05,
      "model_forward_time": 1.7231099605560303,
      "step": 44047
    },
    {
      "epoch": 4.200649261474609e-05,
      "step": 44047,
      "training_step_time": 8.867750644683838
    },
    {
      "epoch": 4.20074462890625e-05,
      "model_forward_time": 1.7164688110351562,
      "step": 44048
    },
    {
      "epoch": 4.20074462890625e-05,
      "step": 44048,
      "training_step_time": 8.871867895126343
    },
    {
      "epoch": 4.2008399963378906e-05,
      "model_forward_time": 1.7204368114471436,
      "step": 44049
    },
    {
      "epoch": 4.2008399963378906e-05,
      "step": 44049,
      "training_step_time": 7.725138187408447
    },
    {
      "epoch": 4.200935363769531e-05,
      "grad_norm": 0.11809780448675156,
      "learning_rate": 6.379228697967702e-06,
      "loss": 0.0618,
      "step": 44050
    },
    {
      "epoch": 4.200935363769531e-05,
      "model_forward_time": 1.7155637741088867,
      "step": 44050
    },
    {
      "dynamics_loss_avg": 0.06239752694964409,
      "epoch": 4.200935363769531e-05,
      "step": 44050
    },
    {
      "action_loss_avg": 0.0028947994665941223,
      "epoch": 4.200935363769531e-05,
      "step": 44050
    },
    {
      "epoch": 4.200935363769531e-05,
      "step": 44050,
      "training_step_time": 6.8107380867004395
    },
    {
      "epoch": 4.201030731201172e-05,
      "model_forward_time": 1.7203869819641113,
      "step": 44051
    },
    {
      "epoch": 4.201030731201172e-05,
      "step": 44051,
      "training_step_time": 7.137113809585571
    },
    {
      "epoch": 4.2011260986328125e-05,
      "model_forward_time": 1.719496488571167,
      "step": 44052
    },
    {
      "epoch": 4.2011260986328125e-05,
      "step": 44052,
      "training_step_time": 7.307549953460693
    },
    {
      "epoch": 4.201221466064453e-05,
      "model_forward_time": 2.280921697616577,
      "step": 44053
    },
    {
      "epoch": 4.201221466064453e-05,
      "step": 44053,
      "training_step_time": 8.233721733093262
    },
    {
      "epoch": 4.201316833496094e-05,
      "model_forward_time": 1.7301907539367676,
      "step": 44054
    },
    {
      "epoch": 4.201316833496094e-05,
      "step": 44054,
      "training_step_time": 7.682807683944702
    },
    {
      "epoch": 4.2014122009277343e-05,
      "model_forward_time": 1.7458715438842773,
      "step": 44055
    },
    {
      "epoch": 4.2014122009277343e-05,
      "step": 44055,
      "training_step_time": 10.266093254089355
    },
    {
      "epoch": 4.201507568359375e-05,
      "model_forward_time": 1.7530431747436523,
      "step": 44056
    },
    {
      "epoch": 4.201507568359375e-05,
      "step": 44056,
      "training_step_time": 7.847285270690918
    },
    {
      "epoch": 4.201602935791016e-05,
      "model_forward_time": 1.7268664836883545,
      "step": 44057
    },
    {
      "epoch": 4.201602935791016e-05,
      "step": 44057,
      "training_step_time": 8.340103387832642
    },
    {
      "epoch": 4.201698303222656e-05,
      "model_forward_time": 1.7130918502807617,
      "step": 44058
    },
    {
      "epoch": 4.201698303222656e-05,
      "step": 44058,
      "training_step_time": 8.245696544647217
    },
    {
      "epoch": 4.2017936706542966e-05,
      "model_forward_time": 1.7144246101379395,
      "step": 44059
    },
    {
      "epoch": 4.2017936706542966e-05,
      "step": 44059,
      "training_step_time": 7.566962003707886
    },
    {
      "epoch": 4.201889038085938e-05,
      "grad_norm": 0.13023394346237183,
      "learning_rate": 6.3776393042206995e-06,
      "loss": 0.0587,
      "step": 44060
    },
    {
      "epoch": 4.201889038085938e-05,
      "model_forward_time": 1.7149441242218018,
      "step": 44060
    },
    {
      "dynamics_loss_avg": 0.049044248089194296,
      "epoch": 4.201889038085938e-05,
      "step": 44060
    },
    {
      "action_loss_avg": 0.0017622028215555474,
      "epoch": 4.201889038085938e-05,
      "step": 44060
    },
    {
      "epoch": 4.201889038085938e-05,
      "step": 44060,
      "training_step_time": 7.226464033126831
    },
    {
      "epoch": 4.201984405517578e-05,
      "model_forward_time": 1.7160685062408447,
      "step": 44061
    },
    {
      "epoch": 4.201984405517578e-05,
      "step": 44061,
      "training_step_time": 7.358339309692383
    },
    {
      "epoch": 4.2020797729492185e-05,
      "model_forward_time": 1.72031831741333,
      "step": 44062
    },
    {
      "epoch": 4.2020797729492185e-05,
      "step": 44062,
      "training_step_time": 8.231741905212402
    },
    {
      "epoch": 4.2021751403808596e-05,
      "model_forward_time": 1.717128038406372,
      "step": 44063
    },
    {
      "epoch": 4.2021751403808596e-05,
      "step": 44063,
      "training_step_time": 8.412512302398682
    },
    {
      "epoch": 4.2022705078125e-05,
      "model_forward_time": 1.7151086330413818,
      "step": 44064
    },
    {
      "epoch": 4.2022705078125e-05,
      "step": 44064,
      "training_step_time": 9.106611251831055
    },
    {
      "epoch": 4.2023658752441404e-05,
      "model_forward_time": 1.7614126205444336,
      "step": 44065
    },
    {
      "epoch": 4.2023658752441404e-05,
      "step": 44065,
      "training_step_time": 9.852179765701294
    },
    {
      "epoch": 4.2024612426757815e-05,
      "model_forward_time": 1.7167439460754395,
      "step": 44066
    },
    {
      "epoch": 4.2024612426757815e-05,
      "step": 44066,
      "training_step_time": 7.7458977699279785
    },
    {
      "epoch": 4.202556610107422e-05,
      "model_forward_time": 1.7174279689788818,
      "step": 44067
    },
    {
      "epoch": 4.202556610107422e-05,
      "step": 44067,
      "training_step_time": 7.849225044250488
    },
    {
      "epoch": 4.202651977539062e-05,
      "model_forward_time": 1.7203404903411865,
      "step": 44068
    },
    {
      "epoch": 4.202651977539062e-05,
      "step": 44068,
      "training_step_time": 7.499510288238525
    },
    {
      "epoch": 4.2027473449707034e-05,
      "model_forward_time": 1.7161662578582764,
      "step": 44069
    },
    {
      "epoch": 4.2027473449707034e-05,
      "step": 44069,
      "training_step_time": 7.12362003326416
    },
    {
      "epoch": 4.202842712402344e-05,
      "grad_norm": 0.1613592952489853,
      "learning_rate": 6.376049759817136e-06,
      "loss": 0.057,
      "step": 44070
    },
    {
      "epoch": 4.202842712402344e-05,
      "model_forward_time": 1.7232708930969238,
      "step": 44070
    },
    {
      "dynamics_loss_avg": 0.07141951005905867,
      "epoch": 4.202842712402344e-05,
      "step": 44070
    },
    {
      "action_loss_avg": 0.002883634844329208,
      "epoch": 4.202842712402344e-05,
      "step": 44070
    },
    {
      "epoch": 4.202842712402344e-05,
      "step": 44070,
      "training_step_time": 7.264315605163574
    },
    {
      "epoch": 4.202938079833984e-05,
      "model_forward_time": 1.7179479598999023,
      "step": 44071
    },
    {
      "epoch": 4.202938079833984e-05,
      "step": 44071,
      "training_step_time": 7.728604316711426
    },
    {
      "epoch": 4.203033447265625e-05,
      "model_forward_time": 2.2419397830963135,
      "step": 44072
    },
    {
      "epoch": 4.203033447265625e-05,
      "step": 44072,
      "training_step_time": 7.700812101364136
    },
    {
      "epoch": 4.203128814697266e-05,
      "model_forward_time": 1.7232985496520996,
      "step": 44073
    },
    {
      "epoch": 4.203128814697266e-05,
      "step": 44073,
      "training_step_time": 8.8179612159729
    },
    {
      "epoch": 4.203224182128906e-05,
      "model_forward_time": 1.7603111267089844,
      "step": 44074
    },
    {
      "epoch": 4.203224182128906e-05,
      "step": 44074,
      "training_step_time": 8.617907285690308
    },
    {
      "epoch": 4.203319549560547e-05,
      "model_forward_time": 1.7055351734161377,
      "step": 44075
    },
    {
      "epoch": 4.203319549560547e-05,
      "step": 44075,
      "training_step_time": 9.151096105575562
    },
    {
      "epoch": 4.2034149169921875e-05,
      "model_forward_time": 1.7002346515655518,
      "step": 44076
    },
    {
      "epoch": 4.2034149169921875e-05,
      "step": 44076,
      "training_step_time": 7.760757207870483
    },
    {
      "epoch": 4.203510284423828e-05,
      "model_forward_time": 1.6974596977233887,
      "step": 44077
    },
    {
      "epoch": 4.203510284423828e-05,
      "step": 44077,
      "training_step_time": 7.562185287475586
    },
    {
      "epoch": 4.203605651855469e-05,
      "model_forward_time": 1.701472520828247,
      "step": 44078
    },
    {
      "epoch": 4.203605651855469e-05,
      "step": 44078,
      "training_step_time": 7.114431381225586
    },
    {
      "epoch": 4.2037010192871094e-05,
      "model_forward_time": 1.6976075172424316,
      "step": 44079
    },
    {
      "epoch": 4.2037010192871094e-05,
      "step": 44079,
      "training_step_time": 7.295104503631592
    },
    {
      "epoch": 4.20379638671875e-05,
      "grad_norm": 0.12908421456813812,
      "learning_rate": 6.374460064930841e-06,
      "loss": 0.0574,
      "step": 44080
    },
    {
      "epoch": 4.20379638671875e-05,
      "model_forward_time": 1.6988115310668945,
      "step": 44080
    },
    {
      "dynamics_loss_avg": 0.05612021945416927,
      "epoch": 4.20379638671875e-05,
      "step": 44080
    },
    {
      "action_loss_avg": 0.0019038034486584365,
      "epoch": 4.20379638671875e-05,
      "step": 44080
    },
    {
      "epoch": 4.20379638671875e-05,
      "step": 44080,
      "training_step_time": 7.795938968658447
    },
    {
      "epoch": 4.203891754150391e-05,
      "model_forward_time": 2.2659358978271484,
      "step": 44081
    },
    {
      "epoch": 4.203891754150391e-05,
      "step": 44081,
      "training_step_time": 8.542586088180542
    },
    {
      "epoch": 4.203987121582031e-05,
      "model_forward_time": 1.7197303771972656,
      "step": 44082
    },
    {
      "epoch": 4.203987121582031e-05,
      "step": 44082,
      "training_step_time": 7.762418270111084
    },
    {
      "epoch": 4.204082489013672e-05,
      "model_forward_time": 1.7015409469604492,
      "step": 44083
    },
    {
      "epoch": 4.204082489013672e-05,
      "step": 44083,
      "training_step_time": 9.255329847335815
    },
    {
      "epoch": 4.204177856445313e-05,
      "model_forward_time": 1.6971113681793213,
      "step": 44084
    },
    {
      "epoch": 4.204177856445313e-05,
      "step": 44084,
      "training_step_time": 10.12315559387207
    },
    {
      "epoch": 4.204273223876953e-05,
      "model_forward_time": 1.702195405960083,
      "step": 44085
    },
    {
      "epoch": 4.204273223876953e-05,
      "step": 44085,
      "training_step_time": 8.564821004867554
    },
    {
      "epoch": 4.2043685913085936e-05,
      "model_forward_time": 1.700319766998291,
      "step": 44086
    },
    {
      "epoch": 4.2043685913085936e-05,
      "step": 44086,
      "training_step_time": 7.699863433837891
    },
    {
      "epoch": 4.204463958740235e-05,
      "model_forward_time": 1.7069542407989502,
      "step": 44087
    },
    {
      "epoch": 4.204463958740235e-05,
      "step": 44087,
      "training_step_time": 7.11480450630188
    },
    {
      "epoch": 4.204559326171875e-05,
      "model_forward_time": 1.725952386856079,
      "step": 44088
    },
    {
      "epoch": 4.204559326171875e-05,
      "step": 44088,
      "training_step_time": 7.286579132080078
    },
    {
      "epoch": 4.2046546936035155e-05,
      "model_forward_time": 1.721480369567871,
      "step": 44089
    },
    {
      "epoch": 4.2046546936035155e-05,
      "step": 44089,
      "training_step_time": 7.6817591190338135
    },
    {
      "epoch": 4.2047500610351566e-05,
      "grad_norm": 0.14007537066936493,
      "learning_rate": 6.372870219735656e-06,
      "loss": 0.06,
      "step": 44090
    },
    {
      "epoch": 4.2047500610351566e-05,
      "model_forward_time": 2.27636981010437,
      "step": 44090
    },
    {
      "dynamics_loss_avg": 0.056954979337751864,
      "epoch": 4.2047500610351566e-05,
      "step": 44090
    },
    {
      "action_loss_avg": 0.0017440015566535295,
      "epoch": 4.2047500610351566e-05,
      "step": 44090
    },
    {
      "epoch": 4.2047500610351566e-05,
      "step": 44090,
      "training_step_time": 8.211992263793945
    },
    {
      "epoch": 4.204845428466797e-05,
      "model_forward_time": 1.770789384841919,
      "step": 44091
    },
    {
      "epoch": 4.204845428466797e-05,
      "step": 44091,
      "training_step_time": 8.620464324951172
    },
    {
      "epoch": 4.2049407958984374e-05,
      "model_forward_time": 1.721400260925293,
      "step": 44092
    },
    {
      "epoch": 4.2049407958984374e-05,
      "step": 44092,
      "training_step_time": 9.26647400856018
    },
    {
      "epoch": 4.2050361633300785e-05,
      "model_forward_time": 1.7192726135253906,
      "step": 44093
    },
    {
      "epoch": 4.2050361633300785e-05,
      "step": 44093,
      "training_step_time": 8.623680353164673
    },
    {
      "epoch": 4.205131530761719e-05,
      "model_forward_time": 1.7145614624023438,
      "step": 44094
    },
    {
      "epoch": 4.205131530761719e-05,
      "step": 44094,
      "training_step_time": 7.664474248886108
    },
    {
      "epoch": 4.205226898193359e-05,
      "model_forward_time": 1.7283310890197754,
      "step": 44095
    },
    {
      "epoch": 4.205226898193359e-05,
      "step": 44095,
      "training_step_time": 8.600683450698853
    },
    {
      "epoch": 4.205322265625e-05,
      "model_forward_time": 1.725642204284668,
      "step": 44096
    },
    {
      "epoch": 4.205322265625e-05,
      "step": 44096,
      "training_step_time": 6.826273441314697
    },
    {
      "epoch": 4.205417633056641e-05,
      "model_forward_time": 1.7234413623809814,
      "step": 44097
    },
    {
      "epoch": 4.205417633056641e-05,
      "step": 44097,
      "training_step_time": 7.348968505859375
    },
    {
      "epoch": 4.205513000488281e-05,
      "model_forward_time": 1.7331688404083252,
      "step": 44098
    },
    {
      "epoch": 4.205513000488281e-05,
      "step": 44098,
      "training_step_time": 7.64983868598938
    },
    {
      "epoch": 4.2056083679199216e-05,
      "model_forward_time": 2.2990562915802,
      "step": 44099
    },
    {
      "epoch": 4.2056083679199216e-05,
      "step": 44099,
      "training_step_time": 7.665023326873779
    },
    {
      "epoch": 4.2057037353515626e-05,
      "grad_norm": 0.18629950284957886,
      "learning_rate": 6.3712802244054515e-06,
      "loss": 0.0609,
      "step": 44100
    },
    {
      "epoch": 4.2057037353515626e-05,
      "model_forward_time": 1.7396700382232666,
      "step": 44100
    },
    {
      "dynamics_loss_avg": 0.05594400875270367,
      "epoch": 4.2057037353515626e-05,
      "step": 44100
    },
    {
      "action_loss_avg": 0.0021238634595647453,
      "epoch": 4.2057037353515626e-05,
      "step": 44100
    },
    {
      "epoch": 4.2057037353515626e-05,
      "step": 44100,
      "training_step_time": 7.744902849197388
    },
    {
      "epoch": 4.205799102783203e-05,
      "model_forward_time": 1.7350842952728271,
      "step": 44101
    },
    {
      "epoch": 4.205799102783203e-05,
      "step": 44101,
      "training_step_time": 7.806887865066528
    },
    {
      "epoch": 4.2058944702148434e-05,
      "model_forward_time": 1.719477891921997,
      "step": 44102
    },
    {
      "epoch": 4.2058944702148434e-05,
      "step": 44102,
      "training_step_time": 9.784383058547974
    },
    {
      "epoch": 4.2059898376464845e-05,
      "model_forward_time": 1.7093524932861328,
      "step": 44103
    },
    {
      "epoch": 4.2059898376464845e-05,
      "step": 44103,
      "training_step_time": 7.70476222038269
    },
    {
      "epoch": 4.206085205078125e-05,
      "model_forward_time": 1.3471364974975586,
      "step": 44104
    },
    {
      "epoch": 4.206085205078125e-05,
      "step": 44104,
      "training_step_time": 8.612484455108643
    },
    {
      "epoch": 4.206180572509765e-05,
      "model_forward_time": 1.7185108661651611,
      "step": 44105
    },
    {
      "epoch": 4.206180572509765e-05,
      "step": 44105,
      "training_step_time": 7.344227075576782
    },
    {
      "epoch": 4.2062759399414064e-05,
      "model_forward_time": 1.7338767051696777,
      "step": 44106
    },
    {
      "epoch": 4.2062759399414064e-05,
      "step": 44106,
      "training_step_time": 7.333043336868286
    },
    {
      "epoch": 4.206371307373047e-05,
      "model_forward_time": 1.7318205833435059,
      "step": 44107
    },
    {
      "epoch": 4.206371307373047e-05,
      "step": 44107,
      "training_step_time": 6.836665868759155
    },
    {
      "epoch": 4.206466674804687e-05,
      "model_forward_time": 1.717318058013916,
      "step": 44108
    },
    {
      "epoch": 4.206466674804687e-05,
      "step": 44108,
      "training_step_time": 7.761441707611084
    },
    {
      "epoch": 4.206562042236328e-05,
      "model_forward_time": 2.2704360485076904,
      "step": 44109
    },
    {
      "epoch": 4.206562042236328e-05,
      "step": 44109,
      "training_step_time": 7.743419885635376
    },
    {
      "epoch": 4.206657409667969e-05,
      "grad_norm": 0.1585087776184082,
      "learning_rate": 6.3696900791141016e-06,
      "loss": 0.0569,
      "step": 44110
    },
    {
      "epoch": 4.206657409667969e-05,
      "model_forward_time": 1.7209224700927734,
      "step": 44110
    },
    {
      "dynamics_loss_avg": 0.05646787267178297,
      "epoch": 4.206657409667969e-05,
      "step": 44110
    },
    {
      "action_loss_avg": 0.0029431794886477292,
      "epoch": 4.206657409667969e-05,
      "step": 44110
    },
    {
      "epoch": 4.206657409667969e-05,
      "step": 44110,
      "training_step_time": 8.72609543800354
    },
    {
      "epoch": 4.206752777099609e-05,
      "model_forward_time": 1.7229480743408203,
      "step": 44111
    },
    {
      "epoch": 4.206752777099609e-05,
      "step": 44111,
      "training_step_time": 7.619901180267334
    },
    {
      "epoch": 4.20684814453125e-05,
      "model_forward_time": 1.722780704498291,
      "step": 44112
    },
    {
      "epoch": 4.20684814453125e-05,
      "step": 44112,
      "training_step_time": 7.736173391342163
    },
    {
      "epoch": 4.2069435119628906e-05,
      "model_forward_time": 1.722649335861206,
      "step": 44113
    },
    {
      "epoch": 4.2069435119628906e-05,
      "step": 44113,
      "training_step_time": 7.569736957550049
    },
    {
      "epoch": 4.207038879394531e-05,
      "model_forward_time": 1.7257835865020752,
      "step": 44114
    },
    {
      "epoch": 4.207038879394531e-05,
      "step": 44114,
      "training_step_time": 7.682247638702393
    },
    {
      "epoch": 4.207134246826172e-05,
      "model_forward_time": 1.735002040863037,
      "step": 44115
    },
    {
      "epoch": 4.207134246826172e-05,
      "step": 44115,
      "training_step_time": 7.38558554649353
    },
    {
      "epoch": 4.2072296142578125e-05,
      "model_forward_time": 1.7271361351013184,
      "step": 44116
    },
    {
      "epoch": 4.2072296142578125e-05,
      "step": 44116,
      "training_step_time": 7.313551425933838
    },
    {
      "epoch": 4.207324981689453e-05,
      "model_forward_time": 1.7149925231933594,
      "step": 44117
    },
    {
      "epoch": 4.207324981689453e-05,
      "step": 44117,
      "training_step_time": 7.780488729476929
    },
    {
      "epoch": 4.207420349121094e-05,
      "model_forward_time": 2.262860059738159,
      "step": 44118
    },
    {
      "epoch": 4.207420349121094e-05,
      "step": 44118,
      "training_step_time": 7.742876052856445
    },
    {
      "epoch": 4.2075157165527344e-05,
      "model_forward_time": 1.7877271175384521,
      "step": 44119
    },
    {
      "epoch": 4.2075157165527344e-05,
      "step": 44119,
      "training_step_time": 7.798468828201294
    },
    {
      "epoch": 4.207611083984375e-05,
      "grad_norm": 0.16384588181972504,
      "learning_rate": 6.368099784035504e-06,
      "loss": 0.0576,
      "step": 44120
    },
    {
      "epoch": 4.207611083984375e-05,
      "model_forward_time": 1.71482515335083,
      "step": 44120
    },
    {
      "dynamics_loss_avg": 0.06730194874107838,
      "epoch": 4.207611083984375e-05,
      "step": 44120
    },
    {
      "action_loss_avg": 0.006589715729933232,
      "epoch": 4.207611083984375e-05,
      "step": 44120
    },
    {
      "epoch": 4.207611083984375e-05,
      "step": 44120,
      "training_step_time": 8.434599161148071
    },
    {
      "epoch": 4.207706451416016e-05,
      "model_forward_time": 1.7229588031768799,
      "step": 44121
    },
    {
      "epoch": 4.207706451416016e-05,
      "step": 44121,
      "training_step_time": 8.43835997581482
    },
    {
      "epoch": 4.207801818847656e-05,
      "model_forward_time": 1.7221708297729492,
      "step": 44122
    },
    {
      "epoch": 4.207801818847656e-05,
      "step": 44122,
      "training_step_time": 8.42729139328003
    },
    {
      "epoch": 4.2078971862792966e-05,
      "model_forward_time": 1.3489742279052734,
      "step": 44123
    },
    {
      "epoch": 4.2078971862792966e-05,
      "step": 44123,
      "training_step_time": 7.7809765338897705
    },
    {
      "epoch": 4.207992553710938e-05,
      "model_forward_time": 1.7487075328826904,
      "step": 44124
    },
    {
      "epoch": 4.207992553710938e-05,
      "step": 44124,
      "training_step_time": 7.3384459018707275
    },
    {
      "epoch": 4.208087921142578e-05,
      "model_forward_time": 1.7401552200317383,
      "step": 44125
    },
    {
      "epoch": 4.208087921142578e-05,
      "step": 44125,
      "training_step_time": 7.37147855758667
    },
    {
      "epoch": 4.2081832885742185e-05,
      "model_forward_time": 1.729017972946167,
      "step": 44126
    },
    {
      "epoch": 4.2081832885742185e-05,
      "step": 44126,
      "training_step_time": 7.726038694381714
    },
    {
      "epoch": 4.2082786560058596e-05,
      "model_forward_time": 2.2842071056365967,
      "step": 44127
    },
    {
      "epoch": 4.2082786560058596e-05,
      "step": 44127,
      "training_step_time": 7.68891978263855
    },
    {
      "epoch": 4.2083740234375e-05,
      "model_forward_time": 1.7805335521697998,
      "step": 44128
    },
    {
      "epoch": 4.2083740234375e-05,
      "step": 44128,
      "training_step_time": 9.39326024055481
    },
    {
      "epoch": 4.2084693908691404e-05,
      "model_forward_time": 1.7472507953643799,
      "step": 44129
    },
    {
      "epoch": 4.2084693908691404e-05,
      "step": 44129,
      "training_step_time": 7.7014007568359375
    },
    {
      "epoch": 4.2085647583007815e-05,
      "grad_norm": 0.12320204824209213,
      "learning_rate": 6.366509339343572e-06,
      "loss": 0.0523,
      "step": 44130
    },
    {
      "epoch": 4.2085647583007815e-05,
      "model_forward_time": 1.7442576885223389,
      "step": 44130
    },
    {
      "dynamics_loss_avg": 0.05423514693975449,
      "epoch": 4.2085647583007815e-05,
      "step": 44130
    },
    {
      "action_loss_avg": 0.002771014359313995,
      "epoch": 4.2085647583007815e-05,
      "step": 44130
    },
    {
      "epoch": 4.2085647583007815e-05,
      "step": 44130,
      "training_step_time": 7.679688453674316
    },
    {
      "epoch": 4.208660125732422e-05,
      "model_forward_time": 1.743269920349121,
      "step": 44131
    },
    {
      "epoch": 4.208660125732422e-05,
      "step": 44131,
      "training_step_time": 7.617916822433472
    },
    {
      "epoch": 4.208755493164062e-05,
      "model_forward_time": 1.7501351833343506,
      "step": 44132
    },
    {
      "epoch": 4.208755493164062e-05,
      "step": 44132,
      "training_step_time": 7.5708043575286865
    },
    {
      "epoch": 4.2088508605957034e-05,
      "model_forward_time": 1.7424054145812988,
      "step": 44133
    },
    {
      "epoch": 4.2088508605957034e-05,
      "step": 44133,
      "training_step_time": 7.173356056213379
    },
    {
      "epoch": 4.208946228027344e-05,
      "model_forward_time": 1.746614694595337,
      "step": 44134
    },
    {
      "epoch": 4.208946228027344e-05,
      "step": 44134,
      "training_step_time": 7.192220449447632
    },
    {
      "epoch": 4.209041595458984e-05,
      "model_forward_time": 1.7409780025482178,
      "step": 44135
    },
    {
      "epoch": 4.209041595458984e-05,
      "step": 44135,
      "training_step_time": 7.471782684326172
    },
    {
      "epoch": 4.209136962890625e-05,
      "model_forward_time": 2.309938430786133,
      "step": 44136
    },
    {
      "epoch": 4.209136962890625e-05,
      "step": 44136,
      "training_step_time": 7.62021541595459
    },
    {
      "epoch": 4.209232330322266e-05,
      "model_forward_time": 1.7693099975585938,
      "step": 44137
    },
    {
      "epoch": 4.209232330322266e-05,
      "step": 44137,
      "training_step_time": 8.119115829467773
    },
    {
      "epoch": 4.209327697753906e-05,
      "model_forward_time": 1.34340238571167,
      "step": 44138
    },
    {
      "epoch": 4.209327697753906e-05,
      "step": 44138,
      "training_step_time": 9.553974628448486
    },
    {
      "epoch": 4.209423065185547e-05,
      "model_forward_time": 1.7273738384246826,
      "step": 44139
    },
    {
      "epoch": 4.209423065185547e-05,
      "step": 44139,
      "training_step_time": 7.805400133132935
    },
    {
      "epoch": 4.2095184326171876e-05,
      "grad_norm": 0.1266506314277649,
      "learning_rate": 6.364918745212232e-06,
      "loss": 0.0572,
      "step": 44140
    },
    {
      "epoch": 4.2095184326171876e-05,
      "model_forward_time": 1.7285842895507812,
      "step": 44140
    },
    {
      "dynamics_loss_avg": 0.07289218073710799,
      "epoch": 4.2095184326171876e-05,
      "step": 44140
    },
    {
      "action_loss_avg": 0.003464825317496434,
      "epoch": 4.2095184326171876e-05,
      "step": 44140
    },
    {
      "epoch": 4.2095184326171876e-05,
      "step": 44140,
      "training_step_time": 8.713340282440186
    },
    {
      "epoch": 4.209613800048828e-05,
      "model_forward_time": 1.7476987838745117,
      "step": 44141
    },
    {
      "epoch": 4.209613800048828e-05,
      "step": 44141,
      "training_step_time": 8.483147859573364
    },
    {
      "epoch": 4.209709167480469e-05,
      "model_forward_time": 1.7392632961273193,
      "step": 44142
    },
    {
      "epoch": 4.209709167480469e-05,
      "step": 44142,
      "training_step_time": 7.430682182312012
    },
    {
      "epoch": 4.2098045349121094e-05,
      "model_forward_time": 1.7464618682861328,
      "step": 44143
    },
    {
      "epoch": 4.2098045349121094e-05,
      "step": 44143,
      "training_step_time": 7.349958658218384
    },
    {
      "epoch": 4.20989990234375e-05,
      "model_forward_time": 1.749091386795044,
      "step": 44144
    },
    {
      "epoch": 4.20989990234375e-05,
      "step": 44144,
      "training_step_time": 6.89215612411499
    },
    {
      "epoch": 4.209995269775391e-05,
      "model_forward_time": 2.28920316696167,
      "step": 44145
    },
    {
      "epoch": 4.209995269775391e-05,
      "step": 44145,
      "training_step_time": 8.986795663833618
    },
    {
      "epoch": 4.210090637207031e-05,
      "model_forward_time": 1.7589313983917236,
      "step": 44146
    },
    {
      "epoch": 4.210090637207031e-05,
      "step": 44146,
      "training_step_time": 7.8102805614471436
    },
    {
      "epoch": 4.210186004638672e-05,
      "model_forward_time": 1.7397291660308838,
      "step": 44147
    },
    {
      "epoch": 4.210186004638672e-05,
      "step": 44147,
      "training_step_time": 8.89555811882019
    },
    {
      "epoch": 4.210281372070313e-05,
      "model_forward_time": 1.742124319076538,
      "step": 44148
    },
    {
      "epoch": 4.210281372070313e-05,
      "step": 44148,
      "training_step_time": 7.854752063751221
    },
    {
      "epoch": 4.210376739501953e-05,
      "model_forward_time": 1.727832317352295,
      "step": 44149
    },
    {
      "epoch": 4.210376739501953e-05,
      "step": 44149,
      "training_step_time": 8.446950674057007
    },
    {
      "epoch": 4.2104721069335936e-05,
      "grad_norm": 0.1652536690235138,
      "learning_rate": 6.363328001815431e-06,
      "loss": 0.0563,
      "step": 44150
    },
    {
      "epoch": 4.2104721069335936e-05,
      "model_forward_time": 1.7398362159729004,
      "step": 44150
    },
    {
      "dynamics_loss_avg": 0.05902513638138771,
      "epoch": 4.2104721069335936e-05,
      "step": 44150
    },
    {
      "action_loss_avg": 0.0013076623785309494,
      "epoch": 4.2104721069335936e-05,
      "step": 44150
    },
    {
      "epoch": 4.2104721069335936e-05,
      "step": 44150,
      "training_step_time": 7.684394598007202
    },
    {
      "epoch": 4.210567474365235e-05,
      "model_forward_time": 1.7445893287658691,
      "step": 44151
    },
    {
      "epoch": 4.210567474365235e-05,
      "step": 44151,
      "training_step_time": 7.7512922286987305
    },
    {
      "epoch": 4.210662841796875e-05,
      "model_forward_time": 1.7532947063446045,
      "step": 44152
    },
    {
      "epoch": 4.210662841796875e-05,
      "step": 44152,
      "training_step_time": 7.351360321044922
    },
    {
      "epoch": 4.2107582092285155e-05,
      "model_forward_time": 1.7330811023712158,
      "step": 44153
    },
    {
      "epoch": 4.2107582092285155e-05,
      "step": 44153,
      "training_step_time": 6.842286825180054
    },
    {
      "epoch": 4.2108535766601566e-05,
      "model_forward_time": 1.7408628463745117,
      "step": 44154
    },
    {
      "epoch": 4.2108535766601566e-05,
      "step": 44154,
      "training_step_time": 7.7385094165802
    },
    {
      "epoch": 4.210948944091797e-05,
      "model_forward_time": 2.289299249649048,
      "step": 44155
    },
    {
      "epoch": 4.210948944091797e-05,
      "step": 44155,
      "training_step_time": 7.8305065631866455
    },
    {
      "epoch": 4.2110443115234374e-05,
      "model_forward_time": 1.8072705268859863,
      "step": 44156
    },
    {
      "epoch": 4.2110443115234374e-05,
      "step": 44156,
      "training_step_time": 7.802692413330078
    },
    {
      "epoch": 4.211139678955078e-05,
      "model_forward_time": 1.737940788269043,
      "step": 44157
    },
    {
      "epoch": 4.211139678955078e-05,
      "step": 44157,
      "training_step_time": 7.836160659790039
    },
    {
      "epoch": 4.211235046386719e-05,
      "model_forward_time": 1.7431244850158691,
      "step": 44158
    },
    {
      "epoch": 4.211235046386719e-05,
      "step": 44158,
      "training_step_time": 8.388237714767456
    },
    {
      "epoch": 4.211330413818359e-05,
      "model_forward_time": 1.7313644886016846,
      "step": 44159
    },
    {
      "epoch": 4.211330413818359e-05,
      "step": 44159,
      "training_step_time": 8.175073623657227
    },
    {
      "epoch": 4.21142578125e-05,
      "grad_norm": 0.19440098106861115,
      "learning_rate": 6.361737109327128e-06,
      "loss": 0.0559,
      "step": 44160
    },
    {
      "epoch": 4.21142578125e-05,
      "model_forward_time": 1.7342121601104736,
      "step": 44160
    },
    {
      "dynamics_loss_avg": 0.05079815667122602,
      "epoch": 4.21142578125e-05,
      "step": 44160
    },
    {
      "action_loss_avg": 0.0015458074340131134,
      "epoch": 4.21142578125e-05,
      "step": 44160
    },
    {
      "epoch": 4.21142578125e-05,
      "step": 44160,
      "training_step_time": 7.737859725952148
    },
    {
      "epoch": 4.211521148681641e-05,
      "model_forward_time": 1.7301604747772217,
      "step": 44161
    },
    {
      "epoch": 4.211521148681641e-05,
      "step": 44161,
      "training_step_time": 7.333754301071167
    },
    {
      "epoch": 4.211616516113281e-05,
      "model_forward_time": 1.7374019622802734,
      "step": 44162
    },
    {
      "epoch": 4.211616516113281e-05,
      "step": 44162,
      "training_step_time": 6.819206476211548
    },
    {
      "epoch": 4.2117118835449216e-05,
      "model_forward_time": 1.7364749908447266,
      "step": 44163
    },
    {
      "epoch": 4.2117118835449216e-05,
      "step": 44163,
      "training_step_time": 7.68224835395813
    },
    {
      "epoch": 4.2118072509765626e-05,
      "model_forward_time": 2.284229278564453,
      "step": 44164
    },
    {
      "epoch": 4.2118072509765626e-05,
      "step": 44164,
      "training_step_time": 8.285868883132935
    },
    {
      "epoch": 4.211902618408203e-05,
      "model_forward_time": 1.7998409271240234,
      "step": 44165
    },
    {
      "epoch": 4.211902618408203e-05,
      "step": 44165,
      "training_step_time": 10.225245475769043
    },
    {
      "epoch": 4.2119979858398434e-05,
      "model_forward_time": 1.7340102195739746,
      "step": 44166
    },
    {
      "epoch": 4.2119979858398434e-05,
      "step": 44166,
      "training_step_time": 7.835958480834961
    },
    {
      "epoch": 4.2120933532714845e-05,
      "model_forward_time": 1.7390146255493164,
      "step": 44167
    },
    {
      "epoch": 4.2120933532714845e-05,
      "step": 44167,
      "training_step_time": 8.820090770721436
    },
    {
      "epoch": 4.212188720703125e-05,
      "model_forward_time": 1.7371087074279785,
      "step": 44168
    },
    {
      "epoch": 4.212188720703125e-05,
      "step": 44168,
      "training_step_time": 7.627405881881714
    },
    {
      "epoch": 4.212284088134765e-05,
      "model_forward_time": 1.7348108291625977,
      "step": 44169
    },
    {
      "epoch": 4.212284088134765e-05,
      "step": 44169,
      "training_step_time": 8.478321075439453
    },
    {
      "epoch": 4.2123794555664064e-05,
      "grad_norm": 0.10120517760515213,
      "learning_rate": 6.360146067921302e-06,
      "loss": 0.0563,
      "step": 44170
    },
    {
      "epoch": 4.2123794555664064e-05,
      "model_forward_time": 1.7434604167938232,
      "step": 44170
    },
    {
      "dynamics_loss_avg": 0.05307023823261261,
      "epoch": 4.2123794555664064e-05,
      "step": 44170
    },
    {
      "action_loss_avg": 0.0022445323411375284,
      "epoch": 4.2123794555664064e-05,
      "step": 44170
    },
    {
      "epoch": 4.2123794555664064e-05,
      "step": 44170,
      "training_step_time": 7.318034887313843
    },
    {
      "epoch": 4.212474822998047e-05,
      "model_forward_time": 1.7109551429748535,
      "step": 44171
    },
    {
      "epoch": 4.212474822998047e-05,
      "step": 44171,
      "training_step_time": 7.210175514221191
    },
    {
      "epoch": 4.212570190429687e-05,
      "model_forward_time": 1.7380216121673584,
      "step": 44172
    },
    {
      "epoch": 4.212570190429687e-05,
      "step": 44172,
      "training_step_time": 7.653977394104004
    },
    {
      "epoch": 4.212665557861328e-05,
      "model_forward_time": 2.2907867431640625,
      "step": 44173
    },
    {
      "epoch": 4.212665557861328e-05,
      "step": 44173,
      "training_step_time": 9.365476846694946
    },
    {
      "epoch": 4.212760925292969e-05,
      "model_forward_time": 1.7876522541046143,
      "step": 44174
    },
    {
      "epoch": 4.212760925292969e-05,
      "step": 44174,
      "training_step_time": 8.682259559631348
    },
    {
      "epoch": 4.212856292724609e-05,
      "model_forward_time": 1.7386572360992432,
      "step": 44175
    },
    {
      "epoch": 4.212856292724609e-05,
      "step": 44175,
      "training_step_time": 9.660398006439209
    },
    {
      "epoch": 4.21295166015625e-05,
      "model_forward_time": 1.738673210144043,
      "step": 44176
    },
    {
      "epoch": 4.21295166015625e-05,
      "step": 44176,
      "training_step_time": 7.968993663787842
    },
    {
      "epoch": 4.2130470275878906e-05,
      "model_forward_time": 1.7183263301849365,
      "step": 44177
    },
    {
      "epoch": 4.2130470275878906e-05,
      "step": 44177,
      "training_step_time": 8.702449083328247
    },
    {
      "epoch": 4.213142395019531e-05,
      "model_forward_time": 1.7493696212768555,
      "step": 44178
    },
    {
      "epoch": 4.213142395019531e-05,
      "step": 44178,
      "training_step_time": 7.645173788070679
    },
    {
      "epoch": 4.213237762451172e-05,
      "model_forward_time": 1.7178435325622559,
      "step": 44179
    },
    {
      "epoch": 4.213237762451172e-05,
      "step": 44179,
      "training_step_time": 7.749791145324707
    },
    {
      "epoch": 4.2133331298828125e-05,
      "grad_norm": 0.11936430633068085,
      "learning_rate": 6.3585548777719495e-06,
      "loss": 0.0547,
      "step": 44180
    },
    {
      "epoch": 4.2133331298828125e-05,
      "model_forward_time": 1.7708053588867188,
      "step": 44180
    },
    {
      "dynamics_loss_avg": 0.07319029960781336,
      "epoch": 4.2133331298828125e-05,
      "step": 44180
    },
    {
      "action_loss_avg": 0.0025744146027136595,
      "epoch": 4.2133331298828125e-05,
      "step": 44180
    },
    {
      "epoch": 4.2133331298828125e-05,
      "step": 44180,
      "training_step_time": 7.382925271987915
    },
    {
      "epoch": 4.213428497314453e-05,
      "model_forward_time": 1.7300002574920654,
      "step": 44181
    },
    {
      "epoch": 4.213428497314453e-05,
      "step": 44181,
      "training_step_time": 7.46622371673584
    },
    {
      "epoch": 4.213523864746094e-05,
      "model_forward_time": 2.284618854522705,
      "step": 44182
    },
    {
      "epoch": 4.213523864746094e-05,
      "step": 44182,
      "training_step_time": 7.776608467102051
    },
    {
      "epoch": 4.2136192321777344e-05,
      "model_forward_time": 1.7441542148590088,
      "step": 44183
    },
    {
      "epoch": 4.2136192321777344e-05,
      "step": 44183,
      "training_step_time": 9.223823308944702
    },
    {
      "epoch": 4.213714599609375e-05,
      "model_forward_time": 1.7301483154296875,
      "step": 44184
    },
    {
      "epoch": 4.213714599609375e-05,
      "step": 44184,
      "training_step_time": 8.536089420318604
    },
    {
      "epoch": 4.213809967041016e-05,
      "model_forward_time": 1.7489120960235596,
      "step": 44185
    },
    {
      "epoch": 4.213809967041016e-05,
      "step": 44185,
      "training_step_time": 8.72637128829956
    },
    {
      "epoch": 4.213905334472656e-05,
      "model_forward_time": 1.7413349151611328,
      "step": 44186
    },
    {
      "epoch": 4.213905334472656e-05,
      "step": 44186,
      "training_step_time": 7.655392169952393
    },
    {
      "epoch": 4.2140007019042966e-05,
      "model_forward_time": 1.7355115413665771,
      "step": 44187
    },
    {
      "epoch": 4.2140007019042966e-05,
      "step": 44187,
      "training_step_time": 7.770009517669678
    },
    {
      "epoch": 4.214096069335938e-05,
      "model_forward_time": 1.743098497390747,
      "step": 44188
    },
    {
      "epoch": 4.214096069335938e-05,
      "step": 44188,
      "training_step_time": 7.739304542541504
    },
    {
      "epoch": 4.214191436767578e-05,
      "model_forward_time": 1.7492961883544922,
      "step": 44189
    },
    {
      "epoch": 4.214191436767578e-05,
      "step": 44189,
      "training_step_time": 7.349832057952881
    },
    {
      "epoch": 4.2142868041992185e-05,
      "grad_norm": 0.16727915406227112,
      "learning_rate": 6.356963539053076e-06,
      "loss": 0.0564,
      "step": 44190
    },
    {
      "epoch": 4.2142868041992185e-05,
      "model_forward_time": 1.733252763748169,
      "step": 44190
    },
    {
      "dynamics_loss_avg": 0.038014763686805966,
      "epoch": 4.2142868041992185e-05,
      "step": 44190
    },
    {
      "action_loss_avg": 0.002229288883972913,
      "epoch": 4.2142868041992185e-05,
      "step": 44190
    },
    {
      "epoch": 4.2142868041992185e-05,
      "step": 44190,
      "training_step_time": 8.644590377807617
    },
    {
      "epoch": 4.2143821716308596e-05,
      "model_forward_time": 2.3424339294433594,
      "step": 44191
    },
    {
      "epoch": 4.2143821716308596e-05,
      "step": 44191,
      "training_step_time": 7.754778146743774
    },
    {
      "epoch": 4.2144775390625e-05,
      "model_forward_time": 1.7299528121948242,
      "step": 44192
    },
    {
      "epoch": 4.2144775390625e-05,
      "step": 44192,
      "training_step_time": 7.807178974151611
    },
    {
      "epoch": 4.2145729064941404e-05,
      "model_forward_time": 1.7857282161712646,
      "step": 44193
    },
    {
      "epoch": 4.2145729064941404e-05,
      "step": 44193,
      "training_step_time": 9.626666069030762
    },
    {
      "epoch": 4.2146682739257815e-05,
      "model_forward_time": 1.727585792541504,
      "step": 44194
    },
    {
      "epoch": 4.2146682739257815e-05,
      "step": 44194,
      "training_step_time": 7.878463506698608
    },
    {
      "epoch": 4.214763641357422e-05,
      "model_forward_time": 1.743457555770874,
      "step": 44195
    },
    {
      "epoch": 4.214763641357422e-05,
      "step": 44195,
      "training_step_time": 7.687216758728027
    },
    {
      "epoch": 4.214859008789062e-05,
      "model_forward_time": 1.736929178237915,
      "step": 44196
    },
    {
      "epoch": 4.214859008789062e-05,
      "step": 44196,
      "training_step_time": 8.063535690307617
    },
    {
      "epoch": 4.2149543762207034e-05,
      "model_forward_time": 1.7373595237731934,
      "step": 44197
    },
    {
      "epoch": 4.2149543762207034e-05,
      "step": 44197,
      "training_step_time": 7.542001008987427
    },
    {
      "epoch": 4.215049743652344e-05,
      "model_forward_time": 1.7436726093292236,
      "step": 44198
    },
    {
      "epoch": 4.215049743652344e-05,
      "step": 44198,
      "training_step_time": 7.240924119949341
    },
    {
      "epoch": 4.215145111083984e-05,
      "model_forward_time": 1.737609624862671,
      "step": 44199
    },
    {
      "epoch": 4.215145111083984e-05,
      "step": 44199,
      "training_step_time": 7.514877080917358
    },
    {
      "epoch": 4.215240478515625e-05,
      "grad_norm": 0.13604597747325897,
      "learning_rate": 6.35537205193871e-06,
      "loss": 0.0556,
      "step": 44200
    },
    {
      "epoch": 4.215240478515625e-05,
      "model_forward_time": 2.281834125518799,
      "step": 44200
    },
    {
      "dynamics_loss_avg": 0.04666252247989178,
      "epoch": 4.215240478515625e-05,
      "step": 44200
    },
    {
      "action_loss_avg": 0.0019107578642433508,
      "epoch": 4.215240478515625e-05,
      "step": 44200
    },
    {
      "epoch": 4.215240478515625e-05,
      "step": 44200,
      "training_step_time": 8.21272325515747
    },
    {
      "epoch": 4.215335845947266e-05,
      "model_forward_time": 1.7375531196594238,
      "step": 44201
    },
    {
      "epoch": 4.215335845947266e-05,
      "step": 44201,
      "training_step_time": 8.369881868362427
    },
    {
      "epoch": 4.215431213378906e-05,
      "model_forward_time": 1.756035566329956,
      "step": 44202
    },
    {
      "epoch": 4.215431213378906e-05,
      "step": 44202,
      "training_step_time": 9.892003774642944
    },
    {
      "epoch": 4.215526580810547e-05,
      "model_forward_time": 1.7379169464111328,
      "step": 44203
    },
    {
      "epoch": 4.215526580810547e-05,
      "step": 44203,
      "training_step_time": 7.766279935836792
    },
    {
      "epoch": 4.2156219482421876e-05,
      "model_forward_time": 1.7353458404541016,
      "step": 44204
    },
    {
      "epoch": 4.2156219482421876e-05,
      "step": 44204,
      "training_step_time": 7.764010906219482
    },
    {
      "epoch": 4.215717315673828e-05,
      "model_forward_time": 1.73380708694458,
      "step": 44205
    },
    {
      "epoch": 4.215717315673828e-05,
      "step": 44205,
      "training_step_time": 7.583037853240967
    },
    {
      "epoch": 4.215812683105469e-05,
      "model_forward_time": 1.750058650970459,
      "step": 44206
    },
    {
      "epoch": 4.215812683105469e-05,
      "step": 44206,
      "training_step_time": 7.736823081970215
    },
    {
      "epoch": 4.2159080505371094e-05,
      "model_forward_time": 1.736480951309204,
      "step": 44207
    },
    {
      "epoch": 4.2159080505371094e-05,
      "step": 44207,
      "training_step_time": 7.598177194595337
    },
    {
      "epoch": 4.21600341796875e-05,
      "model_forward_time": 1.7375171184539795,
      "step": 44208
    },
    {
      "epoch": 4.21600341796875e-05,
      "step": 44208,
      "training_step_time": 7.532848834991455
    },
    {
      "epoch": 4.216098785400391e-05,
      "model_forward_time": 2.2989280223846436,
      "step": 44209
    },
    {
      "epoch": 4.216098785400391e-05,
      "step": 44209,
      "training_step_time": 7.709969758987427
    },
    {
      "epoch": 4.216194152832031e-05,
      "grad_norm": 0.09801334142684937,
      "learning_rate": 6.3537804166028945e-06,
      "loss": 0.0578,
      "step": 44210
    },
    {
      "epoch": 4.216194152832031e-05,
      "model_forward_time": 1.731368064880371,
      "step": 44210
    },
    {
      "dynamics_loss_avg": 0.05744277853518724,
      "epoch": 4.216194152832031e-05,
      "step": 44210
    },
    {
      "action_loss_avg": 0.00161424771649763,
      "epoch": 4.216194152832031e-05,
      "step": 44210
    },
    {
      "epoch": 4.216194152832031e-05,
      "step": 44210,
      "training_step_time": 7.759389162063599
    },
    {
      "epoch": 4.216289520263672e-05,
      "model_forward_time": 1.7320871353149414,
      "step": 44211
    },
    {
      "epoch": 4.216289520263672e-05,
      "step": 44211,
      "training_step_time": 7.636990547180176
    },
    {
      "epoch": 4.216384887695313e-05,
      "model_forward_time": 1.7392051219940186,
      "step": 44212
    },
    {
      "epoch": 4.216384887695313e-05,
      "step": 44212,
      "training_step_time": 9.924647331237793
    },
    {
      "epoch": 4.216480255126953e-05,
      "model_forward_time": 1.7425205707550049,
      "step": 44213
    },
    {
      "epoch": 4.216480255126953e-05,
      "step": 44213,
      "training_step_time": 7.583354473114014
    },
    {
      "epoch": 4.2165756225585936e-05,
      "model_forward_time": 1.7378723621368408,
      "step": 44214
    },
    {
      "epoch": 4.2165756225585936e-05,
      "step": 44214,
      "training_step_time": 7.536874532699585
    },
    {
      "epoch": 4.216670989990235e-05,
      "model_forward_time": 1.7328460216522217,
      "step": 44215
    },
    {
      "epoch": 4.216670989990235e-05,
      "step": 44215,
      "training_step_time": 7.612266302108765
    },
    {
      "epoch": 4.216766357421875e-05,
      "model_forward_time": 1.7393016815185547,
      "step": 44216
    },
    {
      "epoch": 4.216766357421875e-05,
      "step": 44216,
      "training_step_time": 7.531324148178101
    },
    {
      "epoch": 4.2168617248535155e-05,
      "model_forward_time": 1.7316370010375977,
      "step": 44217
    },
    {
      "epoch": 4.2168617248535155e-05,
      "step": 44217,
      "training_step_time": 7.452310562133789
    },
    {
      "epoch": 4.2169570922851566e-05,
      "model_forward_time": 2.2887117862701416,
      "step": 44218
    },
    {
      "epoch": 4.2169570922851566e-05,
      "step": 44218,
      "training_step_time": 7.613252401351929
    },
    {
      "epoch": 4.217052459716797e-05,
      "model_forward_time": 1.7420587539672852,
      "step": 44219
    },
    {
      "epoch": 4.217052459716797e-05,
      "step": 44219,
      "training_step_time": 8.748880386352539
    },
    {
      "epoch": 4.2171478271484374e-05,
      "grad_norm": 0.18491195142269135,
      "learning_rate": 6.3521886332196895e-06,
      "loss": 0.0568,
      "step": 44220
    },
    {
      "epoch": 4.2171478271484374e-05,
      "model_forward_time": 1.727834939956665,
      "step": 44220
    },
    {
      "dynamics_loss_avg": 0.06859936974942685,
      "epoch": 4.2171478271484374e-05,
      "step": 44220
    },
    {
      "action_loss_avg": 0.0026016416435595603,
      "epoch": 4.2171478271484374e-05,
      "step": 44220
    },
    {
      "epoch": 4.2171478271484374e-05,
      "step": 44220,
      "training_step_time": 8.432583093643188
    },
    {
      "epoch": 4.217243194580078e-05,
      "model_forward_time": 1.8193519115447998,
      "step": 44221
    },
    {
      "epoch": 4.217243194580078e-05,
      "step": 44221,
      "training_step_time": 7.690048694610596
    },
    {
      "epoch": 4.217338562011719e-05,
      "model_forward_time": 1.7312140464782715,
      "step": 44222
    },
    {
      "epoch": 4.217338562011719e-05,
      "step": 44222,
      "training_step_time": 8.375535249710083
    },
    {
      "epoch": 4.217433929443359e-05,
      "model_forward_time": 1.716742753982544,
      "step": 44223
    },
    {
      "epoch": 4.217433929443359e-05,
      "step": 44223,
      "training_step_time": 7.787431716918945
    },
    {
      "epoch": 4.217529296875e-05,
      "model_forward_time": 1.7173209190368652,
      "step": 44224
    },
    {
      "epoch": 4.217529296875e-05,
      "step": 44224,
      "training_step_time": 7.552180767059326
    },
    {
      "epoch": 4.217624664306641e-05,
      "model_forward_time": 1.721769094467163,
      "step": 44225
    },
    {
      "epoch": 4.217624664306641e-05,
      "step": 44225,
      "training_step_time": 7.5288166999816895
    },
    {
      "epoch": 4.217720031738281e-05,
      "model_forward_time": 1.7112088203430176,
      "step": 44226
    },
    {
      "epoch": 4.217720031738281e-05,
      "step": 44226,
      "training_step_time": 7.470452547073364
    },
    {
      "epoch": 4.2178153991699216e-05,
      "model_forward_time": 2.2780511379241943,
      "step": 44227
    },
    {
      "epoch": 4.2178153991699216e-05,
      "step": 44227,
      "training_step_time": 7.66852331161499
    },
    {
      "epoch": 4.2179107666015626e-05,
      "model_forward_time": 1.7143020629882812,
      "step": 44228
    },
    {
      "epoch": 4.2179107666015626e-05,
      "step": 44228,
      "training_step_time": 7.677702903747559
    },
    {
      "epoch": 4.218006134033203e-05,
      "model_forward_time": 1.7184078693389893,
      "step": 44229
    },
    {
      "epoch": 4.218006134033203e-05,
      "step": 44229,
      "training_step_time": 8.716334581375122
    },
    {
      "epoch": 4.2181015014648435e-05,
      "grad_norm": 0.1295897364616394,
      "learning_rate": 6.350596701963167e-06,
      "loss": 0.0584,
      "step": 44230
    },
    {
      "epoch": 4.2181015014648435e-05,
      "model_forward_time": 1.7392139434814453,
      "step": 44230
    },
    {
      "dynamics_loss_avg": 0.04955877037718892,
      "epoch": 4.2181015014648435e-05,
      "step": 44230
    },
    {
      "action_loss_avg": 0.0015208852797513827,
      "epoch": 4.2181015014648435e-05,
      "step": 44230
    },
    {
      "epoch": 4.2181015014648435e-05,
      "step": 44230,
      "training_step_time": 9.080922603607178
    },
    {
      "epoch": 4.2181968688964845e-05,
      "model_forward_time": 1.7364840507507324,
      "step": 44231
    },
    {
      "epoch": 4.2181968688964845e-05,
      "step": 44231,
      "training_step_time": 8.49157452583313
    },
    {
      "epoch": 4.218292236328125e-05,
      "model_forward_time": 1.7166602611541748,
      "step": 44232
    },
    {
      "epoch": 4.218292236328125e-05,
      "step": 44232,
      "training_step_time": 7.737239122390747
    },
    {
      "epoch": 4.218387603759765e-05,
      "model_forward_time": 1.7148566246032715,
      "step": 44233
    },
    {
      "epoch": 4.218387603759765e-05,
      "step": 44233,
      "training_step_time": 7.619430780410767
    },
    {
      "epoch": 4.2184829711914064e-05,
      "model_forward_time": 1.7220721244812012,
      "step": 44234
    },
    {
      "epoch": 4.2184829711914064e-05,
      "step": 44234,
      "training_step_time": 7.497762441635132
    },
    {
      "epoch": 4.218578338623047e-05,
      "model_forward_time": 1.7201414108276367,
      "step": 44235
    },
    {
      "epoch": 4.218578338623047e-05,
      "step": 44235,
      "training_step_time": 7.497621059417725
    },
    {
      "epoch": 4.218673706054687e-05,
      "model_forward_time": 2.268979787826538,
      "step": 44236
    },
    {
      "epoch": 4.218673706054687e-05,
      "step": 44236,
      "training_step_time": 7.086023330688477
    },
    {
      "epoch": 4.218769073486328e-05,
      "model_forward_time": 1.7195472717285156,
      "step": 44237
    },
    {
      "epoch": 4.218769073486328e-05,
      "step": 44237,
      "training_step_time": 7.714337348937988
    },
    {
      "epoch": 4.218864440917969e-05,
      "model_forward_time": 1.7186169624328613,
      "step": 44238
    },
    {
      "epoch": 4.218864440917969e-05,
      "step": 44238,
      "training_step_time": 8.722102642059326
    },
    {
      "epoch": 4.218959808349609e-05,
      "model_forward_time": 1.7278225421905518,
      "step": 44239
    },
    {
      "epoch": 4.218959808349609e-05,
      "step": 44239,
      "training_step_time": 9.285205364227295
    },
    {
      "epoch": 4.21905517578125e-05,
      "grad_norm": 0.14952309429645538,
      "learning_rate": 6.349004623007418e-06,
      "loss": 0.0588,
      "step": 44240
    },
    {
      "epoch": 4.21905517578125e-05,
      "model_forward_time": 1.719414234161377,
      "step": 44240
    },
    {
      "dynamics_loss_avg": 0.04800034705549479,
      "epoch": 4.21905517578125e-05,
      "step": 44240
    },
    {
      "action_loss_avg": 0.0017456146713811904,
      "epoch": 4.21905517578125e-05,
      "step": 44240
    },
    {
      "epoch": 4.21905517578125e-05,
      "step": 44240,
      "training_step_time": 7.710813760757446
    },
    {
      "epoch": 4.2191505432128906e-05,
      "model_forward_time": 1.7100462913513184,
      "step": 44241
    },
    {
      "epoch": 4.2191505432128906e-05,
      "step": 44241,
      "training_step_time": 8.655999422073364
    },
    {
      "epoch": 4.219245910644531e-05,
      "model_forward_time": 1.351029634475708,
      "step": 44242
    },
    {
      "epoch": 4.219245910644531e-05,
      "step": 44242,
      "training_step_time": 8.28344202041626
    },
    {
      "epoch": 4.219341278076172e-05,
      "model_forward_time": 1.7134578227996826,
      "step": 44243
    },
    {
      "epoch": 4.219341278076172e-05,
      "step": 44243,
      "training_step_time": 8.613526821136475
    },
    {
      "epoch": 4.2194366455078125e-05,
      "model_forward_time": 1.724491834640503,
      "step": 44244
    },
    {
      "epoch": 4.2194366455078125e-05,
      "step": 44244,
      "training_step_time": 7.502112150192261
    },
    {
      "epoch": 4.219532012939453e-05,
      "model_forward_time": 1.873981237411499,
      "step": 44245
    },
    {
      "epoch": 4.219532012939453e-05,
      "step": 44245,
      "training_step_time": 7.092523813247681
    },
    {
      "epoch": 4.219627380371094e-05,
      "model_forward_time": 1.7226407527923584,
      "step": 44246
    },
    {
      "epoch": 4.219627380371094e-05,
      "step": 44246,
      "training_step_time": 8.359365463256836
    },
    {
      "epoch": 4.2197227478027344e-05,
      "model_forward_time": 1.7186405658721924,
      "step": 44247
    },
    {
      "epoch": 4.2197227478027344e-05,
      "step": 44247,
      "training_step_time": 9.392073631286621
    },
    {
      "epoch": 4.219818115234375e-05,
      "model_forward_time": 1.7174978256225586,
      "step": 44248
    },
    {
      "epoch": 4.219818115234375e-05,
      "step": 44248,
      "training_step_time": 8.251402378082275
    },
    {
      "epoch": 4.219913482666016e-05,
      "model_forward_time": 1.7182395458221436,
      "step": 44249
    },
    {
      "epoch": 4.219913482666016e-05,
      "step": 44249,
      "training_step_time": 8.274576663970947
    },
    {
      "epoch": 4.220008850097656e-05,
      "grad_norm": 0.11248091608285904,
      "learning_rate": 6.3474123965265556e-06,
      "loss": 0.0547,
      "step": 44250
    },
    {
      "epoch": 4.220008850097656e-05,
      "model_forward_time": 1.7132458686828613,
      "step": 44250
    },
    {
      "dynamics_loss_avg": 0.054005602933466436,
      "epoch": 4.220008850097656e-05,
      "step": 44250
    },
    {
      "action_loss_avg": 0.0016203498875256629,
      "epoch": 4.220008850097656e-05,
      "step": 44250
    },
    {
      "epoch": 4.220008850097656e-05,
      "step": 44250,
      "training_step_time": 7.699207782745361
    },
    {
      "epoch": 4.2201042175292967e-05,
      "model_forward_time": 1.7170524597167969,
      "step": 44251
    },
    {
      "epoch": 4.2201042175292967e-05,
      "step": 44251,
      "training_step_time": 7.535926103591919
    },
    {
      "epoch": 4.220199584960938e-05,
      "model_forward_time": 1.719224214553833,
      "step": 44252
    },
    {
      "epoch": 4.220199584960938e-05,
      "step": 44252,
      "training_step_time": 7.668353796005249
    },
    {
      "epoch": 4.220294952392578e-05,
      "model_forward_time": 1.720611572265625,
      "step": 44253
    },
    {
      "epoch": 4.220294952392578e-05,
      "step": 44253,
      "training_step_time": 7.5423643589019775
    },
    {
      "epoch": 4.2203903198242185e-05,
      "model_forward_time": 1.7134068012237549,
      "step": 44254
    },
    {
      "epoch": 4.2203903198242185e-05,
      "step": 44254,
      "training_step_time": 7.0541558265686035
    },
    {
      "epoch": 4.2204856872558596e-05,
      "model_forward_time": 2.260165214538574,
      "step": 44255
    },
    {
      "epoch": 4.2204856872558596e-05,
      "step": 44255,
      "training_step_time": 7.656800985336304
    },
    {
      "epoch": 4.2205810546875e-05,
      "model_forward_time": 1.712200403213501,
      "step": 44256
    },
    {
      "epoch": 4.2205810546875e-05,
      "step": 44256,
      "training_step_time": 7.564882278442383
    },
    {
      "epoch": 4.2206764221191404e-05,
      "model_forward_time": 1.7148544788360596,
      "step": 44257
    },
    {
      "epoch": 4.2206764221191404e-05,
      "step": 44257,
      "training_step_time": 8.182886838912964
    },
    {
      "epoch": 4.2207717895507815e-05,
      "model_forward_time": 1.7119321823120117,
      "step": 44258
    },
    {
      "epoch": 4.2207717895507815e-05,
      "step": 44258,
      "training_step_time": 7.73940110206604
    },
    {
      "epoch": 4.220867156982422e-05,
      "model_forward_time": 1.7307682037353516,
      "step": 44259
    },
    {
      "epoch": 4.220867156982422e-05,
      "step": 44259,
      "training_step_time": 8.167409181594849
    },
    {
      "epoch": 4.220962524414062e-05,
      "grad_norm": 0.12575587630271912,
      "learning_rate": 6.345820022694696e-06,
      "loss": 0.0577,
      "step": 44260
    },
    {
      "epoch": 4.220962524414062e-05,
      "model_forward_time": 1.7148020267486572,
      "step": 44260
    },
    {
      "dynamics_loss_avg": 0.06265581594780087,
      "epoch": 4.220962524414062e-05,
      "step": 44260
    },
    {
      "action_loss_avg": 0.0016672245983500034,
      "epoch": 4.220962524414062e-05,
      "step": 44260
    },
    {
      "epoch": 4.220962524414062e-05,
      "step": 44260,
      "training_step_time": 7.537524223327637
    },
    {
      "epoch": 4.2210578918457034e-05,
      "model_forward_time": 1.3398313522338867,
      "step": 44261
    },
    {
      "epoch": 4.2210578918457034e-05,
      "step": 44261,
      "training_step_time": 7.555821418762207
    },
    {
      "epoch": 4.221153259277344e-05,
      "model_forward_time": 1.692413091659546,
      "step": 44262
    },
    {
      "epoch": 4.221153259277344e-05,
      "step": 44262,
      "training_step_time": 7.551808834075928
    },
    {
      "epoch": 4.221248626708984e-05,
      "model_forward_time": 1.7113268375396729,
      "step": 44263
    },
    {
      "epoch": 4.221248626708984e-05,
      "step": 44263,
      "training_step_time": 7.055233716964722
    },
    {
      "epoch": 4.221343994140625e-05,
      "model_forward_time": 2.2466185092926025,
      "step": 44264
    },
    {
      "epoch": 4.221343994140625e-05,
      "step": 44264,
      "training_step_time": 8.587539911270142
    },
    {
      "epoch": 4.221439361572266e-05,
      "model_forward_time": 1.702129602432251,
      "step": 44265
    },
    {
      "epoch": 4.221439361572266e-05,
      "step": 44265,
      "training_step_time": 7.695935964584351
    },
    {
      "epoch": 4.221534729003906e-05,
      "model_forward_time": 1.6992952823638916,
      "step": 44266
    },
    {
      "epoch": 4.221534729003906e-05,
      "step": 44266,
      "training_step_time": 8.257252216339111
    },
    {
      "epoch": 4.221630096435547e-05,
      "model_forward_time": 1.7182559967041016,
      "step": 44267
    },
    {
      "epoch": 4.221630096435547e-05,
      "step": 44267,
      "training_step_time": 10.487387657165527
    },
    {
      "epoch": 4.2217254638671876e-05,
      "model_forward_time": 1.7161436080932617,
      "step": 44268
    },
    {
      "epoch": 4.2217254638671876e-05,
      "step": 44268,
      "training_step_time": 8.691652297973633
    },
    {
      "epoch": 4.221820831298828e-05,
      "model_forward_time": 1.7026805877685547,
      "step": 44269
    },
    {
      "epoch": 4.221820831298828e-05,
      "step": 44269,
      "training_step_time": 7.638289451599121
    },
    {
      "epoch": 4.221916198730469e-05,
      "grad_norm": 0.13410864770412445,
      "learning_rate": 6.344227501685985e-06,
      "loss": 0.0571,
      "step": 44270
    },
    {
      "epoch": 4.221916198730469e-05,
      "model_forward_time": 1.6961843967437744,
      "step": 44270
    },
    {
      "dynamics_loss_avg": 0.043018007557839154,
      "epoch": 4.221916198730469e-05,
      "step": 44270
    },
    {
      "action_loss_avg": 0.0011852220457512885,
      "epoch": 4.221916198730469e-05,
      "step": 44270
    },
    {
      "epoch": 4.221916198730469e-05,
      "step": 44270,
      "training_step_time": 8.11028528213501
    },
    {
      "epoch": 4.2220115661621095e-05,
      "model_forward_time": 1.7056324481964111,
      "step": 44271
    },
    {
      "epoch": 4.2220115661621095e-05,
      "step": 44271,
      "training_step_time": 8.27980899810791
    },
    {
      "epoch": 4.22210693359375e-05,
      "model_forward_time": 1.7063066959381104,
      "step": 44272
    },
    {
      "epoch": 4.22210693359375e-05,
      "step": 44272,
      "training_step_time": 7.1771399974823
    },
    {
      "epoch": 4.222202301025391e-05,
      "model_forward_time": 2.2690279483795166,
      "step": 44273
    },
    {
      "epoch": 4.222202301025391e-05,
      "step": 44273,
      "training_step_time": 7.09976863861084
    },
    {
      "epoch": 4.222297668457031e-05,
      "model_forward_time": 1.7071003913879395,
      "step": 44274
    },
    {
      "epoch": 4.222297668457031e-05,
      "step": 44274,
      "training_step_time": 7.583715200424194
    },
    {
      "epoch": 4.222393035888672e-05,
      "model_forward_time": 1.7100098133087158,
      "step": 44275
    },
    {
      "epoch": 4.222393035888672e-05,
      "step": 44275,
      "training_step_time": 8.820659399032593
    },
    {
      "epoch": 4.222488403320313e-05,
      "model_forward_time": 1.7130742073059082,
      "step": 44276
    },
    {
      "epoch": 4.222488403320313e-05,
      "step": 44276,
      "training_step_time": 7.6415815353393555
    },
    {
      "epoch": 4.222583770751953e-05,
      "model_forward_time": 1.7159335613250732,
      "step": 44277
    },
    {
      "epoch": 4.222583770751953e-05,
      "step": 44277,
      "training_step_time": 7.759755849838257
    },
    {
      "epoch": 4.2226791381835936e-05,
      "model_forward_time": 1.7042911052703857,
      "step": 44278
    },
    {
      "epoch": 4.2226791381835936e-05,
      "step": 44278,
      "training_step_time": 7.602347135543823
    },
    {
      "epoch": 4.222774505615235e-05,
      "model_forward_time": 1.7019309997558594,
      "step": 44279
    },
    {
      "epoch": 4.222774505615235e-05,
      "step": 44279,
      "training_step_time": 7.456775903701782
    },
    {
      "epoch": 4.222869873046875e-05,
      "grad_norm": 0.1315550059080124,
      "learning_rate": 6.3426348336745725e-06,
      "loss": 0.0578,
      "step": 44280
    },
    {
      "epoch": 4.222869873046875e-05,
      "model_forward_time": 1.7031946182250977,
      "step": 44280
    },
    {
      "dynamics_loss_avg": 0.0725927047431469,
      "epoch": 4.222869873046875e-05,
      "step": 44280
    },
    {
      "action_loss_avg": 0.002875045966356993,
      "epoch": 4.222869873046875e-05,
      "step": 44280
    },
    {
      "epoch": 4.222869873046875e-05,
      "step": 44280,
      "training_step_time": 8.26535701751709
    },
    {
      "epoch": 4.2229652404785155e-05,
      "model_forward_time": 1.7044413089752197,
      "step": 44281
    },
    {
      "epoch": 4.2229652404785155e-05,
      "step": 44281,
      "training_step_time": 7.566733360290527
    },
    {
      "epoch": 4.223060607910156e-05,
      "model_forward_time": 2.282975912094116,
      "step": 44282
    },
    {
      "epoch": 4.223060607910156e-05,
      "step": 44282,
      "training_step_time": 8.118891477584839
    },
    {
      "epoch": 4.223155975341797e-05,
      "model_forward_time": 1.704681634902954,
      "step": 44283
    },
    {
      "epoch": 4.223155975341797e-05,
      "step": 44283,
      "training_step_time": 7.6426074504852295
    },
    {
      "epoch": 4.2232513427734374e-05,
      "model_forward_time": 1.7061595916748047,
      "step": 44284
    },
    {
      "epoch": 4.2232513427734374e-05,
      "step": 44284,
      "training_step_time": 7.675851106643677
    },
    {
      "epoch": 4.223346710205078e-05,
      "model_forward_time": 1.7126660346984863,
      "step": 44285
    },
    {
      "epoch": 4.223346710205078e-05,
      "step": 44285,
      "training_step_time": 7.664626598358154
    },
    {
      "epoch": 4.223442077636719e-05,
      "model_forward_time": 1.7160677909851074,
      "step": 44286
    },
    {
      "epoch": 4.223442077636719e-05,
      "step": 44286,
      "training_step_time": 8.207386016845703
    },
    {
      "epoch": 4.223537445068359e-05,
      "model_forward_time": 1.6963834762573242,
      "step": 44287
    },
    {
      "epoch": 4.223537445068359e-05,
      "step": 44287,
      "training_step_time": 7.578492164611816
    },
    {
      "epoch": 4.2236328125e-05,
      "model_forward_time": 1.6921093463897705,
      "step": 44288
    },
    {
      "epoch": 4.2236328125e-05,
      "step": 44288,
      "training_step_time": 7.517250299453735
    },
    {
      "epoch": 4.223728179931641e-05,
      "model_forward_time": 1.6972224712371826,
      "step": 44289
    },
    {
      "epoch": 4.223728179931641e-05,
      "step": 44289,
      "training_step_time": 7.442840814590454
    },
    {
      "epoch": 4.223823547363281e-05,
      "grad_norm": 0.16365107893943787,
      "learning_rate": 6.341042018834635e-06,
      "loss": 0.0601,
      "step": 44290
    },
    {
      "epoch": 4.223823547363281e-05,
      "model_forward_time": 1.7051165103912354,
      "step": 44290
    },
    {
      "dynamics_loss_avg": 0.0657091949135065,
      "epoch": 4.223823547363281e-05,
      "step": 44290
    },
    {
      "action_loss_avg": 0.002626942191272974,
      "epoch": 4.223823547363281e-05,
      "step": 44290
    },
    {
      "epoch": 4.223823547363281e-05,
      "step": 44290,
      "training_step_time": 8.562800168991089
    },
    {
      "epoch": 4.2239189147949216e-05,
      "model_forward_time": 1.701298713684082,
      "step": 44291
    },
    {
      "epoch": 4.2239189147949216e-05,
      "step": 44291,
      "training_step_time": 7.613287925720215
    },
    {
      "epoch": 4.2240142822265627e-05,
      "model_forward_time": 1.701709508895874,
      "step": 44292
    },
    {
      "epoch": 4.2240142822265627e-05,
      "step": 44292,
      "training_step_time": 8.635898113250732
    },
    {
      "epoch": 4.224109649658203e-05,
      "model_forward_time": 1.7039358615875244,
      "step": 44293
    },
    {
      "epoch": 4.224109649658203e-05,
      "step": 44293,
      "training_step_time": 8.78579592704773
    },
    {
      "epoch": 4.2242050170898435e-05,
      "model_forward_time": 1.6940536499023438,
      "step": 44294
    },
    {
      "epoch": 4.2242050170898435e-05,
      "step": 44294,
      "training_step_time": 8.644808769226074
    },
    {
      "epoch": 4.2243003845214845e-05,
      "model_forward_time": 1.7191412448883057,
      "step": 44295
    },
    {
      "epoch": 4.2243003845214845e-05,
      "step": 44295,
      "training_step_time": 8.813847303390503
    },
    {
      "epoch": 4.224395751953125e-05,
      "model_forward_time": 1.7011504173278809,
      "step": 44296
    },
    {
      "epoch": 4.224395751953125e-05,
      "step": 44296,
      "training_step_time": 8.593672275543213
    },
    {
      "epoch": 4.2244911193847653e-05,
      "model_forward_time": 1.7057414054870605,
      "step": 44297
    },
    {
      "epoch": 4.2244911193847653e-05,
      "step": 44297,
      "training_step_time": 7.503491640090942
    },
    {
      "epoch": 4.2245864868164064e-05,
      "model_forward_time": 1.702392339706421,
      "step": 44298
    },
    {
      "epoch": 4.2245864868164064e-05,
      "step": 44298,
      "training_step_time": 7.425261735916138
    },
    {
      "epoch": 4.224681854248047e-05,
      "model_forward_time": 1.7023870944976807,
      "step": 44299
    },
    {
      "epoch": 4.224681854248047e-05,
      "step": 44299,
      "training_step_time": 7.541778087615967
    },
    {
      "epoch": 4.224777221679687e-05,
      "grad_norm": 0.11662416160106659,
      "learning_rate": 6.339449057340359e-06,
      "loss": 0.0601,
      "step": 44300
    },
    {
      "epoch": 4.224777221679687e-05,
      "model_forward_time": 1.705124855041504,
      "step": 44300
    },
    {
      "dynamics_loss_avg": 0.05806728061288595,
      "epoch": 4.224777221679687e-05,
      "step": 44300
    },
    {
      "action_loss_avg": 0.0024219711020123215,
      "epoch": 4.224777221679687e-05,
      "step": 44300
    },
    {
      "epoch": 4.224777221679687e-05,
      "step": 44300,
      "training_step_time": 8.202888488769531
    },
    {
      "epoch": 4.224872589111328e-05,
      "model_forward_time": 2.2464778423309326,
      "step": 44301
    },
    {
      "epoch": 4.224872589111328e-05,
      "step": 44301,
      "training_step_time": 8.152527809143066
    },
    {
      "epoch": 4.224967956542969e-05,
      "model_forward_time": 1.6873149871826172,
      "step": 44302
    },
    {
      "epoch": 4.224967956542969e-05,
      "step": 44302,
      "training_step_time": 7.665274143218994
    },
    {
      "epoch": 4.225063323974609e-05,
      "model_forward_time": 1.6967437267303467,
      "step": 44303
    },
    {
      "epoch": 4.225063323974609e-05,
      "step": 44303,
      "training_step_time": 7.660078763961792
    },
    {
      "epoch": 4.22515869140625e-05,
      "model_forward_time": 1.7006909847259521,
      "step": 44304
    },
    {
      "epoch": 4.22515869140625e-05,
      "step": 44304,
      "training_step_time": 8.476374864578247
    },
    {
      "epoch": 4.2252540588378906e-05,
      "model_forward_time": 1.7139174938201904,
      "step": 44305
    },
    {
      "epoch": 4.2252540588378906e-05,
      "step": 44305,
      "training_step_time": 7.715016841888428
    },
    {
      "epoch": 4.225349426269531e-05,
      "model_forward_time": 1.6966016292572021,
      "step": 44306
    },
    {
      "epoch": 4.225349426269531e-05,
      "step": 44306,
      "training_step_time": 7.719573736190796
    },
    {
      "epoch": 4.225444793701172e-05,
      "model_forward_time": 1.702235460281372,
      "step": 44307
    },
    {
      "epoch": 4.225444793701172e-05,
      "step": 44307,
      "training_step_time": 7.452808380126953
    },
    {
      "epoch": 4.2255401611328125e-05,
      "model_forward_time": 1.7113914489746094,
      "step": 44308
    },
    {
      "epoch": 4.2255401611328125e-05,
      "step": 44308,
      "training_step_time": 7.700481176376343
    },
    {
      "epoch": 4.225635528564453e-05,
      "model_forward_time": 1.7080192565917969,
      "step": 44309
    },
    {
      "epoch": 4.225635528564453e-05,
      "step": 44309,
      "training_step_time": 7.502063274383545
    },
    {
      "epoch": 4.225730895996094e-05,
      "grad_norm": 0.13384148478507996,
      "learning_rate": 6.3378559493659454e-06,
      "loss": 0.0586,
      "step": 44310
    },
    {
      "epoch": 4.225730895996094e-05,
      "model_forward_time": 2.249423027038574,
      "step": 44310
    },
    {
      "dynamics_loss_avg": 0.0587932663038373,
      "epoch": 4.225730895996094e-05,
      "step": 44310
    },
    {
      "action_loss_avg": 0.0016554761095903814,
      "epoch": 4.225730895996094e-05,
      "step": 44310
    },
    {
      "epoch": 4.225730895996094e-05,
      "step": 44310,
      "training_step_time": 7.373434543609619
    },
    {
      "epoch": 4.2258262634277344e-05,
      "model_forward_time": 1.7024898529052734,
      "step": 44311
    },
    {
      "epoch": 4.2258262634277344e-05,
      "step": 44311,
      "training_step_time": 8.175530672073364
    },
    {
      "epoch": 4.225921630859375e-05,
      "model_forward_time": 1.6994154453277588,
      "step": 44312
    },
    {
      "epoch": 4.225921630859375e-05,
      "step": 44312,
      "training_step_time": 8.539337396621704
    },
    {
      "epoch": 4.226016998291016e-05,
      "model_forward_time": 1.7055583000183105,
      "step": 44313
    },
    {
      "epoch": 4.226016998291016e-05,
      "step": 44313,
      "training_step_time": 8.627724409103394
    },
    {
      "epoch": 4.226112365722656e-05,
      "model_forward_time": 1.7225384712219238,
      "step": 44314
    },
    {
      "epoch": 4.226112365722656e-05,
      "step": 44314,
      "training_step_time": 7.75501823425293
    },
    {
      "epoch": 4.2262077331542967e-05,
      "model_forward_time": 1.699448585510254,
      "step": 44315
    },
    {
      "epoch": 4.2262077331542967e-05,
      "step": 44315,
      "training_step_time": 9.60808515548706
    },
    {
      "epoch": 4.226303100585938e-05,
      "model_forward_time": 1.7054543495178223,
      "step": 44316
    },
    {
      "epoch": 4.226303100585938e-05,
      "step": 44316,
      "training_step_time": 8.102128267288208
    },
    {
      "epoch": 4.226398468017578e-05,
      "model_forward_time": 1.6985371112823486,
      "step": 44317
    },
    {
      "epoch": 4.226398468017578e-05,
      "step": 44317,
      "training_step_time": 7.584000110626221
    },
    {
      "epoch": 4.2264938354492185e-05,
      "model_forward_time": 1.6991095542907715,
      "step": 44318
    },
    {
      "epoch": 4.2264938354492185e-05,
      "step": 44318,
      "training_step_time": 7.565428972244263
    },
    {
      "epoch": 4.2265892028808596e-05,
      "model_forward_time": 2.2501680850982666,
      "step": 44319
    },
    {
      "epoch": 4.2265892028808596e-05,
      "step": 44319,
      "training_step_time": 7.9941723346710205
    },
    {
      "epoch": 4.2266845703125e-05,
      "grad_norm": 0.0823325589299202,
      "learning_rate": 6.336262695085619e-06,
      "loss": 0.0549,
      "step": 44320
    },
    {
      "epoch": 4.2266845703125e-05,
      "model_forward_time": 1.7027926445007324,
      "step": 44320
    },
    {
      "dynamics_loss_avg": 0.04301892658695579,
      "epoch": 4.2266845703125e-05,
      "step": 44320
    },
    {
      "action_loss_avg": 0.002601825160672888,
      "epoch": 4.2266845703125e-05,
      "step": 44320
    },
    {
      "epoch": 4.2266845703125e-05,
      "step": 44320,
      "training_step_time": 8.64194655418396
    },
    {
      "epoch": 4.2267799377441404e-05,
      "model_forward_time": 1.701237678527832,
      "step": 44321
    },
    {
      "epoch": 4.2267799377441404e-05,
      "step": 44321,
      "training_step_time": 8.948472499847412
    },
    {
      "epoch": 4.2268753051757815e-05,
      "model_forward_time": 1.7055761814117432,
      "step": 44322
    },
    {
      "epoch": 4.2268753051757815e-05,
      "step": 44322,
      "training_step_time": 10.440011501312256
    },
    {
      "epoch": 4.226970672607422e-05,
      "model_forward_time": 1.7325878143310547,
      "step": 44323
    },
    {
      "epoch": 4.226970672607422e-05,
      "step": 44323,
      "training_step_time": 8.745718955993652
    },
    {
      "epoch": 4.227066040039062e-05,
      "model_forward_time": 1.7014124393463135,
      "step": 44324
    },
    {
      "epoch": 4.227066040039062e-05,
      "step": 44324,
      "training_step_time": 7.63460636138916
    },
    {
      "epoch": 4.2271614074707034e-05,
      "model_forward_time": 1.700834035873413,
      "step": 44325
    },
    {
      "epoch": 4.2271614074707034e-05,
      "step": 44325,
      "training_step_time": 7.569400787353516
    },
    {
      "epoch": 4.227256774902344e-05,
      "model_forward_time": 1.7027003765106201,
      "step": 44326
    },
    {
      "epoch": 4.227256774902344e-05,
      "step": 44326,
      "training_step_time": 7.571933269500732
    },
    {
      "epoch": 4.227352142333984e-05,
      "model_forward_time": 1.707747459411621,
      "step": 44327
    },
    {
      "epoch": 4.227352142333984e-05,
      "step": 44327,
      "training_step_time": 7.638974905014038
    },
    {
      "epoch": 4.227447509765625e-05,
      "model_forward_time": 2.2450170516967773,
      "step": 44328
    },
    {
      "epoch": 4.227447509765625e-05,
      "step": 44328,
      "training_step_time": 7.37037467956543
    },
    {
      "epoch": 4.227542877197266e-05,
      "model_forward_time": 1.7062058448791504,
      "step": 44329
    },
    {
      "epoch": 4.227542877197266e-05,
      "step": 44329,
      "training_step_time": 8.37288212776184
    },
    {
      "epoch": 4.227638244628906e-05,
      "grad_norm": 0.11448914557695389,
      "learning_rate": 6.3346692946736134e-06,
      "loss": 0.0574,
      "step": 44330
    },
    {
      "epoch": 4.227638244628906e-05,
      "model_forward_time": 1.6932151317596436,
      "step": 44330
    },
    {
      "dynamics_loss_avg": 0.04737493833526969,
      "epoch": 4.227638244628906e-05,
      "step": 44330
    },
    {
      "action_loss_avg": 0.0014959681109758094,
      "epoch": 4.227638244628906e-05,
      "step": 44330
    },
    {
      "epoch": 4.227638244628906e-05,
      "step": 44330,
      "training_step_time": 8.73329496383667
    },
    {
      "epoch": 4.227733612060547e-05,
      "model_forward_time": 1.706425666809082,
      "step": 44331
    },
    {
      "epoch": 4.227733612060547e-05,
      "step": 44331,
      "training_step_time": 7.79577898979187
    },
    {
      "epoch": 4.2278289794921876e-05,
      "model_forward_time": 1.7210357189178467,
      "step": 44332
    },
    {
      "epoch": 4.2278289794921876e-05,
      "step": 44332,
      "training_step_time": 9.040115594863892
    },
    {
      "epoch": 4.227924346923828e-05,
      "model_forward_time": 1.7025952339172363,
      "step": 44333
    },
    {
      "epoch": 4.227924346923828e-05,
      "step": 44333,
      "training_step_time": 8.192292928695679
    },
    {
      "epoch": 4.228019714355469e-05,
      "model_forward_time": 1.7055602073669434,
      "step": 44334
    },
    {
      "epoch": 4.228019714355469e-05,
      "step": 44334,
      "training_step_time": 7.6668713092803955
    },
    {
      "epoch": 4.2281150817871095e-05,
      "model_forward_time": 1.7051937580108643,
      "step": 44335
    },
    {
      "epoch": 4.2281150817871095e-05,
      "step": 44335,
      "training_step_time": 7.4874773025512695
    },
    {
      "epoch": 4.22821044921875e-05,
      "model_forward_time": 1.6972565650939941,
      "step": 44336
    },
    {
      "epoch": 4.22821044921875e-05,
      "step": 44336,
      "training_step_time": 7.592635154724121
    },
    {
      "epoch": 4.228305816650391e-05,
      "model_forward_time": 2.2443268299102783,
      "step": 44337
    },
    {
      "epoch": 4.228305816650391e-05,
      "step": 44337,
      "training_step_time": 7.563706636428833
    },
    {
      "epoch": 4.2284011840820313e-05,
      "model_forward_time": 1.3201570510864258,
      "step": 44338
    },
    {
      "epoch": 4.2284011840820313e-05,
      "step": 44338,
      "training_step_time": 7.834769248962402
    },
    {
      "epoch": 4.228496551513672e-05,
      "model_forward_time": 1.697587013244629,
      "step": 44339
    },
    {
      "epoch": 4.228496551513672e-05,
      "step": 44339,
      "training_step_time": 7.937719345092773
    },
    {
      "epoch": 4.228591918945313e-05,
      "grad_norm": 0.07743266224861145,
      "learning_rate": 6.33307574830418e-06,
      "loss": 0.0571,
      "step": 44340
    },
    {
      "epoch": 4.228591918945313e-05,
      "model_forward_time": 1.6986935138702393,
      "step": 44340
    },
    {
      "dynamics_loss_avg": 0.0599726827815175,
      "epoch": 4.228591918945313e-05,
      "step": 44340
    },
    {
      "action_loss_avg": 0.0014959826221456751,
      "epoch": 4.228591918945313e-05,
      "step": 44340
    },
    {
      "epoch": 4.228591918945313e-05,
      "step": 44340,
      "training_step_time": 8.971046209335327
    },
    {
      "epoch": 4.228687286376953e-05,
      "model_forward_time": 1.7196412086486816,
      "step": 44341
    },
    {
      "epoch": 4.228687286376953e-05,
      "step": 44341,
      "training_step_time": 9.713639974594116
    },
    {
      "epoch": 4.2287826538085936e-05,
      "model_forward_time": 1.7155883312225342,
      "step": 44342
    },
    {
      "epoch": 4.2287826538085936e-05,
      "step": 44342,
      "training_step_time": 8.550430536270142
    },
    {
      "epoch": 4.228878021240235e-05,
      "model_forward_time": 1.7005438804626465,
      "step": 44343
    },
    {
      "epoch": 4.228878021240235e-05,
      "step": 44343,
      "training_step_time": 7.8333423137664795
    },
    {
      "epoch": 4.228973388671875e-05,
      "model_forward_time": 1.7049477100372314,
      "step": 44344
    },
    {
      "epoch": 4.228973388671875e-05,
      "step": 44344,
      "training_step_time": 7.70130181312561
    },
    {
      "epoch": 4.2290687561035155e-05,
      "model_forward_time": 1.7039320468902588,
      "step": 44345
    },
    {
      "epoch": 4.2290687561035155e-05,
      "step": 44345,
      "training_step_time": 7.819591522216797
    },
    {
      "epoch": 4.229164123535156e-05,
      "model_forward_time": 2.2553412914276123,
      "step": 44346
    },
    {
      "epoch": 4.229164123535156e-05,
      "step": 44346,
      "training_step_time": 7.744209051132202
    },
    {
      "epoch": 4.229259490966797e-05,
      "model_forward_time": 1.7038297653198242,
      "step": 44347
    },
    {
      "epoch": 4.229259490966797e-05,
      "step": 44347,
      "training_step_time": 7.848879098892212
    },
    {
      "epoch": 4.2293548583984374e-05,
      "model_forward_time": 1.6946251392364502,
      "step": 44348
    },
    {
      "epoch": 4.2293548583984374e-05,
      "step": 44348,
      "training_step_time": 7.70059609413147
    },
    {
      "epoch": 4.229450225830078e-05,
      "model_forward_time": 1.690971851348877,
      "step": 44349
    },
    {
      "epoch": 4.229450225830078e-05,
      "step": 44349,
      "training_step_time": 9.63629961013794
    },
    {
      "epoch": 4.229545593261719e-05,
      "grad_norm": 0.1207008883357048,
      "learning_rate": 6.3314820561515854e-06,
      "loss": 0.0568,
      "step": 44350
    },
    {
      "epoch": 4.229545593261719e-05,
      "model_forward_time": 1.702739953994751,
      "step": 44350
    },
    {
      "dynamics_loss_avg": 0.06942250058054925,
      "epoch": 4.229545593261719e-05,
      "step": 44350
    },
    {
      "action_loss_avg": 0.0023395997704938054,
      "epoch": 4.229545593261719e-05,
      "step": 44350
    },
    {
      "epoch": 4.229545593261719e-05,
      "step": 44350,
      "training_step_time": 7.852621555328369
    },
    {
      "epoch": 4.229640960693359e-05,
      "model_forward_time": 1.3264176845550537,
      "step": 44351
    },
    {
      "epoch": 4.229640960693359e-05,
      "step": 44351,
      "training_step_time": 7.7119221687316895
    },
    {
      "epoch": 4.229736328125e-05,
      "model_forward_time": 1.6934044361114502,
      "step": 44352
    },
    {
      "epoch": 4.229736328125e-05,
      "step": 44352,
      "training_step_time": 8.765320539474487
    },
    {
      "epoch": 4.229831695556641e-05,
      "model_forward_time": 1.7024805545806885,
      "step": 44353
    },
    {
      "epoch": 4.229831695556641e-05,
      "step": 44353,
      "training_step_time": 7.664571762084961
    },
    {
      "epoch": 4.229927062988281e-05,
      "model_forward_time": 1.6997795104980469,
      "step": 44354
    },
    {
      "epoch": 4.229927062988281e-05,
      "step": 44354,
      "training_step_time": 7.7006237506866455
    },
    {
      "epoch": 4.2300224304199216e-05,
      "model_forward_time": 2.2589476108551025,
      "step": 44355
    },
    {
      "epoch": 4.2300224304199216e-05,
      "step": 44355,
      "training_step_time": 7.63269567489624
    },
    {
      "epoch": 4.2301177978515627e-05,
      "model_forward_time": 1.693390130996704,
      "step": 44356
    },
    {
      "epoch": 4.2301177978515627e-05,
      "step": 44356,
      "training_step_time": 7.777470827102661
    },
    {
      "epoch": 4.230213165283203e-05,
      "model_forward_time": 1.694716215133667,
      "step": 44357
    },
    {
      "epoch": 4.230213165283203e-05,
      "step": 44357,
      "training_step_time": 8.305758953094482
    },
    {
      "epoch": 4.2303085327148435e-05,
      "model_forward_time": 1.696472406387329,
      "step": 44358
    },
    {
      "epoch": 4.2303085327148435e-05,
      "step": 44358,
      "training_step_time": 8.928248405456543
    },
    {
      "epoch": 4.2304039001464845e-05,
      "model_forward_time": 1.6995010375976562,
      "step": 44359
    },
    {
      "epoch": 4.2304039001464845e-05,
      "step": 44359,
      "training_step_time": 10.382379055023193
    },
    {
      "epoch": 4.230499267578125e-05,
      "grad_norm": 0.12370675802230835,
      "learning_rate": 6.329888218390117e-06,
      "loss": 0.0581,
      "step": 44360
    },
    {
      "epoch": 4.230499267578125e-05,
      "model_forward_time": 1.690950870513916,
      "step": 44360
    },
    {
      "dynamics_loss_avg": 0.05375123461708427,
      "epoch": 4.230499267578125e-05,
      "step": 44360
    },
    {
      "action_loss_avg": 0.0018061095848679542,
      "epoch": 4.230499267578125e-05,
      "step": 44360
    },
    {
      "epoch": 4.230499267578125e-05,
      "step": 44360,
      "training_step_time": 9.503661394119263
    },
    {
      "epoch": 4.2305946350097653e-05,
      "model_forward_time": 1.7035393714904785,
      "step": 44361
    },
    {
      "epoch": 4.2305946350097653e-05,
      "step": 44361,
      "training_step_time": 7.771653175354004
    },
    {
      "epoch": 4.2306900024414064e-05,
      "model_forward_time": 1.3192975521087646,
      "step": 44362
    },
    {
      "epoch": 4.2306900024414064e-05,
      "step": 44362,
      "training_step_time": 8.683112144470215
    },
    {
      "epoch": 4.230785369873047e-05,
      "model_forward_time": 1.696209192276001,
      "step": 44363
    },
    {
      "epoch": 4.230785369873047e-05,
      "step": 44363,
      "training_step_time": 7.710397958755493
    },
    {
      "epoch": 4.230880737304687e-05,
      "model_forward_time": 1.698807954788208,
      "step": 44364
    },
    {
      "epoch": 4.230880737304687e-05,
      "step": 44364,
      "training_step_time": 7.645724296569824
    },
    {
      "epoch": 4.230976104736328e-05,
      "model_forward_time": 1.698819637298584,
      "step": 44365
    },
    {
      "epoch": 4.230976104736328e-05,
      "step": 44365,
      "training_step_time": 7.82318639755249
    },
    {
      "epoch": 4.231071472167969e-05,
      "model_forward_time": 1.6981537342071533,
      "step": 44366
    },
    {
      "epoch": 4.231071472167969e-05,
      "step": 44366,
      "training_step_time": 9.206178665161133
    },
    {
      "epoch": 4.231166839599609e-05,
      "model_forward_time": 1.6986136436462402,
      "step": 44367
    },
    {
      "epoch": 4.231166839599609e-05,
      "step": 44367,
      "training_step_time": 7.857140064239502
    },
    {
      "epoch": 4.23126220703125e-05,
      "model_forward_time": 1.699422836303711,
      "step": 44368
    },
    {
      "epoch": 4.23126220703125e-05,
      "step": 44368,
      "training_step_time": 8.871482849121094
    },
    {
      "epoch": 4.2313575744628906e-05,
      "model_forward_time": 1.7006583213806152,
      "step": 44369
    },
    {
      "epoch": 4.2313575744628906e-05,
      "step": 44369,
      "training_step_time": 7.815682649612427
    },
    {
      "epoch": 4.231452941894531e-05,
      "grad_norm": 0.12713763117790222,
      "learning_rate": 6.328294235194073e-06,
      "loss": 0.0582,
      "step": 44370
    },
    {
      "epoch": 4.231452941894531e-05,
      "model_forward_time": 1.7105352878570557,
      "step": 44370
    },
    {
      "dynamics_loss_avg": 0.06018190747126937,
      "epoch": 4.231452941894531e-05,
      "step": 44370
    },
    {
      "action_loss_avg": 0.001767309452407062,
      "epoch": 4.231452941894531e-05,
      "step": 44370
    },
    {
      "epoch": 4.231452941894531e-05,
      "step": 44370,
      "training_step_time": 7.641254425048828
    },
    {
      "epoch": 4.231548309326172e-05,
      "model_forward_time": 1.6972639560699463,
      "step": 44371
    },
    {
      "epoch": 4.231548309326172e-05,
      "step": 44371,
      "training_step_time": 7.765542268753052
    },
    {
      "epoch": 4.2316436767578125e-05,
      "model_forward_time": 1.6928997039794922,
      "step": 44372
    },
    {
      "epoch": 4.2316436767578125e-05,
      "step": 44372,
      "training_step_time": 7.64634108543396
    },
    {
      "epoch": 4.231739044189453e-05,
      "model_forward_time": 1.7036564350128174,
      "step": 44373
    },
    {
      "epoch": 4.231739044189453e-05,
      "step": 44373,
      "training_step_time": 7.614081144332886
    },
    {
      "epoch": 4.231834411621094e-05,
      "model_forward_time": 2.249204397201538,
      "step": 44374
    },
    {
      "epoch": 4.231834411621094e-05,
      "step": 44374,
      "training_step_time": 7.713776588439941
    },
    {
      "epoch": 4.2319297790527344e-05,
      "model_forward_time": 1.6880688667297363,
      "step": 44375
    },
    {
      "epoch": 4.2319297790527344e-05,
      "step": 44375,
      "training_step_time": 8.544310808181763
    },
    {
      "epoch": 4.232025146484375e-05,
      "model_forward_time": 1.6955149173736572,
      "step": 44376
    },
    {
      "epoch": 4.232025146484375e-05,
      "step": 44376,
      "training_step_time": 8.256725788116455
    },
    {
      "epoch": 4.232120513916016e-05,
      "model_forward_time": 1.6976964473724365,
      "step": 44377
    },
    {
      "epoch": 4.232120513916016e-05,
      "step": 44377,
      "training_step_time": 8.683682441711426
    },
    {
      "epoch": 4.232215881347656e-05,
      "model_forward_time": 1.7230021953582764,
      "step": 44378
    },
    {
      "epoch": 4.232215881347656e-05,
      "step": 44378,
      "training_step_time": 8.570637464523315
    },
    {
      "epoch": 4.232311248779297e-05,
      "model_forward_time": 1.7024815082550049,
      "step": 44379
    },
    {
      "epoch": 4.232311248779297e-05,
      "step": 44379,
      "training_step_time": 7.647320032119751
    },
    {
      "epoch": 4.232406616210938e-05,
      "grad_norm": 0.10718893259763718,
      "learning_rate": 6.326700106737765e-06,
      "loss": 0.0587,
      "step": 44380
    },
    {
      "epoch": 4.232406616210938e-05,
      "model_forward_time": 1.6831448078155518,
      "step": 44380
    },
    {
      "dynamics_loss_avg": 0.04841546677052975,
      "epoch": 4.232406616210938e-05,
      "step": 44380
    },
    {
      "action_loss_avg": 0.0014225108840037136,
      "epoch": 4.232406616210938e-05,
      "step": 44380
    },
    {
      "epoch": 4.232406616210938e-05,
      "step": 44380,
      "training_step_time": 7.819513559341431
    },
    {
      "epoch": 4.232501983642578e-05,
      "model_forward_time": 1.6979846954345703,
      "step": 44381
    },
    {
      "epoch": 4.232501983642578e-05,
      "step": 44381,
      "training_step_time": 7.6647748947143555
    },
    {
      "epoch": 4.2325973510742185e-05,
      "model_forward_time": 1.3226749897003174,
      "step": 44382
    },
    {
      "epoch": 4.2325973510742185e-05,
      "step": 44382,
      "training_step_time": 7.545855760574341
    },
    {
      "epoch": 4.2326927185058596e-05,
      "model_forward_time": 2.2511444091796875,
      "step": 44383
    },
    {
      "epoch": 4.2326927185058596e-05,
      "step": 44383,
      "training_step_time": 7.776052951812744
    },
    {
      "epoch": 4.2327880859375e-05,
      "model_forward_time": 1.69818115234375,
      "step": 44384
    },
    {
      "epoch": 4.2327880859375e-05,
      "step": 44384,
      "training_step_time": 7.852763414382935
    },
    {
      "epoch": 4.2328834533691404e-05,
      "model_forward_time": 1.6988697052001953,
      "step": 44385
    },
    {
      "epoch": 4.2328834533691404e-05,
      "step": 44385,
      "training_step_time": 8.572104692459106
    },
    {
      "epoch": 4.2329788208007815e-05,
      "model_forward_time": 1.7009541988372803,
      "step": 44386
    },
    {
      "epoch": 4.2329788208007815e-05,
      "step": 44386,
      "training_step_time": 7.876278877258301
    },
    {
      "epoch": 4.233074188232422e-05,
      "model_forward_time": 1.7164409160614014,
      "step": 44387
    },
    {
      "epoch": 4.233074188232422e-05,
      "step": 44387,
      "training_step_time": 8.829583644866943
    },
    {
      "epoch": 4.233169555664062e-05,
      "model_forward_time": 1.6948800086975098,
      "step": 44388
    },
    {
      "epoch": 4.233169555664062e-05,
      "step": 44388,
      "training_step_time": 8.224947690963745
    },
    {
      "epoch": 4.2332649230957034e-05,
      "model_forward_time": 1.6967637538909912,
      "step": 44389
    },
    {
      "epoch": 4.2332649230957034e-05,
      "step": 44389,
      "training_step_time": 7.802477121353149
    },
    {
      "epoch": 4.233360290527344e-05,
      "grad_norm": 0.1408764272928238,
      "learning_rate": 6.325105833195531e-06,
      "loss": 0.057,
      "step": 44390
    },
    {
      "epoch": 4.233360290527344e-05,
      "model_forward_time": 2.2608301639556885,
      "step": 44390
    },
    {
      "dynamics_loss_avg": 0.05257592387497425,
      "epoch": 4.233360290527344e-05,
      "step": 44390
    },
    {
      "action_loss_avg": 0.003517290169838816,
      "epoch": 4.233360290527344e-05,
      "step": 44390
    },
    {
      "epoch": 4.233360290527344e-05,
      "step": 44390,
      "training_step_time": 9.509625911712646
    },
    {
      "epoch": 4.233455657958984e-05,
      "model_forward_time": 1.7143034934997559,
      "step": 44391
    },
    {
      "epoch": 4.233455657958984e-05,
      "step": 44391,
      "training_step_time": 6.966202974319458
    },
    {
      "epoch": 4.233551025390625e-05,
      "model_forward_time": 1.6981878280639648,
      "step": 44392
    },
    {
      "epoch": 4.233551025390625e-05,
      "step": 44392,
      "training_step_time": 6.950307130813599
    },
    {
      "epoch": 4.233646392822266e-05,
      "model_forward_time": 1.7137179374694824,
      "step": 44393
    },
    {
      "epoch": 4.233646392822266e-05,
      "step": 44393,
      "training_step_time": 6.989372968673706
    },
    {
      "epoch": 4.233741760253906e-05,
      "model_forward_time": 1.7116186618804932,
      "step": 44394
    },
    {
      "epoch": 4.233741760253906e-05,
      "step": 44394,
      "training_step_time": 6.968122959136963
    },
    {
      "epoch": 4.233837127685547e-05,
      "model_forward_time": 1.7037959098815918,
      "step": 44395
    },
    {
      "epoch": 4.233837127685547e-05,
      "step": 44395,
      "training_step_time": 6.969039440155029
    },
    {
      "epoch": 4.2339324951171876e-05,
      "model_forward_time": 1.7114007472991943,
      "step": 44396
    },
    {
      "epoch": 4.2339324951171876e-05,
      "step": 44396,
      "training_step_time": 6.922764301300049
    },
    {
      "epoch": 4.234027862548828e-05,
      "model_forward_time": 1.7139391899108887,
      "step": 44397
    },
    {
      "epoch": 4.234027862548828e-05,
      "step": 44397,
      "training_step_time": 6.789503574371338
    },
    {
      "epoch": 4.234123229980469e-05,
      "model_forward_time": 1.7198402881622314,
      "step": 44398
    },
    {
      "epoch": 4.234123229980469e-05,
      "step": 44398,
      "training_step_time": 7.82013201713562
    },
    {
      "epoch": 4.2342185974121095e-05,
      "model_forward_time": 1.713258981704712,
      "step": 44399
    },
    {
      "epoch": 4.2342185974121095e-05,
      "step": 44399,
      "training_step_time": 10.525944709777832
    },
    {
      "epoch": 4.23431396484375e-05,
      "grad_norm": 0.1282237321138382,
      "learning_rate": 6.323511414741714e-06,
      "loss": 0.0567,
      "step": 44400
    },
    {
      "epoch": 4.23431396484375e-05,
      "model_forward_time": 1.7067785263061523,
      "step": 44400
    },
    {
      "dynamics_loss_avg": 0.05186950173228979,
      "epoch": 4.23431396484375e-05,
      "step": 44400
    },
    {
      "action_loss_avg": 0.002283698820974678,
      "epoch": 4.23431396484375e-05,
      "step": 44400
    },
    {
      "epoch": 4.23431396484375e-05,
      "step": 44400,
      "training_step_time": 7.853261232376099
    },
    {
      "epoch": 4.234409332275391e-05,
      "model_forward_time": 1.7231512069702148,
      "step": 44401
    },
    {
      "epoch": 4.234409332275391e-05,
      "step": 44401,
      "training_step_time": 6.9530930519104
    },
    {
      "epoch": 4.2345046997070313e-05,
      "model_forward_time": 1.7104926109313965,
      "step": 44402
    },
    {
      "epoch": 4.2345046997070313e-05,
      "step": 44402,
      "training_step_time": 6.964182376861572
    },
    {
      "epoch": 4.234600067138672e-05,
      "model_forward_time": 1.721383810043335,
      "step": 44403
    },
    {
      "epoch": 4.234600067138672e-05,
      "step": 44403,
      "training_step_time": 6.958590269088745
    },
    {
      "epoch": 4.234695434570313e-05,
      "model_forward_time": 1.7122383117675781,
      "step": 44404
    },
    {
      "epoch": 4.234695434570313e-05,
      "step": 44404,
      "training_step_time": 6.7064292430877686
    },
    {
      "epoch": 4.234790802001953e-05,
      "model_forward_time": 1.713310956954956,
      "step": 44405
    },
    {
      "epoch": 4.234790802001953e-05,
      "step": 44405,
      "training_step_time": 6.918997764587402
    },
    {
      "epoch": 4.2348861694335936e-05,
      "model_forward_time": 1.7131848335266113,
      "step": 44406
    },
    {
      "epoch": 4.2348861694335936e-05,
      "step": 44406,
      "training_step_time": 6.882204294204712
    },
    {
      "epoch": 4.234981536865234e-05,
      "model_forward_time": 1.714674472808838,
      "step": 44407
    },
    {
      "epoch": 4.234981536865234e-05,
      "step": 44407,
      "training_step_time": 6.936187982559204
    },
    {
      "epoch": 4.235076904296875e-05,
      "model_forward_time": 2.262422561645508,
      "step": 44408
    },
    {
      "epoch": 4.235076904296875e-05,
      "step": 44408,
      "training_step_time": 8.827494382858276
    },
    {
      "epoch": 4.2351722717285155e-05,
      "model_forward_time": 1.7118196487426758,
      "step": 44409
    },
    {
      "epoch": 4.2351722717285155e-05,
      "step": 44409,
      "training_step_time": 7.865408658981323
    },
    {
      "epoch": 4.235267639160156e-05,
      "grad_norm": 0.09152276813983917,
      "learning_rate": 6.321916851550679e-06,
      "loss": 0.0544,
      "step": 44410
    },
    {
      "epoch": 4.235267639160156e-05,
      "model_forward_time": 1.707322597503662,
      "step": 44410
    },
    {
      "dynamics_loss_avg": 0.04630602058023214,
      "epoch": 4.235267639160156e-05,
      "step": 44410
    },
    {
      "action_loss_avg": 0.0021574404381681234,
      "epoch": 4.235267639160156e-05,
      "step": 44410
    },
    {
      "epoch": 4.235267639160156e-05,
      "step": 44410,
      "training_step_time": 6.912908554077148
    },
    {
      "epoch": 4.235363006591797e-05,
      "model_forward_time": 1.7175381183624268,
      "step": 44411
    },
    {
      "epoch": 4.235363006591797e-05,
      "step": 44411,
      "training_step_time": 6.895159006118774
    },
    {
      "epoch": 4.2354583740234374e-05,
      "model_forward_time": 1.714540958404541,
      "step": 44412
    },
    {
      "epoch": 4.2354583740234374e-05,
      "step": 44412,
      "training_step_time": 6.887663841247559
    },
    {
      "epoch": 4.235553741455078e-05,
      "model_forward_time": 1.7131412029266357,
      "step": 44413
    },
    {
      "epoch": 4.235553741455078e-05,
      "step": 44413,
      "training_step_time": 6.892221689224243
    },
    {
      "epoch": 4.235649108886719e-05,
      "model_forward_time": 1.7140722274780273,
      "step": 44414
    },
    {
      "epoch": 4.235649108886719e-05,
      "step": 44414,
      "training_step_time": 6.8976662158966064
    },
    {
      "epoch": 4.235744476318359e-05,
      "model_forward_time": 1.7222058773040771,
      "step": 44415
    },
    {
      "epoch": 4.235744476318359e-05,
      "step": 44415,
      "training_step_time": 6.888182640075684
    },
    {
      "epoch": 4.23583984375e-05,
      "model_forward_time": 1.7144606113433838,
      "step": 44416
    },
    {
      "epoch": 4.23583984375e-05,
      "step": 44416,
      "training_step_time": 6.9620680809021
    },
    {
      "epoch": 4.235935211181641e-05,
      "model_forward_time": 2.278095245361328,
      "step": 44417
    },
    {
      "epoch": 4.235935211181641e-05,
      "step": 44417,
      "training_step_time": 8.838489770889282
    },
    {
      "epoch": 4.236030578613281e-05,
      "model_forward_time": 1.7189712524414062,
      "step": 44418
    },
    {
      "epoch": 4.236030578613281e-05,
      "step": 44418,
      "training_step_time": 9.086199283599854
    },
    {
      "epoch": 4.2361259460449216e-05,
      "model_forward_time": 1.6987402439117432,
      "step": 44419
    },
    {
      "epoch": 4.2361259460449216e-05,
      "step": 44419,
      "training_step_time": 7.618273019790649
    },
    {
      "epoch": 4.236221313476563e-05,
      "grad_norm": 0.15754461288452148,
      "learning_rate": 6.320322143796805e-06,
      "loss": 0.0598,
      "step": 44420
    },
    {
      "epoch": 4.236221313476563e-05,
      "model_forward_time": 1.6966655254364014,
      "step": 44420
    },
    {
      "dynamics_loss_avg": 0.06607517413794994,
      "epoch": 4.236221313476563e-05,
      "step": 44420
    },
    {
      "action_loss_avg": 0.0028661729709710928,
      "epoch": 4.236221313476563e-05,
      "step": 44420
    },
    {
      "epoch": 4.236221313476563e-05,
      "step": 44420,
      "training_step_time": 6.89741849899292
    },
    {
      "epoch": 4.236316680908203e-05,
      "model_forward_time": 1.7223711013793945,
      "step": 44421
    },
    {
      "epoch": 4.236316680908203e-05,
      "step": 44421,
      "training_step_time": 6.919560670852661
    },
    {
      "epoch": 4.2364120483398435e-05,
      "model_forward_time": 1.7085731029510498,
      "step": 44422
    },
    {
      "epoch": 4.2364120483398435e-05,
      "step": 44422,
      "training_step_time": 6.942310333251953
    },
    {
      "epoch": 4.2365074157714845e-05,
      "model_forward_time": 1.715130090713501,
      "step": 44423
    },
    {
      "epoch": 4.2365074157714845e-05,
      "step": 44423,
      "training_step_time": 6.923407316207886
    },
    {
      "epoch": 4.236602783203125e-05,
      "model_forward_time": 1.7108359336853027,
      "step": 44424
    },
    {
      "epoch": 4.236602783203125e-05,
      "step": 44424,
      "training_step_time": 6.995347499847412
    },
    {
      "epoch": 4.2366981506347654e-05,
      "model_forward_time": 1.7114794254302979,
      "step": 44425
    },
    {
      "epoch": 4.2366981506347654e-05,
      "step": 44425,
      "training_step_time": 6.964531183242798
    },
    {
      "epoch": 4.2367935180664064e-05,
      "model_forward_time": 2.2658841609954834,
      "step": 44426
    },
    {
      "epoch": 4.2367935180664064e-05,
      "step": 44426,
      "training_step_time": 7.817797660827637
    },
    {
      "epoch": 4.236888885498047e-05,
      "model_forward_time": 1.7106800079345703,
      "step": 44427
    },
    {
      "epoch": 4.236888885498047e-05,
      "step": 44427,
      "training_step_time": 8.572030305862427
    },
    {
      "epoch": 4.236984252929687e-05,
      "model_forward_time": 1.7084393501281738,
      "step": 44428
    },
    {
      "epoch": 4.236984252929687e-05,
      "step": 44428,
      "training_step_time": 7.7776031494140625
    },
    {
      "epoch": 4.237079620361328e-05,
      "model_forward_time": 1.7110114097595215,
      "step": 44429
    },
    {
      "epoch": 4.237079620361328e-05,
      "step": 44429,
      "training_step_time": 6.914692640304565
    },
    {
      "epoch": 4.237174987792969e-05,
      "grad_norm": 0.12065047770738602,
      "learning_rate": 6.318727291654487e-06,
      "loss": 0.0598,
      "step": 44430
    },
    {
      "epoch": 4.237174987792969e-05,
      "model_forward_time": 1.709441900253296,
      "step": 44430
    },
    {
      "dynamics_loss_avg": 0.07924943696707487,
      "epoch": 4.237174987792969e-05,
      "step": 44430
    },
    {
      "action_loss_avg": 0.002941112214466557,
      "epoch": 4.237174987792969e-05,
      "step": 44430
    },
    {
      "epoch": 4.237174987792969e-05,
      "step": 44430,
      "training_step_time": 6.889704942703247
    },
    {
      "epoch": 4.237270355224609e-05,
      "model_forward_time": 1.712942123413086,
      "step": 44431
    },
    {
      "epoch": 4.237270355224609e-05,
      "step": 44431,
      "training_step_time": 6.950124502182007
    },
    {
      "epoch": 4.23736572265625e-05,
      "model_forward_time": 1.706413984298706,
      "step": 44432
    },
    {
      "epoch": 4.23736572265625e-05,
      "step": 44432,
      "training_step_time": 6.862706899642944
    },
    {
      "epoch": 4.2374610900878906e-05,
      "model_forward_time": 1.7107198238372803,
      "step": 44433
    },
    {
      "epoch": 4.2374610900878906e-05,
      "step": 44433,
      "training_step_time": 6.9205663204193115
    },
    {
      "epoch": 4.237556457519531e-05,
      "model_forward_time": 1.7195453643798828,
      "step": 44434
    },
    {
      "epoch": 4.237556457519531e-05,
      "step": 44434,
      "training_step_time": 6.890118837356567
    },
    {
      "epoch": 4.237651824951172e-05,
      "model_forward_time": 2.259877920150757,
      "step": 44435
    },
    {
      "epoch": 4.237651824951172e-05,
      "step": 44435,
      "training_step_time": 7.758541584014893
    },
    {
      "epoch": 4.2377471923828125e-05,
      "model_forward_time": 1.7136402130126953,
      "step": 44436
    },
    {
      "epoch": 4.2377471923828125e-05,
      "step": 44436,
      "training_step_time": 10.81159257888794
    },
    {
      "epoch": 4.237842559814453e-05,
      "model_forward_time": 1.714416265487671,
      "step": 44437
    },
    {
      "epoch": 4.237842559814453e-05,
      "step": 44437,
      "training_step_time": 7.935209035873413
    },
    {
      "epoch": 4.237937927246094e-05,
      "model_forward_time": 1.7181675434112549,
      "step": 44438
    },
    {
      "epoch": 4.237937927246094e-05,
      "step": 44438,
      "training_step_time": 6.909801006317139
    },
    {
      "epoch": 4.2380332946777344e-05,
      "model_forward_time": 1.6995716094970703,
      "step": 44439
    },
    {
      "epoch": 4.2380332946777344e-05,
      "step": 44439,
      "training_step_time": 6.882473707199097
    },
    {
      "epoch": 4.238128662109375e-05,
      "grad_norm": 0.12967906892299652,
      "learning_rate": 6.317132295298134e-06,
      "loss": 0.0596,
      "step": 44440
    },
    {
      "epoch": 4.238128662109375e-05,
      "model_forward_time": 1.7084195613861084,
      "step": 44440
    },
    {
      "dynamics_loss_avg": 0.06759791728109121,
      "epoch": 4.238128662109375e-05,
      "step": 44440
    },
    {
      "action_loss_avg": 0.0031439314829185606,
      "epoch": 4.238128662109375e-05,
      "step": 44440
    },
    {
      "epoch": 4.238128662109375e-05,
      "step": 44440,
      "training_step_time": 6.910668849945068
    },
    {
      "epoch": 4.238224029541016e-05,
      "model_forward_time": 1.7152678966522217,
      "step": 44441
    },
    {
      "epoch": 4.238224029541016e-05,
      "step": 44441,
      "training_step_time": 6.921224355697632
    },
    {
      "epoch": 4.238319396972656e-05,
      "model_forward_time": 1.715181827545166,
      "step": 44442
    },
    {
      "epoch": 4.238319396972656e-05,
      "step": 44442,
      "training_step_time": 6.918741703033447
    },
    {
      "epoch": 4.238414764404297e-05,
      "model_forward_time": 1.717881679534912,
      "step": 44443
    },
    {
      "epoch": 4.238414764404297e-05,
      "step": 44443,
      "training_step_time": 6.951076984405518
    },
    {
      "epoch": 4.238510131835938e-05,
      "model_forward_time": 2.2397024631500244,
      "step": 44444
    },
    {
      "epoch": 4.238510131835938e-05,
      "step": 44444,
      "training_step_time": 7.675699234008789
    },
    {
      "epoch": 4.238605499267578e-05,
      "model_forward_time": 1.7060251235961914,
      "step": 44445
    },
    {
      "epoch": 4.238605499267578e-05,
      "step": 44445,
      "training_step_time": 9.316710472106934
    },
    {
      "epoch": 4.2387008666992186e-05,
      "model_forward_time": 1.700028657913208,
      "step": 44446
    },
    {
      "epoch": 4.2387008666992186e-05,
      "step": 44446,
      "training_step_time": 9.869330644607544
    },
    {
      "epoch": 4.2387962341308596e-05,
      "model_forward_time": 1.7149319648742676,
      "step": 44447
    },
    {
      "epoch": 4.2387962341308596e-05,
      "step": 44447,
      "training_step_time": 7.263748407363892
    },
    {
      "epoch": 4.2388916015625e-05,
      "model_forward_time": 1.716130018234253,
      "step": 44448
    },
    {
      "epoch": 4.2388916015625e-05,
      "step": 44448,
      "training_step_time": 6.949337720870972
    },
    {
      "epoch": 4.2389869689941404e-05,
      "model_forward_time": 1.704624891281128,
      "step": 44449
    },
    {
      "epoch": 4.2389869689941404e-05,
      "step": 44449,
      "training_step_time": 6.796766757965088
    },
    {
      "epoch": 4.2390823364257815e-05,
      "grad_norm": 0.11246450990438461,
      "learning_rate": 6.315537154902174e-06,
      "loss": 0.0585,
      "step": 44450
    },
    {
      "epoch": 4.2390823364257815e-05,
      "model_forward_time": 1.7158503532409668,
      "step": 44450
    },
    {
      "dynamics_loss_avg": 0.06371742337942124,
      "epoch": 4.2390823364257815e-05,
      "step": 44450
    },
    {
      "action_loss_avg": 0.002222042516223155,
      "epoch": 4.2390823364257815e-05,
      "step": 44450
    },
    {
      "epoch": 4.2390823364257815e-05,
      "step": 44450,
      "training_step_time": 6.96191668510437
    },
    {
      "epoch": 4.239177703857422e-05,
      "model_forward_time": 1.7199554443359375,
      "step": 44451
    },
    {
      "epoch": 4.239177703857422e-05,
      "step": 44451,
      "training_step_time": 6.979902744293213
    },
    {
      "epoch": 4.239273071289062e-05,
      "model_forward_time": 1.7173452377319336,
      "step": 44452
    },
    {
      "epoch": 4.239273071289062e-05,
      "step": 44452,
      "training_step_time": 6.989969253540039
    },
    {
      "epoch": 4.2393684387207034e-05,
      "model_forward_time": 1.7152259349822998,
      "step": 44453
    },
    {
      "epoch": 4.2393684387207034e-05,
      "step": 44453,
      "training_step_time": 7.473336935043335
    },
    {
      "epoch": 4.239463806152344e-05,
      "model_forward_time": 1.732360601425171,
      "step": 44454
    },
    {
      "epoch": 4.239463806152344e-05,
      "step": 44454,
      "training_step_time": 8.381399631500244
    },
    {
      "epoch": 4.239559173583984e-05,
      "model_forward_time": 1.707841157913208,
      "step": 44455
    },
    {
      "epoch": 4.239559173583984e-05,
      "step": 44455,
      "training_step_time": 7.8943400382995605
    },
    {
      "epoch": 4.239654541015625e-05,
      "model_forward_time": 1.7069499492645264,
      "step": 44456
    },
    {
      "epoch": 4.239654541015625e-05,
      "step": 44456,
      "training_step_time": 7.590936660766602
    },
    {
      "epoch": 4.239749908447266e-05,
      "model_forward_time": 1.7197489738464355,
      "step": 44457
    },
    {
      "epoch": 4.239749908447266e-05,
      "step": 44457,
      "training_step_time": 6.917841911315918
    },
    {
      "epoch": 4.239845275878906e-05,
      "model_forward_time": 1.726560115814209,
      "step": 44458
    },
    {
      "epoch": 4.239845275878906e-05,
      "step": 44458,
      "training_step_time": 6.9616782665252686
    },
    {
      "epoch": 4.239940643310547e-05,
      "model_forward_time": 1.7179629802703857,
      "step": 44459
    },
    {
      "epoch": 4.239940643310547e-05,
      "step": 44459,
      "training_step_time": 7.010003328323364
    },
    {
      "epoch": 4.2400360107421876e-05,
      "grad_norm": 0.1248091384768486,
      "learning_rate": 6.313941870641048e-06,
      "loss": 0.0556,
      "step": 44460
    },
    {
      "epoch": 4.2400360107421876e-05,
      "model_forward_time": 1.704174280166626,
      "step": 44460
    },
    {
      "dynamics_loss_avg": 0.05775012653321028,
      "epoch": 4.2400360107421876e-05,
      "step": 44460
    },
    {
      "action_loss_avg": 0.002111250162124634,
      "epoch": 4.2400360107421876e-05,
      "step": 44460
    },
    {
      "epoch": 4.2400360107421876e-05,
      "step": 44460,
      "training_step_time": 7.016972780227661
    },
    {
      "epoch": 4.240131378173828e-05,
      "model_forward_time": 1.7146222591400146,
      "step": 44461
    },
    {
      "epoch": 4.240131378173828e-05,
      "step": 44461,
      "training_step_time": 6.935660123825073
    },
    {
      "epoch": 4.240226745605469e-05,
      "model_forward_time": 1.7135744094848633,
      "step": 44462
    },
    {
      "epoch": 4.240226745605469e-05,
      "step": 44462,
      "training_step_time": 6.977212190628052
    },
    {
      "epoch": 4.2403221130371095e-05,
      "model_forward_time": 2.300719976425171,
      "step": 44463
    },
    {
      "epoch": 4.2403221130371095e-05,
      "step": 44463,
      "training_step_time": 9.461378335952759
    },
    {
      "epoch": 4.24041748046875e-05,
      "model_forward_time": 1.7149338722229004,
      "step": 44464
    },
    {
      "epoch": 4.24041748046875e-05,
      "step": 44464,
      "training_step_time": 9.49645209312439
    },
    {
      "epoch": 4.240512847900391e-05,
      "model_forward_time": 1.713930368423462,
      "step": 44465
    },
    {
      "epoch": 4.240512847900391e-05,
      "step": 44465,
      "training_step_time": 7.600079536437988
    },
    {
      "epoch": 4.2406082153320314e-05,
      "model_forward_time": 1.7132282257080078,
      "step": 44466
    },
    {
      "epoch": 4.2406082153320314e-05,
      "step": 44466,
      "training_step_time": 6.947782039642334
    },
    {
      "epoch": 4.240703582763672e-05,
      "model_forward_time": 1.712268590927124,
      "step": 44467
    },
    {
      "epoch": 4.240703582763672e-05,
      "step": 44467,
      "training_step_time": 7.044010877609253
    },
    {
      "epoch": 4.240798950195313e-05,
      "model_forward_time": 1.749236822128296,
      "step": 44468
    },
    {
      "epoch": 4.240798950195313e-05,
      "step": 44468,
      "training_step_time": 7.032269477844238
    },
    {
      "epoch": 4.240894317626953e-05,
      "model_forward_time": 1.7034413814544678,
      "step": 44469
    },
    {
      "epoch": 4.240894317626953e-05,
      "step": 44469,
      "training_step_time": 6.85843300819397
    },
    {
      "epoch": 4.2409896850585936e-05,
      "grad_norm": 0.11633460968732834,
      "learning_rate": 6.312346442689215e-06,
      "loss": 0.0572,
      "step": 44470
    },
    {
      "epoch": 4.2409896850585936e-05,
      "model_forward_time": 1.7069079875946045,
      "step": 44470
    },
    {
      "dynamics_loss_avg": 0.05250496147200465,
      "epoch": 4.2409896850585936e-05,
      "step": 44470
    },
    {
      "action_loss_avg": 0.002034479327267036,
      "epoch": 4.2409896850585936e-05,
      "step": 44470
    },
    {
      "epoch": 4.2409896850585936e-05,
      "step": 44470,
      "training_step_time": 6.966565370559692
    },
    {
      "epoch": 4.241085052490234e-05,
      "model_forward_time": 1.3284924030303955,
      "step": 44471
    },
    {
      "epoch": 4.241085052490234e-05,
      "step": 44471,
      "training_step_time": 6.993684768676758
    },
    {
      "epoch": 4.241180419921875e-05,
      "model_forward_time": 2.253549098968506,
      "step": 44472
    },
    {
      "epoch": 4.241180419921875e-05,
      "step": 44472,
      "training_step_time": 8.85948133468628
    },
    {
      "epoch": 4.2412757873535155e-05,
      "model_forward_time": 1.7108454704284668,
      "step": 44473
    },
    {
      "epoch": 4.2412757873535155e-05,
      "step": 44473,
      "training_step_time": 8.53602147102356
    },
    {
      "epoch": 4.241371154785156e-05,
      "model_forward_time": 1.700059413909912,
      "step": 44474
    },
    {
      "epoch": 4.241371154785156e-05,
      "step": 44474,
      "training_step_time": 7.8833582401275635
    },
    {
      "epoch": 4.241466522216797e-05,
      "model_forward_time": 1.7178335189819336,
      "step": 44475
    },
    {
      "epoch": 4.241466522216797e-05,
      "step": 44475,
      "training_step_time": 6.92227029800415
    },
    {
      "epoch": 4.2415618896484374e-05,
      "model_forward_time": 1.7197368144989014,
      "step": 44476
    },
    {
      "epoch": 4.2415618896484374e-05,
      "step": 44476,
      "training_step_time": 7.002078533172607
    },
    {
      "epoch": 4.241657257080078e-05,
      "model_forward_time": 1.718132495880127,
      "step": 44477
    },
    {
      "epoch": 4.241657257080078e-05,
      "step": 44477,
      "training_step_time": 6.936118125915527
    },
    {
      "epoch": 4.241752624511719e-05,
      "model_forward_time": 1.7139174938201904,
      "step": 44478
    },
    {
      "epoch": 4.241752624511719e-05,
      "step": 44478,
      "training_step_time": 6.92735743522644
    },
    {
      "epoch": 4.241847991943359e-05,
      "model_forward_time": 1.718029260635376,
      "step": 44479
    },
    {
      "epoch": 4.241847991943359e-05,
      "step": 44479,
      "training_step_time": 6.928673267364502
    },
    {
      "epoch": 4.241943359375e-05,
      "grad_norm": 0.11050307750701904,
      "learning_rate": 6.3107508712211476e-06,
      "loss": 0.0586,
      "step": 44480
    },
    {
      "epoch": 4.241943359375e-05,
      "model_forward_time": 1.71372652053833,
      "step": 44480
    },
    {
      "dynamics_loss_avg": 0.0748596504330635,
      "epoch": 4.241943359375e-05,
      "step": 44480
    },
    {
      "action_loss_avg": 0.003489752113819122,
      "epoch": 4.241943359375e-05,
      "step": 44480
    },
    {
      "epoch": 4.241943359375e-05,
      "step": 44480,
      "training_step_time": 6.927550315856934
    },
    {
      "epoch": 4.242038726806641e-05,
      "model_forward_time": 2.2678020000457764,
      "step": 44481
    },
    {
      "epoch": 4.242038726806641e-05,
      "step": 44481,
      "training_step_time": 7.574814081192017
    },
    {
      "epoch": 4.242134094238281e-05,
      "model_forward_time": 1.7812626361846924,
      "step": 44482
    },
    {
      "epoch": 4.242134094238281e-05,
      "step": 44482,
      "training_step_time": 10.95252799987793
    },
    {
      "epoch": 4.2422294616699216e-05,
      "model_forward_time": 1.7142574787139893,
      "step": 44483
    },
    {
      "epoch": 4.2422294616699216e-05,
      "step": 44483,
      "training_step_time": 7.778872013092041
    },
    {
      "epoch": 4.242324829101563e-05,
      "model_forward_time": 1.7281184196472168,
      "step": 44484
    },
    {
      "epoch": 4.242324829101563e-05,
      "step": 44484,
      "training_step_time": 7.677716493606567
    },
    {
      "epoch": 4.242420196533203e-05,
      "model_forward_time": 1.7174203395843506,
      "step": 44485
    },
    {
      "epoch": 4.242420196533203e-05,
      "step": 44485,
      "training_step_time": 6.887292385101318
    },
    {
      "epoch": 4.2425155639648435e-05,
      "model_forward_time": 1.7211906909942627,
      "step": 44486
    },
    {
      "epoch": 4.2425155639648435e-05,
      "step": 44486,
      "training_step_time": 6.934509038925171
    },
    {
      "epoch": 4.2426109313964846e-05,
      "model_forward_time": 1.7138447761535645,
      "step": 44487
    },
    {
      "epoch": 4.2426109313964846e-05,
      "step": 44487,
      "training_step_time": 6.910696506500244
    },
    {
      "epoch": 4.242706298828125e-05,
      "model_forward_time": 1.7167458534240723,
      "step": 44488
    },
    {
      "epoch": 4.242706298828125e-05,
      "step": 44488,
      "training_step_time": 6.950140714645386
    },
    {
      "epoch": 4.2428016662597654e-05,
      "model_forward_time": 1.713531255722046,
      "step": 44489
    },
    {
      "epoch": 4.2428016662597654e-05,
      "step": 44489,
      "training_step_time": 6.916998624801636
    },
    {
      "epoch": 4.2428970336914064e-05,
      "grad_norm": 0.06612712889909744,
      "learning_rate": 6.309155156411335e-06,
      "loss": 0.0598,
      "step": 44490
    },
    {
      "epoch": 4.2428970336914064e-05,
      "model_forward_time": 2.2813632488250732,
      "step": 44490
    },
    {
      "dynamics_loss_avg": 0.06106764208525419,
      "epoch": 4.2428970336914064e-05,
      "step": 44490
    },
    {
      "action_loss_avg": 0.002037622913485393,
      "epoch": 4.2428970336914064e-05,
      "step": 44490
    },
    {
      "epoch": 4.2428970336914064e-05,
      "step": 44490,
      "training_step_time": 7.469501256942749
    },
    {
      "epoch": 4.242992401123047e-05,
      "model_forward_time": 1.7769031524658203,
      "step": 44491
    },
    {
      "epoch": 4.242992401123047e-05,
      "step": 44491,
      "training_step_time": 11.563446998596191
    },
    {
      "epoch": 4.243087768554687e-05,
      "model_forward_time": 1.7190165519714355,
      "step": 44492
    },
    {
      "epoch": 4.243087768554687e-05,
      "step": 44492,
      "training_step_time": 7.827288866043091
    },
    {
      "epoch": 4.243183135986328e-05,
      "model_forward_time": 1.331223726272583,
      "step": 44493
    },
    {
      "epoch": 4.243183135986328e-05,
      "step": 44493,
      "training_step_time": 7.6874542236328125
    },
    {
      "epoch": 4.243278503417969e-05,
      "model_forward_time": 1.7196359634399414,
      "step": 44494
    },
    {
      "epoch": 4.243278503417969e-05,
      "step": 44494,
      "training_step_time": 7.000771522521973
    },
    {
      "epoch": 4.243373870849609e-05,
      "model_forward_time": 1.7191271781921387,
      "step": 44495
    },
    {
      "epoch": 4.243373870849609e-05,
      "step": 44495,
      "training_step_time": 6.950326442718506
    },
    {
      "epoch": 4.24346923828125e-05,
      "model_forward_time": 1.7144274711608887,
      "step": 44496
    },
    {
      "epoch": 4.24346923828125e-05,
      "step": 44496,
      "training_step_time": 7.041776180267334
    },
    {
      "epoch": 4.2435646057128906e-05,
      "model_forward_time": 1.7214360237121582,
      "step": 44497
    },
    {
      "epoch": 4.2435646057128906e-05,
      "step": 44497,
      "training_step_time": 7.050090789794922
    },
    {
      "epoch": 4.243659973144531e-05,
      "model_forward_time": 1.7186315059661865,
      "step": 44498
    },
    {
      "epoch": 4.243659973144531e-05,
      "step": 44498,
      "training_step_time": 7.041295289993286
    },
    {
      "epoch": 4.243755340576172e-05,
      "model_forward_time": 1.7176790237426758,
      "step": 44499
    },
    {
      "epoch": 4.243755340576172e-05,
      "step": 44499,
      "training_step_time": 8.255147218704224
    },
    {
      "epoch": 4.2438507080078125e-05,
      "grad_norm": 0.10661537200212479,
      "learning_rate": 6.307559298434284e-06,
      "loss": 0.0567,
      "step": 44500
    },
    {
      "epoch": 4.2438507080078125e-05,
      "model_forward_time": 1.754396915435791,
      "step": 44500
    },
    {
      "dynamics_loss_avg": 0.08039636891335249,
      "epoch": 4.2438507080078125e-05,
      "step": 44500
    },
    {
      "action_loss_avg": 0.004327007499523461,
      "epoch": 4.2438507080078125e-05,
      "step": 44500
    },
    {
      "epoch": 4.2438507080078125e-05,
      "step": 44500,
      "training_step_time": 9.129299640655518
    },
    {
      "epoch": 4.243946075439453e-05,
      "model_forward_time": 1.714615821838379,
      "step": 44501
    },
    {
      "epoch": 4.243946075439453e-05,
      "step": 44501,
      "training_step_time": 9.857861995697021
    },
    {
      "epoch": 4.244041442871094e-05,
      "model_forward_time": 1.7188329696655273,
      "step": 44502
    },
    {
      "epoch": 4.244041442871094e-05,
      "step": 44502,
      "training_step_time": 7.894984483718872
    },
    {
      "epoch": 4.2441368103027344e-05,
      "model_forward_time": 1.7174227237701416,
      "step": 44503
    },
    {
      "epoch": 4.2441368103027344e-05,
      "step": 44503,
      "training_step_time": 6.974797248840332
    },
    {
      "epoch": 4.244232177734375e-05,
      "model_forward_time": 1.7108328342437744,
      "step": 44504
    },
    {
      "epoch": 4.244232177734375e-05,
      "step": 44504,
      "training_step_time": 6.947521924972534
    },
    {
      "epoch": 4.244327545166016e-05,
      "model_forward_time": 1.7203054428100586,
      "step": 44505
    },
    {
      "epoch": 4.244327545166016e-05,
      "step": 44505,
      "training_step_time": 6.950339317321777
    },
    {
      "epoch": 4.244422912597656e-05,
      "model_forward_time": 1.7149608135223389,
      "step": 44506
    },
    {
      "epoch": 4.244422912597656e-05,
      "step": 44506,
      "training_step_time": 6.93362021446228
    },
    {
      "epoch": 4.244518280029297e-05,
      "model_forward_time": 1.7090604305267334,
      "step": 44507
    },
    {
      "epoch": 4.244518280029297e-05,
      "step": 44507,
      "training_step_time": 6.8524980545043945
    },
    {
      "epoch": 4.244613647460938e-05,
      "model_forward_time": 1.716543436050415,
      "step": 44508
    },
    {
      "epoch": 4.244613647460938e-05,
      "step": 44508,
      "training_step_time": 8.012640714645386
    },
    {
      "epoch": 4.244709014892578e-05,
      "model_forward_time": 1.7465009689331055,
      "step": 44509
    },
    {
      "epoch": 4.244709014892578e-05,
      "step": 44509,
      "training_step_time": 7.313143730163574
    },
    {
      "epoch": 4.2448043823242186e-05,
      "grad_norm": 0.11808149516582489,
      "learning_rate": 6.3059632974645125e-06,
      "loss": 0.0623,
      "step": 44510
    },
    {
      "epoch": 4.2448043823242186e-05,
      "model_forward_time": 1.707665205001831,
      "step": 44510
    },
    {
      "dynamics_loss_avg": 0.06822836436331273,
      "epoch": 4.2448043823242186e-05,
      "step": 44510
    },
    {
      "action_loss_avg": 0.0017363456718157977,
      "epoch": 4.2448043823242186e-05,
      "step": 44510
    },
    {
      "epoch": 4.2448043823242186e-05,
      "step": 44510,
      "training_step_time": 8.790292501449585
    },
    {
      "epoch": 4.2448997497558596e-05,
      "model_forward_time": 1.7029187679290771,
      "step": 44511
    },
    {
      "epoch": 4.2448997497558596e-05,
      "step": 44511,
      "training_step_time": 7.7275121212005615
    },
    {
      "epoch": 4.2449951171875e-05,
      "model_forward_time": 1.7183253765106201,
      "step": 44512
    },
    {
      "epoch": 4.2449951171875e-05,
      "step": 44512,
      "training_step_time": 6.777167081832886
    },
    {
      "epoch": 4.2450904846191404e-05,
      "model_forward_time": 1.7163405418395996,
      "step": 44513
    },
    {
      "epoch": 4.2450904846191404e-05,
      "step": 44513,
      "training_step_time": 6.765899658203125
    },
    {
      "epoch": 4.2451858520507815e-05,
      "model_forward_time": 1.718533992767334,
      "step": 44514
    },
    {
      "epoch": 4.2451858520507815e-05,
      "step": 44514,
      "training_step_time": 6.792548656463623
    },
    {
      "epoch": 4.245281219482422e-05,
      "model_forward_time": 1.7198097705841064,
      "step": 44515
    },
    {
      "epoch": 4.245281219482422e-05,
      "step": 44515,
      "training_step_time": 6.753309011459351
    },
    {
      "epoch": 4.245376586914062e-05,
      "model_forward_time": 1.7205562591552734,
      "step": 44516
    },
    {
      "epoch": 4.245376586914062e-05,
      "step": 44516,
      "training_step_time": 6.718571662902832
    },
    {
      "epoch": 4.2454719543457034e-05,
      "model_forward_time": 1.711024284362793,
      "step": 44517
    },
    {
      "epoch": 4.2454719543457034e-05,
      "step": 44517,
      "training_step_time": 7.8174872398376465
    },
    {
      "epoch": 4.245567321777344e-05,
      "model_forward_time": 1.7249391078948975,
      "step": 44518
    },
    {
      "epoch": 4.245567321777344e-05,
      "step": 44518,
      "training_step_time": 8.67418360710144
    },
    {
      "epoch": 4.245662689208984e-05,
      "model_forward_time": 1.7199757099151611,
      "step": 44519
    },
    {
      "epoch": 4.245662689208984e-05,
      "step": 44519,
      "training_step_time": 10.280601739883423
    },
    {
      "epoch": 4.245758056640625e-05,
      "grad_norm": 0.10947777330875397,
      "learning_rate": 6.304367153676561e-06,
      "loss": 0.0571,
      "step": 44520
    },
    {
      "epoch": 4.245758056640625e-05,
      "model_forward_time": 1.708298921585083,
      "step": 44520
    },
    {
      "dynamics_loss_avg": 0.061357506737113,
      "epoch": 4.245758056640625e-05,
      "step": 44520
    },
    {
      "action_loss_avg": 0.0014089595875702797,
      "epoch": 4.245758056640625e-05,
      "step": 44520
    },
    {
      "epoch": 4.245758056640625e-05,
      "step": 44520,
      "training_step_time": 7.941779613494873
    },
    {
      "epoch": 4.245853424072266e-05,
      "model_forward_time": 1.7142126560211182,
      "step": 44521
    },
    {
      "epoch": 4.245853424072266e-05,
      "step": 44521,
      "training_step_time": 7.794314384460449
    },
    {
      "epoch": 4.245948791503906e-05,
      "model_forward_time": 1.7138986587524414,
      "step": 44522
    },
    {
      "epoch": 4.245948791503906e-05,
      "step": 44522,
      "training_step_time": 6.894110441207886
    },
    {
      "epoch": 4.246044158935547e-05,
      "model_forward_time": 1.710362434387207,
      "step": 44523
    },
    {
      "epoch": 4.246044158935547e-05,
      "step": 44523,
      "training_step_time": 7.02669882774353
    },
    {
      "epoch": 4.2461395263671876e-05,
      "model_forward_time": 1.7163901329040527,
      "step": 44524
    },
    {
      "epoch": 4.2461395263671876e-05,
      "step": 44524,
      "training_step_time": 6.978788614273071
    },
    {
      "epoch": 4.246234893798828e-05,
      "model_forward_time": 1.7155311107635498,
      "step": 44525
    },
    {
      "epoch": 4.246234893798828e-05,
      "step": 44525,
      "training_step_time": 6.92638087272644
    },
    {
      "epoch": 4.246330261230469e-05,
      "model_forward_time": 1.718130111694336,
      "step": 44526
    },
    {
      "epoch": 4.246330261230469e-05,
      "step": 44526,
      "training_step_time": 7.457247972488403
    },
    {
      "epoch": 4.2464256286621095e-05,
      "model_forward_time": 2.2433159351348877,
      "step": 44527
    },
    {
      "epoch": 4.2464256286621095e-05,
      "step": 44527,
      "training_step_time": 8.273943424224854
    },
    {
      "epoch": 4.24652099609375e-05,
      "model_forward_time": 1.7158944606781006,
      "step": 44528
    },
    {
      "epoch": 4.24652099609375e-05,
      "step": 44528,
      "training_step_time": 8.58057713508606
    },
    {
      "epoch": 4.246616363525391e-05,
      "model_forward_time": 1.718076229095459,
      "step": 44529
    },
    {
      "epoch": 4.246616363525391e-05,
      "step": 44529,
      "training_step_time": 8.505183696746826
    },
    {
      "epoch": 4.2467117309570314e-05,
      "grad_norm": 0.14474742114543915,
      "learning_rate": 6.3027708672449775e-06,
      "loss": 0.0561,
      "step": 44530
    },
    {
      "epoch": 4.2467117309570314e-05,
      "model_forward_time": 1.7060844898223877,
      "step": 44530
    },
    {
      "dynamics_loss_avg": 0.063537691347301,
      "epoch": 4.2467117309570314e-05,
      "step": 44530
    },
    {
      "action_loss_avg": 0.0019966173509601503,
      "epoch": 4.2467117309570314e-05,
      "step": 44530
    },
    {
      "epoch": 4.2467117309570314e-05,
      "step": 44530,
      "training_step_time": 7.707262754440308
    },
    {
      "epoch": 4.246807098388672e-05,
      "model_forward_time": 1.716768503189087,
      "step": 44531
    },
    {
      "epoch": 4.246807098388672e-05,
      "step": 44531,
      "training_step_time": 6.907778978347778
    },
    {
      "epoch": 4.246902465820312e-05,
      "model_forward_time": 1.7114577293395996,
      "step": 44532
    },
    {
      "epoch": 4.246902465820312e-05,
      "step": 44532,
      "training_step_time": 7.000429630279541
    },
    {
      "epoch": 4.246997833251953e-05,
      "model_forward_time": 1.7133734226226807,
      "step": 44533
    },
    {
      "epoch": 4.246997833251953e-05,
      "step": 44533,
      "training_step_time": 6.963245391845703
    },
    {
      "epoch": 4.2470932006835936e-05,
      "model_forward_time": 1.695619821548462,
      "step": 44534
    },
    {
      "epoch": 4.2470932006835936e-05,
      "step": 44534,
      "training_step_time": 6.929413318634033
    },
    {
      "epoch": 4.247188568115234e-05,
      "model_forward_time": 1.7133383750915527,
      "step": 44535
    },
    {
      "epoch": 4.247188568115234e-05,
      "step": 44535,
      "training_step_time": 7.417406797409058
    },
    {
      "epoch": 4.247283935546875e-05,
      "model_forward_time": 2.2538869380950928,
      "step": 44536
    },
    {
      "epoch": 4.247283935546875e-05,
      "step": 44536,
      "training_step_time": 7.7790563106536865
    },
    {
      "epoch": 4.2473793029785155e-05,
      "model_forward_time": 1.7358601093292236,
      "step": 44537
    },
    {
      "epoch": 4.2473793029785155e-05,
      "step": 44537,
      "training_step_time": 10.831188440322876
    },
    {
      "epoch": 4.247474670410156e-05,
      "model_forward_time": 1.7116811275482178,
      "step": 44538
    },
    {
      "epoch": 4.247474670410156e-05,
      "step": 44538,
      "training_step_time": 7.82769250869751
    },
    {
      "epoch": 4.247570037841797e-05,
      "model_forward_time": 1.709531307220459,
      "step": 44539
    },
    {
      "epoch": 4.247570037841797e-05,
      "step": 44539,
      "training_step_time": 7.781915664672852
    },
    {
      "epoch": 4.2476654052734374e-05,
      "grad_norm": 0.12999925017356873,
      "learning_rate": 6.301174438344329e-06,
      "loss": 0.0552,
      "step": 44540
    },
    {
      "epoch": 4.2476654052734374e-05,
      "model_forward_time": 1.705108642578125,
      "step": 44540
    },
    {
      "dynamics_loss_avg": 0.03899664655327797,
      "epoch": 4.2476654052734374e-05,
      "step": 44540
    },
    {
      "action_loss_avg": 0.0018360303598456084,
      "epoch": 4.2476654052734374e-05,
      "step": 44540
    },
    {
      "epoch": 4.2476654052734374e-05,
      "step": 44540,
      "training_step_time": 6.940282583236694
    },
    {
      "epoch": 4.247760772705078e-05,
      "model_forward_time": 1.3192059993743896,
      "step": 44541
    },
    {
      "epoch": 4.247760772705078e-05,
      "step": 44541,
      "training_step_time": 6.8546812534332275
    },
    {
      "epoch": 4.247856140136719e-05,
      "model_forward_time": 1.7141854763031006,
      "step": 44542
    },
    {
      "epoch": 4.247856140136719e-05,
      "step": 44542,
      "training_step_time": 6.968814849853516
    },
    {
      "epoch": 4.247951507568359e-05,
      "model_forward_time": 1.7067632675170898,
      "step": 44543
    },
    {
      "epoch": 4.247951507568359e-05,
      "step": 44543,
      "training_step_time": 6.83295464515686
    },
    {
      "epoch": 4.248046875e-05,
      "model_forward_time": 1.713244915008545,
      "step": 44544
    },
    {
      "epoch": 4.248046875e-05,
      "step": 44544,
      "training_step_time": 7.372175693511963
    },
    {
      "epoch": 4.248142242431641e-05,
      "model_forward_time": 2.286736011505127,
      "step": 44545
    },
    {
      "epoch": 4.248142242431641e-05,
      "step": 44545,
      "training_step_time": 7.635618686676025
    },
    {
      "epoch": 4.248237609863281e-05,
      "model_forward_time": 1.7289700508117676,
      "step": 44546
    },
    {
      "epoch": 4.248237609863281e-05,
      "step": 44546,
      "training_step_time": 10.250408172607422
    },
    {
      "epoch": 4.2483329772949216e-05,
      "model_forward_time": 1.7258141040802002,
      "step": 44547
    },
    {
      "epoch": 4.2483329772949216e-05,
      "step": 44547,
      "training_step_time": 9.186490774154663
    },
    {
      "epoch": 4.248428344726563e-05,
      "model_forward_time": 1.7118256092071533,
      "step": 44548
    },
    {
      "epoch": 4.248428344726563e-05,
      "step": 44548,
      "training_step_time": 7.814604759216309
    },
    {
      "epoch": 4.248523712158203e-05,
      "model_forward_time": 1.719569444656372,
      "step": 44549
    },
    {
      "epoch": 4.248523712158203e-05,
      "step": 44549,
      "training_step_time": 7.7257630825042725
    },
    {
      "epoch": 4.2486190795898435e-05,
      "grad_norm": 0.12101420015096664,
      "learning_rate": 6.299577867149202e-06,
      "loss": 0.0597,
      "step": 44550
    },
    {
      "epoch": 4.2486190795898435e-05,
      "model_forward_time": 1.7053189277648926,
      "step": 44550
    },
    {
      "dynamics_loss_avg": 0.06140443049371243,
      "epoch": 4.2486190795898435e-05,
      "step": 44550
    },
    {
      "action_loss_avg": 0.0015121033589821308,
      "epoch": 4.2486190795898435e-05,
      "step": 44550
    },
    {
      "epoch": 4.2486190795898435e-05,
      "step": 44550,
      "training_step_time": 6.923281669616699
    },
    {
      "epoch": 4.2487144470214846e-05,
      "model_forward_time": 1.7145919799804688,
      "step": 44551
    },
    {
      "epoch": 4.2487144470214846e-05,
      "step": 44551,
      "training_step_time": 6.953708648681641
    },
    {
      "epoch": 4.248809814453125e-05,
      "model_forward_time": 1.7127511501312256,
      "step": 44552
    },
    {
      "epoch": 4.248809814453125e-05,
      "step": 44552,
      "training_step_time": 6.866077661514282
    },
    {
      "epoch": 4.2489051818847654e-05,
      "model_forward_time": 1.7090842723846436,
      "step": 44553
    },
    {
      "epoch": 4.2489051818847654e-05,
      "step": 44553,
      "training_step_time": 7.381572246551514
    },
    {
      "epoch": 4.2490005493164064e-05,
      "model_forward_time": 2.277613878250122,
      "step": 44554
    },
    {
      "epoch": 4.2490005493164064e-05,
      "step": 44554,
      "training_step_time": 7.600982904434204
    },
    {
      "epoch": 4.249095916748047e-05,
      "model_forward_time": 1.7214126586914062,
      "step": 44555
    },
    {
      "epoch": 4.249095916748047e-05,
      "step": 44555,
      "training_step_time": 7.793936491012573
    },
    {
      "epoch": 4.249191284179687e-05,
      "model_forward_time": 1.7174663543701172,
      "step": 44556
    },
    {
      "epoch": 4.249191284179687e-05,
      "step": 44556,
      "training_step_time": 8.733153343200684
    },
    {
      "epoch": 4.249286651611328e-05,
      "model_forward_time": 1.713869333267212,
      "step": 44557
    },
    {
      "epoch": 4.249286651611328e-05,
      "step": 44557,
      "training_step_time": 8.423474311828613
    },
    {
      "epoch": 4.249382019042969e-05,
      "model_forward_time": 1.709632158279419,
      "step": 44558
    },
    {
      "epoch": 4.249382019042969e-05,
      "step": 44558,
      "training_step_time": 7.715742826461792
    },
    {
      "epoch": 4.249477386474609e-05,
      "model_forward_time": 1.7165462970733643,
      "step": 44559
    },
    {
      "epoch": 4.249477386474609e-05,
      "step": 44559,
      "training_step_time": 6.941715240478516
    },
    {
      "epoch": 4.24957275390625e-05,
      "grad_norm": 0.11386626958847046,
      "learning_rate": 6.297981153834193e-06,
      "loss": 0.0591,
      "step": 44560
    },
    {
      "epoch": 4.24957275390625e-05,
      "model_forward_time": 1.7083005905151367,
      "step": 44560
    },
    {
      "dynamics_loss_avg": 0.06084512770175934,
      "epoch": 4.24957275390625e-05,
      "step": 44560
    },
    {
      "action_loss_avg": 0.0012841677671531215,
      "epoch": 4.24957275390625e-05,
      "step": 44560
    },
    {
      "epoch": 4.24957275390625e-05,
      "step": 44560,
      "training_step_time": 6.952190399169922
    },
    {
      "epoch": 4.2496681213378906e-05,
      "model_forward_time": 1.7099525928497314,
      "step": 44561
    },
    {
      "epoch": 4.2496681213378906e-05,
      "step": 44561,
      "training_step_time": 6.913538455963135
    },
    {
      "epoch": 4.249763488769531e-05,
      "model_forward_time": 1.7082905769348145,
      "step": 44562
    },
    {
      "epoch": 4.249763488769531e-05,
      "step": 44562,
      "training_step_time": 7.394328594207764
    },
    {
      "epoch": 4.249858856201172e-05,
      "model_forward_time": 1.703660249710083,
      "step": 44563
    },
    {
      "epoch": 4.249858856201172e-05,
      "step": 44563,
      "training_step_time": 8.10745906829834
    },
    {
      "epoch": 4.2499542236328125e-05,
      "model_forward_time": 1.717193603515625,
      "step": 44564
    },
    {
      "epoch": 4.2499542236328125e-05,
      "step": 44564,
      "training_step_time": 8.307213544845581
    },
    {
      "epoch": 4.250049591064453e-05,
      "model_forward_time": 1.7121219635009766,
      "step": 44565
    },
    {
      "epoch": 4.250049591064453e-05,
      "step": 44565,
      "training_step_time": 9.830469369888306
    },
    {
      "epoch": 4.250144958496094e-05,
      "model_forward_time": 1.7298808097839355,
      "step": 44566
    },
    {
      "epoch": 4.250144958496094e-05,
      "step": 44566,
      "training_step_time": 7.872167348861694
    },
    {
      "epoch": 4.2502403259277344e-05,
      "model_forward_time": 1.7139618396759033,
      "step": 44567
    },
    {
      "epoch": 4.2502403259277344e-05,
      "step": 44567,
      "training_step_time": 7.913775682449341
    },
    {
      "epoch": 4.250335693359375e-05,
      "model_forward_time": 1.7151331901550293,
      "step": 44568
    },
    {
      "epoch": 4.250335693359375e-05,
      "step": 44568,
      "training_step_time": 6.8105103969573975
    },
    {
      "epoch": 4.250431060791016e-05,
      "model_forward_time": 1.7111477851867676,
      "step": 44569
    },
    {
      "epoch": 4.250431060791016e-05,
      "step": 44569,
      "training_step_time": 6.921429395675659
    },
    {
      "epoch": 4.250526428222656e-05,
      "grad_norm": 0.11680477857589722,
      "learning_rate": 6.296384298573916e-06,
      "loss": 0.0563,
      "step": 44570
    },
    {
      "epoch": 4.250526428222656e-05,
      "model_forward_time": 1.6938321590423584,
      "step": 44570
    },
    {
      "dynamics_loss_avg": 0.05787428542971611,
      "epoch": 4.250526428222656e-05,
      "step": 44570
    },
    {
      "action_loss_avg": 0.0020025075296871363,
      "epoch": 4.250526428222656e-05,
      "step": 44570
    },
    {
      "epoch": 4.250526428222656e-05,
      "step": 44570,
      "training_step_time": 6.890505790710449
    },
    {
      "epoch": 4.250621795654297e-05,
      "model_forward_time": 1.7033476829528809,
      "step": 44571
    },
    {
      "epoch": 4.250621795654297e-05,
      "step": 44571,
      "training_step_time": 7.364806652069092
    },
    {
      "epoch": 4.250717163085938e-05,
      "model_forward_time": 1.7095057964324951,
      "step": 44572
    },
    {
      "epoch": 4.250717163085938e-05,
      "step": 44572,
      "training_step_time": 7.694514751434326
    },
    {
      "epoch": 4.250812530517578e-05,
      "model_forward_time": 2.269300699234009,
      "step": 44573
    },
    {
      "epoch": 4.250812530517578e-05,
      "step": 44573,
      "training_step_time": 7.775879621505737
    },
    {
      "epoch": 4.2509078979492186e-05,
      "model_forward_time": 1.7087554931640625,
      "step": 44574
    },
    {
      "epoch": 4.2509078979492186e-05,
      "step": 44574,
      "training_step_time": 8.960089921951294
    },
    {
      "epoch": 4.2510032653808596e-05,
      "model_forward_time": 1.7225418090820312,
      "step": 44575
    },
    {
      "epoch": 4.2510032653808596e-05,
      "step": 44575,
      "training_step_time": 8.501398801803589
    },
    {
      "epoch": 4.2510986328125e-05,
      "model_forward_time": 1.7123839855194092,
      "step": 44576
    },
    {
      "epoch": 4.2510986328125e-05,
      "step": 44576,
      "training_step_time": 7.884047031402588
    },
    {
      "epoch": 4.2511940002441405e-05,
      "model_forward_time": 1.3369169235229492,
      "step": 44577
    },
    {
      "epoch": 4.2511940002441405e-05,
      "step": 44577,
      "training_step_time": 7.717319965362549
    },
    {
      "epoch": 4.2512893676757815e-05,
      "model_forward_time": 1.70924973487854,
      "step": 44578
    },
    {
      "epoch": 4.2512893676757815e-05,
      "step": 44578,
      "training_step_time": 7.016694068908691
    },
    {
      "epoch": 4.251384735107422e-05,
      "model_forward_time": 1.7106499671936035,
      "step": 44579
    },
    {
      "epoch": 4.251384735107422e-05,
      "step": 44579,
      "training_step_time": 6.9866039752960205
    },
    {
      "epoch": 4.251480102539062e-05,
      "grad_norm": 0.08717866241931915,
      "learning_rate": 6.294787301543001e-06,
      "loss": 0.0617,
      "step": 44580
    },
    {
      "epoch": 4.251480102539062e-05,
      "model_forward_time": 1.7034339904785156,
      "step": 44580
    },
    {
      "dynamics_loss_avg": 0.04741642596200109,
      "epoch": 4.251480102539062e-05,
      "step": 44580
    },
    {
      "action_loss_avg": 0.0017007569025736303,
      "epoch": 4.251480102539062e-05,
      "step": 44580
    },
    {
      "epoch": 4.251480102539062e-05,
      "step": 44580,
      "training_step_time": 7.456131935119629
    },
    {
      "epoch": 4.2515754699707034e-05,
      "model_forward_time": 1.709071397781372,
      "step": 44581
    },
    {
      "epoch": 4.2515754699707034e-05,
      "step": 44581,
      "training_step_time": 7.725703954696655
    },
    {
      "epoch": 4.251670837402344e-05,
      "model_forward_time": 1.8909292221069336,
      "step": 44582
    },
    {
      "epoch": 4.251670837402344e-05,
      "step": 44582,
      "training_step_time": 8.758585453033447
    },
    {
      "epoch": 4.251766204833984e-05,
      "model_forward_time": 1.7161283493041992,
      "step": 44583
    },
    {
      "epoch": 4.251766204833984e-05,
      "step": 44583,
      "training_step_time": 8.923256397247314
    },
    {
      "epoch": 4.251861572265625e-05,
      "model_forward_time": 1.7423362731933594,
      "step": 44584
    },
    {
      "epoch": 4.251861572265625e-05,
      "step": 44584,
      "training_step_time": 7.785328388214111
    },
    {
      "epoch": 4.251956939697266e-05,
      "model_forward_time": 1.710284948348999,
      "step": 44585
    },
    {
      "epoch": 4.251956939697266e-05,
      "step": 44585,
      "training_step_time": 7.863302230834961
    },
    {
      "epoch": 4.252052307128906e-05,
      "model_forward_time": 1.7061331272125244,
      "step": 44586
    },
    {
      "epoch": 4.252052307128906e-05,
      "step": 44586,
      "training_step_time": 7.664366722106934
    },
    {
      "epoch": 4.252147674560547e-05,
      "model_forward_time": 1.7087912559509277,
      "step": 44587
    },
    {
      "epoch": 4.252147674560547e-05,
      "step": 44587,
      "training_step_time": 7.011008262634277
    },
    {
      "epoch": 4.2522430419921876e-05,
      "model_forward_time": 1.7115912437438965,
      "step": 44588
    },
    {
      "epoch": 4.2522430419921876e-05,
      "step": 44588,
      "training_step_time": 6.984084129333496
    },
    {
      "epoch": 4.252338409423828e-05,
      "model_forward_time": 1.710254430770874,
      "step": 44589
    },
    {
      "epoch": 4.252338409423828e-05,
      "step": 44589,
      "training_step_time": 7.453617095947266
    },
    {
      "epoch": 4.252433776855469e-05,
      "grad_norm": 0.2424129694700241,
      "learning_rate": 6.293190162916096e-06,
      "loss": 0.0572,
      "step": 44590
    },
    {
      "epoch": 4.252433776855469e-05,
      "model_forward_time": 1.7004046440124512,
      "step": 44590
    },
    {
      "dynamics_loss_avg": 0.06154160033911467,
      "epoch": 4.252433776855469e-05,
      "step": 44590
    },
    {
      "action_loss_avg": 0.001688265142729506,
      "epoch": 4.252433776855469e-05,
      "step": 44590
    },
    {
      "epoch": 4.252433776855469e-05,
      "step": 44590,
      "training_step_time": 7.71946120262146
    },
    {
      "epoch": 4.2525291442871095e-05,
      "model_forward_time": 2.2779765129089355,
      "step": 44591
    },
    {
      "epoch": 4.2525291442871095e-05,
      "step": 44591,
      "training_step_time": 7.761186361312866
    },
    {
      "epoch": 4.25262451171875e-05,
      "model_forward_time": 1.769118309020996,
      "step": 44592
    },
    {
      "epoch": 4.25262451171875e-05,
      "step": 44592,
      "training_step_time": 9.487391471862793
    },
    {
      "epoch": 4.252719879150391e-05,
      "model_forward_time": 1.7197072505950928,
      "step": 44593
    },
    {
      "epoch": 4.252719879150391e-05,
      "step": 44593,
      "training_step_time": 8.488906383514404
    },
    {
      "epoch": 4.2528152465820314e-05,
      "model_forward_time": 1.7315921783447266,
      "step": 44594
    },
    {
      "epoch": 4.2528152465820314e-05,
      "step": 44594,
      "training_step_time": 8.508068561553955
    },
    {
      "epoch": 4.252910614013672e-05,
      "model_forward_time": 1.7186737060546875,
      "step": 44595
    },
    {
      "epoch": 4.252910614013672e-05,
      "step": 44595,
      "training_step_time": 7.961129426956177
    },
    {
      "epoch": 4.253005981445312e-05,
      "model_forward_time": 1.7163405418395996,
      "step": 44596
    },
    {
      "epoch": 4.253005981445312e-05,
      "step": 44596,
      "training_step_time": 6.929426908493042
    },
    {
      "epoch": 4.253101348876953e-05,
      "model_forward_time": 1.724320650100708,
      "step": 44597
    },
    {
      "epoch": 4.253101348876953e-05,
      "step": 44597,
      "training_step_time": 6.982658386230469
    },
    {
      "epoch": 4.2531967163085937e-05,
      "model_forward_time": 1.716799020767212,
      "step": 44598
    },
    {
      "epoch": 4.2531967163085937e-05,
      "step": 44598,
      "training_step_time": 7.428698539733887
    },
    {
      "epoch": 4.253292083740234e-05,
      "model_forward_time": 1.7144019603729248,
      "step": 44599
    },
    {
      "epoch": 4.253292083740234e-05,
      "step": 44599,
      "training_step_time": 7.719911336898804
    },
    {
      "epoch": 4.253387451171875e-05,
      "grad_norm": 0.10595536231994629,
      "learning_rate": 6.291592882867855e-06,
      "loss": 0.0586,
      "step": 44600
    },
    {
      "epoch": 4.253387451171875e-05,
      "model_forward_time": 1.7114179134368896,
      "step": 44600
    },
    {
      "dynamics_loss_avg": 0.06113667022436857,
      "epoch": 4.253387451171875e-05,
      "step": 44600
    },
    {
      "action_loss_avg": 0.0018275587062817067,
      "epoch": 4.253387451171875e-05,
      "step": 44600
    },
    {
      "epoch": 4.253387451171875e-05,
      "step": 44600,
      "training_step_time": 8.201496124267578
    },
    {
      "epoch": 4.2534828186035155e-05,
      "model_forward_time": 1.7211730480194092,
      "step": 44601
    },
    {
      "epoch": 4.2534828186035155e-05,
      "step": 44601,
      "training_step_time": 9.356581449508667
    },
    {
      "epoch": 4.253578186035156e-05,
      "model_forward_time": 1.7189295291900635,
      "step": 44602
    },
    {
      "epoch": 4.253578186035156e-05,
      "step": 44602,
      "training_step_time": 10.304945230484009
    },
    {
      "epoch": 4.253673553466797e-05,
      "model_forward_time": 1.734358549118042,
      "step": 44603
    },
    {
      "epoch": 4.253673553466797e-05,
      "step": 44603,
      "training_step_time": 8.45304799079895
    },
    {
      "epoch": 4.2537689208984374e-05,
      "model_forward_time": 1.7207369804382324,
      "step": 44604
    },
    {
      "epoch": 4.2537689208984374e-05,
      "step": 44604,
      "training_step_time": 7.878958702087402
    },
    {
      "epoch": 4.253864288330078e-05,
      "model_forward_time": 1.7227518558502197,
      "step": 44605
    },
    {
      "epoch": 4.253864288330078e-05,
      "step": 44605,
      "training_step_time": 7.6973042488098145
    },
    {
      "epoch": 4.253959655761719e-05,
      "model_forward_time": 1.719386339187622,
      "step": 44606
    },
    {
      "epoch": 4.253959655761719e-05,
      "step": 44606,
      "training_step_time": 6.953454494476318
    },
    {
      "epoch": 4.254055023193359e-05,
      "model_forward_time": 1.7171237468719482,
      "step": 44607
    },
    {
      "epoch": 4.254055023193359e-05,
      "step": 44607,
      "training_step_time": 7.53301215171814
    },
    {
      "epoch": 4.254150390625e-05,
      "model_forward_time": 1.7190449237823486,
      "step": 44608
    },
    {
      "epoch": 4.254150390625e-05,
      "step": 44608,
      "training_step_time": 7.772099256515503
    },
    {
      "epoch": 4.254245758056641e-05,
      "model_forward_time": 1.7273242473602295,
      "step": 44609
    },
    {
      "epoch": 4.254245758056641e-05,
      "step": 44609,
      "training_step_time": 7.706410884857178
    },
    {
      "epoch": 4.254341125488281e-05,
      "grad_norm": 0.12959304451942444,
      "learning_rate": 6.2899954615729605e-06,
      "loss": 0.0593,
      "step": 44610
    },
    {
      "epoch": 4.254341125488281e-05,
      "model_forward_time": 2.2706551551818848,
      "step": 44610
    },
    {
      "dynamics_loss_avg": 0.045551348477602005,
      "epoch": 4.254341125488281e-05,
      "step": 44610
    },
    {
      "action_loss_avg": 0.0016024651355110108,
      "epoch": 4.254341125488281e-05,
      "step": 44610
    },
    {
      "epoch": 4.254341125488281e-05,
      "step": 44610,
      "training_step_time": 9.428330659866333
    },
    {
      "epoch": 4.2544364929199216e-05,
      "model_forward_time": 1.7189302444458008,
      "step": 44611
    },
    {
      "epoch": 4.2544364929199216e-05,
      "step": 44611,
      "training_step_time": 7.916130065917969
    },
    {
      "epoch": 4.254531860351563e-05,
      "model_forward_time": 1.7327191829681396,
      "step": 44612
    },
    {
      "epoch": 4.254531860351563e-05,
      "step": 44612,
      "training_step_time": 8.564749002456665
    },
    {
      "epoch": 4.254627227783203e-05,
      "model_forward_time": 1.7201828956604004,
      "step": 44613
    },
    {
      "epoch": 4.254627227783203e-05,
      "step": 44613,
      "training_step_time": 7.838699817657471
    },
    {
      "epoch": 4.2547225952148435e-05,
      "model_forward_time": 1.7212178707122803,
      "step": 44614
    },
    {
      "epoch": 4.2547225952148435e-05,
      "step": 44614,
      "training_step_time": 7.668723821640015
    },
    {
      "epoch": 4.2548179626464846e-05,
      "model_forward_time": 1.7354402542114258,
      "step": 44615
    },
    {
      "epoch": 4.2548179626464846e-05,
      "step": 44615,
      "training_step_time": 6.943557024002075
    },
    {
      "epoch": 4.254913330078125e-05,
      "model_forward_time": 1.7324116230010986,
      "step": 44616
    },
    {
      "epoch": 4.254913330078125e-05,
      "step": 44616,
      "training_step_time": 7.443726539611816
    },
    {
      "epoch": 4.2550086975097654e-05,
      "model_forward_time": 1.7359817028045654,
      "step": 44617
    },
    {
      "epoch": 4.2550086975097654e-05,
      "step": 44617,
      "training_step_time": 7.769504547119141
    },
    {
      "epoch": 4.2551040649414065e-05,
      "model_forward_time": 1.7343294620513916,
      "step": 44618
    },
    {
      "epoch": 4.2551040649414065e-05,
      "step": 44618,
      "training_step_time": 7.767663955688477
    },
    {
      "epoch": 4.255199432373047e-05,
      "model_forward_time": 2.2990169525146484,
      "step": 44619
    },
    {
      "epoch": 4.255199432373047e-05,
      "step": 44619,
      "training_step_time": 8.52436351776123
    },
    {
      "epoch": 4.255294799804687e-05,
      "grad_norm": 0.13989318907260895,
      "learning_rate": 6.2883978992061025e-06,
      "loss": 0.0629,
      "step": 44620
    },
    {
      "epoch": 4.255294799804687e-05,
      "model_forward_time": 1.756272554397583,
      "step": 44620
    },
    {
      "dynamics_loss_avg": 0.06046601999551058,
      "epoch": 4.255294799804687e-05,
      "step": 44620
    },
    {
      "action_loss_avg": 0.0018364656600169838,
      "epoch": 4.255294799804687e-05,
      "step": 44620
    },
    {
      "epoch": 4.255294799804687e-05,
      "step": 44620,
      "training_step_time": 8.857586145401001
    },
    {
      "epoch": 4.255390167236328e-05,
      "model_forward_time": 1.7408056259155273,
      "step": 44621
    },
    {
      "epoch": 4.255390167236328e-05,
      "step": 44621,
      "training_step_time": 7.774521112442017
    },
    {
      "epoch": 4.255485534667969e-05,
      "model_forward_time": 1.7319912910461426,
      "step": 44622
    },
    {
      "epoch": 4.255485534667969e-05,
      "step": 44622,
      "training_step_time": 7.891922235488892
    },
    {
      "epoch": 4.255580902099609e-05,
      "model_forward_time": 1.7348740100860596,
      "step": 44623
    },
    {
      "epoch": 4.255580902099609e-05,
      "step": 44623,
      "training_step_time": 6.9523797035217285
    },
    {
      "epoch": 4.25567626953125e-05,
      "model_forward_time": 1.7149643898010254,
      "step": 44624
    },
    {
      "epoch": 4.25567626953125e-05,
      "step": 44624,
      "training_step_time": 7.726049900054932
    },
    {
      "epoch": 4.2557716369628906e-05,
      "model_forward_time": 1.7354412078857422,
      "step": 44625
    },
    {
      "epoch": 4.2557716369628906e-05,
      "step": 44625,
      "training_step_time": 7.412572622299194
    },
    {
      "epoch": 4.255867004394531e-05,
      "model_forward_time": 1.7303979396820068,
      "step": 44626
    },
    {
      "epoch": 4.255867004394531e-05,
      "step": 44626,
      "training_step_time": 7.744867563247681
    },
    {
      "epoch": 4.255962371826172e-05,
      "model_forward_time": 1.7330694198608398,
      "step": 44627
    },
    {
      "epoch": 4.255962371826172e-05,
      "step": 44627,
      "training_step_time": 7.710401296615601
    },
    {
      "epoch": 4.2560577392578125e-05,
      "model_forward_time": 2.3331050872802734,
      "step": 44628
    },
    {
      "epoch": 4.2560577392578125e-05,
      "step": 44628,
      "training_step_time": 8.810437679290771
    },
    {
      "epoch": 4.256153106689453e-05,
      "model_forward_time": 1.7418382167816162,
      "step": 44629
    },
    {
      "epoch": 4.256153106689453e-05,
      "step": 44629,
      "training_step_time": 8.910997152328491
    },
    {
      "epoch": 4.256248474121094e-05,
      "grad_norm": 0.10787405073642731,
      "learning_rate": 6.286800195941984e-06,
      "loss": 0.0554,
      "step": 44630
    },
    {
      "epoch": 4.256248474121094e-05,
      "model_forward_time": 1.7306935787200928,
      "step": 44630
    },
    {
      "dynamics_loss_avg": 0.07112715784460306,
      "epoch": 4.256248474121094e-05,
      "step": 44630
    },
    {
      "action_loss_avg": 0.001700923638418317,
      "epoch": 4.256248474121094e-05,
      "step": 44630
    },
    {
      "epoch": 4.256248474121094e-05,
      "step": 44630,
      "training_step_time": 7.802160024642944
    },
    {
      "epoch": 4.2563438415527344e-05,
      "model_forward_time": 1.734065055847168,
      "step": 44631
    },
    {
      "epoch": 4.2563438415527344e-05,
      "step": 44631,
      "training_step_time": 8.667088270187378
    },
    {
      "epoch": 4.256439208984375e-05,
      "model_forward_time": 1.7192959785461426,
      "step": 44632
    },
    {
      "epoch": 4.256439208984375e-05,
      "step": 44632,
      "training_step_time": 7.025922536849976
    },
    {
      "epoch": 4.256534576416016e-05,
      "model_forward_time": 1.7257487773895264,
      "step": 44633
    },
    {
      "epoch": 4.256534576416016e-05,
      "step": 44633,
      "training_step_time": 7.7539756298065186
    },
    {
      "epoch": 4.256629943847656e-05,
      "model_forward_time": 1.7155640125274658,
      "step": 44634
    },
    {
      "epoch": 4.256629943847656e-05,
      "step": 44634,
      "training_step_time": 7.214852809906006
    },
    {
      "epoch": 4.256725311279297e-05,
      "model_forward_time": 1.7191545963287354,
      "step": 44635
    },
    {
      "epoch": 4.256725311279297e-05,
      "step": 44635,
      "training_step_time": 7.490039348602295
    },
    {
      "epoch": 4.256820678710938e-05,
      "model_forward_time": 1.7119464874267578,
      "step": 44636
    },
    {
      "epoch": 4.256820678710938e-05,
      "step": 44636,
      "training_step_time": 7.699899196624756
    },
    {
      "epoch": 4.256916046142578e-05,
      "model_forward_time": 2.3031411170959473,
      "step": 44637
    },
    {
      "epoch": 4.256916046142578e-05,
      "step": 44637,
      "training_step_time": 9.102397203445435
    },
    {
      "epoch": 4.2570114135742186e-05,
      "model_forward_time": 1.7235586643218994,
      "step": 44638
    },
    {
      "epoch": 4.2570114135742186e-05,
      "step": 44638,
      "training_step_time": 8.402376890182495
    },
    {
      "epoch": 4.2571067810058597e-05,
      "model_forward_time": 1.7199149131774902,
      "step": 44639
    },
    {
      "epoch": 4.2571067810058597e-05,
      "step": 44639,
      "training_step_time": 10.225056886672974
    },
    {
      "epoch": 4.2572021484375e-05,
      "grad_norm": 0.13082127273082733,
      "learning_rate": 6.285202351955335e-06,
      "loss": 0.0582,
      "step": 44640
    },
    {
      "epoch": 4.2572021484375e-05,
      "model_forward_time": 1.730062484741211,
      "step": 44640
    },
    {
      "dynamics_loss_avg": 0.06435217168182135,
      "epoch": 4.2572021484375e-05,
      "step": 44640
    },
    {
      "action_loss_avg": 0.0023320641717873513,
      "epoch": 4.2572021484375e-05,
      "step": 44640
    },
    {
      "epoch": 4.2572021484375e-05,
      "step": 44640,
      "training_step_time": 7.855715751647949
    },
    {
      "epoch": 4.2572975158691405e-05,
      "model_forward_time": 1.7176003456115723,
      "step": 44641
    },
    {
      "epoch": 4.2572975158691405e-05,
      "step": 44641,
      "training_step_time": 7.91446852684021
    },
    {
      "epoch": 4.2573928833007815e-05,
      "model_forward_time": 1.7188067436218262,
      "step": 44642
    },
    {
      "epoch": 4.2573928833007815e-05,
      "step": 44642,
      "training_step_time": 7.665041208267212
    },
    {
      "epoch": 4.257488250732422e-05,
      "model_forward_time": 1.7214436531066895,
      "step": 44643
    },
    {
      "epoch": 4.257488250732422e-05,
      "step": 44643,
      "training_step_time": 7.174866676330566
    },
    {
      "epoch": 4.2575836181640623e-05,
      "model_forward_time": 1.7339379787445068,
      "step": 44644
    },
    {
      "epoch": 4.2575836181640623e-05,
      "step": 44644,
      "training_step_time": 7.550317764282227
    },
    {
      "epoch": 4.2576789855957034e-05,
      "model_forward_time": 1.7206130027770996,
      "step": 44645
    },
    {
      "epoch": 4.2576789855957034e-05,
      "step": 44645,
      "training_step_time": 7.7039220333099365
    },
    {
      "epoch": 4.257774353027344e-05,
      "model_forward_time": 2.280179262161255,
      "step": 44646
    },
    {
      "epoch": 4.257774353027344e-05,
      "step": 44646,
      "training_step_time": 7.832174301147461
    },
    {
      "epoch": 4.257869720458984e-05,
      "model_forward_time": 1.7318782806396484,
      "step": 44647
    },
    {
      "epoch": 4.257869720458984e-05,
      "step": 44647,
      "training_step_time": 9.680300235748291
    },
    {
      "epoch": 4.257965087890625e-05,
      "model_forward_time": 1.7336947917938232,
      "step": 44648
    },
    {
      "epoch": 4.257965087890625e-05,
      "step": 44648,
      "training_step_time": 9.617870092391968
    },
    {
      "epoch": 4.258060455322266e-05,
      "model_forward_time": 1.7479350566864014,
      "step": 44649
    },
    {
      "epoch": 4.258060455322266e-05,
      "step": 44649,
      "training_step_time": 8.868470191955566
    },
    {
      "epoch": 4.258155822753906e-05,
      "grad_norm": 0.13027094304561615,
      "learning_rate": 6.283604367420887e-06,
      "loss": 0.0626,
      "step": 44650
    },
    {
      "epoch": 4.258155822753906e-05,
      "model_forward_time": 1.7367384433746338,
      "step": 44650
    },
    {
      "dynamics_loss_avg": 0.0613176167011261,
      "epoch": 4.258155822753906e-05,
      "step": 44650
    },
    {
      "action_loss_avg": 0.001441733562387526,
      "epoch": 4.258155822753906e-05,
      "step": 44650
    },
    {
      "epoch": 4.258155822753906e-05,
      "step": 44650,
      "training_step_time": 7.77518105506897
    },
    {
      "epoch": 4.258251190185547e-05,
      "model_forward_time": 1.7181744575500488,
      "step": 44651
    },
    {
      "epoch": 4.258251190185547e-05,
      "step": 44651,
      "training_step_time": 6.964260578155518
    },
    {
      "epoch": 4.2583465576171876e-05,
      "model_forward_time": 1.7231924533843994,
      "step": 44652
    },
    {
      "epoch": 4.2583465576171876e-05,
      "step": 44652,
      "training_step_time": 7.263686656951904
    },
    {
      "epoch": 4.258441925048828e-05,
      "model_forward_time": 1.7297041416168213,
      "step": 44653
    },
    {
      "epoch": 4.258441925048828e-05,
      "step": 44653,
      "training_step_time": 7.468560218811035
    },
    {
      "epoch": 4.258537292480469e-05,
      "model_forward_time": 1.7186017036437988,
      "step": 44654
    },
    {
      "epoch": 4.258537292480469e-05,
      "step": 44654,
      "training_step_time": 7.623865842819214
    },
    {
      "epoch": 4.2586326599121095e-05,
      "model_forward_time": 1.724393367767334,
      "step": 44655
    },
    {
      "epoch": 4.2586326599121095e-05,
      "step": 44655,
      "training_step_time": 7.803533315658569
    },
    {
      "epoch": 4.25872802734375e-05,
      "model_forward_time": 2.262253999710083,
      "step": 44656
    },
    {
      "epoch": 4.25872802734375e-05,
      "step": 44656,
      "training_step_time": 8.504974603652954
    },
    {
      "epoch": 4.25882339477539e-05,
      "model_forward_time": 1.7334659099578857,
      "step": 44657
    },
    {
      "epoch": 4.25882339477539e-05,
      "step": 44657,
      "training_step_time": 8.853887796401978
    },
    {
      "epoch": 4.2589187622070314e-05,
      "model_forward_time": 1.714855432510376,
      "step": 44658
    },
    {
      "epoch": 4.2589187622070314e-05,
      "step": 44658,
      "training_step_time": 7.836791753768921
    },
    {
      "epoch": 4.259014129638672e-05,
      "model_forward_time": 1.712472915649414,
      "step": 44659
    },
    {
      "epoch": 4.259014129638672e-05,
      "step": 44659,
      "training_step_time": 7.815793514251709
    },
    {
      "epoch": 4.259109497070312e-05,
      "grad_norm": 0.14436013996601105,
      "learning_rate": 6.282006242513394e-06,
      "loss": 0.0597,
      "step": 44660
    },
    {
      "epoch": 4.259109497070312e-05,
      "model_forward_time": 1.7120237350463867,
      "step": 44660
    },
    {
      "dynamics_loss_avg": 0.062458396144211295,
      "epoch": 4.259109497070312e-05,
      "step": 44660
    },
    {
      "action_loss_avg": 0.0019390879082493485,
      "epoch": 4.259109497070312e-05,
      "step": 44660
    },
    {
      "epoch": 4.259109497070312e-05,
      "step": 44660,
      "training_step_time": 6.898710250854492
    },
    {
      "epoch": 4.259204864501953e-05,
      "model_forward_time": 1.7173545360565186,
      "step": 44661
    },
    {
      "epoch": 4.259204864501953e-05,
      "step": 44661,
      "training_step_time": 7.277602434158325
    },
    {
      "epoch": 4.2593002319335937e-05,
      "model_forward_time": 1.7061452865600586,
      "step": 44662
    },
    {
      "epoch": 4.2593002319335937e-05,
      "step": 44662,
      "training_step_time": 7.414907693862915
    },
    {
      "epoch": 4.259395599365234e-05,
      "model_forward_time": 1.7067539691925049,
      "step": 44663
    },
    {
      "epoch": 4.259395599365234e-05,
      "step": 44663,
      "training_step_time": 7.721842527389526
    },
    {
      "epoch": 4.259490966796875e-05,
      "model_forward_time": 1.7124576568603516,
      "step": 44664
    },
    {
      "epoch": 4.259490966796875e-05,
      "step": 44664,
      "training_step_time": 7.755343437194824
    },
    {
      "epoch": 4.2595863342285155e-05,
      "model_forward_time": 2.2569851875305176,
      "step": 44665
    },
    {
      "epoch": 4.2595863342285155e-05,
      "step": 44665,
      "training_step_time": 9.395286560058594
    },
    {
      "epoch": 4.259681701660156e-05,
      "model_forward_time": 1.719515323638916,
      "step": 44666
    },
    {
      "epoch": 4.259681701660156e-05,
      "step": 44666,
      "training_step_time": 9.646194219589233
    },
    {
      "epoch": 4.259777069091797e-05,
      "model_forward_time": 1.7149171829223633,
      "step": 44667
    },
    {
      "epoch": 4.259777069091797e-05,
      "step": 44667,
      "training_step_time": 7.886054277420044
    },
    {
      "epoch": 4.2598724365234374e-05,
      "model_forward_time": 1.7246520519256592,
      "step": 44668
    },
    {
      "epoch": 4.2598724365234374e-05,
      "step": 44668,
      "training_step_time": 7.905844688415527
    },
    {
      "epoch": 4.259967803955078e-05,
      "model_forward_time": 1.707946538925171,
      "step": 44669
    },
    {
      "epoch": 4.259967803955078e-05,
      "step": 44669,
      "training_step_time": 7.9098427295684814
    },
    {
      "epoch": 4.260063171386719e-05,
      "grad_norm": 0.14873944222927094,
      "learning_rate": 6.2804079774076275e-06,
      "loss": 0.0577,
      "step": 44670
    },
    {
      "epoch": 4.260063171386719e-05,
      "model_forward_time": 1.7067232131958008,
      "step": 44670
    },
    {
      "dynamics_loss_avg": 0.06259618820622563,
      "epoch": 4.260063171386719e-05,
      "step": 44670
    },
    {
      "action_loss_avg": 0.0026656786620151253,
      "epoch": 4.260063171386719e-05,
      "step": 44670
    },
    {
      "epoch": 4.260063171386719e-05,
      "step": 44670,
      "training_step_time": 7.272468090057373
    },
    {
      "epoch": 4.260158538818359e-05,
      "model_forward_time": 1.7076990604400635,
      "step": 44671
    },
    {
      "epoch": 4.260158538818359e-05,
      "step": 44671,
      "training_step_time": 7.011805772781372
    },
    {
      "epoch": 4.26025390625e-05,
      "model_forward_time": 1.7052335739135742,
      "step": 44672
    },
    {
      "epoch": 4.26025390625e-05,
      "step": 44672,
      "training_step_time": 7.748788833618164
    },
    {
      "epoch": 4.260349273681641e-05,
      "model_forward_time": 1.7052702903747559,
      "step": 44673
    },
    {
      "epoch": 4.260349273681641e-05,
      "step": 44673,
      "training_step_time": 8.692039251327515
    },
    {
      "epoch": 4.260444641113281e-05,
      "model_forward_time": 2.268367290496826,
      "step": 44674
    },
    {
      "epoch": 4.260444641113281e-05,
      "step": 44674,
      "training_step_time": 8.385843992233276
    },
    {
      "epoch": 4.2605400085449216e-05,
      "model_forward_time": 1.7050771713256836,
      "step": 44675
    },
    {
      "epoch": 4.2605400085449216e-05,
      "step": 44675,
      "training_step_time": 8.618473768234253
    },
    {
      "epoch": 4.260635375976563e-05,
      "model_forward_time": 1.737663745880127,
      "step": 44676
    },
    {
      "epoch": 4.260635375976563e-05,
      "step": 44676,
      "training_step_time": 9.085554838180542
    },
    {
      "epoch": 4.260730743408203e-05,
      "model_forward_time": 1.7347476482391357,
      "step": 44677
    },
    {
      "epoch": 4.260730743408203e-05,
      "step": 44677,
      "training_step_time": 7.900583744049072
    },
    {
      "epoch": 4.2608261108398435e-05,
      "model_forward_time": 1.713017225265503,
      "step": 44678
    },
    {
      "epoch": 4.2608261108398435e-05,
      "step": 44678,
      "training_step_time": 7.8900065422058105
    },
    {
      "epoch": 4.2609214782714846e-05,
      "model_forward_time": 1.726236343383789,
      "step": 44679
    },
    {
      "epoch": 4.2609214782714846e-05,
      "step": 44679,
      "training_step_time": 7.268457651138306
    },
    {
      "epoch": 4.261016845703125e-05,
      "grad_norm": 0.10849466919898987,
      "learning_rate": 6.27880957227837e-06,
      "loss": 0.0619,
      "step": 44680
    },
    {
      "epoch": 4.261016845703125e-05,
      "model_forward_time": 1.7246043682098389,
      "step": 44680
    },
    {
      "dynamics_loss_avg": 0.08005697764456272,
      "epoch": 4.261016845703125e-05,
      "step": 44680
    },
    {
      "action_loss_avg": 0.003121127450140193,
      "epoch": 4.261016845703125e-05,
      "step": 44680
    },
    {
      "epoch": 4.261016845703125e-05,
      "step": 44680,
      "training_step_time": 6.8748955726623535
    },
    {
      "epoch": 4.2611122131347654e-05,
      "model_forward_time": 1.7162575721740723,
      "step": 44681
    },
    {
      "epoch": 4.2611122131347654e-05,
      "step": 44681,
      "training_step_time": 7.71481728553772
    },
    {
      "epoch": 4.2612075805664065e-05,
      "model_forward_time": 1.7198810577392578,
      "step": 44682
    },
    {
      "epoch": 4.2612075805664065e-05,
      "step": 44682,
      "training_step_time": 7.801384687423706
    },
    {
      "epoch": 4.261302947998047e-05,
      "model_forward_time": 2.306366443634033,
      "step": 44683
    },
    {
      "epoch": 4.261302947998047e-05,
      "step": 44683,
      "training_step_time": 9.456562519073486
    },
    {
      "epoch": 4.261398315429687e-05,
      "model_forward_time": 1.7184081077575684,
      "step": 44684
    },
    {
      "epoch": 4.261398315429687e-05,
      "step": 44684,
      "training_step_time": 9.395285367965698
    },
    {
      "epoch": 4.2614936828613283e-05,
      "model_forward_time": 1.7443816661834717,
      "step": 44685
    },
    {
      "epoch": 4.2614936828613283e-05,
      "step": 44685,
      "training_step_time": 9.236816883087158
    },
    {
      "epoch": 4.261589050292969e-05,
      "model_forward_time": 1.7317745685577393,
      "step": 44686
    },
    {
      "epoch": 4.261589050292969e-05,
      "step": 44686,
      "training_step_time": 7.8925323486328125
    },
    {
      "epoch": 4.261684417724609e-05,
      "model_forward_time": 1.729395866394043,
      "step": 44687
    },
    {
      "epoch": 4.261684417724609e-05,
      "step": 44687,
      "training_step_time": 7.918680191040039
    },
    {
      "epoch": 4.26177978515625e-05,
      "model_forward_time": 1.721428394317627,
      "step": 44688
    },
    {
      "epoch": 4.26177978515625e-05,
      "step": 44688,
      "training_step_time": 7.327351093292236
    },
    {
      "epoch": 4.2618751525878906e-05,
      "model_forward_time": 1.7202603816986084,
      "step": 44689
    },
    {
      "epoch": 4.2618751525878906e-05,
      "step": 44689,
      "training_step_time": 6.939516305923462
    },
    {
      "epoch": 4.261970520019531e-05,
      "grad_norm": 0.12236620485782623,
      "learning_rate": 6.277211027300421e-06,
      "loss": 0.0587,
      "step": 44690
    },
    {
      "epoch": 4.261970520019531e-05,
      "model_forward_time": 1.7214674949645996,
      "step": 44690
    },
    {
      "dynamics_loss_avg": 0.055968566611409185,
      "epoch": 4.261970520019531e-05,
      "step": 44690
    },
    {
      "action_loss_avg": 0.0022051985608413814,
      "epoch": 4.261970520019531e-05,
      "step": 44690
    },
    {
      "epoch": 4.261970520019531e-05,
      "step": 44690,
      "training_step_time": 7.757918119430542
    },
    {
      "epoch": 4.262065887451172e-05,
      "model_forward_time": 1.729731559753418,
      "step": 44691
    },
    {
      "epoch": 4.262065887451172e-05,
      "step": 44691,
      "training_step_time": 8.737919569015503
    },
    {
      "epoch": 4.2621612548828125e-05,
      "model_forward_time": 2.2918951511383057,
      "step": 44692
    },
    {
      "epoch": 4.2621612548828125e-05,
      "step": 44692,
      "training_step_time": 7.798929691314697
    },
    {
      "epoch": 4.262256622314453e-05,
      "model_forward_time": 1.731196641921997,
      "step": 44693
    },
    {
      "epoch": 4.262256622314453e-05,
      "step": 44693,
      "training_step_time": 8.751805067062378
    },
    {
      "epoch": 4.262351989746094e-05,
      "model_forward_time": 1.7449984550476074,
      "step": 44694
    },
    {
      "epoch": 4.262351989746094e-05,
      "step": 44694,
      "training_step_time": 9.217126369476318
    },
    {
      "epoch": 4.2624473571777344e-05,
      "model_forward_time": 1.7358653545379639,
      "step": 44695
    },
    {
      "epoch": 4.2624473571777344e-05,
      "step": 44695,
      "training_step_time": 9.123436212539673
    },
    {
      "epoch": 4.262542724609375e-05,
      "model_forward_time": 1.7323956489562988,
      "step": 44696
    },
    {
      "epoch": 4.262542724609375e-05,
      "step": 44696,
      "training_step_time": 7.826639890670776
    },
    {
      "epoch": 4.262638092041016e-05,
      "model_forward_time": 1.7370567321777344,
      "step": 44697
    },
    {
      "epoch": 4.262638092041016e-05,
      "step": 44697,
      "training_step_time": 8.792771816253662
    },
    {
      "epoch": 4.262733459472656e-05,
      "model_forward_time": 1.7065021991729736,
      "step": 44698
    },
    {
      "epoch": 4.262733459472656e-05,
      "step": 44698,
      "training_step_time": 6.9564807415008545
    },
    {
      "epoch": 4.262828826904297e-05,
      "model_forward_time": 1.7210078239440918,
      "step": 44699
    },
    {
      "epoch": 4.262828826904297e-05,
      "step": 44699,
      "training_step_time": 7.7306389808654785
    },
    {
      "epoch": 4.262924194335938e-05,
      "grad_norm": 0.1061929315328598,
      "learning_rate": 6.2756123426485915e-06,
      "loss": 0.0587,
      "step": 44700
    },
    {
      "epoch": 4.262924194335938e-05,
      "model_forward_time": 1.7233095169067383,
      "step": 44700
    },
    {
      "dynamics_loss_avg": 0.05316089950501919,
      "epoch": 4.262924194335938e-05,
      "step": 44700
    },
    {
      "action_loss_avg": 0.0014010791142936795,
      "epoch": 4.262924194335938e-05,
      "step": 44700
    },
    {
      "epoch": 4.262924194335938e-05,
      "step": 44700,
      "training_step_time": 7.696503400802612
    },
    {
      "epoch": 4.263019561767578e-05,
      "model_forward_time": 1.7268774509429932,
      "step": 44701
    },
    {
      "epoch": 4.263019561767578e-05,
      "step": 44701,
      "training_step_time": 7.775453567504883
    },
    {
      "epoch": 4.2631149291992186e-05,
      "model_forward_time": 2.2684590816497803,
      "step": 44702
    },
    {
      "epoch": 4.2631149291992186e-05,
      "step": 44702,
      "training_step_time": 7.705425262451172
    },
    {
      "epoch": 4.2632102966308597e-05,
      "model_forward_time": 1.7324068546295166,
      "step": 44703
    },
    {
      "epoch": 4.2632102966308597e-05,
      "step": 44703,
      "training_step_time": 10.435005187988281
    },
    {
      "epoch": 4.2633056640625e-05,
      "model_forward_time": 1.7271397113800049,
      "step": 44704
    },
    {
      "epoch": 4.2633056640625e-05,
      "step": 44704,
      "training_step_time": 7.826473712921143
    },
    {
      "epoch": 4.2634010314941405e-05,
      "model_forward_time": 1.7369356155395508,
      "step": 44705
    },
    {
      "epoch": 4.2634010314941405e-05,
      "step": 44705,
      "training_step_time": 7.876055479049683
    },
    {
      "epoch": 4.2634963989257815e-05,
      "model_forward_time": 1.727186679840088,
      "step": 44706
    },
    {
      "epoch": 4.2634963989257815e-05,
      "step": 44706,
      "training_step_time": 7.810575246810913
    },
    {
      "epoch": 4.263591766357422e-05,
      "model_forward_time": 1.7288317680358887,
      "step": 44707
    },
    {
      "epoch": 4.263591766357422e-05,
      "step": 44707,
      "training_step_time": 7.313924074172974
    },
    {
      "epoch": 4.2636871337890623e-05,
      "model_forward_time": 1.7198939323425293,
      "step": 44708
    },
    {
      "epoch": 4.2636871337890623e-05,
      "step": 44708,
      "training_step_time": 7.810935735702515
    },
    {
      "epoch": 4.2637825012207034e-05,
      "model_forward_time": 1.7242496013641357,
      "step": 44709
    },
    {
      "epoch": 4.2637825012207034e-05,
      "step": 44709,
      "training_step_time": 7.710757493972778
    },
    {
      "epoch": 4.263877868652344e-05,
      "grad_norm": 0.15601035952568054,
      "learning_rate": 6.274013518497715e-06,
      "loss": 0.0583,
      "step": 44710
    },
    {
      "epoch": 4.263877868652344e-05,
      "model_forward_time": 1.7200040817260742,
      "step": 44710
    },
    {
      "dynamics_loss_avg": 0.05196753703057766,
      "epoch": 4.263877868652344e-05,
      "step": 44710
    },
    {
      "action_loss_avg": 0.00141439690196421,
      "epoch": 4.263877868652344e-05,
      "step": 44710
    },
    {
      "epoch": 4.263877868652344e-05,
      "step": 44710,
      "training_step_time": 7.931931734085083
    },
    {
      "epoch": 4.263973236083984e-05,
      "model_forward_time": 2.2665047645568848,
      "step": 44711
    },
    {
      "epoch": 4.263973236083984e-05,
      "step": 44711,
      "training_step_time": 7.851599454879761
    },
    {
      "epoch": 4.264068603515625e-05,
      "model_forward_time": 1.7184889316558838,
      "step": 44712
    },
    {
      "epoch": 4.264068603515625e-05,
      "step": 44712,
      "training_step_time": 8.97846794128418
    },
    {
      "epoch": 4.264163970947266e-05,
      "model_forward_time": 1.7204785346984863,
      "step": 44713
    },
    {
      "epoch": 4.264163970947266e-05,
      "step": 44713,
      "training_step_time": 8.898690223693848
    },
    {
      "epoch": 4.264259338378906e-05,
      "model_forward_time": 1.7448475360870361,
      "step": 44714
    },
    {
      "epoch": 4.264259338378906e-05,
      "step": 44714,
      "training_step_time": 7.92477560043335
    },
    {
      "epoch": 4.264354705810547e-05,
      "model_forward_time": 1.7260935306549072,
      "step": 44715
    },
    {
      "epoch": 4.264354705810547e-05,
      "step": 44715,
      "training_step_time": 7.929621934890747
    },
    {
      "epoch": 4.2644500732421876e-05,
      "model_forward_time": 1.7221012115478516,
      "step": 44716
    },
    {
      "epoch": 4.2644500732421876e-05,
      "step": 44716,
      "training_step_time": 7.28086519241333
    },
    {
      "epoch": 4.264545440673828e-05,
      "model_forward_time": 1.7118556499481201,
      "step": 44717
    },
    {
      "epoch": 4.264545440673828e-05,
      "step": 44717,
      "training_step_time": 7.691057205200195
    },
    {
      "epoch": 4.264640808105469e-05,
      "model_forward_time": 1.7234866619110107,
      "step": 44718
    },
    {
      "epoch": 4.264640808105469e-05,
      "step": 44718,
      "training_step_time": 7.649446964263916
    },
    {
      "epoch": 4.2647361755371095e-05,
      "model_forward_time": 1.7179014682769775,
      "step": 44719
    },
    {
      "epoch": 4.2647361755371095e-05,
      "step": 44719,
      "training_step_time": 7.800032138824463
    },
    {
      "epoch": 4.26483154296875e-05,
      "grad_norm": 0.12276490032672882,
      "learning_rate": 6.272414555022637e-06,
      "loss": 0.0564,
      "step": 44720
    },
    {
      "epoch": 4.26483154296875e-05,
      "model_forward_time": 2.2617311477661133,
      "step": 44720
    },
    {
      "dynamics_loss_avg": 0.05820615626871586,
      "epoch": 4.26483154296875e-05,
      "step": 44720
    },
    {
      "action_loss_avg": 0.001844259729841724,
      "epoch": 4.26483154296875e-05,
      "step": 44720
    },
    {
      "epoch": 4.26483154296875e-05,
      "step": 44720,
      "training_step_time": 9.396132707595825
    },
    {
      "epoch": 4.26492691040039e-05,
      "model_forward_time": 1.7161109447479248,
      "step": 44721
    },
    {
      "epoch": 4.26492691040039e-05,
      "step": 44721,
      "training_step_time": 9.637646436691284
    },
    {
      "epoch": 4.2650222778320314e-05,
      "model_forward_time": 1.7464723587036133,
      "step": 44722
    },
    {
      "epoch": 4.2650222778320314e-05,
      "step": 44722,
      "training_step_time": 7.909773349761963
    },
    {
      "epoch": 4.265117645263672e-05,
      "model_forward_time": 1.3633544445037842,
      "step": 44723
    },
    {
      "epoch": 4.265117645263672e-05,
      "step": 44723,
      "training_step_time": 7.82400918006897
    },
    {
      "epoch": 4.265213012695312e-05,
      "model_forward_time": 1.721675157546997,
      "step": 44724
    },
    {
      "epoch": 4.265213012695312e-05,
      "step": 44724,
      "training_step_time": 7.814427137374878
    },
    {
      "epoch": 4.265308380126953e-05,
      "model_forward_time": 1.7255687713623047,
      "step": 44725
    },
    {
      "epoch": 4.265308380126953e-05,
      "step": 44725,
      "training_step_time": 7.290403842926025
    },
    {
      "epoch": 4.265403747558594e-05,
      "model_forward_time": 1.721278429031372,
      "step": 44726
    },
    {
      "epoch": 4.265403747558594e-05,
      "step": 44726,
      "training_step_time": 8.190223455429077
    },
    {
      "epoch": 4.265499114990234e-05,
      "model_forward_time": 1.7245099544525146,
      "step": 44727
    },
    {
      "epoch": 4.265499114990234e-05,
      "step": 44727,
      "training_step_time": 7.448720455169678
    },
    {
      "epoch": 4.265594482421875e-05,
      "model_forward_time": 1.7274811267852783,
      "step": 44728
    },
    {
      "epoch": 4.265594482421875e-05,
      "step": 44728,
      "training_step_time": 7.784809112548828
    },
    {
      "epoch": 4.2656898498535155e-05,
      "model_forward_time": 2.27815580368042,
      "step": 44729
    },
    {
      "epoch": 4.2656898498535155e-05,
      "step": 44729,
      "training_step_time": 7.7055864334106445
    },
    {
      "epoch": 4.265785217285156e-05,
      "grad_norm": 0.09126343578100204,
      "learning_rate": 6.2708154523982155e-06,
      "loss": 0.0583,
      "step": 44730
    },
    {
      "epoch": 4.265785217285156e-05,
      "model_forward_time": 1.7276220321655273,
      "step": 44730
    },
    {
      "dynamics_loss_avg": 0.06432852186262608,
      "epoch": 4.265785217285156e-05,
      "step": 44730
    },
    {
      "action_loss_avg": 0.0020687887794338168,
      "epoch": 4.265785217285156e-05,
      "step": 44730
    },
    {
      "epoch": 4.265785217285156e-05,
      "step": 44730,
      "training_step_time": 7.764858961105347
    },
    {
      "epoch": 4.265880584716797e-05,
      "model_forward_time": 1.7399628162384033,
      "step": 44731
    },
    {
      "epoch": 4.265880584716797e-05,
      "step": 44731,
      "training_step_time": 8.43637204170227
    },
    {
      "epoch": 4.2659759521484374e-05,
      "model_forward_time": 1.7289083003997803,
      "step": 44732
    },
    {
      "epoch": 4.2659759521484374e-05,
      "step": 44732,
      "training_step_time": 8.458078384399414
    },
    {
      "epoch": 4.266071319580078e-05,
      "model_forward_time": 1.7309718132019043,
      "step": 44733
    },
    {
      "epoch": 4.266071319580078e-05,
      "step": 44733,
      "training_step_time": 9.473763465881348
    },
    {
      "epoch": 4.266166687011719e-05,
      "model_forward_time": 1.72232985496521,
      "step": 44734
    },
    {
      "epoch": 4.266166687011719e-05,
      "step": 44734,
      "training_step_time": 7.72632098197937
    },
    {
      "epoch": 4.266262054443359e-05,
      "model_forward_time": 1.7233161926269531,
      "step": 44735
    },
    {
      "epoch": 4.266262054443359e-05,
      "step": 44735,
      "training_step_time": 7.528677701950073
    },
    {
      "epoch": 4.266357421875e-05,
      "model_forward_time": 1.7147088050842285,
      "step": 44736
    },
    {
      "epoch": 4.266357421875e-05,
      "step": 44736,
      "training_step_time": 7.43284010887146
    },
    {
      "epoch": 4.266452789306641e-05,
      "model_forward_time": 1.7245876789093018,
      "step": 44737
    },
    {
      "epoch": 4.266452789306641e-05,
      "step": 44737,
      "training_step_time": 7.7252256870269775
    },
    {
      "epoch": 4.266548156738281e-05,
      "model_forward_time": 1.7219998836517334,
      "step": 44738
    },
    {
      "epoch": 4.266548156738281e-05,
      "step": 44738,
      "training_step_time": 10.482466697692871
    },
    {
      "epoch": 4.2666435241699216e-05,
      "model_forward_time": 1.7238740921020508,
      "step": 44739
    },
    {
      "epoch": 4.2666435241699216e-05,
      "step": 44739,
      "training_step_time": 7.729203701019287
    },
    {
      "epoch": 4.266738891601563e-05,
      "grad_norm": 0.12154959887266159,
      "learning_rate": 6.269216210799326e-06,
      "loss": 0.0612,
      "step": 44740
    },
    {
      "epoch": 4.266738891601563e-05,
      "model_forward_time": 1.734583854675293,
      "step": 44740
    },
    {
      "dynamics_loss_avg": 0.05714439209550619,
      "epoch": 4.266738891601563e-05,
      "step": 44740
    },
    {
      "action_loss_avg": 0.0017324222717434168,
      "epoch": 4.266738891601563e-05,
      "step": 44740
    },
    {
      "epoch": 4.266738891601563e-05,
      "step": 44740,
      "training_step_time": 8.973153114318848
    },
    {
      "epoch": 4.266834259033203e-05,
      "model_forward_time": 1.7217657566070557,
      "step": 44741
    },
    {
      "epoch": 4.266834259033203e-05,
      "step": 44741,
      "training_step_time": 7.778326511383057
    },
    {
      "epoch": 4.2669296264648435e-05,
      "model_forward_time": 1.7199854850769043,
      "step": 44742
    },
    {
      "epoch": 4.2669296264648435e-05,
      "step": 44742,
      "training_step_time": 8.796764612197876
    },
    {
      "epoch": 4.2670249938964846e-05,
      "model_forward_time": 1.7300481796264648,
      "step": 44743
    },
    {
      "epoch": 4.2670249938964846e-05,
      "step": 44743,
      "training_step_time": 7.912755250930786
    },
    {
      "epoch": 4.267120361328125e-05,
      "model_forward_time": 1.734792947769165,
      "step": 44744
    },
    {
      "epoch": 4.267120361328125e-05,
      "step": 44744,
      "training_step_time": 7.482625484466553
    },
    {
      "epoch": 4.2672157287597654e-05,
      "model_forward_time": 1.7382991313934326,
      "step": 44745
    },
    {
      "epoch": 4.2672157287597654e-05,
      "step": 44745,
      "training_step_time": 7.598331689834595
    },
    {
      "epoch": 4.2673110961914065e-05,
      "model_forward_time": 1.7377204895019531,
      "step": 44746
    },
    {
      "epoch": 4.2673110961914065e-05,
      "step": 44746,
      "training_step_time": 7.765251398086548
    },
    {
      "epoch": 4.267406463623047e-05,
      "model_forward_time": 1.7358856201171875,
      "step": 44747
    },
    {
      "epoch": 4.267406463623047e-05,
      "step": 44747,
      "training_step_time": 7.7799060344696045
    },
    {
      "epoch": 4.267501831054687e-05,
      "model_forward_time": 2.2734436988830566,
      "step": 44748
    },
    {
      "epoch": 4.267501831054687e-05,
      "step": 44748,
      "training_step_time": 7.828155517578125
    },
    {
      "epoch": 4.2675971984863283e-05,
      "model_forward_time": 1.7408127784729004,
      "step": 44749
    },
    {
      "epoch": 4.2675971984863283e-05,
      "step": 44749,
      "training_step_time": 9.823389291763306
    },
    {
      "epoch": 4.267692565917969e-05,
      "grad_norm": 0.13356593251228333,
      "learning_rate": 6.26761683040086e-06,
      "loss": 0.0604,
      "step": 44750
    },
    {
      "epoch": 4.267692565917969e-05,
      "model_forward_time": 1.7313015460968018,
      "step": 44750
    },
    {
      "dynamics_loss_avg": 0.04539655148983002,
      "epoch": 4.267692565917969e-05,
      "step": 44750
    },
    {
      "action_loss_avg": 0.0021222716401098295,
      "epoch": 4.267692565917969e-05,
      "step": 44750
    },
    {
      "epoch": 4.267692565917969e-05,
      "step": 44750,
      "training_step_time": 7.764259338378906
    },
    {
      "epoch": 4.267787933349609e-05,
      "model_forward_time": 1.783902883529663,
      "step": 44751
    },
    {
      "epoch": 4.267787933349609e-05,
      "step": 44751,
      "training_step_time": 7.875486373901367
    },
    {
      "epoch": 4.26788330078125e-05,
      "model_forward_time": 1.74057936668396,
      "step": 44752
    },
    {
      "epoch": 4.26788330078125e-05,
      "step": 44752,
      "training_step_time": 8.824678421020508
    },
    {
      "epoch": 4.2679786682128906e-05,
      "model_forward_time": 1.7445588111877441,
      "step": 44753
    },
    {
      "epoch": 4.2679786682128906e-05,
      "step": 44753,
      "training_step_time": 6.971991300582886
    },
    {
      "epoch": 4.268074035644531e-05,
      "model_forward_time": 1.7409281730651855,
      "step": 44754
    },
    {
      "epoch": 4.268074035644531e-05,
      "step": 44754,
      "training_step_time": 8.371195077896118
    },
    {
      "epoch": 4.268169403076172e-05,
      "model_forward_time": 1.749401330947876,
      "step": 44755
    },
    {
      "epoch": 4.268169403076172e-05,
      "step": 44755,
      "training_step_time": 8.64155626296997
    },
    {
      "epoch": 4.2682647705078125e-05,
      "model_forward_time": 1.7475075721740723,
      "step": 44756
    },
    {
      "epoch": 4.2682647705078125e-05,
      "step": 44756,
      "training_step_time": 7.75390625
    },
    {
      "epoch": 4.268360137939453e-05,
      "model_forward_time": 2.2960612773895264,
      "step": 44757
    },
    {
      "epoch": 4.268360137939453e-05,
      "step": 44757,
      "training_step_time": 7.787994623184204
    },
    {
      "epoch": 4.268455505371094e-05,
      "model_forward_time": 1.7504405975341797,
      "step": 44758
    },
    {
      "epoch": 4.268455505371094e-05,
      "step": 44758,
      "training_step_time": 8.707391738891602
    },
    {
      "epoch": 4.2685508728027344e-05,
      "model_forward_time": 1.7409424781799316,
      "step": 44759
    },
    {
      "epoch": 4.2685508728027344e-05,
      "step": 44759,
      "training_step_time": 8.388340473175049
    },
    {
      "epoch": 4.268646240234375e-05,
      "grad_norm": 0.12253974378108978,
      "learning_rate": 6.266017311377724e-06,
      "loss": 0.0613,
      "step": 44760
    },
    {
      "epoch": 4.268646240234375e-05,
      "model_forward_time": 1.7801201343536377,
      "step": 44760
    },
    {
      "dynamics_loss_avg": 0.06361392550170422,
      "epoch": 4.268646240234375e-05,
      "step": 44760
    },
    {
      "action_loss_avg": 0.0028826336201746017,
      "epoch": 4.268646240234375e-05,
      "step": 44760
    },
    {
      "epoch": 4.268646240234375e-05,
      "step": 44760,
      "training_step_time": 7.854785680770874
    },
    {
      "epoch": 4.268741607666016e-05,
      "model_forward_time": 1.740654706954956,
      "step": 44761
    },
    {
      "epoch": 4.268741607666016e-05,
      "step": 44761,
      "training_step_time": 7.5916759967803955
    },
    {
      "epoch": 4.268836975097656e-05,
      "model_forward_time": 1.744776964187622,
      "step": 44762
    },
    {
      "epoch": 4.268836975097656e-05,
      "step": 44762,
      "training_step_time": 7.819749116897583
    },
    {
      "epoch": 4.268932342529297e-05,
      "model_forward_time": 1.7399344444274902,
      "step": 44763
    },
    {
      "epoch": 4.268932342529297e-05,
      "step": 44763,
      "training_step_time": 7.577567100524902
    },
    {
      "epoch": 4.269027709960938e-05,
      "model_forward_time": 1.7469701766967773,
      "step": 44764
    },
    {
      "epoch": 4.269027709960938e-05,
      "step": 44764,
      "training_step_time": 7.786877155303955
    },
    {
      "epoch": 4.269123077392578e-05,
      "model_forward_time": 1.760718584060669,
      "step": 44765
    },
    {
      "epoch": 4.269123077392578e-05,
      "step": 44765,
      "training_step_time": 8.583122730255127
    },
    {
      "epoch": 4.2692184448242186e-05,
      "model_forward_time": 2.2958827018737793,
      "step": 44766
    },
    {
      "epoch": 4.2692184448242186e-05,
      "step": 44766,
      "training_step_time": 8.651518106460571
    },
    {
      "epoch": 4.26931381225586e-05,
      "model_forward_time": 1.742581844329834,
      "step": 44767
    },
    {
      "epoch": 4.26931381225586e-05,
      "step": 44767,
      "training_step_time": 8.427992343902588
    },
    {
      "epoch": 4.2694091796875e-05,
      "model_forward_time": 1.737466812133789,
      "step": 44768
    },
    {
      "epoch": 4.2694091796875e-05,
      "step": 44768,
      "training_step_time": 7.800123691558838
    },
    {
      "epoch": 4.2695045471191405e-05,
      "model_forward_time": 1.7872412204742432,
      "step": 44769
    },
    {
      "epoch": 4.2695045471191405e-05,
      "step": 44769,
      "training_step_time": 7.79979395866394
    },
    {
      "epoch": 4.2695999145507815e-05,
      "grad_norm": 0.09908488392829895,
      "learning_rate": 6.264417653904839e-06,
      "loss": 0.058,
      "step": 44770
    },
    {
      "epoch": 4.2695999145507815e-05,
      "model_forward_time": 1.7387309074401855,
      "step": 44770
    },
    {
      "dynamics_loss_avg": 0.06348070148378611,
      "epoch": 4.2695999145507815e-05,
      "step": 44770
    },
    {
      "action_loss_avg": 0.0022271920286584647,
      "epoch": 4.2695999145507815e-05,
      "step": 44770
    },
    {
      "epoch": 4.2695999145507815e-05,
      "step": 44770,
      "training_step_time": 8.034641981124878
    },
    {
      "epoch": 4.269695281982422e-05,
      "model_forward_time": 1.7488489151000977,
      "step": 44771
    },
    {
      "epoch": 4.269695281982422e-05,
      "step": 44771,
      "training_step_time": 7.611269950866699
    },
    {
      "epoch": 4.2697906494140624e-05,
      "model_forward_time": 1.7449419498443604,
      "step": 44772
    },
    {
      "epoch": 4.2697906494140624e-05,
      "step": 44772,
      "training_step_time": 8.153594732284546
    },
    {
      "epoch": 4.2698860168457034e-05,
      "model_forward_time": 1.7411835193634033,
      "step": 44773
    },
    {
      "epoch": 4.2698860168457034e-05,
      "step": 44773,
      "training_step_time": 7.692897796630859
    },
    {
      "epoch": 4.269981384277344e-05,
      "model_forward_time": 1.7427382469177246,
      "step": 44774
    },
    {
      "epoch": 4.269981384277344e-05,
      "step": 44774,
      "training_step_time": 7.746076822280884
    },
    {
      "epoch": 4.270076751708984e-05,
      "model_forward_time": 2.3228213787078857,
      "step": 44775
    },
    {
      "epoch": 4.270076751708984e-05,
      "step": 44775,
      "training_step_time": 9.389947175979614
    },
    {
      "epoch": 4.270172119140625e-05,
      "model_forward_time": 1.7471661567687988,
      "step": 44776
    },
    {
      "epoch": 4.270172119140625e-05,
      "step": 44776,
      "training_step_time": 7.766030788421631
    },
    {
      "epoch": 4.270267486572266e-05,
      "model_forward_time": 1.7415390014648438,
      "step": 44777
    },
    {
      "epoch": 4.270267486572266e-05,
      "step": 44777,
      "training_step_time": 8.34131145477295
    },
    {
      "epoch": 4.270362854003906e-05,
      "model_forward_time": 1.7862963676452637,
      "step": 44778
    },
    {
      "epoch": 4.270362854003906e-05,
      "step": 44778,
      "training_step_time": 9.187307834625244
    },
    {
      "epoch": 4.270458221435547e-05,
      "model_forward_time": 1.7425510883331299,
      "step": 44779
    },
    {
      "epoch": 4.270458221435547e-05,
      "step": 44779,
      "training_step_time": 7.821060419082642
    },
    {
      "epoch": 4.2705535888671876e-05,
      "grad_norm": 0.12961220741271973,
      "learning_rate": 6.26281785815714e-06,
      "loss": 0.0609,
      "step": 44780
    },
    {
      "epoch": 4.2705535888671876e-05,
      "model_forward_time": 1.7454442977905273,
      "step": 44780
    },
    {
      "dynamics_loss_avg": 0.058604841120541096,
      "epoch": 4.2705535888671876e-05,
      "step": 44780
    },
    {
      "action_loss_avg": 0.0013013672083616256,
      "epoch": 4.2705535888671876e-05,
      "step": 44780
    },
    {
      "epoch": 4.2705535888671876e-05,
      "step": 44780,
      "training_step_time": 7.757550954818726
    },
    {
      "epoch": 4.270648956298828e-05,
      "model_forward_time": 1.7516517639160156,
      "step": 44781
    },
    {
      "epoch": 4.270648956298828e-05,
      "step": 44781,
      "training_step_time": 7.587059736251831
    },
    {
      "epoch": 4.2707443237304684e-05,
      "model_forward_time": 1.7579026222229004,
      "step": 44782
    },
    {
      "epoch": 4.2707443237304684e-05,
      "step": 44782,
      "training_step_time": 7.917236089706421
    },
    {
      "epoch": 4.2708396911621095e-05,
      "model_forward_time": 1.7498784065246582,
      "step": 44783
    },
    {
      "epoch": 4.2708396911621095e-05,
      "step": 44783,
      "training_step_time": 7.682944297790527
    },
    {
      "epoch": 4.27093505859375e-05,
      "model_forward_time": 2.264772891998291,
      "step": 44784
    },
    {
      "epoch": 4.27093505859375e-05,
      "step": 44784,
      "training_step_time": 7.609962224960327
    },
    {
      "epoch": 4.27103042602539e-05,
      "model_forward_time": 1.73722505569458,
      "step": 44785
    },
    {
      "epoch": 4.27103042602539e-05,
      "step": 44785,
      "training_step_time": 7.7571704387664795
    },
    {
      "epoch": 4.2711257934570314e-05,
      "model_forward_time": 1.7348623275756836,
      "step": 44786
    },
    {
      "epoch": 4.2711257934570314e-05,
      "step": 44786,
      "training_step_time": 9.254561424255371
    },
    {
      "epoch": 4.271221160888672e-05,
      "model_forward_time": 1.7811319828033447,
      "step": 44787
    },
    {
      "epoch": 4.271221160888672e-05,
      "step": 44787,
      "training_step_time": 8.525802612304688
    },
    {
      "epoch": 4.271316528320312e-05,
      "model_forward_time": 1.7396321296691895,
      "step": 44788
    },
    {
      "epoch": 4.271316528320312e-05,
      "step": 44788,
      "training_step_time": 7.818558216094971
    },
    {
      "epoch": 4.271411895751953e-05,
      "model_forward_time": 1.7315020561218262,
      "step": 44789
    },
    {
      "epoch": 4.271411895751953e-05,
      "step": 44789,
      "training_step_time": 7.796589612960815
    },
    {
      "epoch": 4.271507263183594e-05,
      "grad_norm": 0.1213206872344017,
      "learning_rate": 6.261217924309576e-06,
      "loss": 0.059,
      "step": 44790
    },
    {
      "epoch": 4.271507263183594e-05,
      "model_forward_time": 1.7298102378845215,
      "step": 44790
    },
    {
      "dynamics_loss_avg": 0.04694638140499592,
      "epoch": 4.271507263183594e-05,
      "step": 44790
    },
    {
      "action_loss_avg": 0.0019464953511487693,
      "epoch": 4.271507263183594e-05,
      "step": 44790
    },
    {
      "epoch": 4.271507263183594e-05,
      "step": 44790,
      "training_step_time": 8.219664573669434
    },
    {
      "epoch": 4.271602630615234e-05,
      "model_forward_time": 1.729079008102417,
      "step": 44791
    },
    {
      "epoch": 4.271602630615234e-05,
      "step": 44791,
      "training_step_time": 7.809020042419434
    },
    {
      "epoch": 4.271697998046875e-05,
      "model_forward_time": 1.7443420886993408,
      "step": 44792
    },
    {
      "epoch": 4.271697998046875e-05,
      "step": 44792,
      "training_step_time": 7.715807676315308
    },
    {
      "epoch": 4.2717933654785156e-05,
      "model_forward_time": 1.7386116981506348,
      "step": 44793
    },
    {
      "epoch": 4.2717933654785156e-05,
      "step": 44793,
      "training_step_time": 8.761028289794922
    },
    {
      "epoch": 4.271888732910156e-05,
      "model_forward_time": 1.7337687015533447,
      "step": 44794
    },
    {
      "epoch": 4.271888732910156e-05,
      "step": 44794,
      "training_step_time": 7.715904712677002
    },
    {
      "epoch": 4.271984100341797e-05,
      "model_forward_time": 1.7325434684753418,
      "step": 44795
    },
    {
      "epoch": 4.271984100341797e-05,
      "step": 44795,
      "training_step_time": 8.83914589881897
    },
    {
      "epoch": 4.2720794677734374e-05,
      "model_forward_time": 1.7835721969604492,
      "step": 44796
    },
    {
      "epoch": 4.2720794677734374e-05,
      "step": 44796,
      "training_step_time": 9.517364740371704
    },
    {
      "epoch": 4.272174835205078e-05,
      "model_forward_time": 1.7377936840057373,
      "step": 44797
    },
    {
      "epoch": 4.272174835205078e-05,
      "step": 44797,
      "training_step_time": 8.331002235412598
    },
    {
      "epoch": 4.272270202636719e-05,
      "model_forward_time": 1.7410433292388916,
      "step": 44798
    },
    {
      "epoch": 4.272270202636719e-05,
      "step": 44798,
      "training_step_time": 8.349310398101807
    },
    {
      "epoch": 4.272365570068359e-05,
      "model_forward_time": 1.7389178276062012,
      "step": 44799
    },
    {
      "epoch": 4.272365570068359e-05,
      "step": 44799,
      "training_step_time": 7.52431845664978
    },
    {
      "epoch": 4.2724609375e-05,
      "grad_norm": 0.10633821785449982,
      "learning_rate": 6.259617852537119e-06,
      "loss": 0.0577,
      "step": 44800
    },
    {
      "epoch": 4.2724609375e-05,
      "model_forward_time": 1.7331242561340332,
      "step": 44800
    },
    {
      "dynamics_loss_avg": 0.06543493084609509,
      "epoch": 4.2724609375e-05,
      "step": 44800
    },
    {
      "action_loss_avg": 0.00225248935748823,
      "epoch": 4.2724609375e-05,
      "step": 44800
    },
    {
      "epoch": 4.2724609375e-05,
      "step": 44800,
      "training_step_time": 7.751116752624512
    },
    {
      "epoch": 4.272556304931641e-05,
      "model_forward_time": 1.7382168769836426,
      "step": 44801
    },
    {
      "epoch": 4.272556304931641e-05,
      "step": 44801,
      "training_step_time": 7.7142157554626465
    },
    {
      "epoch": 4.272651672363281e-05,
      "model_forward_time": 1.7402894496917725,
      "step": 44802
    },
    {
      "epoch": 4.272651672363281e-05,
      "step": 44802,
      "training_step_time": 8.53375768661499
    },
    {
      "epoch": 4.2727470397949216e-05,
      "model_forward_time": 2.2628891468048096,
      "step": 44803
    },
    {
      "epoch": 4.2727470397949216e-05,
      "step": 44803,
      "training_step_time": 7.734748363494873
    },
    {
      "epoch": 4.272842407226563e-05,
      "model_forward_time": 1.71803617477417,
      "step": 44804
    },
    {
      "epoch": 4.272842407226563e-05,
      "step": 44804,
      "training_step_time": 9.50107192993164
    },
    {
      "epoch": 4.272937774658203e-05,
      "model_forward_time": 1.404245138168335,
      "step": 44805
    },
    {
      "epoch": 4.272937774658203e-05,
      "step": 44805,
      "training_step_time": 10.11478042602539
    },
    {
      "epoch": 4.2730331420898435e-05,
      "model_forward_time": 1.7746315002441406,
      "step": 44806
    },
    {
      "epoch": 4.2730331420898435e-05,
      "step": 44806,
      "training_step_time": 9.05448317527771
    },
    {
      "epoch": 4.2731285095214846e-05,
      "model_forward_time": 1.7366478443145752,
      "step": 44807
    },
    {
      "epoch": 4.2731285095214846e-05,
      "step": 44807,
      "training_step_time": 7.846616744995117
    },
    {
      "epoch": 4.273223876953125e-05,
      "model_forward_time": 1.736278772354126,
      "step": 44808
    },
    {
      "epoch": 4.273223876953125e-05,
      "step": 44808,
      "training_step_time": 7.398324012756348
    },
    {
      "epoch": 4.2733192443847654e-05,
      "model_forward_time": 1.7341291904449463,
      "step": 44809
    },
    {
      "epoch": 4.2733192443847654e-05,
      "step": 44809,
      "training_step_time": 7.636658430099487
    },
    {
      "epoch": 4.2734146118164065e-05,
      "grad_norm": 0.14109517633914948,
      "learning_rate": 6.258017643014747e-06,
      "loss": 0.0579,
      "step": 44810
    },
    {
      "epoch": 4.2734146118164065e-05,
      "model_forward_time": 1.718160629272461,
      "step": 44810
    },
    {
      "dynamics_loss_avg": 0.06259255744516849,
      "epoch": 4.2734146118164065e-05,
      "step": 44810
    },
    {
      "action_loss_avg": 0.0023345125664491206,
      "epoch": 4.2734146118164065e-05,
      "step": 44810
    },
    {
      "epoch": 4.2734146118164065e-05,
      "step": 44810,
      "training_step_time": 7.864747762680054
    },
    {
      "epoch": 4.273509979248047e-05,
      "model_forward_time": 1.7375848293304443,
      "step": 44811
    },
    {
      "epoch": 4.273509979248047e-05,
      "step": 44811,
      "training_step_time": 7.764532089233398
    },
    {
      "epoch": 4.273605346679687e-05,
      "model_forward_time": 2.268047571182251,
      "step": 44812
    },
    {
      "epoch": 4.273605346679687e-05,
      "step": 44812,
      "training_step_time": 9.283680438995361
    },
    {
      "epoch": 4.2737007141113284e-05,
      "model_forward_time": 1.7283916473388672,
      "step": 44813
    },
    {
      "epoch": 4.2737007141113284e-05,
      "step": 44813,
      "training_step_time": 7.788258075714111
    },
    {
      "epoch": 4.273796081542969e-05,
      "model_forward_time": 1.7336101531982422,
      "step": 44814
    },
    {
      "epoch": 4.273796081542969e-05,
      "step": 44814,
      "training_step_time": 8.746893644332886
    },
    {
      "epoch": 4.273891448974609e-05,
      "model_forward_time": 1.7798633575439453,
      "step": 44815
    },
    {
      "epoch": 4.273891448974609e-05,
      "step": 44815,
      "training_step_time": 9.102905750274658
    },
    {
      "epoch": 4.27398681640625e-05,
      "model_forward_time": 1.7406795024871826,
      "step": 44816
    },
    {
      "epoch": 4.27398681640625e-05,
      "step": 44816,
      "training_step_time": 7.779007196426392
    },
    {
      "epoch": 4.2740821838378906e-05,
      "model_forward_time": 1.7185003757476807,
      "step": 44817
    },
    {
      "epoch": 4.2740821838378906e-05,
      "step": 44817,
      "training_step_time": 7.479435205459595
    },
    {
      "epoch": 4.274177551269531e-05,
      "model_forward_time": 1.7307360172271729,
      "step": 44818
    },
    {
      "epoch": 4.274177551269531e-05,
      "step": 44818,
      "training_step_time": 7.617993116378784
    },
    {
      "epoch": 4.274272918701172e-05,
      "model_forward_time": 1.7304058074951172,
      "step": 44819
    },
    {
      "epoch": 4.274272918701172e-05,
      "step": 44819,
      "training_step_time": 7.814085006713867
    },
    {
      "epoch": 4.2743682861328125e-05,
      "grad_norm": 0.6774343252182007,
      "learning_rate": 6.256417295917457e-06,
      "loss": 0.0558,
      "step": 44820
    },
    {
      "epoch": 4.2743682861328125e-05,
      "model_forward_time": 1.7324879169464111,
      "step": 44820
    },
    {
      "dynamics_loss_avg": 0.05147806219756603,
      "epoch": 4.2743682861328125e-05,
      "step": 44820
    },
    {
      "action_loss_avg": 0.0013283548119943588,
      "epoch": 4.2743682861328125e-05,
      "step": 44820
    },
    {
      "epoch": 4.2743682861328125e-05,
      "step": 44820,
      "training_step_time": 8.335929870605469
    },
    {
      "epoch": 4.274463653564453e-05,
      "model_forward_time": 2.2698802947998047,
      "step": 44821
    },
    {
      "epoch": 4.274463653564453e-05,
      "step": 44821,
      "training_step_time": 9.303508996963501
    },
    {
      "epoch": 4.274559020996094e-05,
      "model_forward_time": 1.7254574298858643,
      "step": 44822
    },
    {
      "epoch": 4.274559020996094e-05,
      "step": 44822,
      "training_step_time": 7.721682548522949
    },
    {
      "epoch": 4.2746543884277344e-05,
      "model_forward_time": 1.751643180847168,
      "step": 44823
    },
    {
      "epoch": 4.2746543884277344e-05,
      "step": 44823,
      "training_step_time": 9.779297113418579
    },
    {
      "epoch": 4.274749755859375e-05,
      "model_forward_time": 1.7207472324371338,
      "step": 44824
    },
    {
      "epoch": 4.274749755859375e-05,
      "step": 44824,
      "training_step_time": 8.40531039237976
    },
    {
      "epoch": 4.274845123291016e-05,
      "model_forward_time": 1.76381254196167,
      "step": 44825
    },
    {
      "epoch": 4.274845123291016e-05,
      "step": 44825,
      "training_step_time": 7.82590913772583
    },
    {
      "epoch": 4.274940490722656e-05,
      "model_forward_time": 1.0248639583587646,
      "step": 44826
    },
    {
      "epoch": 4.274940490722656e-05,
      "step": 44826,
      "training_step_time": 7.640589237213135
    },
    {
      "epoch": 4.275035858154297e-05,
      "model_forward_time": 1.7372238636016846,
      "step": 44827
    },
    {
      "epoch": 4.275035858154297e-05,
      "step": 44827,
      "training_step_time": 7.56973123550415
    },
    {
      "epoch": 4.275131225585938e-05,
      "model_forward_time": 1.7278430461883545,
      "step": 44828
    },
    {
      "epoch": 4.275131225585938e-05,
      "step": 44828,
      "training_step_time": 7.850382328033447
    },
    {
      "epoch": 4.275226593017578e-05,
      "model_forward_time": 1.7297799587249756,
      "step": 44829
    },
    {
      "epoch": 4.275226593017578e-05,
      "step": 44829,
      "training_step_time": 7.752312183380127
    },
    {
      "epoch": 4.2753219604492186e-05,
      "grad_norm": 0.139688178896904,
      "learning_rate": 6.254816811420259e-06,
      "loss": 0.0602,
      "step": 44830
    },
    {
      "epoch": 4.2753219604492186e-05,
      "model_forward_time": 2.2688119411468506,
      "step": 44830
    },
    {
      "dynamics_loss_avg": 0.06765974164009095,
      "epoch": 4.2753219604492186e-05,
      "step": 44830
    },
    {
      "action_loss_avg": 0.0026506186404731125,
      "epoch": 4.2753219604492186e-05,
      "step": 44830
    },
    {
      "epoch": 4.2753219604492186e-05,
      "step": 44830,
      "training_step_time": 7.571666955947876
    },
    {
      "epoch": 4.27541732788086e-05,
      "model_forward_time": 1.7292759418487549,
      "step": 44831
    },
    {
      "epoch": 4.27541732788086e-05,
      "step": 44831,
      "training_step_time": 8.744764804840088
    },
    {
      "epoch": 4.2755126953125e-05,
      "model_forward_time": 1.7280433177947998,
      "step": 44832
    },
    {
      "epoch": 4.2755126953125e-05,
      "step": 44832,
      "training_step_time": 7.8161842823028564
    },
    {
      "epoch": 4.2756080627441405e-05,
      "model_forward_time": 1.734316110610962,
      "step": 44833
    },
    {
      "epoch": 4.2756080627441405e-05,
      "step": 44833,
      "training_step_time": 7.846994876861572
    },
    {
      "epoch": 4.2757034301757816e-05,
      "model_forward_time": 1.7691583633422852,
      "step": 44834
    },
    {
      "epoch": 4.2757034301757816e-05,
      "step": 44834,
      "training_step_time": 7.8580498695373535
    },
    {
      "epoch": 4.275798797607422e-05,
      "model_forward_time": 1.723524808883667,
      "step": 44835
    },
    {
      "epoch": 4.275798797607422e-05,
      "step": 44835,
      "training_step_time": 7.556921720504761
    },
    {
      "epoch": 4.2758941650390624e-05,
      "model_forward_time": 1.7363476753234863,
      "step": 44836
    },
    {
      "epoch": 4.2758941650390624e-05,
      "step": 44836,
      "training_step_time": 7.593857526779175
    },
    {
      "epoch": 4.2759895324707034e-05,
      "model_forward_time": 1.734602928161621,
      "step": 44837
    },
    {
      "epoch": 4.2759895324707034e-05,
      "step": 44837,
      "training_step_time": 7.865034818649292
    },
    {
      "epoch": 4.276084899902344e-05,
      "model_forward_time": 1.735670804977417,
      "step": 44838
    },
    {
      "epoch": 4.276084899902344e-05,
      "step": 44838,
      "training_step_time": 7.721448183059692
    },
    {
      "epoch": 4.276180267333984e-05,
      "model_forward_time": 1.7420692443847656,
      "step": 44839
    },
    {
      "epoch": 4.276180267333984e-05,
      "step": 44839,
      "training_step_time": 8.438510656356812
    },
    {
      "epoch": 4.276275634765625e-05,
      "grad_norm": 0.12107467651367188,
      "learning_rate": 6.253216189698184e-06,
      "loss": 0.0548,
      "step": 44840
    },
    {
      "epoch": 4.276275634765625e-05,
      "model_forward_time": 2.2748045921325684,
      "step": 44840
    },
    {
      "dynamics_loss_avg": 0.055986224487423894,
      "epoch": 4.276275634765625e-05,
      "step": 44840
    },
    {
      "action_loss_avg": 0.0018723070039413869,
      "epoch": 4.276275634765625e-05,
      "step": 44840
    },
    {
      "epoch": 4.276275634765625e-05,
      "step": 44840,
      "training_step_time": 8.403893232345581
    },
    {
      "epoch": 4.276371002197266e-05,
      "model_forward_time": 1.7446918487548828,
      "step": 44841
    },
    {
      "epoch": 4.276371002197266e-05,
      "step": 44841,
      "training_step_time": 8.774225234985352
    },
    {
      "epoch": 4.276466369628906e-05,
      "model_forward_time": 1.7492280006408691,
      "step": 44842
    },
    {
      "epoch": 4.276466369628906e-05,
      "step": 44842,
      "training_step_time": 8.767187595367432
    },
    {
      "epoch": 4.276561737060547e-05,
      "model_forward_time": 1.7855539321899414,
      "step": 44843
    },
    {
      "epoch": 4.276561737060547e-05,
      "step": 44843,
      "training_step_time": 7.663480043411255
    },
    {
      "epoch": 4.2766571044921876e-05,
      "model_forward_time": 1.7433621883392334,
      "step": 44844
    },
    {
      "epoch": 4.2766571044921876e-05,
      "step": 44844,
      "training_step_time": 7.538172960281372
    },
    {
      "epoch": 4.276752471923828e-05,
      "model_forward_time": 1.738759994506836,
      "step": 44845
    },
    {
      "epoch": 4.276752471923828e-05,
      "step": 44845,
      "training_step_time": 7.577284097671509
    },
    {
      "epoch": 4.2768478393554684e-05,
      "model_forward_time": 1.740257740020752,
      "step": 44846
    },
    {
      "epoch": 4.2768478393554684e-05,
      "step": 44846,
      "training_step_time": 7.85734224319458
    },
    {
      "epoch": 4.2769432067871095e-05,
      "model_forward_time": 1.7427678108215332,
      "step": 44847
    },
    {
      "epoch": 4.2769432067871095e-05,
      "step": 44847,
      "training_step_time": 8.699966669082642
    },
    {
      "epoch": 4.27703857421875e-05,
      "model_forward_time": 1.7045576572418213,
      "step": 44848
    },
    {
      "epoch": 4.27703857421875e-05,
      "step": 44848,
      "training_step_time": 8.51575255393982
    },
    {
      "epoch": 4.27713394165039e-05,
      "model_forward_time": 2.2643239498138428,
      "step": 44849
    },
    {
      "epoch": 4.27713394165039e-05,
      "step": 44849,
      "training_step_time": 7.553675413131714
    },
    {
      "epoch": 4.2772293090820314e-05,
      "grad_norm": 0.08391690254211426,
      "learning_rate": 6.251615430926268e-06,
      "loss": 0.055,
      "step": 44850
    },
    {
      "epoch": 4.2772293090820314e-05,
      "model_forward_time": 1.720102071762085,
      "step": 44850
    },
    {
      "dynamics_loss_avg": 0.06116048898547888,
      "epoch": 4.2772293090820314e-05,
      "step": 44850
    },
    {
      "action_loss_avg": 0.001864926388952881,
      "epoch": 4.2772293090820314e-05,
      "step": 44850
    },
    {
      "epoch": 4.2772293090820314e-05,
      "step": 44850,
      "training_step_time": 8.55203366279602
    },
    {
      "epoch": 4.277324676513672e-05,
      "model_forward_time": 1.7176332473754883,
      "step": 44851
    },
    {
      "epoch": 4.277324676513672e-05,
      "step": 44851,
      "training_step_time": 7.820277452468872
    },
    {
      "epoch": 4.277420043945312e-05,
      "model_forward_time": 1.727172613143921,
      "step": 44852
    },
    {
      "epoch": 4.277420043945312e-05,
      "step": 44852,
      "training_step_time": 8.732378721237183
    },
    {
      "epoch": 4.277515411376953e-05,
      "model_forward_time": 1.7189674377441406,
      "step": 44853
    },
    {
      "epoch": 4.277515411376953e-05,
      "step": 44853,
      "training_step_time": 7.430936336517334
    },
    {
      "epoch": 4.277610778808594e-05,
      "model_forward_time": 1.712864875793457,
      "step": 44854
    },
    {
      "epoch": 4.277610778808594e-05,
      "step": 44854,
      "training_step_time": 7.567222595214844
    },
    {
      "epoch": 4.277706146240234e-05,
      "model_forward_time": 1.731435775756836,
      "step": 44855
    },
    {
      "epoch": 4.277706146240234e-05,
      "step": 44855,
      "training_step_time": 7.671233892440796
    },
    {
      "epoch": 4.277801513671875e-05,
      "model_forward_time": 1.710542917251587,
      "step": 44856
    },
    {
      "epoch": 4.277801513671875e-05,
      "step": 44856,
      "training_step_time": 7.809207916259766
    },
    {
      "epoch": 4.2778968811035156e-05,
      "model_forward_time": 1.7234387397766113,
      "step": 44857
    },
    {
      "epoch": 4.2778968811035156e-05,
      "step": 44857,
      "training_step_time": 8.566622734069824
    },
    {
      "epoch": 4.277992248535156e-05,
      "model_forward_time": 2.2674219608306885,
      "step": 44858
    },
    {
      "epoch": 4.277992248535156e-05,
      "step": 44858,
      "training_step_time": 9.726366996765137
    },
    {
      "epoch": 4.278087615966797e-05,
      "model_forward_time": 1.7241020202636719,
      "step": 44859
    },
    {
      "epoch": 4.278087615966797e-05,
      "step": 44859,
      "training_step_time": 7.866472959518433
    },
    {
      "epoch": 4.2781829833984374e-05,
      "grad_norm": 0.11998821049928665,
      "learning_rate": 6.25001453527957e-06,
      "loss": 0.0562,
      "step": 44860
    },
    {
      "epoch": 4.2781829833984374e-05,
      "model_forward_time": 1.723374843597412,
      "step": 44860
    },
    {
      "dynamics_loss_avg": 0.06141672357916832,
      "epoch": 4.2781829833984374e-05,
      "step": 44860
    },
    {
      "action_loss_avg": 0.002377524133771658,
      "epoch": 4.2781829833984374e-05,
      "step": 44860
    },
    {
      "epoch": 4.2781829833984374e-05,
      "step": 44860,
      "training_step_time": 7.853243589401245
    },
    {
      "epoch": 4.278278350830078e-05,
      "model_forward_time": 1.7371704578399658,
      "step": 44861
    },
    {
      "epoch": 4.278278350830078e-05,
      "step": 44861,
      "training_step_time": 7.876178503036499
    },
    {
      "epoch": 4.278373718261719e-05,
      "model_forward_time": 1.730672836303711,
      "step": 44862
    },
    {
      "epoch": 4.278373718261719e-05,
      "step": 44862,
      "training_step_time": 7.80781364440918
    },
    {
      "epoch": 4.278469085693359e-05,
      "model_forward_time": 1.7266020774841309,
      "step": 44863
    },
    {
      "epoch": 4.278469085693359e-05,
      "step": 44863,
      "training_step_time": 7.787208080291748
    },
    {
      "epoch": 4.278564453125e-05,
      "model_forward_time": 1.725205421447754,
      "step": 44864
    },
    {
      "epoch": 4.278564453125e-05,
      "step": 44864,
      "training_step_time": 7.440679550170898
    },
    {
      "epoch": 4.278659820556641e-05,
      "model_forward_time": 1.7349584102630615,
      "step": 44865
    },
    {
      "epoch": 4.278659820556641e-05,
      "step": 44865,
      "training_step_time": 7.929425954818726
    },
    {
      "epoch": 4.278755187988281e-05,
      "model_forward_time": 1.726196050643921,
      "step": 44866
    },
    {
      "epoch": 4.278755187988281e-05,
      "step": 44866,
      "training_step_time": 7.814968109130859
    },
    {
      "epoch": 4.2788505554199216e-05,
      "model_forward_time": 2.2827978134155273,
      "step": 44867
    },
    {
      "epoch": 4.2788505554199216e-05,
      "step": 44867,
      "training_step_time": 7.62177586555481
    },
    {
      "epoch": 4.278945922851563e-05,
      "model_forward_time": 1.7275354862213135,
      "step": 44868
    },
    {
      "epoch": 4.278945922851563e-05,
      "step": 44868,
      "training_step_time": 7.827847957611084
    },
    {
      "epoch": 4.279041290283203e-05,
      "model_forward_time": 1.7229115962982178,
      "step": 44869
    },
    {
      "epoch": 4.279041290283203e-05,
      "step": 44869,
      "training_step_time": 8.791398286819458
    },
    {
      "epoch": 4.2791366577148435e-05,
      "grad_norm": 0.10227609425783157,
      "learning_rate": 6.248413502933164e-06,
      "loss": 0.0533,
      "step": 44870
    },
    {
      "epoch": 4.2791366577148435e-05,
      "model_forward_time": 1.7158527374267578,
      "step": 44870
    },
    {
      "dynamics_loss_avg": 0.06104520633816719,
      "epoch": 4.2791366577148435e-05,
      "step": 44870
    },
    {
      "action_loss_avg": 0.0024504811473889275,
      "epoch": 4.2791366577148435e-05,
      "step": 44870
    },
    {
      "epoch": 4.2791366577148435e-05,
      "step": 44870,
      "training_step_time": 7.743714809417725
    },
    {
      "epoch": 4.2792320251464846e-05,
      "model_forward_time": 1.7177860736846924,
      "step": 44871
    },
    {
      "epoch": 4.2792320251464846e-05,
      "step": 44871,
      "training_step_time": 7.769512891769409
    },
    {
      "epoch": 4.279327392578125e-05,
      "model_forward_time": 1.7198798656463623,
      "step": 44872
    },
    {
      "epoch": 4.279327392578125e-05,
      "step": 44872,
      "training_step_time": 7.743592977523804
    },
    {
      "epoch": 4.2794227600097654e-05,
      "model_forward_time": 1.7243413925170898,
      "step": 44873
    },
    {
      "epoch": 4.2794227600097654e-05,
      "step": 44873,
      "training_step_time": 7.447826623916626
    },
    {
      "epoch": 4.2795181274414065e-05,
      "model_forward_time": 1.7379422187805176,
      "step": 44874
    },
    {
      "epoch": 4.2795181274414065e-05,
      "step": 44874,
      "training_step_time": 7.824711322784424
    },
    {
      "epoch": 4.279613494873047e-05,
      "model_forward_time": 1.7212889194488525,
      "step": 44875
    },
    {
      "epoch": 4.279613494873047e-05,
      "step": 44875,
      "training_step_time": 8.5756254196167
    },
    {
      "epoch": 4.279708862304687e-05,
      "model_forward_time": 2.2699625492095947,
      "step": 44876
    },
    {
      "epoch": 4.279708862304687e-05,
      "step": 44876,
      "training_step_time": 7.580488681793213
    },
    {
      "epoch": 4.2798042297363284e-05,
      "model_forward_time": 1.7230820655822754,
      "step": 44877
    },
    {
      "epoch": 4.2798042297363284e-05,
      "step": 44877,
      "training_step_time": 8.785695314407349
    },
    {
      "epoch": 4.279899597167969e-05,
      "model_forward_time": 1.7308108806610107,
      "step": 44878
    },
    {
      "epoch": 4.279899597167969e-05,
      "step": 44878,
      "training_step_time": 9.250061511993408
    },
    {
      "epoch": 4.279994964599609e-05,
      "model_forward_time": 1.715367078781128,
      "step": 44879
    },
    {
      "epoch": 4.279994964599609e-05,
      "step": 44879,
      "training_step_time": 8.839774131774902
    },
    {
      "epoch": 4.28009033203125e-05,
      "grad_norm": 0.09433870017528534,
      "learning_rate": 6.246812334062133e-06,
      "loss": 0.0574,
      "step": 44880
    },
    {
      "epoch": 4.28009033203125e-05,
      "model_forward_time": 1.7163474559783936,
      "step": 44880
    },
    {
      "dynamics_loss_avg": 0.06792279370129109,
      "epoch": 4.28009033203125e-05,
      "step": 44880
    },
    {
      "action_loss_avg": 0.002093353064265102,
      "epoch": 4.28009033203125e-05,
      "step": 44880
    },
    {
      "epoch": 4.28009033203125e-05,
      "step": 44880,
      "training_step_time": 8.551653623580933
    },
    {
      "epoch": 4.2801856994628906e-05,
      "model_forward_time": 1.7135851383209229,
      "step": 44881
    },
    {
      "epoch": 4.2801856994628906e-05,
      "step": 44881,
      "training_step_time": 7.724957227706909
    },
    {
      "epoch": 4.280281066894531e-05,
      "model_forward_time": 1.7115414142608643,
      "step": 44882
    },
    {
      "epoch": 4.280281066894531e-05,
      "step": 44882,
      "training_step_time": 7.536936283111572
    },
    {
      "epoch": 4.280376434326172e-05,
      "model_forward_time": 1.7197027206420898,
      "step": 44883
    },
    {
      "epoch": 4.280376434326172e-05,
      "step": 44883,
      "training_step_time": 7.800673961639404
    },
    {
      "epoch": 4.2804718017578125e-05,
      "model_forward_time": 1.7114667892456055,
      "step": 44884
    },
    {
      "epoch": 4.2804718017578125e-05,
      "step": 44884,
      "training_step_time": 7.577335834503174
    },
    {
      "epoch": 4.280567169189453e-05,
      "model_forward_time": 1.7215111255645752,
      "step": 44885
    },
    {
      "epoch": 4.280567169189453e-05,
      "step": 44885,
      "training_step_time": 9.6572585105896
    },
    {
      "epoch": 4.280662536621094e-05,
      "model_forward_time": 1.7197396755218506,
      "step": 44886
    },
    {
      "epoch": 4.280662536621094e-05,
      "step": 44886,
      "training_step_time": 7.498549222946167
    },
    {
      "epoch": 4.2807579040527344e-05,
      "model_forward_time": 1.7202632427215576,
      "step": 44887
    },
    {
      "epoch": 4.2807579040527344e-05,
      "step": 44887,
      "training_step_time": 10.52599811553955
    },
    {
      "epoch": 4.280853271484375e-05,
      "model_forward_time": 1.7119395732879639,
      "step": 44888
    },
    {
      "epoch": 4.280853271484375e-05,
      "step": 44888,
      "training_step_time": 8.192724704742432
    },
    {
      "epoch": 4.280948638916016e-05,
      "model_forward_time": 1.7380890846252441,
      "step": 44889
    },
    {
      "epoch": 4.280948638916016e-05,
      "step": 44889,
      "training_step_time": 8.767205476760864
    },
    {
      "epoch": 4.281044006347656e-05,
      "grad_norm": 0.10053493082523346,
      "learning_rate": 6.2452110288415795e-06,
      "loss": 0.0562,
      "step": 44890
    },
    {
      "epoch": 4.281044006347656e-05,
      "model_forward_time": 1.7241485118865967,
      "step": 44890
    },
    {
      "dynamics_loss_avg": 0.06120954807847738,
      "epoch": 4.281044006347656e-05,
      "step": 44890
    },
    {
      "action_loss_avg": 0.003326089639449492,
      "epoch": 4.281044006347656e-05,
      "step": 44890
    },
    {
      "epoch": 4.281044006347656e-05,
      "step": 44890,
      "training_step_time": 7.623376131057739
    },
    {
      "epoch": 4.281139373779297e-05,
      "model_forward_time": 1.7213380336761475,
      "step": 44891
    },
    {
      "epoch": 4.281139373779297e-05,
      "step": 44891,
      "training_step_time": 7.412769317626953
    },
    {
      "epoch": 4.281234741210938e-05,
      "model_forward_time": 1.720470905303955,
      "step": 44892
    },
    {
      "epoch": 4.281234741210938e-05,
      "step": 44892,
      "training_step_time": 10.187076568603516
    },
    {
      "epoch": 4.281330108642578e-05,
      "model_forward_time": 1.7236146926879883,
      "step": 44893
    },
    {
      "epoch": 4.281330108642578e-05,
      "step": 44893,
      "training_step_time": 7.650441408157349
    },
    {
      "epoch": 4.2814254760742186e-05,
      "model_forward_time": 1.7079646587371826,
      "step": 44894
    },
    {
      "epoch": 4.2814254760742186e-05,
      "step": 44894,
      "training_step_time": 8.285154581069946
    },
    {
      "epoch": 4.28152084350586e-05,
      "model_forward_time": 2.260514974594116,
      "step": 44895
    },
    {
      "epoch": 4.28152084350586e-05,
      "step": 44895,
      "training_step_time": 8.282758474349976
    },
    {
      "epoch": 4.2816162109375e-05,
      "model_forward_time": 1.0209755897521973,
      "step": 44896
    },
    {
      "epoch": 4.2816162109375e-05,
      "step": 44896,
      "training_step_time": 7.763420581817627
    },
    {
      "epoch": 4.2817115783691405e-05,
      "model_forward_time": 1.7194323539733887,
      "step": 44897
    },
    {
      "epoch": 4.2817115783691405e-05,
      "step": 44897,
      "training_step_time": 9.427820920944214
    },
    {
      "epoch": 4.2818069458007816e-05,
      "model_forward_time": 1.7276995182037354,
      "step": 44898
    },
    {
      "epoch": 4.2818069458007816e-05,
      "step": 44898,
      "training_step_time": 7.759523630142212
    },
    {
      "epoch": 4.281902313232422e-05,
      "model_forward_time": 1.7246367931365967,
      "step": 44899
    },
    {
      "epoch": 4.281902313232422e-05,
      "step": 44899,
      "training_step_time": 8.357329607009888
    },
    {
      "epoch": 4.2819976806640624e-05,
      "grad_norm": 0.10991764813661575,
      "learning_rate": 6.243609587446621e-06,
      "loss": 0.0577,
      "step": 44900
    },
    {
      "epoch": 4.2819976806640624e-05,
      "model_forward_time": 1.7207300662994385,
      "step": 44900
    },
    {
      "dynamics_loss_avg": 0.05997736342251301,
      "epoch": 4.2819976806640624e-05,
      "step": 44900
    },
    {
      "action_loss_avg": 0.002441755507607013,
      "epoch": 4.2819976806640624e-05,
      "step": 44900
    },
    {
      "epoch": 4.2819976806640624e-05,
      "step": 44900,
      "training_step_time": 9.286282777786255
    },
    {
      "epoch": 4.2820930480957034e-05,
      "model_forward_time": 1.7338411808013916,
      "step": 44901
    },
    {
      "epoch": 4.2820930480957034e-05,
      "step": 44901,
      "training_step_time": 9.218529224395752
    },
    {
      "epoch": 4.282188415527344e-05,
      "model_forward_time": 1.72072172164917,
      "step": 44902
    },
    {
      "epoch": 4.282188415527344e-05,
      "step": 44902,
      "training_step_time": 7.704995632171631
    },
    {
      "epoch": 4.282283782958984e-05,
      "model_forward_time": 1.718445062637329,
      "step": 44903
    },
    {
      "epoch": 4.282283782958984e-05,
      "step": 44903,
      "training_step_time": 8.086871862411499
    },
    {
      "epoch": 4.282379150390625e-05,
      "model_forward_time": 2.2739527225494385,
      "step": 44904
    },
    {
      "epoch": 4.282379150390625e-05,
      "step": 44904,
      "training_step_time": 8.08001160621643
    },
    {
      "epoch": 4.282474517822266e-05,
      "model_forward_time": 1.7266833782196045,
      "step": 44905
    },
    {
      "epoch": 4.282474517822266e-05,
      "step": 44905,
      "training_step_time": 8.208296537399292
    },
    {
      "epoch": 4.282569885253906e-05,
      "model_forward_time": 1.7166330814361572,
      "step": 44906
    },
    {
      "epoch": 4.282569885253906e-05,
      "step": 44906,
      "training_step_time": 7.729804754257202
    },
    {
      "epoch": 4.2826652526855465e-05,
      "model_forward_time": 1.7250220775604248,
      "step": 44907
    },
    {
      "epoch": 4.2826652526855465e-05,
      "step": 44907,
      "training_step_time": 7.792259693145752
    },
    {
      "epoch": 4.2827606201171876e-05,
      "model_forward_time": 1.7360615730285645,
      "step": 44908
    },
    {
      "epoch": 4.2827606201171876e-05,
      "step": 44908,
      "training_step_time": 7.990169525146484
    },
    {
      "epoch": 4.282855987548828e-05,
      "model_forward_time": 1.7301034927368164,
      "step": 44909
    },
    {
      "epoch": 4.282855987548828e-05,
      "step": 44909,
      "training_step_time": 7.693620443344116
    },
    {
      "epoch": 4.2829513549804684e-05,
      "grad_norm": 0.08379537612199783,
      "learning_rate": 6.242008010052387e-06,
      "loss": 0.0567,
      "step": 44910
    },
    {
      "epoch": 4.2829513549804684e-05,
      "model_forward_time": 1.7259600162506104,
      "step": 44910
    },
    {
      "dynamics_loss_avg": 0.05710740499198437,
      "epoch": 4.2829513549804684e-05,
      "step": 44910
    },
    {
      "action_loss_avg": 0.0014214683789759875,
      "epoch": 4.2829513549804684e-05,
      "step": 44910
    },
    {
      "epoch": 4.2829513549804684e-05,
      "step": 44910,
      "training_step_time": 8.482621669769287
    },
    {
      "epoch": 4.2830467224121095e-05,
      "model_forward_time": 1.7078280448913574,
      "step": 44911
    },
    {
      "epoch": 4.2830467224121095e-05,
      "step": 44911,
      "training_step_time": 8.289509534835815
    },
    {
      "epoch": 4.28314208984375e-05,
      "model_forward_time": 1.7217683792114258,
      "step": 44912
    },
    {
      "epoch": 4.28314208984375e-05,
      "step": 44912,
      "training_step_time": 8.244262456893921
    },
    {
      "epoch": 4.28323745727539e-05,
      "model_forward_time": 2.2796804904937744,
      "step": 44913
    },
    {
      "epoch": 4.28323745727539e-05,
      "step": 44913,
      "training_step_time": 8.08982229232788
    },
    {
      "epoch": 4.2833328247070314e-05,
      "model_forward_time": 1.721869945526123,
      "step": 44914
    },
    {
      "epoch": 4.2833328247070314e-05,
      "step": 44914,
      "training_step_time": 7.655904293060303
    },
    {
      "epoch": 4.283428192138672e-05,
      "model_forward_time": 1.7195475101470947,
      "step": 44915
    },
    {
      "epoch": 4.283428192138672e-05,
      "step": 44915,
      "training_step_time": 10.07608962059021
    },
    {
      "epoch": 4.283523559570312e-05,
      "model_forward_time": 1.7197906970977783,
      "step": 44916
    },
    {
      "epoch": 4.283523559570312e-05,
      "step": 44916,
      "training_step_time": 8.749062538146973
    },
    {
      "epoch": 4.283618927001953e-05,
      "model_forward_time": 1.7361087799072266,
      "step": 44917
    },
    {
      "epoch": 4.283618927001953e-05,
      "step": 44917,
      "training_step_time": 8.275689363479614
    },
    {
      "epoch": 4.283714294433594e-05,
      "model_forward_time": 1.7130534648895264,
      "step": 44918
    },
    {
      "epoch": 4.283714294433594e-05,
      "step": 44918,
      "training_step_time": 8.187851190567017
    },
    {
      "epoch": 4.283809661865234e-05,
      "model_forward_time": 1.7211706638336182,
      "step": 44919
    },
    {
      "epoch": 4.283809661865234e-05,
      "step": 44919,
      "training_step_time": 7.60111141204834
    },
    {
      "epoch": 4.283905029296875e-05,
      "grad_norm": 0.15668480098247528,
      "learning_rate": 6.240406296834024e-06,
      "loss": 0.0599,
      "step": 44920
    },
    {
      "epoch": 4.283905029296875e-05,
      "model_forward_time": 1.72745680809021,
      "step": 44920
    },
    {
      "dynamics_loss_avg": 0.05745121203362942,
      "epoch": 4.283905029296875e-05,
      "step": 44920
    },
    {
      "action_loss_avg": 0.0013645709812408313,
      "epoch": 4.283905029296875e-05,
      "step": 44920
    },
    {
      "epoch": 4.283905029296875e-05,
      "step": 44920,
      "training_step_time": 7.552534103393555
    },
    {
      "epoch": 4.2840003967285156e-05,
      "model_forward_time": 1.7115178108215332,
      "step": 44921
    },
    {
      "epoch": 4.2840003967285156e-05,
      "step": 44921,
      "training_step_time": 7.560299873352051
    },
    {
      "epoch": 4.284095764160156e-05,
      "model_forward_time": 2.2753992080688477,
      "step": 44922
    },
    {
      "epoch": 4.284095764160156e-05,
      "step": 44922,
      "training_step_time": 8.041854619979858
    },
    {
      "epoch": 4.284191131591797e-05,
      "model_forward_time": 1.024346113204956,
      "step": 44923
    },
    {
      "epoch": 4.284191131591797e-05,
      "step": 44923,
      "training_step_time": 7.538570404052734
    },
    {
      "epoch": 4.2842864990234375e-05,
      "model_forward_time": 1.717343807220459,
      "step": 44924
    },
    {
      "epoch": 4.2842864990234375e-05,
      "step": 44924,
      "training_step_time": 7.582035303115845
    },
    {
      "epoch": 4.284381866455078e-05,
      "model_forward_time": 1.7174477577209473,
      "step": 44925
    },
    {
      "epoch": 4.284381866455078e-05,
      "step": 44925,
      "training_step_time": 8.384003162384033
    },
    {
      "epoch": 4.284477233886719e-05,
      "model_forward_time": 1.7125892639160156,
      "step": 44926
    },
    {
      "epoch": 4.284477233886719e-05,
      "step": 44926,
      "training_step_time": 7.657244682312012
    },
    {
      "epoch": 4.284572601318359e-05,
      "model_forward_time": 1.7182860374450684,
      "step": 44927
    },
    {
      "epoch": 4.284572601318359e-05,
      "step": 44927,
      "training_step_time": 7.8478593826293945
    },
    {
      "epoch": 4.28466796875e-05,
      "model_forward_time": 1.7194170951843262,
      "step": 44928
    },
    {
      "epoch": 4.28466796875e-05,
      "step": 44928,
      "training_step_time": 7.6735968589782715
    },
    {
      "epoch": 4.284763336181641e-05,
      "model_forward_time": 1.7357323169708252,
      "step": 44929
    },
    {
      "epoch": 4.284763336181641e-05,
      "step": 44929,
      "training_step_time": 7.865391969680786
    },
    {
      "epoch": 4.284858703613281e-05,
      "grad_norm": 0.13596215844154358,
      "learning_rate": 6.238804447966694e-06,
      "loss": 0.059,
      "step": 44930
    },
    {
      "epoch": 4.284858703613281e-05,
      "model_forward_time": 1.718686819076538,
      "step": 44930
    },
    {
      "dynamics_loss_avg": 0.04487214330583811,
      "epoch": 4.284858703613281e-05,
      "step": 44930
    },
    {
      "action_loss_avg": 0.0016942700924118981,
      "epoch": 4.284858703613281e-05,
      "step": 44930
    },
    {
      "epoch": 4.284858703613281e-05,
      "step": 44930,
      "training_step_time": 8.564653635025024
    },
    {
      "epoch": 4.2849540710449216e-05,
      "model_forward_time": 1.7233402729034424,
      "step": 44931
    },
    {
      "epoch": 4.2849540710449216e-05,
      "step": 44931,
      "training_step_time": 7.703280925750732
    },
    {
      "epoch": 4.285049438476563e-05,
      "model_forward_time": 2.2602462768554688,
      "step": 44932
    },
    {
      "epoch": 4.285049438476563e-05,
      "step": 44932,
      "training_step_time": 7.733087778091431
    },
    {
      "epoch": 4.285144805908203e-05,
      "model_forward_time": 1.7214999198913574,
      "step": 44933
    },
    {
      "epoch": 4.285144805908203e-05,
      "step": 44933,
      "training_step_time": 7.812557220458984
    },
    {
      "epoch": 4.2852401733398435e-05,
      "model_forward_time": 1.7059495449066162,
      "step": 44934
    },
    {
      "epoch": 4.2852401733398435e-05,
      "step": 44934,
      "training_step_time": 9.05055856704712
    },
    {
      "epoch": 4.2853355407714846e-05,
      "model_forward_time": 1.7131688594818115,
      "step": 44935
    },
    {
      "epoch": 4.2853355407714846e-05,
      "step": 44935,
      "training_step_time": 7.698959589004517
    },
    {
      "epoch": 4.285430908203125e-05,
      "model_forward_time": 1.721182107925415,
      "step": 44936
    },
    {
      "epoch": 4.285430908203125e-05,
      "step": 44936,
      "training_step_time": 7.820888042449951
    },
    {
      "epoch": 4.2855262756347654e-05,
      "model_forward_time": 1.7017710208892822,
      "step": 44937
    },
    {
      "epoch": 4.2855262756347654e-05,
      "step": 44937,
      "training_step_time": 7.705796957015991
    },
    {
      "epoch": 4.2856216430664065e-05,
      "model_forward_time": 1.7339985370635986,
      "step": 44938
    },
    {
      "epoch": 4.2856216430664065e-05,
      "step": 44938,
      "training_step_time": 7.8022449016571045
    },
    {
      "epoch": 4.285717010498047e-05,
      "model_forward_time": 1.7069387435913086,
      "step": 44939
    },
    {
      "epoch": 4.285717010498047e-05,
      "step": 44939,
      "training_step_time": 7.633897304534912
    },
    {
      "epoch": 4.285812377929687e-05,
      "grad_norm": 0.1284104436635971,
      "learning_rate": 6.237202463625572e-06,
      "loss": 0.0588,
      "step": 44940
    },
    {
      "epoch": 4.285812377929687e-05,
      "model_forward_time": 1.7106497287750244,
      "step": 44940
    },
    {
      "dynamics_loss_avg": 0.05596745647490024,
      "epoch": 4.285812377929687e-05,
      "step": 44940
    },
    {
      "action_loss_avg": 0.00218070960836485,
      "epoch": 4.285812377929687e-05,
      "step": 44940
    },
    {
      "epoch": 4.285812377929687e-05,
      "step": 44940,
      "training_step_time": 7.698350667953491
    },
    {
      "epoch": 4.2859077453613284e-05,
      "model_forward_time": 2.2628698348999023,
      "step": 44941
    },
    {
      "epoch": 4.2859077453613284e-05,
      "step": 44941,
      "training_step_time": 8.757207155227661
    },
    {
      "epoch": 4.286003112792969e-05,
      "model_forward_time": 1.709585189819336,
      "step": 44942
    },
    {
      "epoch": 4.286003112792969e-05,
      "step": 44942,
      "training_step_time": 8.536959171295166
    },
    {
      "epoch": 4.286098480224609e-05,
      "model_forward_time": 1.7092516422271729,
      "step": 44943
    },
    {
      "epoch": 4.286098480224609e-05,
      "step": 44943,
      "training_step_time": 8.303638935089111
    },
    {
      "epoch": 4.28619384765625e-05,
      "model_forward_time": 1.709141492843628,
      "step": 44944
    },
    {
      "epoch": 4.28619384765625e-05,
      "step": 44944,
      "training_step_time": 8.8773353099823
    },
    {
      "epoch": 4.2862892150878907e-05,
      "model_forward_time": 1.7167470455169678,
      "step": 44945
    },
    {
      "epoch": 4.2862892150878907e-05,
      "step": 44945,
      "training_step_time": 7.631237268447876
    },
    {
      "epoch": 4.286384582519531e-05,
      "model_forward_time": 1.7077252864837646,
      "step": 44946
    },
    {
      "epoch": 4.286384582519531e-05,
      "step": 44946,
      "training_step_time": 7.663253307342529
    },
    {
      "epoch": 4.286479949951172e-05,
      "model_forward_time": 1.7196824550628662,
      "step": 44947
    },
    {
      "epoch": 4.286479949951172e-05,
      "step": 44947,
      "training_step_time": 7.7221763134002686
    },
    {
      "epoch": 4.2865753173828125e-05,
      "model_forward_time": 1.7080976963043213,
      "step": 44948
    },
    {
      "epoch": 4.2865753173828125e-05,
      "step": 44948,
      "training_step_time": 7.702342748641968
    },
    {
      "epoch": 4.286670684814453e-05,
      "model_forward_time": 1.7104496955871582,
      "step": 44949
    },
    {
      "epoch": 4.286670684814453e-05,
      "step": 44949,
      "training_step_time": 7.602582931518555
    },
    {
      "epoch": 4.286766052246094e-05,
      "grad_norm": 0.14053036272525787,
      "learning_rate": 6.235600343985848e-06,
      "loss": 0.0568,
      "step": 44950
    },
    {
      "epoch": 4.286766052246094e-05,
      "model_forward_time": 2.2651166915893555,
      "step": 44950
    },
    {
      "dynamics_loss_avg": 0.06285166256129741,
      "epoch": 4.286766052246094e-05,
      "step": 44950
    },
    {
      "action_loss_avg": 0.0026500888401642443,
      "epoch": 4.286766052246094e-05,
      "step": 44950
    },
    {
      "epoch": 4.286766052246094e-05,
      "step": 44950,
      "training_step_time": 7.719639301300049
    },
    {
      "epoch": 4.2868614196777344e-05,
      "model_forward_time": 1.7049245834350586,
      "step": 44951
    },
    {
      "epoch": 4.2868614196777344e-05,
      "step": 44951,
      "training_step_time": 7.787393569946289
    },
    {
      "epoch": 4.286956787109375e-05,
      "model_forward_time": 1.7029783725738525,
      "step": 44952
    },
    {
      "epoch": 4.286956787109375e-05,
      "step": 44952,
      "training_step_time": 8.376020908355713
    },
    {
      "epoch": 4.287052154541016e-05,
      "model_forward_time": 1.710585355758667,
      "step": 44953
    },
    {
      "epoch": 4.287052154541016e-05,
      "step": 44953,
      "training_step_time": 8.898844718933105
    },
    {
      "epoch": 4.287147521972656e-05,
      "model_forward_time": 1.7165486812591553,
      "step": 44954
    },
    {
      "epoch": 4.287147521972656e-05,
      "step": 44954,
      "training_step_time": 9.177054405212402
    },
    {
      "epoch": 4.287242889404297e-05,
      "model_forward_time": 1.719381332397461,
      "step": 44955
    },
    {
      "epoch": 4.287242889404297e-05,
      "step": 44955,
      "training_step_time": 8.290356874465942
    },
    {
      "epoch": 4.287338256835938e-05,
      "model_forward_time": 1.7245652675628662,
      "step": 44956
    },
    {
      "epoch": 4.287338256835938e-05,
      "step": 44956,
      "training_step_time": 8.734115362167358
    },
    {
      "epoch": 4.287433624267578e-05,
      "model_forward_time": 1.726642370223999,
      "step": 44957
    },
    {
      "epoch": 4.287433624267578e-05,
      "step": 44957,
      "training_step_time": 7.721797227859497
    },
    {
      "epoch": 4.2875289916992186e-05,
      "model_forward_time": 1.7166993618011475,
      "step": 44958
    },
    {
      "epoch": 4.2875289916992186e-05,
      "step": 44958,
      "training_step_time": 7.49118971824646
    },
    {
      "epoch": 4.28762435913086e-05,
      "model_forward_time": 1.7178473472595215,
      "step": 44959
    },
    {
      "epoch": 4.28762435913086e-05,
      "step": 44959,
      "training_step_time": 10.338193893432617
    },
    {
      "epoch": 4.2877197265625e-05,
      "grad_norm": 0.1321651041507721,
      "learning_rate": 6.233998089222729e-06,
      "loss": 0.0601,
      "step": 44960
    },
    {
      "epoch": 4.2877197265625e-05,
      "model_forward_time": 1.7179157733917236,
      "step": 44960
    },
    {
      "dynamics_loss_avg": 0.06821266878396273,
      "epoch": 4.2877197265625e-05,
      "step": 44960
    },
    {
      "action_loss_avg": 0.0025116618489846587,
      "epoch": 4.2877197265625e-05,
      "step": 44960
    },
    {
      "epoch": 4.2877197265625e-05,
      "step": 44960,
      "training_step_time": 7.623590707778931
    },
    {
      "epoch": 4.2878150939941405e-05,
      "model_forward_time": 1.7233173847198486,
      "step": 44961
    },
    {
      "epoch": 4.2878150939941405e-05,
      "step": 44961,
      "training_step_time": 9.564990997314453
    },
    {
      "epoch": 4.2879104614257816e-05,
      "model_forward_time": 1.7031428813934326,
      "step": 44962
    },
    {
      "epoch": 4.2879104614257816e-05,
      "step": 44962,
      "training_step_time": 7.731466054916382
    },
    {
      "epoch": 4.288005828857422e-05,
      "model_forward_time": 1.721574306488037,
      "step": 44963
    },
    {
      "epoch": 4.288005828857422e-05,
      "step": 44963,
      "training_step_time": 8.521547317504883
    },
    {
      "epoch": 4.2881011962890624e-05,
      "model_forward_time": 1.7202420234680176,
      "step": 44964
    },
    {
      "epoch": 4.2881011962890624e-05,
      "step": 44964,
      "training_step_time": 7.98561954498291
    },
    {
      "epoch": 4.2881965637207035e-05,
      "model_forward_time": 1.7231969833374023,
      "step": 44965
    },
    {
      "epoch": 4.2881965637207035e-05,
      "step": 44965,
      "training_step_time": 8.637370109558105
    },
    {
      "epoch": 4.288291931152344e-05,
      "model_forward_time": 1.7288601398468018,
      "step": 44966
    },
    {
      "epoch": 4.288291931152344e-05,
      "step": 44966,
      "training_step_time": 7.817509651184082
    },
    {
      "epoch": 4.288387298583984e-05,
      "model_forward_time": 1.7150673866271973,
      "step": 44967
    },
    {
      "epoch": 4.288387298583984e-05,
      "step": 44967,
      "training_step_time": 7.556220054626465
    },
    {
      "epoch": 4.288482666015625e-05,
      "model_forward_time": 1.7200117111206055,
      "step": 44968
    },
    {
      "epoch": 4.288482666015625e-05,
      "step": 44968,
      "training_step_time": 7.750823974609375
    },
    {
      "epoch": 4.288578033447266e-05,
      "model_forward_time": 2.2605459690093994,
      "step": 44969
    },
    {
      "epoch": 4.288578033447266e-05,
      "step": 44969,
      "training_step_time": 8.222720861434937
    },
    {
      "epoch": 4.288673400878906e-05,
      "grad_norm": 0.1167406290769577,
      "learning_rate": 6.232395699511433e-06,
      "loss": 0.0558,
      "step": 44970
    },
    {
      "epoch": 4.288673400878906e-05,
      "model_forward_time": 1.7179934978485107,
      "step": 44970
    },
    {
      "dynamics_loss_avg": 0.04729957841336727,
      "epoch": 4.288673400878906e-05,
      "step": 44970
    },
    {
      "action_loss_avg": 0.0014897150424076245,
      "epoch": 4.288673400878906e-05,
      "step": 44970
    },
    {
      "epoch": 4.288673400878906e-05,
      "step": 44970,
      "training_step_time": 9.785377264022827
    },
    {
      "epoch": 4.2887687683105465e-05,
      "model_forward_time": 1.723525047302246,
      "step": 44971
    },
    {
      "epoch": 4.2887687683105465e-05,
      "step": 44971,
      "training_step_time": 7.72475004196167
    },
    {
      "epoch": 4.2888641357421876e-05,
      "model_forward_time": 1.732668161392212,
      "step": 44972
    },
    {
      "epoch": 4.2888641357421876e-05,
      "step": 44972,
      "training_step_time": 7.63857364654541
    },
    {
      "epoch": 4.288959503173828e-05,
      "model_forward_time": 1.7244417667388916,
      "step": 44973
    },
    {
      "epoch": 4.288959503173828e-05,
      "step": 44973,
      "training_step_time": 8.274309158325195
    },
    {
      "epoch": 4.2890548706054684e-05,
      "model_forward_time": 1.7013719081878662,
      "step": 44974
    },
    {
      "epoch": 4.2890548706054684e-05,
      "step": 44974,
      "training_step_time": 7.442890405654907
    },
    {
      "epoch": 4.2891502380371095e-05,
      "model_forward_time": 1.7451412677764893,
      "step": 44975
    },
    {
      "epoch": 4.2891502380371095e-05,
      "step": 44975,
      "training_step_time": 8.632583618164062
    },
    {
      "epoch": 4.28924560546875e-05,
      "model_forward_time": 1.3560683727264404,
      "step": 44976
    },
    {
      "epoch": 4.28924560546875e-05,
      "step": 44976,
      "training_step_time": 8.067708015441895
    },
    {
      "epoch": 4.28934097290039e-05,
      "model_forward_time": 1.7257699966430664,
      "step": 44977
    },
    {
      "epoch": 4.28934097290039e-05,
      "step": 44977,
      "training_step_time": 7.5037147998809814
    },
    {
      "epoch": 4.2894363403320314e-05,
      "model_forward_time": 2.2788329124450684,
      "step": 44978
    },
    {
      "epoch": 4.2894363403320314e-05,
      "step": 44978,
      "training_step_time": 7.516269683837891
    },
    {
      "epoch": 4.289531707763672e-05,
      "model_forward_time": 1.7151944637298584,
      "step": 44979
    },
    {
      "epoch": 4.289531707763672e-05,
      "step": 44979,
      "training_step_time": 8.688289165496826
    },
    {
      "epoch": 4.289627075195312e-05,
      "grad_norm": 0.18079857528209686,
      "learning_rate": 6.230793175027199e-06,
      "loss": 0.0558,
      "step": 44980
    },
    {
      "epoch": 4.289627075195312e-05,
      "model_forward_time": 1.7173688411712646,
      "step": 44980
    },
    {
      "dynamics_loss_avg": 0.0683451272547245,
      "epoch": 4.289627075195312e-05,
      "step": 44980
    },
    {
      "action_loss_avg": 0.0014419642626307904,
      "epoch": 4.289627075195312e-05,
      "step": 44980
    },
    {
      "epoch": 4.289627075195312e-05,
      "step": 44980,
      "training_step_time": 8.554107666015625
    },
    {
      "epoch": 4.289722442626953e-05,
      "model_forward_time": 1.0176842212677002,
      "step": 44981
    },
    {
      "epoch": 4.289722442626953e-05,
      "step": 44981,
      "training_step_time": 8.464340209960938
    },
    {
      "epoch": 4.289817810058594e-05,
      "model_forward_time": 1.7235848903656006,
      "step": 44982
    },
    {
      "epoch": 4.289817810058594e-05,
      "step": 44982,
      "training_step_time": 7.691418409347534
    },
    {
      "epoch": 4.289913177490234e-05,
      "model_forward_time": 1.720768690109253,
      "step": 44983
    },
    {
      "epoch": 4.289913177490234e-05,
      "step": 44983,
      "training_step_time": 7.623842716217041
    },
    {
      "epoch": 4.290008544921875e-05,
      "model_forward_time": 1.7436823844909668,
      "step": 44984
    },
    {
      "epoch": 4.290008544921875e-05,
      "step": 44984,
      "training_step_time": 7.548965215682983
    },
    {
      "epoch": 4.2901039123535156e-05,
      "model_forward_time": 1.7173137664794922,
      "step": 44985
    },
    {
      "epoch": 4.2901039123535156e-05,
      "step": 44985,
      "training_step_time": 7.617153882980347
    },
    {
      "epoch": 4.290199279785156e-05,
      "model_forward_time": 1.7294471263885498,
      "step": 44986
    },
    {
      "epoch": 4.290199279785156e-05,
      "step": 44986,
      "training_step_time": 7.515593767166138
    },
    {
      "epoch": 4.290294647216797e-05,
      "model_forward_time": 2.277236223220825,
      "step": 44987
    },
    {
      "epoch": 4.290294647216797e-05,
      "step": 44987,
      "training_step_time": 7.536666393280029
    },
    {
      "epoch": 4.2903900146484375e-05,
      "model_forward_time": 1.7189903259277344,
      "step": 44988
    },
    {
      "epoch": 4.2903900146484375e-05,
      "step": 44988,
      "training_step_time": 8.31990361213684
    },
    {
      "epoch": 4.290485382080078e-05,
      "model_forward_time": 1.7298905849456787,
      "step": 44989
    },
    {
      "epoch": 4.290485382080078e-05,
      "step": 44989,
      "training_step_time": 8.522591352462769
    },
    {
      "epoch": 4.290580749511719e-05,
      "grad_norm": 0.09957462549209595,
      "learning_rate": 6.22919051594527e-06,
      "loss": 0.0556,
      "step": 44990
    },
    {
      "epoch": 4.290580749511719e-05,
      "model_forward_time": 1.7303049564361572,
      "step": 44990
    },
    {
      "dynamics_loss_avg": 0.0691197307780385,
      "epoch": 4.290580749511719e-05,
      "step": 44990
    },
    {
      "action_loss_avg": 0.0020394174294779076,
      "epoch": 4.290580749511719e-05,
      "step": 44990
    },
    {
      "epoch": 4.290580749511719e-05,
      "step": 44990,
      "training_step_time": 7.836328029632568
    },
    {
      "epoch": 4.2906761169433593e-05,
      "model_forward_time": 1.7285993099212646,
      "step": 44991
    },
    {
      "epoch": 4.2906761169433593e-05,
      "step": 44991,
      "training_step_time": 7.692339658737183
    },
    {
      "epoch": 4.290771484375e-05,
      "model_forward_time": 1.7234277725219727,
      "step": 44992
    },
    {
      "epoch": 4.290771484375e-05,
      "step": 44992,
      "training_step_time": 7.807795763015747
    },
    {
      "epoch": 4.290866851806641e-05,
      "model_forward_time": 1.749873161315918,
      "step": 44993
    },
    {
      "epoch": 4.290866851806641e-05,
      "step": 44993,
      "training_step_time": 7.723054885864258
    },
    {
      "epoch": 4.290962219238281e-05,
      "model_forward_time": 1.7257294654846191,
      "step": 44994
    },
    {
      "epoch": 4.290962219238281e-05,
      "step": 44994,
      "training_step_time": 7.740572452545166
    },
    {
      "epoch": 4.2910575866699216e-05,
      "model_forward_time": 1.7284808158874512,
      "step": 44995
    },
    {
      "epoch": 4.2910575866699216e-05,
      "step": 44995,
      "training_step_time": 7.649982213973999
    },
    {
      "epoch": 4.291152954101563e-05,
      "model_forward_time": 1.7240686416625977,
      "step": 44996
    },
    {
      "epoch": 4.291152954101563e-05,
      "step": 44996,
      "training_step_time": 8.746554136276245
    },
    {
      "epoch": 4.291248321533203e-05,
      "model_forward_time": 1.7336399555206299,
      "step": 44997
    },
    {
      "epoch": 4.291248321533203e-05,
      "step": 44997,
      "training_step_time": 8.321561336517334
    },
    {
      "epoch": 4.2913436889648435e-05,
      "model_forward_time": 1.7341697216033936,
      "step": 44998
    },
    {
      "epoch": 4.2913436889648435e-05,
      "step": 44998,
      "training_step_time": 9.117884397506714
    },
    {
      "epoch": 4.2914390563964846e-05,
      "model_forward_time": 1.7208080291748047,
      "step": 44999
    },
    {
      "epoch": 4.2914390563964846e-05,
      "step": 44999,
      "training_step_time": 7.943757772445679
    },
    {
      "epoch": 4.291534423828125e-05,
      "grad_norm": 0.16420210897922516,
      "learning_rate": 6.2275877224409175e-06,
      "loss": 0.0562,
      "step": 45000
    },
    {
      "epoch": 4.291534423828125e-05,
      "model_forward_time": 1.7331032752990723,
      "step": 45000
    },
    {
      "dynamics_loss_avg": 0.05418833661824465,
      "epoch": 4.291534423828125e-05,
      "step": 45000
    },
    {
      "action_loss_avg": 0.0022924928343854845,
      "epoch": 4.291534423828125e-05,
      "step": 45000
    },
    {
      "epoch": 4.291534423828125e-05,
      "step": 45000,
      "training_step_time": 8.883357048034668
    },
    {
      "epoch": 4.2916297912597654e-05,
      "model_forward_time": 1.7107021808624268,
      "step": 45001
    },
    {
      "epoch": 4.2916297912597654e-05,
      "step": 45001,
      "training_step_time": 8.68601655960083
    },
    {
      "epoch": 4.2917251586914065e-05,
      "model_forward_time": 1.7349810600280762,
      "step": 45002
    },
    {
      "epoch": 4.2917251586914065e-05,
      "step": 45002,
      "training_step_time": 8.336642503738403
    },
    {
      "epoch": 4.291820526123047e-05,
      "model_forward_time": 1.7146179676055908,
      "step": 45003
    },
    {
      "epoch": 4.291820526123047e-05,
      "step": 45003,
      "training_step_time": 7.881889343261719
    },
    {
      "epoch": 4.291915893554687e-05,
      "model_forward_time": 1.7272655963897705,
      "step": 45004
    },
    {
      "epoch": 4.291915893554687e-05,
      "step": 45004,
      "training_step_time": 8.377617835998535
    },
    {
      "epoch": 4.2920112609863284e-05,
      "model_forward_time": 2.7953433990478516,
      "step": 45005
    },
    {
      "epoch": 4.2920112609863284e-05,
      "step": 45005,
      "training_step_time": 8.67219877243042
    },
    {
      "epoch": 4.292106628417969e-05,
      "model_forward_time": 1.7213938236236572,
      "step": 45006
    },
    {
      "epoch": 4.292106628417969e-05,
      "step": 45006,
      "training_step_time": 10.235537767410278
    },
    {
      "epoch": 4.292201995849609e-05,
      "model_forward_time": 1.7180545330047607,
      "step": 45007
    },
    {
      "epoch": 4.292201995849609e-05,
      "step": 45007,
      "training_step_time": 8.857745170593262
    },
    {
      "epoch": 4.29229736328125e-05,
      "model_forward_time": 1.7172520160675049,
      "step": 45008
    },
    {
      "epoch": 4.29229736328125e-05,
      "step": 45008,
      "training_step_time": 9.661962509155273
    },
    {
      "epoch": 4.2923927307128907e-05,
      "model_forward_time": 1.719376564025879,
      "step": 45009
    },
    {
      "epoch": 4.2923927307128907e-05,
      "step": 45009,
      "training_step_time": 9.691621541976929
    },
    {
      "epoch": 4.292488098144531e-05,
      "grad_norm": 0.10826326906681061,
      "learning_rate": 6.225984794689414e-06,
      "loss": 0.0577,
      "step": 45010
    },
    {
      "epoch": 4.292488098144531e-05,
      "model_forward_time": 1.7306523323059082,
      "step": 45010
    },
    {
      "dynamics_loss_avg": 0.05764738656580448,
      "epoch": 4.292488098144531e-05,
      "step": 45010
    },
    {
      "action_loss_avg": 0.0020994934369809926,
      "epoch": 4.292488098144531e-05,
      "step": 45010
    },
    {
      "epoch": 4.292488098144531e-05,
      "step": 45010,
      "training_step_time": 8.5133798122406
    },
    {
      "epoch": 4.292583465576172e-05,
      "model_forward_time": 1.7824397087097168,
      "step": 45011
    },
    {
      "epoch": 4.292583465576172e-05,
      "step": 45011,
      "training_step_time": 7.919864654541016
    },
    {
      "epoch": 4.2926788330078125e-05,
      "model_forward_time": 1.7257757186889648,
      "step": 45012
    },
    {
      "epoch": 4.2926788330078125e-05,
      "step": 45012,
      "training_step_time": 7.827470541000366
    },
    {
      "epoch": 4.292774200439453e-05,
      "model_forward_time": 1.7274754047393799,
      "step": 45013
    },
    {
      "epoch": 4.292774200439453e-05,
      "step": 45013,
      "training_step_time": 7.714553117752075
    },
    {
      "epoch": 4.292869567871094e-05,
      "model_forward_time": 2.285754442214966,
      "step": 45014
    },
    {
      "epoch": 4.292869567871094e-05,
      "step": 45014,
      "training_step_time": 8.817518711090088
    },
    {
      "epoch": 4.2929649353027344e-05,
      "model_forward_time": 1.7285890579223633,
      "step": 45015
    },
    {
      "epoch": 4.2929649353027344e-05,
      "step": 45015,
      "training_step_time": 9.403099060058594
    },
    {
      "epoch": 4.293060302734375e-05,
      "model_forward_time": 1.7301714420318604,
      "step": 45016
    },
    {
      "epoch": 4.293060302734375e-05,
      "step": 45016,
      "training_step_time": 9.80710220336914
    },
    {
      "epoch": 4.293155670166016e-05,
      "model_forward_time": 1.7216737270355225,
      "step": 45017
    },
    {
      "epoch": 4.293155670166016e-05,
      "step": 45017,
      "training_step_time": 8.420267820358276
    },
    {
      "epoch": 4.293251037597656e-05,
      "model_forward_time": 1.729262113571167,
      "step": 45018
    },
    {
      "epoch": 4.293251037597656e-05,
      "step": 45018,
      "training_step_time": 8.713964223861694
    },
    {
      "epoch": 4.293346405029297e-05,
      "model_forward_time": 1.698310375213623,
      "step": 45019
    },
    {
      "epoch": 4.293346405029297e-05,
      "step": 45019,
      "training_step_time": 8.593825578689575
    },
    {
      "epoch": 4.293441772460938e-05,
      "grad_norm": 0.12494189292192459,
      "learning_rate": 6.224381732866059e-06,
      "loss": 0.0549,
      "step": 45020
    },
    {
      "epoch": 4.293441772460938e-05,
      "model_forward_time": 1.7721364498138428,
      "step": 45020
    },
    {
      "dynamics_loss_avg": 0.04234970649704337,
      "epoch": 4.293441772460938e-05,
      "step": 45020
    },
    {
      "action_loss_avg": 0.0020810277026612313,
      "epoch": 4.293441772460938e-05,
      "step": 45020
    },
    {
      "epoch": 4.293441772460938e-05,
      "step": 45020,
      "training_step_time": 7.869799375534058
    },
    {
      "epoch": 4.293537139892578e-05,
      "model_forward_time": 1.736931324005127,
      "step": 45021
    },
    {
      "epoch": 4.293537139892578e-05,
      "step": 45021,
      "training_step_time": 8.841991662979126
    },
    {
      "epoch": 4.2936325073242186e-05,
      "model_forward_time": 1.7380757331848145,
      "step": 45022
    },
    {
      "epoch": 4.2936325073242186e-05,
      "step": 45022,
      "training_step_time": 7.518481969833374
    },
    {
      "epoch": 4.29372787475586e-05,
      "model_forward_time": 2.2761898040771484,
      "step": 45023
    },
    {
      "epoch": 4.29372787475586e-05,
      "step": 45023,
      "training_step_time": 8.753933906555176
    },
    {
      "epoch": 4.2938232421875e-05,
      "model_forward_time": 1.7179245948791504,
      "step": 45024
    },
    {
      "epoch": 4.2938232421875e-05,
      "step": 45024,
      "training_step_time": 7.85595965385437
    },
    {
      "epoch": 4.2939186096191405e-05,
      "model_forward_time": 1.7368311882019043,
      "step": 45025
    },
    {
      "epoch": 4.2939186096191405e-05,
      "step": 45025,
      "training_step_time": 10.170299530029297
    },
    {
      "epoch": 4.2940139770507816e-05,
      "model_forward_time": 1.3654100894927979,
      "step": 45026
    },
    {
      "epoch": 4.2940139770507816e-05,
      "step": 45026,
      "training_step_time": 8.633487701416016
    },
    {
      "epoch": 4.294109344482422e-05,
      "model_forward_time": 1.7305338382720947,
      "step": 45027
    },
    {
      "epoch": 4.294109344482422e-05,
      "step": 45027,
      "training_step_time": 7.73321795463562
    },
    {
      "epoch": 4.2942047119140624e-05,
      "model_forward_time": 1.7266464233398438,
      "step": 45028
    },
    {
      "epoch": 4.2942047119140624e-05,
      "step": 45028,
      "training_step_time": 7.77262806892395
    },
    {
      "epoch": 4.2943000793457035e-05,
      "model_forward_time": 1.765049934387207,
      "step": 45029
    },
    {
      "epoch": 4.2943000793457035e-05,
      "step": 45029,
      "training_step_time": 7.856997728347778
    },
    {
      "epoch": 4.294395446777344e-05,
      "grad_norm": 0.12469246238470078,
      "learning_rate": 6.222778537146159e-06,
      "loss": 0.059,
      "step": 45030
    },
    {
      "epoch": 4.294395446777344e-05,
      "model_forward_time": 1.727900505065918,
      "step": 45030
    },
    {
      "dynamics_loss_avg": 0.0530069213360548,
      "epoch": 4.294395446777344e-05,
      "step": 45030
    },
    {
      "action_loss_avg": 0.0023928237525979055,
      "epoch": 4.294395446777344e-05,
      "step": 45030
    },
    {
      "epoch": 4.294395446777344e-05,
      "step": 45030,
      "training_step_time": 7.838019609451294
    },
    {
      "epoch": 4.294490814208984e-05,
      "model_forward_time": 1.7188105583190918,
      "step": 45031
    },
    {
      "epoch": 4.294490814208984e-05,
      "step": 45031,
      "training_step_time": 7.852345943450928
    },
    {
      "epoch": 4.294586181640625e-05,
      "model_forward_time": 1.727480411529541,
      "step": 45032
    },
    {
      "epoch": 4.294586181640625e-05,
      "step": 45032,
      "training_step_time": 7.671907901763916
    },
    {
      "epoch": 4.294681549072266e-05,
      "model_forward_time": 2.2901155948638916,
      "step": 45033
    },
    {
      "epoch": 4.294681549072266e-05,
      "step": 45033,
      "training_step_time": 7.7315497398376465
    },
    {
      "epoch": 4.294776916503906e-05,
      "model_forward_time": 1.7223057746887207,
      "step": 45034
    },
    {
      "epoch": 4.294776916503906e-05,
      "step": 45034,
      "training_step_time": 8.768003225326538
    },
    {
      "epoch": 4.2948722839355465e-05,
      "model_forward_time": 1.7328789234161377,
      "step": 45035
    },
    {
      "epoch": 4.2948722839355465e-05,
      "step": 45035,
      "training_step_time": 8.649027585983276
    },
    {
      "epoch": 4.2949676513671876e-05,
      "model_forward_time": 1.7250330448150635,
      "step": 45036
    },
    {
      "epoch": 4.2949676513671876e-05,
      "step": 45036,
      "training_step_time": 7.912349700927734
    },
    {
      "epoch": 4.295063018798828e-05,
      "model_forward_time": 1.7267765998840332,
      "step": 45037
    },
    {
      "epoch": 4.295063018798828e-05,
      "step": 45037,
      "training_step_time": 7.709204912185669
    },
    {
      "epoch": 4.2951583862304684e-05,
      "model_forward_time": 1.745509147644043,
      "step": 45038
    },
    {
      "epoch": 4.2951583862304684e-05,
      "step": 45038,
      "training_step_time": 8.352157592773438
    },
    {
      "epoch": 4.2952537536621095e-05,
      "model_forward_time": 1.7260029315948486,
      "step": 45039
    },
    {
      "epoch": 4.2952537536621095e-05,
      "step": 45039,
      "training_step_time": 8.48768401145935
    },
    {
      "epoch": 4.29534912109375e-05,
      "grad_norm": 0.15949784219264984,
      "learning_rate": 6.2211752077050366e-06,
      "loss": 0.0614,
      "step": 45040
    },
    {
      "epoch": 4.29534912109375e-05,
      "model_forward_time": 1.71995210647583,
      "step": 45040
    },
    {
      "dynamics_loss_avg": 0.04647956173866987,
      "epoch": 4.29534912109375e-05,
      "step": 45040
    },
    {
      "action_loss_avg": 0.003374523791717365,
      "epoch": 4.29534912109375e-05,
      "step": 45040
    },
    {
      "epoch": 4.29534912109375e-05,
      "step": 45040,
      "training_step_time": 7.769860029220581
    },
    {
      "epoch": 4.29544448852539e-05,
      "model_forward_time": 1.7249057292938232,
      "step": 45041
    },
    {
      "epoch": 4.29544448852539e-05,
      "step": 45041,
      "training_step_time": 7.6129889488220215
    },
    {
      "epoch": 4.2955398559570314e-05,
      "model_forward_time": 2.283418655395508,
      "step": 45042
    },
    {
      "epoch": 4.2955398559570314e-05,
      "step": 45042,
      "training_step_time": 7.66515851020813
    },
    {
      "epoch": 4.295635223388672e-05,
      "model_forward_time": 1.7238643169403076,
      "step": 45043
    },
    {
      "epoch": 4.295635223388672e-05,
      "step": 45043,
      "training_step_time": 11.067163467407227
    },
    {
      "epoch": 4.295730590820312e-05,
      "model_forward_time": 1.7240016460418701,
      "step": 45044
    },
    {
      "epoch": 4.295730590820312e-05,
      "step": 45044,
      "training_step_time": 8.465424537658691
    },
    {
      "epoch": 4.295825958251953e-05,
      "model_forward_time": 1.721900224685669,
      "step": 45045
    },
    {
      "epoch": 4.295825958251953e-05,
      "step": 45045,
      "training_step_time": 7.793576955795288
    },
    {
      "epoch": 4.295921325683594e-05,
      "model_forward_time": 1.7248446941375732,
      "step": 45046
    },
    {
      "epoch": 4.295921325683594e-05,
      "step": 45046,
      "training_step_time": 9.028137922286987
    },
    {
      "epoch": 4.296016693115234e-05,
      "model_forward_time": 1.733898401260376,
      "step": 45047
    },
    {
      "epoch": 4.296016693115234e-05,
      "step": 45047,
      "training_step_time": 8.631841659545898
    },
    {
      "epoch": 4.296112060546875e-05,
      "model_forward_time": 1.7280206680297852,
      "step": 45048
    },
    {
      "epoch": 4.296112060546875e-05,
      "step": 45048,
      "training_step_time": 6.9260413646698
    },
    {
      "epoch": 4.2962074279785156e-05,
      "model_forward_time": 1.725395917892456,
      "step": 45049
    },
    {
      "epoch": 4.2962074279785156e-05,
      "step": 45049,
      "training_step_time": 6.9440906047821045
    },
    {
      "epoch": 4.296302795410156e-05,
      "grad_norm": 0.14568158984184265,
      "learning_rate": 6.219571744718028e-06,
      "loss": 0.0564,
      "step": 45050
    },
    {
      "epoch": 4.296302795410156e-05,
      "model_forward_time": 1.7163257598876953,
      "step": 45050
    },
    {
      "dynamics_loss_avg": 0.05257826577872038,
      "epoch": 4.296302795410156e-05,
      "step": 45050
    },
    {
      "action_loss_avg": 0.0017328607093077153,
      "epoch": 4.296302795410156e-05,
      "step": 45050
    },
    {
      "epoch": 4.296302795410156e-05,
      "step": 45050,
      "training_step_time": 6.970450401306152
    },
    {
      "epoch": 4.296398162841797e-05,
      "model_forward_time": 1.7288212776184082,
      "step": 45051
    },
    {
      "epoch": 4.296398162841797e-05,
      "step": 45051,
      "training_step_time": 6.981400966644287
    },
    {
      "epoch": 4.2964935302734375e-05,
      "model_forward_time": 1.7224500179290771,
      "step": 45052
    },
    {
      "epoch": 4.2964935302734375e-05,
      "step": 45052,
      "training_step_time": 7.035156011581421
    },
    {
      "epoch": 4.296588897705078e-05,
      "model_forward_time": 1.7228155136108398,
      "step": 45053
    },
    {
      "epoch": 4.296588897705078e-05,
      "step": 45053,
      "training_step_time": 6.969566106796265
    },
    {
      "epoch": 4.296684265136719e-05,
      "model_forward_time": 1.7189936637878418,
      "step": 45054
    },
    {
      "epoch": 4.296684265136719e-05,
      "step": 45054,
      "training_step_time": 6.921910524368286
    },
    {
      "epoch": 4.2967796325683593e-05,
      "model_forward_time": 1.7136905193328857,
      "step": 45055
    },
    {
      "epoch": 4.2967796325683593e-05,
      "step": 45055,
      "training_step_time": 6.97503399848938
    },
    {
      "epoch": 4.296875e-05,
      "model_forward_time": 1.9400577545166016,
      "step": 45056
    },
    {
      "epoch": 4.296875e-05,
      "step": 45056,
      "training_step_time": 9.734283208847046
    },
    {
      "epoch": 4.296970367431641e-05,
      "model_forward_time": 1.723294973373413,
      "step": 45057
    },
    {
      "epoch": 4.296970367431641e-05,
      "step": 45057,
      "training_step_time": 7.806843042373657
    },
    {
      "epoch": 4.297065734863281e-05,
      "model_forward_time": 1.7391855716705322,
      "step": 45058
    },
    {
      "epoch": 4.297065734863281e-05,
      "step": 45058,
      "training_step_time": 6.967514753341675
    },
    {
      "epoch": 4.2971611022949216e-05,
      "model_forward_time": 1.7182457447052002,
      "step": 45059
    },
    {
      "epoch": 4.2971611022949216e-05,
      "step": 45059,
      "training_step_time": 6.929041147232056
    },
    {
      "epoch": 4.297256469726563e-05,
      "grad_norm": 0.16906383633613586,
      "learning_rate": 6.217968148360488e-06,
      "loss": 0.056,
      "step": 45060
    },
    {
      "epoch": 4.297256469726563e-05,
      "model_forward_time": 1.7274131774902344,
      "step": 45060
    },
    {
      "dynamics_loss_avg": 0.05865752212703228,
      "epoch": 4.297256469726563e-05,
      "step": 45060
    },
    {
      "action_loss_avg": 0.002008428314002231,
      "epoch": 4.297256469726563e-05,
      "step": 45060
    },
    {
      "epoch": 4.297256469726563e-05,
      "step": 45060,
      "training_step_time": 6.9422876834869385
    },
    {
      "epoch": 4.297351837158203e-05,
      "model_forward_time": 1.3754475116729736,
      "step": 45061
    },
    {
      "epoch": 4.297351837158203e-05,
      "step": 45061,
      "training_step_time": 6.9830334186553955
    },
    {
      "epoch": 4.2974472045898435e-05,
      "model_forward_time": 1.72517991065979,
      "step": 45062
    },
    {
      "epoch": 4.2974472045898435e-05,
      "step": 45062,
      "training_step_time": 6.945084571838379
    },
    {
      "epoch": 4.2975425720214846e-05,
      "model_forward_time": 1.7355420589447021,
      "step": 45063
    },
    {
      "epoch": 4.2975425720214846e-05,
      "step": 45063,
      "training_step_time": 7.004816055297852
    },
    {
      "epoch": 4.297637939453125e-05,
      "model_forward_time": 1.7243154048919678,
      "step": 45064
    },
    {
      "epoch": 4.297637939453125e-05,
      "step": 45064,
      "training_step_time": 6.959244728088379
    },
    {
      "epoch": 4.2977333068847654e-05,
      "model_forward_time": 2.282978057861328,
      "step": 45065
    },
    {
      "epoch": 4.2977333068847654e-05,
      "step": 45065,
      "training_step_time": 8.913084983825684
    },
    {
      "epoch": 4.2978286743164065e-05,
      "model_forward_time": 1.740633249282837,
      "step": 45066
    },
    {
      "epoch": 4.2978286743164065e-05,
      "step": 45066,
      "training_step_time": 7.862893342971802
    },
    {
      "epoch": 4.297924041748047e-05,
      "model_forward_time": 1.373483419418335,
      "step": 45067
    },
    {
      "epoch": 4.297924041748047e-05,
      "step": 45067,
      "training_step_time": 7.805432081222534
    },
    {
      "epoch": 4.298019409179687e-05,
      "model_forward_time": 1.7299597263336182,
      "step": 45068
    },
    {
      "epoch": 4.298019409179687e-05,
      "step": 45068,
      "training_step_time": 6.954068660736084
    },
    {
      "epoch": 4.2981147766113284e-05,
      "model_forward_time": 1.7272372245788574,
      "step": 45069
    },
    {
      "epoch": 4.2981147766113284e-05,
      "step": 45069,
      "training_step_time": 6.964723587036133
    },
    {
      "epoch": 4.298210144042969e-05,
      "grad_norm": 0.19869093596935272,
      "learning_rate": 6.216364418807782e-06,
      "loss": 0.057,
      "step": 45070
    },
    {
      "epoch": 4.298210144042969e-05,
      "model_forward_time": 1.7326643466949463,
      "step": 45070
    },
    {
      "dynamics_loss_avg": 0.048729466088116166,
      "epoch": 4.298210144042969e-05,
      "step": 45070
    },
    {
      "action_loss_avg": 0.0013396483613178135,
      "epoch": 4.298210144042969e-05,
      "step": 45070
    },
    {
      "epoch": 4.298210144042969e-05,
      "step": 45070,
      "training_step_time": 6.912763357162476
    },
    {
      "epoch": 4.298305511474609e-05,
      "model_forward_time": 1.7241804599761963,
      "step": 45071
    },
    {
      "epoch": 4.298305511474609e-05,
      "step": 45071,
      "training_step_time": 6.940306901931763
    },
    {
      "epoch": 4.29840087890625e-05,
      "model_forward_time": 1.726562738418579,
      "step": 45072
    },
    {
      "epoch": 4.29840087890625e-05,
      "step": 45072,
      "training_step_time": 7.045874357223511
    },
    {
      "epoch": 4.298496246337891e-05,
      "model_forward_time": 1.731431245803833,
      "step": 45073
    },
    {
      "epoch": 4.298496246337891e-05,
      "step": 45073,
      "training_step_time": 6.956609725952148
    },
    {
      "epoch": 4.298591613769531e-05,
      "model_forward_time": 1.3818295001983643,
      "step": 45074
    },
    {
      "epoch": 4.298591613769531e-05,
      "step": 45074,
      "training_step_time": 7.815906763076782
    },
    {
      "epoch": 4.298686981201172e-05,
      "model_forward_time": 2.2999746799468994,
      "step": 45075
    },
    {
      "epoch": 4.298686981201172e-05,
      "step": 45075,
      "training_step_time": 8.830417394638062
    },
    {
      "epoch": 4.2987823486328125e-05,
      "model_forward_time": 1.7312254905700684,
      "step": 45076
    },
    {
      "epoch": 4.2987823486328125e-05,
      "step": 45076,
      "training_step_time": 7.784189939498901
    },
    {
      "epoch": 4.298877716064453e-05,
      "model_forward_time": 1.7333247661590576,
      "step": 45077
    },
    {
      "epoch": 4.298877716064453e-05,
      "step": 45077,
      "training_step_time": 6.8106513023376465
    },
    {
      "epoch": 4.298973083496094e-05,
      "model_forward_time": 1.7456629276275635,
      "step": 45078
    },
    {
      "epoch": 4.298973083496094e-05,
      "step": 45078,
      "training_step_time": 6.800377130508423
    },
    {
      "epoch": 4.2990684509277344e-05,
      "model_forward_time": 1.7338299751281738,
      "step": 45079
    },
    {
      "epoch": 4.2990684509277344e-05,
      "step": 45079,
      "training_step_time": 6.77937126159668
    },
    {
      "epoch": 4.299163818359375e-05,
      "grad_norm": 0.1336815059185028,
      "learning_rate": 6.214760556235292e-06,
      "loss": 0.0576,
      "step": 45080
    },
    {
      "epoch": 4.299163818359375e-05,
      "model_forward_time": 1.7237935066223145,
      "step": 45080
    },
    {
      "dynamics_loss_avg": 0.03744803424924612,
      "epoch": 4.299163818359375e-05,
      "step": 45080
    },
    {
      "action_loss_avg": 0.0009347912156954408,
      "epoch": 4.299163818359375e-05,
      "step": 45080
    },
    {
      "epoch": 4.299163818359375e-05,
      "step": 45080,
      "training_step_time": 6.773665428161621
    },
    {
      "epoch": 4.299259185791016e-05,
      "model_forward_time": 1.7237136363983154,
      "step": 45081
    },
    {
      "epoch": 4.299259185791016e-05,
      "step": 45081,
      "training_step_time": 6.781815052032471
    },
    {
      "epoch": 4.299354553222656e-05,
      "model_forward_time": 1.7218403816223145,
      "step": 45082
    },
    {
      "epoch": 4.299354553222656e-05,
      "step": 45082,
      "training_step_time": 6.7286412715911865
    },
    {
      "epoch": 4.299449920654297e-05,
      "model_forward_time": 1.7592754364013672,
      "step": 45083
    },
    {
      "epoch": 4.299449920654297e-05,
      "step": 45083,
      "training_step_time": 7.294785737991333
    },
    {
      "epoch": 4.299545288085938e-05,
      "model_forward_time": 2.27704119682312,
      "step": 45084
    },
    {
      "epoch": 4.299545288085938e-05,
      "step": 45084,
      "training_step_time": 9.7395601272583
    },
    {
      "epoch": 4.299640655517578e-05,
      "model_forward_time": 1.7356367111206055,
      "step": 45085
    },
    {
      "epoch": 4.299640655517578e-05,
      "step": 45085,
      "training_step_time": 7.78939414024353
    },
    {
      "epoch": 4.2997360229492186e-05,
      "model_forward_time": 1.7282590866088867,
      "step": 45086
    },
    {
      "epoch": 4.2997360229492186e-05,
      "step": 45086,
      "training_step_time": 7.606912136077881
    },
    {
      "epoch": 4.29983139038086e-05,
      "model_forward_time": 1.7353262901306152,
      "step": 45087
    },
    {
      "epoch": 4.29983139038086e-05,
      "step": 45087,
      "training_step_time": 6.80400013923645
    },
    {
      "epoch": 4.2999267578125e-05,
      "model_forward_time": 1.7340738773345947,
      "step": 45088
    },
    {
      "epoch": 4.2999267578125e-05,
      "step": 45088,
      "training_step_time": 6.816522121429443
    },
    {
      "epoch": 4.3000221252441405e-05,
      "model_forward_time": 1.7301764488220215,
      "step": 45089
    },
    {
      "epoch": 4.3000221252441405e-05,
      "step": 45089,
      "training_step_time": 6.818304061889648
    },
    {
      "epoch": 4.3001174926757816e-05,
      "grad_norm": 0.15454129874706268,
      "learning_rate": 6.213156560818414e-06,
      "loss": 0.0572,
      "step": 45090
    },
    {
      "epoch": 4.3001174926757816e-05,
      "model_forward_time": 1.7305564880371094,
      "step": 45090
    },
    {
      "dynamics_loss_avg": 0.04352178480476141,
      "epoch": 4.3001174926757816e-05,
      "step": 45090
    },
    {
      "action_loss_avg": 0.0010391955758677796,
      "epoch": 4.3001174926757816e-05,
      "step": 45090
    },
    {
      "epoch": 4.3001174926757816e-05,
      "step": 45090,
      "training_step_time": 6.788272857666016
    },
    {
      "epoch": 4.300212860107422e-05,
      "model_forward_time": 1.7169218063354492,
      "step": 45091
    },
    {
      "epoch": 4.300212860107422e-05,
      "step": 45091,
      "training_step_time": 6.697626113891602
    },
    {
      "epoch": 4.3003082275390624e-05,
      "model_forward_time": 1.7361035346984863,
      "step": 45092
    },
    {
      "epoch": 4.3003082275390624e-05,
      "step": 45092,
      "training_step_time": 7.272558689117432
    },
    {
      "epoch": 4.3004035949707035e-05,
      "model_forward_time": 2.2902321815490723,
      "step": 45093
    },
    {
      "epoch": 4.3004035949707035e-05,
      "step": 45093,
      "training_step_time": 10.632526159286499
    },
    {
      "epoch": 4.300498962402344e-05,
      "model_forward_time": 1.7303290367126465,
      "step": 45094
    },
    {
      "epoch": 4.300498962402344e-05,
      "step": 45094,
      "training_step_time": 7.8746795654296875
    },
    {
      "epoch": 4.300594329833984e-05,
      "model_forward_time": 1.7319529056549072,
      "step": 45095
    },
    {
      "epoch": 4.300594329833984e-05,
      "step": 45095,
      "training_step_time": 7.634432792663574
    },
    {
      "epoch": 4.300689697265625e-05,
      "model_forward_time": 1.7364025115966797,
      "step": 45096
    },
    {
      "epoch": 4.300689697265625e-05,
      "step": 45096,
      "training_step_time": 6.7633092403411865
    },
    {
      "epoch": 4.300785064697266e-05,
      "model_forward_time": 1.7286145687103271,
      "step": 45097
    },
    {
      "epoch": 4.300785064697266e-05,
      "step": 45097,
      "training_step_time": 6.816047668457031
    },
    {
      "epoch": 4.300880432128906e-05,
      "model_forward_time": 1.7299189567565918,
      "step": 45098
    },
    {
      "epoch": 4.300880432128906e-05,
      "step": 45098,
      "training_step_time": 6.823145389556885
    },
    {
      "epoch": 4.3009757995605466e-05,
      "model_forward_time": 1.7336177825927734,
      "step": 45099
    },
    {
      "epoch": 4.3009757995605466e-05,
      "step": 45099,
      "training_step_time": 6.77936315536499
    },
    {
      "epoch": 4.3010711669921876e-05,
      "grad_norm": 0.14237524569034576,
      "learning_rate": 6.211552432732558e-06,
      "loss": 0.0599,
      "step": 45100
    },
    {
      "epoch": 4.3010711669921876e-05,
      "model_forward_time": 1.3725650310516357,
      "step": 45100
    },
    {
      "dynamics_loss_avg": 0.07160263694822788,
      "epoch": 4.3010711669921876e-05,
      "step": 45100
    },
    {
      "action_loss_avg": 0.002035546160186641,
      "epoch": 4.3010711669921876e-05,
      "step": 45100
    },
    {
      "epoch": 4.3010711669921876e-05,
      "step": 45100,
      "training_step_time": 6.813767433166504
    },
    {
      "epoch": 4.301166534423828e-05,
      "model_forward_time": 1.7313165664672852,
      "step": 45101
    },
    {
      "epoch": 4.301166534423828e-05,
      "step": 45101,
      "training_step_time": 6.80261492729187
    },
    {
      "epoch": 4.3012619018554684e-05,
      "model_forward_time": 2.2847955226898193,
      "step": 45102
    },
    {
      "epoch": 4.3012619018554684e-05,
      "step": 45102,
      "training_step_time": 8.613637924194336
    },
    {
      "epoch": 4.3013572692871095e-05,
      "model_forward_time": 1.7325539588928223,
      "step": 45103
    },
    {
      "epoch": 4.3013572692871095e-05,
      "step": 45103,
      "training_step_time": 7.865996599197388
    },
    {
      "epoch": 4.30145263671875e-05,
      "model_forward_time": 1.7337653636932373,
      "step": 45104
    },
    {
      "epoch": 4.30145263671875e-05,
      "step": 45104,
      "training_step_time": 7.685603141784668
    },
    {
      "epoch": 4.30154800415039e-05,
      "model_forward_time": 1.7199578285217285,
      "step": 45105
    },
    {
      "epoch": 4.30154800415039e-05,
      "step": 45105,
      "training_step_time": 6.851216793060303
    },
    {
      "epoch": 4.3016433715820314e-05,
      "model_forward_time": 1.7308316230773926,
      "step": 45106
    },
    {
      "epoch": 4.3016433715820314e-05,
      "step": 45106,
      "training_step_time": 6.79223108291626
    },
    {
      "epoch": 4.301738739013672e-05,
      "model_forward_time": 1.738771677017212,
      "step": 45107
    },
    {
      "epoch": 4.301738739013672e-05,
      "step": 45107,
      "training_step_time": 6.8053412437438965
    },
    {
      "epoch": 4.301834106445312e-05,
      "model_forward_time": 1.7339024543762207,
      "step": 45108
    },
    {
      "epoch": 4.301834106445312e-05,
      "step": 45108,
      "training_step_time": 6.825083255767822
    },
    {
      "epoch": 4.301929473876953e-05,
      "model_forward_time": 1.7419040203094482,
      "step": 45109
    },
    {
      "epoch": 4.301929473876953e-05,
      "step": 45109,
      "training_step_time": 6.802623748779297
    },
    {
      "epoch": 4.302024841308594e-05,
      "grad_norm": 0.10609930008649826,
      "learning_rate": 6.20994817215315e-06,
      "loss": 0.0583,
      "step": 45110
    },
    {
      "epoch": 4.302024841308594e-05,
      "model_forward_time": 1.7392573356628418,
      "step": 45110
    },
    {
      "dynamics_loss_avg": 0.05015111435204744,
      "epoch": 4.302024841308594e-05,
      "step": 45110
    },
    {
      "action_loss_avg": 0.0018419194966554642,
      "epoch": 4.302024841308594e-05,
      "step": 45110
    },
    {
      "epoch": 4.302024841308594e-05,
      "step": 45110,
      "training_step_time": 6.767135143280029
    },
    {
      "epoch": 4.302120208740234e-05,
      "model_forward_time": 1.7773716449737549,
      "step": 45111
    },
    {
      "epoch": 4.302120208740234e-05,
      "step": 45111,
      "training_step_time": 9.005770683288574
    },
    {
      "epoch": 4.302215576171875e-05,
      "model_forward_time": 2.2679243087768555,
      "step": 45112
    },
    {
      "epoch": 4.302215576171875e-05,
      "step": 45112,
      "training_step_time": 7.769594430923462
    },
    {
      "epoch": 4.3023109436035156e-05,
      "model_forward_time": 1.7226223945617676,
      "step": 45113
    },
    {
      "epoch": 4.3023109436035156e-05,
      "step": 45113,
      "training_step_time": 7.768036603927612
    },
    {
      "epoch": 4.302406311035156e-05,
      "model_forward_time": 1.369654893875122,
      "step": 45114
    },
    {
      "epoch": 4.302406311035156e-05,
      "step": 45114,
      "training_step_time": 6.80293607711792
    },
    {
      "epoch": 4.302501678466797e-05,
      "model_forward_time": 1.7288029193878174,
      "step": 45115
    },
    {
      "epoch": 4.302501678466797e-05,
      "step": 45115,
      "training_step_time": 6.809473752975464
    },
    {
      "epoch": 4.3025970458984375e-05,
      "model_forward_time": 1.7297742366790771,
      "step": 45116
    },
    {
      "epoch": 4.3025970458984375e-05,
      "step": 45116,
      "training_step_time": 6.776905298233032
    },
    {
      "epoch": 4.302692413330078e-05,
      "model_forward_time": 1.7318308353424072,
      "step": 45117
    },
    {
      "epoch": 4.302692413330078e-05,
      "step": 45117,
      "training_step_time": 6.824160575866699
    },
    {
      "epoch": 4.302787780761719e-05,
      "model_forward_time": 1.730644941329956,
      "step": 45118
    },
    {
      "epoch": 4.302787780761719e-05,
      "step": 45118,
      "training_step_time": 6.857182264328003
    },
    {
      "epoch": 4.3028831481933594e-05,
      "model_forward_time": 1.7292323112487793,
      "step": 45119
    },
    {
      "epoch": 4.3028831481933594e-05,
      "step": 45119,
      "training_step_time": 6.842198371887207
    },
    {
      "epoch": 4.302978515625e-05,
      "grad_norm": 0.14705196022987366,
      "learning_rate": 6.208343779255629e-06,
      "loss": 0.0571,
      "step": 45120
    },
    {
      "epoch": 4.302978515625e-05,
      "model_forward_time": 1.774489164352417,
      "step": 45120
    },
    {
      "dynamics_loss_avg": 0.0520829239860177,
      "epoch": 4.302978515625e-05,
      "step": 45120
    },
    {
      "action_loss_avg": 0.0017539645545184613,
      "epoch": 4.302978515625e-05,
      "step": 45120
    },
    {
      "epoch": 4.302978515625e-05,
      "step": 45120,
      "training_step_time": 8.50605583190918
    },
    {
      "epoch": 4.303073883056641e-05,
      "model_forward_time": 2.274174690246582,
      "step": 45121
    },
    {
      "epoch": 4.303073883056641e-05,
      "step": 45121,
      "training_step_time": 11.721882343292236
    },
    {
      "epoch": 4.303169250488281e-05,
      "model_forward_time": 1.731525182723999,
      "step": 45122
    },
    {
      "epoch": 4.303169250488281e-05,
      "step": 45122,
      "training_step_time": 7.753952264785767
    },
    {
      "epoch": 4.3032646179199216e-05,
      "model_forward_time": 1.736558198928833,
      "step": 45123
    },
    {
      "epoch": 4.3032646179199216e-05,
      "step": 45123,
      "training_step_time": 7.588478088378906
    },
    {
      "epoch": 4.303359985351563e-05,
      "model_forward_time": 1.7365086078643799,
      "step": 45124
    },
    {
      "epoch": 4.303359985351563e-05,
      "step": 45124,
      "training_step_time": 6.874630689620972
    },
    {
      "epoch": 4.303455352783203e-05,
      "model_forward_time": 1.7443246841430664,
      "step": 45125
    },
    {
      "epoch": 4.303455352783203e-05,
      "step": 45125,
      "training_step_time": 6.8137876987457275
    },
    {
      "epoch": 4.3035507202148435e-05,
      "model_forward_time": 1.7381713390350342,
      "step": 45126
    },
    {
      "epoch": 4.3035507202148435e-05,
      "step": 45126,
      "training_step_time": 6.842316150665283
    },
    {
      "epoch": 4.3036460876464846e-05,
      "model_forward_time": 1.733668327331543,
      "step": 45127
    },
    {
      "epoch": 4.3036460876464846e-05,
      "step": 45127,
      "training_step_time": 6.897844552993774
    },
    {
      "epoch": 4.303741455078125e-05,
      "model_forward_time": 1.7354965209960938,
      "step": 45128
    },
    {
      "epoch": 4.303741455078125e-05,
      "step": 45128,
      "training_step_time": 6.832388639450073
    },
    {
      "epoch": 4.3038368225097654e-05,
      "model_forward_time": 1.75423264503479,
      "step": 45129
    },
    {
      "epoch": 4.3038368225097654e-05,
      "step": 45129,
      "training_step_time": 8.8233163356781
    },
    {
      "epoch": 4.3039321899414065e-05,
      "grad_norm": 0.10241619497537613,
      "learning_rate": 6.206739254215449e-06,
      "loss": 0.057,
      "step": 45130
    },
    {
      "epoch": 4.3039321899414065e-05,
      "model_forward_time": 2.290876626968384,
      "step": 45130
    },
    {
      "dynamics_loss_avg": 0.07724115438759327,
      "epoch": 4.3039321899414065e-05,
      "step": 45130
    },
    {
      "action_loss_avg": 0.00574079126236029,
      "epoch": 4.3039321899414065e-05,
      "step": 45130
    },
    {
      "epoch": 4.3039321899414065e-05,
      "step": 45130,
      "training_step_time": 9.398436546325684
    },
    {
      "epoch": 4.304027557373047e-05,
      "model_forward_time": 1.0367317199707031,
      "step": 45131
    },
    {
      "epoch": 4.304027557373047e-05,
      "step": 45131,
      "training_step_time": 7.797492742538452
    },
    {
      "epoch": 4.304122924804687e-05,
      "model_forward_time": 1.74119234085083,
      "step": 45132
    },
    {
      "epoch": 4.304122924804687e-05,
      "step": 45132,
      "training_step_time": 7.781399726867676
    },
    {
      "epoch": 4.3042182922363284e-05,
      "model_forward_time": 1.7368407249450684,
      "step": 45133
    },
    {
      "epoch": 4.3042182922363284e-05,
      "step": 45133,
      "training_step_time": 6.890392065048218
    },
    {
      "epoch": 4.304313659667969e-05,
      "model_forward_time": 1.3781623840332031,
      "step": 45134
    },
    {
      "epoch": 4.304313659667969e-05,
      "step": 45134,
      "training_step_time": 6.882254600524902
    },
    {
      "epoch": 4.304409027099609e-05,
      "model_forward_time": 1.7393643856048584,
      "step": 45135
    },
    {
      "epoch": 4.304409027099609e-05,
      "step": 45135,
      "training_step_time": 6.905102252960205
    },
    {
      "epoch": 4.30450439453125e-05,
      "model_forward_time": 1.73004150390625,
      "step": 45136
    },
    {
      "epoch": 4.30450439453125e-05,
      "step": 45136,
      "training_step_time": 6.861452102661133
    },
    {
      "epoch": 4.304599761962891e-05,
      "model_forward_time": 1.7439799308776855,
      "step": 45137
    },
    {
      "epoch": 4.304599761962891e-05,
      "step": 45137,
      "training_step_time": 6.855764389038086
    },
    {
      "epoch": 4.304695129394531e-05,
      "model_forward_time": 1.746814489364624,
      "step": 45138
    },
    {
      "epoch": 4.304695129394531e-05,
      "step": 45138,
      "training_step_time": 7.7875075340271
    },
    {
      "epoch": 4.304790496826172e-05,
      "model_forward_time": 1.7350187301635742,
      "step": 45139
    },
    {
      "epoch": 4.304790496826172e-05,
      "step": 45139,
      "training_step_time": 11.079455375671387
    },
    {
      "epoch": 4.3048858642578126e-05,
      "grad_norm": 0.09507086873054504,
      "learning_rate": 6.205134597208078e-06,
      "loss": 0.0571,
      "step": 45140
    },
    {
      "epoch": 4.3048858642578126e-05,
      "model_forward_time": 1.745903491973877,
      "step": 45140
    },
    {
      "dynamics_loss_avg": 0.04739253986626864,
      "epoch": 4.3048858642578126e-05,
      "step": 45140
    },
    {
      "action_loss_avg": 0.0018252143287099897,
      "epoch": 4.3048858642578126e-05,
      "step": 45140
    },
    {
      "epoch": 4.3048858642578126e-05,
      "step": 45140,
      "training_step_time": 7.816860198974609
    },
    {
      "epoch": 4.304981231689453e-05,
      "model_forward_time": 1.7385733127593994,
      "step": 45141
    },
    {
      "epoch": 4.304981231689453e-05,
      "step": 45141,
      "training_step_time": 7.811630010604858
    },
    {
      "epoch": 4.305076599121094e-05,
      "model_forward_time": 1.7504668235778809,
      "step": 45142
    },
    {
      "epoch": 4.305076599121094e-05,
      "step": 45142,
      "training_step_time": 6.845350027084351
    },
    {
      "epoch": 4.3051719665527344e-05,
      "model_forward_time": 1.7335796356201172,
      "step": 45143
    },
    {
      "epoch": 4.3051719665527344e-05,
      "step": 45143,
      "training_step_time": 6.829631567001343
    },
    {
      "epoch": 4.305267333984375e-05,
      "model_forward_time": 1.7350294589996338,
      "step": 45144
    },
    {
      "epoch": 4.305267333984375e-05,
      "step": 45144,
      "training_step_time": 6.862313508987427
    },
    {
      "epoch": 4.305362701416016e-05,
      "model_forward_time": 1.742823839187622,
      "step": 45145
    },
    {
      "epoch": 4.305362701416016e-05,
      "step": 45145,
      "training_step_time": 6.877833366394043
    },
    {
      "epoch": 4.305458068847656e-05,
      "model_forward_time": 1.739990472793579,
      "step": 45146
    },
    {
      "epoch": 4.305458068847656e-05,
      "step": 45146,
      "training_step_time": 6.853485822677612
    },
    {
      "epoch": 4.305553436279297e-05,
      "model_forward_time": 1.742185115814209,
      "step": 45147
    },
    {
      "epoch": 4.305553436279297e-05,
      "step": 45147,
      "training_step_time": 7.80322527885437
    },
    {
      "epoch": 4.305648803710938e-05,
      "model_forward_time": 1.7871625423431396,
      "step": 45148
    },
    {
      "epoch": 4.305648803710938e-05,
      "step": 45148,
      "training_step_time": 9.993339776992798
    },
    {
      "epoch": 4.305744171142578e-05,
      "model_forward_time": 1.7443194389343262,
      "step": 45149
    },
    {
      "epoch": 4.305744171142578e-05,
      "step": 45149,
      "training_step_time": 8.687140941619873
    },
    {
      "epoch": 4.3058395385742186e-05,
      "grad_norm": 0.09259814023971558,
      "learning_rate": 6.203529808409e-06,
      "loss": 0.0586,
      "step": 45150
    },
    {
      "epoch": 4.3058395385742186e-05,
      "model_forward_time": 1.7350046634674072,
      "step": 45150
    },
    {
      "dynamics_loss_avg": 0.055488364212214945,
      "epoch": 4.3058395385742186e-05,
      "step": 45150
    },
    {
      "action_loss_avg": 0.0013729167403653264,
      "epoch": 4.3058395385742186e-05,
      "step": 45150
    },
    {
      "epoch": 4.3058395385742186e-05,
      "step": 45150,
      "training_step_time": 7.836707353591919
    },
    {
      "epoch": 4.30593490600586e-05,
      "model_forward_time": 1.7435781955718994,
      "step": 45151
    },
    {
      "epoch": 4.30593490600586e-05,
      "step": 45151,
      "training_step_time": 6.905091047286987
    },
    {
      "epoch": 4.3060302734375e-05,
      "model_forward_time": 1.744692087173462,
      "step": 45152
    },
    {
      "epoch": 4.3060302734375e-05,
      "step": 45152,
      "training_step_time": 6.899024963378906
    },
    {
      "epoch": 4.3061256408691405e-05,
      "model_forward_time": 1.7351281642913818,
      "step": 45153
    },
    {
      "epoch": 4.3061256408691405e-05,
      "step": 45153,
      "training_step_time": 6.925715208053589
    },
    {
      "epoch": 4.3062210083007816e-05,
      "model_forward_time": 1.717559576034546,
      "step": 45154
    },
    {
      "epoch": 4.3062210083007816e-05,
      "step": 45154,
      "training_step_time": 6.8649537563323975
    },
    {
      "epoch": 4.306316375732422e-05,
      "model_forward_time": 1.740126371383667,
      "step": 45155
    },
    {
      "epoch": 4.306316375732422e-05,
      "step": 45155,
      "training_step_time": 6.92973256111145
    },
    {
      "epoch": 4.3064117431640624e-05,
      "model_forward_time": 1.7470734119415283,
      "step": 45156
    },
    {
      "epoch": 4.3064117431640624e-05,
      "step": 45156,
      "training_step_time": 8.394692659378052
    },
    {
      "epoch": 4.306507110595703e-05,
      "model_forward_time": 1.4228739738464355,
      "step": 45157
    },
    {
      "epoch": 4.306507110595703e-05,
      "step": 45157,
      "training_step_time": 9.2175133228302
    },
    {
      "epoch": 4.306602478027344e-05,
      "model_forward_time": 2.277785539627075,
      "step": 45158
    },
    {
      "epoch": 4.306602478027344e-05,
      "step": 45158,
      "training_step_time": 10.293969869613647
    },
    {
      "epoch": 4.306697845458984e-05,
      "model_forward_time": 1.7466306686401367,
      "step": 45159
    },
    {
      "epoch": 4.306697845458984e-05,
      "step": 45159,
      "training_step_time": 8.850197315216064
    },
    {
      "epoch": 4.306793212890625e-05,
      "grad_norm": 0.17551983892917633,
      "learning_rate": 6.201924887993712e-06,
      "loss": 0.0578,
      "step": 45160
    },
    {
      "epoch": 4.306793212890625e-05,
      "model_forward_time": 1.7374792098999023,
      "step": 45160
    },
    {
      "dynamics_loss_avg": 0.05518244672566652,
      "epoch": 4.306793212890625e-05,
      "step": 45160
    },
    {
      "action_loss_avg": 0.0015303398831747472,
      "epoch": 4.306793212890625e-05,
      "step": 45160
    },
    {
      "epoch": 4.306793212890625e-05,
      "step": 45160,
      "training_step_time": 6.861766576766968
    },
    {
      "epoch": 4.306888580322266e-05,
      "model_forward_time": 1.7443437576293945,
      "step": 45161
    },
    {
      "epoch": 4.306888580322266e-05,
      "step": 45161,
      "training_step_time": 6.879408359527588
    },
    {
      "epoch": 4.306983947753906e-05,
      "model_forward_time": 1.7475090026855469,
      "step": 45162
    },
    {
      "epoch": 4.306983947753906e-05,
      "step": 45162,
      "training_step_time": 6.911496639251709
    },
    {
      "epoch": 4.3070793151855466e-05,
      "model_forward_time": 1.7403364181518555,
      "step": 45163
    },
    {
      "epoch": 4.3070793151855466e-05,
      "step": 45163,
      "training_step_time": 6.8468427658081055
    },
    {
      "epoch": 4.3071746826171876e-05,
      "model_forward_time": 1.7493643760681152,
      "step": 45164
    },
    {
      "epoch": 4.3071746826171876e-05,
      "step": 45164,
      "training_step_time": 6.9060378074646
    },
    {
      "epoch": 4.307270050048828e-05,
      "model_forward_time": 1.7523748874664307,
      "step": 45165
    },
    {
      "epoch": 4.307270050048828e-05,
      "step": 45165,
      "training_step_time": 8.32271432876587
    },
    {
      "epoch": 4.3073654174804684e-05,
      "model_forward_time": 1.7837672233581543,
      "step": 45166
    },
    {
      "epoch": 4.3073654174804684e-05,
      "step": 45166,
      "training_step_time": 8.583521842956543
    },
    {
      "epoch": 4.3074607849121095e-05,
      "model_forward_time": 2.2795798778533936,
      "step": 45167
    },
    {
      "epoch": 4.3074607849121095e-05,
      "step": 45167,
      "training_step_time": 7.917331695556641
    },
    {
      "epoch": 4.30755615234375e-05,
      "model_forward_time": 1.7534222602844238,
      "step": 45168
    },
    {
      "epoch": 4.30755615234375e-05,
      "step": 45168,
      "training_step_time": 7.8350043296813965
    },
    {
      "epoch": 4.30765151977539e-05,
      "model_forward_time": 1.747270107269287,
      "step": 45169
    },
    {
      "epoch": 4.30765151977539e-05,
      "step": 45169,
      "training_step_time": 7.804961681365967
    },
    {
      "epoch": 4.3077468872070314e-05,
      "grad_norm": 0.13054519891738892,
      "learning_rate": 6.200319836137725e-06,
      "loss": 0.0629,
      "step": 45170
    },
    {
      "epoch": 4.3077468872070314e-05,
      "model_forward_time": 1.7334942817687988,
      "step": 45170
    },
    {
      "dynamics_loss_avg": 0.0641755310818553,
      "epoch": 4.3077468872070314e-05,
      "step": 45170
    },
    {
      "action_loss_avg": 0.0015332900715293364,
      "epoch": 4.3077468872070314e-05,
      "step": 45170
    },
    {
      "epoch": 4.3077468872070314e-05,
      "step": 45170,
      "training_step_time": 6.832572937011719
    },
    {
      "epoch": 4.307842254638672e-05,
      "model_forward_time": 1.7471866607666016,
      "step": 45171
    },
    {
      "epoch": 4.307842254638672e-05,
      "step": 45171,
      "training_step_time": 6.849767684936523
    },
    {
      "epoch": 4.307937622070312e-05,
      "model_forward_time": 1.7432563304901123,
      "step": 45172
    },
    {
      "epoch": 4.307937622070312e-05,
      "step": 45172,
      "training_step_time": 6.865522623062134
    },
    {
      "epoch": 4.308032989501953e-05,
      "model_forward_time": 1.7459423542022705,
      "step": 45173
    },
    {
      "epoch": 4.308032989501953e-05,
      "step": 45173,
      "training_step_time": 6.838008403778076
    },
    {
      "epoch": 4.308128356933594e-05,
      "model_forward_time": 1.7470428943634033,
      "step": 45174
    },
    {
      "epoch": 4.308128356933594e-05,
      "step": 45174,
      "training_step_time": 8.201008081436157
    },
    {
      "epoch": 4.308223724365234e-05,
      "model_forward_time": 1.7480273246765137,
      "step": 45175
    },
    {
      "epoch": 4.308223724365234e-05,
      "step": 45175,
      "training_step_time": 10.131041765213013
    },
    {
      "epoch": 4.308319091796875e-05,
      "model_forward_time": 2.2818567752838135,
      "step": 45176
    },
    {
      "epoch": 4.308319091796875e-05,
      "step": 45176,
      "training_step_time": 10.458380460739136
    },
    {
      "epoch": 4.3084144592285156e-05,
      "model_forward_time": 1.7384483814239502,
      "step": 45177
    },
    {
      "epoch": 4.3084144592285156e-05,
      "step": 45177,
      "training_step_time": 8.865256786346436
    },
    {
      "epoch": 4.308509826660156e-05,
      "model_forward_time": 1.7564566135406494,
      "step": 45178
    },
    {
      "epoch": 4.308509826660156e-05,
      "step": 45178,
      "training_step_time": 7.827005624771118
    },
    {
      "epoch": 4.308605194091797e-05,
      "model_forward_time": 1.7528553009033203,
      "step": 45179
    },
    {
      "epoch": 4.308605194091797e-05,
      "step": 45179,
      "training_step_time": 6.878400087356567
    },
    {
      "epoch": 4.3087005615234375e-05,
      "grad_norm": 0.12136305123567581,
      "learning_rate": 6.198714653016564e-06,
      "loss": 0.0598,
      "step": 45180
    },
    {
      "epoch": 4.3087005615234375e-05,
      "model_forward_time": 1.7393600940704346,
      "step": 45180
    },
    {
      "dynamics_loss_avg": 0.04709135517477989,
      "epoch": 4.3087005615234375e-05,
      "step": 45180
    },
    {
      "action_loss_avg": 0.0011839278100524097,
      "epoch": 4.3087005615234375e-05,
      "step": 45180
    },
    {
      "epoch": 4.3087005615234375e-05,
      "step": 45180,
      "training_step_time": 6.903248310089111
    },
    {
      "epoch": 4.308795928955078e-05,
      "model_forward_time": 1.74961256980896,
      "step": 45181
    },
    {
      "epoch": 4.308795928955078e-05,
      "step": 45181,
      "training_step_time": 6.903308868408203
    },
    {
      "epoch": 4.308891296386719e-05,
      "model_forward_time": 1.751509428024292,
      "step": 45182
    },
    {
      "epoch": 4.308891296386719e-05,
      "step": 45182,
      "training_step_time": 6.912774324417114
    },
    {
      "epoch": 4.3089866638183594e-05,
      "model_forward_time": 1.7568309307098389,
      "step": 45183
    },
    {
      "epoch": 4.3089866638183594e-05,
      "step": 45183,
      "training_step_time": 7.761082172393799
    },
    {
      "epoch": 4.30908203125e-05,
      "model_forward_time": 1.744537353515625,
      "step": 45184
    },
    {
      "epoch": 4.30908203125e-05,
      "step": 45184,
      "training_step_time": 7.683108329772949
    },
    {
      "epoch": 4.309177398681641e-05,
      "model_forward_time": 2.2883002758026123,
      "step": 45185
    },
    {
      "epoch": 4.309177398681641e-05,
      "step": 45185,
      "training_step_time": 9.21536898612976
    },
    {
      "epoch": 4.309272766113281e-05,
      "model_forward_time": 1.3787031173706055,
      "step": 45186
    },
    {
      "epoch": 4.309272766113281e-05,
      "step": 45186,
      "training_step_time": 9.997476816177368
    },
    {
      "epoch": 4.3093681335449216e-05,
      "model_forward_time": 1.7304835319519043,
      "step": 45187
    },
    {
      "epoch": 4.3093681335449216e-05,
      "step": 45187,
      "training_step_time": 7.8196892738342285
    },
    {
      "epoch": 4.309463500976563e-05,
      "model_forward_time": 1.7389118671417236,
      "step": 45188
    },
    {
      "epoch": 4.309463500976563e-05,
      "step": 45188,
      "training_step_time": 6.893811225891113
    },
    {
      "epoch": 4.309558868408203e-05,
      "model_forward_time": 1.7340118885040283,
      "step": 45189
    },
    {
      "epoch": 4.309558868408203e-05,
      "step": 45189,
      "training_step_time": 6.92984414100647
    },
    {
      "epoch": 4.3096542358398435e-05,
      "grad_norm": 0.18197672069072723,
      "learning_rate": 6.197109338805773e-06,
      "loss": 0.059,
      "step": 45190
    },
    {
      "epoch": 4.3096542358398435e-05,
      "model_forward_time": 1.7237138748168945,
      "step": 45190
    },
    {
      "dynamics_loss_avg": 0.06145197935402393,
      "epoch": 4.3096542358398435e-05,
      "step": 45190
    },
    {
      "action_loss_avg": 0.002649704576469958,
      "epoch": 4.3096542358398435e-05,
      "step": 45190
    },
    {
      "epoch": 4.3096542358398435e-05,
      "step": 45190,
      "training_step_time": 6.833706617355347
    },
    {
      "epoch": 4.3097496032714846e-05,
      "model_forward_time": 1.755655288696289,
      "step": 45191
    },
    {
      "epoch": 4.3097496032714846e-05,
      "step": 45191,
      "training_step_time": 6.850243091583252
    },
    {
      "epoch": 4.309844970703125e-05,
      "model_forward_time": 1.7300546169281006,
      "step": 45192
    },
    {
      "epoch": 4.309844970703125e-05,
      "step": 45192,
      "training_step_time": 7.666124582290649
    },
    {
      "epoch": 4.3099403381347654e-05,
      "model_forward_time": 1.7525749206542969,
      "step": 45193
    },
    {
      "epoch": 4.3099403381347654e-05,
      "step": 45193,
      "training_step_time": 7.723673582077026
    },
    {
      "epoch": 4.3100357055664065e-05,
      "model_forward_time": 2.3466453552246094,
      "step": 45194
    },
    {
      "epoch": 4.3100357055664065e-05,
      "step": 45194,
      "training_step_time": 12.11700439453125
    },
    {
      "epoch": 4.310131072998047e-05,
      "model_forward_time": 1.76179838180542,
      "step": 45195
    },
    {
      "epoch": 4.310131072998047e-05,
      "step": 45195,
      "training_step_time": 9.319194555282593
    },
    {
      "epoch": 4.310226440429687e-05,
      "model_forward_time": 1.7609446048736572,
      "step": 45196
    },
    {
      "epoch": 4.310226440429687e-05,
      "step": 45196,
      "training_step_time": 7.783824443817139
    },
    {
      "epoch": 4.3103218078613284e-05,
      "model_forward_time": 1.7582337856292725,
      "step": 45197
    },
    {
      "epoch": 4.3103218078613284e-05,
      "step": 45197,
      "training_step_time": 6.845182418823242
    },
    {
      "epoch": 4.310417175292969e-05,
      "model_forward_time": 1.7570350170135498,
      "step": 45198
    },
    {
      "epoch": 4.310417175292969e-05,
      "step": 45198,
      "training_step_time": 6.8058021068573
    },
    {
      "epoch": 4.310512542724609e-05,
      "model_forward_time": 1.759187936782837,
      "step": 45199
    },
    {
      "epoch": 4.310512542724609e-05,
      "step": 45199,
      "training_step_time": 6.767373561859131
    },
    {
      "epoch": 4.31060791015625e-05,
      "grad_norm": 0.175043523311615,
      "learning_rate": 6.195503893680903e-06,
      "loss": 0.0606,
      "step": 45200
    },
    {
      "epoch": 4.31060791015625e-05,
      "model_forward_time": 1.7506742477416992,
      "step": 45200
    },
    {
      "dynamics_loss_avg": 0.05838078577071428,
      "epoch": 4.31060791015625e-05,
      "step": 45200
    },
    {
      "action_loss_avg": 0.0019195365079212935,
      "epoch": 4.31060791015625e-05,
      "step": 45200
    },
    {
      "epoch": 4.31060791015625e-05,
      "step": 45200,
      "training_step_time": 6.845621585845947
    },
    {
      "epoch": 4.310703277587891e-05,
      "model_forward_time": 1.7220733165740967,
      "step": 45201
    },
    {
      "epoch": 4.310703277587891e-05,
      "step": 45201,
      "training_step_time": 7.6816136837005615
    },
    {
      "epoch": 4.310798645019531e-05,
      "model_forward_time": 1.7458395957946777,
      "step": 45202
    },
    {
      "epoch": 4.310798645019531e-05,
      "step": 45202,
      "training_step_time": 7.546161890029907
    },
    {
      "epoch": 4.310894012451172e-05,
      "model_forward_time": 2.3116908073425293,
      "step": 45203
    },
    {
      "epoch": 4.310894012451172e-05,
      "step": 45203,
      "training_step_time": 10.221664428710938
    },
    {
      "epoch": 4.3109893798828126e-05,
      "model_forward_time": 1.730095624923706,
      "step": 45204
    },
    {
      "epoch": 4.3109893798828126e-05,
      "step": 45204,
      "training_step_time": 7.8125176429748535
    },
    {
      "epoch": 4.311084747314453e-05,
      "model_forward_time": 1.3665001392364502,
      "step": 45205
    },
    {
      "epoch": 4.311084747314453e-05,
      "step": 45205,
      "training_step_time": 7.821283340454102
    },
    {
      "epoch": 4.311180114746094e-05,
      "model_forward_time": 1.7492351531982422,
      "step": 45206
    },
    {
      "epoch": 4.311180114746094e-05,
      "step": 45206,
      "training_step_time": 7.0646185874938965
    },
    {
      "epoch": 4.3112754821777344e-05,
      "model_forward_time": 1.734964370727539,
      "step": 45207
    },
    {
      "epoch": 4.3112754821777344e-05,
      "step": 45207,
      "training_step_time": 6.781112194061279
    },
    {
      "epoch": 4.311370849609375e-05,
      "model_forward_time": 1.7410361766815186,
      "step": 45208
    },
    {
      "epoch": 4.311370849609375e-05,
      "step": 45208,
      "training_step_time": 6.839112997055054
    },
    {
      "epoch": 4.311466217041016e-05,
      "model_forward_time": 1.7315118312835693,
      "step": 45209
    },
    {
      "epoch": 4.311466217041016e-05,
      "step": 45209,
      "training_step_time": 6.810642957687378
    },
    {
      "epoch": 4.311561584472656e-05,
      "grad_norm": 0.176021009683609,
      "learning_rate": 6.193898317817525e-06,
      "loss": 0.0609,
      "step": 45210
    },
    {
      "epoch": 4.311561584472656e-05,
      "model_forward_time": 1.7344887256622314,
      "step": 45210
    },
    {
      "dynamics_loss_avg": 0.0677651608362794,
      "epoch": 4.311561584472656e-05,
      "step": 45210
    },
    {
      "action_loss_avg": 0.0020122608402743936,
      "epoch": 4.311561584472656e-05,
      "step": 45210
    },
    {
      "epoch": 4.311561584472656e-05,
      "step": 45210,
      "training_step_time": 7.569594860076904
    },
    {
      "epoch": 4.311656951904297e-05,
      "model_forward_time": 1.7412543296813965,
      "step": 45211
    },
    {
      "epoch": 4.311656951904297e-05,
      "step": 45211,
      "training_step_time": 7.5247111320495605
    },
    {
      "epoch": 4.311752319335938e-05,
      "model_forward_time": 2.3094449043273926,
      "step": 45212
    },
    {
      "epoch": 4.311752319335938e-05,
      "step": 45212,
      "training_step_time": 8.723314046859741
    },
    {
      "epoch": 4.311847686767578e-05,
      "model_forward_time": 1.724869966506958,
      "step": 45213
    },
    {
      "epoch": 4.311847686767578e-05,
      "step": 45213,
      "training_step_time": 9.850053787231445
    },
    {
      "epoch": 4.3119430541992186e-05,
      "model_forward_time": 1.739539384841919,
      "step": 45214
    },
    {
      "epoch": 4.3119430541992186e-05,
      "step": 45214,
      "training_step_time": 8.85596752166748
    },
    {
      "epoch": 4.31203842163086e-05,
      "model_forward_time": 1.7317614555358887,
      "step": 45215
    },
    {
      "epoch": 4.31203842163086e-05,
      "step": 45215,
      "training_step_time": 7.48679256439209
    },
    {
      "epoch": 4.3121337890625e-05,
      "model_forward_time": 1.7308595180511475,
      "step": 45216
    },
    {
      "epoch": 4.3121337890625e-05,
      "step": 45216,
      "training_step_time": 6.831749200820923
    },
    {
      "epoch": 4.3122291564941405e-05,
      "model_forward_time": 1.7458865642547607,
      "step": 45217
    },
    {
      "epoch": 4.3122291564941405e-05,
      "step": 45217,
      "training_step_time": 6.819203615188599
    },
    {
      "epoch": 4.3123245239257816e-05,
      "model_forward_time": 1.736600399017334,
      "step": 45218
    },
    {
      "epoch": 4.3123245239257816e-05,
      "step": 45218,
      "training_step_time": 6.825217962265015
    },
    {
      "epoch": 4.312419891357422e-05,
      "model_forward_time": 1.7383081912994385,
      "step": 45219
    },
    {
      "epoch": 4.312419891357422e-05,
      "step": 45219,
      "training_step_time": 7.708495378494263
    },
    {
      "epoch": 4.3125152587890624e-05,
      "grad_norm": 0.13614726066589355,
      "learning_rate": 6.192292611391222e-06,
      "loss": 0.0571,
      "step": 45220
    },
    {
      "epoch": 4.3125152587890624e-05,
      "model_forward_time": 1.733227252960205,
      "step": 45220
    },
    {
      "dynamics_loss_avg": 0.05690657552331686,
      "epoch": 4.3125152587890624e-05,
      "step": 45220
    },
    {
      "action_loss_avg": 0.0018388765631243587,
      "epoch": 4.3125152587890624e-05,
      "step": 45220
    },
    {
      "epoch": 4.3125152587890624e-05,
      "step": 45220,
      "training_step_time": 7.549249649047852
    },
    {
      "epoch": 4.312610626220703e-05,
      "model_forward_time": 1.7747714519500732,
      "step": 45221
    },
    {
      "epoch": 4.312610626220703e-05,
      "step": 45221,
      "training_step_time": 7.718797445297241
    },
    {
      "epoch": 4.312705993652344e-05,
      "model_forward_time": 2.294593095779419,
      "step": 45222
    },
    {
      "epoch": 4.312705993652344e-05,
      "step": 45222,
      "training_step_time": 7.744121551513672
    },
    {
      "epoch": 4.312801361083984e-05,
      "model_forward_time": 1.749600887298584,
      "step": 45223
    },
    {
      "epoch": 4.312801361083984e-05,
      "step": 45223,
      "training_step_time": 7.801133632659912
    },
    {
      "epoch": 4.312896728515625e-05,
      "model_forward_time": 1.7482411861419678,
      "step": 45224
    },
    {
      "epoch": 4.312896728515625e-05,
      "step": 45224,
      "training_step_time": 7.755474090576172
    },
    {
      "epoch": 4.312992095947266e-05,
      "model_forward_time": 1.7141001224517822,
      "step": 45225
    },
    {
      "epoch": 4.312992095947266e-05,
      "step": 45225,
      "training_step_time": 6.838572978973389
    },
    {
      "epoch": 4.313087463378906e-05,
      "model_forward_time": 1.7349839210510254,
      "step": 45226
    },
    {
      "epoch": 4.313087463378906e-05,
      "step": 45226,
      "training_step_time": 6.837310791015625
    },
    {
      "epoch": 4.3131828308105466e-05,
      "model_forward_time": 1.737222671508789,
      "step": 45227
    },
    {
      "epoch": 4.3131828308105466e-05,
      "step": 45227,
      "training_step_time": 6.812050104141235
    },
    {
      "epoch": 4.3132781982421876e-05,
      "model_forward_time": 1.7434449195861816,
      "step": 45228
    },
    {
      "epoch": 4.3132781982421876e-05,
      "step": 45228,
      "training_step_time": 7.636677265167236
    },
    {
      "epoch": 4.313373565673828e-05,
      "model_forward_time": 1.7437195777893066,
      "step": 45229
    },
    {
      "epoch": 4.313373565673828e-05,
      "step": 45229,
      "training_step_time": 7.4479289054870605
    },
    {
      "epoch": 4.3134689331054685e-05,
      "grad_norm": 0.1375502645969391,
      "learning_rate": 6.190686774577591e-06,
      "loss": 0.0617,
      "step": 45230
    },
    {
      "epoch": 4.3134689331054685e-05,
      "model_forward_time": 1.7412762641906738,
      "step": 45230
    },
    {
      "dynamics_loss_avg": 0.05188619121909142,
      "epoch": 4.3134689331054685e-05,
      "step": 45230
    },
    {
      "action_loss_avg": 0.001916380797047168,
      "epoch": 4.3134689331054685e-05,
      "step": 45230
    },
    {
      "epoch": 4.3134689331054685e-05,
      "step": 45230,
      "training_step_time": 8.507997512817383
    },
    {
      "epoch": 4.3135643005371095e-05,
      "model_forward_time": 2.2835099697113037,
      "step": 45231
    },
    {
      "epoch": 4.3135643005371095e-05,
      "step": 45231,
      "training_step_time": 10.167367935180664
    },
    {
      "epoch": 4.31365966796875e-05,
      "model_forward_time": 1.734311580657959,
      "step": 45232
    },
    {
      "epoch": 4.31365966796875e-05,
      "step": 45232,
      "training_step_time": 9.639941930770874
    },
    {
      "epoch": 4.31375503540039e-05,
      "model_forward_time": 1.7277066707611084,
      "step": 45233
    },
    {
      "epoch": 4.31375503540039e-05,
      "step": 45233,
      "training_step_time": 7.7945592403411865
    },
    {
      "epoch": 4.3138504028320314e-05,
      "model_forward_time": 1.7355666160583496,
      "step": 45234
    },
    {
      "epoch": 4.3138504028320314e-05,
      "step": 45234,
      "training_step_time": 7.035226345062256
    },
    {
      "epoch": 4.313945770263672e-05,
      "model_forward_time": 1.741180658340454,
      "step": 45235
    },
    {
      "epoch": 4.313945770263672e-05,
      "step": 45235,
      "training_step_time": 6.761147975921631
    },
    {
      "epoch": 4.314041137695312e-05,
      "model_forward_time": 1.3783819675445557,
      "step": 45236
    },
    {
      "epoch": 4.314041137695312e-05,
      "step": 45236,
      "training_step_time": 6.757324934005737
    },
    {
      "epoch": 4.314136505126953e-05,
      "model_forward_time": 1.7572548389434814,
      "step": 45237
    },
    {
      "epoch": 4.314136505126953e-05,
      "step": 45237,
      "training_step_time": 7.637335300445557
    },
    {
      "epoch": 4.314231872558594e-05,
      "model_forward_time": 1.7579913139343262,
      "step": 45238
    },
    {
      "epoch": 4.314231872558594e-05,
      "step": 45238,
      "training_step_time": 7.423960447311401
    },
    {
      "epoch": 4.314327239990234e-05,
      "model_forward_time": 1.753448724746704,
      "step": 45239
    },
    {
      "epoch": 4.314327239990234e-05,
      "step": 45239,
      "training_step_time": 8.676016330718994
    },
    {
      "epoch": 4.314422607421875e-05,
      "grad_norm": 0.1542956829071045,
      "learning_rate": 6.189080807552244e-06,
      "loss": 0.0568,
      "step": 45240
    },
    {
      "epoch": 4.314422607421875e-05,
      "model_forward_time": 2.3284871578216553,
      "step": 45240
    },
    {
      "dynamics_loss_avg": 0.0440680424682796,
      "epoch": 4.314422607421875e-05,
      "step": 45240
    },
    {
      "action_loss_avg": 0.0012374538870062678,
      "epoch": 4.314422607421875e-05,
      "step": 45240
    },
    {
      "epoch": 4.314422607421875e-05,
      "step": 45240,
      "training_step_time": 10.179733514785767
    },
    {
      "epoch": 4.3145179748535156e-05,
      "model_forward_time": 1.7555880546569824,
      "step": 45241
    },
    {
      "epoch": 4.3145179748535156e-05,
      "step": 45241,
      "training_step_time": 9.193182706832886
    },
    {
      "epoch": 4.314613342285156e-05,
      "model_forward_time": 1.757251501083374,
      "step": 45242
    },
    {
      "epoch": 4.314613342285156e-05,
      "step": 45242,
      "training_step_time": 7.691375255584717
    },
    {
      "epoch": 4.314708709716797e-05,
      "model_forward_time": 1.7583715915679932,
      "step": 45243
    },
    {
      "epoch": 4.314708709716797e-05,
      "step": 45243,
      "training_step_time": 7.075625658035278
    },
    {
      "epoch": 4.3148040771484375e-05,
      "model_forward_time": 1.761160135269165,
      "step": 45244
    },
    {
      "epoch": 4.3148040771484375e-05,
      "step": 45244,
      "training_step_time": 6.852444648742676
    },
    {
      "epoch": 4.314899444580078e-05,
      "model_forward_time": 1.7455127239227295,
      "step": 45245
    },
    {
      "epoch": 4.314899444580078e-05,
      "step": 45245,
      "training_step_time": 6.770578622817993
    },
    {
      "epoch": 4.314994812011719e-05,
      "model_forward_time": 1.7491214275360107,
      "step": 45246
    },
    {
      "epoch": 4.314994812011719e-05,
      "step": 45246,
      "training_step_time": 7.648671627044678
    },
    {
      "epoch": 4.3150901794433594e-05,
      "model_forward_time": 1.7552387714385986,
      "step": 45247
    },
    {
      "epoch": 4.3150901794433594e-05,
      "step": 45247,
      "training_step_time": 7.419763803482056
    },
    {
      "epoch": 4.315185546875e-05,
      "model_forward_time": 1.7564470767974854,
      "step": 45248
    },
    {
      "epoch": 4.315185546875e-05,
      "step": 45248,
      "training_step_time": 7.526756525039673
    },
    {
      "epoch": 4.315280914306641e-05,
      "model_forward_time": 2.3370413780212402,
      "step": 45249
    },
    {
      "epoch": 4.315280914306641e-05,
      "step": 45249,
      "training_step_time": 8.24489450454712
    },
    {
      "epoch": 4.315376281738281e-05,
      "grad_norm": 0.14169254899024963,
      "learning_rate": 6.1874747104908094e-06,
      "loss": 0.0633,
      "step": 45250
    },
    {
      "epoch": 4.315376281738281e-05,
      "model_forward_time": 1.7399928569793701,
      "step": 45250
    },
    {
      "dynamics_loss_avg": 0.04290193570777774,
      "epoch": 4.315376281738281e-05,
      "step": 45250
    },
    {
      "action_loss_avg": 0.001940345432376489,
      "epoch": 4.315376281738281e-05,
      "step": 45250
    },
    {
      "epoch": 4.315376281738281e-05,
      "step": 45250,
      "training_step_time": 8.850510597229004
    },
    {
      "epoch": 4.3154716491699217e-05,
      "model_forward_time": 1.3747143745422363,
      "step": 45251
    },
    {
      "epoch": 4.3154716491699217e-05,
      "step": 45251,
      "training_step_time": 8.584822654724121
    },
    {
      "epoch": 4.315567016601563e-05,
      "model_forward_time": 1.757899284362793,
      "step": 45252
    },
    {
      "epoch": 4.315567016601563e-05,
      "step": 45252,
      "training_step_time": 7.061422109603882
    },
    {
      "epoch": 4.315662384033203e-05,
      "model_forward_time": 1.3833014965057373,
      "step": 45253
    },
    {
      "epoch": 4.315662384033203e-05,
      "step": 45253,
      "training_step_time": 6.7772955894470215
    },
    {
      "epoch": 4.3157577514648435e-05,
      "model_forward_time": 1.7555994987487793,
      "step": 45254
    },
    {
      "epoch": 4.3157577514648435e-05,
      "step": 45254,
      "training_step_time": 6.784443616867065
    },
    {
      "epoch": 4.3158531188964846e-05,
      "model_forward_time": 1.74642014503479,
      "step": 45255
    },
    {
      "epoch": 4.3158531188964846e-05,
      "step": 45255,
      "training_step_time": 7.741709470748901
    },
    {
      "epoch": 4.315948486328125e-05,
      "model_forward_time": 1.7539598941802979,
      "step": 45256
    },
    {
      "epoch": 4.315948486328125e-05,
      "step": 45256,
      "training_step_time": 7.083348035812378
    },
    {
      "epoch": 4.3160438537597654e-05,
      "model_forward_time": 1.7603511810302734,
      "step": 45257
    },
    {
      "epoch": 4.3160438537597654e-05,
      "step": 45257,
      "training_step_time": 8.076079368591309
    },
    {
      "epoch": 4.3161392211914065e-05,
      "model_forward_time": 1.7984042167663574,
      "step": 45258
    },
    {
      "epoch": 4.3161392211914065e-05,
      "step": 45258,
      "training_step_time": 8.307549953460693
    },
    {
      "epoch": 4.316234588623047e-05,
      "model_forward_time": 2.318157434463501,
      "step": 45259
    },
    {
      "epoch": 4.316234588623047e-05,
      "step": 45259,
      "training_step_time": 8.887485265731812
    },
    {
      "epoch": 4.316329956054687e-05,
      "grad_norm": 0.11118430644273758,
      "learning_rate": 6.185868483568926e-06,
      "loss": 0.0592,
      "step": 45260
    },
    {
      "epoch": 4.316329956054687e-05,
      "model_forward_time": 1.749013900756836,
      "step": 45260
    },
    {
      "dynamics_loss_avg": 0.05799644570797682,
      "epoch": 4.316329956054687e-05,
      "step": 45260
    },
    {
      "action_loss_avg": 0.001838551729451865,
      "epoch": 4.316329956054687e-05,
      "step": 45260
    },
    {
      "epoch": 4.316329956054687e-05,
      "step": 45260,
      "training_step_time": 8.789953708648682
    },
    {
      "epoch": 4.3164253234863284e-05,
      "model_forward_time": 1.7468478679656982,
      "step": 45261
    },
    {
      "epoch": 4.3164253234863284e-05,
      "step": 45261,
      "training_step_time": 8.828544616699219
    },
    {
      "epoch": 4.316520690917969e-05,
      "model_forward_time": 1.7510766983032227,
      "step": 45262
    },
    {
      "epoch": 4.316520690917969e-05,
      "step": 45262,
      "training_step_time": 6.845820188522339
    },
    {
      "epoch": 4.316616058349609e-05,
      "model_forward_time": 1.740229606628418,
      "step": 45263
    },
    {
      "epoch": 4.316616058349609e-05,
      "step": 45263,
      "training_step_time": 6.82179069519043
    },
    {
      "epoch": 4.31671142578125e-05,
      "model_forward_time": 1.735889196395874,
      "step": 45264
    },
    {
      "epoch": 4.31671142578125e-05,
      "step": 45264,
      "training_step_time": 7.707308053970337
    },
    {
      "epoch": 4.316806793212891e-05,
      "model_forward_time": 1.7532901763916016,
      "step": 45265
    },
    {
      "epoch": 4.316806793212891e-05,
      "step": 45265,
      "training_step_time": 7.138679265975952
    },
    {
      "epoch": 4.316902160644531e-05,
      "model_forward_time": 1.7560534477233887,
      "step": 45266
    },
    {
      "epoch": 4.316902160644531e-05,
      "step": 45266,
      "training_step_time": 7.49398136138916
    },
    {
      "epoch": 4.316997528076172e-05,
      "model_forward_time": 1.8058626651763916,
      "step": 45267
    },
    {
      "epoch": 4.316997528076172e-05,
      "step": 45267,
      "training_step_time": 10.246874570846558
    },
    {
      "epoch": 4.3170928955078126e-05,
      "model_forward_time": 2.315511465072632,
      "step": 45268
    },
    {
      "epoch": 4.3170928955078126e-05,
      "step": 45268,
      "training_step_time": 7.742135286331177
    },
    {
      "epoch": 4.317188262939453e-05,
      "model_forward_time": 1.7534401416778564,
      "step": 45269
    },
    {
      "epoch": 4.317188262939453e-05,
      "step": 45269,
      "training_step_time": 7.7306458950042725
    },
    {
      "epoch": 4.317283630371094e-05,
      "grad_norm": 0.10087577253580093,
      "learning_rate": 6.184262126962246e-06,
      "loss": 0.0599,
      "step": 45270
    },
    {
      "epoch": 4.317283630371094e-05,
      "model_forward_time": 1.746854543685913,
      "step": 45270
    },
    {
      "dynamics_loss_avg": 0.05649354476481676,
      "epoch": 4.317283630371094e-05,
      "step": 45270
    },
    {
      "action_loss_avg": 0.0020609458384569733,
      "epoch": 4.317283630371094e-05,
      "step": 45270
    },
    {
      "epoch": 4.317283630371094e-05,
      "step": 45270,
      "training_step_time": 7.641350746154785
    },
    {
      "epoch": 4.3173789978027345e-05,
      "model_forward_time": 1.7565274238586426,
      "step": 45271
    },
    {
      "epoch": 4.3173789978027345e-05,
      "step": 45271,
      "training_step_time": 7.041679382324219
    },
    {
      "epoch": 4.317474365234375e-05,
      "model_forward_time": 1.7571327686309814,
      "step": 45272
    },
    {
      "epoch": 4.317474365234375e-05,
      "step": 45272,
      "training_step_time": 6.7819976806640625
    },
    {
      "epoch": 4.317569732666016e-05,
      "model_forward_time": 1.7482593059539795,
      "step": 45273
    },
    {
      "epoch": 4.317569732666016e-05,
      "step": 45273,
      "training_step_time": 7.652383089065552
    },
    {
      "epoch": 4.317665100097656e-05,
      "model_forward_time": 1.7379469871520996,
      "step": 45274
    },
    {
      "epoch": 4.317665100097656e-05,
      "step": 45274,
      "training_step_time": 7.1056413650512695
    },
    {
      "epoch": 4.317760467529297e-05,
      "model_forward_time": 1.7528464794158936,
      "step": 45275
    },
    {
      "epoch": 4.317760467529297e-05,
      "step": 45275,
      "training_step_time": 7.439521312713623
    },
    {
      "epoch": 4.317855834960938e-05,
      "model_forward_time": 1.8065690994262695,
      "step": 45276
    },
    {
      "epoch": 4.317855834960938e-05,
      "step": 45276,
      "training_step_time": 8.192338228225708
    },
    {
      "epoch": 4.317951202392578e-05,
      "model_forward_time": 2.3489792346954346,
      "step": 45277
    },
    {
      "epoch": 4.317951202392578e-05,
      "step": 45277,
      "training_step_time": 8.654324293136597
    },
    {
      "epoch": 4.3180465698242186e-05,
      "model_forward_time": 1.7408697605133057,
      "step": 45278
    },
    {
      "epoch": 4.3180465698242186e-05,
      "step": 45278,
      "training_step_time": 8.306884050369263
    },
    {
      "epoch": 4.31814193725586e-05,
      "model_forward_time": 1.736220121383667,
      "step": 45279
    },
    {
      "epoch": 4.31814193725586e-05,
      "step": 45279,
      "training_step_time": 7.6766180992126465
    },
    {
      "epoch": 4.3182373046875e-05,
      "grad_norm": 0.10514690726995468,
      "learning_rate": 6.182655640846442e-06,
      "loss": 0.0602,
      "step": 45280
    },
    {
      "epoch": 4.3182373046875e-05,
      "model_forward_time": 1.7511906623840332,
      "step": 45280
    },
    {
      "dynamics_loss_avg": 0.0656390124000609,
      "epoch": 4.3182373046875e-05,
      "step": 45280
    },
    {
      "action_loss_avg": 0.0022644300595857205,
      "epoch": 4.3182373046875e-05,
      "step": 45280
    },
    {
      "epoch": 4.3182373046875e-05,
      "step": 45280,
      "training_step_time": 7.062547206878662
    },
    {
      "epoch": 4.3183326721191405e-05,
      "model_forward_time": 1.3889679908752441,
      "step": 45281
    },
    {
      "epoch": 4.3183326721191405e-05,
      "step": 45281,
      "training_step_time": 6.696145057678223
    },
    {
      "epoch": 4.318428039550781e-05,
      "model_forward_time": 1.7496085166931152,
      "step": 45282
    },
    {
      "epoch": 4.318428039550781e-05,
      "step": 45282,
      "training_step_time": 7.598906517028809
    },
    {
      "epoch": 4.318523406982422e-05,
      "model_forward_time": 1.7507052421569824,
      "step": 45283
    },
    {
      "epoch": 4.318523406982422e-05,
      "step": 45283,
      "training_step_time": 6.711532115936279
    },
    {
      "epoch": 4.3186187744140624e-05,
      "model_forward_time": 1.7501003742218018,
      "step": 45284
    },
    {
      "epoch": 4.3186187744140624e-05,
      "step": 45284,
      "training_step_time": 7.4260358810424805
    },
    {
      "epoch": 4.318714141845703e-05,
      "model_forward_time": 1.8136794567108154,
      "step": 45285
    },
    {
      "epoch": 4.318714141845703e-05,
      "step": 45285,
      "training_step_time": 7.671185255050659
    },
    {
      "epoch": 4.318809509277344e-05,
      "model_forward_time": 2.319478750228882,
      "step": 45286
    },
    {
      "epoch": 4.318809509277344e-05,
      "step": 45286,
      "training_step_time": 9.261701345443726
    },
    {
      "epoch": 4.318904876708984e-05,
      "model_forward_time": 1.7583832740783691,
      "step": 45287
    },
    {
      "epoch": 4.318904876708984e-05,
      "step": 45287,
      "training_step_time": 7.7813520431518555
    },
    {
      "epoch": 4.319000244140625e-05,
      "model_forward_time": 1.7525758743286133,
      "step": 45288
    },
    {
      "epoch": 4.319000244140625e-05,
      "step": 45288,
      "training_step_time": 7.772521734237671
    },
    {
      "epoch": 4.319095611572266e-05,
      "model_forward_time": 1.7550735473632812,
      "step": 45289
    },
    {
      "epoch": 4.319095611572266e-05,
      "step": 45289,
      "training_step_time": 7.050060510635376
    },
    {
      "epoch": 4.319190979003906e-05,
      "grad_norm": 0.14457696676254272,
      "learning_rate": 6.181049025397196e-06,
      "loss": 0.0578,
      "step": 45290
    },
    {
      "epoch": 4.319190979003906e-05,
      "model_forward_time": 1.7476387023925781,
      "step": 45290
    },
    {
      "dynamics_loss_avg": 0.050684197805821896,
      "epoch": 4.319190979003906e-05,
      "step": 45290
    },
    {
      "action_loss_avg": 0.0019117102201562375,
      "epoch": 4.319190979003906e-05,
      "step": 45290
    },
    {
      "epoch": 4.319190979003906e-05,
      "step": 45290,
      "training_step_time": 6.697935581207275
    },
    {
      "epoch": 4.3192863464355466e-05,
      "model_forward_time": 1.7608935832977295,
      "step": 45291
    },
    {
      "epoch": 4.3192863464355466e-05,
      "step": 45291,
      "training_step_time": 7.2219929695129395
    },
    {
      "epoch": 4.3193817138671877e-05,
      "model_forward_time": 1.7479376792907715,
      "step": 45292
    },
    {
      "epoch": 4.3193817138671877e-05,
      "step": 45292,
      "training_step_time": 7.182501792907715
    },
    {
      "epoch": 4.319477081298828e-05,
      "model_forward_time": 1.748976230621338,
      "step": 45293
    },
    {
      "epoch": 4.319477081298828e-05,
      "step": 45293,
      "training_step_time": 7.3525824546813965
    },
    {
      "epoch": 4.3195724487304685e-05,
      "model_forward_time": 1.8031928539276123,
      "step": 45294
    },
    {
      "epoch": 4.3195724487304685e-05,
      "step": 45294,
      "training_step_time": 8.414203882217407
    },
    {
      "epoch": 4.3196678161621095e-05,
      "model_forward_time": 1.7629079818725586,
      "step": 45295
    },
    {
      "epoch": 4.3196678161621095e-05,
      "step": 45295,
      "training_step_time": 8.442245721817017
    },
    {
      "epoch": 4.31976318359375e-05,
      "model_forward_time": 2.2808308601379395,
      "step": 45296
    },
    {
      "epoch": 4.31976318359375e-05,
      "step": 45296,
      "training_step_time": 9.620697259902954
    },
    {
      "epoch": 4.3198585510253903e-05,
      "model_forward_time": 1.390526533126831,
      "step": 45297
    },
    {
      "epoch": 4.3198585510253903e-05,
      "step": 45297,
      "training_step_time": 9.420817136764526
    },
    {
      "epoch": 4.3199539184570314e-05,
      "model_forward_time": 1.762087106704712,
      "step": 45298
    },
    {
      "epoch": 4.3199539184570314e-05,
      "step": 45298,
      "training_step_time": 7.741254806518555
    },
    {
      "epoch": 4.320049285888672e-05,
      "model_forward_time": 1.7511358261108398,
      "step": 45299
    },
    {
      "epoch": 4.320049285888672e-05,
      "step": 45299,
      "training_step_time": 7.005167484283447
    },
    {
      "epoch": 4.320144653320312e-05,
      "grad_norm": 0.13952341675758362,
      "learning_rate": 6.179442280790202e-06,
      "loss": 0.0628,
      "step": 45300
    },
    {
      "epoch": 4.320144653320312e-05,
      "model_forward_time": 1.7517714500427246,
      "step": 45300
    },
    {
      "dynamics_loss_avg": 0.07654469721019268,
      "epoch": 4.320144653320312e-05,
      "step": 45300
    },
    {
      "action_loss_avg": 0.0027291635982692243,
      "epoch": 4.320144653320312e-05,
      "step": 45300
    },
    {
      "epoch": 4.320144653320312e-05,
      "step": 45300,
      "training_step_time": 7.191675901412964
    },
    {
      "epoch": 4.320240020751953e-05,
      "model_forward_time": 1.7586359977722168,
      "step": 45301
    },
    {
      "epoch": 4.320240020751953e-05,
      "step": 45301,
      "training_step_time": 7.685653924942017
    },
    {
      "epoch": 4.320335388183594e-05,
      "model_forward_time": 1.7654151916503906,
      "step": 45302
    },
    {
      "epoch": 4.320335388183594e-05,
      "step": 45302,
      "training_step_time": 7.3347108364105225
    },
    {
      "epoch": 4.320430755615234e-05,
      "model_forward_time": 1.8071463108062744,
      "step": 45303
    },
    {
      "epoch": 4.320430755615234e-05,
      "step": 45303,
      "training_step_time": 7.4856367111206055
    },
    {
      "epoch": 4.320526123046875e-05,
      "model_forward_time": 1.7564115524291992,
      "step": 45304
    },
    {
      "epoch": 4.320526123046875e-05,
      "step": 45304,
      "training_step_time": 8.710970640182495
    },
    {
      "epoch": 4.3206214904785156e-05,
      "model_forward_time": 2.330430746078491,
      "step": 45305
    },
    {
      "epoch": 4.3206214904785156e-05,
      "step": 45305,
      "training_step_time": 8.632988452911377
    },
    {
      "epoch": 4.320716857910156e-05,
      "model_forward_time": 1.7575738430023193,
      "step": 45306
    },
    {
      "epoch": 4.320716857910156e-05,
      "step": 45306,
      "training_step_time": 7.679020643234253
    },
    {
      "epoch": 4.320812225341797e-05,
      "model_forward_time": 1.7599146366119385,
      "step": 45307
    },
    {
      "epoch": 4.320812225341797e-05,
      "step": 45307,
      "training_step_time": 7.709400177001953
    },
    {
      "epoch": 4.3209075927734375e-05,
      "model_forward_time": 1.7530796527862549,
      "step": 45308
    },
    {
      "epoch": 4.3209075927734375e-05,
      "step": 45308,
      "training_step_time": 7.021390676498413
    },
    {
      "epoch": 4.321002960205078e-05,
      "model_forward_time": 1.7424461841583252,
      "step": 45309
    },
    {
      "epoch": 4.321002960205078e-05,
      "step": 45309,
      "training_step_time": 6.685758113861084
    },
    {
      "epoch": 4.321098327636719e-05,
      "grad_norm": 0.15099523961544037,
      "learning_rate": 6.177835407201175e-06,
      "loss": 0.0582,
      "step": 45310
    },
    {
      "epoch": 4.321098327636719e-05,
      "model_forward_time": 1.7540464401245117,
      "step": 45310
    },
    {
      "dynamics_loss_avg": 0.061139749549329284,
      "epoch": 4.321098327636719e-05,
      "step": 45310
    },
    {
      "action_loss_avg": 0.002946035491186194,
      "epoch": 4.321098327636719e-05,
      "step": 45310
    },
    {
      "epoch": 4.321098327636719e-05,
      "step": 45310,
      "training_step_time": 7.297708034515381
    },
    {
      "epoch": 4.3211936950683594e-05,
      "model_forward_time": 1.3883659839630127,
      "step": 45311
    },
    {
      "epoch": 4.3211936950683594e-05,
      "step": 45311,
      "training_step_time": 7.233641147613525
    },
    {
      "epoch": 4.3212890625e-05,
      "model_forward_time": 1.7992753982543945,
      "step": 45312
    },
    {
      "epoch": 4.3212890625e-05,
      "step": 45312,
      "training_step_time": 7.512557506561279
    },
    {
      "epoch": 4.321384429931641e-05,
      "model_forward_time": 1.7358529567718506,
      "step": 45313
    },
    {
      "epoch": 4.321384429931641e-05,
      "step": 45313,
      "training_step_time": 9.015586376190186
    },
    {
      "epoch": 4.321479797363281e-05,
      "model_forward_time": 2.3426918983459473,
      "step": 45314
    },
    {
      "epoch": 4.321479797363281e-05,
      "step": 45314,
      "training_step_time": 9.527101278305054
    },
    {
      "epoch": 4.3215751647949217e-05,
      "model_forward_time": 1.7439286708831787,
      "step": 45315
    },
    {
      "epoch": 4.3215751647949217e-05,
      "step": 45315,
      "training_step_time": 7.7564897537231445
    },
    {
      "epoch": 4.321670532226563e-05,
      "model_forward_time": 1.7495017051696777,
      "step": 45316
    },
    {
      "epoch": 4.321670532226563e-05,
      "step": 45316,
      "training_step_time": 7.678739070892334
    },
    {
      "epoch": 4.321765899658203e-05,
      "model_forward_time": 1.750218391418457,
      "step": 45317
    },
    {
      "epoch": 4.321765899658203e-05,
      "step": 45317,
      "training_step_time": 7.105825185775757
    },
    {
      "epoch": 4.3218612670898435e-05,
      "model_forward_time": 1.7483794689178467,
      "step": 45318
    },
    {
      "epoch": 4.3218612670898435e-05,
      "step": 45318,
      "training_step_time": 6.784644603729248
    },
    {
      "epoch": 4.3219566345214846e-05,
      "model_forward_time": 1.7483208179473877,
      "step": 45319
    },
    {
      "epoch": 4.3219566345214846e-05,
      "step": 45319,
      "training_step_time": 7.344429016113281
    },
    {
      "epoch": 4.322052001953125e-05,
      "grad_norm": 0.10003741830587387,
      "learning_rate": 6.176228404805839e-06,
      "loss": 0.0584,
      "step": 45320
    },
    {
      "epoch": 4.322052001953125e-05,
      "model_forward_time": 1.7566618919372559,
      "step": 45320
    },
    {
      "dynamics_loss_avg": 0.06386203058063984,
      "epoch": 4.322052001953125e-05,
      "step": 45320
    },
    {
      "action_loss_avg": 0.0017010022478643806,
      "epoch": 4.322052001953125e-05,
      "step": 45320
    },
    {
      "epoch": 4.322052001953125e-05,
      "step": 45320,
      "training_step_time": 7.343325614929199
    },
    {
      "epoch": 4.3221473693847654e-05,
      "model_forward_time": 1.7898468971252441,
      "step": 45321
    },
    {
      "epoch": 4.3221473693847654e-05,
      "step": 45321,
      "training_step_time": 7.605211973190308
    },
    {
      "epoch": 4.3222427368164065e-05,
      "model_forward_time": 1.7449498176574707,
      "step": 45322
    },
    {
      "epoch": 4.3222427368164065e-05,
      "step": 45322,
      "training_step_time": 7.828956842422485
    },
    {
      "epoch": 4.322338104248047e-05,
      "model_forward_time": 2.330862045288086,
      "step": 45323
    },
    {
      "epoch": 4.322338104248047e-05,
      "step": 45323,
      "training_step_time": 8.53385591506958
    },
    {
      "epoch": 4.322433471679687e-05,
      "model_forward_time": 1.739330530166626,
      "step": 45324
    },
    {
      "epoch": 4.322433471679687e-05,
      "step": 45324,
      "training_step_time": 9.337114810943604
    },
    {
      "epoch": 4.3225288391113284e-05,
      "model_forward_time": 1.7592949867248535,
      "step": 45325
    },
    {
      "epoch": 4.3225288391113284e-05,
      "step": 45325,
      "training_step_time": 8.655867099761963
    },
    {
      "epoch": 4.322624206542969e-05,
      "model_forward_time": 1.7494769096374512,
      "step": 45326
    },
    {
      "epoch": 4.322624206542969e-05,
      "step": 45326,
      "training_step_time": 7.8923680782318115
    },
    {
      "epoch": 4.322719573974609e-05,
      "model_forward_time": 1.7462615966796875,
      "step": 45327
    },
    {
      "epoch": 4.322719573974609e-05,
      "step": 45327,
      "training_step_time": 6.805969476699829
    },
    {
      "epoch": 4.32281494140625e-05,
      "model_forward_time": 1.7484569549560547,
      "step": 45328
    },
    {
      "epoch": 4.32281494140625e-05,
      "step": 45328,
      "training_step_time": 7.338496208190918
    },
    {
      "epoch": 4.322910308837891e-05,
      "model_forward_time": 1.7498416900634766,
      "step": 45329
    },
    {
      "epoch": 4.322910308837891e-05,
      "step": 45329,
      "training_step_time": 7.694165468215942
    },
    {
      "epoch": 4.323005676269531e-05,
      "grad_norm": 0.11491825431585312,
      "learning_rate": 6.1746212737799315e-06,
      "loss": 0.0613,
      "step": 45330
    },
    {
      "epoch": 4.323005676269531e-05,
      "model_forward_time": 1.7490167617797852,
      "step": 45330
    },
    {
      "dynamics_loss_avg": 0.053430001717060804,
      "epoch": 4.323005676269531e-05,
      "step": 45330
    },
    {
      "action_loss_avg": 0.001176271861186251,
      "epoch": 4.323005676269531e-05,
      "step": 45330
    },
    {
      "epoch": 4.323005676269531e-05,
      "step": 45330,
      "training_step_time": 7.595694303512573
    },
    {
      "epoch": 4.323101043701172e-05,
      "model_forward_time": 1.7558567523956299,
      "step": 45331
    },
    {
      "epoch": 4.323101043701172e-05,
      "step": 45331,
      "training_step_time": 7.73162579536438
    },
    {
      "epoch": 4.3231964111328126e-05,
      "model_forward_time": 1.751127004623413,
      "step": 45332
    },
    {
      "epoch": 4.3231964111328126e-05,
      "step": 45332,
      "training_step_time": 10.675129652023315
    },
    {
      "epoch": 4.323291778564453e-05,
      "model_forward_time": 1.7458837032318115,
      "step": 45333
    },
    {
      "epoch": 4.323291778564453e-05,
      "step": 45333,
      "training_step_time": 9.112077474594116
    },
    {
      "epoch": 4.323387145996094e-05,
      "model_forward_time": 1.7466049194335938,
      "step": 45334
    },
    {
      "epoch": 4.323387145996094e-05,
      "step": 45334,
      "training_step_time": 7.7196409702301025
    },
    {
      "epoch": 4.3234825134277345e-05,
      "model_forward_time": 1.7560834884643555,
      "step": 45335
    },
    {
      "epoch": 4.3234825134277345e-05,
      "step": 45335,
      "training_step_time": 7.576214551925659
    },
    {
      "epoch": 4.323577880859375e-05,
      "model_forward_time": 1.7482733726501465,
      "step": 45336
    },
    {
      "epoch": 4.323577880859375e-05,
      "step": 45336,
      "training_step_time": 6.7541303634643555
    },
    {
      "epoch": 4.323673248291016e-05,
      "model_forward_time": 1.7467541694641113,
      "step": 45337
    },
    {
      "epoch": 4.323673248291016e-05,
      "step": 45337,
      "training_step_time": 7.254600524902344
    },
    {
      "epoch": 4.3237686157226563e-05,
      "model_forward_time": 1.7383368015289307,
      "step": 45338
    },
    {
      "epoch": 4.3237686157226563e-05,
      "step": 45338,
      "training_step_time": 7.278003454208374
    },
    {
      "epoch": 4.323863983154297e-05,
      "model_forward_time": 1.755723237991333,
      "step": 45339
    },
    {
      "epoch": 4.323863983154297e-05,
      "step": 45339,
      "training_step_time": 8.244919061660767
    },
    {
      "epoch": 4.323959350585938e-05,
      "grad_norm": 0.10433512926101685,
      "learning_rate": 6.173014014299209e-06,
      "loss": 0.0613,
      "step": 45340
    },
    {
      "epoch": 4.323959350585938e-05,
      "model_forward_time": 1.439462661743164,
      "step": 45340
    },
    {
      "dynamics_loss_avg": 0.048335440829396246,
      "epoch": 4.323959350585938e-05,
      "step": 45340
    },
    {
      "action_loss_avg": 0.0016648360207909719,
      "epoch": 4.323959350585938e-05,
      "step": 45340
    },
    {
      "epoch": 4.323959350585938e-05,
      "step": 45340,
      "training_step_time": 7.684129953384399
    },
    {
      "epoch": 4.324054718017578e-05,
      "model_forward_time": 1.7513632774353027,
      "step": 45341
    },
    {
      "epoch": 4.324054718017578e-05,
      "step": 45341,
      "training_step_time": 8.356939315795898
    },
    {
      "epoch": 4.3241500854492186e-05,
      "model_forward_time": 2.3246917724609375,
      "step": 45342
    },
    {
      "epoch": 4.3241500854492186e-05,
      "step": 45342,
      "training_step_time": 10.282309770584106
    },
    {
      "epoch": 4.32424545288086e-05,
      "model_forward_time": 1.7386186122894287,
      "step": 45343
    },
    {
      "epoch": 4.32424545288086e-05,
      "step": 45343,
      "training_step_time": 7.663238525390625
    },
    {
      "epoch": 4.3243408203125e-05,
      "model_forward_time": 1.7461168766021729,
      "step": 45344
    },
    {
      "epoch": 4.3243408203125e-05,
      "step": 45344,
      "training_step_time": 7.720045566558838
    },
    {
      "epoch": 4.3244361877441405e-05,
      "model_forward_time": 1.719724178314209,
      "step": 45345
    },
    {
      "epoch": 4.3244361877441405e-05,
      "step": 45345,
      "training_step_time": 7.786420822143555
    },
    {
      "epoch": 4.324531555175781e-05,
      "model_forward_time": 1.7511582374572754,
      "step": 45346
    },
    {
      "epoch": 4.324531555175781e-05,
      "step": 45346,
      "training_step_time": 6.775165319442749
    },
    {
      "epoch": 4.324626922607422e-05,
      "model_forward_time": 1.7363717555999756,
      "step": 45347
    },
    {
      "epoch": 4.324626922607422e-05,
      "step": 45347,
      "training_step_time": 7.331213712692261
    },
    {
      "epoch": 4.3247222900390624e-05,
      "model_forward_time": 1.746840476989746,
      "step": 45348
    },
    {
      "epoch": 4.3247222900390624e-05,
      "step": 45348,
      "training_step_time": 7.701517343521118
    },
    {
      "epoch": 4.324817657470703e-05,
      "model_forward_time": 1.7556943893432617,
      "step": 45349
    },
    {
      "epoch": 4.324817657470703e-05,
      "step": 45349,
      "training_step_time": 9.103684186935425
    },
    {
      "epoch": 4.324913024902344e-05,
      "grad_norm": 0.08805966377258301,
      "learning_rate": 6.171406626539437e-06,
      "loss": 0.0578,
      "step": 45350
    },
    {
      "epoch": 4.324913024902344e-05,
      "model_forward_time": 1.0505683422088623,
      "step": 45350
    },
    {
      "dynamics_loss_avg": 0.06774079389870166,
      "epoch": 4.324913024902344e-05,
      "step": 45350
    },
    {
      "action_loss_avg": 0.0016963055939413607,
      "epoch": 4.324913024902344e-05,
      "step": 45350
    },
    {
      "epoch": 4.324913024902344e-05,
      "step": 45350,
      "training_step_time": 8.746609210968018
    },
    {
      "epoch": 4.325008392333984e-05,
      "model_forward_time": 2.351459264755249,
      "step": 45351
    },
    {
      "epoch": 4.325008392333984e-05,
      "step": 45351,
      "training_step_time": 9.414330244064331
    },
    {
      "epoch": 4.325103759765625e-05,
      "model_forward_time": 1.7476882934570312,
      "step": 45352
    },
    {
      "epoch": 4.325103759765625e-05,
      "step": 45352,
      "training_step_time": 8.420578956604004
    },
    {
      "epoch": 4.325199127197266e-05,
      "model_forward_time": 1.7499058246612549,
      "step": 45353
    },
    {
      "epoch": 4.325199127197266e-05,
      "step": 45353,
      "training_step_time": 7.617562532424927
    },
    {
      "epoch": 4.325294494628906e-05,
      "model_forward_time": 1.752091884613037,
      "step": 45354
    },
    {
      "epoch": 4.325294494628906e-05,
      "step": 45354,
      "training_step_time": 7.741771936416626
    },
    {
      "epoch": 4.3253898620605466e-05,
      "model_forward_time": 1.7460567951202393,
      "step": 45355
    },
    {
      "epoch": 4.3253898620605466e-05,
      "step": 45355,
      "training_step_time": 6.866795301437378
    },
    {
      "epoch": 4.3254852294921877e-05,
      "model_forward_time": 1.7523620128631592,
      "step": 45356
    },
    {
      "epoch": 4.3254852294921877e-05,
      "step": 45356,
      "training_step_time": 7.291236877441406
    },
    {
      "epoch": 4.325580596923828e-05,
      "model_forward_time": 1.7485020160675049,
      "step": 45357
    },
    {
      "epoch": 4.325580596923828e-05,
      "step": 45357,
      "training_step_time": 7.740881681442261
    },
    {
      "epoch": 4.3256759643554685e-05,
      "model_forward_time": 1.7595329284667969,
      "step": 45358
    },
    {
      "epoch": 4.3256759643554685e-05,
      "step": 45358,
      "training_step_time": 7.709211111068726
    },
    {
      "epoch": 4.3257713317871095e-05,
      "model_forward_time": 1.7584850788116455,
      "step": 45359
    },
    {
      "epoch": 4.3257713317871095e-05,
      "step": 45359,
      "training_step_time": 8.46994924545288
    },
    {
      "epoch": 4.32586669921875e-05,
      "grad_norm": 0.12136747688055038,
      "learning_rate": 6.1697991106763975e-06,
      "loss": 0.062,
      "step": 45360
    },
    {
      "epoch": 4.32586669921875e-05,
      "model_forward_time": 1.7416355609893799,
      "step": 45360
    },
    {
      "dynamics_loss_avg": 0.06395981758832932,
      "epoch": 4.32586669921875e-05,
      "step": 45360
    },
    {
      "action_loss_avg": 0.0015421657706610858,
      "epoch": 4.32586669921875e-05,
      "step": 45360
    },
    {
      "epoch": 4.32586669921875e-05,
      "step": 45360,
      "training_step_time": 10.021796941757202
    },
    {
      "epoch": 4.3259620666503903e-05,
      "model_forward_time": 1.7997658252716064,
      "step": 45361
    },
    {
      "epoch": 4.3259620666503903e-05,
      "step": 45361,
      "training_step_time": 7.65963339805603
    },
    {
      "epoch": 4.3260574340820314e-05,
      "model_forward_time": 1.7584259510040283,
      "step": 45362
    },
    {
      "epoch": 4.3260574340820314e-05,
      "step": 45362,
      "training_step_time": 7.7543957233428955
    },
    {
      "epoch": 4.326152801513672e-05,
      "model_forward_time": 1.7515852451324463,
      "step": 45363
    },
    {
      "epoch": 4.326152801513672e-05,
      "step": 45363,
      "training_step_time": 8.765485048294067
    },
    {
      "epoch": 4.326248168945312e-05,
      "model_forward_time": 1.7587714195251465,
      "step": 45364
    },
    {
      "epoch": 4.326248168945312e-05,
      "step": 45364,
      "training_step_time": 6.758870601654053
    },
    {
      "epoch": 4.326343536376953e-05,
      "model_forward_time": 1.7566184997558594,
      "step": 45365
    },
    {
      "epoch": 4.326343536376953e-05,
      "step": 45365,
      "training_step_time": 7.272984027862549
    },
    {
      "epoch": 4.326438903808594e-05,
      "model_forward_time": 1.7533495426177979,
      "step": 45366
    },
    {
      "epoch": 4.326438903808594e-05,
      "step": 45366,
      "training_step_time": 7.670707702636719
    },
    {
      "epoch": 4.326534271240234e-05,
      "model_forward_time": 1.7574787139892578,
      "step": 45367
    },
    {
      "epoch": 4.326534271240234e-05,
      "step": 45367,
      "training_step_time": 7.654972791671753
    },
    {
      "epoch": 4.326629638671875e-05,
      "model_forward_time": 1.7491984367370605,
      "step": 45368
    },
    {
      "epoch": 4.326629638671875e-05,
      "step": 45368,
      "training_step_time": 8.691049575805664
    },
    {
      "epoch": 4.3267250061035156e-05,
      "model_forward_time": 1.7609801292419434,
      "step": 45369
    },
    {
      "epoch": 4.3267250061035156e-05,
      "step": 45369,
      "training_step_time": 9.274627923965454
    },
    {
      "epoch": 4.326820373535156e-05,
      "grad_norm": 0.15569324791431427,
      "learning_rate": 6.168191466885885e-06,
      "loss": 0.061,
      "step": 45370
    },
    {
      "epoch": 4.326820373535156e-05,
      "model_forward_time": 2.3737146854400635,
      "step": 45370
    },
    {
      "dynamics_loss_avg": 0.07247674744576216,
      "epoch": 4.326820373535156e-05,
      "step": 45370
    },
    {
      "action_loss_avg": 0.0021959082514513284,
      "epoch": 4.326820373535156e-05,
      "step": 45370
    },
    {
      "epoch": 4.326820373535156e-05,
      "step": 45370,
      "training_step_time": 8.312018871307373
    },
    {
      "epoch": 4.326915740966797e-05,
      "model_forward_time": 1.747258186340332,
      "step": 45371
    },
    {
      "epoch": 4.326915740966797e-05,
      "step": 45371,
      "training_step_time": 7.658563137054443
    },
    {
      "epoch": 4.3270111083984375e-05,
      "model_forward_time": 1.7624812126159668,
      "step": 45372
    },
    {
      "epoch": 4.3270111083984375e-05,
      "step": 45372,
      "training_step_time": 7.656925916671753
    },
    {
      "epoch": 4.327106475830078e-05,
      "model_forward_time": 1.3826539516448975,
      "step": 45373
    },
    {
      "epoch": 4.327106475830078e-05,
      "step": 45373,
      "training_step_time": 7.8295769691467285
    },
    {
      "epoch": 4.327201843261719e-05,
      "model_forward_time": 1.7516374588012695,
      "step": 45374
    },
    {
      "epoch": 4.327201843261719e-05,
      "step": 45374,
      "training_step_time": 7.369605541229248
    },
    {
      "epoch": 4.3272972106933594e-05,
      "model_forward_time": 1.7590367794036865,
      "step": 45375
    },
    {
      "epoch": 4.3272972106933594e-05,
      "step": 45375,
      "training_step_time": 7.785571575164795
    },
    {
      "epoch": 4.327392578125e-05,
      "model_forward_time": 1.7371437549591064,
      "step": 45376
    },
    {
      "epoch": 4.327392578125e-05,
      "step": 45376,
      "training_step_time": 7.726181507110596
    },
    {
      "epoch": 4.327487945556641e-05,
      "model_forward_time": 1.7392587661743164,
      "step": 45377
    },
    {
      "epoch": 4.327487945556641e-05,
      "step": 45377,
      "training_step_time": 8.539192199707031
    },
    {
      "epoch": 4.327583312988281e-05,
      "model_forward_time": 1.7557551860809326,
      "step": 45378
    },
    {
      "epoch": 4.327583312988281e-05,
      "step": 45378,
      "training_step_time": 8.436340808868408
    },
    {
      "epoch": 4.327678680419922e-05,
      "model_forward_time": 2.365125894546509,
      "step": 45379
    },
    {
      "epoch": 4.327678680419922e-05,
      "step": 45379,
      "training_step_time": 7.967059373855591
    },
    {
      "epoch": 4.327774047851563e-05,
      "grad_norm": 0.1263100951910019,
      "learning_rate": 6.166583695343711e-06,
      "loss": 0.0614,
      "step": 45380
    },
    {
      "epoch": 4.327774047851563e-05,
      "model_forward_time": 1.7461888790130615,
      "step": 45380
    },
    {
      "dynamics_loss_avg": 0.07765981908887624,
      "epoch": 4.327774047851563e-05,
      "step": 45380
    },
    {
      "action_loss_avg": 0.002223906933795661,
      "epoch": 4.327774047851563e-05,
      "step": 45380
    },
    {
      "epoch": 4.327774047851563e-05,
      "step": 45380,
      "training_step_time": 8.872726917266846
    },
    {
      "epoch": 4.327869415283203e-05,
      "model_forward_time": 1.7610094547271729,
      "step": 45381
    },
    {
      "epoch": 4.327869415283203e-05,
      "step": 45381,
      "training_step_time": 7.866570472717285
    },
    {
      "epoch": 4.3279647827148435e-05,
      "model_forward_time": 1.7614455223083496,
      "step": 45382
    },
    {
      "epoch": 4.3279647827148435e-05,
      "step": 45382,
      "training_step_time": 8.735869884490967
    },
    {
      "epoch": 4.3280601501464846e-05,
      "model_forward_time": 1.754495620727539,
      "step": 45383
    },
    {
      "epoch": 4.3280601501464846e-05,
      "step": 45383,
      "training_step_time": 7.456272125244141
    },
    {
      "epoch": 4.328155517578125e-05,
      "model_forward_time": 1.7573764324188232,
      "step": 45384
    },
    {
      "epoch": 4.328155517578125e-05,
      "step": 45384,
      "training_step_time": 7.926500558853149
    },
    {
      "epoch": 4.3282508850097654e-05,
      "model_forward_time": 1.760362148284912,
      "step": 45385
    },
    {
      "epoch": 4.3282508850097654e-05,
      "step": 45385,
      "training_step_time": 8.145249128341675
    },
    {
      "epoch": 4.3283462524414065e-05,
      "model_forward_time": 1.7461237907409668,
      "step": 45386
    },
    {
      "epoch": 4.3283462524414065e-05,
      "step": 45386,
      "training_step_time": 8.691168785095215
    },
    {
      "epoch": 4.328441619873047e-05,
      "model_forward_time": 1.7452688217163086,
      "step": 45387
    },
    {
      "epoch": 4.328441619873047e-05,
      "step": 45387,
      "training_step_time": 11.65474796295166
    },
    {
      "epoch": 4.328536987304687e-05,
      "model_forward_time": 2.368238925933838,
      "step": 45388
    },
    {
      "epoch": 4.328536987304687e-05,
      "step": 45388,
      "training_step_time": 8.685360193252563
    },
    {
      "epoch": 4.3286323547363284e-05,
      "model_forward_time": 1.74574875831604,
      "step": 45389
    },
    {
      "epoch": 4.3286323547363284e-05,
      "step": 45389,
      "training_step_time": 7.768167495727539
    },
    {
      "epoch": 4.328727722167969e-05,
      "grad_norm": 0.12697310745716095,
      "learning_rate": 6.164975796225699e-06,
      "loss": 0.0588,
      "step": 45390
    },
    {
      "epoch": 4.328727722167969e-05,
      "model_forward_time": 1.746103286743164,
      "step": 45390
    },
    {
      "dynamics_loss_avg": 0.06117096543312073,
      "epoch": 4.328727722167969e-05,
      "step": 45390
    },
    {
      "action_loss_avg": 0.0028297261276748033,
      "epoch": 4.328727722167969e-05,
      "step": 45390
    },
    {
      "epoch": 4.328727722167969e-05,
      "step": 45390,
      "training_step_time": 7.798845052719116
    },
    {
      "epoch": 4.328823089599609e-05,
      "model_forward_time": 1.7037663459777832,
      "step": 45391
    },
    {
      "epoch": 4.328823089599609e-05,
      "step": 45391,
      "training_step_time": 7.630739212036133
    },
    {
      "epoch": 4.32891845703125e-05,
      "model_forward_time": 1.7115564346313477,
      "step": 45392
    },
    {
      "epoch": 4.32891845703125e-05,
      "step": 45392,
      "training_step_time": 7.778403997421265
    },
    {
      "epoch": 4.329013824462891e-05,
      "model_forward_time": 1.3471026420593262,
      "step": 45393
    },
    {
      "epoch": 4.329013824462891e-05,
      "step": 45393,
      "training_step_time": 7.363202333450317
    },
    {
      "epoch": 4.329109191894531e-05,
      "model_forward_time": 1.7096128463745117,
      "step": 45394
    },
    {
      "epoch": 4.329109191894531e-05,
      "step": 45394,
      "training_step_time": 8.478084325790405
    },
    {
      "epoch": 4.329204559326172e-05,
      "model_forward_time": 1.7224185466766357,
      "step": 45395
    },
    {
      "epoch": 4.329204559326172e-05,
      "step": 45395,
      "training_step_time": 8.694882154464722
    },
    {
      "epoch": 4.3292999267578126e-05,
      "model_forward_time": 1.7130911350250244,
      "step": 45396
    },
    {
      "epoch": 4.3292999267578126e-05,
      "step": 45396,
      "training_step_time": 8.534812450408936
    },
    {
      "epoch": 4.329395294189453e-05,
      "model_forward_time": 2.325575590133667,
      "step": 45397
    },
    {
      "epoch": 4.329395294189453e-05,
      "step": 45397,
      "training_step_time": 7.865443706512451
    },
    {
      "epoch": 4.329490661621094e-05,
      "model_forward_time": 1.7422161102294922,
      "step": 45398
    },
    {
      "epoch": 4.329490661621094e-05,
      "step": 45398,
      "training_step_time": 8.76463270187378
    },
    {
      "epoch": 4.3295860290527345e-05,
      "model_forward_time": 1.7440509796142578,
      "step": 45399
    },
    {
      "epoch": 4.3295860290527345e-05,
      "step": 45399,
      "training_step_time": 9.130767583847046
    },
    {
      "epoch": 4.329681396484375e-05,
      "grad_norm": 0.12787789106369019,
      "learning_rate": 6.163367769707683e-06,
      "loss": 0.059,
      "step": 45400
    },
    {
      "epoch": 4.329681396484375e-05,
      "model_forward_time": 1.7373175621032715,
      "step": 45400
    },
    {
      "dynamics_loss_avg": 0.050333180278539655,
      "epoch": 4.329681396484375e-05,
      "step": 45400
    },
    {
      "action_loss_avg": 0.0016555739217437803,
      "epoch": 4.329681396484375e-05,
      "step": 45400
    },
    {
      "epoch": 4.329681396484375e-05,
      "step": 45400,
      "training_step_time": 7.804831027984619
    },
    {
      "epoch": 4.329776763916016e-05,
      "model_forward_time": 1.732672929763794,
      "step": 45401
    },
    {
      "epoch": 4.329776763916016e-05,
      "step": 45401,
      "training_step_time": 7.821166515350342
    },
    {
      "epoch": 4.3298721313476563e-05,
      "model_forward_time": 1.747880220413208,
      "step": 45402
    },
    {
      "epoch": 4.3298721313476563e-05,
      "step": 45402,
      "training_step_time": 7.36276912689209
    },
    {
      "epoch": 4.329967498779297e-05,
      "model_forward_time": 1.7408068180084229,
      "step": 45403
    },
    {
      "epoch": 4.329967498779297e-05,
      "step": 45403,
      "training_step_time": 7.788377523422241
    },
    {
      "epoch": 4.330062866210938e-05,
      "model_forward_time": 1.7412970066070557,
      "step": 45404
    },
    {
      "epoch": 4.330062866210938e-05,
      "step": 45404,
      "training_step_time": 7.7648046016693115
    },
    {
      "epoch": 4.330158233642578e-05,
      "model_forward_time": 1.7383382320404053,
      "step": 45405
    },
    {
      "epoch": 4.330158233642578e-05,
      "step": 45405,
      "training_step_time": 8.342788219451904
    },
    {
      "epoch": 4.3302536010742186e-05,
      "model_forward_time": 2.238588809967041,
      "step": 45406
    },
    {
      "epoch": 4.3302536010742186e-05,
      "step": 45406,
      "training_step_time": 8.54641318321228
    },
    {
      "epoch": 4.330348968505859e-05,
      "model_forward_time": 1.704209566116333,
      "step": 45407
    },
    {
      "epoch": 4.330348968505859e-05,
      "step": 45407,
      "training_step_time": 8.806887149810791
    },
    {
      "epoch": 4.3304443359375e-05,
      "model_forward_time": 1.7078416347503662,
      "step": 45408
    },
    {
      "epoch": 4.3304443359375e-05,
      "step": 45408,
      "training_step_time": 7.741051197052002
    },
    {
      "epoch": 4.3305397033691405e-05,
      "model_forward_time": 1.3415710926055908,
      "step": 45409
    },
    {
      "epoch": 4.3305397033691405e-05,
      "step": 45409,
      "training_step_time": 7.795576095581055
    },
    {
      "epoch": 4.330635070800781e-05,
      "grad_norm": 0.15535680949687958,
      "learning_rate": 6.161759615965519e-06,
      "loss": 0.0563,
      "step": 45410
    },
    {
      "epoch": 4.330635070800781e-05,
      "model_forward_time": 1.7048802375793457,
      "step": 45410
    },
    {
      "dynamics_loss_avg": 0.07745761051774025,
      "epoch": 4.330635070800781e-05,
      "step": 45410
    },
    {
      "action_loss_avg": 0.002782778808614239,
      "epoch": 4.330635070800781e-05,
      "step": 45410
    },
    {
      "epoch": 4.330635070800781e-05,
      "step": 45410,
      "training_step_time": 7.8372557163238525
    },
    {
      "epoch": 4.330730438232422e-05,
      "model_forward_time": 1.7095396518707275,
      "step": 45411
    },
    {
      "epoch": 4.330730438232422e-05,
      "step": 45411,
      "training_step_time": 7.40245795249939
    },
    {
      "epoch": 4.3308258056640624e-05,
      "model_forward_time": 1.710357666015625,
      "step": 45412
    },
    {
      "epoch": 4.3308258056640624e-05,
      "step": 45412,
      "training_step_time": 7.877960205078125
    },
    {
      "epoch": 4.330921173095703e-05,
      "model_forward_time": 1.7084503173828125,
      "step": 45413
    },
    {
      "epoch": 4.330921173095703e-05,
      "step": 45413,
      "training_step_time": 7.7136006355285645
    },
    {
      "epoch": 4.331016540527344e-05,
      "model_forward_time": 1.7130413055419922,
      "step": 45414
    },
    {
      "epoch": 4.331016540527344e-05,
      "step": 45414,
      "training_step_time": 7.757069826126099
    },
    {
      "epoch": 4.331111907958984e-05,
      "model_forward_time": 1.7116978168487549,
      "step": 45415
    },
    {
      "epoch": 4.331111907958984e-05,
      "step": 45415,
      "training_step_time": 7.777416944503784
    },
    {
      "epoch": 4.331207275390625e-05,
      "model_forward_time": 2.254850149154663,
      "step": 45416
    },
    {
      "epoch": 4.331207275390625e-05,
      "step": 45416,
      "training_step_time": 8.45618224143982
    },
    {
      "epoch": 4.331302642822266e-05,
      "model_forward_time": 1.7094569206237793,
      "step": 45417
    },
    {
      "epoch": 4.331302642822266e-05,
      "step": 45417,
      "training_step_time": 7.870870590209961
    },
    {
      "epoch": 4.331398010253906e-05,
      "model_forward_time": 1.7091994285583496,
      "step": 45418
    },
    {
      "epoch": 4.331398010253906e-05,
      "step": 45418,
      "training_step_time": 7.823362350463867
    },
    {
      "epoch": 4.3314933776855466e-05,
      "model_forward_time": 1.720719337463379,
      "step": 45419
    },
    {
      "epoch": 4.3314933776855466e-05,
      "step": 45419,
      "training_step_time": 7.801111221313477
    },
    {
      "epoch": 4.331588745117188e-05,
      "grad_norm": 0.09381495416164398,
      "learning_rate": 6.16015133517507e-06,
      "loss": 0.0583,
      "step": 45420
    },
    {
      "epoch": 4.331588745117188e-05,
      "model_forward_time": 1.744103193283081,
      "step": 45420
    },
    {
      "dynamics_loss_avg": 0.06997788734734059,
      "epoch": 4.331588745117188e-05,
      "step": 45420
    },
    {
      "action_loss_avg": 0.0019088911009021103,
      "epoch": 4.331588745117188e-05,
      "step": 45420
    },
    {
      "epoch": 4.331588745117188e-05,
      "step": 45420,
      "training_step_time": 7.357436656951904
    },
    {
      "epoch": 4.331684112548828e-05,
      "model_forward_time": 1.7409718036651611,
      "step": 45421
    },
    {
      "epoch": 4.331684112548828e-05,
      "step": 45421,
      "training_step_time": 7.869077205657959
    },
    {
      "epoch": 4.3317794799804685e-05,
      "model_forward_time": 1.7366869449615479,
      "step": 45422
    },
    {
      "epoch": 4.3317794799804685e-05,
      "step": 45422,
      "training_step_time": 8.426646947860718
    },
    {
      "epoch": 4.3318748474121095e-05,
      "model_forward_time": 1.740267038345337,
      "step": 45423
    },
    {
      "epoch": 4.3318748474121095e-05,
      "step": 45423,
      "training_step_time": 8.722559690475464
    },
    {
      "epoch": 4.33197021484375e-05,
      "model_forward_time": 1.7349185943603516,
      "step": 45424
    },
    {
      "epoch": 4.33197021484375e-05,
      "step": 45424,
      "training_step_time": 8.947408676147461
    },
    {
      "epoch": 4.3320655822753904e-05,
      "model_forward_time": 2.284512996673584,
      "step": 45425
    },
    {
      "epoch": 4.3320655822753904e-05,
      "step": 45425,
      "training_step_time": 9.787876605987549
    },
    {
      "epoch": 4.3321609497070314e-05,
      "model_forward_time": 1.7582895755767822,
      "step": 45426
    },
    {
      "epoch": 4.3321609497070314e-05,
      "step": 45426,
      "training_step_time": 7.805069446563721
    },
    {
      "epoch": 4.332256317138672e-05,
      "model_forward_time": 1.7371950149536133,
      "step": 45427
    },
    {
      "epoch": 4.332256317138672e-05,
      "step": 45427,
      "training_step_time": 7.7805562019348145
    },
    {
      "epoch": 4.332351684570312e-05,
      "model_forward_time": 1.7080276012420654,
      "step": 45428
    },
    {
      "epoch": 4.332351684570312e-05,
      "step": 45428,
      "training_step_time": 7.808841228485107
    },
    {
      "epoch": 4.332447052001953e-05,
      "model_forward_time": 1.7115139961242676,
      "step": 45429
    },
    {
      "epoch": 4.332447052001953e-05,
      "step": 45429,
      "training_step_time": 7.258104562759399
    },
    {
      "epoch": 4.332542419433594e-05,
      "grad_norm": 0.07552237063646317,
      "learning_rate": 6.1585429275122135e-06,
      "loss": 0.0557,
      "step": 45430
    },
    {
      "epoch": 4.332542419433594e-05,
      "model_forward_time": 1.703885555267334,
      "step": 45430
    },
    {
      "dynamics_loss_avg": 0.0562659215182066,
      "epoch": 4.332542419433594e-05,
      "step": 45430
    },
    {
      "action_loss_avg": 0.0017590475676115603,
      "epoch": 4.332542419433594e-05,
      "step": 45430
    },
    {
      "epoch": 4.332542419433594e-05,
      "step": 45430,
      "training_step_time": 7.829275846481323
    },
    {
      "epoch": 4.332637786865234e-05,
      "model_forward_time": 1.7183794975280762,
      "step": 45431
    },
    {
      "epoch": 4.332637786865234e-05,
      "step": 45431,
      "training_step_time": 8.412220001220703
    },
    {
      "epoch": 4.332733154296875e-05,
      "model_forward_time": 1.3855524063110352,
      "step": 45432
    },
    {
      "epoch": 4.332733154296875e-05,
      "step": 45432,
      "training_step_time": 7.829069375991821
    },
    {
      "epoch": 4.3328285217285156e-05,
      "model_forward_time": 1.7322571277618408,
      "step": 45433
    },
    {
      "epoch": 4.3328285217285156e-05,
      "step": 45433,
      "training_step_time": 8.878670930862427
    },
    {
      "epoch": 4.332923889160156e-05,
      "model_forward_time": 2.259490966796875,
      "step": 45434
    },
    {
      "epoch": 4.332923889160156e-05,
      "step": 45434,
      "training_step_time": 8.504219770431519
    },
    {
      "epoch": 4.333019256591797e-05,
      "model_forward_time": 1.7689366340637207,
      "step": 45435
    },
    {
      "epoch": 4.333019256591797e-05,
      "step": 45435,
      "training_step_time": 8.681179523468018
    },
    {
      "epoch": 4.3331146240234375e-05,
      "model_forward_time": 1.353529691696167,
      "step": 45436
    },
    {
      "epoch": 4.3331146240234375e-05,
      "step": 45436,
      "training_step_time": 7.793835401535034
    },
    {
      "epoch": 4.333209991455078e-05,
      "model_forward_time": 1.715527057647705,
      "step": 45437
    },
    {
      "epoch": 4.333209991455078e-05,
      "step": 45437,
      "training_step_time": 7.708428144454956
    },
    {
      "epoch": 4.333305358886719e-05,
      "model_forward_time": 1.7162916660308838,
      "step": 45438
    },
    {
      "epoch": 4.333305358886719e-05,
      "step": 45438,
      "training_step_time": 7.765105247497559
    },
    {
      "epoch": 4.3334007263183594e-05,
      "model_forward_time": 1.719036340713501,
      "step": 45439
    },
    {
      "epoch": 4.3334007263183594e-05,
      "step": 45439,
      "training_step_time": 7.851556062698364
    },
    {
      "epoch": 4.33349609375e-05,
      "grad_norm": 0.11514174938201904,
      "learning_rate": 6.156934393152845e-06,
      "loss": 0.061,
      "step": 45440
    },
    {
      "epoch": 4.33349609375e-05,
      "model_forward_time": 1.723536491394043,
      "step": 45440
    },
    {
      "dynamics_loss_avg": 0.061817492172122,
      "epoch": 4.33349609375e-05,
      "step": 45440
    },
    {
      "action_loss_avg": 0.002574606507550925,
      "epoch": 4.33349609375e-05,
      "step": 45440
    },
    {
      "epoch": 4.33349609375e-05,
      "step": 45440,
      "training_step_time": 6.884248733520508
    },
    {
      "epoch": 4.333591461181641e-05,
      "model_forward_time": 1.7210211753845215,
      "step": 45441
    },
    {
      "epoch": 4.333591461181641e-05,
      "step": 45441,
      "training_step_time": 8.72025179862976
    },
    {
      "epoch": 4.333686828613281e-05,
      "model_forward_time": 1.7403969764709473,
      "step": 45442
    },
    {
      "epoch": 4.333686828613281e-05,
      "step": 45442,
      "training_step_time": 8.87219786643982
    },
    {
      "epoch": 4.333782196044922e-05,
      "model_forward_time": 2.3096439838409424,
      "step": 45443
    },
    {
      "epoch": 4.333782196044922e-05,
      "step": 45443,
      "training_step_time": 8.773254632949829
    },
    {
      "epoch": 4.333877563476563e-05,
      "model_forward_time": 1.7429113388061523,
      "step": 45444
    },
    {
      "epoch": 4.333877563476563e-05,
      "step": 45444,
      "training_step_time": 8.334850549697876
    },
    {
      "epoch": 4.333972930908203e-05,
      "model_forward_time": 1.7275304794311523,
      "step": 45445
    },
    {
      "epoch": 4.333972930908203e-05,
      "step": 45445,
      "training_step_time": 7.877689361572266
    },
    {
      "epoch": 4.3340682983398436e-05,
      "model_forward_time": 1.7452118396759033,
      "step": 45446
    },
    {
      "epoch": 4.3340682983398436e-05,
      "step": 45446,
      "training_step_time": 8.69324803352356
    },
    {
      "epoch": 4.3341636657714846e-05,
      "model_forward_time": 1.0065548419952393,
      "step": 45447
    },
    {
      "epoch": 4.3341636657714846e-05,
      "step": 45447,
      "training_step_time": 7.764708042144775
    },
    {
      "epoch": 4.334259033203125e-05,
      "model_forward_time": 1.7249162197113037,
      "step": 45448
    },
    {
      "epoch": 4.334259033203125e-05,
      "step": 45448,
      "training_step_time": 7.245558261871338
    },
    {
      "epoch": 4.3343544006347654e-05,
      "model_forward_time": 1.7179064750671387,
      "step": 45449
    },
    {
      "epoch": 4.3343544006347654e-05,
      "step": 45449,
      "training_step_time": 7.703522443771362
    },
    {
      "epoch": 4.3344497680664065e-05,
      "grad_norm": 0.1278398633003235,
      "learning_rate": 6.155325732272871e-06,
      "loss": 0.0578,
      "step": 45450
    },
    {
      "epoch": 4.3344497680664065e-05,
      "model_forward_time": 1.7180299758911133,
      "step": 45450
    },
    {
      "dynamics_loss_avg": 0.07177430149167777,
      "epoch": 4.3344497680664065e-05,
      "step": 45450
    },
    {
      "action_loss_avg": 0.002065038518048823,
      "epoch": 4.3344497680664065e-05,
      "step": 45450
    },
    {
      "epoch": 4.3344497680664065e-05,
      "step": 45450,
      "training_step_time": 8.541415452957153
    },
    {
      "epoch": 4.334545135498047e-05,
      "model_forward_time": 1.7206814289093018,
      "step": 45451
    },
    {
      "epoch": 4.334545135498047e-05,
      "step": 45451,
      "training_step_time": 8.764952659606934
    },
    {
      "epoch": 4.334640502929687e-05,
      "model_forward_time": 1.7201957702636719,
      "step": 45452
    },
    {
      "epoch": 4.334640502929687e-05,
      "step": 45452,
      "training_step_time": 8.523032665252686
    },
    {
      "epoch": 4.3347358703613284e-05,
      "model_forward_time": 2.2504453659057617,
      "step": 45453
    },
    {
      "epoch": 4.3347358703613284e-05,
      "step": 45453,
      "training_step_time": 9.39803409576416
    },
    {
      "epoch": 4.334831237792969e-05,
      "model_forward_time": 1.7758703231811523,
      "step": 45454
    },
    {
      "epoch": 4.334831237792969e-05,
      "step": 45454,
      "training_step_time": 8.667698860168457
    },
    {
      "epoch": 4.334926605224609e-05,
      "model_forward_time": 1.7280638217926025,
      "step": 45455
    },
    {
      "epoch": 4.334926605224609e-05,
      "step": 45455,
      "training_step_time": 8.484363079071045
    },
    {
      "epoch": 4.33502197265625e-05,
      "model_forward_time": 1.7306337356567383,
      "step": 45456
    },
    {
      "epoch": 4.33502197265625e-05,
      "step": 45456,
      "training_step_time": 7.707678318023682
    },
    {
      "epoch": 4.335117340087891e-05,
      "model_forward_time": 1.725431203842163,
      "step": 45457
    },
    {
      "epoch": 4.335117340087891e-05,
      "step": 45457,
      "training_step_time": 7.2750163078308105
    },
    {
      "epoch": 4.335212707519531e-05,
      "model_forward_time": 1.7213046550750732,
      "step": 45458
    },
    {
      "epoch": 4.335212707519531e-05,
      "step": 45458,
      "training_step_time": 7.813825368881226
    },
    {
      "epoch": 4.335308074951172e-05,
      "model_forward_time": 1.7281200885772705,
      "step": 45459
    },
    {
      "epoch": 4.335308074951172e-05,
      "step": 45459,
      "training_step_time": 8.15047550201416
    },
    {
      "epoch": 4.3354034423828126e-05,
      "grad_norm": 0.12591727077960968,
      "learning_rate": 6.1537169450482114e-06,
      "loss": 0.0588,
      "step": 45460
    },
    {
      "epoch": 4.3354034423828126e-05,
      "model_forward_time": 1.7377033233642578,
      "step": 45460
    },
    {
      "dynamics_loss_avg": 0.04941449034959078,
      "epoch": 4.3354034423828126e-05,
      "step": 45460
    },
    {
      "action_loss_avg": 0.002713468763977289,
      "epoch": 4.3354034423828126e-05,
      "step": 45460
    },
    {
      "epoch": 4.3354034423828126e-05,
      "step": 45460,
      "training_step_time": 8.71885633468628
    },
    {
      "epoch": 4.335498809814453e-05,
      "model_forward_time": 1.7329967021942139,
      "step": 45461
    },
    {
      "epoch": 4.335498809814453e-05,
      "step": 45461,
      "training_step_time": 8.223788738250732
    },
    {
      "epoch": 4.335594177246094e-05,
      "model_forward_time": 1.885549783706665,
      "step": 45462
    },
    {
      "epoch": 4.335594177246094e-05,
      "step": 45462,
      "training_step_time": 8.544700860977173
    },
    {
      "epoch": 4.3356895446777345e-05,
      "model_forward_time": 1.7744019031524658,
      "step": 45463
    },
    {
      "epoch": 4.3356895446777345e-05,
      "step": 45463,
      "training_step_time": 7.7311577796936035
    },
    {
      "epoch": 4.335784912109375e-05,
      "model_forward_time": 1.7278380393981934,
      "step": 45464
    },
    {
      "epoch": 4.335784912109375e-05,
      "step": 45464,
      "training_step_time": 7.639530658721924
    },
    {
      "epoch": 4.335880279541016e-05,
      "model_forward_time": 1.723510980606079,
      "step": 45465
    },
    {
      "epoch": 4.335880279541016e-05,
      "step": 45465,
      "training_step_time": 8.21637773513794
    },
    {
      "epoch": 4.3359756469726564e-05,
      "model_forward_time": 1.7221646308898926,
      "step": 45466
    },
    {
      "epoch": 4.3359756469726564e-05,
      "step": 45466,
      "training_step_time": 7.361460208892822
    },
    {
      "epoch": 4.336071014404297e-05,
      "model_forward_time": 1.728654384613037,
      "step": 45467
    },
    {
      "epoch": 4.336071014404297e-05,
      "step": 45467,
      "training_step_time": 7.7423927783966064
    },
    {
      "epoch": 4.336166381835938e-05,
      "model_forward_time": 1.7239151000976562,
      "step": 45468
    },
    {
      "epoch": 4.336166381835938e-05,
      "step": 45468,
      "training_step_time": 7.554700613021851
    },
    {
      "epoch": 4.336261749267578e-05,
      "model_forward_time": 1.7285990715026855,
      "step": 45469
    },
    {
      "epoch": 4.336261749267578e-05,
      "step": 45469,
      "training_step_time": 7.657872915267944
    },
    {
      "epoch": 4.3363571166992186e-05,
      "grad_norm": 0.13697607815265656,
      "learning_rate": 6.152108031654802e-06,
      "loss": 0.0554,
      "step": 45470
    },
    {
      "epoch": 4.3363571166992186e-05,
      "model_forward_time": 1.719994068145752,
      "step": 45470
    },
    {
      "dynamics_loss_avg": 0.04954645074903965,
      "epoch": 4.3363571166992186e-05,
      "step": 45470
    },
    {
      "action_loss_avg": 0.0009715037216665223,
      "epoch": 4.3363571166992186e-05,
      "step": 45470
    },
    {
      "epoch": 4.3363571166992186e-05,
      "step": 45470,
      "training_step_time": 9.133212089538574
    },
    {
      "epoch": 4.336452484130859e-05,
      "model_forward_time": 2.2679343223571777,
      "step": 45471
    },
    {
      "epoch": 4.336452484130859e-05,
      "step": 45471,
      "training_step_time": 8.797419309616089
    },
    {
      "epoch": 4.3365478515625e-05,
      "model_forward_time": 1.7664473056793213,
      "step": 45472
    },
    {
      "epoch": 4.3365478515625e-05,
      "step": 45472,
      "training_step_time": 7.595703601837158
    },
    {
      "epoch": 4.3366432189941405e-05,
      "model_forward_time": 1.717947006225586,
      "step": 45473
    },
    {
      "epoch": 4.3366432189941405e-05,
      "step": 45473,
      "training_step_time": 7.600194454193115
    },
    {
      "epoch": 4.336738586425781e-05,
      "model_forward_time": 1.7190039157867432,
      "step": 45474
    },
    {
      "epoch": 4.336738586425781e-05,
      "step": 45474,
      "training_step_time": 8.575504541397095
    },
    {
      "epoch": 4.336833953857422e-05,
      "model_forward_time": 1.7145862579345703,
      "step": 45475
    },
    {
      "epoch": 4.336833953857422e-05,
      "step": 45475,
      "training_step_time": 7.139517545700073
    },
    {
      "epoch": 4.3369293212890624e-05,
      "model_forward_time": 1.7104532718658447,
      "step": 45476
    },
    {
      "epoch": 4.3369293212890624e-05,
      "step": 45476,
      "training_step_time": 7.59488844871521
    },
    {
      "epoch": 4.337024688720703e-05,
      "model_forward_time": 1.7211978435516357,
      "step": 45477
    },
    {
      "epoch": 4.337024688720703e-05,
      "step": 45477,
      "training_step_time": 8.206376552581787
    },
    {
      "epoch": 4.337120056152344e-05,
      "model_forward_time": 1.3571245670318604,
      "step": 45478
    },
    {
      "epoch": 4.337120056152344e-05,
      "step": 45478,
      "training_step_time": 7.644363164901733
    },
    {
      "epoch": 4.337215423583984e-05,
      "model_forward_time": 1.7258703708648682,
      "step": 45479
    },
    {
      "epoch": 4.337215423583984e-05,
      "step": 45479,
      "training_step_time": 9.73148250579834
    },
    {
      "epoch": 4.337310791015625e-05,
      "grad_norm": 0.13033290207386017,
      "learning_rate": 6.1504989922685895e-06,
      "loss": 0.0548,
      "step": 45480
    },
    {
      "epoch": 4.337310791015625e-05,
      "model_forward_time": 1.718837022781372,
      "step": 45480
    },
    {
      "dynamics_loss_avg": 0.05943531543016434,
      "epoch": 4.337310791015625e-05,
      "step": 45480
    },
    {
      "action_loss_avg": 0.0018058509449474514,
      "epoch": 4.337310791015625e-05,
      "step": 45480
    },
    {
      "epoch": 4.337310791015625e-05,
      "step": 45480,
      "training_step_time": 10.384531021118164
    },
    {
      "epoch": 4.337406158447266e-05,
      "model_forward_time": 1.7500746250152588,
      "step": 45481
    },
    {
      "epoch": 4.337406158447266e-05,
      "step": 45481,
      "training_step_time": 7.6000049114227295
    },
    {
      "epoch": 4.337501525878906e-05,
      "model_forward_time": 1.7163386344909668,
      "step": 45482
    },
    {
      "epoch": 4.337501525878906e-05,
      "step": 45482,
      "training_step_time": 8.462459564208984
    },
    {
      "epoch": 4.3375968933105466e-05,
      "model_forward_time": 1.7266967296600342,
      "step": 45483
    },
    {
      "epoch": 4.3375968933105466e-05,
      "step": 45483,
      "training_step_time": 7.588376998901367
    },
    {
      "epoch": 4.337692260742188e-05,
      "model_forward_time": 1.3633484840393066,
      "step": 45484
    },
    {
      "epoch": 4.337692260742188e-05,
      "step": 45484,
      "training_step_time": 7.651180028915405
    },
    {
      "epoch": 4.337787628173828e-05,
      "model_forward_time": 1.7293908596038818,
      "step": 45485
    },
    {
      "epoch": 4.337787628173828e-05,
      "step": 45485,
      "training_step_time": 7.600972414016724
    },
    {
      "epoch": 4.3378829956054685e-05,
      "model_forward_time": 1.731503963470459,
      "step": 45486
    },
    {
      "epoch": 4.3378829956054685e-05,
      "step": 45486,
      "training_step_time": 7.562187433242798
    },
    {
      "epoch": 4.3379783630371096e-05,
      "model_forward_time": 1.7128138542175293,
      "step": 45487
    },
    {
      "epoch": 4.3379783630371096e-05,
      "step": 45487,
      "training_step_time": 7.6896443367004395
    },
    {
      "epoch": 4.33807373046875e-05,
      "model_forward_time": 1.731729507446289,
      "step": 45488
    },
    {
      "epoch": 4.33807373046875e-05,
      "step": 45488,
      "training_step_time": 8.130108118057251
    },
    {
      "epoch": 4.3381690979003904e-05,
      "model_forward_time": 1.7082607746124268,
      "step": 45489
    },
    {
      "epoch": 4.3381690979003904e-05,
      "step": 45489,
      "training_step_time": 8.013670444488525
    },
    {
      "epoch": 4.3382644653320314e-05,
      "grad_norm": 0.1281411200761795,
      "learning_rate": 6.148889827065538e-06,
      "loss": 0.0576,
      "step": 45490
    },
    {
      "epoch": 4.3382644653320314e-05,
      "model_forward_time": 2.2603843212127686,
      "step": 45490
    },
    {
      "dynamics_loss_avg": 0.06591742150485516,
      "epoch": 4.3382644653320314e-05,
      "step": 45490
    },
    {
      "action_loss_avg": 0.001914374198531732,
      "epoch": 4.3382644653320314e-05,
      "step": 45490
    },
    {
      "epoch": 4.3382644653320314e-05,
      "step": 45490,
      "training_step_time": 7.573304653167725
    },
    {
      "epoch": 4.338359832763672e-05,
      "model_forward_time": 1.7202680110931396,
      "step": 45491
    },
    {
      "epoch": 4.338359832763672e-05,
      "step": 45491,
      "training_step_time": 8.643245697021484
    },
    {
      "epoch": 4.338455200195312e-05,
      "model_forward_time": 1.725156307220459,
      "step": 45492
    },
    {
      "epoch": 4.338455200195312e-05,
      "step": 45492,
      "training_step_time": 7.533990144729614
    },
    {
      "epoch": 4.338550567626953e-05,
      "model_forward_time": 1.7309563159942627,
      "step": 45493
    },
    {
      "epoch": 4.338550567626953e-05,
      "step": 45493,
      "training_step_time": 7.619110584259033
    },
    {
      "epoch": 4.338645935058594e-05,
      "model_forward_time": 1.7259573936462402,
      "step": 45494
    },
    {
      "epoch": 4.338645935058594e-05,
      "step": 45494,
      "training_step_time": 7.304300785064697
    },
    {
      "epoch": 4.338741302490234e-05,
      "model_forward_time": 1.721754550933838,
      "step": 45495
    },
    {
      "epoch": 4.338741302490234e-05,
      "step": 45495,
      "training_step_time": 7.56342625617981
    },
    {
      "epoch": 4.338836669921875e-05,
      "model_forward_time": 1.716174840927124,
      "step": 45496
    },
    {
      "epoch": 4.338836669921875e-05,
      "step": 45496,
      "training_step_time": 7.654179096221924
    },
    {
      "epoch": 4.3389320373535156e-05,
      "model_forward_time": 1.7315843105316162,
      "step": 45497
    },
    {
      "epoch": 4.3389320373535156e-05,
      "step": 45497,
      "training_step_time": 9.121036052703857
    },
    {
      "epoch": 4.339027404785156e-05,
      "model_forward_time": 1.7200226783752441,
      "step": 45498
    },
    {
      "epoch": 4.339027404785156e-05,
      "step": 45498,
      "training_step_time": 9.213040351867676
    },
    {
      "epoch": 4.339122772216797e-05,
      "model_forward_time": 2.247037410736084,
      "step": 45499
    },
    {
      "epoch": 4.339122772216797e-05,
      "step": 45499,
      "training_step_time": 7.538071393966675
    },
    {
      "epoch": 4.3392181396484375e-05,
      "grad_norm": 0.3089159429073334,
      "learning_rate": 6.147280536221621e-06,
      "loss": 0.058,
      "step": 45500
    },
    {
      "epoch": 4.3392181396484375e-05,
      "model_forward_time": 1.7646288871765137,
      "step": 45500
    },
    {
      "dynamics_loss_avg": 0.043673469498753545,
      "epoch": 4.3392181396484375e-05,
      "step": 45500
    },
    {
      "action_loss_avg": 0.0009961438365280629,
      "epoch": 4.3392181396484375e-05,
      "step": 45500
    },
    {
      "epoch": 4.3392181396484375e-05,
      "step": 45500,
      "training_step_time": 8.611753463745117
    },
    {
      "epoch": 4.339313507080078e-05,
      "model_forward_time": 1.7318425178527832,
      "step": 45501
    },
    {
      "epoch": 4.339313507080078e-05,
      "step": 45501,
      "training_step_time": 7.620199203491211
    },
    {
      "epoch": 4.339408874511719e-05,
      "model_forward_time": 1.7347862720489502,
      "step": 45502
    },
    {
      "epoch": 4.339408874511719e-05,
      "step": 45502,
      "training_step_time": 7.608250856399536
    },
    {
      "epoch": 4.3395042419433594e-05,
      "model_forward_time": 1.7430896759033203,
      "step": 45503
    },
    {
      "epoch": 4.3395042419433594e-05,
      "step": 45503,
      "training_step_time": 7.2916100025177
    },
    {
      "epoch": 4.339599609375e-05,
      "model_forward_time": 1.7263853549957275,
      "step": 45504
    },
    {
      "epoch": 4.339599609375e-05,
      "step": 45504,
      "training_step_time": 7.62749981880188
    },
    {
      "epoch": 4.339694976806641e-05,
      "model_forward_time": 1.7280917167663574,
      "step": 45505
    },
    {
      "epoch": 4.339694976806641e-05,
      "step": 45505,
      "training_step_time": 8.256174325942993
    },
    {
      "epoch": 4.339790344238281e-05,
      "model_forward_time": 1.7387547492980957,
      "step": 45506
    },
    {
      "epoch": 4.339790344238281e-05,
      "step": 45506,
      "training_step_time": 7.592064142227173
    },
    {
      "epoch": 4.339885711669922e-05,
      "model_forward_time": 1.3705148696899414,
      "step": 45507
    },
    {
      "epoch": 4.339885711669922e-05,
      "step": 45507,
      "training_step_time": 7.489779233932495
    },
    {
      "epoch": 4.339981079101563e-05,
      "model_forward_time": 2.307270050048828,
      "step": 45508
    },
    {
      "epoch": 4.339981079101563e-05,
      "step": 45508,
      "training_step_time": 7.588564157485962
    },
    {
      "epoch": 4.340076446533203e-05,
      "model_forward_time": 1.8023834228515625,
      "step": 45509
    },
    {
      "epoch": 4.340076446533203e-05,
      "step": 45509,
      "training_step_time": 7.635092258453369
    },
    {
      "epoch": 4.3401718139648436e-05,
      "grad_norm": 0.11030647903680801,
      "learning_rate": 6.1456711199128316e-06,
      "loss": 0.0581,
      "step": 45510
    },
    {
      "epoch": 4.3401718139648436e-05,
      "model_forward_time": 1.745499849319458,
      "step": 45510
    },
    {
      "dynamics_loss_avg": 0.044713349733501676,
      "epoch": 4.3401718139648436e-05,
      "step": 45510
    },
    {
      "action_loss_avg": 0.0010693934425944462,
      "epoch": 4.3401718139648436e-05,
      "step": 45510
    },
    {
      "epoch": 4.3401718139648436e-05,
      "step": 45510,
      "training_step_time": 7.568448066711426
    },
    {
      "epoch": 4.3402671813964846e-05,
      "model_forward_time": 1.7429296970367432,
      "step": 45511
    },
    {
      "epoch": 4.3402671813964846e-05,
      "step": 45511,
      "training_step_time": 7.541331768035889
    },
    {
      "epoch": 4.340362548828125e-05,
      "model_forward_time": 1.7434051036834717,
      "step": 45512
    },
    {
      "epoch": 4.340362548828125e-05,
      "step": 45512,
      "training_step_time": 7.35664701461792
    },
    {
      "epoch": 4.3404579162597654e-05,
      "model_forward_time": 1.7501797676086426,
      "step": 45513
    },
    {
      "epoch": 4.3404579162597654e-05,
      "step": 45513,
      "training_step_time": 7.539969444274902
    },
    {
      "epoch": 4.3405532836914065e-05,
      "model_forward_time": 1.7395412921905518,
      "step": 45514
    },
    {
      "epoch": 4.3405532836914065e-05,
      "step": 45514,
      "training_step_time": 8.445022583007812
    },
    {
      "epoch": 4.340648651123047e-05,
      "model_forward_time": 1.7451789379119873,
      "step": 45515
    },
    {
      "epoch": 4.340648651123047e-05,
      "step": 45515,
      "training_step_time": 7.644697427749634
    },
    {
      "epoch": 4.340744018554687e-05,
      "model_forward_time": 1.754833698272705,
      "step": 45516
    },
    {
      "epoch": 4.340744018554687e-05,
      "step": 45516,
      "training_step_time": 8.05129361152649
    },
    {
      "epoch": 4.3408393859863284e-05,
      "model_forward_time": 1.956831455230713,
      "step": 45517
    },
    {
      "epoch": 4.3408393859863284e-05,
      "step": 45517,
      "training_step_time": 7.533144235610962
    },
    {
      "epoch": 4.340934753417969e-05,
      "model_forward_time": 1.8113806247711182,
      "step": 45518
    },
    {
      "epoch": 4.340934753417969e-05,
      "step": 45518,
      "training_step_time": 7.647407054901123
    },
    {
      "epoch": 4.341030120849609e-05,
      "model_forward_time": 1.7554171085357666,
      "step": 45519
    },
    {
      "epoch": 4.341030120849609e-05,
      "step": 45519,
      "training_step_time": 7.533185958862305
    },
    {
      "epoch": 4.34112548828125e-05,
      "grad_norm": 0.1314617246389389,
      "learning_rate": 6.1440615783151695e-06,
      "loss": 0.0588,
      "step": 45520
    },
    {
      "epoch": 4.34112548828125e-05,
      "model_forward_time": 1.7587339878082275,
      "step": 45520
    },
    {
      "dynamics_loss_avg": 0.05257413424551487,
      "epoch": 4.34112548828125e-05,
      "step": 45520
    },
    {
      "action_loss_avg": 0.001381174218840897,
      "epoch": 4.34112548828125e-05,
      "step": 45520
    },
    {
      "epoch": 4.34112548828125e-05,
      "step": 45520,
      "training_step_time": 7.28360652923584
    },
    {
      "epoch": 4.341220855712891e-05,
      "model_forward_time": 1.7793724536895752,
      "step": 45521
    },
    {
      "epoch": 4.341220855712891e-05,
      "step": 45521,
      "training_step_time": 7.636066198348999
    },
    {
      "epoch": 4.341316223144531e-05,
      "model_forward_time": 1.7484087944030762,
      "step": 45522
    },
    {
      "epoch": 4.341316223144531e-05,
      "step": 45522,
      "training_step_time": 7.578704357147217
    },
    {
      "epoch": 4.341411590576172e-05,
      "model_forward_time": 1.7463088035583496,
      "step": 45523
    },
    {
      "epoch": 4.341411590576172e-05,
      "step": 45523,
      "training_step_time": 7.589722156524658
    },
    {
      "epoch": 4.3415069580078126e-05,
      "model_forward_time": 1.748358964920044,
      "step": 45524
    },
    {
      "epoch": 4.3415069580078126e-05,
      "step": 45524,
      "training_step_time": 8.73672366142273
    },
    {
      "epoch": 4.341602325439453e-05,
      "model_forward_time": 1.7382709980010986,
      "step": 45525
    },
    {
      "epoch": 4.341602325439453e-05,
      "step": 45525,
      "training_step_time": 7.665584325790405
    },
    {
      "epoch": 4.341697692871094e-05,
      "model_forward_time": 2.2999603748321533,
      "step": 45526
    },
    {
      "epoch": 4.341697692871094e-05,
      "step": 45526,
      "training_step_time": 8.65891170501709
    },
    {
      "epoch": 4.3417930603027345e-05,
      "model_forward_time": 1.7649414539337158,
      "step": 45527
    },
    {
      "epoch": 4.3417930603027345e-05,
      "step": 45527,
      "training_step_time": 9.427042961120605
    },
    {
      "epoch": 4.341888427734375e-05,
      "model_forward_time": 1.7473609447479248,
      "step": 45528
    },
    {
      "epoch": 4.341888427734375e-05,
      "step": 45528,
      "training_step_time": 7.6506431102752686
    },
    {
      "epoch": 4.341983795166016e-05,
      "model_forward_time": 1.7359211444854736,
      "step": 45529
    },
    {
      "epoch": 4.341983795166016e-05,
      "step": 45529,
      "training_step_time": 8.57601284980774
    },
    {
      "epoch": 4.3420791625976564e-05,
      "grad_norm": 0.1288427859544754,
      "learning_rate": 6.142451911604654e-06,
      "loss": 0.0603,
      "step": 45530
    },
    {
      "epoch": 4.3420791625976564e-05,
      "model_forward_time": 1.7342875003814697,
      "step": 45530
    },
    {
      "dynamics_loss_avg": 0.06775845848023891,
      "epoch": 4.3420791625976564e-05,
      "step": 45530
    },
    {
      "action_loss_avg": 0.0026446566567756235,
      "epoch": 4.3420791625976564e-05,
      "step": 45530
    },
    {
      "epoch": 4.3420791625976564e-05,
      "step": 45530,
      "training_step_time": 7.632390737533569
    },
    {
      "epoch": 4.342174530029297e-05,
      "model_forward_time": 1.7383692264556885,
      "step": 45531
    },
    {
      "epoch": 4.342174530029297e-05,
      "step": 45531,
      "training_step_time": 7.662368535995483
    },
    {
      "epoch": 4.342269897460937e-05,
      "model_forward_time": 1.7371587753295898,
      "step": 45532
    },
    {
      "epoch": 4.342269897460937e-05,
      "step": 45532,
      "training_step_time": 7.714299917221069
    },
    {
      "epoch": 4.342365264892578e-05,
      "model_forward_time": 1.7300450801849365,
      "step": 45533
    },
    {
      "epoch": 4.342365264892578e-05,
      "step": 45533,
      "training_step_time": 8.297038555145264
    },
    {
      "epoch": 4.3424606323242186e-05,
      "model_forward_time": 1.7442259788513184,
      "step": 45534
    },
    {
      "epoch": 4.3424606323242186e-05,
      "step": 45534,
      "training_step_time": 7.669934988021851
    },
    {
      "epoch": 4.342555999755859e-05,
      "model_forward_time": 1.7343859672546387,
      "step": 45535
    },
    {
      "epoch": 4.342555999755859e-05,
      "step": 45535,
      "training_step_time": 9.002058029174805
    },
    {
      "epoch": 4.3426513671875e-05,
      "model_forward_time": 2.28896427154541,
      "step": 45536
    },
    {
      "epoch": 4.3426513671875e-05,
      "step": 45536,
      "training_step_time": 8.976153135299683
    },
    {
      "epoch": 4.3427467346191405e-05,
      "model_forward_time": 1.7449018955230713,
      "step": 45537
    },
    {
      "epoch": 4.3427467346191405e-05,
      "step": 45537,
      "training_step_time": 7.621935606002808
    },
    {
      "epoch": 4.342842102050781e-05,
      "model_forward_time": 1.7406585216522217,
      "step": 45538
    },
    {
      "epoch": 4.342842102050781e-05,
      "step": 45538,
      "training_step_time": 7.714615821838379
    },
    {
      "epoch": 4.342937469482422e-05,
      "model_forward_time": 1.748781442642212,
      "step": 45539
    },
    {
      "epoch": 4.342937469482422e-05,
      "step": 45539,
      "training_step_time": 8.483154058456421
    },
    {
      "epoch": 4.3430328369140624e-05,
      "grad_norm": 0.11974900960922241,
      "learning_rate": 6.1408421199573155e-06,
      "loss": 0.0613,
      "step": 45540
    },
    {
      "epoch": 4.3430328369140624e-05,
      "model_forward_time": 1.739593267440796,
      "step": 45540
    },
    {
      "dynamics_loss_avg": 0.049639513902366164,
      "epoch": 4.3430328369140624e-05,
      "step": 45540
    },
    {
      "action_loss_avg": 0.0016986851318506525,
      "epoch": 4.3430328369140624e-05,
      "step": 45540
    },
    {
      "epoch": 4.3430328369140624e-05,
      "step": 45540,
      "training_step_time": 7.643527030944824
    },
    {
      "epoch": 4.343128204345703e-05,
      "model_forward_time": 1.7440996170043945,
      "step": 45541
    },
    {
      "epoch": 4.343128204345703e-05,
      "step": 45541,
      "training_step_time": 8.711841583251953
    },
    {
      "epoch": 4.343223571777344e-05,
      "model_forward_time": 1.738506555557251,
      "step": 45542
    },
    {
      "epoch": 4.343223571777344e-05,
      "step": 45542,
      "training_step_time": 8.343999862670898
    },
    {
      "epoch": 4.343318939208984e-05,
      "model_forward_time": 1.7424075603485107,
      "step": 45543
    },
    {
      "epoch": 4.343318939208984e-05,
      "step": 45543,
      "training_step_time": 9.495654582977295
    },
    {
      "epoch": 4.343414306640625e-05,
      "model_forward_time": 1.733017921447754,
      "step": 45544
    },
    {
      "epoch": 4.343414306640625e-05,
      "step": 45544,
      "training_step_time": 9.166014671325684
    },
    {
      "epoch": 4.343509674072266e-05,
      "model_forward_time": 2.2833616733551025,
      "step": 45545
    },
    {
      "epoch": 4.343509674072266e-05,
      "step": 45545,
      "training_step_time": 7.6740641593933105
    },
    {
      "epoch": 4.343605041503906e-05,
      "model_forward_time": 1.7850556373596191,
      "step": 45546
    },
    {
      "epoch": 4.343605041503906e-05,
      "step": 45546,
      "training_step_time": 9.518565654754639
    },
    {
      "epoch": 4.3437004089355466e-05,
      "model_forward_time": 1.7886505126953125,
      "step": 45547
    },
    {
      "epoch": 4.3437004089355466e-05,
      "step": 45547,
      "training_step_time": 7.558004856109619
    },
    {
      "epoch": 4.343795776367188e-05,
      "model_forward_time": 1.7414112091064453,
      "step": 45548
    },
    {
      "epoch": 4.343795776367188e-05,
      "step": 45548,
      "training_step_time": 7.41853141784668
    },
    {
      "epoch": 4.343891143798828e-05,
      "model_forward_time": 1.7455353736877441,
      "step": 45549
    },
    {
      "epoch": 4.343891143798828e-05,
      "step": 45549,
      "training_step_time": 8.187506675720215
    },
    {
      "epoch": 4.3439865112304685e-05,
      "grad_norm": 0.11981835216283798,
      "learning_rate": 6.139232203549197e-06,
      "loss": 0.0601,
      "step": 45550
    },
    {
      "epoch": 4.3439865112304685e-05,
      "model_forward_time": 1.7474303245544434,
      "step": 45550
    },
    {
      "dynamics_loss_avg": 0.0512368518859148,
      "epoch": 4.3439865112304685e-05,
      "step": 45550
    },
    {
      "action_loss_avg": 0.0017896332603413612,
      "epoch": 4.3439865112304685e-05,
      "step": 45550
    },
    {
      "epoch": 4.3439865112304685e-05,
      "step": 45550,
      "training_step_time": 8.433018684387207
    },
    {
      "epoch": 4.3440818786621096e-05,
      "model_forward_time": 1.7184998989105225,
      "step": 45551
    },
    {
      "epoch": 4.3440818786621096e-05,
      "step": 45551,
      "training_step_time": 9.054213047027588
    },
    {
      "epoch": 4.34417724609375e-05,
      "model_forward_time": 1.7455670833587646,
      "step": 45552
    },
    {
      "epoch": 4.34417724609375e-05,
      "step": 45552,
      "training_step_time": 9.116116762161255
    },
    {
      "epoch": 4.3442726135253904e-05,
      "model_forward_time": 1.728332281112671,
      "step": 45553
    },
    {
      "epoch": 4.3442726135253904e-05,
      "step": 45553,
      "training_step_time": 7.599451780319214
    },
    {
      "epoch": 4.3443679809570314e-05,
      "model_forward_time": 2.304577589035034,
      "step": 45554
    },
    {
      "epoch": 4.3443679809570314e-05,
      "step": 45554,
      "training_step_time": 8.57003140449524
    },
    {
      "epoch": 4.344463348388672e-05,
      "model_forward_time": 1.7716572284698486,
      "step": 45555
    },
    {
      "epoch": 4.344463348388672e-05,
      "step": 45555,
      "training_step_time": 7.58219575881958
    },
    {
      "epoch": 4.344558715820312e-05,
      "model_forward_time": 1.7211496829986572,
      "step": 45556
    },
    {
      "epoch": 4.344558715820312e-05,
      "step": 45556,
      "training_step_time": 7.664247035980225
    },
    {
      "epoch": 4.344654083251953e-05,
      "model_forward_time": 1.7439682483673096,
      "step": 45557
    },
    {
      "epoch": 4.344654083251953e-05,
      "step": 45557,
      "training_step_time": 7.631934881210327
    },
    {
      "epoch": 4.344749450683594e-05,
      "model_forward_time": 1.7365083694458008,
      "step": 45558
    },
    {
      "epoch": 4.344749450683594e-05,
      "step": 45558,
      "training_step_time": 7.753638744354248
    },
    {
      "epoch": 4.344844818115234e-05,
      "model_forward_time": 1.7365663051605225,
      "step": 45559
    },
    {
      "epoch": 4.344844818115234e-05,
      "step": 45559,
      "training_step_time": 9.102770566940308
    },
    {
      "epoch": 4.344940185546875e-05,
      "grad_norm": 0.12196572870016098,
      "learning_rate": 6.137622162556357e-06,
      "loss": 0.0577,
      "step": 45560
    },
    {
      "epoch": 4.344940185546875e-05,
      "model_forward_time": 1.7138829231262207,
      "step": 45560
    },
    {
      "dynamics_loss_avg": 0.043830201216042045,
      "epoch": 4.344940185546875e-05,
      "step": 45560
    },
    {
      "action_loss_avg": 0.0011543051456101238,
      "epoch": 4.344940185546875e-05,
      "step": 45560
    },
    {
      "epoch": 4.344940185546875e-05,
      "step": 45560,
      "training_step_time": 8.330397129058838
    },
    {
      "epoch": 4.3450355529785156e-05,
      "model_forward_time": 1.725146770477295,
      "step": 45561
    },
    {
      "epoch": 4.3450355529785156e-05,
      "step": 45561,
      "training_step_time": 9.035842657089233
    },
    {
      "epoch": 4.345130920410156e-05,
      "model_forward_time": 1.7419440746307373,
      "step": 45562
    },
    {
      "epoch": 4.345130920410156e-05,
      "step": 45562,
      "training_step_time": 9.199356317520142
    },
    {
      "epoch": 4.345226287841797e-05,
      "model_forward_time": 2.2903993129730225,
      "step": 45563
    },
    {
      "epoch": 4.345226287841797e-05,
      "step": 45563,
      "training_step_time": 9.1881844997406
    },
    {
      "epoch": 4.3453216552734375e-05,
      "model_forward_time": 1.7396821975708008,
      "step": 45564
    },
    {
      "epoch": 4.3453216552734375e-05,
      "step": 45564,
      "training_step_time": 7.591956377029419
    },
    {
      "epoch": 4.345417022705078e-05,
      "model_forward_time": 1.725966453552246,
      "step": 45565
    },
    {
      "epoch": 4.345417022705078e-05,
      "step": 45565,
      "training_step_time": 8.708176136016846
    },
    {
      "epoch": 4.345512390136719e-05,
      "model_forward_time": 1.727736473083496,
      "step": 45566
    },
    {
      "epoch": 4.345512390136719e-05,
      "step": 45566,
      "training_step_time": 8.161589860916138
    },
    {
      "epoch": 4.3456077575683594e-05,
      "model_forward_time": 1.7228646278381348,
      "step": 45567
    },
    {
      "epoch": 4.3456077575683594e-05,
      "step": 45567,
      "training_step_time": 7.751331567764282
    },
    {
      "epoch": 4.345703125e-05,
      "model_forward_time": 1.7250785827636719,
      "step": 45568
    },
    {
      "epoch": 4.345703125e-05,
      "step": 45568,
      "training_step_time": 7.787487745285034
    },
    {
      "epoch": 4.345798492431641e-05,
      "model_forward_time": 1.728989601135254,
      "step": 45569
    },
    {
      "epoch": 4.345798492431641e-05,
      "step": 45569,
      "training_step_time": 9.077548027038574
    },
    {
      "epoch": 4.345893859863281e-05,
      "grad_norm": 0.16883307695388794,
      "learning_rate": 6.136011997154869e-06,
      "loss": 0.0591,
      "step": 45570
    },
    {
      "epoch": 4.345893859863281e-05,
      "model_forward_time": 1.7253823280334473,
      "step": 45570
    },
    {
      "dynamics_loss_avg": 0.05298373717814684,
      "epoch": 4.345893859863281e-05,
      "step": 45570
    },
    {
      "action_loss_avg": 0.0013293535506818444,
      "epoch": 4.345893859863281e-05,
      "step": 45570
    },
    {
      "epoch": 4.345893859863281e-05,
      "step": 45570,
      "training_step_time": 7.652894020080566
    },
    {
      "epoch": 4.345989227294922e-05,
      "model_forward_time": 1.7262394428253174,
      "step": 45571
    },
    {
      "epoch": 4.345989227294922e-05,
      "step": 45571,
      "training_step_time": 8.361396312713623
    },
    {
      "epoch": 4.346084594726563e-05,
      "model_forward_time": 2.2985548973083496,
      "step": 45572
    },
    {
      "epoch": 4.346084594726563e-05,
      "step": 45572,
      "training_step_time": 7.645416498184204
    },
    {
      "epoch": 4.346179962158203e-05,
      "model_forward_time": 1.720921516418457,
      "step": 45573
    },
    {
      "epoch": 4.346179962158203e-05,
      "step": 45573,
      "training_step_time": 7.7016942501068115
    },
    {
      "epoch": 4.3462753295898436e-05,
      "model_forward_time": 1.7806024551391602,
      "step": 45574
    },
    {
      "epoch": 4.3462753295898436e-05,
      "step": 45574,
      "training_step_time": 7.639831304550171
    },
    {
      "epoch": 4.3463706970214846e-05,
      "model_forward_time": 1.733323335647583,
      "step": 45575
    },
    {
      "epoch": 4.3463706970214846e-05,
      "step": 45575,
      "training_step_time": 7.681039094924927
    },
    {
      "epoch": 4.346466064453125e-05,
      "model_forward_time": 1.7206130027770996,
      "step": 45576
    },
    {
      "epoch": 4.346466064453125e-05,
      "step": 45576,
      "training_step_time": 7.658088207244873
    },
    {
      "epoch": 4.3465614318847655e-05,
      "model_forward_time": 1.7261114120483398,
      "step": 45577
    },
    {
      "epoch": 4.3465614318847655e-05,
      "step": 45577,
      "training_step_time": 8.135983943939209
    },
    {
      "epoch": 4.3466567993164065e-05,
      "model_forward_time": 1.7225251197814941,
      "step": 45578
    },
    {
      "epoch": 4.3466567993164065e-05,
      "step": 45578,
      "training_step_time": 8.464413404464722
    },
    {
      "epoch": 4.346752166748047e-05,
      "model_forward_time": 1.3568565845489502,
      "step": 45579
    },
    {
      "epoch": 4.346752166748047e-05,
      "step": 45579,
      "training_step_time": 8.713245391845703
    },
    {
      "epoch": 4.346847534179687e-05,
      "grad_norm": 0.18857410550117493,
      "learning_rate": 6.1344017075208164e-06,
      "loss": 0.0585,
      "step": 45580
    },
    {
      "epoch": 4.346847534179687e-05,
      "model_forward_time": 1.7265350818634033,
      "step": 45580
    },
    {
      "dynamics_loss_avg": 0.055583633854985236,
      "epoch": 4.346847534179687e-05,
      "step": 45580
    },
    {
      "action_loss_avg": 0.0015900009457254782,
      "epoch": 4.346847534179687e-05,
      "step": 45580
    },
    {
      "epoch": 4.346847534179687e-05,
      "step": 45580,
      "training_step_time": 7.626373529434204
    },
    {
      "epoch": 4.3469429016113284e-05,
      "model_forward_time": 1.7260651588439941,
      "step": 45581
    },
    {
      "epoch": 4.3469429016113284e-05,
      "step": 45581,
      "training_step_time": 9.15078091621399
    },
    {
      "epoch": 4.347038269042969e-05,
      "model_forward_time": 2.2816033363342285,
      "step": 45582
    },
    {
      "epoch": 4.347038269042969e-05,
      "step": 45582,
      "training_step_time": 7.730289459228516
    },
    {
      "epoch": 4.347133636474609e-05,
      "model_forward_time": 1.763096570968628,
      "step": 45583
    },
    {
      "epoch": 4.347133636474609e-05,
      "step": 45583,
      "training_step_time": 7.51956033706665
    },
    {
      "epoch": 4.34722900390625e-05,
      "model_forward_time": 1.734525203704834,
      "step": 45584
    },
    {
      "epoch": 4.34722900390625e-05,
      "step": 45584,
      "training_step_time": 7.70551061630249
    },
    {
      "epoch": 4.347324371337891e-05,
      "model_forward_time": 1.7249126434326172,
      "step": 45585
    },
    {
      "epoch": 4.347324371337891e-05,
      "step": 45585,
      "training_step_time": 7.792266845703125
    },
    {
      "epoch": 4.347419738769531e-05,
      "model_forward_time": 1.7325439453125,
      "step": 45586
    },
    {
      "epoch": 4.347419738769531e-05,
      "step": 45586,
      "training_step_time": 7.741799354553223
    },
    {
      "epoch": 4.347515106201172e-05,
      "model_forward_time": 1.7356269359588623,
      "step": 45587
    },
    {
      "epoch": 4.347515106201172e-05,
      "step": 45587,
      "training_step_time": 7.731978178024292
    },
    {
      "epoch": 4.3476104736328126e-05,
      "model_forward_time": 1.7408921718597412,
      "step": 45588
    },
    {
      "epoch": 4.3476104736328126e-05,
      "step": 45588,
      "training_step_time": 7.767478942871094
    },
    {
      "epoch": 4.347705841064453e-05,
      "model_forward_time": 1.7228176593780518,
      "step": 45589
    },
    {
      "epoch": 4.347705841064453e-05,
      "step": 45589,
      "training_step_time": 7.722075462341309
    },
    {
      "epoch": 4.347801208496094e-05,
      "grad_norm": 0.12132488936185837,
      "learning_rate": 6.132791293830298e-06,
      "loss": 0.0601,
      "step": 45590
    },
    {
      "epoch": 4.347801208496094e-05,
      "model_forward_time": 1.7388153076171875,
      "step": 45590
    },
    {
      "dynamics_loss_avg": 0.05968433963134885,
      "epoch": 4.347801208496094e-05,
      "step": 45590
    },
    {
      "action_loss_avg": 0.0017638469755183906,
      "epoch": 4.347801208496094e-05,
      "step": 45590
    },
    {
      "epoch": 4.347801208496094e-05,
      "step": 45590,
      "training_step_time": 8.398475408554077
    },
    {
      "epoch": 4.3478965759277345e-05,
      "model_forward_time": 1.8978230953216553,
      "step": 45591
    },
    {
      "epoch": 4.3478965759277345e-05,
      "step": 45591,
      "training_step_time": 8.687846183776855
    },
    {
      "epoch": 4.347991943359375e-05,
      "model_forward_time": 1.7716617584228516,
      "step": 45592
    },
    {
      "epoch": 4.347991943359375e-05,
      "step": 45592,
      "training_step_time": 8.989139080047607
    },
    {
      "epoch": 4.348087310791016e-05,
      "model_forward_time": 1.734121561050415,
      "step": 45593
    },
    {
      "epoch": 4.348087310791016e-05,
      "step": 45593,
      "training_step_time": 7.693888902664185
    },
    {
      "epoch": 4.3481826782226564e-05,
      "model_forward_time": 1.7320940494537354,
      "step": 45594
    },
    {
      "epoch": 4.3481826782226564e-05,
      "step": 45594,
      "training_step_time": 7.8193442821502686
    },
    {
      "epoch": 4.348278045654297e-05,
      "model_forward_time": 1.7239246368408203,
      "step": 45595
    },
    {
      "epoch": 4.348278045654297e-05,
      "step": 45595,
      "training_step_time": 7.7516772747039795
    },
    {
      "epoch": 4.348373413085937e-05,
      "model_forward_time": 1.7272353172302246,
      "step": 45596
    },
    {
      "epoch": 4.348373413085937e-05,
      "step": 45596,
      "training_step_time": 7.687081336975098
    },
    {
      "epoch": 4.348468780517578e-05,
      "model_forward_time": 1.3555481433868408,
      "step": 45597
    },
    {
      "epoch": 4.348468780517578e-05,
      "step": 45597,
      "training_step_time": 8.698452711105347
    },
    {
      "epoch": 4.3485641479492187e-05,
      "model_forward_time": 1.728393316268921,
      "step": 45598
    },
    {
      "epoch": 4.3485641479492187e-05,
      "step": 45598,
      "training_step_time": 7.713423490524292
    },
    {
      "epoch": 4.348659515380859e-05,
      "model_forward_time": 1.7293758392333984,
      "step": 45599
    },
    {
      "epoch": 4.348659515380859e-05,
      "step": 45599,
      "training_step_time": 8.646122217178345
    },
    {
      "epoch": 4.3487548828125e-05,
      "grad_norm": 0.16703198850154877,
      "learning_rate": 6.131180756259428e-06,
      "loss": 0.0574,
      "step": 45600
    },
    {
      "epoch": 4.3487548828125e-05,
      "model_forward_time": 2.285587787628174,
      "step": 45600
    },
    {
      "dynamics_loss_avg": 0.054290704801678656,
      "epoch": 4.3487548828125e-05,
      "step": 45600
    },
    {
      "action_loss_avg": 0.002925921173300594,
      "epoch": 4.3487548828125e-05,
      "step": 45600
    },
    {
      "epoch": 4.3487548828125e-05,
      "step": 45600,
      "training_step_time": 7.752731800079346
    },
    {
      "epoch": 4.3488502502441405e-05,
      "model_forward_time": 1.7617716789245605,
      "step": 45601
    },
    {
      "epoch": 4.3488502502441405e-05,
      "step": 45601,
      "training_step_time": 8.494632482528687
    },
    {
      "epoch": 4.348945617675781e-05,
      "model_forward_time": 1.732762098312378,
      "step": 45602
    },
    {
      "epoch": 4.348945617675781e-05,
      "step": 45602,
      "training_step_time": 7.779219627380371
    },
    {
      "epoch": 4.349040985107422e-05,
      "model_forward_time": 1.7332556247711182,
      "step": 45603
    },
    {
      "epoch": 4.349040985107422e-05,
      "step": 45603,
      "training_step_time": 7.731672286987305
    },
    {
      "epoch": 4.3491363525390624e-05,
      "model_forward_time": 1.7340266704559326,
      "step": 45604
    },
    {
      "epoch": 4.3491363525390624e-05,
      "step": 45604,
      "training_step_time": 7.707567930221558
    },
    {
      "epoch": 4.349231719970703e-05,
      "model_forward_time": 1.7199804782867432,
      "step": 45605
    },
    {
      "epoch": 4.349231719970703e-05,
      "step": 45605,
      "training_step_time": 7.67084527015686
    },
    {
      "epoch": 4.349327087402344e-05,
      "model_forward_time": 1.733827829360962,
      "step": 45606
    },
    {
      "epoch": 4.349327087402344e-05,
      "step": 45606,
      "training_step_time": 7.633390426635742
    },
    {
      "epoch": 4.349422454833984e-05,
      "model_forward_time": 1.7281544208526611,
      "step": 45607
    },
    {
      "epoch": 4.349422454833984e-05,
      "step": 45607,
      "training_step_time": 7.671483039855957
    },
    {
      "epoch": 4.349517822265625e-05,
      "model_forward_time": 1.7299013137817383,
      "step": 45608
    },
    {
      "epoch": 4.349517822265625e-05,
      "step": 45608,
      "training_step_time": 8.665012836456299
    },
    {
      "epoch": 4.349613189697266e-05,
      "model_forward_time": 2.272019624710083,
      "step": 45609
    },
    {
      "epoch": 4.349613189697266e-05,
      "step": 45609,
      "training_step_time": 7.738198280334473
    },
    {
      "epoch": 4.349708557128906e-05,
      "grad_norm": 0.164231076836586,
      "learning_rate": 6.129570094984331e-06,
      "loss": 0.0528,
      "step": 45610
    },
    {
      "epoch": 4.349708557128906e-05,
      "model_forward_time": 1.3730156421661377,
      "step": 45610
    },
    {
      "dynamics_loss_avg": 0.057868664525449275,
      "epoch": 4.349708557128906e-05,
      "step": 45610
    },
    {
      "action_loss_avg": 0.0025157032010611146,
      "epoch": 4.349708557128906e-05,
      "step": 45610
    },
    {
      "epoch": 4.349708557128906e-05,
      "step": 45610,
      "training_step_time": 8.253173828125
    },
    {
      "epoch": 4.3498039245605466e-05,
      "model_forward_time": 1.735518217086792,
      "step": 45611
    },
    {
      "epoch": 4.3498039245605466e-05,
      "step": 45611,
      "training_step_time": 7.709104061126709
    },
    {
      "epoch": 4.349899291992188e-05,
      "model_forward_time": 1.730494737625122,
      "step": 45612
    },
    {
      "epoch": 4.349899291992188e-05,
      "step": 45612,
      "training_step_time": 7.694751262664795
    },
    {
      "epoch": 4.349994659423828e-05,
      "model_forward_time": 1.7316625118255615,
      "step": 45613
    },
    {
      "epoch": 4.349994659423828e-05,
      "step": 45613,
      "training_step_time": 7.658029079437256
    },
    {
      "epoch": 4.3500900268554685e-05,
      "model_forward_time": 1.7358520030975342,
      "step": 45614
    },
    {
      "epoch": 4.3500900268554685e-05,
      "step": 45614,
      "training_step_time": 7.649960279464722
    },
    {
      "epoch": 4.3501853942871096e-05,
      "model_forward_time": 1.7279255390167236,
      "step": 45615
    },
    {
      "epoch": 4.3501853942871096e-05,
      "step": 45615,
      "training_step_time": 7.671969175338745
    },
    {
      "epoch": 4.35028076171875e-05,
      "model_forward_time": 1.735229730606079,
      "step": 45616
    },
    {
      "epoch": 4.35028076171875e-05,
      "step": 45616,
      "training_step_time": 7.717010736465454
    },
    {
      "epoch": 4.3503761291503904e-05,
      "model_forward_time": 1.7362282276153564,
      "step": 45617
    },
    {
      "epoch": 4.3503761291503904e-05,
      "step": 45617,
      "training_step_time": 7.6764044761657715
    },
    {
      "epoch": 4.3504714965820315e-05,
      "model_forward_time": 1.73345947265625,
      "step": 45618
    },
    {
      "epoch": 4.3504714965820315e-05,
      "step": 45618,
      "training_step_time": 7.654062747955322
    },
    {
      "epoch": 4.350566864013672e-05,
      "model_forward_time": 2.2860593795776367,
      "step": 45619
    },
    {
      "epoch": 4.350566864013672e-05,
      "step": 45619,
      "training_step_time": 8.359122276306152
    },
    {
      "epoch": 4.350662231445312e-05,
      "grad_norm": 0.12044654041528702,
      "learning_rate": 6.127959310181145e-06,
      "loss": 0.0563,
      "step": 45620
    },
    {
      "epoch": 4.350662231445312e-05,
      "model_forward_time": 1.7291676998138428,
      "step": 45620
    },
    {
      "dynamics_loss_avg": 0.041453552059829236,
      "epoch": 4.350662231445312e-05,
      "step": 45620
    },
    {
      "action_loss_avg": 0.0012760190933477134,
      "epoch": 4.350662231445312e-05,
      "step": 45620
    },
    {
      "epoch": 4.350662231445312e-05,
      "step": 45620,
      "training_step_time": 7.736906290054321
    },
    {
      "epoch": 4.350757598876953e-05,
      "model_forward_time": 1.718015193939209,
      "step": 45621
    },
    {
      "epoch": 4.350757598876953e-05,
      "step": 45621,
      "training_step_time": 7.78612494468689
    },
    {
      "epoch": 4.350852966308594e-05,
      "model_forward_time": 1.7268435955047607,
      "step": 45622
    },
    {
      "epoch": 4.350852966308594e-05,
      "step": 45622,
      "training_step_time": 7.725545644760132
    },
    {
      "epoch": 4.350948333740234e-05,
      "model_forward_time": 1.7247333526611328,
      "step": 45623
    },
    {
      "epoch": 4.350948333740234e-05,
      "step": 45623,
      "training_step_time": 8.01736068725586
    },
    {
      "epoch": 4.351043701171875e-05,
      "model_forward_time": 1.3527894020080566,
      "step": 45624
    },
    {
      "epoch": 4.351043701171875e-05,
      "step": 45624,
      "training_step_time": 8.650447845458984
    },
    {
      "epoch": 4.3511390686035156e-05,
      "model_forward_time": 1.7282466888427734,
      "step": 45625
    },
    {
      "epoch": 4.3511390686035156e-05,
      "step": 45625,
      "training_step_time": 8.08369493484497
    },
    {
      "epoch": 4.351234436035156e-05,
      "model_forward_time": 1.734584093093872,
      "step": 45626
    },
    {
      "epoch": 4.351234436035156e-05,
      "step": 45626,
      "training_step_time": 9.33278751373291
    },
    {
      "epoch": 4.351329803466797e-05,
      "model_forward_time": 1.734121322631836,
      "step": 45627
    },
    {
      "epoch": 4.351329803466797e-05,
      "step": 45627,
      "training_step_time": 8.175192832946777
    },
    {
      "epoch": 4.3514251708984375e-05,
      "model_forward_time": 2.2870001792907715,
      "step": 45628
    },
    {
      "epoch": 4.3514251708984375e-05,
      "step": 45628,
      "training_step_time": 7.662680387496948
    },
    {
      "epoch": 4.351520538330078e-05,
      "model_forward_time": 1.7701027393341064,
      "step": 45629
    },
    {
      "epoch": 4.351520538330078e-05,
      "step": 45629,
      "training_step_time": 7.7108049392700195
    },
    {
      "epoch": 4.351615905761719e-05,
      "grad_norm": 0.13579341769218445,
      "learning_rate": 6.126348402026027e-06,
      "loss": 0.058,
      "step": 45630
    },
    {
      "epoch": 4.351615905761719e-05,
      "model_forward_time": 1.7319798469543457,
      "step": 45630
    },
    {
      "dynamics_loss_avg": 0.05273912083357572,
      "epoch": 4.351615905761719e-05,
      "step": 45630
    },
    {
      "action_loss_avg": 0.0011847378918901086,
      "epoch": 4.351615905761719e-05,
      "step": 45630
    },
    {
      "epoch": 4.351615905761719e-05,
      "step": 45630,
      "training_step_time": 7.728288888931274
    },
    {
      "epoch": 4.3517112731933594e-05,
      "model_forward_time": 1.7726218700408936,
      "step": 45631
    },
    {
      "epoch": 4.3517112731933594e-05,
      "step": 45631,
      "training_step_time": 8.44434928894043
    },
    {
      "epoch": 4.351806640625e-05,
      "model_forward_time": 1.7292983531951904,
      "step": 45632
    },
    {
      "epoch": 4.351806640625e-05,
      "step": 45632,
      "training_step_time": 8.574769020080566
    },
    {
      "epoch": 4.351902008056641e-05,
      "model_forward_time": 1.7333645820617676,
      "step": 45633
    },
    {
      "epoch": 4.351902008056641e-05,
      "step": 45633,
      "training_step_time": 7.672348737716675
    },
    {
      "epoch": 4.351997375488281e-05,
      "model_forward_time": 1.714327335357666,
      "step": 45634
    },
    {
      "epoch": 4.351997375488281e-05,
      "step": 45634,
      "training_step_time": 8.258161067962646
    },
    {
      "epoch": 4.352092742919922e-05,
      "model_forward_time": 1.730349063873291,
      "step": 45635
    },
    {
      "epoch": 4.352092742919922e-05,
      "step": 45635,
      "training_step_time": 8.304171800613403
    },
    {
      "epoch": 4.352188110351563e-05,
      "model_forward_time": 1.7327382564544678,
      "step": 45636
    },
    {
      "epoch": 4.352188110351563e-05,
      "step": 45636,
      "training_step_time": 8.613674879074097
    },
    {
      "epoch": 4.352283477783203e-05,
      "model_forward_time": 2.284130334854126,
      "step": 45637
    },
    {
      "epoch": 4.352283477783203e-05,
      "step": 45637,
      "training_step_time": 8.51510763168335
    },
    {
      "epoch": 4.3523788452148436e-05,
      "model_forward_time": 1.7442164421081543,
      "step": 45638
    },
    {
      "epoch": 4.3523788452148436e-05,
      "step": 45638,
      "training_step_time": 9.73299241065979
    },
    {
      "epoch": 4.3524742126464847e-05,
      "model_forward_time": 1.7289061546325684,
      "step": 45639
    },
    {
      "epoch": 4.3524742126464847e-05,
      "step": 45639,
      "training_step_time": 7.785767555236816
    },
    {
      "epoch": 4.352569580078125e-05,
      "grad_norm": 0.10358361154794693,
      "learning_rate": 6.124737370695141e-06,
      "loss": 0.0545,
      "step": 45640
    },
    {
      "epoch": 4.352569580078125e-05,
      "model_forward_time": 1.7484533786773682,
      "step": 45640
    },
    {
      "dynamics_loss_avg": 0.06075358223170042,
      "epoch": 4.352569580078125e-05,
      "step": 45640
    },
    {
      "action_loss_avg": 0.0022661060967948286,
      "epoch": 4.352569580078125e-05,
      "step": 45640
    },
    {
      "epoch": 4.352569580078125e-05,
      "step": 45640,
      "training_step_time": 7.76188325881958
    },
    {
      "epoch": 4.3526649475097655e-05,
      "model_forward_time": 1.3696153163909912,
      "step": 45641
    },
    {
      "epoch": 4.3526649475097655e-05,
      "step": 45641,
      "training_step_time": 8.200451850891113
    },
    {
      "epoch": 4.3527603149414065e-05,
      "model_forward_time": 1.7532782554626465,
      "step": 45642
    },
    {
      "epoch": 4.3527603149414065e-05,
      "step": 45642,
      "training_step_time": 7.685962200164795
    },
    {
      "epoch": 4.352855682373047e-05,
      "model_forward_time": 1.7322063446044922,
      "step": 45643
    },
    {
      "epoch": 4.352855682373047e-05,
      "step": 45643,
      "training_step_time": 7.664279222488403
    },
    {
      "epoch": 4.3529510498046873e-05,
      "model_forward_time": 1.735558032989502,
      "step": 45644
    },
    {
      "epoch": 4.3529510498046873e-05,
      "step": 45644,
      "training_step_time": 9.000773429870605
    },
    {
      "epoch": 4.3530464172363284e-05,
      "model_forward_time": 1.736722707748413,
      "step": 45645
    },
    {
      "epoch": 4.3530464172363284e-05,
      "step": 45645,
      "training_step_time": 7.586822986602783
    },
    {
      "epoch": 4.353141784667969e-05,
      "model_forward_time": 1.7462735176086426,
      "step": 45646
    },
    {
      "epoch": 4.353141784667969e-05,
      "step": 45646,
      "training_step_time": 8.541714668273926
    },
    {
      "epoch": 4.353237152099609e-05,
      "model_forward_time": 2.285630464553833,
      "step": 45647
    },
    {
      "epoch": 4.353237152099609e-05,
      "step": 45647,
      "training_step_time": 9.274475336074829
    },
    {
      "epoch": 4.35333251953125e-05,
      "model_forward_time": 1.7739596366882324,
      "step": 45648
    },
    {
      "epoch": 4.35333251953125e-05,
      "step": 45648,
      "training_step_time": 7.664331674575806
    },
    {
      "epoch": 4.353427886962891e-05,
      "model_forward_time": 1.7374536991119385,
      "step": 45649
    },
    {
      "epoch": 4.353427886962891e-05,
      "step": 45649,
      "training_step_time": 8.258797407150269
    },
    {
      "epoch": 4.353523254394531e-05,
      "grad_norm": 0.1534772515296936,
      "learning_rate": 6.1231262163646656e-06,
      "loss": 0.0574,
      "step": 45650
    },
    {
      "epoch": 4.353523254394531e-05,
      "model_forward_time": 1.737863540649414,
      "step": 45650
    },
    {
      "dynamics_loss_avg": 0.05390291810035706,
      "epoch": 4.353523254394531e-05,
      "step": 45650
    },
    {
      "action_loss_avg": 0.001415459837880917,
      "epoch": 4.353523254394531e-05,
      "step": 45650
    },
    {
      "epoch": 4.353523254394531e-05,
      "step": 45650,
      "training_step_time": 7.411569118499756
    },
    {
      "epoch": 4.353618621826172e-05,
      "model_forward_time": 1.7351009845733643,
      "step": 45651
    },
    {
      "epoch": 4.353618621826172e-05,
      "step": 45651,
      "training_step_time": 7.632248878479004
    },
    {
      "epoch": 4.3537139892578126e-05,
      "model_forward_time": 1.7279853820800781,
      "step": 45652
    },
    {
      "epoch": 4.3537139892578126e-05,
      "step": 45652,
      "training_step_time": 8.515884160995483
    },
    {
      "epoch": 4.353809356689453e-05,
      "model_forward_time": 1.331977128982544,
      "step": 45653
    },
    {
      "epoch": 4.353809356689453e-05,
      "step": 45653,
      "training_step_time": 7.754169940948486
    },
    {
      "epoch": 4.353904724121094e-05,
      "model_forward_time": 1.71639084815979,
      "step": 45654
    },
    {
      "epoch": 4.353904724121094e-05,
      "step": 45654,
      "training_step_time": 7.606304168701172
    },
    {
      "epoch": 4.3540000915527345e-05,
      "model_forward_time": 1.3568742275238037,
      "step": 45655
    },
    {
      "epoch": 4.3540000915527345e-05,
      "step": 45655,
      "training_step_time": 8.40030312538147
    },
    {
      "epoch": 4.354095458984375e-05,
      "model_forward_time": 2.2682080268859863,
      "step": 45656
    },
    {
      "epoch": 4.354095458984375e-05,
      "step": 45656,
      "training_step_time": 8.285701513290405
    },
    {
      "epoch": 4.354190826416015e-05,
      "model_forward_time": 1.743826150894165,
      "step": 45657
    },
    {
      "epoch": 4.354190826416015e-05,
      "step": 45657,
      "training_step_time": 7.7242395877838135
    },
    {
      "epoch": 4.3542861938476564e-05,
      "model_forward_time": 1.3511614799499512,
      "step": 45658
    },
    {
      "epoch": 4.3542861938476564e-05,
      "step": 45658,
      "training_step_time": 7.703792572021484
    },
    {
      "epoch": 4.354381561279297e-05,
      "model_forward_time": 1.347503662109375,
      "step": 45659
    },
    {
      "epoch": 4.354381561279297e-05,
      "step": 45659,
      "training_step_time": 7.468950510025024
    },
    {
      "epoch": 4.354476928710937e-05,
      "grad_norm": 0.13003167510032654,
      "learning_rate": 6.121514939210797e-06,
      "loss": 0.0592,
      "step": 45660
    },
    {
      "epoch": 4.354476928710937e-05,
      "model_forward_time": 1.7193591594696045,
      "step": 45660
    },
    {
      "dynamics_loss_avg": 0.049999567307531835,
      "epoch": 4.354476928710937e-05,
      "step": 45660
    },
    {
      "action_loss_avg": 0.0015848721232032403,
      "epoch": 4.354476928710937e-05,
      "step": 45660
    },
    {
      "epoch": 4.354476928710937e-05,
      "step": 45660,
      "training_step_time": 7.670310020446777
    },
    {
      "epoch": 4.354572296142578e-05,
      "model_forward_time": 1.7269160747528076,
      "step": 45661
    },
    {
      "epoch": 4.354572296142578e-05,
      "step": 45661,
      "training_step_time": 7.649050235748291
    },
    {
      "epoch": 4.3546676635742187e-05,
      "model_forward_time": 1.7380406856536865,
      "step": 45662
    },
    {
      "epoch": 4.3546676635742187e-05,
      "step": 45662,
      "training_step_time": 9.007304191589355
    },
    {
      "epoch": 4.354763031005859e-05,
      "model_forward_time": 1.7321889400482178,
      "step": 45663
    },
    {
      "epoch": 4.354763031005859e-05,
      "step": 45663,
      "training_step_time": 8.488476514816284
    },
    {
      "epoch": 4.3548583984375e-05,
      "model_forward_time": 1.7346484661102295,
      "step": 45664
    },
    {
      "epoch": 4.3548583984375e-05,
      "step": 45664,
      "training_step_time": 7.619177579879761
    },
    {
      "epoch": 4.3549537658691405e-05,
      "model_forward_time": 2.2681005001068115,
      "step": 45665
    },
    {
      "epoch": 4.3549537658691405e-05,
      "step": 45665,
      "training_step_time": 7.673891067504883
    },
    {
      "epoch": 4.355049133300781e-05,
      "model_forward_time": 1.7491052150726318,
      "step": 45666
    },
    {
      "epoch": 4.355049133300781e-05,
      "step": 45666,
      "training_step_time": 7.650100469589233
    },
    {
      "epoch": 4.355144500732422e-05,
      "model_forward_time": 1.3688056468963623,
      "step": 45667
    },
    {
      "epoch": 4.355144500732422e-05,
      "step": 45667,
      "training_step_time": 8.316489458084106
    },
    {
      "epoch": 4.3552398681640624e-05,
      "model_forward_time": 1.7333581447601318,
      "step": 45668
    },
    {
      "epoch": 4.3552398681640624e-05,
      "step": 45668,
      "training_step_time": 7.564307451248169
    },
    {
      "epoch": 4.355335235595703e-05,
      "model_forward_time": 1.7361814975738525,
      "step": 45669
    },
    {
      "epoch": 4.355335235595703e-05,
      "step": 45669,
      "training_step_time": 7.7108376026153564
    },
    {
      "epoch": 4.355430603027344e-05,
      "grad_norm": 0.10797915607690811,
      "learning_rate": 6.11990353940974e-06,
      "loss": 0.0581,
      "step": 45670
    },
    {
      "epoch": 4.355430603027344e-05,
      "model_forward_time": 1.7352252006530762,
      "step": 45670
    },
    {
      "dynamics_loss_avg": 0.05986388293094933,
      "epoch": 4.355430603027344e-05,
      "step": 45670
    },
    {
      "action_loss_avg": 0.0022147867581225,
      "epoch": 4.355430603027344e-05,
      "step": 45670
    },
    {
      "epoch": 4.355430603027344e-05,
      "step": 45670,
      "training_step_time": 8.65372633934021
    },
    {
      "epoch": 4.355525970458984e-05,
      "model_forward_time": 1.7343029975891113,
      "step": 45671
    },
    {
      "epoch": 4.355525970458984e-05,
      "step": 45671,
      "training_step_time": 8.30086874961853
    },
    {
      "epoch": 4.355621337890625e-05,
      "model_forward_time": 1.365602970123291,
      "step": 45672
    },
    {
      "epoch": 4.355621337890625e-05,
      "step": 45672,
      "training_step_time": 7.680272579193115
    },
    {
      "epoch": 4.355716705322266e-05,
      "model_forward_time": 1.7065455913543701,
      "step": 45673
    },
    {
      "epoch": 4.355716705322266e-05,
      "step": 45673,
      "training_step_time": 7.662625551223755
    },
    {
      "epoch": 4.355812072753906e-05,
      "model_forward_time": 1.7256436347961426,
      "step": 45674
    },
    {
      "epoch": 4.355812072753906e-05,
      "step": 45674,
      "training_step_time": 7.719196319580078
    },
    {
      "epoch": 4.3559074401855466e-05,
      "model_forward_time": 2.2826969623565674,
      "step": 45675
    },
    {
      "epoch": 4.3559074401855466e-05,
      "step": 45675,
      "training_step_time": 8.239893436431885
    },
    {
      "epoch": 4.356002807617188e-05,
      "model_forward_time": 1.768660068511963,
      "step": 45676
    },
    {
      "epoch": 4.356002807617188e-05,
      "step": 45676,
      "training_step_time": 7.799752235412598
    },
    {
      "epoch": 4.356098175048828e-05,
      "model_forward_time": 1.7269175052642822,
      "step": 45677
    },
    {
      "epoch": 4.356098175048828e-05,
      "step": 45677,
      "training_step_time": 8.25612735748291
    },
    {
      "epoch": 4.3561935424804685e-05,
      "model_forward_time": 1.7237813472747803,
      "step": 45678
    },
    {
      "epoch": 4.3561935424804685e-05,
      "step": 45678,
      "training_step_time": 7.73140549659729
    },
    {
      "epoch": 4.3562889099121096e-05,
      "model_forward_time": 1.7243857383728027,
      "step": 45679
    },
    {
      "epoch": 4.3562889099121096e-05,
      "step": 45679,
      "training_step_time": 7.58358097076416
    },
    {
      "epoch": 4.35638427734375e-05,
      "grad_norm": 0.1372009515762329,
      "learning_rate": 6.118292017137716e-06,
      "loss": 0.0565,
      "step": 45680
    },
    {
      "epoch": 4.35638427734375e-05,
      "model_forward_time": 1.7107257843017578,
      "step": 45680
    },
    {
      "dynamics_loss_avg": 0.04649876095354557,
      "epoch": 4.35638427734375e-05,
      "step": 45680
    },
    {
      "action_loss_avg": 0.0013867202040273696,
      "epoch": 4.35638427734375e-05,
      "step": 45680
    },
    {
      "epoch": 4.35638427734375e-05,
      "step": 45680,
      "training_step_time": 8.25827431678772
    },
    {
      "epoch": 4.3564796447753904e-05,
      "model_forward_time": 1.726635217666626,
      "step": 45681
    },
    {
      "epoch": 4.3564796447753904e-05,
      "step": 45681,
      "training_step_time": 7.712449550628662
    },
    {
      "epoch": 4.3565750122070315e-05,
      "model_forward_time": 1.7218420505523682,
      "step": 45682
    },
    {
      "epoch": 4.3565750122070315e-05,
      "step": 45682,
      "training_step_time": 7.62322473526001
    },
    {
      "epoch": 4.356670379638672e-05,
      "model_forward_time": 1.724534273147583,
      "step": 45683
    },
    {
      "epoch": 4.356670379638672e-05,
      "step": 45683,
      "training_step_time": 7.688067197799683
    },
    {
      "epoch": 4.356765747070312e-05,
      "model_forward_time": 2.270937442779541,
      "step": 45684
    },
    {
      "epoch": 4.356765747070312e-05,
      "step": 45684,
      "training_step_time": 7.701146602630615
    },
    {
      "epoch": 4.3568611145019533e-05,
      "model_forward_time": 1.758927583694458,
      "step": 45685
    },
    {
      "epoch": 4.3568611145019533e-05,
      "step": 45685,
      "training_step_time": 7.78327488899231
    },
    {
      "epoch": 4.356956481933594e-05,
      "model_forward_time": 1.728381633758545,
      "step": 45686
    },
    {
      "epoch": 4.356956481933594e-05,
      "step": 45686,
      "training_step_time": 7.710821866989136
    },
    {
      "epoch": 4.357051849365234e-05,
      "model_forward_time": 1.0089008808135986,
      "step": 45687
    },
    {
      "epoch": 4.357051849365234e-05,
      "step": 45687,
      "training_step_time": 7.727093935012817
    },
    {
      "epoch": 4.357147216796875e-05,
      "model_forward_time": 1.7156157493591309,
      "step": 45688
    },
    {
      "epoch": 4.357147216796875e-05,
      "step": 45688,
      "training_step_time": 8.476114511489868
    },
    {
      "epoch": 4.3572425842285156e-05,
      "model_forward_time": 1.7228217124938965,
      "step": 45689
    },
    {
      "epoch": 4.3572425842285156e-05,
      "step": 45689,
      "training_step_time": 7.577492952346802
    },
    {
      "epoch": 4.357337951660156e-05,
      "grad_norm": 0.08486898988485336,
      "learning_rate": 6.116680372570959e-06,
      "loss": 0.0593,
      "step": 45690
    },
    {
      "epoch": 4.357337951660156e-05,
      "model_forward_time": 1.3588142395019531,
      "step": 45690
    },
    {
      "dynamics_loss_avg": 0.06095453985035419,
      "epoch": 4.357337951660156e-05,
      "step": 45690
    },
    {
      "action_loss_avg": 0.0013640056800795718,
      "epoch": 4.357337951660156e-05,
      "step": 45690
    },
    {
      "epoch": 4.357337951660156e-05,
      "step": 45690,
      "training_step_time": 8.501291275024414
    },
    {
      "epoch": 4.357433319091797e-05,
      "model_forward_time": 1.72650146484375,
      "step": 45691
    },
    {
      "epoch": 4.357433319091797e-05,
      "step": 45691,
      "training_step_time": 8.071358680725098
    },
    {
      "epoch": 4.3575286865234375e-05,
      "model_forward_time": 1.7350425720214844,
      "step": 45692
    },
    {
      "epoch": 4.3575286865234375e-05,
      "step": 45692,
      "training_step_time": 7.645910024642944
    },
    {
      "epoch": 4.357624053955078e-05,
      "model_forward_time": 1.721405267715454,
      "step": 45693
    },
    {
      "epoch": 4.357624053955078e-05,
      "step": 45693,
      "training_step_time": 8.295660972595215
    },
    {
      "epoch": 4.357719421386719e-05,
      "model_forward_time": 2.27541184425354,
      "step": 45694
    },
    {
      "epoch": 4.357719421386719e-05,
      "step": 45694,
      "training_step_time": 7.688733100891113
    },
    {
      "epoch": 4.3578147888183594e-05,
      "model_forward_time": 1.7114250659942627,
      "step": 45695
    },
    {
      "epoch": 4.3578147888183594e-05,
      "step": 45695,
      "training_step_time": 7.660746812820435
    },
    {
      "epoch": 4.35791015625e-05,
      "model_forward_time": 1.7435224056243896,
      "step": 45696
    },
    {
      "epoch": 4.35791015625e-05,
      "step": 45696,
      "training_step_time": 8.492214918136597
    },
    {
      "epoch": 4.358005523681641e-05,
      "model_forward_time": 1.7354815006256104,
      "step": 45697
    },
    {
      "epoch": 4.358005523681641e-05,
      "step": 45697,
      "training_step_time": 7.6869056224823
    },
    {
      "epoch": 4.358100891113281e-05,
      "model_forward_time": 1.744593858718872,
      "step": 45698
    },
    {
      "epoch": 4.358100891113281e-05,
      "step": 45698,
      "training_step_time": 8.213382720947266
    },
    {
      "epoch": 4.358196258544922e-05,
      "model_forward_time": 1.3606269359588623,
      "step": 45699
    },
    {
      "epoch": 4.358196258544922e-05,
      "step": 45699,
      "training_step_time": 8.262075424194336
    },
    {
      "epoch": 4.358291625976563e-05,
      "grad_norm": 0.09857381135225296,
      "learning_rate": 6.115068605885713e-06,
      "loss": 0.0589,
      "step": 45700
    },
    {
      "epoch": 4.358291625976563e-05,
      "model_forward_time": 1.7580158710479736,
      "step": 45700
    },
    {
      "dynamics_loss_avg": 0.053038550354540345,
      "epoch": 4.358291625976563e-05,
      "step": 45700
    },
    {
      "action_loss_avg": 0.0013669712672708556,
      "epoch": 4.358291625976563e-05,
      "step": 45700
    },
    {
      "epoch": 4.358291625976563e-05,
      "step": 45700,
      "training_step_time": 7.616913795471191
    },
    {
      "epoch": 4.358386993408203e-05,
      "model_forward_time": 1.7301921844482422,
      "step": 45701
    },
    {
      "epoch": 4.358386993408203e-05,
      "step": 45701,
      "training_step_time": 9.110073804855347
    },
    {
      "epoch": 4.3584823608398436e-05,
      "model_forward_time": 1.7348899841308594,
      "step": 45702
    },
    {
      "epoch": 4.3584823608398436e-05,
      "step": 45702,
      "training_step_time": 7.713276624679565
    },
    {
      "epoch": 4.3585777282714847e-05,
      "model_forward_time": 2.909820318222046,
      "step": 45703
    },
    {
      "epoch": 4.3585777282714847e-05,
      "step": 45703,
      "training_step_time": 8.61647915840149
    },
    {
      "epoch": 4.358673095703125e-05,
      "model_forward_time": 1.7459938526153564,
      "step": 45704
    },
    {
      "epoch": 4.358673095703125e-05,
      "step": 45704,
      "training_step_time": 7.733948469161987
    },
    {
      "epoch": 4.3587684631347655e-05,
      "model_forward_time": 1.746307373046875,
      "step": 45705
    },
    {
      "epoch": 4.3587684631347655e-05,
      "step": 45705,
      "training_step_time": 7.459946632385254
    },
    {
      "epoch": 4.3588638305664065e-05,
      "model_forward_time": 1.7453722953796387,
      "step": 45706
    },
    {
      "epoch": 4.3588638305664065e-05,
      "step": 45706,
      "training_step_time": 7.686035394668579
    },
    {
      "epoch": 4.358959197998047e-05,
      "model_forward_time": 1.7543392181396484,
      "step": 45707
    },
    {
      "epoch": 4.358959197998047e-05,
      "step": 45707,
      "training_step_time": 8.432977437973022
    },
    {
      "epoch": 4.3590545654296873e-05,
      "model_forward_time": 1.7627897262573242,
      "step": 45708
    },
    {
      "epoch": 4.3590545654296873e-05,
      "step": 45708,
      "training_step_time": 7.627758502960205
    },
    {
      "epoch": 4.3591499328613284e-05,
      "model_forward_time": 1.803816556930542,
      "step": 45709
    },
    {
      "epoch": 4.3591499328613284e-05,
      "step": 45709,
      "training_step_time": 8.388242959976196
    },
    {
      "epoch": 4.359245300292969e-05,
      "grad_norm": 0.11931926757097244,
      "learning_rate": 6.113456717258244e-06,
      "loss": 0.0586,
      "step": 45710
    },
    {
      "epoch": 4.359245300292969e-05,
      "model_forward_time": 1.7520701885223389,
      "step": 45710
    },
    {
      "dynamics_loss_avg": 0.05791901797056198,
      "epoch": 4.359245300292969e-05,
      "step": 45710
    },
    {
      "action_loss_avg": 0.0012722694780677557,
      "epoch": 4.359245300292969e-05,
      "step": 45710
    },
    {
      "epoch": 4.359245300292969e-05,
      "step": 45710,
      "training_step_time": 8.238173961639404
    },
    {
      "epoch": 4.359340667724609e-05,
      "model_forward_time": 1.3958516120910645,
      "step": 45711
    },
    {
      "epoch": 4.359340667724609e-05,
      "step": 45711,
      "training_step_time": 7.742766380310059
    },
    {
      "epoch": 4.35943603515625e-05,
      "model_forward_time": 2.850990056991577,
      "step": 45712
    },
    {
      "epoch": 4.35943603515625e-05,
      "step": 45712,
      "training_step_time": 8.327763795852661
    },
    {
      "epoch": 4.359531402587891e-05,
      "model_forward_time": 1.7740657329559326,
      "step": 45713
    },
    {
      "epoch": 4.359531402587891e-05,
      "step": 45713,
      "training_step_time": 7.80548095703125
    },
    {
      "epoch": 4.359626770019531e-05,
      "model_forward_time": 1.734009027481079,
      "step": 45714
    },
    {
      "epoch": 4.359626770019531e-05,
      "step": 45714,
      "training_step_time": 7.716773271560669
    },
    {
      "epoch": 4.359722137451172e-05,
      "model_forward_time": 1.7417192459106445,
      "step": 45715
    },
    {
      "epoch": 4.359722137451172e-05,
      "step": 45715,
      "training_step_time": 7.7192676067352295
    },
    {
      "epoch": 4.3598175048828126e-05,
      "model_forward_time": 1.7388744354248047,
      "step": 45716
    },
    {
      "epoch": 4.3598175048828126e-05,
      "step": 45716,
      "training_step_time": 7.587386608123779
    },
    {
      "epoch": 4.359912872314453e-05,
      "model_forward_time": 1.7529675960540771,
      "step": 45717
    },
    {
      "epoch": 4.359912872314453e-05,
      "step": 45717,
      "training_step_time": 7.693877696990967
    },
    {
      "epoch": 4.360008239746094e-05,
      "model_forward_time": 1.8315277099609375,
      "step": 45718
    },
    {
      "epoch": 4.360008239746094e-05,
      "step": 45718,
      "training_step_time": 7.666930913925171
    },
    {
      "epoch": 4.3601036071777345e-05,
      "model_forward_time": 1.756596326828003,
      "step": 45719
    },
    {
      "epoch": 4.3601036071777345e-05,
      "step": 45719,
      "training_step_time": 8.684001684188843
    },
    {
      "epoch": 4.360198974609375e-05,
      "grad_norm": 0.11182565987110138,
      "learning_rate": 6.1118447068648205e-06,
      "loss": 0.0576,
      "step": 45720
    },
    {
      "epoch": 4.360198974609375e-05,
      "model_forward_time": 1.7402129173278809,
      "step": 45720
    },
    {
      "dynamics_loss_avg": 0.060843872278928755,
      "epoch": 4.360198974609375e-05,
      "step": 45720
    },
    {
      "action_loss_avg": 0.0017943775048479438,
      "epoch": 4.360198974609375e-05,
      "step": 45720
    },
    {
      "epoch": 4.360198974609375e-05,
      "step": 45720,
      "training_step_time": 8.767974376678467
    },
    {
      "epoch": 4.360294342041015e-05,
      "model_forward_time": 1.7699129581451416,
      "step": 45721
    },
    {
      "epoch": 4.360294342041015e-05,
      "step": 45721,
      "training_step_time": 7.816400051116943
    },
    {
      "epoch": 4.3603897094726564e-05,
      "model_forward_time": 2.959197759628296,
      "step": 45722
    },
    {
      "epoch": 4.3603897094726564e-05,
      "step": 45722,
      "training_step_time": 7.838671922683716
    },
    {
      "epoch": 4.360485076904297e-05,
      "model_forward_time": 1.733846664428711,
      "step": 45723
    },
    {
      "epoch": 4.360485076904297e-05,
      "step": 45723,
      "training_step_time": 7.739692211151123
    },
    {
      "epoch": 4.360580444335937e-05,
      "model_forward_time": 1.3956892490386963,
      "step": 45724
    },
    {
      "epoch": 4.360580444335937e-05,
      "step": 45724,
      "training_step_time": 7.8914947509765625
    },
    {
      "epoch": 4.360675811767578e-05,
      "model_forward_time": 1.391021728515625,
      "step": 45725
    },
    {
      "epoch": 4.360675811767578e-05,
      "step": 45725,
      "training_step_time": 7.474101305007935
    },
    {
      "epoch": 4.360771179199219e-05,
      "model_forward_time": 1.7464289665222168,
      "step": 45726
    },
    {
      "epoch": 4.360771179199219e-05,
      "step": 45726,
      "training_step_time": 8.540601968765259
    },
    {
      "epoch": 4.360866546630859e-05,
      "model_forward_time": 1.732132911682129,
      "step": 45727
    },
    {
      "epoch": 4.360866546630859e-05,
      "step": 45727,
      "training_step_time": 8.918121337890625
    },
    {
      "epoch": 4.3609619140625e-05,
      "model_forward_time": 1.7433841228485107,
      "step": 45728
    },
    {
      "epoch": 4.3609619140625e-05,
      "step": 45728,
      "training_step_time": 9.84844160079956
    },
    {
      "epoch": 4.3610572814941405e-05,
      "model_forward_time": 1.044663906097412,
      "step": 45729
    },
    {
      "epoch": 4.3610572814941405e-05,
      "step": 45729,
      "training_step_time": 6.768539190292358
    },
    {
      "epoch": 4.361152648925781e-05,
      "grad_norm": 0.0848924070596695,
      "learning_rate": 6.110232574881732e-06,
      "loss": 0.057,
      "step": 45730
    },
    {
      "epoch": 4.361152648925781e-05,
      "model_forward_time": 1.7265312671661377,
      "step": 45730
    },
    {
      "dynamics_loss_avg": 0.05331923700869083,
      "epoch": 4.361152648925781e-05,
      "step": 45730
    },
    {
      "action_loss_avg": 0.0020357706729555503,
      "epoch": 4.361152648925781e-05,
      "step": 45730
    },
    {
      "epoch": 4.361152648925781e-05,
      "step": 45730,
      "training_step_time": 6.8274595737457275
    },
    {
      "epoch": 4.361248016357422e-05,
      "model_forward_time": 1.3742196559906006,
      "step": 45731
    },
    {
      "epoch": 4.361248016357422e-05,
      "step": 45731,
      "training_step_time": 6.817224502563477
    },
    {
      "epoch": 4.3613433837890624e-05,
      "model_forward_time": 1.722973346710205,
      "step": 45732
    },
    {
      "epoch": 4.3613433837890624e-05,
      "step": 45732,
      "training_step_time": 6.7801430225372314
    },
    {
      "epoch": 4.361438751220703e-05,
      "model_forward_time": 1.7199339866638184,
      "step": 45733
    },
    {
      "epoch": 4.361438751220703e-05,
      "step": 45733,
      "training_step_time": 6.779999256134033
    },
    {
      "epoch": 4.361534118652344e-05,
      "model_forward_time": 1.742131233215332,
      "step": 45734
    },
    {
      "epoch": 4.361534118652344e-05,
      "step": 45734,
      "training_step_time": 6.75202202796936
    },
    {
      "epoch": 4.361629486083984e-05,
      "model_forward_time": 1.3838260173797607,
      "step": 45735
    },
    {
      "epoch": 4.361629486083984e-05,
      "step": 45735,
      "training_step_time": 6.771106004714966
    },
    {
      "epoch": 4.361724853515625e-05,
      "model_forward_time": 1.7195243835449219,
      "step": 45736
    },
    {
      "epoch": 4.361724853515625e-05,
      "step": 45736,
      "training_step_time": 7.84807014465332
    },
    {
      "epoch": 4.361820220947266e-05,
      "model_forward_time": 2.8689043521881104,
      "step": 45737
    },
    {
      "epoch": 4.361820220947266e-05,
      "step": 45737,
      "training_step_time": 7.614318370819092
    },
    {
      "epoch": 4.361915588378906e-05,
      "model_forward_time": 1.7331264019012451,
      "step": 45738
    },
    {
      "epoch": 4.361915588378906e-05,
      "step": 45738,
      "training_step_time": 7.679490566253662
    },
    {
      "epoch": 4.3620109558105466e-05,
      "model_forward_time": 1.7296924591064453,
      "step": 45739
    },
    {
      "epoch": 4.3620109558105466e-05,
      "step": 45739,
      "training_step_time": 6.793590545654297
    },
    {
      "epoch": 4.362106323242188e-05,
      "grad_norm": 0.12276444584131241,
      "learning_rate": 6.108620321485278e-06,
      "loss": 0.0588,
      "step": 45740
    },
    {
      "epoch": 4.362106323242188e-05,
      "model_forward_time": 1.7125587463378906,
      "step": 45740
    },
    {
      "dynamics_loss_avg": 0.05830732714384794,
      "epoch": 4.362106323242188e-05,
      "step": 45740
    },
    {
      "action_loss_avg": 0.0028302852530032395,
      "epoch": 4.362106323242188e-05,
      "step": 45740
    },
    {
      "epoch": 4.362106323242188e-05,
      "step": 45740,
      "training_step_time": 6.7813520431518555
    },
    {
      "epoch": 4.362201690673828e-05,
      "model_forward_time": 1.7313408851623535,
      "step": 45741
    },
    {
      "epoch": 4.362201690673828e-05,
      "step": 45741,
      "training_step_time": 6.767038583755493
    },
    {
      "epoch": 4.3622970581054685e-05,
      "model_forward_time": 1.0370988845825195,
      "step": 45742
    },
    {
      "epoch": 4.3622970581054685e-05,
      "step": 45742,
      "training_step_time": 6.82005763053894
    },
    {
      "epoch": 4.3623924255371096e-05,
      "model_forward_time": 1.7347140312194824,
      "step": 45743
    },
    {
      "epoch": 4.3623924255371096e-05,
      "step": 45743,
      "training_step_time": 6.787527322769165
    },
    {
      "epoch": 4.36248779296875e-05,
      "model_forward_time": 1.7273924350738525,
      "step": 45744
    },
    {
      "epoch": 4.36248779296875e-05,
      "step": 45744,
      "training_step_time": 6.7568840980529785
    },
    {
      "epoch": 4.3625831604003904e-05,
      "model_forward_time": 1.354377031326294,
      "step": 45745
    },
    {
      "epoch": 4.3625831604003904e-05,
      "step": 45745,
      "training_step_time": 7.738835334777832
    },
    {
      "epoch": 4.3626785278320315e-05,
      "model_forward_time": 1.7176802158355713,
      "step": 45746
    },
    {
      "epoch": 4.3626785278320315e-05,
      "step": 45746,
      "training_step_time": 7.203518867492676
    },
    {
      "epoch": 4.362773895263672e-05,
      "model_forward_time": 2.9790899753570557,
      "step": 45747
    },
    {
      "epoch": 4.362773895263672e-05,
      "step": 45747,
      "training_step_time": 7.7056238651275635
    },
    {
      "epoch": 4.362869262695312e-05,
      "model_forward_time": 1.7218101024627686,
      "step": 45748
    },
    {
      "epoch": 4.362869262695312e-05,
      "step": 45748,
      "training_step_time": 7.327698469161987
    },
    {
      "epoch": 4.3629646301269533e-05,
      "model_forward_time": 1.7402584552764893,
      "step": 45749
    },
    {
      "epoch": 4.3629646301269533e-05,
      "step": 45749,
      "training_step_time": 6.742333173751831
    },
    {
      "epoch": 4.363059997558594e-05,
      "grad_norm": 0.10794082283973694,
      "learning_rate": 6.107007946851773e-06,
      "loss": 0.0537,
      "step": 45750
    },
    {
      "epoch": 4.363059997558594e-05,
      "model_forward_time": 1.7195250988006592,
      "step": 45750
    },
    {
      "dynamics_loss_avg": 0.057818902656435966,
      "epoch": 4.363059997558594e-05,
      "step": 45750
    },
    {
      "action_loss_avg": 0.001648126303916797,
      "epoch": 4.363059997558594e-05,
      "step": 45750
    },
    {
      "epoch": 4.363059997558594e-05,
      "step": 45750,
      "training_step_time": 6.7549872398376465
    },
    {
      "epoch": 4.363155364990234e-05,
      "model_forward_time": 1.7163655757904053,
      "step": 45751
    },
    {
      "epoch": 4.363155364990234e-05,
      "step": 45751,
      "training_step_time": 6.714697599411011
    },
    {
      "epoch": 4.363250732421875e-05,
      "model_forward_time": 1.725208044052124,
      "step": 45752
    },
    {
      "epoch": 4.363250732421875e-05,
      "step": 45752,
      "training_step_time": 6.78612756729126
    },
    {
      "epoch": 4.3633460998535156e-05,
      "model_forward_time": 1.7297987937927246,
      "step": 45753
    },
    {
      "epoch": 4.3633460998535156e-05,
      "step": 45753,
      "training_step_time": 6.728612184524536
    },
    {
      "epoch": 4.363441467285156e-05,
      "model_forward_time": 1.7151713371276855,
      "step": 45754
    },
    {
      "epoch": 4.363441467285156e-05,
      "step": 45754,
      "training_step_time": 7.6997199058532715
    },
    {
      "epoch": 4.363536834716797e-05,
      "model_forward_time": 1.7105376720428467,
      "step": 45755
    },
    {
      "epoch": 4.363536834716797e-05,
      "step": 45755,
      "training_step_time": 7.1872124671936035
    },
    {
      "epoch": 4.3636322021484375e-05,
      "model_forward_time": 2.5182275772094727,
      "step": 45756
    },
    {
      "epoch": 4.3636322021484375e-05,
      "step": 45756,
      "training_step_time": 8.254902124404907
    },
    {
      "epoch": 4.363727569580078e-05,
      "model_forward_time": 1.394418478012085,
      "step": 45757
    },
    {
      "epoch": 4.363727569580078e-05,
      "step": 45757,
      "training_step_time": 7.371575117111206
    },
    {
      "epoch": 4.363822937011719e-05,
      "model_forward_time": 1.7134697437286377,
      "step": 45758
    },
    {
      "epoch": 4.363822937011719e-05,
      "step": 45758,
      "training_step_time": 6.776819705963135
    },
    {
      "epoch": 4.3639183044433594e-05,
      "model_forward_time": 1.7112336158752441,
      "step": 45759
    },
    {
      "epoch": 4.3639183044433594e-05,
      "step": 45759,
      "training_step_time": 6.747977256774902
    },
    {
      "epoch": 4.364013671875e-05,
      "grad_norm": 0.14210249483585358,
      "learning_rate": 6.105395451157543e-06,
      "loss": 0.0584,
      "step": 45760
    },
    {
      "epoch": 4.364013671875e-05,
      "model_forward_time": 1.3726952075958252,
      "step": 45760
    },
    {
      "dynamics_loss_avg": 0.06341619566082954,
      "epoch": 4.364013671875e-05,
      "step": 45760
    },
    {
      "action_loss_avg": 0.0020627140067517756,
      "epoch": 4.364013671875e-05,
      "step": 45760
    },
    {
      "epoch": 4.364013671875e-05,
      "step": 45760,
      "training_step_time": 6.75860071182251
    },
    {
      "epoch": 4.364109039306641e-05,
      "model_forward_time": 1.7279078960418701,
      "step": 45761
    },
    {
      "epoch": 4.364109039306641e-05,
      "step": 45761,
      "training_step_time": 6.789007663726807
    },
    {
      "epoch": 4.364204406738281e-05,
      "model_forward_time": 1.7433011531829834,
      "step": 45762
    },
    {
      "epoch": 4.364204406738281e-05,
      "step": 45762,
      "training_step_time": 6.777441740036011
    },
    {
      "epoch": 4.364299774169922e-05,
      "model_forward_time": 1.7433381080627441,
      "step": 45763
    },
    {
      "epoch": 4.364299774169922e-05,
      "step": 45763,
      "training_step_time": 6.758983850479126
    },
    {
      "epoch": 4.364395141601563e-05,
      "model_forward_time": 1.7874696254730225,
      "step": 45764
    },
    {
      "epoch": 4.364395141601563e-05,
      "step": 45764,
      "training_step_time": 7.216683387756348
    },
    {
      "epoch": 4.364490509033203e-05,
      "model_forward_time": 1.7327930927276611,
      "step": 45765
    },
    {
      "epoch": 4.364490509033203e-05,
      "step": 45765,
      "training_step_time": 10.883981466293335
    },
    {
      "epoch": 4.3645858764648436e-05,
      "model_forward_time": 1.3703010082244873,
      "step": 45766
    },
    {
      "epoch": 4.3645858764648436e-05,
      "step": 45766,
      "training_step_time": 7.6342453956604
    },
    {
      "epoch": 4.364681243896485e-05,
      "model_forward_time": 1.725844144821167,
      "step": 45767
    },
    {
      "epoch": 4.364681243896485e-05,
      "step": 45767,
      "training_step_time": 6.808718919754028
    },
    {
      "epoch": 4.364776611328125e-05,
      "model_forward_time": 1.7330186367034912,
      "step": 45768
    },
    {
      "epoch": 4.364776611328125e-05,
      "step": 45768,
      "training_step_time": 6.845185995101929
    },
    {
      "epoch": 4.3648719787597655e-05,
      "model_forward_time": 1.384448766708374,
      "step": 45769
    },
    {
      "epoch": 4.3648719787597655e-05,
      "step": 45769,
      "training_step_time": 6.837877035140991
    },
    {
      "epoch": 4.3649673461914065e-05,
      "grad_norm": 0.14441613852977753,
      "learning_rate": 6.1037828345789285e-06,
      "loss": 0.0564,
      "step": 45770
    },
    {
      "epoch": 4.3649673461914065e-05,
      "model_forward_time": 1.7211833000183105,
      "step": 45770
    },
    {
      "dynamics_loss_avg": 0.046642659977078435,
      "epoch": 4.3649673461914065e-05,
      "step": 45770
    },
    {
      "action_loss_avg": 0.0012836543261073529,
      "epoch": 4.3649673461914065e-05,
      "step": 45770
    },
    {
      "epoch": 4.3649673461914065e-05,
      "step": 45770,
      "training_step_time": 6.802225828170776
    },
    {
      "epoch": 4.365062713623047e-05,
      "model_forward_time": 1.7180356979370117,
      "step": 45771
    },
    {
      "epoch": 4.365062713623047e-05,
      "step": 45771,
      "training_step_time": 6.883519887924194
    },
    {
      "epoch": 4.3651580810546874e-05,
      "model_forward_time": 1.725393533706665,
      "step": 45772
    },
    {
      "epoch": 4.3651580810546874e-05,
      "step": 45772,
      "training_step_time": 6.910074234008789
    },
    {
      "epoch": 4.3652534484863284e-05,
      "model_forward_time": 1.7942719459533691,
      "step": 45773
    },
    {
      "epoch": 4.3652534484863284e-05,
      "step": 45773,
      "training_step_time": 7.9613988399505615
    },
    {
      "epoch": 4.365348815917969e-05,
      "model_forward_time": 1.7416601181030273,
      "step": 45774
    },
    {
      "epoch": 4.365348815917969e-05,
      "step": 45774,
      "training_step_time": 8.279848098754883
    },
    {
      "epoch": 4.365444183349609e-05,
      "model_forward_time": 2.8860390186309814,
      "step": 45775
    },
    {
      "epoch": 4.365444183349609e-05,
      "step": 45775,
      "training_step_time": 7.6884260177612305
    },
    {
      "epoch": 4.36553955078125e-05,
      "model_forward_time": 1.7164700031280518,
      "step": 45776
    },
    {
      "epoch": 4.36553955078125e-05,
      "step": 45776,
      "training_step_time": 6.845464468002319
    },
    {
      "epoch": 4.365634918212891e-05,
      "model_forward_time": 1.7288017272949219,
      "step": 45777
    },
    {
      "epoch": 4.365634918212891e-05,
      "step": 45777,
      "training_step_time": 6.835612058639526
    },
    {
      "epoch": 4.365730285644531e-05,
      "model_forward_time": 1.0316741466522217,
      "step": 45778
    },
    {
      "epoch": 4.365730285644531e-05,
      "step": 45778,
      "training_step_time": 6.799120664596558
    },
    {
      "epoch": 4.365825653076172e-05,
      "model_forward_time": 1.7231357097625732,
      "step": 45779
    },
    {
      "epoch": 4.365825653076172e-05,
      "step": 45779,
      "training_step_time": 6.795062303543091
    },
    {
      "epoch": 4.3659210205078126e-05,
      "grad_norm": 0.10360055416822433,
      "learning_rate": 6.1021700972922815e-06,
      "loss": 0.0584,
      "step": 45780
    },
    {
      "epoch": 4.3659210205078126e-05,
      "model_forward_time": 1.7205822467803955,
      "step": 45780
    },
    {
      "dynamics_loss_avg": 0.05481221480295062,
      "epoch": 4.3659210205078126e-05,
      "step": 45780
    },
    {
      "action_loss_avg": 0.0020526145468465983,
      "epoch": 4.3659210205078126e-05,
      "step": 45780
    },
    {
      "epoch": 4.3659210205078126e-05,
      "step": 45780,
      "training_step_time": 6.8381383419036865
    },
    {
      "epoch": 4.366016387939453e-05,
      "model_forward_time": 1.7288756370544434,
      "step": 45781
    },
    {
      "epoch": 4.366016387939453e-05,
      "step": 45781,
      "training_step_time": 6.861131191253662
    },
    {
      "epoch": 4.3661117553710934e-05,
      "model_forward_time": 1.7999091148376465,
      "step": 45782
    },
    {
      "epoch": 4.3661117553710934e-05,
      "step": 45782,
      "training_step_time": 7.134902238845825
    },
    {
      "epoch": 4.3662071228027345e-05,
      "model_forward_time": 1.7147057056427002,
      "step": 45783
    },
    {
      "epoch": 4.3662071228027345e-05,
      "step": 45783,
      "training_step_time": 7.740827322006226
    },
    {
      "epoch": 4.366302490234375e-05,
      "model_forward_time": 2.814995527267456,
      "step": 45784
    },
    {
      "epoch": 4.366302490234375e-05,
      "step": 45784,
      "training_step_time": 8.226285934448242
    },
    {
      "epoch": 4.366397857666015e-05,
      "model_forward_time": 1.3764400482177734,
      "step": 45785
    },
    {
      "epoch": 4.366397857666015e-05,
      "step": 45785,
      "training_step_time": 7.426419734954834
    },
    {
      "epoch": 4.3664932250976564e-05,
      "model_forward_time": 1.7242038249969482,
      "step": 45786
    },
    {
      "epoch": 4.3664932250976564e-05,
      "step": 45786,
      "training_step_time": 6.824488878250122
    },
    {
      "epoch": 4.366588592529297e-05,
      "model_forward_time": 1.723198652267456,
      "step": 45787
    },
    {
      "epoch": 4.366588592529297e-05,
      "step": 45787,
      "training_step_time": 6.7617433071136475
    },
    {
      "epoch": 4.366683959960937e-05,
      "model_forward_time": 1.722203254699707,
      "step": 45788
    },
    {
      "epoch": 4.366683959960937e-05,
      "step": 45788,
      "training_step_time": 6.767510652542114
    },
    {
      "epoch": 4.366779327392578e-05,
      "model_forward_time": 1.7320129871368408,
      "step": 45789
    },
    {
      "epoch": 4.366779327392578e-05,
      "step": 45789,
      "training_step_time": 6.83536696434021
    },
    {
      "epoch": 4.366874694824219e-05,
      "grad_norm": 0.1204838752746582,
      "learning_rate": 6.100557239473972e-06,
      "loss": 0.0559,
      "step": 45790
    },
    {
      "epoch": 4.366874694824219e-05,
      "model_forward_time": 1.735128402709961,
      "step": 45790
    },
    {
      "dynamics_loss_avg": 0.06411743201315404,
      "epoch": 4.366874694824219e-05,
      "step": 45790
    },
    {
      "action_loss_avg": 0.0017547421797644348,
      "epoch": 4.366874694824219e-05,
      "step": 45790
    },
    {
      "epoch": 4.366874694824219e-05,
      "step": 45790,
      "training_step_time": 6.815999984741211
    },
    {
      "epoch": 4.366970062255859e-05,
      "model_forward_time": 1.79701566696167,
      "step": 45791
    },
    {
      "epoch": 4.366970062255859e-05,
      "step": 45791,
      "training_step_time": 7.078491687774658
    },
    {
      "epoch": 4.3670654296875e-05,
      "model_forward_time": 1.719693899154663,
      "step": 45792
    },
    {
      "epoch": 4.3670654296875e-05,
      "step": 45792,
      "training_step_time": 8.99494194984436
    },
    {
      "epoch": 4.3671607971191406e-05,
      "model_forward_time": 1.7277140617370605,
      "step": 45793
    },
    {
      "epoch": 4.3671607971191406e-05,
      "step": 45793,
      "training_step_time": 8.853544235229492
    },
    {
      "epoch": 4.367256164550781e-05,
      "model_forward_time": 2.551204204559326,
      "step": 45794
    },
    {
      "epoch": 4.367256164550781e-05,
      "step": 45794,
      "training_step_time": 7.409325838088989
    },
    {
      "epoch": 4.367351531982422e-05,
      "model_forward_time": 1.719590187072754,
      "step": 45795
    },
    {
      "epoch": 4.367351531982422e-05,
      "step": 45795,
      "training_step_time": 6.806284666061401
    },
    {
      "epoch": 4.3674468994140624e-05,
      "model_forward_time": 1.7288274765014648,
      "step": 45796
    },
    {
      "epoch": 4.3674468994140624e-05,
      "step": 45796,
      "training_step_time": 6.760069847106934
    },
    {
      "epoch": 4.367542266845703e-05,
      "model_forward_time": 1.7409512996673584,
      "step": 45797
    },
    {
      "epoch": 4.367542266845703e-05,
      "step": 45797,
      "training_step_time": 6.84924840927124
    },
    {
      "epoch": 4.367637634277344e-05,
      "model_forward_time": 1.730712652206421,
      "step": 45798
    },
    {
      "epoch": 4.367637634277344e-05,
      "step": 45798,
      "training_step_time": 6.759154558181763
    },
    {
      "epoch": 4.367733001708984e-05,
      "model_forward_time": 1.7251427173614502,
      "step": 45799
    },
    {
      "epoch": 4.367733001708984e-05,
      "step": 45799,
      "training_step_time": 6.783257961273193
    },
    {
      "epoch": 4.367828369140625e-05,
      "grad_norm": 0.10770674794912338,
      "learning_rate": 6.098944261300377e-06,
      "loss": 0.0576,
      "step": 45800
    },
    {
      "epoch": 4.367828369140625e-05,
      "model_forward_time": 1.780247688293457,
      "step": 45800
    },
    {
      "dynamics_loss_avg": 0.04642952401190996,
      "epoch": 4.367828369140625e-05,
      "step": 45800
    },
    {
      "action_loss_avg": 0.001151480615953915,
      "epoch": 4.367828369140625e-05,
      "step": 45800
    },
    {
      "epoch": 4.367828369140625e-05,
      "step": 45800,
      "training_step_time": 7.064528703689575
    },
    {
      "epoch": 4.367923736572266e-05,
      "model_forward_time": 1.7166197299957275,
      "step": 45801
    },
    {
      "epoch": 4.367923736572266e-05,
      "step": 45801,
      "training_step_time": 9.035843849182129
    },
    {
      "epoch": 4.368019104003906e-05,
      "model_forward_time": 1.7218623161315918,
      "step": 45802
    },
    {
      "epoch": 4.368019104003906e-05,
      "step": 45802,
      "training_step_time": 8.917510509490967
    },
    {
      "epoch": 4.3681144714355466e-05,
      "model_forward_time": 2.9108846187591553,
      "step": 45803
    },
    {
      "epoch": 4.3681144714355466e-05,
      "step": 45803,
      "training_step_time": 7.688702821731567
    },
    {
      "epoch": 4.368209838867188e-05,
      "model_forward_time": 1.7302443981170654,
      "step": 45804
    },
    {
      "epoch": 4.368209838867188e-05,
      "step": 45804,
      "training_step_time": 6.790651321411133
    },
    {
      "epoch": 4.368305206298828e-05,
      "model_forward_time": 1.7289438247680664,
      "step": 45805
    },
    {
      "epoch": 4.368305206298828e-05,
      "step": 45805,
      "training_step_time": 6.845403432846069
    },
    {
      "epoch": 4.3684005737304685e-05,
      "model_forward_time": 1.7228314876556396,
      "step": 45806
    },
    {
      "epoch": 4.3684005737304685e-05,
      "step": 45806,
      "training_step_time": 6.824933052062988
    },
    {
      "epoch": 4.3684959411621096e-05,
      "model_forward_time": 1.723301887512207,
      "step": 45807
    },
    {
      "epoch": 4.3684959411621096e-05,
      "step": 45807,
      "training_step_time": 6.782917499542236
    },
    {
      "epoch": 4.36859130859375e-05,
      "model_forward_time": 1.7243430614471436,
      "step": 45808
    },
    {
      "epoch": 4.36859130859375e-05,
      "step": 45808,
      "training_step_time": 6.778225898742676
    },
    {
      "epoch": 4.3686866760253904e-05,
      "model_forward_time": 1.7583632469177246,
      "step": 45809
    },
    {
      "epoch": 4.3686866760253904e-05,
      "step": 45809,
      "training_step_time": 7.076780796051025
    },
    {
      "epoch": 4.3687820434570315e-05,
      "grad_norm": 0.12029842287302017,
      "learning_rate": 6.0973311629478885e-06,
      "loss": 0.0588,
      "step": 45810
    },
    {
      "epoch": 4.3687820434570315e-05,
      "model_forward_time": 1.7329254150390625,
      "step": 45810
    },
    {
      "dynamics_loss_avg": 0.041876418702304366,
      "epoch": 4.3687820434570315e-05,
      "step": 45810
    },
    {
      "action_loss_avg": 0.0010306479613063856,
      "epoch": 4.3687820434570315e-05,
      "step": 45810
    },
    {
      "epoch": 4.3687820434570315e-05,
      "step": 45810,
      "training_step_time": 7.671082258224487
    },
    {
      "epoch": 4.368877410888672e-05,
      "model_forward_time": 1.3717117309570312,
      "step": 45811
    },
    {
      "epoch": 4.368877410888672e-05,
      "step": 45811,
      "training_step_time": 11.152605295181274
    },
    {
      "epoch": 4.368972778320312e-05,
      "model_forward_time": 2.8133468627929688,
      "step": 45812
    },
    {
      "epoch": 4.368972778320312e-05,
      "step": 45812,
      "training_step_time": 7.662486553192139
    },
    {
      "epoch": 4.3690681457519534e-05,
      "model_forward_time": 1.3695135116577148,
      "step": 45813
    },
    {
      "epoch": 4.3690681457519534e-05,
      "step": 45813,
      "training_step_time": 6.818526029586792
    },
    {
      "epoch": 4.369163513183594e-05,
      "model_forward_time": 1.7238996028900146,
      "step": 45814
    },
    {
      "epoch": 4.369163513183594e-05,
      "step": 45814,
      "training_step_time": 6.815108299255371
    },
    {
      "epoch": 4.369258880615234e-05,
      "model_forward_time": 1.7298972606658936,
      "step": 45815
    },
    {
      "epoch": 4.369258880615234e-05,
      "step": 45815,
      "training_step_time": 6.722775459289551
    },
    {
      "epoch": 4.369354248046875e-05,
      "model_forward_time": 1.7698380947113037,
      "step": 45816
    },
    {
      "epoch": 4.369354248046875e-05,
      "step": 45816,
      "training_step_time": 6.733564615249634
    },
    {
      "epoch": 4.3694496154785156e-05,
      "model_forward_time": 1.7204713821411133,
      "step": 45817
    },
    {
      "epoch": 4.3694496154785156e-05,
      "step": 45817,
      "training_step_time": 6.769826650619507
    },
    {
      "epoch": 4.369544982910156e-05,
      "model_forward_time": 1.750605583190918,
      "step": 45818
    },
    {
      "epoch": 4.369544982910156e-05,
      "step": 45818,
      "training_step_time": 7.0423924922943115
    },
    {
      "epoch": 4.369640350341797e-05,
      "model_forward_time": 1.7180750370025635,
      "step": 45819
    },
    {
      "epoch": 4.369640350341797e-05,
      "step": 45819,
      "training_step_time": 7.724793195724487
    },
    {
      "epoch": 4.3697357177734375e-05,
      "grad_norm": 0.11640123277902603,
      "learning_rate": 6.0957179445929135e-06,
      "loss": 0.0597,
      "step": 45820
    },
    {
      "epoch": 4.3697357177734375e-05,
      "model_forward_time": 1.7285165786743164,
      "step": 45820
    },
    {
      "dynamics_loss_avg": 0.06835265532135963,
      "epoch": 4.3697357177734375e-05,
      "step": 45820
    },
    {
      "action_loss_avg": 0.0023101218917872756,
      "epoch": 4.3697357177734375e-05,
      "step": 45820
    },
    {
      "epoch": 4.3697357177734375e-05,
      "step": 45820,
      "training_step_time": 10.251036405563354
    },
    {
      "epoch": 4.369831085205078e-05,
      "model_forward_time": 1.7241272926330566,
      "step": 45821
    },
    {
      "epoch": 4.369831085205078e-05,
      "step": 45821,
      "training_step_time": 7.760340690612793
    },
    {
      "epoch": 4.369926452636719e-05,
      "model_forward_time": 2.908527374267578,
      "step": 45822
    },
    {
      "epoch": 4.369926452636719e-05,
      "step": 45822,
      "training_step_time": 7.674483776092529
    },
    {
      "epoch": 4.3700218200683594e-05,
      "model_forward_time": 1.7290973663330078,
      "step": 45823
    },
    {
      "epoch": 4.3700218200683594e-05,
      "step": 45823,
      "training_step_time": 6.79839301109314
    },
    {
      "epoch": 4.3701171875e-05,
      "model_forward_time": 1.7272636890411377,
      "step": 45824
    },
    {
      "epoch": 4.3701171875e-05,
      "step": 45824,
      "training_step_time": 6.7748894691467285
    },
    {
      "epoch": 4.370212554931641e-05,
      "model_forward_time": 1.7295594215393066,
      "step": 45825
    },
    {
      "epoch": 4.370212554931641e-05,
      "step": 45825,
      "training_step_time": 6.745530605316162
    },
    {
      "epoch": 4.370307922363281e-05,
      "model_forward_time": 1.7250518798828125,
      "step": 45826
    },
    {
      "epoch": 4.370307922363281e-05,
      "step": 45826,
      "training_step_time": 6.765450954437256
    },
    {
      "epoch": 4.370403289794922e-05,
      "model_forward_time": 1.7284011840820312,
      "step": 45827
    },
    {
      "epoch": 4.370403289794922e-05,
      "step": 45827,
      "training_step_time": 6.810899257659912
    },
    {
      "epoch": 4.370498657226563e-05,
      "model_forward_time": 1.787505865097046,
      "step": 45828
    },
    {
      "epoch": 4.370498657226563e-05,
      "step": 45828,
      "training_step_time": 8.562134981155396
    },
    {
      "epoch": 4.370594024658203e-05,
      "model_forward_time": 1.7242708206176758,
      "step": 45829
    },
    {
      "epoch": 4.370594024658203e-05,
      "step": 45829,
      "training_step_time": 8.827270746231079
    },
    {
      "epoch": 4.3706893920898436e-05,
      "grad_norm": 0.09559579938650131,
      "learning_rate": 6.094104606411874e-06,
      "loss": 0.0612,
      "step": 45830
    },
    {
      "epoch": 4.3706893920898436e-05,
      "model_forward_time": 1.722632884979248,
      "step": 45830
    },
    {
      "dynamics_loss_avg": 0.07042398061603308,
      "epoch": 4.3706893920898436e-05,
      "step": 45830
    },
    {
      "action_loss_avg": 0.00197266103932634,
      "epoch": 4.3706893920898436e-05,
      "step": 45830
    },
    {
      "epoch": 4.3706893920898436e-05,
      "step": 45830,
      "training_step_time": 7.750563621520996
    },
    {
      "epoch": 4.370784759521485e-05,
      "model_forward_time": 2.866589069366455,
      "step": 45831
    },
    {
      "epoch": 4.370784759521485e-05,
      "step": 45831,
      "training_step_time": 7.632196426391602
    },
    {
      "epoch": 4.370880126953125e-05,
      "model_forward_time": 1.7334694862365723,
      "step": 45832
    },
    {
      "epoch": 4.370880126953125e-05,
      "step": 45832,
      "training_step_time": 6.772633790969849
    },
    {
      "epoch": 4.3709754943847655e-05,
      "model_forward_time": 1.725799560546875,
      "step": 45833
    },
    {
      "epoch": 4.3709754943847655e-05,
      "step": 45833,
      "training_step_time": 6.822817325592041
    },
    {
      "epoch": 4.3710708618164066e-05,
      "model_forward_time": 1.725987195968628,
      "step": 45834
    },
    {
      "epoch": 4.3710708618164066e-05,
      "step": 45834,
      "training_step_time": 6.801664113998413
    },
    {
      "epoch": 4.371166229248047e-05,
      "model_forward_time": 1.7488689422607422,
      "step": 45835
    },
    {
      "epoch": 4.371166229248047e-05,
      "step": 45835,
      "training_step_time": 6.853031158447266
    },
    {
      "epoch": 4.3712615966796874e-05,
      "model_forward_time": 1.7236559391021729,
      "step": 45836
    },
    {
      "epoch": 4.3712615966796874e-05,
      "step": 45836,
      "training_step_time": 6.862654447555542
    },
    {
      "epoch": 4.3713569641113284e-05,
      "model_forward_time": 1.7833285331726074,
      "step": 45837
    },
    {
      "epoch": 4.3713569641113284e-05,
      "step": 45837,
      "training_step_time": 7.7901976108551025
    },
    {
      "epoch": 4.371452331542969e-05,
      "model_forward_time": 1.7186596393585205,
      "step": 45838
    },
    {
      "epoch": 4.371452331542969e-05,
      "step": 45838,
      "training_step_time": 11.138447999954224
    },
    {
      "epoch": 4.371547698974609e-05,
      "model_forward_time": 1.728266716003418,
      "step": 45839
    },
    {
      "epoch": 4.371547698974609e-05,
      "step": 45839,
      "training_step_time": 8.666832685470581
    },
    {
      "epoch": 4.37164306640625e-05,
      "grad_norm": 0.12442279607057571,
      "learning_rate": 6.092491148581197e-06,
      "loss": 0.0613,
      "step": 45840
    },
    {
      "epoch": 4.37164306640625e-05,
      "model_forward_time": 2.4959230422973633,
      "step": 45840
    },
    {
      "dynamics_loss_avg": 0.05999723840504885,
      "epoch": 4.37164306640625e-05,
      "step": 45840
    },
    {
      "action_loss_avg": 0.00192839577794075,
      "epoch": 4.37164306640625e-05,
      "step": 45840
    },
    {
      "epoch": 4.37164306640625e-05,
      "step": 45840,
      "training_step_time": 8.328214883804321
    },
    {
      "epoch": 4.371738433837891e-05,
      "model_forward_time": 1.7299160957336426,
      "step": 45841
    },
    {
      "epoch": 4.371738433837891e-05,
      "step": 45841,
      "training_step_time": 6.788243293762207
    },
    {
      "epoch": 4.371833801269531e-05,
      "model_forward_time": 1.7266910076141357,
      "step": 45842
    },
    {
      "epoch": 4.371833801269531e-05,
      "step": 45842,
      "training_step_time": 6.828627824783325
    },
    {
      "epoch": 4.371929168701172e-05,
      "model_forward_time": 1.7174999713897705,
      "step": 45843
    },
    {
      "epoch": 4.371929168701172e-05,
      "step": 45843,
      "training_step_time": 6.810574054718018
    },
    {
      "epoch": 4.3720245361328126e-05,
      "model_forward_time": 1.729020357131958,
      "step": 45844
    },
    {
      "epoch": 4.3720245361328126e-05,
      "step": 45844,
      "training_step_time": 6.8218302726745605
    },
    {
      "epoch": 4.372119903564453e-05,
      "model_forward_time": 1.7205147743225098,
      "step": 45845
    },
    {
      "epoch": 4.372119903564453e-05,
      "step": 45845,
      "training_step_time": 6.8337180614471436
    },
    {
      "epoch": 4.3722152709960934e-05,
      "model_forward_time": 1.7994701862335205,
      "step": 45846
    },
    {
      "epoch": 4.3722152709960934e-05,
      "step": 45846,
      "training_step_time": 7.728201150894165
    },
    {
      "epoch": 4.3723106384277345e-05,
      "model_forward_time": 1.736316204071045,
      "step": 45847
    },
    {
      "epoch": 4.3723106384277345e-05,
      "step": 45847,
      "training_step_time": 8.178152561187744
    },
    {
      "epoch": 4.372406005859375e-05,
      "model_forward_time": 1.7248353958129883,
      "step": 45848
    },
    {
      "epoch": 4.372406005859375e-05,
      "step": 45848,
      "training_step_time": 9.937514781951904
    },
    {
      "epoch": 4.372501373291015e-05,
      "model_forward_time": 2.819758415222168,
      "step": 45849
    },
    {
      "epoch": 4.372501373291015e-05,
      "step": 45849,
      "training_step_time": 7.726773500442505
    },
    {
      "epoch": 4.3725967407226564e-05,
      "grad_norm": 0.0982104167342186,
      "learning_rate": 6.090877571277332e-06,
      "loss": 0.0585,
      "step": 45850
    },
    {
      "epoch": 4.3725967407226564e-05,
      "model_forward_time": 1.7348976135253906,
      "step": 45850
    },
    {
      "dynamics_loss_avg": 0.04591566994786263,
      "epoch": 4.3725967407226564e-05,
      "step": 45850
    },
    {
      "action_loss_avg": 0.0017499750887509436,
      "epoch": 4.3725967407226564e-05,
      "step": 45850
    },
    {
      "epoch": 4.3725967407226564e-05,
      "step": 45850,
      "training_step_time": 7.373576641082764
    },
    {
      "epoch": 4.372692108154297e-05,
      "model_forward_time": 1.732353925704956,
      "step": 45851
    },
    {
      "epoch": 4.372692108154297e-05,
      "step": 45851,
      "training_step_time": 6.814270257949829
    },
    {
      "epoch": 4.372787475585937e-05,
      "model_forward_time": 1.728430986404419,
      "step": 45852
    },
    {
      "epoch": 4.372787475585937e-05,
      "step": 45852,
      "training_step_time": 6.879589319229126
    },
    {
      "epoch": 4.372882843017578e-05,
      "model_forward_time": 1.7467842102050781,
      "step": 45853
    },
    {
      "epoch": 4.372882843017578e-05,
      "step": 45853,
      "training_step_time": 6.822924375534058
    },
    {
      "epoch": 4.372978210449219e-05,
      "model_forward_time": 1.736847162246704,
      "step": 45854
    },
    {
      "epoch": 4.372978210449219e-05,
      "step": 45854,
      "training_step_time": 6.875265598297119
    },
    {
      "epoch": 4.373073577880859e-05,
      "model_forward_time": 1.7970795631408691,
      "step": 45855
    },
    {
      "epoch": 4.373073577880859e-05,
      "step": 45855,
      "training_step_time": 7.764775037765503
    },
    {
      "epoch": 4.3731689453125e-05,
      "model_forward_time": 1.7368075847625732,
      "step": 45856
    },
    {
      "epoch": 4.3731689453125e-05,
      "step": 45856,
      "training_step_time": 7.6651930809021
    },
    {
      "epoch": 4.3732643127441406e-05,
      "model_forward_time": 1.7281498908996582,
      "step": 45857
    },
    {
      "epoch": 4.3732643127441406e-05,
      "step": 45857,
      "training_step_time": 9.463518857955933
    },
    {
      "epoch": 4.373359680175781e-05,
      "model_forward_time": 1.7328648567199707,
      "step": 45858
    },
    {
      "epoch": 4.373359680175781e-05,
      "step": 45858,
      "training_step_time": 7.751810073852539
    },
    {
      "epoch": 4.373455047607422e-05,
      "model_forward_time": 2.9478378295898438,
      "step": 45859
    },
    {
      "epoch": 4.373455047607422e-05,
      "step": 45859,
      "training_step_time": 7.703514099121094
    },
    {
      "epoch": 4.3735504150390624e-05,
      "grad_norm": 0.08827420324087143,
      "learning_rate": 6.0892638746767365e-06,
      "loss": 0.0583,
      "step": 45860
    },
    {
      "epoch": 4.3735504150390624e-05,
      "model_forward_time": 1.7270264625549316,
      "step": 45860
    },
    {
      "dynamics_loss_avg": 0.051493226736783984,
      "epoch": 4.3735504150390624e-05,
      "step": 45860
    },
    {
      "action_loss_avg": 0.0014963338064262644,
      "epoch": 4.3735504150390624e-05,
      "step": 45860
    },
    {
      "epoch": 4.3735504150390624e-05,
      "step": 45860,
      "training_step_time": 6.7516632080078125
    },
    {
      "epoch": 4.373645782470703e-05,
      "model_forward_time": 1.744694471359253,
      "step": 45861
    },
    {
      "epoch": 4.373645782470703e-05,
      "step": 45861,
      "training_step_time": 6.8314759731292725
    },
    {
      "epoch": 4.373741149902344e-05,
      "model_forward_time": 1.7180466651916504,
      "step": 45862
    },
    {
      "epoch": 4.373741149902344e-05,
      "step": 45862,
      "training_step_time": 6.813957214355469
    },
    {
      "epoch": 4.373836517333984e-05,
      "model_forward_time": 1.7186264991760254,
      "step": 45863
    },
    {
      "epoch": 4.373836517333984e-05,
      "step": 45863,
      "training_step_time": 6.814018964767456
    },
    {
      "epoch": 4.373931884765625e-05,
      "model_forward_time": 1.4452157020568848,
      "step": 45864
    },
    {
      "epoch": 4.373931884765625e-05,
      "step": 45864,
      "training_step_time": 7.743253469467163
    },
    {
      "epoch": 4.374027252197266e-05,
      "model_forward_time": 1.7293446063995361,
      "step": 45865
    },
    {
      "epoch": 4.374027252197266e-05,
      "step": 45865,
      "training_step_time": 8.620693922042847
    },
    {
      "epoch": 4.374122619628906e-05,
      "model_forward_time": 1.7191972732543945,
      "step": 45866
    },
    {
      "epoch": 4.374122619628906e-05,
      "step": 45866,
      "training_step_time": 9.898441553115845
    },
    {
      "epoch": 4.3742179870605466e-05,
      "model_forward_time": 1.720266580581665,
      "step": 45867
    },
    {
      "epoch": 4.3742179870605466e-05,
      "step": 45867,
      "training_step_time": 8.768293619155884
    },
    {
      "epoch": 4.374313354492188e-05,
      "model_forward_time": 2.885180711746216,
      "step": 45868
    },
    {
      "epoch": 4.374313354492188e-05,
      "step": 45868,
      "training_step_time": 7.698405027389526
    },
    {
      "epoch": 4.374408721923828e-05,
      "model_forward_time": 1.3763725757598877,
      "step": 45869
    },
    {
      "epoch": 4.374408721923828e-05,
      "step": 45869,
      "training_step_time": 6.8639445304870605
    },
    {
      "epoch": 4.3745040893554685e-05,
      "grad_norm": 0.10653078556060791,
      "learning_rate": 6.087650058955879e-06,
      "loss": 0.0636,
      "step": 45870
    },
    {
      "epoch": 4.3745040893554685e-05,
      "model_forward_time": 1.7164580821990967,
      "step": 45870
    },
    {
      "dynamics_loss_avg": 0.05922690872102976,
      "epoch": 4.3745040893554685e-05,
      "step": 45870
    },
    {
      "action_loss_avg": 0.001955939663457684,
      "epoch": 4.3745040893554685e-05,
      "step": 45870
    },
    {
      "epoch": 4.3745040893554685e-05,
      "step": 45870,
      "training_step_time": 6.858644962310791
    },
    {
      "epoch": 4.3745994567871096e-05,
      "model_forward_time": 1.7317173480987549,
      "step": 45871
    },
    {
      "epoch": 4.3745994567871096e-05,
      "step": 45871,
      "training_step_time": 6.911425352096558
    },
    {
      "epoch": 4.37469482421875e-05,
      "model_forward_time": 1.768010139465332,
      "step": 45872
    },
    {
      "epoch": 4.37469482421875e-05,
      "step": 45872,
      "training_step_time": 6.856179714202881
    },
    {
      "epoch": 4.3747901916503904e-05,
      "model_forward_time": 1.7936668395996094,
      "step": 45873
    },
    {
      "epoch": 4.3747901916503904e-05,
      "step": 45873,
      "training_step_time": 7.678588390350342
    },
    {
      "epoch": 4.3748855590820315e-05,
      "model_forward_time": 1.7420828342437744,
      "step": 45874
    },
    {
      "epoch": 4.3748855590820315e-05,
      "step": 45874,
      "training_step_time": 7.607353210449219
    },
    {
      "epoch": 4.374980926513672e-05,
      "model_forward_time": 1.7224140167236328,
      "step": 45875
    },
    {
      "epoch": 4.374980926513672e-05,
      "step": 45875,
      "training_step_time": 9.288686990737915
    },
    {
      "epoch": 4.375076293945312e-05,
      "model_forward_time": 1.742701530456543,
      "step": 45876
    },
    {
      "epoch": 4.375076293945312e-05,
      "step": 45876,
      "training_step_time": 8.786841630935669
    },
    {
      "epoch": 4.3751716613769534e-05,
      "model_forward_time": 2.8512985706329346,
      "step": 45877
    },
    {
      "epoch": 4.3751716613769534e-05,
      "step": 45877,
      "training_step_time": 7.664499044418335
    },
    {
      "epoch": 4.375267028808594e-05,
      "model_forward_time": 1.734895944595337,
      "step": 45878
    },
    {
      "epoch": 4.375267028808594e-05,
      "step": 45878,
      "training_step_time": 6.836199998855591
    },
    {
      "epoch": 4.375362396240234e-05,
      "model_forward_time": 1.7396600246429443,
      "step": 45879
    },
    {
      "epoch": 4.375362396240234e-05,
      "step": 45879,
      "training_step_time": 6.79941725730896
    },
    {
      "epoch": 4.375457763671875e-05,
      "grad_norm": 0.11099331080913544,
      "learning_rate": 6.086036124291248e-06,
      "loss": 0.0607,
      "step": 45880
    },
    {
      "epoch": 4.375457763671875e-05,
      "model_forward_time": 1.7176659107208252,
      "step": 45880
    },
    {
      "dynamics_loss_avg": 0.05342363100498915,
      "epoch": 4.375457763671875e-05,
      "step": 45880
    },
    {
      "action_loss_avg": 0.0016173352778423578,
      "epoch": 4.375457763671875e-05,
      "step": 45880
    },
    {
      "epoch": 4.375457763671875e-05,
      "step": 45880,
      "training_step_time": 6.848912715911865
    },
    {
      "epoch": 4.3755531311035156e-05,
      "model_forward_time": 1.7298929691314697,
      "step": 45881
    },
    {
      "epoch": 4.3755531311035156e-05,
      "step": 45881,
      "training_step_time": 6.844805717468262
    },
    {
      "epoch": 4.375648498535156e-05,
      "model_forward_time": 1.7712929248809814,
      "step": 45882
    },
    {
      "epoch": 4.375648498535156e-05,
      "step": 45882,
      "training_step_time": 7.473556041717529
    },
    {
      "epoch": 4.375743865966797e-05,
      "model_forward_time": 1.3629920482635498,
      "step": 45883
    },
    {
      "epoch": 4.375743865966797e-05,
      "step": 45883,
      "training_step_time": 8.46897029876709
    },
    {
      "epoch": 4.3758392333984375e-05,
      "model_forward_time": 1.7393653392791748,
      "step": 45884
    },
    {
      "epoch": 4.3758392333984375e-05,
      "step": 45884,
      "training_step_time": 10.013503789901733
    },
    {
      "epoch": 4.375934600830078e-05,
      "model_forward_time": 1.7387278079986572,
      "step": 45885
    },
    {
      "epoch": 4.375934600830078e-05,
      "step": 45885,
      "training_step_time": 8.748805522918701
    },
    {
      "epoch": 4.376029968261719e-05,
      "model_forward_time": 1.7452442646026611,
      "step": 45886
    },
    {
      "epoch": 4.376029968261719e-05,
      "step": 45886,
      "training_step_time": 7.783257961273193
    },
    {
      "epoch": 4.3761253356933594e-05,
      "model_forward_time": 2.9055838584899902,
      "step": 45887
    },
    {
      "epoch": 4.3761253356933594e-05,
      "step": 45887,
      "training_step_time": 7.760219097137451
    },
    {
      "epoch": 4.376220703125e-05,
      "model_forward_time": 1.7366406917572021,
      "step": 45888
    },
    {
      "epoch": 4.376220703125e-05,
      "step": 45888,
      "training_step_time": 6.843346357345581
    },
    {
      "epoch": 4.376316070556641e-05,
      "model_forward_time": 1.7405447959899902,
      "step": 45889
    },
    {
      "epoch": 4.376316070556641e-05,
      "step": 45889,
      "training_step_time": 6.880114555358887
    },
    {
      "epoch": 4.376411437988281e-05,
      "grad_norm": 0.16371503472328186,
      "learning_rate": 6.084422070859339e-06,
      "loss": 0.0601,
      "step": 45890
    },
    {
      "epoch": 4.376411437988281e-05,
      "model_forward_time": 1.757493019104004,
      "step": 45890
    },
    {
      "dynamics_loss_avg": 0.058532442059367897,
      "epoch": 4.376411437988281e-05,
      "step": 45890
    },
    {
      "action_loss_avg": 0.0026210566342342646,
      "epoch": 4.376411437988281e-05,
      "step": 45890
    },
    {
      "epoch": 4.376411437988281e-05,
      "step": 45890,
      "training_step_time": 6.861449718475342
    },
    {
      "epoch": 4.376506805419922e-05,
      "model_forward_time": 1.7409143447875977,
      "step": 45891
    },
    {
      "epoch": 4.376506805419922e-05,
      "step": 45891,
      "training_step_time": 7.512117862701416
    },
    {
      "epoch": 4.376602172851563e-05,
      "model_forward_time": 1.4388041496276855,
      "step": 45892
    },
    {
      "epoch": 4.376602172851563e-05,
      "step": 45892,
      "training_step_time": 7.683003664016724
    },
    {
      "epoch": 4.376697540283203e-05,
      "model_forward_time": 1.7343385219573975,
      "step": 45893
    },
    {
      "epoch": 4.376697540283203e-05,
      "step": 45893,
      "training_step_time": 9.230940103530884
    },
    {
      "epoch": 4.3767929077148436e-05,
      "model_forward_time": 1.721801996231079,
      "step": 45894
    },
    {
      "epoch": 4.3767929077148436e-05,
      "step": 45894,
      "training_step_time": 8.768781185150146
    },
    {
      "epoch": 4.376888275146485e-05,
      "model_forward_time": 1.7746281623840332,
      "step": 45895
    },
    {
      "epoch": 4.376888275146485e-05,
      "step": 45895,
      "training_step_time": 8.808618307113647
    },
    {
      "epoch": 4.376983642578125e-05,
      "model_forward_time": 2.877279758453369,
      "step": 45896
    },
    {
      "epoch": 4.376983642578125e-05,
      "step": 45896,
      "training_step_time": 7.67864203453064
    },
    {
      "epoch": 4.3770790100097655e-05,
      "model_forward_time": 1.7263844013214111,
      "step": 45897
    },
    {
      "epoch": 4.3770790100097655e-05,
      "step": 45897,
      "training_step_time": 6.813600778579712
    },
    {
      "epoch": 4.3771743774414066e-05,
      "model_forward_time": 1.385225534439087,
      "step": 45898
    },
    {
      "epoch": 4.3771743774414066e-05,
      "step": 45898,
      "training_step_time": 6.843478441238403
    },
    {
      "epoch": 4.377269744873047e-05,
      "model_forward_time": 1.7388298511505127,
      "step": 45899
    },
    {
      "epoch": 4.377269744873047e-05,
      "step": 45899,
      "training_step_time": 6.79996657371521
    },
    {
      "epoch": 4.3773651123046874e-05,
      "grad_norm": 0.1523858904838562,
      "learning_rate": 6.082807898836664e-06,
      "loss": 0.0621,
      "step": 45900
    },
    {
      "epoch": 4.3773651123046874e-05,
      "model_forward_time": 1.7427613735198975,
      "step": 45900
    },
    {
      "dynamics_loss_avg": 0.07469633165746928,
      "epoch": 4.3773651123046874e-05,
      "step": 45900
    },
    {
      "action_loss_avg": 0.0030052041460294277,
      "epoch": 4.3773651123046874e-05,
      "step": 45900
    },
    {
      "epoch": 4.3773651123046874e-05,
      "step": 45900,
      "training_step_time": 7.543213367462158
    },
    {
      "epoch": 4.3774604797363284e-05,
      "model_forward_time": 1.746112585067749,
      "step": 45901
    },
    {
      "epoch": 4.3774604797363284e-05,
      "step": 45901,
      "training_step_time": 7.852787971496582
    },
    {
      "epoch": 4.377555847167969e-05,
      "model_forward_time": 1.3799018859863281,
      "step": 45902
    },
    {
      "epoch": 4.377555847167969e-05,
      "step": 45902,
      "training_step_time": 8.787161350250244
    },
    {
      "epoch": 4.377651214599609e-05,
      "model_forward_time": 1.729529857635498,
      "step": 45903
    },
    {
      "epoch": 4.377651214599609e-05,
      "step": 45903,
      "training_step_time": 9.977796077728271
    },
    {
      "epoch": 4.37774658203125e-05,
      "model_forward_time": 1.739227533340454,
      "step": 45904
    },
    {
      "epoch": 4.37774658203125e-05,
      "step": 45904,
      "training_step_time": 7.769165515899658
    },
    {
      "epoch": 4.377841949462891e-05,
      "model_forward_time": 2.800194263458252,
      "step": 45905
    },
    {
      "epoch": 4.377841949462891e-05,
      "step": 45905,
      "training_step_time": 7.734839916229248
    },
    {
      "epoch": 4.377937316894531e-05,
      "model_forward_time": 1.7766530513763428,
      "step": 45906
    },
    {
      "epoch": 4.377937316894531e-05,
      "step": 45906,
      "training_step_time": 6.81209397315979
    },
    {
      "epoch": 4.3780326843261715e-05,
      "model_forward_time": 1.7353131771087646,
      "step": 45907
    },
    {
      "epoch": 4.3780326843261715e-05,
      "step": 45907,
      "training_step_time": 6.831200122833252
    },
    {
      "epoch": 4.3781280517578126e-05,
      "model_forward_time": 1.7347643375396729,
      "step": 45908
    },
    {
      "epoch": 4.3781280517578126e-05,
      "step": 45908,
      "training_step_time": 6.840637922286987
    },
    {
      "epoch": 4.378223419189453e-05,
      "model_forward_time": 1.7353389263153076,
      "step": 45909
    },
    {
      "epoch": 4.378223419189453e-05,
      "step": 45909,
      "training_step_time": 7.540423154830933
    },
    {
      "epoch": 4.3783187866210934e-05,
      "grad_norm": 0.1655828058719635,
      "learning_rate": 6.081193608399743e-06,
      "loss": 0.0597,
      "step": 45910
    },
    {
      "epoch": 4.3783187866210934e-05,
      "model_forward_time": 1.3708734512329102,
      "step": 45910
    },
    {
      "dynamics_loss_avg": 0.05903854295611381,
      "epoch": 4.3783187866210934e-05,
      "step": 45910
    },
    {
      "action_loss_avg": 0.002144605590729043,
      "epoch": 4.3783187866210934e-05,
      "step": 45910
    },
    {
      "epoch": 4.3783187866210934e-05,
      "step": 45910,
      "training_step_time": 7.712483167648315
    },
    {
      "epoch": 4.3784141540527345e-05,
      "model_forward_time": 1.8172225952148438,
      "step": 45911
    },
    {
      "epoch": 4.3784141540527345e-05,
      "step": 45911,
      "training_step_time": 8.68211030960083
    },
    {
      "epoch": 4.378509521484375e-05,
      "model_forward_time": 1.7498540878295898,
      "step": 45912
    },
    {
      "epoch": 4.378509521484375e-05,
      "step": 45912,
      "training_step_time": 8.898007154464722
    },
    {
      "epoch": 4.378604888916015e-05,
      "model_forward_time": 1.7352371215820312,
      "step": 45913
    },
    {
      "epoch": 4.378604888916015e-05,
      "step": 45913,
      "training_step_time": 7.77616286277771
    },
    {
      "epoch": 4.3787002563476564e-05,
      "model_forward_time": 1.7270095348358154,
      "step": 45914
    },
    {
      "epoch": 4.3787002563476564e-05,
      "step": 45914,
      "training_step_time": 7.788711071014404
    },
    {
      "epoch": 4.378795623779297e-05,
      "model_forward_time": 2.912505865097046,
      "step": 45915
    },
    {
      "epoch": 4.378795623779297e-05,
      "step": 45915,
      "training_step_time": 7.70639181137085
    },
    {
      "epoch": 4.378890991210937e-05,
      "model_forward_time": 1.7437474727630615,
      "step": 45916
    },
    {
      "epoch": 4.378890991210937e-05,
      "step": 45916,
      "training_step_time": 6.824835777282715
    },
    {
      "epoch": 4.378986358642578e-05,
      "model_forward_time": 1.7412071228027344,
      "step": 45917
    },
    {
      "epoch": 4.378986358642578e-05,
      "step": 45917,
      "training_step_time": 6.8363916873931885
    },
    {
      "epoch": 4.379081726074219e-05,
      "model_forward_time": 1.7371182441711426,
      "step": 45918
    },
    {
      "epoch": 4.379081726074219e-05,
      "step": 45918,
      "training_step_time": 7.550470352172852
    },
    {
      "epoch": 4.379177093505859e-05,
      "model_forward_time": 1.7345869541168213,
      "step": 45919
    },
    {
      "epoch": 4.379177093505859e-05,
      "step": 45919,
      "training_step_time": 7.672492027282715
    },
    {
      "epoch": 4.3792724609375e-05,
      "grad_norm": 0.14923740923404694,
      "learning_rate": 6.079579199725116e-06,
      "loss": 0.0627,
      "step": 45920
    },
    {
      "epoch": 4.3792724609375e-05,
      "model_forward_time": 1.8124051094055176,
      "step": 45920
    },
    {
      "dynamics_loss_avg": 0.05641294978559017,
      "epoch": 4.3792724609375e-05,
      "step": 45920
    },
    {
      "action_loss_avg": 0.0010876926535274833,
      "epoch": 4.3792724609375e-05,
      "step": 45920
    },
    {
      "epoch": 4.3792724609375e-05,
      "step": 45920,
      "training_step_time": 9.021193742752075
    },
    {
      "epoch": 4.3793678283691406e-05,
      "model_forward_time": 1.7478690147399902,
      "step": 45921
    },
    {
      "epoch": 4.3793678283691406e-05,
      "step": 45921,
      "training_step_time": 9.007991790771484
    },
    {
      "epoch": 4.379463195800781e-05,
      "model_forward_time": 1.7304167747497559,
      "step": 45922
    },
    {
      "epoch": 4.379463195800781e-05,
      "step": 45922,
      "training_step_time": 8.849004745483398
    },
    {
      "epoch": 4.379558563232422e-05,
      "model_forward_time": 1.7359108924865723,
      "step": 45923
    },
    {
      "epoch": 4.379558563232422e-05,
      "step": 45923,
      "training_step_time": 7.687792062759399
    },
    {
      "epoch": 4.3796539306640625e-05,
      "model_forward_time": 2.5361316204071045,
      "step": 45924
    },
    {
      "epoch": 4.3796539306640625e-05,
      "step": 45924,
      "training_step_time": 7.330625534057617
    },
    {
      "epoch": 4.379749298095703e-05,
      "model_forward_time": 1.7472209930419922,
      "step": 45925
    },
    {
      "epoch": 4.379749298095703e-05,
      "step": 45925,
      "training_step_time": 6.84402060508728
    },
    {
      "epoch": 4.379844665527344e-05,
      "model_forward_time": 1.7200286388397217,
      "step": 45926
    },
    {
      "epoch": 4.379844665527344e-05,
      "step": 45926,
      "training_step_time": 6.840834856033325
    },
    {
      "epoch": 4.379940032958984e-05,
      "model_forward_time": 1.721266269683838,
      "step": 45927
    },
    {
      "epoch": 4.379940032958984e-05,
      "step": 45927,
      "training_step_time": 7.5366950035095215
    },
    {
      "epoch": 4.380035400390625e-05,
      "model_forward_time": 1.7228460311889648,
      "step": 45928
    },
    {
      "epoch": 4.380035400390625e-05,
      "step": 45928,
      "training_step_time": 7.387528419494629
    },
    {
      "epoch": 4.380130767822266e-05,
      "model_forward_time": 1.7477912902832031,
      "step": 45929
    },
    {
      "epoch": 4.380130767822266e-05,
      "step": 45929,
      "training_step_time": 8.47534704208374
    },
    {
      "epoch": 4.380226135253906e-05,
      "grad_norm": 0.11890362948179245,
      "learning_rate": 6.077964672989329e-06,
      "loss": 0.0614,
      "step": 45930
    },
    {
      "epoch": 4.380226135253906e-05,
      "model_forward_time": 1.719019889831543,
      "step": 45930
    },
    {
      "dynamics_loss_avg": 0.06369164865463972,
      "epoch": 4.380226135253906e-05,
      "step": 45930
    },
    {
      "action_loss_avg": 0.002424909407272935,
      "epoch": 4.380226135253906e-05,
      "step": 45930
    },
    {
      "epoch": 4.380226135253906e-05,
      "step": 45930,
      "training_step_time": 8.767844915390015
    },
    {
      "epoch": 4.3803215026855466e-05,
      "model_forward_time": 1.728241205215454,
      "step": 45931
    },
    {
      "epoch": 4.3803215026855466e-05,
      "step": 45931,
      "training_step_time": 9.233412265777588
    },
    {
      "epoch": 4.380416870117188e-05,
      "model_forward_time": 1.7151405811309814,
      "step": 45932
    },
    {
      "epoch": 4.380416870117188e-05,
      "step": 45932,
      "training_step_time": 9.717207908630371
    },
    {
      "epoch": 4.380512237548828e-05,
      "model_forward_time": 2.8573732376098633,
      "step": 45933
    },
    {
      "epoch": 4.380512237548828e-05,
      "step": 45933,
      "training_step_time": 7.727709531784058
    },
    {
      "epoch": 4.3806076049804685e-05,
      "model_forward_time": 1.7165062427520752,
      "step": 45934
    },
    {
      "epoch": 4.3806076049804685e-05,
      "step": 45934,
      "training_step_time": 6.823962688446045
    },
    {
      "epoch": 4.3807029724121096e-05,
      "model_forward_time": 1.7173352241516113,
      "step": 45935
    },
    {
      "epoch": 4.3807029724121096e-05,
      "step": 45935,
      "training_step_time": 6.8298962116241455
    },
    {
      "epoch": 4.38079833984375e-05,
      "model_forward_time": 1.354130744934082,
      "step": 45936
    },
    {
      "epoch": 4.38079833984375e-05,
      "step": 45936,
      "training_step_time": 7.621665716171265
    },
    {
      "epoch": 4.3808937072753904e-05,
      "model_forward_time": 1.7205836772918701,
      "step": 45937
    },
    {
      "epoch": 4.3808937072753904e-05,
      "step": 45937,
      "training_step_time": 7.460777759552002
    },
    {
      "epoch": 4.3809890747070315e-05,
      "model_forward_time": 1.7191152572631836,
      "step": 45938
    },
    {
      "epoch": 4.3809890747070315e-05,
      "step": 45938,
      "training_step_time": 7.723726511001587
    },
    {
      "epoch": 4.381084442138672e-05,
      "model_forward_time": 1.7268168926239014,
      "step": 45939
    },
    {
      "epoch": 4.381084442138672e-05,
      "step": 45939,
      "training_step_time": 9.20215892791748
    },
    {
      "epoch": 4.381179809570312e-05,
      "grad_norm": 0.10634008795022964,
      "learning_rate": 6.0763500283689484e-06,
      "loss": 0.0585,
      "step": 45940
    },
    {
      "epoch": 4.381179809570312e-05,
      "model_forward_time": 1.71638822555542,
      "step": 45940
    },
    {
      "dynamics_loss_avg": 0.055616606026887894,
      "epoch": 4.381179809570312e-05,
      "step": 45940
    },
    {
      "action_loss_avg": 0.002201060706283897,
      "epoch": 4.381179809570312e-05,
      "step": 45940
    },
    {
      "epoch": 4.381179809570312e-05,
      "step": 45940,
      "training_step_time": 8.725730419158936
    },
    {
      "epoch": 4.3812751770019534e-05,
      "model_forward_time": 1.7094993591308594,
      "step": 45941
    },
    {
      "epoch": 4.3812751770019534e-05,
      "step": 45941,
      "training_step_time": 7.845977067947388
    },
    {
      "epoch": 4.381370544433594e-05,
      "model_forward_time": 1.719475269317627,
      "step": 45942
    },
    {
      "epoch": 4.381370544433594e-05,
      "step": 45942,
      "training_step_time": 7.761408805847168
    },
    {
      "epoch": 4.381465911865234e-05,
      "model_forward_time": 2.909301996231079,
      "step": 45943
    },
    {
      "epoch": 4.381465911865234e-05,
      "step": 45943,
      "training_step_time": 7.7380852699279785
    },
    {
      "epoch": 4.381561279296875e-05,
      "model_forward_time": 1.72501540184021,
      "step": 45944
    },
    {
      "epoch": 4.381561279296875e-05,
      "step": 45944,
      "training_step_time": 6.761615514755249
    },
    {
      "epoch": 4.3816566467285157e-05,
      "model_forward_time": 1.7311820983886719,
      "step": 45945
    },
    {
      "epoch": 4.3816566467285157e-05,
      "step": 45945,
      "training_step_time": 7.540931940078735
    },
    {
      "epoch": 4.381752014160156e-05,
      "model_forward_time": 1.7231669425964355,
      "step": 45946
    },
    {
      "epoch": 4.381752014160156e-05,
      "step": 45946,
      "training_step_time": 7.393948793411255
    },
    {
      "epoch": 4.381847381591797e-05,
      "model_forward_time": 1.7171294689178467,
      "step": 45947
    },
    {
      "epoch": 4.381847381591797e-05,
      "step": 45947,
      "training_step_time": 7.709985971450806
    },
    {
      "epoch": 4.3819427490234375e-05,
      "model_forward_time": 1.7379155158996582,
      "step": 45948
    },
    {
      "epoch": 4.3819427490234375e-05,
      "step": 45948,
      "training_step_time": 7.752754211425781
    },
    {
      "epoch": 4.382038116455078e-05,
      "model_forward_time": 1.7233202457427979,
      "step": 45949
    },
    {
      "epoch": 4.382038116455078e-05,
      "step": 45949,
      "training_step_time": 8.289275169372559
    },
    {
      "epoch": 4.382133483886719e-05,
      "grad_norm": 0.10196837037801743,
      "learning_rate": 6.074735266040546e-06,
      "loss": 0.0568,
      "step": 45950
    },
    {
      "epoch": 4.382133483886719e-05,
      "model_forward_time": 1.7180476188659668,
      "step": 45950
    },
    {
      "dynamics_loss_avg": 0.05879395287483931,
      "epoch": 4.382133483886719e-05,
      "step": 45950
    },
    {
      "action_loss_avg": 0.002656087791547179,
      "epoch": 4.382133483886719e-05,
      "step": 45950
    },
    {
      "epoch": 4.382133483886719e-05,
      "step": 45950,
      "training_step_time": 8.85514497756958
    },
    {
      "epoch": 4.3822288513183594e-05,
      "model_forward_time": 1.7241902351379395,
      "step": 45951
    },
    {
      "epoch": 4.3822288513183594e-05,
      "step": 45951,
      "training_step_time": 8.835865497589111
    },
    {
      "epoch": 4.38232421875e-05,
      "model_forward_time": 2.8807601928710938,
      "step": 45952
    },
    {
      "epoch": 4.38232421875e-05,
      "step": 45952,
      "training_step_time": 7.7359185218811035
    },
    {
      "epoch": 4.382419586181641e-05,
      "model_forward_time": 1.7312068939208984,
      "step": 45953
    },
    {
      "epoch": 4.382419586181641e-05,
      "step": 45953,
      "training_step_time": 6.858741760253906
    },
    {
      "epoch": 4.382514953613281e-05,
      "model_forward_time": 1.7251746654510498,
      "step": 45954
    },
    {
      "epoch": 4.382514953613281e-05,
      "step": 45954,
      "training_step_time": 7.612179279327393
    },
    {
      "epoch": 4.382610321044922e-05,
      "model_forward_time": 1.7495617866516113,
      "step": 45955
    },
    {
      "epoch": 4.382610321044922e-05,
      "step": 45955,
      "training_step_time": 7.540509223937988
    },
    {
      "epoch": 4.382705688476563e-05,
      "model_forward_time": 1.7259695529937744,
      "step": 45956
    },
    {
      "epoch": 4.382705688476563e-05,
      "step": 45956,
      "training_step_time": 8.301946640014648
    },
    {
      "epoch": 4.382801055908203e-05,
      "model_forward_time": 1.7397770881652832,
      "step": 45957
    },
    {
      "epoch": 4.382801055908203e-05,
      "step": 45957,
      "training_step_time": 8.660664558410645
    },
    {
      "epoch": 4.3828964233398436e-05,
      "model_forward_time": 1.7208280563354492,
      "step": 45958
    },
    {
      "epoch": 4.3828964233398436e-05,
      "step": 45958,
      "training_step_time": 9.537471532821655
    },
    {
      "epoch": 4.382991790771485e-05,
      "model_forward_time": 1.721728801727295,
      "step": 45959
    },
    {
      "epoch": 4.382991790771485e-05,
      "step": 45959,
      "training_step_time": 9.987962484359741
    },
    {
      "epoch": 4.383087158203125e-05,
      "grad_norm": 0.09745568037033081,
      "learning_rate": 6.07312038618071e-06,
      "loss": 0.061,
      "step": 45960
    },
    {
      "epoch": 4.383087158203125e-05,
      "model_forward_time": 1.7199127674102783,
      "step": 45960
    },
    {
      "dynamics_loss_avg": 0.05364945800974965,
      "epoch": 4.383087158203125e-05,
      "step": 45960
    },
    {
      "action_loss_avg": 0.0019187742727808654,
      "epoch": 4.383087158203125e-05,
      "step": 45960
    },
    {
      "epoch": 4.383087158203125e-05,
      "step": 45960,
      "training_step_time": 7.57313346862793
    },
    {
      "epoch": 4.3831825256347655e-05,
      "model_forward_time": 2.86141037940979,
      "step": 45961
    },
    {
      "epoch": 4.3831825256347655e-05,
      "step": 45961,
      "training_step_time": 7.693236351013184
    },
    {
      "epoch": 4.3832778930664066e-05,
      "model_forward_time": 1.7216854095458984,
      "step": 45962
    },
    {
      "epoch": 4.3832778930664066e-05,
      "step": 45962,
      "training_step_time": 6.902694225311279
    },
    {
      "epoch": 4.383373260498047e-05,
      "model_forward_time": 1.7211146354675293,
      "step": 45963
    },
    {
      "epoch": 4.383373260498047e-05,
      "step": 45963,
      "training_step_time": 7.599600553512573
    },
    {
      "epoch": 4.3834686279296874e-05,
      "model_forward_time": 1.746480941772461,
      "step": 45964
    },
    {
      "epoch": 4.3834686279296874e-05,
      "step": 45964,
      "training_step_time": 7.50734806060791
    },
    {
      "epoch": 4.3835639953613285e-05,
      "model_forward_time": 1.7251191139221191,
      "step": 45965
    },
    {
      "epoch": 4.3835639953613285e-05,
      "step": 45965,
      "training_step_time": 8.285858154296875
    },
    {
      "epoch": 4.383659362792969e-05,
      "model_forward_time": 1.7309060096740723,
      "step": 45966
    },
    {
      "epoch": 4.383659362792969e-05,
      "step": 45966,
      "training_step_time": 8.725192070007324
    },
    {
      "epoch": 4.383754730224609e-05,
      "model_forward_time": 1.715994119644165,
      "step": 45967
    },
    {
      "epoch": 4.383754730224609e-05,
      "step": 45967,
      "training_step_time": 7.682177305221558
    },
    {
      "epoch": 4.38385009765625e-05,
      "model_forward_time": 1.7273714542388916,
      "step": 45968
    },
    {
      "epoch": 4.38385009765625e-05,
      "step": 45968,
      "training_step_time": 7.764752626419067
    },
    {
      "epoch": 4.383945465087891e-05,
      "model_forward_time": 1.7217833995819092,
      "step": 45969
    },
    {
      "epoch": 4.383945465087891e-05,
      "step": 45969,
      "training_step_time": 7.657166242599487
    },
    {
      "epoch": 4.384040832519531e-05,
      "grad_norm": 0.10657971352338791,
      "learning_rate": 6.071505388966043e-06,
      "loss": 0.0555,
      "step": 45970
    },
    {
      "epoch": 4.384040832519531e-05,
      "model_forward_time": 2.867704391479492,
      "step": 45970
    },
    {
      "dynamics_loss_avg": 0.04473503995686769,
      "epoch": 4.384040832519531e-05,
      "step": 45970
    },
    {
      "action_loss_avg": 0.0014660819491837175,
      "epoch": 4.384040832519531e-05,
      "step": 45970
    },
    {
      "epoch": 4.384040832519531e-05,
      "step": 45970,
      "training_step_time": 7.703593730926514
    },
    {
      "epoch": 4.3841361999511715e-05,
      "model_forward_time": 1.723109245300293,
      "step": 45971
    },
    {
      "epoch": 4.3841361999511715e-05,
      "step": 45971,
      "training_step_time": 6.888835906982422
    },
    {
      "epoch": 4.3842315673828126e-05,
      "model_forward_time": 1.7242846488952637,
      "step": 45972
    },
    {
      "epoch": 4.3842315673828126e-05,
      "step": 45972,
      "training_step_time": 7.620033025741577
    },
    {
      "epoch": 4.384326934814453e-05,
      "model_forward_time": 1.7444617748260498,
      "step": 45973
    },
    {
      "epoch": 4.384326934814453e-05,
      "step": 45973,
      "training_step_time": 7.393420696258545
    },
    {
      "epoch": 4.3844223022460934e-05,
      "model_forward_time": 1.7246067523956299,
      "step": 45974
    },
    {
      "epoch": 4.3844223022460934e-05,
      "step": 45974,
      "training_step_time": 7.626421213150024
    },
    {
      "epoch": 4.3845176696777345e-05,
      "model_forward_time": 1.742459774017334,
      "step": 45975
    },
    {
      "epoch": 4.3845176696777345e-05,
      "step": 45975,
      "training_step_time": 8.27072787284851
    },
    {
      "epoch": 4.384613037109375e-05,
      "model_forward_time": 1.7243618965148926,
      "step": 45976
    },
    {
      "epoch": 4.384613037109375e-05,
      "step": 45976,
      "training_step_time": 7.743540048599243
    },
    {
      "epoch": 4.384708404541015e-05,
      "model_forward_time": 1.7279272079467773,
      "step": 45977
    },
    {
      "epoch": 4.384708404541015e-05,
      "step": 45977,
      "training_step_time": 9.165945768356323
    },
    {
      "epoch": 4.3848037719726564e-05,
      "model_forward_time": 1.7135989665985107,
      "step": 45978
    },
    {
      "epoch": 4.3848037719726564e-05,
      "step": 45978,
      "training_step_time": 7.743899822235107
    },
    {
      "epoch": 4.384899139404297e-05,
      "model_forward_time": 2.8400275707244873,
      "step": 45979
    },
    {
      "epoch": 4.384899139404297e-05,
      "step": 45979,
      "training_step_time": 7.679948091506958
    },
    {
      "epoch": 4.384994506835937e-05,
      "grad_norm": 0.10606015473604202,
      "learning_rate": 6.069890274573157e-06,
      "loss": 0.0603,
      "step": 45980
    },
    {
      "epoch": 4.384994506835937e-05,
      "model_forward_time": 1.0249712467193604,
      "step": 45980
    },
    {
      "dynamics_loss_avg": 0.05400649700313807,
      "epoch": 4.384994506835937e-05,
      "step": 45980
    },
    {
      "action_loss_avg": 0.0023462630168069156,
      "epoch": 4.384994506835937e-05,
      "step": 45980
    },
    {
      "epoch": 4.384994506835937e-05,
      "step": 45980,
      "training_step_time": 7.230716943740845
    },
    {
      "epoch": 4.385089874267578e-05,
      "model_forward_time": 1.7198646068572998,
      "step": 45981
    },
    {
      "epoch": 4.385089874267578e-05,
      "step": 45981,
      "training_step_time": 7.579947471618652
    },
    {
      "epoch": 4.385185241699219e-05,
      "model_forward_time": 1.7338354587554932,
      "step": 45982
    },
    {
      "epoch": 4.385185241699219e-05,
      "step": 45982,
      "training_step_time": 7.445340871810913
    },
    {
      "epoch": 4.385280609130859e-05,
      "model_forward_time": 1.7258987426757812,
      "step": 45983
    },
    {
      "epoch": 4.385280609130859e-05,
      "step": 45983,
      "training_step_time": 7.274761915206909
    },
    {
      "epoch": 4.3853759765625e-05,
      "model_forward_time": 1.722654104232788,
      "step": 45984
    },
    {
      "epoch": 4.3853759765625e-05,
      "step": 45984,
      "training_step_time": 8.470478296279907
    },
    {
      "epoch": 4.3854713439941406e-05,
      "model_forward_time": 1.723059892654419,
      "step": 45985
    },
    {
      "epoch": 4.3854713439941406e-05,
      "step": 45985,
      "training_step_time": 7.77678370475769
    },
    {
      "epoch": 4.385566711425781e-05,
      "model_forward_time": 1.723055362701416,
      "step": 45986
    },
    {
      "epoch": 4.385566711425781e-05,
      "step": 45986,
      "training_step_time": 7.789849519729614
    },
    {
      "epoch": 4.385662078857422e-05,
      "model_forward_time": 1.7257726192474365,
      "step": 45987
    },
    {
      "epoch": 4.385662078857422e-05,
      "step": 45987,
      "training_step_time": 7.718825578689575
    },
    {
      "epoch": 4.3857574462890625e-05,
      "model_forward_time": 1.728801965713501,
      "step": 45988
    },
    {
      "epoch": 4.3857574462890625e-05,
      "step": 45988,
      "training_step_time": 7.714787244796753
    },
    {
      "epoch": 4.385852813720703e-05,
      "model_forward_time": 2.9321553707122803,
      "step": 45989
    },
    {
      "epoch": 4.385852813720703e-05,
      "step": 45989,
      "training_step_time": 7.768985033035278
    },
    {
      "epoch": 4.385948181152344e-05,
      "grad_norm": 0.10658467561006546,
      "learning_rate": 6.068275043178679e-06,
      "loss": 0.0612,
      "step": 45990
    },
    {
      "epoch": 4.385948181152344e-05,
      "model_forward_time": 1.718104362487793,
      "step": 45990
    },
    {
      "dynamics_loss_avg": 0.05414012558758259,
      "epoch": 4.385948181152344e-05,
      "step": 45990
    },
    {
      "action_loss_avg": 0.0016681430046446622,
      "epoch": 4.385948181152344e-05,
      "step": 45990
    },
    {
      "epoch": 4.385948181152344e-05,
      "step": 45990,
      "training_step_time": 7.640256404876709
    },
    {
      "epoch": 4.3860435485839843e-05,
      "model_forward_time": 1.7227163314819336,
      "step": 45991
    },
    {
      "epoch": 4.3860435485839843e-05,
      "step": 45991,
      "training_step_time": 7.70304536819458
    },
    {
      "epoch": 4.386138916015625e-05,
      "model_forward_time": 1.736379861831665,
      "step": 45992
    },
    {
      "epoch": 4.386138916015625e-05,
      "step": 45992,
      "training_step_time": 7.413129091262817
    },
    {
      "epoch": 4.386234283447266e-05,
      "model_forward_time": 1.7327568531036377,
      "step": 45993
    },
    {
      "epoch": 4.386234283447266e-05,
      "step": 45993,
      "training_step_time": 7.653358697891235
    },
    {
      "epoch": 4.386329650878906e-05,
      "model_forward_time": 1.7610619068145752,
      "step": 45994
    },
    {
      "epoch": 4.386329650878906e-05,
      "step": 45994,
      "training_step_time": 8.362994909286499
    },
    {
      "epoch": 4.3864250183105466e-05,
      "model_forward_time": 1.733808994293213,
      "step": 45995
    },
    {
      "epoch": 4.3864250183105466e-05,
      "step": 45995,
      "training_step_time": 9.274484872817993
    },
    {
      "epoch": 4.386520385742188e-05,
      "model_forward_time": 1.3736763000488281,
      "step": 45996
    },
    {
      "epoch": 4.386520385742188e-05,
      "step": 45996,
      "training_step_time": 8.338973999023438
    },
    {
      "epoch": 4.386615753173828e-05,
      "model_forward_time": 1.7692487239837646,
      "step": 45997
    },
    {
      "epoch": 4.386615753173828e-05,
      "step": 45997,
      "training_step_time": 7.7458295822143555
    },
    {
      "epoch": 4.3867111206054685e-05,
      "model_forward_time": 3.033494234085083,
      "step": 45998
    },
    {
      "epoch": 4.3867111206054685e-05,
      "step": 45998,
      "training_step_time": 8.078113317489624
    },
    {
      "epoch": 4.3868064880371096e-05,
      "model_forward_time": 1.7450337409973145,
      "step": 45999
    },
    {
      "epoch": 4.3868064880371096e-05,
      "step": 45999,
      "training_step_time": 7.7363903522491455
    },
    {
      "epoch": 4.38690185546875e-05,
      "grad_norm": 0.13724499940872192,
      "learning_rate": 6.066659694959248e-06,
      "loss": 0.0575,
      "step": 46000
    },
    {
      "epoch": 4.38690185546875e-05,
      "model_forward_time": 1.723578691482544,
      "step": 46000
    },
    {
      "dynamics_loss_avg": 0.059879199042916297,
      "epoch": 4.38690185546875e-05,
      "step": 46000
    },
    {
      "action_loss_avg": 0.0017662010039202869,
      "epoch": 4.38690185546875e-05,
      "step": 46000
    },
    {
      "epoch": 4.38690185546875e-05,
      "step": 46000,
      "training_step_time": 6.637876272201538
    },
    {
      "epoch": 4.3869972229003904e-05,
      "model_forward_time": 1.7253105640411377,
      "step": 46001
    },
    {
      "epoch": 4.3869972229003904e-05,
      "step": 46001,
      "training_step_time": 8.720252275466919
    },
    {
      "epoch": 4.3870925903320315e-05,
      "model_forward_time": 1.7390468120574951,
      "step": 46002
    },
    {
      "epoch": 4.3870925903320315e-05,
      "step": 46002,
      "training_step_time": 8.866642475128174
    },
    {
      "epoch": 4.387187957763672e-05,
      "model_forward_time": 1.7333669662475586,
      "step": 46003
    },
    {
      "epoch": 4.387187957763672e-05,
      "step": 46003,
      "training_step_time": 9.448527336120605
    },
    {
      "epoch": 4.387283325195312e-05,
      "model_forward_time": 1.7457592487335205,
      "step": 46004
    },
    {
      "epoch": 4.387283325195312e-05,
      "step": 46004,
      "training_step_time": 12.363290548324585
    },
    {
      "epoch": 4.3873786926269534e-05,
      "model_forward_time": 1.732372522354126,
      "step": 46005
    },
    {
      "epoch": 4.3873786926269534e-05,
      "step": 46005,
      "training_step_time": 8.687617540359497
    },
    {
      "epoch": 4.387474060058594e-05,
      "model_forward_time": 1.7288622856140137,
      "step": 46006
    },
    {
      "epoch": 4.387474060058594e-05,
      "step": 46006,
      "training_step_time": 8.562885999679565
    },
    {
      "epoch": 4.387569427490234e-05,
      "model_forward_time": 2.717311382293701,
      "step": 46007
    },
    {
      "epoch": 4.387569427490234e-05,
      "step": 46007,
      "training_step_time": 7.572824478149414
    },
    {
      "epoch": 4.387664794921875e-05,
      "model_forward_time": 1.7318775653839111,
      "step": 46008
    },
    {
      "epoch": 4.387664794921875e-05,
      "step": 46008,
      "training_step_time": 7.587377071380615
    },
    {
      "epoch": 4.3877601623535157e-05,
      "model_forward_time": 1.7342925071716309,
      "step": 46009
    },
    {
      "epoch": 4.3877601623535157e-05,
      "step": 46009,
      "training_step_time": 7.5089545249938965
    },
    {
      "epoch": 4.387855529785156e-05,
      "grad_norm": 0.11401335150003433,
      "learning_rate": 6.065044230091518e-06,
      "loss": 0.059,
      "step": 46010
    },
    {
      "epoch": 4.387855529785156e-05,
      "model_forward_time": 1.73291015625,
      "step": 46010
    },
    {
      "dynamics_loss_avg": 0.05864569302648306,
      "epoch": 4.387855529785156e-05,
      "step": 46010
    },
    {
      "action_loss_avg": 0.0013394823065027595,
      "epoch": 4.387855529785156e-05,
      "step": 46010
    },
    {
      "epoch": 4.387855529785156e-05,
      "step": 46010,
      "training_step_time": 7.496800661087036
    },
    {
      "epoch": 4.387950897216797e-05,
      "model_forward_time": 1.7317047119140625,
      "step": 46011
    },
    {
      "epoch": 4.387950897216797e-05,
      "step": 46011,
      "training_step_time": 8.852023124694824
    },
    {
      "epoch": 4.3880462646484375e-05,
      "model_forward_time": 1.7369470596313477,
      "step": 46012
    },
    {
      "epoch": 4.3880462646484375e-05,
      "step": 46012,
      "training_step_time": 7.852511882781982
    },
    {
      "epoch": 4.388141632080078e-05,
      "model_forward_time": 1.733081579208374,
      "step": 46013
    },
    {
      "epoch": 4.388141632080078e-05,
      "step": 46013,
      "training_step_time": 11.574697971343994
    },
    {
      "epoch": 4.388236999511719e-05,
      "model_forward_time": 1.7386865615844727,
      "step": 46014
    },
    {
      "epoch": 4.388236999511719e-05,
      "step": 46014,
      "training_step_time": 8.482817888259888
    },
    {
      "epoch": 4.3883323669433594e-05,
      "model_forward_time": 1.737607479095459,
      "step": 46015
    },
    {
      "epoch": 4.3883323669433594e-05,
      "step": 46015,
      "training_step_time": 8.293882846832275
    },
    {
      "epoch": 4.388427734375e-05,
      "model_forward_time": 2.2801597118377686,
      "step": 46016
    },
    {
      "epoch": 4.388427734375e-05,
      "step": 46016,
      "training_step_time": 7.453141689300537
    },
    {
      "epoch": 4.388523101806641e-05,
      "model_forward_time": 1.7304272651672363,
      "step": 46017
    },
    {
      "epoch": 4.388523101806641e-05,
      "step": 46017,
      "training_step_time": 8.42015552520752
    },
    {
      "epoch": 4.388618469238281e-05,
      "model_forward_time": 1.7382302284240723,
      "step": 46018
    },
    {
      "epoch": 4.388618469238281e-05,
      "step": 46018,
      "training_step_time": 7.557340860366821
    },
    {
      "epoch": 4.388713836669922e-05,
      "model_forward_time": 1.7243614196777344,
      "step": 46019
    },
    {
      "epoch": 4.388713836669922e-05,
      "step": 46019,
      "training_step_time": 7.536356210708618
    },
    {
      "epoch": 4.388809204101563e-05,
      "grad_norm": 0.13817040622234344,
      "learning_rate": 6.063428648752152e-06,
      "loss": 0.0576,
      "step": 46020
    },
    {
      "epoch": 4.388809204101563e-05,
      "model_forward_time": 1.7259457111358643,
      "step": 46020
    },
    {
      "dynamics_loss_avg": 0.06094250492751598,
      "epoch": 4.388809204101563e-05,
      "step": 46020
    },
    {
      "action_loss_avg": 0.0013258900842629373,
      "epoch": 4.388809204101563e-05,
      "step": 46020
    },
    {
      "epoch": 4.388809204101563e-05,
      "step": 46020,
      "training_step_time": 7.709596157073975
    },
    {
      "epoch": 4.388904571533203e-05,
      "model_forward_time": 1.7355077266693115,
      "step": 46021
    },
    {
      "epoch": 4.388904571533203e-05,
      "step": 46021,
      "training_step_time": 8.646739721298218
    },
    {
      "epoch": 4.3889999389648436e-05,
      "model_forward_time": 1.7297759056091309,
      "step": 46022
    },
    {
      "epoch": 4.3889999389648436e-05,
      "step": 46022,
      "training_step_time": 9.78570008277893
    },
    {
      "epoch": 4.389095306396485e-05,
      "model_forward_time": 1.7748563289642334,
      "step": 46023
    },
    {
      "epoch": 4.389095306396485e-05,
      "step": 46023,
      "training_step_time": 8.797495126724243
    },
    {
      "epoch": 4.389190673828125e-05,
      "model_forward_time": 1.7301452159881592,
      "step": 46024
    },
    {
      "epoch": 4.389190673828125e-05,
      "step": 46024,
      "training_step_time": 7.76215124130249
    },
    {
      "epoch": 4.3892860412597655e-05,
      "model_forward_time": 2.2751924991607666,
      "step": 46025
    },
    {
      "epoch": 4.3892860412597655e-05,
      "step": 46025,
      "training_step_time": 8.762193202972412
    },
    {
      "epoch": 4.3893814086914066e-05,
      "model_forward_time": 1.7274353504180908,
      "step": 46026
    },
    {
      "epoch": 4.3893814086914066e-05,
      "step": 46026,
      "training_step_time": 7.477847099304199
    },
    {
      "epoch": 4.389476776123047e-05,
      "model_forward_time": 1.7343039512634277,
      "step": 46027
    },
    {
      "epoch": 4.389476776123047e-05,
      "step": 46027,
      "training_step_time": 7.580163955688477
    },
    {
      "epoch": 4.3895721435546874e-05,
      "model_forward_time": 1.734642505645752,
      "step": 46028
    },
    {
      "epoch": 4.3895721435546874e-05,
      "step": 46028,
      "training_step_time": 7.442782878875732
    },
    {
      "epoch": 4.3896675109863285e-05,
      "model_forward_time": 1.7268486022949219,
      "step": 46029
    },
    {
      "epoch": 4.3896675109863285e-05,
      "step": 46029,
      "training_step_time": 7.751465082168579
    },
    {
      "epoch": 4.389762878417969e-05,
      "grad_norm": 0.1668614149093628,
      "learning_rate": 6.061812951117827e-06,
      "loss": 0.0595,
      "step": 46030
    },
    {
      "epoch": 4.389762878417969e-05,
      "model_forward_time": 1.7324647903442383,
      "step": 46030
    },
    {
      "dynamics_loss_avg": 0.04416138269007206,
      "epoch": 4.389762878417969e-05,
      "step": 46030
    },
    {
      "action_loss_avg": 0.0009435927408048883,
      "epoch": 4.389762878417969e-05,
      "step": 46030
    },
    {
      "epoch": 4.389762878417969e-05,
      "step": 46030,
      "training_step_time": 7.582566261291504
    },
    {
      "epoch": 4.389858245849609e-05,
      "model_forward_time": 1.7328517436981201,
      "step": 46031
    },
    {
      "epoch": 4.389858245849609e-05,
      "step": 46031,
      "training_step_time": 9.742253303527832
    },
    {
      "epoch": 4.38995361328125e-05,
      "model_forward_time": 1.7693190574645996,
      "step": 46032
    },
    {
      "epoch": 4.38995361328125e-05,
      "step": 46032,
      "training_step_time": 8.371458053588867
    },
    {
      "epoch": 4.390048980712891e-05,
      "model_forward_time": 1.7390997409820557,
      "step": 46033
    },
    {
      "epoch": 4.390048980712891e-05,
      "step": 46033,
      "training_step_time": 7.729126691818237
    },
    {
      "epoch": 4.390144348144531e-05,
      "model_forward_time": 2.2639503479003906,
      "step": 46034
    },
    {
      "epoch": 4.390144348144531e-05,
      "step": 46034,
      "training_step_time": 7.874114751815796
    },
    {
      "epoch": 4.3902397155761715e-05,
      "model_forward_time": 1.7259881496429443,
      "step": 46035
    },
    {
      "epoch": 4.3902397155761715e-05,
      "step": 46035,
      "training_step_time": 7.5307934284210205
    },
    {
      "epoch": 4.3903350830078126e-05,
      "model_forward_time": 1.7358341217041016,
      "step": 46036
    },
    {
      "epoch": 4.3903350830078126e-05,
      "step": 46036,
      "training_step_time": 8.408074855804443
    },
    {
      "epoch": 4.390430450439453e-05,
      "model_forward_time": 1.7233514785766602,
      "step": 46037
    },
    {
      "epoch": 4.390430450439453e-05,
      "step": 46037,
      "training_step_time": 7.772083044052124
    },
    {
      "epoch": 4.3905258178710934e-05,
      "model_forward_time": 1.7254743576049805,
      "step": 46038
    },
    {
      "epoch": 4.3905258178710934e-05,
      "step": 46038,
      "training_step_time": 7.759328365325928
    },
    {
      "epoch": 4.3906211853027345e-05,
      "model_forward_time": 1.7397103309631348,
      "step": 46039
    },
    {
      "epoch": 4.3906211853027345e-05,
      "step": 46039,
      "training_step_time": 9.262152910232544
    },
    {
      "epoch": 4.390716552734375e-05,
      "grad_norm": 0.12373468279838562,
      "learning_rate": 6.060197137365235e-06,
      "loss": 0.0575,
      "step": 46040
    },
    {
      "epoch": 4.390716552734375e-05,
      "model_forward_time": 1.725360631942749,
      "step": 46040
    },
    {
      "dynamics_loss_avg": 0.06416052225977183,
      "epoch": 4.390716552734375e-05,
      "step": 46040
    },
    {
      "action_loss_avg": 0.0014828229788690806,
      "epoch": 4.390716552734375e-05,
      "step": 46040
    },
    {
      "epoch": 4.390716552734375e-05,
      "step": 46040,
      "training_step_time": 8.82690167427063
    },
    {
      "epoch": 4.390811920166015e-05,
      "model_forward_time": 1.755195140838623,
      "step": 46041
    },
    {
      "epoch": 4.390811920166015e-05,
      "step": 46041,
      "training_step_time": 9.202305555343628
    },
    {
      "epoch": 4.3909072875976564e-05,
      "model_forward_time": 1.7300567626953125,
      "step": 46042
    },
    {
      "epoch": 4.3909072875976564e-05,
      "step": 46042,
      "training_step_time": 8.815568685531616
    },
    {
      "epoch": 4.391002655029297e-05,
      "model_forward_time": 2.276409387588501,
      "step": 46043
    },
    {
      "epoch": 4.391002655029297e-05,
      "step": 46043,
      "training_step_time": 7.849617004394531
    },
    {
      "epoch": 4.391098022460937e-05,
      "model_forward_time": 1.7284376621246338,
      "step": 46044
    },
    {
      "epoch": 4.391098022460937e-05,
      "step": 46044,
      "training_step_time": 8.621762037277222
    },
    {
      "epoch": 4.391193389892578e-05,
      "model_forward_time": 1.7152130603790283,
      "step": 46045
    },
    {
      "epoch": 4.391193389892578e-05,
      "step": 46045,
      "training_step_time": 7.649016857147217
    },
    {
      "epoch": 4.391288757324219e-05,
      "model_forward_time": 1.730638027191162,
      "step": 46046
    },
    {
      "epoch": 4.391288757324219e-05,
      "step": 46046,
      "training_step_time": 7.565586566925049
    },
    {
      "epoch": 4.391384124755859e-05,
      "model_forward_time": 1.7307839393615723,
      "step": 46047
    },
    {
      "epoch": 4.391384124755859e-05,
      "step": 46047,
      "training_step_time": 7.773055553436279
    },
    {
      "epoch": 4.3914794921875e-05,
      "model_forward_time": 1.7325069904327393,
      "step": 46048
    },
    {
      "epoch": 4.3914794921875e-05,
      "step": 46048,
      "training_step_time": 9.336713314056396
    },
    {
      "epoch": 4.3915748596191406e-05,
      "model_forward_time": 1.730745553970337,
      "step": 46049
    },
    {
      "epoch": 4.3915748596191406e-05,
      "step": 46049,
      "training_step_time": 9.212728023529053
    },
    {
      "epoch": 4.391670227050781e-05,
      "grad_norm": 0.11966816335916519,
      "learning_rate": 6.058581207671077e-06,
      "loss": 0.0563,
      "step": 46050
    },
    {
      "epoch": 4.391670227050781e-05,
      "model_forward_time": 1.768984079360962,
      "step": 46050
    },
    {
      "dynamics_loss_avg": 0.05750465802848339,
      "epoch": 4.391670227050781e-05,
      "step": 46050
    },
    {
      "action_loss_avg": 0.0011744347226340323,
      "epoch": 4.391670227050781e-05,
      "step": 46050
    },
    {
      "epoch": 4.391670227050781e-05,
      "step": 46050,
      "training_step_time": 8.519163608551025
    },
    {
      "epoch": 4.391765594482422e-05,
      "model_forward_time": 1.7218825817108154,
      "step": 46051
    },
    {
      "epoch": 4.391765594482422e-05,
      "step": 46051,
      "training_step_time": 7.8020875453948975
    },
    {
      "epoch": 4.3918609619140625e-05,
      "model_forward_time": 2.275038003921509,
      "step": 46052
    },
    {
      "epoch": 4.3918609619140625e-05,
      "step": 46052,
      "training_step_time": 7.7836525440216064
    },
    {
      "epoch": 4.391956329345703e-05,
      "model_forward_time": 1.731889009475708,
      "step": 46053
    },
    {
      "epoch": 4.391956329345703e-05,
      "step": 46053,
      "training_step_time": 7.807772397994995
    },
    {
      "epoch": 4.392051696777344e-05,
      "model_forward_time": 1.7308506965637207,
      "step": 46054
    },
    {
      "epoch": 4.392051696777344e-05,
      "step": 46054,
      "training_step_time": 7.631057500839233
    },
    {
      "epoch": 4.3921470642089843e-05,
      "model_forward_time": 1.7317235469818115,
      "step": 46055
    },
    {
      "epoch": 4.3921470642089843e-05,
      "step": 46055,
      "training_step_time": 7.639571666717529
    },
    {
      "epoch": 4.392242431640625e-05,
      "model_forward_time": 1.730832576751709,
      "step": 46056
    },
    {
      "epoch": 4.392242431640625e-05,
      "step": 46056,
      "training_step_time": 8.681222677230835
    },
    {
      "epoch": 4.392337799072266e-05,
      "model_forward_time": 1.7300512790679932,
      "step": 46057
    },
    {
      "epoch": 4.392337799072266e-05,
      "step": 46057,
      "training_step_time": 7.620739221572876
    },
    {
      "epoch": 4.392433166503906e-05,
      "model_forward_time": 1.7332475185394287,
      "step": 46058
    },
    {
      "epoch": 4.392433166503906e-05,
      "step": 46058,
      "training_step_time": 7.855045557022095
    },
    {
      "epoch": 4.3925285339355466e-05,
      "model_forward_time": 1.7482736110687256,
      "step": 46059
    },
    {
      "epoch": 4.3925285339355466e-05,
      "step": 46059,
      "training_step_time": 9.629720687866211
    },
    {
      "epoch": 4.392623901367188e-05,
      "grad_norm": 0.1126839742064476,
      "learning_rate": 6.056965162212072e-06,
      "loss": 0.0567,
      "step": 46060
    },
    {
      "epoch": 4.392623901367188e-05,
      "model_forward_time": 1.732733964920044,
      "step": 46060
    },
    {
      "dynamics_loss_avg": 0.06549561601132155,
      "epoch": 4.392623901367188e-05,
      "step": 46060
    },
    {
      "action_loss_avg": 0.0019109867687802761,
      "epoch": 4.392623901367188e-05,
      "step": 46060
    },
    {
      "epoch": 4.392623901367188e-05,
      "step": 46060,
      "training_step_time": 8.839767456054688
    },
    {
      "epoch": 4.392719268798828e-05,
      "model_forward_time": 2.2725229263305664,
      "step": 46061
    },
    {
      "epoch": 4.392719268798828e-05,
      "step": 46061,
      "training_step_time": 8.647572040557861
    },
    {
      "epoch": 4.3928146362304685e-05,
      "model_forward_time": 1.7318191528320312,
      "step": 46062
    },
    {
      "epoch": 4.3928146362304685e-05,
      "step": 46062,
      "training_step_time": 8.70303225517273
    },
    {
      "epoch": 4.3929100036621096e-05,
      "model_forward_time": 1.7267694473266602,
      "step": 46063
    },
    {
      "epoch": 4.3929100036621096e-05,
      "step": 46063,
      "training_step_time": 8.394004583358765
    },
    {
      "epoch": 4.39300537109375e-05,
      "model_forward_time": 1.730111837387085,
      "step": 46064
    },
    {
      "epoch": 4.39300537109375e-05,
      "step": 46064,
      "training_step_time": 7.626570463180542
    },
    {
      "epoch": 4.3931007385253904e-05,
      "model_forward_time": 1.731123447418213,
      "step": 46065
    },
    {
      "epoch": 4.3931007385253904e-05,
      "step": 46065,
      "training_step_time": 7.818229913711548
    },
    {
      "epoch": 4.3931961059570315e-05,
      "model_forward_time": 1.7357196807861328,
      "step": 46066
    },
    {
      "epoch": 4.3931961059570315e-05,
      "step": 46066,
      "training_step_time": 7.586041688919067
    },
    {
      "epoch": 4.393291473388672e-05,
      "model_forward_time": 1.7277238368988037,
      "step": 46067
    },
    {
      "epoch": 4.393291473388672e-05,
      "step": 46067,
      "training_step_time": 8.319367408752441
    },
    {
      "epoch": 4.393386840820312e-05,
      "model_forward_time": 1.7395694255828857,
      "step": 46068
    },
    {
      "epoch": 4.393386840820312e-05,
      "step": 46068,
      "training_step_time": 9.893124341964722
    },
    {
      "epoch": 4.3934822082519534e-05,
      "model_forward_time": 1.72735595703125,
      "step": 46069
    },
    {
      "epoch": 4.3934822082519534e-05,
      "step": 46069,
      "training_step_time": 7.826604843139648
    },
    {
      "epoch": 4.393577575683594e-05,
      "grad_norm": 0.08746278285980225,
      "learning_rate": 6.055349001164943e-06,
      "loss": 0.0632,
      "step": 46070
    },
    {
      "epoch": 4.393577575683594e-05,
      "model_forward_time": 2.2603602409362793,
      "step": 46070
    },
    {
      "dynamics_loss_avg": 0.07552606388926505,
      "epoch": 4.393577575683594e-05,
      "step": 46070
    },
    {
      "action_loss_avg": 0.0016705325338989495,
      "epoch": 4.393577575683594e-05,
      "step": 46070
    },
    {
      "epoch": 4.393577575683594e-05,
      "step": 46070,
      "training_step_time": 8.566348791122437
    },
    {
      "epoch": 4.393672943115234e-05,
      "model_forward_time": 1.7564797401428223,
      "step": 46071
    },
    {
      "epoch": 4.393672943115234e-05,
      "step": 46071,
      "training_step_time": 7.965687036514282
    },
    {
      "epoch": 4.393768310546875e-05,
      "model_forward_time": 1.7478444576263428,
      "step": 46072
    },
    {
      "epoch": 4.393768310546875e-05,
      "step": 46072,
      "training_step_time": 8.374882221221924
    },
    {
      "epoch": 4.393863677978516e-05,
      "model_forward_time": 1.7267780303955078,
      "step": 46073
    },
    {
      "epoch": 4.393863677978516e-05,
      "step": 46073,
      "training_step_time": 7.468173027038574
    },
    {
      "epoch": 4.393959045410156e-05,
      "model_forward_time": 1.7321407794952393,
      "step": 46074
    },
    {
      "epoch": 4.393959045410156e-05,
      "step": 46074,
      "training_step_time": 7.9868245124816895
    },
    {
      "epoch": 4.394054412841797e-05,
      "model_forward_time": 1.7269055843353271,
      "step": 46075
    },
    {
      "epoch": 4.394054412841797e-05,
      "step": 46075,
      "training_step_time": 7.641122579574585
    },
    {
      "epoch": 4.3941497802734375e-05,
      "model_forward_time": 1.7366161346435547,
      "step": 46076
    },
    {
      "epoch": 4.3941497802734375e-05,
      "step": 46076,
      "training_step_time": 7.734426498413086
    },
    {
      "epoch": 4.394245147705078e-05,
      "model_forward_time": 1.7277107238769531,
      "step": 46077
    },
    {
      "epoch": 4.394245147705078e-05,
      "step": 46077,
      "training_step_time": 8.96715259552002
    },
    {
      "epoch": 4.394340515136719e-05,
      "model_forward_time": 1.7810447216033936,
      "step": 46078
    },
    {
      "epoch": 4.394340515136719e-05,
      "step": 46078,
      "training_step_time": 9.230064868927002
    },
    {
      "epoch": 4.3944358825683594e-05,
      "model_forward_time": 2.2769787311553955,
      "step": 46079
    },
    {
      "epoch": 4.3944358825683594e-05,
      "step": 46079,
      "training_step_time": 7.79439902305603
    },
    {
      "epoch": 4.39453125e-05,
      "grad_norm": 0.14507421851158142,
      "learning_rate": 6.053732724706435e-06,
      "loss": 0.0649,
      "step": 46080
    },
    {
      "epoch": 4.39453125e-05,
      "model_forward_time": 1.7301292419433594,
      "step": 46080
    },
    {
      "dynamics_loss_avg": 0.0744290692731738,
      "epoch": 4.39453125e-05,
      "step": 46080
    },
    {
      "action_loss_avg": 0.0014851970598101615,
      "epoch": 4.39453125e-05,
      "step": 46080
    },
    {
      "epoch": 4.39453125e-05,
      "step": 46080,
      "training_step_time": 8.844826698303223
    },
    {
      "epoch": 4.394626617431641e-05,
      "model_forward_time": 1.7346007823944092,
      "step": 46081
    },
    {
      "epoch": 4.394626617431641e-05,
      "step": 46081,
      "training_step_time": 8.438723087310791
    },
    {
      "epoch": 4.394721984863281e-05,
      "model_forward_time": 1.7335963249206543,
      "step": 46082
    },
    {
      "epoch": 4.394721984863281e-05,
      "step": 46082,
      "training_step_time": 7.527784109115601
    },
    {
      "epoch": 4.394817352294922e-05,
      "model_forward_time": 1.7375481128692627,
      "step": 46083
    },
    {
      "epoch": 4.394817352294922e-05,
      "step": 46083,
      "training_step_time": 8.920339822769165
    },
    {
      "epoch": 4.394912719726563e-05,
      "model_forward_time": 1.7311182022094727,
      "step": 46084
    },
    {
      "epoch": 4.394912719726563e-05,
      "step": 46084,
      "training_step_time": 8.242454767227173
    },
    {
      "epoch": 4.395008087158203e-05,
      "model_forward_time": 1.7401883602142334,
      "step": 46085
    },
    {
      "epoch": 4.395008087158203e-05,
      "step": 46085,
      "training_step_time": 8.242444276809692
    },
    {
      "epoch": 4.3951034545898436e-05,
      "model_forward_time": 1.731698751449585,
      "step": 46086
    },
    {
      "epoch": 4.3951034545898436e-05,
      "step": 46086,
      "training_step_time": 8.667281150817871
    },
    {
      "epoch": 4.395198822021485e-05,
      "model_forward_time": 1.7791736125946045,
      "step": 46087
    },
    {
      "epoch": 4.395198822021485e-05,
      "step": 46087,
      "training_step_time": 9.313434600830078
    },
    {
      "epoch": 4.395294189453125e-05,
      "model_forward_time": 2.2663166522979736,
      "step": 46088
    },
    {
      "epoch": 4.395294189453125e-05,
      "step": 46088,
      "training_step_time": 8.449875116348267
    },
    {
      "epoch": 4.3953895568847655e-05,
      "model_forward_time": 1.7398982048034668,
      "step": 46089
    },
    {
      "epoch": 4.3953895568847655e-05,
      "step": 46089,
      "training_step_time": 8.624771118164062
    },
    {
      "epoch": 4.3954849243164066e-05,
      "grad_norm": 0.11537762731313705,
      "learning_rate": 6.052116333013302e-06,
      "loss": 0.0614,
      "step": 46090
    },
    {
      "epoch": 4.3954849243164066e-05,
      "model_forward_time": 1.7312238216400146,
      "step": 46090
    },
    {
      "dynamics_loss_avg": 0.06985711678862572,
      "epoch": 4.3954849243164066e-05,
      "step": 46090
    },
    {
      "action_loss_avg": 0.002110963506856933,
      "epoch": 4.3954849243164066e-05,
      "step": 46090
    },
    {
      "epoch": 4.3954849243164066e-05,
      "step": 46090,
      "training_step_time": 8.733349800109863
    },
    {
      "epoch": 4.395580291748047e-05,
      "model_forward_time": 1.3775036334991455,
      "step": 46091
    },
    {
      "epoch": 4.395580291748047e-05,
      "step": 46091,
      "training_step_time": 7.582449674606323
    },
    {
      "epoch": 4.3956756591796874e-05,
      "model_forward_time": 1.7286453247070312,
      "step": 46092
    },
    {
      "epoch": 4.3956756591796874e-05,
      "step": 46092,
      "training_step_time": 7.855311632156372
    },
    {
      "epoch": 4.3957710266113285e-05,
      "model_forward_time": 1.731844186782837,
      "step": 46093
    },
    {
      "epoch": 4.3957710266113285e-05,
      "step": 46093,
      "training_step_time": 7.86857008934021
    },
    {
      "epoch": 4.395866394042969e-05,
      "model_forward_time": 1.739475965499878,
      "step": 46094
    },
    {
      "epoch": 4.395866394042969e-05,
      "step": 46094,
      "training_step_time": 8.152247905731201
    },
    {
      "epoch": 4.395961761474609e-05,
      "model_forward_time": 1.7340092658996582,
      "step": 46095
    },
    {
      "epoch": 4.395961761474609e-05,
      "step": 46095,
      "training_step_time": 8.64851713180542
    },
    {
      "epoch": 4.39605712890625e-05,
      "model_forward_time": 1.7801105976104736,
      "step": 46096
    },
    {
      "epoch": 4.39605712890625e-05,
      "step": 46096,
      "training_step_time": 8.431095838546753
    },
    {
      "epoch": 4.396152496337891e-05,
      "model_forward_time": 2.270664691925049,
      "step": 46097
    },
    {
      "epoch": 4.396152496337891e-05,
      "step": 46097,
      "training_step_time": 9.283432006835938
    },
    {
      "epoch": 4.396247863769531e-05,
      "model_forward_time": 1.7429468631744385,
      "step": 46098
    },
    {
      "epoch": 4.396247863769531e-05,
      "step": 46098,
      "training_step_time": 9.373597621917725
    },
    {
      "epoch": 4.3963432312011716e-05,
      "model_forward_time": 1.7341251373291016,
      "step": 46099
    },
    {
      "epoch": 4.3963432312011716e-05,
      "step": 46099,
      "training_step_time": 8.486125946044922
    },
    {
      "epoch": 4.3964385986328126e-05,
      "grad_norm": 0.12472014129161835,
      "learning_rate": 6.050499826262306e-06,
      "loss": 0.0623,
      "step": 46100
    },
    {
      "epoch": 4.3964385986328126e-05,
      "model_forward_time": 1.7366724014282227,
      "step": 46100
    },
    {
      "dynamics_loss_avg": 0.06334363501518965,
      "epoch": 4.3964385986328126e-05,
      "step": 46100
    },
    {
      "action_loss_avg": 0.0015606315690092742,
      "epoch": 4.3964385986328126e-05,
      "step": 46100
    },
    {
      "epoch": 4.3964385986328126e-05,
      "step": 46100,
      "training_step_time": 7.520329236984253
    },
    {
      "epoch": 4.396533966064453e-05,
      "model_forward_time": 1.7287781238555908,
      "step": 46101
    },
    {
      "epoch": 4.396533966064453e-05,
      "step": 46101,
      "training_step_time": 7.952190637588501
    },
    {
      "epoch": 4.3966293334960934e-05,
      "model_forward_time": 1.7338602542877197,
      "step": 46102
    },
    {
      "epoch": 4.3966293334960934e-05,
      "step": 46102,
      "training_step_time": 8.459818124771118
    },
    {
      "epoch": 4.3967247009277345e-05,
      "model_forward_time": 1.7343494892120361,
      "step": 46103
    },
    {
      "epoch": 4.3967247009277345e-05,
      "step": 46103,
      "training_step_time": 8.278903245925903
    },
    {
      "epoch": 4.396820068359375e-05,
      "model_forward_time": 1.7293741703033447,
      "step": 46104
    },
    {
      "epoch": 4.396820068359375e-05,
      "step": 46104,
      "training_step_time": 9.928728818893433
    },
    {
      "epoch": 4.396915435791015e-05,
      "model_forward_time": 1.7634756565093994,
      "step": 46105
    },
    {
      "epoch": 4.396915435791015e-05,
      "step": 46105,
      "training_step_time": 8.893115758895874
    },
    {
      "epoch": 4.3970108032226564e-05,
      "model_forward_time": 2.2871077060699463,
      "step": 46106
    },
    {
      "epoch": 4.3970108032226564e-05,
      "step": 46106,
      "training_step_time": 8.357549905776978
    },
    {
      "epoch": 4.397106170654297e-05,
      "model_forward_time": 1.7379553318023682,
      "step": 46107
    },
    {
      "epoch": 4.397106170654297e-05,
      "step": 46107,
      "training_step_time": 7.821699380874634
    },
    {
      "epoch": 4.397201538085937e-05,
      "model_forward_time": 1.7377293109893799,
      "step": 46108
    },
    {
      "epoch": 4.397201538085937e-05,
      "step": 46108,
      "training_step_time": 7.8066160678863525
    },
    {
      "epoch": 4.397296905517578e-05,
      "model_forward_time": 1.7463946342468262,
      "step": 46109
    },
    {
      "epoch": 4.397296905517578e-05,
      "step": 46109,
      "training_step_time": 7.595296859741211
    },
    {
      "epoch": 4.397392272949219e-05,
      "grad_norm": 0.12035048753023148,
      "learning_rate": 6.048883204630229e-06,
      "loss": 0.0611,
      "step": 46110
    },
    {
      "epoch": 4.397392272949219e-05,
      "model_forward_time": 1.7318968772888184,
      "step": 46110
    },
    {
      "dynamics_loss_avg": 0.053387413173913954,
      "epoch": 4.397392272949219e-05,
      "step": 46110
    },
    {
      "action_loss_avg": 0.0008393328287638723,
      "epoch": 4.397392272949219e-05,
      "step": 46110
    },
    {
      "epoch": 4.397392272949219e-05,
      "step": 46110,
      "training_step_time": 7.828255653381348
    },
    {
      "epoch": 4.397487640380859e-05,
      "model_forward_time": 1.7340686321258545,
      "step": 46111
    },
    {
      "epoch": 4.397487640380859e-05,
      "step": 46111,
      "training_step_time": 7.907364845275879
    },
    {
      "epoch": 4.3975830078125e-05,
      "model_forward_time": 1.740839958190918,
      "step": 46112
    },
    {
      "epoch": 4.3975830078125e-05,
      "step": 46112,
      "training_step_time": 7.614634037017822
    },
    {
      "epoch": 4.3976783752441406e-05,
      "model_forward_time": 1.7384638786315918,
      "step": 46113
    },
    {
      "epoch": 4.3976783752441406e-05,
      "step": 46113,
      "training_step_time": 7.7869462966918945
    },
    {
      "epoch": 4.397773742675781e-05,
      "model_forward_time": 1.7452807426452637,
      "step": 46114
    },
    {
      "epoch": 4.397773742675781e-05,
      "step": 46114,
      "training_step_time": 8.72083830833435
    },
    {
      "epoch": 4.397869110107422e-05,
      "model_forward_time": 2.2863645553588867,
      "step": 46115
    },
    {
      "epoch": 4.397869110107422e-05,
      "step": 46115,
      "training_step_time": 8.576138257980347
    },
    {
      "epoch": 4.3979644775390625e-05,
      "model_forward_time": 1.7387235164642334,
      "step": 46116
    },
    {
      "epoch": 4.3979644775390625e-05,
      "step": 46116,
      "training_step_time": 8.844604253768921
    },
    {
      "epoch": 4.398059844970703e-05,
      "model_forward_time": 1.7378716468811035,
      "step": 46117
    },
    {
      "epoch": 4.398059844970703e-05,
      "step": 46117,
      "training_step_time": 8.313331127166748
    },
    {
      "epoch": 4.398155212402344e-05,
      "model_forward_time": 1.7352583408355713,
      "step": 46118
    },
    {
      "epoch": 4.398155212402344e-05,
      "step": 46118,
      "training_step_time": 7.569812774658203
    },
    {
      "epoch": 4.3982505798339844e-05,
      "model_forward_time": 1.746009111404419,
      "step": 46119
    },
    {
      "epoch": 4.3982505798339844e-05,
      "step": 46119,
      "training_step_time": 7.691860675811768
    },
    {
      "epoch": 4.398345947265625e-05,
      "grad_norm": 0.15188123285770416,
      "learning_rate": 6.047266468293863e-06,
      "loss": 0.0653,
      "step": 46120
    },
    {
      "epoch": 4.398345947265625e-05,
      "model_forward_time": 1.7354390621185303,
      "step": 46120
    },
    {
      "dynamics_loss_avg": 0.05931076854467392,
      "epoch": 4.398345947265625e-05,
      "step": 46120
    },
    {
      "action_loss_avg": 0.0017941132828127593,
      "epoch": 4.398345947265625e-05,
      "step": 46120
    },
    {
      "epoch": 4.398345947265625e-05,
      "step": 46120,
      "training_step_time": 8.063696146011353
    },
    {
      "epoch": 4.398441314697266e-05,
      "model_forward_time": 1.7342946529388428,
      "step": 46121
    },
    {
      "epoch": 4.398441314697266e-05,
      "step": 46121,
      "training_step_time": 7.574786186218262
    },
    {
      "epoch": 4.398536682128906e-05,
      "model_forward_time": 1.743689775466919,
      "step": 46122
    },
    {
      "epoch": 4.398536682128906e-05,
      "step": 46122,
      "training_step_time": 8.482300043106079
    },
    {
      "epoch": 4.3986320495605466e-05,
      "model_forward_time": 1.7291028499603271,
      "step": 46123
    },
    {
      "epoch": 4.3986320495605466e-05,
      "step": 46123,
      "training_step_time": 8.9647958278656
    },
    {
      "epoch": 4.398727416992188e-05,
      "model_forward_time": 2.302896499633789,
      "step": 46124
    },
    {
      "epoch": 4.398727416992188e-05,
      "step": 46124,
      "training_step_time": 8.462406158447266
    },
    {
      "epoch": 4.398822784423828e-05,
      "model_forward_time": 1.7347967624664307,
      "step": 46125
    },
    {
      "epoch": 4.398822784423828e-05,
      "step": 46125,
      "training_step_time": 8.887302160263062
    },
    {
      "epoch": 4.3989181518554685e-05,
      "model_forward_time": 1.7294819355010986,
      "step": 46126
    },
    {
      "epoch": 4.3989181518554685e-05,
      "step": 46126,
      "training_step_time": 7.681614398956299
    },
    {
      "epoch": 4.3990135192871096e-05,
      "model_forward_time": 1.7312731742858887,
      "step": 46127
    },
    {
      "epoch": 4.3990135192871096e-05,
      "step": 46127,
      "training_step_time": 7.834098815917969
    },
    {
      "epoch": 4.39910888671875e-05,
      "model_forward_time": 1.7687382698059082,
      "step": 46128
    },
    {
      "epoch": 4.39910888671875e-05,
      "step": 46128,
      "training_step_time": 7.247126340866089
    },
    {
      "epoch": 4.3992042541503904e-05,
      "model_forward_time": 1.737426996231079,
      "step": 46129
    },
    {
      "epoch": 4.3992042541503904e-05,
      "step": 46129,
      "training_step_time": 8.037403583526611
    },
    {
      "epoch": 4.3992996215820315e-05,
      "grad_norm": 0.1251658797264099,
      "learning_rate": 6.045649617430009e-06,
      "loss": 0.0614,
      "step": 46130
    },
    {
      "epoch": 4.3992996215820315e-05,
      "model_forward_time": 1.7345094680786133,
      "step": 46130
    },
    {
      "dynamics_loss_avg": 0.06382241817191243,
      "epoch": 4.3992996215820315e-05,
      "step": 46130
    },
    {
      "action_loss_avg": 0.0018439653431414627,
      "epoch": 4.3992996215820315e-05,
      "step": 46130
    },
    {
      "epoch": 4.3992996215820315e-05,
      "step": 46130,
      "training_step_time": 7.47827935218811
    },
    {
      "epoch": 4.399394989013672e-05,
      "model_forward_time": 1.7349722385406494,
      "step": 46131
    },
    {
      "epoch": 4.399394989013672e-05,
      "step": 46131,
      "training_step_time": 7.632498025894165
    },
    {
      "epoch": 4.399490356445312e-05,
      "model_forward_time": 1.7374775409698486,
      "step": 46132
    },
    {
      "epoch": 4.399490356445312e-05,
      "step": 46132,
      "training_step_time": 8.723182678222656
    },
    {
      "epoch": 4.3995857238769534e-05,
      "model_forward_time": 2.302445888519287,
      "step": 46133
    },
    {
      "epoch": 4.3995857238769534e-05,
      "step": 46133,
      "training_step_time": 8.574013710021973
    },
    {
      "epoch": 4.399681091308594e-05,
      "model_forward_time": 1.73956298828125,
      "step": 46134
    },
    {
      "epoch": 4.399681091308594e-05,
      "step": 46134,
      "training_step_time": 7.821871995925903
    },
    {
      "epoch": 4.399776458740234e-05,
      "model_forward_time": 1.736030101776123,
      "step": 46135
    },
    {
      "epoch": 4.399776458740234e-05,
      "step": 46135,
      "training_step_time": 7.93407940864563
    },
    {
      "epoch": 4.399871826171875e-05,
      "model_forward_time": 1.734473705291748,
      "step": 46136
    },
    {
      "epoch": 4.399871826171875e-05,
      "step": 46136,
      "training_step_time": 7.7622435092926025
    },
    {
      "epoch": 4.399967193603516e-05,
      "model_forward_time": 1.7426435947418213,
      "step": 46137
    },
    {
      "epoch": 4.399967193603516e-05,
      "step": 46137,
      "training_step_time": 7.8099822998046875
    },
    {
      "epoch": 4.400062561035156e-05,
      "model_forward_time": 1.7339835166931152,
      "step": 46138
    },
    {
      "epoch": 4.400062561035156e-05,
      "step": 46138,
      "training_step_time": 8.104663610458374
    },
    {
      "epoch": 4.400157928466797e-05,
      "model_forward_time": 1.7318451404571533,
      "step": 46139
    },
    {
      "epoch": 4.400157928466797e-05,
      "step": 46139,
      "training_step_time": 7.610280990600586
    },
    {
      "epoch": 4.4002532958984376e-05,
      "grad_norm": 0.17240163683891296,
      "learning_rate": 6.044032652215487e-06,
      "loss": 0.0629,
      "step": 46140
    },
    {
      "epoch": 4.4002532958984376e-05,
      "model_forward_time": 1.7423768043518066,
      "step": 46140
    },
    {
      "dynamics_loss_avg": 0.06669718343764544,
      "epoch": 4.4002532958984376e-05,
      "step": 46140
    },
    {
      "action_loss_avg": 0.0016989658091915772,
      "epoch": 4.4002532958984376e-05,
      "step": 46140
    },
    {
      "epoch": 4.4002532958984376e-05,
      "step": 46140,
      "training_step_time": 8.205327987670898
    },
    {
      "epoch": 4.400348663330078e-05,
      "model_forward_time": 1.7276616096496582,
      "step": 46141
    },
    {
      "epoch": 4.400348663330078e-05,
      "step": 46141,
      "training_step_time": 9.440855979919434
    },
    {
      "epoch": 4.400444030761719e-05,
      "model_forward_time": 1.776712417602539,
      "step": 46142
    },
    {
      "epoch": 4.400444030761719e-05,
      "step": 46142,
      "training_step_time": 9.81445050239563
    },
    {
      "epoch": 4.4005393981933594e-05,
      "model_forward_time": 1.7321603298187256,
      "step": 46143
    },
    {
      "epoch": 4.4005393981933594e-05,
      "step": 46143,
      "training_step_time": 8.914753913879395
    },
    {
      "epoch": 4.400634765625e-05,
      "model_forward_time": 1.7340881824493408,
      "step": 46144
    },
    {
      "epoch": 4.400634765625e-05,
      "step": 46144,
      "training_step_time": 8.713756084442139
    },
    {
      "epoch": 4.400730133056641e-05,
      "model_forward_time": 1.7355830669403076,
      "step": 46145
    },
    {
      "epoch": 4.400730133056641e-05,
      "step": 46145,
      "training_step_time": 7.7673704624176025
    },
    {
      "epoch": 4.400825500488281e-05,
      "model_forward_time": 1.7285046577453613,
      "step": 46146
    },
    {
      "epoch": 4.400825500488281e-05,
      "step": 46146,
      "training_step_time": 7.602276802062988
    },
    {
      "epoch": 4.400920867919922e-05,
      "model_forward_time": 1.7358894348144531,
      "step": 46147
    },
    {
      "epoch": 4.400920867919922e-05,
      "step": 46147,
      "training_step_time": 8.107518911361694
    },
    {
      "epoch": 4.401016235351563e-05,
      "model_forward_time": 1.7317800521850586,
      "step": 46148
    },
    {
      "epoch": 4.401016235351563e-05,
      "step": 46148,
      "training_step_time": 7.696894407272339
    },
    {
      "epoch": 4.401111602783203e-05,
      "model_forward_time": 1.721440315246582,
      "step": 46149
    },
    {
      "epoch": 4.401111602783203e-05,
      "step": 46149,
      "training_step_time": 8.248200416564941
    },
    {
      "epoch": 4.4012069702148436e-05,
      "grad_norm": 0.09907301515340805,
      "learning_rate": 6.042415572827123e-06,
      "loss": 0.0582,
      "step": 46150
    },
    {
      "epoch": 4.4012069702148436e-05,
      "model_forward_time": 1.7301297187805176,
      "step": 46150
    },
    {
      "dynamics_loss_avg": 0.06074037216603756,
      "epoch": 4.4012069702148436e-05,
      "step": 46150
    },
    {
      "action_loss_avg": 0.002340075437678024,
      "epoch": 4.4012069702148436e-05,
      "step": 46150
    },
    {
      "epoch": 4.4012069702148436e-05,
      "step": 46150,
      "training_step_time": 8.050345182418823
    },
    {
      "epoch": 4.401302337646485e-05,
      "model_forward_time": 1.7628774642944336,
      "step": 46151
    },
    {
      "epoch": 4.401302337646485e-05,
      "step": 46151,
      "training_step_time": 9.858425855636597
    },
    {
      "epoch": 4.401397705078125e-05,
      "model_forward_time": 1.7293806076049805,
      "step": 46152
    },
    {
      "epoch": 4.401397705078125e-05,
      "step": 46152,
      "training_step_time": 9.416786193847656
    },
    {
      "epoch": 4.4014930725097655e-05,
      "model_forward_time": 1.7271630764007568,
      "step": 46153
    },
    {
      "epoch": 4.4014930725097655e-05,
      "step": 46153,
      "training_step_time": 8.692998886108398
    },
    {
      "epoch": 4.4015884399414066e-05,
      "model_forward_time": 1.7432188987731934,
      "step": 46154
    },
    {
      "epoch": 4.4015884399414066e-05,
      "step": 46154,
      "training_step_time": 8.487866163253784
    },
    {
      "epoch": 4.401683807373047e-05,
      "model_forward_time": 1.7463574409484863,
      "step": 46155
    },
    {
      "epoch": 4.401683807373047e-05,
      "step": 46155,
      "training_step_time": 7.629948139190674
    },
    {
      "epoch": 4.4017791748046874e-05,
      "model_forward_time": 1.7328360080718994,
      "step": 46156
    },
    {
      "epoch": 4.4017791748046874e-05,
      "step": 46156,
      "training_step_time": 9.213927745819092
    },
    {
      "epoch": 4.401874542236328e-05,
      "model_forward_time": 1.7366831302642822,
      "step": 46157
    },
    {
      "epoch": 4.401874542236328e-05,
      "step": 46157,
      "training_step_time": 7.620455741882324
    },
    {
      "epoch": 4.401969909667969e-05,
      "model_forward_time": 1.7332468032836914,
      "step": 46158
    },
    {
      "epoch": 4.401969909667969e-05,
      "step": 46158,
      "training_step_time": 8.533611297607422
    },
    {
      "epoch": 4.402065277099609e-05,
      "model_forward_time": 1.7340998649597168,
      "step": 46159
    },
    {
      "epoch": 4.402065277099609e-05,
      "step": 46159,
      "training_step_time": 8.496594905853271
    },
    {
      "epoch": 4.40216064453125e-05,
      "grad_norm": 0.0936347097158432,
      "learning_rate": 6.040798379441758e-06,
      "loss": 0.0588,
      "step": 46160
    },
    {
      "epoch": 4.40216064453125e-05,
      "model_forward_time": 1.7335679531097412,
      "step": 46160
    },
    {
      "dynamics_loss_avg": 0.06352079398930073,
      "epoch": 4.40216064453125e-05,
      "step": 46160
    },
    {
      "action_loss_avg": 0.0013484063209034502,
      "epoch": 4.40216064453125e-05,
      "step": 46160
    },
    {
      "epoch": 4.40216064453125e-05,
      "step": 46160,
      "training_step_time": 9.410571098327637
    },
    {
      "epoch": 4.402256011962891e-05,
      "model_forward_time": 1.7814080715179443,
      "step": 46161
    },
    {
      "epoch": 4.402256011962891e-05,
      "step": 46161,
      "training_step_time": 8.276682138442993
    },
    {
      "epoch": 4.402351379394531e-05,
      "model_forward_time": 1.7292332649230957,
      "step": 46162
    },
    {
      "epoch": 4.402351379394531e-05,
      "step": 46162,
      "training_step_time": 7.812500715255737
    },
    {
      "epoch": 4.4024467468261716e-05,
      "model_forward_time": 1.7233948707580566,
      "step": 46163
    },
    {
      "epoch": 4.4024467468261716e-05,
      "step": 46163,
      "training_step_time": 8.745173931121826
    },
    {
      "epoch": 4.4025421142578126e-05,
      "model_forward_time": 1.7301816940307617,
      "step": 46164
    },
    {
      "epoch": 4.4025421142578126e-05,
      "step": 46164,
      "training_step_time": 8.462397813796997
    },
    {
      "epoch": 4.402637481689453e-05,
      "model_forward_time": 1.7400290966033936,
      "step": 46165
    },
    {
      "epoch": 4.402637481689453e-05,
      "step": 46165,
      "training_step_time": 7.804196834564209
    },
    {
      "epoch": 4.4027328491210934e-05,
      "model_forward_time": 1.750791311264038,
      "step": 46166
    },
    {
      "epoch": 4.4027328491210934e-05,
      "step": 46166,
      "training_step_time": 8.781585454940796
    },
    {
      "epoch": 4.4028282165527345e-05,
      "model_forward_time": 1.742727279663086,
      "step": 46167
    },
    {
      "epoch": 4.4028282165527345e-05,
      "step": 46167,
      "training_step_time": 7.551158428192139
    },
    {
      "epoch": 4.402923583984375e-05,
      "model_forward_time": 1.7398555278778076,
      "step": 46168
    },
    {
      "epoch": 4.402923583984375e-05,
      "step": 46168,
      "training_step_time": 8.001582860946655
    },
    {
      "epoch": 4.403018951416015e-05,
      "model_forward_time": 1.7196438312530518,
      "step": 46169
    },
    {
      "epoch": 4.403018951416015e-05,
      "step": 46169,
      "training_step_time": 9.577637910842896
    },
    {
      "epoch": 4.4031143188476564e-05,
      "grad_norm": 0.09596794098615646,
      "learning_rate": 6.039181072236249e-06,
      "loss": 0.0601,
      "step": 46170
    },
    {
      "epoch": 4.4031143188476564e-05,
      "model_forward_time": 2.330873727798462,
      "step": 46170
    },
    {
      "dynamics_loss_avg": 0.07603435255587102,
      "epoch": 4.4031143188476564e-05,
      "step": 46170
    },
    {
      "action_loss_avg": 0.0020238348573911936,
      "epoch": 4.4031143188476564e-05,
      "step": 46170
    },
    {
      "epoch": 4.4031143188476564e-05,
      "step": 46170,
      "training_step_time": 8.519676208496094
    },
    {
      "epoch": 4.403209686279297e-05,
      "model_forward_time": 1.7311935424804688,
      "step": 46171
    },
    {
      "epoch": 4.403209686279297e-05,
      "step": 46171,
      "training_step_time": 7.753082752227783
    },
    {
      "epoch": 4.403305053710937e-05,
      "model_forward_time": 1.735802412033081,
      "step": 46172
    },
    {
      "epoch": 4.403305053710937e-05,
      "step": 46172,
      "training_step_time": 8.822861671447754
    },
    {
      "epoch": 4.403400421142578e-05,
      "model_forward_time": 1.7319419384002686,
      "step": 46173
    },
    {
      "epoch": 4.403400421142578e-05,
      "step": 46173,
      "training_step_time": 7.677850961685181
    },
    {
      "epoch": 4.403495788574219e-05,
      "model_forward_time": 1.730302333831787,
      "step": 46174
    },
    {
      "epoch": 4.403495788574219e-05,
      "step": 46174,
      "training_step_time": 7.8233160972595215
    },
    {
      "epoch": 4.403591156005859e-05,
      "model_forward_time": 1.7221202850341797,
      "step": 46175
    },
    {
      "epoch": 4.403591156005859e-05,
      "step": 46175,
      "training_step_time": 8.024412393569946
    },
    {
      "epoch": 4.4036865234375e-05,
      "model_forward_time": 1.7291686534881592,
      "step": 46176
    },
    {
      "epoch": 4.4036865234375e-05,
      "step": 46176,
      "training_step_time": 8.346279859542847
    },
    {
      "epoch": 4.4037818908691406e-05,
      "model_forward_time": 1.750169277191162,
      "step": 46177
    },
    {
      "epoch": 4.4037818908691406e-05,
      "step": 46177,
      "training_step_time": 9.898319959640503
    },
    {
      "epoch": 4.403877258300781e-05,
      "model_forward_time": 1.7192420959472656,
      "step": 46178
    },
    {
      "epoch": 4.403877258300781e-05,
      "step": 46178,
      "training_step_time": 8.683654069900513
    },
    {
      "epoch": 4.403972625732422e-05,
      "model_forward_time": 2.3271260261535645,
      "step": 46179
    },
    {
      "epoch": 4.403972625732422e-05,
      "step": 46179,
      "training_step_time": 8.751212358474731
    },
    {
      "epoch": 4.4040679931640625e-05,
      "grad_norm": 0.11865091323852539,
      "learning_rate": 6.037563651387458e-06,
      "loss": 0.0612,
      "step": 46180
    },
    {
      "epoch": 4.4040679931640625e-05,
      "model_forward_time": 1.731734037399292,
      "step": 46180
    },
    {
      "dynamics_loss_avg": 0.07004651725292206,
      "epoch": 4.4040679931640625e-05,
      "step": 46180
    },
    {
      "action_loss_avg": 0.0012087762501323597,
      "epoch": 4.4040679931640625e-05,
      "step": 46180
    },
    {
      "epoch": 4.4040679931640625e-05,
      "step": 46180,
      "training_step_time": 7.7534263134002686
    },
    {
      "epoch": 4.404163360595703e-05,
      "model_forward_time": 1.7365498542785645,
      "step": 46181
    },
    {
      "epoch": 4.404163360595703e-05,
      "step": 46181,
      "training_step_time": 7.882677793502808
    },
    {
      "epoch": 4.404258728027344e-05,
      "model_forward_time": 1.7206127643585205,
      "step": 46182
    },
    {
      "epoch": 4.404258728027344e-05,
      "step": 46182,
      "training_step_time": 7.901119232177734
    },
    {
      "epoch": 4.4043540954589844e-05,
      "model_forward_time": 1.7326066493988037,
      "step": 46183
    },
    {
      "epoch": 4.4043540954589844e-05,
      "step": 46183,
      "training_step_time": 7.570054292678833
    },
    {
      "epoch": 4.404449462890625e-05,
      "model_forward_time": 1.7346405982971191,
      "step": 46184
    },
    {
      "epoch": 4.404449462890625e-05,
      "step": 46184,
      "training_step_time": 8.520299911499023
    },
    {
      "epoch": 4.404544830322266e-05,
      "model_forward_time": 1.7351701259613037,
      "step": 46185
    },
    {
      "epoch": 4.404544830322266e-05,
      "step": 46185,
      "training_step_time": 7.582281112670898
    },
    {
      "epoch": 4.404640197753906e-05,
      "model_forward_time": 1.7393913269042969,
      "step": 46186
    },
    {
      "epoch": 4.404640197753906e-05,
      "step": 46186,
      "training_step_time": 7.594435214996338
    },
    {
      "epoch": 4.4047355651855466e-05,
      "model_forward_time": 1.7326123714447021,
      "step": 46187
    },
    {
      "epoch": 4.4047355651855466e-05,
      "step": 46187,
      "training_step_time": 7.885618209838867
    },
    {
      "epoch": 4.404830932617188e-05,
      "model_forward_time": 2.3553547859191895,
      "step": 46188
    },
    {
      "epoch": 4.404830932617188e-05,
      "step": 46188,
      "training_step_time": 9.41822361946106
    },
    {
      "epoch": 4.404926300048828e-05,
      "model_forward_time": 1.7373080253601074,
      "step": 46189
    },
    {
      "epoch": 4.404926300048828e-05,
      "step": 46189,
      "training_step_time": 7.817108392715454
    },
    {
      "epoch": 4.4050216674804685e-05,
      "grad_norm": 0.0948929637670517,
      "learning_rate": 6.035946117072267e-06,
      "loss": 0.059,
      "step": 46190
    },
    {
      "epoch": 4.4050216674804685e-05,
      "model_forward_time": 1.7303588390350342,
      "step": 46190
    },
    {
      "dynamics_loss_avg": 0.052849835343658925,
      "epoch": 4.4050216674804685e-05,
      "step": 46190
    },
    {
      "action_loss_avg": 0.0014120847190497443,
      "epoch": 4.4050216674804685e-05,
      "step": 46190
    },
    {
      "epoch": 4.4050216674804685e-05,
      "step": 46190,
      "training_step_time": 8.474056959152222
    },
    {
      "epoch": 4.4051170349121096e-05,
      "model_forward_time": 1.7536163330078125,
      "step": 46191
    },
    {
      "epoch": 4.4051170349121096e-05,
      "step": 46191,
      "training_step_time": 7.8231470584869385
    },
    {
      "epoch": 4.40521240234375e-05,
      "model_forward_time": 1.754242181777954,
      "step": 46192
    },
    {
      "epoch": 4.40521240234375e-05,
      "step": 46192,
      "training_step_time": 7.6865010261535645
    },
    {
      "epoch": 4.4053077697753904e-05,
      "model_forward_time": 1.738097906112671,
      "step": 46193
    },
    {
      "epoch": 4.4053077697753904e-05,
      "step": 46193,
      "training_step_time": 8.08715534210205
    },
    {
      "epoch": 4.4054031372070315e-05,
      "model_forward_time": 1.750943660736084,
      "step": 46194
    },
    {
      "epoch": 4.4054031372070315e-05,
      "step": 46194,
      "training_step_time": 7.599947214126587
    },
    {
      "epoch": 4.405498504638672e-05,
      "model_forward_time": 1.734161615371704,
      "step": 46195
    },
    {
      "epoch": 4.405498504638672e-05,
      "step": 46195,
      "training_step_time": 7.704021215438843
    },
    {
      "epoch": 4.405593872070312e-05,
      "model_forward_time": 1.7428288459777832,
      "step": 46196
    },
    {
      "epoch": 4.405593872070312e-05,
      "step": 46196,
      "training_step_time": 7.894740581512451
    },
    {
      "epoch": 4.4056892395019534e-05,
      "model_forward_time": 2.2970657348632812,
      "step": 46197
    },
    {
      "epoch": 4.4056892395019534e-05,
      "step": 46197,
      "training_step_time": 8.332202434539795
    },
    {
      "epoch": 4.405784606933594e-05,
      "model_forward_time": 1.7397568225860596,
      "step": 46198
    },
    {
      "epoch": 4.405784606933594e-05,
      "step": 46198,
      "training_step_time": 8.737217664718628
    },
    {
      "epoch": 4.405879974365234e-05,
      "model_forward_time": 1.7497053146362305,
      "step": 46199
    },
    {
      "epoch": 4.405879974365234e-05,
      "step": 46199,
      "training_step_time": 7.8270697593688965
    },
    {
      "epoch": 4.405975341796875e-05,
      "grad_norm": 0.140665665268898,
      "learning_rate": 6.034328469467566e-06,
      "loss": 0.0599,
      "step": 46200
    },
    {
      "epoch": 4.405975341796875e-05,
      "model_forward_time": 1.7412188053131104,
      "step": 46200
    },
    {
      "dynamics_loss_avg": 0.0719751924276352,
      "epoch": 4.405975341796875e-05,
      "step": 46200
    },
    {
      "action_loss_avg": 0.0017628888483159244,
      "epoch": 4.405975341796875e-05,
      "step": 46200
    },
    {
      "epoch": 4.405975341796875e-05,
      "step": 46200,
      "training_step_time": 7.767705202102661
    },
    {
      "epoch": 4.406070709228516e-05,
      "model_forward_time": 1.7429711818695068,
      "step": 46201
    },
    {
      "epoch": 4.406070709228516e-05,
      "step": 46201,
      "training_step_time": 7.62767219543457
    },
    {
      "epoch": 4.406166076660156e-05,
      "model_forward_time": 1.7512924671173096,
      "step": 46202
    },
    {
      "epoch": 4.406166076660156e-05,
      "step": 46202,
      "training_step_time": 8.86400318145752
    },
    {
      "epoch": 4.406261444091797e-05,
      "model_forward_time": 1.742760419845581,
      "step": 46203
    },
    {
      "epoch": 4.406261444091797e-05,
      "step": 46203,
      "training_step_time": 7.533055782318115
    },
    {
      "epoch": 4.4063568115234376e-05,
      "model_forward_time": 1.7129361629486084,
      "step": 46204
    },
    {
      "epoch": 4.4063568115234376e-05,
      "step": 46204,
      "training_step_time": 7.591680526733398
    },
    {
      "epoch": 4.406452178955078e-05,
      "model_forward_time": 1.7243237495422363,
      "step": 46205
    },
    {
      "epoch": 4.406452178955078e-05,
      "step": 46205,
      "training_step_time": 10.398582220077515
    },
    {
      "epoch": 4.406547546386719e-05,
      "model_forward_time": 2.297064781188965,
      "step": 46206
    },
    {
      "epoch": 4.406547546386719e-05,
      "step": 46206,
      "training_step_time": 8.95529055595398
    },
    {
      "epoch": 4.4066429138183594e-05,
      "model_forward_time": 1.7737619876861572,
      "step": 46207
    },
    {
      "epoch": 4.4066429138183594e-05,
      "step": 46207,
      "training_step_time": 8.808998584747314
    },
    {
      "epoch": 4.40673828125e-05,
      "model_forward_time": 1.753561019897461,
      "step": 46208
    },
    {
      "epoch": 4.40673828125e-05,
      "step": 46208,
      "training_step_time": 8.49274730682373
    },
    {
      "epoch": 4.406833648681641e-05,
      "model_forward_time": 1.7304420471191406,
      "step": 46209
    },
    {
      "epoch": 4.406833648681641e-05,
      "step": 46209,
      "training_step_time": 8.34644103050232
    },
    {
      "epoch": 4.406929016113281e-05,
      "grad_norm": 0.11040036380290985,
      "learning_rate": 6.0327107087502604e-06,
      "loss": 0.0591,
      "step": 46210
    },
    {
      "epoch": 4.406929016113281e-05,
      "model_forward_time": 1.739450454711914,
      "step": 46210
    },
    {
      "dynamics_loss_avg": 0.07049027793109416,
      "epoch": 4.406929016113281e-05,
      "step": 46210
    },
    {
      "action_loss_avg": 0.001971228315960616,
      "epoch": 4.406929016113281e-05,
      "step": 46210
    },
    {
      "epoch": 4.406929016113281e-05,
      "step": 46210,
      "training_step_time": 7.870167970657349
    },
    {
      "epoch": 4.407024383544922e-05,
      "model_forward_time": 1.7206535339355469,
      "step": 46211
    },
    {
      "epoch": 4.407024383544922e-05,
      "step": 46211,
      "training_step_time": 7.831424951553345
    },
    {
      "epoch": 4.407119750976563e-05,
      "model_forward_time": 1.737008810043335,
      "step": 46212
    },
    {
      "epoch": 4.407119750976563e-05,
      "step": 46212,
      "training_step_time": 7.806699752807617
    },
    {
      "epoch": 4.407215118408203e-05,
      "model_forward_time": 1.744408369064331,
      "step": 46213
    },
    {
      "epoch": 4.407215118408203e-05,
      "step": 46213,
      "training_step_time": 7.488271713256836
    },
    {
      "epoch": 4.4073104858398436e-05,
      "model_forward_time": 1.7298729419708252,
      "step": 46214
    },
    {
      "epoch": 4.4073104858398436e-05,
      "step": 46214,
      "training_step_time": 8.59855604171753
    },
    {
      "epoch": 4.407405853271485e-05,
      "model_forward_time": 1.7249066829681396,
      "step": 46215
    },
    {
      "epoch": 4.407405853271485e-05,
      "step": 46215,
      "training_step_time": 9.65886902809143
    },
    {
      "epoch": 4.407501220703125e-05,
      "model_forward_time": 1.7753119468688965,
      "step": 46216
    },
    {
      "epoch": 4.407501220703125e-05,
      "step": 46216,
      "training_step_time": 7.9123029708862305
    },
    {
      "epoch": 4.4075965881347655e-05,
      "model_forward_time": 1.737497091293335,
      "step": 46217
    },
    {
      "epoch": 4.4075965881347655e-05,
      "step": 46217,
      "training_step_time": 7.895012378692627
    },
    {
      "epoch": 4.4076919555664066e-05,
      "model_forward_time": 1.7400603294372559,
      "step": 46218
    },
    {
      "epoch": 4.4076919555664066e-05,
      "step": 46218,
      "training_step_time": 7.925812244415283
    },
    {
      "epoch": 4.407787322998047e-05,
      "model_forward_time": 1.7360191345214844,
      "step": 46219
    },
    {
      "epoch": 4.407787322998047e-05,
      "step": 46219,
      "training_step_time": 7.847967147827148
    },
    {
      "epoch": 4.4078826904296874e-05,
      "grad_norm": 0.12485742568969727,
      "learning_rate": 6.031092835097262e-06,
      "loss": 0.0595,
      "step": 46220
    },
    {
      "epoch": 4.4078826904296874e-05,
      "model_forward_time": 1.738175868988037,
      "step": 46220
    },
    {
      "dynamics_loss_avg": 0.06439799629151821,
      "epoch": 4.4078826904296874e-05,
      "step": 46220
    },
    {
      "action_loss_avg": 0.0017176469962578266,
      "epoch": 4.4078826904296874e-05,
      "step": 46220
    },
    {
      "epoch": 4.4078826904296874e-05,
      "step": 46220,
      "training_step_time": 7.836355447769165
    },
    {
      "epoch": 4.407978057861328e-05,
      "model_forward_time": 1.7361133098602295,
      "step": 46221
    },
    {
      "epoch": 4.407978057861328e-05,
      "step": 46221,
      "training_step_time": 8.621346235275269
    },
    {
      "epoch": 4.408073425292969e-05,
      "model_forward_time": 1.7339751720428467,
      "step": 46222
    },
    {
      "epoch": 4.408073425292969e-05,
      "step": 46222,
      "training_step_time": 7.515113353729248
    },
    {
      "epoch": 4.408168792724609e-05,
      "model_forward_time": 1.8033101558685303,
      "step": 46223
    },
    {
      "epoch": 4.408168792724609e-05,
      "step": 46223,
      "training_step_time": 8.453873634338379
    },
    {
      "epoch": 4.40826416015625e-05,
      "model_forward_time": 1.8162915706634521,
      "step": 46224
    },
    {
      "epoch": 4.40826416015625e-05,
      "step": 46224,
      "training_step_time": 8.537198066711426
    },
    {
      "epoch": 4.408359527587891e-05,
      "model_forward_time": 2.3574907779693604,
      "step": 46225
    },
    {
      "epoch": 4.408359527587891e-05,
      "step": 46225,
      "training_step_time": 9.328427791595459
    },
    {
      "epoch": 4.408454895019531e-05,
      "model_forward_time": 1.8271424770355225,
      "step": 46226
    },
    {
      "epoch": 4.408454895019531e-05,
      "step": 46226,
      "training_step_time": 7.800926446914673
    },
    {
      "epoch": 4.4085502624511716e-05,
      "model_forward_time": 1.820655345916748,
      "step": 46227
    },
    {
      "epoch": 4.4085502624511716e-05,
      "step": 46227,
      "training_step_time": 8.794151306152344
    },
    {
      "epoch": 4.4086456298828126e-05,
      "model_forward_time": 1.8195123672485352,
      "step": 46228
    },
    {
      "epoch": 4.4086456298828126e-05,
      "step": 46228,
      "training_step_time": 7.790522813796997
    },
    {
      "epoch": 4.408740997314453e-05,
      "model_forward_time": 1.8253672122955322,
      "step": 46229
    },
    {
      "epoch": 4.408740997314453e-05,
      "step": 46229,
      "training_step_time": 7.568164110183716
    },
    {
      "epoch": 4.4088363647460935e-05,
      "grad_norm": 0.15624049305915833,
      "learning_rate": 6.029474848685503e-06,
      "loss": 0.0613,
      "step": 46230
    },
    {
      "epoch": 4.4088363647460935e-05,
      "model_forward_time": 1.828540563583374,
      "step": 46230
    },
    {
      "dynamics_loss_avg": 0.06425484158098697,
      "epoch": 4.4088363647460935e-05,
      "step": 46230
    },
    {
      "action_loss_avg": 0.0015899506775895133,
      "epoch": 4.4088363647460935e-05,
      "step": 46230
    },
    {
      "epoch": 4.4088363647460935e-05,
      "step": 46230,
      "training_step_time": 7.813997983932495
    },
    {
      "epoch": 4.4089317321777345e-05,
      "model_forward_time": 1.8206324577331543,
      "step": 46231
    },
    {
      "epoch": 4.4089317321777345e-05,
      "step": 46231,
      "training_step_time": 7.597581148147583
    },
    {
      "epoch": 4.409027099609375e-05,
      "model_forward_time": 1.8162858486175537,
      "step": 46232
    },
    {
      "epoch": 4.409027099609375e-05,
      "step": 46232,
      "training_step_time": 8.77443528175354
    },
    {
      "epoch": 4.409122467041015e-05,
      "model_forward_time": 1.8177084922790527,
      "step": 46233
    },
    {
      "epoch": 4.409122467041015e-05,
      "step": 46233,
      "training_step_time": 10.407243013381958
    },
    {
      "epoch": 4.4092178344726564e-05,
      "model_forward_time": 2.3573739528656006,
      "step": 46234
    },
    {
      "epoch": 4.4092178344726564e-05,
      "step": 46234,
      "training_step_time": 7.80279016494751
    },
    {
      "epoch": 4.409313201904297e-05,
      "model_forward_time": 1.8187265396118164,
      "step": 46235
    },
    {
      "epoch": 4.409313201904297e-05,
      "step": 46235,
      "training_step_time": 8.398289680480957
    },
    {
      "epoch": 4.409408569335937e-05,
      "model_forward_time": 1.8289854526519775,
      "step": 46236
    },
    {
      "epoch": 4.409408569335937e-05,
      "step": 46236,
      "training_step_time": 7.892346143722534
    },
    {
      "epoch": 4.409503936767578e-05,
      "model_forward_time": 1.8246467113494873,
      "step": 46237
    },
    {
      "epoch": 4.409503936767578e-05,
      "step": 46237,
      "training_step_time": 8.036087989807129
    },
    {
      "epoch": 4.409599304199219e-05,
      "model_forward_time": 1.8238239288330078,
      "step": 46238
    },
    {
      "epoch": 4.409599304199219e-05,
      "step": 46238,
      "training_step_time": 7.602581024169922
    },
    {
      "epoch": 4.409694671630859e-05,
      "model_forward_time": 1.8152077198028564,
      "step": 46239
    },
    {
      "epoch": 4.409694671630859e-05,
      "step": 46239,
      "training_step_time": 8.390326261520386
    },
    {
      "epoch": 4.4097900390625e-05,
      "grad_norm": 0.13860131800174713,
      "learning_rate": 6.027856749691922e-06,
      "loss": 0.0597,
      "step": 46240
    },
    {
      "epoch": 4.4097900390625e-05,
      "model_forward_time": 1.8196358680725098,
      "step": 46240
    },
    {
      "dynamics_loss_avg": 0.04672842500731349,
      "epoch": 4.4097900390625e-05,
      "step": 46240
    },
    {
      "action_loss_avg": 0.0009548147092573345,
      "epoch": 4.4097900390625e-05,
      "step": 46240
    },
    {
      "epoch": 4.4097900390625e-05,
      "step": 46240,
      "training_step_time": 7.598637819290161
    },
    {
      "epoch": 4.4098854064941406e-05,
      "model_forward_time": 1.8214325904846191,
      "step": 46241
    },
    {
      "epoch": 4.4098854064941406e-05,
      "step": 46241,
      "training_step_time": 7.961622953414917
    },
    {
      "epoch": 4.409980773925781e-05,
      "model_forward_time": 1.8296442031860352,
      "step": 46242
    },
    {
      "epoch": 4.409980773925781e-05,
      "step": 46242,
      "training_step_time": 7.983424425125122
    },
    {
      "epoch": 4.410076141357422e-05,
      "model_forward_time": 2.3556723594665527,
      "step": 46243
    },
    {
      "epoch": 4.410076141357422e-05,
      "step": 46243,
      "training_step_time": 9.518909215927124
    },
    {
      "epoch": 4.4101715087890625e-05,
      "model_forward_time": 1.8102571964263916,
      "step": 46244
    },
    {
      "epoch": 4.4101715087890625e-05,
      "step": 46244,
      "training_step_time": 7.930216073989868
    },
    {
      "epoch": 4.410266876220703e-05,
      "model_forward_time": 1.8212172985076904,
      "step": 46245
    },
    {
      "epoch": 4.410266876220703e-05,
      "step": 46245,
      "training_step_time": 8.745235919952393
    },
    {
      "epoch": 4.410362243652344e-05,
      "model_forward_time": 1.826005220413208,
      "step": 46246
    },
    {
      "epoch": 4.410362243652344e-05,
      "step": 46246,
      "training_step_time": 8.717804908752441
    },
    {
      "epoch": 4.4104576110839844e-05,
      "model_forward_time": 1.8243122100830078,
      "step": 46247
    },
    {
      "epoch": 4.4104576110839844e-05,
      "step": 46247,
      "training_step_time": 7.754751205444336
    },
    {
      "epoch": 4.410552978515625e-05,
      "model_forward_time": 1.8262603282928467,
      "step": 46248
    },
    {
      "epoch": 4.410552978515625e-05,
      "step": 46248,
      "training_step_time": 7.827786922454834
    },
    {
      "epoch": 4.410648345947266e-05,
      "model_forward_time": 1.8347094058990479,
      "step": 46249
    },
    {
      "epoch": 4.410648345947266e-05,
      "step": 46249,
      "training_step_time": 7.832735776901245
    },
    {
      "epoch": 4.410743713378906e-05,
      "grad_norm": 0.1046161949634552,
      "learning_rate": 6.026238538293472e-06,
      "loss": 0.0573,
      "step": 46250
    },
    {
      "epoch": 4.410743713378906e-05,
      "model_forward_time": 1.8146536350250244,
      "step": 46250
    },
    {
      "dynamics_loss_avg": 0.04991069436073303,
      "epoch": 4.410743713378906e-05,
      "step": 46250
    },
    {
      "action_loss_avg": 0.0022525603184476496,
      "epoch": 4.410743713378906e-05,
      "step": 46250
    },
    {
      "epoch": 4.410743713378906e-05,
      "step": 46250,
      "training_step_time": 7.993143081665039
    },
    {
      "epoch": 4.4108390808105467e-05,
      "model_forward_time": 1.8250908851623535,
      "step": 46251
    },
    {
      "epoch": 4.4108390808105467e-05,
      "step": 46251,
      "training_step_time": 9.719759941101074
    },
    {
      "epoch": 4.410934448242188e-05,
      "model_forward_time": 2.3664636611938477,
      "step": 46252
    },
    {
      "epoch": 4.410934448242188e-05,
      "step": 46252,
      "training_step_time": 9.627087116241455
    },
    {
      "epoch": 4.411029815673828e-05,
      "model_forward_time": 1.8404762744903564,
      "step": 46253
    },
    {
      "epoch": 4.411029815673828e-05,
      "step": 46253,
      "training_step_time": 8.593407154083252
    },
    {
      "epoch": 4.4111251831054685e-05,
      "model_forward_time": 1.8288302421569824,
      "step": 46254
    },
    {
      "epoch": 4.4111251831054685e-05,
      "step": 46254,
      "training_step_time": 8.637232542037964
    },
    {
      "epoch": 4.4112205505371096e-05,
      "model_forward_time": 1.8268330097198486,
      "step": 46255
    },
    {
      "epoch": 4.4112205505371096e-05,
      "step": 46255,
      "training_step_time": 7.86020302772522
    },
    {
      "epoch": 4.41131591796875e-05,
      "model_forward_time": 1.8163337707519531,
      "step": 46256
    },
    {
      "epoch": 4.41131591796875e-05,
      "step": 46256,
      "training_step_time": 7.948439836502075
    },
    {
      "epoch": 4.4114112854003904e-05,
      "model_forward_time": 1.8229501247406006,
      "step": 46257
    },
    {
      "epoch": 4.4114112854003904e-05,
      "step": 46257,
      "training_step_time": 8.887305736541748
    },
    {
      "epoch": 4.4115066528320315e-05,
      "model_forward_time": 1.8245971202850342,
      "step": 46258
    },
    {
      "epoch": 4.4115066528320315e-05,
      "step": 46258,
      "training_step_time": 8.655166625976562
    },
    {
      "epoch": 4.411602020263672e-05,
      "model_forward_time": 1.8183956146240234,
      "step": 46259
    },
    {
      "epoch": 4.411602020263672e-05,
      "step": 46259,
      "training_step_time": 8.571614265441895
    },
    {
      "epoch": 4.411697387695312e-05,
      "grad_norm": 0.12440123409032822,
      "learning_rate": 6.024620214667119e-06,
      "loss": 0.0595,
      "step": 46260
    },
    {
      "epoch": 4.411697387695312e-05,
      "model_forward_time": 1.8147764205932617,
      "step": 46260
    },
    {
      "dynamics_loss_avg": 0.08840515613555908,
      "epoch": 4.411697387695312e-05,
      "step": 46260
    },
    {
      "action_loss_avg": 0.0018263522244524211,
      "epoch": 4.411697387695312e-05,
      "step": 46260
    },
    {
      "epoch": 4.411697387695312e-05,
      "step": 46260,
      "training_step_time": 7.812622308731079
    },
    {
      "epoch": 4.4117927551269534e-05,
      "model_forward_time": 2.3373799324035645,
      "step": 46261
    },
    {
      "epoch": 4.4117927551269534e-05,
      "step": 46261,
      "training_step_time": 8.528338193893433
    },
    {
      "epoch": 4.411888122558594e-05,
      "model_forward_time": 1.8274431228637695,
      "step": 46262
    },
    {
      "epoch": 4.411888122558594e-05,
      "step": 46262,
      "training_step_time": 7.828833818435669
    },
    {
      "epoch": 4.411983489990234e-05,
      "model_forward_time": 1.8278489112854004,
      "step": 46263
    },
    {
      "epoch": 4.411983489990234e-05,
      "step": 46263,
      "training_step_time": 8.773463487625122
    },
    {
      "epoch": 4.412078857421875e-05,
      "model_forward_time": 1.826958179473877,
      "step": 46264
    },
    {
      "epoch": 4.412078857421875e-05,
      "step": 46264,
      "training_step_time": 9.75325608253479
    },
    {
      "epoch": 4.412174224853516e-05,
      "model_forward_time": 1.825099229812622,
      "step": 46265
    },
    {
      "epoch": 4.412174224853516e-05,
      "step": 46265,
      "training_step_time": 7.860834121704102
    },
    {
      "epoch": 4.412269592285156e-05,
      "model_forward_time": 1.8289084434509277,
      "step": 46266
    },
    {
      "epoch": 4.412269592285156e-05,
      "step": 46266,
      "training_step_time": 8.686357498168945
    },
    {
      "epoch": 4.412364959716797e-05,
      "model_forward_time": 1.8239762783050537,
      "step": 46267
    },
    {
      "epoch": 4.412364959716797e-05,
      "step": 46267,
      "training_step_time": 7.734689235687256
    },
    {
      "epoch": 4.4124603271484376e-05,
      "model_forward_time": 1.8198988437652588,
      "step": 46268
    },
    {
      "epoch": 4.4124603271484376e-05,
      "step": 46268,
      "training_step_time": 8.610212564468384
    },
    {
      "epoch": 4.412555694580078e-05,
      "model_forward_time": 1.8138010501861572,
      "step": 46269
    },
    {
      "epoch": 4.412555694580078e-05,
      "step": 46269,
      "training_step_time": 8.74599289894104
    },
    {
      "epoch": 4.412651062011719e-05,
      "grad_norm": 0.18541878461837769,
      "learning_rate": 6.0230017789898385e-06,
      "loss": 0.0649,
      "step": 46270
    },
    {
      "epoch": 4.412651062011719e-05,
      "model_forward_time": 2.343428134918213,
      "step": 46270
    },
    {
      "dynamics_loss_avg": 0.06497957836836576,
      "epoch": 4.412651062011719e-05,
      "step": 46270
    },
    {
      "action_loss_avg": 0.0011360840348061175,
      "epoch": 4.412651062011719e-05,
      "step": 46270
    },
    {
      "epoch": 4.412651062011719e-05,
      "step": 46270,
      "training_step_time": 7.764926910400391
    },
    {
      "epoch": 4.4127464294433595e-05,
      "model_forward_time": 1.8201227188110352,
      "step": 46271
    },
    {
      "epoch": 4.4127464294433595e-05,
      "step": 46271,
      "training_step_time": 8.442338705062866
    },
    {
      "epoch": 4.412841796875e-05,
      "model_forward_time": 1.8241350650787354,
      "step": 46272
    },
    {
      "epoch": 4.412841796875e-05,
      "step": 46272,
      "training_step_time": 7.810246467590332
    },
    {
      "epoch": 4.412937164306641e-05,
      "model_forward_time": 1.8262791633605957,
      "step": 46273
    },
    {
      "epoch": 4.412937164306641e-05,
      "step": 46273,
      "training_step_time": 7.771032094955444
    },
    {
      "epoch": 4.413032531738281e-05,
      "model_forward_time": 1.8279869556427002,
      "step": 46274
    },
    {
      "epoch": 4.413032531738281e-05,
      "step": 46274,
      "training_step_time": 7.898528575897217
    },
    {
      "epoch": 4.413127899169922e-05,
      "model_forward_time": 1.8217391967773438,
      "step": 46275
    },
    {
      "epoch": 4.413127899169922e-05,
      "step": 46275,
      "training_step_time": 7.595694303512573
    },
    {
      "epoch": 4.413223266601563e-05,
      "model_forward_time": 1.8246285915374756,
      "step": 46276
    },
    {
      "epoch": 4.413223266601563e-05,
      "step": 46276,
      "training_step_time": 8.228891849517822
    },
    {
      "epoch": 4.413318634033203e-05,
      "model_forward_time": 1.8174076080322266,
      "step": 46277
    },
    {
      "epoch": 4.413318634033203e-05,
      "step": 46277,
      "training_step_time": 7.895352125167847
    },
    {
      "epoch": 4.4134140014648436e-05,
      "model_forward_time": 1.8196024894714355,
      "step": 46278
    },
    {
      "epoch": 4.4134140014648436e-05,
      "step": 46278,
      "training_step_time": 8.251832962036133
    },
    {
      "epoch": 4.413509368896485e-05,
      "model_forward_time": 2.3539652824401855,
      "step": 46279
    },
    {
      "epoch": 4.413509368896485e-05,
      "step": 46279,
      "training_step_time": 7.7984418869018555
    },
    {
      "epoch": 4.413604736328125e-05,
      "grad_norm": 0.11892393231391907,
      "learning_rate": 6.021383231438622e-06,
      "loss": 0.0622,
      "step": 46280
    },
    {
      "epoch": 4.413604736328125e-05,
      "model_forward_time": 1.8215277194976807,
      "step": 46280
    },
    {
      "dynamics_loss_avg": 0.061022453010082245,
      "epoch": 4.413604736328125e-05,
      "step": 46280
    },
    {
      "action_loss_avg": 0.001437450060620904,
      "epoch": 4.413604736328125e-05,
      "step": 46280
    },
    {
      "epoch": 4.413604736328125e-05,
      "step": 46280,
      "training_step_time": 8.469725370407104
    },
    {
      "epoch": 4.4137001037597655e-05,
      "model_forward_time": 1.8276491165161133,
      "step": 46281
    },
    {
      "epoch": 4.4137001037597655e-05,
      "step": 46281,
      "training_step_time": 8.16724419593811
    },
    {
      "epoch": 4.413795471191406e-05,
      "model_forward_time": 1.8260385990142822,
      "step": 46282
    },
    {
      "epoch": 4.413795471191406e-05,
      "step": 46282,
      "training_step_time": 7.727516412734985
    },
    {
      "epoch": 4.413890838623047e-05,
      "model_forward_time": 1.8277909755706787,
      "step": 46283
    },
    {
      "epoch": 4.413890838623047e-05,
      "step": 46283,
      "training_step_time": 7.813385725021362
    },
    {
      "epoch": 4.4139862060546874e-05,
      "model_forward_time": 1.830336570739746,
      "step": 46284
    },
    {
      "epoch": 4.4139862060546874e-05,
      "step": 46284,
      "training_step_time": 7.559012174606323
    },
    {
      "epoch": 4.414081573486328e-05,
      "model_forward_time": 1.8167946338653564,
      "step": 46285
    },
    {
      "epoch": 4.414081573486328e-05,
      "step": 46285,
      "training_step_time": 7.671590328216553
    },
    {
      "epoch": 4.414176940917969e-05,
      "model_forward_time": 1.825232982635498,
      "step": 46286
    },
    {
      "epoch": 4.414176940917969e-05,
      "step": 46286,
      "training_step_time": 7.845930576324463
    },
    {
      "epoch": 4.414272308349609e-05,
      "model_forward_time": 1.8243944644927979,
      "step": 46287
    },
    {
      "epoch": 4.414272308349609e-05,
      "step": 46287,
      "training_step_time": 7.570874452590942
    },
    {
      "epoch": 4.41436767578125e-05,
      "model_forward_time": 2.3574068546295166,
      "step": 46288
    },
    {
      "epoch": 4.41436767578125e-05,
      "step": 46288,
      "training_step_time": 7.7667741775512695
    },
    {
      "epoch": 4.414463043212891e-05,
      "model_forward_time": 1.8200819492340088,
      "step": 46289
    },
    {
      "epoch": 4.414463043212891e-05,
      "step": 46289,
      "training_step_time": 8.991187334060669
    },
    {
      "epoch": 4.414558410644531e-05,
      "grad_norm": 0.10880814492702484,
      "learning_rate": 6.019764572190471e-06,
      "loss": 0.0605,
      "step": 46290
    },
    {
      "epoch": 4.414558410644531e-05,
      "model_forward_time": 1.8241424560546875,
      "step": 46290
    },
    {
      "dynamics_loss_avg": 0.0618972897529602,
      "epoch": 4.414558410644531e-05,
      "step": 46290
    },
    {
      "action_loss_avg": 0.001283566464553587,
      "epoch": 4.414558410644531e-05,
      "step": 46290
    },
    {
      "epoch": 4.414558410644531e-05,
      "step": 46290,
      "training_step_time": 8.721502304077148
    },
    {
      "epoch": 4.4146537780761716e-05,
      "model_forward_time": 1.8251965045928955,
      "step": 46291
    },
    {
      "epoch": 4.4146537780761716e-05,
      "step": 46291,
      "training_step_time": 8.363566875457764
    },
    {
      "epoch": 4.4147491455078127e-05,
      "model_forward_time": 1.8264851570129395,
      "step": 46292
    },
    {
      "epoch": 4.4147491455078127e-05,
      "step": 46292,
      "training_step_time": 8.207947969436646
    },
    {
      "epoch": 4.414844512939453e-05,
      "model_forward_time": 1.449648141860962,
      "step": 46293
    },
    {
      "epoch": 4.414844512939453e-05,
      "step": 46293,
      "training_step_time": 7.843874216079712
    },
    {
      "epoch": 4.4149398803710935e-05,
      "model_forward_time": 1.8255031108856201,
      "step": 46294
    },
    {
      "epoch": 4.4149398803710935e-05,
      "step": 46294,
      "training_step_time": 7.753749370574951
    },
    {
      "epoch": 4.4150352478027345e-05,
      "model_forward_time": 1.8151042461395264,
      "step": 46295
    },
    {
      "epoch": 4.4150352478027345e-05,
      "step": 46295,
      "training_step_time": 8.437302589416504
    },
    {
      "epoch": 4.415130615234375e-05,
      "model_forward_time": 1.8221821784973145,
      "step": 46296
    },
    {
      "epoch": 4.415130615234375e-05,
      "step": 46296,
      "training_step_time": 7.558640956878662
    },
    {
      "epoch": 4.4152259826660153e-05,
      "model_forward_time": 2.3458175659179688,
      "step": 46297
    },
    {
      "epoch": 4.4152259826660153e-05,
      "step": 46297,
      "training_step_time": 8.509147644042969
    },
    {
      "epoch": 4.4153213500976564e-05,
      "model_forward_time": 1.8213372230529785,
      "step": 46298
    },
    {
      "epoch": 4.4153213500976564e-05,
      "step": 46298,
      "training_step_time": 8.490671634674072
    },
    {
      "epoch": 4.415416717529297e-05,
      "model_forward_time": 1.8300399780273438,
      "step": 46299
    },
    {
      "epoch": 4.415416717529297e-05,
      "step": 46299,
      "training_step_time": 8.571156024932861
    },
    {
      "epoch": 4.415512084960937e-05,
      "grad_norm": 0.11057251691818237,
      "learning_rate": 6.0181458014224006e-06,
      "loss": 0.0588,
      "step": 46300
    },
    {
      "epoch": 4.415512084960937e-05,
      "model_forward_time": 1.8165812492370605,
      "step": 46300
    },
    {
      "dynamics_loss_avg": 0.07163868732750416,
      "epoch": 4.415512084960937e-05,
      "step": 46300
    },
    {
      "action_loss_avg": 0.0014153188589261846,
      "epoch": 4.415512084960937e-05,
      "step": 46300
    },
    {
      "epoch": 4.415512084960937e-05,
      "step": 46300,
      "training_step_time": 7.781636476516724
    },
    {
      "epoch": 4.415607452392578e-05,
      "model_forward_time": 1.8252301216125488,
      "step": 46301
    },
    {
      "epoch": 4.415607452392578e-05,
      "step": 46301,
      "training_step_time": 8.288678646087646
    },
    {
      "epoch": 4.415702819824219e-05,
      "model_forward_time": 1.8066132068634033,
      "step": 46302
    },
    {
      "epoch": 4.415702819824219e-05,
      "step": 46302,
      "training_step_time": 7.802489519119263
    },
    {
      "epoch": 4.415798187255859e-05,
      "model_forward_time": 1.8251290321350098,
      "step": 46303
    },
    {
      "epoch": 4.415798187255859e-05,
      "step": 46303,
      "training_step_time": 7.721750974655151
    },
    {
      "epoch": 4.4158935546875e-05,
      "model_forward_time": 1.823307752609253,
      "step": 46304
    },
    {
      "epoch": 4.4158935546875e-05,
      "step": 46304,
      "training_step_time": 7.754638195037842
    },
    {
      "epoch": 4.4159889221191406e-05,
      "model_forward_time": 1.8196213245391846,
      "step": 46305
    },
    {
      "epoch": 4.4159889221191406e-05,
      "step": 46305,
      "training_step_time": 7.631258726119995
    },
    {
      "epoch": 4.416084289550781e-05,
      "model_forward_time": 2.3494627475738525,
      "step": 46306
    },
    {
      "epoch": 4.416084289550781e-05,
      "step": 46306,
      "training_step_time": 7.643850326538086
    },
    {
      "epoch": 4.416179656982422e-05,
      "model_forward_time": 1.4585490226745605,
      "step": 46307
    },
    {
      "epoch": 4.416179656982422e-05,
      "step": 46307,
      "training_step_time": 8.6563081741333
    },
    {
      "epoch": 4.4162750244140625e-05,
      "model_forward_time": 1.817643404006958,
      "step": 46308
    },
    {
      "epoch": 4.4162750244140625e-05,
      "step": 46308,
      "training_step_time": 8.36258578300476
    },
    {
      "epoch": 4.416370391845703e-05,
      "model_forward_time": 1.8099682331085205,
      "step": 46309
    },
    {
      "epoch": 4.416370391845703e-05,
      "step": 46309,
      "training_step_time": 7.8061182498931885
    },
    {
      "epoch": 4.416465759277344e-05,
      "grad_norm": 0.14465804398059845,
      "learning_rate": 6.016526919311435e-06,
      "loss": 0.0593,
      "step": 46310
    },
    {
      "epoch": 4.416465759277344e-05,
      "model_forward_time": 1.8026447296142578,
      "step": 46310
    },
    {
      "dynamics_loss_avg": 0.06245719101279974,
      "epoch": 4.416465759277344e-05,
      "step": 46310
    },
    {
      "action_loss_avg": 0.0016796735973912292,
      "epoch": 4.416465759277344e-05,
      "step": 46310
    },
    {
      "epoch": 4.416465759277344e-05,
      "step": 46310,
      "training_step_time": 7.744815349578857
    },
    {
      "epoch": 4.4165611267089844e-05,
      "model_forward_time": 1.8113901615142822,
      "step": 46311
    },
    {
      "epoch": 4.4165611267089844e-05,
      "step": 46311,
      "training_step_time": 7.810915231704712
    },
    {
      "epoch": 4.416656494140625e-05,
      "model_forward_time": 1.810577392578125,
      "step": 46312
    },
    {
      "epoch": 4.416656494140625e-05,
      "step": 46312,
      "training_step_time": 7.655549049377441
    },
    {
      "epoch": 4.416751861572266e-05,
      "model_forward_time": 1.811324119567871,
      "step": 46313
    },
    {
      "epoch": 4.416751861572266e-05,
      "step": 46313,
      "training_step_time": 8.902385234832764
    },
    {
      "epoch": 4.416847229003906e-05,
      "model_forward_time": 1.806999683380127,
      "step": 46314
    },
    {
      "epoch": 4.416847229003906e-05,
      "step": 46314,
      "training_step_time": 7.608242750167847
    },
    {
      "epoch": 4.4169425964355467e-05,
      "model_forward_time": 1.8117542266845703,
      "step": 46315
    },
    {
      "epoch": 4.4169425964355467e-05,
      "step": 46315,
      "training_step_time": 8.180423974990845
    },
    {
      "epoch": 4.417037963867188e-05,
      "model_forward_time": 1.8101320266723633,
      "step": 46316
    },
    {
      "epoch": 4.417037963867188e-05,
      "step": 46316,
      "training_step_time": 9.996423721313477
    },
    {
      "epoch": 4.417133331298828e-05,
      "model_forward_time": 1.8075625896453857,
      "step": 46317
    },
    {
      "epoch": 4.417133331298828e-05,
      "step": 46317,
      "training_step_time": 9.703282594680786
    },
    {
      "epoch": 4.4172286987304685e-05,
      "model_forward_time": 1.7857298851013184,
      "step": 46318
    },
    {
      "epoch": 4.4172286987304685e-05,
      "step": 46318,
      "training_step_time": 7.806167840957642
    },
    {
      "epoch": 4.4173240661621096e-05,
      "model_forward_time": 1.805173397064209,
      "step": 46319
    },
    {
      "epoch": 4.4173240661621096e-05,
      "step": 46319,
      "training_step_time": 7.707869052886963
    },
    {
      "epoch": 4.41741943359375e-05,
      "grad_norm": 0.1627538651227951,
      "learning_rate": 6.014907926034613e-06,
      "loss": 0.0602,
      "step": 46320
    },
    {
      "epoch": 4.41741943359375e-05,
      "model_forward_time": 1.805241346359253,
      "step": 46320
    },
    {
      "dynamics_loss_avg": 0.05735959429293871,
      "epoch": 4.41741943359375e-05,
      "step": 46320
    },
    {
      "action_loss_avg": 0.0017395031463820487,
      "epoch": 4.41741943359375e-05,
      "step": 46320
    },
    {
      "epoch": 4.41741943359375e-05,
      "step": 46320,
      "training_step_time": 7.815510988235474
    },
    {
      "epoch": 4.4175148010253904e-05,
      "model_forward_time": 1.8099792003631592,
      "step": 46321
    },
    {
      "epoch": 4.4175148010253904e-05,
      "step": 46321,
      "training_step_time": 7.5291688442230225
    },
    {
      "epoch": 4.4176101684570315e-05,
      "model_forward_time": 1.8083360195159912,
      "step": 46322
    },
    {
      "epoch": 4.4176101684570315e-05,
      "step": 46322,
      "training_step_time": 7.856399297714233
    },
    {
      "epoch": 4.417705535888672e-05,
      "model_forward_time": 1.8067357540130615,
      "step": 46323
    },
    {
      "epoch": 4.417705535888672e-05,
      "step": 46323,
      "training_step_time": 7.81274151802063
    },
    {
      "epoch": 4.417800903320312e-05,
      "model_forward_time": 1.8080477714538574,
      "step": 46324
    },
    {
      "epoch": 4.417800903320312e-05,
      "step": 46324,
      "training_step_time": 7.670392036437988
    },
    {
      "epoch": 4.4178962707519534e-05,
      "model_forward_time": 2.3426625728607178,
      "step": 46325
    },
    {
      "epoch": 4.4178962707519534e-05,
      "step": 46325,
      "training_step_time": 8.318825483322144
    },
    {
      "epoch": 4.417991638183594e-05,
      "model_forward_time": 1.8162097930908203,
      "step": 46326
    },
    {
      "epoch": 4.417991638183594e-05,
      "step": 46326,
      "training_step_time": 7.788695812225342
    },
    {
      "epoch": 4.418087005615234e-05,
      "model_forward_time": 1.8310327529907227,
      "step": 46327
    },
    {
      "epoch": 4.418087005615234e-05,
      "step": 46327,
      "training_step_time": 9.753833532333374
    },
    {
      "epoch": 4.418182373046875e-05,
      "model_forward_time": 1.796903133392334,
      "step": 46328
    },
    {
      "epoch": 4.418182373046875e-05,
      "step": 46328,
      "training_step_time": 8.822860479354858
    },
    {
      "epoch": 4.418277740478516e-05,
      "model_forward_time": 1.8047032356262207,
      "step": 46329
    },
    {
      "epoch": 4.418277740478516e-05,
      "step": 46329,
      "training_step_time": 7.829138517379761
    },
    {
      "epoch": 4.418373107910156e-05,
      "grad_norm": 0.13927726447582245,
      "learning_rate": 6.013288821768988e-06,
      "loss": 0.0591,
      "step": 46330
    },
    {
      "epoch": 4.418373107910156e-05,
      "model_forward_time": 1.8060812950134277,
      "step": 46330
    },
    {
      "dynamics_loss_avg": 0.09641242921352386,
      "epoch": 4.418373107910156e-05,
      "step": 46330
    },
    {
      "action_loss_avg": 0.0023598863393999637,
      "epoch": 4.418373107910156e-05,
      "step": 46330
    },
    {
      "epoch": 4.418373107910156e-05,
      "step": 46330,
      "training_step_time": 7.567642450332642
    },
    {
      "epoch": 4.418468475341797e-05,
      "model_forward_time": 1.8059043884277344,
      "step": 46331
    },
    {
      "epoch": 4.418468475341797e-05,
      "step": 46331,
      "training_step_time": 8.500009536743164
    },
    {
      "epoch": 4.4185638427734376e-05,
      "model_forward_time": 1.8184726238250732,
      "step": 46332
    },
    {
      "epoch": 4.4185638427734376e-05,
      "step": 46332,
      "training_step_time": 7.7104103565216064
    },
    {
      "epoch": 4.418659210205078e-05,
      "model_forward_time": 1.8054304122924805,
      "step": 46333
    },
    {
      "epoch": 4.418659210205078e-05,
      "step": 46333,
      "training_step_time": 7.683317422866821
    },
    {
      "epoch": 4.418754577636719e-05,
      "model_forward_time": 2.341097831726074,
      "step": 46334
    },
    {
      "epoch": 4.418754577636719e-05,
      "step": 46334,
      "training_step_time": 7.778473615646362
    },
    {
      "epoch": 4.4188499450683595e-05,
      "model_forward_time": 1.8038475513458252,
      "step": 46335
    },
    {
      "epoch": 4.4188499450683595e-05,
      "step": 46335,
      "training_step_time": 9.38757848739624
    },
    {
      "epoch": 4.4189453125e-05,
      "model_forward_time": 1.8123114109039307,
      "step": 46336
    },
    {
      "epoch": 4.4189453125e-05,
      "step": 46336,
      "training_step_time": 7.711735486984253
    },
    {
      "epoch": 4.419040679931641e-05,
      "model_forward_time": 1.8054494857788086,
      "step": 46337
    },
    {
      "epoch": 4.419040679931641e-05,
      "step": 46337,
      "training_step_time": 7.765161991119385
    },
    {
      "epoch": 4.4191360473632813e-05,
      "model_forward_time": 1.8040478229522705,
      "step": 46338
    },
    {
      "epoch": 4.4191360473632813e-05,
      "step": 46338,
      "training_step_time": 8.76785659790039
    },
    {
      "epoch": 4.419231414794922e-05,
      "model_forward_time": 1.8098194599151611,
      "step": 46339
    },
    {
      "epoch": 4.419231414794922e-05,
      "step": 46339,
      "training_step_time": 7.52936315536499
    },
    {
      "epoch": 4.419326782226563e-05,
      "grad_norm": 0.12903694808483124,
      "learning_rate": 6.011669606691621e-06,
      "loss": 0.062,
      "step": 46340
    },
    {
      "epoch": 4.419326782226563e-05,
      "model_forward_time": 1.813230037689209,
      "step": 46340
    },
    {
      "dynamics_loss_avg": 0.0676613450050354,
      "epoch": 4.419326782226563e-05,
      "step": 46340
    },
    {
      "action_loss_avg": 0.0012055695638991891,
      "epoch": 4.419326782226563e-05,
      "step": 46340
    },
    {
      "epoch": 4.419326782226563e-05,
      "step": 46340,
      "training_step_time": 8.815398693084717
    },
    {
      "epoch": 4.419422149658203e-05,
      "model_forward_time": 1.8229022026062012,
      "step": 46341
    },
    {
      "epoch": 4.419422149658203e-05,
      "step": 46341,
      "training_step_time": 7.758019924163818
    },
    {
      "epoch": 4.4195175170898436e-05,
      "model_forward_time": 1.7960481643676758,
      "step": 46342
    },
    {
      "epoch": 4.4195175170898436e-05,
      "step": 46342,
      "training_step_time": 7.734774589538574
    },
    {
      "epoch": 4.419612884521485e-05,
      "model_forward_time": 2.355900526046753,
      "step": 46343
    },
    {
      "epoch": 4.419612884521485e-05,
      "step": 46343,
      "training_step_time": 8.419166326522827
    },
    {
      "epoch": 4.419708251953125e-05,
      "model_forward_time": 1.8117382526397705,
      "step": 46344
    },
    {
      "epoch": 4.419708251953125e-05,
      "step": 46344,
      "training_step_time": 8.980931282043457
    },
    {
      "epoch": 4.4198036193847655e-05,
      "model_forward_time": 1.8149254322052002,
      "step": 46345
    },
    {
      "epoch": 4.4198036193847655e-05,
      "step": 46345,
      "training_step_time": 8.343281745910645
    },
    {
      "epoch": 4.419898986816406e-05,
      "model_forward_time": 1.8187084197998047,
      "step": 46346
    },
    {
      "epoch": 4.419898986816406e-05,
      "step": 46346,
      "training_step_time": 8.66120958328247
    },
    {
      "epoch": 4.419994354248047e-05,
      "model_forward_time": 1.805067777633667,
      "step": 46347
    },
    {
      "epoch": 4.419994354248047e-05,
      "step": 46347,
      "training_step_time": 7.88655948638916
    },
    {
      "epoch": 4.4200897216796874e-05,
      "model_forward_time": 1.8104033470153809,
      "step": 46348
    },
    {
      "epoch": 4.4200897216796874e-05,
      "step": 46348,
      "training_step_time": 8.231584787368774
    },
    {
      "epoch": 4.420185089111328e-05,
      "model_forward_time": 1.8094072341918945,
      "step": 46349
    },
    {
      "epoch": 4.420185089111328e-05,
      "step": 46349,
      "training_step_time": 7.851059436798096
    },
    {
      "epoch": 4.420280456542969e-05,
      "grad_norm": 0.15283755958080292,
      "learning_rate": 6.010050280979585e-06,
      "loss": 0.0598,
      "step": 46350
    },
    {
      "epoch": 4.420280456542969e-05,
      "model_forward_time": 1.808589220046997,
      "step": 46350
    },
    {
      "dynamics_loss_avg": 0.06603662222623825,
      "epoch": 4.420280456542969e-05,
      "step": 46350
    },
    {
      "action_loss_avg": 0.0019887678558006884,
      "epoch": 4.420280456542969e-05,
      "step": 46350
    },
    {
      "epoch": 4.420280456542969e-05,
      "step": 46350,
      "training_step_time": 7.6549391746521
    },
    {
      "epoch": 4.420375823974609e-05,
      "model_forward_time": 1.8085463047027588,
      "step": 46351
    },
    {
      "epoch": 4.420375823974609e-05,
      "step": 46351,
      "training_step_time": 8.613555908203125
    },
    {
      "epoch": 4.42047119140625e-05,
      "model_forward_time": 2.354700803756714,
      "step": 46352
    },
    {
      "epoch": 4.42047119140625e-05,
      "step": 46352,
      "training_step_time": 8.305500030517578
    },
    {
      "epoch": 4.420566558837891e-05,
      "model_forward_time": 1.8049607276916504,
      "step": 46353
    },
    {
      "epoch": 4.420566558837891e-05,
      "step": 46353,
      "training_step_time": 8.770217895507812
    },
    {
      "epoch": 4.420661926269531e-05,
      "model_forward_time": 1.8127858638763428,
      "step": 46354
    },
    {
      "epoch": 4.420661926269531e-05,
      "step": 46354,
      "training_step_time": 7.802609205245972
    },
    {
      "epoch": 4.4207572937011716e-05,
      "model_forward_time": 1.8009428977966309,
      "step": 46355
    },
    {
      "epoch": 4.4207572937011716e-05,
      "step": 46355,
      "training_step_time": 8.44573426246643
    },
    {
      "epoch": 4.4208526611328127e-05,
      "model_forward_time": 1.7963616847991943,
      "step": 46356
    },
    {
      "epoch": 4.4208526611328127e-05,
      "step": 46356,
      "training_step_time": 7.913435220718384
    },
    {
      "epoch": 4.420948028564453e-05,
      "model_forward_time": 1.807633876800537,
      "step": 46357
    },
    {
      "epoch": 4.420948028564453e-05,
      "step": 46357,
      "training_step_time": 7.6132590770721436
    },
    {
      "epoch": 4.4210433959960935e-05,
      "model_forward_time": 1.8066940307617188,
      "step": 46358
    },
    {
      "epoch": 4.4210433959960935e-05,
      "step": 46358,
      "training_step_time": 7.815679550170898
    },
    {
      "epoch": 4.4211387634277345e-05,
      "model_forward_time": 1.8117296695709229,
      "step": 46359
    },
    {
      "epoch": 4.4211387634277345e-05,
      "step": 46359,
      "training_step_time": 7.6876959800720215
    },
    {
      "epoch": 4.421234130859375e-05,
      "grad_norm": 0.09663518518209457,
      "learning_rate": 6.008430844809971e-06,
      "loss": 0.0606,
      "step": 46360
    },
    {
      "epoch": 4.421234130859375e-05,
      "model_forward_time": 1.799199104309082,
      "step": 46360
    },
    {
      "dynamics_loss_avg": 0.04528689533472061,
      "epoch": 4.421234130859375e-05,
      "step": 46360
    },
    {
      "action_loss_avg": 0.0011185991927050055,
      "epoch": 4.421234130859375e-05,
      "step": 46360
    },
    {
      "epoch": 4.421234130859375e-05,
      "step": 46360,
      "training_step_time": 7.72240686416626
    },
    {
      "epoch": 4.4213294982910153e-05,
      "model_forward_time": 1.8090033531188965,
      "step": 46361
    },
    {
      "epoch": 4.4213294982910153e-05,
      "step": 46361,
      "training_step_time": 8.296401739120483
    },
    {
      "epoch": 4.4214248657226564e-05,
      "model_forward_time": 1.802269697189331,
      "step": 46362
    },
    {
      "epoch": 4.4214248657226564e-05,
      "step": 46362,
      "training_step_time": 10.494029760360718
    },
    {
      "epoch": 4.421520233154297e-05,
      "model_forward_time": 1.805640459060669,
      "step": 46363
    },
    {
      "epoch": 4.421520233154297e-05,
      "step": 46363,
      "training_step_time": 9.2032949924469
    },
    {
      "epoch": 4.421615600585937e-05,
      "model_forward_time": 1.8055083751678467,
      "step": 46364
    },
    {
      "epoch": 4.421615600585937e-05,
      "step": 46364,
      "training_step_time": 7.769851922988892
    },
    {
      "epoch": 4.421710968017578e-05,
      "model_forward_time": 1.8038783073425293,
      "step": 46365
    },
    {
      "epoch": 4.421710968017578e-05,
      "step": 46365,
      "training_step_time": 8.80415940284729
    },
    {
      "epoch": 4.421806335449219e-05,
      "model_forward_time": 1.8012785911560059,
      "step": 46366
    },
    {
      "epoch": 4.421806335449219e-05,
      "step": 46366,
      "training_step_time": 8.412143230438232
    },
    {
      "epoch": 4.421901702880859e-05,
      "model_forward_time": 1.8092641830444336,
      "step": 46367
    },
    {
      "epoch": 4.421901702880859e-05,
      "step": 46367,
      "training_step_time": 7.825043439865112
    },
    {
      "epoch": 4.4219970703125e-05,
      "model_forward_time": 1.803070068359375,
      "step": 46368
    },
    {
      "epoch": 4.4219970703125e-05,
      "step": 46368,
      "training_step_time": 7.684788942337036
    },
    {
      "epoch": 4.4220924377441406e-05,
      "model_forward_time": 1.8013339042663574,
      "step": 46369
    },
    {
      "epoch": 4.4220924377441406e-05,
      "step": 46369,
      "training_step_time": 7.684391498565674
    },
    {
      "epoch": 4.422187805175781e-05,
      "grad_norm": 0.11666011810302734,
      "learning_rate": 6.006811298359874e-06,
      "loss": 0.0567,
      "step": 46370
    },
    {
      "epoch": 4.422187805175781e-05,
      "model_forward_time": 1.808976411819458,
      "step": 46370
    },
    {
      "dynamics_loss_avg": 0.044580773636698726,
      "epoch": 4.422187805175781e-05,
      "step": 46370
    },
    {
      "action_loss_avg": 0.0011874136107508093,
      "epoch": 4.422187805175781e-05,
      "step": 46370
    },
    {
      "epoch": 4.422187805175781e-05,
      "step": 46370,
      "training_step_time": 8.323916912078857
    },
    {
      "epoch": 4.422283172607422e-05,
      "model_forward_time": 2.337235450744629,
      "step": 46371
    },
    {
      "epoch": 4.422283172607422e-05,
      "step": 46371,
      "training_step_time": 8.476729154586792
    },
    {
      "epoch": 4.4223785400390625e-05,
      "model_forward_time": 1.802175521850586,
      "step": 46372
    },
    {
      "epoch": 4.4223785400390625e-05,
      "step": 46372,
      "training_step_time": 7.832040786743164
    },
    {
      "epoch": 4.422473907470703e-05,
      "model_forward_time": 1.8091661930084229,
      "step": 46373
    },
    {
      "epoch": 4.422473907470703e-05,
      "step": 46373,
      "training_step_time": 7.83879280090332
    },
    {
      "epoch": 4.422569274902344e-05,
      "model_forward_time": 1.7105603218078613,
      "step": 46374
    },
    {
      "epoch": 4.422569274902344e-05,
      "step": 46374,
      "training_step_time": 7.689068794250488
    },
    {
      "epoch": 4.4226646423339844e-05,
      "model_forward_time": 1.7309484481811523,
      "step": 46375
    },
    {
      "epoch": 4.4226646423339844e-05,
      "step": 46375,
      "training_step_time": 7.832726240158081
    },
    {
      "epoch": 4.422760009765625e-05,
      "model_forward_time": 1.3464741706848145,
      "step": 46376
    },
    {
      "epoch": 4.422760009765625e-05,
      "step": 46376,
      "training_step_time": 7.71782922744751
    },
    {
      "epoch": 4.422855377197266e-05,
      "model_forward_time": 1.7160625457763672,
      "step": 46377
    },
    {
      "epoch": 4.422855377197266e-05,
      "step": 46377,
      "training_step_time": 8.623358726501465
    },
    {
      "epoch": 4.422950744628906e-05,
      "model_forward_time": 1.7155146598815918,
      "step": 46378
    },
    {
      "epoch": 4.422950744628906e-05,
      "step": 46378,
      "training_step_time": 7.699589014053345
    },
    {
      "epoch": 4.423046112060547e-05,
      "model_forward_time": 1.34413480758667,
      "step": 46379
    },
    {
      "epoch": 4.423046112060547e-05,
      "step": 46379,
      "training_step_time": 7.718583106994629
    },
    {
      "epoch": 4.423141479492188e-05,
      "grad_norm": 0.15098464488983154,
      "learning_rate": 6.005191641806407e-06,
      "loss": 0.0644,
      "step": 46380
    },
    {
      "epoch": 4.423141479492188e-05,
      "model_forward_time": 2.2531447410583496,
      "step": 46380
    },
    {
      "dynamics_loss_avg": 0.06414267392829061,
      "epoch": 4.423141479492188e-05,
      "step": 46380
    },
    {
      "action_loss_avg": 0.0011154370673466473,
      "epoch": 4.423141479492188e-05,
      "step": 46380
    },
    {
      "epoch": 4.423141479492188e-05,
      "step": 46380,
      "training_step_time": 7.483432292938232
    },
    {
      "epoch": 4.423236846923828e-05,
      "model_forward_time": 1.7259550094604492,
      "step": 46381
    },
    {
      "epoch": 4.423236846923828e-05,
      "step": 46381,
      "training_step_time": 7.824585914611816
    },
    {
      "epoch": 4.4233322143554685e-05,
      "model_forward_time": 1.7276959419250488,
      "step": 46382
    },
    {
      "epoch": 4.4233322143554685e-05,
      "step": 46382,
      "training_step_time": 9.306202411651611
    },
    {
      "epoch": 4.4234275817871096e-05,
      "model_forward_time": 1.7181460857391357,
      "step": 46383
    },
    {
      "epoch": 4.4234275817871096e-05,
      "step": 46383,
      "training_step_time": 9.605988502502441
    },
    {
      "epoch": 4.42352294921875e-05,
      "model_forward_time": 1.7129523754119873,
      "step": 46384
    },
    {
      "epoch": 4.42352294921875e-05,
      "step": 46384,
      "training_step_time": 6.814948081970215
    },
    {
      "epoch": 4.4236183166503904e-05,
      "model_forward_time": 1.7156405448913574,
      "step": 46385
    },
    {
      "epoch": 4.4236183166503904e-05,
      "step": 46385,
      "training_step_time": 6.867543935775757
    },
    {
      "epoch": 4.4237136840820315e-05,
      "model_forward_time": 1.6973812580108643,
      "step": 46386
    },
    {
      "epoch": 4.4237136840820315e-05,
      "step": 46386,
      "training_step_time": 6.8300135135650635
    },
    {
      "epoch": 4.423809051513672e-05,
      "model_forward_time": 1.7160651683807373,
      "step": 46387
    },
    {
      "epoch": 4.423809051513672e-05,
      "step": 46387,
      "training_step_time": 6.801012992858887
    },
    {
      "epoch": 4.423904418945312e-05,
      "model_forward_time": 1.7069661617279053,
      "step": 46388
    },
    {
      "epoch": 4.423904418945312e-05,
      "step": 46388,
      "training_step_time": 6.8106889724731445
    },
    {
      "epoch": 4.4239997863769534e-05,
      "model_forward_time": 1.715085744857788,
      "step": 46389
    },
    {
      "epoch": 4.4239997863769534e-05,
      "step": 46389,
      "training_step_time": 6.771491050720215
    },
    {
      "epoch": 4.424095153808594e-05,
      "grad_norm": 0.17528420686721802,
      "learning_rate": 6.003571875326695e-06,
      "loss": 0.0652,
      "step": 46390
    },
    {
      "epoch": 4.424095153808594e-05,
      "model_forward_time": 1.7094988822937012,
      "step": 46390
    },
    {
      "dynamics_loss_avg": 0.07226182986050844,
      "epoch": 4.424095153808594e-05,
      "step": 46390
    },
    {
      "action_loss_avg": 0.001777297956869006,
      "epoch": 4.424095153808594e-05,
      "step": 46390
    },
    {
      "epoch": 4.424095153808594e-05,
      "step": 46390,
      "training_step_time": 6.831263065338135
    },
    {
      "epoch": 4.424190521240234e-05,
      "model_forward_time": 1.7146871089935303,
      "step": 46391
    },
    {
      "epoch": 4.424190521240234e-05,
      "step": 46391,
      "training_step_time": 6.867789030075073
    },
    {
      "epoch": 4.424285888671875e-05,
      "model_forward_time": 2.2915945053100586,
      "step": 46392
    },
    {
      "epoch": 4.424285888671875e-05,
      "step": 46392,
      "training_step_time": 7.750956296920776
    },
    {
      "epoch": 4.424381256103516e-05,
      "model_forward_time": 1.718480110168457,
      "step": 46393
    },
    {
      "epoch": 4.424381256103516e-05,
      "step": 46393,
      "training_step_time": 8.028542280197144
    },
    {
      "epoch": 4.424476623535156e-05,
      "model_forward_time": 1.7161357402801514,
      "step": 46394
    },
    {
      "epoch": 4.424476623535156e-05,
      "step": 46394,
      "training_step_time": 6.847833156585693
    },
    {
      "epoch": 4.424571990966797e-05,
      "model_forward_time": 1.7146852016448975,
      "step": 46395
    },
    {
      "epoch": 4.424571990966797e-05,
      "step": 46395,
      "training_step_time": 6.832669258117676
    },
    {
      "epoch": 4.4246673583984376e-05,
      "model_forward_time": 1.7165780067443848,
      "step": 46396
    },
    {
      "epoch": 4.4246673583984376e-05,
      "step": 46396,
      "training_step_time": 6.8217644691467285
    },
    {
      "epoch": 4.424762725830078e-05,
      "model_forward_time": 1.7177314758300781,
      "step": 46397
    },
    {
      "epoch": 4.424762725830078e-05,
      "step": 46397,
      "training_step_time": 6.798993825912476
    },
    {
      "epoch": 4.424858093261719e-05,
      "model_forward_time": 1.7122085094451904,
      "step": 46398
    },
    {
      "epoch": 4.424858093261719e-05,
      "step": 46398,
      "training_step_time": 6.846116065979004
    },
    {
      "epoch": 4.4249534606933595e-05,
      "model_forward_time": 1.7139086723327637,
      "step": 46399
    },
    {
      "epoch": 4.4249534606933595e-05,
      "step": 46399,
      "training_step_time": 6.8343048095703125
    },
    {
      "epoch": 4.425048828125e-05,
      "grad_norm": 0.14127713441848755,
      "learning_rate": 6.00195199909787e-06,
      "loss": 0.0633,
      "step": 46400
    },
    {
      "epoch": 4.425048828125e-05,
      "model_forward_time": 1.731302261352539,
      "step": 46400
    },
    {
      "dynamics_loss_avg": 0.0668504148721695,
      "epoch": 4.425048828125e-05,
      "step": 46400
    },
    {
      "action_loss_avg": 0.0011068581836298109,
      "epoch": 4.425048828125e-05,
      "step": 46400
    },
    {
      "epoch": 4.425048828125e-05,
      "step": 46400,
      "training_step_time": 7.322568416595459
    },
    {
      "epoch": 4.425144195556641e-05,
      "model_forward_time": 2.323621988296509,
      "step": 46401
    },
    {
      "epoch": 4.425144195556641e-05,
      "step": 46401,
      "training_step_time": 7.694658517837524
    },
    {
      "epoch": 4.4252395629882813e-05,
      "model_forward_time": 1.7144298553466797,
      "step": 46402
    },
    {
      "epoch": 4.4252395629882813e-05,
      "step": 46402,
      "training_step_time": 7.990276336669922
    },
    {
      "epoch": 4.425334930419922e-05,
      "model_forward_time": 1.7090001106262207,
      "step": 46403
    },
    {
      "epoch": 4.425334930419922e-05,
      "step": 46403,
      "training_step_time": 7.205231666564941
    },
    {
      "epoch": 4.425430297851563e-05,
      "model_forward_time": 1.72330904006958,
      "step": 46404
    },
    {
      "epoch": 4.425430297851563e-05,
      "step": 46404,
      "training_step_time": 6.775683403015137
    },
    {
      "epoch": 4.425525665283203e-05,
      "model_forward_time": 1.7142918109893799,
      "step": 46405
    },
    {
      "epoch": 4.425525665283203e-05,
      "step": 46405,
      "training_step_time": 6.830630779266357
    },
    {
      "epoch": 4.4256210327148436e-05,
      "model_forward_time": 1.720188856124878,
      "step": 46406
    },
    {
      "epoch": 4.4256210327148436e-05,
      "step": 46406,
      "training_step_time": 6.793895483016968
    },
    {
      "epoch": 4.425716400146484e-05,
      "model_forward_time": 1.717761754989624,
      "step": 46407
    },
    {
      "epoch": 4.425716400146484e-05,
      "step": 46407,
      "training_step_time": 6.776592493057251
    },
    {
      "epoch": 4.425811767578125e-05,
      "model_forward_time": 1.7119698524475098,
      "step": 46408
    },
    {
      "epoch": 4.425811767578125e-05,
      "step": 46408,
      "training_step_time": 6.748806476593018
    },
    {
      "epoch": 4.4259071350097655e-05,
      "model_forward_time": 1.7131528854370117,
      "step": 46409
    },
    {
      "epoch": 4.4259071350097655e-05,
      "step": 46409,
      "training_step_time": 6.724543809890747
    },
    {
      "epoch": 4.426002502441406e-05,
      "grad_norm": 0.14441996812820435,
      "learning_rate": 6.000332013297082e-06,
      "loss": 0.0611,
      "step": 46410
    },
    {
      "epoch": 4.426002502441406e-05,
      "model_forward_time": 2.315197229385376,
      "step": 46410
    },
    {
      "dynamics_loss_avg": 0.04766732333227992,
      "epoch": 4.426002502441406e-05,
      "step": 46410
    },
    {
      "action_loss_avg": 0.0011580704187508673,
      "epoch": 4.426002502441406e-05,
      "step": 46410
    },
    {
      "epoch": 4.426002502441406e-05,
      "step": 46410,
      "training_step_time": 7.566253662109375
    },
    {
      "epoch": 4.426097869873047e-05,
      "model_forward_time": 1.3457043170928955,
      "step": 46411
    },
    {
      "epoch": 4.426097869873047e-05,
      "step": 46411,
      "training_step_time": 8.890355348587036
    },
    {
      "epoch": 4.4261932373046874e-05,
      "model_forward_time": 1.713724136352539,
      "step": 46412
    },
    {
      "epoch": 4.4261932373046874e-05,
      "step": 46412,
      "training_step_time": 7.592159748077393
    },
    {
      "epoch": 4.426288604736328e-05,
      "model_forward_time": 1.717839002609253,
      "step": 46413
    },
    {
      "epoch": 4.426288604736328e-05,
      "step": 46413,
      "training_step_time": 6.751075983047485
    },
    {
      "epoch": 4.426383972167969e-05,
      "model_forward_time": 1.717350959777832,
      "step": 46414
    },
    {
      "epoch": 4.426383972167969e-05,
      "step": 46414,
      "training_step_time": 6.729143381118774
    },
    {
      "epoch": 4.426479339599609e-05,
      "model_forward_time": 1.714062213897705,
      "step": 46415
    },
    {
      "epoch": 4.426479339599609e-05,
      "step": 46415,
      "training_step_time": 6.780534267425537
    },
    {
      "epoch": 4.42657470703125e-05,
      "model_forward_time": 1.7166416645050049,
      "step": 46416
    },
    {
      "epoch": 4.42657470703125e-05,
      "step": 46416,
      "training_step_time": 6.7932655811309814
    },
    {
      "epoch": 4.426670074462891e-05,
      "model_forward_time": 1.7117514610290527,
      "step": 46417
    },
    {
      "epoch": 4.426670074462891e-05,
      "step": 46417,
      "training_step_time": 6.77251935005188
    },
    {
      "epoch": 4.426765441894531e-05,
      "model_forward_time": 1.7177352905273438,
      "step": 46418
    },
    {
      "epoch": 4.426765441894531e-05,
      "step": 46418,
      "training_step_time": 6.811709403991699
    },
    {
      "epoch": 4.4268608093261716e-05,
      "model_forward_time": 2.3073890209198,
      "step": 46419
    },
    {
      "epoch": 4.4268608093261716e-05,
      "step": 46419,
      "training_step_time": 7.623020887374878
    },
    {
      "epoch": 4.426956176757813e-05,
      "grad_norm": 0.17709317803382874,
      "learning_rate": 5.998711918101487e-06,
      "loss": 0.0585,
      "step": 46420
    },
    {
      "epoch": 4.426956176757813e-05,
      "model_forward_time": 1.7082672119140625,
      "step": 46420
    },
    {
      "dynamics_loss_avg": 0.05536725828424096,
      "epoch": 4.426956176757813e-05,
      "step": 46420
    },
    {
      "action_loss_avg": 0.002076480301911943,
      "epoch": 4.426956176757813e-05,
      "step": 46420
    },
    {
      "epoch": 4.426956176757813e-05,
      "step": 46420,
      "training_step_time": 9.306573629379272
    },
    {
      "epoch": 4.427051544189453e-05,
      "model_forward_time": 1.720017433166504,
      "step": 46421
    },
    {
      "epoch": 4.427051544189453e-05,
      "step": 46421,
      "training_step_time": 7.729348421096802
    },
    {
      "epoch": 4.4271469116210935e-05,
      "model_forward_time": 1.7134549617767334,
      "step": 46422
    },
    {
      "epoch": 4.4271469116210935e-05,
      "step": 46422,
      "training_step_time": 6.789598703384399
    },
    {
      "epoch": 4.4272422790527345e-05,
      "model_forward_time": 1.713622808456421,
      "step": 46423
    },
    {
      "epoch": 4.4272422790527345e-05,
      "step": 46423,
      "training_step_time": 6.745706081390381
    },
    {
      "epoch": 4.427337646484375e-05,
      "model_forward_time": 1.7151238918304443,
      "step": 46424
    },
    {
      "epoch": 4.427337646484375e-05,
      "step": 46424,
      "training_step_time": 6.837318658828735
    },
    {
      "epoch": 4.4274330139160154e-05,
      "model_forward_time": 1.7129228115081787,
      "step": 46425
    },
    {
      "epoch": 4.4274330139160154e-05,
      "step": 46425,
      "training_step_time": 6.831489324569702
    },
    {
      "epoch": 4.4275283813476564e-05,
      "model_forward_time": 1.7162420749664307,
      "step": 46426
    },
    {
      "epoch": 4.4275283813476564e-05,
      "step": 46426,
      "training_step_time": 6.812204837799072
    },
    {
      "epoch": 4.427623748779297e-05,
      "model_forward_time": 1.7128016948699951,
      "step": 46427
    },
    {
      "epoch": 4.427623748779297e-05,
      "step": 46427,
      "training_step_time": 6.774379014968872
    },
    {
      "epoch": 4.427719116210937e-05,
      "model_forward_time": 2.2927370071411133,
      "step": 46428
    },
    {
      "epoch": 4.427719116210937e-05,
      "step": 46428,
      "training_step_time": 7.602676153182983
    },
    {
      "epoch": 4.427814483642578e-05,
      "model_forward_time": 1.7131457328796387,
      "step": 46429
    },
    {
      "epoch": 4.427814483642578e-05,
      "step": 46429,
      "training_step_time": 10.744157552719116
    },
    {
      "epoch": 4.427909851074219e-05,
      "grad_norm": 0.20114384591579437,
      "learning_rate": 5.997091713688261e-06,
      "loss": 0.0616,
      "step": 46430
    },
    {
      "epoch": 4.427909851074219e-05,
      "model_forward_time": 1.7098376750946045,
      "step": 46430
    },
    {
      "dynamics_loss_avg": 0.055667629837989806,
      "epoch": 4.427909851074219e-05,
      "step": 46430
    },
    {
      "action_loss_avg": 0.0013065779174212366,
      "epoch": 4.427909851074219e-05,
      "step": 46430
    },
    {
      "epoch": 4.427909851074219e-05,
      "step": 46430,
      "training_step_time": 8.344347715377808
    },
    {
      "epoch": 4.428005218505859e-05,
      "model_forward_time": 1.7194664478302002,
      "step": 46431
    },
    {
      "epoch": 4.428005218505859e-05,
      "step": 46431,
      "training_step_time": 6.743407249450684
    },
    {
      "epoch": 4.4281005859375e-05,
      "model_forward_time": 1.7124316692352295,
      "step": 46432
    },
    {
      "epoch": 4.4281005859375e-05,
      "step": 46432,
      "training_step_time": 6.734449148178101
    },
    {
      "epoch": 4.4281959533691406e-05,
      "model_forward_time": 1.7167298793792725,
      "step": 46433
    },
    {
      "epoch": 4.4281959533691406e-05,
      "step": 46433,
      "training_step_time": 6.761904954910278
    },
    {
      "epoch": 4.428291320800781e-05,
      "model_forward_time": 1.3431735038757324,
      "step": 46434
    },
    {
      "epoch": 4.428291320800781e-05,
      "step": 46434,
      "training_step_time": 6.790269374847412
    },
    {
      "epoch": 4.428386688232422e-05,
      "model_forward_time": 1.711573600769043,
      "step": 46435
    },
    {
      "epoch": 4.428386688232422e-05,
      "step": 46435,
      "training_step_time": 6.78004002571106
    },
    {
      "epoch": 4.4284820556640625e-05,
      "model_forward_time": 1.7092912197113037,
      "step": 46436
    },
    {
      "epoch": 4.4284820556640625e-05,
      "step": 46436,
      "training_step_time": 6.798442602157593
    },
    {
      "epoch": 4.428577423095703e-05,
      "model_forward_time": 1.7179920673370361,
      "step": 46437
    },
    {
      "epoch": 4.428577423095703e-05,
      "step": 46437,
      "training_step_time": 7.708003282546997
    },
    {
      "epoch": 4.428672790527344e-05,
      "model_forward_time": 2.278125524520874,
      "step": 46438
    },
    {
      "epoch": 4.428672790527344e-05,
      "step": 46438,
      "training_step_time": 9.413147211074829
    },
    {
      "epoch": 4.4287681579589844e-05,
      "model_forward_time": 1.8057467937469482,
      "step": 46439
    },
    {
      "epoch": 4.4287681579589844e-05,
      "step": 46439,
      "training_step_time": 7.907709360122681
    },
    {
      "epoch": 4.428863525390625e-05,
      "grad_norm": 0.14136458933353424,
      "learning_rate": 5.995471400234584e-06,
      "loss": 0.0578,
      "step": 46440
    },
    {
      "epoch": 4.428863525390625e-05,
      "model_forward_time": 1.8087663650512695,
      "step": 46440
    },
    {
      "dynamics_loss_avg": 0.055204180628061296,
      "epoch": 4.428863525390625e-05,
      "step": 46440
    },
    {
      "action_loss_avg": 0.0013819111802149564,
      "epoch": 4.428863525390625e-05,
      "step": 46440
    },
    {
      "epoch": 4.428863525390625e-05,
      "step": 46440,
      "training_step_time": 7.637613296508789
    },
    {
      "epoch": 4.428958892822266e-05,
      "model_forward_time": 1.8044841289520264,
      "step": 46441
    },
    {
      "epoch": 4.428958892822266e-05,
      "step": 46441,
      "training_step_time": 6.7634851932525635
    },
    {
      "epoch": 4.429054260253906e-05,
      "model_forward_time": 1.8069021701812744,
      "step": 46442
    },
    {
      "epoch": 4.429054260253906e-05,
      "step": 46442,
      "training_step_time": 6.774867296218872
    },
    {
      "epoch": 4.429149627685547e-05,
      "model_forward_time": 1.8062050342559814,
      "step": 46443
    },
    {
      "epoch": 4.429149627685547e-05,
      "step": 46443,
      "training_step_time": 6.784550428390503
    },
    {
      "epoch": 4.429244995117188e-05,
      "model_forward_time": 1.8073604106903076,
      "step": 46444
    },
    {
      "epoch": 4.429244995117188e-05,
      "step": 46444,
      "training_step_time": 6.762482643127441
    },
    {
      "epoch": 4.429340362548828e-05,
      "model_forward_time": 1.806626558303833,
      "step": 46445
    },
    {
      "epoch": 4.429340362548828e-05,
      "step": 46445,
      "training_step_time": 6.802282810211182
    },
    {
      "epoch": 4.4294357299804686e-05,
      "model_forward_time": 1.8107926845550537,
      "step": 46446
    },
    {
      "epoch": 4.4294357299804686e-05,
      "step": 46446,
      "training_step_time": 7.620004653930664
    },
    {
      "epoch": 4.4295310974121096e-05,
      "model_forward_time": 2.254950523376465,
      "step": 46447
    },
    {
      "epoch": 4.4295310974121096e-05,
      "step": 46447,
      "training_step_time": 7.552242279052734
    },
    {
      "epoch": 4.42962646484375e-05,
      "model_forward_time": 1.7182106971740723,
      "step": 46448
    },
    {
      "epoch": 4.42962646484375e-05,
      "step": 46448,
      "training_step_time": 8.738880157470703
    },
    {
      "epoch": 4.4297218322753904e-05,
      "model_forward_time": 1.7181205749511719,
      "step": 46449
    },
    {
      "epoch": 4.4297218322753904e-05,
      "step": 46449,
      "training_step_time": 7.706568956375122
    },
    {
      "epoch": 4.4298171997070315e-05,
      "grad_norm": 0.1573408991098404,
      "learning_rate": 5.993850977917649e-06,
      "loss": 0.0571,
      "step": 46450
    },
    {
      "epoch": 4.4298171997070315e-05,
      "model_forward_time": 1.7070531845092773,
      "step": 46450
    },
    {
      "dynamics_loss_avg": 0.06639855727553368,
      "epoch": 4.4298171997070315e-05,
      "step": 46450
    },
    {
      "action_loss_avg": 0.001411635079421103,
      "epoch": 4.4298171997070315e-05,
      "step": 46450
    },
    {
      "epoch": 4.4298171997070315e-05,
      "step": 46450,
      "training_step_time": 6.737748146057129
    },
    {
      "epoch": 4.429912567138672e-05,
      "model_forward_time": 1.720280647277832,
      "step": 46451
    },
    {
      "epoch": 4.429912567138672e-05,
      "step": 46451,
      "training_step_time": 6.732699155807495
    },
    {
      "epoch": 4.430007934570312e-05,
      "model_forward_time": 1.7185120582580566,
      "step": 46452
    },
    {
      "epoch": 4.430007934570312e-05,
      "step": 46452,
      "training_step_time": 6.7237019538879395
    },
    {
      "epoch": 4.4301033020019534e-05,
      "model_forward_time": 1.7683210372924805,
      "step": 46453
    },
    {
      "epoch": 4.4301033020019534e-05,
      "step": 46453,
      "training_step_time": 6.757868528366089
    },
    {
      "epoch": 4.430198669433594e-05,
      "model_forward_time": 1.7111315727233887,
      "step": 46454
    },
    {
      "epoch": 4.430198669433594e-05,
      "step": 46454,
      "training_step_time": 6.754611015319824
    },
    {
      "epoch": 4.430294036865234e-05,
      "model_forward_time": 1.7139194011688232,
      "step": 46455
    },
    {
      "epoch": 4.430294036865234e-05,
      "step": 46455,
      "training_step_time": 7.378925800323486
    },
    {
      "epoch": 4.430389404296875e-05,
      "model_forward_time": 2.32511305809021,
      "step": 46456
    },
    {
      "epoch": 4.430389404296875e-05,
      "step": 46456,
      "training_step_time": 7.525261640548706
    },
    {
      "epoch": 4.430484771728516e-05,
      "model_forward_time": 1.7080466747283936,
      "step": 46457
    },
    {
      "epoch": 4.430484771728516e-05,
      "step": 46457,
      "training_step_time": 9.692708253860474
    },
    {
      "epoch": 4.430580139160156e-05,
      "model_forward_time": 1.7176604270935059,
      "step": 46458
    },
    {
      "epoch": 4.430580139160156e-05,
      "step": 46458,
      "training_step_time": 8.90567946434021
    },
    {
      "epoch": 4.430675506591797e-05,
      "model_forward_time": 1.7150859832763672,
      "step": 46459
    },
    {
      "epoch": 4.430675506591797e-05,
      "step": 46459,
      "training_step_time": 6.857608318328857
    },
    {
      "epoch": 4.4307708740234376e-05,
      "grad_norm": 0.16746029257774353,
      "learning_rate": 5.992230446914667e-06,
      "loss": 0.0581,
      "step": 46460
    },
    {
      "epoch": 4.4307708740234376e-05,
      "model_forward_time": 1.710512399673462,
      "step": 46460
    },
    {
      "dynamics_loss_avg": 0.07698039673268794,
      "epoch": 4.4307708740234376e-05,
      "step": 46460
    },
    {
      "action_loss_avg": 0.0018114432925358415,
      "epoch": 4.4307708740234376e-05,
      "step": 46460
    },
    {
      "epoch": 4.4307708740234376e-05,
      "step": 46460,
      "training_step_time": 6.8452467918396
    },
    {
      "epoch": 4.430866241455078e-05,
      "model_forward_time": 1.7151002883911133,
      "step": 46461
    },
    {
      "epoch": 4.430866241455078e-05,
      "step": 46461,
      "training_step_time": 6.887504816055298
    },
    {
      "epoch": 4.430961608886719e-05,
      "model_forward_time": 1.7184109687805176,
      "step": 46462
    },
    {
      "epoch": 4.430961608886719e-05,
      "step": 46462,
      "training_step_time": 6.890684604644775
    },
    {
      "epoch": 4.4310569763183595e-05,
      "model_forward_time": 1.7183029651641846,
      "step": 46463
    },
    {
      "epoch": 4.4310569763183595e-05,
      "step": 46463,
      "training_step_time": 6.860760450363159
    },
    {
      "epoch": 4.43115234375e-05,
      "model_forward_time": 1.717944860458374,
      "step": 46464
    },
    {
      "epoch": 4.43115234375e-05,
      "step": 46464,
      "training_step_time": 6.881156921386719
    },
    {
      "epoch": 4.431247711181641e-05,
      "model_forward_time": 2.2720987796783447,
      "step": 46465
    },
    {
      "epoch": 4.431247711181641e-05,
      "step": 46465,
      "training_step_time": 7.4410505294799805
    },
    {
      "epoch": 4.4313430786132814e-05,
      "model_forward_time": 1.755880355834961,
      "step": 46466
    },
    {
      "epoch": 4.4313430786132814e-05,
      "step": 46466,
      "training_step_time": 10.446492195129395
    },
    {
      "epoch": 4.431438446044922e-05,
      "model_forward_time": 1.7149207592010498,
      "step": 46467
    },
    {
      "epoch": 4.431438446044922e-05,
      "step": 46467,
      "training_step_time": 7.767947435379028
    },
    {
      "epoch": 4.431533813476563e-05,
      "model_forward_time": 1.7223031520843506,
      "step": 46468
    },
    {
      "epoch": 4.431533813476563e-05,
      "step": 46468,
      "training_step_time": 7.156885862350464
    },
    {
      "epoch": 4.431629180908203e-05,
      "model_forward_time": 1.7132880687713623,
      "step": 46469
    },
    {
      "epoch": 4.431629180908203e-05,
      "step": 46469,
      "training_step_time": 6.832918643951416
    },
    {
      "epoch": 4.4317245483398436e-05,
      "grad_norm": 0.1415438950061798,
      "learning_rate": 5.990609807402855e-06,
      "loss": 0.0622,
      "step": 46470
    },
    {
      "epoch": 4.4317245483398436e-05,
      "model_forward_time": 1.7153558731079102,
      "step": 46470
    },
    {
      "dynamics_loss_avg": 0.08345021679997444,
      "epoch": 4.4317245483398436e-05,
      "step": 46470
    },
    {
      "action_loss_avg": 0.002307559159817174,
      "epoch": 4.4317245483398436e-05,
      "step": 46470
    },
    {
      "epoch": 4.4317245483398436e-05,
      "step": 46470,
      "training_step_time": 6.804113864898682
    },
    {
      "epoch": 4.431819915771484e-05,
      "model_forward_time": 1.7251574993133545,
      "step": 46471
    },
    {
      "epoch": 4.431819915771484e-05,
      "step": 46471,
      "training_step_time": 6.802481651306152
    },
    {
      "epoch": 4.431915283203125e-05,
      "model_forward_time": 1.7225522994995117,
      "step": 46472
    },
    {
      "epoch": 4.431915283203125e-05,
      "step": 46472,
      "training_step_time": 6.82540225982666
    },
    {
      "epoch": 4.4320106506347655e-05,
      "model_forward_time": 1.7178804874420166,
      "step": 46473
    },
    {
      "epoch": 4.4320106506347655e-05,
      "step": 46473,
      "training_step_time": 6.768400430679321
    },
    {
      "epoch": 4.432106018066406e-05,
      "model_forward_time": 1.720808506011963,
      "step": 46474
    },
    {
      "epoch": 4.432106018066406e-05,
      "step": 46474,
      "training_step_time": 7.943058490753174
    },
    {
      "epoch": 4.432201385498047e-05,
      "model_forward_time": 1.755643367767334,
      "step": 46475
    },
    {
      "epoch": 4.432201385498047e-05,
      "step": 46475,
      "training_step_time": 8.698221206665039
    },
    {
      "epoch": 4.4322967529296874e-05,
      "model_forward_time": 1.734877347946167,
      "step": 46476
    },
    {
      "epoch": 4.4322967529296874e-05,
      "step": 46476,
      "training_step_time": 7.7723236083984375
    },
    {
      "epoch": 4.432392120361328e-05,
      "model_forward_time": 1.7304139137268066,
      "step": 46477
    },
    {
      "epoch": 4.432392120361328e-05,
      "step": 46477,
      "training_step_time": 7.579850673675537
    },
    {
      "epoch": 4.432487487792969e-05,
      "model_forward_time": 1.726210594177246,
      "step": 46478
    },
    {
      "epoch": 4.432487487792969e-05,
      "step": 46478,
      "training_step_time": 6.777984142303467
    },
    {
      "epoch": 4.432582855224609e-05,
      "model_forward_time": 1.728036642074585,
      "step": 46479
    },
    {
      "epoch": 4.432582855224609e-05,
      "step": 46479,
      "training_step_time": 6.729014873504639
    },
    {
      "epoch": 4.43267822265625e-05,
      "grad_norm": 0.1510469764471054,
      "learning_rate": 5.988989059559443e-06,
      "loss": 0.0553,
      "step": 46480
    },
    {
      "epoch": 4.43267822265625e-05,
      "model_forward_time": 1.7242834568023682,
      "step": 46480
    },
    {
      "dynamics_loss_avg": 0.05946412570774555,
      "epoch": 4.43267822265625e-05,
      "step": 46480
    },
    {
      "action_loss_avg": 0.0013093763089273124,
      "epoch": 4.43267822265625e-05,
      "step": 46480
    },
    {
      "epoch": 4.43267822265625e-05,
      "step": 46480,
      "training_step_time": 6.809779644012451
    },
    {
      "epoch": 4.432773590087891e-05,
      "model_forward_time": 1.7351758480072021,
      "step": 46481
    },
    {
      "epoch": 4.432773590087891e-05,
      "step": 46481,
      "training_step_time": 6.783024311065674
    },
    {
      "epoch": 4.432868957519531e-05,
      "model_forward_time": 1.7298269271850586,
      "step": 46482
    },
    {
      "epoch": 4.432868957519531e-05,
      "step": 46482,
      "training_step_time": 6.746140956878662
    },
    {
      "epoch": 4.4329643249511716e-05,
      "model_forward_time": 1.7312648296356201,
      "step": 46483
    },
    {
      "epoch": 4.4329643249511716e-05,
      "step": 46483,
      "training_step_time": 7.412372350692749
    },
    {
      "epoch": 4.433059692382813e-05,
      "model_forward_time": 2.3105671405792236,
      "step": 46484
    },
    {
      "epoch": 4.433059692382813e-05,
      "step": 46484,
      "training_step_time": 8.608751058578491
    },
    {
      "epoch": 4.433155059814453e-05,
      "model_forward_time": 1.7336704730987549,
      "step": 46485
    },
    {
      "epoch": 4.433155059814453e-05,
      "step": 46485,
      "training_step_time": 8.692171096801758
    },
    {
      "epoch": 4.4332504272460935e-05,
      "model_forward_time": 1.7280747890472412,
      "step": 46486
    },
    {
      "epoch": 4.4332504272460935e-05,
      "step": 46486,
      "training_step_time": 7.604910850524902
    },
    {
      "epoch": 4.4333457946777346e-05,
      "model_forward_time": 1.7305641174316406,
      "step": 46487
    },
    {
      "epoch": 4.4333457946777346e-05,
      "step": 46487,
      "training_step_time": 6.7848944664001465
    },
    {
      "epoch": 4.433441162109375e-05,
      "model_forward_time": 1.730989933013916,
      "step": 46488
    },
    {
      "epoch": 4.433441162109375e-05,
      "step": 46488,
      "training_step_time": 6.743517637252808
    },
    {
      "epoch": 4.4335365295410154e-05,
      "model_forward_time": 1.745685338973999,
      "step": 46489
    },
    {
      "epoch": 4.4335365295410154e-05,
      "step": 46489,
      "training_step_time": 6.820122957229614
    },
    {
      "epoch": 4.4336318969726564e-05,
      "grad_norm": 0.12161616235971451,
      "learning_rate": 5.987368203561676e-06,
      "loss": 0.058,
      "step": 46490
    },
    {
      "epoch": 4.4336318969726564e-05,
      "model_forward_time": 1.735208511352539,
      "step": 46490
    },
    {
      "dynamics_loss_avg": 0.05373336263000965,
      "epoch": 4.4336318969726564e-05,
      "step": 46490
    },
    {
      "action_loss_avg": 0.0012193294649478048,
      "epoch": 4.4336318969726564e-05,
      "step": 46490
    },
    {
      "epoch": 4.4336318969726564e-05,
      "step": 46490,
      "training_step_time": 6.735668897628784
    },
    {
      "epoch": 4.433727264404297e-05,
      "model_forward_time": 1.7313485145568848,
      "step": 46491
    },
    {
      "epoch": 4.433727264404297e-05,
      "step": 46491,
      "training_step_time": 6.784654378890991
    },
    {
      "epoch": 4.433822631835937e-05,
      "model_forward_time": 1.737795352935791,
      "step": 46492
    },
    {
      "epoch": 4.433822631835937e-05,
      "step": 46492,
      "training_step_time": 7.449352502822876
    },
    {
      "epoch": 4.433917999267578e-05,
      "model_forward_time": 2.333298921585083,
      "step": 46493
    },
    {
      "epoch": 4.433917999267578e-05,
      "step": 46493,
      "training_step_time": 9.500412225723267
    },
    {
      "epoch": 4.434013366699219e-05,
      "model_forward_time": 1.740027666091919,
      "step": 46494
    },
    {
      "epoch": 4.434013366699219e-05,
      "step": 46494,
      "training_step_time": 10.21059775352478
    },
    {
      "epoch": 4.434108734130859e-05,
      "model_forward_time": 1.7398285865783691,
      "step": 46495
    },
    {
      "epoch": 4.434108734130859e-05,
      "step": 46495,
      "training_step_time": 7.703914165496826
    },
    {
      "epoch": 4.4342041015625e-05,
      "model_forward_time": 1.7247693538665771,
      "step": 46496
    },
    {
      "epoch": 4.4342041015625e-05,
      "step": 46496,
      "training_step_time": 7.0906081199646
    },
    {
      "epoch": 4.4342994689941406e-05,
      "model_forward_time": 1.727036952972412,
      "step": 46497
    },
    {
      "epoch": 4.4342994689941406e-05,
      "step": 46497,
      "training_step_time": 6.796257972717285
    },
    {
      "epoch": 4.434394836425781e-05,
      "model_forward_time": 1.7319376468658447,
      "step": 46498
    },
    {
      "epoch": 4.434394836425781e-05,
      "step": 46498,
      "training_step_time": 6.781561851501465
    },
    {
      "epoch": 4.434490203857422e-05,
      "model_forward_time": 1.3533313274383545,
      "step": 46499
    },
    {
      "epoch": 4.434490203857422e-05,
      "step": 46499,
      "training_step_time": 6.748847961425781
    },
    {
      "epoch": 4.4345855712890625e-05,
      "grad_norm": 0.13684576749801636,
      "learning_rate": 5.985747239586806e-06,
      "loss": 0.0581,
      "step": 46500
    },
    {
      "epoch": 4.4345855712890625e-05,
      "model_forward_time": 1.7223331928253174,
      "step": 46500
    },
    {
      "dynamics_loss_avg": 0.060264800861477855,
      "epoch": 4.4345855712890625e-05,
      "step": 46500
    },
    {
      "action_loss_avg": 0.001647729880642146,
      "epoch": 4.4345855712890625e-05,
      "step": 46500
    },
    {
      "epoch": 4.4345855712890625e-05,
      "step": 46500,
      "training_step_time": 6.790719747543335
    },
    {
      "epoch": 4.434680938720703e-05,
      "model_forward_time": 1.7242803573608398,
      "step": 46501
    },
    {
      "epoch": 4.434680938720703e-05,
      "step": 46501,
      "training_step_time": 7.10542368888855
    },
    {
      "epoch": 4.434776306152344e-05,
      "model_forward_time": 2.271838426589966,
      "step": 46502
    },
    {
      "epoch": 4.434776306152344e-05,
      "step": 46502,
      "training_step_time": 8.599754333496094
    },
    {
      "epoch": 4.4348716735839844e-05,
      "model_forward_time": 1.7766575813293457,
      "step": 46503
    },
    {
      "epoch": 4.4348716735839844e-05,
      "step": 46503,
      "training_step_time": 10.431155920028687
    },
    {
      "epoch": 4.434967041015625e-05,
      "model_forward_time": 1.732909917831421,
      "step": 46504
    },
    {
      "epoch": 4.434967041015625e-05,
      "step": 46504,
      "training_step_time": 8.624977350234985
    },
    {
      "epoch": 4.435062408447266e-05,
      "model_forward_time": 1.7236924171447754,
      "step": 46505
    },
    {
      "epoch": 4.435062408447266e-05,
      "step": 46505,
      "training_step_time": 7.158896207809448
    },
    {
      "epoch": 4.435157775878906e-05,
      "model_forward_time": 1.725811243057251,
      "step": 46506
    },
    {
      "epoch": 4.435157775878906e-05,
      "step": 46506,
      "training_step_time": 6.85882830619812
    },
    {
      "epoch": 4.435253143310547e-05,
      "model_forward_time": 1.737839937210083,
      "step": 46507
    },
    {
      "epoch": 4.435253143310547e-05,
      "step": 46507,
      "training_step_time": 6.897850275039673
    },
    {
      "epoch": 4.435348510742188e-05,
      "model_forward_time": 1.7319903373718262,
      "step": 46508
    },
    {
      "epoch": 4.435348510742188e-05,
      "step": 46508,
      "training_step_time": 6.91485071182251
    },
    {
      "epoch": 4.435443878173828e-05,
      "model_forward_time": 1.7280933856964111,
      "step": 46509
    },
    {
      "epoch": 4.435443878173828e-05,
      "step": 46509,
      "training_step_time": 6.890707969665527
    },
    {
      "epoch": 4.4355392456054686e-05,
      "grad_norm": 0.10652588307857513,
      "learning_rate": 5.9841261678120985e-06,
      "loss": 0.0577,
      "step": 46510
    },
    {
      "epoch": 4.4355392456054686e-05,
      "model_forward_time": 1.7220566272735596,
      "step": 46510
    },
    {
      "dynamics_loss_avg": 0.08710139896720648,
      "epoch": 4.4355392456054686e-05,
      "step": 46510
    },
    {
      "action_loss_avg": 0.0019048057612963021,
      "epoch": 4.4355392456054686e-05,
      "step": 46510
    },
    {
      "epoch": 4.4355392456054686e-05,
      "step": 46510,
      "training_step_time": 6.884443521499634
    },
    {
      "epoch": 4.4356346130371096e-05,
      "model_forward_time": 2.284748077392578,
      "step": 46511
    },
    {
      "epoch": 4.4356346130371096e-05,
      "step": 46511,
      "training_step_time": 7.782271862030029
    },
    {
      "epoch": 4.43572998046875e-05,
      "model_forward_time": 1.785036563873291,
      "step": 46512
    },
    {
      "epoch": 4.43572998046875e-05,
      "step": 46512,
      "training_step_time": 8.191958904266357
    },
    {
      "epoch": 4.4358253479003904e-05,
      "model_forward_time": 1.7293667793273926,
      "step": 46513
    },
    {
      "epoch": 4.4358253479003904e-05,
      "step": 46513,
      "training_step_time": 8.505117416381836
    },
    {
      "epoch": 4.4359207153320315e-05,
      "model_forward_time": 1.7323050498962402,
      "step": 46514
    },
    {
      "epoch": 4.4359207153320315e-05,
      "step": 46514,
      "training_step_time": 7.582714796066284
    },
    {
      "epoch": 4.436016082763672e-05,
      "model_forward_time": 1.731640338897705,
      "step": 46515
    },
    {
      "epoch": 4.436016082763672e-05,
      "step": 46515,
      "training_step_time": 6.912390232086182
    },
    {
      "epoch": 4.436111450195312e-05,
      "model_forward_time": 1.7302498817443848,
      "step": 46516
    },
    {
      "epoch": 4.436111450195312e-05,
      "step": 46516,
      "training_step_time": 6.897226572036743
    },
    {
      "epoch": 4.4362068176269534e-05,
      "model_forward_time": 1.7283859252929688,
      "step": 46517
    },
    {
      "epoch": 4.4362068176269534e-05,
      "step": 46517,
      "training_step_time": 6.872956991195679
    },
    {
      "epoch": 4.436302185058594e-05,
      "model_forward_time": 1.7305512428283691,
      "step": 46518
    },
    {
      "epoch": 4.436302185058594e-05,
      "step": 46518,
      "training_step_time": 6.874730110168457
    },
    {
      "epoch": 4.436397552490234e-05,
      "model_forward_time": 1.7391741275787354,
      "step": 46519
    },
    {
      "epoch": 4.436397552490234e-05,
      "step": 46519,
      "training_step_time": 6.905696868896484
    },
    {
      "epoch": 4.436492919921875e-05,
      "grad_norm": 0.13102228939533234,
      "learning_rate": 5.982504988414834e-06,
      "loss": 0.0588,
      "step": 46520
    },
    {
      "epoch": 4.436492919921875e-05,
      "model_forward_time": 1.7105300426483154,
      "step": 46520
    },
    {
      "dynamics_loss_avg": 0.06907026041299105,
      "epoch": 4.436492919921875e-05,
      "step": 46520
    },
    {
      "action_loss_avg": 0.0016747546760598198,
      "epoch": 4.436492919921875e-05,
      "step": 46520
    },
    {
      "epoch": 4.436492919921875e-05,
      "step": 46520,
      "training_step_time": 8.100056886672974
    },
    {
      "epoch": 4.436588287353516e-05,
      "model_forward_time": 1.7562072277069092,
      "step": 46521
    },
    {
      "epoch": 4.436588287353516e-05,
      "step": 46521,
      "training_step_time": 12.235292673110962
    },
    {
      "epoch": 4.436683654785156e-05,
      "model_forward_time": 1.7340850830078125,
      "step": 46522
    },
    {
      "epoch": 4.436683654785156e-05,
      "step": 46522,
      "training_step_time": 11.54902982711792
    },
    {
      "epoch": 4.436779022216797e-05,
      "model_forward_time": 1.7236742973327637,
      "step": 46523
    },
    {
      "epoch": 4.436779022216797e-05,
      "step": 46523,
      "training_step_time": 7.801053285598755
    },
    {
      "epoch": 4.4368743896484376e-05,
      "model_forward_time": 1.7370429039001465,
      "step": 46524
    },
    {
      "epoch": 4.4368743896484376e-05,
      "step": 46524,
      "training_step_time": 6.798282861709595
    },
    {
      "epoch": 4.436969757080078e-05,
      "model_forward_time": 1.7345287799835205,
      "step": 46525
    },
    {
      "epoch": 4.436969757080078e-05,
      "step": 46525,
      "training_step_time": 6.916528701782227
    },
    {
      "epoch": 4.437065124511719e-05,
      "model_forward_time": 1.7258169651031494,
      "step": 46526
    },
    {
      "epoch": 4.437065124511719e-05,
      "step": 46526,
      "training_step_time": 6.877323389053345
    },
    {
      "epoch": 4.4371604919433595e-05,
      "model_forward_time": 1.7333030700683594,
      "step": 46527
    },
    {
      "epoch": 4.4371604919433595e-05,
      "step": 46527,
      "training_step_time": 6.8666017055511475
    },
    {
      "epoch": 4.437255859375e-05,
      "model_forward_time": 1.334825038909912,
      "step": 46528
    },
    {
      "epoch": 4.437255859375e-05,
      "step": 46528,
      "training_step_time": 6.825212240219116
    },
    {
      "epoch": 4.437351226806641e-05,
      "model_forward_time": 1.7226738929748535,
      "step": 46529
    },
    {
      "epoch": 4.437351226806641e-05,
      "step": 46529,
      "training_step_time": 8.161011219024658
    },
    {
      "epoch": 4.4374465942382814e-05,
      "grad_norm": 0.1045495942234993,
      "learning_rate": 5.980883701572302e-06,
      "loss": 0.0618,
      "step": 46530
    },
    {
      "epoch": 4.4374465942382814e-05,
      "model_forward_time": 2.3174777030944824,
      "step": 46530
    },
    {
      "dynamics_loss_avg": 0.05277983527630568,
      "epoch": 4.4374465942382814e-05,
      "step": 46530
    },
    {
      "action_loss_avg": 0.0010147588065592573,
      "epoch": 4.4374465942382814e-05,
      "step": 46530
    },
    {
      "epoch": 4.4374465942382814e-05,
      "step": 46530,
      "training_step_time": 9.956672430038452
    },
    {
      "epoch": 4.437541961669922e-05,
      "model_forward_time": 1.7360129356384277,
      "step": 46531
    },
    {
      "epoch": 4.437541961669922e-05,
      "step": 46531,
      "training_step_time": 9.48041558265686
    },
    {
      "epoch": 4.437637329101562e-05,
      "model_forward_time": 1.7221708297729492,
      "step": 46532
    },
    {
      "epoch": 4.437637329101562e-05,
      "step": 46532,
      "training_step_time": 7.773465156555176
    },
    {
      "epoch": 4.437732696533203e-05,
      "model_forward_time": 1.7254867553710938,
      "step": 46533
    },
    {
      "epoch": 4.437732696533203e-05,
      "step": 46533,
      "training_step_time": 6.951670169830322
    },
    {
      "epoch": 4.4378280639648436e-05,
      "model_forward_time": 1.7259764671325684,
      "step": 46534
    },
    {
      "epoch": 4.4378280639648436e-05,
      "step": 46534,
      "training_step_time": 6.863847017288208
    },
    {
      "epoch": 4.437923431396484e-05,
      "model_forward_time": 1.7235291004180908,
      "step": 46535
    },
    {
      "epoch": 4.437923431396484e-05,
      "step": 46535,
      "training_step_time": 6.872594356536865
    },
    {
      "epoch": 4.438018798828125e-05,
      "model_forward_time": 1.7234604358673096,
      "step": 46536
    },
    {
      "epoch": 4.438018798828125e-05,
      "step": 46536,
      "training_step_time": 6.886229753494263
    },
    {
      "epoch": 4.4381141662597655e-05,
      "model_forward_time": 1.7236113548278809,
      "step": 46537
    },
    {
      "epoch": 4.4381141662597655e-05,
      "step": 46537,
      "training_step_time": 6.819399833679199
    },
    {
      "epoch": 4.438209533691406e-05,
      "model_forward_time": 1.722008228302002,
      "step": 46538
    },
    {
      "epoch": 4.438209533691406e-05,
      "step": 46538,
      "training_step_time": 7.494415760040283
    },
    {
      "epoch": 4.438304901123047e-05,
      "model_forward_time": 2.2667012214660645,
      "step": 46539
    },
    {
      "epoch": 4.438304901123047e-05,
      "step": 46539,
      "training_step_time": 10.315484046936035
    },
    {
      "epoch": 4.4384002685546874e-05,
      "grad_norm": 0.11333780735731125,
      "learning_rate": 5.979262307461802e-06,
      "loss": 0.058,
      "step": 46540
    },
    {
      "epoch": 4.4384002685546874e-05,
      "model_forward_time": 1.7246384620666504,
      "step": 46540
    },
    {
      "dynamics_loss_avg": 0.07118719983845949,
      "epoch": 4.4384002685546874e-05,
      "step": 46540
    },
    {
      "action_loss_avg": 0.0012904194125439971,
      "epoch": 4.4384002685546874e-05,
      "step": 46540
    },
    {
      "epoch": 4.4384002685546874e-05,
      "step": 46540,
      "training_step_time": 9.920638799667358
    },
    {
      "epoch": 4.438495635986328e-05,
      "model_forward_time": 1.7223553657531738,
      "step": 46541
    },
    {
      "epoch": 4.438495635986328e-05,
      "step": 46541,
      "training_step_time": 7.782629728317261
    },
    {
      "epoch": 4.438591003417969e-05,
      "model_forward_time": 1.729254961013794,
      "step": 46542
    },
    {
      "epoch": 4.438591003417969e-05,
      "step": 46542,
      "training_step_time": 7.7905919551849365
    },
    {
      "epoch": 4.438686370849609e-05,
      "model_forward_time": 1.7236196994781494,
      "step": 46543
    },
    {
      "epoch": 4.438686370849609e-05,
      "step": 46543,
      "training_step_time": 6.970078706741333
    },
    {
      "epoch": 4.43878173828125e-05,
      "model_forward_time": 1.7244453430175781,
      "step": 46544
    },
    {
      "epoch": 4.43878173828125e-05,
      "step": 46544,
      "training_step_time": 6.9396491050720215
    },
    {
      "epoch": 4.438877105712891e-05,
      "model_forward_time": 1.723381757736206,
      "step": 46545
    },
    {
      "epoch": 4.438877105712891e-05,
      "step": 46545,
      "training_step_time": 6.943030118942261
    },
    {
      "epoch": 4.438972473144531e-05,
      "model_forward_time": 1.7257297039031982,
      "step": 46546
    },
    {
      "epoch": 4.438972473144531e-05,
      "step": 46546,
      "training_step_time": 6.88963508605957
    },
    {
      "epoch": 4.4390678405761716e-05,
      "model_forward_time": 1.7272911071777344,
      "step": 46547
    },
    {
      "epoch": 4.4390678405761716e-05,
      "step": 46547,
      "training_step_time": 7.545106410980225
    },
    {
      "epoch": 4.439163208007813e-05,
      "model_forward_time": 2.2845964431762695,
      "step": 46548
    },
    {
      "epoch": 4.439163208007813e-05,
      "step": 46548,
      "training_step_time": 8.637077331542969
    },
    {
      "epoch": 4.439258575439453e-05,
      "model_forward_time": 1.762366533279419,
      "step": 46549
    },
    {
      "epoch": 4.439258575439453e-05,
      "step": 46549,
      "training_step_time": 10.758291244506836
    },
    {
      "epoch": 4.4393539428710935e-05,
      "grad_norm": 0.11474384367465973,
      "learning_rate": 5.977640806260648e-06,
      "loss": 0.0578,
      "step": 46550
    },
    {
      "epoch": 4.4393539428710935e-05,
      "model_forward_time": 1.7248351573944092,
      "step": 46550
    },
    {
      "dynamics_loss_avg": 0.07796809747815132,
      "epoch": 4.4393539428710935e-05,
      "step": 46550
    },
    {
      "action_loss_avg": 0.0015719560149591416,
      "epoch": 4.4393539428710935e-05,
      "step": 46550
    },
    {
      "epoch": 4.4393539428710935e-05,
      "step": 46550,
      "training_step_time": 8.7915198802948
    },
    {
      "epoch": 4.4394493103027346e-05,
      "model_forward_time": 1.733459234237671,
      "step": 46551
    },
    {
      "epoch": 4.4394493103027346e-05,
      "step": 46551,
      "training_step_time": 7.841224670410156
    },
    {
      "epoch": 4.439544677734375e-05,
      "model_forward_time": 1.7266688346862793,
      "step": 46552
    },
    {
      "epoch": 4.439544677734375e-05,
      "step": 46552,
      "training_step_time": 6.9261767864227295
    },
    {
      "epoch": 4.4396400451660154e-05,
      "model_forward_time": 1.7265617847442627,
      "step": 46553
    },
    {
      "epoch": 4.4396400451660154e-05,
      "step": 46553,
      "training_step_time": 6.89861536026001
    },
    {
      "epoch": 4.4397354125976564e-05,
      "model_forward_time": 1.726926565170288,
      "step": 46554
    },
    {
      "epoch": 4.4397354125976564e-05,
      "step": 46554,
      "training_step_time": 6.922661304473877
    },
    {
      "epoch": 4.439830780029297e-05,
      "model_forward_time": 1.7232978343963623,
      "step": 46555
    },
    {
      "epoch": 4.439830780029297e-05,
      "step": 46555,
      "training_step_time": 6.949678421020508
    },
    {
      "epoch": 4.439926147460937e-05,
      "model_forward_time": 1.730931043624878,
      "step": 46556
    },
    {
      "epoch": 4.439926147460937e-05,
      "step": 46556,
      "training_step_time": 7.422814846038818
    },
    {
      "epoch": 4.440021514892578e-05,
      "model_forward_time": 2.2719240188598633,
      "step": 46557
    },
    {
      "epoch": 4.440021514892578e-05,
      "step": 46557,
      "training_step_time": 8.83877182006836
    },
    {
      "epoch": 4.440116882324219e-05,
      "model_forward_time": 1.779829978942871,
      "step": 46558
    },
    {
      "epoch": 4.440116882324219e-05,
      "step": 46558,
      "training_step_time": 8.024885177612305
    },
    {
      "epoch": 4.440212249755859e-05,
      "model_forward_time": 1.7208714485168457,
      "step": 46559
    },
    {
      "epoch": 4.440212249755859e-05,
      "step": 46559,
      "training_step_time": 10.713993310928345
    },
    {
      "epoch": 4.4403076171875e-05,
      "grad_norm": 0.0825897753238678,
      "learning_rate": 5.976019198146167e-06,
      "loss": 0.0569,
      "step": 46560
    },
    {
      "epoch": 4.4403076171875e-05,
      "model_forward_time": 1.7154953479766846,
      "step": 46560
    },
    {
      "dynamics_loss_avg": 0.06431814674288035,
      "epoch": 4.4403076171875e-05,
      "step": 46560
    },
    {
      "action_loss_avg": 0.0025905203830916433,
      "epoch": 4.4403076171875e-05,
      "step": 46560
    },
    {
      "epoch": 4.4403076171875e-05,
      "step": 46560,
      "training_step_time": 7.73334527015686
    },
    {
      "epoch": 4.4404029846191406e-05,
      "model_forward_time": 1.7257404327392578,
      "step": 46561
    },
    {
      "epoch": 4.4404029846191406e-05,
      "step": 46561,
      "training_step_time": 7.181884288787842
    },
    {
      "epoch": 4.440498352050781e-05,
      "model_forward_time": 1.7304348945617676,
      "step": 46562
    },
    {
      "epoch": 4.440498352050781e-05,
      "step": 46562,
      "training_step_time": 6.901086330413818
    },
    {
      "epoch": 4.440593719482422e-05,
      "model_forward_time": 1.7230987548828125,
      "step": 46563
    },
    {
      "epoch": 4.440593719482422e-05,
      "step": 46563,
      "training_step_time": 6.851324796676636
    },
    {
      "epoch": 4.4406890869140625e-05,
      "model_forward_time": 1.7322602272033691,
      "step": 46564
    },
    {
      "epoch": 4.4406890869140625e-05,
      "step": 46564,
      "training_step_time": 6.905256271362305
    },
    {
      "epoch": 4.440784454345703e-05,
      "model_forward_time": 1.7243340015411377,
      "step": 46565
    },
    {
      "epoch": 4.440784454345703e-05,
      "step": 46565,
      "training_step_time": 7.92798113822937
    },
    {
      "epoch": 4.440879821777344e-05,
      "model_forward_time": 1.7351362705230713,
      "step": 46566
    },
    {
      "epoch": 4.440879821777344e-05,
      "step": 46566,
      "training_step_time": 8.357377052307129
    },
    {
      "epoch": 4.4409751892089844e-05,
      "model_forward_time": 1.9437949657440186,
      "step": 46567
    },
    {
      "epoch": 4.4409751892089844e-05,
      "step": 46567,
      "training_step_time": 9.022871971130371
    },
    {
      "epoch": 4.441070556640625e-05,
      "model_forward_time": 1.7354390621185303,
      "step": 46568
    },
    {
      "epoch": 4.441070556640625e-05,
      "step": 46568,
      "training_step_time": 9.136061668395996
    },
    {
      "epoch": 4.441165924072266e-05,
      "model_forward_time": 1.7334671020507812,
      "step": 46569
    },
    {
      "epoch": 4.441165924072266e-05,
      "step": 46569,
      "training_step_time": 8.240710973739624
    },
    {
      "epoch": 4.441261291503906e-05,
      "grad_norm": 0.09546875208616257,
      "learning_rate": 5.974397483295692e-06,
      "loss": 0.057,
      "step": 46570
    },
    {
      "epoch": 4.441261291503906e-05,
      "model_forward_time": 1.728487491607666,
      "step": 46570
    },
    {
      "dynamics_loss_avg": 0.062105795741081236,
      "epoch": 4.441261291503906e-05,
      "step": 46570
    },
    {
      "action_loss_avg": 0.0014195015770383178,
      "epoch": 4.441261291503906e-05,
      "step": 46570
    },
    {
      "epoch": 4.441261291503906e-05,
      "step": 46570,
      "training_step_time": 7.127800226211548
    },
    {
      "epoch": 4.441356658935547e-05,
      "model_forward_time": 1.730116367340088,
      "step": 46571
    },
    {
      "epoch": 4.441356658935547e-05,
      "step": 46571,
      "training_step_time": 6.748427867889404
    },
    {
      "epoch": 4.441452026367188e-05,
      "model_forward_time": 1.7307322025299072,
      "step": 46572
    },
    {
      "epoch": 4.441452026367188e-05,
      "step": 46572,
      "training_step_time": 6.7461395263671875
    },
    {
      "epoch": 4.441547393798828e-05,
      "model_forward_time": 1.7325859069824219,
      "step": 46573
    },
    {
      "epoch": 4.441547393798828e-05,
      "step": 46573,
      "training_step_time": 6.718345880508423
    },
    {
      "epoch": 4.4416427612304686e-05,
      "model_forward_time": 1.7305223941802979,
      "step": 46574
    },
    {
      "epoch": 4.4416427612304686e-05,
      "step": 46574,
      "training_step_time": 6.72154974937439
    },
    {
      "epoch": 4.4417381286621096e-05,
      "model_forward_time": 1.7309558391571045,
      "step": 46575
    },
    {
      "epoch": 4.4417381286621096e-05,
      "step": 46575,
      "training_step_time": 7.621314287185669
    },
    {
      "epoch": 4.44183349609375e-05,
      "model_forward_time": 2.324273109436035,
      "step": 46576
    },
    {
      "epoch": 4.44183349609375e-05,
      "step": 46576,
      "training_step_time": 7.635557174682617
    },
    {
      "epoch": 4.4419288635253905e-05,
      "model_forward_time": 1.7335841655731201,
      "step": 46577
    },
    {
      "epoch": 4.4419288635253905e-05,
      "step": 46577,
      "training_step_time": 8.354048013687134
    },
    {
      "epoch": 4.4420242309570315e-05,
      "model_forward_time": 1.7312572002410889,
      "step": 46578
    },
    {
      "epoch": 4.4420242309570315e-05,
      "step": 46578,
      "training_step_time": 7.7163379192352295
    },
    {
      "epoch": 4.442119598388672e-05,
      "model_forward_time": 1.7300138473510742,
      "step": 46579
    },
    {
      "epoch": 4.442119598388672e-05,
      "step": 46579,
      "training_step_time": 7.371609449386597
    },
    {
      "epoch": 4.442214965820312e-05,
      "grad_norm": 0.11042650789022446,
      "learning_rate": 5.972775661886575e-06,
      "loss": 0.0576,
      "step": 46580
    },
    {
      "epoch": 4.442214965820312e-05,
      "model_forward_time": 1.7214114665985107,
      "step": 46580
    },
    {
      "dynamics_loss_avg": 0.051674591936171056,
      "epoch": 4.442214965820312e-05,
      "step": 46580
    },
    {
      "action_loss_avg": 0.0015350110916187987,
      "epoch": 4.442214965820312e-05,
      "step": 46580
    },
    {
      "epoch": 4.442214965820312e-05,
      "step": 46580,
      "training_step_time": 6.722943067550659
    },
    {
      "epoch": 4.4423103332519534e-05,
      "model_forward_time": 1.7282254695892334,
      "step": 46581
    },
    {
      "epoch": 4.4423103332519534e-05,
      "step": 46581,
      "training_step_time": 6.754823923110962
    },
    {
      "epoch": 4.442405700683594e-05,
      "model_forward_time": 1.7252321243286133,
      "step": 46582
    },
    {
      "epoch": 4.442405700683594e-05,
      "step": 46582,
      "training_step_time": 6.705977916717529
    },
    {
      "epoch": 4.442501068115234e-05,
      "model_forward_time": 1.745117425918579,
      "step": 46583
    },
    {
      "epoch": 4.442501068115234e-05,
      "step": 46583,
      "training_step_time": 6.731430292129517
    },
    {
      "epoch": 4.442596435546875e-05,
      "model_forward_time": 1.7267394065856934,
      "step": 46584
    },
    {
      "epoch": 4.442596435546875e-05,
      "step": 46584,
      "training_step_time": 8.283672571182251
    },
    {
      "epoch": 4.442691802978516e-05,
      "model_forward_time": 2.2905561923980713,
      "step": 46585
    },
    {
      "epoch": 4.442691802978516e-05,
      "step": 46585,
      "training_step_time": 8.681071519851685
    },
    {
      "epoch": 4.442787170410156e-05,
      "model_forward_time": 1.7760734558105469,
      "step": 46586
    },
    {
      "epoch": 4.442787170410156e-05,
      "step": 46586,
      "training_step_time": 9.5009126663208
    },
    {
      "epoch": 4.442882537841797e-05,
      "model_forward_time": 1.7278528213500977,
      "step": 46587
    },
    {
      "epoch": 4.442882537841797e-05,
      "step": 46587,
      "training_step_time": 7.818605899810791
    },
    {
      "epoch": 4.4429779052734376e-05,
      "model_forward_time": 1.7454924583435059,
      "step": 46588
    },
    {
      "epoch": 4.4429779052734376e-05,
      "step": 46588,
      "training_step_time": 7.541508197784424
    },
    {
      "epoch": 4.443073272705078e-05,
      "model_forward_time": 1.7323665618896484,
      "step": 46589
    },
    {
      "epoch": 4.443073272705078e-05,
      "step": 46589,
      "training_step_time": 7.092007875442505
    },
    {
      "epoch": 4.443168640136719e-05,
      "grad_norm": 0.12783285975456238,
      "learning_rate": 5.971153734096174e-06,
      "loss": 0.0545,
      "step": 46590
    },
    {
      "epoch": 4.443168640136719e-05,
      "model_forward_time": 1.7299613952636719,
      "step": 46590
    },
    {
      "dynamics_loss_avg": 0.04566926099359989,
      "epoch": 4.443168640136719e-05,
      "step": 46590
    },
    {
      "action_loss_avg": 0.0015180460875853896,
      "epoch": 4.443168640136719e-05,
      "step": 46590
    },
    {
      "epoch": 4.443168640136719e-05,
      "step": 46590,
      "training_step_time": 6.78143835067749
    },
    {
      "epoch": 4.4432640075683595e-05,
      "model_forward_time": 1.7311177253723145,
      "step": 46591
    },
    {
      "epoch": 4.4432640075683595e-05,
      "step": 46591,
      "training_step_time": 6.7711286544799805
    },
    {
      "epoch": 4.443359375e-05,
      "model_forward_time": 1.7351274490356445,
      "step": 46592
    },
    {
      "epoch": 4.443359375e-05,
      "step": 46592,
      "training_step_time": 6.752719163894653
    },
    {
      "epoch": 4.443454742431641e-05,
      "model_forward_time": 1.7363693714141846,
      "step": 46593
    },
    {
      "epoch": 4.443454742431641e-05,
      "step": 46593,
      "training_step_time": 8.372633457183838
    },
    {
      "epoch": 4.4435501098632814e-05,
      "model_forward_time": 2.303668975830078,
      "step": 46594
    },
    {
      "epoch": 4.4435501098632814e-05,
      "step": 46594,
      "training_step_time": 8.395822525024414
    },
    {
      "epoch": 4.443645477294922e-05,
      "model_forward_time": 1.7705113887786865,
      "step": 46595
    },
    {
      "epoch": 4.443645477294922e-05,
      "step": 46595,
      "training_step_time": 9.929325580596924
    },
    {
      "epoch": 4.443740844726562e-05,
      "model_forward_time": 1.7261967658996582,
      "step": 46596
    },
    {
      "epoch": 4.443740844726562e-05,
      "step": 46596,
      "training_step_time": 9.272506475448608
    },
    {
      "epoch": 4.443836212158203e-05,
      "model_forward_time": 1.7219018936157227,
      "step": 46597
    },
    {
      "epoch": 4.443836212158203e-05,
      "step": 46597,
      "training_step_time": 7.496268033981323
    },
    {
      "epoch": 4.4439315795898437e-05,
      "model_forward_time": 1.727799415588379,
      "step": 46598
    },
    {
      "epoch": 4.4439315795898437e-05,
      "step": 46598,
      "training_step_time": 7.093271732330322
    },
    {
      "epoch": 4.444026947021484e-05,
      "model_forward_time": 1.73050856590271,
      "step": 46599
    },
    {
      "epoch": 4.444026947021484e-05,
      "step": 46599,
      "training_step_time": 6.751679182052612
    },
    {
      "epoch": 4.444122314453125e-05,
      "grad_norm": 0.13034509122371674,
      "learning_rate": 5.9695317001018595e-06,
      "loss": 0.0581,
      "step": 46600
    },
    {
      "epoch": 4.444122314453125e-05,
      "model_forward_time": 1.7328248023986816,
      "step": 46600
    },
    {
      "dynamics_loss_avg": 0.0501839792355895,
      "epoch": 4.444122314453125e-05,
      "step": 46600
    },
    {
      "action_loss_avg": 0.0009748880518600344,
      "epoch": 4.444122314453125e-05,
      "step": 46600
    },
    {
      "epoch": 4.444122314453125e-05,
      "step": 46600,
      "training_step_time": 6.7620227336883545
    },
    {
      "epoch": 4.4442176818847655e-05,
      "model_forward_time": 1.7320175170898438,
      "step": 46601
    },
    {
      "epoch": 4.4442176818847655e-05,
      "step": 46601,
      "training_step_time": 6.798545837402344
    },
    {
      "epoch": 4.444313049316406e-05,
      "model_forward_time": 1.7218315601348877,
      "step": 46602
    },
    {
      "epoch": 4.444313049316406e-05,
      "step": 46602,
      "training_step_time": 9.063445806503296
    },
    {
      "epoch": 4.444408416748047e-05,
      "model_forward_time": 1.7225959300994873,
      "step": 46603
    },
    {
      "epoch": 4.444408416748047e-05,
      "step": 46603,
      "training_step_time": 9.057742595672607
    },
    {
      "epoch": 4.4445037841796874e-05,
      "model_forward_time": 1.7567827701568604,
      "step": 46604
    },
    {
      "epoch": 4.4445037841796874e-05,
      "step": 46604,
      "training_step_time": 8.819149494171143
    },
    {
      "epoch": 4.444599151611328e-05,
      "model_forward_time": 1.7308523654937744,
      "step": 46605
    },
    {
      "epoch": 4.444599151611328e-05,
      "step": 46605,
      "training_step_time": 8.29246973991394
    },
    {
      "epoch": 4.444694519042969e-05,
      "model_forward_time": 1.721170425415039,
      "step": 46606
    },
    {
      "epoch": 4.444694519042969e-05,
      "step": 46606,
      "training_step_time": 7.8356077671051025
    },
    {
      "epoch": 4.444789886474609e-05,
      "model_forward_time": 1.7244594097137451,
      "step": 46607
    },
    {
      "epoch": 4.444789886474609e-05,
      "step": 46607,
      "training_step_time": 7.908218860626221
    },
    {
      "epoch": 4.44488525390625e-05,
      "model_forward_time": 1.726785659790039,
      "step": 46608
    },
    {
      "epoch": 4.44488525390625e-05,
      "step": 46608,
      "training_step_time": 6.862425327301025
    },
    {
      "epoch": 4.444980621337891e-05,
      "model_forward_time": 1.7252707481384277,
      "step": 46609
    },
    {
      "epoch": 4.444980621337891e-05,
      "step": 46609,
      "training_step_time": 6.9057457447052
    },
    {
      "epoch": 4.445075988769531e-05,
      "grad_norm": 0.13885323703289032,
      "learning_rate": 5.967909560081015e-06,
      "loss": 0.0544,
      "step": 46610
    },
    {
      "epoch": 4.445075988769531e-05,
      "model_forward_time": 1.718472957611084,
      "step": 46610
    },
    {
      "dynamics_loss_avg": 0.0452119616791606,
      "epoch": 4.445075988769531e-05,
      "step": 46610
    },
    {
      "action_loss_avg": 0.0008471402048598975,
      "epoch": 4.445075988769531e-05,
      "step": 46610
    },
    {
      "epoch": 4.445075988769531e-05,
      "step": 46610,
      "training_step_time": 6.842279434204102
    },
    {
      "epoch": 4.4451713562011716e-05,
      "model_forward_time": 1.7243854999542236,
      "step": 46611
    },
    {
      "epoch": 4.4451713562011716e-05,
      "step": 46611,
      "training_step_time": 8.423491954803467
    },
    {
      "epoch": 4.445266723632813e-05,
      "model_forward_time": 1.721799373626709,
      "step": 46612
    },
    {
      "epoch": 4.445266723632813e-05,
      "step": 46612,
      "training_step_time": 8.141995668411255
    },
    {
      "epoch": 4.445362091064453e-05,
      "model_forward_time": 2.3291709423065186,
      "step": 46613
    },
    {
      "epoch": 4.445362091064453e-05,
      "step": 46613,
      "training_step_time": 8.527616024017334
    },
    {
      "epoch": 4.4454574584960935e-05,
      "model_forward_time": 1.7300999164581299,
      "step": 46614
    },
    {
      "epoch": 4.4454574584960935e-05,
      "step": 46614,
      "training_step_time": 8.310121297836304
    },
    {
      "epoch": 4.4455528259277346e-05,
      "model_forward_time": 1.722848892211914,
      "step": 46615
    },
    {
      "epoch": 4.4455528259277346e-05,
      "step": 46615,
      "training_step_time": 9.369693517684937
    },
    {
      "epoch": 4.445648193359375e-05,
      "model_forward_time": 1.735166311264038,
      "step": 46616
    },
    {
      "epoch": 4.445648193359375e-05,
      "step": 46616,
      "training_step_time": 7.335888624191284
    },
    {
      "epoch": 4.4457435607910154e-05,
      "model_forward_time": 1.728734016418457,
      "step": 46617
    },
    {
      "epoch": 4.4457435607910154e-05,
      "step": 46617,
      "training_step_time": 7.01657509803772
    },
    {
      "epoch": 4.4458389282226565e-05,
      "model_forward_time": 1.7339563369750977,
      "step": 46618
    },
    {
      "epoch": 4.4458389282226565e-05,
      "step": 46618,
      "training_step_time": 6.735477924346924
    },
    {
      "epoch": 4.445934295654297e-05,
      "model_forward_time": 1.725189447402954,
      "step": 46619
    },
    {
      "epoch": 4.445934295654297e-05,
      "step": 46619,
      "training_step_time": 6.699071168899536
    },
    {
      "epoch": 4.446029663085937e-05,
      "grad_norm": 0.09883379191160202,
      "learning_rate": 5.9662873142110386e-06,
      "loss": 0.0597,
      "step": 46620
    },
    {
      "epoch": 4.446029663085937e-05,
      "model_forward_time": 1.7295010089874268,
      "step": 46620
    },
    {
      "dynamics_loss_avg": 0.056093609519302846,
      "epoch": 4.446029663085937e-05,
      "step": 46620
    },
    {
      "action_loss_avg": 0.0017407276493031532,
      "epoch": 4.446029663085937e-05,
      "step": 46620
    },
    {
      "epoch": 4.446029663085937e-05,
      "step": 46620,
      "training_step_time": 7.564471960067749
    },
    {
      "epoch": 4.446125030517578e-05,
      "model_forward_time": 1.726957082748413,
      "step": 46621
    },
    {
      "epoch": 4.446125030517578e-05,
      "step": 46621,
      "training_step_time": 7.534967660903931
    },
    {
      "epoch": 4.446220397949219e-05,
      "model_forward_time": 2.2802011966705322,
      "step": 46622
    },
    {
      "epoch": 4.446220397949219e-05,
      "step": 46622,
      "training_step_time": 7.933429479598999
    },
    {
      "epoch": 4.446315765380859e-05,
      "model_forward_time": 1.7680537700653076,
      "step": 46623
    },
    {
      "epoch": 4.446315765380859e-05,
      "step": 46623,
      "training_step_time": 9.438560485839844
    },
    {
      "epoch": 4.4464111328125e-05,
      "model_forward_time": 1.7248828411102295,
      "step": 46624
    },
    {
      "epoch": 4.4464111328125e-05,
      "step": 46624,
      "training_step_time": 7.78717303276062
    },
    {
      "epoch": 4.4465065002441406e-05,
      "model_forward_time": 1.7261464595794678,
      "step": 46625
    },
    {
      "epoch": 4.4465065002441406e-05,
      "step": 46625,
      "training_step_time": 7.353433609008789
    },
    {
      "epoch": 4.446601867675781e-05,
      "model_forward_time": 1.7347948551177979,
      "step": 46626
    },
    {
      "epoch": 4.446601867675781e-05,
      "step": 46626,
      "training_step_time": 7.061841726303101
    },
    {
      "epoch": 4.446697235107422e-05,
      "model_forward_time": 1.7241582870483398,
      "step": 46627
    },
    {
      "epoch": 4.446697235107422e-05,
      "step": 46627,
      "training_step_time": 6.800680875778198
    },
    {
      "epoch": 4.4467926025390625e-05,
      "model_forward_time": 1.731330394744873,
      "step": 46628
    },
    {
      "epoch": 4.4467926025390625e-05,
      "step": 46628,
      "training_step_time": 6.773879051208496
    },
    {
      "epoch": 4.446887969970703e-05,
      "model_forward_time": 1.7262120246887207,
      "step": 46629
    },
    {
      "epoch": 4.446887969970703e-05,
      "step": 46629,
      "training_step_time": 8.388796329498291
    },
    {
      "epoch": 4.446983337402344e-05,
      "grad_norm": 0.11201272904872894,
      "learning_rate": 5.9646649626693335e-06,
      "loss": 0.0584,
      "step": 46630
    },
    {
      "epoch": 4.446983337402344e-05,
      "model_forward_time": 1.7249975204467773,
      "step": 46630
    },
    {
      "dynamics_loss_avg": 0.07014713697135448,
      "epoch": 4.446983337402344e-05,
      "step": 46630
    },
    {
      "action_loss_avg": 0.0026644967205356807,
      "epoch": 4.446983337402344e-05,
      "step": 46630
    },
    {
      "epoch": 4.446983337402344e-05,
      "step": 46630,
      "training_step_time": 8.466125011444092
    },
    {
      "epoch": 4.4470787048339844e-05,
      "model_forward_time": 2.2870371341705322,
      "step": 46631
    },
    {
      "epoch": 4.4470787048339844e-05,
      "step": 46631,
      "training_step_time": 9.534382104873657
    },
    {
      "epoch": 4.447174072265625e-05,
      "model_forward_time": 1.7754955291748047,
      "step": 46632
    },
    {
      "epoch": 4.447174072265625e-05,
      "step": 46632,
      "training_step_time": 9.149999856948853
    },
    {
      "epoch": 4.447269439697266e-05,
      "model_forward_time": 1.7185614109039307,
      "step": 46633
    },
    {
      "epoch": 4.447269439697266e-05,
      "step": 46633,
      "training_step_time": 7.700544357299805
    },
    {
      "epoch": 4.447364807128906e-05,
      "model_forward_time": 1.7244524955749512,
      "step": 46634
    },
    {
      "epoch": 4.447364807128906e-05,
      "step": 46634,
      "training_step_time": 7.607367515563965
    },
    {
      "epoch": 4.447460174560547e-05,
      "model_forward_time": 1.7200133800506592,
      "step": 46635
    },
    {
      "epoch": 4.447460174560547e-05,
      "step": 46635,
      "training_step_time": 7.059001445770264
    },
    {
      "epoch": 4.447555541992188e-05,
      "model_forward_time": 1.7238194942474365,
      "step": 46636
    },
    {
      "epoch": 4.447555541992188e-05,
      "step": 46636,
      "training_step_time": 6.769669055938721
    },
    {
      "epoch": 4.447650909423828e-05,
      "model_forward_time": 1.7240846157073975,
      "step": 46637
    },
    {
      "epoch": 4.447650909423828e-05,
      "step": 46637,
      "training_step_time": 6.79274582862854
    },
    {
      "epoch": 4.4477462768554686e-05,
      "model_forward_time": 1.7417266368865967,
      "step": 46638
    },
    {
      "epoch": 4.4477462768554686e-05,
      "step": 46638,
      "training_step_time": 7.632064580917358
    },
    {
      "epoch": 4.4478416442871097e-05,
      "model_forward_time": 1.7279350757598877,
      "step": 46639
    },
    {
      "epoch": 4.4478416442871097e-05,
      "step": 46639,
      "training_step_time": 7.605617523193359
    },
    {
      "epoch": 4.44793701171875e-05,
      "grad_norm": 0.11749351024627686,
      "learning_rate": 5.963042505633319e-06,
      "loss": 0.0569,
      "step": 46640
    },
    {
      "epoch": 4.44793701171875e-05,
      "model_forward_time": 2.2928249835968018,
      "step": 46640
    },
    {
      "dynamics_loss_avg": 0.05247950702905655,
      "epoch": 4.44793701171875e-05,
      "step": 46640
    },
    {
      "action_loss_avg": 0.0014387927105417476,
      "epoch": 4.44793701171875e-05,
      "step": 46640
    },
    {
      "epoch": 4.44793701171875e-05,
      "step": 46640,
      "training_step_time": 8.179542303085327
    },
    {
      "epoch": 4.4480323791503905e-05,
      "model_forward_time": 1.7602009773254395,
      "step": 46641
    },
    {
      "epoch": 4.4480323791503905e-05,
      "step": 46641,
      "training_step_time": 8.075417518615723
    },
    {
      "epoch": 4.4481277465820315e-05,
      "model_forward_time": 1.7216026782989502,
      "step": 46642
    },
    {
      "epoch": 4.4481277465820315e-05,
      "step": 46642,
      "training_step_time": 8.505972385406494
    },
    {
      "epoch": 4.448223114013672e-05,
      "model_forward_time": 1.7208201885223389,
      "step": 46643
    },
    {
      "epoch": 4.448223114013672e-05,
      "step": 46643,
      "training_step_time": 7.9051408767700195
    },
    {
      "epoch": 4.4483184814453123e-05,
      "model_forward_time": 1.7449357509613037,
      "step": 46644
    },
    {
      "epoch": 4.4483184814453123e-05,
      "step": 46644,
      "training_step_time": 7.29079008102417
    },
    {
      "epoch": 4.4484138488769534e-05,
      "model_forward_time": 1.720313310623169,
      "step": 46645
    },
    {
      "epoch": 4.4484138488769534e-05,
      "step": 46645,
      "training_step_time": 6.946470737457275
    },
    {
      "epoch": 4.448509216308594e-05,
      "model_forward_time": 1.7214670181274414,
      "step": 46646
    },
    {
      "epoch": 4.448509216308594e-05,
      "step": 46646,
      "training_step_time": 6.969027519226074
    },
    {
      "epoch": 4.448604583740234e-05,
      "model_forward_time": 1.7251009941101074,
      "step": 46647
    },
    {
      "epoch": 4.448604583740234e-05,
      "step": 46647,
      "training_step_time": 7.331557750701904
    },
    {
      "epoch": 4.448699951171875e-05,
      "model_forward_time": 1.7332127094268799,
      "step": 46648
    },
    {
      "epoch": 4.448699951171875e-05,
      "step": 46648,
      "training_step_time": 7.818434000015259
    },
    {
      "epoch": 4.448795318603516e-05,
      "model_forward_time": 2.270216464996338,
      "step": 46649
    },
    {
      "epoch": 4.448795318603516e-05,
      "step": 46649,
      "training_step_time": 9.50938367843628
    },
    {
      "epoch": 4.448890686035156e-05,
      "grad_norm": 0.09812554717063904,
      "learning_rate": 5.961419943280422e-06,
      "loss": 0.0561,
      "step": 46650
    },
    {
      "epoch": 4.448890686035156e-05,
      "model_forward_time": 1.7368662357330322,
      "step": 46650
    },
    {
      "dynamics_loss_avg": 0.06175877917557955,
      "epoch": 4.448890686035156e-05,
      "step": 46650
    },
    {
      "action_loss_avg": 0.0011999725829809904,
      "epoch": 4.448890686035156e-05,
      "step": 46650
    },
    {
      "epoch": 4.448890686035156e-05,
      "step": 46650,
      "training_step_time": 8.009318113327026
    },
    {
      "epoch": 4.448986053466797e-05,
      "model_forward_time": 1.7188348770141602,
      "step": 46651
    },
    {
      "epoch": 4.448986053466797e-05,
      "step": 46651,
      "training_step_time": 8.98076581954956
    },
    {
      "epoch": 4.4490814208984376e-05,
      "model_forward_time": 1.721937894821167,
      "step": 46652
    },
    {
      "epoch": 4.4490814208984376e-05,
      "step": 46652,
      "training_step_time": 8.916778564453125
    },
    {
      "epoch": 4.449176788330078e-05,
      "model_forward_time": 1.726588249206543,
      "step": 46653
    },
    {
      "epoch": 4.449176788330078e-05,
      "step": 46653,
      "training_step_time": 7.350640773773193
    },
    {
      "epoch": 4.449272155761719e-05,
      "model_forward_time": 1.7189645767211914,
      "step": 46654
    },
    {
      "epoch": 4.449272155761719e-05,
      "step": 46654,
      "training_step_time": 6.897113800048828
    },
    {
      "epoch": 4.4493675231933595e-05,
      "model_forward_time": 1.731828212738037,
      "step": 46655
    },
    {
      "epoch": 4.4493675231933595e-05,
      "step": 46655,
      "training_step_time": 6.8906402587890625
    },
    {
      "epoch": 4.449462890625e-05,
      "model_forward_time": 1.716439962387085,
      "step": 46656
    },
    {
      "epoch": 4.449462890625e-05,
      "step": 46656,
      "training_step_time": 6.797706604003906
    },
    {
      "epoch": 4.44955825805664e-05,
      "model_forward_time": 1.7215008735656738,
      "step": 46657
    },
    {
      "epoch": 4.44955825805664e-05,
      "step": 46657,
      "training_step_time": 8.48265790939331
    },
    {
      "epoch": 4.4496536254882814e-05,
      "model_forward_time": 1.7120089530944824,
      "step": 46658
    },
    {
      "epoch": 4.4496536254882814e-05,
      "step": 46658,
      "training_step_time": 7.709364891052246
    },
    {
      "epoch": 4.449748992919922e-05,
      "model_forward_time": 2.2758541107177734,
      "step": 46659
    },
    {
      "epoch": 4.449748992919922e-05,
      "step": 46659,
      "training_step_time": 9.452501773834229
    },
    {
      "epoch": 4.449844360351562e-05,
      "grad_norm": 0.12227348238229752,
      "learning_rate": 5.9597972757880876e-06,
      "loss": 0.055,
      "step": 46660
    },
    {
      "epoch": 4.449844360351562e-05,
      "model_forward_time": 1.7157883644104004,
      "step": 46660
    },
    {
      "dynamics_loss_avg": 0.06854774840176106,
      "epoch": 4.449844360351562e-05,
      "step": 46660
    },
    {
      "action_loss_avg": 0.0021307488554157318,
      "epoch": 4.449844360351562e-05,
      "step": 46660
    },
    {
      "epoch": 4.449844360351562e-05,
      "step": 46660,
      "training_step_time": 8.826326131820679
    },
    {
      "epoch": 4.449939727783203e-05,
      "model_forward_time": 1.7234158515930176,
      "step": 46661
    },
    {
      "epoch": 4.449939727783203e-05,
      "step": 46661,
      "training_step_time": 8.866847038269043
    },
    {
      "epoch": 4.4500350952148437e-05,
      "model_forward_time": 1.7168476581573486,
      "step": 46662
    },
    {
      "epoch": 4.4500350952148437e-05,
      "step": 46662,
      "training_step_time": 7.683691740036011
    },
    {
      "epoch": 4.450130462646484e-05,
      "model_forward_time": 1.7234876155853271,
      "step": 46663
    },
    {
      "epoch": 4.450130462646484e-05,
      "step": 46663,
      "training_step_time": 7.239361763000488
    },
    {
      "epoch": 4.450225830078125e-05,
      "model_forward_time": 1.7176072597503662,
      "step": 46664
    },
    {
      "epoch": 4.450225830078125e-05,
      "step": 46664,
      "training_step_time": 6.773790121078491
    },
    {
      "epoch": 4.4503211975097655e-05,
      "model_forward_time": 1.716773271560669,
      "step": 46665
    },
    {
      "epoch": 4.4503211975097655e-05,
      "step": 46665,
      "training_step_time": 6.893765687942505
    },
    {
      "epoch": 4.450416564941406e-05,
      "model_forward_time": 1.717106819152832,
      "step": 46666
    },
    {
      "epoch": 4.450416564941406e-05,
      "step": 46666,
      "training_step_time": 7.706010818481445
    },
    {
      "epoch": 4.450511932373047e-05,
      "model_forward_time": 1.7247905731201172,
      "step": 46667
    },
    {
      "epoch": 4.450511932373047e-05,
      "step": 46667,
      "training_step_time": 7.719723224639893
    },
    {
      "epoch": 4.4506072998046874e-05,
      "model_forward_time": 2.274794816970825,
      "step": 46668
    },
    {
      "epoch": 4.4506072998046874e-05,
      "step": 46668,
      "training_step_time": 10.839529514312744
    },
    {
      "epoch": 4.450702667236328e-05,
      "model_forward_time": 1.768831491470337,
      "step": 46669
    },
    {
      "epoch": 4.450702667236328e-05,
      "step": 46669,
      "training_step_time": 9.739612102508545
    },
    {
      "epoch": 4.450798034667969e-05,
      "grad_norm": 0.09264051914215088,
      "learning_rate": 5.958174503333764e-06,
      "loss": 0.0574,
      "step": 46670
    },
    {
      "epoch": 4.450798034667969e-05,
      "model_forward_time": 1.7278556823730469,
      "step": 46670
    },
    {
      "dynamics_loss_avg": 0.05835385955870152,
      "epoch": 4.450798034667969e-05,
      "step": 46670
    },
    {
      "action_loss_avg": 0.001128818595316261,
      "epoch": 4.450798034667969e-05,
      "step": 46670
    },
    {
      "epoch": 4.450798034667969e-05,
      "step": 46670,
      "training_step_time": 8.586357593536377
    },
    {
      "epoch": 4.450893402099609e-05,
      "model_forward_time": 1.7179656028747559,
      "step": 46671
    },
    {
      "epoch": 4.450893402099609e-05,
      "step": 46671,
      "training_step_time": 7.724349498748779
    },
    {
      "epoch": 4.45098876953125e-05,
      "model_forward_time": 1.7148544788360596,
      "step": 46672
    },
    {
      "epoch": 4.45098876953125e-05,
      "step": 46672,
      "training_step_time": 7.3008527755737305
    },
    {
      "epoch": 4.451084136962891e-05,
      "model_forward_time": 1.7320518493652344,
      "step": 46673
    },
    {
      "epoch": 4.451084136962891e-05,
      "step": 46673,
      "training_step_time": 6.84596848487854
    },
    {
      "epoch": 4.451179504394531e-05,
      "model_forward_time": 1.7091681957244873,
      "step": 46674
    },
    {
      "epoch": 4.451179504394531e-05,
      "step": 46674,
      "training_step_time": 6.8503313064575195
    },
    {
      "epoch": 4.4512748718261716e-05,
      "model_forward_time": 1.7071712017059326,
      "step": 46675
    },
    {
      "epoch": 4.4512748718261716e-05,
      "step": 46675,
      "training_step_time": 7.723156929016113
    },
    {
      "epoch": 4.451370239257813e-05,
      "model_forward_time": 1.7138001918792725,
      "step": 46676
    },
    {
      "epoch": 4.451370239257813e-05,
      "step": 46676,
      "training_step_time": 8.415371656417847
    },
    {
      "epoch": 4.451465606689453e-05,
      "model_forward_time": 2.268886089324951,
      "step": 46677
    },
    {
      "epoch": 4.451465606689453e-05,
      "step": 46677,
      "training_step_time": 8.789764881134033
    },
    {
      "epoch": 4.4515609741210935e-05,
      "model_forward_time": 1.717757225036621,
      "step": 46678
    },
    {
      "epoch": 4.4515609741210935e-05,
      "step": 46678,
      "training_step_time": 9.16130781173706
    },
    {
      "epoch": 4.4516563415527346e-05,
      "model_forward_time": 1.7162516117095947,
      "step": 46679
    },
    {
      "epoch": 4.4516563415527346e-05,
      "step": 46679,
      "training_step_time": 8.388495922088623
    },
    {
      "epoch": 4.451751708984375e-05,
      "grad_norm": 0.12253369390964508,
      "learning_rate": 5.95655162609492e-06,
      "loss": 0.0548,
      "step": 46680
    },
    {
      "epoch": 4.451751708984375e-05,
      "model_forward_time": 1.7182793617248535,
      "step": 46680
    },
    {
      "dynamics_loss_avg": 0.06311504049226642,
      "epoch": 4.451751708984375e-05,
      "step": 46680
    },
    {
      "action_loss_avg": 0.001540910336188972,
      "epoch": 4.451751708984375e-05,
      "step": 46680
    },
    {
      "epoch": 4.451751708984375e-05,
      "step": 46680,
      "training_step_time": 7.836572647094727
    },
    {
      "epoch": 4.4518470764160154e-05,
      "model_forward_time": 1.7177550792694092,
      "step": 46681
    },
    {
      "epoch": 4.4518470764160154e-05,
      "step": 46681,
      "training_step_time": 7.334524154663086
    },
    {
      "epoch": 4.4519424438476565e-05,
      "model_forward_time": 1.7214837074279785,
      "step": 46682
    },
    {
      "epoch": 4.4519424438476565e-05,
      "step": 46682,
      "training_step_time": 6.8722662925720215
    },
    {
      "epoch": 4.452037811279297e-05,
      "model_forward_time": 1.7182376384735107,
      "step": 46683
    },
    {
      "epoch": 4.452037811279297e-05,
      "step": 46683,
      "training_step_time": 6.869755506515503
    },
    {
      "epoch": 4.452133178710937e-05,
      "model_forward_time": 1.7186717987060547,
      "step": 46684
    },
    {
      "epoch": 4.452133178710937e-05,
      "step": 46684,
      "training_step_time": 7.348117828369141
    },
    {
      "epoch": 4.4522285461425783e-05,
      "model_forward_time": 1.7230033874511719,
      "step": 46685
    },
    {
      "epoch": 4.4522285461425783e-05,
      "step": 46685,
      "training_step_time": 7.78344202041626
    },
    {
      "epoch": 4.452323913574219e-05,
      "model_forward_time": 2.2571640014648438,
      "step": 46686
    },
    {
      "epoch": 4.452323913574219e-05,
      "step": 46686,
      "training_step_time": 9.428160905838013
    },
    {
      "epoch": 4.452419281005859e-05,
      "model_forward_time": 1.717118501663208,
      "step": 46687
    },
    {
      "epoch": 4.452419281005859e-05,
      "step": 46687,
      "training_step_time": 8.788232564926147
    },
    {
      "epoch": 4.4525146484375e-05,
      "model_forward_time": 1.392711877822876,
      "step": 46688
    },
    {
      "epoch": 4.4525146484375e-05,
      "step": 46688,
      "training_step_time": 8.511618614196777
    },
    {
      "epoch": 4.4526100158691406e-05,
      "model_forward_time": 1.717839241027832,
      "step": 46689
    },
    {
      "epoch": 4.4526100158691406e-05,
      "step": 46689,
      "training_step_time": 8.952134132385254
    },
    {
      "epoch": 4.452705383300781e-05,
      "grad_norm": 0.14787638187408447,
      "learning_rate": 5.954928644249027e-06,
      "loss": 0.0565,
      "step": 46690
    },
    {
      "epoch": 4.452705383300781e-05,
      "model_forward_time": 1.7093403339385986,
      "step": 46690
    },
    {
      "dynamics_loss_avg": 0.05320542603731156,
      "epoch": 4.452705383300781e-05,
      "step": 46690
    },
    {
      "action_loss_avg": 0.0014952087687561289,
      "epoch": 4.452705383300781e-05,
      "step": 46690
    },
    {
      "epoch": 4.452705383300781e-05,
      "step": 46690,
      "training_step_time": 7.305159568786621
    },
    {
      "epoch": 4.452800750732422e-05,
      "model_forward_time": 1.7082059383392334,
      "step": 46691
    },
    {
      "epoch": 4.452800750732422e-05,
      "step": 46691,
      "training_step_time": 6.858397722244263
    },
    {
      "epoch": 4.4528961181640625e-05,
      "model_forward_time": 1.7154507637023926,
      "step": 46692
    },
    {
      "epoch": 4.4528961181640625e-05,
      "step": 46692,
      "training_step_time": 6.849794864654541
    },
    {
      "epoch": 4.452991485595703e-05,
      "model_forward_time": 1.7158348560333252,
      "step": 46693
    },
    {
      "epoch": 4.452991485595703e-05,
      "step": 46693,
      "training_step_time": 7.330379009246826
    },
    {
      "epoch": 4.453086853027344e-05,
      "model_forward_time": 1.7125568389892578,
      "step": 46694
    },
    {
      "epoch": 4.453086853027344e-05,
      "step": 46694,
      "training_step_time": 7.757217645645142
    },
    {
      "epoch": 4.4531822204589844e-05,
      "model_forward_time": 2.2350831031799316,
      "step": 46695
    },
    {
      "epoch": 4.4531822204589844e-05,
      "step": 46695,
      "training_step_time": 10.52617859840393
    },
    {
      "epoch": 4.453277587890625e-05,
      "model_forward_time": 1.7144312858581543,
      "step": 46696
    },
    {
      "epoch": 4.453277587890625e-05,
      "step": 46696,
      "training_step_time": 8.968742609024048
    },
    {
      "epoch": 4.453372955322266e-05,
      "model_forward_time": 1.7642707824707031,
      "step": 46697
    },
    {
      "epoch": 4.453372955322266e-05,
      "step": 46697,
      "training_step_time": 8.853433847427368
    },
    {
      "epoch": 4.453468322753906e-05,
      "model_forward_time": 1.7155320644378662,
      "step": 46698
    },
    {
      "epoch": 4.453468322753906e-05,
      "step": 46698,
      "training_step_time": 7.907590627670288
    },
    {
      "epoch": 4.453563690185547e-05,
      "model_forward_time": 1.718872308731079,
      "step": 46699
    },
    {
      "epoch": 4.453563690185547e-05,
      "step": 46699,
      "training_step_time": 7.748202085494995
    },
    {
      "epoch": 4.453659057617188e-05,
      "grad_norm": 0.10639094561338425,
      "learning_rate": 5.953305557973572e-06,
      "loss": 0.058,
      "step": 46700
    },
    {
      "epoch": 4.453659057617188e-05,
      "model_forward_time": 1.7167582511901855,
      "step": 46700
    },
    {
      "dynamics_loss_avg": 0.05345322266221046,
      "epoch": 4.453659057617188e-05,
      "step": 46700
    },
    {
      "action_loss_avg": 0.0012457007382181474,
      "epoch": 4.453659057617188e-05,
      "step": 46700
    },
    {
      "epoch": 4.453659057617188e-05,
      "step": 46700,
      "training_step_time": 6.903987407684326
    },
    {
      "epoch": 4.453754425048828e-05,
      "model_forward_time": 1.7105865478515625,
      "step": 46701
    },
    {
      "epoch": 4.453754425048828e-05,
      "step": 46701,
      "training_step_time": 6.875988006591797
    },
    {
      "epoch": 4.4538497924804686e-05,
      "model_forward_time": 1.3470308780670166,
      "step": 46702
    },
    {
      "epoch": 4.4538497924804686e-05,
      "step": 46702,
      "training_step_time": 7.29035210609436
    },
    {
      "epoch": 4.4539451599121097e-05,
      "model_forward_time": 1.7138047218322754,
      "step": 46703
    },
    {
      "epoch": 4.4539451599121097e-05,
      "step": 46703,
      "training_step_time": 8.297191143035889
    },
    {
      "epoch": 4.45404052734375e-05,
      "model_forward_time": 1.712003469467163,
      "step": 46704
    },
    {
      "epoch": 4.45404052734375e-05,
      "step": 46704,
      "training_step_time": 9.666169881820679
    },
    {
      "epoch": 4.4541358947753905e-05,
      "model_forward_time": 2.277663469314575,
      "step": 46705
    },
    {
      "epoch": 4.4541358947753905e-05,
      "step": 46705,
      "training_step_time": 8.870460271835327
    },
    {
      "epoch": 4.4542312622070315e-05,
      "model_forward_time": 1.7545990943908691,
      "step": 46706
    },
    {
      "epoch": 4.4542312622070315e-05,
      "step": 46706,
      "training_step_time": 8.416479349136353
    },
    {
      "epoch": 4.454326629638672e-05,
      "model_forward_time": 1.7253098487854004,
      "step": 46707
    },
    {
      "epoch": 4.454326629638672e-05,
      "step": 46707,
      "training_step_time": 7.9463818073272705
    },
    {
      "epoch": 4.4544219970703123e-05,
      "model_forward_time": 1.7139909267425537,
      "step": 46708
    },
    {
      "epoch": 4.4544219970703123e-05,
      "step": 46708,
      "training_step_time": 7.751063585281372
    },
    {
      "epoch": 4.4545173645019534e-05,
      "model_forward_time": 1.708838939666748,
      "step": 46709
    },
    {
      "epoch": 4.4545173645019534e-05,
      "step": 46709,
      "training_step_time": 7.249568223953247
    },
    {
      "epoch": 4.454612731933594e-05,
      "grad_norm": 0.14868482947349548,
      "learning_rate": 5.951682367446056e-06,
      "loss": 0.0605,
      "step": 46710
    },
    {
      "epoch": 4.454612731933594e-05,
      "model_forward_time": 1.7094814777374268,
      "step": 46710
    },
    {
      "dynamics_loss_avg": 0.06612162347882986,
      "epoch": 4.454612731933594e-05,
      "step": 46710
    },
    {
      "action_loss_avg": 0.0010447004111483692,
      "epoch": 4.454612731933594e-05,
      "step": 46710
    },
    {
      "epoch": 4.454612731933594e-05,
      "step": 46710,
      "training_step_time": 6.941743850708008
    },
    {
      "epoch": 4.454708099365234e-05,
      "model_forward_time": 1.7120997905731201,
      "step": 46711
    },
    {
      "epoch": 4.454708099365234e-05,
      "step": 46711,
      "training_step_time": 7.311680555343628
    },
    {
      "epoch": 4.454803466796875e-05,
      "model_forward_time": 1.7111265659332275,
      "step": 46712
    },
    {
      "epoch": 4.454803466796875e-05,
      "step": 46712,
      "training_step_time": 8.29784083366394
    },
    {
      "epoch": 4.454898834228516e-05,
      "model_forward_time": 1.7136943340301514,
      "step": 46713
    },
    {
      "epoch": 4.454898834228516e-05,
      "step": 46713,
      "training_step_time": 7.777844190597534
    },
    {
      "epoch": 4.454994201660156e-05,
      "model_forward_time": 2.270066261291504,
      "step": 46714
    },
    {
      "epoch": 4.454994201660156e-05,
      "step": 46714,
      "training_step_time": 10.710704565048218
    },
    {
      "epoch": 4.455089569091797e-05,
      "model_forward_time": 1.7378547191619873,
      "step": 46715
    },
    {
      "epoch": 4.455089569091797e-05,
      "step": 46715,
      "training_step_time": 9.555055856704712
    },
    {
      "epoch": 4.4551849365234376e-05,
      "model_forward_time": 1.7131285667419434,
      "step": 46716
    },
    {
      "epoch": 4.4551849365234376e-05,
      "step": 46716,
      "training_step_time": 8.385841131210327
    },
    {
      "epoch": 4.455280303955078e-05,
      "model_forward_time": 1.7166180610656738,
      "step": 46717
    },
    {
      "epoch": 4.455280303955078e-05,
      "step": 46717,
      "training_step_time": 7.705348968505859
    },
    {
      "epoch": 4.455375671386719e-05,
      "model_forward_time": 1.7111289501190186,
      "step": 46718
    },
    {
      "epoch": 4.455375671386719e-05,
      "step": 46718,
      "training_step_time": 7.218373537063599
    },
    {
      "epoch": 4.4554710388183595e-05,
      "model_forward_time": 1.7101728916168213,
      "step": 46719
    },
    {
      "epoch": 4.4554710388183595e-05,
      "step": 46719,
      "training_step_time": 6.820513010025024
    },
    {
      "epoch": 4.45556640625e-05,
      "grad_norm": 0.13169392943382263,
      "learning_rate": 5.950059072843988e-06,
      "loss": 0.0543,
      "step": 46720
    },
    {
      "epoch": 4.45556640625e-05,
      "model_forward_time": 1.7106540203094482,
      "step": 46720
    },
    {
      "dynamics_loss_avg": 0.067647585272789,
      "epoch": 4.45556640625e-05,
      "step": 46720
    },
    {
      "action_loss_avg": 0.0015288769092876465,
      "epoch": 4.45556640625e-05,
      "step": 46720
    },
    {
      "epoch": 4.45556640625e-05,
      "step": 46720,
      "training_step_time": 7.26745080947876
    },
    {
      "epoch": 4.45566177368164e-05,
      "model_forward_time": 1.7198219299316406,
      "step": 46721
    },
    {
      "epoch": 4.45566177368164e-05,
      "step": 46721,
      "training_step_time": 7.67217493057251
    },
    {
      "epoch": 4.4557571411132814e-05,
      "model_forward_time": 1.7143635749816895,
      "step": 46722
    },
    {
      "epoch": 4.4557571411132814e-05,
      "step": 46722,
      "training_step_time": 7.791622877120972
    },
    {
      "epoch": 4.455852508544922e-05,
      "model_forward_time": 2.2868144512176514,
      "step": 46723
    },
    {
      "epoch": 4.455852508544922e-05,
      "step": 46723,
      "training_step_time": 8.813013076782227
    },
    {
      "epoch": 4.455947875976562e-05,
      "model_forward_time": 1.7260994911193848,
      "step": 46724
    },
    {
      "epoch": 4.455947875976562e-05,
      "step": 46724,
      "training_step_time": 8.517918586730957
    },
    {
      "epoch": 4.456043243408203e-05,
      "model_forward_time": 1.7093262672424316,
      "step": 46725
    },
    {
      "epoch": 4.456043243408203e-05,
      "step": 46725,
      "training_step_time": 9.023537158966064
    },
    {
      "epoch": 4.456138610839844e-05,
      "model_forward_time": 1.710005760192871,
      "step": 46726
    },
    {
      "epoch": 4.456138610839844e-05,
      "step": 46726,
      "training_step_time": 9.228678464889526
    },
    {
      "epoch": 4.456233978271484e-05,
      "model_forward_time": 1.7130827903747559,
      "step": 46727
    },
    {
      "epoch": 4.456233978271484e-05,
      "step": 46727,
      "training_step_time": 7.536133527755737
    },
    {
      "epoch": 4.456329345703125e-05,
      "model_forward_time": 1.7166693210601807,
      "step": 46728
    },
    {
      "epoch": 4.456329345703125e-05,
      "step": 46728,
      "training_step_time": 7.2000274658203125
    },
    {
      "epoch": 4.4564247131347655e-05,
      "model_forward_time": 1.7104740142822266,
      "step": 46729
    },
    {
      "epoch": 4.4564247131347655e-05,
      "step": 46729,
      "training_step_time": 7.274246692657471
    },
    {
      "epoch": 4.456520080566406e-05,
      "grad_norm": 0.12917138636112213,
      "learning_rate": 5.948435674344886e-06,
      "loss": 0.0567,
      "step": 46730
    },
    {
      "epoch": 4.456520080566406e-05,
      "model_forward_time": 1.7039363384246826,
      "step": 46730
    },
    {
      "dynamics_loss_avg": 0.04819814218208194,
      "epoch": 4.456520080566406e-05,
      "step": 46730
    },
    {
      "action_loss_avg": 0.001155277481302619,
      "epoch": 4.456520080566406e-05,
      "step": 46730
    },
    {
      "epoch": 4.456520080566406e-05,
      "step": 46730,
      "training_step_time": 9.289442539215088
    },
    {
      "epoch": 4.456615447998047e-05,
      "model_forward_time": 1.7105441093444824,
      "step": 46731
    },
    {
      "epoch": 4.456615447998047e-05,
      "step": 46731,
      "training_step_time": 7.806493043899536
    },
    {
      "epoch": 4.4567108154296874e-05,
      "model_forward_time": 2.2659008502960205,
      "step": 46732
    },
    {
      "epoch": 4.4567108154296874e-05,
      "step": 46732,
      "training_step_time": 8.332060098648071
    },
    {
      "epoch": 4.456806182861328e-05,
      "model_forward_time": 1.716644525527954,
      "step": 46733
    },
    {
      "epoch": 4.456806182861328e-05,
      "step": 46733,
      "training_step_time": 7.728364944458008
    },
    {
      "epoch": 4.456901550292969e-05,
      "model_forward_time": 1.7600090503692627,
      "step": 46734
    },
    {
      "epoch": 4.456901550292969e-05,
      "step": 46734,
      "training_step_time": 8.538221597671509
    },
    {
      "epoch": 4.456996917724609e-05,
      "model_forward_time": 1.71525239944458,
      "step": 46735
    },
    {
      "epoch": 4.456996917724609e-05,
      "step": 46735,
      "training_step_time": 7.791577339172363
    },
    {
      "epoch": 4.45709228515625e-05,
      "model_forward_time": 1.716142177581787,
      "step": 46736
    },
    {
      "epoch": 4.45709228515625e-05,
      "step": 46736,
      "training_step_time": 7.552281141281128
    },
    {
      "epoch": 4.457187652587891e-05,
      "model_forward_time": 1.716130256652832,
      "step": 46737
    },
    {
      "epoch": 4.457187652587891e-05,
      "step": 46737,
      "training_step_time": 7.157069444656372
    },
    {
      "epoch": 4.457283020019531e-05,
      "model_forward_time": 1.7102644443511963,
      "step": 46738
    },
    {
      "epoch": 4.457283020019531e-05,
      "step": 46738,
      "training_step_time": 7.239725589752197
    },
    {
      "epoch": 4.4573783874511716e-05,
      "model_forward_time": 1.718168020248413,
      "step": 46739
    },
    {
      "epoch": 4.4573783874511716e-05,
      "step": 46739,
      "training_step_time": 7.730174541473389
    },
    {
      "epoch": 4.457473754882813e-05,
      "grad_norm": 0.11825016885995865,
      "learning_rate": 5.946812172126285e-06,
      "loss": 0.0571,
      "step": 46740
    },
    {
      "epoch": 4.457473754882813e-05,
      "model_forward_time": 1.7071244716644287,
      "step": 46740
    },
    {
      "dynamics_loss_avg": 0.05616940185427666,
      "epoch": 4.457473754882813e-05,
      "step": 46740
    },
    {
      "action_loss_avg": 0.0012823859840864315,
      "epoch": 4.457473754882813e-05,
      "step": 46740
    },
    {
      "epoch": 4.457473754882813e-05,
      "step": 46740,
      "training_step_time": 7.7535481452941895
    },
    {
      "epoch": 4.457569122314453e-05,
      "model_forward_time": 1.7128520011901855,
      "step": 46741
    },
    {
      "epoch": 4.457569122314453e-05,
      "step": 46741,
      "training_step_time": 11.42074966430664
    },
    {
      "epoch": 4.4576644897460935e-05,
      "model_forward_time": 2.257441282272339,
      "step": 46742
    },
    {
      "epoch": 4.4576644897460935e-05,
      "step": 46742,
      "training_step_time": 7.771345138549805
    },
    {
      "epoch": 4.4577598571777346e-05,
      "model_forward_time": 1.7603552341461182,
      "step": 46743
    },
    {
      "epoch": 4.4577598571777346e-05,
      "step": 46743,
      "training_step_time": 9.131871938705444
    },
    {
      "epoch": 4.457855224609375e-05,
      "model_forward_time": 1.7177462577819824,
      "step": 46744
    },
    {
      "epoch": 4.457855224609375e-05,
      "step": 46744,
      "training_step_time": 8.617465496063232
    },
    {
      "epoch": 4.4579505920410154e-05,
      "model_forward_time": 1.7184135913848877,
      "step": 46745
    },
    {
      "epoch": 4.4579505920410154e-05,
      "step": 46745,
      "training_step_time": 7.462243556976318
    },
    {
      "epoch": 4.4580459594726565e-05,
      "model_forward_time": 1.7082488536834717,
      "step": 46746
    },
    {
      "epoch": 4.4580459594726565e-05,
      "step": 46746,
      "training_step_time": 6.998907804489136
    },
    {
      "epoch": 4.458141326904297e-05,
      "model_forward_time": 1.7118854522705078,
      "step": 46747
    },
    {
      "epoch": 4.458141326904297e-05,
      "step": 46747,
      "training_step_time": 7.139826536178589
    },
    {
      "epoch": 4.458236694335937e-05,
      "model_forward_time": 1.7173357009887695,
      "step": 46748
    },
    {
      "epoch": 4.458236694335937e-05,
      "step": 46748,
      "training_step_time": 7.618145704269409
    },
    {
      "epoch": 4.4583320617675783e-05,
      "model_forward_time": 1.7187063694000244,
      "step": 46749
    },
    {
      "epoch": 4.4583320617675783e-05,
      "step": 46749,
      "training_step_time": 7.622754335403442
    },
    {
      "epoch": 4.458427429199219e-05,
      "grad_norm": 0.10842675715684891,
      "learning_rate": 5.945188566365729e-06,
      "loss": 0.0603,
      "step": 46750
    },
    {
      "epoch": 4.458427429199219e-05,
      "model_forward_time": 1.7105755805969238,
      "step": 46750
    },
    {
      "dynamics_loss_avg": 0.04720853660255671,
      "epoch": 4.458427429199219e-05,
      "step": 46750
    },
    {
      "action_loss_avg": 0.0010493043839232997,
      "epoch": 4.458427429199219e-05,
      "step": 46750
    },
    {
      "epoch": 4.458427429199219e-05,
      "step": 46750,
      "training_step_time": 9.468217372894287
    },
    {
      "epoch": 4.458522796630859e-05,
      "model_forward_time": 2.2899718284606934,
      "step": 46751
    },
    {
      "epoch": 4.458522796630859e-05,
      "step": 46751,
      "training_step_time": 7.611043214797974
    },
    {
      "epoch": 4.4586181640625e-05,
      "model_forward_time": 1.776564359664917,
      "step": 46752
    },
    {
      "epoch": 4.4586181640625e-05,
      "step": 46752,
      "training_step_time": 8.334035396575928
    },
    {
      "epoch": 4.4587135314941406e-05,
      "model_forward_time": 1.715374231338501,
      "step": 46753
    },
    {
      "epoch": 4.4587135314941406e-05,
      "step": 46753,
      "training_step_time": 7.783002853393555
    },
    {
      "epoch": 4.458808898925781e-05,
      "model_forward_time": 1.7181971073150635,
      "step": 46754
    },
    {
      "epoch": 4.458808898925781e-05,
      "step": 46754,
      "training_step_time": 7.712978363037109
    },
    {
      "epoch": 4.458904266357422e-05,
      "model_forward_time": 1.7099120616912842,
      "step": 46755
    },
    {
      "epoch": 4.458904266357422e-05,
      "step": 46755,
      "training_step_time": 7.579854249954224
    },
    {
      "epoch": 4.4589996337890625e-05,
      "model_forward_time": 1.7120966911315918,
      "step": 46756
    },
    {
      "epoch": 4.4589996337890625e-05,
      "step": 46756,
      "training_step_time": 7.1734938621521
    },
    {
      "epoch": 4.459095001220703e-05,
      "model_forward_time": 1.719618320465088,
      "step": 46757
    },
    {
      "epoch": 4.459095001220703e-05,
      "step": 46757,
      "training_step_time": 7.66740083694458
    },
    {
      "epoch": 4.459190368652344e-05,
      "model_forward_time": 1.7039844989776611,
      "step": 46758
    },
    {
      "epoch": 4.459190368652344e-05,
      "step": 46758,
      "training_step_time": 7.542053699493408
    },
    {
      "epoch": 4.4592857360839844e-05,
      "model_forward_time": 1.7095451354980469,
      "step": 46759
    },
    {
      "epoch": 4.4592857360839844e-05,
      "step": 46759,
      "training_step_time": 7.587115526199341
    },
    {
      "epoch": 4.459381103515625e-05,
      "grad_norm": 0.13750380277633667,
      "learning_rate": 5.943564857240773e-06,
      "loss": 0.0579,
      "step": 46760
    },
    {
      "epoch": 4.459381103515625e-05,
      "model_forward_time": 1.8894710540771484,
      "step": 46760
    },
    {
      "dynamics_loss_avg": 0.059075060300529,
      "epoch": 4.459381103515625e-05,
      "step": 46760
    },
    {
      "action_loss_avg": 0.0011208156443899497,
      "epoch": 4.459381103515625e-05,
      "step": 46760
    },
    {
      "epoch": 4.459381103515625e-05,
      "step": 46760,
      "training_step_time": 7.620316505432129
    },
    {
      "epoch": 4.459476470947266e-05,
      "model_forward_time": 1.7575559616088867,
      "step": 46761
    },
    {
      "epoch": 4.459476470947266e-05,
      "step": 46761,
      "training_step_time": 9.235637664794922
    },
    {
      "epoch": 4.459571838378906e-05,
      "model_forward_time": 1.713914155960083,
      "step": 46762
    },
    {
      "epoch": 4.459571838378906e-05,
      "step": 46762,
      "training_step_time": 8.288438558578491
    },
    {
      "epoch": 4.459667205810547e-05,
      "model_forward_time": 1.7115569114685059,
      "step": 46763
    },
    {
      "epoch": 4.459667205810547e-05,
      "step": 46763,
      "training_step_time": 7.694746494293213
    },
    {
      "epoch": 4.459762573242188e-05,
      "model_forward_time": 1.713719129562378,
      "step": 46764
    },
    {
      "epoch": 4.459762573242188e-05,
      "step": 46764,
      "training_step_time": 7.596350908279419
    },
    {
      "epoch": 4.459857940673828e-05,
      "model_forward_time": 1.712477445602417,
      "step": 46765
    },
    {
      "epoch": 4.459857940673828e-05,
      "step": 46765,
      "training_step_time": 7.188741445541382
    },
    {
      "epoch": 4.4599533081054686e-05,
      "model_forward_time": 1.7186431884765625,
      "step": 46766
    },
    {
      "epoch": 4.4599533081054686e-05,
      "step": 46766,
      "training_step_time": 7.6221764087677
    },
    {
      "epoch": 4.46004867553711e-05,
      "model_forward_time": 1.7110581398010254,
      "step": 46767
    },
    {
      "epoch": 4.46004867553711e-05,
      "step": 46767,
      "training_step_time": 7.394128799438477
    },
    {
      "epoch": 4.46014404296875e-05,
      "model_forward_time": 1.7062304019927979,
      "step": 46768
    },
    {
      "epoch": 4.46014404296875e-05,
      "step": 46768,
      "training_step_time": 8.459033727645874
    },
    {
      "epoch": 4.4602394104003905e-05,
      "model_forward_time": 2.290494441986084,
      "step": 46769
    },
    {
      "epoch": 4.4602394104003905e-05,
      "step": 46769,
      "training_step_time": 8.537294387817383
    },
    {
      "epoch": 4.4603347778320315e-05,
      "grad_norm": 0.1371290385723114,
      "learning_rate": 5.941941044928982e-06,
      "loss": 0.0601,
      "step": 46770
    },
    {
      "epoch": 4.4603347778320315e-05,
      "model_forward_time": 1.7632803916931152,
      "step": 46770
    },
    {
      "dynamics_loss_avg": 0.055859333463013174,
      "epoch": 4.4603347778320315e-05,
      "step": 46770
    },
    {
      "action_loss_avg": 0.0008892166399164126,
      "epoch": 4.4603347778320315e-05,
      "step": 46770
    },
    {
      "epoch": 4.4603347778320315e-05,
      "step": 46770,
      "training_step_time": 7.5800957679748535
    },
    {
      "epoch": 4.460430145263672e-05,
      "model_forward_time": 1.7269456386566162,
      "step": 46771
    },
    {
      "epoch": 4.460430145263672e-05,
      "step": 46771,
      "training_step_time": 8.520380973815918
    },
    {
      "epoch": 4.4605255126953124e-05,
      "model_forward_time": 1.720501184463501,
      "step": 46772
    },
    {
      "epoch": 4.4605255126953124e-05,
      "step": 46772,
      "training_step_time": 7.765426397323608
    },
    {
      "epoch": 4.4606208801269534e-05,
      "model_forward_time": 1.7156646251678467,
      "step": 46773
    },
    {
      "epoch": 4.4606208801269534e-05,
      "step": 46773,
      "training_step_time": 7.740865468978882
    },
    {
      "epoch": 4.460716247558594e-05,
      "model_forward_time": 1.7179639339447021,
      "step": 46774
    },
    {
      "epoch": 4.460716247558594e-05,
      "step": 46774,
      "training_step_time": 8.346577167510986
    },
    {
      "epoch": 4.460811614990234e-05,
      "model_forward_time": 1.3439409732818604,
      "step": 46775
    },
    {
      "epoch": 4.460811614990234e-05,
      "step": 46775,
      "training_step_time": 7.770370006561279
    },
    {
      "epoch": 4.460906982421875e-05,
      "model_forward_time": 1.7207021713256836,
      "step": 46776
    },
    {
      "epoch": 4.460906982421875e-05,
      "step": 46776,
      "training_step_time": 8.532861948013306
    },
    {
      "epoch": 4.461002349853516e-05,
      "model_forward_time": 1.7207953929901123,
      "step": 46777
    },
    {
      "epoch": 4.461002349853516e-05,
      "step": 46777,
      "training_step_time": 7.916007995605469
    },
    {
      "epoch": 4.461097717285156e-05,
      "model_forward_time": 1.7156367301940918,
      "step": 46778
    },
    {
      "epoch": 4.461097717285156e-05,
      "step": 46778,
      "training_step_time": 9.309112310409546
    },
    {
      "epoch": 4.461193084716797e-05,
      "model_forward_time": 2.2881596088409424,
      "step": 46779
    },
    {
      "epoch": 4.461193084716797e-05,
      "step": 46779,
      "training_step_time": 7.87527060508728
    },
    {
      "epoch": 4.4612884521484376e-05,
      "grad_norm": 0.12044698745012283,
      "learning_rate": 5.940317129607935e-06,
      "loss": 0.0631,
      "step": 46780
    },
    {
      "epoch": 4.4612884521484376e-05,
      "model_forward_time": 1.6975200176239014,
      "step": 46780
    },
    {
      "dynamics_loss_avg": 0.06312510259449482,
      "epoch": 4.4612884521484376e-05,
      "step": 46780
    },
    {
      "action_loss_avg": 0.0014695384772494436,
      "epoch": 4.4612884521484376e-05,
      "step": 46780
    },
    {
      "epoch": 4.4612884521484376e-05,
      "step": 46780,
      "training_step_time": 7.71683669090271
    },
    {
      "epoch": 4.461383819580078e-05,
      "model_forward_time": 1.7214670181274414,
      "step": 46781
    },
    {
      "epoch": 4.461383819580078e-05,
      "step": 46781,
      "training_step_time": 7.72123384475708
    },
    {
      "epoch": 4.4614791870117184e-05,
      "model_forward_time": 1.7172832489013672,
      "step": 46782
    },
    {
      "epoch": 4.4614791870117184e-05,
      "step": 46782,
      "training_step_time": 7.616188049316406
    },
    {
      "epoch": 4.4615745544433595e-05,
      "model_forward_time": 1.7107248306274414,
      "step": 46783
    },
    {
      "epoch": 4.4615745544433595e-05,
      "step": 46783,
      "training_step_time": 7.21688175201416
    },
    {
      "epoch": 4.461669921875e-05,
      "model_forward_time": 1.71352219581604,
      "step": 46784
    },
    {
      "epoch": 4.461669921875e-05,
      "step": 46784,
      "training_step_time": 7.604938983917236
    },
    {
      "epoch": 4.46176528930664e-05,
      "model_forward_time": 1.7160043716430664,
      "step": 46785
    },
    {
      "epoch": 4.46176528930664e-05,
      "step": 46785,
      "training_step_time": 7.988242149353027
    },
    {
      "epoch": 4.4618606567382814e-05,
      "model_forward_time": 1.7165393829345703,
      "step": 46786
    },
    {
      "epoch": 4.4618606567382814e-05,
      "step": 46786,
      "training_step_time": 7.686146259307861
    },
    {
      "epoch": 4.461956024169922e-05,
      "model_forward_time": 1.7117531299591064,
      "step": 46787
    },
    {
      "epoch": 4.461956024169922e-05,
      "step": 46787,
      "training_step_time": 8.64042615890503
    },
    {
      "epoch": 4.462051391601562e-05,
      "model_forward_time": 2.2700858116149902,
      "step": 46788
    },
    {
      "epoch": 4.462051391601562e-05,
      "step": 46788,
      "training_step_time": 7.6880505084991455
    },
    {
      "epoch": 4.462146759033203e-05,
      "model_forward_time": 1.7156422138214111,
      "step": 46789
    },
    {
      "epoch": 4.462146759033203e-05,
      "step": 46789,
      "training_step_time": 8.601062774658203
    },
    {
      "epoch": 4.462242126464844e-05,
      "grad_norm": 0.16666433215141296,
      "learning_rate": 5.938693111455221e-06,
      "loss": 0.0603,
      "step": 46790
    },
    {
      "epoch": 4.462242126464844e-05,
      "model_forward_time": 1.7064967155456543,
      "step": 46790
    },
    {
      "dynamics_loss_avg": 0.04451285693794489,
      "epoch": 4.462242126464844e-05,
      "step": 46790
    },
    {
      "action_loss_avg": 0.0012962614331627265,
      "epoch": 4.462242126464844e-05,
      "step": 46790
    },
    {
      "epoch": 4.462242126464844e-05,
      "step": 46790,
      "training_step_time": 10.201920986175537
    },
    {
      "epoch": 4.462337493896484e-05,
      "model_forward_time": 1.71531081199646,
      "step": 46791
    },
    {
      "epoch": 4.462337493896484e-05,
      "step": 46791,
      "training_step_time": 8.57346487045288
    },
    {
      "epoch": 4.462432861328125e-05,
      "model_forward_time": 1.7085323333740234,
      "step": 46792
    },
    {
      "epoch": 4.462432861328125e-05,
      "step": 46792,
      "training_step_time": 8.258195877075195
    },
    {
      "epoch": 4.4625282287597656e-05,
      "model_forward_time": 1.7144641876220703,
      "step": 46793
    },
    {
      "epoch": 4.4625282287597656e-05,
      "step": 46793,
      "training_step_time": 6.702030658721924
    },
    {
      "epoch": 4.462623596191406e-05,
      "model_forward_time": 1.724064826965332,
      "step": 46794
    },
    {
      "epoch": 4.462623596191406e-05,
      "step": 46794,
      "training_step_time": 7.582309007644653
    },
    {
      "epoch": 4.462718963623047e-05,
      "model_forward_time": 1.7108774185180664,
      "step": 46795
    },
    {
      "epoch": 4.462718963623047e-05,
      "step": 46795,
      "training_step_time": 8.384349822998047
    },
    {
      "epoch": 4.4628143310546874e-05,
      "model_forward_time": 1.7093491554260254,
      "step": 46796
    },
    {
      "epoch": 4.4628143310546874e-05,
      "step": 46796,
      "training_step_time": 8.992996215820312
    },
    {
      "epoch": 4.462909698486328e-05,
      "model_forward_time": 2.275750160217285,
      "step": 46797
    },
    {
      "epoch": 4.462909698486328e-05,
      "step": 46797,
      "training_step_time": 8.192052125930786
    },
    {
      "epoch": 4.463005065917969e-05,
      "model_forward_time": 1.7092962265014648,
      "step": 46798
    },
    {
      "epoch": 4.463005065917969e-05,
      "step": 46798,
      "training_step_time": 8.248447895050049
    },
    {
      "epoch": 4.463100433349609e-05,
      "model_forward_time": 1.7153563499450684,
      "step": 46799
    },
    {
      "epoch": 4.463100433349609e-05,
      "step": 46799,
      "training_step_time": 7.9176506996154785
    },
    {
      "epoch": 4.46319580078125e-05,
      "grad_norm": 0.19436602294445038,
      "learning_rate": 5.937068990648438e-06,
      "loss": 0.0608,
      "step": 46800
    },
    {
      "epoch": 4.46319580078125e-05,
      "model_forward_time": 1.7056756019592285,
      "step": 46800
    },
    {
      "dynamics_loss_avg": 0.05201813080348074,
      "epoch": 4.46319580078125e-05,
      "step": 46800
    },
    {
      "action_loss_avg": 0.0012375743099255487,
      "epoch": 4.46319580078125e-05,
      "step": 46800
    },
    {
      "epoch": 4.46319580078125e-05,
      "step": 46800,
      "training_step_time": 7.528254985809326
    },
    {
      "epoch": 4.463291168212891e-05,
      "model_forward_time": 1.713909387588501,
      "step": 46801
    },
    {
      "epoch": 4.463291168212891e-05,
      "step": 46801,
      "training_step_time": 7.63409948348999
    },
    {
      "epoch": 4.463386535644531e-05,
      "model_forward_time": 1.7180049419403076,
      "step": 46802
    },
    {
      "epoch": 4.463386535644531e-05,
      "step": 46802,
      "training_step_time": 7.206355571746826
    },
    {
      "epoch": 4.4634819030761716e-05,
      "model_forward_time": 1.7064392566680908,
      "step": 46803
    },
    {
      "epoch": 4.4634819030761716e-05,
      "step": 46803,
      "training_step_time": 7.604510068893433
    },
    {
      "epoch": 4.463577270507813e-05,
      "model_forward_time": 1.7119710445404053,
      "step": 46804
    },
    {
      "epoch": 4.463577270507813e-05,
      "step": 46804,
      "training_step_time": 8.639209032058716
    },
    {
      "epoch": 4.463672637939453e-05,
      "model_forward_time": 1.7134687900543213,
      "step": 46805
    },
    {
      "epoch": 4.463672637939453e-05,
      "step": 46805,
      "training_step_time": 8.429653406143188
    },
    {
      "epoch": 4.4637680053710935e-05,
      "model_forward_time": 2.2978780269622803,
      "step": 46806
    },
    {
      "epoch": 4.4637680053710935e-05,
      "step": 46806,
      "training_step_time": 8.652473211288452
    },
    {
      "epoch": 4.4638633728027346e-05,
      "model_forward_time": 1.7152049541473389,
      "step": 46807
    },
    {
      "epoch": 4.4638633728027346e-05,
      "step": 46807,
      "training_step_time": 8.180337190628052
    },
    {
      "epoch": 4.463958740234375e-05,
      "model_forward_time": 1.7127597332000732,
      "step": 46808
    },
    {
      "epoch": 4.463958740234375e-05,
      "step": 46808,
      "training_step_time": 9.635592222213745
    },
    {
      "epoch": 4.4640541076660154e-05,
      "model_forward_time": 1.7183277606964111,
      "step": 46809
    },
    {
      "epoch": 4.4640541076660154e-05,
      "step": 46809,
      "training_step_time": 8.306695699691772
    },
    {
      "epoch": 4.4641494750976565e-05,
      "grad_norm": 0.13005642592906952,
      "learning_rate": 5.935444767365199e-06,
      "loss": 0.0623,
      "step": 46810
    },
    {
      "epoch": 4.4641494750976565e-05,
      "model_forward_time": 1.7464115619659424,
      "step": 46810
    },
    {
      "dynamics_loss_avg": 0.05820307806134224,
      "epoch": 4.4641494750976565e-05,
      "step": 46810
    },
    {
      "action_loss_avg": 0.0017765626587788574,
      "epoch": 4.4641494750976565e-05,
      "step": 46810
    },
    {
      "epoch": 4.4641494750976565e-05,
      "step": 46810,
      "training_step_time": 7.74731707572937
    },
    {
      "epoch": 4.464244842529297e-05,
      "model_forward_time": 1.7151031494140625,
      "step": 46811
    },
    {
      "epoch": 4.464244842529297e-05,
      "step": 46811,
      "training_step_time": 7.1968138217926025
    },
    {
      "epoch": 4.464340209960937e-05,
      "model_forward_time": 1.711329698562622,
      "step": 46812
    },
    {
      "epoch": 4.464340209960937e-05,
      "step": 46812,
      "training_step_time": 7.839476585388184
    },
    {
      "epoch": 4.4644355773925784e-05,
      "model_forward_time": 1.7109949588775635,
      "step": 46813
    },
    {
      "epoch": 4.4644355773925784e-05,
      "step": 46813,
      "training_step_time": 8.350788831710815
    },
    {
      "epoch": 4.464530944824219e-05,
      "model_forward_time": 1.7249078750610352,
      "step": 46814
    },
    {
      "epoch": 4.464530944824219e-05,
      "step": 46814,
      "training_step_time": 7.81174373626709
    },
    {
      "epoch": 4.464626312255859e-05,
      "model_forward_time": 2.2875607013702393,
      "step": 46815
    },
    {
      "epoch": 4.464626312255859e-05,
      "step": 46815,
      "training_step_time": 7.791739463806152
    },
    {
      "epoch": 4.4647216796875e-05,
      "model_forward_time": 1.711183786392212,
      "step": 46816
    },
    {
      "epoch": 4.4647216796875e-05,
      "step": 46816,
      "training_step_time": 8.692107915878296
    },
    {
      "epoch": 4.4648170471191406e-05,
      "model_forward_time": 1.717320442199707,
      "step": 46817
    },
    {
      "epoch": 4.4648170471191406e-05,
      "step": 46817,
      "training_step_time": 7.9466636180877686
    },
    {
      "epoch": 4.464912414550781e-05,
      "model_forward_time": 1.7175869941711426,
      "step": 46818
    },
    {
      "epoch": 4.464912414550781e-05,
      "step": 46818,
      "training_step_time": 7.766262054443359
    },
    {
      "epoch": 4.465007781982422e-05,
      "model_forward_time": 1.7197771072387695,
      "step": 46819
    },
    {
      "epoch": 4.465007781982422e-05,
      "step": 46819,
      "training_step_time": 7.761418581008911
    },
    {
      "epoch": 4.4651031494140625e-05,
      "grad_norm": 0.14064233005046844,
      "learning_rate": 5.933820441783129e-06,
      "loss": 0.0643,
      "step": 46820
    },
    {
      "epoch": 4.4651031494140625e-05,
      "model_forward_time": 1.7093281745910645,
      "step": 46820
    },
    {
      "dynamics_loss_avg": 0.05113839395344257,
      "epoch": 4.4651031494140625e-05,
      "step": 46820
    },
    {
      "action_loss_avg": 0.0012458072102162986,
      "epoch": 4.4651031494140625e-05,
      "step": 46820
    },
    {
      "epoch": 4.4651031494140625e-05,
      "step": 46820,
      "training_step_time": 7.335894346237183
    },
    {
      "epoch": 4.465198516845703e-05,
      "model_forward_time": 1.713078260421753,
      "step": 46821
    },
    {
      "epoch": 4.465198516845703e-05,
      "step": 46821,
      "training_step_time": 7.820906162261963
    },
    {
      "epoch": 4.465293884277344e-05,
      "model_forward_time": 1.7139625549316406,
      "step": 46822
    },
    {
      "epoch": 4.465293884277344e-05,
      "step": 46822,
      "training_step_time": 7.504043817520142
    },
    {
      "epoch": 4.4653892517089844e-05,
      "model_forward_time": 1.7167408466339111,
      "step": 46823
    },
    {
      "epoch": 4.4653892517089844e-05,
      "step": 46823,
      "training_step_time": 7.947235584259033
    },
    {
      "epoch": 4.465484619140625e-05,
      "model_forward_time": 1.7420949935913086,
      "step": 46824
    },
    {
      "epoch": 4.465484619140625e-05,
      "step": 46824,
      "training_step_time": 8.365302085876465
    },
    {
      "epoch": 4.465579986572266e-05,
      "model_forward_time": 1.722405195236206,
      "step": 46825
    },
    {
      "epoch": 4.465579986572266e-05,
      "step": 46825,
      "training_step_time": 8.719891786575317
    },
    {
      "epoch": 4.465675354003906e-05,
      "model_forward_time": 1.7116003036499023,
      "step": 46826
    },
    {
      "epoch": 4.465675354003906e-05,
      "step": 46826,
      "training_step_time": 8.416716814041138
    },
    {
      "epoch": 4.465770721435547e-05,
      "model_forward_time": 1.7182528972625732,
      "step": 46827
    },
    {
      "epoch": 4.465770721435547e-05,
      "step": 46827,
      "training_step_time": 9.00801157951355
    },
    {
      "epoch": 4.465866088867188e-05,
      "model_forward_time": 1.7129623889923096,
      "step": 46828
    },
    {
      "epoch": 4.465866088867188e-05,
      "step": 46828,
      "training_step_time": 7.823164224624634
    },
    {
      "epoch": 4.465961456298828e-05,
      "model_forward_time": 1.7095506191253662,
      "step": 46829
    },
    {
      "epoch": 4.465961456298828e-05,
      "step": 46829,
      "training_step_time": 7.365682363510132
    },
    {
      "epoch": 4.4660568237304686e-05,
      "grad_norm": 0.08700820803642273,
      "learning_rate": 5.932196014079857e-06,
      "loss": 0.0617,
      "step": 46830
    },
    {
      "epoch": 4.4660568237304686e-05,
      "model_forward_time": 1.7158184051513672,
      "step": 46830
    },
    {
      "dynamics_loss_avg": 0.05514551140367985,
      "epoch": 4.4660568237304686e-05,
      "step": 46830
    },
    {
      "action_loss_avg": 0.0011837372614536435,
      "epoch": 4.4660568237304686e-05,
      "step": 46830
    },
    {
      "epoch": 4.4660568237304686e-05,
      "step": 46830,
      "training_step_time": 7.880041837692261
    },
    {
      "epoch": 4.46615219116211e-05,
      "model_forward_time": 1.7079625129699707,
      "step": 46831
    },
    {
      "epoch": 4.46615219116211e-05,
      "step": 46831,
      "training_step_time": 7.62270712852478
    },
    {
      "epoch": 4.46624755859375e-05,
      "model_forward_time": 1.7130639553070068,
      "step": 46832
    },
    {
      "epoch": 4.46624755859375e-05,
      "step": 46832,
      "training_step_time": 9.33397912979126
    },
    {
      "epoch": 4.4663429260253905e-05,
      "model_forward_time": 1.7197887897491455,
      "step": 46833
    },
    {
      "epoch": 4.4663429260253905e-05,
      "step": 46833,
      "training_step_time": 8.885371685028076
    },
    {
      "epoch": 4.4664382934570316e-05,
      "model_forward_time": 2.3141701221466064,
      "step": 46834
    },
    {
      "epoch": 4.4664382934570316e-05,
      "step": 46834,
      "training_step_time": 8.416653156280518
    },
    {
      "epoch": 4.466533660888672e-05,
      "model_forward_time": 1.7072641849517822,
      "step": 46835
    },
    {
      "epoch": 4.466533660888672e-05,
      "step": 46835,
      "training_step_time": 8.608537197113037
    },
    {
      "epoch": 4.4666290283203124e-05,
      "model_forward_time": 1.7107203006744385,
      "step": 46836
    },
    {
      "epoch": 4.4666290283203124e-05,
      "step": 46836,
      "training_step_time": 8.46784496307373
    },
    {
      "epoch": 4.4667243957519534e-05,
      "model_forward_time": 1.7116584777832031,
      "step": 46837
    },
    {
      "epoch": 4.4667243957519534e-05,
      "step": 46837,
      "training_step_time": 8.734748363494873
    },
    {
      "epoch": 4.466819763183594e-05,
      "model_forward_time": 1.7134106159210205,
      "step": 46838
    },
    {
      "epoch": 4.466819763183594e-05,
      "step": 46838,
      "training_step_time": 8.378610372543335
    },
    {
      "epoch": 4.466915130615234e-05,
      "model_forward_time": 1.7098519802093506,
      "step": 46839
    },
    {
      "epoch": 4.466915130615234e-05,
      "step": 46839,
      "training_step_time": 7.849306344985962
    },
    {
      "epoch": 4.467010498046875e-05,
      "grad_norm": 0.14279988408088684,
      "learning_rate": 5.930571484433032e-06,
      "loss": 0.0605,
      "step": 46840
    },
    {
      "epoch": 4.467010498046875e-05,
      "model_forward_time": 1.7133500576019287,
      "step": 46840
    },
    {
      "dynamics_loss_avg": 0.05870405379682779,
      "epoch": 4.467010498046875e-05,
      "step": 46840
    },
    {
      "action_loss_avg": 0.0012716351629933342,
      "epoch": 4.467010498046875e-05,
      "step": 46840
    },
    {
      "epoch": 4.467010498046875e-05,
      "step": 46840,
      "training_step_time": 7.524967670440674
    },
    {
      "epoch": 4.467105865478516e-05,
      "model_forward_time": 1.337019920349121,
      "step": 46841
    },
    {
      "epoch": 4.467105865478516e-05,
      "step": 46841,
      "training_step_time": 7.926453113555908
    },
    {
      "epoch": 4.467201232910156e-05,
      "model_forward_time": 1.7178144454956055,
      "step": 46842
    },
    {
      "epoch": 4.467201232910156e-05,
      "step": 46842,
      "training_step_time": 9.442535638809204
    },
    {
      "epoch": 4.467296600341797e-05,
      "model_forward_time": 2.3256332874298096,
      "step": 46843
    },
    {
      "epoch": 4.467296600341797e-05,
      "step": 46843,
      "training_step_time": 8.683613777160645
    },
    {
      "epoch": 4.4673919677734376e-05,
      "model_forward_time": 1.716609001159668,
      "step": 46844
    },
    {
      "epoch": 4.4673919677734376e-05,
      "step": 46844,
      "training_step_time": 8.73024320602417
    },
    {
      "epoch": 4.467487335205078e-05,
      "model_forward_time": 1.711620569229126,
      "step": 46845
    },
    {
      "epoch": 4.467487335205078e-05,
      "step": 46845,
      "training_step_time": 7.816827774047852
    },
    {
      "epoch": 4.4675827026367184e-05,
      "model_forward_time": 1.7146127223968506,
      "step": 46846
    },
    {
      "epoch": 4.4675827026367184e-05,
      "step": 46846,
      "training_step_time": 8.688370704650879
    },
    {
      "epoch": 4.4676780700683595e-05,
      "model_forward_time": 1.715132474899292,
      "step": 46847
    },
    {
      "epoch": 4.4676780700683595e-05,
      "step": 46847,
      "training_step_time": 7.665682554244995
    },
    {
      "epoch": 4.4677734375e-05,
      "model_forward_time": 1.7244069576263428,
      "step": 46848
    },
    {
      "epoch": 4.4677734375e-05,
      "step": 46848,
      "training_step_time": 7.09327507019043
    },
    {
      "epoch": 4.46786880493164e-05,
      "model_forward_time": 1.7123234272003174,
      "step": 46849
    },
    {
      "epoch": 4.46786880493164e-05,
      "step": 46849,
      "training_step_time": 7.298238039016724
    },
    {
      "epoch": 4.4679641723632814e-05,
      "grad_norm": 0.13583937287330627,
      "learning_rate": 5.928946853020309e-06,
      "loss": 0.0614,
      "step": 46850
    },
    {
      "epoch": 4.4679641723632814e-05,
      "model_forward_time": 1.7080447673797607,
      "step": 46850
    },
    {
      "dynamics_loss_avg": 0.05849893372505903,
      "epoch": 4.4679641723632814e-05,
      "step": 46850
    },
    {
      "action_loss_avg": 0.0014720700011821464,
      "epoch": 4.4679641723632814e-05,
      "step": 46850
    },
    {
      "epoch": 4.4679641723632814e-05,
      "step": 46850,
      "training_step_time": 7.746977806091309
    },
    {
      "epoch": 4.468059539794922e-05,
      "model_forward_time": 1.7106306552886963,
      "step": 46851
    },
    {
      "epoch": 4.468059539794922e-05,
      "step": 46851,
      "training_step_time": 7.559151649475098
    },
    {
      "epoch": 4.468154907226562e-05,
      "model_forward_time": 2.3311216831207275,
      "step": 46852
    },
    {
      "epoch": 4.468154907226562e-05,
      "step": 46852,
      "training_step_time": 8.654812812805176
    },
    {
      "epoch": 4.468250274658203e-05,
      "model_forward_time": 1.7150421142578125,
      "step": 46853
    },
    {
      "epoch": 4.468250274658203e-05,
      "step": 46853,
      "training_step_time": 8.468826532363892
    },
    {
      "epoch": 4.468345642089844e-05,
      "model_forward_time": 1.7168481349945068,
      "step": 46854
    },
    {
      "epoch": 4.468345642089844e-05,
      "step": 46854,
      "training_step_time": 8.382823944091797
    },
    {
      "epoch": 4.468441009521484e-05,
      "model_forward_time": 1.7132883071899414,
      "step": 46855
    },
    {
      "epoch": 4.468441009521484e-05,
      "step": 46855,
      "training_step_time": 7.9072651863098145
    },
    {
      "epoch": 4.468536376953125e-05,
      "model_forward_time": 1.7118561267852783,
      "step": 46856
    },
    {
      "epoch": 4.468536376953125e-05,
      "step": 46856,
      "training_step_time": 7.725645542144775
    },
    {
      "epoch": 4.4686317443847656e-05,
      "model_forward_time": 1.7026827335357666,
      "step": 46857
    },
    {
      "epoch": 4.4686317443847656e-05,
      "step": 46857,
      "training_step_time": 7.11933159828186
    },
    {
      "epoch": 4.468727111816406e-05,
      "model_forward_time": 1.7159490585327148,
      "step": 46858
    },
    {
      "epoch": 4.468727111816406e-05,
      "step": 46858,
      "training_step_time": 7.502943754196167
    },
    {
      "epoch": 4.468822479248047e-05,
      "model_forward_time": 1.715691089630127,
      "step": 46859
    },
    {
      "epoch": 4.468822479248047e-05,
      "step": 46859,
      "training_step_time": 7.804114580154419
    },
    {
      "epoch": 4.4689178466796874e-05,
      "grad_norm": 0.15013396739959717,
      "learning_rate": 5.927322120019352e-06,
      "loss": 0.0573,
      "step": 46860
    },
    {
      "epoch": 4.4689178466796874e-05,
      "model_forward_time": 1.7115914821624756,
      "step": 46860
    },
    {
      "dynamics_loss_avg": 0.06800103988498449,
      "epoch": 4.4689178466796874e-05,
      "step": 46860
    },
    {
      "action_loss_avg": 0.0014590207574656233,
      "epoch": 4.4689178466796874e-05,
      "step": 46860
    },
    {
      "epoch": 4.4689178466796874e-05,
      "step": 46860,
      "training_step_time": 7.726571083068848
    },
    {
      "epoch": 4.469013214111328e-05,
      "model_forward_time": 2.3101680278778076,
      "step": 46861
    },
    {
      "epoch": 4.469013214111328e-05,
      "step": 46861,
      "training_step_time": 8.743910074234009
    },
    {
      "epoch": 4.469108581542969e-05,
      "model_forward_time": 1.7243733406066895,
      "step": 46862
    },
    {
      "epoch": 4.469108581542969e-05,
      "step": 46862,
      "training_step_time": 8.717705249786377
    },
    {
      "epoch": 4.469203948974609e-05,
      "model_forward_time": 1.7166237831115723,
      "step": 46863
    },
    {
      "epoch": 4.469203948974609e-05,
      "step": 46863,
      "training_step_time": 9.98159122467041
    },
    {
      "epoch": 4.46929931640625e-05,
      "model_forward_time": 1.7157659530639648,
      "step": 46864
    },
    {
      "epoch": 4.46929931640625e-05,
      "step": 46864,
      "training_step_time": 7.838759422302246
    },
    {
      "epoch": 4.469394683837891e-05,
      "model_forward_time": 1.7203192710876465,
      "step": 46865
    },
    {
      "epoch": 4.469394683837891e-05,
      "step": 46865,
      "training_step_time": 7.7209718227386475
    },
    {
      "epoch": 4.469490051269531e-05,
      "model_forward_time": 1.7169084548950195,
      "step": 46866
    },
    {
      "epoch": 4.469490051269531e-05,
      "step": 46866,
      "training_step_time": 7.24350118637085
    },
    {
      "epoch": 4.4695854187011716e-05,
      "model_forward_time": 1.7181830406188965,
      "step": 46867
    },
    {
      "epoch": 4.4695854187011716e-05,
      "step": 46867,
      "training_step_time": 7.543178558349609
    },
    {
      "epoch": 4.469680786132813e-05,
      "model_forward_time": 1.7209374904632568,
      "step": 46868
    },
    {
      "epoch": 4.469680786132813e-05,
      "step": 46868,
      "training_step_time": 8.44684362411499
    },
    {
      "epoch": 4.469776153564453e-05,
      "model_forward_time": 1.7098667621612549,
      "step": 46869
    },
    {
      "epoch": 4.469776153564453e-05,
      "step": 46869,
      "training_step_time": 7.8690149784088135
    },
    {
      "epoch": 4.4698715209960935e-05,
      "grad_norm": 0.15337301790714264,
      "learning_rate": 5.9256972856078445e-06,
      "loss": 0.0602,
      "step": 46870
    },
    {
      "epoch": 4.4698715209960935e-05,
      "model_forward_time": 2.2495932579040527,
      "step": 46870
    },
    {
      "dynamics_loss_avg": 0.05724395792931318,
      "epoch": 4.4698715209960935e-05,
      "step": 46870
    },
    {
      "action_loss_avg": 0.001232915124273859,
      "epoch": 4.4698715209960935e-05,
      "step": 46870
    },
    {
      "epoch": 4.4698715209960935e-05,
      "step": 46870,
      "training_step_time": 8.348109483718872
    },
    {
      "epoch": 4.4699668884277346e-05,
      "model_forward_time": 1.7574036121368408,
      "step": 46871
    },
    {
      "epoch": 4.4699668884277346e-05,
      "step": 46871,
      "training_step_time": 7.707683563232422
    },
    {
      "epoch": 4.470062255859375e-05,
      "model_forward_time": 1.7298526763916016,
      "step": 46872
    },
    {
      "epoch": 4.470062255859375e-05,
      "step": 46872,
      "training_step_time": 8.155272006988525
    },
    {
      "epoch": 4.4701576232910154e-05,
      "model_forward_time": 1.7134010791778564,
      "step": 46873
    },
    {
      "epoch": 4.4701576232910154e-05,
      "step": 46873,
      "training_step_time": 8.294154167175293
    },
    {
      "epoch": 4.4702529907226565e-05,
      "model_forward_time": 1.7176311016082764,
      "step": 46874
    },
    {
      "epoch": 4.4702529907226565e-05,
      "step": 46874,
      "training_step_time": 7.686060190200806
    },
    {
      "epoch": 4.470348358154297e-05,
      "model_forward_time": 1.7215778827667236,
      "step": 46875
    },
    {
      "epoch": 4.470348358154297e-05,
      "step": 46875,
      "training_step_time": 7.242236852645874
    },
    {
      "epoch": 4.470443725585937e-05,
      "model_forward_time": 1.713252305984497,
      "step": 46876
    },
    {
      "epoch": 4.470443725585937e-05,
      "step": 46876,
      "training_step_time": 7.57096004486084
    },
    {
      "epoch": 4.4705390930175784e-05,
      "model_forward_time": 1.7120389938354492,
      "step": 46877
    },
    {
      "epoch": 4.4705390930175784e-05,
      "step": 46877,
      "training_step_time": 7.733034133911133
    },
    {
      "epoch": 4.470634460449219e-05,
      "model_forward_time": 1.7311723232269287,
      "step": 46878
    },
    {
      "epoch": 4.470634460449219e-05,
      "step": 46878,
      "training_step_time": 7.871166944503784
    },
    {
      "epoch": 4.470729827880859e-05,
      "model_forward_time": 1.7228119373321533,
      "step": 46879
    },
    {
      "epoch": 4.470729827880859e-05,
      "step": 46879,
      "training_step_time": 7.605132579803467
    },
    {
      "epoch": 4.4708251953125e-05,
      "grad_norm": 0.1310216188430786,
      "learning_rate": 5.924072349963472e-06,
      "loss": 0.0581,
      "step": 46880
    },
    {
      "epoch": 4.4708251953125e-05,
      "model_forward_time": 2.322077751159668,
      "step": 46880
    },
    {
      "dynamics_loss_avg": 0.06737156603485346,
      "epoch": 4.4708251953125e-05,
      "step": 46880
    },
    {
      "action_loss_avg": 0.001695530512370169,
      "epoch": 4.4708251953125e-05,
      "step": 46880
    },
    {
      "epoch": 4.4708251953125e-05,
      "step": 46880,
      "training_step_time": 7.804704427719116
    },
    {
      "epoch": 4.4709205627441406e-05,
      "model_forward_time": 1.7328224182128906,
      "step": 46881
    },
    {
      "epoch": 4.4709205627441406e-05,
      "step": 46881,
      "training_step_time": 7.828672409057617
    },
    {
      "epoch": 4.471015930175781e-05,
      "model_forward_time": 1.7025880813598633,
      "step": 46882
    },
    {
      "epoch": 4.471015930175781e-05,
      "step": 46882,
      "training_step_time": 9.212641716003418
    },
    {
      "epoch": 4.471111297607422e-05,
      "model_forward_time": 1.72446608543396,
      "step": 46883
    },
    {
      "epoch": 4.471111297607422e-05,
      "step": 46883,
      "training_step_time": 9.47977328300476
    },
    {
      "epoch": 4.4712066650390625e-05,
      "model_forward_time": 1.7171783447265625,
      "step": 46884
    },
    {
      "epoch": 4.4712066650390625e-05,
      "step": 46884,
      "training_step_time": 7.269362211227417
    },
    {
      "epoch": 4.471302032470703e-05,
      "model_forward_time": 1.710665225982666,
      "step": 46885
    },
    {
      "epoch": 4.471302032470703e-05,
      "step": 46885,
      "training_step_time": 7.382941484451294
    },
    {
      "epoch": 4.471397399902344e-05,
      "model_forward_time": 1.7225205898284912,
      "step": 46886
    },
    {
      "epoch": 4.471397399902344e-05,
      "step": 46886,
      "training_step_time": 7.723309516906738
    },
    {
      "epoch": 4.4714927673339844e-05,
      "model_forward_time": 1.707291603088379,
      "step": 46887
    },
    {
      "epoch": 4.4714927673339844e-05,
      "step": 46887,
      "training_step_time": 9.180736064910889
    },
    {
      "epoch": 4.471588134765625e-05,
      "model_forward_time": 1.7169108390808105,
      "step": 46888
    },
    {
      "epoch": 4.471588134765625e-05,
      "step": 46888,
      "training_step_time": 7.419004201889038
    },
    {
      "epoch": 4.471683502197266e-05,
      "model_forward_time": 2.319385051727295,
      "step": 46889
    },
    {
      "epoch": 4.471683502197266e-05,
      "step": 46889,
      "training_step_time": 7.594882249832153
    },
    {
      "epoch": 4.471778869628906e-05,
      "grad_norm": 0.1009678840637207,
      "learning_rate": 5.922447313263938e-06,
      "loss": 0.0589,
      "step": 46890
    },
    {
      "epoch": 4.471778869628906e-05,
      "model_forward_time": 1.7143731117248535,
      "step": 46890
    },
    {
      "dynamics_loss_avg": 0.06467543076723814,
      "epoch": 4.471778869628906e-05,
      "step": 46890
    },
    {
      "action_loss_avg": 0.001098644666490145,
      "epoch": 4.471778869628906e-05,
      "step": 46890
    },
    {
      "epoch": 4.471778869628906e-05,
      "step": 46890,
      "training_step_time": 9.291253089904785
    },
    {
      "epoch": 4.471874237060547e-05,
      "model_forward_time": 1.7216577529907227,
      "step": 46891
    },
    {
      "epoch": 4.471874237060547e-05,
      "step": 46891,
      "training_step_time": 7.786212921142578
    },
    {
      "epoch": 4.471969604492188e-05,
      "model_forward_time": 1.7192282676696777,
      "step": 46892
    },
    {
      "epoch": 4.471969604492188e-05,
      "step": 46892,
      "training_step_time": 7.5890514850616455
    },
    {
      "epoch": 4.472064971923828e-05,
      "model_forward_time": 1.710129976272583,
      "step": 46893
    },
    {
      "epoch": 4.472064971923828e-05,
      "step": 46893,
      "training_step_time": 7.5144829750061035
    },
    {
      "epoch": 4.4721603393554686e-05,
      "model_forward_time": 1.3444385528564453,
      "step": 46894
    },
    {
      "epoch": 4.4721603393554686e-05,
      "step": 46894,
      "training_step_time": 7.26487922668457
    },
    {
      "epoch": 4.47225570678711e-05,
      "model_forward_time": 1.713268518447876,
      "step": 46895
    },
    {
      "epoch": 4.47225570678711e-05,
      "step": 46895,
      "training_step_time": 7.605693340301514
    },
    {
      "epoch": 4.47235107421875e-05,
      "model_forward_time": 1.710679054260254,
      "step": 46896
    },
    {
      "epoch": 4.47235107421875e-05,
      "step": 46896,
      "training_step_time": 7.671773433685303
    },
    {
      "epoch": 4.4724464416503905e-05,
      "model_forward_time": 1.7189030647277832,
      "step": 46897
    },
    {
      "epoch": 4.4724464416503905e-05,
      "step": 46897,
      "training_step_time": 8.28781771659851
    },
    {
      "epoch": 4.4725418090820316e-05,
      "model_forward_time": 2.32857608795166,
      "step": 46898
    },
    {
      "epoch": 4.4725418090820316e-05,
      "step": 46898,
      "training_step_time": 9.766363859176636
    },
    {
      "epoch": 4.472637176513672e-05,
      "model_forward_time": 1.7181735038757324,
      "step": 46899
    },
    {
      "epoch": 4.472637176513672e-05,
      "step": 46899,
      "training_step_time": 7.740862607955933
    },
    {
      "epoch": 4.4727325439453124e-05,
      "grad_norm": 0.11447472870349884,
      "learning_rate": 5.920822175686952e-06,
      "loss": 0.058,
      "step": 46900
    },
    {
      "epoch": 4.4727325439453124e-05,
      "model_forward_time": 1.709465742111206,
      "step": 46900
    },
    {
      "dynamics_loss_avg": 0.06061302684247494,
      "epoch": 4.4727325439453124e-05,
      "step": 46900
    },
    {
      "action_loss_avg": 0.0013619870296679437,
      "epoch": 4.4727325439453124e-05,
      "step": 46900
    },
    {
      "epoch": 4.4727325439453124e-05,
      "step": 46900,
      "training_step_time": 8.315914630889893
    },
    {
      "epoch": 4.4728279113769534e-05,
      "model_forward_time": 1.714263916015625,
      "step": 46901
    },
    {
      "epoch": 4.4728279113769534e-05,
      "step": 46901,
      "training_step_time": 7.816763401031494
    },
    {
      "epoch": 4.472923278808594e-05,
      "model_forward_time": 1.7062156200408936,
      "step": 46902
    },
    {
      "epoch": 4.472923278808594e-05,
      "step": 46902,
      "training_step_time": 7.673171758651733
    },
    {
      "epoch": 4.473018646240234e-05,
      "model_forward_time": 1.7119672298431396,
      "step": 46903
    },
    {
      "epoch": 4.473018646240234e-05,
      "step": 46903,
      "training_step_time": 7.43670392036438
    },
    {
      "epoch": 4.473114013671875e-05,
      "model_forward_time": 1.7147293090820312,
      "step": 46904
    },
    {
      "epoch": 4.473114013671875e-05,
      "step": 46904,
      "training_step_time": 7.652504205703735
    },
    {
      "epoch": 4.473209381103516e-05,
      "model_forward_time": 1.7117836475372314,
      "step": 46905
    },
    {
      "epoch": 4.473209381103516e-05,
      "step": 46905,
      "training_step_time": 9.092925786972046
    },
    {
      "epoch": 4.473304748535156e-05,
      "model_forward_time": 1.715510368347168,
      "step": 46906
    },
    {
      "epoch": 4.473304748535156e-05,
      "step": 46906,
      "training_step_time": 7.9012696743011475
    },
    {
      "epoch": 4.4734001159667965e-05,
      "model_forward_time": 2.284172534942627,
      "step": 46907
    },
    {
      "epoch": 4.4734001159667965e-05,
      "step": 46907,
      "training_step_time": 8.760923147201538
    },
    {
      "epoch": 4.4734954833984376e-05,
      "model_forward_time": 1.761305570602417,
      "step": 46908
    },
    {
      "epoch": 4.4734954833984376e-05,
      "step": 46908,
      "training_step_time": 8.685371398925781
    },
    {
      "epoch": 4.473590850830078e-05,
      "model_forward_time": 1.7264182567596436,
      "step": 46909
    },
    {
      "epoch": 4.473590850830078e-05,
      "step": 46909,
      "training_step_time": 8.866727352142334
    },
    {
      "epoch": 4.4736862182617184e-05,
      "grad_norm": 0.121327705681324,
      "learning_rate": 5.919196937410239e-06,
      "loss": 0.058,
      "step": 46910
    },
    {
      "epoch": 4.4736862182617184e-05,
      "model_forward_time": 1.7098884582519531,
      "step": 46910
    },
    {
      "dynamics_loss_avg": 0.05926470970734954,
      "epoch": 4.4736862182617184e-05,
      "step": 46910
    },
    {
      "action_loss_avg": 0.0014877447247272357,
      "epoch": 4.4736862182617184e-05,
      "step": 46910
    },
    {
      "epoch": 4.4736862182617184e-05,
      "step": 46910,
      "training_step_time": 8.540074348449707
    },
    {
      "epoch": 4.4737815856933595e-05,
      "model_forward_time": 1.7258174419403076,
      "step": 46911
    },
    {
      "epoch": 4.4737815856933595e-05,
      "step": 46911,
      "training_step_time": 7.751958131790161
    },
    {
      "epoch": 4.473876953125e-05,
      "model_forward_time": 1.7157955169677734,
      "step": 46912
    },
    {
      "epoch": 4.473876953125e-05,
      "step": 46912,
      "training_step_time": 7.610810279846191
    },
    {
      "epoch": 4.47397232055664e-05,
      "model_forward_time": 1.7112407684326172,
      "step": 46913
    },
    {
      "epoch": 4.47397232055664e-05,
      "step": 46913,
      "training_step_time": 7.848152160644531
    },
    {
      "epoch": 4.4740676879882814e-05,
      "model_forward_time": 1.7229900360107422,
      "step": 46914
    },
    {
      "epoch": 4.4740676879882814e-05,
      "step": 46914,
      "training_step_time": 8.23935866355896
    },
    {
      "epoch": 4.474163055419922e-05,
      "model_forward_time": 1.7166266441345215,
      "step": 46915
    },
    {
      "epoch": 4.474163055419922e-05,
      "step": 46915,
      "training_step_time": 7.880888938903809
    },
    {
      "epoch": 4.474258422851562e-05,
      "model_forward_time": 1.7104542255401611,
      "step": 46916
    },
    {
      "epoch": 4.474258422851562e-05,
      "step": 46916,
      "training_step_time": 9.805131912231445
    },
    {
      "epoch": 4.474353790283203e-05,
      "model_forward_time": 1.7552542686462402,
      "step": 46917
    },
    {
      "epoch": 4.474353790283203e-05,
      "step": 46917,
      "training_step_time": 8.748474597930908
    },
    {
      "epoch": 4.474449157714844e-05,
      "model_forward_time": 1.337899923324585,
      "step": 46918
    },
    {
      "epoch": 4.474449157714844e-05,
      "step": 46918,
      "training_step_time": 8.773418664932251
    },
    {
      "epoch": 4.474544525146484e-05,
      "model_forward_time": 1.726766586303711,
      "step": 46919
    },
    {
      "epoch": 4.474544525146484e-05,
      "step": 46919,
      "training_step_time": 9.423514604568481
    },
    {
      "epoch": 4.474639892578125e-05,
      "grad_norm": 0.09845903515815735,
      "learning_rate": 5.9175715986115286e-06,
      "loss": 0.0582,
      "step": 46920
    },
    {
      "epoch": 4.474639892578125e-05,
      "model_forward_time": 1.7228538990020752,
      "step": 46920
    },
    {
      "dynamics_loss_avg": 0.060737119242548945,
      "epoch": 4.474639892578125e-05,
      "step": 46920
    },
    {
      "action_loss_avg": 0.001997426841990091,
      "epoch": 4.474639892578125e-05,
      "step": 46920
    },
    {
      "epoch": 4.474639892578125e-05,
      "step": 46920,
      "training_step_time": 8.852135419845581
    },
    {
      "epoch": 4.4747352600097656e-05,
      "model_forward_time": 1.7304198741912842,
      "step": 46921
    },
    {
      "epoch": 4.4747352600097656e-05,
      "step": 46921,
      "training_step_time": 7.571904420852661
    },
    {
      "epoch": 4.474830627441406e-05,
      "model_forward_time": 1.727088212966919,
      "step": 46922
    },
    {
      "epoch": 4.474830627441406e-05,
      "step": 46922,
      "training_step_time": 7.86583137512207
    },
    {
      "epoch": 4.474925994873047e-05,
      "model_forward_time": 1.7194898128509521,
      "step": 46923
    },
    {
      "epoch": 4.474925994873047e-05,
      "step": 46923,
      "training_step_time": 8.109318971633911
    },
    {
      "epoch": 4.4750213623046875e-05,
      "model_forward_time": 1.7281482219696045,
      "step": 46924
    },
    {
      "epoch": 4.4750213623046875e-05,
      "step": 46924,
      "training_step_time": 7.867380857467651
    },
    {
      "epoch": 4.475116729736328e-05,
      "model_forward_time": 1.7239110469818115,
      "step": 46925
    },
    {
      "epoch": 4.475116729736328e-05,
      "step": 46925,
      "training_step_time": 7.76468563079834
    },
    {
      "epoch": 4.475212097167969e-05,
      "model_forward_time": 2.3054299354553223,
      "step": 46926
    },
    {
      "epoch": 4.475212097167969e-05,
      "step": 46926,
      "training_step_time": 8.848270893096924
    },
    {
      "epoch": 4.475307464599609e-05,
      "model_forward_time": 1.7213866710662842,
      "step": 46927
    },
    {
      "epoch": 4.475307464599609e-05,
      "step": 46927,
      "training_step_time": 8.613929033279419
    },
    {
      "epoch": 4.47540283203125e-05,
      "model_forward_time": 1.7197308540344238,
      "step": 46928
    },
    {
      "epoch": 4.47540283203125e-05,
      "step": 46928,
      "training_step_time": 7.801089286804199
    },
    {
      "epoch": 4.475498199462891e-05,
      "model_forward_time": 1.7180609703063965,
      "step": 46929
    },
    {
      "epoch": 4.475498199462891e-05,
      "step": 46929,
      "training_step_time": 7.8253700733184814
    },
    {
      "epoch": 4.475593566894531e-05,
      "grad_norm": 0.13721488416194916,
      "learning_rate": 5.915946159468568e-06,
      "loss": 0.0561,
      "step": 46930
    },
    {
      "epoch": 4.475593566894531e-05,
      "model_forward_time": 1.7264790534973145,
      "step": 46930
    },
    {
      "dynamics_loss_avg": 0.05466090925037861,
      "epoch": 4.475593566894531e-05,
      "step": 46930
    },
    {
      "action_loss_avg": 0.0012203975638840348,
      "epoch": 4.475593566894531e-05,
      "step": 46930
    },
    {
      "epoch": 4.475593566894531e-05,
      "step": 46930,
      "training_step_time": 7.700325012207031
    },
    {
      "epoch": 4.4756889343261716e-05,
      "model_forward_time": 1.734750747680664,
      "step": 46931
    },
    {
      "epoch": 4.4756889343261716e-05,
      "step": 46931,
      "training_step_time": 7.804313898086548
    },
    {
      "epoch": 4.475784301757813e-05,
      "model_forward_time": 1.7115790843963623,
      "step": 46932
    },
    {
      "epoch": 4.475784301757813e-05,
      "step": 46932,
      "training_step_time": 8.148013353347778
    },
    {
      "epoch": 4.475879669189453e-05,
      "model_forward_time": 1.7145357131958008,
      "step": 46933
    },
    {
      "epoch": 4.475879669189453e-05,
      "step": 46933,
      "training_step_time": 7.69134783744812
    },
    {
      "epoch": 4.4759750366210935e-05,
      "model_forward_time": 1.7172632217407227,
      "step": 46934
    },
    {
      "epoch": 4.4759750366210935e-05,
      "step": 46934,
      "training_step_time": 7.358515501022339
    },
    {
      "epoch": 4.4760704040527346e-05,
      "model_forward_time": 2.278610944747925,
      "step": 46935
    },
    {
      "epoch": 4.4760704040527346e-05,
      "step": 46935,
      "training_step_time": 10.317878246307373
    },
    {
      "epoch": 4.476165771484375e-05,
      "model_forward_time": 1.7699644565582275,
      "step": 46936
    },
    {
      "epoch": 4.476165771484375e-05,
      "step": 46936,
      "training_step_time": 7.800050973892212
    },
    {
      "epoch": 4.4762611389160154e-05,
      "model_forward_time": 1.7195403575897217,
      "step": 46937
    },
    {
      "epoch": 4.4762611389160154e-05,
      "step": 46937,
      "training_step_time": 8.400795221328735
    },
    {
      "epoch": 4.4763565063476565e-05,
      "model_forward_time": 1.7162067890167236,
      "step": 46938
    },
    {
      "epoch": 4.4763565063476565e-05,
      "step": 46938,
      "training_step_time": 8.789876937866211
    },
    {
      "epoch": 4.476451873779297e-05,
      "model_forward_time": 1.71822190284729,
      "step": 46939
    },
    {
      "epoch": 4.476451873779297e-05,
      "step": 46939,
      "training_step_time": 8.689801216125488
    },
    {
      "epoch": 4.476547241210937e-05,
      "grad_norm": 0.11926168203353882,
      "learning_rate": 5.914320620159114e-06,
      "loss": 0.0571,
      "step": 46940
    },
    {
      "epoch": 4.476547241210937e-05,
      "model_forward_time": 1.715064525604248,
      "step": 46940
    },
    {
      "dynamics_loss_avg": 0.06600394137203694,
      "epoch": 4.476547241210937e-05,
      "step": 46940
    },
    {
      "action_loss_avg": 0.001276564787258394,
      "epoch": 4.476547241210937e-05,
      "step": 46940
    },
    {
      "epoch": 4.476547241210937e-05,
      "step": 46940,
      "training_step_time": 7.753152847290039
    },
    {
      "epoch": 4.4766426086425784e-05,
      "model_forward_time": 1.7157032489776611,
      "step": 46941
    },
    {
      "epoch": 4.4766426086425784e-05,
      "step": 46941,
      "training_step_time": 8.36455488204956
    },
    {
      "epoch": 4.476737976074219e-05,
      "model_forward_time": 1.7289667129516602,
      "step": 46942
    },
    {
      "epoch": 4.476737976074219e-05,
      "step": 46942,
      "training_step_time": 7.7707459926605225
    },
    {
      "epoch": 4.476833343505859e-05,
      "model_forward_time": 1.720566749572754,
      "step": 46943
    },
    {
      "epoch": 4.476833343505859e-05,
      "step": 46943,
      "training_step_time": 7.749216318130493
    },
    {
      "epoch": 4.4769287109375e-05,
      "model_forward_time": 2.2903151512145996,
      "step": 46944
    },
    {
      "epoch": 4.4769287109375e-05,
      "step": 46944,
      "training_step_time": 8.792924165725708
    },
    {
      "epoch": 4.4770240783691407e-05,
      "model_forward_time": 1.7571511268615723,
      "step": 46945
    },
    {
      "epoch": 4.4770240783691407e-05,
      "step": 46945,
      "training_step_time": 7.7170422077178955
    },
    {
      "epoch": 4.477119445800781e-05,
      "model_forward_time": 1.7112078666687012,
      "step": 46946
    },
    {
      "epoch": 4.477119445800781e-05,
      "step": 46946,
      "training_step_time": 8.605052947998047
    },
    {
      "epoch": 4.477214813232422e-05,
      "model_forward_time": 1.707975149154663,
      "step": 46947
    },
    {
      "epoch": 4.477214813232422e-05,
      "step": 46947,
      "training_step_time": 8.396289110183716
    },
    {
      "epoch": 4.4773101806640625e-05,
      "model_forward_time": 1.7160608768463135,
      "step": 46948
    },
    {
      "epoch": 4.4773101806640625e-05,
      "step": 46948,
      "training_step_time": 8.211404800415039
    },
    {
      "epoch": 4.477405548095703e-05,
      "model_forward_time": 1.723924160003662,
      "step": 46949
    },
    {
      "epoch": 4.477405548095703e-05,
      "step": 46949,
      "training_step_time": 7.700592517852783
    },
    {
      "epoch": 4.477500915527344e-05,
      "grad_norm": 0.13270027935504913,
      "learning_rate": 5.912694980860931e-06,
      "loss": 0.0581,
      "step": 46950
    },
    {
      "epoch": 4.477500915527344e-05,
      "model_forward_time": 1.7142913341522217,
      "step": 46950
    },
    {
      "dynamics_loss_avg": 0.05311568863689899,
      "epoch": 4.477500915527344e-05,
      "step": 46950
    },
    {
      "action_loss_avg": 0.0010799173236591742,
      "epoch": 4.477500915527344e-05,
      "step": 46950
    },
    {
      "epoch": 4.477500915527344e-05,
      "step": 46950,
      "training_step_time": 7.498822927474976
    },
    {
      "epoch": 4.4775962829589844e-05,
      "model_forward_time": 1.717621088027954,
      "step": 46951
    },
    {
      "epoch": 4.4775962829589844e-05,
      "step": 46951,
      "training_step_time": 8.224987030029297
    },
    {
      "epoch": 4.477691650390625e-05,
      "model_forward_time": 1.7197613716125488,
      "step": 46952
    },
    {
      "epoch": 4.477691650390625e-05,
      "step": 46952,
      "training_step_time": 7.7034406661987305
    },
    {
      "epoch": 4.477787017822266e-05,
      "model_forward_time": 2.260969877243042,
      "step": 46953
    },
    {
      "epoch": 4.477787017822266e-05,
      "step": 46953,
      "training_step_time": 7.6016974449157715
    },
    {
      "epoch": 4.477882385253906e-05,
      "model_forward_time": 1.725083589553833,
      "step": 46954
    },
    {
      "epoch": 4.477882385253906e-05,
      "step": 46954,
      "training_step_time": 7.622846364974976
    },
    {
      "epoch": 4.477977752685547e-05,
      "model_forward_time": 1.7334656715393066,
      "step": 46955
    },
    {
      "epoch": 4.477977752685547e-05,
      "step": 46955,
      "training_step_time": 8.20154881477356
    },
    {
      "epoch": 4.478073120117188e-05,
      "model_forward_time": 1.7210853099822998,
      "step": 46956
    },
    {
      "epoch": 4.478073120117188e-05,
      "step": 46956,
      "training_step_time": 7.672744989395142
    },
    {
      "epoch": 4.478168487548828e-05,
      "model_forward_time": 1.7194907665252686,
      "step": 46957
    },
    {
      "epoch": 4.478168487548828e-05,
      "step": 46957,
      "training_step_time": 7.73540997505188
    },
    {
      "epoch": 4.4782638549804686e-05,
      "model_forward_time": 1.718423843383789,
      "step": 46958
    },
    {
      "epoch": 4.4782638549804686e-05,
      "step": 46958,
      "training_step_time": 7.502318382263184
    },
    {
      "epoch": 4.47835922241211e-05,
      "model_forward_time": 1.7160499095916748,
      "step": 46959
    },
    {
      "epoch": 4.47835922241211e-05,
      "step": 46959,
      "training_step_time": 7.645308256149292
    },
    {
      "epoch": 4.47845458984375e-05,
      "grad_norm": 0.11553549021482468,
      "learning_rate": 5.9110692417517965e-06,
      "loss": 0.0596,
      "step": 46960
    },
    {
      "epoch": 4.47845458984375e-05,
      "model_forward_time": 1.7120144367218018,
      "step": 46960
    },
    {
      "dynamics_loss_avg": 0.08761882558465003,
      "epoch": 4.47845458984375e-05,
      "step": 46960
    },
    {
      "action_loss_avg": 0.0018047832389129325,
      "epoch": 4.47845458984375e-05,
      "step": 46960
    },
    {
      "epoch": 4.47845458984375e-05,
      "step": 46960,
      "training_step_time": 7.507998943328857
    },
    {
      "epoch": 4.4785499572753905e-05,
      "model_forward_time": 1.7200160026550293,
      "step": 46961
    },
    {
      "epoch": 4.4785499572753905e-05,
      "step": 46961,
      "training_step_time": 7.800288677215576
    },
    {
      "epoch": 4.4786453247070316e-05,
      "model_forward_time": 1.7230165004730225,
      "step": 46962
    },
    {
      "epoch": 4.4786453247070316e-05,
      "step": 46962,
      "training_step_time": 7.667900800704956
    },
    {
      "epoch": 4.478740692138672e-05,
      "model_forward_time": 2.284618854522705,
      "step": 46963
    },
    {
      "epoch": 4.478740692138672e-05,
      "step": 46963,
      "training_step_time": 8.800987243652344
    },
    {
      "epoch": 4.4788360595703124e-05,
      "model_forward_time": 1.7503507137298584,
      "step": 46964
    },
    {
      "epoch": 4.4788360595703124e-05,
      "step": 46964,
      "training_step_time": 7.691918134689331
    },
    {
      "epoch": 4.4789314270019535e-05,
      "model_forward_time": 1.7122690677642822,
      "step": 46965
    },
    {
      "epoch": 4.4789314270019535e-05,
      "step": 46965,
      "training_step_time": 8.284942150115967
    },
    {
      "epoch": 4.479026794433594e-05,
      "model_forward_time": 1.717365026473999,
      "step": 46966
    },
    {
      "epoch": 4.479026794433594e-05,
      "step": 46966,
      "training_step_time": 7.70148491859436
    },
    {
      "epoch": 4.479122161865234e-05,
      "model_forward_time": 1.7312605381011963,
      "step": 46967
    },
    {
      "epoch": 4.479122161865234e-05,
      "step": 46967,
      "training_step_time": 8.321343421936035
    },
    {
      "epoch": 4.479217529296875e-05,
      "model_forward_time": 1.7201356887817383,
      "step": 46968
    },
    {
      "epoch": 4.479217529296875e-05,
      "step": 46968,
      "training_step_time": 7.704028606414795
    },
    {
      "epoch": 4.479312896728516e-05,
      "model_forward_time": 1.7250912189483643,
      "step": 46969
    },
    {
      "epoch": 4.479312896728516e-05,
      "step": 46969,
      "training_step_time": 7.517561912536621
    },
    {
      "epoch": 4.479408264160156e-05,
      "grad_norm": 0.153293639421463,
      "learning_rate": 5.909443403009501e-06,
      "loss": 0.0608,
      "step": 46970
    },
    {
      "epoch": 4.479408264160156e-05,
      "model_forward_time": 1.7150671482086182,
      "step": 46970
    },
    {
      "dynamics_loss_avg": 0.05852578692138195,
      "epoch": 4.479408264160156e-05,
      "step": 46970
    },
    {
      "action_loss_avg": 0.0011479706066893415,
      "epoch": 4.479408264160156e-05,
      "step": 46970
    },
    {
      "epoch": 4.479408264160156e-05,
      "step": 46970,
      "training_step_time": 7.802061557769775
    },
    {
      "epoch": 4.4795036315917965e-05,
      "model_forward_time": 1.711878776550293,
      "step": 46971
    },
    {
      "epoch": 4.4795036315917965e-05,
      "step": 46971,
      "training_step_time": 8.408189535140991
    },
    {
      "epoch": 4.4795989990234376e-05,
      "model_forward_time": 2.286878824234009,
      "step": 46972
    },
    {
      "epoch": 4.4795989990234376e-05,
      "step": 46972,
      "training_step_time": 7.776304483413696
    },
    {
      "epoch": 4.479694366455078e-05,
      "model_forward_time": 1.7603015899658203,
      "step": 46973
    },
    {
      "epoch": 4.479694366455078e-05,
      "step": 46973,
      "training_step_time": 9.49508285522461
    },
    {
      "epoch": 4.4797897338867184e-05,
      "model_forward_time": 1.736684799194336,
      "step": 46974
    },
    {
      "epoch": 4.4797897338867184e-05,
      "step": 46974,
      "training_step_time": 8.646713018417358
    },
    {
      "epoch": 4.4798851013183595e-05,
      "model_forward_time": 1.7153174877166748,
      "step": 46975
    },
    {
      "epoch": 4.4798851013183595e-05,
      "step": 46975,
      "training_step_time": 8.93327260017395
    },
    {
      "epoch": 4.47998046875e-05,
      "model_forward_time": 1.720916509628296,
      "step": 46976
    },
    {
      "epoch": 4.47998046875e-05,
      "step": 46976,
      "training_step_time": 8.58240556716919
    },
    {
      "epoch": 4.48007583618164e-05,
      "model_forward_time": 1.711479663848877,
      "step": 46977
    },
    {
      "epoch": 4.48007583618164e-05,
      "step": 46977,
      "training_step_time": 7.730246305465698
    },
    {
      "epoch": 4.4801712036132814e-05,
      "model_forward_time": 1.7162318229675293,
      "step": 46978
    },
    {
      "epoch": 4.4801712036132814e-05,
      "step": 46978,
      "training_step_time": 7.503860712051392
    },
    {
      "epoch": 4.480266571044922e-05,
      "model_forward_time": 1.717470407485962,
      "step": 46979
    },
    {
      "epoch": 4.480266571044922e-05,
      "step": 46979,
      "training_step_time": 7.778421878814697
    },
    {
      "epoch": 4.480361938476562e-05,
      "grad_norm": 0.16478288173675537,
      "learning_rate": 5.9078174648118415e-06,
      "loss": 0.0587,
      "step": 46980
    },
    {
      "epoch": 4.480361938476562e-05,
      "model_forward_time": 1.7097601890563965,
      "step": 46980
    },
    {
      "dynamics_loss_avg": 0.04381563831120729,
      "epoch": 4.480361938476562e-05,
      "step": 46980
    },
    {
      "action_loss_avg": 0.0013794505037367343,
      "epoch": 4.480361938476562e-05,
      "step": 46980
    },
    {
      "epoch": 4.480361938476562e-05,
      "step": 46980,
      "training_step_time": 7.715468645095825
    },
    {
      "epoch": 4.480457305908203e-05,
      "model_forward_time": 2.2765867710113525,
      "step": 46981
    },
    {
      "epoch": 4.480457305908203e-05,
      "step": 46981,
      "training_step_time": 7.7243406772613525
    },
    {
      "epoch": 4.480552673339844e-05,
      "model_forward_time": 1.760157585144043,
      "step": 46982
    },
    {
      "epoch": 4.480552673339844e-05,
      "step": 46982,
      "training_step_time": 7.707540512084961
    },
    {
      "epoch": 4.480648040771484e-05,
      "model_forward_time": 1.723210334777832,
      "step": 46983
    },
    {
      "epoch": 4.480648040771484e-05,
      "step": 46983,
      "training_step_time": 8.879279613494873
    },
    {
      "epoch": 4.480743408203125e-05,
      "model_forward_time": 1.721785545349121,
      "step": 46984
    },
    {
      "epoch": 4.480743408203125e-05,
      "step": 46984,
      "training_step_time": 8.483933448791504
    },
    {
      "epoch": 4.4808387756347656e-05,
      "model_forward_time": 1.7215843200683594,
      "step": 46985
    },
    {
      "epoch": 4.4808387756347656e-05,
      "step": 46985,
      "training_step_time": 7.766877174377441
    },
    {
      "epoch": 4.480934143066406e-05,
      "model_forward_time": 1.7254016399383545,
      "step": 46986
    },
    {
      "epoch": 4.480934143066406e-05,
      "step": 46986,
      "training_step_time": 7.747094392776489
    },
    {
      "epoch": 4.481029510498047e-05,
      "model_forward_time": 1.7609620094299316,
      "step": 46987
    },
    {
      "epoch": 4.481029510498047e-05,
      "step": 46987,
      "training_step_time": 7.510235786437988
    },
    {
      "epoch": 4.4811248779296875e-05,
      "model_forward_time": 1.7120819091796875,
      "step": 46988
    },
    {
      "epoch": 4.4811248779296875e-05,
      "step": 46988,
      "training_step_time": 7.761561393737793
    },
    {
      "epoch": 4.481220245361328e-05,
      "model_forward_time": 1.7218749523162842,
      "step": 46989
    },
    {
      "epoch": 4.481220245361328e-05,
      "step": 46989,
      "training_step_time": 7.7997612953186035
    },
    {
      "epoch": 4.481315612792969e-05,
      "grad_norm": 0.12045969069004059,
      "learning_rate": 5.906191427336627e-06,
      "loss": 0.0588,
      "step": 46990
    },
    {
      "epoch": 4.481315612792969e-05,
      "model_forward_time": 2.285069465637207,
      "step": 46990
    },
    {
      "dynamics_loss_avg": 0.05350042805075646,
      "epoch": 4.481315612792969e-05,
      "step": 46990
    },
    {
      "action_loss_avg": 0.0013545456575229764,
      "epoch": 4.481315612792969e-05,
      "step": 46990
    },
    {
      "epoch": 4.481315612792969e-05,
      "step": 46990,
      "training_step_time": 7.703518867492676
    },
    {
      "epoch": 4.4814109802246093e-05,
      "model_forward_time": 1.7251224517822266,
      "step": 46991
    },
    {
      "epoch": 4.4814109802246093e-05,
      "step": 46991,
      "training_step_time": 7.716766357421875
    },
    {
      "epoch": 4.48150634765625e-05,
      "model_forward_time": 1.7678380012512207,
      "step": 46992
    },
    {
      "epoch": 4.48150634765625e-05,
      "step": 46992,
      "training_step_time": 9.185796737670898
    },
    {
      "epoch": 4.481601715087891e-05,
      "model_forward_time": 1.7179853916168213,
      "step": 46993
    },
    {
      "epoch": 4.481601715087891e-05,
      "step": 46993,
      "training_step_time": 8.270508766174316
    },
    {
      "epoch": 4.481697082519531e-05,
      "model_forward_time": 1.728696346282959,
      "step": 46994
    },
    {
      "epoch": 4.481697082519531e-05,
      "step": 46994,
      "training_step_time": 7.697355508804321
    },
    {
      "epoch": 4.4817924499511716e-05,
      "model_forward_time": 1.7317264080047607,
      "step": 46995
    },
    {
      "epoch": 4.4817924499511716e-05,
      "step": 46995,
      "training_step_time": 7.710610866546631
    },
    {
      "epoch": 4.481887817382813e-05,
      "model_forward_time": 1.7212672233581543,
      "step": 46996
    },
    {
      "epoch": 4.481887817382813e-05,
      "step": 46996,
      "training_step_time": 7.5341901779174805
    },
    {
      "epoch": 4.481983184814453e-05,
      "model_forward_time": 1.7312061786651611,
      "step": 46997
    },
    {
      "epoch": 4.481983184814453e-05,
      "step": 46997,
      "training_step_time": 7.762788772583008
    },
    {
      "epoch": 4.4820785522460935e-05,
      "model_forward_time": 1.3612995147705078,
      "step": 46998
    },
    {
      "epoch": 4.4820785522460935e-05,
      "step": 46998,
      "training_step_time": 8.07804822921753
    },
    {
      "epoch": 4.4821739196777346e-05,
      "model_forward_time": 2.2468011379241943,
      "step": 46999
    },
    {
      "epoch": 4.4821739196777346e-05,
      "step": 46999,
      "training_step_time": 7.883687973022461
    },
    {
      "epoch": 4.482269287109375e-05,
      "grad_norm": 0.1647108644247055,
      "learning_rate": 5.904565290761684e-06,
      "loss": 0.0626,
      "step": 47000
    },
    {
      "epoch": 4.482269287109375e-05,
      "model_forward_time": 1.7647924423217773,
      "step": 47000
    },
    {
      "dynamics_loss_avg": 0.04689605850726366,
      "epoch": 4.482269287109375e-05,
      "step": 47000
    },
    {
      "action_loss_avg": 0.0011348281084792688,
      "epoch": 4.482269287109375e-05,
      "step": 47000
    },
    {
      "epoch": 4.482269287109375e-05,
      "step": 47000,
      "training_step_time": 8.805752515792847
    },
    {
      "epoch": 4.4823646545410154e-05,
      "model_forward_time": 1.7365636825561523,
      "step": 47001
    },
    {
      "epoch": 4.4823646545410154e-05,
      "step": 47001,
      "training_step_time": 11.05720567703247
    },
    {
      "epoch": 4.4824600219726565e-05,
      "model_forward_time": 1.7209570407867432,
      "step": 47002
    },
    {
      "epoch": 4.4824600219726565e-05,
      "step": 47002,
      "training_step_time": 7.830969572067261
    },
    {
      "epoch": 4.482555389404297e-05,
      "model_forward_time": 1.7315700054168701,
      "step": 47003
    },
    {
      "epoch": 4.482555389404297e-05,
      "step": 47003,
      "training_step_time": 9.300018787384033
    },
    {
      "epoch": 4.482650756835937e-05,
      "model_forward_time": 1.7249407768249512,
      "step": 47004
    },
    {
      "epoch": 4.482650756835937e-05,
      "step": 47004,
      "training_step_time": 9.689035654067993
    },
    {
      "epoch": 4.4827461242675784e-05,
      "model_forward_time": 1.7207422256469727,
      "step": 47005
    },
    {
      "epoch": 4.4827461242675784e-05,
      "step": 47005,
      "training_step_time": 7.5118207931518555
    },
    {
      "epoch": 4.482841491699219e-05,
      "model_forward_time": 1.7161340713500977,
      "step": 47006
    },
    {
      "epoch": 4.482841491699219e-05,
      "step": 47006,
      "training_step_time": 11.005492925643921
    },
    {
      "epoch": 4.482936859130859e-05,
      "model_forward_time": 1.7266900539398193,
      "step": 47007
    },
    {
      "epoch": 4.482936859130859e-05,
      "step": 47007,
      "training_step_time": 8.925041198730469
    },
    {
      "epoch": 4.4830322265625e-05,
      "model_forward_time": 2.7490017414093018,
      "step": 47008
    },
    {
      "epoch": 4.4830322265625e-05,
      "step": 47008,
      "training_step_time": 9.37179160118103
    },
    {
      "epoch": 4.4831275939941407e-05,
      "model_forward_time": 1.7240829467773438,
      "step": 47009
    },
    {
      "epoch": 4.4831275939941407e-05,
      "step": 47009,
      "training_step_time": 8.559220790863037
    },
    {
      "epoch": 4.483222961425781e-05,
      "grad_norm": 0.13364572823047638,
      "learning_rate": 5.902939055264838e-06,
      "loss": 0.0559,
      "step": 47010
    },
    {
      "epoch": 4.483222961425781e-05,
      "model_forward_time": 1.7288846969604492,
      "step": 47010
    },
    {
      "dynamics_loss_avg": 0.05248395204544067,
      "epoch": 4.483222961425781e-05,
      "step": 47010
    },
    {
      "action_loss_avg": 0.0016599054302787407,
      "epoch": 4.483222961425781e-05,
      "step": 47010
    },
    {
      "epoch": 4.483222961425781e-05,
      "step": 47010,
      "training_step_time": 8.55864405632019
    },
    {
      "epoch": 4.483318328857422e-05,
      "model_forward_time": 1.725865125656128,
      "step": 47011
    },
    {
      "epoch": 4.483318328857422e-05,
      "step": 47011,
      "training_step_time": 7.657146215438843
    },
    {
      "epoch": 4.4834136962890625e-05,
      "model_forward_time": 1.3444788455963135,
      "step": 47012
    },
    {
      "epoch": 4.4834136962890625e-05,
      "step": 47012,
      "training_step_time": 7.707121849060059
    },
    {
      "epoch": 4.483509063720703e-05,
      "model_forward_time": 1.7201685905456543,
      "step": 47013
    },
    {
      "epoch": 4.483509063720703e-05,
      "step": 47013,
      "training_step_time": 8.182182550430298
    },
    {
      "epoch": 4.483604431152344e-05,
      "model_forward_time": 1.720149040222168,
      "step": 47014
    },
    {
      "epoch": 4.483604431152344e-05,
      "step": 47014,
      "training_step_time": 8.431249618530273
    },
    {
      "epoch": 4.4836997985839844e-05,
      "model_forward_time": 1.7263081073760986,
      "step": 47015
    },
    {
      "epoch": 4.4836997985839844e-05,
      "step": 47015,
      "training_step_time": 8.605356693267822
    },
    {
      "epoch": 4.483795166015625e-05,
      "model_forward_time": 1.7252638339996338,
      "step": 47016
    },
    {
      "epoch": 4.483795166015625e-05,
      "step": 47016,
      "training_step_time": 8.422909021377563
    },
    {
      "epoch": 4.483890533447266e-05,
      "model_forward_time": 2.277297258377075,
      "step": 47017
    },
    {
      "epoch": 4.483890533447266e-05,
      "step": 47017,
      "training_step_time": 8.211304903030396
    },
    {
      "epoch": 4.483985900878906e-05,
      "model_forward_time": 1.711355447769165,
      "step": 47018
    },
    {
      "epoch": 4.483985900878906e-05,
      "step": 47018,
      "training_step_time": 7.613618612289429
    },
    {
      "epoch": 4.484081268310547e-05,
      "model_forward_time": 1.717841386795044,
      "step": 47019
    },
    {
      "epoch": 4.484081268310547e-05,
      "step": 47019,
      "training_step_time": 7.726146459579468
    },
    {
      "epoch": 4.484176635742188e-05,
      "grad_norm": 0.11179682612419128,
      "learning_rate": 5.901312721023935e-06,
      "loss": 0.0584,
      "step": 47020
    },
    {
      "epoch": 4.484176635742188e-05,
      "model_forward_time": 1.7683215141296387,
      "step": 47020
    },
    {
      "dynamics_loss_avg": 0.049157388228923085,
      "epoch": 4.484176635742188e-05,
      "step": 47020
    },
    {
      "action_loss_avg": 0.0016320168564561755,
      "epoch": 4.484176635742188e-05,
      "step": 47020
    },
    {
      "epoch": 4.484176635742188e-05,
      "step": 47020,
      "training_step_time": 7.710284948348999
    },
    {
      "epoch": 4.484272003173828e-05,
      "model_forward_time": 1.714237928390503,
      "step": 47021
    },
    {
      "epoch": 4.484272003173828e-05,
      "step": 47021,
      "training_step_time": 7.705460548400879
    },
    {
      "epoch": 4.4843673706054686e-05,
      "model_forward_time": 1.7196319103240967,
      "step": 47022
    },
    {
      "epoch": 4.4843673706054686e-05,
      "step": 47022,
      "training_step_time": 7.68198561668396
    },
    {
      "epoch": 4.48446273803711e-05,
      "model_forward_time": 1.726820707321167,
      "step": 47023
    },
    {
      "epoch": 4.48446273803711e-05,
      "step": 47023,
      "training_step_time": 7.724059820175171
    },
    {
      "epoch": 4.48455810546875e-05,
      "model_forward_time": 1.7141435146331787,
      "step": 47024
    },
    {
      "epoch": 4.48455810546875e-05,
      "step": 47024,
      "training_step_time": 7.52524995803833
    },
    {
      "epoch": 4.4846534729003905e-05,
      "model_forward_time": 1.721210241317749,
      "step": 47025
    },
    {
      "epoch": 4.4846534729003905e-05,
      "step": 47025,
      "training_step_time": 7.764949798583984
    },
    {
      "epoch": 4.4847488403320316e-05,
      "model_forward_time": 1.7282781600952148,
      "step": 47026
    },
    {
      "epoch": 4.4847488403320316e-05,
      "step": 47026,
      "training_step_time": 9.271809577941895
    },
    {
      "epoch": 4.484844207763672e-05,
      "model_forward_time": 2.2622127532958984,
      "step": 47027
    },
    {
      "epoch": 4.484844207763672e-05,
      "step": 47027,
      "training_step_time": 8.259654998779297
    },
    {
      "epoch": 4.4849395751953124e-05,
      "model_forward_time": 1.7287333011627197,
      "step": 47028
    },
    {
      "epoch": 4.4849395751953124e-05,
      "step": 47028,
      "training_step_time": 8.708174705505371
    },
    {
      "epoch": 4.4850349426269535e-05,
      "model_forward_time": 1.7721669673919678,
      "step": 47029
    },
    {
      "epoch": 4.4850349426269535e-05,
      "step": 47029,
      "training_step_time": 9.207084655761719
    },
    {
      "epoch": 4.485130310058594e-05,
      "grad_norm": 0.11687889695167542,
      "learning_rate": 5.899686288216829e-06,
      "loss": 0.0626,
      "step": 47030
    },
    {
      "epoch": 4.485130310058594e-05,
      "model_forward_time": 1.7231707572937012,
      "step": 47030
    },
    {
      "dynamics_loss_avg": 0.05755317248404026,
      "epoch": 4.485130310058594e-05,
      "step": 47030
    },
    {
      "action_loss_avg": 0.00184671517345123,
      "epoch": 4.485130310058594e-05,
      "step": 47030
    },
    {
      "epoch": 4.485130310058594e-05,
      "step": 47030,
      "training_step_time": 9.016793727874756
    },
    {
      "epoch": 4.485225677490234e-05,
      "model_forward_time": 1.7194163799285889,
      "step": 47031
    },
    {
      "epoch": 4.485225677490234e-05,
      "step": 47031,
      "training_step_time": 7.660751819610596
    },
    {
      "epoch": 4.485321044921875e-05,
      "model_forward_time": 1.7282884120941162,
      "step": 47032
    },
    {
      "epoch": 4.485321044921875e-05,
      "step": 47032,
      "training_step_time": 7.656630039215088
    },
    {
      "epoch": 4.485416412353516e-05,
      "model_forward_time": 1.717132329940796,
      "step": 47033
    },
    {
      "epoch": 4.485416412353516e-05,
      "step": 47033,
      "training_step_time": 7.480010032653809
    },
    {
      "epoch": 4.485511779785156e-05,
      "model_forward_time": 1.7213118076324463,
      "step": 47034
    },
    {
      "epoch": 4.485511779785156e-05,
      "step": 47034,
      "training_step_time": 7.724429130554199
    },
    {
      "epoch": 4.4856071472167965e-05,
      "model_forward_time": 1.3416860103607178,
      "step": 47035
    },
    {
      "epoch": 4.4856071472167965e-05,
      "step": 47035,
      "training_step_time": 9.142085313796997
    },
    {
      "epoch": 4.4857025146484376e-05,
      "model_forward_time": 2.275486469268799,
      "step": 47036
    },
    {
      "epoch": 4.4857025146484376e-05,
      "step": 47036,
      "training_step_time": 8.362012147903442
    },
    {
      "epoch": 4.485797882080078e-05,
      "model_forward_time": 1.7300329208374023,
      "step": 47037
    },
    {
      "epoch": 4.485797882080078e-05,
      "step": 47037,
      "training_step_time": 8.66624140739441
    },
    {
      "epoch": 4.4858932495117184e-05,
      "model_forward_time": 1.7686212062835693,
      "step": 47038
    },
    {
      "epoch": 4.4858932495117184e-05,
      "step": 47038,
      "training_step_time": 9.572012186050415
    },
    {
      "epoch": 4.4859886169433595e-05,
      "model_forward_time": 1.72983717918396,
      "step": 47039
    },
    {
      "epoch": 4.4859886169433595e-05,
      "step": 47039,
      "training_step_time": 9.015286445617676
    },
    {
      "epoch": 4.486083984375e-05,
      "grad_norm": 0.12283017486333847,
      "learning_rate": 5.8980597570213824e-06,
      "loss": 0.0631,
      "step": 47040
    },
    {
      "epoch": 4.486083984375e-05,
      "model_forward_time": 1.717175006866455,
      "step": 47040
    },
    {
      "dynamics_loss_avg": 0.06324052661657334,
      "epoch": 4.486083984375e-05,
      "step": 47040
    },
    {
      "action_loss_avg": 0.0028775823884643614,
      "epoch": 4.486083984375e-05,
      "step": 47040
    },
    {
      "epoch": 4.486083984375e-05,
      "step": 47040,
      "training_step_time": 7.797398090362549
    },
    {
      "epoch": 4.48617935180664e-05,
      "model_forward_time": 1.7288756370544434,
      "step": 47041
    },
    {
      "epoch": 4.48617935180664e-05,
      "step": 47041,
      "training_step_time": 7.63004994392395
    },
    {
      "epoch": 4.4862747192382814e-05,
      "model_forward_time": 1.7144758701324463,
      "step": 47042
    },
    {
      "epoch": 4.4862747192382814e-05,
      "step": 47042,
      "training_step_time": 7.43783712387085
    },
    {
      "epoch": 4.486370086669922e-05,
      "model_forward_time": 1.7229490280151367,
      "step": 47043
    },
    {
      "epoch": 4.486370086669922e-05,
      "step": 47043,
      "training_step_time": 8.77930498123169
    },
    {
      "epoch": 4.486465454101562e-05,
      "model_forward_time": 1.7293860912322998,
      "step": 47044
    },
    {
      "epoch": 4.486465454101562e-05,
      "step": 47044,
      "training_step_time": 8.364110231399536
    },
    {
      "epoch": 4.486560821533203e-05,
      "model_forward_time": 2.2767369747161865,
      "step": 47045
    },
    {
      "epoch": 4.486560821533203e-05,
      "step": 47045,
      "training_step_time": 8.638784408569336
    },
    {
      "epoch": 4.486656188964844e-05,
      "model_forward_time": 1.731780767440796,
      "step": 47046
    },
    {
      "epoch": 4.486656188964844e-05,
      "step": 47046,
      "training_step_time": 9.485774040222168
    },
    {
      "epoch": 4.486751556396484e-05,
      "model_forward_time": 1.7236237525939941,
      "step": 47047
    },
    {
      "epoch": 4.486751556396484e-05,
      "step": 47047,
      "training_step_time": 6.921268463134766
    },
    {
      "epoch": 4.486846923828125e-05,
      "model_forward_time": 1.728470802307129,
      "step": 47048
    },
    {
      "epoch": 4.486846923828125e-05,
      "step": 47048,
      "training_step_time": 6.935269832611084
    },
    {
      "epoch": 4.4869422912597656e-05,
      "model_forward_time": 1.7397379875183105,
      "step": 47049
    },
    {
      "epoch": 4.4869422912597656e-05,
      "step": 47049,
      "training_step_time": 6.90470814704895
    },
    {
      "epoch": 4.487037658691406e-05,
      "grad_norm": 0.1669522076845169,
      "learning_rate": 5.896433127615471e-06,
      "loss": 0.0578,
      "step": 47050
    },
    {
      "epoch": 4.487037658691406e-05,
      "model_forward_time": 1.72227144241333,
      "step": 47050
    },
    {
      "dynamics_loss_avg": 0.057522886991500856,
      "epoch": 4.487037658691406e-05,
      "step": 47050
    },
    {
      "action_loss_avg": 0.001208038319600746,
      "epoch": 4.487037658691406e-05,
      "step": 47050
    },
    {
      "epoch": 4.487037658691406e-05,
      "step": 47050,
      "training_step_time": 6.906553506851196
    },
    {
      "epoch": 4.487133026123047e-05,
      "model_forward_time": 1.7238917350769043,
      "step": 47051
    },
    {
      "epoch": 4.487133026123047e-05,
      "step": 47051,
      "training_step_time": 6.930664777755737
    },
    {
      "epoch": 4.4872283935546875e-05,
      "model_forward_time": 1.722081184387207,
      "step": 47052
    },
    {
      "epoch": 4.4872283935546875e-05,
      "step": 47052,
      "training_step_time": 6.909221649169922
    },
    {
      "epoch": 4.487323760986328e-05,
      "model_forward_time": 1.717529535293579,
      "step": 47053
    },
    {
      "epoch": 4.487323760986328e-05,
      "step": 47053,
      "training_step_time": 6.874427795410156
    },
    {
      "epoch": 4.487419128417969e-05,
      "model_forward_time": 1.7367444038391113,
      "step": 47054
    },
    {
      "epoch": 4.487419128417969e-05,
      "step": 47054,
      "training_step_time": 6.898874044418335
    },
    {
      "epoch": 4.4875144958496093e-05,
      "model_forward_time": 1.93585205078125,
      "step": 47055
    },
    {
      "epoch": 4.4875144958496093e-05,
      "step": 47055,
      "training_step_time": 7.522933483123779
    },
    {
      "epoch": 4.48760986328125e-05,
      "model_forward_time": 1.7217152118682861,
      "step": 47056
    },
    {
      "epoch": 4.48760986328125e-05,
      "step": 47056,
      "training_step_time": 7.9225170612335205
    },
    {
      "epoch": 4.487705230712891e-05,
      "model_forward_time": 1.7193386554718018,
      "step": 47057
    },
    {
      "epoch": 4.487705230712891e-05,
      "step": 47057,
      "training_step_time": 6.902442455291748
    },
    {
      "epoch": 4.487800598144531e-05,
      "model_forward_time": 1.7275049686431885,
      "step": 47058
    },
    {
      "epoch": 4.487800598144531e-05,
      "step": 47058,
      "training_step_time": 6.899099826812744
    },
    {
      "epoch": 4.4878959655761716e-05,
      "model_forward_time": 1.7228507995605469,
      "step": 47059
    },
    {
      "epoch": 4.4878959655761716e-05,
      "step": 47059,
      "training_step_time": 6.8786468505859375
    },
    {
      "epoch": 4.487991333007813e-05,
      "grad_norm": 0.14508630335330963,
      "learning_rate": 5.894806400176981e-06,
      "loss": 0.062,
      "step": 47060
    },
    {
      "epoch": 4.487991333007813e-05,
      "model_forward_time": 1.7157747745513916,
      "step": 47060
    },
    {
      "dynamics_loss_avg": 0.06843462847173214,
      "epoch": 4.487991333007813e-05,
      "step": 47060
    },
    {
      "action_loss_avg": 0.0015461608942132442,
      "epoch": 4.487991333007813e-05,
      "step": 47060
    },
    {
      "epoch": 4.487991333007813e-05,
      "step": 47060,
      "training_step_time": 6.877789735794067
    },
    {
      "epoch": 4.488086700439453e-05,
      "model_forward_time": 1.721236228942871,
      "step": 47061
    },
    {
      "epoch": 4.488086700439453e-05,
      "step": 47061,
      "training_step_time": 6.8614442348480225
    },
    {
      "epoch": 4.4881820678710935e-05,
      "model_forward_time": 1.7195115089416504,
      "step": 47062
    },
    {
      "epoch": 4.4881820678710935e-05,
      "step": 47062,
      "training_step_time": 6.846645355224609
    },
    {
      "epoch": 4.4882774353027346e-05,
      "model_forward_time": 1.7309319972991943,
      "step": 47063
    },
    {
      "epoch": 4.4882774353027346e-05,
      "step": 47063,
      "training_step_time": 6.902502059936523
    },
    {
      "epoch": 4.488372802734375e-05,
      "model_forward_time": 2.2646644115448,
      "step": 47064
    },
    {
      "epoch": 4.488372802734375e-05,
      "step": 47064,
      "training_step_time": 7.534012079238892
    },
    {
      "epoch": 4.4884681701660154e-05,
      "model_forward_time": 1.7205517292022705,
      "step": 47065
    },
    {
      "epoch": 4.4884681701660154e-05,
      "step": 47065,
      "training_step_time": 7.771632432937622
    },
    {
      "epoch": 4.4885635375976565e-05,
      "model_forward_time": 1.7239718437194824,
      "step": 47066
    },
    {
      "epoch": 4.4885635375976565e-05,
      "step": 47066,
      "training_step_time": 7.676955938339233
    },
    {
      "epoch": 4.488658905029297e-05,
      "model_forward_time": 1.7248475551605225,
      "step": 47067
    },
    {
      "epoch": 4.488658905029297e-05,
      "step": 47067,
      "training_step_time": 6.922312259674072
    },
    {
      "epoch": 4.488754272460937e-05,
      "model_forward_time": 1.718172550201416,
      "step": 47068
    },
    {
      "epoch": 4.488754272460937e-05,
      "step": 47068,
      "training_step_time": 6.895263195037842
    },
    {
      "epoch": 4.4888496398925784e-05,
      "model_forward_time": 1.7163689136505127,
      "step": 47069
    },
    {
      "epoch": 4.4888496398925784e-05,
      "step": 47069,
      "training_step_time": 6.9384605884552
    },
    {
      "epoch": 4.488945007324219e-05,
      "grad_norm": 0.12400812655687332,
      "learning_rate": 5.893179574883808e-06,
      "loss": 0.0623,
      "step": 47070
    },
    {
      "epoch": 4.488945007324219e-05,
      "model_forward_time": 1.7158622741699219,
      "step": 47070
    },
    {
      "dynamics_loss_avg": 0.07133272513747216,
      "epoch": 4.488945007324219e-05,
      "step": 47070
    },
    {
      "action_loss_avg": 0.001732395745057147,
      "epoch": 4.488945007324219e-05,
      "step": 47070
    },
    {
      "epoch": 4.488945007324219e-05,
      "step": 47070,
      "training_step_time": 6.895609140396118
    },
    {
      "epoch": 4.489040374755859e-05,
      "model_forward_time": 1.7441542148590088,
      "step": 47071
    },
    {
      "epoch": 4.489040374755859e-05,
      "step": 47071,
      "training_step_time": 6.928951263427734
    },
    {
      "epoch": 4.4891357421875e-05,
      "model_forward_time": 1.7168056964874268,
      "step": 47072
    },
    {
      "epoch": 4.4891357421875e-05,
      "step": 47072,
      "training_step_time": 6.862097263336182
    },
    {
      "epoch": 4.489231109619141e-05,
      "model_forward_time": 2.2884562015533447,
      "step": 47073
    },
    {
      "epoch": 4.489231109619141e-05,
      "step": 47073,
      "training_step_time": 7.672952890396118
    },
    {
      "epoch": 4.489326477050781e-05,
      "model_forward_time": 1.7221088409423828,
      "step": 47074
    },
    {
      "epoch": 4.489326477050781e-05,
      "step": 47074,
      "training_step_time": 8.764693021774292
    },
    {
      "epoch": 4.489421844482422e-05,
      "model_forward_time": 1.7162437438964844,
      "step": 47075
    },
    {
      "epoch": 4.489421844482422e-05,
      "step": 47075,
      "training_step_time": 7.725420713424683
    },
    {
      "epoch": 4.4895172119140625e-05,
      "model_forward_time": 1.723625659942627,
      "step": 47076
    },
    {
      "epoch": 4.4895172119140625e-05,
      "step": 47076,
      "training_step_time": 6.91920804977417
    },
    {
      "epoch": 4.489612579345703e-05,
      "model_forward_time": 1.7212560176849365,
      "step": 47077
    },
    {
      "epoch": 4.489612579345703e-05,
      "step": 47077,
      "training_step_time": 6.901723623275757
    },
    {
      "epoch": 4.489707946777344e-05,
      "model_forward_time": 1.729612112045288,
      "step": 47078
    },
    {
      "epoch": 4.489707946777344e-05,
      "step": 47078,
      "training_step_time": 6.92123007774353
    },
    {
      "epoch": 4.4898033142089844e-05,
      "model_forward_time": 1.7285971641540527,
      "step": 47079
    },
    {
      "epoch": 4.4898033142089844e-05,
      "step": 47079,
      "training_step_time": 6.712207078933716
    },
    {
      "epoch": 4.489898681640625e-05,
      "grad_norm": 0.128114253282547,
      "learning_rate": 5.891552651913859e-06,
      "loss": 0.0583,
      "step": 47080
    },
    {
      "epoch": 4.489898681640625e-05,
      "model_forward_time": 1.7124004364013672,
      "step": 47080
    },
    {
      "dynamics_loss_avg": 0.05822870545089245,
      "epoch": 4.489898681640625e-05,
      "step": 47080
    },
    {
      "action_loss_avg": 0.0023695377050898967,
      "epoch": 4.489898681640625e-05,
      "step": 47080
    },
    {
      "epoch": 4.489898681640625e-05,
      "step": 47080,
      "training_step_time": 6.9125025272369385
    },
    {
      "epoch": 4.489994049072266e-05,
      "model_forward_time": 1.7231783866882324,
      "step": 47081
    },
    {
      "epoch": 4.489994049072266e-05,
      "step": 47081,
      "training_step_time": 6.904567241668701
    },
    {
      "epoch": 4.490089416503906e-05,
      "model_forward_time": 2.290222406387329,
      "step": 47082
    },
    {
      "epoch": 4.490089416503906e-05,
      "step": 47082,
      "training_step_time": 7.635523080825806
    },
    {
      "epoch": 4.490184783935547e-05,
      "model_forward_time": 1.7243363857269287,
      "step": 47083
    },
    {
      "epoch": 4.490184783935547e-05,
      "step": 47083,
      "training_step_time": 9.045346021652222
    },
    {
      "epoch": 4.490280151367188e-05,
      "model_forward_time": 1.7190179824829102,
      "step": 47084
    },
    {
      "epoch": 4.490280151367188e-05,
      "step": 47084,
      "training_step_time": 7.899309396743774
    },
    {
      "epoch": 4.490375518798828e-05,
      "model_forward_time": 1.3387141227722168,
      "step": 47085
    },
    {
      "epoch": 4.490375518798828e-05,
      "step": 47085,
      "training_step_time": 7.553420305252075
    },
    {
      "epoch": 4.4904708862304686e-05,
      "model_forward_time": 1.7221519947052002,
      "step": 47086
    },
    {
      "epoch": 4.4904708862304686e-05,
      "step": 47086,
      "training_step_time": 6.876166105270386
    },
    {
      "epoch": 4.49056625366211e-05,
      "model_forward_time": 1.7179522514343262,
      "step": 47087
    },
    {
      "epoch": 4.49056625366211e-05,
      "step": 47087,
      "training_step_time": 6.8723673820495605
    },
    {
      "epoch": 4.49066162109375e-05,
      "model_forward_time": 1.7187447547912598,
      "step": 47088
    },
    {
      "epoch": 4.49066162109375e-05,
      "step": 47088,
      "training_step_time": 6.877811670303345
    },
    {
      "epoch": 4.4907569885253905e-05,
      "model_forward_time": 1.7194445133209229,
      "step": 47089
    },
    {
      "epoch": 4.4907569885253905e-05,
      "step": 47089,
      "training_step_time": 6.885049819946289
    },
    {
      "epoch": 4.4908523559570316e-05,
      "grad_norm": 0.1497461050748825,
      "learning_rate": 5.889925631445054e-06,
      "loss": 0.0614,
      "step": 47090
    },
    {
      "epoch": 4.4908523559570316e-05,
      "model_forward_time": 1.7177443504333496,
      "step": 47090
    },
    {
      "dynamics_loss_avg": 0.05533564444631338,
      "epoch": 4.4908523559570316e-05,
      "step": 47090
    },
    {
      "action_loss_avg": 0.0014939664804842323,
      "epoch": 4.4908523559570316e-05,
      "step": 47090
    },
    {
      "epoch": 4.4908523559570316e-05,
      "step": 47090,
      "training_step_time": 6.87352728843689
    },
    {
      "epoch": 4.490947723388672e-05,
      "model_forward_time": 1.758467674255371,
      "step": 47091
    },
    {
      "epoch": 4.490947723388672e-05,
      "step": 47091,
      "training_step_time": 7.627381801605225
    },
    {
      "epoch": 4.4910430908203124e-05,
      "model_forward_time": 1.873211145401001,
      "step": 47092
    },
    {
      "epoch": 4.4910430908203124e-05,
      "step": 47092,
      "training_step_time": 10.45110297203064
    },
    {
      "epoch": 4.4911384582519535e-05,
      "model_forward_time": 1.7150912284851074,
      "step": 47093
    },
    {
      "epoch": 4.4911384582519535e-05,
      "step": 47093,
      "training_step_time": 8.851897954940796
    },
    {
      "epoch": 4.491233825683594e-05,
      "model_forward_time": 1.7212989330291748,
      "step": 47094
    },
    {
      "epoch": 4.491233825683594e-05,
      "step": 47094,
      "training_step_time": 7.58370304107666
    },
    {
      "epoch": 4.491329193115234e-05,
      "model_forward_time": 1.7459607124328613,
      "step": 47095
    },
    {
      "epoch": 4.491329193115234e-05,
      "step": 47095,
      "training_step_time": 6.835849046707153
    },
    {
      "epoch": 4.491424560546875e-05,
      "model_forward_time": 1.3567891120910645,
      "step": 47096
    },
    {
      "epoch": 4.491424560546875e-05,
      "step": 47096,
      "training_step_time": 6.90752100944519
    },
    {
      "epoch": 4.491519927978516e-05,
      "model_forward_time": 1.7304983139038086,
      "step": 47097
    },
    {
      "epoch": 4.491519927978516e-05,
      "step": 47097,
      "training_step_time": 6.8780248165130615
    },
    {
      "epoch": 4.491615295410156e-05,
      "model_forward_time": 1.7291810512542725,
      "step": 47098
    },
    {
      "epoch": 4.491615295410156e-05,
      "step": 47098,
      "training_step_time": 6.874493837356567
    },
    {
      "epoch": 4.4917106628417966e-05,
      "model_forward_time": 1.7259607315063477,
      "step": 47099
    },
    {
      "epoch": 4.4917106628417966e-05,
      "step": 47099,
      "training_step_time": 6.903004169464111
    },
    {
      "epoch": 4.4918060302734376e-05,
      "grad_norm": 0.12034225463867188,
      "learning_rate": 5.888298513655318e-06,
      "loss": 0.0598,
      "step": 47100
    },
    {
      "epoch": 4.4918060302734376e-05,
      "model_forward_time": 1.727069616317749,
      "step": 47100
    },
    {
      "dynamics_loss_avg": 0.05184829840436578,
      "epoch": 4.4918060302734376e-05,
      "step": 47100
    },
    {
      "action_loss_avg": 0.0018675987201277166,
      "epoch": 4.4918060302734376e-05,
      "step": 47100
    },
    {
      "epoch": 4.4918060302734376e-05,
      "step": 47100,
      "training_step_time": 7.562300682067871
    },
    {
      "epoch": 4.491901397705078e-05,
      "model_forward_time": 2.2743682861328125,
      "step": 47101
    },
    {
      "epoch": 4.491901397705078e-05,
      "step": 47101,
      "training_step_time": 7.887171268463135
    },
    {
      "epoch": 4.4919967651367184e-05,
      "model_forward_time": 1.726975917816162,
      "step": 47102
    },
    {
      "epoch": 4.4919967651367184e-05,
      "step": 47102,
      "training_step_time": 7.9855382442474365
    },
    {
      "epoch": 4.4920921325683595e-05,
      "model_forward_time": 1.727820634841919,
      "step": 47103
    },
    {
      "epoch": 4.4920921325683595e-05,
      "step": 47103,
      "training_step_time": 7.745221376419067
    },
    {
      "epoch": 4.4921875e-05,
      "model_forward_time": 1.7184326648712158,
      "step": 47104
    },
    {
      "epoch": 4.4921875e-05,
      "step": 47104,
      "training_step_time": 6.902693033218384
    },
    {
      "epoch": 4.49228286743164e-05,
      "model_forward_time": 1.7211904525756836,
      "step": 47105
    },
    {
      "epoch": 4.49228286743164e-05,
      "step": 47105,
      "training_step_time": 6.906405925750732
    },
    {
      "epoch": 4.4923782348632814e-05,
      "model_forward_time": 1.722116231918335,
      "step": 47106
    },
    {
      "epoch": 4.4923782348632814e-05,
      "step": 47106,
      "training_step_time": 6.930561065673828
    },
    {
      "epoch": 4.492473602294922e-05,
      "model_forward_time": 1.7141284942626953,
      "step": 47107
    },
    {
      "epoch": 4.492473602294922e-05,
      "step": 47107,
      "training_step_time": 6.898339748382568
    },
    {
      "epoch": 4.492568969726562e-05,
      "model_forward_time": 1.7244832515716553,
      "step": 47108
    },
    {
      "epoch": 4.492568969726562e-05,
      "step": 47108,
      "training_step_time": 6.888203859329224
    },
    {
      "epoch": 4.492664337158203e-05,
      "model_forward_time": 1.7245113849639893,
      "step": 47109
    },
    {
      "epoch": 4.492664337158203e-05,
      "step": 47109,
      "training_step_time": 7.684558391571045
    },
    {
      "epoch": 4.492759704589844e-05,
      "grad_norm": 0.12769466638565063,
      "learning_rate": 5.8866712987225926e-06,
      "loss": 0.0583,
      "step": 47110
    },
    {
      "epoch": 4.492759704589844e-05,
      "model_forward_time": 2.3023457527160645,
      "step": 47110
    },
    {
      "dynamics_loss_avg": 0.049421404115855695,
      "epoch": 4.492759704589844e-05,
      "step": 47110
    },
    {
      "action_loss_avg": 0.0015063521947013213,
      "epoch": 4.492759704589844e-05,
      "step": 47110
    },
    {
      "epoch": 4.492759704589844e-05,
      "step": 47110,
      "training_step_time": 8.592698335647583
    },
    {
      "epoch": 4.492855072021484e-05,
      "model_forward_time": 1.7226436138153076,
      "step": 47111
    },
    {
      "epoch": 4.492855072021484e-05,
      "step": 47111,
      "training_step_time": 10.965832948684692
    },
    {
      "epoch": 4.492950439453125e-05,
      "model_forward_time": 1.724337100982666,
      "step": 47112
    },
    {
      "epoch": 4.492950439453125e-05,
      "step": 47112,
      "training_step_time": 8.954511880874634
    },
    {
      "epoch": 4.4930458068847656e-05,
      "model_forward_time": 1.7239556312561035,
      "step": 47113
    },
    {
      "epoch": 4.4930458068847656e-05,
      "step": 47113,
      "training_step_time": 7.097227096557617
    },
    {
      "epoch": 4.493141174316406e-05,
      "model_forward_time": 1.7214717864990234,
      "step": 47114
    },
    {
      "epoch": 4.493141174316406e-05,
      "step": 47114,
      "training_step_time": 7.055246591567993
    },
    {
      "epoch": 4.493236541748047e-05,
      "model_forward_time": 1.7184221744537354,
      "step": 47115
    },
    {
      "epoch": 4.493236541748047e-05,
      "step": 47115,
      "training_step_time": 7.032815933227539
    },
    {
      "epoch": 4.4933319091796875e-05,
      "model_forward_time": 1.73225736618042,
      "step": 47116
    },
    {
      "epoch": 4.4933319091796875e-05,
      "step": 47116,
      "training_step_time": 7.072699546813965
    },
    {
      "epoch": 4.493427276611328e-05,
      "model_forward_time": 1.7216508388519287,
      "step": 47117
    },
    {
      "epoch": 4.493427276611328e-05,
      "step": 47117,
      "training_step_time": 7.073394298553467
    },
    {
      "epoch": 4.493522644042969e-05,
      "model_forward_time": 1.7138915061950684,
      "step": 47118
    },
    {
      "epoch": 4.493522644042969e-05,
      "step": 47118,
      "training_step_time": 7.619189500808716
    },
    {
      "epoch": 4.4936180114746094e-05,
      "model_forward_time": 2.318326950073242,
      "step": 47119
    },
    {
      "epoch": 4.4936180114746094e-05,
      "step": 47119,
      "training_step_time": 7.908273220062256
    },
    {
      "epoch": 4.49371337890625e-05,
      "grad_norm": 0.14026159048080444,
      "learning_rate": 5.885043986824828e-06,
      "loss": 0.0576,
      "step": 47120
    },
    {
      "epoch": 4.49371337890625e-05,
      "model_forward_time": 1.7293996810913086,
      "step": 47120
    },
    {
      "dynamics_loss_avg": 0.05134914796799421,
      "epoch": 4.49371337890625e-05,
      "step": 47120
    },
    {
      "action_loss_avg": 0.0013752053957432509,
      "epoch": 4.49371337890625e-05,
      "step": 47120
    },
    {
      "epoch": 4.49371337890625e-05,
      "step": 47120,
      "training_step_time": 9.295341968536377
    },
    {
      "epoch": 4.493808746337891e-05,
      "model_forward_time": 1.7255725860595703,
      "step": 47121
    },
    {
      "epoch": 4.493808746337891e-05,
      "step": 47121,
      "training_step_time": 7.975294828414917
    },
    {
      "epoch": 4.493904113769531e-05,
      "model_forward_time": 1.7254698276519775,
      "step": 47122
    },
    {
      "epoch": 4.493904113769531e-05,
      "step": 47122,
      "training_step_time": 7.643576622009277
    },
    {
      "epoch": 4.4939994812011716e-05,
      "model_forward_time": 1.723731279373169,
      "step": 47123
    },
    {
      "epoch": 4.4939994812011716e-05,
      "step": 47123,
      "training_step_time": 7.078485012054443
    },
    {
      "epoch": 4.494094848632813e-05,
      "model_forward_time": 1.7247240543365479,
      "step": 47124
    },
    {
      "epoch": 4.494094848632813e-05,
      "step": 47124,
      "training_step_time": 7.057349920272827
    },
    {
      "epoch": 4.494190216064453e-05,
      "model_forward_time": 1.7220845222473145,
      "step": 47125
    },
    {
      "epoch": 4.494190216064453e-05,
      "step": 47125,
      "training_step_time": 7.145655870437622
    },
    {
      "epoch": 4.4942855834960935e-05,
      "model_forward_time": 1.7282600402832031,
      "step": 47126
    },
    {
      "epoch": 4.4942855834960935e-05,
      "step": 47126,
      "training_step_time": 6.930543422698975
    },
    {
      "epoch": 4.4943809509277346e-05,
      "model_forward_time": 1.728065013885498,
      "step": 47127
    },
    {
      "epoch": 4.4943809509277346e-05,
      "step": 47127,
      "training_step_time": 7.996132135391235
    },
    {
      "epoch": 4.494476318359375e-05,
      "model_forward_time": 1.771285057067871,
      "step": 47128
    },
    {
      "epoch": 4.494476318359375e-05,
      "step": 47128,
      "training_step_time": 9.324727058410645
    },
    {
      "epoch": 4.4945716857910154e-05,
      "model_forward_time": 2.2725443840026855,
      "step": 47129
    },
    {
      "epoch": 4.4945716857910154e-05,
      "step": 47129,
      "training_step_time": 8.647247076034546
    },
    {
      "epoch": 4.4946670532226565e-05,
      "grad_norm": 0.1598224937915802,
      "learning_rate": 5.8834165781399824e-06,
      "loss": 0.0588,
      "step": 47130
    },
    {
      "epoch": 4.4946670532226565e-05,
      "model_forward_time": 1.7186024188995361,
      "step": 47130
    },
    {
      "dynamics_loss_avg": 0.058993093110620974,
      "epoch": 4.4946670532226565e-05,
      "step": 47130
    },
    {
      "action_loss_avg": 0.0020834588271100075,
      "epoch": 4.4946670532226565e-05,
      "step": 47130
    },
    {
      "epoch": 4.4946670532226565e-05,
      "step": 47130,
      "training_step_time": 8.038398265838623
    },
    {
      "epoch": 4.494762420654297e-05,
      "model_forward_time": 1.7261834144592285,
      "step": 47131
    },
    {
      "epoch": 4.494762420654297e-05,
      "step": 47131,
      "training_step_time": 7.842867612838745
    },
    {
      "epoch": 4.494857788085937e-05,
      "model_forward_time": 1.7219741344451904,
      "step": 47132
    },
    {
      "epoch": 4.494857788085937e-05,
      "step": 47132,
      "training_step_time": 7.114078998565674
    },
    {
      "epoch": 4.4949531555175784e-05,
      "model_forward_time": 1.3497223854064941,
      "step": 47133
    },
    {
      "epoch": 4.4949531555175784e-05,
      "step": 47133,
      "training_step_time": 7.070039510726929
    },
    {
      "epoch": 4.495048522949219e-05,
      "model_forward_time": 1.7170801162719727,
      "step": 47134
    },
    {
      "epoch": 4.495048522949219e-05,
      "step": 47134,
      "training_step_time": 7.154036283493042
    },
    {
      "epoch": 4.495143890380859e-05,
      "model_forward_time": 1.7210845947265625,
      "step": 47135
    },
    {
      "epoch": 4.495143890380859e-05,
      "step": 47135,
      "training_step_time": 7.099069833755493
    },
    {
      "epoch": 4.4952392578125e-05,
      "model_forward_time": 1.7265300750732422,
      "step": 47136
    },
    {
      "epoch": 4.4952392578125e-05,
      "step": 47136,
      "training_step_time": 7.048715591430664
    },
    {
      "epoch": 4.495334625244141e-05,
      "model_forward_time": 1.7622337341308594,
      "step": 47137
    },
    {
      "epoch": 4.495334625244141e-05,
      "step": 47137,
      "training_step_time": 8.70408034324646
    },
    {
      "epoch": 4.495429992675781e-05,
      "model_forward_time": 1.9130833148956299,
      "step": 47138
    },
    {
      "epoch": 4.495429992675781e-05,
      "step": 47138,
      "training_step_time": 9.761475563049316
    },
    {
      "epoch": 4.495525360107422e-05,
      "model_forward_time": 1.7293701171875,
      "step": 47139
    },
    {
      "epoch": 4.495525360107422e-05,
      "step": 47139,
      "training_step_time": 7.986419677734375
    },
    {
      "epoch": 4.4956207275390626e-05,
      "grad_norm": 0.1136360839009285,
      "learning_rate": 5.8817890728460295e-06,
      "loss": 0.0585,
      "step": 47140
    },
    {
      "epoch": 4.4956207275390626e-05,
      "model_forward_time": 1.730933427810669,
      "step": 47140
    },
    {
      "dynamics_loss_avg": 0.054839836247265336,
      "epoch": 4.4956207275390626e-05,
      "step": 47140
    },
    {
      "action_loss_avg": 0.002363583486294374,
      "epoch": 4.4956207275390626e-05,
      "step": 47140
    },
    {
      "epoch": 4.4956207275390626e-05,
      "step": 47140,
      "training_step_time": 7.6829752922058105
    },
    {
      "epoch": 4.495716094970703e-05,
      "model_forward_time": 1.7270777225494385,
      "step": 47141
    },
    {
      "epoch": 4.495716094970703e-05,
      "step": 47141,
      "training_step_time": 7.067014217376709
    },
    {
      "epoch": 4.495811462402344e-05,
      "model_forward_time": 1.7302100658416748,
      "step": 47142
    },
    {
      "epoch": 4.495811462402344e-05,
      "step": 47142,
      "training_step_time": 7.039933204650879
    },
    {
      "epoch": 4.4959068298339844e-05,
      "model_forward_time": 1.7293946743011475,
      "step": 47143
    },
    {
      "epoch": 4.4959068298339844e-05,
      "step": 47143,
      "training_step_time": 7.140815258026123
    },
    {
      "epoch": 4.496002197265625e-05,
      "model_forward_time": 1.729759931564331,
      "step": 47144
    },
    {
      "epoch": 4.496002197265625e-05,
      "step": 47144,
      "training_step_time": 7.057400941848755
    },
    {
      "epoch": 4.496097564697266e-05,
      "model_forward_time": 1.7377893924713135,
      "step": 47145
    },
    {
      "epoch": 4.496097564697266e-05,
      "step": 47145,
      "training_step_time": 7.035332679748535
    },
    {
      "epoch": 4.496192932128906e-05,
      "model_forward_time": 1.7404210567474365,
      "step": 47146
    },
    {
      "epoch": 4.496192932128906e-05,
      "step": 47146,
      "training_step_time": 7.897809982299805
    },
    {
      "epoch": 4.496288299560547e-05,
      "model_forward_time": 2.313636302947998,
      "step": 47147
    },
    {
      "epoch": 4.496288299560547e-05,
      "step": 47147,
      "training_step_time": 9.844177961349487
    },
    {
      "epoch": 4.496383666992188e-05,
      "model_forward_time": 1.7280855178833008,
      "step": 47148
    },
    {
      "epoch": 4.496383666992188e-05,
      "step": 47148,
      "training_step_time": 8.809960126876831
    },
    {
      "epoch": 4.496479034423828e-05,
      "model_forward_time": 1.7369706630706787,
      "step": 47149
    },
    {
      "epoch": 4.496479034423828e-05,
      "step": 47149,
      "training_step_time": 8.032659769058228
    },
    {
      "epoch": 4.4965744018554686e-05,
      "grad_norm": 0.11387955397367477,
      "learning_rate": 5.8801614711209476e-06,
      "loss": 0.0596,
      "step": 47150
    },
    {
      "epoch": 4.4965744018554686e-05,
      "model_forward_time": 1.7107667922973633,
      "step": 47150
    },
    {
      "dynamics_loss_avg": 0.06968679372221231,
      "epoch": 4.4965744018554686e-05,
      "step": 47150
    },
    {
      "action_loss_avg": 0.0025312449317425488,
      "epoch": 4.4965744018554686e-05,
      "step": 47150
    },
    {
      "epoch": 4.4965744018554686e-05,
      "step": 47150,
      "training_step_time": 7.812485694885254
    },
    {
      "epoch": 4.49666976928711e-05,
      "model_forward_time": 1.7347853183746338,
      "step": 47151
    },
    {
      "epoch": 4.49666976928711e-05,
      "step": 47151,
      "training_step_time": 7.079214096069336
    },
    {
      "epoch": 4.49676513671875e-05,
      "model_forward_time": 1.7252371311187744,
      "step": 47152
    },
    {
      "epoch": 4.49676513671875e-05,
      "step": 47152,
      "training_step_time": 7.064868688583374
    },
    {
      "epoch": 4.4968605041503905e-05,
      "model_forward_time": 1.7241625785827637,
      "step": 47153
    },
    {
      "epoch": 4.4968605041503905e-05,
      "step": 47153,
      "training_step_time": 7.109868288040161
    },
    {
      "epoch": 4.4969558715820316e-05,
      "model_forward_time": 1.7221550941467285,
      "step": 47154
    },
    {
      "epoch": 4.4969558715820316e-05,
      "step": 47154,
      "training_step_time": 7.085967302322388
    },
    {
      "epoch": 4.497051239013672e-05,
      "model_forward_time": 1.7210254669189453,
      "step": 47155
    },
    {
      "epoch": 4.497051239013672e-05,
      "step": 47155,
      "training_step_time": 8.710634708404541
    },
    {
      "epoch": 4.4971466064453124e-05,
      "model_forward_time": 1.7602484226226807,
      "step": 47156
    },
    {
      "epoch": 4.4971466064453124e-05,
      "step": 47156,
      "training_step_time": 8.375909805297852
    },
    {
      "epoch": 4.497241973876953e-05,
      "model_forward_time": 1.7130987644195557,
      "step": 47157
    },
    {
      "epoch": 4.497241973876953e-05,
      "step": 47157,
      "training_step_time": 8.455217361450195
    },
    {
      "epoch": 4.497337341308594e-05,
      "model_forward_time": 1.7222895622253418,
      "step": 47158
    },
    {
      "epoch": 4.497337341308594e-05,
      "step": 47158,
      "training_step_time": 7.882484197616577
    },
    {
      "epoch": 4.497432708740234e-05,
      "model_forward_time": 1.7250511646270752,
      "step": 47159
    },
    {
      "epoch": 4.497432708740234e-05,
      "step": 47159,
      "training_step_time": 7.72097110748291
    },
    {
      "epoch": 4.497528076171875e-05,
      "grad_norm": 0.11119774729013443,
      "learning_rate": 5.878533773142732e-06,
      "loss": 0.0593,
      "step": 47160
    },
    {
      "epoch": 4.497528076171875e-05,
      "model_forward_time": 1.7320270538330078,
      "step": 47160
    },
    {
      "dynamics_loss_avg": 0.0621557654812932,
      "epoch": 4.497528076171875e-05,
      "step": 47160
    },
    {
      "action_loss_avg": 0.0021888321731239556,
      "epoch": 4.497528076171875e-05,
      "step": 47160
    },
    {
      "epoch": 4.497528076171875e-05,
      "step": 47160,
      "training_step_time": 7.0822649002075195
    },
    {
      "epoch": 4.497623443603516e-05,
      "model_forward_time": 1.7409992218017578,
      "step": 47161
    },
    {
      "epoch": 4.497623443603516e-05,
      "step": 47161,
      "training_step_time": 6.878753900527954
    },
    {
      "epoch": 4.497718811035156e-05,
      "model_forward_time": 1.7450878620147705,
      "step": 47162
    },
    {
      "epoch": 4.497718811035156e-05,
      "step": 47162,
      "training_step_time": 7.0706353187561035
    },
    {
      "epoch": 4.4978141784667966e-05,
      "model_forward_time": 1.7483317852020264,
      "step": 47163
    },
    {
      "epoch": 4.4978141784667966e-05,
      "step": 47163,
      "training_step_time": 7.062496900558472
    },
    {
      "epoch": 4.4979095458984376e-05,
      "model_forward_time": 1.742692232131958,
      "step": 47164
    },
    {
      "epoch": 4.4979095458984376e-05,
      "step": 47164,
      "training_step_time": 7.890902042388916
    },
    {
      "epoch": 4.498004913330078e-05,
      "model_forward_time": 1.734421730041504,
      "step": 47165
    },
    {
      "epoch": 4.498004913330078e-05,
      "step": 47165,
      "training_step_time": 9.510471820831299
    },
    {
      "epoch": 4.4981002807617184e-05,
      "model_forward_time": 2.9724197387695312,
      "step": 47166
    },
    {
      "epoch": 4.4981002807617184e-05,
      "step": 47166,
      "training_step_time": 8.983633041381836
    },
    {
      "epoch": 4.4981956481933595e-05,
      "model_forward_time": 1.7374482154846191,
      "step": 47167
    },
    {
      "epoch": 4.4981956481933595e-05,
      "step": 47167,
      "training_step_time": 7.848158597946167
    },
    {
      "epoch": 4.498291015625e-05,
      "model_forward_time": 1.7532668113708496,
      "step": 47168
    },
    {
      "epoch": 4.498291015625e-05,
      "step": 47168,
      "training_step_time": 8.563629150390625
    },
    {
      "epoch": 4.49838638305664e-05,
      "model_forward_time": 1.7789576053619385,
      "step": 47169
    },
    {
      "epoch": 4.49838638305664e-05,
      "step": 47169,
      "training_step_time": 7.074793100357056
    },
    {
      "epoch": 4.4984817504882814e-05,
      "grad_norm": 0.09996514022350311,
      "learning_rate": 5.876905979089383e-06,
      "loss": 0.0587,
      "step": 47170
    },
    {
      "epoch": 4.4984817504882814e-05,
      "model_forward_time": 1.7669181823730469,
      "step": 47170
    },
    {
      "dynamics_loss_avg": 0.05595319084823132,
      "epoch": 4.4984817504882814e-05,
      "step": 47170
    },
    {
      "action_loss_avg": 0.0014972582692280411,
      "epoch": 4.4984817504882814e-05,
      "step": 47170
    },
    {
      "epoch": 4.4984817504882814e-05,
      "step": 47170,
      "training_step_time": 6.912426471710205
    },
    {
      "epoch": 4.498577117919922e-05,
      "model_forward_time": 1.7592506408691406,
      "step": 47171
    },
    {
      "epoch": 4.498577117919922e-05,
      "step": 47171,
      "training_step_time": 6.908601522445679
    },
    {
      "epoch": 4.498672485351562e-05,
      "model_forward_time": 1.7523183822631836,
      "step": 47172
    },
    {
      "epoch": 4.498672485351562e-05,
      "step": 47172,
      "training_step_time": 7.03227162361145
    },
    {
      "epoch": 4.498767852783203e-05,
      "model_forward_time": 1.7570891380310059,
      "step": 47173
    },
    {
      "epoch": 4.498767852783203e-05,
      "step": 47173,
      "training_step_time": 7.985065221786499
    },
    {
      "epoch": 4.498863220214844e-05,
      "model_forward_time": 1.7402920722961426,
      "step": 47174
    },
    {
      "epoch": 4.498863220214844e-05,
      "step": 47174,
      "training_step_time": 8.411601781845093
    },
    {
      "epoch": 4.498958587646484e-05,
      "model_forward_time": 2.892747163772583,
      "step": 47175
    },
    {
      "epoch": 4.498958587646484e-05,
      "step": 47175,
      "training_step_time": 9.45152235031128
    },
    {
      "epoch": 4.499053955078125e-05,
      "model_forward_time": 1.8181977272033691,
      "step": 47176
    },
    {
      "epoch": 4.499053955078125e-05,
      "step": 47176,
      "training_step_time": 8.993902683258057
    },
    {
      "epoch": 4.4991493225097656e-05,
      "model_forward_time": 1.7552895545959473,
      "step": 47177
    },
    {
      "epoch": 4.4991493225097656e-05,
      "step": 47177,
      "training_step_time": 7.67688512802124
    },
    {
      "epoch": 4.499244689941406e-05,
      "model_forward_time": 1.744490385055542,
      "step": 47178
    },
    {
      "epoch": 4.499244689941406e-05,
      "step": 47178,
      "training_step_time": 7.852739572525024
    },
    {
      "epoch": 4.499340057373047e-05,
      "model_forward_time": 1.7589523792266846,
      "step": 47179
    },
    {
      "epoch": 4.499340057373047e-05,
      "step": 47179,
      "training_step_time": 7.064982652664185
    },
    {
      "epoch": 4.4994354248046875e-05,
      "grad_norm": 0.11543001979589462,
      "learning_rate": 5.875278089138915e-06,
      "loss": 0.0599,
      "step": 47180
    },
    {
      "epoch": 4.4994354248046875e-05,
      "model_forward_time": 1.762810468673706,
      "step": 47180
    },
    {
      "dynamics_loss_avg": 0.07069817520678043,
      "epoch": 4.4994354248046875e-05,
      "step": 47180
    },
    {
      "action_loss_avg": 0.0013944452424766497,
      "epoch": 4.4994354248046875e-05,
      "step": 47180
    },
    {
      "epoch": 4.4994354248046875e-05,
      "step": 47180,
      "training_step_time": 7.091735601425171
    },
    {
      "epoch": 4.499530792236328e-05,
      "model_forward_time": 1.7596609592437744,
      "step": 47181
    },
    {
      "epoch": 4.499530792236328e-05,
      "step": 47181,
      "training_step_time": 7.053223371505737
    },
    {
      "epoch": 4.499626159667969e-05,
      "model_forward_time": 1.768080711364746,
      "step": 47182
    },
    {
      "epoch": 4.499626159667969e-05,
      "step": 47182,
      "training_step_time": 7.984697341918945
    },
    {
      "epoch": 4.4997215270996094e-05,
      "model_forward_time": 1.7660398483276367,
      "step": 47183
    },
    {
      "epoch": 4.4997215270996094e-05,
      "step": 47183,
      "training_step_time": 7.893097877502441
    },
    {
      "epoch": 4.49981689453125e-05,
      "model_forward_time": 2.9113287925720215,
      "step": 47184
    },
    {
      "epoch": 4.49981689453125e-05,
      "step": 47184,
      "training_step_time": 11.048929452896118
    },
    {
      "epoch": 4.499912261962891e-05,
      "model_forward_time": 1.8345181941986084,
      "step": 47185
    },
    {
      "epoch": 4.499912261962891e-05,
      "step": 47185,
      "training_step_time": 7.991333723068237
    },
    {
      "epoch": 4.500007629394531e-05,
      "model_forward_time": 1.7723991870880127,
      "step": 47186
    },
    {
      "epoch": 4.500007629394531e-05,
      "step": 47186,
      "training_step_time": 8.418288707733154
    },
    {
      "epoch": 4.5001029968261716e-05,
      "model_forward_time": 1.7648980617523193,
      "step": 47187
    },
    {
      "epoch": 4.5001029968261716e-05,
      "step": 47187,
      "training_step_time": 7.734381198883057
    },
    {
      "epoch": 4.500198364257813e-05,
      "model_forward_time": 1.7663962841033936,
      "step": 47188
    },
    {
      "epoch": 4.500198364257813e-05,
      "step": 47188,
      "training_step_time": 7.039870738983154
    },
    {
      "epoch": 4.500293731689453e-05,
      "model_forward_time": 1.761812686920166,
      "step": 47189
    },
    {
      "epoch": 4.500293731689453e-05,
      "step": 47189,
      "training_step_time": 7.077937364578247
    },
    {
      "epoch": 4.5003890991210935e-05,
      "grad_norm": 0.11666195839643478,
      "learning_rate": 5.873650103469351e-06,
      "loss": 0.0579,
      "step": 47190
    },
    {
      "epoch": 4.5003890991210935e-05,
      "model_forward_time": 1.7727997303009033,
      "step": 47190
    },
    {
      "dynamics_loss_avg": 0.05809869412332773,
      "epoch": 4.5003890991210935e-05,
      "step": 47190
    },
    {
      "action_loss_avg": 0.0017605081608053296,
      "epoch": 4.5003890991210935e-05,
      "step": 47190
    },
    {
      "epoch": 4.5003890991210935e-05,
      "step": 47190,
      "training_step_time": 7.11984658241272
    },
    {
      "epoch": 4.5004844665527346e-05,
      "model_forward_time": 1.7660565376281738,
      "step": 47191
    },
    {
      "epoch": 4.5004844665527346e-05,
      "step": 47191,
      "training_step_time": 7.98943305015564
    },
    {
      "epoch": 4.500579833984375e-05,
      "model_forward_time": 1.7603960037231445,
      "step": 47192
    },
    {
      "epoch": 4.500579833984375e-05,
      "step": 47192,
      "training_step_time": 9.615603685379028
    },
    {
      "epoch": 4.5006752014160154e-05,
      "model_forward_time": 2.8941586017608643,
      "step": 47193
    },
    {
      "epoch": 4.5006752014160154e-05,
      "step": 47193,
      "training_step_time": 9.669941186904907
    },
    {
      "epoch": 4.5007705688476565e-05,
      "model_forward_time": 1.8113694190979004,
      "step": 47194
    },
    {
      "epoch": 4.5007705688476565e-05,
      "step": 47194,
      "training_step_time": 9.561354637145996
    },
    {
      "epoch": 4.500865936279297e-05,
      "model_forward_time": 1.759796380996704,
      "step": 47195
    },
    {
      "epoch": 4.500865936279297e-05,
      "step": 47195,
      "training_step_time": 7.9140777587890625
    },
    {
      "epoch": 4.500961303710937e-05,
      "model_forward_time": 1.7621138095855713,
      "step": 47196
    },
    {
      "epoch": 4.500961303710937e-05,
      "step": 47196,
      "training_step_time": 7.407156944274902
    },
    {
      "epoch": 4.5010566711425784e-05,
      "model_forward_time": 1.7692062854766846,
      "step": 47197
    },
    {
      "epoch": 4.5010566711425784e-05,
      "step": 47197,
      "training_step_time": 7.846954584121704
    },
    {
      "epoch": 4.501152038574219e-05,
      "model_forward_time": 1.7629151344299316,
      "step": 47198
    },
    {
      "epoch": 4.501152038574219e-05,
      "step": 47198,
      "training_step_time": 7.101089954376221
    },
    {
      "epoch": 4.501247406005859e-05,
      "model_forward_time": 1.758305311203003,
      "step": 47199
    },
    {
      "epoch": 4.501247406005859e-05,
      "step": 47199,
      "training_step_time": 7.147775650024414
    },
    {
      "epoch": 4.5013427734375e-05,
      "grad_norm": 0.12295025587081909,
      "learning_rate": 5.872022022258726e-06,
      "loss": 0.0585,
      "step": 47200
    },
    {
      "epoch": 4.5013427734375e-05,
      "model_forward_time": 1.7629921436309814,
      "step": 47200
    },
    {
      "dynamics_loss_avg": 0.043959683179855345,
      "epoch": 4.5013427734375e-05,
      "step": 47200
    },
    {
      "action_loss_avg": 0.001117324741790071,
      "epoch": 4.5013427734375e-05,
      "step": 47200
    },
    {
      "epoch": 4.5013427734375e-05,
      "step": 47200,
      "training_step_time": 8.047637939453125
    },
    {
      "epoch": 4.501438140869141e-05,
      "model_forward_time": 1.764538049697876,
      "step": 47201
    },
    {
      "epoch": 4.501438140869141e-05,
      "step": 47201,
      "training_step_time": 8.758568525314331
    },
    {
      "epoch": 4.501533508300781e-05,
      "model_forward_time": 2.916246175765991,
      "step": 47202
    },
    {
      "epoch": 4.501533508300781e-05,
      "step": 47202,
      "training_step_time": 9.110861539840698
    },
    {
      "epoch": 4.501628875732422e-05,
      "model_forward_time": 1.7794222831726074,
      "step": 47203
    },
    {
      "epoch": 4.501628875732422e-05,
      "step": 47203,
      "training_step_time": 8.71509337425232
    },
    {
      "epoch": 4.5017242431640626e-05,
      "model_forward_time": 1.7560508251190186,
      "step": 47204
    },
    {
      "epoch": 4.5017242431640626e-05,
      "step": 47204,
      "training_step_time": 8.710562467575073
    },
    {
      "epoch": 4.501819610595703e-05,
      "model_forward_time": 1.7420833110809326,
      "step": 47205
    },
    {
      "epoch": 4.501819610595703e-05,
      "step": 47205,
      "training_step_time": 7.404787063598633
    },
    {
      "epoch": 4.501914978027344e-05,
      "model_forward_time": 1.7547132968902588,
      "step": 47206
    },
    {
      "epoch": 4.501914978027344e-05,
      "step": 47206,
      "training_step_time": 7.46628737449646
    },
    {
      "epoch": 4.5020103454589844e-05,
      "model_forward_time": 1.758955478668213,
      "step": 47207
    },
    {
      "epoch": 4.5020103454589844e-05,
      "step": 47207,
      "training_step_time": 7.016408443450928
    },
    {
      "epoch": 4.502105712890625e-05,
      "model_forward_time": 1.752638339996338,
      "step": 47208
    },
    {
      "epoch": 4.502105712890625e-05,
      "step": 47208,
      "training_step_time": 6.982919454574585
    },
    {
      "epoch": 4.502201080322266e-05,
      "model_forward_time": 1.7582833766937256,
      "step": 47209
    },
    {
      "epoch": 4.502201080322266e-05,
      "step": 47209,
      "training_step_time": 7.974159240722656
    },
    {
      "epoch": 4.502296447753906e-05,
      "grad_norm": 0.10215715318918228,
      "learning_rate": 5.870393845685082e-06,
      "loss": 0.0595,
      "step": 47210
    },
    {
      "epoch": 4.502296447753906e-05,
      "model_forward_time": 1.7490746974945068,
      "step": 47210
    },
    {
      "dynamics_loss_avg": 0.06474089622497559,
      "epoch": 4.502296447753906e-05,
      "step": 47210
    },
    {
      "action_loss_avg": 0.0014154842938296497,
      "epoch": 4.502296447753906e-05,
      "step": 47210
    },
    {
      "epoch": 4.502296447753906e-05,
      "step": 47210,
      "training_step_time": 8.89036512374878
    },
    {
      "epoch": 4.502391815185547e-05,
      "model_forward_time": 2.9372429847717285,
      "step": 47211
    },
    {
      "epoch": 4.502391815185547e-05,
      "step": 47211,
      "training_step_time": 9.513653993606567
    },
    {
      "epoch": 4.502487182617188e-05,
      "model_forward_time": 1.7739431858062744,
      "step": 47212
    },
    {
      "epoch": 4.502487182617188e-05,
      "step": 47212,
      "training_step_time": 9.61952018737793
    },
    {
      "epoch": 4.502582550048828e-05,
      "model_forward_time": 1.8361327648162842,
      "step": 47213
    },
    {
      "epoch": 4.502582550048828e-05,
      "step": 47213,
      "training_step_time": 8.03231954574585
    },
    {
      "epoch": 4.5026779174804686e-05,
      "model_forward_time": 1.772740364074707,
      "step": 47214
    },
    {
      "epoch": 4.5026779174804686e-05,
      "step": 47214,
      "training_step_time": 7.858269929885864
    },
    {
      "epoch": 4.50277328491211e-05,
      "model_forward_time": 1.7730884552001953,
      "step": 47215
    },
    {
      "epoch": 4.50277328491211e-05,
      "step": 47215,
      "training_step_time": 7.897679090499878
    },
    {
      "epoch": 4.50286865234375e-05,
      "model_forward_time": 1.7648422718048096,
      "step": 47216
    },
    {
      "epoch": 4.50286865234375e-05,
      "step": 47216,
      "training_step_time": 7.054696559906006
    },
    {
      "epoch": 4.5029640197753905e-05,
      "model_forward_time": 1.7684850692749023,
      "step": 47217
    },
    {
      "epoch": 4.5029640197753905e-05,
      "step": 47217,
      "training_step_time": 7.118054389953613
    },
    {
      "epoch": 4.5030593872070316e-05,
      "model_forward_time": 1.7736644744873047,
      "step": 47218
    },
    {
      "epoch": 4.5030593872070316e-05,
      "step": 47218,
      "training_step_time": 8.086642026901245
    },
    {
      "epoch": 4.503154754638672e-05,
      "model_forward_time": 1.7624125480651855,
      "step": 47219
    },
    {
      "epoch": 4.503154754638672e-05,
      "step": 47219,
      "training_step_time": 8.734831809997559
    },
    {
      "epoch": 4.5032501220703124e-05,
      "grad_norm": 0.11606749147176743,
      "learning_rate": 5.868765573926477e-06,
      "loss": 0.0603,
      "step": 47220
    },
    {
      "epoch": 4.5032501220703124e-05,
      "model_forward_time": 2.9152443408966064,
      "step": 47220
    },
    {
      "dynamics_loss_avg": 0.04477005898952484,
      "epoch": 4.5032501220703124e-05,
      "step": 47220
    },
    {
      "action_loss_avg": 0.0008482378209009767,
      "epoch": 4.5032501220703124e-05,
      "step": 47220
    },
    {
      "epoch": 4.5032501220703124e-05,
      "step": 47220,
      "training_step_time": 7.991371154785156
    },
    {
      "epoch": 4.503345489501953e-05,
      "model_forward_time": 1.7635729312896729,
      "step": 47221
    },
    {
      "epoch": 4.503345489501953e-05,
      "step": 47221,
      "training_step_time": 8.11856198310852
    },
    {
      "epoch": 4.503440856933594e-05,
      "model_forward_time": 1.8357527256011963,
      "step": 47222
    },
    {
      "epoch": 4.503440856933594e-05,
      "step": 47222,
      "training_step_time": 7.993294715881348
    },
    {
      "epoch": 4.503536224365234e-05,
      "model_forward_time": 1.7762386798858643,
      "step": 47223
    },
    {
      "epoch": 4.503536224365234e-05,
      "step": 47223,
      "training_step_time": 7.693411588668823
    },
    {
      "epoch": 4.503631591796875e-05,
      "model_forward_time": 1.7583277225494385,
      "step": 47224
    },
    {
      "epoch": 4.503631591796875e-05,
      "step": 47224,
      "training_step_time": 7.648559093475342
    },
    {
      "epoch": 4.503726959228516e-05,
      "model_forward_time": 1.7750918865203857,
      "step": 47225
    },
    {
      "epoch": 4.503726959228516e-05,
      "step": 47225,
      "training_step_time": 7.083998918533325
    },
    {
      "epoch": 4.503822326660156e-05,
      "model_forward_time": 1.7907679080963135,
      "step": 47226
    },
    {
      "epoch": 4.503822326660156e-05,
      "step": 47226,
      "training_step_time": 7.036001920700073
    },
    {
      "epoch": 4.5039176940917966e-05,
      "model_forward_time": 1.78875732421875,
      "step": 47227
    },
    {
      "epoch": 4.5039176940917966e-05,
      "step": 47227,
      "training_step_time": 7.938108682632446
    },
    {
      "epoch": 4.5040130615234376e-05,
      "model_forward_time": 1.7668745517730713,
      "step": 47228
    },
    {
      "epoch": 4.5040130615234376e-05,
      "step": 47228,
      "training_step_time": 7.859335422515869
    },
    {
      "epoch": 4.504108428955078e-05,
      "model_forward_time": 2.906357526779175,
      "step": 47229
    },
    {
      "epoch": 4.504108428955078e-05,
      "step": 47229,
      "training_step_time": 9.541745901107788
    },
    {
      "epoch": 4.5042037963867185e-05,
      "grad_norm": 0.09199751913547516,
      "learning_rate": 5.8671372071609745e-06,
      "loss": 0.0582,
      "step": 47230
    },
    {
      "epoch": 4.5042037963867185e-05,
      "model_forward_time": 1.7765893936157227,
      "step": 47230
    },
    {
      "dynamics_loss_avg": 0.07070161178708076,
      "epoch": 4.5042037963867185e-05,
      "step": 47230
    },
    {
      "action_loss_avg": 0.0017111541761551051,
      "epoch": 4.5042037963867185e-05,
      "step": 47230
    },
    {
      "epoch": 4.5042037963867185e-05,
      "step": 47230,
      "training_step_time": 8.558264255523682
    },
    {
      "epoch": 4.5042991638183595e-05,
      "model_forward_time": 1.41729736328125,
      "step": 47231
    },
    {
      "epoch": 4.5042991638183595e-05,
      "step": 47231,
      "training_step_time": 8.039489507675171
    },
    {
      "epoch": 4.50439453125e-05,
      "model_forward_time": 1.7655951976776123,
      "step": 47232
    },
    {
      "epoch": 4.50439453125e-05,
      "step": 47232,
      "training_step_time": 7.895598888397217
    },
    {
      "epoch": 4.50448989868164e-05,
      "model_forward_time": 1.7845664024353027,
      "step": 47233
    },
    {
      "epoch": 4.50448989868164e-05,
      "step": 47233,
      "training_step_time": 7.64113450050354
    },
    {
      "epoch": 4.5045852661132814e-05,
      "model_forward_time": 1.7719545364379883,
      "step": 47234
    },
    {
      "epoch": 4.5045852661132814e-05,
      "step": 47234,
      "training_step_time": 7.779104948043823
    },
    {
      "epoch": 4.504680633544922e-05,
      "model_forward_time": 1.7725722789764404,
      "step": 47235
    },
    {
      "epoch": 4.504680633544922e-05,
      "step": 47235,
      "training_step_time": 7.056523323059082
    },
    {
      "epoch": 4.504776000976562e-05,
      "model_forward_time": 1.7541122436523438,
      "step": 47236
    },
    {
      "epoch": 4.504776000976562e-05,
      "step": 47236,
      "training_step_time": 7.836794376373291
    },
    {
      "epoch": 4.504871368408203e-05,
      "model_forward_time": 1.780203104019165,
      "step": 47237
    },
    {
      "epoch": 4.504871368408203e-05,
      "step": 47237,
      "training_step_time": 7.939378499984741
    },
    {
      "epoch": 4.504966735839844e-05,
      "model_forward_time": 2.909090995788574,
      "step": 47238
    },
    {
      "epoch": 4.504966735839844e-05,
      "step": 47238,
      "training_step_time": 9.169537782669067
    },
    {
      "epoch": 4.505062103271484e-05,
      "model_forward_time": 1.8601911067962646,
      "step": 47239
    },
    {
      "epoch": 4.505062103271484e-05,
      "step": 47239,
      "training_step_time": 8.136070251464844
    },
    {
      "epoch": 4.505157470703125e-05,
      "grad_norm": 0.1322581022977829,
      "learning_rate": 5.8655087455666505e-06,
      "loss": 0.0588,
      "step": 47240
    },
    {
      "epoch": 4.505157470703125e-05,
      "model_forward_time": 1.9101240634918213,
      "step": 47240
    },
    {
      "dynamics_loss_avg": 0.07543733790516853,
      "epoch": 4.505157470703125e-05,
      "step": 47240
    },
    {
      "action_loss_avg": 0.003236292506335303,
      "epoch": 4.505157470703125e-05,
      "step": 47240
    },
    {
      "epoch": 4.505157470703125e-05,
      "step": 47240,
      "training_step_time": 9.598767280578613
    },
    {
      "epoch": 4.5052528381347656e-05,
      "model_forward_time": 1.9003922939300537,
      "step": 47241
    },
    {
      "epoch": 4.5052528381347656e-05,
      "step": 47241,
      "training_step_time": 7.818885087966919
    },
    {
      "epoch": 4.505348205566406e-05,
      "model_forward_time": 1.9263105392456055,
      "step": 47242
    },
    {
      "epoch": 4.505348205566406e-05,
      "step": 47242,
      "training_step_time": 7.613255977630615
    },
    {
      "epoch": 4.505443572998047e-05,
      "model_forward_time": 1.913008213043213,
      "step": 47243
    },
    {
      "epoch": 4.505443572998047e-05,
      "step": 47243,
      "training_step_time": 7.8186798095703125
    },
    {
      "epoch": 4.5055389404296875e-05,
      "model_forward_time": 1.9291956424713135,
      "step": 47244
    },
    {
      "epoch": 4.5055389404296875e-05,
      "step": 47244,
      "training_step_time": 7.020722150802612
    },
    {
      "epoch": 4.505634307861328e-05,
      "model_forward_time": 1.902329921722412,
      "step": 47245
    },
    {
      "epoch": 4.505634307861328e-05,
      "step": 47245,
      "training_step_time": 7.494735956192017
    },
    {
      "epoch": 4.505729675292969e-05,
      "model_forward_time": 1.9241664409637451,
      "step": 47246
    },
    {
      "epoch": 4.505729675292969e-05,
      "step": 47246,
      "training_step_time": 7.924962282180786
    },
    {
      "epoch": 4.5058250427246094e-05,
      "model_forward_time": 3.0067617893218994,
      "step": 47247
    },
    {
      "epoch": 4.5058250427246094e-05,
      "step": 47247,
      "training_step_time": 8.03483247756958
    },
    {
      "epoch": 4.50592041015625e-05,
      "model_forward_time": 1.942042589187622,
      "step": 47248
    },
    {
      "epoch": 4.50592041015625e-05,
      "step": 47248,
      "training_step_time": 7.972170114517212
    },
    {
      "epoch": 4.506015777587891e-05,
      "model_forward_time": 1.9243030548095703,
      "step": 47249
    },
    {
      "epoch": 4.506015777587891e-05,
      "step": 47249,
      "training_step_time": 10.57371187210083
    },
    {
      "epoch": 4.506111145019531e-05,
      "grad_norm": 0.13740451633930206,
      "learning_rate": 5.863880189321592e-06,
      "loss": 0.0607,
      "step": 47250
    },
    {
      "epoch": 4.506111145019531e-05,
      "model_forward_time": 1.9072461128234863,
      "step": 47250
    },
    {
      "dynamics_loss_avg": 0.058364610001444814,
      "epoch": 4.506111145019531e-05,
      "step": 47250
    },
    {
      "action_loss_avg": 0.0016302648873534054,
      "epoch": 4.506111145019531e-05,
      "step": 47250
    },
    {
      "epoch": 4.506111145019531e-05,
      "step": 47250,
      "training_step_time": 8.798701047897339
    },
    {
      "epoch": 4.5062065124511717e-05,
      "model_forward_time": 1.9254426956176758,
      "step": 47251
    },
    {
      "epoch": 4.5062065124511717e-05,
      "step": 47251,
      "training_step_time": 7.67292332649231
    },
    {
      "epoch": 4.506301879882813e-05,
      "model_forward_time": 1.913325309753418,
      "step": 47252
    },
    {
      "epoch": 4.506301879882813e-05,
      "step": 47252,
      "training_step_time": 7.903455495834351
    },
    {
      "epoch": 4.506397247314453e-05,
      "model_forward_time": 1.9378690719604492,
      "step": 47253
    },
    {
      "epoch": 4.506397247314453e-05,
      "step": 47253,
      "training_step_time": 7.022207021713257
    },
    {
      "epoch": 4.5064926147460935e-05,
      "model_forward_time": 1.9310812950134277,
      "step": 47254
    },
    {
      "epoch": 4.5064926147460935e-05,
      "step": 47254,
      "training_step_time": 7.486372709274292
    },
    {
      "epoch": 4.5065879821777346e-05,
      "model_forward_time": 1.8878228664398193,
      "step": 47255
    },
    {
      "epoch": 4.5065879821777346e-05,
      "step": 47255,
      "training_step_time": 7.976404666900635
    },
    {
      "epoch": 4.506683349609375e-05,
      "model_forward_time": 2.5482518672943115,
      "step": 47256
    },
    {
      "epoch": 4.506683349609375e-05,
      "step": 47256,
      "training_step_time": 9.128044366836548
    },
    {
      "epoch": 4.5067787170410154e-05,
      "model_forward_time": 1.7860238552093506,
      "step": 47257
    },
    {
      "epoch": 4.5067787170410154e-05,
      "step": 47257,
      "training_step_time": 7.878710508346558
    },
    {
      "epoch": 4.5068740844726565e-05,
      "model_forward_time": 1.7939741611480713,
      "step": 47258
    },
    {
      "epoch": 4.5068740844726565e-05,
      "step": 47258,
      "training_step_time": 9.816813707351685
    },
    {
      "epoch": 4.506969451904297e-05,
      "model_forward_time": 1.7771174907684326,
      "step": 47259
    },
    {
      "epoch": 4.506969451904297e-05,
      "step": 47259,
      "training_step_time": 7.911345481872559
    },
    {
      "epoch": 4.507064819335937e-05,
      "grad_norm": 0.11970149725675583,
      "learning_rate": 5.862251538603895e-06,
      "loss": 0.0606,
      "step": 47260
    },
    {
      "epoch": 4.507064819335937e-05,
      "model_forward_time": 1.7656238079071045,
      "step": 47260
    },
    {
      "dynamics_loss_avg": 0.06716661341488361,
      "epoch": 4.507064819335937e-05,
      "step": 47260
    },
    {
      "action_loss_avg": 0.002331631211563945,
      "epoch": 4.507064819335937e-05,
      "step": 47260
    },
    {
      "epoch": 4.507064819335937e-05,
      "step": 47260,
      "training_step_time": 7.639114618301392
    },
    {
      "epoch": 4.5071601867675784e-05,
      "model_forward_time": 1.870018720626831,
      "step": 47261
    },
    {
      "epoch": 4.5071601867675784e-05,
      "step": 47261,
      "training_step_time": 7.724508047103882
    },
    {
      "epoch": 4.507255554199219e-05,
      "model_forward_time": 1.9245257377624512,
      "step": 47262
    },
    {
      "epoch": 4.507255554199219e-05,
      "step": 47262,
      "training_step_time": 6.915930509567261
    },
    {
      "epoch": 4.507350921630859e-05,
      "model_forward_time": 1.9477553367614746,
      "step": 47263
    },
    {
      "epoch": 4.507350921630859e-05,
      "step": 47263,
      "training_step_time": 7.425028562545776
    },
    {
      "epoch": 4.5074462890625e-05,
      "model_forward_time": 1.9411301612854004,
      "step": 47264
    },
    {
      "epoch": 4.5074462890625e-05,
      "step": 47264,
      "training_step_time": 7.985426902770996
    },
    {
      "epoch": 4.507541656494141e-05,
      "model_forward_time": 1.9308202266693115,
      "step": 47265
    },
    {
      "epoch": 4.507541656494141e-05,
      "step": 47265,
      "training_step_time": 8.264891386032104
    },
    {
      "epoch": 4.507637023925781e-05,
      "model_forward_time": 1.920098066329956,
      "step": 47266
    },
    {
      "epoch": 4.507637023925781e-05,
      "step": 47266,
      "training_step_time": 8.690317153930664
    },
    {
      "epoch": 4.507732391357422e-05,
      "model_forward_time": 1.9337985515594482,
      "step": 47267
    },
    {
      "epoch": 4.507732391357422e-05,
      "step": 47267,
      "training_step_time": 8.167111873626709
    },
    {
      "epoch": 4.5078277587890626e-05,
      "model_forward_time": 1.9101967811584473,
      "step": 47268
    },
    {
      "epoch": 4.5078277587890626e-05,
      "step": 47268,
      "training_step_time": 8.028762340545654
    },
    {
      "epoch": 4.507923126220703e-05,
      "model_forward_time": 1.9171292781829834,
      "step": 47269
    },
    {
      "epoch": 4.507923126220703e-05,
      "step": 47269,
      "training_step_time": 7.962962627410889
    },
    {
      "epoch": 4.508018493652344e-05,
      "grad_norm": 0.12381279468536377,
      "learning_rate": 5.860622793591666e-06,
      "loss": 0.0626,
      "step": 47270
    },
    {
      "epoch": 4.508018493652344e-05,
      "model_forward_time": 1.9212336540222168,
      "step": 47270
    },
    {
      "dynamics_loss_avg": 0.06978062037378549,
      "epoch": 4.508018493652344e-05,
      "step": 47270
    },
    {
      "action_loss_avg": 0.001369276668992825,
      "epoch": 4.508018493652344e-05,
      "step": 47270
    },
    {
      "epoch": 4.508018493652344e-05,
      "step": 47270,
      "training_step_time": 7.726299285888672
    },
    {
      "epoch": 4.5081138610839845e-05,
      "model_forward_time": 1.9250221252441406,
      "step": 47271
    },
    {
      "epoch": 4.5081138610839845e-05,
      "step": 47271,
      "training_step_time": 7.068750381469727
    },
    {
      "epoch": 4.508209228515625e-05,
      "model_forward_time": 1.9479408264160156,
      "step": 47272
    },
    {
      "epoch": 4.508209228515625e-05,
      "step": 47272,
      "training_step_time": 7.573609828948975
    },
    {
      "epoch": 4.508304595947266e-05,
      "model_forward_time": 1.9193735122680664,
      "step": 47273
    },
    {
      "epoch": 4.508304595947266e-05,
      "step": 47273,
      "training_step_time": 7.986220836639404
    },
    {
      "epoch": 4.508399963378906e-05,
      "model_forward_time": 1.9034326076507568,
      "step": 47274
    },
    {
      "epoch": 4.508399963378906e-05,
      "step": 47274,
      "training_step_time": 8.463524580001831
    },
    {
      "epoch": 4.508495330810547e-05,
      "model_forward_time": 1.9118082523345947,
      "step": 47275
    },
    {
      "epoch": 4.508495330810547e-05,
      "step": 47275,
      "training_step_time": 8.60062289237976
    },
    {
      "epoch": 4.508590698242188e-05,
      "model_forward_time": 1.950775384902954,
      "step": 47276
    },
    {
      "epoch": 4.508590698242188e-05,
      "step": 47276,
      "training_step_time": 8.839680433273315
    },
    {
      "epoch": 4.508686065673828e-05,
      "model_forward_time": 1.9201750755310059,
      "step": 47277
    },
    {
      "epoch": 4.508686065673828e-05,
      "step": 47277,
      "training_step_time": 8.548578262329102
    },
    {
      "epoch": 4.5087814331054686e-05,
      "model_forward_time": 1.9288554191589355,
      "step": 47278
    },
    {
      "epoch": 4.5087814331054686e-05,
      "step": 47278,
      "training_step_time": 7.935671806335449
    },
    {
      "epoch": 4.50887680053711e-05,
      "model_forward_time": 1.9210736751556396,
      "step": 47279
    },
    {
      "epoch": 4.50887680053711e-05,
      "step": 47279,
      "training_step_time": 7.650997638702393
    },
    {
      "epoch": 4.50897216796875e-05,
      "grad_norm": 0.12646572291851044,
      "learning_rate": 5.858993954463021e-06,
      "loss": 0.0618,
      "step": 47280
    },
    {
      "epoch": 4.50897216796875e-05,
      "model_forward_time": 1.9473612308502197,
      "step": 47280
    },
    {
      "dynamics_loss_avg": 0.07112522833049298,
      "epoch": 4.50897216796875e-05,
      "step": 47280
    },
    {
      "action_loss_avg": 0.002367623377358541,
      "epoch": 4.50897216796875e-05,
      "step": 47280
    },
    {
      "epoch": 4.50897216796875e-05,
      "step": 47280,
      "training_step_time": 7.849363327026367
    },
    {
      "epoch": 4.5090675354003905e-05,
      "model_forward_time": 1.9214532375335693,
      "step": 47281
    },
    {
      "epoch": 4.5090675354003905e-05,
      "step": 47281,
      "training_step_time": 7.53244423866272
    },
    {
      "epoch": 4.509162902832031e-05,
      "model_forward_time": 1.9161574840545654,
      "step": 47282
    },
    {
      "epoch": 4.509162902832031e-05,
      "step": 47282,
      "training_step_time": 7.926997184753418
    },
    {
      "epoch": 4.509258270263672e-05,
      "model_forward_time": 1.934732437133789,
      "step": 47283
    },
    {
      "epoch": 4.509258270263672e-05,
      "step": 47283,
      "training_step_time": 8.000496864318848
    },
    {
      "epoch": 4.5093536376953124e-05,
      "model_forward_time": 2.985973358154297,
      "step": 47284
    },
    {
      "epoch": 4.5093536376953124e-05,
      "step": 47284,
      "training_step_time": 7.976835489273071
    },
    {
      "epoch": 4.509449005126953e-05,
      "model_forward_time": 1.785245656967163,
      "step": 47285
    },
    {
      "epoch": 4.509449005126953e-05,
      "step": 47285,
      "training_step_time": 9.014576435089111
    },
    {
      "epoch": 4.509544372558594e-05,
      "model_forward_time": 1.7657382488250732,
      "step": 47286
    },
    {
      "epoch": 4.509544372558594e-05,
      "step": 47286,
      "training_step_time": 9.317685604095459
    },
    {
      "epoch": 4.509639739990234e-05,
      "model_forward_time": 1.7624385356903076,
      "step": 47287
    },
    {
      "epoch": 4.509639739990234e-05,
      "step": 47287,
      "training_step_time": 8.005550622940063
    },
    {
      "epoch": 4.509735107421875e-05,
      "model_forward_time": 1.7173662185668945,
      "step": 47288
    },
    {
      "epoch": 4.509735107421875e-05,
      "step": 47288,
      "training_step_time": 7.853355407714844
    },
    {
      "epoch": 4.509830474853516e-05,
      "model_forward_time": 1.7114663124084473,
      "step": 47289
    },
    {
      "epoch": 4.509830474853516e-05,
      "step": 47289,
      "training_step_time": 7.787103891372681
    },
    {
      "epoch": 4.509925842285156e-05,
      "grad_norm": 0.12514755129814148,
      "learning_rate": 5.85736502139609e-06,
      "loss": 0.0597,
      "step": 47290
    },
    {
      "epoch": 4.509925842285156e-05,
      "model_forward_time": 1.6988246440887451,
      "step": 47290
    },
    {
      "dynamics_loss_avg": 0.06887306589633227,
      "epoch": 4.509925842285156e-05,
      "step": 47290
    },
    {
      "action_loss_avg": 0.0019048583388212136,
      "epoch": 4.509925842285156e-05,
      "step": 47290
    },
    {
      "epoch": 4.509925842285156e-05,
      "step": 47290,
      "training_step_time": 7.4793291091918945
    },
    {
      "epoch": 4.5100212097167966e-05,
      "model_forward_time": 1.71622896194458,
      "step": 47291
    },
    {
      "epoch": 4.5100212097167966e-05,
      "step": 47291,
      "training_step_time": 7.58461332321167
    },
    {
      "epoch": 4.5101165771484377e-05,
      "model_forward_time": 1.7067251205444336,
      "step": 47292
    },
    {
      "epoch": 4.5101165771484377e-05,
      "step": 47292,
      "training_step_time": 8.82893991470337
    },
    {
      "epoch": 4.510211944580078e-05,
      "model_forward_time": 2.3033182621002197,
      "step": 47293
    },
    {
      "epoch": 4.510211944580078e-05,
      "step": 47293,
      "training_step_time": 8.490694284439087
    },
    {
      "epoch": 4.5103073120117185e-05,
      "model_forward_time": 1.7438349723815918,
      "step": 47294
    },
    {
      "epoch": 4.5103073120117185e-05,
      "step": 47294,
      "training_step_time": 9.410852193832397
    },
    {
      "epoch": 4.5104026794433595e-05,
      "model_forward_time": 1.7655675411224365,
      "step": 47295
    },
    {
      "epoch": 4.5104026794433595e-05,
      "step": 47295,
      "training_step_time": 8.619051933288574
    },
    {
      "epoch": 4.510498046875e-05,
      "model_forward_time": 1.7844147682189941,
      "step": 47296
    },
    {
      "epoch": 4.510498046875e-05,
      "step": 47296,
      "training_step_time": 8.874995708465576
    },
    {
      "epoch": 4.5105934143066403e-05,
      "model_forward_time": 1.7280404567718506,
      "step": 47297
    },
    {
      "epoch": 4.5105934143066403e-05,
      "step": 47297,
      "training_step_time": 7.872682571411133
    },
    {
      "epoch": 4.5106887817382814e-05,
      "model_forward_time": 1.733729362487793,
      "step": 47298
    },
    {
      "epoch": 4.5106887817382814e-05,
      "step": 47298,
      "training_step_time": 7.427485466003418
    },
    {
      "epoch": 4.510784149169922e-05,
      "model_forward_time": 1.7344746589660645,
      "step": 47299
    },
    {
      "epoch": 4.510784149169922e-05,
      "step": 47299,
      "training_step_time": 7.560283899307251
    },
    {
      "epoch": 4.510879516601562e-05,
      "grad_norm": 0.151468425989151,
      "learning_rate": 5.855735994569008e-06,
      "loss": 0.0622,
      "step": 47300
    },
    {
      "epoch": 4.510879516601562e-05,
      "model_forward_time": 1.681746006011963,
      "step": 47300
    },
    {
      "dynamics_loss_avg": 0.07033802419900895,
      "epoch": 4.510879516601562e-05,
      "step": 47300
    },
    {
      "action_loss_avg": 0.0016573513858020306,
      "epoch": 4.510879516601562e-05,
      "step": 47300
    },
    {
      "epoch": 4.510879516601562e-05,
      "step": 47300,
      "training_step_time": 7.652675628662109
    },
    {
      "epoch": 4.510974884033203e-05,
      "model_forward_time": 1.6899981498718262,
      "step": 47301
    },
    {
      "epoch": 4.510974884033203e-05,
      "step": 47301,
      "training_step_time": 8.018007278442383
    },
    {
      "epoch": 4.511070251464844e-05,
      "model_forward_time": 2.2262661457061768,
      "step": 47302
    },
    {
      "epoch": 4.511070251464844e-05,
      "step": 47302,
      "training_step_time": 7.948395013809204
    },
    {
      "epoch": 4.511165618896484e-05,
      "model_forward_time": 1.6905322074890137,
      "step": 47303
    },
    {
      "epoch": 4.511165618896484e-05,
      "step": 47303,
      "training_step_time": 9.14249873161316
    },
    {
      "epoch": 4.511260986328125e-05,
      "model_forward_time": 1.6854946613311768,
      "step": 47304
    },
    {
      "epoch": 4.511260986328125e-05,
      "step": 47304,
      "training_step_time": 9.195737838745117
    },
    {
      "epoch": 4.5113563537597656e-05,
      "model_forward_time": 1.7139713764190674,
      "step": 47305
    },
    {
      "epoch": 4.5113563537597656e-05,
      "step": 47305,
      "training_step_time": 8.470147132873535
    },
    {
      "epoch": 4.511451721191406e-05,
      "model_forward_time": 1.6843111515045166,
      "step": 47306
    },
    {
      "epoch": 4.511451721191406e-05,
      "step": 47306,
      "training_step_time": 8.665916204452515
    },
    {
      "epoch": 4.511547088623047e-05,
      "model_forward_time": 1.6971282958984375,
      "step": 47307
    },
    {
      "epoch": 4.511547088623047e-05,
      "step": 47307,
      "training_step_time": 7.1109325885772705
    },
    {
      "epoch": 4.5116424560546875e-05,
      "model_forward_time": 1.689335584640503,
      "step": 47308
    },
    {
      "epoch": 4.5116424560546875e-05,
      "step": 47308,
      "training_step_time": 7.813710927963257
    },
    {
      "epoch": 4.511737823486328e-05,
      "model_forward_time": 1.6854197978973389,
      "step": 47309
    },
    {
      "epoch": 4.511737823486328e-05,
      "step": 47309,
      "training_step_time": 7.553962707519531
    },
    {
      "epoch": 4.511833190917969e-05,
      "grad_norm": 0.10929284989833832,
      "learning_rate": 5.854106874159925e-06,
      "loss": 0.0613,
      "step": 47310
    },
    {
      "epoch": 4.511833190917969e-05,
      "model_forward_time": 1.6912717819213867,
      "step": 47310
    },
    {
      "dynamics_loss_avg": 0.05230939816683531,
      "epoch": 4.511833190917969e-05,
      "step": 47310
    },
    {
      "action_loss_avg": 0.001292413895134814,
      "epoch": 4.511833190917969e-05,
      "step": 47310
    },
    {
      "epoch": 4.511833190917969e-05,
      "step": 47310,
      "training_step_time": 7.893848419189453
    },
    {
      "epoch": 4.5119285583496094e-05,
      "model_forward_time": 2.23433256149292,
      "step": 47311
    },
    {
      "epoch": 4.5119285583496094e-05,
      "step": 47311,
      "training_step_time": 7.898003101348877
    },
    {
      "epoch": 4.51202392578125e-05,
      "model_forward_time": 1.6698365211486816,
      "step": 47312
    },
    {
      "epoch": 4.51202392578125e-05,
      "step": 47312,
      "training_step_time": 9.274080991744995
    },
    {
      "epoch": 4.512119293212891e-05,
      "model_forward_time": 1.706045389175415,
      "step": 47313
    },
    {
      "epoch": 4.512119293212891e-05,
      "step": 47313,
      "training_step_time": 9.774664163589478
    },
    {
      "epoch": 4.512214660644531e-05,
      "model_forward_time": 1.6981146335601807,
      "step": 47314
    },
    {
      "epoch": 4.512214660644531e-05,
      "step": 47314,
      "training_step_time": 7.914410591125488
    },
    {
      "epoch": 4.5123100280761717e-05,
      "model_forward_time": 1.686605453491211,
      "step": 47315
    },
    {
      "epoch": 4.5123100280761717e-05,
      "step": 47315,
      "training_step_time": 7.900285243988037
    },
    {
      "epoch": 4.512405395507813e-05,
      "model_forward_time": 1.6864118576049805,
      "step": 47316
    },
    {
      "epoch": 4.512405395507813e-05,
      "step": 47316,
      "training_step_time": 7.831125974655151
    },
    {
      "epoch": 4.512500762939453e-05,
      "model_forward_time": 1.6901025772094727,
      "step": 47317
    },
    {
      "epoch": 4.512500762939453e-05,
      "step": 47317,
      "training_step_time": 7.753597021102905
    },
    {
      "epoch": 4.5125961303710935e-05,
      "model_forward_time": 1.6843101978302002,
      "step": 47318
    },
    {
      "epoch": 4.5125961303710935e-05,
      "step": 47318,
      "training_step_time": 7.473522186279297
    },
    {
      "epoch": 4.5126914978027346e-05,
      "model_forward_time": 1.6848771572113037,
      "step": 47319
    },
    {
      "epoch": 4.5126914978027346e-05,
      "step": 47319,
      "training_step_time": 8.421339988708496
    },
    {
      "epoch": 4.512786865234375e-05,
      "grad_norm": 0.13849522173404694,
      "learning_rate": 5.852477660346999e-06,
      "loss": 0.0597,
      "step": 47320
    },
    {
      "epoch": 4.512786865234375e-05,
      "model_forward_time": 2.226086378097534,
      "step": 47320
    },
    {
      "dynamics_loss_avg": 0.06843764949589967,
      "epoch": 4.512786865234375e-05,
      "step": 47320
    },
    {
      "action_loss_avg": 0.0012568979116622358,
      "epoch": 4.512786865234375e-05,
      "step": 47320
    },
    {
      "epoch": 4.512786865234375e-05,
      "step": 47320,
      "training_step_time": 8.479933977127075
    },
    {
      "epoch": 4.5128822326660154e-05,
      "model_forward_time": 1.6823382377624512,
      "step": 47321
    },
    {
      "epoch": 4.5128822326660154e-05,
      "step": 47321,
      "training_step_time": 9.575558423995972
    },
    {
      "epoch": 4.5129776000976565e-05,
      "model_forward_time": 1.7002606391906738,
      "step": 47322
    },
    {
      "epoch": 4.5129776000976565e-05,
      "step": 47322,
      "training_step_time": 10.942222833633423
    },
    {
      "epoch": 4.513072967529297e-05,
      "model_forward_time": 1.6822597980499268,
      "step": 47323
    },
    {
      "epoch": 4.513072967529297e-05,
      "step": 47323,
      "training_step_time": 9.643722295761108
    },
    {
      "epoch": 4.513168334960937e-05,
      "model_forward_time": 1.681882381439209,
      "step": 47324
    },
    {
      "epoch": 4.513168334960937e-05,
      "step": 47324,
      "training_step_time": 8.034510612487793
    },
    {
      "epoch": 4.5132637023925784e-05,
      "model_forward_time": 1.6819396018981934,
      "step": 47325
    },
    {
      "epoch": 4.5132637023925784e-05,
      "step": 47325,
      "training_step_time": 7.8585638999938965
    },
    {
      "epoch": 4.513359069824219e-05,
      "model_forward_time": 1.6849687099456787,
      "step": 47326
    },
    {
      "epoch": 4.513359069824219e-05,
      "step": 47326,
      "training_step_time": 7.782308578491211
    },
    {
      "epoch": 4.513454437255859e-05,
      "model_forward_time": 1.6872844696044922,
      "step": 47327
    },
    {
      "epoch": 4.513454437255859e-05,
      "step": 47327,
      "training_step_time": 7.4971654415130615
    },
    {
      "epoch": 4.5135498046875e-05,
      "model_forward_time": 1.693216323852539,
      "step": 47328
    },
    {
      "epoch": 4.5135498046875e-05,
      "step": 47328,
      "training_step_time": 7.9240217208862305
    },
    {
      "epoch": 4.513645172119141e-05,
      "model_forward_time": 2.2312843799591064,
      "step": 47329
    },
    {
      "epoch": 4.513645172119141e-05,
      "step": 47329,
      "training_step_time": 7.9142820835113525
    },
    {
      "epoch": 4.513740539550781e-05,
      "grad_norm": 0.1345919519662857,
      "learning_rate": 5.850848353308397e-06,
      "loss": 0.061,
      "step": 47330
    },
    {
      "epoch": 4.513740539550781e-05,
      "model_forward_time": 1.681004285812378,
      "step": 47330
    },
    {
      "dynamics_loss_avg": 0.0519436314702034,
      "epoch": 4.513740539550781e-05,
      "step": 47330
    },
    {
      "action_loss_avg": 0.0016597834008280187,
      "epoch": 4.513740539550781e-05,
      "step": 47330
    },
    {
      "epoch": 4.513740539550781e-05,
      "step": 47330,
      "training_step_time": 7.924239873886108
    },
    {
      "epoch": 4.513835906982422e-05,
      "model_forward_time": 1.694481611251831,
      "step": 47331
    },
    {
      "epoch": 4.513835906982422e-05,
      "step": 47331,
      "training_step_time": 10.487733602523804
    },
    {
      "epoch": 4.5139312744140626e-05,
      "model_forward_time": 1.3109526634216309,
      "step": 47332
    },
    {
      "epoch": 4.5139312744140626e-05,
      "step": 47332,
      "training_step_time": 8.046154499053955
    },
    {
      "epoch": 4.514026641845703e-05,
      "model_forward_time": 1.692502737045288,
      "step": 47333
    },
    {
      "epoch": 4.514026641845703e-05,
      "step": 47333,
      "training_step_time": 7.93326210975647
    },
    {
      "epoch": 4.514122009277344e-05,
      "model_forward_time": 1.68414306640625,
      "step": 47334
    },
    {
      "epoch": 4.514122009277344e-05,
      "step": 47334,
      "training_step_time": 7.933552980422974
    },
    {
      "epoch": 4.5142173767089845e-05,
      "model_forward_time": 1.6824629306793213,
      "step": 47335
    },
    {
      "epoch": 4.5142173767089845e-05,
      "step": 47335,
      "training_step_time": 8.618818044662476
    },
    {
      "epoch": 4.514312744140625e-05,
      "model_forward_time": 1.682262659072876,
      "step": 47336
    },
    {
      "epoch": 4.514312744140625e-05,
      "step": 47336,
      "training_step_time": 8.546206712722778
    },
    {
      "epoch": 4.514408111572266e-05,
      "model_forward_time": 1.687758445739746,
      "step": 47337
    },
    {
      "epoch": 4.514408111572266e-05,
      "step": 47337,
      "training_step_time": 7.892130136489868
    },
    {
      "epoch": 4.5145034790039063e-05,
      "model_forward_time": 2.2402141094207764,
      "step": 47338
    },
    {
      "epoch": 4.5145034790039063e-05,
      "step": 47338,
      "training_step_time": 8.014352798461914
    },
    {
      "epoch": 4.514598846435547e-05,
      "model_forward_time": 1.6823253631591797,
      "step": 47339
    },
    {
      "epoch": 4.514598846435547e-05,
      "step": 47339,
      "training_step_time": 10.732015371322632
    },
    {
      "epoch": 4.514694213867188e-05,
      "grad_norm": 0.126872256398201,
      "learning_rate": 5.849218953222297e-06,
      "loss": 0.0598,
      "step": 47340
    },
    {
      "epoch": 4.514694213867188e-05,
      "model_forward_time": 1.693037986755371,
      "step": 47340
    },
    {
      "dynamics_loss_avg": 0.06513929050415754,
      "epoch": 4.514694213867188e-05,
      "step": 47340
    },
    {
      "action_loss_avg": 0.0036622979678213596,
      "epoch": 4.514694213867188e-05,
      "step": 47340
    },
    {
      "epoch": 4.514694213867188e-05,
      "step": 47340,
      "training_step_time": 7.887368679046631
    },
    {
      "epoch": 4.514789581298828e-05,
      "model_forward_time": 1.6852188110351562,
      "step": 47341
    },
    {
      "epoch": 4.514789581298828e-05,
      "step": 47341,
      "training_step_time": 9.230759382247925
    },
    {
      "epoch": 4.5148849487304686e-05,
      "model_forward_time": 1.6941580772399902,
      "step": 47342
    },
    {
      "epoch": 4.5148849487304686e-05,
      "step": 47342,
      "training_step_time": 9.379266738891602
    },
    {
      "epoch": 4.51498031616211e-05,
      "model_forward_time": 1.3035228252410889,
      "step": 47343
    },
    {
      "epoch": 4.51498031616211e-05,
      "step": 47343,
      "training_step_time": 7.915175199508667
    },
    {
      "epoch": 4.51507568359375e-05,
      "model_forward_time": 1.6859750747680664,
      "step": 47344
    },
    {
      "epoch": 4.51507568359375e-05,
      "step": 47344,
      "training_step_time": 7.720127105712891
    },
    {
      "epoch": 4.5151710510253905e-05,
      "model_forward_time": 1.693593978881836,
      "step": 47345
    },
    {
      "epoch": 4.5151710510253905e-05,
      "step": 47345,
      "training_step_time": 7.758316278457642
    },
    {
      "epoch": 4.515266418457031e-05,
      "model_forward_time": 1.684147834777832,
      "step": 47346
    },
    {
      "epoch": 4.515266418457031e-05,
      "step": 47346,
      "training_step_time": 7.930538892745972
    },
    {
      "epoch": 4.515361785888672e-05,
      "model_forward_time": 2.247527599334717,
      "step": 47347
    },
    {
      "epoch": 4.515361785888672e-05,
      "step": 47347,
      "training_step_time": 7.973160266876221
    },
    {
      "epoch": 4.5154571533203124e-05,
      "model_forward_time": 1.6891720294952393,
      "step": 47348
    },
    {
      "epoch": 4.5154571533203124e-05,
      "step": 47348,
      "training_step_time": 7.874164581298828
    },
    {
      "epoch": 4.515552520751953e-05,
      "model_forward_time": 1.6787374019622803,
      "step": 47349
    },
    {
      "epoch": 4.515552520751953e-05,
      "step": 47349,
      "training_step_time": 8.738144636154175
    },
    {
      "epoch": 4.515647888183594e-05,
      "grad_norm": 0.16170775890350342,
      "learning_rate": 5.847589460266892e-06,
      "loss": 0.0629,
      "step": 47350
    },
    {
      "epoch": 4.515647888183594e-05,
      "model_forward_time": 1.6939260959625244,
      "step": 47350
    },
    {
      "dynamics_loss_avg": 0.053830991871654986,
      "epoch": 4.515647888183594e-05,
      "step": 47350
    },
    {
      "action_loss_avg": 0.0015564322471618651,
      "epoch": 4.515647888183594e-05,
      "step": 47350
    },
    {
      "epoch": 4.515647888183594e-05,
      "step": 47350,
      "training_step_time": 9.659436702728271
    },
    {
      "epoch": 4.515743255615234e-05,
      "model_forward_time": 1.6918060779571533,
      "step": 47351
    },
    {
      "epoch": 4.515743255615234e-05,
      "step": 47351,
      "training_step_time": 8.031071901321411
    },
    {
      "epoch": 4.515838623046875e-05,
      "model_forward_time": 1.687199592590332,
      "step": 47352
    },
    {
      "epoch": 4.515838623046875e-05,
      "step": 47352,
      "training_step_time": 7.944105386734009
    },
    {
      "epoch": 4.515933990478516e-05,
      "model_forward_time": 1.6829197406768799,
      "step": 47353
    },
    {
      "epoch": 4.515933990478516e-05,
      "step": 47353,
      "training_step_time": 7.883135557174683
    },
    {
      "epoch": 4.516029357910156e-05,
      "model_forward_time": 1.6820478439331055,
      "step": 47354
    },
    {
      "epoch": 4.516029357910156e-05,
      "step": 47354,
      "training_step_time": 7.785629510879517
    },
    {
      "epoch": 4.5161247253417966e-05,
      "model_forward_time": 1.6896288394927979,
      "step": 47355
    },
    {
      "epoch": 4.5161247253417966e-05,
      "step": 47355,
      "training_step_time": 7.973203897476196
    },
    {
      "epoch": 4.5162200927734377e-05,
      "model_forward_time": 1.6866955757141113,
      "step": 47356
    },
    {
      "epoch": 4.5162200927734377e-05,
      "step": 47356,
      "training_step_time": 8.514506816864014
    },
    {
      "epoch": 4.516315460205078e-05,
      "model_forward_time": 1.6917669773101807,
      "step": 47357
    },
    {
      "epoch": 4.516315460205078e-05,
      "step": 47357,
      "training_step_time": 8.545191287994385
    },
    {
      "epoch": 4.5164108276367185e-05,
      "model_forward_time": 1.6842050552368164,
      "step": 47358
    },
    {
      "epoch": 4.5164108276367185e-05,
      "step": 47358,
      "training_step_time": 8.411360502243042
    },
    {
      "epoch": 4.5165061950683595e-05,
      "model_forward_time": 1.703239917755127,
      "step": 47359
    },
    {
      "epoch": 4.5165061950683595e-05,
      "step": 47359,
      "training_step_time": 8.213682174682617
    },
    {
      "epoch": 4.5166015625e-05,
      "grad_norm": 0.13706788420677185,
      "learning_rate": 5.845959874620375e-06,
      "loss": 0.0596,
      "step": 47360
    },
    {
      "epoch": 4.5166015625e-05,
      "model_forward_time": 1.6818652153015137,
      "step": 47360
    },
    {
      "dynamics_loss_avg": 0.06247084345668554,
      "epoch": 4.5166015625e-05,
      "step": 47360
    },
    {
      "action_loss_avg": 0.0019770032027736305,
      "epoch": 4.5166015625e-05,
      "step": 47360
    },
    {
      "epoch": 4.5166015625e-05,
      "step": 47360,
      "training_step_time": 7.889773607254028
    },
    {
      "epoch": 4.5166969299316403e-05,
      "model_forward_time": 1.7074174880981445,
      "step": 47361
    },
    {
      "epoch": 4.5166969299316403e-05,
      "step": 47361,
      "training_step_time": 9.002803564071655
    },
    {
      "epoch": 4.5167922973632814e-05,
      "model_forward_time": 1.684887409210205,
      "step": 47362
    },
    {
      "epoch": 4.5167922973632814e-05,
      "step": 47362,
      "training_step_time": 7.894736289978027
    },
    {
      "epoch": 4.516887664794922e-05,
      "model_forward_time": 1.6784937381744385,
      "step": 47363
    },
    {
      "epoch": 4.516887664794922e-05,
      "step": 47363,
      "training_step_time": 7.832101821899414
    },
    {
      "epoch": 4.516983032226562e-05,
      "model_forward_time": 1.6906776428222656,
      "step": 47364
    },
    {
      "epoch": 4.516983032226562e-05,
      "step": 47364,
      "training_step_time": 7.917587995529175
    },
    {
      "epoch": 4.517078399658203e-05,
      "model_forward_time": 1.3057689666748047,
      "step": 47365
    },
    {
      "epoch": 4.517078399658203e-05,
      "step": 47365,
      "training_step_time": 7.852365970611572
    },
    {
      "epoch": 4.517173767089844e-05,
      "model_forward_time": 2.2273385524749756,
      "step": 47366
    },
    {
      "epoch": 4.517173767089844e-05,
      "step": 47366,
      "training_step_time": 7.8366076946258545
    },
    {
      "epoch": 4.517269134521484e-05,
      "model_forward_time": 1.680112361907959,
      "step": 47367
    },
    {
      "epoch": 4.517269134521484e-05,
      "step": 47367,
      "training_step_time": 9.24080514907837
    },
    {
      "epoch": 4.517364501953125e-05,
      "model_forward_time": 1.6999454498291016,
      "step": 47368
    },
    {
      "epoch": 4.517364501953125e-05,
      "step": 47368,
      "training_step_time": 9.904764652252197
    },
    {
      "epoch": 4.5174598693847656e-05,
      "model_forward_time": 1.6856579780578613,
      "step": 47369
    },
    {
      "epoch": 4.5174598693847656e-05,
      "step": 47369,
      "training_step_time": 7.910651922225952
    },
    {
      "epoch": 4.517555236816406e-05,
      "grad_norm": 0.11085748672485352,
      "learning_rate": 5.84433019646096e-06,
      "loss": 0.0618,
      "step": 47370
    },
    {
      "epoch": 4.517555236816406e-05,
      "model_forward_time": 1.6954717636108398,
      "step": 47370
    },
    {
      "dynamics_loss_avg": 0.05334329605102539,
      "epoch": 4.517555236816406e-05,
      "step": 47370
    },
    {
      "action_loss_avg": 0.0021332833275664598,
      "epoch": 4.517555236816406e-05,
      "step": 47370
    },
    {
      "epoch": 4.517555236816406e-05,
      "step": 47370,
      "training_step_time": 7.965751647949219
    },
    {
      "epoch": 4.517650604248047e-05,
      "model_forward_time": 1.6870641708374023,
      "step": 47371
    },
    {
      "epoch": 4.517650604248047e-05,
      "step": 47371,
      "training_step_time": 8.451614618301392
    },
    {
      "epoch": 4.5177459716796875e-05,
      "model_forward_time": 1.6813328266143799,
      "step": 47372
    },
    {
      "epoch": 4.5177459716796875e-05,
      "step": 47372,
      "training_step_time": 7.661792039871216
    },
    {
      "epoch": 4.517841339111328e-05,
      "model_forward_time": 1.6844165325164795,
      "step": 47373
    },
    {
      "epoch": 4.517841339111328e-05,
      "step": 47373,
      "training_step_time": 8.032924890518188
    },
    {
      "epoch": 4.517936706542969e-05,
      "model_forward_time": 1.6857562065124512,
      "step": 47374
    },
    {
      "epoch": 4.517936706542969e-05,
      "step": 47374,
      "training_step_time": 7.744641304016113
    },
    {
      "epoch": 4.5180320739746094e-05,
      "model_forward_time": 2.2257282733917236,
      "step": 47375
    },
    {
      "epoch": 4.5180320739746094e-05,
      "step": 47375,
      "training_step_time": 7.942580938339233
    },
    {
      "epoch": 4.51812744140625e-05,
      "model_forward_time": 1.6855494976043701,
      "step": 47376
    },
    {
      "epoch": 4.51812744140625e-05,
      "step": 47376,
      "training_step_time": 8.062938451766968
    },
    {
      "epoch": 4.518222808837891e-05,
      "model_forward_time": 1.6923668384552002,
      "step": 47377
    },
    {
      "epoch": 4.518222808837891e-05,
      "step": 47377,
      "training_step_time": 8.861114978790283
    },
    {
      "epoch": 4.518318176269531e-05,
      "model_forward_time": 1.6840143203735352,
      "step": 47378
    },
    {
      "epoch": 4.518318176269531e-05,
      "step": 47378,
      "training_step_time": 7.880058765411377
    },
    {
      "epoch": 4.518413543701172e-05,
      "model_forward_time": 1.308194875717163,
      "step": 47379
    },
    {
      "epoch": 4.518413543701172e-05,
      "step": 47379,
      "training_step_time": 8.903602600097656
    },
    {
      "epoch": 4.518508911132813e-05,
      "grad_norm": 0.10547171533107758,
      "learning_rate": 5.842700425966863e-06,
      "loss": 0.0557,
      "step": 47380
    },
    {
      "epoch": 4.518508911132813e-05,
      "model_forward_time": 1.68064284324646,
      "step": 47380
    },
    {
      "dynamics_loss_avg": 0.06297222478315234,
      "epoch": 4.518508911132813e-05,
      "step": 47380
    },
    {
      "action_loss_avg": 0.0019801496615400537,
      "epoch": 4.518508911132813e-05,
      "step": 47380
    },
    {
      "epoch": 4.518508911132813e-05,
      "step": 47380,
      "training_step_time": 7.87403130531311
    },
    {
      "epoch": 4.518604278564453e-05,
      "model_forward_time": 1.6806352138519287,
      "step": 47381
    },
    {
      "epoch": 4.518604278564453e-05,
      "step": 47381,
      "training_step_time": 7.948266506195068
    },
    {
      "epoch": 4.5186996459960935e-05,
      "model_forward_time": 1.6807036399841309,
      "step": 47382
    },
    {
      "epoch": 4.5186996459960935e-05,
      "step": 47382,
      "training_step_time": 7.851903200149536
    },
    {
      "epoch": 4.5187950134277346e-05,
      "model_forward_time": 1.6822130680084229,
      "step": 47383
    },
    {
      "epoch": 4.5187950134277346e-05,
      "step": 47383,
      "training_step_time": 7.956128835678101
    },
    {
      "epoch": 4.518890380859375e-05,
      "model_forward_time": 2.254154682159424,
      "step": 47384
    },
    {
      "epoch": 4.518890380859375e-05,
      "step": 47384,
      "training_step_time": 8.015133380889893
    },
    {
      "epoch": 4.5189857482910154e-05,
      "model_forward_time": 1.68320631980896,
      "step": 47385
    },
    {
      "epoch": 4.5189857482910154e-05,
      "step": 47385,
      "training_step_time": 7.922720432281494
    },
    {
      "epoch": 4.5190811157226565e-05,
      "model_forward_time": 1.6899781227111816,
      "step": 47386
    },
    {
      "epoch": 4.5190811157226565e-05,
      "step": 47386,
      "training_step_time": 7.925006151199341
    },
    {
      "epoch": 4.519176483154297e-05,
      "model_forward_time": 1.3026080131530762,
      "step": 47387
    },
    {
      "epoch": 4.519176483154297e-05,
      "step": 47387,
      "training_step_time": 8.766129732131958
    },
    {
      "epoch": 4.519271850585937e-05,
      "model_forward_time": 1.307295560836792,
      "step": 47388
    },
    {
      "epoch": 4.519271850585937e-05,
      "step": 47388,
      "training_step_time": 9.015601634979248
    },
    {
      "epoch": 4.5193672180175784e-05,
      "model_forward_time": 1.6837501525878906,
      "step": 47389
    },
    {
      "epoch": 4.5193672180175784e-05,
      "step": 47389,
      "training_step_time": 8.422514915466309
    },
    {
      "epoch": 4.519462585449219e-05,
      "grad_norm": 0.13555139303207397,
      "learning_rate": 5.841070563316316e-06,
      "loss": 0.0551,
      "step": 47390
    },
    {
      "epoch": 4.519462585449219e-05,
      "model_forward_time": 1.681692123413086,
      "step": 47390
    },
    {
      "dynamics_loss_avg": 0.06271015740931034,
      "epoch": 4.519462585449219e-05,
      "step": 47390
    },
    {
      "action_loss_avg": 0.0029980550229083748,
      "epoch": 4.519462585449219e-05,
      "step": 47390
    },
    {
      "epoch": 4.519462585449219e-05,
      "step": 47390,
      "training_step_time": 7.872188329696655
    },
    {
      "epoch": 4.519557952880859e-05,
      "model_forward_time": 1.6910088062286377,
      "step": 47391
    },
    {
      "epoch": 4.519557952880859e-05,
      "step": 47391,
      "training_step_time": 7.7863242626190186
    },
    {
      "epoch": 4.5196533203125e-05,
      "model_forward_time": 1.6865439414978027,
      "step": 47392
    },
    {
      "epoch": 4.5196533203125e-05,
      "step": 47392,
      "training_step_time": 7.850276708602905
    },
    {
      "epoch": 4.519748687744141e-05,
      "model_forward_time": 1.6854205131530762,
      "step": 47393
    },
    {
      "epoch": 4.519748687744141e-05,
      "step": 47393,
      "training_step_time": 7.924422264099121
    },
    {
      "epoch": 4.519844055175781e-05,
      "model_forward_time": 2.2224221229553223,
      "step": 47394
    },
    {
      "epoch": 4.519844055175781e-05,
      "step": 47394,
      "training_step_time": 7.897743463516235
    },
    {
      "epoch": 4.519939422607422e-05,
      "model_forward_time": 1.6831345558166504,
      "step": 47395
    },
    {
      "epoch": 4.519939422607422e-05,
      "step": 47395,
      "training_step_time": 7.913844585418701
    },
    {
      "epoch": 4.5200347900390626e-05,
      "model_forward_time": 1.689514398574829,
      "step": 47396
    },
    {
      "epoch": 4.5200347900390626e-05,
      "step": 47396,
      "training_step_time": 10.295002937316895
    },
    {
      "epoch": 4.520130157470703e-05,
      "model_forward_time": 1.6914794445037842,
      "step": 47397
    },
    {
      "epoch": 4.520130157470703e-05,
      "step": 47397,
      "training_step_time": 8.963583707809448
    },
    {
      "epoch": 4.520225524902344e-05,
      "model_forward_time": 1.6852364540100098,
      "step": 47398
    },
    {
      "epoch": 4.520225524902344e-05,
      "step": 47398,
      "training_step_time": 8.676546573638916
    },
    {
      "epoch": 4.5203208923339845e-05,
      "model_forward_time": 1.692225694656372,
      "step": 47399
    },
    {
      "epoch": 4.5203208923339845e-05,
      "step": 47399,
      "training_step_time": 8.456922054290771
    },
    {
      "epoch": 4.520416259765625e-05,
      "grad_norm": 0.13654376566410065,
      "learning_rate": 5.839440608687555e-06,
      "loss": 0.0584,
      "step": 47400
    },
    {
      "epoch": 4.520416259765625e-05,
      "model_forward_time": 1.6861529350280762,
      "step": 47400
    },
    {
      "dynamics_loss_avg": 0.058509936183691026,
      "epoch": 4.520416259765625e-05,
      "step": 47400
    },
    {
      "action_loss_avg": 0.0017399114149156958,
      "epoch": 4.520416259765625e-05,
      "step": 47400
    },
    {
      "epoch": 4.520416259765625e-05,
      "step": 47400,
      "training_step_time": 7.825017690658569
    },
    {
      "epoch": 4.520511627197266e-05,
      "model_forward_time": 1.6940934658050537,
      "step": 47401
    },
    {
      "epoch": 4.520511627197266e-05,
      "step": 47401,
      "training_step_time": 7.917094707489014
    },
    {
      "epoch": 4.5206069946289063e-05,
      "model_forward_time": 1.6954805850982666,
      "step": 47402
    },
    {
      "epoch": 4.5206069946289063e-05,
      "step": 47402,
      "training_step_time": 8.593979597091675
    },
    {
      "epoch": 4.520702362060547e-05,
      "model_forward_time": 2.2444779872894287,
      "step": 47403
    },
    {
      "epoch": 4.520702362060547e-05,
      "step": 47403,
      "training_step_time": 7.903052806854248
    },
    {
      "epoch": 4.520797729492188e-05,
      "model_forward_time": 1.6873760223388672,
      "step": 47404
    },
    {
      "epoch": 4.520797729492188e-05,
      "step": 47404,
      "training_step_time": 7.9590418338775635
    },
    {
      "epoch": 4.520893096923828e-05,
      "model_forward_time": 1.700387954711914,
      "step": 47405
    },
    {
      "epoch": 4.520893096923828e-05,
      "step": 47405,
      "training_step_time": 9.19779109954834
    },
    {
      "epoch": 4.5209884643554686e-05,
      "model_forward_time": 1.6886191368103027,
      "step": 47406
    },
    {
      "epoch": 4.5209884643554686e-05,
      "step": 47406,
      "training_step_time": 7.995429754257202
    },
    {
      "epoch": 4.521083831787109e-05,
      "model_forward_time": 1.6965816020965576,
      "step": 47407
    },
    {
      "epoch": 4.521083831787109e-05,
      "step": 47407,
      "training_step_time": 7.967937231063843
    },
    {
      "epoch": 4.52117919921875e-05,
      "model_forward_time": 1.6901154518127441,
      "step": 47408
    },
    {
      "epoch": 4.52117919921875e-05,
      "step": 47408,
      "training_step_time": 7.895367383956909
    },
    {
      "epoch": 4.5212745666503905e-05,
      "model_forward_time": 1.6964099407196045,
      "step": 47409
    },
    {
      "epoch": 4.5212745666503905e-05,
      "step": 47409,
      "training_step_time": 7.846688985824585
    },
    {
      "epoch": 4.521369934082031e-05,
      "grad_norm": 0.10749403387308121,
      "learning_rate": 5.837810562258831e-06,
      "loss": 0.0594,
      "step": 47410
    },
    {
      "epoch": 4.521369934082031e-05,
      "model_forward_time": 1.6897258758544922,
      "step": 47410
    },
    {
      "dynamics_loss_avg": 0.06233064737170935,
      "epoch": 4.521369934082031e-05,
      "step": 47410
    },
    {
      "action_loss_avg": 0.0025236239773221312,
      "epoch": 4.521369934082031e-05,
      "step": 47410
    },
    {
      "epoch": 4.521369934082031e-05,
      "step": 47410,
      "training_step_time": 7.04914665222168
    },
    {
      "epoch": 4.521465301513672e-05,
      "model_forward_time": 1.6768290996551514,
      "step": 47411
    },
    {
      "epoch": 4.521465301513672e-05,
      "step": 47411,
      "training_step_time": 8.590936183929443
    },
    {
      "epoch": 4.5215606689453124e-05,
      "model_forward_time": 2.2523982524871826,
      "step": 47412
    },
    {
      "epoch": 4.5215606689453124e-05,
      "step": 47412,
      "training_step_time": 7.883490324020386
    },
    {
      "epoch": 4.521656036376953e-05,
      "model_forward_time": 1.685481309890747,
      "step": 47413
    },
    {
      "epoch": 4.521656036376953e-05,
      "step": 47413,
      "training_step_time": 7.881568670272827
    },
    {
      "epoch": 4.521751403808594e-05,
      "model_forward_time": 1.7042350769042969,
      "step": 47414
    },
    {
      "epoch": 4.521751403808594e-05,
      "step": 47414,
      "training_step_time": 8.237112998962402
    },
    {
      "epoch": 4.521846771240234e-05,
      "model_forward_time": 1.703303575515747,
      "step": 47415
    },
    {
      "epoch": 4.521846771240234e-05,
      "step": 47415,
      "training_step_time": 9.09542441368103
    },
    {
      "epoch": 4.521942138671875e-05,
      "model_forward_time": 1.6986401081085205,
      "step": 47416
    },
    {
      "epoch": 4.521942138671875e-05,
      "step": 47416,
      "training_step_time": 7.95682692527771
    },
    {
      "epoch": 4.522037506103516e-05,
      "model_forward_time": 1.6911447048187256,
      "step": 47417
    },
    {
      "epoch": 4.522037506103516e-05,
      "step": 47417,
      "training_step_time": 7.948607921600342
    },
    {
      "epoch": 4.522132873535156e-05,
      "model_forward_time": 1.6817700862884521,
      "step": 47418
    },
    {
      "epoch": 4.522132873535156e-05,
      "step": 47418,
      "training_step_time": 7.96699333190918
    },
    {
      "epoch": 4.5222282409667966e-05,
      "model_forward_time": 1.6862943172454834,
      "step": 47419
    },
    {
      "epoch": 4.5222282409667966e-05,
      "step": 47419,
      "training_step_time": 7.147278785705566
    },
    {
      "epoch": 4.522323608398438e-05,
      "grad_norm": 0.0941874235868454,
      "learning_rate": 5.836180424208405e-06,
      "loss": 0.057,
      "step": 47420
    },
    {
      "epoch": 4.522323608398438e-05,
      "model_forward_time": 1.6925725936889648,
      "step": 47420
    },
    {
      "dynamics_loss_avg": 0.05817762333899736,
      "epoch": 4.522323608398438e-05,
      "step": 47420
    },
    {
      "action_loss_avg": 0.0020299550902564078,
      "epoch": 4.522323608398438e-05,
      "step": 47420
    },
    {
      "epoch": 4.522323608398438e-05,
      "step": 47420,
      "training_step_time": 8.096429586410522
    },
    {
      "epoch": 4.522418975830078e-05,
      "model_forward_time": 2.2426180839538574,
      "step": 47421
    },
    {
      "epoch": 4.522418975830078e-05,
      "step": 47421,
      "training_step_time": 8.829475164413452
    },
    {
      "epoch": 4.5225143432617185e-05,
      "model_forward_time": 1.69108247756958,
      "step": 47422
    },
    {
      "epoch": 4.5225143432617185e-05,
      "step": 47422,
      "training_step_time": 7.9426538944244385
    },
    {
      "epoch": 4.5226097106933595e-05,
      "model_forward_time": 1.704422950744629,
      "step": 47423
    },
    {
      "epoch": 4.5226097106933595e-05,
      "step": 47423,
      "training_step_time": 9.695540428161621
    },
    {
      "epoch": 4.522705078125e-05,
      "model_forward_time": 1.6981842517852783,
      "step": 47424
    },
    {
      "epoch": 4.522705078125e-05,
      "step": 47424,
      "training_step_time": 8.496463060379028
    },
    {
      "epoch": 4.5228004455566404e-05,
      "model_forward_time": 1.6996946334838867,
      "step": 47425
    },
    {
      "epoch": 4.5228004455566404e-05,
      "step": 47425,
      "training_step_time": 7.8968236446380615
    },
    {
      "epoch": 4.5228958129882814e-05,
      "model_forward_time": 1.6847820281982422,
      "step": 47426
    },
    {
      "epoch": 4.5228958129882814e-05,
      "step": 47426,
      "training_step_time": 7.9306206703186035
    },
    {
      "epoch": 4.522991180419922e-05,
      "model_forward_time": 1.6901841163635254,
      "step": 47427
    },
    {
      "epoch": 4.522991180419922e-05,
      "step": 47427,
      "training_step_time": 8.017800569534302
    },
    {
      "epoch": 4.523086547851562e-05,
      "model_forward_time": 1.6838901042938232,
      "step": 47428
    },
    {
      "epoch": 4.523086547851562e-05,
      "step": 47428,
      "training_step_time": 7.08500337600708
    },
    {
      "epoch": 4.523181915283203e-05,
      "model_forward_time": 1.6847267150878906,
      "step": 47429
    },
    {
      "epoch": 4.523181915283203e-05,
      "step": 47429,
      "training_step_time": 7.976617336273193
    },
    {
      "epoch": 4.523277282714844e-05,
      "grad_norm": 0.10720636695623398,
      "learning_rate": 5.834550194714543e-06,
      "loss": 0.0546,
      "step": 47430
    },
    {
      "epoch": 4.523277282714844e-05,
      "model_forward_time": 1.685826301574707,
      "step": 47430
    },
    {
      "dynamics_loss_avg": 0.07079164460301399,
      "epoch": 4.523277282714844e-05,
      "step": 47430
    },
    {
      "action_loss_avg": 0.0027762269659433515,
      "epoch": 4.523277282714844e-05,
      "step": 47430
    },
    {
      "epoch": 4.523277282714844e-05,
      "step": 47430,
      "training_step_time": 8.566569089889526
    },
    {
      "epoch": 4.523372650146484e-05,
      "model_forward_time": 1.686676263809204,
      "step": 47431
    },
    {
      "epoch": 4.523372650146484e-05,
      "step": 47431,
      "training_step_time": 9.026406288146973
    },
    {
      "epoch": 4.523468017578125e-05,
      "model_forward_time": 1.69913911819458,
      "step": 47432
    },
    {
      "epoch": 4.523468017578125e-05,
      "step": 47432,
      "training_step_time": 8.870237827301025
    },
    {
      "epoch": 4.5235633850097656e-05,
      "model_forward_time": 1.687326431274414,
      "step": 47433
    },
    {
      "epoch": 4.5235633850097656e-05,
      "step": 47433,
      "training_step_time": 8.494869470596313
    },
    {
      "epoch": 4.523658752441406e-05,
      "model_forward_time": 1.6854751110076904,
      "step": 47434
    },
    {
      "epoch": 4.523658752441406e-05,
      "step": 47434,
      "training_step_time": 7.958406209945679
    },
    {
      "epoch": 4.523754119873047e-05,
      "model_forward_time": 1.6866631507873535,
      "step": 47435
    },
    {
      "epoch": 4.523754119873047e-05,
      "step": 47435,
      "training_step_time": 7.924494504928589
    },
    {
      "epoch": 4.5238494873046875e-05,
      "model_forward_time": 1.6950557231903076,
      "step": 47436
    },
    {
      "epoch": 4.5238494873046875e-05,
      "step": 47436,
      "training_step_time": 8.023888349533081
    },
    {
      "epoch": 4.523944854736328e-05,
      "model_forward_time": 1.690903902053833,
      "step": 47437
    },
    {
      "epoch": 4.523944854736328e-05,
      "step": 47437,
      "training_step_time": 7.421814203262329
    },
    {
      "epoch": 4.524040222167969e-05,
      "model_forward_time": 1.681645154953003,
      "step": 47438
    },
    {
      "epoch": 4.524040222167969e-05,
      "step": 47438,
      "training_step_time": 8.02044415473938
    },
    {
      "epoch": 4.5241355895996094e-05,
      "model_forward_time": 1.6849284172058105,
      "step": 47439
    },
    {
      "epoch": 4.5241355895996094e-05,
      "step": 47439,
      "training_step_time": 7.931839227676392
    },
    {
      "epoch": 4.52423095703125e-05,
      "grad_norm": 0.12283221632242203,
      "learning_rate": 5.832919873955528e-06,
      "loss": 0.0562,
      "step": 47440
    },
    {
      "epoch": 4.52423095703125e-05,
      "model_forward_time": 2.224221706390381,
      "step": 47440
    },
    {
      "dynamics_loss_avg": 0.061301720514893535,
      "epoch": 4.52423095703125e-05,
      "step": 47440
    },
    {
      "action_loss_avg": 0.0021076873992569743,
      "epoch": 4.52423095703125e-05,
      "step": 47440
    },
    {
      "epoch": 4.52423095703125e-05,
      "step": 47440,
      "training_step_time": 7.950304269790649
    },
    {
      "epoch": 4.524326324462891e-05,
      "model_forward_time": 1.6878035068511963,
      "step": 47441
    },
    {
      "epoch": 4.524326324462891e-05,
      "step": 47441,
      "training_step_time": 8.40332841873169
    },
    {
      "epoch": 4.524421691894531e-05,
      "model_forward_time": 1.6829378604888916,
      "step": 47442
    },
    {
      "epoch": 4.524421691894531e-05,
      "step": 47442,
      "training_step_time": 9.660956859588623
    },
    {
      "epoch": 4.524517059326172e-05,
      "model_forward_time": 1.6818196773529053,
      "step": 47443
    },
    {
      "epoch": 4.524517059326172e-05,
      "step": 47443,
      "training_step_time": 7.94090723991394
    },
    {
      "epoch": 4.524612426757813e-05,
      "model_forward_time": 1.6718170642852783,
      "step": 47444
    },
    {
      "epoch": 4.524612426757813e-05,
      "step": 47444,
      "training_step_time": 7.914106607437134
    },
    {
      "epoch": 4.524707794189453e-05,
      "model_forward_time": 1.6814591884613037,
      "step": 47445
    },
    {
      "epoch": 4.524707794189453e-05,
      "step": 47445,
      "training_step_time": 7.959815740585327
    },
    {
      "epoch": 4.5248031616210936e-05,
      "model_forward_time": 1.6876847743988037,
      "step": 47446
    },
    {
      "epoch": 4.5248031616210936e-05,
      "step": 47446,
      "training_step_time": 7.745950222015381
    },
    {
      "epoch": 4.5248985290527346e-05,
      "model_forward_time": 1.6872167587280273,
      "step": 47447
    },
    {
      "epoch": 4.5248985290527346e-05,
      "step": 47447,
      "training_step_time": 8.680272340774536
    },
    {
      "epoch": 4.524993896484375e-05,
      "model_forward_time": 1.6970975399017334,
      "step": 47448
    },
    {
      "epoch": 4.524993896484375e-05,
      "step": 47448,
      "training_step_time": 7.914987802505493
    },
    {
      "epoch": 4.5250892639160154e-05,
      "model_forward_time": 2.2296431064605713,
      "step": 47449
    },
    {
      "epoch": 4.5250892639160154e-05,
      "step": 47449,
      "training_step_time": 7.872643947601318
    },
    {
      "epoch": 4.5251846313476565e-05,
      "grad_norm": 0.14461649954319,
      "learning_rate": 5.831289462109648e-06,
      "loss": 0.0606,
      "step": 47450
    },
    {
      "epoch": 4.5251846313476565e-05,
      "model_forward_time": 1.6743824481964111,
      "step": 47450
    },
    {
      "dynamics_loss_avg": 0.06710640285164118,
      "epoch": 4.5251846313476565e-05,
      "step": 47450
    },
    {
      "action_loss_avg": 0.0011462964466772974,
      "epoch": 4.5251846313476565e-05,
      "step": 47450
    },
    {
      "epoch": 4.5251846313476565e-05,
      "step": 47450,
      "training_step_time": 7.795331239700317
    },
    {
      "epoch": 4.525279998779297e-05,
      "model_forward_time": 1.6966040134429932,
      "step": 47451
    },
    {
      "epoch": 4.525279998779297e-05,
      "step": 47451,
      "training_step_time": 7.941012620925903
    },
    {
      "epoch": 4.525375366210937e-05,
      "model_forward_time": 1.6817436218261719,
      "step": 47452
    },
    {
      "epoch": 4.525375366210937e-05,
      "step": 47452,
      "training_step_time": 8.518407821655273
    },
    {
      "epoch": 4.5254707336425784e-05,
      "model_forward_time": 1.7044165134429932,
      "step": 47453
    },
    {
      "epoch": 4.5254707336425784e-05,
      "step": 47453,
      "training_step_time": 7.910288095474243
    },
    {
      "epoch": 4.525566101074219e-05,
      "model_forward_time": 1.6895973682403564,
      "step": 47454
    },
    {
      "epoch": 4.525566101074219e-05,
      "step": 47454,
      "training_step_time": 7.9722371101379395
    },
    {
      "epoch": 4.525661468505859e-05,
      "model_forward_time": 1.6847343444824219,
      "step": 47455
    },
    {
      "epoch": 4.525661468505859e-05,
      "step": 47455,
      "training_step_time": 7.841932535171509
    },
    {
      "epoch": 4.5257568359375e-05,
      "model_forward_time": 1.6883094310760498,
      "step": 47456
    },
    {
      "epoch": 4.5257568359375e-05,
      "step": 47456,
      "training_step_time": 7.964142084121704
    },
    {
      "epoch": 4.525852203369141e-05,
      "model_forward_time": 1.693061113357544,
      "step": 47457
    },
    {
      "epoch": 4.525852203369141e-05,
      "step": 47457,
      "training_step_time": 8.86691951751709
    },
    {
      "epoch": 4.525947570800781e-05,
      "model_forward_time": 2.238726854324341,
      "step": 47458
    },
    {
      "epoch": 4.525947570800781e-05,
      "step": 47458,
      "training_step_time": 8.452124834060669
    },
    {
      "epoch": 4.526042938232422e-05,
      "model_forward_time": 1.6901581287384033,
      "step": 47459
    },
    {
      "epoch": 4.526042938232422e-05,
      "step": 47459,
      "training_step_time": 9.146892309188843
    },
    {
      "epoch": 4.5261383056640626e-05,
      "grad_norm": 0.1125950887799263,
      "learning_rate": 5.829658959355203e-06,
      "loss": 0.0604,
      "step": 47460
    },
    {
      "epoch": 4.5261383056640626e-05,
      "model_forward_time": 1.6983282566070557,
      "step": 47460
    },
    {
      "dynamics_loss_avg": 0.053959932457655665,
      "epoch": 4.5261383056640626e-05,
      "step": 47460
    },
    {
      "action_loss_avg": 0.001573332175030373,
      "epoch": 4.5261383056640626e-05,
      "step": 47460
    },
    {
      "epoch": 4.5261383056640626e-05,
      "step": 47460,
      "training_step_time": 9.440933465957642
    },
    {
      "epoch": 4.526233673095703e-05,
      "model_forward_time": 1.6910200119018555,
      "step": 47461
    },
    {
      "epoch": 4.526233673095703e-05,
      "step": 47461,
      "training_step_time": 8.978830099105835
    },
    {
      "epoch": 4.526329040527344e-05,
      "model_forward_time": 1.6981642246246338,
      "step": 47462
    },
    {
      "epoch": 4.526329040527344e-05,
      "step": 47462,
      "training_step_time": 9.034863471984863
    },
    {
      "epoch": 4.5264244079589845e-05,
      "model_forward_time": 1.682525873184204,
      "step": 47463
    },
    {
      "epoch": 4.5264244079589845e-05,
      "step": 47463,
      "training_step_time": 7.949366331100464
    },
    {
      "epoch": 4.526519775390625e-05,
      "model_forward_time": 1.6818132400512695,
      "step": 47464
    },
    {
      "epoch": 4.526519775390625e-05,
      "step": 47464,
      "training_step_time": 8.08893609046936
    },
    {
      "epoch": 4.526615142822266e-05,
      "model_forward_time": 1.695929765701294,
      "step": 47465
    },
    {
      "epoch": 4.526615142822266e-05,
      "step": 47465,
      "training_step_time": 8.006572961807251
    },
    {
      "epoch": 4.5267105102539064e-05,
      "model_forward_time": 1.6789741516113281,
      "step": 47466
    },
    {
      "epoch": 4.5267105102539064e-05,
      "step": 47466,
      "training_step_time": 7.955460786819458
    },
    {
      "epoch": 4.526805877685547e-05,
      "model_forward_time": 2.2449750900268555,
      "step": 47467
    },
    {
      "epoch": 4.526805877685547e-05,
      "step": 47467,
      "training_step_time": 8.835974931716919
    },
    {
      "epoch": 4.526901245117188e-05,
      "model_forward_time": 1.683375358581543,
      "step": 47468
    },
    {
      "epoch": 4.526901245117188e-05,
      "step": 47468,
      "training_step_time": 9.324755191802979
    },
    {
      "epoch": 4.526996612548828e-05,
      "model_forward_time": 1.708193302154541,
      "step": 47469
    },
    {
      "epoch": 4.526996612548828e-05,
      "step": 47469,
      "training_step_time": 8.693663120269775
    },
    {
      "epoch": 4.5270919799804686e-05,
      "grad_norm": 0.11172918230295181,
      "learning_rate": 5.828028365870499e-06,
      "loss": 0.0587,
      "step": 47470
    },
    {
      "epoch": 4.5270919799804686e-05,
      "model_forward_time": 1.6824307441711426,
      "step": 47470
    },
    {
      "dynamics_loss_avg": 0.040751477144658564,
      "epoch": 4.5270919799804686e-05,
      "step": 47470
    },
    {
      "action_loss_avg": 0.0012056784529704601,
      "epoch": 4.5270919799804686e-05,
      "step": 47470
    },
    {
      "epoch": 4.5270919799804686e-05,
      "step": 47470,
      "training_step_time": 8.745126008987427
    },
    {
      "epoch": 4.527187347412109e-05,
      "model_forward_time": 1.6989994049072266,
      "step": 47471
    },
    {
      "epoch": 4.527187347412109e-05,
      "step": 47471,
      "training_step_time": 7.877416610717773
    },
    {
      "epoch": 4.52728271484375e-05,
      "model_forward_time": 1.6875474452972412,
      "step": 47472
    },
    {
      "epoch": 4.52728271484375e-05,
      "step": 47472,
      "training_step_time": 8.760640621185303
    },
    {
      "epoch": 4.5273780822753905e-05,
      "model_forward_time": 1.6844186782836914,
      "step": 47473
    },
    {
      "epoch": 4.5273780822753905e-05,
      "step": 47473,
      "training_step_time": 7.900699138641357
    },
    {
      "epoch": 4.527473449707031e-05,
      "model_forward_time": 1.6884565353393555,
      "step": 47474
    },
    {
      "epoch": 4.527473449707031e-05,
      "step": 47474,
      "training_step_time": 8.75188398361206
    },
    {
      "epoch": 4.527568817138672e-05,
      "model_forward_time": 1.6822223663330078,
      "step": 47475
    },
    {
      "epoch": 4.527568817138672e-05,
      "step": 47475,
      "training_step_time": 7.95257306098938
    },
    {
      "epoch": 4.5276641845703124e-05,
      "model_forward_time": 2.2311794757843018,
      "step": 47476
    },
    {
      "epoch": 4.5276641845703124e-05,
      "step": 47476,
      "training_step_time": 7.837597131729126
    },
    {
      "epoch": 4.527759552001953e-05,
      "model_forward_time": 1.6814067363739014,
      "step": 47477
    },
    {
      "epoch": 4.527759552001953e-05,
      "step": 47477,
      "training_step_time": 7.973325729370117
    },
    {
      "epoch": 4.527854919433594e-05,
      "model_forward_time": 1.6952240467071533,
      "step": 47478
    },
    {
      "epoch": 4.527854919433594e-05,
      "step": 47478,
      "training_step_time": 8.723275899887085
    },
    {
      "epoch": 4.527950286865234e-05,
      "model_forward_time": 1.6794698238372803,
      "step": 47479
    },
    {
      "epoch": 4.527950286865234e-05,
      "step": 47479,
      "training_step_time": 7.912353277206421
    },
    {
      "epoch": 4.528045654296875e-05,
      "grad_norm": 0.2819436490535736,
      "learning_rate": 5.82639768183386e-06,
      "loss": 0.0585,
      "step": 47480
    },
    {
      "epoch": 4.528045654296875e-05,
      "model_forward_time": 1.685354232788086,
      "step": 47480
    },
    {
      "dynamics_loss_avg": 0.056885307654738426,
      "epoch": 4.528045654296875e-05,
      "step": 47480
    },
    {
      "action_loss_avg": 0.0022679699468426406,
      "epoch": 4.528045654296875e-05,
      "step": 47480
    },
    {
      "epoch": 4.528045654296875e-05,
      "step": 47480,
      "training_step_time": 7.945495367050171
    },
    {
      "epoch": 4.528141021728516e-05,
      "model_forward_time": 1.6850311756134033,
      "step": 47481
    },
    {
      "epoch": 4.528141021728516e-05,
      "step": 47481,
      "training_step_time": 7.689110517501831
    },
    {
      "epoch": 4.528236389160156e-05,
      "model_forward_time": 1.6800503730773926,
      "step": 47482
    },
    {
      "epoch": 4.528236389160156e-05,
      "step": 47482,
      "training_step_time": 7.924495697021484
    },
    {
      "epoch": 4.5283317565917966e-05,
      "model_forward_time": 1.3036043643951416,
      "step": 47483
    },
    {
      "epoch": 4.5283317565917966e-05,
      "step": 47483,
      "training_step_time": 7.932875156402588
    },
    {
      "epoch": 4.528427124023438e-05,
      "model_forward_time": 1.6878721714019775,
      "step": 47484
    },
    {
      "epoch": 4.528427124023438e-05,
      "step": 47484,
      "training_step_time": 7.898337364196777
    },
    {
      "epoch": 4.528522491455078e-05,
      "model_forward_time": 2.2348413467407227,
      "step": 47485
    },
    {
      "epoch": 4.528522491455078e-05,
      "step": 47485,
      "training_step_time": 7.965402126312256
    },
    {
      "epoch": 4.5286178588867185e-05,
      "model_forward_time": 1.6837143898010254,
      "step": 47486
    },
    {
      "epoch": 4.5286178588867185e-05,
      "step": 47486,
      "training_step_time": 8.674270153045654
    },
    {
      "epoch": 4.5287132263183596e-05,
      "model_forward_time": 1.6863842010498047,
      "step": 47487
    },
    {
      "epoch": 4.5287132263183596e-05,
      "step": 47487,
      "training_step_time": 8.393387794494629
    },
    {
      "epoch": 4.52880859375e-05,
      "model_forward_time": 1.6820061206817627,
      "step": 47488
    },
    {
      "epoch": 4.52880859375e-05,
      "step": 47488,
      "training_step_time": 8.721847295761108
    },
    {
      "epoch": 4.5289039611816404e-05,
      "model_forward_time": 1.6757314205169678,
      "step": 47489
    },
    {
      "epoch": 4.5289039611816404e-05,
      "step": 47489,
      "training_step_time": 7.906898260116577
    },
    {
      "epoch": 4.5289993286132814e-05,
      "grad_norm": 0.12864847481250763,
      "learning_rate": 5.824766907423612e-06,
      "loss": 0.0638,
      "step": 47490
    },
    {
      "epoch": 4.5289993286132814e-05,
      "model_forward_time": 1.707139492034912,
      "step": 47490
    },
    {
      "dynamics_loss_avg": 0.06207383135333657,
      "epoch": 4.5289993286132814e-05,
      "step": 47490
    },
    {
      "action_loss_avg": 0.0016126428468851372,
      "epoch": 4.5289993286132814e-05,
      "step": 47490
    },
    {
      "epoch": 4.5289993286132814e-05,
      "step": 47490,
      "training_step_time": 7.904655933380127
    },
    {
      "epoch": 4.529094696044922e-05,
      "model_forward_time": 1.6894264221191406,
      "step": 47491
    },
    {
      "epoch": 4.529094696044922e-05,
      "step": 47491,
      "training_step_time": 7.957299709320068
    },
    {
      "epoch": 4.529190063476562e-05,
      "model_forward_time": 1.6775989532470703,
      "step": 47492
    },
    {
      "epoch": 4.529190063476562e-05,
      "step": 47492,
      "training_step_time": 8.055236577987671
    },
    {
      "epoch": 4.529285430908203e-05,
      "model_forward_time": 1.6811797618865967,
      "step": 47493
    },
    {
      "epoch": 4.529285430908203e-05,
      "step": 47493,
      "training_step_time": 7.815233469009399
    },
    {
      "epoch": 4.529380798339844e-05,
      "model_forward_time": 2.242478132247925,
      "step": 47494
    },
    {
      "epoch": 4.529380798339844e-05,
      "step": 47494,
      "training_step_time": 9.370601892471313
    },
    {
      "epoch": 4.529476165771484e-05,
      "model_forward_time": 1.6868538856506348,
      "step": 47495
    },
    {
      "epoch": 4.529476165771484e-05,
      "step": 47495,
      "training_step_time": 8.654107809066772
    },
    {
      "epoch": 4.529571533203125e-05,
      "model_forward_time": 1.6826376914978027,
      "step": 47496
    },
    {
      "epoch": 4.529571533203125e-05,
      "step": 47496,
      "training_step_time": 8.405987739562988
    },
    {
      "epoch": 4.5296669006347656e-05,
      "model_forward_time": 1.6997289657592773,
      "step": 47497
    },
    {
      "epoch": 4.5296669006347656e-05,
      "step": 47497,
      "training_step_time": 8.726796627044678
    },
    {
      "epoch": 4.529762268066406e-05,
      "model_forward_time": 1.6841306686401367,
      "step": 47498
    },
    {
      "epoch": 4.529762268066406e-05,
      "step": 47498,
      "training_step_time": 9.151551961898804
    },
    {
      "epoch": 4.529857635498047e-05,
      "model_forward_time": 1.6973340511322021,
      "step": 47499
    },
    {
      "epoch": 4.529857635498047e-05,
      "step": 47499,
      "training_step_time": 7.976756811141968
    },
    {
      "epoch": 4.5299530029296875e-05,
      "grad_norm": 0.1091674342751503,
      "learning_rate": 5.823136042818095e-06,
      "loss": 0.061,
      "step": 47500
    },
    {
      "epoch": 4.5299530029296875e-05,
      "model_forward_time": 1.6744999885559082,
      "step": 47500
    },
    {
      "dynamics_loss_avg": 0.04222886860370636,
      "epoch": 4.5299530029296875e-05,
      "step": 47500
    },
    {
      "action_loss_avg": 0.0015874931021244265,
      "epoch": 4.5299530029296875e-05,
      "step": 47500
    },
    {
      "epoch": 4.5299530029296875e-05,
      "step": 47500,
      "training_step_time": 8.37954068183899
    },
    {
      "epoch": 4.530048370361328e-05,
      "model_forward_time": 1.6879799365997314,
      "step": 47501
    },
    {
      "epoch": 4.530048370361328e-05,
      "step": 47501,
      "training_step_time": 7.982736587524414
    },
    {
      "epoch": 4.530143737792969e-05,
      "model_forward_time": 1.6932926177978516,
      "step": 47502
    },
    {
      "epoch": 4.530143737792969e-05,
      "step": 47502,
      "training_step_time": 7.4549150466918945
    },
    {
      "epoch": 4.5302391052246094e-05,
      "model_forward_time": 2.2522239685058594,
      "step": 47503
    },
    {
      "epoch": 4.5302391052246094e-05,
      "step": 47503,
      "training_step_time": 7.9603822231292725
    },
    {
      "epoch": 4.53033447265625e-05,
      "model_forward_time": 1.6818137168884277,
      "step": 47504
    },
    {
      "epoch": 4.53033447265625e-05,
      "step": 47504,
      "training_step_time": 8.526216745376587
    },
    {
      "epoch": 4.530429840087891e-05,
      "model_forward_time": 1.693298578262329,
      "step": 47505
    },
    {
      "epoch": 4.530429840087891e-05,
      "step": 47505,
      "training_step_time": 7.8575119972229
    },
    {
      "epoch": 4.530525207519531e-05,
      "model_forward_time": 1.708392858505249,
      "step": 47506
    },
    {
      "epoch": 4.530525207519531e-05,
      "step": 47506,
      "training_step_time": 8.85044002532959
    },
    {
      "epoch": 4.530620574951172e-05,
      "model_forward_time": 1.6947901248931885,
      "step": 47507
    },
    {
      "epoch": 4.530620574951172e-05,
      "step": 47507,
      "training_step_time": 7.894415378570557
    },
    {
      "epoch": 4.530715942382813e-05,
      "model_forward_time": 1.6894402503967285,
      "step": 47508
    },
    {
      "epoch": 4.530715942382813e-05,
      "step": 47508,
      "training_step_time": 9.59460186958313
    },
    {
      "epoch": 4.530811309814453e-05,
      "model_forward_time": 1.6975319385528564,
      "step": 47509
    },
    {
      "epoch": 4.530811309814453e-05,
      "step": 47509,
      "training_step_time": 8.569036483764648
    },
    {
      "epoch": 4.5309066772460936e-05,
      "grad_norm": 0.14430096745491028,
      "learning_rate": 5.821505088195658e-06,
      "loss": 0.062,
      "step": 47510
    },
    {
      "epoch": 4.5309066772460936e-05,
      "model_forward_time": 1.687007188796997,
      "step": 47510
    },
    {
      "dynamics_loss_avg": 0.07090447396039963,
      "epoch": 4.5309066772460936e-05,
      "step": 47510
    },
    {
      "action_loss_avg": 0.0015755616768728942,
      "epoch": 4.5309066772460936e-05,
      "step": 47510
    },
    {
      "epoch": 4.5309066772460936e-05,
      "step": 47510,
      "training_step_time": 7.758657932281494
    },
    {
      "epoch": 4.5310020446777346e-05,
      "model_forward_time": 1.6956238746643066,
      "step": 47511
    },
    {
      "epoch": 4.5310020446777346e-05,
      "step": 47511,
      "training_step_time": 7.57830023765564
    },
    {
      "epoch": 4.531097412109375e-05,
      "model_forward_time": 2.2328078746795654,
      "step": 47512
    },
    {
      "epoch": 4.531097412109375e-05,
      "step": 47512,
      "training_step_time": 7.985403776168823
    },
    {
      "epoch": 4.5311927795410154e-05,
      "model_forward_time": 1.695685863494873,
      "step": 47513
    },
    {
      "epoch": 4.5311927795410154e-05,
      "step": 47513,
      "training_step_time": 9.67499589920044
    },
    {
      "epoch": 4.5312881469726565e-05,
      "model_forward_time": 1.6938297748565674,
      "step": 47514
    },
    {
      "epoch": 4.5312881469726565e-05,
      "step": 47514,
      "training_step_time": 7.958749055862427
    },
    {
      "epoch": 4.531383514404297e-05,
      "model_forward_time": 1.7178223133087158,
      "step": 47515
    },
    {
      "epoch": 4.531383514404297e-05,
      "step": 47515,
      "training_step_time": 7.866738796234131
    },
    {
      "epoch": 4.531478881835937e-05,
      "model_forward_time": 1.7017889022827148,
      "step": 47516
    },
    {
      "epoch": 4.531478881835937e-05,
      "step": 47516,
      "training_step_time": 7.912506103515625
    },
    {
      "epoch": 4.5315742492675784e-05,
      "model_forward_time": 1.700143575668335,
      "step": 47517
    },
    {
      "epoch": 4.5315742492675784e-05,
      "step": 47517,
      "training_step_time": 7.948654651641846
    },
    {
      "epoch": 4.531669616699219e-05,
      "model_forward_time": 1.6858775615692139,
      "step": 47518
    },
    {
      "epoch": 4.531669616699219e-05,
      "step": 47518,
      "training_step_time": 7.799404621124268
    },
    {
      "epoch": 4.531764984130859e-05,
      "model_forward_time": 1.6985211372375488,
      "step": 47519
    },
    {
      "epoch": 4.531764984130859e-05,
      "step": 47519,
      "training_step_time": 8.004190921783447
    },
    {
      "epoch": 4.5318603515625e-05,
      "grad_norm": 0.13829673826694489,
      "learning_rate": 5.819874043734661e-06,
      "loss": 0.0591,
      "step": 47520
    },
    {
      "epoch": 4.5318603515625e-05,
      "model_forward_time": 1.685074806213379,
      "step": 47520
    },
    {
      "dynamics_loss_avg": 0.06379737947136163,
      "epoch": 4.5318603515625e-05,
      "step": 47520
    },
    {
      "action_loss_avg": 0.0013093682195176371,
      "epoch": 4.5318603515625e-05,
      "step": 47520
    },
    {
      "epoch": 4.5318603515625e-05,
      "step": 47520,
      "training_step_time": 7.803825378417969
    },
    {
      "epoch": 4.531955718994141e-05,
      "model_forward_time": 1.6832551956176758,
      "step": 47521
    },
    {
      "epoch": 4.531955718994141e-05,
      "step": 47521,
      "training_step_time": 7.915415287017822
    },
    {
      "epoch": 4.532051086425781e-05,
      "model_forward_time": 2.2396817207336426,
      "step": 47522
    },
    {
      "epoch": 4.532051086425781e-05,
      "step": 47522,
      "training_step_time": 8.762577772140503
    },
    {
      "epoch": 4.532146453857422e-05,
      "model_forward_time": 1.6931543350219727,
      "step": 47523
    },
    {
      "epoch": 4.532146453857422e-05,
      "step": 47523,
      "training_step_time": 9.495571851730347
    },
    {
      "epoch": 4.5322418212890626e-05,
      "model_forward_time": 1.7048919200897217,
      "step": 47524
    },
    {
      "epoch": 4.5322418212890626e-05,
      "step": 47524,
      "training_step_time": 8.76846718788147
    },
    {
      "epoch": 4.532337188720703e-05,
      "model_forward_time": 1.700927495956421,
      "step": 47525
    },
    {
      "epoch": 4.532337188720703e-05,
      "step": 47525,
      "training_step_time": 7.798001527786255
    },
    {
      "epoch": 4.532432556152344e-05,
      "model_forward_time": 1.7005910873413086,
      "step": 47526
    },
    {
      "epoch": 4.532432556152344e-05,
      "step": 47526,
      "training_step_time": 8.580546617507935
    },
    {
      "epoch": 4.5325279235839845e-05,
      "model_forward_time": 1.6889111995697021,
      "step": 47527
    },
    {
      "epoch": 4.5325279235839845e-05,
      "step": 47527,
      "training_step_time": 8.011370658874512
    },
    {
      "epoch": 4.532623291015625e-05,
      "model_forward_time": 1.6964960098266602,
      "step": 47528
    },
    {
      "epoch": 4.532623291015625e-05,
      "step": 47528,
      "training_step_time": 7.8859803676605225
    },
    {
      "epoch": 4.532718658447266e-05,
      "model_forward_time": 1.3114819526672363,
      "step": 47529
    },
    {
      "epoch": 4.532718658447266e-05,
      "step": 47529,
      "training_step_time": 7.827721118927002
    },
    {
      "epoch": 4.5328140258789064e-05,
      "grad_norm": 0.1467943638563156,
      "learning_rate": 5.81824290961347e-06,
      "loss": 0.0606,
      "step": 47530
    },
    {
      "epoch": 4.5328140258789064e-05,
      "model_forward_time": 1.6945641040802002,
      "step": 47530
    },
    {
      "dynamics_loss_avg": 0.06511954106390476,
      "epoch": 4.5328140258789064e-05,
      "step": 47530
    },
    {
      "action_loss_avg": 0.0013913323229644448,
      "epoch": 4.5328140258789064e-05,
      "step": 47530
    },
    {
      "epoch": 4.5328140258789064e-05,
      "step": 47530,
      "training_step_time": 7.743335247039795
    },
    {
      "epoch": 4.532909393310547e-05,
      "model_forward_time": 2.2478222846984863,
      "step": 47531
    },
    {
      "epoch": 4.532909393310547e-05,
      "step": 47531,
      "training_step_time": 7.805109739303589
    },
    {
      "epoch": 4.533004760742187e-05,
      "model_forward_time": 1.6926696300506592,
      "step": 47532
    },
    {
      "epoch": 4.533004760742187e-05,
      "step": 47532,
      "training_step_time": 7.716163635253906
    },
    {
      "epoch": 4.533100128173828e-05,
      "model_forward_time": 1.7027587890625,
      "step": 47533
    },
    {
      "epoch": 4.533100128173828e-05,
      "step": 47533,
      "training_step_time": 8.515645980834961
    },
    {
      "epoch": 4.5331954956054686e-05,
      "model_forward_time": 1.6911184787750244,
      "step": 47534
    },
    {
      "epoch": 4.5331954956054686e-05,
      "step": 47534,
      "training_step_time": 9.490127563476562
    },
    {
      "epoch": 4.533290863037109e-05,
      "model_forward_time": 1.6919641494750977,
      "step": 47535
    },
    {
      "epoch": 4.533290863037109e-05,
      "step": 47535,
      "training_step_time": 8.542426586151123
    },
    {
      "epoch": 4.53338623046875e-05,
      "model_forward_time": 1.7204539775848389,
      "step": 47536
    },
    {
      "epoch": 4.53338623046875e-05,
      "step": 47536,
      "training_step_time": 8.035331726074219
    },
    {
      "epoch": 4.5334815979003905e-05,
      "model_forward_time": 1.6827375888824463,
      "step": 47537
    },
    {
      "epoch": 4.5334815979003905e-05,
      "step": 47537,
      "training_step_time": 7.836206912994385
    },
    {
      "epoch": 4.533576965332031e-05,
      "model_forward_time": 1.683995008468628,
      "step": 47538
    },
    {
      "epoch": 4.533576965332031e-05,
      "step": 47538,
      "training_step_time": 7.897717475891113
    },
    {
      "epoch": 4.533672332763672e-05,
      "model_forward_time": 1.686084508895874,
      "step": 47539
    },
    {
      "epoch": 4.533672332763672e-05,
      "step": 47539,
      "training_step_time": 7.80053973197937
    },
    {
      "epoch": 4.5337677001953124e-05,
      "grad_norm": 0.16752469539642334,
      "learning_rate": 5.816611686010465e-06,
      "loss": 0.0574,
      "step": 47540
    },
    {
      "epoch": 4.5337677001953124e-05,
      "model_forward_time": 2.2260453701019287,
      "step": 47540
    },
    {
      "dynamics_loss_avg": 0.0622507194057107,
      "epoch": 4.5337677001953124e-05,
      "step": 47540
    },
    {
      "action_loss_avg": 0.0018951206817291676,
      "epoch": 4.5337677001953124e-05,
      "step": 47540
    },
    {
      "epoch": 4.5337677001953124e-05,
      "step": 47540,
      "training_step_time": 8.720263242721558
    },
    {
      "epoch": 4.533863067626953e-05,
      "model_forward_time": 1.6891779899597168,
      "step": 47541
    },
    {
      "epoch": 4.533863067626953e-05,
      "step": 47541,
      "training_step_time": 8.794138669967651
    },
    {
      "epoch": 4.533958435058594e-05,
      "model_forward_time": 1.700922966003418,
      "step": 47542
    },
    {
      "epoch": 4.533958435058594e-05,
      "step": 47542,
      "training_step_time": 7.7702672481536865
    },
    {
      "epoch": 4.534053802490234e-05,
      "model_forward_time": 1.6879935264587402,
      "step": 47543
    },
    {
      "epoch": 4.534053802490234e-05,
      "step": 47543,
      "training_step_time": 7.796131372451782
    },
    {
      "epoch": 4.534149169921875e-05,
      "model_forward_time": 1.6862587928771973,
      "step": 47544
    },
    {
      "epoch": 4.534149169921875e-05,
      "step": 47544,
      "training_step_time": 8.21182370185852
    },
    {
      "epoch": 4.534244537353516e-05,
      "model_forward_time": 1.7016181945800781,
      "step": 47545
    },
    {
      "epoch": 4.534244537353516e-05,
      "step": 47545,
      "training_step_time": 8.749390602111816
    },
    {
      "epoch": 4.534339904785156e-05,
      "model_forward_time": 1.686474084854126,
      "step": 47546
    },
    {
      "epoch": 4.534339904785156e-05,
      "step": 47546,
      "training_step_time": 7.789003372192383
    },
    {
      "epoch": 4.5344352722167966e-05,
      "model_forward_time": 1.6900134086608887,
      "step": 47547
    },
    {
      "epoch": 4.5344352722167966e-05,
      "step": 47547,
      "training_step_time": 7.812325477600098
    },
    {
      "epoch": 4.534530639648438e-05,
      "model_forward_time": 1.6829791069030762,
      "step": 47548
    },
    {
      "epoch": 4.534530639648438e-05,
      "step": 47548,
      "training_step_time": 7.702619791030884
    },
    {
      "epoch": 4.534626007080078e-05,
      "model_forward_time": 2.2396721839904785,
      "step": 47549
    },
    {
      "epoch": 4.534626007080078e-05,
      "step": 47549,
      "training_step_time": 8.719359159469604
    },
    {
      "epoch": 4.5347213745117185e-05,
      "grad_norm": 0.14459533989429474,
      "learning_rate": 5.8149803731040335e-06,
      "loss": 0.0584,
      "step": 47550
    },
    {
      "epoch": 4.5347213745117185e-05,
      "model_forward_time": 1.6906828880310059,
      "step": 47550
    },
    {
      "dynamics_loss_avg": 0.05502675287425518,
      "epoch": 4.5347213745117185e-05,
      "step": 47550
    },
    {
      "action_loss_avg": 0.0013566387933678925,
      "epoch": 4.5347213745117185e-05,
      "step": 47550
    },
    {
      "epoch": 4.5347213745117185e-05,
      "step": 47550,
      "training_step_time": 7.804819822311401
    },
    {
      "epoch": 4.5348167419433596e-05,
      "model_forward_time": 1.7095091342926025,
      "step": 47551
    },
    {
      "epoch": 4.5348167419433596e-05,
      "step": 47551,
      "training_step_time": 8.393036842346191
    },
    {
      "epoch": 4.534912109375e-05,
      "model_forward_time": 1.69657301902771,
      "step": 47552
    },
    {
      "epoch": 4.534912109375e-05,
      "step": 47552,
      "training_step_time": 7.779422998428345
    },
    {
      "epoch": 4.5350074768066404e-05,
      "model_forward_time": 1.692803144454956,
      "step": 47553
    },
    {
      "epoch": 4.5350074768066404e-05,
      "step": 47553,
      "training_step_time": 8.793594360351562
    },
    {
      "epoch": 4.5351028442382814e-05,
      "model_forward_time": 1.712460994720459,
      "step": 47554
    },
    {
      "epoch": 4.5351028442382814e-05,
      "step": 47554,
      "training_step_time": 8.507565021514893
    },
    {
      "epoch": 4.535198211669922e-05,
      "model_forward_time": 1.6956357955932617,
      "step": 47555
    },
    {
      "epoch": 4.535198211669922e-05,
      "step": 47555,
      "training_step_time": 8.431634902954102
    },
    {
      "epoch": 4.535293579101562e-05,
      "model_forward_time": 1.690779209136963,
      "step": 47556
    },
    {
      "epoch": 4.535293579101562e-05,
      "step": 47556,
      "training_step_time": 7.5926008224487305
    },
    {
      "epoch": 4.535388946533203e-05,
      "model_forward_time": 1.682840347290039,
      "step": 47557
    },
    {
      "epoch": 4.535388946533203e-05,
      "step": 47557,
      "training_step_time": 7.480557680130005
    },
    {
      "epoch": 4.535484313964844e-05,
      "model_forward_time": 2.245293378829956,
      "step": 47558
    },
    {
      "epoch": 4.535484313964844e-05,
      "step": 47558,
      "training_step_time": 8.496517658233643
    },
    {
      "epoch": 4.535579681396484e-05,
      "model_forward_time": 1.6934866905212402,
      "step": 47559
    },
    {
      "epoch": 4.535579681396484e-05,
      "step": 47559,
      "training_step_time": 8.742969751358032
    },
    {
      "epoch": 4.535675048828125e-05,
      "grad_norm": 0.1500117927789688,
      "learning_rate": 5.813348971072572e-06,
      "loss": 0.0547,
      "step": 47560
    },
    {
      "epoch": 4.535675048828125e-05,
      "model_forward_time": 1.7067441940307617,
      "step": 47560
    },
    {
      "dynamics_loss_avg": 0.04988380596041679,
      "epoch": 4.535675048828125e-05,
      "step": 47560
    },
    {
      "action_loss_avg": 0.002010709891328588,
      "epoch": 4.535675048828125e-05,
      "step": 47560
    },
    {
      "epoch": 4.535675048828125e-05,
      "step": 47560,
      "training_step_time": 7.847951173782349
    },
    {
      "epoch": 4.5357704162597656e-05,
      "model_forward_time": 1.679793119430542,
      "step": 47561
    },
    {
      "epoch": 4.5357704162597656e-05,
      "step": 47561,
      "training_step_time": 7.842751979827881
    },
    {
      "epoch": 4.535865783691406e-05,
      "model_forward_time": 1.6980273723602295,
      "step": 47562
    },
    {
      "epoch": 4.535865783691406e-05,
      "step": 47562,
      "training_step_time": 9.314931392669678
    },
    {
      "epoch": 4.535961151123047e-05,
      "model_forward_time": 1.706974983215332,
      "step": 47563
    },
    {
      "epoch": 4.535961151123047e-05,
      "step": 47563,
      "training_step_time": 7.863878965377808
    },
    {
      "epoch": 4.5360565185546875e-05,
      "model_forward_time": 1.6968812942504883,
      "step": 47564
    },
    {
      "epoch": 4.5360565185546875e-05,
      "step": 47564,
      "training_step_time": 8.20892858505249
    },
    {
      "epoch": 4.536151885986328e-05,
      "model_forward_time": 1.6881358623504639,
      "step": 47565
    },
    {
      "epoch": 4.536151885986328e-05,
      "step": 47565,
      "training_step_time": 7.926746129989624
    },
    {
      "epoch": 4.536247253417969e-05,
      "model_forward_time": 1.6950428485870361,
      "step": 47566
    },
    {
      "epoch": 4.536247253417969e-05,
      "step": 47566,
      "training_step_time": 7.701178073883057
    },
    {
      "epoch": 4.5363426208496094e-05,
      "model_forward_time": 1.6967291831970215,
      "step": 47567
    },
    {
      "epoch": 4.5363426208496094e-05,
      "step": 47567,
      "training_step_time": 8.276803255081177
    },
    {
      "epoch": 4.53643798828125e-05,
      "model_forward_time": 1.6969280242919922,
      "step": 47568
    },
    {
      "epoch": 4.53643798828125e-05,
      "step": 47568,
      "training_step_time": 8.846550703048706
    },
    {
      "epoch": 4.536533355712891e-05,
      "model_forward_time": 1.7071638107299805,
      "step": 47569
    },
    {
      "epoch": 4.536533355712891e-05,
      "step": 47569,
      "training_step_time": 8.410768270492554
    },
    {
      "epoch": 4.536628723144531e-05,
      "grad_norm": 0.20050360262393951,
      "learning_rate": 5.811717480094492e-06,
      "loss": 0.0546,
      "step": 47570
    },
    {
      "epoch": 4.536628723144531e-05,
      "model_forward_time": 1.6900324821472168,
      "step": 47570
    },
    {
      "dynamics_loss_avg": 0.04122548643499613,
      "epoch": 4.536628723144531e-05,
      "step": 47570
    },
    {
      "action_loss_avg": 0.0012094960489775985,
      "epoch": 4.536628723144531e-05,
      "step": 47570
    },
    {
      "epoch": 4.536628723144531e-05,
      "step": 47570,
      "training_step_time": 8.52018690109253
    },
    {
      "epoch": 4.536724090576172e-05,
      "model_forward_time": 1.6929152011871338,
      "step": 47571
    },
    {
      "epoch": 4.536724090576172e-05,
      "step": 47571,
      "training_step_time": 8.657311201095581
    },
    {
      "epoch": 4.536819458007813e-05,
      "model_forward_time": 1.6931777000427246,
      "step": 47572
    },
    {
      "epoch": 4.536819458007813e-05,
      "step": 47572,
      "training_step_time": 7.645439863204956
    },
    {
      "epoch": 4.536914825439453e-05,
      "model_forward_time": 1.6960170269012451,
      "step": 47573
    },
    {
      "epoch": 4.536914825439453e-05,
      "step": 47573,
      "training_step_time": 7.855408191680908
    },
    {
      "epoch": 4.5370101928710936e-05,
      "model_forward_time": 1.6951262950897217,
      "step": 47574
    },
    {
      "epoch": 4.5370101928710936e-05,
      "step": 47574,
      "training_step_time": 8.148427963256836
    },
    {
      "epoch": 4.5371055603027346e-05,
      "model_forward_time": 1.7080731391906738,
      "step": 47575
    },
    {
      "epoch": 4.5371055603027346e-05,
      "step": 47575,
      "training_step_time": 7.632593393325806
    },
    {
      "epoch": 4.537200927734375e-05,
      "model_forward_time": 1.691636085510254,
      "step": 47576
    },
    {
      "epoch": 4.537200927734375e-05,
      "step": 47576,
      "training_step_time": 8.187544345855713
    },
    {
      "epoch": 4.5372962951660155e-05,
      "model_forward_time": 2.232940912246704,
      "step": 47577
    },
    {
      "epoch": 4.5372962951660155e-05,
      "step": 47577,
      "training_step_time": 8.22339415550232
    },
    {
      "epoch": 4.5373916625976565e-05,
      "model_forward_time": 1.7035326957702637,
      "step": 47578
    },
    {
      "epoch": 4.5373916625976565e-05,
      "step": 47578,
      "training_step_time": 7.543696165084839
    },
    {
      "epoch": 4.537487030029297e-05,
      "model_forward_time": 1.6899468898773193,
      "step": 47579
    },
    {
      "epoch": 4.537487030029297e-05,
      "step": 47579,
      "training_step_time": 7.672526836395264
    },
    {
      "epoch": 4.537582397460937e-05,
      "grad_norm": 0.09002931416034698,
      "learning_rate": 5.810085900348209e-06,
      "loss": 0.0589,
      "step": 47580
    },
    {
      "epoch": 4.537582397460937e-05,
      "model_forward_time": 1.6947526931762695,
      "step": 47580
    },
    {
      "dynamics_loss_avg": 0.07463266272097827,
      "epoch": 4.537582397460937e-05,
      "step": 47580
    },
    {
      "action_loss_avg": 0.0033078983426094053,
      "epoch": 4.537582397460937e-05,
      "step": 47580
    },
    {
      "epoch": 4.537582397460937e-05,
      "step": 47580,
      "training_step_time": 7.603234529495239
    },
    {
      "epoch": 4.5376777648925784e-05,
      "model_forward_time": 1.7116737365722656,
      "step": 47581
    },
    {
      "epoch": 4.5376777648925784e-05,
      "step": 47581,
      "training_step_time": 9.199490070343018
    },
    {
      "epoch": 4.537773132324219e-05,
      "model_forward_time": 1.709714412689209,
      "step": 47582
    },
    {
      "epoch": 4.537773132324219e-05,
      "step": 47582,
      "training_step_time": 7.751087427139282
    },
    {
      "epoch": 4.537868499755859e-05,
      "model_forward_time": 1.6960394382476807,
      "step": 47583
    },
    {
      "epoch": 4.537868499755859e-05,
      "step": 47583,
      "training_step_time": 7.859055280685425
    },
    {
      "epoch": 4.5379638671875e-05,
      "model_forward_time": 1.6955955028533936,
      "step": 47584
    },
    {
      "epoch": 4.5379638671875e-05,
      "step": 47584,
      "training_step_time": 7.673938751220703
    },
    {
      "epoch": 4.538059234619141e-05,
      "model_forward_time": 1.7017335891723633,
      "step": 47585
    },
    {
      "epoch": 4.538059234619141e-05,
      "step": 47585,
      "training_step_time": 8.067621231079102
    },
    {
      "epoch": 4.538154602050781e-05,
      "model_forward_time": 2.253234624862671,
      "step": 47586
    },
    {
      "epoch": 4.538154602050781e-05,
      "step": 47586,
      "training_step_time": 7.846232175827026
    },
    {
      "epoch": 4.538249969482422e-05,
      "model_forward_time": 1.6956298351287842,
      "step": 47587
    },
    {
      "epoch": 4.538249969482422e-05,
      "step": 47587,
      "training_step_time": 8.35574460029602
    },
    {
      "epoch": 4.5383453369140626e-05,
      "model_forward_time": 1.71775484085083,
      "step": 47588
    },
    {
      "epoch": 4.5383453369140626e-05,
      "step": 47588,
      "training_step_time": 8.67650032043457
    },
    {
      "epoch": 4.538440704345703e-05,
      "model_forward_time": 1.6967196464538574,
      "step": 47589
    },
    {
      "epoch": 4.538440704345703e-05,
      "step": 47589,
      "training_step_time": 7.762868642807007
    },
    {
      "epoch": 4.538536071777344e-05,
      "grad_norm": 0.09539996832609177,
      "learning_rate": 5.8084542320121485e-06,
      "loss": 0.0573,
      "step": 47590
    },
    {
      "epoch": 4.538536071777344e-05,
      "model_forward_time": 1.6894340515136719,
      "step": 47590
    },
    {
      "dynamics_loss_avg": 0.055188626796007154,
      "epoch": 4.538536071777344e-05,
      "step": 47590
    },
    {
      "action_loss_avg": 0.0011700060742441565,
      "epoch": 4.538536071777344e-05,
      "step": 47590
    },
    {
      "epoch": 4.538536071777344e-05,
      "step": 47590,
      "training_step_time": 9.515333890914917
    },
    {
      "epoch": 4.5386314392089845e-05,
      "model_forward_time": 1.7218408584594727,
      "step": 47591
    },
    {
      "epoch": 4.5386314392089845e-05,
      "step": 47591,
      "training_step_time": 7.77293848991394
    },
    {
      "epoch": 4.538726806640625e-05,
      "model_forward_time": 1.6957998275756836,
      "step": 47592
    },
    {
      "epoch": 4.538726806640625e-05,
      "step": 47592,
      "training_step_time": 7.956147909164429
    },
    {
      "epoch": 4.538822174072266e-05,
      "model_forward_time": 1.703629493713379,
      "step": 47593
    },
    {
      "epoch": 4.538822174072266e-05,
      "step": 47593,
      "training_step_time": 7.853366374969482
    },
    {
      "epoch": 4.5389175415039064e-05,
      "model_forward_time": 1.6978626251220703,
      "step": 47594
    },
    {
      "epoch": 4.5389175415039064e-05,
      "step": 47594,
      "training_step_time": 7.96379828453064
    },
    {
      "epoch": 4.539012908935547e-05,
      "model_forward_time": 2.249321699142456,
      "step": 47595
    },
    {
      "epoch": 4.539012908935547e-05,
      "step": 47595,
      "training_step_time": 7.940893888473511
    },
    {
      "epoch": 4.539108276367187e-05,
      "model_forward_time": 1.709620475769043,
      "step": 47596
    },
    {
      "epoch": 4.539108276367187e-05,
      "step": 47596,
      "training_step_time": 8.505822658538818
    },
    {
      "epoch": 4.539203643798828e-05,
      "model_forward_time": 1.7150015830993652,
      "step": 47597
    },
    {
      "epoch": 4.539203643798828e-05,
      "step": 47597,
      "training_step_time": 9.552374839782715
    },
    {
      "epoch": 4.5392990112304687e-05,
      "model_forward_time": 1.7010412216186523,
      "step": 47598
    },
    {
      "epoch": 4.5392990112304687e-05,
      "step": 47598,
      "training_step_time": 8.842320680618286
    },
    {
      "epoch": 4.539394378662109e-05,
      "model_forward_time": 1.6962370872497559,
      "step": 47599
    },
    {
      "epoch": 4.539394378662109e-05,
      "step": 47599,
      "training_step_time": 7.875383615493774
    },
    {
      "epoch": 4.53948974609375e-05,
      "grad_norm": 0.1689501255750656,
      "learning_rate": 5.80682247526475e-06,
      "loss": 0.0577,
      "step": 47600
    },
    {
      "epoch": 4.53948974609375e-05,
      "model_forward_time": 1.7106690406799316,
      "step": 47600
    },
    {
      "dynamics_loss_avg": 0.07366010975092649,
      "epoch": 4.53948974609375e-05,
      "step": 47600
    },
    {
      "action_loss_avg": 0.001872633642051369,
      "epoch": 4.53948974609375e-05,
      "step": 47600
    },
    {
      "epoch": 4.53948974609375e-05,
      "step": 47600,
      "training_step_time": 7.775698900222778
    },
    {
      "epoch": 4.5395851135253905e-05,
      "model_forward_time": 1.6953966617584229,
      "step": 47601
    },
    {
      "epoch": 4.5395851135253905e-05,
      "step": 47601,
      "training_step_time": 7.9519362449646
    },
    {
      "epoch": 4.539680480957031e-05,
      "model_forward_time": 1.7033185958862305,
      "step": 47602
    },
    {
      "epoch": 4.539680480957031e-05,
      "step": 47602,
      "training_step_time": 7.79291844367981
    },
    {
      "epoch": 4.539775848388672e-05,
      "model_forward_time": 1.6922752857208252,
      "step": 47603
    },
    {
      "epoch": 4.539775848388672e-05,
      "step": 47603,
      "training_step_time": 7.835383892059326
    },
    {
      "epoch": 4.5398712158203124e-05,
      "model_forward_time": 2.2499594688415527,
      "step": 47604
    },
    {
      "epoch": 4.5398712158203124e-05,
      "step": 47604,
      "training_step_time": 8.588269472122192
    },
    {
      "epoch": 4.539966583251953e-05,
      "model_forward_time": 1.6996965408325195,
      "step": 47605
    },
    {
      "epoch": 4.539966583251953e-05,
      "step": 47605,
      "training_step_time": 7.8474767208099365
    },
    {
      "epoch": 4.540061950683594e-05,
      "model_forward_time": 1.3100686073303223,
      "step": 47606
    },
    {
      "epoch": 4.540061950683594e-05,
      "step": 47606,
      "training_step_time": 9.303180694580078
    },
    {
      "epoch": 4.540157318115234e-05,
      "model_forward_time": 1.6923179626464844,
      "step": 47607
    },
    {
      "epoch": 4.540157318115234e-05,
      "step": 47607,
      "training_step_time": 7.581865549087524
    },
    {
      "epoch": 4.540252685546875e-05,
      "model_forward_time": 1.6924939155578613,
      "step": 47608
    },
    {
      "epoch": 4.540252685546875e-05,
      "step": 47608,
      "training_step_time": 8.164300441741943
    },
    {
      "epoch": 4.540348052978516e-05,
      "model_forward_time": 1.6857318878173828,
      "step": 47609
    },
    {
      "epoch": 4.540348052978516e-05,
      "step": 47609,
      "training_step_time": 7.588737964630127
    },
    {
      "epoch": 4.540443420410156e-05,
      "grad_norm": 0.1337578147649765,
      "learning_rate": 5.805190630284457e-06,
      "loss": 0.0594,
      "step": 47610
    },
    {
      "epoch": 4.540443420410156e-05,
      "model_forward_time": 1.720571756362915,
      "step": 47610
    },
    {
      "dynamics_loss_avg": 0.0473951967433095,
      "epoch": 4.540443420410156e-05,
      "step": 47610
    },
    {
      "action_loss_avg": 0.001213714521145448,
      "epoch": 4.540443420410156e-05,
      "step": 47610
    },
    {
      "epoch": 4.540443420410156e-05,
      "step": 47610,
      "training_step_time": 8.327633380889893
    },
    {
      "epoch": 4.5405387878417966e-05,
      "model_forward_time": 1.6907384395599365,
      "step": 47611
    },
    {
      "epoch": 4.5405387878417966e-05,
      "step": 47611,
      "training_step_time": 7.634918451309204
    },
    {
      "epoch": 4.540634155273438e-05,
      "model_forward_time": 1.689392328262329,
      "step": 47612
    },
    {
      "epoch": 4.540634155273438e-05,
      "step": 47612,
      "training_step_time": 7.7266013622283936
    },
    {
      "epoch": 4.540729522705078e-05,
      "model_forward_time": 2.244933605194092,
      "step": 47613
    },
    {
      "epoch": 4.540729522705078e-05,
      "step": 47613,
      "training_step_time": 8.991668462753296
    },
    {
      "epoch": 4.5408248901367185e-05,
      "model_forward_time": 1.692896842956543,
      "step": 47614
    },
    {
      "epoch": 4.5408248901367185e-05,
      "step": 47614,
      "training_step_time": 7.7478413581848145
    },
    {
      "epoch": 4.5409202575683596e-05,
      "model_forward_time": 1.6997528076171875,
      "step": 47615
    },
    {
      "epoch": 4.5409202575683596e-05,
      "step": 47615,
      "training_step_time": 7.696515798568726
    },
    {
      "epoch": 4.541015625e-05,
      "model_forward_time": 1.6897995471954346,
      "step": 47616
    },
    {
      "epoch": 4.541015625e-05,
      "step": 47616,
      "training_step_time": 8.206845998764038
    },
    {
      "epoch": 4.5411109924316404e-05,
      "model_forward_time": 1.6964099407196045,
      "step": 47617
    },
    {
      "epoch": 4.5411109924316404e-05,
      "step": 47617,
      "training_step_time": 7.694680690765381
    },
    {
      "epoch": 4.5412063598632815e-05,
      "model_forward_time": 1.7485072612762451,
      "step": 47618
    },
    {
      "epoch": 4.5412063598632815e-05,
      "step": 47618,
      "training_step_time": 8.338785886764526
    },
    {
      "epoch": 4.541301727294922e-05,
      "model_forward_time": 1.7717442512512207,
      "step": 47619
    },
    {
      "epoch": 4.541301727294922e-05,
      "step": 47619,
      "training_step_time": 8.398414134979248
    },
    {
      "epoch": 4.541397094726562e-05,
      "grad_norm": 0.11004897207021713,
      "learning_rate": 5.803558697249729e-06,
      "loss": 0.0568,
      "step": 47620
    },
    {
      "epoch": 4.541397094726562e-05,
      "model_forward_time": 1.779395341873169,
      "step": 47620
    },
    {
      "dynamics_loss_avg": 0.06008756905794144,
      "epoch": 4.541397094726562e-05,
      "step": 47620
    },
    {
      "action_loss_avg": 0.0023229556129081173,
      "epoch": 4.541397094726562e-05,
      "step": 47620
    },
    {
      "epoch": 4.541397094726562e-05,
      "step": 47620,
      "training_step_time": 7.683679103851318
    },
    {
      "epoch": 4.541492462158203e-05,
      "model_forward_time": 1.7752022743225098,
      "step": 47621
    },
    {
      "epoch": 4.541492462158203e-05,
      "step": 47621,
      "training_step_time": 7.5915844440460205
    },
    {
      "epoch": 4.541587829589844e-05,
      "model_forward_time": 1.7625958919525146,
      "step": 47622
    },
    {
      "epoch": 4.541587829589844e-05,
      "step": 47622,
      "training_step_time": 7.646483421325684
    },
    {
      "epoch": 4.541683197021484e-05,
      "model_forward_time": 2.29203200340271,
      "step": 47623
    },
    {
      "epoch": 4.541683197021484e-05,
      "step": 47623,
      "training_step_time": 8.45399522781372
    },
    {
      "epoch": 4.541778564453125e-05,
      "model_forward_time": 1.711012601852417,
      "step": 47624
    },
    {
      "epoch": 4.541778564453125e-05,
      "step": 47624,
      "training_step_time": 7.5900866985321045
    },
    {
      "epoch": 4.5418739318847656e-05,
      "model_forward_time": 1.7243366241455078,
      "step": 47625
    },
    {
      "epoch": 4.5418739318847656e-05,
      "step": 47625,
      "training_step_time": 8.929497957229614
    },
    {
      "epoch": 4.541969299316406e-05,
      "model_forward_time": 1.7141389846801758,
      "step": 47626
    },
    {
      "epoch": 4.541969299316406e-05,
      "step": 47626,
      "training_step_time": 8.282236099243164
    },
    {
      "epoch": 4.542064666748047e-05,
      "model_forward_time": 1.7132835388183594,
      "step": 47627
    },
    {
      "epoch": 4.542064666748047e-05,
      "step": 47627,
      "training_step_time": 7.6259446144104
    },
    {
      "epoch": 4.5421600341796875e-05,
      "model_forward_time": 1.7314369678497314,
      "step": 47628
    },
    {
      "epoch": 4.5421600341796875e-05,
      "step": 47628,
      "training_step_time": 7.78273868560791
    },
    {
      "epoch": 4.542255401611328e-05,
      "model_forward_time": 1.71343994140625,
      "step": 47629
    },
    {
      "epoch": 4.542255401611328e-05,
      "step": 47629,
      "training_step_time": 7.654113292694092
    },
    {
      "epoch": 4.542350769042969e-05,
      "grad_norm": 0.11124227195978165,
      "learning_rate": 5.801926676339028e-06,
      "loss": 0.0569,
      "step": 47630
    },
    {
      "epoch": 4.542350769042969e-05,
      "model_forward_time": 1.7060110569000244,
      "step": 47630
    },
    {
      "dynamics_loss_avg": 0.06746612973511219,
      "epoch": 4.542350769042969e-05,
      "step": 47630
    },
    {
      "action_loss_avg": 0.002219690487254411,
      "epoch": 4.542350769042969e-05,
      "step": 47630
    },
    {
      "epoch": 4.542350769042969e-05,
      "step": 47630,
      "training_step_time": 7.512049436569214
    },
    {
      "epoch": 4.5424461364746094e-05,
      "model_forward_time": 1.7160236835479736,
      "step": 47631
    },
    {
      "epoch": 4.5424461364746094e-05,
      "step": 47631,
      "training_step_time": 7.663878917694092
    },
    {
      "epoch": 4.54254150390625e-05,
      "model_forward_time": 2.2769370079040527,
      "step": 47632
    },
    {
      "epoch": 4.54254150390625e-05,
      "step": 47632,
      "training_step_time": 8.651628732681274
    },
    {
      "epoch": 4.542636871337891e-05,
      "model_forward_time": 1.7318620681762695,
      "step": 47633
    },
    {
      "epoch": 4.542636871337891e-05,
      "step": 47633,
      "training_step_time": 8.031415224075317
    },
    {
      "epoch": 4.542732238769531e-05,
      "model_forward_time": 1.778219223022461,
      "step": 47634
    },
    {
      "epoch": 4.542732238769531e-05,
      "step": 47634,
      "training_step_time": 8.108145236968994
    },
    {
      "epoch": 4.542827606201172e-05,
      "model_forward_time": 1.7189280986785889,
      "step": 47635
    },
    {
      "epoch": 4.542827606201172e-05,
      "step": 47635,
      "training_step_time": 8.164589405059814
    },
    {
      "epoch": 4.542922973632813e-05,
      "model_forward_time": 1.7301902770996094,
      "step": 47636
    },
    {
      "epoch": 4.542922973632813e-05,
      "step": 47636,
      "training_step_time": 8.390689611434937
    },
    {
      "epoch": 4.543018341064453e-05,
      "model_forward_time": 1.7117266654968262,
      "step": 47637
    },
    {
      "epoch": 4.543018341064453e-05,
      "step": 47637,
      "training_step_time": 7.834914445877075
    },
    {
      "epoch": 4.5431137084960936e-05,
      "model_forward_time": 1.7234444618225098,
      "step": 47638
    },
    {
      "epoch": 4.5431137084960936e-05,
      "step": 47638,
      "training_step_time": 7.718409061431885
    },
    {
      "epoch": 4.5432090759277347e-05,
      "model_forward_time": 1.7399482727050781,
      "step": 47639
    },
    {
      "epoch": 4.5432090759277347e-05,
      "step": 47639,
      "training_step_time": 8.09305739402771
    },
    {
      "epoch": 4.543304443359375e-05,
      "grad_norm": 0.1712336540222168,
      "learning_rate": 5.800294567730833e-06,
      "loss": 0.0558,
      "step": 47640
    },
    {
      "epoch": 4.543304443359375e-05,
      "model_forward_time": 1.7269248962402344,
      "step": 47640
    },
    {
      "dynamics_loss_avg": 0.03856467567384243,
      "epoch": 4.543304443359375e-05,
      "step": 47640
    },
    {
      "action_loss_avg": 0.0016482492996146903,
      "epoch": 4.543304443359375e-05,
      "step": 47640
    },
    {
      "epoch": 4.543304443359375e-05,
      "step": 47640,
      "training_step_time": 7.710938215255737
    },
    {
      "epoch": 4.5433998107910155e-05,
      "model_forward_time": 2.3075335025787354,
      "step": 47641
    },
    {
      "epoch": 4.5433998107910155e-05,
      "step": 47641,
      "training_step_time": 8.339235067367554
    },
    {
      "epoch": 4.5434951782226565e-05,
      "model_forward_time": 1.722813367843628,
      "step": 47642
    },
    {
      "epoch": 4.5434951782226565e-05,
      "step": 47642,
      "training_step_time": 7.685223579406738
    },
    {
      "epoch": 4.543590545654297e-05,
      "model_forward_time": 1.7220673561096191,
      "step": 47643
    },
    {
      "epoch": 4.543590545654297e-05,
      "step": 47643,
      "training_step_time": 8.421187162399292
    },
    {
      "epoch": 4.5436859130859373e-05,
      "model_forward_time": 1.7380096912384033,
      "step": 47644
    },
    {
      "epoch": 4.5436859130859373e-05,
      "step": 47644,
      "training_step_time": 7.859813928604126
    },
    {
      "epoch": 4.5437812805175784e-05,
      "model_forward_time": 1.7150444984436035,
      "step": 47645
    },
    {
      "epoch": 4.5437812805175784e-05,
      "step": 47645,
      "training_step_time": 8.72996473312378
    },
    {
      "epoch": 4.543876647949219e-05,
      "model_forward_time": 1.7140755653381348,
      "step": 47646
    },
    {
      "epoch": 4.543876647949219e-05,
      "step": 47646,
      "training_step_time": 7.946632385253906
    },
    {
      "epoch": 4.543972015380859e-05,
      "model_forward_time": 1.7117714881896973,
      "step": 47647
    },
    {
      "epoch": 4.543972015380859e-05,
      "step": 47647,
      "training_step_time": 7.898362874984741
    },
    {
      "epoch": 4.5440673828125e-05,
      "model_forward_time": 1.7072746753692627,
      "step": 47648
    },
    {
      "epoch": 4.5440673828125e-05,
      "step": 47648,
      "training_step_time": 7.5175580978393555
    },
    {
      "epoch": 4.544162750244141e-05,
      "model_forward_time": 1.7019941806793213,
      "step": 47649
    },
    {
      "epoch": 4.544162750244141e-05,
      "step": 47649,
      "training_step_time": 7.89616322517395
    },
    {
      "epoch": 4.544258117675781e-05,
      "grad_norm": 0.11495566368103027,
      "learning_rate": 5.798662371603628e-06,
      "loss": 0.0571,
      "step": 47650
    },
    {
      "epoch": 4.544258117675781e-05,
      "model_forward_time": 2.2862744331359863,
      "step": 47650
    },
    {
      "dynamics_loss_avg": 0.06740816757082939,
      "epoch": 4.544258117675781e-05,
      "step": 47650
    },
    {
      "action_loss_avg": 0.0020869885222055016,
      "epoch": 4.544258117675781e-05,
      "step": 47650
    },
    {
      "epoch": 4.544258117675781e-05,
      "step": 47650,
      "training_step_time": 8.626190423965454
    },
    {
      "epoch": 4.544353485107422e-05,
      "model_forward_time": 1.7097315788269043,
      "step": 47651
    },
    {
      "epoch": 4.544353485107422e-05,
      "step": 47651,
      "training_step_time": 8.30374550819397
    },
    {
      "epoch": 4.5444488525390626e-05,
      "model_forward_time": 1.359968900680542,
      "step": 47652
    },
    {
      "epoch": 4.5444488525390626e-05,
      "step": 47652,
      "training_step_time": 7.9386842250823975
    },
    {
      "epoch": 4.544544219970703e-05,
      "model_forward_time": 1.7255265712738037,
      "step": 47653
    },
    {
      "epoch": 4.544544219970703e-05,
      "step": 47653,
      "training_step_time": 8.658419847488403
    },
    {
      "epoch": 4.544639587402344e-05,
      "model_forward_time": 1.7088687419891357,
      "step": 47654
    },
    {
      "epoch": 4.544639587402344e-05,
      "step": 47654,
      "training_step_time": 7.847301006317139
    },
    {
      "epoch": 4.5447349548339845e-05,
      "model_forward_time": 1.7028748989105225,
      "step": 47655
    },
    {
      "epoch": 4.5447349548339845e-05,
      "step": 47655,
      "training_step_time": 8.43283987045288
    },
    {
      "epoch": 4.544830322265625e-05,
      "model_forward_time": 1.7083075046539307,
      "step": 47656
    },
    {
      "epoch": 4.544830322265625e-05,
      "step": 47656,
      "training_step_time": 8.70073676109314
    },
    {
      "epoch": 4.544925689697265e-05,
      "model_forward_time": 1.7030305862426758,
      "step": 47657
    },
    {
      "epoch": 4.544925689697265e-05,
      "step": 47657,
      "training_step_time": 7.447835922241211
    },
    {
      "epoch": 4.5450210571289064e-05,
      "model_forward_time": 1.7101781368255615,
      "step": 47658
    },
    {
      "epoch": 4.5450210571289064e-05,
      "step": 47658,
      "training_step_time": 7.840016603469849
    },
    {
      "epoch": 4.545116424560547e-05,
      "model_forward_time": 2.250850200653076,
      "step": 47659
    },
    {
      "epoch": 4.545116424560547e-05,
      "step": 47659,
      "training_step_time": 7.582302570343018
    },
    {
      "epoch": 4.545211791992187e-05,
      "grad_norm": 0.10856898874044418,
      "learning_rate": 5.797030088135904e-06,
      "loss": 0.0565,
      "step": 47660
    },
    {
      "epoch": 4.545211791992187e-05,
      "model_forward_time": 1.6751422882080078,
      "step": 47660
    },
    {
      "dynamics_loss_avg": 0.07277396321296692,
      "epoch": 4.545211791992187e-05,
      "step": 47660
    },
    {
      "action_loss_avg": 0.002542849650490098,
      "epoch": 4.545211791992187e-05,
      "step": 47660
    },
    {
      "epoch": 4.545211791992187e-05,
      "step": 47660,
      "training_step_time": 7.644556999206543
    },
    {
      "epoch": 4.545307159423828e-05,
      "model_forward_time": 1.7102034091949463,
      "step": 47661
    },
    {
      "epoch": 4.545307159423828e-05,
      "step": 47661,
      "training_step_time": 7.629916191101074
    },
    {
      "epoch": 4.5454025268554687e-05,
      "model_forward_time": 1.6994361877441406,
      "step": 47662
    },
    {
      "epoch": 4.5454025268554687e-05,
      "step": 47662,
      "training_step_time": 7.5229997634887695
    },
    {
      "epoch": 4.545497894287109e-05,
      "model_forward_time": 1.695382833480835,
      "step": 47663
    },
    {
      "epoch": 4.545497894287109e-05,
      "step": 47663,
      "training_step_time": 8.145358324050903
    },
    {
      "epoch": 4.54559326171875e-05,
      "model_forward_time": 1.6954734325408936,
      "step": 47664
    },
    {
      "epoch": 4.54559326171875e-05,
      "step": 47664,
      "training_step_time": 10.355655431747437
    },
    {
      "epoch": 4.5456886291503905e-05,
      "model_forward_time": 1.690286636352539,
      "step": 47665
    },
    {
      "epoch": 4.5456886291503905e-05,
      "step": 47665,
      "training_step_time": 7.709628105163574
    },
    {
      "epoch": 4.545783996582031e-05,
      "model_forward_time": 1.6876511573791504,
      "step": 47666
    },
    {
      "epoch": 4.545783996582031e-05,
      "step": 47666,
      "training_step_time": 7.293227434158325
    },
    {
      "epoch": 4.545879364013672e-05,
      "model_forward_time": 1.6845977306365967,
      "step": 47667
    },
    {
      "epoch": 4.545879364013672e-05,
      "step": 47667,
      "training_step_time": 8.777349710464478
    },
    {
      "epoch": 4.5459747314453124e-05,
      "model_forward_time": 1.6882119178771973,
      "step": 47668
    },
    {
      "epoch": 4.5459747314453124e-05,
      "step": 47668,
      "training_step_time": 7.636182546615601
    },
    {
      "epoch": 4.546070098876953e-05,
      "model_forward_time": 2.2495596408843994,
      "step": 47669
    },
    {
      "epoch": 4.546070098876953e-05,
      "step": 47669,
      "training_step_time": 8.941859722137451
    },
    {
      "epoch": 4.546165466308594e-05,
      "grad_norm": 0.09883477538824081,
      "learning_rate": 5.7953977175061705e-06,
      "loss": 0.0544,
      "step": 47670
    },
    {
      "epoch": 4.546165466308594e-05,
      "model_forward_time": 1.6974232196807861,
      "step": 47670
    },
    {
      "dynamics_loss_avg": 0.06553962752223015,
      "epoch": 4.546165466308594e-05,
      "step": 47670
    },
    {
      "action_loss_avg": 0.0016246542916633188,
      "epoch": 4.546165466308594e-05,
      "step": 47670
    },
    {
      "epoch": 4.546165466308594e-05,
      "step": 47670,
      "training_step_time": 9.620893478393555
    },
    {
      "epoch": 4.546260833740234e-05,
      "model_forward_time": 1.6979665756225586,
      "step": 47671
    },
    {
      "epoch": 4.546260833740234e-05,
      "step": 47671,
      "training_step_time": 7.818260908126831
    },
    {
      "epoch": 4.546356201171875e-05,
      "model_forward_time": 1.7017076015472412,
      "step": 47672
    },
    {
      "epoch": 4.546356201171875e-05,
      "step": 47672,
      "training_step_time": 7.862440347671509
    },
    {
      "epoch": 4.546451568603516e-05,
      "model_forward_time": 1.6917383670806885,
      "step": 47673
    },
    {
      "epoch": 4.546451568603516e-05,
      "step": 47673,
      "training_step_time": 7.798409700393677
    },
    {
      "epoch": 4.546546936035156e-05,
      "model_forward_time": 1.704223871231079,
      "step": 47674
    },
    {
      "epoch": 4.546546936035156e-05,
      "step": 47674,
      "training_step_time": 9.617136001586914
    },
    {
      "epoch": 4.5466423034667966e-05,
      "model_forward_time": 1.686255693435669,
      "step": 47675
    },
    {
      "epoch": 4.5466423034667966e-05,
      "step": 47675,
      "training_step_time": 8.326916933059692
    },
    {
      "epoch": 4.546737670898438e-05,
      "model_forward_time": 1.701101303100586,
      "step": 47676
    },
    {
      "epoch": 4.546737670898438e-05,
      "step": 47676,
      "training_step_time": 7.7153801918029785
    },
    {
      "epoch": 4.546833038330078e-05,
      "model_forward_time": 1.6896145343780518,
      "step": 47677
    },
    {
      "epoch": 4.546833038330078e-05,
      "step": 47677,
      "training_step_time": 7.621679306030273
    },
    {
      "epoch": 4.5469284057617185e-05,
      "model_forward_time": 2.240520477294922,
      "step": 47678
    },
    {
      "epoch": 4.5469284057617185e-05,
      "step": 47678,
      "training_step_time": 7.873357534408569
    },
    {
      "epoch": 4.5470237731933596e-05,
      "model_forward_time": 1.3144478797912598,
      "step": 47679
    },
    {
      "epoch": 4.5470237731933596e-05,
      "step": 47679,
      "training_step_time": 8.348576307296753
    },
    {
      "epoch": 4.547119140625e-05,
      "grad_norm": 0.11042126268148422,
      "learning_rate": 5.7937652598929385e-06,
      "loss": 0.0582,
      "step": 47680
    },
    {
      "epoch": 4.547119140625e-05,
      "model_forward_time": 1.6838648319244385,
      "step": 47680
    },
    {
      "dynamics_loss_avg": 0.05917057301849127,
      "epoch": 4.547119140625e-05,
      "step": 47680
    },
    {
      "action_loss_avg": 0.001623117784038186,
      "epoch": 4.547119140625e-05,
      "step": 47680
    },
    {
      "epoch": 4.547119140625e-05,
      "step": 47680,
      "training_step_time": 7.769427537918091
    },
    {
      "epoch": 4.5472145080566404e-05,
      "model_forward_time": 1.6792383193969727,
      "step": 47681
    },
    {
      "epoch": 4.5472145080566404e-05,
      "step": 47681,
      "training_step_time": 7.820124626159668
    },
    {
      "epoch": 4.5473098754882815e-05,
      "model_forward_time": 1.313042163848877,
      "step": 47682
    },
    {
      "epoch": 4.5473098754882815e-05,
      "step": 47682,
      "training_step_time": 7.791739463806152
    },
    {
      "epoch": 4.547405242919922e-05,
      "model_forward_time": 1.6896965503692627,
      "step": 47683
    },
    {
      "epoch": 4.547405242919922e-05,
      "step": 47683,
      "training_step_time": 7.742218017578125
    },
    {
      "epoch": 4.547500610351562e-05,
      "model_forward_time": 1.7056915760040283,
      "step": 47684
    },
    {
      "epoch": 4.547500610351562e-05,
      "step": 47684,
      "training_step_time": 7.662715435028076
    },
    {
      "epoch": 4.5475959777832033e-05,
      "model_forward_time": 1.703561782836914,
      "step": 47685
    },
    {
      "epoch": 4.5475959777832033e-05,
      "step": 47685,
      "training_step_time": 8.445321083068848
    },
    {
      "epoch": 4.547691345214844e-05,
      "model_forward_time": 1.7010416984558105,
      "step": 47686
    },
    {
      "epoch": 4.547691345214844e-05,
      "step": 47686,
      "training_step_time": 7.829589128494263
    },
    {
      "epoch": 4.547786712646484e-05,
      "model_forward_time": 2.289835214614868,
      "step": 47687
    },
    {
      "epoch": 4.547786712646484e-05,
      "step": 47687,
      "training_step_time": 7.930380821228027
    },
    {
      "epoch": 4.547882080078125e-05,
      "model_forward_time": 1.7005877494812012,
      "step": 47688
    },
    {
      "epoch": 4.547882080078125e-05,
      "step": 47688,
      "training_step_time": 8.37035346031189
    },
    {
      "epoch": 4.5479774475097656e-05,
      "model_forward_time": 1.7083899974822998,
      "step": 47689
    },
    {
      "epoch": 4.5479774475097656e-05,
      "step": 47689,
      "training_step_time": 8.660024881362915
    },
    {
      "epoch": 4.548072814941406e-05,
      "grad_norm": 0.09110327810049057,
      "learning_rate": 5.79213271547473e-06,
      "loss": 0.0546,
      "step": 47690
    },
    {
      "epoch": 4.548072814941406e-05,
      "model_forward_time": 1.6891798973083496,
      "step": 47690
    },
    {
      "dynamics_loss_avg": 0.0528073001652956,
      "epoch": 4.548072814941406e-05,
      "step": 47690
    },
    {
      "action_loss_avg": 0.002178936585551128,
      "epoch": 4.548072814941406e-05,
      "step": 47690
    },
    {
      "epoch": 4.548072814941406e-05,
      "step": 47690,
      "training_step_time": 7.708544731140137
    },
    {
      "epoch": 4.548168182373047e-05,
      "model_forward_time": 1.6922030448913574,
      "step": 47691
    },
    {
      "epoch": 4.548168182373047e-05,
      "step": 47691,
      "training_step_time": 7.742628335952759
    },
    {
      "epoch": 4.5482635498046875e-05,
      "model_forward_time": 1.7031035423278809,
      "step": 47692
    },
    {
      "epoch": 4.5482635498046875e-05,
      "step": 47692,
      "training_step_time": 9.46913743019104
    },
    {
      "epoch": 4.548358917236328e-05,
      "model_forward_time": 1.7249009609222412,
      "step": 47693
    },
    {
      "epoch": 4.548358917236328e-05,
      "step": 47693,
      "training_step_time": 7.720291614532471
    },
    {
      "epoch": 4.548454284667969e-05,
      "model_forward_time": 1.7079105377197266,
      "step": 47694
    },
    {
      "epoch": 4.548454284667969e-05,
      "step": 47694,
      "training_step_time": 7.8862974643707275
    },
    {
      "epoch": 4.5485496520996094e-05,
      "model_forward_time": 1.321958065032959,
      "step": 47695
    },
    {
      "epoch": 4.5485496520996094e-05,
      "step": 47695,
      "training_step_time": 7.828077554702759
    },
    {
      "epoch": 4.54864501953125e-05,
      "model_forward_time": 1.7009611129760742,
      "step": 47696
    },
    {
      "epoch": 4.54864501953125e-05,
      "step": 47696,
      "training_step_time": 8.550566673278809
    },
    {
      "epoch": 4.548740386962891e-05,
      "model_forward_time": 2.260800361633301,
      "step": 47697
    },
    {
      "epoch": 4.548740386962891e-05,
      "step": 47697,
      "training_step_time": 7.779389142990112
    },
    {
      "epoch": 4.548835754394531e-05,
      "model_forward_time": 1.7194766998291016,
      "step": 47698
    },
    {
      "epoch": 4.548835754394531e-05,
      "step": 47698,
      "training_step_time": 8.699624300003052
    },
    {
      "epoch": 4.548931121826172e-05,
      "model_forward_time": 1.698911190032959,
      "step": 47699
    },
    {
      "epoch": 4.548931121826172e-05,
      "step": 47699,
      "training_step_time": 9.337359428405762
    },
    {
      "epoch": 4.549026489257813e-05,
      "grad_norm": 0.11907130479812622,
      "learning_rate": 5.790500084430078e-06,
      "loss": 0.0545,
      "step": 47700
    },
    {
      "epoch": 4.549026489257813e-05,
      "model_forward_time": 1.6985719203948975,
      "step": 47700
    },
    {
      "dynamics_loss_avg": 0.06436180863529443,
      "epoch": 4.549026489257813e-05,
      "step": 47700
    },
    {
      "action_loss_avg": 0.0018662910209968687,
      "epoch": 4.549026489257813e-05,
      "step": 47700
    },
    {
      "epoch": 4.549026489257813e-05,
      "step": 47700,
      "training_step_time": 8.679757833480835
    },
    {
      "epoch": 4.549121856689453e-05,
      "model_forward_time": 1.6977019309997559,
      "step": 47701
    },
    {
      "epoch": 4.549121856689453e-05,
      "step": 47701,
      "training_step_time": 7.757817506790161
    },
    {
      "epoch": 4.5492172241210936e-05,
      "model_forward_time": 1.7134504318237305,
      "step": 47702
    },
    {
      "epoch": 4.5492172241210936e-05,
      "step": 47702,
      "training_step_time": 7.775061130523682
    },
    {
      "epoch": 4.5493125915527347e-05,
      "model_forward_time": 1.7042388916015625,
      "step": 47703
    },
    {
      "epoch": 4.5493125915527347e-05,
      "step": 47703,
      "training_step_time": 8.036811351776123
    },
    {
      "epoch": 4.549407958984375e-05,
      "model_forward_time": 1.7059297561645508,
      "step": 47704
    },
    {
      "epoch": 4.549407958984375e-05,
      "step": 47704,
      "training_step_time": 8.528807163238525
    },
    {
      "epoch": 4.5495033264160155e-05,
      "model_forward_time": 1.6954593658447266,
      "step": 47705
    },
    {
      "epoch": 4.5495033264160155e-05,
      "step": 47705,
      "training_step_time": 7.619602680206299
    },
    {
      "epoch": 4.5495986938476565e-05,
      "model_forward_time": 2.2697484493255615,
      "step": 47706
    },
    {
      "epoch": 4.5495986938476565e-05,
      "step": 47706,
      "training_step_time": 7.723977327346802
    },
    {
      "epoch": 4.549694061279297e-05,
      "model_forward_time": 1.726961612701416,
      "step": 47707
    },
    {
      "epoch": 4.549694061279297e-05,
      "step": 47707,
      "training_step_time": 8.483420372009277
    },
    {
      "epoch": 4.5497894287109373e-05,
      "model_forward_time": 1.7034170627593994,
      "step": 47708
    },
    {
      "epoch": 4.5497894287109373e-05,
      "step": 47708,
      "training_step_time": 7.768909692764282
    },
    {
      "epoch": 4.5498847961425784e-05,
      "model_forward_time": 1.7074954509735107,
      "step": 47709
    },
    {
      "epoch": 4.5498847961425784e-05,
      "step": 47709,
      "training_step_time": 8.469391345977783
    },
    {
      "epoch": 4.549980163574219e-05,
      "grad_norm": 0.0765455812215805,
      "learning_rate": 5.7888673669375276e-06,
      "loss": 0.0583,
      "step": 47710
    },
    {
      "epoch": 4.549980163574219e-05,
      "model_forward_time": 1.6988155841827393,
      "step": 47710
    },
    {
      "dynamics_loss_avg": 0.05559022706001997,
      "epoch": 4.549980163574219e-05,
      "step": 47710
    },
    {
      "action_loss_avg": 0.0009790471580345184,
      "epoch": 4.549980163574219e-05,
      "step": 47710
    },
    {
      "epoch": 4.549980163574219e-05,
      "step": 47710,
      "training_step_time": 7.7871527671813965
    },
    {
      "epoch": 4.550075531005859e-05,
      "model_forward_time": 1.718994140625,
      "step": 47711
    },
    {
      "epoch": 4.550075531005859e-05,
      "step": 47711,
      "training_step_time": 7.7351367473602295
    },
    {
      "epoch": 4.5501708984375e-05,
      "model_forward_time": 1.699533462524414,
      "step": 47712
    },
    {
      "epoch": 4.5501708984375e-05,
      "step": 47712,
      "training_step_time": 7.980526447296143
    },
    {
      "epoch": 4.550266265869141e-05,
      "model_forward_time": 1.703038215637207,
      "step": 47713
    },
    {
      "epoch": 4.550266265869141e-05,
      "step": 47713,
      "training_step_time": 8.874332427978516
    },
    {
      "epoch": 4.550361633300781e-05,
      "model_forward_time": 1.6899216175079346,
      "step": 47714
    },
    {
      "epoch": 4.550361633300781e-05,
      "step": 47714,
      "training_step_time": 8.240418434143066
    },
    {
      "epoch": 4.550457000732422e-05,
      "model_forward_time": 1.692267894744873,
      "step": 47715
    },
    {
      "epoch": 4.550457000732422e-05,
      "step": 47715,
      "training_step_time": 6.91068696975708
    },
    {
      "epoch": 4.5505523681640626e-05,
      "model_forward_time": 1.6914026737213135,
      "step": 47716
    },
    {
      "epoch": 4.5505523681640626e-05,
      "step": 47716,
      "training_step_time": 6.857565402984619
    },
    {
      "epoch": 4.550647735595703e-05,
      "model_forward_time": 1.6898448467254639,
      "step": 47717
    },
    {
      "epoch": 4.550647735595703e-05,
      "step": 47717,
      "training_step_time": 6.835715293884277
    },
    {
      "epoch": 4.550743103027344e-05,
      "model_forward_time": 1.7017452716827393,
      "step": 47718
    },
    {
      "epoch": 4.550743103027344e-05,
      "step": 47718,
      "training_step_time": 6.832780361175537
    },
    {
      "epoch": 4.5508384704589845e-05,
      "model_forward_time": 1.693617582321167,
      "step": 47719
    },
    {
      "epoch": 4.5508384704589845e-05,
      "step": 47719,
      "training_step_time": 6.862590789794922
    },
    {
      "epoch": 4.550933837890625e-05,
      "grad_norm": 0.08700957149267197,
      "learning_rate": 5.787234563175626e-06,
      "loss": 0.0572,
      "step": 47720
    },
    {
      "epoch": 4.550933837890625e-05,
      "model_forward_time": 1.6920862197875977,
      "step": 47720
    },
    {
      "dynamics_loss_avg": 0.06853484418243169,
      "epoch": 4.550933837890625e-05,
      "step": 47720
    },
    {
      "action_loss_avg": 0.002056030661333352,
      "epoch": 4.550933837890625e-05,
      "step": 47720
    },
    {
      "epoch": 4.550933837890625e-05,
      "step": 47720,
      "training_step_time": 6.810100317001343
    },
    {
      "epoch": 4.551029205322265e-05,
      "model_forward_time": 1.6890923976898193,
      "step": 47721
    },
    {
      "epoch": 4.551029205322265e-05,
      "step": 47721,
      "training_step_time": 6.758726358413696
    },
    {
      "epoch": 4.5511245727539064e-05,
      "model_forward_time": 1.6873960494995117,
      "step": 47722
    },
    {
      "epoch": 4.5511245727539064e-05,
      "step": 47722,
      "training_step_time": 6.786024570465088
    },
    {
      "epoch": 4.551219940185547e-05,
      "model_forward_time": 2.2579543590545654,
      "step": 47723
    },
    {
      "epoch": 4.551219940185547e-05,
      "step": 47723,
      "training_step_time": 9.341193437576294
    },
    {
      "epoch": 4.551315307617187e-05,
      "model_forward_time": 1.6893994808197021,
      "step": 47724
    },
    {
      "epoch": 4.551315307617187e-05,
      "step": 47724,
      "training_step_time": 7.699293851852417
    },
    {
      "epoch": 4.551410675048828e-05,
      "model_forward_time": 1.6864888668060303,
      "step": 47725
    },
    {
      "epoch": 4.551410675048828e-05,
      "step": 47725,
      "training_step_time": 6.955585241317749
    },
    {
      "epoch": 4.551506042480469e-05,
      "model_forward_time": 1.69126558303833,
      "step": 47726
    },
    {
      "epoch": 4.551506042480469e-05,
      "step": 47726,
      "training_step_time": 6.916052579879761
    },
    {
      "epoch": 4.551601409912109e-05,
      "model_forward_time": 1.685622215270996,
      "step": 47727
    },
    {
      "epoch": 4.551601409912109e-05,
      "step": 47727,
      "training_step_time": 6.934616327285767
    },
    {
      "epoch": 4.55169677734375e-05,
      "model_forward_time": 1.6886320114135742,
      "step": 47728
    },
    {
      "epoch": 4.55169677734375e-05,
      "step": 47728,
      "training_step_time": 6.933574676513672
    },
    {
      "epoch": 4.5517921447753905e-05,
      "model_forward_time": 1.6867613792419434,
      "step": 47729
    },
    {
      "epoch": 4.5517921447753905e-05,
      "step": 47729,
      "training_step_time": 6.909151792526245
    },
    {
      "epoch": 4.551887512207031e-05,
      "grad_norm": 0.1177808940410614,
      "learning_rate": 5.785601673322936e-06,
      "loss": 0.0584,
      "step": 47730
    },
    {
      "epoch": 4.551887512207031e-05,
      "model_forward_time": 1.684838056564331,
      "step": 47730
    },
    {
      "dynamics_loss_avg": 0.05989463664591312,
      "epoch": 4.551887512207031e-05,
      "step": 47730
    },
    {
      "action_loss_avg": 0.001592738987528719,
      "epoch": 4.551887512207031e-05,
      "step": 47730
    },
    {
      "epoch": 4.551887512207031e-05,
      "step": 47730,
      "training_step_time": 6.916100978851318
    },
    {
      "epoch": 4.551982879638672e-05,
      "model_forward_time": 1.6896247863769531,
      "step": 47731
    },
    {
      "epoch": 4.551982879638672e-05,
      "step": 47731,
      "training_step_time": 6.865194320678711
    },
    {
      "epoch": 4.5520782470703124e-05,
      "model_forward_time": 2.2651190757751465,
      "step": 47732
    },
    {
      "epoch": 4.5520782470703124e-05,
      "step": 47732,
      "training_step_time": 12.134175062179565
    },
    {
      "epoch": 4.552173614501953e-05,
      "model_forward_time": 1.6855297088623047,
      "step": 47733
    },
    {
      "epoch": 4.552173614501953e-05,
      "step": 47733,
      "training_step_time": 7.787975311279297
    },
    {
      "epoch": 4.552268981933594e-05,
      "model_forward_time": 1.6846306324005127,
      "step": 47734
    },
    {
      "epoch": 4.552268981933594e-05,
      "step": 47734,
      "training_step_time": 6.933116674423218
    },
    {
      "epoch": 4.552364349365234e-05,
      "model_forward_time": 1.695760726928711,
      "step": 47735
    },
    {
      "epoch": 4.552364349365234e-05,
      "step": 47735,
      "training_step_time": 6.879837512969971
    },
    {
      "epoch": 4.552459716796875e-05,
      "model_forward_time": 1.693119764328003,
      "step": 47736
    },
    {
      "epoch": 4.552459716796875e-05,
      "step": 47736,
      "training_step_time": 6.922353267669678
    },
    {
      "epoch": 4.552555084228516e-05,
      "model_forward_time": 1.684352159500122,
      "step": 47737
    },
    {
      "epoch": 4.552555084228516e-05,
      "step": 47737,
      "training_step_time": 6.945805788040161
    },
    {
      "epoch": 4.552650451660156e-05,
      "model_forward_time": 1.6948304176330566,
      "step": 47738
    },
    {
      "epoch": 4.552650451660156e-05,
      "step": 47738,
      "training_step_time": 6.854250431060791
    },
    {
      "epoch": 4.5527458190917966e-05,
      "model_forward_time": 1.6954882144927979,
      "step": 47739
    },
    {
      "epoch": 4.5527458190917966e-05,
      "step": 47739,
      "training_step_time": 6.870585680007935
    },
    {
      "epoch": 4.552841186523438e-05,
      "grad_norm": 0.11383076012134552,
      "learning_rate": 5.78396869755803e-06,
      "loss": 0.0564,
      "step": 47740
    },
    {
      "epoch": 4.552841186523438e-05,
      "model_forward_time": 1.69075345993042,
      "step": 47740
    },
    {
      "dynamics_loss_avg": 0.052878227271139625,
      "epoch": 4.552841186523438e-05,
      "step": 47740
    },
    {
      "action_loss_avg": 0.001627022260800004,
      "epoch": 4.552841186523438e-05,
      "step": 47740
    },
    {
      "epoch": 4.552841186523438e-05,
      "step": 47740,
      "training_step_time": 6.896938800811768
    },
    {
      "epoch": 4.552936553955078e-05,
      "model_forward_time": 2.271343469619751,
      "step": 47741
    },
    {
      "epoch": 4.552936553955078e-05,
      "step": 47741,
      "training_step_time": 13.1155686378479
    },
    {
      "epoch": 4.5530319213867185e-05,
      "model_forward_time": 1.6937532424926758,
      "step": 47742
    },
    {
      "epoch": 4.5530319213867185e-05,
      "step": 47742,
      "training_step_time": 7.921266078948975
    },
    {
      "epoch": 4.5531272888183596e-05,
      "model_forward_time": 1.698561429977417,
      "step": 47743
    },
    {
      "epoch": 4.5531272888183596e-05,
      "step": 47743,
      "training_step_time": 6.922515153884888
    },
    {
      "epoch": 4.55322265625e-05,
      "model_forward_time": 1.693258285522461,
      "step": 47744
    },
    {
      "epoch": 4.55322265625e-05,
      "step": 47744,
      "training_step_time": 6.902673721313477
    },
    {
      "epoch": 4.5533180236816404e-05,
      "model_forward_time": 1.6941125392913818,
      "step": 47745
    },
    {
      "epoch": 4.5533180236816404e-05,
      "step": 47745,
      "training_step_time": 6.8629209995269775
    },
    {
      "epoch": 4.5534133911132815e-05,
      "model_forward_time": 1.694035530090332,
      "step": 47746
    },
    {
      "epoch": 4.5534133911132815e-05,
      "step": 47746,
      "training_step_time": 6.944682359695435
    },
    {
      "epoch": 4.553508758544922e-05,
      "model_forward_time": 1.702277421951294,
      "step": 47747
    },
    {
      "epoch": 4.553508758544922e-05,
      "step": 47747,
      "training_step_time": 6.9262285232543945
    },
    {
      "epoch": 4.553604125976562e-05,
      "model_forward_time": 1.6987934112548828,
      "step": 47748
    },
    {
      "epoch": 4.553604125976562e-05,
      "step": 47748,
      "training_step_time": 6.960742712020874
    },
    {
      "epoch": 4.5536994934082033e-05,
      "model_forward_time": 1.6939365863800049,
      "step": 47749
    },
    {
      "epoch": 4.5536994934082033e-05,
      "step": 47749,
      "training_step_time": 6.975382089614868
    },
    {
      "epoch": 4.553794860839844e-05,
      "grad_norm": 0.10351325571537018,
      "learning_rate": 5.782335636059484e-06,
      "loss": 0.0557,
      "step": 47750
    },
    {
      "epoch": 4.553794860839844e-05,
      "model_forward_time": 2.2788538932800293,
      "step": 47750
    },
    {
      "dynamics_loss_avg": 0.05932603310793638,
      "epoch": 4.553794860839844e-05,
      "step": 47750
    },
    {
      "action_loss_avg": 0.0016892279498279094,
      "epoch": 4.553794860839844e-05,
      "step": 47750
    },
    {
      "epoch": 4.553794860839844e-05,
      "step": 47750,
      "training_step_time": 9.143999338150024
    },
    {
      "epoch": 4.553890228271484e-05,
      "model_forward_time": 1.6770551204681396,
      "step": 47751
    },
    {
      "epoch": 4.553890228271484e-05,
      "step": 47751,
      "training_step_time": 8.974436044692993
    },
    {
      "epoch": 4.553985595703125e-05,
      "model_forward_time": 1.6893138885498047,
      "step": 47752
    },
    {
      "epoch": 4.553985595703125e-05,
      "step": 47752,
      "training_step_time": 7.4570088386535645
    },
    {
      "epoch": 4.5540809631347656e-05,
      "model_forward_time": 1.3164215087890625,
      "step": 47753
    },
    {
      "epoch": 4.5540809631347656e-05,
      "step": 47753,
      "training_step_time": 6.930878162384033
    },
    {
      "epoch": 4.554176330566406e-05,
      "model_forward_time": 1.668701410293579,
      "step": 47754
    },
    {
      "epoch": 4.554176330566406e-05,
      "step": 47754,
      "training_step_time": 6.973923683166504
    },
    {
      "epoch": 4.554271697998047e-05,
      "model_forward_time": 1.6927142143249512,
      "step": 47755
    },
    {
      "epoch": 4.554271697998047e-05,
      "step": 47755,
      "training_step_time": 6.977254629135132
    },
    {
      "epoch": 4.5543670654296875e-05,
      "model_forward_time": 1.3133440017700195,
      "step": 47756
    },
    {
      "epoch": 4.5543670654296875e-05,
      "step": 47756,
      "training_step_time": 6.967694282531738
    },
    {
      "epoch": 4.554462432861328e-05,
      "model_forward_time": 1.688690423965454,
      "step": 47757
    },
    {
      "epoch": 4.554462432861328e-05,
      "step": 47757,
      "training_step_time": 6.944964647293091
    },
    {
      "epoch": 4.554557800292969e-05,
      "model_forward_time": 1.6973953247070312,
      "step": 47758
    },
    {
      "epoch": 4.554557800292969e-05,
      "step": 47758,
      "training_step_time": 6.977626085281372
    },
    {
      "epoch": 4.5546531677246094e-05,
      "model_forward_time": 1.6978034973144531,
      "step": 47759
    },
    {
      "epoch": 4.5546531677246094e-05,
      "step": 47759,
      "training_step_time": 7.264546871185303
    },
    {
      "epoch": 4.55474853515625e-05,
      "grad_norm": 0.13508497178554535,
      "learning_rate": 5.780702489005889e-06,
      "loss": 0.0568,
      "step": 47760
    },
    {
      "epoch": 4.55474853515625e-05,
      "model_forward_time": 2.2244584560394287,
      "step": 47760
    },
    {
      "dynamics_loss_avg": 0.0510911051183939,
      "epoch": 4.55474853515625e-05,
      "step": 47760
    },
    {
      "action_loss_avg": 0.0016984389716526494,
      "epoch": 4.55474853515625e-05,
      "step": 47760
    },
    {
      "epoch": 4.55474853515625e-05,
      "step": 47760,
      "training_step_time": 8.746248006820679
    },
    {
      "epoch": 4.554843902587891e-05,
      "model_forward_time": 1.690506935119629,
      "step": 47761
    },
    {
      "epoch": 4.554843902587891e-05,
      "step": 47761,
      "training_step_time": 7.849784851074219
    },
    {
      "epoch": 4.554939270019531e-05,
      "model_forward_time": 1.6931421756744385,
      "step": 47762
    },
    {
      "epoch": 4.554939270019531e-05,
      "step": 47762,
      "training_step_time": 6.980932235717773
    },
    {
      "epoch": 4.555034637451172e-05,
      "model_forward_time": 1.685004472732544,
      "step": 47763
    },
    {
      "epoch": 4.555034637451172e-05,
      "step": 47763,
      "training_step_time": 6.952418565750122
    },
    {
      "epoch": 4.555130004882813e-05,
      "model_forward_time": 1.6833558082580566,
      "step": 47764
    },
    {
      "epoch": 4.555130004882813e-05,
      "step": 47764,
      "training_step_time": 6.8200531005859375
    },
    {
      "epoch": 4.555225372314453e-05,
      "model_forward_time": 1.7024874687194824,
      "step": 47765
    },
    {
      "epoch": 4.555225372314453e-05,
      "step": 47765,
      "training_step_time": 6.930468797683716
    },
    {
      "epoch": 4.5553207397460936e-05,
      "model_forward_time": 1.7089107036590576,
      "step": 47766
    },
    {
      "epoch": 4.5553207397460936e-05,
      "step": 47766,
      "training_step_time": 6.959892511367798
    },
    {
      "epoch": 4.555416107177735e-05,
      "model_forward_time": 1.70267915725708,
      "step": 47767
    },
    {
      "epoch": 4.555416107177735e-05,
      "step": 47767,
      "training_step_time": 6.98098611831665
    },
    {
      "epoch": 4.555511474609375e-05,
      "model_forward_time": 1.7047004699707031,
      "step": 47768
    },
    {
      "epoch": 4.555511474609375e-05,
      "step": 47768,
      "training_step_time": 6.939813137054443
    },
    {
      "epoch": 4.5556068420410155e-05,
      "model_forward_time": 2.2932186126708984,
      "step": 47769
    },
    {
      "epoch": 4.5556068420410155e-05,
      "step": 47769,
      "training_step_time": 10.935591220855713
    },
    {
      "epoch": 4.5557022094726565e-05,
      "grad_norm": 0.13883376121520996,
      "learning_rate": 5.779069256575846e-06,
      "loss": 0.057,
      "step": 47770
    },
    {
      "epoch": 4.5557022094726565e-05,
      "model_forward_time": 1.700399398803711,
      "step": 47770
    },
    {
      "dynamics_loss_avg": 0.05826865239068866,
      "epoch": 4.5557022094726565e-05,
      "step": 47770
    },
    {
      "action_loss_avg": 0.0016745461762184277,
      "epoch": 4.5557022094726565e-05,
      "step": 47770
    },
    {
      "epoch": 4.5557022094726565e-05,
      "step": 47770,
      "training_step_time": 8.413921356201172
    },
    {
      "epoch": 4.555797576904297e-05,
      "model_forward_time": 1.7095067501068115,
      "step": 47771
    },
    {
      "epoch": 4.555797576904297e-05,
      "step": 47771,
      "training_step_time": 6.947027921676636
    },
    {
      "epoch": 4.5558929443359374e-05,
      "model_forward_time": 1.7094860076904297,
      "step": 47772
    },
    {
      "epoch": 4.5558929443359374e-05,
      "step": 47772,
      "training_step_time": 6.982597827911377
    },
    {
      "epoch": 4.5559883117675784e-05,
      "model_forward_time": 1.7071993350982666,
      "step": 47773
    },
    {
      "epoch": 4.5559883117675784e-05,
      "step": 47773,
      "training_step_time": 6.944028615951538
    },
    {
      "epoch": 4.556083679199219e-05,
      "model_forward_time": 1.7075161933898926,
      "step": 47774
    },
    {
      "epoch": 4.556083679199219e-05,
      "step": 47774,
      "training_step_time": 6.9643168449401855
    },
    {
      "epoch": 4.556179046630859e-05,
      "model_forward_time": 1.7023534774780273,
      "step": 47775
    },
    {
      "epoch": 4.556179046630859e-05,
      "step": 47775,
      "training_step_time": 6.983037710189819
    },
    {
      "epoch": 4.5562744140625e-05,
      "model_forward_time": 1.701815128326416,
      "step": 47776
    },
    {
      "epoch": 4.5562744140625e-05,
      "step": 47776,
      "training_step_time": 6.9229230880737305
    },
    {
      "epoch": 4.556369781494141e-05,
      "model_forward_time": 1.7014527320861816,
      "step": 47777
    },
    {
      "epoch": 4.556369781494141e-05,
      "step": 47777,
      "training_step_time": 6.922291278839111
    },
    {
      "epoch": 4.556465148925781e-05,
      "model_forward_time": 2.2927422523498535,
      "step": 47778
    },
    {
      "epoch": 4.556465148925781e-05,
      "step": 47778,
      "training_step_time": 12.343384981155396
    },
    {
      "epoch": 4.556560516357422e-05,
      "model_forward_time": 1.7065584659576416,
      "step": 47779
    },
    {
      "epoch": 4.556560516357422e-05,
      "step": 47779,
      "training_step_time": 7.733471870422363
    },
    {
      "epoch": 4.5566558837890626e-05,
      "grad_norm": 0.1320326030254364,
      "learning_rate": 5.7774359389479574e-06,
      "loss": 0.0579,
      "step": 47780
    },
    {
      "epoch": 4.5566558837890626e-05,
      "model_forward_time": 1.6994085311889648,
      "step": 47780
    },
    {
      "dynamics_loss_avg": 0.04194732662290335,
      "epoch": 4.5566558837890626e-05,
      "step": 47780
    },
    {
      "action_loss_avg": 0.0014421624946407974,
      "epoch": 4.5566558837890626e-05,
      "step": 47780
    },
    {
      "epoch": 4.5566558837890626e-05,
      "step": 47780,
      "training_step_time": 7.571557521820068
    },
    {
      "epoch": 4.556751251220703e-05,
      "model_forward_time": 1.7011268138885498,
      "step": 47781
    },
    {
      "epoch": 4.556751251220703e-05,
      "step": 47781,
      "training_step_time": 6.824662685394287
    },
    {
      "epoch": 4.5568466186523434e-05,
      "model_forward_time": 1.707972526550293,
      "step": 47782
    },
    {
      "epoch": 4.5568466186523434e-05,
      "step": 47782,
      "training_step_time": 6.811563014984131
    },
    {
      "epoch": 4.5569419860839845e-05,
      "model_forward_time": 1.7081298828125,
      "step": 47783
    },
    {
      "epoch": 4.5569419860839845e-05,
      "step": 47783,
      "training_step_time": 6.745374441146851
    },
    {
      "epoch": 4.557037353515625e-05,
      "model_forward_time": 1.702770471572876,
      "step": 47784
    },
    {
      "epoch": 4.557037353515625e-05,
      "step": 47784,
      "training_step_time": 6.857616186141968
    },
    {
      "epoch": 4.557132720947265e-05,
      "model_forward_time": 1.7005336284637451,
      "step": 47785
    },
    {
      "epoch": 4.557132720947265e-05,
      "step": 47785,
      "training_step_time": 6.8390212059021
    },
    {
      "epoch": 4.5572280883789064e-05,
      "model_forward_time": 1.7045443058013916,
      "step": 47786
    },
    {
      "epoch": 4.5572280883789064e-05,
      "step": 47786,
      "training_step_time": 6.838185548782349
    },
    {
      "epoch": 4.557323455810547e-05,
      "model_forward_time": 2.2827186584472656,
      "step": 47787
    },
    {
      "epoch": 4.557323455810547e-05,
      "step": 47787,
      "training_step_time": 9.387084722518921
    },
    {
      "epoch": 4.557418823242187e-05,
      "model_forward_time": 1.6848464012145996,
      "step": 47788
    },
    {
      "epoch": 4.557418823242187e-05,
      "step": 47788,
      "training_step_time": 7.913097620010376
    },
    {
      "epoch": 4.557514190673828e-05,
      "model_forward_time": 1.317039966583252,
      "step": 47789
    },
    {
      "epoch": 4.557514190673828e-05,
      "step": 47789,
      "training_step_time": 7.654892444610596
    },
    {
      "epoch": 4.557609558105469e-05,
      "grad_norm": 0.11952875554561615,
      "learning_rate": 5.7758025363008455e-06,
      "loss": 0.0569,
      "step": 47790
    },
    {
      "epoch": 4.557609558105469e-05,
      "model_forward_time": 1.6839542388916016,
      "step": 47790
    },
    {
      "dynamics_loss_avg": 0.06085825152695179,
      "epoch": 4.557609558105469e-05,
      "step": 47790
    },
    {
      "action_loss_avg": 0.002093964419327676,
      "epoch": 4.557609558105469e-05,
      "step": 47790
    },
    {
      "epoch": 4.557609558105469e-05,
      "step": 47790,
      "training_step_time": 6.886765241622925
    },
    {
      "epoch": 4.557704925537109e-05,
      "model_forward_time": 1.6876587867736816,
      "step": 47791
    },
    {
      "epoch": 4.557704925537109e-05,
      "step": 47791,
      "training_step_time": 6.8596436977386475
    },
    {
      "epoch": 4.55780029296875e-05,
      "model_forward_time": 1.6866750717163086,
      "step": 47792
    },
    {
      "epoch": 4.55780029296875e-05,
      "step": 47792,
      "training_step_time": 6.868623971939087
    },
    {
      "epoch": 4.5578956604003906e-05,
      "model_forward_time": 1.6928315162658691,
      "step": 47793
    },
    {
      "epoch": 4.5578956604003906e-05,
      "step": 47793,
      "training_step_time": 6.824062824249268
    },
    {
      "epoch": 4.557991027832031e-05,
      "model_forward_time": 1.706904649734497,
      "step": 47794
    },
    {
      "epoch": 4.557991027832031e-05,
      "step": 47794,
      "training_step_time": 6.783517360687256
    },
    {
      "epoch": 4.558086395263672e-05,
      "model_forward_time": 1.7099599838256836,
      "step": 47795
    },
    {
      "epoch": 4.558086395263672e-05,
      "step": 47795,
      "training_step_time": 6.854041337966919
    },
    {
      "epoch": 4.5581817626953124e-05,
      "model_forward_time": 1.7075443267822266,
      "step": 47796
    },
    {
      "epoch": 4.5581817626953124e-05,
      "step": 47796,
      "training_step_time": 8.809855699539185
    },
    {
      "epoch": 4.558277130126953e-05,
      "model_forward_time": 2.260556221008301,
      "step": 47797
    },
    {
      "epoch": 4.558277130126953e-05,
      "step": 47797,
      "training_step_time": 8.589675903320312
    },
    {
      "epoch": 4.558372497558594e-05,
      "model_forward_time": 1.70652174949646,
      "step": 47798
    },
    {
      "epoch": 4.558372497558594e-05,
      "step": 47798,
      "training_step_time": 7.658306121826172
    },
    {
      "epoch": 4.558467864990234e-05,
      "model_forward_time": 1.702009677886963,
      "step": 47799
    },
    {
      "epoch": 4.558467864990234e-05,
      "step": 47799,
      "training_step_time": 6.928581714630127
    },
    {
      "epoch": 4.558563232421875e-05,
      "grad_norm": 0.08967269957065582,
      "learning_rate": 5.774169048813134e-06,
      "loss": 0.0559,
      "step": 47800
    },
    {
      "epoch": 4.558563232421875e-05,
      "model_forward_time": 1.705432653427124,
      "step": 47800
    },
    {
      "dynamics_loss_avg": 0.049777747690677644,
      "epoch": 4.558563232421875e-05,
      "step": 47800
    },
    {
      "action_loss_avg": 0.001038628414971754,
      "epoch": 4.558563232421875e-05,
      "step": 47800
    },
    {
      "epoch": 4.558563232421875e-05,
      "step": 47800,
      "training_step_time": 6.899702072143555
    },
    {
      "epoch": 4.558658599853516e-05,
      "model_forward_time": 1.7100441455841064,
      "step": 47801
    },
    {
      "epoch": 4.558658599853516e-05,
      "step": 47801,
      "training_step_time": 6.855032682418823
    },
    {
      "epoch": 4.558753967285156e-05,
      "model_forward_time": 1.708594799041748,
      "step": 47802
    },
    {
      "epoch": 4.558753967285156e-05,
      "step": 47802,
      "training_step_time": 6.8599395751953125
    },
    {
      "epoch": 4.5588493347167966e-05,
      "model_forward_time": 1.706064224243164,
      "step": 47803
    },
    {
      "epoch": 4.5588493347167966e-05,
      "step": 47803,
      "training_step_time": 6.860862731933594
    },
    {
      "epoch": 4.558944702148438e-05,
      "model_forward_time": 1.7016820907592773,
      "step": 47804
    },
    {
      "epoch": 4.558944702148438e-05,
      "step": 47804,
      "training_step_time": 6.888504981994629
    },
    {
      "epoch": 4.559040069580078e-05,
      "model_forward_time": 1.7050833702087402,
      "step": 47805
    },
    {
      "epoch": 4.559040069580078e-05,
      "step": 47805,
      "training_step_time": 9.118358612060547
    },
    {
      "epoch": 4.5591354370117185e-05,
      "model_forward_time": 2.2649855613708496,
      "step": 47806
    },
    {
      "epoch": 4.5591354370117185e-05,
      "step": 47806,
      "training_step_time": 9.947924852371216
    },
    {
      "epoch": 4.5592308044433596e-05,
      "model_forward_time": 1.3376750946044922,
      "step": 47807
    },
    {
      "epoch": 4.5592308044433596e-05,
      "step": 47807,
      "training_step_time": 7.704159498214722
    },
    {
      "epoch": 4.559326171875e-05,
      "model_forward_time": 1.7052247524261475,
      "step": 47808
    },
    {
      "epoch": 4.559326171875e-05,
      "step": 47808,
      "training_step_time": 7.647429466247559
    },
    {
      "epoch": 4.5594215393066404e-05,
      "model_forward_time": 1.708984613418579,
      "step": 47809
    },
    {
      "epoch": 4.5594215393066404e-05,
      "step": 47809,
      "training_step_time": 6.84235405921936
    },
    {
      "epoch": 4.5595169067382815e-05,
      "grad_norm": 0.11403853446245193,
      "learning_rate": 5.77253547666346e-06,
      "loss": 0.0571,
      "step": 47810
    },
    {
      "epoch": 4.5595169067382815e-05,
      "model_forward_time": 1.6989459991455078,
      "step": 47810
    },
    {
      "dynamics_loss_avg": 0.05541607458144426,
      "epoch": 4.5595169067382815e-05,
      "step": 47810
    },
    {
      "action_loss_avg": 0.0020661436079535632,
      "epoch": 4.5595169067382815e-05,
      "step": 47810
    },
    {
      "epoch": 4.5595169067382815e-05,
      "step": 47810,
      "training_step_time": 6.812468767166138
    },
    {
      "epoch": 4.559612274169922e-05,
      "model_forward_time": 1.7087500095367432,
      "step": 47811
    },
    {
      "epoch": 4.559612274169922e-05,
      "step": 47811,
      "training_step_time": 6.789045572280884
    },
    {
      "epoch": 4.559707641601562e-05,
      "model_forward_time": 1.7045221328735352,
      "step": 47812
    },
    {
      "epoch": 4.559707641601562e-05,
      "step": 47812,
      "training_step_time": 6.835195064544678
    },
    {
      "epoch": 4.5598030090332034e-05,
      "model_forward_time": 1.6891045570373535,
      "step": 47813
    },
    {
      "epoch": 4.5598030090332034e-05,
      "step": 47813,
      "training_step_time": 6.901147127151489
    },
    {
      "epoch": 4.559898376464844e-05,
      "model_forward_time": 1.708543300628662,
      "step": 47814
    },
    {
      "epoch": 4.559898376464844e-05,
      "step": 47814,
      "training_step_time": 8.20132327079773
    },
    {
      "epoch": 4.559993743896484e-05,
      "model_forward_time": 2.299201250076294,
      "step": 47815
    },
    {
      "epoch": 4.559993743896484e-05,
      "step": 47815,
      "training_step_time": 11.127301931381226
    },
    {
      "epoch": 4.560089111328125e-05,
      "model_forward_time": 1.70212721824646,
      "step": 47816
    },
    {
      "epoch": 4.560089111328125e-05,
      "step": 47816,
      "training_step_time": 7.677448987960815
    },
    {
      "epoch": 4.5601844787597656e-05,
      "model_forward_time": 1.7050325870513916,
      "step": 47817
    },
    {
      "epoch": 4.5601844787597656e-05,
      "step": 47817,
      "training_step_time": 7.630948066711426
    },
    {
      "epoch": 4.560279846191406e-05,
      "model_forward_time": 1.7068748474121094,
      "step": 47818
    },
    {
      "epoch": 4.560279846191406e-05,
      "step": 47818,
      "training_step_time": 6.857289552688599
    },
    {
      "epoch": 4.560375213623047e-05,
      "model_forward_time": 1.7043166160583496,
      "step": 47819
    },
    {
      "epoch": 4.560375213623047e-05,
      "step": 47819,
      "training_step_time": 6.844101667404175
    },
    {
      "epoch": 4.5604705810546875e-05,
      "grad_norm": 0.11091014742851257,
      "learning_rate": 5.7709018200304656e-06,
      "loss": 0.0562,
      "step": 47820
    },
    {
      "epoch": 4.5604705810546875e-05,
      "model_forward_time": 1.701171875,
      "step": 47820
    },
    {
      "dynamics_loss_avg": 0.0498071551322937,
      "epoch": 4.5604705810546875e-05,
      "step": 47820
    },
    {
      "action_loss_avg": 0.0018968103599036113,
      "epoch": 4.5604705810546875e-05,
      "step": 47820
    },
    {
      "epoch": 4.5604705810546875e-05,
      "step": 47820,
      "training_step_time": 6.861917972564697
    },
    {
      "epoch": 4.560565948486328e-05,
      "model_forward_time": 1.7108042240142822,
      "step": 47821
    },
    {
      "epoch": 4.560565948486328e-05,
      "step": 47821,
      "training_step_time": 6.8607048988342285
    },
    {
      "epoch": 4.560661315917969e-05,
      "model_forward_time": 1.707388162612915,
      "step": 47822
    },
    {
      "epoch": 4.560661315917969e-05,
      "step": 47822,
      "training_step_time": 6.781761884689331
    },
    {
      "epoch": 4.5607566833496094e-05,
      "model_forward_time": 1.7045691013336182,
      "step": 47823
    },
    {
      "epoch": 4.5607566833496094e-05,
      "step": 47823,
      "training_step_time": 8.22581171989441
    },
    {
      "epoch": 4.56085205078125e-05,
      "model_forward_time": 2.255674362182617,
      "step": 47824
    },
    {
      "epoch": 4.56085205078125e-05,
      "step": 47824,
      "training_step_time": 8.643271446228027
    },
    {
      "epoch": 4.560947418212891e-05,
      "model_forward_time": 1.7145402431488037,
      "step": 47825
    },
    {
      "epoch": 4.560947418212891e-05,
      "step": 47825,
      "training_step_time": 7.8844616413116455
    },
    {
      "epoch": 4.561042785644531e-05,
      "model_forward_time": 1.6980116367340088,
      "step": 47826
    },
    {
      "epoch": 4.561042785644531e-05,
      "step": 47826,
      "training_step_time": 7.732895135879517
    },
    {
      "epoch": 4.561138153076172e-05,
      "model_forward_time": 1.7017343044281006,
      "step": 47827
    },
    {
      "epoch": 4.561138153076172e-05,
      "step": 47827,
      "training_step_time": 7.256735801696777
    },
    {
      "epoch": 4.561233520507813e-05,
      "model_forward_time": 1.7052640914916992,
      "step": 47828
    },
    {
      "epoch": 4.561233520507813e-05,
      "step": 47828,
      "training_step_time": 6.918805837631226
    },
    {
      "epoch": 4.561328887939453e-05,
      "model_forward_time": 1.7051446437835693,
      "step": 47829
    },
    {
      "epoch": 4.561328887939453e-05,
      "step": 47829,
      "training_step_time": 6.946718454360962
    },
    {
      "epoch": 4.5614242553710936e-05,
      "grad_norm": 0.11579058319330215,
      "learning_rate": 5.769268079092809e-06,
      "loss": 0.0548,
      "step": 47830
    },
    {
      "epoch": 4.5614242553710936e-05,
      "model_forward_time": 1.3255109786987305,
      "step": 47830
    },
    {
      "dynamics_loss_avg": 0.03703635558485985,
      "epoch": 4.5614242553710936e-05,
      "step": 47830
    },
    {
      "action_loss_avg": 0.0014282273186836392,
      "epoch": 4.5614242553710936e-05,
      "step": 47830
    },
    {
      "epoch": 4.5614242553710936e-05,
      "step": 47830,
      "training_step_time": 6.944056749343872
    },
    {
      "epoch": 4.561519622802735e-05,
      "model_forward_time": 1.7007989883422852,
      "step": 47831
    },
    {
      "epoch": 4.561519622802735e-05,
      "step": 47831,
      "training_step_time": 6.964524745941162
    },
    {
      "epoch": 4.561614990234375e-05,
      "model_forward_time": 1.707566261291504,
      "step": 47832
    },
    {
      "epoch": 4.561614990234375e-05,
      "step": 47832,
      "training_step_time": 7.7244884967803955
    },
    {
      "epoch": 4.5617103576660155e-05,
      "model_forward_time": 1.7150471210479736,
      "step": 47833
    },
    {
      "epoch": 4.5617103576660155e-05,
      "step": 47833,
      "training_step_time": 11.288593292236328
    },
    {
      "epoch": 4.5618057250976566e-05,
      "model_forward_time": 2.2740163803100586,
      "step": 47834
    },
    {
      "epoch": 4.5618057250976566e-05,
      "step": 47834,
      "training_step_time": 9.084393501281738
    },
    {
      "epoch": 4.561901092529297e-05,
      "model_forward_time": 1.70357346534729,
      "step": 47835
    },
    {
      "epoch": 4.561901092529297e-05,
      "step": 47835,
      "training_step_time": 7.766504287719727
    },
    {
      "epoch": 4.5619964599609374e-05,
      "model_forward_time": 1.7070138454437256,
      "step": 47836
    },
    {
      "epoch": 4.5619964599609374e-05,
      "step": 47836,
      "training_step_time": 7.26094913482666
    },
    {
      "epoch": 4.5620918273925784e-05,
      "model_forward_time": 1.7047746181488037,
      "step": 47837
    },
    {
      "epoch": 4.5620918273925784e-05,
      "step": 47837,
      "training_step_time": 7.020207405090332
    },
    {
      "epoch": 4.562187194824219e-05,
      "model_forward_time": 1.6950812339782715,
      "step": 47838
    },
    {
      "epoch": 4.562187194824219e-05,
      "step": 47838,
      "training_step_time": 6.955091953277588
    },
    {
      "epoch": 4.562282562255859e-05,
      "model_forward_time": 1.7043612003326416,
      "step": 47839
    },
    {
      "epoch": 4.562282562255859e-05,
      "step": 47839,
      "training_step_time": 6.976728916168213
    },
    {
      "epoch": 4.5623779296875e-05,
      "grad_norm": 0.08857216686010361,
      "learning_rate": 5.767634254029151e-06,
      "loss": 0.0578,
      "step": 47840
    },
    {
      "epoch": 4.5623779296875e-05,
      "model_forward_time": 1.322908878326416,
      "step": 47840
    },
    {
      "dynamics_loss_avg": 0.052503727935254575,
      "epoch": 4.5623779296875e-05,
      "step": 47840
    },
    {
      "action_loss_avg": 0.002352659543976188,
      "epoch": 4.5623779296875e-05,
      "step": 47840
    },
    {
      "epoch": 4.5623779296875e-05,
      "step": 47840,
      "training_step_time": 6.895745277404785
    },
    {
      "epoch": 4.562473297119141e-05,
      "model_forward_time": 1.702333927154541,
      "step": 47841
    },
    {
      "epoch": 4.562473297119141e-05,
      "step": 47841,
      "training_step_time": 7.588782072067261
    },
    {
      "epoch": 4.562568664550781e-05,
      "model_forward_time": 1.716581106185913,
      "step": 47842
    },
    {
      "epoch": 4.562568664550781e-05,
      "step": 47842,
      "training_step_time": 7.789297819137573
    },
    {
      "epoch": 4.562664031982422e-05,
      "model_forward_time": 2.296741247177124,
      "step": 47843
    },
    {
      "epoch": 4.562664031982422e-05,
      "step": 47843,
      "training_step_time": 9.159484624862671
    },
    {
      "epoch": 4.5627593994140626e-05,
      "model_forward_time": 1.7035727500915527,
      "step": 47844
    },
    {
      "epoch": 4.5627593994140626e-05,
      "step": 47844,
      "training_step_time": 7.704843997955322
    },
    {
      "epoch": 4.562854766845703e-05,
      "model_forward_time": 1.7067649364471436,
      "step": 47845
    },
    {
      "epoch": 4.562854766845703e-05,
      "step": 47845,
      "training_step_time": 7.659087181091309
    },
    {
      "epoch": 4.5629501342773434e-05,
      "model_forward_time": 1.706003189086914,
      "step": 47846
    },
    {
      "epoch": 4.5629501342773434e-05,
      "step": 47846,
      "training_step_time": 6.909996271133423
    },
    {
      "epoch": 4.5630455017089845e-05,
      "model_forward_time": 1.7034835815429688,
      "step": 47847
    },
    {
      "epoch": 4.5630455017089845e-05,
      "step": 47847,
      "training_step_time": 6.872572183609009
    },
    {
      "epoch": 4.563140869140625e-05,
      "model_forward_time": 1.706037998199463,
      "step": 47848
    },
    {
      "epoch": 4.563140869140625e-05,
      "step": 47848,
      "training_step_time": 6.931044578552246
    },
    {
      "epoch": 4.563236236572265e-05,
      "model_forward_time": 1.7023632526397705,
      "step": 47849
    },
    {
      "epoch": 4.563236236572265e-05,
      "step": 47849,
      "training_step_time": 6.900130748748779
    },
    {
      "epoch": 4.5633316040039064e-05,
      "grad_norm": 0.12849432229995728,
      "learning_rate": 5.766000345018166e-06,
      "loss": 0.0577,
      "step": 47850
    },
    {
      "epoch": 4.5633316040039064e-05,
      "model_forward_time": 1.702012062072754,
      "step": 47850
    },
    {
      "dynamics_loss_avg": 0.055363507568836214,
      "epoch": 4.5633316040039064e-05,
      "step": 47850
    },
    {
      "action_loss_avg": 0.0014282001939136535,
      "epoch": 4.5633316040039064e-05,
      "step": 47850
    },
    {
      "epoch": 4.5633316040039064e-05,
      "step": 47850,
      "training_step_time": 7.408761978149414
    },
    {
      "epoch": 4.563426971435547e-05,
      "model_forward_time": 1.7156248092651367,
      "step": 47851
    },
    {
      "epoch": 4.563426971435547e-05,
      "step": 47851,
      "training_step_time": 8.557094812393188
    },
    {
      "epoch": 4.563522338867187e-05,
      "model_forward_time": 2.279738664627075,
      "step": 47852
    },
    {
      "epoch": 4.563522338867187e-05,
      "step": 47852,
      "training_step_time": 10.44471549987793
    },
    {
      "epoch": 4.563617706298828e-05,
      "model_forward_time": 1.7099380493164062,
      "step": 47853
    },
    {
      "epoch": 4.563617706298828e-05,
      "step": 47853,
      "training_step_time": 7.75411319732666
    },
    {
      "epoch": 4.563713073730469e-05,
      "model_forward_time": 1.7064993381500244,
      "step": 47854
    },
    {
      "epoch": 4.563713073730469e-05,
      "step": 47854,
      "training_step_time": 8.660514831542969
    },
    {
      "epoch": 4.563808441162109e-05,
      "model_forward_time": 1.710832118988037,
      "step": 47855
    },
    {
      "epoch": 4.563808441162109e-05,
      "step": 47855,
      "training_step_time": 6.896557092666626
    },
    {
      "epoch": 4.56390380859375e-05,
      "model_forward_time": 1.6857459545135498,
      "step": 47856
    },
    {
      "epoch": 4.56390380859375e-05,
      "step": 47856,
      "training_step_time": 6.787698984146118
    },
    {
      "epoch": 4.5639991760253906e-05,
      "model_forward_time": 1.7052111625671387,
      "step": 47857
    },
    {
      "epoch": 4.5639991760253906e-05,
      "step": 47857,
      "training_step_time": 6.910074710845947
    },
    {
      "epoch": 4.564094543457031e-05,
      "model_forward_time": 1.7113745212554932,
      "step": 47858
    },
    {
      "epoch": 4.564094543457031e-05,
      "step": 47858,
      "training_step_time": 6.9049177169799805
    },
    {
      "epoch": 4.564189910888672e-05,
      "model_forward_time": 1.7056925296783447,
      "step": 47859
    },
    {
      "epoch": 4.564189910888672e-05,
      "step": 47859,
      "training_step_time": 7.504264831542969
    },
    {
      "epoch": 4.5642852783203124e-05,
      "grad_norm": 0.1024773046374321,
      "learning_rate": 5.764366352238534e-06,
      "loss": 0.0624,
      "step": 47860
    },
    {
      "epoch": 4.5642852783203124e-05,
      "model_forward_time": 1.7094855308532715,
      "step": 47860
    },
    {
      "dynamics_loss_avg": 0.07730070687830448,
      "epoch": 4.5642852783203124e-05,
      "step": 47860
    },
    {
      "action_loss_avg": 0.0022611434396822006,
      "epoch": 4.5642852783203124e-05,
      "step": 47860
    },
    {
      "epoch": 4.5642852783203124e-05,
      "step": 47860,
      "training_step_time": 9.27631950378418
    },
    {
      "epoch": 4.564380645751953e-05,
      "model_forward_time": 2.258176803588867,
      "step": 47861
    },
    {
      "epoch": 4.564380645751953e-05,
      "step": 47861,
      "training_step_time": 9.904679775238037
    },
    {
      "epoch": 4.564476013183594e-05,
      "model_forward_time": 1.7165675163269043,
      "step": 47862
    },
    {
      "epoch": 4.564476013183594e-05,
      "step": 47862,
      "training_step_time": 9.229215145111084
    },
    {
      "epoch": 4.564571380615234e-05,
      "model_forward_time": 1.7060737609863281,
      "step": 47863
    },
    {
      "epoch": 4.564571380615234e-05,
      "step": 47863,
      "training_step_time": 7.6790688037872314
    },
    {
      "epoch": 4.564666748046875e-05,
      "model_forward_time": 1.7107782363891602,
      "step": 47864
    },
    {
      "epoch": 4.564666748046875e-05,
      "step": 47864,
      "training_step_time": 7.264016151428223
    },
    {
      "epoch": 4.564762115478516e-05,
      "model_forward_time": 1.7033202648162842,
      "step": 47865
    },
    {
      "epoch": 4.564762115478516e-05,
      "step": 47865,
      "training_step_time": 6.823834657669067
    },
    {
      "epoch": 4.564857482910156e-05,
      "model_forward_time": 1.7014820575714111,
      "step": 47866
    },
    {
      "epoch": 4.564857482910156e-05,
      "step": 47866,
      "training_step_time": 6.8607497215271
    },
    {
      "epoch": 4.5649528503417966e-05,
      "model_forward_time": 1.7050650119781494,
      "step": 47867
    },
    {
      "epoch": 4.5649528503417966e-05,
      "step": 47867,
      "training_step_time": 6.855982542037964
    },
    {
      "epoch": 4.565048217773438e-05,
      "model_forward_time": 1.71000075340271,
      "step": 47868
    },
    {
      "epoch": 4.565048217773438e-05,
      "step": 47868,
      "training_step_time": 7.4111950397491455
    },
    {
      "epoch": 4.565143585205078e-05,
      "model_forward_time": 1.7074456214904785,
      "step": 47869
    },
    {
      "epoch": 4.565143585205078e-05,
      "step": 47869,
      "training_step_time": 8.503461122512817
    },
    {
      "epoch": 4.5652389526367185e-05,
      "grad_norm": 0.10200484097003937,
      "learning_rate": 5.7627322758689476e-06,
      "loss": 0.0578,
      "step": 47870
    },
    {
      "epoch": 4.5652389526367185e-05,
      "model_forward_time": 1.7205724716186523,
      "step": 47870
    },
    {
      "dynamics_loss_avg": 0.05885675884783268,
      "epoch": 4.5652389526367185e-05,
      "step": 47870
    },
    {
      "action_loss_avg": 0.001358816714491695,
      "epoch": 4.5652389526367185e-05,
      "step": 47870
    },
    {
      "epoch": 4.5652389526367185e-05,
      "step": 47870,
      "training_step_time": 8.684197425842285
    },
    {
      "epoch": 4.5653343200683596e-05,
      "model_forward_time": 1.721658706665039,
      "step": 47871
    },
    {
      "epoch": 4.5653343200683596e-05,
      "step": 47871,
      "training_step_time": 9.155025959014893
    },
    {
      "epoch": 4.5654296875e-05,
      "model_forward_time": 1.7068991661071777,
      "step": 47872
    },
    {
      "epoch": 4.5654296875e-05,
      "step": 47872,
      "training_step_time": 7.6678080558776855
    },
    {
      "epoch": 4.5655250549316404e-05,
      "model_forward_time": 1.7054405212402344,
      "step": 47873
    },
    {
      "epoch": 4.5655250549316404e-05,
      "step": 47873,
      "training_step_time": 7.255761623382568
    },
    {
      "epoch": 4.5656204223632815e-05,
      "model_forward_time": 1.7054038047790527,
      "step": 47874
    },
    {
      "epoch": 4.5656204223632815e-05,
      "step": 47874,
      "training_step_time": 6.870289087295532
    },
    {
      "epoch": 4.565715789794922e-05,
      "model_forward_time": 1.7088208198547363,
      "step": 47875
    },
    {
      "epoch": 4.565715789794922e-05,
      "step": 47875,
      "training_step_time": 6.837386608123779
    },
    {
      "epoch": 4.565811157226562e-05,
      "model_forward_time": 1.707080602645874,
      "step": 47876
    },
    {
      "epoch": 4.565811157226562e-05,
      "step": 47876,
      "training_step_time": 6.838855743408203
    },
    {
      "epoch": 4.5659065246582034e-05,
      "model_forward_time": 1.7053000926971436,
      "step": 47877
    },
    {
      "epoch": 4.5659065246582034e-05,
      "step": 47877,
      "training_step_time": 7.496600866317749
    },
    {
      "epoch": 4.566001892089844e-05,
      "model_forward_time": 1.7080776691436768,
      "step": 47878
    },
    {
      "epoch": 4.566001892089844e-05,
      "step": 47878,
      "training_step_time": 7.661383390426636
    },
    {
      "epoch": 4.566097259521484e-05,
      "model_forward_time": 1.715965747833252,
      "step": 47879
    },
    {
      "epoch": 4.566097259521484e-05,
      "step": 47879,
      "training_step_time": 7.714253902435303
    },
    {
      "epoch": 4.566192626953125e-05,
      "grad_norm": 0.16227927803993225,
      "learning_rate": 5.761098116088105e-06,
      "loss": 0.0589,
      "step": 47880
    },
    {
      "epoch": 4.566192626953125e-05,
      "model_forward_time": 2.238767385482788,
      "step": 47880
    },
    {
      "dynamics_loss_avg": 0.057449909206479785,
      "epoch": 4.566192626953125e-05,
      "step": 47880
    },
    {
      "action_loss_avg": 0.0017150670260889455,
      "epoch": 4.566192626953125e-05,
      "step": 47880
    },
    {
      "epoch": 4.566192626953125e-05,
      "step": 47880,
      "training_step_time": 7.8957507610321045
    },
    {
      "epoch": 4.5662879943847656e-05,
      "model_forward_time": 1.698866605758667,
      "step": 47881
    },
    {
      "epoch": 4.5662879943847656e-05,
      "step": 47881,
      "training_step_time": 8.32401418685913
    },
    {
      "epoch": 4.566383361816406e-05,
      "model_forward_time": 1.7137129306793213,
      "step": 47882
    },
    {
      "epoch": 4.566383361816406e-05,
      "step": 47882,
      "training_step_time": 7.331139802932739
    },
    {
      "epoch": 4.566478729248047e-05,
      "model_forward_time": 1.712123155593872,
      "step": 47883
    },
    {
      "epoch": 4.566478729248047e-05,
      "step": 47883,
      "training_step_time": 6.889056444168091
    },
    {
      "epoch": 4.5665740966796875e-05,
      "model_forward_time": 1.7205350399017334,
      "step": 47884
    },
    {
      "epoch": 4.5665740966796875e-05,
      "step": 47884,
      "training_step_time": 6.910767316818237
    },
    {
      "epoch": 4.566669464111328e-05,
      "model_forward_time": 1.712249994277954,
      "step": 47885
    },
    {
      "epoch": 4.566669464111328e-05,
      "step": 47885,
      "training_step_time": 6.904929161071777
    },
    {
      "epoch": 4.566764831542969e-05,
      "model_forward_time": 1.711317777633667,
      "step": 47886
    },
    {
      "epoch": 4.566764831542969e-05,
      "step": 47886,
      "training_step_time": 7.527334928512573
    },
    {
      "epoch": 4.5668601989746094e-05,
      "model_forward_time": 1.7145214080810547,
      "step": 47887
    },
    {
      "epoch": 4.5668601989746094e-05,
      "step": 47887,
      "training_step_time": 8.395314455032349
    },
    {
      "epoch": 4.56695556640625e-05,
      "model_forward_time": 1.7186641693115234,
      "step": 47888
    },
    {
      "epoch": 4.56695556640625e-05,
      "step": 47888,
      "training_step_time": 9.862133502960205
    },
    {
      "epoch": 4.567050933837891e-05,
      "model_forward_time": 2.2732560634613037,
      "step": 47889
    },
    {
      "epoch": 4.567050933837891e-05,
      "step": 47889,
      "training_step_time": 10.943432569503784
    },
    {
      "epoch": 4.567146301269531e-05,
      "grad_norm": 0.18110743165016174,
      "learning_rate": 5.759463873074718e-06,
      "loss": 0.0597,
      "step": 47890
    },
    {
      "epoch": 4.567146301269531e-05,
      "model_forward_time": 1.696589469909668,
      "step": 47890
    },
    {
      "dynamics_loss_avg": 0.07153620086610317,
      "epoch": 4.567146301269531e-05,
      "step": 47890
    },
    {
      "action_loss_avg": 0.0018047222605673597,
      "epoch": 4.567146301269531e-05,
      "step": 47890
    },
    {
      "epoch": 4.567146301269531e-05,
      "step": 47890,
      "training_step_time": 8.24825143814087
    },
    {
      "epoch": 4.567241668701172e-05,
      "model_forward_time": 1.7061069011688232,
      "step": 47891
    },
    {
      "epoch": 4.567241668701172e-05,
      "step": 47891,
      "training_step_time": 7.070263862609863
    },
    {
      "epoch": 4.567337036132813e-05,
      "model_forward_time": 1.7017452716827393,
      "step": 47892
    },
    {
      "epoch": 4.567337036132813e-05,
      "step": 47892,
      "training_step_time": 6.879147052764893
    },
    {
      "epoch": 4.567432403564453e-05,
      "model_forward_time": 1.7007420063018799,
      "step": 47893
    },
    {
      "epoch": 4.567432403564453e-05,
      "step": 47893,
      "training_step_time": 6.819005966186523
    },
    {
      "epoch": 4.5675277709960936e-05,
      "model_forward_time": 1.698268175125122,
      "step": 47894
    },
    {
      "epoch": 4.5675277709960936e-05,
      "step": 47894,
      "training_step_time": 6.751802682876587
    },
    {
      "epoch": 4.567623138427735e-05,
      "model_forward_time": 1.6997754573822021,
      "step": 47895
    },
    {
      "epoch": 4.567623138427735e-05,
      "step": 47895,
      "training_step_time": 7.401049613952637
    },
    {
      "epoch": 4.567718505859375e-05,
      "model_forward_time": 1.3242161273956299,
      "step": 47896
    },
    {
      "epoch": 4.567718505859375e-05,
      "step": 47896,
      "training_step_time": 7.555212736129761
    },
    {
      "epoch": 4.5678138732910155e-05,
      "model_forward_time": 1.7033777236938477,
      "step": 47897
    },
    {
      "epoch": 4.5678138732910155e-05,
      "step": 47897,
      "training_step_time": 7.749521732330322
    },
    {
      "epoch": 4.5679092407226566e-05,
      "model_forward_time": 2.270360231399536,
      "step": 47898
    },
    {
      "epoch": 4.5679092407226566e-05,
      "step": 47898,
      "training_step_time": 10.522506475448608
    },
    {
      "epoch": 4.568004608154297e-05,
      "model_forward_time": 1.701249122619629,
      "step": 47899
    },
    {
      "epoch": 4.568004608154297e-05,
      "step": 47899,
      "training_step_time": 7.724754095077515
    },
    {
      "epoch": 4.5680999755859374e-05,
      "grad_norm": 0.10566641390323639,
      "learning_rate": 5.757829547007504e-06,
      "loss": 0.0598,
      "step": 47900
    },
    {
      "epoch": 4.5680999755859374e-05,
      "model_forward_time": 1.6929197311401367,
      "step": 47900
    },
    {
      "dynamics_loss_avg": 0.06532715149223804,
      "epoch": 4.5680999755859374e-05,
      "step": 47900
    },
    {
      "action_loss_avg": 0.0013308695575688033,
      "epoch": 4.5680999755859374e-05,
      "step": 47900
    },
    {
      "epoch": 4.5680999755859374e-05,
      "step": 47900,
      "training_step_time": 7.764914035797119
    },
    {
      "epoch": 4.5681953430175784e-05,
      "model_forward_time": 1.6943504810333252,
      "step": 47901
    },
    {
      "epoch": 4.5681953430175784e-05,
      "step": 47901,
      "training_step_time": 7.290421485900879
    },
    {
      "epoch": 4.568290710449219e-05,
      "model_forward_time": 1.7023451328277588,
      "step": 47902
    },
    {
      "epoch": 4.568290710449219e-05,
      "step": 47902,
      "training_step_time": 6.781961679458618
    },
    {
      "epoch": 4.568386077880859e-05,
      "model_forward_time": 1.706164836883545,
      "step": 47903
    },
    {
      "epoch": 4.568386077880859e-05,
      "step": 47903,
      "training_step_time": 6.7579779624938965
    },
    {
      "epoch": 4.5684814453125e-05,
      "model_forward_time": 1.6991183757781982,
      "step": 47904
    },
    {
      "epoch": 4.5684814453125e-05,
      "step": 47904,
      "training_step_time": 6.675997018814087
    },
    {
      "epoch": 4.568576812744141e-05,
      "model_forward_time": 1.7010958194732666,
      "step": 47905
    },
    {
      "epoch": 4.568576812744141e-05,
      "step": 47905,
      "training_step_time": 7.61400842666626
    },
    {
      "epoch": 4.568672180175781e-05,
      "model_forward_time": 1.7026934623718262,
      "step": 47906
    },
    {
      "epoch": 4.568672180175781e-05,
      "step": 47906,
      "training_step_time": 9.31115198135376
    },
    {
      "epoch": 4.5687675476074215e-05,
      "model_forward_time": 2.2871813774108887,
      "step": 47907
    },
    {
      "epoch": 4.5687675476074215e-05,
      "step": 47907,
      "training_step_time": 9.916258573532104
    },
    {
      "epoch": 4.5688629150390626e-05,
      "model_forward_time": 1.7218234539031982,
      "step": 47908
    },
    {
      "epoch": 4.5688629150390626e-05,
      "step": 47908,
      "training_step_time": 8.385565280914307
    },
    {
      "epoch": 4.568958282470703e-05,
      "model_forward_time": 1.698044776916504,
      "step": 47909
    },
    {
      "epoch": 4.568958282470703e-05,
      "step": 47909,
      "training_step_time": 7.760461330413818
    },
    {
      "epoch": 4.5690536499023434e-05,
      "grad_norm": 0.12622226774692535,
      "learning_rate": 5.756195138065189e-06,
      "loss": 0.0574,
      "step": 47910
    },
    {
      "epoch": 4.5690536499023434e-05,
      "model_forward_time": 1.696812391281128,
      "step": 47910
    },
    {
      "dynamics_loss_avg": 0.0567359758540988,
      "epoch": 4.5690536499023434e-05,
      "step": 47910
    },
    {
      "action_loss_avg": 0.0012501776160206645,
      "epoch": 4.5690536499023434e-05,
      "step": 47910
    },
    {
      "epoch": 4.5690536499023434e-05,
      "step": 47910,
      "training_step_time": 7.3642778396606445
    },
    {
      "epoch": 4.5691490173339845e-05,
      "model_forward_time": 1.7056879997253418,
      "step": 47911
    },
    {
      "epoch": 4.5691490173339845e-05,
      "step": 47911,
      "training_step_time": 6.832378625869751
    },
    {
      "epoch": 4.569244384765625e-05,
      "model_forward_time": 1.7001657485961914,
      "step": 47912
    },
    {
      "epoch": 4.569244384765625e-05,
      "step": 47912,
      "training_step_time": 6.851116418838501
    },
    {
      "epoch": 4.569339752197265e-05,
      "model_forward_time": 1.7000136375427246,
      "step": 47913
    },
    {
      "epoch": 4.569339752197265e-05,
      "step": 47913,
      "training_step_time": 6.842404127120972
    },
    {
      "epoch": 4.5694351196289064e-05,
      "model_forward_time": 1.7056658267974854,
      "step": 47914
    },
    {
      "epoch": 4.5694351196289064e-05,
      "step": 47914,
      "training_step_time": 7.6935975551605225
    },
    {
      "epoch": 4.569530487060547e-05,
      "model_forward_time": 1.700655221939087,
      "step": 47915
    },
    {
      "epoch": 4.569530487060547e-05,
      "step": 47915,
      "training_step_time": 8.777177333831787
    },
    {
      "epoch": 4.569625854492187e-05,
      "model_forward_time": 1.6933925151824951,
      "step": 47916
    },
    {
      "epoch": 4.569625854492187e-05,
      "step": 47916,
      "training_step_time": 9.325364589691162
    },
    {
      "epoch": 4.569721221923828e-05,
      "model_forward_time": 1.7185862064361572,
      "step": 47917
    },
    {
      "epoch": 4.569721221923828e-05,
      "step": 47917,
      "training_step_time": 9.563199758529663
    },
    {
      "epoch": 4.569816589355469e-05,
      "model_forward_time": 1.7058100700378418,
      "step": 47918
    },
    {
      "epoch": 4.569816589355469e-05,
      "step": 47918,
      "training_step_time": 8.738789796829224
    },
    {
      "epoch": 4.569911956787109e-05,
      "model_forward_time": 1.6997191905975342,
      "step": 47919
    },
    {
      "epoch": 4.569911956787109e-05,
      "step": 47919,
      "training_step_time": 6.832362651824951
    },
    {
      "epoch": 4.57000732421875e-05,
      "grad_norm": 0.11056467890739441,
      "learning_rate": 5.754560646426511e-06,
      "loss": 0.058,
      "step": 47920
    },
    {
      "epoch": 4.57000732421875e-05,
      "model_forward_time": 1.701960563659668,
      "step": 47920
    },
    {
      "dynamics_loss_avg": 0.06672976799309253,
      "epoch": 4.57000732421875e-05,
      "step": 47920
    },
    {
      "action_loss_avg": 0.0018052677332889289,
      "epoch": 4.57000732421875e-05,
      "step": 47920
    },
    {
      "epoch": 4.57000732421875e-05,
      "step": 47920,
      "training_step_time": 7.316161394119263
    },
    {
      "epoch": 4.5701026916503906e-05,
      "model_forward_time": 1.696948766708374,
      "step": 47921
    },
    {
      "epoch": 4.5701026916503906e-05,
      "step": 47921,
      "training_step_time": 6.767485618591309
    },
    {
      "epoch": 4.570198059082031e-05,
      "model_forward_time": 1.7015705108642578,
      "step": 47922
    },
    {
      "epoch": 4.570198059082031e-05,
      "step": 47922,
      "training_step_time": 6.764296770095825
    },
    {
      "epoch": 4.570293426513672e-05,
      "model_forward_time": 1.700979471206665,
      "step": 47923
    },
    {
      "epoch": 4.570293426513672e-05,
      "step": 47923,
      "training_step_time": 7.586161851882935
    },
    {
      "epoch": 4.5703887939453125e-05,
      "model_forward_time": 1.6995396614074707,
      "step": 47924
    },
    {
      "epoch": 4.5703887939453125e-05,
      "step": 47924,
      "training_step_time": 8.646106719970703
    },
    {
      "epoch": 4.570484161376953e-05,
      "model_forward_time": 1.7066295146942139,
      "step": 47925
    },
    {
      "epoch": 4.570484161376953e-05,
      "step": 47925,
      "training_step_time": 8.456841945648193
    },
    {
      "epoch": 4.570579528808594e-05,
      "model_forward_time": 2.2821807861328125,
      "step": 47926
    },
    {
      "epoch": 4.570579528808594e-05,
      "step": 47926,
      "training_step_time": 7.8631064891815186
    },
    {
      "epoch": 4.570674896240234e-05,
      "model_forward_time": 1.706261396408081,
      "step": 47927
    },
    {
      "epoch": 4.570674896240234e-05,
      "step": 47927,
      "training_step_time": 9.151243209838867
    },
    {
      "epoch": 4.570770263671875e-05,
      "model_forward_time": 1.7139861583709717,
      "step": 47928
    },
    {
      "epoch": 4.570770263671875e-05,
      "step": 47928,
      "training_step_time": 7.7433412075042725
    },
    {
      "epoch": 4.570865631103516e-05,
      "model_forward_time": 1.7268524169921875,
      "step": 47929
    },
    {
      "epoch": 4.570865631103516e-05,
      "step": 47929,
      "training_step_time": 7.37143087387085
    },
    {
      "epoch": 4.570960998535156e-05,
      "grad_norm": 0.12943463027477264,
      "learning_rate": 5.752926072270216e-06,
      "loss": 0.0569,
      "step": 47930
    },
    {
      "epoch": 4.570960998535156e-05,
      "model_forward_time": 1.7240474224090576,
      "step": 47930
    },
    {
      "dynamics_loss_avg": 0.06577750239521266,
      "epoch": 4.570960998535156e-05,
      "step": 47930
    },
    {
      "action_loss_avg": 0.002114178758347407,
      "epoch": 4.570960998535156e-05,
      "step": 47930
    },
    {
      "epoch": 4.570960998535156e-05,
      "step": 47930,
      "training_step_time": 6.911091089248657
    },
    {
      "epoch": 4.5710563659667966e-05,
      "model_forward_time": 1.7264273166656494,
      "step": 47931
    },
    {
      "epoch": 4.5710563659667966e-05,
      "step": 47931,
      "training_step_time": 6.85179328918457
    },
    {
      "epoch": 4.571151733398438e-05,
      "model_forward_time": 1.7303197383880615,
      "step": 47932
    },
    {
      "epoch": 4.571151733398438e-05,
      "step": 47932,
      "training_step_time": 8.303390979766846
    },
    {
      "epoch": 4.571247100830078e-05,
      "model_forward_time": 1.7278740406036377,
      "step": 47933
    },
    {
      "epoch": 4.571247100830078e-05,
      "step": 47933,
      "training_step_time": 8.348299741744995
    },
    {
      "epoch": 4.5713424682617185e-05,
      "model_forward_time": 1.732715129852295,
      "step": 47934
    },
    {
      "epoch": 4.5713424682617185e-05,
      "step": 47934,
      "training_step_time": 8.479958772659302
    },
    {
      "epoch": 4.5714378356933596e-05,
      "model_forward_time": 2.344573736190796,
      "step": 47935
    },
    {
      "epoch": 4.5714378356933596e-05,
      "step": 47935,
      "training_step_time": 8.28480339050293
    },
    {
      "epoch": 4.571533203125e-05,
      "model_forward_time": 1.7327990531921387,
      "step": 47936
    },
    {
      "epoch": 4.571533203125e-05,
      "step": 47936,
      "training_step_time": 8.229999542236328
    },
    {
      "epoch": 4.5716285705566404e-05,
      "model_forward_time": 1.7300877571105957,
      "step": 47937
    },
    {
      "epoch": 4.5716285705566404e-05,
      "step": 47937,
      "training_step_time": 7.769981384277344
    },
    {
      "epoch": 4.5717239379882815e-05,
      "model_forward_time": 1.7265751361846924,
      "step": 47938
    },
    {
      "epoch": 4.5717239379882815e-05,
      "step": 47938,
      "training_step_time": 7.330275297164917
    },
    {
      "epoch": 4.571819305419922e-05,
      "model_forward_time": 1.7269344329833984,
      "step": 47939
    },
    {
      "epoch": 4.571819305419922e-05,
      "step": 47939,
      "training_step_time": 6.919482469558716
    },
    {
      "epoch": 4.571914672851562e-05,
      "grad_norm": 0.12767359614372253,
      "learning_rate": 5.7512914157750565e-06,
      "loss": 0.0589,
      "step": 47940
    },
    {
      "epoch": 4.571914672851562e-05,
      "model_forward_time": 1.701894760131836,
      "step": 47940
    },
    {
      "dynamics_loss_avg": 0.047343474067747596,
      "epoch": 4.571914672851562e-05,
      "step": 47940
    },
    {
      "action_loss_avg": 0.0007851371250580996,
      "epoch": 4.571914672851562e-05,
      "step": 47940
    },
    {
      "epoch": 4.571914672851562e-05,
      "step": 47940,
      "training_step_time": 6.856387376785278
    },
    {
      "epoch": 4.5720100402832034e-05,
      "model_forward_time": 1.725818157196045,
      "step": 47941
    },
    {
      "epoch": 4.5720100402832034e-05,
      "step": 47941,
      "training_step_time": 7.517390966415405
    },
    {
      "epoch": 4.572105407714844e-05,
      "model_forward_time": 1.3483803272247314,
      "step": 47942
    },
    {
      "epoch": 4.572105407714844e-05,
      "step": 47942,
      "training_step_time": 7.762604236602783
    },
    {
      "epoch": 4.572200775146484e-05,
      "model_forward_time": 1.7311198711395264,
      "step": 47943
    },
    {
      "epoch": 4.572200775146484e-05,
      "step": 47943,
      "training_step_time": 8.863771915435791
    },
    {
      "epoch": 4.572296142578125e-05,
      "model_forward_time": 2.277966260910034,
      "step": 47944
    },
    {
      "epoch": 4.572296142578125e-05,
      "step": 47944,
      "training_step_time": 9.164486408233643
    },
    {
      "epoch": 4.5723915100097657e-05,
      "model_forward_time": 1.7241675853729248,
      "step": 47945
    },
    {
      "epoch": 4.5723915100097657e-05,
      "step": 47945,
      "training_step_time": 8.547260999679565
    },
    {
      "epoch": 4.572486877441406e-05,
      "model_forward_time": 1.720327615737915,
      "step": 47946
    },
    {
      "epoch": 4.572486877441406e-05,
      "step": 47946,
      "training_step_time": 8.90843915939331
    },
    {
      "epoch": 4.572582244873047e-05,
      "model_forward_time": 1.725205659866333,
      "step": 47947
    },
    {
      "epoch": 4.572582244873047e-05,
      "step": 47947,
      "training_step_time": 8.58652949333191
    },
    {
      "epoch": 4.5726776123046875e-05,
      "model_forward_time": 1.7177493572235107,
      "step": 47948
    },
    {
      "epoch": 4.5726776123046875e-05,
      "step": 47948,
      "training_step_time": 6.910639762878418
    },
    {
      "epoch": 4.572772979736328e-05,
      "model_forward_time": 1.7176237106323242,
      "step": 47949
    },
    {
      "epoch": 4.572772979736328e-05,
      "step": 47949,
      "training_step_time": 6.968695878982544
    },
    {
      "epoch": 4.572868347167969e-05,
      "grad_norm": 0.10140934586524963,
      "learning_rate": 5.749656677119798e-06,
      "loss": 0.0582,
      "step": 47950
    },
    {
      "epoch": 4.572868347167969e-05,
      "model_forward_time": 1.7153306007385254,
      "step": 47950
    },
    {
      "dynamics_loss_avg": 0.06570282969623804,
      "epoch": 4.572868347167969e-05,
      "step": 47950
    },
    {
      "action_loss_avg": 0.0016946412273682655,
      "epoch": 4.572868347167969e-05,
      "step": 47950
    },
    {
      "epoch": 4.572868347167969e-05,
      "step": 47950,
      "training_step_time": 7.505524635314941
    },
    {
      "epoch": 4.5729637145996094e-05,
      "model_forward_time": 1.7150769233703613,
      "step": 47951
    },
    {
      "epoch": 4.5729637145996094e-05,
      "step": 47951,
      "training_step_time": 7.716932773590088
    },
    {
      "epoch": 4.57305908203125e-05,
      "model_forward_time": 1.3401756286621094,
      "step": 47952
    },
    {
      "epoch": 4.57305908203125e-05,
      "step": 47952,
      "training_step_time": 8.467564821243286
    },
    {
      "epoch": 4.573154449462891e-05,
      "model_forward_time": 2.2847750186920166,
      "step": 47953
    },
    {
      "epoch": 4.573154449462891e-05,
      "step": 47953,
      "training_step_time": 8.373572587966919
    },
    {
      "epoch": 4.573249816894531e-05,
      "model_forward_time": 1.7213013172149658,
      "step": 47954
    },
    {
      "epoch": 4.573249816894531e-05,
      "step": 47954,
      "training_step_time": 7.820007801055908
    },
    {
      "epoch": 4.573345184326172e-05,
      "model_forward_time": 1.7181382179260254,
      "step": 47955
    },
    {
      "epoch": 4.573345184326172e-05,
      "step": 47955,
      "training_step_time": 7.801593542098999
    },
    {
      "epoch": 4.573440551757813e-05,
      "model_forward_time": 1.7325584888458252,
      "step": 47956
    },
    {
      "epoch": 4.573440551757813e-05,
      "step": 47956,
      "training_step_time": 7.785802125930786
    },
    {
      "epoch": 4.573535919189453e-05,
      "model_forward_time": 1.7249157428741455,
      "step": 47957
    },
    {
      "epoch": 4.573535919189453e-05,
      "step": 47957,
      "training_step_time": 7.383589506149292
    },
    {
      "epoch": 4.5736312866210936e-05,
      "model_forward_time": 1.721543550491333,
      "step": 47958
    },
    {
      "epoch": 4.5736312866210936e-05,
      "step": 47958,
      "training_step_time": 6.885483026504517
    },
    {
      "epoch": 4.573726654052735e-05,
      "model_forward_time": 1.7233335971832275,
      "step": 47959
    },
    {
      "epoch": 4.573726654052735e-05,
      "step": 47959,
      "training_step_time": 7.551083326339722
    },
    {
      "epoch": 4.573822021484375e-05,
      "grad_norm": 0.10060856491327286,
      "learning_rate": 5.748021856483212e-06,
      "loss": 0.0614,
      "step": 47960
    },
    {
      "epoch": 4.573822021484375e-05,
      "model_forward_time": 1.7257237434387207,
      "step": 47960
    },
    {
      "dynamics_loss_avg": 0.06187118608504534,
      "epoch": 4.573822021484375e-05,
      "step": 47960
    },
    {
      "action_loss_avg": 0.0020322888070950285,
      "epoch": 4.573822021484375e-05,
      "step": 47960
    },
    {
      "epoch": 4.573822021484375e-05,
      "step": 47960,
      "training_step_time": 7.62668251991272
    },
    {
      "epoch": 4.5739173889160155e-05,
      "model_forward_time": 1.718337059020996,
      "step": 47961
    },
    {
      "epoch": 4.5739173889160155e-05,
      "step": 47961,
      "training_step_time": 7.926659107208252
    },
    {
      "epoch": 4.5740127563476566e-05,
      "model_forward_time": 1.7198171615600586,
      "step": 47962
    },
    {
      "epoch": 4.5740127563476566e-05,
      "step": 47962,
      "training_step_time": 8.603277444839478
    },
    {
      "epoch": 4.574108123779297e-05,
      "model_forward_time": 2.2608513832092285,
      "step": 47963
    },
    {
      "epoch": 4.574108123779297e-05,
      "step": 47963,
      "training_step_time": 7.71024489402771
    },
    {
      "epoch": 4.5742034912109374e-05,
      "model_forward_time": 1.7487623691558838,
      "step": 47964
    },
    {
      "epoch": 4.5742034912109374e-05,
      "step": 47964,
      "training_step_time": 7.862075090408325
    },
    {
      "epoch": 4.5742988586425785e-05,
      "model_forward_time": 1.7202904224395752,
      "step": 47965
    },
    {
      "epoch": 4.5742988586425785e-05,
      "step": 47965,
      "training_step_time": 7.704115629196167
    },
    {
      "epoch": 4.574394226074219e-05,
      "model_forward_time": 1.7259645462036133,
      "step": 47966
    },
    {
      "epoch": 4.574394226074219e-05,
      "step": 47966,
      "training_step_time": 7.550981283187866
    },
    {
      "epoch": 4.574489593505859e-05,
      "model_forward_time": 1.7192184925079346,
      "step": 47967
    },
    {
      "epoch": 4.574489593505859e-05,
      "step": 47967,
      "training_step_time": 6.974576234817505
    },
    {
      "epoch": 4.5745849609375e-05,
      "model_forward_time": 1.7237143516540527,
      "step": 47968
    },
    {
      "epoch": 4.5745849609375e-05,
      "step": 47968,
      "training_step_time": 7.787261247634888
    },
    {
      "epoch": 4.574680328369141e-05,
      "model_forward_time": 1.7241296768188477,
      "step": 47969
    },
    {
      "epoch": 4.574680328369141e-05,
      "step": 47969,
      "training_step_time": 8.310742378234863
    },
    {
      "epoch": 4.574775695800781e-05,
      "grad_norm": 0.08369547128677368,
      "learning_rate": 5.746386954044082e-06,
      "loss": 0.0562,
      "step": 47970
    },
    {
      "epoch": 4.574775695800781e-05,
      "model_forward_time": 1.720367431640625,
      "step": 47970
    },
    {
      "dynamics_loss_avg": 0.051216832175850865,
      "epoch": 4.574775695800781e-05,
      "step": 47970
    },
    {
      "action_loss_avg": 0.0015657842974178492,
      "epoch": 4.574775695800781e-05,
      "step": 47970
    },
    {
      "epoch": 4.574775695800781e-05,
      "step": 47970,
      "training_step_time": 8.096270084381104
    },
    {
      "epoch": 4.5748710632324215e-05,
      "model_forward_time": 1.7203755378723145,
      "step": 47971
    },
    {
      "epoch": 4.5748710632324215e-05,
      "step": 47971,
      "training_step_time": 8.342774152755737
    },
    {
      "epoch": 4.5749664306640626e-05,
      "model_forward_time": 2.364819288253784,
      "step": 47972
    },
    {
      "epoch": 4.5749664306640626e-05,
      "step": 47972,
      "training_step_time": 7.7510552406311035
    },
    {
      "epoch": 4.575061798095703e-05,
      "model_forward_time": 1.7159364223480225,
      "step": 47973
    },
    {
      "epoch": 4.575061798095703e-05,
      "step": 47973,
      "training_step_time": 8.458513498306274
    },
    {
      "epoch": 4.5751571655273434e-05,
      "model_forward_time": 1.720811128616333,
      "step": 47974
    },
    {
      "epoch": 4.5751571655273434e-05,
      "step": 47974,
      "training_step_time": 7.734992742538452
    },
    {
      "epoch": 4.5752525329589845e-05,
      "model_forward_time": 1.7245655059814453,
      "step": 47975
    },
    {
      "epoch": 4.5752525329589845e-05,
      "step": 47975,
      "training_step_time": 7.544446706771851
    },
    {
      "epoch": 4.575347900390625e-05,
      "model_forward_time": 1.7198688983917236,
      "step": 47976
    },
    {
      "epoch": 4.575347900390625e-05,
      "step": 47976,
      "training_step_time": 6.958975315093994
    },
    {
      "epoch": 4.575443267822265e-05,
      "model_forward_time": 1.7197480201721191,
      "step": 47977
    },
    {
      "epoch": 4.575443267822265e-05,
      "step": 47977,
      "training_step_time": 6.960368633270264
    },
    {
      "epoch": 4.5755386352539064e-05,
      "model_forward_time": 1.720719337463379,
      "step": 47978
    },
    {
      "epoch": 4.5755386352539064e-05,
      "step": 47978,
      "training_step_time": 7.782124042510986
    },
    {
      "epoch": 4.575634002685547e-05,
      "model_forward_time": 1.7178630828857422,
      "step": 47979
    },
    {
      "epoch": 4.575634002685547e-05,
      "step": 47979,
      "training_step_time": 8.775357007980347
    },
    {
      "epoch": 4.575729370117187e-05,
      "grad_norm": 0.09950833767652512,
      "learning_rate": 5.744751969981195e-06,
      "loss": 0.0587,
      "step": 47980
    },
    {
      "epoch": 4.575729370117187e-05,
      "model_forward_time": 1.7190682888031006,
      "step": 47980
    },
    {
      "dynamics_loss_avg": 0.05787257011979818,
      "epoch": 4.575729370117187e-05,
      "step": 47980
    },
    {
      "action_loss_avg": 0.0013182456721551716,
      "epoch": 4.575729370117187e-05,
      "step": 47980
    },
    {
      "epoch": 4.575729370117187e-05,
      "step": 47980,
      "training_step_time": 8.406890392303467
    },
    {
      "epoch": 4.575824737548828e-05,
      "model_forward_time": 2.335937738418579,
      "step": 47981
    },
    {
      "epoch": 4.575824737548828e-05,
      "step": 47981,
      "training_step_time": 9.386086463928223
    },
    {
      "epoch": 4.575920104980469e-05,
      "model_forward_time": 1.7214233875274658,
      "step": 47982
    },
    {
      "epoch": 4.575920104980469e-05,
      "step": 47982,
      "training_step_time": 7.823294401168823
    },
    {
      "epoch": 4.576015472412109e-05,
      "model_forward_time": 1.7252295017242432,
      "step": 47983
    },
    {
      "epoch": 4.576015472412109e-05,
      "step": 47983,
      "training_step_time": 8.380524158477783
    },
    {
      "epoch": 4.57611083984375e-05,
      "model_forward_time": 1.6893882751464844,
      "step": 47984
    },
    {
      "epoch": 4.57611083984375e-05,
      "step": 47984,
      "training_step_time": 6.959991455078125
    },
    {
      "epoch": 4.5762062072753906e-05,
      "model_forward_time": 1.6904699802398682,
      "step": 47985
    },
    {
      "epoch": 4.5762062072753906e-05,
      "step": 47985,
      "training_step_time": 7.655735969543457
    },
    {
      "epoch": 4.576301574707031e-05,
      "model_forward_time": 1.6888418197631836,
      "step": 47986
    },
    {
      "epoch": 4.576301574707031e-05,
      "step": 47986,
      "training_step_time": 6.99263072013855
    },
    {
      "epoch": 4.576396942138672e-05,
      "model_forward_time": 1.6864919662475586,
      "step": 47987
    },
    {
      "epoch": 4.576396942138672e-05,
      "step": 47987,
      "training_step_time": 7.747044801712036
    },
    {
      "epoch": 4.5764923095703125e-05,
      "model_forward_time": 1.6917998790740967,
      "step": 47988
    },
    {
      "epoch": 4.5764923095703125e-05,
      "step": 47988,
      "training_step_time": 7.835963487625122
    },
    {
      "epoch": 4.576587677001953e-05,
      "model_forward_time": 1.69600248336792,
      "step": 47989
    },
    {
      "epoch": 4.576587677001953e-05,
      "step": 47989,
      "training_step_time": 9.84113597869873
    },
    {
      "epoch": 4.576683044433594e-05,
      "grad_norm": 0.09855729341506958,
      "learning_rate": 5.743116904473353e-06,
      "loss": 0.0616,
      "step": 47990
    },
    {
      "epoch": 4.576683044433594e-05,
      "model_forward_time": 2.2654685974121094,
      "step": 47990
    },
    {
      "dynamics_loss_avg": 0.07106616199016572,
      "epoch": 4.576683044433594e-05,
      "step": 47990
    },
    {
      "action_loss_avg": 0.0016783184837549925,
      "epoch": 4.576683044433594e-05,
      "step": 47990
    },
    {
      "epoch": 4.576683044433594e-05,
      "step": 47990,
      "training_step_time": 10.031025886535645
    },
    {
      "epoch": 4.5767784118652343e-05,
      "model_forward_time": 1.6912341117858887,
      "step": 47991
    },
    {
      "epoch": 4.5767784118652343e-05,
      "step": 47991,
      "training_step_time": 7.80408239364624
    },
    {
      "epoch": 4.576873779296875e-05,
      "model_forward_time": 1.707451581954956,
      "step": 47992
    },
    {
      "epoch": 4.576873779296875e-05,
      "step": 47992,
      "training_step_time": 8.439942359924316
    },
    {
      "epoch": 4.576969146728516e-05,
      "model_forward_time": 1.690669298171997,
      "step": 47993
    },
    {
      "epoch": 4.576969146728516e-05,
      "step": 47993,
      "training_step_time": 7.626111268997192
    },
    {
      "epoch": 4.577064514160156e-05,
      "model_forward_time": 1.68949294090271,
      "step": 47994
    },
    {
      "epoch": 4.577064514160156e-05,
      "step": 47994,
      "training_step_time": 7.605320692062378
    },
    {
      "epoch": 4.5771598815917966e-05,
      "model_forward_time": 1.7008280754089355,
      "step": 47995
    },
    {
      "epoch": 4.5771598815917966e-05,
      "step": 47995,
      "training_step_time": 6.91972804069519
    },
    {
      "epoch": 4.577255249023438e-05,
      "model_forward_time": 1.6901507377624512,
      "step": 47996
    },
    {
      "epoch": 4.577255249023438e-05,
      "step": 47996,
      "training_step_time": 7.530726909637451
    },
    {
      "epoch": 4.577350616455078e-05,
      "model_forward_time": 1.7050461769104004,
      "step": 47997
    },
    {
      "epoch": 4.577350616455078e-05,
      "step": 47997,
      "training_step_time": 7.767218351364136
    },
    {
      "epoch": 4.5774459838867185e-05,
      "model_forward_time": 1.6991009712219238,
      "step": 47998
    },
    {
      "epoch": 4.5774459838867185e-05,
      "step": 47998,
      "training_step_time": 8.96277642250061
    },
    {
      "epoch": 4.5775413513183596e-05,
      "model_forward_time": 2.235313892364502,
      "step": 47999
    },
    {
      "epoch": 4.5775413513183596e-05,
      "step": 47999,
      "training_step_time": 8.718657493591309
    },
    {
      "epoch": 4.57763671875e-05,
      "grad_norm": 0.12839071452617645,
      "learning_rate": 5.741481757699364e-06,
      "loss": 0.0585,
      "step": 48000
    },
    {
      "epoch": 4.57763671875e-05,
      "model_forward_time": 1.7978591918945312,
      "step": 48000
    },
    {
      "dynamics_loss_avg": 0.05651065334677696,
      "epoch": 4.57763671875e-05,
      "step": 48000
    },
    {
      "action_loss_avg": 0.0011793246929300949,
      "epoch": 4.57763671875e-05,
      "step": 48000
    },
    {
      "epoch": 4.57763671875e-05,
      "step": 48000,
      "training_step_time": 6.812419891357422
    },
    {
      "epoch": 4.5777320861816404e-05,
      "model_forward_time": 1.7445945739746094,
      "step": 48001
    },
    {
      "epoch": 4.5777320861816404e-05,
      "step": 48001,
      "training_step_time": 8.5120370388031
    },
    {
      "epoch": 4.5778274536132815e-05,
      "model_forward_time": 1.7625341415405273,
      "step": 48002
    },
    {
      "epoch": 4.5778274536132815e-05,
      "step": 48002,
      "training_step_time": 7.8068318367004395
    },
    {
      "epoch": 4.577922821044922e-05,
      "model_forward_time": 1.757303237915039,
      "step": 48003
    },
    {
      "epoch": 4.577922821044922e-05,
      "step": 48003,
      "training_step_time": 7.621225595474243
    },
    {
      "epoch": 4.578018188476562e-05,
      "model_forward_time": 1.744469165802002,
      "step": 48004
    },
    {
      "epoch": 4.578018188476562e-05,
      "step": 48004,
      "training_step_time": 6.873201370239258
    },
    {
      "epoch": 4.5781135559082034e-05,
      "model_forward_time": 1.7573843002319336,
      "step": 48005
    },
    {
      "epoch": 4.5781135559082034e-05,
      "step": 48005,
      "training_step_time": 8.3231360912323
    },
    {
      "epoch": 4.578208923339844e-05,
      "model_forward_time": 1.7617084980010986,
      "step": 48006
    },
    {
      "epoch": 4.578208923339844e-05,
      "step": 48006,
      "training_step_time": 9.345105171203613
    },
    {
      "epoch": 4.578304290771484e-05,
      "model_forward_time": 1.7557892799377441,
      "step": 48007
    },
    {
      "epoch": 4.578304290771484e-05,
      "step": 48007,
      "training_step_time": 10.0492103099823
    },
    {
      "epoch": 4.578399658203125e-05,
      "model_forward_time": 3.346630811691284,
      "step": 48008
    },
    {
      "epoch": 4.578399658203125e-05,
      "step": 48008,
      "training_step_time": 11.317341089248657
    },
    {
      "epoch": 4.5784950256347657e-05,
      "model_forward_time": 1.7538793087005615,
      "step": 48009
    },
    {
      "epoch": 4.5784950256347657e-05,
      "step": 48009,
      "training_step_time": 8.77961540222168
    },
    {
      "epoch": 4.578590393066406e-05,
      "grad_norm": 0.10187574476003647,
      "learning_rate": 5.739846529838044e-06,
      "loss": 0.0584,
      "step": 48010
    },
    {
      "epoch": 4.578590393066406e-05,
      "model_forward_time": 1.7448334693908691,
      "step": 48010
    },
    {
      "dynamics_loss_avg": 0.045917474664747715,
      "epoch": 4.578590393066406e-05,
      "step": 48010
    },
    {
      "action_loss_avg": 0.0033677008032100275,
      "epoch": 4.578590393066406e-05,
      "step": 48010
    },
    {
      "epoch": 4.578590393066406e-05,
      "step": 48010,
      "training_step_time": 7.915724992752075
    },
    {
      "epoch": 4.578685760498047e-05,
      "model_forward_time": 1.775162935256958,
      "step": 48011
    },
    {
      "epoch": 4.578685760498047e-05,
      "step": 48011,
      "training_step_time": 7.884599924087524
    },
    {
      "epoch": 4.5787811279296875e-05,
      "model_forward_time": 1.7676360607147217,
      "step": 48012
    },
    {
      "epoch": 4.5787811279296875e-05,
      "step": 48012,
      "training_step_time": 7.701131820678711
    },
    {
      "epoch": 4.578876495361328e-05,
      "model_forward_time": 1.779456377029419,
      "step": 48013
    },
    {
      "epoch": 4.578876495361328e-05,
      "step": 48013,
      "training_step_time": 7.028583288192749
    },
    {
      "epoch": 4.578971862792969e-05,
      "model_forward_time": 1.7623860836029053,
      "step": 48014
    },
    {
      "epoch": 4.578971862792969e-05,
      "step": 48014,
      "training_step_time": 7.773503303527832
    },
    {
      "epoch": 4.5790672302246094e-05,
      "model_forward_time": 1.7570955753326416,
      "step": 48015
    },
    {
      "epoch": 4.5790672302246094e-05,
      "step": 48015,
      "training_step_time": 8.659346580505371
    },
    {
      "epoch": 4.57916259765625e-05,
      "model_forward_time": 1.7744014263153076,
      "step": 48016
    },
    {
      "epoch": 4.57916259765625e-05,
      "step": 48016,
      "training_step_time": 10.437562704086304
    },
    {
      "epoch": 4.579257965087891e-05,
      "model_forward_time": 2.9264845848083496,
      "step": 48017
    },
    {
      "epoch": 4.579257965087891e-05,
      "step": 48017,
      "training_step_time": 9.44558072090149
    },
    {
      "epoch": 4.579353332519531e-05,
      "model_forward_time": 1.8731307983398438,
      "step": 48018
    },
    {
      "epoch": 4.579353332519531e-05,
      "step": 48018,
      "training_step_time": 10.804096221923828
    },
    {
      "epoch": 4.579448699951172e-05,
      "model_forward_time": 1.75343918800354,
      "step": 48019
    },
    {
      "epoch": 4.579448699951172e-05,
      "step": 48019,
      "training_step_time": 7.961557626724243
    },
    {
      "epoch": 4.579544067382813e-05,
      "grad_norm": 0.11794441193342209,
      "learning_rate": 5.7382112210682195e-06,
      "loss": 0.0583,
      "step": 48020
    },
    {
      "epoch": 4.579544067382813e-05,
      "model_forward_time": 1.7617521286010742,
      "step": 48020
    },
    {
      "dynamics_loss_avg": 0.04610233837738633,
      "epoch": 4.579544067382813e-05,
      "step": 48020
    },
    {
      "action_loss_avg": 0.0011698568239808082,
      "epoch": 4.579544067382813e-05,
      "step": 48020
    },
    {
      "epoch": 4.579544067382813e-05,
      "step": 48020,
      "training_step_time": 7.831695318222046
    },
    {
      "epoch": 4.579639434814453e-05,
      "model_forward_time": 1.764625072479248,
      "step": 48021
    },
    {
      "epoch": 4.579639434814453e-05,
      "step": 48021,
      "training_step_time": 7.000927925109863
    },
    {
      "epoch": 4.5797348022460936e-05,
      "model_forward_time": 1.75455904006958,
      "step": 48022
    },
    {
      "epoch": 4.5797348022460936e-05,
      "step": 48022,
      "training_step_time": 7.685939311981201
    },
    {
      "epoch": 4.579830169677735e-05,
      "model_forward_time": 1.755458116531372,
      "step": 48023
    },
    {
      "epoch": 4.579830169677735e-05,
      "step": 48023,
      "training_step_time": 7.810466527938843
    },
    {
      "epoch": 4.579925537109375e-05,
      "model_forward_time": 1.7186765670776367,
      "step": 48024
    },
    {
      "epoch": 4.579925537109375e-05,
      "step": 48024,
      "training_step_time": 8.307783126831055
    },
    {
      "epoch": 4.5800209045410155e-05,
      "model_forward_time": 1.7381153106689453,
      "step": 48025
    },
    {
      "epoch": 4.5800209045410155e-05,
      "step": 48025,
      "training_step_time": 7.839348554611206
    },
    {
      "epoch": 4.5801162719726566e-05,
      "model_forward_time": 1.7487001419067383,
      "step": 48026
    },
    {
      "epoch": 4.5801162719726566e-05,
      "step": 48026,
      "training_step_time": 9.88396668434143
    },
    {
      "epoch": 4.580211639404297e-05,
      "model_forward_time": 1.8208706378936768,
      "step": 48027
    },
    {
      "epoch": 4.580211639404297e-05,
      "step": 48027,
      "training_step_time": 10.187464714050293
    },
    {
      "epoch": 4.5803070068359374e-05,
      "model_forward_time": 1.750020980834961,
      "step": 48028
    },
    {
      "epoch": 4.5803070068359374e-05,
      "step": 48028,
      "training_step_time": 7.9370951652526855
    },
    {
      "epoch": 4.5804023742675785e-05,
      "model_forward_time": 1.7561180591583252,
      "step": 48029
    },
    {
      "epoch": 4.5804023742675785e-05,
      "step": 48029,
      "training_step_time": 7.887495756149292
    },
    {
      "epoch": 4.580497741699219e-05,
      "grad_norm": 0.10486268252134323,
      "learning_rate": 5.736575831568727e-06,
      "loss": 0.0616,
      "step": 48030
    },
    {
      "epoch": 4.580497741699219e-05,
      "model_forward_time": 1.7402138710021973,
      "step": 48030
    },
    {
      "dynamics_loss_avg": 0.05406818613409996,
      "epoch": 4.580497741699219e-05,
      "step": 48030
    },
    {
      "action_loss_avg": 0.0029705741384532303,
      "epoch": 4.580497741699219e-05,
      "step": 48030
    },
    {
      "epoch": 4.580497741699219e-05,
      "step": 48030,
      "training_step_time": 6.9455955028533936
    },
    {
      "epoch": 4.580593109130859e-05,
      "model_forward_time": 1.7286498546600342,
      "step": 48031
    },
    {
      "epoch": 4.580593109130859e-05,
      "step": 48031,
      "training_step_time": 7.75574803352356
    },
    {
      "epoch": 4.5806884765625e-05,
      "model_forward_time": 1.7387034893035889,
      "step": 48032
    },
    {
      "epoch": 4.5806884765625e-05,
      "step": 48032,
      "training_step_time": 7.7010931968688965
    },
    {
      "epoch": 4.580783843994141e-05,
      "model_forward_time": 1.729069709777832,
      "step": 48033
    },
    {
      "epoch": 4.580783843994141e-05,
      "step": 48033,
      "training_step_time": 7.7718729972839355
    },
    {
      "epoch": 4.580879211425781e-05,
      "model_forward_time": 1.744291067123413,
      "step": 48034
    },
    {
      "epoch": 4.580879211425781e-05,
      "step": 48034,
      "training_step_time": 7.854169607162476
    },
    {
      "epoch": 4.5809745788574215e-05,
      "model_forward_time": 1.7472219467163086,
      "step": 48035
    },
    {
      "epoch": 4.5809745788574215e-05,
      "step": 48035,
      "training_step_time": 9.66541838645935
    },
    {
      "epoch": 4.5810699462890626e-05,
      "model_forward_time": 1.7831907272338867,
      "step": 48036
    },
    {
      "epoch": 4.5810699462890626e-05,
      "step": 48036,
      "training_step_time": 10.202711582183838
    },
    {
      "epoch": 4.581165313720703e-05,
      "model_forward_time": 1.7440009117126465,
      "step": 48037
    },
    {
      "epoch": 4.581165313720703e-05,
      "step": 48037,
      "training_step_time": 7.9184348583221436
    },
    {
      "epoch": 4.5812606811523434e-05,
      "model_forward_time": 1.7505342960357666,
      "step": 48038
    },
    {
      "epoch": 4.5812606811523434e-05,
      "step": 48038,
      "training_step_time": 8.455490112304688
    },
    {
      "epoch": 4.5813560485839845e-05,
      "model_forward_time": 1.7411797046661377,
      "step": 48039
    },
    {
      "epoch": 4.5813560485839845e-05,
      "step": 48039,
      "training_step_time": 7.271846532821655
    },
    {
      "epoch": 4.581451416015625e-05,
      "grad_norm": 0.10869857668876648,
      "learning_rate": 5.734940361518407e-06,
      "loss": 0.0589,
      "step": 48040
    },
    {
      "epoch": 4.581451416015625e-05,
      "model_forward_time": 1.7364156246185303,
      "step": 48040
    },
    {
      "dynamics_loss_avg": 0.06769234873354435,
      "epoch": 4.581451416015625e-05,
      "step": 48040
    },
    {
      "action_loss_avg": 0.0023514812317444013,
      "epoch": 4.581451416015625e-05,
      "step": 48040
    },
    {
      "epoch": 4.581451416015625e-05,
      "step": 48040,
      "training_step_time": 7.806776523590088
    },
    {
      "epoch": 4.581546783447265e-05,
      "model_forward_time": 1.7489359378814697,
      "step": 48041
    },
    {
      "epoch": 4.581546783447265e-05,
      "step": 48041,
      "training_step_time": 7.754007577896118
    },
    {
      "epoch": 4.5816421508789064e-05,
      "model_forward_time": 1.7338783740997314,
      "step": 48042
    },
    {
      "epoch": 4.5816421508789064e-05,
      "step": 48042,
      "training_step_time": 7.761908531188965
    },
    {
      "epoch": 4.581737518310547e-05,
      "model_forward_time": 1.7425079345703125,
      "step": 48043
    },
    {
      "epoch": 4.581737518310547e-05,
      "step": 48043,
      "training_step_time": 8.835216760635376
    },
    {
      "epoch": 4.581832885742187e-05,
      "model_forward_time": 1.7381384372711182,
      "step": 48044
    },
    {
      "epoch": 4.581832885742187e-05,
      "step": 48044,
      "training_step_time": 9.502657651901245
    },
    {
      "epoch": 4.581928253173828e-05,
      "model_forward_time": 1.7559514045715332,
      "step": 48045
    },
    {
      "epoch": 4.581928253173828e-05,
      "step": 48045,
      "training_step_time": 10.023454427719116
    },
    {
      "epoch": 4.582023620605469e-05,
      "model_forward_time": 1.7388689517974854,
      "step": 48046
    },
    {
      "epoch": 4.582023620605469e-05,
      "step": 48046,
      "training_step_time": 7.938945055007935
    },
    {
      "epoch": 4.582118988037109e-05,
      "model_forward_time": 1.7588400840759277,
      "step": 48047
    },
    {
      "epoch": 4.582118988037109e-05,
      "step": 48047,
      "training_step_time": 8.047849655151367
    },
    {
      "epoch": 4.58221435546875e-05,
      "model_forward_time": 1.7448952198028564,
      "step": 48048
    },
    {
      "epoch": 4.58221435546875e-05,
      "step": 48048,
      "training_step_time": 7.892390727996826
    },
    {
      "epoch": 4.5823097229003906e-05,
      "model_forward_time": 1.7479004859924316,
      "step": 48049
    },
    {
      "epoch": 4.5823097229003906e-05,
      "step": 48049,
      "training_step_time": 7.75723123550415
    },
    {
      "epoch": 4.582405090332031e-05,
      "grad_norm": 0.14089994132518768,
      "learning_rate": 5.733304811096117e-06,
      "loss": 0.06,
      "step": 48050
    },
    {
      "epoch": 4.582405090332031e-05,
      "model_forward_time": 1.732640266418457,
      "step": 48050
    },
    {
      "dynamics_loss_avg": 0.07275263424962759,
      "epoch": 4.582405090332031e-05,
      "step": 48050
    },
    {
      "action_loss_avg": 0.003187439136672765,
      "epoch": 4.582405090332031e-05,
      "step": 48050
    },
    {
      "epoch": 4.582405090332031e-05,
      "step": 48050,
      "training_step_time": 7.645549535751343
    },
    {
      "epoch": 4.582500457763672e-05,
      "model_forward_time": 1.7420616149902344,
      "step": 48051
    },
    {
      "epoch": 4.582500457763672e-05,
      "step": 48051,
      "training_step_time": 7.489200592041016
    },
    {
      "epoch": 4.5825958251953125e-05,
      "model_forward_time": 1.746074914932251,
      "step": 48052
    },
    {
      "epoch": 4.5825958251953125e-05,
      "step": 48052,
      "training_step_time": 7.8516526222229
    },
    {
      "epoch": 4.582691192626953e-05,
      "model_forward_time": 1.762847900390625,
      "step": 48053
    },
    {
      "epoch": 4.582691192626953e-05,
      "step": 48053,
      "training_step_time": 8.528910398483276
    },
    {
      "epoch": 4.582786560058594e-05,
      "model_forward_time": 2.9494805335998535,
      "step": 48054
    },
    {
      "epoch": 4.582786560058594e-05,
      "step": 48054,
      "training_step_time": 9.350077867507935
    },
    {
      "epoch": 4.5828819274902343e-05,
      "model_forward_time": 1.8604083061218262,
      "step": 48055
    },
    {
      "epoch": 4.5828819274902343e-05,
      "step": 48055,
      "training_step_time": 7.9760332107543945
    },
    {
      "epoch": 4.582977294921875e-05,
      "model_forward_time": 1.7400758266448975,
      "step": 48056
    },
    {
      "epoch": 4.582977294921875e-05,
      "step": 48056,
      "training_step_time": 7.948648452758789
    },
    {
      "epoch": 4.583072662353516e-05,
      "model_forward_time": 1.7383158206939697,
      "step": 48057
    },
    {
      "epoch": 4.583072662353516e-05,
      "step": 48057,
      "training_step_time": 7.851276636123657
    },
    {
      "epoch": 4.583168029785156e-05,
      "model_forward_time": 1.7441318035125732,
      "step": 48058
    },
    {
      "epoch": 4.583168029785156e-05,
      "step": 48058,
      "training_step_time": 6.960549354553223
    },
    {
      "epoch": 4.5832633972167966e-05,
      "model_forward_time": 1.734114646911621,
      "step": 48059
    },
    {
      "epoch": 4.5832633972167966e-05,
      "step": 48059,
      "training_step_time": 7.621394157409668
    },
    {
      "epoch": 4.583358764648438e-05,
      "grad_norm": 0.0994332954287529,
      "learning_rate": 5.731669180480713e-06,
      "loss": 0.0581,
      "step": 48060
    },
    {
      "epoch": 4.583358764648438e-05,
      "model_forward_time": 1.733067274093628,
      "step": 48060
    },
    {
      "dynamics_loss_avg": 0.06666232738643885,
      "epoch": 4.583358764648438e-05,
      "step": 48060
    },
    {
      "action_loss_avg": 0.0015929222747217864,
      "epoch": 4.583358764648438e-05,
      "step": 48060
    },
    {
      "epoch": 4.583358764648438e-05,
      "step": 48060,
      "training_step_time": 7.755270719528198
    },
    {
      "epoch": 4.583454132080078e-05,
      "model_forward_time": 1.744081974029541,
      "step": 48061
    },
    {
      "epoch": 4.583454132080078e-05,
      "step": 48061,
      "training_step_time": 7.764968395233154
    },
    {
      "epoch": 4.5835494995117185e-05,
      "model_forward_time": 1.3734047412872314,
      "step": 48062
    },
    {
      "epoch": 4.5835494995117185e-05,
      "step": 48062,
      "training_step_time": 8.763102769851685
    },
    {
      "epoch": 4.5836448669433596e-05,
      "model_forward_time": 2.961085557937622,
      "step": 48063
    },
    {
      "epoch": 4.5836448669433596e-05,
      "step": 48063,
      "training_step_time": 9.315847158432007
    },
    {
      "epoch": 4.583740234375e-05,
      "model_forward_time": 1.8189432621002197,
      "step": 48064
    },
    {
      "epoch": 4.583740234375e-05,
      "step": 48064,
      "training_step_time": 9.69942593574524
    },
    {
      "epoch": 4.5838356018066404e-05,
      "model_forward_time": 1.752892255783081,
      "step": 48065
    },
    {
      "epoch": 4.5838356018066404e-05,
      "step": 48065,
      "training_step_time": 8.47722578048706
    },
    {
      "epoch": 4.5839309692382815e-05,
      "model_forward_time": 1.7416160106658936,
      "step": 48066
    },
    {
      "epoch": 4.5839309692382815e-05,
      "step": 48066,
      "training_step_time": 7.833216190338135
    },
    {
      "epoch": 4.584026336669922e-05,
      "model_forward_time": 1.745359182357788,
      "step": 48067
    },
    {
      "epoch": 4.584026336669922e-05,
      "step": 48067,
      "training_step_time": 7.15634560585022
    },
    {
      "epoch": 4.584121704101562e-05,
      "model_forward_time": 1.738840103149414,
      "step": 48068
    },
    {
      "epoch": 4.584121704101562e-05,
      "step": 48068,
      "training_step_time": 7.509807109832764
    },
    {
      "epoch": 4.5842170715332034e-05,
      "model_forward_time": 1.7455644607543945,
      "step": 48069
    },
    {
      "epoch": 4.5842170715332034e-05,
      "step": 48069,
      "training_step_time": 7.648230075836182
    },
    {
      "epoch": 4.584312438964844e-05,
      "grad_norm": 0.12691065669059753,
      "learning_rate": 5.730033469851068e-06,
      "loss": 0.0615,
      "step": 48070
    },
    {
      "epoch": 4.584312438964844e-05,
      "model_forward_time": 1.725433588027954,
      "step": 48070
    },
    {
      "dynamics_loss_avg": 0.06386163849383593,
      "epoch": 4.584312438964844e-05,
      "step": 48070
    },
    {
      "action_loss_avg": 0.0038444115256424994,
      "epoch": 4.584312438964844e-05,
      "step": 48070
    },
    {
      "epoch": 4.584312438964844e-05,
      "step": 48070,
      "training_step_time": 8.468017578125
    },
    {
      "epoch": 4.584407806396484e-05,
      "model_forward_time": 1.7269794940948486,
      "step": 48071
    },
    {
      "epoch": 4.584407806396484e-05,
      "step": 48071,
      "training_step_time": 7.712346076965332
    },
    {
      "epoch": 4.584503173828125e-05,
      "model_forward_time": 2.908601999282837,
      "step": 48072
    },
    {
      "epoch": 4.584503173828125e-05,
      "step": 48072,
      "training_step_time": 8.714258670806885
    },
    {
      "epoch": 4.584598541259766e-05,
      "model_forward_time": 1.8226749897003174,
      "step": 48073
    },
    {
      "epoch": 4.584598541259766e-05,
      "step": 48073,
      "training_step_time": 7.792991399765015
    },
    {
      "epoch": 4.584693908691406e-05,
      "model_forward_time": 1.9086639881134033,
      "step": 48074
    },
    {
      "epoch": 4.584693908691406e-05,
      "step": 48074,
      "training_step_time": 8.397820711135864
    },
    {
      "epoch": 4.584789276123047e-05,
      "model_forward_time": 1.915412187576294,
      "step": 48075
    },
    {
      "epoch": 4.584789276123047e-05,
      "step": 48075,
      "training_step_time": 7.823359727859497
    },
    {
      "epoch": 4.5848846435546875e-05,
      "model_forward_time": 1.8828692436218262,
      "step": 48076
    },
    {
      "epoch": 4.5848846435546875e-05,
      "step": 48076,
      "training_step_time": 7.347762107849121
    },
    {
      "epoch": 4.584980010986328e-05,
      "model_forward_time": 1.5294451713562012,
      "step": 48077
    },
    {
      "epoch": 4.584980010986328e-05,
      "step": 48077,
      "training_step_time": 7.564633846282959
    },
    {
      "epoch": 4.585075378417969e-05,
      "model_forward_time": 1.9052860736846924,
      "step": 48078
    },
    {
      "epoch": 4.585075378417969e-05,
      "step": 48078,
      "training_step_time": 7.715256929397583
    },
    {
      "epoch": 4.5851707458496094e-05,
      "model_forward_time": 1.9007694721221924,
      "step": 48079
    },
    {
      "epoch": 4.5851707458496094e-05,
      "step": 48079,
      "training_step_time": 7.737421274185181
    },
    {
      "epoch": 4.58526611328125e-05,
      "grad_norm": 0.12471082806587219,
      "learning_rate": 5.7283976793860605e-06,
      "loss": 0.0611,
      "step": 48080
    },
    {
      "epoch": 4.58526611328125e-05,
      "model_forward_time": 1.9061925411224365,
      "step": 48080
    },
    {
      "dynamics_loss_avg": 0.0575684878975153,
      "epoch": 4.58526611328125e-05,
      "step": 48080
    },
    {
      "action_loss_avg": 0.0023641003877855836,
      "epoch": 4.58526611328125e-05,
      "step": 48080
    },
    {
      "epoch": 4.58526611328125e-05,
      "step": 48080,
      "training_step_time": 7.793937683105469
    },
    {
      "epoch": 4.585361480712891e-05,
      "model_forward_time": 3.020636558532715,
      "step": 48081
    },
    {
      "epoch": 4.585361480712891e-05,
      "step": 48081,
      "training_step_time": 8.50499415397644
    },
    {
      "epoch": 4.585456848144531e-05,
      "model_forward_time": 1.9118669033050537,
      "step": 48082
    },
    {
      "epoch": 4.585456848144531e-05,
      "step": 48082,
      "training_step_time": 7.939631700515747
    },
    {
      "epoch": 4.585552215576172e-05,
      "model_forward_time": 1.5371778011322021,
      "step": 48083
    },
    {
      "epoch": 4.585552215576172e-05,
      "step": 48083,
      "training_step_time": 8.717645645141602
    },
    {
      "epoch": 4.585647583007813e-05,
      "model_forward_time": 1.9151976108551025,
      "step": 48084
    },
    {
      "epoch": 4.585647583007813e-05,
      "step": 48084,
      "training_step_time": 8.441006183624268
    },
    {
      "epoch": 4.585742950439453e-05,
      "model_forward_time": 1.9266812801361084,
      "step": 48085
    },
    {
      "epoch": 4.585742950439453e-05,
      "step": 48085,
      "training_step_time": 7.454984426498413
    },
    {
      "epoch": 4.5858383178710936e-05,
      "model_forward_time": 1.920614242553711,
      "step": 48086
    },
    {
      "epoch": 4.5858383178710936e-05,
      "step": 48086,
      "training_step_time": 7.711920261383057
    },
    {
      "epoch": 4.585933685302735e-05,
      "model_forward_time": 1.9317748546600342,
      "step": 48087
    },
    {
      "epoch": 4.585933685302735e-05,
      "step": 48087,
      "training_step_time": 7.761852979660034
    },
    {
      "epoch": 4.586029052734375e-05,
      "model_forward_time": 1.8947112560272217,
      "step": 48088
    },
    {
      "epoch": 4.586029052734375e-05,
      "step": 48088,
      "training_step_time": 7.870203256607056
    },
    {
      "epoch": 4.5861244201660155e-05,
      "model_forward_time": 1.880321741104126,
      "step": 48089
    },
    {
      "epoch": 4.5861244201660155e-05,
      "step": 48089,
      "training_step_time": 7.912980079650879
    },
    {
      "epoch": 4.5862197875976566e-05,
      "grad_norm": 0.2893874943256378,
      "learning_rate": 5.726761809264577e-06,
      "loss": 0.0561,
      "step": 48090
    },
    {
      "epoch": 4.5862197875976566e-05,
      "model_forward_time": 1.9220898151397705,
      "step": 48090
    },
    {
      "dynamics_loss_avg": 0.059166135638952254,
      "epoch": 4.5862197875976566e-05,
      "step": 48090
    },
    {
      "action_loss_avg": 0.001146789058111608,
      "epoch": 4.5862197875976566e-05,
      "step": 48090
    },
    {
      "epoch": 4.5862197875976566e-05,
      "step": 48090,
      "training_step_time": 10.159645318984985
    },
    {
      "epoch": 4.586315155029297e-05,
      "model_forward_time": 2.6007449626922607,
      "step": 48091
    },
    {
      "epoch": 4.586315155029297e-05,
      "step": 48091,
      "training_step_time": 8.992527484893799
    },
    {
      "epoch": 4.5864105224609374e-05,
      "model_forward_time": 1.8903298377990723,
      "step": 48092
    },
    {
      "epoch": 4.5864105224609374e-05,
      "step": 48092,
      "training_step_time": 9.128100872039795
    },
    {
      "epoch": 4.5865058898925785e-05,
      "model_forward_time": 1.9113140106201172,
      "step": 48093
    },
    {
      "epoch": 4.5865058898925785e-05,
      "step": 48093,
      "training_step_time": 7.883957624435425
    },
    {
      "epoch": 4.586601257324219e-05,
      "model_forward_time": 1.926548957824707,
      "step": 48094
    },
    {
      "epoch": 4.586601257324219e-05,
      "step": 48094,
      "training_step_time": 7.8494462966918945
    },
    {
      "epoch": 4.586696624755859e-05,
      "model_forward_time": 1.8866608142852783,
      "step": 48095
    },
    {
      "epoch": 4.586696624755859e-05,
      "step": 48095,
      "training_step_time": 7.638355016708374
    },
    {
      "epoch": 4.5867919921875e-05,
      "model_forward_time": 1.9022257328033447,
      "step": 48096
    },
    {
      "epoch": 4.5867919921875e-05,
      "step": 48096,
      "training_step_time": 7.805956840515137
    },
    {
      "epoch": 4.586887359619141e-05,
      "model_forward_time": 1.8993546962738037,
      "step": 48097
    },
    {
      "epoch": 4.586887359619141e-05,
      "step": 48097,
      "training_step_time": 7.7662670612335205
    },
    {
      "epoch": 4.586982727050781e-05,
      "model_forward_time": 1.8963918685913086,
      "step": 48098
    },
    {
      "epoch": 4.586982727050781e-05,
      "step": 48098,
      "training_step_time": 8.700758934020996
    },
    {
      "epoch": 4.5870780944824216e-05,
      "model_forward_time": 1.9073662757873535,
      "step": 48099
    },
    {
      "epoch": 4.5870780944824216e-05,
      "step": 48099,
      "training_step_time": 7.895128488540649
    },
    {
      "epoch": 4.5871734619140626e-05,
      "grad_norm": 0.12051551789045334,
      "learning_rate": 5.725125859665516e-06,
      "loss": 0.0607,
      "step": 48100
    },
    {
      "epoch": 4.5871734619140626e-05,
      "model_forward_time": 2.9751334190368652,
      "step": 48100
    },
    {
      "dynamics_loss_avg": 0.06185435727238655,
      "epoch": 4.5871734619140626e-05,
      "step": 48100
    },
    {
      "action_loss_avg": 0.001445324526866898,
      "epoch": 4.5871734619140626e-05,
      "step": 48100
    },
    {
      "epoch": 4.5871734619140626e-05,
      "step": 48100,
      "training_step_time": 8.80855655670166
    },
    {
      "epoch": 4.587268829345703e-05,
      "model_forward_time": 1.9129362106323242,
      "step": 48101
    },
    {
      "epoch": 4.587268829345703e-05,
      "step": 48101,
      "training_step_time": 7.9464921951293945
    },
    {
      "epoch": 4.5873641967773434e-05,
      "model_forward_time": 1.546483039855957,
      "step": 48102
    },
    {
      "epoch": 4.5873641967773434e-05,
      "step": 48102,
      "training_step_time": 7.973033666610718
    },
    {
      "epoch": 4.5874595642089845e-05,
      "model_forward_time": 1.9052941799163818,
      "step": 48103
    },
    {
      "epoch": 4.5874595642089845e-05,
      "step": 48103,
      "training_step_time": 7.8593974113464355
    },
    {
      "epoch": 4.587554931640625e-05,
      "model_forward_time": 1.9082775115966797,
      "step": 48104
    },
    {
      "epoch": 4.587554931640625e-05,
      "step": 48104,
      "training_step_time": 7.687586307525635
    },
    {
      "epoch": 4.587650299072265e-05,
      "model_forward_time": 1.9255800247192383,
      "step": 48105
    },
    {
      "epoch": 4.587650299072265e-05,
      "step": 48105,
      "training_step_time": 7.777607679367065
    },
    {
      "epoch": 4.5877456665039064e-05,
      "model_forward_time": 1.9033517837524414,
      "step": 48106
    },
    {
      "epoch": 4.5877456665039064e-05,
      "step": 48106,
      "training_step_time": 7.660569429397583
    },
    {
      "epoch": 4.587841033935547e-05,
      "model_forward_time": 1.9401369094848633,
      "step": 48107
    },
    {
      "epoch": 4.587841033935547e-05,
      "step": 48107,
      "training_step_time": 8.834466934204102
    },
    {
      "epoch": 4.587936401367187e-05,
      "model_forward_time": 1.934673547744751,
      "step": 48108
    },
    {
      "epoch": 4.587936401367187e-05,
      "step": 48108,
      "training_step_time": 8.970759868621826
    },
    {
      "epoch": 4.588031768798828e-05,
      "model_forward_time": 3.078890562057495,
      "step": 48109
    },
    {
      "epoch": 4.588031768798828e-05,
      "step": 48109,
      "training_step_time": 9.135943412780762
    },
    {
      "epoch": 4.588127136230469e-05,
      "grad_norm": 0.12292212247848511,
      "learning_rate": 5.72348983076778e-06,
      "loss": 0.0574,
      "step": 48110
    },
    {
      "epoch": 4.588127136230469e-05,
      "model_forward_time": 1.898193597793579,
      "step": 48110
    },
    {
      "dynamics_loss_avg": 0.059346651285886766,
      "epoch": 4.588127136230469e-05,
      "step": 48110
    },
    {
      "action_loss_avg": 0.001729312221868895,
      "epoch": 4.588127136230469e-05,
      "step": 48110
    },
    {
      "epoch": 4.588127136230469e-05,
      "step": 48110,
      "training_step_time": 8.890128135681152
    },
    {
      "epoch": 4.588222503662109e-05,
      "model_forward_time": 1.891493797302246,
      "step": 48111
    },
    {
      "epoch": 4.588222503662109e-05,
      "step": 48111,
      "training_step_time": 7.934033155441284
    },
    {
      "epoch": 4.58831787109375e-05,
      "model_forward_time": 1.8853859901428223,
      "step": 48112
    },
    {
      "epoch": 4.58831787109375e-05,
      "step": 48112,
      "training_step_time": 7.979945421218872
    },
    {
      "epoch": 4.5884132385253906e-05,
      "model_forward_time": 1.9042677879333496,
      "step": 48113
    },
    {
      "epoch": 4.5884132385253906e-05,
      "step": 48113,
      "training_step_time": 7.746688604354858
    },
    {
      "epoch": 4.588508605957031e-05,
      "model_forward_time": 1.926248550415039,
      "step": 48114
    },
    {
      "epoch": 4.588508605957031e-05,
      "step": 48114,
      "training_step_time": 7.683495283126831
    },
    {
      "epoch": 4.588603973388672e-05,
      "model_forward_time": 1.9265291690826416,
      "step": 48115
    },
    {
      "epoch": 4.588603973388672e-05,
      "step": 48115,
      "training_step_time": 7.71719765663147
    },
    {
      "epoch": 4.5886993408203125e-05,
      "model_forward_time": 1.9167325496673584,
      "step": 48116
    },
    {
      "epoch": 4.5886993408203125e-05,
      "step": 48116,
      "training_step_time": 7.798442125320435
    },
    {
      "epoch": 4.588794708251953e-05,
      "model_forward_time": 1.7645080089569092,
      "step": 48117
    },
    {
      "epoch": 4.588794708251953e-05,
      "step": 48117,
      "training_step_time": 8.72789478302002
    },
    {
      "epoch": 4.588890075683594e-05,
      "model_forward_time": 2.884507656097412,
      "step": 48118
    },
    {
      "epoch": 4.588890075683594e-05,
      "step": 48118,
      "training_step_time": 7.906606912612915
    },
    {
      "epoch": 4.5889854431152344e-05,
      "model_forward_time": 1.8160955905914307,
      "step": 48119
    },
    {
      "epoch": 4.5889854431152344e-05,
      "step": 48119,
      "training_step_time": 8.002412796020508
    },
    {
      "epoch": 4.589080810546875e-05,
      "grad_norm": 0.10798216611146927,
      "learning_rate": 5.7218537227502855e-06,
      "loss": 0.0589,
      "step": 48120
    },
    {
      "epoch": 4.589080810546875e-05,
      "model_forward_time": 1.7414531707763672,
      "step": 48120
    },
    {
      "dynamics_loss_avg": 0.06166671626269817,
      "epoch": 4.589080810546875e-05,
      "step": 48120
    },
    {
      "action_loss_avg": 0.0015013821190223097,
      "epoch": 4.589080810546875e-05,
      "step": 48120
    },
    {
      "epoch": 4.589080810546875e-05,
      "step": 48120,
      "training_step_time": 7.935879945755005
    },
    {
      "epoch": 4.589176177978516e-05,
      "model_forward_time": 1.737241268157959,
      "step": 48121
    },
    {
      "epoch": 4.589176177978516e-05,
      "step": 48121,
      "training_step_time": 7.879786491394043
    },
    {
      "epoch": 4.589271545410156e-05,
      "model_forward_time": 1.7375051975250244,
      "step": 48122
    },
    {
      "epoch": 4.589271545410156e-05,
      "step": 48122,
      "training_step_time": 7.662133693695068
    },
    {
      "epoch": 4.5893669128417966e-05,
      "model_forward_time": 1.7373063564300537,
      "step": 48123
    },
    {
      "epoch": 4.5893669128417966e-05,
      "step": 48123,
      "training_step_time": 7.208751916885376
    },
    {
      "epoch": 4.589462280273438e-05,
      "model_forward_time": 1.7384092807769775,
      "step": 48124
    },
    {
      "epoch": 4.589462280273438e-05,
      "step": 48124,
      "training_step_time": 7.740607976913452
    },
    {
      "epoch": 4.589557647705078e-05,
      "model_forward_time": 1.7444446086883545,
      "step": 48125
    },
    {
      "epoch": 4.589557647705078e-05,
      "step": 48125,
      "training_step_time": 7.763179302215576
    },
    {
      "epoch": 4.5896530151367185e-05,
      "model_forward_time": 1.7268993854522705,
      "step": 48126
    },
    {
      "epoch": 4.5896530151367185e-05,
      "step": 48126,
      "training_step_time": 8.531498193740845
    },
    {
      "epoch": 4.5897483825683596e-05,
      "model_forward_time": 1.7377281188964844,
      "step": 48127
    },
    {
      "epoch": 4.5897483825683596e-05,
      "step": 48127,
      "training_step_time": 8.787428140640259
    },
    {
      "epoch": 4.58984375e-05,
      "model_forward_time": 2.8575336933135986,
      "step": 48128
    },
    {
      "epoch": 4.58984375e-05,
      "step": 48128,
      "training_step_time": 10.016570568084717
    },
    {
      "epoch": 4.5899391174316404e-05,
      "model_forward_time": 1.7422473430633545,
      "step": 48129
    },
    {
      "epoch": 4.5899391174316404e-05,
      "step": 48129,
      "training_step_time": 10.232742071151733
    },
    {
      "epoch": 4.5900344848632815e-05,
      "grad_norm": 0.09775499254465103,
      "learning_rate": 5.720217535791951e-06,
      "loss": 0.06,
      "step": 48130
    },
    {
      "epoch": 4.5900344848632815e-05,
      "model_forward_time": 1.7387523651123047,
      "step": 48130
    },
    {
      "dynamics_loss_avg": 0.06096140276640653,
      "epoch": 4.5900344848632815e-05,
      "step": 48130
    },
    {
      "action_loss_avg": 0.002253643481526524,
      "epoch": 4.5900344848632815e-05,
      "step": 48130
    },
    {
      "epoch": 4.5900344848632815e-05,
      "step": 48130,
      "training_step_time": 7.857269287109375
    },
    {
      "epoch": 4.590129852294922e-05,
      "model_forward_time": 1.7394845485687256,
      "step": 48131
    },
    {
      "epoch": 4.590129852294922e-05,
      "step": 48131,
      "training_step_time": 7.587939262390137
    },
    {
      "epoch": 4.590225219726562e-05,
      "model_forward_time": 1.731241226196289,
      "step": 48132
    },
    {
      "epoch": 4.590225219726562e-05,
      "step": 48132,
      "training_step_time": 7.658929109573364
    },
    {
      "epoch": 4.5903205871582034e-05,
      "model_forward_time": 1.7360143661499023,
      "step": 48133
    },
    {
      "epoch": 4.5903205871582034e-05,
      "step": 48133,
      "training_step_time": 7.731780052185059
    },
    {
      "epoch": 4.590415954589844e-05,
      "model_forward_time": 1.7418773174285889,
      "step": 48134
    },
    {
      "epoch": 4.590415954589844e-05,
      "step": 48134,
      "training_step_time": 8.650595664978027
    },
    {
      "epoch": 4.590511322021484e-05,
      "model_forward_time": 1.3748140335083008,
      "step": 48135
    },
    {
      "epoch": 4.590511322021484e-05,
      "step": 48135,
      "training_step_time": 7.8452675342559814
    },
    {
      "epoch": 4.590606689453125e-05,
      "model_forward_time": 1.7428414821624756,
      "step": 48136
    },
    {
      "epoch": 4.590606689453125e-05,
      "step": 48136,
      "training_step_time": 7.82419228553772
    },
    {
      "epoch": 4.590702056884766e-05,
      "model_forward_time": 2.8933603763580322,
      "step": 48137
    },
    {
      "epoch": 4.590702056884766e-05,
      "step": 48137,
      "training_step_time": 8.754712581634521
    },
    {
      "epoch": 4.590797424316406e-05,
      "model_forward_time": 1.7314486503601074,
      "step": 48138
    },
    {
      "epoch": 4.590797424316406e-05,
      "step": 48138,
      "training_step_time": 7.816823244094849
    },
    {
      "epoch": 4.590892791748047e-05,
      "model_forward_time": 1.732149600982666,
      "step": 48139
    },
    {
      "epoch": 4.590892791748047e-05,
      "step": 48139,
      "training_step_time": 8.588623762130737
    },
    {
      "epoch": 4.5909881591796876e-05,
      "grad_norm": 0.12755821645259857,
      "learning_rate": 5.718581270071711e-06,
      "loss": 0.0568,
      "step": 48140
    },
    {
      "epoch": 4.5909881591796876e-05,
      "model_forward_time": 1.751720666885376,
      "step": 48140
    },
    {
      "dynamics_loss_avg": 0.04726677350699902,
      "epoch": 4.5909881591796876e-05,
      "step": 48140
    },
    {
      "action_loss_avg": 0.005434550627251155,
      "epoch": 4.5909881591796876e-05,
      "step": 48140
    },
    {
      "epoch": 4.5909881591796876e-05,
      "step": 48140,
      "training_step_time": 7.799732446670532
    },
    {
      "epoch": 4.591083526611328e-05,
      "model_forward_time": 1.742548942565918,
      "step": 48141
    },
    {
      "epoch": 4.591083526611328e-05,
      "step": 48141,
      "training_step_time": 7.721312046051025
    },
    {
      "epoch": 4.591178894042969e-05,
      "model_forward_time": 1.7271151542663574,
      "step": 48142
    },
    {
      "epoch": 4.591178894042969e-05,
      "step": 48142,
      "training_step_time": 7.771723747253418
    },
    {
      "epoch": 4.5912742614746094e-05,
      "model_forward_time": 1.741844892501831,
      "step": 48143
    },
    {
      "epoch": 4.5912742614746094e-05,
      "step": 48143,
      "training_step_time": 8.638959169387817
    },
    {
      "epoch": 4.59136962890625e-05,
      "model_forward_time": 1.742126703262329,
      "step": 48144
    },
    {
      "epoch": 4.59136962890625e-05,
      "step": 48144,
      "training_step_time": 9.913135528564453
    },
    {
      "epoch": 4.591464996337891e-05,
      "model_forward_time": 1.7286605834960938,
      "step": 48145
    },
    {
      "epoch": 4.591464996337891e-05,
      "step": 48145,
      "training_step_time": 8.903991460800171
    },
    {
      "epoch": 4.591560363769531e-05,
      "model_forward_time": 2.885655403137207,
      "step": 48146
    },
    {
      "epoch": 4.591560363769531e-05,
      "step": 48146,
      "training_step_time": 8.012313604354858
    },
    {
      "epoch": 4.591655731201172e-05,
      "model_forward_time": 1.7306749820709229,
      "step": 48147
    },
    {
      "epoch": 4.591655731201172e-05,
      "step": 48147,
      "training_step_time": 7.875531911849976
    },
    {
      "epoch": 4.591751098632813e-05,
      "model_forward_time": 1.7427451610565186,
      "step": 48148
    },
    {
      "epoch": 4.591751098632813e-05,
      "step": 48148,
      "training_step_time": 8.024064540863037
    },
    {
      "epoch": 4.591846466064453e-05,
      "model_forward_time": 1.7334465980529785,
      "step": 48149
    },
    {
      "epoch": 4.591846466064453e-05,
      "step": 48149,
      "training_step_time": 7.857845783233643
    },
    {
      "epoch": 4.5919418334960936e-05,
      "grad_norm": 0.1275416761636734,
      "learning_rate": 5.716944925768505e-06,
      "loss": 0.0611,
      "step": 48150
    },
    {
      "epoch": 4.5919418334960936e-05,
      "model_forward_time": 1.7596104145050049,
      "step": 48150
    },
    {
      "dynamics_loss_avg": 0.06300502084195614,
      "epoch": 4.5919418334960936e-05,
      "step": 48150
    },
    {
      "action_loss_avg": 0.0026710529462434353,
      "epoch": 4.5919418334960936e-05,
      "step": 48150
    },
    {
      "epoch": 4.5919418334960936e-05,
      "step": 48150,
      "training_step_time": 7.8282482624053955
    },
    {
      "epoch": 4.592037200927735e-05,
      "model_forward_time": 1.7381258010864258,
      "step": 48151
    },
    {
      "epoch": 4.592037200927735e-05,
      "step": 48151,
      "training_step_time": 7.800740480422974
    },
    {
      "epoch": 4.592132568359375e-05,
      "model_forward_time": 1.7398364543914795,
      "step": 48152
    },
    {
      "epoch": 4.592132568359375e-05,
      "step": 48152,
      "training_step_time": 8.06736445426941
    },
    {
      "epoch": 4.5922279357910155e-05,
      "model_forward_time": 1.7365539073944092,
      "step": 48153
    },
    {
      "epoch": 4.5922279357910155e-05,
      "step": 48153,
      "training_step_time": 7.88013482093811
    },
    {
      "epoch": 4.5923233032226566e-05,
      "model_forward_time": 1.7475595474243164,
      "step": 48154
    },
    {
      "epoch": 4.5923233032226566e-05,
      "step": 48154,
      "training_step_time": 9.509314775466919
    },
    {
      "epoch": 4.592418670654297e-05,
      "model_forward_time": 2.9648144245147705,
      "step": 48155
    },
    {
      "epoch": 4.592418670654297e-05,
      "step": 48155,
      "training_step_time": 7.9342041015625
    },
    {
      "epoch": 4.5925140380859374e-05,
      "model_forward_time": 1.7367541790008545,
      "step": 48156
    },
    {
      "epoch": 4.5925140380859374e-05,
      "step": 48156,
      "training_step_time": 8.53121280670166
    },
    {
      "epoch": 4.592609405517578e-05,
      "model_forward_time": 1.7288737297058105,
      "step": 48157
    },
    {
      "epoch": 4.592609405517578e-05,
      "step": 48157,
      "training_step_time": 8.780138969421387
    },
    {
      "epoch": 4.592704772949219e-05,
      "model_forward_time": 1.7388689517974854,
      "step": 48158
    },
    {
      "epoch": 4.592704772949219e-05,
      "step": 48158,
      "training_step_time": 7.798095703125
    },
    {
      "epoch": 4.592800140380859e-05,
      "model_forward_time": 1.7405481338500977,
      "step": 48159
    },
    {
      "epoch": 4.592800140380859e-05,
      "step": 48159,
      "training_step_time": 7.679754018783569
    },
    {
      "epoch": 4.5928955078125e-05,
      "grad_norm": 0.08914466202259064,
      "learning_rate": 5.715308503061279e-06,
      "loss": 0.0591,
      "step": 48160
    },
    {
      "epoch": 4.5928955078125e-05,
      "model_forward_time": 1.7272703647613525,
      "step": 48160
    },
    {
      "dynamics_loss_avg": 0.05060356045141816,
      "epoch": 4.5928955078125e-05,
      "step": 48160
    },
    {
      "action_loss_avg": 0.0019197808811441063,
      "epoch": 4.5928955078125e-05,
      "step": 48160
    },
    {
      "epoch": 4.5928955078125e-05,
      "step": 48160,
      "training_step_time": 7.402346611022949
    },
    {
      "epoch": 4.592990875244141e-05,
      "model_forward_time": 1.7690422534942627,
      "step": 48161
    },
    {
      "epoch": 4.592990875244141e-05,
      "step": 48161,
      "training_step_time": 7.729981422424316
    },
    {
      "epoch": 4.593086242675781e-05,
      "model_forward_time": 1.7351281642913818,
      "step": 48162
    },
    {
      "epoch": 4.593086242675781e-05,
      "step": 48162,
      "training_step_time": 7.897994756698608
    },
    {
      "epoch": 4.5931816101074216e-05,
      "model_forward_time": 1.747826337814331,
      "step": 48163
    },
    {
      "epoch": 4.5931816101074216e-05,
      "step": 48163,
      "training_step_time": 8.595475912094116
    },
    {
      "epoch": 4.5932769775390626e-05,
      "model_forward_time": 2.9064595699310303,
      "step": 48164
    },
    {
      "epoch": 4.5932769775390626e-05,
      "step": 48164,
      "training_step_time": 9.21170687675476
    },
    {
      "epoch": 4.593372344970703e-05,
      "model_forward_time": 1.7641549110412598,
      "step": 48165
    },
    {
      "epoch": 4.593372344970703e-05,
      "step": 48165,
      "training_step_time": 9.979223251342773
    },
    {
      "epoch": 4.5934677124023434e-05,
      "model_forward_time": 1.746253490447998,
      "step": 48166
    },
    {
      "epoch": 4.5934677124023434e-05,
      "step": 48166,
      "training_step_time": 8.036888360977173
    },
    {
      "epoch": 4.5935630798339845e-05,
      "model_forward_time": 1.7581429481506348,
      "step": 48167
    },
    {
      "epoch": 4.5935630798339845e-05,
      "step": 48167,
      "training_step_time": 7.8890886306762695
    },
    {
      "epoch": 4.593658447265625e-05,
      "model_forward_time": 1.710005283355713,
      "step": 48168
    },
    {
      "epoch": 4.593658447265625e-05,
      "step": 48168,
      "training_step_time": 7.516732215881348
    },
    {
      "epoch": 4.593753814697265e-05,
      "model_forward_time": 1.733675241470337,
      "step": 48169
    },
    {
      "epoch": 4.593753814697265e-05,
      "step": 48169,
      "training_step_time": 7.642864942550659
    },
    {
      "epoch": 4.5938491821289064e-05,
      "grad_norm": 0.09220896661281586,
      "learning_rate": 5.7136720021289895e-06,
      "loss": 0.0628,
      "step": 48170
    },
    {
      "epoch": 4.5938491821289064e-05,
      "model_forward_time": 1.728616714477539,
      "step": 48170
    },
    {
      "dynamics_loss_avg": 0.06100099394097924,
      "epoch": 4.5938491821289064e-05,
      "step": 48170
    },
    {
      "action_loss_avg": 0.001577905530575663,
      "epoch": 4.5938491821289064e-05,
      "step": 48170
    },
    {
      "epoch": 4.5938491821289064e-05,
      "step": 48170,
      "training_step_time": 7.4233808517456055
    },
    {
      "epoch": 4.593944549560547e-05,
      "model_forward_time": 1.737497091293335,
      "step": 48171
    },
    {
      "epoch": 4.593944549560547e-05,
      "step": 48171,
      "training_step_time": 7.81704568862915
    },
    {
      "epoch": 4.594039916992187e-05,
      "model_forward_time": 1.8006184101104736,
      "step": 48172
    },
    {
      "epoch": 4.594039916992187e-05,
      "step": 48172,
      "training_step_time": 7.887692928314209
    },
    {
      "epoch": 4.594135284423828e-05,
      "model_forward_time": 1.7379953861236572,
      "step": 48173
    },
    {
      "epoch": 4.594135284423828e-05,
      "step": 48173,
      "training_step_time": 9.98114275932312
    },
    {
      "epoch": 4.594230651855469e-05,
      "model_forward_time": 1.8066871166229248,
      "step": 48174
    },
    {
      "epoch": 4.594230651855469e-05,
      "step": 48174,
      "training_step_time": 9.576558351516724
    },
    {
      "epoch": 4.594326019287109e-05,
      "model_forward_time": 1.7593483924865723,
      "step": 48175
    },
    {
      "epoch": 4.594326019287109e-05,
      "step": 48175,
      "training_step_time": 8.838268041610718
    },
    {
      "epoch": 4.59442138671875e-05,
      "model_forward_time": 1.7272582054138184,
      "step": 48176
    },
    {
      "epoch": 4.59442138671875e-05,
      "step": 48176,
      "training_step_time": 7.862196922302246
    },
    {
      "epoch": 4.5945167541503906e-05,
      "model_forward_time": 1.7478289604187012,
      "step": 48177
    },
    {
      "epoch": 4.5945167541503906e-05,
      "step": 48177,
      "training_step_time": 7.656539440155029
    },
    {
      "epoch": 4.594612121582031e-05,
      "model_forward_time": 1.7312297821044922,
      "step": 48178
    },
    {
      "epoch": 4.594612121582031e-05,
      "step": 48178,
      "training_step_time": 7.65445876121521
    },
    {
      "epoch": 4.594707489013672e-05,
      "model_forward_time": 1.7290866374969482,
      "step": 48179
    },
    {
      "epoch": 4.594707489013672e-05,
      "step": 48179,
      "training_step_time": 7.248440265655518
    },
    {
      "epoch": 4.5948028564453125e-05,
      "grad_norm": 0.09673792123794556,
      "learning_rate": 5.7120354231506024e-06,
      "loss": 0.0625,
      "step": 48180
    },
    {
      "epoch": 4.5948028564453125e-05,
      "model_forward_time": 1.7349700927734375,
      "step": 48180
    },
    {
      "dynamics_loss_avg": 0.06997611783444882,
      "epoch": 4.5948028564453125e-05,
      "step": 48180
    },
    {
      "action_loss_avg": 0.003557497226574924,
      "epoch": 4.5948028564453125e-05,
      "step": 48180
    },
    {
      "epoch": 4.5948028564453125e-05,
      "step": 48180,
      "training_step_time": 7.809063196182251
    },
    {
      "epoch": 4.594898223876953e-05,
      "model_forward_time": 1.6813633441925049,
      "step": 48181
    },
    {
      "epoch": 4.594898223876953e-05,
      "step": 48181,
      "training_step_time": 7.908451557159424
    },
    {
      "epoch": 4.594993591308594e-05,
      "model_forward_time": 1.685020923614502,
      "step": 48182
    },
    {
      "epoch": 4.594993591308594e-05,
      "step": 48182,
      "training_step_time": 9.505679845809937
    },
    {
      "epoch": 4.5950889587402344e-05,
      "model_forward_time": 2.2309165000915527,
      "step": 48183
    },
    {
      "epoch": 4.5950889587402344e-05,
      "step": 48183,
      "training_step_time": 8.601639986038208
    },
    {
      "epoch": 4.595184326171875e-05,
      "model_forward_time": 1.678175687789917,
      "step": 48184
    },
    {
      "epoch": 4.595184326171875e-05,
      "step": 48184,
      "training_step_time": 8.659713983535767
    },
    {
      "epoch": 4.595279693603516e-05,
      "model_forward_time": 1.6954536437988281,
      "step": 48185
    },
    {
      "epoch": 4.595279693603516e-05,
      "step": 48185,
      "training_step_time": 8.593102931976318
    },
    {
      "epoch": 4.595375061035156e-05,
      "model_forward_time": 1.6817998886108398,
      "step": 48186
    },
    {
      "epoch": 4.595375061035156e-05,
      "step": 48186,
      "training_step_time": 7.613357305526733
    },
    {
      "epoch": 4.5954704284667966e-05,
      "model_forward_time": 1.6895906925201416,
      "step": 48187
    },
    {
      "epoch": 4.5954704284667966e-05,
      "step": 48187,
      "training_step_time": 7.608529806137085
    },
    {
      "epoch": 4.595565795898438e-05,
      "model_forward_time": 1.7044451236724854,
      "step": 48188
    },
    {
      "epoch": 4.595565795898438e-05,
      "step": 48188,
      "training_step_time": 7.39928126335144
    },
    {
      "epoch": 4.595661163330078e-05,
      "model_forward_time": 1.6850390434265137,
      "step": 48189
    },
    {
      "epoch": 4.595661163330078e-05,
      "step": 48189,
      "training_step_time": 7.80059027671814
    },
    {
      "epoch": 4.5957565307617185e-05,
      "grad_norm": 0.13044925034046173,
      "learning_rate": 5.710398766305094e-06,
      "loss": 0.0587,
      "step": 48190
    },
    {
      "epoch": 4.5957565307617185e-05,
      "model_forward_time": 1.6804993152618408,
      "step": 48190
    },
    {
      "dynamics_loss_avg": 0.05282182889059186,
      "epoch": 4.5957565307617185e-05,
      "step": 48190
    },
    {
      "action_loss_avg": 0.0016206393716856836,
      "epoch": 4.5957565307617185e-05,
      "step": 48190
    },
    {
      "epoch": 4.5957565307617185e-05,
      "step": 48190,
      "training_step_time": 7.746456623077393
    },
    {
      "epoch": 4.5958518981933596e-05,
      "model_forward_time": 1.682420253753662,
      "step": 48191
    },
    {
      "epoch": 4.5958518981933596e-05,
      "step": 48191,
      "training_step_time": 9.86669659614563
    },
    {
      "epoch": 4.595947265625e-05,
      "model_forward_time": 2.2288739681243896,
      "step": 48192
    },
    {
      "epoch": 4.595947265625e-05,
      "step": 48192,
      "training_step_time": 7.823617696762085
    },
    {
      "epoch": 4.5960426330566404e-05,
      "model_forward_time": 1.690303087234497,
      "step": 48193
    },
    {
      "epoch": 4.5960426330566404e-05,
      "step": 48193,
      "training_step_time": 7.906430721282959
    },
    {
      "epoch": 4.5961380004882815e-05,
      "model_forward_time": 1.686098575592041,
      "step": 48194
    },
    {
      "epoch": 4.5961380004882815e-05,
      "step": 48194,
      "training_step_time": 7.903994798660278
    },
    {
      "epoch": 4.596233367919922e-05,
      "model_forward_time": 1.706437349319458,
      "step": 48195
    },
    {
      "epoch": 4.596233367919922e-05,
      "step": 48195,
      "training_step_time": 8.677340984344482
    },
    {
      "epoch": 4.596328735351562e-05,
      "model_forward_time": 1.682126522064209,
      "step": 48196
    },
    {
      "epoch": 4.596328735351562e-05,
      "step": 48196,
      "training_step_time": 6.915616512298584
    },
    {
      "epoch": 4.5964241027832034e-05,
      "model_forward_time": 1.7014951705932617,
      "step": 48197
    },
    {
      "epoch": 4.5964241027832034e-05,
      "step": 48197,
      "training_step_time": 7.598793983459473
    },
    {
      "epoch": 4.596519470214844e-05,
      "model_forward_time": 1.6986663341522217,
      "step": 48198
    },
    {
      "epoch": 4.596519470214844e-05,
      "step": 48198,
      "training_step_time": 7.8137047290802
    },
    {
      "epoch": 4.596614837646484e-05,
      "model_forward_time": 1.6872694492340088,
      "step": 48199
    },
    {
      "epoch": 4.596614837646484e-05,
      "step": 48199,
      "training_step_time": 7.846875190734863
    },
    {
      "epoch": 4.596710205078125e-05,
      "grad_norm": 0.13277150690555573,
      "learning_rate": 5.708762031771442e-06,
      "loss": 0.0613,
      "step": 48200
    },
    {
      "epoch": 4.596710205078125e-05,
      "model_forward_time": 1.3137760162353516,
      "step": 48200
    },
    {
      "dynamics_loss_avg": 0.05420871786773205,
      "epoch": 4.596710205078125e-05,
      "step": 48200
    },
    {
      "action_loss_avg": 0.001685576734598726,
      "epoch": 4.596710205078125e-05,
      "step": 48200
    },
    {
      "epoch": 4.596710205078125e-05,
      "step": 48200,
      "training_step_time": 7.792844533920288
    },
    {
      "epoch": 4.596805572509766e-05,
      "model_forward_time": 2.2378273010253906,
      "step": 48201
    },
    {
      "epoch": 4.596805572509766e-05,
      "step": 48201,
      "training_step_time": 8.967320680618286
    },
    {
      "epoch": 4.596900939941406e-05,
      "model_forward_time": 1.686190128326416,
      "step": 48202
    },
    {
      "epoch": 4.596900939941406e-05,
      "step": 48202,
      "training_step_time": 8.600453615188599
    },
    {
      "epoch": 4.596996307373047e-05,
      "model_forward_time": 1.6888141632080078,
      "step": 48203
    },
    {
      "epoch": 4.596996307373047e-05,
      "step": 48203,
      "training_step_time": 9.813744068145752
    },
    {
      "epoch": 4.5970916748046876e-05,
      "model_forward_time": 1.6871800422668457,
      "step": 48204
    },
    {
      "epoch": 4.5970916748046876e-05,
      "step": 48204,
      "training_step_time": 7.883064031600952
    },
    {
      "epoch": 4.597187042236328e-05,
      "model_forward_time": 1.6869568824768066,
      "step": 48205
    },
    {
      "epoch": 4.597187042236328e-05,
      "step": 48205,
      "training_step_time": 7.426178455352783
    },
    {
      "epoch": 4.597282409667969e-05,
      "model_forward_time": 1.682816743850708,
      "step": 48206
    },
    {
      "epoch": 4.597282409667969e-05,
      "step": 48206,
      "training_step_time": 7.604421377182007
    },
    {
      "epoch": 4.5973777770996094e-05,
      "model_forward_time": 1.693981647491455,
      "step": 48207
    },
    {
      "epoch": 4.5973777770996094e-05,
      "step": 48207,
      "training_step_time": 7.794554233551025
    },
    {
      "epoch": 4.59747314453125e-05,
      "model_forward_time": 1.6817281246185303,
      "step": 48208
    },
    {
      "epoch": 4.59747314453125e-05,
      "step": 48208,
      "training_step_time": 7.788761377334595
    },
    {
      "epoch": 4.597568511962891e-05,
      "model_forward_time": 1.6863822937011719,
      "step": 48209
    },
    {
      "epoch": 4.597568511962891e-05,
      "step": 48209,
      "training_step_time": 7.832557201385498
    },
    {
      "epoch": 4.597663879394531e-05,
      "grad_norm": 0.1216433122754097,
      "learning_rate": 5.707125219728642e-06,
      "loss": 0.0629,
      "step": 48210
    },
    {
      "epoch": 4.597663879394531e-05,
      "model_forward_time": 2.2189981937408447,
      "step": 48210
    },
    {
      "dynamics_loss_avg": 0.06325760204344988,
      "epoch": 4.597663879394531e-05,
      "step": 48210
    },
    {
      "action_loss_avg": 0.0016289453371427954,
      "epoch": 4.597663879394531e-05,
      "step": 48210
    },
    {
      "epoch": 4.597663879394531e-05,
      "step": 48210,
      "training_step_time": 7.8201634883880615
    },
    {
      "epoch": 4.597759246826172e-05,
      "model_forward_time": 1.6856889724731445,
      "step": 48211
    },
    {
      "epoch": 4.597759246826172e-05,
      "step": 48211,
      "training_step_time": 9.283439636230469
    },
    {
      "epoch": 4.597854614257813e-05,
      "model_forward_time": 1.6817939281463623,
      "step": 48212
    },
    {
      "epoch": 4.597854614257813e-05,
      "step": 48212,
      "training_step_time": 7.853881120681763
    },
    {
      "epoch": 4.597949981689453e-05,
      "model_forward_time": 1.6815836429595947,
      "step": 48213
    },
    {
      "epoch": 4.597949981689453e-05,
      "step": 48213,
      "training_step_time": 7.925218105316162
    },
    {
      "epoch": 4.5980453491210936e-05,
      "model_forward_time": 1.3268787860870361,
      "step": 48214
    },
    {
      "epoch": 4.5980453491210936e-05,
      "step": 48214,
      "training_step_time": 7.582785129547119
    },
    {
      "epoch": 4.598140716552735e-05,
      "model_forward_time": 1.6819136142730713,
      "step": 48215
    },
    {
      "epoch": 4.598140716552735e-05,
      "step": 48215,
      "training_step_time": 7.48026180267334
    },
    {
      "epoch": 4.598236083984375e-05,
      "model_forward_time": 1.6943042278289795,
      "step": 48216
    },
    {
      "epoch": 4.598236083984375e-05,
      "step": 48216,
      "training_step_time": 7.81804633140564
    },
    {
      "epoch": 4.5983314514160155e-05,
      "model_forward_time": 1.6831321716308594,
      "step": 48217
    },
    {
      "epoch": 4.5983314514160155e-05,
      "step": 48217,
      "training_step_time": 7.816520929336548
    },
    {
      "epoch": 4.5984268188476566e-05,
      "model_forward_time": 1.6910557746887207,
      "step": 48218
    },
    {
      "epoch": 4.5984268188476566e-05,
      "step": 48218,
      "training_step_time": 7.870434045791626
    },
    {
      "epoch": 4.598522186279297e-05,
      "model_forward_time": 1.6819908618927002,
      "step": 48219
    },
    {
      "epoch": 4.598522186279297e-05,
      "step": 48219,
      "training_step_time": 8.803270101547241
    },
    {
      "epoch": 4.5986175537109374e-05,
      "grad_norm": 0.15128086507320404,
      "learning_rate": 5.705488330355691e-06,
      "loss": 0.0601,
      "step": 48220
    },
    {
      "epoch": 4.5986175537109374e-05,
      "model_forward_time": 2.2135210037231445,
      "step": 48220
    },
    {
      "dynamics_loss_avg": 0.04811303876340389,
      "epoch": 4.5986175537109374e-05,
      "step": 48220
    },
    {
      "action_loss_avg": 0.0016422323562437669,
      "epoch": 4.5986175537109374e-05,
      "step": 48220
    },
    {
      "epoch": 4.5986175537109374e-05,
      "step": 48220,
      "training_step_time": 7.889641523361206
    },
    {
      "epoch": 4.598712921142578e-05,
      "model_forward_time": 1.6905782222747803,
      "step": 48221
    },
    {
      "epoch": 4.598712921142578e-05,
      "step": 48221,
      "training_step_time": 7.8972008228302
    },
    {
      "epoch": 4.598808288574219e-05,
      "model_forward_time": 1.6920280456542969,
      "step": 48222
    },
    {
      "epoch": 4.598808288574219e-05,
      "step": 48222,
      "training_step_time": 7.878472328186035
    },
    {
      "epoch": 4.598903656005859e-05,
      "model_forward_time": 1.6910350322723389,
      "step": 48223
    },
    {
      "epoch": 4.598903656005859e-05,
      "step": 48223,
      "training_step_time": 8.711651086807251
    },
    {
      "epoch": 4.5989990234375e-05,
      "model_forward_time": 1.6875004768371582,
      "step": 48224
    },
    {
      "epoch": 4.5989990234375e-05,
      "step": 48224,
      "training_step_time": 7.931813478469849
    },
    {
      "epoch": 4.599094390869141e-05,
      "model_forward_time": 1.6855549812316895,
      "step": 48225
    },
    {
      "epoch": 4.599094390869141e-05,
      "step": 48225,
      "training_step_time": 7.819821834564209
    },
    {
      "epoch": 4.599189758300781e-05,
      "model_forward_time": 1.6877105236053467,
      "step": 48226
    },
    {
      "epoch": 4.599189758300781e-05,
      "step": 48226,
      "training_step_time": 8.855255842208862
    },
    {
      "epoch": 4.5992851257324216e-05,
      "model_forward_time": 1.690659523010254,
      "step": 48227
    },
    {
      "epoch": 4.5992851257324216e-05,
      "step": 48227,
      "training_step_time": 7.881686210632324
    },
    {
      "epoch": 4.5993804931640626e-05,
      "model_forward_time": 1.6865975856781006,
      "step": 48228
    },
    {
      "epoch": 4.5993804931640626e-05,
      "step": 48228,
      "training_step_time": 8.453512191772461
    },
    {
      "epoch": 4.599475860595703e-05,
      "model_forward_time": 2.251204013824463,
      "step": 48229
    },
    {
      "epoch": 4.599475860595703e-05,
      "step": 48229,
      "training_step_time": 7.940553665161133
    },
    {
      "epoch": 4.5995712280273435e-05,
      "grad_norm": 0.11129841208457947,
      "learning_rate": 5.703851363831595e-06,
      "loss": 0.0629,
      "step": 48230
    },
    {
      "epoch": 4.5995712280273435e-05,
      "model_forward_time": 1.6754646301269531,
      "step": 48230
    },
    {
      "dynamics_loss_avg": 0.06362324096262455,
      "epoch": 4.5995712280273435e-05,
      "step": 48230
    },
    {
      "action_loss_avg": 0.0017877462378237396,
      "epoch": 4.5995712280273435e-05,
      "step": 48230
    },
    {
      "epoch": 4.5995712280273435e-05,
      "step": 48230,
      "training_step_time": 7.928731918334961
    },
    {
      "epoch": 4.5996665954589845e-05,
      "model_forward_time": 1.684673547744751,
      "step": 48231
    },
    {
      "epoch": 4.5996665954589845e-05,
      "step": 48231,
      "training_step_time": 9.290448427200317
    },
    {
      "epoch": 4.599761962890625e-05,
      "model_forward_time": 1.6827538013458252,
      "step": 48232
    },
    {
      "epoch": 4.599761962890625e-05,
      "step": 48232,
      "training_step_time": 8.432857513427734
    },
    {
      "epoch": 4.599857330322265e-05,
      "model_forward_time": 1.68800950050354,
      "step": 48233
    },
    {
      "epoch": 4.599857330322265e-05,
      "step": 48233,
      "training_step_time": 7.787788391113281
    },
    {
      "epoch": 4.5999526977539064e-05,
      "model_forward_time": 1.6937532424926758,
      "step": 48234
    },
    {
      "epoch": 4.5999526977539064e-05,
      "step": 48234,
      "training_step_time": 7.8203887939453125
    },
    {
      "epoch": 4.600048065185547e-05,
      "model_forward_time": 1.7098007202148438,
      "step": 48235
    },
    {
      "epoch": 4.600048065185547e-05,
      "step": 48235,
      "training_step_time": 7.75353217124939
    },
    {
      "epoch": 4.600143432617187e-05,
      "model_forward_time": 1.6872897148132324,
      "step": 48236
    },
    {
      "epoch": 4.600143432617187e-05,
      "step": 48236,
      "training_step_time": 7.8091652393341064
    },
    {
      "epoch": 4.600238800048828e-05,
      "model_forward_time": 1.692505121231079,
      "step": 48237
    },
    {
      "epoch": 4.600238800048828e-05,
      "step": 48237,
      "training_step_time": 8.851361274719238
    },
    {
      "epoch": 4.600334167480469e-05,
      "model_forward_time": 2.2300703525543213,
      "step": 48238
    },
    {
      "epoch": 4.600334167480469e-05,
      "step": 48238,
      "training_step_time": 8.376021146774292
    },
    {
      "epoch": 4.600429534912109e-05,
      "model_forward_time": 1.6854197978973389,
      "step": 48239
    },
    {
      "epoch": 4.600429534912109e-05,
      "step": 48239,
      "training_step_time": 8.7513587474823
    },
    {
      "epoch": 4.60052490234375e-05,
      "grad_norm": 0.11753500252962112,
      "learning_rate": 5.702214320335374e-06,
      "loss": 0.0592,
      "step": 48240
    },
    {
      "epoch": 4.60052490234375e-05,
      "model_forward_time": 1.6806824207305908,
      "step": 48240
    },
    {
      "dynamics_loss_avg": 0.05472451020032167,
      "epoch": 4.60052490234375e-05,
      "step": 48240
    },
    {
      "action_loss_avg": 0.0016471248120069504,
      "epoch": 4.60052490234375e-05,
      "step": 48240
    },
    {
      "epoch": 4.60052490234375e-05,
      "step": 48240,
      "training_step_time": 7.749231338500977
    },
    {
      "epoch": 4.6006202697753906e-05,
      "model_forward_time": 1.6889350414276123,
      "step": 48241
    },
    {
      "epoch": 4.6006202697753906e-05,
      "step": 48241,
      "training_step_time": 7.852246522903442
    },
    {
      "epoch": 4.600715637207031e-05,
      "model_forward_time": 1.7040388584136963,
      "step": 48242
    },
    {
      "epoch": 4.600715637207031e-05,
      "step": 48242,
      "training_step_time": 8.147764444351196
    },
    {
      "epoch": 4.600811004638672e-05,
      "model_forward_time": 1.6866049766540527,
      "step": 48243
    },
    {
      "epoch": 4.600811004638672e-05,
      "step": 48243,
      "training_step_time": 7.817467927932739
    },
    {
      "epoch": 4.6009063720703125e-05,
      "model_forward_time": 1.6975288391113281,
      "step": 48244
    },
    {
      "epoch": 4.6009063720703125e-05,
      "step": 48244,
      "training_step_time": 7.85058069229126
    },
    {
      "epoch": 4.601001739501953e-05,
      "model_forward_time": 1.6801223754882812,
      "step": 48245
    },
    {
      "epoch": 4.601001739501953e-05,
      "step": 48245,
      "training_step_time": 7.956689357757568
    },
    {
      "epoch": 4.601097106933594e-05,
      "model_forward_time": 1.6844770908355713,
      "step": 48246
    },
    {
      "epoch": 4.601097106933594e-05,
      "step": 48246,
      "training_step_time": 9.116091251373291
    },
    {
      "epoch": 4.6011924743652344e-05,
      "model_forward_time": 2.2517776489257812,
      "step": 48247
    },
    {
      "epoch": 4.6011924743652344e-05,
      "step": 48247,
      "training_step_time": 8.729838132858276
    },
    {
      "epoch": 4.601287841796875e-05,
      "model_forward_time": 1.6858360767364502,
      "step": 48248
    },
    {
      "epoch": 4.601287841796875e-05,
      "step": 48248,
      "training_step_time": 7.981362581253052
    },
    {
      "epoch": 4.601383209228516e-05,
      "model_forward_time": 1.688781499862671,
      "step": 48249
    },
    {
      "epoch": 4.601383209228516e-05,
      "step": 48249,
      "training_step_time": 7.828770637512207
    },
    {
      "epoch": 4.601478576660156e-05,
      "grad_norm": 0.10790126025676727,
      "learning_rate": 5.700577200046051e-06,
      "loss": 0.0549,
      "step": 48250
    },
    {
      "epoch": 4.601478576660156e-05,
      "model_forward_time": 1.6809265613555908,
      "step": 48250
    },
    {
      "dynamics_loss_avg": 0.05007172478362918,
      "epoch": 4.601478576660156e-05,
      "step": 48250
    },
    {
      "action_loss_avg": 0.001898240193258971,
      "epoch": 4.601478576660156e-05,
      "step": 48250
    },
    {
      "epoch": 4.601478576660156e-05,
      "step": 48250,
      "training_step_time": 7.940720081329346
    },
    {
      "epoch": 4.6015739440917967e-05,
      "model_forward_time": 1.697859525680542,
      "step": 48251
    },
    {
      "epoch": 4.6015739440917967e-05,
      "step": 48251,
      "training_step_time": 7.815743684768677
    },
    {
      "epoch": 4.601669311523438e-05,
      "model_forward_time": 1.6945838928222656,
      "step": 48252
    },
    {
      "epoch": 4.601669311523438e-05,
      "step": 48252,
      "training_step_time": 7.828970432281494
    },
    {
      "epoch": 4.601764678955078e-05,
      "model_forward_time": 1.3161592483520508,
      "step": 48253
    },
    {
      "epoch": 4.601764678955078e-05,
      "step": 48253,
      "training_step_time": 7.742536544799805
    },
    {
      "epoch": 4.6018600463867185e-05,
      "model_forward_time": 1.6863832473754883,
      "step": 48254
    },
    {
      "epoch": 4.6018600463867185e-05,
      "step": 48254,
      "training_step_time": 7.843631029129028
    },
    {
      "epoch": 4.6019554138183596e-05,
      "model_forward_time": 1.6970396041870117,
      "step": 48255
    },
    {
      "epoch": 4.6019554138183596e-05,
      "step": 48255,
      "training_step_time": 7.876155376434326
    },
    {
      "epoch": 4.60205078125e-05,
      "model_forward_time": 1.6832678318023682,
      "step": 48256
    },
    {
      "epoch": 4.60205078125e-05,
      "step": 48256,
      "training_step_time": 8.798710107803345
    },
    {
      "epoch": 4.6021461486816404e-05,
      "model_forward_time": 2.238424062728882,
      "step": 48257
    },
    {
      "epoch": 4.6021461486816404e-05,
      "step": 48257,
      "training_step_time": 8.584908723831177
    },
    {
      "epoch": 4.6022415161132815e-05,
      "model_forward_time": 1.687422513961792,
      "step": 48258
    },
    {
      "epoch": 4.6022415161132815e-05,
      "step": 48258,
      "training_step_time": 7.803363561630249
    },
    {
      "epoch": 4.602336883544922e-05,
      "model_forward_time": 1.6911563873291016,
      "step": 48259
    },
    {
      "epoch": 4.602336883544922e-05,
      "step": 48259,
      "training_step_time": 7.935611248016357
    },
    {
      "epoch": 4.602432250976562e-05,
      "grad_norm": 0.11149264127016068,
      "learning_rate": 5.698940003142657e-06,
      "loss": 0.0591,
      "step": 48260
    },
    {
      "epoch": 4.602432250976562e-05,
      "model_forward_time": 1.7053608894348145,
      "step": 48260
    },
    {
      "dynamics_loss_avg": 0.06703420374542475,
      "epoch": 4.602432250976562e-05,
      "step": 48260
    },
    {
      "action_loss_avg": 0.0033068053337046875,
      "epoch": 4.602432250976562e-05,
      "step": 48260
    },
    {
      "epoch": 4.602432250976562e-05,
      "step": 48260,
      "training_step_time": 7.887246608734131
    },
    {
      "epoch": 4.6025276184082034e-05,
      "model_forward_time": 1.6961219310760498,
      "step": 48261
    },
    {
      "epoch": 4.6025276184082034e-05,
      "step": 48261,
      "training_step_time": 7.960745096206665
    },
    {
      "epoch": 4.602622985839844e-05,
      "model_forward_time": 1.6915032863616943,
      "step": 48262
    },
    {
      "epoch": 4.602622985839844e-05,
      "step": 48262,
      "training_step_time": 7.591110944747925
    },
    {
      "epoch": 4.602718353271484e-05,
      "model_forward_time": 1.7101304531097412,
      "step": 48263
    },
    {
      "epoch": 4.602718353271484e-05,
      "step": 48263,
      "training_step_time": 8.411613464355469
    },
    {
      "epoch": 4.602813720703125e-05,
      "model_forward_time": 1.6775481700897217,
      "step": 48264
    },
    {
      "epoch": 4.602813720703125e-05,
      "step": 48264,
      "training_step_time": 7.774668455123901
    },
    {
      "epoch": 4.602909088134766e-05,
      "model_forward_time": 1.6867468357086182,
      "step": 48265
    },
    {
      "epoch": 4.602909088134766e-05,
      "step": 48265,
      "training_step_time": 8.853723287582397
    },
    {
      "epoch": 4.603004455566406e-05,
      "model_forward_time": 2.2214343547821045,
      "step": 48266
    },
    {
      "epoch": 4.603004455566406e-05,
      "step": 48266,
      "training_step_time": 9.152865648269653
    },
    {
      "epoch": 4.603099822998047e-05,
      "model_forward_time": 1.6868162155151367,
      "step": 48267
    },
    {
      "epoch": 4.603099822998047e-05,
      "step": 48267,
      "training_step_time": 7.800949811935425
    },
    {
      "epoch": 4.6031951904296876e-05,
      "model_forward_time": 1.6877267360687256,
      "step": 48268
    },
    {
      "epoch": 4.6031951904296876e-05,
      "step": 48268,
      "training_step_time": 7.7375428676605225
    },
    {
      "epoch": 4.603290557861328e-05,
      "model_forward_time": 1.6917462348937988,
      "step": 48269
    },
    {
      "epoch": 4.603290557861328e-05,
      "step": 48269,
      "training_step_time": 7.8722851276397705
    },
    {
      "epoch": 4.603385925292969e-05,
      "grad_norm": 0.11204817146062851,
      "learning_rate": 5.697302729804236e-06,
      "loss": 0.0595,
      "step": 48270
    },
    {
      "epoch": 4.603385925292969e-05,
      "model_forward_time": 1.6865770816802979,
      "step": 48270
    },
    {
      "dynamics_loss_avg": 0.07112546376883984,
      "epoch": 4.603385925292969e-05,
      "step": 48270
    },
    {
      "action_loss_avg": 0.0024077111214865,
      "epoch": 4.603385925292969e-05,
      "step": 48270
    },
    {
      "epoch": 4.603385925292969e-05,
      "step": 48270,
      "training_step_time": 7.911834239959717
    },
    {
      "epoch": 4.6034812927246095e-05,
      "model_forward_time": 1.6929168701171875,
      "step": 48271
    },
    {
      "epoch": 4.6034812927246095e-05,
      "step": 48271,
      "training_step_time": 7.5275795459747314
    },
    {
      "epoch": 4.60357666015625e-05,
      "model_forward_time": 1.6999905109405518,
      "step": 48272
    },
    {
      "epoch": 4.60357666015625e-05,
      "step": 48272,
      "training_step_time": 7.807256460189819
    },
    {
      "epoch": 4.603672027587891e-05,
      "model_forward_time": 1.6954617500305176,
      "step": 48273
    },
    {
      "epoch": 4.603672027587891e-05,
      "step": 48273,
      "training_step_time": 7.867084264755249
    },
    {
      "epoch": 4.603767395019531e-05,
      "model_forward_time": 1.6905338764190674,
      "step": 48274
    },
    {
      "epoch": 4.603767395019531e-05,
      "step": 48274,
      "training_step_time": 7.878716230392456
    },
    {
      "epoch": 4.603862762451172e-05,
      "model_forward_time": 2.2242276668548584,
      "step": 48275
    },
    {
      "epoch": 4.603862762451172e-05,
      "step": 48275,
      "training_step_time": 8.845308542251587
    },
    {
      "epoch": 4.603958129882813e-05,
      "model_forward_time": 1.6919019222259521,
      "step": 48276
    },
    {
      "epoch": 4.603958129882813e-05,
      "step": 48276,
      "training_step_time": 8.913130760192871
    },
    {
      "epoch": 4.604053497314453e-05,
      "model_forward_time": 1.6919121742248535,
      "step": 48277
    },
    {
      "epoch": 4.604053497314453e-05,
      "step": 48277,
      "training_step_time": 9.0420560836792
    },
    {
      "epoch": 4.6041488647460936e-05,
      "model_forward_time": 1.6969735622406006,
      "step": 48278
    },
    {
      "epoch": 4.6041488647460936e-05,
      "step": 48278,
      "training_step_time": 8.657444477081299
    },
    {
      "epoch": 4.604244232177735e-05,
      "model_forward_time": 1.7084736824035645,
      "step": 48279
    },
    {
      "epoch": 4.604244232177735e-05,
      "step": 48279,
      "training_step_time": 8.446957349777222
    },
    {
      "epoch": 4.604339599609375e-05,
      "grad_norm": 0.1323048323392868,
      "learning_rate": 5.695665380209837e-06,
      "loss": 0.0567,
      "step": 48280
    },
    {
      "epoch": 4.604339599609375e-05,
      "model_forward_time": 1.6891875267028809,
      "step": 48280
    },
    {
      "dynamics_loss_avg": 0.050653505325317386,
      "epoch": 4.604339599609375e-05,
      "step": 48280
    },
    {
      "action_loss_avg": 0.001524522906402126,
      "epoch": 4.604339599609375e-05,
      "step": 48280
    },
    {
      "epoch": 4.604339599609375e-05,
      "step": 48280,
      "training_step_time": 7.737001180648804
    },
    {
      "epoch": 4.6044349670410155e-05,
      "model_forward_time": 1.6919457912445068,
      "step": 48281
    },
    {
      "epoch": 4.6044349670410155e-05,
      "step": 48281,
      "training_step_time": 8.656059503555298
    },
    {
      "epoch": 4.604530334472656e-05,
      "model_forward_time": 1.718137502670288,
      "step": 48282
    },
    {
      "epoch": 4.604530334472656e-05,
      "step": 48282,
      "training_step_time": 8.45004153251648
    },
    {
      "epoch": 4.604625701904297e-05,
      "model_forward_time": 1.7363958358764648,
      "step": 48283
    },
    {
      "epoch": 4.604625701904297e-05,
      "step": 48283,
      "training_step_time": 7.899170637130737
    },
    {
      "epoch": 4.6047210693359374e-05,
      "model_forward_time": 2.99641489982605,
      "step": 48284
    },
    {
      "epoch": 4.6047210693359374e-05,
      "step": 48284,
      "training_step_time": 7.960791110992432
    },
    {
      "epoch": 4.604816436767578e-05,
      "model_forward_time": 1.7383010387420654,
      "step": 48285
    },
    {
      "epoch": 4.604816436767578e-05,
      "step": 48285,
      "training_step_time": 9.393684148788452
    },
    {
      "epoch": 4.604911804199219e-05,
      "model_forward_time": 1.7392988204956055,
      "step": 48286
    },
    {
      "epoch": 4.604911804199219e-05,
      "step": 48286,
      "training_step_time": 8.802913427352905
    },
    {
      "epoch": 4.605007171630859e-05,
      "model_forward_time": 1.7410881519317627,
      "step": 48287
    },
    {
      "epoch": 4.605007171630859e-05,
      "step": 48287,
      "training_step_time": 7.828066825866699
    },
    {
      "epoch": 4.6051025390625e-05,
      "model_forward_time": 1.7291958332061768,
      "step": 48288
    },
    {
      "epoch": 4.6051025390625e-05,
      "step": 48288,
      "training_step_time": 7.839851140975952
    },
    {
      "epoch": 4.605197906494141e-05,
      "model_forward_time": 1.7561979293823242,
      "step": 48289
    },
    {
      "epoch": 4.605197906494141e-05,
      "step": 48289,
      "training_step_time": 8.291402339935303
    },
    {
      "epoch": 4.605293273925781e-05,
      "grad_norm": 0.10468076169490814,
      "learning_rate": 5.694027954538517e-06,
      "loss": 0.0646,
      "step": 48290
    },
    {
      "epoch": 4.605293273925781e-05,
      "model_forward_time": 1.7567658424377441,
      "step": 48290
    },
    {
      "dynamics_loss_avg": 0.0734854880720377,
      "epoch": 4.605293273925781e-05,
      "step": 48290
    },
    {
      "action_loss_avg": 0.0017753314139554277,
      "epoch": 4.605293273925781e-05,
      "step": 48290
    },
    {
      "epoch": 4.605293273925781e-05,
      "step": 48290,
      "training_step_time": 8.331830263137817
    },
    {
      "epoch": 4.6053886413574216e-05,
      "model_forward_time": 1.758774757385254,
      "step": 48291
    },
    {
      "epoch": 4.6053886413574216e-05,
      "step": 48291,
      "training_step_time": 8.423795938491821
    },
    {
      "epoch": 4.6054840087890627e-05,
      "model_forward_time": 1.7530395984649658,
      "step": 48292
    },
    {
      "epoch": 4.6054840087890627e-05,
      "step": 48292,
      "training_step_time": 8.400540590286255
    },
    {
      "epoch": 4.605579376220703e-05,
      "model_forward_time": 2.943760871887207,
      "step": 48293
    },
    {
      "epoch": 4.605579376220703e-05,
      "step": 48293,
      "training_step_time": 9.640079975128174
    },
    {
      "epoch": 4.6056747436523435e-05,
      "model_forward_time": 1.7570524215698242,
      "step": 48294
    },
    {
      "epoch": 4.6056747436523435e-05,
      "step": 48294,
      "training_step_time": 9.394087314605713
    },
    {
      "epoch": 4.6057701110839845e-05,
      "model_forward_time": 1.7785758972167969,
      "step": 48295
    },
    {
      "epoch": 4.6057701110839845e-05,
      "step": 48295,
      "training_step_time": 8.655387878417969
    },
    {
      "epoch": 4.605865478515625e-05,
      "model_forward_time": 1.7570338249206543,
      "step": 48296
    },
    {
      "epoch": 4.605865478515625e-05,
      "step": 48296,
      "training_step_time": 8.825907707214355
    },
    {
      "epoch": 4.6059608459472653e-05,
      "model_forward_time": 1.7587172985076904,
      "step": 48297
    },
    {
      "epoch": 4.6059608459472653e-05,
      "step": 48297,
      "training_step_time": 7.816682815551758
    },
    {
      "epoch": 4.6060562133789064e-05,
      "model_forward_time": 1.7508466243743896,
      "step": 48298
    },
    {
      "epoch": 4.6060562133789064e-05,
      "step": 48298,
      "training_step_time": 7.486074209213257
    },
    {
      "epoch": 4.606151580810547e-05,
      "model_forward_time": 1.7565350532531738,
      "step": 48299
    },
    {
      "epoch": 4.606151580810547e-05,
      "step": 48299,
      "training_step_time": 8.240949630737305
    },
    {
      "epoch": 4.606246948242187e-05,
      "grad_norm": 0.1354975402355194,
      "learning_rate": 5.692390452969344e-06,
      "loss": 0.0561,
      "step": 48300
    },
    {
      "epoch": 4.606246948242187e-05,
      "model_forward_time": 1.3932216167449951,
      "step": 48300
    },
    {
      "dynamics_loss_avg": 0.05230292771011591,
      "epoch": 4.606246948242187e-05,
      "step": 48300
    },
    {
      "action_loss_avg": 0.0015502599126193672,
      "epoch": 4.606246948242187e-05,
      "step": 48300
    },
    {
      "epoch": 4.606246948242187e-05,
      "step": 48300,
      "training_step_time": 7.8147149085998535
    },
    {
      "epoch": 4.606342315673828e-05,
      "model_forward_time": 1.7386047840118408,
      "step": 48301
    },
    {
      "epoch": 4.606342315673828e-05,
      "step": 48301,
      "training_step_time": 7.841590166091919
    },
    {
      "epoch": 4.606437683105469e-05,
      "model_forward_time": 2.9536356925964355,
      "step": 48302
    },
    {
      "epoch": 4.606437683105469e-05,
      "step": 48302,
      "training_step_time": 8.46425485610962
    },
    {
      "epoch": 4.606533050537109e-05,
      "model_forward_time": 1.8068530559539795,
      "step": 48303
    },
    {
      "epoch": 4.606533050537109e-05,
      "step": 48303,
      "training_step_time": 7.864537715911865
    },
    {
      "epoch": 4.60662841796875e-05,
      "model_forward_time": 1.7521090507507324,
      "step": 48304
    },
    {
      "epoch": 4.60662841796875e-05,
      "step": 48304,
      "training_step_time": 7.723133087158203
    },
    {
      "epoch": 4.6067237854003906e-05,
      "model_forward_time": 1.4018316268920898,
      "step": 48305
    },
    {
      "epoch": 4.6067237854003906e-05,
      "step": 48305,
      "training_step_time": 8.3009774684906
    },
    {
      "epoch": 4.606819152832031e-05,
      "model_forward_time": 1.7476422786712646,
      "step": 48306
    },
    {
      "epoch": 4.606819152832031e-05,
      "step": 48306,
      "training_step_time": 7.842285633087158
    },
    {
      "epoch": 4.606914520263672e-05,
      "model_forward_time": 1.7525694370269775,
      "step": 48307
    },
    {
      "epoch": 4.606914520263672e-05,
      "step": 48307,
      "training_step_time": 7.930859327316284
    },
    {
      "epoch": 4.6070098876953125e-05,
      "model_forward_time": 1.746596097946167,
      "step": 48308
    },
    {
      "epoch": 4.6070098876953125e-05,
      "step": 48308,
      "training_step_time": 7.7919909954071045
    },
    {
      "epoch": 4.607105255126953e-05,
      "model_forward_time": 1.7572119235992432,
      "step": 48309
    },
    {
      "epoch": 4.607105255126953e-05,
      "step": 48309,
      "training_step_time": 7.825629472732544
    },
    {
      "epoch": 4.607200622558594e-05,
      "grad_norm": 0.09115287661552429,
      "learning_rate": 5.690752875681392e-06,
      "loss": 0.0592,
      "step": 48310
    },
    {
      "epoch": 4.607200622558594e-05,
      "model_forward_time": 1.7536628246307373,
      "step": 48310
    },
    {
      "dynamics_loss_avg": 0.05805902490392327,
      "epoch": 4.607200622558594e-05,
      "step": 48310
    },
    {
      "action_loss_avg": 0.0016614065563771874,
      "epoch": 4.607200622558594e-05,
      "step": 48310
    },
    {
      "epoch": 4.607200622558594e-05,
      "step": 48310,
      "training_step_time": 7.867156267166138
    },
    {
      "epoch": 4.6072959899902344e-05,
      "model_forward_time": 2.563523054122925,
      "step": 48311
    },
    {
      "epoch": 4.6072959899902344e-05,
      "step": 48311,
      "training_step_time": 7.920429944992065
    },
    {
      "epoch": 4.607391357421875e-05,
      "model_forward_time": 1.8199093341827393,
      "step": 48312
    },
    {
      "epoch": 4.607391357421875e-05,
      "step": 48312,
      "training_step_time": 7.9784791469573975
    },
    {
      "epoch": 4.607486724853516e-05,
      "model_forward_time": 1.752931833267212,
      "step": 48313
    },
    {
      "epoch": 4.607486724853516e-05,
      "step": 48313,
      "training_step_time": 8.846111297607422
    },
    {
      "epoch": 4.607582092285156e-05,
      "model_forward_time": 1.4028429985046387,
      "step": 48314
    },
    {
      "epoch": 4.607582092285156e-05,
      "step": 48314,
      "training_step_time": 7.84132194519043
    },
    {
      "epoch": 4.6076774597167967e-05,
      "model_forward_time": 1.4020211696624756,
      "step": 48315
    },
    {
      "epoch": 4.6076774597167967e-05,
      "step": 48315,
      "training_step_time": 7.900456190109253
    },
    {
      "epoch": 4.607772827148438e-05,
      "model_forward_time": 1.7586536407470703,
      "step": 48316
    },
    {
      "epoch": 4.607772827148438e-05,
      "step": 48316,
      "training_step_time": 7.865083932876587
    },
    {
      "epoch": 4.607868194580078e-05,
      "model_forward_time": 1.7574362754821777,
      "step": 48317
    },
    {
      "epoch": 4.607868194580078e-05,
      "step": 48317,
      "training_step_time": 7.782214641571045
    },
    {
      "epoch": 4.6079635620117185e-05,
      "model_forward_time": 1.7557542324066162,
      "step": 48318
    },
    {
      "epoch": 4.6079635620117185e-05,
      "step": 48318,
      "training_step_time": 8.345476627349854
    },
    {
      "epoch": 4.6080589294433596e-05,
      "model_forward_time": 1.7544877529144287,
      "step": 48319
    },
    {
      "epoch": 4.6080589294433596e-05,
      "step": 48319,
      "training_step_time": 9.294872045516968
    },
    {
      "epoch": 4.608154296875e-05,
      "grad_norm": 0.12949344515800476,
      "learning_rate": 5.689115222853744e-06,
      "loss": 0.0606,
      "step": 48320
    },
    {
      "epoch": 4.608154296875e-05,
      "model_forward_time": 1.7578818798065186,
      "step": 48320
    },
    {
      "dynamics_loss_avg": 0.06269006337970495,
      "epoch": 4.608154296875e-05,
      "step": 48320
    },
    {
      "action_loss_avg": 0.0025371268042363227,
      "epoch": 4.608154296875e-05,
      "step": 48320
    },
    {
      "epoch": 4.608154296875e-05,
      "step": 48320,
      "training_step_time": 7.832395315170288
    },
    {
      "epoch": 4.6082496643066404e-05,
      "model_forward_time": 2.935393810272217,
      "step": 48321
    },
    {
      "epoch": 4.6082496643066404e-05,
      "step": 48321,
      "training_step_time": 7.897912979125977
    },
    {
      "epoch": 4.6083450317382815e-05,
      "model_forward_time": 1.7600438594818115,
      "step": 48322
    },
    {
      "epoch": 4.6083450317382815e-05,
      "step": 48322,
      "training_step_time": 8.572447776794434
    },
    {
      "epoch": 4.608440399169922e-05,
      "model_forward_time": 1.762127161026001,
      "step": 48323
    },
    {
      "epoch": 4.608440399169922e-05,
      "step": 48323,
      "training_step_time": 7.853280067443848
    },
    {
      "epoch": 4.608535766601562e-05,
      "model_forward_time": 1.752671480178833,
      "step": 48324
    },
    {
      "epoch": 4.608535766601562e-05,
      "step": 48324,
      "training_step_time": 7.7626166343688965
    },
    {
      "epoch": 4.6086311340332034e-05,
      "model_forward_time": 1.7760369777679443,
      "step": 48325
    },
    {
      "epoch": 4.6086311340332034e-05,
      "step": 48325,
      "training_step_time": 7.825936794281006
    },
    {
      "epoch": 4.608726501464844e-05,
      "model_forward_time": 1.7632780075073242,
      "step": 48326
    },
    {
      "epoch": 4.608726501464844e-05,
      "step": 48326,
      "training_step_time": 7.7506067752838135
    },
    {
      "epoch": 4.608821868896484e-05,
      "model_forward_time": 1.7517106533050537,
      "step": 48327
    },
    {
      "epoch": 4.608821868896484e-05,
      "step": 48327,
      "training_step_time": 7.748796701431274
    },
    {
      "epoch": 4.608917236328125e-05,
      "model_forward_time": 1.3987059593200684,
      "step": 48328
    },
    {
      "epoch": 4.608917236328125e-05,
      "step": 48328,
      "training_step_time": 7.84384560585022
    },
    {
      "epoch": 4.609012603759766e-05,
      "model_forward_time": 1.760716199874878,
      "step": 48329
    },
    {
      "epoch": 4.609012603759766e-05,
      "step": 48329,
      "training_step_time": 9.109012126922607
    },
    {
      "epoch": 4.609107971191406e-05,
      "grad_norm": 0.15983352065086365,
      "learning_rate": 5.687477494665492e-06,
      "loss": 0.0596,
      "step": 48330
    },
    {
      "epoch": 4.609107971191406e-05,
      "model_forward_time": 2.949613332748413,
      "step": 48330
    },
    {
      "dynamics_loss_avg": 0.07361813336610794,
      "epoch": 4.609107971191406e-05,
      "step": 48330
    },
    {
      "action_loss_avg": 0.0031209106789901854,
      "epoch": 4.609107971191406e-05,
      "step": 48330
    },
    {
      "epoch": 4.609107971191406e-05,
      "step": 48330,
      "training_step_time": 8.41825819015503
    },
    {
      "epoch": 4.609203338623047e-05,
      "model_forward_time": 1.7634477615356445,
      "step": 48331
    },
    {
      "epoch": 4.609203338623047e-05,
      "step": 48331,
      "training_step_time": 10.149616479873657
    },
    {
      "epoch": 4.6092987060546876e-05,
      "model_forward_time": 1.7478642463684082,
      "step": 48332
    },
    {
      "epoch": 4.6092987060546876e-05,
      "step": 48332,
      "training_step_time": 8.180881261825562
    },
    {
      "epoch": 4.609394073486328e-05,
      "model_forward_time": 1.7602152824401855,
      "step": 48333
    },
    {
      "epoch": 4.609394073486328e-05,
      "step": 48333,
      "training_step_time": 8.693959951400757
    },
    {
      "epoch": 4.609489440917969e-05,
      "model_forward_time": 1.7489757537841797,
      "step": 48334
    },
    {
      "epoch": 4.609489440917969e-05,
      "step": 48334,
      "training_step_time": 7.820477247238159
    },
    {
      "epoch": 4.6095848083496095e-05,
      "model_forward_time": 1.762082815170288,
      "step": 48335
    },
    {
      "epoch": 4.6095848083496095e-05,
      "step": 48335,
      "training_step_time": 7.8666088581085205
    },
    {
      "epoch": 4.60968017578125e-05,
      "model_forward_time": 1.7464468479156494,
      "step": 48336
    },
    {
      "epoch": 4.60968017578125e-05,
      "step": 48336,
      "training_step_time": 7.678193092346191
    },
    {
      "epoch": 4.609775543212891e-05,
      "model_forward_time": 1.7478582859039307,
      "step": 48337
    },
    {
      "epoch": 4.609775543212891e-05,
      "step": 48337,
      "training_step_time": 7.844781398773193
    },
    {
      "epoch": 4.6098709106445313e-05,
      "model_forward_time": 1.7579596042633057,
      "step": 48338
    },
    {
      "epoch": 4.6098709106445313e-05,
      "step": 48338,
      "training_step_time": 7.853496074676514
    },
    {
      "epoch": 4.609966278076172e-05,
      "model_forward_time": 2.9559218883514404,
      "step": 48339
    },
    {
      "epoch": 4.609966278076172e-05,
      "step": 48339,
      "training_step_time": 8.684724807739258
    },
    {
      "epoch": 4.610061645507813e-05,
      "grad_norm": 0.09142874926328659,
      "learning_rate": 5.685839691295734e-06,
      "loss": 0.0603,
      "step": 48340
    },
    {
      "epoch": 4.610061645507813e-05,
      "model_forward_time": 1.752807855606079,
      "step": 48340
    },
    {
      "dynamics_loss_avg": 0.04284931104630232,
      "epoch": 4.610061645507813e-05,
      "step": 48340
    },
    {
      "action_loss_avg": 0.001385236598434858,
      "epoch": 4.610061645507813e-05,
      "step": 48340
    },
    {
      "epoch": 4.610061645507813e-05,
      "step": 48340,
      "training_step_time": 7.928171157836914
    },
    {
      "epoch": 4.610157012939453e-05,
      "model_forward_time": 1.769930124282837,
      "step": 48341
    },
    {
      "epoch": 4.610157012939453e-05,
      "step": 48341,
      "training_step_time": 7.9120705127716064
    },
    {
      "epoch": 4.6102523803710936e-05,
      "model_forward_time": 1.755462408065796,
      "step": 48342
    },
    {
      "epoch": 4.6102523803710936e-05,
      "step": 48342,
      "training_step_time": 7.86573600769043
    },
    {
      "epoch": 4.610347747802735e-05,
      "model_forward_time": 1.7581799030303955,
      "step": 48343
    },
    {
      "epoch": 4.610347747802735e-05,
      "step": 48343,
      "training_step_time": 8.336019515991211
    },
    {
      "epoch": 4.610443115234375e-05,
      "model_forward_time": 1.7439870834350586,
      "step": 48344
    },
    {
      "epoch": 4.610443115234375e-05,
      "step": 48344,
      "training_step_time": 7.865274429321289
    },
    {
      "epoch": 4.6105384826660155e-05,
      "model_forward_time": 1.7547693252563477,
      "step": 48345
    },
    {
      "epoch": 4.6105384826660155e-05,
      "step": 48345,
      "training_step_time": 7.717994689941406
    },
    {
      "epoch": 4.610633850097656e-05,
      "model_forward_time": 1.7533400058746338,
      "step": 48346
    },
    {
      "epoch": 4.610633850097656e-05,
      "step": 48346,
      "training_step_time": 7.874006986618042
    },
    {
      "epoch": 4.610729217529297e-05,
      "model_forward_time": 1.7529704570770264,
      "step": 48347
    },
    {
      "epoch": 4.610729217529297e-05,
      "step": 48347,
      "training_step_time": 7.867290258407593
    },
    {
      "epoch": 4.6108245849609374e-05,
      "model_forward_time": 2.897226333618164,
      "step": 48348
    },
    {
      "epoch": 4.6108245849609374e-05,
      "step": 48348,
      "training_step_time": 9.099940538406372
    },
    {
      "epoch": 4.610919952392578e-05,
      "model_forward_time": 1.8257360458374023,
      "step": 48349
    },
    {
      "epoch": 4.610919952392578e-05,
      "step": 48349,
      "training_step_time": 8.486054420471191
    },
    {
      "epoch": 4.611015319824219e-05,
      "grad_norm": 0.1802120953798294,
      "learning_rate": 5.684201812923578e-06,
      "loss": 0.0576,
      "step": 48350
    },
    {
      "epoch": 4.611015319824219e-05,
      "model_forward_time": 1.7544682025909424,
      "step": 48350
    },
    {
      "dynamics_loss_avg": 0.06287683472037316,
      "epoch": 4.611015319824219e-05,
      "step": 48350
    },
    {
      "action_loss_avg": 0.0022217902762349693,
      "epoch": 4.611015319824219e-05,
      "step": 48350
    },
    {
      "epoch": 4.611015319824219e-05,
      "step": 48350,
      "training_step_time": 7.78404974937439
    },
    {
      "epoch": 4.611110687255859e-05,
      "model_forward_time": 1.7558856010437012,
      "step": 48351
    },
    {
      "epoch": 4.611110687255859e-05,
      "step": 48351,
      "training_step_time": 7.949681043624878
    },
    {
      "epoch": 4.6112060546875e-05,
      "model_forward_time": 1.7440459728240967,
      "step": 48352
    },
    {
      "epoch": 4.6112060546875e-05,
      "step": 48352,
      "training_step_time": 8.355026245117188
    },
    {
      "epoch": 4.611301422119141e-05,
      "model_forward_time": 1.7516250610351562,
      "step": 48353
    },
    {
      "epoch": 4.611301422119141e-05,
      "step": 48353,
      "training_step_time": 7.777495622634888
    },
    {
      "epoch": 4.611396789550781e-05,
      "model_forward_time": 1.7516098022460938,
      "step": 48354
    },
    {
      "epoch": 4.611396789550781e-05,
      "step": 48354,
      "training_step_time": 7.564256429672241
    },
    {
      "epoch": 4.6114921569824216e-05,
      "model_forward_time": 1.7607700824737549,
      "step": 48355
    },
    {
      "epoch": 4.6114921569824216e-05,
      "step": 48355,
      "training_step_time": 8.629163026809692
    },
    {
      "epoch": 4.6115875244140627e-05,
      "model_forward_time": 1.7701656818389893,
      "step": 48356
    },
    {
      "epoch": 4.6115875244140627e-05,
      "step": 48356,
      "training_step_time": 7.70840048789978
    },
    {
      "epoch": 4.611682891845703e-05,
      "model_forward_time": 2.8728067874908447,
      "step": 48357
    },
    {
      "epoch": 4.611682891845703e-05,
      "step": 48357,
      "training_step_time": 7.783396005630493
    },
    {
      "epoch": 4.6117782592773435e-05,
      "model_forward_time": 1.8088386058807373,
      "step": 48358
    },
    {
      "epoch": 4.6117782592773435e-05,
      "step": 48358,
      "training_step_time": 7.934026002883911
    },
    {
      "epoch": 4.6118736267089845e-05,
      "model_forward_time": 1.7664010524749756,
      "step": 48359
    },
    {
      "epoch": 4.6118736267089845e-05,
      "step": 48359,
      "training_step_time": 9.274636268615723
    },
    {
      "epoch": 4.611968994140625e-05,
      "grad_norm": 0.11909427493810654,
      "learning_rate": 5.682563859728141e-06,
      "loss": 0.0578,
      "step": 48360
    },
    {
      "epoch": 4.611968994140625e-05,
      "model_forward_time": 1.7577743530273438,
      "step": 48360
    },
    {
      "dynamics_loss_avg": 0.07284681163728238,
      "epoch": 4.611968994140625e-05,
      "step": 48360
    },
    {
      "action_loss_avg": 0.0024163906608009713,
      "epoch": 4.611968994140625e-05,
      "step": 48360
    },
    {
      "epoch": 4.611968994140625e-05,
      "step": 48360,
      "training_step_time": 8.403153896331787
    },
    {
      "epoch": 4.6120643615722653e-05,
      "model_forward_time": 1.776886224746704,
      "step": 48361
    },
    {
      "epoch": 4.6120643615722653e-05,
      "step": 48361,
      "training_step_time": 8.667320013046265
    },
    {
      "epoch": 4.6121597290039064e-05,
      "model_forward_time": 1.751948595046997,
      "step": 48362
    },
    {
      "epoch": 4.6121597290039064e-05,
      "step": 48362,
      "training_step_time": 7.899062871932983
    },
    {
      "epoch": 4.612255096435547e-05,
      "model_forward_time": 1.7531764507293701,
      "step": 48363
    },
    {
      "epoch": 4.612255096435547e-05,
      "step": 48363,
      "training_step_time": 7.6819610595703125
    },
    {
      "epoch": 4.612350463867187e-05,
      "model_forward_time": 1.7604320049285889,
      "step": 48364
    },
    {
      "epoch": 4.612350463867187e-05,
      "step": 48364,
      "training_step_time": 8.45510482788086
    },
    {
      "epoch": 4.612445831298828e-05,
      "model_forward_time": 1.7534816265106201,
      "step": 48365
    },
    {
      "epoch": 4.612445831298828e-05,
      "step": 48365,
      "training_step_time": 9.06955885887146
    },
    {
      "epoch": 4.612541198730469e-05,
      "model_forward_time": 1.763974666595459,
      "step": 48366
    },
    {
      "epoch": 4.612541198730469e-05,
      "step": 48366,
      "training_step_time": 9.576013326644897
    },
    {
      "epoch": 4.612636566162109e-05,
      "model_forward_time": 2.975233554840088,
      "step": 48367
    },
    {
      "epoch": 4.612636566162109e-05,
      "step": 48367,
      "training_step_time": 8.657894611358643
    },
    {
      "epoch": 4.61273193359375e-05,
      "model_forward_time": 1.7379302978515625,
      "step": 48368
    },
    {
      "epoch": 4.61273193359375e-05,
      "step": 48368,
      "training_step_time": 8.280843496322632
    },
    {
      "epoch": 4.6128273010253906e-05,
      "model_forward_time": 1.7644333839416504,
      "step": 48369
    },
    {
      "epoch": 4.6128273010253906e-05,
      "step": 48369,
      "training_step_time": 7.821353435516357
    },
    {
      "epoch": 4.612922668457031e-05,
      "grad_norm": 0.12832337617874146,
      "learning_rate": 5.680925831888546e-06,
      "loss": 0.061,
      "step": 48370
    },
    {
      "epoch": 4.612922668457031e-05,
      "model_forward_time": 1.772932767868042,
      "step": 48370
    },
    {
      "dynamics_loss_avg": 0.07646542582660913,
      "epoch": 4.612922668457031e-05,
      "step": 48370
    },
    {
      "action_loss_avg": 0.002592145721428096,
      "epoch": 4.612922668457031e-05,
      "step": 48370
    },
    {
      "epoch": 4.612922668457031e-05,
      "step": 48370,
      "training_step_time": 8.807403564453125
    },
    {
      "epoch": 4.613018035888672e-05,
      "model_forward_time": 1.7673795223236084,
      "step": 48371
    },
    {
      "epoch": 4.613018035888672e-05,
      "step": 48371,
      "training_step_time": 6.945245981216431
    },
    {
      "epoch": 4.6131134033203125e-05,
      "model_forward_time": 1.7684824466705322,
      "step": 48372
    },
    {
      "epoch": 4.6131134033203125e-05,
      "step": 48372,
      "training_step_time": 6.879195690155029
    },
    {
      "epoch": 4.613208770751953e-05,
      "model_forward_time": 1.7542343139648438,
      "step": 48373
    },
    {
      "epoch": 4.613208770751953e-05,
      "step": 48373,
      "training_step_time": 6.902331829071045
    },
    {
      "epoch": 4.613304138183594e-05,
      "model_forward_time": 1.7551438808441162,
      "step": 48374
    },
    {
      "epoch": 4.613304138183594e-05,
      "step": 48374,
      "training_step_time": 6.984186887741089
    },
    {
      "epoch": 4.6133995056152344e-05,
      "model_forward_time": 1.770395040512085,
      "step": 48375
    },
    {
      "epoch": 4.6133995056152344e-05,
      "step": 48375,
      "training_step_time": 6.911972522735596
    },
    {
      "epoch": 4.613494873046875e-05,
      "model_forward_time": 1.758267879486084,
      "step": 48376
    },
    {
      "epoch": 4.613494873046875e-05,
      "step": 48376,
      "training_step_time": 6.929486036300659
    },
    {
      "epoch": 4.613590240478516e-05,
      "model_forward_time": 1.7388958930969238,
      "step": 48377
    },
    {
      "epoch": 4.613590240478516e-05,
      "step": 48377,
      "training_step_time": 6.938643455505371
    },
    {
      "epoch": 4.613685607910156e-05,
      "model_forward_time": 1.75789213180542,
      "step": 48378
    },
    {
      "epoch": 4.613685607910156e-05,
      "step": 48378,
      "training_step_time": 6.927825689315796
    },
    {
      "epoch": 4.613780975341797e-05,
      "model_forward_time": 2.950268030166626,
      "step": 48379
    },
    {
      "epoch": 4.613780975341797e-05,
      "step": 48379,
      "training_step_time": 11.218378067016602
    },
    {
      "epoch": 4.613876342773438e-05,
      "grad_norm": 0.11519110947847366,
      "learning_rate": 5.679287729583927e-06,
      "loss": 0.0614,
      "step": 48380
    },
    {
      "epoch": 4.613876342773438e-05,
      "model_forward_time": 1.7751824855804443,
      "step": 48380
    },
    {
      "dynamics_loss_avg": 0.0570702712982893,
      "epoch": 4.613876342773438e-05,
      "step": 48380
    },
    {
      "action_loss_avg": 0.0018069055746309458,
      "epoch": 4.613876342773438e-05,
      "step": 48380
    },
    {
      "epoch": 4.613876342773438e-05,
      "step": 48380,
      "training_step_time": 7.888701677322388
    },
    {
      "epoch": 4.613971710205078e-05,
      "model_forward_time": 1.7560999393463135,
      "step": 48381
    },
    {
      "epoch": 4.613971710205078e-05,
      "step": 48381,
      "training_step_time": 6.887465000152588
    },
    {
      "epoch": 4.6140670776367185e-05,
      "model_forward_time": 1.762188196182251,
      "step": 48382
    },
    {
      "epoch": 4.6140670776367185e-05,
      "step": 48382,
      "training_step_time": 6.852587938308716
    },
    {
      "epoch": 4.6141624450683596e-05,
      "model_forward_time": 1.763956069946289,
      "step": 48383
    },
    {
      "epoch": 4.6141624450683596e-05,
      "step": 48383,
      "training_step_time": 6.870707750320435
    },
    {
      "epoch": 4.6142578125e-05,
      "model_forward_time": 1.7652161121368408,
      "step": 48384
    },
    {
      "epoch": 4.6142578125e-05,
      "step": 48384,
      "training_step_time": 6.885218381881714
    },
    {
      "epoch": 4.6143531799316404e-05,
      "model_forward_time": 1.7564775943756104,
      "step": 48385
    },
    {
      "epoch": 4.6143531799316404e-05,
      "step": 48385,
      "training_step_time": 6.885544300079346
    },
    {
      "epoch": 4.6144485473632815e-05,
      "model_forward_time": 1.784515380859375,
      "step": 48386
    },
    {
      "epoch": 4.6144485473632815e-05,
      "step": 48386,
      "training_step_time": 6.880156993865967
    },
    {
      "epoch": 4.614543914794922e-05,
      "model_forward_time": 1.7635512351989746,
      "step": 48387
    },
    {
      "epoch": 4.614543914794922e-05,
      "step": 48387,
      "training_step_time": 6.84320330619812
    },
    {
      "epoch": 4.614639282226562e-05,
      "model_forward_time": 2.961289882659912,
      "step": 48388
    },
    {
      "epoch": 4.614639282226562e-05,
      "step": 48388,
      "training_step_time": 9.501519918441772
    },
    {
      "epoch": 4.6147346496582034e-05,
      "model_forward_time": 1.7357261180877686,
      "step": 48389
    },
    {
      "epoch": 4.6147346496582034e-05,
      "step": 48389,
      "training_step_time": 7.888837575912476
    },
    {
      "epoch": 4.614830017089844e-05,
      "grad_norm": 0.11599061638116837,
      "learning_rate": 5.677649552993423e-06,
      "loss": 0.0615,
      "step": 48390
    },
    {
      "epoch": 4.614830017089844e-05,
      "model_forward_time": 1.759568214416504,
      "step": 48390
    },
    {
      "dynamics_loss_avg": 0.048038069251924755,
      "epoch": 4.614830017089844e-05,
      "step": 48390
    },
    {
      "action_loss_avg": 0.002455512504093349,
      "epoch": 4.614830017089844e-05,
      "step": 48390
    },
    {
      "epoch": 4.614830017089844e-05,
      "step": 48390,
      "training_step_time": 6.915689468383789
    },
    {
      "epoch": 4.614925384521484e-05,
      "model_forward_time": 1.77587890625,
      "step": 48391
    },
    {
      "epoch": 4.614925384521484e-05,
      "step": 48391,
      "training_step_time": 6.768256902694702
    },
    {
      "epoch": 4.615020751953125e-05,
      "model_forward_time": 1.7701871395111084,
      "step": 48392
    },
    {
      "epoch": 4.615020751953125e-05,
      "step": 48392,
      "training_step_time": 6.876184940338135
    },
    {
      "epoch": 4.615116119384766e-05,
      "model_forward_time": 1.7490367889404297,
      "step": 48393
    },
    {
      "epoch": 4.615116119384766e-05,
      "step": 48393,
      "training_step_time": 6.895701169967651
    },
    {
      "epoch": 4.615211486816406e-05,
      "model_forward_time": 1.748262882232666,
      "step": 48394
    },
    {
      "epoch": 4.615211486816406e-05,
      "step": 48394,
      "training_step_time": 6.866731882095337
    },
    {
      "epoch": 4.615306854248047e-05,
      "model_forward_time": 1.7617249488830566,
      "step": 48395
    },
    {
      "epoch": 4.615306854248047e-05,
      "step": 48395,
      "training_step_time": 6.85950779914856
    },
    {
      "epoch": 4.6154022216796876e-05,
      "model_forward_time": 1.7619671821594238,
      "step": 48396
    },
    {
      "epoch": 4.6154022216796876e-05,
      "step": 48396,
      "training_step_time": 6.911052227020264
    },
    {
      "epoch": 4.615497589111328e-05,
      "model_forward_time": 2.9643728733062744,
      "step": 48397
    },
    {
      "epoch": 4.615497589111328e-05,
      "step": 48397,
      "training_step_time": 8.930102825164795
    },
    {
      "epoch": 4.615592956542969e-05,
      "model_forward_time": 1.8161749839782715,
      "step": 48398
    },
    {
      "epoch": 4.615592956542969e-05,
      "step": 48398,
      "training_step_time": 9.633248567581177
    },
    {
      "epoch": 4.6156883239746095e-05,
      "model_forward_time": 1.763401985168457,
      "step": 48399
    },
    {
      "epoch": 4.6156883239746095e-05,
      "step": 48399,
      "training_step_time": 7.31432843208313
    },
    {
      "epoch": 4.61578369140625e-05,
      "grad_norm": 0.09129853546619415,
      "learning_rate": 5.676011302296183e-06,
      "loss": 0.0587,
      "step": 48400
    },
    {
      "epoch": 4.61578369140625e-05,
      "model_forward_time": 1.7616941928863525,
      "step": 48400
    },
    {
      "dynamics_loss_avg": 0.06769003681838512,
      "epoch": 4.61578369140625e-05,
      "step": 48400
    },
    {
      "action_loss_avg": 0.0024055838148342445,
      "epoch": 4.61578369140625e-05,
      "step": 48400
    },
    {
      "epoch": 4.61578369140625e-05,
      "step": 48400,
      "training_step_time": 6.9040303230285645
    },
    {
      "epoch": 4.615879058837891e-05,
      "model_forward_time": 1.7635700702667236,
      "step": 48401
    },
    {
      "epoch": 4.615879058837891e-05,
      "step": 48401,
      "training_step_time": 6.931260347366333
    },
    {
      "epoch": 4.6159744262695313e-05,
      "model_forward_time": 1.7504465579986572,
      "step": 48402
    },
    {
      "epoch": 4.6159744262695313e-05,
      "step": 48402,
      "training_step_time": 6.881246566772461
    },
    {
      "epoch": 4.616069793701172e-05,
      "model_forward_time": 1.756037712097168,
      "step": 48403
    },
    {
      "epoch": 4.616069793701172e-05,
      "step": 48403,
      "training_step_time": 6.892793655395508
    },
    {
      "epoch": 4.616165161132813e-05,
      "model_forward_time": 1.8014435768127441,
      "step": 48404
    },
    {
      "epoch": 4.616165161132813e-05,
      "step": 48404,
      "training_step_time": 6.937460899353027
    },
    {
      "epoch": 4.616260528564453e-05,
      "model_forward_time": 1.7605931758880615,
      "step": 48405
    },
    {
      "epoch": 4.616260528564453e-05,
      "step": 48405,
      "training_step_time": 6.972198009490967
    },
    {
      "epoch": 4.6163558959960936e-05,
      "model_forward_time": 2.8840854167938232,
      "step": 48406
    },
    {
      "epoch": 4.6163558959960936e-05,
      "step": 48406,
      "training_step_time": 7.888099431991577
    },
    {
      "epoch": 4.616451263427734e-05,
      "model_forward_time": 1.804382085800171,
      "step": 48407
    },
    {
      "epoch": 4.616451263427734e-05,
      "step": 48407,
      "training_step_time": 8.55202579498291
    },
    {
      "epoch": 4.616546630859375e-05,
      "model_forward_time": 1.758958101272583,
      "step": 48408
    },
    {
      "epoch": 4.616546630859375e-05,
      "step": 48408,
      "training_step_time": 7.822344064712524
    },
    {
      "epoch": 4.6166419982910155e-05,
      "model_forward_time": 1.3976824283599854,
      "step": 48409
    },
    {
      "epoch": 4.6166419982910155e-05,
      "step": 48409,
      "training_step_time": 6.999135494232178
    },
    {
      "epoch": 4.616737365722656e-05,
      "grad_norm": 0.156333327293396,
      "learning_rate": 5.674372977671362e-06,
      "loss": 0.0615,
      "step": 48410
    },
    {
      "epoch": 4.616737365722656e-05,
      "model_forward_time": 1.7564570903778076,
      "step": 48410
    },
    {
      "dynamics_loss_avg": 0.06481075864285231,
      "epoch": 4.616737365722656e-05,
      "step": 48410
    },
    {
      "action_loss_avg": 0.0019903695734683423,
      "epoch": 4.616737365722656e-05,
      "step": 48410
    },
    {
      "epoch": 4.616737365722656e-05,
      "step": 48410,
      "training_step_time": 7.001620292663574
    },
    {
      "epoch": 4.616832733154297e-05,
      "model_forward_time": 1.7621407508850098,
      "step": 48411
    },
    {
      "epoch": 4.616832733154297e-05,
      "step": 48411,
      "training_step_time": 6.8808465003967285
    },
    {
      "epoch": 4.6169281005859374e-05,
      "model_forward_time": 1.761131763458252,
      "step": 48412
    },
    {
      "epoch": 4.6169281005859374e-05,
      "step": 48412,
      "training_step_time": 6.907010555267334
    },
    {
      "epoch": 4.617023468017578e-05,
      "model_forward_time": 1.7589027881622314,
      "step": 48413
    },
    {
      "epoch": 4.617023468017578e-05,
      "step": 48413,
      "training_step_time": 6.895095586776733
    },
    {
      "epoch": 4.617118835449219e-05,
      "model_forward_time": 1.7480595111846924,
      "step": 48414
    },
    {
      "epoch": 4.617118835449219e-05,
      "step": 48414,
      "training_step_time": 6.916059255599976
    },
    {
      "epoch": 4.617214202880859e-05,
      "model_forward_time": 1.7426786422729492,
      "step": 48415
    },
    {
      "epoch": 4.617214202880859e-05,
      "step": 48415,
      "training_step_time": 7.679956912994385
    },
    {
      "epoch": 4.6173095703125e-05,
      "model_forward_time": 3.0003089904785156,
      "step": 48416
    },
    {
      "epoch": 4.6173095703125e-05,
      "step": 48416,
      "training_step_time": 10.4930260181427
    },
    {
      "epoch": 4.617404937744141e-05,
      "model_forward_time": 1.752812385559082,
      "step": 48417
    },
    {
      "epoch": 4.617404937744141e-05,
      "step": 48417,
      "training_step_time": 7.884085178375244
    },
    {
      "epoch": 4.617500305175781e-05,
      "model_forward_time": 1.7625799179077148,
      "step": 48418
    },
    {
      "epoch": 4.617500305175781e-05,
      "step": 48418,
      "training_step_time": 6.930511474609375
    },
    {
      "epoch": 4.6175956726074216e-05,
      "model_forward_time": 1.7553212642669678,
      "step": 48419
    },
    {
      "epoch": 4.6175956726074216e-05,
      "step": 48419,
      "training_step_time": 6.863257884979248
    },
    {
      "epoch": 4.617691040039063e-05,
      "grad_norm": 0.1274520605802536,
      "learning_rate": 5.672734579298126e-06,
      "loss": 0.057,
      "step": 48420
    },
    {
      "epoch": 4.617691040039063e-05,
      "model_forward_time": 1.7599117755889893,
      "step": 48420
    },
    {
      "dynamics_loss_avg": 0.06810519080609083,
      "epoch": 4.617691040039063e-05,
      "step": 48420
    },
    {
      "action_loss_avg": 0.001786107401130721,
      "epoch": 4.617691040039063e-05,
      "step": 48420
    },
    {
      "epoch": 4.617691040039063e-05,
      "step": 48420,
      "training_step_time": 6.916147232055664
    },
    {
      "epoch": 4.617786407470703e-05,
      "model_forward_time": 1.7540953159332275,
      "step": 48421
    },
    {
      "epoch": 4.617786407470703e-05,
      "step": 48421,
      "training_step_time": 6.893301486968994
    },
    {
      "epoch": 4.6178817749023435e-05,
      "model_forward_time": 1.3898334503173828,
      "step": 48422
    },
    {
      "epoch": 4.6178817749023435e-05,
      "step": 48422,
      "training_step_time": 6.867730617523193
    },
    {
      "epoch": 4.6179771423339845e-05,
      "model_forward_time": 1.7647860050201416,
      "step": 48423
    },
    {
      "epoch": 4.6179771423339845e-05,
      "step": 48423,
      "training_step_time": 6.937896966934204
    },
    {
      "epoch": 4.618072509765625e-05,
      "model_forward_time": 1.7602412700653076,
      "step": 48424
    },
    {
      "epoch": 4.618072509765625e-05,
      "step": 48424,
      "training_step_time": 6.788058280944824
    },
    {
      "epoch": 4.6181678771972654e-05,
      "model_forward_time": 2.9618611335754395,
      "step": 48425
    },
    {
      "epoch": 4.6181678771972654e-05,
      "step": 48425,
      "training_step_time": 10.040026426315308
    },
    {
      "epoch": 4.6182632446289064e-05,
      "model_forward_time": 1.8134005069732666,
      "step": 48426
    },
    {
      "epoch": 4.6182632446289064e-05,
      "step": 48426,
      "training_step_time": 7.938758850097656
    },
    {
      "epoch": 4.618358612060547e-05,
      "model_forward_time": 1.7548563480377197,
      "step": 48427
    },
    {
      "epoch": 4.618358612060547e-05,
      "step": 48427,
      "training_step_time": 6.932816505432129
    },
    {
      "epoch": 4.618453979492187e-05,
      "model_forward_time": 1.7528526782989502,
      "step": 48428
    },
    {
      "epoch": 4.618453979492187e-05,
      "step": 48428,
      "training_step_time": 6.907515048980713
    },
    {
      "epoch": 4.618549346923828e-05,
      "model_forward_time": 1.7524003982543945,
      "step": 48429
    },
    {
      "epoch": 4.618549346923828e-05,
      "step": 48429,
      "training_step_time": 6.775351524353027
    },
    {
      "epoch": 4.618644714355469e-05,
      "grad_norm": 0.08394096791744232,
      "learning_rate": 5.671096107355649e-06,
      "loss": 0.0584,
      "step": 48430
    },
    {
      "epoch": 4.618644714355469e-05,
      "model_forward_time": 1.7674262523651123,
      "step": 48430
    },
    {
      "dynamics_loss_avg": 0.0707126684486866,
      "epoch": 4.618644714355469e-05,
      "step": 48430
    },
    {
      "action_loss_avg": 0.0014449837428401225,
      "epoch": 4.618644714355469e-05,
      "step": 48430
    },
    {
      "epoch": 4.618644714355469e-05,
      "step": 48430,
      "training_step_time": 6.765183687210083
    },
    {
      "epoch": 4.618740081787109e-05,
      "model_forward_time": 1.7555639743804932,
      "step": 48431
    },
    {
      "epoch": 4.618740081787109e-05,
      "step": 48431,
      "training_step_time": 6.868466377258301
    },
    {
      "epoch": 4.61883544921875e-05,
      "model_forward_time": 1.7546136379241943,
      "step": 48432
    },
    {
      "epoch": 4.61883544921875e-05,
      "step": 48432,
      "training_step_time": 6.901172161102295
    },
    {
      "epoch": 4.6189308166503906e-05,
      "model_forward_time": 1.394279956817627,
      "step": 48433
    },
    {
      "epoch": 4.6189308166503906e-05,
      "step": 48433,
      "training_step_time": 6.886751890182495
    },
    {
      "epoch": 4.619026184082031e-05,
      "model_forward_time": 2.8680331707000732,
      "step": 48434
    },
    {
      "epoch": 4.619026184082031e-05,
      "step": 48434,
      "training_step_time": 9.79056191444397
    },
    {
      "epoch": 4.619121551513672e-05,
      "model_forward_time": 1.8252127170562744,
      "step": 48435
    },
    {
      "epoch": 4.619121551513672e-05,
      "step": 48435,
      "training_step_time": 8.016296148300171
    },
    {
      "epoch": 4.6192169189453125e-05,
      "model_forward_time": 1.7508432865142822,
      "step": 48436
    },
    {
      "epoch": 4.6192169189453125e-05,
      "step": 48436,
      "training_step_time": 7.886305570602417
    },
    {
      "epoch": 4.619312286376953e-05,
      "model_forward_time": 1.7644267082214355,
      "step": 48437
    },
    {
      "epoch": 4.619312286376953e-05,
      "step": 48437,
      "training_step_time": 6.933926820755005
    },
    {
      "epoch": 4.619407653808594e-05,
      "model_forward_time": 1.7510089874267578,
      "step": 48438
    },
    {
      "epoch": 4.619407653808594e-05,
      "step": 48438,
      "training_step_time": 6.821749925613403
    },
    {
      "epoch": 4.6195030212402344e-05,
      "model_forward_time": 1.738027572631836,
      "step": 48439
    },
    {
      "epoch": 4.6195030212402344e-05,
      "step": 48439,
      "training_step_time": 7.002033472061157
    },
    {
      "epoch": 4.619598388671875e-05,
      "grad_norm": 0.09069979935884476,
      "learning_rate": 5.669457562023111e-06,
      "loss": 0.0597,
      "step": 48440
    },
    {
      "epoch": 4.619598388671875e-05,
      "model_forward_time": 1.7557957172393799,
      "step": 48440
    },
    {
      "dynamics_loss_avg": 0.05860929246991873,
      "epoch": 4.619598388671875e-05,
      "step": 48440
    },
    {
      "action_loss_avg": 0.0016780315374489874,
      "epoch": 4.619598388671875e-05,
      "step": 48440
    },
    {
      "epoch": 4.619598388671875e-05,
      "step": 48440,
      "training_step_time": 7.0457048416137695
    },
    {
      "epoch": 4.619693756103516e-05,
      "model_forward_time": 1.7495670318603516,
      "step": 48441
    },
    {
      "epoch": 4.619693756103516e-05,
      "step": 48441,
      "training_step_time": 6.994044303894043
    },
    {
      "epoch": 4.619789123535156e-05,
      "model_forward_time": 1.7677268981933594,
      "step": 48442
    },
    {
      "epoch": 4.619789123535156e-05,
      "step": 48442,
      "training_step_time": 7.005908489227295
    },
    {
      "epoch": 4.619884490966797e-05,
      "model_forward_time": 1.7585313320159912,
      "step": 48443
    },
    {
      "epoch": 4.619884490966797e-05,
      "step": 48443,
      "training_step_time": 9.446833372116089
    },
    {
      "epoch": 4.619979858398438e-05,
      "model_forward_time": 1.8347578048706055,
      "step": 48444
    },
    {
      "epoch": 4.619979858398438e-05,
      "step": 48444,
      "training_step_time": 8.585039138793945
    },
    {
      "epoch": 4.620075225830078e-05,
      "model_forward_time": 1.7589266300201416,
      "step": 48445
    },
    {
      "epoch": 4.620075225830078e-05,
      "step": 48445,
      "training_step_time": 7.949040651321411
    },
    {
      "epoch": 4.6201705932617186e-05,
      "model_forward_time": 1.7608497142791748,
      "step": 48446
    },
    {
      "epoch": 4.6201705932617186e-05,
      "step": 48446,
      "training_step_time": 6.969552040100098
    },
    {
      "epoch": 4.6202659606933596e-05,
      "model_forward_time": 1.7624130249023438,
      "step": 48447
    },
    {
      "epoch": 4.6202659606933596e-05,
      "step": 48447,
      "training_step_time": 6.900068044662476
    },
    {
      "epoch": 4.620361328125e-05,
      "model_forward_time": 1.7592108249664307,
      "step": 48448
    },
    {
      "epoch": 4.620361328125e-05,
      "step": 48448,
      "training_step_time": 6.965300798416138
    },
    {
      "epoch": 4.6204566955566404e-05,
      "model_forward_time": 1.7566494941711426,
      "step": 48449
    },
    {
      "epoch": 4.6204566955566404e-05,
      "step": 48449,
      "training_step_time": 6.836446523666382
    },
    {
      "epoch": 4.6205520629882815e-05,
      "grad_norm": 0.09139019250869751,
      "learning_rate": 5.667818943479699e-06,
      "loss": 0.0554,
      "step": 48450
    },
    {
      "epoch": 4.6205520629882815e-05,
      "model_forward_time": 1.766294002532959,
      "step": 48450
    },
    {
      "dynamics_loss_avg": 0.05077326446771622,
      "epoch": 4.6205520629882815e-05,
      "step": 48450
    },
    {
      "action_loss_avg": 0.001387222169432789,
      "epoch": 4.6205520629882815e-05,
      "step": 48450
    },
    {
      "epoch": 4.6205520629882815e-05,
      "step": 48450,
      "training_step_time": 6.966614007949829
    },
    {
      "epoch": 4.620647430419922e-05,
      "model_forward_time": 1.762446641921997,
      "step": 48451
    },
    {
      "epoch": 4.620647430419922e-05,
      "step": 48451,
      "training_step_time": 6.927780389785767
    },
    {
      "epoch": 4.620742797851562e-05,
      "model_forward_time": 1.7641186714172363,
      "step": 48452
    },
    {
      "epoch": 4.620742797851562e-05,
      "step": 48452,
      "training_step_time": 7.911316633224487
    },
    {
      "epoch": 4.6208381652832034e-05,
      "model_forward_time": 3.0180108547210693,
      "step": 48453
    },
    {
      "epoch": 4.6208381652832034e-05,
      "step": 48453,
      "training_step_time": 11.307898759841919
    },
    {
      "epoch": 4.620933532714844e-05,
      "model_forward_time": 1.761336326599121,
      "step": 48454
    },
    {
      "epoch": 4.620933532714844e-05,
      "step": 48454,
      "training_step_time": 7.870559453964233
    },
    {
      "epoch": 4.621028900146484e-05,
      "model_forward_time": 1.7537474632263184,
      "step": 48455
    },
    {
      "epoch": 4.621028900146484e-05,
      "step": 48455,
      "training_step_time": 7.847793340682983
    },
    {
      "epoch": 4.621124267578125e-05,
      "model_forward_time": 1.7603018283843994,
      "step": 48456
    },
    {
      "epoch": 4.621124267578125e-05,
      "step": 48456,
      "training_step_time": 6.927162408828735
    },
    {
      "epoch": 4.621219635009766e-05,
      "model_forward_time": 1.7700130939483643,
      "step": 48457
    },
    {
      "epoch": 4.621219635009766e-05,
      "step": 48457,
      "training_step_time": 6.927720546722412
    },
    {
      "epoch": 4.621315002441406e-05,
      "model_forward_time": 1.7864253520965576,
      "step": 48458
    },
    {
      "epoch": 4.621315002441406e-05,
      "step": 48458,
      "training_step_time": 6.928327798843384
    },
    {
      "epoch": 4.621410369873047e-05,
      "model_forward_time": 1.7548120021820068,
      "step": 48459
    },
    {
      "epoch": 4.621410369873047e-05,
      "step": 48459,
      "training_step_time": 6.932398796081543
    },
    {
      "epoch": 4.6215057373046876e-05,
      "grad_norm": 0.12591633200645447,
      "learning_rate": 5.666180251904613e-06,
      "loss": 0.0551,
      "step": 48460
    },
    {
      "epoch": 4.6215057373046876e-05,
      "model_forward_time": 1.7600274085998535,
      "step": 48460
    },
    {
      "dynamics_loss_avg": 0.06794752031564713,
      "epoch": 4.6215057373046876e-05,
      "step": 48460
    },
    {
      "action_loss_avg": 0.001592081575654447,
      "epoch": 4.6215057373046876e-05,
      "step": 48460
    },
    {
      "epoch": 4.6215057373046876e-05,
      "step": 48460,
      "training_step_time": 6.963308334350586
    },
    {
      "epoch": 4.621601104736328e-05,
      "model_forward_time": 1.7603809833526611,
      "step": 48461
    },
    {
      "epoch": 4.621601104736328e-05,
      "step": 48461,
      "training_step_time": 7.850787401199341
    },
    {
      "epoch": 4.621696472167969e-05,
      "model_forward_time": 3.0481743812561035,
      "step": 48462
    },
    {
      "epoch": 4.621696472167969e-05,
      "step": 48462,
      "training_step_time": 11.764207601547241
    },
    {
      "epoch": 4.6217918395996095e-05,
      "model_forward_time": 1.7622430324554443,
      "step": 48463
    },
    {
      "epoch": 4.6217918395996095e-05,
      "step": 48463,
      "training_step_time": 7.935177326202393
    },
    {
      "epoch": 4.62188720703125e-05,
      "model_forward_time": 1.7496623992919922,
      "step": 48464
    },
    {
      "epoch": 4.62188720703125e-05,
      "step": 48464,
      "training_step_time": 7.851738214492798
    },
    {
      "epoch": 4.621982574462891e-05,
      "model_forward_time": 1.7601993083953857,
      "step": 48465
    },
    {
      "epoch": 4.621982574462891e-05,
      "step": 48465,
      "training_step_time": 6.883723497390747
    },
    {
      "epoch": 4.6220779418945314e-05,
      "model_forward_time": 1.4126429557800293,
      "step": 48466
    },
    {
      "epoch": 4.6220779418945314e-05,
      "step": 48466,
      "training_step_time": 6.843236446380615
    },
    {
      "epoch": 4.622173309326172e-05,
      "model_forward_time": 1.7633330821990967,
      "step": 48467
    },
    {
      "epoch": 4.622173309326172e-05,
      "step": 48467,
      "training_step_time": 6.864579200744629
    },
    {
      "epoch": 4.622268676757813e-05,
      "model_forward_time": 1.7577238082885742,
      "step": 48468
    },
    {
      "epoch": 4.622268676757813e-05,
      "step": 48468,
      "training_step_time": 6.88686203956604
    },
    {
      "epoch": 4.622364044189453e-05,
      "model_forward_time": 1.7611517906188965,
      "step": 48469
    },
    {
      "epoch": 4.622364044189453e-05,
      "step": 48469,
      "training_step_time": 6.902937650680542
    },
    {
      "epoch": 4.6224594116210936e-05,
      "grad_norm": 0.11639879643917084,
      "learning_rate": 5.664541487477056e-06,
      "loss": 0.0604,
      "step": 48470
    },
    {
      "epoch": 4.6224594116210936e-05,
      "model_forward_time": 1.7420551776885986,
      "step": 48470
    },
    {
      "dynamics_loss_avg": 0.05690645687282085,
      "epoch": 4.6224594116210936e-05,
      "step": 48470
    },
    {
      "action_loss_avg": 0.002682298835134134,
      "epoch": 4.6224594116210936e-05,
      "step": 48470
    },
    {
      "epoch": 4.6224594116210936e-05,
      "step": 48470,
      "training_step_time": 7.769803762435913
    },
    {
      "epoch": 4.622554779052734e-05,
      "model_forward_time": 2.3376259803771973,
      "step": 48471
    },
    {
      "epoch": 4.622554779052734e-05,
      "step": 48471,
      "training_step_time": 9.896457195281982
    },
    {
      "epoch": 4.622650146484375e-05,
      "model_forward_time": 1.6923401355743408,
      "step": 48472
    },
    {
      "epoch": 4.622650146484375e-05,
      "step": 48472,
      "training_step_time": 7.89258337020874
    },
    {
      "epoch": 4.6227455139160155e-05,
      "model_forward_time": 1.7141103744506836,
      "step": 48473
    },
    {
      "epoch": 4.6227455139160155e-05,
      "step": 48473,
      "training_step_time": 7.898438453674316
    },
    {
      "epoch": 4.622840881347656e-05,
      "model_forward_time": 1.7113254070281982,
      "step": 48474
    },
    {
      "epoch": 4.622840881347656e-05,
      "step": 48474,
      "training_step_time": 7.266441583633423
    },
    {
      "epoch": 4.622936248779297e-05,
      "model_forward_time": 1.7150187492370605,
      "step": 48475
    },
    {
      "epoch": 4.622936248779297e-05,
      "step": 48475,
      "training_step_time": 6.881046772003174
    },
    {
      "epoch": 4.6230316162109374e-05,
      "model_forward_time": 1.7366795539855957,
      "step": 48476
    },
    {
      "epoch": 4.6230316162109374e-05,
      "step": 48476,
      "training_step_time": 6.932739496231079
    },
    {
      "epoch": 4.623126983642578e-05,
      "model_forward_time": 1.7220568656921387,
      "step": 48477
    },
    {
      "epoch": 4.623126983642578e-05,
      "step": 48477,
      "training_step_time": 6.797363996505737
    },
    {
      "epoch": 4.623222351074219e-05,
      "model_forward_time": 1.7112348079681396,
      "step": 48478
    },
    {
      "epoch": 4.623222351074219e-05,
      "step": 48478,
      "training_step_time": 6.8310325145721436
    },
    {
      "epoch": 4.623317718505859e-05,
      "model_forward_time": 1.7189064025878906,
      "step": 48479
    },
    {
      "epoch": 4.623317718505859e-05,
      "step": 48479,
      "training_step_time": 7.813064813613892
    },
    {
      "epoch": 4.6234130859375e-05,
      "grad_norm": 0.12359632551670074,
      "learning_rate": 5.66290265037624e-06,
      "loss": 0.0541,
      "step": 48480
    },
    {
      "epoch": 4.6234130859375e-05,
      "model_forward_time": 2.280160665512085,
      "step": 48480
    },
    {
      "dynamics_loss_avg": 0.041676444560289384,
      "epoch": 4.6234130859375e-05,
      "step": 48480
    },
    {
      "action_loss_avg": 0.001806460472289473,
      "epoch": 4.6234130859375e-05,
      "step": 48480
    },
    {
      "epoch": 4.6234130859375e-05,
      "step": 48480,
      "training_step_time": 8.235023975372314
    },
    {
      "epoch": 4.623508453369141e-05,
      "model_forward_time": 1.7228002548217773,
      "step": 48481
    },
    {
      "epoch": 4.623508453369141e-05,
      "step": 48481,
      "training_step_time": 9.766992092132568
    },
    {
      "epoch": 4.623603820800781e-05,
      "model_forward_time": 1.724717140197754,
      "step": 48482
    },
    {
      "epoch": 4.623603820800781e-05,
      "step": 48482,
      "training_step_time": 7.837768316268921
    },
    {
      "epoch": 4.6236991882324216e-05,
      "model_forward_time": 1.722808599472046,
      "step": 48483
    },
    {
      "epoch": 4.6236991882324216e-05,
      "step": 48483,
      "training_step_time": 7.5250842571258545
    },
    {
      "epoch": 4.623794555664063e-05,
      "model_forward_time": 1.7110114097595215,
      "step": 48484
    },
    {
      "epoch": 4.623794555664063e-05,
      "step": 48484,
      "training_step_time": 6.890019178390503
    },
    {
      "epoch": 4.623889923095703e-05,
      "model_forward_time": 1.7125523090362549,
      "step": 48485
    },
    {
      "epoch": 4.623889923095703e-05,
      "step": 48485,
      "training_step_time": 6.908367872238159
    },
    {
      "epoch": 4.6239852905273435e-05,
      "model_forward_time": 1.7066473960876465,
      "step": 48486
    },
    {
      "epoch": 4.6239852905273435e-05,
      "step": 48486,
      "training_step_time": 6.89568567276001
    },
    {
      "epoch": 4.6240806579589846e-05,
      "model_forward_time": 1.7219808101654053,
      "step": 48487
    },
    {
      "epoch": 4.6240806579589846e-05,
      "step": 48487,
      "training_step_time": 6.945560455322266
    },
    {
      "epoch": 4.624176025390625e-05,
      "model_forward_time": 1.7238671779632568,
      "step": 48488
    },
    {
      "epoch": 4.624176025390625e-05,
      "step": 48488,
      "training_step_time": 7.73747181892395
    },
    {
      "epoch": 4.6242713928222654e-05,
      "model_forward_time": 1.750967025756836,
      "step": 48489
    },
    {
      "epoch": 4.6242713928222654e-05,
      "step": 48489,
      "training_step_time": 10.319077253341675
    },
    {
      "epoch": 4.6243667602539064e-05,
      "grad_norm": 0.11927983164787292,
      "learning_rate": 5.661263740781387e-06,
      "loss": 0.0576,
      "step": 48490
    },
    {
      "epoch": 4.6243667602539064e-05,
      "model_forward_time": 1.7143890857696533,
      "step": 48490
    },
    {
      "dynamics_loss_avg": 0.06889649890363217,
      "epoch": 4.6243667602539064e-05,
      "step": 48490
    },
    {
      "action_loss_avg": 0.0028658777591772377,
      "epoch": 4.6243667602539064e-05,
      "step": 48490
    },
    {
      "epoch": 4.6243667602539064e-05,
      "step": 48490,
      "training_step_time": 10.383469343185425
    },
    {
      "epoch": 4.624462127685547e-05,
      "model_forward_time": 1.7166979312896729,
      "step": 48491
    },
    {
      "epoch": 4.624462127685547e-05,
      "step": 48491,
      "training_step_time": 7.741073846817017
    },
    {
      "epoch": 4.624557495117187e-05,
      "model_forward_time": 1.7032761573791504,
      "step": 48492
    },
    {
      "epoch": 4.624557495117187e-05,
      "step": 48492,
      "training_step_time": 7.596165418624878
    },
    {
      "epoch": 4.624652862548828e-05,
      "model_forward_time": 1.705204963684082,
      "step": 48493
    },
    {
      "epoch": 4.624652862548828e-05,
      "step": 48493,
      "training_step_time": 7.137955904006958
    },
    {
      "epoch": 4.624748229980469e-05,
      "model_forward_time": 1.70951247215271,
      "step": 48494
    },
    {
      "epoch": 4.624748229980469e-05,
      "step": 48494,
      "training_step_time": 6.756670236587524
    },
    {
      "epoch": 4.624843597412109e-05,
      "model_forward_time": 1.708198070526123,
      "step": 48495
    },
    {
      "epoch": 4.624843597412109e-05,
      "step": 48495,
      "training_step_time": 6.834795713424683
    },
    {
      "epoch": 4.62493896484375e-05,
      "model_forward_time": 1.707308053970337,
      "step": 48496
    },
    {
      "epoch": 4.62493896484375e-05,
      "step": 48496,
      "training_step_time": 6.80420708656311
    },
    {
      "epoch": 4.6250343322753906e-05,
      "model_forward_time": 1.7082724571228027,
      "step": 48497
    },
    {
      "epoch": 4.6250343322753906e-05,
      "step": 48497,
      "training_step_time": 7.796266555786133
    },
    {
      "epoch": 4.625129699707031e-05,
      "model_forward_time": 1.7132549285888672,
      "step": 48498
    },
    {
      "epoch": 4.625129699707031e-05,
      "step": 48498,
      "training_step_time": 7.81949257850647
    },
    {
      "epoch": 4.625225067138672e-05,
      "model_forward_time": 2.2890422344207764,
      "step": 48499
    },
    {
      "epoch": 4.625225067138672e-05,
      "step": 48499,
      "training_step_time": 8.602668762207031
    },
    {
      "epoch": 4.6253204345703125e-05,
      "grad_norm": 0.1069609746336937,
      "learning_rate": 5.659624758871726e-06,
      "loss": 0.0568,
      "step": 48500
    },
    {
      "epoch": 4.6253204345703125e-05,
      "model_forward_time": 1.7115774154663086,
      "step": 48500
    },
    {
      "dynamics_loss_avg": 0.044402312859892844,
      "epoch": 4.6253204345703125e-05,
      "step": 48500
    },
    {
      "action_loss_avg": 0.0009555409720633179,
      "epoch": 4.6253204345703125e-05,
      "step": 48500
    },
    {
      "epoch": 4.6253204345703125e-05,
      "step": 48500,
      "training_step_time": 7.822321653366089
    },
    {
      "epoch": 4.625415802001953e-05,
      "model_forward_time": 1.7043259143829346,
      "step": 48501
    },
    {
      "epoch": 4.625415802001953e-05,
      "step": 48501,
      "training_step_time": 7.804581642150879
    },
    {
      "epoch": 4.625511169433594e-05,
      "model_forward_time": 1.3363029956817627,
      "step": 48502
    },
    {
      "epoch": 4.625511169433594e-05,
      "step": 48502,
      "training_step_time": 7.292997121810913
    },
    {
      "epoch": 4.6256065368652344e-05,
      "model_forward_time": 1.7110862731933594,
      "step": 48503
    },
    {
      "epoch": 4.6256065368652344e-05,
      "step": 48503,
      "training_step_time": 6.768457651138306
    },
    {
      "epoch": 4.625701904296875e-05,
      "model_forward_time": 1.711130142211914,
      "step": 48504
    },
    {
      "epoch": 4.625701904296875e-05,
      "step": 48504,
      "training_step_time": 6.923521280288696
    },
    {
      "epoch": 4.625797271728516e-05,
      "model_forward_time": 1.7024877071380615,
      "step": 48505
    },
    {
      "epoch": 4.625797271728516e-05,
      "step": 48505,
      "training_step_time": 6.9056031703948975
    },
    {
      "epoch": 4.625892639160156e-05,
      "model_forward_time": 1.721222162246704,
      "step": 48506
    },
    {
      "epoch": 4.625892639160156e-05,
      "step": 48506,
      "training_step_time": 7.718399524688721
    },
    {
      "epoch": 4.625988006591797e-05,
      "model_forward_time": 1.7150943279266357,
      "step": 48507
    },
    {
      "epoch": 4.625988006591797e-05,
      "step": 48507,
      "training_step_time": 7.595154762268066
    },
    {
      "epoch": 4.626083374023438e-05,
      "model_forward_time": 2.272298574447632,
      "step": 48508
    },
    {
      "epoch": 4.626083374023438e-05,
      "step": 48508,
      "training_step_time": 8.786100387573242
    },
    {
      "epoch": 4.626178741455078e-05,
      "model_forward_time": 1.750556468963623,
      "step": 48509
    },
    {
      "epoch": 4.626178741455078e-05,
      "step": 48509,
      "training_step_time": 8.915130376815796
    },
    {
      "epoch": 4.6262741088867186e-05,
      "grad_norm": 0.10270195454359055,
      "learning_rate": 5.657985704826493e-06,
      "loss": 0.052,
      "step": 48510
    },
    {
      "epoch": 4.6262741088867186e-05,
      "model_forward_time": 1.7138569355010986,
      "step": 48510
    },
    {
      "dynamics_loss_avg": 0.045313626434653996,
      "epoch": 4.6262741088867186e-05,
      "step": 48510
    },
    {
      "action_loss_avg": 0.0010950387280900032,
      "epoch": 4.6262741088867186e-05,
      "step": 48510
    },
    {
      "epoch": 4.6262741088867186e-05,
      "step": 48510,
      "training_step_time": 7.790996313095093
    },
    {
      "epoch": 4.6263694763183596e-05,
      "model_forward_time": 1.712522268295288,
      "step": 48511
    },
    {
      "epoch": 4.6263694763183596e-05,
      "step": 48511,
      "training_step_time": 7.243931293487549
    },
    {
      "epoch": 4.62646484375e-05,
      "model_forward_time": 1.7298879623413086,
      "step": 48512
    },
    {
      "epoch": 4.62646484375e-05,
      "step": 48512,
      "training_step_time": 6.863339424133301
    },
    {
      "epoch": 4.6265602111816404e-05,
      "model_forward_time": 1.7191517353057861,
      "step": 48513
    },
    {
      "epoch": 4.6265602111816404e-05,
      "step": 48513,
      "training_step_time": 6.834054470062256
    },
    {
      "epoch": 4.6266555786132815e-05,
      "model_forward_time": 1.7182402610778809,
      "step": 48514
    },
    {
      "epoch": 4.6266555786132815e-05,
      "step": 48514,
      "training_step_time": 6.869122743606567
    },
    {
      "epoch": 4.626750946044922e-05,
      "model_forward_time": 1.7160007953643799,
      "step": 48515
    },
    {
      "epoch": 4.626750946044922e-05,
      "step": 48515,
      "training_step_time": 7.760383367538452
    },
    {
      "epoch": 4.626846313476562e-05,
      "model_forward_time": 1.7283997535705566,
      "step": 48516
    },
    {
      "epoch": 4.626846313476562e-05,
      "step": 48516,
      "training_step_time": 8.133069038391113
    },
    {
      "epoch": 4.6269416809082034e-05,
      "model_forward_time": 2.308732509613037,
      "step": 48517
    },
    {
      "epoch": 4.6269416809082034e-05,
      "step": 48517,
      "training_step_time": 8.778878927230835
    },
    {
      "epoch": 4.627037048339844e-05,
      "model_forward_time": 1.7672266960144043,
      "step": 48518
    },
    {
      "epoch": 4.627037048339844e-05,
      "step": 48518,
      "training_step_time": 9.59842824935913
    },
    {
      "epoch": 4.627132415771484e-05,
      "model_forward_time": 1.7463953495025635,
      "step": 48519
    },
    {
      "epoch": 4.627132415771484e-05,
      "step": 48519,
      "training_step_time": 7.725741863250732
    },
    {
      "epoch": 4.627227783203125e-05,
      "grad_norm": 0.08383598923683167,
      "learning_rate": 5.656346578824932e-06,
      "loss": 0.059,
      "step": 48520
    },
    {
      "epoch": 4.627227783203125e-05,
      "model_forward_time": 1.7469124794006348,
      "step": 48520
    },
    {
      "dynamics_loss_avg": 0.06415499141439795,
      "epoch": 4.627227783203125e-05,
      "step": 48520
    },
    {
      "action_loss_avg": 0.002405159006593749,
      "epoch": 4.627227783203125e-05,
      "step": 48520
    },
    {
      "epoch": 4.627227783203125e-05,
      "step": 48520,
      "training_step_time": 7.729159355163574
    },
    {
      "epoch": 4.627323150634766e-05,
      "model_forward_time": 1.7670793533325195,
      "step": 48521
    },
    {
      "epoch": 4.627323150634766e-05,
      "step": 48521,
      "training_step_time": 6.843794345855713
    },
    {
      "epoch": 4.627418518066406e-05,
      "model_forward_time": 1.7386887073516846,
      "step": 48522
    },
    {
      "epoch": 4.627418518066406e-05,
      "step": 48522,
      "training_step_time": 6.90085768699646
    },
    {
      "epoch": 4.627513885498047e-05,
      "model_forward_time": 1.7467079162597656,
      "step": 48523
    },
    {
      "epoch": 4.627513885498047e-05,
      "step": 48523,
      "training_step_time": 6.874097108840942
    },
    {
      "epoch": 4.6276092529296876e-05,
      "model_forward_time": 1.746483564376831,
      "step": 48524
    },
    {
      "epoch": 4.6276092529296876e-05,
      "step": 48524,
      "training_step_time": 7.757798433303833
    },
    {
      "epoch": 4.627704620361328e-05,
      "model_forward_time": 1.7469902038574219,
      "step": 48525
    },
    {
      "epoch": 4.627704620361328e-05,
      "step": 48525,
      "training_step_time": 8.255590438842773
    },
    {
      "epoch": 4.627799987792969e-05,
      "model_forward_time": 2.938352346420288,
      "step": 48526
    },
    {
      "epoch": 4.627799987792969e-05,
      "step": 48526,
      "training_step_time": 9.238012790679932
    },
    {
      "epoch": 4.6278953552246095e-05,
      "model_forward_time": 1.741393804550171,
      "step": 48527
    },
    {
      "epoch": 4.6278953552246095e-05,
      "step": 48527,
      "training_step_time": 7.783854007720947
    },
    {
      "epoch": 4.62799072265625e-05,
      "model_forward_time": 1.7571921348571777,
      "step": 48528
    },
    {
      "epoch": 4.62799072265625e-05,
      "step": 48528,
      "training_step_time": 7.66209077835083
    },
    {
      "epoch": 4.628086090087891e-05,
      "model_forward_time": 1.7523553371429443,
      "step": 48529
    },
    {
      "epoch": 4.628086090087891e-05,
      "step": 48529,
      "training_step_time": 7.593601226806641
    },
    {
      "epoch": 4.6281814575195314e-05,
      "grad_norm": 0.09149639308452606,
      "learning_rate": 5.654707381046296e-06,
      "loss": 0.0578,
      "step": 48530
    },
    {
      "epoch": 4.6281814575195314e-05,
      "model_forward_time": 1.7416880130767822,
      "step": 48530
    },
    {
      "dynamics_loss_avg": 0.04928045812994242,
      "epoch": 4.6281814575195314e-05,
      "step": 48530
    },
    {
      "action_loss_avg": 0.0016117835621116683,
      "epoch": 4.6281814575195314e-05,
      "step": 48530
    },
    {
      "epoch": 4.6281814575195314e-05,
      "step": 48530,
      "training_step_time": 7.18310546875
    },
    {
      "epoch": 4.628276824951172e-05,
      "model_forward_time": 1.7525732517242432,
      "step": 48531
    },
    {
      "epoch": 4.628276824951172e-05,
      "step": 48531,
      "training_step_time": 6.804465055465698
    },
    {
      "epoch": 4.628372192382812e-05,
      "model_forward_time": 1.7618906497955322,
      "step": 48532
    },
    {
      "epoch": 4.628372192382812e-05,
      "step": 48532,
      "training_step_time": 6.761251926422119
    },
    {
      "epoch": 4.628467559814453e-05,
      "model_forward_time": 1.7568342685699463,
      "step": 48533
    },
    {
      "epoch": 4.628467559814453e-05,
      "step": 48533,
      "training_step_time": 7.687392234802246
    },
    {
      "epoch": 4.6285629272460936e-05,
      "model_forward_time": 1.7412948608398438,
      "step": 48534
    },
    {
      "epoch": 4.6285629272460936e-05,
      "step": 48534,
      "training_step_time": 7.568468332290649
    },
    {
      "epoch": 4.628658294677734e-05,
      "model_forward_time": 2.8864212036132812,
      "step": 48535
    },
    {
      "epoch": 4.628658294677734e-05,
      "step": 48535,
      "training_step_time": 9.89026141166687
    },
    {
      "epoch": 4.628753662109375e-05,
      "model_forward_time": 1.7498514652252197,
      "step": 48536
    },
    {
      "epoch": 4.628753662109375e-05,
      "step": 48536,
      "training_step_time": 9.570478677749634
    },
    {
      "epoch": 4.6288490295410155e-05,
      "model_forward_time": 1.7481119632720947,
      "step": 48537
    },
    {
      "epoch": 4.6288490295410155e-05,
      "step": 48537,
      "training_step_time": 8.299459457397461
    },
    {
      "epoch": 4.628944396972656e-05,
      "model_forward_time": 1.7422657012939453,
      "step": 48538
    },
    {
      "epoch": 4.628944396972656e-05,
      "step": 48538,
      "training_step_time": 7.636312007904053
    },
    {
      "epoch": 4.629039764404297e-05,
      "model_forward_time": 1.748703956604004,
      "step": 48539
    },
    {
      "epoch": 4.629039764404297e-05,
      "step": 48539,
      "training_step_time": 7.217848300933838
    },
    {
      "epoch": 4.6291351318359374e-05,
      "grad_norm": 0.0959700495004654,
      "learning_rate": 5.653068111669846e-06,
      "loss": 0.0572,
      "step": 48540
    },
    {
      "epoch": 4.6291351318359374e-05,
      "model_forward_time": 1.7463316917419434,
      "step": 48540
    },
    {
      "dynamics_loss_avg": 0.057128687389194965,
      "epoch": 4.6291351318359374e-05,
      "step": 48540
    },
    {
      "action_loss_avg": 0.0013006372493691743,
      "epoch": 4.6291351318359374e-05,
      "step": 48540
    },
    {
      "epoch": 4.6291351318359374e-05,
      "step": 48540,
      "training_step_time": 6.797321796417236
    },
    {
      "epoch": 4.629230499267578e-05,
      "model_forward_time": 1.754819631576538,
      "step": 48541
    },
    {
      "epoch": 4.629230499267578e-05,
      "step": 48541,
      "training_step_time": 6.8635241985321045
    },
    {
      "epoch": 4.629325866699219e-05,
      "model_forward_time": 1.7346572875976562,
      "step": 48542
    },
    {
      "epoch": 4.629325866699219e-05,
      "step": 48542,
      "training_step_time": 7.59600305557251
    },
    {
      "epoch": 4.629421234130859e-05,
      "model_forward_time": 1.7546775341033936,
      "step": 48543
    },
    {
      "epoch": 4.629421234130859e-05,
      "step": 48543,
      "training_step_time": 7.610388278961182
    },
    {
      "epoch": 4.6295166015625e-05,
      "model_forward_time": 2.833479166030884,
      "step": 48544
    },
    {
      "epoch": 4.6295166015625e-05,
      "step": 48544,
      "training_step_time": 10.504141330718994
    },
    {
      "epoch": 4.629611968994141e-05,
      "model_forward_time": 1.866403341293335,
      "step": 48545
    },
    {
      "epoch": 4.629611968994141e-05,
      "step": 48545,
      "training_step_time": 8.980741739273071
    },
    {
      "epoch": 4.629707336425781e-05,
      "model_forward_time": 1.7485008239746094,
      "step": 48546
    },
    {
      "epoch": 4.629707336425781e-05,
      "step": 48546,
      "training_step_time": 7.675306797027588
    },
    {
      "epoch": 4.6298027038574216e-05,
      "model_forward_time": 1.7578003406524658,
      "step": 48547
    },
    {
      "epoch": 4.6298027038574216e-05,
      "step": 48547,
      "training_step_time": 7.599489212036133
    },
    {
      "epoch": 4.629898071289063e-05,
      "model_forward_time": 1.7403240203857422,
      "step": 48548
    },
    {
      "epoch": 4.629898071289063e-05,
      "step": 48548,
      "training_step_time": 7.049185037612915
    },
    {
      "epoch": 4.629993438720703e-05,
      "model_forward_time": 1.7542157173156738,
      "step": 48549
    },
    {
      "epoch": 4.629993438720703e-05,
      "step": 48549,
      "training_step_time": 6.762070894241333
    },
    {
      "epoch": 4.6300888061523435e-05,
      "grad_norm": 0.08527680486440659,
      "learning_rate": 5.6514287708748484e-06,
      "loss": 0.0574,
      "step": 48550
    },
    {
      "epoch": 4.6300888061523435e-05,
      "model_forward_time": 1.7610435485839844,
      "step": 48550
    },
    {
      "dynamics_loss_avg": 0.05949665307998657,
      "epoch": 4.6300888061523435e-05,
      "step": 48550
    },
    {
      "action_loss_avg": 0.001498167566023767,
      "epoch": 4.6300888061523435e-05,
      "step": 48550
    },
    {
      "epoch": 4.6300888061523435e-05,
      "step": 48550,
      "training_step_time": 6.734862565994263
    },
    {
      "epoch": 4.6301841735839846e-05,
      "model_forward_time": 1.751779556274414,
      "step": 48551
    },
    {
      "epoch": 4.6301841735839846e-05,
      "step": 48551,
      "training_step_time": 7.438276052474976
    },
    {
      "epoch": 4.630279541015625e-05,
      "model_forward_time": 1.7636594772338867,
      "step": 48552
    },
    {
      "epoch": 4.630279541015625e-05,
      "step": 48552,
      "training_step_time": 7.434150457382202
    },
    {
      "epoch": 4.6303749084472654e-05,
      "model_forward_time": 1.7588481903076172,
      "step": 48553
    },
    {
      "epoch": 4.6303749084472654e-05,
      "step": 48553,
      "training_step_time": 7.674589395523071
    },
    {
      "epoch": 4.6304702758789064e-05,
      "model_forward_time": 2.9043967723846436,
      "step": 48554
    },
    {
      "epoch": 4.6304702758789064e-05,
      "step": 48554,
      "training_step_time": 9.684406518936157
    },
    {
      "epoch": 4.630565643310547e-05,
      "model_forward_time": 1.748417615890503,
      "step": 48555
    },
    {
      "epoch": 4.630565643310547e-05,
      "step": 48555,
      "training_step_time": 8.400151252746582
    },
    {
      "epoch": 4.630661010742187e-05,
      "model_forward_time": 1.7429521083831787,
      "step": 48556
    },
    {
      "epoch": 4.630661010742187e-05,
      "step": 48556,
      "training_step_time": 7.647760629653931
    },
    {
      "epoch": 4.630756378173828e-05,
      "model_forward_time": 1.7559287548065186,
      "step": 48557
    },
    {
      "epoch": 4.630756378173828e-05,
      "step": 48557,
      "training_step_time": 7.073037147521973
    },
    {
      "epoch": 4.630851745605469e-05,
      "model_forward_time": 1.7473676204681396,
      "step": 48558
    },
    {
      "epoch": 4.630851745605469e-05,
      "step": 48558,
      "training_step_time": 7.154256820678711
    },
    {
      "epoch": 4.630947113037109e-05,
      "model_forward_time": 1.742340087890625,
      "step": 48559
    },
    {
      "epoch": 4.630947113037109e-05,
      "step": 48559,
      "training_step_time": 6.674915075302124
    },
    {
      "epoch": 4.63104248046875e-05,
      "grad_norm": 0.13444550335407257,
      "learning_rate": 5.6497893588405805e-06,
      "loss": 0.0561,
      "step": 48560
    },
    {
      "epoch": 4.63104248046875e-05,
      "model_forward_time": 1.7361788749694824,
      "step": 48560
    },
    {
      "dynamics_loss_avg": 0.07203175704926253,
      "epoch": 4.63104248046875e-05,
      "step": 48560
    },
    {
      "action_loss_avg": 0.0019294458092190326,
      "epoch": 4.63104248046875e-05,
      "step": 48560
    },
    {
      "epoch": 4.63104248046875e-05,
      "step": 48560,
      "training_step_time": 7.617164134979248
    },
    {
      "epoch": 4.6311378479003906e-05,
      "model_forward_time": 1.7642502784729004,
      "step": 48561
    },
    {
      "epoch": 4.6311378479003906e-05,
      "step": 48561,
      "training_step_time": 7.5627007484436035
    },
    {
      "epoch": 4.631233215332031e-05,
      "model_forward_time": 1.7576181888580322,
      "step": 48562
    },
    {
      "epoch": 4.631233215332031e-05,
      "step": 48562,
      "training_step_time": 8.367507219314575
    },
    {
      "epoch": 4.631328582763672e-05,
      "model_forward_time": 2.887727975845337,
      "step": 48563
    },
    {
      "epoch": 4.631328582763672e-05,
      "step": 48563,
      "training_step_time": 8.656951427459717
    },
    {
      "epoch": 4.6314239501953125e-05,
      "model_forward_time": 1.7528259754180908,
      "step": 48564
    },
    {
      "epoch": 4.6314239501953125e-05,
      "step": 48564,
      "training_step_time": 8.211235761642456
    },
    {
      "epoch": 4.631519317626953e-05,
      "model_forward_time": 1.7429957389831543,
      "step": 48565
    },
    {
      "epoch": 4.631519317626953e-05,
      "step": 48565,
      "training_step_time": 7.693017244338989
    },
    {
      "epoch": 4.631614685058594e-05,
      "model_forward_time": 1.7618391513824463,
      "step": 48566
    },
    {
      "epoch": 4.631614685058594e-05,
      "step": 48566,
      "training_step_time": 7.07707667350769
    },
    {
      "epoch": 4.6317100524902344e-05,
      "model_forward_time": 1.7465012073516846,
      "step": 48567
    },
    {
      "epoch": 4.6317100524902344e-05,
      "step": 48567,
      "training_step_time": 7.14658260345459
    },
    {
      "epoch": 4.631805419921875e-05,
      "model_forward_time": 1.7472271919250488,
      "step": 48568
    },
    {
      "epoch": 4.631805419921875e-05,
      "step": 48568,
      "training_step_time": 6.750891447067261
    },
    {
      "epoch": 4.631900787353516e-05,
      "model_forward_time": 1.382817268371582,
      "step": 48569
    },
    {
      "epoch": 4.631900787353516e-05,
      "step": 48569,
      "training_step_time": 7.674337387084961
    },
    {
      "epoch": 4.631996154785156e-05,
      "grad_norm": 0.0978577733039856,
      "learning_rate": 5.648149875746325e-06,
      "loss": 0.0562,
      "step": 48570
    },
    {
      "epoch": 4.631996154785156e-05,
      "model_forward_time": 1.755922794342041,
      "step": 48570
    },
    {
      "dynamics_loss_avg": 0.07309906296432019,
      "epoch": 4.631996154785156e-05,
      "step": 48570
    },
    {
      "action_loss_avg": 0.0026236527482979,
      "epoch": 4.631996154785156e-05,
      "step": 48570
    },
    {
      "epoch": 4.631996154785156e-05,
      "step": 48570,
      "training_step_time": 7.660193681716919
    },
    {
      "epoch": 4.632091522216797e-05,
      "model_forward_time": 1.7493102550506592,
      "step": 48571
    },
    {
      "epoch": 4.632091522216797e-05,
      "step": 48571,
      "training_step_time": 8.42781686782837
    },
    {
      "epoch": 4.632186889648438e-05,
      "model_forward_time": 2.892526388168335,
      "step": 48572
    },
    {
      "epoch": 4.632186889648438e-05,
      "step": 48572,
      "training_step_time": 8.891227006912231
    },
    {
      "epoch": 4.632282257080078e-05,
      "model_forward_time": 1.7666826248168945,
      "step": 48573
    },
    {
      "epoch": 4.632282257080078e-05,
      "step": 48573,
      "training_step_time": 7.869720697402954
    },
    {
      "epoch": 4.6323776245117186e-05,
      "model_forward_time": 1.7339239120483398,
      "step": 48574
    },
    {
      "epoch": 4.6323776245117186e-05,
      "step": 48574,
      "training_step_time": 8.660696268081665
    },
    {
      "epoch": 4.6324729919433596e-05,
      "model_forward_time": 1.7458882331848145,
      "step": 48575
    },
    {
      "epoch": 4.6324729919433596e-05,
      "step": 48575,
      "training_step_time": 7.7003700733184814
    },
    {
      "epoch": 4.632568359375e-05,
      "model_forward_time": 1.743906021118164,
      "step": 48576
    },
    {
      "epoch": 4.632568359375e-05,
      "step": 48576,
      "training_step_time": 7.193551301956177
    },
    {
      "epoch": 4.6326637268066405e-05,
      "model_forward_time": 1.7497706413269043,
      "step": 48577
    },
    {
      "epoch": 4.6326637268066405e-05,
      "step": 48577,
      "training_step_time": 7.319605588912964
    },
    {
      "epoch": 4.6327590942382815e-05,
      "model_forward_time": 1.7495360374450684,
      "step": 48578
    },
    {
      "epoch": 4.6327590942382815e-05,
      "step": 48578,
      "training_step_time": 7.679888486862183
    },
    {
      "epoch": 4.632854461669922e-05,
      "model_forward_time": 1.7412092685699463,
      "step": 48579
    },
    {
      "epoch": 4.632854461669922e-05,
      "step": 48579,
      "training_step_time": 7.604731559753418
    },
    {
      "epoch": 4.632949829101562e-05,
      "grad_norm": 0.09635145962238312,
      "learning_rate": 5.646510321771373e-06,
      "loss": 0.055,
      "step": 48580
    },
    {
      "epoch": 4.632949829101562e-05,
      "model_forward_time": 1.7396647930145264,
      "step": 48580
    },
    {
      "dynamics_loss_avg": 0.03943390361964703,
      "epoch": 4.632949829101562e-05,
      "step": 48580
    },
    {
      "action_loss_avg": 0.0018732546363025905,
      "epoch": 4.632949829101562e-05,
      "step": 48580
    },
    {
      "epoch": 4.632949829101562e-05,
      "step": 48580,
      "training_step_time": 7.67588472366333
    },
    {
      "epoch": 4.6330451965332034e-05,
      "model_forward_time": 2.8719472885131836,
      "step": 48581
    },
    {
      "epoch": 4.6330451965332034e-05,
      "step": 48581,
      "training_step_time": 10.272939682006836
    },
    {
      "epoch": 4.633140563964844e-05,
      "model_forward_time": 1.7357664108276367,
      "step": 48582
    },
    {
      "epoch": 4.633140563964844e-05,
      "step": 48582,
      "training_step_time": 8.365488767623901
    },
    {
      "epoch": 4.633235931396484e-05,
      "model_forward_time": 1.7417492866516113,
      "step": 48583
    },
    {
      "epoch": 4.633235931396484e-05,
      "step": 48583,
      "training_step_time": 7.843283653259277
    },
    {
      "epoch": 4.633331298828125e-05,
      "model_forward_time": 1.7267482280731201,
      "step": 48584
    },
    {
      "epoch": 4.633331298828125e-05,
      "step": 48584,
      "training_step_time": 7.486395359039307
    },
    {
      "epoch": 4.633426666259766e-05,
      "model_forward_time": 1.7358880043029785,
      "step": 48585
    },
    {
      "epoch": 4.633426666259766e-05,
      "step": 48585,
      "training_step_time": 7.206760406494141
    },
    {
      "epoch": 4.633522033691406e-05,
      "model_forward_time": 1.7513208389282227,
      "step": 48586
    },
    {
      "epoch": 4.633522033691406e-05,
      "step": 48586,
      "training_step_time": 6.898322105407715
    },
    {
      "epoch": 4.633617401123047e-05,
      "model_forward_time": 1.7267682552337646,
      "step": 48587
    },
    {
      "epoch": 4.633617401123047e-05,
      "step": 48587,
      "training_step_time": 7.675227403640747
    },
    {
      "epoch": 4.6337127685546876e-05,
      "model_forward_time": 1.7411930561065674,
      "step": 48588
    },
    {
      "epoch": 4.6337127685546876e-05,
      "step": 48588,
      "training_step_time": 7.617210626602173
    },
    {
      "epoch": 4.633808135986328e-05,
      "model_forward_time": 1.7418560981750488,
      "step": 48589
    },
    {
      "epoch": 4.633808135986328e-05,
      "step": 48589,
      "training_step_time": 7.624764919281006
    },
    {
      "epoch": 4.633903503417969e-05,
      "grad_norm": 0.13467654585838318,
      "learning_rate": 5.644870697095024e-06,
      "loss": 0.0585,
      "step": 48590
    },
    {
      "epoch": 4.633903503417969e-05,
      "model_forward_time": 1.7749524116516113,
      "step": 48590
    },
    {
      "dynamics_loss_avg": 0.07381486482918262,
      "epoch": 4.633903503417969e-05,
      "step": 48590
    },
    {
      "action_loss_avg": 0.0020746886788401754,
      "epoch": 4.633903503417969e-05,
      "step": 48590
    },
    {
      "epoch": 4.633903503417969e-05,
      "step": 48590,
      "training_step_time": 9.019362449645996
    },
    {
      "epoch": 4.6339988708496095e-05,
      "model_forward_time": 1.7322630882263184,
      "step": 48591
    },
    {
      "epoch": 4.6339988708496095e-05,
      "step": 48591,
      "training_step_time": 9.761714696884155
    },
    {
      "epoch": 4.63409423828125e-05,
      "model_forward_time": 1.7404828071594238,
      "step": 48592
    },
    {
      "epoch": 4.63409423828125e-05,
      "step": 48592,
      "training_step_time": 7.88751220703125
    },
    {
      "epoch": 4.634189605712891e-05,
      "model_forward_time": 1.7358372211456299,
      "step": 48593
    },
    {
      "epoch": 4.634189605712891e-05,
      "step": 48593,
      "training_step_time": 8.770864725112915
    },
    {
      "epoch": 4.6342849731445314e-05,
      "model_forward_time": 1.7365350723266602,
      "step": 48594
    },
    {
      "epoch": 4.6342849731445314e-05,
      "step": 48594,
      "training_step_time": 7.372382640838623
    },
    {
      "epoch": 4.634380340576172e-05,
      "model_forward_time": 1.7324557304382324,
      "step": 48595
    },
    {
      "epoch": 4.634380340576172e-05,
      "step": 48595,
      "training_step_time": 7.340871572494507
    },
    {
      "epoch": 4.634475708007812e-05,
      "model_forward_time": 1.734142541885376,
      "step": 48596
    },
    {
      "epoch": 4.634475708007812e-05,
      "step": 48596,
      "training_step_time": 7.666801929473877
    },
    {
      "epoch": 4.634571075439453e-05,
      "model_forward_time": 1.7374389171600342,
      "step": 48597
    },
    {
      "epoch": 4.634571075439453e-05,
      "step": 48597,
      "training_step_time": 7.56663966178894
    },
    {
      "epoch": 4.6346664428710937e-05,
      "model_forward_time": 1.7366125583648682,
      "step": 48598
    },
    {
      "epoch": 4.6346664428710937e-05,
      "step": 48598,
      "training_step_time": 7.623867034912109
    },
    {
      "epoch": 4.634761810302734e-05,
      "model_forward_time": 1.7526509761810303,
      "step": 48599
    },
    {
      "epoch": 4.634761810302734e-05,
      "step": 48599,
      "training_step_time": 7.960888624191284
    },
    {
      "epoch": 4.634857177734375e-05,
      "grad_norm": 0.13149301707744598,
      "learning_rate": 5.643231001896587e-06,
      "loss": 0.0565,
      "step": 48600
    },
    {
      "epoch": 4.634857177734375e-05,
      "model_forward_time": 2.9576303958892822,
      "step": 48600
    },
    {
      "dynamics_loss_avg": 0.05897950604557991,
      "epoch": 4.634857177734375e-05,
      "step": 48600
    },
    {
      "action_loss_avg": 0.001171419964521192,
      "epoch": 4.634857177734375e-05,
      "step": 48600
    },
    {
      "epoch": 4.634857177734375e-05,
      "step": 48600,
      "training_step_time": 9.619670867919922
    },
    {
      "epoch": 4.6349525451660155e-05,
      "model_forward_time": 1.7297720909118652,
      "step": 48601
    },
    {
      "epoch": 4.6349525451660155e-05,
      "step": 48601,
      "training_step_time": 7.82111668586731
    },
    {
      "epoch": 4.635047912597656e-05,
      "model_forward_time": 1.7530598640441895,
      "step": 48602
    },
    {
      "epoch": 4.635047912597656e-05,
      "step": 48602,
      "training_step_time": 7.799943208694458
    },
    {
      "epoch": 4.635143280029297e-05,
      "model_forward_time": 1.781134843826294,
      "step": 48603
    },
    {
      "epoch": 4.635143280029297e-05,
      "step": 48603,
      "training_step_time": 7.714510202407837
    },
    {
      "epoch": 4.6352386474609374e-05,
      "model_forward_time": 1.754805088043213,
      "step": 48604
    },
    {
      "epoch": 4.6352386474609374e-05,
      "step": 48604,
      "training_step_time": 7.293982267379761
    },
    {
      "epoch": 4.635334014892578e-05,
      "model_forward_time": 1.7511601448059082,
      "step": 48605
    },
    {
      "epoch": 4.635334014892578e-05,
      "step": 48605,
      "training_step_time": 7.658027410507202
    },
    {
      "epoch": 4.635429382324219e-05,
      "model_forward_time": 1.7535841464996338,
      "step": 48606
    },
    {
      "epoch": 4.635429382324219e-05,
      "step": 48606,
      "training_step_time": 7.507541179656982
    },
    {
      "epoch": 4.635524749755859e-05,
      "model_forward_time": 1.7530779838562012,
      "step": 48607
    },
    {
      "epoch": 4.635524749755859e-05,
      "step": 48607,
      "training_step_time": 7.544976234436035
    },
    {
      "epoch": 4.6356201171875e-05,
      "model_forward_time": 1.7510788440704346,
      "step": 48608
    },
    {
      "epoch": 4.6356201171875e-05,
      "step": 48608,
      "training_step_time": 8.54328179359436
    },
    {
      "epoch": 4.635715484619141e-05,
      "model_forward_time": 2.866419553756714,
      "step": 48609
    },
    {
      "epoch": 4.635715484619141e-05,
      "step": 48609,
      "training_step_time": 9.573436498641968
    },
    {
      "epoch": 4.635810852050781e-05,
      "grad_norm": 0.16931478679180145,
      "learning_rate": 5.641591236355373e-06,
      "loss": 0.0545,
      "step": 48610
    },
    {
      "epoch": 4.635810852050781e-05,
      "model_forward_time": 1.767134428024292,
      "step": 48610
    },
    {
      "dynamics_loss_avg": 0.06365383975207806,
      "epoch": 4.635810852050781e-05,
      "step": 48610
    },
    {
      "action_loss_avg": 0.0014830372936557978,
      "epoch": 4.635810852050781e-05,
      "step": 48610
    },
    {
      "epoch": 4.635810852050781e-05,
      "step": 48610,
      "training_step_time": 8.705004930496216
    },
    {
      "epoch": 4.6359062194824216e-05,
      "model_forward_time": 1.7480747699737549,
      "step": 48611
    },
    {
      "epoch": 4.6359062194824216e-05,
      "step": 48611,
      "training_step_time": 7.794013023376465
    },
    {
      "epoch": 4.636001586914063e-05,
      "model_forward_time": 1.7397007942199707,
      "step": 48612
    },
    {
      "epoch": 4.636001586914063e-05,
      "step": 48612,
      "training_step_time": 7.802587270736694
    },
    {
      "epoch": 4.636096954345703e-05,
      "model_forward_time": 1.752138614654541,
      "step": 48613
    },
    {
      "epoch": 4.636096954345703e-05,
      "step": 48613,
      "training_step_time": 7.262624502182007
    },
    {
      "epoch": 4.6361923217773435e-05,
      "model_forward_time": 1.7471165657043457,
      "step": 48614
    },
    {
      "epoch": 4.6361923217773435e-05,
      "step": 48614,
      "training_step_time": 7.669755697250366
    },
    {
      "epoch": 4.6362876892089846e-05,
      "model_forward_time": 1.7515511512756348,
      "step": 48615
    },
    {
      "epoch": 4.6362876892089846e-05,
      "step": 48615,
      "training_step_time": 7.609972953796387
    },
    {
      "epoch": 4.636383056640625e-05,
      "model_forward_time": 1.752253770828247,
      "step": 48616
    },
    {
      "epoch": 4.636383056640625e-05,
      "step": 48616,
      "training_step_time": 7.615671157836914
    },
    {
      "epoch": 4.6364784240722654e-05,
      "model_forward_time": 1.7483222484588623,
      "step": 48617
    },
    {
      "epoch": 4.6364784240722654e-05,
      "step": 48617,
      "training_step_time": 9.450213432312012
    },
    {
      "epoch": 4.6365737915039065e-05,
      "model_forward_time": 2.876999855041504,
      "step": 48618
    },
    {
      "epoch": 4.6365737915039065e-05,
      "step": 48618,
      "training_step_time": 10.89241075515747
    },
    {
      "epoch": 4.636669158935547e-05,
      "model_forward_time": 1.7351810932159424,
      "step": 48619
    },
    {
      "epoch": 4.636669158935547e-05,
      "step": 48619,
      "training_step_time": 9.755467414855957
    },
    {
      "epoch": 4.636764526367187e-05,
      "grad_norm": 0.15800561010837555,
      "learning_rate": 5.639951400650706e-06,
      "loss": 0.0606,
      "step": 48620
    },
    {
      "epoch": 4.636764526367187e-05,
      "model_forward_time": 1.7441775798797607,
      "step": 48620
    },
    {
      "dynamics_loss_avg": 0.06848203651607036,
      "epoch": 4.636764526367187e-05,
      "step": 48620
    },
    {
      "action_loss_avg": 0.0022459743602667006,
      "epoch": 4.636764526367187e-05,
      "step": 48620
    },
    {
      "epoch": 4.636764526367187e-05,
      "step": 48620,
      "training_step_time": 9.263304471969604
    },
    {
      "epoch": 4.636859893798828e-05,
      "model_forward_time": 1.7411844730377197,
      "step": 48621
    },
    {
      "epoch": 4.636859893798828e-05,
      "step": 48621,
      "training_step_time": 7.889288425445557
    },
    {
      "epoch": 4.636955261230469e-05,
      "model_forward_time": 1.7452750205993652,
      "step": 48622
    },
    {
      "epoch": 4.636955261230469e-05,
      "step": 48622,
      "training_step_time": 7.37592339515686
    },
    {
      "epoch": 4.637050628662109e-05,
      "model_forward_time": 1.7801909446716309,
      "step": 48623
    },
    {
      "epoch": 4.637050628662109e-05,
      "step": 48623,
      "training_step_time": 7.669041872024536
    },
    {
      "epoch": 4.63714599609375e-05,
      "model_forward_time": 1.739612102508545,
      "step": 48624
    },
    {
      "epoch": 4.63714599609375e-05,
      "step": 48624,
      "training_step_time": 7.367086887359619
    },
    {
      "epoch": 4.6372413635253906e-05,
      "model_forward_time": 1.7459921836853027,
      "step": 48625
    },
    {
      "epoch": 4.6372413635253906e-05,
      "step": 48625,
      "training_step_time": 7.620602607727051
    },
    {
      "epoch": 4.637336730957031e-05,
      "model_forward_time": 1.7468829154968262,
      "step": 48626
    },
    {
      "epoch": 4.637336730957031e-05,
      "step": 48626,
      "training_step_time": 7.707901239395142
    },
    {
      "epoch": 4.637432098388672e-05,
      "model_forward_time": 2.9205989837646484,
      "step": 48627
    },
    {
      "epoch": 4.637432098388672e-05,
      "step": 48627,
      "training_step_time": 9.904008626937866
    },
    {
      "epoch": 4.6375274658203125e-05,
      "model_forward_time": 1.7432653903961182,
      "step": 48628
    },
    {
      "epoch": 4.6375274658203125e-05,
      "step": 48628,
      "training_step_time": 8.675987243652344
    },
    {
      "epoch": 4.637622833251953e-05,
      "model_forward_time": 1.743786096572876,
      "step": 48629
    },
    {
      "epoch": 4.637622833251953e-05,
      "step": 48629,
      "training_step_time": 7.852584362030029
    },
    {
      "epoch": 4.637718200683594e-05,
      "grad_norm": 0.21149468421936035,
      "learning_rate": 5.638311494961916e-06,
      "loss": 0.0588,
      "step": 48630
    },
    {
      "epoch": 4.637718200683594e-05,
      "model_forward_time": 1.7421600818634033,
      "step": 48630
    },
    {
      "dynamics_loss_avg": 0.06204524049535394,
      "epoch": 4.637718200683594e-05,
      "step": 48630
    },
    {
      "action_loss_avg": 0.0027890364115592094,
      "epoch": 4.637718200683594e-05,
      "step": 48630
    },
    {
      "epoch": 4.637718200683594e-05,
      "step": 48630,
      "training_step_time": 7.792051792144775
    },
    {
      "epoch": 4.6378135681152344e-05,
      "model_forward_time": 1.7439560890197754,
      "step": 48631
    },
    {
      "epoch": 4.6378135681152344e-05,
      "step": 48631,
      "training_step_time": 7.375808954238892
    },
    {
      "epoch": 4.637908935546875e-05,
      "model_forward_time": 1.7476942539215088,
      "step": 48632
    },
    {
      "epoch": 4.637908935546875e-05,
      "step": 48632,
      "training_step_time": 7.723473310470581
    },
    {
      "epoch": 4.638004302978516e-05,
      "model_forward_time": 1.7506544589996338,
      "step": 48633
    },
    {
      "epoch": 4.638004302978516e-05,
      "step": 48633,
      "training_step_time": 7.380918264389038
    },
    {
      "epoch": 4.638099670410156e-05,
      "model_forward_time": 1.731872320175171,
      "step": 48634
    },
    {
      "epoch": 4.638099670410156e-05,
      "step": 48634,
      "training_step_time": 7.620249271392822
    },
    {
      "epoch": 4.638195037841797e-05,
      "model_forward_time": 1.74688720703125,
      "step": 48635
    },
    {
      "epoch": 4.638195037841797e-05,
      "step": 48635,
      "training_step_time": 8.651203632354736
    },
    {
      "epoch": 4.638290405273438e-05,
      "model_forward_time": 2.8204751014709473,
      "step": 48636
    },
    {
      "epoch": 4.638290405273438e-05,
      "step": 48636,
      "training_step_time": 8.780028820037842
    },
    {
      "epoch": 4.638385772705078e-05,
      "model_forward_time": 1.8718950748443604,
      "step": 48637
    },
    {
      "epoch": 4.638385772705078e-05,
      "step": 48637,
      "training_step_time": 9.676491498947144
    },
    {
      "epoch": 4.6384811401367186e-05,
      "model_forward_time": 1.7486202716827393,
      "step": 48638
    },
    {
      "epoch": 4.6384811401367186e-05,
      "step": 48638,
      "training_step_time": 7.832753658294678
    },
    {
      "epoch": 4.6385765075683597e-05,
      "model_forward_time": 1.7508559226989746,
      "step": 48639
    },
    {
      "epoch": 4.6385765075683597e-05,
      "step": 48639,
      "training_step_time": 8.54172134399414
    },
    {
      "epoch": 4.638671875e-05,
      "grad_norm": 0.11057192832231522,
      "learning_rate": 5.636671519468342e-06,
      "loss": 0.0582,
      "step": 48640
    },
    {
      "epoch": 4.638671875e-05,
      "model_forward_time": 1.7448451519012451,
      "step": 48640
    },
    {
      "dynamics_loss_avg": 0.05915588028728962,
      "epoch": 4.638671875e-05,
      "step": 48640
    },
    {
      "action_loss_avg": 0.0017232668935321272,
      "epoch": 4.638671875e-05,
      "step": 48640
    },
    {
      "epoch": 4.638671875e-05,
      "step": 48640,
      "training_step_time": 7.793771028518677
    },
    {
      "epoch": 4.6387672424316405e-05,
      "model_forward_time": 1.748260498046875,
      "step": 48641
    },
    {
      "epoch": 4.6387672424316405e-05,
      "step": 48641,
      "training_step_time": 7.568008899688721
    },
    {
      "epoch": 4.6388626098632815e-05,
      "model_forward_time": 1.7544183731079102,
      "step": 48642
    },
    {
      "epoch": 4.6388626098632815e-05,
      "step": 48642,
      "training_step_time": 7.567215442657471
    },
    {
      "epoch": 4.638957977294922e-05,
      "model_forward_time": 1.747643232345581,
      "step": 48643
    },
    {
      "epoch": 4.638957977294922e-05,
      "step": 48643,
      "training_step_time": 7.541025638580322
    },
    {
      "epoch": 4.6390533447265623e-05,
      "model_forward_time": 1.744642734527588,
      "step": 48644
    },
    {
      "epoch": 4.6390533447265623e-05,
      "step": 48644,
      "training_step_time": 7.561168670654297
    },
    {
      "epoch": 4.6391487121582034e-05,
      "model_forward_time": 1.7450144290924072,
      "step": 48645
    },
    {
      "epoch": 4.6391487121582034e-05,
      "step": 48645,
      "training_step_time": 9.919963598251343
    },
    {
      "epoch": 4.639244079589844e-05,
      "model_forward_time": 2.8953919410705566,
      "step": 48646
    },
    {
      "epoch": 4.639244079589844e-05,
      "step": 48646,
      "training_step_time": 7.811845064163208
    },
    {
      "epoch": 4.639339447021484e-05,
      "model_forward_time": 1.7457125186920166,
      "step": 48647
    },
    {
      "epoch": 4.639339447021484e-05,
      "step": 48647,
      "training_step_time": 7.877672433853149
    },
    {
      "epoch": 4.639434814453125e-05,
      "model_forward_time": 1.7412683963775635,
      "step": 48648
    },
    {
      "epoch": 4.639434814453125e-05,
      "step": 48648,
      "training_step_time": 8.412552833557129
    },
    {
      "epoch": 4.639530181884766e-05,
      "model_forward_time": 1.745382308959961,
      "step": 48649
    },
    {
      "epoch": 4.639530181884766e-05,
      "step": 48649,
      "training_step_time": 7.816677093505859
    },
    {
      "epoch": 4.639625549316406e-05,
      "grad_norm": 0.13382560014724731,
      "learning_rate": 5.635031474349327e-06,
      "loss": 0.0618,
      "step": 48650
    },
    {
      "epoch": 4.639625549316406e-05,
      "model_forward_time": 1.7442755699157715,
      "step": 48650
    },
    {
      "dynamics_loss_avg": 0.048194673284888266,
      "epoch": 4.639625549316406e-05,
      "step": 48650
    },
    {
      "action_loss_avg": 0.0013185558826080523,
      "epoch": 4.639625549316406e-05,
      "step": 48650
    },
    {
      "epoch": 4.639625549316406e-05,
      "step": 48650,
      "training_step_time": 7.700488090515137
    },
    {
      "epoch": 4.639720916748047e-05,
      "model_forward_time": 1.7433362007141113,
      "step": 48651
    },
    {
      "epoch": 4.639720916748047e-05,
      "step": 48651,
      "training_step_time": 7.383447647094727
    },
    {
      "epoch": 4.6398162841796876e-05,
      "model_forward_time": 1.7476921081542969,
      "step": 48652
    },
    {
      "epoch": 4.6398162841796876e-05,
      "step": 48652,
      "training_step_time": 7.57498025894165
    },
    {
      "epoch": 4.639911651611328e-05,
      "model_forward_time": 1.7348105907440186,
      "step": 48653
    },
    {
      "epoch": 4.639911651611328e-05,
      "step": 48653,
      "training_step_time": 8.336541414260864
    },
    {
      "epoch": 4.640007019042969e-05,
      "model_forward_time": 1.7441251277923584,
      "step": 48654
    },
    {
      "epoch": 4.640007019042969e-05,
      "step": 48654,
      "training_step_time": 8.797234058380127
    },
    {
      "epoch": 4.6401023864746095e-05,
      "model_forward_time": 2.8794515132904053,
      "step": 48655
    },
    {
      "epoch": 4.6401023864746095e-05,
      "step": 48655,
      "training_step_time": 9.593332767486572
    },
    {
      "epoch": 4.64019775390625e-05,
      "model_forward_time": 1.7438974380493164,
      "step": 48656
    },
    {
      "epoch": 4.64019775390625e-05,
      "step": 48656,
      "training_step_time": 8.425085544586182
    },
    {
      "epoch": 4.64029312133789e-05,
      "model_forward_time": 1.737924337387085,
      "step": 48657
    },
    {
      "epoch": 4.64029312133789e-05,
      "step": 48657,
      "training_step_time": 7.850977182388306
    },
    {
      "epoch": 4.6403884887695314e-05,
      "model_forward_time": 1.7446906566619873,
      "step": 48658
    },
    {
      "epoch": 4.6403884887695314e-05,
      "step": 48658,
      "training_step_time": 7.738678932189941
    },
    {
      "epoch": 4.640483856201172e-05,
      "model_forward_time": 1.7901504039764404,
      "step": 48659
    },
    {
      "epoch": 4.640483856201172e-05,
      "step": 48659,
      "training_step_time": 7.701863050460815
    },
    {
      "epoch": 4.640579223632812e-05,
      "grad_norm": 0.1415470391511917,
      "learning_rate": 5.633391359784225e-06,
      "loss": 0.0571,
      "step": 48660
    },
    {
      "epoch": 4.640579223632812e-05,
      "model_forward_time": 1.7516272068023682,
      "step": 48660
    },
    {
      "dynamics_loss_avg": 0.046739500109106304,
      "epoch": 4.640579223632812e-05,
      "step": 48660
    },
    {
      "action_loss_avg": 0.0009247393812984228,
      "epoch": 4.640579223632812e-05,
      "step": 48660
    },
    {
      "epoch": 4.640579223632812e-05,
      "step": 48660,
      "training_step_time": 7.294852018356323
    },
    {
      "epoch": 4.640674591064453e-05,
      "model_forward_time": 1.7549428939819336,
      "step": 48661
    },
    {
      "epoch": 4.640674591064453e-05,
      "step": 48661,
      "training_step_time": 7.5047149658203125
    },
    {
      "epoch": 4.6407699584960937e-05,
      "model_forward_time": 1.751004695892334,
      "step": 48662
    },
    {
      "epoch": 4.6407699584960937e-05,
      "step": 48662,
      "training_step_time": 7.615121603012085
    },
    {
      "epoch": 4.640865325927734e-05,
      "model_forward_time": 1.757380723953247,
      "step": 48663
    },
    {
      "epoch": 4.640865325927734e-05,
      "step": 48663,
      "training_step_time": 8.771287441253662
    },
    {
      "epoch": 4.640960693359375e-05,
      "model_forward_time": 2.8717164993286133,
      "step": 48664
    },
    {
      "epoch": 4.640960693359375e-05,
      "step": 48664,
      "training_step_time": 8.577629566192627
    },
    {
      "epoch": 4.6410560607910155e-05,
      "model_forward_time": 1.7444920539855957,
      "step": 48665
    },
    {
      "epoch": 4.6410560607910155e-05,
      "step": 48665,
      "training_step_time": 9.672693967819214
    },
    {
      "epoch": 4.641151428222656e-05,
      "model_forward_time": 1.7621371746063232,
      "step": 48666
    },
    {
      "epoch": 4.641151428222656e-05,
      "step": 48666,
      "training_step_time": 7.938700199127197
    },
    {
      "epoch": 4.641246795654297e-05,
      "model_forward_time": 1.3952279090881348,
      "step": 48667
    },
    {
      "epoch": 4.641246795654297e-05,
      "step": 48667,
      "training_step_time": 7.786865234375
    },
    {
      "epoch": 4.6413421630859374e-05,
      "model_forward_time": 1.7316677570343018,
      "step": 48668
    },
    {
      "epoch": 4.6413421630859374e-05,
      "step": 48668,
      "training_step_time": 7.832605361938477
    },
    {
      "epoch": 4.641437530517578e-05,
      "model_forward_time": 1.749023199081421,
      "step": 48669
    },
    {
      "epoch": 4.641437530517578e-05,
      "step": 48669,
      "training_step_time": 7.3395469188690186
    },
    {
      "epoch": 4.641532897949219e-05,
      "grad_norm": 0.12999369204044342,
      "learning_rate": 5.631751175952396e-06,
      "loss": 0.058,
      "step": 48670
    },
    {
      "epoch": 4.641532897949219e-05,
      "model_forward_time": 1.7440483570098877,
      "step": 48670
    },
    {
      "dynamics_loss_avg": 0.05407355595380068,
      "epoch": 4.641532897949219e-05,
      "step": 48670
    },
    {
      "action_loss_avg": 0.0021017017133999616,
      "epoch": 4.641532897949219e-05,
      "step": 48670
    },
    {
      "epoch": 4.641532897949219e-05,
      "step": 48670,
      "training_step_time": 7.426788330078125
    },
    {
      "epoch": 4.641628265380859e-05,
      "model_forward_time": 1.7541840076446533,
      "step": 48671
    },
    {
      "epoch": 4.641628265380859e-05,
      "step": 48671,
      "training_step_time": 7.70316481590271
    },
    {
      "epoch": 4.6417236328125e-05,
      "model_forward_time": 1.759836196899414,
      "step": 48672
    },
    {
      "epoch": 4.6417236328125e-05,
      "step": 48672,
      "training_step_time": 7.712308406829834
    },
    {
      "epoch": 4.641819000244141e-05,
      "model_forward_time": 2.9100022315979004,
      "step": 48673
    },
    {
      "epoch": 4.641819000244141e-05,
      "step": 48673,
      "training_step_time": 9.230747938156128
    },
    {
      "epoch": 4.641914367675781e-05,
      "model_forward_time": 1.7518794536590576,
      "step": 48674
    },
    {
      "epoch": 4.641914367675781e-05,
      "step": 48674,
      "training_step_time": 9.57188606262207
    },
    {
      "epoch": 4.6420097351074216e-05,
      "model_forward_time": 1.7264111042022705,
      "step": 48675
    },
    {
      "epoch": 4.6420097351074216e-05,
      "step": 48675,
      "training_step_time": 7.89035439491272
    },
    {
      "epoch": 4.642105102539063e-05,
      "model_forward_time": 1.7457633018493652,
      "step": 48676
    },
    {
      "epoch": 4.642105102539063e-05,
      "step": 48676,
      "training_step_time": 7.774383306503296
    },
    {
      "epoch": 4.642200469970703e-05,
      "model_forward_time": 1.740504503250122,
      "step": 48677
    },
    {
      "epoch": 4.642200469970703e-05,
      "step": 48677,
      "training_step_time": 7.775940656661987
    },
    {
      "epoch": 4.6422958374023435e-05,
      "model_forward_time": 1.7479639053344727,
      "step": 48678
    },
    {
      "epoch": 4.6422958374023435e-05,
      "step": 48678,
      "training_step_time": 7.399572134017944
    },
    {
      "epoch": 4.6423912048339846e-05,
      "model_forward_time": 1.7461202144622803,
      "step": 48679
    },
    {
      "epoch": 4.6423912048339846e-05,
      "step": 48679,
      "training_step_time": 6.926926136016846
    },
    {
      "epoch": 4.642486572265625e-05,
      "grad_norm": 0.12852314114570618,
      "learning_rate": 5.6301109230332075e-06,
      "loss": 0.057,
      "step": 48680
    },
    {
      "epoch": 4.642486572265625e-05,
      "model_forward_time": 1.7482500076293945,
      "step": 48680
    },
    {
      "dynamics_loss_avg": 0.05799370184540749,
      "epoch": 4.642486572265625e-05,
      "step": 48680
    },
    {
      "action_loss_avg": 0.0033245508559048174,
      "epoch": 4.642486572265625e-05,
      "step": 48680
    },
    {
      "epoch": 4.642486572265625e-05,
      "step": 48680,
      "training_step_time": 7.514772176742554
    },
    {
      "epoch": 4.6425819396972654e-05,
      "model_forward_time": 1.7501003742218018,
      "step": 48681
    },
    {
      "epoch": 4.6425819396972654e-05,
      "step": 48681,
      "training_step_time": 8.729207038879395
    },
    {
      "epoch": 4.6426773071289065e-05,
      "model_forward_time": 1.8065333366394043,
      "step": 48682
    },
    {
      "epoch": 4.6426773071289065e-05,
      "step": 48682,
      "training_step_time": 8.953214168548584
    },
    {
      "epoch": 4.642772674560547e-05,
      "model_forward_time": 1.7587871551513672,
      "step": 48683
    },
    {
      "epoch": 4.642772674560547e-05,
      "step": 48683,
      "training_step_time": 9.191832304000854
    },
    {
      "epoch": 4.642868041992187e-05,
      "model_forward_time": 1.7467005252838135,
      "step": 48684
    },
    {
      "epoch": 4.642868041992187e-05,
      "step": 48684,
      "training_step_time": 8.612390756607056
    },
    {
      "epoch": 4.6429634094238283e-05,
      "model_forward_time": 1.741894006729126,
      "step": 48685
    },
    {
      "epoch": 4.6429634094238283e-05,
      "step": 48685,
      "training_step_time": 7.650235652923584
    },
    {
      "epoch": 4.643058776855469e-05,
      "model_forward_time": 1.7493922710418701,
      "step": 48686
    },
    {
      "epoch": 4.643058776855469e-05,
      "step": 48686,
      "training_step_time": 7.777606964111328
    },
    {
      "epoch": 4.643154144287109e-05,
      "model_forward_time": 1.7633512020111084,
      "step": 48687
    },
    {
      "epoch": 4.643154144287109e-05,
      "step": 48687,
      "training_step_time": 7.347151279449463
    },
    {
      "epoch": 4.64324951171875e-05,
      "model_forward_time": 1.7408757209777832,
      "step": 48688
    },
    {
      "epoch": 4.64324951171875e-05,
      "step": 48688,
      "training_step_time": 7.274724006652832
    },
    {
      "epoch": 4.6433448791503906e-05,
      "model_forward_time": 1.7463231086730957,
      "step": 48689
    },
    {
      "epoch": 4.6433448791503906e-05,
      "step": 48689,
      "training_step_time": 7.540540456771851
    },
    {
      "epoch": 4.643440246582031e-05,
      "grad_norm": 0.13560964167118073,
      "learning_rate": 5.6284706012060365e-06,
      "loss": 0.056,
      "step": 48690
    },
    {
      "epoch": 4.643440246582031e-05,
      "model_forward_time": 1.7440440654754639,
      "step": 48690
    },
    {
      "dynamics_loss_avg": 0.0731329970061779,
      "epoch": 4.643440246582031e-05,
      "step": 48690
    },
    {
      "action_loss_avg": 0.0011504083784529939,
      "epoch": 4.643440246582031e-05,
      "step": 48690
    },
    {
      "epoch": 4.643440246582031e-05,
      "step": 48690,
      "training_step_time": 7.658723831176758
    },
    {
      "epoch": 4.643535614013672e-05,
      "model_forward_time": 1.7760298252105713,
      "step": 48691
    },
    {
      "epoch": 4.643535614013672e-05,
      "step": 48691,
      "training_step_time": 7.882542371749878
    },
    {
      "epoch": 4.6436309814453125e-05,
      "model_forward_time": 2.912517786026001,
      "step": 48692
    },
    {
      "epoch": 4.6436309814453125e-05,
      "step": 48692,
      "training_step_time": 9.524657487869263
    },
    {
      "epoch": 4.643726348876953e-05,
      "model_forward_time": 1.7428414821624756,
      "step": 48693
    },
    {
      "epoch": 4.643726348876953e-05,
      "step": 48693,
      "training_step_time": 8.462876558303833
    },
    {
      "epoch": 4.643821716308594e-05,
      "model_forward_time": 1.738973617553711,
      "step": 48694
    },
    {
      "epoch": 4.643821716308594e-05,
      "step": 48694,
      "training_step_time": 7.782109498977661
    },
    {
      "epoch": 4.6439170837402344e-05,
      "model_forward_time": 1.7422170639038086,
      "step": 48695
    },
    {
      "epoch": 4.6439170837402344e-05,
      "step": 48695,
      "training_step_time": 7.708393335342407
    },
    {
      "epoch": 4.644012451171875e-05,
      "model_forward_time": 1.7482450008392334,
      "step": 48696
    },
    {
      "epoch": 4.644012451171875e-05,
      "step": 48696,
      "training_step_time": 7.353015184402466
    },
    {
      "epoch": 4.644107818603516e-05,
      "model_forward_time": 1.7743303775787354,
      "step": 48697
    },
    {
      "epoch": 4.644107818603516e-05,
      "step": 48697,
      "training_step_time": 7.31021785736084
    },
    {
      "epoch": 4.644203186035156e-05,
      "model_forward_time": 1.7490053176879883,
      "step": 48698
    },
    {
      "epoch": 4.644203186035156e-05,
      "step": 48698,
      "training_step_time": 7.581238031387329
    },
    {
      "epoch": 4.644298553466797e-05,
      "model_forward_time": 1.745636224746704,
      "step": 48699
    },
    {
      "epoch": 4.644298553466797e-05,
      "step": 48699,
      "training_step_time": 8.289831638336182
    },
    {
      "epoch": 4.644393920898438e-05,
      "grad_norm": 0.11467006057500839,
      "learning_rate": 5.626830210650266e-06,
      "loss": 0.0586,
      "step": 48700
    },
    {
      "epoch": 4.644393920898438e-05,
      "model_forward_time": 1.7452030181884766,
      "step": 48700
    },
    {
      "dynamics_loss_avg": 0.07382077239453792,
      "epoch": 4.644393920898438e-05,
      "step": 48700
    },
    {
      "action_loss_avg": 0.0022484035609522833,
      "epoch": 4.644393920898438e-05,
      "step": 48700
    },
    {
      "epoch": 4.644393920898438e-05,
      "step": 48700,
      "training_step_time": 7.941997289657593
    },
    {
      "epoch": 4.644489288330078e-05,
      "model_forward_time": 2.8800997734069824,
      "step": 48701
    },
    {
      "epoch": 4.644489288330078e-05,
      "step": 48701,
      "training_step_time": 9.689507007598877
    },
    {
      "epoch": 4.6445846557617186e-05,
      "model_forward_time": 1.7504425048828125,
      "step": 48702
    },
    {
      "epoch": 4.6445846557617186e-05,
      "step": 48702,
      "training_step_time": 9.23478388786316
    },
    {
      "epoch": 4.6446800231933597e-05,
      "model_forward_time": 1.7460944652557373,
      "step": 48703
    },
    {
      "epoch": 4.6446800231933597e-05,
      "step": 48703,
      "training_step_time": 7.908732175827026
    },
    {
      "epoch": 4.644775390625e-05,
      "model_forward_time": 1.729933500289917,
      "step": 48704
    },
    {
      "epoch": 4.644775390625e-05,
      "step": 48704,
      "training_step_time": 7.793386697769165
    },
    {
      "epoch": 4.6448707580566405e-05,
      "model_forward_time": 1.7485568523406982,
      "step": 48705
    },
    {
      "epoch": 4.6448707580566405e-05,
      "step": 48705,
      "training_step_time": 7.780743598937988
    },
    {
      "epoch": 4.6449661254882815e-05,
      "model_forward_time": 1.7481486797332764,
      "step": 48706
    },
    {
      "epoch": 4.6449661254882815e-05,
      "step": 48706,
      "training_step_time": 7.3936767578125
    },
    {
      "epoch": 4.645061492919922e-05,
      "model_forward_time": 1.7505543231964111,
      "step": 48707
    },
    {
      "epoch": 4.645061492919922e-05,
      "step": 48707,
      "training_step_time": 7.571120262145996
    },
    {
      "epoch": 4.6451568603515623e-05,
      "model_forward_time": 1.7507143020629883,
      "step": 48708
    },
    {
      "epoch": 4.6451568603515623e-05,
      "step": 48708,
      "training_step_time": 7.651817798614502
    },
    {
      "epoch": 4.6452522277832034e-05,
      "model_forward_time": 1.7490394115447998,
      "step": 48709
    },
    {
      "epoch": 4.6452522277832034e-05,
      "step": 48709,
      "training_step_time": 8.473883867263794
    },
    {
      "epoch": 4.645347595214844e-05,
      "grad_norm": 0.13788390159606934,
      "learning_rate": 5.625189751545285e-06,
      "loss": 0.0567,
      "step": 48710
    },
    {
      "epoch": 4.645347595214844e-05,
      "model_forward_time": 2.8621015548706055,
      "step": 48710
    },
    {
      "dynamics_loss_avg": 0.05792049653828144,
      "epoch": 4.645347595214844e-05,
      "step": 48710
    },
    {
      "action_loss_avg": 0.002298158721532673,
      "epoch": 4.645347595214844e-05,
      "step": 48710
    },
    {
      "epoch": 4.645347595214844e-05,
      "step": 48710,
      "training_step_time": 7.863855600357056
    },
    {
      "epoch": 4.645442962646484e-05,
      "model_forward_time": 1.7430496215820312,
      "step": 48711
    },
    {
      "epoch": 4.645442962646484e-05,
      "step": 48711,
      "training_step_time": 8.441643953323364
    },
    {
      "epoch": 4.645538330078125e-05,
      "model_forward_time": 1.749847412109375,
      "step": 48712
    },
    {
      "epoch": 4.645538330078125e-05,
      "step": 48712,
      "training_step_time": 7.960845470428467
    },
    {
      "epoch": 4.645633697509766e-05,
      "model_forward_time": 1.7458465099334717,
      "step": 48713
    },
    {
      "epoch": 4.645633697509766e-05,
      "step": 48713,
      "training_step_time": 7.826564788818359
    },
    {
      "epoch": 4.645729064941406e-05,
      "model_forward_time": 1.7458372116088867,
      "step": 48714
    },
    {
      "epoch": 4.645729064941406e-05,
      "step": 48714,
      "training_step_time": 7.89910888671875
    },
    {
      "epoch": 4.645824432373047e-05,
      "model_forward_time": 1.7463722229003906,
      "step": 48715
    },
    {
      "epoch": 4.645824432373047e-05,
      "step": 48715,
      "training_step_time": 7.422403573989868
    },
    {
      "epoch": 4.6459197998046876e-05,
      "model_forward_time": 1.7465274333953857,
      "step": 48716
    },
    {
      "epoch": 4.6459197998046876e-05,
      "step": 48716,
      "training_step_time": 7.379445791244507
    },
    {
      "epoch": 4.646015167236328e-05,
      "model_forward_time": 1.7417590618133545,
      "step": 48717
    },
    {
      "epoch": 4.646015167236328e-05,
      "step": 48717,
      "training_step_time": 7.6883509159088135
    },
    {
      "epoch": 4.646110534667969e-05,
      "model_forward_time": 1.7476813793182373,
      "step": 48718
    },
    {
      "epoch": 4.646110534667969e-05,
      "step": 48718,
      "training_step_time": 8.294146060943604
    },
    {
      "epoch": 4.6462059020996095e-05,
      "model_forward_time": 2.860330820083618,
      "step": 48719
    },
    {
      "epoch": 4.6462059020996095e-05,
      "step": 48719,
      "training_step_time": 9.090975761413574
    },
    {
      "epoch": 4.64630126953125e-05,
      "grad_norm": 0.11129381507635117,
      "learning_rate": 5.623549224070494e-06,
      "loss": 0.0563,
      "step": 48720
    },
    {
      "epoch": 4.64630126953125e-05,
      "model_forward_time": 1.741441249847412,
      "step": 48720
    },
    {
      "dynamics_loss_avg": 0.06565963048487902,
      "epoch": 4.64630126953125e-05,
      "step": 48720
    },
    {
      "action_loss_avg": 0.0015666221268475055,
      "epoch": 4.64630126953125e-05,
      "step": 48720
    },
    {
      "epoch": 4.64630126953125e-05,
      "step": 48720,
      "training_step_time": 10.443203926086426
    },
    {
      "epoch": 4.64639663696289e-05,
      "model_forward_time": 1.7420148849487305,
      "step": 48721
    },
    {
      "epoch": 4.64639663696289e-05,
      "step": 48721,
      "training_step_time": 7.956531763076782
    },
    {
      "epoch": 4.6464920043945314e-05,
      "model_forward_time": 1.7473745346069336,
      "step": 48722
    },
    {
      "epoch": 4.6464920043945314e-05,
      "step": 48722,
      "training_step_time": 7.893767833709717
    },
    {
      "epoch": 4.646587371826172e-05,
      "model_forward_time": 1.7433266639709473,
      "step": 48723
    },
    {
      "epoch": 4.646587371826172e-05,
      "step": 48723,
      "training_step_time": 7.837213754653931
    },
    {
      "epoch": 4.646682739257812e-05,
      "model_forward_time": 1.7426238059997559,
      "step": 48724
    },
    {
      "epoch": 4.646682739257812e-05,
      "step": 48724,
      "training_step_time": 7.417819499969482
    },
    {
      "epoch": 4.646778106689453e-05,
      "model_forward_time": 1.7366878986358643,
      "step": 48725
    },
    {
      "epoch": 4.646778106689453e-05,
      "step": 48725,
      "training_step_time": 7.382223129272461
    },
    {
      "epoch": 4.646873474121094e-05,
      "model_forward_time": 1.773885726928711,
      "step": 48726
    },
    {
      "epoch": 4.646873474121094e-05,
      "step": 48726,
      "training_step_time": 7.688830375671387
    },
    {
      "epoch": 4.646968841552734e-05,
      "model_forward_time": 1.7277638912200928,
      "step": 48727
    },
    {
      "epoch": 4.646968841552734e-05,
      "step": 48727,
      "training_step_time": 8.56225323677063
    },
    {
      "epoch": 4.647064208984375e-05,
      "model_forward_time": 2.8635339736938477,
      "step": 48728
    },
    {
      "epoch": 4.647064208984375e-05,
      "step": 48728,
      "training_step_time": 8.807389259338379
    },
    {
      "epoch": 4.6471595764160155e-05,
      "model_forward_time": 1.7418396472930908,
      "step": 48729
    },
    {
      "epoch": 4.6471595764160155e-05,
      "step": 48729,
      "training_step_time": 7.747331142425537
    },
    {
      "epoch": 4.647254943847656e-05,
      "grad_norm": 0.08435610681772232,
      "learning_rate": 5.621908628405296e-06,
      "loss": 0.0557,
      "step": 48730
    },
    {
      "epoch": 4.647254943847656e-05,
      "model_forward_time": 1.745673418045044,
      "step": 48730
    },
    {
      "dynamics_loss_avg": 0.06282009724527597,
      "epoch": 4.647254943847656e-05,
      "step": 48730
    },
    {
      "action_loss_avg": 0.0019733027031179516,
      "epoch": 4.647254943847656e-05,
      "step": 48730
    },
    {
      "epoch": 4.647254943847656e-05,
      "step": 48730,
      "training_step_time": 7.895540714263916
    },
    {
      "epoch": 4.647350311279297e-05,
      "model_forward_time": 1.7404823303222656,
      "step": 48731
    },
    {
      "epoch": 4.647350311279297e-05,
      "step": 48731,
      "training_step_time": 7.854735374450684
    },
    {
      "epoch": 4.6474456787109374e-05,
      "model_forward_time": 1.7505543231964111,
      "step": 48732
    },
    {
      "epoch": 4.6474456787109374e-05,
      "step": 48732,
      "training_step_time": 7.778390884399414
    },
    {
      "epoch": 4.647541046142578e-05,
      "model_forward_time": 1.8135418891906738,
      "step": 48733
    },
    {
      "epoch": 4.647541046142578e-05,
      "step": 48733,
      "training_step_time": 7.434486627578735
    },
    {
      "epoch": 4.647636413574219e-05,
      "model_forward_time": 1.7400879859924316,
      "step": 48734
    },
    {
      "epoch": 4.647636413574219e-05,
      "step": 48734,
      "training_step_time": 7.218288898468018
    },
    {
      "epoch": 4.647731781005859e-05,
      "model_forward_time": 1.7442238330841064,
      "step": 48735
    },
    {
      "epoch": 4.647731781005859e-05,
      "step": 48735,
      "training_step_time": 7.630444288253784
    },
    {
      "epoch": 4.6478271484375e-05,
      "model_forward_time": 1.7480580806732178,
      "step": 48736
    },
    {
      "epoch": 4.6478271484375e-05,
      "step": 48736,
      "training_step_time": 7.727679014205933
    },
    {
      "epoch": 4.647922515869141e-05,
      "model_forward_time": 1.7112140655517578,
      "step": 48737
    },
    {
      "epoch": 4.647922515869141e-05,
      "step": 48737,
      "training_step_time": 7.800871849060059
    },
    {
      "epoch": 4.648017883300781e-05,
      "model_forward_time": 2.9166829586029053,
      "step": 48738
    },
    {
      "epoch": 4.648017883300781e-05,
      "step": 48738,
      "training_step_time": 8.487120151519775
    },
    {
      "epoch": 4.6481132507324216e-05,
      "model_forward_time": 1.3688504695892334,
      "step": 48739
    },
    {
      "epoch": 4.6481132507324216e-05,
      "step": 48739,
      "training_step_time": 9.180214166641235
    },
    {
      "epoch": 4.648208618164063e-05,
      "grad_norm": 0.14447742700576782,
      "learning_rate": 5.620267964729106e-06,
      "loss": 0.0569,
      "step": 48740
    },
    {
      "epoch": 4.648208618164063e-05,
      "model_forward_time": 1.7218906879425049,
      "step": 48740
    },
    {
      "dynamics_loss_avg": 0.05108576063066721,
      "epoch": 4.648208618164063e-05,
      "step": 48740
    },
    {
      "action_loss_avg": 0.0024751366407144816,
      "epoch": 4.648208618164063e-05,
      "step": 48740
    },
    {
      "epoch": 4.648208618164063e-05,
      "step": 48740,
      "training_step_time": 7.659355401992798
    },
    {
      "epoch": 4.648303985595703e-05,
      "model_forward_time": 1.3542068004608154,
      "step": 48741
    },
    {
      "epoch": 4.648303985595703e-05,
      "step": 48741,
      "training_step_time": 8.085317373275757
    },
    {
      "epoch": 4.6483993530273435e-05,
      "model_forward_time": 1.7244975566864014,
      "step": 48742
    },
    {
      "epoch": 4.6483993530273435e-05,
      "step": 48742,
      "training_step_time": 9.034181118011475
    },
    {
      "epoch": 4.6484947204589846e-05,
      "model_forward_time": 1.733402967453003,
      "step": 48743
    },
    {
      "epoch": 4.6484947204589846e-05,
      "step": 48743,
      "training_step_time": 7.248493432998657
    },
    {
      "epoch": 4.648590087890625e-05,
      "model_forward_time": 1.7250514030456543,
      "step": 48744
    },
    {
      "epoch": 4.648590087890625e-05,
      "step": 48744,
      "training_step_time": 7.511061191558838
    },
    {
      "epoch": 4.6486854553222654e-05,
      "model_forward_time": 1.721524953842163,
      "step": 48745
    },
    {
      "epoch": 4.6486854553222654e-05,
      "step": 48745,
      "training_step_time": 8.389350652694702
    },
    {
      "epoch": 4.6487808227539065e-05,
      "model_forward_time": 1.7302131652832031,
      "step": 48746
    },
    {
      "epoch": 4.6487808227539065e-05,
      "step": 48746,
      "training_step_time": 10.067691326141357
    },
    {
      "epoch": 4.648876190185547e-05,
      "model_forward_time": 2.882009983062744,
      "step": 48747
    },
    {
      "epoch": 4.648876190185547e-05,
      "step": 48747,
      "training_step_time": 7.668314218521118
    },
    {
      "epoch": 4.648971557617187e-05,
      "model_forward_time": 1.7232236862182617,
      "step": 48748
    },
    {
      "epoch": 4.648971557617187e-05,
      "step": 48748,
      "training_step_time": 8.716949224472046
    },
    {
      "epoch": 4.6490669250488283e-05,
      "model_forward_time": 1.722015380859375,
      "step": 48749
    },
    {
      "epoch": 4.6490669250488283e-05,
      "step": 48749,
      "training_step_time": 7.72451376914978
    },
    {
      "epoch": 4.649162292480469e-05,
      "grad_norm": 0.12014119327068329,
      "learning_rate": 5.618627233221344e-06,
      "loss": 0.0524,
      "step": 48750
    },
    {
      "epoch": 4.649162292480469e-05,
      "model_forward_time": 1.7571487426757812,
      "step": 48750
    },
    {
      "dynamics_loss_avg": 0.059647209197282794,
      "epoch": 4.649162292480469e-05,
      "step": 48750
    },
    {
      "action_loss_avg": 0.004613050085026771,
      "epoch": 4.649162292480469e-05,
      "step": 48750
    },
    {
      "epoch": 4.649162292480469e-05,
      "step": 48750,
      "training_step_time": 7.67350697517395
    },
    {
      "epoch": 4.649257659912109e-05,
      "model_forward_time": 1.7138245105743408,
      "step": 48751
    },
    {
      "epoch": 4.649257659912109e-05,
      "step": 48751,
      "training_step_time": 7.678182601928711
    },
    {
      "epoch": 4.64935302734375e-05,
      "model_forward_time": 1.7296700477600098,
      "step": 48752
    },
    {
      "epoch": 4.64935302734375e-05,
      "step": 48752,
      "training_step_time": 7.25193977355957
    },
    {
      "epoch": 4.6494483947753906e-05,
      "model_forward_time": 1.7199833393096924,
      "step": 48753
    },
    {
      "epoch": 4.6494483947753906e-05,
      "step": 48753,
      "training_step_time": 7.494927406311035
    },
    {
      "epoch": 4.649543762207031e-05,
      "model_forward_time": 1.7201204299926758,
      "step": 48754
    },
    {
      "epoch": 4.649543762207031e-05,
      "step": 48754,
      "training_step_time": 7.672182559967041
    },
    {
      "epoch": 4.649639129638672e-05,
      "model_forward_time": 1.721787929534912,
      "step": 48755
    },
    {
      "epoch": 4.649639129638672e-05,
      "step": 48755,
      "training_step_time": 8.53486680984497
    },
    {
      "epoch": 4.6497344970703125e-05,
      "model_forward_time": 2.9310121536254883,
      "step": 48756
    },
    {
      "epoch": 4.6497344970703125e-05,
      "step": 48756,
      "training_step_time": 8.732343673706055
    },
    {
      "epoch": 4.649829864501953e-05,
      "model_forward_time": 1.7205724716186523,
      "step": 48757
    },
    {
      "epoch": 4.649829864501953e-05,
      "step": 48757,
      "training_step_time": 7.700629949569702
    },
    {
      "epoch": 4.649925231933594e-05,
      "model_forward_time": 1.7125205993652344,
      "step": 48758
    },
    {
      "epoch": 4.649925231933594e-05,
      "step": 48758,
      "training_step_time": 7.6661200523376465
    },
    {
      "epoch": 4.6500205993652344e-05,
      "model_forward_time": 1.3623378276824951,
      "step": 48759
    },
    {
      "epoch": 4.6500205993652344e-05,
      "step": 48759,
      "training_step_time": 7.59356164932251
    },
    {
      "epoch": 4.650115966796875e-05,
      "grad_norm": 0.09513552486896515,
      "learning_rate": 5.616986434061438e-06,
      "loss": 0.0582,
      "step": 48760
    },
    {
      "epoch": 4.650115966796875e-05,
      "model_forward_time": 1.7321739196777344,
      "step": 48760
    },
    {
      "dynamics_loss_avg": 0.04626327306032181,
      "epoch": 4.650115966796875e-05,
      "step": 48760
    },
    {
      "action_loss_avg": 0.0012066969356965274,
      "epoch": 4.650115966796875e-05,
      "step": 48760
    },
    {
      "epoch": 4.650115966796875e-05,
      "step": 48760,
      "training_step_time": 7.5305328369140625
    },
    {
      "epoch": 4.650211334228516e-05,
      "model_forward_time": 1.7511730194091797,
      "step": 48761
    },
    {
      "epoch": 4.650211334228516e-05,
      "step": 48761,
      "training_step_time": 7.563383102416992
    },
    {
      "epoch": 4.650306701660156e-05,
      "model_forward_time": 1.7430648803710938,
      "step": 48762
    },
    {
      "epoch": 4.650306701660156e-05,
      "step": 48762,
      "training_step_time": 7.511730909347534
    },
    {
      "epoch": 4.650402069091797e-05,
      "model_forward_time": 1.7549786567687988,
      "step": 48763
    },
    {
      "epoch": 4.650402069091797e-05,
      "step": 48763,
      "training_step_time": 10.018791198730469
    },
    {
      "epoch": 4.650497436523438e-05,
      "model_forward_time": 1.7471606731414795,
      "step": 48764
    },
    {
      "epoch": 4.650497436523438e-05,
      "step": 48764,
      "training_step_time": 8.402285814285278
    },
    {
      "epoch": 4.650592803955078e-05,
      "model_forward_time": 2.8356523513793945,
      "step": 48765
    },
    {
      "epoch": 4.650592803955078e-05,
      "step": 48765,
      "training_step_time": 8.675308227539062
    },
    {
      "epoch": 4.6506881713867186e-05,
      "model_forward_time": 1.8684818744659424,
      "step": 48766
    },
    {
      "epoch": 4.6506881713867186e-05,
      "step": 48766,
      "training_step_time": 7.672686576843262
    },
    {
      "epoch": 4.65078353881836e-05,
      "model_forward_time": 1.751244306564331,
      "step": 48767
    },
    {
      "epoch": 4.65078353881836e-05,
      "step": 48767,
      "training_step_time": 7.5801615715026855
    },
    {
      "epoch": 4.65087890625e-05,
      "model_forward_time": 1.7495133876800537,
      "step": 48768
    },
    {
      "epoch": 4.65087890625e-05,
      "step": 48768,
      "training_step_time": 7.603545904159546
    },
    {
      "epoch": 4.6509742736816405e-05,
      "model_forward_time": 1.7562756538391113,
      "step": 48769
    },
    {
      "epoch": 4.6509742736816405e-05,
      "step": 48769,
      "training_step_time": 7.562107801437378
    },
    {
      "epoch": 4.6510696411132815e-05,
      "grad_norm": 0.10500955581665039,
      "learning_rate": 5.615345567428822e-06,
      "loss": 0.0559,
      "step": 48770
    },
    {
      "epoch": 4.6510696411132815e-05,
      "model_forward_time": 1.738431453704834,
      "step": 48770
    },
    {
      "dynamics_loss_avg": 0.06310414914041758,
      "epoch": 4.6510696411132815e-05,
      "step": 48770
    },
    {
      "action_loss_avg": 0.002014008688274771,
      "epoch": 4.6510696411132815e-05,
      "step": 48770
    },
    {
      "epoch": 4.6510696411132815e-05,
      "step": 48770,
      "training_step_time": 7.584663391113281
    },
    {
      "epoch": 4.651165008544922e-05,
      "model_forward_time": 1.7428696155548096,
      "step": 48771
    },
    {
      "epoch": 4.651165008544922e-05,
      "step": 48771,
      "training_step_time": 7.471934795379639
    },
    {
      "epoch": 4.6512603759765624e-05,
      "model_forward_time": 1.7387440204620361,
      "step": 48772
    },
    {
      "epoch": 4.6512603759765624e-05,
      "step": 48772,
      "training_step_time": 8.290931940078735
    },
    {
      "epoch": 4.6513557434082034e-05,
      "model_forward_time": 1.7462797164916992,
      "step": 48773
    },
    {
      "epoch": 4.6513557434082034e-05,
      "step": 48773,
      "training_step_time": 7.491084098815918
    },
    {
      "epoch": 4.651451110839844e-05,
      "model_forward_time": 1.7451276779174805,
      "step": 48774
    },
    {
      "epoch": 4.651451110839844e-05,
      "step": 48774,
      "training_step_time": 7.710670471191406
    },
    {
      "epoch": 4.651546478271484e-05,
      "model_forward_time": 2.8565616607666016,
      "step": 48775
    },
    {
      "epoch": 4.651546478271484e-05,
      "step": 48775,
      "training_step_time": 7.685664653778076
    },
    {
      "epoch": 4.651641845703125e-05,
      "model_forward_time": 1.7376811504364014,
      "step": 48776
    },
    {
      "epoch": 4.651641845703125e-05,
      "step": 48776,
      "training_step_time": 7.673562526702881
    },
    {
      "epoch": 4.651737213134766e-05,
      "model_forward_time": 1.748871088027954,
      "step": 48777
    },
    {
      "epoch": 4.651737213134766e-05,
      "step": 48777,
      "training_step_time": 8.625798463821411
    },
    {
      "epoch": 4.651832580566406e-05,
      "model_forward_time": 1.7460057735443115,
      "step": 48778
    },
    {
      "epoch": 4.651832580566406e-05,
      "step": 48778,
      "training_step_time": 7.576623439788818
    },
    {
      "epoch": 4.651927947998047e-05,
      "model_forward_time": 1.746697187423706,
      "step": 48779
    },
    {
      "epoch": 4.651927947998047e-05,
      "step": 48779,
      "training_step_time": 7.67330265045166
    },
    {
      "epoch": 4.6520233154296876e-05,
      "grad_norm": 0.14986686408519745,
      "learning_rate": 5.613704633502941e-06,
      "loss": 0.0551,
      "step": 48780
    },
    {
      "epoch": 4.6520233154296876e-05,
      "model_forward_time": 1.7376632690429688,
      "step": 48780
    },
    {
      "dynamics_loss_avg": 0.05002124886959791,
      "epoch": 4.6520233154296876e-05,
      "step": 48780
    },
    {
      "action_loss_avg": 0.0014223489444702864,
      "epoch": 4.6520233154296876e-05,
      "step": 48780
    },
    {
      "epoch": 4.6520233154296876e-05,
      "step": 48780,
      "training_step_time": 7.531679153442383
    },
    {
      "epoch": 4.652118682861328e-05,
      "model_forward_time": 1.7474751472473145,
      "step": 48781
    },
    {
      "epoch": 4.652118682861328e-05,
      "step": 48781,
      "training_step_time": 7.683602571487427
    },
    {
      "epoch": 4.6522140502929684e-05,
      "model_forward_time": 1.7469494342803955,
      "step": 48782
    },
    {
      "epoch": 4.6522140502929684e-05,
      "step": 48782,
      "training_step_time": 7.550193786621094
    },
    {
      "epoch": 4.6523094177246095e-05,
      "model_forward_time": 1.7511694431304932,
      "step": 48783
    },
    {
      "epoch": 4.6523094177246095e-05,
      "step": 48783,
      "training_step_time": 7.778847932815552
    },
    {
      "epoch": 4.65240478515625e-05,
      "model_forward_time": 2.8861396312713623,
      "step": 48784
    },
    {
      "epoch": 4.65240478515625e-05,
      "step": 48784,
      "training_step_time": 7.751538038253784
    },
    {
      "epoch": 4.65250015258789e-05,
      "model_forward_time": 1.7406730651855469,
      "step": 48785
    },
    {
      "epoch": 4.65250015258789e-05,
      "step": 48785,
      "training_step_time": 8.468915700912476
    },
    {
      "epoch": 4.6525955200195314e-05,
      "model_forward_time": 1.7501776218414307,
      "step": 48786
    },
    {
      "epoch": 4.6525955200195314e-05,
      "step": 48786,
      "training_step_time": 8.28679633140564
    },
    {
      "epoch": 4.652690887451172e-05,
      "model_forward_time": 1.7519633769989014,
      "step": 48787
    },
    {
      "epoch": 4.652690887451172e-05,
      "step": 48787,
      "training_step_time": 7.594393730163574
    },
    {
      "epoch": 4.652786254882812e-05,
      "model_forward_time": 1.7519569396972656,
      "step": 48788
    },
    {
      "epoch": 4.652786254882812e-05,
      "step": 48788,
      "training_step_time": 7.611728668212891
    },
    {
      "epoch": 4.652881622314453e-05,
      "model_forward_time": 1.7517998218536377,
      "step": 48789
    },
    {
      "epoch": 4.652881622314453e-05,
      "step": 48789,
      "training_step_time": 8.400954008102417
    },
    {
      "epoch": 4.652976989746094e-05,
      "grad_norm": 0.1251574009656906,
      "learning_rate": 5.612063632463245e-06,
      "loss": 0.0573,
      "step": 48790
    },
    {
      "epoch": 4.652976989746094e-05,
      "model_forward_time": 1.7472620010375977,
      "step": 48790
    },
    {
      "dynamics_loss_avg": 0.053993231244385244,
      "epoch": 4.652976989746094e-05,
      "step": 48790
    },
    {
      "action_loss_avg": 0.0013850752206053586,
      "epoch": 4.652976989746094e-05,
      "step": 48790
    },
    {
      "epoch": 4.652976989746094e-05,
      "step": 48790,
      "training_step_time": 7.588657379150391
    },
    {
      "epoch": 4.653072357177734e-05,
      "model_forward_time": 1.7475037574768066,
      "step": 48791
    },
    {
      "epoch": 4.653072357177734e-05,
      "step": 48791,
      "training_step_time": 7.592449903488159
    },
    {
      "epoch": 4.653167724609375e-05,
      "model_forward_time": 1.751042127609253,
      "step": 48792
    },
    {
      "epoch": 4.653167724609375e-05,
      "step": 48792,
      "training_step_time": 8.356173515319824
    },
    {
      "epoch": 4.6532630920410156e-05,
      "model_forward_time": 2.880458116531372,
      "step": 48793
    },
    {
      "epoch": 4.6532630920410156e-05,
      "step": 48793,
      "training_step_time": 10.188339233398438
    },
    {
      "epoch": 4.653358459472656e-05,
      "model_forward_time": 1.7413756847381592,
      "step": 48794
    },
    {
      "epoch": 4.653358459472656e-05,
      "step": 48794,
      "training_step_time": 8.500812292098999
    },
    {
      "epoch": 4.653453826904297e-05,
      "model_forward_time": 1.7559685707092285,
      "step": 48795
    },
    {
      "epoch": 4.653453826904297e-05,
      "step": 48795,
      "training_step_time": 9.448228359222412
    },
    {
      "epoch": 4.6535491943359374e-05,
      "model_forward_time": 1.7593021392822266,
      "step": 48796
    },
    {
      "epoch": 4.6535491943359374e-05,
      "step": 48796,
      "training_step_time": 9.399148941040039
    },
    {
      "epoch": 4.653644561767578e-05,
      "model_forward_time": 1.3906052112579346,
      "step": 48797
    },
    {
      "epoch": 4.653644561767578e-05,
      "step": 48797,
      "training_step_time": 7.740550994873047
    },
    {
      "epoch": 4.653739929199219e-05,
      "model_forward_time": 1.748178482055664,
      "step": 48798
    },
    {
      "epoch": 4.653739929199219e-05,
      "step": 48798,
      "training_step_time": 7.673725843429565
    },
    {
      "epoch": 4.653835296630859e-05,
      "model_forward_time": 1.75994873046875,
      "step": 48799
    },
    {
      "epoch": 4.653835296630859e-05,
      "step": 48799,
      "training_step_time": 7.558861255645752
    },
    {
      "epoch": 4.6539306640625e-05,
      "grad_norm": 0.1582919955253601,
      "learning_rate": 5.6104225644891884e-06,
      "loss": 0.0575,
      "step": 48800
    },
    {
      "epoch": 4.6539306640625e-05,
      "model_forward_time": 1.740732192993164,
      "step": 48800
    },
    {
      "dynamics_loss_avg": 0.0453519631177187,
      "epoch": 4.6539306640625e-05,
      "step": 48800
    },
    {
      "action_loss_avg": 0.0027698310615960507,
      "epoch": 4.6539306640625e-05,
      "step": 48800
    },
    {
      "epoch": 4.6539306640625e-05,
      "step": 48800,
      "training_step_time": 7.658206462860107
    },
    {
      "epoch": 4.654026031494141e-05,
      "model_forward_time": 1.7404980659484863,
      "step": 48801
    },
    {
      "epoch": 4.654026031494141e-05,
      "step": 48801,
      "training_step_time": 8.944462299346924
    },
    {
      "epoch": 4.654121398925781e-05,
      "model_forward_time": 2.8728554248809814,
      "step": 48802
    },
    {
      "epoch": 4.654121398925781e-05,
      "step": 48802,
      "training_step_time": 9.761245727539062
    },
    {
      "epoch": 4.6542167663574216e-05,
      "model_forward_time": 1.7437520027160645,
      "step": 48803
    },
    {
      "epoch": 4.6542167663574216e-05,
      "step": 48803,
      "training_step_time": 9.34907341003418
    },
    {
      "epoch": 4.654312133789063e-05,
      "model_forward_time": 1.7552740573883057,
      "step": 48804
    },
    {
      "epoch": 4.654312133789063e-05,
      "step": 48804,
      "training_step_time": 7.796136140823364
    },
    {
      "epoch": 4.654407501220703e-05,
      "model_forward_time": 1.3938543796539307,
      "step": 48805
    },
    {
      "epoch": 4.654407501220703e-05,
      "step": 48805,
      "training_step_time": 7.796941757202148
    },
    {
      "epoch": 4.6545028686523435e-05,
      "model_forward_time": 1.7378172874450684,
      "step": 48806
    },
    {
      "epoch": 4.6545028686523435e-05,
      "step": 48806,
      "training_step_time": 7.752464294433594
    },
    {
      "epoch": 4.6545982360839846e-05,
      "model_forward_time": 1.7382240295410156,
      "step": 48807
    },
    {
      "epoch": 4.6545982360839846e-05,
      "step": 48807,
      "training_step_time": 7.729807376861572
    },
    {
      "epoch": 4.654693603515625e-05,
      "model_forward_time": 1.7553746700286865,
      "step": 48808
    },
    {
      "epoch": 4.654693603515625e-05,
      "step": 48808,
      "training_step_time": 8.565099239349365
    },
    {
      "epoch": 4.6547889709472654e-05,
      "model_forward_time": 1.7384395599365234,
      "step": 48809
    },
    {
      "epoch": 4.6547889709472654e-05,
      "step": 48809,
      "training_step_time": 7.648159503936768
    },
    {
      "epoch": 4.6548843383789065e-05,
      "grad_norm": 0.10426925867795944,
      "learning_rate": 5.608781429760239e-06,
      "loss": 0.0594,
      "step": 48810
    },
    {
      "epoch": 4.6548843383789065e-05,
      "model_forward_time": 1.747558355331421,
      "step": 48810
    },
    {
      "dynamics_loss_avg": 0.04348755553364754,
      "epoch": 4.6548843383789065e-05,
      "step": 48810
    },
    {
      "action_loss_avg": 0.0018199298006948083,
      "epoch": 4.6548843383789065e-05,
      "step": 48810
    },
    {
      "epoch": 4.6548843383789065e-05,
      "step": 48810,
      "training_step_time": 8.648746490478516
    },
    {
      "epoch": 4.654979705810547e-05,
      "model_forward_time": 1.7671656608581543,
      "step": 48811
    },
    {
      "epoch": 4.654979705810547e-05,
      "step": 48811,
      "training_step_time": 9.422895193099976
    },
    {
      "epoch": 4.655075073242187e-05,
      "model_forward_time": 2.8999924659729004,
      "step": 48812
    },
    {
      "epoch": 4.655075073242187e-05,
      "step": 48812,
      "training_step_time": 8.506662368774414
    },
    {
      "epoch": 4.6551704406738284e-05,
      "model_forward_time": 1.7607331275939941,
      "step": 48813
    },
    {
      "epoch": 4.6551704406738284e-05,
      "step": 48813,
      "training_step_time": 7.649572134017944
    },
    {
      "epoch": 4.655265808105469e-05,
      "model_forward_time": 1.7499430179595947,
      "step": 48814
    },
    {
      "epoch": 4.655265808105469e-05,
      "step": 48814,
      "training_step_time": 8.643800973892212
    },
    {
      "epoch": 4.655361175537109e-05,
      "model_forward_time": 1.7421019077301025,
      "step": 48815
    },
    {
      "epoch": 4.655361175537109e-05,
      "step": 48815,
      "training_step_time": 7.594624280929565
    },
    {
      "epoch": 4.65545654296875e-05,
      "model_forward_time": 1.7609033584594727,
      "step": 48816
    },
    {
      "epoch": 4.65545654296875e-05,
      "step": 48816,
      "training_step_time": 7.731092214584351
    },
    {
      "epoch": 4.6555519104003906e-05,
      "model_forward_time": 1.7341957092285156,
      "step": 48817
    },
    {
      "epoch": 4.6555519104003906e-05,
      "step": 48817,
      "training_step_time": 7.379178285598755
    },
    {
      "epoch": 4.655647277832031e-05,
      "model_forward_time": 1.7510497570037842,
      "step": 48818
    },
    {
      "epoch": 4.655647277832031e-05,
      "step": 48818,
      "training_step_time": 7.717841863632202
    },
    {
      "epoch": 4.655742645263672e-05,
      "model_forward_time": 1.7559428215026855,
      "step": 48819
    },
    {
      "epoch": 4.655742645263672e-05,
      "step": 48819,
      "training_step_time": 8.43034815788269
    },
    {
      "epoch": 4.6558380126953125e-05,
      "grad_norm": 0.133926123380661,
      "learning_rate": 5.607140228455866e-06,
      "loss": 0.057,
      "step": 48820
    },
    {
      "epoch": 4.6558380126953125e-05,
      "model_forward_time": 1.7499375343322754,
      "step": 48820
    },
    {
      "dynamics_loss_avg": 0.05670727416872978,
      "epoch": 4.6558380126953125e-05,
      "step": 48820
    },
    {
      "action_loss_avg": 0.002505150323850103,
      "epoch": 4.6558380126953125e-05,
      "step": 48820
    },
    {
      "epoch": 4.6558380126953125e-05,
      "step": 48820,
      "training_step_time": 8.71571660041809
    },
    {
      "epoch": 4.655933380126953e-05,
      "model_forward_time": 2.838902235031128,
      "step": 48821
    },
    {
      "epoch": 4.655933380126953e-05,
      "step": 48821,
      "training_step_time": 8.369261026382446
    },
    {
      "epoch": 4.656028747558594e-05,
      "model_forward_time": 1.7193641662597656,
      "step": 48822
    },
    {
      "epoch": 4.656028747558594e-05,
      "step": 48822,
      "training_step_time": 7.726215124130249
    },
    {
      "epoch": 4.6561241149902344e-05,
      "model_forward_time": 1.726550817489624,
      "step": 48823
    },
    {
      "epoch": 4.6561241149902344e-05,
      "step": 48823,
      "training_step_time": 8.876261234283447
    },
    {
      "epoch": 4.656219482421875e-05,
      "model_forward_time": 1.7212183475494385,
      "step": 48824
    },
    {
      "epoch": 4.656219482421875e-05,
      "step": 48824,
      "training_step_time": 8.69158124923706
    },
    {
      "epoch": 4.656314849853516e-05,
      "model_forward_time": 1.7195219993591309,
      "step": 48825
    },
    {
      "epoch": 4.656314849853516e-05,
      "step": 48825,
      "training_step_time": 7.752964973449707
    },
    {
      "epoch": 4.656410217285156e-05,
      "model_forward_time": 1.366405725479126,
      "step": 48826
    },
    {
      "epoch": 4.656410217285156e-05,
      "step": 48826,
      "training_step_time": 7.772181510925293
    },
    {
      "epoch": 4.656505584716797e-05,
      "model_forward_time": 1.3599514961242676,
      "step": 48827
    },
    {
      "epoch": 4.656505584716797e-05,
      "step": 48827,
      "training_step_time": 7.847527980804443
    },
    {
      "epoch": 4.656600952148438e-05,
      "model_forward_time": 1.7241437435150146,
      "step": 48828
    },
    {
      "epoch": 4.656600952148438e-05,
      "step": 48828,
      "training_step_time": 7.656519889831543
    },
    {
      "epoch": 4.656696319580078e-05,
      "model_forward_time": 1.7180311679840088,
      "step": 48829
    },
    {
      "epoch": 4.656696319580078e-05,
      "step": 48829,
      "training_step_time": 9.072311401367188
    },
    {
      "epoch": 4.6567916870117186e-05,
      "grad_norm": 0.15873515605926514,
      "learning_rate": 5.605498960755553e-06,
      "loss": 0.0585,
      "step": 48830
    },
    {
      "epoch": 4.6567916870117186e-05,
      "model_forward_time": 2.8602731227874756,
      "step": 48830
    },
    {
      "dynamics_loss_avg": 0.06618930827826261,
      "epoch": 4.6567916870117186e-05,
      "step": 48830
    },
    {
      "action_loss_avg": 0.0038495556102134286,
      "epoch": 4.6567916870117186e-05,
      "step": 48830
    },
    {
      "epoch": 4.6567916870117186e-05,
      "step": 48830,
      "training_step_time": 8.377695083618164
    },
    {
      "epoch": 4.65688705444336e-05,
      "model_forward_time": 1.7227027416229248,
      "step": 48831
    },
    {
      "epoch": 4.65688705444336e-05,
      "step": 48831,
      "training_step_time": 9.259476900100708
    },
    {
      "epoch": 4.656982421875e-05,
      "model_forward_time": 1.7231414318084717,
      "step": 48832
    },
    {
      "epoch": 4.656982421875e-05,
      "step": 48832,
      "training_step_time": 8.56929636001587
    },
    {
      "epoch": 4.6570777893066405e-05,
      "model_forward_time": 1.7227959632873535,
      "step": 48833
    },
    {
      "epoch": 4.6570777893066405e-05,
      "step": 48833,
      "training_step_time": 7.913610458374023
    },
    {
      "epoch": 4.6571731567382816e-05,
      "model_forward_time": 1.7171576023101807,
      "step": 48834
    },
    {
      "epoch": 4.6571731567382816e-05,
      "step": 48834,
      "training_step_time": 7.82142972946167
    },
    {
      "epoch": 4.657268524169922e-05,
      "model_forward_time": 1.7200307846069336,
      "step": 48835
    },
    {
      "epoch": 4.657268524169922e-05,
      "step": 48835,
      "training_step_time": 7.795626878738403
    },
    {
      "epoch": 4.6573638916015624e-05,
      "model_forward_time": 1.7279064655303955,
      "step": 48836
    },
    {
      "epoch": 4.6573638916015624e-05,
      "step": 48836,
      "training_step_time": 8.577779293060303
    },
    {
      "epoch": 4.6574592590332034e-05,
      "model_forward_time": 1.7384486198425293,
      "step": 48837
    },
    {
      "epoch": 4.6574592590332034e-05,
      "step": 48837,
      "training_step_time": 7.716322660446167
    },
    {
      "epoch": 4.657554626464844e-05,
      "model_forward_time": 1.7265098094940186,
      "step": 48838
    },
    {
      "epoch": 4.657554626464844e-05,
      "step": 48838,
      "training_step_time": 8.810226917266846
    },
    {
      "epoch": 4.657649993896484e-05,
      "model_forward_time": 2.8757340908050537,
      "step": 48839
    },
    {
      "epoch": 4.657649993896484e-05,
      "step": 48839,
      "training_step_time": 8.540770053863525
    },
    {
      "epoch": 4.657745361328125e-05,
      "grad_norm": 0.13847041130065918,
      "learning_rate": 5.603857626838782e-06,
      "loss": 0.0589,
      "step": 48840
    },
    {
      "epoch": 4.657745361328125e-05,
      "model_forward_time": 1.7133994102478027,
      "step": 48840
    },
    {
      "dynamics_loss_avg": 0.04369408376514912,
      "epoch": 4.657745361328125e-05,
      "step": 48840
    },
    {
      "action_loss_avg": 0.0013918324693804607,
      "epoch": 4.657745361328125e-05,
      "step": 48840
    },
    {
      "epoch": 4.657745361328125e-05,
      "step": 48840,
      "training_step_time": 8.563154935836792
    },
    {
      "epoch": 4.657840728759766e-05,
      "model_forward_time": 1.7132508754730225,
      "step": 48841
    },
    {
      "epoch": 4.657840728759766e-05,
      "step": 48841,
      "training_step_time": 7.789590358734131
    },
    {
      "epoch": 4.657936096191406e-05,
      "model_forward_time": 1.7380211353302002,
      "step": 48842
    },
    {
      "epoch": 4.657936096191406e-05,
      "step": 48842,
      "training_step_time": 7.815651893615723
    },
    {
      "epoch": 4.658031463623047e-05,
      "model_forward_time": 1.7165277004241943,
      "step": 48843
    },
    {
      "epoch": 4.658031463623047e-05,
      "step": 48843,
      "training_step_time": 7.67302131652832
    },
    {
      "epoch": 4.6581268310546876e-05,
      "model_forward_time": 1.7234256267547607,
      "step": 48844
    },
    {
      "epoch": 4.6581268310546876e-05,
      "step": 48844,
      "training_step_time": 7.8142008781433105
    },
    {
      "epoch": 4.658222198486328e-05,
      "model_forward_time": 1.7181556224822998,
      "step": 48845
    },
    {
      "epoch": 4.658222198486328e-05,
      "step": 48845,
      "training_step_time": 7.975653886795044
    },
    {
      "epoch": 4.6583175659179684e-05,
      "model_forward_time": 1.7205414772033691,
      "step": 48846
    },
    {
      "epoch": 4.6583175659179684e-05,
      "step": 48846,
      "training_step_time": 8.123670816421509
    },
    {
      "epoch": 4.6584129333496095e-05,
      "model_forward_time": 1.7275586128234863,
      "step": 48847
    },
    {
      "epoch": 4.6584129333496095e-05,
      "step": 48847,
      "training_step_time": 7.9615888595581055
    },
    {
      "epoch": 4.65850830078125e-05,
      "model_forward_time": 1.7547950744628906,
      "step": 48848
    },
    {
      "epoch": 4.65850830078125e-05,
      "step": 48848,
      "training_step_time": 8.94258427619934
    },
    {
      "epoch": 4.65860366821289e-05,
      "model_forward_time": 2.8457484245300293,
      "step": 48849
    },
    {
      "epoch": 4.65860366821289e-05,
      "step": 48849,
      "training_step_time": 7.806229829788208
    },
    {
      "epoch": 4.6586990356445314e-05,
      "grad_norm": 0.1400703489780426,
      "learning_rate": 5.60221622688505e-06,
      "loss": 0.0587,
      "step": 48850
    },
    {
      "epoch": 4.6586990356445314e-05,
      "model_forward_time": 1.7282383441925049,
      "step": 48850
    },
    {
      "dynamics_loss_avg": 0.05855163857340813,
      "epoch": 4.6586990356445314e-05,
      "step": 48850
    },
    {
      "action_loss_avg": 0.0024744445574469863,
      "epoch": 4.6586990356445314e-05,
      "step": 48850
    },
    {
      "epoch": 4.6586990356445314e-05,
      "step": 48850,
      "training_step_time": 7.792832374572754
    },
    {
      "epoch": 4.658794403076172e-05,
      "model_forward_time": 1.7254090309143066,
      "step": 48851
    },
    {
      "epoch": 4.658794403076172e-05,
      "step": 48851,
      "training_step_time": 7.819857835769653
    },
    {
      "epoch": 4.658889770507812e-05,
      "model_forward_time": 1.7057979106903076,
      "step": 48852
    },
    {
      "epoch": 4.658889770507812e-05,
      "step": 48852,
      "training_step_time": 7.739635467529297
    },
    {
      "epoch": 4.658985137939453e-05,
      "model_forward_time": 1.7170305252075195,
      "step": 48853
    },
    {
      "epoch": 4.658985137939453e-05,
      "step": 48853,
      "training_step_time": 7.600115776062012
    },
    {
      "epoch": 4.659080505371094e-05,
      "model_forward_time": 1.731943130493164,
      "step": 48854
    },
    {
      "epoch": 4.659080505371094e-05,
      "step": 48854,
      "training_step_time": 7.457279682159424
    },
    {
      "epoch": 4.659175872802734e-05,
      "model_forward_time": 1.3619980812072754,
      "step": 48855
    },
    {
      "epoch": 4.659175872802734e-05,
      "step": 48855,
      "training_step_time": 8.745827674865723
    },
    {
      "epoch": 4.659271240234375e-05,
      "model_forward_time": 1.7158825397491455,
      "step": 48856
    },
    {
      "epoch": 4.659271240234375e-05,
      "step": 48856,
      "training_step_time": 8.075564861297607
    },
    {
      "epoch": 4.6593666076660156e-05,
      "model_forward_time": 1.7266123294830322,
      "step": 48857
    },
    {
      "epoch": 4.6593666076660156e-05,
      "step": 48857,
      "training_step_time": 8.797302961349487
    },
    {
      "epoch": 4.659461975097656e-05,
      "model_forward_time": 2.850374221801758,
      "step": 48858
    },
    {
      "epoch": 4.659461975097656e-05,
      "step": 48858,
      "training_step_time": 7.907110691070557
    },
    {
      "epoch": 4.659557342529297e-05,
      "model_forward_time": 1.7277607917785645,
      "step": 48859
    },
    {
      "epoch": 4.659557342529297e-05,
      "step": 48859,
      "training_step_time": 8.593428611755371
    },
    {
      "epoch": 4.6596527099609374e-05,
      "grad_norm": 0.128477081656456,
      "learning_rate": 5.600574761073857e-06,
      "loss": 0.055,
      "step": 48860
    },
    {
      "epoch": 4.6596527099609374e-05,
      "model_forward_time": 1.7215495109558105,
      "step": 48860
    },
    {
      "dynamics_loss_avg": 0.051790892332792285,
      "epoch": 4.6596527099609374e-05,
      "step": 48860
    },
    {
      "action_loss_avg": 0.0018034716951660812,
      "epoch": 4.6596527099609374e-05,
      "step": 48860
    },
    {
      "epoch": 4.6596527099609374e-05,
      "step": 48860,
      "training_step_time": 7.879234075546265
    },
    {
      "epoch": 4.659748077392578e-05,
      "model_forward_time": 1.72263765335083,
      "step": 48861
    },
    {
      "epoch": 4.659748077392578e-05,
      "step": 48861,
      "training_step_time": 7.869229078292847
    },
    {
      "epoch": 4.659843444824219e-05,
      "model_forward_time": 1.736048698425293,
      "step": 48862
    },
    {
      "epoch": 4.659843444824219e-05,
      "step": 48862,
      "training_step_time": 7.78942346572876
    },
    {
      "epoch": 4.659938812255859e-05,
      "model_forward_time": 1.7229056358337402,
      "step": 48863
    },
    {
      "epoch": 4.659938812255859e-05,
      "step": 48863,
      "training_step_time": 7.467926025390625
    },
    {
      "epoch": 4.6600341796875e-05,
      "model_forward_time": 1.7355642318725586,
      "step": 48864
    },
    {
      "epoch": 4.6600341796875e-05,
      "step": 48864,
      "training_step_time": 8.64694595336914
    },
    {
      "epoch": 4.660129547119141e-05,
      "model_forward_time": 1.734435796737671,
      "step": 48865
    },
    {
      "epoch": 4.660129547119141e-05,
      "step": 48865,
      "training_step_time": 8.966066360473633
    },
    {
      "epoch": 4.660224914550781e-05,
      "model_forward_time": 1.7357454299926758,
      "step": 48866
    },
    {
      "epoch": 4.660224914550781e-05,
      "step": 48866,
      "training_step_time": 8.80479621887207
    },
    {
      "epoch": 4.6603202819824216e-05,
      "model_forward_time": 2.918921947479248,
      "step": 48867
    },
    {
      "epoch": 4.6603202819824216e-05,
      "step": 48867,
      "training_step_time": 8.341338872909546
    },
    {
      "epoch": 4.660415649414063e-05,
      "model_forward_time": 1.8872125148773193,
      "step": 48868
    },
    {
      "epoch": 4.660415649414063e-05,
      "step": 48868,
      "training_step_time": 7.842235803604126
    },
    {
      "epoch": 4.660511016845703e-05,
      "model_forward_time": 1.8726134300231934,
      "step": 48869
    },
    {
      "epoch": 4.660511016845703e-05,
      "step": 48869,
      "training_step_time": 7.739078998565674
    },
    {
      "epoch": 4.6606063842773435e-05,
      "grad_norm": 0.12818847596645355,
      "learning_rate": 5.59893322958471e-06,
      "loss": 0.0564,
      "step": 48870
    },
    {
      "epoch": 4.6606063842773435e-05,
      "model_forward_time": 1.8643059730529785,
      "step": 48870
    },
    {
      "dynamics_loss_avg": 0.048828852083534005,
      "epoch": 4.6606063842773435e-05,
      "step": 48870
    },
    {
      "action_loss_avg": 0.0011613455426413566,
      "epoch": 4.6606063842773435e-05,
      "step": 48870
    },
    {
      "epoch": 4.6606063842773435e-05,
      "step": 48870,
      "training_step_time": 8.701191902160645
    },
    {
      "epoch": 4.6607017517089846e-05,
      "model_forward_time": 1.8679790496826172,
      "step": 48871
    },
    {
      "epoch": 4.6607017517089846e-05,
      "step": 48871,
      "training_step_time": 7.610645055770874
    },
    {
      "epoch": 4.660797119140625e-05,
      "model_forward_time": 1.7381081581115723,
      "step": 48872
    },
    {
      "epoch": 4.660797119140625e-05,
      "step": 48872,
      "training_step_time": 7.61467981338501
    },
    {
      "epoch": 4.6608924865722654e-05,
      "model_forward_time": 1.7184233665466309,
      "step": 48873
    },
    {
      "epoch": 4.6608924865722654e-05,
      "step": 48873,
      "training_step_time": 7.596197843551636
    },
    {
      "epoch": 4.6609878540039065e-05,
      "model_forward_time": 1.723222255706787,
      "step": 48874
    },
    {
      "epoch": 4.6609878540039065e-05,
      "step": 48874,
      "training_step_time": 8.559234857559204
    },
    {
      "epoch": 4.661083221435547e-05,
      "model_forward_time": 1.7286596298217773,
      "step": 48875
    },
    {
      "epoch": 4.661083221435547e-05,
      "step": 48875,
      "training_step_time": 7.88983416557312
    },
    {
      "epoch": 4.661178588867187e-05,
      "model_forward_time": 2.92268967628479,
      "step": 48876
    },
    {
      "epoch": 4.661178588867187e-05,
      "step": 48876,
      "training_step_time": 7.8406946659088135
    },
    {
      "epoch": 4.6612739562988284e-05,
      "model_forward_time": 1.724376916885376,
      "step": 48877
    },
    {
      "epoch": 4.6612739562988284e-05,
      "step": 48877,
      "training_step_time": 7.7756664752960205
    },
    {
      "epoch": 4.661369323730469e-05,
      "model_forward_time": 1.7189781665802002,
      "step": 48878
    },
    {
      "epoch": 4.661369323730469e-05,
      "step": 48878,
      "training_step_time": 9.528520584106445
    },
    {
      "epoch": 4.661464691162109e-05,
      "model_forward_time": 1.722855567932129,
      "step": 48879
    },
    {
      "epoch": 4.661464691162109e-05,
      "step": 48879,
      "training_step_time": 8.411308288574219
    },
    {
      "epoch": 4.66156005859375e-05,
      "grad_norm": 0.08610551059246063,
      "learning_rate": 5.597291632597126e-06,
      "loss": 0.0582,
      "step": 48880
    },
    {
      "epoch": 4.66156005859375e-05,
      "model_forward_time": 1.7176477909088135,
      "step": 48880
    },
    {
      "dynamics_loss_avg": 0.06150101087987423,
      "epoch": 4.66156005859375e-05,
      "step": 48880
    },
    {
      "action_loss_avg": 0.0011678475886583329,
      "epoch": 4.66156005859375e-05,
      "step": 48880
    },
    {
      "epoch": 4.66156005859375e-05,
      "step": 48880,
      "training_step_time": 7.8417627811431885
    },
    {
      "epoch": 4.6616554260253906e-05,
      "model_forward_time": 1.7203547954559326,
      "step": 48881
    },
    {
      "epoch": 4.6616554260253906e-05,
      "step": 48881,
      "training_step_time": 7.460428237915039
    },
    {
      "epoch": 4.661750793457031e-05,
      "model_forward_time": 1.722710371017456,
      "step": 48882
    },
    {
      "epoch": 4.661750793457031e-05,
      "step": 48882,
      "training_step_time": 7.678844451904297
    },
    {
      "epoch": 4.661846160888672e-05,
      "model_forward_time": 1.7196216583251953,
      "step": 48883
    },
    {
      "epoch": 4.661846160888672e-05,
      "step": 48883,
      "training_step_time": 7.732821941375732
    },
    {
      "epoch": 4.6619415283203125e-05,
      "model_forward_time": 1.7218399047851562,
      "step": 48884
    },
    {
      "epoch": 4.6619415283203125e-05,
      "step": 48884,
      "training_step_time": 8.57030701637268
    },
    {
      "epoch": 4.662036895751953e-05,
      "model_forward_time": 1.8478012084960938,
      "step": 48885
    },
    {
      "epoch": 4.662036895751953e-05,
      "step": 48885,
      "training_step_time": 9.451436519622803
    },
    {
      "epoch": 4.662132263183594e-05,
      "model_forward_time": 1.834364414215088,
      "step": 48886
    },
    {
      "epoch": 4.662132263183594e-05,
      "step": 48886,
      "training_step_time": 7.846737861633301
    },
    {
      "epoch": 4.6622276306152344e-05,
      "model_forward_time": 1.7219657897949219,
      "step": 48887
    },
    {
      "epoch": 4.6622276306152344e-05,
      "step": 48887,
      "training_step_time": 9.914506912231445
    },
    {
      "epoch": 4.662322998046875e-05,
      "model_forward_time": 1.7211644649505615,
      "step": 48888
    },
    {
      "epoch": 4.662322998046875e-05,
      "step": 48888,
      "training_step_time": 7.834853410720825
    },
    {
      "epoch": 4.662418365478516e-05,
      "model_forward_time": 1.7201833724975586,
      "step": 48889
    },
    {
      "epoch": 4.662418365478516e-05,
      "step": 48889,
      "training_step_time": 7.859743595123291
    },
    {
      "epoch": 4.662513732910156e-05,
      "grad_norm": 0.13865578174591064,
      "learning_rate": 5.595649970290629e-06,
      "loss": 0.0563,
      "step": 48890
    },
    {
      "epoch": 4.662513732910156e-05,
      "model_forward_time": 1.7191083431243896,
      "step": 48890
    },
    {
      "dynamics_loss_avg": 0.05040975660085678,
      "epoch": 4.662513732910156e-05,
      "step": 48890
    },
    {
      "action_loss_avg": 0.001451313038705848,
      "epoch": 4.662513732910156e-05,
      "step": 48890
    },
    {
      "epoch": 4.662513732910156e-05,
      "step": 48890,
      "training_step_time": 7.278848648071289
    },
    {
      "epoch": 4.662609100341797e-05,
      "model_forward_time": 1.7164812088012695,
      "step": 48891
    },
    {
      "epoch": 4.662609100341797e-05,
      "step": 48891,
      "training_step_time": 8.291495561599731
    },
    {
      "epoch": 4.662704467773438e-05,
      "model_forward_time": 1.7374794483184814,
      "step": 48892
    },
    {
      "epoch": 4.662704467773438e-05,
      "step": 48892,
      "training_step_time": 8.32764983177185
    },
    {
      "epoch": 4.662799835205078e-05,
      "model_forward_time": 1.7243101596832275,
      "step": 48893
    },
    {
      "epoch": 4.662799835205078e-05,
      "step": 48893,
      "training_step_time": 8.649885654449463
    },
    {
      "epoch": 4.6628952026367186e-05,
      "model_forward_time": 1.8507518768310547,
      "step": 48894
    },
    {
      "epoch": 4.6628952026367186e-05,
      "step": 48894,
      "training_step_time": 8.502249717712402
    },
    {
      "epoch": 4.66299057006836e-05,
      "model_forward_time": 2.834703207015991,
      "step": 48895
    },
    {
      "epoch": 4.66299057006836e-05,
      "step": 48895,
      "training_step_time": 8.76983642578125
    },
    {
      "epoch": 4.6630859375e-05,
      "model_forward_time": 1.7199687957763672,
      "step": 48896
    },
    {
      "epoch": 4.6630859375e-05,
      "step": 48896,
      "training_step_time": 8.195196390151978
    },
    {
      "epoch": 4.6631813049316405e-05,
      "model_forward_time": 1.7328076362609863,
      "step": 48897
    },
    {
      "epoch": 4.6631813049316405e-05,
      "step": 48897,
      "training_step_time": 7.79594349861145
    },
    {
      "epoch": 4.6632766723632816e-05,
      "model_forward_time": 1.7178280353546143,
      "step": 48898
    },
    {
      "epoch": 4.6632766723632816e-05,
      "step": 48898,
      "training_step_time": 7.8271484375
    },
    {
      "epoch": 4.663372039794922e-05,
      "model_forward_time": 1.7272019386291504,
      "step": 48899
    },
    {
      "epoch": 4.663372039794922e-05,
      "step": 48899,
      "training_step_time": 7.2255637645721436
    },
    {
      "epoch": 4.6634674072265624e-05,
      "grad_norm": 0.12415813654661179,
      "learning_rate": 5.594008242844743e-06,
      "loss": 0.0598,
      "step": 48900
    },
    {
      "epoch": 4.6634674072265624e-05,
      "model_forward_time": 1.7153515815734863,
      "step": 48900
    },
    {
      "dynamics_loss_avg": 0.06218748725950718,
      "epoch": 4.6634674072265624e-05,
      "step": 48900
    },
    {
      "action_loss_avg": 0.0023301266075577585,
      "epoch": 4.6634674072265624e-05,
      "step": 48900
    },
    {
      "epoch": 4.6634674072265624e-05,
      "step": 48900,
      "training_step_time": 7.660490989685059
    },
    {
      "epoch": 4.6635627746582034e-05,
      "model_forward_time": 1.734513282775879,
      "step": 48901
    },
    {
      "epoch": 4.6635627746582034e-05,
      "step": 48901,
      "training_step_time": 7.659643888473511
    },
    {
      "epoch": 4.663658142089844e-05,
      "model_forward_time": 1.363703727722168,
      "step": 48902
    },
    {
      "epoch": 4.663658142089844e-05,
      "step": 48902,
      "training_step_time": 8.867161273956299
    },
    {
      "epoch": 4.663753509521484e-05,
      "model_forward_time": 1.7954661846160889,
      "step": 48903
    },
    {
      "epoch": 4.663753509521484e-05,
      "step": 48903,
      "training_step_time": 8.749769926071167
    },
    {
      "epoch": 4.663848876953125e-05,
      "model_forward_time": 2.903560161590576,
      "step": 48904
    },
    {
      "epoch": 4.663848876953125e-05,
      "step": 48904,
      "training_step_time": 7.832727909088135
    },
    {
      "epoch": 4.663944244384766e-05,
      "model_forward_time": 1.7326064109802246,
      "step": 48905
    },
    {
      "epoch": 4.663944244384766e-05,
      "step": 48905,
      "training_step_time": 7.82507586479187
    },
    {
      "epoch": 4.664039611816406e-05,
      "model_forward_time": 1.7363801002502441,
      "step": 48906
    },
    {
      "epoch": 4.664039611816406e-05,
      "step": 48906,
      "training_step_time": 8.297035217285156
    },
    {
      "epoch": 4.6641349792480465e-05,
      "model_forward_time": 1.7171716690063477,
      "step": 48907
    },
    {
      "epoch": 4.6641349792480465e-05,
      "step": 48907,
      "training_step_time": 7.7523581981658936
    },
    {
      "epoch": 4.6642303466796876e-05,
      "model_forward_time": 1.726804256439209,
      "step": 48908
    },
    {
      "epoch": 4.6642303466796876e-05,
      "step": 48908,
      "training_step_time": 7.664809226989746
    },
    {
      "epoch": 4.664325714111328e-05,
      "model_forward_time": 1.7650890350341797,
      "step": 48909
    },
    {
      "epoch": 4.664325714111328e-05,
      "step": 48909,
      "training_step_time": 7.441779613494873
    },
    {
      "epoch": 4.6644210815429684e-05,
      "grad_norm": 0.14306308329105377,
      "learning_rate": 5.592366450439012e-06,
      "loss": 0.0597,
      "step": 48910
    },
    {
      "epoch": 4.6644210815429684e-05,
      "model_forward_time": 1.7204625606536865,
      "step": 48910
    },
    {
      "dynamics_loss_avg": 0.07148579955101013,
      "epoch": 4.6644210815429684e-05,
      "step": 48910
    },
    {
      "action_loss_avg": 0.0019953839073423296,
      "epoch": 4.6644210815429684e-05,
      "step": 48910
    },
    {
      "epoch": 4.6644210815429684e-05,
      "step": 48910,
      "training_step_time": 8.3788902759552
    },
    {
      "epoch": 4.6645164489746095e-05,
      "model_forward_time": 1.8073346614837646,
      "step": 48911
    },
    {
      "epoch": 4.6645164489746095e-05,
      "step": 48911,
      "training_step_time": 7.758351564407349
    },
    {
      "epoch": 4.66461181640625e-05,
      "model_forward_time": 1.7358686923980713,
      "step": 48912
    },
    {
      "epoch": 4.66461181640625e-05,
      "step": 48912,
      "training_step_time": 8.669029474258423
    },
    {
      "epoch": 4.66470718383789e-05,
      "model_forward_time": 2.8659772872924805,
      "step": 48913
    },
    {
      "epoch": 4.66470718383789e-05,
      "step": 48913,
      "training_step_time": 9.068825960159302
    },
    {
      "epoch": 4.6648025512695314e-05,
      "model_forward_time": 1.7341563701629639,
      "step": 48914
    },
    {
      "epoch": 4.6648025512695314e-05,
      "step": 48914,
      "training_step_time": 8.505716323852539
    },
    {
      "epoch": 4.664897918701172e-05,
      "model_forward_time": 1.727137804031372,
      "step": 48915
    },
    {
      "epoch": 4.664897918701172e-05,
      "step": 48915,
      "training_step_time": 8.303070068359375
    },
    {
      "epoch": 4.664993286132812e-05,
      "model_forward_time": 1.729067325592041,
      "step": 48916
    },
    {
      "epoch": 4.664993286132812e-05,
      "step": 48916,
      "training_step_time": 8.62372875213623
    },
    {
      "epoch": 4.665088653564453e-05,
      "model_forward_time": 1.7333242893218994,
      "step": 48917
    },
    {
      "epoch": 4.665088653564453e-05,
      "step": 48917,
      "training_step_time": 7.7708563804626465
    },
    {
      "epoch": 4.665184020996094e-05,
      "model_forward_time": 1.750617265701294,
      "step": 48918
    },
    {
      "epoch": 4.665184020996094e-05,
      "step": 48918,
      "training_step_time": 7.364988327026367
    },
    {
      "epoch": 4.665279388427734e-05,
      "model_forward_time": 1.7486546039581299,
      "step": 48919
    },
    {
      "epoch": 4.665279388427734e-05,
      "step": 48919,
      "training_step_time": 8.772578477859497
    },
    {
      "epoch": 4.665374755859375e-05,
      "grad_norm": 0.09901970624923706,
      "learning_rate": 5.590724593252975e-06,
      "loss": 0.0603,
      "step": 48920
    },
    {
      "epoch": 4.665374755859375e-05,
      "model_forward_time": 1.3692047595977783,
      "step": 48920
    },
    {
      "dynamics_loss_avg": 0.06410469990223647,
      "epoch": 4.665374755859375e-05,
      "step": 48920
    },
    {
      "action_loss_avg": 0.0019477439782349393,
      "epoch": 4.665374755859375e-05,
      "step": 48920
    },
    {
      "epoch": 4.665374755859375e-05,
      "step": 48920,
      "training_step_time": 7.726781845092773
    },
    {
      "epoch": 4.6654701232910156e-05,
      "model_forward_time": 1.7458958625793457,
      "step": 48921
    },
    {
      "epoch": 4.6654701232910156e-05,
      "step": 48921,
      "training_step_time": 7.841856241226196
    },
    {
      "epoch": 4.665565490722656e-05,
      "model_forward_time": 2.9315123558044434,
      "step": 48922
    },
    {
      "epoch": 4.665565490722656e-05,
      "step": 48922,
      "training_step_time": 7.823374032974243
    },
    {
      "epoch": 4.665660858154297e-05,
      "model_forward_time": 1.7451443672180176,
      "step": 48923
    },
    {
      "epoch": 4.665660858154297e-05,
      "step": 48923,
      "training_step_time": 7.760522365570068
    },
    {
      "epoch": 4.6657562255859375e-05,
      "model_forward_time": 1.735584020614624,
      "step": 48924
    },
    {
      "epoch": 4.6657562255859375e-05,
      "step": 48924,
      "training_step_time": 8.472687482833862
    },
    {
      "epoch": 4.665851593017578e-05,
      "model_forward_time": 1.7454833984375,
      "step": 48925
    },
    {
      "epoch": 4.665851593017578e-05,
      "step": 48925,
      "training_step_time": 8.593018293380737
    },
    {
      "epoch": 4.665946960449219e-05,
      "model_forward_time": 1.7441959381103516,
      "step": 48926
    },
    {
      "epoch": 4.665946960449219e-05,
      "step": 48926,
      "training_step_time": 7.772852897644043
    },
    {
      "epoch": 4.666042327880859e-05,
      "model_forward_time": 1.373833417892456,
      "step": 48927
    },
    {
      "epoch": 4.666042327880859e-05,
      "step": 48927,
      "training_step_time": 7.90433931350708
    },
    {
      "epoch": 4.6661376953125e-05,
      "model_forward_time": 1.376854658126831,
      "step": 48928
    },
    {
      "epoch": 4.6661376953125e-05,
      "step": 48928,
      "training_step_time": 7.683916807174683
    },
    {
      "epoch": 4.666233062744141e-05,
      "model_forward_time": 1.7481653690338135,
      "step": 48929
    },
    {
      "epoch": 4.666233062744141e-05,
      "step": 48929,
      "training_step_time": 8.642065286636353
    },
    {
      "epoch": 4.666328430175781e-05,
      "grad_norm": 0.11749550700187683,
      "learning_rate": 5.589082671466184e-06,
      "loss": 0.0609,
      "step": 48930
    },
    {
      "epoch": 4.666328430175781e-05,
      "model_forward_time": 1.748260498046875,
      "step": 48930
    },
    {
      "dynamics_loss_avg": 0.0666646471247077,
      "epoch": 4.666328430175781e-05,
      "step": 48930
    },
    {
      "action_loss_avg": 0.003052532923175022,
      "epoch": 4.666328430175781e-05,
      "step": 48930
    },
    {
      "epoch": 4.666328430175781e-05,
      "step": 48930,
      "training_step_time": 8.752059698104858
    },
    {
      "epoch": 4.6664237976074216e-05,
      "model_forward_time": 1.804445743560791,
      "step": 48931
    },
    {
      "epoch": 4.6664237976074216e-05,
      "step": 48931,
      "training_step_time": 9.569415092468262
    },
    {
      "epoch": 4.666519165039063e-05,
      "model_forward_time": 2.9161205291748047,
      "step": 48932
    },
    {
      "epoch": 4.666519165039063e-05,
      "step": 48932,
      "training_step_time": 8.832711219787598
    },
    {
      "epoch": 4.666614532470703e-05,
      "model_forward_time": 1.747352123260498,
      "step": 48933
    },
    {
      "epoch": 4.666614532470703e-05,
      "step": 48933,
      "training_step_time": 7.783820629119873
    },
    {
      "epoch": 4.6667098999023435e-05,
      "model_forward_time": 1.7444953918457031,
      "step": 48934
    },
    {
      "epoch": 4.6667098999023435e-05,
      "step": 48934,
      "training_step_time": 7.796281337738037
    },
    {
      "epoch": 4.6668052673339846e-05,
      "model_forward_time": 1.744626522064209,
      "step": 48935
    },
    {
      "epoch": 4.6668052673339846e-05,
      "step": 48935,
      "training_step_time": 7.745624542236328
    },
    {
      "epoch": 4.666900634765625e-05,
      "model_forward_time": 1.7318737506866455,
      "step": 48936
    },
    {
      "epoch": 4.666900634765625e-05,
      "step": 48936,
      "training_step_time": 7.519693613052368
    },
    {
      "epoch": 4.6669960021972654e-05,
      "model_forward_time": 1.7321114540100098,
      "step": 48937
    },
    {
      "epoch": 4.6669960021972654e-05,
      "step": 48937,
      "training_step_time": 7.61566424369812
    },
    {
      "epoch": 4.6670913696289065e-05,
      "model_forward_time": 1.768537998199463,
      "step": 48938
    },
    {
      "epoch": 4.6670913696289065e-05,
      "step": 48938,
      "training_step_time": 8.495691299438477
    },
    {
      "epoch": 4.667186737060547e-05,
      "model_forward_time": 1.744560718536377,
      "step": 48939
    },
    {
      "epoch": 4.667186737060547e-05,
      "step": 48939,
      "training_step_time": 8.010398626327515
    },
    {
      "epoch": 4.667282104492187e-05,
      "grad_norm": 0.13424929976463318,
      "learning_rate": 5.587440685258199e-06,
      "loss": 0.0593,
      "step": 48940
    },
    {
      "epoch": 4.667282104492187e-05,
      "model_forward_time": 1.7440376281738281,
      "step": 48940
    },
    {
      "dynamics_loss_avg": 0.04914116058498621,
      "epoch": 4.667282104492187e-05,
      "step": 48940
    },
    {
      "action_loss_avg": 0.0012507815466960893,
      "epoch": 4.667282104492187e-05,
      "step": 48940
    },
    {
      "epoch": 4.667282104492187e-05,
      "step": 48940,
      "training_step_time": 9.681006908416748
    },
    {
      "epoch": 4.6673774719238284e-05,
      "model_forward_time": 2.9004735946655273,
      "step": 48941
    },
    {
      "epoch": 4.6673774719238284e-05,
      "step": 48941,
      "training_step_time": 8.669244766235352
    },
    {
      "epoch": 4.667472839355469e-05,
      "model_forward_time": 1.7483618259429932,
      "step": 48942
    },
    {
      "epoch": 4.667472839355469e-05,
      "step": 48942,
      "training_step_time": 7.849331617355347
    },
    {
      "epoch": 4.667568206787109e-05,
      "model_forward_time": 1.7344024181365967,
      "step": 48943
    },
    {
      "epoch": 4.667568206787109e-05,
      "step": 48943,
      "training_step_time": 8.382718563079834
    },
    {
      "epoch": 4.66766357421875e-05,
      "model_forward_time": 1.7366318702697754,
      "step": 48944
    },
    {
      "epoch": 4.66766357421875e-05,
      "step": 48944,
      "training_step_time": 7.775670528411865
    },
    {
      "epoch": 4.6677589416503907e-05,
      "model_forward_time": 1.3684277534484863,
      "step": 48945
    },
    {
      "epoch": 4.6677589416503907e-05,
      "step": 48945,
      "training_step_time": 7.360071420669556
    },
    {
      "epoch": 4.667854309082031e-05,
      "model_forward_time": 1.7337939739227295,
      "step": 48946
    },
    {
      "epoch": 4.667854309082031e-05,
      "step": 48946,
      "training_step_time": 7.542837619781494
    },
    {
      "epoch": 4.667949676513672e-05,
      "model_forward_time": 1.7510707378387451,
      "step": 48947
    },
    {
      "epoch": 4.667949676513672e-05,
      "step": 48947,
      "training_step_time": 8.811605215072632
    },
    {
      "epoch": 4.6680450439453125e-05,
      "model_forward_time": 1.7550265789031982,
      "step": 48948
    },
    {
      "epoch": 4.6680450439453125e-05,
      "step": 48948,
      "training_step_time": 8.631215572357178
    },
    {
      "epoch": 4.668140411376953e-05,
      "model_forward_time": 1.740997552871704,
      "step": 48949
    },
    {
      "epoch": 4.668140411376953e-05,
      "step": 48949,
      "training_step_time": 8.858057737350464
    },
    {
      "epoch": 4.668235778808594e-05,
      "grad_norm": 0.10297878831624985,
      "learning_rate": 5.5857986348085834e-06,
      "loss": 0.058,
      "step": 48950
    },
    {
      "epoch": 4.668235778808594e-05,
      "model_forward_time": 2.9661383628845215,
      "step": 48950
    },
    {
      "dynamics_loss_avg": 0.05399039164185524,
      "epoch": 4.668235778808594e-05,
      "step": 48950
    },
    {
      "action_loss_avg": 0.0014636593055911361,
      "epoch": 4.668235778808594e-05,
      "step": 48950
    },
    {
      "epoch": 4.668235778808594e-05,
      "step": 48950,
      "training_step_time": 9.185031414031982
    },
    {
      "epoch": 4.6683311462402344e-05,
      "model_forward_time": 1.732466220855713,
      "step": 48951
    },
    {
      "epoch": 4.6683311462402344e-05,
      "step": 48951,
      "training_step_time": 7.802670955657959
    },
    {
      "epoch": 4.668426513671875e-05,
      "model_forward_time": 1.7236292362213135,
      "step": 48952
    },
    {
      "epoch": 4.668426513671875e-05,
      "step": 48952,
      "training_step_time": 7.754281520843506
    },
    {
      "epoch": 4.668521881103516e-05,
      "model_forward_time": 1.7219877243041992,
      "step": 48953
    },
    {
      "epoch": 4.668521881103516e-05,
      "step": 48953,
      "training_step_time": 7.765794277191162
    },
    {
      "epoch": 4.668617248535156e-05,
      "model_forward_time": 1.737513780593872,
      "step": 48954
    },
    {
      "epoch": 4.668617248535156e-05,
      "step": 48954,
      "training_step_time": 7.86073112487793
    },
    {
      "epoch": 4.668712615966797e-05,
      "model_forward_time": 1.7499043941497803,
      "step": 48955
    },
    {
      "epoch": 4.668712615966797e-05,
      "step": 48955,
      "training_step_time": 7.470491886138916
    },
    {
      "epoch": 4.668807983398438e-05,
      "model_forward_time": 1.734748363494873,
      "step": 48956
    },
    {
      "epoch": 4.668807983398438e-05,
      "step": 48956,
      "training_step_time": 7.932465076446533
    },
    {
      "epoch": 4.668903350830078e-05,
      "model_forward_time": 1.7228188514709473,
      "step": 48957
    },
    {
      "epoch": 4.668903350830078e-05,
      "step": 48957,
      "training_step_time": 9.062870979309082
    },
    {
      "epoch": 4.6689987182617186e-05,
      "model_forward_time": 1.736823558807373,
      "step": 48958
    },
    {
      "epoch": 4.6689987182617186e-05,
      "step": 48958,
      "training_step_time": 9.31195878982544
    },
    {
      "epoch": 4.66909408569336e-05,
      "model_forward_time": 1.8347864151000977,
      "step": 48959
    },
    {
      "epoch": 4.66909408569336e-05,
      "step": 48959,
      "training_step_time": 9.521803617477417
    },
    {
      "epoch": 4.669189453125e-05,
      "grad_norm": 0.08966337889432907,
      "learning_rate": 5.584156520296909e-06,
      "loss": 0.0565,
      "step": 48960
    },
    {
      "epoch": 4.669189453125e-05,
      "model_forward_time": 2.8836042881011963,
      "step": 48960
    },
    {
      "dynamics_loss_avg": 0.04642555937170982,
      "epoch": 4.669189453125e-05,
      "step": 48960
    },
    {
      "action_loss_avg": 0.0008808716986095533,
      "epoch": 4.669189453125e-05,
      "step": 48960
    },
    {
      "epoch": 4.669189453125e-05,
      "step": 48960,
      "training_step_time": 8.814515113830566
    },
    {
      "epoch": 4.6692848205566405e-05,
      "model_forward_time": 1.7475392818450928,
      "step": 48961
    },
    {
      "epoch": 4.6692848205566405e-05,
      "step": 48961,
      "training_step_time": 7.828839063644409
    },
    {
      "epoch": 4.6693801879882816e-05,
      "model_forward_time": 1.7489423751831055,
      "step": 48962
    },
    {
      "epoch": 4.6693801879882816e-05,
      "step": 48962,
      "training_step_time": 7.762209177017212
    },
    {
      "epoch": 4.669475555419922e-05,
      "model_forward_time": 1.7459123134613037,
      "step": 48963
    },
    {
      "epoch": 4.669475555419922e-05,
      "step": 48963,
      "training_step_time": 7.781995058059692
    },
    {
      "epoch": 4.6695709228515624e-05,
      "model_forward_time": 1.7508490085601807,
      "step": 48964
    },
    {
      "epoch": 4.6695709228515624e-05,
      "step": 48964,
      "training_step_time": 7.249293327331543
    },
    {
      "epoch": 4.6696662902832035e-05,
      "model_forward_time": 1.738870620727539,
      "step": 48965
    },
    {
      "epoch": 4.6696662902832035e-05,
      "step": 48965,
      "training_step_time": 7.727237224578857
    },
    {
      "epoch": 4.669761657714844e-05,
      "model_forward_time": 1.7495648860931396,
      "step": 48966
    },
    {
      "epoch": 4.669761657714844e-05,
      "step": 48966,
      "training_step_time": 8.802985906600952
    },
    {
      "epoch": 4.669857025146484e-05,
      "model_forward_time": 1.7516236305236816,
      "step": 48967
    },
    {
      "epoch": 4.669857025146484e-05,
      "step": 48967,
      "training_step_time": 7.866772174835205
    },
    {
      "epoch": 4.669952392578125e-05,
      "model_forward_time": 1.819950819015503,
      "step": 48968
    },
    {
      "epoch": 4.669952392578125e-05,
      "step": 48968,
      "training_step_time": 8.57664442062378
    },
    {
      "epoch": 4.670047760009766e-05,
      "model_forward_time": 2.9160962104797363,
      "step": 48969
    },
    {
      "epoch": 4.670047760009766e-05,
      "step": 48969,
      "training_step_time": 7.802702188491821
    },
    {
      "epoch": 4.670143127441406e-05,
      "grad_norm": 0.11312832683324814,
      "learning_rate": 5.582514341902757e-06,
      "loss": 0.0594,
      "step": 48970
    },
    {
      "epoch": 4.670143127441406e-05,
      "model_forward_time": 1.7290947437286377,
      "step": 48970
    },
    {
      "dynamics_loss_avg": 0.06519283698871732,
      "epoch": 4.670143127441406e-05,
      "step": 48970
    },
    {
      "action_loss_avg": 0.0015813601086847485,
      "epoch": 4.670143127441406e-05,
      "step": 48970
    },
    {
      "epoch": 4.670143127441406e-05,
      "step": 48970,
      "training_step_time": 7.850668907165527
    },
    {
      "epoch": 4.6702384948730465e-05,
      "model_forward_time": 1.3764452934265137,
      "step": 48971
    },
    {
      "epoch": 4.6702384948730465e-05,
      "step": 48971,
      "training_step_time": 7.630148649215698
    },
    {
      "epoch": 4.6703338623046876e-05,
      "model_forward_time": 1.7314958572387695,
      "step": 48972
    },
    {
      "epoch": 4.6703338623046876e-05,
      "step": 48972,
      "training_step_time": 7.581474304199219
    },
    {
      "epoch": 4.670429229736328e-05,
      "model_forward_time": 1.7441823482513428,
      "step": 48973
    },
    {
      "epoch": 4.670429229736328e-05,
      "step": 48973,
      "training_step_time": 7.841378688812256
    },
    {
      "epoch": 4.6705245971679684e-05,
      "model_forward_time": 1.7369484901428223,
      "step": 48974
    },
    {
      "epoch": 4.6705245971679684e-05,
      "step": 48974,
      "training_step_time": 7.654718399047852
    },
    {
      "epoch": 4.6706199645996095e-05,
      "model_forward_time": 1.7485363483428955,
      "step": 48975
    },
    {
      "epoch": 4.6706199645996095e-05,
      "step": 48975,
      "training_step_time": 9.295997381210327
    },
    {
      "epoch": 4.67071533203125e-05,
      "model_forward_time": 1.7442340850830078,
      "step": 48976
    },
    {
      "epoch": 4.67071533203125e-05,
      "step": 48976,
      "training_step_time": 8.552116870880127
    },
    {
      "epoch": 4.67081069946289e-05,
      "model_forward_time": 1.8082916736602783,
      "step": 48977
    },
    {
      "epoch": 4.67081069946289e-05,
      "step": 48977,
      "training_step_time": 8.255529880523682
    },
    {
      "epoch": 4.6709060668945314e-05,
      "model_forward_time": 2.871361255645752,
      "step": 48978
    },
    {
      "epoch": 4.6709060668945314e-05,
      "step": 48978,
      "training_step_time": 8.41959023475647
    },
    {
      "epoch": 4.671001434326172e-05,
      "model_forward_time": 1.7416532039642334,
      "step": 48979
    },
    {
      "epoch": 4.671001434326172e-05,
      "step": 48979,
      "training_step_time": 7.743690490722656
    },
    {
      "epoch": 4.671096801757812e-05,
      "grad_norm": 0.12034915387630463,
      "learning_rate": 5.580872099805713e-06,
      "loss": 0.0589,
      "step": 48980
    },
    {
      "epoch": 4.671096801757812e-05,
      "model_forward_time": 1.739652156829834,
      "step": 48980
    },
    {
      "dynamics_loss_avg": 0.07854211665689945,
      "epoch": 4.671096801757812e-05,
      "step": 48980
    },
    {
      "action_loss_avg": 0.0024352087755687537,
      "epoch": 4.671096801757812e-05,
      "step": 48980
    },
    {
      "epoch": 4.671096801757812e-05,
      "step": 48980,
      "training_step_time": 7.825138330459595
    },
    {
      "epoch": 4.671192169189453e-05,
      "model_forward_time": 1.7439846992492676,
      "step": 48981
    },
    {
      "epoch": 4.671192169189453e-05,
      "step": 48981,
      "training_step_time": 7.708981990814209
    },
    {
      "epoch": 4.671287536621094e-05,
      "model_forward_time": 1.7287628650665283,
      "step": 48982
    },
    {
      "epoch": 4.671287536621094e-05,
      "step": 48982,
      "training_step_time": 7.729556322097778
    },
    {
      "epoch": 4.671382904052734e-05,
      "model_forward_time": 1.7532517910003662,
      "step": 48983
    },
    {
      "epoch": 4.671382904052734e-05,
      "step": 48983,
      "training_step_time": 7.5868611335754395
    },
    {
      "epoch": 4.671478271484375e-05,
      "model_forward_time": 1.7270259857177734,
      "step": 48984
    },
    {
      "epoch": 4.671478271484375e-05,
      "step": 48984,
      "training_step_time": 8.021907567977905
    },
    {
      "epoch": 4.6715736389160156e-05,
      "model_forward_time": 1.3585147857666016,
      "step": 48985
    },
    {
      "epoch": 4.6715736389160156e-05,
      "step": 48985,
      "training_step_time": 7.73228907585144
    },
    {
      "epoch": 4.671669006347656e-05,
      "model_forward_time": 1.7517294883728027,
      "step": 48986
    },
    {
      "epoch": 4.671669006347656e-05,
      "step": 48986,
      "training_step_time": 8.510427474975586
    },
    {
      "epoch": 4.671764373779297e-05,
      "model_forward_time": 2.8482937812805176,
      "step": 48987
    },
    {
      "epoch": 4.671764373779297e-05,
      "step": 48987,
      "training_step_time": 9.011961698532104
    },
    {
      "epoch": 4.6718597412109375e-05,
      "model_forward_time": 1.7218775749206543,
      "step": 48988
    },
    {
      "epoch": 4.6718597412109375e-05,
      "step": 48988,
      "training_step_time": 7.81752347946167
    },
    {
      "epoch": 4.671955108642578e-05,
      "model_forward_time": 1.7234385013580322,
      "step": 48989
    },
    {
      "epoch": 4.671955108642578e-05,
      "step": 48989,
      "training_step_time": 7.723164319992065
    },
    {
      "epoch": 4.672050476074219e-05,
      "grad_norm": 0.11448941379785538,
      "learning_rate": 5.579229794185368e-06,
      "loss": 0.0586,
      "step": 48990
    },
    {
      "epoch": 4.672050476074219e-05,
      "model_forward_time": 1.720123529434204,
      "step": 48990
    },
    {
      "dynamics_loss_avg": 0.04727024305611849,
      "epoch": 4.672050476074219e-05,
      "step": 48990
    },
    {
      "action_loss_avg": 0.0011010007176082582,
      "epoch": 4.672050476074219e-05,
      "step": 48990
    },
    {
      "epoch": 4.672050476074219e-05,
      "step": 48990,
      "training_step_time": 8.811223030090332
    },
    {
      "epoch": 4.6721458435058593e-05,
      "model_forward_time": 1.7235643863677979,
      "step": 48991
    },
    {
      "epoch": 4.6721458435058593e-05,
      "step": 48991,
      "training_step_time": 7.776705026626587
    },
    {
      "epoch": 4.6722412109375e-05,
      "model_forward_time": 1.76121187210083,
      "step": 48992
    },
    {
      "epoch": 4.6722412109375e-05,
      "step": 48992,
      "training_step_time": 7.541736364364624
    },
    {
      "epoch": 4.672336578369141e-05,
      "model_forward_time": 1.7250568866729736,
      "step": 48993
    },
    {
      "epoch": 4.672336578369141e-05,
      "step": 48993,
      "training_step_time": 8.598266363143921
    },
    {
      "epoch": 4.672431945800781e-05,
      "model_forward_time": 1.7225933074951172,
      "step": 48994
    },
    {
      "epoch": 4.672431945800781e-05,
      "step": 48994,
      "training_step_time": 7.846377372741699
    },
    {
      "epoch": 4.6725273132324216e-05,
      "model_forward_time": 1.7231922149658203,
      "step": 48995
    },
    {
      "epoch": 4.6725273132324216e-05,
      "step": 48995,
      "training_step_time": 7.761136531829834
    },
    {
      "epoch": 4.672622680664063e-05,
      "model_forward_time": 1.8296470642089844,
      "step": 48996
    },
    {
      "epoch": 4.672622680664063e-05,
      "step": 48996,
      "training_step_time": 10.523908376693726
    },
    {
      "epoch": 4.672718048095703e-05,
      "model_forward_time": 1.736820936203003,
      "step": 48997
    },
    {
      "epoch": 4.672718048095703e-05,
      "step": 48997,
      "training_step_time": 7.931548357009888
    },
    {
      "epoch": 4.6728134155273435e-05,
      "model_forward_time": 1.7344353199005127,
      "step": 48998
    },
    {
      "epoch": 4.6728134155273435e-05,
      "step": 48998,
      "training_step_time": 8.869322538375854
    },
    {
      "epoch": 4.6729087829589846e-05,
      "model_forward_time": 1.7178795337677002,
      "step": 48999
    },
    {
      "epoch": 4.6729087829589846e-05,
      "step": 48999,
      "training_step_time": 7.693138360977173
    },
    {
      "epoch": 4.673004150390625e-05,
      "grad_norm": 0.11874082684516907,
      "learning_rate": 5.577587425221325e-06,
      "loss": 0.0562,
      "step": 49000
    },
    {
      "epoch": 4.673004150390625e-05,
      "model_forward_time": 1.754594087600708,
      "step": 49000
    },
    {
      "dynamics_loss_avg": 0.05374551750719547,
      "epoch": 4.673004150390625e-05,
      "step": 49000
    },
    {
      "action_loss_avg": 0.001542764667829033,
      "epoch": 4.673004150390625e-05,
      "step": 49000
    },
    {
      "epoch": 4.673004150390625e-05,
      "step": 49000,
      "training_step_time": 7.280569076538086
    },
    {
      "epoch": 4.6730995178222654e-05,
      "model_forward_time": 1.7126505374908447,
      "step": 49001
    },
    {
      "epoch": 4.6730995178222654e-05,
      "step": 49001,
      "training_step_time": 8.319494724273682
    },
    {
      "epoch": 4.6731948852539065e-05,
      "model_forward_time": 1.7173471450805664,
      "step": 49002
    },
    {
      "epoch": 4.6731948852539065e-05,
      "step": 49002,
      "training_step_time": 8.730790853500366
    },
    {
      "epoch": 4.673290252685547e-05,
      "model_forward_time": 1.7144660949707031,
      "step": 49003
    },
    {
      "epoch": 4.673290252685547e-05,
      "step": 49003,
      "training_step_time": 9.739249229431152
    },
    {
      "epoch": 4.673385620117187e-05,
      "model_forward_time": 1.7082061767578125,
      "step": 49004
    },
    {
      "epoch": 4.673385620117187e-05,
      "step": 49004,
      "training_step_time": 12.921600103378296
    },
    {
      "epoch": 4.6734809875488284e-05,
      "model_forward_time": 2.7560811042785645,
      "step": 49005
    },
    {
      "epoch": 4.6734809875488284e-05,
      "step": 49005,
      "training_step_time": 8.713813304901123
    },
    {
      "epoch": 4.673576354980469e-05,
      "model_forward_time": 1.7143309116363525,
      "step": 49006
    },
    {
      "epoch": 4.673576354980469e-05,
      "step": 49006,
      "training_step_time": 8.28462553024292
    },
    {
      "epoch": 4.673671722412109e-05,
      "model_forward_time": 1.7290425300598145,
      "step": 49007
    },
    {
      "epoch": 4.673671722412109e-05,
      "step": 49007,
      "training_step_time": 10.43004560470581
    },
    {
      "epoch": 4.67376708984375e-05,
      "model_forward_time": 1.7110145092010498,
      "step": 49008
    },
    {
      "epoch": 4.67376708984375e-05,
      "step": 49008,
      "training_step_time": 7.83709192276001
    },
    {
      "epoch": 4.6738624572753907e-05,
      "model_forward_time": 1.729520320892334,
      "step": 49009
    },
    {
      "epoch": 4.6738624572753907e-05,
      "step": 49009,
      "training_step_time": 7.790180444717407
    },
    {
      "epoch": 4.673957824707031e-05,
      "grad_norm": 0.11671137809753418,
      "learning_rate": 5.575944993093188e-06,
      "loss": 0.0565,
      "step": 49010
    },
    {
      "epoch": 4.673957824707031e-05,
      "model_forward_time": 1.7139272689819336,
      "step": 49010
    },
    {
      "dynamics_loss_avg": 0.054005368053913115,
      "epoch": 4.673957824707031e-05,
      "step": 49010
    },
    {
      "action_loss_avg": 0.0015933094779029488,
      "epoch": 4.673957824707031e-05,
      "step": 49010
    },
    {
      "epoch": 4.673957824707031e-05,
      "step": 49010,
      "training_step_time": 7.517087697982788
    },
    {
      "epoch": 4.674053192138672e-05,
      "model_forward_time": 1.7208404541015625,
      "step": 49011
    },
    {
      "epoch": 4.674053192138672e-05,
      "step": 49011,
      "training_step_time": 9.459956407546997
    },
    {
      "epoch": 4.6741485595703125e-05,
      "model_forward_time": 1.7214975357055664,
      "step": 49012
    },
    {
      "epoch": 4.6741485595703125e-05,
      "step": 49012,
      "training_step_time": 7.880638599395752
    },
    {
      "epoch": 4.674243927001953e-05,
      "model_forward_time": 1.7202484607696533,
      "step": 49013
    },
    {
      "epoch": 4.674243927001953e-05,
      "step": 49013,
      "training_step_time": 10.996768712997437
    },
    {
      "epoch": 4.674339294433594e-05,
      "model_forward_time": 2.2702407836914062,
      "step": 49014
    },
    {
      "epoch": 4.674339294433594e-05,
      "step": 49014,
      "training_step_time": 8.865902662277222
    },
    {
      "epoch": 4.6744346618652344e-05,
      "model_forward_time": 1.7121853828430176,
      "step": 49015
    },
    {
      "epoch": 4.6744346618652344e-05,
      "step": 49015,
      "training_step_time": 7.845832586288452
    },
    {
      "epoch": 4.674530029296875e-05,
      "model_forward_time": 1.727003812789917,
      "step": 49016
    },
    {
      "epoch": 4.674530029296875e-05,
      "step": 49016,
      "training_step_time": 8.630755186080933
    },
    {
      "epoch": 4.674625396728516e-05,
      "model_forward_time": 1.705934762954712,
      "step": 49017
    },
    {
      "epoch": 4.674625396728516e-05,
      "step": 49017,
      "training_step_time": 7.849206447601318
    },
    {
      "epoch": 4.674720764160156e-05,
      "model_forward_time": 1.7502589225769043,
      "step": 49018
    },
    {
      "epoch": 4.674720764160156e-05,
      "step": 49018,
      "training_step_time": 7.82441258430481
    },
    {
      "epoch": 4.674816131591797e-05,
      "model_forward_time": 1.7162389755249023,
      "step": 49019
    },
    {
      "epoch": 4.674816131591797e-05,
      "step": 49019,
      "training_step_time": 7.526323318481445
    },
    {
      "epoch": 4.674911499023438e-05,
      "grad_norm": 0.11917905509471893,
      "learning_rate": 5.574302497980574e-06,
      "loss": 0.0578,
      "step": 49020
    },
    {
      "epoch": 4.674911499023438e-05,
      "model_forward_time": 1.7129848003387451,
      "step": 49020
    },
    {
      "dynamics_loss_avg": 0.056687625125050545,
      "epoch": 4.674911499023438e-05,
      "step": 49020
    },
    {
      "action_loss_avg": 0.0016505993437021971,
      "epoch": 4.674911499023438e-05,
      "step": 49020
    },
    {
      "epoch": 4.674911499023438e-05,
      "step": 49020,
      "training_step_time": 7.7685253620147705
    },
    {
      "epoch": 4.675006866455078e-05,
      "model_forward_time": 1.7128620147705078,
      "step": 49021
    },
    {
      "epoch": 4.675006866455078e-05,
      "step": 49021,
      "training_step_time": 8.827913284301758
    },
    {
      "epoch": 4.6751022338867186e-05,
      "model_forward_time": 1.7121188640594482,
      "step": 49022
    },
    {
      "epoch": 4.6751022338867186e-05,
      "step": 49022,
      "training_step_time": 8.516194820404053
    },
    {
      "epoch": 4.67519760131836e-05,
      "model_forward_time": 1.7054767608642578,
      "step": 49023
    },
    {
      "epoch": 4.67519760131836e-05,
      "step": 49023,
      "training_step_time": 8.379108190536499
    },
    {
      "epoch": 4.67529296875e-05,
      "model_forward_time": 1.6994423866271973,
      "step": 49024
    },
    {
      "epoch": 4.67529296875e-05,
      "step": 49024,
      "training_step_time": 7.858346223831177
    },
    {
      "epoch": 4.6753883361816405e-05,
      "model_forward_time": 1.7180376052856445,
      "step": 49025
    },
    {
      "epoch": 4.6753883361816405e-05,
      "step": 49025,
      "training_step_time": 12.01954960823059
    },
    {
      "epoch": 4.6754837036132816e-05,
      "model_forward_time": 1.6981847286224365,
      "step": 49026
    },
    {
      "epoch": 4.6754837036132816e-05,
      "step": 49026,
      "training_step_time": 7.023168087005615
    },
    {
      "epoch": 4.675579071044922e-05,
      "model_forward_time": 1.711029291152954,
      "step": 49027
    },
    {
      "epoch": 4.675579071044922e-05,
      "step": 49027,
      "training_step_time": 7.037369966506958
    },
    {
      "epoch": 4.6756744384765624e-05,
      "model_forward_time": 1.7033350467681885,
      "step": 49028
    },
    {
      "epoch": 4.6756744384765624e-05,
      "step": 49028,
      "training_step_time": 7.042972803115845
    },
    {
      "epoch": 4.6757698059082035e-05,
      "model_forward_time": 1.7139801979064941,
      "step": 49029
    },
    {
      "epoch": 4.6757698059082035e-05,
      "step": 49029,
      "training_step_time": 6.979543447494507
    },
    {
      "epoch": 4.675865173339844e-05,
      "grad_norm": 0.1111224815249443,
      "learning_rate": 5.5726599400631005e-06,
      "loss": 0.0547,
      "step": 49030
    },
    {
      "epoch": 4.675865173339844e-05,
      "model_forward_time": 1.7196815013885498,
      "step": 49030
    },
    {
      "dynamics_loss_avg": 0.05331124924123287,
      "epoch": 4.675865173339844e-05,
      "step": 49030
    },
    {
      "action_loss_avg": 0.0019398903707042337,
      "epoch": 4.675865173339844e-05,
      "step": 49030
    },
    {
      "epoch": 4.675865173339844e-05,
      "step": 49030,
      "training_step_time": 6.988006591796875
    },
    {
      "epoch": 4.675960540771484e-05,
      "model_forward_time": 1.6971817016601562,
      "step": 49031
    },
    {
      "epoch": 4.675960540771484e-05,
      "step": 49031,
      "training_step_time": 7.012215614318848
    },
    {
      "epoch": 4.676055908203125e-05,
      "model_forward_time": 1.7016303539276123,
      "step": 49032
    },
    {
      "epoch": 4.676055908203125e-05,
      "step": 49032,
      "training_step_time": 6.919691324234009
    },
    {
      "epoch": 4.676151275634766e-05,
      "model_forward_time": 1.3387925624847412,
      "step": 49033
    },
    {
      "epoch": 4.676151275634766e-05,
      "step": 49033,
      "training_step_time": 6.94301438331604
    },
    {
      "epoch": 4.676246643066406e-05,
      "model_forward_time": 1.7170734405517578,
      "step": 49034
    },
    {
      "epoch": 4.676246643066406e-05,
      "step": 49034,
      "training_step_time": 8.858604907989502
    },
    {
      "epoch": 4.6763420104980465e-05,
      "model_forward_time": 2.262822389602661,
      "step": 49035
    },
    {
      "epoch": 4.6763420104980465e-05,
      "step": 49035,
      "training_step_time": 7.915226459503174
    },
    {
      "epoch": 4.6764373779296876e-05,
      "model_forward_time": 1.7062416076660156,
      "step": 49036
    },
    {
      "epoch": 4.6764373779296876e-05,
      "step": 49036,
      "training_step_time": 7.008971691131592
    },
    {
      "epoch": 4.676532745361328e-05,
      "model_forward_time": 1.7107057571411133,
      "step": 49037
    },
    {
      "epoch": 4.676532745361328e-05,
      "step": 49037,
      "training_step_time": 6.9691481590271
    },
    {
      "epoch": 4.6766281127929684e-05,
      "model_forward_time": 1.713129997253418,
      "step": 49038
    },
    {
      "epoch": 4.6766281127929684e-05,
      "step": 49038,
      "training_step_time": 6.905283451080322
    },
    {
      "epoch": 4.6767234802246095e-05,
      "model_forward_time": 1.710784912109375,
      "step": 49039
    },
    {
      "epoch": 4.6767234802246095e-05,
      "step": 49039,
      "training_step_time": 7.021902561187744
    },
    {
      "epoch": 4.67681884765625e-05,
      "grad_norm": 0.1389106661081314,
      "learning_rate": 5.571017319520399e-06,
      "loss": 0.0553,
      "step": 49040
    },
    {
      "epoch": 4.67681884765625e-05,
      "model_forward_time": 1.717993974685669,
      "step": 49040
    },
    {
      "dynamics_loss_avg": 0.04860502853989601,
      "epoch": 4.67681884765625e-05,
      "step": 49040
    },
    {
      "action_loss_avg": 0.0015122179203899578,
      "epoch": 4.67681884765625e-05,
      "step": 49040
    },
    {
      "epoch": 4.67681884765625e-05,
      "step": 49040,
      "training_step_time": 6.983841180801392
    },
    {
      "epoch": 4.67691421508789e-05,
      "model_forward_time": 1.709071397781372,
      "step": 49041
    },
    {
      "epoch": 4.67691421508789e-05,
      "step": 49041,
      "training_step_time": 6.973644018173218
    },
    {
      "epoch": 4.6770095825195314e-05,
      "model_forward_time": 1.7149040699005127,
      "step": 49042
    },
    {
      "epoch": 4.6770095825195314e-05,
      "step": 49042,
      "training_step_time": 7.018416881561279
    },
    {
      "epoch": 4.677104949951172e-05,
      "model_forward_time": 1.7145140171051025,
      "step": 49043
    },
    {
      "epoch": 4.677104949951172e-05,
      "step": 49043,
      "training_step_time": 6.944604873657227
    },
    {
      "epoch": 4.677200317382812e-05,
      "model_forward_time": 2.287731647491455,
      "step": 49044
    },
    {
      "epoch": 4.677200317382812e-05,
      "step": 49044,
      "training_step_time": 7.949289321899414
    },
    {
      "epoch": 4.677295684814453e-05,
      "model_forward_time": 1.7099409103393555,
      "step": 49045
    },
    {
      "epoch": 4.677295684814453e-05,
      "step": 49045,
      "training_step_time": 6.9479265213012695
    },
    {
      "epoch": 4.677391052246094e-05,
      "model_forward_time": 1.7033467292785645,
      "step": 49046
    },
    {
      "epoch": 4.677391052246094e-05,
      "step": 49046,
      "training_step_time": 6.978963613510132
    },
    {
      "epoch": 4.677486419677734e-05,
      "model_forward_time": 1.7255728244781494,
      "step": 49047
    },
    {
      "epoch": 4.677486419677734e-05,
      "step": 49047,
      "training_step_time": 6.9722630977630615
    },
    {
      "epoch": 4.677581787109375e-05,
      "model_forward_time": 1.7090625762939453,
      "step": 49048
    },
    {
      "epoch": 4.677581787109375e-05,
      "step": 49048,
      "training_step_time": 6.995972633361816
    },
    {
      "epoch": 4.6776771545410156e-05,
      "model_forward_time": 1.7135627269744873,
      "step": 49049
    },
    {
      "epoch": 4.6776771545410156e-05,
      "step": 49049,
      "training_step_time": 6.933980226516724
    },
    {
      "epoch": 4.677772521972656e-05,
      "grad_norm": 0.11122363805770874,
      "learning_rate": 5.569374636532099e-06,
      "loss": 0.057,
      "step": 49050
    },
    {
      "epoch": 4.677772521972656e-05,
      "model_forward_time": 1.713439702987671,
      "step": 49050
    },
    {
      "dynamics_loss_avg": 0.05848991274833679,
      "epoch": 4.677772521972656e-05,
      "step": 49050
    },
    {
      "action_loss_avg": 0.001893238688353449,
      "epoch": 4.677772521972656e-05,
      "step": 49050
    },
    {
      "epoch": 4.677772521972656e-05,
      "step": 49050,
      "training_step_time": 7.0042030811309814
    },
    {
      "epoch": 4.677867889404297e-05,
      "model_forward_time": 1.721442699432373,
      "step": 49051
    },
    {
      "epoch": 4.677867889404297e-05,
      "step": 49051,
      "training_step_time": 6.982347249984741
    },
    {
      "epoch": 4.6779632568359375e-05,
      "model_forward_time": 1.7236313819885254,
      "step": 49052
    },
    {
      "epoch": 4.6779632568359375e-05,
      "step": 49052,
      "training_step_time": 7.079266309738159
    },
    {
      "epoch": 4.678058624267578e-05,
      "model_forward_time": 2.2836132049560547,
      "step": 49053
    },
    {
      "epoch": 4.678058624267578e-05,
      "step": 49053,
      "training_step_time": 9.647730350494385
    },
    {
      "epoch": 4.678153991699219e-05,
      "model_forward_time": 1.7133457660675049,
      "step": 49054
    },
    {
      "epoch": 4.678153991699219e-05,
      "step": 49054,
      "training_step_time": 7.713107347488403
    },
    {
      "epoch": 4.6782493591308593e-05,
      "model_forward_time": 1.7130537033081055,
      "step": 49055
    },
    {
      "epoch": 4.6782493591308593e-05,
      "step": 49055,
      "training_step_time": 7.076375961303711
    },
    {
      "epoch": 4.6783447265625e-05,
      "model_forward_time": 1.7030038833618164,
      "step": 49056
    },
    {
      "epoch": 4.6783447265625e-05,
      "step": 49056,
      "training_step_time": 6.997352600097656
    },
    {
      "epoch": 4.678440093994141e-05,
      "model_forward_time": 1.71134614944458,
      "step": 49057
    },
    {
      "epoch": 4.678440093994141e-05,
      "step": 49057,
      "training_step_time": 6.927705764770508
    },
    {
      "epoch": 4.678535461425781e-05,
      "model_forward_time": 1.7075819969177246,
      "step": 49058
    },
    {
      "epoch": 4.678535461425781e-05,
      "step": 49058,
      "training_step_time": 6.952193737030029
    },
    {
      "epoch": 4.6786308288574216e-05,
      "model_forward_time": 1.718515396118164,
      "step": 49059
    },
    {
      "epoch": 4.6786308288574216e-05,
      "step": 49059,
      "training_step_time": 6.947827577590942
    },
    {
      "epoch": 4.678726196289063e-05,
      "grad_norm": 0.12483329325914383,
      "learning_rate": 5.567731891277848e-06,
      "loss": 0.0584,
      "step": 49060
    },
    {
      "epoch": 4.678726196289063e-05,
      "model_forward_time": 1.7053096294403076,
      "step": 49060
    },
    {
      "dynamics_loss_avg": 0.045153630618005994,
      "epoch": 4.678726196289063e-05,
      "step": 49060
    },
    {
      "action_loss_avg": 0.0015376470430055634,
      "epoch": 4.678726196289063e-05,
      "step": 49060
    },
    {
      "epoch": 4.678726196289063e-05,
      "step": 49060,
      "training_step_time": 6.912276983261108
    },
    {
      "epoch": 4.678821563720703e-05,
      "model_forward_time": 1.70493483543396,
      "step": 49061
    },
    {
      "epoch": 4.678821563720703e-05,
      "step": 49061,
      "training_step_time": 6.986560583114624
    },
    {
      "epoch": 4.6789169311523435e-05,
      "model_forward_time": 2.2982845306396484,
      "step": 49062
    },
    {
      "epoch": 4.6789169311523435e-05,
      "step": 49062,
      "training_step_time": 9.463492155075073
    },
    {
      "epoch": 4.6790122985839846e-05,
      "model_forward_time": 1.7098445892333984,
      "step": 49063
    },
    {
      "epoch": 4.6790122985839846e-05,
      "step": 49063,
      "training_step_time": 7.849250078201294
    },
    {
      "epoch": 4.679107666015625e-05,
      "model_forward_time": 1.7133488655090332,
      "step": 49064
    },
    {
      "epoch": 4.679107666015625e-05,
      "step": 49064,
      "training_step_time": 6.917705297470093
    },
    {
      "epoch": 4.6792030334472654e-05,
      "model_forward_time": 1.71474289894104,
      "step": 49065
    },
    {
      "epoch": 4.6792030334472654e-05,
      "step": 49065,
      "training_step_time": 6.988453149795532
    },
    {
      "epoch": 4.6792984008789065e-05,
      "model_forward_time": 1.712716817855835,
      "step": 49066
    },
    {
      "epoch": 4.6792984008789065e-05,
      "step": 49066,
      "training_step_time": 6.978089332580566
    },
    {
      "epoch": 4.679393768310547e-05,
      "model_forward_time": 1.3373353481292725,
      "step": 49067
    },
    {
      "epoch": 4.679393768310547e-05,
      "step": 49067,
      "training_step_time": 6.940768718719482
    },
    {
      "epoch": 4.679489135742187e-05,
      "model_forward_time": 1.712059497833252,
      "step": 49068
    },
    {
      "epoch": 4.679489135742187e-05,
      "step": 49068,
      "training_step_time": 6.949892282485962
    },
    {
      "epoch": 4.6795845031738284e-05,
      "model_forward_time": 1.7134969234466553,
      "step": 49069
    },
    {
      "epoch": 4.6795845031738284e-05,
      "step": 49069,
      "training_step_time": 6.966588973999023
    },
    {
      "epoch": 4.679679870605469e-05,
      "grad_norm": 0.10728706419467926,
      "learning_rate": 5.566089083937288e-06,
      "loss": 0.055,
      "step": 49070
    },
    {
      "epoch": 4.679679870605469e-05,
      "model_forward_time": 1.7088770866394043,
      "step": 49070
    },
    {
      "dynamics_loss_avg": 0.052632554434239864,
      "epoch": 4.679679870605469e-05,
      "step": 49070
    },
    {
      "action_loss_avg": 0.0023425965919159353,
      "epoch": 4.679679870605469e-05,
      "step": 49070
    },
    {
      "epoch": 4.679679870605469e-05,
      "step": 49070,
      "training_step_time": 6.9427478313446045
    },
    {
      "epoch": 4.679775238037109e-05,
      "model_forward_time": 1.7223255634307861,
      "step": 49071
    },
    {
      "epoch": 4.679775238037109e-05,
      "step": 49071,
      "training_step_time": 10.831522703170776
    },
    {
      "epoch": 4.67987060546875e-05,
      "model_forward_time": 2.2447431087493896,
      "step": 49072
    },
    {
      "epoch": 4.67987060546875e-05,
      "step": 49072,
      "training_step_time": 7.857557058334351
    },
    {
      "epoch": 4.679965972900391e-05,
      "model_forward_time": 1.713808536529541,
      "step": 49073
    },
    {
      "epoch": 4.679965972900391e-05,
      "step": 49073,
      "training_step_time": 6.995851039886475
    },
    {
      "epoch": 4.680061340332031e-05,
      "model_forward_time": 1.7083210945129395,
      "step": 49074
    },
    {
      "epoch": 4.680061340332031e-05,
      "step": 49074,
      "training_step_time": 6.9356608390808105
    },
    {
      "epoch": 4.680156707763672e-05,
      "model_forward_time": 1.7080950736999512,
      "step": 49075
    },
    {
      "epoch": 4.680156707763672e-05,
      "step": 49075,
      "training_step_time": 6.912656307220459
    },
    {
      "epoch": 4.6802520751953125e-05,
      "model_forward_time": 1.7121689319610596,
      "step": 49076
    },
    {
      "epoch": 4.6802520751953125e-05,
      "step": 49076,
      "training_step_time": 6.945528984069824
    },
    {
      "epoch": 4.680347442626953e-05,
      "model_forward_time": 1.7082107067108154,
      "step": 49077
    },
    {
      "epoch": 4.680347442626953e-05,
      "step": 49077,
      "training_step_time": 6.928544044494629
    },
    {
      "epoch": 4.680442810058594e-05,
      "model_forward_time": 1.7183127403259277,
      "step": 49078
    },
    {
      "epoch": 4.680442810058594e-05,
      "step": 49078,
      "training_step_time": 6.944010019302368
    },
    {
      "epoch": 4.6805381774902344e-05,
      "model_forward_time": 1.712134599685669,
      "step": 49079
    },
    {
      "epoch": 4.6805381774902344e-05,
      "step": 49079,
      "training_step_time": 6.952963590621948
    },
    {
      "epoch": 4.680633544921875e-05,
      "grad_norm": 0.13905884325504303,
      "learning_rate": 5.564446214690079e-06,
      "loss": 0.0568,
      "step": 49080
    },
    {
      "epoch": 4.680633544921875e-05,
      "model_forward_time": 1.7090716361999512,
      "step": 49080
    },
    {
      "dynamics_loss_avg": 0.054264478385448456,
      "epoch": 4.680633544921875e-05,
      "step": 49080
    },
    {
      "action_loss_avg": 0.0017380881748977118,
      "epoch": 4.680633544921875e-05,
      "step": 49080
    },
    {
      "epoch": 4.680633544921875e-05,
      "step": 49080,
      "training_step_time": 10.92581820487976
    },
    {
      "epoch": 4.680728912353516e-05,
      "model_forward_time": 2.2657930850982666,
      "step": 49081
    },
    {
      "epoch": 4.680728912353516e-05,
      "step": 49081,
      "training_step_time": 8.691455125808716
    },
    {
      "epoch": 4.680824279785156e-05,
      "model_forward_time": 1.7096893787384033,
      "step": 49082
    },
    {
      "epoch": 4.680824279785156e-05,
      "step": 49082,
      "training_step_time": 6.946399211883545
    },
    {
      "epoch": 4.680919647216797e-05,
      "model_forward_time": 1.7193660736083984,
      "step": 49083
    },
    {
      "epoch": 4.680919647216797e-05,
      "step": 49083,
      "training_step_time": 6.954692125320435
    },
    {
      "epoch": 4.681015014648438e-05,
      "model_forward_time": 1.7111454010009766,
      "step": 49084
    },
    {
      "epoch": 4.681015014648438e-05,
      "step": 49084,
      "training_step_time": 6.994349956512451
    },
    {
      "epoch": 4.681110382080078e-05,
      "model_forward_time": 1.7089955806732178,
      "step": 49085
    },
    {
      "epoch": 4.681110382080078e-05,
      "step": 49085,
      "training_step_time": 6.948655605316162
    },
    {
      "epoch": 4.6812057495117186e-05,
      "model_forward_time": 1.7132582664489746,
      "step": 49086
    },
    {
      "epoch": 4.6812057495117186e-05,
      "step": 49086,
      "training_step_time": 6.9647393226623535
    },
    {
      "epoch": 4.68130111694336e-05,
      "model_forward_time": 1.7102861404418945,
      "step": 49087
    },
    {
      "epoch": 4.68130111694336e-05,
      "step": 49087,
      "training_step_time": 7.0005903244018555
    },
    {
      "epoch": 4.681396484375e-05,
      "model_forward_time": 1.7181334495544434,
      "step": 49088
    },
    {
      "epoch": 4.681396484375e-05,
      "step": 49088,
      "training_step_time": 6.937992811203003
    },
    {
      "epoch": 4.6814918518066405e-05,
      "model_forward_time": 1.7068932056427002,
      "step": 49089
    },
    {
      "epoch": 4.6814918518066405e-05,
      "step": 49089,
      "training_step_time": 10.771481275558472
    },
    {
      "epoch": 4.6815872192382816e-05,
      "grad_norm": 0.11842992901802063,
      "learning_rate": 5.56280328371588e-06,
      "loss": 0.0612,
      "step": 49090
    },
    {
      "epoch": 4.6815872192382816e-05,
      "model_forward_time": 2.2795209884643555,
      "step": 49090
    },
    {
      "dynamics_loss_avg": 0.05322712566703558,
      "epoch": 4.6815872192382816e-05,
      "step": 49090
    },
    {
      "action_loss_avg": 0.0017896699602715671,
      "epoch": 4.6815872192382816e-05,
      "step": 49090
    },
    {
      "epoch": 4.6815872192382816e-05,
      "step": 49090,
      "training_step_time": 7.876690149307251
    },
    {
      "epoch": 4.681682586669922e-05,
      "model_forward_time": 1.7167437076568604,
      "step": 49091
    },
    {
      "epoch": 4.681682586669922e-05,
      "step": 49091,
      "training_step_time": 7.705111026763916
    },
    {
      "epoch": 4.6817779541015624e-05,
      "model_forward_time": 0.9970753192901611,
      "step": 49092
    },
    {
      "epoch": 4.6817779541015624e-05,
      "step": 49092,
      "training_step_time": 6.935850381851196
    },
    {
      "epoch": 4.6818733215332035e-05,
      "model_forward_time": 1.7111947536468506,
      "step": 49093
    },
    {
      "epoch": 4.6818733215332035e-05,
      "step": 49093,
      "training_step_time": 6.947539329528809
    },
    {
      "epoch": 4.681968688964844e-05,
      "model_forward_time": 1.7163972854614258,
      "step": 49094
    },
    {
      "epoch": 4.681968688964844e-05,
      "step": 49094,
      "training_step_time": 6.9800636768341064
    },
    {
      "epoch": 4.682064056396484e-05,
      "model_forward_time": 1.7080726623535156,
      "step": 49095
    },
    {
      "epoch": 4.682064056396484e-05,
      "step": 49095,
      "training_step_time": 6.940840482711792
    },
    {
      "epoch": 4.682159423828125e-05,
      "model_forward_time": 1.7104051113128662,
      "step": 49096
    },
    {
      "epoch": 4.682159423828125e-05,
      "step": 49096,
      "training_step_time": 6.928438901901245
    },
    {
      "epoch": 4.682254791259766e-05,
      "model_forward_time": 1.6998255252838135,
      "step": 49097
    },
    {
      "epoch": 4.682254791259766e-05,
      "step": 49097,
      "training_step_time": 6.985961675643921
    },
    {
      "epoch": 4.682350158691406e-05,
      "model_forward_time": 1.7004432678222656,
      "step": 49098
    },
    {
      "epoch": 4.682350158691406e-05,
      "step": 49098,
      "training_step_time": 9.799388647079468
    },
    {
      "epoch": 4.6824455261230466e-05,
      "model_forward_time": 1.736128807067871,
      "step": 49099
    },
    {
      "epoch": 4.6824455261230466e-05,
      "step": 49099,
      "training_step_time": 10.625462293624878
    },
    {
      "epoch": 4.6825408935546876e-05,
      "grad_norm": 0.13332431018352509,
      "learning_rate": 5.56116029119436e-06,
      "loss": 0.0582,
      "step": 49100
    },
    {
      "epoch": 4.6825408935546876e-05,
      "model_forward_time": 2.293940305709839,
      "step": 49100
    },
    {
      "dynamics_loss_avg": 0.06519629433751106,
      "epoch": 4.6825408935546876e-05,
      "step": 49100
    },
    {
      "action_loss_avg": 0.0019769508042372765,
      "epoch": 4.6825408935546876e-05,
      "step": 49100
    },
    {
      "epoch": 4.6825408935546876e-05,
      "step": 49100,
      "training_step_time": 7.802083253860474
    },
    {
      "epoch": 4.682636260986328e-05,
      "model_forward_time": 1.33345627784729,
      "step": 49101
    },
    {
      "epoch": 4.682636260986328e-05,
      "step": 49101,
      "training_step_time": 6.983221769332886
    },
    {
      "epoch": 4.6827316284179684e-05,
      "model_forward_time": 1.7074720859527588,
      "step": 49102
    },
    {
      "epoch": 4.6827316284179684e-05,
      "step": 49102,
      "training_step_time": 6.969395637512207
    },
    {
      "epoch": 4.6828269958496095e-05,
      "model_forward_time": 1.710723876953125,
      "step": 49103
    },
    {
      "epoch": 4.6828269958496095e-05,
      "step": 49103,
      "training_step_time": 7.048226356506348
    },
    {
      "epoch": 4.68292236328125e-05,
      "model_forward_time": 1.7074766159057617,
      "step": 49104
    },
    {
      "epoch": 4.68292236328125e-05,
      "step": 49104,
      "training_step_time": 6.985445499420166
    },
    {
      "epoch": 4.68301773071289e-05,
      "model_forward_time": 1.7109792232513428,
      "step": 49105
    },
    {
      "epoch": 4.68301773071289e-05,
      "step": 49105,
      "training_step_time": 6.934506893157959
    },
    {
      "epoch": 4.6831130981445314e-05,
      "model_forward_time": 1.7125661373138428,
      "step": 49106
    },
    {
      "epoch": 4.6831130981445314e-05,
      "step": 49106,
      "training_step_time": 6.989574909210205
    },
    {
      "epoch": 4.683208465576172e-05,
      "model_forward_time": 1.7076025009155273,
      "step": 49107
    },
    {
      "epoch": 4.683208465576172e-05,
      "step": 49107,
      "training_step_time": 8.58385157585144
    },
    {
      "epoch": 4.683303833007812e-05,
      "model_forward_time": 1.716679334640503,
      "step": 49108
    },
    {
      "epoch": 4.683303833007812e-05,
      "step": 49108,
      "training_step_time": 7.847082614898682
    },
    {
      "epoch": 4.683399200439453e-05,
      "model_forward_time": 2.276137351989746,
      "step": 49109
    },
    {
      "epoch": 4.683399200439453e-05,
      "step": 49109,
      "training_step_time": 7.843714237213135
    },
    {
      "epoch": 4.683494567871094e-05,
      "grad_norm": 0.09993310272693634,
      "learning_rate": 5.559517237305194e-06,
      "loss": 0.058,
      "step": 49110
    },
    {
      "epoch": 4.683494567871094e-05,
      "model_forward_time": 1.7069346904754639,
      "step": 49110
    },
    {
      "dynamics_loss_avg": 0.05042820516973734,
      "epoch": 4.683494567871094e-05,
      "step": 49110
    },
    {
      "action_loss_avg": 0.0017978703443077392,
      "epoch": 4.683494567871094e-05,
      "step": 49110
    },
    {
      "epoch": 4.683494567871094e-05,
      "step": 49110,
      "training_step_time": 6.9079203605651855
    },
    {
      "epoch": 4.683589935302734e-05,
      "model_forward_time": 1.7093901634216309,
      "step": 49111
    },
    {
      "epoch": 4.683589935302734e-05,
      "step": 49111,
      "training_step_time": 6.974810361862183
    },
    {
      "epoch": 4.683685302734375e-05,
      "model_forward_time": 1.7092134952545166,
      "step": 49112
    },
    {
      "epoch": 4.683685302734375e-05,
      "step": 49112,
      "training_step_time": 6.994534254074097
    },
    {
      "epoch": 4.6837806701660156e-05,
      "model_forward_time": 1.7039921283721924,
      "step": 49113
    },
    {
      "epoch": 4.6837806701660156e-05,
      "step": 49113,
      "training_step_time": 6.958519697189331
    },
    {
      "epoch": 4.683876037597656e-05,
      "model_forward_time": 1.7133471965789795,
      "step": 49114
    },
    {
      "epoch": 4.683876037597656e-05,
      "step": 49114,
      "training_step_time": 6.951684951782227
    },
    {
      "epoch": 4.683971405029297e-05,
      "model_forward_time": 1.7113497257232666,
      "step": 49115
    },
    {
      "epoch": 4.683971405029297e-05,
      "step": 49115,
      "training_step_time": 6.99278450012207
    },
    {
      "epoch": 4.6840667724609375e-05,
      "model_forward_time": 1.7105941772460938,
      "step": 49116
    },
    {
      "epoch": 4.6840667724609375e-05,
      "step": 49116,
      "training_step_time": 8.002282857894897
    },
    {
      "epoch": 4.684162139892578e-05,
      "model_forward_time": 1.7096703052520752,
      "step": 49117
    },
    {
      "epoch": 4.684162139892578e-05,
      "step": 49117,
      "training_step_time": 11.880014419555664
    },
    {
      "epoch": 4.684257507324219e-05,
      "model_forward_time": 2.2657558917999268,
      "step": 49118
    },
    {
      "epoch": 4.684257507324219e-05,
      "step": 49118,
      "training_step_time": 8.816389083862305
    },
    {
      "epoch": 4.6843528747558594e-05,
      "model_forward_time": 1.3346607685089111,
      "step": 49119
    },
    {
      "epoch": 4.6843528747558594e-05,
      "step": 49119,
      "training_step_time": 7.69968843460083
    },
    {
      "epoch": 4.6844482421875e-05,
      "grad_norm": 0.10334786027669907,
      "learning_rate": 5.557874122228064e-06,
      "loss": 0.0592,
      "step": 49120
    },
    {
      "epoch": 4.6844482421875e-05,
      "model_forward_time": 1.7081713676452637,
      "step": 49120
    },
    {
      "dynamics_loss_avg": 0.04450668087229133,
      "epoch": 4.6844482421875e-05,
      "step": 49120
    },
    {
      "action_loss_avg": 0.002022940694587305,
      "epoch": 4.6844482421875e-05,
      "step": 49120
    },
    {
      "epoch": 4.6844482421875e-05,
      "step": 49120,
      "training_step_time": 6.943166255950928
    },
    {
      "epoch": 4.684543609619141e-05,
      "model_forward_time": 1.712327241897583,
      "step": 49121
    },
    {
      "epoch": 4.684543609619141e-05,
      "step": 49121,
      "training_step_time": 6.906529188156128
    },
    {
      "epoch": 4.684638977050781e-05,
      "model_forward_time": 1.7091376781463623,
      "step": 49122
    },
    {
      "epoch": 4.684638977050781e-05,
      "step": 49122,
      "training_step_time": 6.9159300327301025
    },
    {
      "epoch": 4.6847343444824216e-05,
      "model_forward_time": 1.7097327709197998,
      "step": 49123
    },
    {
      "epoch": 4.6847343444824216e-05,
      "step": 49123,
      "training_step_time": 6.926990509033203
    },
    {
      "epoch": 4.684829711914063e-05,
      "model_forward_time": 1.715151309967041,
      "step": 49124
    },
    {
      "epoch": 4.684829711914063e-05,
      "step": 49124,
      "training_step_time": 6.964441299438477
    },
    {
      "epoch": 4.684925079345703e-05,
      "model_forward_time": 1.7061665058135986,
      "step": 49125
    },
    {
      "epoch": 4.684925079345703e-05,
      "step": 49125,
      "training_step_time": 10.130085468292236
    },
    {
      "epoch": 4.6850204467773435e-05,
      "model_forward_time": 1.718130111694336,
      "step": 49126
    },
    {
      "epoch": 4.6850204467773435e-05,
      "step": 49126,
      "training_step_time": 10.0226891040802
    },
    {
      "epoch": 4.6851158142089846e-05,
      "model_forward_time": 2.280261754989624,
      "step": 49127
    },
    {
      "epoch": 4.6851158142089846e-05,
      "step": 49127,
      "training_step_time": 10.55633544921875
    },
    {
      "epoch": 4.685211181640625e-05,
      "model_forward_time": 1.7056918144226074,
      "step": 49128
    },
    {
      "epoch": 4.685211181640625e-05,
      "step": 49128,
      "training_step_time": 7.754839181900024
    },
    {
      "epoch": 4.6853065490722654e-05,
      "model_forward_time": 1.7054994106292725,
      "step": 49129
    },
    {
      "epoch": 4.6853065490722654e-05,
      "step": 49129,
      "training_step_time": 6.930791616439819
    },
    {
      "epoch": 4.6854019165039065e-05,
      "grad_norm": 0.18660035729408264,
      "learning_rate": 5.55623094614266e-06,
      "loss": 0.0605,
      "step": 49130
    },
    {
      "epoch": 4.6854019165039065e-05,
      "model_forward_time": 1.7106826305389404,
      "step": 49130
    },
    {
      "dynamics_loss_avg": 0.05605129543691874,
      "epoch": 4.6854019165039065e-05,
      "step": 49130
    },
    {
      "action_loss_avg": 0.0018712746212258935,
      "epoch": 4.6854019165039065e-05,
      "step": 49130
    },
    {
      "epoch": 4.6854019165039065e-05,
      "step": 49130,
      "training_step_time": 6.927931785583496
    },
    {
      "epoch": 4.685497283935547e-05,
      "model_forward_time": 1.7095410823822021,
      "step": 49131
    },
    {
      "epoch": 4.685497283935547e-05,
      "step": 49131,
      "training_step_time": 7.009260177612305
    },
    {
      "epoch": 4.685592651367187e-05,
      "model_forward_time": 1.707343578338623,
      "step": 49132
    },
    {
      "epoch": 4.685592651367187e-05,
      "step": 49132,
      "training_step_time": 6.966309070587158
    },
    {
      "epoch": 4.6856880187988284e-05,
      "model_forward_time": 1.711857557296753,
      "step": 49133
    },
    {
      "epoch": 4.6856880187988284e-05,
      "step": 49133,
      "training_step_time": 6.988722801208496
    },
    {
      "epoch": 4.685783386230469e-05,
      "model_forward_time": 1.7053039073944092,
      "step": 49134
    },
    {
      "epoch": 4.685783386230469e-05,
      "step": 49134,
      "training_step_time": 9.247439861297607
    },
    {
      "epoch": 4.685878753662109e-05,
      "model_forward_time": 1.712524652481079,
      "step": 49135
    },
    {
      "epoch": 4.685878753662109e-05,
      "step": 49135,
      "training_step_time": 7.812920808792114
    },
    {
      "epoch": 4.68597412109375e-05,
      "model_forward_time": 2.285914897918701,
      "step": 49136
    },
    {
      "epoch": 4.68597412109375e-05,
      "step": 49136,
      "training_step_time": 7.98661470413208
    },
    {
      "epoch": 4.686069488525391e-05,
      "model_forward_time": 1.7122910022735596,
      "step": 49137
    },
    {
      "epoch": 4.686069488525391e-05,
      "step": 49137,
      "training_step_time": 7.829586029052734
    },
    {
      "epoch": 4.686164855957031e-05,
      "model_forward_time": 1.710766315460205,
      "step": 49138
    },
    {
      "epoch": 4.686164855957031e-05,
      "step": 49138,
      "training_step_time": 6.932152271270752
    },
    {
      "epoch": 4.686260223388672e-05,
      "model_forward_time": 1.7132604122161865,
      "step": 49139
    },
    {
      "epoch": 4.686260223388672e-05,
      "step": 49139,
      "training_step_time": 7.015033721923828
    },
    {
      "epoch": 4.6863555908203126e-05,
      "grad_norm": 0.20628060400485992,
      "learning_rate": 5.554587709228675e-06,
      "loss": 0.0572,
      "step": 49140
    },
    {
      "epoch": 4.6863555908203126e-05,
      "model_forward_time": 1.708174705505371,
      "step": 49140
    },
    {
      "dynamics_loss_avg": 0.059797149151563644,
      "epoch": 4.6863555908203126e-05,
      "step": 49140
    },
    {
      "action_loss_avg": 0.0021709966531489044,
      "epoch": 4.6863555908203126e-05,
      "step": 49140
    },
    {
      "epoch": 4.6863555908203126e-05,
      "step": 49140,
      "training_step_time": 6.961467027664185
    },
    {
      "epoch": 4.686450958251953e-05,
      "model_forward_time": 1.7139945030212402,
      "step": 49141
    },
    {
      "epoch": 4.686450958251953e-05,
      "step": 49141,
      "training_step_time": 6.999346494674683
    },
    {
      "epoch": 4.686546325683594e-05,
      "model_forward_time": 1.708432674407959,
      "step": 49142
    },
    {
      "epoch": 4.686546325683594e-05,
      "step": 49142,
      "training_step_time": 6.985172271728516
    },
    {
      "epoch": 4.6866416931152344e-05,
      "model_forward_time": 1.7084269523620605,
      "step": 49143
    },
    {
      "epoch": 4.6866416931152344e-05,
      "step": 49143,
      "training_step_time": 9.080363988876343
    },
    {
      "epoch": 4.686737060546875e-05,
      "model_forward_time": 1.7128705978393555,
      "step": 49144
    },
    {
      "epoch": 4.686737060546875e-05,
      "step": 49144,
      "training_step_time": 7.810447692871094
    },
    {
      "epoch": 4.686832427978516e-05,
      "model_forward_time": 1.7188997268676758,
      "step": 49145
    },
    {
      "epoch": 4.686832427978516e-05,
      "step": 49145,
      "training_step_time": 7.94318699836731
    },
    {
      "epoch": 4.686927795410156e-05,
      "model_forward_time": 2.268718719482422,
      "step": 49146
    },
    {
      "epoch": 4.686927795410156e-05,
      "step": 49146,
      "training_step_time": 7.76571249961853
    },
    {
      "epoch": 4.687023162841797e-05,
      "model_forward_time": 1.7185256481170654,
      "step": 49147
    },
    {
      "epoch": 4.687023162841797e-05,
      "step": 49147,
      "training_step_time": 7.766388893127441
    },
    {
      "epoch": 4.687118530273438e-05,
      "model_forward_time": 1.7184734344482422,
      "step": 49148
    },
    {
      "epoch": 4.687118530273438e-05,
      "step": 49148,
      "training_step_time": 6.989580154418945
    },
    {
      "epoch": 4.687213897705078e-05,
      "model_forward_time": 1.7089483737945557,
      "step": 49149
    },
    {
      "epoch": 4.687213897705078e-05,
      "step": 49149,
      "training_step_time": 7.01673698425293
    },
    {
      "epoch": 4.6873092651367186e-05,
      "grad_norm": 0.0956825464963913,
      "learning_rate": 5.552944411665811e-06,
      "loss": 0.0571,
      "step": 49150
    },
    {
      "epoch": 4.6873092651367186e-05,
      "model_forward_time": 1.6873724460601807,
      "step": 49150
    },
    {
      "dynamics_loss_avg": 0.05768154375255108,
      "epoch": 4.6873092651367186e-05,
      "step": 49150
    },
    {
      "action_loss_avg": 0.0014555570553056896,
      "epoch": 4.6873092651367186e-05,
      "step": 49150
    },
    {
      "epoch": 4.6873092651367186e-05,
      "step": 49150,
      "training_step_time": 6.9380104541778564
    },
    {
      "epoch": 4.68740463256836e-05,
      "model_forward_time": 1.7065043449401855,
      "step": 49151
    },
    {
      "epoch": 4.68740463256836e-05,
      "step": 49151,
      "training_step_time": 7.00295352935791
    },
    {
      "epoch": 4.6875e-05,
      "model_forward_time": 1.6955559253692627,
      "step": 49152
    },
    {
      "epoch": 4.6875e-05,
      "step": 49152,
      "training_step_time": 9.669307708740234
    },
    {
      "epoch": 4.6875953674316405e-05,
      "model_forward_time": 1.7021899223327637,
      "step": 49153
    },
    {
      "epoch": 4.6875953674316405e-05,
      "step": 49153,
      "training_step_time": 8.859774112701416
    },
    {
      "epoch": 4.6876907348632816e-05,
      "model_forward_time": 1.704115390777588,
      "step": 49154
    },
    {
      "epoch": 4.6876907348632816e-05,
      "step": 49154,
      "training_step_time": 7.855319023132324
    },
    {
      "epoch": 4.687786102294922e-05,
      "model_forward_time": 2.2600886821746826,
      "step": 49155
    },
    {
      "epoch": 4.687786102294922e-05,
      "step": 49155,
      "training_step_time": 7.884254217147827
    },
    {
      "epoch": 4.6878814697265624e-05,
      "model_forward_time": 1.6990656852722168,
      "step": 49156
    },
    {
      "epoch": 4.6878814697265624e-05,
      "step": 49156,
      "training_step_time": 7.752016544342041
    },
    {
      "epoch": 4.687976837158203e-05,
      "model_forward_time": 1.7008142471313477,
      "step": 49157
    },
    {
      "epoch": 4.687976837158203e-05,
      "step": 49157,
      "training_step_time": 6.960900068283081
    },
    {
      "epoch": 4.688072204589844e-05,
      "model_forward_time": 1.6980371475219727,
      "step": 49158
    },
    {
      "epoch": 4.688072204589844e-05,
      "step": 49158,
      "training_step_time": 6.982477426528931
    },
    {
      "epoch": 4.688167572021484e-05,
      "model_forward_time": 1.6994702816009521,
      "step": 49159
    },
    {
      "epoch": 4.688167572021484e-05,
      "step": 49159,
      "training_step_time": 6.971993923187256
    },
    {
      "epoch": 4.688262939453125e-05,
      "grad_norm": 0.09729472547769547,
      "learning_rate": 5.55130105363378e-06,
      "loss": 0.0597,
      "step": 49160
    },
    {
      "epoch": 4.688262939453125e-05,
      "model_forward_time": 1.700730323791504,
      "step": 49160
    },
    {
      "dynamics_loss_avg": 0.06013540271669626,
      "epoch": 4.688262939453125e-05,
      "step": 49160
    },
    {
      "action_loss_avg": 0.002638071571709588,
      "epoch": 4.688262939453125e-05,
      "step": 49160
    },
    {
      "epoch": 4.688262939453125e-05,
      "step": 49160,
      "training_step_time": 6.979717254638672
    },
    {
      "epoch": 4.688358306884766e-05,
      "model_forward_time": 1.699521541595459,
      "step": 49161
    },
    {
      "epoch": 4.688358306884766e-05,
      "step": 49161,
      "training_step_time": 8.416628122329712
    },
    {
      "epoch": 4.688453674316406e-05,
      "model_forward_time": 1.704211950302124,
      "step": 49162
    },
    {
      "epoch": 4.688453674316406e-05,
      "step": 49162,
      "training_step_time": 8.899356365203857
    },
    {
      "epoch": 4.6885490417480466e-05,
      "model_forward_time": 1.7014427185058594,
      "step": 49163
    },
    {
      "epoch": 4.6885490417480466e-05,
      "step": 49163,
      "training_step_time": 8.31899881362915
    },
    {
      "epoch": 4.6886444091796876e-05,
      "model_forward_time": 2.2873034477233887,
      "step": 49164
    },
    {
      "epoch": 4.6886444091796876e-05,
      "step": 49164,
      "training_step_time": 8.510274648666382
    },
    {
      "epoch": 4.688739776611328e-05,
      "model_forward_time": 1.7022557258605957,
      "step": 49165
    },
    {
      "epoch": 4.688739776611328e-05,
      "step": 49165,
      "training_step_time": 7.638094663619995
    },
    {
      "epoch": 4.6888351440429684e-05,
      "model_forward_time": 1.7027194499969482,
      "step": 49166
    },
    {
      "epoch": 4.6888351440429684e-05,
      "step": 49166,
      "training_step_time": 6.9513609409332275
    },
    {
      "epoch": 4.6889305114746095e-05,
      "model_forward_time": 1.7007880210876465,
      "step": 49167
    },
    {
      "epoch": 4.6889305114746095e-05,
      "step": 49167,
      "training_step_time": 6.966093063354492
    },
    {
      "epoch": 4.68902587890625e-05,
      "model_forward_time": 1.706347942352295,
      "step": 49168
    },
    {
      "epoch": 4.68902587890625e-05,
      "step": 49168,
      "training_step_time": 6.974455118179321
    },
    {
      "epoch": 4.68912124633789e-05,
      "model_forward_time": 1.7022068500518799,
      "step": 49169
    },
    {
      "epoch": 4.68912124633789e-05,
      "step": 49169,
      "training_step_time": 6.989208221435547
    },
    {
      "epoch": 4.6892166137695314e-05,
      "grad_norm": 0.12376841902732849,
      "learning_rate": 5.549657635312294e-06,
      "loss": 0.0625,
      "step": 49170
    },
    {
      "epoch": 4.6892166137695314e-05,
      "model_forward_time": 1.7060201168060303,
      "step": 49170
    },
    {
      "dynamics_loss_avg": 0.06016799099743366,
      "epoch": 4.6892166137695314e-05,
      "step": 49170
    },
    {
      "action_loss_avg": 0.0014926789561286568,
      "epoch": 4.6892166137695314e-05,
      "step": 49170
    },
    {
      "epoch": 4.6892166137695314e-05,
      "step": 49170,
      "training_step_time": 9.866567373275757
    },
    {
      "epoch": 4.689311981201172e-05,
      "model_forward_time": 1.714392900466919,
      "step": 49171
    },
    {
      "epoch": 4.689311981201172e-05,
      "step": 49171,
      "training_step_time": 7.885242938995361
    },
    {
      "epoch": 4.689407348632812e-05,
      "model_forward_time": 1.715057373046875,
      "step": 49172
    },
    {
      "epoch": 4.689407348632812e-05,
      "step": 49172,
      "training_step_time": 8.398096323013306
    },
    {
      "epoch": 4.689502716064453e-05,
      "model_forward_time": 2.320507049560547,
      "step": 49173
    },
    {
      "epoch": 4.689502716064453e-05,
      "step": 49173,
      "training_step_time": 7.7236573696136475
    },
    {
      "epoch": 4.689598083496094e-05,
      "model_forward_time": 1.722815990447998,
      "step": 49174
    },
    {
      "epoch": 4.689598083496094e-05,
      "step": 49174,
      "training_step_time": 7.789057970046997
    },
    {
      "epoch": 4.689693450927734e-05,
      "model_forward_time": 1.7192823886871338,
      "step": 49175
    },
    {
      "epoch": 4.689693450927734e-05,
      "step": 49175,
      "training_step_time": 6.910820007324219
    },
    {
      "epoch": 4.689788818359375e-05,
      "model_forward_time": 1.7142300605773926,
      "step": 49176
    },
    {
      "epoch": 4.689788818359375e-05,
      "step": 49176,
      "training_step_time": 6.9351747035980225
    },
    {
      "epoch": 4.6898841857910156e-05,
      "model_forward_time": 1.7147343158721924,
      "step": 49177
    },
    {
      "epoch": 4.6898841857910156e-05,
      "step": 49177,
      "training_step_time": 6.950932741165161
    },
    {
      "epoch": 4.689979553222656e-05,
      "model_forward_time": 1.719923973083496,
      "step": 49178
    },
    {
      "epoch": 4.689979553222656e-05,
      "step": 49178,
      "training_step_time": 6.993047475814819
    },
    {
      "epoch": 4.690074920654297e-05,
      "model_forward_time": 1.7193632125854492,
      "step": 49179
    },
    {
      "epoch": 4.690074920654297e-05,
      "step": 49179,
      "training_step_time": 8.675883054733276
    },
    {
      "epoch": 4.6901702880859375e-05,
      "grad_norm": 0.1602788269519806,
      "learning_rate": 5.5480141568810745e-06,
      "loss": 0.0567,
      "step": 49180
    },
    {
      "epoch": 4.6901702880859375e-05,
      "model_forward_time": 1.723503828048706,
      "step": 49180
    },
    {
      "dynamics_loss_avg": 0.050816225446760656,
      "epoch": 4.6901702880859375e-05,
      "step": 49180
    },
    {
      "action_loss_avg": 0.0019760163442697374,
      "epoch": 4.6901702880859375e-05,
      "step": 49180
    },
    {
      "epoch": 4.6901702880859375e-05,
      "step": 49180,
      "training_step_time": 7.8967907428741455
    },
    {
      "epoch": 4.690265655517578e-05,
      "model_forward_time": 1.7233178615570068,
      "step": 49181
    },
    {
      "epoch": 4.690265655517578e-05,
      "step": 49181,
      "training_step_time": 8.762048244476318
    },
    {
      "epoch": 4.690361022949219e-05,
      "model_forward_time": 2.2609081268310547,
      "step": 49182
    },
    {
      "epoch": 4.690361022949219e-05,
      "step": 49182,
      "training_step_time": 7.849481821060181
    },
    {
      "epoch": 4.6904563903808594e-05,
      "model_forward_time": 1.7728712558746338,
      "step": 49183
    },
    {
      "epoch": 4.6904563903808594e-05,
      "step": 49183,
      "training_step_time": 7.925203800201416
    },
    {
      "epoch": 4.6905517578125e-05,
      "model_forward_time": 1.8178205490112305,
      "step": 49184
    },
    {
      "epoch": 4.6905517578125e-05,
      "step": 49184,
      "training_step_time": 7.45904541015625
    },
    {
      "epoch": 4.690647125244141e-05,
      "model_forward_time": 1.8192031383514404,
      "step": 49185
    },
    {
      "epoch": 4.690647125244141e-05,
      "step": 49185,
      "training_step_time": 7.046048402786255
    },
    {
      "epoch": 4.690742492675781e-05,
      "model_forward_time": 1.8205554485321045,
      "step": 49186
    },
    {
      "epoch": 4.690742492675781e-05,
      "step": 49186,
      "training_step_time": 6.9298951625823975
    },
    {
      "epoch": 4.6908378601074216e-05,
      "model_forward_time": 1.8199317455291748,
      "step": 49187
    },
    {
      "epoch": 4.6908378601074216e-05,
      "step": 49187,
      "training_step_time": 6.963089942932129
    },
    {
      "epoch": 4.690933227539063e-05,
      "model_forward_time": 1.8125648498535156,
      "step": 49188
    },
    {
      "epoch": 4.690933227539063e-05,
      "step": 49188,
      "training_step_time": 7.861940860748291
    },
    {
      "epoch": 4.691028594970703e-05,
      "model_forward_time": 1.8184773921966553,
      "step": 49189
    },
    {
      "epoch": 4.691028594970703e-05,
      "step": 49189,
      "training_step_time": 8.69295334815979
    },
    {
      "epoch": 4.6911239624023435e-05,
      "grad_norm": 0.0977383702993393,
      "learning_rate": 5.546370618519849e-06,
      "loss": 0.0587,
      "step": 49190
    },
    {
      "epoch": 4.6911239624023435e-05,
      "model_forward_time": 1.8225233554840088,
      "step": 49190
    },
    {
      "dynamics_loss_avg": 0.054987074248492716,
      "epoch": 4.6911239624023435e-05,
      "step": 49190
    },
    {
      "action_loss_avg": 0.0017666990650468505,
      "epoch": 4.6911239624023435e-05,
      "step": 49190
    },
    {
      "epoch": 4.6911239624023435e-05,
      "step": 49190,
      "training_step_time": 9.469544172286987
    },
    {
      "epoch": 4.6912193298339846e-05,
      "model_forward_time": 1.815889835357666,
      "step": 49191
    },
    {
      "epoch": 4.6912193298339846e-05,
      "step": 49191,
      "training_step_time": 8.695215702056885
    },
    {
      "epoch": 4.691314697265625e-05,
      "model_forward_time": 2.3464243412017822,
      "step": 49192
    },
    {
      "epoch": 4.691314697265625e-05,
      "step": 49192,
      "training_step_time": 8.59872317314148
    },
    {
      "epoch": 4.6914100646972654e-05,
      "model_forward_time": 1.8032803535461426,
      "step": 49193
    },
    {
      "epoch": 4.6914100646972654e-05,
      "step": 49193,
      "training_step_time": 7.832540988922119
    },
    {
      "epoch": 4.6915054321289065e-05,
      "model_forward_time": 1.8073437213897705,
      "step": 49194
    },
    {
      "epoch": 4.6915054321289065e-05,
      "step": 49194,
      "training_step_time": 6.929999113082886
    },
    {
      "epoch": 4.691600799560547e-05,
      "model_forward_time": 1.8131799697875977,
      "step": 49195
    },
    {
      "epoch": 4.691600799560547e-05,
      "step": 49195,
      "training_step_time": 6.92199444770813
    },
    {
      "epoch": 4.691696166992187e-05,
      "model_forward_time": 1.8096225261688232,
      "step": 49196
    },
    {
      "epoch": 4.691696166992187e-05,
      "step": 49196,
      "training_step_time": 6.961353778839111
    },
    {
      "epoch": 4.6917915344238284e-05,
      "model_forward_time": 1.8073718547821045,
      "step": 49197
    },
    {
      "epoch": 4.6917915344238284e-05,
      "step": 49197,
      "training_step_time": 8.758568048477173
    },
    {
      "epoch": 4.691886901855469e-05,
      "model_forward_time": 1.828087568283081,
      "step": 49198
    },
    {
      "epoch": 4.691886901855469e-05,
      "step": 49198,
      "training_step_time": 8.799592971801758
    },
    {
      "epoch": 4.691982269287109e-05,
      "model_forward_time": 1.8188755512237549,
      "step": 49199
    },
    {
      "epoch": 4.691982269287109e-05,
      "step": 49199,
      "training_step_time": 8.33872103691101
    },
    {
      "epoch": 4.69207763671875e-05,
      "grad_norm": 0.08970190584659576,
      "learning_rate": 5.544727020408358e-06,
      "loss": 0.0597,
      "step": 49200
    },
    {
      "epoch": 4.69207763671875e-05,
      "model_forward_time": 1.8224279880523682,
      "step": 49200
    },
    {
      "dynamics_loss_avg": 0.04636334497481585,
      "epoch": 4.69207763671875e-05,
      "step": 49200
    },
    {
      "action_loss_avg": 0.0013551284064305946,
      "epoch": 4.69207763671875e-05,
      "step": 49200
    },
    {
      "epoch": 4.69207763671875e-05,
      "step": 49200,
      "training_step_time": 10.074689388275146
    },
    {
      "epoch": 4.692173004150391e-05,
      "model_forward_time": 2.3485288619995117,
      "step": 49201
    },
    {
      "epoch": 4.692173004150391e-05,
      "step": 49201,
      "training_step_time": 8.565944194793701
    },
    {
      "epoch": 4.692268371582031e-05,
      "model_forward_time": 1.7230236530303955,
      "step": 49202
    },
    {
      "epoch": 4.692268371582031e-05,
      "step": 49202,
      "training_step_time": 7.925945997238159
    },
    {
      "epoch": 4.692363739013672e-05,
      "model_forward_time": 1.7103006839752197,
      "step": 49203
    },
    {
      "epoch": 4.692363739013672e-05,
      "step": 49203,
      "training_step_time": 6.991618633270264
    },
    {
      "epoch": 4.6924591064453126e-05,
      "model_forward_time": 1.7143163681030273,
      "step": 49204
    },
    {
      "epoch": 4.6924591064453126e-05,
      "step": 49204,
      "training_step_time": 7.036296606063843
    },
    {
      "epoch": 4.692554473876953e-05,
      "model_forward_time": 1.7178080081939697,
      "step": 49205
    },
    {
      "epoch": 4.692554473876953e-05,
      "step": 49205,
      "training_step_time": 6.988483428955078
    },
    {
      "epoch": 4.692649841308594e-05,
      "model_forward_time": 1.725235939025879,
      "step": 49206
    },
    {
      "epoch": 4.692649841308594e-05,
      "step": 49206,
      "training_step_time": 8.61976933479309
    },
    {
      "epoch": 4.6927452087402344e-05,
      "model_forward_time": 1.7139413356781006,
      "step": 49207
    },
    {
      "epoch": 4.6927452087402344e-05,
      "step": 49207,
      "training_step_time": 8.887128829956055
    },
    {
      "epoch": 4.692840576171875e-05,
      "model_forward_time": 1.7173829078674316,
      "step": 49208
    },
    {
      "epoch": 4.692840576171875e-05,
      "step": 49208,
      "training_step_time": 8.470178604125977
    },
    {
      "epoch": 4.692935943603516e-05,
      "model_forward_time": 1.7137072086334229,
      "step": 49209
    },
    {
      "epoch": 4.692935943603516e-05,
      "step": 49209,
      "training_step_time": 8.152915477752686
    },
    {
      "epoch": 4.693031311035156e-05,
      "grad_norm": 0.09697956591844559,
      "learning_rate": 5.543083362726337e-06,
      "loss": 0.0574,
      "step": 49210
    },
    {
      "epoch": 4.693031311035156e-05,
      "model_forward_time": 2.2604072093963623,
      "step": 49210
    },
    {
      "dynamics_loss_avg": 0.04720928370952606,
      "epoch": 4.693031311035156e-05,
      "step": 49210
    },
    {
      "action_loss_avg": 0.0012886483629699797,
      "epoch": 4.693031311035156e-05,
      "step": 49210
    },
    {
      "epoch": 4.693031311035156e-05,
      "step": 49210,
      "training_step_time": 8.6239755153656
    },
    {
      "epoch": 4.693126678466797e-05,
      "model_forward_time": 1.7157025337219238,
      "step": 49211
    },
    {
      "epoch": 4.693126678466797e-05,
      "step": 49211,
      "training_step_time": 7.859827518463135
    },
    {
      "epoch": 4.693222045898438e-05,
      "model_forward_time": 1.7141847610473633,
      "step": 49212
    },
    {
      "epoch": 4.693222045898438e-05,
      "step": 49212,
      "training_step_time": 6.822323560714722
    },
    {
      "epoch": 4.693317413330078e-05,
      "model_forward_time": 1.7147767543792725,
      "step": 49213
    },
    {
      "epoch": 4.693317413330078e-05,
      "step": 49213,
      "training_step_time": 6.9276955127716064
    },
    {
      "epoch": 4.6934127807617186e-05,
      "model_forward_time": 1.7127647399902344,
      "step": 49214
    },
    {
      "epoch": 4.6934127807617186e-05,
      "step": 49214,
      "training_step_time": 6.971437215805054
    },
    {
      "epoch": 4.69350814819336e-05,
      "model_forward_time": 1.7125060558319092,
      "step": 49215
    },
    {
      "epoch": 4.69350814819336e-05,
      "step": 49215,
      "training_step_time": 7.746974468231201
    },
    {
      "epoch": 4.693603515625e-05,
      "model_forward_time": 1.715148687362671,
      "step": 49216
    },
    {
      "epoch": 4.693603515625e-05,
      "step": 49216,
      "training_step_time": 8.394122123718262
    },
    {
      "epoch": 4.6936988830566405e-05,
      "model_forward_time": 1.717059850692749,
      "step": 49217
    },
    {
      "epoch": 4.6936988830566405e-05,
      "step": 49217,
      "training_step_time": 7.796598672866821
    },
    {
      "epoch": 4.6937942504882816e-05,
      "model_forward_time": 1.714047908782959,
      "step": 49218
    },
    {
      "epoch": 4.6937942504882816e-05,
      "step": 49218,
      "training_step_time": 8.877790689468384
    },
    {
      "epoch": 4.693889617919922e-05,
      "model_forward_time": 2.3026838302612305,
      "step": 49219
    },
    {
      "epoch": 4.693889617919922e-05,
      "step": 49219,
      "training_step_time": 8.40992546081543
    },
    {
      "epoch": 4.6939849853515624e-05,
      "grad_norm": 0.14239995181560516,
      "learning_rate": 5.541439645653536e-06,
      "loss": 0.0616,
      "step": 49220
    },
    {
      "epoch": 4.6939849853515624e-05,
      "model_forward_time": 1.702796459197998,
      "step": 49220
    },
    {
      "dynamics_loss_avg": 0.03958521131426096,
      "epoch": 4.6939849853515624e-05,
      "step": 49220
    },
    {
      "action_loss_avg": 0.0009760412271134555,
      "epoch": 4.6939849853515624e-05,
      "step": 49220
    },
    {
      "epoch": 4.6939849853515624e-05,
      "step": 49220,
      "training_step_time": 7.727595090866089
    },
    {
      "epoch": 4.694080352783203e-05,
      "model_forward_time": 1.7088148593902588,
      "step": 49221
    },
    {
      "epoch": 4.694080352783203e-05,
      "step": 49221,
      "training_step_time": 7.24866795539856
    },
    {
      "epoch": 4.694175720214844e-05,
      "model_forward_time": 1.7142939567565918,
      "step": 49222
    },
    {
      "epoch": 4.694175720214844e-05,
      "step": 49222,
      "training_step_time": 6.990741491317749
    },
    {
      "epoch": 4.694271087646484e-05,
      "model_forward_time": 1.7088274955749512,
      "step": 49223
    },
    {
      "epoch": 4.694271087646484e-05,
      "step": 49223,
      "training_step_time": 6.929007291793823
    },
    {
      "epoch": 4.694366455078125e-05,
      "model_forward_time": 1.7075164318084717,
      "step": 49224
    },
    {
      "epoch": 4.694366455078125e-05,
      "step": 49224,
      "training_step_time": 7.810838222503662
    },
    {
      "epoch": 4.694461822509766e-05,
      "model_forward_time": 1.7179479598999023,
      "step": 49225
    },
    {
      "epoch": 4.694461822509766e-05,
      "step": 49225,
      "training_step_time": 7.88573431968689
    },
    {
      "epoch": 4.694557189941406e-05,
      "model_forward_time": 1.7087745666503906,
      "step": 49226
    },
    {
      "epoch": 4.694557189941406e-05,
      "step": 49226,
      "training_step_time": 8.629862070083618
    },
    {
      "epoch": 4.6946525573730466e-05,
      "model_forward_time": 1.7084221839904785,
      "step": 49227
    },
    {
      "epoch": 4.6946525573730466e-05,
      "step": 49227,
      "training_step_time": 8.572075605392456
    },
    {
      "epoch": 4.6947479248046876e-05,
      "model_forward_time": 2.2913432121276855,
      "step": 49228
    },
    {
      "epoch": 4.6947479248046876e-05,
      "step": 49228,
      "training_step_time": 9.469521760940552
    },
    {
      "epoch": 4.694843292236328e-05,
      "model_forward_time": 1.7057111263275146,
      "step": 49229
    },
    {
      "epoch": 4.694843292236328e-05,
      "step": 49229,
      "training_step_time": 7.779303312301636
    },
    {
      "epoch": 4.6949386596679685e-05,
      "grad_norm": 0.16112598776817322,
      "learning_rate": 5.53979586936971e-06,
      "loss": 0.0584,
      "step": 49230
    },
    {
      "epoch": 4.6949386596679685e-05,
      "model_forward_time": 1.702399492263794,
      "step": 49230
    },
    {
      "dynamics_loss_avg": 0.046018242835998535,
      "epoch": 4.6949386596679685e-05,
      "step": 49230
    },
    {
      "action_loss_avg": 0.0018545822298619897,
      "epoch": 4.6949386596679685e-05,
      "step": 49230
    },
    {
      "epoch": 4.6949386596679685e-05,
      "step": 49230,
      "training_step_time": 7.6529457569122314
    },
    {
      "epoch": 4.6950340270996095e-05,
      "model_forward_time": 1.7161102294921875,
      "step": 49231
    },
    {
      "epoch": 4.6950340270996095e-05,
      "step": 49231,
      "training_step_time": 6.945635795593262
    },
    {
      "epoch": 4.69512939453125e-05,
      "model_forward_time": 1.7201805114746094,
      "step": 49232
    },
    {
      "epoch": 4.69512939453125e-05,
      "step": 49232,
      "training_step_time": 6.982123613357544
    },
    {
      "epoch": 4.69522476196289e-05,
      "model_forward_time": 1.7136530876159668,
      "step": 49233
    },
    {
      "epoch": 4.69522476196289e-05,
      "step": 49233,
      "training_step_time": 7.792853593826294
    },
    {
      "epoch": 4.6953201293945314e-05,
      "model_forward_time": 1.721808910369873,
      "step": 49234
    },
    {
      "epoch": 4.6953201293945314e-05,
      "step": 49234,
      "training_step_time": 7.820806503295898
    },
    {
      "epoch": 4.695415496826172e-05,
      "model_forward_time": 1.7275497913360596,
      "step": 49235
    },
    {
      "epoch": 4.695415496826172e-05,
      "step": 49235,
      "training_step_time": 10.46280813217163
    },
    {
      "epoch": 4.695510864257812e-05,
      "model_forward_time": 1.6986157894134521,
      "step": 49236
    },
    {
      "epoch": 4.695510864257812e-05,
      "step": 49236,
      "training_step_time": 7.8556928634643555
    },
    {
      "epoch": 4.695606231689453e-05,
      "model_forward_time": 1.7181930541992188,
      "step": 49237
    },
    {
      "epoch": 4.695606231689453e-05,
      "step": 49237,
      "training_step_time": 11.208641767501831
    },
    {
      "epoch": 4.695701599121094e-05,
      "model_forward_time": 1.7177090644836426,
      "step": 49238
    },
    {
      "epoch": 4.695701599121094e-05,
      "step": 49238,
      "training_step_time": 8.914841651916504
    },
    {
      "epoch": 4.695796966552734e-05,
      "model_forward_time": 1.7153258323669434,
      "step": 49239
    },
    {
      "epoch": 4.695796966552734e-05,
      "step": 49239,
      "training_step_time": 7.689942836761475
    },
    {
      "epoch": 4.695892333984375e-05,
      "grad_norm": 0.12051030248403549,
      "learning_rate": 5.5381520340546176e-06,
      "loss": 0.0597,
      "step": 49240
    },
    {
      "epoch": 4.695892333984375e-05,
      "model_forward_time": 1.7098727226257324,
      "step": 49240
    },
    {
      "dynamics_loss_avg": 0.06049739271402359,
      "epoch": 4.695892333984375e-05,
      "step": 49240
    },
    {
      "action_loss_avg": 0.0017028586065862329,
      "epoch": 4.695892333984375e-05,
      "step": 49240
    },
    {
      "epoch": 4.695892333984375e-05,
      "step": 49240,
      "training_step_time": 6.907986879348755
    },
    {
      "epoch": 4.6959877014160156e-05,
      "model_forward_time": 1.333200216293335,
      "step": 49241
    },
    {
      "epoch": 4.6959877014160156e-05,
      "step": 49241,
      "training_step_time": 7.05063009262085
    },
    {
      "epoch": 4.696083068847656e-05,
      "model_forward_time": 1.7233295440673828,
      "step": 49242
    },
    {
      "epoch": 4.696083068847656e-05,
      "step": 49242,
      "training_step_time": 7.808718681335449
    },
    {
      "epoch": 4.696178436279297e-05,
      "model_forward_time": 1.7148075103759766,
      "step": 49243
    },
    {
      "epoch": 4.696178436279297e-05,
      "step": 49243,
      "training_step_time": 7.90226149559021
    },
    {
      "epoch": 4.6962738037109375e-05,
      "model_forward_time": 1.7147295475006104,
      "step": 49244
    },
    {
      "epoch": 4.6962738037109375e-05,
      "step": 49244,
      "training_step_time": 7.911491632461548
    },
    {
      "epoch": 4.696369171142578e-05,
      "model_forward_time": 1.7197949886322021,
      "step": 49245
    },
    {
      "epoch": 4.696369171142578e-05,
      "step": 49245,
      "training_step_time": 8.721258640289307
    },
    {
      "epoch": 4.696464538574219e-05,
      "model_forward_time": 1.7197632789611816,
      "step": 49246
    },
    {
      "epoch": 4.696464538574219e-05,
      "step": 49246,
      "training_step_time": 8.458519458770752
    },
    {
      "epoch": 4.6965599060058594e-05,
      "model_forward_time": 2.267003059387207,
      "step": 49247
    },
    {
      "epoch": 4.6965599060058594e-05,
      "step": 49247,
      "training_step_time": 7.870283365249634
    },
    {
      "epoch": 4.6966552734375e-05,
      "model_forward_time": 1.7149744033813477,
      "step": 49248
    },
    {
      "epoch": 4.6966552734375e-05,
      "step": 49248,
      "training_step_time": 7.779761791229248
    },
    {
      "epoch": 4.696750640869141e-05,
      "model_forward_time": 1.7117092609405518,
      "step": 49249
    },
    {
      "epoch": 4.696750640869141e-05,
      "step": 49249,
      "training_step_time": 6.966302871704102
    },
    {
      "epoch": 4.696846008300781e-05,
      "grad_norm": 0.10447292774915695,
      "learning_rate": 5.536508139888028e-06,
      "loss": 0.0594,
      "step": 49250
    },
    {
      "epoch": 4.696846008300781e-05,
      "model_forward_time": 1.715747594833374,
      "step": 49250
    },
    {
      "dynamics_loss_avg": 0.0565833305940032,
      "epoch": 4.696846008300781e-05,
      "step": 49250
    },
    {
      "action_loss_avg": 0.0019297078484669327,
      "epoch": 4.696846008300781e-05,
      "step": 49250
    },
    {
      "epoch": 4.696846008300781e-05,
      "step": 49250,
      "training_step_time": 6.915688514709473
    },
    {
      "epoch": 4.6969413757324217e-05,
      "model_forward_time": 1.7177672386169434,
      "step": 49251
    },
    {
      "epoch": 4.6969413757324217e-05,
      "step": 49251,
      "training_step_time": 8.326957941055298
    },
    {
      "epoch": 4.697036743164063e-05,
      "model_forward_time": 1.7196314334869385,
      "step": 49252
    },
    {
      "epoch": 4.697036743164063e-05,
      "step": 49252,
      "training_step_time": 7.819528579711914
    },
    {
      "epoch": 4.697132110595703e-05,
      "model_forward_time": 1.7154295444488525,
      "step": 49253
    },
    {
      "epoch": 4.697132110595703e-05,
      "step": 49253,
      "training_step_time": 7.860679864883423
    },
    {
      "epoch": 4.6972274780273435e-05,
      "model_forward_time": 1.7190139293670654,
      "step": 49254
    },
    {
      "epoch": 4.6972274780273435e-05,
      "step": 49254,
      "training_step_time": 7.776710510253906
    },
    {
      "epoch": 4.6973228454589846e-05,
      "model_forward_time": 1.7133615016937256,
      "step": 49255
    },
    {
      "epoch": 4.6973228454589846e-05,
      "step": 49255,
      "training_step_time": 8.915613889694214
    },
    {
      "epoch": 4.697418212890625e-05,
      "model_forward_time": 2.294806957244873,
      "step": 49256
    },
    {
      "epoch": 4.697418212890625e-05,
      "step": 49256,
      "training_step_time": 8.89625072479248
    },
    {
      "epoch": 4.6975135803222654e-05,
      "model_forward_time": 1.714379072189331,
      "step": 49257
    },
    {
      "epoch": 4.6975135803222654e-05,
      "step": 49257,
      "training_step_time": 7.831308126449585
    },
    {
      "epoch": 4.6976089477539065e-05,
      "model_forward_time": 1.7129802703857422,
      "step": 49258
    },
    {
      "epoch": 4.6976089477539065e-05,
      "step": 49258,
      "training_step_time": 7.432265996932983
    },
    {
      "epoch": 4.697704315185547e-05,
      "model_forward_time": 1.7111351490020752,
      "step": 49259
    },
    {
      "epoch": 4.697704315185547e-05,
      "step": 49259,
      "training_step_time": 7.020411491394043
    },
    {
      "epoch": 4.697799682617187e-05,
      "grad_norm": 0.14502738416194916,
      "learning_rate": 5.5348641870497165e-06,
      "loss": 0.0586,
      "step": 49260
    },
    {
      "epoch": 4.697799682617187e-05,
      "model_forward_time": 1.7185969352722168,
      "step": 49260
    },
    {
      "dynamics_loss_avg": 0.057644883170723915,
      "epoch": 4.697799682617187e-05,
      "step": 49260
    },
    {
      "action_loss_avg": 0.0020061812159838156,
      "epoch": 4.697799682617187e-05,
      "step": 49260
    },
    {
      "epoch": 4.697799682617187e-05,
      "step": 49260,
      "training_step_time": 7.763868808746338
    },
    {
      "epoch": 4.6978950500488284e-05,
      "model_forward_time": 1.7140166759490967,
      "step": 49261
    },
    {
      "epoch": 4.6978950500488284e-05,
      "step": 49261,
      "training_step_time": 8.60962438583374
    },
    {
      "epoch": 4.697990417480469e-05,
      "model_forward_time": 1.7161428928375244,
      "step": 49262
    },
    {
      "epoch": 4.697990417480469e-05,
      "step": 49262,
      "training_step_time": 9.783206701278687
    },
    {
      "epoch": 4.698085784912109e-05,
      "model_forward_time": 1.7203516960144043,
      "step": 49263
    },
    {
      "epoch": 4.698085784912109e-05,
      "step": 49263,
      "training_step_time": 8.884244680404663
    },
    {
      "epoch": 4.69818115234375e-05,
      "model_forward_time": 1.710993766784668,
      "step": 49264
    },
    {
      "epoch": 4.69818115234375e-05,
      "step": 49264,
      "training_step_time": 8.33073115348816
    },
    {
      "epoch": 4.698276519775391e-05,
      "model_forward_time": 2.315028667449951,
      "step": 49265
    },
    {
      "epoch": 4.698276519775391e-05,
      "step": 49265,
      "training_step_time": 7.838284015655518
    },
    {
      "epoch": 4.698371887207031e-05,
      "model_forward_time": 1.7093746662139893,
      "step": 49266
    },
    {
      "epoch": 4.698371887207031e-05,
      "step": 49266,
      "training_step_time": 7.767697095870972
    },
    {
      "epoch": 4.698467254638672e-05,
      "model_forward_time": 1.715423822402954,
      "step": 49267
    },
    {
      "epoch": 4.698467254638672e-05,
      "step": 49267,
      "training_step_time": 7.645973205566406
    },
    {
      "epoch": 4.6985626220703126e-05,
      "model_forward_time": 1.723728895187378,
      "step": 49268
    },
    {
      "epoch": 4.6985626220703126e-05,
      "step": 49268,
      "training_step_time": 6.934443235397339
    },
    {
      "epoch": 4.698657989501953e-05,
      "model_forward_time": 1.7087757587432861,
      "step": 49269
    },
    {
      "epoch": 4.698657989501953e-05,
      "step": 49269,
      "training_step_time": 7.726311683654785
    },
    {
      "epoch": 4.698753356933594e-05,
      "grad_norm": 0.09979972243309021,
      "learning_rate": 5.533220175719461e-06,
      "loss": 0.0606,
      "step": 49270
    },
    {
      "epoch": 4.698753356933594e-05,
      "model_forward_time": 1.706303596496582,
      "step": 49270
    },
    {
      "dynamics_loss_avg": 0.048418745212256906,
      "epoch": 4.698753356933594e-05,
      "step": 49270
    },
    {
      "action_loss_avg": 0.0009920793818309903,
      "epoch": 4.698753356933594e-05,
      "step": 49270
    },
    {
      "epoch": 4.698753356933594e-05,
      "step": 49270,
      "training_step_time": 7.697096109390259
    },
    {
      "epoch": 4.6988487243652345e-05,
      "model_forward_time": 1.7170648574829102,
      "step": 49271
    },
    {
      "epoch": 4.6988487243652345e-05,
      "step": 49271,
      "training_step_time": 7.957710266113281
    },
    {
      "epoch": 4.698944091796875e-05,
      "model_forward_time": 1.7194271087646484,
      "step": 49272
    },
    {
      "epoch": 4.698944091796875e-05,
      "step": 49272,
      "training_step_time": 8.691148042678833
    },
    {
      "epoch": 4.699039459228516e-05,
      "model_forward_time": 1.7177894115447998,
      "step": 49273
    },
    {
      "epoch": 4.699039459228516e-05,
      "step": 49273,
      "training_step_time": 9.755549192428589
    },
    {
      "epoch": 4.699134826660156e-05,
      "model_forward_time": 2.288484573364258,
      "step": 49274
    },
    {
      "epoch": 4.699134826660156e-05,
      "step": 49274,
      "training_step_time": 8.933987379074097
    },
    {
      "epoch": 4.699230194091797e-05,
      "model_forward_time": 1.7248787879943848,
      "step": 49275
    },
    {
      "epoch": 4.699230194091797e-05,
      "step": 49275,
      "training_step_time": 8.93746829032898
    },
    {
      "epoch": 4.699325561523438e-05,
      "model_forward_time": 1.7128562927246094,
      "step": 49276
    },
    {
      "epoch": 4.699325561523438e-05,
      "step": 49276,
      "training_step_time": 7.8466477394104
    },
    {
      "epoch": 4.699420928955078e-05,
      "model_forward_time": 1.7147376537322998,
      "step": 49277
    },
    {
      "epoch": 4.699420928955078e-05,
      "step": 49277,
      "training_step_time": 7.072639465332031
    },
    {
      "epoch": 4.6995162963867186e-05,
      "model_forward_time": 1.718459129333496,
      "step": 49278
    },
    {
      "epoch": 4.6995162963867186e-05,
      "step": 49278,
      "training_step_time": 7.759838104248047
    },
    {
      "epoch": 4.69961166381836e-05,
      "model_forward_time": 1.7215242385864258,
      "step": 49279
    },
    {
      "epoch": 4.69961166381836e-05,
      "step": 49279,
      "training_step_time": 7.794515132904053
    },
    {
      "epoch": 4.69970703125e-05,
      "grad_norm": 0.11670099943876266,
      "learning_rate": 5.531576106077049e-06,
      "loss": 0.06,
      "step": 49280
    },
    {
      "epoch": 4.69970703125e-05,
      "model_forward_time": 1.7246863842010498,
      "step": 49280
    },
    {
      "dynamics_loss_avg": 0.04865267872810364,
      "epoch": 4.69970703125e-05,
      "step": 49280
    },
    {
      "action_loss_avg": 0.0019114681068458594,
      "epoch": 4.69970703125e-05,
      "step": 49280
    },
    {
      "epoch": 4.69970703125e-05,
      "step": 49280,
      "training_step_time": 7.8522114753723145
    },
    {
      "epoch": 4.6998023986816405e-05,
      "model_forward_time": 1.709920883178711,
      "step": 49281
    },
    {
      "epoch": 4.6998023986816405e-05,
      "step": 49281,
      "training_step_time": 7.762632131576538
    },
    {
      "epoch": 4.699897766113281e-05,
      "model_forward_time": 1.7140560150146484,
      "step": 49282
    },
    {
      "epoch": 4.699897766113281e-05,
      "step": 49282,
      "training_step_time": 8.266764402389526
    },
    {
      "epoch": 4.699993133544922e-05,
      "model_forward_time": 2.2956340312957764,
      "step": 49283
    },
    {
      "epoch": 4.699993133544922e-05,
      "step": 49283,
      "training_step_time": 9.856320858001709
    },
    {
      "epoch": 4.7000885009765624e-05,
      "model_forward_time": 1.3356826305389404,
      "step": 49284
    },
    {
      "epoch": 4.7000885009765624e-05,
      "step": 49284,
      "training_step_time": 7.818609237670898
    },
    {
      "epoch": 4.700183868408203e-05,
      "model_forward_time": 1.719569206237793,
      "step": 49285
    },
    {
      "epoch": 4.700183868408203e-05,
      "step": 49285,
      "training_step_time": 7.794313430786133
    },
    {
      "epoch": 4.700279235839844e-05,
      "model_forward_time": 1.7167210578918457,
      "step": 49286
    },
    {
      "epoch": 4.700279235839844e-05,
      "step": 49286,
      "training_step_time": 7.062504529953003
    },
    {
      "epoch": 4.700374603271484e-05,
      "model_forward_time": 1.7128243446350098,
      "step": 49287
    },
    {
      "epoch": 4.700374603271484e-05,
      "step": 49287,
      "training_step_time": 8.624109029769897
    },
    {
      "epoch": 4.700469970703125e-05,
      "model_forward_time": 1.7231338024139404,
      "step": 49288
    },
    {
      "epoch": 4.700469970703125e-05,
      "step": 49288,
      "training_step_time": 7.754412651062012
    },
    {
      "epoch": 4.700565338134766e-05,
      "model_forward_time": 1.7176439762115479,
      "step": 49289
    },
    {
      "epoch": 4.700565338134766e-05,
      "step": 49289,
      "training_step_time": 7.66907811164856
    },
    {
      "epoch": 4.700660705566406e-05,
      "grad_norm": 0.15895353257656097,
      "learning_rate": 5.529931978302272e-06,
      "loss": 0.0566,
      "step": 49290
    },
    {
      "epoch": 4.700660705566406e-05,
      "model_forward_time": 1.7055892944335938,
      "step": 49290
    },
    {
      "dynamics_loss_avg": 0.04608513033017516,
      "epoch": 4.700660705566406e-05,
      "step": 49290
    },
    {
      "action_loss_avg": 0.0017931503389263526,
      "epoch": 4.700660705566406e-05,
      "step": 49290
    },
    {
      "epoch": 4.700660705566406e-05,
      "step": 49290,
      "training_step_time": 8.513261556625366
    },
    {
      "epoch": 4.7007560729980466e-05,
      "model_forward_time": 1.7157611846923828,
      "step": 49291
    },
    {
      "epoch": 4.7007560729980466e-05,
      "step": 49291,
      "training_step_time": 9.30040693283081
    },
    {
      "epoch": 4.7008514404296877e-05,
      "model_forward_time": 1.7154948711395264,
      "step": 49292
    },
    {
      "epoch": 4.7008514404296877e-05,
      "step": 49292,
      "training_step_time": 8.862463474273682
    },
    {
      "epoch": 4.700946807861328e-05,
      "model_forward_time": 2.2605133056640625,
      "step": 49293
    },
    {
      "epoch": 4.700946807861328e-05,
      "step": 49293,
      "training_step_time": 7.91213321685791
    },
    {
      "epoch": 4.7010421752929685e-05,
      "model_forward_time": 1.7144691944122314,
      "step": 49294
    },
    {
      "epoch": 4.7010421752929685e-05,
      "step": 49294,
      "training_step_time": 8.370815992355347
    },
    {
      "epoch": 4.7011375427246095e-05,
      "model_forward_time": 1.7128949165344238,
      "step": 49295
    },
    {
      "epoch": 4.7011375427246095e-05,
      "step": 49295,
      "training_step_time": 7.720354795455933
    },
    {
      "epoch": 4.70123291015625e-05,
      "model_forward_time": 1.703862190246582,
      "step": 49296
    },
    {
      "epoch": 4.70123291015625e-05,
      "step": 49296,
      "training_step_time": 7.78442907333374
    },
    {
      "epoch": 4.7013282775878903e-05,
      "model_forward_time": 1.7123475074768066,
      "step": 49297
    },
    {
      "epoch": 4.7013282775878903e-05,
      "step": 49297,
      "training_step_time": 7.666289806365967
    },
    {
      "epoch": 4.7014236450195314e-05,
      "model_forward_time": 1.7122344970703125,
      "step": 49298
    },
    {
      "epoch": 4.7014236450195314e-05,
      "step": 49298,
      "training_step_time": 8.091743230819702
    },
    {
      "epoch": 4.701519012451172e-05,
      "model_forward_time": 1.7010447978973389,
      "step": 49299
    },
    {
      "epoch": 4.701519012451172e-05,
      "step": 49299,
      "training_step_time": 7.846064567565918
    },
    {
      "epoch": 4.701614379882812e-05,
      "grad_norm": 0.14670182764530182,
      "learning_rate": 5.528287792574932e-06,
      "loss": 0.0573,
      "step": 49300
    },
    {
      "epoch": 4.701614379882812e-05,
      "model_forward_time": 1.7020165920257568,
      "step": 49300
    },
    {
      "dynamics_loss_avg": 0.05658695846796036,
      "epoch": 4.701614379882812e-05,
      "step": 49300
    },
    {
      "action_loss_avg": 0.001706316031049937,
      "epoch": 4.701614379882812e-05,
      "step": 49300
    },
    {
      "epoch": 4.701614379882812e-05,
      "step": 49300,
      "training_step_time": 9.38816499710083
    },
    {
      "epoch": 4.701709747314453e-05,
      "model_forward_time": 1.7092795372009277,
      "step": 49301
    },
    {
      "epoch": 4.701709747314453e-05,
      "step": 49301,
      "training_step_time": 8.671361923217773
    },
    {
      "epoch": 4.701805114746094e-05,
      "model_forward_time": 2.3247406482696533,
      "step": 49302
    },
    {
      "epoch": 4.701805114746094e-05,
      "step": 49302,
      "training_step_time": 7.9260640144348145
    },
    {
      "epoch": 4.701900482177734e-05,
      "model_forward_time": 1.7081482410430908,
      "step": 49303
    },
    {
      "epoch": 4.701900482177734e-05,
      "step": 49303,
      "training_step_time": 7.663153648376465
    },
    {
      "epoch": 4.701995849609375e-05,
      "model_forward_time": 1.7023735046386719,
      "step": 49304
    },
    {
      "epoch": 4.701995849609375e-05,
      "step": 49304,
      "training_step_time": 7.756157875061035
    },
    {
      "epoch": 4.7020912170410156e-05,
      "model_forward_time": 1.7127630710601807,
      "step": 49305
    },
    {
      "epoch": 4.7020912170410156e-05,
      "step": 49305,
      "training_step_time": 7.764425277709961
    },
    {
      "epoch": 4.702186584472656e-05,
      "model_forward_time": 1.7111456394195557,
      "step": 49306
    },
    {
      "epoch": 4.702186584472656e-05,
      "step": 49306,
      "training_step_time": 7.568096160888672
    },
    {
      "epoch": 4.702281951904297e-05,
      "model_forward_time": 1.6984686851501465,
      "step": 49307
    },
    {
      "epoch": 4.702281951904297e-05,
      "step": 49307,
      "training_step_time": 7.75478458404541
    },
    {
      "epoch": 4.7023773193359375e-05,
      "model_forward_time": 1.7067649364471436,
      "step": 49308
    },
    {
      "epoch": 4.7023773193359375e-05,
      "step": 49308,
      "training_step_time": 7.811999082565308
    },
    {
      "epoch": 4.702472686767578e-05,
      "model_forward_time": 1.7194461822509766,
      "step": 49309
    },
    {
      "epoch": 4.702472686767578e-05,
      "step": 49309,
      "training_step_time": 9.042557716369629
    },
    {
      "epoch": 4.702568054199219e-05,
      "grad_norm": 0.1778631955385208,
      "learning_rate": 5.526643549074833e-06,
      "loss": 0.0592,
      "step": 49310
    },
    {
      "epoch": 4.702568054199219e-05,
      "model_forward_time": 1.7049729824066162,
      "step": 49310
    },
    {
      "dynamics_loss_avg": 0.05326390601694584,
      "epoch": 4.702568054199219e-05,
      "step": 49310
    },
    {
      "action_loss_avg": 0.0017050427675712853,
      "epoch": 4.702568054199219e-05,
      "step": 49310
    },
    {
      "epoch": 4.702568054199219e-05,
      "step": 49310,
      "training_step_time": 8.60892105102539
    },
    {
      "epoch": 4.7026634216308594e-05,
      "model_forward_time": 2.3239119052886963,
      "step": 49311
    },
    {
      "epoch": 4.7026634216308594e-05,
      "step": 49311,
      "training_step_time": 8.49378514289856
    },
    {
      "epoch": 4.7027587890625e-05,
      "model_forward_time": 1.7095823287963867,
      "step": 49312
    },
    {
      "epoch": 4.7027587890625e-05,
      "step": 49312,
      "training_step_time": 7.749551296234131
    },
    {
      "epoch": 4.702854156494141e-05,
      "model_forward_time": 1.7175309658050537,
      "step": 49313
    },
    {
      "epoch": 4.702854156494141e-05,
      "step": 49313,
      "training_step_time": 7.746415615081787
    },
    {
      "epoch": 4.702949523925781e-05,
      "model_forward_time": 1.7148873805999756,
      "step": 49314
    },
    {
      "epoch": 4.702949523925781e-05,
      "step": 49314,
      "training_step_time": 7.780644416809082
    },
    {
      "epoch": 4.7030448913574217e-05,
      "model_forward_time": 1.7137224674224854,
      "step": 49315
    },
    {
      "epoch": 4.7030448913574217e-05,
      "step": 49315,
      "training_step_time": 7.5886242389678955
    },
    {
      "epoch": 4.703140258789063e-05,
      "model_forward_time": 1.7071428298950195,
      "step": 49316
    },
    {
      "epoch": 4.703140258789063e-05,
      "step": 49316,
      "training_step_time": 8.736479759216309
    },
    {
      "epoch": 4.703235626220703e-05,
      "model_forward_time": 1.7188644409179688,
      "step": 49317
    },
    {
      "epoch": 4.703235626220703e-05,
      "step": 49317,
      "training_step_time": 8.42678427696228
    },
    {
      "epoch": 4.7033309936523435e-05,
      "model_forward_time": 1.733736276626587,
      "step": 49318
    },
    {
      "epoch": 4.7033309936523435e-05,
      "step": 49318,
      "training_step_time": 8.427288293838501
    },
    {
      "epoch": 4.7034263610839846e-05,
      "model_forward_time": 1.7176494598388672,
      "step": 49319
    },
    {
      "epoch": 4.7034263610839846e-05,
      "step": 49319,
      "training_step_time": 7.794143199920654
    },
    {
      "epoch": 4.703521728515625e-05,
      "grad_norm": 0.09996138513088226,
      "learning_rate": 5.524999247981787e-06,
      "loss": 0.0576,
      "step": 49320
    },
    {
      "epoch": 4.703521728515625e-05,
      "model_forward_time": 2.3095054626464844,
      "step": 49320
    },
    {
      "dynamics_loss_avg": 0.05693375263363123,
      "epoch": 4.703521728515625e-05,
      "step": 49320
    },
    {
      "action_loss_avg": 0.0013518734020181,
      "epoch": 4.703521728515625e-05,
      "step": 49320
    },
    {
      "epoch": 4.703521728515625e-05,
      "step": 49320,
      "training_step_time": 9.956559896469116
    },
    {
      "epoch": 4.7036170959472654e-05,
      "model_forward_time": 1.711639642715454,
      "step": 49321
    },
    {
      "epoch": 4.7036170959472654e-05,
      "step": 49321,
      "training_step_time": 7.72580623626709
    },
    {
      "epoch": 4.7037124633789065e-05,
      "model_forward_time": 1.7165944576263428,
      "step": 49322
    },
    {
      "epoch": 4.7037124633789065e-05,
      "step": 49322,
      "training_step_time": 7.752207040786743
    },
    {
      "epoch": 4.703807830810547e-05,
      "model_forward_time": 1.7184803485870361,
      "step": 49323
    },
    {
      "epoch": 4.703807830810547e-05,
      "step": 49323,
      "training_step_time": 7.736854791641235
    },
    {
      "epoch": 4.703903198242187e-05,
      "model_forward_time": 1.7238762378692627,
      "step": 49324
    },
    {
      "epoch": 4.703903198242187e-05,
      "step": 49324,
      "training_step_time": 7.570482969284058
    },
    {
      "epoch": 4.7039985656738284e-05,
      "model_forward_time": 1.7068111896514893,
      "step": 49325
    },
    {
      "epoch": 4.7039985656738284e-05,
      "step": 49325,
      "training_step_time": 9.290732860565186
    },
    {
      "epoch": 4.704093933105469e-05,
      "model_forward_time": 1.707728385925293,
      "step": 49326
    },
    {
      "epoch": 4.704093933105469e-05,
      "step": 49326,
      "training_step_time": 7.839763641357422
    },
    {
      "epoch": 4.704189300537109e-05,
      "model_forward_time": 1.7134170532226562,
      "step": 49327
    },
    {
      "epoch": 4.704189300537109e-05,
      "step": 49327,
      "training_step_time": 7.726983070373535
    },
    {
      "epoch": 4.70428466796875e-05,
      "model_forward_time": 1.7098486423492432,
      "step": 49328
    },
    {
      "epoch": 4.70428466796875e-05,
      "step": 49328,
      "training_step_time": 8.753938674926758
    },
    {
      "epoch": 4.704380035400391e-05,
      "model_forward_time": 1.7054228782653809,
      "step": 49329
    },
    {
      "epoch": 4.704380035400391e-05,
      "step": 49329,
      "training_step_time": 9.71591854095459
    },
    {
      "epoch": 4.704475402832031e-05,
      "grad_norm": 0.0920245498418808,
      "learning_rate": 5.523354889475613e-06,
      "loss": 0.0594,
      "step": 49330
    },
    {
      "epoch": 4.704475402832031e-05,
      "model_forward_time": 1.7775230407714844,
      "step": 49330
    },
    {
      "dynamics_loss_avg": 0.04652295354753733,
      "epoch": 4.704475402832031e-05,
      "step": 49330
    },
    {
      "action_loss_avg": 0.0019854676094837485,
      "epoch": 4.704475402832031e-05,
      "step": 49330
    },
    {
      "epoch": 4.704475402832031e-05,
      "step": 49330,
      "training_step_time": 8.897921562194824
    },
    {
      "epoch": 4.704570770263672e-05,
      "model_forward_time": 1.715620756149292,
      "step": 49331
    },
    {
      "epoch": 4.704570770263672e-05,
      "step": 49331,
      "training_step_time": 8.788408279418945
    },
    {
      "epoch": 4.7046661376953126e-05,
      "model_forward_time": 1.713477611541748,
      "step": 49332
    },
    {
      "epoch": 4.7046661376953126e-05,
      "step": 49332,
      "training_step_time": 7.773347854614258
    },
    {
      "epoch": 4.704761505126953e-05,
      "model_forward_time": 1.7115106582641602,
      "step": 49333
    },
    {
      "epoch": 4.704761505126953e-05,
      "step": 49333,
      "training_step_time": 7.603811025619507
    },
    {
      "epoch": 4.704856872558594e-05,
      "model_forward_time": 1.7107131481170654,
      "step": 49334
    },
    {
      "epoch": 4.704856872558594e-05,
      "step": 49334,
      "training_step_time": 7.680664539337158
    },
    {
      "epoch": 4.7049522399902345e-05,
      "model_forward_time": 1.7178294658660889,
      "step": 49335
    },
    {
      "epoch": 4.7049522399902345e-05,
      "step": 49335,
      "training_step_time": 8.690431356430054
    },
    {
      "epoch": 4.705047607421875e-05,
      "model_forward_time": 1.7022819519042969,
      "step": 49336
    },
    {
      "epoch": 4.705047607421875e-05,
      "step": 49336,
      "training_step_time": 7.893773078918457
    },
    {
      "epoch": 4.705142974853516e-05,
      "model_forward_time": 1.7137033939361572,
      "step": 49337
    },
    {
      "epoch": 4.705142974853516e-05,
      "step": 49337,
      "training_step_time": 8.836337566375732
    },
    {
      "epoch": 4.7052383422851563e-05,
      "model_forward_time": 1.700206995010376,
      "step": 49338
    },
    {
      "epoch": 4.7052383422851563e-05,
      "step": 49338,
      "training_step_time": 7.85812783241272
    },
    {
      "epoch": 4.705333709716797e-05,
      "model_forward_time": 2.301982879638672,
      "step": 49339
    },
    {
      "epoch": 4.705333709716797e-05,
      "step": 49339,
      "training_step_time": 8.060941696166992
    },
    {
      "epoch": 4.705429077148438e-05,
      "grad_norm": 0.15183351933956146,
      "learning_rate": 5.521710473736134e-06,
      "loss": 0.0577,
      "step": 49340
    },
    {
      "epoch": 4.705429077148438e-05,
      "model_forward_time": 1.6963257789611816,
      "step": 49340
    },
    {
      "dynamics_loss_avg": 0.04437137246131897,
      "epoch": 4.705429077148438e-05,
      "step": 49340
    },
    {
      "action_loss_avg": 0.001629482480348088,
      "epoch": 4.705429077148438e-05,
      "step": 49340
    },
    {
      "epoch": 4.705429077148438e-05,
      "step": 49340,
      "training_step_time": 8.39363169670105
    },
    {
      "epoch": 4.705524444580078e-05,
      "model_forward_time": 1.709425449371338,
      "step": 49341
    },
    {
      "epoch": 4.705524444580078e-05,
      "step": 49341,
      "training_step_time": 7.645610094070435
    },
    {
      "epoch": 4.7056198120117186e-05,
      "model_forward_time": 1.7072772979736328,
      "step": 49342
    },
    {
      "epoch": 4.7056198120117186e-05,
      "step": 49342,
      "training_step_time": 7.597200393676758
    },
    {
      "epoch": 4.70571517944336e-05,
      "model_forward_time": 1.7109184265136719,
      "step": 49343
    },
    {
      "epoch": 4.70571517944336e-05,
      "step": 49343,
      "training_step_time": 7.759577751159668
    },
    {
      "epoch": 4.705810546875e-05,
      "model_forward_time": 1.7168095111846924,
      "step": 49344
    },
    {
      "epoch": 4.705810546875e-05,
      "step": 49344,
      "training_step_time": 7.816959857940674
    },
    {
      "epoch": 4.7059059143066405e-05,
      "model_forward_time": 1.736619472503662,
      "step": 49345
    },
    {
      "epoch": 4.7059059143066405e-05,
      "step": 49345,
      "training_step_time": 9.486406564712524
    },
    {
      "epoch": 4.706001281738281e-05,
      "model_forward_time": 1.7133660316467285,
      "step": 49346
    },
    {
      "epoch": 4.706001281738281e-05,
      "step": 49346,
      "training_step_time": 9.042882919311523
    },
    {
      "epoch": 4.706096649169922e-05,
      "model_forward_time": 1.6856193542480469,
      "step": 49347
    },
    {
      "epoch": 4.706096649169922e-05,
      "step": 49347,
      "training_step_time": 9.190752267837524
    },
    {
      "epoch": 4.7061920166015624e-05,
      "model_forward_time": 2.337822914123535,
      "step": 49348
    },
    {
      "epoch": 4.7061920166015624e-05,
      "step": 49348,
      "training_step_time": 9.450829029083252
    },
    {
      "epoch": 4.706287384033203e-05,
      "model_forward_time": 1.709655523300171,
      "step": 49349
    },
    {
      "epoch": 4.706287384033203e-05,
      "step": 49349,
      "training_step_time": 9.495111465454102
    },
    {
      "epoch": 4.706382751464844e-05,
      "grad_norm": 0.13887545466423035,
      "learning_rate": 5.520066000943182e-06,
      "loss": 0.0579,
      "step": 49350
    },
    {
      "epoch": 4.706382751464844e-05,
      "model_forward_time": 1.7194135189056396,
      "step": 49350
    },
    {
      "dynamics_loss_avg": 0.06198568567633629,
      "epoch": 4.706382751464844e-05,
      "step": 49350
    },
    {
      "action_loss_avg": 0.0014726366207469255,
      "epoch": 4.706382751464844e-05,
      "step": 49350
    },
    {
      "epoch": 4.706382751464844e-05,
      "step": 49350,
      "training_step_time": 7.743035316467285
    },
    {
      "epoch": 4.706478118896484e-05,
      "model_forward_time": 1.7097420692443848,
      "step": 49351
    },
    {
      "epoch": 4.706478118896484e-05,
      "step": 49351,
      "training_step_time": 7.511126518249512
    },
    {
      "epoch": 4.706573486328125e-05,
      "model_forward_time": 1.7146940231323242,
      "step": 49352
    },
    {
      "epoch": 4.706573486328125e-05,
      "step": 49352,
      "training_step_time": 7.326916933059692
    },
    {
      "epoch": 4.706668853759766e-05,
      "model_forward_time": 1.720357894897461,
      "step": 49353
    },
    {
      "epoch": 4.706668853759766e-05,
      "step": 49353,
      "training_step_time": 7.784076929092407
    },
    {
      "epoch": 4.706764221191406e-05,
      "model_forward_time": 1.7178480625152588,
      "step": 49354
    },
    {
      "epoch": 4.706764221191406e-05,
      "step": 49354,
      "training_step_time": 7.876110076904297
    },
    {
      "epoch": 4.7068595886230466e-05,
      "model_forward_time": 1.710508108139038,
      "step": 49355
    },
    {
      "epoch": 4.7068595886230466e-05,
      "step": 49355,
      "training_step_time": 8.76203203201294
    },
    {
      "epoch": 4.7069549560546877e-05,
      "model_forward_time": 1.7186994552612305,
      "step": 49356
    },
    {
      "epoch": 4.7069549560546877e-05,
      "step": 49356,
      "training_step_time": 7.8617401123046875
    },
    {
      "epoch": 4.707050323486328e-05,
      "model_forward_time": 2.3059744834899902,
      "step": 49357
    },
    {
      "epoch": 4.707050323486328e-05,
      "step": 49357,
      "training_step_time": 9.914589643478394
    },
    {
      "epoch": 4.7071456909179685e-05,
      "model_forward_time": 1.713561773300171,
      "step": 49358
    },
    {
      "epoch": 4.7071456909179685e-05,
      "step": 49358,
      "training_step_time": 7.784169673919678
    },
    {
      "epoch": 4.7072410583496095e-05,
      "model_forward_time": 1.7265677452087402,
      "step": 49359
    },
    {
      "epoch": 4.7072410583496095e-05,
      "step": 49359,
      "training_step_time": 7.7770116329193115
    },
    {
      "epoch": 4.70733642578125e-05,
      "grad_norm": 0.0826001837849617,
      "learning_rate": 5.518421471276596e-06,
      "loss": 0.0581,
      "step": 49360
    },
    {
      "epoch": 4.70733642578125e-05,
      "model_forward_time": 1.7124278545379639,
      "step": 49360
    },
    {
      "dynamics_loss_avg": 0.0483827612362802,
      "epoch": 4.70733642578125e-05,
      "step": 49360
    },
    {
      "action_loss_avg": 0.002542699407786131,
      "epoch": 4.70733642578125e-05,
      "step": 49360
    },
    {
      "epoch": 4.70733642578125e-05,
      "step": 49360,
      "training_step_time": 7.457372665405273
    },
    {
      "epoch": 4.7074317932128903e-05,
      "model_forward_time": 1.713836431503296,
      "step": 49361
    },
    {
      "epoch": 4.7074317932128903e-05,
      "step": 49361,
      "training_step_time": 7.612292289733887
    },
    {
      "epoch": 4.7075271606445314e-05,
      "model_forward_time": 1.712932825088501,
      "step": 49362
    },
    {
      "epoch": 4.7075271606445314e-05,
      "step": 49362,
      "training_step_time": 7.841656923294067
    },
    {
      "epoch": 4.707622528076172e-05,
      "model_forward_time": 1.711881399154663,
      "step": 49363
    },
    {
      "epoch": 4.707622528076172e-05,
      "step": 49363,
      "training_step_time": 9.633691787719727
    },
    {
      "epoch": 4.707717895507812e-05,
      "model_forward_time": 1.7054402828216553,
      "step": 49364
    },
    {
      "epoch": 4.707717895507812e-05,
      "step": 49364,
      "training_step_time": 8.87900972366333
    },
    {
      "epoch": 4.707813262939453e-05,
      "model_forward_time": 1.722550392150879,
      "step": 49365
    },
    {
      "epoch": 4.707813262939453e-05,
      "step": 49365,
      "training_step_time": 7.808726072311401
    },
    {
      "epoch": 4.707908630371094e-05,
      "model_forward_time": 2.2872860431671143,
      "step": 49366
    },
    {
      "epoch": 4.707908630371094e-05,
      "step": 49366,
      "training_step_time": 8.923715114593506
    },
    {
      "epoch": 4.708003997802734e-05,
      "model_forward_time": 1.6945221424102783,
      "step": 49367
    },
    {
      "epoch": 4.708003997802734e-05,
      "step": 49367,
      "training_step_time": 9.515427112579346
    },
    {
      "epoch": 4.708099365234375e-05,
      "model_forward_time": 1.7064049243927002,
      "step": 49368
    },
    {
      "epoch": 4.708099365234375e-05,
      "step": 49368,
      "training_step_time": 7.8046653270721436
    },
    {
      "epoch": 4.7081947326660156e-05,
      "model_forward_time": 1.7066988945007324,
      "step": 49369
    },
    {
      "epoch": 4.7081947326660156e-05,
      "step": 49369,
      "training_step_time": 7.561257362365723
    },
    {
      "epoch": 4.708290100097656e-05,
      "grad_norm": 0.11673230677843094,
      "learning_rate": 5.516776884916218e-06,
      "loss": 0.0595,
      "step": 49370
    },
    {
      "epoch": 4.708290100097656e-05,
      "model_forward_time": 1.7047631740570068,
      "step": 49370
    },
    {
      "dynamics_loss_avg": 0.05828109700232744,
      "epoch": 4.708290100097656e-05,
      "step": 49370
    },
    {
      "action_loss_avg": 0.0020630226237699388,
      "epoch": 4.708290100097656e-05,
      "step": 49370
    },
    {
      "epoch": 4.708290100097656e-05,
      "step": 49370,
      "training_step_time": 7.628878593444824
    },
    {
      "epoch": 4.708385467529297e-05,
      "model_forward_time": 1.7055160999298096,
      "step": 49371
    },
    {
      "epoch": 4.708385467529297e-05,
      "step": 49371,
      "training_step_time": 7.7378058433532715
    },
    {
      "epoch": 4.7084808349609375e-05,
      "model_forward_time": 1.7071495056152344,
      "step": 49372
    },
    {
      "epoch": 4.7084808349609375e-05,
      "step": 49372,
      "training_step_time": 7.880125045776367
    },
    {
      "epoch": 4.708576202392578e-05,
      "model_forward_time": 1.7099015712738037,
      "step": 49373
    },
    {
      "epoch": 4.708576202392578e-05,
      "step": 49373,
      "training_step_time": 9.348335981369019
    },
    {
      "epoch": 4.708671569824219e-05,
      "model_forward_time": 1.7062277793884277,
      "step": 49374
    },
    {
      "epoch": 4.708671569824219e-05,
      "step": 49374,
      "training_step_time": 8.694387435913086
    },
    {
      "epoch": 4.7087669372558594e-05,
      "model_forward_time": 2.2466163635253906,
      "step": 49375
    },
    {
      "epoch": 4.7087669372558594e-05,
      "step": 49375,
      "training_step_time": 7.900994300842285
    },
    {
      "epoch": 4.7088623046875e-05,
      "model_forward_time": 1.782482624053955,
      "step": 49376
    },
    {
      "epoch": 4.7088623046875e-05,
      "step": 49376,
      "training_step_time": 7.9397969245910645
    },
    {
      "epoch": 4.708957672119141e-05,
      "model_forward_time": 1.7096693515777588,
      "step": 49377
    },
    {
      "epoch": 4.708957672119141e-05,
      "step": 49377,
      "training_step_time": 7.836683988571167
    },
    {
      "epoch": 4.709053039550781e-05,
      "model_forward_time": 1.7051000595092773,
      "step": 49378
    },
    {
      "epoch": 4.709053039550781e-05,
      "step": 49378,
      "training_step_time": 7.8628785610198975
    },
    {
      "epoch": 4.709148406982422e-05,
      "model_forward_time": 1.7068724632263184,
      "step": 49379
    },
    {
      "epoch": 4.709148406982422e-05,
      "step": 49379,
      "training_step_time": 7.647019624710083
    },
    {
      "epoch": 4.709243774414063e-05,
      "grad_norm": 0.12163444608449936,
      "learning_rate": 5.515132242041893e-06,
      "loss": 0.0591,
      "step": 49380
    },
    {
      "epoch": 4.709243774414063e-05,
      "model_forward_time": 1.7036659717559814,
      "step": 49380
    },
    {
      "dynamics_loss_avg": 0.06239931583404541,
      "epoch": 4.709243774414063e-05,
      "step": 49380
    },
    {
      "action_loss_avg": 0.0020162125045317227,
      "epoch": 4.709243774414063e-05,
      "step": 49380
    },
    {
      "epoch": 4.709243774414063e-05,
      "step": 49380,
      "training_step_time": 7.744168043136597
    },
    {
      "epoch": 4.709339141845703e-05,
      "model_forward_time": 1.721769094467163,
      "step": 49381
    },
    {
      "epoch": 4.709339141845703e-05,
      "step": 49381,
      "training_step_time": 7.763367414474487
    },
    {
      "epoch": 4.7094345092773435e-05,
      "model_forward_time": 1.709130048751831,
      "step": 49382
    },
    {
      "epoch": 4.7094345092773435e-05,
      "step": 49382,
      "training_step_time": 7.870246410369873
    },
    {
      "epoch": 4.7095298767089846e-05,
      "model_forward_time": 1.7144441604614258,
      "step": 49383
    },
    {
      "epoch": 4.7095298767089846e-05,
      "step": 49383,
      "training_step_time": 9.476032257080078
    },
    {
      "epoch": 4.709625244140625e-05,
      "model_forward_time": 1.7102246284484863,
      "step": 49384
    },
    {
      "epoch": 4.709625244140625e-05,
      "step": 49384,
      "training_step_time": 7.845134496688843
    },
    {
      "epoch": 4.7097206115722654e-05,
      "model_forward_time": 2.332308530807495,
      "step": 49385
    },
    {
      "epoch": 4.7097206115722654e-05,
      "step": 49385,
      "training_step_time": 8.01005220413208
    },
    {
      "epoch": 4.7098159790039065e-05,
      "model_forward_time": 1.7027227878570557,
      "step": 49386
    },
    {
      "epoch": 4.7098159790039065e-05,
      "step": 49386,
      "training_step_time": 7.828991174697876
    },
    {
      "epoch": 4.709911346435547e-05,
      "model_forward_time": 1.7061645984649658,
      "step": 49387
    },
    {
      "epoch": 4.709911346435547e-05,
      "step": 49387,
      "training_step_time": 7.690451145172119
    },
    {
      "epoch": 4.710006713867187e-05,
      "model_forward_time": 1.710618019104004,
      "step": 49388
    },
    {
      "epoch": 4.710006713867187e-05,
      "step": 49388,
      "training_step_time": 7.574711799621582
    },
    {
      "epoch": 4.7101020812988284e-05,
      "model_forward_time": 1.7090981006622314,
      "step": 49389
    },
    {
      "epoch": 4.7101020812988284e-05,
      "step": 49389,
      "training_step_time": 8.399091720581055
    },
    {
      "epoch": 4.710197448730469e-05,
      "grad_norm": 0.14018110930919647,
      "learning_rate": 5.5134875428334835e-06,
      "loss": 0.0564,
      "step": 49390
    },
    {
      "epoch": 4.710197448730469e-05,
      "model_forward_time": 1.7030420303344727,
      "step": 49390
    },
    {
      "dynamics_loss_avg": 0.06439425721764565,
      "epoch": 4.710197448730469e-05,
      "step": 49390
    },
    {
      "action_loss_avg": 0.0023231859842780976,
      "epoch": 4.710197448730469e-05,
      "step": 49390
    },
    {
      "epoch": 4.710197448730469e-05,
      "step": 49390,
      "training_step_time": 7.7353315353393555
    },
    {
      "epoch": 4.710292816162109e-05,
      "model_forward_time": 1.722686767578125,
      "step": 49391
    },
    {
      "epoch": 4.710292816162109e-05,
      "step": 49391,
      "training_step_time": 9.36986231803894
    },
    {
      "epoch": 4.71038818359375e-05,
      "model_forward_time": 1.7109947204589844,
      "step": 49392
    },
    {
      "epoch": 4.71038818359375e-05,
      "step": 49392,
      "training_step_time": 8.927034616470337
    },
    {
      "epoch": 4.710483551025391e-05,
      "model_forward_time": 1.7108337879180908,
      "step": 49393
    },
    {
      "epoch": 4.710483551025391e-05,
      "step": 49393,
      "training_step_time": 8.489995241165161
    },
    {
      "epoch": 4.710578918457031e-05,
      "model_forward_time": 2.3120713233947754,
      "step": 49394
    },
    {
      "epoch": 4.710578918457031e-05,
      "step": 49394,
      "training_step_time": 8.946861028671265
    },
    {
      "epoch": 4.710674285888672e-05,
      "model_forward_time": 1.7022628784179688,
      "step": 49395
    },
    {
      "epoch": 4.710674285888672e-05,
      "step": 49395,
      "training_step_time": 7.906655788421631
    },
    {
      "epoch": 4.7107696533203126e-05,
      "model_forward_time": 1.3312017917633057,
      "step": 49396
    },
    {
      "epoch": 4.7107696533203126e-05,
      "step": 49396,
      "training_step_time": 8.533498764038086
    },
    {
      "epoch": 4.710865020751953e-05,
      "model_forward_time": 1.7099668979644775,
      "step": 49397
    },
    {
      "epoch": 4.710865020751953e-05,
      "step": 49397,
      "training_step_time": 8.31346607208252
    },
    {
      "epoch": 4.710960388183594e-05,
      "model_forward_time": 1.704723834991455,
      "step": 49398
    },
    {
      "epoch": 4.710960388183594e-05,
      "step": 49398,
      "training_step_time": 7.300387620925903
    },
    {
      "epoch": 4.7110557556152345e-05,
      "model_forward_time": 1.7272286415100098,
      "step": 49399
    },
    {
      "epoch": 4.7110557556152345e-05,
      "step": 49399,
      "training_step_time": 7.756089687347412
    },
    {
      "epoch": 4.711151123046875e-05,
      "grad_norm": 0.14241327345371246,
      "learning_rate": 5.511842787470847e-06,
      "loss": 0.0562,
      "step": 49400
    },
    {
      "epoch": 4.711151123046875e-05,
      "model_forward_time": 1.7157537937164307,
      "step": 49400
    },
    {
      "dynamics_loss_avg": 0.0644175574183464,
      "epoch": 4.711151123046875e-05,
      "step": 49400
    },
    {
      "action_loss_avg": 0.001593673537718132,
      "epoch": 4.711151123046875e-05,
      "step": 49400
    },
    {
      "epoch": 4.711151123046875e-05,
      "step": 49400,
      "training_step_time": 7.8571937084198
    },
    {
      "epoch": 4.711246490478516e-05,
      "model_forward_time": 1.721696138381958,
      "step": 49401
    },
    {
      "epoch": 4.711246490478516e-05,
      "step": 49401,
      "training_step_time": 8.49217700958252
    },
    {
      "epoch": 4.7113418579101563e-05,
      "model_forward_time": 1.7188198566436768,
      "step": 49402
    },
    {
      "epoch": 4.7113418579101563e-05,
      "step": 49402,
      "training_step_time": 8.012023687362671
    },
    {
      "epoch": 4.711437225341797e-05,
      "model_forward_time": 2.3279879093170166,
      "step": 49403
    },
    {
      "epoch": 4.711437225341797e-05,
      "step": 49403,
      "training_step_time": 7.922912359237671
    },
    {
      "epoch": 4.711532592773438e-05,
      "model_forward_time": 1.7160730361938477,
      "step": 49404
    },
    {
      "epoch": 4.711532592773438e-05,
      "step": 49404,
      "training_step_time": 7.992480754852295
    },
    {
      "epoch": 4.711627960205078e-05,
      "model_forward_time": 1.7246818542480469,
      "step": 49405
    },
    {
      "epoch": 4.711627960205078e-05,
      "step": 49405,
      "training_step_time": 7.65028977394104
    },
    {
      "epoch": 4.7117233276367186e-05,
      "model_forward_time": 1.6942532062530518,
      "step": 49406
    },
    {
      "epoch": 4.7117233276367186e-05,
      "step": 49406,
      "training_step_time": 7.6592347621917725
    },
    {
      "epoch": 4.711818695068359e-05,
      "model_forward_time": 1.7182338237762451,
      "step": 49407
    },
    {
      "epoch": 4.711818695068359e-05,
      "step": 49407,
      "training_step_time": 7.5898051261901855
    },
    {
      "epoch": 4.7119140625e-05,
      "model_forward_time": 1.7231101989746094,
      "step": 49408
    },
    {
      "epoch": 4.7119140625e-05,
      "step": 49408,
      "training_step_time": 7.844730615615845
    },
    {
      "epoch": 4.7120094299316405e-05,
      "model_forward_time": 1.723463535308838,
      "step": 49409
    },
    {
      "epoch": 4.7120094299316405e-05,
      "step": 49409,
      "training_step_time": 9.449441194534302
    },
    {
      "epoch": 4.712104797363281e-05,
      "grad_norm": 0.10750459879636765,
      "learning_rate": 5.510197976133853e-06,
      "loss": 0.0587,
      "step": 49410
    },
    {
      "epoch": 4.712104797363281e-05,
      "model_forward_time": 1.7118120193481445,
      "step": 49410
    },
    {
      "dynamics_loss_avg": 0.057142074964940545,
      "epoch": 4.712104797363281e-05,
      "step": 49410
    },
    {
      "action_loss_avg": 0.0011208242416614667,
      "epoch": 4.712104797363281e-05,
      "step": 49410
    },
    {
      "epoch": 4.712104797363281e-05,
      "step": 49410,
      "training_step_time": 8.76818037033081
    },
    {
      "epoch": 4.712200164794922e-05,
      "model_forward_time": 1.7165215015411377,
      "step": 49411
    },
    {
      "epoch": 4.712200164794922e-05,
      "step": 49411,
      "training_step_time": 8.449116706848145
    },
    {
      "epoch": 4.7122955322265624e-05,
      "model_forward_time": 2.2652323246002197,
      "step": 49412
    },
    {
      "epoch": 4.7122955322265624e-05,
      "step": 49412,
      "training_step_time": 8.44949722290039
    },
    {
      "epoch": 4.712390899658203e-05,
      "model_forward_time": 1.7019658088684082,
      "step": 49413
    },
    {
      "epoch": 4.712390899658203e-05,
      "step": 49413,
      "training_step_time": 9.5566987991333
    },
    {
      "epoch": 4.712486267089844e-05,
      "model_forward_time": 1.6979410648345947,
      "step": 49414
    },
    {
      "epoch": 4.712486267089844e-05,
      "step": 49414,
      "training_step_time": 7.753397226333618
    },
    {
      "epoch": 4.712581634521484e-05,
      "model_forward_time": 1.7024221420288086,
      "step": 49415
    },
    {
      "epoch": 4.712581634521484e-05,
      "step": 49415,
      "training_step_time": 7.69864821434021
    },
    {
      "epoch": 4.712677001953125e-05,
      "model_forward_time": 1.701667308807373,
      "step": 49416
    },
    {
      "epoch": 4.712677001953125e-05,
      "step": 49416,
      "training_step_time": 7.736230373382568
    },
    {
      "epoch": 4.712772369384766e-05,
      "model_forward_time": 1.7030081748962402,
      "step": 49417
    },
    {
      "epoch": 4.712772369384766e-05,
      "step": 49417,
      "training_step_time": 7.8045008182525635
    },
    {
      "epoch": 4.712867736816406e-05,
      "model_forward_time": 1.7043910026550293,
      "step": 49418
    },
    {
      "epoch": 4.712867736816406e-05,
      "step": 49418,
      "training_step_time": 7.828907251358032
    },
    {
      "epoch": 4.7129631042480466e-05,
      "model_forward_time": 1.6999123096466064,
      "step": 49419
    },
    {
      "epoch": 4.7129631042480466e-05,
      "step": 49419,
      "training_step_time": 7.837896823883057
    },
    {
      "epoch": 4.713058471679688e-05,
      "grad_norm": 0.21086111664772034,
      "learning_rate": 5.508553109002376e-06,
      "loss": 0.0585,
      "step": 49420
    },
    {
      "epoch": 4.713058471679688e-05,
      "model_forward_time": 1.6954433917999268,
      "step": 49420
    },
    {
      "dynamics_loss_avg": 0.044246337842196226,
      "epoch": 4.713058471679688e-05,
      "step": 49420
    },
    {
      "action_loss_avg": 0.0013574079195677769,
      "epoch": 4.713058471679688e-05,
      "step": 49420
    },
    {
      "epoch": 4.713058471679688e-05,
      "step": 49420,
      "training_step_time": 8.861948251724243
    },
    {
      "epoch": 4.713153839111328e-05,
      "model_forward_time": 1.7010829448699951,
      "step": 49421
    },
    {
      "epoch": 4.713153839111328e-05,
      "step": 49421,
      "training_step_time": 8.935482740402222
    },
    {
      "epoch": 4.7132492065429685e-05,
      "model_forward_time": 1.7171411514282227,
      "step": 49422
    },
    {
      "epoch": 4.7132492065429685e-05,
      "step": 49422,
      "training_step_time": 9.811393976211548
    },
    {
      "epoch": 4.7133445739746095e-05,
      "model_forward_time": 1.6998376846313477,
      "step": 49423
    },
    {
      "epoch": 4.7133445739746095e-05,
      "step": 49423,
      "training_step_time": 8.534143686294556
    },
    {
      "epoch": 4.71343994140625e-05,
      "model_forward_time": 1.7043895721435547,
      "step": 49424
    },
    {
      "epoch": 4.71343994140625e-05,
      "step": 49424,
      "training_step_time": 8.502917528152466
    },
    {
      "epoch": 4.7135353088378904e-05,
      "model_forward_time": 1.6996214389801025,
      "step": 49425
    },
    {
      "epoch": 4.7135353088378904e-05,
      "step": 49425,
      "training_step_time": 7.863919973373413
    },
    {
      "epoch": 4.7136306762695314e-05,
      "model_forward_time": 1.722611665725708,
      "step": 49426
    },
    {
      "epoch": 4.7136306762695314e-05,
      "step": 49426,
      "training_step_time": 7.796550273895264
    },
    {
      "epoch": 4.713726043701172e-05,
      "model_forward_time": 1.7235822677612305,
      "step": 49427
    },
    {
      "epoch": 4.713726043701172e-05,
      "step": 49427,
      "training_step_time": 8.743897199630737
    },
    {
      "epoch": 4.713821411132812e-05,
      "model_forward_time": 1.722687005996704,
      "step": 49428
    },
    {
      "epoch": 4.713821411132812e-05,
      "step": 49428,
      "training_step_time": 9.628960609436035
    },
    {
      "epoch": 4.713916778564453e-05,
      "model_forward_time": 1.713367223739624,
      "step": 49429
    },
    {
      "epoch": 4.713916778564453e-05,
      "step": 49429,
      "training_step_time": 8.636621475219727
    },
    {
      "epoch": 4.714012145996094e-05,
      "grad_norm": 0.1711406260728836,
      "learning_rate": 5.506908186256296e-06,
      "loss": 0.054,
      "step": 49430
    },
    {
      "epoch": 4.714012145996094e-05,
      "model_forward_time": 1.7128159999847412,
      "step": 49430
    },
    {
      "dynamics_loss_avg": 0.047399905230849984,
      "epoch": 4.714012145996094e-05,
      "step": 49430
    },
    {
      "action_loss_avg": 0.001373853936092928,
      "epoch": 4.714012145996094e-05,
      "step": 49430
    },
    {
      "epoch": 4.714012145996094e-05,
      "step": 49430,
      "training_step_time": 7.915418386459351
    },
    {
      "epoch": 4.714107513427734e-05,
      "model_forward_time": 2.3746390342712402,
      "step": 49431
    },
    {
      "epoch": 4.714107513427734e-05,
      "step": 49431,
      "training_step_time": 7.950418472290039
    },
    {
      "epoch": 4.714202880859375e-05,
      "model_forward_time": 1.7210521697998047,
      "step": 49432
    },
    {
      "epoch": 4.714202880859375e-05,
      "step": 49432,
      "training_step_time": 7.909884929656982
    },
    {
      "epoch": 4.7142982482910156e-05,
      "model_forward_time": 1.7215125560760498,
      "step": 49433
    },
    {
      "epoch": 4.7142982482910156e-05,
      "step": 49433,
      "training_step_time": 7.731675148010254
    },
    {
      "epoch": 4.714393615722656e-05,
      "model_forward_time": 1.7231719493865967,
      "step": 49434
    },
    {
      "epoch": 4.714393615722656e-05,
      "step": 49434,
      "training_step_time": 7.7772111892700195
    },
    {
      "epoch": 4.714488983154297e-05,
      "model_forward_time": 1.7285447120666504,
      "step": 49435
    },
    {
      "epoch": 4.714488983154297e-05,
      "step": 49435,
      "training_step_time": 7.72655463218689
    },
    {
      "epoch": 4.7145843505859375e-05,
      "model_forward_time": 1.7167842388153076,
      "step": 49436
    },
    {
      "epoch": 4.7145843505859375e-05,
      "step": 49436,
      "training_step_time": 7.787067413330078
    },
    {
      "epoch": 4.714679718017578e-05,
      "model_forward_time": 1.7191002368927002,
      "step": 49437
    },
    {
      "epoch": 4.714679718017578e-05,
      "step": 49437,
      "training_step_time": 8.744970560073853
    },
    {
      "epoch": 4.714775085449219e-05,
      "model_forward_time": 1.3467462062835693,
      "step": 49438
    },
    {
      "epoch": 4.714775085449219e-05,
      "step": 49438,
      "training_step_time": 7.850518465042114
    },
    {
      "epoch": 4.7148704528808594e-05,
      "model_forward_time": 1.7222836017608643,
      "step": 49439
    },
    {
      "epoch": 4.7148704528808594e-05,
      "step": 49439,
      "training_step_time": 8.40344786643982
    },
    {
      "epoch": 4.7149658203125e-05,
      "grad_norm": 0.11089739203453064,
      "learning_rate": 5.505263208075496e-06,
      "loss": 0.0561,
      "step": 49440
    },
    {
      "epoch": 4.7149658203125e-05,
      "model_forward_time": 2.337791919708252,
      "step": 49440
    },
    {
      "dynamics_loss_avg": 0.05073348917067051,
      "epoch": 4.7149658203125e-05,
      "step": 49440
    },
    {
      "action_loss_avg": 0.00198784374515526,
      "epoch": 4.7149658203125e-05,
      "step": 49440
    },
    {
      "epoch": 4.7149658203125e-05,
      "step": 49440,
      "training_step_time": 8.602323532104492
    },
    {
      "epoch": 4.715061187744141e-05,
      "model_forward_time": 1.7150113582611084,
      "step": 49441
    },
    {
      "epoch": 4.715061187744141e-05,
      "step": 49441,
      "training_step_time": 7.8492372035980225
    },
    {
      "epoch": 4.715156555175781e-05,
      "model_forward_time": 1.717498779296875,
      "step": 49442
    },
    {
      "epoch": 4.715156555175781e-05,
      "step": 49442,
      "training_step_time": 7.728571653366089
    },
    {
      "epoch": 4.715251922607422e-05,
      "model_forward_time": 1.7173793315887451,
      "step": 49443
    },
    {
      "epoch": 4.715251922607422e-05,
      "step": 49443,
      "training_step_time": 7.673865556716919
    },
    {
      "epoch": 4.715347290039063e-05,
      "model_forward_time": 1.7214443683624268,
      "step": 49444
    },
    {
      "epoch": 4.715347290039063e-05,
      "step": 49444,
      "training_step_time": 7.709772825241089
    },
    {
      "epoch": 4.715442657470703e-05,
      "model_forward_time": 1.6991617679595947,
      "step": 49445
    },
    {
      "epoch": 4.715442657470703e-05,
      "step": 49445,
      "training_step_time": 8.252628564834595
    },
    {
      "epoch": 4.7155380249023436e-05,
      "model_forward_time": 1.7221319675445557,
      "step": 49446
    },
    {
      "epoch": 4.7155380249023436e-05,
      "step": 49446,
      "training_step_time": 7.871487379074097
    },
    {
      "epoch": 4.7156333923339846e-05,
      "model_forward_time": 1.729445457458496,
      "step": 49447
    },
    {
      "epoch": 4.7156333923339846e-05,
      "step": 49447,
      "training_step_time": 8.884263277053833
    },
    {
      "epoch": 4.715728759765625e-05,
      "model_forward_time": 1.7197585105895996,
      "step": 49448
    },
    {
      "epoch": 4.715728759765625e-05,
      "step": 49448,
      "training_step_time": 7.692137718200684
    },
    {
      "epoch": 4.7158241271972654e-05,
      "model_forward_time": 2.3062329292297363,
      "step": 49449
    },
    {
      "epoch": 4.7158241271972654e-05,
      "step": 49449,
      "training_step_time": 8.825117588043213
    },
    {
      "epoch": 4.7159194946289065e-05,
      "grad_norm": 0.1178862452507019,
      "learning_rate": 5.503618174639874e-06,
      "loss": 0.0533,
      "step": 49450
    },
    {
      "epoch": 4.7159194946289065e-05,
      "model_forward_time": 1.7151553630828857,
      "step": 49450
    },
    {
      "dynamics_loss_avg": 0.057824190519750115,
      "epoch": 4.7159194946289065e-05,
      "step": 49450
    },
    {
      "action_loss_avg": 0.002339445648249239,
      "epoch": 4.7159194946289065e-05,
      "step": 49450
    },
    {
      "epoch": 4.7159194946289065e-05,
      "step": 49450,
      "training_step_time": 8.502462387084961
    },
    {
      "epoch": 4.716014862060547e-05,
      "model_forward_time": 1.7255191802978516,
      "step": 49451
    },
    {
      "epoch": 4.716014862060547e-05,
      "step": 49451,
      "training_step_time": 7.69573712348938
    },
    {
      "epoch": 4.716110229492187e-05,
      "model_forward_time": 1.7179961204528809,
      "step": 49452
    },
    {
      "epoch": 4.716110229492187e-05,
      "step": 49452,
      "training_step_time": 7.4963343143463135
    },
    {
      "epoch": 4.7162055969238284e-05,
      "model_forward_time": 1.7245762348175049,
      "step": 49453
    },
    {
      "epoch": 4.7162055969238284e-05,
      "step": 49453,
      "training_step_time": 7.72199273109436
    },
    {
      "epoch": 4.716300964355469e-05,
      "model_forward_time": 1.7243831157684326,
      "step": 49454
    },
    {
      "epoch": 4.716300964355469e-05,
      "step": 49454,
      "training_step_time": 7.801878213882446
    },
    {
      "epoch": 4.716396331787109e-05,
      "model_forward_time": 1.7263283729553223,
      "step": 49455
    },
    {
      "epoch": 4.716396331787109e-05,
      "step": 49455,
      "training_step_time": 7.836399555206299
    },
    {
      "epoch": 4.71649169921875e-05,
      "model_forward_time": 1.7230360507965088,
      "step": 49456
    },
    {
      "epoch": 4.71649169921875e-05,
      "step": 49456,
      "training_step_time": 7.7069478034973145
    },
    {
      "epoch": 4.716587066650391e-05,
      "model_forward_time": 1.7202935218811035,
      "step": 49457
    },
    {
      "epoch": 4.716587066650391e-05,
      "step": 49457,
      "training_step_time": 7.923573732376099
    },
    {
      "epoch": 4.716682434082031e-05,
      "model_forward_time": 2.281954765319824,
      "step": 49458
    },
    {
      "epoch": 4.716682434082031e-05,
      "step": 49458,
      "training_step_time": 10.688435316085815
    },
    {
      "epoch": 4.716777801513672e-05,
      "model_forward_time": 1.811122179031372,
      "step": 49459
    },
    {
      "epoch": 4.716777801513672e-05,
      "step": 49459,
      "training_step_time": 7.945918560028076
    },
    {
      "epoch": 4.7168731689453126e-05,
      "grad_norm": 0.07032834738492966,
      "learning_rate": 5.501973086129325e-06,
      "loss": 0.0533,
      "step": 49460
    },
    {
      "epoch": 4.7168731689453126e-05,
      "model_forward_time": 1.7137119770050049,
      "step": 49460
    },
    {
      "dynamics_loss_avg": 0.05557625945657492,
      "epoch": 4.7168731689453126e-05,
      "step": 49460
    },
    {
      "action_loss_avg": 0.0018796029151417315,
      "epoch": 4.7168731689453126e-05,
      "step": 49460
    },
    {
      "epoch": 4.7168731689453126e-05,
      "step": 49460,
      "training_step_time": 7.876407861709595
    },
    {
      "epoch": 4.716968536376953e-05,
      "model_forward_time": 1.7190592288970947,
      "step": 49461
    },
    {
      "epoch": 4.716968536376953e-05,
      "step": 49461,
      "training_step_time": 7.6999733448028564
    },
    {
      "epoch": 4.717063903808594e-05,
      "model_forward_time": 1.7231659889221191,
      "step": 49462
    },
    {
      "epoch": 4.717063903808594e-05,
      "step": 49462,
      "training_step_time": 8.419625520706177
    },
    {
      "epoch": 4.7171592712402345e-05,
      "model_forward_time": 1.7157971858978271,
      "step": 49463
    },
    {
      "epoch": 4.7171592712402345e-05,
      "step": 49463,
      "training_step_time": 7.805395841598511
    },
    {
      "epoch": 4.717254638671875e-05,
      "model_forward_time": 1.7214348316192627,
      "step": 49464
    },
    {
      "epoch": 4.717254638671875e-05,
      "step": 49464,
      "training_step_time": 7.944225549697876
    },
    {
      "epoch": 4.717350006103516e-05,
      "model_forward_time": 1.728515863418579,
      "step": 49465
    },
    {
      "epoch": 4.717350006103516e-05,
      "step": 49465,
      "training_step_time": 8.989476680755615
    },
    {
      "epoch": 4.7174453735351564e-05,
      "model_forward_time": 1.7181477546691895,
      "step": 49466
    },
    {
      "epoch": 4.7174453735351564e-05,
      "step": 49466,
      "training_step_time": 8.929019212722778
    },
    {
      "epoch": 4.717540740966797e-05,
      "model_forward_time": 1.7293143272399902,
      "step": 49467
    },
    {
      "epoch": 4.717540740966797e-05,
      "step": 49467,
      "training_step_time": 7.911971569061279
    },
    {
      "epoch": 4.717636108398438e-05,
      "model_forward_time": 2.378796100616455,
      "step": 49468
    },
    {
      "epoch": 4.717636108398438e-05,
      "step": 49468,
      "training_step_time": 9.642678499221802
    },
    {
      "epoch": 4.717731475830078e-05,
      "model_forward_time": 1.727013111114502,
      "step": 49469
    },
    {
      "epoch": 4.717731475830078e-05,
      "step": 49469,
      "training_step_time": 7.831241130828857
    },
    {
      "epoch": 4.7178268432617186e-05,
      "grad_norm": 0.11545029282569885,
      "learning_rate": 5.500327942723754e-06,
      "loss": 0.0578,
      "step": 49470
    },
    {
      "epoch": 4.7178268432617186e-05,
      "model_forward_time": 1.7329704761505127,
      "step": 49470
    },
    {
      "dynamics_loss_avg": 0.05034846914932132,
      "epoch": 4.7178268432617186e-05,
      "step": 49470
    },
    {
      "action_loss_avg": 0.002311302494490519,
      "epoch": 4.7178268432617186e-05,
      "step": 49470
    },
    {
      "epoch": 4.7178268432617186e-05,
      "step": 49470,
      "training_step_time": 7.5269858837127686
    },
    {
      "epoch": 4.717922210693359e-05,
      "model_forward_time": 1.7368578910827637,
      "step": 49471
    },
    {
      "epoch": 4.717922210693359e-05,
      "step": 49471,
      "training_step_time": 7.825354099273682
    },
    {
      "epoch": 4.718017578125e-05,
      "model_forward_time": 1.7302587032318115,
      "step": 49472
    },
    {
      "epoch": 4.718017578125e-05,
      "step": 49472,
      "training_step_time": 7.72022557258606
    },
    {
      "epoch": 4.7181129455566405e-05,
      "model_forward_time": 1.7390515804290771,
      "step": 49473
    },
    {
      "epoch": 4.7181129455566405e-05,
      "step": 49473,
      "training_step_time": 7.901817798614502
    },
    {
      "epoch": 4.718208312988281e-05,
      "model_forward_time": 1.7301106452941895,
      "step": 49474
    },
    {
      "epoch": 4.718208312988281e-05,
      "step": 49474,
      "training_step_time": 8.673927783966064
    },
    {
      "epoch": 4.718303680419922e-05,
      "model_forward_time": 1.7339746952056885,
      "step": 49475
    },
    {
      "epoch": 4.718303680419922e-05,
      "step": 49475,
      "training_step_time": 8.856033086776733
    },
    {
      "epoch": 4.7183990478515624e-05,
      "model_forward_time": 1.7356030941009521,
      "step": 49476
    },
    {
      "epoch": 4.7183990478515624e-05,
      "step": 49476,
      "training_step_time": 8.040122270584106
    },
    {
      "epoch": 4.718494415283203e-05,
      "model_forward_time": 2.316478967666626,
      "step": 49477
    },
    {
      "epoch": 4.718494415283203e-05,
      "step": 49477,
      "training_step_time": 8.592909336090088
    },
    {
      "epoch": 4.718589782714844e-05,
      "model_forward_time": 1.7163491249084473,
      "step": 49478
    },
    {
      "epoch": 4.718589782714844e-05,
      "step": 49478,
      "training_step_time": 7.664704322814941
    },
    {
      "epoch": 4.718685150146484e-05,
      "model_forward_time": 1.7449719905853271,
      "step": 49479
    },
    {
      "epoch": 4.718685150146484e-05,
      "step": 49479,
      "training_step_time": 7.8081910610198975
    },
    {
      "epoch": 4.718780517578125e-05,
      "grad_norm": 0.09850375354290009,
      "learning_rate": 5.498682744603071e-06,
      "loss": 0.0589,
      "step": 49480
    },
    {
      "epoch": 4.718780517578125e-05,
      "model_forward_time": 1.7326161861419678,
      "step": 49480
    },
    {
      "dynamics_loss_avg": 0.055038582812994716,
      "epoch": 4.718780517578125e-05,
      "step": 49480
    },
    {
      "action_loss_avg": 0.0024471674289088695,
      "epoch": 4.718780517578125e-05,
      "step": 49480
    },
    {
      "epoch": 4.718780517578125e-05,
      "step": 49480,
      "training_step_time": 7.612660646438599
    },
    {
      "epoch": 4.718875885009766e-05,
      "model_forward_time": 1.7314832210540771,
      "step": 49481
    },
    {
      "epoch": 4.718875885009766e-05,
      "step": 49481,
      "training_step_time": 9.071350336074829
    },
    {
      "epoch": 4.718971252441406e-05,
      "model_forward_time": 1.7331738471984863,
      "step": 49482
    },
    {
      "epoch": 4.718971252441406e-05,
      "step": 49482,
      "training_step_time": 7.830114841461182
    },
    {
      "epoch": 4.7190666198730466e-05,
      "model_forward_time": 1.7307217121124268,
      "step": 49483
    },
    {
      "epoch": 4.7190666198730466e-05,
      "step": 49483,
      "training_step_time": 9.340996742248535
    },
    {
      "epoch": 4.719161987304688e-05,
      "model_forward_time": 1.7417407035827637,
      "step": 49484
    },
    {
      "epoch": 4.719161987304688e-05,
      "step": 49484,
      "training_step_time": 9.083422899246216
    },
    {
      "epoch": 4.719257354736328e-05,
      "model_forward_time": 1.7332091331481934,
      "step": 49485
    },
    {
      "epoch": 4.719257354736328e-05,
      "step": 49485,
      "training_step_time": 8.81936764717102
    },
    {
      "epoch": 4.7193527221679685e-05,
      "model_forward_time": 2.3480405807495117,
      "step": 49486
    },
    {
      "epoch": 4.7193527221679685e-05,
      "step": 49486,
      "training_step_time": 7.677762508392334
    },
    {
      "epoch": 4.7194480895996096e-05,
      "model_forward_time": 1.7325522899627686,
      "step": 49487
    },
    {
      "epoch": 4.7194480895996096e-05,
      "step": 49487,
      "training_step_time": 7.658918619155884
    },
    {
      "epoch": 4.71954345703125e-05,
      "model_forward_time": 1.739396095275879,
      "step": 49488
    },
    {
      "epoch": 4.71954345703125e-05,
      "step": 49488,
      "training_step_time": 7.574753999710083
    },
    {
      "epoch": 4.7196388244628904e-05,
      "model_forward_time": 1.7592225074768066,
      "step": 49489
    },
    {
      "epoch": 4.7196388244628904e-05,
      "step": 49489,
      "training_step_time": 7.5303754806518555
    },
    {
      "epoch": 4.7197341918945314e-05,
      "grad_norm": 0.11177922040224075,
      "learning_rate": 5.4970374919471935e-06,
      "loss": 0.0586,
      "step": 49490
    },
    {
      "epoch": 4.7197341918945314e-05,
      "model_forward_time": 1.7017664909362793,
      "step": 49490
    },
    {
      "dynamics_loss_avg": 0.046491789445281026,
      "epoch": 4.7197341918945314e-05,
      "step": 49490
    },
    {
      "action_loss_avg": 0.0014444280706811697,
      "epoch": 4.7197341918945314e-05,
      "step": 49490
    },
    {
      "epoch": 4.7197341918945314e-05,
      "step": 49490,
      "training_step_time": 7.5226359367370605
    },
    {
      "epoch": 4.719829559326172e-05,
      "model_forward_time": 1.7216720581054688,
      "step": 49491
    },
    {
      "epoch": 4.719829559326172e-05,
      "step": 49491,
      "training_step_time": 7.560436964035034
    },
    {
      "epoch": 4.719924926757812e-05,
      "model_forward_time": 1.7177844047546387,
      "step": 49492
    },
    {
      "epoch": 4.719924926757812e-05,
      "step": 49492,
      "training_step_time": 7.4770894050598145
    },
    {
      "epoch": 4.720020294189453e-05,
      "model_forward_time": 1.7213962078094482,
      "step": 49493
    },
    {
      "epoch": 4.720020294189453e-05,
      "step": 49493,
      "training_step_time": 7.64955735206604
    },
    {
      "epoch": 4.720115661621094e-05,
      "model_forward_time": 1.720712661743164,
      "step": 49494
    },
    {
      "epoch": 4.720115661621094e-05,
      "step": 49494,
      "training_step_time": 8.823543310165405
    },
    {
      "epoch": 4.720211029052734e-05,
      "model_forward_time": 2.294100522994995,
      "step": 49495
    },
    {
      "epoch": 4.720211029052734e-05,
      "step": 49495,
      "training_step_time": 8.746627569198608
    },
    {
      "epoch": 4.720306396484375e-05,
      "model_forward_time": 1.796353816986084,
      "step": 49496
    },
    {
      "epoch": 4.720306396484375e-05,
      "step": 49496,
      "training_step_time": 7.71151065826416
    },
    {
      "epoch": 4.7204017639160156e-05,
      "model_forward_time": 1.724128007888794,
      "step": 49497
    },
    {
      "epoch": 4.7204017639160156e-05,
      "step": 49497,
      "training_step_time": 7.6437671184539795
    },
    {
      "epoch": 4.720497131347656e-05,
      "model_forward_time": 1.7208592891693115,
      "step": 49498
    },
    {
      "epoch": 4.720497131347656e-05,
      "step": 49498,
      "training_step_time": 8.354783296585083
    },
    {
      "epoch": 4.720592498779297e-05,
      "model_forward_time": 1.7248814105987549,
      "step": 49499
    },
    {
      "epoch": 4.720592498779297e-05,
      "step": 49499,
      "training_step_time": 7.506377220153809
    },
    {
      "epoch": 4.7206878662109375e-05,
      "grad_norm": 0.11844251304864883,
      "learning_rate": 5.495392184936043e-06,
      "loss": 0.0596,
      "step": 49500
    },
    {
      "epoch": 4.7206878662109375e-05,
      "model_forward_time": 1.7147412300109863,
      "step": 49500
    },
    {
      "dynamics_loss_avg": 0.04900681413710117,
      "epoch": 4.7206878662109375e-05,
      "step": 49500
    },
    {
      "action_loss_avg": 0.001561567337193992,
      "epoch": 4.7206878662109375e-05,
      "step": 49500
    },
    {
      "epoch": 4.7206878662109375e-05,
      "step": 49500,
      "training_step_time": 7.512298822402954
    },
    {
      "epoch": 4.720783233642578e-05,
      "model_forward_time": 1.7229907512664795,
      "step": 49501
    },
    {
      "epoch": 4.720783233642578e-05,
      "step": 49501,
      "training_step_time": 7.69490385055542
    },
    {
      "epoch": 4.720878601074219e-05,
      "model_forward_time": 1.7152159214019775,
      "step": 49502
    },
    {
      "epoch": 4.720878601074219e-05,
      "step": 49502,
      "training_step_time": 7.614877700805664
    },
    {
      "epoch": 4.7209739685058594e-05,
      "model_forward_time": 1.7317707538604736,
      "step": 49503
    },
    {
      "epoch": 4.7209739685058594e-05,
      "step": 49503,
      "training_step_time": 8.510847806930542
    },
    {
      "epoch": 4.7210693359375e-05,
      "model_forward_time": 2.2951817512512207,
      "step": 49504
    },
    {
      "epoch": 4.7210693359375e-05,
      "step": 49504,
      "training_step_time": 8.155565738677979
    },
    {
      "epoch": 4.721164703369141e-05,
      "model_forward_time": 1.8012661933898926,
      "step": 49505
    },
    {
      "epoch": 4.721164703369141e-05,
      "step": 49505,
      "training_step_time": 7.694243669509888
    },
    {
      "epoch": 4.721260070800781e-05,
      "model_forward_time": 1.7094533443450928,
      "step": 49506
    },
    {
      "epoch": 4.721260070800781e-05,
      "step": 49506,
      "training_step_time": 7.654200792312622
    },
    {
      "epoch": 4.721355438232422e-05,
      "model_forward_time": 1.7140493392944336,
      "step": 49507
    },
    {
      "epoch": 4.721355438232422e-05,
      "step": 49507,
      "training_step_time": 8.187201738357544
    },
    {
      "epoch": 4.721450805664063e-05,
      "model_forward_time": 1.7378480434417725,
      "step": 49508
    },
    {
      "epoch": 4.721450805664063e-05,
      "step": 49508,
      "training_step_time": 7.555221080780029
    },
    {
      "epoch": 4.721546173095703e-05,
      "model_forward_time": 1.3406126499176025,
      "step": 49509
    },
    {
      "epoch": 4.721546173095703e-05,
      "step": 49509,
      "training_step_time": 7.560673236846924
    },
    {
      "epoch": 4.7216415405273436e-05,
      "grad_norm": 0.129705548286438,
      "learning_rate": 5.493746823749548e-06,
      "loss": 0.0599,
      "step": 49510
    },
    {
      "epoch": 4.7216415405273436e-05,
      "model_forward_time": 1.713728427886963,
      "step": 49510
    },
    {
      "dynamics_loss_avg": 0.04604381117969751,
      "epoch": 4.7216415405273436e-05,
      "step": 49510
    },
    {
      "action_loss_avg": 0.0013289444585097953,
      "epoch": 4.7216415405273436e-05,
      "step": 49510
    },
    {
      "epoch": 4.7216415405273436e-05,
      "step": 49510,
      "training_step_time": 7.689172029495239
    },
    {
      "epoch": 4.7217369079589846e-05,
      "model_forward_time": 1.721745252609253,
      "step": 49511
    },
    {
      "epoch": 4.7217369079589846e-05,
      "step": 49511,
      "training_step_time": 7.55046272277832
    },
    {
      "epoch": 4.721832275390625e-05,
      "model_forward_time": 1.7260851860046387,
      "step": 49512
    },
    {
      "epoch": 4.721832275390625e-05,
      "step": 49512,
      "training_step_time": 8.327152490615845
    },
    {
      "epoch": 4.7219276428222654e-05,
      "model_forward_time": 1.7221472263336182,
      "step": 49513
    },
    {
      "epoch": 4.7219276428222654e-05,
      "step": 49513,
      "training_step_time": 9.076347351074219
    },
    {
      "epoch": 4.7220230102539065e-05,
      "model_forward_time": 2.3533172607421875,
      "step": 49514
    },
    {
      "epoch": 4.7220230102539065e-05,
      "step": 49514,
      "training_step_time": 8.62938380241394
    },
    {
      "epoch": 4.722118377685547e-05,
      "model_forward_time": 1.718240737915039,
      "step": 49515
    },
    {
      "epoch": 4.722118377685547e-05,
      "step": 49515,
      "training_step_time": 8.470099449157715
    },
    {
      "epoch": 4.722213745117187e-05,
      "model_forward_time": 1.7306795120239258,
      "step": 49516
    },
    {
      "epoch": 4.722213745117187e-05,
      "step": 49516,
      "training_step_time": 8.23803424835205
    },
    {
      "epoch": 4.7223091125488284e-05,
      "model_forward_time": 1.7255053520202637,
      "step": 49517
    },
    {
      "epoch": 4.7223091125488284e-05,
      "step": 49517,
      "training_step_time": 8.138365983963013
    },
    {
      "epoch": 4.722404479980469e-05,
      "model_forward_time": 1.7203452587127686,
      "step": 49518
    },
    {
      "epoch": 4.722404479980469e-05,
      "step": 49518,
      "training_step_time": 7.627909898757935
    },
    {
      "epoch": 4.722499847412109e-05,
      "model_forward_time": 1.7235827445983887,
      "step": 49519
    },
    {
      "epoch": 4.722499847412109e-05,
      "step": 49519,
      "training_step_time": 8.48573350906372
    },
    {
      "epoch": 4.72259521484375e-05,
      "grad_norm": 0.10353294014930725,
      "learning_rate": 5.49210140856764e-06,
      "loss": 0.0584,
      "step": 49520
    },
    {
      "epoch": 4.72259521484375e-05,
      "model_forward_time": 1.7140824794769287,
      "step": 49520
    },
    {
      "dynamics_loss_avg": 0.050248862244188786,
      "epoch": 4.72259521484375e-05,
      "step": 49520
    },
    {
      "action_loss_avg": 0.0024543712963350115,
      "epoch": 4.72259521484375e-05,
      "step": 49520
    },
    {
      "epoch": 4.72259521484375e-05,
      "step": 49520,
      "training_step_time": 8.290173530578613
    },
    {
      "epoch": 4.722690582275391e-05,
      "model_forward_time": 1.7318196296691895,
      "step": 49521
    },
    {
      "epoch": 4.722690582275391e-05,
      "step": 49521,
      "training_step_time": 7.721508264541626
    },
    {
      "epoch": 4.722785949707031e-05,
      "model_forward_time": 1.724231243133545,
      "step": 49522
    },
    {
      "epoch": 4.722785949707031e-05,
      "step": 49522,
      "training_step_time": 7.730318069458008
    },
    {
      "epoch": 4.722881317138672e-05,
      "model_forward_time": 2.336784601211548,
      "step": 49523
    },
    {
      "epoch": 4.722881317138672e-05,
      "step": 49523,
      "training_step_time": 7.71639347076416
    },
    {
      "epoch": 4.7229766845703126e-05,
      "model_forward_time": 1.732062578201294,
      "step": 49524
    },
    {
      "epoch": 4.7229766845703126e-05,
      "step": 49524,
      "training_step_time": 8.766315460205078
    },
    {
      "epoch": 4.723072052001953e-05,
      "model_forward_time": 1.7272708415985107,
      "step": 49525
    },
    {
      "epoch": 4.723072052001953e-05,
      "step": 49525,
      "training_step_time": 7.370660781860352
    },
    {
      "epoch": 4.723167419433594e-05,
      "model_forward_time": 1.7235691547393799,
      "step": 49526
    },
    {
      "epoch": 4.723167419433594e-05,
      "step": 49526,
      "training_step_time": 7.4874279499053955
    },
    {
      "epoch": 4.7232627868652345e-05,
      "model_forward_time": 1.7266287803649902,
      "step": 49527
    },
    {
      "epoch": 4.7232627868652345e-05,
      "step": 49527,
      "training_step_time": 7.59869647026062
    },
    {
      "epoch": 4.723358154296875e-05,
      "model_forward_time": 1.732801914215088,
      "step": 49528
    },
    {
      "epoch": 4.723358154296875e-05,
      "step": 49528,
      "training_step_time": 8.560550212860107
    },
    {
      "epoch": 4.723453521728516e-05,
      "model_forward_time": 1.7240631580352783,
      "step": 49529
    },
    {
      "epoch": 4.723453521728516e-05,
      "step": 49529,
      "training_step_time": 8.680453062057495
    },
    {
      "epoch": 4.7235488891601564e-05,
      "grad_norm": 0.11651845276355743,
      "learning_rate": 5.490455939570265e-06,
      "loss": 0.0645,
      "step": 49530
    },
    {
      "epoch": 4.7235488891601564e-05,
      "model_forward_time": 1.7151758670806885,
      "step": 49530
    },
    {
      "dynamics_loss_avg": 0.04588477145880461,
      "epoch": 4.7235488891601564e-05,
      "step": 49530
    },
    {
      "action_loss_avg": 0.0015235874510835857,
      "epoch": 4.7235488891601564e-05,
      "step": 49530
    },
    {
      "epoch": 4.7235488891601564e-05,
      "step": 49530,
      "training_step_time": 7.668489933013916
    },
    {
      "epoch": 4.723644256591797e-05,
      "model_forward_time": 1.7174310684204102,
      "step": 49531
    },
    {
      "epoch": 4.723644256591797e-05,
      "step": 49531,
      "training_step_time": 9.528874397277832
    },
    {
      "epoch": 4.723739624023437e-05,
      "model_forward_time": 2.295114278793335,
      "step": 49532
    },
    {
      "epoch": 4.723739624023437e-05,
      "step": 49532,
      "training_step_time": 7.749496936798096
    },
    {
      "epoch": 4.723834991455078e-05,
      "model_forward_time": 1.7216174602508545,
      "step": 49533
    },
    {
      "epoch": 4.723834991455078e-05,
      "step": 49533,
      "training_step_time": 8.413569688796997
    },
    {
      "epoch": 4.7239303588867186e-05,
      "model_forward_time": 1.722956895828247,
      "step": 49534
    },
    {
      "epoch": 4.7239303588867186e-05,
      "step": 49534,
      "training_step_time": 8.527804613113403
    },
    {
      "epoch": 4.724025726318359e-05,
      "model_forward_time": 1.7256734371185303,
      "step": 49535
    },
    {
      "epoch": 4.724025726318359e-05,
      "step": 49535,
      "training_step_time": 7.380378484725952
    },
    {
      "epoch": 4.72412109375e-05,
      "model_forward_time": 1.721634864807129,
      "step": 49536
    },
    {
      "epoch": 4.72412109375e-05,
      "step": 49536,
      "training_step_time": 7.593607664108276
    },
    {
      "epoch": 4.7242164611816405e-05,
      "model_forward_time": 1.7262721061706543,
      "step": 49537
    },
    {
      "epoch": 4.7242164611816405e-05,
      "step": 49537,
      "training_step_time": 8.233092784881592
    },
    {
      "epoch": 4.724311828613281e-05,
      "model_forward_time": 1.7299096584320068,
      "step": 49538
    },
    {
      "epoch": 4.724311828613281e-05,
      "step": 49538,
      "training_step_time": 8.344192504882812
    },
    {
      "epoch": 4.724407196044922e-05,
      "model_forward_time": 1.7181549072265625,
      "step": 49539
    },
    {
      "epoch": 4.724407196044922e-05,
      "step": 49539,
      "training_step_time": 8.353328704833984
    },
    {
      "epoch": 4.7245025634765624e-05,
      "grad_norm": 0.14442074298858643,
      "learning_rate": 5.488810416937364e-06,
      "loss": 0.0617,
      "step": 49540
    },
    {
      "epoch": 4.7245025634765624e-05,
      "model_forward_time": 1.7151689529418945,
      "step": 49540
    },
    {
      "dynamics_loss_avg": 0.06903073396533728,
      "epoch": 4.7245025634765624e-05,
      "step": 49540
    },
    {
      "action_loss_avg": 0.0014462646387983114,
      "epoch": 4.7245025634765624e-05,
      "step": 49540
    },
    {
      "epoch": 4.7245025634765624e-05,
      "step": 49540,
      "training_step_time": 8.570626020431519
    },
    {
      "epoch": 4.724597930908203e-05,
      "model_forward_time": 2.301670551300049,
      "step": 49541
    },
    {
      "epoch": 4.724597930908203e-05,
      "step": 49541,
      "training_step_time": 7.693665504455566
    },
    {
      "epoch": 4.724693298339844e-05,
      "model_forward_time": 1.7959799766540527,
      "step": 49542
    },
    {
      "epoch": 4.724693298339844e-05,
      "step": 49542,
      "training_step_time": 9.14516282081604
    },
    {
      "epoch": 4.724788665771484e-05,
      "model_forward_time": 1.7266592979431152,
      "step": 49543
    },
    {
      "epoch": 4.724788665771484e-05,
      "step": 49543,
      "training_step_time": 7.655553817749023
    },
    {
      "epoch": 4.724884033203125e-05,
      "model_forward_time": 1.7259435653686523,
      "step": 49544
    },
    {
      "epoch": 4.724884033203125e-05,
      "step": 49544,
      "training_step_time": 7.530032634735107
    },
    {
      "epoch": 4.724979400634766e-05,
      "model_forward_time": 1.7256190776824951,
      "step": 49545
    },
    {
      "epoch": 4.724979400634766e-05,
      "step": 49545,
      "training_step_time": 7.563921689987183
    },
    {
      "epoch": 4.725074768066406e-05,
      "model_forward_time": 1.7233388423919678,
      "step": 49546
    },
    {
      "epoch": 4.725074768066406e-05,
      "step": 49546,
      "training_step_time": 7.573653697967529
    },
    {
      "epoch": 4.7251701354980466e-05,
      "model_forward_time": 1.722437858581543,
      "step": 49547
    },
    {
      "epoch": 4.7251701354980466e-05,
      "step": 49547,
      "training_step_time": 8.352986574172974
    },
    {
      "epoch": 4.725265502929688e-05,
      "model_forward_time": 1.7244012355804443,
      "step": 49548
    },
    {
      "epoch": 4.725265502929688e-05,
      "step": 49548,
      "training_step_time": 7.63403844833374
    },
    {
      "epoch": 4.725360870361328e-05,
      "model_forward_time": 1.3440539836883545,
      "step": 49549
    },
    {
      "epoch": 4.725360870361328e-05,
      "step": 49549,
      "training_step_time": 8.371500253677368
    },
    {
      "epoch": 4.7254562377929685e-05,
      "grad_norm": 0.1757710874080658,
      "learning_rate": 5.487164840848893e-06,
      "loss": 0.0622,
      "step": 49550
    },
    {
      "epoch": 4.7254562377929685e-05,
      "model_forward_time": 2.2379720211029053,
      "step": 49550
    },
    {
      "dynamics_loss_avg": 0.042872222419828174,
      "epoch": 4.7254562377929685e-05,
      "step": 49550
    },
    {
      "action_loss_avg": 0.0016018131194869057,
      "epoch": 4.7254562377929685e-05,
      "step": 49550
    },
    {
      "epoch": 4.7254562377929685e-05,
      "step": 49550,
      "training_step_time": 8.582804918289185
    },
    {
      "epoch": 4.7255516052246096e-05,
      "model_forward_time": 1.775991439819336,
      "step": 49551
    },
    {
      "epoch": 4.7255516052246096e-05,
      "step": 49551,
      "training_step_time": 8.228978872299194
    },
    {
      "epoch": 4.72564697265625e-05,
      "model_forward_time": 1.7221746444702148,
      "step": 49552
    },
    {
      "epoch": 4.72564697265625e-05,
      "step": 49552,
      "training_step_time": 7.683839321136475
    },
    {
      "epoch": 4.7257423400878904e-05,
      "model_forward_time": 1.7177529335021973,
      "step": 49553
    },
    {
      "epoch": 4.7257423400878904e-05,
      "step": 49553,
      "training_step_time": 8.371477365493774
    },
    {
      "epoch": 4.7258377075195314e-05,
      "model_forward_time": 1.7250230312347412,
      "step": 49554
    },
    {
      "epoch": 4.7258377075195314e-05,
      "step": 49554,
      "training_step_time": 7.575518369674683
    },
    {
      "epoch": 4.725933074951172e-05,
      "model_forward_time": 1.7306125164031982,
      "step": 49555
    },
    {
      "epoch": 4.725933074951172e-05,
      "step": 49555,
      "training_step_time": 7.619280576705933
    },
    {
      "epoch": 4.726028442382812e-05,
      "model_forward_time": 1.7230091094970703,
      "step": 49556
    },
    {
      "epoch": 4.726028442382812e-05,
      "step": 49556,
      "training_step_time": 7.790709495544434
    },
    {
      "epoch": 4.726123809814453e-05,
      "model_forward_time": 1.7249937057495117,
      "step": 49557
    },
    {
      "epoch": 4.726123809814453e-05,
      "step": 49557,
      "training_step_time": 8.138017654418945
    },
    {
      "epoch": 4.726219177246094e-05,
      "model_forward_time": 1.7281179428100586,
      "step": 49558
    },
    {
      "epoch": 4.726219177246094e-05,
      "step": 49558,
      "training_step_time": 7.678608417510986
    },
    {
      "epoch": 4.726314544677734e-05,
      "model_forward_time": 1.721865177154541,
      "step": 49559
    },
    {
      "epoch": 4.726314544677734e-05,
      "step": 49559,
      "training_step_time": 7.754428386688232
    },
    {
      "epoch": 4.726409912109375e-05,
      "grad_norm": 0.11002035439014435,
      "learning_rate": 5.485519211484807e-06,
      "loss": 0.0617,
      "step": 49560
    },
    {
      "epoch": 4.726409912109375e-05,
      "model_forward_time": 2.3490302562713623,
      "step": 49560
    },
    {
      "dynamics_loss_avg": 0.056333750300109384,
      "epoch": 4.726409912109375e-05,
      "step": 49560
    },
    {
      "action_loss_avg": 0.00117913301801309,
      "epoch": 4.726409912109375e-05,
      "step": 49560
    },
    {
      "epoch": 4.726409912109375e-05,
      "step": 49560,
      "training_step_time": 7.761308670043945
    },
    {
      "epoch": 4.7265052795410156e-05,
      "model_forward_time": 1.724921703338623,
      "step": 49561
    },
    {
      "epoch": 4.7265052795410156e-05,
      "step": 49561,
      "training_step_time": 7.776776552200317
    },
    {
      "epoch": 4.726600646972656e-05,
      "model_forward_time": 1.7181751728057861,
      "step": 49562
    },
    {
      "epoch": 4.726600646972656e-05,
      "step": 49562,
      "training_step_time": 7.746062517166138
    },
    {
      "epoch": 4.726696014404297e-05,
      "model_forward_time": 1.7248668670654297,
      "step": 49563
    },
    {
      "epoch": 4.726696014404297e-05,
      "step": 49563,
      "training_step_time": 7.6218202114105225
    },
    {
      "epoch": 4.7267913818359375e-05,
      "model_forward_time": 1.730229377746582,
      "step": 49564
    },
    {
      "epoch": 4.7267913818359375e-05,
      "step": 49564,
      "training_step_time": 7.685530662536621
    },
    {
      "epoch": 4.726886749267578e-05,
      "model_forward_time": 1.7194747924804688,
      "step": 49565
    },
    {
      "epoch": 4.726886749267578e-05,
      "step": 49565,
      "training_step_time": 7.76396369934082
    },
    {
      "epoch": 4.726982116699219e-05,
      "model_forward_time": 1.7203443050384521,
      "step": 49566
    },
    {
      "epoch": 4.726982116699219e-05,
      "step": 49566,
      "training_step_time": 7.594936847686768
    },
    {
      "epoch": 4.7270774841308594e-05,
      "model_forward_time": 1.7254140377044678,
      "step": 49567
    },
    {
      "epoch": 4.7270774841308594e-05,
      "step": 49567,
      "training_step_time": 8.31986117362976
    },
    {
      "epoch": 4.7271728515625e-05,
      "model_forward_time": 1.724360704421997,
      "step": 49568
    },
    {
      "epoch": 4.7271728515625e-05,
      "step": 49568,
      "training_step_time": 7.714899301528931
    },
    {
      "epoch": 4.727268218994141e-05,
      "model_forward_time": 2.3183066844940186,
      "step": 49569
    },
    {
      "epoch": 4.727268218994141e-05,
      "step": 49569,
      "training_step_time": 9.37756061553955
    },
    {
      "epoch": 4.727363586425781e-05,
      "grad_norm": 0.1039012148976326,
      "learning_rate": 5.483873529025068e-06,
      "loss": 0.0612,
      "step": 49570
    },
    {
      "epoch": 4.727363586425781e-05,
      "model_forward_time": 1.7165193557739258,
      "step": 49570
    },
    {
      "dynamics_loss_avg": 0.04292056718841195,
      "epoch": 4.727363586425781e-05,
      "step": 49570
    },
    {
      "action_loss_avg": 0.0014061340305488556,
      "epoch": 4.727363586425781e-05,
      "step": 49570
    },
    {
      "epoch": 4.727363586425781e-05,
      "step": 49570,
      "training_step_time": 7.77481746673584
    },
    {
      "epoch": 4.727458953857422e-05,
      "model_forward_time": 1.723606824874878,
      "step": 49571
    },
    {
      "epoch": 4.727458953857422e-05,
      "step": 49571,
      "training_step_time": 7.756545543670654
    },
    {
      "epoch": 4.727554321289063e-05,
      "model_forward_time": 1.7251560688018799,
      "step": 49572
    },
    {
      "epoch": 4.727554321289063e-05,
      "step": 49572,
      "training_step_time": 8.603415966033936
    },
    {
      "epoch": 4.727649688720703e-05,
      "model_forward_time": 1.7232332229614258,
      "step": 49573
    },
    {
      "epoch": 4.727649688720703e-05,
      "step": 49573,
      "training_step_time": 7.578595876693726
    },
    {
      "epoch": 4.7277450561523436e-05,
      "model_forward_time": 1.7188334465026855,
      "step": 49574
    },
    {
      "epoch": 4.7277450561523436e-05,
      "step": 49574,
      "training_step_time": 7.63902735710144
    },
    {
      "epoch": 4.7278404235839846e-05,
      "model_forward_time": 1.7360131740570068,
      "step": 49575
    },
    {
      "epoch": 4.7278404235839846e-05,
      "step": 49575,
      "training_step_time": 7.5575408935546875
    },
    {
      "epoch": 4.727935791015625e-05,
      "model_forward_time": 1.73726487159729,
      "step": 49576
    },
    {
      "epoch": 4.727935791015625e-05,
      "step": 49576,
      "training_step_time": 8.226000308990479
    },
    {
      "epoch": 4.7280311584472655e-05,
      "model_forward_time": 1.715054988861084,
      "step": 49577
    },
    {
      "epoch": 4.7280311584472655e-05,
      "step": 49577,
      "training_step_time": 7.825777292251587
    },
    {
      "epoch": 4.7281265258789065e-05,
      "model_forward_time": 2.2854933738708496,
      "step": 49578
    },
    {
      "epoch": 4.7281265258789065e-05,
      "step": 49578,
      "training_step_time": 9.294277429580688
    },
    {
      "epoch": 4.728221893310547e-05,
      "model_forward_time": 1.8047583103179932,
      "step": 49579
    },
    {
      "epoch": 4.728221893310547e-05,
      "step": 49579,
      "training_step_time": 8.236915588378906
    },
    {
      "epoch": 4.728317260742187e-05,
      "grad_norm": 0.09273247420787811,
      "learning_rate": 5.482227793649649e-06,
      "loss": 0.0582,
      "step": 49580
    },
    {
      "epoch": 4.728317260742187e-05,
      "model_forward_time": 1.7423150539398193,
      "step": 49580
    },
    {
      "dynamics_loss_avg": 0.04914348684251309,
      "epoch": 4.728317260742187e-05,
      "step": 49580
    },
    {
      "action_loss_avg": 0.0020593176712282004,
      "epoch": 4.728317260742187e-05,
      "step": 49580
    },
    {
      "epoch": 4.728317260742187e-05,
      "step": 49580,
      "training_step_time": 7.651555299758911
    },
    {
      "epoch": 4.7284126281738284e-05,
      "model_forward_time": 1.735147476196289,
      "step": 49581
    },
    {
      "epoch": 4.7284126281738284e-05,
      "step": 49581,
      "training_step_time": 8.282115697860718
    },
    {
      "epoch": 4.728507995605469e-05,
      "model_forward_time": 1.7312235832214355,
      "step": 49582
    },
    {
      "epoch": 4.728507995605469e-05,
      "step": 49582,
      "training_step_time": 7.52810001373291
    },
    {
      "epoch": 4.728603363037109e-05,
      "model_forward_time": 1.7375588417053223,
      "step": 49583
    },
    {
      "epoch": 4.728603363037109e-05,
      "step": 49583,
      "training_step_time": 8.20748233795166
    },
    {
      "epoch": 4.72869873046875e-05,
      "model_forward_time": 1.729339838027954,
      "step": 49584
    },
    {
      "epoch": 4.72869873046875e-05,
      "step": 49584,
      "training_step_time": 8.375256299972534
    },
    {
      "epoch": 4.728794097900391e-05,
      "model_forward_time": 1.7371416091918945,
      "step": 49585
    },
    {
      "epoch": 4.728794097900391e-05,
      "step": 49585,
      "training_step_time": 7.6071457862854
    },
    {
      "epoch": 4.728889465332031e-05,
      "model_forward_time": 1.7559726238250732,
      "step": 49586
    },
    {
      "epoch": 4.728889465332031e-05,
      "step": 49586,
      "training_step_time": 8.776086568832397
    },
    {
      "epoch": 4.728984832763672e-05,
      "model_forward_time": 2.2941906452178955,
      "step": 49587
    },
    {
      "epoch": 4.728984832763672e-05,
      "step": 49587,
      "training_step_time": 7.7971251010894775
    },
    {
      "epoch": 4.7290802001953126e-05,
      "model_forward_time": 1.8131842613220215,
      "step": 49588
    },
    {
      "epoch": 4.7290802001953126e-05,
      "step": 49588,
      "training_step_time": 8.718168020248413
    },
    {
      "epoch": 4.729175567626953e-05,
      "model_forward_time": 1.7323367595672607,
      "step": 49589
    },
    {
      "epoch": 4.729175567626953e-05,
      "step": 49589,
      "training_step_time": 8.627596616744995
    },
    {
      "epoch": 4.729270935058594e-05,
      "grad_norm": 0.11519668996334076,
      "learning_rate": 5.4805820055385236e-06,
      "loss": 0.0595,
      "step": 49590
    },
    {
      "epoch": 4.729270935058594e-05,
      "model_forward_time": 1.739293098449707,
      "step": 49590
    },
    {
      "dynamics_loss_avg": 0.06320575680583715,
      "epoch": 4.729270935058594e-05,
      "step": 49590
    },
    {
      "action_loss_avg": 0.0014509431261103601,
      "epoch": 4.729270935058594e-05,
      "step": 49590
    },
    {
      "epoch": 4.729270935058594e-05,
      "step": 49590,
      "training_step_time": 7.632741451263428
    },
    {
      "epoch": 4.7293663024902345e-05,
      "model_forward_time": 1.7406349182128906,
      "step": 49591
    },
    {
      "epoch": 4.7293663024902345e-05,
      "step": 49591,
      "training_step_time": 7.61382269859314
    },
    {
      "epoch": 4.729461669921875e-05,
      "model_forward_time": 1.731358289718628,
      "step": 49592
    },
    {
      "epoch": 4.729461669921875e-05,
      "step": 49592,
      "training_step_time": 7.716131210327148
    },
    {
      "epoch": 4.729557037353516e-05,
      "model_forward_time": 1.7396056652069092,
      "step": 49593
    },
    {
      "epoch": 4.729557037353516e-05,
      "step": 49593,
      "training_step_time": 7.6691954135894775
    },
    {
      "epoch": 4.7296524047851564e-05,
      "model_forward_time": 1.7317838668823242,
      "step": 49594
    },
    {
      "epoch": 4.7296524047851564e-05,
      "step": 49594,
      "training_step_time": 7.567725658416748
    },
    {
      "epoch": 4.729747772216797e-05,
      "model_forward_time": 1.734154462814331,
      "step": 49595
    },
    {
      "epoch": 4.729747772216797e-05,
      "step": 49595,
      "training_step_time": 7.661994457244873
    },
    {
      "epoch": 4.729843139648437e-05,
      "model_forward_time": 2.2975668907165527,
      "step": 49596
    },
    {
      "epoch": 4.729843139648437e-05,
      "step": 49596,
      "training_step_time": 7.68857479095459
    },
    {
      "epoch": 4.729938507080078e-05,
      "model_forward_time": 1.782043218612671,
      "step": 49597
    },
    {
      "epoch": 4.729938507080078e-05,
      "step": 49597,
      "training_step_time": 7.653834819793701
    },
    {
      "epoch": 4.7300338745117187e-05,
      "model_forward_time": 1.7299885749816895,
      "step": 49598
    },
    {
      "epoch": 4.7300338745117187e-05,
      "step": 49598,
      "training_step_time": 7.699817180633545
    },
    {
      "epoch": 4.730129241943359e-05,
      "model_forward_time": 1.723646640777588,
      "step": 49599
    },
    {
      "epoch": 4.730129241943359e-05,
      "step": 49599,
      "training_step_time": 7.606733798980713
    },
    {
      "epoch": 4.730224609375e-05,
      "grad_norm": 0.11106213927268982,
      "learning_rate": 5.478936164871671e-06,
      "loss": 0.061,
      "step": 49600
    },
    {
      "epoch": 4.730224609375e-05,
      "model_forward_time": 1.7622010707855225,
      "step": 49600
    },
    {
      "dynamics_loss_avg": 0.0630403988994658,
      "epoch": 4.730224609375e-05,
      "step": 49600
    },
    {
      "action_loss_avg": 0.001448277337476611,
      "epoch": 4.730224609375e-05,
      "step": 49600
    },
    {
      "epoch": 4.730224609375e-05,
      "step": 49600,
      "training_step_time": 7.5007102489471436
    },
    {
      "epoch": 4.7303199768066405e-05,
      "model_forward_time": 1.7314865589141846,
      "step": 49601
    },
    {
      "epoch": 4.7303199768066405e-05,
      "step": 49601,
      "training_step_time": 7.601521730422974
    },
    {
      "epoch": 4.730415344238281e-05,
      "model_forward_time": 1.7325482368469238,
      "step": 49602
    },
    {
      "epoch": 4.730415344238281e-05,
      "step": 49602,
      "training_step_time": 9.421653747558594
    },
    {
      "epoch": 4.730510711669922e-05,
      "model_forward_time": 1.7302868366241455,
      "step": 49603
    },
    {
      "epoch": 4.730510711669922e-05,
      "step": 49603,
      "training_step_time": 8.57558298110962
    },
    {
      "epoch": 4.7306060791015624e-05,
      "model_forward_time": 1.73305082321167,
      "step": 49604
    },
    {
      "epoch": 4.7306060791015624e-05,
      "step": 49604,
      "training_step_time": 8.30520510673523
    },
    {
      "epoch": 4.730701446533203e-05,
      "model_forward_time": 1.7333378791809082,
      "step": 49605
    },
    {
      "epoch": 4.730701446533203e-05,
      "step": 49605,
      "training_step_time": 7.68133544921875
    },
    {
      "epoch": 4.730796813964844e-05,
      "model_forward_time": 2.3280694484710693,
      "step": 49606
    },
    {
      "epoch": 4.730796813964844e-05,
      "step": 49606,
      "training_step_time": 8.391427755355835
    },
    {
      "epoch": 4.730892181396484e-05,
      "model_forward_time": 1.735614538192749,
      "step": 49607
    },
    {
      "epoch": 4.730892181396484e-05,
      "step": 49607,
      "training_step_time": 7.650554656982422
    },
    {
      "epoch": 4.730987548828125e-05,
      "model_forward_time": 1.7304883003234863,
      "step": 49608
    },
    {
      "epoch": 4.730987548828125e-05,
      "step": 49608,
      "training_step_time": 7.5765297412872314
    },
    {
      "epoch": 4.731082916259766e-05,
      "model_forward_time": 1.7065637111663818,
      "step": 49609
    },
    {
      "epoch": 4.731082916259766e-05,
      "step": 49609,
      "training_step_time": 7.556942701339722
    },
    {
      "epoch": 4.731178283691406e-05,
      "grad_norm": 0.08906393498182297,
      "learning_rate": 5.477290271829081e-06,
      "loss": 0.0576,
      "step": 49610
    },
    {
      "epoch": 4.731178283691406e-05,
      "model_forward_time": 1.7127306461334229,
      "step": 49610
    },
    {
      "dynamics_loss_avg": 0.061726159602403644,
      "epoch": 4.731178283691406e-05,
      "step": 49610
    },
    {
      "action_loss_avg": 0.0017320665472652763,
      "epoch": 4.731178283691406e-05,
      "step": 49610
    },
    {
      "epoch": 4.731178283691406e-05,
      "step": 49610,
      "training_step_time": 8.564533948898315
    },
    {
      "epoch": 4.7312736511230466e-05,
      "model_forward_time": 1.7192659378051758,
      "step": 49611
    },
    {
      "epoch": 4.7312736511230466e-05,
      "step": 49611,
      "training_step_time": 7.711277723312378
    },
    {
      "epoch": 4.731369018554688e-05,
      "model_forward_time": 1.7205605506896973,
      "step": 49612
    },
    {
      "epoch": 4.731369018554688e-05,
      "step": 49612,
      "training_step_time": 7.645133018493652
    },
    {
      "epoch": 4.731464385986328e-05,
      "model_forward_time": 1.7254014015197754,
      "step": 49613
    },
    {
      "epoch": 4.731464385986328e-05,
      "step": 49613,
      "training_step_time": 8.816526412963867
    },
    {
      "epoch": 4.7315597534179685e-05,
      "model_forward_time": 1.7200350761413574,
      "step": 49614
    },
    {
      "epoch": 4.7315597534179685e-05,
      "step": 49614,
      "training_step_time": 8.254889249801636
    },
    {
      "epoch": 4.7316551208496096e-05,
      "model_forward_time": 2.2660415172576904,
      "step": 49615
    },
    {
      "epoch": 4.7316551208496096e-05,
      "step": 49615,
      "training_step_time": 8.481332302093506
    },
    {
      "epoch": 4.73175048828125e-05,
      "model_forward_time": 1.7198221683502197,
      "step": 49616
    },
    {
      "epoch": 4.73175048828125e-05,
      "step": 49616,
      "training_step_time": 7.760459661483765
    },
    {
      "epoch": 4.7318458557128904e-05,
      "model_forward_time": 1.7249112129211426,
      "step": 49617
    },
    {
      "epoch": 4.7318458557128904e-05,
      "step": 49617,
      "training_step_time": 8.198277950286865
    },
    {
      "epoch": 4.7319412231445315e-05,
      "model_forward_time": 1.7266716957092285,
      "step": 49618
    },
    {
      "epoch": 4.7319412231445315e-05,
      "step": 49618,
      "training_step_time": 8.466923236846924
    },
    {
      "epoch": 4.732036590576172e-05,
      "model_forward_time": 1.7231967449188232,
      "step": 49619
    },
    {
      "epoch": 4.732036590576172e-05,
      "step": 49619,
      "training_step_time": 7.577164173126221
    },
    {
      "epoch": 4.732131958007812e-05,
      "grad_norm": 0.10327982902526855,
      "learning_rate": 5.475644326590744e-06,
      "loss": 0.0594,
      "step": 49620
    },
    {
      "epoch": 4.732131958007812e-05,
      "model_forward_time": 1.7105302810668945,
      "step": 49620
    },
    {
      "dynamics_loss_avg": 0.04453539121896029,
      "epoch": 4.732131958007812e-05,
      "step": 49620
    },
    {
      "action_loss_avg": 0.0014586957928258925,
      "epoch": 4.732131958007812e-05,
      "step": 49620
    },
    {
      "epoch": 4.732131958007812e-05,
      "step": 49620,
      "training_step_time": 8.182531356811523
    },
    {
      "epoch": 4.732227325439453e-05,
      "model_forward_time": 1.7275974750518799,
      "step": 49621
    },
    {
      "epoch": 4.732227325439453e-05,
      "step": 49621,
      "training_step_time": 7.664742469787598
    },
    {
      "epoch": 4.732322692871094e-05,
      "model_forward_time": 1.7196989059448242,
      "step": 49622
    },
    {
      "epoch": 4.732322692871094e-05,
      "step": 49622,
      "training_step_time": 7.76892614364624
    },
    {
      "epoch": 4.732418060302734e-05,
      "model_forward_time": 1.7195630073547363,
      "step": 49623
    },
    {
      "epoch": 4.732418060302734e-05,
      "step": 49623,
      "training_step_time": 8.9724280834198
    },
    {
      "epoch": 4.732513427734375e-05,
      "model_forward_time": 2.276655673980713,
      "step": 49624
    },
    {
      "epoch": 4.732513427734375e-05,
      "step": 49624,
      "training_step_time": 8.314885377883911
    },
    {
      "epoch": 4.7326087951660156e-05,
      "model_forward_time": 1.7851369380950928,
      "step": 49625
    },
    {
      "epoch": 4.7326087951660156e-05,
      "step": 49625,
      "training_step_time": 9.517588138580322
    },
    {
      "epoch": 4.732704162597656e-05,
      "model_forward_time": 1.7180287837982178,
      "step": 49626
    },
    {
      "epoch": 4.732704162597656e-05,
      "step": 49626,
      "training_step_time": 7.759384870529175
    },
    {
      "epoch": 4.732799530029297e-05,
      "model_forward_time": 1.7222084999084473,
      "step": 49627
    },
    {
      "epoch": 4.732799530029297e-05,
      "step": 49627,
      "training_step_time": 7.653242588043213
    },
    {
      "epoch": 4.7328948974609375e-05,
      "model_forward_time": 1.7218005657196045,
      "step": 49628
    },
    {
      "epoch": 4.7328948974609375e-05,
      "step": 49628,
      "training_step_time": 7.768336057662964
    },
    {
      "epoch": 4.732990264892578e-05,
      "model_forward_time": 1.7213501930236816,
      "step": 49629
    },
    {
      "epoch": 4.732990264892578e-05,
      "step": 49629,
      "training_step_time": 7.831529855728149
    },
    {
      "epoch": 4.733085632324219e-05,
      "grad_norm": 0.3889705538749695,
      "learning_rate": 5.473998329336658e-06,
      "loss": 0.0566,
      "step": 49630
    },
    {
      "epoch": 4.733085632324219e-05,
      "model_forward_time": 1.711906909942627,
      "step": 49630
    },
    {
      "dynamics_loss_avg": 0.05744669111445546,
      "epoch": 4.733085632324219e-05,
      "step": 49630
    },
    {
      "action_loss_avg": 0.0017617212899494917,
      "epoch": 4.733085632324219e-05,
      "step": 49630
    },
    {
      "epoch": 4.733085632324219e-05,
      "step": 49630,
      "training_step_time": 8.766811609268188
    },
    {
      "epoch": 4.7331809997558594e-05,
      "model_forward_time": 1.7176353931427002,
      "step": 49631
    },
    {
      "epoch": 4.7331809997558594e-05,
      "step": 49631,
      "training_step_time": 7.7404985427856445
    },
    {
      "epoch": 4.7332763671875e-05,
      "model_forward_time": 1.723123550415039,
      "step": 49632
    },
    {
      "epoch": 4.7332763671875e-05,
      "step": 49632,
      "training_step_time": 8.838255167007446
    },
    {
      "epoch": 4.733371734619141e-05,
      "model_forward_time": 2.3170878887176514,
      "step": 49633
    },
    {
      "epoch": 4.733371734619141e-05,
      "step": 49633,
      "training_step_time": 9.344439268112183
    },
    {
      "epoch": 4.733467102050781e-05,
      "model_forward_time": 1.7486250400543213,
      "step": 49634
    },
    {
      "epoch": 4.733467102050781e-05,
      "step": 49634,
      "training_step_time": 8.931159257888794
    },
    {
      "epoch": 4.733562469482422e-05,
      "model_forward_time": 1.7101690769195557,
      "step": 49635
    },
    {
      "epoch": 4.733562469482422e-05,
      "step": 49635,
      "training_step_time": 7.788553476333618
    },
    {
      "epoch": 4.733657836914063e-05,
      "model_forward_time": 1.719402551651001,
      "step": 49636
    },
    {
      "epoch": 4.733657836914063e-05,
      "step": 49636,
      "training_step_time": 7.784689903259277
    },
    {
      "epoch": 4.733753204345703e-05,
      "model_forward_time": 1.7313318252563477,
      "step": 49637
    },
    {
      "epoch": 4.733753204345703e-05,
      "step": 49637,
      "training_step_time": 9.556932926177979
    },
    {
      "epoch": 4.7338485717773436e-05,
      "model_forward_time": 1.7184245586395264,
      "step": 49638
    },
    {
      "epoch": 4.7338485717773436e-05,
      "step": 49638,
      "training_step_time": 7.830259323120117
    },
    {
      "epoch": 4.7339439392089847e-05,
      "model_forward_time": 1.7232706546783447,
      "step": 49639
    },
    {
      "epoch": 4.7339439392089847e-05,
      "step": 49639,
      "training_step_time": 7.769455432891846
    },
    {
      "epoch": 4.734039306640625e-05,
      "grad_norm": 0.15116329491138458,
      "learning_rate": 5.472352280246829e-06,
      "loss": 0.0587,
      "step": 49640
    },
    {
      "epoch": 4.734039306640625e-05,
      "model_forward_time": 1.715467929840088,
      "step": 49640
    },
    {
      "dynamics_loss_avg": 0.043419108353555204,
      "epoch": 4.734039306640625e-05,
      "step": 49640
    },
    {
      "action_loss_avg": 0.0016244085083599203,
      "epoch": 4.734039306640625e-05,
      "step": 49640
    },
    {
      "epoch": 4.734039306640625e-05,
      "step": 49640,
      "training_step_time": 8.697031736373901
    },
    {
      "epoch": 4.7341346740722655e-05,
      "model_forward_time": 1.7209622859954834,
      "step": 49641
    },
    {
      "epoch": 4.7341346740722655e-05,
      "step": 49641,
      "training_step_time": 7.795177698135376
    },
    {
      "epoch": 4.7342300415039065e-05,
      "model_forward_time": 2.2863404750823975,
      "step": 49642
    },
    {
      "epoch": 4.7342300415039065e-05,
      "step": 49642,
      "training_step_time": 8.37697958946228
    },
    {
      "epoch": 4.734325408935547e-05,
      "model_forward_time": 1.7169561386108398,
      "step": 49643
    },
    {
      "epoch": 4.734325408935547e-05,
      "step": 49643,
      "training_step_time": 8.79936671257019
    },
    {
      "epoch": 4.7344207763671873e-05,
      "model_forward_time": 1.7250070571899414,
      "step": 49644
    },
    {
      "epoch": 4.7344207763671873e-05,
      "step": 49644,
      "training_step_time": 7.813822507858276
    },
    {
      "epoch": 4.7345161437988284e-05,
      "model_forward_time": 1.7152955532073975,
      "step": 49645
    },
    {
      "epoch": 4.7345161437988284e-05,
      "step": 49645,
      "training_step_time": 7.883708953857422
    },
    {
      "epoch": 4.734611511230469e-05,
      "model_forward_time": 1.7186880111694336,
      "step": 49646
    },
    {
      "epoch": 4.734611511230469e-05,
      "step": 49646,
      "training_step_time": 7.753398656845093
    },
    {
      "epoch": 4.734706878662109e-05,
      "model_forward_time": 1.729156255722046,
      "step": 49647
    },
    {
      "epoch": 4.734706878662109e-05,
      "step": 49647,
      "training_step_time": 7.825246810913086
    },
    {
      "epoch": 4.73480224609375e-05,
      "model_forward_time": 1.715872049331665,
      "step": 49648
    },
    {
      "epoch": 4.73480224609375e-05,
      "step": 49648,
      "training_step_time": 7.864847421646118
    },
    {
      "epoch": 4.734897613525391e-05,
      "model_forward_time": 1.7240588665008545,
      "step": 49649
    },
    {
      "epoch": 4.734897613525391e-05,
      "step": 49649,
      "training_step_time": 8.328046798706055
    },
    {
      "epoch": 4.734992980957031e-05,
      "grad_norm": 0.1331363469362259,
      "learning_rate": 5.4707061795012636e-06,
      "loss": 0.0581,
      "step": 49650
    },
    {
      "epoch": 4.734992980957031e-05,
      "model_forward_time": 1.7169556617736816,
      "step": 49650
    },
    {
      "dynamics_loss_avg": 0.0588338416069746,
      "epoch": 4.734992980957031e-05,
      "step": 49650
    },
    {
      "action_loss_avg": 0.0013628170752781443,
      "epoch": 4.734992980957031e-05,
      "step": 49650
    },
    {
      "epoch": 4.734992980957031e-05,
      "step": 49650,
      "training_step_time": 7.841997385025024
    },
    {
      "epoch": 4.735088348388672e-05,
      "model_forward_time": 1.7251057624816895,
      "step": 49651
    },
    {
      "epoch": 4.735088348388672e-05,
      "step": 49651,
      "training_step_time": 9.386225938796997
    },
    {
      "epoch": 4.7351837158203126e-05,
      "model_forward_time": 1.7187893390655518,
      "step": 49652
    },
    {
      "epoch": 4.7351837158203126e-05,
      "step": 49652,
      "training_step_time": 8.591395854949951
    },
    {
      "epoch": 4.735279083251953e-05,
      "model_forward_time": 1.7918579578399658,
      "step": 49653
    },
    {
      "epoch": 4.735279083251953e-05,
      "step": 49653,
      "training_step_time": 8.495733261108398
    },
    {
      "epoch": 4.735374450683594e-05,
      "model_forward_time": 1.7266130447387695,
      "step": 49654
    },
    {
      "epoch": 4.735374450683594e-05,
      "step": 49654,
      "training_step_time": 7.831076622009277
    },
    {
      "epoch": 4.7354698181152345e-05,
      "model_forward_time": 1.7145559787750244,
      "step": 49655
    },
    {
      "epoch": 4.7354698181152345e-05,
      "step": 49655,
      "training_step_time": 9.365830659866333
    },
    {
      "epoch": 4.735565185546875e-05,
      "model_forward_time": 1.7194342613220215,
      "step": 49656
    },
    {
      "epoch": 4.735565185546875e-05,
      "step": 49656,
      "training_step_time": 7.727674961090088
    },
    {
      "epoch": 4.735660552978515e-05,
      "model_forward_time": 1.7195305824279785,
      "step": 49657
    },
    {
      "epoch": 4.735660552978515e-05,
      "step": 49657,
      "training_step_time": 9.211510181427002
    },
    {
      "epoch": 4.7357559204101564e-05,
      "model_forward_time": 1.7210423946380615,
      "step": 49658
    },
    {
      "epoch": 4.7357559204101564e-05,
      "step": 49658,
      "training_step_time": 7.826210975646973
    },
    {
      "epoch": 4.735851287841797e-05,
      "model_forward_time": 1.7244391441345215,
      "step": 49659
    },
    {
      "epoch": 4.735851287841797e-05,
      "step": 49659,
      "training_step_time": 9.691748142242432
    },
    {
      "epoch": 4.735946655273437e-05,
      "grad_norm": 0.13530388474464417,
      "learning_rate": 5.469060027279977e-06,
      "loss": 0.0596,
      "step": 49660
    },
    {
      "epoch": 4.735946655273437e-05,
      "model_forward_time": 1.708439826965332,
      "step": 49660
    },
    {
      "dynamics_loss_avg": 0.05039874548092484,
      "epoch": 4.735946655273437e-05,
      "step": 49660
    },
    {
      "action_loss_avg": 0.0008978135447250679,
      "epoch": 4.735946655273437e-05,
      "step": 49660
    },
    {
      "epoch": 4.735946655273437e-05,
      "step": 49660,
      "training_step_time": 8.306843042373657
    },
    {
      "epoch": 4.736042022705078e-05,
      "model_forward_time": 2.2553822994232178,
      "step": 49661
    },
    {
      "epoch": 4.736042022705078e-05,
      "step": 49661,
      "training_step_time": 8.9436936378479
    },
    {
      "epoch": 4.7361373901367187e-05,
      "model_forward_time": 1.7701599597930908,
      "step": 49662
    },
    {
      "epoch": 4.7361373901367187e-05,
      "step": 49662,
      "training_step_time": 7.912315368652344
    },
    {
      "epoch": 4.736232757568359e-05,
      "model_forward_time": 1.7164511680603027,
      "step": 49663
    },
    {
      "epoch": 4.736232757568359e-05,
      "step": 49663,
      "training_step_time": 7.845196962356567
    },
    {
      "epoch": 4.736328125e-05,
      "model_forward_time": 1.721031665802002,
      "step": 49664
    },
    {
      "epoch": 4.736328125e-05,
      "step": 49664,
      "training_step_time": 8.548819780349731
    },
    {
      "epoch": 4.7364234924316405e-05,
      "model_forward_time": 1.7249999046325684,
      "step": 49665
    },
    {
      "epoch": 4.7364234924316405e-05,
      "step": 49665,
      "training_step_time": 7.851957559585571
    },
    {
      "epoch": 4.736518859863281e-05,
      "model_forward_time": 1.716193675994873,
      "step": 49666
    },
    {
      "epoch": 4.736518859863281e-05,
      "step": 49666,
      "training_step_time": 8.59381914138794
    },
    {
      "epoch": 4.736614227294922e-05,
      "model_forward_time": 1.7281630039215088,
      "step": 49667
    },
    {
      "epoch": 4.736614227294922e-05,
      "step": 49667,
      "training_step_time": 8.715377807617188
    },
    {
      "epoch": 4.7367095947265624e-05,
      "model_forward_time": 1.7160117626190186,
      "step": 49668
    },
    {
      "epoch": 4.7367095947265624e-05,
      "step": 49668,
      "training_step_time": 7.84137487411499
    },
    {
      "epoch": 4.736804962158203e-05,
      "model_forward_time": 1.723242998123169,
      "step": 49669
    },
    {
      "epoch": 4.736804962158203e-05,
      "step": 49669,
      "training_step_time": 7.857247829437256
    },
    {
      "epoch": 4.736900329589844e-05,
      "grad_norm": 0.12674511969089508,
      "learning_rate": 5.467413823762994e-06,
      "loss": 0.0627,
      "step": 49670
    },
    {
      "epoch": 4.736900329589844e-05,
      "model_forward_time": 2.261378765106201,
      "step": 49670
    },
    {
      "dynamics_loss_avg": 0.054016124457120895,
      "epoch": 4.736900329589844e-05,
      "step": 49670
    },
    {
      "action_loss_avg": 0.0015012631803983823,
      "epoch": 4.736900329589844e-05,
      "step": 49670
    },
    {
      "epoch": 4.736900329589844e-05,
      "step": 49670,
      "training_step_time": 8.94182014465332
    },
    {
      "epoch": 4.736995697021484e-05,
      "model_forward_time": 1.7423491477966309,
      "step": 49671
    },
    {
      "epoch": 4.736995697021484e-05,
      "step": 49671,
      "training_step_time": 8.441979169845581
    },
    {
      "epoch": 4.737091064453125e-05,
      "model_forward_time": 1.730823278427124,
      "step": 49672
    },
    {
      "epoch": 4.737091064453125e-05,
      "step": 49672,
      "training_step_time": 8.661275386810303
    },
    {
      "epoch": 4.737186431884766e-05,
      "model_forward_time": 1.712151288986206,
      "step": 49673
    },
    {
      "epoch": 4.737186431884766e-05,
      "step": 49673,
      "training_step_time": 7.754993677139282
    },
    {
      "epoch": 4.737281799316406e-05,
      "model_forward_time": 1.7158472537994385,
      "step": 49674
    },
    {
      "epoch": 4.737281799316406e-05,
      "step": 49674,
      "training_step_time": 9.546203374862671
    },
    {
      "epoch": 4.7373771667480466e-05,
      "model_forward_time": 1.7151947021484375,
      "step": 49675
    },
    {
      "epoch": 4.7373771667480466e-05,
      "step": 49675,
      "training_step_time": 9.123446464538574
    },
    {
      "epoch": 4.737472534179688e-05,
      "model_forward_time": 1.725278377532959,
      "step": 49676
    },
    {
      "epoch": 4.737472534179688e-05,
      "step": 49676,
      "training_step_time": 8.423445701599121
    },
    {
      "epoch": 4.737567901611328e-05,
      "model_forward_time": 1.717688798904419,
      "step": 49677
    },
    {
      "epoch": 4.737567901611328e-05,
      "step": 49677,
      "training_step_time": 9.17733097076416
    },
    {
      "epoch": 4.7376632690429685e-05,
      "model_forward_time": 1.7248916625976562,
      "step": 49678
    },
    {
      "epoch": 4.7376632690429685e-05,
      "step": 49678,
      "training_step_time": 8.855283260345459
    },
    {
      "epoch": 4.7377586364746096e-05,
      "model_forward_time": 2.279925584793091,
      "step": 49679
    },
    {
      "epoch": 4.7377586364746096e-05,
      "step": 49679,
      "training_step_time": 9.91530179977417
    },
    {
      "epoch": 4.73785400390625e-05,
      "grad_norm": 0.10499774664640427,
      "learning_rate": 5.465767569130336e-06,
      "loss": 0.0595,
      "step": 49680
    },
    {
      "epoch": 4.73785400390625e-05,
      "model_forward_time": 1.7277863025665283,
      "step": 49680
    },
    {
      "dynamics_loss_avg": 0.04373033549636603,
      "epoch": 4.73785400390625e-05,
      "step": 49680
    },
    {
      "action_loss_avg": 0.0015219079068629071,
      "epoch": 4.73785400390625e-05,
      "step": 49680
    },
    {
      "epoch": 4.73785400390625e-05,
      "step": 49680,
      "training_step_time": 7.987390995025635
    },
    {
      "epoch": 4.7379493713378904e-05,
      "model_forward_time": 1.7158913612365723,
      "step": 49681
    },
    {
      "epoch": 4.7379493713378904e-05,
      "step": 49681,
      "training_step_time": 7.738302946090698
    },
    {
      "epoch": 4.7380447387695315e-05,
      "model_forward_time": 1.7210521697998047,
      "step": 49682
    },
    {
      "epoch": 4.7380447387695315e-05,
      "step": 49682,
      "training_step_time": 7.766275405883789
    },
    {
      "epoch": 4.738140106201172e-05,
      "model_forward_time": 1.7140016555786133,
      "step": 49683
    },
    {
      "epoch": 4.738140106201172e-05,
      "step": 49683,
      "training_step_time": 8.060534238815308
    },
    {
      "epoch": 4.738235473632812e-05,
      "model_forward_time": 1.724815845489502,
      "step": 49684
    },
    {
      "epoch": 4.738235473632812e-05,
      "step": 49684,
      "training_step_time": 9.549732446670532
    },
    {
      "epoch": 4.7383308410644533e-05,
      "model_forward_time": 1.720733880996704,
      "step": 49685
    },
    {
      "epoch": 4.7383308410644533e-05,
      "step": 49685,
      "training_step_time": 8.204218864440918
    },
    {
      "epoch": 4.738426208496094e-05,
      "model_forward_time": 1.7239899635314941,
      "step": 49686
    },
    {
      "epoch": 4.738426208496094e-05,
      "step": 49686,
      "training_step_time": 7.653216361999512
    },
    {
      "epoch": 4.738521575927734e-05,
      "model_forward_time": 1.7194459438323975,
      "step": 49687
    },
    {
      "epoch": 4.738521575927734e-05,
      "step": 49687,
      "training_step_time": 8.66185450553894
    },
    {
      "epoch": 4.738616943359375e-05,
      "model_forward_time": 2.2932283878326416,
      "step": 49688
    },
    {
      "epoch": 4.738616943359375e-05,
      "step": 49688,
      "training_step_time": 7.654958009719849
    },
    {
      "epoch": 4.7387123107910156e-05,
      "model_forward_time": 1.7112541198730469,
      "step": 49689
    },
    {
      "epoch": 4.7387123107910156e-05,
      "step": 49689,
      "training_step_time": 8.240050315856934
    },
    {
      "epoch": 4.738807678222656e-05,
      "grad_norm": 0.1215064749121666,
      "learning_rate": 5.4641212635620365e-06,
      "loss": 0.0525,
      "step": 49690
    },
    {
      "epoch": 4.738807678222656e-05,
      "model_forward_time": 1.777256727218628,
      "step": 49690
    },
    {
      "dynamics_loss_avg": 0.03550703609362245,
      "epoch": 4.738807678222656e-05,
      "step": 49690
    },
    {
      "action_loss_avg": 0.001503438811050728,
      "epoch": 4.738807678222656e-05,
      "step": 49690
    },
    {
      "epoch": 4.738807678222656e-05,
      "step": 49690,
      "training_step_time": 7.702419996261597
    },
    {
      "epoch": 4.738903045654297e-05,
      "model_forward_time": 1.7187812328338623,
      "step": 49691
    },
    {
      "epoch": 4.738903045654297e-05,
      "step": 49691,
      "training_step_time": 7.704277038574219
    },
    {
      "epoch": 4.7389984130859375e-05,
      "model_forward_time": 1.713244915008545,
      "step": 49692
    },
    {
      "epoch": 4.7389984130859375e-05,
      "step": 49692,
      "training_step_time": 7.470500469207764
    },
    {
      "epoch": 4.739093780517578e-05,
      "model_forward_time": 1.7190511226654053,
      "step": 49693
    },
    {
      "epoch": 4.739093780517578e-05,
      "step": 49693,
      "training_step_time": 8.509235382080078
    },
    {
      "epoch": 4.739189147949219e-05,
      "model_forward_time": 1.7117409706115723,
      "step": 49694
    },
    {
      "epoch": 4.739189147949219e-05,
      "step": 49694,
      "training_step_time": 7.6225080490112305
    },
    {
      "epoch": 4.7392845153808594e-05,
      "model_forward_time": 1.7132306098937988,
      "step": 49695
    },
    {
      "epoch": 4.7392845153808594e-05,
      "step": 49695,
      "training_step_time": 8.19413685798645
    },
    {
      "epoch": 4.7393798828125e-05,
      "model_forward_time": 1.7157559394836426,
      "step": 49696
    },
    {
      "epoch": 4.7393798828125e-05,
      "step": 49696,
      "training_step_time": 8.126583814620972
    },
    {
      "epoch": 4.739475250244141e-05,
      "model_forward_time": 2.266798257827759,
      "step": 49697
    },
    {
      "epoch": 4.739475250244141e-05,
      "step": 49697,
      "training_step_time": 10.255192518234253
    },
    {
      "epoch": 4.739570617675781e-05,
      "model_forward_time": 1.71610426902771,
      "step": 49698
    },
    {
      "epoch": 4.739570617675781e-05,
      "step": 49698,
      "training_step_time": 6.7181315422058105
    },
    {
      "epoch": 4.739665985107422e-05,
      "model_forward_time": 1.7082099914550781,
      "step": 49699
    },
    {
      "epoch": 4.739665985107422e-05,
      "step": 49699,
      "training_step_time": 6.712961196899414
    },
    {
      "epoch": 4.739761352539063e-05,
      "grad_norm": 0.08791393041610718,
      "learning_rate": 5.462474907238134e-06,
      "loss": 0.0558,
      "step": 49700
    },
    {
      "epoch": 4.739761352539063e-05,
      "model_forward_time": 1.7042264938354492,
      "step": 49700
    },
    {
      "dynamics_loss_avg": 0.05002124067395926,
      "epoch": 4.739761352539063e-05,
      "step": 49700
    },
    {
      "action_loss_avg": 0.0011829598559415899,
      "epoch": 4.739761352539063e-05,
      "step": 49700
    },
    {
      "epoch": 4.739761352539063e-05,
      "step": 49700,
      "training_step_time": 6.741334915161133
    },
    {
      "epoch": 4.739856719970703e-05,
      "model_forward_time": 1.7059922218322754,
      "step": 49701
    },
    {
      "epoch": 4.739856719970703e-05,
      "step": 49701,
      "training_step_time": 6.705000638961792
    },
    {
      "epoch": 4.7399520874023436e-05,
      "model_forward_time": 1.7187299728393555,
      "step": 49702
    },
    {
      "epoch": 4.7399520874023436e-05,
      "step": 49702,
      "training_step_time": 6.737200498580933
    },
    {
      "epoch": 4.7400474548339847e-05,
      "model_forward_time": 1.7154037952423096,
      "step": 49703
    },
    {
      "epoch": 4.7400474548339847e-05,
      "step": 49703,
      "training_step_time": 6.7486724853515625
    },
    {
      "epoch": 4.740142822265625e-05,
      "model_forward_time": 1.7062218189239502,
      "step": 49704
    },
    {
      "epoch": 4.740142822265625e-05,
      "step": 49704,
      "training_step_time": 6.6695616245269775
    },
    {
      "epoch": 4.7402381896972655e-05,
      "model_forward_time": 1.7044572830200195,
      "step": 49705
    },
    {
      "epoch": 4.7402381896972655e-05,
      "step": 49705,
      "training_step_time": 8.160145282745361
    },
    {
      "epoch": 4.7403335571289065e-05,
      "model_forward_time": 1.7124700546264648,
      "step": 49706
    },
    {
      "epoch": 4.7403335571289065e-05,
      "step": 49706,
      "training_step_time": 9.70839786529541
    },
    {
      "epoch": 4.740428924560547e-05,
      "model_forward_time": 1.7088260650634766,
      "step": 49707
    },
    {
      "epoch": 4.740428924560547e-05,
      "step": 49707,
      "training_step_time": 7.592267751693726
    },
    {
      "epoch": 4.7405242919921873e-05,
      "model_forward_time": 1.713735818862915,
      "step": 49708
    },
    {
      "epoch": 4.7405242919921873e-05,
      "step": 49708,
      "training_step_time": 6.846965551376343
    },
    {
      "epoch": 4.7406196594238284e-05,
      "model_forward_time": 1.7086763381958008,
      "step": 49709
    },
    {
      "epoch": 4.7406196594238284e-05,
      "step": 49709,
      "training_step_time": 6.777871370315552
    },
    {
      "epoch": 4.740715026855469e-05,
      "grad_norm": 0.10291038453578949,
      "learning_rate": 5.460828500338672e-06,
      "loss": 0.0603,
      "step": 49710
    },
    {
      "epoch": 4.740715026855469e-05,
      "model_forward_time": 1.6964952945709229,
      "step": 49710
    },
    {
      "dynamics_loss_avg": 0.05595534648746252,
      "epoch": 4.740715026855469e-05,
      "step": 49710
    },
    {
      "action_loss_avg": 0.0017882861779071391,
      "epoch": 4.740715026855469e-05,
      "step": 49710
    },
    {
      "epoch": 4.740715026855469e-05,
      "step": 49710,
      "training_step_time": 6.856217622756958
    },
    {
      "epoch": 4.740810394287109e-05,
      "model_forward_time": 1.710705041885376,
      "step": 49711
    },
    {
      "epoch": 4.740810394287109e-05,
      "step": 49711,
      "training_step_time": 6.733305215835571
    },
    {
      "epoch": 4.74090576171875e-05,
      "model_forward_time": 1.7048516273498535,
      "step": 49712
    },
    {
      "epoch": 4.74090576171875e-05,
      "step": 49712,
      "training_step_time": 6.729224920272827
    },
    {
      "epoch": 4.741001129150391e-05,
      "model_forward_time": 1.6962783336639404,
      "step": 49713
    },
    {
      "epoch": 4.741001129150391e-05,
      "step": 49713,
      "training_step_time": 6.756218433380127
    },
    {
      "epoch": 4.741096496582031e-05,
      "model_forward_time": 1.3409724235534668,
      "step": 49714
    },
    {
      "epoch": 4.741096496582031e-05,
      "step": 49714,
      "training_step_time": 6.750288963317871
    },
    {
      "epoch": 4.741191864013672e-05,
      "model_forward_time": 2.2644639015197754,
      "step": 49715
    },
    {
      "epoch": 4.741191864013672e-05,
      "step": 49715,
      "training_step_time": 7.612445116043091
    },
    {
      "epoch": 4.7412872314453126e-05,
      "model_forward_time": 1.7119503021240234,
      "step": 49716
    },
    {
      "epoch": 4.7412872314453126e-05,
      "step": 49716,
      "training_step_time": 7.704579830169678
    },
    {
      "epoch": 4.741382598876953e-05,
      "model_forward_time": 1.7150685787200928,
      "step": 49717
    },
    {
      "epoch": 4.741382598876953e-05,
      "step": 49717,
      "training_step_time": 6.72999382019043
    },
    {
      "epoch": 4.741477966308594e-05,
      "model_forward_time": 1.3376944065093994,
      "step": 49718
    },
    {
      "epoch": 4.741477966308594e-05,
      "step": 49718,
      "training_step_time": 6.820430517196655
    },
    {
      "epoch": 4.7415733337402345e-05,
      "model_forward_time": 1.7081570625305176,
      "step": 49719
    },
    {
      "epoch": 4.7415733337402345e-05,
      "step": 49719,
      "training_step_time": 6.83567476272583
    },
    {
      "epoch": 4.741668701171875e-05,
      "grad_norm": 0.09502507001161575,
      "learning_rate": 5.459182043043698e-06,
      "loss": 0.0577,
      "step": 49720
    },
    {
      "epoch": 4.741668701171875e-05,
      "model_forward_time": 1.701359510421753,
      "step": 49720
    },
    {
      "dynamics_loss_avg": 0.05547726582735777,
      "epoch": 4.741668701171875e-05,
      "step": 49720
    },
    {
      "action_loss_avg": 0.0015553797828033566,
      "epoch": 4.741668701171875e-05,
      "step": 49720
    },
    {
      "epoch": 4.741668701171875e-05,
      "step": 49720,
      "training_step_time": 6.737461090087891
    },
    {
      "epoch": 4.741764068603515e-05,
      "model_forward_time": 1.6948933601379395,
      "step": 49721
    },
    {
      "epoch": 4.741764068603515e-05,
      "step": 49721,
      "training_step_time": 6.831778287887573
    },
    {
      "epoch": 4.7418594360351564e-05,
      "model_forward_time": 1.708467960357666,
      "step": 49722
    },
    {
      "epoch": 4.7418594360351564e-05,
      "step": 49722,
      "training_step_time": 6.871929168701172
    },
    {
      "epoch": 4.741954803466797e-05,
      "model_forward_time": 1.7028400897979736,
      "step": 49723
    },
    {
      "epoch": 4.741954803466797e-05,
      "step": 49723,
      "training_step_time": 6.886105060577393
    },
    {
      "epoch": 4.742050170898437e-05,
      "model_forward_time": 2.2657032012939453,
      "step": 49724
    },
    {
      "epoch": 4.742050170898437e-05,
      "step": 49724,
      "training_step_time": 7.598007917404175
    },
    {
      "epoch": 4.742145538330078e-05,
      "model_forward_time": 1.7150120735168457,
      "step": 49725
    },
    {
      "epoch": 4.742145538330078e-05,
      "step": 49725,
      "training_step_time": 9.048712730407715
    },
    {
      "epoch": 4.742240905761719e-05,
      "model_forward_time": 1.7026877403259277,
      "step": 49726
    },
    {
      "epoch": 4.742240905761719e-05,
      "step": 49726,
      "training_step_time": 7.589545488357544
    },
    {
      "epoch": 4.742336273193359e-05,
      "model_forward_time": 1.7072389125823975,
      "step": 49727
    },
    {
      "epoch": 4.742336273193359e-05,
      "step": 49727,
      "training_step_time": 6.777357816696167
    },
    {
      "epoch": 4.742431640625e-05,
      "model_forward_time": 1.7153759002685547,
      "step": 49728
    },
    {
      "epoch": 4.742431640625e-05,
      "step": 49728,
      "training_step_time": 6.843258380889893
    },
    {
      "epoch": 4.7425270080566405e-05,
      "model_forward_time": 1.7016456127166748,
      "step": 49729
    },
    {
      "epoch": 4.7425270080566405e-05,
      "step": 49729,
      "training_step_time": 6.862902879714966
    },
    {
      "epoch": 4.742622375488281e-05,
      "grad_norm": 0.11115521192550659,
      "learning_rate": 5.4575355355332645e-06,
      "loss": 0.0602,
      "step": 49730
    },
    {
      "epoch": 4.742622375488281e-05,
      "model_forward_time": 1.6985843181610107,
      "step": 49730
    },
    {
      "dynamics_loss_avg": 0.04982773340307176,
      "epoch": 4.742622375488281e-05,
      "step": 49730
    },
    {
      "action_loss_avg": 0.0010792687498906162,
      "epoch": 4.742622375488281e-05,
      "step": 49730
    },
    {
      "epoch": 4.742622375488281e-05,
      "step": 49730,
      "training_step_time": 6.800870656967163
    },
    {
      "epoch": 4.742717742919922e-05,
      "model_forward_time": 1.7178895473480225,
      "step": 49731
    },
    {
      "epoch": 4.742717742919922e-05,
      "step": 49731,
      "training_step_time": 6.744919300079346
    },
    {
      "epoch": 4.7428131103515624e-05,
      "model_forward_time": 1.7150905132293701,
      "step": 49732
    },
    {
      "epoch": 4.7428131103515624e-05,
      "step": 49732,
      "training_step_time": 6.788212060928345
    },
    {
      "epoch": 4.742908477783203e-05,
      "model_forward_time": 2.285601854324341,
      "step": 49733
    },
    {
      "epoch": 4.742908477783203e-05,
      "step": 49733,
      "training_step_time": 7.536570072174072
    },
    {
      "epoch": 4.743003845214844e-05,
      "model_forward_time": 1.7033271789550781,
      "step": 49734
    },
    {
      "epoch": 4.743003845214844e-05,
      "step": 49734,
      "training_step_time": 8.744955062866211
    },
    {
      "epoch": 4.743099212646484e-05,
      "model_forward_time": 1.717442274093628,
      "step": 49735
    },
    {
      "epoch": 4.743099212646484e-05,
      "step": 49735,
      "training_step_time": 7.724676847457886
    },
    {
      "epoch": 4.743194580078125e-05,
      "model_forward_time": 1.7223925590515137,
      "step": 49736
    },
    {
      "epoch": 4.743194580078125e-05,
      "step": 49736,
      "training_step_time": 6.880983114242554
    },
    {
      "epoch": 4.743289947509766e-05,
      "model_forward_time": 1.7163965702056885,
      "step": 49737
    },
    {
      "epoch": 4.743289947509766e-05,
      "step": 49737,
      "training_step_time": 6.8623881340026855
    },
    {
      "epoch": 4.743385314941406e-05,
      "model_forward_time": 1.7127416133880615,
      "step": 49738
    },
    {
      "epoch": 4.743385314941406e-05,
      "step": 49738,
      "training_step_time": 6.807803630828857
    },
    {
      "epoch": 4.7434806823730466e-05,
      "model_forward_time": 1.720350742340088,
      "step": 49739
    },
    {
      "epoch": 4.7434806823730466e-05,
      "step": 49739,
      "training_step_time": 6.849758625030518
    },
    {
      "epoch": 4.743576049804688e-05,
      "grad_norm": 0.08209020644426346,
      "learning_rate": 5.455888977987434e-06,
      "loss": 0.0589,
      "step": 49740
    },
    {
      "epoch": 4.743576049804688e-05,
      "model_forward_time": 1.7242012023925781,
      "step": 49740
    },
    {
      "dynamics_loss_avg": 0.06763767953962088,
      "epoch": 4.743576049804688e-05,
      "step": 49740
    },
    {
      "action_loss_avg": 0.0016563145720283501,
      "epoch": 4.743576049804688e-05,
      "step": 49740
    },
    {
      "epoch": 4.743576049804688e-05,
      "step": 49740,
      "training_step_time": 6.812112331390381
    },
    {
      "epoch": 4.743671417236328e-05,
      "model_forward_time": 1.712622880935669,
      "step": 49741
    },
    {
      "epoch": 4.743671417236328e-05,
      "step": 49741,
      "training_step_time": 6.80375337600708
    },
    {
      "epoch": 4.7437667846679685e-05,
      "model_forward_time": 2.341916561126709,
      "step": 49742
    },
    {
      "epoch": 4.7437667846679685e-05,
      "step": 49742,
      "training_step_time": 7.603867292404175
    },
    {
      "epoch": 4.7438621520996096e-05,
      "model_forward_time": 1.712038278579712,
      "step": 49743
    },
    {
      "epoch": 4.7438621520996096e-05,
      "step": 49743,
      "training_step_time": 10.522371292114258
    },
    {
      "epoch": 4.74395751953125e-05,
      "model_forward_time": 1.7212722301483154,
      "step": 49744
    },
    {
      "epoch": 4.74395751953125e-05,
      "step": 49744,
      "training_step_time": 7.717064142227173
    },
    {
      "epoch": 4.7440528869628904e-05,
      "model_forward_time": 1.7160680294036865,
      "step": 49745
    },
    {
      "epoch": 4.7440528869628904e-05,
      "step": 49745,
      "training_step_time": 6.765362501144409
    },
    {
      "epoch": 4.7441482543945315e-05,
      "model_forward_time": 1.720885992050171,
      "step": 49746
    },
    {
      "epoch": 4.7441482543945315e-05,
      "step": 49746,
      "training_step_time": 6.785082578659058
    },
    {
      "epoch": 4.744243621826172e-05,
      "model_forward_time": 1.7184550762176514,
      "step": 49747
    },
    {
      "epoch": 4.744243621826172e-05,
      "step": 49747,
      "training_step_time": 6.8130035400390625
    },
    {
      "epoch": 4.744338989257812e-05,
      "model_forward_time": 1.7229118347167969,
      "step": 49748
    },
    {
      "epoch": 4.744338989257812e-05,
      "step": 49748,
      "training_step_time": 6.8106465339660645
    },
    {
      "epoch": 4.7444343566894533e-05,
      "model_forward_time": 1.723597526550293,
      "step": 49749
    },
    {
      "epoch": 4.7444343566894533e-05,
      "step": 49749,
      "training_step_time": 6.756539344787598
    },
    {
      "epoch": 4.744529724121094e-05,
      "grad_norm": 0.09997232258319855,
      "learning_rate": 5.454242370586269e-06,
      "loss": 0.0562,
      "step": 49750
    },
    {
      "epoch": 4.744529724121094e-05,
      "model_forward_time": 1.711047649383545,
      "step": 49750
    },
    {
      "dynamics_loss_avg": 0.05469266623258591,
      "epoch": 4.744529724121094e-05,
      "step": 49750
    },
    {
      "action_loss_avg": 0.0012863097421359272,
      "epoch": 4.744529724121094e-05,
      "step": 49750
    },
    {
      "epoch": 4.744529724121094e-05,
      "step": 49750,
      "training_step_time": 6.805172443389893
    },
    {
      "epoch": 4.744625091552734e-05,
      "model_forward_time": 2.3076789379119873,
      "step": 49751
    },
    {
      "epoch": 4.744625091552734e-05,
      "step": 49751,
      "training_step_time": 7.605067729949951
    },
    {
      "epoch": 4.744720458984375e-05,
      "model_forward_time": 1.7174813747406006,
      "step": 49752
    },
    {
      "epoch": 4.744720458984375e-05,
      "step": 49752,
      "training_step_time": 10.565980911254883
    },
    {
      "epoch": 4.7448158264160156e-05,
      "model_forward_time": 1.7156522274017334,
      "step": 49753
    },
    {
      "epoch": 4.7448158264160156e-05,
      "step": 49753,
      "training_step_time": 7.6759278774261475
    },
    {
      "epoch": 4.744911193847656e-05,
      "model_forward_time": 1.7223389148712158,
      "step": 49754
    },
    {
      "epoch": 4.744911193847656e-05,
      "step": 49754,
      "training_step_time": 7.594117879867554
    },
    {
      "epoch": 4.745006561279297e-05,
      "model_forward_time": 1.7196249961853027,
      "step": 49755
    },
    {
      "epoch": 4.745006561279297e-05,
      "step": 49755,
      "training_step_time": 6.87026047706604
    },
    {
      "epoch": 4.7451019287109375e-05,
      "model_forward_time": 1.712073564529419,
      "step": 49756
    },
    {
      "epoch": 4.7451019287109375e-05,
      "step": 49756,
      "training_step_time": 6.8671324253082275
    },
    {
      "epoch": 4.745197296142578e-05,
      "model_forward_time": 1.7112348079681396,
      "step": 49757
    },
    {
      "epoch": 4.745197296142578e-05,
      "step": 49757,
      "training_step_time": 6.90489935874939
    },
    {
      "epoch": 4.745292663574219e-05,
      "model_forward_time": 1.7286360263824463,
      "step": 49758
    },
    {
      "epoch": 4.745292663574219e-05,
      "step": 49758,
      "training_step_time": 6.720990180969238
    },
    {
      "epoch": 4.7453880310058594e-05,
      "model_forward_time": 1.7198066711425781,
      "step": 49759
    },
    {
      "epoch": 4.7453880310058594e-05,
      "step": 49759,
      "training_step_time": 6.8256447315216064
    },
    {
      "epoch": 4.7454833984375e-05,
      "grad_norm": 0.08572888374328613,
      "learning_rate": 5.452595713509843e-06,
      "loss": 0.0582,
      "step": 49760
    },
    {
      "epoch": 4.7454833984375e-05,
      "model_forward_time": 1.7160546779632568,
      "step": 49760
    },
    {
      "dynamics_loss_avg": 0.051583843864500524,
      "epoch": 4.7454833984375e-05,
      "step": 49760
    },
    {
      "action_loss_avg": 0.0015678597032092513,
      "epoch": 4.7454833984375e-05,
      "step": 49760
    },
    {
      "epoch": 4.7454833984375e-05,
      "step": 49760,
      "training_step_time": 7.8425703048706055
    },
    {
      "epoch": 4.745578765869141e-05,
      "model_forward_time": 1.7963006496429443,
      "step": 49761
    },
    {
      "epoch": 4.745578765869141e-05,
      "step": 49761,
      "training_step_time": 9.357128620147705
    },
    {
      "epoch": 4.745674133300781e-05,
      "model_forward_time": 1.7150990962982178,
      "step": 49762
    },
    {
      "epoch": 4.745674133300781e-05,
      "step": 49762,
      "training_step_time": 7.676537036895752
    },
    {
      "epoch": 4.745769500732422e-05,
      "model_forward_time": 1.699885368347168,
      "step": 49763
    },
    {
      "epoch": 4.745769500732422e-05,
      "step": 49763,
      "training_step_time": 7.628742218017578
    },
    {
      "epoch": 4.745864868164063e-05,
      "model_forward_time": 1.715005874633789,
      "step": 49764
    },
    {
      "epoch": 4.745864868164063e-05,
      "step": 49764,
      "training_step_time": 6.929271221160889
    },
    {
      "epoch": 4.745960235595703e-05,
      "model_forward_time": 1.7221870422363281,
      "step": 49765
    },
    {
      "epoch": 4.745960235595703e-05,
      "step": 49765,
      "training_step_time": 6.733410358428955
    },
    {
      "epoch": 4.7460556030273436e-05,
      "model_forward_time": 1.7205431461334229,
      "step": 49766
    },
    {
      "epoch": 4.7460556030273436e-05,
      "step": 49766,
      "training_step_time": 6.7634289264678955
    },
    {
      "epoch": 4.746150970458985e-05,
      "model_forward_time": 1.714353084564209,
      "step": 49767
    },
    {
      "epoch": 4.746150970458985e-05,
      "step": 49767,
      "training_step_time": 6.850957870483398
    },
    {
      "epoch": 4.746246337890625e-05,
      "model_forward_time": 1.7136125564575195,
      "step": 49768
    },
    {
      "epoch": 4.746246337890625e-05,
      "step": 49768,
      "training_step_time": 6.7884814739227295
    },
    {
      "epoch": 4.7463417053222655e-05,
      "model_forward_time": 1.7203845977783203,
      "step": 49769
    },
    {
      "epoch": 4.7463417053222655e-05,
      "step": 49769,
      "training_step_time": 7.313358545303345
    },
    {
      "epoch": 4.7464370727539065e-05,
      "grad_norm": 0.11761409789323807,
      "learning_rate": 5.450949006938228e-06,
      "loss": 0.0587,
      "step": 49770
    },
    {
      "epoch": 4.7464370727539065e-05,
      "model_forward_time": 2.3291330337524414,
      "step": 49770
    },
    {
      "dynamics_loss_avg": 0.049981831572949886,
      "epoch": 4.7464370727539065e-05,
      "step": 49770
    },
    {
      "action_loss_avg": 0.0019038446014747023,
      "epoch": 4.7464370727539065e-05,
      "step": 49770
    },
    {
      "epoch": 4.7464370727539065e-05,
      "step": 49770,
      "training_step_time": 9.01021957397461
    },
    {
      "epoch": 4.746532440185547e-05,
      "model_forward_time": 1.7190964221954346,
      "step": 49771
    },
    {
      "epoch": 4.746532440185547e-05,
      "step": 49771,
      "training_step_time": 8.246652126312256
    },
    {
      "epoch": 4.7466278076171874e-05,
      "model_forward_time": 1.71535325050354,
      "step": 49772
    },
    {
      "epoch": 4.7466278076171874e-05,
      "step": 49772,
      "training_step_time": 7.518802165985107
    },
    {
      "epoch": 4.7467231750488284e-05,
      "model_forward_time": 1.7189371585845947,
      "step": 49773
    },
    {
      "epoch": 4.7467231750488284e-05,
      "step": 49773,
      "training_step_time": 7.148597717285156
    },
    {
      "epoch": 4.746818542480469e-05,
      "model_forward_time": 1.7228355407714844,
      "step": 49774
    },
    {
      "epoch": 4.746818542480469e-05,
      "step": 49774,
      "training_step_time": 6.75327730178833
    },
    {
      "epoch": 4.746913909912109e-05,
      "model_forward_time": 1.7141449451446533,
      "step": 49775
    },
    {
      "epoch": 4.746913909912109e-05,
      "step": 49775,
      "training_step_time": 6.778757333755493
    },
    {
      "epoch": 4.74700927734375e-05,
      "model_forward_time": 1.6847097873687744,
      "step": 49776
    },
    {
      "epoch": 4.74700927734375e-05,
      "step": 49776,
      "training_step_time": 6.791884422302246
    },
    {
      "epoch": 4.747104644775391e-05,
      "model_forward_time": 1.7150864601135254,
      "step": 49777
    },
    {
      "epoch": 4.747104644775391e-05,
      "step": 49777,
      "training_step_time": 6.763025522232056
    },
    {
      "epoch": 4.747200012207031e-05,
      "model_forward_time": 1.3342781066894531,
      "step": 49778
    },
    {
      "epoch": 4.747200012207031e-05,
      "step": 49778,
      "training_step_time": 7.326425075531006
    },
    {
      "epoch": 4.747295379638672e-05,
      "model_forward_time": 2.301887035369873,
      "step": 49779
    },
    {
      "epoch": 4.747295379638672e-05,
      "step": 49779,
      "training_step_time": 8.544822454452515
    },
    {
      "epoch": 4.7473907470703126e-05,
      "grad_norm": 0.11070691049098969,
      "learning_rate": 5.449302251051509e-06,
      "loss": 0.0621,
      "step": 49780
    },
    {
      "epoch": 4.7473907470703126e-05,
      "model_forward_time": 1.7040612697601318,
      "step": 49780
    },
    {
      "dynamics_loss_avg": 0.0593074107542634,
      "epoch": 4.7473907470703126e-05,
      "step": 49780
    },
    {
      "action_loss_avg": 0.0018489737238269299,
      "epoch": 4.7473907470703126e-05,
      "step": 49780
    },
    {
      "epoch": 4.7473907470703126e-05,
      "step": 49780,
      "training_step_time": 8.491224527359009
    },
    {
      "epoch": 4.747486114501953e-05,
      "model_forward_time": 1.7081997394561768,
      "step": 49781
    },
    {
      "epoch": 4.747486114501953e-05,
      "step": 49781,
      "training_step_time": 7.657942533493042
    },
    {
      "epoch": 4.7475814819335934e-05,
      "model_forward_time": 1.710402011871338,
      "step": 49782
    },
    {
      "epoch": 4.7475814819335934e-05,
      "step": 49782,
      "training_step_time": 7.506715774536133
    },
    {
      "epoch": 4.7476768493652345e-05,
      "model_forward_time": 1.7042887210845947,
      "step": 49783
    },
    {
      "epoch": 4.7476768493652345e-05,
      "step": 49783,
      "training_step_time": 6.812768936157227
    },
    {
      "epoch": 4.747772216796875e-05,
      "model_forward_time": 1.7219822406768799,
      "step": 49784
    },
    {
      "epoch": 4.747772216796875e-05,
      "step": 49784,
      "training_step_time": 6.770234107971191
    },
    {
      "epoch": 4.747867584228515e-05,
      "model_forward_time": 1.7071731090545654,
      "step": 49785
    },
    {
      "epoch": 4.747867584228515e-05,
      "step": 49785,
      "training_step_time": 6.792989253997803
    },
    {
      "epoch": 4.7479629516601564e-05,
      "model_forward_time": 1.703840970993042,
      "step": 49786
    },
    {
      "epoch": 4.7479629516601564e-05,
      "step": 49786,
      "training_step_time": 6.824881076812744
    },
    {
      "epoch": 4.748058319091797e-05,
      "model_forward_time": 1.7071046829223633,
      "step": 49787
    },
    {
      "epoch": 4.748058319091797e-05,
      "step": 49787,
      "training_step_time": 7.4092395305633545
    },
    {
      "epoch": 4.748153686523437e-05,
      "model_forward_time": 2.3125734329223633,
      "step": 49788
    },
    {
      "epoch": 4.748153686523437e-05,
      "step": 49788,
      "training_step_time": 8.658251285552979
    },
    {
      "epoch": 4.748249053955078e-05,
      "model_forward_time": 1.7083361148834229,
      "step": 49789
    },
    {
      "epoch": 4.748249053955078e-05,
      "step": 49789,
      "training_step_time": 10.455881118774414
    },
    {
      "epoch": 4.748344421386719e-05,
      "grad_norm": 0.0825967937707901,
      "learning_rate": 5.4476554460297706e-06,
      "loss": 0.0608,
      "step": 49790
    },
    {
      "epoch": 4.748344421386719e-05,
      "model_forward_time": 1.7114369869232178,
      "step": 49790
    },
    {
      "dynamics_loss_avg": 0.05657298732548952,
      "epoch": 4.748344421386719e-05,
      "step": 49790
    },
    {
      "action_loss_avg": 0.001601934217615053,
      "epoch": 4.748344421386719e-05,
      "step": 49790
    },
    {
      "epoch": 4.748344421386719e-05,
      "step": 49790,
      "training_step_time": 7.671943664550781
    },
    {
      "epoch": 4.748439788818359e-05,
      "model_forward_time": 1.709089994430542,
      "step": 49791
    },
    {
      "epoch": 4.748439788818359e-05,
      "step": 49791,
      "training_step_time": 7.561213493347168
    },
    {
      "epoch": 4.74853515625e-05,
      "model_forward_time": 1.7110474109649658,
      "step": 49792
    },
    {
      "epoch": 4.74853515625e-05,
      "step": 49792,
      "training_step_time": 6.9322192668914795
    },
    {
      "epoch": 4.7486305236816406e-05,
      "model_forward_time": 1.7200913429260254,
      "step": 49793
    },
    {
      "epoch": 4.7486305236816406e-05,
      "step": 49793,
      "training_step_time": 6.753087759017944
    },
    {
      "epoch": 4.748725891113281e-05,
      "model_forward_time": 1.7107536792755127,
      "step": 49794
    },
    {
      "epoch": 4.748725891113281e-05,
      "step": 49794,
      "training_step_time": 6.705769777297974
    },
    {
      "epoch": 4.748821258544922e-05,
      "model_forward_time": 1.7104823589324951,
      "step": 49795
    },
    {
      "epoch": 4.748821258544922e-05,
      "step": 49795,
      "training_step_time": 6.7593162059783936
    },
    {
      "epoch": 4.7489166259765624e-05,
      "model_forward_time": 1.707029104232788,
      "step": 49796
    },
    {
      "epoch": 4.7489166259765624e-05,
      "step": 49796,
      "training_step_time": 7.343419313430786
    },
    {
      "epoch": 4.749011993408203e-05,
      "model_forward_time": 2.2891194820404053,
      "step": 49797
    },
    {
      "epoch": 4.749011993408203e-05,
      "step": 49797,
      "training_step_time": 8.964238405227661
    },
    {
      "epoch": 4.749107360839844e-05,
      "model_forward_time": 1.7113089561462402,
      "step": 49798
    },
    {
      "epoch": 4.749107360839844e-05,
      "step": 49798,
      "training_step_time": 7.7994866371154785
    },
    {
      "epoch": 4.749202728271484e-05,
      "model_forward_time": 1.7076210975646973,
      "step": 49799
    },
    {
      "epoch": 4.749202728271484e-05,
      "step": 49799,
      "training_step_time": 7.761155605316162
    },
    {
      "epoch": 4.749298095703125e-05,
      "grad_norm": 0.11471840739250183,
      "learning_rate": 5.446008592053107e-06,
      "loss": 0.0602,
      "step": 49800
    },
    {
      "epoch": 4.749298095703125e-05,
      "model_forward_time": 1.711956262588501,
      "step": 49800
    },
    {
      "dynamics_loss_avg": 0.06026044487953186,
      "epoch": 4.749298095703125e-05,
      "step": 49800
    },
    {
      "action_loss_avg": 0.0020277686999179423,
      "epoch": 4.749298095703125e-05,
      "step": 49800
    },
    {
      "epoch": 4.749298095703125e-05,
      "step": 49800,
      "training_step_time": 7.584881067276001
    },
    {
      "epoch": 4.749393463134766e-05,
      "model_forward_time": 1.7054815292358398,
      "step": 49801
    },
    {
      "epoch": 4.749393463134766e-05,
      "step": 49801,
      "training_step_time": 7.252694129943848
    },
    {
      "epoch": 4.749488830566406e-05,
      "model_forward_time": 1.7115740776062012,
      "step": 49802
    },
    {
      "epoch": 4.749488830566406e-05,
      "step": 49802,
      "training_step_time": 6.833151340484619
    },
    {
      "epoch": 4.7495841979980466e-05,
      "model_forward_time": 1.7057750225067139,
      "step": 49803
    },
    {
      "epoch": 4.7495841979980466e-05,
      "step": 49803,
      "training_step_time": 6.820508003234863
    },
    {
      "epoch": 4.749679565429688e-05,
      "model_forward_time": 1.7109084129333496,
      "step": 49804
    },
    {
      "epoch": 4.749679565429688e-05,
      "step": 49804,
      "training_step_time": 6.837085008621216
    },
    {
      "epoch": 4.749774932861328e-05,
      "model_forward_time": 1.7174675464630127,
      "step": 49805
    },
    {
      "epoch": 4.749774932861328e-05,
      "step": 49805,
      "training_step_time": 6.86282205581665
    },
    {
      "epoch": 4.7498703002929685e-05,
      "model_forward_time": 1.7267870903015137,
      "step": 49806
    },
    {
      "epoch": 4.7498703002929685e-05,
      "step": 49806,
      "training_step_time": 9.571840047836304
    },
    {
      "epoch": 4.7499656677246096e-05,
      "model_forward_time": 1.7052037715911865,
      "step": 49807
    },
    {
      "epoch": 4.7499656677246096e-05,
      "step": 49807,
      "training_step_time": 8.563255071640015
    },
    {
      "epoch": 4.75006103515625e-05,
      "model_forward_time": 1.7126045227050781,
      "step": 49808
    },
    {
      "epoch": 4.75006103515625e-05,
      "step": 49808,
      "training_step_time": 9.130763530731201
    },
    {
      "epoch": 4.7501564025878904e-05,
      "model_forward_time": 1.7137815952301025,
      "step": 49809
    },
    {
      "epoch": 4.7501564025878904e-05,
      "step": 49809,
      "training_step_time": 7.540239095687866
    },
    {
      "epoch": 4.7502517700195315e-05,
      "grad_norm": 0.10333838313817978,
      "learning_rate": 5.444361689301614e-06,
      "loss": 0.0603,
      "step": 49810
    },
    {
      "epoch": 4.7502517700195315e-05,
      "model_forward_time": 1.7006134986877441,
      "step": 49810
    },
    {
      "dynamics_loss_avg": 0.048549641110003,
      "epoch": 4.7502517700195315e-05,
      "step": 49810
    },
    {
      "action_loss_avg": 0.0012034935585688799,
      "epoch": 4.7502517700195315e-05,
      "step": 49810
    },
    {
      "epoch": 4.7502517700195315e-05,
      "step": 49810,
      "training_step_time": 6.966911315917969
    },
    {
      "epoch": 4.750347137451172e-05,
      "model_forward_time": 1.71262526512146,
      "step": 49811
    },
    {
      "epoch": 4.750347137451172e-05,
      "step": 49811,
      "training_step_time": 6.7915754318237305
    },
    {
      "epoch": 4.750442504882812e-05,
      "model_forward_time": 1.7111215591430664,
      "step": 49812
    },
    {
      "epoch": 4.750442504882812e-05,
      "step": 49812,
      "training_step_time": 6.791326284408569
    },
    {
      "epoch": 4.7505378723144534e-05,
      "model_forward_time": 1.7064173221588135,
      "step": 49813
    },
    {
      "epoch": 4.7505378723144534e-05,
      "step": 49813,
      "training_step_time": 6.786724090576172
    },
    {
      "epoch": 4.750633239746094e-05,
      "model_forward_time": 1.7110517024993896,
      "step": 49814
    },
    {
      "epoch": 4.750633239746094e-05,
      "step": 49814,
      "training_step_time": 6.74821138381958
    },
    {
      "epoch": 4.750728607177734e-05,
      "model_forward_time": 1.7147777080535889,
      "step": 49815
    },
    {
      "epoch": 4.750728607177734e-05,
      "step": 49815,
      "training_step_time": 8.162125825881958
    },
    {
      "epoch": 4.750823974609375e-05,
      "model_forward_time": 2.2482497692108154,
      "step": 49816
    },
    {
      "epoch": 4.750823974609375e-05,
      "step": 49816,
      "training_step_time": 8.713822841644287
    },
    {
      "epoch": 4.7509193420410156e-05,
      "model_forward_time": 1.7143220901489258,
      "step": 49817
    },
    {
      "epoch": 4.7509193420410156e-05,
      "step": 49817,
      "training_step_time": 9.67954707145691
    },
    {
      "epoch": 4.751014709472656e-05,
      "model_forward_time": 1.7052228450775146,
      "step": 49818
    },
    {
      "epoch": 4.751014709472656e-05,
      "step": 49818,
      "training_step_time": 7.751938104629517
    },
    {
      "epoch": 4.751110076904297e-05,
      "model_forward_time": 1.6951699256896973,
      "step": 49819
    },
    {
      "epoch": 4.751110076904297e-05,
      "step": 49819,
      "training_step_time": 7.634048223495483
    },
    {
      "epoch": 4.7512054443359375e-05,
      "grad_norm": 0.13933244347572327,
      "learning_rate": 5.442714737955394e-06,
      "loss": 0.0595,
      "step": 49820
    },
    {
      "epoch": 4.7512054443359375e-05,
      "model_forward_time": 1.7063376903533936,
      "step": 49820
    },
    {
      "dynamics_loss_avg": 0.06348061244934797,
      "epoch": 4.7512054443359375e-05,
      "step": 49820
    },
    {
      "action_loss_avg": 0.0017370709945680573,
      "epoch": 4.7512054443359375e-05,
      "step": 49820
    },
    {
      "epoch": 4.7512054443359375e-05,
      "step": 49820,
      "training_step_time": 6.862303018569946
    },
    {
      "epoch": 4.751300811767578e-05,
      "model_forward_time": 1.7101171016693115,
      "step": 49821
    },
    {
      "epoch": 4.751300811767578e-05,
      "step": 49821,
      "training_step_time": 6.9115800857543945
    },
    {
      "epoch": 4.751396179199219e-05,
      "model_forward_time": 1.7117335796356201,
      "step": 49822
    },
    {
      "epoch": 4.751396179199219e-05,
      "step": 49822,
      "training_step_time": 6.866129398345947
    },
    {
      "epoch": 4.7514915466308594e-05,
      "model_forward_time": 1.7098455429077148,
      "step": 49823
    },
    {
      "epoch": 4.7514915466308594e-05,
      "step": 49823,
      "training_step_time": 6.862381935119629
    },
    {
      "epoch": 4.7515869140625e-05,
      "model_forward_time": 1.7123308181762695,
      "step": 49824
    },
    {
      "epoch": 4.7515869140625e-05,
      "step": 49824,
      "training_step_time": 8.718935012817383
    },
    {
      "epoch": 4.751682281494141e-05,
      "model_forward_time": 1.9777469635009766,
      "step": 49825
    },
    {
      "epoch": 4.751682281494141e-05,
      "step": 49825,
      "training_step_time": 9.400108814239502
    },
    {
      "epoch": 4.751777648925781e-05,
      "model_forward_time": 1.7105305194854736,
      "step": 49826
    },
    {
      "epoch": 4.751777648925781e-05,
      "step": 49826,
      "training_step_time": 9.453290700912476
    },
    {
      "epoch": 4.751873016357422e-05,
      "model_forward_time": 1.7109785079956055,
      "step": 49827
    },
    {
      "epoch": 4.751873016357422e-05,
      "step": 49827,
      "training_step_time": 7.786195993423462
    },
    {
      "epoch": 4.751968383789063e-05,
      "model_forward_time": 1.7138535976409912,
      "step": 49828
    },
    {
      "epoch": 4.751968383789063e-05,
      "step": 49828,
      "training_step_time": 7.653596639633179
    },
    {
      "epoch": 4.752063751220703e-05,
      "model_forward_time": 1.7099895477294922,
      "step": 49829
    },
    {
      "epoch": 4.752063751220703e-05,
      "step": 49829,
      "training_step_time": 7.33116888999939
    },
    {
      "epoch": 4.7521591186523436e-05,
      "grad_norm": 0.10385783016681671,
      "learning_rate": 5.441067738194557e-06,
      "loss": 0.0579,
      "step": 49830
    },
    {
      "epoch": 4.7521591186523436e-05,
      "model_forward_time": 1.7039799690246582,
      "step": 49830
    },
    {
      "dynamics_loss_avg": 0.051209048833698034,
      "epoch": 4.7521591186523436e-05,
      "step": 49830
    },
    {
      "action_loss_avg": 0.0010797267052112148,
      "epoch": 4.7521591186523436e-05,
      "step": 49830
    },
    {
      "epoch": 4.7521591186523436e-05,
      "step": 49830,
      "training_step_time": 6.7652060985565186
    },
    {
      "epoch": 4.752254486083985e-05,
      "model_forward_time": 1.7102184295654297,
      "step": 49831
    },
    {
      "epoch": 4.752254486083985e-05,
      "step": 49831,
      "training_step_time": 6.737219572067261
    },
    {
      "epoch": 4.752349853515625e-05,
      "model_forward_time": 1.7102866172790527,
      "step": 49832
    },
    {
      "epoch": 4.752349853515625e-05,
      "step": 49832,
      "training_step_time": 6.733559846878052
    },
    {
      "epoch": 4.7524452209472655e-05,
      "model_forward_time": 1.71030855178833,
      "step": 49833
    },
    {
      "epoch": 4.7524452209472655e-05,
      "step": 49833,
      "training_step_time": 10.034160375595093
    },
    {
      "epoch": 4.7525405883789066e-05,
      "model_forward_time": 2.307150363922119,
      "step": 49834
    },
    {
      "epoch": 4.7525405883789066e-05,
      "step": 49834,
      "training_step_time": 8.583277225494385
    },
    {
      "epoch": 4.752635955810547e-05,
      "model_forward_time": 1.7087962627410889,
      "step": 49835
    },
    {
      "epoch": 4.752635955810547e-05,
      "step": 49835,
      "training_step_time": 10.320664405822754
    },
    {
      "epoch": 4.7527313232421874e-05,
      "model_forward_time": 1.7205536365509033,
      "step": 49836
    },
    {
      "epoch": 4.7527313232421874e-05,
      "step": 49836,
      "training_step_time": 9.308140993118286
    },
    {
      "epoch": 4.7528266906738284e-05,
      "model_forward_time": 1.7047638893127441,
      "step": 49837
    },
    {
      "epoch": 4.7528266906738284e-05,
      "step": 49837,
      "training_step_time": 7.603976488113403
    },
    {
      "epoch": 4.752922058105469e-05,
      "model_forward_time": 1.7142105102539062,
      "step": 49838
    },
    {
      "epoch": 4.752922058105469e-05,
      "step": 49838,
      "training_step_time": 7.2403059005737305
    },
    {
      "epoch": 4.753017425537109e-05,
      "model_forward_time": 1.683220386505127,
      "step": 49839
    },
    {
      "epoch": 4.753017425537109e-05,
      "step": 49839,
      "training_step_time": 6.53922700881958
    },
    {
      "epoch": 4.75311279296875e-05,
      "grad_norm": 0.1425296813249588,
      "learning_rate": 5.439420690199214e-06,
      "loss": 0.058,
      "step": 49840
    },
    {
      "epoch": 4.75311279296875e-05,
      "model_forward_time": 1.6975679397583008,
      "step": 49840
    },
    {
      "dynamics_loss_avg": 0.0631744446232915,
      "epoch": 4.75311279296875e-05,
      "step": 49840
    },
    {
      "action_loss_avg": 0.002226937303203158,
      "epoch": 4.75311279296875e-05,
      "step": 49840
    },
    {
      "epoch": 4.75311279296875e-05,
      "step": 49840,
      "training_step_time": 6.614773750305176
    },
    {
      "epoch": 4.753208160400391e-05,
      "model_forward_time": 1.7049753665924072,
      "step": 49841
    },
    {
      "epoch": 4.753208160400391e-05,
      "step": 49841,
      "training_step_time": 6.657277584075928
    },
    {
      "epoch": 4.753303527832031e-05,
      "model_forward_time": 1.7129466533660889,
      "step": 49842
    },
    {
      "epoch": 4.753303527832031e-05,
      "step": 49842,
      "training_step_time": 7.58819842338562
    },
    {
      "epoch": 4.753398895263672e-05,
      "model_forward_time": 2.3369011878967285,
      "step": 49843
    },
    {
      "epoch": 4.753398895263672e-05,
      "step": 49843,
      "training_step_time": 7.6102135181427
    },
    {
      "epoch": 4.7534942626953126e-05,
      "model_forward_time": 1.7103588581085205,
      "step": 49844
    },
    {
      "epoch": 4.7534942626953126e-05,
      "step": 49844,
      "training_step_time": 9.068902254104614
    },
    {
      "epoch": 4.753589630126953e-05,
      "model_forward_time": 1.7052977085113525,
      "step": 49845
    },
    {
      "epoch": 4.753589630126953e-05,
      "step": 49845,
      "training_step_time": 8.999739408493042
    },
    {
      "epoch": 4.7536849975585934e-05,
      "model_forward_time": 1.7108795642852783,
      "step": 49846
    },
    {
      "epoch": 4.7536849975585934e-05,
      "step": 49846,
      "training_step_time": 7.624161243438721
    },
    {
      "epoch": 4.7537803649902345e-05,
      "model_forward_time": 1.7147047519683838,
      "step": 49847
    },
    {
      "epoch": 4.7537803649902345e-05,
      "step": 49847,
      "training_step_time": 7.205686092376709
    },
    {
      "epoch": 4.753875732421875e-05,
      "model_forward_time": 1.7131521701812744,
      "step": 49848
    },
    {
      "epoch": 4.753875732421875e-05,
      "step": 49848,
      "training_step_time": 6.6449103355407715
    },
    {
      "epoch": 4.753971099853515e-05,
      "model_forward_time": 1.7070975303649902,
      "step": 49849
    },
    {
      "epoch": 4.753971099853515e-05,
      "step": 49849,
      "training_step_time": 6.656651258468628
    },
    {
      "epoch": 4.7540664672851564e-05,
      "grad_norm": 0.11380663514137268,
      "learning_rate": 5.437773594149485e-06,
      "loss": 0.0551,
      "step": 49850
    },
    {
      "epoch": 4.7540664672851564e-05,
      "model_forward_time": 1.6816432476043701,
      "step": 49850
    },
    {
      "dynamics_loss_avg": 0.04643588699400425,
      "epoch": 4.7540664672851564e-05,
      "step": 49850
    },
    {
      "action_loss_avg": 0.002045763688511215,
      "epoch": 4.7540664672851564e-05,
      "step": 49850
    },
    {
      "epoch": 4.7540664672851564e-05,
      "step": 49850,
      "training_step_time": 6.5605340003967285
    },
    {
      "epoch": 4.754161834716797e-05,
      "model_forward_time": 1.7134804725646973,
      "step": 49851
    },
    {
      "epoch": 4.754161834716797e-05,
      "step": 49851,
      "training_step_time": 7.642480850219727
    },
    {
      "epoch": 4.754257202148437e-05,
      "model_forward_time": 2.2679412364959717,
      "step": 49852
    },
    {
      "epoch": 4.754257202148437e-05,
      "step": 49852,
      "training_step_time": 7.595479965209961
    },
    {
      "epoch": 4.754352569580078e-05,
      "model_forward_time": 1.71360182762146,
      "step": 49853
    },
    {
      "epoch": 4.754352569580078e-05,
      "step": 49853,
      "training_step_time": 8.271665573120117
    },
    {
      "epoch": 4.754447937011719e-05,
      "model_forward_time": 1.7038943767547607,
      "step": 49854
    },
    {
      "epoch": 4.754447937011719e-05,
      "step": 49854,
      "training_step_time": 8.629578828811646
    },
    {
      "epoch": 4.754543304443359e-05,
      "model_forward_time": 1.7077925205230713,
      "step": 49855
    },
    {
      "epoch": 4.754543304443359e-05,
      "step": 49855,
      "training_step_time": 7.5921876430511475
    },
    {
      "epoch": 4.754638671875e-05,
      "model_forward_time": 1.7126812934875488,
      "step": 49856
    },
    {
      "epoch": 4.754638671875e-05,
      "step": 49856,
      "training_step_time": 7.560307264328003
    },
    {
      "epoch": 4.7547340393066406e-05,
      "model_forward_time": 1.7008354663848877,
      "step": 49857
    },
    {
      "epoch": 4.7547340393066406e-05,
      "step": 49857,
      "training_step_time": 6.981063365936279
    },
    {
      "epoch": 4.754829406738281e-05,
      "model_forward_time": 1.7020034790039062,
      "step": 49858
    },
    {
      "epoch": 4.754829406738281e-05,
      "step": 49858,
      "training_step_time": 6.657077789306641
    },
    {
      "epoch": 4.754924774169922e-05,
      "model_forward_time": 1.7071621417999268,
      "step": 49859
    },
    {
      "epoch": 4.754924774169922e-05,
      "step": 49859,
      "training_step_time": 6.655916213989258
    },
    {
      "epoch": 4.7550201416015624e-05,
      "grad_norm": 0.10804694890975952,
      "learning_rate": 5.436126450225495e-06,
      "loss": 0.0594,
      "step": 49860
    },
    {
      "epoch": 4.7550201416015624e-05,
      "model_forward_time": 1.710719347000122,
      "step": 49860
    },
    {
      "dynamics_loss_avg": 0.0629750156775117,
      "epoch": 4.7550201416015624e-05,
      "step": 49860
    },
    {
      "action_loss_avg": 0.0017012073221849278,
      "epoch": 4.7550201416015624e-05,
      "step": 49860
    },
    {
      "epoch": 4.7550201416015624e-05,
      "step": 49860,
      "training_step_time": 8.339755296707153
    },
    {
      "epoch": 4.755115509033203e-05,
      "model_forward_time": 2.2311291694641113,
      "step": 49861
    },
    {
      "epoch": 4.755115509033203e-05,
      "step": 49861,
      "training_step_time": 8.22055983543396
    },
    {
      "epoch": 4.755210876464844e-05,
      "model_forward_time": 1.708282232284546,
      "step": 49862
    },
    {
      "epoch": 4.755210876464844e-05,
      "step": 49862,
      "training_step_time": 8.743223190307617
    },
    {
      "epoch": 4.755306243896484e-05,
      "model_forward_time": 1.7098500728607178,
      "step": 49863
    },
    {
      "epoch": 4.755306243896484e-05,
      "step": 49863,
      "training_step_time": 9.376492500305176
    },
    {
      "epoch": 4.755401611328125e-05,
      "model_forward_time": 1.714496374130249,
      "step": 49864
    },
    {
      "epoch": 4.755401611328125e-05,
      "step": 49864,
      "training_step_time": 7.6978185176849365
    },
    {
      "epoch": 4.755496978759766e-05,
      "model_forward_time": 1.7053980827331543,
      "step": 49865
    },
    {
      "epoch": 4.755496978759766e-05,
      "step": 49865,
      "training_step_time": 8.28180718421936
    },
    {
      "epoch": 4.755592346191406e-05,
      "model_forward_time": 1.7088737487792969,
      "step": 49866
    },
    {
      "epoch": 4.755592346191406e-05,
      "step": 49866,
      "training_step_time": 6.9933555126190186
    },
    {
      "epoch": 4.7556877136230466e-05,
      "model_forward_time": 1.709507703781128,
      "step": 49867
    },
    {
      "epoch": 4.7556877136230466e-05,
      "step": 49867,
      "training_step_time": 6.739840269088745
    },
    {
      "epoch": 4.755783081054688e-05,
      "model_forward_time": 1.707763433456421,
      "step": 49868
    },
    {
      "epoch": 4.755783081054688e-05,
      "step": 49868,
      "training_step_time": 6.7039854526519775
    },
    {
      "epoch": 4.755878448486328e-05,
      "model_forward_time": 1.6982953548431396,
      "step": 49869
    },
    {
      "epoch": 4.755878448486328e-05,
      "step": 49869,
      "training_step_time": 8.40639877319336
    },
    {
      "epoch": 4.7559738159179685e-05,
      "grad_norm": 0.0845237746834755,
      "learning_rate": 5.434479258607371e-06,
      "loss": 0.0582,
      "step": 49870
    },
    {
      "epoch": 4.7559738159179685e-05,
      "model_forward_time": 1.7341115474700928,
      "step": 49870
    },
    {
      "dynamics_loss_avg": 0.057333859242498876,
      "epoch": 4.7559738159179685e-05,
      "step": 49870
    },
    {
      "action_loss_avg": 0.0021689399873139337,
      "epoch": 4.7559738159179685e-05,
      "step": 49870
    },
    {
      "epoch": 4.7559738159179685e-05,
      "step": 49870,
      "training_step_time": 8.744051694869995
    },
    {
      "epoch": 4.7560691833496096e-05,
      "model_forward_time": 2.3153562545776367,
      "step": 49871
    },
    {
      "epoch": 4.7560691833496096e-05,
      "step": 49871,
      "training_step_time": 8.21708607673645
    },
    {
      "epoch": 4.75616455078125e-05,
      "model_forward_time": 1.695784568786621,
      "step": 49872
    },
    {
      "epoch": 4.75616455078125e-05,
      "step": 49872,
      "training_step_time": 8.24177861213684
    },
    {
      "epoch": 4.7562599182128904e-05,
      "model_forward_time": 1.7119009494781494,
      "step": 49873
    },
    {
      "epoch": 4.7562599182128904e-05,
      "step": 49873,
      "training_step_time": 8.24774169921875
    },
    {
      "epoch": 4.7563552856445315e-05,
      "model_forward_time": 1.701793909072876,
      "step": 49874
    },
    {
      "epoch": 4.7563552856445315e-05,
      "step": 49874,
      "training_step_time": 7.554064989089966
    },
    {
      "epoch": 4.756450653076172e-05,
      "model_forward_time": 1.7060928344726562,
      "step": 49875
    },
    {
      "epoch": 4.756450653076172e-05,
      "step": 49875,
      "training_step_time": 7.243326663970947
    },
    {
      "epoch": 4.756546020507812e-05,
      "model_forward_time": 1.712400197982788,
      "step": 49876
    },
    {
      "epoch": 4.756546020507812e-05,
      "step": 49876,
      "training_step_time": 6.697695016860962
    },
    {
      "epoch": 4.7566413879394534e-05,
      "model_forward_time": 1.7067415714263916,
      "step": 49877
    },
    {
      "epoch": 4.7566413879394534e-05,
      "step": 49877,
      "training_step_time": 6.655112028121948
    },
    {
      "epoch": 4.756736755371094e-05,
      "model_forward_time": 1.704918622970581,
      "step": 49878
    },
    {
      "epoch": 4.756736755371094e-05,
      "step": 49878,
      "training_step_time": 7.619826555252075
    },
    {
      "epoch": 4.756832122802734e-05,
      "model_forward_time": 1.7042934894561768,
      "step": 49879
    },
    {
      "epoch": 4.756832122802734e-05,
      "step": 49879,
      "training_step_time": 8.397762775421143
    },
    {
      "epoch": 4.756927490234375e-05,
      "grad_norm": 0.09600339829921722,
      "learning_rate": 5.432832019475251e-06,
      "loss": 0.0551,
      "step": 49880
    },
    {
      "epoch": 4.756927490234375e-05,
      "model_forward_time": 2.317324161529541,
      "step": 49880
    },
    {
      "dynamics_loss_avg": 0.0663253277540207,
      "epoch": 4.756927490234375e-05,
      "step": 49880
    },
    {
      "action_loss_avg": 0.001633782871067524,
      "epoch": 4.756927490234375e-05,
      "step": 49880
    },
    {
      "epoch": 4.756927490234375e-05,
      "step": 49880,
      "training_step_time": 8.914390802383423
    },
    {
      "epoch": 4.7570228576660156e-05,
      "model_forward_time": 1.7049026489257812,
      "step": 49881
    },
    {
      "epoch": 4.7570228576660156e-05,
      "step": 49881,
      "training_step_time": 10.10486626625061
    },
    {
      "epoch": 4.757118225097656e-05,
      "model_forward_time": 1.7033312320709229,
      "step": 49882
    },
    {
      "epoch": 4.757118225097656e-05,
      "step": 49882,
      "training_step_time": 7.689859390258789
    },
    {
      "epoch": 4.757213592529297e-05,
      "model_forward_time": 1.7037310600280762,
      "step": 49883
    },
    {
      "epoch": 4.757213592529297e-05,
      "step": 49883,
      "training_step_time": 7.656816005706787
    },
    {
      "epoch": 4.7573089599609375e-05,
      "model_forward_time": 1.7126708030700684,
      "step": 49884
    },
    {
      "epoch": 4.7573089599609375e-05,
      "step": 49884,
      "training_step_time": 7.324492692947388
    },
    {
      "epoch": 4.757404327392578e-05,
      "model_forward_time": 1.7072100639343262,
      "step": 49885
    },
    {
      "epoch": 4.757404327392578e-05,
      "step": 49885,
      "training_step_time": 7.219571352005005
    },
    {
      "epoch": 4.757499694824219e-05,
      "model_forward_time": 1.705761432647705,
      "step": 49886
    },
    {
      "epoch": 4.757499694824219e-05,
      "step": 49886,
      "training_step_time": 6.692682981491089
    },
    {
      "epoch": 4.7575950622558594e-05,
      "model_forward_time": 1.707874059677124,
      "step": 49887
    },
    {
      "epoch": 4.7575950622558594e-05,
      "step": 49887,
      "training_step_time": 7.652342319488525
    },
    {
      "epoch": 4.7576904296875e-05,
      "model_forward_time": 1.705690860748291,
      "step": 49888
    },
    {
      "epoch": 4.7576904296875e-05,
      "step": 49888,
      "training_step_time": 7.620285272598267
    },
    {
      "epoch": 4.757785797119141e-05,
      "model_forward_time": 2.3086917400360107,
      "step": 49889
    },
    {
      "epoch": 4.757785797119141e-05,
      "step": 49889,
      "training_step_time": 7.746981859207153
    },
    {
      "epoch": 4.757881164550781e-05,
      "grad_norm": 0.09428927302360535,
      "learning_rate": 5.431184733009268e-06,
      "loss": 0.0553,
      "step": 49890
    },
    {
      "epoch": 4.757881164550781e-05,
      "model_forward_time": 1.7073454856872559,
      "step": 49890
    },
    {
      "dynamics_loss_avg": 0.0594082061201334,
      "epoch": 4.757881164550781e-05,
      "step": 49890
    },
    {
      "action_loss_avg": 0.0026213800971163436,
      "epoch": 4.757881164550781e-05,
      "step": 49890
    },
    {
      "epoch": 4.757881164550781e-05,
      "step": 49890,
      "training_step_time": 10.759212493896484
    },
    {
      "epoch": 4.757976531982422e-05,
      "model_forward_time": 1.7088067531585693,
      "step": 49891
    },
    {
      "epoch": 4.757976531982422e-05,
      "step": 49891,
      "training_step_time": 7.6004884243011475
    },
    {
      "epoch": 4.758071899414063e-05,
      "model_forward_time": 1.7153534889221191,
      "step": 49892
    },
    {
      "epoch": 4.758071899414063e-05,
      "step": 49892,
      "training_step_time": 7.665625810623169
    },
    {
      "epoch": 4.758167266845703e-05,
      "model_forward_time": 1.7175474166870117,
      "step": 49893
    },
    {
      "epoch": 4.758167266845703e-05,
      "step": 49893,
      "training_step_time": 7.538248062133789
    },
    {
      "epoch": 4.7582626342773436e-05,
      "model_forward_time": 1.7095344066619873,
      "step": 49894
    },
    {
      "epoch": 4.7582626342773436e-05,
      "step": 49894,
      "training_step_time": 7.193316221237183
    },
    {
      "epoch": 4.758358001708985e-05,
      "model_forward_time": 1.7184436321258545,
      "step": 49895
    },
    {
      "epoch": 4.758358001708985e-05,
      "step": 49895,
      "training_step_time": 6.639561653137207
    },
    {
      "epoch": 4.758453369140625e-05,
      "model_forward_time": 1.7075984477996826,
      "step": 49896
    },
    {
      "epoch": 4.758453369140625e-05,
      "step": 49896,
      "training_step_time": 7.041651964187622
    },
    {
      "epoch": 4.7585487365722655e-05,
      "model_forward_time": 1.7255737781524658,
      "step": 49897
    },
    {
      "epoch": 4.7585487365722655e-05,
      "step": 49897,
      "training_step_time": 7.593902111053467
    },
    {
      "epoch": 4.7586441040039066e-05,
      "model_forward_time": 2.3056020736694336,
      "step": 49898
    },
    {
      "epoch": 4.7586441040039066e-05,
      "step": 49898,
      "training_step_time": 8.213748216629028
    },
    {
      "epoch": 4.758739471435547e-05,
      "model_forward_time": 1.7946929931640625,
      "step": 49899
    },
    {
      "epoch": 4.758739471435547e-05,
      "step": 49899,
      "training_step_time": 7.670565605163574
    },
    {
      "epoch": 4.7588348388671874e-05,
      "grad_norm": 0.07980974018573761,
      "learning_rate": 5.429537399389574e-06,
      "loss": 0.0557,
      "step": 49900
    },
    {
      "epoch": 4.7588348388671874e-05,
      "model_forward_time": 1.7251145839691162,
      "step": 49900
    },
    {
      "dynamics_loss_avg": 0.053221140429377554,
      "epoch": 4.7588348388671874e-05,
      "step": 49900
    },
    {
      "action_loss_avg": 0.0015320756356231868,
      "epoch": 4.7588348388671874e-05,
      "step": 49900
    },
    {
      "epoch": 4.7588348388671874e-05,
      "step": 49900,
      "training_step_time": 8.64673399925232
    },
    {
      "epoch": 4.7589302062988284e-05,
      "model_forward_time": 1.7272753715515137,
      "step": 49901
    },
    {
      "epoch": 4.7589302062988284e-05,
      "step": 49901,
      "training_step_time": 9.409162759780884
    },
    {
      "epoch": 4.759025573730469e-05,
      "model_forward_time": 1.7387783527374268,
      "step": 49902
    },
    {
      "epoch": 4.759025573730469e-05,
      "step": 49902,
      "training_step_time": 7.553098440170288
    },
    {
      "epoch": 4.759120941162109e-05,
      "model_forward_time": 1.7264866828918457,
      "step": 49903
    },
    {
      "epoch": 4.759120941162109e-05,
      "step": 49903,
      "training_step_time": 7.2460527420043945
    },
    {
      "epoch": 4.75921630859375e-05,
      "model_forward_time": 1.7248282432556152,
      "step": 49904
    },
    {
      "epoch": 4.75921630859375e-05,
      "step": 49904,
      "training_step_time": 6.690274477005005
    },
    {
      "epoch": 4.759311676025391e-05,
      "model_forward_time": 1.7207615375518799,
      "step": 49905
    },
    {
      "epoch": 4.759311676025391e-05,
      "step": 49905,
      "training_step_time": 7.069867372512817
    },
    {
      "epoch": 4.759407043457031e-05,
      "model_forward_time": 1.7319278717041016,
      "step": 49906
    },
    {
      "epoch": 4.759407043457031e-05,
      "step": 49906,
      "training_step_time": 7.700194835662842
    },
    {
      "epoch": 4.7595024108886715e-05,
      "model_forward_time": 1.877945899963379,
      "step": 49907
    },
    {
      "epoch": 4.7595024108886715e-05,
      "step": 49907,
      "training_step_time": 7.707218170166016
    },
    {
      "epoch": 4.7595977783203126e-05,
      "model_forward_time": 1.7690582275390625,
      "step": 49908
    },
    {
      "epoch": 4.7595977783203126e-05,
      "step": 49908,
      "training_step_time": 7.62383508682251
    },
    {
      "epoch": 4.759693145751953e-05,
      "model_forward_time": 1.7260196208953857,
      "step": 49909
    },
    {
      "epoch": 4.759693145751953e-05,
      "step": 49909,
      "training_step_time": 9.51034164428711
    },
    {
      "epoch": 4.7597885131835934e-05,
      "grad_norm": 0.1207614466547966,
      "learning_rate": 5.427890018796315e-06,
      "loss": 0.0552,
      "step": 49910
    },
    {
      "epoch": 4.7597885131835934e-05,
      "model_forward_time": 1.7207765579223633,
      "step": 49910
    },
    {
      "dynamics_loss_avg": 0.04441819433122873,
      "epoch": 4.7597885131835934e-05,
      "step": 49910
    },
    {
      "action_loss_avg": 0.0020110680750804024,
      "epoch": 4.7597885131835934e-05,
      "step": 49910
    },
    {
      "epoch": 4.7597885131835934e-05,
      "step": 49910,
      "training_step_time": 7.693047285079956
    },
    {
      "epoch": 4.7598838806152345e-05,
      "model_forward_time": 1.7265052795410156,
      "step": 49911
    },
    {
      "epoch": 4.7598838806152345e-05,
      "step": 49911,
      "training_step_time": 7.616971969604492
    },
    {
      "epoch": 4.759979248046875e-05,
      "model_forward_time": 1.728557825088501,
      "step": 49912
    },
    {
      "epoch": 4.759979248046875e-05,
      "step": 49912,
      "training_step_time": 7.187703847885132
    },
    {
      "epoch": 4.760074615478515e-05,
      "model_forward_time": 1.7257959842681885,
      "step": 49913
    },
    {
      "epoch": 4.760074615478515e-05,
      "step": 49913,
      "training_step_time": 6.726954460144043
    },
    {
      "epoch": 4.7601699829101564e-05,
      "model_forward_time": 1.7260634899139404,
      "step": 49914
    },
    {
      "epoch": 4.7601699829101564e-05,
      "step": 49914,
      "training_step_time": 7.066338539123535
    },
    {
      "epoch": 4.760265350341797e-05,
      "model_forward_time": 1.7265138626098633,
      "step": 49915
    },
    {
      "epoch": 4.760265350341797e-05,
      "step": 49915,
      "training_step_time": 8.30186915397644
    },
    {
      "epoch": 4.760360717773437e-05,
      "model_forward_time": 1.325420618057251,
      "step": 49916
    },
    {
      "epoch": 4.760360717773437e-05,
      "step": 49916,
      "training_step_time": 7.639307737350464
    },
    {
      "epoch": 4.760456085205078e-05,
      "model_forward_time": 2.351146936416626,
      "step": 49917
    },
    {
      "epoch": 4.760456085205078e-05,
      "step": 49917,
      "training_step_time": 8.356873750686646
    },
    {
      "epoch": 4.760551452636719e-05,
      "model_forward_time": 1.7399542331695557,
      "step": 49918
    },
    {
      "epoch": 4.760551452636719e-05,
      "step": 49918,
      "training_step_time": 7.625396966934204
    },
    {
      "epoch": 4.760646820068359e-05,
      "model_forward_time": 1.7291920185089111,
      "step": 49919
    },
    {
      "epoch": 4.760646820068359e-05,
      "step": 49919,
      "training_step_time": 8.639196634292603
    },
    {
      "epoch": 4.7607421875e-05,
      "grad_norm": 0.0840516984462738,
      "learning_rate": 5.426242591409646e-06,
      "loss": 0.0527,
      "step": 49920
    },
    {
      "epoch": 4.7607421875e-05,
      "model_forward_time": 1.719008207321167,
      "step": 49920
    },
    {
      "dynamics_loss_avg": 0.04958855472505093,
      "epoch": 4.7607421875e-05,
      "step": 49920
    },
    {
      "action_loss_avg": 0.0012782420380972326,
      "epoch": 4.7607421875e-05,
      "step": 49920
    },
    {
      "epoch": 4.7607421875e-05,
      "step": 49920,
      "training_step_time": 7.624726057052612
    },
    {
      "epoch": 4.7608375549316406e-05,
      "model_forward_time": 1.7306969165802002,
      "step": 49921
    },
    {
      "epoch": 4.7608375549316406e-05,
      "step": 49921,
      "training_step_time": 6.7347986698150635
    },
    {
      "epoch": 4.760932922363281e-05,
      "model_forward_time": 1.7395155429840088,
      "step": 49922
    },
    {
      "epoch": 4.760932922363281e-05,
      "step": 49922,
      "training_step_time": 7.141072750091553
    },
    {
      "epoch": 4.761028289794922e-05,
      "model_forward_time": 1.7251121997833252,
      "step": 49923
    },
    {
      "epoch": 4.761028289794922e-05,
      "step": 49923,
      "training_step_time": 7.055273771286011
    },
    {
      "epoch": 4.7611236572265625e-05,
      "model_forward_time": 1.7229650020599365,
      "step": 49924
    },
    {
      "epoch": 4.7611236572265625e-05,
      "step": 49924,
      "training_step_time": 7.259519815444946
    },
    {
      "epoch": 4.761219024658203e-05,
      "model_forward_time": 1.7232131958007812,
      "step": 49925
    },
    {
      "epoch": 4.761219024658203e-05,
      "step": 49925,
      "training_step_time": 8.515480518341064
    },
    {
      "epoch": 4.761314392089844e-05,
      "model_forward_time": 2.2747063636779785,
      "step": 49926
    },
    {
      "epoch": 4.761314392089844e-05,
      "step": 49926,
      "training_step_time": 7.660121440887451
    },
    {
      "epoch": 4.761409759521484e-05,
      "model_forward_time": 1.7281980514526367,
      "step": 49927
    },
    {
      "epoch": 4.761409759521484e-05,
      "step": 49927,
      "training_step_time": 8.162537813186646
    },
    {
      "epoch": 4.761505126953125e-05,
      "model_forward_time": 1.7294785976409912,
      "step": 49928
    },
    {
      "epoch": 4.761505126953125e-05,
      "step": 49928,
      "training_step_time": 8.144025325775146
    },
    {
      "epoch": 4.761600494384766e-05,
      "model_forward_time": 1.7257318496704102,
      "step": 49929
    },
    {
      "epoch": 4.761600494384766e-05,
      "step": 49929,
      "training_step_time": 7.654992580413818
    },
    {
      "epoch": 4.761695861816406e-05,
      "grad_norm": 0.10151886940002441,
      "learning_rate": 5.42459511740973e-06,
      "loss": 0.0546,
      "step": 49930
    },
    {
      "epoch": 4.761695861816406e-05,
      "model_forward_time": 1.7169504165649414,
      "step": 49930
    },
    {
      "dynamics_loss_avg": 0.0498763820156455,
      "epoch": 4.761695861816406e-05,
      "step": 49930
    },
    {
      "action_loss_avg": 0.0015355699812062084,
      "epoch": 4.761695861816406e-05,
      "step": 49930
    },
    {
      "epoch": 4.761695861816406e-05,
      "step": 49930,
      "training_step_time": 7.398745775222778
    },
    {
      "epoch": 4.7617912292480466e-05,
      "model_forward_time": 1.7209789752960205,
      "step": 49931
    },
    {
      "epoch": 4.7617912292480466e-05,
      "step": 49931,
      "training_step_time": 6.98064923286438
    },
    {
      "epoch": 4.761886596679688e-05,
      "model_forward_time": 1.724898338317871,
      "step": 49932
    },
    {
      "epoch": 4.761886596679688e-05,
      "step": 49932,
      "training_step_time": 7.090442180633545
    },
    {
      "epoch": 4.761981964111328e-05,
      "model_forward_time": 1.7201881408691406,
      "step": 49933
    },
    {
      "epoch": 4.761981964111328e-05,
      "step": 49933,
      "training_step_time": 6.67200779914856
    },
    {
      "epoch": 4.7620773315429685e-05,
      "model_forward_time": 1.720193862915039,
      "step": 49934
    },
    {
      "epoch": 4.7620773315429685e-05,
      "step": 49934,
      "training_step_time": 8.2312593460083
    },
    {
      "epoch": 4.7621726989746096e-05,
      "model_forward_time": 2.2837915420532227,
      "step": 49935
    },
    {
      "epoch": 4.7621726989746096e-05,
      "step": 49935,
      "training_step_time": 7.600708246231079
    },
    {
      "epoch": 4.76226806640625e-05,
      "model_forward_time": 1.8089122772216797,
      "step": 49936
    },
    {
      "epoch": 4.76226806640625e-05,
      "step": 49936,
      "training_step_time": 7.611124753952026
    },
    {
      "epoch": 4.7623634338378904e-05,
      "model_forward_time": 1.733680009841919,
      "step": 49937
    },
    {
      "epoch": 4.7623634338378904e-05,
      "step": 49937,
      "training_step_time": 8.539180278778076
    },
    {
      "epoch": 4.7624588012695315e-05,
      "model_forward_time": 1.7107603549957275,
      "step": 49938
    },
    {
      "epoch": 4.7624588012695315e-05,
      "step": 49938,
      "training_step_time": 8.193051815032959
    },
    {
      "epoch": 4.762554168701172e-05,
      "model_forward_time": 1.7223117351531982,
      "step": 49939
    },
    {
      "epoch": 4.762554168701172e-05,
      "step": 49939,
      "training_step_time": 7.363755226135254
    },
    {
      "epoch": 4.762649536132812e-05,
      "grad_norm": 0.1057247593998909,
      "learning_rate": 5.422947596976729e-06,
      "loss": 0.0517,
      "step": 49940
    },
    {
      "epoch": 4.762649536132812e-05,
      "model_forward_time": 1.731250286102295,
      "step": 49940
    },
    {
      "dynamics_loss_avg": 0.07086027823388577,
      "epoch": 4.762649536132812e-05,
      "step": 49940
    },
    {
      "action_loss_avg": 0.0022575019276700912,
      "epoch": 4.762649536132812e-05,
      "step": 49940
    },
    {
      "epoch": 4.762649536132812e-05,
      "step": 49940,
      "training_step_time": 7.15936541557312
    },
    {
      "epoch": 4.7627449035644534e-05,
      "model_forward_time": 1.7278859615325928,
      "step": 49941
    },
    {
      "epoch": 4.7627449035644534e-05,
      "step": 49941,
      "training_step_time": 7.098961591720581
    },
    {
      "epoch": 4.762840270996094e-05,
      "model_forward_time": 1.7230603694915771,
      "step": 49942
    },
    {
      "epoch": 4.762840270996094e-05,
      "step": 49942,
      "training_step_time": 6.699668884277344
    },
    {
      "epoch": 4.762935638427734e-05,
      "model_forward_time": 1.7225234508514404,
      "step": 49943
    },
    {
      "epoch": 4.762935638427734e-05,
      "step": 49943,
      "training_step_time": 8.554637432098389
    },
    {
      "epoch": 4.763031005859375e-05,
      "model_forward_time": 2.2924094200134277,
      "step": 49944
    },
    {
      "epoch": 4.763031005859375e-05,
      "step": 49944,
      "training_step_time": 7.602296829223633
    },
    {
      "epoch": 4.7631263732910157e-05,
      "model_forward_time": 1.7794134616851807,
      "step": 49945
    },
    {
      "epoch": 4.7631263732910157e-05,
      "step": 49945,
      "training_step_time": 7.640181064605713
    },
    {
      "epoch": 4.763221740722656e-05,
      "model_forward_time": 1.7254528999328613,
      "step": 49946
    },
    {
      "epoch": 4.763221740722656e-05,
      "step": 49946,
      "training_step_time": 8.701500415802002
    },
    {
      "epoch": 4.763317108154297e-05,
      "model_forward_time": 1.724344253540039,
      "step": 49947
    },
    {
      "epoch": 4.763317108154297e-05,
      "step": 49947,
      "training_step_time": 8.244955778121948
    },
    {
      "epoch": 4.7634124755859375e-05,
      "model_forward_time": 1.722998857498169,
      "step": 49948
    },
    {
      "epoch": 4.7634124755859375e-05,
      "step": 49948,
      "training_step_time": 7.563117980957031
    },
    {
      "epoch": 4.763507843017578e-05,
      "model_forward_time": 1.727696180343628,
      "step": 49949
    },
    {
      "epoch": 4.763507843017578e-05,
      "step": 49949,
      "training_step_time": 7.205801010131836
    },
    {
      "epoch": 4.763603210449219e-05,
      "grad_norm": 0.08956954628229141,
      "learning_rate": 5.421300030290814e-06,
      "loss": 0.0539,
      "step": 49950
    },
    {
      "epoch": 4.763603210449219e-05,
      "model_forward_time": 1.7225050926208496,
      "step": 49950
    },
    {
      "dynamics_loss_avg": 0.0379774296656251,
      "epoch": 4.763603210449219e-05,
      "step": 49950
    },
    {
      "action_loss_avg": 0.0011097432754468173,
      "epoch": 4.763603210449219e-05,
      "step": 49950
    },
    {
      "epoch": 4.763603210449219e-05,
      "step": 49950,
      "training_step_time": 7.21633768081665
    },
    {
      "epoch": 4.7636985778808594e-05,
      "model_forward_time": 1.7190930843353271,
      "step": 49951
    },
    {
      "epoch": 4.7636985778808594e-05,
      "step": 49951,
      "training_step_time": 6.655861854553223
    },
    {
      "epoch": 4.7637939453125e-05,
      "model_forward_time": 1.726402997970581,
      "step": 49952
    },
    {
      "epoch": 4.7637939453125e-05,
      "step": 49952,
      "training_step_time": 7.759684324264526
    },
    {
      "epoch": 4.763889312744141e-05,
      "model_forward_time": 1.7271382808685303,
      "step": 49953
    },
    {
      "epoch": 4.763889312744141e-05,
      "step": 49953,
      "training_step_time": 8.36209225654602
    },
    {
      "epoch": 4.763984680175781e-05,
      "model_forward_time": 1.7746813297271729,
      "step": 49954
    },
    {
      "epoch": 4.763984680175781e-05,
      "step": 49954,
      "training_step_time": 7.631159067153931
    },
    {
      "epoch": 4.764080047607422e-05,
      "model_forward_time": 1.7234978675842285,
      "step": 49955
    },
    {
      "epoch": 4.764080047607422e-05,
      "step": 49955,
      "training_step_time": 10.328577756881714
    },
    {
      "epoch": 4.764175415039063e-05,
      "model_forward_time": 1.7224817276000977,
      "step": 49956
    },
    {
      "epoch": 4.764175415039063e-05,
      "step": 49956,
      "training_step_time": 7.646865606307983
    },
    {
      "epoch": 4.764270782470703e-05,
      "model_forward_time": 1.7318918704986572,
      "step": 49957
    },
    {
      "epoch": 4.764270782470703e-05,
      "step": 49957,
      "training_step_time": 7.631193161010742
    },
    {
      "epoch": 4.7643661499023436e-05,
      "model_forward_time": 1.7401936054229736,
      "step": 49958
    },
    {
      "epoch": 4.7643661499023436e-05,
      "step": 49958,
      "training_step_time": 6.691691875457764
    },
    {
      "epoch": 4.764461517333985e-05,
      "model_forward_time": 1.7273404598236084,
      "step": 49959
    },
    {
      "epoch": 4.764461517333985e-05,
      "step": 49959,
      "training_step_time": 7.786861181259155
    },
    {
      "epoch": 4.764556884765625e-05,
      "grad_norm": 0.11329040676355362,
      "learning_rate": 5.419652417532162e-06,
      "loss": 0.0566,
      "step": 49960
    },
    {
      "epoch": 4.764556884765625e-05,
      "model_forward_time": 1.718291997909546,
      "step": 49960
    },
    {
      "dynamics_loss_avg": 0.04740730160847306,
      "epoch": 4.764556884765625e-05,
      "step": 49960
    },
    {
      "action_loss_avg": 0.0011460267705842853,
      "epoch": 4.764556884765625e-05,
      "step": 49960
    },
    {
      "epoch": 4.764556884765625e-05,
      "step": 49960,
      "training_step_time": 6.6757566928863525
    },
    {
      "epoch": 4.7646522521972655e-05,
      "model_forward_time": 1.7300491333007812,
      "step": 49961
    },
    {
      "epoch": 4.7646522521972655e-05,
      "step": 49961,
      "training_step_time": 7.748902082443237
    },
    {
      "epoch": 4.7647476196289066e-05,
      "model_forward_time": 1.7272312641143799,
      "step": 49962
    },
    {
      "epoch": 4.7647476196289066e-05,
      "step": 49962,
      "training_step_time": 9.290940999984741
    },
    {
      "epoch": 4.764842987060547e-05,
      "model_forward_time": 2.3083086013793945,
      "step": 49963
    },
    {
      "epoch": 4.764842987060547e-05,
      "step": 49963,
      "training_step_time": 9.169361591339111
    },
    {
      "epoch": 4.7649383544921874e-05,
      "model_forward_time": 1.735651969909668,
      "step": 49964
    },
    {
      "epoch": 4.7649383544921874e-05,
      "step": 49964,
      "training_step_time": 8.643885612487793
    },
    {
      "epoch": 4.7650337219238285e-05,
      "model_forward_time": 1.7292170524597168,
      "step": 49965
    },
    {
      "epoch": 4.7650337219238285e-05,
      "step": 49965,
      "training_step_time": 8.128196954727173
    },
    {
      "epoch": 4.765129089355469e-05,
      "model_forward_time": 1.7311427593231201,
      "step": 49966
    },
    {
      "epoch": 4.765129089355469e-05,
      "step": 49966,
      "training_step_time": 8.281371831893921
    },
    {
      "epoch": 4.765224456787109e-05,
      "model_forward_time": 1.7282907962799072,
      "step": 49967
    },
    {
      "epoch": 4.765224456787109e-05,
      "step": 49967,
      "training_step_time": 7.386565923690796
    },
    {
      "epoch": 4.76531982421875e-05,
      "model_forward_time": 1.720374584197998,
      "step": 49968
    },
    {
      "epoch": 4.76531982421875e-05,
      "step": 49968,
      "training_step_time": 7.265423536300659
    },
    {
      "epoch": 4.765415191650391e-05,
      "model_forward_time": 1.739022970199585,
      "step": 49969
    },
    {
      "epoch": 4.765415191650391e-05,
      "step": 49969,
      "training_step_time": 7.112600326538086
    },
    {
      "epoch": 4.765510559082031e-05,
      "grad_norm": 0.10740450769662857,
      "learning_rate": 5.418004758880953e-06,
      "loss": 0.0591,
      "step": 49970
    },
    {
      "epoch": 4.765510559082031e-05,
      "model_forward_time": 1.717555046081543,
      "step": 49970
    },
    {
      "dynamics_loss_avg": 0.05122681260108948,
      "epoch": 4.765510559082031e-05,
      "step": 49970
    },
    {
      "action_loss_avg": 0.001289973233360797,
      "epoch": 4.765510559082031e-05,
      "step": 49970
    },
    {
      "epoch": 4.765510559082031e-05,
      "step": 49970,
      "training_step_time": 7.739221096038818
    },
    {
      "epoch": 4.7656059265136715e-05,
      "model_forward_time": 1.7213494777679443,
      "step": 49971
    },
    {
      "epoch": 4.7656059265136715e-05,
      "step": 49971,
      "training_step_time": 9.025576114654541
    },
    {
      "epoch": 4.7657012939453126e-05,
      "model_forward_time": 2.281373977661133,
      "step": 49972
    },
    {
      "epoch": 4.7657012939453126e-05,
      "step": 49972,
      "training_step_time": 8.46180534362793
    },
    {
      "epoch": 4.765796661376953e-05,
      "model_forward_time": 1.6992924213409424,
      "step": 49973
    },
    {
      "epoch": 4.765796661376953e-05,
      "step": 49973,
      "training_step_time": 8.486916303634644
    },
    {
      "epoch": 4.7658920288085934e-05,
      "model_forward_time": 1.718465805053711,
      "step": 49974
    },
    {
      "epoch": 4.7658920288085934e-05,
      "step": 49974,
      "training_step_time": 8.524310827255249
    },
    {
      "epoch": 4.7659873962402345e-05,
      "model_forward_time": 1.7153375148773193,
      "step": 49975
    },
    {
      "epoch": 4.7659873962402345e-05,
      "step": 49975,
      "training_step_time": 7.763248920440674
    },
    {
      "epoch": 4.766082763671875e-05,
      "model_forward_time": 1.7099215984344482,
      "step": 49976
    },
    {
      "epoch": 4.766082763671875e-05,
      "step": 49976,
      "training_step_time": 7.522280216217041
    },
    {
      "epoch": 4.766178131103515e-05,
      "model_forward_time": 1.7212238311767578,
      "step": 49977
    },
    {
      "epoch": 4.766178131103515e-05,
      "step": 49977,
      "training_step_time": 7.1867592334747314
    },
    {
      "epoch": 4.7662734985351564e-05,
      "model_forward_time": 1.3365378379821777,
      "step": 49978
    },
    {
      "epoch": 4.7662734985351564e-05,
      "step": 49978,
      "training_step_time": 7.083784818649292
    },
    {
      "epoch": 4.766368865966797e-05,
      "model_forward_time": 1.7185242176055908,
      "step": 49979
    },
    {
      "epoch": 4.766368865966797e-05,
      "step": 49979,
      "training_step_time": 7.621279954910278
    },
    {
      "epoch": 4.766464233398437e-05,
      "grad_norm": 0.08115153014659882,
      "learning_rate": 5.416357054517371e-06,
      "loss": 0.0578,
      "step": 49980
    },
    {
      "epoch": 4.766464233398437e-05,
      "model_forward_time": 1.7168662548065186,
      "step": 49980
    },
    {
      "dynamics_loss_avg": 0.0534134341403842,
      "epoch": 4.766464233398437e-05,
      "step": 49980
    },
    {
      "action_loss_avg": 0.0018382818379905075,
      "epoch": 4.766464233398437e-05,
      "step": 49980
    },
    {
      "epoch": 4.766464233398437e-05,
      "step": 49980,
      "training_step_time": 8.480406761169434
    },
    {
      "epoch": 4.766559600830078e-05,
      "model_forward_time": 2.2674624919891357,
      "step": 49981
    },
    {
      "epoch": 4.766559600830078e-05,
      "step": 49981,
      "training_step_time": 9.926011323928833
    },
    {
      "epoch": 4.766654968261719e-05,
      "model_forward_time": 1.7857306003570557,
      "step": 49982
    },
    {
      "epoch": 4.766654968261719e-05,
      "step": 49982,
      "training_step_time": 7.683958530426025
    },
    {
      "epoch": 4.766750335693359e-05,
      "model_forward_time": 1.7207584381103516,
      "step": 49983
    },
    {
      "epoch": 4.766750335693359e-05,
      "step": 49983,
      "training_step_time": 8.200064420700073
    },
    {
      "epoch": 4.766845703125e-05,
      "model_forward_time": 1.7217273712158203,
      "step": 49984
    },
    {
      "epoch": 4.766845703125e-05,
      "step": 49984,
      "training_step_time": 7.780992269515991
    },
    {
      "epoch": 4.7669410705566406e-05,
      "model_forward_time": 1.707627534866333,
      "step": 49985
    },
    {
      "epoch": 4.7669410705566406e-05,
      "step": 49985,
      "training_step_time": 7.525221347808838
    },
    {
      "epoch": 4.767036437988281e-05,
      "model_forward_time": 1.712341547012329,
      "step": 49986
    },
    {
      "epoch": 4.767036437988281e-05,
      "step": 49986,
      "training_step_time": 6.691802263259888
    },
    {
      "epoch": 4.767131805419922e-05,
      "model_forward_time": 1.3364839553833008,
      "step": 49987
    },
    {
      "epoch": 4.767131805419922e-05,
      "step": 49987,
      "training_step_time": 7.328261852264404
    },
    {
      "epoch": 4.7672271728515625e-05,
      "model_forward_time": 1.7110276222229004,
      "step": 49988
    },
    {
      "epoch": 4.7672271728515625e-05,
      "step": 49988,
      "training_step_time": 7.629587650299072
    },
    {
      "epoch": 4.767322540283203e-05,
      "model_forward_time": 1.7201566696166992,
      "step": 49989
    },
    {
      "epoch": 4.767322540283203e-05,
      "step": 49989,
      "training_step_time": 8.320194959640503
    },
    {
      "epoch": 4.767417907714844e-05,
      "grad_norm": 0.10643845051527023,
      "learning_rate": 5.4147093046216086e-06,
      "loss": 0.0577,
      "step": 49990
    },
    {
      "epoch": 4.767417907714844e-05,
      "model_forward_time": 2.2320241928100586,
      "step": 49990
    },
    {
      "dynamics_loss_avg": 0.04831713605672121,
      "epoch": 4.767417907714844e-05,
      "step": 49990
    },
    {
      "action_loss_avg": 0.0014357039064634592,
      "epoch": 4.767417907714844e-05,
      "step": 49990
    },
    {
      "epoch": 4.767417907714844e-05,
      "step": 49990,
      "training_step_time": 7.682518720626831
    },
    {
      "epoch": 4.7675132751464843e-05,
      "model_forward_time": 1.7657620906829834,
      "step": 49991
    },
    {
      "epoch": 4.7675132751464843e-05,
      "step": 49991,
      "training_step_time": 9.50527548789978
    },
    {
      "epoch": 4.767608642578125e-05,
      "model_forward_time": 1.7006535530090332,
      "step": 49992
    },
    {
      "epoch": 4.767608642578125e-05,
      "step": 49992,
      "training_step_time": 9.456840753555298
    },
    {
      "epoch": 4.767704010009766e-05,
      "model_forward_time": 1.6977319717407227,
      "step": 49993
    },
    {
      "epoch": 4.767704010009766e-05,
      "step": 49993,
      "training_step_time": 7.661780595779419
    },
    {
      "epoch": 4.767799377441406e-05,
      "model_forward_time": 1.709538221359253,
      "step": 49994
    },
    {
      "epoch": 4.767799377441406e-05,
      "step": 49994,
      "training_step_time": 7.513300180435181
    },
    {
      "epoch": 4.7678947448730466e-05,
      "model_forward_time": 1.711681842803955,
      "step": 49995
    },
    {
      "epoch": 4.7678947448730466e-05,
      "step": 49995,
      "training_step_time": 6.950203895568848
    },
    {
      "epoch": 4.767990112304688e-05,
      "model_forward_time": 1.7130532264709473,
      "step": 49996
    },
    {
      "epoch": 4.767990112304688e-05,
      "step": 49996,
      "training_step_time": 7.127453327178955
    },
    {
      "epoch": 4.768085479736328e-05,
      "model_forward_time": 1.7228431701660156,
      "step": 49997
    },
    {
      "epoch": 4.768085479736328e-05,
      "step": 49997,
      "training_step_time": 7.581279993057251
    },
    {
      "epoch": 4.7681808471679685e-05,
      "model_forward_time": 1.755716323852539,
      "step": 49998
    },
    {
      "epoch": 4.7681808471679685e-05,
      "step": 49998,
      "training_step_time": 8.28560471534729
    },
    {
      "epoch": 4.7682762145996096e-05,
      "model_forward_time": 1.7117271423339844,
      "step": 49999
    },
    {
      "epoch": 4.7682762145996096e-05,
      "step": 49999,
      "training_step_time": 8.00117039680481
    },
    {
      "epoch": 4.76837158203125e-05,
      "grad_norm": 0.10224605351686478,
      "learning_rate": 5.413061509373858e-06,
      "loss": 0.0615,
      "step": 50000
    },
    {
      "epoch": 4.76837158203125e-05,
      "model_forward_time": 1.7567052841186523,
      "step": 50000
    },
    {
      "dynamics_loss_avg": 0.056872112676501274,
      "epoch": 4.76837158203125e-05,
      "step": 50000
    },
    {
      "action_loss_avg": 0.0014428528142161668,
      "epoch": 4.76837158203125e-05,
      "step": 50000
    },
    {
      "epoch": 4.76837158203125e-05,
      "step": 50000,
      "training_step_time": 11.302716732025146
    },
    {
      "epoch": 4.7684669494628904e-05,
      "model_forward_time": 1.7251436710357666,
      "step": 50001
    },
    {
      "epoch": 4.7684669494628904e-05,
      "step": 50001,
      "training_step_time": 10.47501826286316
    },
    {
      "epoch": 4.7685623168945315e-05,
      "model_forward_time": 1.7230939865112305,
      "step": 50002
    },
    {
      "epoch": 4.7685623168945315e-05,
      "step": 50002,
      "training_step_time": 9.811557054519653
    },
    {
      "epoch": 4.768657684326172e-05,
      "model_forward_time": 1.722416639328003,
      "step": 50003
    },
    {
      "epoch": 4.768657684326172e-05,
      "step": 50003,
      "training_step_time": 7.651434421539307
    },
    {
      "epoch": 4.768753051757812e-05,
      "model_forward_time": 1.721372127532959,
      "step": 50004
    },
    {
      "epoch": 4.768753051757812e-05,
      "step": 50004,
      "training_step_time": 7.372648239135742
    },
    {
      "epoch": 4.7688484191894534e-05,
      "model_forward_time": 1.7431864738464355,
      "step": 50005
    },
    {
      "epoch": 4.7688484191894534e-05,
      "step": 50005,
      "training_step_time": 7.625680208206177
    },
    {
      "epoch": 4.768943786621094e-05,
      "model_forward_time": 1.7231359481811523,
      "step": 50006
    },
    {
      "epoch": 4.768943786621094e-05,
      "step": 50006,
      "training_step_time": 8.258110046386719
    },
    {
      "epoch": 4.769039154052734e-05,
      "model_forward_time": 1.7247843742370605,
      "step": 50007
    },
    {
      "epoch": 4.769039154052734e-05,
      "step": 50007,
      "training_step_time": 8.440950632095337
    },
    {
      "epoch": 4.769134521484375e-05,
      "model_forward_time": 2.7307558059692383,
      "step": 50008
    },
    {
      "epoch": 4.769134521484375e-05,
      "step": 50008,
      "training_step_time": 9.690454959869385
    },
    {
      "epoch": 4.7692298889160157e-05,
      "model_forward_time": 1.7287867069244385,
      "step": 50009
    },
    {
      "epoch": 4.7692298889160157e-05,
      "step": 50009,
      "training_step_time": 9.629385471343994
    },
    {
      "epoch": 4.769325256347656e-05,
      "grad_norm": 0.09328895807266235,
      "learning_rate": 5.411413668954324e-06,
      "loss": 0.0632,
      "step": 50010
    },
    {
      "epoch": 4.769325256347656e-05,
      "model_forward_time": 1.715148687362671,
      "step": 50010
    },
    {
      "dynamics_loss_avg": 0.04994794260710478,
      "epoch": 4.769325256347656e-05,
      "step": 50010
    },
    {
      "action_loss_avg": 0.002058042772114277,
      "epoch": 4.769325256347656e-05,
      "step": 50010
    },
    {
      "epoch": 4.769325256347656e-05,
      "step": 50010,
      "training_step_time": 8.532367706298828
    },
    {
      "epoch": 4.769420623779297e-05,
      "model_forward_time": 1.7186007499694824,
      "step": 50011
    },
    {
      "epoch": 4.769420623779297e-05,
      "step": 50011,
      "training_step_time": 8.229995965957642
    },
    {
      "epoch": 4.7695159912109375e-05,
      "model_forward_time": 1.727257490158081,
      "step": 50012
    },
    {
      "epoch": 4.7695159912109375e-05,
      "step": 50012,
      "training_step_time": 7.695413827896118
    },
    {
      "epoch": 4.769611358642578e-05,
      "model_forward_time": 1.7201781272888184,
      "step": 50013
    },
    {
      "epoch": 4.769611358642578e-05,
      "step": 50013,
      "training_step_time": 7.5085413455963135
    },
    {
      "epoch": 4.769706726074219e-05,
      "model_forward_time": 1.706803798675537,
      "step": 50014
    },
    {
      "epoch": 4.769706726074219e-05,
      "step": 50014,
      "training_step_time": 7.1267266273498535
    },
    {
      "epoch": 4.7698020935058594e-05,
      "model_forward_time": 1.737621545791626,
      "step": 50015
    },
    {
      "epoch": 4.7698020935058594e-05,
      "step": 50015,
      "training_step_time": 7.778438568115234
    },
    {
      "epoch": 4.7698974609375e-05,
      "model_forward_time": 1.7411048412322998,
      "step": 50016
    },
    {
      "epoch": 4.7698974609375e-05,
      "step": 50016,
      "training_step_time": 8.47592806816101
    },
    {
      "epoch": 4.769992828369141e-05,
      "model_forward_time": 2.2776336669921875,
      "step": 50017
    },
    {
      "epoch": 4.769992828369141e-05,
      "step": 50017,
      "training_step_time": 8.437044382095337
    },
    {
      "epoch": 4.770088195800781e-05,
      "model_forward_time": 1.7015776634216309,
      "step": 50018
    },
    {
      "epoch": 4.770088195800781e-05,
      "step": 50018,
      "training_step_time": 8.16956090927124
    },
    {
      "epoch": 4.770183563232422e-05,
      "model_forward_time": 1.8120648860931396,
      "step": 50019
    },
    {
      "epoch": 4.770183563232422e-05,
      "step": 50019,
      "training_step_time": 7.625888109207153
    },
    {
      "epoch": 4.770278930664063e-05,
      "grad_norm": 0.14754487574100494,
      "learning_rate": 5.4097657835432085e-06,
      "loss": 0.0613,
      "step": 50020
    },
    {
      "epoch": 4.770278930664063e-05,
      "model_forward_time": 1.7291169166564941,
      "step": 50020
    },
    {
      "dynamics_loss_avg": 0.061622010357677934,
      "epoch": 4.770278930664063e-05,
      "step": 50020
    },
    {
      "action_loss_avg": 0.002269429567968473,
      "epoch": 4.770278930664063e-05,
      "step": 50020
    },
    {
      "epoch": 4.770278930664063e-05,
      "step": 50020,
      "training_step_time": 8.736414909362793
    },
    {
      "epoch": 4.770374298095703e-05,
      "model_forward_time": 1.7372190952301025,
      "step": 50021
    },
    {
      "epoch": 4.770374298095703e-05,
      "step": 50021,
      "training_step_time": 7.763992071151733
    },
    {
      "epoch": 4.7704696655273436e-05,
      "model_forward_time": 1.7341539859771729,
      "step": 50022
    },
    {
      "epoch": 4.7704696655273436e-05,
      "step": 50022,
      "training_step_time": 7.555142641067505
    },
    {
      "epoch": 4.770565032958985e-05,
      "model_forward_time": 1.7460992336273193,
      "step": 50023
    },
    {
      "epoch": 4.770565032958985e-05,
      "step": 50023,
      "training_step_time": 7.139089822769165
    },
    {
      "epoch": 4.770660400390625e-05,
      "model_forward_time": 1.7368838787078857,
      "step": 50024
    },
    {
      "epoch": 4.770660400390625e-05,
      "step": 50024,
      "training_step_time": 7.2321696281433105
    },
    {
      "epoch": 4.7707557678222655e-05,
      "model_forward_time": 1.7378296852111816,
      "step": 50025
    },
    {
      "epoch": 4.7707557678222655e-05,
      "step": 50025,
      "training_step_time": 7.781964540481567
    },
    {
      "epoch": 4.7708511352539066e-05,
      "model_forward_time": 1.7030248641967773,
      "step": 50026
    },
    {
      "epoch": 4.7708511352539066e-05,
      "step": 50026,
      "training_step_time": 7.633445978164673
    },
    {
      "epoch": 4.770946502685547e-05,
      "model_forward_time": 2.252991199493408,
      "step": 50027
    },
    {
      "epoch": 4.770946502685547e-05,
      "step": 50027,
      "training_step_time": 7.662264823913574
    },
    {
      "epoch": 4.7710418701171874e-05,
      "model_forward_time": 1.7697138786315918,
      "step": 50028
    },
    {
      "epoch": 4.7710418701171874e-05,
      "step": 50028,
      "training_step_time": 9.211427211761475
    },
    {
      "epoch": 4.7711372375488285e-05,
      "model_forward_time": 1.7279822826385498,
      "step": 50029
    },
    {
      "epoch": 4.7711372375488285e-05,
      "step": 50029,
      "training_step_time": 7.656182527542114
    },
    {
      "epoch": 4.771232604980469e-05,
      "grad_norm": 0.11017889529466629,
      "learning_rate": 5.408117853320723e-06,
      "loss": 0.0627,
      "step": 50030
    },
    {
      "epoch": 4.771232604980469e-05,
      "model_forward_time": 1.7135009765625,
      "step": 50030
    },
    {
      "dynamics_loss_avg": 0.07773884981870652,
      "epoch": 4.771232604980469e-05,
      "step": 50030
    },
    {
      "action_loss_avg": 0.002826012496370822,
      "epoch": 4.771232604980469e-05,
      "step": 50030
    },
    {
      "epoch": 4.771232604980469e-05,
      "step": 50030,
      "training_step_time": 8.345858812332153
    },
    {
      "epoch": 4.771327972412109e-05,
      "model_forward_time": 1.721092700958252,
      "step": 50031
    },
    {
      "epoch": 4.771327972412109e-05,
      "step": 50031,
      "training_step_time": 7.549494743347168
    },
    {
      "epoch": 4.77142333984375e-05,
      "model_forward_time": 1.7261130809783936,
      "step": 50032
    },
    {
      "epoch": 4.77142333984375e-05,
      "step": 50032,
      "training_step_time": 7.003634929656982
    },
    {
      "epoch": 4.771518707275391e-05,
      "model_forward_time": 1.7211759090423584,
      "step": 50033
    },
    {
      "epoch": 4.771518707275391e-05,
      "step": 50033,
      "training_step_time": 7.190161228179932
    },
    {
      "epoch": 4.771614074707031e-05,
      "model_forward_time": 1.7235157489776611,
      "step": 50034
    },
    {
      "epoch": 4.771614074707031e-05,
      "step": 50034,
      "training_step_time": 7.736634731292725
    },
    {
      "epoch": 4.7717094421386715e-05,
      "model_forward_time": 1.716951608657837,
      "step": 50035
    },
    {
      "epoch": 4.7717094421386715e-05,
      "step": 50035,
      "training_step_time": 8.04923963546753
    },
    {
      "epoch": 4.7718048095703126e-05,
      "model_forward_time": 2.2715911865234375,
      "step": 50036
    },
    {
      "epoch": 4.7718048095703126e-05,
      "step": 50036,
      "training_step_time": 10.742145776748657
    },
    {
      "epoch": 4.771900177001953e-05,
      "model_forward_time": 1.7623043060302734,
      "step": 50037
    },
    {
      "epoch": 4.771900177001953e-05,
      "step": 50037,
      "training_step_time": 10.271338939666748
    },
    {
      "epoch": 4.7719955444335934e-05,
      "model_forward_time": 1.7186875343322754,
      "step": 50038
    },
    {
      "epoch": 4.7719955444335934e-05,
      "step": 50038,
      "training_step_time": 7.603851556777954
    },
    {
      "epoch": 4.7720909118652345e-05,
      "model_forward_time": 1.7189621925354004,
      "step": 50039
    },
    {
      "epoch": 4.7720909118652345e-05,
      "step": 50039,
      "training_step_time": 7.584888458251953
    },
    {
      "epoch": 4.772186279296875e-05,
      "grad_norm": 0.16840243339538574,
      "learning_rate": 5.4064698784670845e-06,
      "loss": 0.0641,
      "step": 50040
    },
    {
      "epoch": 4.772186279296875e-05,
      "model_forward_time": 1.70900297164917,
      "step": 50040
    },
    {
      "dynamics_loss_avg": 0.06693492401391268,
      "epoch": 4.772186279296875e-05,
      "step": 50040
    },
    {
      "action_loss_avg": 0.0023098251956980677,
      "epoch": 4.772186279296875e-05,
      "step": 50040
    },
    {
      "epoch": 4.772186279296875e-05,
      "step": 50040,
      "training_step_time": 7.627592325210571
    },
    {
      "epoch": 4.772281646728515e-05,
      "model_forward_time": 1.7167253494262695,
      "step": 50041
    },
    {
      "epoch": 4.772281646728515e-05,
      "step": 50041,
      "training_step_time": 7.506549835205078
    },
    {
      "epoch": 4.7723770141601564e-05,
      "model_forward_time": 1.7259178161621094,
      "step": 50042
    },
    {
      "epoch": 4.7723770141601564e-05,
      "step": 50042,
      "training_step_time": 7.288744688034058
    },
    {
      "epoch": 4.772472381591797e-05,
      "model_forward_time": 1.7151408195495605,
      "step": 50043
    },
    {
      "epoch": 4.772472381591797e-05,
      "step": 50043,
      "training_step_time": 8.54631519317627
    },
    {
      "epoch": 4.772567749023437e-05,
      "model_forward_time": 1.7142348289489746,
      "step": 50044
    },
    {
      "epoch": 4.772567749023437e-05,
      "step": 50044,
      "training_step_time": 7.755822658538818
    },
    {
      "epoch": 4.772663116455078e-05,
      "model_forward_time": 2.257885694503784,
      "step": 50045
    },
    {
      "epoch": 4.772663116455078e-05,
      "step": 50045,
      "training_step_time": 8.345382452011108
    },
    {
      "epoch": 4.772758483886719e-05,
      "model_forward_time": 1.7264833450317383,
      "step": 50046
    },
    {
      "epoch": 4.772758483886719e-05,
      "step": 50046,
      "training_step_time": 8.281434297561646
    },
    {
      "epoch": 4.772853851318359e-05,
      "model_forward_time": 1.7217967510223389,
      "step": 50047
    },
    {
      "epoch": 4.772853851318359e-05,
      "step": 50047,
      "training_step_time": 7.727219343185425
    },
    {
      "epoch": 4.77294921875e-05,
      "model_forward_time": 1.7167110443115234,
      "step": 50048
    },
    {
      "epoch": 4.77294921875e-05,
      "step": 50048,
      "training_step_time": 7.700551748275757
    },
    {
      "epoch": 4.7730445861816406e-05,
      "model_forward_time": 1.7136495113372803,
      "step": 50049
    },
    {
      "epoch": 4.7730445861816406e-05,
      "step": 50049,
      "training_step_time": 7.702665090560913
    },
    {
      "epoch": 4.773139953613281e-05,
      "grad_norm": 0.12959478795528412,
      "learning_rate": 5.404821859162509e-06,
      "loss": 0.0636,
      "step": 50050
    },
    {
      "epoch": 4.773139953613281e-05,
      "model_forward_time": 1.7083196640014648,
      "step": 50050
    },
    {
      "dynamics_loss_avg": 0.07843004427850246,
      "epoch": 4.773139953613281e-05,
      "step": 50050
    },
    {
      "action_loss_avg": 0.0029413278098218143,
      "epoch": 4.773139953613281e-05,
      "step": 50050
    },
    {
      "epoch": 4.773139953613281e-05,
      "step": 50050,
      "training_step_time": 7.409470796585083
    },
    {
      "epoch": 4.773235321044922e-05,
      "model_forward_time": 1.7200987339019775,
      "step": 50051
    },
    {
      "epoch": 4.773235321044922e-05,
      "step": 50051,
      "training_step_time": 7.258429288864136
    },
    {
      "epoch": 4.7733306884765625e-05,
      "model_forward_time": 1.7168152332305908,
      "step": 50052
    },
    {
      "epoch": 4.7733306884765625e-05,
      "step": 50052,
      "training_step_time": 8.386711359024048
    },
    {
      "epoch": 4.773426055908203e-05,
      "model_forward_time": 1.7141897678375244,
      "step": 50053
    },
    {
      "epoch": 4.773426055908203e-05,
      "step": 50053,
      "training_step_time": 7.770695209503174
    },
    {
      "epoch": 4.773521423339844e-05,
      "model_forward_time": 2.2643847465515137,
      "step": 50054
    },
    {
      "epoch": 4.773521423339844e-05,
      "step": 50054,
      "training_step_time": 8.48517656326294
    },
    {
      "epoch": 4.7736167907714843e-05,
      "model_forward_time": 1.7217092514038086,
      "step": 50055
    },
    {
      "epoch": 4.7736167907714843e-05,
      "step": 50055,
      "training_step_time": 8.331616640090942
    },
    {
      "epoch": 4.773712158203125e-05,
      "model_forward_time": 1.717310905456543,
      "step": 50056
    },
    {
      "epoch": 4.773712158203125e-05,
      "step": 50056,
      "training_step_time": 9.55640196800232
    },
    {
      "epoch": 4.773807525634766e-05,
      "model_forward_time": 1.711623191833496,
      "step": 50057
    },
    {
      "epoch": 4.773807525634766e-05,
      "step": 50057,
      "training_step_time": 9.356725692749023
    },
    {
      "epoch": 4.773902893066406e-05,
      "model_forward_time": 1.7213654518127441,
      "step": 50058
    },
    {
      "epoch": 4.773902893066406e-05,
      "step": 50058,
      "training_step_time": 7.730629205703735
    },
    {
      "epoch": 4.7739982604980466e-05,
      "model_forward_time": 1.34897780418396,
      "step": 50059
    },
    {
      "epoch": 4.7739982604980466e-05,
      "step": 50059,
      "training_step_time": 7.215391159057617
    },
    {
      "epoch": 4.774093627929688e-05,
      "grad_norm": 0.10930043458938599,
      "learning_rate": 5.403173795587225e-06,
      "loss": 0.0624,
      "step": 50060
    },
    {
      "epoch": 4.774093627929688e-05,
      "model_forward_time": 1.716353416442871,
      "step": 50060
    },
    {
      "dynamics_loss_avg": 0.07806241307407617,
      "epoch": 4.774093627929688e-05,
      "step": 50060
    },
    {
      "action_loss_avg": 0.0035194940282963216,
      "epoch": 4.774093627929688e-05,
      "step": 50060
    },
    {
      "epoch": 4.774093627929688e-05,
      "step": 50060,
      "training_step_time": 7.287785291671753
    },
    {
      "epoch": 4.774188995361328e-05,
      "model_forward_time": 1.7289705276489258,
      "step": 50061
    },
    {
      "epoch": 4.774188995361328e-05,
      "step": 50061,
      "training_step_time": 7.89273738861084
    },
    {
      "epoch": 4.7742843627929685e-05,
      "model_forward_time": 1.7186334133148193,
      "step": 50062
    },
    {
      "epoch": 4.7742843627929685e-05,
      "step": 50062,
      "training_step_time": 7.905000925064087
    },
    {
      "epoch": 4.7743797302246096e-05,
      "model_forward_time": 2.251413345336914,
      "step": 50063
    },
    {
      "epoch": 4.7743797302246096e-05,
      "step": 50063,
      "training_step_time": 7.704545974731445
    },
    {
      "epoch": 4.77447509765625e-05,
      "model_forward_time": 1.713402271270752,
      "step": 50064
    },
    {
      "epoch": 4.77447509765625e-05,
      "step": 50064,
      "training_step_time": 8.718863010406494
    },
    {
      "epoch": 4.7745704650878904e-05,
      "model_forward_time": 1.7962157726287842,
      "step": 50065
    },
    {
      "epoch": 4.7745704650878904e-05,
      "step": 50065,
      "training_step_time": 10.879163265228271
    },
    {
      "epoch": 4.7746658325195315e-05,
      "model_forward_time": 1.7200567722320557,
      "step": 50066
    },
    {
      "epoch": 4.7746658325195315e-05,
      "step": 50066,
      "training_step_time": 8.258514165878296
    },
    {
      "epoch": 4.774761199951172e-05,
      "model_forward_time": 1.7237374782562256,
      "step": 50067
    },
    {
      "epoch": 4.774761199951172e-05,
      "step": 50067,
      "training_step_time": 7.756398916244507
    },
    {
      "epoch": 4.774856567382812e-05,
      "model_forward_time": 1.7174487113952637,
      "step": 50068
    },
    {
      "epoch": 4.774856567382812e-05,
      "step": 50068,
      "training_step_time": 7.838544607162476
    },
    {
      "epoch": 4.7749519348144534e-05,
      "model_forward_time": 1.7257325649261475,
      "step": 50069
    },
    {
      "epoch": 4.7749519348144534e-05,
      "step": 50069,
      "training_step_time": 7.473537921905518
    },
    {
      "epoch": 4.775047302246094e-05,
      "grad_norm": 0.11341043561697006,
      "learning_rate": 5.4015256879214604e-06,
      "loss": 0.0581,
      "step": 50070
    },
    {
      "epoch": 4.775047302246094e-05,
      "model_forward_time": 1.7083940505981445,
      "step": 50070
    },
    {
      "dynamics_loss_avg": 0.04972348641604185,
      "epoch": 4.775047302246094e-05,
      "step": 50070
    },
    {
      "action_loss_avg": 0.001517255016369745,
      "epoch": 4.775047302246094e-05,
      "step": 50070
    },
    {
      "epoch": 4.775047302246094e-05,
      "step": 50070,
      "training_step_time": 7.864412784576416
    },
    {
      "epoch": 4.775142669677734e-05,
      "model_forward_time": 1.7190115451812744,
      "step": 50071
    },
    {
      "epoch": 4.775142669677734e-05,
      "step": 50071,
      "training_step_time": 7.813831567764282
    },
    {
      "epoch": 4.775238037109375e-05,
      "model_forward_time": 1.7233695983886719,
      "step": 50072
    },
    {
      "epoch": 4.775238037109375e-05,
      "step": 50072,
      "training_step_time": 8.481983423233032
    },
    {
      "epoch": 4.775333404541016e-05,
      "model_forward_time": 2.270191192626953,
      "step": 50073
    },
    {
      "epoch": 4.775333404541016e-05,
      "step": 50073,
      "training_step_time": 7.854757308959961
    },
    {
      "epoch": 4.775428771972656e-05,
      "model_forward_time": 1.7818005084991455,
      "step": 50074
    },
    {
      "epoch": 4.775428771972656e-05,
      "step": 50074,
      "training_step_time": 8.717386722564697
    },
    {
      "epoch": 4.775524139404297e-05,
      "model_forward_time": 1.342200517654419,
      "step": 50075
    },
    {
      "epoch": 4.775524139404297e-05,
      "step": 50075,
      "training_step_time": 9.71585988998413
    },
    {
      "epoch": 4.7756195068359375e-05,
      "model_forward_time": 1.7115132808685303,
      "step": 50076
    },
    {
      "epoch": 4.7756195068359375e-05,
      "step": 50076,
      "training_step_time": 7.66253137588501
    },
    {
      "epoch": 4.775714874267578e-05,
      "model_forward_time": 1.718698263168335,
      "step": 50077
    },
    {
      "epoch": 4.775714874267578e-05,
      "step": 50077,
      "training_step_time": 8.179372787475586
    },
    {
      "epoch": 4.775810241699219e-05,
      "model_forward_time": 1.7198708057403564,
      "step": 50078
    },
    {
      "epoch": 4.775810241699219e-05,
      "step": 50078,
      "training_step_time": 7.448332071304321
    },
    {
      "epoch": 4.7759056091308594e-05,
      "model_forward_time": 1.7195265293121338,
      "step": 50079
    },
    {
      "epoch": 4.7759056091308594e-05,
      "step": 50079,
      "training_step_time": 7.851887941360474
    },
    {
      "epoch": 4.7760009765625e-05,
      "grad_norm": 0.12822844088077545,
      "learning_rate": 5.39987753634545e-06,
      "loss": 0.0585,
      "step": 50080
    },
    {
      "epoch": 4.7760009765625e-05,
      "model_forward_time": 1.7108182907104492,
      "step": 50080
    },
    {
      "dynamics_loss_avg": 0.061815212666988376,
      "epoch": 4.7760009765625e-05,
      "step": 50080
    },
    {
      "action_loss_avg": 0.0026196917780907823,
      "epoch": 4.7760009765625e-05,
      "step": 50080
    },
    {
      "epoch": 4.7760009765625e-05,
      "step": 50080,
      "training_step_time": 7.640343427658081
    },
    {
      "epoch": 4.776096343994141e-05,
      "model_forward_time": 1.7103030681610107,
      "step": 50081
    },
    {
      "epoch": 4.776096343994141e-05,
      "step": 50081,
      "training_step_time": 8.622719049453735
    },
    {
      "epoch": 4.776191711425781e-05,
      "model_forward_time": 2.2666642665863037,
      "step": 50082
    },
    {
      "epoch": 4.776191711425781e-05,
      "step": 50082,
      "training_step_time": 7.726916790008545
    },
    {
      "epoch": 4.776287078857422e-05,
      "model_forward_time": 1.7630817890167236,
      "step": 50083
    },
    {
      "epoch": 4.776287078857422e-05,
      "step": 50083,
      "training_step_time": 9.342628002166748
    },
    {
      "epoch": 4.776382446289063e-05,
      "model_forward_time": 1.7230579853057861,
      "step": 50084
    },
    {
      "epoch": 4.776382446289063e-05,
      "step": 50084,
      "training_step_time": 7.691362142562866
    },
    {
      "epoch": 4.776477813720703e-05,
      "model_forward_time": 1.7266464233398438,
      "step": 50085
    },
    {
      "epoch": 4.776477813720703e-05,
      "step": 50085,
      "training_step_time": 7.663865566253662
    },
    {
      "epoch": 4.7765731811523436e-05,
      "model_forward_time": 1.715240716934204,
      "step": 50086
    },
    {
      "epoch": 4.7765731811523436e-05,
      "step": 50086,
      "training_step_time": 7.653100252151489
    },
    {
      "epoch": 4.776668548583985e-05,
      "model_forward_time": 1.715714454650879,
      "step": 50087
    },
    {
      "epoch": 4.776668548583985e-05,
      "step": 50087,
      "training_step_time": 8.23153829574585
    },
    {
      "epoch": 4.776763916015625e-05,
      "model_forward_time": 1.7181646823883057,
      "step": 50088
    },
    {
      "epoch": 4.776763916015625e-05,
      "step": 50088,
      "training_step_time": 7.211361646652222
    },
    {
      "epoch": 4.7768592834472655e-05,
      "model_forward_time": 1.7258741855621338,
      "step": 50089
    },
    {
      "epoch": 4.7768592834472655e-05,
      "step": 50089,
      "training_step_time": 7.797696828842163
    },
    {
      "epoch": 4.7769546508789066e-05,
      "grad_norm": 0.09212100505828857,
      "learning_rate": 5.3982293410394335e-06,
      "loss": 0.0605,
      "step": 50090
    },
    {
      "epoch": 4.7769546508789066e-05,
      "model_forward_time": 1.721987009048462,
      "step": 50090
    },
    {
      "dynamics_loss_avg": 0.0818481869995594,
      "epoch": 4.7769546508789066e-05,
      "step": 50090
    },
    {
      "action_loss_avg": 0.0023090726521331818,
      "epoch": 4.7769546508789066e-05,
      "step": 50090
    },
    {
      "epoch": 4.7769546508789066e-05,
      "step": 50090,
      "training_step_time": 7.666206121444702
    },
    {
      "epoch": 4.777050018310547e-05,
      "model_forward_time": 2.27339243888855,
      "step": 50091
    },
    {
      "epoch": 4.777050018310547e-05,
      "step": 50091,
      "training_step_time": 9.006471872329712
    },
    {
      "epoch": 4.7771453857421874e-05,
      "model_forward_time": 1.7363250255584717,
      "step": 50092
    },
    {
      "epoch": 4.7771453857421874e-05,
      "step": 50092,
      "training_step_time": 8.43791651725769
    },
    {
      "epoch": 4.7772407531738285e-05,
      "model_forward_time": 1.732496976852417,
      "step": 50093
    },
    {
      "epoch": 4.7772407531738285e-05,
      "step": 50093,
      "training_step_time": 7.579440593719482
    },
    {
      "epoch": 4.777336120605469e-05,
      "model_forward_time": 1.7206203937530518,
      "step": 50094
    },
    {
      "epoch": 4.777336120605469e-05,
      "step": 50094,
      "training_step_time": 7.5675153732299805
    },
    {
      "epoch": 4.777431488037109e-05,
      "model_forward_time": 1.7265143394470215,
      "step": 50095
    },
    {
      "epoch": 4.777431488037109e-05,
      "step": 50095,
      "training_step_time": 7.566911697387695
    },
    {
      "epoch": 4.77752685546875e-05,
      "model_forward_time": 1.734776496887207,
      "step": 50096
    },
    {
      "epoch": 4.77752685546875e-05,
      "step": 50096,
      "training_step_time": 7.493517637252808
    },
    {
      "epoch": 4.777622222900391e-05,
      "model_forward_time": 1.7288720607757568,
      "step": 50097
    },
    {
      "epoch": 4.777622222900391e-05,
      "step": 50097,
      "training_step_time": 7.904428958892822
    },
    {
      "epoch": 4.777717590332031e-05,
      "model_forward_time": 1.7275254726409912,
      "step": 50098
    },
    {
      "epoch": 4.777717590332031e-05,
      "step": 50098,
      "training_step_time": 7.600419998168945
    },
    {
      "epoch": 4.7778129577636716e-05,
      "model_forward_time": 1.727165699005127,
      "step": 50099
    },
    {
      "epoch": 4.7778129577636716e-05,
      "step": 50099,
      "training_step_time": 7.552013158798218
    },
    {
      "epoch": 4.7779083251953126e-05,
      "grad_norm": 0.09228457510471344,
      "learning_rate": 5.396581102183656e-06,
      "loss": 0.0596,
      "step": 50100
    },
    {
      "epoch": 4.7779083251953126e-05,
      "model_forward_time": 2.2719480991363525,
      "step": 50100
    },
    {
      "dynamics_loss_avg": 0.060490965843200684,
      "epoch": 4.7779083251953126e-05,
      "step": 50100
    },
    {
      "action_loss_avg": 0.0016578615963226185,
      "epoch": 4.7779083251953126e-05,
      "step": 50100
    },
    {
      "epoch": 4.7779083251953126e-05,
      "step": 50100,
      "training_step_time": 7.58462119102478
    },
    {
      "epoch": 4.778003692626953e-05,
      "model_forward_time": 1.7176566123962402,
      "step": 50101
    },
    {
      "epoch": 4.778003692626953e-05,
      "step": 50101,
      "training_step_time": 9.734866380691528
    },
    {
      "epoch": 4.7780990600585934e-05,
      "model_forward_time": 1.808763027191162,
      "step": 50102
    },
    {
      "epoch": 4.7780990600585934e-05,
      "step": 50102,
      "training_step_time": 7.630110263824463
    },
    {
      "epoch": 4.7781944274902345e-05,
      "model_forward_time": 1.7217977046966553,
      "step": 50103
    },
    {
      "epoch": 4.7781944274902345e-05,
      "step": 50103,
      "training_step_time": 7.6977949142456055
    },
    {
      "epoch": 4.778289794921875e-05,
      "model_forward_time": 1.7252552509307861,
      "step": 50104
    },
    {
      "epoch": 4.778289794921875e-05,
      "step": 50104,
      "training_step_time": 7.546618223190308
    },
    {
      "epoch": 4.778385162353515e-05,
      "model_forward_time": 1.7265379428863525,
      "step": 50105
    },
    {
      "epoch": 4.778385162353515e-05,
      "step": 50105,
      "training_step_time": 8.066540718078613
    },
    {
      "epoch": 4.7784805297851564e-05,
      "model_forward_time": 1.722707748413086,
      "step": 50106
    },
    {
      "epoch": 4.7784805297851564e-05,
      "step": 50106,
      "training_step_time": 7.486287355422974
    },
    {
      "epoch": 4.778575897216797e-05,
      "model_forward_time": 1.7251169681549072,
      "step": 50107
    },
    {
      "epoch": 4.778575897216797e-05,
      "step": 50107,
      "training_step_time": 7.858944654464722
    },
    {
      "epoch": 4.778671264648437e-05,
      "model_forward_time": 1.7287931442260742,
      "step": 50108
    },
    {
      "epoch": 4.778671264648437e-05,
      "step": 50108,
      "training_step_time": 7.5976784229278564
    },
    {
      "epoch": 4.778766632080078e-05,
      "model_forward_time": 1.723921298980713,
      "step": 50109
    },
    {
      "epoch": 4.778766632080078e-05,
      "step": 50109,
      "training_step_time": 8.114906787872314
    },
    {
      "epoch": 4.778861999511719e-05,
      "grad_norm": 0.12492670863866806,
      "learning_rate": 5.394932819958364e-06,
      "loss": 0.0571,
      "step": 50110
    },
    {
      "epoch": 4.778861999511719e-05,
      "model_forward_time": 1.7136096954345703,
      "step": 50110
    },
    {
      "dynamics_loss_avg": 0.062429520301520824,
      "epoch": 4.778861999511719e-05,
      "step": 50110
    },
    {
      "action_loss_avg": 0.0025267131393775346,
      "epoch": 4.778861999511719e-05,
      "step": 50110
    },
    {
      "epoch": 4.778861999511719e-05,
      "step": 50110,
      "training_step_time": 8.239911794662476
    },
    {
      "epoch": 4.778957366943359e-05,
      "model_forward_time": 1.8148527145385742,
      "step": 50111
    },
    {
      "epoch": 4.778957366943359e-05,
      "step": 50111,
      "training_step_time": 9.20032525062561
    },
    {
      "epoch": 4.779052734375e-05,
      "model_forward_time": 1.7005670070648193,
      "step": 50112
    },
    {
      "epoch": 4.779052734375e-05,
      "step": 50112,
      "training_step_time": 8.44574785232544
    },
    {
      "epoch": 4.7791481018066406e-05,
      "model_forward_time": 1.7205462455749512,
      "step": 50113
    },
    {
      "epoch": 4.7791481018066406e-05,
      "step": 50113,
      "training_step_time": 7.652292490005493
    },
    {
      "epoch": 4.779243469238281e-05,
      "model_forward_time": 1.718933343887329,
      "step": 50114
    },
    {
      "epoch": 4.779243469238281e-05,
      "step": 50114,
      "training_step_time": 7.501318454742432
    },
    {
      "epoch": 4.779338836669922e-05,
      "model_forward_time": 1.7255210876464844,
      "step": 50115
    },
    {
      "epoch": 4.779338836669922e-05,
      "step": 50115,
      "training_step_time": 8.19795846939087
    },
    {
      "epoch": 4.7794342041015625e-05,
      "model_forward_time": 1.7213053703308105,
      "step": 50116
    },
    {
      "epoch": 4.7794342041015625e-05,
      "step": 50116,
      "training_step_time": 8.296876192092896
    },
    {
      "epoch": 4.779529571533203e-05,
      "model_forward_time": 1.7155299186706543,
      "step": 50117
    },
    {
      "epoch": 4.779529571533203e-05,
      "step": 50117,
      "training_step_time": 7.437209844589233
    },
    {
      "epoch": 4.779624938964844e-05,
      "model_forward_time": 1.7199363708496094,
      "step": 50118
    },
    {
      "epoch": 4.779624938964844e-05,
      "step": 50118,
      "training_step_time": 8.21491289138794
    },
    {
      "epoch": 4.7797203063964844e-05,
      "model_forward_time": 2.2448389530181885,
      "step": 50119
    },
    {
      "epoch": 4.7797203063964844e-05,
      "step": 50119,
      "training_step_time": 8.684471368789673
    },
    {
      "epoch": 4.779815673828125e-05,
      "grad_norm": 0.1539372056722641,
      "learning_rate": 5.393284494543813e-06,
      "loss": 0.0594,
      "step": 50120
    },
    {
      "epoch": 4.779815673828125e-05,
      "model_forward_time": 1.792712926864624,
      "step": 50120
    },
    {
      "dynamics_loss_avg": 0.05880921557545662,
      "epoch": 4.779815673828125e-05,
      "step": 50120
    },
    {
      "action_loss_avg": 0.0016781112470198422,
      "epoch": 4.779815673828125e-05,
      "step": 50120
    },
    {
      "epoch": 4.779815673828125e-05,
      "step": 50120,
      "training_step_time": 7.608238935470581
    },
    {
      "epoch": 4.779911041259766e-05,
      "model_forward_time": 1.7221777439117432,
      "step": 50121
    },
    {
      "epoch": 4.779911041259766e-05,
      "step": 50121,
      "training_step_time": 8.136495113372803
    },
    {
      "epoch": 4.780006408691406e-05,
      "model_forward_time": 1.7169482707977295,
      "step": 50122
    },
    {
      "epoch": 4.780006408691406e-05,
      "step": 50122,
      "training_step_time": 7.603513956069946
    },
    {
      "epoch": 4.7801017761230466e-05,
      "model_forward_time": 1.7231614589691162,
      "step": 50123
    },
    {
      "epoch": 4.7801017761230466e-05,
      "step": 50123,
      "training_step_time": 7.562852144241333
    },
    {
      "epoch": 4.780197143554688e-05,
      "model_forward_time": 1.7249505519866943,
      "step": 50124
    },
    {
      "epoch": 4.780197143554688e-05,
      "step": 50124,
      "training_step_time": 7.591367483139038
    },
    {
      "epoch": 4.780292510986328e-05,
      "model_forward_time": 1.7211601734161377,
      "step": 50125
    },
    {
      "epoch": 4.780292510986328e-05,
      "step": 50125,
      "training_step_time": 7.269491672515869
    },
    {
      "epoch": 4.7803878784179685e-05,
      "model_forward_time": 1.705904483795166,
      "step": 50126
    },
    {
      "epoch": 4.7803878784179685e-05,
      "step": 50126,
      "training_step_time": 7.8264148235321045
    },
    {
      "epoch": 4.7804832458496096e-05,
      "model_forward_time": 1.714735746383667,
      "step": 50127
    },
    {
      "epoch": 4.7804832458496096e-05,
      "step": 50127,
      "training_step_time": 8.347075700759888
    },
    {
      "epoch": 4.78057861328125e-05,
      "model_forward_time": 2.258551597595215,
      "step": 50128
    },
    {
      "epoch": 4.78057861328125e-05,
      "step": 50128,
      "training_step_time": 9.44968581199646
    },
    {
      "epoch": 4.7806739807128904e-05,
      "model_forward_time": 1.7577180862426758,
      "step": 50129
    },
    {
      "epoch": 4.7806739807128904e-05,
      "step": 50129,
      "training_step_time": 9.277791738510132
    },
    {
      "epoch": 4.7807693481445315e-05,
      "grad_norm": 0.12542638182640076,
      "learning_rate": 5.391636126120262e-06,
      "loss": 0.0525,
      "step": 50130
    },
    {
      "epoch": 4.7807693481445315e-05,
      "model_forward_time": 1.7187416553497314,
      "step": 50130
    },
    {
      "dynamics_loss_avg": 0.06023929845541716,
      "epoch": 4.7807693481445315e-05,
      "step": 50130
    },
    {
      "action_loss_avg": 0.0022950935002882034,
      "epoch": 4.7807693481445315e-05,
      "step": 50130
    },
    {
      "epoch": 4.7807693481445315e-05,
      "step": 50130,
      "training_step_time": 7.655232906341553
    },
    {
      "epoch": 4.780864715576172e-05,
      "model_forward_time": 1.727203369140625,
      "step": 50131
    },
    {
      "epoch": 4.780864715576172e-05,
      "step": 50131,
      "training_step_time": 8.647951602935791
    },
    {
      "epoch": 4.780960083007812e-05,
      "model_forward_time": 1.7212669849395752,
      "step": 50132
    },
    {
      "epoch": 4.780960083007812e-05,
      "step": 50132,
      "training_step_time": 7.500454664230347
    },
    {
      "epoch": 4.7810554504394534e-05,
      "model_forward_time": 1.7263734340667725,
      "step": 50133
    },
    {
      "epoch": 4.7810554504394534e-05,
      "step": 50133,
      "training_step_time": 7.530934810638428
    },
    {
      "epoch": 4.781150817871094e-05,
      "model_forward_time": 1.72658109664917,
      "step": 50134
    },
    {
      "epoch": 4.781150817871094e-05,
      "step": 50134,
      "training_step_time": 7.316654682159424
    },
    {
      "epoch": 4.781246185302734e-05,
      "model_forward_time": 1.717895269393921,
      "step": 50135
    },
    {
      "epoch": 4.781246185302734e-05,
      "step": 50135,
      "training_step_time": 7.761842489242554
    },
    {
      "epoch": 4.781341552734375e-05,
      "model_forward_time": 1.7224159240722656,
      "step": 50136
    },
    {
      "epoch": 4.781341552734375e-05,
      "step": 50136,
      "training_step_time": 8.827121257781982
    },
    {
      "epoch": 4.781436920166016e-05,
      "model_forward_time": 2.278374195098877,
      "step": 50137
    },
    {
      "epoch": 4.781436920166016e-05,
      "step": 50137,
      "training_step_time": 9.455009937286377
    },
    {
      "epoch": 4.781532287597656e-05,
      "model_forward_time": 1.788224458694458,
      "step": 50138
    },
    {
      "epoch": 4.781532287597656e-05,
      "step": 50138,
      "training_step_time": 7.660678863525391
    },
    {
      "epoch": 4.781627655029297e-05,
      "model_forward_time": 1.7247090339660645,
      "step": 50139
    },
    {
      "epoch": 4.781627655029297e-05,
      "step": 50139,
      "training_step_time": 8.950249433517456
    },
    {
      "epoch": 4.7817230224609376e-05,
      "grad_norm": 0.12525784969329834,
      "learning_rate": 5.389987714867971e-06,
      "loss": 0.0558,
      "step": 50140
    },
    {
      "epoch": 4.7817230224609376e-05,
      "model_forward_time": 1.7188622951507568,
      "step": 50140
    },
    {
      "dynamics_loss_avg": 0.046095486357808114,
      "epoch": 4.7817230224609376e-05,
      "step": 50140
    },
    {
      "action_loss_avg": 0.0024755216552875938,
      "epoch": 4.7817230224609376e-05,
      "step": 50140
    },
    {
      "epoch": 4.7817230224609376e-05,
      "step": 50140,
      "training_step_time": 7.644649505615234
    },
    {
      "epoch": 4.781818389892578e-05,
      "model_forward_time": 1.732130527496338,
      "step": 50141
    },
    {
      "epoch": 4.781818389892578e-05,
      "step": 50141,
      "training_step_time": 7.677637338638306
    },
    {
      "epoch": 4.781913757324219e-05,
      "model_forward_time": 1.7246053218841553,
      "step": 50142
    },
    {
      "epoch": 4.781913757324219e-05,
      "step": 50142,
      "training_step_time": 7.142292737960815
    },
    {
      "epoch": 4.7820091247558594e-05,
      "model_forward_time": 1.7237935066223145,
      "step": 50143
    },
    {
      "epoch": 4.7820091247558594e-05,
      "step": 50143,
      "training_step_time": 7.398422002792358
    },
    {
      "epoch": 4.7821044921875e-05,
      "model_forward_time": 1.7270135879516602,
      "step": 50144
    },
    {
      "epoch": 4.7821044921875e-05,
      "step": 50144,
      "training_step_time": 8.537550210952759
    },
    {
      "epoch": 4.782199859619141e-05,
      "model_forward_time": 1.7332849502563477,
      "step": 50145
    },
    {
      "epoch": 4.782199859619141e-05,
      "step": 50145,
      "training_step_time": 7.770992994308472
    },
    {
      "epoch": 4.782295227050781e-05,
      "model_forward_time": 2.2948505878448486,
      "step": 50146
    },
    {
      "epoch": 4.782295227050781e-05,
      "step": 50146,
      "training_step_time": 7.73920202255249
    },
    {
      "epoch": 4.782390594482422e-05,
      "model_forward_time": 1.805788278579712,
      "step": 50147
    },
    {
      "epoch": 4.782390594482422e-05,
      "step": 50147,
      "training_step_time": 7.604535341262817
    },
    {
      "epoch": 4.782485961914063e-05,
      "model_forward_time": 1.7342185974121094,
      "step": 50148
    },
    {
      "epoch": 4.782485961914063e-05,
      "step": 50148,
      "training_step_time": 8.999077558517456
    },
    {
      "epoch": 4.782581329345703e-05,
      "model_forward_time": 1.722273349761963,
      "step": 50149
    },
    {
      "epoch": 4.782581329345703e-05,
      "step": 50149,
      "training_step_time": 7.579380035400391
    },
    {
      "epoch": 4.7826766967773436e-05,
      "grad_norm": 0.08306311070919037,
      "learning_rate": 5.388339260967211e-06,
      "loss": 0.0524,
      "step": 50150
    },
    {
      "epoch": 4.7826766967773436e-05,
      "model_forward_time": 1.7162840366363525,
      "step": 50150
    },
    {
      "dynamics_loss_avg": 0.05351780131459236,
      "epoch": 4.7826766967773436e-05,
      "step": 50150
    },
    {
      "action_loss_avg": 0.002972503146156669,
      "epoch": 4.7826766967773436e-05,
      "step": 50150
    },
    {
      "epoch": 4.7826766967773436e-05,
      "step": 50150,
      "training_step_time": 7.543403148651123
    },
    {
      "epoch": 4.782772064208985e-05,
      "model_forward_time": 1.7222778797149658,
      "step": 50151
    },
    {
      "epoch": 4.782772064208985e-05,
      "step": 50151,
      "training_step_time": 7.533299922943115
    },
    {
      "epoch": 4.782867431640625e-05,
      "model_forward_time": 1.7204289436340332,
      "step": 50152
    },
    {
      "epoch": 4.782867431640625e-05,
      "step": 50152,
      "training_step_time": 7.540133714675903
    },
    {
      "epoch": 4.7829627990722655e-05,
      "model_forward_time": 1.7161986827850342,
      "step": 50153
    },
    {
      "epoch": 4.7829627990722655e-05,
      "step": 50153,
      "training_step_time": 7.867759704589844
    },
    {
      "epoch": 4.7830581665039066e-05,
      "model_forward_time": 1.72929048538208,
      "step": 50154
    },
    {
      "epoch": 4.7830581665039066e-05,
      "step": 50154,
      "training_step_time": 7.6874635219573975
    },
    {
      "epoch": 4.783153533935547e-05,
      "model_forward_time": 2.28621506690979,
      "step": 50155
    },
    {
      "epoch": 4.783153533935547e-05,
      "step": 50155,
      "training_step_time": 7.634565353393555
    },
    {
      "epoch": 4.7832489013671874e-05,
      "model_forward_time": 1.7872908115386963,
      "step": 50156
    },
    {
      "epoch": 4.7832489013671874e-05,
      "step": 50156,
      "training_step_time": 9.644176244735718
    },
    {
      "epoch": 4.783344268798828e-05,
      "model_forward_time": 1.7146494388580322,
      "step": 50157
    },
    {
      "epoch": 4.783344268798828e-05,
      "step": 50157,
      "training_step_time": 7.663598299026489
    },
    {
      "epoch": 4.783439636230469e-05,
      "model_forward_time": 1.712214708328247,
      "step": 50158
    },
    {
      "epoch": 4.783439636230469e-05,
      "step": 50158,
      "training_step_time": 7.613563776016235
    },
    {
      "epoch": 4.783535003662109e-05,
      "model_forward_time": 1.7241544723510742,
      "step": 50159
    },
    {
      "epoch": 4.783535003662109e-05,
      "step": 50159,
      "training_step_time": 7.608658790588379
    },
    {
      "epoch": 4.78363037109375e-05,
      "grad_norm": 0.14507615566253662,
      "learning_rate": 5.386690764598255e-06,
      "loss": 0.0563,
      "step": 50160
    },
    {
      "epoch": 4.78363037109375e-05,
      "model_forward_time": 1.7144582271575928,
      "step": 50160
    },
    {
      "dynamics_loss_avg": 0.06642504036426544,
      "epoch": 4.78363037109375e-05,
      "step": 50160
    },
    {
      "action_loss_avg": 0.003323051624465734,
      "epoch": 4.78363037109375e-05,
      "step": 50160
    },
    {
      "epoch": 4.78363037109375e-05,
      "step": 50160,
      "training_step_time": 7.4814612865448
    },
    {
      "epoch": 4.783725738525391e-05,
      "model_forward_time": 1.7202584743499756,
      "step": 50161
    },
    {
      "epoch": 4.783725738525391e-05,
      "step": 50161,
      "training_step_time": 7.4553070068359375
    },
    {
      "epoch": 4.783821105957031e-05,
      "model_forward_time": 1.7191767692565918,
      "step": 50162
    },
    {
      "epoch": 4.783821105957031e-05,
      "step": 50162,
      "training_step_time": 7.815820693969727
    },
    {
      "epoch": 4.7839164733886716e-05,
      "model_forward_time": 1.7169549465179443,
      "step": 50163
    },
    {
      "epoch": 4.7839164733886716e-05,
      "step": 50163,
      "training_step_time": 7.415497779846191
    },
    {
      "epoch": 4.7840118408203126e-05,
      "model_forward_time": 1.7222189903259277,
      "step": 50164
    },
    {
      "epoch": 4.7840118408203126e-05,
      "step": 50164,
      "training_step_time": 8.176991701126099
    },
    {
      "epoch": 4.784107208251953e-05,
      "model_forward_time": 1.7693493366241455,
      "step": 50165
    },
    {
      "epoch": 4.784107208251953e-05,
      "step": 50165,
      "training_step_time": 8.997650146484375
    },
    {
      "epoch": 4.7842025756835934e-05,
      "model_forward_time": 1.7196228504180908,
      "step": 50166
    },
    {
      "epoch": 4.7842025756835934e-05,
      "step": 50166,
      "training_step_time": 8.115024089813232
    },
    {
      "epoch": 4.7842979431152345e-05,
      "model_forward_time": 1.720524787902832,
      "step": 50167
    },
    {
      "epoch": 4.7842979431152345e-05,
      "step": 50167,
      "training_step_time": 7.605902910232544
    },
    {
      "epoch": 4.784393310546875e-05,
      "model_forward_time": 1.7262423038482666,
      "step": 50168
    },
    {
      "epoch": 4.784393310546875e-05,
      "step": 50168,
      "training_step_time": 7.590475082397461
    },
    {
      "epoch": 4.784488677978515e-05,
      "model_forward_time": 1.7192020416259766,
      "step": 50169
    },
    {
      "epoch": 4.784488677978515e-05,
      "step": 50169,
      "training_step_time": 7.561229944229126
    },
    {
      "epoch": 4.7845840454101564e-05,
      "grad_norm": 0.10283783823251724,
      "learning_rate": 5.385042225941375e-06,
      "loss": 0.0565,
      "step": 50170
    },
    {
      "epoch": 4.7845840454101564e-05,
      "model_forward_time": 1.718078374862671,
      "step": 50170
    },
    {
      "dynamics_loss_avg": 0.07300984710454941,
      "epoch": 4.7845840454101564e-05,
      "step": 50170
    },
    {
      "action_loss_avg": 0.0022583659389056264,
      "epoch": 4.7845840454101564e-05,
      "step": 50170
    },
    {
      "epoch": 4.7845840454101564e-05,
      "step": 50170,
      "training_step_time": 7.433290004730225
    },
    {
      "epoch": 4.784679412841797e-05,
      "model_forward_time": 1.723973035812378,
      "step": 50171
    },
    {
      "epoch": 4.784679412841797e-05,
      "step": 50171,
      "training_step_time": 7.7577784061431885
    },
    {
      "epoch": 4.784774780273437e-05,
      "model_forward_time": 1.7204394340515137,
      "step": 50172
    },
    {
      "epoch": 4.784774780273437e-05,
      "step": 50172,
      "training_step_time": 7.4047064781188965
    },
    {
      "epoch": 4.784870147705078e-05,
      "model_forward_time": 1.7224326133728027,
      "step": 50173
    },
    {
      "epoch": 4.784870147705078e-05,
      "step": 50173,
      "training_step_time": 7.798309087753296
    },
    {
      "epoch": 4.784965515136719e-05,
      "model_forward_time": 2.3304920196533203,
      "step": 50174
    },
    {
      "epoch": 4.784965515136719e-05,
      "step": 50174,
      "training_step_time": 8.38581395149231
    },
    {
      "epoch": 4.785060882568359e-05,
      "model_forward_time": 1.7186939716339111,
      "step": 50175
    },
    {
      "epoch": 4.785060882568359e-05,
      "step": 50175,
      "training_step_time": 9.199169397354126
    },
    {
      "epoch": 4.78515625e-05,
      "model_forward_time": 1.7243735790252686,
      "step": 50176
    },
    {
      "epoch": 4.78515625e-05,
      "step": 50176,
      "training_step_time": 8.481736421585083
    },
    {
      "epoch": 4.7852516174316406e-05,
      "model_forward_time": 1.7187907695770264,
      "step": 50177
    },
    {
      "epoch": 4.7852516174316406e-05,
      "step": 50177,
      "training_step_time": 8.40266227722168
    },
    {
      "epoch": 4.785346984863281e-05,
      "model_forward_time": 1.724613904953003,
      "step": 50178
    },
    {
      "epoch": 4.785346984863281e-05,
      "step": 50178,
      "training_step_time": 7.5536534786224365
    },
    {
      "epoch": 4.785442352294922e-05,
      "model_forward_time": 1.715895652770996,
      "step": 50179
    },
    {
      "epoch": 4.785442352294922e-05,
      "step": 50179,
      "training_step_time": 7.159273386001587
    },
    {
      "epoch": 4.7855377197265625e-05,
      "grad_norm": 0.11548355966806412,
      "learning_rate": 5.3833936451768585e-06,
      "loss": 0.0574,
      "step": 50180
    },
    {
      "epoch": 4.7855377197265625e-05,
      "model_forward_time": 1.7143993377685547,
      "step": 50180
    },
    {
      "dynamics_loss_avg": 0.08219782225787639,
      "epoch": 4.7855377197265625e-05,
      "step": 50180
    },
    {
      "action_loss_avg": 0.003921640396583825,
      "epoch": 4.7855377197265625e-05,
      "step": 50180
    },
    {
      "epoch": 4.7855377197265625e-05,
      "step": 50180,
      "training_step_time": 7.520766496658325
    },
    {
      "epoch": 4.785633087158203e-05,
      "model_forward_time": 1.7190577983856201,
      "step": 50181
    },
    {
      "epoch": 4.785633087158203e-05,
      "step": 50181,
      "training_step_time": 7.873850107192993
    },
    {
      "epoch": 4.785728454589844e-05,
      "model_forward_time": 1.7283110618591309,
      "step": 50182
    },
    {
      "epoch": 4.785728454589844e-05,
      "step": 50182,
      "training_step_time": 8.735657930374146
    },
    {
      "epoch": 4.7858238220214844e-05,
      "model_forward_time": 2.2905466556549072,
      "step": 50183
    },
    {
      "epoch": 4.7858238220214844e-05,
      "step": 50183,
      "training_step_time": 8.428740978240967
    },
    {
      "epoch": 4.785919189453125e-05,
      "model_forward_time": 1.7251830101013184,
      "step": 50184
    },
    {
      "epoch": 4.785919189453125e-05,
      "step": 50184,
      "training_step_time": 9.014262437820435
    },
    {
      "epoch": 4.786014556884766e-05,
      "model_forward_time": 1.7142176628112793,
      "step": 50185
    },
    {
      "epoch": 4.786014556884766e-05,
      "step": 50185,
      "training_step_time": 7.696900844573975
    },
    {
      "epoch": 4.786109924316406e-05,
      "model_forward_time": 1.7148549556732178,
      "step": 50186
    },
    {
      "epoch": 4.786109924316406e-05,
      "step": 50186,
      "training_step_time": 7.6712024211883545
    },
    {
      "epoch": 4.7862052917480466e-05,
      "model_forward_time": 1.7178668975830078,
      "step": 50187
    },
    {
      "epoch": 4.7862052917480466e-05,
      "step": 50187,
      "training_step_time": 8.365846633911133
    },
    {
      "epoch": 4.786300659179688e-05,
      "model_forward_time": 1.7234947681427002,
      "step": 50188
    },
    {
      "epoch": 4.786300659179688e-05,
      "step": 50188,
      "training_step_time": 7.224243879318237
    },
    {
      "epoch": 4.786396026611328e-05,
      "model_forward_time": 1.7148730754852295,
      "step": 50189
    },
    {
      "epoch": 4.786396026611328e-05,
      "step": 50189,
      "training_step_time": 7.64514946937561
    },
    {
      "epoch": 4.7864913940429685e-05,
      "grad_norm": 0.11361552029848099,
      "learning_rate": 5.381745022484989e-06,
      "loss": 0.0561,
      "step": 50190
    },
    {
      "epoch": 4.7864913940429685e-05,
      "model_forward_time": 1.695791482925415,
      "step": 50190
    },
    {
      "dynamics_loss_avg": 0.058185098506510255,
      "epoch": 4.7864913940429685e-05,
      "step": 50190
    },
    {
      "action_loss_avg": 0.0024470070260576905,
      "epoch": 4.7864913940429685e-05,
      "step": 50190
    },
    {
      "epoch": 4.7864913940429685e-05,
      "step": 50190,
      "training_step_time": 7.778592586517334
    },
    {
      "epoch": 4.7865867614746096e-05,
      "model_forward_time": 1.7209224700927734,
      "step": 50191
    },
    {
      "epoch": 4.7865867614746096e-05,
      "step": 50191,
      "training_step_time": 7.78689169883728
    },
    {
      "epoch": 4.78668212890625e-05,
      "model_forward_time": 2.263266086578369,
      "step": 50192
    },
    {
      "epoch": 4.78668212890625e-05,
      "step": 50192,
      "training_step_time": 8.473318576812744
    },
    {
      "epoch": 4.7867774963378904e-05,
      "model_forward_time": 1.7218260765075684,
      "step": 50193
    },
    {
      "epoch": 4.7867774963378904e-05,
      "step": 50193,
      "training_step_time": 8.496948003768921
    },
    {
      "epoch": 4.7868728637695315e-05,
      "model_forward_time": 1.719752311706543,
      "step": 50194
    },
    {
      "epoch": 4.7868728637695315e-05,
      "step": 50194,
      "training_step_time": 8.246958494186401
    },
    {
      "epoch": 4.786968231201172e-05,
      "model_forward_time": 1.7207841873168945,
      "step": 50195
    },
    {
      "epoch": 4.786968231201172e-05,
      "step": 50195,
      "training_step_time": 8.562194347381592
    },
    {
      "epoch": 4.787063598632812e-05,
      "model_forward_time": 1.7173199653625488,
      "step": 50196
    },
    {
      "epoch": 4.787063598632812e-05,
      "step": 50196,
      "training_step_time": 8.662999153137207
    },
    {
      "epoch": 4.7871589660644534e-05,
      "model_forward_time": 1.7172069549560547,
      "step": 50197
    },
    {
      "epoch": 4.7871589660644534e-05,
      "step": 50197,
      "training_step_time": 7.237618446350098
    },
    {
      "epoch": 4.787254333496094e-05,
      "model_forward_time": 1.719315528869629,
      "step": 50198
    },
    {
      "epoch": 4.787254333496094e-05,
      "step": 50198,
      "training_step_time": 7.7334678173065186
    },
    {
      "epoch": 4.787349700927734e-05,
      "model_forward_time": 1.7206928730010986,
      "step": 50199
    },
    {
      "epoch": 4.787349700927734e-05,
      "step": 50199,
      "training_step_time": 7.9048686027526855
    },
    {
      "epoch": 4.787445068359375e-05,
      "grad_norm": 0.1365019977092743,
      "learning_rate": 5.380096358046057e-06,
      "loss": 0.0527,
      "step": 50200
    },
    {
      "epoch": 4.787445068359375e-05,
      "model_forward_time": 1.7188184261322021,
      "step": 50200
    },
    {
      "dynamics_loss_avg": 0.05481870975345373,
      "epoch": 4.787445068359375e-05,
      "step": 50200
    },
    {
      "action_loss_avg": 0.0018474484706530348,
      "epoch": 4.787445068359375e-05,
      "step": 50200
    },
    {
      "epoch": 4.787445068359375e-05,
      "step": 50200,
      "training_step_time": 8.287742853164673
    },
    {
      "epoch": 4.787540435791016e-05,
      "model_forward_time": 2.285484552383423,
      "step": 50201
    },
    {
      "epoch": 4.787540435791016e-05,
      "step": 50201,
      "training_step_time": 7.78287935256958
    },
    {
      "epoch": 4.787635803222656e-05,
      "model_forward_time": 1.784067153930664,
      "step": 50202
    },
    {
      "epoch": 4.787635803222656e-05,
      "step": 50202,
      "training_step_time": 8.329098463058472
    },
    {
      "epoch": 4.787731170654297e-05,
      "model_forward_time": 1.7147870063781738,
      "step": 50203
    },
    {
      "epoch": 4.787731170654297e-05,
      "step": 50203,
      "training_step_time": 8.672215700149536
    },
    {
      "epoch": 4.7878265380859376e-05,
      "model_forward_time": 1.7113392353057861,
      "step": 50204
    },
    {
      "epoch": 4.7878265380859376e-05,
      "step": 50204,
      "training_step_time": 9.328477144241333
    },
    {
      "epoch": 4.787921905517578e-05,
      "model_forward_time": 1.718554973602295,
      "step": 50205
    },
    {
      "epoch": 4.787921905517578e-05,
      "step": 50205,
      "training_step_time": 7.6905364990234375
    },
    {
      "epoch": 4.788017272949219e-05,
      "model_forward_time": 1.7200212478637695,
      "step": 50206
    },
    {
      "epoch": 4.788017272949219e-05,
      "step": 50206,
      "training_step_time": 7.58234429359436
    },
    {
      "epoch": 4.7881126403808594e-05,
      "model_forward_time": 1.7305889129638672,
      "step": 50207
    },
    {
      "epoch": 4.7881126403808594e-05,
      "step": 50207,
      "training_step_time": 7.746323823928833
    },
    {
      "epoch": 4.7882080078125e-05,
      "model_forward_time": 1.7197136878967285,
      "step": 50208
    },
    {
      "epoch": 4.7882080078125e-05,
      "step": 50208,
      "training_step_time": 7.918731927871704
    },
    {
      "epoch": 4.788303375244141e-05,
      "model_forward_time": 1.7198171615600586,
      "step": 50209
    },
    {
      "epoch": 4.788303375244141e-05,
      "step": 50209,
      "training_step_time": 7.716707706451416
    },
    {
      "epoch": 4.788398742675781e-05,
      "grad_norm": 0.14760485291481018,
      "learning_rate": 5.378447652040359e-06,
      "loss": 0.0566,
      "step": 50210
    },
    {
      "epoch": 4.788398742675781e-05,
      "model_forward_time": 2.258150100708008,
      "step": 50210
    },
    {
      "dynamics_loss_avg": 0.06500684525817632,
      "epoch": 4.788398742675781e-05,
      "step": 50210
    },
    {
      "action_loss_avg": 0.0023153773043304684,
      "epoch": 4.788398742675781e-05,
      "step": 50210
    },
    {
      "epoch": 4.788398742675781e-05,
      "step": 50210,
      "training_step_time": 7.790353059768677
    },
    {
      "epoch": 4.788494110107422e-05,
      "model_forward_time": 1.375753402709961,
      "step": 50211
    },
    {
      "epoch": 4.788494110107422e-05,
      "step": 50211,
      "training_step_time": 9.664183616638184
    },
    {
      "epoch": 4.788589477539063e-05,
      "model_forward_time": 1.7190113067626953,
      "step": 50212
    },
    {
      "epoch": 4.788589477539063e-05,
      "step": 50212,
      "training_step_time": 8.191161394119263
    },
    {
      "epoch": 4.788684844970703e-05,
      "model_forward_time": 1.7155604362487793,
      "step": 50213
    },
    {
      "epoch": 4.788684844970703e-05,
      "step": 50213,
      "training_step_time": 7.663508892059326
    },
    {
      "epoch": 4.7887802124023436e-05,
      "model_forward_time": 1.7232491970062256,
      "step": 50214
    },
    {
      "epoch": 4.7887802124023436e-05,
      "step": 50214,
      "training_step_time": 9.912775993347168
    },
    {
      "epoch": 4.788875579833985e-05,
      "model_forward_time": 1.7204675674438477,
      "step": 50215
    },
    {
      "epoch": 4.788875579833985e-05,
      "step": 50215,
      "training_step_time": 7.605477333068848
    },
    {
      "epoch": 4.788970947265625e-05,
      "model_forward_time": 1.7238638401031494,
      "step": 50216
    },
    {
      "epoch": 4.788970947265625e-05,
      "step": 50216,
      "training_step_time": 7.690391540527344
    },
    {
      "epoch": 4.7890663146972655e-05,
      "model_forward_time": 1.722024917602539,
      "step": 50217
    },
    {
      "epoch": 4.7890663146972655e-05,
      "step": 50217,
      "training_step_time": 7.872777223587036
    },
    {
      "epoch": 4.7891616821289066e-05,
      "model_forward_time": 1.7225384712219238,
      "step": 50218
    },
    {
      "epoch": 4.7891616821289066e-05,
      "step": 50218,
      "training_step_time": 7.648083686828613
    },
    {
      "epoch": 4.789257049560547e-05,
      "model_forward_time": 1.7218966484069824,
      "step": 50219
    },
    {
      "epoch": 4.789257049560547e-05,
      "step": 50219,
      "training_step_time": 9.186736822128296
    },
    {
      "epoch": 4.7893524169921874e-05,
      "grad_norm": 0.11087676137685776,
      "learning_rate": 5.376798904648195e-06,
      "loss": 0.0565,
      "step": 50220
    },
    {
      "epoch": 4.7893524169921874e-05,
      "model_forward_time": 2.3257393836975098,
      "step": 50220
    },
    {
      "dynamics_loss_avg": 0.056286356039345266,
      "epoch": 4.7893524169921874e-05,
      "step": 50220
    },
    {
      "action_loss_avg": 0.002330865227850154,
      "epoch": 4.7893524169921874e-05,
      "step": 50220
    },
    {
      "epoch": 4.7893524169921874e-05,
      "step": 50220,
      "training_step_time": 8.269460916519165
    },
    {
      "epoch": 4.789447784423828e-05,
      "model_forward_time": 1.7192914485931396,
      "step": 50221
    },
    {
      "epoch": 4.789447784423828e-05,
      "step": 50221,
      "training_step_time": 8.218889951705933
    },
    {
      "epoch": 4.789543151855469e-05,
      "model_forward_time": 1.7113101482391357,
      "step": 50222
    },
    {
      "epoch": 4.789543151855469e-05,
      "step": 50222,
      "training_step_time": 7.58954381942749
    },
    {
      "epoch": 4.789638519287109e-05,
      "model_forward_time": 1.7202167510986328,
      "step": 50223
    },
    {
      "epoch": 4.789638519287109e-05,
      "step": 50223,
      "training_step_time": 8.7337167263031
    },
    {
      "epoch": 4.78973388671875e-05,
      "model_forward_time": 1.7297379970550537,
      "step": 50224
    },
    {
      "epoch": 4.78973388671875e-05,
      "step": 50224,
      "training_step_time": 7.690673589706421
    },
    {
      "epoch": 4.789829254150391e-05,
      "model_forward_time": 1.7174334526062012,
      "step": 50225
    },
    {
      "epoch": 4.789829254150391e-05,
      "step": 50225,
      "training_step_time": 8.64464545249939
    },
    {
      "epoch": 4.789924621582031e-05,
      "model_forward_time": 1.7138779163360596,
      "step": 50226
    },
    {
      "epoch": 4.789924621582031e-05,
      "step": 50226,
      "training_step_time": 7.566564321517944
    },
    {
      "epoch": 4.7900199890136716e-05,
      "model_forward_time": 1.7198073863983154,
      "step": 50227
    },
    {
      "epoch": 4.7900199890136716e-05,
      "step": 50227,
      "training_step_time": 7.82367730140686
    },
    {
      "epoch": 4.7901153564453126e-05,
      "model_forward_time": 1.7215440273284912,
      "step": 50228
    },
    {
      "epoch": 4.7901153564453126e-05,
      "step": 50228,
      "training_step_time": 7.534389495849609
    },
    {
      "epoch": 4.790210723876953e-05,
      "model_forward_time": 2.3023455142974854,
      "step": 50229
    },
    {
      "epoch": 4.790210723876953e-05,
      "step": 50229,
      "training_step_time": 7.77547812461853
    },
    {
      "epoch": 4.7903060913085935e-05,
      "grad_norm": 0.08529239892959595,
      "learning_rate": 5.375150116049869e-06,
      "loss": 0.0566,
      "step": 50230
    },
    {
      "epoch": 4.7903060913085935e-05,
      "model_forward_time": 1.708968162536621,
      "step": 50230
    },
    {
      "dynamics_loss_avg": 0.06744573339819908,
      "epoch": 4.7903060913085935e-05,
      "step": 50230
    },
    {
      "action_loss_avg": 0.001995180791709572,
      "epoch": 4.7903060913085935e-05,
      "step": 50230
    },
    {
      "epoch": 4.7903060913085935e-05,
      "step": 50230,
      "training_step_time": 8.543528079986572
    },
    {
      "epoch": 4.7904014587402345e-05,
      "model_forward_time": 1.697115182876587,
      "step": 50231
    },
    {
      "epoch": 4.7904014587402345e-05,
      "step": 50231,
      "training_step_time": 9.52049446105957
    },
    {
      "epoch": 4.790496826171875e-05,
      "model_forward_time": 1.7087972164154053,
      "step": 50232
    },
    {
      "epoch": 4.790496826171875e-05,
      "step": 50232,
      "training_step_time": 7.659178018569946
    },
    {
      "epoch": 4.790592193603515e-05,
      "model_forward_time": 1.6908695697784424,
      "step": 50233
    },
    {
      "epoch": 4.790592193603515e-05,
      "step": 50233,
      "training_step_time": 8.28964614868164
    },
    {
      "epoch": 4.7906875610351564e-05,
      "model_forward_time": 1.7193093299865723,
      "step": 50234
    },
    {
      "epoch": 4.7906875610351564e-05,
      "step": 50234,
      "training_step_time": 7.376144170761108
    },
    {
      "epoch": 4.790782928466797e-05,
      "model_forward_time": 1.3325796127319336,
      "step": 50235
    },
    {
      "epoch": 4.790782928466797e-05,
      "step": 50235,
      "training_step_time": 7.636927604675293
    },
    {
      "epoch": 4.790878295898437e-05,
      "model_forward_time": 1.7124080657958984,
      "step": 50236
    },
    {
      "epoch": 4.790878295898437e-05,
      "step": 50236,
      "training_step_time": 7.832516431808472
    },
    {
      "epoch": 4.790973663330078e-05,
      "model_forward_time": 1.7128286361694336,
      "step": 50237
    },
    {
      "epoch": 4.790973663330078e-05,
      "step": 50237,
      "training_step_time": 7.494017601013184
    },
    {
      "epoch": 4.791069030761719e-05,
      "model_forward_time": 2.292022943496704,
      "step": 50238
    },
    {
      "epoch": 4.791069030761719e-05,
      "step": 50238,
      "training_step_time": 7.696152925491333
    },
    {
      "epoch": 4.791164398193359e-05,
      "model_forward_time": 1.708390474319458,
      "step": 50239
    },
    {
      "epoch": 4.791164398193359e-05,
      "step": 50239,
      "training_step_time": 8.481500625610352
    },
    {
      "epoch": 4.791259765625e-05,
      "grad_norm": 0.08634334057569504,
      "learning_rate": 5.373501286425691e-06,
      "loss": 0.057,
      "step": 50240
    },
    {
      "epoch": 4.791259765625e-05,
      "model_forward_time": 1.7105646133422852,
      "step": 50240
    },
    {
      "dynamics_loss_avg": 0.05809774026274681,
      "epoch": 4.791259765625e-05,
      "step": 50240
    },
    {
      "action_loss_avg": 0.0012538765149656683,
      "epoch": 4.791259765625e-05,
      "step": 50240
    },
    {
      "epoch": 4.791259765625e-05,
      "step": 50240,
      "training_step_time": 7.6458446979522705
    },
    {
      "epoch": 4.7913551330566406e-05,
      "model_forward_time": 1.7108547687530518,
      "step": 50241
    },
    {
      "epoch": 4.7913551330566406e-05,
      "step": 50241,
      "training_step_time": 8.355168581008911
    },
    {
      "epoch": 4.791450500488281e-05,
      "model_forward_time": 1.7126355171203613,
      "step": 50242
    },
    {
      "epoch": 4.791450500488281e-05,
      "step": 50242,
      "training_step_time": 7.668874979019165
    },
    {
      "epoch": 4.791545867919922e-05,
      "model_forward_time": 1.7115731239318848,
      "step": 50243
    },
    {
      "epoch": 4.791545867919922e-05,
      "step": 50243,
      "training_step_time": 7.366845369338989
    },
    {
      "epoch": 4.7916412353515625e-05,
      "model_forward_time": 1.7058143615722656,
      "step": 50244
    },
    {
      "epoch": 4.7916412353515625e-05,
      "step": 50244,
      "training_step_time": 7.735333204269409
    },
    {
      "epoch": 4.791736602783203e-05,
      "model_forward_time": 1.7088086605072021,
      "step": 50245
    },
    {
      "epoch": 4.791736602783203e-05,
      "step": 50245,
      "training_step_time": 7.828997850418091
    },
    {
      "epoch": 4.791831970214844e-05,
      "model_forward_time": 1.7179615497589111,
      "step": 50246
    },
    {
      "epoch": 4.791831970214844e-05,
      "step": 50246,
      "training_step_time": 7.6353843212127686
    },
    {
      "epoch": 4.7919273376464844e-05,
      "model_forward_time": 1.7167489528656006,
      "step": 50247
    },
    {
      "epoch": 4.7919273376464844e-05,
      "step": 50247,
      "training_step_time": 8.2512788772583
    },
    {
      "epoch": 4.792022705078125e-05,
      "model_forward_time": 1.7772877216339111,
      "step": 50248
    },
    {
      "epoch": 4.792022705078125e-05,
      "step": 50248,
      "training_step_time": 10.274173021316528
    },
    {
      "epoch": 4.792118072509766e-05,
      "model_forward_time": 1.7154650688171387,
      "step": 50249
    },
    {
      "epoch": 4.792118072509766e-05,
      "step": 50249,
      "training_step_time": 8.495409488677979
    },
    {
      "epoch": 4.792213439941406e-05,
      "grad_norm": 0.11014318466186523,
      "learning_rate": 5.3718524159559724e-06,
      "loss": 0.0569,
      "step": 50250
    },
    {
      "epoch": 4.792213439941406e-05,
      "model_forward_time": 1.72157621383667,
      "step": 50250
    },
    {
      "dynamics_loss_avg": 0.054018386825919154,
      "epoch": 4.792213439941406e-05,
      "step": 50250
    },
    {
      "action_loss_avg": 0.0016812459216453135,
      "epoch": 4.792213439941406e-05,
      "step": 50250
    },
    {
      "epoch": 4.792213439941406e-05,
      "step": 50250,
      "training_step_time": 8.21058177947998
    },
    {
      "epoch": 4.7923088073730467e-05,
      "model_forward_time": 1.7067577838897705,
      "step": 50251
    },
    {
      "epoch": 4.7923088073730467e-05,
      "step": 50251,
      "training_step_time": 8.688954830169678
    },
    {
      "epoch": 4.792404174804688e-05,
      "model_forward_time": 1.7004191875457764,
      "step": 50252
    },
    {
      "epoch": 4.792404174804688e-05,
      "step": 50252,
      "training_step_time": 7.403457164764404
    },
    {
      "epoch": 4.792499542236328e-05,
      "model_forward_time": 1.7194511890411377,
      "step": 50253
    },
    {
      "epoch": 4.792499542236328e-05,
      "step": 50253,
      "training_step_time": 7.735857248306274
    },
    {
      "epoch": 4.7925949096679685e-05,
      "model_forward_time": 1.718545913696289,
      "step": 50254
    },
    {
      "epoch": 4.7925949096679685e-05,
      "step": 50254,
      "training_step_time": 7.8524627685546875
    },
    {
      "epoch": 4.7926902770996096e-05,
      "model_forward_time": 1.7097740173339844,
      "step": 50255
    },
    {
      "epoch": 4.7926902770996096e-05,
      "step": 50255,
      "training_step_time": 7.6319944858551025
    },
    {
      "epoch": 4.79278564453125e-05,
      "model_forward_time": 1.7088444232940674,
      "step": 50256
    },
    {
      "epoch": 4.79278564453125e-05,
      "step": 50256,
      "training_step_time": 8.333509683609009
    },
    {
      "epoch": 4.7928810119628904e-05,
      "model_forward_time": 2.319697141647339,
      "step": 50257
    },
    {
      "epoch": 4.7928810119628904e-05,
      "step": 50257,
      "training_step_time": 8.519334077835083
    },
    {
      "epoch": 4.7929763793945315e-05,
      "model_forward_time": 1.719390630722046,
      "step": 50258
    },
    {
      "epoch": 4.7929763793945315e-05,
      "step": 50258,
      "training_step_time": 8.471848487854004
    },
    {
      "epoch": 4.793071746826172e-05,
      "model_forward_time": 1.7078578472137451,
      "step": 50259
    },
    {
      "epoch": 4.793071746826172e-05,
      "step": 50259,
      "training_step_time": 7.676320552825928
    },
    {
      "epoch": 4.793167114257812e-05,
      "grad_norm": 0.10721225291490555,
      "learning_rate": 5.370203504821034e-06,
      "loss": 0.0562,
      "step": 50260
    },
    {
      "epoch": 4.793167114257812e-05,
      "model_forward_time": 1.705491065979004,
      "step": 50260
    },
    {
      "dynamics_loss_avg": 0.060423812456429006,
      "epoch": 4.793167114257812e-05,
      "step": 50260
    },
    {
      "action_loss_avg": 0.0019023744389414787,
      "epoch": 4.793167114257812e-05,
      "step": 50260
    },
    {
      "epoch": 4.793167114257812e-05,
      "step": 50260,
      "training_step_time": 7.6249589920043945
    },
    {
      "epoch": 4.7932624816894534e-05,
      "model_forward_time": 1.7171313762664795,
      "step": 50261
    },
    {
      "epoch": 4.7932624816894534e-05,
      "step": 50261,
      "training_step_time": 8.298875331878662
    },
    {
      "epoch": 4.793357849121094e-05,
      "model_forward_time": 1.7174129486083984,
      "step": 50262
    },
    {
      "epoch": 4.793357849121094e-05,
      "step": 50262,
      "training_step_time": 8.341683626174927
    },
    {
      "epoch": 4.793453216552734e-05,
      "model_forward_time": 1.7165253162384033,
      "step": 50263
    },
    {
      "epoch": 4.793453216552734e-05,
      "step": 50263,
      "training_step_time": 8.707380294799805
    },
    {
      "epoch": 4.793548583984375e-05,
      "model_forward_time": 1.7120742797851562,
      "step": 50264
    },
    {
      "epoch": 4.793548583984375e-05,
      "step": 50264,
      "training_step_time": 7.57115364074707
    },
    {
      "epoch": 4.793643951416016e-05,
      "model_forward_time": 1.7146551609039307,
      "step": 50265
    },
    {
      "epoch": 4.793643951416016e-05,
      "step": 50265,
      "training_step_time": 8.465867757797241
    },
    {
      "epoch": 4.793739318847656e-05,
      "model_forward_time": 2.3132269382476807,
      "step": 50266
    },
    {
      "epoch": 4.793739318847656e-05,
      "step": 50266,
      "training_step_time": 8.850368976593018
    },
    {
      "epoch": 4.793834686279297e-05,
      "model_forward_time": 1.7102901935577393,
      "step": 50267
    },
    {
      "epoch": 4.793834686279297e-05,
      "step": 50267,
      "training_step_time": 8.173557043075562
    },
    {
      "epoch": 4.7939300537109376e-05,
      "model_forward_time": 1.7164771556854248,
      "step": 50268
    },
    {
      "epoch": 4.7939300537109376e-05,
      "step": 50268,
      "training_step_time": 8.411094665527344
    },
    {
      "epoch": 4.794025421142578e-05,
      "model_forward_time": 1.7158467769622803,
      "step": 50269
    },
    {
      "epoch": 4.794025421142578e-05,
      "step": 50269,
      "training_step_time": 7.788433313369751
    },
    {
      "epoch": 4.794120788574219e-05,
      "grad_norm": 0.10206808149814606,
      "learning_rate": 5.368554553201196e-06,
      "loss": 0.0597,
      "step": 50270
    },
    {
      "epoch": 4.794120788574219e-05,
      "model_forward_time": 1.7149527072906494,
      "step": 50270
    },
    {
      "dynamics_loss_avg": 0.055023737251758575,
      "epoch": 4.794120788574219e-05,
      "step": 50270
    },
    {
      "action_loss_avg": 0.0020986610965337605,
      "epoch": 4.794120788574219e-05,
      "step": 50270
    },
    {
      "epoch": 4.794120788574219e-05,
      "step": 50270,
      "training_step_time": 7.557704210281372
    },
    {
      "epoch": 4.7942161560058595e-05,
      "model_forward_time": 1.7116823196411133,
      "step": 50271
    },
    {
      "epoch": 4.7942161560058595e-05,
      "step": 50271,
      "training_step_time": 7.474807977676392
    },
    {
      "epoch": 4.7943115234375e-05,
      "model_forward_time": 1.714608907699585,
      "step": 50272
    },
    {
      "epoch": 4.7943115234375e-05,
      "step": 50272,
      "training_step_time": 7.864459037780762
    },
    {
      "epoch": 4.794406890869141e-05,
      "model_forward_time": 1.718846321105957,
      "step": 50273
    },
    {
      "epoch": 4.794406890869141e-05,
      "step": 50273,
      "training_step_time": 7.5810582637786865
    },
    {
      "epoch": 4.794502258300781e-05,
      "model_forward_time": 1.7199301719665527,
      "step": 50274
    },
    {
      "epoch": 4.794502258300781e-05,
      "step": 50274,
      "training_step_time": 8.17520523071289
    },
    {
      "epoch": 4.794597625732422e-05,
      "model_forward_time": 2.2963576316833496,
      "step": 50275
    },
    {
      "epoch": 4.794597625732422e-05,
      "step": 50275,
      "training_step_time": 7.736966371536255
    },
    {
      "epoch": 4.794692993164063e-05,
      "model_forward_time": 1.724501371383667,
      "step": 50276
    },
    {
      "epoch": 4.794692993164063e-05,
      "step": 50276,
      "training_step_time": 7.655573129653931
    },
    {
      "epoch": 4.794788360595703e-05,
      "model_forward_time": 1.6986939907073975,
      "step": 50277
    },
    {
      "epoch": 4.794788360595703e-05,
      "step": 50277,
      "training_step_time": 8.37479567527771
    },
    {
      "epoch": 4.7948837280273436e-05,
      "model_forward_time": 1.7137372493743896,
      "step": 50278
    },
    {
      "epoch": 4.7948837280273436e-05,
      "step": 50278,
      "training_step_time": 7.740241050720215
    },
    {
      "epoch": 4.794979095458985e-05,
      "model_forward_time": 1.7168352603912354,
      "step": 50279
    },
    {
      "epoch": 4.794979095458985e-05,
      "step": 50279,
      "training_step_time": 8.26292109489441
    },
    {
      "epoch": 4.795074462890625e-05,
      "grad_norm": 0.08526530116796494,
      "learning_rate": 5.366905561276787e-06,
      "loss": 0.0551,
      "step": 50280
    },
    {
      "epoch": 4.795074462890625e-05,
      "model_forward_time": 1.7072131633758545,
      "step": 50280
    },
    {
      "dynamics_loss_avg": 0.05971305221319199,
      "epoch": 4.795074462890625e-05,
      "step": 50280
    },
    {
      "action_loss_avg": 0.0019978401018306614,
      "epoch": 4.795074462890625e-05,
      "step": 50280
    },
    {
      "epoch": 4.795074462890625e-05,
      "step": 50280,
      "training_step_time": 7.406810760498047
    },
    {
      "epoch": 4.7951698303222655e-05,
      "model_forward_time": 1.7106430530548096,
      "step": 50281
    },
    {
      "epoch": 4.7951698303222655e-05,
      "step": 50281,
      "training_step_time": 8.215477466583252
    },
    {
      "epoch": 4.795265197753906e-05,
      "model_forward_time": 1.712451457977295,
      "step": 50282
    },
    {
      "epoch": 4.795265197753906e-05,
      "step": 50282,
      "training_step_time": 7.680263519287109
    },
    {
      "epoch": 4.795360565185547e-05,
      "model_forward_time": 1.7030341625213623,
      "step": 50283
    },
    {
      "epoch": 4.795360565185547e-05,
      "step": 50283,
      "training_step_time": 7.696256637573242
    },
    {
      "epoch": 4.7954559326171874e-05,
      "model_forward_time": 2.2814741134643555,
      "step": 50284
    },
    {
      "epoch": 4.7954559326171874e-05,
      "step": 50284,
      "training_step_time": 7.723694086074829
    },
    {
      "epoch": 4.795551300048828e-05,
      "model_forward_time": 1.7285091876983643,
      "step": 50285
    },
    {
      "epoch": 4.795551300048828e-05,
      "step": 50285,
      "training_step_time": 8.319234609603882
    },
    {
      "epoch": 4.795646667480469e-05,
      "model_forward_time": 1.7042224407196045,
      "step": 50286
    },
    {
      "epoch": 4.795646667480469e-05,
      "step": 50286,
      "training_step_time": 7.66438889503479
    },
    {
      "epoch": 4.795742034912109e-05,
      "model_forward_time": 1.7105469703674316,
      "step": 50287
    },
    {
      "epoch": 4.795742034912109e-05,
      "step": 50287,
      "training_step_time": 7.769921779632568
    },
    {
      "epoch": 4.79583740234375e-05,
      "model_forward_time": 1.7122421264648438,
      "step": 50288
    },
    {
      "epoch": 4.79583740234375e-05,
      "step": 50288,
      "training_step_time": 7.672137260437012
    },
    {
      "epoch": 4.795932769775391e-05,
      "model_forward_time": 1.7115471363067627,
      "step": 50289
    },
    {
      "epoch": 4.795932769775391e-05,
      "step": 50289,
      "training_step_time": 7.399553298950195
    },
    {
      "epoch": 4.796028137207031e-05,
      "grad_norm": 0.12349478900432587,
      "learning_rate": 5.365256529228138e-06,
      "loss": 0.06,
      "step": 50290
    },
    {
      "epoch": 4.796028137207031e-05,
      "model_forward_time": 1.7159595489501953,
      "step": 50290
    },
    {
      "dynamics_loss_avg": 0.06802064199000597,
      "epoch": 4.796028137207031e-05,
      "step": 50290
    },
    {
      "action_loss_avg": 0.002665456722024828,
      "epoch": 4.796028137207031e-05,
      "step": 50290
    },
    {
      "epoch": 4.796028137207031e-05,
      "step": 50290,
      "training_step_time": 7.665343523025513
    },
    {
      "epoch": 4.7961235046386716e-05,
      "model_forward_time": 1.7286419868469238,
      "step": 50291
    },
    {
      "epoch": 4.7961235046386716e-05,
      "step": 50291,
      "training_step_time": 7.8547585010528564
    },
    {
      "epoch": 4.7962188720703127e-05,
      "model_forward_time": 1.72214937210083,
      "step": 50292
    },
    {
      "epoch": 4.7962188720703127e-05,
      "step": 50292,
      "training_step_time": 7.738072395324707
    },
    {
      "epoch": 4.796314239501953e-05,
      "model_forward_time": 2.275836706161499,
      "step": 50293
    },
    {
      "epoch": 4.796314239501953e-05,
      "step": 50293,
      "training_step_time": 7.645150423049927
    },
    {
      "epoch": 4.7964096069335935e-05,
      "model_forward_time": 1.7854559421539307,
      "step": 50294
    },
    {
      "epoch": 4.7964096069335935e-05,
      "step": 50294,
      "training_step_time": 9.783833503723145
    },
    {
      "epoch": 4.7965049743652345e-05,
      "model_forward_time": 1.7103331089019775,
      "step": 50295
    },
    {
      "epoch": 4.7965049743652345e-05,
      "step": 50295,
      "training_step_time": 8.712440013885498
    },
    {
      "epoch": 4.796600341796875e-05,
      "model_forward_time": 0.9940934181213379,
      "step": 50296
    },
    {
      "epoch": 4.796600341796875e-05,
      "step": 50296,
      "training_step_time": 10.189996242523193
    },
    {
      "epoch": 4.7966957092285153e-05,
      "model_forward_time": 1.7292561531066895,
      "step": 50297
    },
    {
      "epoch": 4.7966957092285153e-05,
      "step": 50297,
      "training_step_time": 7.857162714004517
    },
    {
      "epoch": 4.7967910766601564e-05,
      "model_forward_time": 1.7242200374603271,
      "step": 50298
    },
    {
      "epoch": 4.7967910766601564e-05,
      "step": 50298,
      "training_step_time": 7.7722413539886475
    },
    {
      "epoch": 4.796886444091797e-05,
      "model_forward_time": 1.7234065532684326,
      "step": 50299
    },
    {
      "epoch": 4.796886444091797e-05,
      "step": 50299,
      "training_step_time": 7.853009939193726
    },
    {
      "epoch": 4.796981811523437e-05,
      "grad_norm": 0.08540946990251541,
      "learning_rate": 5.363607457235582e-06,
      "loss": 0.0527,
      "step": 50300
    },
    {
      "epoch": 4.796981811523437e-05,
      "model_forward_time": 1.731168270111084,
      "step": 50300
    },
    {
      "dynamics_loss_avg": 0.049134825728833674,
      "epoch": 4.796981811523437e-05,
      "step": 50300
    },
    {
      "action_loss_avg": 0.0018333853222429752,
      "epoch": 4.796981811523437e-05,
      "step": 50300
    },
    {
      "epoch": 4.796981811523437e-05,
      "step": 50300,
      "training_step_time": 8.062264680862427
    },
    {
      "epoch": 4.797077178955078e-05,
      "model_forward_time": 1.7273805141448975,
      "step": 50301
    },
    {
      "epoch": 4.797077178955078e-05,
      "step": 50301,
      "training_step_time": 7.857152462005615
    },
    {
      "epoch": 4.797172546386719e-05,
      "model_forward_time": 1.7277438640594482,
      "step": 50302
    },
    {
      "epoch": 4.797172546386719e-05,
      "step": 50302,
      "training_step_time": 8.357692003250122
    },
    {
      "epoch": 4.797267913818359e-05,
      "model_forward_time": 2.3500444889068604,
      "step": 50303
    },
    {
      "epoch": 4.797267913818359e-05,
      "step": 50303,
      "training_step_time": 8.764517545700073
    },
    {
      "epoch": 4.79736328125e-05,
      "model_forward_time": 1.7191994190216064,
      "step": 50304
    },
    {
      "epoch": 4.79736328125e-05,
      "step": 50304,
      "training_step_time": 7.643280029296875
    },
    {
      "epoch": 4.7974586486816406e-05,
      "model_forward_time": 1.7273192405700684,
      "step": 50305
    },
    {
      "epoch": 4.7974586486816406e-05,
      "step": 50305,
      "training_step_time": 8.899604320526123
    },
    {
      "epoch": 4.797554016113281e-05,
      "model_forward_time": 1.722538948059082,
      "step": 50306
    },
    {
      "epoch": 4.797554016113281e-05,
      "step": 50306,
      "training_step_time": 7.866169452667236
    },
    {
      "epoch": 4.797649383544922e-05,
      "model_forward_time": 1.726839542388916,
      "step": 50307
    },
    {
      "epoch": 4.797649383544922e-05,
      "step": 50307,
      "training_step_time": 7.786059141159058
    },
    {
      "epoch": 4.7977447509765625e-05,
      "model_forward_time": 1.7197351455688477,
      "step": 50308
    },
    {
      "epoch": 4.7977447509765625e-05,
      "step": 50308,
      "training_step_time": 9.219708919525146
    },
    {
      "epoch": 4.797840118408203e-05,
      "model_forward_time": 1.0121965408325195,
      "step": 50309
    },
    {
      "epoch": 4.797840118408203e-05,
      "step": 50309,
      "training_step_time": 8.10705828666687
    },
    {
      "epoch": 4.797935485839844e-05,
      "grad_norm": 0.11995099484920502,
      "learning_rate": 5.3619583454794635e-06,
      "loss": 0.0581,
      "step": 50310
    },
    {
      "epoch": 4.797935485839844e-05,
      "model_forward_time": 1.696721076965332,
      "step": 50310
    },
    {
      "dynamics_loss_avg": 0.06155152805149555,
      "epoch": 4.797935485839844e-05,
      "step": 50310
    },
    {
      "action_loss_avg": 0.0025211596162989735,
      "epoch": 4.797935485839844e-05,
      "step": 50310
    },
    {
      "epoch": 4.797935485839844e-05,
      "step": 50310,
      "training_step_time": 7.792318105697632
    },
    {
      "epoch": 4.7980308532714844e-05,
      "model_forward_time": 1.726386308670044,
      "step": 50311
    },
    {
      "epoch": 4.7980308532714844e-05,
      "step": 50311,
      "training_step_time": 9.61062741279602
    },
    {
      "epoch": 4.798126220703125e-05,
      "model_forward_time": 2.2958903312683105,
      "step": 50312
    },
    {
      "epoch": 4.798126220703125e-05,
      "step": 50312,
      "training_step_time": 8.208101749420166
    },
    {
      "epoch": 4.798221588134766e-05,
      "model_forward_time": 1.720717191696167,
      "step": 50313
    },
    {
      "epoch": 4.798221588134766e-05,
      "step": 50313,
      "training_step_time": 8.714335441589355
    },
    {
      "epoch": 4.798316955566406e-05,
      "model_forward_time": 1.7316701412200928,
      "step": 50314
    },
    {
      "epoch": 4.798316955566406e-05,
      "step": 50314,
      "training_step_time": 9.395711898803711
    },
    {
      "epoch": 4.7984123229980467e-05,
      "model_forward_time": 1.7244982719421387,
      "step": 50315
    },
    {
      "epoch": 4.7984123229980467e-05,
      "step": 50315,
      "training_step_time": 7.924495220184326
    },
    {
      "epoch": 4.798507690429688e-05,
      "model_forward_time": 1.7333812713623047,
      "step": 50316
    },
    {
      "epoch": 4.798507690429688e-05,
      "step": 50316,
      "training_step_time": 7.841877698898315
    },
    {
      "epoch": 4.798603057861328e-05,
      "model_forward_time": 1.732375144958496,
      "step": 50317
    },
    {
      "epoch": 4.798603057861328e-05,
      "step": 50317,
      "training_step_time": 7.5851149559021
    },
    {
      "epoch": 4.7986984252929685e-05,
      "model_forward_time": 1.728445053100586,
      "step": 50318
    },
    {
      "epoch": 4.7986984252929685e-05,
      "step": 50318,
      "training_step_time": 8.151426553726196
    },
    {
      "epoch": 4.7987937927246096e-05,
      "model_forward_time": 1.7281203269958496,
      "step": 50319
    },
    {
      "epoch": 4.7987937927246096e-05,
      "step": 50319,
      "training_step_time": 7.711061000823975
    },
    {
      "epoch": 4.79888916015625e-05,
      "grad_norm": 0.10376516729593277,
      "learning_rate": 5.360309194140123e-06,
      "loss": 0.0562,
      "step": 50320
    },
    {
      "epoch": 4.79888916015625e-05,
      "model_forward_time": 1.7275760173797607,
      "step": 50320
    },
    {
      "dynamics_loss_avg": 0.056933865323662756,
      "epoch": 4.79888916015625e-05,
      "step": 50320
    },
    {
      "action_loss_avg": 0.002439815262914635,
      "epoch": 4.79888916015625e-05,
      "step": 50320
    },
    {
      "epoch": 4.79888916015625e-05,
      "step": 50320,
      "training_step_time": 7.93777322769165
    },
    {
      "epoch": 4.7989845275878904e-05,
      "model_forward_time": 1.3547108173370361,
      "step": 50321
    },
    {
      "epoch": 4.7989845275878904e-05,
      "step": 50321,
      "training_step_time": 8.809141635894775
    },
    {
      "epoch": 4.7990798950195315e-05,
      "model_forward_time": 2.268691062927246,
      "step": 50322
    },
    {
      "epoch": 4.7990798950195315e-05,
      "step": 50322,
      "training_step_time": 8.702736139297485
    },
    {
      "epoch": 4.799175262451172e-05,
      "model_forward_time": 1.7337076663970947,
      "step": 50323
    },
    {
      "epoch": 4.799175262451172e-05,
      "step": 50323,
      "training_step_time": 8.448455572128296
    },
    {
      "epoch": 4.799270629882812e-05,
      "model_forward_time": 1.7304370403289795,
      "step": 50324
    },
    {
      "epoch": 4.799270629882812e-05,
      "step": 50324,
      "training_step_time": 7.981262683868408
    },
    {
      "epoch": 4.7993659973144534e-05,
      "model_forward_time": 1.734494686126709,
      "step": 50325
    },
    {
      "epoch": 4.7993659973144534e-05,
      "step": 50325,
      "training_step_time": 7.8226237297058105
    },
    {
      "epoch": 4.799461364746094e-05,
      "model_forward_time": 1.7395930290222168,
      "step": 50326
    },
    {
      "epoch": 4.799461364746094e-05,
      "step": 50326,
      "training_step_time": 7.544020891189575
    },
    {
      "epoch": 4.799556732177734e-05,
      "model_forward_time": 1.7335317134857178,
      "step": 50327
    },
    {
      "epoch": 4.799556732177734e-05,
      "step": 50327,
      "training_step_time": 8.066309928894043
    },
    {
      "epoch": 4.799652099609375e-05,
      "model_forward_time": 1.732562780380249,
      "step": 50328
    },
    {
      "epoch": 4.799652099609375e-05,
      "step": 50328,
      "training_step_time": 7.78014063835144
    },
    {
      "epoch": 4.799747467041016e-05,
      "model_forward_time": 1.7375192642211914,
      "step": 50329
    },
    {
      "epoch": 4.799747467041016e-05,
      "step": 50329,
      "training_step_time": 7.961278915405273
    },
    {
      "epoch": 4.799842834472656e-05,
      "grad_norm": 0.10395698994398117,
      "learning_rate": 5.3586600033979106e-06,
      "loss": 0.0551,
      "step": 50330
    },
    {
      "epoch": 4.799842834472656e-05,
      "model_forward_time": 1.721409559249878,
      "step": 50330
    },
    {
      "dynamics_loss_avg": 0.0670424297451973,
      "epoch": 4.799842834472656e-05,
      "step": 50330
    },
    {
      "action_loss_avg": 0.0032437104731798173,
      "epoch": 4.799842834472656e-05,
      "step": 50330
    },
    {
      "epoch": 4.799842834472656e-05,
      "step": 50330,
      "training_step_time": 8.716456651687622
    },
    {
      "epoch": 4.799938201904297e-05,
      "model_forward_time": 2.297157049179077,
      "step": 50331
    },
    {
      "epoch": 4.799938201904297e-05,
      "step": 50331,
      "training_step_time": 7.670535326004028
    },
    {
      "epoch": 4.8000335693359376e-05,
      "model_forward_time": 1.7292397022247314,
      "step": 50332
    },
    {
      "epoch": 4.8000335693359376e-05,
      "step": 50332,
      "training_step_time": 7.755782127380371
    },
    {
      "epoch": 4.800128936767578e-05,
      "model_forward_time": 1.728369951248169,
      "step": 50333
    },
    {
      "epoch": 4.800128936767578e-05,
      "step": 50333,
      "training_step_time": 7.77109432220459
    },
    {
      "epoch": 4.800224304199219e-05,
      "model_forward_time": 1.7317750453948975,
      "step": 50334
    },
    {
      "epoch": 4.800224304199219e-05,
      "step": 50334,
      "training_step_time": 7.6563849449157715
    },
    {
      "epoch": 4.8003196716308595e-05,
      "model_forward_time": 1.7338931560516357,
      "step": 50335
    },
    {
      "epoch": 4.8003196716308595e-05,
      "step": 50335,
      "training_step_time": 7.3764588832855225
    },
    {
      "epoch": 4.8004150390625e-05,
      "model_forward_time": 1.7319679260253906,
      "step": 50336
    },
    {
      "epoch": 4.8004150390625e-05,
      "step": 50336,
      "training_step_time": 7.718081712722778
    },
    {
      "epoch": 4.800510406494141e-05,
      "model_forward_time": 1.7285518646240234,
      "step": 50337
    },
    {
      "epoch": 4.800510406494141e-05,
      "step": 50337,
      "training_step_time": 7.880182266235352
    },
    {
      "epoch": 4.8006057739257813e-05,
      "model_forward_time": 1.7388474941253662,
      "step": 50338
    },
    {
      "epoch": 4.8006057739257813e-05,
      "step": 50338,
      "training_step_time": 7.7785420417785645
    },
    {
      "epoch": 4.800701141357422e-05,
      "model_forward_time": 1.7162678241729736,
      "step": 50339
    },
    {
      "epoch": 4.800701141357422e-05,
      "step": 50339,
      "training_step_time": 8.61570405960083
    },
    {
      "epoch": 4.800796508789063e-05,
      "grad_norm": 0.10724268108606339,
      "learning_rate": 5.357010773433181e-06,
      "loss": 0.0584,
      "step": 50340
    },
    {
      "epoch": 4.800796508789063e-05,
      "model_forward_time": 2.332439661026001,
      "step": 50340
    },
    {
      "dynamics_loss_avg": 0.06634458750486374,
      "epoch": 4.800796508789063e-05,
      "step": 50340
    },
    {
      "action_loss_avg": 0.002205653878627345,
      "epoch": 4.800796508789063e-05,
      "step": 50340
    },
    {
      "epoch": 4.800796508789063e-05,
      "step": 50340,
      "training_step_time": 8.480912923812866
    },
    {
      "epoch": 4.800891876220703e-05,
      "model_forward_time": 1.727210521697998,
      "step": 50341
    },
    {
      "epoch": 4.800891876220703e-05,
      "step": 50341,
      "training_step_time": 7.770198345184326
    },
    {
      "epoch": 4.8009872436523436e-05,
      "model_forward_time": 1.7289187908172607,
      "step": 50342
    },
    {
      "epoch": 4.8009872436523436e-05,
      "step": 50342,
      "training_step_time": 8.306273698806763
    },
    {
      "epoch": 4.801082611083985e-05,
      "model_forward_time": 1.7411866188049316,
      "step": 50343
    },
    {
      "epoch": 4.801082611083985e-05,
      "step": 50343,
      "training_step_time": 9.576678037643433
    },
    {
      "epoch": 4.801177978515625e-05,
      "model_forward_time": 1.7384560108184814,
      "step": 50344
    },
    {
      "epoch": 4.801177978515625e-05,
      "step": 50344,
      "training_step_time": 7.413748025894165
    },
    {
      "epoch": 4.8012733459472655e-05,
      "model_forward_time": 1.7437238693237305,
      "step": 50345
    },
    {
      "epoch": 4.8012733459472655e-05,
      "step": 50345,
      "training_step_time": 7.682107448577881
    },
    {
      "epoch": 4.801368713378906e-05,
      "model_forward_time": 1.7300162315368652,
      "step": 50346
    },
    {
      "epoch": 4.801368713378906e-05,
      "step": 50346,
      "training_step_time": 7.562309503555298
    },
    {
      "epoch": 4.801464080810547e-05,
      "model_forward_time": 1.7329847812652588,
      "step": 50347
    },
    {
      "epoch": 4.801464080810547e-05,
      "step": 50347,
      "training_step_time": 7.761179685592651
    },
    {
      "epoch": 4.8015594482421874e-05,
      "model_forward_time": 1.733046531677246,
      "step": 50348
    },
    {
      "epoch": 4.8015594482421874e-05,
      "step": 50348,
      "training_step_time": 7.705363988876343
    },
    {
      "epoch": 4.801654815673828e-05,
      "model_forward_time": 2.3195688724517822,
      "step": 50349
    },
    {
      "epoch": 4.801654815673828e-05,
      "step": 50349,
      "training_step_time": 8.788987398147583
    },
    {
      "epoch": 4.801750183105469e-05,
      "grad_norm": 0.15194180607795715,
      "learning_rate": 5.355361504426289e-06,
      "loss": 0.0603,
      "step": 50350
    },
    {
      "epoch": 4.801750183105469e-05,
      "model_forward_time": 1.7245502471923828,
      "step": 50350
    },
    {
      "dynamics_loss_avg": 0.06372104752808809,
      "epoch": 4.801750183105469e-05,
      "step": 50350
    },
    {
      "action_loss_avg": 0.0027185143087990584,
      "epoch": 4.801750183105469e-05,
      "step": 50350
    },
    {
      "epoch": 4.801750183105469e-05,
      "step": 50350,
      "training_step_time": 7.7614874839782715
    },
    {
      "epoch": 4.801845550537109e-05,
      "model_forward_time": 1.732025146484375,
      "step": 50351
    },
    {
      "epoch": 4.801845550537109e-05,
      "step": 50351,
      "training_step_time": 7.693674087524414
    },
    {
      "epoch": 4.80194091796875e-05,
      "model_forward_time": 1.7346489429473877,
      "step": 50352
    },
    {
      "epoch": 4.80194091796875e-05,
      "step": 50352,
      "training_step_time": 8.805794954299927
    },
    {
      "epoch": 4.802036285400391e-05,
      "model_forward_time": 1.7465391159057617,
      "step": 50353
    },
    {
      "epoch": 4.802036285400391e-05,
      "step": 50353,
      "training_step_time": 7.729501485824585
    },
    {
      "epoch": 4.802131652832031e-05,
      "model_forward_time": 1.7290081977844238,
      "step": 50354
    },
    {
      "epoch": 4.802131652832031e-05,
      "step": 50354,
      "training_step_time": 7.737190008163452
    },
    {
      "epoch": 4.8022270202636716e-05,
      "model_forward_time": 1.72774076461792,
      "step": 50355
    },
    {
      "epoch": 4.8022270202636716e-05,
      "step": 50355,
      "training_step_time": 7.819809913635254
    },
    {
      "epoch": 4.8023223876953127e-05,
      "model_forward_time": 1.7379834651947021,
      "step": 50356
    },
    {
      "epoch": 4.8023223876953127e-05,
      "step": 50356,
      "training_step_time": 7.699005126953125
    },
    {
      "epoch": 4.802417755126953e-05,
      "model_forward_time": 1.72894287109375,
      "step": 50357
    },
    {
      "epoch": 4.802417755126953e-05,
      "step": 50357,
      "training_step_time": 8.76446008682251
    },
    {
      "epoch": 4.8025131225585935e-05,
      "model_forward_time": 1.779909610748291,
      "step": 50358
    },
    {
      "epoch": 4.8025131225585935e-05,
      "step": 50358,
      "training_step_time": 9.394851684570312
    },
    {
      "epoch": 4.8026084899902345e-05,
      "model_forward_time": 1.7297494411468506,
      "step": 50359
    },
    {
      "epoch": 4.8026084899902345e-05,
      "step": 50359,
      "training_step_time": 8.61033010482788
    },
    {
      "epoch": 4.802703857421875e-05,
      "grad_norm": 0.10088959336280823,
      "learning_rate": 5.3537121965575966e-06,
      "loss": 0.0565,
      "step": 50360
    },
    {
      "epoch": 4.802703857421875e-05,
      "model_forward_time": 1.7277796268463135,
      "step": 50360
    },
    {
      "dynamics_loss_avg": 0.0662451408803463,
      "epoch": 4.802703857421875e-05,
      "step": 50360
    },
    {
      "action_loss_avg": 0.0027756038878578694,
      "epoch": 4.802703857421875e-05,
      "step": 50360
    },
    {
      "epoch": 4.802703857421875e-05,
      "step": 50360,
      "training_step_time": 7.656754016876221
    },
    {
      "epoch": 4.8027992248535153e-05,
      "model_forward_time": 1.722196340560913,
      "step": 50361
    },
    {
      "epoch": 4.8027992248535153e-05,
      "step": 50361,
      "training_step_time": 9.650148630142212
    },
    {
      "epoch": 4.8028945922851564e-05,
      "model_forward_time": 1.7296006679534912,
      "step": 50362
    },
    {
      "epoch": 4.8028945922851564e-05,
      "step": 50362,
      "training_step_time": 9.20082402229309
    },
    {
      "epoch": 4.802989959716797e-05,
      "model_forward_time": 1.7294702529907227,
      "step": 50363
    },
    {
      "epoch": 4.802989959716797e-05,
      "step": 50363,
      "training_step_time": 6.73725962638855
    },
    {
      "epoch": 4.803085327148437e-05,
      "model_forward_time": 1.7300043106079102,
      "step": 50364
    },
    {
      "epoch": 4.803085327148437e-05,
      "step": 50364,
      "training_step_time": 6.7676050662994385
    },
    {
      "epoch": 4.803180694580078e-05,
      "model_forward_time": 1.7278697490692139,
      "step": 50365
    },
    {
      "epoch": 4.803180694580078e-05,
      "step": 50365,
      "training_step_time": 6.808845281600952
    },
    {
      "epoch": 4.803276062011719e-05,
      "model_forward_time": 1.7365117073059082,
      "step": 50366
    },
    {
      "epoch": 4.803276062011719e-05,
      "step": 50366,
      "training_step_time": 6.7687458992004395
    },
    {
      "epoch": 4.803371429443359e-05,
      "model_forward_time": 1.7334144115447998,
      "step": 50367
    },
    {
      "epoch": 4.803371429443359e-05,
      "step": 50367,
      "training_step_time": 6.719685792922974
    },
    {
      "epoch": 4.803466796875e-05,
      "model_forward_time": 1.7304778099060059,
      "step": 50368
    },
    {
      "epoch": 4.803466796875e-05,
      "step": 50368,
      "training_step_time": 6.730896711349487
    },
    {
      "epoch": 4.8035621643066406e-05,
      "model_forward_time": 1.7391386032104492,
      "step": 50369
    },
    {
      "epoch": 4.8035621643066406e-05,
      "step": 50369,
      "training_step_time": 6.7660839557647705
    },
    {
      "epoch": 4.803657531738281e-05,
      "grad_norm": 0.1259099841117859,
      "learning_rate": 5.352062850007471e-06,
      "loss": 0.0563,
      "step": 50370
    },
    {
      "epoch": 4.803657531738281e-05,
      "model_forward_time": 1.352994680404663,
      "step": 50370
    },
    {
      "dynamics_loss_avg": 0.051250686123967173,
      "epoch": 4.803657531738281e-05,
      "step": 50370
    },
    {
      "action_loss_avg": 0.0016785956453531981,
      "epoch": 4.803657531738281e-05,
      "step": 50370
    },
    {
      "epoch": 4.803657531738281e-05,
      "step": 50370,
      "training_step_time": 6.72315788269043
    },
    {
      "epoch": 4.803752899169922e-05,
      "model_forward_time": 2.29496431350708,
      "step": 50371
    },
    {
      "epoch": 4.803752899169922e-05,
      "step": 50371,
      "training_step_time": 7.689284801483154
    },
    {
      "epoch": 4.8038482666015625e-05,
      "model_forward_time": 1.7379579544067383,
      "step": 50372
    },
    {
      "epoch": 4.8038482666015625e-05,
      "step": 50372,
      "training_step_time": 7.802842617034912
    },
    {
      "epoch": 4.803943634033203e-05,
      "model_forward_time": 1.733586311340332,
      "step": 50373
    },
    {
      "epoch": 4.803943634033203e-05,
      "step": 50373,
      "training_step_time": 6.766894817352295
    },
    {
      "epoch": 4.804039001464844e-05,
      "model_forward_time": 1.7319703102111816,
      "step": 50374
    },
    {
      "epoch": 4.804039001464844e-05,
      "step": 50374,
      "training_step_time": 6.756126880645752
    },
    {
      "epoch": 4.8041343688964844e-05,
      "model_forward_time": 1.7311327457427979,
      "step": 50375
    },
    {
      "epoch": 4.8041343688964844e-05,
      "step": 50375,
      "training_step_time": 6.775095462799072
    },
    {
      "epoch": 4.804229736328125e-05,
      "model_forward_time": 1.7410640716552734,
      "step": 50376
    },
    {
      "epoch": 4.804229736328125e-05,
      "step": 50376,
      "training_step_time": 6.751654148101807
    },
    {
      "epoch": 4.804325103759766e-05,
      "model_forward_time": 1.7350237369537354,
      "step": 50377
    },
    {
      "epoch": 4.804325103759766e-05,
      "step": 50377,
      "training_step_time": 6.749706745147705
    },
    {
      "epoch": 4.804420471191406e-05,
      "model_forward_time": 1.7319035530090332,
      "step": 50378
    },
    {
      "epoch": 4.804420471191406e-05,
      "step": 50378,
      "training_step_time": 6.739747524261475
    },
    {
      "epoch": 4.804515838623047e-05,
      "model_forward_time": 1.7289330959320068,
      "step": 50379
    },
    {
      "epoch": 4.804515838623047e-05,
      "step": 50379,
      "training_step_time": 6.77833890914917
    },
    {
      "epoch": 4.804611206054688e-05,
      "grad_norm": 0.11489678919315338,
      "learning_rate": 5.350413464956283e-06,
      "loss": 0.0567,
      "step": 50380
    },
    {
      "epoch": 4.804611206054688e-05,
      "model_forward_time": 2.3155884742736816,
      "step": 50380
    },
    {
      "dynamics_loss_avg": 0.06846118289977313,
      "epoch": 4.804611206054688e-05,
      "step": 50380
    },
    {
      "action_loss_avg": 0.003090945922303945,
      "epoch": 4.804611206054688e-05,
      "step": 50380
    },
    {
      "epoch": 4.804611206054688e-05,
      "step": 50380,
      "training_step_time": 7.67098331451416
    },
    {
      "epoch": 4.804706573486328e-05,
      "model_forward_time": 1.7326045036315918,
      "step": 50381
    },
    {
      "epoch": 4.804706573486328e-05,
      "step": 50381,
      "training_step_time": 8.297073125839233
    },
    {
      "epoch": 4.8048019409179685e-05,
      "model_forward_time": 1.7284607887268066,
      "step": 50382
    },
    {
      "epoch": 4.8048019409179685e-05,
      "step": 50382,
      "training_step_time": 7.388598918914795
    },
    {
      "epoch": 4.8048973083496096e-05,
      "model_forward_time": 1.7291455268859863,
      "step": 50383
    },
    {
      "epoch": 4.8048973083496096e-05,
      "step": 50383,
      "training_step_time": 6.7810142040252686
    },
    {
      "epoch": 4.80499267578125e-05,
      "model_forward_time": 1.7280101776123047,
      "step": 50384
    },
    {
      "epoch": 4.80499267578125e-05,
      "step": 50384,
      "training_step_time": 6.730216979980469
    },
    {
      "epoch": 4.8050880432128904e-05,
      "model_forward_time": 1.7366607189178467,
      "step": 50385
    },
    {
      "epoch": 4.8050880432128904e-05,
      "step": 50385,
      "training_step_time": 6.7485761642456055
    },
    {
      "epoch": 4.8051834106445315e-05,
      "model_forward_time": 1.7419116497039795,
      "step": 50386
    },
    {
      "epoch": 4.8051834106445315e-05,
      "step": 50386,
      "training_step_time": 6.790518045425415
    },
    {
      "epoch": 4.805278778076172e-05,
      "model_forward_time": 1.72231125831604,
      "step": 50387
    },
    {
      "epoch": 4.805278778076172e-05,
      "step": 50387,
      "training_step_time": 6.695588111877441
    },
    {
      "epoch": 4.805374145507812e-05,
      "model_forward_time": 1.7302265167236328,
      "step": 50388
    },
    {
      "epoch": 4.805374145507812e-05,
      "step": 50388,
      "training_step_time": 6.802843809127808
    },
    {
      "epoch": 4.8054695129394534e-05,
      "model_forward_time": 2.3133933544158936,
      "step": 50389
    },
    {
      "epoch": 4.8054695129394534e-05,
      "step": 50389,
      "training_step_time": 7.707472324371338
    },
    {
      "epoch": 4.805564880371094e-05,
      "grad_norm": 0.12107191979885101,
      "learning_rate": 5.3487640415844034e-06,
      "loss": 0.0538,
      "step": 50390
    },
    {
      "epoch": 4.805564880371094e-05,
      "model_forward_time": 1.7325553894042969,
      "step": 50390
    },
    {
      "dynamics_loss_avg": 0.061728462018072604,
      "epoch": 4.805564880371094e-05,
      "step": 50390
    },
    {
      "action_loss_avg": 0.002302403427893296,
      "epoch": 4.805564880371094e-05,
      "step": 50390
    },
    {
      "epoch": 4.805564880371094e-05,
      "step": 50390,
      "training_step_time": 9.611209630966187
    },
    {
      "epoch": 4.805660247802734e-05,
      "model_forward_time": 1.7298707962036133,
      "step": 50391
    },
    {
      "epoch": 4.805660247802734e-05,
      "step": 50391,
      "training_step_time": 7.66012978553772
    },
    {
      "epoch": 4.805755615234375e-05,
      "model_forward_time": 1.7342510223388672,
      "step": 50392
    },
    {
      "epoch": 4.805755615234375e-05,
      "step": 50392,
      "training_step_time": 6.744811296463013
    },
    {
      "epoch": 4.805850982666016e-05,
      "model_forward_time": 1.728205919265747,
      "step": 50393
    },
    {
      "epoch": 4.805850982666016e-05,
      "step": 50393,
      "training_step_time": 6.726245164871216
    },
    {
      "epoch": 4.805946350097656e-05,
      "model_forward_time": 1.734117031097412,
      "step": 50394
    },
    {
      "epoch": 4.805946350097656e-05,
      "step": 50394,
      "training_step_time": 6.741374254226685
    },
    {
      "epoch": 4.806041717529297e-05,
      "model_forward_time": 1.7434990406036377,
      "step": 50395
    },
    {
      "epoch": 4.806041717529297e-05,
      "step": 50395,
      "training_step_time": 6.760684490203857
    },
    {
      "epoch": 4.8061370849609376e-05,
      "model_forward_time": 1.7432594299316406,
      "step": 50396
    },
    {
      "epoch": 4.8061370849609376e-05,
      "step": 50396,
      "training_step_time": 6.728022813796997
    },
    {
      "epoch": 4.806232452392578e-05,
      "model_forward_time": 1.7376468181610107,
      "step": 50397
    },
    {
      "epoch": 4.806232452392578e-05,
      "step": 50397,
      "training_step_time": 6.794600486755371
    },
    {
      "epoch": 4.806327819824219e-05,
      "model_forward_time": 2.3551623821258545,
      "step": 50398
    },
    {
      "epoch": 4.806327819824219e-05,
      "step": 50398,
      "training_step_time": 7.773321628570557
    },
    {
      "epoch": 4.8064231872558595e-05,
      "model_forward_time": 1.7446002960205078,
      "step": 50399
    },
    {
      "epoch": 4.8064231872558595e-05,
      "step": 50399,
      "training_step_time": 7.841921806335449
    },
    {
      "epoch": 4.8065185546875e-05,
      "grad_norm": 0.11702201515436172,
      "learning_rate": 5.347114580072212e-06,
      "loss": 0.0575,
      "step": 50400
    },
    {
      "epoch": 4.8065185546875e-05,
      "model_forward_time": 1.7279882431030273,
      "step": 50400
    },
    {
      "dynamics_loss_avg": 0.0848271882161498,
      "epoch": 4.8065185546875e-05,
      "step": 50400
    },
    {
      "action_loss_avg": 0.002101456298260018,
      "epoch": 4.8065185546875e-05,
      "step": 50400
    },
    {
      "epoch": 4.8065185546875e-05,
      "step": 50400,
      "training_step_time": 7.712721347808838
    },
    {
      "epoch": 4.806613922119141e-05,
      "model_forward_time": 1.7490060329437256,
      "step": 50401
    },
    {
      "epoch": 4.806613922119141e-05,
      "step": 50401,
      "training_step_time": 6.729906797409058
    },
    {
      "epoch": 4.8067092895507813e-05,
      "model_forward_time": 1.7365832328796387,
      "step": 50402
    },
    {
      "epoch": 4.8067092895507813e-05,
      "step": 50402,
      "training_step_time": 6.762405633926392
    },
    {
      "epoch": 4.806804656982422e-05,
      "model_forward_time": 1.739537000656128,
      "step": 50403
    },
    {
      "epoch": 4.806804656982422e-05,
      "step": 50403,
      "training_step_time": 6.753549098968506
    },
    {
      "epoch": 4.806900024414063e-05,
      "model_forward_time": 1.7406857013702393,
      "step": 50404
    },
    {
      "epoch": 4.806900024414063e-05,
      "step": 50404,
      "training_step_time": 6.758094072341919
    },
    {
      "epoch": 4.806995391845703e-05,
      "model_forward_time": 1.7479939460754395,
      "step": 50405
    },
    {
      "epoch": 4.806995391845703e-05,
      "step": 50405,
      "training_step_time": 6.781971454620361
    },
    {
      "epoch": 4.8070907592773436e-05,
      "model_forward_time": 1.7395033836364746,
      "step": 50406
    },
    {
      "epoch": 4.8070907592773436e-05,
      "step": 50406,
      "training_step_time": 6.777679443359375
    },
    {
      "epoch": 4.807186126708984e-05,
      "model_forward_time": 1.7988581657409668,
      "step": 50407
    },
    {
      "epoch": 4.807186126708984e-05,
      "step": 50407,
      "training_step_time": 8.262352228164673
    },
    {
      "epoch": 4.807281494140625e-05,
      "model_forward_time": 1.7418386936187744,
      "step": 50408
    },
    {
      "epoch": 4.807281494140625e-05,
      "step": 50408,
      "training_step_time": 7.72410249710083
    },
    {
      "epoch": 4.8073768615722655e-05,
      "model_forward_time": 1.748471975326538,
      "step": 50409
    },
    {
      "epoch": 4.8073768615722655e-05,
      "step": 50409,
      "training_step_time": 8.424269437789917
    },
    {
      "epoch": 4.807472229003906e-05,
      "grad_norm": 0.08854496479034424,
      "learning_rate": 5.345465080600093e-06,
      "loss": 0.0563,
      "step": 50410
    },
    {
      "epoch": 4.807472229003906e-05,
      "model_forward_time": 1.736548662185669,
      "step": 50410
    },
    {
      "dynamics_loss_avg": 0.05700323283672333,
      "epoch": 4.807472229003906e-05,
      "step": 50410
    },
    {
      "action_loss_avg": 0.0018513027869630605,
      "epoch": 4.807472229003906e-05,
      "step": 50410
    },
    {
      "epoch": 4.807472229003906e-05,
      "step": 50410,
      "training_step_time": 6.6877758502960205
    },
    {
      "epoch": 4.807567596435547e-05,
      "model_forward_time": 1.7406668663024902,
      "step": 50411
    },
    {
      "epoch": 4.807567596435547e-05,
      "step": 50411,
      "training_step_time": 6.699608325958252
    },
    {
      "epoch": 4.8076629638671874e-05,
      "model_forward_time": 1.745079755783081,
      "step": 50412
    },
    {
      "epoch": 4.8076629638671874e-05,
      "step": 50412,
      "training_step_time": 6.679228782653809
    },
    {
      "epoch": 4.807758331298828e-05,
      "model_forward_time": 1.7437994480133057,
      "step": 50413
    },
    {
      "epoch": 4.807758331298828e-05,
      "step": 50413,
      "training_step_time": 6.695040464401245
    },
    {
      "epoch": 4.807853698730469e-05,
      "model_forward_time": 1.7417175769805908,
      "step": 50414
    },
    {
      "epoch": 4.807853698730469e-05,
      "step": 50414,
      "training_step_time": 6.702211856842041
    },
    {
      "epoch": 4.807949066162109e-05,
      "model_forward_time": 1.7518551349639893,
      "step": 50415
    },
    {
      "epoch": 4.807949066162109e-05,
      "step": 50415,
      "training_step_time": 6.686102867126465
    },
    {
      "epoch": 4.80804443359375e-05,
      "model_forward_time": 1.3913793563842773,
      "step": 50416
    },
    {
      "epoch": 4.80804443359375e-05,
      "step": 50416,
      "training_step_time": 7.653470993041992
    },
    {
      "epoch": 4.808139801025391e-05,
      "model_forward_time": 2.286367177963257,
      "step": 50417
    },
    {
      "epoch": 4.808139801025391e-05,
      "step": 50417,
      "training_step_time": 7.666024446487427
    },
    {
      "epoch": 4.808235168457031e-05,
      "model_forward_time": 1.7403254508972168,
      "step": 50418
    },
    {
      "epoch": 4.808235168457031e-05,
      "step": 50418,
      "training_step_time": 9.720019102096558
    },
    {
      "epoch": 4.8083305358886716e-05,
      "model_forward_time": 1.7383179664611816,
      "step": 50419
    },
    {
      "epoch": 4.8083305358886716e-05,
      "step": 50419,
      "training_step_time": 7.6211159229278564
    },
    {
      "epoch": 4.808425903320313e-05,
      "grad_norm": 0.12061280012130737,
      "learning_rate": 5.343815543348433e-06,
      "loss": 0.0587,
      "step": 50420
    },
    {
      "epoch": 4.808425903320313e-05,
      "model_forward_time": 1.7331860065460205,
      "step": 50420
    },
    {
      "dynamics_loss_avg": 0.061175462789833546,
      "epoch": 4.808425903320313e-05,
      "step": 50420
    },
    {
      "action_loss_avg": 0.002402378572151065,
      "epoch": 4.808425903320313e-05,
      "step": 50420
    },
    {
      "epoch": 4.808425903320313e-05,
      "step": 50420,
      "training_step_time": 6.758850336074829
    },
    {
      "epoch": 4.808521270751953e-05,
      "model_forward_time": 1.7518608570098877,
      "step": 50421
    },
    {
      "epoch": 4.808521270751953e-05,
      "step": 50421,
      "training_step_time": 6.71924352645874
    },
    {
      "epoch": 4.8086166381835935e-05,
      "model_forward_time": 1.7459936141967773,
      "step": 50422
    },
    {
      "epoch": 4.8086166381835935e-05,
      "step": 50422,
      "training_step_time": 6.753946781158447
    },
    {
      "epoch": 4.8087120056152345e-05,
      "model_forward_time": 1.7355928421020508,
      "step": 50423
    },
    {
      "epoch": 4.8087120056152345e-05,
      "step": 50423,
      "training_step_time": 6.7209086418151855
    },
    {
      "epoch": 4.808807373046875e-05,
      "model_forward_time": 1.7392942905426025,
      "step": 50424
    },
    {
      "epoch": 4.808807373046875e-05,
      "step": 50424,
      "training_step_time": 6.7441747188568115
    },
    {
      "epoch": 4.8089027404785154e-05,
      "model_forward_time": 1.769545316696167,
      "step": 50425
    },
    {
      "epoch": 4.8089027404785154e-05,
      "step": 50425,
      "training_step_time": 7.699913263320923
    },
    {
      "epoch": 4.8089981079101564e-05,
      "model_forward_time": 2.3042004108428955,
      "step": 50426
    },
    {
      "epoch": 4.8089981079101564e-05,
      "step": 50426,
      "training_step_time": 7.598422050476074
    },
    {
      "epoch": 4.809093475341797e-05,
      "model_forward_time": 1.7444157600402832,
      "step": 50427
    },
    {
      "epoch": 4.809093475341797e-05,
      "step": 50427,
      "training_step_time": 8.858471155166626
    },
    {
      "epoch": 4.809188842773437e-05,
      "model_forward_time": 1.7347338199615479,
      "step": 50428
    },
    {
      "epoch": 4.809188842773437e-05,
      "step": 50428,
      "training_step_time": 7.686547040939331
    },
    {
      "epoch": 4.809284210205078e-05,
      "model_forward_time": 1.7476294040679932,
      "step": 50429
    },
    {
      "epoch": 4.809284210205078e-05,
      "step": 50429,
      "training_step_time": 6.770037412643433
    },
    {
      "epoch": 4.809379577636719e-05,
      "grad_norm": 0.22140620648860931,
      "learning_rate": 5.3421659684976205e-06,
      "loss": 0.0591,
      "step": 50430
    },
    {
      "epoch": 4.809379577636719e-05,
      "model_forward_time": 1.7250018119812012,
      "step": 50430
    },
    {
      "dynamics_loss_avg": 0.06852155700325965,
      "epoch": 4.809379577636719e-05,
      "step": 50430
    },
    {
      "action_loss_avg": 0.0017739023664034904,
      "epoch": 4.809379577636719e-05,
      "step": 50430
    },
    {
      "epoch": 4.809379577636719e-05,
      "step": 50430,
      "training_step_time": 6.729223012924194
    },
    {
      "epoch": 4.809474945068359e-05,
      "model_forward_time": 1.7405836582183838,
      "step": 50431
    },
    {
      "epoch": 4.809474945068359e-05,
      "step": 50431,
      "training_step_time": 6.672675848007202
    },
    {
      "epoch": 4.8095703125e-05,
      "model_forward_time": 1.7402067184448242,
      "step": 50432
    },
    {
      "epoch": 4.8095703125e-05,
      "step": 50432,
      "training_step_time": 6.761648416519165
    },
    {
      "epoch": 4.8096656799316406e-05,
      "model_forward_time": 1.7381420135498047,
      "step": 50433
    },
    {
      "epoch": 4.8096656799316406e-05,
      "step": 50433,
      "training_step_time": 6.80200457572937
    },
    {
      "epoch": 4.809761047363281e-05,
      "model_forward_time": 1.7526464462280273,
      "step": 50434
    },
    {
      "epoch": 4.809761047363281e-05,
      "step": 50434,
      "training_step_time": 7.710980653762817
    },
    {
      "epoch": 4.809856414794922e-05,
      "model_forward_time": 2.307771921157837,
      "step": 50435
    },
    {
      "epoch": 4.809856414794922e-05,
      "step": 50435,
      "training_step_time": 7.646025896072388
    },
    {
      "epoch": 4.8099517822265625e-05,
      "model_forward_time": 1.7354135513305664,
      "step": 50436
    },
    {
      "epoch": 4.8099517822265625e-05,
      "step": 50436,
      "training_step_time": 10.6307954788208
    },
    {
      "epoch": 4.810047149658203e-05,
      "model_forward_time": 1.749739646911621,
      "step": 50437
    },
    {
      "epoch": 4.810047149658203e-05,
      "step": 50437,
      "training_step_time": 7.801685333251953
    },
    {
      "epoch": 4.810142517089844e-05,
      "model_forward_time": 1.7445614337921143,
      "step": 50438
    },
    {
      "epoch": 4.810142517089844e-05,
      "step": 50438,
      "training_step_time": 6.869664430618286
    },
    {
      "epoch": 4.8102378845214844e-05,
      "model_forward_time": 1.7372705936431885,
      "step": 50439
    },
    {
      "epoch": 4.8102378845214844e-05,
      "step": 50439,
      "training_step_time": 6.879077911376953
    },
    {
      "epoch": 4.810333251953125e-05,
      "grad_norm": 0.12750859558582306,
      "learning_rate": 5.340516356228053e-06,
      "loss": 0.0608,
      "step": 50440
    },
    {
      "epoch": 4.810333251953125e-05,
      "model_forward_time": 1.7404203414916992,
      "step": 50440
    },
    {
      "dynamics_loss_avg": 0.060608938336372375,
      "epoch": 4.810333251953125e-05,
      "step": 50440
    },
    {
      "action_loss_avg": 0.0014556699083186687,
      "epoch": 4.810333251953125e-05,
      "step": 50440
    },
    {
      "epoch": 4.810333251953125e-05,
      "step": 50440,
      "training_step_time": 6.896066427230835
    },
    {
      "epoch": 4.810428619384766e-05,
      "model_forward_time": 1.7396457195281982,
      "step": 50441
    },
    {
      "epoch": 4.810428619384766e-05,
      "step": 50441,
      "training_step_time": 6.869482040405273
    },
    {
      "epoch": 4.810523986816406e-05,
      "model_forward_time": 1.7417049407958984,
      "step": 50442
    },
    {
      "epoch": 4.810523986816406e-05,
      "step": 50442,
      "training_step_time": 6.9101598262786865
    },
    {
      "epoch": 4.810619354248047e-05,
      "model_forward_time": 1.729750394821167,
      "step": 50443
    },
    {
      "epoch": 4.810619354248047e-05,
      "step": 50443,
      "training_step_time": 7.574547529220581
    },
    {
      "epoch": 4.810714721679688e-05,
      "model_forward_time": 2.3325111865997314,
      "step": 50444
    },
    {
      "epoch": 4.810714721679688e-05,
      "step": 50444,
      "training_step_time": 7.843871831893921
    },
    {
      "epoch": 4.810810089111328e-05,
      "model_forward_time": 1.7414052486419678,
      "step": 50445
    },
    {
      "epoch": 4.810810089111328e-05,
      "step": 50445,
      "training_step_time": 9.814012050628662
    },
    {
      "epoch": 4.8109054565429686e-05,
      "model_forward_time": 1.7502398490905762,
      "step": 50446
    },
    {
      "epoch": 4.8109054565429686e-05,
      "step": 50446,
      "training_step_time": 8.332911729812622
    },
    {
      "epoch": 4.8110008239746096e-05,
      "model_forward_time": 1.735595941543579,
      "step": 50447
    },
    {
      "epoch": 4.8110008239746096e-05,
      "step": 50447,
      "training_step_time": 7.47386622428894
    },
    {
      "epoch": 4.81109619140625e-05,
      "model_forward_time": 1.7356162071228027,
      "step": 50448
    },
    {
      "epoch": 4.81109619140625e-05,
      "step": 50448,
      "training_step_time": 6.871553659439087
    },
    {
      "epoch": 4.8111915588378904e-05,
      "model_forward_time": 1.7454800605773926,
      "step": 50449
    },
    {
      "epoch": 4.8111915588378904e-05,
      "step": 50449,
      "training_step_time": 6.86512303352356
    },
    {
      "epoch": 4.8112869262695315e-05,
      "grad_norm": 0.14308896660804749,
      "learning_rate": 5.338866706720128e-06,
      "loss": 0.0607,
      "step": 50450
    },
    {
      "epoch": 4.8112869262695315e-05,
      "model_forward_time": 1.7329216003417969,
      "step": 50450
    },
    {
      "dynamics_loss_avg": 0.0720956452190876,
      "epoch": 4.8112869262695315e-05,
      "step": 50450
    },
    {
      "action_loss_avg": 0.0022438357991632074,
      "epoch": 4.8112869262695315e-05,
      "step": 50450
    },
    {
      "epoch": 4.8112869262695315e-05,
      "step": 50450,
      "training_step_time": 6.867801904678345
    },
    {
      "epoch": 4.811382293701172e-05,
      "model_forward_time": 1.7346532344818115,
      "step": 50451
    },
    {
      "epoch": 4.811382293701172e-05,
      "step": 50451,
      "training_step_time": 6.856544733047485
    },
    {
      "epoch": 4.811477661132812e-05,
      "model_forward_time": 1.7416810989379883,
      "step": 50452
    },
    {
      "epoch": 4.811477661132812e-05,
      "step": 50452,
      "training_step_time": 7.486641883850098
    },
    {
      "epoch": 4.8115730285644534e-05,
      "model_forward_time": 1.781686544418335,
      "step": 50453
    },
    {
      "epoch": 4.8115730285644534e-05,
      "step": 50453,
      "training_step_time": 8.309738874435425
    },
    {
      "epoch": 4.811668395996094e-05,
      "model_forward_time": 1.7328681945800781,
      "step": 50454
    },
    {
      "epoch": 4.811668395996094e-05,
      "step": 50454,
      "training_step_time": 9.508794069290161
    },
    {
      "epoch": 4.811763763427734e-05,
      "model_forward_time": 1.7432641983032227,
      "step": 50455
    },
    {
      "epoch": 4.811763763427734e-05,
      "step": 50455,
      "training_step_time": 8.518606662750244
    },
    {
      "epoch": 4.811859130859375e-05,
      "model_forward_time": 1.743288278579712,
      "step": 50456
    },
    {
      "epoch": 4.811859130859375e-05,
      "step": 50456,
      "training_step_time": 7.692563772201538
    },
    {
      "epoch": 4.811954498291016e-05,
      "model_forward_time": 1.7360033988952637,
      "step": 50457
    },
    {
      "epoch": 4.811954498291016e-05,
      "step": 50457,
      "training_step_time": 6.829268932342529
    },
    {
      "epoch": 4.812049865722656e-05,
      "model_forward_time": 1.739011287689209,
      "step": 50458
    },
    {
      "epoch": 4.812049865722656e-05,
      "step": 50458,
      "training_step_time": 6.8408074378967285
    },
    {
      "epoch": 4.812145233154297e-05,
      "model_forward_time": 1.7486984729766846,
      "step": 50459
    },
    {
      "epoch": 4.812145233154297e-05,
      "step": 50459,
      "training_step_time": 6.906003475189209
    },
    {
      "epoch": 4.8122406005859376e-05,
      "grad_norm": 0.09816519170999527,
      "learning_rate": 5.33721702015425e-06,
      "loss": 0.0591,
      "step": 50460
    },
    {
      "epoch": 4.8122406005859376e-05,
      "model_forward_time": 1.7262511253356934,
      "step": 50460
    },
    {
      "dynamics_loss_avg": 0.0582484757527709,
      "epoch": 4.8122406005859376e-05,
      "step": 50460
    },
    {
      "action_loss_avg": 0.00157224852591753,
      "epoch": 4.8122406005859376e-05,
      "step": 50460
    },
    {
      "epoch": 4.8122406005859376e-05,
      "step": 50460,
      "training_step_time": 6.840117692947388
    },
    {
      "epoch": 4.812335968017578e-05,
      "model_forward_time": 1.742021083831787,
      "step": 50461
    },
    {
      "epoch": 4.812335968017578e-05,
      "step": 50461,
      "training_step_time": 7.552758693695068
    },
    {
      "epoch": 4.812431335449219e-05,
      "model_forward_time": 1.766371250152588,
      "step": 50462
    },
    {
      "epoch": 4.812431335449219e-05,
      "step": 50462,
      "training_step_time": 7.878262758255005
    },
    {
      "epoch": 4.8125267028808595e-05,
      "model_forward_time": 2.311563014984131,
      "step": 50463
    },
    {
      "epoch": 4.8125267028808595e-05,
      "step": 50463,
      "training_step_time": 8.716969728469849
    },
    {
      "epoch": 4.8126220703125e-05,
      "model_forward_time": 1.743100881576538,
      "step": 50464
    },
    {
      "epoch": 4.8126220703125e-05,
      "step": 50464,
      "training_step_time": 8.601962804794312
    },
    {
      "epoch": 4.812717437744141e-05,
      "model_forward_time": 1.7497451305389404,
      "step": 50465
    },
    {
      "epoch": 4.812717437744141e-05,
      "step": 50465,
      "training_step_time": 7.832425117492676
    },
    {
      "epoch": 4.8128128051757814e-05,
      "model_forward_time": 1.7158594131469727,
      "step": 50466
    },
    {
      "epoch": 4.8128128051757814e-05,
      "step": 50466,
      "training_step_time": 6.95002293586731
    },
    {
      "epoch": 4.812908172607422e-05,
      "model_forward_time": 1.7311673164367676,
      "step": 50467
    },
    {
      "epoch": 4.812908172607422e-05,
      "step": 50467,
      "training_step_time": 6.979172706604004
    },
    {
      "epoch": 4.813003540039063e-05,
      "model_forward_time": 1.7279810905456543,
      "step": 50468
    },
    {
      "epoch": 4.813003540039063e-05,
      "step": 50468,
      "training_step_time": 6.9746012687683105
    },
    {
      "epoch": 4.813098907470703e-05,
      "model_forward_time": 1.7334108352661133,
      "step": 50469
    },
    {
      "epoch": 4.813098907470703e-05,
      "step": 50469,
      "training_step_time": 6.939753770828247
    },
    {
      "epoch": 4.8131942749023436e-05,
      "grad_norm": 0.15551623702049255,
      "learning_rate": 5.335567296710825e-06,
      "loss": 0.0607,
      "step": 50470
    },
    {
      "epoch": 4.8131942749023436e-05,
      "model_forward_time": 1.7266860008239746,
      "step": 50470
    },
    {
      "dynamics_loss_avg": 0.05912880357354879,
      "epoch": 4.8131942749023436e-05,
      "step": 50470
    },
    {
      "action_loss_avg": 0.001834081142442301,
      "epoch": 4.8131942749023436e-05,
      "step": 50470
    },
    {
      "epoch": 4.8131942749023436e-05,
      "step": 50470,
      "training_step_time": 6.987388372421265
    },
    {
      "epoch": 4.813289642333984e-05,
      "model_forward_time": 1.734945297241211,
      "step": 50471
    },
    {
      "epoch": 4.813289642333984e-05,
      "step": 50471,
      "training_step_time": 7.946184158325195
    },
    {
      "epoch": 4.813385009765625e-05,
      "model_forward_time": 2.2770752906799316,
      "step": 50472
    },
    {
      "epoch": 4.813385009765625e-05,
      "step": 50472,
      "training_step_time": 7.979320287704468
    },
    {
      "epoch": 4.8134803771972655e-05,
      "model_forward_time": 1.7395431995391846,
      "step": 50473
    },
    {
      "epoch": 4.8134803771972655e-05,
      "step": 50473,
      "training_step_time": 12.331958293914795
    },
    {
      "epoch": 4.813575744628906e-05,
      "model_forward_time": 1.7353463172912598,
      "step": 50474
    },
    {
      "epoch": 4.813575744628906e-05,
      "step": 50474,
      "training_step_time": 8.750155210494995
    },
    {
      "epoch": 4.813671112060547e-05,
      "model_forward_time": 1.7302794456481934,
      "step": 50475
    },
    {
      "epoch": 4.813671112060547e-05,
      "step": 50475,
      "training_step_time": 7.205657005310059
    },
    {
      "epoch": 4.8137664794921874e-05,
      "model_forward_time": 1.3590478897094727,
      "step": 50476
    },
    {
      "epoch": 4.8137664794921874e-05,
      "step": 50476,
      "training_step_time": 6.956920385360718
    },
    {
      "epoch": 4.813861846923828e-05,
      "model_forward_time": 1.3663511276245117,
      "step": 50477
    },
    {
      "epoch": 4.813861846923828e-05,
      "step": 50477,
      "training_step_time": 6.820450067520142
    },
    {
      "epoch": 4.813957214355469e-05,
      "model_forward_time": 1.733055591583252,
      "step": 50478
    },
    {
      "epoch": 4.813957214355469e-05,
      "step": 50478,
      "training_step_time": 6.885097026824951
    },
    {
      "epoch": 4.814052581787109e-05,
      "model_forward_time": 1.7289278507232666,
      "step": 50479
    },
    {
      "epoch": 4.814052581787109e-05,
      "step": 50479,
      "training_step_time": 6.9235005378723145
    },
    {
      "epoch": 4.81414794921875e-05,
      "grad_norm": 0.11886120587587357,
      "learning_rate": 5.333917536570266e-06,
      "loss": 0.059,
      "step": 50480
    },
    {
      "epoch": 4.81414794921875e-05,
      "model_forward_time": 1.7216053009033203,
      "step": 50480
    },
    {
      "dynamics_loss_avg": 0.049055477418005464,
      "epoch": 4.81414794921875e-05,
      "step": 50480
    },
    {
      "action_loss_avg": 0.0014944207505322993,
      "epoch": 4.81414794921875e-05,
      "step": 50480
    },
    {
      "epoch": 4.81414794921875e-05,
      "step": 50480,
      "training_step_time": 7.923010587692261
    },
    {
      "epoch": 4.814243316650391e-05,
      "model_forward_time": 2.3295774459838867,
      "step": 50481
    },
    {
      "epoch": 4.814243316650391e-05,
      "step": 50481,
      "training_step_time": 7.912973642349243
    },
    {
      "epoch": 4.814338684082031e-05,
      "model_forward_time": 1.7161836624145508,
      "step": 50482
    },
    {
      "epoch": 4.814338684082031e-05,
      "step": 50482,
      "training_step_time": 8.953316450119019
    },
    {
      "epoch": 4.8144340515136716e-05,
      "model_forward_time": 1.7368602752685547,
      "step": 50483
    },
    {
      "epoch": 4.8144340515136716e-05,
      "step": 50483,
      "training_step_time": 9.66852355003357
    },
    {
      "epoch": 4.814529418945313e-05,
      "model_forward_time": 1.7300844192504883,
      "step": 50484
    },
    {
      "epoch": 4.814529418945313e-05,
      "step": 50484,
      "training_step_time": 7.5803306102752686
    },
    {
      "epoch": 4.814624786376953e-05,
      "model_forward_time": 1.7357714176177979,
      "step": 50485
    },
    {
      "epoch": 4.814624786376953e-05,
      "step": 50485,
      "training_step_time": 6.930642127990723
    },
    {
      "epoch": 4.8147201538085935e-05,
      "model_forward_time": 1.7428603172302246,
      "step": 50486
    },
    {
      "epoch": 4.8147201538085935e-05,
      "step": 50486,
      "training_step_time": 6.917224407196045
    },
    {
      "epoch": 4.8148155212402346e-05,
      "model_forward_time": 1.713517189025879,
      "step": 50487
    },
    {
      "epoch": 4.8148155212402346e-05,
      "step": 50487,
      "training_step_time": 6.845318078994751
    },
    {
      "epoch": 4.814910888671875e-05,
      "model_forward_time": 1.7387197017669678,
      "step": 50488
    },
    {
      "epoch": 4.814910888671875e-05,
      "step": 50488,
      "training_step_time": 6.929987668991089
    },
    {
      "epoch": 4.8150062561035154e-05,
      "model_forward_time": 1.7358736991882324,
      "step": 50489
    },
    {
      "epoch": 4.8150062561035154e-05,
      "step": 50489,
      "training_step_time": 7.892821550369263
    },
    {
      "epoch": 4.8151016235351564e-05,
      "grad_norm": 0.13651011884212494,
      "learning_rate": 5.332267739912986e-06,
      "loss": 0.0578,
      "step": 50490
    },
    {
      "epoch": 4.8151016235351564e-05,
      "model_forward_time": 2.341416120529175,
      "step": 50490
    },
    {
      "dynamics_loss_avg": 0.06962072793394328,
      "epoch": 4.8151016235351564e-05,
      "step": 50490
    },
    {
      "action_loss_avg": 0.002675941656343639,
      "epoch": 4.8151016235351564e-05,
      "step": 50490
    },
    {
      "epoch": 4.8151016235351564e-05,
      "step": 50490,
      "training_step_time": 8.473074913024902
    },
    {
      "epoch": 4.815196990966797e-05,
      "model_forward_time": 1.7352774143218994,
      "step": 50491
    },
    {
      "epoch": 4.815196990966797e-05,
      "step": 50491,
      "training_step_time": 10.963285446166992
    },
    {
      "epoch": 4.815292358398437e-05,
      "model_forward_time": 1.7500739097595215,
      "step": 50492
    },
    {
      "epoch": 4.815292358398437e-05,
      "step": 50492,
      "training_step_time": 7.932204008102417
    },
    {
      "epoch": 4.815387725830078e-05,
      "model_forward_time": 1.7250938415527344,
      "step": 50493
    },
    {
      "epoch": 4.815387725830078e-05,
      "step": 50493,
      "training_step_time": 7.858097076416016
    },
    {
      "epoch": 4.815483093261719e-05,
      "model_forward_time": 1.7383134365081787,
      "step": 50494
    },
    {
      "epoch": 4.815483093261719e-05,
      "step": 50494,
      "training_step_time": 6.952550411224365
    },
    {
      "epoch": 4.815578460693359e-05,
      "model_forward_time": 1.358661413192749,
      "step": 50495
    },
    {
      "epoch": 4.815578460693359e-05,
      "step": 50495,
      "training_step_time": 6.991123199462891
    },
    {
      "epoch": 4.815673828125e-05,
      "model_forward_time": 1.7296936511993408,
      "step": 50496
    },
    {
      "epoch": 4.815673828125e-05,
      "step": 50496,
      "training_step_time": 6.9836015701293945
    },
    {
      "epoch": 4.8157691955566406e-05,
      "model_forward_time": 1.7312161922454834,
      "step": 50497
    },
    {
      "epoch": 4.8157691955566406e-05,
      "step": 50497,
      "training_step_time": 6.9705188274383545
    },
    {
      "epoch": 4.815864562988281e-05,
      "model_forward_time": 1.730327844619751,
      "step": 50498
    },
    {
      "epoch": 4.815864562988281e-05,
      "step": 50498,
      "training_step_time": 7.8497748374938965
    },
    {
      "epoch": 4.815959930419922e-05,
      "model_forward_time": 1.789926290512085,
      "step": 50499
    },
    {
      "epoch": 4.815959930419922e-05,
      "step": 50499,
      "training_step_time": 8.553033113479614
    },
    {
      "epoch": 4.8160552978515625e-05,
      "grad_norm": 0.13185574114322662,
      "learning_rate": 5.330617906919405e-06,
      "loss": 0.0587,
      "step": 50500
    },
    {
      "epoch": 4.8160552978515625e-05,
      "model_forward_time": 1.712120771408081,
      "step": 50500
    },
    {
      "dynamics_loss_avg": 0.06998645905405283,
      "epoch": 4.8160552978515625e-05,
      "step": 50500
    },
    {
      "action_loss_avg": 0.002876809280132875,
      "epoch": 4.8160552978515625e-05,
      "step": 50500
    },
    {
      "epoch": 4.8160552978515625e-05,
      "step": 50500,
      "training_step_time": 8.570487976074219
    },
    {
      "epoch": 4.816150665283203e-05,
      "model_forward_time": 1.7312355041503906,
      "step": 50501
    },
    {
      "epoch": 4.816150665283203e-05,
      "step": 50501,
      "training_step_time": 7.96718430519104
    },
    {
      "epoch": 4.816246032714844e-05,
      "model_forward_time": 1.7275991439819336,
      "step": 50502
    },
    {
      "epoch": 4.816246032714844e-05,
      "step": 50502,
      "training_step_time": 7.81439995765686
    },
    {
      "epoch": 4.8163414001464844e-05,
      "model_forward_time": 1.728609561920166,
      "step": 50503
    },
    {
      "epoch": 4.8163414001464844e-05,
      "step": 50503,
      "training_step_time": 6.927867650985718
    },
    {
      "epoch": 4.816436767578125e-05,
      "model_forward_time": 1.7328565120697021,
      "step": 50504
    },
    {
      "epoch": 4.816436767578125e-05,
      "step": 50504,
      "training_step_time": 6.929872989654541
    },
    {
      "epoch": 4.816532135009766e-05,
      "model_forward_time": 1.7318599224090576,
      "step": 50505
    },
    {
      "epoch": 4.816532135009766e-05,
      "step": 50505,
      "training_step_time": 6.9271509647369385
    },
    {
      "epoch": 4.816627502441406e-05,
      "model_forward_time": 1.7249715328216553,
      "step": 50506
    },
    {
      "epoch": 4.816627502441406e-05,
      "step": 50506,
      "training_step_time": 6.911879301071167
    },
    {
      "epoch": 4.816722869873047e-05,
      "model_forward_time": 1.7281334400177002,
      "step": 50507
    },
    {
      "epoch": 4.816722869873047e-05,
      "step": 50507,
      "training_step_time": 7.625702619552612
    },
    {
      "epoch": 4.816818237304688e-05,
      "model_forward_time": 1.779970407485962,
      "step": 50508
    },
    {
      "epoch": 4.816818237304688e-05,
      "step": 50508,
      "training_step_time": 7.9224536418914795
    },
    {
      "epoch": 4.816913604736328e-05,
      "model_forward_time": 2.2863998413085938,
      "step": 50509
    },
    {
      "epoch": 4.816913604736328e-05,
      "step": 50509,
      "training_step_time": 9.500088453292847
    },
    {
      "epoch": 4.8170089721679686e-05,
      "grad_norm": 0.11076857894659042,
      "learning_rate": 5.32896803776995e-06,
      "loss": 0.0619,
      "step": 50510
    },
    {
      "epoch": 4.8170089721679686e-05,
      "model_forward_time": 1.7179014682769775,
      "step": 50510
    },
    {
      "dynamics_loss_avg": 0.05606624912470579,
      "epoch": 4.8170089721679686e-05,
      "step": 50510
    },
    {
      "action_loss_avg": 0.0021778650290798394,
      "epoch": 4.8170089721679686e-05,
      "step": 50510
    },
    {
      "epoch": 4.8170089721679686e-05,
      "step": 50510,
      "training_step_time": 9.001461744308472
    },
    {
      "epoch": 4.8171043395996096e-05,
      "model_forward_time": 1.7228937149047852,
      "step": 50511
    },
    {
      "epoch": 4.8171043395996096e-05,
      "step": 50511,
      "training_step_time": 7.8538854122161865
    },
    {
      "epoch": 4.81719970703125e-05,
      "model_forward_time": 1.7242810726165771,
      "step": 50512
    },
    {
      "epoch": 4.81719970703125e-05,
      "step": 50512,
      "training_step_time": 7.58403754234314
    },
    {
      "epoch": 4.8172950744628904e-05,
      "model_forward_time": 1.7293779850006104,
      "step": 50513
    },
    {
      "epoch": 4.8172950744628904e-05,
      "step": 50513,
      "training_step_time": 6.925512790679932
    },
    {
      "epoch": 4.8173904418945315e-05,
      "model_forward_time": 1.7257859706878662,
      "step": 50514
    },
    {
      "epoch": 4.8173904418945315e-05,
      "step": 50514,
      "training_step_time": 6.955101728439331
    },
    {
      "epoch": 4.817485809326172e-05,
      "model_forward_time": 1.7247517108917236,
      "step": 50515
    },
    {
      "epoch": 4.817485809326172e-05,
      "step": 50515,
      "training_step_time": 6.940991640090942
    },
    {
      "epoch": 4.817581176757812e-05,
      "model_forward_time": 1.7250113487243652,
      "step": 50516
    },
    {
      "epoch": 4.817581176757812e-05,
      "step": 50516,
      "training_step_time": 7.581699371337891
    },
    {
      "epoch": 4.8176765441894534e-05,
      "model_forward_time": 1.7579753398895264,
      "step": 50517
    },
    {
      "epoch": 4.8176765441894534e-05,
      "step": 50517,
      "training_step_time": 7.8953657150268555
    },
    {
      "epoch": 4.817771911621094e-05,
      "model_forward_time": 2.299791097640991,
      "step": 50518
    },
    {
      "epoch": 4.817771911621094e-05,
      "step": 50518,
      "training_step_time": 8.607027292251587
    },
    {
      "epoch": 4.817867279052734e-05,
      "model_forward_time": 1.722867488861084,
      "step": 50519
    },
    {
      "epoch": 4.817867279052734e-05,
      "step": 50519,
      "training_step_time": 9.821138143539429
    },
    {
      "epoch": 4.817962646484375e-05,
      "grad_norm": 0.09822223335504532,
      "learning_rate": 5.327318132645042e-06,
      "loss": 0.0568,
      "step": 50520
    },
    {
      "epoch": 4.817962646484375e-05,
      "model_forward_time": 1.7179503440856934,
      "step": 50520
    },
    {
      "dynamics_loss_avg": 0.07336502149701118,
      "epoch": 4.817962646484375e-05,
      "step": 50520
    },
    {
      "action_loss_avg": 0.0020040289615280925,
      "epoch": 4.817962646484375e-05,
      "step": 50520
    },
    {
      "epoch": 4.817962646484375e-05,
      "step": 50520,
      "training_step_time": 7.778438091278076
    },
    {
      "epoch": 4.818058013916016e-05,
      "model_forward_time": 1.7060589790344238,
      "step": 50521
    },
    {
      "epoch": 4.818058013916016e-05,
      "step": 50521,
      "training_step_time": 7.810484409332275
    },
    {
      "epoch": 4.818153381347656e-05,
      "model_forward_time": 1.7097179889678955,
      "step": 50522
    },
    {
      "epoch": 4.818153381347656e-05,
      "step": 50522,
      "training_step_time": 6.8688390254974365
    },
    {
      "epoch": 4.818248748779297e-05,
      "model_forward_time": 1.7251553535461426,
      "step": 50523
    },
    {
      "epoch": 4.818248748779297e-05,
      "step": 50523,
      "training_step_time": 6.999558687210083
    },
    {
      "epoch": 4.8183441162109376e-05,
      "model_forward_time": 1.7283782958984375,
      "step": 50524
    },
    {
      "epoch": 4.8183441162109376e-05,
      "step": 50524,
      "training_step_time": 7.001656532287598
    },
    {
      "epoch": 4.818439483642578e-05,
      "model_forward_time": 1.721510410308838,
      "step": 50525
    },
    {
      "epoch": 4.818439483642578e-05,
      "step": 50525,
      "training_step_time": 6.976882219314575
    },
    {
      "epoch": 4.818534851074219e-05,
      "model_forward_time": 1.749880075454712,
      "step": 50526
    },
    {
      "epoch": 4.818534851074219e-05,
      "step": 50526,
      "training_step_time": 8.022425413131714
    },
    {
      "epoch": 4.8186302185058595e-05,
      "model_forward_time": 2.305593490600586,
      "step": 50527
    },
    {
      "epoch": 4.8186302185058595e-05,
      "step": 50527,
      "training_step_time": 7.7951414585113525
    },
    {
      "epoch": 4.8187255859375e-05,
      "model_forward_time": 1.7261021137237549,
      "step": 50528
    },
    {
      "epoch": 4.8187255859375e-05,
      "step": 50528,
      "training_step_time": 9.855599641799927
    },
    {
      "epoch": 4.818820953369141e-05,
      "model_forward_time": 1.7212164402008057,
      "step": 50529
    },
    {
      "epoch": 4.818820953369141e-05,
      "step": 50529,
      "training_step_time": 9.14991569519043
    },
    {
      "epoch": 4.8189163208007814e-05,
      "grad_norm": 0.14791293442249298,
      "learning_rate": 5.325668191725116e-06,
      "loss": 0.0577,
      "step": 50530
    },
    {
      "epoch": 4.8189163208007814e-05,
      "model_forward_time": 1.7254400253295898,
      "step": 50530
    },
    {
      "dynamics_loss_avg": 0.05287481267005205,
      "epoch": 4.8189163208007814e-05,
      "step": 50530
    },
    {
      "action_loss_avg": 0.0026386124431155623,
      "epoch": 4.8189163208007814e-05,
      "step": 50530
    },
    {
      "epoch": 4.8189163208007814e-05,
      "step": 50530,
      "training_step_time": 7.824399709701538
    },
    {
      "epoch": 4.819011688232422e-05,
      "model_forward_time": 1.3507909774780273,
      "step": 50531
    },
    {
      "epoch": 4.819011688232422e-05,
      "step": 50531,
      "training_step_time": 6.98923397064209
    },
    {
      "epoch": 4.819107055664062e-05,
      "model_forward_time": 1.7309010028839111,
      "step": 50532
    },
    {
      "epoch": 4.819107055664062e-05,
      "step": 50532,
      "training_step_time": 7.011677265167236
    },
    {
      "epoch": 4.819202423095703e-05,
      "model_forward_time": 1.7293450832366943,
      "step": 50533
    },
    {
      "epoch": 4.819202423095703e-05,
      "step": 50533,
      "training_step_time": 6.984241247177124
    },
    {
      "epoch": 4.8192977905273436e-05,
      "model_forward_time": 1.734433889389038,
      "step": 50534
    },
    {
      "epoch": 4.8192977905273436e-05,
      "step": 50534,
      "training_step_time": 6.9790990352630615
    },
    {
      "epoch": 4.819393157958984e-05,
      "model_forward_time": 1.7158207893371582,
      "step": 50535
    },
    {
      "epoch": 4.819393157958984e-05,
      "step": 50535,
      "training_step_time": 8.43922734260559
    },
    {
      "epoch": 4.819488525390625e-05,
      "model_forward_time": 2.2648303508758545,
      "step": 50536
    },
    {
      "epoch": 4.819488525390625e-05,
      "step": 50536,
      "training_step_time": 7.823710680007935
    },
    {
      "epoch": 4.8195838928222655e-05,
      "model_forward_time": 1.736166000366211,
      "step": 50537
    },
    {
      "epoch": 4.8195838928222655e-05,
      "step": 50537,
      "training_step_time": 9.595060348510742
    },
    {
      "epoch": 4.819679260253906e-05,
      "model_forward_time": 1.7256252765655518,
      "step": 50538
    },
    {
      "epoch": 4.819679260253906e-05,
      "step": 50538,
      "training_step_time": 8.916611194610596
    },
    {
      "epoch": 4.819774627685547e-05,
      "model_forward_time": 1.3454704284667969,
      "step": 50539
    },
    {
      "epoch": 4.819774627685547e-05,
      "step": 50539,
      "training_step_time": 7.805497407913208
    },
    {
      "epoch": 4.8198699951171874e-05,
      "grad_norm": 0.12144986540079117,
      "learning_rate": 5.324018215190606e-06,
      "loss": 0.0573,
      "step": 50540
    },
    {
      "epoch": 4.8198699951171874e-05,
      "model_forward_time": 1.7183549404144287,
      "step": 50540
    },
    {
      "dynamics_loss_avg": 0.0768771730363369,
      "epoch": 4.8198699951171874e-05,
      "step": 50540
    },
    {
      "action_loss_avg": 0.003001774742733687,
      "epoch": 4.8198699951171874e-05,
      "step": 50540
    },
    {
      "epoch": 4.8198699951171874e-05,
      "step": 50540,
      "training_step_time": 6.991789102554321
    },
    {
      "epoch": 4.819965362548828e-05,
      "model_forward_time": 1.7294416427612305,
      "step": 50541
    },
    {
      "epoch": 4.819965362548828e-05,
      "step": 50541,
      "training_step_time": 6.737414360046387
    },
    {
      "epoch": 4.820060729980469e-05,
      "model_forward_time": 1.741591215133667,
      "step": 50542
    },
    {
      "epoch": 4.820060729980469e-05,
      "step": 50542,
      "training_step_time": 6.997697830200195
    },
    {
      "epoch": 4.820156097412109e-05,
      "model_forward_time": 1.7295711040496826,
      "step": 50543
    },
    {
      "epoch": 4.820156097412109e-05,
      "step": 50543,
      "training_step_time": 6.942936897277832
    },
    {
      "epoch": 4.82025146484375e-05,
      "model_forward_time": 1.727264165878296,
      "step": 50544
    },
    {
      "epoch": 4.82025146484375e-05,
      "step": 50544,
      "training_step_time": 7.625977277755737
    },
    {
      "epoch": 4.820346832275391e-05,
      "model_forward_time": 1.8128118515014648,
      "step": 50545
    },
    {
      "epoch": 4.820346832275391e-05,
      "step": 50545,
      "training_step_time": 8.544187545776367
    },
    {
      "epoch": 4.820442199707031e-05,
      "model_forward_time": 2.302107095718384,
      "step": 50546
    },
    {
      "epoch": 4.820442199707031e-05,
      "step": 50546,
      "training_step_time": 9.695353507995605
    },
    {
      "epoch": 4.8205375671386716e-05,
      "model_forward_time": 1.7314436435699463,
      "step": 50547
    },
    {
      "epoch": 4.8205375671386716e-05,
      "step": 50547,
      "training_step_time": 10.538516759872437
    },
    {
      "epoch": 4.820632934570313e-05,
      "model_forward_time": 1.7315020561218262,
      "step": 50548
    },
    {
      "epoch": 4.820632934570313e-05,
      "step": 50548,
      "training_step_time": 8.66621446609497
    },
    {
      "epoch": 4.820728302001953e-05,
      "model_forward_time": 1.7423343658447266,
      "step": 50549
    },
    {
      "epoch": 4.820728302001953e-05,
      "step": 50549,
      "training_step_time": 6.99377965927124
    },
    {
      "epoch": 4.8208236694335935e-05,
      "grad_norm": 0.09691489487886429,
      "learning_rate": 5.3223682032219515e-06,
      "loss": 0.0593,
      "step": 50550
    },
    {
      "epoch": 4.8208236694335935e-05,
      "model_forward_time": 1.7264702320098877,
      "step": 50550
    },
    {
      "dynamics_loss_avg": 0.053206710144877434,
      "epoch": 4.8208236694335935e-05,
      "step": 50550
    },
    {
      "action_loss_avg": 0.0022138670552521943,
      "epoch": 4.8208236694335935e-05,
      "step": 50550
    },
    {
      "epoch": 4.8208236694335935e-05,
      "step": 50550,
      "training_step_time": 6.971782922744751
    },
    {
      "epoch": 4.8209190368652346e-05,
      "model_forward_time": 1.7375693321228027,
      "step": 50551
    },
    {
      "epoch": 4.8209190368652346e-05,
      "step": 50551,
      "training_step_time": 6.927755355834961
    },
    {
      "epoch": 4.821014404296875e-05,
      "model_forward_time": 1.7324576377868652,
      "step": 50552
    },
    {
      "epoch": 4.821014404296875e-05,
      "step": 50552,
      "training_step_time": 6.95603609085083
    },
    {
      "epoch": 4.8211097717285154e-05,
      "model_forward_time": 1.7147009372711182,
      "step": 50553
    },
    {
      "epoch": 4.8211097717285154e-05,
      "step": 50553,
      "training_step_time": 7.637676954269409
    },
    {
      "epoch": 4.8212051391601564e-05,
      "model_forward_time": 1.7981605529785156,
      "step": 50554
    },
    {
      "epoch": 4.8212051391601564e-05,
      "step": 50554,
      "training_step_time": 7.9517741203308105
    },
    {
      "epoch": 4.821300506591797e-05,
      "model_forward_time": 2.294828414916992,
      "step": 50555
    },
    {
      "epoch": 4.821300506591797e-05,
      "step": 50555,
      "training_step_time": 9.473840236663818
    },
    {
      "epoch": 4.821395874023437e-05,
      "model_forward_time": 1.736579179763794,
      "step": 50556
    },
    {
      "epoch": 4.821395874023437e-05,
      "step": 50556,
      "training_step_time": 8.801676988601685
    },
    {
      "epoch": 4.821491241455078e-05,
      "model_forward_time": 1.7357549667358398,
      "step": 50557
    },
    {
      "epoch": 4.821491241455078e-05,
      "step": 50557,
      "training_step_time": 8.842710256576538
    },
    {
      "epoch": 4.821586608886719e-05,
      "model_forward_time": 1.7069308757781982,
      "step": 50558
    },
    {
      "epoch": 4.821586608886719e-05,
      "step": 50558,
      "training_step_time": 7.659543037414551
    },
    {
      "epoch": 4.821681976318359e-05,
      "model_forward_time": 1.7312595844268799,
      "step": 50559
    },
    {
      "epoch": 4.821681976318359e-05,
      "step": 50559,
      "training_step_time": 6.933533668518066
    },
    {
      "epoch": 4.82177734375e-05,
      "grad_norm": 0.09429267793893814,
      "learning_rate": 5.320718155999594e-06,
      "loss": 0.0551,
      "step": 50560
    },
    {
      "epoch": 4.82177734375e-05,
      "model_forward_time": 1.7227787971496582,
      "step": 50560
    },
    {
      "dynamics_loss_avg": 0.0411419315263629,
      "epoch": 4.82177734375e-05,
      "step": 50560
    },
    {
      "action_loss_avg": 0.002155242869048379,
      "epoch": 4.82177734375e-05,
      "step": 50560
    },
    {
      "epoch": 4.82177734375e-05,
      "step": 50560,
      "training_step_time": 6.928873300552368
    },
    {
      "epoch": 4.8218727111816406e-05,
      "model_forward_time": 1.7317707538604736,
      "step": 50561
    },
    {
      "epoch": 4.8218727111816406e-05,
      "step": 50561,
      "training_step_time": 6.935421705245972
    },
    {
      "epoch": 4.821968078613281e-05,
      "model_forward_time": 1.7335569858551025,
      "step": 50562
    },
    {
      "epoch": 4.821968078613281e-05,
      "step": 50562,
      "training_step_time": 7.851755857467651
    },
    {
      "epoch": 4.822063446044922e-05,
      "model_forward_time": 1.3602542877197266,
      "step": 50563
    },
    {
      "epoch": 4.822063446044922e-05,
      "step": 50563,
      "training_step_time": 7.896261930465698
    },
    {
      "epoch": 4.8221588134765625e-05,
      "model_forward_time": 1.9071314334869385,
      "step": 50564
    },
    {
      "epoch": 4.8221588134765625e-05,
      "step": 50564,
      "training_step_time": 8.297194480895996
    },
    {
      "epoch": 4.822254180908203e-05,
      "model_forward_time": 1.7336199283599854,
      "step": 50565
    },
    {
      "epoch": 4.822254180908203e-05,
      "step": 50565,
      "training_step_time": 8.880284547805786
    },
    {
      "epoch": 4.822349548339844e-05,
      "model_forward_time": 1.7345354557037354,
      "step": 50566
    },
    {
      "epoch": 4.822349548339844e-05,
      "step": 50566,
      "training_step_time": 8.523096323013306
    },
    {
      "epoch": 4.8224449157714844e-05,
      "model_forward_time": 1.737919807434082,
      "step": 50567
    },
    {
      "epoch": 4.8224449157714844e-05,
      "step": 50567,
      "training_step_time": 7.810540437698364
    },
    {
      "epoch": 4.822540283203125e-05,
      "model_forward_time": 1.7364459037780762,
      "step": 50568
    },
    {
      "epoch": 4.822540283203125e-05,
      "step": 50568,
      "training_step_time": 6.957633018493652
    },
    {
      "epoch": 4.822635650634766e-05,
      "model_forward_time": 1.7284870147705078,
      "step": 50569
    },
    {
      "epoch": 4.822635650634766e-05,
      "step": 50569,
      "training_step_time": 7.0167295932769775
    },
    {
      "epoch": 4.822731018066406e-05,
      "grad_norm": 0.12224321812391281,
      "learning_rate": 5.319068073703984e-06,
      "loss": 0.0562,
      "step": 50570
    },
    {
      "epoch": 4.822731018066406e-05,
      "model_forward_time": 1.7264420986175537,
      "step": 50570
    },
    {
      "dynamics_loss_avg": 0.050963260466232896,
      "epoch": 4.822731018066406e-05,
      "step": 50570
    },
    {
      "action_loss_avg": 0.001674422129872255,
      "epoch": 4.822731018066406e-05,
      "step": 50570
    },
    {
      "epoch": 4.822731018066406e-05,
      "step": 50570,
      "training_step_time": 7.003153085708618
    },
    {
      "epoch": 4.822826385498047e-05,
      "model_forward_time": 1.7264125347137451,
      "step": 50571
    },
    {
      "epoch": 4.822826385498047e-05,
      "step": 50571,
      "training_step_time": 7.922434568405151
    },
    {
      "epoch": 4.822921752929688e-05,
      "model_forward_time": 1.7553679943084717,
      "step": 50572
    },
    {
      "epoch": 4.822921752929688e-05,
      "step": 50572,
      "training_step_time": 7.893902063369751
    },
    {
      "epoch": 4.823017120361328e-05,
      "model_forward_time": 1.7358863353729248,
      "step": 50573
    },
    {
      "epoch": 4.823017120361328e-05,
      "step": 50573,
      "training_step_time": 7.831055641174316
    },
    {
      "epoch": 4.8231124877929686e-05,
      "model_forward_time": 2.301156759262085,
      "step": 50574
    },
    {
      "epoch": 4.8231124877929686e-05,
      "step": 50574,
      "training_step_time": 10.359873294830322
    },
    {
      "epoch": 4.8232078552246096e-05,
      "model_forward_time": 1.7291176319122314,
      "step": 50575
    },
    {
      "epoch": 4.8232078552246096e-05,
      "step": 50575,
      "training_step_time": 8.368459463119507
    },
    {
      "epoch": 4.82330322265625e-05,
      "model_forward_time": 1.7304117679595947,
      "step": 50576
    },
    {
      "epoch": 4.82330322265625e-05,
      "step": 50576,
      "training_step_time": 8.655397415161133
    },
    {
      "epoch": 4.8233985900878905e-05,
      "model_forward_time": 1.739912986755371,
      "step": 50577
    },
    {
      "epoch": 4.8233985900878905e-05,
      "step": 50577,
      "training_step_time": 6.990658760070801
    },
    {
      "epoch": 4.8234939575195315e-05,
      "model_forward_time": 1.7331421375274658,
      "step": 50578
    },
    {
      "epoch": 4.8234939575195315e-05,
      "step": 50578,
      "training_step_time": 6.9446587562561035
    },
    {
      "epoch": 4.823589324951172e-05,
      "model_forward_time": 1.7397024631500244,
      "step": 50579
    },
    {
      "epoch": 4.823589324951172e-05,
      "step": 50579,
      "training_step_time": 6.958534479141235
    },
    {
      "epoch": 4.823684692382812e-05,
      "grad_norm": 0.09798179566860199,
      "learning_rate": 5.317417956515567e-06,
      "loss": 0.0548,
      "step": 50580
    },
    {
      "epoch": 4.823684692382812e-05,
      "model_forward_time": 1.720884084701538,
      "step": 50580
    },
    {
      "dynamics_loss_avg": 0.06928216889500619,
      "epoch": 4.823684692382812e-05,
      "step": 50580
    },
    {
      "action_loss_avg": 0.0026153957354836167,
      "epoch": 4.823684692382812e-05,
      "step": 50580
    },
    {
      "epoch": 4.823684692382812e-05,
      "step": 50580,
      "training_step_time": 7.889297246932983
    },
    {
      "epoch": 4.8237800598144534e-05,
      "model_forward_time": 1.737621784210205,
      "step": 50581
    },
    {
      "epoch": 4.8237800598144534e-05,
      "step": 50581,
      "training_step_time": 7.933794260025024
    },
    {
      "epoch": 4.823875427246094e-05,
      "model_forward_time": 1.802536964416504,
      "step": 50582
    },
    {
      "epoch": 4.823875427246094e-05,
      "step": 50582,
      "training_step_time": 7.848839521408081
    },
    {
      "epoch": 4.823970794677734e-05,
      "model_forward_time": 2.294917106628418,
      "step": 50583
    },
    {
      "epoch": 4.823970794677734e-05,
      "step": 50583,
      "training_step_time": 9.754125118255615
    },
    {
      "epoch": 4.824066162109375e-05,
      "model_forward_time": 1.7299017906188965,
      "step": 50584
    },
    {
      "epoch": 4.824066162109375e-05,
      "step": 50584,
      "training_step_time": 7.928530693054199
    },
    {
      "epoch": 4.824161529541016e-05,
      "model_forward_time": 1.7311971187591553,
      "step": 50585
    },
    {
      "epoch": 4.824161529541016e-05,
      "step": 50585,
      "training_step_time": 7.871637344360352
    },
    {
      "epoch": 4.824256896972656e-05,
      "model_forward_time": 1.7391574382781982,
      "step": 50586
    },
    {
      "epoch": 4.824256896972656e-05,
      "step": 50586,
      "training_step_time": 7.536296367645264
    },
    {
      "epoch": 4.824352264404297e-05,
      "model_forward_time": 1.732562780380249,
      "step": 50587
    },
    {
      "epoch": 4.824352264404297e-05,
      "step": 50587,
      "training_step_time": 6.889318466186523
    },
    {
      "epoch": 4.8244476318359376e-05,
      "model_forward_time": 1.726954460144043,
      "step": 50588
    },
    {
      "epoch": 4.8244476318359376e-05,
      "step": 50588,
      "training_step_time": 6.940843343734741
    },
    {
      "epoch": 4.824542999267578e-05,
      "model_forward_time": 1.7335689067840576,
      "step": 50589
    },
    {
      "epoch": 4.824542999267578e-05,
      "step": 50589,
      "training_step_time": 7.867280721664429
    },
    {
      "epoch": 4.824638366699219e-05,
      "grad_norm": 0.09982801973819733,
      "learning_rate": 5.315767804614799e-06,
      "loss": 0.0564,
      "step": 50590
    },
    {
      "epoch": 4.824638366699219e-05,
      "model_forward_time": 1.722393274307251,
      "step": 50590
    },
    {
      "dynamics_loss_avg": 0.05329894796013832,
      "epoch": 4.824638366699219e-05,
      "step": 50590
    },
    {
      "action_loss_avg": 0.0018921168491942808,
      "epoch": 4.824638366699219e-05,
      "step": 50590
    },
    {
      "epoch": 4.824638366699219e-05,
      "step": 50590,
      "training_step_time": 7.902807950973511
    },
    {
      "epoch": 4.8247337341308595e-05,
      "model_forward_time": 1.8056809902191162,
      "step": 50591
    },
    {
      "epoch": 4.8247337341308595e-05,
      "step": 50591,
      "training_step_time": 9.292696714401245
    },
    {
      "epoch": 4.8248291015625e-05,
      "model_forward_time": 2.3135409355163574,
      "step": 50592
    },
    {
      "epoch": 4.8248291015625e-05,
      "step": 50592,
      "training_step_time": 10.542692422866821
    },
    {
      "epoch": 4.824924468994141e-05,
      "model_forward_time": 1.7383613586425781,
      "step": 50593
    },
    {
      "epoch": 4.824924468994141e-05,
      "step": 50593,
      "training_step_time": 9.367859840393066
    },
    {
      "epoch": 4.8250198364257814e-05,
      "model_forward_time": 1.735091209411621,
      "step": 50594
    },
    {
      "epoch": 4.8250198364257814e-05,
      "step": 50594,
      "training_step_time": 7.876862287521362
    },
    {
      "epoch": 4.825115203857422e-05,
      "model_forward_time": 1.7394404411315918,
      "step": 50595
    },
    {
      "epoch": 4.825115203857422e-05,
      "step": 50595,
      "training_step_time": 7.666185617446899
    },
    {
      "epoch": 4.825210571289062e-05,
      "model_forward_time": 1.7324259281158447,
      "step": 50596
    },
    {
      "epoch": 4.825210571289062e-05,
      "step": 50596,
      "training_step_time": 6.896118879318237
    },
    {
      "epoch": 4.825305938720703e-05,
      "model_forward_time": 1.7392749786376953,
      "step": 50597
    },
    {
      "epoch": 4.825305938720703e-05,
      "step": 50597,
      "training_step_time": 6.890924692153931
    },
    {
      "epoch": 4.8254013061523437e-05,
      "model_forward_time": 1.7397055625915527,
      "step": 50598
    },
    {
      "epoch": 4.8254013061523437e-05,
      "step": 50598,
      "training_step_time": 8.736204624176025
    },
    {
      "epoch": 4.825496673583984e-05,
      "model_forward_time": 1.7353620529174805,
      "step": 50599
    },
    {
      "epoch": 4.825496673583984e-05,
      "step": 50599,
      "training_step_time": 8.078159093856812
    },
    {
      "epoch": 4.825592041015625e-05,
      "grad_norm": 0.10166630148887634,
      "learning_rate": 5.31411761818214e-06,
      "loss": 0.0571,
      "step": 50600
    },
    {
      "epoch": 4.825592041015625e-05,
      "model_forward_time": 1.8056933879852295,
      "step": 50600
    },
    {
      "dynamics_loss_avg": 0.07353381775319576,
      "epoch": 4.825592041015625e-05,
      "step": 50600
    },
    {
      "action_loss_avg": 0.0019728509709239007,
      "epoch": 4.825592041015625e-05,
      "step": 50600
    },
    {
      "epoch": 4.825592041015625e-05,
      "step": 50600,
      "training_step_time": 7.848130464553833
    },
    {
      "epoch": 4.8256874084472655e-05,
      "model_forward_time": 2.3381874561309814,
      "step": 50601
    },
    {
      "epoch": 4.8256874084472655e-05,
      "step": 50601,
      "training_step_time": 8.89650011062622
    },
    {
      "epoch": 4.825782775878906e-05,
      "model_forward_time": 1.7446746826171875,
      "step": 50602
    },
    {
      "epoch": 4.825782775878906e-05,
      "step": 50602,
      "training_step_time": 9.055591821670532
    },
    {
      "epoch": 4.825878143310547e-05,
      "model_forward_time": 1.7407259941101074,
      "step": 50603
    },
    {
      "epoch": 4.825878143310547e-05,
      "step": 50603,
      "training_step_time": 8.565359830856323
    },
    {
      "epoch": 4.8259735107421874e-05,
      "model_forward_time": 1.7331106662750244,
      "step": 50604
    },
    {
      "epoch": 4.8259735107421874e-05,
      "step": 50604,
      "training_step_time": 7.7512102127075195
    },
    {
      "epoch": 4.826068878173828e-05,
      "model_forward_time": 1.7356443405151367,
      "step": 50605
    },
    {
      "epoch": 4.826068878173828e-05,
      "step": 50605,
      "training_step_time": 6.946847438812256
    },
    {
      "epoch": 4.826164245605469e-05,
      "model_forward_time": 1.753157138824463,
      "step": 50606
    },
    {
      "epoch": 4.826164245605469e-05,
      "step": 50606,
      "training_step_time": 6.948528528213501
    },
    {
      "epoch": 4.826259613037109e-05,
      "model_forward_time": 1.7346291542053223,
      "step": 50607
    },
    {
      "epoch": 4.826259613037109e-05,
      "step": 50607,
      "training_step_time": 7.927048683166504
    },
    {
      "epoch": 4.82635498046875e-05,
      "model_forward_time": 1.7317378520965576,
      "step": 50608
    },
    {
      "epoch": 4.82635498046875e-05,
      "step": 50608,
      "training_step_time": 8.03229022026062
    },
    {
      "epoch": 4.826450347900391e-05,
      "model_forward_time": 1.8209497928619385,
      "step": 50609
    },
    {
      "epoch": 4.826450347900391e-05,
      "step": 50609,
      "training_step_time": 9.655977964401245
    },
    {
      "epoch": 4.826545715332031e-05,
      "grad_norm": 0.1166326105594635,
      "learning_rate": 5.31246739739805e-06,
      "loss": 0.0574,
      "step": 50610
    },
    {
      "epoch": 4.826545715332031e-05,
      "model_forward_time": 2.2934775352478027,
      "step": 50610
    },
    {
      "dynamics_loss_avg": 0.06748966537415982,
      "epoch": 4.826545715332031e-05,
      "step": 50610
    },
    {
      "action_loss_avg": 0.001772467658156529,
      "epoch": 4.826545715332031e-05,
      "step": 50610
    },
    {
      "epoch": 4.826545715332031e-05,
      "step": 50610,
      "training_step_time": 8.600456237792969
    },
    {
      "epoch": 4.8266410827636716e-05,
      "model_forward_time": 1.7409427165985107,
      "step": 50611
    },
    {
      "epoch": 4.8266410827636716e-05,
      "step": 50611,
      "training_step_time": 9.91826343536377
    },
    {
      "epoch": 4.826736450195313e-05,
      "model_forward_time": 1.7339487075805664,
      "step": 50612
    },
    {
      "epoch": 4.826736450195313e-05,
      "step": 50612,
      "training_step_time": 7.897449970245361
    },
    {
      "epoch": 4.826831817626953e-05,
      "model_forward_time": 1.7329933643341064,
      "step": 50613
    },
    {
      "epoch": 4.826831817626953e-05,
      "step": 50613,
      "training_step_time": 7.867274761199951
    },
    {
      "epoch": 4.8269271850585935e-05,
      "model_forward_time": 1.73484468460083,
      "step": 50614
    },
    {
      "epoch": 4.8269271850585935e-05,
      "step": 50614,
      "training_step_time": 6.950496673583984
    },
    {
      "epoch": 4.8270225524902346e-05,
      "model_forward_time": 1.7326507568359375,
      "step": 50615
    },
    {
      "epoch": 4.8270225524902346e-05,
      "step": 50615,
      "training_step_time": 6.893819332122803
    },
    {
      "epoch": 4.827117919921875e-05,
      "model_forward_time": 1.7366461753845215,
      "step": 50616
    },
    {
      "epoch": 4.827117919921875e-05,
      "step": 50616,
      "training_step_time": 7.868356466293335
    },
    {
      "epoch": 4.8272132873535154e-05,
      "model_forward_time": 1.7300629615783691,
      "step": 50617
    },
    {
      "epoch": 4.8272132873535154e-05,
      "step": 50617,
      "training_step_time": 7.981524705886841
    },
    {
      "epoch": 4.8273086547851565e-05,
      "model_forward_time": 1.7895448207855225,
      "step": 50618
    },
    {
      "epoch": 4.8273086547851565e-05,
      "step": 50618,
      "training_step_time": 8.843024492263794
    },
    {
      "epoch": 4.827404022216797e-05,
      "model_forward_time": 1.7419688701629639,
      "step": 50619
    },
    {
      "epoch": 4.827404022216797e-05,
      "step": 50619,
      "training_step_time": 8.981834411621094
    },
    {
      "epoch": 4.827499389648437e-05,
      "grad_norm": 0.10374724119901657,
      "learning_rate": 5.310817142442995e-06,
      "loss": 0.0536,
      "step": 50620
    },
    {
      "epoch": 4.827499389648437e-05,
      "model_forward_time": 1.7260181903839111,
      "step": 50620
    },
    {
      "dynamics_loss_avg": 0.05868036458268762,
      "epoch": 4.827499389648437e-05,
      "step": 50620
    },
    {
      "action_loss_avg": 0.002340964235190768,
      "epoch": 4.827499389648437e-05,
      "step": 50620
    },
    {
      "epoch": 4.827499389648437e-05,
      "step": 50620,
      "training_step_time": 10.239061832427979
    },
    {
      "epoch": 4.827594757080078e-05,
      "model_forward_time": 1.73714280128479,
      "step": 50621
    },
    {
      "epoch": 4.827594757080078e-05,
      "step": 50621,
      "training_step_time": 9.142329692840576
    },
    {
      "epoch": 4.827690124511719e-05,
      "model_forward_time": 1.7413086891174316,
      "step": 50622
    },
    {
      "epoch": 4.827690124511719e-05,
      "step": 50622,
      "training_step_time": 7.824152231216431
    },
    {
      "epoch": 4.827785491943359e-05,
      "model_forward_time": 1.7413053512573242,
      "step": 50623
    },
    {
      "epoch": 4.827785491943359e-05,
      "step": 50623,
      "training_step_time": 7.637897253036499
    },
    {
      "epoch": 4.827880859375e-05,
      "model_forward_time": 1.7397637367248535,
      "step": 50624
    },
    {
      "epoch": 4.827880859375e-05,
      "step": 50624,
      "training_step_time": 6.934894561767578
    },
    {
      "epoch": 4.8279762268066406e-05,
      "model_forward_time": 1.7301371097564697,
      "step": 50625
    },
    {
      "epoch": 4.8279762268066406e-05,
      "step": 50625,
      "training_step_time": 8.542805671691895
    },
    {
      "epoch": 4.828071594238281e-05,
      "model_forward_time": 1.7348425388336182,
      "step": 50626
    },
    {
      "epoch": 4.828071594238281e-05,
      "step": 50626,
      "training_step_time": 7.934044599533081
    },
    {
      "epoch": 4.828166961669922e-05,
      "model_forward_time": 1.7388372421264648,
      "step": 50627
    },
    {
      "epoch": 4.828166961669922e-05,
      "step": 50627,
      "training_step_time": 7.7018210887908936
    },
    {
      "epoch": 4.8282623291015625e-05,
      "model_forward_time": 1.7457540035247803,
      "step": 50628
    },
    {
      "epoch": 4.8282623291015625e-05,
      "step": 50628,
      "training_step_time": 9.140183210372925
    },
    {
      "epoch": 4.828357696533203e-05,
      "model_forward_time": 2.3012003898620605,
      "step": 50629
    },
    {
      "epoch": 4.828357696533203e-05,
      "step": 50629,
      "training_step_time": 9.45354962348938
    },
    {
      "epoch": 4.828453063964844e-05,
      "grad_norm": 0.09574254602193832,
      "learning_rate": 5.309166853497445e-06,
      "loss": 0.0551,
      "step": 50630
    },
    {
      "epoch": 4.828453063964844e-05,
      "model_forward_time": 1.727980136871338,
      "step": 50630
    },
    {
      "dynamics_loss_avg": 0.049778456892818214,
      "epoch": 4.828453063964844e-05,
      "step": 50630
    },
    {
      "action_loss_avg": 0.003141203775885515,
      "epoch": 4.828453063964844e-05,
      "step": 50630
    },
    {
      "epoch": 4.828453063964844e-05,
      "step": 50630,
      "training_step_time": 8.501437187194824
    },
    {
      "epoch": 4.8285484313964844e-05,
      "model_forward_time": 1.742222785949707,
      "step": 50631
    },
    {
      "epoch": 4.8285484313964844e-05,
      "step": 50631,
      "training_step_time": 7.894616365432739
    },
    {
      "epoch": 4.828643798828125e-05,
      "model_forward_time": 1.7333998680114746,
      "step": 50632
    },
    {
      "epoch": 4.828643798828125e-05,
      "step": 50632,
      "training_step_time": 7.810034513473511
    },
    {
      "epoch": 4.828739166259766e-05,
      "model_forward_time": 1.7386250495910645,
      "step": 50633
    },
    {
      "epoch": 4.828739166259766e-05,
      "step": 50633,
      "training_step_time": 6.970723628997803
    },
    {
      "epoch": 4.828834533691406e-05,
      "model_forward_time": 1.7386412620544434,
      "step": 50634
    },
    {
      "epoch": 4.828834533691406e-05,
      "step": 50634,
      "training_step_time": 7.938754558563232
    },
    {
      "epoch": 4.828929901123047e-05,
      "model_forward_time": 1.7496542930603027,
      "step": 50635
    },
    {
      "epoch": 4.828929901123047e-05,
      "step": 50635,
      "training_step_time": 7.923675537109375
    },
    {
      "epoch": 4.829025268554688e-05,
      "model_forward_time": 1.7464418411254883,
      "step": 50636
    },
    {
      "epoch": 4.829025268554688e-05,
      "step": 50636,
      "training_step_time": 7.830703496932983
    },
    {
      "epoch": 4.829120635986328e-05,
      "model_forward_time": 1.737107515335083,
      "step": 50637
    },
    {
      "epoch": 4.829120635986328e-05,
      "step": 50637,
      "training_step_time": 8.596256971359253
    },
    {
      "epoch": 4.8292160034179686e-05,
      "model_forward_time": 2.2690348625183105,
      "step": 50638
    },
    {
      "epoch": 4.8292160034179686e-05,
      "step": 50638,
      "training_step_time": 9.079817056655884
    },
    {
      "epoch": 4.8293113708496097e-05,
      "model_forward_time": 1.7373542785644531,
      "step": 50639
    },
    {
      "epoch": 4.8293113708496097e-05,
      "step": 50639,
      "training_step_time": 7.726064205169678
    },
    {
      "epoch": 4.82940673828125e-05,
      "grad_norm": 0.11988529562950134,
      "learning_rate": 5.307516530741873e-06,
      "loss": 0.058,
      "step": 50640
    },
    {
      "epoch": 4.82940673828125e-05,
      "model_forward_time": 1.7312736511230469,
      "step": 50640
    },
    {
      "dynamics_loss_avg": 0.06246078219264746,
      "epoch": 4.82940673828125e-05,
      "step": 50640
    },
    {
      "action_loss_avg": 0.0018704980335314757,
      "epoch": 4.82940673828125e-05,
      "step": 50640
    },
    {
      "epoch": 4.82940673828125e-05,
      "step": 50640,
      "training_step_time": 7.911445379257202
    },
    {
      "epoch": 4.8295021057128905e-05,
      "model_forward_time": 1.7410624027252197,
      "step": 50641
    },
    {
      "epoch": 4.8295021057128905e-05,
      "step": 50641,
      "training_step_time": 7.89600682258606
    },
    {
      "epoch": 4.8295974731445315e-05,
      "model_forward_time": 1.7368874549865723,
      "step": 50642
    },
    {
      "epoch": 4.8295974731445315e-05,
      "step": 50642,
      "training_step_time": 6.967170000076294
    },
    {
      "epoch": 4.829692840576172e-05,
      "model_forward_time": 1.741647720336914,
      "step": 50643
    },
    {
      "epoch": 4.829692840576172e-05,
      "step": 50643,
      "training_step_time": 8.013100624084473
    },
    {
      "epoch": 4.8297882080078123e-05,
      "model_forward_time": 1.7391486167907715,
      "step": 50644
    },
    {
      "epoch": 4.8297882080078123e-05,
      "step": 50644,
      "training_step_time": 7.8590404987335205
    },
    {
      "epoch": 4.8298835754394534e-05,
      "model_forward_time": 1.7294251918792725,
      "step": 50645
    },
    {
      "epoch": 4.8298835754394534e-05,
      "step": 50645,
      "training_step_time": 7.820251941680908
    },
    {
      "epoch": 4.829978942871094e-05,
      "model_forward_time": 1.8157994747161865,
      "step": 50646
    },
    {
      "epoch": 4.829978942871094e-05,
      "step": 50646,
      "training_step_time": 7.8894665241241455
    },
    {
      "epoch": 4.830074310302734e-05,
      "model_forward_time": 2.2938435077667236,
      "step": 50647
    },
    {
      "epoch": 4.830074310302734e-05,
      "step": 50647,
      "training_step_time": 8.548954725265503
    },
    {
      "epoch": 4.830169677734375e-05,
      "model_forward_time": 1.747830867767334,
      "step": 50648
    },
    {
      "epoch": 4.830169677734375e-05,
      "step": 50648,
      "training_step_time": 8.940213203430176
    },
    {
      "epoch": 4.830265045166016e-05,
      "model_forward_time": 1.7343029975891113,
      "step": 50649
    },
    {
      "epoch": 4.830265045166016e-05,
      "step": 50649,
      "training_step_time": 9.249140501022339
    },
    {
      "epoch": 4.830360412597656e-05,
      "grad_norm": 0.0811140388250351,
      "learning_rate": 5.305866174356754e-06,
      "loss": 0.0582,
      "step": 50650
    },
    {
      "epoch": 4.830360412597656e-05,
      "model_forward_time": 1.7297053337097168,
      "step": 50650
    },
    {
      "dynamics_loss_avg": 0.037530267983675,
      "epoch": 4.830360412597656e-05,
      "step": 50650
    },
    {
      "action_loss_avg": 0.0015864894492551684,
      "epoch": 4.830360412597656e-05,
      "step": 50650
    },
    {
      "epoch": 4.830360412597656e-05,
      "step": 50650,
      "training_step_time": 7.768589496612549
    },
    {
      "epoch": 4.830455780029297e-05,
      "model_forward_time": 1.7359778881072998,
      "step": 50651
    },
    {
      "epoch": 4.830455780029297e-05,
      "step": 50651,
      "training_step_time": 7.2099573612213135
    },
    {
      "epoch": 4.8305511474609376e-05,
      "model_forward_time": 1.3660645484924316,
      "step": 50652
    },
    {
      "epoch": 4.8305511474609376e-05,
      "step": 50652,
      "training_step_time": 7.872346878051758
    },
    {
      "epoch": 4.830646514892578e-05,
      "model_forward_time": 1.7468934059143066,
      "step": 50653
    },
    {
      "epoch": 4.830646514892578e-05,
      "step": 50653,
      "training_step_time": 7.324256896972656
    },
    {
      "epoch": 4.830741882324219e-05,
      "model_forward_time": 1.7372918128967285,
      "step": 50654
    },
    {
      "epoch": 4.830741882324219e-05,
      "step": 50654,
      "training_step_time": 7.905531406402588
    },
    {
      "epoch": 4.8308372497558595e-05,
      "model_forward_time": 1.798271894454956,
      "step": 50655
    },
    {
      "epoch": 4.8308372497558595e-05,
      "step": 50655,
      "training_step_time": 7.889038562774658
    },
    {
      "epoch": 4.8309326171875e-05,
      "model_forward_time": 2.331352949142456,
      "step": 50656
    },
    {
      "epoch": 4.8309326171875e-05,
      "step": 50656,
      "training_step_time": 7.86174464225769
    },
    {
      "epoch": 4.83102798461914e-05,
      "model_forward_time": 1.7366724014282227,
      "step": 50657
    },
    {
      "epoch": 4.83102798461914e-05,
      "step": 50657,
      "training_step_time": 10.075710535049438
    },
    {
      "epoch": 4.8311233520507814e-05,
      "model_forward_time": 1.7437329292297363,
      "step": 50658
    },
    {
      "epoch": 4.8311233520507814e-05,
      "step": 50658,
      "training_step_time": 7.799588680267334
    },
    {
      "epoch": 4.831218719482422e-05,
      "model_forward_time": 1.740201473236084,
      "step": 50659
    },
    {
      "epoch": 4.831218719482422e-05,
      "step": 50659,
      "training_step_time": 7.79717755317688
    },
    {
      "epoch": 4.831314086914062e-05,
      "grad_norm": 0.13431088626384735,
      "learning_rate": 5.3042157845225715e-06,
      "loss": 0.0605,
      "step": 50660
    },
    {
      "epoch": 4.831314086914062e-05,
      "model_forward_time": 1.7281198501586914,
      "step": 50660
    },
    {
      "dynamics_loss_avg": 0.05848715286701918,
      "epoch": 4.831314086914062e-05,
      "step": 50660
    },
    {
      "action_loss_avg": 0.0024126615200657396,
      "epoch": 4.831314086914062e-05,
      "step": 50660
    },
    {
      "epoch": 4.831314086914062e-05,
      "step": 50660,
      "training_step_time": 7.23930025100708
    },
    {
      "epoch": 4.831409454345703e-05,
      "model_forward_time": 1.7485706806182861,
      "step": 50661
    },
    {
      "epoch": 4.831409454345703e-05,
      "step": 50661,
      "training_step_time": 7.869740009307861
    },
    {
      "epoch": 4.8315048217773437e-05,
      "model_forward_time": 1.7253446578979492,
      "step": 50662
    },
    {
      "epoch": 4.8315048217773437e-05,
      "step": 50662,
      "training_step_time": 7.337427616119385
    },
    {
      "epoch": 4.831600189208984e-05,
      "model_forward_time": 1.7341523170471191,
      "step": 50663
    },
    {
      "epoch": 4.831600189208984e-05,
      "step": 50663,
      "training_step_time": 8.012802600860596
    },
    {
      "epoch": 4.831695556640625e-05,
      "model_forward_time": 1.7903945446014404,
      "step": 50664
    },
    {
      "epoch": 4.831695556640625e-05,
      "step": 50664,
      "training_step_time": 7.88686203956604
    },
    {
      "epoch": 4.8317909240722655e-05,
      "model_forward_time": 1.7531969547271729,
      "step": 50665
    },
    {
      "epoch": 4.8317909240722655e-05,
      "step": 50665,
      "training_step_time": 8.885194540023804
    },
    {
      "epoch": 4.831886291503906e-05,
      "model_forward_time": 1.7531850337982178,
      "step": 50666
    },
    {
      "epoch": 4.831886291503906e-05,
      "step": 50666,
      "training_step_time": 8.602807521820068
    },
    {
      "epoch": 4.831981658935547e-05,
      "model_forward_time": 1.7374136447906494,
      "step": 50667
    },
    {
      "epoch": 4.831981658935547e-05,
      "step": 50667,
      "training_step_time": 7.851346254348755
    },
    {
      "epoch": 4.8320770263671874e-05,
      "model_forward_time": 1.7496333122253418,
      "step": 50668
    },
    {
      "epoch": 4.8320770263671874e-05,
      "step": 50668,
      "training_step_time": 7.804532766342163
    },
    {
      "epoch": 4.832172393798828e-05,
      "model_forward_time": 1.739870548248291,
      "step": 50669
    },
    {
      "epoch": 4.832172393798828e-05,
      "step": 50669,
      "training_step_time": 7.79211163520813
    },
    {
      "epoch": 4.832267761230469e-05,
      "grad_norm": 0.10516173392534256,
      "learning_rate": 5.302565361419808e-06,
      "loss": 0.0602,
      "step": 50670
    },
    {
      "epoch": 4.832267761230469e-05,
      "model_forward_time": 1.7275848388671875,
      "step": 50670
    },
    {
      "dynamics_loss_avg": 0.05503534059971571,
      "epoch": 4.832267761230469e-05,
      "step": 50670
    },
    {
      "action_loss_avg": 0.00228200092096813,
      "epoch": 4.832267761230469e-05,
      "step": 50670
    },
    {
      "epoch": 4.832267761230469e-05,
      "step": 50670,
      "training_step_time": 7.849309206008911
    },
    {
      "epoch": 4.832363128662109e-05,
      "model_forward_time": 1.7305409908294678,
      "step": 50671
    },
    {
      "epoch": 4.832363128662109e-05,
      "step": 50671,
      "training_step_time": 7.212414979934692
    },
    {
      "epoch": 4.83245849609375e-05,
      "model_forward_time": 1.745711326599121,
      "step": 50672
    },
    {
      "epoch": 4.83245849609375e-05,
      "step": 50672,
      "training_step_time": 7.919142723083496
    },
    {
      "epoch": 4.832553863525391e-05,
      "model_forward_time": 1.7569725513458252,
      "step": 50673
    },
    {
      "epoch": 4.832553863525391e-05,
      "step": 50673,
      "training_step_time": 8.427012205123901
    },
    {
      "epoch": 4.832649230957031e-05,
      "model_forward_time": 1.744199275970459,
      "step": 50674
    },
    {
      "epoch": 4.832649230957031e-05,
      "step": 50674,
      "training_step_time": 8.602732181549072
    },
    {
      "epoch": 4.8327445983886716e-05,
      "model_forward_time": 2.2928638458251953,
      "step": 50675
    },
    {
      "epoch": 4.8327445983886716e-05,
      "step": 50675,
      "training_step_time": 9.845603704452515
    },
    {
      "epoch": 4.832839965820313e-05,
      "model_forward_time": 1.7421109676361084,
      "step": 50676
    },
    {
      "epoch": 4.832839965820313e-05,
      "step": 50676,
      "training_step_time": 7.866986274719238
    },
    {
      "epoch": 4.832935333251953e-05,
      "model_forward_time": 1.7320537567138672,
      "step": 50677
    },
    {
      "epoch": 4.832935333251953e-05,
      "step": 50677,
      "training_step_time": 9.048499345779419
    },
    {
      "epoch": 4.8330307006835935e-05,
      "model_forward_time": 1.7450625896453857,
      "step": 50678
    },
    {
      "epoch": 4.8330307006835935e-05,
      "step": 50678,
      "training_step_time": 7.838038682937622
    },
    {
      "epoch": 4.8331260681152346e-05,
      "model_forward_time": 1.7343335151672363,
      "step": 50679
    },
    {
      "epoch": 4.8331260681152346e-05,
      "step": 50679,
      "training_step_time": 7.792050361633301
    },
    {
      "epoch": 4.833221435546875e-05,
      "grad_norm": 0.13302814960479736,
      "learning_rate": 5.300914905228951e-06,
      "loss": 0.0617,
      "step": 50680
    },
    {
      "epoch": 4.833221435546875e-05,
      "model_forward_time": 1.7218074798583984,
      "step": 50680
    },
    {
      "dynamics_loss_avg": 0.06853020135313273,
      "epoch": 4.833221435546875e-05,
      "step": 50680
    },
    {
      "action_loss_avg": 0.0021247378608677536,
      "epoch": 4.833221435546875e-05,
      "step": 50680
    },
    {
      "epoch": 4.833221435546875e-05,
      "step": 50680,
      "training_step_time": 7.227487087249756
    },
    {
      "epoch": 4.8333168029785154e-05,
      "model_forward_time": 1.7260043621063232,
      "step": 50681
    },
    {
      "epoch": 4.8333168029785154e-05,
      "step": 50681,
      "training_step_time": 7.905824422836304
    },
    {
      "epoch": 4.8334121704101565e-05,
      "model_forward_time": 1.3520448207855225,
      "step": 50682
    },
    {
      "epoch": 4.8334121704101565e-05,
      "step": 50682,
      "training_step_time": 7.861509561538696
    },
    {
      "epoch": 4.833507537841797e-05,
      "model_forward_time": 1.726208209991455,
      "step": 50683
    },
    {
      "epoch": 4.833507537841797e-05,
      "step": 50683,
      "training_step_time": 7.831236839294434
    },
    {
      "epoch": 4.833602905273437e-05,
      "model_forward_time": 2.297218084335327,
      "step": 50684
    },
    {
      "epoch": 4.833602905273437e-05,
      "step": 50684,
      "training_step_time": 10.394738912582397
    },
    {
      "epoch": 4.8336982727050783e-05,
      "model_forward_time": 1.7367186546325684,
      "step": 50685
    },
    {
      "epoch": 4.8336982727050783e-05,
      "step": 50685,
      "training_step_time": 9.356674194335938
    },
    {
      "epoch": 4.833793640136719e-05,
      "model_forward_time": 1.7381436824798584,
      "step": 50686
    },
    {
      "epoch": 4.833793640136719e-05,
      "step": 50686,
      "training_step_time": 7.861668348312378
    },
    {
      "epoch": 4.833889007568359e-05,
      "model_forward_time": 1.7304375171661377,
      "step": 50687
    },
    {
      "epoch": 4.833889007568359e-05,
      "step": 50687,
      "training_step_time": 7.8670432567596436
    },
    {
      "epoch": 4.833984375e-05,
      "model_forward_time": 1.730788230895996,
      "step": 50688
    },
    {
      "epoch": 4.833984375e-05,
      "step": 50688,
      "training_step_time": 6.937735080718994
    },
    {
      "epoch": 4.8340797424316406e-05,
      "model_forward_time": 1.731464147567749,
      "step": 50689
    },
    {
      "epoch": 4.8340797424316406e-05,
      "step": 50689,
      "training_step_time": 7.7771031856536865
    },
    {
      "epoch": 4.834175109863281e-05,
      "grad_norm": 0.15309052169322968,
      "learning_rate": 5.299264416130493e-06,
      "loss": 0.0538,
      "step": 50690
    },
    {
      "epoch": 4.834175109863281e-05,
      "model_forward_time": 1.7200758457183838,
      "step": 50690
    },
    {
      "dynamics_loss_avg": 0.060125276632606986,
      "epoch": 4.834175109863281e-05,
      "step": 50690
    },
    {
      "action_loss_avg": 0.0031823583121877164,
      "epoch": 4.834175109863281e-05,
      "step": 50690
    },
    {
      "epoch": 4.834175109863281e-05,
      "step": 50690,
      "training_step_time": 7.904427766799927
    },
    {
      "epoch": 4.834270477294922e-05,
      "model_forward_time": 1.7376782894134521,
      "step": 50691
    },
    {
      "epoch": 4.834270477294922e-05,
      "step": 50691,
      "training_step_time": 7.836654186248779
    },
    {
      "epoch": 4.8343658447265625e-05,
      "model_forward_time": 1.8067593574523926,
      "step": 50692
    },
    {
      "epoch": 4.8343658447265625e-05,
      "step": 50692,
      "training_step_time": 7.921719789505005
    },
    {
      "epoch": 4.834461212158203e-05,
      "model_forward_time": 2.287754774093628,
      "step": 50693
    },
    {
      "epoch": 4.834461212158203e-05,
      "step": 50693,
      "training_step_time": 10.245593070983887
    },
    {
      "epoch": 4.834556579589844e-05,
      "model_forward_time": 1.7309091091156006,
      "step": 50694
    },
    {
      "epoch": 4.834556579589844e-05,
      "step": 50694,
      "training_step_time": 9.58464241027832
    },
    {
      "epoch": 4.8346519470214844e-05,
      "model_forward_time": 1.7246840000152588,
      "step": 50695
    },
    {
      "epoch": 4.8346519470214844e-05,
      "step": 50695,
      "training_step_time": 7.801752328872681
    },
    {
      "epoch": 4.834747314453125e-05,
      "model_forward_time": 1.7225584983825684,
      "step": 50696
    },
    {
      "epoch": 4.834747314453125e-05,
      "step": 50696,
      "training_step_time": 7.8656463623046875
    },
    {
      "epoch": 4.834842681884766e-05,
      "model_forward_time": 1.724461317062378,
      "step": 50697
    },
    {
      "epoch": 4.834842681884766e-05,
      "step": 50697,
      "training_step_time": 7.875962257385254
    },
    {
      "epoch": 4.834938049316406e-05,
      "model_forward_time": 1.7279818058013916,
      "step": 50698
    },
    {
      "epoch": 4.834938049316406e-05,
      "step": 50698,
      "training_step_time": 7.8403639793396
    },
    {
      "epoch": 4.835033416748047e-05,
      "model_forward_time": 1.726975917816162,
      "step": 50699
    },
    {
      "epoch": 4.835033416748047e-05,
      "step": 50699,
      "training_step_time": 7.868760585784912
    },
    {
      "epoch": 4.835128784179688e-05,
      "grad_norm": 0.13401475548744202,
      "learning_rate": 5.297613894304928e-06,
      "loss": 0.0559,
      "step": 50700
    },
    {
      "epoch": 4.835128784179688e-05,
      "model_forward_time": 1.7164738178253174,
      "step": 50700
    },
    {
      "dynamics_loss_avg": 0.08212780468165874,
      "epoch": 4.835128784179688e-05,
      "step": 50700
    },
    {
      "action_loss_avg": 0.0030117336893454192,
      "epoch": 4.835128784179688e-05,
      "step": 50700
    },
    {
      "epoch": 4.835128784179688e-05,
      "step": 50700,
      "training_step_time": 7.807946443557739
    },
    {
      "epoch": 4.835224151611328e-05,
      "model_forward_time": 1.7796382904052734,
      "step": 50701
    },
    {
      "epoch": 4.835224151611328e-05,
      "step": 50701,
      "training_step_time": 9.598353385925293
    },
    {
      "epoch": 4.8353195190429686e-05,
      "model_forward_time": 2.3304200172424316,
      "step": 50702
    },
    {
      "epoch": 4.8353195190429686e-05,
      "step": 50702,
      "training_step_time": 9.689588069915771
    },
    {
      "epoch": 4.8354148864746097e-05,
      "model_forward_time": 1.7237193584442139,
      "step": 50703
    },
    {
      "epoch": 4.8354148864746097e-05,
      "step": 50703,
      "training_step_time": 9.754737615585327
    },
    {
      "epoch": 4.83551025390625e-05,
      "model_forward_time": 1.7015149593353271,
      "step": 50704
    },
    {
      "epoch": 4.83551025390625e-05,
      "step": 50704,
      "training_step_time": 9.122640609741211
    },
    {
      "epoch": 4.8356056213378905e-05,
      "model_forward_time": 1.7316977977752686,
      "step": 50705
    },
    {
      "epoch": 4.8356056213378905e-05,
      "step": 50705,
      "training_step_time": 7.80324649810791
    },
    {
      "epoch": 4.8357009887695315e-05,
      "model_forward_time": 1.7307567596435547,
      "step": 50706
    },
    {
      "epoch": 4.8357009887695315e-05,
      "step": 50706,
      "training_step_time": 7.699144124984741
    },
    {
      "epoch": 4.835796356201172e-05,
      "model_forward_time": 1.7372586727142334,
      "step": 50707
    },
    {
      "epoch": 4.835796356201172e-05,
      "step": 50707,
      "training_step_time": 7.942929029464722
    },
    {
      "epoch": 4.8358917236328123e-05,
      "model_forward_time": 1.7295856475830078,
      "step": 50708
    },
    {
      "epoch": 4.8358917236328123e-05,
      "step": 50708,
      "training_step_time": 7.997326135635376
    },
    {
      "epoch": 4.8359870910644534e-05,
      "model_forward_time": 1.7363696098327637,
      "step": 50709
    },
    {
      "epoch": 4.8359870910644534e-05,
      "step": 50709,
      "training_step_time": 7.835151672363281
    },
    {
      "epoch": 4.836082458496094e-05,
      "grad_norm": 0.10579424351453781,
      "learning_rate": 5.295963339932753e-06,
      "loss": 0.059,
      "step": 50710
    },
    {
      "epoch": 4.836082458496094e-05,
      "model_forward_time": 1.8056907653808594,
      "step": 50710
    },
    {
      "dynamics_loss_avg": 0.05613056439906359,
      "epoch": 4.836082458496094e-05,
      "step": 50710
    },
    {
      "action_loss_avg": 0.0016580433875788003,
      "epoch": 4.836082458496094e-05,
      "step": 50710
    },
    {
      "epoch": 4.836082458496094e-05,
      "step": 50710,
      "training_step_time": 10.244129180908203
    },
    {
      "epoch": 4.836177825927734e-05,
      "model_forward_time": 2.300715923309326,
      "step": 50711
    },
    {
      "epoch": 4.836177825927734e-05,
      "step": 50711,
      "training_step_time": 8.766793012619019
    },
    {
      "epoch": 4.836273193359375e-05,
      "model_forward_time": 1.727482795715332,
      "step": 50712
    },
    {
      "epoch": 4.836273193359375e-05,
      "step": 50712,
      "training_step_time": 8.04302167892456
    },
    {
      "epoch": 4.836368560791016e-05,
      "model_forward_time": 1.732513189315796,
      "step": 50713
    },
    {
      "epoch": 4.836368560791016e-05,
      "step": 50713,
      "training_step_time": 8.436628818511963
    },
    {
      "epoch": 4.836463928222656e-05,
      "model_forward_time": 1.7320196628570557,
      "step": 50714
    },
    {
      "epoch": 4.836463928222656e-05,
      "step": 50714,
      "training_step_time": 7.817699909210205
    },
    {
      "epoch": 4.836559295654297e-05,
      "model_forward_time": 1.743089199066162,
      "step": 50715
    },
    {
      "epoch": 4.836559295654297e-05,
      "step": 50715,
      "training_step_time": 7.940187692642212
    },
    {
      "epoch": 4.8366546630859376e-05,
      "model_forward_time": 1.7292749881744385,
      "step": 50716
    },
    {
      "epoch": 4.8366546630859376e-05,
      "step": 50716,
      "training_step_time": 7.903390169143677
    },
    {
      "epoch": 4.836750030517578e-05,
      "model_forward_time": 1.7328181266784668,
      "step": 50717
    },
    {
      "epoch": 4.836750030517578e-05,
      "step": 50717,
      "training_step_time": 8.007498741149902
    },
    {
      "epoch": 4.836845397949219e-05,
      "model_forward_time": 1.732619047164917,
      "step": 50718
    },
    {
      "epoch": 4.836845397949219e-05,
      "step": 50718,
      "training_step_time": 8.658120393753052
    },
    {
      "epoch": 4.8369407653808595e-05,
      "model_forward_time": 1.7863636016845703,
      "step": 50719
    },
    {
      "epoch": 4.8369407653808595e-05,
      "step": 50719,
      "training_step_time": 8.878898859024048
    },
    {
      "epoch": 4.8370361328125e-05,
      "grad_norm": 0.11158832162618637,
      "learning_rate": 5.294312753194476e-06,
      "loss": 0.0578,
      "step": 50720
    },
    {
      "epoch": 4.8370361328125e-05,
      "model_forward_time": 1.7201974391937256,
      "step": 50720
    },
    {
      "dynamics_loss_avg": 0.05571749173104763,
      "epoch": 4.8370361328125e-05,
      "step": 50720
    },
    {
      "action_loss_avg": 0.0021609092422295363,
      "epoch": 4.8370361328125e-05,
      "step": 50720
    },
    {
      "epoch": 4.8370361328125e-05,
      "step": 50720,
      "training_step_time": 8.975786447525024
    },
    {
      "epoch": 4.83713150024414e-05,
      "model_forward_time": 1.7254202365875244,
      "step": 50721
    },
    {
      "epoch": 4.83713150024414e-05,
      "step": 50721,
      "training_step_time": 8.893774271011353
    },
    {
      "epoch": 4.8372268676757814e-05,
      "model_forward_time": 1.7244646549224854,
      "step": 50722
    },
    {
      "epoch": 4.8372268676757814e-05,
      "step": 50722,
      "training_step_time": 7.81919002532959
    },
    {
      "epoch": 4.837322235107422e-05,
      "model_forward_time": 1.7221786975860596,
      "step": 50723
    },
    {
      "epoch": 4.837322235107422e-05,
      "step": 50723,
      "training_step_time": 7.878970623016357
    },
    {
      "epoch": 4.837417602539062e-05,
      "model_forward_time": 1.7264995574951172,
      "step": 50724
    },
    {
      "epoch": 4.837417602539062e-05,
      "step": 50724,
      "training_step_time": 7.8466198444366455
    },
    {
      "epoch": 4.837512969970703e-05,
      "model_forward_time": 1.7270398139953613,
      "step": 50725
    },
    {
      "epoch": 4.837512969970703e-05,
      "step": 50725,
      "training_step_time": 7.220146894454956
    },
    {
      "epoch": 4.837608337402344e-05,
      "model_forward_time": 1.7266793251037598,
      "step": 50726
    },
    {
      "epoch": 4.837608337402344e-05,
      "step": 50726,
      "training_step_time": 7.976219654083252
    },
    {
      "epoch": 4.837703704833984e-05,
      "model_forward_time": 1.7200422286987305,
      "step": 50727
    },
    {
      "epoch": 4.837703704833984e-05,
      "step": 50727,
      "training_step_time": 7.604732990264893
    },
    {
      "epoch": 4.837799072265625e-05,
      "model_forward_time": 1.808750867843628,
      "step": 50728
    },
    {
      "epoch": 4.837799072265625e-05,
      "step": 50728,
      "training_step_time": 8.846350908279419
    },
    {
      "epoch": 4.8378944396972655e-05,
      "model_forward_time": 1.7139222621917725,
      "step": 50729
    },
    {
      "epoch": 4.8378944396972655e-05,
      "step": 50729,
      "training_step_time": 7.930906057357788
    },
    {
      "epoch": 4.837989807128906e-05,
      "grad_norm": 0.1402432918548584,
      "learning_rate": 5.292662134270597e-06,
      "loss": 0.056,
      "step": 50730
    },
    {
      "epoch": 4.837989807128906e-05,
      "model_forward_time": 2.272850513458252,
      "step": 50730
    },
    {
      "dynamics_loss_avg": 0.07488944418728352,
      "epoch": 4.837989807128906e-05,
      "step": 50730
    },
    {
      "action_loss_avg": 0.0019612095842603595,
      "epoch": 4.837989807128906e-05,
      "step": 50730
    },
    {
      "epoch": 4.837989807128906e-05,
      "step": 50730,
      "training_step_time": 9.032800436019897
    },
    {
      "epoch": 4.838085174560547e-05,
      "model_forward_time": 1.7306902408599854,
      "step": 50731
    },
    {
      "epoch": 4.838085174560547e-05,
      "step": 50731,
      "training_step_time": 7.833181142807007
    },
    {
      "epoch": 4.8381805419921874e-05,
      "model_forward_time": 1.7317860126495361,
      "step": 50732
    },
    {
      "epoch": 4.8381805419921874e-05,
      "step": 50732,
      "training_step_time": 9.679486751556396
    },
    {
      "epoch": 4.838275909423828e-05,
      "model_forward_time": 1.7284986972808838,
      "step": 50733
    },
    {
      "epoch": 4.838275909423828e-05,
      "step": 50733,
      "training_step_time": 7.819122552871704
    },
    {
      "epoch": 4.838371276855469e-05,
      "model_forward_time": 1.7362136840820312,
      "step": 50734
    },
    {
      "epoch": 4.838371276855469e-05,
      "step": 50734,
      "training_step_time": 7.373477220535278
    },
    {
      "epoch": 4.838466644287109e-05,
      "model_forward_time": 1.7263858318328857,
      "step": 50735
    },
    {
      "epoch": 4.838466644287109e-05,
      "step": 50735,
      "training_step_time": 8.477583885192871
    },
    {
      "epoch": 4.83856201171875e-05,
      "model_forward_time": 1.7271571159362793,
      "step": 50736
    },
    {
      "epoch": 4.83856201171875e-05,
      "step": 50736,
      "training_step_time": 7.520537853240967
    },
    {
      "epoch": 4.838657379150391e-05,
      "model_forward_time": 1.7974340915679932,
      "step": 50737
    },
    {
      "epoch": 4.838657379150391e-05,
      "step": 50737,
      "training_step_time": 8.876676797866821
    },
    {
      "epoch": 4.838752746582031e-05,
      "model_forward_time": 1.7282488346099854,
      "step": 50738
    },
    {
      "epoch": 4.838752746582031e-05,
      "step": 50738,
      "training_step_time": 8.655287504196167
    },
    {
      "epoch": 4.8388481140136716e-05,
      "model_forward_time": 2.291583299636841,
      "step": 50739
    },
    {
      "epoch": 4.8388481140136716e-05,
      "step": 50739,
      "training_step_time": 7.959286212921143
    },
    {
      "epoch": 4.838943481445313e-05,
      "grad_norm": 0.10091273486614227,
      "learning_rate": 5.291011483341626e-06,
      "loss": 0.0539,
      "step": 50740
    },
    {
      "epoch": 4.838943481445313e-05,
      "model_forward_time": 1.7237961292266846,
      "step": 50740
    },
    {
      "dynamics_loss_avg": 0.042503093369305134,
      "epoch": 4.838943481445313e-05,
      "step": 50740
    },
    {
      "action_loss_avg": 0.0011264766799286008,
      "epoch": 4.838943481445313e-05,
      "step": 50740
    },
    {
      "epoch": 4.838943481445313e-05,
      "step": 50740,
      "training_step_time": 8.425503253936768
    },
    {
      "epoch": 4.839038848876953e-05,
      "model_forward_time": 1.7231316566467285,
      "step": 50741
    },
    {
      "epoch": 4.839038848876953e-05,
      "step": 50741,
      "training_step_time": 8.912944793701172
    },
    {
      "epoch": 4.8391342163085935e-05,
      "model_forward_time": 1.720550775527954,
      "step": 50742
    },
    {
      "epoch": 4.8391342163085935e-05,
      "step": 50742,
      "training_step_time": 7.823120594024658
    },
    {
      "epoch": 4.8392295837402346e-05,
      "model_forward_time": 1.7253572940826416,
      "step": 50743
    },
    {
      "epoch": 4.8392295837402346e-05,
      "step": 50743,
      "training_step_time": 7.845018625259399
    },
    {
      "epoch": 4.839324951171875e-05,
      "model_forward_time": 1.7273306846618652,
      "step": 50744
    },
    {
      "epoch": 4.839324951171875e-05,
      "step": 50744,
      "training_step_time": 7.944383859634399
    },
    {
      "epoch": 4.8394203186035154e-05,
      "model_forward_time": 1.7203974723815918,
      "step": 50745
    },
    {
      "epoch": 4.8394203186035154e-05,
      "step": 50745,
      "training_step_time": 7.7881176471710205
    },
    {
      "epoch": 4.8395156860351565e-05,
      "model_forward_time": 1.7757785320281982,
      "step": 50746
    },
    {
      "epoch": 4.8395156860351565e-05,
      "step": 50746,
      "training_step_time": 8.517648696899414
    },
    {
      "epoch": 4.839611053466797e-05,
      "model_forward_time": 1.7247352600097656,
      "step": 50747
    },
    {
      "epoch": 4.839611053466797e-05,
      "step": 50747,
      "training_step_time": 7.837808132171631
    },
    {
      "epoch": 4.839706420898437e-05,
      "model_forward_time": 2.278029203414917,
      "step": 50748
    },
    {
      "epoch": 4.839706420898437e-05,
      "step": 50748,
      "training_step_time": 7.821670055389404
    },
    {
      "epoch": 4.8398017883300783e-05,
      "model_forward_time": 1.7334465980529785,
      "step": 50749
    },
    {
      "epoch": 4.8398017883300783e-05,
      "step": 50749,
      "training_step_time": 8.50034213066101
    },
    {
      "epoch": 4.839897155761719e-05,
      "grad_norm": 0.08747100830078125,
      "learning_rate": 5.289360800588077e-06,
      "loss": 0.0526,
      "step": 50750
    },
    {
      "epoch": 4.839897155761719e-05,
      "model_forward_time": 1.716277837753296,
      "step": 50750
    },
    {
      "dynamics_loss_avg": 0.060860713478177786,
      "epoch": 4.839897155761719e-05,
      "step": 50750
    },
    {
      "action_loss_avg": 0.0017979667929466814,
      "epoch": 4.839897155761719e-05,
      "step": 50750
    },
    {
      "epoch": 4.839897155761719e-05,
      "step": 50750,
      "training_step_time": 7.857860326766968
    },
    {
      "epoch": 4.839992523193359e-05,
      "model_forward_time": 1.7255301475524902,
      "step": 50751
    },
    {
      "epoch": 4.839992523193359e-05,
      "step": 50751,
      "training_step_time": 7.8240814208984375
    },
    {
      "epoch": 4.840087890625e-05,
      "model_forward_time": 1.728583574295044,
      "step": 50752
    },
    {
      "epoch": 4.840087890625e-05,
      "step": 50752,
      "training_step_time": 7.928663969039917
    },
    {
      "epoch": 4.8401832580566406e-05,
      "model_forward_time": 1.7282006740570068,
      "step": 50753
    },
    {
      "epoch": 4.8401832580566406e-05,
      "step": 50753,
      "training_step_time": 8.025598764419556
    },
    {
      "epoch": 4.840278625488281e-05,
      "model_forward_time": 1.7371535301208496,
      "step": 50754
    },
    {
      "epoch": 4.840278625488281e-05,
      "step": 50754,
      "training_step_time": 7.892494201660156
    },
    {
      "epoch": 4.840373992919922e-05,
      "model_forward_time": 1.716717004776001,
      "step": 50755
    },
    {
      "epoch": 4.840373992919922e-05,
      "step": 50755,
      "training_step_time": 7.7797324657440186
    },
    {
      "epoch": 4.8404693603515625e-05,
      "model_forward_time": 1.7211427688598633,
      "step": 50756
    },
    {
      "epoch": 4.8404693603515625e-05,
      "step": 50756,
      "training_step_time": 7.946039438247681
    },
    {
      "epoch": 4.840564727783203e-05,
      "model_forward_time": 2.2805049419403076,
      "step": 50757
    },
    {
      "epoch": 4.840564727783203e-05,
      "step": 50757,
      "training_step_time": 7.885491609573364
    },
    {
      "epoch": 4.840660095214844e-05,
      "model_forward_time": 1.7339565753936768,
      "step": 50758
    },
    {
      "epoch": 4.840660095214844e-05,
      "step": 50758,
      "training_step_time": 9.03465461730957
    },
    {
      "epoch": 4.8407554626464844e-05,
      "model_forward_time": 1.7319393157958984,
      "step": 50759
    },
    {
      "epoch": 4.8407554626464844e-05,
      "step": 50759,
      "training_step_time": 8.889918565750122
    },
    {
      "epoch": 4.840850830078125e-05,
      "grad_norm": 0.16619467735290527,
      "learning_rate": 5.287710086190467e-06,
      "loss": 0.0525,
      "step": 50760
    },
    {
      "epoch": 4.840850830078125e-05,
      "model_forward_time": 1.7254564762115479,
      "step": 50760
    },
    {
      "dynamics_loss_avg": 0.0653862901031971,
      "epoch": 4.840850830078125e-05,
      "step": 50760
    },
    {
      "action_loss_avg": 0.0020791052520507947,
      "epoch": 4.840850830078125e-05,
      "step": 50760
    },
    {
      "epoch": 4.840850830078125e-05,
      "step": 50760,
      "training_step_time": 7.8770551681518555
    },
    {
      "epoch": 4.840946197509766e-05,
      "model_forward_time": 1.7322924137115479,
      "step": 50761
    },
    {
      "epoch": 4.840946197509766e-05,
      "step": 50761,
      "training_step_time": 7.888612985610962
    },
    {
      "epoch": 4.841041564941406e-05,
      "model_forward_time": 1.7397243976593018,
      "step": 50762
    },
    {
      "epoch": 4.841041564941406e-05,
      "step": 50762,
      "training_step_time": 9.01777958869934
    },
    {
      "epoch": 4.841136932373047e-05,
      "model_forward_time": 1.7269618511199951,
      "step": 50763
    },
    {
      "epoch": 4.841136932373047e-05,
      "step": 50763,
      "training_step_time": 7.82053804397583
    },
    {
      "epoch": 4.841232299804688e-05,
      "model_forward_time": 1.7354273796081543,
      "step": 50764
    },
    {
      "epoch": 4.841232299804688e-05,
      "step": 50764,
      "training_step_time": 8.078569650650024
    },
    {
      "epoch": 4.841327667236328e-05,
      "model_forward_time": 1.7314426898956299,
      "step": 50765
    },
    {
      "epoch": 4.841327667236328e-05,
      "step": 50765,
      "training_step_time": 7.844436883926392
    },
    {
      "epoch": 4.8414230346679686e-05,
      "model_forward_time": 2.2537007331848145,
      "step": 50766
    },
    {
      "epoch": 4.8414230346679686e-05,
      "step": 50766,
      "training_step_time": 9.626060009002686
    },
    {
      "epoch": 4.84151840209961e-05,
      "model_forward_time": 1.7283592224121094,
      "step": 50767
    },
    {
      "epoch": 4.84151840209961e-05,
      "step": 50767,
      "training_step_time": 9.863640308380127
    },
    {
      "epoch": 4.84161376953125e-05,
      "model_forward_time": 1.7331478595733643,
      "step": 50768
    },
    {
      "epoch": 4.84161376953125e-05,
      "step": 50768,
      "training_step_time": 8.461633920669556
    },
    {
      "epoch": 4.8417091369628905e-05,
      "model_forward_time": 1.7193214893341064,
      "step": 50769
    },
    {
      "epoch": 4.8417091369628905e-05,
      "step": 50769,
      "training_step_time": 8.641120433807373
    },
    {
      "epoch": 4.8418045043945315e-05,
      "grad_norm": 0.10310359299182892,
      "learning_rate": 5.286059340329315e-06,
      "loss": 0.0563,
      "step": 50770
    },
    {
      "epoch": 4.8418045043945315e-05,
      "model_forward_time": 1.7263424396514893,
      "step": 50770
    },
    {
      "dynamics_loss_avg": 0.07105825860053301,
      "epoch": 4.8418045043945315e-05,
      "step": 50770
    },
    {
      "action_loss_avg": 0.0016742969513870777,
      "epoch": 4.8418045043945315e-05,
      "step": 50770
    },
    {
      "epoch": 4.8418045043945315e-05,
      "step": 50770,
      "training_step_time": 7.785855054855347
    },
    {
      "epoch": 4.841899871826172e-05,
      "model_forward_time": 1.7263751029968262,
      "step": 50771
    },
    {
      "epoch": 4.841899871826172e-05,
      "step": 50771,
      "training_step_time": 8.008638381958008
    },
    {
      "epoch": 4.8419952392578124e-05,
      "model_forward_time": 1.7155959606170654,
      "step": 50772
    },
    {
      "epoch": 4.8419952392578124e-05,
      "step": 50772,
      "training_step_time": 7.693877220153809
    },
    {
      "epoch": 4.8420906066894534e-05,
      "model_forward_time": 1.7254815101623535,
      "step": 50773
    },
    {
      "epoch": 4.8420906066894534e-05,
      "step": 50773,
      "training_step_time": 7.459655284881592
    },
    {
      "epoch": 4.842185974121094e-05,
      "model_forward_time": 1.7682411670684814,
      "step": 50774
    },
    {
      "epoch": 4.842185974121094e-05,
      "step": 50774,
      "training_step_time": 7.917558670043945
    },
    {
      "epoch": 4.842281341552734e-05,
      "model_forward_time": 2.279252767562866,
      "step": 50775
    },
    {
      "epoch": 4.842281341552734e-05,
      "step": 50775,
      "training_step_time": 9.138525247573853
    },
    {
      "epoch": 4.842376708984375e-05,
      "model_forward_time": 1.7191886901855469,
      "step": 50776
    },
    {
      "epoch": 4.842376708984375e-05,
      "step": 50776,
      "training_step_time": 8.531230688095093
    },
    {
      "epoch": 4.842472076416016e-05,
      "model_forward_time": 1.7265315055847168,
      "step": 50777
    },
    {
      "epoch": 4.842472076416016e-05,
      "step": 50777,
      "training_step_time": 7.904367446899414
    },
    {
      "epoch": 4.842567443847656e-05,
      "model_forward_time": 1.7247068881988525,
      "step": 50778
    },
    {
      "epoch": 4.842567443847656e-05,
      "step": 50778,
      "training_step_time": 7.796996355056763
    },
    {
      "epoch": 4.842662811279297e-05,
      "model_forward_time": 1.7236275672912598,
      "step": 50779
    },
    {
      "epoch": 4.842662811279297e-05,
      "step": 50779,
      "training_step_time": 7.835129499435425
    },
    {
      "epoch": 4.8427581787109376e-05,
      "grad_norm": 0.10954651236534119,
      "learning_rate": 5.2844085631851415e-06,
      "loss": 0.0555,
      "step": 50780
    },
    {
      "epoch": 4.8427581787109376e-05,
      "model_forward_time": 1.7107906341552734,
      "step": 50780
    },
    {
      "dynamics_loss_avg": 0.0549815284088254,
      "epoch": 4.8427581787109376e-05,
      "step": 50780
    },
    {
      "action_loss_avg": 0.0016673196165356784,
      "epoch": 4.8427581787109376e-05,
      "step": 50780
    },
    {
      "epoch": 4.8427581787109376e-05,
      "step": 50780,
      "training_step_time": 7.891214370727539
    },
    {
      "epoch": 4.842853546142578e-05,
      "model_forward_time": 1.7235450744628906,
      "step": 50781
    },
    {
      "epoch": 4.842853546142578e-05,
      "step": 50781,
      "training_step_time": 7.966073036193848
    },
    {
      "epoch": 4.8429489135742184e-05,
      "model_forward_time": 1.7222304344177246,
      "step": 50782
    },
    {
      "epoch": 4.8429489135742184e-05,
      "step": 50782,
      "training_step_time": 7.534372329711914
    },
    {
      "epoch": 4.8430442810058595e-05,
      "model_forward_time": 1.7578248977661133,
      "step": 50783
    },
    {
      "epoch": 4.8430442810058595e-05,
      "step": 50783,
      "training_step_time": 7.9075658321380615
    },
    {
      "epoch": 4.8431396484375e-05,
      "model_forward_time": 2.282151699066162,
      "step": 50784
    },
    {
      "epoch": 4.8431396484375e-05,
      "step": 50784,
      "training_step_time": 7.7868430614471436
    },
    {
      "epoch": 4.84323501586914e-05,
      "model_forward_time": 1.7251460552215576,
      "step": 50785
    },
    {
      "epoch": 4.84323501586914e-05,
      "step": 50785,
      "training_step_time": 9.026215076446533
    },
    {
      "epoch": 4.8433303833007814e-05,
      "model_forward_time": 1.7101078033447266,
      "step": 50786
    },
    {
      "epoch": 4.8433303833007814e-05,
      "step": 50786,
      "training_step_time": 9.722371339797974
    },
    {
      "epoch": 4.843425750732422e-05,
      "model_forward_time": 1.7231965065002441,
      "step": 50787
    },
    {
      "epoch": 4.843425750732422e-05,
      "step": 50787,
      "training_step_time": 8.975555181503296
    },
    {
      "epoch": 4.843521118164062e-05,
      "model_forward_time": 1.728938341140747,
      "step": 50788
    },
    {
      "epoch": 4.843521118164062e-05,
      "step": 50788,
      "training_step_time": 8.470102548599243
    },
    {
      "epoch": 4.843616485595703e-05,
      "model_forward_time": 1.7329678535461426,
      "step": 50789
    },
    {
      "epoch": 4.843616485595703e-05,
      "step": 50789,
      "training_step_time": 8.956347465515137
    },
    {
      "epoch": 4.843711853027344e-05,
      "grad_norm": 0.08409823477268219,
      "learning_rate": 5.282757754938476e-06,
      "loss": 0.0553,
      "step": 50790
    },
    {
      "epoch": 4.843711853027344e-05,
      "model_forward_time": 1.7235984802246094,
      "step": 50790
    },
    {
      "dynamics_loss_avg": 0.048098836187273265,
      "epoch": 4.843711853027344e-05,
      "step": 50790
    },
    {
      "action_loss_avg": 0.0011508060328196735,
      "epoch": 4.843711853027344e-05,
      "step": 50790
    },
    {
      "epoch": 4.843711853027344e-05,
      "step": 50790,
      "training_step_time": 8.465250253677368
    },
    {
      "epoch": 4.843807220458984e-05,
      "model_forward_time": 1.7314999103546143,
      "step": 50791
    },
    {
      "epoch": 4.843807220458984e-05,
      "step": 50791,
      "training_step_time": 7.448371410369873
    },
    {
      "epoch": 4.843902587890625e-05,
      "model_forward_time": 1.7742650508880615,
      "step": 50792
    },
    {
      "epoch": 4.843902587890625e-05,
      "step": 50792,
      "training_step_time": 7.813241243362427
    },
    {
      "epoch": 4.8439979553222656e-05,
      "model_forward_time": 2.282050371170044,
      "step": 50793
    },
    {
      "epoch": 4.8439979553222656e-05,
      "step": 50793,
      "training_step_time": 7.841481685638428
    },
    {
      "epoch": 4.844093322753906e-05,
      "model_forward_time": 1.7312850952148438,
      "step": 50794
    },
    {
      "epoch": 4.844093322753906e-05,
      "step": 50794,
      "training_step_time": 7.824781656265259
    },
    {
      "epoch": 4.844188690185547e-05,
      "model_forward_time": 1.7407703399658203,
      "step": 50795
    },
    {
      "epoch": 4.844188690185547e-05,
      "step": 50795,
      "training_step_time": 9.54443621635437
    },
    {
      "epoch": 4.8442840576171874e-05,
      "model_forward_time": 1.7279963493347168,
      "step": 50796
    },
    {
      "epoch": 4.8442840576171874e-05,
      "step": 50796,
      "training_step_time": 8.897548913955688
    },
    {
      "epoch": 4.844379425048828e-05,
      "model_forward_time": 1.7074460983276367,
      "step": 50797
    },
    {
      "epoch": 4.844379425048828e-05,
      "step": 50797,
      "training_step_time": 8.43326210975647
    },
    {
      "epoch": 4.844474792480469e-05,
      "model_forward_time": 1.728424310684204,
      "step": 50798
    },
    {
      "epoch": 4.844474792480469e-05,
      "step": 50798,
      "training_step_time": 7.921016693115234
    },
    {
      "epoch": 4.844570159912109e-05,
      "model_forward_time": 1.7365524768829346,
      "step": 50799
    },
    {
      "epoch": 4.844570159912109e-05,
      "step": 50799,
      "training_step_time": 7.910540342330933
    },
    {
      "epoch": 4.84466552734375e-05,
      "grad_norm": 0.11494386196136475,
      "learning_rate": 5.2811069157698495e-06,
      "loss": 0.0574,
      "step": 50800
    },
    {
      "epoch": 4.84466552734375e-05,
      "model_forward_time": 1.730081558227539,
      "step": 50800
    },
    {
      "dynamics_loss_avg": 0.05232552215456963,
      "epoch": 4.84466552734375e-05,
      "step": 50800
    },
    {
      "action_loss_avg": 0.0018570707354228944,
      "epoch": 4.84466552734375e-05,
      "step": 50800
    },
    {
      "epoch": 4.84466552734375e-05,
      "step": 50800,
      "training_step_time": 7.7397780418396
    },
    {
      "epoch": 4.844760894775391e-05,
      "model_forward_time": 1.7563984394073486,
      "step": 50801
    },
    {
      "epoch": 4.844760894775391e-05,
      "step": 50801,
      "training_step_time": 7.833385705947876
    },
    {
      "epoch": 4.844856262207031e-05,
      "model_forward_time": 2.278904438018799,
      "step": 50802
    },
    {
      "epoch": 4.844856262207031e-05,
      "step": 50802,
      "training_step_time": 7.878448724746704
    },
    {
      "epoch": 4.8449516296386716e-05,
      "model_forward_time": 1.7325499057769775,
      "step": 50803
    },
    {
      "epoch": 4.8449516296386716e-05,
      "step": 50803,
      "training_step_time": 8.254481792449951
    },
    {
      "epoch": 4.845046997070313e-05,
      "model_forward_time": 1.731107234954834,
      "step": 50804
    },
    {
      "epoch": 4.845046997070313e-05,
      "step": 50804,
      "training_step_time": 8.875128030776978
    },
    {
      "epoch": 4.845142364501953e-05,
      "model_forward_time": 1.7369437217712402,
      "step": 50805
    },
    {
      "epoch": 4.845142364501953e-05,
      "step": 50805,
      "training_step_time": 7.874891996383667
    },
    {
      "epoch": 4.8452377319335935e-05,
      "model_forward_time": 1.7327861785888672,
      "step": 50806
    },
    {
      "epoch": 4.8452377319335935e-05,
      "step": 50806,
      "training_step_time": 7.758485794067383
    },
    {
      "epoch": 4.8453330993652346e-05,
      "model_forward_time": 1.734266996383667,
      "step": 50807
    },
    {
      "epoch": 4.8453330993652346e-05,
      "step": 50807,
      "training_step_time": 7.864790916442871
    },
    {
      "epoch": 4.845428466796875e-05,
      "model_forward_time": 1.7390038967132568,
      "step": 50808
    },
    {
      "epoch": 4.845428466796875e-05,
      "step": 50808,
      "training_step_time": 7.8738086223602295
    },
    {
      "epoch": 4.8455238342285154e-05,
      "model_forward_time": 1.736513376235962,
      "step": 50809
    },
    {
      "epoch": 4.8455238342285154e-05,
      "step": 50809,
      "training_step_time": 7.84615159034729
    },
    {
      "epoch": 4.8456192016601565e-05,
      "grad_norm": 0.1320076286792755,
      "learning_rate": 5.27945604585979e-06,
      "loss": 0.0562,
      "step": 50810
    },
    {
      "epoch": 4.8456192016601565e-05,
      "model_forward_time": 1.734917163848877,
      "step": 50810
    },
    {
      "dynamics_loss_avg": 0.04859923953190446,
      "epoch": 4.8456192016601565e-05,
      "step": 50810
    },
    {
      "action_loss_avg": 0.0015147743630222975,
      "epoch": 4.8456192016601565e-05,
      "step": 50810
    },
    {
      "epoch": 4.8456192016601565e-05,
      "step": 50810,
      "training_step_time": 7.822940111160278
    },
    {
      "epoch": 4.845714569091797e-05,
      "model_forward_time": 2.276310443878174,
      "step": 50811
    },
    {
      "epoch": 4.845714569091797e-05,
      "step": 50811,
      "training_step_time": 7.808251619338989
    },
    {
      "epoch": 4.845809936523437e-05,
      "model_forward_time": 1.7317955493927002,
      "step": 50812
    },
    {
      "epoch": 4.845809936523437e-05,
      "step": 50812,
      "training_step_time": 7.85315465927124
    },
    {
      "epoch": 4.8459053039550784e-05,
      "model_forward_time": 1.7424583435058594,
      "step": 50813
    },
    {
      "epoch": 4.8459053039550784e-05,
      "step": 50813,
      "training_step_time": 8.083020210266113
    },
    {
      "epoch": 4.846000671386719e-05,
      "model_forward_time": 1.7311947345733643,
      "step": 50814
    },
    {
      "epoch": 4.846000671386719e-05,
      "step": 50814,
      "training_step_time": 8.918832063674927
    },
    {
      "epoch": 4.846096038818359e-05,
      "model_forward_time": 1.7422988414764404,
      "step": 50815
    },
    {
      "epoch": 4.846096038818359e-05,
      "step": 50815,
      "training_step_time": 7.871874094009399
    },
    {
      "epoch": 4.84619140625e-05,
      "model_forward_time": 1.7319724559783936,
      "step": 50816
    },
    {
      "epoch": 4.84619140625e-05,
      "step": 50816,
      "training_step_time": 8.567484855651855
    },
    {
      "epoch": 4.8462867736816406e-05,
      "model_forward_time": 1.733098030090332,
      "step": 50817
    },
    {
      "epoch": 4.8462867736816406e-05,
      "step": 50817,
      "training_step_time": 7.911259651184082
    },
    {
      "epoch": 4.846382141113281e-05,
      "model_forward_time": 1.7256157398223877,
      "step": 50818
    },
    {
      "epoch": 4.846382141113281e-05,
      "step": 50818,
      "training_step_time": 7.868175268173218
    },
    {
      "epoch": 4.846477508544922e-05,
      "model_forward_time": 1.7379138469696045,
      "step": 50819
    },
    {
      "epoch": 4.846477508544922e-05,
      "step": 50819,
      "training_step_time": 7.90481424331665
    },
    {
      "epoch": 4.8465728759765625e-05,
      "grad_norm": 0.11337389796972275,
      "learning_rate": 5.27780514538884e-06,
      "loss": 0.0565,
      "step": 50820
    },
    {
      "epoch": 4.8465728759765625e-05,
      "model_forward_time": 2.3640637397766113,
      "step": 50820
    },
    {
      "dynamics_loss_avg": 0.050084117241203784,
      "epoch": 4.8465728759765625e-05,
      "step": 50820
    },
    {
      "action_loss_avg": 0.0010704247717512772,
      "epoch": 4.8465728759765625e-05,
      "step": 50820
    },
    {
      "epoch": 4.8465728759765625e-05,
      "step": 50820,
      "training_step_time": 7.896944046020508
    },
    {
      "epoch": 4.846668243408203e-05,
      "model_forward_time": 1.7267124652862549,
      "step": 50821
    },
    {
      "epoch": 4.846668243408203e-05,
      "step": 50821,
      "training_step_time": 8.709438800811768
    },
    {
      "epoch": 4.846763610839844e-05,
      "model_forward_time": 1.734438419342041,
      "step": 50822
    },
    {
      "epoch": 4.846763610839844e-05,
      "step": 50822,
      "training_step_time": 9.122148513793945
    },
    {
      "epoch": 4.8468589782714844e-05,
      "model_forward_time": 1.7320387363433838,
      "step": 50823
    },
    {
      "epoch": 4.8468589782714844e-05,
      "step": 50823,
      "training_step_time": 7.922786235809326
    },
    {
      "epoch": 4.846954345703125e-05,
      "model_forward_time": 1.7396321296691895,
      "step": 50824
    },
    {
      "epoch": 4.846954345703125e-05,
      "step": 50824,
      "training_step_time": 7.734821319580078
    },
    {
      "epoch": 4.847049713134766e-05,
      "model_forward_time": 1.7395877838134766,
      "step": 50825
    },
    {
      "epoch": 4.847049713134766e-05,
      "step": 50825,
      "training_step_time": 9.34426474571228
    },
    {
      "epoch": 4.847145080566406e-05,
      "model_forward_time": 1.7262938022613525,
      "step": 50826
    },
    {
      "epoch": 4.847145080566406e-05,
      "step": 50826,
      "training_step_time": 7.963277816772461
    },
    {
      "epoch": 4.847240447998047e-05,
      "model_forward_time": 1.719944953918457,
      "step": 50827
    },
    {
      "epoch": 4.847240447998047e-05,
      "step": 50827,
      "training_step_time": 7.944932699203491
    },
    {
      "epoch": 4.847335815429688e-05,
      "model_forward_time": 1.73508882522583,
      "step": 50828
    },
    {
      "epoch": 4.847335815429688e-05,
      "step": 50828,
      "training_step_time": 8.456075668334961
    },
    {
      "epoch": 4.847431182861328e-05,
      "model_forward_time": 2.3515233993530273,
      "step": 50829
    },
    {
      "epoch": 4.847431182861328e-05,
      "step": 50829,
      "training_step_time": 7.9600701332092285
    },
    {
      "epoch": 4.8475265502929686e-05,
      "grad_norm": 0.09284619987010956,
      "learning_rate": 5.276154214537537e-06,
      "loss": 0.0599,
      "step": 50830
    },
    {
      "epoch": 4.8475265502929686e-05,
      "model_forward_time": 1.703599452972412,
      "step": 50830
    },
    {
      "dynamics_loss_avg": 0.07171178851276636,
      "epoch": 4.8475265502929686e-05,
      "step": 50830
    },
    {
      "action_loss_avg": 0.0015344996732892468,
      "epoch": 4.8475265502929686e-05,
      "step": 50830
    },
    {
      "epoch": 4.8475265502929686e-05,
      "step": 50830,
      "training_step_time": 7.943132162094116
    },
    {
      "epoch": 4.84762191772461e-05,
      "model_forward_time": 1.715045690536499,
      "step": 50831
    },
    {
      "epoch": 4.84762191772461e-05,
      "step": 50831,
      "training_step_time": 9.484562158584595
    },
    {
      "epoch": 4.84771728515625e-05,
      "model_forward_time": 1.711057186126709,
      "step": 50832
    },
    {
      "epoch": 4.84771728515625e-05,
      "step": 50832,
      "training_step_time": 9.138736724853516
    },
    {
      "epoch": 4.8478126525878905e-05,
      "model_forward_time": 1.715782880783081,
      "step": 50833
    },
    {
      "epoch": 4.8478126525878905e-05,
      "step": 50833,
      "training_step_time": 9.666791677474976
    },
    {
      "epoch": 4.8479080200195316e-05,
      "model_forward_time": 1.7096045017242432,
      "step": 50834
    },
    {
      "epoch": 4.8479080200195316e-05,
      "step": 50834,
      "training_step_time": 7.922996520996094
    },
    {
      "epoch": 4.848003387451172e-05,
      "model_forward_time": 1.7119388580322266,
      "step": 50835
    },
    {
      "epoch": 4.848003387451172e-05,
      "step": 50835,
      "training_step_time": 7.916644811630249
    },
    {
      "epoch": 4.8480987548828124e-05,
      "model_forward_time": 1.720104694366455,
      "step": 50836
    },
    {
      "epoch": 4.8480987548828124e-05,
      "step": 50836,
      "training_step_time": 7.89661169052124
    },
    {
      "epoch": 4.8481941223144534e-05,
      "model_forward_time": 1.7248454093933105,
      "step": 50837
    },
    {
      "epoch": 4.8481941223144534e-05,
      "step": 50837,
      "training_step_time": 8.589365720748901
    },
    {
      "epoch": 4.848289489746094e-05,
      "model_forward_time": 2.2850987911224365,
      "step": 50838
    },
    {
      "epoch": 4.848289489746094e-05,
      "step": 50838,
      "training_step_time": 7.8828394412994385
    },
    {
      "epoch": 4.848384857177734e-05,
      "model_forward_time": 1.7291474342346191,
      "step": 50839
    },
    {
      "epoch": 4.848384857177734e-05,
      "step": 50839,
      "training_step_time": 8.92548680305481
    },
    {
      "epoch": 4.848480224609375e-05,
      "grad_norm": 0.12281998246908188,
      "learning_rate": 5.274503253486421e-06,
      "loss": 0.0565,
      "step": 50840
    },
    {
      "epoch": 4.848480224609375e-05,
      "model_forward_time": 1.7197349071502686,
      "step": 50840
    },
    {
      "dynamics_loss_avg": 0.05626606047153473,
      "epoch": 4.848480224609375e-05,
      "step": 50840
    },
    {
      "action_loss_avg": 0.0024049794767051933,
      "epoch": 4.848480224609375e-05,
      "step": 50840
    },
    {
      "epoch": 4.848480224609375e-05,
      "step": 50840,
      "training_step_time": 8.331510543823242
    },
    {
      "epoch": 4.848575592041016e-05,
      "model_forward_time": 1.707521915435791,
      "step": 50841
    },
    {
      "epoch": 4.848575592041016e-05,
      "step": 50841,
      "training_step_time": 9.072293519973755
    },
    {
      "epoch": 4.848670959472656e-05,
      "model_forward_time": 1.7128410339355469,
      "step": 50842
    },
    {
      "epoch": 4.848670959472656e-05,
      "step": 50842,
      "training_step_time": 7.789653539657593
    },
    {
      "epoch": 4.848766326904297e-05,
      "model_forward_time": 1.7194111347198486,
      "step": 50843
    },
    {
      "epoch": 4.848766326904297e-05,
      "step": 50843,
      "training_step_time": 8.831254005432129
    },
    {
      "epoch": 4.8488616943359376e-05,
      "model_forward_time": 1.7092618942260742,
      "step": 50844
    },
    {
      "epoch": 4.8488616943359376e-05,
      "step": 50844,
      "training_step_time": 8.76457953453064
    },
    {
      "epoch": 4.848957061767578e-05,
      "model_forward_time": 1.711700677871704,
      "step": 50845
    },
    {
      "epoch": 4.848957061767578e-05,
      "step": 50845,
      "training_step_time": 7.85338568687439
    },
    {
      "epoch": 4.8490524291992184e-05,
      "model_forward_time": 1.7066905498504639,
      "step": 50846
    },
    {
      "epoch": 4.8490524291992184e-05,
      "step": 50846,
      "training_step_time": 7.91020131111145
    },
    {
      "epoch": 4.8491477966308595e-05,
      "model_forward_time": 2.2859771251678467,
      "step": 50847
    },
    {
      "epoch": 4.8491477966308595e-05,
      "step": 50847,
      "training_step_time": 8.490370750427246
    },
    {
      "epoch": 4.8492431640625e-05,
      "model_forward_time": 1.7298943996429443,
      "step": 50848
    },
    {
      "epoch": 4.8492431640625e-05,
      "step": 50848,
      "training_step_time": 7.884947776794434
    },
    {
      "epoch": 4.84933853149414e-05,
      "model_forward_time": 1.7059319019317627,
      "step": 50849
    },
    {
      "epoch": 4.84933853149414e-05,
      "step": 50849,
      "training_step_time": 8.535848140716553
    },
    {
      "epoch": 4.8494338989257814e-05,
      "grad_norm": 0.11383729428052902,
      "learning_rate": 5.272852262416046e-06,
      "loss": 0.0565,
      "step": 50850
    },
    {
      "epoch": 4.8494338989257814e-05,
      "model_forward_time": 1.7126877307891846,
      "step": 50850
    },
    {
      "dynamics_loss_avg": 0.0645672908052802,
      "epoch": 4.8494338989257814e-05,
      "step": 50850
    },
    {
      "action_loss_avg": 0.002851452195318416,
      "epoch": 4.8494338989257814e-05,
      "step": 50850
    },
    {
      "epoch": 4.8494338989257814e-05,
      "step": 50850,
      "training_step_time": 8.786363363265991
    },
    {
      "epoch": 4.849529266357422e-05,
      "model_forward_time": 1.7165496349334717,
      "step": 50851
    },
    {
      "epoch": 4.849529266357422e-05,
      "step": 50851,
      "training_step_time": 8.759695291519165
    },
    {
      "epoch": 4.849624633789062e-05,
      "model_forward_time": 1.7171955108642578,
      "step": 50852
    },
    {
      "epoch": 4.849624633789062e-05,
      "step": 50852,
      "training_step_time": 7.892094373703003
    },
    {
      "epoch": 4.849720001220703e-05,
      "model_forward_time": 1.70943284034729,
      "step": 50853
    },
    {
      "epoch": 4.849720001220703e-05,
      "step": 50853,
      "training_step_time": 8.518531560897827
    },
    {
      "epoch": 4.849815368652344e-05,
      "model_forward_time": 1.71517014503479,
      "step": 50854
    },
    {
      "epoch": 4.849815368652344e-05,
      "step": 50854,
      "training_step_time": 7.841346740722656
    },
    {
      "epoch": 4.849910736083984e-05,
      "model_forward_time": 1.7094385623931885,
      "step": 50855
    },
    {
      "epoch": 4.849910736083984e-05,
      "step": 50855,
      "training_step_time": 7.876881122589111
    },
    {
      "epoch": 4.850006103515625e-05,
      "model_forward_time": 2.28367280960083,
      "step": 50856
    },
    {
      "epoch": 4.850006103515625e-05,
      "step": 50856,
      "training_step_time": 8.509894132614136
    },
    {
      "epoch": 4.8501014709472656e-05,
      "model_forward_time": 1.7209208011627197,
      "step": 50857
    },
    {
      "epoch": 4.8501014709472656e-05,
      "step": 50857,
      "training_step_time": 7.883933067321777
    },
    {
      "epoch": 4.850196838378906e-05,
      "model_forward_time": 1.7079627513885498,
      "step": 50858
    },
    {
      "epoch": 4.850196838378906e-05,
      "step": 50858,
      "training_step_time": 8.435834884643555
    },
    {
      "epoch": 4.850292205810547e-05,
      "model_forward_time": 1.72102689743042,
      "step": 50859
    },
    {
      "epoch": 4.850292205810547e-05,
      "step": 50859,
      "training_step_time": 10.094002962112427
    },
    {
      "epoch": 4.8503875732421874e-05,
      "grad_norm": 0.09606266021728516,
      "learning_rate": 5.271201241506956e-06,
      "loss": 0.0572,
      "step": 50860
    },
    {
      "epoch": 4.8503875732421874e-05,
      "model_forward_time": 1.7165956497192383,
      "step": 50860
    },
    {
      "dynamics_loss_avg": 0.057605483662337066,
      "epoch": 4.8503875732421874e-05,
      "step": 50860
    },
    {
      "action_loss_avg": 0.002120556798763573,
      "epoch": 4.8503875732421874e-05,
      "step": 50860
    },
    {
      "epoch": 4.8503875732421874e-05,
      "step": 50860,
      "training_step_time": 8.590829610824585
    },
    {
      "epoch": 4.850482940673828e-05,
      "model_forward_time": 1.721924066543579,
      "step": 50861
    },
    {
      "epoch": 4.850482940673828e-05,
      "step": 50861,
      "training_step_time": 7.9204323291778564
    },
    {
      "epoch": 4.850578308105469e-05,
      "model_forward_time": 1.726888656616211,
      "step": 50862
    },
    {
      "epoch": 4.850578308105469e-05,
      "step": 50862,
      "training_step_time": 8.953716039657593
    },
    {
      "epoch": 4.850673675537109e-05,
      "model_forward_time": 1.7292184829711914,
      "step": 50863
    },
    {
      "epoch": 4.850673675537109e-05,
      "step": 50863,
      "training_step_time": 7.904359340667725
    },
    {
      "epoch": 4.85076904296875e-05,
      "model_forward_time": 1.734473705291748,
      "step": 50864
    },
    {
      "epoch": 4.85076904296875e-05,
      "step": 50864,
      "training_step_time": 7.965736150741577
    },
    {
      "epoch": 4.850864410400391e-05,
      "model_forward_time": 2.2994930744171143,
      "step": 50865
    },
    {
      "epoch": 4.850864410400391e-05,
      "step": 50865,
      "training_step_time": 7.651251792907715
    },
    {
      "epoch": 4.850959777832031e-05,
      "model_forward_time": 1.7373015880584717,
      "step": 50866
    },
    {
      "epoch": 4.850959777832031e-05,
      "step": 50866,
      "training_step_time": 7.912594556808472
    },
    {
      "epoch": 4.8510551452636716e-05,
      "model_forward_time": 1.7739074230194092,
      "step": 50867
    },
    {
      "epoch": 4.8510551452636716e-05,
      "step": 50867,
      "training_step_time": 7.882333993911743
    },
    {
      "epoch": 4.851150512695313e-05,
      "model_forward_time": 1.7346735000610352,
      "step": 50868
    },
    {
      "epoch": 4.851150512695313e-05,
      "step": 50868,
      "training_step_time": 7.93506646156311
    },
    {
      "epoch": 4.851245880126953e-05,
      "model_forward_time": 1.3686697483062744,
      "step": 50869
    },
    {
      "epoch": 4.851245880126953e-05,
      "step": 50869,
      "training_step_time": 8.071651697158813
    },
    {
      "epoch": 4.8513412475585935e-05,
      "grad_norm": 0.1361132711172104,
      "learning_rate": 5.269550190939705e-06,
      "loss": 0.0565,
      "step": 50870
    },
    {
      "epoch": 4.8513412475585935e-05,
      "model_forward_time": 1.734726905822754,
      "step": 50870
    },
    {
      "dynamics_loss_avg": 0.04913940466940403,
      "epoch": 4.8513412475585935e-05,
      "step": 50870
    },
    {
      "action_loss_avg": 0.0021535839478019625,
      "epoch": 4.8513412475585935e-05,
      "step": 50870
    },
    {
      "epoch": 4.8513412475585935e-05,
      "step": 50870,
      "training_step_time": 8.544983148574829
    },
    {
      "epoch": 4.8514366149902346e-05,
      "model_forward_time": 1.7296535968780518,
      "step": 50871
    },
    {
      "epoch": 4.8514366149902346e-05,
      "step": 50871,
      "training_step_time": 8.558636665344238
    },
    {
      "epoch": 4.851531982421875e-05,
      "model_forward_time": 1.72605562210083,
      "step": 50872
    },
    {
      "epoch": 4.851531982421875e-05,
      "step": 50872,
      "training_step_time": 7.88277268409729
    },
    {
      "epoch": 4.8516273498535154e-05,
      "model_forward_time": 1.7245042324066162,
      "step": 50873
    },
    {
      "epoch": 4.8516273498535154e-05,
      "step": 50873,
      "training_step_time": 7.899837017059326
    },
    {
      "epoch": 4.8517227172851565e-05,
      "model_forward_time": 2.283109664916992,
      "step": 50874
    },
    {
      "epoch": 4.8517227172851565e-05,
      "step": 50874,
      "training_step_time": 7.7017199993133545
    },
    {
      "epoch": 4.851818084716797e-05,
      "model_forward_time": 1.720911979675293,
      "step": 50875
    },
    {
      "epoch": 4.851818084716797e-05,
      "step": 50875,
      "training_step_time": 7.865887880325317
    },
    {
      "epoch": 4.851913452148437e-05,
      "model_forward_time": 1.7695093154907227,
      "step": 50876
    },
    {
      "epoch": 4.851913452148437e-05,
      "step": 50876,
      "training_step_time": 8.880460023880005
    },
    {
      "epoch": 4.8520088195800784e-05,
      "model_forward_time": 1.7205002307891846,
      "step": 50877
    },
    {
      "epoch": 4.8520088195800784e-05,
      "step": 50877,
      "training_step_time": 8.93239951133728
    },
    {
      "epoch": 4.852104187011719e-05,
      "model_forward_time": 1.7333600521087646,
      "step": 50878
    },
    {
      "epoch": 4.852104187011719e-05,
      "step": 50878,
      "training_step_time": 7.905550479888916
    },
    {
      "epoch": 4.852199554443359e-05,
      "model_forward_time": 1.7294347286224365,
      "step": 50879
    },
    {
      "epoch": 4.852199554443359e-05,
      "step": 50879,
      "training_step_time": 8.4768226146698
    },
    {
      "epoch": 4.852294921875e-05,
      "grad_norm": 0.11271622031927109,
      "learning_rate": 5.267899110894852e-06,
      "loss": 0.0593,
      "step": 50880
    },
    {
      "epoch": 4.852294921875e-05,
      "model_forward_time": 1.72395920753479,
      "step": 50880
    },
    {
      "dynamics_loss_avg": 0.05806230343878269,
      "epoch": 4.852294921875e-05,
      "step": 50880
    },
    {
      "action_loss_avg": 0.0016418514191173018,
      "epoch": 4.852294921875e-05,
      "step": 50880
    },
    {
      "epoch": 4.852294921875e-05,
      "step": 50880,
      "training_step_time": 9.33532428741455
    },
    {
      "epoch": 4.8523902893066406e-05,
      "model_forward_time": 1.7311408519744873,
      "step": 50881
    },
    {
      "epoch": 4.8523902893066406e-05,
      "step": 50881,
      "training_step_time": 8.984598159790039
    },
    {
      "epoch": 4.852485656738281e-05,
      "model_forward_time": 1.735919713973999,
      "step": 50882
    },
    {
      "epoch": 4.852485656738281e-05,
      "step": 50882,
      "training_step_time": 8.933292150497437
    },
    {
      "epoch": 4.852581024169922e-05,
      "model_forward_time": 2.297785520553589,
      "step": 50883
    },
    {
      "epoch": 4.852581024169922e-05,
      "step": 50883,
      "training_step_time": 7.939711570739746
    },
    {
      "epoch": 4.8526763916015625e-05,
      "model_forward_time": 1.7393832206726074,
      "step": 50884
    },
    {
      "epoch": 4.8526763916015625e-05,
      "step": 50884,
      "training_step_time": 7.642035007476807
    },
    {
      "epoch": 4.852771759033203e-05,
      "model_forward_time": 1.750657558441162,
      "step": 50885
    },
    {
      "epoch": 4.852771759033203e-05,
      "step": 50885,
      "training_step_time": 7.815711259841919
    },
    {
      "epoch": 4.852867126464844e-05,
      "model_forward_time": 1.7390210628509521,
      "step": 50886
    },
    {
      "epoch": 4.852867126464844e-05,
      "step": 50886,
      "training_step_time": 7.904641151428223
    },
    {
      "epoch": 4.8529624938964844e-05,
      "model_forward_time": 1.7457449436187744,
      "step": 50887
    },
    {
      "epoch": 4.8529624938964844e-05,
      "step": 50887,
      "training_step_time": 8.125175476074219
    },
    {
      "epoch": 4.853057861328125e-05,
      "model_forward_time": 1.737882137298584,
      "step": 50888
    },
    {
      "epoch": 4.853057861328125e-05,
      "step": 50888,
      "training_step_time": 8.88350510597229
    },
    {
      "epoch": 4.853153228759766e-05,
      "model_forward_time": 1.7400686740875244,
      "step": 50889
    },
    {
      "epoch": 4.853153228759766e-05,
      "step": 50889,
      "training_step_time": 9.171794414520264
    },
    {
      "epoch": 4.853248596191406e-05,
      "grad_norm": 0.10521769523620605,
      "learning_rate": 5.2662480015529546e-06,
      "loss": 0.0574,
      "step": 50890
    },
    {
      "epoch": 4.853248596191406e-05,
      "model_forward_time": 1.7144489288330078,
      "step": 50890
    },
    {
      "dynamics_loss_avg": 0.050917374528944494,
      "epoch": 4.853248596191406e-05,
      "step": 50890
    },
    {
      "action_loss_avg": 0.001509105833247304,
      "epoch": 4.853248596191406e-05,
      "step": 50890
    },
    {
      "epoch": 4.853248596191406e-05,
      "step": 50890,
      "training_step_time": 7.861848831176758
    },
    {
      "epoch": 4.853343963623047e-05,
      "model_forward_time": 1.7389791011810303,
      "step": 50891
    },
    {
      "epoch": 4.853343963623047e-05,
      "step": 50891,
      "training_step_time": 7.899570941925049
    },
    {
      "epoch": 4.853439331054688e-05,
      "model_forward_time": 2.3238494396209717,
      "step": 50892
    },
    {
      "epoch": 4.853439331054688e-05,
      "step": 50892,
      "training_step_time": 7.9404754638671875
    },
    {
      "epoch": 4.853534698486328e-05,
      "model_forward_time": 1.7434227466583252,
      "step": 50893
    },
    {
      "epoch": 4.853534698486328e-05,
      "step": 50893,
      "training_step_time": 8.480018377304077
    },
    {
      "epoch": 4.8536300659179686e-05,
      "model_forward_time": 1.7568800449371338,
      "step": 50894
    },
    {
      "epoch": 4.8536300659179686e-05,
      "step": 50894,
      "training_step_time": 7.895501136779785
    },
    {
      "epoch": 4.85372543334961e-05,
      "model_forward_time": 1.7334413528442383,
      "step": 50895
    },
    {
      "epoch": 4.85372543334961e-05,
      "step": 50895,
      "training_step_time": 9.865288496017456
    },
    {
      "epoch": 4.85382080078125e-05,
      "model_forward_time": 1.7355363368988037,
      "step": 50896
    },
    {
      "epoch": 4.85382080078125e-05,
      "step": 50896,
      "training_step_time": 9.498234987258911
    },
    {
      "epoch": 4.8539161682128905e-05,
      "model_forward_time": 1.7321951389312744,
      "step": 50897
    },
    {
      "epoch": 4.8539161682128905e-05,
      "step": 50897,
      "training_step_time": 7.888663291931152
    },
    {
      "epoch": 4.8540115356445316e-05,
      "model_forward_time": 1.742138147354126,
      "step": 50898
    },
    {
      "epoch": 4.8540115356445316e-05,
      "step": 50898,
      "training_step_time": 8.025553226470947
    },
    {
      "epoch": 4.854106903076172e-05,
      "model_forward_time": 1.3645951747894287,
      "step": 50899
    },
    {
      "epoch": 4.854106903076172e-05,
      "step": 50899,
      "training_step_time": 8.602467060089111
    },
    {
      "epoch": 4.8542022705078124e-05,
      "grad_norm": 0.10205712169408798,
      "learning_rate": 5.264596863094575e-06,
      "loss": 0.0551,
      "step": 50900
    },
    {
      "epoch": 4.8542022705078124e-05,
      "model_forward_time": 1.3655595779418945,
      "step": 50900
    },
    {
      "dynamics_loss_avg": 0.06587176658213138,
      "epoch": 4.8542022705078124e-05,
      "step": 50900
    },
    {
      "action_loss_avg": 0.003052194288466126,
      "epoch": 4.8542022705078124e-05,
      "step": 50900
    },
    {
      "epoch": 4.8542022705078124e-05,
      "step": 50900,
      "training_step_time": 8.746799945831299
    },
    {
      "epoch": 4.8542976379394534e-05,
      "model_forward_time": 1.7386689186096191,
      "step": 50901
    },
    {
      "epoch": 4.8542976379394534e-05,
      "step": 50901,
      "training_step_time": 7.927453517913818
    },
    {
      "epoch": 4.854393005371094e-05,
      "model_forward_time": 2.290742874145508,
      "step": 50902
    },
    {
      "epoch": 4.854393005371094e-05,
      "step": 50902,
      "training_step_time": 8.216349840164185
    },
    {
      "epoch": 4.854488372802734e-05,
      "model_forward_time": 1.7380094528198242,
      "step": 50903
    },
    {
      "epoch": 4.854488372802734e-05,
      "step": 50903,
      "training_step_time": 7.89206600189209
    },
    {
      "epoch": 4.854583740234375e-05,
      "model_forward_time": 1.772824764251709,
      "step": 50904
    },
    {
      "epoch": 4.854583740234375e-05,
      "step": 50904,
      "training_step_time": 8.422256231307983
    },
    {
      "epoch": 4.854679107666016e-05,
      "model_forward_time": 1.7365868091583252,
      "step": 50905
    },
    {
      "epoch": 4.854679107666016e-05,
      "step": 50905,
      "training_step_time": 8.622510433197021
    },
    {
      "epoch": 4.854774475097656e-05,
      "model_forward_time": 1.7339439392089844,
      "step": 50906
    },
    {
      "epoch": 4.854774475097656e-05,
      "step": 50906,
      "training_step_time": 7.9110705852508545
    },
    {
      "epoch": 4.8548698425292965e-05,
      "model_forward_time": 1.7336816787719727,
      "step": 50907
    },
    {
      "epoch": 4.8548698425292965e-05,
      "step": 50907,
      "training_step_time": 8.515727996826172
    },
    {
      "epoch": 4.8549652099609376e-05,
      "model_forward_time": 1.7337758541107178,
      "step": 50908
    },
    {
      "epoch": 4.8549652099609376e-05,
      "step": 50908,
      "training_step_time": 7.967272043228149
    },
    {
      "epoch": 4.855060577392578e-05,
      "model_forward_time": 1.7378933429718018,
      "step": 50909
    },
    {
      "epoch": 4.855060577392578e-05,
      "step": 50909,
      "training_step_time": 8.248475790023804
    },
    {
      "epoch": 4.8551559448242184e-05,
      "grad_norm": 0.10493361949920654,
      "learning_rate": 5.262945695700282e-06,
      "loss": 0.0562,
      "step": 50910
    },
    {
      "epoch": 4.8551559448242184e-05,
      "model_forward_time": 1.7434592247009277,
      "step": 50910
    },
    {
      "dynamics_loss_avg": 0.07753931544721127,
      "epoch": 4.8551559448242184e-05,
      "step": 50910
    },
    {
      "action_loss_avg": 0.001902320224326104,
      "epoch": 4.8551559448242184e-05,
      "step": 50910
    },
    {
      "epoch": 4.8551559448242184e-05,
      "step": 50910,
      "training_step_time": 7.9925713539123535
    },
    {
      "epoch": 4.8552513122558595e-05,
      "model_forward_time": 2.3031201362609863,
      "step": 50911
    },
    {
      "epoch": 4.8552513122558595e-05,
      "step": 50911,
      "training_step_time": 8.330625772476196
    },
    {
      "epoch": 4.8553466796875e-05,
      "model_forward_time": 1.7351481914520264,
      "step": 50912
    },
    {
      "epoch": 4.8553466796875e-05,
      "step": 50912,
      "training_step_time": 7.951459646224976
    },
    {
      "epoch": 4.85544204711914e-05,
      "model_forward_time": 1.7736706733703613,
      "step": 50913
    },
    {
      "epoch": 4.85544204711914e-05,
      "step": 50913,
      "training_step_time": 8.829631090164185
    },
    {
      "epoch": 4.8555374145507814e-05,
      "model_forward_time": 1.7357182502746582,
      "step": 50914
    },
    {
      "epoch": 4.8555374145507814e-05,
      "step": 50914,
      "training_step_time": 9.349956274032593
    },
    {
      "epoch": 4.855632781982422e-05,
      "model_forward_time": 1.7370457649230957,
      "step": 50915
    },
    {
      "epoch": 4.855632781982422e-05,
      "step": 50915,
      "training_step_time": 7.986090183258057
    },
    {
      "epoch": 4.855728149414062e-05,
      "model_forward_time": 1.739145040512085,
      "step": 50916
    },
    {
      "epoch": 4.855728149414062e-05,
      "step": 50916,
      "training_step_time": 8.851919174194336
    },
    {
      "epoch": 4.855823516845703e-05,
      "model_forward_time": 1.7384600639343262,
      "step": 50917
    },
    {
      "epoch": 4.855823516845703e-05,
      "step": 50917,
      "training_step_time": 7.9271016120910645
    },
    {
      "epoch": 4.855918884277344e-05,
      "model_forward_time": 1.7376503944396973,
      "step": 50918
    },
    {
      "epoch": 4.855918884277344e-05,
      "step": 50918,
      "training_step_time": 8.533023118972778
    },
    {
      "epoch": 4.856014251708984e-05,
      "model_forward_time": 1.7441315650939941,
      "step": 50919
    },
    {
      "epoch": 4.856014251708984e-05,
      "step": 50919,
      "training_step_time": 8.748315811157227
    },
    {
      "epoch": 4.856109619140625e-05,
      "grad_norm": 0.09029624611139297,
      "learning_rate": 5.261294499550643e-06,
      "loss": 0.0569,
      "step": 50920
    },
    {
      "epoch": 4.856109619140625e-05,
      "model_forward_time": 2.315563678741455,
      "step": 50920
    },
    {
      "dynamics_loss_avg": 0.05842613652348518,
      "epoch": 4.856109619140625e-05,
      "step": 50920
    },
    {
      "action_loss_avg": 0.001529589493293315,
      "epoch": 4.856109619140625e-05,
      "step": 50920
    },
    {
      "epoch": 4.856109619140625e-05,
      "step": 50920,
      "training_step_time": 8.556058168411255
    },
    {
      "epoch": 4.8562049865722656e-05,
      "model_forward_time": 1.7339870929718018,
      "step": 50921
    },
    {
      "epoch": 4.8562049865722656e-05,
      "step": 50921,
      "training_step_time": 7.916887998580933
    },
    {
      "epoch": 4.856300354003906e-05,
      "model_forward_time": 1.7771704196929932,
      "step": 50922
    },
    {
      "epoch": 4.856300354003906e-05,
      "step": 50922,
      "training_step_time": 7.960598945617676
    },
    {
      "epoch": 4.856395721435547e-05,
      "model_forward_time": 1.714369297027588,
      "step": 50923
    },
    {
      "epoch": 4.856395721435547e-05,
      "step": 50923,
      "training_step_time": 7.9884655475616455
    },
    {
      "epoch": 4.8564910888671875e-05,
      "model_forward_time": 1.7028210163116455,
      "step": 50924
    },
    {
      "epoch": 4.8564910888671875e-05,
      "step": 50924,
      "training_step_time": 8.061745882034302
    },
    {
      "epoch": 4.856586456298828e-05,
      "model_forward_time": 1.7267491817474365,
      "step": 50925
    },
    {
      "epoch": 4.856586456298828e-05,
      "step": 50925,
      "training_step_time": 8.020187377929688
    },
    {
      "epoch": 4.856681823730469e-05,
      "model_forward_time": 1.7313745021820068,
      "step": 50926
    },
    {
      "epoch": 4.856681823730469e-05,
      "step": 50926,
      "training_step_time": 8.971234798431396
    },
    {
      "epoch": 4.856777191162109e-05,
      "model_forward_time": 1.7292754650115967,
      "step": 50927
    },
    {
      "epoch": 4.856777191162109e-05,
      "step": 50927,
      "training_step_time": 10.523266792297363
    },
    {
      "epoch": 4.85687255859375e-05,
      "model_forward_time": 1.7310125827789307,
      "step": 50928
    },
    {
      "epoch": 4.85687255859375e-05,
      "step": 50928,
      "training_step_time": 8.459109544754028
    },
    {
      "epoch": 4.856967926025391e-05,
      "model_forward_time": 2.335057258605957,
      "step": 50929
    },
    {
      "epoch": 4.856967926025391e-05,
      "step": 50929,
      "training_step_time": 7.859297752380371
    },
    {
      "epoch": 4.857063293457031e-05,
      "grad_norm": 0.14817872643470764,
      "learning_rate": 5.25964327482623e-06,
      "loss": 0.057,
      "step": 50930
    },
    {
      "epoch": 4.857063293457031e-05,
      "model_forward_time": 1.7329144477844238,
      "step": 50930
    },
    {
      "dynamics_loss_avg": 0.05898720100522041,
      "epoch": 4.857063293457031e-05,
      "step": 50930
    },
    {
      "action_loss_avg": 0.0021173074608668687,
      "epoch": 4.857063293457031e-05,
      "step": 50930
    },
    {
      "epoch": 4.857063293457031e-05,
      "step": 50930,
      "training_step_time": 7.928712368011475
    },
    {
      "epoch": 4.8571586608886716e-05,
      "model_forward_time": 1.7599594593048096,
      "step": 50931
    },
    {
      "epoch": 4.8571586608886716e-05,
      "step": 50931,
      "training_step_time": 9.266787767410278
    },
    {
      "epoch": 4.857254028320313e-05,
      "model_forward_time": 1.7404115200042725,
      "step": 50932
    },
    {
      "epoch": 4.857254028320313e-05,
      "step": 50932,
      "training_step_time": 7.917916774749756
    },
    {
      "epoch": 4.857349395751953e-05,
      "model_forward_time": 1.741412878036499,
      "step": 50933
    },
    {
      "epoch": 4.857349395751953e-05,
      "step": 50933,
      "training_step_time": 9.203454732894897
    },
    {
      "epoch": 4.8574447631835935e-05,
      "model_forward_time": 1.7451715469360352,
      "step": 50934
    },
    {
      "epoch": 4.8574447631835935e-05,
      "step": 50934,
      "training_step_time": 8.005059719085693
    },
    {
      "epoch": 4.8575401306152346e-05,
      "model_forward_time": 1.7425696849822998,
      "step": 50935
    },
    {
      "epoch": 4.8575401306152346e-05,
      "step": 50935,
      "training_step_time": 8.524834871292114
    },
    {
      "epoch": 4.857635498046875e-05,
      "model_forward_time": 1.7424201965332031,
      "step": 50936
    },
    {
      "epoch": 4.857635498046875e-05,
      "step": 50936,
      "training_step_time": 7.914163827896118
    },
    {
      "epoch": 4.8577308654785154e-05,
      "model_forward_time": 1.7422637939453125,
      "step": 50937
    },
    {
      "epoch": 4.8577308654785154e-05,
      "step": 50937,
      "training_step_time": 7.8772196769714355
    },
    {
      "epoch": 4.8578262329101565e-05,
      "model_forward_time": 2.288701057434082,
      "step": 50938
    },
    {
      "epoch": 4.8578262329101565e-05,
      "step": 50938,
      "training_step_time": 7.83580207824707
    },
    {
      "epoch": 4.857921600341797e-05,
      "model_forward_time": 1.7218818664550781,
      "step": 50939
    },
    {
      "epoch": 4.857921600341797e-05,
      "step": 50939,
      "training_step_time": 7.852780818939209
    },
    {
      "epoch": 4.858016967773437e-05,
      "grad_norm": 0.12239456921815872,
      "learning_rate": 5.257992021707617e-06,
      "loss": 0.0557,
      "step": 50940
    },
    {
      "epoch": 4.858016967773437e-05,
      "model_forward_time": 1.7197349071502686,
      "step": 50940
    },
    {
      "dynamics_loss_avg": 0.04804356563836336,
      "epoch": 4.858016967773437e-05,
      "step": 50940
    },
    {
      "action_loss_avg": 0.0013734122359892354,
      "epoch": 4.858016967773437e-05,
      "step": 50940
    },
    {
      "epoch": 4.858016967773437e-05,
      "step": 50940,
      "training_step_time": 7.994504928588867
    },
    {
      "epoch": 4.8581123352050784e-05,
      "model_forward_time": 1.7347323894500732,
      "step": 50941
    },
    {
      "epoch": 4.8581123352050784e-05,
      "step": 50941,
      "training_step_time": 8.43810510635376
    },
    {
      "epoch": 4.858207702636719e-05,
      "model_forward_time": 1.7371001243591309,
      "step": 50942
    },
    {
      "epoch": 4.858207702636719e-05,
      "step": 50942,
      "training_step_time": 8.045690536499023
    },
    {
      "epoch": 4.858303070068359e-05,
      "model_forward_time": 1.7285184860229492,
      "step": 50943
    },
    {
      "epoch": 4.858303070068359e-05,
      "step": 50943,
      "training_step_time": 7.877994775772095
    },
    {
      "epoch": 4.8583984375e-05,
      "model_forward_time": 1.739755392074585,
      "step": 50944
    },
    {
      "epoch": 4.8583984375e-05,
      "step": 50944,
      "training_step_time": 8.088176965713501
    },
    {
      "epoch": 4.8584938049316407e-05,
      "model_forward_time": 1.7219762802124023,
      "step": 50945
    },
    {
      "epoch": 4.8584938049316407e-05,
      "step": 50945,
      "training_step_time": 8.879722356796265
    },
    {
      "epoch": 4.858589172363281e-05,
      "model_forward_time": 1.7337369918823242,
      "step": 50946
    },
    {
      "epoch": 4.858589172363281e-05,
      "step": 50946,
      "training_step_time": 7.858152151107788
    },
    {
      "epoch": 4.858684539794922e-05,
      "model_forward_time": 1.716191291809082,
      "step": 50947
    },
    {
      "epoch": 4.858684539794922e-05,
      "step": 50947,
      "training_step_time": 7.913865804672241
    },
    {
      "epoch": 4.8587799072265625e-05,
      "model_forward_time": 2.260530710220337,
      "step": 50948
    },
    {
      "epoch": 4.8587799072265625e-05,
      "step": 50948,
      "training_step_time": 8.029272079467773
    },
    {
      "epoch": 4.858875274658203e-05,
      "model_forward_time": 1.7231671810150146,
      "step": 50949
    },
    {
      "epoch": 4.858875274658203e-05,
      "step": 50949,
      "training_step_time": 7.830717325210571
    },
    {
      "epoch": 4.858970642089844e-05,
      "grad_norm": 0.15793730318546295,
      "learning_rate": 5.256340740375387e-06,
      "loss": 0.0571,
      "step": 50950
    },
    {
      "epoch": 4.858970642089844e-05,
      "model_forward_time": 1.7571053504943848,
      "step": 50950
    },
    {
      "dynamics_loss_avg": 0.07167275324463844,
      "epoch": 4.858970642089844e-05,
      "step": 50950
    },
    {
      "action_loss_avg": 0.003042996657313779,
      "epoch": 4.858970642089844e-05,
      "step": 50950
    },
    {
      "epoch": 4.858970642089844e-05,
      "step": 50950,
      "training_step_time": 7.942043304443359
    },
    {
      "epoch": 4.8590660095214844e-05,
      "model_forward_time": 1.7185983657836914,
      "step": 50951
    },
    {
      "epoch": 4.8590660095214844e-05,
      "step": 50951,
      "training_step_time": 8.063172817230225
    },
    {
      "epoch": 4.859161376953125e-05,
      "model_forward_time": 1.717820644378662,
      "step": 50952
    },
    {
      "epoch": 4.859161376953125e-05,
      "step": 50952,
      "training_step_time": 8.479763507843018
    },
    {
      "epoch": 4.859256744384766e-05,
      "model_forward_time": 1.7184438705444336,
      "step": 50953
    },
    {
      "epoch": 4.859256744384766e-05,
      "step": 50953,
      "training_step_time": 8.884095907211304
    },
    {
      "epoch": 4.859352111816406e-05,
      "model_forward_time": 1.7190663814544678,
      "step": 50954
    },
    {
      "epoch": 4.859352111816406e-05,
      "step": 50954,
      "training_step_time": 7.8221611976623535
    },
    {
      "epoch": 4.859447479248047e-05,
      "model_forward_time": 1.713167428970337,
      "step": 50955
    },
    {
      "epoch": 4.859447479248047e-05,
      "step": 50955,
      "training_step_time": 7.760719060897827
    },
    {
      "epoch": 4.859542846679688e-05,
      "model_forward_time": 1.7180595397949219,
      "step": 50956
    },
    {
      "epoch": 4.859542846679688e-05,
      "step": 50956,
      "training_step_time": 7.710723161697388
    },
    {
      "epoch": 4.859638214111328e-05,
      "model_forward_time": 2.270001173019409,
      "step": 50957
    },
    {
      "epoch": 4.859638214111328e-05,
      "step": 50957,
      "training_step_time": 7.819677829742432
    },
    {
      "epoch": 4.8597335815429686e-05,
      "model_forward_time": 1.7134110927581787,
      "step": 50958
    },
    {
      "epoch": 4.8597335815429686e-05,
      "step": 50958,
      "training_step_time": 8.164746284484863
    },
    {
      "epoch": 4.85982894897461e-05,
      "model_forward_time": 1.7645173072814941,
      "step": 50959
    },
    {
      "epoch": 4.85982894897461e-05,
      "step": 50959,
      "training_step_time": 8.485371828079224
    },
    {
      "epoch": 4.85992431640625e-05,
      "grad_norm": 0.09899751096963882,
      "learning_rate": 5.254689431010117e-06,
      "loss": 0.0566,
      "step": 50960
    },
    {
      "epoch": 4.85992431640625e-05,
      "model_forward_time": 1.711972951889038,
      "step": 50960
    },
    {
      "dynamics_loss_avg": 0.050895626470446584,
      "epoch": 4.85992431640625e-05,
      "step": 50960
    },
    {
      "action_loss_avg": 0.0014550431456882506,
      "epoch": 4.85992431640625e-05,
      "step": 50960
    },
    {
      "epoch": 4.85992431640625e-05,
      "step": 50960,
      "training_step_time": 7.7450621128082275
    },
    {
      "epoch": 4.8600196838378905e-05,
      "model_forward_time": 1.7177443504333496,
      "step": 50961
    },
    {
      "epoch": 4.8600196838378905e-05,
      "step": 50961,
      "training_step_time": 8.013754844665527
    },
    {
      "epoch": 4.8601150512695316e-05,
      "model_forward_time": 1.723642110824585,
      "step": 50962
    },
    {
      "epoch": 4.8601150512695316e-05,
      "step": 50962,
      "training_step_time": 7.799872159957886
    },
    {
      "epoch": 4.860210418701172e-05,
      "model_forward_time": 1.7158498764038086,
      "step": 50963
    },
    {
      "epoch": 4.860210418701172e-05,
      "step": 50963,
      "training_step_time": 7.718317270278931
    },
    {
      "epoch": 4.8603057861328124e-05,
      "model_forward_time": 1.7220656871795654,
      "step": 50964
    },
    {
      "epoch": 4.8603057861328124e-05,
      "step": 50964,
      "training_step_time": 7.682774066925049
    },
    {
      "epoch": 4.8604011535644535e-05,
      "model_forward_time": 1.7295362949371338,
      "step": 50965
    },
    {
      "epoch": 4.8604011535644535e-05,
      "step": 50965,
      "training_step_time": 8.30681562423706
    },
    {
      "epoch": 4.860496520996094e-05,
      "model_forward_time": 2.275237798690796,
      "step": 50966
    },
    {
      "epoch": 4.860496520996094e-05,
      "step": 50966,
      "training_step_time": 7.736961364746094
    },
    {
      "epoch": 4.860591888427734e-05,
      "model_forward_time": 1.7191681861877441,
      "step": 50967
    },
    {
      "epoch": 4.860591888427734e-05,
      "step": 50967,
      "training_step_time": 7.2700769901275635
    },
    {
      "epoch": 4.860687255859375e-05,
      "model_forward_time": 1.7540614604949951,
      "step": 50968
    },
    {
      "epoch": 4.860687255859375e-05,
      "step": 50968,
      "training_step_time": 7.619948148727417
    },
    {
      "epoch": 4.860782623291016e-05,
      "model_forward_time": 1.7159202098846436,
      "step": 50969
    },
    {
      "epoch": 4.860782623291016e-05,
      "step": 50969,
      "training_step_time": 8.368393182754517
    },
    {
      "epoch": 4.860877990722656e-05,
      "grad_norm": 0.08931146562099457,
      "learning_rate": 5.2530380937923955e-06,
      "loss": 0.0556,
      "step": 50970
    },
    {
      "epoch": 4.860877990722656e-05,
      "model_forward_time": 1.7193067073822021,
      "step": 50970
    },
    {
      "dynamics_loss_avg": 0.053510510548949244,
      "epoch": 4.860877990722656e-05,
      "step": 50970
    },
    {
      "action_loss_avg": 0.001355619533569552,
      "epoch": 4.860877990722656e-05,
      "step": 50970
    },
    {
      "epoch": 4.860877990722656e-05,
      "step": 50970,
      "training_step_time": 8.678088426589966
    },
    {
      "epoch": 4.8609733581542965e-05,
      "model_forward_time": 1.7220306396484375,
      "step": 50971
    },
    {
      "epoch": 4.8609733581542965e-05,
      "step": 50971,
      "training_step_time": 7.710520267486572
    },
    {
      "epoch": 4.8610687255859376e-05,
      "model_forward_time": 1.7188398838043213,
      "step": 50972
    },
    {
      "epoch": 4.8610687255859376e-05,
      "step": 50972,
      "training_step_time": 7.741954326629639
    },
    {
      "epoch": 4.861164093017578e-05,
      "model_forward_time": 1.718902587890625,
      "step": 50973
    },
    {
      "epoch": 4.861164093017578e-05,
      "step": 50973,
      "training_step_time": 7.726250171661377
    },
    {
      "epoch": 4.8612594604492184e-05,
      "model_forward_time": 1.7242631912231445,
      "step": 50974
    },
    {
      "epoch": 4.8612594604492184e-05,
      "step": 50974,
      "training_step_time": 9.060850381851196
    },
    {
      "epoch": 4.8613548278808595e-05,
      "model_forward_time": 2.3037641048431396,
      "step": 50975
    },
    {
      "epoch": 4.8613548278808595e-05,
      "step": 50975,
      "training_step_time": 8.680670738220215
    },
    {
      "epoch": 4.8614501953125e-05,
      "model_forward_time": 1.7316091060638428,
      "step": 50976
    },
    {
      "epoch": 4.8614501953125e-05,
      "step": 50976,
      "training_step_time": 7.422759056091309
    },
    {
      "epoch": 4.86154556274414e-05,
      "model_forward_time": 1.7502543926239014,
      "step": 50977
    },
    {
      "epoch": 4.86154556274414e-05,
      "step": 50977,
      "training_step_time": 8.49637508392334
    },
    {
      "epoch": 4.8616409301757814e-05,
      "model_forward_time": 1.7337768077850342,
      "step": 50978
    },
    {
      "epoch": 4.8616409301757814e-05,
      "step": 50978,
      "training_step_time": 9.816410303115845
    },
    {
      "epoch": 4.861736297607422e-05,
      "model_forward_time": 1.734522819519043,
      "step": 50979
    },
    {
      "epoch": 4.861736297607422e-05,
      "step": 50979,
      "training_step_time": 7.951807737350464
    },
    {
      "epoch": 4.861831665039062e-05,
      "grad_norm": 0.11054389923810959,
      "learning_rate": 5.251386728902806e-06,
      "loss": 0.0545,
      "step": 50980
    },
    {
      "epoch": 4.861831665039062e-05,
      "model_forward_time": 1.7320654392242432,
      "step": 50980
    },
    {
      "dynamics_loss_avg": 0.0641956316307187,
      "epoch": 4.861831665039062e-05,
      "step": 50980
    },
    {
      "action_loss_avg": 0.00238894778303802,
      "epoch": 4.861831665039062e-05,
      "step": 50980
    },
    {
      "epoch": 4.861831665039062e-05,
      "step": 50980,
      "training_step_time": 9.466739654541016
    },
    {
      "epoch": 4.861927032470703e-05,
      "model_forward_time": 1.7245218753814697,
      "step": 50981
    },
    {
      "epoch": 4.861927032470703e-05,
      "step": 50981,
      "training_step_time": 8.787447452545166
    },
    {
      "epoch": 4.862022399902344e-05,
      "model_forward_time": 1.7284603118896484,
      "step": 50982
    },
    {
      "epoch": 4.862022399902344e-05,
      "step": 50982,
      "training_step_time": 7.67375111579895
    },
    {
      "epoch": 4.862117767333984e-05,
      "model_forward_time": 1.7255258560180664,
      "step": 50983
    },
    {
      "epoch": 4.862117767333984e-05,
      "step": 50983,
      "training_step_time": 8.276325702667236
    },
    {
      "epoch": 4.862213134765625e-05,
      "model_forward_time": 2.296867609024048,
      "step": 50984
    },
    {
      "epoch": 4.862213134765625e-05,
      "step": 50984,
      "training_step_time": 7.705386161804199
    },
    {
      "epoch": 4.8623085021972656e-05,
      "model_forward_time": 1.7254853248596191,
      "step": 50985
    },
    {
      "epoch": 4.8623085021972656e-05,
      "step": 50985,
      "training_step_time": 7.666799306869507
    },
    {
      "epoch": 4.862403869628906e-05,
      "model_forward_time": 1.7243380546569824,
      "step": 50986
    },
    {
      "epoch": 4.862403869628906e-05,
      "step": 50986,
      "training_step_time": 7.598426818847656
    },
    {
      "epoch": 4.862499237060547e-05,
      "model_forward_time": 1.7690443992614746,
      "step": 50987
    },
    {
      "epoch": 4.862499237060547e-05,
      "step": 50987,
      "training_step_time": 8.31844687461853
    },
    {
      "epoch": 4.8625946044921875e-05,
      "model_forward_time": 1.7245965003967285,
      "step": 50988
    },
    {
      "epoch": 4.8625946044921875e-05,
      "step": 50988,
      "training_step_time": 8.397772789001465
    },
    {
      "epoch": 4.862689971923828e-05,
      "model_forward_time": 1.7328846454620361,
      "step": 50989
    },
    {
      "epoch": 4.862689971923828e-05,
      "step": 50989,
      "training_step_time": 9.650768041610718
    },
    {
      "epoch": 4.862785339355469e-05,
      "grad_norm": 0.1729501634836197,
      "learning_rate": 5.249735336521945e-06,
      "loss": 0.0534,
      "step": 50990
    },
    {
      "epoch": 4.862785339355469e-05,
      "model_forward_time": 1.7299320697784424,
      "step": 50990
    },
    {
      "dynamics_loss_avg": 0.05974591765552759,
      "epoch": 4.862785339355469e-05,
      "step": 50990
    },
    {
      "action_loss_avg": 0.001886303024366498,
      "epoch": 4.862785339355469e-05,
      "step": 50990
    },
    {
      "epoch": 4.862785339355469e-05,
      "step": 50990,
      "training_step_time": 7.817662954330444
    },
    {
      "epoch": 4.8628807067871093e-05,
      "model_forward_time": 1.7342338562011719,
      "step": 50991
    },
    {
      "epoch": 4.8628807067871093e-05,
      "step": 50991,
      "training_step_time": 7.592928171157837
    },
    {
      "epoch": 4.86297607421875e-05,
      "model_forward_time": 1.7211661338806152,
      "step": 50992
    },
    {
      "epoch": 4.86297607421875e-05,
      "step": 50992,
      "training_step_time": 7.720312118530273
    },
    {
      "epoch": 4.863071441650391e-05,
      "model_forward_time": 1.7010810375213623,
      "step": 50993
    },
    {
      "epoch": 4.863071441650391e-05,
      "step": 50993,
      "training_step_time": 7.721985816955566
    },
    {
      "epoch": 4.863166809082031e-05,
      "model_forward_time": 2.272986650466919,
      "step": 50994
    },
    {
      "epoch": 4.863166809082031e-05,
      "step": 50994,
      "training_step_time": 7.587417364120483
    },
    {
      "epoch": 4.8632621765136716e-05,
      "model_forward_time": 1.707190990447998,
      "step": 50995
    },
    {
      "epoch": 4.8632621765136716e-05,
      "step": 50995,
      "training_step_time": 7.551546573638916
    },
    {
      "epoch": 4.863357543945313e-05,
      "model_forward_time": 1.7527127265930176,
      "step": 50996
    },
    {
      "epoch": 4.863357543945313e-05,
      "step": 50996,
      "training_step_time": 7.647014856338501
    },
    {
      "epoch": 4.863452911376953e-05,
      "model_forward_time": 1.7263760566711426,
      "step": 50997
    },
    {
      "epoch": 4.863452911376953e-05,
      "step": 50997,
      "training_step_time": 9.971821069717407
    },
    {
      "epoch": 4.8635482788085935e-05,
      "model_forward_time": 1.7519316673278809,
      "step": 50998
    },
    {
      "epoch": 4.8635482788085935e-05,
      "step": 50998,
      "training_step_time": 9.038831949234009
    },
    {
      "epoch": 4.8636436462402346e-05,
      "model_forward_time": 1.7121903896331787,
      "step": 50999
    },
    {
      "epoch": 4.8636436462402346e-05,
      "step": 50999,
      "training_step_time": 7.930569648742676
    },
    {
      "epoch": 4.863739013671875e-05,
      "grad_norm": 0.1337307244539261,
      "learning_rate": 5.2480839168304e-06,
      "loss": 0.0563,
      "step": 51000
    },
    {
      "epoch": 4.863739013671875e-05,
      "model_forward_time": 1.7618682384490967,
      "step": 51000
    },
    {
      "dynamics_loss_avg": 0.07310494743287563,
      "epoch": 4.863739013671875e-05,
      "step": 51000
    },
    {
      "action_loss_avg": 0.0019572849792893975,
      "epoch": 4.863739013671875e-05,
      "step": 51000
    },
    {
      "epoch": 4.863739013671875e-05,
      "step": 51000,
      "training_step_time": 7.5457117557525635
    },
    {
      "epoch": 4.8638343811035154e-05,
      "model_forward_time": 1.7686803340911865,
      "step": 51001
    },
    {
      "epoch": 4.8638343811035154e-05,
      "step": 51001,
      "training_step_time": 9.251506567001343
    },
    {
      "epoch": 4.8639297485351565e-05,
      "model_forward_time": 2.73504376411438,
      "step": 51002
    },
    {
      "epoch": 4.8639297485351565e-05,
      "step": 51002,
      "training_step_time": 7.977530002593994
    },
    {
      "epoch": 4.864025115966797e-05,
      "model_forward_time": 1.7319355010986328,
      "step": 51003
    },
    {
      "epoch": 4.864025115966797e-05,
      "step": 51003,
      "training_step_time": 9.474355220794678
    },
    {
      "epoch": 4.864120483398437e-05,
      "model_forward_time": 1.7239632606506348,
      "step": 51004
    },
    {
      "epoch": 4.864120483398437e-05,
      "step": 51004,
      "training_step_time": 7.79978084564209
    },
    {
      "epoch": 4.8642158508300784e-05,
      "model_forward_time": 1.7640366554260254,
      "step": 51005
    },
    {
      "epoch": 4.8642158508300784e-05,
      "step": 51005,
      "training_step_time": 8.487435817718506
    },
    {
      "epoch": 4.864311218261719e-05,
      "model_forward_time": 1.7222542762756348,
      "step": 51006
    },
    {
      "epoch": 4.864311218261719e-05,
      "step": 51006,
      "training_step_time": 10.916588306427002
    },
    {
      "epoch": 4.864406585693359e-05,
      "model_forward_time": 1.7230637073516846,
      "step": 51007
    },
    {
      "epoch": 4.864406585693359e-05,
      "step": 51007,
      "training_step_time": 10.045523405075073
    },
    {
      "epoch": 4.864501953125e-05,
      "model_forward_time": 1.7358269691467285,
      "step": 51008
    },
    {
      "epoch": 4.864501953125e-05,
      "step": 51008,
      "training_step_time": 9.384153842926025
    },
    {
      "epoch": 4.8645973205566407e-05,
      "model_forward_time": 1.7261242866516113,
      "step": 51009
    },
    {
      "epoch": 4.8645973205566407e-05,
      "step": 51009,
      "training_step_time": 8.553825378417969
    },
    {
      "epoch": 4.864692687988281e-05,
      "grad_norm": 0.09083765745162964,
      "learning_rate": 5.246432470008773e-06,
      "loss": 0.0559,
      "step": 51010
    },
    {
      "epoch": 4.864692687988281e-05,
      "model_forward_time": 1.7223801612854004,
      "step": 51010
    },
    {
      "dynamics_loss_avg": 0.058189929276704785,
      "epoch": 4.864692687988281e-05,
      "step": 51010
    },
    {
      "action_loss_avg": 0.0023609806259628384,
      "epoch": 4.864692687988281e-05,
      "step": 51010
    },
    {
      "epoch": 4.864692687988281e-05,
      "step": 51010,
      "training_step_time": 8.528137922286987
    },
    {
      "epoch": 4.864788055419922e-05,
      "model_forward_time": 2.286005973815918,
      "step": 51011
    },
    {
      "epoch": 4.864788055419922e-05,
      "step": 51011,
      "training_step_time": 7.961347818374634
    },
    {
      "epoch": 4.8648834228515625e-05,
      "model_forward_time": 1.7357168197631836,
      "step": 51012
    },
    {
      "epoch": 4.8648834228515625e-05,
      "step": 51012,
      "training_step_time": 7.523054122924805
    },
    {
      "epoch": 4.864978790283203e-05,
      "model_forward_time": 2.910698652267456,
      "step": 51013
    },
    {
      "epoch": 4.864978790283203e-05,
      "step": 51013,
      "training_step_time": 11.40169644355774
    },
    {
      "epoch": 4.865074157714844e-05,
      "model_forward_time": 1.71848726272583,
      "step": 51014
    },
    {
      "epoch": 4.865074157714844e-05,
      "step": 51014,
      "training_step_time": 7.121514320373535
    },
    {
      "epoch": 4.8651695251464844e-05,
      "model_forward_time": 1.714460849761963,
      "step": 51015
    },
    {
      "epoch": 4.8651695251464844e-05,
      "step": 51015,
      "training_step_time": 7.090878248214722
    },
    {
      "epoch": 4.865264892578125e-05,
      "model_forward_time": 1.7163007259368896,
      "step": 51016
    },
    {
      "epoch": 4.865264892578125e-05,
      "step": 51016,
      "training_step_time": 7.072211503982544
    },
    {
      "epoch": 4.865360260009766e-05,
      "model_forward_time": 1.7160007953643799,
      "step": 51017
    },
    {
      "epoch": 4.865360260009766e-05,
      "step": 51017,
      "training_step_time": 7.103298187255859
    },
    {
      "epoch": 4.865455627441406e-05,
      "model_forward_time": 1.705883502960205,
      "step": 51018
    },
    {
      "epoch": 4.865455627441406e-05,
      "step": 51018,
      "training_step_time": 7.015384197235107
    },
    {
      "epoch": 4.865550994873047e-05,
      "model_forward_time": 1.713876485824585,
      "step": 51019
    },
    {
      "epoch": 4.865550994873047e-05,
      "step": 51019,
      "training_step_time": 7.056737899780273
    },
    {
      "epoch": 4.865646362304688e-05,
      "grad_norm": 0.08523086458444595,
      "learning_rate": 5.244780996237662e-06,
      "loss": 0.0552,
      "step": 51020
    },
    {
      "epoch": 4.865646362304688e-05,
      "model_forward_time": 1.7087759971618652,
      "step": 51020
    },
    {
      "dynamics_loss_avg": 0.043649666011333466,
      "epoch": 4.865646362304688e-05,
      "step": 51020
    },
    {
      "action_loss_avg": 0.0022698151122312995,
      "epoch": 4.865646362304688e-05,
      "step": 51020
    },
    {
      "epoch": 4.865646362304688e-05,
      "step": 51020,
      "training_step_time": 7.10528039932251
    },
    {
      "epoch": 4.865741729736328e-05,
      "model_forward_time": 1.7167978286743164,
      "step": 51021
    },
    {
      "epoch": 4.865741729736328e-05,
      "step": 51021,
      "training_step_time": 7.075563907623291
    },
    {
      "epoch": 4.8658370971679686e-05,
      "model_forward_time": 2.279892683029175,
      "step": 51022
    },
    {
      "epoch": 4.8658370971679686e-05,
      "step": 51022,
      "training_step_time": 7.67303729057312
    },
    {
      "epoch": 4.86593246459961e-05,
      "model_forward_time": 1.7101798057556152,
      "step": 51023
    },
    {
      "epoch": 4.86593246459961e-05,
      "step": 51023,
      "training_step_time": 8.15052342414856
    },
    {
      "epoch": 4.86602783203125e-05,
      "model_forward_time": 1.7153592109680176,
      "step": 51024
    },
    {
      "epoch": 4.86602783203125e-05,
      "step": 51024,
      "training_step_time": 7.137415170669556
    },
    {
      "epoch": 4.8661231994628905e-05,
      "model_forward_time": 1.722926378250122,
      "step": 51025
    },
    {
      "epoch": 4.8661231994628905e-05,
      "step": 51025,
      "training_step_time": 7.150887727737427
    },
    {
      "epoch": 4.8662185668945316e-05,
      "model_forward_time": 1.7115099430084229,
      "step": 51026
    },
    {
      "epoch": 4.8662185668945316e-05,
      "step": 51026,
      "training_step_time": 7.170771598815918
    },
    {
      "epoch": 4.866313934326172e-05,
      "model_forward_time": 1.7183856964111328,
      "step": 51027
    },
    {
      "epoch": 4.866313934326172e-05,
      "step": 51027,
      "training_step_time": 7.180726766586304
    },
    {
      "epoch": 4.8664093017578124e-05,
      "model_forward_time": 1.718022346496582,
      "step": 51028
    },
    {
      "epoch": 4.8664093017578124e-05,
      "step": 51028,
      "training_step_time": 7.213884115219116
    },
    {
      "epoch": 4.8665046691894535e-05,
      "model_forward_time": 1.713970422744751,
      "step": 51029
    },
    {
      "epoch": 4.8665046691894535e-05,
      "step": 51029,
      "training_step_time": 7.147160291671753
    },
    {
      "epoch": 4.866600036621094e-05,
      "grad_norm": 0.142441987991333,
      "learning_rate": 5.243129495697666e-06,
      "loss": 0.0547,
      "step": 51030
    },
    {
      "epoch": 4.866600036621094e-05,
      "model_forward_time": 1.7113990783691406,
      "step": 51030
    },
    {
      "dynamics_loss_avg": 0.05506626768037677,
      "epoch": 4.866600036621094e-05,
      "step": 51030
    },
    {
      "action_loss_avg": 0.0027680049824994057,
      "epoch": 4.866600036621094e-05,
      "step": 51030
    },
    {
      "epoch": 4.866600036621094e-05,
      "step": 51030,
      "training_step_time": 7.158789157867432
    },
    {
      "epoch": 4.866695404052734e-05,
      "model_forward_time": 2.2737793922424316,
      "step": 51031
    },
    {
      "epoch": 4.866695404052734e-05,
      "step": 51031,
      "training_step_time": 8.875603675842285
    },
    {
      "epoch": 4.866790771484375e-05,
      "model_forward_time": 1.7171661853790283,
      "step": 51032
    },
    {
      "epoch": 4.866790771484375e-05,
      "step": 51032,
      "training_step_time": 8.174293518066406
    },
    {
      "epoch": 4.866886138916016e-05,
      "model_forward_time": 1.7111928462982178,
      "step": 51033
    },
    {
      "epoch": 4.866886138916016e-05,
      "step": 51033,
      "training_step_time": 7.428409099578857
    },
    {
      "epoch": 4.866981506347656e-05,
      "model_forward_time": 1.7052991390228271,
      "step": 51034
    },
    {
      "epoch": 4.866981506347656e-05,
      "step": 51034,
      "training_step_time": 7.1527791023254395
    },
    {
      "epoch": 4.8670768737792965e-05,
      "model_forward_time": 1.7187952995300293,
      "step": 51035
    },
    {
      "epoch": 4.8670768737792965e-05,
      "step": 51035,
      "training_step_time": 7.176695823669434
    },
    {
      "epoch": 4.8671722412109376e-05,
      "model_forward_time": 1.721984624862671,
      "step": 51036
    },
    {
      "epoch": 4.8671722412109376e-05,
      "step": 51036,
      "training_step_time": 6.895105361938477
    },
    {
      "epoch": 4.867267608642578e-05,
      "model_forward_time": 1.715625286102295,
      "step": 51037
    },
    {
      "epoch": 4.867267608642578e-05,
      "step": 51037,
      "training_step_time": 7.097517013549805
    },
    {
      "epoch": 4.8673629760742184e-05,
      "model_forward_time": 1.7182538509368896,
      "step": 51038
    },
    {
      "epoch": 4.8673629760742184e-05,
      "step": 51038,
      "training_step_time": 7.09410834312439
    },
    {
      "epoch": 4.8674583435058595e-05,
      "model_forward_time": 1.7073371410369873,
      "step": 51039
    },
    {
      "epoch": 4.8674583435058595e-05,
      "step": 51039,
      "training_step_time": 7.0783374309539795
    },
    {
      "epoch": 4.8675537109375e-05,
      "grad_norm": 0.10352472215890884,
      "learning_rate": 5.241477968569397e-06,
      "loss": 0.0563,
      "step": 51040
    },
    {
      "epoch": 4.8675537109375e-05,
      "model_forward_time": 1.729095697402954,
      "step": 51040
    },
    {
      "dynamics_loss_avg": 0.06425108052790166,
      "epoch": 4.8675537109375e-05,
      "step": 51040
    },
    {
      "action_loss_avg": 0.003777646867092699,
      "epoch": 4.8675537109375e-05,
      "step": 51040
    },
    {
      "epoch": 4.8675537109375e-05,
      "step": 51040,
      "training_step_time": 8.219076871871948
    },
    {
      "epoch": 4.86764907836914e-05,
      "model_forward_time": 1.716400146484375,
      "step": 51041
    },
    {
      "epoch": 4.86764907836914e-05,
      "step": 51041,
      "training_step_time": 8.184648275375366
    },
    {
      "epoch": 4.8677444458007814e-05,
      "model_forward_time": 1.7224631309509277,
      "step": 51042
    },
    {
      "epoch": 4.8677444458007814e-05,
      "step": 51042,
      "training_step_time": 7.90250563621521
    },
    {
      "epoch": 4.867839813232422e-05,
      "model_forward_time": 1.73093581199646,
      "step": 51043
    },
    {
      "epoch": 4.867839813232422e-05,
      "step": 51043,
      "training_step_time": 7.081867933273315
    },
    {
      "epoch": 4.867935180664062e-05,
      "model_forward_time": 1.7214055061340332,
      "step": 51044
    },
    {
      "epoch": 4.867935180664062e-05,
      "step": 51044,
      "training_step_time": 7.14076042175293
    },
    {
      "epoch": 4.868030548095703e-05,
      "model_forward_time": 1.7217333316802979,
      "step": 51045
    },
    {
      "epoch": 4.868030548095703e-05,
      "step": 51045,
      "training_step_time": 7.0187225341796875
    },
    {
      "epoch": 4.868125915527344e-05,
      "model_forward_time": 1.7245385646820068,
      "step": 51046
    },
    {
      "epoch": 4.868125915527344e-05,
      "step": 51046,
      "training_step_time": 7.076747179031372
    },
    {
      "epoch": 4.868221282958984e-05,
      "model_forward_time": 1.7337210178375244,
      "step": 51047
    },
    {
      "epoch": 4.868221282958984e-05,
      "step": 51047,
      "training_step_time": 7.063085079193115
    },
    {
      "epoch": 4.868316650390625e-05,
      "model_forward_time": 1.7242920398712158,
      "step": 51048
    },
    {
      "epoch": 4.868316650390625e-05,
      "step": 51048,
      "training_step_time": 7.096544504165649
    },
    {
      "epoch": 4.8684120178222656e-05,
      "model_forward_time": 1.7210807800292969,
      "step": 51049
    },
    {
      "epoch": 4.8684120178222656e-05,
      "step": 51049,
      "training_step_time": 7.061256408691406
    },
    {
      "epoch": 4.868507385253906e-05,
      "grad_norm": 0.11870850622653961,
      "learning_rate": 5.2398264150334595e-06,
      "loss": 0.0608,
      "step": 51050
    },
    {
      "epoch": 4.868507385253906e-05,
      "model_forward_time": 2.267807722091675,
      "step": 51050
    },
    {
      "dynamics_loss_avg": 0.04591535665094852,
      "epoch": 4.868507385253906e-05,
      "step": 51050
    },
    {
      "action_loss_avg": 0.0019136747345328332,
      "epoch": 4.868507385253906e-05,
      "step": 51050
    },
    {
      "epoch": 4.868507385253906e-05,
      "step": 51050,
      "training_step_time": 9.1022367477417
    },
    {
      "epoch": 4.868602752685547e-05,
      "model_forward_time": 1.733426809310913,
      "step": 51051
    },
    {
      "epoch": 4.868602752685547e-05,
      "step": 51051,
      "training_step_time": 7.807748317718506
    },
    {
      "epoch": 4.8686981201171875e-05,
      "model_forward_time": 1.7172338962554932,
      "step": 51052
    },
    {
      "epoch": 4.8686981201171875e-05,
      "step": 51052,
      "training_step_time": 7.084499359130859
    },
    {
      "epoch": 4.868793487548828e-05,
      "model_forward_time": 1.7199475765228271,
      "step": 51053
    },
    {
      "epoch": 4.868793487548828e-05,
      "step": 51053,
      "training_step_time": 7.063046932220459
    },
    {
      "epoch": 4.868888854980469e-05,
      "model_forward_time": 1.7261919975280762,
      "step": 51054
    },
    {
      "epoch": 4.868888854980469e-05,
      "step": 51054,
      "training_step_time": 7.084184646606445
    },
    {
      "epoch": 4.8689842224121093e-05,
      "model_forward_time": 1.7114269733428955,
      "step": 51055
    },
    {
      "epoch": 4.8689842224121093e-05,
      "step": 51055,
      "training_step_time": 7.060964822769165
    },
    {
      "epoch": 4.86907958984375e-05,
      "model_forward_time": 1.7050602436065674,
      "step": 51056
    },
    {
      "epoch": 4.86907958984375e-05,
      "step": 51056,
      "training_step_time": 7.108086109161377
    },
    {
      "epoch": 4.869174957275391e-05,
      "model_forward_time": 1.7053251266479492,
      "step": 51057
    },
    {
      "epoch": 4.869174957275391e-05,
      "step": 51057,
      "training_step_time": 7.046972751617432
    },
    {
      "epoch": 4.869270324707031e-05,
      "model_forward_time": 1.7073607444763184,
      "step": 51058
    },
    {
      "epoch": 4.869270324707031e-05,
      "step": 51058,
      "training_step_time": 7.048574924468994
    },
    {
      "epoch": 4.8693656921386716e-05,
      "model_forward_time": 2.2614591121673584,
      "step": 51059
    },
    {
      "epoch": 4.8693656921386716e-05,
      "step": 51059,
      "training_step_time": 8.998663187026978
    },
    {
      "epoch": 4.869461059570313e-05,
      "grad_norm": 0.11288503557443619,
      "learning_rate": 5.238174835270466e-06,
      "loss": 0.061,
      "step": 51060
    },
    {
      "epoch": 4.869461059570313e-05,
      "model_forward_time": 0.993849515914917,
      "step": 51060
    },
    {
      "dynamics_loss_avg": 0.058773414418101314,
      "epoch": 4.869461059570313e-05,
      "step": 51060
    },
    {
      "action_loss_avg": 0.0020290524524170907,
      "epoch": 4.869461059570313e-05,
      "step": 51060
    },
    {
      "epoch": 4.869461059570313e-05,
      "step": 51060,
      "training_step_time": 8.116982698440552
    },
    {
      "epoch": 4.869556427001953e-05,
      "model_forward_time": 1.7162179946899414,
      "step": 51061
    },
    {
      "epoch": 4.869556427001953e-05,
      "step": 51061,
      "training_step_time": 7.374922752380371
    },
    {
      "epoch": 4.8696517944335935e-05,
      "model_forward_time": 1.7085990905761719,
      "step": 51062
    },
    {
      "epoch": 4.8696517944335935e-05,
      "step": 51062,
      "training_step_time": 7.056359529495239
    },
    {
      "epoch": 4.8697471618652346e-05,
      "model_forward_time": 1.7106366157531738,
      "step": 51063
    },
    {
      "epoch": 4.8697471618652346e-05,
      "step": 51063,
      "training_step_time": 7.059844017028809
    },
    {
      "epoch": 4.869842529296875e-05,
      "model_forward_time": 1.7123520374298096,
      "step": 51064
    },
    {
      "epoch": 4.869842529296875e-05,
      "step": 51064,
      "training_step_time": 7.07315468788147
    },
    {
      "epoch": 4.8699378967285154e-05,
      "model_forward_time": 1.7123494148254395,
      "step": 51065
    },
    {
      "epoch": 4.8699378967285154e-05,
      "step": 51065,
      "training_step_time": 7.042010068893433
    },
    {
      "epoch": 4.8700332641601565e-05,
      "model_forward_time": 1.7115321159362793,
      "step": 51066
    },
    {
      "epoch": 4.8700332641601565e-05,
      "step": 51066,
      "training_step_time": 7.094775915145874
    },
    {
      "epoch": 4.870128631591797e-05,
      "model_forward_time": 1.70796537399292,
      "step": 51067
    },
    {
      "epoch": 4.870128631591797e-05,
      "step": 51067,
      "training_step_time": 7.088727235794067
    },
    {
      "epoch": 4.870223999023437e-05,
      "model_forward_time": 2.2891249656677246,
      "step": 51068
    },
    {
      "epoch": 4.870223999023437e-05,
      "step": 51068,
      "training_step_time": 8.979014158248901
    },
    {
      "epoch": 4.8703193664550784e-05,
      "model_forward_time": 1.7070903778076172,
      "step": 51069
    },
    {
      "epoch": 4.8703193664550784e-05,
      "step": 51069,
      "training_step_time": 9.186721086502075
    },
    {
      "epoch": 4.870414733886719e-05,
      "grad_norm": 0.1036958247423172,
      "learning_rate": 5.23652322946103e-06,
      "loss": 0.0609,
      "step": 51070
    },
    {
      "epoch": 4.870414733886719e-05,
      "model_forward_time": 1.690129280090332,
      "step": 51070
    },
    {
      "dynamics_loss_avg": 0.06965895872563124,
      "epoch": 4.870414733886719e-05,
      "step": 51070
    },
    {
      "action_loss_avg": 0.0023601843509823083,
      "epoch": 4.870414733886719e-05,
      "step": 51070
    },
    {
      "epoch": 4.870414733886719e-05,
      "step": 51070,
      "training_step_time": 7.530947208404541
    },
    {
      "epoch": 4.870510101318359e-05,
      "model_forward_time": 1.701023817062378,
      "step": 51071
    },
    {
      "epoch": 4.870510101318359e-05,
      "step": 51071,
      "training_step_time": 7.096601724624634
    },
    {
      "epoch": 4.87060546875e-05,
      "model_forward_time": 1.6973905563354492,
      "step": 51072
    },
    {
      "epoch": 4.87060546875e-05,
      "step": 51072,
      "training_step_time": 7.098355531692505
    },
    {
      "epoch": 4.870700836181641e-05,
      "model_forward_time": 1.6963424682617188,
      "step": 51073
    },
    {
      "epoch": 4.870700836181641e-05,
      "step": 51073,
      "training_step_time": 7.086066961288452
    },
    {
      "epoch": 4.870796203613281e-05,
      "model_forward_time": 1.6971955299377441,
      "step": 51074
    },
    {
      "epoch": 4.870796203613281e-05,
      "step": 51074,
      "training_step_time": 7.074701547622681
    },
    {
      "epoch": 4.870891571044922e-05,
      "model_forward_time": 1.6995222568511963,
      "step": 51075
    },
    {
      "epoch": 4.870891571044922e-05,
      "step": 51075,
      "training_step_time": 7.056674480438232
    },
    {
      "epoch": 4.8709869384765625e-05,
      "model_forward_time": 1.7007927894592285,
      "step": 51076
    },
    {
      "epoch": 4.8709869384765625e-05,
      "step": 51076,
      "training_step_time": 7.115292310714722
    },
    {
      "epoch": 4.871082305908203e-05,
      "model_forward_time": 2.2252824306488037,
      "step": 51077
    },
    {
      "epoch": 4.871082305908203e-05,
      "step": 51077,
      "training_step_time": 8.47788119316101
    },
    {
      "epoch": 4.871177673339844e-05,
      "model_forward_time": 1.7017138004302979,
      "step": 51078
    },
    {
      "epoch": 4.871177673339844e-05,
      "step": 51078,
      "training_step_time": 8.0915367603302
    },
    {
      "epoch": 4.8712730407714844e-05,
      "model_forward_time": 1.6954765319824219,
      "step": 51079
    },
    {
      "epoch": 4.8712730407714844e-05,
      "step": 51079,
      "training_step_time": 7.837873458862305
    },
    {
      "epoch": 4.871368408203125e-05,
      "grad_norm": 0.09241443127393723,
      "learning_rate": 5.234871597785771e-06,
      "loss": 0.06,
      "step": 51080
    },
    {
      "epoch": 4.871368408203125e-05,
      "model_forward_time": 1.6873135566711426,
      "step": 51080
    },
    {
      "dynamics_loss_avg": 0.050532248243689536,
      "epoch": 4.871368408203125e-05,
      "step": 51080
    },
    {
      "action_loss_avg": 0.001974875424639322,
      "epoch": 4.871368408203125e-05,
      "step": 51080
    },
    {
      "epoch": 4.871368408203125e-05,
      "step": 51080,
      "training_step_time": 6.820375680923462
    },
    {
      "epoch": 4.871463775634766e-05,
      "model_forward_time": 1.6968135833740234,
      "step": 51081
    },
    {
      "epoch": 4.871463775634766e-05,
      "step": 51081,
      "training_step_time": 7.064261436462402
    },
    {
      "epoch": 4.871559143066406e-05,
      "model_forward_time": 1.7006995677947998,
      "step": 51082
    },
    {
      "epoch": 4.871559143066406e-05,
      "step": 51082,
      "training_step_time": 7.0968406200408936
    },
    {
      "epoch": 4.871654510498047e-05,
      "model_forward_time": 1.6972615718841553,
      "step": 51083
    },
    {
      "epoch": 4.871654510498047e-05,
      "step": 51083,
      "training_step_time": 7.123935222625732
    },
    {
      "epoch": 4.871749877929688e-05,
      "model_forward_time": 1.7042806148529053,
      "step": 51084
    },
    {
      "epoch": 4.871749877929688e-05,
      "step": 51084,
      "training_step_time": 7.1311399936676025
    },
    {
      "epoch": 4.871845245361328e-05,
      "model_forward_time": 1.7008593082427979,
      "step": 51085
    },
    {
      "epoch": 4.871845245361328e-05,
      "step": 51085,
      "training_step_time": 7.148166656494141
    },
    {
      "epoch": 4.8719406127929686e-05,
      "model_forward_time": 1.7192485332489014,
      "step": 51086
    },
    {
      "epoch": 4.8719406127929686e-05,
      "step": 51086,
      "training_step_time": 7.969818830490112
    },
    {
      "epoch": 4.87203598022461e-05,
      "model_forward_time": 2.270829677581787,
      "step": 51087
    },
    {
      "epoch": 4.87203598022461e-05,
      "step": 51087,
      "training_step_time": 10.455669403076172
    },
    {
      "epoch": 4.87213134765625e-05,
      "model_forward_time": 1.7000923156738281,
      "step": 51088
    },
    {
      "epoch": 4.87213134765625e-05,
      "step": 51088,
      "training_step_time": 8.11162281036377
    },
    {
      "epoch": 4.8722267150878905e-05,
      "model_forward_time": 1.6992690563201904,
      "step": 51089
    },
    {
      "epoch": 4.8722267150878905e-05,
      "step": 51089,
      "training_step_time": 7.4011242389678955
    },
    {
      "epoch": 4.8723220825195316e-05,
      "grad_norm": 0.10629520565271378,
      "learning_rate": 5.233219940425306e-06,
      "loss": 0.0595,
      "step": 51090
    },
    {
      "epoch": 4.8723220825195316e-05,
      "model_forward_time": 1.7000691890716553,
      "step": 51090
    },
    {
      "dynamics_loss_avg": 0.046663519367575644,
      "epoch": 4.8723220825195316e-05,
      "step": 51090
    },
    {
      "action_loss_avg": 0.0014722187363076956,
      "epoch": 4.8723220825195316e-05,
      "step": 51090
    },
    {
      "epoch": 4.8723220825195316e-05,
      "step": 51090,
      "training_step_time": 7.051701307296753
    },
    {
      "epoch": 4.872417449951172e-05,
      "model_forward_time": 1.7044978141784668,
      "step": 51091
    },
    {
      "epoch": 4.872417449951172e-05,
      "step": 51091,
      "training_step_time": 7.067194700241089
    },
    {
      "epoch": 4.8725128173828124e-05,
      "model_forward_time": 1.694977045059204,
      "step": 51092
    },
    {
      "epoch": 4.8725128173828124e-05,
      "step": 51092,
      "training_step_time": 7.077914714813232
    },
    {
      "epoch": 4.8726081848144535e-05,
      "model_forward_time": 1.7028474807739258,
      "step": 51093
    },
    {
      "epoch": 4.8726081848144535e-05,
      "step": 51093,
      "training_step_time": 7.050695896148682
    },
    {
      "epoch": 4.872703552246094e-05,
      "model_forward_time": 1.7004938125610352,
      "step": 51094
    },
    {
      "epoch": 4.872703552246094e-05,
      "step": 51094,
      "training_step_time": 7.08463454246521
    },
    {
      "epoch": 4.872798919677734e-05,
      "model_forward_time": 1.7093174457550049,
      "step": 51095
    },
    {
      "epoch": 4.872798919677734e-05,
      "step": 51095,
      "training_step_time": 7.992796421051025
    },
    {
      "epoch": 4.872894287109375e-05,
      "model_forward_time": 2.2658333778381348,
      "step": 51096
    },
    {
      "epoch": 4.872894287109375e-05,
      "step": 51096,
      "training_step_time": 9.839315176010132
    },
    {
      "epoch": 4.872989654541016e-05,
      "model_forward_time": 1.7073817253112793,
      "step": 51097
    },
    {
      "epoch": 4.872989654541016e-05,
      "step": 51097,
      "training_step_time": 8.201787233352661
    },
    {
      "epoch": 4.873085021972656e-05,
      "model_forward_time": 1.7005813121795654,
      "step": 51098
    },
    {
      "epoch": 4.873085021972656e-05,
      "step": 51098,
      "training_step_time": 7.9891932010650635
    },
    {
      "epoch": 4.8731803894042966e-05,
      "model_forward_time": 1.6988441944122314,
      "step": 51099
    },
    {
      "epoch": 4.8731803894042966e-05,
      "step": 51099,
      "training_step_time": 7.12226676940918
    },
    {
      "epoch": 4.8732757568359376e-05,
      "grad_norm": 0.09947412461042404,
      "learning_rate": 5.231568257560258e-06,
      "loss": 0.0568,
      "step": 51100
    },
    {
      "epoch": 4.8732757568359376e-05,
      "model_forward_time": 1.7032194137573242,
      "step": 51100
    },
    {
      "dynamics_loss_avg": 0.04789381716400385,
      "epoch": 4.8732757568359376e-05,
      "step": 51100
    },
    {
      "action_loss_avg": 0.002704247315705288,
      "epoch": 4.8732757568359376e-05,
      "step": 51100
    },
    {
      "epoch": 4.8732757568359376e-05,
      "step": 51100,
      "training_step_time": 7.090850114822388
    },
    {
      "epoch": 4.873371124267578e-05,
      "model_forward_time": 1.705512523651123,
      "step": 51101
    },
    {
      "epoch": 4.873371124267578e-05,
      "step": 51101,
      "training_step_time": 7.096890211105347
    },
    {
      "epoch": 4.8734664916992184e-05,
      "model_forward_time": 1.6907832622528076,
      "step": 51102
    },
    {
      "epoch": 4.8734664916992184e-05,
      "step": 51102,
      "training_step_time": 7.007331848144531
    },
    {
      "epoch": 4.8735618591308595e-05,
      "model_forward_time": 1.702059030532837,
      "step": 51103
    },
    {
      "epoch": 4.8735618591308595e-05,
      "step": 51103,
      "training_step_time": 7.049948692321777
    },
    {
      "epoch": 4.8736572265625e-05,
      "model_forward_time": 1.7042269706726074,
      "step": 51104
    },
    {
      "epoch": 4.8736572265625e-05,
      "step": 51104,
      "training_step_time": 7.912779331207275
    },
    {
      "epoch": 4.87375259399414e-05,
      "model_forward_time": 2.293482780456543,
      "step": 51105
    },
    {
      "epoch": 4.87375259399414e-05,
      "step": 51105,
      "training_step_time": 8.81494402885437
    },
    {
      "epoch": 4.8738479614257814e-05,
      "model_forward_time": 1.7042701244354248,
      "step": 51106
    },
    {
      "epoch": 4.8738479614257814e-05,
      "step": 51106,
      "training_step_time": 8.14396333694458
    },
    {
      "epoch": 4.873943328857422e-05,
      "model_forward_time": 1.7039988040924072,
      "step": 51107
    },
    {
      "epoch": 4.873943328857422e-05,
      "step": 51107,
      "training_step_time": 7.960442543029785
    },
    {
      "epoch": 4.874038696289062e-05,
      "model_forward_time": 1.7027943134307861,
      "step": 51108
    },
    {
      "epoch": 4.874038696289062e-05,
      "step": 51108,
      "training_step_time": 7.029028654098511
    },
    {
      "epoch": 4.874134063720703e-05,
      "model_forward_time": 1.7077691555023193,
      "step": 51109
    },
    {
      "epoch": 4.874134063720703e-05,
      "step": 51109,
      "training_step_time": 7.0582215785980225
    },
    {
      "epoch": 4.874229431152344e-05,
      "grad_norm": 0.0977008268237114,
      "learning_rate": 5.229916549371255e-06,
      "loss": 0.057,
      "step": 51110
    },
    {
      "epoch": 4.874229431152344e-05,
      "model_forward_time": 1.700817346572876,
      "step": 51110
    },
    {
      "dynamics_loss_avg": 0.043518321588635445,
      "epoch": 4.874229431152344e-05,
      "step": 51110
    },
    {
      "action_loss_avg": 0.0023456291819456965,
      "epoch": 4.874229431152344e-05,
      "step": 51110
    },
    {
      "epoch": 4.874229431152344e-05,
      "step": 51110,
      "training_step_time": 7.047353506088257
    },
    {
      "epoch": 4.874324798583984e-05,
      "model_forward_time": 1.703324556350708,
      "step": 51111
    },
    {
      "epoch": 4.874324798583984e-05,
      "step": 51111,
      "training_step_time": 7.0541465282440186
    },
    {
      "epoch": 4.874420166015625e-05,
      "model_forward_time": 1.702570915222168,
      "step": 51112
    },
    {
      "epoch": 4.874420166015625e-05,
      "step": 51112,
      "training_step_time": 7.021718978881836
    },
    {
      "epoch": 4.8745155334472656e-05,
      "model_forward_time": 1.698484182357788,
      "step": 51113
    },
    {
      "epoch": 4.8745155334472656e-05,
      "step": 51113,
      "training_step_time": 7.933894157409668
    },
    {
      "epoch": 4.874610900878906e-05,
      "model_forward_time": 1.893754005432129,
      "step": 51114
    },
    {
      "epoch": 4.874610900878906e-05,
      "step": 51114,
      "training_step_time": 9.497798442840576
    },
    {
      "epoch": 4.874706268310547e-05,
      "model_forward_time": 1.695033073425293,
      "step": 51115
    },
    {
      "epoch": 4.874706268310547e-05,
      "step": 51115,
      "training_step_time": 10.057338953018188
    },
    {
      "epoch": 4.8748016357421875e-05,
      "model_forward_time": 1.6941611766815186,
      "step": 51116
    },
    {
      "epoch": 4.8748016357421875e-05,
      "step": 51116,
      "training_step_time": 7.941101312637329
    },
    {
      "epoch": 4.874897003173828e-05,
      "model_forward_time": 1.698007583618164,
      "step": 51117
    },
    {
      "epoch": 4.874897003173828e-05,
      "step": 51117,
      "training_step_time": 7.367416143417358
    },
    {
      "epoch": 4.874992370605469e-05,
      "model_forward_time": 1.7025229930877686,
      "step": 51118
    },
    {
      "epoch": 4.874992370605469e-05,
      "step": 51118,
      "training_step_time": 7.051093339920044
    },
    {
      "epoch": 4.8750877380371094e-05,
      "model_forward_time": 1.697343111038208,
      "step": 51119
    },
    {
      "epoch": 4.8750877380371094e-05,
      "step": 51119,
      "training_step_time": 7.078021764755249
    },
    {
      "epoch": 4.87518310546875e-05,
      "grad_norm": 0.11961058527231216,
      "learning_rate": 5.2282648160389235e-06,
      "loss": 0.062,
      "step": 51120
    },
    {
      "epoch": 4.87518310546875e-05,
      "model_forward_time": 1.6920084953308105,
      "step": 51120
    },
    {
      "dynamics_loss_avg": 0.0802593968808651,
      "epoch": 4.87518310546875e-05,
      "step": 51120
    },
    {
      "action_loss_avg": 0.0047058641066541895,
      "epoch": 4.87518310546875e-05,
      "step": 51120
    },
    {
      "epoch": 4.87518310546875e-05,
      "step": 51120,
      "training_step_time": 7.01094388961792
    },
    {
      "epoch": 4.875278472900391e-05,
      "model_forward_time": 1.6953814029693604,
      "step": 51121
    },
    {
      "epoch": 4.875278472900391e-05,
      "step": 51121,
      "training_step_time": 7.083791494369507
    },
    {
      "epoch": 4.875373840332031e-05,
      "model_forward_time": 1.6922433376312256,
      "step": 51122
    },
    {
      "epoch": 4.875373840332031e-05,
      "step": 51122,
      "training_step_time": 7.945380449295044
    },
    {
      "epoch": 4.8754692077636716e-05,
      "model_forward_time": 2.2540271282196045,
      "step": 51123
    },
    {
      "epoch": 4.8754692077636716e-05,
      "step": 51123,
      "training_step_time": 8.914790868759155
    },
    {
      "epoch": 4.875564575195313e-05,
      "model_forward_time": 1.6960980892181396,
      "step": 51124
    },
    {
      "epoch": 4.875564575195313e-05,
      "step": 51124,
      "training_step_time": 8.875075817108154
    },
    {
      "epoch": 4.875659942626953e-05,
      "model_forward_time": 1.6990554332733154,
      "step": 51125
    },
    {
      "epoch": 4.875659942626953e-05,
      "step": 51125,
      "training_step_time": 8.042295455932617
    },
    {
      "epoch": 4.8757553100585935e-05,
      "model_forward_time": 1.6981637477874756,
      "step": 51126
    },
    {
      "epoch": 4.8757553100585935e-05,
      "step": 51126,
      "training_step_time": 7.341077089309692
    },
    {
      "epoch": 4.8758506774902346e-05,
      "model_forward_time": 1.6983611583709717,
      "step": 51127
    },
    {
      "epoch": 4.8758506774902346e-05,
      "step": 51127,
      "training_step_time": 6.9955432415008545
    },
    {
      "epoch": 4.875946044921875e-05,
      "model_forward_time": 1.7010750770568848,
      "step": 51128
    },
    {
      "epoch": 4.875946044921875e-05,
      "step": 51128,
      "training_step_time": 6.9803550243377686
    },
    {
      "epoch": 4.8760414123535154e-05,
      "model_forward_time": 1.6954212188720703,
      "step": 51129
    },
    {
      "epoch": 4.8760414123535154e-05,
      "step": 51129,
      "training_step_time": 7.020109415054321
    },
    {
      "epoch": 4.8761367797851565e-05,
      "grad_norm": 0.13920368254184723,
      "learning_rate": 5.2266130577438935e-06,
      "loss": 0.0601,
      "step": 51130
    },
    {
      "epoch": 4.8761367797851565e-05,
      "model_forward_time": 1.6928892135620117,
      "step": 51130
    },
    {
      "dynamics_loss_avg": 0.047515802457928656,
      "epoch": 4.8761367797851565e-05,
      "step": 51130
    },
    {
      "action_loss_avg": 0.004517476778710261,
      "epoch": 4.8761367797851565e-05,
      "step": 51130
    },
    {
      "epoch": 4.8761367797851565e-05,
      "step": 51130,
      "training_step_time": 7.041944742202759
    },
    {
      "epoch": 4.876232147216797e-05,
      "model_forward_time": 1.6969869136810303,
      "step": 51131
    },
    {
      "epoch": 4.876232147216797e-05,
      "step": 51131,
      "training_step_time": 8.016210556030273
    },
    {
      "epoch": 4.876327514648437e-05,
      "model_forward_time": 1.7024486064910889,
      "step": 51132
    },
    {
      "epoch": 4.876327514648437e-05,
      "step": 51132,
      "training_step_time": 7.847919702529907
    },
    {
      "epoch": 4.8764228820800784e-05,
      "model_forward_time": 2.272069215774536,
      "step": 51133
    },
    {
      "epoch": 4.8764228820800784e-05,
      "step": 51133,
      "training_step_time": 9.868722438812256
    },
    {
      "epoch": 4.876518249511719e-05,
      "model_forward_time": 1.6988279819488525,
      "step": 51134
    },
    {
      "epoch": 4.876518249511719e-05,
      "step": 51134,
      "training_step_time": 7.911207675933838
    },
    {
      "epoch": 4.876613616943359e-05,
      "model_forward_time": 1.697056770324707,
      "step": 51135
    },
    {
      "epoch": 4.876613616943359e-05,
      "step": 51135,
      "training_step_time": 7.742670059204102
    },
    {
      "epoch": 4.876708984375e-05,
      "model_forward_time": 1.6977641582489014,
      "step": 51136
    },
    {
      "epoch": 4.876708984375e-05,
      "step": 51136,
      "training_step_time": 7.069932460784912
    },
    {
      "epoch": 4.876804351806641e-05,
      "model_forward_time": 1.7021517753601074,
      "step": 51137
    },
    {
      "epoch": 4.876804351806641e-05,
      "step": 51137,
      "training_step_time": 7.000845432281494
    },
    {
      "epoch": 4.876899719238281e-05,
      "model_forward_time": 1.6749911308288574,
      "step": 51138
    },
    {
      "epoch": 4.876899719238281e-05,
      "step": 51138,
      "training_step_time": 7.012484788894653
    },
    {
      "epoch": 4.876995086669922e-05,
      "model_forward_time": 1.6972110271453857,
      "step": 51139
    },
    {
      "epoch": 4.876995086669922e-05,
      "step": 51139,
      "training_step_time": 7.003586530685425
    },
    {
      "epoch": 4.8770904541015626e-05,
      "grad_norm": 0.15901830792427063,
      "learning_rate": 5.224961274666802e-06,
      "loss": 0.061,
      "step": 51140
    },
    {
      "epoch": 4.8770904541015626e-05,
      "model_forward_time": 1.6930372714996338,
      "step": 51140
    },
    {
      "dynamics_loss_avg": 0.07629032880067825,
      "epoch": 4.8770904541015626e-05,
      "step": 51140
    },
    {
      "action_loss_avg": 0.0041966777294874195,
      "epoch": 4.8770904541015626e-05,
      "step": 51140
    },
    {
      "epoch": 4.8770904541015626e-05,
      "step": 51140,
      "training_step_time": 7.96186089515686
    },
    {
      "epoch": 4.877185821533203e-05,
      "model_forward_time": 1.702735424041748,
      "step": 51141
    },
    {
      "epoch": 4.877185821533203e-05,
      "step": 51141,
      "training_step_time": 8.411020994186401
    },
    {
      "epoch": 4.877281188964844e-05,
      "model_forward_time": 2.272205114364624,
      "step": 51142
    },
    {
      "epoch": 4.877281188964844e-05,
      "step": 51142,
      "training_step_time": 9.619921207427979
    },
    {
      "epoch": 4.8773765563964844e-05,
      "model_forward_time": 1.7009837627410889,
      "step": 51143
    },
    {
      "epoch": 4.8773765563964844e-05,
      "step": 51143,
      "training_step_time": 8.173213481903076
    },
    {
      "epoch": 4.877471923828125e-05,
      "model_forward_time": 1.694990634918213,
      "step": 51144
    },
    {
      "epoch": 4.877471923828125e-05,
      "step": 51144,
      "training_step_time": 7.890588283538818
    },
    {
      "epoch": 4.877567291259766e-05,
      "model_forward_time": 1.7006771564483643,
      "step": 51145
    },
    {
      "epoch": 4.877567291259766e-05,
      "step": 51145,
      "training_step_time": 7.0668230056762695
    },
    {
      "epoch": 4.877662658691406e-05,
      "model_forward_time": 1.6983840465545654,
      "step": 51146
    },
    {
      "epoch": 4.877662658691406e-05,
      "step": 51146,
      "training_step_time": 7.069921493530273
    },
    {
      "epoch": 4.877758026123047e-05,
      "model_forward_time": 1.6939351558685303,
      "step": 51147
    },
    {
      "epoch": 4.877758026123047e-05,
      "step": 51147,
      "training_step_time": 7.060818433761597
    },
    {
      "epoch": 4.877853393554688e-05,
      "model_forward_time": 1.6976029872894287,
      "step": 51148
    },
    {
      "epoch": 4.877853393554688e-05,
      "step": 51148,
      "training_step_time": 7.034221172332764
    },
    {
      "epoch": 4.877948760986328e-05,
      "model_forward_time": 1.6965341567993164,
      "step": 51149
    },
    {
      "epoch": 4.877948760986328e-05,
      "step": 51149,
      "training_step_time": 7.9487998485565186
    },
    {
      "epoch": 4.8780441284179686e-05,
      "grad_norm": 0.10605153441429138,
      "learning_rate": 5.223309466988283e-06,
      "loss": 0.0597,
      "step": 51150
    },
    {
      "epoch": 4.8780441284179686e-05,
      "model_forward_time": 1.6953012943267822,
      "step": 51150
    },
    {
      "dynamics_loss_avg": 0.06583248861134053,
      "epoch": 4.8780441284179686e-05,
      "step": 51150
    },
    {
      "action_loss_avg": 0.0030159888439811764,
      "epoch": 4.8780441284179686e-05,
      "step": 51150
    },
    {
      "epoch": 4.8780441284179686e-05,
      "step": 51150,
      "training_step_time": 7.566194295883179
    },
    {
      "epoch": 4.87813949584961e-05,
      "model_forward_time": 2.2822976112365723,
      "step": 51151
    },
    {
      "epoch": 4.87813949584961e-05,
      "step": 51151,
      "training_step_time": 9.978619575500488
    },
    {
      "epoch": 4.87823486328125e-05,
      "model_forward_time": 1.70050048828125,
      "step": 51152
    },
    {
      "epoch": 4.87823486328125e-05,
      "step": 51152,
      "training_step_time": 9.048822402954102
    },
    {
      "epoch": 4.8783302307128905e-05,
      "model_forward_time": 1.6956944465637207,
      "step": 51153
    },
    {
      "epoch": 4.8783302307128905e-05,
      "step": 51153,
      "training_step_time": 7.936854362487793
    },
    {
      "epoch": 4.8784255981445316e-05,
      "model_forward_time": 1.704596996307373,
      "step": 51154
    },
    {
      "epoch": 4.8784255981445316e-05,
      "step": 51154,
      "training_step_time": 7.002912759780884
    },
    {
      "epoch": 4.878520965576172e-05,
      "model_forward_time": 1.6923246383666992,
      "step": 51155
    },
    {
      "epoch": 4.878520965576172e-05,
      "step": 51155,
      "training_step_time": 7.0283098220825195
    },
    {
      "epoch": 4.8786163330078124e-05,
      "model_forward_time": 1.6957051753997803,
      "step": 51156
    },
    {
      "epoch": 4.8786163330078124e-05,
      "step": 51156,
      "training_step_time": 7.065936803817749
    },
    {
      "epoch": 4.878711700439453e-05,
      "model_forward_time": 1.6973161697387695,
      "step": 51157
    },
    {
      "epoch": 4.878711700439453e-05,
      "step": 51157,
      "training_step_time": 7.046306848526001
    },
    {
      "epoch": 4.878807067871094e-05,
      "model_forward_time": 1.7027373313903809,
      "step": 51158
    },
    {
      "epoch": 4.878807067871094e-05,
      "step": 51158,
      "training_step_time": 7.035871744155884
    },
    {
      "epoch": 4.878902435302734e-05,
      "model_forward_time": 1.701275110244751,
      "step": 51159
    },
    {
      "epoch": 4.878902435302734e-05,
      "step": 51159,
      "training_step_time": 8.027250528335571
    },
    {
      "epoch": 4.878997802734375e-05,
      "grad_norm": 0.09125610440969467,
      "learning_rate": 5.221657634888976e-06,
      "loss": 0.0581,
      "step": 51160
    },
    {
      "epoch": 4.878997802734375e-05,
      "model_forward_time": 2.24636173248291,
      "step": 51160
    },
    {
      "dynamics_loss_avg": 0.07138515766710044,
      "epoch": 4.878997802734375e-05,
      "step": 51160
    },
    {
      "action_loss_avg": 0.0027987589244730772,
      "epoch": 4.878997802734375e-05,
      "step": 51160
    },
    {
      "epoch": 4.878997802734375e-05,
      "step": 51160,
      "training_step_time": 9.10430121421814
    },
    {
      "epoch": 4.879093170166016e-05,
      "model_forward_time": 1.7281153202056885,
      "step": 51161
    },
    {
      "epoch": 4.879093170166016e-05,
      "step": 51161,
      "training_step_time": 9.107414484024048
    },
    {
      "epoch": 4.879188537597656e-05,
      "model_forward_time": 1.6916704177856445,
      "step": 51162
    },
    {
      "epoch": 4.879188537597656e-05,
      "step": 51162,
      "training_step_time": 8.109524726867676
    },
    {
      "epoch": 4.8792839050292966e-05,
      "model_forward_time": 1.7041285037994385,
      "step": 51163
    },
    {
      "epoch": 4.8792839050292966e-05,
      "step": 51163,
      "training_step_time": 7.754289150238037
    },
    {
      "epoch": 4.8793792724609376e-05,
      "model_forward_time": 1.7009954452514648,
      "step": 51164
    },
    {
      "epoch": 4.8793792724609376e-05,
      "step": 51164,
      "training_step_time": 7.084981203079224
    },
    {
      "epoch": 4.879474639892578e-05,
      "model_forward_time": 1.6965947151184082,
      "step": 51165
    },
    {
      "epoch": 4.879474639892578e-05,
      "step": 51165,
      "training_step_time": 7.0995564460754395
    },
    {
      "epoch": 4.8795700073242184e-05,
      "model_forward_time": 1.6930639743804932,
      "step": 51166
    },
    {
      "epoch": 4.8795700073242184e-05,
      "step": 51166,
      "training_step_time": 7.051705837249756
    },
    {
      "epoch": 4.8796653747558595e-05,
      "model_forward_time": 1.7077996730804443,
      "step": 51167
    },
    {
      "epoch": 4.8796653747558595e-05,
      "step": 51167,
      "training_step_time": 6.791810989379883
    },
    {
      "epoch": 4.8797607421875e-05,
      "model_forward_time": 1.6996078491210938,
      "step": 51168
    },
    {
      "epoch": 4.8797607421875e-05,
      "step": 51168,
      "training_step_time": 8.058536529541016
    },
    {
      "epoch": 4.87985610961914e-05,
      "model_forward_time": 1.7019004821777344,
      "step": 51169
    },
    {
      "epoch": 4.87985610961914e-05,
      "step": 51169,
      "training_step_time": 10.959760427474976
    },
    {
      "epoch": 4.8799514770507814e-05,
      "grad_norm": 0.09362386912107468,
      "learning_rate": 5.220005778549524e-06,
      "loss": 0.0548,
      "step": 51170
    },
    {
      "epoch": 4.8799514770507814e-05,
      "model_forward_time": 1.7104008197784424,
      "step": 51170
    },
    {
      "dynamics_loss_avg": 0.051519674528390166,
      "epoch": 4.8799514770507814e-05,
      "step": 51170
    },
    {
      "action_loss_avg": 0.0028371091873850675,
      "epoch": 4.8799514770507814e-05,
      "step": 51170
    },
    {
      "epoch": 4.8799514770507814e-05,
      "step": 51170,
      "training_step_time": 8.076941013336182
    },
    {
      "epoch": 4.880046844482422e-05,
      "model_forward_time": 1.7013163566589355,
      "step": 51171
    },
    {
      "epoch": 4.880046844482422e-05,
      "step": 51171,
      "training_step_time": 7.824886798858643
    },
    {
      "epoch": 4.880142211914062e-05,
      "model_forward_time": 0.9811289310455322,
      "step": 51172
    },
    {
      "epoch": 4.880142211914062e-05,
      "step": 51172,
      "training_step_time": 7.883939981460571
    },
    {
      "epoch": 4.880237579345703e-05,
      "model_forward_time": 1.7015151977539062,
      "step": 51173
    },
    {
      "epoch": 4.880237579345703e-05,
      "step": 51173,
      "training_step_time": 6.996180295944214
    },
    {
      "epoch": 4.880332946777344e-05,
      "model_forward_time": 1.699720859527588,
      "step": 51174
    },
    {
      "epoch": 4.880332946777344e-05,
      "step": 51174,
      "training_step_time": 7.199672222137451
    },
    {
      "epoch": 4.880428314208984e-05,
      "model_forward_time": 1.6969690322875977,
      "step": 51175
    },
    {
      "epoch": 4.880428314208984e-05,
      "step": 51175,
      "training_step_time": 7.19270658493042
    },
    {
      "epoch": 4.880523681640625e-05,
      "model_forward_time": 1.6987419128417969,
      "step": 51176
    },
    {
      "epoch": 4.880523681640625e-05,
      "step": 51176,
      "training_step_time": 7.038211822509766
    },
    {
      "epoch": 4.8806190490722656e-05,
      "model_forward_time": 1.692077398300171,
      "step": 51177
    },
    {
      "epoch": 4.8806190490722656e-05,
      "step": 51177,
      "training_step_time": 8.073697566986084
    },
    {
      "epoch": 4.880714416503906e-05,
      "model_forward_time": 1.7035365104675293,
      "step": 51178
    },
    {
      "epoch": 4.880714416503906e-05,
      "step": 51178,
      "training_step_time": 9.416694164276123
    },
    {
      "epoch": 4.880809783935547e-05,
      "model_forward_time": 2.2498488426208496,
      "step": 51179
    },
    {
      "epoch": 4.880809783935547e-05,
      "step": 51179,
      "training_step_time": 10.287565469741821
    },
    {
      "epoch": 4.8809051513671875e-05,
      "grad_norm": 0.1299564093351364,
      "learning_rate": 5.218353898150571e-06,
      "loss": 0.0587,
      "step": 51180
    },
    {
      "epoch": 4.8809051513671875e-05,
      "model_forward_time": 1.6932382583618164,
      "step": 51180
    },
    {
      "dynamics_loss_avg": 0.08095220997929573,
      "epoch": 4.8809051513671875e-05,
      "step": 51180
    },
    {
      "action_loss_avg": 0.003561486332910135,
      "epoch": 4.8809051513671875e-05,
      "step": 51180
    },
    {
      "epoch": 4.8809051513671875e-05,
      "step": 51180,
      "training_step_time": 8.022676944732666
    },
    {
      "epoch": 4.881000518798828e-05,
      "model_forward_time": 1.6916186809539795,
      "step": 51181
    },
    {
      "epoch": 4.881000518798828e-05,
      "step": 51181,
      "training_step_time": 7.570996999740601
    },
    {
      "epoch": 4.881095886230469e-05,
      "model_forward_time": 1.6988818645477295,
      "step": 51182
    },
    {
      "epoch": 4.881095886230469e-05,
      "step": 51182,
      "training_step_time": 7.378357648849487
    },
    {
      "epoch": 4.8811912536621094e-05,
      "model_forward_time": 1.6917972564697266,
      "step": 51183
    },
    {
      "epoch": 4.8811912536621094e-05,
      "step": 51183,
      "training_step_time": 6.997628450393677
    },
    {
      "epoch": 4.88128662109375e-05,
      "model_forward_time": 1.6922640800476074,
      "step": 51184
    },
    {
      "epoch": 4.88128662109375e-05,
      "step": 51184,
      "training_step_time": 6.993328809738159
    },
    {
      "epoch": 4.881381988525391e-05,
      "model_forward_time": 1.6920993328094482,
      "step": 51185
    },
    {
      "epoch": 4.881381988525391e-05,
      "step": 51185,
      "training_step_time": 6.984729766845703
    },
    {
      "epoch": 4.881477355957031e-05,
      "model_forward_time": 1.31760835647583,
      "step": 51186
    },
    {
      "epoch": 4.881477355957031e-05,
      "step": 51186,
      "training_step_time": 7.986993789672852
    },
    {
      "epoch": 4.8815727233886716e-05,
      "model_forward_time": 1.6928069591522217,
      "step": 51187
    },
    {
      "epoch": 4.8815727233886716e-05,
      "step": 51187,
      "training_step_time": 7.908519983291626
    },
    {
      "epoch": 4.881668090820313e-05,
      "model_forward_time": 2.2792108058929443,
      "step": 51188
    },
    {
      "epoch": 4.881668090820313e-05,
      "step": 51188,
      "training_step_time": 9.403793811798096
    },
    {
      "epoch": 4.881763458251953e-05,
      "model_forward_time": 1.705446481704712,
      "step": 51189
    },
    {
      "epoch": 4.881763458251953e-05,
      "step": 51189,
      "training_step_time": 9.537206411361694
    },
    {
      "epoch": 4.8818588256835935e-05,
      "grad_norm": 0.11751854419708252,
      "learning_rate": 5.216701993872763e-06,
      "loss": 0.0555,
      "step": 51190
    },
    {
      "epoch": 4.8818588256835935e-05,
      "model_forward_time": 1.7031700611114502,
      "step": 51190
    },
    {
      "dynamics_loss_avg": 0.06662366334348917,
      "epoch": 4.8818588256835935e-05,
      "step": 51190
    },
    {
      "action_loss_avg": 0.001766736072022468,
      "epoch": 4.8818588256835935e-05,
      "step": 51190
    },
    {
      "epoch": 4.8818588256835935e-05,
      "step": 51190,
      "training_step_time": 8.67101764678955
    },
    {
      "epoch": 4.8819541931152346e-05,
      "model_forward_time": 1.7033257484436035,
      "step": 51191
    },
    {
      "epoch": 4.8819541931152346e-05,
      "step": 51191,
      "training_step_time": 7.804672718048096
    },
    {
      "epoch": 4.882049560546875e-05,
      "model_forward_time": 1.7028672695159912,
      "step": 51192
    },
    {
      "epoch": 4.882049560546875e-05,
      "step": 51192,
      "training_step_time": 6.95285439491272
    },
    {
      "epoch": 4.8821449279785154e-05,
      "model_forward_time": 1.716005802154541,
      "step": 51193
    },
    {
      "epoch": 4.8821449279785154e-05,
      "step": 51193,
      "training_step_time": 7.02389669418335
    },
    {
      "epoch": 4.8822402954101565e-05,
      "model_forward_time": 1.7092523574829102,
      "step": 51194
    },
    {
      "epoch": 4.8822402954101565e-05,
      "step": 51194,
      "training_step_time": 6.996461629867554
    },
    {
      "epoch": 4.882335662841797e-05,
      "model_forward_time": 1.7118823528289795,
      "step": 51195
    },
    {
      "epoch": 4.882335662841797e-05,
      "step": 51195,
      "training_step_time": 7.983576536178589
    },
    {
      "epoch": 4.882431030273437e-05,
      "model_forward_time": 1.70444655418396,
      "step": 51196
    },
    {
      "epoch": 4.882431030273437e-05,
      "step": 51196,
      "training_step_time": 8.496222734451294
    },
    {
      "epoch": 4.8825263977050784e-05,
      "model_forward_time": 1.7055542469024658,
      "step": 51197
    },
    {
      "epoch": 4.8825263977050784e-05,
      "step": 51197,
      "training_step_time": 8.13968276977539
    },
    {
      "epoch": 4.882621765136719e-05,
      "model_forward_time": 2.2613866329193115,
      "step": 51198
    },
    {
      "epoch": 4.882621765136719e-05,
      "step": 51198,
      "training_step_time": 8.571231603622437
    },
    {
      "epoch": 4.882717132568359e-05,
      "model_forward_time": 1.7107465267181396,
      "step": 51199
    },
    {
      "epoch": 4.882717132568359e-05,
      "step": 51199,
      "training_step_time": 7.824028491973877
    },
    {
      "epoch": 4.8828125e-05,
      "grad_norm": 0.11898910254240036,
      "learning_rate": 5.215050065896752e-06,
      "loss": 0.0549,
      "step": 51200
    },
    {
      "epoch": 4.8828125e-05,
      "model_forward_time": 1.7038822174072266,
      "step": 51200
    },
    {
      "dynamics_loss_avg": 0.07018936276435853,
      "epoch": 4.8828125e-05,
      "step": 51200
    },
    {
      "action_loss_avg": 0.0020236124750226737,
      "epoch": 4.8828125e-05,
      "step": 51200
    },
    {
      "epoch": 4.8828125e-05,
      "step": 51200,
      "training_step_time": 7.773803472518921
    },
    {
      "epoch": 4.882907867431641e-05,
      "model_forward_time": 1.7051212787628174,
      "step": 51201
    },
    {
      "epoch": 4.882907867431641e-05,
      "step": 51201,
      "training_step_time": 6.998192548751831
    },
    {
      "epoch": 4.883003234863281e-05,
      "model_forward_time": 1.7237694263458252,
      "step": 51202
    },
    {
      "epoch": 4.883003234863281e-05,
      "step": 51202,
      "training_step_time": 6.991778373718262
    },
    {
      "epoch": 4.883098602294922e-05,
      "model_forward_time": 1.7150564193725586,
      "step": 51203
    },
    {
      "epoch": 4.883098602294922e-05,
      "step": 51203,
      "training_step_time": 6.997892379760742
    },
    {
      "epoch": 4.8831939697265626e-05,
      "model_forward_time": 1.7047934532165527,
      "step": 51204
    },
    {
      "epoch": 4.8831939697265626e-05,
      "step": 51204,
      "training_step_time": 7.5506603717803955
    },
    {
      "epoch": 4.883289337158203e-05,
      "model_forward_time": 1.7062125205993652,
      "step": 51205
    },
    {
      "epoch": 4.883289337158203e-05,
      "step": 51205,
      "training_step_time": 8.025240182876587
    },
    {
      "epoch": 4.883384704589844e-05,
      "model_forward_time": 1.7367308139801025,
      "step": 51206
    },
    {
      "epoch": 4.883384704589844e-05,
      "step": 51206,
      "training_step_time": 9.504168510437012
    },
    {
      "epoch": 4.8834800720214844e-05,
      "model_forward_time": 2.2597482204437256,
      "step": 51207
    },
    {
      "epoch": 4.8834800720214844e-05,
      "step": 51207,
      "training_step_time": 9.782163858413696
    },
    {
      "epoch": 4.883575439453125e-05,
      "model_forward_time": 1.7057180404663086,
      "step": 51208
    },
    {
      "epoch": 4.883575439453125e-05,
      "step": 51208,
      "training_step_time": 8.104437351226807
    },
    {
      "epoch": 4.883670806884766e-05,
      "model_forward_time": 1.70328950881958,
      "step": 51209
    },
    {
      "epoch": 4.883670806884766e-05,
      "step": 51209,
      "training_step_time": 7.759664058685303
    },
    {
      "epoch": 4.883766174316406e-05,
      "grad_norm": 0.10683228075504303,
      "learning_rate": 5.213398114403187e-06,
      "loss": 0.0525,
      "step": 51210
    },
    {
      "epoch": 4.883766174316406e-05,
      "model_forward_time": 1.7055933475494385,
      "step": 51210
    },
    {
      "dynamics_loss_avg": 0.059195257164537905,
      "epoch": 4.883766174316406e-05,
      "step": 51210
    },
    {
      "action_loss_avg": 0.0012922313588205725,
      "epoch": 4.883766174316406e-05,
      "step": 51210
    },
    {
      "epoch": 4.883766174316406e-05,
      "step": 51210,
      "training_step_time": 7.8385655879974365
    },
    {
      "epoch": 4.883861541748047e-05,
      "model_forward_time": 1.7132418155670166,
      "step": 51211
    },
    {
      "epoch": 4.883861541748047e-05,
      "step": 51211,
      "training_step_time": 6.96763801574707
    },
    {
      "epoch": 4.883956909179688e-05,
      "model_forward_time": 1.7043468952178955,
      "step": 51212
    },
    {
      "epoch": 4.883956909179688e-05,
      "step": 51212,
      "training_step_time": 7.027822971343994
    },
    {
      "epoch": 4.884052276611328e-05,
      "model_forward_time": 1.7114591598510742,
      "step": 51213
    },
    {
      "epoch": 4.884052276611328e-05,
      "step": 51213,
      "training_step_time": 7.566051006317139
    },
    {
      "epoch": 4.8841476440429686e-05,
      "model_forward_time": 1.708362102508545,
      "step": 51214
    },
    {
      "epoch": 4.8841476440429686e-05,
      "step": 51214,
      "training_step_time": 8.078177690505981
    },
    {
      "epoch": 4.88424301147461e-05,
      "model_forward_time": 1.7491505146026611,
      "step": 51215
    },
    {
      "epoch": 4.88424301147461e-05,
      "step": 51215,
      "training_step_time": 8.993944644927979
    },
    {
      "epoch": 4.88433837890625e-05,
      "model_forward_time": 2.256504774093628,
      "step": 51216
    },
    {
      "epoch": 4.88433837890625e-05,
      "step": 51216,
      "training_step_time": 8.701700210571289
    },
    {
      "epoch": 4.8844337463378905e-05,
      "model_forward_time": 1.709460973739624,
      "step": 51217
    },
    {
      "epoch": 4.8844337463378905e-05,
      "step": 51217,
      "training_step_time": 8.116451025009155
    },
    {
      "epoch": 4.8845291137695316e-05,
      "model_forward_time": 1.7121164798736572,
      "step": 51218
    },
    {
      "epoch": 4.8845291137695316e-05,
      "step": 51218,
      "training_step_time": 7.833939790725708
    },
    {
      "epoch": 4.884624481201172e-05,
      "model_forward_time": 1.7071397304534912,
      "step": 51219
    },
    {
      "epoch": 4.884624481201172e-05,
      "step": 51219,
      "training_step_time": 8.022563219070435
    },
    {
      "epoch": 4.8847198486328124e-05,
      "grad_norm": 0.09337859600782394,
      "learning_rate": 5.2117461395727275e-06,
      "loss": 0.0548,
      "step": 51220
    },
    {
      "epoch": 4.8847198486328124e-05,
      "model_forward_time": 1.7072441577911377,
      "step": 51220
    },
    {
      "dynamics_loss_avg": 0.0661160258576274,
      "epoch": 4.8847198486328124e-05,
      "step": 51220
    },
    {
      "action_loss_avg": 0.005732868635095656,
      "epoch": 4.8847198486328124e-05,
      "step": 51220
    },
    {
      "epoch": 4.8847198486328124e-05,
      "step": 51220,
      "training_step_time": 7.0194127559661865
    },
    {
      "epoch": 4.884815216064453e-05,
      "model_forward_time": 1.7464635372161865,
      "step": 51221
    },
    {
      "epoch": 4.884815216064453e-05,
      "step": 51221,
      "training_step_time": 7.096221446990967
    },
    {
      "epoch": 4.884910583496094e-05,
      "model_forward_time": 1.73756742477417,
      "step": 51222
    },
    {
      "epoch": 4.884910583496094e-05,
      "step": 51222,
      "training_step_time": 7.577021837234497
    },
    {
      "epoch": 4.885005950927734e-05,
      "model_forward_time": 1.7333612442016602,
      "step": 51223
    },
    {
      "epoch": 4.885005950927734e-05,
      "step": 51223,
      "training_step_time": 7.960130453109741
    },
    {
      "epoch": 4.885101318359375e-05,
      "model_forward_time": 1.775507926940918,
      "step": 51224
    },
    {
      "epoch": 4.885101318359375e-05,
      "step": 51224,
      "training_step_time": 10.264235734939575
    },
    {
      "epoch": 4.885196685791016e-05,
      "model_forward_time": 2.327812433242798,
      "step": 51225
    },
    {
      "epoch": 4.885196685791016e-05,
      "step": 51225,
      "training_step_time": 8.916160821914673
    },
    {
      "epoch": 4.885292053222656e-05,
      "model_forward_time": 1.7337253093719482,
      "step": 51226
    },
    {
      "epoch": 4.885292053222656e-05,
      "step": 51226,
      "training_step_time": 7.967101335525513
    },
    {
      "epoch": 4.8853874206542966e-05,
      "model_forward_time": 1.7315869331359863,
      "step": 51227
    },
    {
      "epoch": 4.8853874206542966e-05,
      "step": 51227,
      "training_step_time": 8.577890157699585
    },
    {
      "epoch": 4.8854827880859376e-05,
      "model_forward_time": 1.7437782287597656,
      "step": 51228
    },
    {
      "epoch": 4.8854827880859376e-05,
      "step": 51228,
      "training_step_time": 8.015498638153076
    },
    {
      "epoch": 4.885578155517578e-05,
      "model_forward_time": 1.7105722427368164,
      "step": 51229
    },
    {
      "epoch": 4.885578155517578e-05,
      "step": 51229,
      "training_step_time": 7.085332870483398
    },
    {
      "epoch": 4.8856735229492185e-05,
      "grad_norm": 0.15483014285564423,
      "learning_rate": 5.210094141586026e-06,
      "loss": 0.0537,
      "step": 51230
    },
    {
      "epoch": 4.8856735229492185e-05,
      "model_forward_time": 1.7177226543426514,
      "step": 51230
    },
    {
      "dynamics_loss_avg": 0.06367627382278443,
      "epoch": 4.8856735229492185e-05,
      "step": 51230
    },
    {
      "action_loss_avg": 0.002742917137220502,
      "epoch": 4.8856735229492185e-05,
      "step": 51230
    },
    {
      "epoch": 4.8856735229492185e-05,
      "step": 51230,
      "training_step_time": 7.049286603927612
    },
    {
      "epoch": 4.8857688903808595e-05,
      "model_forward_time": 1.729074478149414,
      "step": 51231
    },
    {
      "epoch": 4.8857688903808595e-05,
      "step": 51231,
      "training_step_time": 7.659687280654907
    },
    {
      "epoch": 4.8858642578125e-05,
      "model_forward_time": 1.7381012439727783,
      "step": 51232
    },
    {
      "epoch": 4.8858642578125e-05,
      "step": 51232,
      "training_step_time": 8.161170959472656
    },
    {
      "epoch": 4.88595962524414e-05,
      "model_forward_time": 1.7674894332885742,
      "step": 51233
    },
    {
      "epoch": 4.88595962524414e-05,
      "step": 51233,
      "training_step_time": 8.085334539413452
    },
    {
      "epoch": 4.8860549926757814e-05,
      "model_forward_time": 2.300246238708496,
      "step": 51234
    },
    {
      "epoch": 4.8860549926757814e-05,
      "step": 51234,
      "training_step_time": 8.903224468231201
    },
    {
      "epoch": 4.886150360107422e-05,
      "model_forward_time": 1.7278265953063965,
      "step": 51235
    },
    {
      "epoch": 4.886150360107422e-05,
      "step": 51235,
      "training_step_time": 8.091249942779541
    },
    {
      "epoch": 4.886245727539062e-05,
      "model_forward_time": 1.7319583892822266,
      "step": 51236
    },
    {
      "epoch": 4.886245727539062e-05,
      "step": 51236,
      "training_step_time": 8.9767165184021
    },
    {
      "epoch": 4.886341094970703e-05,
      "model_forward_time": 1.737760066986084,
      "step": 51237
    },
    {
      "epoch": 4.886341094970703e-05,
      "step": 51237,
      "training_step_time": 7.951913595199585
    },
    {
      "epoch": 4.886436462402344e-05,
      "model_forward_time": 1.7320172786712646,
      "step": 51238
    },
    {
      "epoch": 4.886436462402344e-05,
      "step": 51238,
      "training_step_time": 7.039957284927368
    },
    {
      "epoch": 4.886531829833984e-05,
      "model_forward_time": 1.7050862312316895,
      "step": 51239
    },
    {
      "epoch": 4.886531829833984e-05,
      "step": 51239,
      "training_step_time": 6.870062828063965
    },
    {
      "epoch": 4.886627197265625e-05,
      "grad_norm": 0.09877566993236542,
      "learning_rate": 5.208442120623748e-06,
      "loss": 0.0522,
      "step": 51240
    },
    {
      "epoch": 4.886627197265625e-05,
      "model_forward_time": 1.702024221420288,
      "step": 51240
    },
    {
      "dynamics_loss_avg": 0.046803544834256174,
      "epoch": 4.886627197265625e-05,
      "step": 51240
    },
    {
      "action_loss_avg": 0.005159740673843771,
      "epoch": 4.886627197265625e-05,
      "step": 51240
    },
    {
      "epoch": 4.886627197265625e-05,
      "step": 51240,
      "training_step_time": 7.783217430114746
    },
    {
      "epoch": 4.8867225646972656e-05,
      "model_forward_time": 1.7084147930145264,
      "step": 51241
    },
    {
      "epoch": 4.8867225646972656e-05,
      "step": 51241,
      "training_step_time": 9.037927389144897
    },
    {
      "epoch": 4.886817932128906e-05,
      "model_forward_time": 1.7544081211090088,
      "step": 51242
    },
    {
      "epoch": 4.886817932128906e-05,
      "step": 51242,
      "training_step_time": 9.000079154968262
    },
    {
      "epoch": 4.886913299560547e-05,
      "model_forward_time": 1.7062299251556396,
      "step": 51243
    },
    {
      "epoch": 4.886913299560547e-05,
      "step": 51243,
      "training_step_time": 10.320208311080933
    },
    {
      "epoch": 4.8870086669921875e-05,
      "model_forward_time": 2.255615711212158,
      "step": 51244
    },
    {
      "epoch": 4.8870086669921875e-05,
      "step": 51244,
      "training_step_time": 9.2531099319458
    },
    {
      "epoch": 4.887104034423828e-05,
      "model_forward_time": 1.712864637374878,
      "step": 51245
    },
    {
      "epoch": 4.887104034423828e-05,
      "step": 51245,
      "training_step_time": 8.083699703216553
    },
    {
      "epoch": 4.887199401855469e-05,
      "model_forward_time": 1.705359935760498,
      "step": 51246
    },
    {
      "epoch": 4.887199401855469e-05,
      "step": 51246,
      "training_step_time": 8.015808343887329
    },
    {
      "epoch": 4.8872947692871094e-05,
      "model_forward_time": 1.699706792831421,
      "step": 51247
    },
    {
      "epoch": 4.8872947692871094e-05,
      "step": 51247,
      "training_step_time": 7.116016626358032
    },
    {
      "epoch": 4.88739013671875e-05,
      "model_forward_time": 1.7123136520385742,
      "step": 51248
    },
    {
      "epoch": 4.88739013671875e-05,
      "step": 51248,
      "training_step_time": 7.098027467727661
    },
    {
      "epoch": 4.887485504150391e-05,
      "model_forward_time": 1.7066192626953125,
      "step": 51249
    },
    {
      "epoch": 4.887485504150391e-05,
      "step": 51249,
      "training_step_time": 7.64655327796936
    },
    {
      "epoch": 4.887580871582031e-05,
      "grad_norm": 0.13842982053756714,
      "learning_rate": 5.20679007686655e-06,
      "loss": 0.0543,
      "step": 51250
    },
    {
      "epoch": 4.887580871582031e-05,
      "model_forward_time": 1.7100696563720703,
      "step": 51250
    },
    {
      "dynamics_loss_avg": 0.055142666958272454,
      "epoch": 4.887580871582031e-05,
      "step": 51250
    },
    {
      "action_loss_avg": 0.0013062382116913796,
      "epoch": 4.887580871582031e-05,
      "step": 51250
    },
    {
      "epoch": 4.887580871582031e-05,
      "step": 51250,
      "training_step_time": 7.645820379257202
    },
    {
      "epoch": 4.8876762390136717e-05,
      "model_forward_time": 1.7530224323272705,
      "step": 51251
    },
    {
      "epoch": 4.8876762390136717e-05,
      "step": 51251,
      "training_step_time": 7.910966634750366
    },
    {
      "epoch": 4.887771606445313e-05,
      "model_forward_time": 1.7099976539611816,
      "step": 51252
    },
    {
      "epoch": 4.887771606445313e-05,
      "step": 51252,
      "training_step_time": 10.2868812084198
    },
    {
      "epoch": 4.887866973876953e-05,
      "model_forward_time": 2.2624974250793457,
      "step": 51253
    },
    {
      "epoch": 4.887866973876953e-05,
      "step": 51253,
      "training_step_time": 8.591281652450562
    },
    {
      "epoch": 4.8879623413085935e-05,
      "model_forward_time": 1.7086968421936035,
      "step": 51254
    },
    {
      "epoch": 4.8879623413085935e-05,
      "step": 51254,
      "training_step_time": 9.132451295852661
    },
    {
      "epoch": 4.8880577087402346e-05,
      "model_forward_time": 1.7081208229064941,
      "step": 51255
    },
    {
      "epoch": 4.8880577087402346e-05,
      "step": 51255,
      "training_step_time": 8.87822151184082
    },
    {
      "epoch": 4.888153076171875e-05,
      "model_forward_time": 1.7086827754974365,
      "step": 51256
    },
    {
      "epoch": 4.888153076171875e-05,
      "step": 51256,
      "training_step_time": 7.56281590461731
    },
    {
      "epoch": 4.8882484436035154e-05,
      "model_forward_time": 1.7047491073608398,
      "step": 51257
    },
    {
      "epoch": 4.8882484436035154e-05,
      "step": 51257,
      "training_step_time": 7.034374952316284
    },
    {
      "epoch": 4.8883438110351565e-05,
      "model_forward_time": 1.7088685035705566,
      "step": 51258
    },
    {
      "epoch": 4.8883438110351565e-05,
      "step": 51258,
      "training_step_time": 7.52253794670105
    },
    {
      "epoch": 4.888439178466797e-05,
      "model_forward_time": 1.3335611820220947,
      "step": 51259
    },
    {
      "epoch": 4.888439178466797e-05,
      "step": 51259,
      "training_step_time": 7.531923055648804
    },
    {
      "epoch": 4.888534545898437e-05,
      "grad_norm": 0.08659973740577698,
      "learning_rate": 5.205138010495102e-06,
      "loss": 0.0537,
      "step": 51260
    },
    {
      "epoch": 4.888534545898437e-05,
      "model_forward_time": 1.72979736328125,
      "step": 51260
    },
    {
      "dynamics_loss_avg": 0.054355927463620904,
      "epoch": 4.888534545898437e-05,
      "step": 51260
    },
    {
      "action_loss_avg": 0.004210190154844895,
      "epoch": 4.888534545898437e-05,
      "step": 51260
    },
    {
      "epoch": 4.888534545898437e-05,
      "step": 51260,
      "training_step_time": 7.993937730789185
    },
    {
      "epoch": 4.8886299133300784e-05,
      "model_forward_time": 1.710557460784912,
      "step": 51261
    },
    {
      "epoch": 4.8886299133300784e-05,
      "step": 51261,
      "training_step_time": 8.6641685962677
    },
    {
      "epoch": 4.888725280761719e-05,
      "model_forward_time": 2.2728700637817383,
      "step": 51262
    },
    {
      "epoch": 4.888725280761719e-05,
      "step": 51262,
      "training_step_time": 8.983372449874878
    },
    {
      "epoch": 4.888820648193359e-05,
      "model_forward_time": 1.7080979347229004,
      "step": 51263
    },
    {
      "epoch": 4.888820648193359e-05,
      "step": 51263,
      "training_step_time": 8.06099820137024
    },
    {
      "epoch": 4.888916015625e-05,
      "model_forward_time": 1.7044174671173096,
      "step": 51264
    },
    {
      "epoch": 4.888916015625e-05,
      "step": 51264,
      "training_step_time": 7.677307367324829
    },
    {
      "epoch": 4.889011383056641e-05,
      "model_forward_time": 1.7122831344604492,
      "step": 51265
    },
    {
      "epoch": 4.889011383056641e-05,
      "step": 51265,
      "training_step_time": 7.948964595794678
    },
    {
      "epoch": 4.889106750488281e-05,
      "model_forward_time": 1.7069368362426758,
      "step": 51266
    },
    {
      "epoch": 4.889106750488281e-05,
      "step": 51266,
      "training_step_time": 7.057873964309692
    },
    {
      "epoch": 4.889202117919922e-05,
      "model_forward_time": 1.7119414806365967,
      "step": 51267
    },
    {
      "epoch": 4.889202117919922e-05,
      "step": 51267,
      "training_step_time": 7.5772624015808105
    },
    {
      "epoch": 4.8892974853515626e-05,
      "model_forward_time": 1.7055389881134033,
      "step": 51268
    },
    {
      "epoch": 4.8892974853515626e-05,
      "step": 51268,
      "training_step_time": 7.642775058746338
    },
    {
      "epoch": 4.889392852783203e-05,
      "model_forward_time": 1.7127103805541992,
      "step": 51269
    },
    {
      "epoch": 4.889392852783203e-05,
      "step": 51269,
      "training_step_time": 8.948052883148193
    },
    {
      "epoch": 4.889488220214844e-05,
      "grad_norm": 0.11643250286579132,
      "learning_rate": 5.2034859216900704e-06,
      "loss": 0.0525,
      "step": 51270
    },
    {
      "epoch": 4.889488220214844e-05,
      "model_forward_time": 1.7046599388122559,
      "step": 51270
    },
    {
      "dynamics_loss_avg": 0.04489622469991446,
      "epoch": 4.889488220214844e-05,
      "step": 51270
    },
    {
      "action_loss_avg": 0.0026976090390235185,
      "epoch": 4.889488220214844e-05,
      "step": 51270
    },
    {
      "epoch": 4.889488220214844e-05,
      "step": 51270,
      "training_step_time": 9.170003414154053
    },
    {
      "epoch": 4.8895835876464845e-05,
      "model_forward_time": 2.2800848484039307,
      "step": 51271
    },
    {
      "epoch": 4.8895835876464845e-05,
      "step": 51271,
      "training_step_time": 9.843185663223267
    },
    {
      "epoch": 4.889678955078125e-05,
      "model_forward_time": 1.7104966640472412,
      "step": 51272
    },
    {
      "epoch": 4.889678955078125e-05,
      "step": 51272,
      "training_step_time": 8.140662670135498
    },
    {
      "epoch": 4.889774322509766e-05,
      "model_forward_time": 1.6995141506195068,
      "step": 51273
    },
    {
      "epoch": 4.889774322509766e-05,
      "step": 51273,
      "training_step_time": 7.774951696395874
    },
    {
      "epoch": 4.889869689941406e-05,
      "model_forward_time": 1.7065000534057617,
      "step": 51274
    },
    {
      "epoch": 4.889869689941406e-05,
      "step": 51274,
      "training_step_time": 8.143691778182983
    },
    {
      "epoch": 4.889965057373047e-05,
      "model_forward_time": 1.7065105438232422,
      "step": 51275
    },
    {
      "epoch": 4.889965057373047e-05,
      "step": 51275,
      "training_step_time": 7.00614857673645
    },
    {
      "epoch": 4.890060424804688e-05,
      "model_forward_time": 1.711893081665039,
      "step": 51276
    },
    {
      "epoch": 4.890060424804688e-05,
      "step": 51276,
      "training_step_time": 7.567967891693115
    },
    {
      "epoch": 4.890155792236328e-05,
      "model_forward_time": 1.712522029876709,
      "step": 51277
    },
    {
      "epoch": 4.890155792236328e-05,
      "step": 51277,
      "training_step_time": 7.703310012817383
    },
    {
      "epoch": 4.8902511596679686e-05,
      "model_forward_time": 1.713982105255127,
      "step": 51278
    },
    {
      "epoch": 4.8902511596679686e-05,
      "step": 51278,
      "training_step_time": 7.95650839805603
    },
    {
      "epoch": 4.89034652709961e-05,
      "model_forward_time": 1.751988410949707,
      "step": 51279
    },
    {
      "epoch": 4.89034652709961e-05,
      "step": 51279,
      "training_step_time": 9.243531227111816
    },
    {
      "epoch": 4.89044189453125e-05,
      "grad_norm": 0.10023128986358643,
      "learning_rate": 5.201833810632123e-06,
      "loss": 0.057,
      "step": 51280
    },
    {
      "epoch": 4.89044189453125e-05,
      "model_forward_time": 1.7133371829986572,
      "step": 51280
    },
    {
      "dynamics_loss_avg": 0.05310789067298174,
      "epoch": 4.89044189453125e-05,
      "step": 51280
    },
    {
      "action_loss_avg": 0.0018177577876485884,
      "epoch": 4.89044189453125e-05,
      "step": 51280
    },
    {
      "epoch": 4.89044189453125e-05,
      "step": 51280,
      "training_step_time": 9.067682027816772
    },
    {
      "epoch": 4.8905372619628905e-05,
      "model_forward_time": 1.7050626277923584,
      "step": 51281
    },
    {
      "epoch": 4.8905372619628905e-05,
      "step": 51281,
      "training_step_time": 8.77552580833435
    },
    {
      "epoch": 4.890632629394531e-05,
      "model_forward_time": 1.7134475708007812,
      "step": 51282
    },
    {
      "epoch": 4.890632629394531e-05,
      "step": 51282,
      "training_step_time": 7.981398105621338
    },
    {
      "epoch": 4.890727996826172e-05,
      "model_forward_time": 1.7139735221862793,
      "step": 51283
    },
    {
      "epoch": 4.890727996826172e-05,
      "step": 51283,
      "training_step_time": 7.938862323760986
    },
    {
      "epoch": 4.8908233642578124e-05,
      "model_forward_time": 1.7026333808898926,
      "step": 51284
    },
    {
      "epoch": 4.8908233642578124e-05,
      "step": 51284,
      "training_step_time": 7.586471796035767
    },
    {
      "epoch": 4.890918731689453e-05,
      "model_forward_time": 1.7079627513885498,
      "step": 51285
    },
    {
      "epoch": 4.890918731689453e-05,
      "step": 51285,
      "training_step_time": 7.604300260543823
    },
    {
      "epoch": 4.891014099121094e-05,
      "model_forward_time": 1.700730800628662,
      "step": 51286
    },
    {
      "epoch": 4.891014099121094e-05,
      "step": 51286,
      "training_step_time": 7.614021062850952
    },
    {
      "epoch": 4.891109466552734e-05,
      "model_forward_time": 1.7074825763702393,
      "step": 51287
    },
    {
      "epoch": 4.891109466552734e-05,
      "step": 51287,
      "training_step_time": 7.97890305519104
    },
    {
      "epoch": 4.891204833984375e-05,
      "model_forward_time": 1.7518012523651123,
      "step": 51288
    },
    {
      "epoch": 4.891204833984375e-05,
      "step": 51288,
      "training_step_time": 8.80660080909729
    },
    {
      "epoch": 4.891300201416016e-05,
      "model_forward_time": 1.7046091556549072,
      "step": 51289
    },
    {
      "epoch": 4.891300201416016e-05,
      "step": 51289,
      "training_step_time": 9.050542831420898
    },
    {
      "epoch": 4.891395568847656e-05,
      "grad_norm": 0.10668645799160004,
      "learning_rate": 5.200181677501933e-06,
      "loss": 0.0566,
      "step": 51290
    },
    {
      "epoch": 4.891395568847656e-05,
      "model_forward_time": 2.259950637817383,
      "step": 51290
    },
    {
      "dynamics_loss_avg": 0.05305348318070173,
      "epoch": 4.891395568847656e-05,
      "step": 51290
    },
    {
      "action_loss_avg": 0.002419552730862051,
      "epoch": 4.891395568847656e-05,
      "step": 51290
    },
    {
      "epoch": 4.891395568847656e-05,
      "step": 51290,
      "training_step_time": 7.966439723968506
    },
    {
      "epoch": 4.8914909362792966e-05,
      "model_forward_time": 1.7026209831237793,
      "step": 51291
    },
    {
      "epoch": 4.8914909362792966e-05,
      "step": 51291,
      "training_step_time": 8.037271499633789
    },
    {
      "epoch": 4.8915863037109377e-05,
      "model_forward_time": 1.708477258682251,
      "step": 51292
    },
    {
      "epoch": 4.8915863037109377e-05,
      "step": 51292,
      "training_step_time": 7.9319093227386475
    },
    {
      "epoch": 4.891681671142578e-05,
      "model_forward_time": 1.707033634185791,
      "step": 51293
    },
    {
      "epoch": 4.891681671142578e-05,
      "step": 51293,
      "training_step_time": 7.575807809829712
    },
    {
      "epoch": 4.8917770385742185e-05,
      "model_forward_time": 1.704021692276001,
      "step": 51294
    },
    {
      "epoch": 4.8917770385742185e-05,
      "step": 51294,
      "training_step_time": 7.679647445678711
    },
    {
      "epoch": 4.8918724060058595e-05,
      "model_forward_time": 1.7080471515655518,
      "step": 51295
    },
    {
      "epoch": 4.8918724060058595e-05,
      "step": 51295,
      "training_step_time": 7.747009754180908
    },
    {
      "epoch": 4.8919677734375e-05,
      "model_forward_time": 1.7027041912078857,
      "step": 51296
    },
    {
      "epoch": 4.8919677734375e-05,
      "step": 51296,
      "training_step_time": 7.951502561569214
    },
    {
      "epoch": 4.8920631408691403e-05,
      "model_forward_time": 1.7556579113006592,
      "step": 51297
    },
    {
      "epoch": 4.8920631408691403e-05,
      "step": 51297,
      "training_step_time": 7.999807834625244
    },
    {
      "epoch": 4.8921585083007814e-05,
      "model_forward_time": 1.7117729187011719,
      "step": 51298
    },
    {
      "epoch": 4.8921585083007814e-05,
      "step": 51298,
      "training_step_time": 9.548597574234009
    },
    {
      "epoch": 4.892253875732422e-05,
      "model_forward_time": 2.2714648246765137,
      "step": 51299
    },
    {
      "epoch": 4.892253875732422e-05,
      "step": 51299,
      "training_step_time": 8.666640996932983
    },
    {
      "epoch": 4.892349243164062e-05,
      "grad_norm": 0.10421624779701233,
      "learning_rate": 5.198529522480178e-06,
      "loss": 0.0581,
      "step": 51300
    },
    {
      "epoch": 4.892349243164062e-05,
      "model_forward_time": 1.7074871063232422,
      "step": 51300
    },
    {
      "dynamics_loss_avg": 0.061229277215898034,
      "epoch": 4.892349243164062e-05,
      "step": 51300
    },
    {
      "action_loss_avg": 0.0026897124596871437,
      "epoch": 4.892349243164062e-05,
      "step": 51300
    },
    {
      "epoch": 4.892349243164062e-05,
      "step": 51300,
      "training_step_time": 9.356066703796387
    },
    {
      "epoch": 4.892444610595703e-05,
      "model_forward_time": 1.7013957500457764,
      "step": 51301
    },
    {
      "epoch": 4.892444610595703e-05,
      "step": 51301,
      "training_step_time": 7.78522515296936
    },
    {
      "epoch": 4.892539978027344e-05,
      "model_forward_time": 1.711130142211914,
      "step": 51302
    },
    {
      "epoch": 4.892539978027344e-05,
      "step": 51302,
      "training_step_time": 7.4746928215026855
    },
    {
      "epoch": 4.892635345458984e-05,
      "model_forward_time": 1.7072739601135254,
      "step": 51303
    },
    {
      "epoch": 4.892635345458984e-05,
      "step": 51303,
      "training_step_time": 7.5950539112091064
    },
    {
      "epoch": 4.892730712890625e-05,
      "model_forward_time": 1.7037253379821777,
      "step": 51304
    },
    {
      "epoch": 4.892730712890625e-05,
      "step": 51304,
      "training_step_time": 7.682158708572388
    },
    {
      "epoch": 4.8928260803222656e-05,
      "model_forward_time": 1.7142741680145264,
      "step": 51305
    },
    {
      "epoch": 4.8928260803222656e-05,
      "step": 51305,
      "training_step_time": 7.944332122802734
    },
    {
      "epoch": 4.892921447753906e-05,
      "model_forward_time": 1.7461669445037842,
      "step": 51306
    },
    {
      "epoch": 4.892921447753906e-05,
      "step": 51306,
      "training_step_time": 7.965614080429077
    },
    {
      "epoch": 4.893016815185547e-05,
      "model_forward_time": 1.7029359340667725,
      "step": 51307
    },
    {
      "epoch": 4.893016815185547e-05,
      "step": 51307,
      "training_step_time": 8.090563774108887
    },
    {
      "epoch": 4.8931121826171875e-05,
      "model_forward_time": 2.2766237258911133,
      "step": 51308
    },
    {
      "epoch": 4.8931121826171875e-05,
      "step": 51308,
      "training_step_time": 8.558014154434204
    },
    {
      "epoch": 4.893207550048828e-05,
      "model_forward_time": 1.7075023651123047,
      "step": 51309
    },
    {
      "epoch": 4.893207550048828e-05,
      "step": 51309,
      "training_step_time": 8.09037733078003
    },
    {
      "epoch": 4.893302917480469e-05,
      "grad_norm": 0.12341850996017456,
      "learning_rate": 5.196877345747531e-06,
      "loss": 0.0553,
      "step": 51310
    },
    {
      "epoch": 4.893302917480469e-05,
      "model_forward_time": 1.7191333770751953,
      "step": 51310
    },
    {
      "dynamics_loss_avg": 0.057339420355856416,
      "epoch": 4.893302917480469e-05,
      "step": 51310
    },
    {
      "action_loss_avg": 0.0024417439417447895,
      "epoch": 4.893302917480469e-05,
      "step": 51310
    },
    {
      "epoch": 4.893302917480469e-05,
      "step": 51310,
      "training_step_time": 7.756379127502441
    },
    {
      "epoch": 4.8933982849121094e-05,
      "model_forward_time": 1.7114577293395996,
      "step": 51311
    },
    {
      "epoch": 4.8933982849121094e-05,
      "step": 51311,
      "training_step_time": 7.557158946990967
    },
    {
      "epoch": 4.89349365234375e-05,
      "model_forward_time": 1.707364797592163,
      "step": 51312
    },
    {
      "epoch": 4.89349365234375e-05,
      "step": 51312,
      "training_step_time": 7.551692485809326
    },
    {
      "epoch": 4.893589019775391e-05,
      "model_forward_time": 1.714099407196045,
      "step": 51313
    },
    {
      "epoch": 4.893589019775391e-05,
      "step": 51313,
      "training_step_time": 7.703928709030151
    },
    {
      "epoch": 4.893684387207031e-05,
      "model_forward_time": 1.7089018821716309,
      "step": 51314
    },
    {
      "epoch": 4.893684387207031e-05,
      "step": 51314,
      "training_step_time": 8.067237138748169
    },
    {
      "epoch": 4.8937797546386717e-05,
      "model_forward_time": 1.7212131023406982,
      "step": 51315
    },
    {
      "epoch": 4.8937797546386717e-05,
      "step": 51315,
      "training_step_time": 7.880279779434204
    },
    {
      "epoch": 4.893875122070313e-05,
      "model_forward_time": 1.7124733924865723,
      "step": 51316
    },
    {
      "epoch": 4.893875122070313e-05,
      "step": 51316,
      "training_step_time": 9.048900604248047
    },
    {
      "epoch": 4.893970489501953e-05,
      "model_forward_time": 2.2751662731170654,
      "step": 51317
    },
    {
      "epoch": 4.893970489501953e-05,
      "step": 51317,
      "training_step_time": 7.989538192749023
    },
    {
      "epoch": 4.8940658569335935e-05,
      "model_forward_time": 1.712716817855835,
      "step": 51318
    },
    {
      "epoch": 4.8940658569335935e-05,
      "step": 51318,
      "training_step_time": 8.9633629322052
    },
    {
      "epoch": 4.8941612243652346e-05,
      "model_forward_time": 1.3337748050689697,
      "step": 51319
    },
    {
      "epoch": 4.8941612243652346e-05,
      "step": 51319,
      "training_step_time": 7.947982549667358
    },
    {
      "epoch": 4.894256591796875e-05,
      "grad_norm": 0.10968185216188431,
      "learning_rate": 5.195225147484674e-06,
      "loss": 0.0593,
      "step": 51320
    },
    {
      "epoch": 4.894256591796875e-05,
      "model_forward_time": 1.7130613327026367,
      "step": 51320
    },
    {
      "dynamics_loss_avg": 0.08083447776734828,
      "epoch": 4.894256591796875e-05,
      "step": 51320
    },
    {
      "action_loss_avg": 0.0031404883484356105,
      "epoch": 4.894256591796875e-05,
      "step": 51320
    },
    {
      "epoch": 4.894256591796875e-05,
      "step": 51320,
      "training_step_time": 7.8490660190582275
    },
    {
      "epoch": 4.8943519592285154e-05,
      "model_forward_time": 1.7052762508392334,
      "step": 51321
    },
    {
      "epoch": 4.8943519592285154e-05,
      "step": 51321,
      "training_step_time": 7.637165546417236
    },
    {
      "epoch": 4.8944473266601565e-05,
      "model_forward_time": 1.7074429988861084,
      "step": 51322
    },
    {
      "epoch": 4.8944473266601565e-05,
      "step": 51322,
      "training_step_time": 7.569074630737305
    },
    {
      "epoch": 4.894542694091797e-05,
      "model_forward_time": 1.7059879302978516,
      "step": 51323
    },
    {
      "epoch": 4.894542694091797e-05,
      "step": 51323,
      "training_step_time": 7.612775087356567
    },
    {
      "epoch": 4.894638061523437e-05,
      "model_forward_time": 1.7018427848815918,
      "step": 51324
    },
    {
      "epoch": 4.894638061523437e-05,
      "step": 51324,
      "training_step_time": 8.904665470123291
    },
    {
      "epoch": 4.8947334289550784e-05,
      "model_forward_time": 1.7130708694458008,
      "step": 51325
    },
    {
      "epoch": 4.8947334289550784e-05,
      "step": 51325,
      "training_step_time": 9.242026329040527
    },
    {
      "epoch": 4.894828796386719e-05,
      "model_forward_time": 1.7093956470489502,
      "step": 51326
    },
    {
      "epoch": 4.894828796386719e-05,
      "step": 51326,
      "training_step_time": 8.474039554595947
    },
    {
      "epoch": 4.894924163818359e-05,
      "model_forward_time": 2.234067678451538,
      "step": 51327
    },
    {
      "epoch": 4.894924163818359e-05,
      "step": 51327,
      "training_step_time": 8.924829006195068
    },
    {
      "epoch": 4.89501953125e-05,
      "model_forward_time": 1.706590175628662,
      "step": 51328
    },
    {
      "epoch": 4.89501953125e-05,
      "step": 51328,
      "training_step_time": 8.079074621200562
    },
    {
      "epoch": 4.895114898681641e-05,
      "model_forward_time": 1.7116987705230713,
      "step": 51329
    },
    {
      "epoch": 4.895114898681641e-05,
      "step": 51329,
      "training_step_time": 7.725861549377441
    },
    {
      "epoch": 4.895210266113281e-05,
      "grad_norm": 0.12778843939304352,
      "learning_rate": 5.193572927872288e-06,
      "loss": 0.0567,
      "step": 51330
    },
    {
      "epoch": 4.895210266113281e-05,
      "model_forward_time": 1.7209692001342773,
      "step": 51330
    },
    {
      "dynamics_loss_avg": 0.06263637654483319,
      "epoch": 4.895210266113281e-05,
      "step": 51330
    },
    {
      "action_loss_avg": 0.0017683438025414943,
      "epoch": 4.895210266113281e-05,
      "step": 51330
    },
    {
      "epoch": 4.895210266113281e-05,
      "step": 51330,
      "training_step_time": 7.85423731803894
    },
    {
      "epoch": 4.895305633544922e-05,
      "model_forward_time": 1.7037928104400635,
      "step": 51331
    },
    {
      "epoch": 4.895305633544922e-05,
      "step": 51331,
      "training_step_time": 7.621150732040405
    },
    {
      "epoch": 4.8954010009765626e-05,
      "model_forward_time": 1.7089564800262451,
      "step": 51332
    },
    {
      "epoch": 4.8954010009765626e-05,
      "step": 51332,
      "training_step_time": 7.965266227722168
    },
    {
      "epoch": 4.895496368408203e-05,
      "model_forward_time": 1.7072234153747559,
      "step": 51333
    },
    {
      "epoch": 4.895496368408203e-05,
      "step": 51333,
      "training_step_time": 8.057997941970825
    },
    {
      "epoch": 4.895591735839844e-05,
      "model_forward_time": 1.7619318962097168,
      "step": 51334
    },
    {
      "epoch": 4.895591735839844e-05,
      "step": 51334,
      "training_step_time": 9.081703186035156
    },
    {
      "epoch": 4.8956871032714845e-05,
      "model_forward_time": 1.7364981174468994,
      "step": 51335
    },
    {
      "epoch": 4.8956871032714845e-05,
      "step": 51335,
      "training_step_time": 9.45905613899231
    },
    {
      "epoch": 4.895782470703125e-05,
      "model_forward_time": 2.2839555740356445,
      "step": 51336
    },
    {
      "epoch": 4.895782470703125e-05,
      "step": 51336,
      "training_step_time": 8.61572003364563
    },
    {
      "epoch": 4.895877838134766e-05,
      "model_forward_time": 1.7094182968139648,
      "step": 51337
    },
    {
      "epoch": 4.895877838134766e-05,
      "step": 51337,
      "training_step_time": 8.953271865844727
    },
    {
      "epoch": 4.8959732055664063e-05,
      "model_forward_time": 1.7108497619628906,
      "step": 51338
    },
    {
      "epoch": 4.8959732055664063e-05,
      "step": 51338,
      "training_step_time": 7.65490984916687
    },
    {
      "epoch": 4.896068572998047e-05,
      "model_forward_time": 1.7169640064239502,
      "step": 51339
    },
    {
      "epoch": 4.896068572998047e-05,
      "step": 51339,
      "training_step_time": 7.620145320892334
    },
    {
      "epoch": 4.896163940429688e-05,
      "grad_norm": 0.10744287818670273,
      "learning_rate": 5.191920687091059e-06,
      "loss": 0.0571,
      "step": 51340
    },
    {
      "epoch": 4.896163940429688e-05,
      "model_forward_time": 1.7082123756408691,
      "step": 51340
    },
    {
      "dynamics_loss_avg": 0.055073955841362474,
      "epoch": 4.896163940429688e-05,
      "step": 51340
    },
    {
      "action_loss_avg": 0.0021026048925705254,
      "epoch": 4.896163940429688e-05,
      "step": 51340
    },
    {
      "epoch": 4.896163940429688e-05,
      "step": 51340,
      "training_step_time": 7.827877521514893
    },
    {
      "epoch": 4.896259307861328e-05,
      "model_forward_time": 1.717787265777588,
      "step": 51341
    },
    {
      "epoch": 4.896259307861328e-05,
      "step": 51341,
      "training_step_time": 7.923288345336914
    },
    {
      "epoch": 4.8963546752929686e-05,
      "model_forward_time": 1.7246160507202148,
      "step": 51342
    },
    {
      "epoch": 4.8963546752929686e-05,
      "step": 51342,
      "training_step_time": 7.790812015533447
    },
    {
      "epoch": 4.89645004272461e-05,
      "model_forward_time": 1.732853651046753,
      "step": 51343
    },
    {
      "epoch": 4.89645004272461e-05,
      "step": 51343,
      "training_step_time": 7.989107608795166
    },
    {
      "epoch": 4.89654541015625e-05,
      "model_forward_time": 1.7305784225463867,
      "step": 51344
    },
    {
      "epoch": 4.89654541015625e-05,
      "step": 51344,
      "training_step_time": 9.135705947875977
    },
    {
      "epoch": 4.8966407775878905e-05,
      "model_forward_time": 2.297609567642212,
      "step": 51345
    },
    {
      "epoch": 4.8966407775878905e-05,
      "step": 51345,
      "training_step_time": 8.560076236724854
    },
    {
      "epoch": 4.896736145019531e-05,
      "model_forward_time": 1.7186129093170166,
      "step": 51346
    },
    {
      "epoch": 4.896736145019531e-05,
      "step": 51346,
      "training_step_time": 7.986583232879639
    },
    {
      "epoch": 4.896831512451172e-05,
      "model_forward_time": 1.7250995635986328,
      "step": 51347
    },
    {
      "epoch": 4.896831512451172e-05,
      "step": 51347,
      "training_step_time": 7.7393999099731445
    },
    {
      "epoch": 4.8969268798828124e-05,
      "model_forward_time": 1.7162928581237793,
      "step": 51348
    },
    {
      "epoch": 4.8969268798828124e-05,
      "step": 51348,
      "training_step_time": 7.803316831588745
    },
    {
      "epoch": 4.897022247314453e-05,
      "model_forward_time": 1.6819591522216797,
      "step": 51349
    },
    {
      "epoch": 4.897022247314453e-05,
      "step": 51349,
      "training_step_time": 7.82404637336731
    },
    {
      "epoch": 4.897117614746094e-05,
      "grad_norm": 0.11728941649198532,
      "learning_rate": 5.1902684253216695e-06,
      "loss": 0.0538,
      "step": 51350
    },
    {
      "epoch": 4.897117614746094e-05,
      "model_forward_time": 1.6914644241333008,
      "step": 51350
    },
    {
      "dynamics_loss_avg": 0.05154178496450186,
      "epoch": 4.897117614746094e-05,
      "step": 51350
    },
    {
      "action_loss_avg": 0.0017389316228218377,
      "epoch": 4.897117614746094e-05,
      "step": 51350
    },
    {
      "epoch": 4.897117614746094e-05,
      "step": 51350,
      "training_step_time": 8.926455974578857
    },
    {
      "epoch": 4.897212982177734e-05,
      "model_forward_time": 1.6882009506225586,
      "step": 51351
    },
    {
      "epoch": 4.897212982177734e-05,
      "step": 51351,
      "training_step_time": 7.79097843170166
    },
    {
      "epoch": 4.897308349609375e-05,
      "model_forward_time": 1.703138828277588,
      "step": 51352
    },
    {
      "epoch": 4.897308349609375e-05,
      "step": 51352,
      "training_step_time": 8.964105606079102
    },
    {
      "epoch": 4.897403717041016e-05,
      "model_forward_time": 1.6956017017364502,
      "step": 51353
    },
    {
      "epoch": 4.897403717041016e-05,
      "step": 51353,
      "training_step_time": 8.821356773376465
    },
    {
      "epoch": 4.897499084472656e-05,
      "model_forward_time": 2.2408485412597656,
      "step": 51354
    },
    {
      "epoch": 4.897499084472656e-05,
      "step": 51354,
      "training_step_time": 8.545367002487183
    },
    {
      "epoch": 4.8975944519042966e-05,
      "model_forward_time": 1.6826601028442383,
      "step": 51355
    },
    {
      "epoch": 4.8975944519042966e-05,
      "step": 51355,
      "training_step_time": 8.04169487953186
    },
    {
      "epoch": 4.8976898193359377e-05,
      "model_forward_time": 1.6918437480926514,
      "step": 51356
    },
    {
      "epoch": 4.8976898193359377e-05,
      "step": 51356,
      "training_step_time": 8.158109664916992
    },
    {
      "epoch": 4.897785186767578e-05,
      "model_forward_time": 1.6840097904205322,
      "step": 51357
    },
    {
      "epoch": 4.897785186767578e-05,
      "step": 51357,
      "training_step_time": 7.874027729034424
    },
    {
      "epoch": 4.8978805541992185e-05,
      "model_forward_time": 1.6827099323272705,
      "step": 51358
    },
    {
      "epoch": 4.8978805541992185e-05,
      "step": 51358,
      "training_step_time": 7.868872404098511
    },
    {
      "epoch": 4.8979759216308595e-05,
      "model_forward_time": 1.686194658279419,
      "step": 51359
    },
    {
      "epoch": 4.8979759216308595e-05,
      "step": 51359,
      "training_step_time": 7.992130994796753
    },
    {
      "epoch": 4.8980712890625e-05,
      "grad_norm": 0.1457979530096054,
      "learning_rate": 5.188616142744813e-06,
      "loss": 0.0579,
      "step": 51360
    },
    {
      "epoch": 4.8980712890625e-05,
      "model_forward_time": 1.6861188411712646,
      "step": 51360
    },
    {
      "dynamics_loss_avg": 0.08521685265004635,
      "epoch": 4.8980712890625e-05,
      "step": 51360
    },
    {
      "action_loss_avg": 0.0036994597467128187,
      "epoch": 4.8980712890625e-05,
      "step": 51360
    },
    {
      "epoch": 4.8980712890625e-05,
      "step": 51360,
      "training_step_time": 7.780997037887573
    },
    {
      "epoch": 4.8981666564941403e-05,
      "model_forward_time": 1.6893911361694336,
      "step": 51361
    },
    {
      "epoch": 4.8981666564941403e-05,
      "step": 51361,
      "training_step_time": 7.983990430831909
    },
    {
      "epoch": 4.8982620239257814e-05,
      "model_forward_time": 1.7046573162078857,
      "step": 51362
    },
    {
      "epoch": 4.8982620239257814e-05,
      "step": 51362,
      "training_step_time": 9.911601066589355
    },
    {
      "epoch": 4.898357391357422e-05,
      "model_forward_time": 2.2654576301574707,
      "step": 51363
    },
    {
      "epoch": 4.898357391357422e-05,
      "step": 51363,
      "training_step_time": 9.780494928359985
    },
    {
      "epoch": 4.898452758789062e-05,
      "model_forward_time": 1.6852943897247314,
      "step": 51364
    },
    {
      "epoch": 4.898452758789062e-05,
      "step": 51364,
      "training_step_time": 9.05713152885437
    },
    {
      "epoch": 4.898548126220703e-05,
      "model_forward_time": 1.6891279220581055,
      "step": 51365
    },
    {
      "epoch": 4.898548126220703e-05,
      "step": 51365,
      "training_step_time": 8.009470462799072
    },
    {
      "epoch": 4.898643493652344e-05,
      "model_forward_time": 1.690448522567749,
      "step": 51366
    },
    {
      "epoch": 4.898643493652344e-05,
      "step": 51366,
      "training_step_time": 7.731766700744629
    },
    {
      "epoch": 4.898738861083984e-05,
      "model_forward_time": 1.6895875930786133,
      "step": 51367
    },
    {
      "epoch": 4.898738861083984e-05,
      "step": 51367,
      "training_step_time": 7.811629295349121
    },
    {
      "epoch": 4.898834228515625e-05,
      "model_forward_time": 1.6736252307891846,
      "step": 51368
    },
    {
      "epoch": 4.898834228515625e-05,
      "step": 51368,
      "training_step_time": 7.984316825866699
    },
    {
      "epoch": 4.8989295959472656e-05,
      "model_forward_time": 1.684237003326416,
      "step": 51369
    },
    {
      "epoch": 4.8989295959472656e-05,
      "step": 51369,
      "training_step_time": 7.84437370300293
    },
    {
      "epoch": 4.899024963378906e-05,
      "grad_norm": 0.14019154012203217,
      "learning_rate": 5.186963839541179e-06,
      "loss": 0.0563,
      "step": 51370
    },
    {
      "epoch": 4.899024963378906e-05,
      "model_forward_time": 1.691161870956421,
      "step": 51370
    },
    {
      "dynamics_loss_avg": 0.07562551498413086,
      "epoch": 4.899024963378906e-05,
      "step": 51370
    },
    {
      "action_loss_avg": 0.005112314748112112,
      "epoch": 4.899024963378906e-05,
      "step": 51370
    },
    {
      "epoch": 4.899024963378906e-05,
      "step": 51370,
      "training_step_time": 7.5915138721466064
    },
    {
      "epoch": 4.899120330810547e-05,
      "model_forward_time": 1.6978673934936523,
      "step": 51371
    },
    {
      "epoch": 4.899120330810547e-05,
      "step": 51371,
      "training_step_time": 8.920078992843628
    },
    {
      "epoch": 4.8992156982421875e-05,
      "model_forward_time": 2.2429425716400146,
      "step": 51372
    },
    {
      "epoch": 4.8992156982421875e-05,
      "step": 51372,
      "training_step_time": 9.880368947982788
    },
    {
      "epoch": 4.899311065673828e-05,
      "model_forward_time": 1.6985132694244385,
      "step": 51373
    },
    {
      "epoch": 4.899311065673828e-05,
      "step": 51373,
      "training_step_time": 7.962914943695068
    },
    {
      "epoch": 4.899406433105469e-05,
      "model_forward_time": 1.7028751373291016,
      "step": 51374
    },
    {
      "epoch": 4.899406433105469e-05,
      "step": 51374,
      "training_step_time": 8.013343334197998
    },
    {
      "epoch": 4.8995018005371094e-05,
      "model_forward_time": 1.6996557712554932,
      "step": 51375
    },
    {
      "epoch": 4.8995018005371094e-05,
      "step": 51375,
      "training_step_time": 7.68713903427124
    },
    {
      "epoch": 4.89959716796875e-05,
      "model_forward_time": 1.6948871612548828,
      "step": 51376
    },
    {
      "epoch": 4.89959716796875e-05,
      "step": 51376,
      "training_step_time": 7.699417352676392
    },
    {
      "epoch": 4.899692535400391e-05,
      "model_forward_time": 1.7044804096221924,
      "step": 51377
    },
    {
      "epoch": 4.899692535400391e-05,
      "step": 51377,
      "training_step_time": 7.877325534820557
    },
    {
      "epoch": 4.899787902832031e-05,
      "model_forward_time": 1.6991591453552246,
      "step": 51378
    },
    {
      "epoch": 4.899787902832031e-05,
      "step": 51378,
      "training_step_time": 7.748960256576538
    },
    {
      "epoch": 4.899883270263672e-05,
      "model_forward_time": 1.7033250331878662,
      "step": 51379
    },
    {
      "epoch": 4.899883270263672e-05,
      "step": 51379,
      "training_step_time": 7.867782831192017
    },
    {
      "epoch": 4.899978637695313e-05,
      "grad_norm": 0.1278602033853531,
      "learning_rate": 5.1853115158914595e-06,
      "loss": 0.0554,
      "step": 51380
    },
    {
      "epoch": 4.899978637695313e-05,
      "model_forward_time": 1.7432351112365723,
      "step": 51380
    },
    {
      "dynamics_loss_avg": 0.06107878442853689,
      "epoch": 4.899978637695313e-05,
      "step": 51380
    },
    {
      "action_loss_avg": 0.0037896058725891634,
      "epoch": 4.899978637695313e-05,
      "step": 51380
    },
    {
      "epoch": 4.899978637695313e-05,
      "step": 51380,
      "training_step_time": 8.49250078201294
    },
    {
      "epoch": 4.900074005126953e-05,
      "model_forward_time": 1.6945858001708984,
      "step": 51381
    },
    {
      "epoch": 4.900074005126953e-05,
      "step": 51381,
      "training_step_time": 9.076698303222656
    },
    {
      "epoch": 4.9001693725585935e-05,
      "model_forward_time": 2.2560107707977295,
      "step": 51382
    },
    {
      "epoch": 4.9001693725585935e-05,
      "step": 51382,
      "training_step_time": 8.009938478469849
    },
    {
      "epoch": 4.9002647399902346e-05,
      "model_forward_time": 1.7008607387542725,
      "step": 51383
    },
    {
      "epoch": 4.9002647399902346e-05,
      "step": 51383,
      "training_step_time": 9.044333219528198
    },
    {
      "epoch": 4.900360107421875e-05,
      "model_forward_time": 1.705564022064209,
      "step": 51384
    },
    {
      "epoch": 4.900360107421875e-05,
      "step": 51384,
      "training_step_time": 7.956090927124023
    },
    {
      "epoch": 4.9004554748535154e-05,
      "model_forward_time": 1.6981019973754883,
      "step": 51385
    },
    {
      "epoch": 4.9004554748535154e-05,
      "step": 51385,
      "training_step_time": 7.843977689743042
    },
    {
      "epoch": 4.9005508422851565e-05,
      "model_forward_time": 1.70261549949646,
      "step": 51386
    },
    {
      "epoch": 4.9005508422851565e-05,
      "step": 51386,
      "training_step_time": 7.981288433074951
    },
    {
      "epoch": 4.900646209716797e-05,
      "model_forward_time": 1.700554609298706,
      "step": 51387
    },
    {
      "epoch": 4.900646209716797e-05,
      "step": 51387,
      "training_step_time": 7.803071022033691
    },
    {
      "epoch": 4.900741577148437e-05,
      "model_forward_time": 1.696385383605957,
      "step": 51388
    },
    {
      "epoch": 4.900741577148437e-05,
      "step": 51388,
      "training_step_time": 7.8975770473480225
    },
    {
      "epoch": 4.9008369445800784e-05,
      "model_forward_time": 1.7346174716949463,
      "step": 51389
    },
    {
      "epoch": 4.9008369445800784e-05,
      "step": 51389,
      "training_step_time": 7.996391534805298
    },
    {
      "epoch": 4.900932312011719e-05,
      "grad_norm": 0.11390388756990433,
      "learning_rate": 5.183659171976351e-06,
      "loss": 0.0549,
      "step": 51390
    },
    {
      "epoch": 4.900932312011719e-05,
      "model_forward_time": 1.6976635456085205,
      "step": 51390
    },
    {
      "dynamics_loss_avg": 0.0733697710558772,
      "epoch": 4.900932312011719e-05,
      "step": 51390
    },
    {
      "action_loss_avg": 0.0031716435973066837,
      "epoch": 4.900932312011719e-05,
      "step": 51390
    },
    {
      "epoch": 4.900932312011719e-05,
      "step": 51390,
      "training_step_time": 8.082406044006348
    },
    {
      "epoch": 4.901027679443359e-05,
      "model_forward_time": 2.2402970790863037,
      "step": 51391
    },
    {
      "epoch": 4.901027679443359e-05,
      "step": 51391,
      "training_step_time": 9.113484621047974
    },
    {
      "epoch": 4.901123046875e-05,
      "model_forward_time": 1.7028274536132812,
      "step": 51392
    },
    {
      "epoch": 4.901123046875e-05,
      "step": 51392,
      "training_step_time": 8.917332887649536
    },
    {
      "epoch": 4.901218414306641e-05,
      "model_forward_time": 1.6935575008392334,
      "step": 51393
    },
    {
      "epoch": 4.901218414306641e-05,
      "step": 51393,
      "training_step_time": 8.095717906951904
    },
    {
      "epoch": 4.901313781738281e-05,
      "model_forward_time": 1.704101800918579,
      "step": 51394
    },
    {
      "epoch": 4.901313781738281e-05,
      "step": 51394,
      "training_step_time": 7.7427592277526855
    },
    {
      "epoch": 4.901409149169922e-05,
      "model_forward_time": 1.6971900463104248,
      "step": 51395
    },
    {
      "epoch": 4.901409149169922e-05,
      "step": 51395,
      "training_step_time": 7.9429357051849365
    },
    {
      "epoch": 4.9015045166015626e-05,
      "model_forward_time": 1.6989567279815674,
      "step": 51396
    },
    {
      "epoch": 4.9015045166015626e-05,
      "step": 51396,
      "training_step_time": 7.833806276321411
    },
    {
      "epoch": 4.901599884033203e-05,
      "model_forward_time": 1.7000839710235596,
      "step": 51397
    },
    {
      "epoch": 4.901599884033203e-05,
      "step": 51397,
      "training_step_time": 8.657898187637329
    },
    {
      "epoch": 4.901695251464844e-05,
      "model_forward_time": 1.7084364891052246,
      "step": 51398
    },
    {
      "epoch": 4.901695251464844e-05,
      "step": 51398,
      "training_step_time": 7.979095697402954
    },
    {
      "epoch": 4.9017906188964845e-05,
      "model_forward_time": 1.706040859222412,
      "step": 51399
    },
    {
      "epoch": 4.9017906188964845e-05,
      "step": 51399,
      "training_step_time": 9.505549669265747
    },
    {
      "epoch": 4.901885986328125e-05,
      "grad_norm": 0.11046170443296432,
      "learning_rate": 5.182006807976554e-06,
      "loss": 0.0579,
      "step": 51400
    },
    {
      "epoch": 4.901885986328125e-05,
      "model_forward_time": 2.253915786743164,
      "step": 51400
    },
    {
      "dynamics_loss_avg": 0.07343541048467159,
      "epoch": 4.901885986328125e-05,
      "step": 51400
    },
    {
      "action_loss_avg": 0.0037354064639657735,
      "epoch": 4.901885986328125e-05,
      "step": 51400
    },
    {
      "epoch": 4.901885986328125e-05,
      "step": 51400,
      "training_step_time": 8.577012062072754
    },
    {
      "epoch": 4.901981353759766e-05,
      "model_forward_time": 1.7024705410003662,
      "step": 51401
    },
    {
      "epoch": 4.901981353759766e-05,
      "step": 51401,
      "training_step_time": 7.986171722412109
    },
    {
      "epoch": 4.9020767211914063e-05,
      "model_forward_time": 1.6992554664611816,
      "step": 51402
    },
    {
      "epoch": 4.9020767211914063e-05,
      "step": 51402,
      "training_step_time": 7.722620010375977
    },
    {
      "epoch": 4.902172088623047e-05,
      "model_forward_time": 1.695181131362915,
      "step": 51403
    },
    {
      "epoch": 4.902172088623047e-05,
      "step": 51403,
      "training_step_time": 7.480091333389282
    },
    {
      "epoch": 4.902267456054688e-05,
      "model_forward_time": 1.7099931240081787,
      "step": 51404
    },
    {
      "epoch": 4.902267456054688e-05,
      "step": 51404,
      "training_step_time": 7.663362264633179
    },
    {
      "epoch": 4.902362823486328e-05,
      "model_forward_time": 1.6998989582061768,
      "step": 51405
    },
    {
      "epoch": 4.902362823486328e-05,
      "step": 51405,
      "training_step_time": 7.640424966812134
    },
    {
      "epoch": 4.9024581909179686e-05,
      "model_forward_time": 1.6963062286376953,
      "step": 51406
    },
    {
      "epoch": 4.9024581909179686e-05,
      "step": 51406,
      "training_step_time": 7.680443286895752
    },
    {
      "epoch": 4.902553558349609e-05,
      "model_forward_time": 1.700141429901123,
      "step": 51407
    },
    {
      "epoch": 4.902553558349609e-05,
      "step": 51407,
      "training_step_time": 7.707683324813843
    },
    {
      "epoch": 4.90264892578125e-05,
      "model_forward_time": 1.7410836219787598,
      "step": 51408
    },
    {
      "epoch": 4.90264892578125e-05,
      "step": 51408,
      "training_step_time": 9.524138450622559
    },
    {
      "epoch": 4.9027442932128905e-05,
      "model_forward_time": 2.2766759395599365,
      "step": 51409
    },
    {
      "epoch": 4.9027442932128905e-05,
      "step": 51409,
      "training_step_time": 9.310350179672241
    },
    {
      "epoch": 4.902839660644531e-05,
      "grad_norm": 0.10487346351146698,
      "learning_rate": 5.180354424072763e-06,
      "loss": 0.0565,
      "step": 51410
    },
    {
      "epoch": 4.902839660644531e-05,
      "model_forward_time": 1.6947431564331055,
      "step": 51410
    },
    {
      "dynamics_loss_avg": 0.046963535342365505,
      "epoch": 4.902839660644531e-05,
      "step": 51410
    },
    {
      "action_loss_avg": 0.0014277675072662532,
      "epoch": 4.902839660644531e-05,
      "step": 51410
    },
    {
      "epoch": 4.902839660644531e-05,
      "step": 51410,
      "training_step_time": 7.7659220695495605
    },
    {
      "epoch": 4.902935028076172e-05,
      "model_forward_time": 1.7076244354248047,
      "step": 51411
    },
    {
      "epoch": 4.902935028076172e-05,
      "step": 51411,
      "training_step_time": 8.751806259155273
    },
    {
      "epoch": 4.9030303955078124e-05,
      "model_forward_time": 1.6906406879425049,
      "step": 51412
    },
    {
      "epoch": 4.9030303955078124e-05,
      "step": 51412,
      "training_step_time": 7.969998598098755
    },
    {
      "epoch": 4.903125762939453e-05,
      "model_forward_time": 1.7044274806976318,
      "step": 51413
    },
    {
      "epoch": 4.903125762939453e-05,
      "step": 51413,
      "training_step_time": 7.6581103801727295
    },
    {
      "epoch": 4.903221130371094e-05,
      "model_forward_time": 1.7152595520019531,
      "step": 51414
    },
    {
      "epoch": 4.903221130371094e-05,
      "step": 51414,
      "training_step_time": 7.632462024688721
    },
    {
      "epoch": 4.903316497802734e-05,
      "model_forward_time": 1.6976537704467773,
      "step": 51415
    },
    {
      "epoch": 4.903316497802734e-05,
      "step": 51415,
      "training_step_time": 6.847691774368286
    },
    {
      "epoch": 4.903411865234375e-05,
      "model_forward_time": 1.6985344886779785,
      "step": 51416
    },
    {
      "epoch": 4.903411865234375e-05,
      "step": 51416,
      "training_step_time": 7.745002508163452
    },
    {
      "epoch": 4.903507232666016e-05,
      "model_forward_time": 1.7479543685913086,
      "step": 51417
    },
    {
      "epoch": 4.903507232666016e-05,
      "step": 51417,
      "training_step_time": 7.833584308624268
    },
    {
      "epoch": 4.903602600097656e-05,
      "model_forward_time": 2.2386720180511475,
      "step": 51418
    },
    {
      "epoch": 4.903602600097656e-05,
      "step": 51418,
      "training_step_time": 8.96466588973999
    },
    {
      "epoch": 4.9036979675292966e-05,
      "model_forward_time": 1.7201955318450928,
      "step": 51419
    },
    {
      "epoch": 4.9036979675292966e-05,
      "step": 51419,
      "training_step_time": 9.233474731445312
    },
    {
      "epoch": 4.903793334960938e-05,
      "grad_norm": 0.07529179751873016,
      "learning_rate": 5.178702020445686e-06,
      "loss": 0.0531,
      "step": 51420
    },
    {
      "epoch": 4.903793334960938e-05,
      "model_forward_time": 1.6911535263061523,
      "step": 51420
    },
    {
      "dynamics_loss_avg": 0.057034118101000786,
      "epoch": 4.903793334960938e-05,
      "step": 51420
    },
    {
      "action_loss_avg": 0.0031211689056362957,
      "epoch": 4.903793334960938e-05,
      "step": 51420
    },
    {
      "epoch": 4.903793334960938e-05,
      "step": 51420,
      "training_step_time": 7.686877012252808
    },
    {
      "epoch": 4.903888702392578e-05,
      "model_forward_time": 1.7067592144012451,
      "step": 51421
    },
    {
      "epoch": 4.903888702392578e-05,
      "step": 51421,
      "training_step_time": 7.671977996826172
    },
    {
      "epoch": 4.9039840698242185e-05,
      "model_forward_time": 1.6989707946777344,
      "step": 51422
    },
    {
      "epoch": 4.9039840698242185e-05,
      "step": 51422,
      "training_step_time": 8.629528522491455
    },
    {
      "epoch": 4.9040794372558595e-05,
      "model_forward_time": 1.7064247131347656,
      "step": 51423
    },
    {
      "epoch": 4.9040794372558595e-05,
      "step": 51423,
      "training_step_time": 7.600037574768066
    },
    {
      "epoch": 4.9041748046875e-05,
      "model_forward_time": 1.3315205574035645,
      "step": 51424
    },
    {
      "epoch": 4.9041748046875e-05,
      "step": 51424,
      "training_step_time": 6.756722688674927
    },
    {
      "epoch": 4.9042701721191404e-05,
      "model_forward_time": 1.7050433158874512,
      "step": 51425
    },
    {
      "epoch": 4.9042701721191404e-05,
      "step": 51425,
      "training_step_time": 7.726674556732178
    },
    {
      "epoch": 4.9043655395507814e-05,
      "model_forward_time": 1.7039308547973633,
      "step": 51426
    },
    {
      "epoch": 4.9043655395507814e-05,
      "step": 51426,
      "training_step_time": 8.7420973777771
    },
    {
      "epoch": 4.904460906982422e-05,
      "model_forward_time": 1.7148253917694092,
      "step": 51427
    },
    {
      "epoch": 4.904460906982422e-05,
      "step": 51427,
      "training_step_time": 8.804640531539917
    },
    {
      "epoch": 4.904556274414062e-05,
      "model_forward_time": 2.261482000350952,
      "step": 51428
    },
    {
      "epoch": 4.904556274414062e-05,
      "step": 51428,
      "training_step_time": 8.509563207626343
    },
    {
      "epoch": 4.904651641845703e-05,
      "model_forward_time": 1.7131648063659668,
      "step": 51429
    },
    {
      "epoch": 4.904651641845703e-05,
      "step": 51429,
      "training_step_time": 8.459210634231567
    },
    {
      "epoch": 4.904747009277344e-05,
      "grad_norm": 0.11588888615369797,
      "learning_rate": 5.1770495972760234e-06,
      "loss": 0.0564,
      "step": 51430
    },
    {
      "epoch": 4.904747009277344e-05,
      "model_forward_time": 1.7023389339447021,
      "step": 51430
    },
    {
      "dynamics_loss_avg": 0.07140580974519253,
      "epoch": 4.904747009277344e-05,
      "step": 51430
    },
    {
      "action_loss_avg": 0.0018017128575593233,
      "epoch": 4.904747009277344e-05,
      "step": 51430
    },
    {
      "epoch": 4.904747009277344e-05,
      "step": 51430,
      "training_step_time": 8.215399026870728
    },
    {
      "epoch": 4.904842376708984e-05,
      "model_forward_time": 1.712991714477539,
      "step": 51431
    },
    {
      "epoch": 4.904842376708984e-05,
      "step": 51431,
      "training_step_time": 7.051380634307861
    },
    {
      "epoch": 4.904937744140625e-05,
      "model_forward_time": 1.712538242340088,
      "step": 51432
    },
    {
      "epoch": 4.904937744140625e-05,
      "step": 51432,
      "training_step_time": 7.622002363204956
    },
    {
      "epoch": 4.9050331115722656e-05,
      "model_forward_time": 1.705376148223877,
      "step": 51433
    },
    {
      "epoch": 4.9050331115722656e-05,
      "step": 51433,
      "training_step_time": 7.258471727371216
    },
    {
      "epoch": 4.905128479003906e-05,
      "model_forward_time": 1.7039124965667725,
      "step": 51434
    },
    {
      "epoch": 4.905128479003906e-05,
      "step": 51434,
      "training_step_time": 7.706219673156738
    },
    {
      "epoch": 4.905223846435547e-05,
      "model_forward_time": 1.7228851318359375,
      "step": 51435
    },
    {
      "epoch": 4.905223846435547e-05,
      "step": 51435,
      "training_step_time": 7.658679723739624
    },
    {
      "epoch": 4.9053192138671875e-05,
      "model_forward_time": 1.7163732051849365,
      "step": 51436
    },
    {
      "epoch": 4.9053192138671875e-05,
      "step": 51436,
      "training_step_time": 9.602483034133911
    },
    {
      "epoch": 4.905414581298828e-05,
      "model_forward_time": 2.2892725467681885,
      "step": 51437
    },
    {
      "epoch": 4.905414581298828e-05,
      "step": 51437,
      "training_step_time": 8.87852430343628
    },
    {
      "epoch": 4.905509948730469e-05,
      "model_forward_time": 1.6998827457427979,
      "step": 51438
    },
    {
      "epoch": 4.905509948730469e-05,
      "step": 51438,
      "training_step_time": 7.748225688934326
    },
    {
      "epoch": 4.9056053161621094e-05,
      "model_forward_time": 1.7020106315612793,
      "step": 51439
    },
    {
      "epoch": 4.9056053161621094e-05,
      "step": 51439,
      "training_step_time": 7.709240913391113
    },
    {
      "epoch": 4.90570068359375e-05,
      "grad_norm": 0.08535255491733551,
      "learning_rate": 5.175397154744483e-06,
      "loss": 0.0572,
      "step": 51440
    },
    {
      "epoch": 4.90570068359375e-05,
      "model_forward_time": 1.699491024017334,
      "step": 51440
    },
    {
      "dynamics_loss_avg": 0.0662528458982706,
      "epoch": 4.90570068359375e-05,
      "step": 51440
    },
    {
      "action_loss_avg": 0.002649120695423335,
      "epoch": 4.90570068359375e-05,
      "step": 51440
    },
    {
      "epoch": 4.90570068359375e-05,
      "step": 51440,
      "training_step_time": 7.36327862739563
    },
    {
      "epoch": 4.905796051025391e-05,
      "model_forward_time": 1.6949326992034912,
      "step": 51441
    },
    {
      "epoch": 4.905796051025391e-05,
      "step": 51441,
      "training_step_time": 7.6428868770599365
    },
    {
      "epoch": 4.905891418457031e-05,
      "model_forward_time": 1.6959481239318848,
      "step": 51442
    },
    {
      "epoch": 4.905891418457031e-05,
      "step": 51442,
      "training_step_time": 7.431149244308472
    },
    {
      "epoch": 4.905986785888672e-05,
      "model_forward_time": 1.7000632286071777,
      "step": 51443
    },
    {
      "epoch": 4.905986785888672e-05,
      "step": 51443,
      "training_step_time": 8.327648162841797
    },
    {
      "epoch": 4.906082153320313e-05,
      "model_forward_time": 1.6930723190307617,
      "step": 51444
    },
    {
      "epoch": 4.906082153320313e-05,
      "step": 51444,
      "training_step_time": 8.422568082809448
    },
    {
      "epoch": 4.906177520751953e-05,
      "model_forward_time": 1.7125473022460938,
      "step": 51445
    },
    {
      "epoch": 4.906177520751953e-05,
      "step": 51445,
      "training_step_time": 9.364803314208984
    },
    {
      "epoch": 4.9062728881835936e-05,
      "model_forward_time": 2.2661170959472656,
      "step": 51446
    },
    {
      "epoch": 4.9062728881835936e-05,
      "step": 51446,
      "training_step_time": 8.736960649490356
    },
    {
      "epoch": 4.9063682556152346e-05,
      "model_forward_time": 1.714087724685669,
      "step": 51447
    },
    {
      "epoch": 4.9063682556152346e-05,
      "step": 51447,
      "training_step_time": 7.726013660430908
    },
    {
      "epoch": 4.906463623046875e-05,
      "model_forward_time": 1.7093822956085205,
      "step": 51448
    },
    {
      "epoch": 4.906463623046875e-05,
      "step": 51448,
      "training_step_time": 7.688060283660889
    },
    {
      "epoch": 4.9065589904785154e-05,
      "model_forward_time": 1.7030904293060303,
      "step": 51449
    },
    {
      "epoch": 4.9065589904785154e-05,
      "step": 51449,
      "training_step_time": 7.4514665603637695
    },
    {
      "epoch": 4.9066543579101565e-05,
      "grad_norm": 0.07956372201442719,
      "learning_rate": 5.173744693031773e-06,
      "loss": 0.0593,
      "step": 51450
    },
    {
      "epoch": 4.9066543579101565e-05,
      "model_forward_time": 1.7054336071014404,
      "step": 51450
    },
    {
      "dynamics_loss_avg": 0.0532345337793231,
      "epoch": 4.9066543579101565e-05,
      "step": 51450
    },
    {
      "action_loss_avg": 0.0026246572189847937,
      "epoch": 4.9066543579101565e-05,
      "step": 51450
    },
    {
      "epoch": 4.9066543579101565e-05,
      "step": 51450,
      "training_step_time": 8.646400690078735
    },
    {
      "epoch": 4.906749725341797e-05,
      "model_forward_time": 1.701648473739624,
      "step": 51451
    },
    {
      "epoch": 4.906749725341797e-05,
      "step": 51451,
      "training_step_time": 7.346462249755859
    },
    {
      "epoch": 4.906845092773437e-05,
      "model_forward_time": 1.6997389793395996,
      "step": 51452
    },
    {
      "epoch": 4.906845092773437e-05,
      "step": 51452,
      "training_step_time": 8.391813278198242
    },
    {
      "epoch": 4.9069404602050784e-05,
      "model_forward_time": 1.7089123725891113,
      "step": 51453
    },
    {
      "epoch": 4.9069404602050784e-05,
      "step": 51453,
      "training_step_time": 7.6219482421875
    },
    {
      "epoch": 4.907035827636719e-05,
      "model_forward_time": 1.7438197135925293,
      "step": 51454
    },
    {
      "epoch": 4.907035827636719e-05,
      "step": 51454,
      "training_step_time": 8.38272762298584
    },
    {
      "epoch": 4.907131195068359e-05,
      "model_forward_time": 2.275038719177246,
      "step": 51455
    },
    {
      "epoch": 4.907131195068359e-05,
      "step": 51455,
      "training_step_time": 8.642014503479004
    },
    {
      "epoch": 4.9072265625e-05,
      "model_forward_time": 1.7029798030853271,
      "step": 51456
    },
    {
      "epoch": 4.9072265625e-05,
      "step": 51456,
      "training_step_time": 7.6250081062316895
    },
    {
      "epoch": 4.907321929931641e-05,
      "model_forward_time": 1.7043426036834717,
      "step": 51457
    },
    {
      "epoch": 4.907321929931641e-05,
      "step": 51457,
      "training_step_time": 8.689966917037964
    },
    {
      "epoch": 4.907417297363281e-05,
      "model_forward_time": 1.7056999206542969,
      "step": 51458
    },
    {
      "epoch": 4.907417297363281e-05,
      "step": 51458,
      "training_step_time": 7.342503786087036
    },
    {
      "epoch": 4.907512664794922e-05,
      "model_forward_time": 1.7038993835449219,
      "step": 51459
    },
    {
      "epoch": 4.907512664794922e-05,
      "step": 51459,
      "training_step_time": 7.411025524139404
    },
    {
      "epoch": 4.9076080322265626e-05,
      "grad_norm": 0.08092699199914932,
      "learning_rate": 5.172092212318605e-06,
      "loss": 0.0554,
      "step": 51460
    },
    {
      "epoch": 4.9076080322265626e-05,
      "model_forward_time": 1.6984527111053467,
      "step": 51460
    },
    {
      "dynamics_loss_avg": 0.05342306373640895,
      "epoch": 4.9076080322265626e-05,
      "step": 51460
    },
    {
      "action_loss_avg": 0.003668856514559593,
      "epoch": 4.9076080322265626e-05,
      "step": 51460
    },
    {
      "epoch": 4.9076080322265626e-05,
      "step": 51460,
      "training_step_time": 8.433260679244995
    },
    {
      "epoch": 4.907703399658203e-05,
      "model_forward_time": 1.7033352851867676,
      "step": 51461
    },
    {
      "epoch": 4.907703399658203e-05,
      "step": 51461,
      "training_step_time": 7.598112344741821
    },
    {
      "epoch": 4.907798767089844e-05,
      "model_forward_time": 1.702488660812378,
      "step": 51462
    },
    {
      "epoch": 4.907798767089844e-05,
      "step": 51462,
      "training_step_time": 8.464115619659424
    },
    {
      "epoch": 4.9078941345214845e-05,
      "model_forward_time": 1.7378265857696533,
      "step": 51463
    },
    {
      "epoch": 4.9078941345214845e-05,
      "step": 51463,
      "training_step_time": 7.629073619842529
    },
    {
      "epoch": 4.907989501953125e-05,
      "model_forward_time": 1.696674108505249,
      "step": 51464
    },
    {
      "epoch": 4.907989501953125e-05,
      "step": 51464,
      "training_step_time": 10.336970329284668
    },
    {
      "epoch": 4.908084869384766e-05,
      "model_forward_time": 1.7024250030517578,
      "step": 51465
    },
    {
      "epoch": 4.908084869384766e-05,
      "step": 51465,
      "training_step_time": 8.731539487838745
    },
    {
      "epoch": 4.9081802368164064e-05,
      "model_forward_time": 1.6985437870025635,
      "step": 51466
    },
    {
      "epoch": 4.9081802368164064e-05,
      "step": 51466,
      "training_step_time": 8.742830991744995
    },
    {
      "epoch": 4.908275604248047e-05,
      "model_forward_time": 1.7048332691192627,
      "step": 51467
    },
    {
      "epoch": 4.908275604248047e-05,
      "step": 51467,
      "training_step_time": 7.6868414878845215
    },
    {
      "epoch": 4.908370971679688e-05,
      "model_forward_time": 1.7015185356140137,
      "step": 51468
    },
    {
      "epoch": 4.908370971679688e-05,
      "step": 51468,
      "training_step_time": 8.377111196517944
    },
    {
      "epoch": 4.908466339111328e-05,
      "model_forward_time": 1.7056608200073242,
      "step": 51469
    },
    {
      "epoch": 4.908466339111328e-05,
      "step": 51469,
      "training_step_time": 7.570080041885376
    },
    {
      "epoch": 4.9085617065429686e-05,
      "grad_norm": 0.1014472246170044,
      "learning_rate": 5.170439712785692e-06,
      "loss": 0.0577,
      "step": 51470
    },
    {
      "epoch": 4.9085617065429686e-05,
      "model_forward_time": 1.704514980316162,
      "step": 51470
    },
    {
      "dynamics_loss_avg": 0.06253404784947633,
      "epoch": 4.9085617065429686e-05,
      "step": 51470
    },
    {
      "action_loss_avg": 0.0030709819664480166,
      "epoch": 4.9085617065429686e-05,
      "step": 51470
    },
    {
      "epoch": 4.9085617065429686e-05,
      "step": 51470,
      "training_step_time": 7.511953115463257
    },
    {
      "epoch": 4.908657073974609e-05,
      "model_forward_time": 1.7048118114471436,
      "step": 51471
    },
    {
      "epoch": 4.908657073974609e-05,
      "step": 51471,
      "training_step_time": 9.39358925819397
    },
    {
      "epoch": 4.90875244140625e-05,
      "model_forward_time": 1.704538106918335,
      "step": 51472
    },
    {
      "epoch": 4.90875244140625e-05,
      "step": 51472,
      "training_step_time": 7.6453330516815186
    },
    {
      "epoch": 4.9088478088378905e-05,
      "model_forward_time": 1.6986267566680908,
      "step": 51473
    },
    {
      "epoch": 4.9088478088378905e-05,
      "step": 51473,
      "training_step_time": 7.7679948806762695
    },
    {
      "epoch": 4.908943176269531e-05,
      "model_forward_time": 2.2543137073516846,
      "step": 51474
    },
    {
      "epoch": 4.908943176269531e-05,
      "step": 51474,
      "training_step_time": 9.923566102981567
    },
    {
      "epoch": 4.909038543701172e-05,
      "model_forward_time": 1.705742359161377,
      "step": 51475
    },
    {
      "epoch": 4.909038543701172e-05,
      "step": 51475,
      "training_step_time": 8.677520751953125
    },
    {
      "epoch": 4.9091339111328124e-05,
      "model_forward_time": 1.713404655456543,
      "step": 51476
    },
    {
      "epoch": 4.9091339111328124e-05,
      "step": 51476,
      "training_step_time": 8.196799278259277
    },
    {
      "epoch": 4.909229278564453e-05,
      "model_forward_time": 1.7025372982025146,
      "step": 51477
    },
    {
      "epoch": 4.909229278564453e-05,
      "step": 51477,
      "training_step_time": 7.363050222396851
    },
    {
      "epoch": 4.909324645996094e-05,
      "model_forward_time": 1.701164722442627,
      "step": 51478
    },
    {
      "epoch": 4.909324645996094e-05,
      "step": 51478,
      "training_step_time": 7.597363710403442
    },
    {
      "epoch": 4.909420013427734e-05,
      "model_forward_time": 1.711150884628296,
      "step": 51479
    },
    {
      "epoch": 4.909420013427734e-05,
      "step": 51479,
      "training_step_time": 7.602326393127441
    },
    {
      "epoch": 4.909515380859375e-05,
      "grad_norm": 0.10959625989198685,
      "learning_rate": 5.168787194613747e-06,
      "loss": 0.0535,
      "step": 51480
    },
    {
      "epoch": 4.909515380859375e-05,
      "model_forward_time": 1.6972689628601074,
      "step": 51480
    },
    {
      "dynamics_loss_avg": 0.05928036328405142,
      "epoch": 4.909515380859375e-05,
      "step": 51480
    },
    {
      "action_loss_avg": 0.002655868750298396,
      "epoch": 4.909515380859375e-05,
      "step": 51480
    },
    {
      "epoch": 4.909515380859375e-05,
      "step": 51480,
      "training_step_time": 7.519055128097534
    },
    {
      "epoch": 4.909610748291016e-05,
      "model_forward_time": 1.7418639659881592,
      "step": 51481
    },
    {
      "epoch": 4.909610748291016e-05,
      "step": 51481,
      "training_step_time": 8.440744876861572
    },
    {
      "epoch": 4.909706115722656e-05,
      "model_forward_time": 1.749457836151123,
      "step": 51482
    },
    {
      "epoch": 4.909706115722656e-05,
      "step": 51482,
      "training_step_time": 8.59229564666748
    },
    {
      "epoch": 4.9098014831542966e-05,
      "model_forward_time": 1.5241518020629883,
      "step": 51483
    },
    {
      "epoch": 4.9098014831542966e-05,
      "step": 51483,
      "training_step_time": 10.270160913467407
    },
    {
      "epoch": 4.909896850585938e-05,
      "model_forward_time": 1.7080183029174805,
      "step": 51484
    },
    {
      "epoch": 4.909896850585938e-05,
      "step": 51484,
      "training_step_time": 8.663750410079956
    },
    {
      "epoch": 4.909992218017578e-05,
      "model_forward_time": 1.704982042312622,
      "step": 51485
    },
    {
      "epoch": 4.909992218017578e-05,
      "step": 51485,
      "training_step_time": 8.689009189605713
    },
    {
      "epoch": 4.9100875854492185e-05,
      "model_forward_time": 1.7043793201446533,
      "step": 51486
    },
    {
      "epoch": 4.9100875854492185e-05,
      "step": 51486,
      "training_step_time": 7.357389211654663
    },
    {
      "epoch": 4.9101829528808596e-05,
      "model_forward_time": 1.704505205154419,
      "step": 51487
    },
    {
      "epoch": 4.9101829528808596e-05,
      "step": 51487,
      "training_step_time": 7.514249563217163
    },
    {
      "epoch": 4.9102783203125e-05,
      "model_forward_time": 1.7066638469696045,
      "step": 51488
    },
    {
      "epoch": 4.9102783203125e-05,
      "step": 51488,
      "training_step_time": 7.55213737487793
    },
    {
      "epoch": 4.9103736877441404e-05,
      "model_forward_time": 1.7077052593231201,
      "step": 51489
    },
    {
      "epoch": 4.9103736877441404e-05,
      "step": 51489,
      "training_step_time": 8.183635950088501
    },
    {
      "epoch": 4.9104690551757814e-05,
      "grad_norm": 0.11409195512533188,
      "learning_rate": 5.16713465798349e-06,
      "loss": 0.0575,
      "step": 51490
    },
    {
      "epoch": 4.9104690551757814e-05,
      "model_forward_time": 1.695793867111206,
      "step": 51490
    },
    {
      "dynamics_loss_avg": 0.08409631457179785,
      "epoch": 4.9104690551757814e-05,
      "step": 51490
    },
    {
      "action_loss_avg": 0.004101802024524659,
      "epoch": 4.9104690551757814e-05,
      "step": 51490
    },
    {
      "epoch": 4.9104690551757814e-05,
      "step": 51490,
      "training_step_time": 7.549381494522095
    },
    {
      "epoch": 4.910564422607422e-05,
      "model_forward_time": 1.7290947437286377,
      "step": 51491
    },
    {
      "epoch": 4.910564422607422e-05,
      "step": 51491,
      "training_step_time": 7.6511499881744385
    },
    {
      "epoch": 4.910659790039062e-05,
      "model_forward_time": 2.2604775428771973,
      "step": 51492
    },
    {
      "epoch": 4.910659790039062e-05,
      "step": 51492,
      "training_step_time": 10.98607611656189
    },
    {
      "epoch": 4.910755157470703e-05,
      "model_forward_time": 1.712503433227539,
      "step": 51493
    },
    {
      "epoch": 4.910755157470703e-05,
      "step": 51493,
      "training_step_time": 7.669247150421143
    },
    {
      "epoch": 4.910850524902344e-05,
      "model_forward_time": 1.7044460773468018,
      "step": 51494
    },
    {
      "epoch": 4.910850524902344e-05,
      "step": 51494,
      "training_step_time": 8.249389886856079
    },
    {
      "epoch": 4.910945892333984e-05,
      "model_forward_time": 1.706907033920288,
      "step": 51495
    },
    {
      "epoch": 4.910945892333984e-05,
      "step": 51495,
      "training_step_time": 7.3811256885528564
    },
    {
      "epoch": 4.911041259765625e-05,
      "model_forward_time": 1.7123618125915527,
      "step": 51496
    },
    {
      "epoch": 4.911041259765625e-05,
      "step": 51496,
      "training_step_time": 7.664424896240234
    },
    {
      "epoch": 4.9111366271972656e-05,
      "model_forward_time": 1.7044165134429932,
      "step": 51497
    },
    {
      "epoch": 4.9111366271972656e-05,
      "step": 51497,
      "training_step_time": 7.6217522621154785
    },
    {
      "epoch": 4.911231994628906e-05,
      "model_forward_time": 1.706329345703125,
      "step": 51498
    },
    {
      "epoch": 4.911231994628906e-05,
      "step": 51498,
      "training_step_time": 7.5640459060668945
    },
    {
      "epoch": 4.911327362060547e-05,
      "model_forward_time": 1.7134113311767578,
      "step": 51499
    },
    {
      "epoch": 4.911327362060547e-05,
      "step": 51499,
      "training_step_time": 7.654082775115967
    },
    {
      "epoch": 4.9114227294921875e-05,
      "grad_norm": 0.12577493488788605,
      "learning_rate": 5.165482103075637e-06,
      "loss": 0.0595,
      "step": 51500
    },
    {
      "epoch": 4.9114227294921875e-05,
      "model_forward_time": 1.710932970046997,
      "step": 51500
    },
    {
      "dynamics_loss_avg": 0.06414613798260689,
      "epoch": 4.9114227294921875e-05,
      "step": 51500
    },
    {
      "action_loss_avg": 0.0034123136312700806,
      "epoch": 4.9114227294921875e-05,
      "step": 51500
    },
    {
      "epoch": 4.9114227294921875e-05,
      "step": 51500,
      "training_step_time": 7.574726104736328
    },
    {
      "epoch": 4.911518096923828e-05,
      "model_forward_time": 1.6984901428222656,
      "step": 51501
    },
    {
      "epoch": 4.911518096923828e-05,
      "step": 51501,
      "training_step_time": 8.776894807815552
    },
    {
      "epoch": 4.911613464355469e-05,
      "model_forward_time": 2.263049840927124,
      "step": 51502
    },
    {
      "epoch": 4.911613464355469e-05,
      "step": 51502,
      "training_step_time": 8.136099815368652
    },
    {
      "epoch": 4.9117088317871094e-05,
      "model_forward_time": 1.700141429901123,
      "step": 51503
    },
    {
      "epoch": 4.9117088317871094e-05,
      "step": 51503,
      "training_step_time": 7.67413854598999
    },
    {
      "epoch": 4.91180419921875e-05,
      "model_forward_time": 1.70235276222229,
      "step": 51504
    },
    {
      "epoch": 4.91180419921875e-05,
      "step": 51504,
      "training_step_time": 7.415316343307495
    },
    {
      "epoch": 4.911899566650391e-05,
      "model_forward_time": 1.7065763473510742,
      "step": 51505
    },
    {
      "epoch": 4.911899566650391e-05,
      "step": 51505,
      "training_step_time": 8.42999529838562
    },
    {
      "epoch": 4.911994934082031e-05,
      "model_forward_time": 1.7014684677124023,
      "step": 51506
    },
    {
      "epoch": 4.911994934082031e-05,
      "step": 51506,
      "training_step_time": 8.005839586257935
    },
    {
      "epoch": 4.912090301513672e-05,
      "model_forward_time": 1.750166654586792,
      "step": 51507
    },
    {
      "epoch": 4.912090301513672e-05,
      "step": 51507,
      "training_step_time": 7.54904317855835
    },
    {
      "epoch": 4.912185668945313e-05,
      "model_forward_time": 1.706725835800171,
      "step": 51508
    },
    {
      "epoch": 4.912185668945313e-05,
      "step": 51508,
      "training_step_time": 8.370295286178589
    },
    {
      "epoch": 4.912281036376953e-05,
      "model_forward_time": 1.7020180225372314,
      "step": 51509
    },
    {
      "epoch": 4.912281036376953e-05,
      "step": 51509,
      "training_step_time": 7.547369003295898
    },
    {
      "epoch": 4.9123764038085936e-05,
      "grad_norm": 0.10765143483877182,
      "learning_rate": 5.16382953007091e-06,
      "loss": 0.0551,
      "step": 51510
    },
    {
      "epoch": 4.9123764038085936e-05,
      "model_forward_time": 1.7495372295379639,
      "step": 51510
    },
    {
      "dynamics_loss_avg": 0.06002675322815776,
      "epoch": 4.9123764038085936e-05,
      "step": 51510
    },
    {
      "action_loss_avg": 0.00302429887233302,
      "epoch": 4.9123764038085936e-05,
      "step": 51510
    },
    {
      "epoch": 4.9123764038085936e-05,
      "step": 51510,
      "training_step_time": 7.692876577377319
    },
    {
      "epoch": 4.9124717712402346e-05,
      "model_forward_time": 2.264136552810669,
      "step": 51511
    },
    {
      "epoch": 4.9124717712402346e-05,
      "step": 51511,
      "training_step_time": 8.964832544326782
    },
    {
      "epoch": 4.912567138671875e-05,
      "model_forward_time": 1.696073293685913,
      "step": 51512
    },
    {
      "epoch": 4.912567138671875e-05,
      "step": 51512,
      "training_step_time": 8.481679677963257
    },
    {
      "epoch": 4.9126625061035154e-05,
      "model_forward_time": 1.706824541091919,
      "step": 51513
    },
    {
      "epoch": 4.9126625061035154e-05,
      "step": 51513,
      "training_step_time": 7.592792272567749
    },
    {
      "epoch": 4.9127578735351565e-05,
      "model_forward_time": 1.7221028804779053,
      "step": 51514
    },
    {
      "epoch": 4.9127578735351565e-05,
      "step": 51514,
      "training_step_time": 7.3348352909088135
    },
    {
      "epoch": 4.912853240966797e-05,
      "model_forward_time": 1.7236318588256836,
      "step": 51515
    },
    {
      "epoch": 4.912853240966797e-05,
      "step": 51515,
      "training_step_time": 7.535954713821411
    },
    {
      "epoch": 4.912948608398437e-05,
      "model_forward_time": 1.7141482830047607,
      "step": 51516
    },
    {
      "epoch": 4.912948608398437e-05,
      "step": 51516,
      "training_step_time": 7.6886656284332275
    },
    {
      "epoch": 4.9130439758300784e-05,
      "model_forward_time": 1.7334785461425781,
      "step": 51517
    },
    {
      "epoch": 4.9130439758300784e-05,
      "step": 51517,
      "training_step_time": 9.379974126815796
    },
    {
      "epoch": 4.913139343261719e-05,
      "model_forward_time": 1.7634906768798828,
      "step": 51518
    },
    {
      "epoch": 4.913139343261719e-05,
      "step": 51518,
      "training_step_time": 8.770660638809204
    },
    {
      "epoch": 4.913234710693359e-05,
      "model_forward_time": 1.7690248489379883,
      "step": 51519
    },
    {
      "epoch": 4.913234710693359e-05,
      "step": 51519,
      "training_step_time": 7.714207410812378
    },
    {
      "epoch": 4.913330078125e-05,
      "grad_norm": 0.150406152009964,
      "learning_rate": 5.162176939150032e-06,
      "loss": 0.058,
      "step": 51520
    },
    {
      "epoch": 4.913330078125e-05,
      "model_forward_time": 2.942988634109497,
      "step": 51520
    },
    {
      "dynamics_loss_avg": 0.04989903401583433,
      "epoch": 4.913330078125e-05,
      "step": 51520
    },
    {
      "action_loss_avg": 0.003297355119138956,
      "epoch": 4.913330078125e-05,
      "step": 51520
    },
    {
      "epoch": 4.913330078125e-05,
      "step": 51520,
      "training_step_time": 8.391059637069702
    },
    {
      "epoch": 4.913425445556641e-05,
      "model_forward_time": 1.7462701797485352,
      "step": 51521
    },
    {
      "epoch": 4.913425445556641e-05,
      "step": 51521,
      "training_step_time": 7.6898298263549805
    },
    {
      "epoch": 4.913520812988281e-05,
      "model_forward_time": 1.7522709369659424,
      "step": 51522
    },
    {
      "epoch": 4.913520812988281e-05,
      "step": 51522,
      "training_step_time": 7.780693769454956
    },
    {
      "epoch": 4.913616180419922e-05,
      "model_forward_time": 1.7489368915557861,
      "step": 51523
    },
    {
      "epoch": 4.913616180419922e-05,
      "step": 51523,
      "training_step_time": 7.382994890213013
    },
    {
      "epoch": 4.9137115478515626e-05,
      "model_forward_time": 1.753525733947754,
      "step": 51524
    },
    {
      "epoch": 4.9137115478515626e-05,
      "step": 51524,
      "training_step_time": 8.397827386856079
    },
    {
      "epoch": 4.913806915283203e-05,
      "model_forward_time": 1.758007526397705,
      "step": 51525
    },
    {
      "epoch": 4.913806915283203e-05,
      "step": 51525,
      "training_step_time": 7.697020769119263
    },
    {
      "epoch": 4.913902282714844e-05,
      "model_forward_time": 1.7588560581207275,
      "step": 51526
    },
    {
      "epoch": 4.913902282714844e-05,
      "step": 51526,
      "training_step_time": 7.693925857543945
    },
    {
      "epoch": 4.9139976501464845e-05,
      "model_forward_time": 1.8153982162475586,
      "step": 51527
    },
    {
      "epoch": 4.9139976501464845e-05,
      "step": 51527,
      "training_step_time": 8.300391435623169
    },
    {
      "epoch": 4.914093017578125e-05,
      "model_forward_time": 1.754178762435913,
      "step": 51528
    },
    {
      "epoch": 4.914093017578125e-05,
      "step": 51528,
      "training_step_time": 8.752673864364624
    },
    {
      "epoch": 4.914188385009766e-05,
      "model_forward_time": 2.896719455718994,
      "step": 51529
    },
    {
      "epoch": 4.914188385009766e-05,
      "step": 51529,
      "training_step_time": 9.126450300216675
    },
    {
      "epoch": 4.9142837524414064e-05,
      "grad_norm": 0.09488257765769958,
      "learning_rate": 5.160524330493728e-06,
      "loss": 0.057,
      "step": 51530
    },
    {
      "epoch": 4.9142837524414064e-05,
      "model_forward_time": 1.758906602859497,
      "step": 51530
    },
    {
      "dynamics_loss_avg": 0.08302320688962936,
      "epoch": 4.9142837524414064e-05,
      "step": 51530
    },
    {
      "action_loss_avg": 0.0025388661597389728,
      "epoch": 4.9142837524414064e-05,
      "step": 51530
    },
    {
      "epoch": 4.9142837524414064e-05,
      "step": 51530,
      "training_step_time": 7.639495372772217
    },
    {
      "epoch": 4.914379119873047e-05,
      "model_forward_time": 1.7644648551940918,
      "step": 51531
    },
    {
      "epoch": 4.914379119873047e-05,
      "step": 51531,
      "training_step_time": 7.718102693557739
    },
    {
      "epoch": 4.914474487304687e-05,
      "model_forward_time": 1.4032270908355713,
      "step": 51532
    },
    {
      "epoch": 4.914474487304687e-05,
      "step": 51532,
      "training_step_time": 8.251415252685547
    },
    {
      "epoch": 4.914569854736328e-05,
      "model_forward_time": 1.750190019607544,
      "step": 51533
    },
    {
      "epoch": 4.914569854736328e-05,
      "step": 51533,
      "training_step_time": 8.537165403366089
    },
    {
      "epoch": 4.9146652221679686e-05,
      "model_forward_time": 1.7598841190338135,
      "step": 51534
    },
    {
      "epoch": 4.9146652221679686e-05,
      "step": 51534,
      "training_step_time": 7.553808927536011
    },
    {
      "epoch": 4.914760589599609e-05,
      "model_forward_time": 1.7585806846618652,
      "step": 51535
    },
    {
      "epoch": 4.914760589599609e-05,
      "step": 51535,
      "training_step_time": 7.608484745025635
    },
    {
      "epoch": 4.91485595703125e-05,
      "model_forward_time": 1.815246343612671,
      "step": 51536
    },
    {
      "epoch": 4.91485595703125e-05,
      "step": 51536,
      "training_step_time": 7.557218790054321
    },
    {
      "epoch": 4.9149513244628905e-05,
      "model_forward_time": 1.7625491619110107,
      "step": 51537
    },
    {
      "epoch": 4.9149513244628905e-05,
      "step": 51537,
      "training_step_time": 7.467797756195068
    },
    {
      "epoch": 4.915046691894531e-05,
      "model_forward_time": 2.8501455783843994,
      "step": 51538
    },
    {
      "epoch": 4.915046691894531e-05,
      "step": 51538,
      "training_step_time": 7.684053897857666
    },
    {
      "epoch": 4.915142059326172e-05,
      "model_forward_time": 1.7547783851623535,
      "step": 51539
    },
    {
      "epoch": 4.915142059326172e-05,
      "step": 51539,
      "training_step_time": 7.741950273513794
    },
    {
      "epoch": 4.9152374267578124e-05,
      "grad_norm": 0.07863230258226395,
      "learning_rate": 5.158871704282724e-06,
      "loss": 0.059,
      "step": 51540
    },
    {
      "epoch": 4.9152374267578124e-05,
      "model_forward_time": 1.759164571762085,
      "step": 51540
    },
    {
      "dynamics_loss_avg": 0.0729465913027525,
      "epoch": 4.9152374267578124e-05,
      "step": 51540
    },
    {
      "action_loss_avg": 0.0028562331979628654,
      "epoch": 4.9152374267578124e-05,
      "step": 51540
    },
    {
      "epoch": 4.9152374267578124e-05,
      "step": 51540,
      "training_step_time": 8.251188516616821
    },
    {
      "epoch": 4.915332794189453e-05,
      "model_forward_time": 1.7554931640625,
      "step": 51541
    },
    {
      "epoch": 4.915332794189453e-05,
      "step": 51541,
      "training_step_time": 7.5325117111206055
    },
    {
      "epoch": 4.915428161621094e-05,
      "model_forward_time": 1.7499299049377441,
      "step": 51542
    },
    {
      "epoch": 4.915428161621094e-05,
      "step": 51542,
      "training_step_time": 8.103845357894897
    },
    {
      "epoch": 4.915523529052734e-05,
      "model_forward_time": 1.7508938312530518,
      "step": 51543
    },
    {
      "epoch": 4.915523529052734e-05,
      "step": 51543,
      "training_step_time": 7.6003804206848145
    },
    {
      "epoch": 4.915618896484375e-05,
      "model_forward_time": 1.7694315910339355,
      "step": 51544
    },
    {
      "epoch": 4.915618896484375e-05,
      "step": 51544,
      "training_step_time": 7.6394922733306885
    },
    {
      "epoch": 4.915714263916016e-05,
      "model_forward_time": 1.8205993175506592,
      "step": 51545
    },
    {
      "epoch": 4.915714263916016e-05,
      "step": 51545,
      "training_step_time": 8.636689186096191
    },
    {
      "epoch": 4.915809631347656e-05,
      "model_forward_time": 1.744755506515503,
      "step": 51546
    },
    {
      "epoch": 4.915809631347656e-05,
      "step": 51546,
      "training_step_time": 8.395538091659546
    },
    {
      "epoch": 4.9159049987792966e-05,
      "model_forward_time": 1.7560667991638184,
      "step": 51547
    },
    {
      "epoch": 4.9159049987792966e-05,
      "step": 51547,
      "training_step_time": 8.673705339431763
    },
    {
      "epoch": 4.916000366210938e-05,
      "model_forward_time": 2.957263708114624,
      "step": 51548
    },
    {
      "epoch": 4.916000366210938e-05,
      "step": 51548,
      "training_step_time": 8.29376220703125
    },
    {
      "epoch": 4.916095733642578e-05,
      "model_forward_time": 1.750061273574829,
      "step": 51549
    },
    {
      "epoch": 4.916095733642578e-05,
      "step": 51549,
      "training_step_time": 7.717556476593018
    },
    {
      "epoch": 4.9161911010742185e-05,
      "grad_norm": 0.1054609939455986,
      "learning_rate": 5.15721906069775e-06,
      "loss": 0.0579,
      "step": 51550
    },
    {
      "epoch": 4.9161911010742185e-05,
      "model_forward_time": 1.7652626037597656,
      "step": 51550
    },
    {
      "dynamics_loss_avg": 0.05373234879225493,
      "epoch": 4.9161911010742185e-05,
      "step": 51550
    },
    {
      "action_loss_avg": 0.002930798748275265,
      "epoch": 4.9161911010742185e-05,
      "step": 51550
    },
    {
      "epoch": 4.9161911010742185e-05,
      "step": 51550,
      "training_step_time": 7.562293767929077
    },
    {
      "epoch": 4.9162864685058596e-05,
      "model_forward_time": 1.7450296878814697,
      "step": 51551
    },
    {
      "epoch": 4.9162864685058596e-05,
      "step": 51551,
      "training_step_time": 7.478480100631714
    },
    {
      "epoch": 4.9163818359375e-05,
      "model_forward_time": 1.756138563156128,
      "step": 51552
    },
    {
      "epoch": 4.9163818359375e-05,
      "step": 51552,
      "training_step_time": 7.623483657836914
    },
    {
      "epoch": 4.9164772033691404e-05,
      "model_forward_time": 1.7489910125732422,
      "step": 51553
    },
    {
      "epoch": 4.9164772033691404e-05,
      "step": 51553,
      "training_step_time": 7.607304573059082
    },
    {
      "epoch": 4.9165725708007814e-05,
      "model_forward_time": 1.822507381439209,
      "step": 51554
    },
    {
      "epoch": 4.9165725708007814e-05,
      "step": 51554,
      "training_step_time": 7.479903936386108
    },
    {
      "epoch": 4.916667938232422e-05,
      "model_forward_time": 1.7451753616333008,
      "step": 51555
    },
    {
      "epoch": 4.916667938232422e-05,
      "step": 51555,
      "training_step_time": 7.530372858047485
    },
    {
      "epoch": 4.916763305664062e-05,
      "model_forward_time": 1.7599601745605469,
      "step": 51556
    },
    {
      "epoch": 4.916763305664062e-05,
      "step": 51556,
      "training_step_time": 8.669026851654053
    },
    {
      "epoch": 4.916858673095703e-05,
      "model_forward_time": 2.9464573860168457,
      "step": 51557
    },
    {
      "epoch": 4.916858673095703e-05,
      "step": 51557,
      "training_step_time": 8.939933776855469
    },
    {
      "epoch": 4.916954040527344e-05,
      "model_forward_time": 1.7549786567687988,
      "step": 51558
    },
    {
      "epoch": 4.916954040527344e-05,
      "step": 51558,
      "training_step_time": 7.598167657852173
    },
    {
      "epoch": 4.917049407958984e-05,
      "model_forward_time": 1.7572917938232422,
      "step": 51559
    },
    {
      "epoch": 4.917049407958984e-05,
      "step": 51559,
      "training_step_time": 7.551436424255371
    },
    {
      "epoch": 4.917144775390625e-05,
      "grad_norm": 0.10948410630226135,
      "learning_rate": 5.155566399919536e-06,
      "loss": 0.0587,
      "step": 51560
    },
    {
      "epoch": 4.917144775390625e-05,
      "model_forward_time": 1.7525627613067627,
      "step": 51560
    },
    {
      "dynamics_loss_avg": 0.06728282030671835,
      "epoch": 4.917144775390625e-05,
      "step": 51560
    },
    {
      "action_loss_avg": 0.0014171645569149404,
      "epoch": 4.917144775390625e-05,
      "step": 51560
    },
    {
      "epoch": 4.917144775390625e-05,
      "step": 51560,
      "training_step_time": 7.868429183959961
    },
    {
      "epoch": 4.9172401428222656e-05,
      "model_forward_time": 1.7468328475952148,
      "step": 51561
    },
    {
      "epoch": 4.9172401428222656e-05,
      "step": 51561,
      "training_step_time": 7.649232625961304
    },
    {
      "epoch": 4.917335510253906e-05,
      "model_forward_time": 1.7693276405334473,
      "step": 51562
    },
    {
      "epoch": 4.917335510253906e-05,
      "step": 51562,
      "training_step_time": 7.581855773925781
    },
    {
      "epoch": 4.917430877685547e-05,
      "model_forward_time": 1.8211970329284668,
      "step": 51563
    },
    {
      "epoch": 4.917430877685547e-05,
      "step": 51563,
      "training_step_time": 7.474849462509155
    },
    {
      "epoch": 4.9175262451171875e-05,
      "model_forward_time": 1.767317771911621,
      "step": 51564
    },
    {
      "epoch": 4.9175262451171875e-05,
      "step": 51564,
      "training_step_time": 7.582064867019653
    },
    {
      "epoch": 4.917621612548828e-05,
      "model_forward_time": 1.7612383365631104,
      "step": 51565
    },
    {
      "epoch": 4.917621612548828e-05,
      "step": 51565,
      "training_step_time": 8.628573417663574
    },
    {
      "epoch": 4.917716979980469e-05,
      "model_forward_time": 2.887239933013916,
      "step": 51566
    },
    {
      "epoch": 4.917716979980469e-05,
      "step": 51566,
      "training_step_time": 7.692528486251831
    },
    {
      "epoch": 4.9178123474121094e-05,
      "model_forward_time": 1.7709252834320068,
      "step": 51567
    },
    {
      "epoch": 4.9178123474121094e-05,
      "step": 51567,
      "training_step_time": 7.616205930709839
    },
    {
      "epoch": 4.91790771484375e-05,
      "model_forward_time": 1.7649390697479248,
      "step": 51568
    },
    {
      "epoch": 4.91790771484375e-05,
      "step": 51568,
      "training_step_time": 7.660923719406128
    },
    {
      "epoch": 4.918003082275391e-05,
      "model_forward_time": 1.767733097076416,
      "step": 51569
    },
    {
      "epoch": 4.918003082275391e-05,
      "step": 51569,
      "training_step_time": 7.192229509353638
    },
    {
      "epoch": 4.918098449707031e-05,
      "grad_norm": 0.1456485539674759,
      "learning_rate": 5.153913722128813e-06,
      "loss": 0.0617,
      "step": 51570
    },
    {
      "epoch": 4.918098449707031e-05,
      "model_forward_time": 1.7675070762634277,
      "step": 51570
    },
    {
      "dynamics_loss_avg": 0.06151497270911932,
      "epoch": 4.918098449707031e-05,
      "step": 51570
    },
    {
      "action_loss_avg": 0.002458857439341955,
      "epoch": 4.918098449707031e-05,
      "step": 51570
    },
    {
      "epoch": 4.918098449707031e-05,
      "step": 51570,
      "training_step_time": 7.745062351226807
    },
    {
      "epoch": 4.918193817138672e-05,
      "model_forward_time": 1.772092342376709,
      "step": 51571
    },
    {
      "epoch": 4.918193817138672e-05,
      "step": 51571,
      "training_step_time": 7.672380447387695
    },
    {
      "epoch": 4.918289184570313e-05,
      "model_forward_time": 1.8372888565063477,
      "step": 51572
    },
    {
      "epoch": 4.918289184570313e-05,
      "step": 51572,
      "training_step_time": 7.534886837005615
    },
    {
      "epoch": 4.918384552001953e-05,
      "model_forward_time": 1.7633171081542969,
      "step": 51573
    },
    {
      "epoch": 4.918384552001953e-05,
      "step": 51573,
      "training_step_time": 7.591078519821167
    },
    {
      "epoch": 4.9184799194335936e-05,
      "model_forward_time": 1.7662758827209473,
      "step": 51574
    },
    {
      "epoch": 4.9184799194335936e-05,
      "step": 51574,
      "training_step_time": 7.612786293029785
    },
    {
      "epoch": 4.9185752868652346e-05,
      "model_forward_time": 2.892577886581421,
      "step": 51575
    },
    {
      "epoch": 4.9185752868652346e-05,
      "step": 51575,
      "training_step_time": 8.553219318389893
    },
    {
      "epoch": 4.918670654296875e-05,
      "model_forward_time": 1.7636382579803467,
      "step": 51576
    },
    {
      "epoch": 4.918670654296875e-05,
      "step": 51576,
      "training_step_time": 8.282260179519653
    },
    {
      "epoch": 4.9187660217285155e-05,
      "model_forward_time": 1.7787532806396484,
      "step": 51577
    },
    {
      "epoch": 4.9187660217285155e-05,
      "step": 51577,
      "training_step_time": 7.62878942489624
    },
    {
      "epoch": 4.9188613891601565e-05,
      "model_forward_time": 1.7699873447418213,
      "step": 51578
    },
    {
      "epoch": 4.9188613891601565e-05,
      "step": 51578,
      "training_step_time": 7.546578645706177
    },
    {
      "epoch": 4.918956756591797e-05,
      "model_forward_time": 1.7626476287841797,
      "step": 51579
    },
    {
      "epoch": 4.918956756591797e-05,
      "step": 51579,
      "training_step_time": 7.70419979095459
    },
    {
      "epoch": 4.919052124023437e-05,
      "grad_norm": 0.10600478947162628,
      "learning_rate": 5.152261027506318e-06,
      "loss": 0.0614,
      "step": 51580
    },
    {
      "epoch": 4.919052124023437e-05,
      "model_forward_time": 1.7660021781921387,
      "step": 51580
    },
    {
      "dynamics_loss_avg": 0.05331774987280369,
      "epoch": 4.919052124023437e-05,
      "step": 51580
    },
    {
      "action_loss_avg": 0.0015802299894858152,
      "epoch": 4.919052124023437e-05,
      "step": 51580
    },
    {
      "epoch": 4.919052124023437e-05,
      "step": 51580,
      "training_step_time": 7.606522083282471
    },
    {
      "epoch": 4.9191474914550784e-05,
      "model_forward_time": 1.8477835655212402,
      "step": 51581
    },
    {
      "epoch": 4.9191474914550784e-05,
      "step": 51581,
      "training_step_time": 7.99918270111084
    },
    {
      "epoch": 4.919242858886719e-05,
      "model_forward_time": 1.7726340293884277,
      "step": 51582
    },
    {
      "epoch": 4.919242858886719e-05,
      "step": 51582,
      "training_step_time": 7.503798246383667
    },
    {
      "epoch": 4.919338226318359e-05,
      "model_forward_time": 1.7648427486419678,
      "step": 51583
    },
    {
      "epoch": 4.919338226318359e-05,
      "step": 51583,
      "training_step_time": 7.670067071914673
    },
    {
      "epoch": 4.91943359375e-05,
      "model_forward_time": 2.8581337928771973,
      "step": 51584
    },
    {
      "epoch": 4.91943359375e-05,
      "step": 51584,
      "training_step_time": 9.177714109420776
    },
    {
      "epoch": 4.919528961181641e-05,
      "model_forward_time": 1.7683601379394531,
      "step": 51585
    },
    {
      "epoch": 4.919528961181641e-05,
      "step": 51585,
      "training_step_time": 8.861404418945312
    },
    {
      "epoch": 4.919624328613281e-05,
      "model_forward_time": 1.7741711139678955,
      "step": 51586
    },
    {
      "epoch": 4.919624328613281e-05,
      "step": 51586,
      "training_step_time": 7.702185153961182
    },
    {
      "epoch": 4.919719696044922e-05,
      "model_forward_time": 1.7798664569854736,
      "step": 51587
    },
    {
      "epoch": 4.919719696044922e-05,
      "step": 51587,
      "training_step_time": 7.603874921798706
    },
    {
      "epoch": 4.9198150634765626e-05,
      "model_forward_time": 1.7725167274475098,
      "step": 51588
    },
    {
      "epoch": 4.9198150634765626e-05,
      "step": 51588,
      "training_step_time": 7.669777154922485
    },
    {
      "epoch": 4.919910430908203e-05,
      "model_forward_time": 1.7744576930999756,
      "step": 51589
    },
    {
      "epoch": 4.919910430908203e-05,
      "step": 51589,
      "training_step_time": 8.476142644882202
    },
    {
      "epoch": 4.920005798339844e-05,
      "grad_norm": 0.11616850644350052,
      "learning_rate": 5.150608316232785e-06,
      "loss": 0.0585,
      "step": 51590
    },
    {
      "epoch": 4.920005798339844e-05,
      "model_forward_time": 1.8215153217315674,
      "step": 51590
    },
    {
      "dynamics_loss_avg": 0.0654384981840849,
      "epoch": 4.920005798339844e-05,
      "step": 51590
    },
    {
      "action_loss_avg": 0.0032484076800756156,
      "epoch": 4.920005798339844e-05,
      "step": 51590
    },
    {
      "epoch": 4.920005798339844e-05,
      "step": 51590,
      "training_step_time": 7.425449371337891
    },
    {
      "epoch": 4.9201011657714845e-05,
      "model_forward_time": 1.7648096084594727,
      "step": 51591
    },
    {
      "epoch": 4.9201011657714845e-05,
      "step": 51591,
      "training_step_time": 7.545291423797607
    },
    {
      "epoch": 4.920196533203125e-05,
      "model_forward_time": 1.7689149379730225,
      "step": 51592
    },
    {
      "epoch": 4.920196533203125e-05,
      "step": 51592,
      "training_step_time": 10.062612771987915
    },
    {
      "epoch": 4.920291900634766e-05,
      "model_forward_time": 1.7740230560302734,
      "step": 51593
    },
    {
      "epoch": 4.920291900634766e-05,
      "step": 51593,
      "training_step_time": 7.647373914718628
    },
    {
      "epoch": 4.9203872680664064e-05,
      "model_forward_time": 2.909113645553589,
      "step": 51594
    },
    {
      "epoch": 4.9203872680664064e-05,
      "step": 51594,
      "training_step_time": 7.6934263706207275
    },
    {
      "epoch": 4.920482635498047e-05,
      "model_forward_time": 1.7692523002624512,
      "step": 51595
    },
    {
      "epoch": 4.920482635498047e-05,
      "step": 51595,
      "training_step_time": 8.616073846817017
    },
    {
      "epoch": 4.920578002929687e-05,
      "model_forward_time": 1.7635586261749268,
      "step": 51596
    },
    {
      "epoch": 4.920578002929687e-05,
      "step": 51596,
      "training_step_time": 7.6586079597473145
    },
    {
      "epoch": 4.920673370361328e-05,
      "model_forward_time": 1.7698123455047607,
      "step": 51597
    },
    {
      "epoch": 4.920673370361328e-05,
      "step": 51597,
      "training_step_time": 7.60867714881897
    },
    {
      "epoch": 4.9207687377929687e-05,
      "model_forward_time": 1.770409107208252,
      "step": 51598
    },
    {
      "epoch": 4.9207687377929687e-05,
      "step": 51598,
      "training_step_time": 7.603441953659058
    },
    {
      "epoch": 4.920864105224609e-05,
      "model_forward_time": 1.8279078006744385,
      "step": 51599
    },
    {
      "epoch": 4.920864105224609e-05,
      "step": 51599,
      "training_step_time": 7.46195650100708
    },
    {
      "epoch": 4.92095947265625e-05,
      "grad_norm": 0.10613211989402771,
      "learning_rate": 5.148955588488955e-06,
      "loss": 0.0616,
      "step": 51600
    },
    {
      "epoch": 4.92095947265625e-05,
      "model_forward_time": 1.7624731063842773,
      "step": 51600
    },
    {
      "dynamics_loss_avg": 0.0715498710051179,
      "epoch": 4.92095947265625e-05,
      "step": 51600
    },
    {
      "action_loss_avg": 0.003290517147979699,
      "epoch": 4.92095947265625e-05,
      "step": 51600
    },
    {
      "epoch": 4.92095947265625e-05,
      "step": 51600,
      "training_step_time": 7.549008131027222
    },
    {
      "epoch": 4.9210548400878905e-05,
      "model_forward_time": 1.768089771270752,
      "step": 51601
    },
    {
      "epoch": 4.9210548400878905e-05,
      "step": 51601,
      "training_step_time": 8.434020042419434
    },
    {
      "epoch": 4.921150207519531e-05,
      "model_forward_time": 1.7629051208496094,
      "step": 51602
    },
    {
      "epoch": 4.921150207519531e-05,
      "step": 51602,
      "training_step_time": 7.763468503952026
    },
    {
      "epoch": 4.921245574951172e-05,
      "model_forward_time": 2.9698903560638428,
      "step": 51603
    },
    {
      "epoch": 4.921245574951172e-05,
      "step": 51603,
      "training_step_time": 10.120616436004639
    },
    {
      "epoch": 4.9213409423828124e-05,
      "model_forward_time": 1.763810634613037,
      "step": 51604
    },
    {
      "epoch": 4.9213409423828124e-05,
      "step": 51604,
      "training_step_time": 8.704137325286865
    },
    {
      "epoch": 4.921436309814453e-05,
      "model_forward_time": 1.7696194648742676,
      "step": 51605
    },
    {
      "epoch": 4.921436309814453e-05,
      "step": 51605,
      "training_step_time": 7.609873056411743
    },
    {
      "epoch": 4.921531677246094e-05,
      "model_forward_time": 1.7720966339111328,
      "step": 51606
    },
    {
      "epoch": 4.921531677246094e-05,
      "step": 51606,
      "training_step_time": 8.63943338394165
    },
    {
      "epoch": 4.921627044677734e-05,
      "model_forward_time": 1.766897439956665,
      "step": 51607
    },
    {
      "epoch": 4.921627044677734e-05,
      "step": 51607,
      "training_step_time": 7.48967719078064
    },
    {
      "epoch": 4.921722412109375e-05,
      "model_forward_time": 1.8300464153289795,
      "step": 51608
    },
    {
      "epoch": 4.921722412109375e-05,
      "step": 51608,
      "training_step_time": 9.20691728591919
    },
    {
      "epoch": 4.921817779541016e-05,
      "model_forward_time": 1.7777435779571533,
      "step": 51609
    },
    {
      "epoch": 4.921817779541016e-05,
      "step": 51609,
      "training_step_time": 8.278082132339478
    },
    {
      "epoch": 4.921913146972656e-05,
      "grad_norm": 0.1356232613325119,
      "learning_rate": 5.1473028444555635e-06,
      "loss": 0.0584,
      "step": 51610
    },
    {
      "epoch": 4.921913146972656e-05,
      "model_forward_time": 1.7434155941009521,
      "step": 51610
    },
    {
      "dynamics_loss_avg": 0.0666197357699275,
      "epoch": 4.921913146972656e-05,
      "step": 51610
    },
    {
      "action_loss_avg": 0.003266327996971086,
      "epoch": 4.921913146972656e-05,
      "step": 51610
    },
    {
      "epoch": 4.921913146972656e-05,
      "step": 51610,
      "training_step_time": 7.50102162361145
    },
    {
      "epoch": 4.9220085144042966e-05,
      "model_forward_time": 1.7509806156158447,
      "step": 51611
    },
    {
      "epoch": 4.9220085144042966e-05,
      "step": 51611,
      "training_step_time": 8.661890268325806
    },
    {
      "epoch": 4.922103881835938e-05,
      "model_forward_time": 2.900113582611084,
      "step": 51612
    },
    {
      "epoch": 4.922103881835938e-05,
      "step": 51612,
      "training_step_time": 8.525011777877808
    },
    {
      "epoch": 4.922199249267578e-05,
      "model_forward_time": 1.7758541107177734,
      "step": 51613
    },
    {
      "epoch": 4.922199249267578e-05,
      "step": 51613,
      "training_step_time": 9.080073833465576
    },
    {
      "epoch": 4.9222946166992185e-05,
      "model_forward_time": 1.764399766921997,
      "step": 51614
    },
    {
      "epoch": 4.9222946166992185e-05,
      "step": 51614,
      "training_step_time": 7.583644390106201
    },
    {
      "epoch": 4.9223899841308596e-05,
      "model_forward_time": 1.7670238018035889,
      "step": 51615
    },
    {
      "epoch": 4.9223899841308596e-05,
      "step": 51615,
      "training_step_time": 7.660584926605225
    },
    {
      "epoch": 4.9224853515625e-05,
      "model_forward_time": 1.7593169212341309,
      "step": 51616
    },
    {
      "epoch": 4.9224853515625e-05,
      "step": 51616,
      "training_step_time": 8.236635208129883
    },
    {
      "epoch": 4.9225807189941404e-05,
      "model_forward_time": 1.8287341594696045,
      "step": 51617
    },
    {
      "epoch": 4.9225807189941404e-05,
      "step": 51617,
      "training_step_time": 7.507655143737793
    },
    {
      "epoch": 4.9226760864257815e-05,
      "model_forward_time": 1.7564361095428467,
      "step": 51618
    },
    {
      "epoch": 4.9226760864257815e-05,
      "step": 51618,
      "training_step_time": 8.162007808685303
    },
    {
      "epoch": 4.922771453857422e-05,
      "model_forward_time": 1.762315273284912,
      "step": 51619
    },
    {
      "epoch": 4.922771453857422e-05,
      "step": 51619,
      "training_step_time": 9.161763429641724
    },
    {
      "epoch": 4.922866821289062e-05,
      "grad_norm": 0.13542722165584564,
      "learning_rate": 5.145650084313356e-06,
      "loss": 0.0601,
      "step": 51620
    },
    {
      "epoch": 4.922866821289062e-05,
      "model_forward_time": 1.7522380352020264,
      "step": 51620
    },
    {
      "dynamics_loss_avg": 0.06812679562717676,
      "epoch": 4.922866821289062e-05,
      "step": 51620
    },
    {
      "action_loss_avg": 0.0030260424013249577,
      "epoch": 4.922866821289062e-05,
      "step": 51620
    },
    {
      "epoch": 4.922866821289062e-05,
      "step": 51620,
      "training_step_time": 7.638677597045898
    },
    {
      "epoch": 4.922962188720703e-05,
      "model_forward_time": 2.875088691711426,
      "step": 51621
    },
    {
      "epoch": 4.922962188720703e-05,
      "step": 51621,
      "training_step_time": 7.702345848083496
    },
    {
      "epoch": 4.923057556152344e-05,
      "model_forward_time": 1.7629132270812988,
      "step": 51622
    },
    {
      "epoch": 4.923057556152344e-05,
      "step": 51622,
      "training_step_time": 7.684337139129639
    },
    {
      "epoch": 4.923152923583984e-05,
      "model_forward_time": 1.7592380046844482,
      "step": 51623
    },
    {
      "epoch": 4.923152923583984e-05,
      "step": 51623,
      "training_step_time": 7.629103660583496
    },
    {
      "epoch": 4.923248291015625e-05,
      "model_forward_time": 1.7530980110168457,
      "step": 51624
    },
    {
      "epoch": 4.923248291015625e-05,
      "step": 51624,
      "training_step_time": 7.66042685508728
    },
    {
      "epoch": 4.9233436584472656e-05,
      "model_forward_time": 1.7608015537261963,
      "step": 51625
    },
    {
      "epoch": 4.9233436584472656e-05,
      "step": 51625,
      "training_step_time": 7.560291290283203
    },
    {
      "epoch": 4.923439025878906e-05,
      "model_forward_time": 1.8167438507080078,
      "step": 51626
    },
    {
      "epoch": 4.923439025878906e-05,
      "step": 51626,
      "training_step_time": 7.622204065322876
    },
    {
      "epoch": 4.923534393310547e-05,
      "model_forward_time": 1.7523484230041504,
      "step": 51627
    },
    {
      "epoch": 4.923534393310547e-05,
      "step": 51627,
      "training_step_time": 7.423994064331055
    },
    {
      "epoch": 4.9236297607421875e-05,
      "model_forward_time": 1.7567009925842285,
      "step": 51628
    },
    {
      "epoch": 4.9236297607421875e-05,
      "step": 51628,
      "training_step_time": 8.532051801681519
    },
    {
      "epoch": 4.923725128173828e-05,
      "model_forward_time": 1.7626583576202393,
      "step": 51629
    },
    {
      "epoch": 4.923725128173828e-05,
      "step": 51629,
      "training_step_time": 8.141661405563354
    },
    {
      "epoch": 4.923820495605469e-05,
      "grad_norm": 0.1194954439997673,
      "learning_rate": 5.143997308243074e-06,
      "loss": 0.0612,
      "step": 51630
    },
    {
      "epoch": 4.923820495605469e-05,
      "model_forward_time": 2.8826510906219482,
      "step": 51630
    },
    {
      "dynamics_loss_avg": 0.04405550323426723,
      "epoch": 4.923820495605469e-05,
      "step": 51630
    },
    {
      "action_loss_avg": 0.002064101360156201,
      "epoch": 4.923820495605469e-05,
      "step": 51630
    },
    {
      "epoch": 4.923820495605469e-05,
      "step": 51630,
      "training_step_time": 8.552386283874512
    },
    {
      "epoch": 4.9239158630371094e-05,
      "model_forward_time": 1.7590906620025635,
      "step": 51631
    },
    {
      "epoch": 4.9239158630371094e-05,
      "step": 51631,
      "training_step_time": 7.671902418136597
    },
    {
      "epoch": 4.92401123046875e-05,
      "model_forward_time": 1.7564561367034912,
      "step": 51632
    },
    {
      "epoch": 4.92401123046875e-05,
      "step": 51632,
      "training_step_time": 7.697389602661133
    },
    {
      "epoch": 4.924106597900391e-05,
      "model_forward_time": 1.7611165046691895,
      "step": 51633
    },
    {
      "epoch": 4.924106597900391e-05,
      "step": 51633,
      "training_step_time": 9.724064111709595
    },
    {
      "epoch": 4.924201965332031e-05,
      "model_forward_time": 1.7661349773406982,
      "step": 51634
    },
    {
      "epoch": 4.924201965332031e-05,
      "step": 51634,
      "training_step_time": 8.271431922912598
    },
    {
      "epoch": 4.924297332763672e-05,
      "model_forward_time": 1.8318495750427246,
      "step": 51635
    },
    {
      "epoch": 4.924297332763672e-05,
      "step": 51635,
      "training_step_time": 8.448409795761108
    },
    {
      "epoch": 4.924392700195313e-05,
      "model_forward_time": 1.7668628692626953,
      "step": 51636
    },
    {
      "epoch": 4.924392700195313e-05,
      "step": 51636,
      "training_step_time": 7.413950681686401
    },
    {
      "epoch": 4.924488067626953e-05,
      "model_forward_time": 1.762937307357788,
      "step": 51637
    },
    {
      "epoch": 4.924488067626953e-05,
      "step": 51637,
      "training_step_time": 7.582432985305786
    },
    {
      "epoch": 4.9245834350585936e-05,
      "model_forward_time": 1.7620048522949219,
      "step": 51638
    },
    {
      "epoch": 4.9245834350585936e-05,
      "step": 51638,
      "training_step_time": 8.144096851348877
    },
    {
      "epoch": 4.9246788024902347e-05,
      "model_forward_time": 1.7716388702392578,
      "step": 51639
    },
    {
      "epoch": 4.9246788024902347e-05,
      "step": 51639,
      "training_step_time": 8.28998064994812
    },
    {
      "epoch": 4.924774169921875e-05,
      "grad_norm": 0.11381812393665314,
      "learning_rate": 5.142344516425461e-06,
      "loss": 0.062,
      "step": 51640
    },
    {
      "epoch": 4.924774169921875e-05,
      "model_forward_time": 1.7667036056518555,
      "step": 51640
    },
    {
      "dynamics_loss_avg": 0.05900550745427609,
      "epoch": 4.924774169921875e-05,
      "step": 51640
    },
    {
      "action_loss_avg": 0.0014600697002606466,
      "epoch": 4.924774169921875e-05,
      "step": 51640
    },
    {
      "epoch": 4.924774169921875e-05,
      "step": 51640,
      "training_step_time": 8.748265981674194
    },
    {
      "epoch": 4.9248695373535155e-05,
      "model_forward_time": 1.7772982120513916,
      "step": 51641
    },
    {
      "epoch": 4.9248695373535155e-05,
      "step": 51641,
      "training_step_time": 7.615801095962524
    },
    {
      "epoch": 4.9249649047851565e-05,
      "model_forward_time": 1.772132396697998,
      "step": 51642
    },
    {
      "epoch": 4.9249649047851565e-05,
      "step": 51642,
      "training_step_time": 7.611371994018555
    },
    {
      "epoch": 4.925060272216797e-05,
      "model_forward_time": 1.7724485397338867,
      "step": 51643
    },
    {
      "epoch": 4.925060272216797e-05,
      "step": 51643,
      "training_step_time": 7.618525981903076
    },
    {
      "epoch": 4.9251556396484373e-05,
      "model_forward_time": 1.8051800727844238,
      "step": 51644
    },
    {
      "epoch": 4.9251556396484373e-05,
      "step": 51644,
      "training_step_time": 7.608380556106567
    },
    {
      "epoch": 4.9252510070800784e-05,
      "model_forward_time": 1.7730505466461182,
      "step": 51645
    },
    {
      "epoch": 4.9252510070800784e-05,
      "step": 51645,
      "training_step_time": 7.3784019947052
    },
    {
      "epoch": 4.925346374511719e-05,
      "model_forward_time": 1.7658913135528564,
      "step": 51646
    },
    {
      "epoch": 4.925346374511719e-05,
      "step": 51646,
      "training_step_time": 8.608972311019897
    },
    {
      "epoch": 4.925441741943359e-05,
      "model_forward_time": 1.7674000263214111,
      "step": 51647
    },
    {
      "epoch": 4.925441741943359e-05,
      "step": 51647,
      "training_step_time": 7.633220672607422
    },
    {
      "epoch": 4.925537109375e-05,
      "model_forward_time": 1.772087812423706,
      "step": 51648
    },
    {
      "epoch": 4.925537109375e-05,
      "step": 51648,
      "training_step_time": 7.781395435333252
    },
    {
      "epoch": 4.925632476806641e-05,
      "model_forward_time": 2.574885368347168,
      "step": 51649
    },
    {
      "epoch": 4.925632476806641e-05,
      "step": 51649,
      "training_step_time": 7.638457775115967
    },
    {
      "epoch": 4.925727844238281e-05,
      "grad_norm": 0.10033906251192093,
      "learning_rate": 5.1406917090412675e-06,
      "loss": 0.0581,
      "step": 51650
    },
    {
      "epoch": 4.925727844238281e-05,
      "model_forward_time": 1.7622549533843994,
      "step": 51650
    },
    {
      "dynamics_loss_avg": 0.05347295897081494,
      "epoch": 4.925727844238281e-05,
      "step": 51650
    },
    {
      "action_loss_avg": 0.0030415477929636838,
      "epoch": 4.925727844238281e-05,
      "step": 51650
    },
    {
      "epoch": 4.925727844238281e-05,
      "step": 51650,
      "training_step_time": 9.508231163024902
    },
    {
      "epoch": 4.925823211669922e-05,
      "model_forward_time": 1.7744131088256836,
      "step": 51651
    },
    {
      "epoch": 4.925823211669922e-05,
      "step": 51651,
      "training_step_time": 7.606600284576416
    },
    {
      "epoch": 4.9259185791015626e-05,
      "model_forward_time": 1.7737135887145996,
      "step": 51652
    },
    {
      "epoch": 4.9259185791015626e-05,
      "step": 51652,
      "training_step_time": 7.590822219848633
    },
    {
      "epoch": 4.926013946533203e-05,
      "model_forward_time": 1.7740962505340576,
      "step": 51653
    },
    {
      "epoch": 4.926013946533203e-05,
      "step": 51653,
      "training_step_time": 7.598725080490112
    },
    {
      "epoch": 4.926109313964844e-05,
      "model_forward_time": 1.7690176963806152,
      "step": 51654
    },
    {
      "epoch": 4.926109313964844e-05,
      "step": 51654,
      "training_step_time": 7.696070671081543
    },
    {
      "epoch": 4.9262046813964845e-05,
      "model_forward_time": 1.7718889713287354,
      "step": 51655
    },
    {
      "epoch": 4.9262046813964845e-05,
      "step": 51655,
      "training_step_time": 7.544321298599243
    },
    {
      "epoch": 4.926300048828125e-05,
      "model_forward_time": 1.765108585357666,
      "step": 51656
    },
    {
      "epoch": 4.926300048828125e-05,
      "step": 51656,
      "training_step_time": 8.59747862815857
    },
    {
      "epoch": 4.926395416259765e-05,
      "model_forward_time": 1.7659022808074951,
      "step": 51657
    },
    {
      "epoch": 4.926395416259765e-05,
      "step": 51657,
      "training_step_time": 7.666789293289185
    },
    {
      "epoch": 4.9264907836914064e-05,
      "model_forward_time": 2.928529739379883,
      "step": 51658
    },
    {
      "epoch": 4.9264907836914064e-05,
      "step": 51658,
      "training_step_time": 7.702124118804932
    },
    {
      "epoch": 4.926586151123047e-05,
      "model_forward_time": 1.7746217250823975,
      "step": 51659
    },
    {
      "epoch": 4.926586151123047e-05,
      "step": 51659,
      "training_step_time": 8.615210771560669
    },
    {
      "epoch": 4.926681518554687e-05,
      "grad_norm": 0.09726937115192413,
      "learning_rate": 5.13903888627124e-06,
      "loss": 0.0572,
      "step": 51660
    },
    {
      "epoch": 4.926681518554687e-05,
      "model_forward_time": 1.7797391414642334,
      "step": 51660
    },
    {
      "dynamics_loss_avg": 0.06058464795351028,
      "epoch": 4.926681518554687e-05,
      "step": 51660
    },
    {
      "action_loss_avg": 0.0020063008676515894,
      "epoch": 4.926681518554687e-05,
      "step": 51660
    },
    {
      "epoch": 4.926681518554687e-05,
      "step": 51660,
      "training_step_time": 7.723202705383301
    },
    {
      "epoch": 4.926776885986328e-05,
      "model_forward_time": 1.7744624614715576,
      "step": 51661
    },
    {
      "epoch": 4.926776885986328e-05,
      "step": 51661,
      "training_step_time": 7.619805812835693
    },
    {
      "epoch": 4.9268722534179687e-05,
      "model_forward_time": 1.7728545665740967,
      "step": 51662
    },
    {
      "epoch": 4.9268722534179687e-05,
      "step": 51662,
      "training_step_time": 7.603551149368286
    },
    {
      "epoch": 4.926967620849609e-05,
      "model_forward_time": 1.8317525386810303,
      "step": 51663
    },
    {
      "epoch": 4.926967620849609e-05,
      "step": 51663,
      "training_step_time": 7.720866441726685
    },
    {
      "epoch": 4.92706298828125e-05,
      "model_forward_time": 1.7764713764190674,
      "step": 51664
    },
    {
      "epoch": 4.92706298828125e-05,
      "step": 51664,
      "training_step_time": 8.457592248916626
    },
    {
      "epoch": 4.9271583557128905e-05,
      "model_forward_time": 1.7691004276275635,
      "step": 51665
    },
    {
      "epoch": 4.9271583557128905e-05,
      "step": 51665,
      "training_step_time": 7.593210220336914
    },
    {
      "epoch": 4.927253723144531e-05,
      "model_forward_time": 1.7817461490631104,
      "step": 51666
    },
    {
      "epoch": 4.927253723144531e-05,
      "step": 51666,
      "training_step_time": 7.679793357849121
    },
    {
      "epoch": 4.927349090576172e-05,
      "model_forward_time": 2.8851895332336426,
      "step": 51667
    },
    {
      "epoch": 4.927349090576172e-05,
      "step": 51667,
      "training_step_time": 7.666755676269531
    },
    {
      "epoch": 4.9274444580078124e-05,
      "model_forward_time": 1.9500625133514404,
      "step": 51668
    },
    {
      "epoch": 4.9274444580078124e-05,
      "step": 51668,
      "training_step_time": 7.801879167556763
    },
    {
      "epoch": 4.927539825439453e-05,
      "model_forward_time": 1.9390184879302979,
      "step": 51669
    },
    {
      "epoch": 4.927539825439453e-05,
      "step": 51669,
      "training_step_time": 8.731825590133667
    },
    {
      "epoch": 4.927635192871094e-05,
      "grad_norm": 0.10464329272508621,
      "learning_rate": 5.137386048296129e-06,
      "loss": 0.0581,
      "step": 51670
    },
    {
      "epoch": 4.927635192871094e-05,
      "model_forward_time": 1.9329214096069336,
      "step": 51670
    },
    {
      "dynamics_loss_avg": 0.05490293875336647,
      "epoch": 4.927635192871094e-05,
      "step": 51670
    },
    {
      "action_loss_avg": 0.003019136912189424,
      "epoch": 4.927635192871094e-05,
      "step": 51670
    },
    {
      "epoch": 4.927635192871094e-05,
      "step": 51670,
      "training_step_time": 7.5890443325042725
    },
    {
      "epoch": 4.927730560302734e-05,
      "model_forward_time": 1.9268150329589844,
      "step": 51671
    },
    {
      "epoch": 4.927730560302734e-05,
      "step": 51671,
      "training_step_time": 8.33536696434021
    },
    {
      "epoch": 4.927825927734375e-05,
      "model_forward_time": 1.9109716415405273,
      "step": 51672
    },
    {
      "epoch": 4.927825927734375e-05,
      "step": 51672,
      "training_step_time": 7.695653676986694
    },
    {
      "epoch": 4.927921295166016e-05,
      "model_forward_time": 1.9405415058135986,
      "step": 51673
    },
    {
      "epoch": 4.927921295166016e-05,
      "step": 51673,
      "training_step_time": 7.547377347946167
    },
    {
      "epoch": 4.928016662597656e-05,
      "model_forward_time": 1.923166275024414,
      "step": 51674
    },
    {
      "epoch": 4.928016662597656e-05,
      "step": 51674,
      "training_step_time": 9.245006799697876
    },
    {
      "epoch": 4.9281120300292966e-05,
      "model_forward_time": 1.935431718826294,
      "step": 51675
    },
    {
      "epoch": 4.9281120300292966e-05,
      "step": 51675,
      "training_step_time": 9.45359992980957
    },
    {
      "epoch": 4.928207397460938e-05,
      "model_forward_time": 3.029221296310425,
      "step": 51676
    },
    {
      "epoch": 4.928207397460938e-05,
      "step": 51676,
      "training_step_time": 8.803018569946289
    },
    {
      "epoch": 4.928302764892578e-05,
      "model_forward_time": 1.9192118644714355,
      "step": 51677
    },
    {
      "epoch": 4.928302764892578e-05,
      "step": 51677,
      "training_step_time": 7.76913046836853
    },
    {
      "epoch": 4.9283981323242185e-05,
      "model_forward_time": 1.9134340286254883,
      "step": 51678
    },
    {
      "epoch": 4.9283981323242185e-05,
      "step": 51678,
      "training_step_time": 7.710419416427612
    },
    {
      "epoch": 4.9284934997558596e-05,
      "model_forward_time": 1.8991520404815674,
      "step": 51679
    },
    {
      "epoch": 4.9284934997558596e-05,
      "step": 51679,
      "training_step_time": 7.501515626907349
    },
    {
      "epoch": 4.9285888671875e-05,
      "grad_norm": 0.10166992247104645,
      "learning_rate": 5.135733195296686e-06,
      "loss": 0.0585,
      "step": 51680
    },
    {
      "epoch": 4.9285888671875e-05,
      "model_forward_time": 1.9412541389465332,
      "step": 51680
    },
    {
      "dynamics_loss_avg": 0.08434128612279893,
      "epoch": 4.9285888671875e-05,
      "step": 51680
    },
    {
      "action_loss_avg": 0.0042245887103490535,
      "epoch": 4.9285888671875e-05,
      "step": 51680
    },
    {
      "epoch": 4.9285888671875e-05,
      "step": 51680,
      "training_step_time": 7.688822507858276
    },
    {
      "epoch": 4.9286842346191404e-05,
      "model_forward_time": 1.9235546588897705,
      "step": 51681
    },
    {
      "epoch": 4.9286842346191404e-05,
      "step": 51681,
      "training_step_time": 7.589801073074341
    },
    {
      "epoch": 4.9287796020507815e-05,
      "model_forward_time": 1.9336974620819092,
      "step": 51682
    },
    {
      "epoch": 4.9287796020507815e-05,
      "step": 51682,
      "training_step_time": 7.483334541320801
    },
    {
      "epoch": 4.928874969482422e-05,
      "model_forward_time": 1.930661916732788,
      "step": 51683
    },
    {
      "epoch": 4.928874969482422e-05,
      "step": 51683,
      "training_step_time": 7.527181386947632
    },
    {
      "epoch": 4.928970336914062e-05,
      "model_forward_time": 1.9477899074554443,
      "step": 51684
    },
    {
      "epoch": 4.928970336914062e-05,
      "step": 51684,
      "training_step_time": 7.6265504360198975
    },
    {
      "epoch": 4.9290657043457033e-05,
      "model_forward_time": 1.935828685760498,
      "step": 51685
    },
    {
      "epoch": 4.9290657043457033e-05,
      "step": 51685,
      "training_step_time": 8.966845989227295
    },
    {
      "epoch": 4.929161071777344e-05,
      "model_forward_time": 2.9960663318634033,
      "step": 51686
    },
    {
      "epoch": 4.929161071777344e-05,
      "step": 51686,
      "training_step_time": 7.765208721160889
    },
    {
      "epoch": 4.929256439208984e-05,
      "model_forward_time": 1.914703369140625,
      "step": 51687
    },
    {
      "epoch": 4.929256439208984e-05,
      "step": 51687,
      "training_step_time": 7.707547664642334
    },
    {
      "epoch": 4.929351806640625e-05,
      "model_forward_time": 1.8875305652618408,
      "step": 51688
    },
    {
      "epoch": 4.929351806640625e-05,
      "step": 51688,
      "training_step_time": 8.532964706420898
    },
    {
      "epoch": 4.9294471740722656e-05,
      "model_forward_time": 1.9452953338623047,
      "step": 51689
    },
    {
      "epoch": 4.9294471740722656e-05,
      "step": 51689,
      "training_step_time": 8.628598928451538
    },
    {
      "epoch": 4.929542541503906e-05,
      "grad_norm": 0.12695352733135223,
      "learning_rate": 5.134080327453664e-06,
      "loss": 0.0596,
      "step": 51690
    },
    {
      "epoch": 4.929542541503906e-05,
      "model_forward_time": 1.9293286800384521,
      "step": 51690
    },
    {
      "dynamics_loss_avg": 0.06910132560878993,
      "epoch": 4.929542541503906e-05,
      "step": 51690
    },
    {
      "action_loss_avg": 0.0030325723040732556,
      "epoch": 4.929542541503906e-05,
      "step": 51690
    },
    {
      "epoch": 4.929542541503906e-05,
      "step": 51690,
      "training_step_time": 6.77809739112854
    },
    {
      "epoch": 4.929637908935547e-05,
      "model_forward_time": 1.9016945362091064,
      "step": 51691
    },
    {
      "epoch": 4.929637908935547e-05,
      "step": 51691,
      "training_step_time": 6.764737606048584
    },
    {
      "epoch": 4.9297332763671875e-05,
      "model_forward_time": 1.8757648468017578,
      "step": 51692
    },
    {
      "epoch": 4.9297332763671875e-05,
      "step": 51692,
      "training_step_time": 6.6950483322143555
    },
    {
      "epoch": 4.929828643798828e-05,
      "model_forward_time": 1.9380056858062744,
      "step": 51693
    },
    {
      "epoch": 4.929828643798828e-05,
      "step": 51693,
      "training_step_time": 6.768961191177368
    },
    {
      "epoch": 4.929924011230469e-05,
      "model_forward_time": 1.908703327178955,
      "step": 51694
    },
    {
      "epoch": 4.929924011230469e-05,
      "step": 51694,
      "training_step_time": 6.731504678726196
    },
    {
      "epoch": 4.9300193786621094e-05,
      "model_forward_time": 1.9216721057891846,
      "step": 51695
    },
    {
      "epoch": 4.9300193786621094e-05,
      "step": 51695,
      "training_step_time": 6.7530083656311035
    },
    {
      "epoch": 4.93011474609375e-05,
      "model_forward_time": 1.9061133861541748,
      "step": 51696
    },
    {
      "epoch": 4.93011474609375e-05,
      "step": 51696,
      "training_step_time": 6.770649671554565
    },
    {
      "epoch": 4.930210113525391e-05,
      "model_forward_time": 1.9165823459625244,
      "step": 51697
    },
    {
      "epoch": 4.930210113525391e-05,
      "step": 51697,
      "training_step_time": 6.783663034439087
    },
    {
      "epoch": 4.930305480957031e-05,
      "model_forward_time": 3.004413366317749,
      "step": 51698
    },
    {
      "epoch": 4.930305480957031e-05,
      "step": 51698,
      "training_step_time": 8.354644536972046
    },
    {
      "epoch": 4.930400848388672e-05,
      "model_forward_time": 1.9203453063964844,
      "step": 51699
    },
    {
      "epoch": 4.930400848388672e-05,
      "step": 51699,
      "training_step_time": 7.7647316455841064
    },
    {
      "epoch": 4.930496215820313e-05,
      "grad_norm": 0.2281980961561203,
      "learning_rate": 5.132427444947821e-06,
      "loss": 0.0561,
      "step": 51700
    },
    {
      "epoch": 4.930496215820313e-05,
      "model_forward_time": 1.920565128326416,
      "step": 51700
    },
    {
      "dynamics_loss_avg": 0.0579437455162406,
      "epoch": 4.930496215820313e-05,
      "step": 51700
    },
    {
      "action_loss_avg": 0.003131205873796716,
      "epoch": 4.930496215820313e-05,
      "step": 51700
    },
    {
      "epoch": 4.930496215820313e-05,
      "step": 51700,
      "training_step_time": 6.7839131355285645
    },
    {
      "epoch": 4.930591583251953e-05,
      "model_forward_time": 1.9179415702819824,
      "step": 51701
    },
    {
      "epoch": 4.930591583251953e-05,
      "step": 51701,
      "training_step_time": 6.762378692626953
    },
    {
      "epoch": 4.9306869506835936e-05,
      "model_forward_time": 1.9034645557403564,
      "step": 51702
    },
    {
      "epoch": 4.9306869506835936e-05,
      "step": 51702,
      "training_step_time": 6.797656059265137
    },
    {
      "epoch": 4.9307823181152347e-05,
      "model_forward_time": 1.9426767826080322,
      "step": 51703
    },
    {
      "epoch": 4.9307823181152347e-05,
      "step": 51703,
      "training_step_time": 6.783764839172363
    },
    {
      "epoch": 4.930877685546875e-05,
      "model_forward_time": 1.9235985279083252,
      "step": 51704
    },
    {
      "epoch": 4.930877685546875e-05,
      "step": 51704,
      "training_step_time": 6.786332607269287
    },
    {
      "epoch": 4.9309730529785155e-05,
      "model_forward_time": 1.8971240520477295,
      "step": 51705
    },
    {
      "epoch": 4.9309730529785155e-05,
      "step": 51705,
      "training_step_time": 6.780216693878174
    },
    {
      "epoch": 4.9310684204101565e-05,
      "model_forward_time": 1.94203782081604,
      "step": 51706
    },
    {
      "epoch": 4.9310684204101565e-05,
      "step": 51706,
      "training_step_time": 6.810583829879761
    },
    {
      "epoch": 4.931163787841797e-05,
      "model_forward_time": 3.055032968521118,
      "step": 51707
    },
    {
      "epoch": 4.931163787841797e-05,
      "step": 51707,
      "training_step_time": 8.362736463546753
    },
    {
      "epoch": 4.9312591552734373e-05,
      "model_forward_time": 1.9368128776550293,
      "step": 51708
    },
    {
      "epoch": 4.9312591552734373e-05,
      "step": 51708,
      "training_step_time": 8.867932081222534
    },
    {
      "epoch": 4.9313545227050784e-05,
      "model_forward_time": 1.9155690670013428,
      "step": 51709
    },
    {
      "epoch": 4.9313545227050784e-05,
      "step": 51709,
      "training_step_time": 6.783013582229614
    },
    {
      "epoch": 4.931449890136719e-05,
      "grad_norm": 0.10637661069631577,
      "learning_rate": 5.130774547959909e-06,
      "loss": 0.0551,
      "step": 51710
    },
    {
      "epoch": 4.931449890136719e-05,
      "model_forward_time": 1.9277610778808594,
      "step": 51710
    },
    {
      "dynamics_loss_avg": 0.05376816419884563,
      "epoch": 4.931449890136719e-05,
      "step": 51710
    },
    {
      "action_loss_avg": 0.001686624699505046,
      "epoch": 4.931449890136719e-05,
      "step": 51710
    },
    {
      "epoch": 4.931449890136719e-05,
      "step": 51710,
      "training_step_time": 6.799214601516724
    },
    {
      "epoch": 4.931545257568359e-05,
      "model_forward_time": 1.9087860584259033,
      "step": 51711
    },
    {
      "epoch": 4.931545257568359e-05,
      "step": 51711,
      "training_step_time": 6.7677459716796875
    },
    {
      "epoch": 4.931640625e-05,
      "model_forward_time": 1.9163670539855957,
      "step": 51712
    },
    {
      "epoch": 4.931640625e-05,
      "step": 51712,
      "training_step_time": 6.79253077507019
    },
    {
      "epoch": 4.931735992431641e-05,
      "model_forward_time": 1.9338176250457764,
      "step": 51713
    },
    {
      "epoch": 4.931735992431641e-05,
      "step": 51713,
      "training_step_time": 6.870544195175171
    },
    {
      "epoch": 4.931831359863281e-05,
      "model_forward_time": 1.938737392425537,
      "step": 51714
    },
    {
      "epoch": 4.931831359863281e-05,
      "step": 51714,
      "training_step_time": 6.842667102813721
    },
    {
      "epoch": 4.931926727294922e-05,
      "model_forward_time": 1.903428316116333,
      "step": 51715
    },
    {
      "epoch": 4.931926727294922e-05,
      "step": 51715,
      "training_step_time": 6.707616806030273
    },
    {
      "epoch": 4.9320220947265626e-05,
      "model_forward_time": 3.016794204711914,
      "step": 51716
    },
    {
      "epoch": 4.9320220947265626e-05,
      "step": 51716,
      "training_step_time": 7.768895864486694
    },
    {
      "epoch": 4.932117462158203e-05,
      "model_forward_time": 1.9423527717590332,
      "step": 51717
    },
    {
      "epoch": 4.932117462158203e-05,
      "step": 51717,
      "training_step_time": 8.823347568511963
    },
    {
      "epoch": 4.932212829589844e-05,
      "model_forward_time": 1.9350757598876953,
      "step": 51718
    },
    {
      "epoch": 4.932212829589844e-05,
      "step": 51718,
      "training_step_time": 7.395228385925293
    },
    {
      "epoch": 4.9323081970214845e-05,
      "model_forward_time": 1.9274842739105225,
      "step": 51719
    },
    {
      "epoch": 4.9323081970214845e-05,
      "step": 51719,
      "training_step_time": 6.776078939437866
    },
    {
      "epoch": 4.932403564453125e-05,
      "grad_norm": 0.1497914046049118,
      "learning_rate": 5.1291216366706916e-06,
      "loss": 0.0586,
      "step": 51720
    },
    {
      "epoch": 4.932403564453125e-05,
      "model_forward_time": 1.9382460117340088,
      "step": 51720
    },
    {
      "dynamics_loss_avg": 0.04963619085028768,
      "epoch": 4.932403564453125e-05,
      "step": 51720
    },
    {
      "action_loss_avg": 0.0016115612263092772,
      "epoch": 4.932403564453125e-05,
      "step": 51720
    },
    {
      "epoch": 4.932403564453125e-05,
      "step": 51720,
      "training_step_time": 6.8059797286987305
    },
    {
      "epoch": 4.932498931884765e-05,
      "model_forward_time": 1.9208171367645264,
      "step": 51721
    },
    {
      "epoch": 4.932498931884765e-05,
      "step": 51721,
      "training_step_time": 6.739321231842041
    },
    {
      "epoch": 4.9325942993164064e-05,
      "model_forward_time": 1.9072115421295166,
      "step": 51722
    },
    {
      "epoch": 4.9325942993164064e-05,
      "step": 51722,
      "training_step_time": 6.845660924911499
    },
    {
      "epoch": 4.932689666748047e-05,
      "model_forward_time": 1.9051029682159424,
      "step": 51723
    },
    {
      "epoch": 4.932689666748047e-05,
      "step": 51723,
      "training_step_time": 6.795575857162476
    },
    {
      "epoch": 4.932785034179687e-05,
      "model_forward_time": 1.9264039993286133,
      "step": 51724
    },
    {
      "epoch": 4.932785034179687e-05,
      "step": 51724,
      "training_step_time": 6.774758577346802
    },
    {
      "epoch": 4.932880401611328e-05,
      "model_forward_time": 3.0284414291381836,
      "step": 51725
    },
    {
      "epoch": 4.932880401611328e-05,
      "step": 51725,
      "training_step_time": 7.838988780975342
    },
    {
      "epoch": 4.932975769042969e-05,
      "model_forward_time": 1.8908255100250244,
      "step": 51726
    },
    {
      "epoch": 4.932975769042969e-05,
      "step": 51726,
      "training_step_time": 9.372034788131714
    },
    {
      "epoch": 4.933071136474609e-05,
      "model_forward_time": 1.8980963230133057,
      "step": 51727
    },
    {
      "epoch": 4.933071136474609e-05,
      "step": 51727,
      "training_step_time": 7.8062050342559814
    },
    {
      "epoch": 4.93316650390625e-05,
      "model_forward_time": 1.9109869003295898,
      "step": 51728
    },
    {
      "epoch": 4.93316650390625e-05,
      "step": 51728,
      "training_step_time": 6.801548957824707
    },
    {
      "epoch": 4.9332618713378905e-05,
      "model_forward_time": 1.9173803329467773,
      "step": 51729
    },
    {
      "epoch": 4.9332618713378905e-05,
      "step": 51729,
      "training_step_time": 6.843694686889648
    },
    {
      "epoch": 4.933357238769531e-05,
      "grad_norm": 0.12945674359798431,
      "learning_rate": 5.127468711260924e-06,
      "loss": 0.0576,
      "step": 51730
    },
    {
      "epoch": 4.933357238769531e-05,
      "model_forward_time": 1.9314196109771729,
      "step": 51730
    },
    {
      "dynamics_loss_avg": 0.05839965119957924,
      "epoch": 4.933357238769531e-05,
      "step": 51730
    },
    {
      "action_loss_avg": 0.0033113417157437652,
      "epoch": 4.933357238769531e-05,
      "step": 51730
    },
    {
      "epoch": 4.933357238769531e-05,
      "step": 51730,
      "training_step_time": 6.87888503074646
    },
    {
      "epoch": 4.933452606201172e-05,
      "model_forward_time": 1.9050099849700928,
      "step": 51731
    },
    {
      "epoch": 4.933452606201172e-05,
      "step": 51731,
      "training_step_time": 6.820014953613281
    },
    {
      "epoch": 4.9335479736328124e-05,
      "model_forward_time": 1.9317920207977295,
      "step": 51732
    },
    {
      "epoch": 4.9335479736328124e-05,
      "step": 51732,
      "training_step_time": 6.78397274017334
    },
    {
      "epoch": 4.933643341064453e-05,
      "model_forward_time": 1.9212610721588135,
      "step": 51733
    },
    {
      "epoch": 4.933643341064453e-05,
      "step": 51733,
      "training_step_time": 6.828132152557373
    },
    {
      "epoch": 4.933738708496094e-05,
      "model_forward_time": 3.0517373085021973,
      "step": 51734
    },
    {
      "epoch": 4.933738708496094e-05,
      "step": 51734,
      "training_step_time": 7.8503172397613525
    },
    {
      "epoch": 4.933834075927734e-05,
      "model_forward_time": 1.9033739566802979,
      "step": 51735
    },
    {
      "epoch": 4.933834075927734e-05,
      "step": 51735,
      "training_step_time": 9.232284784317017
    },
    {
      "epoch": 4.933929443359375e-05,
      "model_forward_time": 1.916337013244629,
      "step": 51736
    },
    {
      "epoch": 4.933929443359375e-05,
      "step": 51736,
      "training_step_time": 8.784157276153564
    },
    {
      "epoch": 4.934024810791016e-05,
      "model_forward_time": 1.925525426864624,
      "step": 51737
    },
    {
      "epoch": 4.934024810791016e-05,
      "step": 51737,
      "training_step_time": 7.435967206954956
    },
    {
      "epoch": 4.934120178222656e-05,
      "model_forward_time": 1.918722152709961,
      "step": 51738
    },
    {
      "epoch": 4.934120178222656e-05,
      "step": 51738,
      "training_step_time": 6.824902296066284
    },
    {
      "epoch": 4.9342155456542966e-05,
      "model_forward_time": 1.9017422199249268,
      "step": 51739
    },
    {
      "epoch": 4.9342155456542966e-05,
      "step": 51739,
      "training_step_time": 6.786170721054077
    },
    {
      "epoch": 4.934310913085938e-05,
      "grad_norm": 0.1225292980670929,
      "learning_rate": 5.1258157719113726e-06,
      "loss": 0.0601,
      "step": 51740
    },
    {
      "epoch": 4.934310913085938e-05,
      "model_forward_time": 1.9101717472076416,
      "step": 51740
    },
    {
      "dynamics_loss_avg": 0.06421153005212546,
      "epoch": 4.934310913085938e-05,
      "step": 51740
    },
    {
      "action_loss_avg": 0.004143306936020963,
      "epoch": 4.934310913085938e-05,
      "step": 51740
    },
    {
      "epoch": 4.934310913085938e-05,
      "step": 51740,
      "training_step_time": 6.8064000606536865
    },
    {
      "epoch": 4.934406280517578e-05,
      "model_forward_time": 1.8897767066955566,
      "step": 51741
    },
    {
      "epoch": 4.934406280517578e-05,
      "step": 51741,
      "training_step_time": 6.774976015090942
    },
    {
      "epoch": 4.9345016479492185e-05,
      "model_forward_time": 1.8875620365142822,
      "step": 51742
    },
    {
      "epoch": 4.9345016479492185e-05,
      "step": 51742,
      "training_step_time": 6.7842419147491455
    },
    {
      "epoch": 4.9345970153808596e-05,
      "model_forward_time": 1.9236714839935303,
      "step": 51743
    },
    {
      "epoch": 4.9345970153808596e-05,
      "step": 51743,
      "training_step_time": 7.258168458938599
    },
    {
      "epoch": 4.9346923828125e-05,
      "model_forward_time": 1.925239086151123,
      "step": 51744
    },
    {
      "epoch": 4.9346923828125e-05,
      "step": 51744,
      "training_step_time": 8.219017267227173
    },
    {
      "epoch": 4.9347877502441404e-05,
      "model_forward_time": 1.9067387580871582,
      "step": 51745
    },
    {
      "epoch": 4.9347877502441404e-05,
      "step": 51745,
      "training_step_time": 7.782421827316284
    },
    {
      "epoch": 4.9348831176757815e-05,
      "model_forward_time": 1.9035906791687012,
      "step": 51746
    },
    {
      "epoch": 4.9348831176757815e-05,
      "step": 51746,
      "training_step_time": 7.518091678619385
    },
    {
      "epoch": 4.934978485107422e-05,
      "model_forward_time": 1.906750202178955,
      "step": 51747
    },
    {
      "epoch": 4.934978485107422e-05,
      "step": 51747,
      "training_step_time": 6.770598888397217
    },
    {
      "epoch": 4.935073852539062e-05,
      "model_forward_time": 1.9196014404296875,
      "step": 51748
    },
    {
      "epoch": 4.935073852539062e-05,
      "step": 51748,
      "training_step_time": 6.851453542709351
    },
    {
      "epoch": 4.9351692199707033e-05,
      "model_forward_time": 1.7573940753936768,
      "step": 51749
    },
    {
      "epoch": 4.9351692199707033e-05,
      "step": 51749,
      "training_step_time": 6.761439323425293
    },
    {
      "epoch": 4.935264587402344e-05,
      "grad_norm": 0.13897760212421417,
      "learning_rate": 5.124162818802796e-06,
      "loss": 0.0615,
      "step": 51750
    },
    {
      "epoch": 4.935264587402344e-05,
      "model_forward_time": 1.7670066356658936,
      "step": 51750
    },
    {
      "dynamics_loss_avg": 0.0670383708551526,
      "epoch": 4.935264587402344e-05,
      "step": 51750
    },
    {
      "action_loss_avg": 0.0030750014091609048,
      "epoch": 4.935264587402344e-05,
      "step": 51750
    },
    {
      "epoch": 4.935264587402344e-05,
      "step": 51750,
      "training_step_time": 6.777189254760742
    },
    {
      "epoch": 4.935359954833984e-05,
      "model_forward_time": 1.7655510902404785,
      "step": 51751
    },
    {
      "epoch": 4.935359954833984e-05,
      "step": 51751,
      "training_step_time": 6.795192003250122
    },
    {
      "epoch": 4.935455322265625e-05,
      "model_forward_time": 1.7569918632507324,
      "step": 51752
    },
    {
      "epoch": 4.935455322265625e-05,
      "step": 51752,
      "training_step_time": 7.324769973754883
    },
    {
      "epoch": 4.9355506896972656e-05,
      "model_forward_time": 1.804236650466919,
      "step": 51753
    },
    {
      "epoch": 4.9355506896972656e-05,
      "step": 51753,
      "training_step_time": 7.6708502769470215
    },
    {
      "epoch": 4.935646057128906e-05,
      "model_forward_time": 1.7532362937927246,
      "step": 51754
    },
    {
      "epoch": 4.935646057128906e-05,
      "step": 51754,
      "training_step_time": 7.743060350418091
    },
    {
      "epoch": 4.935741424560547e-05,
      "model_forward_time": 1.7587039470672607,
      "step": 51755
    },
    {
      "epoch": 4.935741424560547e-05,
      "step": 51755,
      "training_step_time": 7.663116455078125
    },
    {
      "epoch": 4.9358367919921875e-05,
      "model_forward_time": 1.754892349243164,
      "step": 51756
    },
    {
      "epoch": 4.9358367919921875e-05,
      "step": 51756,
      "training_step_time": 6.787884473800659
    },
    {
      "epoch": 4.935932159423828e-05,
      "model_forward_time": 1.7575721740722656,
      "step": 51757
    },
    {
      "epoch": 4.935932159423828e-05,
      "step": 51757,
      "training_step_time": 6.767589092254639
    },
    {
      "epoch": 4.936027526855469e-05,
      "model_forward_time": 1.760443925857544,
      "step": 51758
    },
    {
      "epoch": 4.936027526855469e-05,
      "step": 51758,
      "training_step_time": 6.8075385093688965
    },
    {
      "epoch": 4.9361228942871094e-05,
      "model_forward_time": 1.751478910446167,
      "step": 51759
    },
    {
      "epoch": 4.9361228942871094e-05,
      "step": 51759,
      "training_step_time": 6.820911645889282
    },
    {
      "epoch": 4.93621826171875e-05,
      "grad_norm": 0.12344919890165329,
      "learning_rate": 5.122509852115959e-06,
      "loss": 0.0598,
      "step": 51760
    },
    {
      "epoch": 4.93621826171875e-05,
      "model_forward_time": 1.759174108505249,
      "step": 51760
    },
    {
      "dynamics_loss_avg": 0.059934092685580254,
      "epoch": 4.93621826171875e-05,
      "step": 51760
    },
    {
      "action_loss_avg": 0.002782076763105579,
      "epoch": 4.93621826171875e-05,
      "step": 51760
    },
    {
      "epoch": 4.93621826171875e-05,
      "step": 51760,
      "training_step_time": 6.801325559616089
    },
    {
      "epoch": 4.936313629150391e-05,
      "model_forward_time": 1.7558813095092773,
      "step": 51761
    },
    {
      "epoch": 4.936313629150391e-05,
      "step": 51761,
      "training_step_time": 7.72142481803894
    },
    {
      "epoch": 4.936408996582031e-05,
      "model_forward_time": 1.8093600273132324,
      "step": 51762
    },
    {
      "epoch": 4.936408996582031e-05,
      "step": 51762,
      "training_step_time": 8.731071472167969
    },
    {
      "epoch": 4.936504364013672e-05,
      "model_forward_time": 1.7467725276947021,
      "step": 51763
    },
    {
      "epoch": 4.936504364013672e-05,
      "step": 51763,
      "training_step_time": 8.402455806732178
    },
    {
      "epoch": 4.936599731445313e-05,
      "model_forward_time": 1.7514424324035645,
      "step": 51764
    },
    {
      "epoch": 4.936599731445313e-05,
      "step": 51764,
      "training_step_time": 7.659255504608154
    },
    {
      "epoch": 4.936695098876953e-05,
      "model_forward_time": 1.7689926624298096,
      "step": 51765
    },
    {
      "epoch": 4.936695098876953e-05,
      "step": 51765,
      "training_step_time": 6.727316617965698
    },
    {
      "epoch": 4.9367904663085936e-05,
      "model_forward_time": 1.757448673248291,
      "step": 51766
    },
    {
      "epoch": 4.9367904663085936e-05,
      "step": 51766,
      "training_step_time": 6.754805564880371
    },
    {
      "epoch": 4.936885833740235e-05,
      "model_forward_time": 1.7565333843231201,
      "step": 51767
    },
    {
      "epoch": 4.936885833740235e-05,
      "step": 51767,
      "training_step_time": 6.77810525894165
    },
    {
      "epoch": 4.936981201171875e-05,
      "model_forward_time": 1.756859540939331,
      "step": 51768
    },
    {
      "epoch": 4.936981201171875e-05,
      "step": 51768,
      "training_step_time": 6.736112356185913
    },
    {
      "epoch": 4.9370765686035155e-05,
      "model_forward_time": 1.751054286956787,
      "step": 51769
    },
    {
      "epoch": 4.9370765686035155e-05,
      "step": 51769,
      "training_step_time": 6.749728679656982
    },
    {
      "epoch": 4.9371719360351565e-05,
      "grad_norm": 0.14495877921581268,
      "learning_rate": 5.12085687203163e-06,
      "loss": 0.0587,
      "step": 51770
    },
    {
      "epoch": 4.9371719360351565e-05,
      "model_forward_time": 1.7601957321166992,
      "step": 51770
    },
    {
      "dynamics_loss_avg": 0.06506386883556843,
      "epoch": 4.9371719360351565e-05,
      "step": 51770
    },
    {
      "action_loss_avg": 0.002896432584384456,
      "epoch": 4.9371719360351565e-05,
      "step": 51770
    },
    {
      "epoch": 4.9371719360351565e-05,
      "step": 51770,
      "training_step_time": 7.282803535461426
    },
    {
      "epoch": 4.937267303466797e-05,
      "model_forward_time": 1.7504441738128662,
      "step": 51771
    },
    {
      "epoch": 4.937267303466797e-05,
      "step": 51771,
      "training_step_time": 7.58929181098938
    },
    {
      "epoch": 4.9373626708984374e-05,
      "model_forward_time": 1.8097379207611084,
      "step": 51772
    },
    {
      "epoch": 4.9373626708984374e-05,
      "step": 51772,
      "training_step_time": 8.893417596817017
    },
    {
      "epoch": 4.9374580383300784e-05,
      "model_forward_time": 1.7707931995391846,
      "step": 51773
    },
    {
      "epoch": 4.9374580383300784e-05,
      "step": 51773,
      "training_step_time": 7.755825519561768
    },
    {
      "epoch": 4.937553405761719e-05,
      "model_forward_time": 1.7660834789276123,
      "step": 51774
    },
    {
      "epoch": 4.937553405761719e-05,
      "step": 51774,
      "training_step_time": 6.729633569717407
    },
    {
      "epoch": 4.937648773193359e-05,
      "model_forward_time": 1.7523460388183594,
      "step": 51775
    },
    {
      "epoch": 4.937648773193359e-05,
      "step": 51775,
      "training_step_time": 6.737436056137085
    },
    {
      "epoch": 4.937744140625e-05,
      "model_forward_time": 1.7564358711242676,
      "step": 51776
    },
    {
      "epoch": 4.937744140625e-05,
      "step": 51776,
      "training_step_time": 6.726135730743408
    },
    {
      "epoch": 4.937839508056641e-05,
      "model_forward_time": 1.756377935409546,
      "step": 51777
    },
    {
      "epoch": 4.937839508056641e-05,
      "step": 51777,
      "training_step_time": 6.716205358505249
    },
    {
      "epoch": 4.937934875488281e-05,
      "model_forward_time": 1.760643482208252,
      "step": 51778
    },
    {
      "epoch": 4.937934875488281e-05,
      "step": 51778,
      "training_step_time": 6.731799602508545
    },
    {
      "epoch": 4.938030242919922e-05,
      "model_forward_time": 1.7554559707641602,
      "step": 51779
    },
    {
      "epoch": 4.938030242919922e-05,
      "step": 51779,
      "training_step_time": 6.804089307785034
    },
    {
      "epoch": 4.9381256103515626e-05,
      "grad_norm": 0.11973275989294052,
      "learning_rate": 5.119203878730575e-06,
      "loss": 0.0598,
      "step": 51780
    },
    {
      "epoch": 4.9381256103515626e-05,
      "model_forward_time": 2.8899409770965576,
      "step": 51780
    },
    {
      "dynamics_loss_avg": 0.06684283306822181,
      "epoch": 4.9381256103515626e-05,
      "step": 51780
    },
    {
      "action_loss_avg": 0.0033479205216281117,
      "epoch": 4.9381256103515626e-05,
      "step": 51780
    },
    {
      "epoch": 4.9381256103515626e-05,
      "step": 51780,
      "training_step_time": 7.6473729610443115
    },
    {
      "epoch": 4.938220977783203e-05,
      "model_forward_time": 1.760789155960083,
      "step": 51781
    },
    {
      "epoch": 4.938220977783203e-05,
      "step": 51781,
      "training_step_time": 9.209898233413696
    },
    {
      "epoch": 4.9383163452148434e-05,
      "model_forward_time": 1.7407727241516113,
      "step": 51782
    },
    {
      "epoch": 4.9383163452148434e-05,
      "step": 51782,
      "training_step_time": 8.449166774749756
    },
    {
      "epoch": 4.9384117126464845e-05,
      "model_forward_time": 1.743992805480957,
      "step": 51783
    },
    {
      "epoch": 4.9384117126464845e-05,
      "step": 51783,
      "training_step_time": 7.550419092178345
    },
    {
      "epoch": 4.938507080078125e-05,
      "model_forward_time": 1.7345483303070068,
      "step": 51784
    },
    {
      "epoch": 4.938507080078125e-05,
      "step": 51784,
      "training_step_time": 7.039537191390991
    },
    {
      "epoch": 4.938602447509765e-05,
      "model_forward_time": 1.7384395599365234,
      "step": 51785
    },
    {
      "epoch": 4.938602447509765e-05,
      "step": 51785,
      "training_step_time": 6.734506130218506
    },
    {
      "epoch": 4.9386978149414064e-05,
      "model_forward_time": 1.7326021194458008,
      "step": 51786
    },
    {
      "epoch": 4.9386978149414064e-05,
      "step": 51786,
      "training_step_time": 6.760392904281616
    },
    {
      "epoch": 4.938793182373047e-05,
      "model_forward_time": 1.7376861572265625,
      "step": 51787
    },
    {
      "epoch": 4.938793182373047e-05,
      "step": 51787,
      "training_step_time": 6.72555136680603
    },
    {
      "epoch": 4.938888549804687e-05,
      "model_forward_time": 1.7342095375061035,
      "step": 51788
    },
    {
      "epoch": 4.938888549804687e-05,
      "step": 51788,
      "training_step_time": 6.73273229598999
    },
    {
      "epoch": 4.938983917236328e-05,
      "model_forward_time": 2.9199631214141846,
      "step": 51789
    },
    {
      "epoch": 4.938983917236328e-05,
      "step": 51789,
      "training_step_time": 8.479434251785278
    },
    {
      "epoch": 4.939079284667969e-05,
      "grad_norm": 0.10208901017904282,
      "learning_rate": 5.117550872393564e-06,
      "loss": 0.0601,
      "step": 51790
    },
    {
      "epoch": 4.939079284667969e-05,
      "model_forward_time": 1.8003995418548584,
      "step": 51790
    },
    {
      "dynamics_loss_avg": 0.06790239959955216,
      "epoch": 4.939079284667969e-05,
      "step": 51790
    },
    {
      "action_loss_avg": 0.0021052255237009377,
      "epoch": 4.939079284667969e-05,
      "step": 51790
    },
    {
      "epoch": 4.939079284667969e-05,
      "step": 51790,
      "training_step_time": 7.705927133560181
    },
    {
      "epoch": 4.939174652099609e-05,
      "model_forward_time": 1.7353401184082031,
      "step": 51791
    },
    {
      "epoch": 4.939174652099609e-05,
      "step": 51791,
      "training_step_time": 8.743165016174316
    },
    {
      "epoch": 4.93927001953125e-05,
      "model_forward_time": 1.7371940612792969,
      "step": 51792
    },
    {
      "epoch": 4.93927001953125e-05,
      "step": 51792,
      "training_step_time": 7.558396816253662
    },
    {
      "epoch": 4.9393653869628906e-05,
      "model_forward_time": 1.7383511066436768,
      "step": 51793
    },
    {
      "epoch": 4.9393653869628906e-05,
      "step": 51793,
      "training_step_time": 7.035240411758423
    },
    {
      "epoch": 4.939460754394531e-05,
      "model_forward_time": 1.7356343269348145,
      "step": 51794
    },
    {
      "epoch": 4.939460754394531e-05,
      "step": 51794,
      "training_step_time": 6.765951156616211
    },
    {
      "epoch": 4.939556121826172e-05,
      "model_forward_time": 1.7308988571166992,
      "step": 51795
    },
    {
      "epoch": 4.939556121826172e-05,
      "step": 51795,
      "training_step_time": 6.741689205169678
    },
    {
      "epoch": 4.9396514892578124e-05,
      "model_forward_time": 1.7440474033355713,
      "step": 51796
    },
    {
      "epoch": 4.9396514892578124e-05,
      "step": 51796,
      "training_step_time": 6.7342369556427
    },
    {
      "epoch": 4.939746856689453e-05,
      "model_forward_time": 1.7375261783599854,
      "step": 51797
    },
    {
      "epoch": 4.939746856689453e-05,
      "step": 51797,
      "training_step_time": 6.782949924468994
    },
    {
      "epoch": 4.939842224121094e-05,
      "model_forward_time": 2.884678602218628,
      "step": 51798
    },
    {
      "epoch": 4.939842224121094e-05,
      "step": 51798,
      "training_step_time": 8.294339895248413
    },
    {
      "epoch": 4.939937591552734e-05,
      "model_forward_time": 1.7602550983428955,
      "step": 51799
    },
    {
      "epoch": 4.939937591552734e-05,
      "step": 51799,
      "training_step_time": 7.6042094230651855
    },
    {
      "epoch": 4.940032958984375e-05,
      "grad_norm": 0.1887282431125641,
      "learning_rate": 5.1158978532013645e-06,
      "loss": 0.0582,
      "step": 51800
    },
    {
      "epoch": 4.940032958984375e-05,
      "model_forward_time": 1.7420217990875244,
      "step": 51800
    },
    {
      "dynamics_loss_avg": 0.060407713428139685,
      "epoch": 4.940032958984375e-05,
      "step": 51800
    },
    {
      "action_loss_avg": 0.003091649612179026,
      "epoch": 4.940032958984375e-05,
      "step": 51800
    },
    {
      "epoch": 4.940032958984375e-05,
      "step": 51800,
      "training_step_time": 9.477096557617188
    },
    {
      "epoch": 4.940128326416016e-05,
      "model_forward_time": 1.725203037261963,
      "step": 51801
    },
    {
      "epoch": 4.940128326416016e-05,
      "step": 51801,
      "training_step_time": 8.647064208984375
    },
    {
      "epoch": 4.940223693847656e-05,
      "model_forward_time": 1.741288185119629,
      "step": 51802
    },
    {
      "epoch": 4.940223693847656e-05,
      "step": 51802,
      "training_step_time": 6.982239484786987
    },
    {
      "epoch": 4.9403190612792966e-05,
      "model_forward_time": 1.738189458847046,
      "step": 51803
    },
    {
      "epoch": 4.9403190612792966e-05,
      "step": 51803,
      "training_step_time": 6.718251943588257
    },
    {
      "epoch": 4.940414428710938e-05,
      "model_forward_time": 1.7374722957611084,
      "step": 51804
    },
    {
      "epoch": 4.940414428710938e-05,
      "step": 51804,
      "training_step_time": 6.707352161407471
    },
    {
      "epoch": 4.940509796142578e-05,
      "model_forward_time": 1.7445731163024902,
      "step": 51805
    },
    {
      "epoch": 4.940509796142578e-05,
      "step": 51805,
      "training_step_time": 6.7077882289886475
    },
    {
      "epoch": 4.9406051635742185e-05,
      "model_forward_time": 1.7443573474884033,
      "step": 51806
    },
    {
      "epoch": 4.9406051635742185e-05,
      "step": 51806,
      "training_step_time": 6.689322233200073
    },
    {
      "epoch": 4.9407005310058596e-05,
      "model_forward_time": 2.8655264377593994,
      "step": 51807
    },
    {
      "epoch": 4.9407005310058596e-05,
      "step": 51807,
      "training_step_time": 7.595380067825317
    },
    {
      "epoch": 4.9407958984375e-05,
      "model_forward_time": 1.7636609077453613,
      "step": 51808
    },
    {
      "epoch": 4.9407958984375e-05,
      "step": 51808,
      "training_step_time": 7.524962902069092
    },
    {
      "epoch": 4.9408912658691404e-05,
      "model_forward_time": 1.7400612831115723,
      "step": 51809
    },
    {
      "epoch": 4.9408912658691404e-05,
      "step": 51809,
      "training_step_time": 8.791332721710205
    },
    {
      "epoch": 4.9409866333007815e-05,
      "grad_norm": 0.11220768094062805,
      "learning_rate": 5.114244821334752e-06,
      "loss": 0.058,
      "step": 51810
    },
    {
      "epoch": 4.9409866333007815e-05,
      "model_forward_time": 1.7495448589324951,
      "step": 51810
    },
    {
      "dynamics_loss_avg": 0.07952514104545116,
      "epoch": 4.9409866333007815e-05,
      "step": 51810
    },
    {
      "action_loss_avg": 0.001912192176678218,
      "epoch": 4.9409866333007815e-05,
      "step": 51810
    },
    {
      "epoch": 4.9409866333007815e-05,
      "step": 51810,
      "training_step_time": 7.686568260192871
    },
    {
      "epoch": 4.941082000732422e-05,
      "model_forward_time": 1.741795301437378,
      "step": 51811
    },
    {
      "epoch": 4.941082000732422e-05,
      "step": 51811,
      "training_step_time": 7.558673858642578
    },
    {
      "epoch": 4.941177368164062e-05,
      "model_forward_time": 1.736687183380127,
      "step": 51812
    },
    {
      "epoch": 4.941177368164062e-05,
      "step": 51812,
      "training_step_time": 6.793189287185669
    },
    {
      "epoch": 4.9412727355957034e-05,
      "model_forward_time": 1.7512218952178955,
      "step": 51813
    },
    {
      "epoch": 4.9412727355957034e-05,
      "step": 51813,
      "training_step_time": 6.828771352767944
    },
    {
      "epoch": 4.941368103027344e-05,
      "model_forward_time": 1.7454524040222168,
      "step": 51814
    },
    {
      "epoch": 4.941368103027344e-05,
      "step": 51814,
      "training_step_time": 6.79875111579895
    },
    {
      "epoch": 4.941463470458984e-05,
      "model_forward_time": 1.73201322555542,
      "step": 51815
    },
    {
      "epoch": 4.941463470458984e-05,
      "step": 51815,
      "training_step_time": 6.752605199813843
    },
    {
      "epoch": 4.941558837890625e-05,
      "model_forward_time": 2.8352296352386475,
      "step": 51816
    },
    {
      "epoch": 4.941558837890625e-05,
      "step": 51816,
      "training_step_time": 7.661244869232178
    },
    {
      "epoch": 4.9416542053222656e-05,
      "model_forward_time": 1.7303740978240967,
      "step": 51817
    },
    {
      "epoch": 4.9416542053222656e-05,
      "step": 51817,
      "training_step_time": 8.988751411437988
    },
    {
      "epoch": 4.941749572753906e-05,
      "model_forward_time": 1.7320597171783447,
      "step": 51818
    },
    {
      "epoch": 4.941749572753906e-05,
      "step": 51818,
      "training_step_time": 9.129726886749268
    },
    {
      "epoch": 4.941844940185547e-05,
      "model_forward_time": 1.748462438583374,
      "step": 51819
    },
    {
      "epoch": 4.941844940185547e-05,
      "step": 51819,
      "training_step_time": 7.769479990005493
    },
    {
      "epoch": 4.9419403076171875e-05,
      "grad_norm": 0.15504860877990723,
      "learning_rate": 5.112591776974497e-06,
      "loss": 0.0579,
      "step": 51820
    },
    {
      "epoch": 4.9419403076171875e-05,
      "model_forward_time": 1.754791498184204,
      "step": 51820
    },
    {
      "dynamics_loss_avg": 0.07119166012853384,
      "epoch": 4.9419403076171875e-05,
      "step": 51820
    },
    {
      "action_loss_avg": 0.0024105708755087106,
      "epoch": 4.9419403076171875e-05,
      "step": 51820
    },
    {
      "epoch": 4.9419403076171875e-05,
      "step": 51820,
      "training_step_time": 7.61996054649353
    },
    {
      "epoch": 4.942035675048828e-05,
      "model_forward_time": 1.7436506748199463,
      "step": 51821
    },
    {
      "epoch": 4.942035675048828e-05,
      "step": 51821,
      "training_step_time": 7.102458953857422
    },
    {
      "epoch": 4.942131042480469e-05,
      "model_forward_time": 1.0576798915863037,
      "step": 51822
    },
    {
      "epoch": 4.942131042480469e-05,
      "step": 51822,
      "training_step_time": 6.849045276641846
    },
    {
      "epoch": 4.9422264099121094e-05,
      "model_forward_time": 1.7533602714538574,
      "step": 51823
    },
    {
      "epoch": 4.9422264099121094e-05,
      "step": 51823,
      "training_step_time": 6.867807149887085
    },
    {
      "epoch": 4.94232177734375e-05,
      "model_forward_time": 1.7428016662597656,
      "step": 51824
    },
    {
      "epoch": 4.94232177734375e-05,
      "step": 51824,
      "training_step_time": 6.840238094329834
    },
    {
      "epoch": 4.942417144775391e-05,
      "model_forward_time": 1.741640329360962,
      "step": 51825
    },
    {
      "epoch": 4.942417144775391e-05,
      "step": 51825,
      "training_step_time": 7.8035807609558105
    },
    {
      "epoch": 4.942512512207031e-05,
      "model_forward_time": 2.897857904434204,
      "step": 51826
    },
    {
      "epoch": 4.942512512207031e-05,
      "step": 51826,
      "training_step_time": 9.63447117805481
    },
    {
      "epoch": 4.942607879638672e-05,
      "model_forward_time": 1.7896883487701416,
      "step": 51827
    },
    {
      "epoch": 4.942607879638672e-05,
      "step": 51827,
      "training_step_time": 8.844946384429932
    },
    {
      "epoch": 4.942703247070313e-05,
      "model_forward_time": 1.7423858642578125,
      "step": 51828
    },
    {
      "epoch": 4.942703247070313e-05,
      "step": 51828,
      "training_step_time": 7.808364629745483
    },
    {
      "epoch": 4.942798614501953e-05,
      "model_forward_time": 1.7437200546264648,
      "step": 51829
    },
    {
      "epoch": 4.942798614501953e-05,
      "step": 51829,
      "training_step_time": 7.6643967628479
    },
    {
      "epoch": 4.9428939819335936e-05,
      "grad_norm": 0.13224650919437408,
      "learning_rate": 5.110938720301374e-06,
      "loss": 0.0554,
      "step": 51830
    },
    {
      "epoch": 4.9428939819335936e-05,
      "model_forward_time": 1.7529411315917969,
      "step": 51830
    },
    {
      "dynamics_loss_avg": 0.043401670455932614,
      "epoch": 4.9428939819335936e-05,
      "step": 51830
    },
    {
      "action_loss_avg": 0.002557388404966332,
      "epoch": 4.9428939819335936e-05,
      "step": 51830
    },
    {
      "epoch": 4.9428939819335936e-05,
      "step": 51830,
      "training_step_time": 7.132323741912842
    },
    {
      "epoch": 4.942989349365235e-05,
      "model_forward_time": 1.7379310131072998,
      "step": 51831
    },
    {
      "epoch": 4.942989349365235e-05,
      "step": 51831,
      "training_step_time": 6.808210134506226
    },
    {
      "epoch": 4.943084716796875e-05,
      "model_forward_time": 1.7385520935058594,
      "step": 51832
    },
    {
      "epoch": 4.943084716796875e-05,
      "step": 51832,
      "training_step_time": 6.802032709121704
    },
    {
      "epoch": 4.9431800842285155e-05,
      "model_forward_time": 1.740196704864502,
      "step": 51833
    },
    {
      "epoch": 4.9431800842285155e-05,
      "step": 51833,
      "training_step_time": 6.799857139587402
    },
    {
      "epoch": 4.9432754516601566e-05,
      "model_forward_time": 1.763392686843872,
      "step": 51834
    },
    {
      "epoch": 4.9432754516601566e-05,
      "step": 51834,
      "training_step_time": 7.559442520141602
    },
    {
      "epoch": 4.943370819091797e-05,
      "model_forward_time": 2.861419439315796,
      "step": 51835
    },
    {
      "epoch": 4.943370819091797e-05,
      "step": 51835,
      "training_step_time": 8.189292669296265
    },
    {
      "epoch": 4.9434661865234374e-05,
      "model_forward_time": 1.7920265197753906,
      "step": 51836
    },
    {
      "epoch": 4.9434661865234374e-05,
      "step": 51836,
      "training_step_time": 9.50951862335205
    },
    {
      "epoch": 4.9435615539550784e-05,
      "model_forward_time": 1.738234519958496,
      "step": 51837
    },
    {
      "epoch": 4.9435615539550784e-05,
      "step": 51837,
      "training_step_time": 7.679968595504761
    },
    {
      "epoch": 4.943656921386719e-05,
      "model_forward_time": 1.7400994300842285,
      "step": 51838
    },
    {
      "epoch": 4.943656921386719e-05,
      "step": 51838,
      "training_step_time": 7.559266090393066
    },
    {
      "epoch": 4.943752288818359e-05,
      "model_forward_time": 1.7475199699401855,
      "step": 51839
    },
    {
      "epoch": 4.943752288818359e-05,
      "step": 51839,
      "training_step_time": 7.510594129562378
    },
    {
      "epoch": 4.94384765625e-05,
      "grad_norm": 0.13099445402622223,
      "learning_rate": 5.10928565149616e-06,
      "loss": 0.0599,
      "step": 51840
    },
    {
      "epoch": 4.94384765625e-05,
      "model_forward_time": 1.7344353199005127,
      "step": 51840
    },
    {
      "dynamics_loss_avg": 0.07090285122394561,
      "epoch": 4.94384765625e-05,
      "step": 51840
    },
    {
      "action_loss_avg": 0.004301448015030474,
      "epoch": 4.94384765625e-05,
      "step": 51840
    },
    {
      "epoch": 4.94384765625e-05,
      "step": 51840,
      "training_step_time": 6.8094635009765625
    },
    {
      "epoch": 4.943943023681641e-05,
      "model_forward_time": 1.737104892730713,
      "step": 51841
    },
    {
      "epoch": 4.943943023681641e-05,
      "step": 51841,
      "training_step_time": 6.750264406204224
    },
    {
      "epoch": 4.944038391113281e-05,
      "model_forward_time": 1.7422866821289062,
      "step": 51842
    },
    {
      "epoch": 4.944038391113281e-05,
      "step": 51842,
      "training_step_time": 6.789372682571411
    },
    {
      "epoch": 4.944133758544922e-05,
      "model_forward_time": 1.7375414371490479,
      "step": 51843
    },
    {
      "epoch": 4.944133758544922e-05,
      "step": 51843,
      "training_step_time": 8.6048583984375
    },
    {
      "epoch": 4.9442291259765626e-05,
      "model_forward_time": 2.8308093547821045,
      "step": 51844
    },
    {
      "epoch": 4.9442291259765626e-05,
      "step": 51844,
      "training_step_time": 7.587179899215698
    },
    {
      "epoch": 4.944324493408203e-05,
      "model_forward_time": 1.7967796325683594,
      "step": 51845
    },
    {
      "epoch": 4.944324493408203e-05,
      "step": 51845,
      "training_step_time": 9.186200618743896
    },
    {
      "epoch": 4.9444198608398434e-05,
      "model_forward_time": 1.7389695644378662,
      "step": 51846
    },
    {
      "epoch": 4.9444198608398434e-05,
      "step": 51846,
      "training_step_time": 12.193611860275269
    },
    {
      "epoch": 4.9445152282714845e-05,
      "model_forward_time": 1.740154504776001,
      "step": 51847
    },
    {
      "epoch": 4.9445152282714845e-05,
      "step": 51847,
      "training_step_time": 7.720162391662598
    },
    {
      "epoch": 4.944610595703125e-05,
      "model_forward_time": 1.740119218826294,
      "step": 51848
    },
    {
      "epoch": 4.944610595703125e-05,
      "step": 51848,
      "training_step_time": 7.530053377151489
    },
    {
      "epoch": 4.944705963134765e-05,
      "model_forward_time": 1.7482304573059082,
      "step": 51849
    },
    {
      "epoch": 4.944705963134765e-05,
      "step": 51849,
      "training_step_time": 7.025293588638306
    },
    {
      "epoch": 4.9448013305664064e-05,
      "grad_norm": 0.14313815534114838,
      "learning_rate": 5.107632570739631e-06,
      "loss": 0.0561,
      "step": 51850
    },
    {
      "epoch": 4.9448013305664064e-05,
      "model_forward_time": 1.741697072982788,
      "step": 51850
    },
    {
      "dynamics_loss_avg": 0.05801241435110569,
      "epoch": 4.9448013305664064e-05,
      "step": 51850
    },
    {
      "action_loss_avg": 0.0022595963877392934,
      "epoch": 4.9448013305664064e-05,
      "step": 51850
    },
    {
      "epoch": 4.9448013305664064e-05,
      "step": 51850,
      "training_step_time": 6.806094169616699
    },
    {
      "epoch": 4.944896697998047e-05,
      "model_forward_time": 1.734529972076416,
      "step": 51851
    },
    {
      "epoch": 4.944896697998047e-05,
      "step": 51851,
      "training_step_time": 6.7484540939331055
    },
    {
      "epoch": 4.944992065429687e-05,
      "model_forward_time": 1.738034725189209,
      "step": 51852
    },
    {
      "epoch": 4.944992065429687e-05,
      "step": 51852,
      "training_step_time": 7.318172931671143
    },
    {
      "epoch": 4.945087432861328e-05,
      "model_forward_time": 2.8267102241516113,
      "step": 51853
    },
    {
      "epoch": 4.945087432861328e-05,
      "step": 51853,
      "training_step_time": 7.578870058059692
    },
    {
      "epoch": 4.945182800292969e-05,
      "model_forward_time": 1.801008701324463,
      "step": 51854
    },
    {
      "epoch": 4.945182800292969e-05,
      "step": 51854,
      "training_step_time": 8.951661109924316
    },
    {
      "epoch": 4.945278167724609e-05,
      "model_forward_time": 1.7385711669921875,
      "step": 51855
    },
    {
      "epoch": 4.945278167724609e-05,
      "step": 51855,
      "training_step_time": 8.801822662353516
    },
    {
      "epoch": 4.94537353515625e-05,
      "model_forward_time": 1.7483208179473877,
      "step": 51856
    },
    {
      "epoch": 4.94537353515625e-05,
      "step": 51856,
      "training_step_time": 7.783218145370483
    },
    {
      "epoch": 4.9454689025878906e-05,
      "model_forward_time": 1.7347373962402344,
      "step": 51857
    },
    {
      "epoch": 4.9454689025878906e-05,
      "step": 51857,
      "training_step_time": 7.5003838539123535
    },
    {
      "epoch": 4.945564270019531e-05,
      "model_forward_time": 1.747657060623169,
      "step": 51858
    },
    {
      "epoch": 4.945564270019531e-05,
      "step": 51858,
      "training_step_time": 6.959987640380859
    },
    {
      "epoch": 4.945659637451172e-05,
      "model_forward_time": 1.7355594635009766,
      "step": 51859
    },
    {
      "epoch": 4.945659637451172e-05,
      "step": 51859,
      "training_step_time": 6.7023725509643555
    },
    {
      "epoch": 4.9457550048828124e-05,
      "grad_norm": 0.10002726316452026,
      "learning_rate": 5.105979478212565e-06,
      "loss": 0.0607,
      "step": 51860
    },
    {
      "epoch": 4.9457550048828124e-05,
      "model_forward_time": 1.7365834712982178,
      "step": 51860
    },
    {
      "dynamics_loss_avg": 0.0606233075261116,
      "epoch": 4.9457550048828124e-05,
      "step": 51860
    },
    {
      "action_loss_avg": 0.003540134316426702,
      "epoch": 4.9457550048828124e-05,
      "step": 51860
    },
    {
      "epoch": 4.9457550048828124e-05,
      "step": 51860,
      "training_step_time": 6.695889472961426
    },
    {
      "epoch": 4.945850372314453e-05,
      "model_forward_time": 1.7340455055236816,
      "step": 51861
    },
    {
      "epoch": 4.945850372314453e-05,
      "step": 51861,
      "training_step_time": 7.225435256958008
    },
    {
      "epoch": 4.945945739746094e-05,
      "model_forward_time": 1.741687536239624,
      "step": 51862
    },
    {
      "epoch": 4.945945739746094e-05,
      "step": 51862,
      "training_step_time": 7.557804107666016
    },
    {
      "epoch": 4.946041107177734e-05,
      "model_forward_time": 2.947256326675415,
      "step": 51863
    },
    {
      "epoch": 4.946041107177734e-05,
      "step": 51863,
      "training_step_time": 7.661794185638428
    },
    {
      "epoch": 4.946136474609375e-05,
      "model_forward_time": 1.736537218093872,
      "step": 51864
    },
    {
      "epoch": 4.946136474609375e-05,
      "step": 51864,
      "training_step_time": 8.541783332824707
    },
    {
      "epoch": 4.946231842041016e-05,
      "model_forward_time": 1.734128713607788,
      "step": 51865
    },
    {
      "epoch": 4.946231842041016e-05,
      "step": 51865,
      "training_step_time": 8.21176791191101
    },
    {
      "epoch": 4.946327209472656e-05,
      "model_forward_time": 1.7531182765960693,
      "step": 51866
    },
    {
      "epoch": 4.946327209472656e-05,
      "step": 51866,
      "training_step_time": 7.568552732467651
    },
    {
      "epoch": 4.9464225769042966e-05,
      "model_forward_time": 1.7374229431152344,
      "step": 51867
    },
    {
      "epoch": 4.9464225769042966e-05,
      "step": 51867,
      "training_step_time": 6.950340032577515
    },
    {
      "epoch": 4.946517944335938e-05,
      "model_forward_time": 1.7513492107391357,
      "step": 51868
    },
    {
      "epoch": 4.946517944335938e-05,
      "step": 51868,
      "training_step_time": 6.74131965637207
    },
    {
      "epoch": 4.946613311767578e-05,
      "model_forward_time": 1.7441883087158203,
      "step": 51869
    },
    {
      "epoch": 4.946613311767578e-05,
      "step": 51869,
      "training_step_time": 6.724776029586792
    },
    {
      "epoch": 4.9467086791992185e-05,
      "grad_norm": 0.14594322443008423,
      "learning_rate": 5.104326374095744e-06,
      "loss": 0.0613,
      "step": 51870
    },
    {
      "epoch": 4.9467086791992185e-05,
      "model_forward_time": 1.743537187576294,
      "step": 51870
    },
    {
      "dynamics_loss_avg": 0.06881772819906473,
      "epoch": 4.9467086791992185e-05,
      "step": 51870
    },
    {
      "action_loss_avg": 0.0029994807089678945,
      "epoch": 4.9467086791992185e-05,
      "step": 51870
    },
    {
      "epoch": 4.9467086791992185e-05,
      "step": 51870,
      "training_step_time": 7.241699934005737
    },
    {
      "epoch": 4.9468040466308596e-05,
      "model_forward_time": 1.731933355331421,
      "step": 51871
    },
    {
      "epoch": 4.9468040466308596e-05,
      "step": 51871,
      "training_step_time": 7.5901312828063965
    },
    {
      "epoch": 4.9468994140625e-05,
      "model_forward_time": 2.1301722526550293,
      "step": 51872
    },
    {
      "epoch": 4.9468994140625e-05,
      "step": 51872,
      "training_step_time": 8.239923000335693
    },
    {
      "epoch": 4.9469947814941404e-05,
      "model_forward_time": 1.797905445098877,
      "step": 51873
    },
    {
      "epoch": 4.9469947814941404e-05,
      "step": 51873,
      "training_step_time": 9.044214010238647
    },
    {
      "epoch": 4.9470901489257815e-05,
      "model_forward_time": 1.740142822265625,
      "step": 51874
    },
    {
      "epoch": 4.9470901489257815e-05,
      "step": 51874,
      "training_step_time": 8.728431463241577
    },
    {
      "epoch": 4.947185516357422e-05,
      "model_forward_time": 1.7323062419891357,
      "step": 51875
    },
    {
      "epoch": 4.947185516357422e-05,
      "step": 51875,
      "training_step_time": 8.369216442108154
    },
    {
      "epoch": 4.947280883789062e-05,
      "model_forward_time": 1.7376458644866943,
      "step": 51876
    },
    {
      "epoch": 4.947280883789062e-05,
      "step": 51876,
      "training_step_time": 6.993889570236206
    },
    {
      "epoch": 4.9473762512207034e-05,
      "model_forward_time": 1.7405548095703125,
      "step": 51877
    },
    {
      "epoch": 4.9473762512207034e-05,
      "step": 51877,
      "training_step_time": 6.723458528518677
    },
    {
      "epoch": 4.947471618652344e-05,
      "model_forward_time": 1.7402169704437256,
      "step": 51878
    },
    {
      "epoch": 4.947471618652344e-05,
      "step": 51878,
      "training_step_time": 6.7251222133636475
    },
    {
      "epoch": 4.947566986083984e-05,
      "model_forward_time": 1.7356784343719482,
      "step": 51879
    },
    {
      "epoch": 4.947566986083984e-05,
      "step": 51879,
      "training_step_time": 7.259034633636475
    },
    {
      "epoch": 4.947662353515625e-05,
      "grad_norm": 0.10976189374923706,
      "learning_rate": 5.102673258569947e-06,
      "loss": 0.0582,
      "step": 51880
    },
    {
      "epoch": 4.947662353515625e-05,
      "model_forward_time": 1.734886884689331,
      "step": 51880
    },
    {
      "dynamics_loss_avg": 0.05912239421159029,
      "epoch": 4.947662353515625e-05,
      "step": 51880
    },
    {
      "action_loss_avg": 0.0020013124216347934,
      "epoch": 4.947662353515625e-05,
      "step": 51880
    },
    {
      "epoch": 4.947662353515625e-05,
      "step": 51880,
      "training_step_time": 7.548110008239746
    },
    {
      "epoch": 4.9477577209472656e-05,
      "model_forward_time": 2.8668212890625,
      "step": 51881
    },
    {
      "epoch": 4.9477577209472656e-05,
      "step": 51881,
      "training_step_time": 8.172106266021729
    },
    {
      "epoch": 4.947853088378906e-05,
      "model_forward_time": 1.7957384586334229,
      "step": 51882
    },
    {
      "epoch": 4.947853088378906e-05,
      "step": 51882,
      "training_step_time": 8.646517515182495
    },
    {
      "epoch": 4.947948455810547e-05,
      "model_forward_time": 1.757239580154419,
      "step": 51883
    },
    {
      "epoch": 4.947948455810547e-05,
      "step": 51883,
      "training_step_time": 7.696109056472778
    },
    {
      "epoch": 4.9480438232421875e-05,
      "model_forward_time": 1.7674295902252197,
      "step": 51884
    },
    {
      "epoch": 4.9480438232421875e-05,
      "step": 51884,
      "training_step_time": 8.623807907104492
    },
    {
      "epoch": 4.948139190673828e-05,
      "model_forward_time": 1.7646150588989258,
      "step": 51885
    },
    {
      "epoch": 4.948139190673828e-05,
      "step": 51885,
      "training_step_time": 8.602642059326172
    },
    {
      "epoch": 4.948234558105469e-05,
      "model_forward_time": 1.756605863571167,
      "step": 51886
    },
    {
      "epoch": 4.948234558105469e-05,
      "step": 51886,
      "training_step_time": 6.775846004486084
    },
    {
      "epoch": 4.9483299255371094e-05,
      "model_forward_time": 1.7552833557128906,
      "step": 51887
    },
    {
      "epoch": 4.9483299255371094e-05,
      "step": 51887,
      "training_step_time": 6.770739555358887
    },
    {
      "epoch": 4.94842529296875e-05,
      "model_forward_time": 1.7510004043579102,
      "step": 51888
    },
    {
      "epoch": 4.94842529296875e-05,
      "step": 51888,
      "training_step_time": 7.345018625259399
    },
    {
      "epoch": 4.948520660400391e-05,
      "model_forward_time": 1.759544849395752,
      "step": 51889
    },
    {
      "epoch": 4.948520660400391e-05,
      "step": 51889,
      "training_step_time": 8.44554591178894
    },
    {
      "epoch": 4.948616027832031e-05,
      "grad_norm": 0.15761032700538635,
      "learning_rate": 5.1010201318159555e-06,
      "loss": 0.0607,
      "step": 51890
    },
    {
      "epoch": 4.948616027832031e-05,
      "model_forward_time": 2.838456392288208,
      "step": 51890
    },
    {
      "dynamics_loss_avg": 0.08150683809071779,
      "epoch": 4.948616027832031e-05,
      "step": 51890
    },
    {
      "action_loss_avg": 0.003161462547723204,
      "epoch": 4.948616027832031e-05,
      "step": 51890
    },
    {
      "epoch": 4.948616027832031e-05,
      "step": 51890,
      "training_step_time": 7.652880430221558
    },
    {
      "epoch": 4.948711395263672e-05,
      "model_forward_time": 1.7995240688323975,
      "step": 51891
    },
    {
      "epoch": 4.948711395263672e-05,
      "step": 51891,
      "training_step_time": 8.342226028442383
    },
    {
      "epoch": 4.948806762695313e-05,
      "model_forward_time": 1.7457950115203857,
      "step": 51892
    },
    {
      "epoch": 4.948806762695313e-05,
      "step": 51892,
      "training_step_time": 8.736285209655762
    },
    {
      "epoch": 4.948902130126953e-05,
      "model_forward_time": 1.7446556091308594,
      "step": 51893
    },
    {
      "epoch": 4.948902130126953e-05,
      "step": 51893,
      "training_step_time": 7.702624320983887
    },
    {
      "epoch": 4.9489974975585936e-05,
      "model_forward_time": 1.7296290397644043,
      "step": 51894
    },
    {
      "epoch": 4.9489974975585936e-05,
      "step": 51894,
      "training_step_time": 7.5921525955200195
    },
    {
      "epoch": 4.949092864990235e-05,
      "model_forward_time": 1.7517006397247314,
      "step": 51895
    },
    {
      "epoch": 4.949092864990235e-05,
      "step": 51895,
      "training_step_time": 7.0070037841796875
    },
    {
      "epoch": 4.949188232421875e-05,
      "model_forward_time": 1.739197015762329,
      "step": 51896
    },
    {
      "epoch": 4.949188232421875e-05,
      "step": 51896,
      "training_step_time": 6.736457824707031
    },
    {
      "epoch": 4.9492835998535155e-05,
      "model_forward_time": 1.7366750240325928,
      "step": 51897
    },
    {
      "epoch": 4.9492835998535155e-05,
      "step": 51897,
      "training_step_time": 7.299470901489258
    },
    {
      "epoch": 4.9493789672851566e-05,
      "model_forward_time": 1.7299613952636719,
      "step": 51898
    },
    {
      "epoch": 4.9493789672851566e-05,
      "step": 51898,
      "training_step_time": 7.570804119110107
    },
    {
      "epoch": 4.949474334716797e-05,
      "model_forward_time": 2.8358633518218994,
      "step": 51899
    },
    {
      "epoch": 4.949474334716797e-05,
      "step": 51899,
      "training_step_time": 7.595633268356323
    },
    {
      "epoch": 4.9495697021484374e-05,
      "grad_norm": 0.18799781799316406,
      "learning_rate": 5.0993669940145564e-06,
      "loss": 0.0579,
      "step": 51900
    },
    {
      "epoch": 4.9495697021484374e-05,
      "model_forward_time": 1.8223247528076172,
      "step": 51900
    },
    {
      "dynamics_loss_avg": 0.05839464534074068,
      "epoch": 4.9495697021484374e-05,
      "step": 51900
    },
    {
      "action_loss_avg": 0.001655112986918539,
      "epoch": 4.9495697021484374e-05,
      "step": 51900
    },
    {
      "epoch": 4.9495697021484374e-05,
      "step": 51900,
      "training_step_time": 8.407136917114258
    },
    {
      "epoch": 4.9496650695800784e-05,
      "model_forward_time": 1.7715494632720947,
      "step": 51901
    },
    {
      "epoch": 4.9496650695800784e-05,
      "step": 51901,
      "training_step_time": 9.549360275268555
    },
    {
      "epoch": 4.949760437011719e-05,
      "model_forward_time": 1.7634518146514893,
      "step": 51902
    },
    {
      "epoch": 4.949760437011719e-05,
      "step": 51902,
      "training_step_time": 8.695027351379395
    },
    {
      "epoch": 4.949855804443359e-05,
      "model_forward_time": 1.766805648803711,
      "step": 51903
    },
    {
      "epoch": 4.949855804443359e-05,
      "step": 51903,
      "training_step_time": 7.797746181488037
    },
    {
      "epoch": 4.949951171875e-05,
      "model_forward_time": 1.769239902496338,
      "step": 51904
    },
    {
      "epoch": 4.949951171875e-05,
      "step": 51904,
      "training_step_time": 7.0059654712677
    },
    {
      "epoch": 4.950046539306641e-05,
      "model_forward_time": 1.7594659328460693,
      "step": 51905
    },
    {
      "epoch": 4.950046539306641e-05,
      "step": 51905,
      "training_step_time": 6.7243123054504395
    },
    {
      "epoch": 4.950141906738281e-05,
      "model_forward_time": 1.7606241703033447,
      "step": 51906
    },
    {
      "epoch": 4.950141906738281e-05,
      "step": 51906,
      "training_step_time": 7.27379584312439
    },
    {
      "epoch": 4.9502372741699215e-05,
      "model_forward_time": 1.769010066986084,
      "step": 51907
    },
    {
      "epoch": 4.9502372741699215e-05,
      "step": 51907,
      "training_step_time": 7.576064109802246
    },
    {
      "epoch": 4.9503326416015626e-05,
      "model_forward_time": 1.7678310871124268,
      "step": 51908
    },
    {
      "epoch": 4.9503326416015626e-05,
      "step": 51908,
      "training_step_time": 7.486137390136719
    },
    {
      "epoch": 4.950428009033203e-05,
      "model_forward_time": 2.9844536781311035,
      "step": 51909
    },
    {
      "epoch": 4.950428009033203e-05,
      "step": 51909,
      "training_step_time": 8.595983505249023
    },
    {
      "epoch": 4.9505233764648434e-05,
      "grad_norm": 0.12563663721084595,
      "learning_rate": 5.097713845346533e-06,
      "loss": 0.0609,
      "step": 51910
    },
    {
      "epoch": 4.9505233764648434e-05,
      "model_forward_time": 1.7669305801391602,
      "step": 51910
    },
    {
      "dynamics_loss_avg": 0.06194385662674904,
      "epoch": 4.9505233764648434e-05,
      "step": 51910
    },
    {
      "action_loss_avg": 0.0017709792969981208,
      "epoch": 4.9505233764648434e-05,
      "step": 51910
    },
    {
      "epoch": 4.9505233764648434e-05,
      "step": 51910,
      "training_step_time": 8.290647268295288
    },
    {
      "epoch": 4.9506187438964845e-05,
      "model_forward_time": 1.762814998626709,
      "step": 51911
    },
    {
      "epoch": 4.9506187438964845e-05,
      "step": 51911,
      "training_step_time": 7.774512052536011
    },
    {
      "epoch": 4.950714111328125e-05,
      "model_forward_time": 1.768770456314087,
      "step": 51912
    },
    {
      "epoch": 4.950714111328125e-05,
      "step": 51912,
      "training_step_time": 7.8771820068359375
    },
    {
      "epoch": 4.950809478759765e-05,
      "model_forward_time": 1.7616331577301025,
      "step": 51913
    },
    {
      "epoch": 4.950809478759765e-05,
      "step": 51913,
      "training_step_time": 7.685893297195435
    },
    {
      "epoch": 4.9509048461914064e-05,
      "model_forward_time": 1.7306416034698486,
      "step": 51914
    },
    {
      "epoch": 4.9509048461914064e-05,
      "step": 51914,
      "training_step_time": 6.84461236000061
    },
    {
      "epoch": 4.951000213623047e-05,
      "model_forward_time": 1.7505841255187988,
      "step": 51915
    },
    {
      "epoch": 4.951000213623047e-05,
      "step": 51915,
      "training_step_time": 7.419287443161011
    },
    {
      "epoch": 4.951095581054687e-05,
      "model_forward_time": 1.745800256729126,
      "step": 51916
    },
    {
      "epoch": 4.951095581054687e-05,
      "step": 51916,
      "training_step_time": 7.690490484237671
    },
    {
      "epoch": 4.951190948486328e-05,
      "model_forward_time": 1.745703935623169,
      "step": 51917
    },
    {
      "epoch": 4.951190948486328e-05,
      "step": 51917,
      "training_step_time": 7.557952404022217
    },
    {
      "epoch": 4.951286315917969e-05,
      "model_forward_time": 2.8704216480255127,
      "step": 51918
    },
    {
      "epoch": 4.951286315917969e-05,
      "step": 51918,
      "training_step_time": 8.267229080200195
    },
    {
      "epoch": 4.951381683349609e-05,
      "model_forward_time": 1.736487865447998,
      "step": 51919
    },
    {
      "epoch": 4.951381683349609e-05,
      "step": 51919,
      "training_step_time": 10.435988187789917
    },
    {
      "epoch": 4.95147705078125e-05,
      "grad_norm": 0.1263139247894287,
      "learning_rate": 5.096060685992667e-06,
      "loss": 0.0593,
      "step": 51920
    },
    {
      "epoch": 4.95147705078125e-05,
      "model_forward_time": 1.7376179695129395,
      "step": 51920
    },
    {
      "dynamics_loss_avg": 0.06782410200685263,
      "epoch": 4.95147705078125e-05,
      "step": 51920
    },
    {
      "action_loss_avg": 0.0022154408739879726,
      "epoch": 4.95147705078125e-05,
      "step": 51920
    },
    {
      "epoch": 4.95147705078125e-05,
      "step": 51920,
      "training_step_time": 7.784134149551392
    },
    {
      "epoch": 4.9515724182128906e-05,
      "model_forward_time": 1.7432260513305664,
      "step": 51921
    },
    {
      "epoch": 4.9515724182128906e-05,
      "step": 51921,
      "training_step_time": 7.825359106063843
    },
    {
      "epoch": 4.951667785644531e-05,
      "model_forward_time": 1.7718799114227295,
      "step": 51922
    },
    {
      "epoch": 4.951667785644531e-05,
      "step": 51922,
      "training_step_time": 7.69357442855835
    },
    {
      "epoch": 4.951763153076172e-05,
      "model_forward_time": 1.7378754615783691,
      "step": 51923
    },
    {
      "epoch": 4.951763153076172e-05,
      "step": 51923,
      "training_step_time": 7.078811883926392
    },
    {
      "epoch": 4.9518585205078125e-05,
      "model_forward_time": 1.7423267364501953,
      "step": 51924
    },
    {
      "epoch": 4.9518585205078125e-05,
      "step": 51924,
      "training_step_time": 7.410318374633789
    },
    {
      "epoch": 4.951953887939453e-05,
      "model_forward_time": 1.7491838932037354,
      "step": 51925
    },
    {
      "epoch": 4.951953887939453e-05,
      "step": 51925,
      "training_step_time": 8.382262706756592
    },
    {
      "epoch": 4.952049255371094e-05,
      "model_forward_time": 1.7391929626464844,
      "step": 51926
    },
    {
      "epoch": 4.952049255371094e-05,
      "step": 51926,
      "training_step_time": 7.552978038787842
    },
    {
      "epoch": 4.952144622802734e-05,
      "model_forward_time": 2.896594762802124,
      "step": 51927
    },
    {
      "epoch": 4.952144622802734e-05,
      "step": 51927,
      "training_step_time": 7.812024831771851
    },
    {
      "epoch": 4.952239990234375e-05,
      "model_forward_time": 1.818716049194336,
      "step": 51928
    },
    {
      "epoch": 4.952239990234375e-05,
      "step": 51928,
      "training_step_time": 8.709580421447754
    },
    {
      "epoch": 4.952335357666016e-05,
      "model_forward_time": 1.762451171875,
      "step": 51929
    },
    {
      "epoch": 4.952335357666016e-05,
      "step": 51929,
      "training_step_time": 7.860058784484863
    },
    {
      "epoch": 4.952430725097656e-05,
      "grad_norm": 0.10855825990438461,
      "learning_rate": 5.094407516133751e-06,
      "loss": 0.0626,
      "step": 51930
    },
    {
      "epoch": 4.952430725097656e-05,
      "model_forward_time": 1.770646095275879,
      "step": 51930
    },
    {
      "dynamics_loss_avg": 0.07334565073251724,
      "epoch": 4.952430725097656e-05,
      "step": 51930
    },
    {
      "action_loss_avg": 0.0022180978732649235,
      "epoch": 4.952430725097656e-05,
      "step": 51930
    },
    {
      "epoch": 4.952430725097656e-05,
      "step": 51930,
      "training_step_time": 7.741487264633179
    },
    {
      "epoch": 4.9525260925292966e-05,
      "model_forward_time": 1.7625658512115479,
      "step": 51931
    },
    {
      "epoch": 4.9525260925292966e-05,
      "step": 51931,
      "training_step_time": 7.777381896972656
    },
    {
      "epoch": 4.952621459960938e-05,
      "model_forward_time": 1.771407127380371,
      "step": 51932
    },
    {
      "epoch": 4.952621459960938e-05,
      "step": 51932,
      "training_step_time": 7.073523283004761
    },
    {
      "epoch": 4.952716827392578e-05,
      "model_forward_time": 1.767448902130127,
      "step": 51933
    },
    {
      "epoch": 4.952716827392578e-05,
      "step": 51933,
      "training_step_time": 7.366457939147949
    },
    {
      "epoch": 4.9528121948242185e-05,
      "model_forward_time": 1.7658836841583252,
      "step": 51934
    },
    {
      "epoch": 4.9528121948242185e-05,
      "step": 51934,
      "training_step_time": 7.675455331802368
    },
    {
      "epoch": 4.9529075622558596e-05,
      "model_forward_time": 1.722963571548462,
      "step": 51935
    },
    {
      "epoch": 4.9529075622558596e-05,
      "step": 51935,
      "training_step_time": 7.514133453369141
    },
    {
      "epoch": 4.9530029296875e-05,
      "model_forward_time": 2.8596041202545166,
      "step": 51936
    },
    {
      "epoch": 4.9530029296875e-05,
      "step": 51936,
      "training_step_time": 9.86194634437561
    },
    {
      "epoch": 4.9530982971191404e-05,
      "model_forward_time": 1.798537254333496,
      "step": 51937
    },
    {
      "epoch": 4.9530982971191404e-05,
      "step": 51937,
      "training_step_time": 8.316858768463135
    },
    {
      "epoch": 4.9531936645507815e-05,
      "model_forward_time": 1.7417690753936768,
      "step": 51938
    },
    {
      "epoch": 4.9531936645507815e-05,
      "step": 51938,
      "training_step_time": 8.814922571182251
    },
    {
      "epoch": 4.953289031982422e-05,
      "model_forward_time": 1.3809056282043457,
      "step": 51939
    },
    {
      "epoch": 4.953289031982422e-05,
      "step": 51939,
      "training_step_time": 8.43830132484436
    },
    {
      "epoch": 4.953384399414062e-05,
      "grad_norm": 0.09976199269294739,
      "learning_rate": 5.092754335950571e-06,
      "loss": 0.0571,
      "step": 51940
    },
    {
      "epoch": 4.953384399414062e-05,
      "model_forward_time": 1.7451777458190918,
      "step": 51940
    },
    {
      "dynamics_loss_avg": 0.059135600551962854,
      "epoch": 4.953384399414062e-05,
      "step": 51940
    },
    {
      "action_loss_avg": 0.0023079797043465077,
      "epoch": 4.953384399414062e-05,
      "step": 51940
    },
    {
      "epoch": 4.953384399414062e-05,
      "step": 51940,
      "training_step_time": 7.887930870056152
    },
    {
      "epoch": 4.9534797668457034e-05,
      "model_forward_time": 1.7477662563323975,
      "step": 51941
    },
    {
      "epoch": 4.9534797668457034e-05,
      "step": 51941,
      "training_step_time": 7.6714019775390625
    },
    {
      "epoch": 4.953575134277344e-05,
      "model_forward_time": 1.7294197082519531,
      "step": 51942
    },
    {
      "epoch": 4.953575134277344e-05,
      "step": 51942,
      "training_step_time": 7.37299108505249
    },
    {
      "epoch": 4.953670501708984e-05,
      "model_forward_time": 1.7322063446044922,
      "step": 51943
    },
    {
      "epoch": 4.953670501708984e-05,
      "step": 51943,
      "training_step_time": 8.692151069641113
    },
    {
      "epoch": 4.953765869140625e-05,
      "model_forward_time": 1.7643778324127197,
      "step": 51944
    },
    {
      "epoch": 4.953765869140625e-05,
      "step": 51944,
      "training_step_time": 8.286277055740356
    },
    {
      "epoch": 4.9538612365722657e-05,
      "model_forward_time": 1.73541259765625,
      "step": 51945
    },
    {
      "epoch": 4.9538612365722657e-05,
      "step": 51945,
      "training_step_time": 8.10618782043457
    },
    {
      "epoch": 4.953956604003906e-05,
      "model_forward_time": 1.7829673290252686,
      "step": 51946
    },
    {
      "epoch": 4.953956604003906e-05,
      "step": 51946,
      "training_step_time": 10.186441898345947
    },
    {
      "epoch": 4.954051971435547e-05,
      "model_forward_time": 1.741560459136963,
      "step": 51947
    },
    {
      "epoch": 4.954051971435547e-05,
      "step": 51947,
      "training_step_time": 8.838099241256714
    },
    {
      "epoch": 4.9541473388671875e-05,
      "model_forward_time": 1.7371222972869873,
      "step": 51948
    },
    {
      "epoch": 4.9541473388671875e-05,
      "step": 51948,
      "training_step_time": 10.2308669090271
    },
    {
      "epoch": 4.954242706298828e-05,
      "model_forward_time": 1.732412338256836,
      "step": 51949
    },
    {
      "epoch": 4.954242706298828e-05,
      "step": 51949,
      "training_step_time": 8.012109994888306
    },
    {
      "epoch": 4.954338073730469e-05,
      "grad_norm": 0.11458611488342285,
      "learning_rate": 5.091101145623916e-06,
      "loss": 0.0591,
      "step": 51950
    },
    {
      "epoch": 4.954338073730469e-05,
      "model_forward_time": 1.7376105785369873,
      "step": 51950
    },
    {
      "dynamics_loss_avg": 0.0634861659258604,
      "epoch": 4.954338073730469e-05,
      "step": 51950
    },
    {
      "action_loss_avg": 0.0016491596616106108,
      "epoch": 4.954338073730469e-05,
      "step": 51950
    },
    {
      "epoch": 4.954338073730469e-05,
      "step": 51950,
      "training_step_time": 8.386541604995728
    },
    {
      "epoch": 4.9544334411621094e-05,
      "model_forward_time": 1.7365708351135254,
      "step": 51951
    },
    {
      "epoch": 4.9544334411621094e-05,
      "step": 51951,
      "training_step_time": 7.428533315658569
    },
    {
      "epoch": 4.95452880859375e-05,
      "model_forward_time": 1.746248722076416,
      "step": 51952
    },
    {
      "epoch": 4.95452880859375e-05,
      "step": 51952,
      "training_step_time": 7.671682357788086
    },
    {
      "epoch": 4.954624176025391e-05,
      "model_forward_time": 1.7384283542633057,
      "step": 51953
    },
    {
      "epoch": 4.954624176025391e-05,
      "step": 51953,
      "training_step_time": 7.333756446838379
    },
    {
      "epoch": 4.954719543457031e-05,
      "model_forward_time": 1.734837532043457,
      "step": 51954
    },
    {
      "epoch": 4.954719543457031e-05,
      "step": 51954,
      "training_step_time": 7.5896852016448975
    },
    {
      "epoch": 4.954814910888672e-05,
      "model_forward_time": 2.952484130859375,
      "step": 51955
    },
    {
      "epoch": 4.954814910888672e-05,
      "step": 51955,
      "training_step_time": 9.223658084869385
    },
    {
      "epoch": 4.954910278320313e-05,
      "model_forward_time": 1.7389531135559082,
      "step": 51956
    },
    {
      "epoch": 4.954910278320313e-05,
      "step": 51956,
      "training_step_time": 10.123384952545166
    },
    {
      "epoch": 4.955005645751953e-05,
      "model_forward_time": 1.7367033958435059,
      "step": 51957
    },
    {
      "epoch": 4.955005645751953e-05,
      "step": 51957,
      "training_step_time": 8.858604431152344
    },
    {
      "epoch": 4.9551010131835936e-05,
      "model_forward_time": 1.739891767501831,
      "step": 51958
    },
    {
      "epoch": 4.9551010131835936e-05,
      "step": 51958,
      "training_step_time": 7.915626287460327
    },
    {
      "epoch": 4.955196380615235e-05,
      "model_forward_time": 1.3880839347839355,
      "step": 51959
    },
    {
      "epoch": 4.955196380615235e-05,
      "step": 51959,
      "training_step_time": 8.3557608127594
    },
    {
      "epoch": 4.955291748046875e-05,
      "grad_norm": 0.12836012244224548,
      "learning_rate": 5.089447945334574e-06,
      "loss": 0.0573,
      "step": 51960
    },
    {
      "epoch": 4.955291748046875e-05,
      "model_forward_time": 1.7444486618041992,
      "step": 51960
    },
    {
      "dynamics_loss_avg": 0.06532427947968245,
      "epoch": 4.955291748046875e-05,
      "step": 51960
    },
    {
      "action_loss_avg": 0.0030970426741987467,
      "epoch": 4.955291748046875e-05,
      "step": 51960
    },
    {
      "epoch": 4.955291748046875e-05,
      "step": 51960,
      "training_step_time": 7.411287546157837
    },
    {
      "epoch": 4.9553871154785155e-05,
      "model_forward_time": 1.740541696548462,
      "step": 51961
    },
    {
      "epoch": 4.9553871154785155e-05,
      "step": 51961,
      "training_step_time": 7.5690391063690186
    },
    {
      "epoch": 4.9554824829101566e-05,
      "model_forward_time": 1.7291736602783203,
      "step": 51962
    },
    {
      "epoch": 4.9554824829101566e-05,
      "step": 51962,
      "training_step_time": 7.2758097648620605
    },
    {
      "epoch": 4.955577850341797e-05,
      "model_forward_time": 1.7397136688232422,
      "step": 51963
    },
    {
      "epoch": 4.955577850341797e-05,
      "step": 51963,
      "training_step_time": 8.576748609542847
    },
    {
      "epoch": 4.9556732177734374e-05,
      "model_forward_time": 2.8616116046905518,
      "step": 51964
    },
    {
      "epoch": 4.9556732177734374e-05,
      "step": 51964,
      "training_step_time": 7.697187900543213
    },
    {
      "epoch": 4.9557685852050785e-05,
      "model_forward_time": 1.8065543174743652,
      "step": 51965
    },
    {
      "epoch": 4.9557685852050785e-05,
      "step": 51965,
      "training_step_time": 9.207569122314453
    },
    {
      "epoch": 4.955863952636719e-05,
      "model_forward_time": 1.7527508735656738,
      "step": 51966
    },
    {
      "epoch": 4.955863952636719e-05,
      "step": 51966,
      "training_step_time": 7.829950571060181
    },
    {
      "epoch": 4.955959320068359e-05,
      "model_forward_time": 1.7307243347167969,
      "step": 51967
    },
    {
      "epoch": 4.955959320068359e-05,
      "step": 51967,
      "training_step_time": 8.376441955566406
    },
    {
      "epoch": 4.9560546875e-05,
      "model_forward_time": 1.738729476928711,
      "step": 51968
    },
    {
      "epoch": 4.9560546875e-05,
      "step": 51968,
      "training_step_time": 7.652260065078735
    },
    {
      "epoch": 4.956150054931641e-05,
      "model_forward_time": 1.7509191036224365,
      "step": 51969
    },
    {
      "epoch": 4.956150054931641e-05,
      "step": 51969,
      "training_step_time": 7.581918716430664
    },
    {
      "epoch": 4.956245422363281e-05,
      "grad_norm": 0.10121279209852219,
      "learning_rate": 5.087794735263342e-06,
      "loss": 0.0598,
      "step": 51970
    },
    {
      "epoch": 4.956245422363281e-05,
      "model_forward_time": 1.7335712909698486,
      "step": 51970
    },
    {
      "dynamics_loss_avg": 0.06514461543411017,
      "epoch": 4.956245422363281e-05,
      "step": 51970
    },
    {
      "action_loss_avg": 0.0023356467514531686,
      "epoch": 4.956245422363281e-05,
      "step": 51970
    },
    {
      "epoch": 4.956245422363281e-05,
      "step": 51970,
      "training_step_time": 6.853616714477539
    },
    {
      "epoch": 4.9563407897949215e-05,
      "model_forward_time": 1.7453746795654297,
      "step": 51971
    },
    {
      "epoch": 4.9563407897949215e-05,
      "step": 51971,
      "training_step_time": 7.618805408477783
    },
    {
      "epoch": 4.9564361572265626e-05,
      "model_forward_time": 1.7456731796264648,
      "step": 51972
    },
    {
      "epoch": 4.9564361572265626e-05,
      "step": 51972,
      "training_step_time": 8.559022903442383
    },
    {
      "epoch": 4.956531524658203e-05,
      "model_forward_time": 2.8438830375671387,
      "step": 51973
    },
    {
      "epoch": 4.956531524658203e-05,
      "step": 51973,
      "training_step_time": 7.665006875991821
    },
    {
      "epoch": 4.9566268920898434e-05,
      "model_forward_time": 1.8217661380767822,
      "step": 51974
    },
    {
      "epoch": 4.9566268920898434e-05,
      "step": 51974,
      "training_step_time": 9.562122821807861
    },
    {
      "epoch": 4.9567222595214845e-05,
      "model_forward_time": 1.7403383255004883,
      "step": 51975
    },
    {
      "epoch": 4.9567222595214845e-05,
      "step": 51975,
      "training_step_time": 9.35763168334961
    },
    {
      "epoch": 4.956817626953125e-05,
      "model_forward_time": 1.7517781257629395,
      "step": 51976
    },
    {
      "epoch": 4.956817626953125e-05,
      "step": 51976,
      "training_step_time": 9.807865142822266
    },
    {
      "epoch": 4.956912994384765e-05,
      "model_forward_time": 1.7363393306732178,
      "step": 51977
    },
    {
      "epoch": 4.956912994384765e-05,
      "step": 51977,
      "training_step_time": 7.6304731369018555
    },
    {
      "epoch": 4.9570083618164064e-05,
      "model_forward_time": 1.7502357959747314,
      "step": 51978
    },
    {
      "epoch": 4.9570083618164064e-05,
      "step": 51978,
      "training_step_time": 7.717648267745972
    },
    {
      "epoch": 4.957103729248047e-05,
      "model_forward_time": 1.7365736961364746,
      "step": 51979
    },
    {
      "epoch": 4.957103729248047e-05,
      "step": 51979,
      "training_step_time": 6.739439487457275
    },
    {
      "epoch": 4.957199096679687e-05,
      "grad_norm": 0.12689274549484253,
      "learning_rate": 5.086141515591011e-06,
      "loss": 0.0581,
      "step": 51980
    },
    {
      "epoch": 4.957199096679687e-05,
      "model_forward_time": 1.7460627555847168,
      "step": 51980
    },
    {
      "dynamics_loss_avg": 0.08640829361975193,
      "epoch": 4.957199096679687e-05,
      "step": 51980
    },
    {
      "action_loss_avg": 0.0049269288312643765,
      "epoch": 4.957199096679687e-05,
      "step": 51980
    },
    {
      "epoch": 4.957199096679687e-05,
      "step": 51980,
      "training_step_time": 8.284124851226807
    },
    {
      "epoch": 4.957294464111328e-05,
      "model_forward_time": 1.7376644611358643,
      "step": 51981
    },
    {
      "epoch": 4.957294464111328e-05,
      "step": 51981,
      "training_step_time": 7.576683521270752
    },
    {
      "epoch": 4.957389831542969e-05,
      "model_forward_time": 1.7310349941253662,
      "step": 51982
    },
    {
      "epoch": 4.957389831542969e-05,
      "step": 51982,
      "training_step_time": 9.261469841003418
    },
    {
      "epoch": 4.957485198974609e-05,
      "model_forward_time": 1.794658899307251,
      "step": 51983
    },
    {
      "epoch": 4.957485198974609e-05,
      "step": 51983,
      "training_step_time": 8.528129816055298
    },
    {
      "epoch": 4.95758056640625e-05,
      "model_forward_time": 1.7328031063079834,
      "step": 51984
    },
    {
      "epoch": 4.95758056640625e-05,
      "step": 51984,
      "training_step_time": 8.781578302383423
    },
    {
      "epoch": 4.9576759338378906e-05,
      "model_forward_time": 1.7426352500915527,
      "step": 51985
    },
    {
      "epoch": 4.9576759338378906e-05,
      "step": 51985,
      "training_step_time": 8.366693258285522
    },
    {
      "epoch": 4.957771301269531e-05,
      "model_forward_time": 1.7535791397094727,
      "step": 51986
    },
    {
      "epoch": 4.957771301269531e-05,
      "step": 51986,
      "training_step_time": 7.893023252487183
    },
    {
      "epoch": 4.957866668701172e-05,
      "model_forward_time": 1.7529315948486328,
      "step": 51987
    },
    {
      "epoch": 4.957866668701172e-05,
      "step": 51987,
      "training_step_time": 7.550533771514893
    },
    {
      "epoch": 4.9579620361328125e-05,
      "model_forward_time": 1.7380499839782715,
      "step": 51988
    },
    {
      "epoch": 4.9579620361328125e-05,
      "step": 51988,
      "training_step_time": 7.3133156299591064
    },
    {
      "epoch": 4.958057403564453e-05,
      "model_forward_time": 1.7486932277679443,
      "step": 51989
    },
    {
      "epoch": 4.958057403564453e-05,
      "step": 51989,
      "training_step_time": 7.665742874145508
    },
    {
      "epoch": 4.958152770996094e-05,
      "grad_norm": 0.17907559871673584,
      "learning_rate": 5.084488286498372e-06,
      "loss": 0.0601,
      "step": 51990
    },
    {
      "epoch": 4.958152770996094e-05,
      "model_forward_time": 1.7430250644683838,
      "step": 51990
    },
    {
      "dynamics_loss_avg": 0.06186882741749287,
      "epoch": 4.958152770996094e-05,
      "step": 51990
    },
    {
      "action_loss_avg": 0.002595308009767905,
      "epoch": 4.958152770996094e-05,
      "step": 51990
    },
    {
      "epoch": 4.958152770996094e-05,
      "step": 51990,
      "training_step_time": 7.501535415649414
    },
    {
      "epoch": 4.9582481384277343e-05,
      "model_forward_time": 1.7547614574432373,
      "step": 51991
    },
    {
      "epoch": 4.9582481384277343e-05,
      "step": 51991,
      "training_step_time": 9.889285564422607
    },
    {
      "epoch": 4.958343505859375e-05,
      "model_forward_time": 2.9840168952941895,
      "step": 51992
    },
    {
      "epoch": 4.958343505859375e-05,
      "step": 51992,
      "training_step_time": 8.629055738449097
    },
    {
      "epoch": 4.958438873291016e-05,
      "model_forward_time": 1.7449381351470947,
      "step": 51993
    },
    {
      "epoch": 4.958438873291016e-05,
      "step": 51993,
      "training_step_time": 9.620863437652588
    },
    {
      "epoch": 4.958534240722656e-05,
      "model_forward_time": 1.7403507232666016,
      "step": 51994
    },
    {
      "epoch": 4.958534240722656e-05,
      "step": 51994,
      "training_step_time": 7.695416450500488
    },
    {
      "epoch": 4.9586296081542966e-05,
      "model_forward_time": 1.7343249320983887,
      "step": 51995
    },
    {
      "epoch": 4.9586296081542966e-05,
      "step": 51995,
      "training_step_time": 8.597522258758545
    },
    {
      "epoch": 4.958724975585938e-05,
      "model_forward_time": 1.732846736907959,
      "step": 51996
    },
    {
      "epoch": 4.958724975585938e-05,
      "step": 51996,
      "training_step_time": 8.285082817077637
    },
    {
      "epoch": 4.958820343017578e-05,
      "model_forward_time": 1.73907470703125,
      "step": 51997
    },
    {
      "epoch": 4.958820343017578e-05,
      "step": 51997,
      "training_step_time": 7.300192356109619
    },
    {
      "epoch": 4.9589157104492185e-05,
      "model_forward_time": 1.805058479309082,
      "step": 51998
    },
    {
      "epoch": 4.9589157104492185e-05,
      "step": 51998,
      "training_step_time": 7.510179758071899
    },
    {
      "epoch": 4.9590110778808596e-05,
      "model_forward_time": 1.7789828777313232,
      "step": 51999
    },
    {
      "epoch": 4.9590110778808596e-05,
      "step": 51999,
      "training_step_time": 7.4688639640808105
    },
    {
      "epoch": 4.9591064453125e-05,
      "grad_norm": 0.13985319435596466,
      "learning_rate": 5.08283504816622e-06,
      "loss": 0.062,
      "step": 52000
    },
    {
      "epoch": 4.9591064453125e-05,
      "model_forward_time": 3.234393835067749,
      "step": 52000
    },
    {
      "dynamics_loss_avg": 0.054794187843799594,
      "epoch": 4.9591064453125e-05,
      "step": 52000
    },
    {
      "action_loss_avg": 0.001442260801559314,
      "epoch": 4.9591064453125e-05,
      "step": 52000
    },
    {
      "epoch": 4.9591064453125e-05,
      "step": 52000,
      "training_step_time": 11.339149475097656
    },
    {
      "epoch": 4.9592018127441404e-05,
      "model_forward_time": 1.748847484588623,
      "step": 52001
    },
    {
      "epoch": 4.9592018127441404e-05,
      "step": 52001,
      "training_step_time": 9.385860919952393
    },
    {
      "epoch": 4.9592971801757815e-05,
      "model_forward_time": 1.747455358505249,
      "step": 52002
    },
    {
      "epoch": 4.9592971801757815e-05,
      "step": 52002,
      "training_step_time": 11.9677734375
    },
    {
      "epoch": 4.959392547607422e-05,
      "model_forward_time": 1.7344367504119873,
      "step": 52003
    },
    {
      "epoch": 4.959392547607422e-05,
      "step": 52003,
      "training_step_time": 9.74832534790039
    },
    {
      "epoch": 4.959487915039062e-05,
      "model_forward_time": 1.7409260272979736,
      "step": 52004
    },
    {
      "epoch": 4.959487915039062e-05,
      "step": 52004,
      "training_step_time": 8.571906805038452
    },
    {
      "epoch": 4.9595832824707034e-05,
      "model_forward_time": 1.742004156112671,
      "step": 52005
    },
    {
      "epoch": 4.9595832824707034e-05,
      "step": 52005,
      "training_step_time": 7.502936840057373
    },
    {
      "epoch": 4.959678649902344e-05,
      "model_forward_time": 1.7503719329833984,
      "step": 52006
    },
    {
      "epoch": 4.959678649902344e-05,
      "step": 52006,
      "training_step_time": 7.550886392593384
    },
    {
      "epoch": 4.959774017333984e-05,
      "model_forward_time": 1.7310454845428467,
      "step": 52007
    },
    {
      "epoch": 4.959774017333984e-05,
      "step": 52007,
      "training_step_time": 8.6434907913208
    },
    {
      "epoch": 4.959869384765625e-05,
      "model_forward_time": 1.7348434925079346,
      "step": 52008
    },
    {
      "epoch": 4.959869384765625e-05,
      "step": 52008,
      "training_step_time": 8.27365779876709
    },
    {
      "epoch": 4.9599647521972657e-05,
      "model_forward_time": 2.7985711097717285,
      "step": 52009
    },
    {
      "epoch": 4.9599647521972657e-05,
      "step": 52009,
      "training_step_time": 8.872107744216919
    },
    {
      "epoch": 4.960060119628906e-05,
      "grad_norm": 0.16764962673187256,
      "learning_rate": 5.081181800775353e-06,
      "loss": 0.0575,
      "step": 52010
    },
    {
      "epoch": 4.960060119628906e-05,
      "model_forward_time": 1.7423839569091797,
      "step": 52010
    },
    {
      "dynamics_loss_avg": 0.048536678217351435,
      "epoch": 4.960060119628906e-05,
      "step": 52010
    },
    {
      "action_loss_avg": 0.0016493946022819728,
      "epoch": 4.960060119628906e-05,
      "step": 52010
    },
    {
      "epoch": 4.960060119628906e-05,
      "step": 52010,
      "training_step_time": 8.82906174659729
    },
    {
      "epoch": 4.960155487060547e-05,
      "model_forward_time": 1.8046915531158447,
      "step": 52011
    },
    {
      "epoch": 4.960155487060547e-05,
      "step": 52011,
      "training_step_time": 10.304898500442505
    },
    {
      "epoch": 4.9602508544921875e-05,
      "model_forward_time": 1.731558084487915,
      "step": 52012
    },
    {
      "epoch": 4.9602508544921875e-05,
      "step": 52012,
      "training_step_time": 7.753430604934692
    },
    {
      "epoch": 4.960346221923828e-05,
      "model_forward_time": 1.741086483001709,
      "step": 52013
    },
    {
      "epoch": 4.960346221923828e-05,
      "step": 52013,
      "training_step_time": 7.665844440460205
    },
    {
      "epoch": 4.960441589355469e-05,
      "model_forward_time": 1.751065969467163,
      "step": 52014
    },
    {
      "epoch": 4.960441589355469e-05,
      "step": 52014,
      "training_step_time": 7.655497312545776
    },
    {
      "epoch": 4.9605369567871094e-05,
      "model_forward_time": 1.739201307296753,
      "step": 52015
    },
    {
      "epoch": 4.9605369567871094e-05,
      "step": 52015,
      "training_step_time": 7.565490007400513
    },
    {
      "epoch": 4.96063232421875e-05,
      "model_forward_time": 1.384009599685669,
      "step": 52016
    },
    {
      "epoch": 4.96063232421875e-05,
      "step": 52016,
      "training_step_time": 7.792311668395996
    },
    {
      "epoch": 4.960727691650391e-05,
      "model_forward_time": 1.7395496368408203,
      "step": 52017
    },
    {
      "epoch": 4.960727691650391e-05,
      "step": 52017,
      "training_step_time": 7.565727710723877
    },
    {
      "epoch": 4.960823059082031e-05,
      "model_forward_time": 1.7456884384155273,
      "step": 52018
    },
    {
      "epoch": 4.960823059082031e-05,
      "step": 52018,
      "training_step_time": 7.494020938873291
    },
    {
      "epoch": 4.960918426513672e-05,
      "model_forward_time": 2.927142381668091,
      "step": 52019
    },
    {
      "epoch": 4.960918426513672e-05,
      "step": 52019,
      "training_step_time": 8.620025873184204
    },
    {
      "epoch": 4.961013793945313e-05,
      "grad_norm": 0.1236577257514,
      "learning_rate": 5.079528544506566e-06,
      "loss": 0.0582,
      "step": 52020
    },
    {
      "epoch": 4.961013793945313e-05,
      "model_forward_time": 1.8101093769073486,
      "step": 52020
    },
    {
      "dynamics_loss_avg": 0.056045131757855415,
      "epoch": 4.961013793945313e-05,
      "step": 52020
    },
    {
      "action_loss_avg": 0.0017369060369674116,
      "epoch": 4.961013793945313e-05,
      "step": 52020
    },
    {
      "epoch": 4.961013793945313e-05,
      "step": 52020,
      "training_step_time": 8.525892734527588
    },
    {
      "epoch": 4.961109161376953e-05,
      "model_forward_time": 1.7450141906738281,
      "step": 52021
    },
    {
      "epoch": 4.961109161376953e-05,
      "step": 52021,
      "training_step_time": 7.749429941177368
    },
    {
      "epoch": 4.9612045288085936e-05,
      "model_forward_time": 1.7384943962097168,
      "step": 52022
    },
    {
      "epoch": 4.9612045288085936e-05,
      "step": 52022,
      "training_step_time": 7.6733558177948
    },
    {
      "epoch": 4.961299896240235e-05,
      "model_forward_time": 1.7381055355072021,
      "step": 52023
    },
    {
      "epoch": 4.961299896240235e-05,
      "step": 52023,
      "training_step_time": 7.550225734710693
    },
    {
      "epoch": 4.961395263671875e-05,
      "model_forward_time": 1.7396879196166992,
      "step": 52024
    },
    {
      "epoch": 4.961395263671875e-05,
      "step": 52024,
      "training_step_time": 7.465646028518677
    },
    {
      "epoch": 4.9614906311035155e-05,
      "model_forward_time": 1.7419452667236328,
      "step": 52025
    },
    {
      "epoch": 4.9614906311035155e-05,
      "step": 52025,
      "training_step_time": 7.178070068359375
    },
    {
      "epoch": 4.9615859985351566e-05,
      "model_forward_time": 1.735917568206787,
      "step": 52026
    },
    {
      "epoch": 4.9615859985351566e-05,
      "step": 52026,
      "training_step_time": 7.467589378356934
    },
    {
      "epoch": 4.961681365966797e-05,
      "model_forward_time": 1.7353556156158447,
      "step": 52027
    },
    {
      "epoch": 4.961681365966797e-05,
      "step": 52027,
      "training_step_time": 8.161559820175171
    },
    {
      "epoch": 4.9617767333984374e-05,
      "model_forward_time": 2.892085313796997,
      "step": 52028
    },
    {
      "epoch": 4.9617767333984374e-05,
      "step": 52028,
      "training_step_time": 7.629671573638916
    },
    {
      "epoch": 4.9618721008300785e-05,
      "model_forward_time": 1.8026306629180908,
      "step": 52029
    },
    {
      "epoch": 4.9618721008300785e-05,
      "step": 52029,
      "training_step_time": 7.577699422836304
    },
    {
      "epoch": 4.961967468261719e-05,
      "grad_norm": 0.10947049409151077,
      "learning_rate": 5.077875279540657e-06,
      "loss": 0.0625,
      "step": 52030
    },
    {
      "epoch": 4.961967468261719e-05,
      "model_forward_time": 1.7427849769592285,
      "step": 52030
    },
    {
      "dynamics_loss_avg": 0.04890483561903238,
      "epoch": 4.961967468261719e-05,
      "step": 52030
    },
    {
      "action_loss_avg": 0.0017238394473679363,
      "epoch": 4.961967468261719e-05,
      "step": 52030
    },
    {
      "epoch": 4.961967468261719e-05,
      "step": 52030,
      "training_step_time": 7.594415664672852
    },
    {
      "epoch": 4.962062835693359e-05,
      "model_forward_time": 1.7435989379882812,
      "step": 52031
    },
    {
      "epoch": 4.962062835693359e-05,
      "step": 52031,
      "training_step_time": 8.628669261932373
    },
    {
      "epoch": 4.962158203125e-05,
      "model_forward_time": 1.7483198642730713,
      "step": 52032
    },
    {
      "epoch": 4.962158203125e-05,
      "step": 52032,
      "training_step_time": 7.817301988601685
    },
    {
      "epoch": 4.962253570556641e-05,
      "model_forward_time": 1.7454991340637207,
      "step": 52033
    },
    {
      "epoch": 4.962253570556641e-05,
      "step": 52033,
      "training_step_time": 7.605175495147705
    },
    {
      "epoch": 4.962348937988281e-05,
      "model_forward_time": 1.7393970489501953,
      "step": 52034
    },
    {
      "epoch": 4.962348937988281e-05,
      "step": 52034,
      "training_step_time": 7.316355228424072
    },
    {
      "epoch": 4.9624443054199215e-05,
      "model_forward_time": 1.7395074367523193,
      "step": 52035
    },
    {
      "epoch": 4.9624443054199215e-05,
      "step": 52035,
      "training_step_time": 7.534233570098877
    },
    {
      "epoch": 4.9625396728515626e-05,
      "model_forward_time": 1.7515537738800049,
      "step": 52036
    },
    {
      "epoch": 4.9625396728515626e-05,
      "step": 52036,
      "training_step_time": 8.515599727630615
    },
    {
      "epoch": 4.962635040283203e-05,
      "model_forward_time": 2.9364726543426514,
      "step": 52037
    },
    {
      "epoch": 4.962635040283203e-05,
      "step": 52037,
      "training_step_time": 8.327752828598022
    },
    {
      "epoch": 4.9627304077148434e-05,
      "model_forward_time": 1.8221826553344727,
      "step": 52038
    },
    {
      "epoch": 4.9627304077148434e-05,
      "step": 52038,
      "training_step_time": 8.469562768936157
    },
    {
      "epoch": 4.9628257751464845e-05,
      "model_forward_time": 1.7655136585235596,
      "step": 52039
    },
    {
      "epoch": 4.9628257751464845e-05,
      "step": 52039,
      "training_step_time": 7.66718602180481
    },
    {
      "epoch": 4.962921142578125e-05,
      "grad_norm": 0.09989569336175919,
      "learning_rate": 5.076222006058424e-06,
      "loss": 0.0571,
      "step": 52040
    },
    {
      "epoch": 4.962921142578125e-05,
      "model_forward_time": 1.7648048400878906,
      "step": 52040
    },
    {
      "dynamics_loss_avg": 0.0539530660957098,
      "epoch": 4.962921142578125e-05,
      "step": 52040
    },
    {
      "action_loss_avg": 0.002034702029777691,
      "epoch": 4.962921142578125e-05,
      "step": 52040
    },
    {
      "epoch": 4.962921142578125e-05,
      "step": 52040,
      "training_step_time": 7.788479804992676
    },
    {
      "epoch": 4.963016510009765e-05,
      "model_forward_time": 1.7784364223480225,
      "step": 52041
    },
    {
      "epoch": 4.963016510009765e-05,
      "step": 52041,
      "training_step_time": 8.687839031219482
    },
    {
      "epoch": 4.9631118774414064e-05,
      "model_forward_time": 1.761528491973877,
      "step": 52042
    },
    {
      "epoch": 4.9631118774414064e-05,
      "step": 52042,
      "training_step_time": 7.627386569976807
    },
    {
      "epoch": 4.963207244873047e-05,
      "model_forward_time": 1.7560861110687256,
      "step": 52043
    },
    {
      "epoch": 4.963207244873047e-05,
      "step": 52043,
      "training_step_time": 7.369554758071899
    },
    {
      "epoch": 4.963302612304687e-05,
      "model_forward_time": 1.7628657817840576,
      "step": 52044
    },
    {
      "epoch": 4.963302612304687e-05,
      "step": 52044,
      "training_step_time": 7.5700366497039795
    },
    {
      "epoch": 4.963397979736328e-05,
      "model_forward_time": 1.7570059299468994,
      "step": 52045
    },
    {
      "epoch": 4.963397979736328e-05,
      "step": 52045,
      "training_step_time": 8.320516109466553
    },
    {
      "epoch": 4.963493347167969e-05,
      "model_forward_time": 2.897554636001587,
      "step": 52046
    },
    {
      "epoch": 4.963493347167969e-05,
      "step": 52046,
      "training_step_time": 7.727420091629028
    },
    {
      "epoch": 4.963588714599609e-05,
      "model_forward_time": 1.8243298530578613,
      "step": 52047
    },
    {
      "epoch": 4.963588714599609e-05,
      "step": 52047,
      "training_step_time": 8.557876825332642
    },
    {
      "epoch": 4.96368408203125e-05,
      "model_forward_time": 1.753319501876831,
      "step": 52048
    },
    {
      "epoch": 4.96368408203125e-05,
      "step": 52048,
      "training_step_time": 7.719880104064941
    },
    {
      "epoch": 4.9637794494628906e-05,
      "model_forward_time": 1.771092176437378,
      "step": 52049
    },
    {
      "epoch": 4.9637794494628906e-05,
      "step": 52049,
      "training_step_time": 8.705771207809448
    },
    {
      "epoch": 4.963874816894531e-05,
      "grad_norm": 0.11015887558460236,
      "learning_rate": 5.074568724240667e-06,
      "loss": 0.0615,
      "step": 52050
    },
    {
      "epoch": 4.963874816894531e-05,
      "model_forward_time": 1.7637269496917725,
      "step": 52050
    },
    {
      "dynamics_loss_avg": 0.05132490042597056,
      "epoch": 4.963874816894531e-05,
      "step": 52050
    },
    {
      "action_loss_avg": 0.001657029625494033,
      "epoch": 4.963874816894531e-05,
      "step": 52050
    },
    {
      "epoch": 4.963874816894531e-05,
      "step": 52050,
      "training_step_time": 7.71656346321106
    },
    {
      "epoch": 4.963970184326172e-05,
      "model_forward_time": 1.7638771533966064,
      "step": 52051
    },
    {
      "epoch": 4.963970184326172e-05,
      "step": 52051,
      "training_step_time": 8.185662269592285
    },
    {
      "epoch": 4.9640655517578125e-05,
      "model_forward_time": 1.7647955417633057,
      "step": 52052
    },
    {
      "epoch": 4.9640655517578125e-05,
      "step": 52052,
      "training_step_time": 7.25703239440918
    },
    {
      "epoch": 4.964160919189453e-05,
      "model_forward_time": 1.765747308731079,
      "step": 52053
    },
    {
      "epoch": 4.964160919189453e-05,
      "step": 52053,
      "training_step_time": 7.372264385223389
    },
    {
      "epoch": 4.964256286621094e-05,
      "model_forward_time": 1.759150743484497,
      "step": 52054
    },
    {
      "epoch": 4.964256286621094e-05,
      "step": 52054,
      "training_step_time": 7.265638589859009
    },
    {
      "epoch": 4.9643516540527343e-05,
      "model_forward_time": 2.84598970413208,
      "step": 52055
    },
    {
      "epoch": 4.9643516540527343e-05,
      "step": 52055,
      "training_step_time": 7.56075119972229
    },
    {
      "epoch": 4.964447021484375e-05,
      "model_forward_time": 1.8328897953033447,
      "step": 52056
    },
    {
      "epoch": 4.964447021484375e-05,
      "step": 52056,
      "training_step_time": 8.659693717956543
    },
    {
      "epoch": 4.964542388916016e-05,
      "model_forward_time": 1.7690269947052002,
      "step": 52057
    },
    {
      "epoch": 4.964542388916016e-05,
      "step": 52057,
      "training_step_time": 8.401738405227661
    },
    {
      "epoch": 4.964637756347656e-05,
      "model_forward_time": 1.7649741172790527,
      "step": 52058
    },
    {
      "epoch": 4.964637756347656e-05,
      "step": 52058,
      "training_step_time": 8.608898878097534
    },
    {
      "epoch": 4.9647331237792966e-05,
      "model_forward_time": 1.761669635772705,
      "step": 52059
    },
    {
      "epoch": 4.9647331237792966e-05,
      "step": 52059,
      "training_step_time": 8.867280721664429
    },
    {
      "epoch": 4.964828491210938e-05,
      "grad_norm": 0.11721228063106537,
      "learning_rate": 5.072915434268188e-06,
      "loss": 0.0612,
      "step": 52060
    },
    {
      "epoch": 4.964828491210938e-05,
      "model_forward_time": 1.761063814163208,
      "step": 52060
    },
    {
      "dynamics_loss_avg": 0.061917076818645,
      "epoch": 4.964828491210938e-05,
      "step": 52060
    },
    {
      "action_loss_avg": 0.0015107460727449506,
      "epoch": 4.964828491210938e-05,
      "step": 52060
    },
    {
      "epoch": 4.964828491210938e-05,
      "step": 52060,
      "training_step_time": 7.5049543380737305
    },
    {
      "epoch": 4.964923858642578e-05,
      "model_forward_time": 1.7669486999511719,
      "step": 52061
    },
    {
      "epoch": 4.964923858642578e-05,
      "step": 52061,
      "training_step_time": 7.578783273696899
    },
    {
      "epoch": 4.9650192260742185e-05,
      "model_forward_time": 1.4004271030426025,
      "step": 52062
    },
    {
      "epoch": 4.9650192260742185e-05,
      "step": 52062,
      "training_step_time": 7.302197456359863
    },
    {
      "epoch": 4.9651145935058596e-05,
      "model_forward_time": 1.7663662433624268,
      "step": 52063
    },
    {
      "epoch": 4.9651145935058596e-05,
      "step": 52063,
      "training_step_time": 7.481709957122803
    },
    {
      "epoch": 4.9652099609375e-05,
      "model_forward_time": 1.7649869918823242,
      "step": 52064
    },
    {
      "epoch": 4.9652099609375e-05,
      "step": 52064,
      "training_step_time": 8.19106936454773
    },
    {
      "epoch": 4.9653053283691404e-05,
      "model_forward_time": 2.9686713218688965,
      "step": 52065
    },
    {
      "epoch": 4.9653053283691404e-05,
      "step": 52065,
      "training_step_time": 8.369506597518921
    },
    {
      "epoch": 4.9654006958007815e-05,
      "model_forward_time": 1.8287131786346436,
      "step": 52066
    },
    {
      "epoch": 4.9654006958007815e-05,
      "step": 52066,
      "training_step_time": 9.030966758728027
    },
    {
      "epoch": 4.965496063232422e-05,
      "model_forward_time": 1.7665493488311768,
      "step": 52067
    },
    {
      "epoch": 4.965496063232422e-05,
      "step": 52067,
      "training_step_time": 7.634164333343506
    },
    {
      "epoch": 4.965591430664062e-05,
      "model_forward_time": 1.7682600021362305,
      "step": 52068
    },
    {
      "epoch": 4.965591430664062e-05,
      "step": 52068,
      "training_step_time": 8.601985931396484
    },
    {
      "epoch": 4.9656867980957034e-05,
      "model_forward_time": 1.768878698348999,
      "step": 52069
    },
    {
      "epoch": 4.9656867980957034e-05,
      "step": 52069,
      "training_step_time": 7.551726579666138
    },
    {
      "epoch": 4.965782165527344e-05,
      "grad_norm": 0.12818355858325958,
      "learning_rate": 5.071262136321785e-06,
      "loss": 0.0623,
      "step": 52070
    },
    {
      "epoch": 4.965782165527344e-05,
      "model_forward_time": 1.7654426097869873,
      "step": 52070
    },
    {
      "dynamics_loss_avg": 0.0564732525497675,
      "epoch": 4.965782165527344e-05,
      "step": 52070
    },
    {
      "action_loss_avg": 0.0016859059629496186,
      "epoch": 4.965782165527344e-05,
      "step": 52070
    },
    {
      "epoch": 4.965782165527344e-05,
      "step": 52070,
      "training_step_time": 7.5126121044158936
    },
    {
      "epoch": 4.965877532958984e-05,
      "model_forward_time": 1.7662134170532227,
      "step": 52071
    },
    {
      "epoch": 4.965877532958984e-05,
      "step": 52071,
      "training_step_time": 7.281394004821777
    },
    {
      "epoch": 4.965972900390625e-05,
      "model_forward_time": 1.7663378715515137,
      "step": 52072
    },
    {
      "epoch": 4.965972900390625e-05,
      "step": 52072,
      "training_step_time": 7.71549916267395
    },
    {
      "epoch": 4.966068267822266e-05,
      "model_forward_time": 1.762739658355713,
      "step": 52073
    },
    {
      "epoch": 4.966068267822266e-05,
      "step": 52073,
      "training_step_time": 8.143851041793823
    },
    {
      "epoch": 4.966163635253906e-05,
      "model_forward_time": 2.9315593242645264,
      "step": 52074
    },
    {
      "epoch": 4.966163635253906e-05,
      "step": 52074,
      "training_step_time": 8.310926914215088
    },
    {
      "epoch": 4.966259002685547e-05,
      "model_forward_time": 1.8340234756469727,
      "step": 52075
    },
    {
      "epoch": 4.966259002685547e-05,
      "step": 52075,
      "training_step_time": 7.823267459869385
    },
    {
      "epoch": 4.9663543701171875e-05,
      "model_forward_time": 1.7674782276153564,
      "step": 52076
    },
    {
      "epoch": 4.9663543701171875e-05,
      "step": 52076,
      "training_step_time": 7.627699851989746
    },
    {
      "epoch": 4.966449737548828e-05,
      "model_forward_time": 1.732630968093872,
      "step": 52077
    },
    {
      "epoch": 4.966449737548828e-05,
      "step": 52077,
      "training_step_time": 9.162244319915771
    },
    {
      "epoch": 4.966545104980469e-05,
      "model_forward_time": 1.743405818939209,
      "step": 52078
    },
    {
      "epoch": 4.966545104980469e-05,
      "step": 52078,
      "training_step_time": 8.525020599365234
    },
    {
      "epoch": 4.9666404724121094e-05,
      "model_forward_time": 1.734553337097168,
      "step": 52079
    },
    {
      "epoch": 4.9666404724121094e-05,
      "step": 52079,
      "training_step_time": 7.489771842956543
    },
    {
      "epoch": 4.96673583984375e-05,
      "grad_norm": 0.15774524211883545,
      "learning_rate": 5.069608830582262e-06,
      "loss": 0.0611,
      "step": 52080
    },
    {
      "epoch": 4.96673583984375e-05,
      "model_forward_time": 1.7472913265228271,
      "step": 52080
    },
    {
      "dynamics_loss_avg": 0.06619147434830666,
      "epoch": 4.96673583984375e-05,
      "step": 52080
    },
    {
      "action_loss_avg": 0.0014862304436974228,
      "epoch": 4.96673583984375e-05,
      "step": 52080
    },
    {
      "epoch": 4.96673583984375e-05,
      "step": 52080,
      "training_step_time": 7.265333652496338
    },
    {
      "epoch": 4.966831207275391e-05,
      "model_forward_time": 1.7445228099822998,
      "step": 52081
    },
    {
      "epoch": 4.966831207275391e-05,
      "step": 52081,
      "training_step_time": 7.676197290420532
    },
    {
      "epoch": 4.966926574707031e-05,
      "model_forward_time": 1.7449333667755127,
      "step": 52082
    },
    {
      "epoch": 4.966926574707031e-05,
      "step": 52082,
      "training_step_time": 7.4352617263793945
    },
    {
      "epoch": 4.967021942138672e-05,
      "model_forward_time": 2.837780714035034,
      "step": 52083
    },
    {
      "epoch": 4.967021942138672e-05,
      "step": 52083,
      "training_step_time": 9.574913024902344
    },
    {
      "epoch": 4.967117309570313e-05,
      "model_forward_time": 1.8075203895568848,
      "step": 52084
    },
    {
      "epoch": 4.967117309570313e-05,
      "step": 52084,
      "training_step_time": 8.596572160720825
    },
    {
      "epoch": 4.967212677001953e-05,
      "model_forward_time": 1.7359435558319092,
      "step": 52085
    },
    {
      "epoch": 4.967212677001953e-05,
      "step": 52085,
      "training_step_time": 8.658054113388062
    },
    {
      "epoch": 4.9673080444335936e-05,
      "model_forward_time": 1.7384099960327148,
      "step": 52086
    },
    {
      "epoch": 4.9673080444335936e-05,
      "step": 52086,
      "training_step_time": 8.802022457122803
    },
    {
      "epoch": 4.967403411865235e-05,
      "model_forward_time": 1.7397739887237549,
      "step": 52087
    },
    {
      "epoch": 4.967403411865235e-05,
      "step": 52087,
      "training_step_time": 8.973913431167603
    },
    {
      "epoch": 4.967498779296875e-05,
      "model_forward_time": 1.789374589920044,
      "step": 52088
    },
    {
      "epoch": 4.967498779296875e-05,
      "step": 52088,
      "training_step_time": 8.463992357254028
    },
    {
      "epoch": 4.9675941467285155e-05,
      "model_forward_time": 1.7328290939331055,
      "step": 52089
    },
    {
      "epoch": 4.9675941467285155e-05,
      "step": 52089,
      "training_step_time": 7.317937850952148
    },
    {
      "epoch": 4.9676895141601566e-05,
      "grad_norm": 0.1499473750591278,
      "learning_rate": 5.0679555172304225e-06,
      "loss": 0.0575,
      "step": 52090
    },
    {
      "epoch": 4.9676895141601566e-05,
      "model_forward_time": 1.744478464126587,
      "step": 52090
    },
    {
      "dynamics_loss_avg": 0.06380466017872095,
      "epoch": 4.9676895141601566e-05,
      "step": 52090
    },
    {
      "action_loss_avg": 0.0015344392741099,
      "epoch": 4.9676895141601566e-05,
      "step": 52090
    },
    {
      "epoch": 4.9676895141601566e-05,
      "step": 52090,
      "training_step_time": 7.7136924266815186
    },
    {
      "epoch": 4.967784881591797e-05,
      "model_forward_time": 1.7390055656433105,
      "step": 52091
    },
    {
      "epoch": 4.967784881591797e-05,
      "step": 52091,
      "training_step_time": 7.550246953964233
    },
    {
      "epoch": 4.9678802490234374e-05,
      "model_forward_time": 2.8401687145233154,
      "step": 52092
    },
    {
      "epoch": 4.9678802490234374e-05,
      "step": 52092,
      "training_step_time": 7.710467100143433
    },
    {
      "epoch": 4.9679756164550785e-05,
      "model_forward_time": 1.8122236728668213,
      "step": 52093
    },
    {
      "epoch": 4.9679756164550785e-05,
      "step": 52093,
      "training_step_time": 9.522321939468384
    },
    {
      "epoch": 4.968070983886719e-05,
      "model_forward_time": 1.735755205154419,
      "step": 52094
    },
    {
      "epoch": 4.968070983886719e-05,
      "step": 52094,
      "training_step_time": 8.241764783859253
    },
    {
      "epoch": 4.968166351318359e-05,
      "model_forward_time": 1.732344627380371,
      "step": 52095
    },
    {
      "epoch": 4.968166351318359e-05,
      "step": 52095,
      "training_step_time": 9.764796257019043
    },
    {
      "epoch": 4.96826171875e-05,
      "model_forward_time": 1.7382187843322754,
      "step": 52096
    },
    {
      "epoch": 4.96826171875e-05,
      "step": 52096,
      "training_step_time": 7.687758207321167
    },
    {
      "epoch": 4.968357086181641e-05,
      "model_forward_time": 1.8017876148223877,
      "step": 52097
    },
    {
      "epoch": 4.968357086181641e-05,
      "step": 52097,
      "training_step_time": 7.618407964706421
    },
    {
      "epoch": 4.968452453613281e-05,
      "model_forward_time": 1.7423810958862305,
      "step": 52098
    },
    {
      "epoch": 4.968452453613281e-05,
      "step": 52098,
      "training_step_time": 7.342143774032593
    },
    {
      "epoch": 4.9685478210449216e-05,
      "model_forward_time": 1.738732099533081,
      "step": 52099
    },
    {
      "epoch": 4.9685478210449216e-05,
      "step": 52099,
      "training_step_time": 7.673588037490845
    },
    {
      "epoch": 4.9686431884765626e-05,
      "grad_norm": 0.13431303203105927,
      "learning_rate": 5.066302196447071e-06,
      "loss": 0.0595,
      "step": 52100
    },
    {
      "epoch": 4.9686431884765626e-05,
      "model_forward_time": 1.7395191192626953,
      "step": 52100
    },
    {
      "dynamics_loss_avg": 0.06021004542708397,
      "epoch": 4.9686431884765626e-05,
      "step": 52100
    },
    {
      "action_loss_avg": 0.0022209267248399557,
      "epoch": 4.9686431884765626e-05,
      "step": 52100
    },
    {
      "epoch": 4.9686431884765626e-05,
      "step": 52100,
      "training_step_time": 7.554425001144409
    },
    {
      "epoch": 4.968738555908203e-05,
      "model_forward_time": 1.7411994934082031,
      "step": 52101
    },
    {
      "epoch": 4.968738555908203e-05,
      "step": 52101,
      "training_step_time": 8.458123683929443
    },
    {
      "epoch": 4.9688339233398434e-05,
      "model_forward_time": 1.7226495742797852,
      "step": 52102
    },
    {
      "epoch": 4.9688339233398434e-05,
      "step": 52102,
      "training_step_time": 8.173414468765259
    },
    {
      "epoch": 4.9689292907714845e-05,
      "model_forward_time": 1.740105390548706,
      "step": 52103
    },
    {
      "epoch": 4.9689292907714845e-05,
      "step": 52103,
      "training_step_time": 8.674963235855103
    },
    {
      "epoch": 4.969024658203125e-05,
      "model_forward_time": 1.7390756607055664,
      "step": 52104
    },
    {
      "epoch": 4.969024658203125e-05,
      "step": 52104,
      "training_step_time": 7.635375738143921
    },
    {
      "epoch": 4.969120025634765e-05,
      "model_forward_time": 1.7489242553710938,
      "step": 52105
    },
    {
      "epoch": 4.969120025634765e-05,
      "step": 52105,
      "training_step_time": 7.656376361846924
    },
    {
      "epoch": 4.9692153930664064e-05,
      "model_forward_time": 1.7446868419647217,
      "step": 52106
    },
    {
      "epoch": 4.9692153930664064e-05,
      "step": 52106,
      "training_step_time": 8.668938398361206
    },
    {
      "epoch": 4.969310760498047e-05,
      "model_forward_time": 1.7425189018249512,
      "step": 52107
    },
    {
      "epoch": 4.969310760498047e-05,
      "step": 52107,
      "training_step_time": 7.864529371261597
    },
    {
      "epoch": 4.969406127929687e-05,
      "model_forward_time": 1.7298438549041748,
      "step": 52108
    },
    {
      "epoch": 4.969406127929687e-05,
      "step": 52108,
      "training_step_time": 7.604813098907471
    },
    {
      "epoch": 4.969501495361328e-05,
      "model_forward_time": 1.733093023300171,
      "step": 52109
    },
    {
      "epoch": 4.969501495361328e-05,
      "step": 52109,
      "training_step_time": 7.488054990768433
    },
    {
      "epoch": 4.969596862792969e-05,
      "grad_norm": 0.10813842713832855,
      "learning_rate": 5.064648868413008e-06,
      "loss": 0.0586,
      "step": 52110
    },
    {
      "epoch": 4.969596862792969e-05,
      "model_forward_time": 1.7337143421173096,
      "step": 52110
    },
    {
      "dynamics_loss_avg": 0.0570804612711072,
      "epoch": 4.969596862792969e-05,
      "step": 52110
    },
    {
      "action_loss_avg": 0.0019693479669513183,
      "epoch": 4.969596862792969e-05,
      "step": 52110
    },
    {
      "epoch": 4.969596862792969e-05,
      "step": 52110,
      "training_step_time": 7.396138429641724
    },
    {
      "epoch": 4.969692230224609e-05,
      "model_forward_time": 2.8714025020599365,
      "step": 52111
    },
    {
      "epoch": 4.969692230224609e-05,
      "step": 52111,
      "training_step_time": 8.151318788528442
    },
    {
      "epoch": 4.96978759765625e-05,
      "model_forward_time": 1.7916524410247803,
      "step": 52112
    },
    {
      "epoch": 4.96978759765625e-05,
      "step": 52112,
      "training_step_time": 7.6752378940582275
    },
    {
      "epoch": 4.9698829650878906e-05,
      "model_forward_time": 1.7556369304656982,
      "step": 52113
    },
    {
      "epoch": 4.9698829650878906e-05,
      "step": 52113,
      "training_step_time": 8.382648944854736
    },
    {
      "epoch": 4.969978332519531e-05,
      "model_forward_time": 1.7440423965454102,
      "step": 52114
    },
    {
      "epoch": 4.969978332519531e-05,
      "step": 52114,
      "training_step_time": 7.63155722618103
    },
    {
      "epoch": 4.970073699951172e-05,
      "model_forward_time": 1.7429749965667725,
      "step": 52115
    },
    {
      "epoch": 4.970073699951172e-05,
      "step": 52115,
      "training_step_time": 8.63846230506897
    },
    {
      "epoch": 4.9701690673828125e-05,
      "model_forward_time": 1.7405645847320557,
      "step": 52116
    },
    {
      "epoch": 4.9701690673828125e-05,
      "step": 52116,
      "training_step_time": 8.133392572402954
    },
    {
      "epoch": 4.970264434814453e-05,
      "model_forward_time": 1.7385857105255127,
      "step": 52117
    },
    {
      "epoch": 4.970264434814453e-05,
      "step": 52117,
      "training_step_time": 7.5937159061431885
    },
    {
      "epoch": 4.970359802246094e-05,
      "model_forward_time": 1.7501990795135498,
      "step": 52118
    },
    {
      "epoch": 4.970359802246094e-05,
      "step": 52118,
      "training_step_time": 8.434314250946045
    },
    {
      "epoch": 4.9704551696777344e-05,
      "model_forward_time": 1.74709153175354,
      "step": 52119
    },
    {
      "epoch": 4.9704551696777344e-05,
      "step": 52119,
      "training_step_time": 7.365971326828003
    },
    {
      "epoch": 4.970550537109375e-05,
      "grad_norm": 0.10364845395088196,
      "learning_rate": 5.0629955333090445e-06,
      "loss": 0.0612,
      "step": 52120
    },
    {
      "epoch": 4.970550537109375e-05,
      "model_forward_time": 2.9254229068756104,
      "step": 52120
    },
    {
      "dynamics_loss_avg": 0.06279468294233084,
      "epoch": 4.970550537109375e-05,
      "step": 52120
    },
    {
      "action_loss_avg": 0.002006175077985972,
      "epoch": 4.970550537109375e-05,
      "step": 52120
    },
    {
      "epoch": 4.970550537109375e-05,
      "step": 52120,
      "training_step_time": 7.679809093475342
    },
    {
      "epoch": 4.970645904541016e-05,
      "model_forward_time": 1.80507493019104,
      "step": 52121
    },
    {
      "epoch": 4.970645904541016e-05,
      "step": 52121,
      "training_step_time": 8.499213695526123
    },
    {
      "epoch": 4.970741271972656e-05,
      "model_forward_time": 1.7473392486572266,
      "step": 52122
    },
    {
      "epoch": 4.970741271972656e-05,
      "step": 52122,
      "training_step_time": 8.462376117706299
    },
    {
      "epoch": 4.9708366394042966e-05,
      "model_forward_time": 1.7177364826202393,
      "step": 52123
    },
    {
      "epoch": 4.9708366394042966e-05,
      "step": 52123,
      "training_step_time": 8.530848026275635
    },
    {
      "epoch": 4.970932006835938e-05,
      "model_forward_time": 1.746161699295044,
      "step": 52124
    },
    {
      "epoch": 4.970932006835938e-05,
      "step": 52124,
      "training_step_time": 8.209875106811523
    },
    {
      "epoch": 4.971027374267578e-05,
      "model_forward_time": 1.7386863231658936,
      "step": 52125
    },
    {
      "epoch": 4.971027374267578e-05,
      "step": 52125,
      "training_step_time": 7.495344400405884
    },
    {
      "epoch": 4.9711227416992185e-05,
      "model_forward_time": 1.730790376663208,
      "step": 52126
    },
    {
      "epoch": 4.9711227416992185e-05,
      "step": 52126,
      "training_step_time": 7.559379577636719
    },
    {
      "epoch": 4.9712181091308596e-05,
      "model_forward_time": 1.7392470836639404,
      "step": 52127
    },
    {
      "epoch": 4.9712181091308596e-05,
      "step": 52127,
      "training_step_time": 7.6246421337127686
    },
    {
      "epoch": 4.9713134765625e-05,
      "model_forward_time": 1.7334723472595215,
      "step": 52128
    },
    {
      "epoch": 4.9713134765625e-05,
      "step": 52128,
      "training_step_time": 7.421358346939087
    },
    {
      "epoch": 4.9714088439941404e-05,
      "model_forward_time": 2.8983240127563477,
      "step": 52129
    },
    {
      "epoch": 4.9714088439941404e-05,
      "step": 52129,
      "training_step_time": 8.9282546043396
    },
    {
      "epoch": 4.9715042114257815e-05,
      "grad_norm": 0.10274892300367355,
      "learning_rate": 5.0613421913159835e-06,
      "loss": 0.0579,
      "step": 52130
    },
    {
      "epoch": 4.9715042114257815e-05,
      "model_forward_time": 1.804478406906128,
      "step": 52130
    },
    {
      "dynamics_loss_avg": 0.06648732237517833,
      "epoch": 4.9715042114257815e-05,
      "step": 52130
    },
    {
      "action_loss_avg": 0.0018672627164050936,
      "epoch": 4.9715042114257815e-05,
      "step": 52130
    },
    {
      "epoch": 4.9715042114257815e-05,
      "step": 52130,
      "training_step_time": 7.677798748016357
    },
    {
      "epoch": 4.971599578857422e-05,
      "model_forward_time": 1.7296559810638428,
      "step": 52131
    },
    {
      "epoch": 4.971599578857422e-05,
      "step": 52131,
      "training_step_time": 10.095539808273315
    },
    {
      "epoch": 4.971694946289062e-05,
      "model_forward_time": 1.734764575958252,
      "step": 52132
    },
    {
      "epoch": 4.971694946289062e-05,
      "step": 52132,
      "training_step_time": 7.7536022663116455
    },
    {
      "epoch": 4.9717903137207034e-05,
      "model_forward_time": 1.7420780658721924,
      "step": 52133
    },
    {
      "epoch": 4.9717903137207034e-05,
      "step": 52133,
      "training_step_time": 8.659992456436157
    },
    {
      "epoch": 4.971885681152344e-05,
      "model_forward_time": 1.7419769763946533,
      "step": 52134
    },
    {
      "epoch": 4.971885681152344e-05,
      "step": 52134,
      "training_step_time": 7.737095832824707
    },
    {
      "epoch": 4.971981048583984e-05,
      "model_forward_time": 1.7344934940338135,
      "step": 52135
    },
    {
      "epoch": 4.971981048583984e-05,
      "step": 52135,
      "training_step_time": 7.662290573120117
    },
    {
      "epoch": 4.972076416015625e-05,
      "model_forward_time": 1.7359733581542969,
      "step": 52136
    },
    {
      "epoch": 4.972076416015625e-05,
      "step": 52136,
      "training_step_time": 8.132116794586182
    },
    {
      "epoch": 4.972171783447266e-05,
      "model_forward_time": 1.740168809890747,
      "step": 52137
    },
    {
      "epoch": 4.972171783447266e-05,
      "step": 52137,
      "training_step_time": 8.178937196731567
    },
    {
      "epoch": 4.972267150878906e-05,
      "model_forward_time": 2.8541126251220703,
      "step": 52138
    },
    {
      "epoch": 4.972267150878906e-05,
      "step": 52138,
      "training_step_time": 8.446568965911865
    },
    {
      "epoch": 4.972362518310547e-05,
      "model_forward_time": 1.7622389793395996,
      "step": 52139
    },
    {
      "epoch": 4.972362518310547e-05,
      "step": 52139,
      "training_step_time": 7.8050384521484375
    },
    {
      "epoch": 4.9724578857421876e-05,
      "grad_norm": 0.10134166479110718,
      "learning_rate": 5.0596888426146325e-06,
      "loss": 0.0591,
      "step": 52140
    },
    {
      "epoch": 4.9724578857421876e-05,
      "model_forward_time": 1.7391376495361328,
      "step": 52140
    },
    {
      "dynamics_loss_avg": 0.04525437708944082,
      "epoch": 4.9724578857421876e-05,
      "step": 52140
    },
    {
      "action_loss_avg": 0.0009238088183337823,
      "epoch": 4.9724578857421876e-05,
      "step": 52140
    },
    {
      "epoch": 4.9724578857421876e-05,
      "step": 52140,
      "training_step_time": 8.337045431137085
    },
    {
      "epoch": 4.972553253173828e-05,
      "model_forward_time": 1.7397994995117188,
      "step": 52141
    },
    {
      "epoch": 4.972553253173828e-05,
      "step": 52141,
      "training_step_time": 8.458876609802246
    },
    {
      "epoch": 4.972648620605469e-05,
      "model_forward_time": 1.7365672588348389,
      "step": 52142
    },
    {
      "epoch": 4.972648620605469e-05,
      "step": 52142,
      "training_step_time": 9.038862705230713
    },
    {
      "epoch": 4.9727439880371094e-05,
      "model_forward_time": 1.7406768798828125,
      "step": 52143
    },
    {
      "epoch": 4.9727439880371094e-05,
      "step": 52143,
      "training_step_time": 8.376540422439575
    },
    {
      "epoch": 4.97283935546875e-05,
      "model_forward_time": 1.729525089263916,
      "step": 52144
    },
    {
      "epoch": 4.97283935546875e-05,
      "step": 52144,
      "training_step_time": 7.575041055679321
    },
    {
      "epoch": 4.972934722900391e-05,
      "model_forward_time": 1.7413840293884277,
      "step": 52145
    },
    {
      "epoch": 4.972934722900391e-05,
      "step": 52145,
      "training_step_time": 7.58970832824707
    },
    {
      "epoch": 4.973030090332031e-05,
      "model_forward_time": 1.7450189590454102,
      "step": 52146
    },
    {
      "epoch": 4.973030090332031e-05,
      "step": 52146,
      "training_step_time": 7.560760021209717
    },
    {
      "epoch": 4.973125457763672e-05,
      "model_forward_time": 2.831925868988037,
      "step": 52147
    },
    {
      "epoch": 4.973125457763672e-05,
      "step": 52147,
      "training_step_time": 8.456110715866089
    },
    {
      "epoch": 4.973220825195313e-05,
      "model_forward_time": 1.7453374862670898,
      "step": 52148
    },
    {
      "epoch": 4.973220825195313e-05,
      "step": 52148,
      "training_step_time": 8.463815450668335
    },
    {
      "epoch": 4.973316192626953e-05,
      "model_forward_time": 1.7406914234161377,
      "step": 52149
    },
    {
      "epoch": 4.973316192626953e-05,
      "step": 52149,
      "training_step_time": 7.7290003299713135
    },
    {
      "epoch": 4.9734115600585936e-05,
      "grad_norm": 0.12535932660102844,
      "learning_rate": 5.058035487385798e-06,
      "loss": 0.0558,
      "step": 52150
    },
    {
      "epoch": 4.9734115600585936e-05,
      "model_forward_time": 1.743753433227539,
      "step": 52150
    },
    {
      "dynamics_loss_avg": 0.052648926712572576,
      "epoch": 4.9734115600585936e-05,
      "step": 52150
    },
    {
      "action_loss_avg": 0.0011574783333344386,
      "epoch": 4.9734115600585936e-05,
      "step": 52150
    },
    {
      "epoch": 4.9734115600585936e-05,
      "step": 52150,
      "training_step_time": 7.702053546905518
    },
    {
      "epoch": 4.973506927490235e-05,
      "model_forward_time": 1.7516505718231201,
      "step": 52151
    },
    {
      "epoch": 4.973506927490235e-05,
      "step": 52151,
      "training_step_time": 8.81477427482605
    },
    {
      "epoch": 4.973602294921875e-05,
      "model_forward_time": 1.7345843315124512,
      "step": 52152
    },
    {
      "epoch": 4.973602294921875e-05,
      "step": 52152,
      "training_step_time": 8.136993885040283
    },
    {
      "epoch": 4.9736976623535155e-05,
      "model_forward_time": 1.7451305389404297,
      "step": 52153
    },
    {
      "epoch": 4.9736976623535155e-05,
      "step": 52153,
      "training_step_time": 7.714006662368774
    },
    {
      "epoch": 4.9737930297851566e-05,
      "model_forward_time": 1.371016502380371,
      "step": 52154
    },
    {
      "epoch": 4.9737930297851566e-05,
      "step": 52154,
      "training_step_time": 7.775836229324341
    },
    {
      "epoch": 4.973888397216797e-05,
      "model_forward_time": 1.741206169128418,
      "step": 52155
    },
    {
      "epoch": 4.973888397216797e-05,
      "step": 52155,
      "training_step_time": 7.736151218414307
    },
    {
      "epoch": 4.9739837646484374e-05,
      "model_forward_time": 1.7430768013000488,
      "step": 52156
    },
    {
      "epoch": 4.9739837646484374e-05,
      "step": 52156,
      "training_step_time": 7.521525859832764
    },
    {
      "epoch": 4.974079132080078e-05,
      "model_forward_time": 2.8820478916168213,
      "step": 52157
    },
    {
      "epoch": 4.974079132080078e-05,
      "step": 52157,
      "training_step_time": 7.741668939590454
    },
    {
      "epoch": 4.974174499511719e-05,
      "model_forward_time": 1.7914128303527832,
      "step": 52158
    },
    {
      "epoch": 4.974174499511719e-05,
      "step": 52158,
      "training_step_time": 8.687083005905151
    },
    {
      "epoch": 4.974269866943359e-05,
      "model_forward_time": 1.7430739402770996,
      "step": 52159
    },
    {
      "epoch": 4.974269866943359e-05,
      "step": 52159,
      "training_step_time": 7.73743462562561
    },
    {
      "epoch": 4.974365234375e-05,
      "grad_norm": 0.10710681974887848,
      "learning_rate": 5.056382125810292e-06,
      "loss": 0.0555,
      "step": 52160
    },
    {
      "epoch": 4.974365234375e-05,
      "model_forward_time": 1.7670745849609375,
      "step": 52160
    },
    {
      "dynamics_loss_avg": 0.061208198219537734,
      "epoch": 4.974365234375e-05,
      "step": 52160
    },
    {
      "action_loss_avg": 0.001362116987002082,
      "epoch": 4.974365234375e-05,
      "step": 52160
    },
    {
      "epoch": 4.974365234375e-05,
      "step": 52160,
      "training_step_time": 7.770709276199341
    },
    {
      "epoch": 4.974460601806641e-05,
      "model_forward_time": 1.7452142238616943,
      "step": 52161
    },
    {
      "epoch": 4.974460601806641e-05,
      "step": 52161,
      "training_step_time": 7.733098745346069
    },
    {
      "epoch": 4.974555969238281e-05,
      "model_forward_time": 1.7361228466033936,
      "step": 52162
    },
    {
      "epoch": 4.974555969238281e-05,
      "step": 52162,
      "training_step_time": 7.688518762588501
    },
    {
      "epoch": 4.9746513366699216e-05,
      "model_forward_time": 1.7390284538269043,
      "step": 52163
    },
    {
      "epoch": 4.9746513366699216e-05,
      "step": 52163,
      "training_step_time": 8.52020788192749
    },
    {
      "epoch": 4.9747467041015626e-05,
      "model_forward_time": 1.7389605045318604,
      "step": 52164
    },
    {
      "epoch": 4.9747467041015626e-05,
      "step": 52164,
      "training_step_time": 7.759145021438599
    },
    {
      "epoch": 4.974842071533203e-05,
      "model_forward_time": 1.742114543914795,
      "step": 52165
    },
    {
      "epoch": 4.974842071533203e-05,
      "step": 52165,
      "training_step_time": 7.547587871551514
    },
    {
      "epoch": 4.9749374389648434e-05,
      "model_forward_time": 2.939051866531372,
      "step": 52166
    },
    {
      "epoch": 4.9749374389648434e-05,
      "step": 52166,
      "training_step_time": 7.798221111297607
    },
    {
      "epoch": 4.9750328063964845e-05,
      "model_forward_time": 1.815540075302124,
      "step": 52167
    },
    {
      "epoch": 4.9750328063964845e-05,
      "step": 52167,
      "training_step_time": 7.784725189208984
    },
    {
      "epoch": 4.975128173828125e-05,
      "model_forward_time": 1.734553337097168,
      "step": 52168
    },
    {
      "epoch": 4.975128173828125e-05,
      "step": 52168,
      "training_step_time": 8.586026430130005
    },
    {
      "epoch": 4.975223541259765e-05,
      "model_forward_time": 1.7383227348327637,
      "step": 52169
    },
    {
      "epoch": 4.975223541259765e-05,
      "step": 52169,
      "training_step_time": 9.225002765655518
    },
    {
      "epoch": 4.9753189086914064e-05,
      "grad_norm": 0.119458869099617,
      "learning_rate": 5.05472875806892e-06,
      "loss": 0.0563,
      "step": 52170
    },
    {
      "epoch": 4.9753189086914064e-05,
      "model_forward_time": 1.7487316131591797,
      "step": 52170
    },
    {
      "dynamics_loss_avg": 0.05159520525485277,
      "epoch": 4.9753189086914064e-05,
      "step": 52170
    },
    {
      "action_loss_avg": 0.0010230335930828006,
      "epoch": 4.9753189086914064e-05,
      "step": 52170
    },
    {
      "epoch": 4.9753189086914064e-05,
      "step": 52170,
      "training_step_time": 7.579001426696777
    },
    {
      "epoch": 4.975414276123047e-05,
      "model_forward_time": 1.7419838905334473,
      "step": 52171
    },
    {
      "epoch": 4.975414276123047e-05,
      "step": 52171,
      "training_step_time": 8.230514526367188
    },
    {
      "epoch": 4.975509643554687e-05,
      "model_forward_time": 1.737743616104126,
      "step": 52172
    },
    {
      "epoch": 4.975509643554687e-05,
      "step": 52172,
      "training_step_time": 7.757366180419922
    },
    {
      "epoch": 4.975605010986328e-05,
      "model_forward_time": 1.7427141666412354,
      "step": 52173
    },
    {
      "epoch": 4.975605010986328e-05,
      "step": 52173,
      "training_step_time": 7.719019174575806
    },
    {
      "epoch": 4.975700378417969e-05,
      "model_forward_time": 1.738842248916626,
      "step": 52174
    },
    {
      "epoch": 4.975700378417969e-05,
      "step": 52174,
      "training_step_time": 7.527411222457886
    },
    {
      "epoch": 4.975795745849609e-05,
      "model_forward_time": 2.8724968433380127,
      "step": 52175
    },
    {
      "epoch": 4.975795745849609e-05,
      "step": 52175,
      "training_step_time": 8.661174774169922
    },
    {
      "epoch": 4.97589111328125e-05,
      "model_forward_time": 1.7993848323822021,
      "step": 52176
    },
    {
      "epoch": 4.97589111328125e-05,
      "step": 52176,
      "training_step_time": 7.6341331005096436
    },
    {
      "epoch": 4.9759864807128906e-05,
      "model_forward_time": 1.7458202838897705,
      "step": 52177
    },
    {
      "epoch": 4.9759864807128906e-05,
      "step": 52177,
      "training_step_time": 7.7076263427734375
    },
    {
      "epoch": 4.976081848144531e-05,
      "model_forward_time": 1.7383651733398438,
      "step": 52178
    },
    {
      "epoch": 4.976081848144531e-05,
      "step": 52178,
      "training_step_time": 8.47933053970337
    },
    {
      "epoch": 4.976177215576172e-05,
      "model_forward_time": 1.74141526222229,
      "step": 52179
    },
    {
      "epoch": 4.976177215576172e-05,
      "step": 52179,
      "training_step_time": 9.286936044692993
    },
    {
      "epoch": 4.9762725830078125e-05,
      "grad_norm": 0.28932029008865356,
      "learning_rate": 5.053075384342494e-06,
      "loss": 0.0593,
      "step": 52180
    },
    {
      "epoch": 4.9762725830078125e-05,
      "model_forward_time": 1.7475249767303467,
      "step": 52180
    },
    {
      "dynamics_loss_avg": 0.07694372590631246,
      "epoch": 4.9762725830078125e-05,
      "step": 52180
    },
    {
      "action_loss_avg": 0.002345411205897108,
      "epoch": 4.9762725830078125e-05,
      "step": 52180
    },
    {
      "epoch": 4.9762725830078125e-05,
      "step": 52180,
      "training_step_time": 8.649774551391602
    },
    {
      "epoch": 4.976367950439453e-05,
      "model_forward_time": 1.7422661781311035,
      "step": 52181
    },
    {
      "epoch": 4.976367950439453e-05,
      "step": 52181,
      "training_step_time": 7.573724746704102
    },
    {
      "epoch": 4.976463317871094e-05,
      "model_forward_time": 1.7405667304992676,
      "step": 52182
    },
    {
      "epoch": 4.976463317871094e-05,
      "step": 52182,
      "training_step_time": 7.565784454345703
    },
    {
      "epoch": 4.9765586853027344e-05,
      "model_forward_time": 1.7432045936584473,
      "step": 52183
    },
    {
      "epoch": 4.9765586853027344e-05,
      "step": 52183,
      "training_step_time": 7.412584066390991
    },
    {
      "epoch": 4.976654052734375e-05,
      "model_forward_time": 2.8321456909179688,
      "step": 52184
    },
    {
      "epoch": 4.976654052734375e-05,
      "step": 52184,
      "training_step_time": 7.55597710609436
    },
    {
      "epoch": 4.976749420166016e-05,
      "model_forward_time": 1.7656536102294922,
      "step": 52185
    },
    {
      "epoch": 4.976749420166016e-05,
      "step": 52185,
      "training_step_time": 8.47823715209961
    },
    {
      "epoch": 4.976844787597656e-05,
      "model_forward_time": 1.7422873973846436,
      "step": 52186
    },
    {
      "epoch": 4.976844787597656e-05,
      "step": 52186,
      "training_step_time": 9.168158054351807
    },
    {
      "epoch": 4.9769401550292966e-05,
      "model_forward_time": 1.7456622123718262,
      "step": 52187
    },
    {
      "epoch": 4.9769401550292966e-05,
      "step": 52187,
      "training_step_time": 7.634705066680908
    },
    {
      "epoch": 4.977035522460938e-05,
      "model_forward_time": 1.7389945983886719,
      "step": 52188
    },
    {
      "epoch": 4.977035522460938e-05,
      "step": 52188,
      "training_step_time": 7.773231029510498
    },
    {
      "epoch": 4.977130889892578e-05,
      "model_forward_time": 1.7316536903381348,
      "step": 52189
    },
    {
      "epoch": 4.977130889892578e-05,
      "step": 52189,
      "training_step_time": 8.223000764846802
    },
    {
      "epoch": 4.9772262573242185e-05,
      "grad_norm": 0.13020315766334534,
      "learning_rate": 5.051422004811824e-06,
      "loss": 0.0541,
      "step": 52190
    },
    {
      "epoch": 4.9772262573242185e-05,
      "model_forward_time": 1.730950117111206,
      "step": 52190
    },
    {
      "dynamics_loss_avg": 0.044212513230741025,
      "epoch": 4.9772262573242185e-05,
      "step": 52190
    },
    {
      "action_loss_avg": 0.0014660745742730795,
      "epoch": 4.9772262573242185e-05,
      "step": 52190
    },
    {
      "epoch": 4.9772262573242185e-05,
      "step": 52190,
      "training_step_time": 7.613648176193237
    },
    {
      "epoch": 4.9773216247558596e-05,
      "model_forward_time": 1.7368848323822021,
      "step": 52191
    },
    {
      "epoch": 4.9773216247558596e-05,
      "step": 52191,
      "training_step_time": 7.56801700592041
    },
    {
      "epoch": 4.9774169921875e-05,
      "model_forward_time": 1.768373727798462,
      "step": 52192
    },
    {
      "epoch": 4.9774169921875e-05,
      "step": 52192,
      "training_step_time": 7.4573140144348145
    },
    {
      "epoch": 4.9775123596191404e-05,
      "model_forward_time": 2.8371808528900146,
      "step": 52193
    },
    {
      "epoch": 4.9775123596191404e-05,
      "step": 52193,
      "training_step_time": 7.6043381690979
    },
    {
      "epoch": 4.9776077270507815e-05,
      "model_forward_time": 1.7393474578857422,
      "step": 52194
    },
    {
      "epoch": 4.9776077270507815e-05,
      "step": 52194,
      "training_step_time": 8.49464464187622
    },
    {
      "epoch": 4.977703094482422e-05,
      "model_forward_time": 1.7882566452026367,
      "step": 52195
    },
    {
      "epoch": 4.977703094482422e-05,
      "step": 52195,
      "training_step_time": 10.259271383285522
    },
    {
      "epoch": 4.977798461914062e-05,
      "model_forward_time": 1.7427546977996826,
      "step": 52196
    },
    {
      "epoch": 4.977798461914062e-05,
      "step": 52196,
      "training_step_time": 7.642950773239136
    },
    {
      "epoch": 4.9778938293457034e-05,
      "model_forward_time": 1.7649362087249756,
      "step": 52197
    },
    {
      "epoch": 4.9778938293457034e-05,
      "step": 52197,
      "training_step_time": 7.855950593948364
    },
    {
      "epoch": 4.977989196777344e-05,
      "model_forward_time": 1.7407536506652832,
      "step": 52198
    },
    {
      "epoch": 4.977989196777344e-05,
      "step": 52198,
      "training_step_time": 7.778698921203613
    },
    {
      "epoch": 4.978084564208984e-05,
      "model_forward_time": 1.7335636615753174,
      "step": 52199
    },
    {
      "epoch": 4.978084564208984e-05,
      "step": 52199,
      "training_step_time": 7.7197723388671875
    },
    {
      "epoch": 4.978179931640625e-05,
      "grad_norm": 0.0834202691912651,
      "learning_rate": 5.049768619657721e-06,
      "loss": 0.0544,
      "step": 52200
    },
    {
      "epoch": 4.978179931640625e-05,
      "model_forward_time": 1.7396314144134521,
      "step": 52200
    },
    {
      "dynamics_loss_avg": 0.049872261565178634,
      "epoch": 4.978179931640625e-05,
      "step": 52200
    },
    {
      "action_loss_avg": 0.001508622661640402,
      "epoch": 4.978179931640625e-05,
      "step": 52200
    },
    {
      "epoch": 4.978179931640625e-05,
      "step": 52200,
      "training_step_time": 7.793236494064331
    },
    {
      "epoch": 4.978275299072266e-05,
      "model_forward_time": 1.743783712387085,
      "step": 52201
    },
    {
      "epoch": 4.978275299072266e-05,
      "step": 52201,
      "training_step_time": 7.649133682250977
    },
    {
      "epoch": 4.978370666503906e-05,
      "model_forward_time": 1.7411072254180908,
      "step": 52202
    },
    {
      "epoch": 4.978370666503906e-05,
      "step": 52202,
      "training_step_time": 9.154505014419556
    },
    {
      "epoch": 4.978466033935547e-05,
      "model_forward_time": 1.739833116531372,
      "step": 52203
    },
    {
      "epoch": 4.978466033935547e-05,
      "step": 52203,
      "training_step_time": 7.477752685546875
    },
    {
      "epoch": 4.9785614013671876e-05,
      "model_forward_time": 1.8084807395935059,
      "step": 52204
    },
    {
      "epoch": 4.9785614013671876e-05,
      "step": 52204,
      "training_step_time": 7.685389518737793
    },
    {
      "epoch": 4.978656768798828e-05,
      "model_forward_time": 1.7446532249450684,
      "step": 52205
    },
    {
      "epoch": 4.978656768798828e-05,
      "step": 52205,
      "training_step_time": 8.40867567062378
    },
    {
      "epoch": 4.978752136230469e-05,
      "model_forward_time": 1.7363777160644531,
      "step": 52206
    },
    {
      "epoch": 4.978752136230469e-05,
      "step": 52206,
      "training_step_time": 8.948680877685547
    },
    {
      "epoch": 4.9788475036621094e-05,
      "model_forward_time": 1.7391166687011719,
      "step": 52207
    },
    {
      "epoch": 4.9788475036621094e-05,
      "step": 52207,
      "training_step_time": 7.769701242446899
    },
    {
      "epoch": 4.97894287109375e-05,
      "model_forward_time": 1.747086763381958,
      "step": 52208
    },
    {
      "epoch": 4.97894287109375e-05,
      "step": 52208,
      "training_step_time": 7.584590673446655
    },
    {
      "epoch": 4.979038238525391e-05,
      "model_forward_time": 1.732508897781372,
      "step": 52209
    },
    {
      "epoch": 4.979038238525391e-05,
      "step": 52209,
      "training_step_time": 7.799138307571411
    },
    {
      "epoch": 4.979133605957031e-05,
      "grad_norm": 0.08302980661392212,
      "learning_rate": 5.048115229060996e-06,
      "loss": 0.0568,
      "step": 52210
    },
    {
      "epoch": 4.979133605957031e-05,
      "model_forward_time": 1.7443058490753174,
      "step": 52210
    },
    {
      "dynamics_loss_avg": 0.05193764697760343,
      "epoch": 4.979133605957031e-05,
      "step": 52210
    },
    {
      "action_loss_avg": 0.0012436732882633804,
      "epoch": 4.979133605957031e-05,
      "step": 52210
    },
    {
      "epoch": 4.979133605957031e-05,
      "step": 52210,
      "training_step_time": 7.575285911560059
    },
    {
      "epoch": 4.979228973388672e-05,
      "model_forward_time": 1.7360596656799316,
      "step": 52211
    },
    {
      "epoch": 4.979228973388672e-05,
      "step": 52211,
      "training_step_time": 7.519807577133179
    },
    {
      "epoch": 4.979324340820313e-05,
      "model_forward_time": 2.858907699584961,
      "step": 52212
    },
    {
      "epoch": 4.979324340820313e-05,
      "step": 52212,
      "training_step_time": 8.72597336769104
    },
    {
      "epoch": 4.979419708251953e-05,
      "model_forward_time": 1.7984511852264404,
      "step": 52213
    },
    {
      "epoch": 4.979419708251953e-05,
      "step": 52213,
      "training_step_time": 8.7619948387146
    },
    {
      "epoch": 4.9795150756835936e-05,
      "model_forward_time": 1.7393343448638916,
      "step": 52214
    },
    {
      "epoch": 4.9795150756835936e-05,
      "step": 52214,
      "training_step_time": 8.725616931915283
    },
    {
      "epoch": 4.979610443115235e-05,
      "model_forward_time": 1.7399132251739502,
      "step": 52215
    },
    {
      "epoch": 4.979610443115235e-05,
      "step": 52215,
      "training_step_time": 7.821879863739014
    },
    {
      "epoch": 4.979705810546875e-05,
      "model_forward_time": 1.7330131530761719,
      "step": 52216
    },
    {
      "epoch": 4.979705810546875e-05,
      "step": 52216,
      "training_step_time": 7.758322477340698
    },
    {
      "epoch": 4.9798011779785155e-05,
      "model_forward_time": 1.7302911281585693,
      "step": 52217
    },
    {
      "epoch": 4.9798011779785155e-05,
      "step": 52217,
      "training_step_time": 7.635523796081543
    },
    {
      "epoch": 4.9798965454101566e-05,
      "model_forward_time": 1.728844165802002,
      "step": 52218
    },
    {
      "epoch": 4.9798965454101566e-05,
      "step": 52218,
      "training_step_time": 8.53870439529419
    },
    {
      "epoch": 4.979991912841797e-05,
      "model_forward_time": 1.7644133567810059,
      "step": 52219
    },
    {
      "epoch": 4.979991912841797e-05,
      "step": 52219,
      "training_step_time": 7.568861722946167
    },
    {
      "epoch": 4.9800872802734374e-05,
      "grad_norm": 0.1047336682677269,
      "learning_rate": 5.046461833202462e-06,
      "loss": 0.0567,
      "step": 52220
    },
    {
      "epoch": 4.9800872802734374e-05,
      "model_forward_time": 1.738605260848999,
      "step": 52220
    },
    {
      "dynamics_loss_avg": 0.05995711833238602,
      "epoch": 4.9800872802734374e-05,
      "step": 52220
    },
    {
      "action_loss_avg": 0.0016504644299857318,
      "epoch": 4.9800872802734374e-05,
      "step": 52220
    },
    {
      "epoch": 4.9800872802734374e-05,
      "step": 52220,
      "training_step_time": 7.4588940143585205
    },
    {
      "epoch": 4.980182647705078e-05,
      "model_forward_time": 2.9047646522521973,
      "step": 52221
    },
    {
      "epoch": 4.980182647705078e-05,
      "step": 52221,
      "training_step_time": 7.71361517906189
    },
    {
      "epoch": 4.980278015136719e-05,
      "model_forward_time": 1.7547461986541748,
      "step": 52222
    },
    {
      "epoch": 4.980278015136719e-05,
      "step": 52222,
      "training_step_time": 8.403784275054932
    },
    {
      "epoch": 4.980373382568359e-05,
      "model_forward_time": 1.7364237308502197,
      "step": 52223
    },
    {
      "epoch": 4.980373382568359e-05,
      "step": 52223,
      "training_step_time": 9.572058200836182
    },
    {
      "epoch": 4.98046875e-05,
      "model_forward_time": 1.7376468181610107,
      "step": 52224
    },
    {
      "epoch": 4.98046875e-05,
      "step": 52224,
      "training_step_time": 7.711780071258545
    },
    {
      "epoch": 4.980564117431641e-05,
      "model_forward_time": 1.7425167560577393,
      "step": 52225
    },
    {
      "epoch": 4.980564117431641e-05,
      "step": 52225,
      "training_step_time": 9.740967512130737
    },
    {
      "epoch": 4.980659484863281e-05,
      "model_forward_time": 1.742063283920288,
      "step": 52226
    },
    {
      "epoch": 4.980659484863281e-05,
      "step": 52226,
      "training_step_time": 7.671459913253784
    },
    {
      "epoch": 4.9807548522949216e-05,
      "model_forward_time": 1.728219747543335,
      "step": 52227
    },
    {
      "epoch": 4.9807548522949216e-05,
      "step": 52227,
      "training_step_time": 7.645723819732666
    },
    {
      "epoch": 4.9808502197265626e-05,
      "model_forward_time": 1.7428901195526123,
      "step": 52228
    },
    {
      "epoch": 4.9808502197265626e-05,
      "step": 52228,
      "training_step_time": 7.648075580596924
    },
    {
      "epoch": 4.980945587158203e-05,
      "model_forward_time": 1.735105276107788,
      "step": 52229
    },
    {
      "epoch": 4.980945587158203e-05,
      "step": 52229,
      "training_step_time": 7.434917449951172
    },
    {
      "epoch": 4.9810409545898435e-05,
      "grad_norm": 0.13310721516609192,
      "learning_rate": 5.044808432262932e-06,
      "loss": 0.059,
      "step": 52230
    },
    {
      "epoch": 4.9810409545898435e-05,
      "model_forward_time": 2.901587724685669,
      "step": 52230
    },
    {
      "dynamics_loss_avg": 0.042699347250163555,
      "epoch": 4.9810409545898435e-05,
      "step": 52230
    },
    {
      "action_loss_avg": 0.0014584023330826313,
      "epoch": 4.9810409545898435e-05,
      "step": 52230
    },
    {
      "epoch": 4.9810409545898435e-05,
      "step": 52230,
      "training_step_time": 7.740412712097168
    },
    {
      "epoch": 4.9811363220214845e-05,
      "model_forward_time": 1.769688606262207,
      "step": 52231
    },
    {
      "epoch": 4.9811363220214845e-05,
      "step": 52231,
      "training_step_time": 8.766782283782959
    },
    {
      "epoch": 4.981231689453125e-05,
      "model_forward_time": 1.8076972961425781,
      "step": 52232
    },
    {
      "epoch": 4.981231689453125e-05,
      "step": 52232,
      "training_step_time": 7.9382383823394775
    },
    {
      "epoch": 4.981327056884765e-05,
      "model_forward_time": 1.7445039749145508,
      "step": 52233
    },
    {
      "epoch": 4.981327056884765e-05,
      "step": 52233,
      "training_step_time": 8.67743444442749
    },
    {
      "epoch": 4.9814224243164064e-05,
      "model_forward_time": 1.746842622756958,
      "step": 52234
    },
    {
      "epoch": 4.9814224243164064e-05,
      "step": 52234,
      "training_step_time": 8.580434322357178
    },
    {
      "epoch": 4.981517791748047e-05,
      "model_forward_time": 1.7443387508392334,
      "step": 52235
    },
    {
      "epoch": 4.981517791748047e-05,
      "step": 52235,
      "training_step_time": 7.722302198410034
    },
    {
      "epoch": 4.981613159179687e-05,
      "model_forward_time": 1.7285244464874268,
      "step": 52236
    },
    {
      "epoch": 4.981613159179687e-05,
      "step": 52236,
      "training_step_time": 8.19957947731018
    },
    {
      "epoch": 4.981708526611328e-05,
      "model_forward_time": 1.7510192394256592,
      "step": 52237
    },
    {
      "epoch": 4.981708526611328e-05,
      "step": 52237,
      "training_step_time": 7.7761452198028564
    },
    {
      "epoch": 4.981803894042969e-05,
      "model_forward_time": 1.745847463607788,
      "step": 52238
    },
    {
      "epoch": 4.981803894042969e-05,
      "step": 52238,
      "training_step_time": 7.555703401565552
    },
    {
      "epoch": 4.981899261474609e-05,
      "model_forward_time": 2.8551788330078125,
      "step": 52239
    },
    {
      "epoch": 4.981899261474609e-05,
      "step": 52239,
      "training_step_time": 7.6757917404174805
    },
    {
      "epoch": 4.98199462890625e-05,
      "grad_norm": 0.10588329285383224,
      "learning_rate": 5.043155026423219e-06,
      "loss": 0.0562,
      "step": 52240
    },
    {
      "epoch": 4.98199462890625e-05,
      "model_forward_time": 1.7374603748321533,
      "step": 52240
    },
    {
      "dynamics_loss_avg": 0.05220065433531999,
      "epoch": 4.98199462890625e-05,
      "step": 52240
    },
    {
      "action_loss_avg": 0.0016197793593164533,
      "epoch": 4.98199462890625e-05,
      "step": 52240
    },
    {
      "epoch": 4.98199462890625e-05,
      "step": 52240,
      "training_step_time": 8.559664487838745
    },
    {
      "epoch": 4.9820899963378906e-05,
      "model_forward_time": 1.8079283237457275,
      "step": 52241
    },
    {
      "epoch": 4.9820899963378906e-05,
      "step": 52241,
      "training_step_time": 7.75502347946167
    },
    {
      "epoch": 4.982185363769531e-05,
      "model_forward_time": 1.7396211624145508,
      "step": 52242
    },
    {
      "epoch": 4.982185363769531e-05,
      "step": 52242,
      "training_step_time": 7.6933794021606445
    },
    {
      "epoch": 4.982280731201172e-05,
      "model_forward_time": 1.7414929866790771,
      "step": 52243
    },
    {
      "epoch": 4.982280731201172e-05,
      "step": 52243,
      "training_step_time": 7.728330373764038
    },
    {
      "epoch": 4.9823760986328125e-05,
      "model_forward_time": 1.74241042137146,
      "step": 52244
    },
    {
      "epoch": 4.9823760986328125e-05,
      "step": 52244,
      "training_step_time": 8.739332437515259
    },
    {
      "epoch": 4.982471466064453e-05,
      "model_forward_time": 1.7459702491760254,
      "step": 52245
    },
    {
      "epoch": 4.982471466064453e-05,
      "step": 52245,
      "training_step_time": 7.62668776512146
    },
    {
      "epoch": 4.982566833496094e-05,
      "model_forward_time": 1.7456879615783691,
      "step": 52246
    },
    {
      "epoch": 4.982566833496094e-05,
      "step": 52246,
      "training_step_time": 7.708514213562012
    },
    {
      "epoch": 4.9826622009277344e-05,
      "model_forward_time": 1.7322719097137451,
      "step": 52247
    },
    {
      "epoch": 4.9826622009277344e-05,
      "step": 52247,
      "training_step_time": 7.452667474746704
    },
    {
      "epoch": 4.982757568359375e-05,
      "model_forward_time": 2.8456180095672607,
      "step": 52248
    },
    {
      "epoch": 4.982757568359375e-05,
      "step": 52248,
      "training_step_time": 8.403575658798218
    },
    {
      "epoch": 4.982852935791016e-05,
      "model_forward_time": 1.763362169265747,
      "step": 52249
    },
    {
      "epoch": 4.982852935791016e-05,
      "step": 52249,
      "training_step_time": 7.503538608551025
    },
    {
      "epoch": 4.982948303222656e-05,
      "grad_norm": 0.11343158781528473,
      "learning_rate": 5.0415016158641375e-06,
      "loss": 0.0599,
      "step": 52250
    },
    {
      "epoch": 4.982948303222656e-05,
      "model_forward_time": 1.782214641571045,
      "step": 52250
    },
    {
      "dynamics_loss_avg": 0.0516071755439043,
      "epoch": 4.982948303222656e-05,
      "step": 52250
    },
    {
      "action_loss_avg": 0.001510590559337288,
      "epoch": 4.982948303222656e-05,
      "step": 52250
    },
    {
      "epoch": 4.982948303222656e-05,
      "step": 52250,
      "training_step_time": 7.832578420639038
    },
    {
      "epoch": 4.9830436706542967e-05,
      "model_forward_time": 1.7344436645507812,
      "step": 52251
    },
    {
      "epoch": 4.9830436706542967e-05,
      "step": 52251,
      "training_step_time": 8.81623911857605
    },
    {
      "epoch": 4.983139038085938e-05,
      "model_forward_time": 1.7465355396270752,
      "step": 52252
    },
    {
      "epoch": 4.983139038085938e-05,
      "step": 52252,
      "training_step_time": 8.384680271148682
    },
    {
      "epoch": 4.983234405517578e-05,
      "model_forward_time": 1.7437639236450195,
      "step": 52253
    },
    {
      "epoch": 4.983234405517578e-05,
      "step": 52253,
      "training_step_time": 7.994017839431763
    },
    {
      "epoch": 4.9833297729492185e-05,
      "model_forward_time": 1.7370414733886719,
      "step": 52254
    },
    {
      "epoch": 4.9833297729492185e-05,
      "step": 52254,
      "training_step_time": 7.739712238311768
    },
    {
      "epoch": 4.9834251403808596e-05,
      "model_forward_time": 1.7448110580444336,
      "step": 52255
    },
    {
      "epoch": 4.9834251403808596e-05,
      "step": 52255,
      "training_step_time": 8.51240086555481
    },
    {
      "epoch": 4.9835205078125e-05,
      "model_forward_time": 1.7393546104431152,
      "step": 52256
    },
    {
      "epoch": 4.9835205078125e-05,
      "step": 52256,
      "training_step_time": 7.470317840576172
    },
    {
      "epoch": 4.9836158752441404e-05,
      "model_forward_time": 1.7471978664398193,
      "step": 52257
    },
    {
      "epoch": 4.9836158752441404e-05,
      "step": 52257,
      "training_step_time": 8.640965700149536
    },
    {
      "epoch": 4.9837112426757815e-05,
      "model_forward_time": 1.7352735996246338,
      "step": 52258
    },
    {
      "epoch": 4.9837112426757815e-05,
      "step": 52258,
      "training_step_time": 7.558546781539917
    },
    {
      "epoch": 4.983806610107422e-05,
      "model_forward_time": 1.7524034976959229,
      "step": 52259
    },
    {
      "epoch": 4.983806610107422e-05,
      "step": 52259,
      "training_step_time": 8.491022109985352
    },
    {
      "epoch": 4.983901977539062e-05,
      "grad_norm": 0.11661522090435028,
      "learning_rate": 5.039848200766502e-06,
      "loss": 0.0559,
      "step": 52260
    },
    {
      "epoch": 4.983901977539062e-05,
      "model_forward_time": 1.7455685138702393,
      "step": 52260
    },
    {
      "dynamics_loss_avg": 0.04977133460342884,
      "epoch": 4.983901977539062e-05,
      "step": 52260
    },
    {
      "action_loss_avg": 0.0011674488720018416,
      "epoch": 4.983901977539062e-05,
      "step": 52260
    },
    {
      "epoch": 4.983901977539062e-05,
      "step": 52260,
      "training_step_time": 7.725205898284912
    },
    {
      "epoch": 4.9839973449707034e-05,
      "model_forward_time": 1.739262342453003,
      "step": 52261
    },
    {
      "epoch": 4.9839973449707034e-05,
      "step": 52261,
      "training_step_time": 8.714957475662231
    },
    {
      "epoch": 4.984092712402344e-05,
      "model_forward_time": 1.7469301223754883,
      "step": 52262
    },
    {
      "epoch": 4.984092712402344e-05,
      "step": 52262,
      "training_step_time": 7.883707761764526
    },
    {
      "epoch": 4.984188079833984e-05,
      "model_forward_time": 1.743079423904419,
      "step": 52263
    },
    {
      "epoch": 4.984188079833984e-05,
      "step": 52263,
      "training_step_time": 8.657285451889038
    },
    {
      "epoch": 4.984283447265625e-05,
      "model_forward_time": 1.7413296699523926,
      "step": 52264
    },
    {
      "epoch": 4.984283447265625e-05,
      "step": 52264,
      "training_step_time": 7.675600290298462
    },
    {
      "epoch": 4.984378814697266e-05,
      "model_forward_time": 1.7507450580596924,
      "step": 52265
    },
    {
      "epoch": 4.984378814697266e-05,
      "step": 52265,
      "training_step_time": 7.715378284454346
    },
    {
      "epoch": 4.984474182128906e-05,
      "model_forward_time": 1.7464463710784912,
      "step": 52266
    },
    {
      "epoch": 4.984474182128906e-05,
      "step": 52266,
      "training_step_time": 7.533651351928711
    },
    {
      "epoch": 4.984569549560547e-05,
      "model_forward_time": 2.89756178855896,
      "step": 52267
    },
    {
      "epoch": 4.984569549560547e-05,
      "step": 52267,
      "training_step_time": 9.168167352676392
    },
    {
      "epoch": 4.9846649169921876e-05,
      "model_forward_time": 1.3834128379821777,
      "step": 52268
    },
    {
      "epoch": 4.9846649169921876e-05,
      "step": 52268,
      "training_step_time": 8.247253179550171
    },
    {
      "epoch": 4.984760284423828e-05,
      "model_forward_time": 1.8101623058319092,
      "step": 52269
    },
    {
      "epoch": 4.984760284423828e-05,
      "step": 52269,
      "training_step_time": 8.562739849090576
    },
    {
      "epoch": 4.984855651855469e-05,
      "grad_norm": 0.12100545316934586,
      "learning_rate": 5.038194781311125e-06,
      "loss": 0.0599,
      "step": 52270
    },
    {
      "epoch": 4.984855651855469e-05,
      "model_forward_time": 1.7401576042175293,
      "step": 52270
    },
    {
      "dynamics_loss_avg": 0.05608753822743893,
      "epoch": 4.984855651855469e-05,
      "step": 52270
    },
    {
      "action_loss_avg": 0.002232231356902048,
      "epoch": 4.984855651855469e-05,
      "step": 52270
    },
    {
      "epoch": 4.984855651855469e-05,
      "step": 52270,
      "training_step_time": 7.680909633636475
    },
    {
      "epoch": 4.9849510192871095e-05,
      "model_forward_time": 1.7512423992156982,
      "step": 52271
    },
    {
      "epoch": 4.9849510192871095e-05,
      "step": 52271,
      "training_step_time": 7.821767091751099
    },
    {
      "epoch": 4.98504638671875e-05,
      "model_forward_time": 1.7425522804260254,
      "step": 52272
    },
    {
      "epoch": 4.98504638671875e-05,
      "step": 52272,
      "training_step_time": 7.637434959411621
    },
    {
      "epoch": 4.985141754150391e-05,
      "model_forward_time": 1.7345659732818604,
      "step": 52273
    },
    {
      "epoch": 4.985141754150391e-05,
      "step": 52273,
      "training_step_time": 8.651010036468506
    },
    {
      "epoch": 4.985237121582031e-05,
      "model_forward_time": 1.7469890117645264,
      "step": 52274
    },
    {
      "epoch": 4.985237121582031e-05,
      "step": 52274,
      "training_step_time": 7.6934075355529785
    },
    {
      "epoch": 4.985332489013672e-05,
      "model_forward_time": 1.747018575668335,
      "step": 52275
    },
    {
      "epoch": 4.985332489013672e-05,
      "step": 52275,
      "training_step_time": 7.950532674789429
    },
    {
      "epoch": 4.985427856445313e-05,
      "model_forward_time": 2.892124891281128,
      "step": 52276
    },
    {
      "epoch": 4.985427856445313e-05,
      "step": 52276,
      "training_step_time": 7.7554473876953125
    },
    {
      "epoch": 4.985523223876953e-05,
      "model_forward_time": 1.7412924766540527,
      "step": 52277
    },
    {
      "epoch": 4.985523223876953e-05,
      "step": 52277,
      "training_step_time": 7.685718536376953
    },
    {
      "epoch": 4.9856185913085936e-05,
      "model_forward_time": 1.8194022178649902,
      "step": 52278
    },
    {
      "epoch": 4.9856185913085936e-05,
      "step": 52278,
      "training_step_time": 8.271252155303955
    },
    {
      "epoch": 4.985713958740235e-05,
      "model_forward_time": 1.770484209060669,
      "step": 52279
    },
    {
      "epoch": 4.985713958740235e-05,
      "step": 52279,
      "training_step_time": 8.90751051902771
    },
    {
      "epoch": 4.985809326171875e-05,
      "grad_norm": 0.12766966223716736,
      "learning_rate": 5.0365413576788266e-06,
      "loss": 0.0583,
      "step": 52280
    },
    {
      "epoch": 4.985809326171875e-05,
      "model_forward_time": 1.73805570602417,
      "step": 52280
    },
    {
      "dynamics_loss_avg": 0.05671066138893366,
      "epoch": 4.985809326171875e-05,
      "step": 52280
    },
    {
      "action_loss_avg": 0.002042869245633483,
      "epoch": 4.985809326171875e-05,
      "step": 52280
    },
    {
      "epoch": 4.985809326171875e-05,
      "step": 52280,
      "training_step_time": 7.894614934921265
    },
    {
      "epoch": 4.9859046936035155e-05,
      "model_forward_time": 1.742079496383667,
      "step": 52281
    },
    {
      "epoch": 4.9859046936035155e-05,
      "step": 52281,
      "training_step_time": 9.162356615066528
    },
    {
      "epoch": 4.986000061035156e-05,
      "model_forward_time": 1.7426297664642334,
      "step": 52282
    },
    {
      "epoch": 4.986000061035156e-05,
      "step": 52282,
      "training_step_time": 7.738566875457764
    },
    {
      "epoch": 4.986095428466797e-05,
      "model_forward_time": 1.742088794708252,
      "step": 52283
    },
    {
      "epoch": 4.986095428466797e-05,
      "step": 52283,
      "training_step_time": 7.700880289077759
    },
    {
      "epoch": 4.9861907958984374e-05,
      "model_forward_time": 1.7433037757873535,
      "step": 52284
    },
    {
      "epoch": 4.9861907958984374e-05,
      "step": 52284,
      "training_step_time": 8.407611608505249
    },
    {
      "epoch": 4.986286163330078e-05,
      "model_forward_time": 2.8112053871154785,
      "step": 52285
    },
    {
      "epoch": 4.986286163330078e-05,
      "step": 52285,
      "training_step_time": 8.529258251190186
    },
    {
      "epoch": 4.986381530761719e-05,
      "model_forward_time": 1.7412710189819336,
      "step": 52286
    },
    {
      "epoch": 4.986381530761719e-05,
      "step": 52286,
      "training_step_time": 7.61621356010437
    },
    {
      "epoch": 4.986476898193359e-05,
      "model_forward_time": 1.8137426376342773,
      "step": 52287
    },
    {
      "epoch": 4.986476898193359e-05,
      "step": 52287,
      "training_step_time": 10.04748272895813
    },
    {
      "epoch": 4.986572265625e-05,
      "model_forward_time": 1.7427105903625488,
      "step": 52288
    },
    {
      "epoch": 4.986572265625e-05,
      "step": 52288,
      "training_step_time": 8.592679738998413
    },
    {
      "epoch": 4.986667633056641e-05,
      "model_forward_time": 1.738069772720337,
      "step": 52289
    },
    {
      "epoch": 4.986667633056641e-05,
      "step": 52289,
      "training_step_time": 8.925548315048218
    },
    {
      "epoch": 4.986763000488281e-05,
      "grad_norm": 0.1378113478422165,
      "learning_rate": 5.0348879300504195e-06,
      "loss": 0.0579,
      "step": 52290
    },
    {
      "epoch": 4.986763000488281e-05,
      "model_forward_time": 1.7391674518585205,
      "step": 52290
    },
    {
      "dynamics_loss_avg": 0.06823898255825042,
      "epoch": 4.986763000488281e-05,
      "step": 52290
    },
    {
      "action_loss_avg": 0.0020707822928670793,
      "epoch": 4.986763000488281e-05,
      "step": 52290
    },
    {
      "epoch": 4.986763000488281e-05,
      "step": 52290,
      "training_step_time": 7.808875799179077
    },
    {
      "epoch": 4.9868583679199216e-05,
      "model_forward_time": 1.734048843383789,
      "step": 52291
    },
    {
      "epoch": 4.9868583679199216e-05,
      "step": 52291,
      "training_step_time": 9.268354177474976
    },
    {
      "epoch": 4.9869537353515627e-05,
      "model_forward_time": 1.7403738498687744,
      "step": 52292
    },
    {
      "epoch": 4.9869537353515627e-05,
      "step": 52292,
      "training_step_time": 8.431320428848267
    },
    {
      "epoch": 4.987049102783203e-05,
      "model_forward_time": 1.7444889545440674,
      "step": 52293
    },
    {
      "epoch": 4.987049102783203e-05,
      "step": 52293,
      "training_step_time": 7.651371717453003
    },
    {
      "epoch": 4.9871444702148435e-05,
      "model_forward_time": 2.2248215675354004,
      "step": 52294
    },
    {
      "epoch": 4.9871444702148435e-05,
      "step": 52294,
      "training_step_time": 7.43113374710083
    },
    {
      "epoch": 4.9872398376464845e-05,
      "model_forward_time": 1.7107839584350586,
      "step": 52295
    },
    {
      "epoch": 4.9872398376464845e-05,
      "step": 52295,
      "training_step_time": 8.416774988174438
    },
    {
      "epoch": 4.987335205078125e-05,
      "model_forward_time": 1.6886670589447021,
      "step": 52296
    },
    {
      "epoch": 4.987335205078125e-05,
      "step": 52296,
      "training_step_time": 8.272152662277222
    },
    {
      "epoch": 4.9874305725097653e-05,
      "model_forward_time": 1.6916866302490234,
      "step": 52297
    },
    {
      "epoch": 4.9874305725097653e-05,
      "step": 52297,
      "training_step_time": 8.343781232833862
    },
    {
      "epoch": 4.9875259399414064e-05,
      "model_forward_time": 1.3157925605773926,
      "step": 52298
    },
    {
      "epoch": 4.9875259399414064e-05,
      "step": 52298,
      "training_step_time": 7.694486856460571
    },
    {
      "epoch": 4.987621307373047e-05,
      "model_forward_time": 1.6892318725585938,
      "step": 52299
    },
    {
      "epoch": 4.987621307373047e-05,
      "step": 52299,
      "training_step_time": 9.209145069122314
    },
    {
      "epoch": 4.987716674804687e-05,
      "grad_norm": 0.24948835372924805,
      "learning_rate": 5.033234498606722e-06,
      "loss": 0.0608,
      "step": 52300
    },
    {
      "epoch": 4.987716674804687e-05,
      "model_forward_time": 1.7193424701690674,
      "step": 52300
    },
    {
      "dynamics_loss_avg": 0.044264578819274904,
      "epoch": 4.987716674804687e-05,
      "step": 52300
    },
    {
      "action_loss_avg": 0.0009575549920555204,
      "epoch": 4.987716674804687e-05,
      "step": 52300
    },
    {
      "epoch": 4.987716674804687e-05,
      "step": 52300,
      "training_step_time": 7.446057319641113
    },
    {
      "epoch": 4.987812042236328e-05,
      "model_forward_time": 1.703336238861084,
      "step": 52301
    },
    {
      "epoch": 4.987812042236328e-05,
      "step": 52301,
      "training_step_time": 7.591981887817383
    },
    {
      "epoch": 4.987907409667969e-05,
      "model_forward_time": 1.713747262954712,
      "step": 52302
    },
    {
      "epoch": 4.987907409667969e-05,
      "step": 52302,
      "training_step_time": 7.4317193031311035
    },
    {
      "epoch": 4.988002777099609e-05,
      "model_forward_time": 1.7008976936340332,
      "step": 52303
    },
    {
      "epoch": 4.988002777099609e-05,
      "step": 52303,
      "training_step_time": 7.44830060005188
    },
    {
      "epoch": 4.98809814453125e-05,
      "model_forward_time": 2.2670414447784424,
      "step": 52304
    },
    {
      "epoch": 4.98809814453125e-05,
      "step": 52304,
      "training_step_time": 8.426246643066406
    },
    {
      "epoch": 4.9881935119628906e-05,
      "model_forward_time": 1.7068064212799072,
      "step": 52305
    },
    {
      "epoch": 4.9881935119628906e-05,
      "step": 52305,
      "training_step_time": 7.551012992858887
    },
    {
      "epoch": 4.988288879394531e-05,
      "model_forward_time": 1.7000095844268799,
      "step": 52306
    },
    {
      "epoch": 4.988288879394531e-05,
      "step": 52306,
      "training_step_time": 7.644815683364868
    },
    {
      "epoch": 4.988384246826172e-05,
      "model_forward_time": 1.7073843479156494,
      "step": 52307
    },
    {
      "epoch": 4.988384246826172e-05,
      "step": 52307,
      "training_step_time": 7.699576377868652
    },
    {
      "epoch": 4.9884796142578125e-05,
      "model_forward_time": 1.721975326538086,
      "step": 52308
    },
    {
      "epoch": 4.9884796142578125e-05,
      "step": 52308,
      "training_step_time": 9.270188808441162
    },
    {
      "epoch": 4.988574981689453e-05,
      "model_forward_time": 1.711296796798706,
      "step": 52309
    },
    {
      "epoch": 4.988574981689453e-05,
      "step": 52309,
      "training_step_time": 8.443307876586914
    },
    {
      "epoch": 4.988670349121094e-05,
      "grad_norm": 0.13518117368221283,
      "learning_rate": 5.0315810635285476e-06,
      "loss": 0.0559,
      "step": 52310
    },
    {
      "epoch": 4.988670349121094e-05,
      "model_forward_time": 1.7073063850402832,
      "step": 52310
    },
    {
      "dynamics_loss_avg": 0.06314704790711403,
      "epoch": 4.988670349121094e-05,
      "step": 52310
    },
    {
      "action_loss_avg": 0.0018355846026679502,
      "epoch": 4.988670349121094e-05,
      "step": 52310
    },
    {
      "epoch": 4.988670349121094e-05,
      "step": 52310,
      "training_step_time": 8.411718606948853
    },
    {
      "epoch": 4.9887657165527344e-05,
      "model_forward_time": 1.697216272354126,
      "step": 52311
    },
    {
      "epoch": 4.9887657165527344e-05,
      "step": 52311,
      "training_step_time": 7.486424207687378
    },
    {
      "epoch": 4.988861083984375e-05,
      "model_forward_time": 1.7044332027435303,
      "step": 52312
    },
    {
      "epoch": 4.988861083984375e-05,
      "step": 52312,
      "training_step_time": 7.378559112548828
    },
    {
      "epoch": 4.988956451416016e-05,
      "model_forward_time": 2.2726898193359375,
      "step": 52313
    },
    {
      "epoch": 4.988956451416016e-05,
      "step": 52313,
      "training_step_time": 7.343076229095459
    },
    {
      "epoch": 4.989051818847656e-05,
      "model_forward_time": 1.696448564529419,
      "step": 52314
    },
    {
      "epoch": 4.989051818847656e-05,
      "step": 52314,
      "training_step_time": 7.620826721191406
    },
    {
      "epoch": 4.9891471862792967e-05,
      "model_forward_time": 1.6922094821929932,
      "step": 52315
    },
    {
      "epoch": 4.9891471862792967e-05,
      "step": 52315,
      "training_step_time": 7.846137523651123
    },
    {
      "epoch": 4.989242553710938e-05,
      "model_forward_time": 1.7072038650512695,
      "step": 52316
    },
    {
      "epoch": 4.989242553710938e-05,
      "step": 52316,
      "training_step_time": 7.914285182952881
    },
    {
      "epoch": 4.989337921142578e-05,
      "model_forward_time": 1.6989128589630127,
      "step": 52317
    },
    {
      "epoch": 4.989337921142578e-05,
      "step": 52317,
      "training_step_time": 7.841003179550171
    },
    {
      "epoch": 4.9894332885742185e-05,
      "model_forward_time": 1.7085528373718262,
      "step": 52318
    },
    {
      "epoch": 4.9894332885742185e-05,
      "step": 52318,
      "training_step_time": 8.631009578704834
    },
    {
      "epoch": 4.9895286560058596e-05,
      "model_forward_time": 1.6997301578521729,
      "step": 52319
    },
    {
      "epoch": 4.9895286560058596e-05,
      "step": 52319,
      "training_step_time": 7.734433889389038
    },
    {
      "epoch": 4.9896240234375e-05,
      "grad_norm": 0.14173035323619843,
      "learning_rate": 5.0299276249967165e-06,
      "loss": 0.0531,
      "step": 52320
    },
    {
      "epoch": 4.9896240234375e-05,
      "model_forward_time": 1.7102539539337158,
      "step": 52320
    },
    {
      "dynamics_loss_avg": 0.04009000426158309,
      "epoch": 4.9896240234375e-05,
      "step": 52320
    },
    {
      "action_loss_avg": 0.0013489603239577263,
      "epoch": 4.9896240234375e-05,
      "step": 52320
    },
    {
      "epoch": 4.9896240234375e-05,
      "step": 52320,
      "training_step_time": 7.601189136505127
    },
    {
      "epoch": 4.9897193908691404e-05,
      "model_forward_time": 1.715876817703247,
      "step": 52321
    },
    {
      "epoch": 4.9897193908691404e-05,
      "step": 52321,
      "training_step_time": 8.276672601699829
    },
    {
      "epoch": 4.9898147583007815e-05,
      "model_forward_time": 2.3018226623535156,
      "step": 52322
    },
    {
      "epoch": 4.9898147583007815e-05,
      "step": 52322,
      "training_step_time": 10.332092761993408
    },
    {
      "epoch": 4.989910125732422e-05,
      "model_forward_time": 1.717301607131958,
      "step": 52323
    },
    {
      "epoch": 4.989910125732422e-05,
      "step": 52323,
      "training_step_time": 9.187707901000977
    },
    {
      "epoch": 4.990005493164062e-05,
      "model_forward_time": 1.3324894905090332,
      "step": 52324
    },
    {
      "epoch": 4.990005493164062e-05,
      "step": 52324,
      "training_step_time": 8.858553409576416
    },
    {
      "epoch": 4.9901008605957034e-05,
      "model_forward_time": 1.7058796882629395,
      "step": 52325
    },
    {
      "epoch": 4.9901008605957034e-05,
      "step": 52325,
      "training_step_time": 8.886587858200073
    },
    {
      "epoch": 4.990196228027344e-05,
      "model_forward_time": 1.702204942703247,
      "step": 52326
    },
    {
      "epoch": 4.990196228027344e-05,
      "step": 52326,
      "training_step_time": 7.786473751068115
    },
    {
      "epoch": 4.990291595458984e-05,
      "model_forward_time": 1.7086107730865479,
      "step": 52327
    },
    {
      "epoch": 4.990291595458984e-05,
      "step": 52327,
      "training_step_time": 7.692054033279419
    },
    {
      "epoch": 4.990386962890625e-05,
      "model_forward_time": 1.7239158153533936,
      "step": 52328
    },
    {
      "epoch": 4.990386962890625e-05,
      "step": 52328,
      "training_step_time": 7.610040664672852
    },
    {
      "epoch": 4.990482330322266e-05,
      "model_forward_time": 1.7145164012908936,
      "step": 52329
    },
    {
      "epoch": 4.990482330322266e-05,
      "step": 52329,
      "training_step_time": 7.590050220489502
    },
    {
      "epoch": 4.990577697753906e-05,
      "grad_norm": 0.1257433295249939,
      "learning_rate": 5.028274183192046e-06,
      "loss": 0.0581,
      "step": 52330
    },
    {
      "epoch": 4.990577697753906e-05,
      "model_forward_time": 1.7134130001068115,
      "step": 52330
    },
    {
      "dynamics_loss_avg": 0.07143500708043575,
      "epoch": 4.990577697753906e-05,
      "step": 52330
    },
    {
      "action_loss_avg": 0.002549310075119138,
      "epoch": 4.990577697753906e-05,
      "step": 52330
    },
    {
      "epoch": 4.990577697753906e-05,
      "step": 52330,
      "training_step_time": 7.575430154800415
    },
    {
      "epoch": 4.990673065185547e-05,
      "model_forward_time": 2.238192081451416,
      "step": 52331
    },
    {
      "epoch": 4.990673065185547e-05,
      "step": 52331,
      "training_step_time": 8.356945753097534
    },
    {
      "epoch": 4.9907684326171876e-05,
      "model_forward_time": 1.7076137065887451,
      "step": 52332
    },
    {
      "epoch": 4.9907684326171876e-05,
      "step": 52332,
      "training_step_time": 7.5093584060668945
    },
    {
      "epoch": 4.990863800048828e-05,
      "model_forward_time": 1.711501121520996,
      "step": 52333
    },
    {
      "epoch": 4.990863800048828e-05,
      "step": 52333,
      "training_step_time": 9.405907154083252
    },
    {
      "epoch": 4.990959167480469e-05,
      "model_forward_time": 1.7162220478057861,
      "step": 52334
    },
    {
      "epoch": 4.990959167480469e-05,
      "step": 52334,
      "training_step_time": 7.704524278640747
    },
    {
      "epoch": 4.9910545349121095e-05,
      "model_forward_time": 1.7092571258544922,
      "step": 52335
    },
    {
      "epoch": 4.9910545349121095e-05,
      "step": 52335,
      "training_step_time": 9.891620635986328
    },
    {
      "epoch": 4.99114990234375e-05,
      "model_forward_time": 1.6991312503814697,
      "step": 52336
    },
    {
      "epoch": 4.99114990234375e-05,
      "step": 52336,
      "training_step_time": 7.787790536880493
    },
    {
      "epoch": 4.991245269775391e-05,
      "model_forward_time": 1.7110779285430908,
      "step": 52337
    },
    {
      "epoch": 4.991245269775391e-05,
      "step": 52337,
      "training_step_time": 7.509098529815674
    },
    {
      "epoch": 4.9913406372070313e-05,
      "model_forward_time": 1.7119641304016113,
      "step": 52338
    },
    {
      "epoch": 4.9913406372070313e-05,
      "step": 52338,
      "training_step_time": 9.048474788665771
    },
    {
      "epoch": 4.991436004638672e-05,
      "model_forward_time": 1.7066597938537598,
      "step": 52339
    },
    {
      "epoch": 4.991436004638672e-05,
      "step": 52339,
      "training_step_time": 6.830152273178101
    },
    {
      "epoch": 4.991531372070313e-05,
      "grad_norm": 0.1261555701494217,
      "learning_rate": 5.026620738295351e-06,
      "loss": 0.0585,
      "step": 52340
    },
    {
      "epoch": 4.991531372070313e-05,
      "model_forward_time": 1.7078261375427246,
      "step": 52340
    },
    {
      "dynamics_loss_avg": 0.05175515208393335,
      "epoch": 4.991531372070313e-05,
      "step": 52340
    },
    {
      "action_loss_avg": 0.0014290444960352033,
      "epoch": 4.991531372070313e-05,
      "step": 52340
    },
    {
      "epoch": 4.991531372070313e-05,
      "step": 52340,
      "training_step_time": 7.4861674308776855
    },
    {
      "epoch": 4.991626739501953e-05,
      "model_forward_time": 2.2844269275665283,
      "step": 52341
    },
    {
      "epoch": 4.991626739501953e-05,
      "step": 52341,
      "training_step_time": 7.482975006103516
    },
    {
      "epoch": 4.9917221069335936e-05,
      "model_forward_time": 1.6936628818511963,
      "step": 52342
    },
    {
      "epoch": 4.9917221069335936e-05,
      "step": 52342,
      "training_step_time": 7.66801905632019
    },
    {
      "epoch": 4.991817474365235e-05,
      "model_forward_time": 1.7048466205596924,
      "step": 52343
    },
    {
      "epoch": 4.991817474365235e-05,
      "step": 52343,
      "training_step_time": 8.995000839233398
    },
    {
      "epoch": 4.991912841796875e-05,
      "model_forward_time": 1.7045693397521973,
      "step": 52344
    },
    {
      "epoch": 4.991912841796875e-05,
      "step": 52344,
      "training_step_time": 7.5972065925598145
    },
    {
      "epoch": 4.9920082092285155e-05,
      "model_forward_time": 1.6923549175262451,
      "step": 52345
    },
    {
      "epoch": 4.9920082092285155e-05,
      "step": 52345,
      "training_step_time": 7.633184432983398
    },
    {
      "epoch": 4.992103576660156e-05,
      "model_forward_time": 1.6884541511535645,
      "step": 52346
    },
    {
      "epoch": 4.992103576660156e-05,
      "step": 52346,
      "training_step_time": 7.404040098190308
    },
    {
      "epoch": 4.992198944091797e-05,
      "model_forward_time": 1.6968119144439697,
      "step": 52347
    },
    {
      "epoch": 4.992198944091797e-05,
      "step": 52347,
      "training_step_time": 7.448263883590698
    },
    {
      "epoch": 4.9922943115234374e-05,
      "model_forward_time": 1.6970551013946533,
      "step": 52348
    },
    {
      "epoch": 4.9922943115234374e-05,
      "step": 52348,
      "training_step_time": 7.5933849811553955
    },
    {
      "epoch": 4.992389678955078e-05,
      "model_forward_time": 1.6875965595245361,
      "step": 52349
    },
    {
      "epoch": 4.992389678955078e-05,
      "step": 52349,
      "training_step_time": 7.551614999771118
    },
    {
      "epoch": 4.992485046386719e-05,
      "grad_norm": 0.08272241801023483,
      "learning_rate": 5.024967290487455e-06,
      "loss": 0.0558,
      "step": 52350
    },
    {
      "epoch": 4.992485046386719e-05,
      "model_forward_time": 2.251798629760742,
      "step": 52350
    },
    {
      "dynamics_loss_avg": 0.04551154151558876,
      "epoch": 4.992485046386719e-05,
      "step": 52350
    },
    {
      "action_loss_avg": 0.001796392851974815,
      "epoch": 4.992485046386719e-05,
      "step": 52350
    },
    {
      "epoch": 4.992485046386719e-05,
      "step": 52350,
      "training_step_time": 7.411077976226807
    },
    {
      "epoch": 4.992580413818359e-05,
      "model_forward_time": 1.679018497467041,
      "step": 52351
    },
    {
      "epoch": 4.992580413818359e-05,
      "step": 52351,
      "training_step_time": 8.779308795928955
    },
    {
      "epoch": 4.99267578125e-05,
      "model_forward_time": 1.6872107982635498,
      "step": 52352
    },
    {
      "epoch": 4.99267578125e-05,
      "step": 52352,
      "training_step_time": 9.38917851448059
    },
    {
      "epoch": 4.992771148681641e-05,
      "model_forward_time": 1.6890981197357178,
      "step": 52353
    },
    {
      "epoch": 4.992771148681641e-05,
      "step": 52353,
      "training_step_time": 6.7283852100372314
    },
    {
      "epoch": 4.992866516113281e-05,
      "model_forward_time": 1.6942682266235352,
      "step": 52354
    },
    {
      "epoch": 4.992866516113281e-05,
      "step": 52354,
      "training_step_time": 6.710377216339111
    },
    {
      "epoch": 4.9929618835449216e-05,
      "model_forward_time": 1.6940481662750244,
      "step": 52355
    },
    {
      "epoch": 4.9929618835449216e-05,
      "step": 52355,
      "training_step_time": 6.740464687347412
    },
    {
      "epoch": 4.9930572509765627e-05,
      "model_forward_time": 1.688791275024414,
      "step": 52356
    },
    {
      "epoch": 4.9930572509765627e-05,
      "step": 52356,
      "training_step_time": 6.723775863647461
    },
    {
      "epoch": 4.993152618408203e-05,
      "model_forward_time": 1.6903729438781738,
      "step": 52357
    },
    {
      "epoch": 4.993152618408203e-05,
      "step": 52357,
      "training_step_time": 6.704449415206909
    },
    {
      "epoch": 4.9932479858398435e-05,
      "model_forward_time": 1.6937134265899658,
      "step": 52358
    },
    {
      "epoch": 4.9932479858398435e-05,
      "step": 52358,
      "training_step_time": 6.6954185962677
    },
    {
      "epoch": 4.9933433532714845e-05,
      "model_forward_time": 1.6836490631103516,
      "step": 52359
    },
    {
      "epoch": 4.9933433532714845e-05,
      "step": 52359,
      "training_step_time": 6.7141196727752686
    },
    {
      "epoch": 4.993438720703125e-05,
      "grad_norm": 0.12646673619747162,
      "learning_rate": 5.023313839949172e-06,
      "loss": 0.059,
      "step": 52360
    },
    {
      "epoch": 4.993438720703125e-05,
      "model_forward_time": 1.689119815826416,
      "step": 52360
    },
    {
      "dynamics_loss_avg": 0.058983008936047555,
      "epoch": 4.993438720703125e-05,
      "step": 52360
    },
    {
      "action_loss_avg": 0.0016646652453346177,
      "epoch": 4.993438720703125e-05,
      "step": 52360
    },
    {
      "epoch": 4.993438720703125e-05,
      "step": 52360,
      "training_step_time": 6.672078609466553
    },
    {
      "epoch": 4.9935340881347653e-05,
      "model_forward_time": 2.258854866027832,
      "step": 52361
    },
    {
      "epoch": 4.9935340881347653e-05,
      "step": 52361,
      "training_step_time": 7.159581661224365
    },
    {
      "epoch": 4.9936294555664064e-05,
      "model_forward_time": 1.6920864582061768,
      "step": 52362
    },
    {
      "epoch": 4.9936294555664064e-05,
      "step": 52362,
      "training_step_time": 7.618001461029053
    },
    {
      "epoch": 4.993724822998047e-05,
      "model_forward_time": 1.6939947605133057,
      "step": 52363
    },
    {
      "epoch": 4.993724822998047e-05,
      "step": 52363,
      "training_step_time": 6.726116180419922
    },
    {
      "epoch": 4.993820190429687e-05,
      "model_forward_time": 1.6905696392059326,
      "step": 52364
    },
    {
      "epoch": 4.993820190429687e-05,
      "step": 52364,
      "training_step_time": 6.725393056869507
    },
    {
      "epoch": 4.993915557861328e-05,
      "model_forward_time": 1.6902031898498535,
      "step": 52365
    },
    {
      "epoch": 4.993915557861328e-05,
      "step": 52365,
      "training_step_time": 6.756520748138428
    },
    {
      "epoch": 4.994010925292969e-05,
      "model_forward_time": 1.6892566680908203,
      "step": 52366
    },
    {
      "epoch": 4.994010925292969e-05,
      "step": 52366,
      "training_step_time": 6.730974435806274
    },
    {
      "epoch": 4.994106292724609e-05,
      "model_forward_time": 1.6867609024047852,
      "step": 52367
    },
    {
      "epoch": 4.994106292724609e-05,
      "step": 52367,
      "training_step_time": 6.72836446762085
    },
    {
      "epoch": 4.99420166015625e-05,
      "model_forward_time": 1.6898834705352783,
      "step": 52368
    },
    {
      "epoch": 4.99420166015625e-05,
      "step": 52368,
      "training_step_time": 6.75276255607605
    },
    {
      "epoch": 4.9942970275878906e-05,
      "model_forward_time": 1.6876158714294434,
      "step": 52369
    },
    {
      "epoch": 4.9942970275878906e-05,
      "step": 52369,
      "training_step_time": 6.820490121841431
    },
    {
      "epoch": 4.994392395019531e-05,
      "grad_norm": 0.08048127591609955,
      "learning_rate": 5.021660386861322e-06,
      "loss": 0.0618,
      "step": 52370
    },
    {
      "epoch": 4.994392395019531e-05,
      "model_forward_time": 2.2325475215911865,
      "step": 52370
    },
    {
      "dynamics_loss_avg": 0.06976596154272556,
      "epoch": 4.994392395019531e-05,
      "step": 52370
    },
    {
      "action_loss_avg": 0.0020844856509938834,
      "epoch": 4.994392395019531e-05,
      "step": 52370
    },
    {
      "epoch": 4.994392395019531e-05,
      "step": 52370,
      "training_step_time": 7.165253639221191
    },
    {
      "epoch": 4.994487762451172e-05,
      "model_forward_time": 1.7009851932525635,
      "step": 52371
    },
    {
      "epoch": 4.994487762451172e-05,
      "step": 52371,
      "training_step_time": 7.648087501525879
    },
    {
      "epoch": 4.9945831298828125e-05,
      "model_forward_time": 1.6900694370269775,
      "step": 52372
    },
    {
      "epoch": 4.9945831298828125e-05,
      "step": 52372,
      "training_step_time": 6.795170307159424
    },
    {
      "epoch": 4.994678497314453e-05,
      "model_forward_time": 1.6868908405303955,
      "step": 52373
    },
    {
      "epoch": 4.994678497314453e-05,
      "step": 52373,
      "training_step_time": 6.78535532951355
    },
    {
      "epoch": 4.994773864746094e-05,
      "model_forward_time": 1.688180923461914,
      "step": 52374
    },
    {
      "epoch": 4.994773864746094e-05,
      "step": 52374,
      "training_step_time": 6.823099613189697
    },
    {
      "epoch": 4.9948692321777344e-05,
      "model_forward_time": 1.6910977363586426,
      "step": 52375
    },
    {
      "epoch": 4.9948692321777344e-05,
      "step": 52375,
      "training_step_time": 6.8145060539245605
    },
    {
      "epoch": 4.994964599609375e-05,
      "model_forward_time": 1.6943068504333496,
      "step": 52376
    },
    {
      "epoch": 4.994964599609375e-05,
      "step": 52376,
      "training_step_time": 6.836957216262817
    },
    {
      "epoch": 4.995059967041016e-05,
      "model_forward_time": 1.6988544464111328,
      "step": 52377
    },
    {
      "epoch": 4.995059967041016e-05,
      "step": 52377,
      "training_step_time": 6.812313079833984
    },
    {
      "epoch": 4.995155334472656e-05,
      "model_forward_time": 1.7236788272857666,
      "step": 52378
    },
    {
      "epoch": 4.995155334472656e-05,
      "step": 52378,
      "training_step_time": 6.795308589935303
    },
    {
      "epoch": 4.995250701904297e-05,
      "model_forward_time": 2.3031506538391113,
      "step": 52379
    },
    {
      "epoch": 4.995250701904297e-05,
      "step": 52379,
      "training_step_time": 7.564464569091797
    },
    {
      "epoch": 4.995346069335938e-05,
      "grad_norm": 0.10769091546535492,
      "learning_rate": 5.020006931404726e-06,
      "loss": 0.0558,
      "step": 52380
    },
    {
      "epoch": 4.995346069335938e-05,
      "model_forward_time": 1.707610845565796,
      "step": 52380
    },
    {
      "dynamics_loss_avg": 0.04284850936383009,
      "epoch": 4.995346069335938e-05,
      "step": 52380
    },
    {
      "action_loss_avg": 0.001523455063579604,
      "epoch": 4.995346069335938e-05,
      "step": 52380
    },
    {
      "epoch": 4.995346069335938e-05,
      "step": 52380,
      "training_step_time": 7.696916341781616
    },
    {
      "epoch": 4.995441436767578e-05,
      "model_forward_time": 1.7065045833587646,
      "step": 52381
    },
    {
      "epoch": 4.995441436767578e-05,
      "step": 52381,
      "training_step_time": 7.372668266296387
    },
    {
      "epoch": 4.9955368041992185e-05,
      "model_forward_time": 1.7047510147094727,
      "step": 52382
    },
    {
      "epoch": 4.9955368041992185e-05,
      "step": 52382,
      "training_step_time": 6.716435194015503
    },
    {
      "epoch": 4.9956321716308596e-05,
      "model_forward_time": 1.7039568424224854,
      "step": 52383
    },
    {
      "epoch": 4.9956321716308596e-05,
      "step": 52383,
      "training_step_time": 6.794663667678833
    },
    {
      "epoch": 4.9957275390625e-05,
      "model_forward_time": 1.6915249824523926,
      "step": 52384
    },
    {
      "epoch": 4.9957275390625e-05,
      "step": 52384,
      "training_step_time": 6.673776865005493
    },
    {
      "epoch": 4.9958229064941404e-05,
      "model_forward_time": 1.7052390575408936,
      "step": 52385
    },
    {
      "epoch": 4.9958229064941404e-05,
      "step": 52385,
      "training_step_time": 6.6628499031066895
    },
    {
      "epoch": 4.9959182739257815e-05,
      "model_forward_time": 1.7122797966003418,
      "step": 52386
    },
    {
      "epoch": 4.9959182739257815e-05,
      "step": 52386,
      "training_step_time": 6.770928621292114
    },
    {
      "epoch": 4.996013641357422e-05,
      "model_forward_time": 1.7060081958770752,
      "step": 52387
    },
    {
      "epoch": 4.996013641357422e-05,
      "step": 52387,
      "training_step_time": 6.74425745010376
    },
    {
      "epoch": 4.996109008789062e-05,
      "model_forward_time": 2.3123908042907715,
      "step": 52388
    },
    {
      "epoch": 4.996109008789062e-05,
      "step": 52388,
      "training_step_time": 7.5114850997924805
    },
    {
      "epoch": 4.9962043762207034e-05,
      "model_forward_time": 1.703263759613037,
      "step": 52389
    },
    {
      "epoch": 4.9962043762207034e-05,
      "step": 52389,
      "training_step_time": 8.317840576171875
    },
    {
      "epoch": 4.996299743652344e-05,
      "grad_norm": 0.10886965692043304,
      "learning_rate": 5.018353473760201e-06,
      "loss": 0.0586,
      "step": 52390
    },
    {
      "epoch": 4.996299743652344e-05,
      "model_forward_time": 1.7085413932800293,
      "step": 52390
    },
    {
      "dynamics_loss_avg": 0.05326221249997616,
      "epoch": 4.996299743652344e-05,
      "step": 52390
    },
    {
      "action_loss_avg": 0.001692401577020064,
      "epoch": 4.996299743652344e-05,
      "step": 52390
    },
    {
      "epoch": 4.996299743652344e-05,
      "step": 52390,
      "training_step_time": 7.523853302001953
    },
    {
      "epoch": 4.996395111083984e-05,
      "model_forward_time": 1.7004659175872803,
      "step": 52391
    },
    {
      "epoch": 4.996395111083984e-05,
      "step": 52391,
      "training_step_time": 6.740546703338623
    },
    {
      "epoch": 4.996490478515625e-05,
      "model_forward_time": 1.6984951496124268,
      "step": 52392
    },
    {
      "epoch": 4.996490478515625e-05,
      "step": 52392,
      "training_step_time": 6.753621578216553
    },
    {
      "epoch": 4.996585845947266e-05,
      "model_forward_time": 1.7041001319885254,
      "step": 52393
    },
    {
      "epoch": 4.996585845947266e-05,
      "step": 52393,
      "training_step_time": 6.78914475440979
    },
    {
      "epoch": 4.996681213378906e-05,
      "model_forward_time": 1.7016472816467285,
      "step": 52394
    },
    {
      "epoch": 4.996681213378906e-05,
      "step": 52394,
      "training_step_time": 6.717336893081665
    },
    {
      "epoch": 4.996776580810547e-05,
      "model_forward_time": 1.7088046073913574,
      "step": 52395
    },
    {
      "epoch": 4.996776580810547e-05,
      "step": 52395,
      "training_step_time": 6.718567609786987
    },
    {
      "epoch": 4.9968719482421876e-05,
      "model_forward_time": 1.7033684253692627,
      "step": 52396
    },
    {
      "epoch": 4.9968719482421876e-05,
      "step": 52396,
      "training_step_time": 6.788311004638672
    },
    {
      "epoch": 4.996967315673828e-05,
      "model_forward_time": 2.2990822792053223,
      "step": 52397
    },
    {
      "epoch": 4.996967315673828e-05,
      "step": 52397,
      "training_step_time": 7.490047454833984
    },
    {
      "epoch": 4.997062683105469e-05,
      "model_forward_time": 1.7034647464752197,
      "step": 52398
    },
    {
      "epoch": 4.997062683105469e-05,
      "step": 52398,
      "training_step_time": 9.833936214447021
    },
    {
      "epoch": 4.9971580505371095e-05,
      "model_forward_time": 1.6934943199157715,
      "step": 52399
    },
    {
      "epoch": 4.9971580505371095e-05,
      "step": 52399,
      "training_step_time": 7.54820990562439
    },
    {
      "epoch": 4.99725341796875e-05,
      "grad_norm": 0.10213207453489304,
      "learning_rate": 5.0167000141085685e-06,
      "loss": 0.0575,
      "step": 52400
    },
    {
      "epoch": 4.99725341796875e-05,
      "model_forward_time": 1.7073183059692383,
      "step": 52400
    },
    {
      "dynamics_loss_avg": 0.04779152665287256,
      "epoch": 4.99725341796875e-05,
      "step": 52400
    },
    {
      "action_loss_avg": 0.0012817691371310503,
      "epoch": 4.99725341796875e-05,
      "step": 52400
    },
    {
      "epoch": 4.99725341796875e-05,
      "step": 52400,
      "training_step_time": 6.734732151031494
    },
    {
      "epoch": 4.997348785400391e-05,
      "model_forward_time": 1.7147212028503418,
      "step": 52401
    },
    {
      "epoch": 4.997348785400391e-05,
      "step": 52401,
      "training_step_time": 6.736981630325317
    },
    {
      "epoch": 4.9974441528320313e-05,
      "model_forward_time": 1.7109153270721436,
      "step": 52402
    },
    {
      "epoch": 4.9974441528320313e-05,
      "step": 52402,
      "training_step_time": 6.7517616748809814
    },
    {
      "epoch": 4.997539520263672e-05,
      "model_forward_time": 1.7129464149475098,
      "step": 52403
    },
    {
      "epoch": 4.997539520263672e-05,
      "step": 52403,
      "training_step_time": 6.756151437759399
    },
    {
      "epoch": 4.997634887695313e-05,
      "model_forward_time": 1.7009532451629639,
      "step": 52404
    },
    {
      "epoch": 4.997634887695313e-05,
      "step": 52404,
      "training_step_time": 6.67888879776001
    },
    {
      "epoch": 4.997730255126953e-05,
      "model_forward_time": 1.7077970504760742,
      "step": 52405
    },
    {
      "epoch": 4.997730255126953e-05,
      "step": 52405,
      "training_step_time": 6.71362829208374
    },
    {
      "epoch": 4.9978256225585936e-05,
      "model_forward_time": 2.249969959259033,
      "step": 52406
    },
    {
      "epoch": 4.9978256225585936e-05,
      "step": 52406,
      "training_step_time": 8.460072040557861
    },
    {
      "epoch": 4.997920989990234e-05,
      "model_forward_time": 1.6972100734710693,
      "step": 52407
    },
    {
      "epoch": 4.997920989990234e-05,
      "step": 52407,
      "training_step_time": 8.618692874908447
    },
    {
      "epoch": 4.998016357421875e-05,
      "model_forward_time": 1.6908907890319824,
      "step": 52408
    },
    {
      "epoch": 4.998016357421875e-05,
      "step": 52408,
      "training_step_time": 8.255313634872437
    },
    {
      "epoch": 4.9981117248535155e-05,
      "model_forward_time": 1.6930325031280518,
      "step": 52409
    },
    {
      "epoch": 4.9981117248535155e-05,
      "step": 52409,
      "training_step_time": 7.264664173126221
    },
    {
      "epoch": 4.998207092285156e-05,
      "grad_norm": 0.0996638759970665,
      "learning_rate": 5.015046552630647e-06,
      "loss": 0.0553,
      "step": 52410
    },
    {
      "epoch": 4.998207092285156e-05,
      "model_forward_time": 1.6907415390014648,
      "step": 52410
    },
    {
      "dynamics_loss_avg": 0.05039182286709547,
      "epoch": 4.998207092285156e-05,
      "step": 52410
    },
    {
      "action_loss_avg": 0.001264500255638268,
      "epoch": 4.998207092285156e-05,
      "step": 52410
    },
    {
      "epoch": 4.998207092285156e-05,
      "step": 52410,
      "training_step_time": 6.713855028152466
    },
    {
      "epoch": 4.998302459716797e-05,
      "model_forward_time": 1.6898150444030762,
      "step": 52411
    },
    {
      "epoch": 4.998302459716797e-05,
      "step": 52411,
      "training_step_time": 6.768933296203613
    },
    {
      "epoch": 4.9983978271484374e-05,
      "model_forward_time": 1.691483497619629,
      "step": 52412
    },
    {
      "epoch": 4.9983978271484374e-05,
      "step": 52412,
      "training_step_time": 6.732708692550659
    },
    {
      "epoch": 4.998493194580078e-05,
      "model_forward_time": 1.6922414302825928,
      "step": 52413
    },
    {
      "epoch": 4.998493194580078e-05,
      "step": 52413,
      "training_step_time": 6.734145879745483
    },
    {
      "epoch": 4.998588562011719e-05,
      "model_forward_time": 1.6872451305389404,
      "step": 52414
    },
    {
      "epoch": 4.998588562011719e-05,
      "step": 52414,
      "training_step_time": 6.7741477489471436
    },
    {
      "epoch": 4.998683929443359e-05,
      "model_forward_time": 1.7020354270935059,
      "step": 52415
    },
    {
      "epoch": 4.998683929443359e-05,
      "step": 52415,
      "training_step_time": 7.191821336746216
    },
    {
      "epoch": 4.998779296875e-05,
      "model_forward_time": 2.2565672397613525,
      "step": 52416
    },
    {
      "epoch": 4.998779296875e-05,
      "step": 52416,
      "training_step_time": 8.15339469909668
    },
    {
      "epoch": 4.998874664306641e-05,
      "model_forward_time": 1.6910791397094727,
      "step": 52417
    },
    {
      "epoch": 4.998874664306641e-05,
      "step": 52417,
      "training_step_time": 7.726269006729126
    },
    {
      "epoch": 4.998970031738281e-05,
      "model_forward_time": 1.6888904571533203,
      "step": 52418
    },
    {
      "epoch": 4.998970031738281e-05,
      "step": 52418,
      "training_step_time": 7.5716471672058105
    },
    {
      "epoch": 4.9990653991699216e-05,
      "model_forward_time": 1.6898248195648193,
      "step": 52419
    },
    {
      "epoch": 4.9990653991699216e-05,
      "step": 52419,
      "training_step_time": 6.796913146972656
    },
    {
      "epoch": 4.999160766601563e-05,
      "grad_norm": 0.13004720211029053,
      "learning_rate": 5.013393089507258e-06,
      "loss": 0.053,
      "step": 52420
    },
    {
      "epoch": 4.999160766601563e-05,
      "model_forward_time": 1.688483715057373,
      "step": 52420
    },
    {
      "dynamics_loss_avg": 0.06318702045828103,
      "epoch": 4.999160766601563e-05,
      "step": 52420
    },
    {
      "action_loss_avg": 0.0016045112861320376,
      "epoch": 4.999160766601563e-05,
      "step": 52420
    },
    {
      "epoch": 4.999160766601563e-05,
      "step": 52420,
      "training_step_time": 6.764591217041016
    },
    {
      "epoch": 4.999256134033203e-05,
      "model_forward_time": 1.6976375579833984,
      "step": 52421
    },
    {
      "epoch": 4.999256134033203e-05,
      "step": 52421,
      "training_step_time": 6.730071544647217
    },
    {
      "epoch": 4.9993515014648435e-05,
      "model_forward_time": 1.6984803676605225,
      "step": 52422
    },
    {
      "epoch": 4.9993515014648435e-05,
      "step": 52422,
      "training_step_time": 6.7584381103515625
    },
    {
      "epoch": 4.9994468688964845e-05,
      "model_forward_time": 1.7014644145965576,
      "step": 52423
    },
    {
      "epoch": 4.9994468688964845e-05,
      "step": 52423,
      "training_step_time": 6.736242771148682
    },
    {
      "epoch": 4.999542236328125e-05,
      "model_forward_time": 1.716740369796753,
      "step": 52424
    },
    {
      "epoch": 4.999542236328125e-05,
      "step": 52424,
      "training_step_time": 7.171070337295532
    },
    {
      "epoch": 4.9996376037597654e-05,
      "model_forward_time": 2.327834367752075,
      "step": 52425
    },
    {
      "epoch": 4.9996376037597654e-05,
      "step": 52425,
      "training_step_time": 8.643499374389648
    },
    {
      "epoch": 4.9997329711914064e-05,
      "model_forward_time": 1.7185561656951904,
      "step": 52426
    },
    {
      "epoch": 4.9997329711914064e-05,
      "step": 52426,
      "training_step_time": 11.6987144947052
    },
    {
      "epoch": 4.999828338623047e-05,
      "model_forward_time": 1.716052770614624,
      "step": 52427
    },
    {
      "epoch": 4.999828338623047e-05,
      "step": 52427,
      "training_step_time": 7.559150457382202
    },
    {
      "epoch": 4.999923706054687e-05,
      "model_forward_time": 1.724579095840454,
      "step": 52428
    },
    {
      "epoch": 4.999923706054687e-05,
      "step": 52428,
      "training_step_time": 6.748505115509033
    },
    {
      "epoch": 5.000019073486328e-05,
      "model_forward_time": 1.7003576755523682,
      "step": 52429
    },
    {
      "epoch": 5.000019073486328e-05,
      "step": 52429,
      "training_step_time": 6.7499001026153564
    },
    {
      "epoch": 5.000114440917969e-05,
      "grad_norm": 0.09873046725988388,
      "learning_rate": 5.01173962491922e-06,
      "loss": 0.0538,
      "step": 52430
    },
    {
      "epoch": 5.000114440917969e-05,
      "model_forward_time": 1.71773362159729,
      "step": 52430
    },
    {
      "dynamics_loss_avg": 0.05631773062050342,
      "epoch": 5.000114440917969e-05,
      "step": 52430
    },
    {
      "action_loss_avg": 0.001688768420717679,
      "epoch": 5.000114440917969e-05,
      "step": 52430
    },
    {
      "epoch": 5.000114440917969e-05,
      "step": 52430,
      "training_step_time": 6.72691535949707
    },
    {
      "epoch": 5.000209808349609e-05,
      "model_forward_time": 1.7196431159973145,
      "step": 52431
    },
    {
      "epoch": 5.000209808349609e-05,
      "step": 52431,
      "training_step_time": 6.733769416809082
    },
    {
      "epoch": 5.00030517578125e-05,
      "model_forward_time": 1.7131288051605225,
      "step": 52432
    },
    {
      "epoch": 5.00030517578125e-05,
      "step": 52432,
      "training_step_time": 6.7437052726745605
    },
    {
      "epoch": 5.0004005432128906e-05,
      "model_forward_time": 1.7186956405639648,
      "step": 52433
    },
    {
      "epoch": 5.0004005432128906e-05,
      "step": 52433,
      "training_step_time": 6.700430154800415
    },
    {
      "epoch": 5.000495910644531e-05,
      "model_forward_time": 2.343994140625,
      "step": 52434
    },
    {
      "epoch": 5.000495910644531e-05,
      "step": 52434,
      "training_step_time": 7.5544116497039795
    },
    {
      "epoch": 5.000591278076172e-05,
      "model_forward_time": 1.763782262802124,
      "step": 52435
    },
    {
      "epoch": 5.000591278076172e-05,
      "step": 52435,
      "training_step_time": 11.75656795501709
    },
    {
      "epoch": 5.0006866455078125e-05,
      "model_forward_time": 1.7554104328155518,
      "step": 52436
    },
    {
      "epoch": 5.0006866455078125e-05,
      "step": 52436,
      "training_step_time": 8.149038553237915
    },
    {
      "epoch": 5.000782012939453e-05,
      "model_forward_time": 1.7485229969024658,
      "step": 52437
    },
    {
      "epoch": 5.000782012939453e-05,
      "step": 52437,
      "training_step_time": 7.284358263015747
    },
    {
      "epoch": 5.000877380371094e-05,
      "model_forward_time": 1.746302843093872,
      "step": 52438
    },
    {
      "epoch": 5.000877380371094e-05,
      "step": 52438,
      "training_step_time": 6.735607862472534
    },
    {
      "epoch": 5.0009727478027344e-05,
      "model_forward_time": 1.7484104633331299,
      "step": 52439
    },
    {
      "epoch": 5.0009727478027344e-05,
      "step": 52439,
      "training_step_time": 6.800429582595825
    },
    {
      "epoch": 5.001068115234375e-05,
      "grad_norm": 0.06661082059144974,
      "learning_rate": 5.010086159047355e-06,
      "loss": 0.0579,
      "step": 52440
    },
    {
      "epoch": 5.001068115234375e-05,
      "model_forward_time": 1.7556700706481934,
      "step": 52440
    },
    {
      "dynamics_loss_avg": 0.05865422785282135,
      "epoch": 5.001068115234375e-05,
      "step": 52440
    },
    {
      "action_loss_avg": 0.0018460479739587754,
      "epoch": 5.001068115234375e-05,
      "step": 52440
    },
    {
      "epoch": 5.001068115234375e-05,
      "step": 52440,
      "training_step_time": 6.778766393661499
    },
    {
      "epoch": 5.001163482666016e-05,
      "model_forward_time": 1.7528784275054932,
      "step": 52441
    },
    {
      "epoch": 5.001163482666016e-05,
      "step": 52441,
      "training_step_time": 6.769593715667725
    },
    {
      "epoch": 5.001258850097656e-05,
      "model_forward_time": 1.7440028190612793,
      "step": 52442
    },
    {
      "epoch": 5.001258850097656e-05,
      "step": 52442,
      "training_step_time": 6.79833197593689
    },
    {
      "epoch": 5.001354217529297e-05,
      "model_forward_time": 2.867307424545288,
      "step": 52443
    },
    {
      "epoch": 5.001354217529297e-05,
      "step": 52443,
      "training_step_time": 7.671244144439697
    },
    {
      "epoch": 5.001449584960938e-05,
      "model_forward_time": 1.7841875553131104,
      "step": 52444
    },
    {
      "epoch": 5.001449584960938e-05,
      "step": 52444,
      "training_step_time": 9.32793378829956
    },
    {
      "epoch": 5.001544952392578e-05,
      "model_forward_time": 1.7663109302520752,
      "step": 52445
    },
    {
      "epoch": 5.001544952392578e-05,
      "step": 52445,
      "training_step_time": 8.416805744171143
    },
    {
      "epoch": 5.0016403198242186e-05,
      "model_forward_time": 1.768977165222168,
      "step": 52446
    },
    {
      "epoch": 5.0016403198242186e-05,
      "step": 52446,
      "training_step_time": 7.7191479206085205
    },
    {
      "epoch": 5.0017356872558596e-05,
      "model_forward_time": 1.763357400894165,
      "step": 52447
    },
    {
      "epoch": 5.0017356872558596e-05,
      "step": 52447,
      "training_step_time": 6.811150074005127
    },
    {
      "epoch": 5.0018310546875e-05,
      "model_forward_time": 1.7625494003295898,
      "step": 52448
    },
    {
      "epoch": 5.0018310546875e-05,
      "step": 52448,
      "training_step_time": 6.83602499961853
    },
    {
      "epoch": 5.0019264221191404e-05,
      "model_forward_time": 1.7634291648864746,
      "step": 52449
    },
    {
      "epoch": 5.0019264221191404e-05,
      "step": 52449,
      "training_step_time": 6.765032768249512
    },
    {
      "epoch": 5.0020217895507815e-05,
      "grad_norm": 0.11097452044487,
      "learning_rate": 5.008432692072482e-06,
      "loss": 0.0559,
      "step": 52450
    },
    {
      "epoch": 5.0020217895507815e-05,
      "model_forward_time": 1.7711687088012695,
      "step": 52450
    },
    {
      "dynamics_loss_avg": 0.05555263813585043,
      "epoch": 5.0020217895507815e-05,
      "step": 52450
    },
    {
      "action_loss_avg": 0.0015098580537596718,
      "epoch": 5.0020217895507815e-05,
      "step": 52450
    },
    {
      "epoch": 5.0020217895507815e-05,
      "step": 52450,
      "training_step_time": 6.754123210906982
    },
    {
      "epoch": 5.002117156982422e-05,
      "model_forward_time": 1.7619287967681885,
      "step": 52451
    },
    {
      "epoch": 5.002117156982422e-05,
      "step": 52451,
      "training_step_time": 6.739832401275635
    },
    {
      "epoch": 5.002212524414062e-05,
      "model_forward_time": 2.8823652267456055,
      "step": 52452
    },
    {
      "epoch": 5.002212524414062e-05,
      "step": 52452,
      "training_step_time": 7.658253192901611
    },
    {
      "epoch": 5.0023078918457034e-05,
      "model_forward_time": 1.8268511295318604,
      "step": 52453
    },
    {
      "epoch": 5.0023078918457034e-05,
      "step": 52453,
      "training_step_time": 10.29283356666565
    },
    {
      "epoch": 5.002403259277344e-05,
      "model_forward_time": 1.7663705348968506,
      "step": 52454
    },
    {
      "epoch": 5.002403259277344e-05,
      "step": 52454,
      "training_step_time": 8.678348541259766
    },
    {
      "epoch": 5.002498626708984e-05,
      "model_forward_time": 1.7598850727081299,
      "step": 52455
    },
    {
      "epoch": 5.002498626708984e-05,
      "step": 52455,
      "training_step_time": 7.4694132804870605
    },
    {
      "epoch": 5.002593994140625e-05,
      "model_forward_time": 1.75909423828125,
      "step": 52456
    },
    {
      "epoch": 5.002593994140625e-05,
      "step": 52456,
      "training_step_time": 6.785176038742065
    },
    {
      "epoch": 5.002689361572266e-05,
      "model_forward_time": 1.7597801685333252,
      "step": 52457
    },
    {
      "epoch": 5.002689361572266e-05,
      "step": 52457,
      "training_step_time": 6.783364772796631
    },
    {
      "epoch": 5.002784729003906e-05,
      "model_forward_time": 1.7605478763580322,
      "step": 52458
    },
    {
      "epoch": 5.002784729003906e-05,
      "step": 52458,
      "training_step_time": 6.763236999511719
    },
    {
      "epoch": 5.002880096435547e-05,
      "model_forward_time": 1.7551004886627197,
      "step": 52459
    },
    {
      "epoch": 5.002880096435547e-05,
      "step": 52459,
      "training_step_time": 6.806216239929199
    },
    {
      "epoch": 5.0029754638671876e-05,
      "grad_norm": 0.1023336872458458,
      "learning_rate": 5.006779224175423e-06,
      "loss": 0.0579,
      "step": 52460
    },
    {
      "epoch": 5.0029754638671876e-05,
      "model_forward_time": 1.7805507183074951,
      "step": 52460
    },
    {
      "dynamics_loss_avg": 0.05390158593654633,
      "epoch": 5.0029754638671876e-05,
      "step": 52460
    },
    {
      "action_loss_avg": 0.0013641027733683586,
      "epoch": 5.0029754638671876e-05,
      "step": 52460
    },
    {
      "epoch": 5.0029754638671876e-05,
      "step": 52460,
      "training_step_time": 6.785133361816406
    },
    {
      "epoch": 5.003070831298828e-05,
      "model_forward_time": 1.7911067008972168,
      "step": 52461
    },
    {
      "epoch": 5.003070831298828e-05,
      "step": 52461,
      "training_step_time": 9.21893048286438
    },
    {
      "epoch": 5.003166198730469e-05,
      "model_forward_time": 1.832794427871704,
      "step": 52462
    },
    {
      "epoch": 5.003166198730469e-05,
      "step": 52462,
      "training_step_time": 8.154995679855347
    },
    {
      "epoch": 5.0032615661621095e-05,
      "model_forward_time": 1.7899882793426514,
      "step": 52463
    },
    {
      "epoch": 5.0032615661621095e-05,
      "step": 52463,
      "training_step_time": 9.095451354980469
    },
    {
      "epoch": 5.00335693359375e-05,
      "model_forward_time": 1.7587158679962158,
      "step": 52464
    },
    {
      "epoch": 5.00335693359375e-05,
      "step": 52464,
      "training_step_time": 7.502247333526611
    },
    {
      "epoch": 5.003452301025391e-05,
      "model_forward_time": 1.7535769939422607,
      "step": 52465
    },
    {
      "epoch": 5.003452301025391e-05,
      "step": 52465,
      "training_step_time": 6.756723403930664
    },
    {
      "epoch": 5.0035476684570314e-05,
      "model_forward_time": 1.7646443843841553,
      "step": 52466
    },
    {
      "epoch": 5.0035476684570314e-05,
      "step": 52466,
      "training_step_time": 6.690225601196289
    },
    {
      "epoch": 5.003643035888672e-05,
      "model_forward_time": 1.7663054466247559,
      "step": 52467
    },
    {
      "epoch": 5.003643035888672e-05,
      "step": 52467,
      "training_step_time": 6.72957706451416
    },
    {
      "epoch": 5.003738403320313e-05,
      "model_forward_time": 1.7572317123413086,
      "step": 52468
    },
    {
      "epoch": 5.003738403320313e-05,
      "step": 52468,
      "training_step_time": 6.771583318710327
    },
    {
      "epoch": 5.003833770751953e-05,
      "model_forward_time": 1.765941858291626,
      "step": 52469
    },
    {
      "epoch": 5.003833770751953e-05,
      "step": 52469,
      "training_step_time": 6.7368738651275635
    },
    {
      "epoch": 5.0039291381835936e-05,
      "grad_norm": 0.12394394725561142,
      "learning_rate": 5.005125755536998e-06,
      "loss": 0.0581,
      "step": 52470
    },
    {
      "epoch": 5.0039291381835936e-05,
      "model_forward_time": 1.7725698947906494,
      "step": 52470
    },
    {
      "dynamics_loss_avg": 0.05361623615026474,
      "epoch": 5.0039291381835936e-05,
      "step": 52470
    },
    {
      "action_loss_avg": 0.0011817127699032426,
      "epoch": 5.0039291381835936e-05,
      "step": 52470
    },
    {
      "epoch": 5.0039291381835936e-05,
      "step": 52470,
      "training_step_time": 7.521266222000122
    },
    {
      "epoch": 5.004024505615234e-05,
      "model_forward_time": 2.9212727546691895,
      "step": 52471
    },
    {
      "epoch": 5.004024505615234e-05,
      "step": 52471,
      "training_step_time": 8.664608240127563
    },
    {
      "epoch": 5.004119873046875e-05,
      "model_forward_time": 1.7568786144256592,
      "step": 52472
    },
    {
      "epoch": 5.004119873046875e-05,
      "step": 52472,
      "training_step_time": 10.380025148391724
    },
    {
      "epoch": 5.0042152404785155e-05,
      "model_forward_time": 1.7540004253387451,
      "step": 52473
    },
    {
      "epoch": 5.0042152404785155e-05,
      "step": 52473,
      "training_step_time": 8.431987762451172
    },
    {
      "epoch": 5.004310607910156e-05,
      "model_forward_time": 1.7587511539459229,
      "step": 52474
    },
    {
      "epoch": 5.004310607910156e-05,
      "step": 52474,
      "training_step_time": 7.0204808712005615
    },
    {
      "epoch": 5.004405975341797e-05,
      "model_forward_time": 1.7706267833709717,
      "step": 52475
    },
    {
      "epoch": 5.004405975341797e-05,
      "step": 52475,
      "training_step_time": 6.732126712799072
    },
    {
      "epoch": 5.0045013427734374e-05,
      "model_forward_time": 1.7695403099060059,
      "step": 52476
    },
    {
      "epoch": 5.0045013427734374e-05,
      "step": 52476,
      "training_step_time": 6.755503177642822
    },
    {
      "epoch": 5.004596710205078e-05,
      "model_forward_time": 1.7511043548583984,
      "step": 52477
    },
    {
      "epoch": 5.004596710205078e-05,
      "step": 52477,
      "training_step_time": 6.756092548370361
    },
    {
      "epoch": 5.004692077636719e-05,
      "model_forward_time": 1.7683212757110596,
      "step": 52478
    },
    {
      "epoch": 5.004692077636719e-05,
      "step": 52478,
      "training_step_time": 6.745071172714233
    },
    {
      "epoch": 5.004787445068359e-05,
      "model_forward_time": 1.761350154876709,
      "step": 52479
    },
    {
      "epoch": 5.004787445068359e-05,
      "step": 52479,
      "training_step_time": 7.55253791809082
    },
    {
      "epoch": 5.0048828125e-05,
      "grad_norm": 0.10625236481428146,
      "learning_rate": 5.003472286338029e-06,
      "loss": 0.0584,
      "step": 52480
    },
    {
      "epoch": 5.0048828125e-05,
      "model_forward_time": 2.950836420059204,
      "step": 52480
    },
    {
      "dynamics_loss_avg": 0.047528442554175855,
      "epoch": 5.0048828125e-05,
      "step": 52480
    },
    {
      "action_loss_avg": 0.0013926709652878344,
      "epoch": 5.0048828125e-05,
      "step": 52480
    },
    {
      "epoch": 5.0048828125e-05,
      "step": 52480,
      "training_step_time": 9.962400674819946
    },
    {
      "epoch": 5.004978179931641e-05,
      "model_forward_time": 1.767561435699463,
      "step": 52481
    },
    {
      "epoch": 5.004978179931641e-05,
      "step": 52481,
      "training_step_time": 8.660088539123535
    },
    {
      "epoch": 5.005073547363281e-05,
      "model_forward_time": 1.7643272876739502,
      "step": 52482
    },
    {
      "epoch": 5.005073547363281e-05,
      "step": 52482,
      "training_step_time": 9.528198957443237
    },
    {
      "epoch": 5.0051689147949216e-05,
      "model_forward_time": 1.7588183879852295,
      "step": 52483
    },
    {
      "epoch": 5.0051689147949216e-05,
      "step": 52483,
      "training_step_time": 7.392491340637207
    },
    {
      "epoch": 5.005264282226563e-05,
      "model_forward_time": 1.762589454650879,
      "step": 52484
    },
    {
      "epoch": 5.005264282226563e-05,
      "step": 52484,
      "training_step_time": 6.7374067306518555
    },
    {
      "epoch": 5.005359649658203e-05,
      "model_forward_time": 1.7613604068756104,
      "step": 52485
    },
    {
      "epoch": 5.005359649658203e-05,
      "step": 52485,
      "training_step_time": 6.780961751937866
    },
    {
      "epoch": 5.0054550170898435e-05,
      "model_forward_time": 1.7629814147949219,
      "step": 52486
    },
    {
      "epoch": 5.0054550170898435e-05,
      "step": 52486,
      "training_step_time": 6.759684801101685
    },
    {
      "epoch": 5.0055503845214846e-05,
      "model_forward_time": 1.7648985385894775,
      "step": 52487
    },
    {
      "epoch": 5.0055503845214846e-05,
      "step": 52487,
      "training_step_time": 6.744218826293945
    },
    {
      "epoch": 5.005645751953125e-05,
      "model_forward_time": 1.7628130912780762,
      "step": 52488
    },
    {
      "epoch": 5.005645751953125e-05,
      "step": 52488,
      "training_step_time": 8.301069498062134
    },
    {
      "epoch": 5.0057411193847654e-05,
      "model_forward_time": 2.9882967472076416,
      "step": 52489
    },
    {
      "epoch": 5.0057411193847654e-05,
      "step": 52489,
      "training_step_time": 8.79521918296814
    },
    {
      "epoch": 5.0058364868164064e-05,
      "grad_norm": 0.10409778356552124,
      "learning_rate": 5.001818816759335e-06,
      "loss": 0.0581,
      "step": 52490
    },
    {
      "epoch": 5.0058364868164064e-05,
      "model_forward_time": 1.7491645812988281,
      "step": 52490
    },
    {
      "dynamics_loss_avg": 0.057182779908180235,
      "epoch": 5.0058364868164064e-05,
      "step": 52490
    },
    {
      "action_loss_avg": 0.0012290547485463322,
      "epoch": 5.0058364868164064e-05,
      "step": 52490
    },
    {
      "epoch": 5.0058364868164064e-05,
      "step": 52490,
      "training_step_time": 8.31035566329956
    },
    {
      "epoch": 5.005931854248047e-05,
      "model_forward_time": 1.7449793815612793,
      "step": 52491
    },
    {
      "epoch": 5.005931854248047e-05,
      "step": 52491,
      "training_step_time": 9.834680557250977
    },
    {
      "epoch": 5.006027221679687e-05,
      "model_forward_time": 1.7364082336425781,
      "step": 52492
    },
    {
      "epoch": 5.006027221679687e-05,
      "step": 52492,
      "training_step_time": 7.342359304428101
    },
    {
      "epoch": 5.006122589111328e-05,
      "model_forward_time": 1.7507667541503906,
      "step": 52493
    },
    {
      "epoch": 5.006122589111328e-05,
      "step": 52493,
      "training_step_time": 6.761449575424194
    },
    {
      "epoch": 5.006217956542969e-05,
      "model_forward_time": 1.7498669624328613,
      "step": 52494
    },
    {
      "epoch": 5.006217956542969e-05,
      "step": 52494,
      "training_step_time": 6.767281770706177
    },
    {
      "epoch": 5.006313323974609e-05,
      "model_forward_time": 1.7447972297668457,
      "step": 52495
    },
    {
      "epoch": 5.006313323974609e-05,
      "step": 52495,
      "training_step_time": 6.714966058731079
    },
    {
      "epoch": 5.00640869140625e-05,
      "model_forward_time": 1.8328132629394531,
      "step": 52496
    },
    {
      "epoch": 5.00640869140625e-05,
      "step": 52496,
      "training_step_time": 6.795267581939697
    },
    {
      "epoch": 5.0065040588378906e-05,
      "model_forward_time": 1.7358770370483398,
      "step": 52497
    },
    {
      "epoch": 5.0065040588378906e-05,
      "step": 52497,
      "training_step_time": 7.535463333129883
    },
    {
      "epoch": 5.006599426269531e-05,
      "model_forward_time": 2.3113536834716797,
      "step": 52498
    },
    {
      "epoch": 5.006599426269531e-05,
      "step": 52498,
      "training_step_time": 9.6673104763031
    },
    {
      "epoch": 5.006694793701172e-05,
      "model_forward_time": 1.734710693359375,
      "step": 52499
    },
    {
      "epoch": 5.006694793701172e-05,
      "step": 52499,
      "training_step_time": 8.409907341003418
    },
    {
      "epoch": 5.0067901611328125e-05,
      "grad_norm": 0.10998297482728958,
      "learning_rate": 5.000165346981738e-06,
      "loss": 0.0586,
      "step": 52500
    },
    {
      "epoch": 5.0067901611328125e-05,
      "model_forward_time": 1.716930627822876,
      "step": 52500
    },
    {
      "dynamics_loss_avg": 0.05060949623584747,
      "epoch": 5.0067901611328125e-05,
      "step": 52500
    },
    {
      "action_loss_avg": 0.001281427351932507,
      "epoch": 5.0067901611328125e-05,
      "step": 52500
    },
    {
      "epoch": 5.0067901611328125e-05,
      "step": 52500,
      "training_step_time": 9.259964942932129
    },
    {
      "epoch": 5.006885528564453e-05,
      "model_forward_time": 1.7370426654815674,
      "step": 52501
    },
    {
      "epoch": 5.006885528564453e-05,
      "step": 52501,
      "training_step_time": 7.670509576797485
    },
    {
      "epoch": 5.006980895996094e-05,
      "model_forward_time": 1.7330384254455566,
      "step": 52502
    },
    {
      "epoch": 5.006980895996094e-05,
      "step": 52502,
      "training_step_time": 7.392521858215332
    },
    {
      "epoch": 5.0070762634277344e-05,
      "model_forward_time": 1.350315809249878,
      "step": 52503
    },
    {
      "epoch": 5.0070762634277344e-05,
      "step": 52503,
      "training_step_time": 6.665294885635376
    },
    {
      "epoch": 5.007171630859375e-05,
      "model_forward_time": 1.7341277599334717,
      "step": 52504
    },
    {
      "epoch": 5.007171630859375e-05,
      "step": 52504,
      "training_step_time": 6.8763298988342285
    },
    {
      "epoch": 5.007266998291016e-05,
      "model_forward_time": 1.743279218673706,
      "step": 52505
    },
    {
      "epoch": 5.007266998291016e-05,
      "step": 52505,
      "training_step_time": 6.808590412139893
    },
    {
      "epoch": 5.007362365722656e-05,
      "model_forward_time": 1.7541449069976807,
      "step": 52506
    },
    {
      "epoch": 5.007362365722656e-05,
      "step": 52506,
      "training_step_time": 7.680486679077148
    },
    {
      "epoch": 5.007457733154297e-05,
      "model_forward_time": 2.314924716949463,
      "step": 52507
    },
    {
      "epoch": 5.007457733154297e-05,
      "step": 52507,
      "training_step_time": 8.956225156784058
    },
    {
      "epoch": 5.007553100585938e-05,
      "model_forward_time": 1.738990068435669,
      "step": 52508
    },
    {
      "epoch": 5.007553100585938e-05,
      "step": 52508,
      "training_step_time": 9.657191753387451
    },
    {
      "epoch": 5.007648468017578e-05,
      "model_forward_time": 1.7402327060699463,
      "step": 52509
    },
    {
      "epoch": 5.007648468017578e-05,
      "step": 52509,
      "training_step_time": 9.328193187713623
    },
    {
      "epoch": 5.0077438354492186e-05,
      "grad_norm": 0.1112099140882492,
      "learning_rate": 4.9985118771860595e-06,
      "loss": 0.0587,
      "step": 52510
    },
    {
      "epoch": 5.0077438354492186e-05,
      "model_forward_time": 1.739164113998413,
      "step": 52510
    },
    {
      "dynamics_loss_avg": 0.05194799359887838,
      "epoch": 5.0077438354492186e-05,
      "step": 52510
    },
    {
      "action_loss_avg": 0.0010994526499416679,
      "epoch": 5.0077438354492186e-05,
      "step": 52510
    },
    {
      "epoch": 5.0077438354492186e-05,
      "step": 52510,
      "training_step_time": 9.141971826553345
    },
    {
      "epoch": 5.0078392028808596e-05,
      "model_forward_time": 1.738983392715454,
      "step": 52511
    },
    {
      "epoch": 5.0078392028808596e-05,
      "step": 52511,
      "training_step_time": 7.3836143016815186
    },
    {
      "epoch": 5.0079345703125e-05,
      "model_forward_time": 1.742767572402954,
      "step": 52512
    },
    {
      "epoch": 5.0079345703125e-05,
      "step": 52512,
      "training_step_time": 6.763232707977295
    },
    {
      "epoch": 5.0080299377441404e-05,
      "model_forward_time": 1.7428314685821533,
      "step": 52513
    },
    {
      "epoch": 5.0080299377441404e-05,
      "step": 52513,
      "training_step_time": 6.77657413482666
    },
    {
      "epoch": 5.0081253051757815e-05,
      "model_forward_time": 1.7390332221984863,
      "step": 52514
    },
    {
      "epoch": 5.0081253051757815e-05,
      "step": 52514,
      "training_step_time": 6.821770906448364
    },
    {
      "epoch": 5.008220672607422e-05,
      "model_forward_time": 1.7354540824890137,
      "step": 52515
    },
    {
      "epoch": 5.008220672607422e-05,
      "step": 52515,
      "training_step_time": 7.533217668533325
    },
    {
      "epoch": 5.008316040039062e-05,
      "model_forward_time": 2.3275575637817383,
      "step": 52516
    },
    {
      "epoch": 5.008316040039062e-05,
      "step": 52516,
      "training_step_time": 7.543132781982422
    },
    {
      "epoch": 5.0084114074707034e-05,
      "model_forward_time": 1.748741865158081,
      "step": 52517
    },
    {
      "epoch": 5.0084114074707034e-05,
      "step": 52517,
      "training_step_time": 9.688185930252075
    },
    {
      "epoch": 5.008506774902344e-05,
      "model_forward_time": 1.7286412715911865,
      "step": 52518
    },
    {
      "epoch": 5.008506774902344e-05,
      "step": 52518,
      "training_step_time": 9.105093240737915
    },
    {
      "epoch": 5.008602142333984e-05,
      "model_forward_time": 1.7286741733551025,
      "step": 52519
    },
    {
      "epoch": 5.008602142333984e-05,
      "step": 52519,
      "training_step_time": 7.802810430526733
    },
    {
      "epoch": 5.008697509765625e-05,
      "grad_norm": 0.10395222157239914,
      "learning_rate": 4.996858407553119e-06,
      "loss": 0.0632,
      "step": 52520
    },
    {
      "epoch": 5.008697509765625e-05,
      "model_forward_time": 1.7300431728363037,
      "step": 52520
    },
    {
      "dynamics_loss_avg": 0.06318080481141805,
      "epoch": 5.008697509765625e-05,
      "step": 52520
    },
    {
      "action_loss_avg": 0.0018274825473781676,
      "epoch": 5.008697509765625e-05,
      "step": 52520
    },
    {
      "epoch": 5.008697509765625e-05,
      "step": 52520,
      "training_step_time": 7.525953531265259
    },
    {
      "epoch": 5.008792877197266e-05,
      "model_forward_time": 1.7417080402374268,
      "step": 52521
    },
    {
      "epoch": 5.008792877197266e-05,
      "step": 52521,
      "training_step_time": 6.828792333602905
    },
    {
      "epoch": 5.008888244628906e-05,
      "model_forward_time": 1.7525551319122314,
      "step": 52522
    },
    {
      "epoch": 5.008888244628906e-05,
      "step": 52522,
      "training_step_time": 6.772287130355835
    },
    {
      "epoch": 5.008983612060547e-05,
      "model_forward_time": 1.7397692203521729,
      "step": 52523
    },
    {
      "epoch": 5.008983612060547e-05,
      "step": 52523,
      "training_step_time": 6.870238304138184
    },
    {
      "epoch": 5.0090789794921876e-05,
      "model_forward_time": 1.7442524433135986,
      "step": 52524
    },
    {
      "epoch": 5.0090789794921876e-05,
      "step": 52524,
      "training_step_time": 7.501067638397217
    },
    {
      "epoch": 5.009174346923828e-05,
      "model_forward_time": 2.332396984100342,
      "step": 52525
    },
    {
      "epoch": 5.009174346923828e-05,
      "step": 52525,
      "training_step_time": 8.029944896697998
    },
    {
      "epoch": 5.009269714355469e-05,
      "model_forward_time": 1.727853536605835,
      "step": 52526
    },
    {
      "epoch": 5.009269714355469e-05,
      "step": 52526,
      "training_step_time": 9.0957670211792
    },
    {
      "epoch": 5.0093650817871095e-05,
      "model_forward_time": 1.7254629135131836,
      "step": 52527
    },
    {
      "epoch": 5.0093650817871095e-05,
      "step": 52527,
      "training_step_time": 9.180542230606079
    },
    {
      "epoch": 5.00946044921875e-05,
      "model_forward_time": 1.7319786548614502,
      "step": 52528
    },
    {
      "epoch": 5.00946044921875e-05,
      "step": 52528,
      "training_step_time": 9.271541118621826
    },
    {
      "epoch": 5.009555816650391e-05,
      "model_forward_time": 1.7327518463134766,
      "step": 52529
    },
    {
      "epoch": 5.009555816650391e-05,
      "step": 52529,
      "training_step_time": 7.698758840560913
    },
    {
      "epoch": 5.0096511840820314e-05,
      "grad_norm": 0.14388877153396606,
      "learning_rate": 4.995204938263741e-06,
      "loss": 0.059,
      "step": 52530
    },
    {
      "epoch": 5.0096511840820314e-05,
      "model_forward_time": 1.7282078266143799,
      "step": 52530
    },
    {
      "dynamics_loss_avg": 0.05958023685961962,
      "epoch": 5.0096511840820314e-05,
      "step": 52530
    },
    {
      "action_loss_avg": 0.0016320171940606087,
      "epoch": 5.0096511840820314e-05,
      "step": 52530
    },
    {
      "epoch": 5.0096511840820314e-05,
      "step": 52530,
      "training_step_time": 7.428738355636597
    },
    {
      "epoch": 5.009746551513672e-05,
      "model_forward_time": 1.7231974601745605,
      "step": 52531
    },
    {
      "epoch": 5.009746551513672e-05,
      "step": 52531,
      "training_step_time": 6.84462833404541
    },
    {
      "epoch": 5.009841918945312e-05,
      "model_forward_time": 1.726703405380249,
      "step": 52532
    },
    {
      "epoch": 5.009841918945312e-05,
      "step": 52532,
      "training_step_time": 6.811296463012695
    },
    {
      "epoch": 5.009937286376953e-05,
      "model_forward_time": 1.721813678741455,
      "step": 52533
    },
    {
      "epoch": 5.009937286376953e-05,
      "step": 52533,
      "training_step_time": 7.5238401889801025
    },
    {
      "epoch": 5.0100326538085936e-05,
      "model_forward_time": 2.303535223007202,
      "step": 52534
    },
    {
      "epoch": 5.0100326538085936e-05,
      "step": 52534,
      "training_step_time": 7.370972633361816
    },
    {
      "epoch": 5.010128021240234e-05,
      "model_forward_time": 1.7275118827819824,
      "step": 52535
    },
    {
      "epoch": 5.010128021240234e-05,
      "step": 52535,
      "training_step_time": 7.791419982910156
    },
    {
      "epoch": 5.010223388671875e-05,
      "model_forward_time": 1.3239381313323975,
      "step": 52536
    },
    {
      "epoch": 5.010223388671875e-05,
      "step": 52536,
      "training_step_time": 9.378000259399414
    },
    {
      "epoch": 5.0103187561035155e-05,
      "model_forward_time": 1.7110486030578613,
      "step": 52537
    },
    {
      "epoch": 5.0103187561035155e-05,
      "step": 52537,
      "training_step_time": 9.122589826583862
    },
    {
      "epoch": 5.010414123535156e-05,
      "model_forward_time": 1.7233784198760986,
      "step": 52538
    },
    {
      "epoch": 5.010414123535156e-05,
      "step": 52538,
      "training_step_time": 7.905742883682251
    },
    {
      "epoch": 5.010509490966797e-05,
      "model_forward_time": 1.7270972728729248,
      "step": 52539
    },
    {
      "epoch": 5.010509490966797e-05,
      "step": 52539,
      "training_step_time": 7.578502893447876
    },
    {
      "epoch": 5.0106048583984374e-05,
      "grad_norm": 0.1092832013964653,
      "learning_rate": 4.993551469498739e-06,
      "loss": 0.0587,
      "step": 52540
    },
    {
      "epoch": 5.0106048583984374e-05,
      "model_forward_time": 1.716193437576294,
      "step": 52540
    },
    {
      "dynamics_loss_avg": 0.05222876248881221,
      "epoch": 5.0106048583984374e-05,
      "step": 52540
    },
    {
      "action_loss_avg": 0.0015441906987689435,
      "epoch": 5.0106048583984374e-05,
      "step": 52540
    },
    {
      "epoch": 5.0106048583984374e-05,
      "step": 52540,
      "training_step_time": 6.9487144947052
    },
    {
      "epoch": 5.010700225830078e-05,
      "model_forward_time": 1.7287139892578125,
      "step": 52541
    },
    {
      "epoch": 5.010700225830078e-05,
      "step": 52541,
      "training_step_time": 7.014368534088135
    },
    {
      "epoch": 5.010795593261719e-05,
      "model_forward_time": 1.7279226779937744,
      "step": 52542
    },
    {
      "epoch": 5.010795593261719e-05,
      "step": 52542,
      "training_step_time": 7.552002429962158
    },
    {
      "epoch": 5.010890960693359e-05,
      "model_forward_time": 1.7737596035003662,
      "step": 52543
    },
    {
      "epoch": 5.010890960693359e-05,
      "step": 52543,
      "training_step_time": 7.740377426147461
    },
    {
      "epoch": 5.010986328125e-05,
      "model_forward_time": 2.2955100536346436,
      "step": 52544
    },
    {
      "epoch": 5.010986328125e-05,
      "step": 52544,
      "training_step_time": 7.79287576675415
    },
    {
      "epoch": 5.011081695556641e-05,
      "model_forward_time": 1.7367973327636719,
      "step": 52545
    },
    {
      "epoch": 5.011081695556641e-05,
      "step": 52545,
      "training_step_time": 8.418994665145874
    },
    {
      "epoch": 5.011177062988281e-05,
      "model_forward_time": 1.7309887409210205,
      "step": 52546
    },
    {
      "epoch": 5.011177062988281e-05,
      "step": 52546,
      "training_step_time": 8.469934940338135
    },
    {
      "epoch": 5.0112724304199216e-05,
      "model_forward_time": 1.7451210021972656,
      "step": 52547
    },
    {
      "epoch": 5.0112724304199216e-05,
      "step": 52547,
      "training_step_time": 7.612008571624756
    },
    {
      "epoch": 5.011367797851563e-05,
      "model_forward_time": 1.7540488243103027,
      "step": 52548
    },
    {
      "epoch": 5.011367797851563e-05,
      "step": 52548,
      "training_step_time": 7.333819389343262
    },
    {
      "epoch": 5.011463165283203e-05,
      "model_forward_time": 1.741398572921753,
      "step": 52549
    },
    {
      "epoch": 5.011463165283203e-05,
      "step": 52549,
      "training_step_time": 6.775240421295166
    },
    {
      "epoch": 5.0115585327148435e-05,
      "grad_norm": 0.10164550691843033,
      "learning_rate": 4.991898001438941e-06,
      "loss": 0.06,
      "step": 52550
    },
    {
      "epoch": 5.0115585327148435e-05,
      "model_forward_time": 1.7291216850280762,
      "step": 52550
    },
    {
      "dynamics_loss_avg": 0.05903088096529245,
      "epoch": 5.0115585327148435e-05,
      "step": 52550
    },
    {
      "action_loss_avg": 0.0017214938445249573,
      "epoch": 5.0115585327148435e-05,
      "step": 52550
    },
    {
      "epoch": 5.0115585327148435e-05,
      "step": 52550,
      "training_step_time": 6.7566070556640625
    },
    {
      "epoch": 5.0116539001464846e-05,
      "model_forward_time": 1.741344928741455,
      "step": 52551
    },
    {
      "epoch": 5.0116539001464846e-05,
      "step": 52551,
      "training_step_time": 7.290665864944458
    },
    {
      "epoch": 5.011749267578125e-05,
      "model_forward_time": 1.7972373962402344,
      "step": 52552
    },
    {
      "epoch": 5.011749267578125e-05,
      "step": 52552,
      "training_step_time": 7.525434494018555
    },
    {
      "epoch": 5.0118446350097654e-05,
      "model_forward_time": 2.303739070892334,
      "step": 52553
    },
    {
      "epoch": 5.0118446350097654e-05,
      "step": 52553,
      "training_step_time": 9.208366632461548
    },
    {
      "epoch": 5.0119400024414064e-05,
      "model_forward_time": 1.7396972179412842,
      "step": 52554
    },
    {
      "epoch": 5.0119400024414064e-05,
      "step": 52554,
      "training_step_time": 7.672156095504761
    },
    {
      "epoch": 5.012035369873047e-05,
      "model_forward_time": 1.7394475936889648,
      "step": 52555
    },
    {
      "epoch": 5.012035369873047e-05,
      "step": 52555,
      "training_step_time": 8.398797988891602
    },
    {
      "epoch": 5.012130737304687e-05,
      "model_forward_time": 1.7397024631500244,
      "step": 52556
    },
    {
      "epoch": 5.012130737304687e-05,
      "step": 52556,
      "training_step_time": 8.479874849319458
    },
    {
      "epoch": 5.012226104736328e-05,
      "model_forward_time": 1.7438154220581055,
      "step": 52557
    },
    {
      "epoch": 5.012226104736328e-05,
      "step": 52557,
      "training_step_time": 7.354785203933716
    },
    {
      "epoch": 5.012321472167969e-05,
      "model_forward_time": 1.3540613651275635,
      "step": 52558
    },
    {
      "epoch": 5.012321472167969e-05,
      "step": 52558,
      "training_step_time": 6.6971235275268555
    },
    {
      "epoch": 5.012416839599609e-05,
      "model_forward_time": 1.7405030727386475,
      "step": 52559
    },
    {
      "epoch": 5.012416839599609e-05,
      "step": 52559,
      "training_step_time": 6.778494358062744
    },
    {
      "epoch": 5.01251220703125e-05,
      "grad_norm": 0.11126905679702759,
      "learning_rate": 4.990244534265166e-06,
      "loss": 0.0574,
      "step": 52560
    },
    {
      "epoch": 5.01251220703125e-05,
      "model_forward_time": 1.732938289642334,
      "step": 52560
    },
    {
      "dynamics_loss_avg": 0.05787776652723551,
      "epoch": 5.01251220703125e-05,
      "step": 52560
    },
    {
      "action_loss_avg": 0.0018846433376893401,
      "epoch": 5.01251220703125e-05,
      "step": 52560
    },
    {
      "epoch": 5.01251220703125e-05,
      "step": 52560,
      "training_step_time": 7.167065620422363
    },
    {
      "epoch": 5.0126075744628906e-05,
      "model_forward_time": 1.7912209033966064,
      "step": 52561
    },
    {
      "epoch": 5.0126075744628906e-05,
      "step": 52561,
      "training_step_time": 7.520928859710693
    },
    {
      "epoch": 5.012702941894531e-05,
      "model_forward_time": 2.3331034183502197,
      "step": 52562
    },
    {
      "epoch": 5.012702941894531e-05,
      "step": 52562,
      "training_step_time": 7.511561393737793
    },
    {
      "epoch": 5.012798309326172e-05,
      "model_forward_time": 1.7434358596801758,
      "step": 52563
    },
    {
      "epoch": 5.012798309326172e-05,
      "step": 52563,
      "training_step_time": 8.965384721755981
    },
    {
      "epoch": 5.0128936767578125e-05,
      "model_forward_time": 1.7511720657348633,
      "step": 52564
    },
    {
      "epoch": 5.0128936767578125e-05,
      "step": 52564,
      "training_step_time": 8.770464420318604
    },
    {
      "epoch": 5.012989044189453e-05,
      "model_forward_time": 1.7390506267547607,
      "step": 52565
    },
    {
      "epoch": 5.012989044189453e-05,
      "step": 52565,
      "training_step_time": 8.130422353744507
    },
    {
      "epoch": 5.013084411621094e-05,
      "model_forward_time": 1.7530083656311035,
      "step": 52566
    },
    {
      "epoch": 5.013084411621094e-05,
      "step": 52566,
      "training_step_time": 7.562443256378174
    },
    {
      "epoch": 5.0131797790527344e-05,
      "model_forward_time": 1.7347817420959473,
      "step": 52567
    },
    {
      "epoch": 5.0131797790527344e-05,
      "step": 52567,
      "training_step_time": 7.352908372879028
    },
    {
      "epoch": 5.013275146484375e-05,
      "model_forward_time": 1.7329962253570557,
      "step": 52568
    },
    {
      "epoch": 5.013275146484375e-05,
      "step": 52568,
      "training_step_time": 6.759605646133423
    },
    {
      "epoch": 5.013370513916016e-05,
      "model_forward_time": 1.7309556007385254,
      "step": 52569
    },
    {
      "epoch": 5.013370513916016e-05,
      "step": 52569,
      "training_step_time": 7.171348333358765
    },
    {
      "epoch": 5.013465881347656e-05,
      "grad_norm": 0.10560812056064606,
      "learning_rate": 4.9885910681582336e-06,
      "loss": 0.0551,
      "step": 52570
    },
    {
      "epoch": 5.013465881347656e-05,
      "model_forward_time": 1.7823877334594727,
      "step": 52570
    },
    {
      "dynamics_loss_avg": 0.05782692357897758,
      "epoch": 5.013465881347656e-05,
      "step": 52570
    },
    {
      "action_loss_avg": 0.001665045419940725,
      "epoch": 5.013465881347656e-05,
      "step": 52570
    },
    {
      "epoch": 5.013465881347656e-05,
      "step": 52570,
      "training_step_time": 8.362261772155762
    },
    {
      "epoch": 5.013561248779297e-05,
      "model_forward_time": 1.7327029705047607,
      "step": 52571
    },
    {
      "epoch": 5.013561248779297e-05,
      "step": 52571,
      "training_step_time": 8.8470299243927
    },
    {
      "epoch": 5.013656616210938e-05,
      "model_forward_time": 1.7402541637420654,
      "step": 52572
    },
    {
      "epoch": 5.013656616210938e-05,
      "step": 52572,
      "training_step_time": 9.387099981307983
    },
    {
      "epoch": 5.013751983642578e-05,
      "model_forward_time": 1.730755090713501,
      "step": 52573
    },
    {
      "epoch": 5.013751983642578e-05,
      "step": 52573,
      "training_step_time": 9.154505252838135
    },
    {
      "epoch": 5.0138473510742186e-05,
      "model_forward_time": 1.739353895187378,
      "step": 52574
    },
    {
      "epoch": 5.0138473510742186e-05,
      "step": 52574,
      "training_step_time": 9.229308605194092
    },
    {
      "epoch": 5.0139427185058596e-05,
      "model_forward_time": 1.7324299812316895,
      "step": 52575
    },
    {
      "epoch": 5.0139427185058596e-05,
      "step": 52575,
      "training_step_time": 7.563255548477173
    },
    {
      "epoch": 5.0140380859375e-05,
      "model_forward_time": 1.7412700653076172,
      "step": 52576
    },
    {
      "epoch": 5.0140380859375e-05,
      "step": 52576,
      "training_step_time": 7.427204847335815
    },
    {
      "epoch": 5.0141334533691405e-05,
      "model_forward_time": 1.734846830368042,
      "step": 52577
    },
    {
      "epoch": 5.0141334533691405e-05,
      "step": 52577,
      "training_step_time": 6.791037559509277
    },
    {
      "epoch": 5.0142288208007815e-05,
      "model_forward_time": 1.7289743423461914,
      "step": 52578
    },
    {
      "epoch": 5.0142288208007815e-05,
      "step": 52578,
      "training_step_time": 7.193423509597778
    },
    {
      "epoch": 5.014324188232422e-05,
      "model_forward_time": 1.7827141284942627,
      "step": 52579
    },
    {
      "epoch": 5.014324188232422e-05,
      "step": 52579,
      "training_step_time": 7.440586090087891
    },
    {
      "epoch": 5.014419555664062e-05,
      "grad_norm": 0.09286066889762878,
      "learning_rate": 4.986937603298965e-06,
      "loss": 0.057,
      "step": 52580
    },
    {
      "epoch": 5.014419555664062e-05,
      "model_forward_time": 1.736980676651001,
      "step": 52580
    },
    {
      "dynamics_loss_avg": 0.054571051709353925,
      "epoch": 5.014419555664062e-05,
      "step": 52580
    },
    {
      "action_loss_avg": 0.0016844558645971119,
      "epoch": 5.014419555664062e-05,
      "step": 52580
    },
    {
      "epoch": 5.014419555664062e-05,
      "step": 52580,
      "training_step_time": 7.614580869674683
    },
    {
      "epoch": 5.0145149230957034e-05,
      "model_forward_time": 2.2902796268463135,
      "step": 52581
    },
    {
      "epoch": 5.0145149230957034e-05,
      "step": 52581,
      "training_step_time": 8.272475719451904
    },
    {
      "epoch": 5.014610290527344e-05,
      "model_forward_time": 1.732064962387085,
      "step": 52582
    },
    {
      "epoch": 5.014610290527344e-05,
      "step": 52582,
      "training_step_time": 8.602400302886963
    },
    {
      "epoch": 5.014705657958984e-05,
      "model_forward_time": 1.7365295886993408,
      "step": 52583
    },
    {
      "epoch": 5.014705657958984e-05,
      "step": 52583,
      "training_step_time": 8.530621528625488
    },
    {
      "epoch": 5.014801025390625e-05,
      "model_forward_time": 1.7318220138549805,
      "step": 52584
    },
    {
      "epoch": 5.014801025390625e-05,
      "step": 52584,
      "training_step_time": 8.722534418106079
    },
    {
      "epoch": 5.014896392822266e-05,
      "model_forward_time": 1.7330095767974854,
      "step": 52585
    },
    {
      "epoch": 5.014896392822266e-05,
      "step": 52585,
      "training_step_time": 7.433392524719238
    },
    {
      "epoch": 5.014991760253906e-05,
      "model_forward_time": 1.7443714141845703,
      "step": 52586
    },
    {
      "epoch": 5.014991760253906e-05,
      "step": 52586,
      "training_step_time": 6.820466041564941
    },
    {
      "epoch": 5.015087127685547e-05,
      "model_forward_time": 1.732630729675293,
      "step": 52587
    },
    {
      "epoch": 5.015087127685547e-05,
      "step": 52587,
      "training_step_time": 7.223387002944946
    },
    {
      "epoch": 5.0151824951171876e-05,
      "model_forward_time": 1.7811064720153809,
      "step": 52588
    },
    {
      "epoch": 5.0151824951171876e-05,
      "step": 52588,
      "training_step_time": 7.427990913391113
    },
    {
      "epoch": 5.015277862548828e-05,
      "model_forward_time": 1.7353951930999756,
      "step": 52589
    },
    {
      "epoch": 5.015277862548828e-05,
      "step": 52589,
      "training_step_time": 7.629432201385498
    },
    {
      "epoch": 5.015373229980469e-05,
      "grad_norm": 0.09081736207008362,
      "learning_rate": 4.985284139868181e-06,
      "loss": 0.0566,
      "step": 52590
    },
    {
      "epoch": 5.015373229980469e-05,
      "model_forward_time": 2.2862160205841064,
      "step": 52590
    },
    {
      "dynamics_loss_avg": 0.05910988701507449,
      "epoch": 5.015373229980469e-05,
      "step": 52590
    },
    {
      "action_loss_avg": 0.0015571062656817957,
      "epoch": 5.015373229980469e-05,
      "step": 52590
    },
    {
      "epoch": 5.015373229980469e-05,
      "step": 52590,
      "training_step_time": 9.717724084854126
    },
    {
      "epoch": 5.0154685974121095e-05,
      "model_forward_time": 1.7304041385650635,
      "step": 52591
    },
    {
      "epoch": 5.0154685974121095e-05,
      "step": 52591,
      "training_step_time": 8.497581958770752
    },
    {
      "epoch": 5.01556396484375e-05,
      "model_forward_time": 1.7343909740447998,
      "step": 52592
    },
    {
      "epoch": 5.01556396484375e-05,
      "step": 52592,
      "training_step_time": 7.649526596069336
    },
    {
      "epoch": 5.015659332275391e-05,
      "model_forward_time": 1.7453806400299072,
      "step": 52593
    },
    {
      "epoch": 5.015659332275391e-05,
      "step": 52593,
      "training_step_time": 7.5904011726379395
    },
    {
      "epoch": 5.0157546997070314e-05,
      "model_forward_time": 1.732525110244751,
      "step": 52594
    },
    {
      "epoch": 5.0157546997070314e-05,
      "step": 52594,
      "training_step_time": 8.208679676055908
    },
    {
      "epoch": 5.015850067138672e-05,
      "model_forward_time": 1.7326374053955078,
      "step": 52595
    },
    {
      "epoch": 5.015850067138672e-05,
      "step": 52595,
      "training_step_time": 7.203979253768921
    },
    {
      "epoch": 5.015945434570312e-05,
      "model_forward_time": 1.7295348644256592,
      "step": 52596
    },
    {
      "epoch": 5.015945434570312e-05,
      "step": 52596,
      "training_step_time": 6.765893936157227
    },
    {
      "epoch": 5.016040802001953e-05,
      "model_forward_time": 1.771599292755127,
      "step": 52597
    },
    {
      "epoch": 5.016040802001953e-05,
      "step": 52597,
      "training_step_time": 7.303211688995361
    },
    {
      "epoch": 5.0161361694335937e-05,
      "model_forward_time": 1.7277636528015137,
      "step": 52598
    },
    {
      "epoch": 5.0161361694335937e-05,
      "step": 52598,
      "training_step_time": 7.596797466278076
    },
    {
      "epoch": 5.016231536865234e-05,
      "model_forward_time": 1.8890800476074219,
      "step": 52599
    },
    {
      "epoch": 5.016231536865234e-05,
      "step": 52599,
      "training_step_time": 8.09826111793518
    },
    {
      "epoch": 5.016326904296875e-05,
      "grad_norm": 0.08707858622074127,
      "learning_rate": 4.9836306780467e-06,
      "loss": 0.0537,
      "step": 52600
    },
    {
      "epoch": 5.016326904296875e-05,
      "model_forward_time": 1.7200860977172852,
      "step": 52600
    },
    {
      "dynamics_loss_avg": 0.0530230063945055,
      "epoch": 5.016326904296875e-05,
      "step": 52600
    },
    {
      "action_loss_avg": 0.0019290382449980825,
      "epoch": 5.016326904296875e-05,
      "step": 52600
    },
    {
      "epoch": 5.016326904296875e-05,
      "step": 52600,
      "training_step_time": 9.459920167922974
    },
    {
      "epoch": 5.0164222717285155e-05,
      "model_forward_time": 1.7318074703216553,
      "step": 52601
    },
    {
      "epoch": 5.0164222717285155e-05,
      "step": 52601,
      "training_step_time": 9.221816778182983
    },
    {
      "epoch": 5.016517639160156e-05,
      "model_forward_time": 1.7271969318389893,
      "step": 52602
    },
    {
      "epoch": 5.016517639160156e-05,
      "step": 52602,
      "training_step_time": 10.030428171157837
    },
    {
      "epoch": 5.016613006591797e-05,
      "model_forward_time": 1.736607551574707,
      "step": 52603
    },
    {
      "epoch": 5.016613006591797e-05,
      "step": 52603,
      "training_step_time": 7.653086185455322
    },
    {
      "epoch": 5.0167083740234374e-05,
      "model_forward_time": 1.7318434715270996,
      "step": 52604
    },
    {
      "epoch": 5.0167083740234374e-05,
      "step": 52604,
      "training_step_time": 7.448754787445068
    },
    {
      "epoch": 5.016803741455078e-05,
      "model_forward_time": 1.7282803058624268,
      "step": 52605
    },
    {
      "epoch": 5.016803741455078e-05,
      "step": 52605,
      "training_step_time": 6.889314889907837
    },
    {
      "epoch": 5.016899108886719e-05,
      "model_forward_time": 1.7520508766174316,
      "step": 52606
    },
    {
      "epoch": 5.016899108886719e-05,
      "step": 52606,
      "training_step_time": 7.226824760437012
    },
    {
      "epoch": 5.016994476318359e-05,
      "model_forward_time": 1.734098196029663,
      "step": 52607
    },
    {
      "epoch": 5.016994476318359e-05,
      "step": 52607,
      "training_step_time": 7.6446850299835205
    },
    {
      "epoch": 5.01708984375e-05,
      "model_forward_time": 2.294062614440918,
      "step": 52608
    },
    {
      "epoch": 5.01708984375e-05,
      "step": 52608,
      "training_step_time": 8.562865018844604
    },
    {
      "epoch": 5.017185211181641e-05,
      "model_forward_time": 1.7295811176300049,
      "step": 52609
    },
    {
      "epoch": 5.017185211181641e-05,
      "step": 52609,
      "training_step_time": 9.980976343154907
    },
    {
      "epoch": 5.017280578613281e-05,
      "grad_norm": 0.10623158514499664,
      "learning_rate": 4.981977218015343e-06,
      "loss": 0.0524,
      "step": 52610
    },
    {
      "epoch": 5.017280578613281e-05,
      "model_forward_time": 1.7193045616149902,
      "step": 52610
    },
    {
      "dynamics_loss_avg": 0.04684268031269312,
      "epoch": 5.017280578613281e-05,
      "step": 52610
    },
    {
      "action_loss_avg": 0.0014286660138168373,
      "epoch": 5.017280578613281e-05,
      "step": 52610
    },
    {
      "epoch": 5.017280578613281e-05,
      "step": 52610,
      "training_step_time": 9.38996434211731
    },
    {
      "epoch": 5.0173759460449216e-05,
      "model_forward_time": 1.7042591571807861,
      "step": 52611
    },
    {
      "epoch": 5.0173759460449216e-05,
      "step": 52611,
      "training_step_time": 9.670649290084839
    },
    {
      "epoch": 5.017471313476563e-05,
      "model_forward_time": 1.7299778461456299,
      "step": 52612
    },
    {
      "epoch": 5.017471313476563e-05,
      "step": 52612,
      "training_step_time": 7.578056335449219
    },
    {
      "epoch": 5.017566680908203e-05,
      "model_forward_time": 1.7296802997589111,
      "step": 52613
    },
    {
      "epoch": 5.017566680908203e-05,
      "step": 52613,
      "training_step_time": 7.441544532775879
    },
    {
      "epoch": 5.0176620483398435e-05,
      "model_forward_time": 1.7260289192199707,
      "step": 52614
    },
    {
      "epoch": 5.0176620483398435e-05,
      "step": 52614,
      "training_step_time": 6.773972988128662
    },
    {
      "epoch": 5.0177574157714846e-05,
      "model_forward_time": 1.733464241027832,
      "step": 52615
    },
    {
      "epoch": 5.0177574157714846e-05,
      "step": 52615,
      "training_step_time": 7.162090063095093
    },
    {
      "epoch": 5.017852783203125e-05,
      "model_forward_time": 1.7370834350585938,
      "step": 52616
    },
    {
      "epoch": 5.017852783203125e-05,
      "step": 52616,
      "training_step_time": 7.615802049636841
    },
    {
      "epoch": 5.0179481506347654e-05,
      "model_forward_time": 1.7145652770996094,
      "step": 52617
    },
    {
      "epoch": 5.0179481506347654e-05,
      "step": 52617,
      "training_step_time": 7.951870441436768
    },
    {
      "epoch": 5.0180435180664065e-05,
      "model_forward_time": 1.7156624794006348,
      "step": 52618
    },
    {
      "epoch": 5.0180435180664065e-05,
      "step": 52618,
      "training_step_time": 8.525796890258789
    },
    {
      "epoch": 5.018138885498047e-05,
      "model_forward_time": 1.7292499542236328,
      "step": 52619
    },
    {
      "epoch": 5.018138885498047e-05,
      "step": 52619,
      "training_step_time": 8.556763172149658
    },
    {
      "epoch": 5.018234252929687e-05,
      "grad_norm": 0.11418702453374863,
      "learning_rate": 4.980323759954932e-06,
      "loss": 0.0548,
      "step": 52620
    },
    {
      "epoch": 5.018234252929687e-05,
      "model_forward_time": 1.7117502689361572,
      "step": 52620
    },
    {
      "dynamics_loss_avg": 0.05600889455527067,
      "epoch": 5.018234252929687e-05,
      "step": 52620
    },
    {
      "action_loss_avg": 0.0011085350110079163,
      "epoch": 5.018234252929687e-05,
      "step": 52620
    },
    {
      "epoch": 5.018234252929687e-05,
      "step": 52620,
      "training_step_time": 7.695084095001221
    },
    {
      "epoch": 5.018329620361328e-05,
      "model_forward_time": 1.7293767929077148,
      "step": 52621
    },
    {
      "epoch": 5.018329620361328e-05,
      "step": 52621,
      "training_step_time": 7.6970295906066895
    },
    {
      "epoch": 5.018424987792969e-05,
      "model_forward_time": 1.7150547504425049,
      "step": 52622
    },
    {
      "epoch": 5.018424987792969e-05,
      "step": 52622,
      "training_step_time": 7.320621490478516
    },
    {
      "epoch": 5.018520355224609e-05,
      "model_forward_time": 1.7301623821258545,
      "step": 52623
    },
    {
      "epoch": 5.018520355224609e-05,
      "step": 52623,
      "training_step_time": 6.798730373382568
    },
    {
      "epoch": 5.01861572265625e-05,
      "model_forward_time": 1.7323532104492188,
      "step": 52624
    },
    {
      "epoch": 5.01861572265625e-05,
      "step": 52624,
      "training_step_time": 7.181772708892822
    },
    {
      "epoch": 5.0187110900878906e-05,
      "model_forward_time": 1.7830653190612793,
      "step": 52625
    },
    {
      "epoch": 5.0187110900878906e-05,
      "step": 52625,
      "training_step_time": 7.4271628856658936
    },
    {
      "epoch": 5.018806457519531e-05,
      "model_forward_time": 1.7307806015014648,
      "step": 52626
    },
    {
      "epoch": 5.018806457519531e-05,
      "step": 52626,
      "training_step_time": 7.382872819900513
    },
    {
      "epoch": 5.018901824951172e-05,
      "model_forward_time": 2.2771155834198,
      "step": 52627
    },
    {
      "epoch": 5.018901824951172e-05,
      "step": 52627,
      "training_step_time": 8.935714960098267
    },
    {
      "epoch": 5.0189971923828125e-05,
      "model_forward_time": 1.736975908279419,
      "step": 52628
    },
    {
      "epoch": 5.0189971923828125e-05,
      "step": 52628,
      "training_step_time": 7.947134971618652
    },
    {
      "epoch": 5.019092559814453e-05,
      "model_forward_time": 1.7243213653564453,
      "step": 52629
    },
    {
      "epoch": 5.019092559814453e-05,
      "step": 52629,
      "training_step_time": 9.995676755905151
    },
    {
      "epoch": 5.019187927246094e-05,
      "grad_norm": 0.10102636367082596,
      "learning_rate": 4.9786703040462846e-06,
      "loss": 0.0553,
      "step": 52630
    },
    {
      "epoch": 5.019187927246094e-05,
      "model_forward_time": 1.7269048690795898,
      "step": 52630
    },
    {
      "dynamics_loss_avg": 0.04868840761482716,
      "epoch": 5.019187927246094e-05,
      "step": 52630
    },
    {
      "action_loss_avg": 0.0013405291188973933,
      "epoch": 5.019187927246094e-05,
      "step": 52630
    },
    {
      "epoch": 5.019187927246094e-05,
      "step": 52630,
      "training_step_time": 8.293322324752808
    },
    {
      "epoch": 5.0192832946777344e-05,
      "model_forward_time": 1.7307343482971191,
      "step": 52631
    },
    {
      "epoch": 5.0192832946777344e-05,
      "step": 52631,
      "training_step_time": 7.357282400131226
    },
    {
      "epoch": 5.019378662109375e-05,
      "model_forward_time": 1.7340278625488281,
      "step": 52632
    },
    {
      "epoch": 5.019378662109375e-05,
      "step": 52632,
      "training_step_time": 6.814958810806274
    },
    {
      "epoch": 5.019474029541016e-05,
      "model_forward_time": 1.7396082878112793,
      "step": 52633
    },
    {
      "epoch": 5.019474029541016e-05,
      "step": 52633,
      "training_step_time": 7.180259704589844
    },
    {
      "epoch": 5.019569396972656e-05,
      "model_forward_time": 1.7796063423156738,
      "step": 52634
    },
    {
      "epoch": 5.019569396972656e-05,
      "step": 52634,
      "training_step_time": 7.460610866546631
    },
    {
      "epoch": 5.019664764404297e-05,
      "model_forward_time": 1.7297444343566895,
      "step": 52635
    },
    {
      "epoch": 5.019664764404297e-05,
      "step": 52635,
      "training_step_time": 7.558080434799194
    },
    {
      "epoch": 5.019760131835938e-05,
      "model_forward_time": 2.3138346672058105,
      "step": 52636
    },
    {
      "epoch": 5.019760131835938e-05,
      "step": 52636,
      "training_step_time": 8.809583187103271
    },
    {
      "epoch": 5.019855499267578e-05,
      "model_forward_time": 1.7273175716400146,
      "step": 52637
    },
    {
      "epoch": 5.019855499267578e-05,
      "step": 52637,
      "training_step_time": 8.56131100654602
    },
    {
      "epoch": 5.0199508666992186e-05,
      "model_forward_time": 1.7344880104064941,
      "step": 52638
    },
    {
      "epoch": 5.0199508666992186e-05,
      "step": 52638,
      "training_step_time": 9.888827562332153
    },
    {
      "epoch": 5.0200462341308597e-05,
      "model_forward_time": 1.7988710403442383,
      "step": 52639
    },
    {
      "epoch": 5.0200462341308597e-05,
      "step": 52639,
      "training_step_time": 8.630079507827759
    },
    {
      "epoch": 5.0201416015625e-05,
      "grad_norm": 0.10340961068868637,
      "learning_rate": 4.97701685047022e-06,
      "loss": 0.0535,
      "step": 52640
    },
    {
      "epoch": 5.0201416015625e-05,
      "model_forward_time": 1.734607219696045,
      "step": 52640
    },
    {
      "dynamics_loss_avg": 0.06769892070442438,
      "epoch": 5.0201416015625e-05,
      "step": 52640
    },
    {
      "action_loss_avg": 0.003563496586866677,
      "epoch": 5.0201416015625e-05,
      "step": 52640
    },
    {
      "epoch": 5.0201416015625e-05,
      "step": 52640,
      "training_step_time": 8.28687310218811
    },
    {
      "epoch": 5.0202369689941405e-05,
      "model_forward_time": 1.7321109771728516,
      "step": 52641
    },
    {
      "epoch": 5.0202369689941405e-05,
      "step": 52641,
      "training_step_time": 7.3671393394470215
    },
    {
      "epoch": 5.0203323364257815e-05,
      "model_forward_time": 1.7295477390289307,
      "step": 52642
    },
    {
      "epoch": 5.0203323364257815e-05,
      "step": 52642,
      "training_step_time": 7.175655126571655
    },
    {
      "epoch": 5.020427703857422e-05,
      "model_forward_time": 1.7589592933654785,
      "step": 52643
    },
    {
      "epoch": 5.020427703857422e-05,
      "step": 52643,
      "training_step_time": 7.439173936843872
    },
    {
      "epoch": 5.0205230712890623e-05,
      "model_forward_time": 1.7323639392852783,
      "step": 52644
    },
    {
      "epoch": 5.0205230712890623e-05,
      "step": 52644,
      "training_step_time": 7.450782060623169
    },
    {
      "epoch": 5.0206184387207034e-05,
      "model_forward_time": 2.3045785427093506,
      "step": 52645
    },
    {
      "epoch": 5.0206184387207034e-05,
      "step": 52645,
      "training_step_time": 8.457046270370483
    },
    {
      "epoch": 5.020713806152344e-05,
      "model_forward_time": 1.7349774837493896,
      "step": 52646
    },
    {
      "epoch": 5.020713806152344e-05,
      "step": 52646,
      "training_step_time": 7.862128734588623
    },
    {
      "epoch": 5.020809173583984e-05,
      "model_forward_time": 1.7313213348388672,
      "step": 52647
    },
    {
      "epoch": 5.020809173583984e-05,
      "step": 52647,
      "training_step_time": 8.552590131759644
    },
    {
      "epoch": 5.020904541015625e-05,
      "model_forward_time": 1.7281842231750488,
      "step": 52648
    },
    {
      "epoch": 5.020904541015625e-05,
      "step": 52648,
      "training_step_time": 7.689419746398926
    },
    {
      "epoch": 5.020999908447266e-05,
      "model_forward_time": 1.7383766174316406,
      "step": 52649
    },
    {
      "epoch": 5.020999908447266e-05,
      "step": 52649,
      "training_step_time": 7.692326307296753
    },
    {
      "epoch": 5.021095275878906e-05,
      "grad_norm": 0.09498956054449081,
      "learning_rate": 4.975363399407559e-06,
      "loss": 0.051,
      "step": 52650
    },
    {
      "epoch": 5.021095275878906e-05,
      "model_forward_time": 1.723097801208496,
      "step": 52650
    },
    {
      "dynamics_loss_avg": 0.05123465545475483,
      "epoch": 5.021095275878906e-05,
      "step": 52650
    },
    {
      "action_loss_avg": 0.0018333345942664891,
      "epoch": 5.021095275878906e-05,
      "step": 52650
    },
    {
      "epoch": 5.021095275878906e-05,
      "step": 52650,
      "training_step_time": 7.396487474441528
    },
    {
      "epoch": 5.021190643310547e-05,
      "model_forward_time": 1.7324955463409424,
      "step": 52651
    },
    {
      "epoch": 5.021190643310547e-05,
      "step": 52651,
      "training_step_time": 7.190442323684692
    },
    {
      "epoch": 5.0212860107421876e-05,
      "model_forward_time": 1.7580583095550537,
      "step": 52652
    },
    {
      "epoch": 5.0212860107421876e-05,
      "step": 52652,
      "training_step_time": 7.403058767318726
    },
    {
      "epoch": 5.021381378173828e-05,
      "model_forward_time": 1.7826805114746094,
      "step": 52653
    },
    {
      "epoch": 5.021381378173828e-05,
      "step": 52653,
      "training_step_time": 8.315575361251831
    },
    {
      "epoch": 5.021476745605469e-05,
      "model_forward_time": 2.3049283027648926,
      "step": 52654
    },
    {
      "epoch": 5.021476745605469e-05,
      "step": 52654,
      "training_step_time": 8.220543622970581
    },
    {
      "epoch": 5.0215721130371095e-05,
      "model_forward_time": 1.7323217391967773,
      "step": 52655
    },
    {
      "epoch": 5.0215721130371095e-05,
      "step": 52655,
      "training_step_time": 8.592494487762451
    },
    {
      "epoch": 5.02166748046875e-05,
      "model_forward_time": 1.7310750484466553,
      "step": 52656
    },
    {
      "epoch": 5.02166748046875e-05,
      "step": 52656,
      "training_step_time": 7.6049675941467285
    },
    {
      "epoch": 5.02176284790039e-05,
      "model_forward_time": 1.7320153713226318,
      "step": 52657
    },
    {
      "epoch": 5.02176284790039e-05,
      "step": 52657,
      "training_step_time": 9.396560907363892
    },
    {
      "epoch": 5.0218582153320314e-05,
      "model_forward_time": 1.7389767169952393,
      "step": 52658
    },
    {
      "epoch": 5.0218582153320314e-05,
      "step": 52658,
      "training_step_time": 7.545307159423828
    },
    {
      "epoch": 5.021953582763672e-05,
      "model_forward_time": 1.733567476272583,
      "step": 52659
    },
    {
      "epoch": 5.021953582763672e-05,
      "step": 52659,
      "training_step_time": 7.369368076324463
    },
    {
      "epoch": 5.022048950195312e-05,
      "grad_norm": 0.10176065564155579,
      "learning_rate": 4.97370995103912e-06,
      "loss": 0.0559,
      "step": 52660
    },
    {
      "epoch": 5.022048950195312e-05,
      "model_forward_time": 1.732299566268921,
      "step": 52660
    },
    {
      "dynamics_loss_avg": 0.05811265325173735,
      "epoch": 5.022048950195312e-05,
      "step": 52660
    },
    {
      "action_loss_avg": 0.0010320888162823395,
      "epoch": 5.022048950195312e-05,
      "step": 52660
    },
    {
      "epoch": 5.022048950195312e-05,
      "step": 52660,
      "training_step_time": 7.151177167892456
    },
    {
      "epoch": 5.022144317626953e-05,
      "model_forward_time": 1.7312407493591309,
      "step": 52661
    },
    {
      "epoch": 5.022144317626953e-05,
      "step": 52661,
      "training_step_time": 7.3204755783081055
    },
    {
      "epoch": 5.0222396850585937e-05,
      "model_forward_time": 1.7755215167999268,
      "step": 52662
    },
    {
      "epoch": 5.0222396850585937e-05,
      "step": 52662,
      "training_step_time": 7.4038310050964355
    },
    {
      "epoch": 5.022335052490234e-05,
      "model_forward_time": 2.2966716289520264,
      "step": 52663
    },
    {
      "epoch": 5.022335052490234e-05,
      "step": 52663,
      "training_step_time": 7.522874593734741
    },
    {
      "epoch": 5.022430419921875e-05,
      "model_forward_time": 1.7359144687652588,
      "step": 52664
    },
    {
      "epoch": 5.022430419921875e-05,
      "step": 52664,
      "training_step_time": 8.503972053527832
    },
    {
      "epoch": 5.0225257873535155e-05,
      "model_forward_time": 1.729231357574463,
      "step": 52665
    },
    {
      "epoch": 5.0225257873535155e-05,
      "step": 52665,
      "training_step_time": 7.734012842178345
    },
    {
      "epoch": 5.022621154785156e-05,
      "model_forward_time": 1.7369575500488281,
      "step": 52666
    },
    {
      "epoch": 5.022621154785156e-05,
      "step": 52666,
      "training_step_time": 7.693467140197754
    },
    {
      "epoch": 5.022716522216797e-05,
      "model_forward_time": 1.7421927452087402,
      "step": 52667
    },
    {
      "epoch": 5.022716522216797e-05,
      "step": 52667,
      "training_step_time": 7.565056562423706
    },
    {
      "epoch": 5.0228118896484374e-05,
      "model_forward_time": 1.7327544689178467,
      "step": 52668
    },
    {
      "epoch": 5.0228118896484374e-05,
      "step": 52668,
      "training_step_time": 7.605798959732056
    },
    {
      "epoch": 5.022907257080078e-05,
      "model_forward_time": 1.7331476211547852,
      "step": 52669
    },
    {
      "epoch": 5.022907257080078e-05,
      "step": 52669,
      "training_step_time": 7.759180307388306
    },
    {
      "epoch": 5.023002624511719e-05,
      "grad_norm": 0.0903833732008934,
      "learning_rate": 4.972056505545719e-06,
      "loss": 0.0532,
      "step": 52670
    },
    {
      "epoch": 5.023002624511719e-05,
      "model_forward_time": 1.7256865501403809,
      "step": 52670
    },
    {
      "dynamics_loss_avg": 0.0594493567943573,
      "epoch": 5.023002624511719e-05,
      "step": 52670
    },
    {
      "action_loss_avg": 0.001960312877781689,
      "epoch": 5.023002624511719e-05,
      "step": 52670
    },
    {
      "epoch": 5.023002624511719e-05,
      "step": 52670,
      "training_step_time": 7.2610554695129395
    },
    {
      "epoch": 5.023097991943359e-05,
      "model_forward_time": 1.7764723300933838,
      "step": 52671
    },
    {
      "epoch": 5.023097991943359e-05,
      "step": 52671,
      "training_step_time": 7.399062871932983
    },
    {
      "epoch": 5.023193359375e-05,
      "model_forward_time": 2.3066627979278564,
      "step": 52672
    },
    {
      "epoch": 5.023193359375e-05,
      "step": 52672,
      "training_step_time": 7.552658557891846
    },
    {
      "epoch": 5.023288726806641e-05,
      "model_forward_time": 1.7353103160858154,
      "step": 52673
    },
    {
      "epoch": 5.023288726806641e-05,
      "step": 52673,
      "training_step_time": 8.01227617263794
    },
    {
      "epoch": 5.023384094238281e-05,
      "model_forward_time": 1.7314765453338623,
      "step": 52674
    },
    {
      "epoch": 5.023384094238281e-05,
      "step": 52674,
      "training_step_time": 8.694440126419067
    },
    {
      "epoch": 5.0234794616699216e-05,
      "model_forward_time": 1.7341177463531494,
      "step": 52675
    },
    {
      "epoch": 5.0234794616699216e-05,
      "step": 52675,
      "training_step_time": 9.462728500366211
    },
    {
      "epoch": 5.023574829101563e-05,
      "model_forward_time": 1.735844612121582,
      "step": 52676
    },
    {
      "epoch": 5.023574829101563e-05,
      "step": 52676,
      "training_step_time": 7.620495557785034
    },
    {
      "epoch": 5.023670196533203e-05,
      "model_forward_time": 1.7315583229064941,
      "step": 52677
    },
    {
      "epoch": 5.023670196533203e-05,
      "step": 52677,
      "training_step_time": 8.677088499069214
    },
    {
      "epoch": 5.0237655639648435e-05,
      "model_forward_time": 1.7281897068023682,
      "step": 52678
    },
    {
      "epoch": 5.0237655639648435e-05,
      "step": 52678,
      "training_step_time": 7.680436134338379
    },
    {
      "epoch": 5.0238609313964846e-05,
      "model_forward_time": 1.7330195903778076,
      "step": 52679
    },
    {
      "epoch": 5.0238609313964846e-05,
      "step": 52679,
      "training_step_time": 7.189038991928101
    },
    {
      "epoch": 5.023956298828125e-05,
      "grad_norm": 0.09234913438558578,
      "learning_rate": 4.970403063108178e-06,
      "loss": 0.0523,
      "step": 52680
    },
    {
      "epoch": 5.023956298828125e-05,
      "model_forward_time": 1.7662031650543213,
      "step": 52680
    },
    {
      "dynamics_loss_avg": 0.05027256291359663,
      "epoch": 5.023956298828125e-05,
      "step": 52680
    },
    {
      "action_loss_avg": 0.001338636054424569,
      "epoch": 5.023956298828125e-05,
      "step": 52680
    },
    {
      "epoch": 5.023956298828125e-05,
      "step": 52680,
      "training_step_time": 7.351834774017334
    },
    {
      "epoch": 5.0240516662597654e-05,
      "model_forward_time": 2.304760694503784,
      "step": 52681
    },
    {
      "epoch": 5.0240516662597654e-05,
      "step": 52681,
      "training_step_time": 8.097126483917236
    },
    {
      "epoch": 5.0241470336914065e-05,
      "model_forward_time": 1.7295880317687988,
      "step": 52682
    },
    {
      "epoch": 5.0241470336914065e-05,
      "step": 52682,
      "training_step_time": 8.303855180740356
    },
    {
      "epoch": 5.024242401123047e-05,
      "model_forward_time": 1.7418596744537354,
      "step": 52683
    },
    {
      "epoch": 5.024242401123047e-05,
      "step": 52683,
      "training_step_time": 9.410100936889648
    },
    {
      "epoch": 5.024337768554687e-05,
      "model_forward_time": 1.7299106121063232,
      "step": 52684
    },
    {
      "epoch": 5.024337768554687e-05,
      "step": 52684,
      "training_step_time": 8.360852718353271
    },
    {
      "epoch": 5.0244331359863283e-05,
      "model_forward_time": 1.7400052547454834,
      "step": 52685
    },
    {
      "epoch": 5.0244331359863283e-05,
      "step": 52685,
      "training_step_time": 8.870508670806885
    },
    {
      "epoch": 5.024528503417969e-05,
      "model_forward_time": 1.7314350605010986,
      "step": 52686
    },
    {
      "epoch": 5.024528503417969e-05,
      "step": 52686,
      "training_step_time": 7.612645864486694
    },
    {
      "epoch": 5.024623870849609e-05,
      "model_forward_time": 1.71213698387146,
      "step": 52687
    },
    {
      "epoch": 5.024623870849609e-05,
      "step": 52687,
      "training_step_time": 7.7996110916137695
    },
    {
      "epoch": 5.02471923828125e-05,
      "model_forward_time": 1.7051591873168945,
      "step": 52688
    },
    {
      "epoch": 5.02471923828125e-05,
      "step": 52688,
      "training_step_time": 6.6900835037231445
    },
    {
      "epoch": 5.0248146057128906e-05,
      "model_forward_time": 1.7394523620605469,
      "step": 52689
    },
    {
      "epoch": 5.0248146057128906e-05,
      "step": 52689,
      "training_step_time": 7.425007581710815
    },
    {
      "epoch": 5.024909973144531e-05,
      "grad_norm": 0.10505523532629013,
      "learning_rate": 4.9687496239073126e-06,
      "loss": 0.0567,
      "step": 52690
    },
    {
      "epoch": 5.024909973144531e-05,
      "model_forward_time": 2.286379337310791,
      "step": 52690
    },
    {
      "dynamics_loss_avg": 0.05835949741303921,
      "epoch": 5.024909973144531e-05,
      "step": 52690
    },
    {
      "action_loss_avg": 0.0012040063156746328,
      "epoch": 5.024909973144531e-05,
      "step": 52690
    },
    {
      "epoch": 5.024909973144531e-05,
      "step": 52690,
      "training_step_time": 7.413298606872559
    },
    {
      "epoch": 5.025005340576172e-05,
      "model_forward_time": 1.7230861186981201,
      "step": 52691
    },
    {
      "epoch": 5.025005340576172e-05,
      "step": 52691,
      "training_step_time": 7.56308388710022
    },
    {
      "epoch": 5.0251007080078125e-05,
      "model_forward_time": 1.7273955345153809,
      "step": 52692
    },
    {
      "epoch": 5.0251007080078125e-05,
      "step": 52692,
      "training_step_time": 9.163679122924805
    },
    {
      "epoch": 5.025196075439453e-05,
      "model_forward_time": 1.729318380355835,
      "step": 52693
    },
    {
      "epoch": 5.025196075439453e-05,
      "step": 52693,
      "training_step_time": 8.317870378494263
    },
    {
      "epoch": 5.025291442871094e-05,
      "model_forward_time": 1.731447458267212,
      "step": 52694
    },
    {
      "epoch": 5.025291442871094e-05,
      "step": 52694,
      "training_step_time": 7.633289575576782
    },
    {
      "epoch": 5.0253868103027344e-05,
      "model_forward_time": 1.7291555404663086,
      "step": 52695
    },
    {
      "epoch": 5.0253868103027344e-05,
      "step": 52695,
      "training_step_time": 7.5501813888549805
    },
    {
      "epoch": 5.025482177734375e-05,
      "model_forward_time": 1.733445644378662,
      "step": 52696
    },
    {
      "epoch": 5.025482177734375e-05,
      "step": 52696,
      "training_step_time": 7.756442546844482
    },
    {
      "epoch": 5.025577545166016e-05,
      "model_forward_time": 1.7408640384674072,
      "step": 52697
    },
    {
      "epoch": 5.025577545166016e-05,
      "step": 52697,
      "training_step_time": 6.752044677734375
    },
    {
      "epoch": 5.025672912597656e-05,
      "model_forward_time": 1.7267353534698486,
      "step": 52698
    },
    {
      "epoch": 5.025672912597656e-05,
      "step": 52698,
      "training_step_time": 8.44558072090149
    },
    {
      "epoch": 5.025768280029297e-05,
      "model_forward_time": 2.2757644653320312,
      "step": 52699
    },
    {
      "epoch": 5.025768280029297e-05,
      "step": 52699,
      "training_step_time": 8.304829597473145
    },
    {
      "epoch": 5.025863647460938e-05,
      "grad_norm": 0.1184503510594368,
      "learning_rate": 4.967096188123943e-06,
      "loss": 0.0557,
      "step": 52700
    },
    {
      "epoch": 5.025863647460938e-05,
      "model_forward_time": 1.72676682472229,
      "step": 52700
    },
    {
      "dynamics_loss_avg": 0.037340473011136056,
      "epoch": 5.025863647460938e-05,
      "step": 52700
    },
    {
      "action_loss_avg": 0.0014821926190052182,
      "epoch": 5.025863647460938e-05,
      "step": 52700
    },
    {
      "epoch": 5.025863647460938e-05,
      "step": 52700,
      "training_step_time": 8.045212745666504
    },
    {
      "epoch": 5.025959014892578e-05,
      "model_forward_time": 1.7280852794647217,
      "step": 52701
    },
    {
      "epoch": 5.025959014892578e-05,
      "step": 52701,
      "training_step_time": 9.157166719436646
    },
    {
      "epoch": 5.0260543823242186e-05,
      "model_forward_time": 1.7223975658416748,
      "step": 52702
    },
    {
      "epoch": 5.0260543823242186e-05,
      "step": 52702,
      "training_step_time": 8.725359201431274
    },
    {
      "epoch": 5.0261497497558597e-05,
      "model_forward_time": 1.728736162185669,
      "step": 52703
    },
    {
      "epoch": 5.0261497497558597e-05,
      "step": 52703,
      "training_step_time": 8.677491426467896
    },
    {
      "epoch": 5.0262451171875e-05,
      "model_forward_time": 1.7259666919708252,
      "step": 52704
    },
    {
      "epoch": 5.0262451171875e-05,
      "step": 52704,
      "training_step_time": 7.667737245559692
    },
    {
      "epoch": 5.0263404846191405e-05,
      "model_forward_time": 1.7300426959991455,
      "step": 52705
    },
    {
      "epoch": 5.0263404846191405e-05,
      "step": 52705,
      "training_step_time": 7.745432376861572
    },
    {
      "epoch": 5.0264358520507815e-05,
      "model_forward_time": 1.7380740642547607,
      "step": 52706
    },
    {
      "epoch": 5.0264358520507815e-05,
      "step": 52706,
      "training_step_time": 7.371352910995483
    },
    {
      "epoch": 5.026531219482422e-05,
      "model_forward_time": 1.7286686897277832,
      "step": 52707
    },
    {
      "epoch": 5.026531219482422e-05,
      "step": 52707,
      "training_step_time": 7.415940761566162
    },
    {
      "epoch": 5.0266265869140623e-05,
      "model_forward_time": 2.305248498916626,
      "step": 52708
    },
    {
      "epoch": 5.0266265869140623e-05,
      "step": 52708,
      "training_step_time": 7.434033393859863
    },
    {
      "epoch": 5.0267219543457034e-05,
      "model_forward_time": 1.7247939109802246,
      "step": 52709
    },
    {
      "epoch": 5.0267219543457034e-05,
      "step": 52709,
      "training_step_time": 7.609643459320068
    },
    {
      "epoch": 5.026817321777344e-05,
      "grad_norm": 0.11074762046337128,
      "learning_rate": 4.9654427559388845e-06,
      "loss": 0.0553,
      "step": 52710
    },
    {
      "epoch": 5.026817321777344e-05,
      "model_forward_time": 1.722841501235962,
      "step": 52710
    },
    {
      "dynamics_loss_avg": 0.049244361370801924,
      "epoch": 5.026817321777344e-05,
      "step": 52710
    },
    {
      "action_loss_avg": 0.0015362412319518625,
      "epoch": 5.026817321777344e-05,
      "step": 52710
    },
    {
      "epoch": 5.026817321777344e-05,
      "step": 52710,
      "training_step_time": 8.0546875
    },
    {
      "epoch": 5.026912689208984e-05,
      "model_forward_time": 1.7205541133880615,
      "step": 52711
    },
    {
      "epoch": 5.026912689208984e-05,
      "step": 52711,
      "training_step_time": 7.693576097488403
    },
    {
      "epoch": 5.027008056640625e-05,
      "model_forward_time": 1.722285270690918,
      "step": 52712
    },
    {
      "epoch": 5.027008056640625e-05,
      "step": 52712,
      "training_step_time": 7.586138010025024
    },
    {
      "epoch": 5.027103424072266e-05,
      "model_forward_time": 1.7306029796600342,
      "step": 52713
    },
    {
      "epoch": 5.027103424072266e-05,
      "step": 52713,
      "training_step_time": 8.710295677185059
    },
    {
      "epoch": 5.027198791503906e-05,
      "model_forward_time": 1.720555305480957,
      "step": 52714
    },
    {
      "epoch": 5.027198791503906e-05,
      "step": 52714,
      "training_step_time": 10.454095840454102
    },
    {
      "epoch": 5.027294158935547e-05,
      "model_forward_time": 1.7215321063995361,
      "step": 52715
    },
    {
      "epoch": 5.027294158935547e-05,
      "step": 52715,
      "training_step_time": 7.368817567825317
    },
    {
      "epoch": 5.0273895263671876e-05,
      "model_forward_time": 1.7210514545440674,
      "step": 52716
    },
    {
      "epoch": 5.0273895263671876e-05,
      "step": 52716,
      "training_step_time": 7.328125238418579
    },
    {
      "epoch": 5.027484893798828e-05,
      "model_forward_time": 2.3312647342681885,
      "step": 52717
    },
    {
      "epoch": 5.027484893798828e-05,
      "step": 52717,
      "training_step_time": 7.330604553222656
    },
    {
      "epoch": 5.027580261230469e-05,
      "model_forward_time": 1.7275002002716064,
      "step": 52718
    },
    {
      "epoch": 5.027580261230469e-05,
      "step": 52718,
      "training_step_time": 7.508185386657715
    },
    {
      "epoch": 5.0276756286621095e-05,
      "model_forward_time": 1.7207036018371582,
      "step": 52719
    },
    {
      "epoch": 5.0276756286621095e-05,
      "step": 52719,
      "training_step_time": 7.469433307647705
    },
    {
      "epoch": 5.02777099609375e-05,
      "grad_norm": 0.12434542179107666,
      "learning_rate": 4.963789327532954e-06,
      "loss": 0.055,
      "step": 52720
    },
    {
      "epoch": 5.02777099609375e-05,
      "model_forward_time": 1.7124433517456055,
      "step": 52720
    },
    {
      "dynamics_loss_avg": 0.060784902051091196,
      "epoch": 5.02777099609375e-05,
      "step": 52720
    },
    {
      "action_loss_avg": 0.0015046318178065122,
      "epoch": 5.02777099609375e-05,
      "step": 52720
    },
    {
      "epoch": 5.02777099609375e-05,
      "step": 52720,
      "training_step_time": 8.823929786682129
    },
    {
      "epoch": 5.02786636352539e-05,
      "model_forward_time": 1.7213776111602783,
      "step": 52721
    },
    {
      "epoch": 5.02786636352539e-05,
      "step": 52721,
      "training_step_time": 8.721251249313354
    },
    {
      "epoch": 5.0279617309570314e-05,
      "model_forward_time": 1.7194342613220215,
      "step": 52722
    },
    {
      "epoch": 5.0279617309570314e-05,
      "step": 52722,
      "training_step_time": 8.40486741065979
    },
    {
      "epoch": 5.028057098388672e-05,
      "model_forward_time": 1.3391673564910889,
      "step": 52723
    },
    {
      "epoch": 5.028057098388672e-05,
      "step": 52723,
      "training_step_time": 7.576627969741821
    },
    {
      "epoch": 5.028152465820312e-05,
      "model_forward_time": 1.7234864234924316,
      "step": 52724
    },
    {
      "epoch": 5.028152465820312e-05,
      "step": 52724,
      "training_step_time": 7.340921401977539
    },
    {
      "epoch": 5.028247833251953e-05,
      "model_forward_time": 1.7031309604644775,
      "step": 52725
    },
    {
      "epoch": 5.028247833251953e-05,
      "step": 52725,
      "training_step_time": 7.33837628364563
    },
    {
      "epoch": 5.028343200683594e-05,
      "model_forward_time": 2.3458058834075928,
      "step": 52726
    },
    {
      "epoch": 5.028343200683594e-05,
      "step": 52726,
      "training_step_time": 7.170706510543823
    },
    {
      "epoch": 5.028438568115234e-05,
      "model_forward_time": 1.337038516998291,
      "step": 52727
    },
    {
      "epoch": 5.028438568115234e-05,
      "step": 52727,
      "training_step_time": 7.368832111358643
    },
    {
      "epoch": 5.028533935546875e-05,
      "model_forward_time": 1.7322328090667725,
      "step": 52728
    },
    {
      "epoch": 5.028533935546875e-05,
      "step": 52728,
      "training_step_time": 7.584565162658691
    },
    {
      "epoch": 5.0286293029785155e-05,
      "model_forward_time": 1.7284278869628906,
      "step": 52729
    },
    {
      "epoch": 5.0286293029785155e-05,
      "step": 52729,
      "training_step_time": 7.787554740905762
    },
    {
      "epoch": 5.028724670410156e-05,
      "grad_norm": 0.12654633820056915,
      "learning_rate": 4.962135903086968e-06,
      "loss": 0.0524,
      "step": 52730
    },
    {
      "epoch": 5.028724670410156e-05,
      "model_forward_time": 1.7127079963684082,
      "step": 52730
    },
    {
      "dynamics_loss_avg": 0.05015672016888857,
      "epoch": 5.028724670410156e-05,
      "step": 52730
    },
    {
      "action_loss_avg": 0.0018337926303502172,
      "epoch": 5.028724670410156e-05,
      "step": 52730
    },
    {
      "epoch": 5.028724670410156e-05,
      "step": 52730,
      "training_step_time": 7.599420070648193
    },
    {
      "epoch": 5.028820037841797e-05,
      "model_forward_time": 1.7240197658538818,
      "step": 52731
    },
    {
      "epoch": 5.028820037841797e-05,
      "step": 52731,
      "training_step_time": 7.672420263290405
    },
    {
      "epoch": 5.0289154052734374e-05,
      "model_forward_time": 1.7255232334136963,
      "step": 52732
    },
    {
      "epoch": 5.0289154052734374e-05,
      "step": 52732,
      "training_step_time": 9.601951360702515
    },
    {
      "epoch": 5.029010772705078e-05,
      "model_forward_time": 1.7235381603240967,
      "step": 52733
    },
    {
      "epoch": 5.029010772705078e-05,
      "step": 52733,
      "training_step_time": 7.474799156188965
    },
    {
      "epoch": 5.029106140136719e-05,
      "model_forward_time": 1.7226459980010986,
      "step": 52734
    },
    {
      "epoch": 5.029106140136719e-05,
      "step": 52734,
      "training_step_time": 8.238399744033813
    },
    {
      "epoch": 5.029201507568359e-05,
      "model_forward_time": 2.3259835243225098,
      "step": 52735
    },
    {
      "epoch": 5.029201507568359e-05,
      "step": 52735,
      "training_step_time": 7.219140291213989
    },
    {
      "epoch": 5.029296875e-05,
      "model_forward_time": 1.7204201221466064,
      "step": 52736
    },
    {
      "epoch": 5.029296875e-05,
      "step": 52736,
      "training_step_time": 7.961782932281494
    },
    {
      "epoch": 5.029392242431641e-05,
      "model_forward_time": 1.7205772399902344,
      "step": 52737
    },
    {
      "epoch": 5.029392242431641e-05,
      "step": 52737,
      "training_step_time": 8.438500881195068
    },
    {
      "epoch": 5.029487609863281e-05,
      "model_forward_time": 1.7238266468048096,
      "step": 52738
    },
    {
      "epoch": 5.029487609863281e-05,
      "step": 52738,
      "training_step_time": 10.056742906570435
    },
    {
      "epoch": 5.0295829772949216e-05,
      "model_forward_time": 1.7314157485961914,
      "step": 52739
    },
    {
      "epoch": 5.0295829772949216e-05,
      "step": 52739,
      "training_step_time": 7.552999973297119
    },
    {
      "epoch": 5.029678344726563e-05,
      "grad_norm": 0.13199369609355927,
      "learning_rate": 4.960482482781741e-06,
      "loss": 0.0564,
      "step": 52740
    },
    {
      "epoch": 5.029678344726563e-05,
      "model_forward_time": 1.7194461822509766,
      "step": 52740
    },
    {
      "dynamics_loss_avg": 0.0596026448532939,
      "epoch": 5.029678344726563e-05,
      "step": 52740
    },
    {
      "action_loss_avg": 0.001768342690775171,
      "epoch": 5.029678344726563e-05,
      "step": 52740
    },
    {
      "epoch": 5.029678344726563e-05,
      "step": 52740,
      "training_step_time": 7.660177946090698
    },
    {
      "epoch": 5.029773712158203e-05,
      "model_forward_time": 1.7245681285858154,
      "step": 52741
    },
    {
      "epoch": 5.029773712158203e-05,
      "step": 52741,
      "training_step_time": 9.330632209777832
    },
    {
      "epoch": 5.0298690795898435e-05,
      "model_forward_time": 1.7334506511688232,
      "step": 52742
    },
    {
      "epoch": 5.0298690795898435e-05,
      "step": 52742,
      "training_step_time": 7.567710876464844
    },
    {
      "epoch": 5.0299644470214846e-05,
      "model_forward_time": 1.7281737327575684,
      "step": 52743
    },
    {
      "epoch": 5.0299644470214846e-05,
      "step": 52743,
      "training_step_time": 7.386315584182739
    },
    {
      "epoch": 5.030059814453125e-05,
      "model_forward_time": 2.3304507732391357,
      "step": 52744
    },
    {
      "epoch": 5.030059814453125e-05,
      "step": 52744,
      "training_step_time": 7.225835800170898
    },
    {
      "epoch": 5.0301551818847654e-05,
      "model_forward_time": 1.7366759777069092,
      "step": 52745
    },
    {
      "epoch": 5.0301551818847654e-05,
      "step": 52745,
      "training_step_time": 7.480378866195679
    },
    {
      "epoch": 5.0302505493164065e-05,
      "model_forward_time": 1.7197318077087402,
      "step": 52746
    },
    {
      "epoch": 5.0302505493164065e-05,
      "step": 52746,
      "training_step_time": 8.269742965698242
    },
    {
      "epoch": 5.030345916748047e-05,
      "model_forward_time": 1.7215697765350342,
      "step": 52747
    },
    {
      "epoch": 5.030345916748047e-05,
      "step": 52747,
      "training_step_time": 7.833606243133545
    },
    {
      "epoch": 5.030441284179687e-05,
      "model_forward_time": 1.7261500358581543,
      "step": 52748
    },
    {
      "epoch": 5.030441284179687e-05,
      "step": 52748,
      "training_step_time": 7.573236465454102
    },
    {
      "epoch": 5.0305366516113283e-05,
      "model_forward_time": 1.718418836593628,
      "step": 52749
    },
    {
      "epoch": 5.0305366516113283e-05,
      "step": 52749,
      "training_step_time": 7.611573934555054
    },
    {
      "epoch": 5.030632019042969e-05,
      "grad_norm": 0.08809414505958557,
      "learning_rate": 4.958829066798092e-06,
      "loss": 0.057,
      "step": 52750
    },
    {
      "epoch": 5.030632019042969e-05,
      "model_forward_time": 1.7137293815612793,
      "step": 52750
    },
    {
      "dynamics_loss_avg": 0.05232125706970692,
      "epoch": 5.030632019042969e-05,
      "step": 52750
    },
    {
      "action_loss_avg": 0.0011540235194843263,
      "epoch": 5.030632019042969e-05,
      "step": 52750
    },
    {
      "epoch": 5.030632019042969e-05,
      "step": 52750,
      "training_step_time": 7.531586647033691
    },
    {
      "epoch": 5.030727386474609e-05,
      "model_forward_time": 1.7315049171447754,
      "step": 52751
    },
    {
      "epoch": 5.030727386474609e-05,
      "step": 52751,
      "training_step_time": 7.514834880828857
    },
    {
      "epoch": 5.03082275390625e-05,
      "model_forward_time": 1.696260690689087,
      "step": 52752
    },
    {
      "epoch": 5.03082275390625e-05,
      "step": 52752,
      "training_step_time": 7.685861587524414
    },
    {
      "epoch": 5.0309181213378906e-05,
      "model_forward_time": 2.3073599338531494,
      "step": 52753
    },
    {
      "epoch": 5.0309181213378906e-05,
      "step": 52753,
      "training_step_time": 7.385858535766602
    },
    {
      "epoch": 5.031013488769531e-05,
      "model_forward_time": 1.7401869297027588,
      "step": 52754
    },
    {
      "epoch": 5.031013488769531e-05,
      "step": 52754,
      "training_step_time": 7.46674370765686
    },
    {
      "epoch": 5.031108856201172e-05,
      "model_forward_time": 1.7098889350891113,
      "step": 52755
    },
    {
      "epoch": 5.031108856201172e-05,
      "step": 52755,
      "training_step_time": 8.414159059524536
    },
    {
      "epoch": 5.0312042236328125e-05,
      "model_forward_time": 1.7123844623565674,
      "step": 52756
    },
    {
      "epoch": 5.0312042236328125e-05,
      "step": 52756,
      "training_step_time": 7.85633659362793
    },
    {
      "epoch": 5.031299591064453e-05,
      "model_forward_time": 1.725797176361084,
      "step": 52757
    },
    {
      "epoch": 5.031299591064453e-05,
      "step": 52757,
      "training_step_time": 7.527129888534546
    },
    {
      "epoch": 5.031394958496094e-05,
      "model_forward_time": 1.71431303024292,
      "step": 52758
    },
    {
      "epoch": 5.031394958496094e-05,
      "step": 52758,
      "training_step_time": 8.113178491592407
    },
    {
      "epoch": 5.0314903259277344e-05,
      "model_forward_time": 1.7080094814300537,
      "step": 52759
    },
    {
      "epoch": 5.0314903259277344e-05,
      "step": 52759,
      "training_step_time": 8.430619239807129
    },
    {
      "epoch": 5.031585693359375e-05,
      "grad_norm": 0.08812527358531952,
      "learning_rate": 4.957175655316835e-06,
      "loss": 0.0605,
      "step": 52760
    },
    {
      "epoch": 5.031585693359375e-05,
      "model_forward_time": 1.704155445098877,
      "step": 52760
    },
    {
      "dynamics_loss_avg": 0.041396180633455515,
      "epoch": 5.031585693359375e-05,
      "step": 52760
    },
    {
      "action_loss_avg": 0.0014158920996123925,
      "epoch": 5.031585693359375e-05,
      "step": 52760
    },
    {
      "epoch": 5.031585693359375e-05,
      "step": 52760,
      "training_step_time": 7.4021501541137695
    },
    {
      "epoch": 5.031681060791016e-05,
      "model_forward_time": 1.711000919342041,
      "step": 52761
    },
    {
      "epoch": 5.031681060791016e-05,
      "step": 52761,
      "training_step_time": 7.678074359893799
    },
    {
      "epoch": 5.031776428222656e-05,
      "model_forward_time": 1.712047815322876,
      "step": 52762
    },
    {
      "epoch": 5.031776428222656e-05,
      "step": 52762,
      "training_step_time": 7.334099054336548
    },
    {
      "epoch": 5.031871795654297e-05,
      "model_forward_time": 2.3064346313476562,
      "step": 52763
    },
    {
      "epoch": 5.031871795654297e-05,
      "step": 52763,
      "training_step_time": 7.356098175048828
    },
    {
      "epoch": 5.031967163085938e-05,
      "model_forward_time": 1.709822416305542,
      "step": 52764
    },
    {
      "epoch": 5.031967163085938e-05,
      "step": 52764,
      "training_step_time": 8.376444339752197
    },
    {
      "epoch": 5.032062530517578e-05,
      "model_forward_time": 1.7147889137268066,
      "step": 52765
    },
    {
      "epoch": 5.032062530517578e-05,
      "step": 52765,
      "training_step_time": 7.900748014450073
    },
    {
      "epoch": 5.0321578979492186e-05,
      "model_forward_time": 1.7147297859191895,
      "step": 52766
    },
    {
      "epoch": 5.0321578979492186e-05,
      "step": 52766,
      "training_step_time": 8.562163591384888
    },
    {
      "epoch": 5.03225326538086e-05,
      "model_forward_time": 1.7152750492095947,
      "step": 52767
    },
    {
      "epoch": 5.03225326538086e-05,
      "step": 52767,
      "training_step_time": 9.25598430633545
    },
    {
      "epoch": 5.0323486328125e-05,
      "model_forward_time": 1.71553635597229,
      "step": 52768
    },
    {
      "epoch": 5.0323486328125e-05,
      "step": 52768,
      "training_step_time": 8.181236028671265
    },
    {
      "epoch": 5.0324440002441405e-05,
      "model_forward_time": 1.6898608207702637,
      "step": 52769
    },
    {
      "epoch": 5.0324440002441405e-05,
      "step": 52769,
      "training_step_time": 7.780633211135864
    },
    {
      "epoch": 5.0325393676757815e-05,
      "grad_norm": 0.11721646040678024,
      "learning_rate": 4.955522248518784e-06,
      "loss": 0.06,
      "step": 52770
    },
    {
      "epoch": 5.0325393676757815e-05,
      "model_forward_time": 1.7061278820037842,
      "step": 52770
    },
    {
      "dynamics_loss_avg": 0.04408329036086798,
      "epoch": 5.0325393676757815e-05,
      "step": 52770
    },
    {
      "action_loss_avg": 0.0010327273863367735,
      "epoch": 5.0325393676757815e-05,
      "step": 52770
    },
    {
      "epoch": 5.0325393676757815e-05,
      "step": 52770,
      "training_step_time": 7.710405111312866
    },
    {
      "epoch": 5.032634735107422e-05,
      "model_forward_time": 1.7124135494232178,
      "step": 52771
    },
    {
      "epoch": 5.032634735107422e-05,
      "step": 52771,
      "training_step_time": 7.395845890045166
    },
    {
      "epoch": 5.0327301025390624e-05,
      "model_forward_time": 2.2860982418060303,
      "step": 52772
    },
    {
      "epoch": 5.0327301025390624e-05,
      "step": 52772,
      "training_step_time": 8.204075813293457
    },
    {
      "epoch": 5.0328254699707034e-05,
      "model_forward_time": 1.726440668106079,
      "step": 52773
    },
    {
      "epoch": 5.0328254699707034e-05,
      "step": 52773,
      "training_step_time": 8.953384637832642
    },
    {
      "epoch": 5.032920837402344e-05,
      "model_forward_time": 1.713388204574585,
      "step": 52774
    },
    {
      "epoch": 5.032920837402344e-05,
      "step": 52774,
      "training_step_time": 8.108488321304321
    },
    {
      "epoch": 5.033016204833984e-05,
      "model_forward_time": 1.7115356922149658,
      "step": 52775
    },
    {
      "epoch": 5.033016204833984e-05,
      "step": 52775,
      "training_step_time": 7.90475869178772
    },
    {
      "epoch": 5.033111572265625e-05,
      "model_forward_time": 1.700500726699829,
      "step": 52776
    },
    {
      "epoch": 5.033111572265625e-05,
      "step": 52776,
      "training_step_time": 7.9819722175598145
    },
    {
      "epoch": 5.033206939697266e-05,
      "model_forward_time": 1.7046432495117188,
      "step": 52777
    },
    {
      "epoch": 5.033206939697266e-05,
      "step": 52777,
      "training_step_time": 8.766120195388794
    },
    {
      "epoch": 5.033302307128906e-05,
      "model_forward_time": 1.7167418003082275,
      "step": 52778
    },
    {
      "epoch": 5.033302307128906e-05,
      "step": 52778,
      "training_step_time": 7.696513891220093
    },
    {
      "epoch": 5.033397674560547e-05,
      "model_forward_time": 1.716763973236084,
      "step": 52779
    },
    {
      "epoch": 5.033397674560547e-05,
      "step": 52779,
      "training_step_time": 9.11967658996582
    },
    {
      "epoch": 5.0334930419921876e-05,
      "grad_norm": 0.1393107920885086,
      "learning_rate": 4.9538688465847526e-06,
      "loss": 0.0607,
      "step": 52780
    },
    {
      "epoch": 5.0334930419921876e-05,
      "model_forward_time": 1.7100188732147217,
      "step": 52780
    },
    {
      "dynamics_loss_avg": 0.056659061647951606,
      "epoch": 5.0334930419921876e-05,
      "step": 52780
    },
    {
      "action_loss_avg": 0.0019813470949884503,
      "epoch": 5.0334930419921876e-05,
      "step": 52780
    },
    {
      "epoch": 5.0334930419921876e-05,
      "step": 52780,
      "training_step_time": 7.672422170639038
    },
    {
      "epoch": 5.033588409423828e-05,
      "model_forward_time": 2.2609405517578125,
      "step": 52781
    },
    {
      "epoch": 5.033588409423828e-05,
      "step": 52781,
      "training_step_time": 7.120614051818848
    },
    {
      "epoch": 5.0336837768554684e-05,
      "model_forward_time": 1.7580091953277588,
      "step": 52782
    },
    {
      "epoch": 5.0336837768554684e-05,
      "step": 52782,
      "training_step_time": 7.551289796829224
    },
    {
      "epoch": 5.0337791442871095e-05,
      "model_forward_time": 1.7132954597473145,
      "step": 52783
    },
    {
      "epoch": 5.0337791442871095e-05,
      "step": 52783,
      "training_step_time": 7.6631669998168945
    },
    {
      "epoch": 5.03387451171875e-05,
      "model_forward_time": 1.716702938079834,
      "step": 52784
    },
    {
      "epoch": 5.03387451171875e-05,
      "step": 52784,
      "training_step_time": 8.991408586502075
    },
    {
      "epoch": 5.03396987915039e-05,
      "model_forward_time": 1.7122650146484375,
      "step": 52785
    },
    {
      "epoch": 5.03396987915039e-05,
      "step": 52785,
      "training_step_time": 7.613842964172363
    },
    {
      "epoch": 5.0340652465820314e-05,
      "model_forward_time": 1.721717119216919,
      "step": 52786
    },
    {
      "epoch": 5.0340652465820314e-05,
      "step": 52786,
      "training_step_time": 8.33035922050476
    },
    {
      "epoch": 5.034160614013672e-05,
      "model_forward_time": 1.704483985900879,
      "step": 52787
    },
    {
      "epoch": 5.034160614013672e-05,
      "step": 52787,
      "training_step_time": 7.776457071304321
    },
    {
      "epoch": 5.034255981445312e-05,
      "model_forward_time": 1.7082390785217285,
      "step": 52788
    },
    {
      "epoch": 5.034255981445312e-05,
      "step": 52788,
      "training_step_time": 7.692590713500977
    },
    {
      "epoch": 5.034351348876953e-05,
      "model_forward_time": 1.721691608428955,
      "step": 52789
    },
    {
      "epoch": 5.034351348876953e-05,
      "step": 52789,
      "training_step_time": 7.711999893188477
    },
    {
      "epoch": 5.034446716308594e-05,
      "grad_norm": 0.10776230692863464,
      "learning_rate": 4.952215449695554e-06,
      "loss": 0.06,
      "step": 52790
    },
    {
      "epoch": 5.034446716308594e-05,
      "model_forward_time": 1.8557896614074707,
      "step": 52790
    },
    {
      "dynamics_loss_avg": 0.060396565683186056,
      "epoch": 5.034446716308594e-05,
      "step": 52790
    },
    {
      "action_loss_avg": 0.001362315655569546,
      "epoch": 5.034446716308594e-05,
      "step": 52790
    },
    {
      "epoch": 5.034446716308594e-05,
      "step": 52790,
      "training_step_time": 7.471387147903442
    },
    {
      "epoch": 5.034542083740234e-05,
      "model_forward_time": 1.7559168338775635,
      "step": 52791
    },
    {
      "epoch": 5.034542083740234e-05,
      "step": 52791,
      "training_step_time": 7.5512309074401855
    },
    {
      "epoch": 5.034637451171875e-05,
      "model_forward_time": 1.7269353866577148,
      "step": 52792
    },
    {
      "epoch": 5.034637451171875e-05,
      "step": 52792,
      "training_step_time": 7.698330402374268
    },
    {
      "epoch": 5.0347328186035156e-05,
      "model_forward_time": 1.7154541015625,
      "step": 52793
    },
    {
      "epoch": 5.0347328186035156e-05,
      "step": 52793,
      "training_step_time": 9.266858577728271
    },
    {
      "epoch": 5.034828186035156e-05,
      "model_forward_time": 1.7231369018554688,
      "step": 52794
    },
    {
      "epoch": 5.034828186035156e-05,
      "step": 52794,
      "training_step_time": 9.484071254730225
    },
    {
      "epoch": 5.034923553466797e-05,
      "model_forward_time": 1.7107124328613281,
      "step": 52795
    },
    {
      "epoch": 5.034923553466797e-05,
      "step": 52795,
      "training_step_time": 9.167846441268921
    },
    {
      "epoch": 5.0350189208984374e-05,
      "model_forward_time": 1.7152094841003418,
      "step": 52796
    },
    {
      "epoch": 5.0350189208984374e-05,
      "step": 52796,
      "training_step_time": 8.2673978805542
    },
    {
      "epoch": 5.035114288330078e-05,
      "model_forward_time": 1.7128868103027344,
      "step": 52797
    },
    {
      "epoch": 5.035114288330078e-05,
      "step": 52797,
      "training_step_time": 8.53059697151184
    },
    {
      "epoch": 5.035209655761719e-05,
      "model_forward_time": 1.7204816341400146,
      "step": 52798
    },
    {
      "epoch": 5.035209655761719e-05,
      "step": 52798,
      "training_step_time": 7.7874908447265625
    },
    {
      "epoch": 5.035305023193359e-05,
      "model_forward_time": 2.241947650909424,
      "step": 52799
    },
    {
      "epoch": 5.035305023193359e-05,
      "step": 52799,
      "training_step_time": 7.4601452350616455
    },
    {
      "epoch": 5.035400390625e-05,
      "grad_norm": 0.09212855249643326,
      "learning_rate": 4.950562058032002e-06,
      "loss": 0.06,
      "step": 52800
    },
    {
      "epoch": 5.035400390625e-05,
      "model_forward_time": 1.7373161315917969,
      "step": 52800
    },
    {
      "dynamics_loss_avg": 0.05331417080014944,
      "epoch": 5.035400390625e-05,
      "step": 52800
    },
    {
      "action_loss_avg": 0.001467065088218078,
      "epoch": 5.035400390625e-05,
      "step": 52800
    },
    {
      "epoch": 5.035400390625e-05,
      "step": 52800,
      "training_step_time": 8.903695344924927
    },
    {
      "epoch": 5.035495758056641e-05,
      "model_forward_time": 1.6966979503631592,
      "step": 52801
    },
    {
      "epoch": 5.035495758056641e-05,
      "step": 52801,
      "training_step_time": 7.627939224243164
    },
    {
      "epoch": 5.035591125488281e-05,
      "model_forward_time": 1.7137625217437744,
      "step": 52802
    },
    {
      "epoch": 5.035591125488281e-05,
      "step": 52802,
      "training_step_time": 8.593626976013184
    },
    {
      "epoch": 5.0356864929199216e-05,
      "model_forward_time": 1.717623233795166,
      "step": 52803
    },
    {
      "epoch": 5.0356864929199216e-05,
      "step": 52803,
      "training_step_time": 8.724491357803345
    },
    {
      "epoch": 5.035781860351563e-05,
      "model_forward_time": 1.715853214263916,
      "step": 52804
    },
    {
      "epoch": 5.035781860351563e-05,
      "step": 52804,
      "training_step_time": 9.313810110092163
    },
    {
      "epoch": 5.035877227783203e-05,
      "model_forward_time": 1.7140491008758545,
      "step": 52805
    },
    {
      "epoch": 5.035877227783203e-05,
      "step": 52805,
      "training_step_time": 7.764868259429932
    },
    {
      "epoch": 5.0359725952148435e-05,
      "model_forward_time": 1.7108805179595947,
      "step": 52806
    },
    {
      "epoch": 5.0359725952148435e-05,
      "step": 52806,
      "training_step_time": 7.844987392425537
    },
    {
      "epoch": 5.0360679626464846e-05,
      "model_forward_time": 1.7184467315673828,
      "step": 52807
    },
    {
      "epoch": 5.0360679626464846e-05,
      "step": 52807,
      "training_step_time": 7.956708669662476
    },
    {
      "epoch": 5.036163330078125e-05,
      "model_forward_time": 1.6925222873687744,
      "step": 52808
    },
    {
      "epoch": 5.036163330078125e-05,
      "step": 52808,
      "training_step_time": 7.426788330078125
    },
    {
      "epoch": 5.0362586975097654e-05,
      "model_forward_time": 2.293846607208252,
      "step": 52809
    },
    {
      "epoch": 5.0362586975097654e-05,
      "step": 52809,
      "training_step_time": 7.55562424659729
    },
    {
      "epoch": 5.0363540649414065e-05,
      "grad_norm": 0.13171181082725525,
      "learning_rate": 4.948908671774909e-06,
      "loss": 0.0572,
      "step": 52810
    },
    {
      "epoch": 5.0363540649414065e-05,
      "model_forward_time": 1.7009761333465576,
      "step": 52810
    },
    {
      "dynamics_loss_avg": 0.057497231382876636,
      "epoch": 5.0363540649414065e-05,
      "step": 52810
    },
    {
      "action_loss_avg": 0.0017577481048647314,
      "epoch": 5.0363540649414065e-05,
      "step": 52810
    },
    {
      "epoch": 5.0363540649414065e-05,
      "step": 52810,
      "training_step_time": 7.638537406921387
    },
    {
      "epoch": 5.036449432373047e-05,
      "model_forward_time": 1.716951847076416,
      "step": 52811
    },
    {
      "epoch": 5.036449432373047e-05,
      "step": 52811,
      "training_step_time": 8.522665977478027
    },
    {
      "epoch": 5.036544799804687e-05,
      "model_forward_time": 1.7115168571472168,
      "step": 52812
    },
    {
      "epoch": 5.036544799804687e-05,
      "step": 52812,
      "training_step_time": 9.25416374206543
    },
    {
      "epoch": 5.0366401672363284e-05,
      "model_forward_time": 1.7062206268310547,
      "step": 52813
    },
    {
      "epoch": 5.0366401672363284e-05,
      "step": 52813,
      "training_step_time": 7.74707293510437
    },
    {
      "epoch": 5.036735534667969e-05,
      "model_forward_time": 1.7079002857208252,
      "step": 52814
    },
    {
      "epoch": 5.036735534667969e-05,
      "step": 52814,
      "training_step_time": 8.514601707458496
    },
    {
      "epoch": 5.036830902099609e-05,
      "model_forward_time": 1.7233567237854004,
      "step": 52815
    },
    {
      "epoch": 5.036830902099609e-05,
      "step": 52815,
      "training_step_time": 8.875414848327637
    },
    {
      "epoch": 5.03692626953125e-05,
      "model_forward_time": 1.7080872058868408,
      "step": 52816
    },
    {
      "epoch": 5.03692626953125e-05,
      "step": 52816,
      "training_step_time": 7.850417852401733
    },
    {
      "epoch": 5.0370216369628906e-05,
      "model_forward_time": 1.711902141571045,
      "step": 52817
    },
    {
      "epoch": 5.0370216369628906e-05,
      "step": 52817,
      "training_step_time": 7.539949655532837
    },
    {
      "epoch": 5.037117004394531e-05,
      "model_forward_time": 2.2978172302246094,
      "step": 52818
    },
    {
      "epoch": 5.037117004394531e-05,
      "step": 52818,
      "training_step_time": 7.3917529582977295
    },
    {
      "epoch": 5.037212371826172e-05,
      "model_forward_time": 1.7132577896118164,
      "step": 52819
    },
    {
      "epoch": 5.037212371826172e-05,
      "step": 52819,
      "training_step_time": 7.719780445098877
    },
    {
      "epoch": 5.0373077392578125e-05,
      "grad_norm": 0.08541733026504517,
      "learning_rate": 4.947255291105088e-06,
      "loss": 0.059,
      "step": 52820
    },
    {
      "epoch": 5.0373077392578125e-05,
      "model_forward_time": 1.702967882156372,
      "step": 52820
    },
    {
      "dynamics_loss_avg": 0.05593653954565525,
      "epoch": 5.0373077392578125e-05,
      "step": 52820
    },
    {
      "action_loss_avg": 0.0009076363581698388,
      "epoch": 5.0373077392578125e-05,
      "step": 52820
    },
    {
      "epoch": 5.0373077392578125e-05,
      "step": 52820,
      "training_step_time": 8.800495862960815
    },
    {
      "epoch": 5.037403106689453e-05,
      "model_forward_time": 1.711003303527832,
      "step": 52821
    },
    {
      "epoch": 5.037403106689453e-05,
      "step": 52821,
      "training_step_time": 7.663917303085327
    },
    {
      "epoch": 5.037498474121094e-05,
      "model_forward_time": 1.7052898406982422,
      "step": 52822
    },
    {
      "epoch": 5.037498474121094e-05,
      "step": 52822,
      "training_step_time": 7.778667211532593
    },
    {
      "epoch": 5.0375938415527344e-05,
      "model_forward_time": 1.7154390811920166,
      "step": 52823
    },
    {
      "epoch": 5.0375938415527344e-05,
      "step": 52823,
      "training_step_time": 8.347342729568481
    },
    {
      "epoch": 5.037689208984375e-05,
      "model_forward_time": 1.7053258419036865,
      "step": 52824
    },
    {
      "epoch": 5.037689208984375e-05,
      "step": 52824,
      "training_step_time": 8.689149379730225
    },
    {
      "epoch": 5.037784576416016e-05,
      "model_forward_time": 1.7136414051055908,
      "step": 52825
    },
    {
      "epoch": 5.037784576416016e-05,
      "step": 52825,
      "training_step_time": 8.825288534164429
    },
    {
      "epoch": 5.037879943847656e-05,
      "model_forward_time": 1.7113368511199951,
      "step": 52826
    },
    {
      "epoch": 5.037879943847656e-05,
      "step": 52826,
      "training_step_time": 7.543273448944092
    },
    {
      "epoch": 5.037975311279297e-05,
      "model_forward_time": 2.2674689292907715,
      "step": 52827
    },
    {
      "epoch": 5.037975311279297e-05,
      "step": 52827,
      "training_step_time": 8.51555609703064
    },
    {
      "epoch": 5.038070678710938e-05,
      "model_forward_time": 1.7189817428588867,
      "step": 52828
    },
    {
      "epoch": 5.038070678710938e-05,
      "step": 52828,
      "training_step_time": 7.669857978820801
    },
    {
      "epoch": 5.038166046142578e-05,
      "model_forward_time": 1.708709478378296,
      "step": 52829
    },
    {
      "epoch": 5.038166046142578e-05,
      "step": 52829,
      "training_step_time": 7.832130670547485
    },
    {
      "epoch": 5.0382614135742186e-05,
      "grad_norm": 0.11453618854284286,
      "learning_rate": 4.945601916203348e-06,
      "loss": 0.0578,
      "step": 52830
    },
    {
      "epoch": 5.0382614135742186e-05,
      "model_forward_time": 1.6978678703308105,
      "step": 52830
    },
    {
      "dynamics_loss_avg": 0.06441906206309796,
      "epoch": 5.0382614135742186e-05,
      "step": 52830
    },
    {
      "action_loss_avg": 0.001987414760515094,
      "epoch": 5.0382614135742186e-05,
      "step": 52830
    },
    {
      "epoch": 5.0382614135742186e-05,
      "step": 52830,
      "training_step_time": 8.357485055923462
    },
    {
      "epoch": 5.03835678100586e-05,
      "model_forward_time": 1.7089686393737793,
      "step": 52831
    },
    {
      "epoch": 5.03835678100586e-05,
      "step": 52831,
      "training_step_time": 7.818305969238281
    },
    {
      "epoch": 5.0384521484375e-05,
      "model_forward_time": 1.711341142654419,
      "step": 52832
    },
    {
      "epoch": 5.0384521484375e-05,
      "step": 52832,
      "training_step_time": 8.27110743522644
    },
    {
      "epoch": 5.0385475158691405e-05,
      "model_forward_time": 1.7068538665771484,
      "step": 52833
    },
    {
      "epoch": 5.0385475158691405e-05,
      "step": 52833,
      "training_step_time": 7.858226537704468
    },
    {
      "epoch": 5.0386428833007816e-05,
      "model_forward_time": 1.712024211883545,
      "step": 52834
    },
    {
      "epoch": 5.0386428833007816e-05,
      "step": 52834,
      "training_step_time": 7.918579339981079
    },
    {
      "epoch": 5.038738250732422e-05,
      "model_forward_time": 1.705336570739746,
      "step": 52835
    },
    {
      "epoch": 5.038738250732422e-05,
      "step": 52835,
      "training_step_time": 7.875364542007446
    },
    {
      "epoch": 5.0388336181640624e-05,
      "model_forward_time": 2.313242197036743,
      "step": 52836
    },
    {
      "epoch": 5.0388336181640624e-05,
      "step": 52836,
      "training_step_time": 7.574854612350464
    },
    {
      "epoch": 5.0389289855957034e-05,
      "model_forward_time": 1.7522284984588623,
      "step": 52837
    },
    {
      "epoch": 5.0389289855957034e-05,
      "step": 52837,
      "training_step_time": 7.6996910572052
    },
    {
      "epoch": 5.039024353027344e-05,
      "model_forward_time": 1.710494041442871,
      "step": 52838
    },
    {
      "epoch": 5.039024353027344e-05,
      "step": 52838,
      "training_step_time": 10.471936702728271
    },
    {
      "epoch": 5.039119720458984e-05,
      "model_forward_time": 1.7103393077850342,
      "step": 52839
    },
    {
      "epoch": 5.039119720458984e-05,
      "step": 52839,
      "training_step_time": 7.7000133991241455
    },
    {
      "epoch": 5.039215087890625e-05,
      "grad_norm": 0.09582837671041489,
      "learning_rate": 4.943948547250502e-06,
      "loss": 0.0567,
      "step": 52840
    },
    {
      "epoch": 5.039215087890625e-05,
      "model_forward_time": 1.6931850910186768,
      "step": 52840
    },
    {
      "dynamics_loss_avg": 0.05548765081912279,
      "epoch": 5.039215087890625e-05,
      "step": 52840
    },
    {
      "action_loss_avg": 0.0017438666400266811,
      "epoch": 5.039215087890625e-05,
      "step": 52840
    },
    {
      "epoch": 5.039215087890625e-05,
      "step": 52840,
      "training_step_time": 8.35403323173523
    },
    {
      "epoch": 5.039310455322266e-05,
      "model_forward_time": 1.7092828750610352,
      "step": 52841
    },
    {
      "epoch": 5.039310455322266e-05,
      "step": 52841,
      "training_step_time": 8.394587755203247
    },
    {
      "epoch": 5.039405822753906e-05,
      "model_forward_time": 1.7081668376922607,
      "step": 52842
    },
    {
      "epoch": 5.039405822753906e-05,
      "step": 52842,
      "training_step_time": 8.677900314331055
    },
    {
      "epoch": 5.039501190185547e-05,
      "model_forward_time": 1.7083919048309326,
      "step": 52843
    },
    {
      "epoch": 5.039501190185547e-05,
      "step": 52843,
      "training_step_time": 7.741215705871582
    },
    {
      "epoch": 5.0395965576171876e-05,
      "model_forward_time": 1.7064802646636963,
      "step": 52844
    },
    {
      "epoch": 5.0395965576171876e-05,
      "step": 52844,
      "training_step_time": 8.722655057907104
    },
    {
      "epoch": 5.039691925048828e-05,
      "model_forward_time": 2.268871307373047,
      "step": 52845
    },
    {
      "epoch": 5.039691925048828e-05,
      "step": 52845,
      "training_step_time": 7.490062713623047
    },
    {
      "epoch": 5.0397872924804684e-05,
      "model_forward_time": 1.7437856197357178,
      "step": 52846
    },
    {
      "epoch": 5.0397872924804684e-05,
      "step": 52846,
      "training_step_time": 7.698288679122925
    },
    {
      "epoch": 5.0398826599121095e-05,
      "model_forward_time": 1.704514503479004,
      "step": 52847
    },
    {
      "epoch": 5.0398826599121095e-05,
      "step": 52847,
      "training_step_time": 7.651519060134888
    },
    {
      "epoch": 5.03997802734375e-05,
      "model_forward_time": 1.7097094058990479,
      "step": 52848
    },
    {
      "epoch": 5.03997802734375e-05,
      "step": 52848,
      "training_step_time": 8.549014806747437
    },
    {
      "epoch": 5.04007339477539e-05,
      "model_forward_time": 1.7683451175689697,
      "step": 52849
    },
    {
      "epoch": 5.04007339477539e-05,
      "step": 52849,
      "training_step_time": 8.602442741394043
    },
    {
      "epoch": 5.0401687622070314e-05,
      "grad_norm": 0.11840125173330307,
      "learning_rate": 4.9422951844273565e-06,
      "loss": 0.0588,
      "step": 52850
    },
    {
      "epoch": 5.0401687622070314e-05,
      "model_forward_time": 1.7117931842803955,
      "step": 52850
    },
    {
      "dynamics_loss_avg": 0.04479398401454091,
      "epoch": 5.0401687622070314e-05,
      "step": 52850
    },
    {
      "action_loss_avg": 0.0012219373020343482,
      "epoch": 5.0401687622070314e-05,
      "step": 52850
    },
    {
      "epoch": 5.0401687622070314e-05,
      "step": 52850,
      "training_step_time": 7.815697193145752
    },
    {
      "epoch": 5.040264129638672e-05,
      "model_forward_time": 1.7000539302825928,
      "step": 52851
    },
    {
      "epoch": 5.040264129638672e-05,
      "step": 52851,
      "training_step_time": 8.266344547271729
    },
    {
      "epoch": 5.040359497070312e-05,
      "model_forward_time": 1.7072498798370361,
      "step": 52852
    },
    {
      "epoch": 5.040359497070312e-05,
      "step": 52852,
      "training_step_time": 8.704111099243164
    },
    {
      "epoch": 5.040454864501953e-05,
      "model_forward_time": 1.7069568634033203,
      "step": 52853
    },
    {
      "epoch": 5.040454864501953e-05,
      "step": 52853,
      "training_step_time": 7.880643844604492
    },
    {
      "epoch": 5.040550231933594e-05,
      "model_forward_time": 1.7054879665374756,
      "step": 52854
    },
    {
      "epoch": 5.040550231933594e-05,
      "step": 52854,
      "training_step_time": 8.111129999160767
    },
    {
      "epoch": 5.040645599365234e-05,
      "model_forward_time": 1.7061612606048584,
      "step": 52855
    },
    {
      "epoch": 5.040645599365234e-05,
      "step": 52855,
      "training_step_time": 9.25124192237854
    },
    {
      "epoch": 5.040740966796875e-05,
      "model_forward_time": 1.7080934047698975,
      "step": 52856
    },
    {
      "epoch": 5.040740966796875e-05,
      "step": 52856,
      "training_step_time": 8.526973485946655
    },
    {
      "epoch": 5.0408363342285156e-05,
      "model_forward_time": 1.7043302059173584,
      "step": 52857
    },
    {
      "epoch": 5.0408363342285156e-05,
      "step": 52857,
      "training_step_time": 10.412023544311523
    },
    {
      "epoch": 5.040931701660156e-05,
      "model_forward_time": 1.7088801860809326,
      "step": 52858
    },
    {
      "epoch": 5.040931701660156e-05,
      "step": 52858,
      "training_step_time": 7.916696786880493
    },
    {
      "epoch": 5.041027069091797e-05,
      "model_forward_time": 1.7090063095092773,
      "step": 52859
    },
    {
      "epoch": 5.041027069091797e-05,
      "step": 52859,
      "training_step_time": 9.575613021850586
    },
    {
      "epoch": 5.0411224365234374e-05,
      "grad_norm": 0.1054377406835556,
      "learning_rate": 4.9406418279147236e-06,
      "loss": 0.058,
      "step": 52860
    },
    {
      "epoch": 5.0411224365234374e-05,
      "model_forward_time": 1.6932682991027832,
      "step": 52860
    },
    {
      "dynamics_loss_avg": 0.059000276029109955,
      "epoch": 5.0411224365234374e-05,
      "step": 52860
    },
    {
      "action_loss_avg": 0.0016373073565773665,
      "epoch": 5.0411224365234374e-05,
      "step": 52860
    },
    {
      "epoch": 5.0411224365234374e-05,
      "step": 52860,
      "training_step_time": 8.29835319519043
    },
    {
      "epoch": 5.041217803955078e-05,
      "model_forward_time": 1.70322847366333,
      "step": 52861
    },
    {
      "epoch": 5.041217803955078e-05,
      "step": 52861,
      "training_step_time": 8.485261678695679
    },
    {
      "epoch": 5.041313171386719e-05,
      "model_forward_time": 1.7021574974060059,
      "step": 52862
    },
    {
      "epoch": 5.041313171386719e-05,
      "step": 52862,
      "training_step_time": 8.88273811340332
    },
    {
      "epoch": 5.041408538818359e-05,
      "model_forward_time": 1.7032740116119385,
      "step": 52863
    },
    {
      "epoch": 5.041408538818359e-05,
      "step": 52863,
      "training_step_time": 7.598424196243286
    },
    {
      "epoch": 5.04150390625e-05,
      "model_forward_time": 2.258183240890503,
      "step": 52864
    },
    {
      "epoch": 5.04150390625e-05,
      "step": 52864,
      "training_step_time": 7.682800769805908
    },
    {
      "epoch": 5.041599273681641e-05,
      "model_forward_time": 1.7502262592315674,
      "step": 52865
    },
    {
      "epoch": 5.041599273681641e-05,
      "step": 52865,
      "training_step_time": 7.698566913604736
    },
    {
      "epoch": 5.041694641113281e-05,
      "model_forward_time": 1.7058706283569336,
      "step": 52866
    },
    {
      "epoch": 5.041694641113281e-05,
      "step": 52866,
      "training_step_time": 8.381340503692627
    },
    {
      "epoch": 5.0417900085449216e-05,
      "model_forward_time": 1.7044601440429688,
      "step": 52867
    },
    {
      "epoch": 5.0417900085449216e-05,
      "step": 52867,
      "training_step_time": 8.803377151489258
    },
    {
      "epoch": 5.041885375976563e-05,
      "model_forward_time": 1.7026493549346924,
      "step": 52868
    },
    {
      "epoch": 5.041885375976563e-05,
      "step": 52868,
      "training_step_time": 9.164976119995117
    },
    {
      "epoch": 5.041980743408203e-05,
      "model_forward_time": 1.7045385837554932,
      "step": 52869
    },
    {
      "epoch": 5.041980743408203e-05,
      "step": 52869,
      "training_step_time": 7.777508497238159
    },
    {
      "epoch": 5.0420761108398435e-05,
      "grad_norm": 0.15057995915412903,
      "learning_rate": 4.9389884778934115e-06,
      "loss": 0.0581,
      "step": 52870
    },
    {
      "epoch": 5.0420761108398435e-05,
      "model_forward_time": 1.7082922458648682,
      "step": 52870
    },
    {
      "dynamics_loss_avg": 0.05155292097479105,
      "epoch": 5.0420761108398435e-05,
      "step": 52870
    },
    {
      "action_loss_avg": 0.002064358495408669,
      "epoch": 5.0420761108398435e-05,
      "step": 52870
    },
    {
      "epoch": 5.0420761108398435e-05,
      "step": 52870,
      "training_step_time": 7.716182470321655
    },
    {
      "epoch": 5.0421714782714846e-05,
      "model_forward_time": 1.6996614933013916,
      "step": 52871
    },
    {
      "epoch": 5.0421714782714846e-05,
      "step": 52871,
      "training_step_time": 7.80042290687561
    },
    {
      "epoch": 5.042266845703125e-05,
      "model_forward_time": 1.698427677154541,
      "step": 52872
    },
    {
      "epoch": 5.042266845703125e-05,
      "step": 52872,
      "training_step_time": 7.605559587478638
    },
    {
      "epoch": 5.0423622131347654e-05,
      "model_forward_time": 2.2701985836029053,
      "step": 52873
    },
    {
      "epoch": 5.0423622131347654e-05,
      "step": 52873,
      "training_step_time": 7.665290832519531
    },
    {
      "epoch": 5.0424575805664065e-05,
      "model_forward_time": 1.7467334270477295,
      "step": 52874
    },
    {
      "epoch": 5.0424575805664065e-05,
      "step": 52874,
      "training_step_time": 8.280960083007812
    },
    {
      "epoch": 5.042552947998047e-05,
      "model_forward_time": 1.7003979682922363,
      "step": 52875
    },
    {
      "epoch": 5.042552947998047e-05,
      "step": 52875,
      "training_step_time": 7.750789403915405
    },
    {
      "epoch": 5.042648315429687e-05,
      "model_forward_time": 1.7083861827850342,
      "step": 52876
    },
    {
      "epoch": 5.042648315429687e-05,
      "step": 52876,
      "training_step_time": 7.9886438846588135
    },
    {
      "epoch": 5.0427436828613284e-05,
      "model_forward_time": 1.7022809982299805,
      "step": 52877
    },
    {
      "epoch": 5.0427436828613284e-05,
      "step": 52877,
      "training_step_time": 9.256799936294556
    },
    {
      "epoch": 5.042839050292969e-05,
      "model_forward_time": 1.7136354446411133,
      "step": 52878
    },
    {
      "epoch": 5.042839050292969e-05,
      "step": 52878,
      "training_step_time": 8.884334087371826
    },
    {
      "epoch": 5.042934417724609e-05,
      "model_forward_time": 1.7005126476287842,
      "step": 52879
    },
    {
      "epoch": 5.042934417724609e-05,
      "step": 52879,
      "training_step_time": 7.896891117095947
    },
    {
      "epoch": 5.04302978515625e-05,
      "grad_norm": 0.09966332465410233,
      "learning_rate": 4.937335134544228e-06,
      "loss": 0.0557,
      "step": 52880
    },
    {
      "epoch": 5.04302978515625e-05,
      "model_forward_time": 1.7116053104400635,
      "step": 52880
    },
    {
      "dynamics_loss_avg": 0.043367363326251504,
      "epoch": 5.04302978515625e-05,
      "step": 52880
    },
    {
      "action_loss_avg": 0.001387450605398044,
      "epoch": 5.04302978515625e-05,
      "step": 52880
    },
    {
      "epoch": 5.04302978515625e-05,
      "step": 52880,
      "training_step_time": 8.26856017112732
    },
    {
      "epoch": 5.0431251525878906e-05,
      "model_forward_time": 1.7067952156066895,
      "step": 52881
    },
    {
      "epoch": 5.0431251525878906e-05,
      "step": 52881,
      "training_step_time": 7.801149845123291
    },
    {
      "epoch": 5.043220520019531e-05,
      "model_forward_time": 2.272306442260742,
      "step": 52882
    },
    {
      "epoch": 5.043220520019531e-05,
      "step": 52882,
      "training_step_time": 7.6463799476623535
    },
    {
      "epoch": 5.043315887451172e-05,
      "model_forward_time": 1.7482731342315674,
      "step": 52883
    },
    {
      "epoch": 5.043315887451172e-05,
      "step": 52883,
      "training_step_time": 8.203707218170166
    },
    {
      "epoch": 5.0434112548828125e-05,
      "model_forward_time": 1.692636251449585,
      "step": 52884
    },
    {
      "epoch": 5.0434112548828125e-05,
      "step": 52884,
      "training_step_time": 7.698094844818115
    },
    {
      "epoch": 5.043506622314453e-05,
      "model_forward_time": 1.706437349319458,
      "step": 52885
    },
    {
      "epoch": 5.043506622314453e-05,
      "step": 52885,
      "training_step_time": 8.98978328704834
    },
    {
      "epoch": 5.043601989746094e-05,
      "model_forward_time": 1.7062690258026123,
      "step": 52886
    },
    {
      "epoch": 5.043601989746094e-05,
      "step": 52886,
      "training_step_time": 7.803493499755859
    },
    {
      "epoch": 5.0436973571777344e-05,
      "model_forward_time": 1.7062506675720215,
      "step": 52887
    },
    {
      "epoch": 5.0436973571777344e-05,
      "step": 52887,
      "training_step_time": 9.12231183052063
    },
    {
      "epoch": 5.043792724609375e-05,
      "model_forward_time": 1.712822437286377,
      "step": 52888
    },
    {
      "epoch": 5.043792724609375e-05,
      "step": 52888,
      "training_step_time": 8.280647277832031
    },
    {
      "epoch": 5.043888092041016e-05,
      "model_forward_time": 1.7101759910583496,
      "step": 52889
    },
    {
      "epoch": 5.043888092041016e-05,
      "step": 52889,
      "training_step_time": 8.32297658920288
    },
    {
      "epoch": 5.043983459472656e-05,
      "grad_norm": 0.12025101482868195,
      "learning_rate": 4.93568179804798e-06,
      "loss": 0.0569,
      "step": 52890
    },
    {
      "epoch": 5.043983459472656e-05,
      "model_forward_time": 1.704019546508789,
      "step": 52890
    },
    {
      "dynamics_loss_avg": 0.056868623942136765,
      "epoch": 5.043983459472656e-05,
      "step": 52890
    },
    {
      "action_loss_avg": 0.0018381487025180832,
      "epoch": 5.043983459472656e-05,
      "step": 52890
    },
    {
      "epoch": 5.043983459472656e-05,
      "step": 52890,
      "training_step_time": 8.529799938201904
    },
    {
      "epoch": 5.044078826904297e-05,
      "model_forward_time": 2.2824625968933105,
      "step": 52891
    },
    {
      "epoch": 5.044078826904297e-05,
      "step": 52891,
      "training_step_time": 7.667508602142334
    },
    {
      "epoch": 5.044174194335938e-05,
      "model_forward_time": 1.7238543033599854,
      "step": 52892
    },
    {
      "epoch": 5.044174194335938e-05,
      "step": 52892,
      "training_step_time": 7.652925968170166
    },
    {
      "epoch": 5.044269561767578e-05,
      "model_forward_time": 1.7003822326660156,
      "step": 52893
    },
    {
      "epoch": 5.044269561767578e-05,
      "step": 52893,
      "training_step_time": 8.545312404632568
    },
    {
      "epoch": 5.0443649291992186e-05,
      "model_forward_time": 1.7029838562011719,
      "step": 52894
    },
    {
      "epoch": 5.0443649291992186e-05,
      "step": 52894,
      "training_step_time": 7.770663499832153
    },
    {
      "epoch": 5.04446029663086e-05,
      "model_forward_time": 1.705268383026123,
      "step": 52895
    },
    {
      "epoch": 5.04446029663086e-05,
      "step": 52895,
      "training_step_time": 7.769635915756226
    },
    {
      "epoch": 5.0445556640625e-05,
      "model_forward_time": 1.7046091556549072,
      "step": 52896
    },
    {
      "epoch": 5.0445556640625e-05,
      "step": 52896,
      "training_step_time": 9.547301769256592
    },
    {
      "epoch": 5.0446510314941405e-05,
      "model_forward_time": 1.707078218460083,
      "step": 52897
    },
    {
      "epoch": 5.0446510314941405e-05,
      "step": 52897,
      "training_step_time": 8.362435579299927
    },
    {
      "epoch": 5.0447463989257816e-05,
      "model_forward_time": 1.7049484252929688,
      "step": 52898
    },
    {
      "epoch": 5.0447463989257816e-05,
      "step": 52898,
      "training_step_time": 7.774685382843018
    },
    {
      "epoch": 5.044841766357422e-05,
      "model_forward_time": 1.7064411640167236,
      "step": 52899
    },
    {
      "epoch": 5.044841766357422e-05,
      "step": 52899,
      "training_step_time": 7.754980802536011
    },
    {
      "epoch": 5.0449371337890624e-05,
      "grad_norm": 0.13491970300674438,
      "learning_rate": 4.934028468585473e-06,
      "loss": 0.0595,
      "step": 52900
    },
    {
      "epoch": 5.0449371337890624e-05,
      "model_forward_time": 2.2745683193206787,
      "step": 52900
    },
    {
      "dynamics_loss_avg": 0.06118845492601395,
      "epoch": 5.0449371337890624e-05,
      "step": 52900
    },
    {
      "action_loss_avg": 0.0017560353153385222,
      "epoch": 5.0449371337890624e-05,
      "step": 52900
    },
    {
      "epoch": 5.0449371337890624e-05,
      "step": 52900,
      "training_step_time": 7.727276802062988
    },
    {
      "epoch": 5.0450325012207034e-05,
      "model_forward_time": 1.7081587314605713,
      "step": 52901
    },
    {
      "epoch": 5.0450325012207034e-05,
      "step": 52901,
      "training_step_time": 8.175915956497192
    },
    {
      "epoch": 5.045127868652344e-05,
      "model_forward_time": 1.702357292175293,
      "step": 52902
    },
    {
      "epoch": 5.045127868652344e-05,
      "step": 52902,
      "training_step_time": 8.580707788467407
    },
    {
      "epoch": 5.045223236083984e-05,
      "model_forward_time": 1.7082316875457764,
      "step": 52903
    },
    {
      "epoch": 5.045223236083984e-05,
      "step": 52903,
      "training_step_time": 7.74875545501709
    },
    {
      "epoch": 5.045318603515625e-05,
      "model_forward_time": 1.7031986713409424,
      "step": 52904
    },
    {
      "epoch": 5.045318603515625e-05,
      "step": 52904,
      "training_step_time": 8.220618486404419
    },
    {
      "epoch": 5.045413970947266e-05,
      "model_forward_time": 1.7055282592773438,
      "step": 52905
    },
    {
      "epoch": 5.045413970947266e-05,
      "step": 52905,
      "training_step_time": 8.304339170455933
    },
    {
      "epoch": 5.045509338378906e-05,
      "model_forward_time": 1.7039422988891602,
      "step": 52906
    },
    {
      "epoch": 5.045509338378906e-05,
      "step": 52906,
      "training_step_time": 8.362313270568848
    },
    {
      "epoch": 5.0456047058105465e-05,
      "model_forward_time": 1.7032482624053955,
      "step": 52907
    },
    {
      "epoch": 5.0456047058105465e-05,
      "step": 52907,
      "training_step_time": 8.332353591918945
    },
    {
      "epoch": 5.0457000732421876e-05,
      "model_forward_time": 1.702892541885376,
      "step": 52908
    },
    {
      "epoch": 5.0457000732421876e-05,
      "step": 52908,
      "training_step_time": 8.252299308776855
    },
    {
      "epoch": 5.045795440673828e-05,
      "model_forward_time": 1.7044591903686523,
      "step": 52909
    },
    {
      "epoch": 5.045795440673828e-05,
      "step": 52909,
      "training_step_time": 8.20919418334961
    },
    {
      "epoch": 5.0458908081054684e-05,
      "grad_norm": 0.12534767389297485,
      "learning_rate": 4.9323751463375145e-06,
      "loss": 0.0592,
      "step": 52910
    },
    {
      "epoch": 5.0458908081054684e-05,
      "model_forward_time": 1.7089171409606934,
      "step": 52910
    },
    {
      "dynamics_loss_avg": 0.05307907722890377,
      "epoch": 5.0458908081054684e-05,
      "step": 52910
    },
    {
      "action_loss_avg": 0.0016706739792425652,
      "epoch": 5.0458908081054684e-05,
      "step": 52910
    },
    {
      "epoch": 5.0458908081054684e-05,
      "step": 52910,
      "training_step_time": 7.625586032867432
    },
    {
      "epoch": 5.0459861755371095e-05,
      "model_forward_time": 1.753638505935669,
      "step": 52911
    },
    {
      "epoch": 5.0459861755371095e-05,
      "step": 52911,
      "training_step_time": 7.698927164077759
    },
    {
      "epoch": 5.04608154296875e-05,
      "model_forward_time": 1.699105978012085,
      "step": 52912
    },
    {
      "epoch": 5.04608154296875e-05,
      "step": 52912,
      "training_step_time": 8.45276141166687
    },
    {
      "epoch": 5.04617691040039e-05,
      "model_forward_time": 1.7010643482208252,
      "step": 52913
    },
    {
      "epoch": 5.04617691040039e-05,
      "step": 52913,
      "training_step_time": 7.706125259399414
    },
    {
      "epoch": 5.0462722778320314e-05,
      "model_forward_time": 1.7031149864196777,
      "step": 52914
    },
    {
      "epoch": 5.0462722778320314e-05,
      "step": 52914,
      "training_step_time": 8.892099380493164
    },
    {
      "epoch": 5.046367645263672e-05,
      "model_forward_time": 1.3194248676300049,
      "step": 52915
    },
    {
      "epoch": 5.046367645263672e-05,
      "step": 52915,
      "training_step_time": 7.788449287414551
    },
    {
      "epoch": 5.046463012695312e-05,
      "model_forward_time": 1.7044014930725098,
      "step": 52916
    },
    {
      "epoch": 5.046463012695312e-05,
      "step": 52916,
      "training_step_time": 7.778110980987549
    },
    {
      "epoch": 5.046558380126953e-05,
      "model_forward_time": 1.7027015686035156,
      "step": 52917
    },
    {
      "epoch": 5.046558380126953e-05,
      "step": 52917,
      "training_step_time": 8.848897457122803
    },
    {
      "epoch": 5.046653747558594e-05,
      "model_forward_time": 1.6991703510284424,
      "step": 52918
    },
    {
      "epoch": 5.046653747558594e-05,
      "step": 52918,
      "training_step_time": 7.709395885467529
    },
    {
      "epoch": 5.046749114990234e-05,
      "model_forward_time": 2.247403144836426,
      "step": 52919
    },
    {
      "epoch": 5.046749114990234e-05,
      "step": 52919,
      "training_step_time": 7.681041240692139
    },
    {
      "epoch": 5.046844482421875e-05,
      "grad_norm": 0.10881519317626953,
      "learning_rate": 4.930721831484908e-06,
      "loss": 0.0584,
      "step": 52920
    },
    {
      "epoch": 5.046844482421875e-05,
      "model_forward_time": 1.741201639175415,
      "step": 52920
    },
    {
      "dynamics_loss_avg": 0.0729133989661932,
      "epoch": 5.046844482421875e-05,
      "step": 52920
    },
    {
      "action_loss_avg": 0.0018954343744553626,
      "epoch": 5.046844482421875e-05,
      "step": 52920
    },
    {
      "epoch": 5.046844482421875e-05,
      "step": 52920,
      "training_step_time": 8.608757019042969
    },
    {
      "epoch": 5.0469398498535156e-05,
      "model_forward_time": 1.7035105228424072,
      "step": 52921
    },
    {
      "epoch": 5.0469398498535156e-05,
      "step": 52921,
      "training_step_time": 8.3961923122406
    },
    {
      "epoch": 5.047035217285156e-05,
      "model_forward_time": 1.7022511959075928,
      "step": 52922
    },
    {
      "epoch": 5.047035217285156e-05,
      "step": 52922,
      "training_step_time": 7.779865026473999
    },
    {
      "epoch": 5.047130584716797e-05,
      "model_forward_time": 1.7079648971557617,
      "step": 52923
    },
    {
      "epoch": 5.047130584716797e-05,
      "step": 52923,
      "training_step_time": 7.855090618133545
    },
    {
      "epoch": 5.0472259521484375e-05,
      "model_forward_time": 1.7103683948516846,
      "step": 52924
    },
    {
      "epoch": 5.0472259521484375e-05,
      "step": 52924,
      "training_step_time": 8.464220523834229
    },
    {
      "epoch": 5.047321319580078e-05,
      "model_forward_time": 1.7090744972229004,
      "step": 52925
    },
    {
      "epoch": 5.047321319580078e-05,
      "step": 52925,
      "training_step_time": 7.73104453086853
    },
    {
      "epoch": 5.047416687011719e-05,
      "model_forward_time": 1.7001261711120605,
      "step": 52926
    },
    {
      "epoch": 5.047416687011719e-05,
      "step": 52926,
      "training_step_time": 7.737124443054199
    },
    {
      "epoch": 5.047512054443359e-05,
      "model_forward_time": 1.7052948474884033,
      "step": 52927
    },
    {
      "epoch": 5.047512054443359e-05,
      "step": 52927,
      "training_step_time": 8.804598808288574
    },
    {
      "epoch": 5.047607421875e-05,
      "model_forward_time": 2.2611141204833984,
      "step": 52928
    },
    {
      "epoch": 5.047607421875e-05,
      "step": 52928,
      "training_step_time": 8.652131080627441
    },
    {
      "epoch": 5.047702789306641e-05,
      "model_forward_time": 1.7443726062774658,
      "step": 52929
    },
    {
      "epoch": 5.047702789306641e-05,
      "step": 52929,
      "training_step_time": 7.688266754150391
    },
    {
      "epoch": 5.047798156738281e-05,
      "grad_norm": 0.07658559828996658,
      "learning_rate": 4.929068524208456e-06,
      "loss": 0.0568,
      "step": 52930
    },
    {
      "epoch": 5.047798156738281e-05,
      "model_forward_time": 1.690859317779541,
      "step": 52930
    },
    {
      "dynamics_loss_avg": 0.050059709046036,
      "epoch": 5.047798156738281e-05,
      "step": 52930
    },
    {
      "action_loss_avg": 0.00144256295170635,
      "epoch": 5.047798156738281e-05,
      "step": 52930
    },
    {
      "epoch": 5.047798156738281e-05,
      "step": 52930,
      "training_step_time": 7.764115571975708
    },
    {
      "epoch": 5.0478935241699216e-05,
      "model_forward_time": 1.6995365619659424,
      "step": 52931
    },
    {
      "epoch": 5.0478935241699216e-05,
      "step": 52931,
      "training_step_time": 7.850957870483398
    },
    {
      "epoch": 5.047988891601563e-05,
      "model_forward_time": 1.7063422203063965,
      "step": 52932
    },
    {
      "epoch": 5.047988891601563e-05,
      "step": 52932,
      "training_step_time": 7.923036336898804
    },
    {
      "epoch": 5.048084259033203e-05,
      "model_forward_time": 1.704594373703003,
      "step": 52933
    },
    {
      "epoch": 5.048084259033203e-05,
      "step": 52933,
      "training_step_time": 8.518263578414917
    },
    {
      "epoch": 5.0481796264648435e-05,
      "model_forward_time": 1.70135498046875,
      "step": 52934
    },
    {
      "epoch": 5.0481796264648435e-05,
      "step": 52934,
      "training_step_time": 7.781466484069824
    },
    {
      "epoch": 5.0482749938964846e-05,
      "model_forward_time": 1.701833724975586,
      "step": 52935
    },
    {
      "epoch": 5.0482749938964846e-05,
      "step": 52935,
      "training_step_time": 7.756591081619263
    },
    {
      "epoch": 5.048370361328125e-05,
      "model_forward_time": 1.7044272422790527,
      "step": 52936
    },
    {
      "epoch": 5.048370361328125e-05,
      "step": 52936,
      "training_step_time": 7.875073194503784
    },
    {
      "epoch": 5.0484657287597654e-05,
      "model_forward_time": 2.269766092300415,
      "step": 52937
    },
    {
      "epoch": 5.0484657287597654e-05,
      "step": 52937,
      "training_step_time": 7.7646379470825195
    },
    {
      "epoch": 5.0485610961914065e-05,
      "model_forward_time": 1.7296867370605469,
      "step": 52938
    },
    {
      "epoch": 5.0485610961914065e-05,
      "step": 52938,
      "training_step_time": 7.669594049453735
    },
    {
      "epoch": 5.048656463623047e-05,
      "model_forward_time": 1.7070646286010742,
      "step": 52939
    },
    {
      "epoch": 5.048656463623047e-05,
      "step": 52939,
      "training_step_time": 7.787475824356079
    },
    {
      "epoch": 5.048751831054687e-05,
      "grad_norm": 0.08906599134206772,
      "learning_rate": 4.927415224688965e-06,
      "loss": 0.0558,
      "step": 52940
    },
    {
      "epoch": 5.048751831054687e-05,
      "model_forward_time": 1.7041692733764648,
      "step": 52940
    },
    {
      "dynamics_loss_avg": 0.06005477495491505,
      "epoch": 5.048751831054687e-05,
      "step": 52940
    },
    {
      "action_loss_avg": 0.0023069661925546824,
      "epoch": 5.048751831054687e-05,
      "step": 52940
    },
    {
      "epoch": 5.048751831054687e-05,
      "step": 52940,
      "training_step_time": 7.7505106925964355
    },
    {
      "epoch": 5.0488471984863284e-05,
      "model_forward_time": 1.6993117332458496,
      "step": 52941
    },
    {
      "epoch": 5.0488471984863284e-05,
      "step": 52941,
      "training_step_time": 8.792741537094116
    },
    {
      "epoch": 5.048942565917969e-05,
      "model_forward_time": 1.7044017314910889,
      "step": 52942
    },
    {
      "epoch": 5.048942565917969e-05,
      "step": 52942,
      "training_step_time": 8.743595600128174
    },
    {
      "epoch": 5.049037933349609e-05,
      "model_forward_time": 1.7022628784179688,
      "step": 52943
    },
    {
      "epoch": 5.049037933349609e-05,
      "step": 52943,
      "training_step_time": 7.773729562759399
    },
    {
      "epoch": 5.04913330078125e-05,
      "model_forward_time": 1.7048156261444092,
      "step": 52944
    },
    {
      "epoch": 5.04913330078125e-05,
      "step": 52944,
      "training_step_time": 8.3457350730896
    },
    {
      "epoch": 5.0492286682128907e-05,
      "model_forward_time": 1.7007355690002441,
      "step": 52945
    },
    {
      "epoch": 5.0492286682128907e-05,
      "step": 52945,
      "training_step_time": 8.71848440170288
    },
    {
      "epoch": 5.049324035644531e-05,
      "model_forward_time": 2.2496607303619385,
      "step": 52946
    },
    {
      "epoch": 5.049324035644531e-05,
      "step": 52946,
      "training_step_time": 7.724798679351807
    },
    {
      "epoch": 5.049419403076172e-05,
      "model_forward_time": 1.7100696563720703,
      "step": 52947
    },
    {
      "epoch": 5.049419403076172e-05,
      "step": 52947,
      "training_step_time": 7.626899003982544
    },
    {
      "epoch": 5.0495147705078125e-05,
      "model_forward_time": 1.6995632648468018,
      "step": 52948
    },
    {
      "epoch": 5.0495147705078125e-05,
      "step": 52948,
      "training_step_time": 7.8228631019592285
    },
    {
      "epoch": 5.049610137939453e-05,
      "model_forward_time": 1.7021384239196777,
      "step": 52949
    },
    {
      "epoch": 5.049610137939453e-05,
      "step": 52949,
      "training_step_time": 8.581204175949097
    },
    {
      "epoch": 5.049705505371094e-05,
      "grad_norm": 0.1187581941485405,
      "learning_rate": 4.925761933107238e-06,
      "loss": 0.0579,
      "step": 52950
    },
    {
      "epoch": 5.049705505371094e-05,
      "model_forward_time": 1.70511794090271,
      "step": 52950
    },
    {
      "dynamics_loss_avg": 0.053656382113695146,
      "epoch": 5.049705505371094e-05,
      "step": 52950
    },
    {
      "action_loss_avg": 0.0012161225895397365,
      "epoch": 5.049705505371094e-05,
      "step": 52950
    },
    {
      "epoch": 5.049705505371094e-05,
      "step": 52950,
      "training_step_time": 8.406636714935303
    },
    {
      "epoch": 5.0498008728027344e-05,
      "model_forward_time": 1.701319932937622,
      "step": 52951
    },
    {
      "epoch": 5.0498008728027344e-05,
      "step": 52951,
      "training_step_time": 8.440890312194824
    },
    {
      "epoch": 5.049896240234375e-05,
      "model_forward_time": 1.698981761932373,
      "step": 52952
    },
    {
      "epoch": 5.049896240234375e-05,
      "step": 52952,
      "training_step_time": 7.576599597930908
    },
    {
      "epoch": 5.049991607666016e-05,
      "model_forward_time": 1.7028706073760986,
      "step": 52953
    },
    {
      "epoch": 5.049991607666016e-05,
      "step": 52953,
      "training_step_time": 7.611963987350464
    },
    {
      "epoch": 5.050086975097656e-05,
      "model_forward_time": 1.6907756328582764,
      "step": 52954
    },
    {
      "epoch": 5.050086975097656e-05,
      "step": 52954,
      "training_step_time": 7.656869649887085
    },
    {
      "epoch": 5.050182342529297e-05,
      "model_forward_time": 1.7071707248687744,
      "step": 52955
    },
    {
      "epoch": 5.050182342529297e-05,
      "step": 52955,
      "training_step_time": 8.419911623001099
    },
    {
      "epoch": 5.050277709960938e-05,
      "model_forward_time": 2.2538323402404785,
      "step": 52956
    },
    {
      "epoch": 5.050277709960938e-05,
      "step": 52956,
      "training_step_time": 8.425815105438232
    },
    {
      "epoch": 5.050373077392578e-05,
      "model_forward_time": 1.7563111782073975,
      "step": 52957
    },
    {
      "epoch": 5.050373077392578e-05,
      "step": 52957,
      "training_step_time": 7.676788568496704
    },
    {
      "epoch": 5.0504684448242186e-05,
      "model_forward_time": 1.7021162509918213,
      "step": 52958
    },
    {
      "epoch": 5.0504684448242186e-05,
      "step": 52958,
      "training_step_time": 9.479233503341675
    },
    {
      "epoch": 5.05056381225586e-05,
      "model_forward_time": 1.700216293334961,
      "step": 52959
    },
    {
      "epoch": 5.05056381225586e-05,
      "step": 52959,
      "training_step_time": 8.47277307510376
    },
    {
      "epoch": 5.0506591796875e-05,
      "grad_norm": 0.11162352561950684,
      "learning_rate": 4.924108649644072e-06,
      "loss": 0.0624,
      "step": 52960
    },
    {
      "epoch": 5.0506591796875e-05,
      "model_forward_time": 1.7001383304595947,
      "step": 52960
    },
    {
      "dynamics_loss_avg": 0.06547078378498554,
      "epoch": 5.0506591796875e-05,
      "step": 52960
    },
    {
      "action_loss_avg": 0.002090266384766437,
      "epoch": 5.0506591796875e-05,
      "step": 52960
    },
    {
      "epoch": 5.0506591796875e-05,
      "step": 52960,
      "training_step_time": 7.724442005157471
    },
    {
      "epoch": 5.0507545471191405e-05,
      "model_forward_time": 1.70526123046875,
      "step": 52961
    },
    {
      "epoch": 5.0507545471191405e-05,
      "step": 52961,
      "training_step_time": 7.760055780410767
    },
    {
      "epoch": 5.0508499145507816e-05,
      "model_forward_time": 1.7086422443389893,
      "step": 52962
    },
    {
      "epoch": 5.0508499145507816e-05,
      "step": 52962,
      "training_step_time": 8.298532485961914
    },
    {
      "epoch": 5.050945281982422e-05,
      "model_forward_time": 1.6932246685028076,
      "step": 52963
    },
    {
      "epoch": 5.050945281982422e-05,
      "step": 52963,
      "training_step_time": 7.671898365020752
    },
    {
      "epoch": 5.0510406494140624e-05,
      "model_forward_time": 1.7014288902282715,
      "step": 52964
    },
    {
      "epoch": 5.0510406494140624e-05,
      "step": 52964,
      "training_step_time": 7.771353244781494
    },
    {
      "epoch": 5.0511360168457035e-05,
      "model_forward_time": 2.245495557785034,
      "step": 52965
    },
    {
      "epoch": 5.0511360168457035e-05,
      "step": 52965,
      "training_step_time": 7.639467239379883
    },
    {
      "epoch": 5.051231384277344e-05,
      "model_forward_time": 1.7404627799987793,
      "step": 52966
    },
    {
      "epoch": 5.051231384277344e-05,
      "step": 52966,
      "training_step_time": 7.736180067062378
    },
    {
      "epoch": 5.051326751708984e-05,
      "model_forward_time": 1.7409138679504395,
      "step": 52967
    },
    {
      "epoch": 5.051326751708984e-05,
      "step": 52967,
      "training_step_time": 7.82201361656189
    },
    {
      "epoch": 5.051422119140625e-05,
      "model_forward_time": 1.708162784576416,
      "step": 52968
    },
    {
      "epoch": 5.051422119140625e-05,
      "step": 52968,
      "training_step_time": 7.79433798789978
    },
    {
      "epoch": 5.051517486572266e-05,
      "model_forward_time": 1.6946399211883545,
      "step": 52969
    },
    {
      "epoch": 5.051517486572266e-05,
      "step": 52969,
      "training_step_time": 8.727649927139282
    },
    {
      "epoch": 5.051612854003906e-05,
      "grad_norm": 0.12026525288820267,
      "learning_rate": 4.922455374480271e-06,
      "loss": 0.0578,
      "step": 52970
    },
    {
      "epoch": 5.051612854003906e-05,
      "model_forward_time": 1.7027738094329834,
      "step": 52970
    },
    {
      "dynamics_loss_avg": 0.04893063008785248,
      "epoch": 5.051612854003906e-05,
      "step": 52970
    },
    {
      "action_loss_avg": 0.0010123568237759173,
      "epoch": 5.051612854003906e-05,
      "step": 52970
    },
    {
      "epoch": 5.051612854003906e-05,
      "step": 52970,
      "training_step_time": 7.97093391418457
    },
    {
      "epoch": 5.0517082214355465e-05,
      "model_forward_time": 1.710423469543457,
      "step": 52971
    },
    {
      "epoch": 5.0517082214355465e-05,
      "step": 52971,
      "training_step_time": 7.848602771759033
    },
    {
      "epoch": 5.0518035888671876e-05,
      "model_forward_time": 1.6936736106872559,
      "step": 52972
    },
    {
      "epoch": 5.0518035888671876e-05,
      "step": 52972,
      "training_step_time": 8.631293535232544
    },
    {
      "epoch": 5.051898956298828e-05,
      "model_forward_time": 1.7021989822387695,
      "step": 52973
    },
    {
      "epoch": 5.051898956298828e-05,
      "step": 52973,
      "training_step_time": 8.343145847320557
    },
    {
      "epoch": 5.0519943237304684e-05,
      "model_forward_time": 2.231980562210083,
      "step": 52974
    },
    {
      "epoch": 5.0519943237304684e-05,
      "step": 52974,
      "training_step_time": 8.224342584609985
    },
    {
      "epoch": 5.0520896911621095e-05,
      "model_forward_time": 1.7181901931762695,
      "step": 52975
    },
    {
      "epoch": 5.0520896911621095e-05,
      "step": 52975,
      "training_step_time": 8.199625015258789
    },
    {
      "epoch": 5.05218505859375e-05,
      "model_forward_time": 1.7042407989501953,
      "step": 52976
    },
    {
      "epoch": 5.05218505859375e-05,
      "step": 52976,
      "training_step_time": 8.608821392059326
    },
    {
      "epoch": 5.05228042602539e-05,
      "model_forward_time": 1.6989567279815674,
      "step": 52977
    },
    {
      "epoch": 5.05228042602539e-05,
      "step": 52977,
      "training_step_time": 7.559327840805054
    },
    {
      "epoch": 5.0523757934570314e-05,
      "model_forward_time": 1.7051820755004883,
      "step": 52978
    },
    {
      "epoch": 5.0523757934570314e-05,
      "step": 52978,
      "training_step_time": 7.752846717834473
    },
    {
      "epoch": 5.052471160888672e-05,
      "model_forward_time": 1.6992559432983398,
      "step": 52979
    },
    {
      "epoch": 5.052471160888672e-05,
      "step": 52979,
      "training_step_time": 7.732745409011841
    },
    {
      "epoch": 5.052566528320312e-05,
      "grad_norm": 0.0855347216129303,
      "learning_rate": 4.920802107796634e-06,
      "loss": 0.0603,
      "step": 52980
    },
    {
      "epoch": 5.052566528320312e-05,
      "model_forward_time": 1.698948621749878,
      "step": 52980
    },
    {
      "dynamics_loss_avg": 0.058754749689251186,
      "epoch": 5.052566528320312e-05,
      "step": 52980
    },
    {
      "action_loss_avg": 0.0017725854442687706,
      "epoch": 5.052566528320312e-05,
      "step": 52980
    },
    {
      "epoch": 5.052566528320312e-05,
      "step": 52980,
      "training_step_time": 7.568744897842407
    },
    {
      "epoch": 5.052661895751953e-05,
      "model_forward_time": 1.703601598739624,
      "step": 52981
    },
    {
      "epoch": 5.052661895751953e-05,
      "step": 52981,
      "training_step_time": 8.292361736297607
    },
    {
      "epoch": 5.052757263183594e-05,
      "model_forward_time": 1.6961724758148193,
      "step": 52982
    },
    {
      "epoch": 5.052757263183594e-05,
      "step": 52982,
      "training_step_time": 7.60986328125
    },
    {
      "epoch": 5.052852630615234e-05,
      "model_forward_time": 2.246115207672119,
      "step": 52983
    },
    {
      "epoch": 5.052852630615234e-05,
      "step": 52983,
      "training_step_time": 7.746779441833496
    },
    {
      "epoch": 5.052947998046875e-05,
      "model_forward_time": 1.6970405578613281,
      "step": 52984
    },
    {
      "epoch": 5.052947998046875e-05,
      "step": 52984,
      "training_step_time": 8.22544813156128
    },
    {
      "epoch": 5.0530433654785156e-05,
      "model_forward_time": 1.6955091953277588,
      "step": 52985
    },
    {
      "epoch": 5.0530433654785156e-05,
      "step": 52985,
      "training_step_time": 7.764399528503418
    },
    {
      "epoch": 5.053138732910156e-05,
      "model_forward_time": 1.702869176864624,
      "step": 52986
    },
    {
      "epoch": 5.053138732910156e-05,
      "step": 52986,
      "training_step_time": 8.406933784484863
    },
    {
      "epoch": 5.053234100341797e-05,
      "model_forward_time": 1.7015137672424316,
      "step": 52987
    },
    {
      "epoch": 5.053234100341797e-05,
      "step": 52987,
      "training_step_time": 8.476984977722168
    },
    {
      "epoch": 5.0533294677734375e-05,
      "model_forward_time": 1.6965739727020264,
      "step": 52988
    },
    {
      "epoch": 5.0533294677734375e-05,
      "step": 52988,
      "training_step_time": 8.921186923980713
    },
    {
      "epoch": 5.053424835205078e-05,
      "model_forward_time": 1.6945433616638184,
      "step": 52989
    },
    {
      "epoch": 5.053424835205078e-05,
      "step": 52989,
      "training_step_time": 7.728421688079834
    },
    {
      "epoch": 5.053520202636719e-05,
      "grad_norm": 0.11080774664878845,
      "learning_rate": 4.9191488497739566e-06,
      "loss": 0.0618,
      "step": 52990
    },
    {
      "epoch": 5.053520202636719e-05,
      "model_forward_time": 1.6989030838012695,
      "step": 52990
    },
    {
      "dynamics_loss_avg": 0.05524159567430616,
      "epoch": 5.053520202636719e-05,
      "step": 52990
    },
    {
      "action_loss_avg": 0.0014584923686925323,
      "epoch": 5.053520202636719e-05,
      "step": 52990
    },
    {
      "epoch": 5.053520202636719e-05,
      "step": 52990,
      "training_step_time": 8.162063360214233
    },
    {
      "epoch": 5.0536155700683593e-05,
      "model_forward_time": 1.6947073936462402,
      "step": 52991
    },
    {
      "epoch": 5.0536155700683593e-05,
      "step": 52991,
      "training_step_time": 8.35721206665039
    },
    {
      "epoch": 5.0537109375e-05,
      "model_forward_time": 2.2629003524780273,
      "step": 52992
    },
    {
      "epoch": 5.0537109375e-05,
      "step": 52992,
      "training_step_time": 8.773620843887329
    },
    {
      "epoch": 5.053806304931641e-05,
      "model_forward_time": 1.7195110321044922,
      "step": 52993
    },
    {
      "epoch": 5.053806304931641e-05,
      "step": 52993,
      "training_step_time": 7.712713241577148
    },
    {
      "epoch": 5.053901672363281e-05,
      "model_forward_time": 1.752990961074829,
      "step": 52994
    },
    {
      "epoch": 5.053901672363281e-05,
      "step": 52994,
      "training_step_time": 7.72917628288269
    },
    {
      "epoch": 5.0539970397949216e-05,
      "model_forward_time": 1.7003161907196045,
      "step": 52995
    },
    {
      "epoch": 5.0539970397949216e-05,
      "step": 52995,
      "training_step_time": 8.239449501037598
    },
    {
      "epoch": 5.054092407226563e-05,
      "model_forward_time": 1.710517168045044,
      "step": 52996
    },
    {
      "epoch": 5.054092407226563e-05,
      "step": 52996,
      "training_step_time": 9.869687795639038
    },
    {
      "epoch": 5.054187774658203e-05,
      "model_forward_time": 1.719454050064087,
      "step": 52997
    },
    {
      "epoch": 5.054187774658203e-05,
      "step": 52997,
      "training_step_time": 7.525098085403442
    },
    {
      "epoch": 5.0542831420898435e-05,
      "model_forward_time": 1.718982458114624,
      "step": 52998
    },
    {
      "epoch": 5.0542831420898435e-05,
      "step": 52998,
      "training_step_time": 8.879508018493652
    },
    {
      "epoch": 5.0543785095214846e-05,
      "model_forward_time": 1.6956140995025635,
      "step": 52999
    },
    {
      "epoch": 5.0543785095214846e-05,
      "step": 52999,
      "training_step_time": 7.572339773178101
    },
    {
      "epoch": 5.054473876953125e-05,
      "grad_norm": 0.08270535618066788,
      "learning_rate": 4.917495600593042e-06,
      "loss": 0.058,
      "step": 53000
    },
    {
      "epoch": 5.054473876953125e-05,
      "model_forward_time": 1.7535996437072754,
      "step": 53000
    },
    {
      "dynamics_loss_avg": 0.04391483282670379,
      "epoch": 5.054473876953125e-05,
      "step": 53000
    },
    {
      "action_loss_avg": 0.0010776164417620748,
      "epoch": 5.054473876953125e-05,
      "step": 53000
    },
    {
      "epoch": 5.054473876953125e-05,
      "step": 53000,
      "training_step_time": 6.540182828903198
    },
    {
      "epoch": 5.0545692443847654e-05,
      "model_forward_time": 2.7381720542907715,
      "step": 53001
    },
    {
      "epoch": 5.0545692443847654e-05,
      "step": 53001,
      "training_step_time": 9.275329113006592
    },
    {
      "epoch": 5.0546646118164065e-05,
      "model_forward_time": 1.7010374069213867,
      "step": 53002
    },
    {
      "epoch": 5.0546646118164065e-05,
      "step": 53002,
      "training_step_time": 9.302824974060059
    },
    {
      "epoch": 5.054759979248047e-05,
      "model_forward_time": 1.6998236179351807,
      "step": 53003
    },
    {
      "epoch": 5.054759979248047e-05,
      "step": 53003,
      "training_step_time": 7.639843702316284
    },
    {
      "epoch": 5.054855346679687e-05,
      "model_forward_time": 1.694319486618042,
      "step": 53004
    },
    {
      "epoch": 5.054855346679687e-05,
      "step": 53004,
      "training_step_time": 9.594873189926147
    },
    {
      "epoch": 5.0549507141113284e-05,
      "model_forward_time": 1.7081682682037354,
      "step": 53005
    },
    {
      "epoch": 5.0549507141113284e-05,
      "step": 53005,
      "training_step_time": 8.974489688873291
    },
    {
      "epoch": 5.055046081542969e-05,
      "model_forward_time": 1.324051856994629,
      "step": 53006
    },
    {
      "epoch": 5.055046081542969e-05,
      "step": 53006,
      "training_step_time": 8.441105127334595
    },
    {
      "epoch": 5.055141448974609e-05,
      "model_forward_time": 1.6949455738067627,
      "step": 53007
    },
    {
      "epoch": 5.055141448974609e-05,
      "step": 53007,
      "training_step_time": 8.160428762435913
    },
    {
      "epoch": 5.05523681640625e-05,
      "model_forward_time": 1.702453851699829,
      "step": 53008
    },
    {
      "epoch": 5.05523681640625e-05,
      "step": 53008,
      "training_step_time": 8.073822736740112
    },
    {
      "epoch": 5.0553321838378907e-05,
      "model_forward_time": 0.9846866130828857,
      "step": 53009
    },
    {
      "epoch": 5.0553321838378907e-05,
      "step": 53009,
      "training_step_time": 7.492219924926758
    },
    {
      "epoch": 5.055427551269531e-05,
      "grad_norm": 0.08739771693944931,
      "learning_rate": 4.915842360434684e-06,
      "loss": 0.0614,
      "step": 53010
    },
    {
      "epoch": 5.055427551269531e-05,
      "model_forward_time": 2.242504596710205,
      "step": 53010
    },
    {
      "dynamics_loss_avg": 0.056816482823342083,
      "epoch": 5.055427551269531e-05,
      "step": 53010
    },
    {
      "action_loss_avg": 0.002659609983675182,
      "epoch": 5.055427551269531e-05,
      "step": 53010
    },
    {
      "epoch": 5.055427551269531e-05,
      "step": 53010,
      "training_step_time": 7.6012122631073
    },
    {
      "epoch": 5.055522918701172e-05,
      "model_forward_time": 1.697847843170166,
      "step": 53011
    },
    {
      "epoch": 5.055522918701172e-05,
      "step": 53011,
      "training_step_time": 7.711566925048828
    },
    {
      "epoch": 5.0556182861328125e-05,
      "model_forward_time": 1.7447795867919922,
      "step": 53012
    },
    {
      "epoch": 5.0556182861328125e-05,
      "step": 53012,
      "training_step_time": 7.714333772659302
    },
    {
      "epoch": 5.055713653564453e-05,
      "model_forward_time": 1.703775405883789,
      "step": 53013
    },
    {
      "epoch": 5.055713653564453e-05,
      "step": 53013,
      "training_step_time": 8.317212581634521
    },
    {
      "epoch": 5.055809020996094e-05,
      "model_forward_time": 1.699591875076294,
      "step": 53014
    },
    {
      "epoch": 5.055809020996094e-05,
      "step": 53014,
      "training_step_time": 7.683047294616699
    },
    {
      "epoch": 5.0559043884277344e-05,
      "model_forward_time": 1.7020208835601807,
      "step": 53015
    },
    {
      "epoch": 5.0559043884277344e-05,
      "step": 53015,
      "training_step_time": 9.583171129226685
    },
    {
      "epoch": 5.055999755859375e-05,
      "model_forward_time": 1.702878475189209,
      "step": 53016
    },
    {
      "epoch": 5.055999755859375e-05,
      "step": 53016,
      "training_step_time": 9.400492191314697
    },
    {
      "epoch": 5.056095123291016e-05,
      "model_forward_time": 1.7036428451538086,
      "step": 53017
    },
    {
      "epoch": 5.056095123291016e-05,
      "step": 53017,
      "training_step_time": 6.726334810256958
    },
    {
      "epoch": 5.056190490722656e-05,
      "model_forward_time": 1.7007930278778076,
      "step": 53018
    },
    {
      "epoch": 5.056190490722656e-05,
      "step": 53018,
      "training_step_time": 6.75250244140625
    },
    {
      "epoch": 5.056285858154297e-05,
      "model_forward_time": 1.700425624847412,
      "step": 53019
    },
    {
      "epoch": 5.056285858154297e-05,
      "step": 53019,
      "training_step_time": 6.736884117126465
    },
    {
      "epoch": 5.056381225585938e-05,
      "grad_norm": 0.0986228659749031,
      "learning_rate": 4.914189129479679e-06,
      "loss": 0.0604,
      "step": 53020
    },
    {
      "epoch": 5.056381225585938e-05,
      "model_forward_time": 1.7010810375213623,
      "step": 53020
    },
    {
      "dynamics_loss_avg": 0.0607730133458972,
      "epoch": 5.056381225585938e-05,
      "step": 53020
    },
    {
      "action_loss_avg": 0.0023729812528472394,
      "epoch": 5.056381225585938e-05,
      "step": 53020
    },
    {
      "epoch": 5.056381225585938e-05,
      "step": 53020,
      "training_step_time": 6.75954794883728
    },
    {
      "epoch": 5.056476593017578e-05,
      "model_forward_time": 1.6926355361938477,
      "step": 53021
    },
    {
      "epoch": 5.056476593017578e-05,
      "step": 53021,
      "training_step_time": 6.7880859375
    },
    {
      "epoch": 5.0565719604492186e-05,
      "model_forward_time": 1.7032086849212646,
      "step": 53022
    },
    {
      "epoch": 5.0565719604492186e-05,
      "step": 53022,
      "training_step_time": 6.781682968139648
    },
    {
      "epoch": 5.05666732788086e-05,
      "model_forward_time": 0.9817678928375244,
      "step": 53023
    },
    {
      "epoch": 5.05666732788086e-05,
      "step": 53023,
      "training_step_time": 6.77894926071167
    },
    {
      "epoch": 5.0567626953125e-05,
      "model_forward_time": 1.7000129222869873,
      "step": 53024
    },
    {
      "epoch": 5.0567626953125e-05,
      "step": 53024,
      "training_step_time": 6.739552021026611
    },
    {
      "epoch": 5.0568580627441405e-05,
      "model_forward_time": 2.235722064971924,
      "step": 53025
    },
    {
      "epoch": 5.0568580627441405e-05,
      "step": 53025,
      "training_step_time": 7.398487567901611
    },
    {
      "epoch": 5.0569534301757816e-05,
      "model_forward_time": 1.7084753513336182,
      "step": 53026
    },
    {
      "epoch": 5.0569534301757816e-05,
      "step": 53026,
      "training_step_time": 8.394920349121094
    },
    {
      "epoch": 5.057048797607422e-05,
      "model_forward_time": 1.701537847518921,
      "step": 53027
    },
    {
      "epoch": 5.057048797607422e-05,
      "step": 53027,
      "training_step_time": 6.759284257888794
    },
    {
      "epoch": 5.0571441650390624e-05,
      "model_forward_time": 1.7010529041290283,
      "step": 53028
    },
    {
      "epoch": 5.0571441650390624e-05,
      "step": 53028,
      "training_step_time": 6.7927329540252686
    },
    {
      "epoch": 5.0572395324707035e-05,
      "model_forward_time": 1.6919736862182617,
      "step": 53029
    },
    {
      "epoch": 5.0572395324707035e-05,
      "step": 53029,
      "training_step_time": 6.759891748428345
    },
    {
      "epoch": 5.057334899902344e-05,
      "grad_norm": 0.13843129575252533,
      "learning_rate": 4.912535907908822e-06,
      "loss": 0.0568,
      "step": 53030
    },
    {
      "epoch": 5.057334899902344e-05,
      "model_forward_time": 1.7094805240631104,
      "step": 53030
    },
    {
      "dynamics_loss_avg": 0.053791137598454955,
      "epoch": 5.057334899902344e-05,
      "step": 53030
    },
    {
      "action_loss_avg": 0.0019589185947552325,
      "epoch": 5.057334899902344e-05,
      "step": 53030
    },
    {
      "epoch": 5.057334899902344e-05,
      "step": 53030,
      "training_step_time": 6.776648998260498
    },
    {
      "epoch": 5.057430267333984e-05,
      "model_forward_time": 1.7047502994537354,
      "step": 53031
    },
    {
      "epoch": 5.057430267333984e-05,
      "step": 53031,
      "training_step_time": 6.8418965339660645
    },
    {
      "epoch": 5.057525634765625e-05,
      "model_forward_time": 1.3264987468719482,
      "step": 53032
    },
    {
      "epoch": 5.057525634765625e-05,
      "step": 53032,
      "training_step_time": 6.805761098861694
    },
    {
      "epoch": 5.057621002197266e-05,
      "model_forward_time": 1.7049355506896973,
      "step": 53033
    },
    {
      "epoch": 5.057621002197266e-05,
      "step": 53033,
      "training_step_time": 6.812707185745239
    },
    {
      "epoch": 5.057716369628906e-05,
      "model_forward_time": 2.2483432292938232,
      "step": 53034
    },
    {
      "epoch": 5.057716369628906e-05,
      "step": 53034,
      "training_step_time": 7.432582378387451
    },
    {
      "epoch": 5.0578117370605465e-05,
      "model_forward_time": 1.7031304836273193,
      "step": 53035
    },
    {
      "epoch": 5.0578117370605465e-05,
      "step": 53035,
      "training_step_time": 7.7915565967559814
    },
    {
      "epoch": 5.0579071044921876e-05,
      "model_forward_time": 1.7055978775024414,
      "step": 53036
    },
    {
      "epoch": 5.0579071044921876e-05,
      "step": 53036,
      "training_step_time": 7.262292861938477
    },
    {
      "epoch": 5.058002471923828e-05,
      "model_forward_time": 1.7073085308074951,
      "step": 53037
    },
    {
      "epoch": 5.058002471923828e-05,
      "step": 53037,
      "training_step_time": 6.7900614738464355
    },
    {
      "epoch": 5.0580978393554684e-05,
      "model_forward_time": 1.699902057647705,
      "step": 53038
    },
    {
      "epoch": 5.0580978393554684e-05,
      "step": 53038,
      "training_step_time": 6.838632106781006
    },
    {
      "epoch": 5.0581932067871095e-05,
      "model_forward_time": 1.7028193473815918,
      "step": 53039
    },
    {
      "epoch": 5.0581932067871095e-05,
      "step": 53039,
      "training_step_time": 6.800076961517334
    },
    {
      "epoch": 5.05828857421875e-05,
      "grad_norm": 0.09190123528242111,
      "learning_rate": 4.910882695902907e-06,
      "loss": 0.0586,
      "step": 53040
    },
    {
      "epoch": 5.05828857421875e-05,
      "model_forward_time": 1.7060050964355469,
      "step": 53040
    },
    {
      "dynamics_loss_avg": 0.04766590036451816,
      "epoch": 5.05828857421875e-05,
      "step": 53040
    },
    {
      "action_loss_avg": 0.0012495664646849035,
      "epoch": 5.05828857421875e-05,
      "step": 53040
    },
    {
      "epoch": 5.05828857421875e-05,
      "step": 53040,
      "training_step_time": 6.8263373374938965
    },
    {
      "epoch": 5.05838394165039e-05,
      "model_forward_time": 1.698042631149292,
      "step": 53041
    },
    {
      "epoch": 5.05838394165039e-05,
      "step": 53041,
      "training_step_time": 6.814221620559692
    },
    {
      "epoch": 5.0584793090820314e-05,
      "model_forward_time": 1.7022733688354492,
      "step": 53042
    },
    {
      "epoch": 5.0584793090820314e-05,
      "step": 53042,
      "training_step_time": 6.755753755569458
    },
    {
      "epoch": 5.058574676513672e-05,
      "model_forward_time": 2.2278964519500732,
      "step": 53043
    },
    {
      "epoch": 5.058574676513672e-05,
      "step": 53043,
      "training_step_time": 7.399693965911865
    },
    {
      "epoch": 5.058670043945312e-05,
      "model_forward_time": 1.701549768447876,
      "step": 53044
    },
    {
      "epoch": 5.058670043945312e-05,
      "step": 53044,
      "training_step_time": 7.7826268672943115
    },
    {
      "epoch": 5.058765411376953e-05,
      "model_forward_time": 1.7052991390228271,
      "step": 53045
    },
    {
      "epoch": 5.058765411376953e-05,
      "step": 53045,
      "training_step_time": 7.665283918380737
    },
    {
      "epoch": 5.058860778808594e-05,
      "model_forward_time": 1.7022418975830078,
      "step": 53046
    },
    {
      "epoch": 5.058860778808594e-05,
      "step": 53046,
      "training_step_time": 6.76293420791626
    },
    {
      "epoch": 5.058956146240234e-05,
      "model_forward_time": 1.705822467803955,
      "step": 53047
    },
    {
      "epoch": 5.058956146240234e-05,
      "step": 53047,
      "training_step_time": 6.768129110336304
    },
    {
      "epoch": 5.059051513671875e-05,
      "model_forward_time": 1.7020959854125977,
      "step": 53048
    },
    {
      "epoch": 5.059051513671875e-05,
      "step": 53048,
      "training_step_time": 6.779752254486084
    },
    {
      "epoch": 5.0591468811035156e-05,
      "model_forward_time": 1.7015841007232666,
      "step": 53049
    },
    {
      "epoch": 5.0591468811035156e-05,
      "step": 53049,
      "training_step_time": 6.719610691070557
    },
    {
      "epoch": 5.059242248535156e-05,
      "grad_norm": 0.09510403126478195,
      "learning_rate": 4.909229493642724e-06,
      "loss": 0.0558,
      "step": 53050
    },
    {
      "epoch": 5.059242248535156e-05,
      "model_forward_time": 1.6945452690124512,
      "step": 53050
    },
    {
      "dynamics_loss_avg": 0.06620650794357061,
      "epoch": 5.059242248535156e-05,
      "step": 53050
    },
    {
      "action_loss_avg": 0.0020611694431863724,
      "epoch": 5.059242248535156e-05,
      "step": 53050
    },
    {
      "epoch": 5.059242248535156e-05,
      "step": 53050,
      "training_step_time": 6.783537149429321
    },
    {
      "epoch": 5.059337615966797e-05,
      "model_forward_time": 1.6977505683898926,
      "step": 53051
    },
    {
      "epoch": 5.059337615966797e-05,
      "step": 53051,
      "training_step_time": 6.8045549392700195
    },
    {
      "epoch": 5.0594329833984375e-05,
      "model_forward_time": 1.707756757736206,
      "step": 53052
    },
    {
      "epoch": 5.0594329833984375e-05,
      "step": 53052,
      "training_step_time": 7.391335964202881
    },
    {
      "epoch": 5.059528350830078e-05,
      "model_forward_time": 2.237194061279297,
      "step": 53053
    },
    {
      "epoch": 5.059528350830078e-05,
      "step": 53053,
      "training_step_time": 8.699376583099365
    },
    {
      "epoch": 5.059623718261719e-05,
      "model_forward_time": 1.7017486095428467,
      "step": 53054
    },
    {
      "epoch": 5.059623718261719e-05,
      "step": 53054,
      "training_step_time": 7.636656284332275
    },
    {
      "epoch": 5.0597190856933593e-05,
      "model_forward_time": 1.7005095481872559,
      "step": 53055
    },
    {
      "epoch": 5.0597190856933593e-05,
      "step": 53055,
      "training_step_time": 7.565547943115234
    },
    {
      "epoch": 5.059814453125e-05,
      "model_forward_time": 1.7106297016143799,
      "step": 53056
    },
    {
      "epoch": 5.059814453125e-05,
      "step": 53056,
      "training_step_time": 6.8038859367370605
    },
    {
      "epoch": 5.059909820556641e-05,
      "model_forward_time": 1.6932249069213867,
      "step": 53057
    },
    {
      "epoch": 5.059909820556641e-05,
      "step": 53057,
      "training_step_time": 6.76725697517395
    },
    {
      "epoch": 5.060005187988281e-05,
      "model_forward_time": 1.7183027267456055,
      "step": 53058
    },
    {
      "epoch": 5.060005187988281e-05,
      "step": 53058,
      "training_step_time": 6.793324708938599
    },
    {
      "epoch": 5.0601005554199216e-05,
      "model_forward_time": 1.7042651176452637,
      "step": 53059
    },
    {
      "epoch": 5.0601005554199216e-05,
      "step": 53059,
      "training_step_time": 6.798248052597046
    },
    {
      "epoch": 5.060195922851563e-05,
      "grad_norm": 0.0950305238366127,
      "learning_rate": 4.9075763013090685e-06,
      "loss": 0.0557,
      "step": 53060
    },
    {
      "epoch": 5.060195922851563e-05,
      "model_forward_time": 1.6908791065216064,
      "step": 53060
    },
    {
      "dynamics_loss_avg": 0.05298160649836063,
      "epoch": 5.060195922851563e-05,
      "step": 53060
    },
    {
      "action_loss_avg": 0.0013101077114697547,
      "epoch": 5.060195922851563e-05,
      "step": 53060
    },
    {
      "epoch": 5.060195922851563e-05,
      "step": 53060,
      "training_step_time": 6.737164735794067
    },
    {
      "epoch": 5.060291290283203e-05,
      "model_forward_time": 1.7065114974975586,
      "step": 53061
    },
    {
      "epoch": 5.060291290283203e-05,
      "step": 53061,
      "training_step_time": 7.3655924797058105
    },
    {
      "epoch": 5.0603866577148435e-05,
      "model_forward_time": 2.235175132751465,
      "step": 53062
    },
    {
      "epoch": 5.0603866577148435e-05,
      "step": 53062,
      "training_step_time": 9.26401424407959
    },
    {
      "epoch": 5.0604820251464846e-05,
      "model_forward_time": 1.6999866962432861,
      "step": 53063
    },
    {
      "epoch": 5.0604820251464846e-05,
      "step": 53063,
      "training_step_time": 7.798923969268799
    },
    {
      "epoch": 5.060577392578125e-05,
      "model_forward_time": 1.7047152519226074,
      "step": 53064
    },
    {
      "epoch": 5.060577392578125e-05,
      "step": 53064,
      "training_step_time": 7.5622313022613525
    },
    {
      "epoch": 5.0606727600097654e-05,
      "model_forward_time": 1.7086477279663086,
      "step": 53065
    },
    {
      "epoch": 5.0606727600097654e-05,
      "step": 53065,
      "training_step_time": 6.74021577835083
    },
    {
      "epoch": 5.0607681274414065e-05,
      "model_forward_time": 1.7013187408447266,
      "step": 53066
    },
    {
      "epoch": 5.0607681274414065e-05,
      "step": 53066,
      "training_step_time": 6.688114881515503
    },
    {
      "epoch": 5.060863494873047e-05,
      "model_forward_time": 1.704406499862671,
      "step": 53067
    },
    {
      "epoch": 5.060863494873047e-05,
      "step": 53067,
      "training_step_time": 6.724780559539795
    },
    {
      "epoch": 5.060958862304687e-05,
      "model_forward_time": 1.7042059898376465,
      "step": 53068
    },
    {
      "epoch": 5.060958862304687e-05,
      "step": 53068,
      "training_step_time": 6.722356796264648
    },
    {
      "epoch": 5.0610542297363284e-05,
      "model_forward_time": 1.6973919868469238,
      "step": 53069
    },
    {
      "epoch": 5.0610542297363284e-05,
      "step": 53069,
      "training_step_time": 6.664107322692871
    },
    {
      "epoch": 5.061149597167969e-05,
      "grad_norm": 0.1468207985162735,
      "learning_rate": 4.90592311908273e-06,
      "loss": 0.0531,
      "step": 53070
    },
    {
      "epoch": 5.061149597167969e-05,
      "model_forward_time": 1.7021901607513428,
      "step": 53070
    },
    {
      "dynamics_loss_avg": 0.047761211171746254,
      "epoch": 5.061149597167969e-05,
      "step": 53070
    },
    {
      "action_loss_avg": 0.0015139218186959625,
      "epoch": 5.061149597167969e-05,
      "step": 53070
    },
    {
      "epoch": 5.061149597167969e-05,
      "step": 53070,
      "training_step_time": 7.213023662567139
    },
    {
      "epoch": 5.061244964599609e-05,
      "model_forward_time": 1.8645434379577637,
      "step": 53071
    },
    {
      "epoch": 5.061244964599609e-05,
      "step": 53071,
      "training_step_time": 9.211260080337524
    },
    {
      "epoch": 5.06134033203125e-05,
      "model_forward_time": 1.6950857639312744,
      "step": 53072
    },
    {
      "epoch": 5.06134033203125e-05,
      "step": 53072,
      "training_step_time": 8.481261730194092
    },
    {
      "epoch": 5.061435699462891e-05,
      "model_forward_time": 1.709395408630371,
      "step": 53073
    },
    {
      "epoch": 5.061435699462891e-05,
      "step": 53073,
      "training_step_time": 7.652895212173462
    },
    {
      "epoch": 5.061531066894531e-05,
      "model_forward_time": 1.6930978298187256,
      "step": 53074
    },
    {
      "epoch": 5.061531066894531e-05,
      "step": 53074,
      "training_step_time": 6.701783180236816
    },
    {
      "epoch": 5.061626434326172e-05,
      "model_forward_time": 1.6970012187957764,
      "step": 53075
    },
    {
      "epoch": 5.061626434326172e-05,
      "step": 53075,
      "training_step_time": 6.737947940826416
    },
    {
      "epoch": 5.0617218017578125e-05,
      "model_forward_time": 1.6993725299835205,
      "step": 53076
    },
    {
      "epoch": 5.0617218017578125e-05,
      "step": 53076,
      "training_step_time": 6.7090935707092285
    },
    {
      "epoch": 5.061817169189453e-05,
      "model_forward_time": 1.7027480602264404,
      "step": 53077
    },
    {
      "epoch": 5.061817169189453e-05,
      "step": 53077,
      "training_step_time": 6.708969354629517
    },
    {
      "epoch": 5.061912536621094e-05,
      "model_forward_time": 1.6984190940856934,
      "step": 53078
    },
    {
      "epoch": 5.061912536621094e-05,
      "step": 53078,
      "training_step_time": 6.7313032150268555
    },
    {
      "epoch": 5.0620079040527344e-05,
      "model_forward_time": 1.6985232830047607,
      "step": 53079
    },
    {
      "epoch": 5.0620079040527344e-05,
      "step": 53079,
      "training_step_time": 6.698901414871216
    },
    {
      "epoch": 5.062103271484375e-05,
      "grad_norm": 0.1093098595738411,
      "learning_rate": 4.904269947144499e-06,
      "loss": 0.0584,
      "step": 53080
    },
    {
      "epoch": 5.062103271484375e-05,
      "model_forward_time": 2.2522919178009033,
      "step": 53080
    },
    {
      "dynamics_loss_avg": 0.04654060900211334,
      "epoch": 5.062103271484375e-05,
      "step": 53080
    },
    {
      "action_loss_avg": 0.0013734301217482425,
      "epoch": 5.062103271484375e-05,
      "step": 53080
    },
    {
      "epoch": 5.062103271484375e-05,
      "step": 53080,
      "training_step_time": 8.330042600631714
    },
    {
      "epoch": 5.062198638916016e-05,
      "model_forward_time": 1.6998720169067383,
      "step": 53081
    },
    {
      "epoch": 5.062198638916016e-05,
      "step": 53081,
      "training_step_time": 11.365387678146362
    },
    {
      "epoch": 5.062294006347656e-05,
      "model_forward_time": 1.6951212882995605,
      "step": 53082
    },
    {
      "epoch": 5.062294006347656e-05,
      "step": 53082,
      "training_step_time": 7.53243088722229
    },
    {
      "epoch": 5.062389373779297e-05,
      "model_forward_time": 1.692596197128296,
      "step": 53083
    },
    {
      "epoch": 5.062389373779297e-05,
      "step": 53083,
      "training_step_time": 6.699971437454224
    },
    {
      "epoch": 5.062484741210938e-05,
      "model_forward_time": 1.6955885887145996,
      "step": 53084
    },
    {
      "epoch": 5.062484741210938e-05,
      "step": 53084,
      "training_step_time": 6.697290658950806
    },
    {
      "epoch": 5.062580108642578e-05,
      "model_forward_time": 1.6955771446228027,
      "step": 53085
    },
    {
      "epoch": 5.062580108642578e-05,
      "step": 53085,
      "training_step_time": 6.742400407791138
    },
    {
      "epoch": 5.0626754760742186e-05,
      "model_forward_time": 1.6931228637695312,
      "step": 53086
    },
    {
      "epoch": 5.0626754760742186e-05,
      "step": 53086,
      "training_step_time": 6.707093954086304
    },
    {
      "epoch": 5.06277084350586e-05,
      "model_forward_time": 1.700594425201416,
      "step": 53087
    },
    {
      "epoch": 5.06277084350586e-05,
      "step": 53087,
      "training_step_time": 6.697669506072998
    },
    {
      "epoch": 5.0628662109375e-05,
      "model_forward_time": 1.7001488208770752,
      "step": 53088
    },
    {
      "epoch": 5.0628662109375e-05,
      "step": 53088,
      "training_step_time": 6.767374038696289
    },
    {
      "epoch": 5.0629615783691405e-05,
      "model_forward_time": 1.725161075592041,
      "step": 53089
    },
    {
      "epoch": 5.0629615783691405e-05,
      "step": 53089,
      "training_step_time": 7.5362467765808105
    },
    {
      "epoch": 5.0630569458007816e-05,
      "grad_norm": 0.0864764153957367,
      "learning_rate": 4.902616785675162e-06,
      "loss": 0.0533,
      "step": 53090
    },
    {
      "epoch": 5.0630569458007816e-05,
      "model_forward_time": 2.226069688796997,
      "step": 53090
    },
    {
      "dynamics_loss_avg": 0.03989538783207536,
      "epoch": 5.0630569458007816e-05,
      "step": 53090
    },
    {
      "action_loss_avg": 0.0013029870635364205,
      "epoch": 5.0630569458007816e-05,
      "step": 53090
    },
    {
      "epoch": 5.0630569458007816e-05,
      "step": 53090,
      "training_step_time": 9.073008060455322
    },
    {
      "epoch": 5.063152313232422e-05,
      "model_forward_time": 1.699638843536377,
      "step": 53091
    },
    {
      "epoch": 5.063152313232422e-05,
      "step": 53091,
      "training_step_time": 7.77821683883667
    },
    {
      "epoch": 5.0632476806640624e-05,
      "model_forward_time": 1.7030651569366455,
      "step": 53092
    },
    {
      "epoch": 5.0632476806640624e-05,
      "step": 53092,
      "training_step_time": 7.558115720748901
    },
    {
      "epoch": 5.0633430480957035e-05,
      "model_forward_time": 1.3250725269317627,
      "step": 53093
    },
    {
      "epoch": 5.0633430480957035e-05,
      "step": 53093,
      "training_step_time": 6.699460029602051
    },
    {
      "epoch": 5.063438415527344e-05,
      "model_forward_time": 1.7037618160247803,
      "step": 53094
    },
    {
      "epoch": 5.063438415527344e-05,
      "step": 53094,
      "training_step_time": 6.702622413635254
    },
    {
      "epoch": 5.063533782958984e-05,
      "model_forward_time": 1.7024004459381104,
      "step": 53095
    },
    {
      "epoch": 5.063533782958984e-05,
      "step": 53095,
      "training_step_time": 6.746126651763916
    },
    {
      "epoch": 5.063629150390625e-05,
      "model_forward_time": 1.6995608806610107,
      "step": 53096
    },
    {
      "epoch": 5.063629150390625e-05,
      "step": 53096,
      "training_step_time": 6.589014053344727
    },
    {
      "epoch": 5.063724517822266e-05,
      "model_forward_time": 1.6996870040893555,
      "step": 53097
    },
    {
      "epoch": 5.063724517822266e-05,
      "step": 53097,
      "training_step_time": 6.69304633140564
    },
    {
      "epoch": 5.063819885253906e-05,
      "model_forward_time": 1.7064778804779053,
      "step": 53098
    },
    {
      "epoch": 5.063819885253906e-05,
      "step": 53098,
      "training_step_time": 7.42449688911438
    },
    {
      "epoch": 5.0639152526855466e-05,
      "model_forward_time": 2.242945909500122,
      "step": 53099
    },
    {
      "epoch": 5.0639152526855466e-05,
      "step": 53099,
      "training_step_time": 10.224539995193481
    },
    {
      "epoch": 5.0640106201171876e-05,
      "grad_norm": 0.09442320466041565,
      "learning_rate": 4.900963634855505e-06,
      "loss": 0.056,
      "step": 53100
    },
    {
      "epoch": 5.0640106201171876e-05,
      "model_forward_time": 1.7050743103027344,
      "step": 53100
    },
    {
      "dynamics_loss_avg": 0.04449308104813099,
      "epoch": 5.0640106201171876e-05,
      "step": 53100
    },
    {
      "action_loss_avg": 0.0010189974680542945,
      "epoch": 5.0640106201171876e-05,
      "step": 53100
    },
    {
      "epoch": 5.0640106201171876e-05,
      "step": 53100,
      "training_step_time": 7.701136112213135
    },
    {
      "epoch": 5.064105987548828e-05,
      "model_forward_time": 1.702974796295166,
      "step": 53101
    },
    {
      "epoch": 5.064105987548828e-05,
      "step": 53101,
      "training_step_time": 7.61881947517395
    },
    {
      "epoch": 5.0642013549804684e-05,
      "model_forward_time": 1.703193187713623,
      "step": 53102
    },
    {
      "epoch": 5.0642013549804684e-05,
      "step": 53102,
      "training_step_time": 6.692134618759155
    },
    {
      "epoch": 5.0642967224121095e-05,
      "model_forward_time": 1.7045190334320068,
      "step": 53103
    },
    {
      "epoch": 5.0642967224121095e-05,
      "step": 53103,
      "training_step_time": 6.643297910690308
    },
    {
      "epoch": 5.06439208984375e-05,
      "model_forward_time": 1.7065415382385254,
      "step": 53104
    },
    {
      "epoch": 5.06439208984375e-05,
      "step": 53104,
      "training_step_time": 6.641149282455444
    },
    {
      "epoch": 5.06448745727539e-05,
      "model_forward_time": 1.6913392543792725,
      "step": 53105
    },
    {
      "epoch": 5.06448745727539e-05,
      "step": 53105,
      "training_step_time": 6.620940923690796
    },
    {
      "epoch": 5.0645828247070314e-05,
      "model_forward_time": 1.703340768814087,
      "step": 53106
    },
    {
      "epoch": 5.0645828247070314e-05,
      "step": 53106,
      "training_step_time": 6.656957626342773
    },
    {
      "epoch": 5.064678192138672e-05,
      "model_forward_time": 1.7011475563049316,
      "step": 53107
    },
    {
      "epoch": 5.064678192138672e-05,
      "step": 53107,
      "training_step_time": 7.355019807815552
    },
    {
      "epoch": 5.064773559570312e-05,
      "model_forward_time": 2.264512062072754,
      "step": 53108
    },
    {
      "epoch": 5.064773559570312e-05,
      "step": 53108,
      "training_step_time": 8.574265241622925
    },
    {
      "epoch": 5.064868927001953e-05,
      "model_forward_time": 1.7053730487823486,
      "step": 53109
    },
    {
      "epoch": 5.064868927001953e-05,
      "step": 53109,
      "training_step_time": 7.637312889099121
    },
    {
      "epoch": 5.064964294433594e-05,
      "grad_norm": 0.07830382883548737,
      "learning_rate": 4.899310494866318e-06,
      "loss": 0.0549,
      "step": 53110
    },
    {
      "epoch": 5.064964294433594e-05,
      "model_forward_time": 1.7005395889282227,
      "step": 53110
    },
    {
      "dynamics_loss_avg": 0.05280447155237198,
      "epoch": 5.064964294433594e-05,
      "step": 53110
    },
    {
      "action_loss_avg": 0.0020126227609580383,
      "epoch": 5.064964294433594e-05,
      "step": 53110
    },
    {
      "epoch": 5.064964294433594e-05,
      "step": 53110,
      "training_step_time": 7.567200183868408
    },
    {
      "epoch": 5.065059661865234e-05,
      "model_forward_time": 1.7031278610229492,
      "step": 53111
    },
    {
      "epoch": 5.065059661865234e-05,
      "step": 53111,
      "training_step_time": 6.672027826309204
    },
    {
      "epoch": 5.065155029296875e-05,
      "model_forward_time": 1.705826759338379,
      "step": 53112
    },
    {
      "epoch": 5.065155029296875e-05,
      "step": 53112,
      "training_step_time": 6.62431788444519
    },
    {
      "epoch": 5.0652503967285156e-05,
      "model_forward_time": 1.7119958400726318,
      "step": 53113
    },
    {
      "epoch": 5.0652503967285156e-05,
      "step": 53113,
      "training_step_time": 6.626958608627319
    },
    {
      "epoch": 5.065345764160156e-05,
      "model_forward_time": 1.7034289836883545,
      "step": 53114
    },
    {
      "epoch": 5.065345764160156e-05,
      "step": 53114,
      "training_step_time": 6.621780157089233
    },
    {
      "epoch": 5.065441131591797e-05,
      "model_forward_time": 1.702415943145752,
      "step": 53115
    },
    {
      "epoch": 5.065441131591797e-05,
      "step": 53115,
      "training_step_time": 6.671237230300903
    },
    {
      "epoch": 5.0655364990234375e-05,
      "model_forward_time": 1.6980054378509521,
      "step": 53116
    },
    {
      "epoch": 5.0655364990234375e-05,
      "step": 53116,
      "training_step_time": 6.976401090621948
    },
    {
      "epoch": 5.065631866455078e-05,
      "model_forward_time": 2.265629529953003,
      "step": 53117
    },
    {
      "epoch": 5.065631866455078e-05,
      "step": 53117,
      "training_step_time": 7.578347444534302
    },
    {
      "epoch": 5.065727233886719e-05,
      "model_forward_time": 1.7101950645446777,
      "step": 53118
    },
    {
      "epoch": 5.065727233886719e-05,
      "step": 53118,
      "training_step_time": 8.603925466537476
    },
    {
      "epoch": 5.0658226013183594e-05,
      "model_forward_time": 1.7038938999176025,
      "step": 53119
    },
    {
      "epoch": 5.0658226013183594e-05,
      "step": 53119,
      "training_step_time": 7.7315473556518555
    },
    {
      "epoch": 5.06591796875e-05,
      "grad_norm": 0.11533722281455994,
      "learning_rate": 4.8976573658883825e-06,
      "loss": 0.0538,
      "step": 53120
    },
    {
      "epoch": 5.06591796875e-05,
      "model_forward_time": 1.6927170753479004,
      "step": 53120
    },
    {
      "dynamics_loss_avg": 0.047744087688624856,
      "epoch": 5.06591796875e-05,
      "step": 53120
    },
    {
      "action_loss_avg": 0.0019209265301469713,
      "epoch": 5.06591796875e-05,
      "step": 53120
    },
    {
      "epoch": 5.06591796875e-05,
      "step": 53120,
      "training_step_time": 7.376637935638428
    },
    {
      "epoch": 5.066013336181641e-05,
      "model_forward_time": 1.7072103023529053,
      "step": 53121
    },
    {
      "epoch": 5.066013336181641e-05,
      "step": 53121,
      "training_step_time": 6.674043416976929
    },
    {
      "epoch": 5.066108703613281e-05,
      "model_forward_time": 0.9856228828430176,
      "step": 53122
    },
    {
      "epoch": 5.066108703613281e-05,
      "step": 53122,
      "training_step_time": 6.603532552719116
    },
    {
      "epoch": 5.0662040710449216e-05,
      "model_forward_time": 1.7121732234954834,
      "step": 53123
    },
    {
      "epoch": 5.0662040710449216e-05,
      "step": 53123,
      "training_step_time": 6.667367219924927
    },
    {
      "epoch": 5.066299438476563e-05,
      "model_forward_time": 1.6980335712432861,
      "step": 53124
    },
    {
      "epoch": 5.066299438476563e-05,
      "step": 53124,
      "training_step_time": 6.667447090148926
    },
    {
      "epoch": 5.066394805908203e-05,
      "model_forward_time": 1.7046091556549072,
      "step": 53125
    },
    {
      "epoch": 5.066394805908203e-05,
      "step": 53125,
      "training_step_time": 6.651720285415649
    },
    {
      "epoch": 5.0664901733398435e-05,
      "model_forward_time": 1.7316806316375732,
      "step": 53126
    },
    {
      "epoch": 5.0664901733398435e-05,
      "step": 53126,
      "training_step_time": 9.134829998016357
    },
    {
      "epoch": 5.0665855407714846e-05,
      "model_forward_time": 2.241189479827881,
      "step": 53127
    },
    {
      "epoch": 5.0665855407714846e-05,
      "step": 53127,
      "training_step_time": 10.394116640090942
    },
    {
      "epoch": 5.066680908203125e-05,
      "model_forward_time": 1.70841383934021,
      "step": 53128
    },
    {
      "epoch": 5.066680908203125e-05,
      "step": 53128,
      "training_step_time": 7.6607630252838135
    },
    {
      "epoch": 5.0667762756347654e-05,
      "model_forward_time": 1.7042927742004395,
      "step": 53129
    },
    {
      "epoch": 5.0667762756347654e-05,
      "step": 53129,
      "training_step_time": 7.5707197189331055
    },
    {
      "epoch": 5.0668716430664065e-05,
      "grad_norm": 0.0938306674361229,
      "learning_rate": 4.896004248102485e-06,
      "loss": 0.0541,
      "step": 53130
    },
    {
      "epoch": 5.0668716430664065e-05,
      "model_forward_time": 1.709125280380249,
      "step": 53130
    },
    {
      "dynamics_loss_avg": 0.04425557628273964,
      "epoch": 5.0668716430664065e-05,
      "step": 53130
    },
    {
      "action_loss_avg": 0.0023135428069508634,
      "epoch": 5.0668716430664065e-05,
      "step": 53130
    },
    {
      "epoch": 5.0668716430664065e-05,
      "step": 53130,
      "training_step_time": 6.661091327667236
    },
    {
      "epoch": 5.066967010498047e-05,
      "model_forward_time": 1.7012789249420166,
      "step": 53131
    },
    {
      "epoch": 5.066967010498047e-05,
      "step": 53131,
      "training_step_time": 6.653958559036255
    },
    {
      "epoch": 5.067062377929687e-05,
      "model_forward_time": 1.7111623287200928,
      "step": 53132
    },
    {
      "epoch": 5.067062377929687e-05,
      "step": 53132,
      "training_step_time": 6.663355588912964
    },
    {
      "epoch": 5.0671577453613284e-05,
      "model_forward_time": 1.7068779468536377,
      "step": 53133
    },
    {
      "epoch": 5.0671577453613284e-05,
      "step": 53133,
      "training_step_time": 6.66968035697937
    },
    {
      "epoch": 5.067253112792969e-05,
      "model_forward_time": 1.700146198272705,
      "step": 53134
    },
    {
      "epoch": 5.067253112792969e-05,
      "step": 53134,
      "training_step_time": 6.673314571380615
    },
    {
      "epoch": 5.067348480224609e-05,
      "model_forward_time": 1.3201758861541748,
      "step": 53135
    },
    {
      "epoch": 5.067348480224609e-05,
      "step": 53135,
      "training_step_time": 7.561836242675781
    },
    {
      "epoch": 5.06744384765625e-05,
      "model_forward_time": 2.2436695098876953,
      "step": 53136
    },
    {
      "epoch": 5.06744384765625e-05,
      "step": 53136,
      "training_step_time": 9.005248785018921
    },
    {
      "epoch": 5.067539215087891e-05,
      "model_forward_time": 1.705451488494873,
      "step": 53137
    },
    {
      "epoch": 5.067539215087891e-05,
      "step": 53137,
      "training_step_time": 7.668371200561523
    },
    {
      "epoch": 5.067634582519531e-05,
      "model_forward_time": 1.7003540992736816,
      "step": 53138
    },
    {
      "epoch": 5.067634582519531e-05,
      "step": 53138,
      "training_step_time": 7.7212135791778564
    },
    {
      "epoch": 5.067729949951172e-05,
      "model_forward_time": 1.7158691883087158,
      "step": 53139
    },
    {
      "epoch": 5.067729949951172e-05,
      "step": 53139,
      "training_step_time": 6.742716550827026
    },
    {
      "epoch": 5.0678253173828126e-05,
      "grad_norm": 0.08977507799863815,
      "learning_rate": 4.894351141689405e-06,
      "loss": 0.0572,
      "step": 53140
    },
    {
      "epoch": 5.0678253173828126e-05,
      "model_forward_time": 1.7083923816680908,
      "step": 53140
    },
    {
      "dynamics_loss_avg": 0.0660347431898117,
      "epoch": 5.0678253173828126e-05,
      "step": 53140
    },
    {
      "action_loss_avg": 0.001738058211049065,
      "epoch": 5.0678253173828126e-05,
      "step": 53140
    },
    {
      "epoch": 5.0678253173828126e-05,
      "step": 53140,
      "training_step_time": 6.742202997207642
    },
    {
      "epoch": 5.067920684814453e-05,
      "model_forward_time": 1.710461139678955,
      "step": 53141
    },
    {
      "epoch": 5.067920684814453e-05,
      "step": 53141,
      "training_step_time": 6.835011959075928
    },
    {
      "epoch": 5.068016052246094e-05,
      "model_forward_time": 1.708890438079834,
      "step": 53142
    },
    {
      "epoch": 5.068016052246094e-05,
      "step": 53142,
      "training_step_time": 6.836346626281738
    },
    {
      "epoch": 5.0681114196777344e-05,
      "model_forward_time": 1.7099781036376953,
      "step": 53143
    },
    {
      "epoch": 5.0681114196777344e-05,
      "step": 53143,
      "training_step_time": 6.818342447280884
    },
    {
      "epoch": 5.068206787109375e-05,
      "model_forward_time": 1.7051599025726318,
      "step": 53144
    },
    {
      "epoch": 5.068206787109375e-05,
      "step": 53144,
      "training_step_time": 8.372022867202759
    },
    {
      "epoch": 5.068302154541016e-05,
      "model_forward_time": 2.249911308288574,
      "step": 53145
    },
    {
      "epoch": 5.068302154541016e-05,
      "step": 53145,
      "training_step_time": 9.720913887023926
    },
    {
      "epoch": 5.068397521972656e-05,
      "model_forward_time": 1.689460039138794,
      "step": 53146
    },
    {
      "epoch": 5.068397521972656e-05,
      "step": 53146,
      "training_step_time": 7.758816480636597
    },
    {
      "epoch": 5.068492889404297e-05,
      "model_forward_time": 1.6978929042816162,
      "step": 53147
    },
    {
      "epoch": 5.068492889404297e-05,
      "step": 53147,
      "training_step_time": 7.885096788406372
    },
    {
      "epoch": 5.068588256835938e-05,
      "model_forward_time": 1.7033085823059082,
      "step": 53148
    },
    {
      "epoch": 5.068588256835938e-05,
      "step": 53148,
      "training_step_time": 6.832485675811768
    },
    {
      "epoch": 5.068683624267578e-05,
      "model_forward_time": 1.6976652145385742,
      "step": 53149
    },
    {
      "epoch": 5.068683624267578e-05,
      "step": 53149,
      "training_step_time": 6.8503007888793945
    },
    {
      "epoch": 5.0687789916992186e-05,
      "grad_norm": 0.10133615136146545,
      "learning_rate": 4.892698046829926e-06,
      "loss": 0.0566,
      "step": 53150
    },
    {
      "epoch": 5.0687789916992186e-05,
      "model_forward_time": 1.7131001949310303,
      "step": 53150
    },
    {
      "dynamics_loss_avg": 0.05240945089608431,
      "epoch": 5.0687789916992186e-05,
      "step": 53150
    },
    {
      "action_loss_avg": 0.0023521580034866927,
      "epoch": 5.0687789916992186e-05,
      "step": 53150
    },
    {
      "epoch": 5.0687789916992186e-05,
      "step": 53150,
      "training_step_time": 6.761597394943237
    },
    {
      "epoch": 5.06887435913086e-05,
      "model_forward_time": 1.6982338428497314,
      "step": 53151
    },
    {
      "epoch": 5.06887435913086e-05,
      "step": 53151,
      "training_step_time": 6.797598838806152
    },
    {
      "epoch": 5.0689697265625e-05,
      "model_forward_time": 1.6963844299316406,
      "step": 53152
    },
    {
      "epoch": 5.0689697265625e-05,
      "step": 53152,
      "training_step_time": 6.74198842048645
    },
    {
      "epoch": 5.0690650939941405e-05,
      "model_forward_time": 1.6981098651885986,
      "step": 53153
    },
    {
      "epoch": 5.0690650939941405e-05,
      "step": 53153,
      "training_step_time": 7.764291763305664
    },
    {
      "epoch": 5.0691604614257816e-05,
      "model_forward_time": 2.271939992904663,
      "step": 53154
    },
    {
      "epoch": 5.0691604614257816e-05,
      "step": 53154,
      "training_step_time": 10.138562202453613
    },
    {
      "epoch": 5.069255828857422e-05,
      "model_forward_time": 1.6997628211975098,
      "step": 53155
    },
    {
      "epoch": 5.069255828857422e-05,
      "step": 53155,
      "training_step_time": 8.8782377243042
    },
    {
      "epoch": 5.0693511962890624e-05,
      "model_forward_time": 1.7101352214813232,
      "step": 53156
    },
    {
      "epoch": 5.0693511962890624e-05,
      "step": 53156,
      "training_step_time": 8.906284809112549
    },
    {
      "epoch": 5.069446563720703e-05,
      "model_forward_time": 1.7043066024780273,
      "step": 53157
    },
    {
      "epoch": 5.069446563720703e-05,
      "step": 53157,
      "training_step_time": 6.794736862182617
    },
    {
      "epoch": 5.069541931152344e-05,
      "model_forward_time": 1.7069120407104492,
      "step": 53158
    },
    {
      "epoch": 5.069541931152344e-05,
      "step": 53158,
      "training_step_time": 6.7692201137542725
    },
    {
      "epoch": 5.069637298583984e-05,
      "model_forward_time": 1.7111339569091797,
      "step": 53159
    },
    {
      "epoch": 5.069637298583984e-05,
      "step": 53159,
      "training_step_time": 6.7800517082214355
    },
    {
      "epoch": 5.069732666015625e-05,
      "grad_norm": 0.10124655067920685,
      "learning_rate": 4.891044963704828e-06,
      "loss": 0.0544,
      "step": 53160
    },
    {
      "epoch": 5.069732666015625e-05,
      "model_forward_time": 1.6893763542175293,
      "step": 53160
    },
    {
      "dynamics_loss_avg": 0.0410288117825985,
      "epoch": 5.069732666015625e-05,
      "step": 53160
    },
    {
      "action_loss_avg": 0.001101040141656995,
      "epoch": 5.069732666015625e-05,
      "step": 53160
    },
    {
      "epoch": 5.069732666015625e-05,
      "step": 53160,
      "training_step_time": 6.802062511444092
    },
    {
      "epoch": 5.069828033447266e-05,
      "model_forward_time": 1.6986870765686035,
      "step": 53161
    },
    {
      "epoch": 5.069828033447266e-05,
      "step": 53161,
      "training_step_time": 6.767478942871094
    },
    {
      "epoch": 5.069923400878906e-05,
      "model_forward_time": 1.6964266300201416,
      "step": 53162
    },
    {
      "epoch": 5.069923400878906e-05,
      "step": 53162,
      "training_step_time": 9.187300682067871
    },
    {
      "epoch": 5.0700187683105466e-05,
      "model_forward_time": 2.265941858291626,
      "step": 53163
    },
    {
      "epoch": 5.0700187683105466e-05,
      "step": 53163,
      "training_step_time": 9.782068252563477
    },
    {
      "epoch": 5.0701141357421876e-05,
      "model_forward_time": 1.699812889099121,
      "step": 53164
    },
    {
      "epoch": 5.0701141357421876e-05,
      "step": 53164,
      "training_step_time": 9.152890920639038
    },
    {
      "epoch": 5.070209503173828e-05,
      "model_forward_time": 1.6999084949493408,
      "step": 53165
    },
    {
      "epoch": 5.070209503173828e-05,
      "step": 53165,
      "training_step_time": 7.839817523956299
    },
    {
      "epoch": 5.0703048706054684e-05,
      "model_forward_time": 1.703235149383545,
      "step": 53166
    },
    {
      "epoch": 5.0703048706054684e-05,
      "step": 53166,
      "training_step_time": 7.6142189502716064
    },
    {
      "epoch": 5.0704002380371095e-05,
      "model_forward_time": 1.7056927680969238,
      "step": 53167
    },
    {
      "epoch": 5.0704002380371095e-05,
      "step": 53167,
      "training_step_time": 6.710932016372681
    },
    {
      "epoch": 5.07049560546875e-05,
      "model_forward_time": 1.7036759853363037,
      "step": 53168
    },
    {
      "epoch": 5.07049560546875e-05,
      "step": 53168,
      "training_step_time": 6.721622943878174
    },
    {
      "epoch": 5.07059097290039e-05,
      "model_forward_time": 1.7052710056304932,
      "step": 53169
    },
    {
      "epoch": 5.07059097290039e-05,
      "step": 53169,
      "training_step_time": 6.743706941604614
    },
    {
      "epoch": 5.0706863403320314e-05,
      "grad_norm": 0.13378150761127472,
      "learning_rate": 4.889391892494888e-06,
      "loss": 0.0584,
      "step": 53170
    },
    {
      "epoch": 5.0706863403320314e-05,
      "model_forward_time": 1.7009129524230957,
      "step": 53170
    },
    {
      "dynamics_loss_avg": 0.06689237523823977,
      "epoch": 5.0706863403320314e-05,
      "step": 53170
    },
    {
      "action_loss_avg": 0.0020628361206036063,
      "epoch": 5.0706863403320314e-05,
      "step": 53170
    },
    {
      "epoch": 5.0706863403320314e-05,
      "step": 53170,
      "training_step_time": 6.7646355628967285
    },
    {
      "epoch": 5.070781707763672e-05,
      "model_forward_time": 1.7043516635894775,
      "step": 53171
    },
    {
      "epoch": 5.070781707763672e-05,
      "step": 53171,
      "training_step_time": 8.52400827407837
    },
    {
      "epoch": 5.070877075195312e-05,
      "model_forward_time": 1.721193790435791,
      "step": 53172
    },
    {
      "epoch": 5.070877075195312e-05,
      "step": 53172,
      "training_step_time": 9.274436950683594
    },
    {
      "epoch": 5.070972442626953e-05,
      "model_forward_time": 1.712264060974121,
      "step": 53173
    },
    {
      "epoch": 5.070972442626953e-05,
      "step": 53173,
      "training_step_time": 9.607440948486328
    },
    {
      "epoch": 5.071067810058594e-05,
      "model_forward_time": 1.7018840312957764,
      "step": 53174
    },
    {
      "epoch": 5.071067810058594e-05,
      "step": 53174,
      "training_step_time": 7.630359172821045
    },
    {
      "epoch": 5.071163177490234e-05,
      "model_forward_time": 1.7043697834014893,
      "step": 53175
    },
    {
      "epoch": 5.071163177490234e-05,
      "step": 53175,
      "training_step_time": 7.680305242538452
    },
    {
      "epoch": 5.071258544921875e-05,
      "model_forward_time": 1.71372652053833,
      "step": 53176
    },
    {
      "epoch": 5.071258544921875e-05,
      "step": 53176,
      "training_step_time": 6.759015083312988
    },
    {
      "epoch": 5.0713539123535156e-05,
      "model_forward_time": 1.700087547302246,
      "step": 53177
    },
    {
      "epoch": 5.0713539123535156e-05,
      "step": 53177,
      "training_step_time": 6.688776969909668
    },
    {
      "epoch": 5.071449279785156e-05,
      "model_forward_time": 1.7082769870758057,
      "step": 53178
    },
    {
      "epoch": 5.071449279785156e-05,
      "step": 53178,
      "training_step_time": 6.721731185913086
    },
    {
      "epoch": 5.071544647216797e-05,
      "model_forward_time": 1.7066576480865479,
      "step": 53179
    },
    {
      "epoch": 5.071544647216797e-05,
      "step": 53179,
      "training_step_time": 6.6806321144104
    },
    {
      "epoch": 5.0716400146484375e-05,
      "grad_norm": 0.1275882124900818,
      "learning_rate": 4.887738833380883e-06,
      "loss": 0.06,
      "step": 53180
    },
    {
      "epoch": 5.0716400146484375e-05,
      "model_forward_time": 1.6996197700500488,
      "step": 53180
    },
    {
      "dynamics_loss_avg": 0.04770464468747378,
      "epoch": 5.0716400146484375e-05,
      "step": 53180
    },
    {
      "action_loss_avg": 0.0013965388265205548,
      "epoch": 5.0716400146484375e-05,
      "step": 53180
    },
    {
      "epoch": 5.0716400146484375e-05,
      "step": 53180,
      "training_step_time": 7.550105571746826
    },
    {
      "epoch": 5.071735382080078e-05,
      "model_forward_time": 1.705030918121338,
      "step": 53181
    },
    {
      "epoch": 5.071735382080078e-05,
      "step": 53181,
      "training_step_time": 9.178850889205933
    },
    {
      "epoch": 5.071830749511719e-05,
      "model_forward_time": 2.272803544998169,
      "step": 53182
    },
    {
      "epoch": 5.071830749511719e-05,
      "step": 53182,
      "training_step_time": 8.447727680206299
    },
    {
      "epoch": 5.0719261169433594e-05,
      "model_forward_time": 1.7120625972747803,
      "step": 53183
    },
    {
      "epoch": 5.0719261169433594e-05,
      "step": 53183,
      "training_step_time": 7.6094300746917725
    },
    {
      "epoch": 5.072021484375e-05,
      "model_forward_time": 1.708475112915039,
      "step": 53184
    },
    {
      "epoch": 5.072021484375e-05,
      "step": 53184,
      "training_step_time": 7.514681339263916
    },
    {
      "epoch": 5.072116851806641e-05,
      "model_forward_time": 1.702723741531372,
      "step": 53185
    },
    {
      "epoch": 5.072116851806641e-05,
      "step": 53185,
      "training_step_time": 7.690474271774292
    },
    {
      "epoch": 5.072212219238281e-05,
      "model_forward_time": 1.7104122638702393,
      "step": 53186
    },
    {
      "epoch": 5.072212219238281e-05,
      "step": 53186,
      "training_step_time": 6.693002939224243
    },
    {
      "epoch": 5.0723075866699216e-05,
      "model_forward_time": 1.7138521671295166,
      "step": 53187
    },
    {
      "epoch": 5.0723075866699216e-05,
      "step": 53187,
      "training_step_time": 6.642658233642578
    },
    {
      "epoch": 5.072402954101563e-05,
      "model_forward_time": 1.706101655960083,
      "step": 53188
    },
    {
      "epoch": 5.072402954101563e-05,
      "step": 53188,
      "training_step_time": 6.6553874015808105
    },
    {
      "epoch": 5.072498321533203e-05,
      "model_forward_time": 1.700584888458252,
      "step": 53189
    },
    {
      "epoch": 5.072498321533203e-05,
      "step": 53189,
      "training_step_time": 7.544269561767578
    },
    {
      "epoch": 5.0725936889648435e-05,
      "grad_norm": 0.11167360097169876,
      "learning_rate": 4.886085786543593e-06,
      "loss": 0.0616,
      "step": 53190
    },
    {
      "epoch": 5.0725936889648435e-05,
      "model_forward_time": 1.6925158500671387,
      "step": 53190
    },
    {
      "dynamics_loss_avg": 0.056469501741230485,
      "epoch": 5.0725936889648435e-05,
      "step": 53190
    },
    {
      "action_loss_avg": 0.0018649122852366419,
      "epoch": 5.0725936889648435e-05,
      "step": 53190
    },
    {
      "epoch": 5.0725936889648435e-05,
      "step": 53190,
      "training_step_time": 7.587798118591309
    },
    {
      "epoch": 5.0726890563964846e-05,
      "model_forward_time": 2.2851622104644775,
      "step": 53191
    },
    {
      "epoch": 5.0726890563964846e-05,
      "step": 53191,
      "training_step_time": 10.448176145553589
    },
    {
      "epoch": 5.072784423828125e-05,
      "model_forward_time": 1.6987004280090332,
      "step": 53192
    },
    {
      "epoch": 5.072784423828125e-05,
      "step": 53192,
      "training_step_time": 8.249608755111694
    },
    {
      "epoch": 5.0728797912597654e-05,
      "model_forward_time": 1.711763620376587,
      "step": 53193
    },
    {
      "epoch": 5.0728797912597654e-05,
      "step": 53193,
      "training_step_time": 7.391390323638916
    },
    {
      "epoch": 5.0729751586914065e-05,
      "model_forward_time": 1.7082595825195312,
      "step": 53194
    },
    {
      "epoch": 5.0729751586914065e-05,
      "step": 53194,
      "training_step_time": 7.722756624221802
    },
    {
      "epoch": 5.073070526123047e-05,
      "model_forward_time": 1.7057828903198242,
      "step": 53195
    },
    {
      "epoch": 5.073070526123047e-05,
      "step": 53195,
      "training_step_time": 6.651555061340332
    },
    {
      "epoch": 5.073165893554687e-05,
      "model_forward_time": 1.7086236476898193,
      "step": 53196
    },
    {
      "epoch": 5.073165893554687e-05,
      "step": 53196,
      "training_step_time": 6.737486839294434
    },
    {
      "epoch": 5.0732612609863284e-05,
      "model_forward_time": 1.7166810035705566,
      "step": 53197
    },
    {
      "epoch": 5.0732612609863284e-05,
      "step": 53197,
      "training_step_time": 6.714553594589233
    },
    {
      "epoch": 5.073356628417969e-05,
      "model_forward_time": 1.7094814777374268,
      "step": 53198
    },
    {
      "epoch": 5.073356628417969e-05,
      "step": 53198,
      "training_step_time": 7.632341146469116
    },
    {
      "epoch": 5.073451995849609e-05,
      "model_forward_time": 1.702214002609253,
      "step": 53199
    },
    {
      "epoch": 5.073451995849609e-05,
      "step": 53199,
      "training_step_time": 7.6768317222595215
    },
    {
      "epoch": 5.07354736328125e-05,
      "grad_norm": 0.1042608991265297,
      "learning_rate": 4.8844327521637895e-06,
      "loss": 0.0586,
      "step": 53200
    },
    {
      "epoch": 5.07354736328125e-05,
      "model_forward_time": 2.2859201431274414,
      "step": 53200
    },
    {
      "dynamics_loss_avg": 0.05257473085075617,
      "epoch": 5.07354736328125e-05,
      "step": 53200
    },
    {
      "action_loss_avg": 0.0017850755015388132,
      "epoch": 5.07354736328125e-05,
      "step": 53200
    },
    {
      "epoch": 5.07354736328125e-05,
      "step": 53200,
      "training_step_time": 7.746708869934082
    },
    {
      "epoch": 5.073642730712891e-05,
      "model_forward_time": 1.7097060680389404,
      "step": 53201
    },
    {
      "epoch": 5.073642730712891e-05,
      "step": 53201,
      "training_step_time": 8.349570035934448
    },
    {
      "epoch": 5.073738098144531e-05,
      "model_forward_time": 1.7069785594940186,
      "step": 53202
    },
    {
      "epoch": 5.073738098144531e-05,
      "step": 53202,
      "training_step_time": 7.545022487640381
    },
    {
      "epoch": 5.073833465576172e-05,
      "model_forward_time": 1.34108567237854,
      "step": 53203
    },
    {
      "epoch": 5.073833465576172e-05,
      "step": 53203,
      "training_step_time": 7.599687576293945
    },
    {
      "epoch": 5.0739288330078126e-05,
      "model_forward_time": 1.7189626693725586,
      "step": 53204
    },
    {
      "epoch": 5.0739288330078126e-05,
      "step": 53204,
      "training_step_time": 6.659604787826538
    },
    {
      "epoch": 5.074024200439453e-05,
      "model_forward_time": 1.708754539489746,
      "step": 53205
    },
    {
      "epoch": 5.074024200439453e-05,
      "step": 53205,
      "training_step_time": 6.597979545593262
    },
    {
      "epoch": 5.074119567871094e-05,
      "model_forward_time": 1.7073287963867188,
      "step": 53206
    },
    {
      "epoch": 5.074119567871094e-05,
      "step": 53206,
      "training_step_time": 6.626260995864868
    },
    {
      "epoch": 5.0742149353027344e-05,
      "model_forward_time": 1.7076821327209473,
      "step": 53207
    },
    {
      "epoch": 5.0742149353027344e-05,
      "step": 53207,
      "training_step_time": 8.185861110687256
    },
    {
      "epoch": 5.074310302734375e-05,
      "model_forward_time": 1.710400104522705,
      "step": 53208
    },
    {
      "epoch": 5.074310302734375e-05,
      "step": 53208,
      "training_step_time": 9.113319635391235
    },
    {
      "epoch": 5.074405670166016e-05,
      "model_forward_time": 2.260317802429199,
      "step": 53209
    },
    {
      "epoch": 5.074405670166016e-05,
      "step": 53209,
      "training_step_time": 10.335801362991333
    },
    {
      "epoch": 5.074501037597656e-05,
      "grad_norm": 0.16698437929153442,
      "learning_rate": 4.882779730422247e-06,
      "loss": 0.0612,
      "step": 53210
    },
    {
      "epoch": 5.074501037597656e-05,
      "model_forward_time": 1.3409149646759033,
      "step": 53210
    },
    {
      "dynamics_loss_avg": 0.06001436244696379,
      "epoch": 5.074501037597656e-05,
      "step": 53210
    },
    {
      "action_loss_avg": 0.0012268968537682667,
      "epoch": 5.074501037597656e-05,
      "step": 53210
    },
    {
      "epoch": 5.074501037597656e-05,
      "step": 53210,
      "training_step_time": 8.71609616279602
    },
    {
      "epoch": 5.074596405029297e-05,
      "model_forward_time": 1.7174506187438965,
      "step": 53211
    },
    {
      "epoch": 5.074596405029297e-05,
      "step": 53211,
      "training_step_time": 7.684599161148071
    },
    {
      "epoch": 5.074691772460938e-05,
      "model_forward_time": 1.7083098888397217,
      "step": 53212
    },
    {
      "epoch": 5.074691772460938e-05,
      "step": 53212,
      "training_step_time": 7.833020925521851
    },
    {
      "epoch": 5.074787139892578e-05,
      "model_forward_time": 1.700225830078125,
      "step": 53213
    },
    {
      "epoch": 5.074787139892578e-05,
      "step": 53213,
      "training_step_time": 7.497212886810303
    },
    {
      "epoch": 5.0748825073242186e-05,
      "model_forward_time": 1.7169277667999268,
      "step": 53214
    },
    {
      "epoch": 5.0748825073242186e-05,
      "step": 53214,
      "training_step_time": 6.768844842910767
    },
    {
      "epoch": 5.07497787475586e-05,
      "model_forward_time": 1.7362213134765625,
      "step": 53215
    },
    {
      "epoch": 5.07497787475586e-05,
      "step": 53215,
      "training_step_time": 6.726625442504883
    },
    {
      "epoch": 5.0750732421875e-05,
      "model_forward_time": 1.713801622390747,
      "step": 53216
    },
    {
      "epoch": 5.0750732421875e-05,
      "step": 53216,
      "training_step_time": 8.309168100357056
    },
    {
      "epoch": 5.0751686096191405e-05,
      "model_forward_time": 1.7085671424865723,
      "step": 53217
    },
    {
      "epoch": 5.0751686096191405e-05,
      "step": 53217,
      "training_step_time": 7.698297739028931
    },
    {
      "epoch": 5.0752639770507816e-05,
      "model_forward_time": 1.718759298324585,
      "step": 53218
    },
    {
      "epoch": 5.0752639770507816e-05,
      "step": 53218,
      "training_step_time": 9.657739639282227
    },
    {
      "epoch": 5.075359344482422e-05,
      "model_forward_time": 2.256868839263916,
      "step": 53219
    },
    {
      "epoch": 5.075359344482422e-05,
      "step": 53219,
      "training_step_time": 8.453445196151733
    },
    {
      "epoch": 5.0754547119140624e-05,
      "grad_norm": 0.09961792826652527,
      "learning_rate": 4.881126721499736e-06,
      "loss": 0.0618,
      "step": 53220
    },
    {
      "epoch": 5.0754547119140624e-05,
      "model_forward_time": 1.7191495895385742,
      "step": 53220
    },
    {
      "dynamics_loss_avg": 0.056479760259389875,
      "epoch": 5.0754547119140624e-05,
      "step": 53220
    },
    {
      "action_loss_avg": 0.0014155986660625785,
      "epoch": 5.0754547119140624e-05,
      "step": 53220
    },
    {
      "epoch": 5.0754547119140624e-05,
      "step": 53220,
      "training_step_time": 7.592779874801636
    },
    {
      "epoch": 5.075550079345703e-05,
      "model_forward_time": 1.7247107028961182,
      "step": 53221
    },
    {
      "epoch": 5.075550079345703e-05,
      "step": 53221,
      "training_step_time": 7.737670660018921
    },
    {
      "epoch": 5.075645446777344e-05,
      "model_forward_time": 1.7120258808135986,
      "step": 53222
    },
    {
      "epoch": 5.075645446777344e-05,
      "step": 53222,
      "training_step_time": 7.493319034576416
    },
    {
      "epoch": 5.075740814208984e-05,
      "model_forward_time": 1.7211201190948486,
      "step": 53223
    },
    {
      "epoch": 5.075740814208984e-05,
      "step": 53223,
      "training_step_time": 6.730259895324707
    },
    {
      "epoch": 5.075836181640625e-05,
      "model_forward_time": 1.720198631286621,
      "step": 53224
    },
    {
      "epoch": 5.075836181640625e-05,
      "step": 53224,
      "training_step_time": 6.765751838684082
    },
    {
      "epoch": 5.075931549072266e-05,
      "model_forward_time": 1.7121543884277344,
      "step": 53225
    },
    {
      "epoch": 5.075931549072266e-05,
      "step": 53225,
      "training_step_time": 7.6206159591674805
    },
    {
      "epoch": 5.076026916503906e-05,
      "model_forward_time": 1.712930679321289,
      "step": 53226
    },
    {
      "epoch": 5.076026916503906e-05,
      "step": 53226,
      "training_step_time": 7.658645868301392
    },
    {
      "epoch": 5.0761222839355466e-05,
      "model_forward_time": 1.7149474620819092,
      "step": 53227
    },
    {
      "epoch": 5.0761222839355466e-05,
      "step": 53227,
      "training_step_time": 9.678757190704346
    },
    {
      "epoch": 5.0762176513671876e-05,
      "model_forward_time": 2.2704269886016846,
      "step": 53228
    },
    {
      "epoch": 5.0762176513671876e-05,
      "step": 53228,
      "training_step_time": 11.685377836227417
    },
    {
      "epoch": 5.076313018798828e-05,
      "model_forward_time": 1.716252326965332,
      "step": 53229
    },
    {
      "epoch": 5.076313018798828e-05,
      "step": 53229,
      "training_step_time": 9.344477653503418
    },
    {
      "epoch": 5.0764083862304685e-05,
      "grad_norm": 0.11212984472513199,
      "learning_rate": 4.879473725577029e-06,
      "loss": 0.0558,
      "step": 53230
    },
    {
      "epoch": 5.0764083862304685e-05,
      "model_forward_time": 1.701354742050171,
      "step": 53230
    },
    {
      "dynamics_loss_avg": 0.053283934481441975,
      "epoch": 5.0764083862304685e-05,
      "step": 53230
    },
    {
      "action_loss_avg": 0.0017375193478073926,
      "epoch": 5.0764083862304685e-05,
      "step": 53230
    },
    {
      "epoch": 5.0764083862304685e-05,
      "step": 53230,
      "training_step_time": 7.506048202514648
    },
    {
      "epoch": 5.0765037536621095e-05,
      "model_forward_time": 1.7165441513061523,
      "step": 53231
    },
    {
      "epoch": 5.0765037536621095e-05,
      "step": 53231,
      "training_step_time": 7.785917520523071
    },
    {
      "epoch": 5.07659912109375e-05,
      "model_forward_time": 1.71223783493042,
      "step": 53232
    },
    {
      "epoch": 5.07659912109375e-05,
      "step": 53232,
      "training_step_time": 6.77772855758667
    },
    {
      "epoch": 5.07669448852539e-05,
      "model_forward_time": 1.7142465114593506,
      "step": 53233
    },
    {
      "epoch": 5.07669448852539e-05,
      "step": 53233,
      "training_step_time": 6.757540464401245
    },
    {
      "epoch": 5.0767898559570314e-05,
      "model_forward_time": 1.3379433155059814,
      "step": 53234
    },
    {
      "epoch": 5.0767898559570314e-05,
      "step": 53234,
      "training_step_time": 7.592836856842041
    },
    {
      "epoch": 5.076885223388672e-05,
      "model_forward_time": 1.7156496047973633,
      "step": 53235
    },
    {
      "epoch": 5.076885223388672e-05,
      "step": 53235,
      "training_step_time": 7.588840961456299
    },
    {
      "epoch": 5.076980590820312e-05,
      "model_forward_time": 1.7128496170043945,
      "step": 53236
    },
    {
      "epoch": 5.076980590820312e-05,
      "step": 53236,
      "training_step_time": 7.596423625946045
    },
    {
      "epoch": 5.077075958251953e-05,
      "model_forward_time": 2.265132188796997,
      "step": 53237
    },
    {
      "epoch": 5.077075958251953e-05,
      "step": 53237,
      "training_step_time": 8.804991483688354
    },
    {
      "epoch": 5.077171325683594e-05,
      "model_forward_time": 1.7170288562774658,
      "step": 53238
    },
    {
      "epoch": 5.077171325683594e-05,
      "step": 53238,
      "training_step_time": 8.408688306808472
    },
    {
      "epoch": 5.077266693115234e-05,
      "model_forward_time": 1.7172417640686035,
      "step": 53239
    },
    {
      "epoch": 5.077266693115234e-05,
      "step": 53239,
      "training_step_time": 7.608357906341553
    },
    {
      "epoch": 5.077362060546875e-05,
      "grad_norm": 0.11299606412649155,
      "learning_rate": 4.877820742834891e-06,
      "loss": 0.0589,
      "step": 53240
    },
    {
      "epoch": 5.077362060546875e-05,
      "model_forward_time": 1.7178585529327393,
      "step": 53240
    },
    {
      "dynamics_loss_avg": 0.04913394451141358,
      "epoch": 5.077362060546875e-05,
      "step": 53240
    },
    {
      "action_loss_avg": 0.001966516626998782,
      "epoch": 5.077362060546875e-05,
      "step": 53240
    },
    {
      "epoch": 5.077362060546875e-05,
      "step": 53240,
      "training_step_time": 7.846877813339233
    },
    {
      "epoch": 5.0774574279785156e-05,
      "model_forward_time": 1.7149674892425537,
      "step": 53241
    },
    {
      "epoch": 5.0774574279785156e-05,
      "step": 53241,
      "training_step_time": 6.7568559646606445
    },
    {
      "epoch": 5.077552795410156e-05,
      "model_forward_time": 1.715625286102295,
      "step": 53242
    },
    {
      "epoch": 5.077552795410156e-05,
      "step": 53242,
      "training_step_time": 6.7742578983306885
    },
    {
      "epoch": 5.077648162841797e-05,
      "model_forward_time": 1.7174723148345947,
      "step": 53243
    },
    {
      "epoch": 5.077648162841797e-05,
      "step": 53243,
      "training_step_time": 7.62191367149353
    },
    {
      "epoch": 5.0777435302734375e-05,
      "model_forward_time": 1.716259241104126,
      "step": 53244
    },
    {
      "epoch": 5.0777435302734375e-05,
      "step": 53244,
      "training_step_time": 10.01755952835083
    },
    {
      "epoch": 5.077838897705078e-05,
      "model_forward_time": 1.7156174182891846,
      "step": 53245
    },
    {
      "epoch": 5.077838897705078e-05,
      "step": 53245,
      "training_step_time": 7.680914878845215
    },
    {
      "epoch": 5.077934265136719e-05,
      "model_forward_time": 2.2509982585906982,
      "step": 53246
    },
    {
      "epoch": 5.077934265136719e-05,
      "step": 53246,
      "training_step_time": 8.926515102386475
    },
    {
      "epoch": 5.0780296325683594e-05,
      "model_forward_time": 1.7218778133392334,
      "step": 53247
    },
    {
      "epoch": 5.0780296325683594e-05,
      "step": 53247,
      "training_step_time": 8.804079055786133
    },
    {
      "epoch": 5.078125e-05,
      "model_forward_time": 1.7130649089813232,
      "step": 53248
    },
    {
      "epoch": 5.078125e-05,
      "step": 53248,
      "training_step_time": 7.748722076416016
    },
    {
      "epoch": 5.078220367431641e-05,
      "model_forward_time": 1.716017246246338,
      "step": 53249
    },
    {
      "epoch": 5.078220367431641e-05,
      "step": 53249,
      "training_step_time": 7.707255601882935
    },
    {
      "epoch": 5.078315734863281e-05,
      "grad_norm": 0.12737400829792023,
      "learning_rate": 4.876167773454096e-06,
      "loss": 0.0611,
      "step": 53250
    },
    {
      "epoch": 5.078315734863281e-05,
      "model_forward_time": 1.717421054840088,
      "step": 53250
    },
    {
      "dynamics_loss_avg": 0.04593812674283981,
      "epoch": 5.078315734863281e-05,
      "step": 53250
    },
    {
      "action_loss_avg": 0.0011154886335134507,
      "epoch": 5.078315734863281e-05,
      "step": 53250
    },
    {
      "epoch": 5.078315734863281e-05,
      "step": 53250,
      "training_step_time": 7.622459173202515
    },
    {
      "epoch": 5.0784111022949217e-05,
      "model_forward_time": 1.7131941318511963,
      "step": 53251
    },
    {
      "epoch": 5.0784111022949217e-05,
      "step": 53251,
      "training_step_time": 6.736006498336792
    },
    {
      "epoch": 5.078506469726563e-05,
      "model_forward_time": 1.71516752243042,
      "step": 53252
    },
    {
      "epoch": 5.078506469726563e-05,
      "step": 53252,
      "training_step_time": 7.570108652114868
    },
    {
      "epoch": 5.078601837158203e-05,
      "model_forward_time": 1.7235033512115479,
      "step": 53253
    },
    {
      "epoch": 5.078601837158203e-05,
      "step": 53253,
      "training_step_time": 7.568838119506836
    },
    {
      "epoch": 5.0786972045898435e-05,
      "model_forward_time": 1.7143614292144775,
      "step": 53254
    },
    {
      "epoch": 5.0786972045898435e-05,
      "step": 53254,
      "training_step_time": 8.481139659881592
    },
    {
      "epoch": 5.0787925720214846e-05,
      "model_forward_time": 1.7254295349121094,
      "step": 53255
    },
    {
      "epoch": 5.0787925720214846e-05,
      "step": 53255,
      "training_step_time": 7.637846231460571
    },
    {
      "epoch": 5.078887939453125e-05,
      "model_forward_time": 2.2604949474334717,
      "step": 53256
    },
    {
      "epoch": 5.078887939453125e-05,
      "step": 53256,
      "training_step_time": 9.052912950515747
    },
    {
      "epoch": 5.0789833068847654e-05,
      "model_forward_time": 1.7149145603179932,
      "step": 53257
    },
    {
      "epoch": 5.0789833068847654e-05,
      "step": 53257,
      "training_step_time": 7.755523443222046
    },
    {
      "epoch": 5.0790786743164065e-05,
      "model_forward_time": 1.7169826030731201,
      "step": 53258
    },
    {
      "epoch": 5.0790786743164065e-05,
      "step": 53258,
      "training_step_time": 7.63361930847168
    },
    {
      "epoch": 5.079174041748047e-05,
      "model_forward_time": 1.7193162441253662,
      "step": 53259
    },
    {
      "epoch": 5.079174041748047e-05,
      "step": 53259,
      "training_step_time": 7.755606174468994
    },
    {
      "epoch": 5.079269409179687e-05,
      "grad_norm": 0.13764722645282745,
      "learning_rate": 4.874514817615407e-06,
      "loss": 0.0615,
      "step": 53260
    },
    {
      "epoch": 5.079269409179687e-05,
      "model_forward_time": 1.714954137802124,
      "step": 53260
    },
    {
      "dynamics_loss_avg": 0.060584304854273796,
      "epoch": 5.079269409179687e-05,
      "step": 53260
    },
    {
      "action_loss_avg": 0.0015823496389202774,
      "epoch": 5.079269409179687e-05,
      "step": 53260
    },
    {
      "epoch": 5.079269409179687e-05,
      "step": 53260,
      "training_step_time": 6.810250997543335
    },
    {
      "epoch": 5.0793647766113284e-05,
      "model_forward_time": 1.712829351425171,
      "step": 53261
    },
    {
      "epoch": 5.0793647766113284e-05,
      "step": 53261,
      "training_step_time": 7.67101263999939
    },
    {
      "epoch": 5.079460144042969e-05,
      "model_forward_time": 1.7175920009613037,
      "step": 53262
    },
    {
      "epoch": 5.079460144042969e-05,
      "step": 53262,
      "training_step_time": 8.648314476013184
    },
    {
      "epoch": 5.079555511474609e-05,
      "model_forward_time": 1.7089998722076416,
      "step": 53263
    },
    {
      "epoch": 5.079555511474609e-05,
      "step": 53263,
      "training_step_time": 7.933989763259888
    },
    {
      "epoch": 5.07965087890625e-05,
      "model_forward_time": 1.721616506576538,
      "step": 53264
    },
    {
      "epoch": 5.07965087890625e-05,
      "step": 53264,
      "training_step_time": 7.6813435554504395
    },
    {
      "epoch": 5.079746246337891e-05,
      "model_forward_time": 2.2475693225860596,
      "step": 53265
    },
    {
      "epoch": 5.079746246337891e-05,
      "step": 53265,
      "training_step_time": 7.812215805053711
    },
    {
      "epoch": 5.079841613769531e-05,
      "model_forward_time": 1.7186682224273682,
      "step": 53266
    },
    {
      "epoch": 5.079841613769531e-05,
      "step": 53266,
      "training_step_time": 7.697024822235107
    },
    {
      "epoch": 5.079936981201172e-05,
      "model_forward_time": 1.7193515300750732,
      "step": 53267
    },
    {
      "epoch": 5.079936981201172e-05,
      "step": 53267,
      "training_step_time": 7.6866395473480225
    },
    {
      "epoch": 5.0800323486328126e-05,
      "model_forward_time": 1.714458703994751,
      "step": 53268
    },
    {
      "epoch": 5.0800323486328126e-05,
      "step": 53268,
      "training_step_time": 7.769325017929077
    },
    {
      "epoch": 5.080127716064453e-05,
      "model_forward_time": 1.7084810733795166,
      "step": 53269
    },
    {
      "epoch": 5.080127716064453e-05,
      "step": 53269,
      "training_step_time": 6.7258055210113525
    },
    {
      "epoch": 5.080223083496094e-05,
      "grad_norm": 0.12240060418844223,
      "learning_rate": 4.872861875499588e-06,
      "loss": 0.0578,
      "step": 53270
    },
    {
      "epoch": 5.080223083496094e-05,
      "model_forward_time": 1.7131810188293457,
      "step": 53270
    },
    {
      "dynamics_loss_avg": 0.05214682221412659,
      "epoch": 5.080223083496094e-05,
      "step": 53270
    },
    {
      "action_loss_avg": 0.0015848579874727874,
      "epoch": 5.080223083496094e-05,
      "step": 53270
    },
    {
      "epoch": 5.080223083496094e-05,
      "step": 53270,
      "training_step_time": 7.624900817871094
    },
    {
      "epoch": 5.0803184509277345e-05,
      "model_forward_time": 1.7189826965332031,
      "step": 53271
    },
    {
      "epoch": 5.0803184509277345e-05,
      "step": 53271,
      "training_step_time": 7.618145227432251
    },
    {
      "epoch": 5.080413818359375e-05,
      "model_forward_time": 1.7136235237121582,
      "step": 53272
    },
    {
      "epoch": 5.080413818359375e-05,
      "step": 53272,
      "training_step_time": 7.720122337341309
    },
    {
      "epoch": 5.080509185791016e-05,
      "model_forward_time": 1.7098424434661865,
      "step": 53273
    },
    {
      "epoch": 5.080509185791016e-05,
      "step": 53273,
      "training_step_time": 9.417173385620117
    },
    {
      "epoch": 5.080604553222656e-05,
      "model_forward_time": 2.2660090923309326,
      "step": 53274
    },
    {
      "epoch": 5.080604553222656e-05,
      "step": 53274,
      "training_step_time": 8.78218150138855
    },
    {
      "epoch": 5.080699920654297e-05,
      "model_forward_time": 1.7190847396850586,
      "step": 53275
    },
    {
      "epoch": 5.080699920654297e-05,
      "step": 53275,
      "training_step_time": 7.812586069107056
    },
    {
      "epoch": 5.080795288085938e-05,
      "model_forward_time": 1.7147560119628906,
      "step": 53276
    },
    {
      "epoch": 5.080795288085938e-05,
      "step": 53276,
      "training_step_time": 7.741037368774414
    },
    {
      "epoch": 5.080890655517578e-05,
      "model_forward_time": 1.7129192352294922,
      "step": 53277
    },
    {
      "epoch": 5.080890655517578e-05,
      "step": 53277,
      "training_step_time": 7.8818559646606445
    },
    {
      "epoch": 5.0809860229492186e-05,
      "model_forward_time": 1.7199006080627441,
      "step": 53278
    },
    {
      "epoch": 5.0809860229492186e-05,
      "step": 53278,
      "training_step_time": 7.7487359046936035
    },
    {
      "epoch": 5.08108139038086e-05,
      "model_forward_time": 1.714979887008667,
      "step": 53279
    },
    {
      "epoch": 5.08108139038086e-05,
      "step": 53279,
      "training_step_time": 7.608372449874878
    },
    {
      "epoch": 5.0811767578125e-05,
      "grad_norm": 0.12327927350997925,
      "learning_rate": 4.871208947287404e-06,
      "loss": 0.06,
      "step": 53280
    },
    {
      "epoch": 5.0811767578125e-05,
      "model_forward_time": 1.7163708209991455,
      "step": 53280
    },
    {
      "dynamics_loss_avg": 0.05741795990616083,
      "epoch": 5.0811767578125e-05,
      "step": 53280
    },
    {
      "action_loss_avg": 0.0017941332713235169,
      "epoch": 5.0811767578125e-05,
      "step": 53280
    },
    {
      "epoch": 5.0811767578125e-05,
      "step": 53280,
      "training_step_time": 7.130078077316284
    },
    {
      "epoch": 5.0812721252441405e-05,
      "model_forward_time": 1.7127139568328857,
      "step": 53281
    },
    {
      "epoch": 5.0812721252441405e-05,
      "step": 53281,
      "training_step_time": 7.7482216358184814
    },
    {
      "epoch": 5.081367492675781e-05,
      "model_forward_time": 1.7171282768249512,
      "step": 53282
    },
    {
      "epoch": 5.081367492675781e-05,
      "step": 53282,
      "training_step_time": 8.681165933609009
    },
    {
      "epoch": 5.081462860107422e-05,
      "model_forward_time": 1.906092643737793,
      "step": 53283
    },
    {
      "epoch": 5.081462860107422e-05,
      "step": 53283,
      "training_step_time": 9.2682044506073
    },
    {
      "epoch": 5.0815582275390624e-05,
      "model_forward_time": 1.6940844058990479,
      "step": 53284
    },
    {
      "epoch": 5.0815582275390624e-05,
      "step": 53284,
      "training_step_time": 7.768182992935181
    },
    {
      "epoch": 5.081653594970703e-05,
      "model_forward_time": 1.7064943313598633,
      "step": 53285
    },
    {
      "epoch": 5.081653594970703e-05,
      "step": 53285,
      "training_step_time": 9.239214897155762
    },
    {
      "epoch": 5.081748962402344e-05,
      "model_forward_time": 1.711458444595337,
      "step": 53286
    },
    {
      "epoch": 5.081748962402344e-05,
      "step": 53286,
      "training_step_time": 7.71713924407959
    },
    {
      "epoch": 5.081844329833984e-05,
      "model_forward_time": 1.7059218883514404,
      "step": 53287
    },
    {
      "epoch": 5.081844329833984e-05,
      "step": 53287,
      "training_step_time": 7.225134372711182
    },
    {
      "epoch": 5.081939697265625e-05,
      "model_forward_time": 1.7112975120544434,
      "step": 53288
    },
    {
      "epoch": 5.081939697265625e-05,
      "step": 53288,
      "training_step_time": 7.509824752807617
    },
    {
      "epoch": 5.082035064697266e-05,
      "model_forward_time": 1.7155025005340576,
      "step": 53289
    },
    {
      "epoch": 5.082035064697266e-05,
      "step": 53289,
      "training_step_time": 7.082211971282959
    },
    {
      "epoch": 5.082130432128906e-05,
      "grad_norm": 0.11938546597957611,
      "learning_rate": 4.869556033159616e-06,
      "loss": 0.0577,
      "step": 53290
    },
    {
      "epoch": 5.082130432128906e-05,
      "model_forward_time": 1.7006340026855469,
      "step": 53290
    },
    {
      "dynamics_loss_avg": 0.050411699898540975,
      "epoch": 5.082130432128906e-05,
      "step": 53290
    },
    {
      "action_loss_avg": 0.0009226497291820124,
      "epoch": 5.082130432128906e-05,
      "step": 53290
    },
    {
      "epoch": 5.082130432128906e-05,
      "step": 53290,
      "training_step_time": 7.576539039611816
    },
    {
      "epoch": 5.0822257995605466e-05,
      "model_forward_time": 1.7075235843658447,
      "step": 53291
    },
    {
      "epoch": 5.0822257995605466e-05,
      "step": 53291,
      "training_step_time": 7.640789985656738
    },
    {
      "epoch": 5.0823211669921877e-05,
      "model_forward_time": 2.2734901905059814,
      "step": 53292
    },
    {
      "epoch": 5.0823211669921877e-05,
      "step": 53292,
      "training_step_time": 7.634753465652466
    },
    {
      "epoch": 5.082416534423828e-05,
      "model_forward_time": 1.6899871826171875,
      "step": 53293
    },
    {
      "epoch": 5.082416534423828e-05,
      "step": 53293,
      "training_step_time": 8.213555097579956
    },
    {
      "epoch": 5.0825119018554685e-05,
      "model_forward_time": 1.7106616497039795,
      "step": 53294
    },
    {
      "epoch": 5.0825119018554685e-05,
      "step": 53294,
      "training_step_time": 9.041609287261963
    },
    {
      "epoch": 5.0826072692871095e-05,
      "model_forward_time": 1.710327386856079,
      "step": 53295
    },
    {
      "epoch": 5.0826072692871095e-05,
      "step": 53295,
      "training_step_time": 8.4799964427948
    },
    {
      "epoch": 5.08270263671875e-05,
      "model_forward_time": 1.708080768585205,
      "step": 53296
    },
    {
      "epoch": 5.08270263671875e-05,
      "step": 53296,
      "training_step_time": 7.496604919433594
    },
    {
      "epoch": 5.0827980041503903e-05,
      "model_forward_time": 1.7145171165466309,
      "step": 53297
    },
    {
      "epoch": 5.0827980041503903e-05,
      "step": 53297,
      "training_step_time": 7.519340753555298
    },
    {
      "epoch": 5.0828933715820314e-05,
      "model_forward_time": 1.3336207866668701,
      "step": 53298
    },
    {
      "epoch": 5.0828933715820314e-05,
      "step": 53298,
      "training_step_time": 7.037476301193237
    },
    {
      "epoch": 5.082988739013672e-05,
      "model_forward_time": 1.7151093482971191,
      "step": 53299
    },
    {
      "epoch": 5.082988739013672e-05,
      "step": 53299,
      "training_step_time": 9.278302431106567
    },
    {
      "epoch": 5.083084106445312e-05,
      "grad_norm": 0.125936359167099,
      "learning_rate": 4.867903133296983e-06,
      "loss": 0.0605,
      "step": 53300
    },
    {
      "epoch": 5.083084106445312e-05,
      "model_forward_time": 1.7104032039642334,
      "step": 53300
    },
    {
      "dynamics_loss_avg": 0.06048140674829483,
      "epoch": 5.083084106445312e-05,
      "step": 53300
    },
    {
      "action_loss_avg": 0.001024742389563471,
      "epoch": 5.083084106445312e-05,
      "step": 53300
    },
    {
      "epoch": 5.083084106445312e-05,
      "step": 53300,
      "training_step_time": 7.630268812179565
    },
    {
      "epoch": 5.083179473876953e-05,
      "model_forward_time": 1.7371809482574463,
      "step": 53301
    },
    {
      "epoch": 5.083179473876953e-05,
      "step": 53301,
      "training_step_time": 10.069653511047363
    },
    {
      "epoch": 5.083274841308594e-05,
      "model_forward_time": 2.266845464706421,
      "step": 53302
    },
    {
      "epoch": 5.083274841308594e-05,
      "step": 53302,
      "training_step_time": 9.425770044326782
    },
    {
      "epoch": 5.083370208740234e-05,
      "model_forward_time": 1.7164945602416992,
      "step": 53303
    },
    {
      "epoch": 5.083370208740234e-05,
      "step": 53303,
      "training_step_time": 7.767899513244629
    },
    {
      "epoch": 5.083465576171875e-05,
      "model_forward_time": 1.7237989902496338,
      "step": 53304
    },
    {
      "epoch": 5.083465576171875e-05,
      "step": 53304,
      "training_step_time": 7.677687168121338
    },
    {
      "epoch": 5.0835609436035156e-05,
      "model_forward_time": 1.7133095264434814,
      "step": 53305
    },
    {
      "epoch": 5.0835609436035156e-05,
      "step": 53305,
      "training_step_time": 7.707436561584473
    },
    {
      "epoch": 5.083656311035156e-05,
      "model_forward_time": 1.712745189666748,
      "step": 53306
    },
    {
      "epoch": 5.083656311035156e-05,
      "step": 53306,
      "training_step_time": 7.5533225536346436
    },
    {
      "epoch": 5.083751678466797e-05,
      "model_forward_time": 1.7216362953186035,
      "step": 53307
    },
    {
      "epoch": 5.083751678466797e-05,
      "step": 53307,
      "training_step_time": 6.729045867919922
    },
    {
      "epoch": 5.0838470458984375e-05,
      "model_forward_time": 1.7197329998016357,
      "step": 53308
    },
    {
      "epoch": 5.0838470458984375e-05,
      "step": 53308,
      "training_step_time": 7.644976854324341
    },
    {
      "epoch": 5.083942413330078e-05,
      "model_forward_time": 1.7131917476654053,
      "step": 53309
    },
    {
      "epoch": 5.083942413330078e-05,
      "step": 53309,
      "training_step_time": 8.730587244033813
    },
    {
      "epoch": 5.084037780761719e-05,
      "grad_norm": 0.08886532485485077,
      "learning_rate": 4.866250247880264e-06,
      "loss": 0.061,
      "step": 53310
    },
    {
      "epoch": 5.084037780761719e-05,
      "model_forward_time": 1.7090725898742676,
      "step": 53310
    },
    {
      "dynamics_loss_avg": 0.05286718215793371,
      "epoch": 5.084037780761719e-05,
      "step": 53310
    },
    {
      "action_loss_avg": 0.00189080024138093,
      "epoch": 5.084037780761719e-05,
      "step": 53310
    },
    {
      "epoch": 5.084037780761719e-05,
      "step": 53310,
      "training_step_time": 9.06437063217163
    },
    {
      "epoch": 5.0841331481933594e-05,
      "model_forward_time": 2.2595021724700928,
      "step": 53311
    },
    {
      "epoch": 5.0841331481933594e-05,
      "step": 53311,
      "training_step_time": 10.23216462135315
    },
    {
      "epoch": 5.084228515625e-05,
      "model_forward_time": 1.7034692764282227,
      "step": 53312
    },
    {
      "epoch": 5.084228515625e-05,
      "step": 53312,
      "training_step_time": 8.31201696395874
    },
    {
      "epoch": 5.084323883056641e-05,
      "model_forward_time": 1.6915383338928223,
      "step": 53313
    },
    {
      "epoch": 5.084323883056641e-05,
      "step": 53313,
      "training_step_time": 7.370034456253052
    },
    {
      "epoch": 5.084419250488281e-05,
      "model_forward_time": 1.3424673080444336,
      "step": 53314
    },
    {
      "epoch": 5.084419250488281e-05,
      "step": 53314,
      "training_step_time": 7.6723856925964355
    },
    {
      "epoch": 5.0845146179199217e-05,
      "model_forward_time": 1.7092394828796387,
      "step": 53315
    },
    {
      "epoch": 5.0845146179199217e-05,
      "step": 53315,
      "training_step_time": 8.208009243011475
    },
    {
      "epoch": 5.084609985351563e-05,
      "model_forward_time": 1.7118408679962158,
      "step": 53316
    },
    {
      "epoch": 5.084609985351563e-05,
      "step": 53316,
      "training_step_time": 6.763738632202148
    },
    {
      "epoch": 5.084705352783203e-05,
      "model_forward_time": 1.7116830348968506,
      "step": 53317
    },
    {
      "epoch": 5.084705352783203e-05,
      "step": 53317,
      "training_step_time": 7.532581090927124
    },
    {
      "epoch": 5.0848007202148435e-05,
      "model_forward_time": 1.712937355041504,
      "step": 53318
    },
    {
      "epoch": 5.0848007202148435e-05,
      "step": 53318,
      "training_step_time": 7.6973748207092285
    },
    {
      "epoch": 5.0848960876464846e-05,
      "model_forward_time": 1.6998670101165771,
      "step": 53319
    },
    {
      "epoch": 5.0848960876464846e-05,
      "step": 53319,
      "training_step_time": 8.818847179412842
    },
    {
      "epoch": 5.084991455078125e-05,
      "grad_norm": 0.1115991398692131,
      "learning_rate": 4.864597377090218e-06,
      "loss": 0.0568,
      "step": 53320
    },
    {
      "epoch": 5.084991455078125e-05,
      "model_forward_time": 2.2856357097625732,
      "step": 53320
    },
    {
      "dynamics_loss_avg": 0.0631721718236804,
      "epoch": 5.084991455078125e-05,
      "step": 53320
    },
    {
      "action_loss_avg": 0.0020350266131572425,
      "epoch": 5.084991455078125e-05,
      "step": 53320
    },
    {
      "epoch": 5.084991455078125e-05,
      "step": 53320,
      "training_step_time": 9.602418661117554
    },
    {
      "epoch": 5.0850868225097654e-05,
      "model_forward_time": 1.7049777507781982,
      "step": 53321
    },
    {
      "epoch": 5.0850868225097654e-05,
      "step": 53321,
      "training_step_time": 8.778089761734009
    },
    {
      "epoch": 5.0851821899414065e-05,
      "model_forward_time": 1.7093632221221924,
      "step": 53322
    },
    {
      "epoch": 5.0851821899414065e-05,
      "step": 53322,
      "training_step_time": 7.699316024780273
    },
    {
      "epoch": 5.085277557373047e-05,
      "model_forward_time": 1.712092399597168,
      "step": 53323
    },
    {
      "epoch": 5.085277557373047e-05,
      "step": 53323,
      "training_step_time": 7.492945194244385
    },
    {
      "epoch": 5.085372924804687e-05,
      "model_forward_time": 1.7053799629211426,
      "step": 53324
    },
    {
      "epoch": 5.085372924804687e-05,
      "step": 53324,
      "training_step_time": 8.228439092636108
    },
    {
      "epoch": 5.0854682922363284e-05,
      "model_forward_time": 1.6994531154632568,
      "step": 53325
    },
    {
      "epoch": 5.0854682922363284e-05,
      "step": 53325,
      "training_step_time": 6.742016553878784
    },
    {
      "epoch": 5.085563659667969e-05,
      "model_forward_time": 1.7087628841400146,
      "step": 53326
    },
    {
      "epoch": 5.085563659667969e-05,
      "step": 53326,
      "training_step_time": 7.500420808792114
    },
    {
      "epoch": 5.085659027099609e-05,
      "model_forward_time": 1.710385799407959,
      "step": 53327
    },
    {
      "epoch": 5.085659027099609e-05,
      "step": 53327,
      "training_step_time": 8.657056331634521
    },
    {
      "epoch": 5.08575439453125e-05,
      "model_forward_time": 1.7031407356262207,
      "step": 53328
    },
    {
      "epoch": 5.08575439453125e-05,
      "step": 53328,
      "training_step_time": 8.869359254837036
    },
    {
      "epoch": 5.085849761962891e-05,
      "model_forward_time": 1.7500934600830078,
      "step": 53329
    },
    {
      "epoch": 5.085849761962891e-05,
      "step": 53329,
      "training_step_time": 8.837814569473267
    },
    {
      "epoch": 5.085945129394531e-05,
      "grad_norm": 0.11676253378391266,
      "learning_rate": 4.862944521107599e-06,
      "loss": 0.0578,
      "step": 53330
    },
    {
      "epoch": 5.085945129394531e-05,
      "model_forward_time": 1.6996667385101318,
      "step": 53330
    },
    {
      "dynamics_loss_avg": 0.05401543844491243,
      "epoch": 5.085945129394531e-05,
      "step": 53330
    },
    {
      "action_loss_avg": 0.001701467263046652,
      "epoch": 5.085945129394531e-05,
      "step": 53330
    },
    {
      "epoch": 5.085945129394531e-05,
      "step": 53330,
      "training_step_time": 8.201224565505981
    },
    {
      "epoch": 5.086040496826172e-05,
      "model_forward_time": 1.707348346710205,
      "step": 53331
    },
    {
      "epoch": 5.086040496826172e-05,
      "step": 53331,
      "training_step_time": 8.858386039733887
    },
    {
      "epoch": 5.0861358642578126e-05,
      "model_forward_time": 1.3300669193267822,
      "step": 53332
    },
    {
      "epoch": 5.0861358642578126e-05,
      "step": 53332,
      "training_step_time": 7.748809099197388
    },
    {
      "epoch": 5.086231231689453e-05,
      "model_forward_time": 1.7022294998168945,
      "step": 53333
    },
    {
      "epoch": 5.086231231689453e-05,
      "step": 53333,
      "training_step_time": 7.798584938049316
    },
    {
      "epoch": 5.086326599121094e-05,
      "model_forward_time": 1.7064218521118164,
      "step": 53334
    },
    {
      "epoch": 5.086326599121094e-05,
      "step": 53334,
      "training_step_time": 7.157004356384277
    },
    {
      "epoch": 5.0864219665527345e-05,
      "model_forward_time": 1.7067766189575195,
      "step": 53335
    },
    {
      "epoch": 5.0864219665527345e-05,
      "step": 53335,
      "training_step_time": 7.553212404251099
    },
    {
      "epoch": 5.086517333984375e-05,
      "model_forward_time": 1.7085795402526855,
      "step": 53336
    },
    {
      "epoch": 5.086517333984375e-05,
      "step": 53336,
      "training_step_time": 7.747178316116333
    },
    {
      "epoch": 5.086612701416016e-05,
      "model_forward_time": 1.7048366069793701,
      "step": 53337
    },
    {
      "epoch": 5.086612701416016e-05,
      "step": 53337,
      "training_step_time": 7.80376935005188
    },
    {
      "epoch": 5.0867080688476563e-05,
      "model_forward_time": 1.7621774673461914,
      "step": 53338
    },
    {
      "epoch": 5.0867080688476563e-05,
      "step": 53338,
      "training_step_time": 7.7459495067596436
    },
    {
      "epoch": 5.086803436279297e-05,
      "model_forward_time": 2.254472255706787,
      "step": 53339
    },
    {
      "epoch": 5.086803436279297e-05,
      "step": 53339,
      "training_step_time": 8.71611738204956
    },
    {
      "epoch": 5.086898803710938e-05,
      "grad_norm": 0.09909458458423615,
      "learning_rate": 4.861291680113162e-06,
      "loss": 0.0569,
      "step": 53340
    },
    {
      "epoch": 5.086898803710938e-05,
      "model_forward_time": 1.7034308910369873,
      "step": 53340
    },
    {
      "dynamics_loss_avg": 0.06350189615041017,
      "epoch": 5.086898803710938e-05,
      "step": 53340
    },
    {
      "action_loss_avg": 0.0018604736193083226,
      "epoch": 5.086898803710938e-05,
      "step": 53340
    },
    {
      "epoch": 5.086898803710938e-05,
      "step": 53340,
      "training_step_time": 7.783057928085327
    },
    {
      "epoch": 5.086994171142578e-05,
      "model_forward_time": 1.6866936683654785,
      "step": 53341
    },
    {
      "epoch": 5.086994171142578e-05,
      "step": 53341,
      "training_step_time": 7.699311017990112
    },
    {
      "epoch": 5.0870895385742186e-05,
      "model_forward_time": 1.7055952548980713,
      "step": 53342
    },
    {
      "epoch": 5.0870895385742186e-05,
      "step": 53342,
      "training_step_time": 8.724770545959473
    },
    {
      "epoch": 5.08718490600586e-05,
      "model_forward_time": 1.7106475830078125,
      "step": 53343
    },
    {
      "epoch": 5.08718490600586e-05,
      "step": 53343,
      "training_step_time": 7.134016275405884
    },
    {
      "epoch": 5.0872802734375e-05,
      "model_forward_time": 1.706528902053833,
      "step": 53344
    },
    {
      "epoch": 5.0872802734375e-05,
      "step": 53344,
      "training_step_time": 7.569953441619873
    },
    {
      "epoch": 5.0873756408691405e-05,
      "model_forward_time": 1.6912274360656738,
      "step": 53345
    },
    {
      "epoch": 5.0873756408691405e-05,
      "step": 53345,
      "training_step_time": 7.693222284317017
    },
    {
      "epoch": 5.087471008300781e-05,
      "model_forward_time": 1.70790433883667,
      "step": 53346
    },
    {
      "epoch": 5.087471008300781e-05,
      "step": 53346,
      "training_step_time": 8.628440856933594
    },
    {
      "epoch": 5.087566375732422e-05,
      "model_forward_time": 1.7371366024017334,
      "step": 53347
    },
    {
      "epoch": 5.087566375732422e-05,
      "step": 53347,
      "training_step_time": 8.424770832061768
    },
    {
      "epoch": 5.0876617431640624e-05,
      "model_forward_time": 2.268934965133667,
      "step": 53348
    },
    {
      "epoch": 5.0876617431640624e-05,
      "step": 53348,
      "training_step_time": 8.391838550567627
    },
    {
      "epoch": 5.087757110595703e-05,
      "model_forward_time": 1.7014381885528564,
      "step": 53349
    },
    {
      "epoch": 5.087757110595703e-05,
      "step": 53349,
      "training_step_time": 8.310490846633911
    },
    {
      "epoch": 5.087852478027344e-05,
      "grad_norm": 0.09161823242902756,
      "learning_rate": 4.859638854287656e-06,
      "loss": 0.0582,
      "step": 53350
    },
    {
      "epoch": 5.087852478027344e-05,
      "model_forward_time": 1.6992995738983154,
      "step": 53350
    },
    {
      "dynamics_loss_avg": 0.057351083308458325,
      "epoch": 5.087852478027344e-05,
      "step": 53350
    },
    {
      "action_loss_avg": 0.001323982939356938,
      "epoch": 5.087852478027344e-05,
      "step": 53350
    },
    {
      "epoch": 5.087852478027344e-05,
      "step": 53350,
      "training_step_time": 7.822826147079468
    },
    {
      "epoch": 5.087947845458984e-05,
      "model_forward_time": 1.7229182720184326,
      "step": 53351
    },
    {
      "epoch": 5.087947845458984e-05,
      "step": 53351,
      "training_step_time": 7.759690999984741
    },
    {
      "epoch": 5.088043212890625e-05,
      "model_forward_time": 1.702770471572876,
      "step": 53352
    },
    {
      "epoch": 5.088043212890625e-05,
      "step": 53352,
      "training_step_time": 7.709322214126587
    },
    {
      "epoch": 5.088138580322266e-05,
      "model_forward_time": 1.705543041229248,
      "step": 53353
    },
    {
      "epoch": 5.088138580322266e-05,
      "step": 53353,
      "training_step_time": 7.529150485992432
    },
    {
      "epoch": 5.088233947753906e-05,
      "model_forward_time": 1.715012788772583,
      "step": 53354
    },
    {
      "epoch": 5.088233947753906e-05,
      "step": 53354,
      "training_step_time": 7.658760070800781
    },
    {
      "epoch": 5.0883293151855466e-05,
      "model_forward_time": 1.7053658962249756,
      "step": 53355
    },
    {
      "epoch": 5.0883293151855466e-05,
      "step": 53355,
      "training_step_time": 8.470072269439697
    },
    {
      "epoch": 5.0884246826171877e-05,
      "model_forward_time": 1.7038955688476562,
      "step": 53356
    },
    {
      "epoch": 5.0884246826171877e-05,
      "step": 53356,
      "training_step_time": 7.721905469894409
    },
    {
      "epoch": 5.088520050048828e-05,
      "model_forward_time": 2.279395818710327,
      "step": 53357
    },
    {
      "epoch": 5.088520050048828e-05,
      "step": 53357,
      "training_step_time": 8.880063533782959
    },
    {
      "epoch": 5.0886154174804685e-05,
      "model_forward_time": 1.7078309059143066,
      "step": 53358
    },
    {
      "epoch": 5.0886154174804685e-05,
      "step": 53358,
      "training_step_time": 8.262547731399536
    },
    {
      "epoch": 5.0887107849121095e-05,
      "model_forward_time": 1.7111382484436035,
      "step": 53359
    },
    {
      "epoch": 5.0887107849121095e-05,
      "step": 53359,
      "training_step_time": 7.727705478668213
    },
    {
      "epoch": 5.08880615234375e-05,
      "grad_norm": 0.10187245905399323,
      "learning_rate": 4.857986043811836e-06,
      "loss": 0.0604,
      "step": 53360
    },
    {
      "epoch": 5.08880615234375e-05,
      "model_forward_time": 1.70135498046875,
      "step": 53360
    },
    {
      "dynamics_loss_avg": 0.06015982329845428,
      "epoch": 5.08880615234375e-05,
      "step": 53360
    },
    {
      "action_loss_avg": 0.0015415673435200006,
      "epoch": 5.08880615234375e-05,
      "step": 53360
    },
    {
      "epoch": 5.08880615234375e-05,
      "step": 53360,
      "training_step_time": 7.724460124969482
    },
    {
      "epoch": 5.0889015197753903e-05,
      "model_forward_time": 1.7128398418426514,
      "step": 53361
    },
    {
      "epoch": 5.0889015197753903e-05,
      "step": 53361,
      "training_step_time": 7.698117733001709
    },
    {
      "epoch": 5.0889968872070314e-05,
      "model_forward_time": 1.7195079326629639,
      "step": 53362
    },
    {
      "epoch": 5.0889968872070314e-05,
      "step": 53362,
      "training_step_time": 7.493331432342529
    },
    {
      "epoch": 5.089092254638672e-05,
      "model_forward_time": 1.708672285079956,
      "step": 53363
    },
    {
      "epoch": 5.089092254638672e-05,
      "step": 53363,
      "training_step_time": 8.568432092666626
    },
    {
      "epoch": 5.089187622070312e-05,
      "model_forward_time": 1.7068390846252441,
      "step": 53364
    },
    {
      "epoch": 5.089187622070312e-05,
      "step": 53364,
      "training_step_time": 8.488424301147461
    },
    {
      "epoch": 5.089282989501953e-05,
      "model_forward_time": 1.7036888599395752,
      "step": 53365
    },
    {
      "epoch": 5.089282989501953e-05,
      "step": 53365,
      "training_step_time": 7.742149114608765
    },
    {
      "epoch": 5.089378356933594e-05,
      "model_forward_time": 2.2785465717315674,
      "step": 53366
    },
    {
      "epoch": 5.089378356933594e-05,
      "step": 53366,
      "training_step_time": 8.35330080986023
    },
    {
      "epoch": 5.089473724365234e-05,
      "model_forward_time": 1.7177424430847168,
      "step": 53367
    },
    {
      "epoch": 5.089473724365234e-05,
      "step": 53367,
      "training_step_time": 8.488629341125488
    },
    {
      "epoch": 5.089569091796875e-05,
      "model_forward_time": 1.712949275970459,
      "step": 53368
    },
    {
      "epoch": 5.089569091796875e-05,
      "step": 53368,
      "training_step_time": 7.742748498916626
    },
    {
      "epoch": 5.0896644592285156e-05,
      "model_forward_time": 1.710906982421875,
      "step": 53369
    },
    {
      "epoch": 5.0896644592285156e-05,
      "step": 53369,
      "training_step_time": 7.744518518447876
    },
    {
      "epoch": 5.089759826660156e-05,
      "grad_norm": 0.11529958993196487,
      "learning_rate": 4.856333248866448e-06,
      "loss": 0.0568,
      "step": 53370
    },
    {
      "epoch": 5.089759826660156e-05,
      "model_forward_time": 1.7090697288513184,
      "step": 53370
    },
    {
      "dynamics_loss_avg": 0.05072435922920704,
      "epoch": 5.089759826660156e-05,
      "step": 53370
    },
    {
      "action_loss_avg": 0.001129072526236996,
      "epoch": 5.089759826660156e-05,
      "step": 53370
    },
    {
      "epoch": 5.089759826660156e-05,
      "step": 53370,
      "training_step_time": 7.655881881713867
    },
    {
      "epoch": 5.089855194091797e-05,
      "model_forward_time": 1.7125627994537354,
      "step": 53371
    },
    {
      "epoch": 5.089855194091797e-05,
      "step": 53371,
      "training_step_time": 7.1685473918914795
    },
    {
      "epoch": 5.0899505615234375e-05,
      "model_forward_time": 1.7114574909210205,
      "step": 53372
    },
    {
      "epoch": 5.0899505615234375e-05,
      "step": 53372,
      "training_step_time": 7.569069147109985
    },
    {
      "epoch": 5.090045928955078e-05,
      "model_forward_time": 1.7199575901031494,
      "step": 53373
    },
    {
      "epoch": 5.090045928955078e-05,
      "step": 53373,
      "training_step_time": 7.669971466064453
    },
    {
      "epoch": 5.090141296386719e-05,
      "model_forward_time": 1.7117846012115479,
      "step": 53374
    },
    {
      "epoch": 5.090141296386719e-05,
      "step": 53374,
      "training_step_time": 8.729723453521729
    },
    {
      "epoch": 5.0902366638183594e-05,
      "model_forward_time": 1.7474181652069092,
      "step": 53375
    },
    {
      "epoch": 5.0902366638183594e-05,
      "step": 53375,
      "training_step_time": 7.747947454452515
    },
    {
      "epoch": 5.09033203125e-05,
      "model_forward_time": 2.2573935985565186,
      "step": 53376
    },
    {
      "epoch": 5.09033203125e-05,
      "step": 53376,
      "training_step_time": 9.41663932800293
    },
    {
      "epoch": 5.090427398681641e-05,
      "model_forward_time": 1.7099413871765137,
      "step": 53377
    },
    {
      "epoch": 5.090427398681641e-05,
      "step": 53377,
      "training_step_time": 7.789673566818237
    },
    {
      "epoch": 5.090522766113281e-05,
      "model_forward_time": 1.712831974029541,
      "step": 53378
    },
    {
      "epoch": 5.090522766113281e-05,
      "step": 53378,
      "training_step_time": 7.795490026473999
    },
    {
      "epoch": 5.090618133544922e-05,
      "model_forward_time": 1.7109737396240234,
      "step": 53379
    },
    {
      "epoch": 5.090618133544922e-05,
      "step": 53379,
      "training_step_time": 7.656677961349487
    },
    {
      "epoch": 5.090713500976563e-05,
      "grad_norm": 0.10475005209445953,
      "learning_rate": 4.85468046963224e-06,
      "loss": 0.0566,
      "step": 53380
    },
    {
      "epoch": 5.090713500976563e-05,
      "model_forward_time": 1.7113010883331299,
      "step": 53380
    },
    {
      "dynamics_loss_avg": 0.05090235387906432,
      "epoch": 5.090713500976563e-05,
      "step": 53380
    },
    {
      "action_loss_avg": 0.001538386329775676,
      "epoch": 5.090713500976563e-05,
      "step": 53380
    },
    {
      "epoch": 5.090713500976563e-05,
      "step": 53380,
      "training_step_time": 7.445984601974487
    },
    {
      "epoch": 5.090808868408203e-05,
      "model_forward_time": 1.7165729999542236,
      "step": 53381
    },
    {
      "epoch": 5.090808868408203e-05,
      "step": 53381,
      "training_step_time": 7.624727249145508
    },
    {
      "epoch": 5.0909042358398435e-05,
      "model_forward_time": 1.7073025703430176,
      "step": 53382
    },
    {
      "epoch": 5.0909042358398435e-05,
      "step": 53382,
      "training_step_time": 9.147757768630981
    },
    {
      "epoch": 5.0909996032714846e-05,
      "model_forward_time": 1.717223882675171,
      "step": 53383
    },
    {
      "epoch": 5.0909996032714846e-05,
      "step": 53383,
      "training_step_time": 8.894140720367432
    },
    {
      "epoch": 5.091094970703125e-05,
      "model_forward_time": 1.7417051792144775,
      "step": 53384
    },
    {
      "epoch": 5.091094970703125e-05,
      "step": 53384,
      "training_step_time": 9.382606506347656
    },
    {
      "epoch": 5.0911903381347654e-05,
      "model_forward_time": 2.247002124786377,
      "step": 53385
    },
    {
      "epoch": 5.0911903381347654e-05,
      "step": 53385,
      "training_step_time": 7.76006817817688
    },
    {
      "epoch": 5.0912857055664065e-05,
      "model_forward_time": 1.707547664642334,
      "step": 53386
    },
    {
      "epoch": 5.0912857055664065e-05,
      "step": 53386,
      "training_step_time": 7.750414848327637
    },
    {
      "epoch": 5.091381072998047e-05,
      "model_forward_time": 1.7153477668762207,
      "step": 53387
    },
    {
      "epoch": 5.091381072998047e-05,
      "step": 53387,
      "training_step_time": 7.640178203582764
    },
    {
      "epoch": 5.091476440429687e-05,
      "model_forward_time": 1.7059969902038574,
      "step": 53388
    },
    {
      "epoch": 5.091476440429687e-05,
      "step": 53388,
      "training_step_time": 7.809911012649536
    },
    {
      "epoch": 5.0915718078613284e-05,
      "model_forward_time": 1.7065458297729492,
      "step": 53389
    },
    {
      "epoch": 5.0915718078613284e-05,
      "step": 53389,
      "training_step_time": 7.639873504638672
    },
    {
      "epoch": 5.091667175292969e-05,
      "grad_norm": 0.1562512218952179,
      "learning_rate": 4.8530277062899575e-06,
      "loss": 0.054,
      "step": 53390
    },
    {
      "epoch": 5.091667175292969e-05,
      "model_forward_time": 1.3390614986419678,
      "step": 53390
    },
    {
      "dynamics_loss_avg": 0.05118371173739433,
      "epoch": 5.091667175292969e-05,
      "step": 53390
    },
    {
      "action_loss_avg": 0.0017060368525562807,
      "epoch": 5.091667175292969e-05,
      "step": 53390
    },
    {
      "epoch": 5.091667175292969e-05,
      "step": 53390,
      "training_step_time": 7.66498589515686
    },
    {
      "epoch": 5.091762542724609e-05,
      "model_forward_time": 1.7122070789337158,
      "step": 53391
    },
    {
      "epoch": 5.091762542724609e-05,
      "step": 53391,
      "training_step_time": 8.53207802772522
    },
    {
      "epoch": 5.09185791015625e-05,
      "model_forward_time": 1.7130310535430908,
      "step": 53392
    },
    {
      "epoch": 5.09185791015625e-05,
      "step": 53392,
      "training_step_time": 8.76880145072937
    },
    {
      "epoch": 5.091953277587891e-05,
      "model_forward_time": 1.739861249923706,
      "step": 53393
    },
    {
      "epoch": 5.091953277587891e-05,
      "step": 53393,
      "training_step_time": 8.25140905380249
    },
    {
      "epoch": 5.092048645019531e-05,
      "model_forward_time": 2.264940023422241,
      "step": 53394
    },
    {
      "epoch": 5.092048645019531e-05,
      "step": 53394,
      "training_step_time": 7.8579137325286865
    },
    {
      "epoch": 5.092144012451172e-05,
      "model_forward_time": 1.7192912101745605,
      "step": 53395
    },
    {
      "epoch": 5.092144012451172e-05,
      "step": 53395,
      "training_step_time": 7.704343795776367
    },
    {
      "epoch": 5.0922393798828126e-05,
      "model_forward_time": 1.7190747261047363,
      "step": 53396
    },
    {
      "epoch": 5.0922393798828126e-05,
      "step": 53396,
      "training_step_time": 8.312920331954956
    },
    {
      "epoch": 5.092334747314453e-05,
      "model_forward_time": 1.7157299518585205,
      "step": 53397
    },
    {
      "epoch": 5.092334747314453e-05,
      "step": 53397,
      "training_step_time": 8.521258115768433
    },
    {
      "epoch": 5.092430114746094e-05,
      "model_forward_time": 1.724757432937622,
      "step": 53398
    },
    {
      "epoch": 5.092430114746094e-05,
      "step": 53398,
      "training_step_time": 7.533207178115845
    },
    {
      "epoch": 5.0925254821777345e-05,
      "model_forward_time": 1.7180049419403076,
      "step": 53399
    },
    {
      "epoch": 5.0925254821777345e-05,
      "step": 53399,
      "training_step_time": 8.447859764099121
    },
    {
      "epoch": 5.092620849609375e-05,
      "grad_norm": 0.15939533710479736,
      "learning_rate": 4.851374959020344e-06,
      "loss": 0.0595,
      "step": 53400
    },
    {
      "epoch": 5.092620849609375e-05,
      "model_forward_time": 1.7160050868988037,
      "step": 53400
    },
    {
      "dynamics_loss_avg": 0.05240828134119511,
      "epoch": 5.092620849609375e-05,
      "step": 53400
    },
    {
      "action_loss_avg": 0.001766318891895935,
      "epoch": 5.092620849609375e-05,
      "step": 53400
    },
    {
      "epoch": 5.092620849609375e-05,
      "step": 53400,
      "training_step_time": 7.757830381393433
    },
    {
      "epoch": 5.092716217041016e-05,
      "model_forward_time": 1.7196872234344482,
      "step": 53401
    },
    {
      "epoch": 5.092716217041016e-05,
      "step": 53401,
      "training_step_time": 7.793259382247925
    },
    {
      "epoch": 5.0928115844726563e-05,
      "model_forward_time": 1.3459930419921875,
      "step": 53402
    },
    {
      "epoch": 5.0928115844726563e-05,
      "step": 53402,
      "training_step_time": 8.31828236579895
    },
    {
      "epoch": 5.092906951904297e-05,
      "model_forward_time": 2.3016457557678223,
      "step": 53403
    },
    {
      "epoch": 5.092906951904297e-05,
      "step": 53403,
      "training_step_time": 8.718489646911621
    },
    {
      "epoch": 5.093002319335938e-05,
      "model_forward_time": 1.7203710079193115,
      "step": 53404
    },
    {
      "epoch": 5.093002319335938e-05,
      "step": 53404,
      "training_step_time": 8.493730783462524
    },
    {
      "epoch": 5.093097686767578e-05,
      "model_forward_time": 1.7165122032165527,
      "step": 53405
    },
    {
      "epoch": 5.093097686767578e-05,
      "step": 53405,
      "training_step_time": 7.841103792190552
    },
    {
      "epoch": 5.0931930541992186e-05,
      "model_forward_time": 1.715923547744751,
      "step": 53406
    },
    {
      "epoch": 5.0931930541992186e-05,
      "step": 53406,
      "training_step_time": 7.9140119552612305
    },
    {
      "epoch": 5.093288421630859e-05,
      "model_forward_time": 1.7178449630737305,
      "step": 53407
    },
    {
      "epoch": 5.093288421630859e-05,
      "step": 53407,
      "training_step_time": 7.750699996948242
    },
    {
      "epoch": 5.0933837890625e-05,
      "model_forward_time": 1.720041036605835,
      "step": 53408
    },
    {
      "epoch": 5.0933837890625e-05,
      "step": 53408,
      "training_step_time": 7.668834924697876
    },
    {
      "epoch": 5.0934791564941405e-05,
      "model_forward_time": 1.7213175296783447,
      "step": 53409
    },
    {
      "epoch": 5.0934791564941405e-05,
      "step": 53409,
      "training_step_time": 8.034582614898682
    },
    {
      "epoch": 5.093574523925781e-05,
      "grad_norm": 0.1452232152223587,
      "learning_rate": 4.849722228004142e-06,
      "loss": 0.056,
      "step": 53410
    },
    {
      "epoch": 5.093574523925781e-05,
      "model_forward_time": 1.7198612689971924,
      "step": 53410
    },
    {
      "dynamics_loss_avg": 0.04790365397930145,
      "epoch": 5.093574523925781e-05,
      "step": 53410
    },
    {
      "action_loss_avg": 0.0010281721770297736,
      "epoch": 5.093574523925781e-05,
      "step": 53410
    },
    {
      "epoch": 5.093574523925781e-05,
      "step": 53410,
      "training_step_time": 7.727829933166504
    },
    {
      "epoch": 5.093669891357422e-05,
      "model_forward_time": 1.7102227210998535,
      "step": 53411
    },
    {
      "epoch": 5.093669891357422e-05,
      "step": 53411,
      "training_step_time": 7.699186325073242
    },
    {
      "epoch": 5.0937652587890624e-05,
      "model_forward_time": 1.7499921321868896,
      "step": 53412
    },
    {
      "epoch": 5.0937652587890624e-05,
      "step": 53412,
      "training_step_time": 8.305017948150635
    },
    {
      "epoch": 5.093860626220703e-05,
      "model_forward_time": 1.7195847034454346,
      "step": 53413
    },
    {
      "epoch": 5.093860626220703e-05,
      "step": 53413,
      "training_step_time": 8.513477563858032
    },
    {
      "epoch": 5.093955993652344e-05,
      "model_forward_time": 1.7210919857025146,
      "step": 53414
    },
    {
      "epoch": 5.093955993652344e-05,
      "step": 53414,
      "training_step_time": 8.355461597442627
    },
    {
      "epoch": 5.094051361083984e-05,
      "model_forward_time": 1.7163431644439697,
      "step": 53415
    },
    {
      "epoch": 5.094051361083984e-05,
      "step": 53415,
      "training_step_time": 7.756626844406128
    },
    {
      "epoch": 5.094146728515625e-05,
      "model_forward_time": 1.7161660194396973,
      "step": 53416
    },
    {
      "epoch": 5.094146728515625e-05,
      "step": 53416,
      "training_step_time": 7.712944746017456
    },
    {
      "epoch": 5.094242095947266e-05,
      "model_forward_time": 1.7148566246032715,
      "step": 53417
    },
    {
      "epoch": 5.094242095947266e-05,
      "step": 53417,
      "training_step_time": 7.8281354904174805
    },
    {
      "epoch": 5.094337463378906e-05,
      "model_forward_time": 1.7156240940093994,
      "step": 53418
    },
    {
      "epoch": 5.094337463378906e-05,
      "step": 53418,
      "training_step_time": 8.670216798782349
    },
    {
      "epoch": 5.0944328308105466e-05,
      "model_forward_time": 1.7188327312469482,
      "step": 53419
    },
    {
      "epoch": 5.0944328308105466e-05,
      "step": 53419,
      "training_step_time": 8.621315002441406
    },
    {
      "epoch": 5.094528198242188e-05,
      "grad_norm": 0.10432665795087814,
      "learning_rate": 4.848069513422091e-06,
      "loss": 0.0576,
      "step": 53420
    },
    {
      "epoch": 5.094528198242188e-05,
      "model_forward_time": 1.7130718231201172,
      "step": 53420
    },
    {
      "dynamics_loss_avg": 0.04128199107944965,
      "epoch": 5.094528198242188e-05,
      "step": 53420
    },
    {
      "action_loss_avg": 0.0011771669844165445,
      "epoch": 5.094528198242188e-05,
      "step": 53420
    },
    {
      "epoch": 5.094528198242188e-05,
      "step": 53420,
      "training_step_time": 7.775591135025024
    },
    {
      "epoch": 5.094623565673828e-05,
      "model_forward_time": 1.761702537536621,
      "step": 53421
    },
    {
      "epoch": 5.094623565673828e-05,
      "step": 53421,
      "training_step_time": 8.595167398452759
    },
    {
      "epoch": 5.0947189331054685e-05,
      "model_forward_time": 2.2574968338012695,
      "step": 53422
    },
    {
      "epoch": 5.0947189331054685e-05,
      "step": 53422,
      "training_step_time": 8.736260890960693
    },
    {
      "epoch": 5.0948143005371095e-05,
      "model_forward_time": 1.7150108814239502,
      "step": 53423
    },
    {
      "epoch": 5.0948143005371095e-05,
      "step": 53423,
      "training_step_time": 8.387486457824707
    },
    {
      "epoch": 5.09490966796875e-05,
      "model_forward_time": 1.7189795970916748,
      "step": 53424
    },
    {
      "epoch": 5.09490966796875e-05,
      "step": 53424,
      "training_step_time": 7.780088901519775
    },
    {
      "epoch": 5.0950050354003904e-05,
      "model_forward_time": 1.7037277221679688,
      "step": 53425
    },
    {
      "epoch": 5.0950050354003904e-05,
      "step": 53425,
      "training_step_time": 7.72350001335144
    },
    {
      "epoch": 5.0951004028320314e-05,
      "model_forward_time": 1.7140967845916748,
      "step": 53426
    },
    {
      "epoch": 5.0951004028320314e-05,
      "step": 53426,
      "training_step_time": 7.739717483520508
    },
    {
      "epoch": 5.095195770263672e-05,
      "model_forward_time": 1.709963321685791,
      "step": 53427
    },
    {
      "epoch": 5.095195770263672e-05,
      "step": 53427,
      "training_step_time": 7.662144899368286
    },
    {
      "epoch": 5.095291137695312e-05,
      "model_forward_time": 1.713017225265503,
      "step": 53428
    },
    {
      "epoch": 5.095291137695312e-05,
      "step": 53428,
      "training_step_time": 8.4043710231781
    },
    {
      "epoch": 5.095386505126953e-05,
      "model_forward_time": 1.7171728610992432,
      "step": 53429
    },
    {
      "epoch": 5.095386505126953e-05,
      "step": 53429,
      "training_step_time": 7.782547235488892
    },
    {
      "epoch": 5.095481872558594e-05,
      "grad_norm": 0.10049348324537277,
      "learning_rate": 4.84641681545493e-06,
      "loss": 0.0587,
      "step": 53430
    },
    {
      "epoch": 5.095481872558594e-05,
      "model_forward_time": 1.7613770961761475,
      "step": 53430
    },
    {
      "dynamics_loss_avg": 0.05597638711333275,
      "epoch": 5.095481872558594e-05,
      "step": 53430
    },
    {
      "action_loss_avg": 0.0013298711361130699,
      "epoch": 5.095481872558594e-05,
      "step": 53430
    },
    {
      "epoch": 5.095481872558594e-05,
      "step": 53430,
      "training_step_time": 9.30650019645691
    },
    {
      "epoch": 5.095577239990234e-05,
      "model_forward_time": 2.2594804763793945,
      "step": 53431
    },
    {
      "epoch": 5.095577239990234e-05,
      "step": 53431,
      "training_step_time": 7.885516881942749
    },
    {
      "epoch": 5.095672607421875e-05,
      "model_forward_time": 1.7168078422546387,
      "step": 53432
    },
    {
      "epoch": 5.095672607421875e-05,
      "step": 53432,
      "training_step_time": 8.283527374267578
    },
    {
      "epoch": 5.0957679748535156e-05,
      "model_forward_time": 1.7037761211395264,
      "step": 53433
    },
    {
      "epoch": 5.0957679748535156e-05,
      "step": 53433,
      "training_step_time": 7.958219528198242
    },
    {
      "epoch": 5.095863342285156e-05,
      "model_forward_time": 1.7235305309295654,
      "step": 53434
    },
    {
      "epoch": 5.095863342285156e-05,
      "step": 53434,
      "training_step_time": 7.7397496700286865
    },
    {
      "epoch": 5.095958709716797e-05,
      "model_forward_time": 1.7283856868743896,
      "step": 53435
    },
    {
      "epoch": 5.095958709716797e-05,
      "step": 53435,
      "training_step_time": 7.848579168319702
    },
    {
      "epoch": 5.0960540771484375e-05,
      "model_forward_time": 1.7189867496490479,
      "step": 53436
    },
    {
      "epoch": 5.0960540771484375e-05,
      "step": 53436,
      "training_step_time": 7.527223587036133
    },
    {
      "epoch": 5.096149444580078e-05,
      "model_forward_time": 1.7283685207366943,
      "step": 53437
    },
    {
      "epoch": 5.096149444580078e-05,
      "step": 53437,
      "training_step_time": 8.775691747665405
    },
    {
      "epoch": 5.096244812011719e-05,
      "model_forward_time": 1.715088129043579,
      "step": 53438
    },
    {
      "epoch": 5.096244812011719e-05,
      "step": 53438,
      "training_step_time": 8.3806471824646
    },
    {
      "epoch": 5.0963401794433594e-05,
      "model_forward_time": 1.7249562740325928,
      "step": 53439
    },
    {
      "epoch": 5.0963401794433594e-05,
      "step": 53439,
      "training_step_time": 7.766841650009155
    },
    {
      "epoch": 5.096435546875e-05,
      "grad_norm": 0.0676329955458641,
      "learning_rate": 4.844764134283392e-06,
      "loss": 0.0549,
      "step": 53440
    },
    {
      "epoch": 5.096435546875e-05,
      "model_forward_time": 2.2735238075256348,
      "step": 53440
    },
    {
      "dynamics_loss_avg": 0.05489203697070479,
      "epoch": 5.096435546875e-05,
      "step": 53440
    },
    {
      "action_loss_avg": 0.0015868040034547448,
      "epoch": 5.096435546875e-05,
      "step": 53440
    },
    {
      "epoch": 5.096435546875e-05,
      "step": 53440,
      "training_step_time": 8.240553617477417
    },
    {
      "epoch": 5.096530914306641e-05,
      "model_forward_time": 1.7151989936828613,
      "step": 53441
    },
    {
      "epoch": 5.096530914306641e-05,
      "step": 53441,
      "training_step_time": 7.794948101043701
    },
    {
      "epoch": 5.096626281738281e-05,
      "model_forward_time": 1.342421293258667,
      "step": 53442
    },
    {
      "epoch": 5.096626281738281e-05,
      "step": 53442,
      "training_step_time": 7.898851156234741
    },
    {
      "epoch": 5.096721649169922e-05,
      "model_forward_time": 1.7164268493652344,
      "step": 53443
    },
    {
      "epoch": 5.096721649169922e-05,
      "step": 53443,
      "training_step_time": 7.721694469451904
    },
    {
      "epoch": 5.096817016601563e-05,
      "model_forward_time": 1.7125542163848877,
      "step": 53444
    },
    {
      "epoch": 5.096817016601563e-05,
      "step": 53444,
      "training_step_time": 7.783736228942871
    },
    {
      "epoch": 5.096912384033203e-05,
      "model_forward_time": 1.7223656177520752,
      "step": 53445
    },
    {
      "epoch": 5.096912384033203e-05,
      "step": 53445,
      "training_step_time": 7.724623680114746
    },
    {
      "epoch": 5.0970077514648436e-05,
      "model_forward_time": 1.7166340351104736,
      "step": 53446
    },
    {
      "epoch": 5.0970077514648436e-05,
      "step": 53446,
      "training_step_time": 7.741546869277954
    },
    {
      "epoch": 5.0971031188964846e-05,
      "model_forward_time": 1.7204194068908691,
      "step": 53447
    },
    {
      "epoch": 5.0971031188964846e-05,
      "step": 53447,
      "training_step_time": 7.653318881988525
    },
    {
      "epoch": 5.097198486328125e-05,
      "model_forward_time": 1.707883596420288,
      "step": 53448
    },
    {
      "epoch": 5.097198486328125e-05,
      "step": 53448,
      "training_step_time": 9.305667877197266
    },
    {
      "epoch": 5.0972938537597654e-05,
      "model_forward_time": 2.2725393772125244,
      "step": 53449
    },
    {
      "epoch": 5.0972938537597654e-05,
      "step": 53449,
      "training_step_time": 7.804670095443726
    },
    {
      "epoch": 5.0973892211914065e-05,
      "grad_norm": 0.09052810817956924,
      "learning_rate": 4.843111470088218e-06,
      "loss": 0.0551,
      "step": 53450
    },
    {
      "epoch": 5.0973892211914065e-05,
      "model_forward_time": 1.3474571704864502,
      "step": 53450
    },
    {
      "dynamics_loss_avg": 0.04243422131985426,
      "epoch": 5.0973892211914065e-05,
      "step": 53450
    },
    {
      "action_loss_avg": 0.0014957261155359447,
      "epoch": 5.0973892211914065e-05,
      "step": 53450
    },
    {
      "epoch": 5.0973892211914065e-05,
      "step": 53450,
      "training_step_time": 9.45188021659851
    },
    {
      "epoch": 5.097484588623047e-05,
      "model_forward_time": 1.7204647064208984,
      "step": 53451
    },
    {
      "epoch": 5.097484588623047e-05,
      "step": 53451,
      "training_step_time": 7.770625591278076
    },
    {
      "epoch": 5.097579956054687e-05,
      "model_forward_time": 1.718358039855957,
      "step": 53452
    },
    {
      "epoch": 5.097579956054687e-05,
      "step": 53452,
      "training_step_time": 8.476011753082275
    },
    {
      "epoch": 5.0976753234863284e-05,
      "model_forward_time": 1.7119619846343994,
      "step": 53453
    },
    {
      "epoch": 5.0976753234863284e-05,
      "step": 53453,
      "training_step_time": 7.638528347015381
    },
    {
      "epoch": 5.097770690917969e-05,
      "model_forward_time": 1.7187261581420898,
      "step": 53454
    },
    {
      "epoch": 5.097770690917969e-05,
      "step": 53454,
      "training_step_time": 7.708808898925781
    },
    {
      "epoch": 5.097866058349609e-05,
      "model_forward_time": 1.7226741313934326,
      "step": 53455
    },
    {
      "epoch": 5.097866058349609e-05,
      "step": 53455,
      "training_step_time": 7.640576362609863
    },
    {
      "epoch": 5.09796142578125e-05,
      "model_forward_time": 1.7205381393432617,
      "step": 53456
    },
    {
      "epoch": 5.09796142578125e-05,
      "step": 53456,
      "training_step_time": 8.67844557762146
    },
    {
      "epoch": 5.098056793212891e-05,
      "model_forward_time": 1.7105381488800049,
      "step": 53457
    },
    {
      "epoch": 5.098056793212891e-05,
      "step": 53457,
      "training_step_time": 7.70902419090271
    },
    {
      "epoch": 5.098152160644531e-05,
      "model_forward_time": 1.768284559249878,
      "step": 53458
    },
    {
      "epoch": 5.098152160644531e-05,
      "step": 53458,
      "training_step_time": 7.7862958908081055
    },
    {
      "epoch": 5.098247528076172e-05,
      "model_forward_time": 2.2828214168548584,
      "step": 53459
    },
    {
      "epoch": 5.098247528076172e-05,
      "step": 53459,
      "training_step_time": 8.529987812042236
    },
    {
      "epoch": 5.0983428955078126e-05,
      "grad_norm": 0.14296409487724304,
      "learning_rate": 4.841458823050137e-06,
      "loss": 0.0565,
      "step": 53460
    },
    {
      "epoch": 5.0983428955078126e-05,
      "model_forward_time": 1.7077083587646484,
      "step": 53460
    },
    {
      "dynamics_loss_avg": 0.055642199516296384,
      "epoch": 5.0983428955078126e-05,
      "step": 53460
    },
    {
      "action_loss_avg": 0.0015741651615826413,
      "epoch": 5.0983428955078126e-05,
      "step": 53460
    },
    {
      "epoch": 5.0983428955078126e-05,
      "step": 53460,
      "training_step_time": 7.844820499420166
    },
    {
      "epoch": 5.098438262939453e-05,
      "model_forward_time": 1.709263801574707,
      "step": 53461
    },
    {
      "epoch": 5.098438262939453e-05,
      "step": 53461,
      "training_step_time": 8.574347019195557
    },
    {
      "epoch": 5.098533630371094e-05,
      "model_forward_time": 1.713179588317871,
      "step": 53462
    },
    {
      "epoch": 5.098533630371094e-05,
      "step": 53462,
      "training_step_time": 8.742784976959229
    },
    {
      "epoch": 5.0986289978027345e-05,
      "model_forward_time": 1.7065577507019043,
      "step": 53463
    },
    {
      "epoch": 5.0986289978027345e-05,
      "step": 53463,
      "training_step_time": 7.7089433670043945
    },
    {
      "epoch": 5.098724365234375e-05,
      "model_forward_time": 1.7179958820343018,
      "step": 53464
    },
    {
      "epoch": 5.098724365234375e-05,
      "step": 53464,
      "training_step_time": 7.566417694091797
    },
    {
      "epoch": 5.098819732666016e-05,
      "model_forward_time": 1.717005968093872,
      "step": 53465
    },
    {
      "epoch": 5.098819732666016e-05,
      "step": 53465,
      "training_step_time": 7.774328708648682
    },
    {
      "epoch": 5.0989151000976564e-05,
      "model_forward_time": 1.707113265991211,
      "step": 53466
    },
    {
      "epoch": 5.0989151000976564e-05,
      "step": 53466,
      "training_step_time": 8.672767162322998
    },
    {
      "epoch": 5.099010467529297e-05,
      "model_forward_time": 1.7615468502044678,
      "step": 53467
    },
    {
      "epoch": 5.099010467529297e-05,
      "step": 53467,
      "training_step_time": 7.795040130615234
    },
    {
      "epoch": 5.099105834960938e-05,
      "model_forward_time": 2.255821943283081,
      "step": 53468
    },
    {
      "epoch": 5.099105834960938e-05,
      "step": 53468,
      "training_step_time": 8.360105037689209
    },
    {
      "epoch": 5.099201202392578e-05,
      "model_forward_time": 1.7194466590881348,
      "step": 53469
    },
    {
      "epoch": 5.099201202392578e-05,
      "step": 53469,
      "training_step_time": 7.843191146850586
    },
    {
      "epoch": 5.0992965698242186e-05,
      "grad_norm": 0.12828710675239563,
      "learning_rate": 4.839806193349882e-06,
      "loss": 0.0596,
      "step": 53470
    },
    {
      "epoch": 5.0992965698242186e-05,
      "model_forward_time": 1.716909408569336,
      "step": 53470
    },
    {
      "dynamics_loss_avg": 0.05216026231646538,
      "epoch": 5.0992965698242186e-05,
      "step": 53470
    },
    {
      "action_loss_avg": 0.001226409457740374,
      "epoch": 5.0992965698242186e-05,
      "step": 53470
    },
    {
      "epoch": 5.0992965698242186e-05,
      "step": 53470,
      "training_step_time": 7.741748571395874
    },
    {
      "epoch": 5.099391937255859e-05,
      "model_forward_time": 1.7148780822753906,
      "step": 53471
    },
    {
      "epoch": 5.099391937255859e-05,
      "step": 53471,
      "training_step_time": 7.66228985786438
    },
    {
      "epoch": 5.0994873046875e-05,
      "model_forward_time": 1.7142510414123535,
      "step": 53472
    },
    {
      "epoch": 5.0994873046875e-05,
      "step": 53472,
      "training_step_time": 7.635006904602051
    },
    {
      "epoch": 5.0995826721191405e-05,
      "model_forward_time": 1.7106590270996094,
      "step": 53473
    },
    {
      "epoch": 5.0995826721191405e-05,
      "step": 53473,
      "training_step_time": 7.5151050090789795
    },
    {
      "epoch": 5.099678039550781e-05,
      "model_forward_time": 1.7127056121826172,
      "step": 53474
    },
    {
      "epoch": 5.099678039550781e-05,
      "step": 53474,
      "training_step_time": 8.573862552642822
    },
    {
      "epoch": 5.099773406982422e-05,
      "model_forward_time": 1.7104766368865967,
      "step": 53475
    },
    {
      "epoch": 5.099773406982422e-05,
      "step": 53475,
      "training_step_time": 7.782868385314941
    },
    {
      "epoch": 5.0998687744140624e-05,
      "model_forward_time": 1.7424201965332031,
      "step": 53476
    },
    {
      "epoch": 5.0998687744140624e-05,
      "step": 53476,
      "training_step_time": 7.769721269607544
    },
    {
      "epoch": 5.099964141845703e-05,
      "model_forward_time": 2.259373664855957,
      "step": 53477
    },
    {
      "epoch": 5.099964141845703e-05,
      "step": 53477,
      "training_step_time": 7.744126319885254
    },
    {
      "epoch": 5.100059509277344e-05,
      "model_forward_time": 1.7111601829528809,
      "step": 53478
    },
    {
      "epoch": 5.100059509277344e-05,
      "step": 53478,
      "training_step_time": 7.781262397766113
    },
    {
      "epoch": 5.100154876708984e-05,
      "model_forward_time": 1.7135848999023438,
      "step": 53479
    },
    {
      "epoch": 5.100154876708984e-05,
      "step": 53479,
      "training_step_time": 8.411689519882202
    },
    {
      "epoch": 5.100250244140625e-05,
      "grad_norm": 0.15174803137779236,
      "learning_rate": 4.83815358116818e-06,
      "loss": 0.0566,
      "step": 53480
    },
    {
      "epoch": 5.100250244140625e-05,
      "model_forward_time": 1.7060656547546387,
      "step": 53480
    },
    {
      "dynamics_loss_avg": 0.05454461202025414,
      "epoch": 5.100250244140625e-05,
      "step": 53480
    },
    {
      "action_loss_avg": 0.0019347752851899714,
      "epoch": 5.100250244140625e-05,
      "step": 53480
    },
    {
      "epoch": 5.100250244140625e-05,
      "step": 53480,
      "training_step_time": 7.645875930786133
    },
    {
      "epoch": 5.100345611572266e-05,
      "model_forward_time": 1.7100565433502197,
      "step": 53481
    },
    {
      "epoch": 5.100345611572266e-05,
      "step": 53481,
      "training_step_time": 7.6438047885894775
    },
    {
      "epoch": 5.100440979003906e-05,
      "model_forward_time": 1.7135910987854004,
      "step": 53482
    },
    {
      "epoch": 5.100440979003906e-05,
      "step": 53482,
      "training_step_time": 7.579454660415649
    },
    {
      "epoch": 5.1005363464355466e-05,
      "model_forward_time": 1.7031028270721436,
      "step": 53483
    },
    {
      "epoch": 5.1005363464355466e-05,
      "step": 53483,
      "training_step_time": 8.604161024093628
    },
    {
      "epoch": 5.100631713867188e-05,
      "model_forward_time": 1.7110669612884521,
      "step": 53484
    },
    {
      "epoch": 5.100631713867188e-05,
      "step": 53484,
      "training_step_time": 7.725947380065918
    },
    {
      "epoch": 5.100727081298828e-05,
      "model_forward_time": 1.716038465499878,
      "step": 53485
    },
    {
      "epoch": 5.100727081298828e-05,
      "step": 53485,
      "training_step_time": 7.667588710784912
    },
    {
      "epoch": 5.1008224487304685e-05,
      "model_forward_time": 2.261054277420044,
      "step": 53486
    },
    {
      "epoch": 5.1008224487304685e-05,
      "step": 53486,
      "training_step_time": 7.74062442779541
    },
    {
      "epoch": 5.1009178161621096e-05,
      "model_forward_time": 1.7107365131378174,
      "step": 53487
    },
    {
      "epoch": 5.1009178161621096e-05,
      "step": 53487,
      "training_step_time": 8.333276271820068
    },
    {
      "epoch": 5.10101318359375e-05,
      "model_forward_time": 1.7101860046386719,
      "step": 53488
    },
    {
      "epoch": 5.10101318359375e-05,
      "step": 53488,
      "training_step_time": 7.843944311141968
    },
    {
      "epoch": 5.1011085510253904e-05,
      "model_forward_time": 1.712303638458252,
      "step": 53489
    },
    {
      "epoch": 5.1011085510253904e-05,
      "step": 53489,
      "training_step_time": 7.689752817153931
    },
    {
      "epoch": 5.1012039184570314e-05,
      "grad_norm": 0.10809171199798584,
      "learning_rate": 4.836500986685759e-06,
      "loss": 0.0592,
      "step": 53490
    },
    {
      "epoch": 5.1012039184570314e-05,
      "model_forward_time": 1.7070012092590332,
      "step": 53490
    },
    {
      "dynamics_loss_avg": 0.06767555437982083,
      "epoch": 5.1012039184570314e-05,
      "step": 53490
    },
    {
      "action_loss_avg": 0.0013768646284006536,
      "epoch": 5.1012039184570314e-05,
      "step": 53490
    },
    {
      "epoch": 5.1012039184570314e-05,
      "step": 53490,
      "training_step_time": 7.775915622711182
    },
    {
      "epoch": 5.101299285888672e-05,
      "model_forward_time": 1.711935043334961,
      "step": 53491
    },
    {
      "epoch": 5.101299285888672e-05,
      "step": 53491,
      "training_step_time": 7.586487770080566
    },
    {
      "epoch": 5.101394653320312e-05,
      "model_forward_time": 1.708374261856079,
      "step": 53492
    },
    {
      "epoch": 5.101394653320312e-05,
      "step": 53492,
      "training_step_time": 9.468825340270996
    },
    {
      "epoch": 5.101490020751953e-05,
      "model_forward_time": 1.706805944442749,
      "step": 53493
    },
    {
      "epoch": 5.101490020751953e-05,
      "step": 53493,
      "training_step_time": 8.601083040237427
    },
    {
      "epoch": 5.101585388183594e-05,
      "model_forward_time": 1.7090609073638916,
      "step": 53494
    },
    {
      "epoch": 5.101585388183594e-05,
      "step": 53494,
      "training_step_time": 9.03495192527771
    },
    {
      "epoch": 5.101680755615234e-05,
      "model_forward_time": 2.2838869094848633,
      "step": 53495
    },
    {
      "epoch": 5.101680755615234e-05,
      "step": 53495,
      "training_step_time": 8.83538007736206
    },
    {
      "epoch": 5.101776123046875e-05,
      "model_forward_time": 1.7108948230743408,
      "step": 53496
    },
    {
      "epoch": 5.101776123046875e-05,
      "step": 53496,
      "training_step_time": 7.831246376037598
    },
    {
      "epoch": 5.1018714904785156e-05,
      "model_forward_time": 1.7094566822052002,
      "step": 53497
    },
    {
      "epoch": 5.1018714904785156e-05,
      "step": 53497,
      "training_step_time": 7.912339448928833
    },
    {
      "epoch": 5.101966857910156e-05,
      "model_forward_time": 1.7004687786102295,
      "step": 53498
    },
    {
      "epoch": 5.101966857910156e-05,
      "step": 53498,
      "training_step_time": 8.4499032497406
    },
    {
      "epoch": 5.102062225341797e-05,
      "model_forward_time": 1.7097852230072021,
      "step": 53499
    },
    {
      "epoch": 5.102062225341797e-05,
      "step": 53499,
      "training_step_time": 9.787796020507812
    },
    {
      "epoch": 5.1021575927734375e-05,
      "grad_norm": 0.09742014110088348,
      "learning_rate": 4.834848410083342e-06,
      "loss": 0.0555,
      "step": 53500
    },
    {
      "epoch": 5.1021575927734375e-05,
      "model_forward_time": 1.7145333290100098,
      "step": 53500
    },
    {
      "dynamics_loss_avg": 0.04813941838219762,
      "epoch": 5.1021575927734375e-05,
      "step": 53500
    },
    {
      "action_loss_avg": 0.001213380292756483,
      "epoch": 5.1021575927734375e-05,
      "step": 53500
    },
    {
      "epoch": 5.1021575927734375e-05,
      "step": 53500,
      "training_step_time": 7.623136758804321
    },
    {
      "epoch": 5.102252960205078e-05,
      "model_forward_time": 1.704310417175293,
      "step": 53501
    },
    {
      "epoch": 5.102252960205078e-05,
      "step": 53501,
      "training_step_time": 7.812614679336548
    },
    {
      "epoch": 5.102348327636719e-05,
      "model_forward_time": 1.7053141593933105,
      "step": 53502
    },
    {
      "epoch": 5.102348327636719e-05,
      "step": 53502,
      "training_step_time": 8.419609785079956
    },
    {
      "epoch": 5.1024436950683594e-05,
      "model_forward_time": 1.701188325881958,
      "step": 53503
    },
    {
      "epoch": 5.1024436950683594e-05,
      "step": 53503,
      "training_step_time": 7.877942085266113
    },
    {
      "epoch": 5.1025390625e-05,
      "model_forward_time": 2.2801530361175537,
      "step": 53504
    },
    {
      "epoch": 5.1025390625e-05,
      "step": 53504,
      "training_step_time": 7.7963547706604
    },
    {
      "epoch": 5.102634429931641e-05,
      "model_forward_time": 1.7235398292541504,
      "step": 53505
    },
    {
      "epoch": 5.102634429931641e-05,
      "step": 53505,
      "training_step_time": 7.824350118637085
    },
    {
      "epoch": 5.102729797363281e-05,
      "model_forward_time": 1.7055656909942627,
      "step": 53506
    },
    {
      "epoch": 5.102729797363281e-05,
      "step": 53506,
      "training_step_time": 7.739087104797363
    },
    {
      "epoch": 5.102825164794922e-05,
      "model_forward_time": 1.6995902061462402,
      "step": 53507
    },
    {
      "epoch": 5.102825164794922e-05,
      "step": 53507,
      "training_step_time": 7.937987804412842
    },
    {
      "epoch": 5.102920532226563e-05,
      "model_forward_time": 1.7206060886383057,
      "step": 53508
    },
    {
      "epoch": 5.102920532226563e-05,
      "step": 53508,
      "training_step_time": 8.713677883148193
    },
    {
      "epoch": 5.103015899658203e-05,
      "model_forward_time": 1.7075159549713135,
      "step": 53509
    },
    {
      "epoch": 5.103015899658203e-05,
      "step": 53509,
      "training_step_time": 8.093515872955322
    },
    {
      "epoch": 5.1031112670898436e-05,
      "grad_norm": 0.1248134970664978,
      "learning_rate": 4.833195851541657e-06,
      "loss": 0.0543,
      "step": 53510
    },
    {
      "epoch": 5.1031112670898436e-05,
      "model_forward_time": 1.6934983730316162,
      "step": 53510
    },
    {
      "dynamics_loss_avg": 0.043920913711190224,
      "epoch": 5.1031112670898436e-05,
      "step": 53510
    },
    {
      "action_loss_avg": 0.0011501501183374784,
      "epoch": 5.1031112670898436e-05,
      "step": 53510
    },
    {
      "epoch": 5.1031112670898436e-05,
      "step": 53510,
      "training_step_time": 7.571439027786255
    },
    {
      "epoch": 5.1032066345214846e-05,
      "model_forward_time": 1.7216298580169678,
      "step": 53511
    },
    {
      "epoch": 5.1032066345214846e-05,
      "step": 53511,
      "training_step_time": 7.7747483253479
    },
    {
      "epoch": 5.103302001953125e-05,
      "model_forward_time": 1.7060608863830566,
      "step": 53512
    },
    {
      "epoch": 5.103302001953125e-05,
      "step": 53512,
      "training_step_time": 7.78931999206543
    },
    {
      "epoch": 5.1033973693847654e-05,
      "model_forward_time": 1.755664348602295,
      "step": 53513
    },
    {
      "epoch": 5.1033973693847654e-05,
      "step": 53513,
      "training_step_time": 11.263805389404297
    },
    {
      "epoch": 5.1034927368164065e-05,
      "model_forward_time": 2.2422797679901123,
      "step": 53514
    },
    {
      "epoch": 5.1034927368164065e-05,
      "step": 53514,
      "training_step_time": 8.776405334472656
    },
    {
      "epoch": 5.103588104248047e-05,
      "model_forward_time": 1.711789846420288,
      "step": 53515
    },
    {
      "epoch": 5.103588104248047e-05,
      "step": 53515,
      "training_step_time": 7.716978073120117
    },
    {
      "epoch": 5.103683471679687e-05,
      "model_forward_time": 1.807992935180664,
      "step": 53516
    },
    {
      "epoch": 5.103683471679687e-05,
      "step": 53516,
      "training_step_time": 7.861834287643433
    },
    {
      "epoch": 5.1037788391113284e-05,
      "model_forward_time": 1.8083951473236084,
      "step": 53517
    },
    {
      "epoch": 5.1037788391113284e-05,
      "step": 53517,
      "training_step_time": 7.646829605102539
    },
    {
      "epoch": 5.103874206542969e-05,
      "model_forward_time": 1.8086178302764893,
      "step": 53518
    },
    {
      "epoch": 5.103874206542969e-05,
      "step": 53518,
      "training_step_time": 7.671804428100586
    },
    {
      "epoch": 5.103969573974609e-05,
      "model_forward_time": 1.8046214580535889,
      "step": 53519
    },
    {
      "epoch": 5.103969573974609e-05,
      "step": 53519,
      "training_step_time": 7.535886287689209
    },
    {
      "epoch": 5.10406494140625e-05,
      "grad_norm": 0.15250851213932037,
      "learning_rate": 4.8315433112414225e-06,
      "loss": 0.0549,
      "step": 53520
    },
    {
      "epoch": 5.10406494140625e-05,
      "model_forward_time": 1.813380479812622,
      "step": 53520
    },
    {
      "dynamics_loss_avg": 0.05082733891904354,
      "epoch": 5.10406494140625e-05,
      "step": 53520
    },
    {
      "action_loss_avg": 0.0015932625625282525,
      "epoch": 5.10406494140625e-05,
      "step": 53520
    },
    {
      "epoch": 5.10406494140625e-05,
      "step": 53520,
      "training_step_time": 7.707583665847778
    },
    {
      "epoch": 5.104160308837891e-05,
      "model_forward_time": 1.8084325790405273,
      "step": 53521
    },
    {
      "epoch": 5.104160308837891e-05,
      "step": 53521,
      "training_step_time": 9.230411291122437
    },
    {
      "epoch": 5.104255676269531e-05,
      "model_forward_time": 1.8067569732666016,
      "step": 53522
    },
    {
      "epoch": 5.104255676269531e-05,
      "step": 53522,
      "training_step_time": 7.781536102294922
    },
    {
      "epoch": 5.104351043701172e-05,
      "model_forward_time": 2.3343420028686523,
      "step": 53523
    },
    {
      "epoch": 5.104351043701172e-05,
      "step": 53523,
      "training_step_time": 9.17513394355774
    },
    {
      "epoch": 5.1044464111328126e-05,
      "model_forward_time": 1.8067576885223389,
      "step": 53524
    },
    {
      "epoch": 5.1044464111328126e-05,
      "step": 53524,
      "training_step_time": 7.777270078659058
    },
    {
      "epoch": 5.104541778564453e-05,
      "model_forward_time": 1.835857629776001,
      "step": 53525
    },
    {
      "epoch": 5.104541778564453e-05,
      "step": 53525,
      "training_step_time": 7.779141664505005
    },
    {
      "epoch": 5.104637145996094e-05,
      "model_forward_time": 1.828688383102417,
      "step": 53526
    },
    {
      "epoch": 5.104637145996094e-05,
      "step": 53526,
      "training_step_time": 8.740735530853271
    },
    {
      "epoch": 5.1047325134277345e-05,
      "model_forward_time": 1.8265888690948486,
      "step": 53527
    },
    {
      "epoch": 5.1047325134277345e-05,
      "step": 53527,
      "training_step_time": 7.584137678146362
    },
    {
      "epoch": 5.104827880859375e-05,
      "model_forward_time": 1.8313102722167969,
      "step": 53528
    },
    {
      "epoch": 5.104827880859375e-05,
      "step": 53528,
      "training_step_time": 7.5163819789886475
    },
    {
      "epoch": 5.104923248291016e-05,
      "model_forward_time": 1.8326725959777832,
      "step": 53529
    },
    {
      "epoch": 5.104923248291016e-05,
      "step": 53529,
      "training_step_time": 8.416473388671875
    },
    {
      "epoch": 5.1050186157226564e-05,
      "grad_norm": 0.12734633684158325,
      "learning_rate": 4.829890789363359e-06,
      "loss": 0.0544,
      "step": 53530
    },
    {
      "epoch": 5.1050186157226564e-05,
      "model_forward_time": 1.829322338104248,
      "step": 53530
    },
    {
      "dynamics_loss_avg": 0.04458579272031784,
      "epoch": 5.1050186157226564e-05,
      "step": 53530
    },
    {
      "action_loss_avg": 0.0012501261895522474,
      "epoch": 5.1050186157226564e-05,
      "step": 53530
    },
    {
      "epoch": 5.1050186157226564e-05,
      "step": 53530,
      "training_step_time": 7.68678092956543
    },
    {
      "epoch": 5.105113983154297e-05,
      "model_forward_time": 1.8250815868377686,
      "step": 53531
    },
    {
      "epoch": 5.105113983154297e-05,
      "step": 53531,
      "training_step_time": 9.425863265991211
    },
    {
      "epoch": 5.105209350585937e-05,
      "model_forward_time": 2.3588902950286865,
      "step": 53532
    },
    {
      "epoch": 5.105209350585937e-05,
      "step": 53532,
      "training_step_time": 8.594197750091553
    },
    {
      "epoch": 5.105304718017578e-05,
      "model_forward_time": 1.827822208404541,
      "step": 53533
    },
    {
      "epoch": 5.105304718017578e-05,
      "step": 53533,
      "training_step_time": 7.7431182861328125
    },
    {
      "epoch": 5.1054000854492186e-05,
      "model_forward_time": 1.8308589458465576,
      "step": 53534
    },
    {
      "epoch": 5.1054000854492186e-05,
      "step": 53534,
      "training_step_time": 7.78408145904541
    },
    {
      "epoch": 5.105495452880859e-05,
      "model_forward_time": 1.8358190059661865,
      "step": 53535
    },
    {
      "epoch": 5.105495452880859e-05,
      "step": 53535,
      "training_step_time": 8.307817697525024
    },
    {
      "epoch": 5.1055908203125e-05,
      "model_forward_time": 1.8031306266784668,
      "step": 53536
    },
    {
      "epoch": 5.1055908203125e-05,
      "step": 53536,
      "training_step_time": 7.65518045425415
    },
    {
      "epoch": 5.1056861877441405e-05,
      "model_forward_time": 1.805804967880249,
      "step": 53537
    },
    {
      "epoch": 5.1056861877441405e-05,
      "step": 53537,
      "training_step_time": 7.573246717453003
    },
    {
      "epoch": 5.105781555175781e-05,
      "model_forward_time": 1.8057506084442139,
      "step": 53538
    },
    {
      "epoch": 5.105781555175781e-05,
      "step": 53538,
      "training_step_time": 8.415727615356445
    },
    {
      "epoch": 5.105876922607422e-05,
      "model_forward_time": 1.8053524494171143,
      "step": 53539
    },
    {
      "epoch": 5.105876922607422e-05,
      "step": 53539,
      "training_step_time": 8.882608652114868
    },
    {
      "epoch": 5.1059722900390624e-05,
      "grad_norm": 0.12465445697307587,
      "learning_rate": 4.828238286088183e-06,
      "loss": 0.0601,
      "step": 53540
    },
    {
      "epoch": 5.1059722900390624e-05,
      "model_forward_time": 1.8002924919128418,
      "step": 53540
    },
    {
      "dynamics_loss_avg": 0.06324572134763003,
      "epoch": 5.1059722900390624e-05,
      "step": 53540
    },
    {
      "action_loss_avg": 0.0017673161724815144,
      "epoch": 5.1059722900390624e-05,
      "step": 53540
    },
    {
      "epoch": 5.1059722900390624e-05,
      "step": 53540,
      "training_step_time": 8.385085105895996
    },
    {
      "epoch": 5.106067657470703e-05,
      "model_forward_time": 2.3607687950134277,
      "step": 53541
    },
    {
      "epoch": 5.106067657470703e-05,
      "step": 53541,
      "training_step_time": 7.8235156536102295
    },
    {
      "epoch": 5.106163024902344e-05,
      "model_forward_time": 1.8116400241851807,
      "step": 53542
    },
    {
      "epoch": 5.106163024902344e-05,
      "step": 53542,
      "training_step_time": 8.483433961868286
    },
    {
      "epoch": 5.106258392333984e-05,
      "model_forward_time": 1.8034288883209229,
      "step": 53543
    },
    {
      "epoch": 5.106258392333984e-05,
      "step": 53543,
      "training_step_time": 7.7150797843933105
    },
    {
      "epoch": 5.106353759765625e-05,
      "model_forward_time": 1.8073625564575195,
      "step": 53544
    },
    {
      "epoch": 5.106353759765625e-05,
      "step": 53544,
      "training_step_time": 7.816403865814209
    },
    {
      "epoch": 5.106449127197266e-05,
      "model_forward_time": 1.8080594539642334,
      "step": 53545
    },
    {
      "epoch": 5.106449127197266e-05,
      "step": 53545,
      "training_step_time": 8.211987495422363
    },
    {
      "epoch": 5.106544494628906e-05,
      "model_forward_time": 1.8077092170715332,
      "step": 53546
    },
    {
      "epoch": 5.106544494628906e-05,
      "step": 53546,
      "training_step_time": 8.160626411437988
    },
    {
      "epoch": 5.1066398620605466e-05,
      "model_forward_time": 1.8176543712615967,
      "step": 53547
    },
    {
      "epoch": 5.1066398620605466e-05,
      "step": 53547,
      "training_step_time": 7.706061363220215
    },
    {
      "epoch": 5.106735229492188e-05,
      "model_forward_time": 1.8089795112609863,
      "step": 53548
    },
    {
      "epoch": 5.106735229492188e-05,
      "step": 53548,
      "training_step_time": 7.763102769851685
    },
    {
      "epoch": 5.106830596923828e-05,
      "model_forward_time": 1.8036649227142334,
      "step": 53549
    },
    {
      "epoch": 5.106830596923828e-05,
      "step": 53549,
      "training_step_time": 8.788810729980469
    },
    {
      "epoch": 5.1069259643554685e-05,
      "grad_norm": 0.10054842382669449,
      "learning_rate": 4.826585801596609e-06,
      "loss": 0.0554,
      "step": 53550
    },
    {
      "epoch": 5.1069259643554685e-05,
      "model_forward_time": 2.3586244583129883,
      "step": 53550
    },
    {
      "dynamics_loss_avg": 0.03902257513254881,
      "epoch": 5.1069259643554685e-05,
      "step": 53550
    },
    {
      "action_loss_avg": 0.0014863261894788593,
      "epoch": 5.1069259643554685e-05,
      "step": 53550
    },
    {
      "epoch": 5.1069259643554685e-05,
      "step": 53550,
      "training_step_time": 8.657230615615845
    },
    {
      "epoch": 5.1070213317871096e-05,
      "model_forward_time": 1.8090765476226807,
      "step": 53551
    },
    {
      "epoch": 5.1070213317871096e-05,
      "step": 53551,
      "training_step_time": 10.233594417572021
    },
    {
      "epoch": 5.10711669921875e-05,
      "model_forward_time": 1.8044779300689697,
      "step": 53552
    },
    {
      "epoch": 5.10711669921875e-05,
      "step": 53552,
      "training_step_time": 7.733339309692383
    },
    {
      "epoch": 5.1072120666503904e-05,
      "model_forward_time": 1.8033263683319092,
      "step": 53553
    },
    {
      "epoch": 5.1072120666503904e-05,
      "step": 53553,
      "training_step_time": 7.684971570968628
    },
    {
      "epoch": 5.1073074340820314e-05,
      "model_forward_time": 1.0840866565704346,
      "step": 53554
    },
    {
      "epoch": 5.1073074340820314e-05,
      "step": 53554,
      "training_step_time": 8.524608135223389
    },
    {
      "epoch": 5.107402801513672e-05,
      "model_forward_time": 1.8023450374603271,
      "step": 53555
    },
    {
      "epoch": 5.107402801513672e-05,
      "step": 53555,
      "training_step_time": 7.609966516494751
    },
    {
      "epoch": 5.107498168945312e-05,
      "model_forward_time": 1.8125743865966797,
      "step": 53556
    },
    {
      "epoch": 5.107498168945312e-05,
      "step": 53556,
      "training_step_time": 7.507490634918213
    },
    {
      "epoch": 5.107593536376953e-05,
      "model_forward_time": 1.8060801029205322,
      "step": 53557
    },
    {
      "epoch": 5.107593536376953e-05,
      "step": 53557,
      "training_step_time": 7.7245378494262695
    },
    {
      "epoch": 5.107688903808594e-05,
      "model_forward_time": 1.8139753341674805,
      "step": 53558
    },
    {
      "epoch": 5.107688903808594e-05,
      "step": 53558,
      "training_step_time": 8.454699754714966
    },
    {
      "epoch": 5.107784271240234e-05,
      "model_forward_time": 1.8104524612426758,
      "step": 53559
    },
    {
      "epoch": 5.107784271240234e-05,
      "step": 53559,
      "training_step_time": 7.767446041107178
    },
    {
      "epoch": 5.107879638671875e-05,
      "grad_norm": 0.13276150822639465,
      "learning_rate": 4.824933336069349e-06,
      "loss": 0.0565,
      "step": 53560
    },
    {
      "epoch": 5.107879638671875e-05,
      "model_forward_time": 2.3372888565063477,
      "step": 53560
    },
    {
      "dynamics_loss_avg": 0.040328220929950476,
      "epoch": 5.107879638671875e-05,
      "step": 53560
    },
    {
      "action_loss_avg": 0.0009755477571161464,
      "epoch": 5.107879638671875e-05,
      "step": 53560
    },
    {
      "epoch": 5.107879638671875e-05,
      "step": 53560,
      "training_step_time": 8.620798826217651
    },
    {
      "epoch": 5.1079750061035156e-05,
      "model_forward_time": 1.8063337802886963,
      "step": 53561
    },
    {
      "epoch": 5.1079750061035156e-05,
      "step": 53561,
      "training_step_time": 8.812949419021606
    },
    {
      "epoch": 5.108070373535156e-05,
      "model_forward_time": 1.8091039657592773,
      "step": 53562
    },
    {
      "epoch": 5.108070373535156e-05,
      "step": 53562,
      "training_step_time": 8.437501668930054
    },
    {
      "epoch": 5.108165740966797e-05,
      "model_forward_time": 1.797729730606079,
      "step": 53563
    },
    {
      "epoch": 5.108165740966797e-05,
      "step": 53563,
      "training_step_time": 7.707067251205444
    },
    {
      "epoch": 5.1082611083984375e-05,
      "model_forward_time": 1.8074822425842285,
      "step": 53564
    },
    {
      "epoch": 5.1082611083984375e-05,
      "step": 53564,
      "training_step_time": 7.593103408813477
    },
    {
      "epoch": 5.108356475830078e-05,
      "model_forward_time": 1.7941322326660156,
      "step": 53565
    },
    {
      "epoch": 5.108356475830078e-05,
      "step": 53565,
      "training_step_time": 7.655205011367798
    },
    {
      "epoch": 5.108451843261719e-05,
      "model_forward_time": 1.8137798309326172,
      "step": 53566
    },
    {
      "epoch": 5.108451843261719e-05,
      "step": 53566,
      "training_step_time": 8.740150451660156
    },
    {
      "epoch": 5.1085472106933594e-05,
      "model_forward_time": 1.4585340023040771,
      "step": 53567
    },
    {
      "epoch": 5.1085472106933594e-05,
      "step": 53567,
      "training_step_time": 8.509523391723633
    },
    {
      "epoch": 5.108642578125e-05,
      "model_forward_time": 1.8345286846160889,
      "step": 53568
    },
    {
      "epoch": 5.108642578125e-05,
      "step": 53568,
      "training_step_time": 8.434936046600342
    },
    {
      "epoch": 5.108737945556641e-05,
      "model_forward_time": 2.4033243656158447,
      "step": 53569
    },
    {
      "epoch": 5.108737945556641e-05,
      "step": 53569,
      "training_step_time": 8.619899988174438
    },
    {
      "epoch": 5.108833312988281e-05,
      "grad_norm": 0.13200494647026062,
      "learning_rate": 4.823280889687119e-06,
      "loss": 0.0589,
      "step": 53570
    },
    {
      "epoch": 5.108833312988281e-05,
      "model_forward_time": 1.8423030376434326,
      "step": 53570
    },
    {
      "dynamics_loss_avg": 0.06069393996149301,
      "epoch": 5.108833312988281e-05,
      "step": 53570
    },
    {
      "action_loss_avg": 0.0018418616615235805,
      "epoch": 5.108833312988281e-05,
      "step": 53570
    },
    {
      "epoch": 5.108833312988281e-05,
      "step": 53570,
      "training_step_time": 7.714960336685181
    },
    {
      "epoch": 5.108928680419922e-05,
      "model_forward_time": 1.8286404609680176,
      "step": 53571
    },
    {
      "epoch": 5.108928680419922e-05,
      "step": 53571,
      "training_step_time": 7.722227334976196
    },
    {
      "epoch": 5.109024047851563e-05,
      "model_forward_time": 1.831801176071167,
      "step": 53572
    },
    {
      "epoch": 5.109024047851563e-05,
      "step": 53572,
      "training_step_time": 8.81263542175293
    },
    {
      "epoch": 5.109119415283203e-05,
      "model_forward_time": 1.8339614868164062,
      "step": 53573
    },
    {
      "epoch": 5.109119415283203e-05,
      "step": 53573,
      "training_step_time": 8.635213851928711
    },
    {
      "epoch": 5.1092147827148436e-05,
      "model_forward_time": 1.833155632019043,
      "step": 53574
    },
    {
      "epoch": 5.1092147827148436e-05,
      "step": 53574,
      "training_step_time": 7.826335430145264
    },
    {
      "epoch": 5.1093101501464846e-05,
      "model_forward_time": 1.8158140182495117,
      "step": 53575
    },
    {
      "epoch": 5.1093101501464846e-05,
      "step": 53575,
      "training_step_time": 7.806198835372925
    },
    {
      "epoch": 5.109405517578125e-05,
      "model_forward_time": 1.808685302734375,
      "step": 53576
    },
    {
      "epoch": 5.109405517578125e-05,
      "step": 53576,
      "training_step_time": 7.88214635848999
    },
    {
      "epoch": 5.1095008850097655e-05,
      "model_forward_time": 1.8066070079803467,
      "step": 53577
    },
    {
      "epoch": 5.1095008850097655e-05,
      "step": 53577,
      "training_step_time": 9.393340349197388
    },
    {
      "epoch": 5.1095962524414065e-05,
      "model_forward_time": 2.356290817260742,
      "step": 53578
    },
    {
      "epoch": 5.1095962524414065e-05,
      "step": 53578,
      "training_step_time": 7.7833168506622314
    },
    {
      "epoch": 5.109691619873047e-05,
      "model_forward_time": 1.810943365097046,
      "step": 53579
    },
    {
      "epoch": 5.109691619873047e-05,
      "step": 53579,
      "training_step_time": 8.716448545455933
    },
    {
      "epoch": 5.109786987304687e-05,
      "grad_norm": 0.19380700588226318,
      "learning_rate": 4.821628462630624e-06,
      "loss": 0.0608,
      "step": 53580
    },
    {
      "epoch": 5.109786987304687e-05,
      "model_forward_time": 1.811933994293213,
      "step": 53580
    },
    {
      "dynamics_loss_avg": 0.05265944208949804,
      "epoch": 5.109786987304687e-05,
      "step": 53580
    },
    {
      "action_loss_avg": 0.0014618179673561826,
      "epoch": 5.109786987304687e-05,
      "step": 53580
    },
    {
      "epoch": 5.109786987304687e-05,
      "step": 53580,
      "training_step_time": 7.744124889373779
    },
    {
      "epoch": 5.1098823547363284e-05,
      "model_forward_time": 1.8113548755645752,
      "step": 53581
    },
    {
      "epoch": 5.1098823547363284e-05,
      "step": 53581,
      "training_step_time": 8.715094566345215
    },
    {
      "epoch": 5.109977722167969e-05,
      "model_forward_time": 1.4535839557647705,
      "step": 53582
    },
    {
      "epoch": 5.109977722167969e-05,
      "step": 53582,
      "training_step_time": 7.848254442214966
    },
    {
      "epoch": 5.110073089599609e-05,
      "model_forward_time": 1.807673692703247,
      "step": 53583
    },
    {
      "epoch": 5.110073089599609e-05,
      "step": 53583,
      "training_step_time": 7.718496799468994
    },
    {
      "epoch": 5.11016845703125e-05,
      "model_forward_time": 1.8106133937835693,
      "step": 53584
    },
    {
      "epoch": 5.11016845703125e-05,
      "step": 53584,
      "training_step_time": 8.609713554382324
    },
    {
      "epoch": 5.110263824462891e-05,
      "model_forward_time": 1.8093738555908203,
      "step": 53585
    },
    {
      "epoch": 5.110263824462891e-05,
      "step": 53585,
      "training_step_time": 7.807778835296631
    },
    {
      "epoch": 5.110359191894531e-05,
      "model_forward_time": 1.8133771419525146,
      "step": 53586
    },
    {
      "epoch": 5.110359191894531e-05,
      "step": 53586,
      "training_step_time": 7.851245164871216
    },
    {
      "epoch": 5.110454559326172e-05,
      "model_forward_time": 1.815861701965332,
      "step": 53587
    },
    {
      "epoch": 5.110454559326172e-05,
      "step": 53587,
      "training_step_time": 8.783557653427124
    },
    {
      "epoch": 5.1105499267578126e-05,
      "model_forward_time": 2.334791421890259,
      "step": 53588
    },
    {
      "epoch": 5.1105499267578126e-05,
      "step": 53588,
      "training_step_time": 9.80791974067688
    },
    {
      "epoch": 5.110645294189453e-05,
      "model_forward_time": 1.7903752326965332,
      "step": 53589
    },
    {
      "epoch": 5.110645294189453e-05,
      "step": 53589,
      "training_step_time": 8.366793155670166
    },
    {
      "epoch": 5.110740661621094e-05,
      "grad_norm": 0.11240603029727936,
      "learning_rate": 4.819976055080574e-06,
      "loss": 0.0622,
      "step": 53590
    },
    {
      "epoch": 5.110740661621094e-05,
      "model_forward_time": 1.3402373790740967,
      "step": 53590
    },
    {
      "dynamics_loss_avg": 0.06447495762258768,
      "epoch": 5.110740661621094e-05,
      "step": 53590
    },
    {
      "action_loss_avg": 0.0012990113202249632,
      "epoch": 5.110740661621094e-05,
      "step": 53590
    },
    {
      "epoch": 5.110740661621094e-05,
      "step": 53590,
      "training_step_time": 7.82802939414978
    },
    {
      "epoch": 5.1108360290527345e-05,
      "model_forward_time": 1.7221362590789795,
      "step": 53591
    },
    {
      "epoch": 5.1108360290527345e-05,
      "step": 53591,
      "training_step_time": 9.159707307815552
    },
    {
      "epoch": 5.110931396484375e-05,
      "model_forward_time": 1.7125189304351807,
      "step": 53592
    },
    {
      "epoch": 5.110931396484375e-05,
      "step": 53592,
      "training_step_time": 8.248632431030273
    },
    {
      "epoch": 5.111026763916016e-05,
      "model_forward_time": 1.7046220302581787,
      "step": 53593
    },
    {
      "epoch": 5.111026763916016e-05,
      "step": 53593,
      "training_step_time": 9.578883647918701
    },
    {
      "epoch": 5.1111221313476564e-05,
      "model_forward_time": 1.7407350540161133,
      "step": 53594
    },
    {
      "epoch": 5.1111221313476564e-05,
      "step": 53594,
      "training_step_time": 7.7119059562683105
    },
    {
      "epoch": 5.111217498779297e-05,
      "model_forward_time": 1.7522368431091309,
      "step": 53595
    },
    {
      "epoch": 5.111217498779297e-05,
      "step": 53595,
      "training_step_time": 7.8022894859313965
    },
    {
      "epoch": 5.111312866210937e-05,
      "model_forward_time": 1.7493257522583008,
      "step": 53596
    },
    {
      "epoch": 5.111312866210937e-05,
      "step": 53596,
      "training_step_time": 7.645413398742676
    },
    {
      "epoch": 5.111408233642578e-05,
      "model_forward_time": 2.319467544555664,
      "step": 53597
    },
    {
      "epoch": 5.111408233642578e-05,
      "step": 53597,
      "training_step_time": 7.7578558921813965
    },
    {
      "epoch": 5.1115036010742187e-05,
      "model_forward_time": 1.7484302520751953,
      "step": 53598
    },
    {
      "epoch": 5.1115036010742187e-05,
      "step": 53598,
      "training_step_time": 8.525995016098022
    },
    {
      "epoch": 5.111598968505859e-05,
      "model_forward_time": 1.7458617687225342,
      "step": 53599
    },
    {
      "epoch": 5.111598968505859e-05,
      "step": 53599,
      "training_step_time": 7.740381717681885
    },
    {
      "epoch": 5.1116943359375e-05,
      "grad_norm": 0.11846823990345001,
      "learning_rate": 4.818323667217669e-06,
      "loss": 0.0578,
      "step": 53600
    },
    {
      "epoch": 5.1116943359375e-05,
      "model_forward_time": 1.7336199283599854,
      "step": 53600
    },
    {
      "dynamics_loss_avg": 0.05104355290532112,
      "epoch": 5.1116943359375e-05,
      "step": 53600
    },
    {
      "action_loss_avg": 0.001198366525932215,
      "epoch": 5.1116943359375e-05,
      "step": 53600
    },
    {
      "epoch": 5.1116943359375e-05,
      "step": 53600,
      "training_step_time": 7.77048397064209
    },
    {
      "epoch": 5.1117897033691405e-05,
      "model_forward_time": 1.7422537803649902,
      "step": 53601
    },
    {
      "epoch": 5.1117897033691405e-05,
      "step": 53601,
      "training_step_time": 9.186317205429077
    },
    {
      "epoch": 5.111885070800781e-05,
      "model_forward_time": 1.7440993785858154,
      "step": 53602
    },
    {
      "epoch": 5.111885070800781e-05,
      "step": 53602,
      "training_step_time": 7.554753065109253
    },
    {
      "epoch": 5.111980438232422e-05,
      "model_forward_time": 1.7448952198028564,
      "step": 53603
    },
    {
      "epoch": 5.111980438232422e-05,
      "step": 53603,
      "training_step_time": 8.603673696517944
    },
    {
      "epoch": 5.1120758056640624e-05,
      "model_forward_time": 1.746584177017212,
      "step": 53604
    },
    {
      "epoch": 5.1120758056640624e-05,
      "step": 53604,
      "training_step_time": 9.403913497924805
    },
    {
      "epoch": 5.112171173095703e-05,
      "model_forward_time": 1.8178317546844482,
      "step": 53605
    },
    {
      "epoch": 5.112171173095703e-05,
      "step": 53605,
      "training_step_time": 8.347296476364136
    },
    {
      "epoch": 5.112266540527344e-05,
      "model_forward_time": 2.286222457885742,
      "step": 53606
    },
    {
      "epoch": 5.112266540527344e-05,
      "step": 53606,
      "training_step_time": 7.6420817375183105
    },
    {
      "epoch": 5.112361907958984e-05,
      "model_forward_time": 1.7435855865478516,
      "step": 53607
    },
    {
      "epoch": 5.112361907958984e-05,
      "step": 53607,
      "training_step_time": 8.565024852752686
    },
    {
      "epoch": 5.112457275390625e-05,
      "model_forward_time": 1.7485237121582031,
      "step": 53608
    },
    {
      "epoch": 5.112457275390625e-05,
      "step": 53608,
      "training_step_time": 7.609604597091675
    },
    {
      "epoch": 5.112552642822266e-05,
      "model_forward_time": 1.7438442707061768,
      "step": 53609
    },
    {
      "epoch": 5.112552642822266e-05,
      "step": 53609,
      "training_step_time": 7.791245460510254
    },
    {
      "epoch": 5.112648010253906e-05,
      "grad_norm": 0.11073750257492065,
      "learning_rate": 4.8166712992226165e-06,
      "loss": 0.0618,
      "step": 53610
    },
    {
      "epoch": 5.112648010253906e-05,
      "model_forward_time": 1.7318663597106934,
      "step": 53610
    },
    {
      "dynamics_loss_avg": 0.0523621616885066,
      "epoch": 5.112648010253906e-05,
      "step": 53610
    },
    {
      "action_loss_avg": 0.0012111549498513342,
      "epoch": 5.112648010253906e-05,
      "step": 53610
    },
    {
      "epoch": 5.112648010253906e-05,
      "step": 53610,
      "training_step_time": 8.128398418426514
    },
    {
      "epoch": 5.1127433776855466e-05,
      "model_forward_time": 1.7593247890472412,
      "step": 53611
    },
    {
      "epoch": 5.1127433776855466e-05,
      "step": 53611,
      "training_step_time": 7.702124834060669
    },
    {
      "epoch": 5.112838745117188e-05,
      "model_forward_time": 1.7411909103393555,
      "step": 53612
    },
    {
      "epoch": 5.112838745117188e-05,
      "step": 53612,
      "training_step_time": 8.554308891296387
    },
    {
      "epoch": 5.112934112548828e-05,
      "model_forward_time": 1.7413115501403809,
      "step": 53613
    },
    {
      "epoch": 5.112934112548828e-05,
      "step": 53613,
      "training_step_time": 8.53816294670105
    },
    {
      "epoch": 5.1130294799804685e-05,
      "model_forward_time": 1.8159205913543701,
      "step": 53614
    },
    {
      "epoch": 5.1130294799804685e-05,
      "step": 53614,
      "training_step_time": 8.37076711654663
    },
    {
      "epoch": 5.1131248474121096e-05,
      "model_forward_time": 2.3271546363830566,
      "step": 53615
    },
    {
      "epoch": 5.1131248474121096e-05,
      "step": 53615,
      "training_step_time": 7.75379753112793
    },
    {
      "epoch": 5.11322021484375e-05,
      "model_forward_time": 1.755054235458374,
      "step": 53616
    },
    {
      "epoch": 5.11322021484375e-05,
      "step": 53616,
      "training_step_time": 8.634618043899536
    },
    {
      "epoch": 5.1133155822753904e-05,
      "model_forward_time": 1.745147705078125,
      "step": 53617
    },
    {
      "epoch": 5.1133155822753904e-05,
      "step": 53617,
      "training_step_time": 7.684011220932007
    },
    {
      "epoch": 5.1134109497070315e-05,
      "model_forward_time": 1.7447991371154785,
      "step": 53618
    },
    {
      "epoch": 5.1134109497070315e-05,
      "step": 53618,
      "training_step_time": 7.74083137512207
    },
    {
      "epoch": 5.113506317138672e-05,
      "model_forward_time": 1.742687702178955,
      "step": 53619
    },
    {
      "epoch": 5.113506317138672e-05,
      "step": 53619,
      "training_step_time": 7.765888214111328
    },
    {
      "epoch": 5.113601684570312e-05,
      "grad_norm": 0.10530200600624084,
      "learning_rate": 4.815018951276115e-06,
      "loss": 0.0602,
      "step": 53620
    },
    {
      "epoch": 5.113601684570312e-05,
      "model_forward_time": 1.7389748096466064,
      "step": 53620
    },
    {
      "dynamics_loss_avg": 0.05679342746734619,
      "epoch": 5.113601684570312e-05,
      "step": 53620
    },
    {
      "action_loss_avg": 0.0014749703637789934,
      "epoch": 5.113601684570312e-05,
      "step": 53620
    },
    {
      "epoch": 5.113601684570312e-05,
      "step": 53620,
      "training_step_time": 7.768735885620117
    },
    {
      "epoch": 5.113697052001953e-05,
      "model_forward_time": 1.747309684753418,
      "step": 53621
    },
    {
      "epoch": 5.113697052001953e-05,
      "step": 53621,
      "training_step_time": 7.688745975494385
    },
    {
      "epoch": 5.113792419433594e-05,
      "model_forward_time": 1.7425017356872559,
      "step": 53622
    },
    {
      "epoch": 5.113792419433594e-05,
      "step": 53622,
      "training_step_time": 8.6142897605896
    },
    {
      "epoch": 5.113887786865234e-05,
      "model_forward_time": 1.8147876262664795,
      "step": 53623
    },
    {
      "epoch": 5.113887786865234e-05,
      "step": 53623,
      "training_step_time": 7.796247959136963
    },
    {
      "epoch": 5.113983154296875e-05,
      "model_forward_time": 2.286313056945801,
      "step": 53624
    },
    {
      "epoch": 5.113983154296875e-05,
      "step": 53624,
      "training_step_time": 8.445666551589966
    },
    {
      "epoch": 5.1140785217285156e-05,
      "model_forward_time": 1.7297775745391846,
      "step": 53625
    },
    {
      "epoch": 5.1140785217285156e-05,
      "step": 53625,
      "training_step_time": 7.716490983963013
    },
    {
      "epoch": 5.114173889160156e-05,
      "model_forward_time": 1.3570315837860107,
      "step": 53626
    },
    {
      "epoch": 5.114173889160156e-05,
      "step": 53626,
      "training_step_time": 7.640974044799805
    },
    {
      "epoch": 5.114269256591797e-05,
      "model_forward_time": 1.74495267868042,
      "step": 53627
    },
    {
      "epoch": 5.114269256591797e-05,
      "step": 53627,
      "training_step_time": 7.718912363052368
    },
    {
      "epoch": 5.1143646240234375e-05,
      "model_forward_time": 1.7451386451721191,
      "step": 53628
    },
    {
      "epoch": 5.1143646240234375e-05,
      "step": 53628,
      "training_step_time": 8.582780361175537
    },
    {
      "epoch": 5.114459991455078e-05,
      "model_forward_time": 1.7514917850494385,
      "step": 53629
    },
    {
      "epoch": 5.114459991455078e-05,
      "step": 53629,
      "training_step_time": 7.659580945968628
    },
    {
      "epoch": 5.114555358886719e-05,
      "grad_norm": 0.0818176344037056,
      "learning_rate": 4.8133666235588614e-06,
      "loss": 0.0586,
      "step": 53630
    },
    {
      "epoch": 5.114555358886719e-05,
      "model_forward_time": 1.7406322956085205,
      "step": 53630
    },
    {
      "dynamics_loss_avg": 0.06978541538119316,
      "epoch": 5.114555358886719e-05,
      "step": 53630
    },
    {
      "action_loss_avg": 0.0017029856593580917,
      "epoch": 5.114555358886719e-05,
      "step": 53630
    },
    {
      "epoch": 5.114555358886719e-05,
      "step": 53630,
      "training_step_time": 8.341945171356201
    },
    {
      "epoch": 5.1146507263183594e-05,
      "model_forward_time": 1.7436730861663818,
      "step": 53631
    },
    {
      "epoch": 5.1146507263183594e-05,
      "step": 53631,
      "training_step_time": 8.619096040725708
    },
    {
      "epoch": 5.11474609375e-05,
      "model_forward_time": 1.7838780879974365,
      "step": 53632
    },
    {
      "epoch": 5.11474609375e-05,
      "step": 53632,
      "training_step_time": 7.7634923458099365
    },
    {
      "epoch": 5.114841461181641e-05,
      "model_forward_time": 1.7333738803863525,
      "step": 53633
    },
    {
      "epoch": 5.114841461181641e-05,
      "step": 53633,
      "training_step_time": 7.732156276702881
    },
    {
      "epoch": 5.114936828613281e-05,
      "model_forward_time": 1.9032492637634277,
      "step": 53634
    },
    {
      "epoch": 5.114936828613281e-05,
      "step": 53634,
      "training_step_time": 8.577557802200317
    },
    {
      "epoch": 5.115032196044922e-05,
      "model_forward_time": 1.7327654361724854,
      "step": 53635
    },
    {
      "epoch": 5.115032196044922e-05,
      "step": 53635,
      "training_step_time": 8.632285356521606
    },
    {
      "epoch": 5.115127563476563e-05,
      "model_forward_time": 1.742558240890503,
      "step": 53636
    },
    {
      "epoch": 5.115127563476563e-05,
      "step": 53636,
      "training_step_time": 8.768061637878418
    },
    {
      "epoch": 5.115222930908203e-05,
      "model_forward_time": 1.7529497146606445,
      "step": 53637
    },
    {
      "epoch": 5.115222930908203e-05,
      "step": 53637,
      "training_step_time": 7.568368911743164
    },
    {
      "epoch": 5.1153182983398436e-05,
      "model_forward_time": 1.7389490604400635,
      "step": 53638
    },
    {
      "epoch": 5.1153182983398436e-05,
      "step": 53638,
      "training_step_time": 8.747769832611084
    },
    {
      "epoch": 5.1154136657714847e-05,
      "model_forward_time": 1.7461493015289307,
      "step": 53639
    },
    {
      "epoch": 5.1154136657714847e-05,
      "step": 53639,
      "training_step_time": 7.657132625579834
    },
    {
      "epoch": 5.115509033203125e-05,
      "grad_norm": 0.1163363978266716,
      "learning_rate": 4.811714316251554e-06,
      "loss": 0.0576,
      "step": 53640
    },
    {
      "epoch": 5.115509033203125e-05,
      "model_forward_time": 1.3647253513336182,
      "step": 53640
    },
    {
      "dynamics_loss_avg": 0.05873875878751278,
      "epoch": 5.115509033203125e-05,
      "step": 53640
    },
    {
      "action_loss_avg": 0.0015886394074186682,
      "epoch": 5.115509033203125e-05,
      "step": 53640
    },
    {
      "epoch": 5.115509033203125e-05,
      "step": 53640,
      "training_step_time": 7.747349262237549
    },
    {
      "epoch": 5.1156044006347655e-05,
      "model_forward_time": 1.760399341583252,
      "step": 53641
    },
    {
      "epoch": 5.1156044006347655e-05,
      "step": 53641,
      "training_step_time": 7.702244997024536
    },
    {
      "epoch": 5.1156997680664065e-05,
      "model_forward_time": 1.7393383979797363,
      "step": 53642
    },
    {
      "epoch": 5.1156997680664065e-05,
      "step": 53642,
      "training_step_time": 7.7241363525390625
    },
    {
      "epoch": 5.115795135498047e-05,
      "model_forward_time": 2.334429979324341,
      "step": 53643
    },
    {
      "epoch": 5.115795135498047e-05,
      "step": 53643,
      "training_step_time": 7.74136757850647
    },
    {
      "epoch": 5.1158905029296873e-05,
      "model_forward_time": 1.7385799884796143,
      "step": 53644
    },
    {
      "epoch": 5.1158905029296873e-05,
      "step": 53644,
      "training_step_time": 7.750790596008301
    },
    {
      "epoch": 5.1159858703613284e-05,
      "model_forward_time": 1.743438959121704,
      "step": 53645
    },
    {
      "epoch": 5.1159858703613284e-05,
      "step": 53645,
      "training_step_time": 8.783396244049072
    },
    {
      "epoch": 5.116081237792969e-05,
      "model_forward_time": 1.7412714958190918,
      "step": 53646
    },
    {
      "epoch": 5.116081237792969e-05,
      "step": 53646,
      "training_step_time": 7.634807109832764
    },
    {
      "epoch": 5.116176605224609e-05,
      "model_forward_time": 1.7434463500976562,
      "step": 53647
    },
    {
      "epoch": 5.116176605224609e-05,
      "step": 53647,
      "training_step_time": 8.129358053207397
    },
    {
      "epoch": 5.11627197265625e-05,
      "model_forward_time": 1.744262456893921,
      "step": 53648
    },
    {
      "epoch": 5.11627197265625e-05,
      "step": 53648,
      "training_step_time": 7.612032175064087
    },
    {
      "epoch": 5.116367340087891e-05,
      "model_forward_time": 1.73934006690979,
      "step": 53649
    },
    {
      "epoch": 5.116367340087891e-05,
      "step": 53649,
      "training_step_time": 9.18605899810791
    },
    {
      "epoch": 5.116462707519531e-05,
      "grad_norm": 0.1373997926712036,
      "learning_rate": 4.810062029534886e-06,
      "loss": 0.0543,
      "step": 53650
    },
    {
      "epoch": 5.116462707519531e-05,
      "model_forward_time": 1.737318992614746,
      "step": 53650
    },
    {
      "dynamics_loss_avg": 0.0568889494985342,
      "epoch": 5.116462707519531e-05,
      "step": 53650
    },
    {
      "action_loss_avg": 0.0010523337812628597,
      "epoch": 5.116462707519531e-05,
      "step": 53650
    },
    {
      "epoch": 5.116462707519531e-05,
      "step": 53650,
      "training_step_time": 8.36888861656189
    },
    {
      "epoch": 5.116558074951172e-05,
      "model_forward_time": 1.749072790145874,
      "step": 53651
    },
    {
      "epoch": 5.116558074951172e-05,
      "step": 53651,
      "training_step_time": 7.747896909713745
    },
    {
      "epoch": 5.1166534423828126e-05,
      "model_forward_time": 2.309823751449585,
      "step": 53652
    },
    {
      "epoch": 5.1166534423828126e-05,
      "step": 53652,
      "training_step_time": 9.07372498512268
    },
    {
      "epoch": 5.116748809814453e-05,
      "model_forward_time": 1.7508940696716309,
      "step": 53653
    },
    {
      "epoch": 5.116748809814453e-05,
      "step": 53653,
      "training_step_time": 7.795741319656372
    },
    {
      "epoch": 5.116844177246094e-05,
      "model_forward_time": 1.7526166439056396,
      "step": 53654
    },
    {
      "epoch": 5.116844177246094e-05,
      "step": 53654,
      "training_step_time": 7.765214204788208
    },
    {
      "epoch": 5.1169395446777345e-05,
      "model_forward_time": 1.7559788227081299,
      "step": 53655
    },
    {
      "epoch": 5.1169395446777345e-05,
      "step": 53655,
      "training_step_time": 7.6790056228637695
    },
    {
      "epoch": 5.117034912109375e-05,
      "model_forward_time": 1.7524008750915527,
      "step": 53656
    },
    {
      "epoch": 5.117034912109375e-05,
      "step": 53656,
      "training_step_time": 8.638046264648438
    },
    {
      "epoch": 5.117130279541015e-05,
      "model_forward_time": 1.7525074481964111,
      "step": 53657
    },
    {
      "epoch": 5.117130279541015e-05,
      "step": 53657,
      "training_step_time": 7.642051935195923
    },
    {
      "epoch": 5.1172256469726564e-05,
      "model_forward_time": 1.7483491897583008,
      "step": 53658
    },
    {
      "epoch": 5.1172256469726564e-05,
      "step": 53658,
      "training_step_time": 7.681498050689697
    },
    {
      "epoch": 5.117321014404297e-05,
      "model_forward_time": 1.7404770851135254,
      "step": 53659
    },
    {
      "epoch": 5.117321014404297e-05,
      "step": 53659,
      "training_step_time": 7.784534692764282
    },
    {
      "epoch": 5.117416381835937e-05,
      "grad_norm": 0.12375946342945099,
      "learning_rate": 4.8084097635895505e-06,
      "loss": 0.0528,
      "step": 53660
    },
    {
      "epoch": 5.117416381835937e-05,
      "model_forward_time": 1.8103504180908203,
      "step": 53660
    },
    {
      "dynamics_loss_avg": 0.05231576636433601,
      "epoch": 5.117416381835937e-05,
      "step": 53660
    },
    {
      "action_loss_avg": 0.0024463282723445447,
      "epoch": 5.117416381835937e-05,
      "step": 53660
    },
    {
      "epoch": 5.117416381835937e-05,
      "step": 53660,
      "training_step_time": 8.87586522102356
    },
    {
      "epoch": 5.117511749267578e-05,
      "model_forward_time": 1.7428383827209473,
      "step": 53661
    },
    {
      "epoch": 5.117511749267578e-05,
      "step": 53661,
      "training_step_time": 9.273104667663574
    },
    {
      "epoch": 5.1176071166992187e-05,
      "model_forward_time": 2.2809367179870605,
      "step": 53662
    },
    {
      "epoch": 5.1176071166992187e-05,
      "step": 53662,
      "training_step_time": 8.434937477111816
    },
    {
      "epoch": 5.117702484130859e-05,
      "model_forward_time": 1.7509574890136719,
      "step": 53663
    },
    {
      "epoch": 5.117702484130859e-05,
      "step": 53663,
      "training_step_time": 8.295721292495728
    },
    {
      "epoch": 5.1177978515625e-05,
      "model_forward_time": 1.7514173984527588,
      "step": 53664
    },
    {
      "epoch": 5.1177978515625e-05,
      "step": 53664,
      "training_step_time": 7.716466665267944
    },
    {
      "epoch": 5.1178932189941405e-05,
      "model_forward_time": 1.7552037239074707,
      "step": 53665
    },
    {
      "epoch": 5.1178932189941405e-05,
      "step": 53665,
      "training_step_time": 7.639265298843384
    },
    {
      "epoch": 5.117988586425781e-05,
      "model_forward_time": 1.7483727931976318,
      "step": 53666
    },
    {
      "epoch": 5.117988586425781e-05,
      "step": 53666,
      "training_step_time": 8.394595861434937
    },
    {
      "epoch": 5.118083953857422e-05,
      "model_forward_time": 1.7495133876800537,
      "step": 53667
    },
    {
      "epoch": 5.118083953857422e-05,
      "step": 53667,
      "training_step_time": 7.630266189575195
    },
    {
      "epoch": 5.1181793212890624e-05,
      "model_forward_time": 1.7509827613830566,
      "step": 53668
    },
    {
      "epoch": 5.1181793212890624e-05,
      "step": 53668,
      "training_step_time": 7.7682836055755615
    },
    {
      "epoch": 5.118274688720703e-05,
      "model_forward_time": 1.805147647857666,
      "step": 53669
    },
    {
      "epoch": 5.118274688720703e-05,
      "step": 53669,
      "training_step_time": 7.756715774536133
    },
    {
      "epoch": 5.118370056152344e-05,
      "grad_norm": 0.10662756115198135,
      "learning_rate": 4.806757518596234e-06,
      "loss": 0.057,
      "step": 53670
    },
    {
      "epoch": 5.118370056152344e-05,
      "model_forward_time": 1.7406206130981445,
      "step": 53670
    },
    {
      "dynamics_loss_avg": 0.04732789434492588,
      "epoch": 5.118370056152344e-05,
      "step": 53670
    },
    {
      "action_loss_avg": 0.0011491876299260183,
      "epoch": 5.118370056152344e-05,
      "step": 53670
    },
    {
      "epoch": 5.118370056152344e-05,
      "step": 53670,
      "training_step_time": 8.479682445526123
    },
    {
      "epoch": 5.118465423583984e-05,
      "model_forward_time": 2.308681011199951,
      "step": 53671
    },
    {
      "epoch": 5.118465423583984e-05,
      "step": 53671,
      "training_step_time": 7.693916082382202
    },
    {
      "epoch": 5.118560791015625e-05,
      "model_forward_time": 1.7466819286346436,
      "step": 53672
    },
    {
      "epoch": 5.118560791015625e-05,
      "step": 53672,
      "training_step_time": 7.613820791244507
    },
    {
      "epoch": 5.118656158447266e-05,
      "model_forward_time": 1.7482109069824219,
      "step": 53673
    },
    {
      "epoch": 5.118656158447266e-05,
      "step": 53673,
      "training_step_time": 7.720587491989136
    },
    {
      "epoch": 5.118751525878906e-05,
      "model_forward_time": 1.7471468448638916,
      "step": 53674
    },
    {
      "epoch": 5.118751525878906e-05,
      "step": 53674,
      "training_step_time": 7.569587707519531
    },
    {
      "epoch": 5.1188468933105466e-05,
      "model_forward_time": 1.7437994480133057,
      "step": 53675
    },
    {
      "epoch": 5.1188468933105466e-05,
      "step": 53675,
      "training_step_time": 8.272046327590942
    },
    {
      "epoch": 5.118942260742188e-05,
      "model_forward_time": 1.7425589561462402,
      "step": 53676
    },
    {
      "epoch": 5.118942260742188e-05,
      "step": 53676,
      "training_step_time": 7.671569108963013
    },
    {
      "epoch": 5.119037628173828e-05,
      "model_forward_time": 1.7478790283203125,
      "step": 53677
    },
    {
      "epoch": 5.119037628173828e-05,
      "step": 53677,
      "training_step_time": 7.697386264801025
    },
    {
      "epoch": 5.1191329956054685e-05,
      "model_forward_time": 1.7728054523468018,
      "step": 53678
    },
    {
      "epoch": 5.1191329956054685e-05,
      "step": 53678,
      "training_step_time": 9.085692167282104
    },
    {
      "epoch": 5.1192283630371096e-05,
      "model_forward_time": 1.7352728843688965,
      "step": 53679
    },
    {
      "epoch": 5.1192283630371096e-05,
      "step": 53679,
      "training_step_time": 7.689383029937744
    },
    {
      "epoch": 5.11932373046875e-05,
      "grad_norm": 0.1442524492740631,
      "learning_rate": 4.805105294735625e-06,
      "loss": 0.0558,
      "step": 53680
    },
    {
      "epoch": 5.11932373046875e-05,
      "model_forward_time": 2.318087577819824,
      "step": 53680
    },
    {
      "dynamics_loss_avg": 0.05609774626791477,
      "epoch": 5.11932373046875e-05,
      "step": 53680
    },
    {
      "action_loss_avg": 0.001451779913622886,
      "epoch": 5.11932373046875e-05,
      "step": 53680
    },
    {
      "epoch": 5.11932373046875e-05,
      "step": 53680,
      "training_step_time": 7.607390880584717
    },
    {
      "epoch": 5.1194190979003904e-05,
      "model_forward_time": 1.746096134185791,
      "step": 53681
    },
    {
      "epoch": 5.1194190979003904e-05,
      "step": 53681,
      "training_step_time": 7.652923345565796
    },
    {
      "epoch": 5.1195144653320315e-05,
      "model_forward_time": 1.7467472553253174,
      "step": 53682
    },
    {
      "epoch": 5.1195144653320315e-05,
      "step": 53682,
      "training_step_time": 7.680803060531616
    },
    {
      "epoch": 5.119609832763672e-05,
      "model_forward_time": 1.7439384460449219,
      "step": 53683
    },
    {
      "epoch": 5.119609832763672e-05,
      "step": 53683,
      "training_step_time": 8.500738620758057
    },
    {
      "epoch": 5.119705200195312e-05,
      "model_forward_time": 1.7459545135498047,
      "step": 53684
    },
    {
      "epoch": 5.119705200195312e-05,
      "step": 53684,
      "training_step_time": 7.749202013015747
    },
    {
      "epoch": 5.1198005676269533e-05,
      "model_forward_time": 1.7483866214752197,
      "step": 53685
    },
    {
      "epoch": 5.1198005676269533e-05,
      "step": 53685,
      "training_step_time": 7.640662908554077
    },
    {
      "epoch": 5.119895935058594e-05,
      "model_forward_time": 1.747706651687622,
      "step": 53686
    },
    {
      "epoch": 5.119895935058594e-05,
      "step": 53686,
      "training_step_time": 7.629247426986694
    },
    {
      "epoch": 5.119991302490234e-05,
      "model_forward_time": 1.768340826034546,
      "step": 53687
    },
    {
      "epoch": 5.119991302490234e-05,
      "step": 53687,
      "training_step_time": 7.707513809204102
    },
    {
      "epoch": 5.120086669921875e-05,
      "model_forward_time": 1.7383029460906982,
      "step": 53688
    },
    {
      "epoch": 5.120086669921875e-05,
      "step": 53688,
      "training_step_time": 8.696643352508545
    },
    {
      "epoch": 5.1201820373535156e-05,
      "model_forward_time": 2.3141252994537354,
      "step": 53689
    },
    {
      "epoch": 5.1201820373535156e-05,
      "step": 53689,
      "training_step_time": 9.063455581665039
    },
    {
      "epoch": 5.120277404785156e-05,
      "grad_norm": 0.14722004532814026,
      "learning_rate": 4.803453092188406e-06,
      "loss": 0.0541,
      "step": 53690
    },
    {
      "epoch": 5.120277404785156e-05,
      "model_forward_time": 1.738269329071045,
      "step": 53690
    },
    {
      "dynamics_loss_avg": 0.04980481192469597,
      "epoch": 5.120277404785156e-05,
      "step": 53690
    },
    {
      "action_loss_avg": 0.0012968449445907027,
      "epoch": 5.120277404785156e-05,
      "step": 53690
    },
    {
      "epoch": 5.120277404785156e-05,
      "step": 53690,
      "training_step_time": 7.5472118854522705
    },
    {
      "epoch": 5.120372772216797e-05,
      "model_forward_time": 1.7421152591705322,
      "step": 53691
    },
    {
      "epoch": 5.120372772216797e-05,
      "step": 53691,
      "training_step_time": 7.654006242752075
    },
    {
      "epoch": 5.1204681396484375e-05,
      "model_forward_time": 1.7443301677703857,
      "step": 53692
    },
    {
      "epoch": 5.1204681396484375e-05,
      "step": 53692,
      "training_step_time": 7.583056449890137
    },
    {
      "epoch": 5.120563507080078e-05,
      "model_forward_time": 1.7387902736663818,
      "step": 53693
    },
    {
      "epoch": 5.120563507080078e-05,
      "step": 53693,
      "training_step_time": 7.712480545043945
    },
    {
      "epoch": 5.120658874511719e-05,
      "model_forward_time": 1.7422637939453125,
      "step": 53694
    },
    {
      "epoch": 5.120658874511719e-05,
      "step": 53694,
      "training_step_time": 7.56816291809082
    },
    {
      "epoch": 5.1207542419433594e-05,
      "model_forward_time": 1.360299825668335,
      "step": 53695
    },
    {
      "epoch": 5.1207542419433594e-05,
      "step": 53695,
      "training_step_time": 7.619493722915649
    },
    {
      "epoch": 5.120849609375e-05,
      "model_forward_time": 1.7518908977508545,
      "step": 53696
    },
    {
      "epoch": 5.120849609375e-05,
      "step": 53696,
      "training_step_time": 7.866714239120483
    },
    {
      "epoch": 5.120944976806641e-05,
      "model_forward_time": 1.7512288093566895,
      "step": 53697
    },
    {
      "epoch": 5.120944976806641e-05,
      "step": 53697,
      "training_step_time": 8.161203145980835
    },
    {
      "epoch": 5.121040344238281e-05,
      "model_forward_time": 2.273463726043701,
      "step": 53698
    },
    {
      "epoch": 5.121040344238281e-05,
      "step": 53698,
      "training_step_time": 8.834691047668457
    },
    {
      "epoch": 5.121135711669922e-05,
      "model_forward_time": 1.7485578060150146,
      "step": 53699
    },
    {
      "epoch": 5.121135711669922e-05,
      "step": 53699,
      "training_step_time": 9.859324932098389
    },
    {
      "epoch": 5.121231079101563e-05,
      "grad_norm": 0.11120906472206116,
      "learning_rate": 4.8018009111352635e-06,
      "loss": 0.0575,
      "step": 53700
    },
    {
      "epoch": 5.121231079101563e-05,
      "model_forward_time": 1.7367331981658936,
      "step": 53700
    },
    {
      "dynamics_loss_avg": 0.04168876688927412,
      "epoch": 5.121231079101563e-05,
      "step": 53700
    },
    {
      "action_loss_avg": 0.0017032903269864619,
      "epoch": 5.121231079101563e-05,
      "step": 53700
    },
    {
      "epoch": 5.121231079101563e-05,
      "step": 53700,
      "training_step_time": 6.847772836685181
    },
    {
      "epoch": 5.121326446533203e-05,
      "model_forward_time": 1.7415239810943604,
      "step": 53701
    },
    {
      "epoch": 5.121326446533203e-05,
      "step": 53701,
      "training_step_time": 6.8232262134552
    },
    {
      "epoch": 5.1214218139648436e-05,
      "model_forward_time": 1.742920160293579,
      "step": 53702
    },
    {
      "epoch": 5.1214218139648436e-05,
      "step": 53702,
      "training_step_time": 6.904320001602173
    },
    {
      "epoch": 5.1215171813964847e-05,
      "model_forward_time": 1.744152307510376,
      "step": 53703
    },
    {
      "epoch": 5.1215171813964847e-05,
      "step": 53703,
      "training_step_time": 6.892516136169434
    },
    {
      "epoch": 5.121612548828125e-05,
      "model_forward_time": 1.7447850704193115,
      "step": 53704
    },
    {
      "epoch": 5.121612548828125e-05,
      "step": 53704,
      "training_step_time": 6.866883993148804
    },
    {
      "epoch": 5.1217079162597655e-05,
      "model_forward_time": 1.748779058456421,
      "step": 53705
    },
    {
      "epoch": 5.1217079162597655e-05,
      "step": 53705,
      "training_step_time": 6.86035418510437
    },
    {
      "epoch": 5.1218032836914065e-05,
      "model_forward_time": 1.3708868026733398,
      "step": 53706
    },
    {
      "epoch": 5.1218032836914065e-05,
      "step": 53706,
      "training_step_time": 6.906490325927734
    },
    {
      "epoch": 5.121898651123047e-05,
      "model_forward_time": 1.7459211349487305,
      "step": 53707
    },
    {
      "epoch": 5.121898651123047e-05,
      "step": 53707,
      "training_step_time": 6.947956085205078
    },
    {
      "epoch": 5.1219940185546873e-05,
      "model_forward_time": 2.2858681678771973,
      "step": 53708
    },
    {
      "epoch": 5.1219940185546873e-05,
      "step": 53708,
      "training_step_time": 7.655138254165649
    },
    {
      "epoch": 5.1220893859863284e-05,
      "model_forward_time": 1.7458314895629883,
      "step": 53709
    },
    {
      "epoch": 5.1220893859863284e-05,
      "step": 53709,
      "training_step_time": 7.906433582305908
    },
    {
      "epoch": 5.122184753417969e-05,
      "grad_norm": 0.07120849192142487,
      "learning_rate": 4.800148751756874e-06,
      "loss": 0.0559,
      "step": 53710
    },
    {
      "epoch": 5.122184753417969e-05,
      "model_forward_time": 1.7132673263549805,
      "step": 53710
    },
    {
      "dynamics_loss_avg": 0.0482821486890316,
      "epoch": 5.122184753417969e-05,
      "step": 53710
    },
    {
      "action_loss_avg": 0.0013094370951876044,
      "epoch": 5.122184753417969e-05,
      "step": 53710
    },
    {
      "epoch": 5.122184753417969e-05,
      "step": 53710,
      "training_step_time": 6.898442983627319
    },
    {
      "epoch": 5.122280120849609e-05,
      "model_forward_time": 1.7418887615203857,
      "step": 53711
    },
    {
      "epoch": 5.122280120849609e-05,
      "step": 53711,
      "training_step_time": 6.853952407836914
    },
    {
      "epoch": 5.12237548828125e-05,
      "model_forward_time": 1.741441249847412,
      "step": 53712
    },
    {
      "epoch": 5.12237548828125e-05,
      "step": 53712,
      "training_step_time": 7.029258489608765
    },
    {
      "epoch": 5.122470855712891e-05,
      "model_forward_time": 1.740309715270996,
      "step": 53713
    },
    {
      "epoch": 5.122470855712891e-05,
      "step": 53713,
      "training_step_time": 6.920699596405029
    },
    {
      "epoch": 5.122566223144531e-05,
      "model_forward_time": 1.7363653182983398,
      "step": 53714
    },
    {
      "epoch": 5.122566223144531e-05,
      "step": 53714,
      "training_step_time": 6.921797752380371
    },
    {
      "epoch": 5.122661590576172e-05,
      "model_forward_time": 1.7440102100372314,
      "step": 53715
    },
    {
      "epoch": 5.122661590576172e-05,
      "step": 53715,
      "training_step_time": 6.910655975341797
    },
    {
      "epoch": 5.1227569580078126e-05,
      "model_forward_time": 1.75351881980896,
      "step": 53716
    },
    {
      "epoch": 5.1227569580078126e-05,
      "step": 53716,
      "training_step_time": 6.887418270111084
    },
    {
      "epoch": 5.122852325439453e-05,
      "model_forward_time": 2.2809503078460693,
      "step": 53717
    },
    {
      "epoch": 5.122852325439453e-05,
      "step": 53717,
      "training_step_time": 7.685586214065552
    },
    {
      "epoch": 5.122947692871094e-05,
      "model_forward_time": 1.732468843460083,
      "step": 53718
    },
    {
      "epoch": 5.122947692871094e-05,
      "step": 53718,
      "training_step_time": 8.932692527770996
    },
    {
      "epoch": 5.1230430603027345e-05,
      "model_forward_time": 1.7441420555114746,
      "step": 53719
    },
    {
      "epoch": 5.1230430603027345e-05,
      "step": 53719,
      "training_step_time": 7.282326936721802
    },
    {
      "epoch": 5.123138427734375e-05,
      "grad_norm": 0.11334454268217087,
      "learning_rate": 4.798496614233919e-06,
      "loss": 0.0563,
      "step": 53720
    },
    {
      "epoch": 5.123138427734375e-05,
      "model_forward_time": 1.7153394222259521,
      "step": 53720
    },
    {
      "dynamics_loss_avg": 0.043874206766486165,
      "epoch": 5.123138427734375e-05,
      "step": 53720
    },
    {
      "action_loss_avg": 0.0009565419459249824,
      "epoch": 5.123138427734375e-05,
      "step": 53720
    },
    {
      "epoch": 5.123138427734375e-05,
      "step": 53720,
      "training_step_time": 6.893801927566528
    },
    {
      "epoch": 5.123233795166015e-05,
      "model_forward_time": 1.7365682125091553,
      "step": 53721
    },
    {
      "epoch": 5.123233795166015e-05,
      "step": 53721,
      "training_step_time": 6.87377667427063
    },
    {
      "epoch": 5.1233291625976564e-05,
      "model_forward_time": 1.7331583499908447,
      "step": 53722
    },
    {
      "epoch": 5.1233291625976564e-05,
      "step": 53722,
      "training_step_time": 6.854450225830078
    },
    {
      "epoch": 5.123424530029297e-05,
      "model_forward_time": 1.7329645156860352,
      "step": 53723
    },
    {
      "epoch": 5.123424530029297e-05,
      "step": 53723,
      "training_step_time": 6.913888692855835
    },
    {
      "epoch": 5.123519897460937e-05,
      "model_forward_time": 1.7449951171875,
      "step": 53724
    },
    {
      "epoch": 5.123519897460937e-05,
      "step": 53724,
      "training_step_time": 6.878559112548828
    },
    {
      "epoch": 5.123615264892578e-05,
      "model_forward_time": 1.7282624244689941,
      "step": 53725
    },
    {
      "epoch": 5.123615264892578e-05,
      "step": 53725,
      "training_step_time": 6.877691030502319
    },
    {
      "epoch": 5.123710632324219e-05,
      "model_forward_time": 2.308643341064453,
      "step": 53726
    },
    {
      "epoch": 5.123710632324219e-05,
      "step": 53726,
      "training_step_time": 7.797738313674927
    },
    {
      "epoch": 5.123805999755859e-05,
      "model_forward_time": 1.733518362045288,
      "step": 53727
    },
    {
      "epoch": 5.123805999755859e-05,
      "step": 53727,
      "training_step_time": 11.644635677337646
    },
    {
      "epoch": 5.1239013671875e-05,
      "model_forward_time": 1.7390193939208984,
      "step": 53728
    },
    {
      "epoch": 5.1239013671875e-05,
      "step": 53728,
      "training_step_time": 7.657355546951294
    },
    {
      "epoch": 5.1239967346191405e-05,
      "model_forward_time": 1.7415380477905273,
      "step": 53729
    },
    {
      "epoch": 5.1239967346191405e-05,
      "step": 53729,
      "training_step_time": 6.880089521408081
    },
    {
      "epoch": 5.124092102050781e-05,
      "grad_norm": 0.1466057002544403,
      "learning_rate": 4.796844498747069e-06,
      "loss": 0.0588,
      "step": 53730
    },
    {
      "epoch": 5.124092102050781e-05,
      "model_forward_time": 1.7059574127197266,
      "step": 53730
    },
    {
      "dynamics_loss_avg": 0.045746417436748746,
      "epoch": 5.124092102050781e-05,
      "step": 53730
    },
    {
      "action_loss_avg": 0.0017692289693513885,
      "epoch": 5.124092102050781e-05,
      "step": 53730
    },
    {
      "epoch": 5.124092102050781e-05,
      "step": 53730,
      "training_step_time": 6.876110076904297
    },
    {
      "epoch": 5.124187469482422e-05,
      "model_forward_time": 1.7375431060791016,
      "step": 53731
    },
    {
      "epoch": 5.124187469482422e-05,
      "step": 53731,
      "training_step_time": 6.84858512878418
    },
    {
      "epoch": 5.1242828369140624e-05,
      "model_forward_time": 1.7347080707550049,
      "step": 53732
    },
    {
      "epoch": 5.1242828369140624e-05,
      "step": 53732,
      "training_step_time": 6.742054224014282
    },
    {
      "epoch": 5.124378204345703e-05,
      "model_forward_time": 1.3561513423919678,
      "step": 53733
    },
    {
      "epoch": 5.124378204345703e-05,
      "step": 53733,
      "training_step_time": 6.848968505859375
    },
    {
      "epoch": 5.124473571777344e-05,
      "model_forward_time": 1.7342569828033447,
      "step": 53734
    },
    {
      "epoch": 5.124473571777344e-05,
      "step": 53734,
      "training_step_time": 6.884478807449341
    },
    {
      "epoch": 5.124568939208984e-05,
      "model_forward_time": 2.288353681564331,
      "step": 53735
    },
    {
      "epoch": 5.124568939208984e-05,
      "step": 53735,
      "training_step_time": 7.720739126205444
    },
    {
      "epoch": 5.124664306640625e-05,
      "model_forward_time": 1.7279908657073975,
      "step": 53736
    },
    {
      "epoch": 5.124664306640625e-05,
      "step": 53736,
      "training_step_time": 10.468009233474731
    },
    {
      "epoch": 5.124759674072266e-05,
      "model_forward_time": 1.3572015762329102,
      "step": 53737
    },
    {
      "epoch": 5.124759674072266e-05,
      "step": 53737,
      "training_step_time": 7.869184970855713
    },
    {
      "epoch": 5.124855041503906e-05,
      "model_forward_time": 1.7330443859100342,
      "step": 53738
    },
    {
      "epoch": 5.124855041503906e-05,
      "step": 53738,
      "training_step_time": 6.900445461273193
    },
    {
      "epoch": 5.1249504089355466e-05,
      "model_forward_time": 1.735288381576538,
      "step": 53739
    },
    {
      "epoch": 5.1249504089355466e-05,
      "step": 53739,
      "training_step_time": 6.860333204269409
    },
    {
      "epoch": 5.125045776367188e-05,
      "grad_norm": 0.12446279078722,
      "learning_rate": 4.795192405477e-06,
      "loss": 0.057,
      "step": 53740
    },
    {
      "epoch": 5.125045776367188e-05,
      "model_forward_time": 1.7276651859283447,
      "step": 53740
    },
    {
      "dynamics_loss_avg": 0.053920775558799504,
      "epoch": 5.125045776367188e-05,
      "step": 53740
    },
    {
      "action_loss_avg": 0.0012412554991897196,
      "epoch": 5.125045776367188e-05,
      "step": 53740
    },
    {
      "epoch": 5.125045776367188e-05,
      "step": 53740,
      "training_step_time": 6.969516277313232
    },
    {
      "epoch": 5.125141143798828e-05,
      "model_forward_time": 1.7243666648864746,
      "step": 53741
    },
    {
      "epoch": 5.125141143798828e-05,
      "step": 53741,
      "training_step_time": 6.913585424423218
    },
    {
      "epoch": 5.1252365112304685e-05,
      "model_forward_time": 1.3558547496795654,
      "step": 53742
    },
    {
      "epoch": 5.1252365112304685e-05,
      "step": 53742,
      "training_step_time": 6.970213890075684
    },
    {
      "epoch": 5.1253318786621096e-05,
      "model_forward_time": 1.7360632419586182,
      "step": 53743
    },
    {
      "epoch": 5.1253318786621096e-05,
      "step": 53743,
      "training_step_time": 6.9175896644592285
    },
    {
      "epoch": 5.12542724609375e-05,
      "model_forward_time": 1.7302372455596924,
      "step": 53744
    },
    {
      "epoch": 5.12542724609375e-05,
      "step": 53744,
      "training_step_time": 7.768687009811401
    },
    {
      "epoch": 5.1255226135253904e-05,
      "model_forward_time": 2.2939915657043457,
      "step": 53745
    },
    {
      "epoch": 5.1255226135253904e-05,
      "step": 53745,
      "training_step_time": 10.542402744293213
    },
    {
      "epoch": 5.1256179809570315e-05,
      "model_forward_time": 1.739847183227539,
      "step": 53746
    },
    {
      "epoch": 5.1256179809570315e-05,
      "step": 53746,
      "training_step_time": 7.902131795883179
    },
    {
      "epoch": 5.125713348388672e-05,
      "model_forward_time": 1.722330093383789,
      "step": 53747
    },
    {
      "epoch": 5.125713348388672e-05,
      "step": 53747,
      "training_step_time": 7.718670845031738
    },
    {
      "epoch": 5.125808715820312e-05,
      "model_forward_time": 1.7422935962677002,
      "step": 53748
    },
    {
      "epoch": 5.125808715820312e-05,
      "step": 53748,
      "training_step_time": 6.8899617195129395
    },
    {
      "epoch": 5.1259040832519533e-05,
      "model_forward_time": 1.7461764812469482,
      "step": 53749
    },
    {
      "epoch": 5.1259040832519533e-05,
      "step": 53749,
      "training_step_time": 6.884382009506226
    },
    {
      "epoch": 5.125999450683594e-05,
      "grad_norm": 0.12603259086608887,
      "learning_rate": 4.79354033460438e-06,
      "loss": 0.0571,
      "step": 53750
    },
    {
      "epoch": 5.125999450683594e-05,
      "model_forward_time": 1.7311735153198242,
      "step": 53750
    },
    {
      "dynamics_loss_avg": 0.04813250824809075,
      "epoch": 5.125999450683594e-05,
      "step": 53750
    },
    {
      "action_loss_avg": 0.00116944465553388,
      "epoch": 5.125999450683594e-05,
      "step": 53750
    },
    {
      "epoch": 5.125999450683594e-05,
      "step": 53750,
      "training_step_time": 6.919149398803711
    },
    {
      "epoch": 5.126094818115234e-05,
      "model_forward_time": 1.7486827373504639,
      "step": 53751
    },
    {
      "epoch": 5.126094818115234e-05,
      "step": 53751,
      "training_step_time": 6.876773357391357
    },
    {
      "epoch": 5.126190185546875e-05,
      "model_forward_time": 1.723459243774414,
      "step": 53752
    },
    {
      "epoch": 5.126190185546875e-05,
      "step": 53752,
      "training_step_time": 6.927185297012329
    },
    {
      "epoch": 5.1262855529785156e-05,
      "model_forward_time": 1.7352898120880127,
      "step": 53753
    },
    {
      "epoch": 5.1262855529785156e-05,
      "step": 53753,
      "training_step_time": 6.950603723526001
    },
    {
      "epoch": 5.126380920410156e-05,
      "model_forward_time": 2.3071205615997314,
      "step": 53754
    },
    {
      "epoch": 5.126380920410156e-05,
      "step": 53754,
      "training_step_time": 7.612452745437622
    },
    {
      "epoch": 5.126476287841797e-05,
      "model_forward_time": 1.7372972965240479,
      "step": 53755
    },
    {
      "epoch": 5.126476287841797e-05,
      "step": 53755,
      "training_step_time": 7.907331705093384
    },
    {
      "epoch": 5.1265716552734375e-05,
      "model_forward_time": 1.729733943939209,
      "step": 53756
    },
    {
      "epoch": 5.1265716552734375e-05,
      "step": 53756,
      "training_step_time": 7.87137770652771
    },
    {
      "epoch": 5.126667022705078e-05,
      "model_forward_time": 1.7352464199066162,
      "step": 53757
    },
    {
      "epoch": 5.126667022705078e-05,
      "step": 53757,
      "training_step_time": 6.897499322891235
    },
    {
      "epoch": 5.126762390136719e-05,
      "model_forward_time": 1.7346127033233643,
      "step": 53758
    },
    {
      "epoch": 5.126762390136719e-05,
      "step": 53758,
      "training_step_time": 6.840227365493774
    },
    {
      "epoch": 5.1268577575683594e-05,
      "model_forward_time": 1.7329010963439941,
      "step": 53759
    },
    {
      "epoch": 5.1268577575683594e-05,
      "step": 53759,
      "training_step_time": 6.912537574768066
    },
    {
      "epoch": 5.126953125e-05,
      "grad_norm": 0.10587777197360992,
      "learning_rate": 4.79188828630988e-06,
      "loss": 0.0582,
      "step": 53760
    },
    {
      "epoch": 5.126953125e-05,
      "model_forward_time": 1.72257399559021,
      "step": 53760
    },
    {
      "dynamics_loss_avg": 0.05348287196829915,
      "epoch": 5.126953125e-05,
      "step": 53760
    },
    {
      "action_loss_avg": 0.0012129344133427367,
      "epoch": 5.126953125e-05,
      "step": 53760
    },
    {
      "epoch": 5.126953125e-05,
      "step": 53760,
      "training_step_time": 6.940541982650757
    },
    {
      "epoch": 5.127048492431641e-05,
      "model_forward_time": 1.730334758758545,
      "step": 53761
    },
    {
      "epoch": 5.127048492431641e-05,
      "step": 53761,
      "training_step_time": 6.880764961242676
    },
    {
      "epoch": 5.127143859863281e-05,
      "model_forward_time": 1.734266757965088,
      "step": 53762
    },
    {
      "epoch": 5.127143859863281e-05,
      "step": 53762,
      "training_step_time": 6.938383340835571
    },
    {
      "epoch": 5.127239227294922e-05,
      "model_forward_time": 2.304314136505127,
      "step": 53763
    },
    {
      "epoch": 5.127239227294922e-05,
      "step": 53763,
      "training_step_time": 7.653510332107544
    },
    {
      "epoch": 5.127334594726563e-05,
      "model_forward_time": 1.7452521324157715,
      "step": 53764
    },
    {
      "epoch": 5.127334594726563e-05,
      "step": 53764,
      "training_step_time": 9.899944305419922
    },
    {
      "epoch": 5.127429962158203e-05,
      "model_forward_time": 1.73561429977417,
      "step": 53765
    },
    {
      "epoch": 5.127429962158203e-05,
      "step": 53765,
      "training_step_time": 7.925331115722656
    },
    {
      "epoch": 5.1275253295898436e-05,
      "model_forward_time": 1.7397563457489014,
      "step": 53766
    },
    {
      "epoch": 5.1275253295898436e-05,
      "step": 53766,
      "training_step_time": 6.9181506633758545
    },
    {
      "epoch": 5.127620697021485e-05,
      "model_forward_time": 1.7301855087280273,
      "step": 53767
    },
    {
      "epoch": 5.127620697021485e-05,
      "step": 53767,
      "training_step_time": 6.9254419803619385
    },
    {
      "epoch": 5.127716064453125e-05,
      "model_forward_time": 1.7369778156280518,
      "step": 53768
    },
    {
      "epoch": 5.127716064453125e-05,
      "step": 53768,
      "training_step_time": 6.908839464187622
    },
    {
      "epoch": 5.1278114318847655e-05,
      "model_forward_time": 1.7342441082000732,
      "step": 53769
    },
    {
      "epoch": 5.1278114318847655e-05,
      "step": 53769,
      "training_step_time": 6.911494493484497
    },
    {
      "epoch": 5.1279067993164065e-05,
      "grad_norm": 0.14530400931835175,
      "learning_rate": 4.7902362607741645e-06,
      "loss": 0.0571,
      "step": 53770
    },
    {
      "epoch": 5.1279067993164065e-05,
      "model_forward_time": 1.3441028594970703,
      "step": 53770
    },
    {
      "dynamics_loss_avg": 0.058782851323485374,
      "epoch": 5.1279067993164065e-05,
      "step": 53770
    },
    {
      "action_loss_avg": 0.0032921981226536447,
      "epoch": 5.1279067993164065e-05,
      "step": 53770
    },
    {
      "epoch": 5.1279067993164065e-05,
      "step": 53770,
      "training_step_time": 6.93013334274292
    },
    {
      "epoch": 5.128002166748047e-05,
      "model_forward_time": 1.7394564151763916,
      "step": 53771
    },
    {
      "epoch": 5.128002166748047e-05,
      "step": 53771,
      "training_step_time": 6.873225927352905
    },
    {
      "epoch": 5.1280975341796874e-05,
      "model_forward_time": 1.6980957984924316,
      "step": 53772
    },
    {
      "epoch": 5.1280975341796874e-05,
      "step": 53772,
      "training_step_time": 7.7935895919799805
    },
    {
      "epoch": 5.1281929016113284e-05,
      "model_forward_time": 2.2857959270477295,
      "step": 53773
    },
    {
      "epoch": 5.1281929016113284e-05,
      "step": 53773,
      "training_step_time": 9.430079936981201
    },
    {
      "epoch": 5.128288269042969e-05,
      "model_forward_time": 1.7318549156188965,
      "step": 53774
    },
    {
      "epoch": 5.128288269042969e-05,
      "step": 53774,
      "training_step_time": 7.913809299468994
    },
    {
      "epoch": 5.128383636474609e-05,
      "model_forward_time": 1.73347806930542,
      "step": 53775
    },
    {
      "epoch": 5.128383636474609e-05,
      "step": 53775,
      "training_step_time": 7.6449878215789795
    },
    {
      "epoch": 5.12847900390625e-05,
      "model_forward_time": 1.7301113605499268,
      "step": 53776
    },
    {
      "epoch": 5.12847900390625e-05,
      "step": 53776,
      "training_step_time": 6.799469470977783
    },
    {
      "epoch": 5.128574371337891e-05,
      "model_forward_time": 1.7433454990386963,
      "step": 53777
    },
    {
      "epoch": 5.128574371337891e-05,
      "step": 53777,
      "training_step_time": 6.823504686355591
    },
    {
      "epoch": 5.128669738769531e-05,
      "model_forward_time": 1.7363810539245605,
      "step": 53778
    },
    {
      "epoch": 5.128669738769531e-05,
      "step": 53778,
      "training_step_time": 6.780820846557617
    },
    {
      "epoch": 5.128765106201172e-05,
      "model_forward_time": 1.7320952415466309,
      "step": 53779
    },
    {
      "epoch": 5.128765106201172e-05,
      "step": 53779,
      "training_step_time": 6.7874321937561035
    },
    {
      "epoch": 5.1288604736328126e-05,
      "grad_norm": 0.11997327208518982,
      "learning_rate": 4.788584258177896e-06,
      "loss": 0.0565,
      "step": 53780
    },
    {
      "epoch": 5.1288604736328126e-05,
      "model_forward_time": 1.7360758781433105,
      "step": 53780
    },
    {
      "dynamics_loss_avg": 0.03984717391431332,
      "epoch": 5.1288604736328126e-05,
      "step": 53780
    },
    {
      "action_loss_avg": 0.0010086561640491709,
      "epoch": 5.1288604736328126e-05,
      "step": 53780
    },
    {
      "epoch": 5.1288604736328126e-05,
      "step": 53780,
      "training_step_time": 6.82167911529541
    },
    {
      "epoch": 5.128955841064453e-05,
      "model_forward_time": 1.7346158027648926,
      "step": 53781
    },
    {
      "epoch": 5.128955841064453e-05,
      "step": 53781,
      "training_step_time": 7.296679496765137
    },
    {
      "epoch": 5.1290512084960934e-05,
      "model_forward_time": 2.293895959854126,
      "step": 53782
    },
    {
      "epoch": 5.1290512084960934e-05,
      "step": 53782,
      "training_step_time": 11.21170449256897
    },
    {
      "epoch": 5.1291465759277345e-05,
      "model_forward_time": 1.7343664169311523,
      "step": 53783
    },
    {
      "epoch": 5.1291465759277345e-05,
      "step": 53783,
      "training_step_time": 8.869145154953003
    },
    {
      "epoch": 5.129241943359375e-05,
      "model_forward_time": 1.7336623668670654,
      "step": 53784
    },
    {
      "epoch": 5.129241943359375e-05,
      "step": 53784,
      "training_step_time": 7.641355752944946
    },
    {
      "epoch": 5.129337310791015e-05,
      "model_forward_time": 1.7327485084533691,
      "step": 53785
    },
    {
      "epoch": 5.129337310791015e-05,
      "step": 53785,
      "training_step_time": 6.7908148765563965
    },
    {
      "epoch": 5.1294326782226564e-05,
      "model_forward_time": 1.729802131652832,
      "step": 53786
    },
    {
      "epoch": 5.1294326782226564e-05,
      "step": 53786,
      "training_step_time": 6.803416013717651
    },
    {
      "epoch": 5.129528045654297e-05,
      "model_forward_time": 1.7547800540924072,
      "step": 53787
    },
    {
      "epoch": 5.129528045654297e-05,
      "step": 53787,
      "training_step_time": 6.813459157943726
    },
    {
      "epoch": 5.129623413085937e-05,
      "model_forward_time": 1.727097511291504,
      "step": 53788
    },
    {
      "epoch": 5.129623413085937e-05,
      "step": 53788,
      "training_step_time": 6.805942535400391
    },
    {
      "epoch": 5.129718780517578e-05,
      "model_forward_time": 1.731621265411377,
      "step": 53789
    },
    {
      "epoch": 5.129718780517578e-05,
      "step": 53789,
      "training_step_time": 6.82620096206665
    },
    {
      "epoch": 5.129814147949219e-05,
      "grad_norm": 0.11492504924535751,
      "learning_rate": 4.786932278701736e-06,
      "loss": 0.0603,
      "step": 53790
    },
    {
      "epoch": 5.129814147949219e-05,
      "model_forward_time": 1.7279109954833984,
      "step": 53790
    },
    {
      "dynamics_loss_avg": 0.04227786734700203,
      "epoch": 5.129814147949219e-05,
      "step": 53790
    },
    {
      "action_loss_avg": 0.001262194142327644,
      "epoch": 5.129814147949219e-05,
      "step": 53790
    },
    {
      "epoch": 5.129814147949219e-05,
      "step": 53790,
      "training_step_time": 6.8342413902282715
    },
    {
      "epoch": 5.129909515380859e-05,
      "model_forward_time": 2.3250632286071777,
      "step": 53791
    },
    {
      "epoch": 5.129909515380859e-05,
      "step": 53791,
      "training_step_time": 9.199368476867676
    },
    {
      "epoch": 5.1300048828125e-05,
      "model_forward_time": 1.3445024490356445,
      "step": 53792
    },
    {
      "epoch": 5.1300048828125e-05,
      "step": 53792,
      "training_step_time": 8.508044958114624
    },
    {
      "epoch": 5.1301002502441406e-05,
      "model_forward_time": 1.7370107173919678,
      "step": 53793
    },
    {
      "epoch": 5.1301002502441406e-05,
      "step": 53793,
      "training_step_time": 7.658783912658691
    },
    {
      "epoch": 5.130195617675781e-05,
      "model_forward_time": 1.7422394752502441,
      "step": 53794
    },
    {
      "epoch": 5.130195617675781e-05,
      "step": 53794,
      "training_step_time": 6.859262228012085
    },
    {
      "epoch": 5.130290985107422e-05,
      "model_forward_time": 1.7105507850646973,
      "step": 53795
    },
    {
      "epoch": 5.130290985107422e-05,
      "step": 53795,
      "training_step_time": 6.816219329833984
    },
    {
      "epoch": 5.1303863525390624e-05,
      "model_forward_time": 1.7333858013153076,
      "step": 53796
    },
    {
      "epoch": 5.1303863525390624e-05,
      "step": 53796,
      "training_step_time": 6.849429368972778
    },
    {
      "epoch": 5.130481719970703e-05,
      "model_forward_time": 1.7394068241119385,
      "step": 53797
    },
    {
      "epoch": 5.130481719970703e-05,
      "step": 53797,
      "training_step_time": 6.834048271179199
    },
    {
      "epoch": 5.130577087402344e-05,
      "model_forward_time": 1.7386889457702637,
      "step": 53798
    },
    {
      "epoch": 5.130577087402344e-05,
      "step": 53798,
      "training_step_time": 6.836819171905518
    },
    {
      "epoch": 5.130672454833984e-05,
      "model_forward_time": 1.7377841472625732,
      "step": 53799
    },
    {
      "epoch": 5.130672454833984e-05,
      "step": 53799,
      "training_step_time": 6.881208181381226
    },
    {
      "epoch": 5.130767822265625e-05,
      "grad_norm": 0.11854642629623413,
      "learning_rate": 4.7852803225263405e-06,
      "loss": 0.0569,
      "step": 53800
    },
    {
      "epoch": 5.130767822265625e-05,
      "model_forward_time": 2.3305280208587646,
      "step": 53800
    },
    {
      "dynamics_loss_avg": 0.055702554434537886,
      "epoch": 5.130767822265625e-05,
      "step": 53800
    },
    {
      "action_loss_avg": 0.0034201860893517733,
      "epoch": 5.130767822265625e-05,
      "step": 53800
    },
    {
      "epoch": 5.130767822265625e-05,
      "step": 53800,
      "training_step_time": 8.703187704086304
    },
    {
      "epoch": 5.130863189697266e-05,
      "model_forward_time": 1.7332441806793213,
      "step": 53801
    },
    {
      "epoch": 5.130863189697266e-05,
      "step": 53801,
      "training_step_time": 8.854278087615967
    },
    {
      "epoch": 5.130958557128906e-05,
      "model_forward_time": 1.7420172691345215,
      "step": 53802
    },
    {
      "epoch": 5.130958557128906e-05,
      "step": 53802,
      "training_step_time": 7.879795789718628
    },
    {
      "epoch": 5.1310539245605466e-05,
      "model_forward_time": 1.7303111553192139,
      "step": 53803
    },
    {
      "epoch": 5.1310539245605466e-05,
      "step": 53803,
      "training_step_time": 7.795085668563843
    },
    {
      "epoch": 5.131149291992188e-05,
      "model_forward_time": 1.7314245700836182,
      "step": 53804
    },
    {
      "epoch": 5.131149291992188e-05,
      "step": 53804,
      "training_step_time": 6.879061937332153
    },
    {
      "epoch": 5.131244659423828e-05,
      "model_forward_time": 1.7462644577026367,
      "step": 53805
    },
    {
      "epoch": 5.131244659423828e-05,
      "step": 53805,
      "training_step_time": 6.832616329193115
    },
    {
      "epoch": 5.1313400268554685e-05,
      "model_forward_time": 1.7299516201019287,
      "step": 53806
    },
    {
      "epoch": 5.1313400268554685e-05,
      "step": 53806,
      "training_step_time": 6.807535648345947
    },
    {
      "epoch": 5.1314353942871096e-05,
      "model_forward_time": 1.7261395454406738,
      "step": 53807
    },
    {
      "epoch": 5.1314353942871096e-05,
      "step": 53807,
      "training_step_time": 6.816911935806274
    },
    {
      "epoch": 5.13153076171875e-05,
      "model_forward_time": 1.732511043548584,
      "step": 53808
    },
    {
      "epoch": 5.13153076171875e-05,
      "step": 53808,
      "training_step_time": 6.877739667892456
    },
    {
      "epoch": 5.1316261291503904e-05,
      "model_forward_time": 1.7369961738586426,
      "step": 53809
    },
    {
      "epoch": 5.1316261291503904e-05,
      "step": 53809,
      "training_step_time": 10.218358516693115
    },
    {
      "epoch": 5.1317214965820315e-05,
      "grad_norm": 0.10087776929140091,
      "learning_rate": 4.783628389832366e-06,
      "loss": 0.0599,
      "step": 53810
    },
    {
      "epoch": 5.1317214965820315e-05,
      "model_forward_time": 1.7218003273010254,
      "step": 53810
    },
    {
      "dynamics_loss_avg": 0.04397244043648243,
      "epoch": 5.1317214965820315e-05,
      "step": 53810
    },
    {
      "action_loss_avg": 0.0016909785190364346,
      "epoch": 5.1317214965820315e-05,
      "step": 53810
    },
    {
      "epoch": 5.1317214965820315e-05,
      "step": 53810,
      "training_step_time": 8.293902158737183
    },
    {
      "epoch": 5.131816864013672e-05,
      "model_forward_time": 1.7355554103851318,
      "step": 53811
    },
    {
      "epoch": 5.131816864013672e-05,
      "step": 53811,
      "training_step_time": 7.802396535873413
    },
    {
      "epoch": 5.131912231445312e-05,
      "model_forward_time": 1.7132642269134521,
      "step": 53812
    },
    {
      "epoch": 5.131912231445312e-05,
      "step": 53812,
      "training_step_time": 7.454266786575317
    },
    {
      "epoch": 5.1320075988769534e-05,
      "model_forward_time": 1.734227180480957,
      "step": 53813
    },
    {
      "epoch": 5.1320075988769534e-05,
      "step": 53813,
      "training_step_time": 6.85412859916687
    },
    {
      "epoch": 5.132102966308594e-05,
      "model_forward_time": 1.727555751800537,
      "step": 53814
    },
    {
      "epoch": 5.132102966308594e-05,
      "step": 53814,
      "training_step_time": 6.819436550140381
    },
    {
      "epoch": 5.132198333740234e-05,
      "model_forward_time": 1.7291159629821777,
      "step": 53815
    },
    {
      "epoch": 5.132198333740234e-05,
      "step": 53815,
      "training_step_time": 6.85718035697937
    },
    {
      "epoch": 5.132293701171875e-05,
      "model_forward_time": 1.7284832000732422,
      "step": 53816
    },
    {
      "epoch": 5.132293701171875e-05,
      "step": 53816,
      "training_step_time": 6.816066741943359
    },
    {
      "epoch": 5.1323890686035156e-05,
      "model_forward_time": 1.7331304550170898,
      "step": 53817
    },
    {
      "epoch": 5.1323890686035156e-05,
      "step": 53817,
      "training_step_time": 6.851512432098389
    },
    {
      "epoch": 5.132484436035156e-05,
      "model_forward_time": 1.7491676807403564,
      "step": 53818
    },
    {
      "epoch": 5.132484436035156e-05,
      "step": 53818,
      "training_step_time": 7.760778903961182
    },
    {
      "epoch": 5.132579803466797e-05,
      "model_forward_time": 2.256242036819458,
      "step": 53819
    },
    {
      "epoch": 5.132579803466797e-05,
      "step": 53819,
      "training_step_time": 8.528331756591797
    },
    {
      "epoch": 5.1326751708984375e-05,
      "grad_norm": 0.11202135682106018,
      "learning_rate": 4.781976480800465e-06,
      "loss": 0.0604,
      "step": 53820
    },
    {
      "epoch": 5.1326751708984375e-05,
      "model_forward_time": 1.714033603668213,
      "step": 53820
    },
    {
      "dynamics_loss_avg": 0.045037051662802696,
      "epoch": 5.1326751708984375e-05,
      "step": 53820
    },
    {
      "action_loss_avg": 0.0015349030058132485,
      "epoch": 5.1326751708984375e-05,
      "step": 53820
    },
    {
      "epoch": 5.1326751708984375e-05,
      "step": 53820,
      "training_step_time": 7.936254262924194
    },
    {
      "epoch": 5.132770538330078e-05,
      "model_forward_time": 1.7173523902893066,
      "step": 53821
    },
    {
      "epoch": 5.132770538330078e-05,
      "step": 53821,
      "training_step_time": 7.7624382972717285
    },
    {
      "epoch": 5.132865905761719e-05,
      "model_forward_time": 1.7281770706176758,
      "step": 53822
    },
    {
      "epoch": 5.132865905761719e-05,
      "step": 53822,
      "training_step_time": 6.89802885055542
    },
    {
      "epoch": 5.1329612731933594e-05,
      "model_forward_time": 1.7307097911834717,
      "step": 53823
    },
    {
      "epoch": 5.1329612731933594e-05,
      "step": 53823,
      "training_step_time": 6.951681137084961
    },
    {
      "epoch": 5.133056640625e-05,
      "model_forward_time": 1.7244925498962402,
      "step": 53824
    },
    {
      "epoch": 5.133056640625e-05,
      "step": 53824,
      "training_step_time": 6.943545579910278
    },
    {
      "epoch": 5.133152008056641e-05,
      "model_forward_time": 1.7308130264282227,
      "step": 53825
    },
    {
      "epoch": 5.133152008056641e-05,
      "step": 53825,
      "training_step_time": 6.973879098892212
    },
    {
      "epoch": 5.133247375488281e-05,
      "model_forward_time": 1.7296793460845947,
      "step": 53826
    },
    {
      "epoch": 5.133247375488281e-05,
      "step": 53826,
      "training_step_time": 6.950270891189575
    },
    {
      "epoch": 5.133342742919922e-05,
      "model_forward_time": 1.7279798984527588,
      "step": 53827
    },
    {
      "epoch": 5.133342742919922e-05,
      "step": 53827,
      "training_step_time": 7.866644620895386
    },
    {
      "epoch": 5.133438110351563e-05,
      "model_forward_time": 2.2623119354248047,
      "step": 53828
    },
    {
      "epoch": 5.133438110351563e-05,
      "step": 53828,
      "training_step_time": 11.065440893173218
    },
    {
      "epoch": 5.133533477783203e-05,
      "model_forward_time": 1.713618278503418,
      "step": 53829
    },
    {
      "epoch": 5.133533477783203e-05,
      "step": 53829,
      "training_step_time": 8.777292490005493
    },
    {
      "epoch": 5.1336288452148436e-05,
      "grad_norm": 0.12039417773485184,
      "learning_rate": 4.780324595611289e-06,
      "loss": 0.061,
      "step": 53830
    },
    {
      "epoch": 5.1336288452148436e-05,
      "model_forward_time": 1.716033697128296,
      "step": 53830
    },
    {
      "dynamics_loss_avg": 0.06506644506007433,
      "epoch": 5.1336288452148436e-05,
      "step": 53830
    },
    {
      "action_loss_avg": 0.0015832040517125279,
      "epoch": 5.1336288452148436e-05,
      "step": 53830
    },
    {
      "epoch": 5.1336288452148436e-05,
      "step": 53830,
      "training_step_time": 7.887031555175781
    },
    {
      "epoch": 5.133724212646485e-05,
      "model_forward_time": 1.7315866947174072,
      "step": 53831
    },
    {
      "epoch": 5.133724212646485e-05,
      "step": 53831,
      "training_step_time": 6.772024154663086
    },
    {
      "epoch": 5.133819580078125e-05,
      "model_forward_time": 1.350895643234253,
      "step": 53832
    },
    {
      "epoch": 5.133819580078125e-05,
      "step": 53832,
      "training_step_time": 6.98320746421814
    },
    {
      "epoch": 5.1339149475097655e-05,
      "model_forward_time": 1.728656530380249,
      "step": 53833
    },
    {
      "epoch": 5.1339149475097655e-05,
      "step": 53833,
      "training_step_time": 6.9206953048706055
    },
    {
      "epoch": 5.1340103149414066e-05,
      "model_forward_time": 1.733806848526001,
      "step": 53834
    },
    {
      "epoch": 5.1340103149414066e-05,
      "step": 53834,
      "training_step_time": 6.872018098831177
    },
    {
      "epoch": 5.134105682373047e-05,
      "model_forward_time": 1.7151908874511719,
      "step": 53835
    },
    {
      "epoch": 5.134105682373047e-05,
      "step": 53835,
      "training_step_time": 6.870251417160034
    },
    {
      "epoch": 5.1342010498046874e-05,
      "model_forward_time": 1.7287697792053223,
      "step": 53836
    },
    {
      "epoch": 5.1342010498046874e-05,
      "step": 53836,
      "training_step_time": 7.693117380142212
    },
    {
      "epoch": 5.1342964172363284e-05,
      "model_forward_time": 2.3097846508026123,
      "step": 53837
    },
    {
      "epoch": 5.1342964172363284e-05,
      "step": 53837,
      "training_step_time": 9.41557240486145
    },
    {
      "epoch": 5.134391784667969e-05,
      "model_forward_time": 1.7090673446655273,
      "step": 53838
    },
    {
      "epoch": 5.134391784667969e-05,
      "step": 53838,
      "training_step_time": 8.747089385986328
    },
    {
      "epoch": 5.134487152099609e-05,
      "model_forward_time": 1.7210423946380615,
      "step": 53839
    },
    {
      "epoch": 5.134487152099609e-05,
      "step": 53839,
      "training_step_time": 7.634160995483398
    },
    {
      "epoch": 5.13458251953125e-05,
      "grad_norm": 0.08168211579322815,
      "learning_rate": 4.778672734445485e-06,
      "loss": 0.062,
      "step": 53840
    },
    {
      "epoch": 5.13458251953125e-05,
      "model_forward_time": 1.7171428203582764,
      "step": 53840
    },
    {
      "dynamics_loss_avg": 0.06651752553880215,
      "epoch": 5.13458251953125e-05,
      "step": 53840
    },
    {
      "action_loss_avg": 0.001548359275329858,
      "epoch": 5.13458251953125e-05,
      "step": 53840
    },
    {
      "epoch": 5.13458251953125e-05,
      "step": 53840,
      "training_step_time": 7.407705068588257
    },
    {
      "epoch": 5.134677886962891e-05,
      "model_forward_time": 1.7232353687286377,
      "step": 53841
    },
    {
      "epoch": 5.134677886962891e-05,
      "step": 53841,
      "training_step_time": 6.869787216186523
    },
    {
      "epoch": 5.134773254394531e-05,
      "model_forward_time": 1.7286410331726074,
      "step": 53842
    },
    {
      "epoch": 5.134773254394531e-05,
      "step": 53842,
      "training_step_time": 6.905196905136108
    },
    {
      "epoch": 5.134868621826172e-05,
      "model_forward_time": 1.7165307998657227,
      "step": 53843
    },
    {
      "epoch": 5.134868621826172e-05,
      "step": 53843,
      "training_step_time": 6.933063745498657
    },
    {
      "epoch": 5.1349639892578126e-05,
      "model_forward_time": 1.7223145961761475,
      "step": 53844
    },
    {
      "epoch": 5.1349639892578126e-05,
      "step": 53844,
      "training_step_time": 6.895346164703369
    },
    {
      "epoch": 5.135059356689453e-05,
      "model_forward_time": 1.7239289283752441,
      "step": 53845
    },
    {
      "epoch": 5.135059356689453e-05,
      "step": 53845,
      "training_step_time": 7.707146167755127
    },
    {
      "epoch": 5.1351547241210934e-05,
      "model_forward_time": 2.310723304748535,
      "step": 53846
    },
    {
      "epoch": 5.1351547241210934e-05,
      "step": 53846,
      "training_step_time": 8.569380044937134
    },
    {
      "epoch": 5.1352500915527345e-05,
      "model_forward_time": 1.7298274040222168,
      "step": 53847
    },
    {
      "epoch": 5.1352500915527345e-05,
      "step": 53847,
      "training_step_time": 8.649734258651733
    },
    {
      "epoch": 5.135345458984375e-05,
      "model_forward_time": 1.7164816856384277,
      "step": 53848
    },
    {
      "epoch": 5.135345458984375e-05,
      "step": 53848,
      "training_step_time": 7.863200664520264
    },
    {
      "epoch": 5.135440826416015e-05,
      "model_forward_time": 1.720978021621704,
      "step": 53849
    },
    {
      "epoch": 5.135440826416015e-05,
      "step": 53849,
      "training_step_time": 7.7335662841796875
    },
    {
      "epoch": 5.1355361938476564e-05,
      "grad_norm": 0.09097708016633987,
      "learning_rate": 4.777020897483697e-06,
      "loss": 0.0583,
      "step": 53850
    },
    {
      "epoch": 5.1355361938476564e-05,
      "model_forward_time": 1.7122018337249756,
      "step": 53850
    },
    {
      "dynamics_loss_avg": 0.046013259515166285,
      "epoch": 5.1355361938476564e-05,
      "step": 53850
    },
    {
      "action_loss_avg": 0.0010657996637746692,
      "epoch": 5.1355361938476564e-05,
      "step": 53850
    },
    {
      "epoch": 5.1355361938476564e-05,
      "step": 53850,
      "training_step_time": 6.969415903091431
    },
    {
      "epoch": 5.135631561279297e-05,
      "model_forward_time": 1.7254931926727295,
      "step": 53851
    },
    {
      "epoch": 5.135631561279297e-05,
      "step": 53851,
      "training_step_time": 6.9405176639556885
    },
    {
      "epoch": 5.135726928710937e-05,
      "model_forward_time": 1.7277250289916992,
      "step": 53852
    },
    {
      "epoch": 5.135726928710937e-05,
      "step": 53852,
      "training_step_time": 6.908445119857788
    },
    {
      "epoch": 5.135822296142578e-05,
      "model_forward_time": 1.718144416809082,
      "step": 53853
    },
    {
      "epoch": 5.135822296142578e-05,
      "step": 53853,
      "training_step_time": 7.011331081390381
    },
    {
      "epoch": 5.135917663574219e-05,
      "model_forward_time": 1.7239434719085693,
      "step": 53854
    },
    {
      "epoch": 5.135917663574219e-05,
      "step": 53854,
      "training_step_time": 7.705755949020386
    },
    {
      "epoch": 5.136013031005859e-05,
      "model_forward_time": 2.28403902053833,
      "step": 53855
    },
    {
      "epoch": 5.136013031005859e-05,
      "step": 53855,
      "training_step_time": 10.239068984985352
    },
    {
      "epoch": 5.1361083984375e-05,
      "model_forward_time": 1.72328519821167,
      "step": 53856
    },
    {
      "epoch": 5.1361083984375e-05,
      "step": 53856,
      "training_step_time": 9.278909683227539
    },
    {
      "epoch": 5.1362037658691406e-05,
      "model_forward_time": 1.7252600193023682,
      "step": 53857
    },
    {
      "epoch": 5.1362037658691406e-05,
      "step": 53857,
      "training_step_time": 9.429242610931396
    },
    {
      "epoch": 5.136299133300781e-05,
      "model_forward_time": 1.7270503044128418,
      "step": 53858
    },
    {
      "epoch": 5.136299133300781e-05,
      "step": 53858,
      "training_step_time": 7.755566358566284
    },
    {
      "epoch": 5.136394500732422e-05,
      "model_forward_time": 1.7340795993804932,
      "step": 53859
    },
    {
      "epoch": 5.136394500732422e-05,
      "step": 53859,
      "training_step_time": 7.020833730697632
    },
    {
      "epoch": 5.1364898681640624e-05,
      "grad_norm": 0.0994831845164299,
      "learning_rate": 4.77536908490657e-06,
      "loss": 0.0604,
      "step": 53860
    },
    {
      "epoch": 5.1364898681640624e-05,
      "model_forward_time": 1.7181358337402344,
      "step": 53860
    },
    {
      "dynamics_loss_avg": 0.07025284338742495,
      "epoch": 5.1364898681640624e-05,
      "step": 53860
    },
    {
      "action_loss_avg": 0.001822917873505503,
      "epoch": 5.1364898681640624e-05,
      "step": 53860
    },
    {
      "epoch": 5.1364898681640624e-05,
      "step": 53860,
      "training_step_time": 6.784433603286743
    },
    {
      "epoch": 5.136585235595703e-05,
      "model_forward_time": 1.7239232063293457,
      "step": 53861
    },
    {
      "epoch": 5.136585235595703e-05,
      "step": 53861,
      "training_step_time": 6.9372193813323975
    },
    {
      "epoch": 5.136680603027344e-05,
      "model_forward_time": 1.729910135269165,
      "step": 53862
    },
    {
      "epoch": 5.136680603027344e-05,
      "step": 53862,
      "training_step_time": 6.9543516635894775
    },
    {
      "epoch": 5.136775970458984e-05,
      "model_forward_time": 1.7240746021270752,
      "step": 53863
    },
    {
      "epoch": 5.136775970458984e-05,
      "step": 53863,
      "training_step_time": 7.742367267608643
    },
    {
      "epoch": 5.136871337890625e-05,
      "model_forward_time": 2.2962379455566406,
      "step": 53864
    },
    {
      "epoch": 5.136871337890625e-05,
      "step": 53864,
      "training_step_time": 8.75433349609375
    },
    {
      "epoch": 5.136966705322266e-05,
      "model_forward_time": 1.8174922466278076,
      "step": 53865
    },
    {
      "epoch": 5.136966705322266e-05,
      "step": 53865,
      "training_step_time": 9.413604497909546
    },
    {
      "epoch": 5.137062072753906e-05,
      "model_forward_time": 1.825765609741211,
      "step": 53866
    },
    {
      "epoch": 5.137062072753906e-05,
      "step": 53866,
      "training_step_time": 7.852824687957764
    },
    {
      "epoch": 5.1371574401855466e-05,
      "model_forward_time": 1.8207015991210938,
      "step": 53867
    },
    {
      "epoch": 5.1371574401855466e-05,
      "step": 53867,
      "training_step_time": 7.7731757164001465
    },
    {
      "epoch": 5.137252807617188e-05,
      "model_forward_time": 1.821897029876709,
      "step": 53868
    },
    {
      "epoch": 5.137252807617188e-05,
      "step": 53868,
      "training_step_time": 6.9437034130096436
    },
    {
      "epoch": 5.137348175048828e-05,
      "model_forward_time": 1.7772412300109863,
      "step": 53869
    },
    {
      "epoch": 5.137348175048828e-05,
      "step": 53869,
      "training_step_time": 6.873722553253174
    },
    {
      "epoch": 5.1374435424804685e-05,
      "grad_norm": 0.1113753393292427,
      "learning_rate": 4.773717296894742e-06,
      "loss": 0.0554,
      "step": 53870
    },
    {
      "epoch": 5.1374435424804685e-05,
      "model_forward_time": 1.7246379852294922,
      "step": 53870
    },
    {
      "dynamics_loss_avg": 0.04155667871236801,
      "epoch": 5.1374435424804685e-05,
      "step": 53870
    },
    {
      "action_loss_avg": 0.0018916646658908576,
      "epoch": 5.1374435424804685e-05,
      "step": 53870
    },
    {
      "epoch": 5.1374435424804685e-05,
      "step": 53870,
      "training_step_time": 6.947827100753784
    },
    {
      "epoch": 5.1375389099121096e-05,
      "model_forward_time": 1.7308025360107422,
      "step": 53871
    },
    {
      "epoch": 5.1375389099121096e-05,
      "step": 53871,
      "training_step_time": 6.900899171829224
    },
    {
      "epoch": 5.13763427734375e-05,
      "model_forward_time": 1.7243621349334717,
      "step": 53872
    },
    {
      "epoch": 5.13763427734375e-05,
      "step": 53872,
      "training_step_time": 7.698427438735962
    },
    {
      "epoch": 5.1377296447753904e-05,
      "model_forward_time": 1.7274765968322754,
      "step": 53873
    },
    {
      "epoch": 5.1377296447753904e-05,
      "step": 53873,
      "training_step_time": 7.7643187046051025
    },
    {
      "epoch": 5.1378250122070315e-05,
      "model_forward_time": 2.2864089012145996,
      "step": 53874
    },
    {
      "epoch": 5.1378250122070315e-05,
      "step": 53874,
      "training_step_time": 8.2924063205719
    },
    {
      "epoch": 5.137920379638672e-05,
      "model_forward_time": 1.7225890159606934,
      "step": 53875
    },
    {
      "epoch": 5.137920379638672e-05,
      "step": 53875,
      "training_step_time": 8.863936185836792
    },
    {
      "epoch": 5.138015747070312e-05,
      "model_forward_time": 1.7334001064300537,
      "step": 53876
    },
    {
      "epoch": 5.138015747070312e-05,
      "step": 53876,
      "training_step_time": 7.804258346557617
    },
    {
      "epoch": 5.1381111145019534e-05,
      "model_forward_time": 1.015915870666504,
      "step": 53877
    },
    {
      "epoch": 5.1381111145019534e-05,
      "step": 53877,
      "training_step_time": 6.929410219192505
    },
    {
      "epoch": 5.138206481933594e-05,
      "model_forward_time": 1.7296867370605469,
      "step": 53878
    },
    {
      "epoch": 5.138206481933594e-05,
      "step": 53878,
      "training_step_time": 6.934189558029175
    },
    {
      "epoch": 5.138301849365234e-05,
      "model_forward_time": 1.7290127277374268,
      "step": 53879
    },
    {
      "epoch": 5.138301849365234e-05,
      "step": 53879,
      "training_step_time": 6.893539905548096
    },
    {
      "epoch": 5.138397216796875e-05,
      "grad_norm": 0.11978227645158768,
      "learning_rate": 4.772065533628848e-06,
      "loss": 0.0624,
      "step": 53880
    },
    {
      "epoch": 5.138397216796875e-05,
      "model_forward_time": 1.7281603813171387,
      "step": 53880
    },
    {
      "dynamics_loss_avg": 0.043750308267772195,
      "epoch": 5.138397216796875e-05,
      "step": 53880
    },
    {
      "action_loss_avg": 0.0014145664375973866,
      "epoch": 5.138397216796875e-05,
      "step": 53880
    },
    {
      "epoch": 5.138397216796875e-05,
      "step": 53880,
      "training_step_time": 6.9556708335876465
    },
    {
      "epoch": 5.1384925842285156e-05,
      "model_forward_time": 1.7179391384124756,
      "step": 53881
    },
    {
      "epoch": 5.1384925842285156e-05,
      "step": 53881,
      "training_step_time": 6.924758195877075
    },
    {
      "epoch": 5.138587951660156e-05,
      "model_forward_time": 1.7266602516174316,
      "step": 53882
    },
    {
      "epoch": 5.138587951660156e-05,
      "step": 53882,
      "training_step_time": 7.711664438247681
    },
    {
      "epoch": 5.138683319091797e-05,
      "model_forward_time": 2.314804792404175,
      "step": 53883
    },
    {
      "epoch": 5.138683319091797e-05,
      "step": 53883,
      "training_step_time": 10.203295469284058
    },
    {
      "epoch": 5.1387786865234375e-05,
      "model_forward_time": 1.7254807949066162,
      "step": 53884
    },
    {
      "epoch": 5.1387786865234375e-05,
      "step": 53884,
      "training_step_time": 8.87477445602417
    },
    {
      "epoch": 5.138874053955078e-05,
      "model_forward_time": 1.7188093662261963,
      "step": 53885
    },
    {
      "epoch": 5.138874053955078e-05,
      "step": 53885,
      "training_step_time": 7.932759523391724
    },
    {
      "epoch": 5.138969421386719e-05,
      "model_forward_time": 1.721552848815918,
      "step": 53886
    },
    {
      "epoch": 5.138969421386719e-05,
      "step": 53886,
      "training_step_time": 7.6932854652404785
    },
    {
      "epoch": 5.1390647888183594e-05,
      "model_forward_time": 1.715674877166748,
      "step": 53887
    },
    {
      "epoch": 5.1390647888183594e-05,
      "step": 53887,
      "training_step_time": 6.989985942840576
    },
    {
      "epoch": 5.13916015625e-05,
      "model_forward_time": 1.714259147644043,
      "step": 53888
    },
    {
      "epoch": 5.13916015625e-05,
      "step": 53888,
      "training_step_time": 6.939736604690552
    },
    {
      "epoch": 5.139255523681641e-05,
      "model_forward_time": 1.7216606140136719,
      "step": 53889
    },
    {
      "epoch": 5.139255523681641e-05,
      "step": 53889,
      "training_step_time": 6.97618293762207
    },
    {
      "epoch": 5.139350891113281e-05,
      "grad_norm": 0.12958163022994995,
      "learning_rate": 4.770413795289526e-06,
      "loss": 0.0561,
      "step": 53890
    },
    {
      "epoch": 5.139350891113281e-05,
      "model_forward_time": 1.7056379318237305,
      "step": 53890
    },
    {
      "dynamics_loss_avg": 0.0651096636429429,
      "epoch": 5.139350891113281e-05,
      "step": 53890
    },
    {
      "action_loss_avg": 0.0017488633922766895,
      "epoch": 5.139350891113281e-05,
      "step": 53890
    },
    {
      "epoch": 5.139350891113281e-05,
      "step": 53890,
      "training_step_time": 6.941690683364868
    },
    {
      "epoch": 5.139446258544922e-05,
      "model_forward_time": 1.7234866619110107,
      "step": 53891
    },
    {
      "epoch": 5.139446258544922e-05,
      "step": 53891,
      "training_step_time": 7.755783796310425
    },
    {
      "epoch": 5.139541625976563e-05,
      "model_forward_time": 2.301452875137329,
      "step": 53892
    },
    {
      "epoch": 5.139541625976563e-05,
      "step": 53892,
      "training_step_time": 8.816304206848145
    },
    {
      "epoch": 5.139636993408203e-05,
      "model_forward_time": 1.7079622745513916,
      "step": 53893
    },
    {
      "epoch": 5.139636993408203e-05,
      "step": 53893,
      "training_step_time": 8.932164907455444
    },
    {
      "epoch": 5.1397323608398436e-05,
      "model_forward_time": 1.7099945545196533,
      "step": 53894
    },
    {
      "epoch": 5.1397323608398436e-05,
      "step": 53894,
      "training_step_time": 9.050923824310303
    },
    {
      "epoch": 5.139827728271485e-05,
      "model_forward_time": 1.7081122398376465,
      "step": 53895
    },
    {
      "epoch": 5.139827728271485e-05,
      "step": 53895,
      "training_step_time": 7.695616006851196
    },
    {
      "epoch": 5.139923095703125e-05,
      "model_forward_time": 1.7049472332000732,
      "step": 53896
    },
    {
      "epoch": 5.139923095703125e-05,
      "step": 53896,
      "training_step_time": 6.892384767532349
    },
    {
      "epoch": 5.1400184631347655e-05,
      "model_forward_time": 1.7094337940216064,
      "step": 53897
    },
    {
      "epoch": 5.1400184631347655e-05,
      "step": 53897,
      "training_step_time": 6.928682088851929
    },
    {
      "epoch": 5.1401138305664066e-05,
      "model_forward_time": 1.7154769897460938,
      "step": 53898
    },
    {
      "epoch": 5.1401138305664066e-05,
      "step": 53898,
      "training_step_time": 6.799370527267456
    },
    {
      "epoch": 5.140209197998047e-05,
      "model_forward_time": 1.6991698741912842,
      "step": 53899
    },
    {
      "epoch": 5.140209197998047e-05,
      "step": 53899,
      "training_step_time": 6.825263977050781
    },
    {
      "epoch": 5.1403045654296874e-05,
      "grad_norm": 0.11669640243053436,
      "learning_rate": 4.768762082057407e-06,
      "loss": 0.0573,
      "step": 53900
    },
    {
      "epoch": 5.1403045654296874e-05,
      "model_forward_time": 1.6968674659729004,
      "step": 53900
    },
    {
      "dynamics_loss_avg": 0.06100521273910999,
      "epoch": 5.1403045654296874e-05,
      "step": 53900
    },
    {
      "action_loss_avg": 0.0014357654203195126,
      "epoch": 5.1403045654296874e-05,
      "step": 53900
    },
    {
      "epoch": 5.1403045654296874e-05,
      "step": 53900,
      "training_step_time": 7.671834230422974
    },
    {
      "epoch": 5.1403999328613284e-05,
      "model_forward_time": 1.7398438453674316,
      "step": 53901
    },
    {
      "epoch": 5.1403999328613284e-05,
      "step": 53901,
      "training_step_time": 10.704243183135986
    },
    {
      "epoch": 5.140495300292969e-05,
      "model_forward_time": 1.7079417705535889,
      "step": 53902
    },
    {
      "epoch": 5.140495300292969e-05,
      "step": 53902,
      "training_step_time": 9.323502779006958
    },
    {
      "epoch": 5.140590667724609e-05,
      "model_forward_time": 1.7162811756134033,
      "step": 53903
    },
    {
      "epoch": 5.140590667724609e-05,
      "step": 53903,
      "training_step_time": 10.235773086547852
    },
    {
      "epoch": 5.14068603515625e-05,
      "model_forward_time": 1.713209867477417,
      "step": 53904
    },
    {
      "epoch": 5.14068603515625e-05,
      "step": 53904,
      "training_step_time": 7.773158311843872
    },
    {
      "epoch": 5.140781402587891e-05,
      "model_forward_time": 1.7103910446166992,
      "step": 53905
    },
    {
      "epoch": 5.140781402587891e-05,
      "step": 53905,
      "training_step_time": 6.911982297897339
    },
    {
      "epoch": 5.140876770019531e-05,
      "model_forward_time": 1.7089760303497314,
      "step": 53906
    },
    {
      "epoch": 5.140876770019531e-05,
      "step": 53906,
      "training_step_time": 6.919760465621948
    },
    {
      "epoch": 5.1409721374511715e-05,
      "model_forward_time": 1.7145142555236816,
      "step": 53907
    },
    {
      "epoch": 5.1409721374511715e-05,
      "step": 53907,
      "training_step_time": 6.890719652175903
    },
    {
      "epoch": 5.1410675048828126e-05,
      "model_forward_time": 1.712541103363037,
      "step": 53908
    },
    {
      "epoch": 5.1410675048828126e-05,
      "step": 53908,
      "training_step_time": 6.91677451133728
    },
    {
      "epoch": 5.141162872314453e-05,
      "model_forward_time": 1.7151062488555908,
      "step": 53909
    },
    {
      "epoch": 5.141162872314453e-05,
      "step": 53909,
      "training_step_time": 7.723669528961182
    },
    {
      "epoch": 5.1412582397460934e-05,
      "grad_norm": 0.09620746970176697,
      "learning_rate": 4.7671103941131186e-06,
      "loss": 0.0583,
      "step": 53910
    },
    {
      "epoch": 5.1412582397460934e-05,
      "model_forward_time": 1.728182077407837,
      "step": 53910
    },
    {
      "dynamics_loss_avg": 0.05167752094566822,
      "epoch": 5.1412582397460934e-05,
      "step": 53910
    },
    {
      "action_loss_avg": 0.0010972709656925871,
      "epoch": 5.1412582397460934e-05,
      "step": 53910
    },
    {
      "epoch": 5.1412582397460934e-05,
      "step": 53910,
      "training_step_time": 9.425161123275757
    },
    {
      "epoch": 5.1413536071777345e-05,
      "model_forward_time": 2.2691009044647217,
      "step": 53911
    },
    {
      "epoch": 5.1413536071777345e-05,
      "step": 53911,
      "training_step_time": 9.665923595428467
    },
    {
      "epoch": 5.141448974609375e-05,
      "model_forward_time": 1.7215967178344727,
      "step": 53912
    },
    {
      "epoch": 5.141448974609375e-05,
      "step": 53912,
      "training_step_time": 7.923962593078613
    },
    {
      "epoch": 5.141544342041015e-05,
      "model_forward_time": 1.720870018005371,
      "step": 53913
    },
    {
      "epoch": 5.141544342041015e-05,
      "step": 53913,
      "training_step_time": 7.987655878067017
    },
    {
      "epoch": 5.1416397094726564e-05,
      "model_forward_time": 1.720932960510254,
      "step": 53914
    },
    {
      "epoch": 5.1416397094726564e-05,
      "step": 53914,
      "training_step_time": 6.970441102981567
    },
    {
      "epoch": 5.141735076904297e-05,
      "model_forward_time": 1.722688913345337,
      "step": 53915
    },
    {
      "epoch": 5.141735076904297e-05,
      "step": 53915,
      "training_step_time": 6.971857309341431
    },
    {
      "epoch": 5.141830444335937e-05,
      "model_forward_time": 1.7255451679229736,
      "step": 53916
    },
    {
      "epoch": 5.141830444335937e-05,
      "step": 53916,
      "training_step_time": 6.9175660610198975
    },
    {
      "epoch": 5.141925811767578e-05,
      "model_forward_time": 1.7153239250183105,
      "step": 53917
    },
    {
      "epoch": 5.141925811767578e-05,
      "step": 53917,
      "training_step_time": 6.950501203536987
    },
    {
      "epoch": 5.142021179199219e-05,
      "model_forward_time": 1.7216427326202393,
      "step": 53918
    },
    {
      "epoch": 5.142021179199219e-05,
      "step": 53918,
      "training_step_time": 8.45665955543518
    },
    {
      "epoch": 5.142116546630859e-05,
      "model_forward_time": 1.7240784168243408,
      "step": 53919
    },
    {
      "epoch": 5.142116546630859e-05,
      "step": 53919,
      "training_step_time": 9.308504343032837
    },
    {
      "epoch": 5.1422119140625e-05,
      "grad_norm": 0.10205479711294174,
      "learning_rate": 4.765458731637287e-06,
      "loss": 0.061,
      "step": 53920
    },
    {
      "epoch": 5.1422119140625e-05,
      "model_forward_time": 2.2868170738220215,
      "step": 53920
    },
    {
      "dynamics_loss_avg": 0.04214052017778158,
      "epoch": 5.1422119140625e-05,
      "step": 53920
    },
    {
      "action_loss_avg": 0.0013206503164838069,
      "epoch": 5.1422119140625e-05,
      "step": 53920
    },
    {
      "epoch": 5.1422119140625e-05,
      "step": 53920,
      "training_step_time": 9.180548906326294
    },
    {
      "epoch": 5.1423072814941406e-05,
      "model_forward_time": 1.7177326679229736,
      "step": 53921
    },
    {
      "epoch": 5.1423072814941406e-05,
      "step": 53921,
      "training_step_time": 7.880616664886475
    },
    {
      "epoch": 5.142402648925781e-05,
      "model_forward_time": 1.7222483158111572,
      "step": 53922
    },
    {
      "epoch": 5.142402648925781e-05,
      "step": 53922,
      "training_step_time": 7.806239604949951
    },
    {
      "epoch": 5.142498016357422e-05,
      "model_forward_time": 1.7218983173370361,
      "step": 53923
    },
    {
      "epoch": 5.142498016357422e-05,
      "step": 53923,
      "training_step_time": 7.719306468963623
    },
    {
      "epoch": 5.1425933837890625e-05,
      "model_forward_time": 1.720820665359497,
      "step": 53924
    },
    {
      "epoch": 5.1425933837890625e-05,
      "step": 53924,
      "training_step_time": 6.8042449951171875
    },
    {
      "epoch": 5.142688751220703e-05,
      "model_forward_time": 1.6979384422302246,
      "step": 53925
    },
    {
      "epoch": 5.142688751220703e-05,
      "step": 53925,
      "training_step_time": 6.85149359703064
    },
    {
      "epoch": 5.142784118652344e-05,
      "model_forward_time": 1.7272560596466064,
      "step": 53926
    },
    {
      "epoch": 5.142784118652344e-05,
      "step": 53926,
      "training_step_time": 6.905474662780762
    },
    {
      "epoch": 5.142879486083984e-05,
      "model_forward_time": 1.694077730178833,
      "step": 53927
    },
    {
      "epoch": 5.142879486083984e-05,
      "step": 53927,
      "training_step_time": 7.51503324508667
    },
    {
      "epoch": 5.142974853515625e-05,
      "model_forward_time": 1.3379554748535156,
      "step": 53928
    },
    {
      "epoch": 5.142974853515625e-05,
      "step": 53928,
      "training_step_time": 7.699641704559326
    },
    {
      "epoch": 5.143070220947266e-05,
      "model_forward_time": 2.3262791633605957,
      "step": 53929
    },
    {
      "epoch": 5.143070220947266e-05,
      "step": 53929,
      "training_step_time": 8.833256959915161
    },
    {
      "epoch": 5.143165588378906e-05,
      "grad_norm": 0.10278326272964478,
      "learning_rate": 4.7638070948105366e-06,
      "loss": 0.0597,
      "step": 53930
    },
    {
      "epoch": 5.143165588378906e-05,
      "model_forward_time": 1.7126362323760986,
      "step": 53930
    },
    {
      "dynamics_loss_avg": 0.04831251055002213,
      "epoch": 5.143165588378906e-05,
      "step": 53930
    },
    {
      "action_loss_avg": 0.0012975273537449538,
      "epoch": 5.143165588378906e-05,
      "step": 53930
    },
    {
      "epoch": 5.143165588378906e-05,
      "step": 53930,
      "training_step_time": 7.812170505523682
    },
    {
      "epoch": 5.1432609558105466e-05,
      "model_forward_time": 1.721364974975586,
      "step": 53931
    },
    {
      "epoch": 5.1432609558105466e-05,
      "step": 53931,
      "training_step_time": 8.875447750091553
    },
    {
      "epoch": 5.143356323242188e-05,
      "model_forward_time": 1.719559907913208,
      "step": 53932
    },
    {
      "epoch": 5.143356323242188e-05,
      "step": 53932,
      "training_step_time": 7.803763151168823
    },
    {
      "epoch": 5.143451690673828e-05,
      "model_forward_time": 1.7205710411071777,
      "step": 53933
    },
    {
      "epoch": 5.143451690673828e-05,
      "step": 53933,
      "training_step_time": 6.926418304443359
    },
    {
      "epoch": 5.1435470581054685e-05,
      "model_forward_time": 1.7176504135131836,
      "step": 53934
    },
    {
      "epoch": 5.1435470581054685e-05,
      "step": 53934,
      "training_step_time": 6.925197601318359
    },
    {
      "epoch": 5.1436424255371096e-05,
      "model_forward_time": 1.7217178344726562,
      "step": 53935
    },
    {
      "epoch": 5.1436424255371096e-05,
      "step": 53935,
      "training_step_time": 6.8607399463653564
    },
    {
      "epoch": 5.14373779296875e-05,
      "model_forward_time": 1.7238242626190186,
      "step": 53936
    },
    {
      "epoch": 5.14373779296875e-05,
      "step": 53936,
      "training_step_time": 7.5675013065338135
    },
    {
      "epoch": 5.1438331604003904e-05,
      "model_forward_time": 1.7243385314941406,
      "step": 53937
    },
    {
      "epoch": 5.1438331604003904e-05,
      "step": 53937,
      "training_step_time": 7.748018264770508
    },
    {
      "epoch": 5.1439285278320315e-05,
      "model_forward_time": 2.306638479232788,
      "step": 53938
    },
    {
      "epoch": 5.1439285278320315e-05,
      "step": 53938,
      "training_step_time": 7.854181528091431
    },
    {
      "epoch": 5.144023895263672e-05,
      "model_forward_time": 1.7218763828277588,
      "step": 53939
    },
    {
      "epoch": 5.144023895263672e-05,
      "step": 53939,
      "training_step_time": 8.795732021331787
    },
    {
      "epoch": 5.144119262695312e-05,
      "grad_norm": 0.13454459607601166,
      "learning_rate": 4.762155483813487e-06,
      "loss": 0.0565,
      "step": 53940
    },
    {
      "epoch": 5.144119262695312e-05,
      "model_forward_time": 1.7123517990112305,
      "step": 53940
    },
    {
      "dynamics_loss_avg": 0.05398642625659704,
      "epoch": 5.144119262695312e-05,
      "step": 53940
    },
    {
      "action_loss_avg": 0.0013740987051278353,
      "epoch": 5.144119262695312e-05,
      "step": 53940
    },
    {
      "epoch": 5.144119262695312e-05,
      "step": 53940,
      "training_step_time": 9.159334659576416
    },
    {
      "epoch": 5.1442146301269534e-05,
      "model_forward_time": 1.7203385829925537,
      "step": 53941
    },
    {
      "epoch": 5.1442146301269534e-05,
      "step": 53941,
      "training_step_time": 7.904654502868652
    },
    {
      "epoch": 5.144309997558594e-05,
      "model_forward_time": 1.7182557582855225,
      "step": 53942
    },
    {
      "epoch": 5.144309997558594e-05,
      "step": 53942,
      "training_step_time": 6.958462953567505
    },
    {
      "epoch": 5.144405364990234e-05,
      "model_forward_time": 1.717160940170288,
      "step": 53943
    },
    {
      "epoch": 5.144405364990234e-05,
      "step": 53943,
      "training_step_time": 6.900583744049072
    },
    {
      "epoch": 5.144500732421875e-05,
      "model_forward_time": 1.719930648803711,
      "step": 53944
    },
    {
      "epoch": 5.144500732421875e-05,
      "step": 53944,
      "training_step_time": 6.882995128631592
    },
    {
      "epoch": 5.1445960998535157e-05,
      "model_forward_time": 1.724278450012207,
      "step": 53945
    },
    {
      "epoch": 5.1445960998535157e-05,
      "step": 53945,
      "training_step_time": 7.651293992996216
    },
    {
      "epoch": 5.144691467285156e-05,
      "model_forward_time": 1.7219715118408203,
      "step": 53946
    },
    {
      "epoch": 5.144691467285156e-05,
      "step": 53946,
      "training_step_time": 8.732413291931152
    },
    {
      "epoch": 5.144786834716797e-05,
      "model_forward_time": 1.7240862846374512,
      "step": 53947
    },
    {
      "epoch": 5.144786834716797e-05,
      "step": 53947,
      "training_step_time": 7.808403968811035
    },
    {
      "epoch": 5.1448822021484375e-05,
      "model_forward_time": 2.2578041553497314,
      "step": 53948
    },
    {
      "epoch": 5.1448822021484375e-05,
      "step": 53948,
      "training_step_time": 10.173292875289917
    },
    {
      "epoch": 5.144977569580078e-05,
      "model_forward_time": 1.7186486721038818,
      "step": 53949
    },
    {
      "epoch": 5.144977569580078e-05,
      "step": 53949,
      "training_step_time": 8.766758441925049
    },
    {
      "epoch": 5.145072937011719e-05,
      "grad_norm": 0.0939558893442154,
      "learning_rate": 4.760503898826754e-06,
      "loss": 0.0546,
      "step": 53950
    },
    {
      "epoch": 5.145072937011719e-05,
      "model_forward_time": 1.7133963108062744,
      "step": 53950
    },
    {
      "dynamics_loss_avg": 0.04726590346544981,
      "epoch": 5.145072937011719e-05,
      "step": 53950
    },
    {
      "action_loss_avg": 0.0012816036760341376,
      "epoch": 5.145072937011719e-05,
      "step": 53950
    },
    {
      "epoch": 5.145072937011719e-05,
      "step": 53950,
      "training_step_time": 7.829952716827393
    },
    {
      "epoch": 5.1451683044433594e-05,
      "model_forward_time": 1.7214410305023193,
      "step": 53951
    },
    {
      "epoch": 5.1451683044433594e-05,
      "step": 53951,
      "training_step_time": 7.475375652313232
    },
    {
      "epoch": 5.145263671875e-05,
      "model_forward_time": 1.7278282642364502,
      "step": 53952
    },
    {
      "epoch": 5.145263671875e-05,
      "step": 53952,
      "training_step_time": 6.873122692108154
    },
    {
      "epoch": 5.145359039306641e-05,
      "model_forward_time": 1.7248868942260742,
      "step": 53953
    },
    {
      "epoch": 5.145359039306641e-05,
      "step": 53953,
      "training_step_time": 6.851407051086426
    },
    {
      "epoch": 5.145454406738281e-05,
      "model_forward_time": 1.725928783416748,
      "step": 53954
    },
    {
      "epoch": 5.145454406738281e-05,
      "step": 53954,
      "training_step_time": 7.570857048034668
    },
    {
      "epoch": 5.145549774169922e-05,
      "model_forward_time": 1.7258954048156738,
      "step": 53955
    },
    {
      "epoch": 5.145549774169922e-05,
      "step": 53955,
      "training_step_time": 7.6604163646698
    },
    {
      "epoch": 5.145645141601563e-05,
      "model_forward_time": 1.7220470905303955,
      "step": 53956
    },
    {
      "epoch": 5.145645141601563e-05,
      "step": 53956,
      "training_step_time": 7.700548410415649
    },
    {
      "epoch": 5.145740509033203e-05,
      "model_forward_time": 2.3192620277404785,
      "step": 53957
    },
    {
      "epoch": 5.145740509033203e-05,
      "step": 53957,
      "training_step_time": 8.465427160263062
    },
    {
      "epoch": 5.1458358764648436e-05,
      "model_forward_time": 1.7182512283325195,
      "step": 53958
    },
    {
      "epoch": 5.1458358764648436e-05,
      "step": 53958,
      "training_step_time": 9.669273614883423
    },
    {
      "epoch": 5.145931243896485e-05,
      "model_forward_time": 1.7156524658203125,
      "step": 53959
    },
    {
      "epoch": 5.145931243896485e-05,
      "step": 53959,
      "training_step_time": 8.666808128356934
    },
    {
      "epoch": 5.146026611328125e-05,
      "grad_norm": 0.19881784915924072,
      "learning_rate": 4.758852340030957e-06,
      "loss": 0.0585,
      "step": 53960
    },
    {
      "epoch": 5.146026611328125e-05,
      "model_forward_time": 1.7154972553253174,
      "step": 53960
    },
    {
      "dynamics_loss_avg": 0.05521753486245871,
      "epoch": 5.146026611328125e-05,
      "step": 53960
    },
    {
      "action_loss_avg": 0.0012855471402872353,
      "epoch": 5.146026611328125e-05,
      "step": 53960
    },
    {
      "epoch": 5.146026611328125e-05,
      "step": 53960,
      "training_step_time": 7.835902452468872
    },
    {
      "epoch": 5.1461219787597655e-05,
      "model_forward_time": 1.0100631713867188,
      "step": 53961
    },
    {
      "epoch": 5.1461219787597655e-05,
      "step": 53961,
      "training_step_time": 6.8881635665893555
    },
    {
      "epoch": 5.1462173461914066e-05,
      "model_forward_time": 1.711139440536499,
      "step": 53962
    },
    {
      "epoch": 5.1462173461914066e-05,
      "step": 53962,
      "training_step_time": 6.878598928451538
    },
    {
      "epoch": 5.146312713623047e-05,
      "model_forward_time": 1.7203302383422852,
      "step": 53963
    },
    {
      "epoch": 5.146312713623047e-05,
      "step": 53963,
      "training_step_time": 7.598785400390625
    },
    {
      "epoch": 5.1464080810546874e-05,
      "model_forward_time": 1.7239437103271484,
      "step": 53964
    },
    {
      "epoch": 5.1464080810546874e-05,
      "step": 53964,
      "training_step_time": 7.4988744258880615
    },
    {
      "epoch": 5.1465034484863285e-05,
      "model_forward_time": 1.722127914428711,
      "step": 53965
    },
    {
      "epoch": 5.1465034484863285e-05,
      "step": 53965,
      "training_step_time": 9.145133256912231
    },
    {
      "epoch": 5.146598815917969e-05,
      "model_forward_time": 2.307325601577759,
      "step": 53966
    },
    {
      "epoch": 5.146598815917969e-05,
      "step": 53966,
      "training_step_time": 8.464259147644043
    },
    {
      "epoch": 5.146694183349609e-05,
      "model_forward_time": 1.7156987190246582,
      "step": 53967
    },
    {
      "epoch": 5.146694183349609e-05,
      "step": 53967,
      "training_step_time": 9.264038324356079
    },
    {
      "epoch": 5.14678955078125e-05,
      "model_forward_time": 1.7192022800445557,
      "step": 53968
    },
    {
      "epoch": 5.14678955078125e-05,
      "step": 53968,
      "training_step_time": 7.877965688705444
    },
    {
      "epoch": 5.146884918212891e-05,
      "model_forward_time": 1.7234814167022705,
      "step": 53969
    },
    {
      "epoch": 5.146884918212891e-05,
      "step": 53969,
      "training_step_time": 7.836693048477173
    },
    {
      "epoch": 5.146980285644531e-05,
      "grad_norm": 0.12559883296489716,
      "learning_rate": 4.757200807606705e-06,
      "loss": 0.057,
      "step": 53970
    },
    {
      "epoch": 5.146980285644531e-05,
      "model_forward_time": 1.711047887802124,
      "step": 53970
    },
    {
      "dynamics_loss_avg": 0.050589813850820065,
      "epoch": 5.146980285644531e-05,
      "step": 53970
    },
    {
      "action_loss_avg": 0.0013435778790153563,
      "epoch": 5.146980285644531e-05,
      "step": 53970
    },
    {
      "epoch": 5.146980285644531e-05,
      "step": 53970,
      "training_step_time": 6.917442798614502
    },
    {
      "epoch": 5.1470756530761715e-05,
      "model_forward_time": 1.7203645706176758,
      "step": 53971
    },
    {
      "epoch": 5.1470756530761715e-05,
      "step": 53971,
      "training_step_time": 6.941622734069824
    },
    {
      "epoch": 5.1471710205078126e-05,
      "model_forward_time": 1.7148096561431885,
      "step": 53972
    },
    {
      "epoch": 5.1471710205078126e-05,
      "step": 53972,
      "training_step_time": 7.615714073181152
    },
    {
      "epoch": 5.147266387939453e-05,
      "model_forward_time": 1.7179481983184814,
      "step": 53973
    },
    {
      "epoch": 5.147266387939453e-05,
      "step": 53973,
      "training_step_time": 7.423468828201294
    },
    {
      "epoch": 5.1473617553710934e-05,
      "model_forward_time": 1.7245891094207764,
      "step": 53974
    },
    {
      "epoch": 5.1473617553710934e-05,
      "step": 53974,
      "training_step_time": 8.796096324920654
    },
    {
      "epoch": 5.1474571228027345e-05,
      "model_forward_time": 2.311875820159912,
      "step": 53975
    },
    {
      "epoch": 5.1474571228027345e-05,
      "step": 53975,
      "training_step_time": 11.061810731887817
    },
    {
      "epoch": 5.147552490234375e-05,
      "model_forward_time": 1.7196247577667236,
      "step": 53976
    },
    {
      "epoch": 5.147552490234375e-05,
      "step": 53976,
      "training_step_time": 8.823338270187378
    },
    {
      "epoch": 5.147647857666015e-05,
      "model_forward_time": 1.7216713428497314,
      "step": 53977
    },
    {
      "epoch": 5.147647857666015e-05,
      "step": 53977,
      "training_step_time": 7.890198469161987
    },
    {
      "epoch": 5.1477432250976564e-05,
      "model_forward_time": 1.7231478691101074,
      "step": 53978
    },
    {
      "epoch": 5.1477432250976564e-05,
      "step": 53978,
      "training_step_time": 7.889169931411743
    },
    {
      "epoch": 5.147838592529297e-05,
      "model_forward_time": 1.7191333770751953,
      "step": 53979
    },
    {
      "epoch": 5.147838592529297e-05,
      "step": 53979,
      "training_step_time": 7.516982316970825
    },
    {
      "epoch": 5.147933959960937e-05,
      "grad_norm": 0.09677055478096008,
      "learning_rate": 4.755549301734608e-06,
      "loss": 0.0588,
      "step": 53980
    },
    {
      "epoch": 5.147933959960937e-05,
      "model_forward_time": 1.7044987678527832,
      "step": 53980
    },
    {
      "dynamics_loss_avg": 0.045744766294956204,
      "epoch": 5.147933959960937e-05,
      "step": 53980
    },
    {
      "action_loss_avg": 0.0011616800882620737,
      "epoch": 5.147933959960937e-05,
      "step": 53980
    },
    {
      "epoch": 5.147933959960937e-05,
      "step": 53980,
      "training_step_time": 6.911591291427612
    },
    {
      "epoch": 5.148029327392578e-05,
      "model_forward_time": 1.7199547290802002,
      "step": 53981
    },
    {
      "epoch": 5.148029327392578e-05,
      "step": 53981,
      "training_step_time": 7.592597007751465
    },
    {
      "epoch": 5.148124694824219e-05,
      "model_forward_time": 1.724308729171753,
      "step": 53982
    },
    {
      "epoch": 5.148124694824219e-05,
      "step": 53982,
      "training_step_time": 7.348456144332886
    },
    {
      "epoch": 5.148220062255859e-05,
      "model_forward_time": 1.719348669052124,
      "step": 53983
    },
    {
      "epoch": 5.148220062255859e-05,
      "step": 53983,
      "training_step_time": 8.267298460006714
    },
    {
      "epoch": 5.1483154296875e-05,
      "model_forward_time": 1.7932219505310059,
      "step": 53984
    },
    {
      "epoch": 5.1483154296875e-05,
      "step": 53984,
      "training_step_time": 8.633886814117432
    },
    {
      "epoch": 5.1484107971191406e-05,
      "model_forward_time": 2.267185688018799,
      "step": 53985
    },
    {
      "epoch": 5.1484107971191406e-05,
      "step": 53985,
      "training_step_time": 8.634376764297485
    },
    {
      "epoch": 5.148506164550781e-05,
      "model_forward_time": 1.7224113941192627,
      "step": 53986
    },
    {
      "epoch": 5.148506164550781e-05,
      "step": 53986,
      "training_step_time": 8.860323667526245
    },
    {
      "epoch": 5.148601531982422e-05,
      "model_forward_time": 1.7308566570281982,
      "step": 53987
    },
    {
      "epoch": 5.148601531982422e-05,
      "step": 53987,
      "training_step_time": 7.813480377197266
    },
    {
      "epoch": 5.1486968994140625e-05,
      "model_forward_time": 1.7270057201385498,
      "step": 53988
    },
    {
      "epoch": 5.1486968994140625e-05,
      "step": 53988,
      "training_step_time": 7.423391103744507
    },
    {
      "epoch": 5.148792266845703e-05,
      "model_forward_time": 1.7217984199523926,
      "step": 53989
    },
    {
      "epoch": 5.148792266845703e-05,
      "step": 53989,
      "training_step_time": 6.892759561538696
    },
    {
      "epoch": 5.148887634277344e-05,
      "grad_norm": 0.11303611099720001,
      "learning_rate": 4.753897822595271e-06,
      "loss": 0.0571,
      "step": 53990
    },
    {
      "epoch": 5.148887634277344e-05,
      "model_forward_time": 1.7155847549438477,
      "step": 53990
    },
    {
      "dynamics_loss_avg": 0.06668877620249987,
      "epoch": 5.148887634277344e-05,
      "step": 53990
    },
    {
      "action_loss_avg": 0.001688451407244429,
      "epoch": 5.148887634277344e-05,
      "step": 53990
    },
    {
      "epoch": 5.148887634277344e-05,
      "step": 53990,
      "training_step_time": 7.555952072143555
    },
    {
      "epoch": 5.1489830017089843e-05,
      "model_forward_time": 1.7242140769958496,
      "step": 53991
    },
    {
      "epoch": 5.1489830017089843e-05,
      "step": 53991,
      "training_step_time": 6.9444420337677
    },
    {
      "epoch": 5.149078369140625e-05,
      "model_forward_time": 1.721602439880371,
      "step": 53992
    },
    {
      "epoch": 5.149078369140625e-05,
      "step": 53992,
      "training_step_time": 8.229548454284668
    },
    {
      "epoch": 5.149173736572266e-05,
      "model_forward_time": 1.7991893291473389,
      "step": 53993
    },
    {
      "epoch": 5.149173736572266e-05,
      "step": 53993,
      "training_step_time": 8.309403896331787
    },
    {
      "epoch": 5.149269104003906e-05,
      "model_forward_time": 2.2734127044677734,
      "step": 53994
    },
    {
      "epoch": 5.149269104003906e-05,
      "step": 53994,
      "training_step_time": 8.345932245254517
    },
    {
      "epoch": 5.1493644714355466e-05,
      "model_forward_time": 1.7234320640563965,
      "step": 53995
    },
    {
      "epoch": 5.1493644714355466e-05,
      "step": 53995,
      "training_step_time": 8.713307857513428
    },
    {
      "epoch": 5.149459838867188e-05,
      "model_forward_time": 1.7256386280059814,
      "step": 53996
    },
    {
      "epoch": 5.149459838867188e-05,
      "step": 53996,
      "training_step_time": 7.815860033035278
    },
    {
      "epoch": 5.149555206298828e-05,
      "model_forward_time": 1.743419885635376,
      "step": 53997
    },
    {
      "epoch": 5.149555206298828e-05,
      "step": 53997,
      "training_step_time": 7.665365934371948
    },
    {
      "epoch": 5.1496505737304685e-05,
      "model_forward_time": 1.757246971130371,
      "step": 53998
    },
    {
      "epoch": 5.1496505737304685e-05,
      "step": 53998,
      "training_step_time": 7.899958372116089
    },
    {
      "epoch": 5.1497459411621096e-05,
      "model_forward_time": 1.759131908416748,
      "step": 53999
    },
    {
      "epoch": 5.1497459411621096e-05,
      "step": 53999,
      "training_step_time": 7.581468343734741
    },
    {
      "epoch": 5.14984130859375e-05,
      "grad_norm": 0.09907599538564682,
      "learning_rate": 4.7522463703693e-06,
      "loss": 0.0561,
      "step": 54000
    },
    {
      "epoch": 5.14984130859375e-05,
      "model_forward_time": 1.748046636581421,
      "step": 54000
    },
    {
      "dynamics_loss_avg": 0.05046522412449121,
      "epoch": 5.14984130859375e-05,
      "step": 54000
    },
    {
      "action_loss_avg": 0.0031735814234707506,
      "epoch": 5.14984130859375e-05,
      "step": 54000
    },
    {
      "epoch": 5.14984130859375e-05,
      "step": 54000,
      "training_step_time": 7.6432106494903564
    },
    {
      "epoch": 5.1499366760253904e-05,
      "model_forward_time": 1.7135944366455078,
      "step": 54001
    },
    {
      "epoch": 5.1499366760253904e-05,
      "step": 54001,
      "training_step_time": 11.863757848739624
    },
    {
      "epoch": 5.1500320434570315e-05,
      "model_forward_time": 1.348585844039917,
      "step": 54002
    },
    {
      "epoch": 5.1500320434570315e-05,
      "step": 54002,
      "training_step_time": 10.651779890060425
    },
    {
      "epoch": 5.150127410888672e-05,
      "model_forward_time": 1.7117562294006348,
      "step": 54003
    },
    {
      "epoch": 5.150127410888672e-05,
      "step": 54003,
      "training_step_time": 10.001092672348022
    },
    {
      "epoch": 5.150222778320312e-05,
      "model_forward_time": 1.7082016468048096,
      "step": 54004
    },
    {
      "epoch": 5.150222778320312e-05,
      "step": 54004,
      "training_step_time": 10.55303144454956
    },
    {
      "epoch": 5.1503181457519534e-05,
      "model_forward_time": 1.6907060146331787,
      "step": 54005
    },
    {
      "epoch": 5.1503181457519534e-05,
      "step": 54005,
      "training_step_time": 7.750943422317505
    },
    {
      "epoch": 5.150413513183594e-05,
      "model_forward_time": 1.6799781322479248,
      "step": 54006
    },
    {
      "epoch": 5.150413513183594e-05,
      "step": 54006,
      "training_step_time": 7.8493406772613525
    },
    {
      "epoch": 5.150508880615234e-05,
      "model_forward_time": 1.7039992809295654,
      "step": 54007
    },
    {
      "epoch": 5.150508880615234e-05,
      "step": 54007,
      "training_step_time": 7.484020709991455
    },
    {
      "epoch": 5.150604248046875e-05,
      "model_forward_time": 1.7138516902923584,
      "step": 54008
    },
    {
      "epoch": 5.150604248046875e-05,
      "step": 54008,
      "training_step_time": 8.328633308410645
    },
    {
      "epoch": 5.1506996154785157e-05,
      "model_forward_time": 1.7106215953826904,
      "step": 54009
    },
    {
      "epoch": 5.1506996154785157e-05,
      "step": 54009,
      "training_step_time": 7.9578857421875
    },
    {
      "epoch": 5.150794982910156e-05,
      "grad_norm": 0.1321217268705368,
      "learning_rate": 4.75059494523729e-06,
      "loss": 0.0579,
      "step": 54010
    },
    {
      "epoch": 5.150794982910156e-05,
      "model_forward_time": 1.7032928466796875,
      "step": 54010
    },
    {
      "dynamics_loss_avg": 0.051584658399224284,
      "epoch": 5.150794982910156e-05,
      "step": 54010
    },
    {
      "action_loss_avg": 0.001728128723334521,
      "epoch": 5.150794982910156e-05,
      "step": 54010
    },
    {
      "epoch": 5.150794982910156e-05,
      "step": 54010,
      "training_step_time": 9.09361720085144
    },
    {
      "epoch": 5.150890350341797e-05,
      "model_forward_time": 1.7050867080688477,
      "step": 54011
    },
    {
      "epoch": 5.150890350341797e-05,
      "step": 54011,
      "training_step_time": 7.847937345504761
    },
    {
      "epoch": 5.1509857177734375e-05,
      "model_forward_time": 2.273033857345581,
      "step": 54012
    },
    {
      "epoch": 5.1509857177734375e-05,
      "step": 54012,
      "training_step_time": 8.657951593399048
    },
    {
      "epoch": 5.151081085205078e-05,
      "model_forward_time": 1.7052528858184814,
      "step": 54013
    },
    {
      "epoch": 5.151081085205078e-05,
      "step": 54013,
      "training_step_time": 10.673787593841553
    },
    {
      "epoch": 5.151176452636719e-05,
      "model_forward_time": 1.718592882156372,
      "step": 54014
    },
    {
      "epoch": 5.151176452636719e-05,
      "step": 54014,
      "training_step_time": 7.798458576202393
    },
    {
      "epoch": 5.1512718200683594e-05,
      "model_forward_time": 1.7057616710662842,
      "step": 54015
    },
    {
      "epoch": 5.1512718200683594e-05,
      "step": 54015,
      "training_step_time": 7.846960783004761
    },
    {
      "epoch": 5.1513671875e-05,
      "model_forward_time": 1.705862045288086,
      "step": 54016
    },
    {
      "epoch": 5.1513671875e-05,
      "step": 54016,
      "training_step_time": 7.423676013946533
    },
    {
      "epoch": 5.151462554931641e-05,
      "model_forward_time": 1.7094497680664062,
      "step": 54017
    },
    {
      "epoch": 5.151462554931641e-05,
      "step": 54017,
      "training_step_time": 7.54180645942688
    },
    {
      "epoch": 5.151557922363281e-05,
      "model_forward_time": 1.7062442302703857,
      "step": 54018
    },
    {
      "epoch": 5.151557922363281e-05,
      "step": 54018,
      "training_step_time": 7.590519666671753
    },
    {
      "epoch": 5.151653289794922e-05,
      "model_forward_time": 1.707655429840088,
      "step": 54019
    },
    {
      "epoch": 5.151653289794922e-05,
      "step": 54019,
      "training_step_time": 8.010986804962158
    },
    {
      "epoch": 5.151748657226563e-05,
      "grad_norm": 0.1289503425359726,
      "learning_rate": 4.7489435473798435e-06,
      "loss": 0.0592,
      "step": 54020
    },
    {
      "epoch": 5.151748657226563e-05,
      "model_forward_time": 1.7092838287353516,
      "step": 54020
    },
    {
      "dynamics_loss_avg": 0.0605049105361104,
      "epoch": 5.151748657226563e-05,
      "step": 54020
    },
    {
      "action_loss_avg": 0.001197967171901837,
      "epoch": 5.151748657226563e-05,
      "step": 54020
    },
    {
      "epoch": 5.151748657226563e-05,
      "step": 54020,
      "training_step_time": 7.880566358566284
    },
    {
      "epoch": 5.151844024658203e-05,
      "model_forward_time": 2.259154796600342,
      "step": 54021
    },
    {
      "epoch": 5.151844024658203e-05,
      "step": 54021,
      "training_step_time": 10.131186962127686
    },
    {
      "epoch": 5.1519393920898436e-05,
      "model_forward_time": 1.7092938423156738,
      "step": 54022
    },
    {
      "epoch": 5.1519393920898436e-05,
      "step": 54022,
      "training_step_time": 8.528190851211548
    },
    {
      "epoch": 5.152034759521485e-05,
      "model_forward_time": 1.6825072765350342,
      "step": 54023
    },
    {
      "epoch": 5.152034759521485e-05,
      "step": 54023,
      "training_step_time": 7.848994016647339
    },
    {
      "epoch": 5.152130126953125e-05,
      "model_forward_time": 1.695530652999878,
      "step": 54024
    },
    {
      "epoch": 5.152130126953125e-05,
      "step": 54024,
      "training_step_time": 7.774855375289917
    },
    {
      "epoch": 5.1522254943847655e-05,
      "model_forward_time": 1.695716142654419,
      "step": 54025
    },
    {
      "epoch": 5.1522254943847655e-05,
      "step": 54025,
      "training_step_time": 7.252920150756836
    },
    {
      "epoch": 5.1523208618164066e-05,
      "model_forward_time": 1.6950836181640625,
      "step": 54026
    },
    {
      "epoch": 5.1523208618164066e-05,
      "step": 54026,
      "training_step_time": 7.648670196533203
    },
    {
      "epoch": 5.152416229248047e-05,
      "model_forward_time": 1.6967594623565674,
      "step": 54027
    },
    {
      "epoch": 5.152416229248047e-05,
      "step": 54027,
      "training_step_time": 7.595225095748901
    },
    {
      "epoch": 5.1525115966796874e-05,
      "model_forward_time": 1.693716287612915,
      "step": 54028
    },
    {
      "epoch": 5.1525115966796874e-05,
      "step": 54028,
      "training_step_time": 7.782742500305176
    },
    {
      "epoch": 5.1526069641113285e-05,
      "model_forward_time": 1.7000267505645752,
      "step": 54029
    },
    {
      "epoch": 5.1526069641113285e-05,
      "step": 54029,
      "training_step_time": 9.00711178779602
    },
    {
      "epoch": 5.152702331542969e-05,
      "grad_norm": 0.11189686506986618,
      "learning_rate": 4.747292176977554e-06,
      "loss": 0.0579,
      "step": 54030
    },
    {
      "epoch": 5.152702331542969e-05,
      "model_forward_time": 2.256030559539795,
      "step": 54030
    },
    {
      "dynamics_loss_avg": 0.05185411870479584,
      "epoch": 5.152702331542969e-05,
      "step": 54030
    },
    {
      "action_loss_avg": 0.0018717554805334658,
      "epoch": 5.152702331542969e-05,
      "step": 54030
    },
    {
      "epoch": 5.152702331542969e-05,
      "step": 54030,
      "training_step_time": 8.56168508529663
    },
    {
      "epoch": 5.152797698974609e-05,
      "model_forward_time": 1.7126030921936035,
      "step": 54031
    },
    {
      "epoch": 5.152797698974609e-05,
      "step": 54031,
      "training_step_time": 10.23364520072937
    },
    {
      "epoch": 5.15289306640625e-05,
      "model_forward_time": 1.6911695003509521,
      "step": 54032
    },
    {
      "epoch": 5.15289306640625e-05,
      "step": 54032,
      "training_step_time": 9.362594842910767
    },
    {
      "epoch": 5.152988433837891e-05,
      "model_forward_time": 1.3246471881866455,
      "step": 54033
    },
    {
      "epoch": 5.152988433837891e-05,
      "step": 54033,
      "training_step_time": 7.868555307388306
    },
    {
      "epoch": 5.153083801269531e-05,
      "model_forward_time": 1.718317985534668,
      "step": 54034
    },
    {
      "epoch": 5.153083801269531e-05,
      "step": 54034,
      "training_step_time": 7.909770727157593
    },
    {
      "epoch": 5.1531791687011715e-05,
      "model_forward_time": 1.705428123474121,
      "step": 54035
    },
    {
      "epoch": 5.1531791687011715e-05,
      "step": 54035,
      "training_step_time": 7.665773868560791
    },
    {
      "epoch": 5.1532745361328126e-05,
      "model_forward_time": 1.7005696296691895,
      "step": 54036
    },
    {
      "epoch": 5.1532745361328126e-05,
      "step": 54036,
      "training_step_time": 7.531697034835815
    },
    {
      "epoch": 5.153369903564453e-05,
      "model_forward_time": 1.7062547206878662,
      "step": 54037
    },
    {
      "epoch": 5.153369903564453e-05,
      "step": 54037,
      "training_step_time": 7.676893711090088
    },
    {
      "epoch": 5.1534652709960934e-05,
      "model_forward_time": 1.7101666927337646,
      "step": 54038
    },
    {
      "epoch": 5.1534652709960934e-05,
      "step": 54038,
      "training_step_time": 7.999806880950928
    },
    {
      "epoch": 5.1535606384277345e-05,
      "model_forward_time": 2.2711212635040283,
      "step": 54039
    },
    {
      "epoch": 5.1535606384277345e-05,
      "step": 54039,
      "training_step_time": 8.888343811035156
    },
    {
      "epoch": 5.153656005859375e-05,
      "grad_norm": 0.14441749453544617,
      "learning_rate": 4.745640834211012e-06,
      "loss": 0.0566,
      "step": 54040
    },
    {
      "epoch": 5.153656005859375e-05,
      "model_forward_time": 1.7210369110107422,
      "step": 54040
    },
    {
      "dynamics_loss_avg": 0.04468973064795136,
      "epoch": 5.153656005859375e-05,
      "step": 54040
    },
    {
      "action_loss_avg": 0.0011017517826985568,
      "epoch": 5.153656005859375e-05,
      "step": 54040
    },
    {
      "epoch": 5.153656005859375e-05,
      "step": 54040,
      "training_step_time": 7.751277923583984
    },
    {
      "epoch": 5.153751373291015e-05,
      "model_forward_time": 1.7013134956359863,
      "step": 54041
    },
    {
      "epoch": 5.153751373291015e-05,
      "step": 54041,
      "training_step_time": 8.631984949111938
    },
    {
      "epoch": 5.1538467407226564e-05,
      "model_forward_time": 1.709592580795288,
      "step": 54042
    },
    {
      "epoch": 5.1538467407226564e-05,
      "step": 54042,
      "training_step_time": 7.543081998825073
    },
    {
      "epoch": 5.153942108154297e-05,
      "model_forward_time": 1.7083189487457275,
      "step": 54043
    },
    {
      "epoch": 5.153942108154297e-05,
      "step": 54043,
      "training_step_time": 7.808181047439575
    },
    {
      "epoch": 5.154037475585937e-05,
      "model_forward_time": 1.7020697593688965,
      "step": 54044
    },
    {
      "epoch": 5.154037475585937e-05,
      "step": 54044,
      "training_step_time": 7.645389080047607
    },
    {
      "epoch": 5.154132843017578e-05,
      "model_forward_time": 1.7009074687957764,
      "step": 54045
    },
    {
      "epoch": 5.154132843017578e-05,
      "step": 54045,
      "training_step_time": 7.552731275558472
    },
    {
      "epoch": 5.154228210449219e-05,
      "model_forward_time": 1.6955451965332031,
      "step": 54046
    },
    {
      "epoch": 5.154228210449219e-05,
      "step": 54046,
      "training_step_time": 7.683897972106934
    },
    {
      "epoch": 5.154323577880859e-05,
      "model_forward_time": 1.6943628787994385,
      "step": 54047
    },
    {
      "epoch": 5.154323577880859e-05,
      "step": 54047,
      "training_step_time": 8.03396463394165
    },
    {
      "epoch": 5.1544189453125e-05,
      "model_forward_time": 2.2583844661712646,
      "step": 54048
    },
    {
      "epoch": 5.1544189453125e-05,
      "step": 54048,
      "training_step_time": 8.474850416183472
    },
    {
      "epoch": 5.1545143127441406e-05,
      "model_forward_time": 1.7028257846832275,
      "step": 54049
    },
    {
      "epoch": 5.1545143127441406e-05,
      "step": 54049,
      "training_step_time": 9.41583514213562
    },
    {
      "epoch": 5.154609680175781e-05,
      "grad_norm": 0.2066119760274887,
      "learning_rate": 4.743989519260805e-06,
      "loss": 0.0584,
      "step": 54050
    },
    {
      "epoch": 5.154609680175781e-05,
      "model_forward_time": 1.695098638534546,
      "step": 54050
    },
    {
      "dynamics_loss_avg": 0.05730032604187727,
      "epoch": 5.154609680175781e-05,
      "step": 54050
    },
    {
      "action_loss_avg": 0.0012384082947392017,
      "epoch": 5.154609680175781e-05,
      "step": 54050
    },
    {
      "epoch": 5.154609680175781e-05,
      "step": 54050,
      "training_step_time": 8.636813640594482
    },
    {
      "epoch": 5.154705047607422e-05,
      "model_forward_time": 1.6893651485443115,
      "step": 54051
    },
    {
      "epoch": 5.154705047607422e-05,
      "step": 54051,
      "training_step_time": 7.728731393814087
    },
    {
      "epoch": 5.1548004150390625e-05,
      "model_forward_time": 1.6973025798797607,
      "step": 54052
    },
    {
      "epoch": 5.1548004150390625e-05,
      "step": 54052,
      "training_step_time": 7.627631902694702
    },
    {
      "epoch": 5.154895782470703e-05,
      "model_forward_time": 1.7002794742584229,
      "step": 54053
    },
    {
      "epoch": 5.154895782470703e-05,
      "step": 54053,
      "training_step_time": 7.38399600982666
    },
    {
      "epoch": 5.154991149902344e-05,
      "model_forward_time": 1.694199562072754,
      "step": 54054
    },
    {
      "epoch": 5.154991149902344e-05,
      "step": 54054,
      "training_step_time": 7.719388484954834
    },
    {
      "epoch": 5.1550865173339843e-05,
      "model_forward_time": 1.6904876232147217,
      "step": 54055
    },
    {
      "epoch": 5.1550865173339843e-05,
      "step": 54055,
      "training_step_time": 7.725372314453125
    },
    {
      "epoch": 5.155181884765625e-05,
      "model_forward_time": 1.6914317607879639,
      "step": 54056
    },
    {
      "epoch": 5.155181884765625e-05,
      "step": 54056,
      "training_step_time": 8.83373498916626
    },
    {
      "epoch": 5.155277252197266e-05,
      "model_forward_time": 2.2427239418029785,
      "step": 54057
    },
    {
      "epoch": 5.155277252197266e-05,
      "step": 54057,
      "training_step_time": 8.974035263061523
    },
    {
      "epoch": 5.155372619628906e-05,
      "model_forward_time": 1.6963000297546387,
      "step": 54058
    },
    {
      "epoch": 5.155372619628906e-05,
      "step": 54058,
      "training_step_time": 8.609188556671143
    },
    {
      "epoch": 5.1554679870605466e-05,
      "model_forward_time": 1.6938416957855225,
      "step": 54059
    },
    {
      "epoch": 5.1554679870605466e-05,
      "step": 54059,
      "training_step_time": 8.695279836654663
    },
    {
      "epoch": 5.155563354492188e-05,
      "grad_norm": 0.13244083523750305,
      "learning_rate": 4.742338232307519e-06,
      "loss": 0.0571,
      "step": 54060
    },
    {
      "epoch": 5.155563354492188e-05,
      "model_forward_time": 1.6955914497375488,
      "step": 54060
    },
    {
      "dynamics_loss_avg": 0.044879466947168115,
      "epoch": 5.155563354492188e-05,
      "step": 54060
    },
    {
      "action_loss_avg": 0.0020158638537395744,
      "epoch": 5.155563354492188e-05,
      "step": 54060
    },
    {
      "epoch": 5.155563354492188e-05,
      "step": 54060,
      "training_step_time": 7.670059680938721
    },
    {
      "epoch": 5.155658721923828e-05,
      "model_forward_time": 1.6966993808746338,
      "step": 54061
    },
    {
      "epoch": 5.155658721923828e-05,
      "step": 54061,
      "training_step_time": 7.884161472320557
    },
    {
      "epoch": 5.1557540893554685e-05,
      "model_forward_time": 1.692633867263794,
      "step": 54062
    },
    {
      "epoch": 5.1557540893554685e-05,
      "step": 54062,
      "training_step_time": 6.992055654525757
    },
    {
      "epoch": 5.1558494567871096e-05,
      "model_forward_time": 1.6999664306640625,
      "step": 54063
    },
    {
      "epoch": 5.1558494567871096e-05,
      "step": 54063,
      "training_step_time": 7.745924711227417
    },
    {
      "epoch": 5.15594482421875e-05,
      "model_forward_time": 1.6974289417266846,
      "step": 54064
    },
    {
      "epoch": 5.15594482421875e-05,
      "step": 54064,
      "training_step_time": 8.330659627914429
    },
    {
      "epoch": 5.1560401916503904e-05,
      "model_forward_time": 1.6983826160430908,
      "step": 54065
    },
    {
      "epoch": 5.1560401916503904e-05,
      "step": 54065,
      "training_step_time": 8.854184865951538
    },
    {
      "epoch": 5.1561355590820315e-05,
      "model_forward_time": 1.7036290168762207,
      "step": 54066
    },
    {
      "epoch": 5.1561355590820315e-05,
      "step": 54066,
      "training_step_time": 10.118829250335693
    },
    {
      "epoch": 5.156230926513672e-05,
      "model_forward_time": 1.7219667434692383,
      "step": 54067
    },
    {
      "epoch": 5.156230926513672e-05,
      "step": 54067,
      "training_step_time": 10.602414608001709
    },
    {
      "epoch": 5.156326293945312e-05,
      "model_forward_time": 1.692352056503296,
      "step": 54068
    },
    {
      "epoch": 5.156326293945312e-05,
      "step": 54068,
      "training_step_time": 7.8758158683776855
    },
    {
      "epoch": 5.1564216613769534e-05,
      "model_forward_time": 1.6911799907684326,
      "step": 54069
    },
    {
      "epoch": 5.1564216613769534e-05,
      "step": 54069,
      "training_step_time": 7.766303062438965
    },
    {
      "epoch": 5.156517028808594e-05,
      "grad_norm": 0.1584935039281845,
      "learning_rate": 4.740686973531737e-06,
      "loss": 0.0574,
      "step": 54070
    },
    {
      "epoch": 5.156517028808594e-05,
      "model_forward_time": 1.6908364295959473,
      "step": 54070
    },
    {
      "dynamics_loss_avg": 0.062466617301106454,
      "epoch": 5.156517028808594e-05,
      "step": 54070
    },
    {
      "action_loss_avg": 0.0025916976388543843,
      "epoch": 5.156517028808594e-05,
      "step": 54070
    },
    {
      "epoch": 5.156517028808594e-05,
      "step": 54070,
      "training_step_time": 7.901721477508545
    },
    {
      "epoch": 5.156612396240234e-05,
      "model_forward_time": 1.6866123676300049,
      "step": 54071
    },
    {
      "epoch": 5.156612396240234e-05,
      "step": 54071,
      "training_step_time": 7.026426553726196
    },
    {
      "epoch": 5.156707763671875e-05,
      "model_forward_time": 1.694145917892456,
      "step": 54072
    },
    {
      "epoch": 5.156707763671875e-05,
      "step": 54072,
      "training_step_time": 7.682513952255249
    },
    {
      "epoch": 5.156803131103516e-05,
      "model_forward_time": 1.6960830688476562,
      "step": 54073
    },
    {
      "epoch": 5.156803131103516e-05,
      "step": 54073,
      "training_step_time": 7.781567335128784
    },
    {
      "epoch": 5.156898498535156e-05,
      "model_forward_time": 1.6951262950897217,
      "step": 54074
    },
    {
      "epoch": 5.156898498535156e-05,
      "step": 54074,
      "training_step_time": 9.097583055496216
    },
    {
      "epoch": 5.156993865966797e-05,
      "model_forward_time": 1.7050232887268066,
      "step": 54075
    },
    {
      "epoch": 5.156993865966797e-05,
      "step": 54075,
      "training_step_time": 10.49783968925476
    },
    {
      "epoch": 5.1570892333984375e-05,
      "model_forward_time": 2.2520580291748047,
      "step": 54076
    },
    {
      "epoch": 5.1570892333984375e-05,
      "step": 54076,
      "training_step_time": 9.083880186080933
    },
    {
      "epoch": 5.157184600830078e-05,
      "model_forward_time": 1.6911160945892334,
      "step": 54077
    },
    {
      "epoch": 5.157184600830078e-05,
      "step": 54077,
      "training_step_time": 8.635795593261719
    },
    {
      "epoch": 5.157279968261719e-05,
      "model_forward_time": 1.697265625,
      "step": 54078
    },
    {
      "epoch": 5.157279968261719e-05,
      "step": 54078,
      "training_step_time": 8.745133399963379
    },
    {
      "epoch": 5.1573753356933594e-05,
      "model_forward_time": 1.6908812522888184,
      "step": 54079
    },
    {
      "epoch": 5.1573753356933594e-05,
      "step": 54079,
      "training_step_time": 7.689612150192261
    },
    {
      "epoch": 5.157470703125e-05,
      "grad_norm": 0.11165209859609604,
      "learning_rate": 4.739035743114037e-06,
      "loss": 0.0568,
      "step": 54080
    },
    {
      "epoch": 5.157470703125e-05,
      "model_forward_time": 1.6914374828338623,
      "step": 54080
    },
    {
      "dynamics_loss_avg": 0.05980336330831051,
      "epoch": 5.157470703125e-05,
      "step": 54080
    },
    {
      "action_loss_avg": 0.002011760196182877,
      "epoch": 5.157470703125e-05,
      "step": 54080
    },
    {
      "epoch": 5.157470703125e-05,
      "step": 54080,
      "training_step_time": 7.325462579727173
    },
    {
      "epoch": 5.157566070556641e-05,
      "model_forward_time": 1.6954596042633057,
      "step": 54081
    },
    {
      "epoch": 5.157566070556641e-05,
      "step": 54081,
      "training_step_time": 7.019495248794556
    },
    {
      "epoch": 5.157661437988281e-05,
      "model_forward_time": 1.694396734237671,
      "step": 54082
    },
    {
      "epoch": 5.157661437988281e-05,
      "step": 54082,
      "training_step_time": 7.725816011428833
    },
    {
      "epoch": 5.157756805419922e-05,
      "model_forward_time": 1.690925121307373,
      "step": 54083
    },
    {
      "epoch": 5.157756805419922e-05,
      "step": 54083,
      "training_step_time": 8.090851545333862
    },
    {
      "epoch": 5.157852172851563e-05,
      "model_forward_time": 1.6981589794158936,
      "step": 54084
    },
    {
      "epoch": 5.157852172851563e-05,
      "step": 54084,
      "training_step_time": 8.523689031600952
    },
    {
      "epoch": 5.157947540283203e-05,
      "model_forward_time": 1.8773915767669678,
      "step": 54085
    },
    {
      "epoch": 5.157947540283203e-05,
      "step": 54085,
      "training_step_time": 10.521015882492065
    },
    {
      "epoch": 5.1580429077148436e-05,
      "model_forward_time": 1.6915791034698486,
      "step": 54086
    },
    {
      "epoch": 5.1580429077148436e-05,
      "step": 54086,
      "training_step_time": 8.965674877166748
    },
    {
      "epoch": 5.158138275146485e-05,
      "model_forward_time": 1.6939496994018555,
      "step": 54087
    },
    {
      "epoch": 5.158138275146485e-05,
      "step": 54087,
      "training_step_time": 8.82648515701294
    },
    {
      "epoch": 5.158233642578125e-05,
      "model_forward_time": 1.314544439315796,
      "step": 54088
    },
    {
      "epoch": 5.158233642578125e-05,
      "step": 54088,
      "training_step_time": 7.78720498085022
    },
    {
      "epoch": 5.1583290100097655e-05,
      "model_forward_time": 1.6934006214141846,
      "step": 54089
    },
    {
      "epoch": 5.1583290100097655e-05,
      "step": 54089,
      "training_step_time": 7.1484949588775635
    },
    {
      "epoch": 5.1584243774414066e-05,
      "grad_norm": 0.11196411401033401,
      "learning_rate": 4.737384541234998e-06,
      "loss": 0.0553,
      "step": 54090
    },
    {
      "epoch": 5.1584243774414066e-05,
      "model_forward_time": 1.6942451000213623,
      "step": 54090
    },
    {
      "dynamics_loss_avg": 0.04928673766553402,
      "epoch": 5.1584243774414066e-05,
      "step": 54090
    },
    {
      "action_loss_avg": 0.0011958552320720627,
      "epoch": 5.1584243774414066e-05,
      "step": 54090
    },
    {
      "epoch": 5.1584243774414066e-05,
      "step": 54090,
      "training_step_time": 6.910075426101685
    },
    {
      "epoch": 5.158519744873047e-05,
      "model_forward_time": 1.6981046199798584,
      "step": 54091
    },
    {
      "epoch": 5.158519744873047e-05,
      "step": 54091,
      "training_step_time": 7.6159772872924805
    },
    {
      "epoch": 5.1586151123046874e-05,
      "model_forward_time": 1.691169023513794,
      "step": 54092
    },
    {
      "epoch": 5.1586151123046874e-05,
      "step": 54092,
      "training_step_time": 8.028503179550171
    },
    {
      "epoch": 5.1587104797363285e-05,
      "model_forward_time": 1.6971991062164307,
      "step": 54093
    },
    {
      "epoch": 5.1587104797363285e-05,
      "step": 54093,
      "training_step_time": 8.156149625778198
    },
    {
      "epoch": 5.158805847167969e-05,
      "model_forward_time": 2.2795753479003906,
      "step": 54094
    },
    {
      "epoch": 5.158805847167969e-05,
      "step": 54094,
      "training_step_time": 8.640862464904785
    },
    {
      "epoch": 5.158901214599609e-05,
      "model_forward_time": 1.698410987854004,
      "step": 54095
    },
    {
      "epoch": 5.158901214599609e-05,
      "step": 54095,
      "training_step_time": 9.0149507522583
    },
    {
      "epoch": 5.15899658203125e-05,
      "model_forward_time": 1.69091796875,
      "step": 54096
    },
    {
      "epoch": 5.15899658203125e-05,
      "step": 54096,
      "training_step_time": 8.668696403503418
    },
    {
      "epoch": 5.159091949462891e-05,
      "model_forward_time": 1.69651198387146,
      "step": 54097
    },
    {
      "epoch": 5.159091949462891e-05,
      "step": 54097,
      "training_step_time": 7.727608919143677
    },
    {
      "epoch": 5.159187316894531e-05,
      "model_forward_time": 1.703195571899414,
      "step": 54098
    },
    {
      "epoch": 5.159187316894531e-05,
      "step": 54098,
      "training_step_time": 7.49320912361145
    },
    {
      "epoch": 5.1592826843261716e-05,
      "model_forward_time": 1.696695327758789,
      "step": 54099
    },
    {
      "epoch": 5.1592826843261716e-05,
      "step": 54099,
      "training_step_time": 6.832854270935059
    },
    {
      "epoch": 5.1593780517578126e-05,
      "grad_norm": 0.12816089391708374,
      "learning_rate": 4.735733368075189e-06,
      "loss": 0.0539,
      "step": 54100
    },
    {
      "epoch": 5.1593780517578126e-05,
      "model_forward_time": 1.6951205730438232,
      "step": 54100
    },
    {
      "dynamics_loss_avg": 0.06056588413193822,
      "epoch": 5.1593780517578126e-05,
      "step": 54100
    },
    {
      "action_loss_avg": 0.0009525490051601083,
      "epoch": 5.1593780517578126e-05,
      "step": 54100
    },
    {
      "epoch": 5.1593780517578126e-05,
      "step": 54100,
      "training_step_time": 7.656102657318115
    },
    {
      "epoch": 5.159473419189453e-05,
      "model_forward_time": 1.3177368640899658,
      "step": 54101
    },
    {
      "epoch": 5.159473419189453e-05,
      "step": 54101,
      "training_step_time": 8.927343845367432
    },
    {
      "epoch": 5.1595687866210934e-05,
      "model_forward_time": 1.6973402500152588,
      "step": 54102
    },
    {
      "epoch": 5.1595687866210934e-05,
      "step": 54102,
      "training_step_time": 8.714824199676514
    },
    {
      "epoch": 5.1596641540527345e-05,
      "model_forward_time": 1.7256455421447754,
      "step": 54103
    },
    {
      "epoch": 5.1596641540527345e-05,
      "step": 54103,
      "training_step_time": 7.754357576370239
    },
    {
      "epoch": 5.159759521484375e-05,
      "model_forward_time": 2.233988046646118,
      "step": 54104
    },
    {
      "epoch": 5.159759521484375e-05,
      "step": 54104,
      "training_step_time": 9.145945310592651
    },
    {
      "epoch": 5.159854888916015e-05,
      "model_forward_time": 1.6965315341949463,
      "step": 54105
    },
    {
      "epoch": 5.159854888916015e-05,
      "step": 54105,
      "training_step_time": 7.61990761756897
    },
    {
      "epoch": 5.1599502563476564e-05,
      "model_forward_time": 1.7000398635864258,
      "step": 54106
    },
    {
      "epoch": 5.1599502563476564e-05,
      "step": 54106,
      "training_step_time": 7.789381265640259
    },
    {
      "epoch": 5.160045623779297e-05,
      "model_forward_time": 1.6953907012939453,
      "step": 54107
    },
    {
      "epoch": 5.160045623779297e-05,
      "step": 54107,
      "training_step_time": 7.451261281967163
    },
    {
      "epoch": 5.160140991210937e-05,
      "model_forward_time": 1.6928691864013672,
      "step": 54108
    },
    {
      "epoch": 5.160140991210937e-05,
      "step": 54108,
      "training_step_time": 6.7956883907318115
    },
    {
      "epoch": 5.160236358642578e-05,
      "model_forward_time": 1.6969521045684814,
      "step": 54109
    },
    {
      "epoch": 5.160236358642578e-05,
      "step": 54109,
      "training_step_time": 7.9074907302856445
    },
    {
      "epoch": 5.160331726074219e-05,
      "grad_norm": 0.11466234177350998,
      "learning_rate": 4.734082223815182e-06,
      "loss": 0.0514,
      "step": 54110
    },
    {
      "epoch": 5.160331726074219e-05,
      "model_forward_time": 1.6926884651184082,
      "step": 54110
    },
    {
      "dynamics_loss_avg": 0.05715671665966511,
      "epoch": 5.160331726074219e-05,
      "step": 54110
    },
    {
      "action_loss_avg": 0.0014145880297292024,
      "epoch": 5.160331726074219e-05,
      "step": 54110
    },
    {
      "epoch": 5.160331726074219e-05,
      "step": 54110,
      "training_step_time": 8.673633575439453
    },
    {
      "epoch": 5.160427093505859e-05,
      "model_forward_time": 1.6963386535644531,
      "step": 54111
    },
    {
      "epoch": 5.160427093505859e-05,
      "step": 54111,
      "training_step_time": 7.62511134147644
    },
    {
      "epoch": 5.1605224609375e-05,
      "model_forward_time": 1.7168869972229004,
      "step": 54112
    },
    {
      "epoch": 5.1605224609375e-05,
      "step": 54112,
      "training_step_time": 7.766285419464111
    },
    {
      "epoch": 5.1606178283691406e-05,
      "model_forward_time": 2.2410337924957275,
      "step": 54113
    },
    {
      "epoch": 5.1606178283691406e-05,
      "step": 54113,
      "training_step_time": 8.235451936721802
    },
    {
      "epoch": 5.160713195800781e-05,
      "model_forward_time": 1.6940016746520996,
      "step": 54114
    },
    {
      "epoch": 5.160713195800781e-05,
      "step": 54114,
      "training_step_time": 8.62024736404419
    },
    {
      "epoch": 5.160808563232422e-05,
      "model_forward_time": 1.6927943229675293,
      "step": 54115
    },
    {
      "epoch": 5.160808563232422e-05,
      "step": 54115,
      "training_step_time": 7.755110502243042
    },
    {
      "epoch": 5.1609039306640625e-05,
      "model_forward_time": 1.693809986114502,
      "step": 54116
    },
    {
      "epoch": 5.1609039306640625e-05,
      "step": 54116,
      "training_step_time": 7.623188018798828
    },
    {
      "epoch": 5.160999298095703e-05,
      "model_forward_time": 1.6947567462921143,
      "step": 54117
    },
    {
      "epoch": 5.160999298095703e-05,
      "step": 54117,
      "training_step_time": 7.109926700592041
    },
    {
      "epoch": 5.161094665527344e-05,
      "model_forward_time": 1.7002692222595215,
      "step": 54118
    },
    {
      "epoch": 5.161094665527344e-05,
      "step": 54118,
      "training_step_time": 7.635185480117798
    },
    {
      "epoch": 5.1611900329589844e-05,
      "model_forward_time": 1.686337947845459,
      "step": 54119
    },
    {
      "epoch": 5.1611900329589844e-05,
      "step": 54119,
      "training_step_time": 7.5941784381866455
    },
    {
      "epoch": 5.161285400390625e-05,
      "grad_norm": 0.12024765461683273,
      "learning_rate": 4.732431108635544e-06,
      "loss": 0.0586,
      "step": 54120
    },
    {
      "epoch": 5.161285400390625e-05,
      "model_forward_time": 1.7041852474212646,
      "step": 54120
    },
    {
      "dynamics_loss_avg": 0.05766057223081589,
      "epoch": 5.161285400390625e-05,
      "step": 54120
    },
    {
      "action_loss_avg": 0.0020975010236725213,
      "epoch": 5.161285400390625e-05,
      "step": 54120
    },
    {
      "epoch": 5.161285400390625e-05,
      "step": 54120,
      "training_step_time": 8.285393714904785
    },
    {
      "epoch": 5.161380767822266e-05,
      "model_forward_time": 1.7394871711730957,
      "step": 54121
    },
    {
      "epoch": 5.161380767822266e-05,
      "step": 54121,
      "training_step_time": 8.567839622497559
    },
    {
      "epoch": 5.161476135253906e-05,
      "model_forward_time": 2.2894513607025146,
      "step": 54122
    },
    {
      "epoch": 5.161476135253906e-05,
      "step": 54122,
      "training_step_time": 7.697726726531982
    },
    {
      "epoch": 5.1615715026855466e-05,
      "model_forward_time": 1.7060961723327637,
      "step": 54123
    },
    {
      "epoch": 5.1615715026855466e-05,
      "step": 54123,
      "training_step_time": 9.109538316726685
    },
    {
      "epoch": 5.161666870117188e-05,
      "model_forward_time": 1.704164981842041,
      "step": 54124
    },
    {
      "epoch": 5.161666870117188e-05,
      "step": 54124,
      "training_step_time": 7.7069480419158936
    },
    {
      "epoch": 5.161762237548828e-05,
      "model_forward_time": 1.701737880706787,
      "step": 54125
    },
    {
      "epoch": 5.161762237548828e-05,
      "step": 54125,
      "training_step_time": 7.613078594207764
    },
    {
      "epoch": 5.1618576049804685e-05,
      "model_forward_time": 1.6945548057556152,
      "step": 54126
    },
    {
      "epoch": 5.1618576049804685e-05,
      "step": 54126,
      "training_step_time": 7.423657178878784
    },
    {
      "epoch": 5.1619529724121096e-05,
      "model_forward_time": 1.703780174255371,
      "step": 54127
    },
    {
      "epoch": 5.1619529724121096e-05,
      "step": 54127,
      "training_step_time": 7.534315347671509
    },
    {
      "epoch": 5.16204833984375e-05,
      "model_forward_time": 1.6967697143554688,
      "step": 54128
    },
    {
      "epoch": 5.16204833984375e-05,
      "step": 54128,
      "training_step_time": 7.538677930831909
    },
    {
      "epoch": 5.1621437072753904e-05,
      "model_forward_time": 1.701591968536377,
      "step": 54129
    },
    {
      "epoch": 5.1621437072753904e-05,
      "step": 54129,
      "training_step_time": 8.901879072189331
    },
    {
      "epoch": 5.1622390747070315e-05,
      "grad_norm": 0.08999618887901306,
      "learning_rate": 4.730780022716837e-06,
      "loss": 0.0595,
      "step": 54130
    },
    {
      "epoch": 5.1622390747070315e-05,
      "model_forward_time": 1.7062697410583496,
      "step": 54130
    },
    {
      "dynamics_loss_avg": 0.05982136968523264,
      "epoch": 5.1622390747070315e-05,
      "step": 54130
    },
    {
      "action_loss_avg": 0.0014612640719860793,
      "epoch": 5.1622390747070315e-05,
      "step": 54130
    },
    {
      "epoch": 5.1622390747070315e-05,
      "step": 54130,
      "training_step_time": 9.094993352890015
    },
    {
      "epoch": 5.162334442138672e-05,
      "model_forward_time": 2.2591731548309326,
      "step": 54131
    },
    {
      "epoch": 5.162334442138672e-05,
      "step": 54131,
      "training_step_time": 8.56497311592102
    },
    {
      "epoch": 5.162429809570312e-05,
      "model_forward_time": 1.6973073482513428,
      "step": 54132
    },
    {
      "epoch": 5.162429809570312e-05,
      "step": 54132,
      "training_step_time": 10.043539047241211
    },
    {
      "epoch": 5.1625251770019534e-05,
      "model_forward_time": 1.7039978504180908,
      "step": 54133
    },
    {
      "epoch": 5.1625251770019534e-05,
      "step": 54133,
      "training_step_time": 8.282851457595825
    },
    {
      "epoch": 5.162620544433594e-05,
      "model_forward_time": 1.7028658390045166,
      "step": 54134
    },
    {
      "epoch": 5.162620544433594e-05,
      "step": 54134,
      "training_step_time": 7.842562675476074
    },
    {
      "epoch": 5.162715911865234e-05,
      "model_forward_time": 1.7021939754486084,
      "step": 54135
    },
    {
      "epoch": 5.162715911865234e-05,
      "step": 54135,
      "training_step_time": 7.495707988739014
    },
    {
      "epoch": 5.162811279296875e-05,
      "model_forward_time": 1.7002856731414795,
      "step": 54136
    },
    {
      "epoch": 5.162811279296875e-05,
      "step": 54136,
      "training_step_time": 6.8908326625823975
    },
    {
      "epoch": 5.162906646728516e-05,
      "model_forward_time": 1.697871208190918,
      "step": 54137
    },
    {
      "epoch": 5.162906646728516e-05,
      "step": 54137,
      "training_step_time": 7.636152744293213
    },
    {
      "epoch": 5.163002014160156e-05,
      "model_forward_time": 1.7017414569854736,
      "step": 54138
    },
    {
      "epoch": 5.163002014160156e-05,
      "step": 54138,
      "training_step_time": 8.041529178619385
    },
    {
      "epoch": 5.163097381591797e-05,
      "model_forward_time": 1.7209923267364502,
      "step": 54139
    },
    {
      "epoch": 5.163097381591797e-05,
      "step": 54139,
      "training_step_time": 7.799072027206421
    },
    {
      "epoch": 5.1631927490234376e-05,
      "grad_norm": 0.12585145235061646,
      "learning_rate": 4.729128966239619e-06,
      "loss": 0.0575,
      "step": 54140
    },
    {
      "epoch": 5.1631927490234376e-05,
      "model_forward_time": 2.261601448059082,
      "step": 54140
    },
    {
      "dynamics_loss_avg": 0.04631195198744535,
      "epoch": 5.1631927490234376e-05,
      "step": 54140
    },
    {
      "action_loss_avg": 0.0011642510158708318,
      "epoch": 5.1631927490234376e-05,
      "step": 54140
    },
    {
      "epoch": 5.1631927490234376e-05,
      "step": 54140,
      "training_step_time": 7.846766710281372
    },
    {
      "epoch": 5.163288116455078e-05,
      "model_forward_time": 1.7029731273651123,
      "step": 54141
    },
    {
      "epoch": 5.163288116455078e-05,
      "step": 54141,
      "training_step_time": 10.747792959213257
    },
    {
      "epoch": 5.163383483886719e-05,
      "model_forward_time": 1.6965577602386475,
      "step": 54142
    },
    {
      "epoch": 5.163383483886719e-05,
      "step": 54142,
      "training_step_time": 8.542799234390259
    },
    {
      "epoch": 5.1634788513183594e-05,
      "model_forward_time": 1.7070038318634033,
      "step": 54143
    },
    {
      "epoch": 5.1634788513183594e-05,
      "step": 54143,
      "training_step_time": 9.523076295852661
    },
    {
      "epoch": 5.16357421875e-05,
      "model_forward_time": 1.699561357498169,
      "step": 54144
    },
    {
      "epoch": 5.16357421875e-05,
      "step": 54144,
      "training_step_time": 7.5276219844818115
    },
    {
      "epoch": 5.163669586181641e-05,
      "model_forward_time": 1.6988742351531982,
      "step": 54145
    },
    {
      "epoch": 5.163669586181641e-05,
      "step": 54145,
      "training_step_time": 6.906556606292725
    },
    {
      "epoch": 5.163764953613281e-05,
      "model_forward_time": 1.698981523513794,
      "step": 54146
    },
    {
      "epoch": 5.163764953613281e-05,
      "step": 54146,
      "training_step_time": 8.623310565948486
    },
    {
      "epoch": 5.163860321044922e-05,
      "model_forward_time": 1.698462963104248,
      "step": 54147
    },
    {
      "epoch": 5.163860321044922e-05,
      "step": 54147,
      "training_step_time": 8.063710451126099
    },
    {
      "epoch": 5.163955688476563e-05,
      "model_forward_time": 1.7248480319976807,
      "step": 54148
    },
    {
      "epoch": 5.163955688476563e-05,
      "step": 54148,
      "training_step_time": 8.702879905700684
    },
    {
      "epoch": 5.164051055908203e-05,
      "model_forward_time": 1.714425802230835,
      "step": 54149
    },
    {
      "epoch": 5.164051055908203e-05,
      "step": 54149,
      "training_step_time": 7.859143018722534
    },
    {
      "epoch": 5.1641464233398436e-05,
      "grad_norm": 0.14623242616653442,
      "learning_rate": 4.727477939384454e-06,
      "loss": 0.0572,
      "step": 54150
    },
    {
      "epoch": 5.1641464233398436e-05,
      "model_forward_time": 2.253124237060547,
      "step": 54150
    },
    {
      "dynamics_loss_avg": 0.052369114197790625,
      "epoch": 5.1641464233398436e-05,
      "step": 54150
    },
    {
      "action_loss_avg": 0.001643211458576843,
      "epoch": 5.1641464233398436e-05,
      "step": 54150
    },
    {
      "epoch": 5.1641464233398436e-05,
      "step": 54150,
      "training_step_time": 7.820951700210571
    },
    {
      "epoch": 5.164241790771485e-05,
      "model_forward_time": 1.6962330341339111,
      "step": 54151
    },
    {
      "epoch": 5.164241790771485e-05,
      "step": 54151,
      "training_step_time": 9.589595556259155
    },
    {
      "epoch": 5.164337158203125e-05,
      "model_forward_time": 1.7075226306915283,
      "step": 54152
    },
    {
      "epoch": 5.164337158203125e-05,
      "step": 54152,
      "training_step_time": 8.828399658203125
    },
    {
      "epoch": 5.1644325256347655e-05,
      "model_forward_time": 1.718200445175171,
      "step": 54153
    },
    {
      "epoch": 5.1644325256347655e-05,
      "step": 54153,
      "training_step_time": 8.161479949951172
    },
    {
      "epoch": 5.1645278930664066e-05,
      "model_forward_time": 1.716388463973999,
      "step": 54154
    },
    {
      "epoch": 5.1645278930664066e-05,
      "step": 54154,
      "training_step_time": 7.466386079788208
    },
    {
      "epoch": 5.164623260498047e-05,
      "model_forward_time": 1.723479986190796,
      "step": 54155
    },
    {
      "epoch": 5.164623260498047e-05,
      "step": 54155,
      "training_step_time": 8.505932569503784
    },
    {
      "epoch": 5.1647186279296874e-05,
      "model_forward_time": 1.7335503101348877,
      "step": 54156
    },
    {
      "epoch": 5.1647186279296874e-05,
      "step": 54156,
      "training_step_time": 8.008199691772461
    },
    {
      "epoch": 5.164813995361328e-05,
      "model_forward_time": 1.7849841117858887,
      "step": 54157
    },
    {
      "epoch": 5.164813995361328e-05,
      "step": 54157,
      "training_step_time": 7.743795394897461
    },
    {
      "epoch": 5.164909362792969e-05,
      "model_forward_time": 1.706273078918457,
      "step": 54158
    },
    {
      "epoch": 5.164909362792969e-05,
      "step": 54158,
      "training_step_time": 8.277758121490479
    },
    {
      "epoch": 5.165004730224609e-05,
      "model_forward_time": 2.2810254096984863,
      "step": 54159
    },
    {
      "epoch": 5.165004730224609e-05,
      "step": 54159,
      "training_step_time": 8.739586114883423
    },
    {
      "epoch": 5.16510009765625e-05,
      "grad_norm": 0.1320691853761673,
      "learning_rate": 4.725826942331891e-06,
      "loss": 0.0596,
      "step": 54160
    },
    {
      "epoch": 5.16510009765625e-05,
      "model_forward_time": 1.6966040134429932,
      "step": 54160
    },
    {
      "dynamics_loss_avg": 0.03936080988496542,
      "epoch": 5.16510009765625e-05,
      "step": 54160
    },
    {
      "action_loss_avg": 0.001240584655897692,
      "epoch": 5.16510009765625e-05,
      "step": 54160
    },
    {
      "epoch": 5.16510009765625e-05,
      "step": 54160,
      "training_step_time": 8.724401712417603
    },
    {
      "epoch": 5.165195465087891e-05,
      "model_forward_time": 1.7121663093566895,
      "step": 54161
    },
    {
      "epoch": 5.165195465087891e-05,
      "step": 54161,
      "training_step_time": 7.702757358551025
    },
    {
      "epoch": 5.165290832519531e-05,
      "model_forward_time": 1.7130427360534668,
      "step": 54162
    },
    {
      "epoch": 5.165290832519531e-05,
      "step": 54162,
      "training_step_time": 7.65503191947937
    },
    {
      "epoch": 5.1653861999511716e-05,
      "model_forward_time": 1.3419082164764404,
      "step": 54163
    },
    {
      "epoch": 5.1653861999511716e-05,
      "step": 54163,
      "training_step_time": 7.4368672370910645
    },
    {
      "epoch": 5.1654815673828126e-05,
      "model_forward_time": 1.7194702625274658,
      "step": 54164
    },
    {
      "epoch": 5.1654815673828126e-05,
      "step": 54164,
      "training_step_time": 7.573190689086914
    },
    {
      "epoch": 5.165576934814453e-05,
      "model_forward_time": 1.7145476341247559,
      "step": 54165
    },
    {
      "epoch": 5.165576934814453e-05,
      "step": 54165,
      "training_step_time": 9.272762298583984
    },
    {
      "epoch": 5.1656723022460934e-05,
      "model_forward_time": 1.7735936641693115,
      "step": 54166
    },
    {
      "epoch": 5.1656723022460934e-05,
      "step": 54166,
      "training_step_time": 8.788810968399048
    },
    {
      "epoch": 5.1657676696777345e-05,
      "model_forward_time": 1.712705373764038,
      "step": 54167
    },
    {
      "epoch": 5.1657676696777345e-05,
      "step": 54167,
      "training_step_time": 8.563673496246338
    },
    {
      "epoch": 5.165863037109375e-05,
      "model_forward_time": 2.281327486038208,
      "step": 54168
    },
    {
      "epoch": 5.165863037109375e-05,
      "step": 54168,
      "training_step_time": 8.458189249038696
    },
    {
      "epoch": 5.165958404541015e-05,
      "model_forward_time": 1.3489043712615967,
      "step": 54169
    },
    {
      "epoch": 5.165958404541015e-05,
      "step": 54169,
      "training_step_time": 8.67357587814331
    },
    {
      "epoch": 5.1660537719726564e-05,
      "grad_norm": 0.10554435104131699,
      "learning_rate": 4.724175975262481e-06,
      "loss": 0.0587,
      "step": 54170
    },
    {
      "epoch": 5.1660537719726564e-05,
      "model_forward_time": 1.7244057655334473,
      "step": 54170
    },
    {
      "dynamics_loss_avg": 0.05004062503576279,
      "epoch": 5.1660537719726564e-05,
      "step": 54170
    },
    {
      "action_loss_avg": 0.0015643224585801363,
      "epoch": 5.1660537719726564e-05,
      "step": 54170
    },
    {
      "epoch": 5.1660537719726564e-05,
      "step": 54170,
      "training_step_time": 7.820086240768433
    },
    {
      "epoch": 5.166149139404297e-05,
      "model_forward_time": 1.7169816493988037,
      "step": 54171
    },
    {
      "epoch": 5.166149139404297e-05,
      "step": 54171,
      "training_step_time": 7.63936185836792
    },
    {
      "epoch": 5.166244506835937e-05,
      "model_forward_time": 1.7297499179840088,
      "step": 54172
    },
    {
      "epoch": 5.166244506835937e-05,
      "step": 54172,
      "training_step_time": 7.494145631790161
    },
    {
      "epoch": 5.166339874267578e-05,
      "model_forward_time": 1.722168207168579,
      "step": 54173
    },
    {
      "epoch": 5.166339874267578e-05,
      "step": 54173,
      "training_step_time": 6.859072685241699
    },
    {
      "epoch": 5.166435241699219e-05,
      "model_forward_time": 1.719292163848877,
      "step": 54174
    },
    {
      "epoch": 5.166435241699219e-05,
      "step": 54174,
      "training_step_time": 7.6757893562316895
    },
    {
      "epoch": 5.166530609130859e-05,
      "model_forward_time": 1.7910435199737549,
      "step": 54175
    },
    {
      "epoch": 5.166530609130859e-05,
      "step": 54175,
      "training_step_time": 7.981493234634399
    },
    {
      "epoch": 5.1666259765625e-05,
      "model_forward_time": 1.7403342723846436,
      "step": 54176
    },
    {
      "epoch": 5.1666259765625e-05,
      "step": 54176,
      "training_step_time": 7.612898111343384
    },
    {
      "epoch": 5.1667213439941406e-05,
      "model_forward_time": 2.2975213527679443,
      "step": 54177
    },
    {
      "epoch": 5.1667213439941406e-05,
      "step": 54177,
      "training_step_time": 9.070544242858887
    },
    {
      "epoch": 5.166816711425781e-05,
      "model_forward_time": 1.7400338649749756,
      "step": 54178
    },
    {
      "epoch": 5.166816711425781e-05,
      "step": 54178,
      "training_step_time": 7.704943895339966
    },
    {
      "epoch": 5.166912078857422e-05,
      "model_forward_time": 1.7370543479919434,
      "step": 54179
    },
    {
      "epoch": 5.166912078857422e-05,
      "step": 54179,
      "training_step_time": 8.55403757095337
    },
    {
      "epoch": 5.1670074462890625e-05,
      "grad_norm": 0.11174672096967697,
      "learning_rate": 4.7225250383567725e-06,
      "loss": 0.0584,
      "step": 54180
    },
    {
      "epoch": 5.1670074462890625e-05,
      "model_forward_time": 1.7194080352783203,
      "step": 54180
    },
    {
      "dynamics_loss_avg": 0.05135247204452753,
      "epoch": 5.1670074462890625e-05,
      "step": 54180
    },
    {
      "action_loss_avg": 0.0011208761803572997,
      "epoch": 5.1670074462890625e-05,
      "step": 54180
    },
    {
      "epoch": 5.1670074462890625e-05,
      "step": 54180,
      "training_step_time": 8.55081558227539
    },
    {
      "epoch": 5.167102813720703e-05,
      "model_forward_time": 1.7369670867919922,
      "step": 54181
    },
    {
      "epoch": 5.167102813720703e-05,
      "step": 54181,
      "training_step_time": 7.442233562469482
    },
    {
      "epoch": 5.167198181152344e-05,
      "model_forward_time": 1.7401320934295654,
      "step": 54182
    },
    {
      "epoch": 5.167198181152344e-05,
      "step": 54182,
      "training_step_time": 7.444406270980835
    },
    {
      "epoch": 5.1672935485839844e-05,
      "model_forward_time": 1.7361018657684326,
      "step": 54183
    },
    {
      "epoch": 5.1672935485839844e-05,
      "step": 54183,
      "training_step_time": 7.678267955780029
    },
    {
      "epoch": 5.167388916015625e-05,
      "model_forward_time": 1.4277098178863525,
      "step": 54184
    },
    {
      "epoch": 5.167388916015625e-05,
      "step": 54184,
      "training_step_time": 7.937098503112793
    },
    {
      "epoch": 5.167484283447266e-05,
      "model_forward_time": 1.745382308959961,
      "step": 54185
    },
    {
      "epoch": 5.167484283447266e-05,
      "step": 54185,
      "training_step_time": 8.641634941101074
    },
    {
      "epoch": 5.167579650878906e-05,
      "model_forward_time": 1.7494421005249023,
      "step": 54186
    },
    {
      "epoch": 5.167579650878906e-05,
      "step": 54186,
      "training_step_time": 7.754140138626099
    },
    {
      "epoch": 5.1676750183105466e-05,
      "model_forward_time": 2.3028173446655273,
      "step": 54187
    },
    {
      "epoch": 5.1676750183105466e-05,
      "step": 54187,
      "training_step_time": 8.462783098220825
    },
    {
      "epoch": 5.167770385742188e-05,
      "model_forward_time": 1.746736764907837,
      "step": 54188
    },
    {
      "epoch": 5.167770385742188e-05,
      "step": 54188,
      "training_step_time": 7.8049538135528564
    },
    {
      "epoch": 5.167865753173828e-05,
      "model_forward_time": 1.7450098991394043,
      "step": 54189
    },
    {
      "epoch": 5.167865753173828e-05,
      "step": 54189,
      "training_step_time": 7.627704858779907
    },
    {
      "epoch": 5.1679611206054685e-05,
      "grad_norm": 0.11074122786521912,
      "learning_rate": 4.7208741317953075e-06,
      "loss": 0.0614,
      "step": 54190
    },
    {
      "epoch": 5.1679611206054685e-05,
      "model_forward_time": 1.7380330562591553,
      "step": 54190
    },
    {
      "dynamics_loss_avg": 0.0531156687065959,
      "epoch": 5.1679611206054685e-05,
      "step": 54190
    },
    {
      "action_loss_avg": 0.0016764369269367308,
      "epoch": 5.1679611206054685e-05,
      "step": 54190
    },
    {
      "epoch": 5.1679611206054685e-05,
      "step": 54190,
      "training_step_time": 7.446789741516113
    },
    {
      "epoch": 5.1680564880371096e-05,
      "model_forward_time": 1.7455787658691406,
      "step": 54191
    },
    {
      "epoch": 5.1680564880371096e-05,
      "step": 54191,
      "training_step_time": 7.0362067222595215
    },
    {
      "epoch": 5.16815185546875e-05,
      "model_forward_time": 1.7413866519927979,
      "step": 54192
    },
    {
      "epoch": 5.16815185546875e-05,
      "step": 54192,
      "training_step_time": 7.7834532260894775
    },
    {
      "epoch": 5.1682472229003904e-05,
      "model_forward_time": 1.777942419052124,
      "step": 54193
    },
    {
      "epoch": 5.1682472229003904e-05,
      "step": 54193,
      "training_step_time": 7.893331050872803
    },
    {
      "epoch": 5.1683425903320315e-05,
      "model_forward_time": 1.7279977798461914,
      "step": 54194
    },
    {
      "epoch": 5.1683425903320315e-05,
      "step": 54194,
      "training_step_time": 8.769827365875244
    },
    {
      "epoch": 5.168437957763672e-05,
      "model_forward_time": 1.7364568710327148,
      "step": 54195
    },
    {
      "epoch": 5.168437957763672e-05,
      "step": 54195,
      "training_step_time": 8.600985765457153
    },
    {
      "epoch": 5.168533325195312e-05,
      "model_forward_time": 2.287522077560425,
      "step": 54196
    },
    {
      "epoch": 5.168533325195312e-05,
      "step": 54196,
      "training_step_time": 8.243318796157837
    },
    {
      "epoch": 5.1686286926269534e-05,
      "model_forward_time": 1.7263123989105225,
      "step": 54197
    },
    {
      "epoch": 5.1686286926269534e-05,
      "step": 54197,
      "training_step_time": 9.539287567138672
    },
    {
      "epoch": 5.168724060058594e-05,
      "model_forward_time": 1.7131168842315674,
      "step": 54198
    },
    {
      "epoch": 5.168724060058594e-05,
      "step": 54198,
      "training_step_time": 8.716209411621094
    },
    {
      "epoch": 5.168819427490234e-05,
      "model_forward_time": 1.7253975868225098,
      "step": 54199
    },
    {
      "epoch": 5.168819427490234e-05,
      "step": 54199,
      "training_step_time": 7.631420373916626
    },
    {
      "epoch": 5.168914794921875e-05,
      "grad_norm": 0.10920605808496475,
      "learning_rate": 4.719223255758626e-06,
      "loss": 0.0603,
      "step": 54200
    },
    {
      "epoch": 5.168914794921875e-05,
      "model_forward_time": 1.7321946620941162,
      "step": 54200
    },
    {
      "dynamics_loss_avg": 0.042660395987331866,
      "epoch": 5.168914794921875e-05,
      "step": 54200
    },
    {
      "action_loss_avg": 0.001541382775758393,
      "epoch": 5.168914794921875e-05,
      "step": 54200
    },
    {
      "epoch": 5.168914794921875e-05,
      "step": 54200,
      "training_step_time": 7.4744062423706055
    },
    {
      "epoch": 5.169010162353516e-05,
      "model_forward_time": 1.7376577854156494,
      "step": 54201
    },
    {
      "epoch": 5.169010162353516e-05,
      "step": 54201,
      "training_step_time": 7.480963230133057
    },
    {
      "epoch": 5.169105529785156e-05,
      "model_forward_time": 1.7674953937530518,
      "step": 54202
    },
    {
      "epoch": 5.169105529785156e-05,
      "step": 54202,
      "training_step_time": 7.7261128425598145
    },
    {
      "epoch": 5.169200897216797e-05,
      "model_forward_time": 1.72418212890625,
      "step": 54203
    },
    {
      "epoch": 5.169200897216797e-05,
      "step": 54203,
      "training_step_time": 7.872247695922852
    },
    {
      "epoch": 5.1692962646484376e-05,
      "model_forward_time": 1.7376518249511719,
      "step": 54204
    },
    {
      "epoch": 5.1692962646484376e-05,
      "step": 54204,
      "training_step_time": 8.490843057632446
    },
    {
      "epoch": 5.169391632080078e-05,
      "model_forward_time": 2.3009543418884277,
      "step": 54205
    },
    {
      "epoch": 5.169391632080078e-05,
      "step": 54205,
      "training_step_time": 9.37166166305542
    },
    {
      "epoch": 5.169486999511719e-05,
      "model_forward_time": 1.7269577980041504,
      "step": 54206
    },
    {
      "epoch": 5.169486999511719e-05,
      "step": 54206,
      "training_step_time": 7.981748819351196
    },
    {
      "epoch": 5.1695823669433594e-05,
      "model_forward_time": 1.7332019805908203,
      "step": 54207
    },
    {
      "epoch": 5.1695823669433594e-05,
      "step": 54207,
      "training_step_time": 7.827755928039551
    },
    {
      "epoch": 5.169677734375e-05,
      "model_forward_time": 1.733386516571045,
      "step": 54208
    },
    {
      "epoch": 5.169677734375e-05,
      "step": 54208,
      "training_step_time": 7.5252697467803955
    },
    {
      "epoch": 5.169773101806641e-05,
      "model_forward_time": 1.7337136268615723,
      "step": 54209
    },
    {
      "epoch": 5.169773101806641e-05,
      "step": 54209,
      "training_step_time": 7.436628580093384
    },
    {
      "epoch": 5.169868469238281e-05,
      "grad_norm": 0.12266308069229126,
      "learning_rate": 4.71757241042727e-06,
      "loss": 0.0562,
      "step": 54210
    },
    {
      "epoch": 5.169868469238281e-05,
      "model_forward_time": 1.7293193340301514,
      "step": 54210
    },
    {
      "dynamics_loss_avg": 0.051802316680550575,
      "epoch": 5.169868469238281e-05,
      "step": 54210
    },
    {
      "action_loss_avg": 0.0015339473087806255,
      "epoch": 5.169868469238281e-05,
      "step": 54210
    },
    {
      "epoch": 5.169868469238281e-05,
      "step": 54210,
      "training_step_time": 7.376940488815308
    },
    {
      "epoch": 5.169963836669922e-05,
      "model_forward_time": 1.7436351776123047,
      "step": 54211
    },
    {
      "epoch": 5.169963836669922e-05,
      "step": 54211,
      "training_step_time": 7.701625823974609
    },
    {
      "epoch": 5.170059204101563e-05,
      "model_forward_time": 1.720573902130127,
      "step": 54212
    },
    {
      "epoch": 5.170059204101563e-05,
      "step": 54212,
      "training_step_time": 7.942439556121826
    },
    {
      "epoch": 5.170154571533203e-05,
      "model_forward_time": 1.7258214950561523,
      "step": 54213
    },
    {
      "epoch": 5.170154571533203e-05,
      "step": 54213,
      "training_step_time": 7.710955858230591
    },
    {
      "epoch": 5.1702499389648436e-05,
      "model_forward_time": 2.3029584884643555,
      "step": 54214
    },
    {
      "epoch": 5.1702499389648436e-05,
      "step": 54214,
      "training_step_time": 9.250669240951538
    },
    {
      "epoch": 5.170345306396485e-05,
      "model_forward_time": 1.7280795574188232,
      "step": 54215
    },
    {
      "epoch": 5.170345306396485e-05,
      "step": 54215,
      "training_step_time": 10.619338274002075
    },
    {
      "epoch": 5.170440673828125e-05,
      "model_forward_time": 1.736170768737793,
      "step": 54216
    },
    {
      "epoch": 5.170440673828125e-05,
      "step": 54216,
      "training_step_time": 8.5737624168396
    },
    {
      "epoch": 5.1705360412597655e-05,
      "model_forward_time": 1.7382190227508545,
      "step": 54217
    },
    {
      "epoch": 5.1705360412597655e-05,
      "step": 54217,
      "training_step_time": 7.537467956542969
    },
    {
      "epoch": 5.1706314086914066e-05,
      "model_forward_time": 1.730616807937622,
      "step": 54218
    },
    {
      "epoch": 5.1706314086914066e-05,
      "step": 54218,
      "training_step_time": 7.443668365478516
    },
    {
      "epoch": 5.170726776123047e-05,
      "model_forward_time": 1.7279932498931885,
      "step": 54219
    },
    {
      "epoch": 5.170726776123047e-05,
      "step": 54219,
      "training_step_time": 7.4136810302734375
    },
    {
      "epoch": 5.1708221435546874e-05,
      "grad_norm": 0.10438592731952667,
      "learning_rate": 4.71592159598177e-06,
      "loss": 0.0571,
      "step": 54220
    },
    {
      "epoch": 5.1708221435546874e-05,
      "model_forward_time": 1.731421709060669,
      "step": 54220
    },
    {
      "dynamics_loss_avg": 0.04931331854313612,
      "epoch": 5.1708221435546874e-05,
      "step": 54220
    },
    {
      "action_loss_avg": 0.0011248986556893215,
      "epoch": 5.1708221435546874e-05,
      "step": 54220
    },
    {
      "epoch": 5.1708221435546874e-05,
      "step": 54220,
      "training_step_time": 8.709718704223633
    },
    {
      "epoch": 5.170917510986328e-05,
      "model_forward_time": 1.7261443138122559,
      "step": 54221
    },
    {
      "epoch": 5.170917510986328e-05,
      "step": 54221,
      "training_step_time": 7.963715553283691
    },
    {
      "epoch": 5.171012878417969e-05,
      "model_forward_time": 1.7313816547393799,
      "step": 54222
    },
    {
      "epoch": 5.171012878417969e-05,
      "step": 54222,
      "training_step_time": 7.666684865951538
    },
    {
      "epoch": 5.171108245849609e-05,
      "model_forward_time": 1.7371089458465576,
      "step": 54223
    },
    {
      "epoch": 5.171108245849609e-05,
      "step": 54223,
      "training_step_time": 7.795062780380249
    },
    {
      "epoch": 5.17120361328125e-05,
      "model_forward_time": 1.725125789642334,
      "step": 54224
    },
    {
      "epoch": 5.17120361328125e-05,
      "step": 54224,
      "training_step_time": 8.358953714370728
    },
    {
      "epoch": 5.171298980712891e-05,
      "model_forward_time": 1.7274227142333984,
      "step": 54225
    },
    {
      "epoch": 5.171298980712891e-05,
      "step": 54225,
      "training_step_time": 7.64979887008667
    },
    {
      "epoch": 5.171394348144531e-05,
      "model_forward_time": 1.7266833782196045,
      "step": 54226
    },
    {
      "epoch": 5.171394348144531e-05,
      "step": 54226,
      "training_step_time": 8.384072542190552
    },
    {
      "epoch": 5.1714897155761716e-05,
      "model_forward_time": 1.737208366394043,
      "step": 54227
    },
    {
      "epoch": 5.1714897155761716e-05,
      "step": 54227,
      "training_step_time": 7.926456928253174
    },
    {
      "epoch": 5.1715850830078126e-05,
      "model_forward_time": 1.7479593753814697,
      "step": 54228
    },
    {
      "epoch": 5.1715850830078126e-05,
      "step": 54228,
      "training_step_time": 7.4380269050598145
    },
    {
      "epoch": 5.171680450439453e-05,
      "model_forward_time": 1.7348191738128662,
      "step": 54229
    },
    {
      "epoch": 5.171680450439453e-05,
      "step": 54229,
      "training_step_time": 7.6056084632873535
    },
    {
      "epoch": 5.1717758178710935e-05,
      "grad_norm": 0.11757022142410278,
      "learning_rate": 4.714270812602657e-06,
      "loss": 0.0563,
      "step": 54230
    },
    {
      "epoch": 5.1717758178710935e-05,
      "model_forward_time": 1.7898013591766357,
      "step": 54230
    },
    {
      "dynamics_loss_avg": 0.06359846815466881,
      "epoch": 5.1717758178710935e-05,
      "step": 54230
    },
    {
      "action_loss_avg": 0.002469664439558983,
      "epoch": 5.1717758178710935e-05,
      "step": 54230
    },
    {
      "epoch": 5.1717758178710935e-05,
      "step": 54230,
      "training_step_time": 7.783239364624023
    },
    {
      "epoch": 5.1718711853027345e-05,
      "model_forward_time": 1.7479054927825928,
      "step": 54231
    },
    {
      "epoch": 5.1718711853027345e-05,
      "step": 54231,
      "training_step_time": 7.529288053512573
    },
    {
      "epoch": 5.171966552734375e-05,
      "model_forward_time": 1.7284197807312012,
      "step": 54232
    },
    {
      "epoch": 5.171966552734375e-05,
      "step": 54232,
      "training_step_time": 7.610356569290161
    },
    {
      "epoch": 5.172061920166015e-05,
      "model_forward_time": 2.2970001697540283,
      "step": 54233
    },
    {
      "epoch": 5.172061920166015e-05,
      "step": 54233,
      "training_step_time": 8.480041980743408
    },
    {
      "epoch": 5.1721572875976564e-05,
      "model_forward_time": 1.738964319229126,
      "step": 54234
    },
    {
      "epoch": 5.1721572875976564e-05,
      "step": 54234,
      "training_step_time": 7.6977338790893555
    },
    {
      "epoch": 5.172252655029297e-05,
      "model_forward_time": 1.7335319519042969,
      "step": 54235
    },
    {
      "epoch": 5.172252655029297e-05,
      "step": 54235,
      "training_step_time": 7.602903842926025
    },
    {
      "epoch": 5.172348022460937e-05,
      "model_forward_time": 1.733949899673462,
      "step": 54236
    },
    {
      "epoch": 5.172348022460937e-05,
      "step": 54236,
      "training_step_time": 7.595146179199219
    },
    {
      "epoch": 5.172443389892578e-05,
      "model_forward_time": 1.7352092266082764,
      "step": 54237
    },
    {
      "epoch": 5.172443389892578e-05,
      "step": 54237,
      "training_step_time": 7.378347873687744
    },
    {
      "epoch": 5.172538757324219e-05,
      "model_forward_time": 1.7368273735046387,
      "step": 54238
    },
    {
      "epoch": 5.172538757324219e-05,
      "step": 54238,
      "training_step_time": 8.5453200340271
    },
    {
      "epoch": 5.172634124755859e-05,
      "model_forward_time": 1.795165777206421,
      "step": 54239
    },
    {
      "epoch": 5.172634124755859e-05,
      "step": 54239,
      "training_step_time": 7.73000693321228
    },
    {
      "epoch": 5.1727294921875e-05,
      "grad_norm": 0.12883958220481873,
      "learning_rate": 4.712620060470459e-06,
      "loss": 0.0551,
      "step": 54240
    },
    {
      "epoch": 5.1727294921875e-05,
      "model_forward_time": 1.7399799823760986,
      "step": 54240
    },
    {
      "dynamics_loss_avg": 0.06429297104477882,
      "epoch": 5.1727294921875e-05,
      "step": 54240
    },
    {
      "action_loss_avg": 0.0015046101820189507,
      "epoch": 5.1727294921875e-05,
      "step": 54240
    },
    {
      "epoch": 5.1727294921875e-05,
      "step": 54240,
      "training_step_time": 7.934632778167725
    },
    {
      "epoch": 5.1728248596191406e-05,
      "model_forward_time": 1.7224087715148926,
      "step": 54241
    },
    {
      "epoch": 5.1728248596191406e-05,
      "step": 54241,
      "training_step_time": 8.401094675064087
    },
    {
      "epoch": 5.172920227050781e-05,
      "model_forward_time": 2.929001569747925,
      "step": 54242
    },
    {
      "epoch": 5.172920227050781e-05,
      "step": 54242,
      "training_step_time": 8.29586124420166
    },
    {
      "epoch": 5.173015594482422e-05,
      "model_forward_time": 1.7246320247650146,
      "step": 54243
    },
    {
      "epoch": 5.173015594482422e-05,
      "step": 54243,
      "training_step_time": 8.561362981796265
    },
    {
      "epoch": 5.1731109619140625e-05,
      "model_forward_time": 1.7869458198547363,
      "step": 54244
    },
    {
      "epoch": 5.1731109619140625e-05,
      "step": 54244,
      "training_step_time": 7.58773398399353
    },
    {
      "epoch": 5.173206329345703e-05,
      "model_forward_time": 1.7382335662841797,
      "step": 54245
    },
    {
      "epoch": 5.173206329345703e-05,
      "step": 54245,
      "training_step_time": 7.5198259353637695
    },
    {
      "epoch": 5.173301696777344e-05,
      "model_forward_time": 1.7360808849334717,
      "step": 54246
    },
    {
      "epoch": 5.173301696777344e-05,
      "step": 54246,
      "training_step_time": 8.082399129867554
    },
    {
      "epoch": 5.1733970642089844e-05,
      "model_forward_time": 1.7459027767181396,
      "step": 54247
    },
    {
      "epoch": 5.1733970642089844e-05,
      "step": 54247,
      "training_step_time": 7.624987363815308
    },
    {
      "epoch": 5.173492431640625e-05,
      "model_forward_time": 1.3697903156280518,
      "step": 54248
    },
    {
      "epoch": 5.173492431640625e-05,
      "step": 54248,
      "training_step_time": 7.693439722061157
    },
    {
      "epoch": 5.173587799072266e-05,
      "model_forward_time": 1.7390234470367432,
      "step": 54249
    },
    {
      "epoch": 5.173587799072266e-05,
      "step": 54249,
      "training_step_time": 7.910057306289673
    },
    {
      "epoch": 5.173683166503906e-05,
      "grad_norm": 0.11740191280841827,
      "learning_rate": 4.7109693397656984e-06,
      "loss": 0.0571,
      "step": 54250
    },
    {
      "epoch": 5.173683166503906e-05,
      "model_forward_time": 1.7386133670806885,
      "step": 54250
    },
    {
      "dynamics_loss_avg": 0.04618446938693523,
      "epoch": 5.173683166503906e-05,
      "step": 54250
    },
    {
      "action_loss_avg": 0.0024942867748904973,
      "epoch": 5.173683166503906e-05,
      "step": 54250
    },
    {
      "epoch": 5.173683166503906e-05,
      "step": 54250,
      "training_step_time": 8.997040271759033
    },
    {
      "epoch": 5.1737785339355467e-05,
      "model_forward_time": 2.3321938514709473,
      "step": 54251
    },
    {
      "epoch": 5.1737785339355467e-05,
      "step": 54251,
      "training_step_time": 9.20948338508606
    },
    {
      "epoch": 5.173873901367188e-05,
      "model_forward_time": 1.753800392150879,
      "step": 54252
    },
    {
      "epoch": 5.173873901367188e-05,
      "step": 54252,
      "training_step_time": 8.884051322937012
    },
    {
      "epoch": 5.173969268798828e-05,
      "model_forward_time": 1.7421810626983643,
      "step": 54253
    },
    {
      "epoch": 5.173969268798828e-05,
      "step": 54253,
      "training_step_time": 8.69542145729065
    },
    {
      "epoch": 5.1740646362304685e-05,
      "model_forward_time": 1.7504267692565918,
      "step": 54254
    },
    {
      "epoch": 5.1740646362304685e-05,
      "step": 54254,
      "training_step_time": 7.496758937835693
    },
    {
      "epoch": 5.1741600036621096e-05,
      "model_forward_time": 1.7579724788665771,
      "step": 54255
    },
    {
      "epoch": 5.1741600036621096e-05,
      "step": 54255,
      "training_step_time": 7.34960150718689
    },
    {
      "epoch": 5.17425537109375e-05,
      "model_forward_time": 1.7454681396484375,
      "step": 54256
    },
    {
      "epoch": 5.17425537109375e-05,
      "step": 54256,
      "training_step_time": 8.573782920837402
    },
    {
      "epoch": 5.1743507385253904e-05,
      "model_forward_time": 1.7533411979675293,
      "step": 54257
    },
    {
      "epoch": 5.1743507385253904e-05,
      "step": 54257,
      "training_step_time": 7.773539066314697
    },
    {
      "epoch": 5.1744461059570315e-05,
      "model_forward_time": 1.7975683212280273,
      "step": 54258
    },
    {
      "epoch": 5.1744461059570315e-05,
      "step": 54258,
      "training_step_time": 8.781155109405518
    },
    {
      "epoch": 5.174541473388672e-05,
      "model_forward_time": 1.7531242370605469,
      "step": 54259
    },
    {
      "epoch": 5.174541473388672e-05,
      "step": 54259,
      "training_step_time": 7.597951889038086
    },
    {
      "epoch": 5.174636840820312e-05,
      "grad_norm": 0.14997121691703796,
      "learning_rate": 4.709318650668894e-06,
      "loss": 0.0591,
      "step": 54260
    },
    {
      "epoch": 5.174636840820312e-05,
      "model_forward_time": 2.3022210597991943,
      "step": 54260
    },
    {
      "dynamics_loss_avg": 0.03785826778039336,
      "epoch": 5.174636840820312e-05,
      "step": 54260
    },
    {
      "action_loss_avg": 0.0014329169207485393,
      "epoch": 5.174636840820312e-05,
      "step": 54260
    },
    {
      "epoch": 5.174636840820312e-05,
      "step": 54260,
      "training_step_time": 9.83938455581665
    },
    {
      "epoch": 5.1747322082519534e-05,
      "model_forward_time": 1.7433228492736816,
      "step": 54261
    },
    {
      "epoch": 5.1747322082519534e-05,
      "step": 54261,
      "training_step_time": 10.042710065841675
    },
    {
      "epoch": 5.174827575683594e-05,
      "model_forward_time": 1.7401084899902344,
      "step": 54262
    },
    {
      "epoch": 5.174827575683594e-05,
      "step": 54262,
      "training_step_time": 7.697826623916626
    },
    {
      "epoch": 5.174922943115234e-05,
      "model_forward_time": 1.735443115234375,
      "step": 54263
    },
    {
      "epoch": 5.174922943115234e-05,
      "step": 54263,
      "training_step_time": 7.618358135223389
    },
    {
      "epoch": 5.175018310546875e-05,
      "model_forward_time": 1.7416365146636963,
      "step": 54264
    },
    {
      "epoch": 5.175018310546875e-05,
      "step": 54264,
      "training_step_time": 7.583442687988281
    },
    {
      "epoch": 5.175113677978516e-05,
      "model_forward_time": 1.7560510635375977,
      "step": 54265
    },
    {
      "epoch": 5.175113677978516e-05,
      "step": 54265,
      "training_step_time": 7.448626279830933
    },
    {
      "epoch": 5.175209045410156e-05,
      "model_forward_time": 1.7331573963165283,
      "step": 54266
    },
    {
      "epoch": 5.175209045410156e-05,
      "step": 54266,
      "training_step_time": 7.740076303482056
    },
    {
      "epoch": 5.175304412841797e-05,
      "model_forward_time": 1.7969443798065186,
      "step": 54267
    },
    {
      "epoch": 5.175304412841797e-05,
      "step": 54267,
      "training_step_time": 7.885702133178711
    },
    {
      "epoch": 5.1753997802734376e-05,
      "model_forward_time": 1.7204434871673584,
      "step": 54268
    },
    {
      "epoch": 5.1753997802734376e-05,
      "step": 54268,
      "training_step_time": 9.335173845291138
    },
    {
      "epoch": 5.175495147705078e-05,
      "model_forward_time": 1.713623285293579,
      "step": 54269
    },
    {
      "epoch": 5.175495147705078e-05,
      "step": 54269,
      "training_step_time": 9.086306095123291
    },
    {
      "epoch": 5.175590515136719e-05,
      "grad_norm": 0.08415595442056656,
      "learning_rate": 4.707667993360568e-06,
      "loss": 0.058,
      "step": 54270
    },
    {
      "epoch": 5.175590515136719e-05,
      "model_forward_time": 2.2611963748931885,
      "step": 54270
    },
    {
      "dynamics_loss_avg": 0.05450892634689808,
      "epoch": 5.175590515136719e-05,
      "step": 54270
    },
    {
      "action_loss_avg": 0.0013424434029730037,
      "epoch": 5.175590515136719e-05,
      "step": 54270
    },
    {
      "epoch": 5.175590515136719e-05,
      "step": 54270,
      "training_step_time": 8.55626106262207
    },
    {
      "epoch": 5.1756858825683595e-05,
      "model_forward_time": 1.719071865081787,
      "step": 54271
    },
    {
      "epoch": 5.1756858825683595e-05,
      "step": 54271,
      "training_step_time": 7.731571435928345
    },
    {
      "epoch": 5.17578125e-05,
      "model_forward_time": 1.7231016159057617,
      "step": 54272
    },
    {
      "epoch": 5.17578125e-05,
      "step": 54272,
      "training_step_time": 7.601153373718262
    },
    {
      "epoch": 5.175876617431641e-05,
      "model_forward_time": 1.7259724140167236,
      "step": 54273
    },
    {
      "epoch": 5.175876617431641e-05,
      "step": 54273,
      "training_step_time": 7.569713115692139
    },
    {
      "epoch": 5.175971984863281e-05,
      "model_forward_time": 1.7245240211486816,
      "step": 54274
    },
    {
      "epoch": 5.175971984863281e-05,
      "step": 54274,
      "training_step_time": 7.6709535121917725
    },
    {
      "epoch": 5.176067352294922e-05,
      "model_forward_time": 1.7182629108428955,
      "step": 54275
    },
    {
      "epoch": 5.176067352294922e-05,
      "step": 54275,
      "training_step_time": 8.238720655441284
    },
    {
      "epoch": 5.176162719726563e-05,
      "model_forward_time": 1.7805273532867432,
      "step": 54276
    },
    {
      "epoch": 5.176162719726563e-05,
      "step": 54276,
      "training_step_time": 9.361221313476562
    },
    {
      "epoch": 5.176258087158203e-05,
      "model_forward_time": 1.7180800437927246,
      "step": 54277
    },
    {
      "epoch": 5.176258087158203e-05,
      "step": 54277,
      "training_step_time": 8.437258005142212
    },
    {
      "epoch": 5.1763534545898436e-05,
      "model_forward_time": 1.6827332973480225,
      "step": 54278
    },
    {
      "epoch": 5.1763534545898436e-05,
      "step": 54278,
      "training_step_time": 8.454755306243896
    },
    {
      "epoch": 5.176448822021485e-05,
      "model_forward_time": 2.240244150161743,
      "step": 54279
    },
    {
      "epoch": 5.176448822021485e-05,
      "step": 54279,
      "training_step_time": 7.748968124389648
    },
    {
      "epoch": 5.176544189453125e-05,
      "grad_norm": 0.1359652429819107,
      "learning_rate": 4.70601736802123e-06,
      "loss": 0.0603,
      "step": 54280
    },
    {
      "epoch": 5.176544189453125e-05,
      "model_forward_time": 1.7054221630096436,
      "step": 54280
    },
    {
      "dynamics_loss_avg": 0.05437382776290178,
      "epoch": 5.176544189453125e-05,
      "step": 54280
    },
    {
      "action_loss_avg": 0.0018064911244437098,
      "epoch": 5.176544189453125e-05,
      "step": 54280
    },
    {
      "epoch": 5.176544189453125e-05,
      "step": 54280,
      "training_step_time": 8.746166229248047
    },
    {
      "epoch": 5.1766395568847655e-05,
      "model_forward_time": 1.6975204944610596,
      "step": 54281
    },
    {
      "epoch": 5.1766395568847655e-05,
      "step": 54281,
      "training_step_time": 7.62481689453125
    },
    {
      "epoch": 5.176734924316406e-05,
      "model_forward_time": 1.6983757019042969,
      "step": 54282
    },
    {
      "epoch": 5.176734924316406e-05,
      "step": 54282,
      "training_step_time": 7.560219049453735
    },
    {
      "epoch": 5.176830291748047e-05,
      "model_forward_time": 1.7002911567687988,
      "step": 54283
    },
    {
      "epoch": 5.176830291748047e-05,
      "step": 54283,
      "training_step_time": 7.7068030834198
    },
    {
      "epoch": 5.1769256591796874e-05,
      "model_forward_time": 1.6989173889160156,
      "step": 54284
    },
    {
      "epoch": 5.1769256591796874e-05,
      "step": 54284,
      "training_step_time": 8.84415054321289
    },
    {
      "epoch": 5.177021026611328e-05,
      "model_forward_time": 1.7445061206817627,
      "step": 54285
    },
    {
      "epoch": 5.177021026611328e-05,
      "step": 54285,
      "training_step_time": 7.742233514785767
    },
    {
      "epoch": 5.177116394042969e-05,
      "model_forward_time": 1.6915488243103027,
      "step": 54286
    },
    {
      "epoch": 5.177116394042969e-05,
      "step": 54286,
      "training_step_time": 7.773428678512573
    },
    {
      "epoch": 5.177211761474609e-05,
      "model_forward_time": 1.6968731880187988,
      "step": 54287
    },
    {
      "epoch": 5.177211761474609e-05,
      "step": 54287,
      "training_step_time": 7.8449625968933105
    },
    {
      "epoch": 5.17730712890625e-05,
      "model_forward_time": 1.8571596145629883,
      "step": 54288
    },
    {
      "epoch": 5.17730712890625e-05,
      "step": 54288,
      "training_step_time": 8.807055234909058
    },
    {
      "epoch": 5.177402496337891e-05,
      "model_forward_time": 1.707676887512207,
      "step": 54289
    },
    {
      "epoch": 5.177402496337891e-05,
      "step": 54289,
      "training_step_time": 8.570405721664429
    },
    {
      "epoch": 5.177497863769531e-05,
      "grad_norm": 0.10851803421974182,
      "learning_rate": 4.704366774831391e-06,
      "loss": 0.0607,
      "step": 54290
    },
    {
      "epoch": 5.177497863769531e-05,
      "model_forward_time": 1.7016773223876953,
      "step": 54290
    },
    {
      "dynamics_loss_avg": 0.04882721733301878,
      "epoch": 5.177497863769531e-05,
      "step": 54290
    },
    {
      "action_loss_avg": 0.0018150629533920438,
      "epoch": 5.177497863769531e-05,
      "step": 54290
    },
    {
      "epoch": 5.177497863769531e-05,
      "step": 54290,
      "training_step_time": 8.600772380828857
    },
    {
      "epoch": 5.1775932312011716e-05,
      "model_forward_time": 1.6983578205108643,
      "step": 54291
    },
    {
      "epoch": 5.1775932312011716e-05,
      "step": 54291,
      "training_step_time": 7.521551609039307
    },
    {
      "epoch": 5.1776885986328127e-05,
      "model_forward_time": 1.7082452774047852,
      "step": 54292
    },
    {
      "epoch": 5.1776885986328127e-05,
      "step": 54292,
      "training_step_time": 7.712688207626343
    },
    {
      "epoch": 5.177783966064453e-05,
      "model_forward_time": 1.7046339511871338,
      "step": 54293
    },
    {
      "epoch": 5.177783966064453e-05,
      "step": 54293,
      "training_step_time": 7.513302326202393
    },
    {
      "epoch": 5.1778793334960935e-05,
      "model_forward_time": 1.7217285633087158,
      "step": 54294
    },
    {
      "epoch": 5.1778793334960935e-05,
      "step": 54294,
      "training_step_time": 7.743658065795898
    },
    {
      "epoch": 5.1779747009277345e-05,
      "model_forward_time": 1.7042510509490967,
      "step": 54295
    },
    {
      "epoch": 5.1779747009277345e-05,
      "step": 54295,
      "training_step_time": 7.881715297698975
    },
    {
      "epoch": 5.178070068359375e-05,
      "model_forward_time": 1.7047123908996582,
      "step": 54296
    },
    {
      "epoch": 5.178070068359375e-05,
      "step": 54296,
      "training_step_time": 7.863811731338501
    },
    {
      "epoch": 5.1781654357910153e-05,
      "model_forward_time": 2.2504799365997314,
      "step": 54297
    },
    {
      "epoch": 5.1781654357910153e-05,
      "step": 54297,
      "training_step_time": 10.197404623031616
    },
    {
      "epoch": 5.1782608032226564e-05,
      "model_forward_time": 1.70103120803833,
      "step": 54298
    },
    {
      "epoch": 5.1782608032226564e-05,
      "step": 54298,
      "training_step_time": 8.19015622138977
    },
    {
      "epoch": 5.178356170654297e-05,
      "model_forward_time": 1.7047975063323975,
      "step": 54299
    },
    {
      "epoch": 5.178356170654297e-05,
      "step": 54299,
      "training_step_time": 9.664923667907715
    },
    {
      "epoch": 5.178451538085937e-05,
      "grad_norm": 0.19007635116577148,
      "learning_rate": 4.702716213971557e-06,
      "loss": 0.0573,
      "step": 54300
    },
    {
      "epoch": 5.178451538085937e-05,
      "model_forward_time": 1.703597068786621,
      "step": 54300
    },
    {
      "dynamics_loss_avg": 0.057538633979856965,
      "epoch": 5.178451538085937e-05,
      "step": 54300
    },
    {
      "action_loss_avg": 0.002232578897383064,
      "epoch": 5.178451538085937e-05,
      "step": 54300
    },
    {
      "epoch": 5.178451538085937e-05,
      "step": 54300,
      "training_step_time": 7.646124601364136
    },
    {
      "epoch": 5.178546905517578e-05,
      "model_forward_time": 1.7016136646270752,
      "step": 54301
    },
    {
      "epoch": 5.178546905517578e-05,
      "step": 54301,
      "training_step_time": 8.56035327911377
    },
    {
      "epoch": 5.178642272949219e-05,
      "model_forward_time": 1.7028179168701172,
      "step": 54302
    },
    {
      "epoch": 5.178642272949219e-05,
      "step": 54302,
      "training_step_time": 7.627196550369263
    },
    {
      "epoch": 5.178737640380859e-05,
      "model_forward_time": 1.6995341777801514,
      "step": 54303
    },
    {
      "epoch": 5.178737640380859e-05,
      "step": 54303,
      "training_step_time": 8.43077039718628
    },
    {
      "epoch": 5.1788330078125e-05,
      "model_forward_time": 1.7063076496124268,
      "step": 54304
    },
    {
      "epoch": 5.1788330078125e-05,
      "step": 54304,
      "training_step_time": 8.345620393753052
    },
    {
      "epoch": 5.1789283752441406e-05,
      "model_forward_time": 1.6987018585205078,
      "step": 54305
    },
    {
      "epoch": 5.1789283752441406e-05,
      "step": 54305,
      "training_step_time": 8.44473910331726
    },
    {
      "epoch": 5.179023742675781e-05,
      "model_forward_time": 2.252197027206421,
      "step": 54306
    },
    {
      "epoch": 5.179023742675781e-05,
      "step": 54306,
      "training_step_time": 7.7765703201293945
    },
    {
      "epoch": 5.179119110107422e-05,
      "model_forward_time": 1.6977462768554688,
      "step": 54307
    },
    {
      "epoch": 5.179119110107422e-05,
      "step": 54307,
      "training_step_time": 9.039262056350708
    },
    {
      "epoch": 5.1792144775390625e-05,
      "model_forward_time": 1.6953132152557373,
      "step": 54308
    },
    {
      "epoch": 5.1792144775390625e-05,
      "step": 54308,
      "training_step_time": 7.7179481983184814
    },
    {
      "epoch": 5.179309844970703e-05,
      "model_forward_time": 1.7272717952728271,
      "step": 54309
    },
    {
      "epoch": 5.179309844970703e-05,
      "step": 54309,
      "training_step_time": 7.6224071979522705
    },
    {
      "epoch": 5.179405212402344e-05,
      "grad_norm": 0.13503599166870117,
      "learning_rate": 4.70106568562223e-06,
      "loss": 0.0573,
      "step": 54310
    },
    {
      "epoch": 5.179405212402344e-05,
      "model_forward_time": 1.7212650775909424,
      "step": 54310
    },
    {
      "dynamics_loss_avg": 0.04761027991771698,
      "epoch": 5.179405212402344e-05,
      "step": 54310
    },
    {
      "action_loss_avg": 0.0012124224827857687,
      "epoch": 5.179405212402344e-05,
      "step": 54310
    },
    {
      "epoch": 5.179405212402344e-05,
      "step": 54310,
      "training_step_time": 7.573957920074463
    },
    {
      "epoch": 5.1795005798339844e-05,
      "model_forward_time": 1.7171483039855957,
      "step": 54311
    },
    {
      "epoch": 5.1795005798339844e-05,
      "step": 54311,
      "training_step_time": 7.717725515365601
    },
    {
      "epoch": 5.179595947265625e-05,
      "model_forward_time": 1.7339246273040771,
      "step": 54312
    },
    {
      "epoch": 5.179595947265625e-05,
      "step": 54312,
      "training_step_time": 7.7149107456207275
    },
    {
      "epoch": 5.179691314697266e-05,
      "model_forward_time": 1.4332165718078613,
      "step": 54313
    },
    {
      "epoch": 5.179691314697266e-05,
      "step": 54313,
      "training_step_time": 7.79637336730957
    },
    {
      "epoch": 5.179786682128906e-05,
      "model_forward_time": 1.7198143005371094,
      "step": 54314
    },
    {
      "epoch": 5.179786682128906e-05,
      "step": 54314,
      "training_step_time": 8.712211847305298
    },
    {
      "epoch": 5.1798820495605467e-05,
      "model_forward_time": 1.718250036239624,
      "step": 54315
    },
    {
      "epoch": 5.1798820495605467e-05,
      "step": 54315,
      "training_step_time": 8.372822046279907
    },
    {
      "epoch": 5.179977416992188e-05,
      "model_forward_time": 2.279784679412842,
      "step": 54316
    },
    {
      "epoch": 5.179977416992188e-05,
      "step": 54316,
      "training_step_time": 9.186202049255371
    },
    {
      "epoch": 5.180072784423828e-05,
      "model_forward_time": 1.6931076049804688,
      "step": 54317
    },
    {
      "epoch": 5.180072784423828e-05,
      "step": 54317,
      "training_step_time": 8.450361013412476
    },
    {
      "epoch": 5.1801681518554685e-05,
      "model_forward_time": 1.3481431007385254,
      "step": 54318
    },
    {
      "epoch": 5.1801681518554685e-05,
      "step": 54318,
      "training_step_time": 7.625641345977783
    },
    {
      "epoch": 5.1802635192871096e-05,
      "model_forward_time": 1.7204062938690186,
      "step": 54319
    },
    {
      "epoch": 5.1802635192871096e-05,
      "step": 54319,
      "training_step_time": 7.666332006454468
    },
    {
      "epoch": 5.18035888671875e-05,
      "grad_norm": 0.17475754022598267,
      "learning_rate": 4.699415189963911e-06,
      "loss": 0.0598,
      "step": 54320
    },
    {
      "epoch": 5.18035888671875e-05,
      "model_forward_time": 1.3653898239135742,
      "step": 54320
    },
    {
      "dynamics_loss_avg": 0.05973143093287945,
      "epoch": 5.18035888671875e-05,
      "step": 54320
    },
    {
      "action_loss_avg": 0.001728754793293774,
      "epoch": 5.18035888671875e-05,
      "step": 54320
    },
    {
      "epoch": 5.18035888671875e-05,
      "step": 54320,
      "training_step_time": 7.689563035964966
    },
    {
      "epoch": 5.1804542541503904e-05,
      "model_forward_time": 1.7148377895355225,
      "step": 54321
    },
    {
      "epoch": 5.1804542541503904e-05,
      "step": 54321,
      "training_step_time": 7.797831773757935
    },
    {
      "epoch": 5.1805496215820315e-05,
      "model_forward_time": 1.7838704586029053,
      "step": 54322
    },
    {
      "epoch": 5.1805496215820315e-05,
      "step": 54322,
      "training_step_time": 8.513734102249146
    },
    {
      "epoch": 5.180644989013672e-05,
      "model_forward_time": 1.7233946323394775,
      "step": 54323
    },
    {
      "epoch": 5.180644989013672e-05,
      "step": 54323,
      "training_step_time": 8.63628625869751
    },
    {
      "epoch": 5.180740356445312e-05,
      "model_forward_time": 1.720038890838623,
      "step": 54324
    },
    {
      "epoch": 5.180740356445312e-05,
      "step": 54324,
      "training_step_time": 7.770171642303467
    },
    {
      "epoch": 5.1808357238769534e-05,
      "model_forward_time": 2.294177770614624,
      "step": 54325
    },
    {
      "epoch": 5.1808357238769534e-05,
      "step": 54325,
      "training_step_time": 9.476931095123291
    },
    {
      "epoch": 5.180931091308594e-05,
      "model_forward_time": 1.717947244644165,
      "step": 54326
    },
    {
      "epoch": 5.180931091308594e-05,
      "step": 54326,
      "training_step_time": 7.627453565597534
    },
    {
      "epoch": 5.181026458740234e-05,
      "model_forward_time": 1.7312071323394775,
      "step": 54327
    },
    {
      "epoch": 5.181026458740234e-05,
      "step": 54327,
      "training_step_time": 7.5683839321136475
    },
    {
      "epoch": 5.181121826171875e-05,
      "model_forward_time": 1.7226054668426514,
      "step": 54328
    },
    {
      "epoch": 5.181121826171875e-05,
      "step": 54328,
      "training_step_time": 7.655876159667969
    },
    {
      "epoch": 5.181217193603516e-05,
      "model_forward_time": 1.7215290069580078,
      "step": 54329
    },
    {
      "epoch": 5.181217193603516e-05,
      "step": 54329,
      "training_step_time": 8.431283712387085
    },
    {
      "epoch": 5.181312561035156e-05,
      "grad_norm": 0.1253005415201187,
      "learning_rate": 4.697764727177093e-06,
      "loss": 0.0604,
      "step": 54330
    },
    {
      "epoch": 5.181312561035156e-05,
      "model_forward_time": 1.7324309349060059,
      "step": 54330
    },
    {
      "dynamics_loss_avg": 0.04970014477148652,
      "epoch": 5.181312561035156e-05,
      "step": 54330
    },
    {
      "action_loss_avg": 0.0020646541786845775,
      "epoch": 5.181312561035156e-05,
      "step": 54330
    },
    {
      "epoch": 5.181312561035156e-05,
      "step": 54330,
      "training_step_time": 7.676581859588623
    },
    {
      "epoch": 5.181407928466797e-05,
      "model_forward_time": 1.7684307098388672,
      "step": 54331
    },
    {
      "epoch": 5.181407928466797e-05,
      "step": 54331,
      "training_step_time": 7.917928218841553
    },
    {
      "epoch": 5.1815032958984376e-05,
      "model_forward_time": 1.7141852378845215,
      "step": 54332
    },
    {
      "epoch": 5.1815032958984376e-05,
      "step": 54332,
      "training_step_time": 7.711394548416138
    },
    {
      "epoch": 5.181598663330078e-05,
      "model_forward_time": 1.7213077545166016,
      "step": 54333
    },
    {
      "epoch": 5.181598663330078e-05,
      "step": 54333,
      "training_step_time": 9.8468918800354
    },
    {
      "epoch": 5.181694030761719e-05,
      "model_forward_time": 2.278578758239746,
      "step": 54334
    },
    {
      "epoch": 5.181694030761719e-05,
      "step": 54334,
      "training_step_time": 7.765275239944458
    },
    {
      "epoch": 5.1817893981933595e-05,
      "model_forward_time": 1.7190756797790527,
      "step": 54335
    },
    {
      "epoch": 5.1817893981933595e-05,
      "step": 54335,
      "training_step_time": 9.025564670562744
    },
    {
      "epoch": 5.181884765625e-05,
      "model_forward_time": 1.7275264263153076,
      "step": 54336
    },
    {
      "epoch": 5.181884765625e-05,
      "step": 54336,
      "training_step_time": 7.756407976150513
    },
    {
      "epoch": 5.181980133056641e-05,
      "model_forward_time": 1.7351853847503662,
      "step": 54337
    },
    {
      "epoch": 5.181980133056641e-05,
      "step": 54337,
      "training_step_time": 7.678920269012451
    },
    {
      "epoch": 5.1820755004882813e-05,
      "model_forward_time": 1.7269790172576904,
      "step": 54338
    },
    {
      "epoch": 5.1820755004882813e-05,
      "step": 54338,
      "training_step_time": 7.672435760498047
    },
    {
      "epoch": 5.182170867919922e-05,
      "model_forward_time": 1.7254490852355957,
      "step": 54339
    },
    {
      "epoch": 5.182170867919922e-05,
      "step": 54339,
      "training_step_time": 7.7280449867248535
    },
    {
      "epoch": 5.182266235351563e-05,
      "grad_norm": 0.10357912629842758,
      "learning_rate": 4.696114297442273e-06,
      "loss": 0.0591,
      "step": 54340
    },
    {
      "epoch": 5.182266235351563e-05,
      "model_forward_time": 1.7543206214904785,
      "step": 54340
    },
    {
      "dynamics_loss_avg": 0.046239778958261014,
      "epoch": 5.182266235351563e-05,
      "step": 54340
    },
    {
      "action_loss_avg": 0.0022913159104064108,
      "epoch": 5.182266235351563e-05,
      "step": 54340
    },
    {
      "epoch": 5.182266235351563e-05,
      "step": 54340,
      "training_step_time": 7.645043611526489
    },
    {
      "epoch": 5.182361602783203e-05,
      "model_forward_time": 1.730128288269043,
      "step": 54341
    },
    {
      "epoch": 5.182361602783203e-05,
      "step": 54341,
      "training_step_time": 7.960481405258179
    },
    {
      "epoch": 5.1824569702148436e-05,
      "model_forward_time": 1.7231745719909668,
      "step": 54342
    },
    {
      "epoch": 5.1824569702148436e-05,
      "step": 54342,
      "training_step_time": 8.81493878364563
    },
    {
      "epoch": 5.182552337646485e-05,
      "model_forward_time": 2.267561435699463,
      "step": 54343
    },
    {
      "epoch": 5.182552337646485e-05,
      "step": 54343,
      "training_step_time": 8.34612512588501
    },
    {
      "epoch": 5.182647705078125e-05,
      "model_forward_time": 1.726768970489502,
      "step": 54344
    },
    {
      "epoch": 5.182647705078125e-05,
      "step": 54344,
      "training_step_time": 8.704853057861328
    },
    {
      "epoch": 5.1827430725097655e-05,
      "model_forward_time": 1.7267043590545654,
      "step": 54345
    },
    {
      "epoch": 5.1827430725097655e-05,
      "step": 54345,
      "training_step_time": 7.65484356880188
    },
    {
      "epoch": 5.182838439941406e-05,
      "model_forward_time": 1.7248218059539795,
      "step": 54346
    },
    {
      "epoch": 5.182838439941406e-05,
      "step": 54346,
      "training_step_time": 7.748340129852295
    },
    {
      "epoch": 5.182933807373047e-05,
      "model_forward_time": 1.6964654922485352,
      "step": 54347
    },
    {
      "epoch": 5.182933807373047e-05,
      "step": 54347,
      "training_step_time": 7.619151830673218
    },
    {
      "epoch": 5.1830291748046874e-05,
      "model_forward_time": 1.7082536220550537,
      "step": 54348
    },
    {
      "epoch": 5.1830291748046874e-05,
      "step": 54348,
      "training_step_time": 8.595906972885132
    },
    {
      "epoch": 5.183124542236328e-05,
      "model_forward_time": 1.7008466720581055,
      "step": 54349
    },
    {
      "epoch": 5.183124542236328e-05,
      "step": 54349,
      "training_step_time": 8.323969841003418
    },
    {
      "epoch": 5.183219909667969e-05,
      "grad_norm": 0.11076939105987549,
      "learning_rate": 4.6944639009399334e-06,
      "loss": 0.0603,
      "step": 54350
    },
    {
      "epoch": 5.183219909667969e-05,
      "model_forward_time": 1.7056176662445068,
      "step": 54350
    },
    {
      "dynamics_loss_avg": 0.06069226767867804,
      "epoch": 5.183219909667969e-05,
      "step": 54350
    },
    {
      "action_loss_avg": 0.0012327109783655033,
      "epoch": 5.183219909667969e-05,
      "step": 54350
    },
    {
      "epoch": 5.183219909667969e-05,
      "step": 54350,
      "training_step_time": 7.866520881652832
    },
    {
      "epoch": 5.183315277099609e-05,
      "model_forward_time": 1.7019426822662354,
      "step": 54351
    },
    {
      "epoch": 5.183315277099609e-05,
      "step": 54351,
      "training_step_time": 8.538089275360107
    },
    {
      "epoch": 5.18341064453125e-05,
      "model_forward_time": 1.697861671447754,
      "step": 54352
    },
    {
      "epoch": 5.18341064453125e-05,
      "step": 54352,
      "training_step_time": 7.749931812286377
    },
    {
      "epoch": 5.183506011962891e-05,
      "model_forward_time": 2.2480974197387695,
      "step": 54353
    },
    {
      "epoch": 5.183506011962891e-05,
      "step": 54353,
      "training_step_time": 7.618321657180786
    },
    {
      "epoch": 5.183601379394531e-05,
      "model_forward_time": 1.7029554843902588,
      "step": 54354
    },
    {
      "epoch": 5.183601379394531e-05,
      "step": 54354,
      "training_step_time": 7.788477182388306
    },
    {
      "epoch": 5.1836967468261716e-05,
      "model_forward_time": 1.6978142261505127,
      "step": 54355
    },
    {
      "epoch": 5.1836967468261716e-05,
      "step": 54355,
      "training_step_time": 7.742162704467773
    },
    {
      "epoch": 5.1837921142578127e-05,
      "model_forward_time": 1.7024126052856445,
      "step": 54356
    },
    {
      "epoch": 5.1837921142578127e-05,
      "step": 54356,
      "training_step_time": 7.584505558013916
    },
    {
      "epoch": 5.183887481689453e-05,
      "model_forward_time": 1.7198586463928223,
      "step": 54357
    },
    {
      "epoch": 5.183887481689453e-05,
      "step": 54357,
      "training_step_time": 7.660516023635864
    },
    {
      "epoch": 5.1839828491210935e-05,
      "model_forward_time": 1.6993563175201416,
      "step": 54358
    },
    {
      "epoch": 5.1839828491210935e-05,
      "step": 54358,
      "training_step_time": 8.747089147567749
    },
    {
      "epoch": 5.1840782165527345e-05,
      "model_forward_time": 1.704282522201538,
      "step": 54359
    },
    {
      "epoch": 5.1840782165527345e-05,
      "step": 54359,
      "training_step_time": 8.692946195602417
    },
    {
      "epoch": 5.184173583984375e-05,
      "grad_norm": 0.09786435216665268,
      "learning_rate": 4.692813537850565e-06,
      "loss": 0.0587,
      "step": 54360
    },
    {
      "epoch": 5.184173583984375e-05,
      "model_forward_time": 1.7001564502716064,
      "step": 54360
    },
    {
      "dynamics_loss_avg": 0.082749087177217,
      "epoch": 5.184173583984375e-05,
      "step": 54360
    },
    {
      "action_loss_avg": 0.0018790481612086297,
      "epoch": 5.184173583984375e-05,
      "step": 54360
    },
    {
      "epoch": 5.184173583984375e-05,
      "step": 54360,
      "training_step_time": 6.762774229049683
    },
    {
      "epoch": 5.1842689514160153e-05,
      "model_forward_time": 1.6947672367095947,
      "step": 54361
    },
    {
      "epoch": 5.1842689514160153e-05,
      "step": 54361,
      "training_step_time": 6.746297121047974
    },
    {
      "epoch": 5.1843643188476564e-05,
      "model_forward_time": 1.7059955596923828,
      "step": 54362
    },
    {
      "epoch": 5.1843643188476564e-05,
      "step": 54362,
      "training_step_time": 6.779834508895874
    },
    {
      "epoch": 5.184459686279297e-05,
      "model_forward_time": 1.7586331367492676,
      "step": 54363
    },
    {
      "epoch": 5.184459686279297e-05,
      "step": 54363,
      "training_step_time": 6.767976760864258
    },
    {
      "epoch": 5.184555053710937e-05,
      "model_forward_time": 1.6998205184936523,
      "step": 54364
    },
    {
      "epoch": 5.184555053710937e-05,
      "step": 54364,
      "training_step_time": 6.729298830032349
    },
    {
      "epoch": 5.184650421142578e-05,
      "model_forward_time": 1.6994667053222656,
      "step": 54365
    },
    {
      "epoch": 5.184650421142578e-05,
      "step": 54365,
      "training_step_time": 6.784035921096802
    },
    {
      "epoch": 5.184745788574219e-05,
      "model_forward_time": 1.7220118045806885,
      "step": 54366
    },
    {
      "epoch": 5.184745788574219e-05,
      "step": 54366,
      "training_step_time": 6.772027015686035
    },
    {
      "epoch": 5.184841156005859e-05,
      "model_forward_time": 1.7017650604248047,
      "step": 54367
    },
    {
      "epoch": 5.184841156005859e-05,
      "step": 54367,
      "training_step_time": 6.75089168548584
    },
    {
      "epoch": 5.1849365234375e-05,
      "model_forward_time": 2.2993197441101074,
      "step": 54368
    },
    {
      "epoch": 5.1849365234375e-05,
      "step": 54368,
      "training_step_time": 8.640564918518066
    },
    {
      "epoch": 5.1850318908691406e-05,
      "model_forward_time": 1.7005393505096436,
      "step": 54369
    },
    {
      "epoch": 5.1850318908691406e-05,
      "step": 54369,
      "training_step_time": 7.917412042617798
    },
    {
      "epoch": 5.185127258300781e-05,
      "grad_norm": 0.21226315200328827,
      "learning_rate": 4.691163208354645e-06,
      "loss": 0.0618,
      "step": 54370
    },
    {
      "epoch": 5.185127258300781e-05,
      "model_forward_time": 1.6962902545928955,
      "step": 54370
    },
    {
      "dynamics_loss_avg": 0.07027153223752976,
      "epoch": 5.185127258300781e-05,
      "step": 54370
    },
    {
      "action_loss_avg": 0.002515249399584718,
      "epoch": 5.185127258300781e-05,
      "step": 54370
    },
    {
      "epoch": 5.185127258300781e-05,
      "step": 54370,
      "training_step_time": 7.295292615890503
    },
    {
      "epoch": 5.185222625732422e-05,
      "model_forward_time": 1.7060754299163818,
      "step": 54371
    },
    {
      "epoch": 5.185222625732422e-05,
      "step": 54371,
      "training_step_time": 6.750760555267334
    },
    {
      "epoch": 5.1853179931640625e-05,
      "model_forward_time": 1.3485140800476074,
      "step": 54372
    },
    {
      "epoch": 5.1853179931640625e-05,
      "step": 54372,
      "training_step_time": 6.782573223114014
    },
    {
      "epoch": 5.185413360595703e-05,
      "model_forward_time": 1.7063632011413574,
      "step": 54373
    },
    {
      "epoch": 5.185413360595703e-05,
      "step": 54373,
      "training_step_time": 6.790560245513916
    },
    {
      "epoch": 5.185508728027344e-05,
      "model_forward_time": 1.6987371444702148,
      "step": 54374
    },
    {
      "epoch": 5.185508728027344e-05,
      "step": 54374,
      "training_step_time": 6.773505449295044
    },
    {
      "epoch": 5.1856040954589844e-05,
      "model_forward_time": 1.7000072002410889,
      "step": 54375
    },
    {
      "epoch": 5.1856040954589844e-05,
      "step": 54375,
      "training_step_time": 6.785085439682007
    },
    {
      "epoch": 5.185699462890625e-05,
      "model_forward_time": 1.724435806274414,
      "step": 54376
    },
    {
      "epoch": 5.185699462890625e-05,
      "step": 54376,
      "training_step_time": 6.774472236633301
    },
    {
      "epoch": 5.185794830322266e-05,
      "model_forward_time": 2.3601996898651123,
      "step": 54377
    },
    {
      "epoch": 5.185794830322266e-05,
      "step": 54377,
      "training_step_time": 7.631131649017334
    },
    {
      "epoch": 5.185890197753906e-05,
      "model_forward_time": 1.7286572456359863,
      "step": 54378
    },
    {
      "epoch": 5.185890197753906e-05,
      "step": 54378,
      "training_step_time": 9.107050657272339
    },
    {
      "epoch": 5.185985565185547e-05,
      "model_forward_time": 1.7170674800872803,
      "step": 54379
    },
    {
      "epoch": 5.185985565185547e-05,
      "step": 54379,
      "training_step_time": 7.321487665176392
    },
    {
      "epoch": 5.186080932617188e-05,
      "grad_norm": 0.12562870979309082,
      "learning_rate": 4.689512912632653e-06,
      "loss": 0.058,
      "step": 54380
    },
    {
      "epoch": 5.186080932617188e-05,
      "model_forward_time": 1.7215232849121094,
      "step": 54380
    },
    {
      "dynamics_loss_avg": 0.0532130379229784,
      "epoch": 5.186080932617188e-05,
      "step": 54380
    },
    {
      "action_loss_avg": 0.0016149951028637589,
      "epoch": 5.186080932617188e-05,
      "step": 54380
    },
    {
      "epoch": 5.186080932617188e-05,
      "step": 54380,
      "training_step_time": 6.775355339050293
    },
    {
      "epoch": 5.186176300048828e-05,
      "model_forward_time": 1.7099456787109375,
      "step": 54381
    },
    {
      "epoch": 5.186176300048828e-05,
      "step": 54381,
      "training_step_time": 6.789123296737671
    },
    {
      "epoch": 5.1862716674804685e-05,
      "model_forward_time": 1.698897123336792,
      "step": 54382
    },
    {
      "epoch": 5.1862716674804685e-05,
      "step": 54382,
      "training_step_time": 6.77143406867981
    },
    {
      "epoch": 5.1863670349121096e-05,
      "model_forward_time": 1.7036917209625244,
      "step": 54383
    },
    {
      "epoch": 5.1863670349121096e-05,
      "step": 54383,
      "training_step_time": 6.789446592330933
    },
    {
      "epoch": 5.18646240234375e-05,
      "model_forward_time": 1.7039241790771484,
      "step": 54384
    },
    {
      "epoch": 5.18646240234375e-05,
      "step": 54384,
      "training_step_time": 6.80674147605896
    },
    {
      "epoch": 5.1865577697753904e-05,
      "model_forward_time": 1.7028000354766846,
      "step": 54385
    },
    {
      "epoch": 5.1865577697753904e-05,
      "step": 54385,
      "training_step_time": 6.782968759536743
    },
    {
      "epoch": 5.1866531372070315e-05,
      "model_forward_time": 2.3328070640563965,
      "step": 54386
    },
    {
      "epoch": 5.1866531372070315e-05,
      "step": 54386,
      "training_step_time": 8.66922116279602
    },
    {
      "epoch": 5.186748504638672e-05,
      "model_forward_time": 1.71250319480896,
      "step": 54387
    },
    {
      "epoch": 5.186748504638672e-05,
      "step": 54387,
      "training_step_time": 9.93192195892334
    },
    {
      "epoch": 5.186843872070312e-05,
      "model_forward_time": 1.717444896697998,
      "step": 54388
    },
    {
      "epoch": 5.186843872070312e-05,
      "step": 54388,
      "training_step_time": 7.5136449337005615
    },
    {
      "epoch": 5.1869392395019534e-05,
      "model_forward_time": 1.7193524837493896,
      "step": 54389
    },
    {
      "epoch": 5.1869392395019534e-05,
      "step": 54389,
      "training_step_time": 6.832884311676025
    },
    {
      "epoch": 5.187034606933594e-05,
      "grad_norm": 0.09828324615955353,
      "learning_rate": 4.687862650865059e-06,
      "loss": 0.06,
      "step": 54390
    },
    {
      "epoch": 5.187034606933594e-05,
      "model_forward_time": 1.712096929550171,
      "step": 54390
    },
    {
      "dynamics_loss_avg": 0.03802067348733544,
      "epoch": 5.187034606933594e-05,
      "step": 54390
    },
    {
      "action_loss_avg": 0.0015452317777089775,
      "epoch": 5.187034606933594e-05,
      "step": 54390
    },
    {
      "epoch": 5.187034606933594e-05,
      "step": 54390,
      "training_step_time": 6.797575235366821
    },
    {
      "epoch": 5.187129974365234e-05,
      "model_forward_time": 1.7221591472625732,
      "step": 54391
    },
    {
      "epoch": 5.187129974365234e-05,
      "step": 54391,
      "training_step_time": 6.835800409317017
    },
    {
      "epoch": 5.187225341796875e-05,
      "model_forward_time": 1.7495973110198975,
      "step": 54392
    },
    {
      "epoch": 5.187225341796875e-05,
      "step": 54392,
      "training_step_time": 6.811980962753296
    },
    {
      "epoch": 5.187320709228516e-05,
      "model_forward_time": 1.7173857688903809,
      "step": 54393
    },
    {
      "epoch": 5.187320709228516e-05,
      "step": 54393,
      "training_step_time": 6.795209646224976
    },
    {
      "epoch": 5.187416076660156e-05,
      "model_forward_time": 1.7129433155059814,
      "step": 54394
    },
    {
      "epoch": 5.187416076660156e-05,
      "step": 54394,
      "training_step_time": 6.795223951339722
    },
    {
      "epoch": 5.187511444091797e-05,
      "model_forward_time": 2.327573537826538,
      "step": 54395
    },
    {
      "epoch": 5.187511444091797e-05,
      "step": 54395,
      "training_step_time": 8.433083534240723
    },
    {
      "epoch": 5.1876068115234376e-05,
      "model_forward_time": 1.7432968616485596,
      "step": 54396
    },
    {
      "epoch": 5.1876068115234376e-05,
      "step": 54396,
      "training_step_time": 10.957741260528564
    },
    {
      "epoch": 5.187702178955078e-05,
      "model_forward_time": 1.7439563274383545,
      "step": 54397
    },
    {
      "epoch": 5.187702178955078e-05,
      "step": 54397,
      "training_step_time": 7.6775062084198
    },
    {
      "epoch": 5.187797546386719e-05,
      "model_forward_time": 1.7331197261810303,
      "step": 54398
    },
    {
      "epoch": 5.187797546386719e-05,
      "step": 54398,
      "training_step_time": 7.381622076034546
    },
    {
      "epoch": 5.1878929138183595e-05,
      "model_forward_time": 1.3834004402160645,
      "step": 54399
    },
    {
      "epoch": 5.1878929138183595e-05,
      "step": 54399,
      "training_step_time": 6.7581939697265625
    },
    {
      "epoch": 5.18798828125e-05,
      "grad_norm": 0.10377772152423859,
      "learning_rate": 4.6862124232323394e-06,
      "loss": 0.0626,
      "step": 54400
    },
    {
      "epoch": 5.18798828125e-05,
      "model_forward_time": 1.7349426746368408,
      "step": 54400
    },
    {
      "dynamics_loss_avg": 0.06022841855883598,
      "epoch": 5.18798828125e-05,
      "step": 54400
    },
    {
      "action_loss_avg": 0.0014737832359969616,
      "epoch": 5.18798828125e-05,
      "step": 54400
    },
    {
      "epoch": 5.18798828125e-05,
      "step": 54400,
      "training_step_time": 6.78161883354187
    },
    {
      "epoch": 5.188083648681641e-05,
      "model_forward_time": 1.7379894256591797,
      "step": 54401
    },
    {
      "epoch": 5.188083648681641e-05,
      "step": 54401,
      "training_step_time": 6.858903646469116
    },
    {
      "epoch": 5.1881790161132813e-05,
      "model_forward_time": 1.7530248165130615,
      "step": 54402
    },
    {
      "epoch": 5.1881790161132813e-05,
      "step": 54402,
      "training_step_time": 6.7588465213775635
    },
    {
      "epoch": 5.188274383544922e-05,
      "model_forward_time": 1.7403438091278076,
      "step": 54403
    },
    {
      "epoch": 5.188274383544922e-05,
      "step": 54403,
      "training_step_time": 6.777526617050171
    },
    {
      "epoch": 5.188369750976563e-05,
      "model_forward_time": 2.2641074657440186,
      "step": 54404
    },
    {
      "epoch": 5.188369750976563e-05,
      "step": 54404,
      "training_step_time": 8.147678136825562
    },
    {
      "epoch": 5.188465118408203e-05,
      "model_forward_time": 1.7918307781219482,
      "step": 54405
    },
    {
      "epoch": 5.188465118408203e-05,
      "step": 54405,
      "training_step_time": 8.87084674835205
    },
    {
      "epoch": 5.1885604858398436e-05,
      "model_forward_time": 1.7355015277862549,
      "step": 54406
    },
    {
      "epoch": 5.1885604858398436e-05,
      "step": 54406,
      "training_step_time": 8.731186628341675
    },
    {
      "epoch": 5.188655853271484e-05,
      "model_forward_time": 1.7416107654571533,
      "step": 54407
    },
    {
      "epoch": 5.188655853271484e-05,
      "step": 54407,
      "training_step_time": 7.3724446296691895
    },
    {
      "epoch": 5.188751220703125e-05,
      "model_forward_time": 1.7468876838684082,
      "step": 54408
    },
    {
      "epoch": 5.188751220703125e-05,
      "step": 54408,
      "training_step_time": 6.763514757156372
    },
    {
      "epoch": 5.1888465881347655e-05,
      "model_forward_time": 1.7435495853424072,
      "step": 54409
    },
    {
      "epoch": 5.1888465881347655e-05,
      "step": 54409,
      "training_step_time": 6.803573369979858
    },
    {
      "epoch": 5.188941955566406e-05,
      "grad_norm": 0.12324853241443634,
      "learning_rate": 4.684562229914957e-06,
      "loss": 0.0586,
      "step": 54410
    },
    {
      "epoch": 5.188941955566406e-05,
      "model_forward_time": 1.7385945320129395,
      "step": 54410
    },
    {
      "dynamics_loss_avg": 0.06063385475426912,
      "epoch": 5.188941955566406e-05,
      "step": 54410
    },
    {
      "action_loss_avg": 0.001696895735221915,
      "epoch": 5.188941955566406e-05,
      "step": 54410
    },
    {
      "epoch": 5.188941955566406e-05,
      "step": 54410,
      "training_step_time": 6.74833083152771
    },
    {
      "epoch": 5.189037322998047e-05,
      "model_forward_time": 1.7388215065002441,
      "step": 54411
    },
    {
      "epoch": 5.189037322998047e-05,
      "step": 54411,
      "training_step_time": 6.788343667984009
    },
    {
      "epoch": 5.1891326904296874e-05,
      "model_forward_time": 1.7316005229949951,
      "step": 54412
    },
    {
      "epoch": 5.1891326904296874e-05,
      "step": 54412,
      "training_step_time": 6.7892937660217285
    },
    {
      "epoch": 5.189228057861328e-05,
      "model_forward_time": 1.7323479652404785,
      "step": 54413
    },
    {
      "epoch": 5.189228057861328e-05,
      "step": 54413,
      "training_step_time": 7.677801609039307
    },
    {
      "epoch": 5.189323425292969e-05,
      "model_forward_time": 2.3113441467285156,
      "step": 54414
    },
    {
      "epoch": 5.189323425292969e-05,
      "step": 54414,
      "training_step_time": 8.749157428741455
    },
    {
      "epoch": 5.189418792724609e-05,
      "model_forward_time": 1.716334581375122,
      "step": 54415
    },
    {
      "epoch": 5.189418792724609e-05,
      "step": 54415,
      "training_step_time": 8.630602598190308
    },
    {
      "epoch": 5.18951416015625e-05,
      "model_forward_time": 1.7368943691253662,
      "step": 54416
    },
    {
      "epoch": 5.18951416015625e-05,
      "step": 54416,
      "training_step_time": 7.643737077713013
    },
    {
      "epoch": 5.189609527587891e-05,
      "model_forward_time": 1.7215046882629395,
      "step": 54417
    },
    {
      "epoch": 5.189609527587891e-05,
      "step": 54417,
      "training_step_time": 6.838015556335449
    },
    {
      "epoch": 5.189704895019531e-05,
      "model_forward_time": 1.7211365699768066,
      "step": 54418
    },
    {
      "epoch": 5.189704895019531e-05,
      "step": 54418,
      "training_step_time": 6.818066596984863
    },
    {
      "epoch": 5.1898002624511716e-05,
      "model_forward_time": 1.7145962715148926,
      "step": 54419
    },
    {
      "epoch": 5.1898002624511716e-05,
      "step": 54419,
      "training_step_time": 6.844249725341797
    },
    {
      "epoch": 5.189895629882813e-05,
      "grad_norm": 0.11741811782121658,
      "learning_rate": 4.682912071093374e-06,
      "loss": 0.0584,
      "step": 54420
    },
    {
      "epoch": 5.189895629882813e-05,
      "model_forward_time": 1.7273554801940918,
      "step": 54420
    },
    {
      "dynamics_loss_avg": 0.06575075387954712,
      "epoch": 5.189895629882813e-05,
      "step": 54420
    },
    {
      "action_loss_avg": 0.0018286508682649582,
      "epoch": 5.189895629882813e-05,
      "step": 54420
    },
    {
      "epoch": 5.189895629882813e-05,
      "step": 54420,
      "training_step_time": 6.830629110336304
    },
    {
      "epoch": 5.189990997314453e-05,
      "model_forward_time": 1.719433307647705,
      "step": 54421
    },
    {
      "epoch": 5.189990997314453e-05,
      "step": 54421,
      "training_step_time": 6.839471817016602
    },
    {
      "epoch": 5.1900863647460935e-05,
      "model_forward_time": 1.7153494358062744,
      "step": 54422
    },
    {
      "epoch": 5.1900863647460935e-05,
      "step": 54422,
      "training_step_time": 7.637448787689209
    },
    {
      "epoch": 5.1901817321777345e-05,
      "model_forward_time": 2.317289113998413,
      "step": 54423
    },
    {
      "epoch": 5.1901817321777345e-05,
      "step": 54423,
      "training_step_time": 7.732877969741821
    },
    {
      "epoch": 5.190277099609375e-05,
      "model_forward_time": 1.7237343788146973,
      "step": 54424
    },
    {
      "epoch": 5.190277099609375e-05,
      "step": 54424,
      "training_step_time": 9.31346082687378
    },
    {
      "epoch": 5.1903724670410154e-05,
      "model_forward_time": 1.7209703922271729,
      "step": 54425
    },
    {
      "epoch": 5.1903724670410154e-05,
      "step": 54425,
      "training_step_time": 7.555262565612793
    },
    {
      "epoch": 5.1904678344726564e-05,
      "model_forward_time": 1.7160098552703857,
      "step": 54426
    },
    {
      "epoch": 5.1904678344726564e-05,
      "step": 54426,
      "training_step_time": 7.330869674682617
    },
    {
      "epoch": 5.190563201904297e-05,
      "model_forward_time": 1.714836597442627,
      "step": 54427
    },
    {
      "epoch": 5.190563201904297e-05,
      "step": 54427,
      "training_step_time": 6.7776734828948975
    },
    {
      "epoch": 5.190658569335937e-05,
      "model_forward_time": 1.7226567268371582,
      "step": 54428
    },
    {
      "epoch": 5.190658569335937e-05,
      "step": 54428,
      "training_step_time": 6.766618251800537
    },
    {
      "epoch": 5.190753936767578e-05,
      "model_forward_time": 1.7159454822540283,
      "step": 54429
    },
    {
      "epoch": 5.190753936767578e-05,
      "step": 54429,
      "training_step_time": 6.729117393493652
    },
    {
      "epoch": 5.190849304199219e-05,
      "grad_norm": 0.14098644256591797,
      "learning_rate": 4.681261946948052e-06,
      "loss": 0.0578,
      "step": 54430
    },
    {
      "epoch": 5.190849304199219e-05,
      "model_forward_time": 1.7159678936004639,
      "step": 54430
    },
    {
      "dynamics_loss_avg": 0.049478701502084735,
      "epoch": 5.190849304199219e-05,
      "step": 54430
    },
    {
      "action_loss_avg": 0.0016294721164740622,
      "epoch": 5.190849304199219e-05,
      "step": 54430
    },
    {
      "epoch": 5.190849304199219e-05,
      "step": 54430,
      "training_step_time": 6.768706560134888
    },
    {
      "epoch": 5.190944671630859e-05,
      "model_forward_time": 1.697413444519043,
      "step": 54431
    },
    {
      "epoch": 5.190944671630859e-05,
      "step": 54431,
      "training_step_time": 7.562657356262207
    },
    {
      "epoch": 5.1910400390625e-05,
      "model_forward_time": 2.328700542449951,
      "step": 54432
    },
    {
      "epoch": 5.1910400390625e-05,
      "step": 54432,
      "training_step_time": 8.210862159729004
    },
    {
      "epoch": 5.1911354064941406e-05,
      "model_forward_time": 1.7221472263336182,
      "step": 54433
    },
    {
      "epoch": 5.1911354064941406e-05,
      "step": 54433,
      "training_step_time": 8.784457921981812
    },
    {
      "epoch": 5.191230773925781e-05,
      "model_forward_time": 1.7220406532287598,
      "step": 54434
    },
    {
      "epoch": 5.191230773925781e-05,
      "step": 54434,
      "training_step_time": 7.698888301849365
    },
    {
      "epoch": 5.191326141357422e-05,
      "model_forward_time": 1.720266580581665,
      "step": 54435
    },
    {
      "epoch": 5.191326141357422e-05,
      "step": 54435,
      "training_step_time": 7.336576461791992
    },
    {
      "epoch": 5.1914215087890625e-05,
      "model_forward_time": 1.7144286632537842,
      "step": 54436
    },
    {
      "epoch": 5.1914215087890625e-05,
      "step": 54436,
      "training_step_time": 6.784497261047363
    },
    {
      "epoch": 5.191516876220703e-05,
      "model_forward_time": 1.706369400024414,
      "step": 54437
    },
    {
      "epoch": 5.191516876220703e-05,
      "step": 54437,
      "training_step_time": 6.743921279907227
    },
    {
      "epoch": 5.191612243652344e-05,
      "model_forward_time": 1.7134342193603516,
      "step": 54438
    },
    {
      "epoch": 5.191612243652344e-05,
      "step": 54438,
      "training_step_time": 6.803083181381226
    },
    {
      "epoch": 5.1917076110839844e-05,
      "model_forward_time": 1.720691442489624,
      "step": 54439
    },
    {
      "epoch": 5.1917076110839844e-05,
      "step": 54439,
      "training_step_time": 6.739706754684448
    },
    {
      "epoch": 5.191802978515625e-05,
      "grad_norm": 0.14981722831726074,
      "learning_rate": 4.679611857659443e-06,
      "loss": 0.0596,
      "step": 54440
    },
    {
      "epoch": 5.191802978515625e-05,
      "model_forward_time": 1.7009389400482178,
      "step": 54440
    },
    {
      "dynamics_loss_avg": 0.05796752255409956,
      "epoch": 5.191802978515625e-05,
      "step": 54440
    },
    {
      "action_loss_avg": 0.0015340115525759757,
      "epoch": 5.191802978515625e-05,
      "step": 54440
    },
    {
      "epoch": 5.191802978515625e-05,
      "step": 54440,
      "training_step_time": 7.371121644973755
    },
    {
      "epoch": 5.191898345947266e-05,
      "model_forward_time": 2.277066230773926,
      "step": 54441
    },
    {
      "epoch": 5.191898345947266e-05,
      "step": 54441,
      "training_step_time": 9.214203834533691
    },
    {
      "epoch": 5.191993713378906e-05,
      "model_forward_time": 1.7218303680419922,
      "step": 54442
    },
    {
      "epoch": 5.191993713378906e-05,
      "step": 54442,
      "training_step_time": 8.365541219711304
    },
    {
      "epoch": 5.192089080810547e-05,
      "model_forward_time": 1.7254817485809326,
      "step": 54443
    },
    {
      "epoch": 5.192089080810547e-05,
      "step": 54443,
      "training_step_time": 7.725480079650879
    },
    {
      "epoch": 5.192184448242188e-05,
      "model_forward_time": 1.7222893238067627,
      "step": 54444
    },
    {
      "epoch": 5.192184448242188e-05,
      "step": 54444,
      "training_step_time": 7.299212694168091
    },
    {
      "epoch": 5.192279815673828e-05,
      "model_forward_time": 1.7344176769256592,
      "step": 54445
    },
    {
      "epoch": 5.192279815673828e-05,
      "step": 54445,
      "training_step_time": 6.766692399978638
    },
    {
      "epoch": 5.1923751831054686e-05,
      "model_forward_time": 1.7200028896331787,
      "step": 54446
    },
    {
      "epoch": 5.1923751831054686e-05,
      "step": 54446,
      "training_step_time": 6.720249176025391
    },
    {
      "epoch": 5.1924705505371096e-05,
      "model_forward_time": 1.716796875,
      "step": 54447
    },
    {
      "epoch": 5.1924705505371096e-05,
      "step": 54447,
      "training_step_time": 6.764779567718506
    },
    {
      "epoch": 5.19256591796875e-05,
      "model_forward_time": 1.7293052673339844,
      "step": 54448
    },
    {
      "epoch": 5.19256591796875e-05,
      "step": 54448,
      "training_step_time": 6.783009052276611
    },
    {
      "epoch": 5.1926612854003904e-05,
      "model_forward_time": 1.7198197841644287,
      "step": 54449
    },
    {
      "epoch": 5.1926612854003904e-05,
      "step": 54449,
      "training_step_time": 6.762621641159058
    },
    {
      "epoch": 5.1927566528320315e-05,
      "grad_norm": 0.16418789327144623,
      "learning_rate": 4.6779618034079975e-06,
      "loss": 0.0584,
      "step": 54450
    },
    {
      "epoch": 5.1927566528320315e-05,
      "model_forward_time": 2.265028715133667,
      "step": 54450
    },
    {
      "dynamics_loss_avg": 0.06564881037920714,
      "epoch": 5.1927566528320315e-05,
      "step": 54450
    },
    {
      "action_loss_avg": 0.0017550292541272938,
      "epoch": 5.1927566528320315e-05,
      "step": 54450
    },
    {
      "epoch": 5.1927566528320315e-05,
      "step": 54450,
      "training_step_time": 8.557240724563599
    },
    {
      "epoch": 5.192852020263672e-05,
      "model_forward_time": 1.7778370380401611,
      "step": 54451
    },
    {
      "epoch": 5.192852020263672e-05,
      "step": 54451,
      "training_step_time": 9.40895962715149
    },
    {
      "epoch": 5.192947387695312e-05,
      "model_forward_time": 1.7323753833770752,
      "step": 54452
    },
    {
      "epoch": 5.192947387695312e-05,
      "step": 54452,
      "training_step_time": 7.930213689804077
    },
    {
      "epoch": 5.1930427551269534e-05,
      "model_forward_time": 1.7222700119018555,
      "step": 54453
    },
    {
      "epoch": 5.1930427551269534e-05,
      "step": 54453,
      "training_step_time": 7.59370756149292
    },
    {
      "epoch": 5.193138122558594e-05,
      "model_forward_time": 1.7156131267547607,
      "step": 54454
    },
    {
      "epoch": 5.193138122558594e-05,
      "step": 54454,
      "training_step_time": 6.715664386749268
    },
    {
      "epoch": 5.193233489990234e-05,
      "model_forward_time": 1.7208764553070068,
      "step": 54455
    },
    {
      "epoch": 5.193233489990234e-05,
      "step": 54455,
      "training_step_time": 6.744553327560425
    },
    {
      "epoch": 5.193328857421875e-05,
      "model_forward_time": 1.7412049770355225,
      "step": 54456
    },
    {
      "epoch": 5.193328857421875e-05,
      "step": 54456,
      "training_step_time": 6.7390711307525635
    },
    {
      "epoch": 5.193424224853516e-05,
      "model_forward_time": 1.7189226150512695,
      "step": 54457
    },
    {
      "epoch": 5.193424224853516e-05,
      "step": 54457,
      "training_step_time": 6.742332696914673
    },
    {
      "epoch": 5.193519592285156e-05,
      "model_forward_time": 1.8140912055969238,
      "step": 54458
    },
    {
      "epoch": 5.193519592285156e-05,
      "step": 54458,
      "training_step_time": 6.768033981323242
    },
    {
      "epoch": 5.193614959716797e-05,
      "model_forward_time": 1.8229751586914062,
      "step": 54459
    },
    {
      "epoch": 5.193614959716797e-05,
      "step": 54459,
      "training_step_time": 7.635270595550537
    },
    {
      "epoch": 5.1937103271484376e-05,
      "grad_norm": 0.10097180306911469,
      "learning_rate": 4.6763117843741675e-06,
      "loss": 0.0546,
      "step": 54460
    },
    {
      "epoch": 5.1937103271484376e-05,
      "model_forward_time": 2.354037046432495,
      "step": 54460
    },
    {
      "dynamics_loss_avg": 0.051332973409444094,
      "epoch": 5.1937103271484376e-05,
      "step": 54460
    },
    {
      "action_loss_avg": 0.0016619030182482674,
      "epoch": 5.1937103271484376e-05,
      "step": 54460
    },
    {
      "epoch": 5.1937103271484376e-05,
      "step": 54460,
      "training_step_time": 9.215001583099365
    },
    {
      "epoch": 5.193805694580078e-05,
      "model_forward_time": 1.8112328052520752,
      "step": 54461
    },
    {
      "epoch": 5.193805694580078e-05,
      "step": 54461,
      "training_step_time": 9.795685529708862
    },
    {
      "epoch": 5.193901062011719e-05,
      "model_forward_time": 1.8295586109161377,
      "step": 54462
    },
    {
      "epoch": 5.193901062011719e-05,
      "step": 54462,
      "training_step_time": 7.734657049179077
    },
    {
      "epoch": 5.1939964294433595e-05,
      "model_forward_time": 1.8291704654693604,
      "step": 54463
    },
    {
      "epoch": 5.1939964294433595e-05,
      "step": 54463,
      "training_step_time": 7.344383716583252
    },
    {
      "epoch": 5.194091796875e-05,
      "model_forward_time": 1.8011271953582764,
      "step": 54464
    },
    {
      "epoch": 5.194091796875e-05,
      "step": 54464,
      "training_step_time": 6.817116022109985
    },
    {
      "epoch": 5.194187164306641e-05,
      "model_forward_time": 1.8109896183013916,
      "step": 54465
    },
    {
      "epoch": 5.194187164306641e-05,
      "step": 54465,
      "training_step_time": 6.777938365936279
    },
    {
      "epoch": 5.1942825317382814e-05,
      "model_forward_time": 1.8089828491210938,
      "step": 54466
    },
    {
      "epoch": 5.1942825317382814e-05,
      "step": 54466,
      "training_step_time": 6.759476184844971
    },
    {
      "epoch": 5.194377899169922e-05,
      "model_forward_time": 1.4402353763580322,
      "step": 54467
    },
    {
      "epoch": 5.194377899169922e-05,
      "step": 54467,
      "training_step_time": 6.734379291534424
    },
    {
      "epoch": 5.194473266601563e-05,
      "model_forward_time": 1.4494574069976807,
      "step": 54468
    },
    {
      "epoch": 5.194473266601563e-05,
      "step": 54468,
      "training_step_time": 7.552526473999023
    },
    {
      "epoch": 5.194568634033203e-05,
      "model_forward_time": 2.345015048980713,
      "step": 54469
    },
    {
      "epoch": 5.194568634033203e-05,
      "step": 54469,
      "training_step_time": 9.35768175125122
    },
    {
      "epoch": 5.1946640014648436e-05,
      "grad_norm": 0.11080928146839142,
      "learning_rate": 4.674661800738394e-06,
      "loss": 0.0558,
      "step": 54470
    },
    {
      "epoch": 5.1946640014648436e-05,
      "model_forward_time": 1.8138444423675537,
      "step": 54470
    },
    {
      "dynamics_loss_avg": 0.05103464741259813,
      "epoch": 5.1946640014648436e-05,
      "step": 54470
    },
    {
      "action_loss_avg": 0.0015302915242500603,
      "epoch": 5.1946640014648436e-05,
      "step": 54470
    },
    {
      "epoch": 5.1946640014648436e-05,
      "step": 54470,
      "training_step_time": 8.432870864868164
    },
    {
      "epoch": 5.194759368896484e-05,
      "model_forward_time": 1.8139076232910156,
      "step": 54471
    },
    {
      "epoch": 5.194759368896484e-05,
      "step": 54471,
      "training_step_time": 8.640348672866821
    },
    {
      "epoch": 5.194854736328125e-05,
      "model_forward_time": 1.8076233863830566,
      "step": 54472
    },
    {
      "epoch": 5.194854736328125e-05,
      "step": 54472,
      "training_step_time": 7.296499490737915
    },
    {
      "epoch": 5.1949501037597655e-05,
      "model_forward_time": 1.8159146308898926,
      "step": 54473
    },
    {
      "epoch": 5.1949501037597655e-05,
      "step": 54473,
      "training_step_time": 6.763460159301758
    },
    {
      "epoch": 5.195045471191406e-05,
      "model_forward_time": 1.800346851348877,
      "step": 54474
    },
    {
      "epoch": 5.195045471191406e-05,
      "step": 54474,
      "training_step_time": 6.708196401596069
    },
    {
      "epoch": 5.195140838623047e-05,
      "model_forward_time": 1.807497501373291,
      "step": 54475
    },
    {
      "epoch": 5.195140838623047e-05,
      "step": 54475,
      "training_step_time": 6.763365983963013
    },
    {
      "epoch": 5.1952362060546874e-05,
      "model_forward_time": 1.4480509757995605,
      "step": 54476
    },
    {
      "epoch": 5.1952362060546874e-05,
      "step": 54476,
      "training_step_time": 6.762601375579834
    },
    {
      "epoch": 5.195331573486328e-05,
      "model_forward_time": 1.805177927017212,
      "step": 54477
    },
    {
      "epoch": 5.195331573486328e-05,
      "step": 54477,
      "training_step_time": 7.512681722640991
    },
    {
      "epoch": 5.195426940917969e-05,
      "model_forward_time": 2.383455276489258,
      "step": 54478
    },
    {
      "epoch": 5.195426940917969e-05,
      "step": 54478,
      "training_step_time": 8.486090421676636
    },
    {
      "epoch": 5.195522308349609e-05,
      "model_forward_time": 1.8252785205841064,
      "step": 54479
    },
    {
      "epoch": 5.195522308349609e-05,
      "step": 54479,
      "training_step_time": 7.892175197601318
    },
    {
      "epoch": 5.19561767578125e-05,
      "grad_norm": 0.1285623461008072,
      "learning_rate": 4.673011852681117e-06,
      "loss": 0.0583,
      "step": 54480
    },
    {
      "epoch": 5.19561767578125e-05,
      "model_forward_time": 1.8209514617919922,
      "step": 54480
    },
    {
      "dynamics_loss_avg": 0.054052668064832686,
      "epoch": 5.19561767578125e-05,
      "step": 54480
    },
    {
      "action_loss_avg": 0.0014989722752943636,
      "epoch": 5.19561767578125e-05,
      "step": 54480
    },
    {
      "epoch": 5.19561767578125e-05,
      "step": 54480,
      "training_step_time": 7.626682281494141
    },
    {
      "epoch": 5.195713043212891e-05,
      "model_forward_time": 1.8259789943695068,
      "step": 54481
    },
    {
      "epoch": 5.195713043212891e-05,
      "step": 54481,
      "training_step_time": 7.545602321624756
    },
    {
      "epoch": 5.195808410644531e-05,
      "model_forward_time": 1.8255751132965088,
      "step": 54482
    },
    {
      "epoch": 5.195808410644531e-05,
      "step": 54482,
      "training_step_time": 6.709402561187744
    },
    {
      "epoch": 5.1959037780761716e-05,
      "model_forward_time": 1.8263907432556152,
      "step": 54483
    },
    {
      "epoch": 5.1959037780761716e-05,
      "step": 54483,
      "training_step_time": 6.7495691776275635
    },
    {
      "epoch": 5.195999145507813e-05,
      "model_forward_time": 1.82279372215271,
      "step": 54484
    },
    {
      "epoch": 5.195999145507813e-05,
      "step": 54484,
      "training_step_time": 6.79677677154541
    },
    {
      "epoch": 5.196094512939453e-05,
      "model_forward_time": 1.8253300189971924,
      "step": 54485
    },
    {
      "epoch": 5.196094512939453e-05,
      "step": 54485,
      "training_step_time": 6.767715215682983
    },
    {
      "epoch": 5.1961898803710935e-05,
      "model_forward_time": 1.8165161609649658,
      "step": 54486
    },
    {
      "epoch": 5.1961898803710935e-05,
      "step": 54486,
      "training_step_time": 8.062852382659912
    },
    {
      "epoch": 5.1962852478027346e-05,
      "model_forward_time": 2.372378349304199,
      "step": 54487
    },
    {
      "epoch": 5.1962852478027346e-05,
      "step": 54487,
      "training_step_time": 9.527107954025269
    },
    {
      "epoch": 5.196380615234375e-05,
      "model_forward_time": 1.8213560581207275,
      "step": 54488
    },
    {
      "epoch": 5.196380615234375e-05,
      "step": 54488,
      "training_step_time": 9.561970949172974
    },
    {
      "epoch": 5.1964759826660154e-05,
      "model_forward_time": 1.8200323581695557,
      "step": 54489
    },
    {
      "epoch": 5.1964759826660154e-05,
      "step": 54489,
      "training_step_time": 7.900935173034668
    },
    {
      "epoch": 5.1965713500976564e-05,
      "grad_norm": 0.12585443258285522,
      "learning_rate": 4.671361940382772e-06,
      "loss": 0.0586,
      "step": 54490
    },
    {
      "epoch": 5.1965713500976564e-05,
      "model_forward_time": 1.8257226943969727,
      "step": 54490
    },
    {
      "dynamics_loss_avg": 0.05447957832366228,
      "epoch": 5.1965713500976564e-05,
      "step": 54490
    },
    {
      "action_loss_avg": 0.00213605759490747,
      "epoch": 5.1965713500976564e-05,
      "step": 54490
    },
    {
      "epoch": 5.1965713500976564e-05,
      "step": 54490,
      "training_step_time": 7.580688953399658
    },
    {
      "epoch": 5.196666717529297e-05,
      "model_forward_time": 1.828927755355835,
      "step": 54491
    },
    {
      "epoch": 5.196666717529297e-05,
      "step": 54491,
      "training_step_time": 7.325107574462891
    },
    {
      "epoch": 5.196762084960937e-05,
      "model_forward_time": 1.8300635814666748,
      "step": 54492
    },
    {
      "epoch": 5.196762084960937e-05,
      "step": 54492,
      "training_step_time": 6.743719577789307
    },
    {
      "epoch": 5.196857452392578e-05,
      "model_forward_time": 1.8288757801055908,
      "step": 54493
    },
    {
      "epoch": 5.196857452392578e-05,
      "step": 54493,
      "training_step_time": 6.737477779388428
    },
    {
      "epoch": 5.196952819824219e-05,
      "model_forward_time": 1.8235852718353271,
      "step": 54494
    },
    {
      "epoch": 5.196952819824219e-05,
      "step": 54494,
      "training_step_time": 6.723376274108887
    },
    {
      "epoch": 5.197048187255859e-05,
      "model_forward_time": 1.821594476699829,
      "step": 54495
    },
    {
      "epoch": 5.197048187255859e-05,
      "step": 54495,
      "training_step_time": 7.2384352684021
    },
    {
      "epoch": 5.1971435546875e-05,
      "model_forward_time": 1.8236966133117676,
      "step": 54496
    },
    {
      "epoch": 5.1971435546875e-05,
      "step": 54496,
      "training_step_time": 8.198595762252808
    },
    {
      "epoch": 5.1972389221191406e-05,
      "model_forward_time": 2.367238759994507,
      "step": 54497
    },
    {
      "epoch": 5.1972389221191406e-05,
      "step": 54497,
      "training_step_time": 9.336786985397339
    },
    {
      "epoch": 5.197334289550781e-05,
      "model_forward_time": 1.8233954906463623,
      "step": 54498
    },
    {
      "epoch": 5.197334289550781e-05,
      "step": 54498,
      "training_step_time": 7.8505470752716064
    },
    {
      "epoch": 5.197429656982422e-05,
      "model_forward_time": 1.827913761138916,
      "step": 54499
    },
    {
      "epoch": 5.197429656982422e-05,
      "step": 54499,
      "training_step_time": 7.621181011199951
    },
    {
      "epoch": 5.1975250244140625e-05,
      "grad_norm": 0.1044745072722435,
      "learning_rate": 4.669712064023791e-06,
      "loss": 0.0579,
      "step": 54500
    },
    {
      "epoch": 5.1975250244140625e-05,
      "model_forward_time": 1.8232181072235107,
      "step": 54500
    },
    {
      "dynamics_loss_avg": 0.06487645506858826,
      "epoch": 5.1975250244140625e-05,
      "step": 54500
    },
    {
      "action_loss_avg": 0.0029706297034863384,
      "epoch": 5.1975250244140625e-05,
      "step": 54500
    },
    {
      "epoch": 5.1975250244140625e-05,
      "step": 54500,
      "training_step_time": 7.260154962539673
    },
    {
      "epoch": 5.197620391845703e-05,
      "model_forward_time": 1.8229663372039795,
      "step": 54501
    },
    {
      "epoch": 5.197620391845703e-05,
      "step": 54501,
      "training_step_time": 6.700626373291016
    },
    {
      "epoch": 5.197715759277344e-05,
      "model_forward_time": 1.8178739547729492,
      "step": 54502
    },
    {
      "epoch": 5.197715759277344e-05,
      "step": 54502,
      "training_step_time": 6.706473112106323
    },
    {
      "epoch": 5.1978111267089844e-05,
      "model_forward_time": 1.8309226036071777,
      "step": 54503
    },
    {
      "epoch": 5.1978111267089844e-05,
      "step": 54503,
      "training_step_time": 6.7292022705078125
    },
    {
      "epoch": 5.197906494140625e-05,
      "model_forward_time": 1.826889991760254,
      "step": 54504
    },
    {
      "epoch": 5.197906494140625e-05,
      "step": 54504,
      "training_step_time": 7.186825752258301
    },
    {
      "epoch": 5.198001861572266e-05,
      "model_forward_time": 1.7999038696289062,
      "step": 54505
    },
    {
      "epoch": 5.198001861572266e-05,
      "step": 54505,
      "training_step_time": 7.639198541641235
    },
    {
      "epoch": 5.198097229003906e-05,
      "model_forward_time": 2.3518245220184326,
      "step": 54506
    },
    {
      "epoch": 5.198097229003906e-05,
      "step": 54506,
      "training_step_time": 10.31830382347107
    },
    {
      "epoch": 5.198192596435547e-05,
      "model_forward_time": 1.799863338470459,
      "step": 54507
    },
    {
      "epoch": 5.198192596435547e-05,
      "step": 54507,
      "training_step_time": 9.181315660476685
    },
    {
      "epoch": 5.198287963867188e-05,
      "model_forward_time": 1.8139512538909912,
      "step": 54508
    },
    {
      "epoch": 5.198287963867188e-05,
      "step": 54508,
      "training_step_time": 7.736125946044922
    },
    {
      "epoch": 5.198383331298828e-05,
      "model_forward_time": 1.815091848373413,
      "step": 54509
    },
    {
      "epoch": 5.198383331298828e-05,
      "step": 54509,
      "training_step_time": 7.429750919342041
    },
    {
      "epoch": 5.1984786987304686e-05,
      "grad_norm": 0.1144738644361496,
      "learning_rate": 4.6680622237846025e-06,
      "loss": 0.0586,
      "step": 54510
    },
    {
      "epoch": 5.1984786987304686e-05,
      "model_forward_time": 1.811103105545044,
      "step": 54510
    },
    {
      "dynamics_loss_avg": 0.04246235992759466,
      "epoch": 5.1984786987304686e-05,
      "step": 54510
    },
    {
      "action_loss_avg": 0.000887978175887838,
      "epoch": 5.1984786987304686e-05,
      "step": 54510
    },
    {
      "epoch": 5.1984786987304686e-05,
      "step": 54510,
      "training_step_time": 6.847656011581421
    },
    {
      "epoch": 5.1985740661621096e-05,
      "model_forward_time": 1.819180965423584,
      "step": 54511
    },
    {
      "epoch": 5.1985740661621096e-05,
      "step": 54511,
      "training_step_time": 6.720795392990112
    },
    {
      "epoch": 5.19866943359375e-05,
      "model_forward_time": 1.8163325786590576,
      "step": 54512
    },
    {
      "epoch": 5.19866943359375e-05,
      "step": 54512,
      "training_step_time": 6.706704139709473
    },
    {
      "epoch": 5.1987648010253904e-05,
      "model_forward_time": 1.7799403667449951,
      "step": 54513
    },
    {
      "epoch": 5.1987648010253904e-05,
      "step": 54513,
      "training_step_time": 7.154372215270996
    },
    {
      "epoch": 5.1988601684570315e-05,
      "model_forward_time": 1.8275210857391357,
      "step": 54514
    },
    {
      "epoch": 5.1988601684570315e-05,
      "step": 54514,
      "training_step_time": 8.386853694915771
    },
    {
      "epoch": 5.198955535888672e-05,
      "model_forward_time": 2.375809907913208,
      "step": 54515
    },
    {
      "epoch": 5.198955535888672e-05,
      "step": 54515,
      "training_step_time": 8.61599850654602
    },
    {
      "epoch": 5.199050903320312e-05,
      "model_forward_time": 1.8109517097473145,
      "step": 54516
    },
    {
      "epoch": 5.199050903320312e-05,
      "step": 54516,
      "training_step_time": 7.837841033935547
    },
    {
      "epoch": 5.1991462707519534e-05,
      "model_forward_time": 1.8153045177459717,
      "step": 54517
    },
    {
      "epoch": 5.1991462707519534e-05,
      "step": 54517,
      "training_step_time": 8.661668539047241
    },
    {
      "epoch": 5.199241638183594e-05,
      "model_forward_time": 1.8089444637298584,
      "step": 54518
    },
    {
      "epoch": 5.199241638183594e-05,
      "step": 54518,
      "training_step_time": 7.535583019256592
    },
    {
      "epoch": 5.199337005615234e-05,
      "model_forward_time": 1.8035681247711182,
      "step": 54519
    },
    {
      "epoch": 5.199337005615234e-05,
      "step": 54519,
      "training_step_time": 7.253887414932251
    },
    {
      "epoch": 5.199432373046875e-05,
      "grad_norm": 0.09210379421710968,
      "learning_rate": 4.666412419845627e-06,
      "loss": 0.0612,
      "step": 54520
    },
    {
      "epoch": 5.199432373046875e-05,
      "model_forward_time": 1.8158462047576904,
      "step": 54520
    },
    {
      "dynamics_loss_avg": 0.0553392780944705,
      "epoch": 5.199432373046875e-05,
      "step": 54520
    },
    {
      "action_loss_avg": 0.002745743084233254,
      "epoch": 5.199432373046875e-05,
      "step": 54520
    },
    {
      "epoch": 5.199432373046875e-05,
      "step": 54520,
      "training_step_time": 6.713024377822876
    },
    {
      "epoch": 5.199527740478516e-05,
      "model_forward_time": 1.8304052352905273,
      "step": 54521
    },
    {
      "epoch": 5.199527740478516e-05,
      "step": 54521,
      "training_step_time": 6.698627233505249
    },
    {
      "epoch": 5.199623107910156e-05,
      "model_forward_time": 1.460376262664795,
      "step": 54522
    },
    {
      "epoch": 5.199623107910156e-05,
      "step": 54522,
      "training_step_time": 7.275449752807617
    },
    {
      "epoch": 5.199718475341797e-05,
      "model_forward_time": 1.82252836227417,
      "step": 54523
    },
    {
      "epoch": 5.199718475341797e-05,
      "step": 54523,
      "training_step_time": 8.655840873718262
    },
    {
      "epoch": 5.1998138427734376e-05,
      "model_forward_time": 2.3889708518981934,
      "step": 54524
    },
    {
      "epoch": 5.1998138427734376e-05,
      "step": 54524,
      "training_step_time": 8.745084524154663
    },
    {
      "epoch": 5.199909210205078e-05,
      "model_forward_time": 1.8179078102111816,
      "step": 54525
    },
    {
      "epoch": 5.199909210205078e-05,
      "step": 54525,
      "training_step_time": 8.792169094085693
    },
    {
      "epoch": 5.200004577636719e-05,
      "model_forward_time": 1.8239011764526367,
      "step": 54526
    },
    {
      "epoch": 5.200004577636719e-05,
      "step": 54526,
      "training_step_time": 8.68738842010498
    },
    {
      "epoch": 5.2000999450683595e-05,
      "model_forward_time": 1.824477195739746,
      "step": 54527
    },
    {
      "epoch": 5.2000999450683595e-05,
      "step": 54527,
      "training_step_time": 7.520592927932739
    },
    {
      "epoch": 5.2001953125e-05,
      "model_forward_time": 1.8168940544128418,
      "step": 54528
    },
    {
      "epoch": 5.2001953125e-05,
      "step": 54528,
      "training_step_time": 7.3886330127716064
    },
    {
      "epoch": 5.200290679931641e-05,
      "model_forward_time": 1.8237366676330566,
      "step": 54529
    },
    {
      "epoch": 5.200290679931641e-05,
      "step": 54529,
      "training_step_time": 6.6910622119903564
    },
    {
      "epoch": 5.2003860473632814e-05,
      "grad_norm": 0.13366413116455078,
      "learning_rate": 4.664762652387291e-06,
      "loss": 0.0618,
      "step": 54530
    },
    {
      "epoch": 5.2003860473632814e-05,
      "model_forward_time": 1.825742244720459,
      "step": 54530
    },
    {
      "dynamics_loss_avg": 0.05365965124219656,
      "epoch": 5.2003860473632814e-05,
      "step": 54530
    },
    {
      "action_loss_avg": 0.0018902306095696987,
      "epoch": 5.2003860473632814e-05,
      "step": 54530
    },
    {
      "epoch": 5.2003860473632814e-05,
      "step": 54530,
      "training_step_time": 6.742690086364746
    },
    {
      "epoch": 5.200481414794922e-05,
      "model_forward_time": 1.8212475776672363,
      "step": 54531
    },
    {
      "epoch": 5.200481414794922e-05,
      "step": 54531,
      "training_step_time": 7.1755852699279785
    },
    {
      "epoch": 5.200576782226562e-05,
      "model_forward_time": 1.8220701217651367,
      "step": 54532
    },
    {
      "epoch": 5.200576782226562e-05,
      "step": 54532,
      "training_step_time": 7.60312557220459
    },
    {
      "epoch": 5.200672149658203e-05,
      "model_forward_time": 1.822145938873291,
      "step": 54533
    },
    {
      "epoch": 5.200672149658203e-05,
      "step": 54533,
      "training_step_time": 10.254671335220337
    },
    {
      "epoch": 5.2007675170898436e-05,
      "model_forward_time": 1.8206758499145508,
      "step": 54534
    },
    {
      "epoch": 5.2007675170898436e-05,
      "step": 54534,
      "training_step_time": 10.423481464385986
    },
    {
      "epoch": 5.200862884521484e-05,
      "model_forward_time": 1.8202259540557861,
      "step": 54535
    },
    {
      "epoch": 5.200862884521484e-05,
      "step": 54535,
      "training_step_time": 8.94050407409668
    },
    {
      "epoch": 5.200958251953125e-05,
      "model_forward_time": 1.8238248825073242,
      "step": 54536
    },
    {
      "epoch": 5.200958251953125e-05,
      "step": 54536,
      "training_step_time": 7.627447605133057
    },
    {
      "epoch": 5.2010536193847655e-05,
      "model_forward_time": 1.8165831565856934,
      "step": 54537
    },
    {
      "epoch": 5.2010536193847655e-05,
      "step": 54537,
      "training_step_time": 7.418853282928467
    },
    {
      "epoch": 5.201148986816406e-05,
      "model_forward_time": 1.8206815719604492,
      "step": 54538
    },
    {
      "epoch": 5.201148986816406e-05,
      "step": 54538,
      "training_step_time": 6.755640983581543
    },
    {
      "epoch": 5.201244354248047e-05,
      "model_forward_time": 1.8227958679199219,
      "step": 54539
    },
    {
      "epoch": 5.201244354248047e-05,
      "step": 54539,
      "training_step_time": 6.714966058731079
    },
    {
      "epoch": 5.2013397216796874e-05,
      "grad_norm": 0.10172062367200851,
      "learning_rate": 4.663112921590007e-06,
      "loss": 0.0642,
      "step": 54540
    },
    {
      "epoch": 5.2013397216796874e-05,
      "model_forward_time": 1.8214428424835205,
      "step": 54540
    },
    {
      "dynamics_loss_avg": 0.056142687983810904,
      "epoch": 5.2013397216796874e-05,
      "step": 54540
    },
    {
      "action_loss_avg": 0.0014784661994781344,
      "epoch": 5.2013397216796874e-05,
      "step": 54540
    },
    {
      "epoch": 5.2013397216796874e-05,
      "step": 54540,
      "training_step_time": 7.216393709182739
    },
    {
      "epoch": 5.201435089111328e-05,
      "model_forward_time": 1.8170905113220215,
      "step": 54541
    },
    {
      "epoch": 5.201435089111328e-05,
      "step": 54541,
      "training_step_time": 7.490885019302368
    },
    {
      "epoch": 5.201530456542969e-05,
      "model_forward_time": 1.829890489578247,
      "step": 54542
    },
    {
      "epoch": 5.201530456542969e-05,
      "step": 54542,
      "training_step_time": 8.357396602630615
    },
    {
      "epoch": 5.201625823974609e-05,
      "model_forward_time": 2.3751418590545654,
      "step": 54543
    },
    {
      "epoch": 5.201625823974609e-05,
      "step": 54543,
      "training_step_time": 10.145940065383911
    },
    {
      "epoch": 5.20172119140625e-05,
      "model_forward_time": 1.8224680423736572,
      "step": 54544
    },
    {
      "epoch": 5.20172119140625e-05,
      "step": 54544,
      "training_step_time": 9.31583571434021
    },
    {
      "epoch": 5.201816558837891e-05,
      "model_forward_time": 1.815969705581665,
      "step": 54545
    },
    {
      "epoch": 5.201816558837891e-05,
      "step": 54545,
      "training_step_time": 7.57243800163269
    },
    {
      "epoch": 5.201911926269531e-05,
      "model_forward_time": 1.8218271732330322,
      "step": 54546
    },
    {
      "epoch": 5.201911926269531e-05,
      "step": 54546,
      "training_step_time": 7.355143308639526
    },
    {
      "epoch": 5.2020072937011716e-05,
      "model_forward_time": 1.8193390369415283,
      "step": 54547
    },
    {
      "epoch": 5.2020072937011716e-05,
      "step": 54547,
      "training_step_time": 6.704437255859375
    },
    {
      "epoch": 5.202102661132813e-05,
      "model_forward_time": 1.832427740097046,
      "step": 54548
    },
    {
      "epoch": 5.202102661132813e-05,
      "step": 54548,
      "training_step_time": 6.682642936706543
    },
    {
      "epoch": 5.202198028564453e-05,
      "model_forward_time": 1.8212976455688477,
      "step": 54549
    },
    {
      "epoch": 5.202198028564453e-05,
      "step": 54549,
      "training_step_time": 7.224678993225098
    },
    {
      "epoch": 5.2022933959960935e-05,
      "grad_norm": 0.11052676290273666,
      "learning_rate": 4.661463227634187e-06,
      "loss": 0.0608,
      "step": 54550
    },
    {
      "epoch": 5.2022933959960935e-05,
      "model_forward_time": 1.8342204093933105,
      "step": 54550
    },
    {
      "dynamics_loss_avg": 0.051835459657013416,
      "epoch": 5.2022933959960935e-05,
      "step": 54550
    },
    {
      "action_loss_avg": 0.0019139492069371045,
      "epoch": 5.2022933959960935e-05,
      "step": 54550
    },
    {
      "epoch": 5.2022933959960935e-05,
      "step": 54550,
      "training_step_time": 7.152349948883057
    },
    {
      "epoch": 5.2023887634277346e-05,
      "model_forward_time": 1.462156057357788,
      "step": 54551
    },
    {
      "epoch": 5.2023887634277346e-05,
      "step": 54551,
      "training_step_time": 7.626683950424194
    },
    {
      "epoch": 5.202484130859375e-05,
      "model_forward_time": 2.3926711082458496,
      "step": 54552
    },
    {
      "epoch": 5.202484130859375e-05,
      "step": 54552,
      "training_step_time": 9.328402280807495
    },
    {
      "epoch": 5.2025794982910154e-05,
      "model_forward_time": 1.8256218433380127,
      "step": 54553
    },
    {
      "epoch": 5.2025794982910154e-05,
      "step": 54553,
      "training_step_time": 7.800985097885132
    },
    {
      "epoch": 5.2026748657226564e-05,
      "model_forward_time": 1.8283724784851074,
      "step": 54554
    },
    {
      "epoch": 5.2026748657226564e-05,
      "step": 54554,
      "training_step_time": 8.496703863143921
    },
    {
      "epoch": 5.202770233154297e-05,
      "model_forward_time": 1.8287582397460938,
      "step": 54555
    },
    {
      "epoch": 5.202770233154297e-05,
      "step": 54555,
      "training_step_time": 7.605138778686523
    },
    {
      "epoch": 5.202865600585937e-05,
      "model_forward_time": 1.8321888446807861,
      "step": 54556
    },
    {
      "epoch": 5.202865600585937e-05,
      "step": 54556,
      "training_step_time": 6.720174312591553
    },
    {
      "epoch": 5.202960968017578e-05,
      "model_forward_time": 1.827852487564087,
      "step": 54557
    },
    {
      "epoch": 5.202960968017578e-05,
      "step": 54557,
      "training_step_time": 6.7031495571136475
    },
    {
      "epoch": 5.203056335449219e-05,
      "model_forward_time": 1.8310935497283936,
      "step": 54558
    },
    {
      "epoch": 5.203056335449219e-05,
      "step": 54558,
      "training_step_time": 6.78061842918396
    },
    {
      "epoch": 5.203151702880859e-05,
      "model_forward_time": 1.838287115097046,
      "step": 54559
    },
    {
      "epoch": 5.203151702880859e-05,
      "step": 54559,
      "training_step_time": 7.209735155105591
    },
    {
      "epoch": 5.2032470703125e-05,
      "grad_norm": 0.10735645145177841,
      "learning_rate": 4.6598135707002385e-06,
      "loss": 0.0629,
      "step": 54560
    },
    {
      "epoch": 5.2032470703125e-05,
      "model_forward_time": 1.8410680294036865,
      "step": 54560
    },
    {
      "dynamics_loss_avg": 0.06128524895757437,
      "epoch": 5.2032470703125e-05,
      "step": 54560
    },
    {
      "action_loss_avg": 0.0021658409968949854,
      "epoch": 5.2032470703125e-05,
      "step": 54560
    },
    {
      "epoch": 5.2032470703125e-05,
      "step": 54560,
      "training_step_time": 9.847606897354126
    },
    {
      "epoch": 5.2033424377441406e-05,
      "model_forward_time": 2.3918848037719727,
      "step": 54561
    },
    {
      "epoch": 5.2033424377441406e-05,
      "step": 54561,
      "training_step_time": 9.794743537902832
    },
    {
      "epoch": 5.203437805175781e-05,
      "model_forward_time": 1.8335356712341309,
      "step": 54562
    },
    {
      "epoch": 5.203437805175781e-05,
      "step": 54562,
      "training_step_time": 8.448825359344482
    },
    {
      "epoch": 5.203533172607422e-05,
      "model_forward_time": 1.8282299041748047,
      "step": 54563
    },
    {
      "epoch": 5.203533172607422e-05,
      "step": 54563,
      "training_step_time": 7.6096179485321045
    },
    {
      "epoch": 5.2036285400390625e-05,
      "model_forward_time": 1.4384682178497314,
      "step": 54564
    },
    {
      "epoch": 5.2036285400390625e-05,
      "step": 54564,
      "training_step_time": 7.7850563526153564
    },
    {
      "epoch": 5.203723907470703e-05,
      "model_forward_time": 1.829648733139038,
      "step": 54565
    },
    {
      "epoch": 5.203723907470703e-05,
      "step": 54565,
      "training_step_time": 7.388216733932495
    },
    {
      "epoch": 5.203819274902344e-05,
      "model_forward_time": 1.8279576301574707,
      "step": 54566
    },
    {
      "epoch": 5.203819274902344e-05,
      "step": 54566,
      "training_step_time": 6.772638320922852
    },
    {
      "epoch": 5.2039146423339844e-05,
      "model_forward_time": 1.8292601108551025,
      "step": 54567
    },
    {
      "epoch": 5.2039146423339844e-05,
      "step": 54567,
      "training_step_time": 6.7995445728302
    },
    {
      "epoch": 5.204010009765625e-05,
      "model_forward_time": 1.8263988494873047,
      "step": 54568
    },
    {
      "epoch": 5.204010009765625e-05,
      "step": 54568,
      "training_step_time": 7.471735239028931
    },
    {
      "epoch": 5.204105377197266e-05,
      "model_forward_time": 1.825225830078125,
      "step": 54569
    },
    {
      "epoch": 5.204105377197266e-05,
      "step": 54569,
      "training_step_time": 9.291526317596436
    },
    {
      "epoch": 5.204200744628906e-05,
      "grad_norm": 0.1324559897184372,
      "learning_rate": 4.658163950968567e-06,
      "loss": 0.0617,
      "step": 54570
    },
    {
      "epoch": 5.204200744628906e-05,
      "model_forward_time": 2.367445468902588,
      "step": 54570
    },
    {
      "dynamics_loss_avg": 0.04267939664423466,
      "epoch": 5.204200744628906e-05,
      "step": 54570
    },
    {
      "action_loss_avg": 0.0017160132032586262,
      "epoch": 5.204200744628906e-05,
      "step": 54570
    },
    {
      "epoch": 5.204200744628906e-05,
      "step": 54570,
      "training_step_time": 8.230530977249146
    },
    {
      "epoch": 5.204296112060547e-05,
      "model_forward_time": 1.829984188079834,
      "step": 54571
    },
    {
      "epoch": 5.204296112060547e-05,
      "step": 54571,
      "training_step_time": 9.406513214111328
    },
    {
      "epoch": 5.204391479492188e-05,
      "model_forward_time": 1.824937343597412,
      "step": 54572
    },
    {
      "epoch": 5.204391479492188e-05,
      "step": 54572,
      "training_step_time": 7.686927318572998
    },
    {
      "epoch": 5.204486846923828e-05,
      "model_forward_time": 1.8270907402038574,
      "step": 54573
    },
    {
      "epoch": 5.204486846923828e-05,
      "step": 54573,
      "training_step_time": 7.914450645446777
    },
    {
      "epoch": 5.2045822143554686e-05,
      "model_forward_time": 1.824059009552002,
      "step": 54574
    },
    {
      "epoch": 5.2045822143554686e-05,
      "step": 54574,
      "training_step_time": 7.482990503311157
    },
    {
      "epoch": 5.2046775817871096e-05,
      "model_forward_time": 1.827981948852539,
      "step": 54575
    },
    {
      "epoch": 5.2046775817871096e-05,
      "step": 54575,
      "training_step_time": 6.803409814834595
    },
    {
      "epoch": 5.20477294921875e-05,
      "model_forward_time": 1.4539966583251953,
      "step": 54576
    },
    {
      "epoch": 5.20477294921875e-05,
      "step": 54576,
      "training_step_time": 6.826688766479492
    },
    {
      "epoch": 5.2048683166503905e-05,
      "model_forward_time": 1.8249964714050293,
      "step": 54577
    },
    {
      "epoch": 5.2048683166503905e-05,
      "step": 54577,
      "training_step_time": 7.590309143066406
    },
    {
      "epoch": 5.2049636840820315e-05,
      "model_forward_time": 1.8321905136108398,
      "step": 54578
    },
    {
      "epoch": 5.2049636840820315e-05,
      "step": 54578,
      "training_step_time": 8.736285209655762
    },
    {
      "epoch": 5.205059051513672e-05,
      "model_forward_time": 1.82899808883667,
      "step": 54579
    },
    {
      "epoch": 5.205059051513672e-05,
      "step": 54579,
      "training_step_time": 8.19840955734253
    },
    {
      "epoch": 5.205154418945312e-05,
      "grad_norm": 0.11143726110458374,
      "learning_rate": 4.656514368619572e-06,
      "loss": 0.0632,
      "step": 54580
    },
    {
      "epoch": 5.205154418945312e-05,
      "model_forward_time": 2.3168118000030518,
      "step": 54580
    },
    {
      "dynamics_loss_avg": 0.07566801467910408,
      "epoch": 5.205154418945312e-05,
      "step": 54580
    },
    {
      "action_loss_avg": 0.0027155522839166224,
      "epoch": 5.205154418945312e-05,
      "step": 54580
    },
    {
      "epoch": 5.205154418945312e-05,
      "step": 54580,
      "training_step_time": 7.8040454387664795
    },
    {
      "epoch": 5.2052497863769534e-05,
      "model_forward_time": 1.7281408309936523,
      "step": 54581
    },
    {
      "epoch": 5.2052497863769534e-05,
      "step": 54581,
      "training_step_time": 7.830021858215332
    },
    {
      "epoch": 5.205345153808594e-05,
      "model_forward_time": 1.731569528579712,
      "step": 54582
    },
    {
      "epoch": 5.205345153808594e-05,
      "step": 54582,
      "training_step_time": 7.786969184875488
    },
    {
      "epoch": 5.205440521240234e-05,
      "model_forward_time": 1.727783203125,
      "step": 54583
    },
    {
      "epoch": 5.205440521240234e-05,
      "step": 54583,
      "training_step_time": 7.5187156200408936
    },
    {
      "epoch": 5.205535888671875e-05,
      "model_forward_time": 1.737360954284668,
      "step": 54584
    },
    {
      "epoch": 5.205535888671875e-05,
      "step": 54584,
      "training_step_time": 6.778709650039673
    },
    {
      "epoch": 5.205631256103516e-05,
      "model_forward_time": 1.7350471019744873,
      "step": 54585
    },
    {
      "epoch": 5.205631256103516e-05,
      "step": 54585,
      "training_step_time": 6.769761800765991
    },
    {
      "epoch": 5.205726623535156e-05,
      "model_forward_time": 1.7332754135131836,
      "step": 54586
    },
    {
      "epoch": 5.205726623535156e-05,
      "step": 54586,
      "training_step_time": 7.3969104290008545
    },
    {
      "epoch": 5.205821990966797e-05,
      "model_forward_time": 1.7315618991851807,
      "step": 54587
    },
    {
      "epoch": 5.205821990966797e-05,
      "step": 54587,
      "training_step_time": 7.563337802886963
    },
    {
      "epoch": 5.2059173583984376e-05,
      "model_forward_time": 1.781975269317627,
      "step": 54588
    },
    {
      "epoch": 5.2059173583984376e-05,
      "step": 54588,
      "training_step_time": 7.710415840148926
    },
    {
      "epoch": 5.206012725830078e-05,
      "model_forward_time": 2.2871174812316895,
      "step": 54589
    },
    {
      "epoch": 5.206012725830078e-05,
      "step": 54589,
      "training_step_time": 9.378813028335571
    },
    {
      "epoch": 5.206108093261719e-05,
      "grad_norm": 0.15490615367889404,
      "learning_rate": 4.65486482383365e-06,
      "loss": 0.0575,
      "step": 54590
    },
    {
      "epoch": 5.206108093261719e-05,
      "model_forward_time": 1.3643803596496582,
      "step": 54590
    },
    {
      "dynamics_loss_avg": 0.0530835272744298,
      "epoch": 5.206108093261719e-05,
      "step": 54590
    },
    {
      "action_loss_avg": 0.0018528815649915486,
      "epoch": 5.206108093261719e-05,
      "step": 54590
    },
    {
      "epoch": 5.206108093261719e-05,
      "step": 54590,
      "training_step_time": 8.608462810516357
    },
    {
      "epoch": 5.2062034606933595e-05,
      "model_forward_time": 1.7323153018951416,
      "step": 54591
    },
    {
      "epoch": 5.2062034606933595e-05,
      "step": 54591,
      "training_step_time": 7.630457878112793
    },
    {
      "epoch": 5.206298828125e-05,
      "model_forward_time": 1.7362644672393799,
      "step": 54592
    },
    {
      "epoch": 5.206298828125e-05,
      "step": 54592,
      "training_step_time": 7.875152826309204
    },
    {
      "epoch": 5.206394195556641e-05,
      "model_forward_time": 1.7432246208190918,
      "step": 54593
    },
    {
      "epoch": 5.206394195556641e-05,
      "step": 54593,
      "training_step_time": 7.27424430847168
    },
    {
      "epoch": 5.2064895629882814e-05,
      "model_forward_time": 1.7276794910430908,
      "step": 54594
    },
    {
      "epoch": 5.2064895629882814e-05,
      "step": 54594,
      "training_step_time": 6.757645130157471
    },
    {
      "epoch": 5.206584930419922e-05,
      "model_forward_time": 1.7318310737609863,
      "step": 54595
    },
    {
      "epoch": 5.206584930419922e-05,
      "step": 54595,
      "training_step_time": 7.47497820854187
    },
    {
      "epoch": 5.206680297851562e-05,
      "model_forward_time": 1.7368700504302979,
      "step": 54596
    },
    {
      "epoch": 5.206680297851562e-05,
      "step": 54596,
      "training_step_time": 7.259562730789185
    },
    {
      "epoch": 5.206775665283203e-05,
      "model_forward_time": 1.7788872718811035,
      "step": 54597
    },
    {
      "epoch": 5.206775665283203e-05,
      "step": 54597,
      "training_step_time": 8.743476629257202
    },
    {
      "epoch": 5.2068710327148437e-05,
      "model_forward_time": 2.311394691467285,
      "step": 54598
    },
    {
      "epoch": 5.2068710327148437e-05,
      "step": 54598,
      "training_step_time": 8.755714178085327
    },
    {
      "epoch": 5.206966400146484e-05,
      "model_forward_time": 1.735351800918579,
      "step": 54599
    },
    {
      "epoch": 5.206966400146484e-05,
      "step": 54599,
      "training_step_time": 8.809130191802979
    },
    {
      "epoch": 5.207061767578125e-05,
      "grad_norm": 0.13730774819850922,
      "learning_rate": 4.6532153167911904e-06,
      "loss": 0.0603,
      "step": 54600
    },
    {
      "epoch": 5.207061767578125e-05,
      "model_forward_time": 1.73594331741333,
      "step": 54600
    },
    {
      "dynamics_loss_avg": 0.06667821295559406,
      "epoch": 5.207061767578125e-05,
      "step": 54600
    },
    {
      "action_loss_avg": 0.0023423407576046882,
      "epoch": 5.207061767578125e-05,
      "step": 54600
    },
    {
      "epoch": 5.207061767578125e-05,
      "step": 54600,
      "training_step_time": 7.583884954452515
    },
    {
      "epoch": 5.2071571350097655e-05,
      "model_forward_time": 1.7292389869689941,
      "step": 54601
    },
    {
      "epoch": 5.2071571350097655e-05,
      "step": 54601,
      "training_step_time": 7.741496801376343
    },
    {
      "epoch": 5.207252502441406e-05,
      "model_forward_time": 1.7315456867218018,
      "step": 54602
    },
    {
      "epoch": 5.207252502441406e-05,
      "step": 54602,
      "training_step_time": 7.385716199874878
    },
    {
      "epoch": 5.207347869873047e-05,
      "model_forward_time": 1.765073299407959,
      "step": 54603
    },
    {
      "epoch": 5.207347869873047e-05,
      "step": 54603,
      "training_step_time": 6.760490417480469
    },
    {
      "epoch": 5.2074432373046874e-05,
      "model_forward_time": 1.7347419261932373,
      "step": 54604
    },
    {
      "epoch": 5.2074432373046874e-05,
      "step": 54604,
      "training_step_time": 7.489858388900757
    },
    {
      "epoch": 5.207538604736328e-05,
      "model_forward_time": 1.73895263671875,
      "step": 54605
    },
    {
      "epoch": 5.207538604736328e-05,
      "step": 54605,
      "training_step_time": 7.158356428146362
    },
    {
      "epoch": 5.207633972167969e-05,
      "model_forward_time": 1.7730920314788818,
      "step": 54606
    },
    {
      "epoch": 5.207633972167969e-05,
      "step": 54606,
      "training_step_time": 8.76201343536377
    },
    {
      "epoch": 5.207729339599609e-05,
      "model_forward_time": 2.279325008392334,
      "step": 54607
    },
    {
      "epoch": 5.207729339599609e-05,
      "step": 54607,
      "training_step_time": 9.543640851974487
    },
    {
      "epoch": 5.20782470703125e-05,
      "model_forward_time": 1.770751953125,
      "step": 54608
    },
    {
      "epoch": 5.20782470703125e-05,
      "step": 54608,
      "training_step_time": 8.762500524520874
    },
    {
      "epoch": 5.207920074462891e-05,
      "model_forward_time": 1.7307453155517578,
      "step": 54609
    },
    {
      "epoch": 5.207920074462891e-05,
      "step": 54609,
      "training_step_time": 8.25292706489563
    },
    {
      "epoch": 5.208015441894531e-05,
      "grad_norm": 0.09367505460977554,
      "learning_rate": 4.651565847672584e-06,
      "loss": 0.0652,
      "step": 54610
    },
    {
      "epoch": 5.208015441894531e-05,
      "model_forward_time": 1.7305247783660889,
      "step": 54610
    },
    {
      "dynamics_loss_avg": 0.05927540026605129,
      "epoch": 5.208015441894531e-05,
      "step": 54610
    },
    {
      "action_loss_avg": 0.0010205311817117035,
      "epoch": 5.208015441894531e-05,
      "step": 54610
    },
    {
      "epoch": 5.208015441894531e-05,
      "step": 54610,
      "training_step_time": 7.863550901412964
    },
    {
      "epoch": 5.2081108093261716e-05,
      "model_forward_time": 1.737706184387207,
      "step": 54611
    },
    {
      "epoch": 5.2081108093261716e-05,
      "step": 54611,
      "training_step_time": 7.4370667934417725
    },
    {
      "epoch": 5.208206176757813e-05,
      "model_forward_time": 1.7349662780761719,
      "step": 54612
    },
    {
      "epoch": 5.208206176757813e-05,
      "step": 54612,
      "training_step_time": 6.789499998092651
    },
    {
      "epoch": 5.208301544189453e-05,
      "model_forward_time": 1.7375741004943848,
      "step": 54613
    },
    {
      "epoch": 5.208301544189453e-05,
      "step": 54613,
      "training_step_time": 7.387429475784302
    },
    {
      "epoch": 5.2083969116210935e-05,
      "model_forward_time": 1.7362456321716309,
      "step": 54614
    },
    {
      "epoch": 5.2083969116210935e-05,
      "step": 54614,
      "training_step_time": 7.049800157546997
    },
    {
      "epoch": 5.2084922790527346e-05,
      "model_forward_time": 1.7344872951507568,
      "step": 54615
    },
    {
      "epoch": 5.2084922790527346e-05,
      "step": 54615,
      "training_step_time": 8.307964086532593
    },
    {
      "epoch": 5.208587646484375e-05,
      "model_forward_time": 1.7656595706939697,
      "step": 54616
    },
    {
      "epoch": 5.208587646484375e-05,
      "step": 54616,
      "training_step_time": 8.386661767959595
    },
    {
      "epoch": 5.2086830139160154e-05,
      "model_forward_time": 2.2935562133789062,
      "step": 54617
    },
    {
      "epoch": 5.2086830139160154e-05,
      "step": 54617,
      "training_step_time": 9.023199796676636
    },
    {
      "epoch": 5.2087783813476565e-05,
      "model_forward_time": 1.7338061332702637,
      "step": 54618
    },
    {
      "epoch": 5.2087783813476565e-05,
      "step": 54618,
      "training_step_time": 9.54135513305664
    },
    {
      "epoch": 5.208873748779297e-05,
      "model_forward_time": 1.7320737838745117,
      "step": 54619
    },
    {
      "epoch": 5.208873748779297e-05,
      "step": 54619,
      "training_step_time": 7.8022894859313965
    },
    {
      "epoch": 5.208969116210937e-05,
      "grad_norm": 0.11612745374441147,
      "learning_rate": 4.649916416658212e-06,
      "loss": 0.0553,
      "step": 54620
    },
    {
      "epoch": 5.208969116210937e-05,
      "model_forward_time": 1.7332439422607422,
      "step": 54620
    },
    {
      "dynamics_loss_avg": 0.056969990395009516,
      "epoch": 5.208969116210937e-05,
      "step": 54620
    },
    {
      "action_loss_avg": 0.0018551833927631378,
      "epoch": 5.208969116210937e-05,
      "step": 54620
    },
    {
      "epoch": 5.208969116210937e-05,
      "step": 54620,
      "training_step_time": 7.265466213226318
    },
    {
      "epoch": 5.209064483642578e-05,
      "model_forward_time": 1.740499496459961,
      "step": 54621
    },
    {
      "epoch": 5.209064483642578e-05,
      "step": 54621,
      "training_step_time": 6.801290035247803
    },
    {
      "epoch": 5.209159851074219e-05,
      "model_forward_time": 1.7363243103027344,
      "step": 54622
    },
    {
      "epoch": 5.209159851074219e-05,
      "step": 54622,
      "training_step_time": 7.411790370941162
    },
    {
      "epoch": 5.209255218505859e-05,
      "model_forward_time": 1.7360153198242188,
      "step": 54623
    },
    {
      "epoch": 5.209255218505859e-05,
      "step": 54623,
      "training_step_time": 7.050810098648071
    },
    {
      "epoch": 5.2093505859375e-05,
      "model_forward_time": 1.7366933822631836,
      "step": 54624
    },
    {
      "epoch": 5.2093505859375e-05,
      "step": 54624,
      "training_step_time": 8.564002513885498
    },
    {
      "epoch": 5.2094459533691406e-05,
      "model_forward_time": 1.7640256881713867,
      "step": 54625
    },
    {
      "epoch": 5.2094459533691406e-05,
      "step": 54625,
      "training_step_time": 9.248211860656738
    },
    {
      "epoch": 5.209541320800781e-05,
      "model_forward_time": 2.283834218978882,
      "step": 54626
    },
    {
      "epoch": 5.209541320800781e-05,
      "step": 54626,
      "training_step_time": 9.490877151489258
    },
    {
      "epoch": 5.209636688232422e-05,
      "model_forward_time": 1.7376389503479004,
      "step": 54627
    },
    {
      "epoch": 5.209636688232422e-05,
      "step": 54627,
      "training_step_time": 8.518317222595215
    },
    {
      "epoch": 5.2097320556640625e-05,
      "model_forward_time": 1.7341864109039307,
      "step": 54628
    },
    {
      "epoch": 5.2097320556640625e-05,
      "step": 54628,
      "training_step_time": 7.581883430480957
    },
    {
      "epoch": 5.209827423095703e-05,
      "model_forward_time": 1.737001895904541,
      "step": 54629
    },
    {
      "epoch": 5.209827423095703e-05,
      "step": 54629,
      "training_step_time": 7.6845316886901855
    },
    {
      "epoch": 5.209922790527344e-05,
      "grad_norm": 0.1687576025724411,
      "learning_rate": 4.648267023928455e-06,
      "loss": 0.061,
      "step": 54630
    },
    {
      "epoch": 5.209922790527344e-05,
      "model_forward_time": 1.7340342998504639,
      "step": 54630
    },
    {
      "dynamics_loss_avg": 0.058981760032474996,
      "epoch": 5.209922790527344e-05,
      "step": 54630
    },
    {
      "action_loss_avg": 0.002523663971805945,
      "epoch": 5.209922790527344e-05,
      "step": 54630
    },
    {
      "epoch": 5.209922790527344e-05,
      "step": 54630,
      "training_step_time": 7.17418909072876
    },
    {
      "epoch": 5.2100181579589844e-05,
      "model_forward_time": 1.733985424041748,
      "step": 54631
    },
    {
      "epoch": 5.2100181579589844e-05,
      "step": 54631,
      "training_step_time": 7.4061806201934814
    },
    {
      "epoch": 5.210113525390625e-05,
      "model_forward_time": 1.7344090938568115,
      "step": 54632
    },
    {
      "epoch": 5.210113525390625e-05,
      "step": 54632,
      "training_step_time": 6.979034423828125
    },
    {
      "epoch": 5.210208892822266e-05,
      "model_forward_time": 1.731889247894287,
      "step": 54633
    },
    {
      "epoch": 5.210208892822266e-05,
      "step": 54633,
      "training_step_time": 7.5256078243255615
    },
    {
      "epoch": 5.210304260253906e-05,
      "model_forward_time": 1.7740204334259033,
      "step": 54634
    },
    {
      "epoch": 5.210304260253906e-05,
      "step": 54634,
      "training_step_time": 8.814915418624878
    },
    {
      "epoch": 5.210399627685547e-05,
      "model_forward_time": 2.274237871170044,
      "step": 54635
    },
    {
      "epoch": 5.210399627685547e-05,
      "step": 54635,
      "training_step_time": 9.189589262008667
    },
    {
      "epoch": 5.210494995117188e-05,
      "model_forward_time": 1.7308111190795898,
      "step": 54636
    },
    {
      "epoch": 5.210494995117188e-05,
      "step": 54636,
      "training_step_time": 8.787433624267578
    },
    {
      "epoch": 5.210590362548828e-05,
      "model_forward_time": 1.7384090423583984,
      "step": 54637
    },
    {
      "epoch": 5.210590362548828e-05,
      "step": 54637,
      "training_step_time": 8.17069697380066
    },
    {
      "epoch": 5.2106857299804686e-05,
      "model_forward_time": 1.7350690364837646,
      "step": 54638
    },
    {
      "epoch": 5.2106857299804686e-05,
      "step": 54638,
      "training_step_time": 7.896388530731201
    },
    {
      "epoch": 5.2107810974121097e-05,
      "model_forward_time": 1.7282755374908447,
      "step": 54639
    },
    {
      "epoch": 5.2107810974121097e-05,
      "step": 54639,
      "training_step_time": 6.826007127761841
    },
    {
      "epoch": 5.21087646484375e-05,
      "grad_norm": 0.11572004109621048,
      "learning_rate": 4.646617669663687e-06,
      "loss": 0.0593,
      "step": 54640
    },
    {
      "epoch": 5.21087646484375e-05,
      "model_forward_time": 1.7201917171478271,
      "step": 54640
    },
    {
      "dynamics_loss_avg": 0.05161555716767907,
      "epoch": 5.21087646484375e-05,
      "step": 54640
    },
    {
      "action_loss_avg": 0.001414275857678149,
      "epoch": 5.21087646484375e-05,
      "step": 54640
    },
    {
      "epoch": 5.21087646484375e-05,
      "step": 54640,
      "training_step_time": 7.510416746139526
    },
    {
      "epoch": 5.2109718322753905e-05,
      "model_forward_time": 1.7301270961761475,
      "step": 54641
    },
    {
      "epoch": 5.2109718322753905e-05,
      "step": 54641,
      "training_step_time": 6.812493324279785
    },
    {
      "epoch": 5.2110671997070315e-05,
      "model_forward_time": 1.7317252159118652,
      "step": 54642
    },
    {
      "epoch": 5.2110671997070315e-05,
      "step": 54642,
      "training_step_time": 7.59444260597229
    },
    {
      "epoch": 5.211162567138672e-05,
      "model_forward_time": 1.7645456790924072,
      "step": 54643
    },
    {
      "epoch": 5.211162567138672e-05,
      "step": 54643,
      "training_step_time": 8.459492206573486
    },
    {
      "epoch": 5.2112579345703123e-05,
      "model_forward_time": 2.2722010612487793,
      "step": 54644
    },
    {
      "epoch": 5.2112579345703123e-05,
      "step": 54644,
      "training_step_time": 8.39115047454834
    },
    {
      "epoch": 5.2113533020019534e-05,
      "model_forward_time": 1.7337687015533447,
      "step": 54645
    },
    {
      "epoch": 5.2113533020019534e-05,
      "step": 54645,
      "training_step_time": 8.239207029342651
    },
    {
      "epoch": 5.211448669433594e-05,
      "model_forward_time": 1.7288446426391602,
      "step": 54646
    },
    {
      "epoch": 5.211448669433594e-05,
      "step": 54646,
      "training_step_time": 8.384793281555176
    },
    {
      "epoch": 5.211544036865234e-05,
      "model_forward_time": 1.7333664894104004,
      "step": 54647
    },
    {
      "epoch": 5.211544036865234e-05,
      "step": 54647,
      "training_step_time": 7.594715356826782
    },
    {
      "epoch": 5.211639404296875e-05,
      "model_forward_time": 1.7307891845703125,
      "step": 54648
    },
    {
      "epoch": 5.211639404296875e-05,
      "step": 54648,
      "training_step_time": 7.762322425842285
    },
    {
      "epoch": 5.211734771728516e-05,
      "model_forward_time": 1.731370449066162,
      "step": 54649
    },
    {
      "epoch": 5.211734771728516e-05,
      "step": 54649,
      "training_step_time": 7.401207447052002
    },
    {
      "epoch": 5.211830139160156e-05,
      "grad_norm": 0.07962484657764435,
      "learning_rate": 4.644968354044277e-06,
      "loss": 0.0582,
      "step": 54650
    },
    {
      "epoch": 5.211830139160156e-05,
      "model_forward_time": 1.3534212112426758,
      "step": 54650
    },
    {
      "dynamics_loss_avg": 0.07186590507626534,
      "epoch": 5.211830139160156e-05,
      "step": 54650
    },
    {
      "action_loss_avg": 0.002098316268529743,
      "epoch": 5.211830139160156e-05,
      "step": 54650
    },
    {
      "epoch": 5.211830139160156e-05,
      "step": 54650,
      "training_step_time": 6.7795329093933105
    },
    {
      "epoch": 5.211925506591797e-05,
      "model_forward_time": 1.7226428985595703,
      "step": 54651
    },
    {
      "epoch": 5.211925506591797e-05,
      "step": 54651,
      "training_step_time": 7.62254524230957
    },
    {
      "epoch": 5.2120208740234376e-05,
      "model_forward_time": 1.781341552734375,
      "step": 54652
    },
    {
      "epoch": 5.2120208740234376e-05,
      "step": 54652,
      "training_step_time": 7.6720709800720215
    },
    {
      "epoch": 5.212116241455078e-05,
      "model_forward_time": 2.2702388763427734,
      "step": 54653
    },
    {
      "epoch": 5.212116241455078e-05,
      "step": 54653,
      "training_step_time": 10.645848512649536
    },
    {
      "epoch": 5.212211608886719e-05,
      "model_forward_time": 1.734708547592163,
      "step": 54654
    },
    {
      "epoch": 5.212211608886719e-05,
      "step": 54654,
      "training_step_time": 8.612813949584961
    },
    {
      "epoch": 5.2123069763183595e-05,
      "model_forward_time": 1.7340943813323975,
      "step": 54655
    },
    {
      "epoch": 5.2123069763183595e-05,
      "step": 54655,
      "training_step_time": 7.655671119689941
    },
    {
      "epoch": 5.21240234375e-05,
      "model_forward_time": 1.732701063156128,
      "step": 54656
    },
    {
      "epoch": 5.21240234375e-05,
      "step": 54656,
      "training_step_time": 7.598352432250977
    },
    {
      "epoch": 5.21249771118164e-05,
      "model_forward_time": 1.751793622970581,
      "step": 54657
    },
    {
      "epoch": 5.21249771118164e-05,
      "step": 54657,
      "training_step_time": 7.852527141571045
    },
    {
      "epoch": 5.2125930786132814e-05,
      "model_forward_time": 1.7314693927764893,
      "step": 54658
    },
    {
      "epoch": 5.2125930786132814e-05,
      "step": 54658,
      "training_step_time": 7.353212118148804
    },
    {
      "epoch": 5.212688446044922e-05,
      "model_forward_time": 1.7324192523956299,
      "step": 54659
    },
    {
      "epoch": 5.212688446044922e-05,
      "step": 54659,
      "training_step_time": 6.76292872428894
    },
    {
      "epoch": 5.212783813476562e-05,
      "grad_norm": 0.11354178935289383,
      "learning_rate": 4.643319077250596e-06,
      "loss": 0.0565,
      "step": 54660
    },
    {
      "epoch": 5.212783813476562e-05,
      "model_forward_time": 1.724780559539795,
      "step": 54660
    },
    {
      "dynamics_loss_avg": 0.05703513938933611,
      "epoch": 5.212783813476562e-05,
      "step": 54660
    },
    {
      "action_loss_avg": 0.0012797597475582734,
      "epoch": 5.212783813476562e-05,
      "step": 54660
    },
    {
      "epoch": 5.212783813476562e-05,
      "step": 54660,
      "training_step_time": 8.094101905822754
    },
    {
      "epoch": 5.212879180908203e-05,
      "model_forward_time": 1.777395248413086,
      "step": 54661
    },
    {
      "epoch": 5.212879180908203e-05,
      "step": 54661,
      "training_step_time": 8.710609436035156
    },
    {
      "epoch": 5.2129745483398437e-05,
      "model_forward_time": 1.7393085956573486,
      "step": 54662
    },
    {
      "epoch": 5.2129745483398437e-05,
      "step": 54662,
      "training_step_time": 9.144975900650024
    },
    {
      "epoch": 5.213069915771484e-05,
      "model_forward_time": 2.288285255432129,
      "step": 54663
    },
    {
      "epoch": 5.213069915771484e-05,
      "step": 54663,
      "training_step_time": 8.74789810180664
    },
    {
      "epoch": 5.213165283203125e-05,
      "model_forward_time": 1.7395858764648438,
      "step": 54664
    },
    {
      "epoch": 5.213165283203125e-05,
      "step": 54664,
      "training_step_time": 8.736984014511108
    },
    {
      "epoch": 5.2132606506347655e-05,
      "model_forward_time": 1.735856056213379,
      "step": 54665
    },
    {
      "epoch": 5.2132606506347655e-05,
      "step": 54665,
      "training_step_time": 7.406314849853516
    },
    {
      "epoch": 5.213356018066406e-05,
      "model_forward_time": 1.735489845275879,
      "step": 54666
    },
    {
      "epoch": 5.213356018066406e-05,
      "step": 54666,
      "training_step_time": 7.62470006942749
    },
    {
      "epoch": 5.213451385498047e-05,
      "model_forward_time": 1.6986944675445557,
      "step": 54667
    },
    {
      "epoch": 5.213451385498047e-05,
      "step": 54667,
      "training_step_time": 7.3550074100494385
    },
    {
      "epoch": 5.2135467529296874e-05,
      "model_forward_time": 1.7217998504638672,
      "step": 54668
    },
    {
      "epoch": 5.2135467529296874e-05,
      "step": 54668,
      "training_step_time": 7.486636161804199
    },
    {
      "epoch": 5.213642120361328e-05,
      "model_forward_time": 1.7192816734313965,
      "step": 54669
    },
    {
      "epoch": 5.213642120361328e-05,
      "step": 54669,
      "training_step_time": 7.493495225906372
    },
    {
      "epoch": 5.213737487792969e-05,
      "grad_norm": 0.11658375710248947,
      "learning_rate": 4.6416698394630055e-06,
      "loss": 0.0598,
      "step": 54670
    },
    {
      "epoch": 5.213737487792969e-05,
      "model_forward_time": 1.7556002140045166,
      "step": 54670
    },
    {
      "dynamics_loss_avg": 0.0521842647343874,
      "epoch": 5.213737487792969e-05,
      "step": 54670
    },
    {
      "action_loss_avg": 0.0014177757228026166,
      "epoch": 5.213737487792969e-05,
      "step": 54670
    },
    {
      "epoch": 5.213737487792969e-05,
      "step": 54670,
      "training_step_time": 7.62866997718811
    },
    {
      "epoch": 5.213832855224609e-05,
      "model_forward_time": 1.717097282409668,
      "step": 54671
    },
    {
      "epoch": 5.213832855224609e-05,
      "step": 54671,
      "training_step_time": 8.228400945663452
    },
    {
      "epoch": 5.21392822265625e-05,
      "model_forward_time": 1.8667147159576416,
      "step": 54672
    },
    {
      "epoch": 5.21392822265625e-05,
      "step": 54672,
      "training_step_time": 9.892482280731201
    },
    {
      "epoch": 5.214023590087891e-05,
      "model_forward_time": 1.728196620941162,
      "step": 54673
    },
    {
      "epoch": 5.214023590087891e-05,
      "step": 54673,
      "training_step_time": 8.367127180099487
    },
    {
      "epoch": 5.214118957519531e-05,
      "model_forward_time": 1.7080917358398438,
      "step": 54674
    },
    {
      "epoch": 5.214118957519531e-05,
      "step": 54674,
      "training_step_time": 8.47549557685852
    },
    {
      "epoch": 5.2142143249511716e-05,
      "model_forward_time": 1.7201974391937256,
      "step": 54675
    },
    {
      "epoch": 5.2142143249511716e-05,
      "step": 54675,
      "training_step_time": 7.597980737686157
    },
    {
      "epoch": 5.214309692382813e-05,
      "model_forward_time": 1.7342727184295654,
      "step": 54676
    },
    {
      "epoch": 5.214309692382813e-05,
      "step": 54676,
      "training_step_time": 7.2416675090789795
    },
    {
      "epoch": 5.214405059814453e-05,
      "model_forward_time": 1.7171180248260498,
      "step": 54677
    },
    {
      "epoch": 5.214405059814453e-05,
      "step": 54677,
      "training_step_time": 7.067155122756958
    },
    {
      "epoch": 5.2145004272460935e-05,
      "model_forward_time": 1.724395513534546,
      "step": 54678
    },
    {
      "epoch": 5.2145004272460935e-05,
      "step": 54678,
      "training_step_time": 7.477789402008057
    },
    {
      "epoch": 5.2145957946777346e-05,
      "model_forward_time": 1.7542736530303955,
      "step": 54679
    },
    {
      "epoch": 5.2145957946777346e-05,
      "step": 54679,
      "training_step_time": 7.69497013092041
    },
    {
      "epoch": 5.214691162109375e-05,
      "grad_norm": 0.16381488740444183,
      "learning_rate": 4.640020640861862e-06,
      "loss": 0.0602,
      "step": 54680
    },
    {
      "epoch": 5.214691162109375e-05,
      "model_forward_time": 1.718794584274292,
      "step": 54680
    },
    {
      "dynamics_loss_avg": 0.058973353728652,
      "epoch": 5.214691162109375e-05,
      "step": 54680
    },
    {
      "action_loss_avg": 0.0015342467348091304,
      "epoch": 5.214691162109375e-05,
      "step": 54680
    },
    {
      "epoch": 5.214691162109375e-05,
      "step": 54680,
      "training_step_time": 7.6367974281311035
    },
    {
      "epoch": 5.2147865295410154e-05,
      "model_forward_time": 2.2584240436553955,
      "step": 54681
    },
    {
      "epoch": 5.2147865295410154e-05,
      "step": 54681,
      "training_step_time": 10.236001014709473
    },
    {
      "epoch": 5.2148818969726565e-05,
      "model_forward_time": 1.7097628116607666,
      "step": 54682
    },
    {
      "epoch": 5.2148818969726565e-05,
      "step": 54682,
      "training_step_time": 8.53947901725769
    },
    {
      "epoch": 5.214977264404297e-05,
      "model_forward_time": 1.7212350368499756,
      "step": 54683
    },
    {
      "epoch": 5.214977264404297e-05,
      "step": 54683,
      "training_step_time": 8.176722049713135
    },
    {
      "epoch": 5.215072631835937e-05,
      "model_forward_time": 1.703472375869751,
      "step": 54684
    },
    {
      "epoch": 5.215072631835937e-05,
      "step": 54684,
      "training_step_time": 7.502459764480591
    },
    {
      "epoch": 5.2151679992675783e-05,
      "model_forward_time": 1.722580909729004,
      "step": 54685
    },
    {
      "epoch": 5.2151679992675783e-05,
      "step": 54685,
      "training_step_time": 7.362038850784302
    },
    {
      "epoch": 5.215263366699219e-05,
      "model_forward_time": 1.7308285236358643,
      "step": 54686
    },
    {
      "epoch": 5.215263366699219e-05,
      "step": 54686,
      "training_step_time": 7.449642658233643
    },
    {
      "epoch": 5.215358734130859e-05,
      "model_forward_time": 1.7406418323516846,
      "step": 54687
    },
    {
      "epoch": 5.215358734130859e-05,
      "step": 54687,
      "training_step_time": 7.458423137664795
    },
    {
      "epoch": 5.2154541015625e-05,
      "model_forward_time": 1.7585906982421875,
      "step": 54688
    },
    {
      "epoch": 5.2154541015625e-05,
      "step": 54688,
      "training_step_time": 8.248471021652222
    },
    {
      "epoch": 5.2155494689941406e-05,
      "model_forward_time": 1.7359943389892578,
      "step": 54689
    },
    {
      "epoch": 5.2155494689941406e-05,
      "step": 54689,
      "training_step_time": 9.44079065322876
    },
    {
      "epoch": 5.215644836425781e-05,
      "grad_norm": 0.10527708381414413,
      "learning_rate": 4.63837148162752e-06,
      "loss": 0.061,
      "step": 54690
    },
    {
      "epoch": 5.215644836425781e-05,
      "model_forward_time": 2.28973650932312,
      "step": 54690
    },
    {
      "dynamics_loss_avg": 0.04278363678604365,
      "epoch": 5.215644836425781e-05,
      "step": 54690
    },
    {
      "action_loss_avg": 0.0017673942318651825,
      "epoch": 5.215644836425781e-05,
      "step": 54690
    },
    {
      "epoch": 5.215644836425781e-05,
      "step": 54690,
      "training_step_time": 8.472900629043579
    },
    {
      "epoch": 5.215740203857422e-05,
      "model_forward_time": 1.7301888465881348,
      "step": 54691
    },
    {
      "epoch": 5.215740203857422e-05,
      "step": 54691,
      "training_step_time": 8.375357866287231
    },
    {
      "epoch": 5.2158355712890625e-05,
      "model_forward_time": 1.026733636856079,
      "step": 54692
    },
    {
      "epoch": 5.2158355712890625e-05,
      "step": 54692,
      "training_step_time": 7.708153486251831
    },
    {
      "epoch": 5.215930938720703e-05,
      "model_forward_time": 1.7368323802947998,
      "step": 54693
    },
    {
      "epoch": 5.215930938720703e-05,
      "step": 54693,
      "training_step_time": 8.32881236076355
    },
    {
      "epoch": 5.216026306152344e-05,
      "model_forward_time": 1.7356443405151367,
      "step": 54694
    },
    {
      "epoch": 5.216026306152344e-05,
      "step": 54694,
      "training_step_time": 7.63288950920105
    },
    {
      "epoch": 5.2161216735839844e-05,
      "model_forward_time": 1.7398128509521484,
      "step": 54695
    },
    {
      "epoch": 5.2161216735839844e-05,
      "step": 54695,
      "training_step_time": 7.16123104095459
    },
    {
      "epoch": 5.216217041015625e-05,
      "model_forward_time": 1.7359025478363037,
      "step": 54696
    },
    {
      "epoch": 5.216217041015625e-05,
      "step": 54696,
      "training_step_time": 7.470899343490601
    },
    {
      "epoch": 5.216312408447266e-05,
      "model_forward_time": 1.7353622913360596,
      "step": 54697
    },
    {
      "epoch": 5.216312408447266e-05,
      "step": 54697,
      "training_step_time": 7.647035360336304
    },
    {
      "epoch": 5.216407775878906e-05,
      "model_forward_time": 1.7497341632843018,
      "step": 54698
    },
    {
      "epoch": 5.216407775878906e-05,
      "step": 54698,
      "training_step_time": 7.659706115722656
    },
    {
      "epoch": 5.216503143310547e-05,
      "model_forward_time": 1.7388036251068115,
      "step": 54699
    },
    {
      "epoch": 5.216503143310547e-05,
      "step": 54699,
      "training_step_time": 8.739518642425537
    },
    {
      "epoch": 5.216598510742188e-05,
      "grad_norm": 0.17372502386569977,
      "learning_rate": 4.636722361940329e-06,
      "loss": 0.0597,
      "step": 54700
    },
    {
      "epoch": 5.216598510742188e-05,
      "model_forward_time": 2.2716457843780518,
      "step": 54700
    },
    {
      "dynamics_loss_avg": 0.05416969247162342,
      "epoch": 5.216598510742188e-05,
      "step": 54700
    },
    {
      "action_loss_avg": 0.0034084051148965955,
      "epoch": 5.216598510742188e-05,
      "step": 54700
    },
    {
      "epoch": 5.216598510742188e-05,
      "step": 54700,
      "training_step_time": 8.627967119216919
    },
    {
      "epoch": 5.216693878173828e-05,
      "model_forward_time": 1.7409110069274902,
      "step": 54701
    },
    {
      "epoch": 5.216693878173828e-05,
      "step": 54701,
      "training_step_time": 7.6717798709869385
    },
    {
      "epoch": 5.2167892456054686e-05,
      "model_forward_time": 1.7038657665252686,
      "step": 54702
    },
    {
      "epoch": 5.2167892456054686e-05,
      "step": 54702,
      "training_step_time": 7.490429162979126
    },
    {
      "epoch": 5.2168846130371097e-05,
      "model_forward_time": 1.717376708984375,
      "step": 54703
    },
    {
      "epoch": 5.2168846130371097e-05,
      "step": 54703,
      "training_step_time": 7.601823329925537
    },
    {
      "epoch": 5.21697998046875e-05,
      "model_forward_time": 1.6970393657684326,
      "step": 54704
    },
    {
      "epoch": 5.21697998046875e-05,
      "step": 54704,
      "training_step_time": 7.26551079750061
    },
    {
      "epoch": 5.2170753479003905e-05,
      "model_forward_time": 1.7062337398529053,
      "step": 54705
    },
    {
      "epoch": 5.2170753479003905e-05,
      "step": 54705,
      "training_step_time": 7.541069269180298
    },
    {
      "epoch": 5.2171707153320315e-05,
      "model_forward_time": 1.7163069248199463,
      "step": 54706
    },
    {
      "epoch": 5.2171707153320315e-05,
      "step": 54706,
      "training_step_time": 7.674037456512451
    },
    {
      "epoch": 5.217266082763672e-05,
      "model_forward_time": 1.7668373584747314,
      "step": 54707
    },
    {
      "epoch": 5.217266082763672e-05,
      "step": 54707,
      "training_step_time": 8.379358291625977
    },
    {
      "epoch": 5.2173614501953123e-05,
      "model_forward_time": 1.7143235206604004,
      "step": 54708
    },
    {
      "epoch": 5.2173614501953123e-05,
      "step": 54708,
      "training_step_time": 8.59489631652832
    },
    {
      "epoch": 5.2174568176269534e-05,
      "model_forward_time": 2.268299102783203,
      "step": 54709
    },
    {
      "epoch": 5.2174568176269534e-05,
      "step": 54709,
      "training_step_time": 8.304001331329346
    },
    {
      "epoch": 5.217552185058594e-05,
      "grad_norm": 0.14972546696662903,
      "learning_rate": 4.635073281980632e-06,
      "loss": 0.0612,
      "step": 54710
    },
    {
      "epoch": 5.217552185058594e-05,
      "model_forward_time": 1.7145416736602783,
      "step": 54710
    },
    {
      "dynamics_loss_avg": 0.04623628612607718,
      "epoch": 5.217552185058594e-05,
      "step": 54710
    },
    {
      "action_loss_avg": 0.0016800925455754622,
      "epoch": 5.217552185058594e-05,
      "step": 54710
    },
    {
      "epoch": 5.217552185058594e-05,
      "step": 54710,
      "training_step_time": 8.704181432723999
    },
    {
      "epoch": 5.217647552490234e-05,
      "model_forward_time": 1.7104859352111816,
      "step": 54711
    },
    {
      "epoch": 5.217647552490234e-05,
      "step": 54711,
      "training_step_time": 7.4459943771362305
    },
    {
      "epoch": 5.217742919921875e-05,
      "model_forward_time": 1.7140283584594727,
      "step": 54712
    },
    {
      "epoch": 5.217742919921875e-05,
      "step": 54712,
      "training_step_time": 8.512365818023682
    },
    {
      "epoch": 5.217838287353516e-05,
      "model_forward_time": 1.710918664932251,
      "step": 54713
    },
    {
      "epoch": 5.217838287353516e-05,
      "step": 54713,
      "training_step_time": 7.390130043029785
    },
    {
      "epoch": 5.217933654785156e-05,
      "model_forward_time": 1.709745168685913,
      "step": 54714
    },
    {
      "epoch": 5.217933654785156e-05,
      "step": 54714,
      "training_step_time": 7.550259590148926
    },
    {
      "epoch": 5.218029022216797e-05,
      "model_forward_time": 1.7116613388061523,
      "step": 54715
    },
    {
      "epoch": 5.218029022216797e-05,
      "step": 54715,
      "training_step_time": 7.764033555984497
    },
    {
      "epoch": 5.2181243896484376e-05,
      "model_forward_time": 1.7562541961669922,
      "step": 54716
    },
    {
      "epoch": 5.2181243896484376e-05,
      "step": 54716,
      "training_step_time": 9.392130374908447
    },
    {
      "epoch": 5.218219757080078e-05,
      "model_forward_time": 1.7209935188293457,
      "step": 54717
    },
    {
      "epoch": 5.218219757080078e-05,
      "step": 54717,
      "training_step_time": 9.77835464477539
    },
    {
      "epoch": 5.218315124511719e-05,
      "model_forward_time": 2.252901792526245,
      "step": 54718
    },
    {
      "epoch": 5.218315124511719e-05,
      "step": 54718,
      "training_step_time": 9.042572259902954
    },
    {
      "epoch": 5.2184104919433595e-05,
      "model_forward_time": 1.7155606746673584,
      "step": 54719
    },
    {
      "epoch": 5.2184104919433595e-05,
      "step": 54719,
      "training_step_time": 8.25536060333252
    },
    {
      "epoch": 5.218505859375e-05,
      "grad_norm": 0.11428368836641312,
      "learning_rate": 4.633424241928775e-06,
      "loss": 0.0632,
      "step": 54720
    },
    {
      "epoch": 5.218505859375e-05,
      "model_forward_time": 1.719574213027954,
      "step": 54720
    },
    {
      "dynamics_loss_avg": 0.05447809398174286,
      "epoch": 5.218505859375e-05,
      "step": 54720
    },
    {
      "action_loss_avg": 0.001927491492824629,
      "epoch": 5.218505859375e-05,
      "step": 54720
    },
    {
      "epoch": 5.218505859375e-05,
      "step": 54720,
      "training_step_time": 7.568422794342041
    },
    {
      "epoch": 5.21860122680664e-05,
      "model_forward_time": 1.7183952331542969,
      "step": 54721
    },
    {
      "epoch": 5.21860122680664e-05,
      "step": 54721,
      "training_step_time": 7.5210559368133545
    },
    {
      "epoch": 5.2186965942382814e-05,
      "model_forward_time": 1.7204921245574951,
      "step": 54722
    },
    {
      "epoch": 5.2186965942382814e-05,
      "step": 54722,
      "training_step_time": 7.51385760307312
    },
    {
      "epoch": 5.218791961669922e-05,
      "model_forward_time": 1.7117633819580078,
      "step": 54723
    },
    {
      "epoch": 5.218791961669922e-05,
      "step": 54723,
      "training_step_time": 7.516094923019409
    },
    {
      "epoch": 5.218887329101562e-05,
      "model_forward_time": 1.711848497390747,
      "step": 54724
    },
    {
      "epoch": 5.218887329101562e-05,
      "step": 54724,
      "training_step_time": 7.668253660202026
    },
    {
      "epoch": 5.218982696533203e-05,
      "model_forward_time": 1.7781322002410889,
      "step": 54725
    },
    {
      "epoch": 5.218982696533203e-05,
      "step": 54725,
      "training_step_time": 8.688677310943604
    },
    {
      "epoch": 5.219078063964844e-05,
      "model_forward_time": 1.3373544216156006,
      "step": 54726
    },
    {
      "epoch": 5.219078063964844e-05,
      "step": 54726,
      "training_step_time": 7.704463958740234
    },
    {
      "epoch": 5.219173431396484e-05,
      "model_forward_time": 1.7115423679351807,
      "step": 54727
    },
    {
      "epoch": 5.219173431396484e-05,
      "step": 54727,
      "training_step_time": 8.52007532119751
    },
    {
      "epoch": 5.219268798828125e-05,
      "model_forward_time": 1.7047295570373535,
      "step": 54728
    },
    {
      "epoch": 5.219268798828125e-05,
      "step": 54728,
      "training_step_time": 9.397469758987427
    },
    {
      "epoch": 5.2193641662597655e-05,
      "model_forward_time": 1.7117412090301514,
      "step": 54729
    },
    {
      "epoch": 5.2193641662597655e-05,
      "step": 54729,
      "training_step_time": 8.83088231086731
    },
    {
      "epoch": 5.219459533691406e-05,
      "grad_norm": 0.09819357097148895,
      "learning_rate": 4.631775241965091e-06,
      "loss": 0.0603,
      "step": 54730
    },
    {
      "epoch": 5.219459533691406e-05,
      "model_forward_time": 1.7018771171569824,
      "step": 54730
    },
    {
      "dynamics_loss_avg": 0.051032324135303495,
      "epoch": 5.219459533691406e-05,
      "step": 54730
    },
    {
      "action_loss_avg": 0.002350553619908169,
      "epoch": 5.219459533691406e-05,
      "step": 54730
    },
    {
      "epoch": 5.219459533691406e-05,
      "step": 54730,
      "training_step_time": 7.467153549194336
    },
    {
      "epoch": 5.219554901123047e-05,
      "model_forward_time": 1.7113690376281738,
      "step": 54731
    },
    {
      "epoch": 5.219554901123047e-05,
      "step": 54731,
      "training_step_time": 7.491279363632202
    },
    {
      "epoch": 5.2196502685546874e-05,
      "model_forward_time": 1.744527816772461,
      "step": 54732
    },
    {
      "epoch": 5.2196502685546874e-05,
      "step": 54732,
      "training_step_time": 7.423343658447266
    },
    {
      "epoch": 5.219745635986328e-05,
      "model_forward_time": 1.7080702781677246,
      "step": 54733
    },
    {
      "epoch": 5.219745635986328e-05,
      "step": 54733,
      "training_step_time": 7.767439603805542
    },
    {
      "epoch": 5.219841003417969e-05,
      "model_forward_time": 1.7173404693603516,
      "step": 54734
    },
    {
      "epoch": 5.219841003417969e-05,
      "step": 54734,
      "training_step_time": 7.604326248168945
    },
    {
      "epoch": 5.219936370849609e-05,
      "model_forward_time": 1.7561719417572021,
      "step": 54735
    },
    {
      "epoch": 5.219936370849609e-05,
      "step": 54735,
      "training_step_time": 7.59227442741394
    },
    {
      "epoch": 5.22003173828125e-05,
      "model_forward_time": 1.7080326080322266,
      "step": 54736
    },
    {
      "epoch": 5.22003173828125e-05,
      "step": 54736,
      "training_step_time": 8.16794753074646
    },
    {
      "epoch": 5.220127105712891e-05,
      "model_forward_time": 2.2609143257141113,
      "step": 54737
    },
    {
      "epoch": 5.220127105712891e-05,
      "step": 54737,
      "training_step_time": 7.862355470657349
    },
    {
      "epoch": 5.220222473144531e-05,
      "model_forward_time": 1.7149786949157715,
      "step": 54738
    },
    {
      "epoch": 5.220222473144531e-05,
      "step": 54738,
      "training_step_time": 7.561841726303101
    },
    {
      "epoch": 5.2203178405761716e-05,
      "model_forward_time": 1.7125728130340576,
      "step": 54739
    },
    {
      "epoch": 5.2203178405761716e-05,
      "step": 54739,
      "training_step_time": 7.544709920883179
    },
    {
      "epoch": 5.220413208007813e-05,
      "grad_norm": 0.08187603205442429,
      "learning_rate": 4.630126282269914e-06,
      "loss": 0.0565,
      "step": 54740
    },
    {
      "epoch": 5.220413208007813e-05,
      "model_forward_time": 1.7024798393249512,
      "step": 54740
    },
    {
      "dynamics_loss_avg": 0.06795349437743425,
      "epoch": 5.220413208007813e-05,
      "step": 54740
    },
    {
      "action_loss_avg": 0.0016126766393426805,
      "epoch": 5.220413208007813e-05,
      "step": 54740
    },
    {
      "epoch": 5.220413208007813e-05,
      "step": 54740,
      "training_step_time": 7.561459541320801
    },
    {
      "epoch": 5.220508575439453e-05,
      "model_forward_time": 1.7091622352600098,
      "step": 54741
    },
    {
      "epoch": 5.220508575439453e-05,
      "step": 54741,
      "training_step_time": 7.378798007965088
    },
    {
      "epoch": 5.2206039428710935e-05,
      "model_forward_time": 1.70479154586792,
      "step": 54742
    },
    {
      "epoch": 5.2206039428710935e-05,
      "step": 54742,
      "training_step_time": 7.706506967544556
    },
    {
      "epoch": 5.2206993103027346e-05,
      "model_forward_time": 1.71217679977417,
      "step": 54743
    },
    {
      "epoch": 5.2206993103027346e-05,
      "step": 54743,
      "training_step_time": 8.993481874465942
    },
    {
      "epoch": 5.220794677734375e-05,
      "model_forward_time": 1.7595558166503906,
      "step": 54744
    },
    {
      "epoch": 5.220794677734375e-05,
      "step": 54744,
      "training_step_time": 7.571215629577637
    },
    {
      "epoch": 5.2208900451660154e-05,
      "model_forward_time": 1.701904535293579,
      "step": 54745
    },
    {
      "epoch": 5.2208900451660154e-05,
      "step": 54745,
      "training_step_time": 8.61651349067688
    },
    {
      "epoch": 5.2209854125976565e-05,
      "model_forward_time": 2.238882303237915,
      "step": 54746
    },
    {
      "epoch": 5.2209854125976565e-05,
      "step": 54746,
      "training_step_time": 7.780728578567505
    },
    {
      "epoch": 5.221080780029297e-05,
      "model_forward_time": 1.7035410404205322,
      "step": 54747
    },
    {
      "epoch": 5.221080780029297e-05,
      "step": 54747,
      "training_step_time": 8.375792264938354
    },
    {
      "epoch": 5.221176147460937e-05,
      "model_forward_time": 1.7042717933654785,
      "step": 54748
    },
    {
      "epoch": 5.221176147460937e-05,
      "step": 54748,
      "training_step_time": 7.526527166366577
    },
    {
      "epoch": 5.2212715148925783e-05,
      "model_forward_time": 1.7137696743011475,
      "step": 54749
    },
    {
      "epoch": 5.2212715148925783e-05,
      "step": 54749,
      "training_step_time": 8.036948680877686
    },
    {
      "epoch": 5.221366882324219e-05,
      "grad_norm": 0.09369074553251266,
      "learning_rate": 4.6284773630235695e-06,
      "loss": 0.0636,
      "step": 54750
    },
    {
      "epoch": 5.221366882324219e-05,
      "model_forward_time": 1.7072412967681885,
      "step": 54750
    },
    {
      "dynamics_loss_avg": 0.07597291730344295,
      "epoch": 5.221366882324219e-05,
      "step": 54750
    },
    {
      "action_loss_avg": 0.0022335366840707136,
      "epoch": 5.221366882324219e-05,
      "step": 54750
    },
    {
      "epoch": 5.221366882324219e-05,
      "step": 54750,
      "training_step_time": 7.310166358947754
    },
    {
      "epoch": 5.221462249755859e-05,
      "model_forward_time": 1.7043147087097168,
      "step": 54751
    },
    {
      "epoch": 5.221462249755859e-05,
      "step": 54751,
      "training_step_time": 8.093347311019897
    },
    {
      "epoch": 5.2215576171875e-05,
      "model_forward_time": 1.7086536884307861,
      "step": 54752
    },
    {
      "epoch": 5.2215576171875e-05,
      "step": 54752,
      "training_step_time": 7.749571084976196
    },
    {
      "epoch": 5.2216529846191406e-05,
      "model_forward_time": 1.751305341720581,
      "step": 54753
    },
    {
      "epoch": 5.2216529846191406e-05,
      "step": 54753,
      "training_step_time": 7.510474920272827
    },
    {
      "epoch": 5.221748352050781e-05,
      "model_forward_time": 1.338564395904541,
      "step": 54754
    },
    {
      "epoch": 5.221748352050781e-05,
      "step": 54754,
      "training_step_time": 8.596683740615845
    },
    {
      "epoch": 5.221843719482422e-05,
      "model_forward_time": 2.25213623046875,
      "step": 54755
    },
    {
      "epoch": 5.221843719482422e-05,
      "step": 54755,
      "training_step_time": 7.731744289398193
    },
    {
      "epoch": 5.2219390869140625e-05,
      "model_forward_time": 1.7080481052398682,
      "step": 54756
    },
    {
      "epoch": 5.2219390869140625e-05,
      "step": 54756,
      "training_step_time": 7.5918495655059814
    },
    {
      "epoch": 5.222034454345703e-05,
      "model_forward_time": 1.7073748111724854,
      "step": 54757
    },
    {
      "epoch": 5.222034454345703e-05,
      "step": 54757,
      "training_step_time": 7.525005340576172
    },
    {
      "epoch": 5.222129821777344e-05,
      "model_forward_time": 1.7124826908111572,
      "step": 54758
    },
    {
      "epoch": 5.222129821777344e-05,
      "step": 54758,
      "training_step_time": 7.537142038345337
    },
    {
      "epoch": 5.2222251892089844e-05,
      "model_forward_time": 1.707808256149292,
      "step": 54759
    },
    {
      "epoch": 5.2222251892089844e-05,
      "step": 54759,
      "training_step_time": 8.245308637619019
    },
    {
      "epoch": 5.222320556640625e-05,
      "grad_norm": 0.12214013189077377,
      "learning_rate": 4.626828484406383e-06,
      "loss": 0.0575,
      "step": 54760
    },
    {
      "epoch": 5.222320556640625e-05,
      "model_forward_time": 1.708939552307129,
      "step": 54760
    },
    {
      "dynamics_loss_avg": 0.056286837160587314,
      "epoch": 5.222320556640625e-05,
      "step": 54760
    },
    {
      "action_loss_avg": 0.0015862594736972824,
      "epoch": 5.222320556640625e-05,
      "step": 54760
    },
    {
      "epoch": 5.222320556640625e-05,
      "step": 54760,
      "training_step_time": 7.515177488327026
    },
    {
      "epoch": 5.222415924072266e-05,
      "model_forward_time": 1.7102546691894531,
      "step": 54761
    },
    {
      "epoch": 5.222415924072266e-05,
      "step": 54761,
      "training_step_time": 8.392311096191406
    },
    {
      "epoch": 5.222511291503906e-05,
      "model_forward_time": 1.7437682151794434,
      "step": 54762
    },
    {
      "epoch": 5.222511291503906e-05,
      "step": 54762,
      "training_step_time": 7.595618724822998
    },
    {
      "epoch": 5.222606658935547e-05,
      "model_forward_time": 1.7210521697998047,
      "step": 54763
    },
    {
      "epoch": 5.222606658935547e-05,
      "step": 54763,
      "training_step_time": 7.787278890609741
    },
    {
      "epoch": 5.222702026367188e-05,
      "model_forward_time": 2.2696993350982666,
      "step": 54764
    },
    {
      "epoch": 5.222702026367188e-05,
      "step": 54764,
      "training_step_time": 8.853790760040283
    },
    {
      "epoch": 5.222797393798828e-05,
      "model_forward_time": 1.707026481628418,
      "step": 54765
    },
    {
      "epoch": 5.222797393798828e-05,
      "step": 54765,
      "training_step_time": 7.747946262359619
    },
    {
      "epoch": 5.2228927612304686e-05,
      "model_forward_time": 1.71171236038208,
      "step": 54766
    },
    {
      "epoch": 5.2228927612304686e-05,
      "step": 54766,
      "training_step_time": 8.39052939414978
    },
    {
      "epoch": 5.22298812866211e-05,
      "model_forward_time": 1.7085497379302979,
      "step": 54767
    },
    {
      "epoch": 5.22298812866211e-05,
      "step": 54767,
      "training_step_time": 7.531816482543945
    },
    {
      "epoch": 5.22308349609375e-05,
      "model_forward_time": 1.7134795188903809,
      "step": 54768
    },
    {
      "epoch": 5.22308349609375e-05,
      "step": 54768,
      "training_step_time": 7.652215242385864
    },
    {
      "epoch": 5.2231788635253905e-05,
      "model_forward_time": 1.7038309574127197,
      "step": 54769
    },
    {
      "epoch": 5.2231788635253905e-05,
      "step": 54769,
      "training_step_time": 7.509117841720581
    },
    {
      "epoch": 5.2232742309570315e-05,
      "grad_norm": 0.08260832726955414,
      "learning_rate": 4.62517964659867e-06,
      "loss": 0.0594,
      "step": 54770
    },
    {
      "epoch": 5.2232742309570315e-05,
      "model_forward_time": 1.7070443630218506,
      "step": 54770
    },
    {
      "dynamics_loss_avg": 0.049424076545983554,
      "epoch": 5.2232742309570315e-05,
      "step": 54770
    },
    {
      "action_loss_avg": 0.001743098208680749,
      "epoch": 5.2232742309570315e-05,
      "step": 54770
    },
    {
      "epoch": 5.2232742309570315e-05,
      "step": 54770,
      "training_step_time": 8.71017599105835
    },
    {
      "epoch": 5.223369598388672e-05,
      "model_forward_time": 1.7220664024353027,
      "step": 54771
    },
    {
      "epoch": 5.223369598388672e-05,
      "step": 54771,
      "training_step_time": 9.182667016983032
    },
    {
      "epoch": 5.2234649658203124e-05,
      "model_forward_time": 1.7076480388641357,
      "step": 54772
    },
    {
      "epoch": 5.2234649658203124e-05,
      "step": 54772,
      "training_step_time": 8.501072883605957
    },
    {
      "epoch": 5.2235603332519534e-05,
      "model_forward_time": 1.7035598754882812,
      "step": 54773
    },
    {
      "epoch": 5.2235603332519534e-05,
      "step": 54773,
      "training_step_time": 9.136624097824097
    },
    {
      "epoch": 5.223655700683594e-05,
      "model_forward_time": 1.69008207321167,
      "step": 54774
    },
    {
      "epoch": 5.223655700683594e-05,
      "step": 54774,
      "training_step_time": 8.56235647201538
    },
    {
      "epoch": 5.223751068115234e-05,
      "model_forward_time": 1.7162468433380127,
      "step": 54775
    },
    {
      "epoch": 5.223751068115234e-05,
      "step": 54775,
      "training_step_time": 7.76903772354126
    },
    {
      "epoch": 5.223846435546875e-05,
      "model_forward_time": 1.7195990085601807,
      "step": 54776
    },
    {
      "epoch": 5.223846435546875e-05,
      "step": 54776,
      "training_step_time": 7.5783610343933105
    },
    {
      "epoch": 5.223941802978516e-05,
      "model_forward_time": 1.712839126586914,
      "step": 54777
    },
    {
      "epoch": 5.223941802978516e-05,
      "step": 54777,
      "training_step_time": 7.669893980026245
    },
    {
      "epoch": 5.224037170410156e-05,
      "model_forward_time": 1.7152886390686035,
      "step": 54778
    },
    {
      "epoch": 5.224037170410156e-05,
      "step": 54778,
      "training_step_time": 7.470567226409912
    },
    {
      "epoch": 5.224132537841797e-05,
      "model_forward_time": 1.7212634086608887,
      "step": 54779
    },
    {
      "epoch": 5.224132537841797e-05,
      "step": 54779,
      "training_step_time": 7.768221378326416
    },
    {
      "epoch": 5.2242279052734376e-05,
      "grad_norm": 0.11985848098993301,
      "learning_rate": 4.623530849780749e-06,
      "loss": 0.0602,
      "step": 54780
    },
    {
      "epoch": 5.2242279052734376e-05,
      "model_forward_time": 1.708430290222168,
      "step": 54780
    },
    {
      "dynamics_loss_avg": 0.049418647028505805,
      "epoch": 5.2242279052734376e-05,
      "step": 54780
    },
    {
      "action_loss_avg": 0.0019884245411958545,
      "epoch": 5.2242279052734376e-05,
      "step": 54780
    },
    {
      "epoch": 5.2242279052734376e-05,
      "step": 54780,
      "training_step_time": 8.223714590072632
    },
    {
      "epoch": 5.224323272705078e-05,
      "model_forward_time": 1.7083661556243896,
      "step": 54781
    },
    {
      "epoch": 5.224323272705078e-05,
      "step": 54781,
      "training_step_time": 7.617967367172241
    },
    {
      "epoch": 5.2244186401367184e-05,
      "model_forward_time": 1.7157611846923828,
      "step": 54782
    },
    {
      "epoch": 5.2244186401367184e-05,
      "step": 54782,
      "training_step_time": 7.588779449462891
    },
    {
      "epoch": 5.2245140075683595e-05,
      "model_forward_time": 2.253293514251709,
      "step": 54783
    },
    {
      "epoch": 5.2245140075683595e-05,
      "step": 54783,
      "training_step_time": 8.641111612319946
    },
    {
      "epoch": 5.224609375e-05,
      "model_forward_time": 1.7105882167816162,
      "step": 54784
    },
    {
      "epoch": 5.224609375e-05,
      "step": 54784,
      "training_step_time": 7.641448497772217
    },
    {
      "epoch": 5.22470474243164e-05,
      "model_forward_time": 1.716191053390503,
      "step": 54785
    },
    {
      "epoch": 5.22470474243164e-05,
      "step": 54785,
      "training_step_time": 7.583983898162842
    },
    {
      "epoch": 5.2248001098632814e-05,
      "model_forward_time": 1.7097842693328857,
      "step": 54786
    },
    {
      "epoch": 5.2248001098632814e-05,
      "step": 54786,
      "training_step_time": 8.258027076721191
    },
    {
      "epoch": 5.224895477294922e-05,
      "model_forward_time": 1.712759017944336,
      "step": 54787
    },
    {
      "epoch": 5.224895477294922e-05,
      "step": 54787,
      "training_step_time": 7.452027797698975
    },
    {
      "epoch": 5.224990844726562e-05,
      "model_forward_time": 1.721949815750122,
      "step": 54788
    },
    {
      "epoch": 5.224990844726562e-05,
      "step": 54788,
      "training_step_time": 7.667962312698364
    },
    {
      "epoch": 5.225086212158203e-05,
      "model_forward_time": 1.7152423858642578,
      "step": 54789
    },
    {
      "epoch": 5.225086212158203e-05,
      "step": 54789,
      "training_step_time": 7.6584367752075195
    },
    {
      "epoch": 5.225181579589844e-05,
      "grad_norm": 0.16290490329265594,
      "learning_rate": 4.621882094132928e-06,
      "loss": 0.062,
      "step": 54790
    },
    {
      "epoch": 5.225181579589844e-05,
      "model_forward_time": 1.752713680267334,
      "step": 54790
    },
    {
      "dynamics_loss_avg": 0.07731728255748749,
      "epoch": 5.225181579589844e-05,
      "step": 54790
    },
    {
      "action_loss_avg": 0.002715281321434304,
      "epoch": 5.225181579589844e-05,
      "step": 54790
    },
    {
      "epoch": 5.225181579589844e-05,
      "step": 54790,
      "training_step_time": 8.211059808731079
    },
    {
      "epoch": 5.225276947021484e-05,
      "model_forward_time": 1.7059955596923828,
      "step": 54791
    },
    {
      "epoch": 5.225276947021484e-05,
      "step": 54791,
      "training_step_time": 9.18453311920166
    },
    {
      "epoch": 5.225372314453125e-05,
      "model_forward_time": 2.250398874282837,
      "step": 54792
    },
    {
      "epoch": 5.225372314453125e-05,
      "step": 54792,
      "training_step_time": 7.83234977722168
    },
    {
      "epoch": 5.2254676818847656e-05,
      "model_forward_time": 1.7149581909179688,
      "step": 54793
    },
    {
      "epoch": 5.2254676818847656e-05,
      "step": 54793,
      "training_step_time": 7.570902347564697
    },
    {
      "epoch": 5.225563049316406e-05,
      "model_forward_time": 1.716169834136963,
      "step": 54794
    },
    {
      "epoch": 5.225563049316406e-05,
      "step": 54794,
      "training_step_time": 7.566607236862183
    },
    {
      "epoch": 5.225658416748047e-05,
      "model_forward_time": 1.7142934799194336,
      "step": 54795
    },
    {
      "epoch": 5.225658416748047e-05,
      "step": 54795,
      "training_step_time": 7.536503553390503
    },
    {
      "epoch": 5.2257537841796874e-05,
      "model_forward_time": 1.7136342525482178,
      "step": 54796
    },
    {
      "epoch": 5.2257537841796874e-05,
      "step": 54796,
      "training_step_time": 7.505424499511719
    },
    {
      "epoch": 5.225849151611328e-05,
      "model_forward_time": 1.7118580341339111,
      "step": 54797
    },
    {
      "epoch": 5.225849151611328e-05,
      "step": 54797,
      "training_step_time": 7.443131446838379
    },
    {
      "epoch": 5.225944519042969e-05,
      "model_forward_time": 1.7271442413330078,
      "step": 54798
    },
    {
      "epoch": 5.225944519042969e-05,
      "step": 54798,
      "training_step_time": 8.138070821762085
    },
    {
      "epoch": 5.226039886474609e-05,
      "model_forward_time": 1.3745553493499756,
      "step": 54799
    },
    {
      "epoch": 5.226039886474609e-05,
      "step": 54799,
      "training_step_time": 7.603233098983765
    },
    {
      "epoch": 5.22613525390625e-05,
      "grad_norm": 0.10461647808551788,
      "learning_rate": 4.620233379835513e-06,
      "loss": 0.0601,
      "step": 54800
    },
    {
      "epoch": 5.22613525390625e-05,
      "model_forward_time": 1.7409882545471191,
      "step": 54800
    },
    {
      "dynamics_loss_avg": 0.053541152738034725,
      "epoch": 5.22613525390625e-05,
      "step": 54800
    },
    {
      "action_loss_avg": 0.0018703254754655063,
      "epoch": 5.22613525390625e-05,
      "step": 54800
    },
    {
      "epoch": 5.22613525390625e-05,
      "step": 54800,
      "training_step_time": 7.644725799560547
    },
    {
      "epoch": 5.226230621337891e-05,
      "model_forward_time": 2.2955548763275146,
      "step": 54801
    },
    {
      "epoch": 5.226230621337891e-05,
      "step": 54801,
      "training_step_time": 8.699799299240112
    },
    {
      "epoch": 5.226325988769531e-05,
      "model_forward_time": 1.7338459491729736,
      "step": 54802
    },
    {
      "epoch": 5.226325988769531e-05,
      "step": 54802,
      "training_step_time": 9.195026636123657
    },
    {
      "epoch": 5.2264213562011716e-05,
      "model_forward_time": 1.7262513637542725,
      "step": 54803
    },
    {
      "epoch": 5.2264213562011716e-05,
      "step": 54803,
      "training_step_time": 7.55248498916626
    },
    {
      "epoch": 5.226516723632813e-05,
      "model_forward_time": 1.7243940830230713,
      "step": 54804
    },
    {
      "epoch": 5.226516723632813e-05,
      "step": 54804,
      "training_step_time": 7.590677976608276
    },
    {
      "epoch": 5.226612091064453e-05,
      "model_forward_time": 1.730414867401123,
      "step": 54805
    },
    {
      "epoch": 5.226612091064453e-05,
      "step": 54805,
      "training_step_time": 7.603838205337524
    },
    {
      "epoch": 5.2267074584960935e-05,
      "model_forward_time": 1.724625825881958,
      "step": 54806
    },
    {
      "epoch": 5.2267074584960935e-05,
      "step": 54806,
      "training_step_time": 7.579349517822266
    },
    {
      "epoch": 5.2268028259277346e-05,
      "model_forward_time": 1.7284367084503174,
      "step": 54807
    },
    {
      "epoch": 5.2268028259277346e-05,
      "step": 54807,
      "training_step_time": 7.595618963241577
    },
    {
      "epoch": 5.226898193359375e-05,
      "model_forward_time": 1.7433538436889648,
      "step": 54808
    },
    {
      "epoch": 5.226898193359375e-05,
      "step": 54808,
      "training_step_time": 7.650607109069824
    },
    {
      "epoch": 5.2269935607910154e-05,
      "model_forward_time": 1.7286643981933594,
      "step": 54809
    },
    {
      "epoch": 5.2269935607910154e-05,
      "step": 54809,
      "training_step_time": 8.789600849151611
    },
    {
      "epoch": 5.2270889282226565e-05,
      "grad_norm": 0.09645086526870728,
      "learning_rate": 4.618584707068804e-06,
      "loss": 0.0563,
      "step": 54810
    },
    {
      "epoch": 5.2270889282226565e-05,
      "model_forward_time": 1.88592529296875,
      "step": 54810
    },
    {
      "dynamics_loss_avg": 0.05474947541952133,
      "epoch": 5.2270889282226565e-05,
      "step": 54810
    },
    {
      "action_loss_avg": 0.0019789055513683707,
      "epoch": 5.2270889282226565e-05,
      "step": 54810
    },
    {
      "epoch": 5.2270889282226565e-05,
      "step": 54810,
      "training_step_time": 9.114226579666138
    },
    {
      "epoch": 5.227184295654297e-05,
      "model_forward_time": 1.7306289672851562,
      "step": 54811
    },
    {
      "epoch": 5.227184295654297e-05,
      "step": 54811,
      "training_step_time": 8.802082777023315
    },
    {
      "epoch": 5.227279663085937e-05,
      "model_forward_time": 1.7275865077972412,
      "step": 54812
    },
    {
      "epoch": 5.227279663085937e-05,
      "step": 54812,
      "training_step_time": 8.287293434143066
    },
    {
      "epoch": 5.2273750305175784e-05,
      "model_forward_time": 1.3583440780639648,
      "step": 54813
    },
    {
      "epoch": 5.2273750305175784e-05,
      "step": 54813,
      "training_step_time": 7.561238765716553
    },
    {
      "epoch": 5.227470397949219e-05,
      "model_forward_time": 1.779116153717041,
      "step": 54814
    },
    {
      "epoch": 5.227470397949219e-05,
      "step": 54814,
      "training_step_time": 7.693118572235107
    },
    {
      "epoch": 5.227565765380859e-05,
      "model_forward_time": 1.7304770946502686,
      "step": 54815
    },
    {
      "epoch": 5.227565765380859e-05,
      "step": 54815,
      "training_step_time": 7.531845569610596
    },
    {
      "epoch": 5.2276611328125e-05,
      "model_forward_time": 1.357849359512329,
      "step": 54816
    },
    {
      "epoch": 5.2276611328125e-05,
      "step": 54816,
      "training_step_time": 7.817800283432007
    },
    {
      "epoch": 5.2277565002441406e-05,
      "model_forward_time": 1.7306156158447266,
      "step": 54817
    },
    {
      "epoch": 5.2277565002441406e-05,
      "step": 54817,
      "training_step_time": 7.78419041633606
    },
    {
      "epoch": 5.227851867675781e-05,
      "model_forward_time": 1.7905073165893555,
      "step": 54818
    },
    {
      "epoch": 5.227851867675781e-05,
      "step": 54818,
      "training_step_time": 10.446457386016846
    },
    {
      "epoch": 5.227947235107422e-05,
      "model_forward_time": 1.7397148609161377,
      "step": 54819
    },
    {
      "epoch": 5.227947235107422e-05,
      "step": 54819,
      "training_step_time": 8.554551839828491
    },
    {
      "epoch": 5.2280426025390625e-05,
      "grad_norm": 0.11728796362876892,
      "learning_rate": 4.616936076013099e-06,
      "loss": 0.0578,
      "step": 54820
    },
    {
      "epoch": 5.2280426025390625e-05,
      "model_forward_time": 2.293652296066284,
      "step": 54820
    },
    {
      "dynamics_loss_avg": 0.04940680023282766,
      "epoch": 5.2280426025390625e-05,
      "step": 54820
    },
    {
      "action_loss_avg": 0.0013061302073765547,
      "epoch": 5.2280426025390625e-05,
      "step": 54820
    },
    {
      "epoch": 5.2280426025390625e-05,
      "step": 54820,
      "training_step_time": 8.000113725662231
    },
    {
      "epoch": 5.228137969970703e-05,
      "model_forward_time": 1.7351348400115967,
      "step": 54821
    },
    {
      "epoch": 5.228137969970703e-05,
      "step": 54821,
      "training_step_time": 9.262061357498169
    },
    {
      "epoch": 5.228233337402344e-05,
      "model_forward_time": 1.7255885601043701,
      "step": 54822
    },
    {
      "epoch": 5.228233337402344e-05,
      "step": 54822,
      "training_step_time": 7.592459201812744
    },
    {
      "epoch": 5.2283287048339844e-05,
      "model_forward_time": 1.730466365814209,
      "step": 54823
    },
    {
      "epoch": 5.2283287048339844e-05,
      "step": 54823,
      "training_step_time": 7.7460877895355225
    },
    {
      "epoch": 5.228424072265625e-05,
      "model_forward_time": 1.7213411331176758,
      "step": 54824
    },
    {
      "epoch": 5.228424072265625e-05,
      "step": 54824,
      "training_step_time": 8.34326457977295
    },
    {
      "epoch": 5.228519439697266e-05,
      "model_forward_time": 1.7278616428375244,
      "step": 54825
    },
    {
      "epoch": 5.228519439697266e-05,
      "step": 54825,
      "training_step_time": 7.8158042430877686
    },
    {
      "epoch": 5.228614807128906e-05,
      "model_forward_time": 1.7119081020355225,
      "step": 54826
    },
    {
      "epoch": 5.228614807128906e-05,
      "step": 54826,
      "training_step_time": 7.761659383773804
    },
    {
      "epoch": 5.228710174560547e-05,
      "model_forward_time": 1.7698793411254883,
      "step": 54827
    },
    {
      "epoch": 5.228710174560547e-05,
      "step": 54827,
      "training_step_time": 7.731380939483643
    },
    {
      "epoch": 5.228805541992188e-05,
      "model_forward_time": 1.7916367053985596,
      "step": 54828
    },
    {
      "epoch": 5.228805541992188e-05,
      "step": 54828,
      "training_step_time": 8.517778158187866
    },
    {
      "epoch": 5.228900909423828e-05,
      "model_forward_time": 2.294717788696289,
      "step": 54829
    },
    {
      "epoch": 5.228900909423828e-05,
      "step": 54829,
      "training_step_time": 8.795948028564453
    },
    {
      "epoch": 5.2289962768554686e-05,
      "grad_norm": 0.09012893587350845,
      "learning_rate": 4.615287486848688e-06,
      "loss": 0.0562,
      "step": 54830
    },
    {
      "epoch": 5.2289962768554686e-05,
      "model_forward_time": 1.7224080562591553,
      "step": 54830
    },
    {
      "dynamics_loss_avg": 0.06825436316430569,
      "epoch": 5.2289962768554686e-05,
      "step": 54830
    },
    {
      "action_loss_avg": 0.0022050043975468726,
      "epoch": 5.2289962768554686e-05,
      "step": 54830
    },
    {
      "epoch": 5.2289962768554686e-05,
      "step": 54830,
      "training_step_time": 7.824978590011597
    },
    {
      "epoch": 5.22909164428711e-05,
      "model_forward_time": 1.6932573318481445,
      "step": 54831
    },
    {
      "epoch": 5.22909164428711e-05,
      "step": 54831,
      "training_step_time": 8.567928791046143
    },
    {
      "epoch": 5.22918701171875e-05,
      "model_forward_time": 1.7166643142700195,
      "step": 54832
    },
    {
      "epoch": 5.22918701171875e-05,
      "step": 54832,
      "training_step_time": 7.738846302032471
    },
    {
      "epoch": 5.2292823791503905e-05,
      "model_forward_time": 1.7120702266693115,
      "step": 54833
    },
    {
      "epoch": 5.2292823791503905e-05,
      "step": 54833,
      "training_step_time": 7.596216201782227
    },
    {
      "epoch": 5.2293777465820316e-05,
      "model_forward_time": 1.6907715797424316,
      "step": 54834
    },
    {
      "epoch": 5.2293777465820316e-05,
      "step": 54834,
      "training_step_time": 7.815638542175293
    },
    {
      "epoch": 5.229473114013672e-05,
      "model_forward_time": 1.7142233848571777,
      "step": 54835
    },
    {
      "epoch": 5.229473114013672e-05,
      "step": 54835,
      "training_step_time": 7.7305989265441895
    },
    {
      "epoch": 5.2295684814453124e-05,
      "model_forward_time": 1.7175564765930176,
      "step": 54836
    },
    {
      "epoch": 5.2295684814453124e-05,
      "step": 54836,
      "training_step_time": 7.76440691947937
    },
    {
      "epoch": 5.2296638488769534e-05,
      "model_forward_time": 1.7556483745574951,
      "step": 54837
    },
    {
      "epoch": 5.2296638488769534e-05,
      "step": 54837,
      "training_step_time": 8.354711294174194
    },
    {
      "epoch": 5.229759216308594e-05,
      "model_forward_time": 2.253537178039551,
      "step": 54838
    },
    {
      "epoch": 5.229759216308594e-05,
      "step": 54838,
      "training_step_time": 7.892412424087524
    },
    {
      "epoch": 5.229854583740234e-05,
      "model_forward_time": 1.7114927768707275,
      "step": 54839
    },
    {
      "epoch": 5.229854583740234e-05,
      "step": 54839,
      "training_step_time": 7.687507390975952
    },
    {
      "epoch": 5.229949951171875e-05,
      "grad_norm": 0.09711300581693649,
      "learning_rate": 4.613638939755858e-06,
      "loss": 0.0614,
      "step": 54840
    },
    {
      "epoch": 5.229949951171875e-05,
      "model_forward_time": 1.6986911296844482,
      "step": 54840
    },
    {
      "dynamics_loss_avg": 0.06274861060082912,
      "epoch": 5.229949951171875e-05,
      "step": 54840
    },
    {
      "action_loss_avg": 0.0016163306194357575,
      "epoch": 5.229949951171875e-05,
      "step": 54840
    },
    {
      "epoch": 5.229949951171875e-05,
      "step": 54840,
      "training_step_time": 8.44347596168518
    },
    {
      "epoch": 5.230045318603516e-05,
      "model_forward_time": 1.7145514488220215,
      "step": 54841
    },
    {
      "epoch": 5.230045318603516e-05,
      "step": 54841,
      "training_step_time": 7.512721061706543
    },
    {
      "epoch": 5.230140686035156e-05,
      "model_forward_time": 1.7138569355010986,
      "step": 54842
    },
    {
      "epoch": 5.230140686035156e-05,
      "step": 54842,
      "training_step_time": 7.647070646286011
    },
    {
      "epoch": 5.230236053466797e-05,
      "model_forward_time": 1.7121615409851074,
      "step": 54843
    },
    {
      "epoch": 5.230236053466797e-05,
      "step": 54843,
      "training_step_time": 7.607476472854614
    },
    {
      "epoch": 5.2303314208984376e-05,
      "model_forward_time": 1.7047615051269531,
      "step": 54844
    },
    {
      "epoch": 5.2303314208984376e-05,
      "step": 54844,
      "training_step_time": 7.795747518539429
    },
    {
      "epoch": 5.230426788330078e-05,
      "model_forward_time": 1.7131669521331787,
      "step": 54845
    },
    {
      "epoch": 5.230426788330078e-05,
      "step": 54845,
      "training_step_time": 7.657838582992554
    },
    {
      "epoch": 5.2305221557617184e-05,
      "model_forward_time": 1.74983549118042,
      "step": 54846
    },
    {
      "epoch": 5.2305221557617184e-05,
      "step": 54846,
      "training_step_time": 8.654727458953857
    },
    {
      "epoch": 5.2306175231933595e-05,
      "model_forward_time": 1.7069387435913086,
      "step": 54847
    },
    {
      "epoch": 5.2306175231933595e-05,
      "step": 54847,
      "training_step_time": 8.385766744613647
    },
    {
      "epoch": 5.230712890625e-05,
      "model_forward_time": 1.7081458568572998,
      "step": 54848
    },
    {
      "epoch": 5.230712890625e-05,
      "step": 54848,
      "training_step_time": 8.51855993270874
    },
    {
      "epoch": 5.23080825805664e-05,
      "model_forward_time": 1.7121422290802002,
      "step": 54849
    },
    {
      "epoch": 5.23080825805664e-05,
      "step": 54849,
      "training_step_time": 7.60456395149231
    },
    {
      "epoch": 5.2309036254882814e-05,
      "grad_norm": 0.21648511290550232,
      "learning_rate": 4.611990434914894e-06,
      "loss": 0.0568,
      "step": 54850
    },
    {
      "epoch": 5.2309036254882814e-05,
      "model_forward_time": 1.3347384929656982,
      "step": 54850
    },
    {
      "dynamics_loss_avg": 0.06370179988443851,
      "epoch": 5.2309036254882814e-05,
      "step": 54850
    },
    {
      "action_loss_avg": 0.0023510493629146367,
      "epoch": 5.2309036254882814e-05,
      "step": 54850
    },
    {
      "epoch": 5.2309036254882814e-05,
      "step": 54850,
      "training_step_time": 8.375314712524414
    },
    {
      "epoch": 5.230998992919922e-05,
      "model_forward_time": 1.714399814605713,
      "step": 54851
    },
    {
      "epoch": 5.230998992919922e-05,
      "step": 54851,
      "training_step_time": 8.929137945175171
    },
    {
      "epoch": 5.231094360351562e-05,
      "model_forward_time": 1.705526351928711,
      "step": 54852
    },
    {
      "epoch": 5.231094360351562e-05,
      "step": 54852,
      "training_step_time": 7.662503004074097
    },
    {
      "epoch": 5.231189727783203e-05,
      "model_forward_time": 1.717456579208374,
      "step": 54853
    },
    {
      "epoch": 5.231189727783203e-05,
      "step": 54853,
      "training_step_time": 7.748029470443726
    },
    {
      "epoch": 5.231285095214844e-05,
      "model_forward_time": 1.7481286525726318,
      "step": 54854
    },
    {
      "epoch": 5.231285095214844e-05,
      "step": 54854,
      "training_step_time": 9.966911315917969
    },
    {
      "epoch": 5.231380462646484e-05,
      "model_forward_time": 1.7030513286590576,
      "step": 54855
    },
    {
      "epoch": 5.231380462646484e-05,
      "step": 54855,
      "training_step_time": 7.622610807418823
    },
    {
      "epoch": 5.231475830078125e-05,
      "model_forward_time": 1.7583074569702148,
      "step": 54856
    },
    {
      "epoch": 5.231475830078125e-05,
      "step": 54856,
      "training_step_time": 9.235589742660522
    },
    {
      "epoch": 5.2315711975097656e-05,
      "model_forward_time": 2.255542755126953,
      "step": 54857
    },
    {
      "epoch": 5.2315711975097656e-05,
      "step": 54857,
      "training_step_time": 7.810477256774902
    },
    {
      "epoch": 5.231666564941406e-05,
      "model_forward_time": 1.7148785591125488,
      "step": 54858
    },
    {
      "epoch": 5.231666564941406e-05,
      "step": 54858,
      "training_step_time": 7.67242169380188
    },
    {
      "epoch": 5.231761932373047e-05,
      "model_forward_time": 1.7179253101348877,
      "step": 54859
    },
    {
      "epoch": 5.231761932373047e-05,
      "step": 54859,
      "training_step_time": 7.581077575683594
    },
    {
      "epoch": 5.2318572998046874e-05,
      "grad_norm": 0.09998584538698196,
      "learning_rate": 4.6103419725060725e-06,
      "loss": 0.0598,
      "step": 54860
    },
    {
      "epoch": 5.2318572998046874e-05,
      "model_forward_time": 1.699960470199585,
      "step": 54860
    },
    {
      "dynamics_loss_avg": 0.048006010614335536,
      "epoch": 5.2318572998046874e-05,
      "step": 54860
    },
    {
      "action_loss_avg": 0.001392795395804569,
      "epoch": 5.2318572998046874e-05,
      "step": 54860
    },
    {
      "epoch": 5.2318572998046874e-05,
      "step": 54860,
      "training_step_time": 8.501083612442017
    },
    {
      "epoch": 5.231952667236328e-05,
      "model_forward_time": 1.7136719226837158,
      "step": 54861
    },
    {
      "epoch": 5.231952667236328e-05,
      "step": 54861,
      "training_step_time": 8.264702081680298
    },
    {
      "epoch": 5.232048034667969e-05,
      "model_forward_time": 1.7143595218658447,
      "step": 54862
    },
    {
      "epoch": 5.232048034667969e-05,
      "step": 54862,
      "training_step_time": 8.241730690002441
    },
    {
      "epoch": 5.232143402099609e-05,
      "model_forward_time": 1.7086660861968994,
      "step": 54863
    },
    {
      "epoch": 5.232143402099609e-05,
      "step": 54863,
      "training_step_time": 8.69974684715271
    },
    {
      "epoch": 5.23223876953125e-05,
      "model_forward_time": 1.7086217403411865,
      "step": 54864
    },
    {
      "epoch": 5.23223876953125e-05,
      "step": 54864,
      "training_step_time": 7.710187911987305
    },
    {
      "epoch": 5.232334136962891e-05,
      "model_forward_time": 1.7510380744934082,
      "step": 54865
    },
    {
      "epoch": 5.232334136962891e-05,
      "step": 54865,
      "training_step_time": 9.296015739440918
    },
    {
      "epoch": 5.232429504394531e-05,
      "model_forward_time": 1.8611507415771484,
      "step": 54866
    },
    {
      "epoch": 5.232429504394531e-05,
      "step": 54866,
      "training_step_time": 7.931218862533569
    },
    {
      "epoch": 5.2325248718261716e-05,
      "model_forward_time": 1.7058873176574707,
      "step": 54867
    },
    {
      "epoch": 5.2325248718261716e-05,
      "step": 54867,
      "training_step_time": 8.573813915252686
    },
    {
      "epoch": 5.232620239257813e-05,
      "model_forward_time": 1.7085371017456055,
      "step": 54868
    },
    {
      "epoch": 5.232620239257813e-05,
      "step": 54868,
      "training_step_time": 7.620829820632935
    },
    {
      "epoch": 5.232715606689453e-05,
      "model_forward_time": 1.7125685214996338,
      "step": 54869
    },
    {
      "epoch": 5.232715606689453e-05,
      "step": 54869,
      "training_step_time": 7.560812473297119
    },
    {
      "epoch": 5.2328109741210935e-05,
      "grad_norm": 0.09903495013713837,
      "learning_rate": 4.608693552709668e-06,
      "loss": 0.0617,
      "step": 54870
    },
    {
      "epoch": 5.2328109741210935e-05,
      "model_forward_time": 1.704498529434204,
      "step": 54870
    },
    {
      "dynamics_loss_avg": 0.06124347783625126,
      "epoch": 5.2328109741210935e-05,
      "step": 54870
    },
    {
      "action_loss_avg": 0.0013209728873334824,
      "epoch": 5.2328109741210935e-05,
      "step": 54870
    },
    {
      "epoch": 5.2328109741210935e-05,
      "step": 54870,
      "training_step_time": 7.854944467544556
    },
    {
      "epoch": 5.2329063415527346e-05,
      "model_forward_time": 1.7140147686004639,
      "step": 54871
    },
    {
      "epoch": 5.2329063415527346e-05,
      "step": 54871,
      "training_step_time": 7.83884859085083
    },
    {
      "epoch": 5.233001708984375e-05,
      "model_forward_time": 1.693181037902832,
      "step": 54872
    },
    {
      "epoch": 5.233001708984375e-05,
      "step": 54872,
      "training_step_time": 7.788719892501831
    },
    {
      "epoch": 5.2330970764160154e-05,
      "model_forward_time": 1.7235329151153564,
      "step": 54873
    },
    {
      "epoch": 5.2330970764160154e-05,
      "step": 54873,
      "training_step_time": 8.861708402633667
    },
    {
      "epoch": 5.2331924438476565e-05,
      "model_forward_time": 1.7493577003479004,
      "step": 54874
    },
    {
      "epoch": 5.2331924438476565e-05,
      "step": 54874,
      "training_step_time": 7.73174262046814
    },
    {
      "epoch": 5.233287811279297e-05,
      "model_forward_time": 2.284529447555542,
      "step": 54875
    },
    {
      "epoch": 5.233287811279297e-05,
      "step": 54875,
      "training_step_time": 8.041754484176636
    },
    {
      "epoch": 5.233383178710937e-05,
      "model_forward_time": 1.7256970405578613,
      "step": 54876
    },
    {
      "epoch": 5.233383178710937e-05,
      "step": 54876,
      "training_step_time": 8.356444835662842
    },
    {
      "epoch": 5.2334785461425784e-05,
      "model_forward_time": 1.7298462390899658,
      "step": 54877
    },
    {
      "epoch": 5.2334785461425784e-05,
      "step": 54877,
      "training_step_time": 7.627603530883789
    },
    {
      "epoch": 5.233573913574219e-05,
      "model_forward_time": 1.725240707397461,
      "step": 54878
    },
    {
      "epoch": 5.233573913574219e-05,
      "step": 54878,
      "training_step_time": 7.465820074081421
    },
    {
      "epoch": 5.233669281005859e-05,
      "model_forward_time": 1.7311089038848877,
      "step": 54879
    },
    {
      "epoch": 5.233669281005859e-05,
      "step": 54879,
      "training_step_time": 7.94426703453064
    },
    {
      "epoch": 5.2337646484375e-05,
      "grad_norm": 0.09308166801929474,
      "learning_rate": 4.607045175705948e-06,
      "loss": 0.0538,
      "step": 54880
    },
    {
      "epoch": 5.2337646484375e-05,
      "model_forward_time": 1.701249361038208,
      "step": 54880
    },
    {
      "dynamics_loss_avg": 0.05354027897119522,
      "epoch": 5.2337646484375e-05,
      "step": 54880
    },
    {
      "action_loss_avg": 0.0018071196536766365,
      "epoch": 5.2337646484375e-05,
      "step": 54880
    },
    {
      "epoch": 5.2337646484375e-05,
      "step": 54880,
      "training_step_time": 7.825356483459473
    },
    {
      "epoch": 5.2338600158691406e-05,
      "model_forward_time": 1.7107329368591309,
      "step": 54881
    },
    {
      "epoch": 5.2338600158691406e-05,
      "step": 54881,
      "training_step_time": 8.520066976547241
    },
    {
      "epoch": 5.233955383300781e-05,
      "model_forward_time": 1.7071647644042969,
      "step": 54882
    },
    {
      "epoch": 5.233955383300781e-05,
      "step": 54882,
      "training_step_time": 9.33996057510376
    },
    {
      "epoch": 5.234050750732422e-05,
      "model_forward_time": 1.71990966796875,
      "step": 54883
    },
    {
      "epoch": 5.234050750732422e-05,
      "step": 54883,
      "training_step_time": 8.665948152542114
    },
    {
      "epoch": 5.2341461181640625e-05,
      "model_forward_time": 1.7482366561889648,
      "step": 54884
    },
    {
      "epoch": 5.2341461181640625e-05,
      "step": 54884,
      "training_step_time": 7.951332330703735
    },
    {
      "epoch": 5.234241485595703e-05,
      "model_forward_time": 2.243762254714966,
      "step": 54885
    },
    {
      "epoch": 5.234241485595703e-05,
      "step": 54885,
      "training_step_time": 7.6828367710113525
    },
    {
      "epoch": 5.234336853027344e-05,
      "model_forward_time": 1.7126004695892334,
      "step": 54886
    },
    {
      "epoch": 5.234336853027344e-05,
      "step": 54886,
      "training_step_time": 7.668805837631226
    },
    {
      "epoch": 5.2344322204589844e-05,
      "model_forward_time": 1.7107257843017578,
      "step": 54887
    },
    {
      "epoch": 5.2344322204589844e-05,
      "step": 54887,
      "training_step_time": 7.6251935958862305
    },
    {
      "epoch": 5.234527587890625e-05,
      "model_forward_time": 1.7062501907348633,
      "step": 54888
    },
    {
      "epoch": 5.234527587890625e-05,
      "step": 54888,
      "training_step_time": 8.933943271636963
    },
    {
      "epoch": 5.234622955322266e-05,
      "model_forward_time": 1.7104182243347168,
      "step": 54889
    },
    {
      "epoch": 5.234622955322266e-05,
      "step": 54889,
      "training_step_time": 7.773974895477295
    },
    {
      "epoch": 5.234718322753906e-05,
      "grad_norm": 0.12581485509872437,
      "learning_rate": 4.605396841675176e-06,
      "loss": 0.0584,
      "step": 54890
    },
    {
      "epoch": 5.234718322753906e-05,
      "model_forward_time": 1.7009053230285645,
      "step": 54890
    },
    {
      "dynamics_loss_avg": 0.05865322770550847,
      "epoch": 5.234718322753906e-05,
      "step": 54890
    },
    {
      "action_loss_avg": 0.0011753183498512954,
      "epoch": 5.234718322753906e-05,
      "step": 54890
    },
    {
      "epoch": 5.234718322753906e-05,
      "step": 54890,
      "training_step_time": 7.664653539657593
    },
    {
      "epoch": 5.234813690185547e-05,
      "model_forward_time": 1.71616792678833,
      "step": 54891
    },
    {
      "epoch": 5.234813690185547e-05,
      "step": 54891,
      "training_step_time": 7.7644264698028564
    },
    {
      "epoch": 5.234909057617188e-05,
      "model_forward_time": 1.7059147357940674,
      "step": 54892
    },
    {
      "epoch": 5.234909057617188e-05,
      "step": 54892,
      "training_step_time": 7.638151407241821
    },
    {
      "epoch": 5.235004425048828e-05,
      "model_forward_time": 1.7563085556030273,
      "step": 54893
    },
    {
      "epoch": 5.235004425048828e-05,
      "step": 54893,
      "training_step_time": 7.711449861526489
    },
    {
      "epoch": 5.2350997924804686e-05,
      "model_forward_time": 2.267091989517212,
      "step": 54894
    },
    {
      "epoch": 5.2350997924804686e-05,
      "step": 54894,
      "training_step_time": 8.94762921333313
    },
    {
      "epoch": 5.23519515991211e-05,
      "model_forward_time": 1.7039563655853271,
      "step": 54895
    },
    {
      "epoch": 5.23519515991211e-05,
      "step": 54895,
      "training_step_time": 8.75105333328247
    },
    {
      "epoch": 5.23529052734375e-05,
      "model_forward_time": 1.7250800132751465,
      "step": 54896
    },
    {
      "epoch": 5.23529052734375e-05,
      "step": 54896,
      "training_step_time": 8.411033630371094
    },
    {
      "epoch": 5.2353858947753905e-05,
      "model_forward_time": 1.7109100818634033,
      "step": 54897
    },
    {
      "epoch": 5.2353858947753905e-05,
      "step": 54897,
      "training_step_time": 7.812419652938843
    },
    {
      "epoch": 5.2354812622070316e-05,
      "model_forward_time": 1.7076101303100586,
      "step": 54898
    },
    {
      "epoch": 5.2354812622070316e-05,
      "step": 54898,
      "training_step_time": 7.834107398986816
    },
    {
      "epoch": 5.235576629638672e-05,
      "model_forward_time": 1.7066380977630615,
      "step": 54899
    },
    {
      "epoch": 5.235576629638672e-05,
      "step": 54899,
      "training_step_time": 7.64885139465332
    },
    {
      "epoch": 5.2356719970703124e-05,
      "grad_norm": 0.10695282369852066,
      "learning_rate": 4.60374855079761e-06,
      "loss": 0.057,
      "step": 54900
    },
    {
      "epoch": 5.2356719970703124e-05,
      "model_forward_time": 1.7038390636444092,
      "step": 54900
    },
    {
      "dynamics_loss_avg": 0.04142897706478834,
      "epoch": 5.2356719970703124e-05,
      "step": 54900
    },
    {
      "action_loss_avg": 0.001891649523167871,
      "epoch": 5.2356719970703124e-05,
      "step": 54900
    },
    {
      "epoch": 5.2356719970703124e-05,
      "step": 54900,
      "training_step_time": 8.70628023147583
    },
    {
      "epoch": 5.2357673645019534e-05,
      "model_forward_time": 1.7128775119781494,
      "step": 54901
    },
    {
      "epoch": 5.2357673645019534e-05,
      "step": 54901,
      "training_step_time": 7.5625855922698975
    },
    {
      "epoch": 5.235862731933594e-05,
      "model_forward_time": 1.7409749031066895,
      "step": 54902
    },
    {
      "epoch": 5.235862731933594e-05,
      "step": 54902,
      "training_step_time": 8.508303165435791
    },
    {
      "epoch": 5.235958099365234e-05,
      "model_forward_time": 2.2457597255706787,
      "step": 54903
    },
    {
      "epoch": 5.235958099365234e-05,
      "step": 54903,
      "training_step_time": 8.733776092529297
    },
    {
      "epoch": 5.236053466796875e-05,
      "model_forward_time": 1.6855614185333252,
      "step": 54904
    },
    {
      "epoch": 5.236053466796875e-05,
      "step": 54904,
      "training_step_time": 7.728604316711426
    },
    {
      "epoch": 5.236148834228516e-05,
      "model_forward_time": 1.706881046295166,
      "step": 54905
    },
    {
      "epoch": 5.236148834228516e-05,
      "step": 54905,
      "training_step_time": 7.524607181549072
    },
    {
      "epoch": 5.236244201660156e-05,
      "model_forward_time": 1.707176923751831,
      "step": 54906
    },
    {
      "epoch": 5.236244201660156e-05,
      "step": 54906,
      "training_step_time": 8.37722659111023
    },
    {
      "epoch": 5.2363395690917965e-05,
      "model_forward_time": 1.7263972759246826,
      "step": 54907
    },
    {
      "epoch": 5.2363395690917965e-05,
      "step": 54907,
      "training_step_time": 8.430109024047852
    },
    {
      "epoch": 5.2364349365234376e-05,
      "model_forward_time": 1.7287311553955078,
      "step": 54908
    },
    {
      "epoch": 5.2364349365234376e-05,
      "step": 54908,
      "training_step_time": 7.6020283699035645
    },
    {
      "epoch": 5.236530303955078e-05,
      "model_forward_time": 1.7315545082092285,
      "step": 54909
    },
    {
      "epoch": 5.236530303955078e-05,
      "step": 54909,
      "training_step_time": 9.26729702949524
    },
    {
      "epoch": 5.2366256713867184e-05,
      "grad_norm": 0.08730459213256836,
      "learning_rate": 4.60210030325351e-06,
      "loss": 0.0526,
      "step": 54910
    },
    {
      "epoch": 5.2366256713867184e-05,
      "model_forward_time": 1.7148606777191162,
      "step": 54910
    },
    {
      "dynamics_loss_avg": 0.04885344142094254,
      "epoch": 5.2366256713867184e-05,
      "step": 54910
    },
    {
      "action_loss_avg": 0.0017780904105165972,
      "epoch": 5.2366256713867184e-05,
      "step": 54910
    },
    {
      "epoch": 5.2366256713867184e-05,
      "step": 54910,
      "training_step_time": 8.448895454406738
    },
    {
      "epoch": 5.2367210388183595e-05,
      "model_forward_time": 1.752530813217163,
      "step": 54911
    },
    {
      "epoch": 5.2367210388183595e-05,
      "step": 54911,
      "training_step_time": 8.476114511489868
    },
    {
      "epoch": 5.23681640625e-05,
      "model_forward_time": 2.299673318862915,
      "step": 54912
    },
    {
      "epoch": 5.23681640625e-05,
      "step": 54912,
      "training_step_time": 8.608481407165527
    },
    {
      "epoch": 5.23691177368164e-05,
      "model_forward_time": 1.7342796325683594,
      "step": 54913
    },
    {
      "epoch": 5.23691177368164e-05,
      "step": 54913,
      "training_step_time": 7.782384157180786
    },
    {
      "epoch": 5.2370071411132814e-05,
      "model_forward_time": 1.7245903015136719,
      "step": 54914
    },
    {
      "epoch": 5.2370071411132814e-05,
      "step": 54914,
      "training_step_time": 8.129769325256348
    },
    {
      "epoch": 5.237102508544922e-05,
      "model_forward_time": 1.7265625,
      "step": 54915
    },
    {
      "epoch": 5.237102508544922e-05,
      "step": 54915,
      "training_step_time": 7.510643482208252
    },
    {
      "epoch": 5.237197875976562e-05,
      "model_forward_time": 1.733332633972168,
      "step": 54916
    },
    {
      "epoch": 5.237197875976562e-05,
      "step": 54916,
      "training_step_time": 7.791315078735352
    },
    {
      "epoch": 5.237293243408203e-05,
      "model_forward_time": 1.7286365032196045,
      "step": 54917
    },
    {
      "epoch": 5.237293243408203e-05,
      "step": 54917,
      "training_step_time": 7.65718936920166
    },
    {
      "epoch": 5.237388610839844e-05,
      "model_forward_time": 1.7225594520568848,
      "step": 54918
    },
    {
      "epoch": 5.237388610839844e-05,
      "step": 54918,
      "training_step_time": 7.6655871868133545
    },
    {
      "epoch": 5.237483978271484e-05,
      "model_forward_time": 1.7364771366119385,
      "step": 54919
    },
    {
      "epoch": 5.237483978271484e-05,
      "step": 54919,
      "training_step_time": 7.708387851715088
    },
    {
      "epoch": 5.237579345703125e-05,
      "grad_norm": 0.12951484322547913,
      "learning_rate": 4.6004520992231214e-06,
      "loss": 0.0585,
      "step": 54920
    },
    {
      "epoch": 5.237579345703125e-05,
      "model_forward_time": 1.7122156620025635,
      "step": 54920
    },
    {
      "dynamics_loss_avg": 0.05006434954702854,
      "epoch": 5.237579345703125e-05,
      "step": 54920
    },
    {
      "action_loss_avg": 0.0010807679347635713,
      "epoch": 5.237579345703125e-05,
      "step": 54920
    },
    {
      "epoch": 5.237579345703125e-05,
      "step": 54920,
      "training_step_time": 7.629055023193359
    },
    {
      "epoch": 5.2376747131347656e-05,
      "model_forward_time": 2.292055606842041,
      "step": 54921
    },
    {
      "epoch": 5.2376747131347656e-05,
      "step": 54921,
      "training_step_time": 8.633832454681396
    },
    {
      "epoch": 5.237770080566406e-05,
      "model_forward_time": 1.345973014831543,
      "step": 54922
    },
    {
      "epoch": 5.237770080566406e-05,
      "step": 54922,
      "training_step_time": 7.78230619430542
    },
    {
      "epoch": 5.237865447998047e-05,
      "model_forward_time": 1.7243142127990723,
      "step": 54923
    },
    {
      "epoch": 5.237865447998047e-05,
      "step": 54923,
      "training_step_time": 7.690713405609131
    },
    {
      "epoch": 5.2379608154296875e-05,
      "model_forward_time": 1.7238953113555908,
      "step": 54924
    },
    {
      "epoch": 5.2379608154296875e-05,
      "step": 54924,
      "training_step_time": 7.554524660110474
    },
    {
      "epoch": 5.238056182861328e-05,
      "model_forward_time": 1.722747564315796,
      "step": 54925
    },
    {
      "epoch": 5.238056182861328e-05,
      "step": 54925,
      "training_step_time": 8.325629234313965
    },
    {
      "epoch": 5.238151550292969e-05,
      "model_forward_time": 1.719590663909912,
      "step": 54926
    },
    {
      "epoch": 5.238151550292969e-05,
      "step": 54926,
      "training_step_time": 7.800847768783569
    },
    {
      "epoch": 5.238246917724609e-05,
      "model_forward_time": 1.7220327854156494,
      "step": 54927
    },
    {
      "epoch": 5.238246917724609e-05,
      "step": 54927,
      "training_step_time": 8.705169916152954
    },
    {
      "epoch": 5.23834228515625e-05,
      "model_forward_time": 1.725386381149292,
      "step": 54928
    },
    {
      "epoch": 5.23834228515625e-05,
      "step": 54928,
      "training_step_time": 8.255924940109253
    },
    {
      "epoch": 5.238437652587891e-05,
      "model_forward_time": 1.7200002670288086,
      "step": 54929
    },
    {
      "epoch": 5.238437652587891e-05,
      "step": 54929,
      "training_step_time": 7.6353983879089355
    },
    {
      "epoch": 5.238533020019531e-05,
      "grad_norm": 0.11068776249885559,
      "learning_rate": 4.5988039388866915e-06,
      "loss": 0.0557,
      "step": 54930
    },
    {
      "epoch": 5.238533020019531e-05,
      "model_forward_time": 1.7591197490692139,
      "step": 54930
    },
    {
      "dynamics_loss_avg": 0.060135450959205625,
      "epoch": 5.238533020019531e-05,
      "step": 54930
    },
    {
      "action_loss_avg": 0.0017771093174815179,
      "epoch": 5.238533020019531e-05,
      "step": 54930
    },
    {
      "epoch": 5.238533020019531e-05,
      "step": 54930,
      "training_step_time": 7.624657392501831
    },
    {
      "epoch": 5.2386283874511716e-05,
      "model_forward_time": 2.2647206783294678,
      "step": 54931
    },
    {
      "epoch": 5.2386283874511716e-05,
      "step": 54931,
      "training_step_time": 8.447245121002197
    },
    {
      "epoch": 5.238723754882813e-05,
      "model_forward_time": 1.720822811126709,
      "step": 54932
    },
    {
      "epoch": 5.238723754882813e-05,
      "step": 54932,
      "training_step_time": 8.20081877708435
    },
    {
      "epoch": 5.238819122314453e-05,
      "model_forward_time": 1.7187485694885254,
      "step": 54933
    },
    {
      "epoch": 5.238819122314453e-05,
      "step": 54933,
      "training_step_time": 7.53247857093811
    },
    {
      "epoch": 5.2389144897460935e-05,
      "model_forward_time": 1.7243561744689941,
      "step": 54934
    },
    {
      "epoch": 5.2389144897460935e-05,
      "step": 54934,
      "training_step_time": 7.335070371627808
    },
    {
      "epoch": 5.2390098571777346e-05,
      "model_forward_time": 1.7207207679748535,
      "step": 54935
    },
    {
      "epoch": 5.2390098571777346e-05,
      "step": 54935,
      "training_step_time": 7.767454624176025
    },
    {
      "epoch": 5.239105224609375e-05,
      "model_forward_time": 1.7181980609893799,
      "step": 54936
    },
    {
      "epoch": 5.239105224609375e-05,
      "step": 54936,
      "training_step_time": 7.739278793334961
    },
    {
      "epoch": 5.2392005920410154e-05,
      "model_forward_time": 1.7209057807922363,
      "step": 54937
    },
    {
      "epoch": 5.2392005920410154e-05,
      "step": 54937,
      "training_step_time": 8.450285911560059
    },
    {
      "epoch": 5.2392959594726565e-05,
      "model_forward_time": 1.7383005619049072,
      "step": 54938
    },
    {
      "epoch": 5.2392959594726565e-05,
      "step": 54938,
      "training_step_time": 7.601499319076538
    },
    {
      "epoch": 5.239391326904297e-05,
      "model_forward_time": 1.7618827819824219,
      "step": 54939
    },
    {
      "epoch": 5.239391326904297e-05,
      "step": 54939,
      "training_step_time": 7.638907432556152
    },
    {
      "epoch": 5.239486694335937e-05,
      "grad_norm": 0.08482588827610016,
      "learning_rate": 4.597155822424459e-06,
      "loss": 0.0591,
      "step": 54940
    },
    {
      "epoch": 5.239486694335937e-05,
      "model_forward_time": 2.276719808578491,
      "step": 54940
    },
    {
      "dynamics_loss_avg": 0.07736912034451962,
      "epoch": 5.239486694335937e-05,
      "step": 54940
    },
    {
      "action_loss_avg": 0.002128676150459796,
      "epoch": 5.239486694335937e-05,
      "step": 54940
    },
    {
      "epoch": 5.239486694335937e-05,
      "step": 54940,
      "training_step_time": 8.424093246459961
    },
    {
      "epoch": 5.2395820617675784e-05,
      "model_forward_time": 1.7319087982177734,
      "step": 54941
    },
    {
      "epoch": 5.2395820617675784e-05,
      "step": 54941,
      "training_step_time": 7.69256067276001
    },
    {
      "epoch": 5.239677429199219e-05,
      "model_forward_time": 1.7254478931427002,
      "step": 54942
    },
    {
      "epoch": 5.239677429199219e-05,
      "step": 54942,
      "training_step_time": 8.225088834762573
    },
    {
      "epoch": 5.239772796630859e-05,
      "model_forward_time": 1.7201998233795166,
      "step": 54943
    },
    {
      "epoch": 5.239772796630859e-05,
      "step": 54943,
      "training_step_time": 7.434072494506836
    },
    {
      "epoch": 5.2398681640625e-05,
      "model_forward_time": 1.7310888767242432,
      "step": 54944
    },
    {
      "epoch": 5.2398681640625e-05,
      "step": 54944,
      "training_step_time": 7.842301607131958
    },
    {
      "epoch": 5.2399635314941407e-05,
      "model_forward_time": 1.7229652404785156,
      "step": 54945
    },
    {
      "epoch": 5.2399635314941407e-05,
      "step": 54945,
      "training_step_time": 9.084898710250854
    },
    {
      "epoch": 5.240058898925781e-05,
      "model_forward_time": 1.7343063354492188,
      "step": 54946
    },
    {
      "epoch": 5.240058898925781e-05,
      "step": 54946,
      "training_step_time": 8.13394021987915
    },
    {
      "epoch": 5.240154266357422e-05,
      "model_forward_time": 1.7108807563781738,
      "step": 54947
    },
    {
      "epoch": 5.240154266357422e-05,
      "step": 54947,
      "training_step_time": 7.629846811294556
    },
    {
      "epoch": 5.2402496337890625e-05,
      "model_forward_time": 1.7590217590332031,
      "step": 54948
    },
    {
      "epoch": 5.2402496337890625e-05,
      "step": 54948,
      "training_step_time": 8.256580591201782
    },
    {
      "epoch": 5.240345001220703e-05,
      "model_forward_time": 2.2806406021118164,
      "step": 54949
    },
    {
      "epoch": 5.240345001220703e-05,
      "step": 54949,
      "training_step_time": 7.642892360687256
    },
    {
      "epoch": 5.240440368652344e-05,
      "grad_norm": 0.0938069149851799,
      "learning_rate": 4.59550775001666e-06,
      "loss": 0.0554,
      "step": 54950
    },
    {
      "epoch": 5.240440368652344e-05,
      "model_forward_time": 1.7165448665618896,
      "step": 54950
    },
    {
      "dynamics_loss_avg": 0.04558801911771297,
      "epoch": 5.240440368652344e-05,
      "step": 54950
    },
    {
      "action_loss_avg": 0.0013776924344711006,
      "epoch": 5.240440368652344e-05,
      "step": 54950
    },
    {
      "epoch": 5.240440368652344e-05,
      "step": 54950,
      "training_step_time": 8.796834707260132
    },
    {
      "epoch": 5.2405357360839844e-05,
      "model_forward_time": 1.7365920543670654,
      "step": 54951
    },
    {
      "epoch": 5.2405357360839844e-05,
      "step": 54951,
      "training_step_time": 8.09753131866455
    },
    {
      "epoch": 5.240631103515625e-05,
      "model_forward_time": 1.7223525047302246,
      "step": 54952
    },
    {
      "epoch": 5.240631103515625e-05,
      "step": 54952,
      "training_step_time": 8.296428203582764
    },
    {
      "epoch": 5.240726470947266e-05,
      "model_forward_time": 1.7268366813659668,
      "step": 54953
    },
    {
      "epoch": 5.240726470947266e-05,
      "step": 54953,
      "training_step_time": 8.726836442947388
    },
    {
      "epoch": 5.240821838378906e-05,
      "model_forward_time": 1.7293789386749268,
      "step": 54954
    },
    {
      "epoch": 5.240821838378906e-05,
      "step": 54954,
      "training_step_time": 7.72164249420166
    },
    {
      "epoch": 5.240917205810547e-05,
      "model_forward_time": 1.7318089008331299,
      "step": 54955
    },
    {
      "epoch": 5.240917205810547e-05,
      "step": 54955,
      "training_step_time": 9.001519680023193
    },
    {
      "epoch": 5.241012573242188e-05,
      "model_forward_time": 1.732595443725586,
      "step": 54956
    },
    {
      "epoch": 5.241012573242188e-05,
      "step": 54956,
      "training_step_time": 9.451573848724365
    },
    {
      "epoch": 5.241107940673828e-05,
      "model_forward_time": 1.748763084411621,
      "step": 54957
    },
    {
      "epoch": 5.241107940673828e-05,
      "step": 54957,
      "training_step_time": 8.16414213180542
    },
    {
      "epoch": 5.2412033081054686e-05,
      "model_forward_time": 2.3124308586120605,
      "step": 54958
    },
    {
      "epoch": 5.2412033081054686e-05,
      "step": 54958,
      "training_step_time": 7.564502954483032
    },
    {
      "epoch": 5.24129867553711e-05,
      "model_forward_time": 1.7152001857757568,
      "step": 54959
    },
    {
      "epoch": 5.24129867553711e-05,
      "step": 54959,
      "training_step_time": 7.741179704666138
    },
    {
      "epoch": 5.24139404296875e-05,
      "grad_norm": 0.13517068326473236,
      "learning_rate": 4.593859721843523e-06,
      "loss": 0.0572,
      "step": 54960
    },
    {
      "epoch": 5.24139404296875e-05,
      "model_forward_time": 1.717829704284668,
      "step": 54960
    },
    {
      "dynamics_loss_avg": 0.04868734749034047,
      "epoch": 5.24139404296875e-05,
      "step": 54960
    },
    {
      "action_loss_avg": 0.0016766974003985524,
      "epoch": 5.24139404296875e-05,
      "step": 54960
    },
    {
      "epoch": 5.24139404296875e-05,
      "step": 54960,
      "training_step_time": 7.503349781036377
    },
    {
      "epoch": 5.2414894104003905e-05,
      "model_forward_time": 1.741011381149292,
      "step": 54961
    },
    {
      "epoch": 5.2414894104003905e-05,
      "step": 54961,
      "training_step_time": 7.4962239265441895
    },
    {
      "epoch": 5.2415847778320316e-05,
      "model_forward_time": 1.7228341102600098,
      "step": 54962
    },
    {
      "epoch": 5.2415847778320316e-05,
      "step": 54962,
      "training_step_time": 7.535522937774658
    },
    {
      "epoch": 5.241680145263672e-05,
      "model_forward_time": 1.7242298126220703,
      "step": 54963
    },
    {
      "epoch": 5.241680145263672e-05,
      "step": 54963,
      "training_step_time": 8.41580605506897
    },
    {
      "epoch": 5.2417755126953124e-05,
      "model_forward_time": 1.7290396690368652,
      "step": 54964
    },
    {
      "epoch": 5.2417755126953124e-05,
      "step": 54964,
      "training_step_time": 7.671183347702026
    },
    {
      "epoch": 5.2418708801269535e-05,
      "model_forward_time": 1.727663516998291,
      "step": 54965
    },
    {
      "epoch": 5.2418708801269535e-05,
      "step": 54965,
      "training_step_time": 7.701988697052002
    },
    {
      "epoch": 5.241966247558594e-05,
      "model_forward_time": 1.727442979812622,
      "step": 54966
    },
    {
      "epoch": 5.241966247558594e-05,
      "step": 54966,
      "training_step_time": 9.572370052337646
    },
    {
      "epoch": 5.242061614990234e-05,
      "model_forward_time": 2.2979347705841064,
      "step": 54967
    },
    {
      "epoch": 5.242061614990234e-05,
      "step": 54967,
      "training_step_time": 7.774343490600586
    },
    {
      "epoch": 5.242156982421875e-05,
      "model_forward_time": 1.7222323417663574,
      "step": 54968
    },
    {
      "epoch": 5.242156982421875e-05,
      "step": 54968,
      "training_step_time": 8.775085926055908
    },
    {
      "epoch": 5.242252349853516e-05,
      "model_forward_time": 1.7289206981658936,
      "step": 54969
    },
    {
      "epoch": 5.242252349853516e-05,
      "step": 54969,
      "training_step_time": 7.571821451187134
    },
    {
      "epoch": 5.242347717285156e-05,
      "grad_norm": 0.1272992044687271,
      "learning_rate": 4.592211738085279e-06,
      "loss": 0.0598,
      "step": 54970
    },
    {
      "epoch": 5.242347717285156e-05,
      "model_forward_time": 1.7348175048828125,
      "step": 54970
    },
    {
      "dynamics_loss_avg": 0.057652748003602025,
      "epoch": 5.242347717285156e-05,
      "step": 54970
    },
    {
      "action_loss_avg": 0.0013638574338983745,
      "epoch": 5.242347717285156e-05,
      "step": 54970
    },
    {
      "epoch": 5.242347717285156e-05,
      "step": 54970,
      "training_step_time": 7.586238384246826
    },
    {
      "epoch": 5.2424430847167965e-05,
      "model_forward_time": 1.7280917167663574,
      "step": 54971
    },
    {
      "epoch": 5.2424430847167965e-05,
      "step": 54971,
      "training_step_time": 7.645569801330566
    },
    {
      "epoch": 5.2425384521484376e-05,
      "model_forward_time": 1.7347412109375,
      "step": 54972
    },
    {
      "epoch": 5.2425384521484376e-05,
      "step": 54972,
      "training_step_time": 7.698215961456299
    },
    {
      "epoch": 5.242633819580078e-05,
      "model_forward_time": 1.7321500778198242,
      "step": 54973
    },
    {
      "epoch": 5.242633819580078e-05,
      "step": 54973,
      "training_step_time": 7.694051027297974
    },
    {
      "epoch": 5.2427291870117184e-05,
      "model_forward_time": 1.737800121307373,
      "step": 54974
    },
    {
      "epoch": 5.2427291870117184e-05,
      "step": 54974,
      "training_step_time": 8.566849708557129
    },
    {
      "epoch": 5.2428245544433595e-05,
      "model_forward_time": 1.7409567832946777,
      "step": 54975
    },
    {
      "epoch": 5.2428245544433595e-05,
      "step": 54975,
      "training_step_time": 9.206050395965576
    },
    {
      "epoch": 5.242919921875e-05,
      "model_forward_time": 1.7821087837219238,
      "step": 54976
    },
    {
      "epoch": 5.242919921875e-05,
      "step": 54976,
      "training_step_time": 8.609270334243774
    },
    {
      "epoch": 5.24301528930664e-05,
      "model_forward_time": 2.2850513458251953,
      "step": 54977
    },
    {
      "epoch": 5.24301528930664e-05,
      "step": 54977,
      "training_step_time": 8.517534255981445
    },
    {
      "epoch": 5.2431106567382814e-05,
      "model_forward_time": 1.7419164180755615,
      "step": 54978
    },
    {
      "epoch": 5.2431106567382814e-05,
      "step": 54978,
      "training_step_time": 7.714188575744629
    },
    {
      "epoch": 5.243206024169922e-05,
      "model_forward_time": 1.746429204940796,
      "step": 54979
    },
    {
      "epoch": 5.243206024169922e-05,
      "step": 54979,
      "training_step_time": 7.624113082885742
    },
    {
      "epoch": 5.243301391601562e-05,
      "grad_norm": 0.10024872422218323,
      "learning_rate": 4.590563798922146e-06,
      "loss": 0.0577,
      "step": 54980
    },
    {
      "epoch": 5.243301391601562e-05,
      "model_forward_time": 1.724506139755249,
      "step": 54980
    },
    {
      "dynamics_loss_avg": 0.04418092295527458,
      "epoch": 5.243301391601562e-05,
      "step": 54980
    },
    {
      "action_loss_avg": 0.0013328924527741038,
      "epoch": 5.243301391601562e-05,
      "step": 54980
    },
    {
      "epoch": 5.243301391601562e-05,
      "step": 54980,
      "training_step_time": 7.749571323394775
    },
    {
      "epoch": 5.243396759033203e-05,
      "model_forward_time": 1.7416157722473145,
      "step": 54981
    },
    {
      "epoch": 5.243396759033203e-05,
      "step": 54981,
      "training_step_time": 7.842166185379028
    },
    {
      "epoch": 5.243492126464844e-05,
      "model_forward_time": 1.7375755310058594,
      "step": 54982
    },
    {
      "epoch": 5.243492126464844e-05,
      "step": 54982,
      "training_step_time": 7.775946855545044
    },
    {
      "epoch": 5.243587493896484e-05,
      "model_forward_time": 1.7134971618652344,
      "step": 54983
    },
    {
      "epoch": 5.243587493896484e-05,
      "step": 54983,
      "training_step_time": 8.585657119750977
    },
    {
      "epoch": 5.243682861328125e-05,
      "model_forward_time": 1.7308425903320312,
      "step": 54984
    },
    {
      "epoch": 5.243682861328125e-05,
      "step": 54984,
      "training_step_time": 7.5443878173828125
    },
    {
      "epoch": 5.2437782287597656e-05,
      "model_forward_time": 1.7621197700500488,
      "step": 54985
    },
    {
      "epoch": 5.2437782287597656e-05,
      "step": 54985,
      "training_step_time": 7.711775541305542
    },
    {
      "epoch": 5.243873596191406e-05,
      "model_forward_time": 2.2627153396606445,
      "step": 54986
    },
    {
      "epoch": 5.243873596191406e-05,
      "step": 54986,
      "training_step_time": 7.802269697189331
    },
    {
      "epoch": 5.243968963623047e-05,
      "model_forward_time": 1.7369933128356934,
      "step": 54987
    },
    {
      "epoch": 5.243968963623047e-05,
      "step": 54987,
      "training_step_time": 7.862412929534912
    },
    {
      "epoch": 5.2440643310546875e-05,
      "model_forward_time": 1.7290771007537842,
      "step": 54988
    },
    {
      "epoch": 5.2440643310546875e-05,
      "step": 54988,
      "training_step_time": 7.644012451171875
    },
    {
      "epoch": 5.244159698486328e-05,
      "model_forward_time": 1.7150237560272217,
      "step": 54989
    },
    {
      "epoch": 5.244159698486328e-05,
      "step": 54989,
      "training_step_time": 7.664176940917969
    },
    {
      "epoch": 5.244255065917969e-05,
      "grad_norm": 0.12570706009864807,
      "learning_rate": 4.588915904534341e-06,
      "loss": 0.058,
      "step": 54990
    },
    {
      "epoch": 5.244255065917969e-05,
      "model_forward_time": 1.732212781906128,
      "step": 54990
    },
    {
      "dynamics_loss_avg": 0.0519476305693388,
      "epoch": 5.244255065917969e-05,
      "step": 54990
    },
    {
      "action_loss_avg": 0.0014511933317407967,
      "epoch": 5.244255065917969e-05,
      "step": 54990
    },
    {
      "epoch": 5.244255065917969e-05,
      "step": 54990,
      "training_step_time": 8.895563125610352
    },
    {
      "epoch": 5.2443504333496093e-05,
      "model_forward_time": 1.7266502380371094,
      "step": 54991
    },
    {
      "epoch": 5.2443504333496093e-05,
      "step": 54991,
      "training_step_time": 7.999126195907593
    },
    {
      "epoch": 5.24444580078125e-05,
      "model_forward_time": 1.7314178943634033,
      "step": 54992
    },
    {
      "epoch": 5.24444580078125e-05,
      "step": 54992,
      "training_step_time": 7.936386585235596
    },
    {
      "epoch": 5.244541168212891e-05,
      "model_forward_time": 1.7256755828857422,
      "step": 54993
    },
    {
      "epoch": 5.244541168212891e-05,
      "step": 54993,
      "training_step_time": 8.77512264251709
    },
    {
      "epoch": 5.244636535644531e-05,
      "model_forward_time": 1.764700174331665,
      "step": 54994
    },
    {
      "epoch": 5.244636535644531e-05,
      "step": 54994,
      "training_step_time": 7.874972820281982
    },
    {
      "epoch": 5.2447319030761716e-05,
      "model_forward_time": 2.2779669761657715,
      "step": 54995
    },
    {
      "epoch": 5.2447319030761716e-05,
      "step": 54995,
      "training_step_time": 7.8513734340667725
    },
    {
      "epoch": 5.244827270507813e-05,
      "model_forward_time": 1.718144416809082,
      "step": 54996
    },
    {
      "epoch": 5.244827270507813e-05,
      "step": 54996,
      "training_step_time": 8.058900594711304
    },
    {
      "epoch": 5.244922637939453e-05,
      "model_forward_time": 1.7341933250427246,
      "step": 54997
    },
    {
      "epoch": 5.244922637939453e-05,
      "step": 54997,
      "training_step_time": 7.682382345199585
    },
    {
      "epoch": 5.2450180053710935e-05,
      "model_forward_time": 1.7430665493011475,
      "step": 54998
    },
    {
      "epoch": 5.2450180053710935e-05,
      "step": 54998,
      "training_step_time": 7.730106592178345
    },
    {
      "epoch": 5.2451133728027346e-05,
      "model_forward_time": 1.7423310279846191,
      "step": 54999
    },
    {
      "epoch": 5.2451133728027346e-05,
      "step": 54999,
      "training_step_time": 9.085930585861206
    },
    {
      "epoch": 5.245208740234375e-05,
      "grad_norm": 0.10788732022047043,
      "learning_rate": 4.587268055102073e-06,
      "loss": 0.0587,
      "step": 55000
    },
    {
      "epoch": 5.245208740234375e-05,
      "model_forward_time": 1.7549147605895996,
      "step": 55000
    },
    {
      "dynamics_loss_avg": 0.03491480629891157,
      "epoch": 5.245208740234375e-05,
      "step": 55000
    },
    {
      "action_loss_avg": 0.0011311769281746819,
      "epoch": 5.245208740234375e-05,
      "step": 55000
    },
    {
      "epoch": 5.245208740234375e-05,
      "step": 55000,
      "training_step_time": 7.578790903091431
    },
    {
      "epoch": 5.2453041076660154e-05,
      "model_forward_time": 1.7280457019805908,
      "step": 55001
    },
    {
      "epoch": 5.2453041076660154e-05,
      "step": 55001,
      "training_step_time": 7.649662494659424
    },
    {
      "epoch": 5.2453994750976565e-05,
      "model_forward_time": 1.7323296070098877,
      "step": 55002
    },
    {
      "epoch": 5.2453994750976565e-05,
      "step": 55002,
      "training_step_time": 8.306989669799805
    },
    {
      "epoch": 5.245494842529297e-05,
      "model_forward_time": 1.7191812992095947,
      "step": 55003
    },
    {
      "epoch": 5.245494842529297e-05,
      "step": 55003,
      "training_step_time": 10.027576923370361
    },
    {
      "epoch": 5.245590209960937e-05,
      "model_forward_time": 1.720346450805664,
      "step": 55004
    },
    {
      "epoch": 5.245590209960937e-05,
      "step": 55004,
      "training_step_time": 10.662562847137451
    },
    {
      "epoch": 5.2456855773925784e-05,
      "model_forward_time": 1.728369951248169,
      "step": 55005
    },
    {
      "epoch": 5.2456855773925784e-05,
      "step": 55005,
      "training_step_time": 7.668047904968262
    },
    {
      "epoch": 5.245780944824219e-05,
      "model_forward_time": 1.7199828624725342,
      "step": 55006
    },
    {
      "epoch": 5.245780944824219e-05,
      "step": 55006,
      "training_step_time": 8.567463159561157
    },
    {
      "epoch": 5.245876312255859e-05,
      "model_forward_time": 1.7249581813812256,
      "step": 55007
    },
    {
      "epoch": 5.245876312255859e-05,
      "step": 55007,
      "training_step_time": 8.126287460327148
    },
    {
      "epoch": 5.2459716796875e-05,
      "model_forward_time": 1.721829891204834,
      "step": 55008
    },
    {
      "epoch": 5.2459716796875e-05,
      "step": 55008,
      "training_step_time": 9.679925918579102
    },
    {
      "epoch": 5.2460670471191407e-05,
      "model_forward_time": 1.7268273830413818,
      "step": 55009
    },
    {
      "epoch": 5.2460670471191407e-05,
      "step": 55009,
      "training_step_time": 9.454534769058228
    },
    {
      "epoch": 5.246162414550781e-05,
      "grad_norm": 0.09803103655576706,
      "learning_rate": 4.585620250805552e-06,
      "loss": 0.0583,
      "step": 55010
    },
    {
      "epoch": 5.246162414550781e-05,
      "model_forward_time": 1.7120325565338135,
      "step": 55010
    },
    {
      "dynamics_loss_avg": 0.07225541137158871,
      "epoch": 5.246162414550781e-05,
      "step": 55010
    },
    {
      "action_loss_avg": 0.0016714958735974505,
      "epoch": 5.246162414550781e-05,
      "step": 55010
    },
    {
      "epoch": 5.246162414550781e-05,
      "step": 55010,
      "training_step_time": 8.47656798362732
    },
    {
      "epoch": 5.246257781982422e-05,
      "model_forward_time": 1.7250909805297852,
      "step": 55011
    },
    {
      "epoch": 5.246257781982422e-05,
      "step": 55011,
      "training_step_time": 7.588341474533081
    },
    {
      "epoch": 5.2463531494140625e-05,
      "model_forward_time": 1.723442554473877,
      "step": 55012
    },
    {
      "epoch": 5.2463531494140625e-05,
      "step": 55012,
      "training_step_time": 8.660919189453125
    },
    {
      "epoch": 5.246448516845703e-05,
      "model_forward_time": 2.289611339569092,
      "step": 55013
    },
    {
      "epoch": 5.246448516845703e-05,
      "step": 55013,
      "training_step_time": 8.52778697013855
    },
    {
      "epoch": 5.246543884277344e-05,
      "model_forward_time": 1.7196991443634033,
      "step": 55014
    },
    {
      "epoch": 5.246543884277344e-05,
      "step": 55014,
      "training_step_time": 7.709063291549683
    },
    {
      "epoch": 5.2466392517089844e-05,
      "model_forward_time": 1.7137115001678467,
      "step": 55015
    },
    {
      "epoch": 5.2466392517089844e-05,
      "step": 55015,
      "training_step_time": 8.244599103927612
    },
    {
      "epoch": 5.246734619140625e-05,
      "model_forward_time": 1.352226972579956,
      "step": 55016
    },
    {
      "epoch": 5.246734619140625e-05,
      "step": 55016,
      "training_step_time": 7.995966672897339
    },
    {
      "epoch": 5.246829986572266e-05,
      "model_forward_time": 1.7293345928192139,
      "step": 55017
    },
    {
      "epoch": 5.246829986572266e-05,
      "step": 55017,
      "training_step_time": 7.56615686416626
    },
    {
      "epoch": 5.246925354003906e-05,
      "model_forward_time": 1.700441598892212,
      "step": 55018
    },
    {
      "epoch": 5.246925354003906e-05,
      "step": 55018,
      "training_step_time": 8.609118938446045
    },
    {
      "epoch": 5.247020721435547e-05,
      "model_forward_time": 1.7153778076171875,
      "step": 55019
    },
    {
      "epoch": 5.247020721435547e-05,
      "step": 55019,
      "training_step_time": 8.688237190246582
    },
    {
      "epoch": 5.247116088867188e-05,
      "grad_norm": 0.20890557765960693,
      "learning_rate": 4.583972491824975e-06,
      "loss": 0.059,
      "step": 55020
    },
    {
      "epoch": 5.247116088867188e-05,
      "model_forward_time": 1.7091000080108643,
      "step": 55020
    },
    {
      "dynamics_loss_avg": 0.045350554026663305,
      "epoch": 5.247116088867188e-05,
      "step": 55020
    },
    {
      "action_loss_avg": 0.0013185637653805315,
      "epoch": 5.247116088867188e-05,
      "step": 55020
    },
    {
      "epoch": 5.247116088867188e-05,
      "step": 55020,
      "training_step_time": 7.744306802749634
    },
    {
      "epoch": 5.247211456298828e-05,
      "model_forward_time": 1.7220659255981445,
      "step": 55021
    },
    {
      "epoch": 5.247211456298828e-05,
      "step": 55021,
      "training_step_time": 9.902740240097046
    },
    {
      "epoch": 5.2473068237304686e-05,
      "model_forward_time": 2.282309055328369,
      "step": 55022
    },
    {
      "epoch": 5.2473068237304686e-05,
      "step": 55022,
      "training_step_time": 9.77168869972229
    },
    {
      "epoch": 5.24740219116211e-05,
      "model_forward_time": 1.719740629196167,
      "step": 55023
    },
    {
      "epoch": 5.24740219116211e-05,
      "step": 55023,
      "training_step_time": 8.577503442764282
    },
    {
      "epoch": 5.24749755859375e-05,
      "model_forward_time": 1.722661018371582,
      "step": 55024
    },
    {
      "epoch": 5.24749755859375e-05,
      "step": 55024,
      "training_step_time": 6.914914846420288
    },
    {
      "epoch": 5.2475929260253905e-05,
      "model_forward_time": 1.7371644973754883,
      "step": 55025
    },
    {
      "epoch": 5.2475929260253905e-05,
      "step": 55025,
      "training_step_time": 6.9029786586761475
    },
    {
      "epoch": 5.2476882934570316e-05,
      "model_forward_time": 1.7163465023040771,
      "step": 55026
    },
    {
      "epoch": 5.2476882934570316e-05,
      "step": 55026,
      "training_step_time": 6.951090335845947
    },
    {
      "epoch": 5.247783660888672e-05,
      "model_forward_time": 1.7172183990478516,
      "step": 55027
    },
    {
      "epoch": 5.247783660888672e-05,
      "step": 55027,
      "training_step_time": 6.92945122718811
    },
    {
      "epoch": 5.2478790283203124e-05,
      "model_forward_time": 1.7169222831726074,
      "step": 55028
    },
    {
      "epoch": 5.2478790283203124e-05,
      "step": 55028,
      "training_step_time": 6.93749475479126
    },
    {
      "epoch": 5.2479743957519535e-05,
      "model_forward_time": 1.7180845737457275,
      "step": 55029
    },
    {
      "epoch": 5.2479743957519535e-05,
      "step": 55029,
      "training_step_time": 6.986644268035889
    },
    {
      "epoch": 5.248069763183594e-05,
      "grad_norm": 0.0951901525259018,
      "learning_rate": 4.582324778340541e-06,
      "loss": 0.059,
      "step": 55030
    },
    {
      "epoch": 5.248069763183594e-05,
      "model_forward_time": 1.7124037742614746,
      "step": 55030
    },
    {
      "dynamics_loss_avg": 0.057970835082232955,
      "epoch": 5.248069763183594e-05,
      "step": 55030
    },
    {
      "action_loss_avg": 0.0018485797918401659,
      "epoch": 5.248069763183594e-05,
      "step": 55030
    },
    {
      "epoch": 5.248069763183594e-05,
      "step": 55030,
      "training_step_time": 6.972763538360596
    },
    {
      "epoch": 5.248165130615234e-05,
      "model_forward_time": 1.3475673198699951,
      "step": 55031
    },
    {
      "epoch": 5.248165130615234e-05,
      "step": 55031,
      "training_step_time": 7.9085853099823
    },
    {
      "epoch": 5.248260498046875e-05,
      "model_forward_time": 2.273102283477783,
      "step": 55032
    },
    {
      "epoch": 5.248260498046875e-05,
      "step": 55032,
      "training_step_time": 8.028783798217773
    },
    {
      "epoch": 5.248355865478516e-05,
      "model_forward_time": 1.7190120220184326,
      "step": 55033
    },
    {
      "epoch": 5.248355865478516e-05,
      "step": 55033,
      "training_step_time": 7.743873357772827
    },
    {
      "epoch": 5.248451232910156e-05,
      "model_forward_time": 1.716700553894043,
      "step": 55034
    },
    {
      "epoch": 5.248451232910156e-05,
      "step": 55034,
      "training_step_time": 6.962040185928345
    },
    {
      "epoch": 5.2485466003417965e-05,
      "model_forward_time": 1.7173545360565186,
      "step": 55035
    },
    {
      "epoch": 5.2485466003417965e-05,
      "step": 55035,
      "training_step_time": 6.984382629394531
    },
    {
      "epoch": 5.2486419677734376e-05,
      "model_forward_time": 1.0114312171936035,
      "step": 55036
    },
    {
      "epoch": 5.2486419677734376e-05,
      "step": 55036,
      "training_step_time": 6.948550224304199
    },
    {
      "epoch": 5.248737335205078e-05,
      "model_forward_time": 1.7182908058166504,
      "step": 55037
    },
    {
      "epoch": 5.248737335205078e-05,
      "step": 55037,
      "training_step_time": 6.947926044464111
    },
    {
      "epoch": 5.2488327026367184e-05,
      "model_forward_time": 1.7124402523040771,
      "step": 55038
    },
    {
      "epoch": 5.2488327026367184e-05,
      "step": 55038,
      "training_step_time": 6.960149049758911
    },
    {
      "epoch": 5.2489280700683595e-05,
      "model_forward_time": 1.7203104496002197,
      "step": 55039
    },
    {
      "epoch": 5.2489280700683595e-05,
      "step": 55039,
      "training_step_time": 6.984173059463501
    },
    {
      "epoch": 5.2490234375e-05,
      "grad_norm": 0.09302845597267151,
      "learning_rate": 4.580677110532443e-06,
      "loss": 0.0606,
      "step": 55040
    },
    {
      "epoch": 5.2490234375e-05,
      "model_forward_time": 1.7480931282043457,
      "step": 55040
    },
    {
      "dynamics_loss_avg": 0.04682696759700775,
      "epoch": 5.2490234375e-05,
      "step": 55040
    },
    {
      "action_loss_avg": 0.0013614575465908274,
      "epoch": 5.2490234375e-05,
      "step": 55040
    },
    {
      "epoch": 5.2490234375e-05,
      "step": 55040,
      "training_step_time": 6.887247323989868
    },
    {
      "epoch": 5.24911880493164e-05,
      "model_forward_time": 2.2579784393310547,
      "step": 55041
    },
    {
      "epoch": 5.24911880493164e-05,
      "step": 55041,
      "training_step_time": 11.039328813552856
    },
    {
      "epoch": 5.2492141723632814e-05,
      "model_forward_time": 1.715662956237793,
      "step": 55042
    },
    {
      "epoch": 5.2492141723632814e-05,
      "step": 55042,
      "training_step_time": 7.854955673217773
    },
    {
      "epoch": 5.249309539794922e-05,
      "model_forward_time": 1.7208192348480225,
      "step": 55043
    },
    {
      "epoch": 5.249309539794922e-05,
      "step": 55043,
      "training_step_time": 6.943942308425903
    },
    {
      "epoch": 5.249404907226562e-05,
      "model_forward_time": 1.7153584957122803,
      "step": 55044
    },
    {
      "epoch": 5.249404907226562e-05,
      "step": 55044,
      "training_step_time": 6.897150039672852
    },
    {
      "epoch": 5.249500274658203e-05,
      "model_forward_time": 1.7052619457244873,
      "step": 55045
    },
    {
      "epoch": 5.249500274658203e-05,
      "step": 55045,
      "training_step_time": 6.948684453964233
    },
    {
      "epoch": 5.249595642089844e-05,
      "model_forward_time": 1.3499505519866943,
      "step": 55046
    },
    {
      "epoch": 5.249595642089844e-05,
      "step": 55046,
      "training_step_time": 6.915465831756592
    },
    {
      "epoch": 5.249691009521484e-05,
      "model_forward_time": 1.7323362827301025,
      "step": 55047
    },
    {
      "epoch": 5.249691009521484e-05,
      "step": 55047,
      "training_step_time": 6.972294807434082
    },
    {
      "epoch": 5.249786376953125e-05,
      "model_forward_time": 1.7239406108856201,
      "step": 55048
    },
    {
      "epoch": 5.249786376953125e-05,
      "step": 55048,
      "training_step_time": 6.933873414993286
    },
    {
      "epoch": 5.2498817443847656e-05,
      "model_forward_time": 1.7173888683319092,
      "step": 55049
    },
    {
      "epoch": 5.2498817443847656e-05,
      "step": 55049,
      "training_step_time": 6.947190523147583
    },
    {
      "epoch": 5.249977111816406e-05,
      "grad_norm": 0.12402047216892242,
      "learning_rate": 4.579029488580865e-06,
      "loss": 0.0576,
      "step": 55050
    },
    {
      "epoch": 5.249977111816406e-05,
      "model_forward_time": 1.7091262340545654,
      "step": 55050
    },
    {
      "dynamics_loss_avg": 0.05040663685649634,
      "epoch": 5.249977111816406e-05,
      "step": 55050
    },
    {
      "action_loss_avg": 0.00175006955396384,
      "epoch": 5.249977111816406e-05,
      "step": 55050
    },
    {
      "epoch": 5.249977111816406e-05,
      "step": 55050,
      "training_step_time": 10.131383180618286
    },
    {
      "epoch": 5.250072479248047e-05,
      "model_forward_time": 2.2506587505340576,
      "step": 55051
    },
    {
      "epoch": 5.250072479248047e-05,
      "step": 55051,
      "training_step_time": 7.915747165679932
    },
    {
      "epoch": 5.2501678466796875e-05,
      "model_forward_time": 1.720900058746338,
      "step": 55052
    },
    {
      "epoch": 5.2501678466796875e-05,
      "step": 55052,
      "training_step_time": 7.237229585647583
    },
    {
      "epoch": 5.250263214111328e-05,
      "model_forward_time": 1.7226884365081787,
      "step": 55053
    },
    {
      "epoch": 5.250263214111328e-05,
      "step": 55053,
      "training_step_time": 6.893427133560181
    },
    {
      "epoch": 5.250358581542969e-05,
      "model_forward_time": 1.7290217876434326,
      "step": 55054
    },
    {
      "epoch": 5.250358581542969e-05,
      "step": 55054,
      "training_step_time": 6.794013500213623
    },
    {
      "epoch": 5.2504539489746093e-05,
      "model_forward_time": 1.7340459823608398,
      "step": 55055
    },
    {
      "epoch": 5.2504539489746093e-05,
      "step": 55055,
      "training_step_time": 6.784182786941528
    },
    {
      "epoch": 5.25054931640625e-05,
      "model_forward_time": 1.7362234592437744,
      "step": 55056
    },
    {
      "epoch": 5.25054931640625e-05,
      "step": 55056,
      "training_step_time": 6.859573125839233
    },
    {
      "epoch": 5.250644683837891e-05,
      "model_forward_time": 1.72575044631958,
      "step": 55057
    },
    {
      "epoch": 5.250644683837891e-05,
      "step": 55057,
      "training_step_time": 6.799229621887207
    },
    {
      "epoch": 5.250740051269531e-05,
      "model_forward_time": 1.7283716201782227,
      "step": 55058
    },
    {
      "epoch": 5.250740051269531e-05,
      "step": 55058,
      "training_step_time": 6.7727625370025635
    },
    {
      "epoch": 5.2508354187011716e-05,
      "model_forward_time": 1.7323250770568848,
      "step": 55059
    },
    {
      "epoch": 5.2508354187011716e-05,
      "step": 55059,
      "training_step_time": 11.234116554260254
    },
    {
      "epoch": 5.250930786132813e-05,
      "grad_norm": 0.09439779818058014,
      "learning_rate": 4.577381912665988e-06,
      "loss": 0.0627,
      "step": 55060
    },
    {
      "epoch": 5.250930786132813e-05,
      "model_forward_time": 2.2808327674865723,
      "step": 55060
    },
    {
      "dynamics_loss_avg": 0.05626495145261288,
      "epoch": 5.250930786132813e-05,
      "step": 55060
    },
    {
      "action_loss_avg": 0.001567959869862534,
      "epoch": 5.250930786132813e-05,
      "step": 55060
    },
    {
      "epoch": 5.250930786132813e-05,
      "step": 55060,
      "training_step_time": 7.724655628204346
    },
    {
      "epoch": 5.251026153564453e-05,
      "model_forward_time": 1.7194392681121826,
      "step": 55061
    },
    {
      "epoch": 5.251026153564453e-05,
      "step": 55061,
      "training_step_time": 7.572282075881958
    },
    {
      "epoch": 5.2511215209960935e-05,
      "model_forward_time": 1.7227394580841064,
      "step": 55062
    },
    {
      "epoch": 5.2511215209960935e-05,
      "step": 55062,
      "training_step_time": 6.84477686882019
    },
    {
      "epoch": 5.2512168884277346e-05,
      "model_forward_time": 1.3510971069335938,
      "step": 55063
    },
    {
      "epoch": 5.2512168884277346e-05,
      "step": 55063,
      "training_step_time": 6.787980318069458
    },
    {
      "epoch": 5.251312255859375e-05,
      "model_forward_time": 1.7086608409881592,
      "step": 55064
    },
    {
      "epoch": 5.251312255859375e-05,
      "step": 55064,
      "training_step_time": 6.797883033752441
    },
    {
      "epoch": 5.2514076232910154e-05,
      "model_forward_time": 1.734591007232666,
      "step": 55065
    },
    {
      "epoch": 5.2514076232910154e-05,
      "step": 55065,
      "training_step_time": 6.832372665405273
    },
    {
      "epoch": 5.2515029907226565e-05,
      "model_forward_time": 1.7066493034362793,
      "step": 55066
    },
    {
      "epoch": 5.2515029907226565e-05,
      "step": 55066,
      "training_step_time": 6.796650409698486
    },
    {
      "epoch": 5.251598358154297e-05,
      "model_forward_time": 1.7292072772979736,
      "step": 55067
    },
    {
      "epoch": 5.251598358154297e-05,
      "step": 55067,
      "training_step_time": 6.8313307762146
    },
    {
      "epoch": 5.251693725585937e-05,
      "model_forward_time": 1.7239124774932861,
      "step": 55068
    },
    {
      "epoch": 5.251693725585937e-05,
      "step": 55068,
      "training_step_time": 8.376535654067993
    },
    {
      "epoch": 5.2517890930175784e-05,
      "model_forward_time": 2.3069002628326416,
      "step": 55069
    },
    {
      "epoch": 5.2517890930175784e-05,
      "step": 55069,
      "training_step_time": 8.369736194610596
    },
    {
      "epoch": 5.251884460449219e-05,
      "grad_norm": 0.08420775085687637,
      "learning_rate": 4.5757343829679914e-06,
      "loss": 0.0579,
      "step": 55070
    },
    {
      "epoch": 5.251884460449219e-05,
      "model_forward_time": 1.712719202041626,
      "step": 55070
    },
    {
      "dynamics_loss_avg": 0.07347831688821316,
      "epoch": 5.251884460449219e-05,
      "step": 55070
    },
    {
      "action_loss_avg": 0.0021389161818660797,
      "epoch": 5.251884460449219e-05,
      "step": 55070
    },
    {
      "epoch": 5.251884460449219e-05,
      "step": 55070,
      "training_step_time": 7.795433521270752
    },
    {
      "epoch": 5.251979827880859e-05,
      "model_forward_time": 1.7206792831420898,
      "step": 55071
    },
    {
      "epoch": 5.251979827880859e-05,
      "step": 55071,
      "training_step_time": 6.835359334945679
    },
    {
      "epoch": 5.2520751953125e-05,
      "model_forward_time": 1.7064838409423828,
      "step": 55072
    },
    {
      "epoch": 5.2520751953125e-05,
      "step": 55072,
      "training_step_time": 6.787533760070801
    },
    {
      "epoch": 5.252170562744141e-05,
      "model_forward_time": 1.719316005706787,
      "step": 55073
    },
    {
      "epoch": 5.252170562744141e-05,
      "step": 55073,
      "training_step_time": 6.811316013336182
    },
    {
      "epoch": 5.252265930175781e-05,
      "model_forward_time": 1.7245707511901855,
      "step": 55074
    },
    {
      "epoch": 5.252265930175781e-05,
      "step": 55074,
      "training_step_time": 6.692535638809204
    },
    {
      "epoch": 5.252361297607422e-05,
      "model_forward_time": 1.346343755722046,
      "step": 55075
    },
    {
      "epoch": 5.252361297607422e-05,
      "step": 55075,
      "training_step_time": 6.700601577758789
    },
    {
      "epoch": 5.2524566650390625e-05,
      "model_forward_time": 1.7285335063934326,
      "step": 55076
    },
    {
      "epoch": 5.2524566650390625e-05,
      "step": 55076,
      "training_step_time": 6.797502756118774
    },
    {
      "epoch": 5.252552032470703e-05,
      "model_forward_time": 1.723931074142456,
      "step": 55077
    },
    {
      "epoch": 5.252552032470703e-05,
      "step": 55077,
      "training_step_time": 9.661517858505249
    },
    {
      "epoch": 5.252647399902344e-05,
      "model_forward_time": 1.7537481784820557,
      "step": 55078
    },
    {
      "epoch": 5.252647399902344e-05,
      "step": 55078,
      "training_step_time": 9.872753143310547
    },
    {
      "epoch": 5.2527427673339844e-05,
      "model_forward_time": 1.716860294342041,
      "step": 55079
    },
    {
      "epoch": 5.2527427673339844e-05,
      "step": 55079,
      "training_step_time": 7.666082859039307
    },
    {
      "epoch": 5.252838134765625e-05,
      "grad_norm": 0.08585051447153091,
      "learning_rate": 4.574086899667044e-06,
      "loss": 0.0614,
      "step": 55080
    },
    {
      "epoch": 5.252838134765625e-05,
      "model_forward_time": 1.7098166942596436,
      "step": 55080
    },
    {
      "dynamics_loss_avg": 0.0692112198099494,
      "epoch": 5.252838134765625e-05,
      "step": 55080
    },
    {
      "action_loss_avg": 0.0015184354153461755,
      "epoch": 5.252838134765625e-05,
      "step": 55080
    },
    {
      "epoch": 5.252838134765625e-05,
      "step": 55080,
      "training_step_time": 6.7542405128479
    },
    {
      "epoch": 5.252933502197266e-05,
      "model_forward_time": 1.7308540344238281,
      "step": 55081
    },
    {
      "epoch": 5.252933502197266e-05,
      "step": 55081,
      "training_step_time": 6.796428680419922
    },
    {
      "epoch": 5.253028869628906e-05,
      "model_forward_time": 1.711113452911377,
      "step": 55082
    },
    {
      "epoch": 5.253028869628906e-05,
      "step": 55082,
      "training_step_time": 6.78058123588562
    },
    {
      "epoch": 5.253124237060547e-05,
      "model_forward_time": 1.7274670600891113,
      "step": 55083
    },
    {
      "epoch": 5.253124237060547e-05,
      "step": 55083,
      "training_step_time": 6.777288913726807
    },
    {
      "epoch": 5.253219604492188e-05,
      "model_forward_time": 1.7180211544036865,
      "step": 55084
    },
    {
      "epoch": 5.253219604492188e-05,
      "step": 55084,
      "training_step_time": 6.80632758140564
    },
    {
      "epoch": 5.253314971923828e-05,
      "model_forward_time": 1.726527214050293,
      "step": 55085
    },
    {
      "epoch": 5.253314971923828e-05,
      "step": 55085,
      "training_step_time": 6.782161712646484
    },
    {
      "epoch": 5.2534103393554686e-05,
      "model_forward_time": 1.7215065956115723,
      "step": 55086
    },
    {
      "epoch": 5.2534103393554686e-05,
      "step": 55086,
      "training_step_time": 7.700442790985107
    },
    {
      "epoch": 5.25350570678711e-05,
      "model_forward_time": 1.7265031337738037,
      "step": 55087
    },
    {
      "epoch": 5.25350570678711e-05,
      "step": 55087,
      "training_step_time": 8.301734447479248
    },
    {
      "epoch": 5.25360107421875e-05,
      "model_forward_time": 2.2659525871276855,
      "step": 55088
    },
    {
      "epoch": 5.25360107421875e-05,
      "step": 55088,
      "training_step_time": 7.612064838409424
    },
    {
      "epoch": 5.2536964416503905e-05,
      "model_forward_time": 1.7264657020568848,
      "step": 55089
    },
    {
      "epoch": 5.2536964416503905e-05,
      "step": 55089,
      "training_step_time": 7.2517335414886475
    },
    {
      "epoch": 5.2537918090820316e-05,
      "grad_norm": 0.10710569471120834,
      "learning_rate": 4.572439462943311e-06,
      "loss": 0.0543,
      "step": 55090
    },
    {
      "epoch": 5.2537918090820316e-05,
      "model_forward_time": 1.7105982303619385,
      "step": 55090
    },
    {
      "dynamics_loss_avg": 0.054426942579448225,
      "epoch": 5.2537918090820316e-05,
      "step": 55090
    },
    {
      "action_loss_avg": 0.001955208566505462,
      "epoch": 5.2537918090820316e-05,
      "step": 55090
    },
    {
      "epoch": 5.2537918090820316e-05,
      "step": 55090,
      "training_step_time": 6.658801078796387
    },
    {
      "epoch": 5.253887176513672e-05,
      "model_forward_time": 1.7245352268218994,
      "step": 55091
    },
    {
      "epoch": 5.253887176513672e-05,
      "step": 55091,
      "training_step_time": 6.734612226486206
    },
    {
      "epoch": 5.2539825439453124e-05,
      "model_forward_time": 1.724379062652588,
      "step": 55092
    },
    {
      "epoch": 5.2539825439453124e-05,
      "step": 55092,
      "training_step_time": 6.68237566947937
    },
    {
      "epoch": 5.2540779113769535e-05,
      "model_forward_time": 1.7242252826690674,
      "step": 55093
    },
    {
      "epoch": 5.2540779113769535e-05,
      "step": 55093,
      "training_step_time": 6.695735454559326
    },
    {
      "epoch": 5.254173278808594e-05,
      "model_forward_time": 1.7183952331542969,
      "step": 55094
    },
    {
      "epoch": 5.254173278808594e-05,
      "step": 55094,
      "training_step_time": 6.703717947006226
    },
    {
      "epoch": 5.254268646240234e-05,
      "model_forward_time": 1.715336561203003,
      "step": 55095
    },
    {
      "epoch": 5.254268646240234e-05,
      "step": 55095,
      "training_step_time": 7.633890151977539
    },
    {
      "epoch": 5.254364013671875e-05,
      "model_forward_time": 1.7347066402435303,
      "step": 55096
    },
    {
      "epoch": 5.254364013671875e-05,
      "step": 55096,
      "training_step_time": 9.610692262649536
    },
    {
      "epoch": 5.254459381103516e-05,
      "model_forward_time": 2.2715494632720947,
      "step": 55097
    },
    {
      "epoch": 5.254459381103516e-05,
      "step": 55097,
      "training_step_time": 9.428964853286743
    },
    {
      "epoch": 5.254554748535156e-05,
      "model_forward_time": 1.7266967296600342,
      "step": 55098
    },
    {
      "epoch": 5.254554748535156e-05,
      "step": 55098,
      "training_step_time": 7.2858827114105225
    },
    {
      "epoch": 5.2546501159667966e-05,
      "model_forward_time": 1.7039179801940918,
      "step": 55099
    },
    {
      "epoch": 5.2546501159667966e-05,
      "step": 55099,
      "training_step_time": 6.7178428173065186
    },
    {
      "epoch": 5.2547454833984376e-05,
      "grad_norm": 0.1193331852555275,
      "learning_rate": 4.570792072976958e-06,
      "loss": 0.059,
      "step": 55100
    },
    {
      "epoch": 5.2547454833984376e-05,
      "model_forward_time": 1.7096476554870605,
      "step": 55100
    },
    {
      "dynamics_loss_avg": 0.06056381352245808,
      "epoch": 5.2547454833984376e-05,
      "step": 55100
    },
    {
      "action_loss_avg": 0.001921299030072987,
      "epoch": 5.2547454833984376e-05,
      "step": 55100
    },
    {
      "epoch": 5.2547454833984376e-05,
      "step": 55100,
      "training_step_time": 6.717033386230469
    },
    {
      "epoch": 5.254840850830078e-05,
      "model_forward_time": 1.7259521484375,
      "step": 55101
    },
    {
      "epoch": 5.254840850830078e-05,
      "step": 55101,
      "training_step_time": 6.67065691947937
    },
    {
      "epoch": 5.2549362182617184e-05,
      "model_forward_time": 1.7270162105560303,
      "step": 55102
    },
    {
      "epoch": 5.2549362182617184e-05,
      "step": 55102,
      "training_step_time": 6.698563814163208
    },
    {
      "epoch": 5.2550315856933595e-05,
      "model_forward_time": 1.7171943187713623,
      "step": 55103
    },
    {
      "epoch": 5.2550315856933595e-05,
      "step": 55103,
      "training_step_time": 6.920606851577759
    },
    {
      "epoch": 5.255126953125e-05,
      "model_forward_time": 1.7291948795318604,
      "step": 55104
    },
    {
      "epoch": 5.255126953125e-05,
      "step": 55104,
      "training_step_time": 7.8748695850372314
    },
    {
      "epoch": 5.25522232055664e-05,
      "model_forward_time": 1.7145776748657227,
      "step": 55105
    },
    {
      "epoch": 5.25522232055664e-05,
      "step": 55105,
      "training_step_time": 10.427621364593506
    },
    {
      "epoch": 5.2553176879882814e-05,
      "model_forward_time": 2.301382303237915,
      "step": 55106
    },
    {
      "epoch": 5.2553176879882814e-05,
      "step": 55106,
      "training_step_time": 7.724693059921265
    },
    {
      "epoch": 5.255413055419922e-05,
      "model_forward_time": 1.716820240020752,
      "step": 55107
    },
    {
      "epoch": 5.255413055419922e-05,
      "step": 55107,
      "training_step_time": 7.678112745285034
    },
    {
      "epoch": 5.255508422851562e-05,
      "model_forward_time": 1.7151224613189697,
      "step": 55108
    },
    {
      "epoch": 5.255508422851562e-05,
      "step": 55108,
      "training_step_time": 6.730952501296997
    },
    {
      "epoch": 5.255603790283203e-05,
      "model_forward_time": 1.7252507209777832,
      "step": 55109
    },
    {
      "epoch": 5.255603790283203e-05,
      "step": 55109,
      "training_step_time": 6.69559383392334
    },
    {
      "epoch": 5.255699157714844e-05,
      "grad_norm": 0.11548689752817154,
      "learning_rate": 4.569144729948138e-06,
      "loss": 0.0572,
      "step": 55110
    },
    {
      "epoch": 5.255699157714844e-05,
      "model_forward_time": 1.7042272090911865,
      "step": 55110
    },
    {
      "dynamics_loss_avg": 0.052372615039348605,
      "epoch": 5.255699157714844e-05,
      "step": 55110
    },
    {
      "action_loss_avg": 0.001619792776182294,
      "epoch": 5.255699157714844e-05,
      "step": 55110
    },
    {
      "epoch": 5.255699157714844e-05,
      "step": 55110,
      "training_step_time": 6.712561368942261
    },
    {
      "epoch": 5.255794525146484e-05,
      "model_forward_time": 1.7105131149291992,
      "step": 55111
    },
    {
      "epoch": 5.255794525146484e-05,
      "step": 55111,
      "training_step_time": 6.710742235183716
    },
    {
      "epoch": 5.255889892578125e-05,
      "model_forward_time": 1.719059944152832,
      "step": 55112
    },
    {
      "epoch": 5.255889892578125e-05,
      "step": 55112,
      "training_step_time": 6.6689770221710205
    },
    {
      "epoch": 5.2559852600097656e-05,
      "model_forward_time": 1.7186808586120605,
      "step": 55113
    },
    {
      "epoch": 5.2559852600097656e-05,
      "step": 55113,
      "training_step_time": 7.662103176116943
    },
    {
      "epoch": 5.256080627441406e-05,
      "model_forward_time": 1.7034118175506592,
      "step": 55114
    },
    {
      "epoch": 5.256080627441406e-05,
      "step": 55114,
      "training_step_time": 10.306121826171875
    },
    {
      "epoch": 5.256175994873047e-05,
      "model_forward_time": 2.3013548851013184,
      "step": 55115
    },
    {
      "epoch": 5.256175994873047e-05,
      "step": 55115,
      "training_step_time": 8.748368978500366
    },
    {
      "epoch": 5.2562713623046875e-05,
      "model_forward_time": 1.710235834121704,
      "step": 55116
    },
    {
      "epoch": 5.2562713623046875e-05,
      "step": 55116,
      "training_step_time": 7.69209098815918
    },
    {
      "epoch": 5.256366729736328e-05,
      "model_forward_time": 1.7093522548675537,
      "step": 55117
    },
    {
      "epoch": 5.256366729736328e-05,
      "step": 55117,
      "training_step_time": 7.200375556945801
    },
    {
      "epoch": 5.256462097167969e-05,
      "model_forward_time": 1.7239959239959717,
      "step": 55118
    },
    {
      "epoch": 5.256462097167969e-05,
      "step": 55118,
      "training_step_time": 6.7107765674591064
    },
    {
      "epoch": 5.2565574645996094e-05,
      "model_forward_time": 1.7028656005859375,
      "step": 55119
    },
    {
      "epoch": 5.2565574645996094e-05,
      "step": 55119,
      "training_step_time": 6.724228858947754
    },
    {
      "epoch": 5.25665283203125e-05,
      "grad_norm": 0.10630525648593903,
      "learning_rate": 4.567497434037004e-06,
      "loss": 0.0557,
      "step": 55120
    },
    {
      "epoch": 5.25665283203125e-05,
      "model_forward_time": 1.7067523002624512,
      "step": 55120
    },
    {
      "dynamics_loss_avg": 0.056839559227228165,
      "epoch": 5.25665283203125e-05,
      "step": 55120
    },
    {
      "action_loss_avg": 0.0012413726304657756,
      "epoch": 5.25665283203125e-05,
      "step": 55120
    },
    {
      "epoch": 5.25665283203125e-05,
      "step": 55120,
      "training_step_time": 6.715318441390991
    },
    {
      "epoch": 5.256748199462891e-05,
      "model_forward_time": 1.7191755771636963,
      "step": 55121
    },
    {
      "epoch": 5.256748199462891e-05,
      "step": 55121,
      "training_step_time": 6.734838485717773
    },
    {
      "epoch": 5.256843566894531e-05,
      "model_forward_time": 1.7141716480255127,
      "step": 55122
    },
    {
      "epoch": 5.256843566894531e-05,
      "step": 55122,
      "training_step_time": 7.590830326080322
    },
    {
      "epoch": 5.2569389343261716e-05,
      "model_forward_time": 1.726017713546753,
      "step": 55123
    },
    {
      "epoch": 5.2569389343261716e-05,
      "step": 55123,
      "training_step_time": 8.98470401763916
    },
    {
      "epoch": 5.257034301757813e-05,
      "model_forward_time": 2.2812559604644775,
      "step": 55124
    },
    {
      "epoch": 5.257034301757813e-05,
      "step": 55124,
      "training_step_time": 8.83490777015686
    },
    {
      "epoch": 5.257129669189453e-05,
      "model_forward_time": 1.7200260162353516,
      "step": 55125
    },
    {
      "epoch": 5.257129669189453e-05,
      "step": 55125,
      "training_step_time": 8.554377317428589
    },
    {
      "epoch": 5.2572250366210935e-05,
      "model_forward_time": 1.7194690704345703,
      "step": 55126
    },
    {
      "epoch": 5.2572250366210935e-05,
      "step": 55126,
      "training_step_time": 7.5179431438446045
    },
    {
      "epoch": 5.2573204040527346e-05,
      "model_forward_time": 1.7118070125579834,
      "step": 55127
    },
    {
      "epoch": 5.2573204040527346e-05,
      "step": 55127,
      "training_step_time": 6.75911545753479
    },
    {
      "epoch": 5.257415771484375e-05,
      "model_forward_time": 1.7232882976531982,
      "step": 55128
    },
    {
      "epoch": 5.257415771484375e-05,
      "step": 55128,
      "training_step_time": 6.704923391342163
    },
    {
      "epoch": 5.2575111389160154e-05,
      "model_forward_time": 1.720047950744629,
      "step": 55129
    },
    {
      "epoch": 5.2575111389160154e-05,
      "step": 55129,
      "training_step_time": 6.748448371887207
    },
    {
      "epoch": 5.2576065063476565e-05,
      "grad_norm": 0.08288919180631638,
      "learning_rate": 4.565850185423699e-06,
      "loss": 0.0611,
      "step": 55130
    },
    {
      "epoch": 5.2576065063476565e-05,
      "model_forward_time": 1.7105910778045654,
      "step": 55130
    },
    {
      "dynamics_loss_avg": 0.06292641162872314,
      "epoch": 5.2576065063476565e-05,
      "step": 55130
    },
    {
      "action_loss_avg": 0.0037464338005520404,
      "epoch": 5.2576065063476565e-05,
      "step": 55130
    },
    {
      "epoch": 5.2576065063476565e-05,
      "step": 55130,
      "training_step_time": 6.730932712554932
    },
    {
      "epoch": 5.257701873779297e-05,
      "model_forward_time": 1.7201368808746338,
      "step": 55131
    },
    {
      "epoch": 5.257701873779297e-05,
      "step": 55131,
      "training_step_time": 7.360388517379761
    },
    {
      "epoch": 5.257797241210937e-05,
      "model_forward_time": 1.7223358154296875,
      "step": 55132
    },
    {
      "epoch": 5.257797241210937e-05,
      "step": 55132,
      "training_step_time": 8.840979099273682
    },
    {
      "epoch": 5.2578926086425784e-05,
      "model_forward_time": 1.7221720218658447,
      "step": 55133
    },
    {
      "epoch": 5.2578926086425784e-05,
      "step": 55133,
      "training_step_time": 10.18089771270752
    },
    {
      "epoch": 5.257987976074219e-05,
      "model_forward_time": 1.7699029445648193,
      "step": 55134
    },
    {
      "epoch": 5.257987976074219e-05,
      "step": 55134,
      "training_step_time": 7.6830809116363525
    },
    {
      "epoch": 5.258083343505859e-05,
      "model_forward_time": 1.7177422046661377,
      "step": 55135
    },
    {
      "epoch": 5.258083343505859e-05,
      "step": 55135,
      "training_step_time": 7.658661842346191
    },
    {
      "epoch": 5.2581787109375e-05,
      "model_forward_time": 1.727457046508789,
      "step": 55136
    },
    {
      "epoch": 5.2581787109375e-05,
      "step": 55136,
      "training_step_time": 6.695950984954834
    },
    {
      "epoch": 5.258274078369141e-05,
      "model_forward_time": 1.7216439247131348,
      "step": 55137
    },
    {
      "epoch": 5.258274078369141e-05,
      "step": 55137,
      "training_step_time": 6.703208923339844
    },
    {
      "epoch": 5.258369445800781e-05,
      "model_forward_time": 1.7419898509979248,
      "step": 55138
    },
    {
      "epoch": 5.258369445800781e-05,
      "step": 55138,
      "training_step_time": 6.694664001464844
    },
    {
      "epoch": 5.258464813232422e-05,
      "model_forward_time": 1.7233898639678955,
      "step": 55139
    },
    {
      "epoch": 5.258464813232422e-05,
      "step": 55139,
      "training_step_time": 6.7211830615997314
    },
    {
      "epoch": 5.2585601806640626e-05,
      "grad_norm": 0.09741774201393127,
      "learning_rate": 4.564202984288365e-06,
      "loss": 0.0573,
      "step": 55140
    },
    {
      "epoch": 5.2585601806640626e-05,
      "model_forward_time": 1.717597484588623,
      "step": 55140
    },
    {
      "dynamics_loss_avg": 0.04795999228954315,
      "epoch": 5.2585601806640626e-05,
      "step": 55140
    },
    {
      "action_loss_avg": 0.0012153490795753896,
      "epoch": 5.2585601806640626e-05,
      "step": 55140
    },
    {
      "epoch": 5.2585601806640626e-05,
      "step": 55140,
      "training_step_time": 7.2908124923706055
    },
    {
      "epoch": 5.258655548095703e-05,
      "model_forward_time": 1.3444373607635498,
      "step": 55141
    },
    {
      "epoch": 5.258655548095703e-05,
      "step": 55141,
      "training_step_time": 8.311480045318604
    },
    {
      "epoch": 5.258750915527344e-05,
      "model_forward_time": 1.730600118637085,
      "step": 55142
    },
    {
      "epoch": 5.258750915527344e-05,
      "step": 55142,
      "training_step_time": 7.791540145874023
    },
    {
      "epoch": 5.2588462829589844e-05,
      "model_forward_time": 2.2719271183013916,
      "step": 55143
    },
    {
      "epoch": 5.2588462829589844e-05,
      "step": 55143,
      "training_step_time": 7.679246425628662
    },
    {
      "epoch": 5.258941650390625e-05,
      "model_forward_time": 1.7208023071289062,
      "step": 55144
    },
    {
      "epoch": 5.258941650390625e-05,
      "step": 55144,
      "training_step_time": 7.623956203460693
    },
    {
      "epoch": 5.259037017822266e-05,
      "model_forward_time": 1.7241461277008057,
      "step": 55145
    },
    {
      "epoch": 5.259037017822266e-05,
      "step": 55145,
      "training_step_time": 7.1565046310424805
    },
    {
      "epoch": 5.259132385253906e-05,
      "model_forward_time": 1.7125959396362305,
      "step": 55146
    },
    {
      "epoch": 5.259132385253906e-05,
      "step": 55146,
      "training_step_time": 6.679314374923706
    },
    {
      "epoch": 5.259227752685547e-05,
      "model_forward_time": 1.701282262802124,
      "step": 55147
    },
    {
      "epoch": 5.259227752685547e-05,
      "step": 55147,
      "training_step_time": 6.790029525756836
    },
    {
      "epoch": 5.259323120117188e-05,
      "model_forward_time": 1.7160592079162598,
      "step": 55148
    },
    {
      "epoch": 5.259323120117188e-05,
      "step": 55148,
      "training_step_time": 6.760549783706665
    },
    {
      "epoch": 5.259418487548828e-05,
      "model_forward_time": 1.7167878150939941,
      "step": 55149
    },
    {
      "epoch": 5.259418487548828e-05,
      "step": 55149,
      "training_step_time": 7.440211772918701
    },
    {
      "epoch": 5.2595138549804686e-05,
      "grad_norm": 0.09173346310853958,
      "learning_rate": 4.562555830811136e-06,
      "loss": 0.0584,
      "step": 55150
    },
    {
      "epoch": 5.2595138549804686e-05,
      "model_forward_time": 1.706885576248169,
      "step": 55150
    },
    {
      "dynamics_loss_avg": 0.04667705642059446,
      "epoch": 5.2595138549804686e-05,
      "step": 55150
    },
    {
      "action_loss_avg": 0.0014468832407146693,
      "epoch": 5.2595138549804686e-05,
      "step": 55150
    },
    {
      "epoch": 5.2595138549804686e-05,
      "step": 55150,
      "training_step_time": 7.545498847961426
    },
    {
      "epoch": 5.25960922241211e-05,
      "model_forward_time": 1.7223336696624756,
      "step": 55151
    },
    {
      "epoch": 5.25960922241211e-05,
      "step": 55151,
      "training_step_time": 9.511759042739868
    },
    {
      "epoch": 5.25970458984375e-05,
      "model_forward_time": 2.326326608657837,
      "step": 55152
    },
    {
      "epoch": 5.25970458984375e-05,
      "step": 55152,
      "training_step_time": 8.799293994903564
    },
    {
      "epoch": 5.2597999572753905e-05,
      "model_forward_time": 1.711092233657837,
      "step": 55153
    },
    {
      "epoch": 5.2597999572753905e-05,
      "step": 55153,
      "training_step_time": 7.75188684463501
    },
    {
      "epoch": 5.2598953247070316e-05,
      "model_forward_time": 1.719886302947998,
      "step": 55154
    },
    {
      "epoch": 5.2598953247070316e-05,
      "step": 55154,
      "training_step_time": 7.365635395050049
    },
    {
      "epoch": 5.259990692138672e-05,
      "model_forward_time": 1.7274229526519775,
      "step": 55155
    },
    {
      "epoch": 5.259990692138672e-05,
      "step": 55155,
      "training_step_time": 6.756353378295898
    },
    {
      "epoch": 5.2600860595703124e-05,
      "model_forward_time": 1.716545581817627,
      "step": 55156
    },
    {
      "epoch": 5.2600860595703124e-05,
      "step": 55156,
      "training_step_time": 6.72929048538208
    },
    {
      "epoch": 5.260181427001953e-05,
      "model_forward_time": 1.7098021507263184,
      "step": 55157
    },
    {
      "epoch": 5.260181427001953e-05,
      "step": 55157,
      "training_step_time": 6.703930377960205
    },
    {
      "epoch": 5.260276794433594e-05,
      "model_forward_time": 1.7178471088409424,
      "step": 55158
    },
    {
      "epoch": 5.260276794433594e-05,
      "step": 55158,
      "training_step_time": 7.3940911293029785
    },
    {
      "epoch": 5.260372161865234e-05,
      "model_forward_time": 1.7173776626586914,
      "step": 55159
    },
    {
      "epoch": 5.260372161865234e-05,
      "step": 55159,
      "training_step_time": 7.502532958984375
    },
    {
      "epoch": 5.260467529296875e-05,
      "grad_norm": 0.19027629494667053,
      "learning_rate": 4.560908725172145e-06,
      "loss": 0.0585,
      "step": 55160
    },
    {
      "epoch": 5.260467529296875e-05,
      "model_forward_time": 1.7109344005584717,
      "step": 55160
    },
    {
      "dynamics_loss_avg": 0.04109015474095941,
      "epoch": 5.260467529296875e-05,
      "step": 55160
    },
    {
      "action_loss_avg": 0.0015297767327865585,
      "epoch": 5.260467529296875e-05,
      "step": 55160
    },
    {
      "epoch": 5.260467529296875e-05,
      "step": 55160,
      "training_step_time": 9.179490566253662
    },
    {
      "epoch": 5.260562896728516e-05,
      "model_forward_time": 2.286634922027588,
      "step": 55161
    },
    {
      "epoch": 5.260562896728516e-05,
      "step": 55161,
      "training_step_time": 7.7866528034210205
    },
    {
      "epoch": 5.260658264160156e-05,
      "model_forward_time": 1.7520253658294678,
      "step": 55162
    },
    {
      "epoch": 5.260658264160156e-05,
      "step": 55162,
      "training_step_time": 7.784083604812622
    },
    {
      "epoch": 5.2607536315917966e-05,
      "model_forward_time": 1.71820068359375,
      "step": 55163
    },
    {
      "epoch": 5.2607536315917966e-05,
      "step": 55163,
      "training_step_time": 7.527737140655518
    },
    {
      "epoch": 5.2608489990234376e-05,
      "model_forward_time": 1.7008605003356934,
      "step": 55164
    },
    {
      "epoch": 5.2608489990234376e-05,
      "step": 55164,
      "training_step_time": 6.766344785690308
    },
    {
      "epoch": 5.260944366455078e-05,
      "model_forward_time": 1.717966079711914,
      "step": 55165
    },
    {
      "epoch": 5.260944366455078e-05,
      "step": 55165,
      "training_step_time": 6.766647577285767
    },
    {
      "epoch": 5.2610397338867184e-05,
      "model_forward_time": 1.7165286540985107,
      "step": 55166
    },
    {
      "epoch": 5.2610397338867184e-05,
      "step": 55166,
      "training_step_time": 6.738471508026123
    },
    {
      "epoch": 5.2611351013183595e-05,
      "model_forward_time": 1.7377843856811523,
      "step": 55167
    },
    {
      "epoch": 5.2611351013183595e-05,
      "step": 55167,
      "training_step_time": 7.382697820663452
    },
    {
      "epoch": 5.26123046875e-05,
      "model_forward_time": 1.3518104553222656,
      "step": 55168
    },
    {
      "epoch": 5.26123046875e-05,
      "step": 55168,
      "training_step_time": 7.542841911315918
    },
    {
      "epoch": 5.26132583618164e-05,
      "model_forward_time": 1.713611364364624,
      "step": 55169
    },
    {
      "epoch": 5.26132583618164e-05,
      "step": 55169,
      "training_step_time": 9.681839942932129
    },
    {
      "epoch": 5.2614212036132814e-05,
      "grad_norm": 0.11763349175453186,
      "learning_rate": 4.559261667551516e-06,
      "loss": 0.0552,
      "step": 55170
    },
    {
      "epoch": 5.2614212036132814e-05,
      "model_forward_time": 1.711874008178711,
      "step": 55170
    },
    {
      "dynamics_loss_avg": 0.044796106219291684,
      "epoch": 5.2614212036132814e-05,
      "step": 55170
    },
    {
      "action_loss_avg": 0.0011898856522748248,
      "epoch": 5.2614212036132814e-05,
      "step": 55170
    },
    {
      "epoch": 5.2614212036132814e-05,
      "step": 55170,
      "training_step_time": 8.777329444885254
    },
    {
      "epoch": 5.261516571044922e-05,
      "model_forward_time": 1.7380282878875732,
      "step": 55171
    },
    {
      "epoch": 5.261516571044922e-05,
      "step": 55171,
      "training_step_time": 8.252060651779175
    },
    {
      "epoch": 5.261611938476562e-05,
      "model_forward_time": 1.6986804008483887,
      "step": 55172
    },
    {
      "epoch": 5.261611938476562e-05,
      "step": 55172,
      "training_step_time": 7.398162126541138
    },
    {
      "epoch": 5.261707305908203e-05,
      "model_forward_time": 1.7207067012786865,
      "step": 55173
    },
    {
      "epoch": 5.261707305908203e-05,
      "step": 55173,
      "training_step_time": 7.255327939987183
    },
    {
      "epoch": 5.261802673339844e-05,
      "model_forward_time": 1.7248365879058838,
      "step": 55174
    },
    {
      "epoch": 5.261802673339844e-05,
      "step": 55174,
      "training_step_time": 6.745970726013184
    },
    {
      "epoch": 5.261898040771484e-05,
      "model_forward_time": 1.7319743633270264,
      "step": 55175
    },
    {
      "epoch": 5.261898040771484e-05,
      "step": 55175,
      "training_step_time": 6.712655305862427
    },
    {
      "epoch": 5.261993408203125e-05,
      "model_forward_time": 1.7276661396026611,
      "step": 55176
    },
    {
      "epoch": 5.261993408203125e-05,
      "step": 55176,
      "training_step_time": 6.9294257164001465
    },
    {
      "epoch": 5.2620887756347656e-05,
      "model_forward_time": 1.723290205001831,
      "step": 55177
    },
    {
      "epoch": 5.2620887756347656e-05,
      "step": 55177,
      "training_step_time": 8.243293046951294
    },
    {
      "epoch": 5.262184143066406e-05,
      "model_forward_time": 1.725600004196167,
      "step": 55178
    },
    {
      "epoch": 5.262184143066406e-05,
      "step": 55178,
      "training_step_time": 8.51789116859436
    },
    {
      "epoch": 5.262279510498047e-05,
      "model_forward_time": 1.7215256690979004,
      "step": 55179
    },
    {
      "epoch": 5.262279510498047e-05,
      "step": 55179,
      "training_step_time": 9.595344305038452
    },
    {
      "epoch": 5.2623748779296875e-05,
      "grad_norm": 0.11435772478580475,
      "learning_rate": 4.557614658129369e-06,
      "loss": 0.0554,
      "step": 55180
    },
    {
      "epoch": 5.2623748779296875e-05,
      "model_forward_time": 2.2531394958496094,
      "step": 55180
    },
    {
      "dynamics_loss_avg": 0.0707111980766058,
      "epoch": 5.2623748779296875e-05,
      "step": 55180
    },
    {
      "action_loss_avg": 0.0020013685687445106,
      "epoch": 5.2623748779296875e-05,
      "step": 55180
    },
    {
      "epoch": 5.2623748779296875e-05,
      "step": 55180,
      "training_step_time": 9.369405746459961
    },
    {
      "epoch": 5.262470245361328e-05,
      "model_forward_time": 1.723618507385254,
      "step": 55181
    },
    {
      "epoch": 5.262470245361328e-05,
      "step": 55181,
      "training_step_time": 7.518935441970825
    },
    {
      "epoch": 5.262565612792969e-05,
      "model_forward_time": 1.345458745956421,
      "step": 55182
    },
    {
      "epoch": 5.262565612792969e-05,
      "step": 55182,
      "training_step_time": 7.2753379344940186
    },
    {
      "epoch": 5.2626609802246094e-05,
      "model_forward_time": 1.7307782173156738,
      "step": 55183
    },
    {
      "epoch": 5.2626609802246094e-05,
      "step": 55183,
      "training_step_time": 6.738940238952637
    },
    {
      "epoch": 5.26275634765625e-05,
      "model_forward_time": 1.7232859134674072,
      "step": 55184
    },
    {
      "epoch": 5.26275634765625e-05,
      "step": 55184,
      "training_step_time": 6.720597267150879
    },
    {
      "epoch": 5.262851715087891e-05,
      "model_forward_time": 1.716493844985962,
      "step": 55185
    },
    {
      "epoch": 5.262851715087891e-05,
      "step": 55185,
      "training_step_time": 6.732653379440308
    },
    {
      "epoch": 5.262947082519531e-05,
      "model_forward_time": 1.7119619846343994,
      "step": 55186
    },
    {
      "epoch": 5.262947082519531e-05,
      "step": 55186,
      "training_step_time": 7.338915109634399
    },
    {
      "epoch": 5.2630424499511716e-05,
      "model_forward_time": 1.725813865661621,
      "step": 55187
    },
    {
      "epoch": 5.2630424499511716e-05,
      "step": 55187,
      "training_step_time": 7.707555294036865
    },
    {
      "epoch": 5.263137817382813e-05,
      "model_forward_time": 1.7278261184692383,
      "step": 55188
    },
    {
      "epoch": 5.263137817382813e-05,
      "step": 55188,
      "training_step_time": 10.41091275215149
    },
    {
      "epoch": 5.263233184814453e-05,
      "model_forward_time": 2.2969658374786377,
      "step": 55189
    },
    {
      "epoch": 5.263233184814453e-05,
      "step": 55189,
      "training_step_time": 9.553105115890503
    },
    {
      "epoch": 5.2633285522460935e-05,
      "grad_norm": 0.07591819763183594,
      "learning_rate": 4.555967697085817e-06,
      "loss": 0.0604,
      "step": 55190
    },
    {
      "epoch": 5.2633285522460935e-05,
      "model_forward_time": 1.7573983669281006,
      "step": 55190
    },
    {
      "dynamics_loss_avg": 0.06018587350845337,
      "epoch": 5.2633285522460935e-05,
      "step": 55190
    },
    {
      "action_loss_avg": 0.001130746805574745,
      "epoch": 5.2633285522460935e-05,
      "step": 55190
    },
    {
      "epoch": 5.2633285522460935e-05,
      "step": 55190,
      "training_step_time": 7.490130662918091
    },
    {
      "epoch": 5.2634239196777346e-05,
      "model_forward_time": 1.7092092037200928,
      "step": 55191
    },
    {
      "epoch": 5.2634239196777346e-05,
      "step": 55191,
      "training_step_time": 7.3116774559021
    },
    {
      "epoch": 5.263519287109375e-05,
      "model_forward_time": 1.7269647121429443,
      "step": 55192
    },
    {
      "epoch": 5.263519287109375e-05,
      "step": 55192,
      "training_step_time": 6.743621349334717
    },
    {
      "epoch": 5.2636146545410154e-05,
      "model_forward_time": 1.731290340423584,
      "step": 55193
    },
    {
      "epoch": 5.2636146545410154e-05,
      "step": 55193,
      "training_step_time": 6.76298975944519
    },
    {
      "epoch": 5.2637100219726565e-05,
      "model_forward_time": 1.7275705337524414,
      "step": 55194
    },
    {
      "epoch": 5.2637100219726565e-05,
      "step": 55194,
      "training_step_time": 6.75088095664978
    },
    {
      "epoch": 5.263805389404297e-05,
      "model_forward_time": 1.7272701263427734,
      "step": 55195
    },
    {
      "epoch": 5.263805389404297e-05,
      "step": 55195,
      "training_step_time": 7.559862375259399
    },
    {
      "epoch": 5.263900756835937e-05,
      "model_forward_time": 1.727492094039917,
      "step": 55196
    },
    {
      "epoch": 5.263900756835937e-05,
      "step": 55196,
      "training_step_time": 8.69403886795044
    },
    {
      "epoch": 5.2639961242675784e-05,
      "model_forward_time": 1.727039098739624,
      "step": 55197
    },
    {
      "epoch": 5.2639961242675784e-05,
      "step": 55197,
      "training_step_time": 7.861189842224121
    },
    {
      "epoch": 5.264091491699219e-05,
      "model_forward_time": 2.2930662631988525,
      "step": 55198
    },
    {
      "epoch": 5.264091491699219e-05,
      "step": 55198,
      "training_step_time": 8.789481163024902
    },
    {
      "epoch": 5.264186859130859e-05,
      "model_forward_time": 1.7679312229156494,
      "step": 55199
    },
    {
      "epoch": 5.264186859130859e-05,
      "step": 55199,
      "training_step_time": 7.609435558319092
    },
    {
      "epoch": 5.2642822265625e-05,
      "grad_norm": 0.1088259294629097,
      "learning_rate": 4.5543207846009705e-06,
      "loss": 0.0623,
      "step": 55200
    },
    {
      "epoch": 5.2642822265625e-05,
      "model_forward_time": 1.715050458908081,
      "step": 55200
    },
    {
      "dynamics_loss_avg": 0.06319230925291777,
      "epoch": 5.2642822265625e-05,
      "step": 55200
    },
    {
      "action_loss_avg": 0.002127293759258464,
      "epoch": 5.2642822265625e-05,
      "step": 55200
    },
    {
      "epoch": 5.2642822265625e-05,
      "step": 55200,
      "training_step_time": 7.568972110748291
    },
    {
      "epoch": 5.264377593994141e-05,
      "model_forward_time": 1.7294325828552246,
      "step": 55201
    },
    {
      "epoch": 5.264377593994141e-05,
      "step": 55201,
      "training_step_time": 7.258063077926636
    },
    {
      "epoch": 5.264472961425781e-05,
      "model_forward_time": 1.7343783378601074,
      "step": 55202
    },
    {
      "epoch": 5.264472961425781e-05,
      "step": 55202,
      "training_step_time": 6.800370931625366
    },
    {
      "epoch": 5.264568328857422e-05,
      "model_forward_time": 1.728442907333374,
      "step": 55203
    },
    {
      "epoch": 5.264568328857422e-05,
      "step": 55203,
      "training_step_time": 6.802905797958374
    },
    {
      "epoch": 5.2646636962890626e-05,
      "model_forward_time": 1.7103674411773682,
      "step": 55204
    },
    {
      "epoch": 5.2646636962890626e-05,
      "step": 55204,
      "training_step_time": 7.676928520202637
    },
    {
      "epoch": 5.264759063720703e-05,
      "model_forward_time": 1.7206041812896729,
      "step": 55205
    },
    {
      "epoch": 5.264759063720703e-05,
      "step": 55205,
      "training_step_time": 8.47812557220459
    },
    {
      "epoch": 5.264854431152344e-05,
      "model_forward_time": 1.7173998355865479,
      "step": 55206
    },
    {
      "epoch": 5.264854431152344e-05,
      "step": 55206,
      "training_step_time": 8.662148475646973
    },
    {
      "epoch": 5.2649497985839844e-05,
      "model_forward_time": 2.28090763092041,
      "step": 55207
    },
    {
      "epoch": 5.2649497985839844e-05,
      "step": 55207,
      "training_step_time": 8.825058937072754
    },
    {
      "epoch": 5.265045166015625e-05,
      "model_forward_time": 1.7670409679412842,
      "step": 55208
    },
    {
      "epoch": 5.265045166015625e-05,
      "step": 55208,
      "training_step_time": 7.794583559036255
    },
    {
      "epoch": 5.265140533447266e-05,
      "model_forward_time": 1.7251482009887695,
      "step": 55209
    },
    {
      "epoch": 5.265140533447266e-05,
      "step": 55209,
      "training_step_time": 8.438789129257202
    },
    {
      "epoch": 5.265235900878906e-05,
      "grad_norm": 0.11709662526845932,
      "learning_rate": 4.552673920854933e-06,
      "loss": 0.0594,
      "step": 55210
    },
    {
      "epoch": 5.265235900878906e-05,
      "model_forward_time": 1.7182297706604004,
      "step": 55210
    },
    {
      "dynamics_loss_avg": 0.060427144169807434,
      "epoch": 5.265235900878906e-05,
      "step": 55210
    },
    {
      "action_loss_avg": 0.0012415168806910516,
      "epoch": 5.265235900878906e-05,
      "step": 55210
    },
    {
      "epoch": 5.265235900878906e-05,
      "step": 55210,
      "training_step_time": 7.2563636302948
    },
    {
      "epoch": 5.265331268310547e-05,
      "model_forward_time": 1.7172884941101074,
      "step": 55211
    },
    {
      "epoch": 5.265331268310547e-05,
      "step": 55211,
      "training_step_time": 6.832256078720093
    },
    {
      "epoch": 5.265426635742188e-05,
      "model_forward_time": 1.7198684215545654,
      "step": 55212
    },
    {
      "epoch": 5.265426635742188e-05,
      "step": 55212,
      "training_step_time": 6.850217819213867
    },
    {
      "epoch": 5.265522003173828e-05,
      "model_forward_time": 1.7311506271362305,
      "step": 55213
    },
    {
      "epoch": 5.265522003173828e-05,
      "step": 55213,
      "training_step_time": 7.711937665939331
    },
    {
      "epoch": 5.2656173706054686e-05,
      "model_forward_time": 1.7281835079193115,
      "step": 55214
    },
    {
      "epoch": 5.2656173706054686e-05,
      "step": 55214,
      "training_step_time": 7.702622413635254
    },
    {
      "epoch": 5.26571273803711e-05,
      "model_forward_time": 1.7345588207244873,
      "step": 55215
    },
    {
      "epoch": 5.26571273803711e-05,
      "step": 55215,
      "training_step_time": 9.555164337158203
    },
    {
      "epoch": 5.26580810546875e-05,
      "model_forward_time": 1.7265856266021729,
      "step": 55216
    },
    {
      "epoch": 5.26580810546875e-05,
      "step": 55216,
      "training_step_time": 8.863565683364868
    },
    {
      "epoch": 5.2659034729003905e-05,
      "model_forward_time": 1.7320506572723389,
      "step": 55217
    },
    {
      "epoch": 5.2659034729003905e-05,
      "step": 55217,
      "training_step_time": 7.777910232543945
    },
    {
      "epoch": 5.2659988403320316e-05,
      "model_forward_time": 1.72590970993042,
      "step": 55218
    },
    {
      "epoch": 5.2659988403320316e-05,
      "step": 55218,
      "training_step_time": 7.57929253578186
    },
    {
      "epoch": 5.266094207763672e-05,
      "model_forward_time": 1.7253191471099854,
      "step": 55219
    },
    {
      "epoch": 5.266094207763672e-05,
      "step": 55219,
      "training_step_time": 7.268200874328613
    },
    {
      "epoch": 5.2661895751953124e-05,
      "grad_norm": 0.15416847169399261,
      "learning_rate": 4.5510271060278005e-06,
      "loss": 0.057,
      "step": 55220
    },
    {
      "epoch": 5.2661895751953124e-05,
      "model_forward_time": 1.713179588317871,
      "step": 55220
    },
    {
      "dynamics_loss_avg": 0.04825735911726951,
      "epoch": 5.2661895751953124e-05,
      "step": 55220
    },
    {
      "action_loss_avg": 0.0019736861810088157,
      "epoch": 5.2661895751953124e-05,
      "step": 55220
    },
    {
      "epoch": 5.2661895751953124e-05,
      "step": 55220,
      "training_step_time": 6.7223827838897705
    },
    {
      "epoch": 5.266284942626953e-05,
      "model_forward_time": 1.7298469543457031,
      "step": 55221
    },
    {
      "epoch": 5.266284942626953e-05,
      "step": 55221,
      "training_step_time": 6.771446228027344
    },
    {
      "epoch": 5.266380310058594e-05,
      "model_forward_time": 1.725527286529541,
      "step": 55222
    },
    {
      "epoch": 5.266380310058594e-05,
      "step": 55222,
      "training_step_time": 7.458964824676514
    },
    {
      "epoch": 5.266475677490234e-05,
      "model_forward_time": 1.7284948825836182,
      "step": 55223
    },
    {
      "epoch": 5.266475677490234e-05,
      "step": 55223,
      "training_step_time": 7.350460767745972
    },
    {
      "epoch": 5.266571044921875e-05,
      "model_forward_time": 1.725860357284546,
      "step": 55224
    },
    {
      "epoch": 5.266571044921875e-05,
      "step": 55224,
      "training_step_time": 9.866766214370728
    },
    {
      "epoch": 5.266666412353516e-05,
      "model_forward_time": 1.7255570888519287,
      "step": 55225
    },
    {
      "epoch": 5.266666412353516e-05,
      "step": 55225,
      "training_step_time": 9.29203987121582
    },
    {
      "epoch": 5.266761779785156e-05,
      "model_forward_time": 2.269357204437256,
      "step": 55226
    },
    {
      "epoch": 5.266761779785156e-05,
      "step": 55226,
      "training_step_time": 7.77798056602478
    },
    {
      "epoch": 5.2668571472167966e-05,
      "model_forward_time": 1.761521816253662,
      "step": 55227
    },
    {
      "epoch": 5.2668571472167966e-05,
      "step": 55227,
      "training_step_time": 7.7453413009643555
    },
    {
      "epoch": 5.2669525146484376e-05,
      "model_forward_time": 1.72471284866333,
      "step": 55228
    },
    {
      "epoch": 5.2669525146484376e-05,
      "step": 55228,
      "training_step_time": 7.66328239440918
    },
    {
      "epoch": 5.267047882080078e-05,
      "model_forward_time": 1.7222774028778076,
      "step": 55229
    },
    {
      "epoch": 5.267047882080078e-05,
      "step": 55229,
      "training_step_time": 6.861220359802246
    },
    {
      "epoch": 5.2671432495117185e-05,
      "grad_norm": 0.08650361746549606,
      "learning_rate": 4.54938034029967e-06,
      "loss": 0.0565,
      "step": 55230
    },
    {
      "epoch": 5.2671432495117185e-05,
      "model_forward_time": 1.7122113704681396,
      "step": 55230
    },
    {
      "dynamics_loss_avg": 0.0646589620038867,
      "epoch": 5.2671432495117185e-05,
      "step": 55230
    },
    {
      "action_loss_avg": 0.0013261144806165249,
      "epoch": 5.2671432495117185e-05,
      "step": 55230
    },
    {
      "epoch": 5.2671432495117185e-05,
      "step": 55230,
      "training_step_time": 6.863752126693726
    },
    {
      "epoch": 5.2672386169433595e-05,
      "model_forward_time": 1.7295384407043457,
      "step": 55231
    },
    {
      "epoch": 5.2672386169433595e-05,
      "step": 55231,
      "training_step_time": 7.376955270767212
    },
    {
      "epoch": 5.267333984375e-05,
      "model_forward_time": 1.3493261337280273,
      "step": 55232
    },
    {
      "epoch": 5.267333984375e-05,
      "step": 55232,
      "training_step_time": 8.51267123222351
    },
    {
      "epoch": 5.26742935180664e-05,
      "model_forward_time": 1.7178707122802734,
      "step": 55233
    },
    {
      "epoch": 5.26742935180664e-05,
      "step": 55233,
      "training_step_time": 8.86070966720581
    },
    {
      "epoch": 5.2675247192382814e-05,
      "model_forward_time": 1.722355604171753,
      "step": 55234
    },
    {
      "epoch": 5.2675247192382814e-05,
      "step": 55234,
      "training_step_time": 7.900239944458008
    },
    {
      "epoch": 5.267620086669922e-05,
      "model_forward_time": 2.271487236022949,
      "step": 55235
    },
    {
      "epoch": 5.267620086669922e-05,
      "step": 55235,
      "training_step_time": 7.827617406845093
    },
    {
      "epoch": 5.267715454101562e-05,
      "model_forward_time": 1.766714334487915,
      "step": 55236
    },
    {
      "epoch": 5.267715454101562e-05,
      "step": 55236,
      "training_step_time": 7.7914416790008545
    },
    {
      "epoch": 5.267810821533203e-05,
      "model_forward_time": 1.7205142974853516,
      "step": 55237
    },
    {
      "epoch": 5.267810821533203e-05,
      "step": 55237,
      "training_step_time": 7.636754274368286
    },
    {
      "epoch": 5.267906188964844e-05,
      "model_forward_time": 1.7219884395599365,
      "step": 55238
    },
    {
      "epoch": 5.267906188964844e-05,
      "step": 55238,
      "training_step_time": 7.290477514266968
    },
    {
      "epoch": 5.268001556396484e-05,
      "model_forward_time": 1.7362449169158936,
      "step": 55239
    },
    {
      "epoch": 5.268001556396484e-05,
      "step": 55239,
      "training_step_time": 6.847711086273193
    },
    {
      "epoch": 5.268096923828125e-05,
      "grad_norm": 0.09505315124988556,
      "learning_rate": 4.54773362385063e-06,
      "loss": 0.0567,
      "step": 55240
    },
    {
      "epoch": 5.268096923828125e-05,
      "model_forward_time": 1.714186429977417,
      "step": 55240
    },
    {
      "dynamics_loss_avg": 0.05034697176888585,
      "epoch": 5.268096923828125e-05,
      "step": 55240
    },
    {
      "action_loss_avg": 0.001783341348345857,
      "epoch": 5.268096923828125e-05,
      "step": 55240
    },
    {
      "epoch": 5.268096923828125e-05,
      "step": 55240,
      "training_step_time": 6.8135716915130615
    },
    {
      "epoch": 5.2681922912597656e-05,
      "model_forward_time": 1.723078727722168,
      "step": 55241
    },
    {
      "epoch": 5.2681922912597656e-05,
      "step": 55241,
      "training_step_time": 8.367603540420532
    },
    {
      "epoch": 5.268287658691406e-05,
      "model_forward_time": 1.729909896850586,
      "step": 55242
    },
    {
      "epoch": 5.268287658691406e-05,
      "step": 55242,
      "training_step_time": 8.589062929153442
    },
    {
      "epoch": 5.268383026123047e-05,
      "model_forward_time": 1.7196729183197021,
      "step": 55243
    },
    {
      "epoch": 5.268383026123047e-05,
      "step": 55243,
      "training_step_time": 8.785512924194336
    },
    {
      "epoch": 5.2684783935546875e-05,
      "model_forward_time": 2.283795118331909,
      "step": 55244
    },
    {
      "epoch": 5.2684783935546875e-05,
      "step": 55244,
      "training_step_time": 8.751540422439575
    },
    {
      "epoch": 5.268573760986328e-05,
      "model_forward_time": 1.7718093395233154,
      "step": 55245
    },
    {
      "epoch": 5.268573760986328e-05,
      "step": 55245,
      "training_step_time": 7.759040832519531
    },
    {
      "epoch": 5.268669128417969e-05,
      "model_forward_time": 1.722125768661499,
      "step": 55246
    },
    {
      "epoch": 5.268669128417969e-05,
      "step": 55246,
      "training_step_time": 7.6224448680877686
    },
    {
      "epoch": 5.2687644958496094e-05,
      "model_forward_time": 1.7384915351867676,
      "step": 55247
    },
    {
      "epoch": 5.2687644958496094e-05,
      "step": 55247,
      "training_step_time": 7.262634992599487
    },
    {
      "epoch": 5.26885986328125e-05,
      "model_forward_time": 1.7255611419677734,
      "step": 55248
    },
    {
      "epoch": 5.26885986328125e-05,
      "step": 55248,
      "training_step_time": 6.819775104522705
    },
    {
      "epoch": 5.268955230712891e-05,
      "model_forward_time": 1.7250688076019287,
      "step": 55249
    },
    {
      "epoch": 5.268955230712891e-05,
      "step": 55249,
      "training_step_time": 6.8528735637664795
    },
    {
      "epoch": 5.269050598144531e-05,
      "grad_norm": 0.1375196874141693,
      "learning_rate": 4.546086956860759e-06,
      "loss": 0.0596,
      "step": 55250
    },
    {
      "epoch": 5.269050598144531e-05,
      "model_forward_time": 1.7123758792877197,
      "step": 55250
    },
    {
      "dynamics_loss_avg": 0.053166982531547544,
      "epoch": 5.269050598144531e-05,
      "step": 55250
    },
    {
      "action_loss_avg": 0.0013931637513451278,
      "epoch": 5.269050598144531e-05,
      "step": 55250
    },
    {
      "epoch": 5.269050598144531e-05,
      "step": 55250,
      "training_step_time": 7.624307155609131
    },
    {
      "epoch": 5.2691459655761717e-05,
      "model_forward_time": 1.723947525024414,
      "step": 55251
    },
    {
      "epoch": 5.2691459655761717e-05,
      "step": 55251,
      "training_step_time": 8.302063465118408
    },
    {
      "epoch": 5.269241333007813e-05,
      "model_forward_time": 1.7227561473846436,
      "step": 55252
    },
    {
      "epoch": 5.269241333007813e-05,
      "step": 55252,
      "training_step_time": 9.507628679275513
    },
    {
      "epoch": 5.269336700439453e-05,
      "model_forward_time": 2.3025968074798584,
      "step": 55253
    },
    {
      "epoch": 5.269336700439453e-05,
      "step": 55253,
      "training_step_time": 7.78963565826416
    },
    {
      "epoch": 5.2694320678710935e-05,
      "model_forward_time": 1.755842924118042,
      "step": 55254
    },
    {
      "epoch": 5.2694320678710935e-05,
      "step": 55254,
      "training_step_time": 7.863493204116821
    },
    {
      "epoch": 5.2695274353027346e-05,
      "model_forward_time": 1.7250335216522217,
      "step": 55255
    },
    {
      "epoch": 5.2695274353027346e-05,
      "step": 55255,
      "training_step_time": 7.707697153091431
    },
    {
      "epoch": 5.269622802734375e-05,
      "model_forward_time": 1.727407693862915,
      "step": 55256
    },
    {
      "epoch": 5.269622802734375e-05,
      "step": 55256,
      "training_step_time": 7.2990124225616455
    },
    {
      "epoch": 5.2697181701660154e-05,
      "model_forward_time": 1.7255477905273438,
      "step": 55257
    },
    {
      "epoch": 5.2697181701660154e-05,
      "step": 55257,
      "training_step_time": 6.815459489822388
    },
    {
      "epoch": 5.2698135375976565e-05,
      "model_forward_time": 1.7255961894989014,
      "step": 55258
    },
    {
      "epoch": 5.2698135375976565e-05,
      "step": 55258,
      "training_step_time": 6.865580797195435
    },
    {
      "epoch": 5.269908905029297e-05,
      "model_forward_time": 1.7279937267303467,
      "step": 55259
    },
    {
      "epoch": 5.269908905029297e-05,
      "step": 55259,
      "training_step_time": 7.600530624389648
    },
    {
      "epoch": 5.270004272460937e-05,
      "grad_norm": 0.09669500589370728,
      "learning_rate": 4.544440339510137e-06,
      "loss": 0.0598,
      "step": 55260
    },
    {
      "epoch": 5.270004272460937e-05,
      "model_forward_time": 1.7214024066925049,
      "step": 55260
    },
    {
      "dynamics_loss_avg": 0.05762608293443918,
      "epoch": 5.270004272460937e-05,
      "step": 55260
    },
    {
      "action_loss_avg": 0.0019641643011709675,
      "epoch": 5.270004272460937e-05,
      "step": 55260
    },
    {
      "epoch": 5.270004272460937e-05,
      "step": 55260,
      "training_step_time": 7.746631145477295
    },
    {
      "epoch": 5.2700996398925784e-05,
      "model_forward_time": 1.7342071533203125,
      "step": 55261
    },
    {
      "epoch": 5.2700996398925784e-05,
      "step": 55261,
      "training_step_time": 7.955786466598511
    },
    {
      "epoch": 5.270195007324219e-05,
      "model_forward_time": 1.737401008605957,
      "step": 55262
    },
    {
      "epoch": 5.270195007324219e-05,
      "step": 55262,
      "training_step_time": 9.679817199707031
    },
    {
      "epoch": 5.270290374755859e-05,
      "model_forward_time": 1.7221431732177734,
      "step": 55263
    },
    {
      "epoch": 5.270290374755859e-05,
      "step": 55263,
      "training_step_time": 10.424890756607056
    },
    {
      "epoch": 5.2703857421875e-05,
      "model_forward_time": 1.7394866943359375,
      "step": 55264
    },
    {
      "epoch": 5.2703857421875e-05,
      "step": 55264,
      "training_step_time": 7.709044694900513
    },
    {
      "epoch": 5.270481109619141e-05,
      "model_forward_time": 1.7391002178192139,
      "step": 55265
    },
    {
      "epoch": 5.270481109619141e-05,
      "step": 55265,
      "training_step_time": 7.600307464599609
    },
    {
      "epoch": 5.270576477050781e-05,
      "model_forward_time": 1.7194914817810059,
      "step": 55266
    },
    {
      "epoch": 5.270576477050781e-05,
      "step": 55266,
      "training_step_time": 6.849202632904053
    },
    {
      "epoch": 5.270671844482422e-05,
      "model_forward_time": 1.7401201725006104,
      "step": 55267
    },
    {
      "epoch": 5.270671844482422e-05,
      "step": 55267,
      "training_step_time": 6.913347005844116
    },
    {
      "epoch": 5.2707672119140626e-05,
      "model_forward_time": 1.733025312423706,
      "step": 55268
    },
    {
      "epoch": 5.2707672119140626e-05,
      "step": 55268,
      "training_step_time": 7.475196838378906
    },
    {
      "epoch": 5.270862579345703e-05,
      "model_forward_time": 1.7291314601898193,
      "step": 55269
    },
    {
      "epoch": 5.270862579345703e-05,
      "step": 55269,
      "training_step_time": 8.73050594329834
    },
    {
      "epoch": 5.270957946777344e-05,
      "grad_norm": 0.10534616559743881,
      "learning_rate": 4.542793771978834e-06,
      "loss": 0.0633,
      "step": 55270
    },
    {
      "epoch": 5.270957946777344e-05,
      "model_forward_time": 1.7208847999572754,
      "step": 55270
    },
    {
      "dynamics_loss_avg": 0.05985780283808708,
      "epoch": 5.270957946777344e-05,
      "step": 55270
    },
    {
      "action_loss_avg": 0.0011329522763844579,
      "epoch": 5.270957946777344e-05,
      "step": 55270
    },
    {
      "epoch": 5.270957946777344e-05,
      "step": 55270,
      "training_step_time": 8.453123092651367
    },
    {
      "epoch": 5.2710533142089845e-05,
      "model_forward_time": 1.7266089916229248,
      "step": 55271
    },
    {
      "epoch": 5.2710533142089845e-05,
      "step": 55271,
      "training_step_time": 8.59358263015747
    },
    {
      "epoch": 5.271148681640625e-05,
      "model_forward_time": 2.3239634037017822,
      "step": 55272
    },
    {
      "epoch": 5.271148681640625e-05,
      "step": 55272,
      "training_step_time": 7.763030290603638
    },
    {
      "epoch": 5.271244049072266e-05,
      "model_forward_time": 1.7858409881591797,
      "step": 55273
    },
    {
      "epoch": 5.271244049072266e-05,
      "step": 55273,
      "training_step_time": 7.638315200805664
    },
    {
      "epoch": 5.271339416503906e-05,
      "model_forward_time": 1.7333984375,
      "step": 55274
    },
    {
      "epoch": 5.271339416503906e-05,
      "step": 55274,
      "training_step_time": 7.559520959854126
    },
    {
      "epoch": 5.271434783935547e-05,
      "model_forward_time": 1.7417724132537842,
      "step": 55275
    },
    {
      "epoch": 5.271434783935547e-05,
      "step": 55275,
      "training_step_time": 6.784080982208252
    },
    {
      "epoch": 5.271530151367188e-05,
      "model_forward_time": 1.721203327178955,
      "step": 55276
    },
    {
      "epoch": 5.271530151367188e-05,
      "step": 55276,
      "training_step_time": 6.69624662399292
    },
    {
      "epoch": 5.271625518798828e-05,
      "model_forward_time": 1.7258281707763672,
      "step": 55277
    },
    {
      "epoch": 5.271625518798828e-05,
      "step": 55277,
      "training_step_time": 8.325016260147095
    },
    {
      "epoch": 5.2717208862304686e-05,
      "model_forward_time": 1.727522850036621,
      "step": 55278
    },
    {
      "epoch": 5.2717208862304686e-05,
      "step": 55278,
      "training_step_time": 7.709004878997803
    },
    {
      "epoch": 5.27181625366211e-05,
      "model_forward_time": 1.7298412322998047,
      "step": 55279
    },
    {
      "epoch": 5.27181625366211e-05,
      "step": 55279,
      "training_step_time": 7.873112201690674
    },
    {
      "epoch": 5.27191162109375e-05,
      "grad_norm": 0.0877012312412262,
      "learning_rate": 4.541147254446916e-06,
      "loss": 0.0578,
      "step": 55280
    },
    {
      "epoch": 5.27191162109375e-05,
      "model_forward_time": 1.7021801471710205,
      "step": 55280
    },
    {
      "dynamics_loss_avg": 0.06002974882721901,
      "epoch": 5.27191162109375e-05,
      "step": 55280
    },
    {
      "action_loss_avg": 0.0008918464591260999,
      "epoch": 5.27191162109375e-05,
      "step": 55280
    },
    {
      "epoch": 5.27191162109375e-05,
      "step": 55280,
      "training_step_time": 8.543447971343994
    },
    {
      "epoch": 5.2720069885253905e-05,
      "model_forward_time": 2.257723331451416,
      "step": 55281
    },
    {
      "epoch": 5.2720069885253905e-05,
      "step": 55281,
      "training_step_time": 8.22869086265564
    },
    {
      "epoch": 5.272102355957031e-05,
      "model_forward_time": 1.7486944198608398,
      "step": 55282
    },
    {
      "epoch": 5.272102355957031e-05,
      "step": 55282,
      "training_step_time": 7.748331069946289
    },
    {
      "epoch": 5.272197723388672e-05,
      "model_forward_time": 1.7089948654174805,
      "step": 55283
    },
    {
      "epoch": 5.272197723388672e-05,
      "step": 55283,
      "training_step_time": 7.5759382247924805
    },
    {
      "epoch": 5.2722930908203124e-05,
      "model_forward_time": 1.712219476699829,
      "step": 55284
    },
    {
      "epoch": 5.2722930908203124e-05,
      "step": 55284,
      "training_step_time": 7.244889974594116
    },
    {
      "epoch": 5.272388458251953e-05,
      "model_forward_time": 1.7143075466156006,
      "step": 55285
    },
    {
      "epoch": 5.272388458251953e-05,
      "step": 55285,
      "training_step_time": 6.79627251625061
    },
    {
      "epoch": 5.272483825683594e-05,
      "model_forward_time": 1.7099506855010986,
      "step": 55286
    },
    {
      "epoch": 5.272483825683594e-05,
      "step": 55286,
      "training_step_time": 7.367600202560425
    },
    {
      "epoch": 5.272579193115234e-05,
      "model_forward_time": 1.707263708114624,
      "step": 55287
    },
    {
      "epoch": 5.272579193115234e-05,
      "step": 55287,
      "training_step_time": 7.615497589111328
    },
    {
      "epoch": 5.272674560546875e-05,
      "model_forward_time": 1.7147877216339111,
      "step": 55288
    },
    {
      "epoch": 5.272674560546875e-05,
      "step": 55288,
      "training_step_time": 9.668760299682617
    },
    {
      "epoch": 5.272769927978516e-05,
      "model_forward_time": 1.7102503776550293,
      "step": 55289
    },
    {
      "epoch": 5.272769927978516e-05,
      "step": 55289,
      "training_step_time": 8.838881731033325
    },
    {
      "epoch": 5.272865295410156e-05,
      "grad_norm": 0.1180998906493187,
      "learning_rate": 4.539500787094445e-06,
      "loss": 0.0583,
      "step": 55290
    },
    {
      "epoch": 5.272865295410156e-05,
      "model_forward_time": 2.2525908946990967,
      "step": 55290
    },
    {
      "dynamics_loss_avg": 0.061559206992387774,
      "epoch": 5.272865295410156e-05,
      "step": 55290
    },
    {
      "action_loss_avg": 0.0018328209727769718,
      "epoch": 5.272865295410156e-05,
      "step": 55290
    },
    {
      "epoch": 5.272865295410156e-05,
      "step": 55290,
      "training_step_time": 8.549560785293579
    },
    {
      "epoch": 5.2729606628417966e-05,
      "model_forward_time": 1.723017930984497,
      "step": 55291
    },
    {
      "epoch": 5.2729606628417966e-05,
      "step": 55291,
      "training_step_time": 8.770828247070312
    },
    {
      "epoch": 5.2730560302734377e-05,
      "model_forward_time": 1.7114360332489014,
      "step": 55292
    },
    {
      "epoch": 5.2730560302734377e-05,
      "step": 55292,
      "training_step_time": 7.513540029525757
    },
    {
      "epoch": 5.273151397705078e-05,
      "model_forward_time": 1.7103710174560547,
      "step": 55293
    },
    {
      "epoch": 5.273151397705078e-05,
      "step": 55293,
      "training_step_time": 7.228066921234131
    },
    {
      "epoch": 5.2732467651367185e-05,
      "model_forward_time": 1.7037992477416992,
      "step": 55294
    },
    {
      "epoch": 5.2732467651367185e-05,
      "step": 55294,
      "training_step_time": 6.722419500350952
    },
    {
      "epoch": 5.2733421325683595e-05,
      "model_forward_time": 1.70847749710083,
      "step": 55295
    },
    {
      "epoch": 5.2733421325683595e-05,
      "step": 55295,
      "training_step_time": 7.330928564071655
    },
    {
      "epoch": 5.2734375e-05,
      "model_forward_time": 1.7445929050445557,
      "step": 55296
    },
    {
      "epoch": 5.2734375e-05,
      "step": 55296,
      "training_step_time": 7.625558853149414
    },
    {
      "epoch": 5.2735328674316403e-05,
      "model_forward_time": 1.7169344425201416,
      "step": 55297
    },
    {
      "epoch": 5.2735328674316403e-05,
      "step": 55297,
      "training_step_time": 7.668941259384155
    },
    {
      "epoch": 5.2736282348632814e-05,
      "model_forward_time": 1.712805986404419,
      "step": 55298
    },
    {
      "epoch": 5.2736282348632814e-05,
      "step": 55298,
      "training_step_time": 8.604620456695557
    },
    {
      "epoch": 5.273723602294922e-05,
      "model_forward_time": 2.239790201187134,
      "step": 55299
    },
    {
      "epoch": 5.273723602294922e-05,
      "step": 55299,
      "training_step_time": 9.94537901878357
    },
    {
      "epoch": 5.273818969726562e-05,
      "grad_norm": 0.0982346460223198,
      "learning_rate": 4.537854370101475e-06,
      "loss": 0.056,
      "step": 55300
    },
    {
      "epoch": 5.273818969726562e-05,
      "model_forward_time": 1.712935447692871,
      "step": 55300
    },
    {
      "dynamics_loss_avg": 0.049196720123291016,
      "epoch": 5.273818969726562e-05,
      "step": 55300
    },
    {
      "action_loss_avg": 0.001703360621468164,
      "epoch": 5.273818969726562e-05,
      "step": 55300
    },
    {
      "epoch": 5.273818969726562e-05,
      "step": 55300,
      "training_step_time": 8.790318965911865
    },
    {
      "epoch": 5.273914337158203e-05,
      "model_forward_time": 1.7534470558166504,
      "step": 55301
    },
    {
      "epoch": 5.273914337158203e-05,
      "step": 55301,
      "training_step_time": 7.635220289230347
    },
    {
      "epoch": 5.274009704589844e-05,
      "model_forward_time": 1.7075302600860596,
      "step": 55302
    },
    {
      "epoch": 5.274009704589844e-05,
      "step": 55302,
      "training_step_time": 7.189733505249023
    },
    {
      "epoch": 5.274105072021484e-05,
      "model_forward_time": 1.7070469856262207,
      "step": 55303
    },
    {
      "epoch": 5.274105072021484e-05,
      "step": 55303,
      "training_step_time": 6.741976261138916
    },
    {
      "epoch": 5.274200439453125e-05,
      "model_forward_time": 1.7071104049682617,
      "step": 55304
    },
    {
      "epoch": 5.274200439453125e-05,
      "step": 55304,
      "training_step_time": 7.4212329387664795
    },
    {
      "epoch": 5.2742958068847656e-05,
      "model_forward_time": 1.7015819549560547,
      "step": 55305
    },
    {
      "epoch": 5.2742958068847656e-05,
      "step": 55305,
      "training_step_time": 7.5596771240234375
    },
    {
      "epoch": 5.274391174316406e-05,
      "model_forward_time": 1.7037632465362549,
      "step": 55306
    },
    {
      "epoch": 5.274391174316406e-05,
      "step": 55306,
      "training_step_time": 7.7820143699646
    },
    {
      "epoch": 5.274486541748047e-05,
      "model_forward_time": 1.7047200202941895,
      "step": 55307
    },
    {
      "epoch": 5.274486541748047e-05,
      "step": 55307,
      "training_step_time": 7.864992141723633
    },
    {
      "epoch": 5.2745819091796875e-05,
      "model_forward_time": 2.2293736934661865,
      "step": 55308
    },
    {
      "epoch": 5.2745819091796875e-05,
      "step": 55308,
      "training_step_time": 9.465914249420166
    },
    {
      "epoch": 5.274677276611328e-05,
      "model_forward_time": 1.706099033355713,
      "step": 55309
    },
    {
      "epoch": 5.274677276611328e-05,
      "step": 55309,
      "training_step_time": 7.898052453994751
    },
    {
      "epoch": 5.274772644042969e-05,
      "grad_norm": 0.10260768234729767,
      "learning_rate": 4.5362080036480575e-06,
      "loss": 0.0579,
      "step": 55310
    },
    {
      "epoch": 5.274772644042969e-05,
      "model_forward_time": 1.7446844577789307,
      "step": 55310
    },
    {
      "dynamics_loss_avg": 0.049903082102537154,
      "epoch": 5.274772644042969e-05,
      "step": 55310
    },
    {
      "action_loss_avg": 0.0017852155957370996,
      "epoch": 5.274772644042969e-05,
      "step": 55310
    },
    {
      "epoch": 5.274772644042969e-05,
      "step": 55310,
      "training_step_time": 7.817840099334717
    },
    {
      "epoch": 5.2748680114746094e-05,
      "model_forward_time": 1.70318603515625,
      "step": 55311
    },
    {
      "epoch": 5.2748680114746094e-05,
      "step": 55311,
      "training_step_time": 7.305156946182251
    },
    {
      "epoch": 5.27496337890625e-05,
      "model_forward_time": 1.7103958129882812,
      "step": 55312
    },
    {
      "epoch": 5.27496337890625e-05,
      "step": 55312,
      "training_step_time": 6.9320549964904785
    },
    {
      "epoch": 5.275058746337891e-05,
      "model_forward_time": 1.711625576019287,
      "step": 55313
    },
    {
      "epoch": 5.275058746337891e-05,
      "step": 55313,
      "training_step_time": 7.562044858932495
    },
    {
      "epoch": 5.275154113769531e-05,
      "model_forward_time": 1.7341642379760742,
      "step": 55314
    },
    {
      "epoch": 5.275154113769531e-05,
      "step": 55314,
      "training_step_time": 7.634836673736572
    },
    {
      "epoch": 5.2752494812011717e-05,
      "model_forward_time": 1.3593494892120361,
      "step": 55315
    },
    {
      "epoch": 5.2752494812011717e-05,
      "step": 55315,
      "training_step_time": 7.992352724075317
    },
    {
      "epoch": 5.275344848632813e-05,
      "model_forward_time": 1.7431399822235107,
      "step": 55316
    },
    {
      "epoch": 5.275344848632813e-05,
      "step": 55316,
      "training_step_time": 9.18464183807373
    },
    {
      "epoch": 5.275440216064453e-05,
      "model_forward_time": 2.273538589477539,
      "step": 55317
    },
    {
      "epoch": 5.275440216064453e-05,
      "step": 55317,
      "training_step_time": 9.24171233177185
    },
    {
      "epoch": 5.2755355834960935e-05,
      "model_forward_time": 1.7354176044464111,
      "step": 55318
    },
    {
      "epoch": 5.2755355834960935e-05,
      "step": 55318,
      "training_step_time": 9.967485666275024
    },
    {
      "epoch": 5.2756309509277346e-05,
      "model_forward_time": 1.729362964630127,
      "step": 55319
    },
    {
      "epoch": 5.2756309509277346e-05,
      "step": 55319,
      "training_step_time": 8.297323942184448
    },
    {
      "epoch": 5.275726318359375e-05,
      "grad_norm": 0.09441641718149185,
      "learning_rate": 4.534561687914235e-06,
      "loss": 0.0638,
      "step": 55320
    },
    {
      "epoch": 5.275726318359375e-05,
      "model_forward_time": 1.719878911972046,
      "step": 55320
    },
    {
      "dynamics_loss_avg": 0.07085061632096767,
      "epoch": 5.275726318359375e-05,
      "step": 55320
    },
    {
      "action_loss_avg": 0.001867910398868844,
      "epoch": 5.275726318359375e-05,
      "step": 55320
    },
    {
      "epoch": 5.275726318359375e-05,
      "step": 55320,
      "training_step_time": 8.085728645324707
    },
    {
      "epoch": 5.2758216857910154e-05,
      "model_forward_time": 1.7386443614959717,
      "step": 55321
    },
    {
      "epoch": 5.2758216857910154e-05,
      "step": 55321,
      "training_step_time": 7.335984468460083
    },
    {
      "epoch": 5.2759170532226565e-05,
      "model_forward_time": 1.6888878345489502,
      "step": 55322
    },
    {
      "epoch": 5.2759170532226565e-05,
      "step": 55322,
      "training_step_time": 7.522547483444214
    },
    {
      "epoch": 5.276012420654297e-05,
      "model_forward_time": 1.698143720626831,
      "step": 55323
    },
    {
      "epoch": 5.276012420654297e-05,
      "step": 55323,
      "training_step_time": 8.049845218658447
    },
    {
      "epoch": 5.276107788085937e-05,
      "model_forward_time": 1.7008073329925537,
      "step": 55324
    },
    {
      "epoch": 5.276107788085937e-05,
      "step": 55324,
      "training_step_time": 7.917720794677734
    },
    {
      "epoch": 5.2762031555175784e-05,
      "model_forward_time": 1.7068417072296143,
      "step": 55325
    },
    {
      "epoch": 5.2762031555175784e-05,
      "step": 55325,
      "training_step_time": 8.579116106033325
    },
    {
      "epoch": 5.276298522949219e-05,
      "model_forward_time": 2.2413175106048584,
      "step": 55326
    },
    {
      "epoch": 5.276298522949219e-05,
      "step": 55326,
      "training_step_time": 9.681789875030518
    },
    {
      "epoch": 5.276393890380859e-05,
      "model_forward_time": 1.6997287273406982,
      "step": 55327
    },
    {
      "epoch": 5.276393890380859e-05,
      "step": 55327,
      "training_step_time": 8.815765619277954
    },
    {
      "epoch": 5.2764892578125e-05,
      "model_forward_time": 1.7056324481964111,
      "step": 55328
    },
    {
      "epoch": 5.2764892578125e-05,
      "step": 55328,
      "training_step_time": 7.7011754512786865
    },
    {
      "epoch": 5.276584625244141e-05,
      "model_forward_time": 1.7478828430175781,
      "step": 55329
    },
    {
      "epoch": 5.276584625244141e-05,
      "step": 55329,
      "training_step_time": 7.633256196975708
    },
    {
      "epoch": 5.276679992675781e-05,
      "grad_norm": 0.11175505071878433,
      "learning_rate": 4.532915423080046e-06,
      "loss": 0.0643,
      "step": 55330
    },
    {
      "epoch": 5.276679992675781e-05,
      "model_forward_time": 1.7060070037841797,
      "step": 55330
    },
    {
      "dynamics_loss_avg": 0.07185333520174027,
      "epoch": 5.276679992675781e-05,
      "step": 55330
    },
    {
      "action_loss_avg": 0.0017933158669620752,
      "epoch": 5.276679992675781e-05,
      "step": 55330
    },
    {
      "epoch": 5.276679992675781e-05,
      "step": 55330,
      "training_step_time": 7.221474885940552
    },
    {
      "epoch": 5.276775360107422e-05,
      "model_forward_time": 1.7034122943878174,
      "step": 55331
    },
    {
      "epoch": 5.276775360107422e-05,
      "step": 55331,
      "training_step_time": 7.381826400756836
    },
    {
      "epoch": 5.2768707275390626e-05,
      "model_forward_time": 1.73736572265625,
      "step": 55332
    },
    {
      "epoch": 5.2768707275390626e-05,
      "step": 55332,
      "training_step_time": 7.354939699172974
    },
    {
      "epoch": 5.276966094970703e-05,
      "model_forward_time": 1.730846643447876,
      "step": 55333
    },
    {
      "epoch": 5.276966094970703e-05,
      "step": 55333,
      "training_step_time": 7.901454925537109
    },
    {
      "epoch": 5.277061462402344e-05,
      "model_forward_time": 1.7310926914215088,
      "step": 55334
    },
    {
      "epoch": 5.277061462402344e-05,
      "step": 55334,
      "training_step_time": 8.493413925170898
    },
    {
      "epoch": 5.2771568298339845e-05,
      "model_forward_time": 2.3083293437957764,
      "step": 55335
    },
    {
      "epoch": 5.2771568298339845e-05,
      "step": 55335,
      "training_step_time": 8.759544372558594
    },
    {
      "epoch": 5.277252197265625e-05,
      "model_forward_time": 1.7332158088684082,
      "step": 55336
    },
    {
      "epoch": 5.277252197265625e-05,
      "step": 55336,
      "training_step_time": 8.79212498664856
    },
    {
      "epoch": 5.277347564697266e-05,
      "model_forward_time": 1.7332255840301514,
      "step": 55337
    },
    {
      "epoch": 5.277347564697266e-05,
      "step": 55337,
      "training_step_time": 7.754567384719849
    },
    {
      "epoch": 5.2774429321289063e-05,
      "model_forward_time": 1.7799558639526367,
      "step": 55338
    },
    {
      "epoch": 5.2774429321289063e-05,
      "step": 55338,
      "training_step_time": 7.489747047424316
    },
    {
      "epoch": 5.277538299560547e-05,
      "model_forward_time": 1.7327966690063477,
      "step": 55339
    },
    {
      "epoch": 5.277538299560547e-05,
      "step": 55339,
      "training_step_time": 7.787243843078613
    },
    {
      "epoch": 5.277633666992188e-05,
      "grad_norm": 0.10778552293777466,
      "learning_rate": 4.5312692093255255e-06,
      "loss": 0.0574,
      "step": 55340
    },
    {
      "epoch": 5.277633666992188e-05,
      "model_forward_time": 1.6836879253387451,
      "step": 55340
    },
    {
      "dynamics_loss_avg": 0.05675065703690052,
      "epoch": 5.277633666992188e-05,
      "step": 55340
    },
    {
      "action_loss_avg": 0.001361671049380675,
      "epoch": 5.277633666992188e-05,
      "step": 55340
    },
    {
      "epoch": 5.277633666992188e-05,
      "step": 55340,
      "training_step_time": 6.7692341804504395
    },
    {
      "epoch": 5.277729034423828e-05,
      "model_forward_time": 1.6987850666046143,
      "step": 55341
    },
    {
      "epoch": 5.277729034423828e-05,
      "step": 55341,
      "training_step_time": 7.384406328201294
    },
    {
      "epoch": 5.2778244018554686e-05,
      "model_forward_time": 1.7021098136901855,
      "step": 55342
    },
    {
      "epoch": 5.2778244018554686e-05,
      "step": 55342,
      "training_step_time": 7.866090297698975
    },
    {
      "epoch": 5.27791976928711e-05,
      "model_forward_time": 1.7063848972320557,
      "step": 55343
    },
    {
      "epoch": 5.27791976928711e-05,
      "step": 55343,
      "training_step_time": 8.463679313659668
    },
    {
      "epoch": 5.27801513671875e-05,
      "model_forward_time": 2.236753463745117,
      "step": 55344
    },
    {
      "epoch": 5.27801513671875e-05,
      "step": 55344,
      "training_step_time": 7.852656126022339
    },
    {
      "epoch": 5.2781105041503905e-05,
      "model_forward_time": 1.704453468322754,
      "step": 55345
    },
    {
      "epoch": 5.2781105041503905e-05,
      "step": 55345,
      "training_step_time": 8.674644231796265
    },
    {
      "epoch": 5.278205871582031e-05,
      "model_forward_time": 1.7068228721618652,
      "step": 55346
    },
    {
      "epoch": 5.278205871582031e-05,
      "step": 55346,
      "training_step_time": 8.5588698387146
    },
    {
      "epoch": 5.278301239013672e-05,
      "model_forward_time": 1.7482476234436035,
      "step": 55347
    },
    {
      "epoch": 5.278301239013672e-05,
      "step": 55347,
      "training_step_time": 7.6637043952941895
    },
    {
      "epoch": 5.2783966064453124e-05,
      "model_forward_time": 1.7054338455200195,
      "step": 55348
    },
    {
      "epoch": 5.2783966064453124e-05,
      "step": 55348,
      "training_step_time": 7.396360397338867
    },
    {
      "epoch": 5.278491973876953e-05,
      "model_forward_time": 1.336984395980835,
      "step": 55349
    },
    {
      "epoch": 5.278491973876953e-05,
      "step": 55349,
      "training_step_time": 7.1699113845825195
    },
    {
      "epoch": 5.278587341308594e-05,
      "grad_norm": 0.1298900544643402,
      "learning_rate": 4.529623046830697e-06,
      "loss": 0.0628,
      "step": 55350
    },
    {
      "epoch": 5.278587341308594e-05,
      "model_forward_time": 1.7105886936187744,
      "step": 55350
    },
    {
      "dynamics_loss_avg": 0.0465726213529706,
      "epoch": 5.278587341308594e-05,
      "step": 55350
    },
    {
      "action_loss_avg": 0.0017820172477513551,
      "epoch": 5.278587341308594e-05,
      "step": 55350
    },
    {
      "epoch": 5.278587341308594e-05,
      "step": 55350,
      "training_step_time": 7.352267026901245
    },
    {
      "epoch": 5.278682708740234e-05,
      "model_forward_time": 1.7183308601379395,
      "step": 55351
    },
    {
      "epoch": 5.278682708740234e-05,
      "step": 55351,
      "training_step_time": 7.759356737136841
    },
    {
      "epoch": 5.278778076171875e-05,
      "model_forward_time": 1.7126407623291016,
      "step": 55352
    },
    {
      "epoch": 5.278778076171875e-05,
      "step": 55352,
      "training_step_time": 9.037490129470825
    },
    {
      "epoch": 5.278873443603516e-05,
      "model_forward_time": 1.7135109901428223,
      "step": 55353
    },
    {
      "epoch": 5.278873443603516e-05,
      "step": 55353,
      "training_step_time": 9.12079405784607
    },
    {
      "epoch": 5.278968811035156e-05,
      "model_forward_time": 2.250230550765991,
      "step": 55354
    },
    {
      "epoch": 5.278968811035156e-05,
      "step": 55354,
      "training_step_time": 8.254545211791992
    },
    {
      "epoch": 5.2790641784667966e-05,
      "model_forward_time": 1.7175681591033936,
      "step": 55355
    },
    {
      "epoch": 5.2790641784667966e-05,
      "step": 55355,
      "training_step_time": 8.332444667816162
    },
    {
      "epoch": 5.2791595458984377e-05,
      "model_forward_time": 1.3531556129455566,
      "step": 55356
    },
    {
      "epoch": 5.2791595458984377e-05,
      "step": 55356,
      "training_step_time": 10.368905544281006
    },
    {
      "epoch": 5.279254913330078e-05,
      "model_forward_time": 1.7162373065948486,
      "step": 55357
    },
    {
      "epoch": 5.279254913330078e-05,
      "step": 55357,
      "training_step_time": 7.477931976318359
    },
    {
      "epoch": 5.2793502807617185e-05,
      "model_forward_time": 1.6996839046478271,
      "step": 55358
    },
    {
      "epoch": 5.2793502807617185e-05,
      "step": 55358,
      "training_step_time": 7.162311792373657
    },
    {
      "epoch": 5.2794456481933595e-05,
      "model_forward_time": 1.694267988204956,
      "step": 55359
    },
    {
      "epoch": 5.2794456481933595e-05,
      "step": 55359,
      "training_step_time": 7.448168039321899
    },
    {
      "epoch": 5.279541015625e-05,
      "grad_norm": 0.08958438783884048,
      "learning_rate": 4.527976935775586e-06,
      "loss": 0.057,
      "step": 55360
    },
    {
      "epoch": 5.279541015625e-05,
      "model_forward_time": 1.7055087089538574,
      "step": 55360
    },
    {
      "dynamics_loss_avg": 0.04508175514638424,
      "epoch": 5.279541015625e-05,
      "step": 55360
    },
    {
      "action_loss_avg": 0.0014600664522731676,
      "epoch": 5.279541015625e-05,
      "step": 55360
    },
    {
      "epoch": 5.279541015625e-05,
      "step": 55360,
      "training_step_time": 7.360805511474609
    },
    {
      "epoch": 5.2796363830566403e-05,
      "model_forward_time": 1.737304925918579,
      "step": 55361
    },
    {
      "epoch": 5.2796363830566403e-05,
      "step": 55361,
      "training_step_time": 8.627902746200562
    },
    {
      "epoch": 5.2797317504882814e-05,
      "model_forward_time": 1.7046024799346924,
      "step": 55362
    },
    {
      "epoch": 5.2797317504882814e-05,
      "step": 55362,
      "training_step_time": 9.118651151657104
    },
    {
      "epoch": 5.279827117919922e-05,
      "model_forward_time": 2.244577407836914,
      "step": 55363
    },
    {
      "epoch": 5.279827117919922e-05,
      "step": 55363,
      "training_step_time": 9.662230730056763
    },
    {
      "epoch": 5.279922485351562e-05,
      "model_forward_time": 1.7020683288574219,
      "step": 55364
    },
    {
      "epoch": 5.279922485351562e-05,
      "step": 55364,
      "training_step_time": 9.311580657958984
    },
    {
      "epoch": 5.280017852783203e-05,
      "model_forward_time": 1.7049593925476074,
      "step": 55365
    },
    {
      "epoch": 5.280017852783203e-05,
      "step": 55365,
      "training_step_time": 9.329619884490967
    },
    {
      "epoch": 5.280113220214844e-05,
      "model_forward_time": 1.4001119136810303,
      "step": 55366
    },
    {
      "epoch": 5.280113220214844e-05,
      "step": 55366,
      "training_step_time": 7.3730926513671875
    },
    {
      "epoch": 5.280208587646484e-05,
      "model_forward_time": 1.7072787284851074,
      "step": 55367
    },
    {
      "epoch": 5.280208587646484e-05,
      "step": 55367,
      "training_step_time": 7.197153806686401
    },
    {
      "epoch": 5.280303955078125e-05,
      "model_forward_time": 1.703155755996704,
      "step": 55368
    },
    {
      "epoch": 5.280303955078125e-05,
      "step": 55368,
      "training_step_time": 7.463356971740723
    },
    {
      "epoch": 5.2803993225097656e-05,
      "model_forward_time": 1.7083277702331543,
      "step": 55369
    },
    {
      "epoch": 5.2803993225097656e-05,
      "step": 55369,
      "training_step_time": 7.482417583465576
    },
    {
      "epoch": 5.280494689941406e-05,
      "grad_norm": 0.08816041797399521,
      "learning_rate": 4.526330876340209e-06,
      "loss": 0.0575,
      "step": 55370
    },
    {
      "epoch": 5.280494689941406e-05,
      "model_forward_time": 1.7031598091125488,
      "step": 55370
    },
    {
      "dynamics_loss_avg": 0.05168572273105383,
      "epoch": 5.280494689941406e-05,
      "step": 55370
    },
    {
      "action_loss_avg": 0.001669521987787448,
      "epoch": 5.280494689941406e-05,
      "step": 55370
    },
    {
      "epoch": 5.280494689941406e-05,
      "step": 55370,
      "training_step_time": 7.92769718170166
    },
    {
      "epoch": 5.280590057373047e-05,
      "model_forward_time": 1.7004587650299072,
      "step": 55371
    },
    {
      "epoch": 5.280590057373047e-05,
      "step": 55371,
      "training_step_time": 7.701036214828491
    },
    {
      "epoch": 5.2806854248046875e-05,
      "model_forward_time": 2.2649035453796387,
      "step": 55372
    },
    {
      "epoch": 5.2806854248046875e-05,
      "step": 55372,
      "training_step_time": 8.861250162124634
    },
    {
      "epoch": 5.280780792236328e-05,
      "model_forward_time": 1.7138144969940186,
      "step": 55373
    },
    {
      "epoch": 5.280780792236328e-05,
      "step": 55373,
      "training_step_time": 9.17875051498413
    },
    {
      "epoch": 5.280876159667969e-05,
      "model_forward_time": 1.700850009918213,
      "step": 55374
    },
    {
      "epoch": 5.280876159667969e-05,
      "step": 55374,
      "training_step_time": 7.965530633926392
    },
    {
      "epoch": 5.2809715270996094e-05,
      "model_forward_time": 1.7398364543914795,
      "step": 55375
    },
    {
      "epoch": 5.2809715270996094e-05,
      "step": 55375,
      "training_step_time": 7.696546792984009
    },
    {
      "epoch": 5.28106689453125e-05,
      "model_forward_time": 1.705073356628418,
      "step": 55376
    },
    {
      "epoch": 5.28106689453125e-05,
      "step": 55376,
      "training_step_time": 7.553003787994385
    },
    {
      "epoch": 5.281162261962891e-05,
      "model_forward_time": 1.7033169269561768,
      "step": 55377
    },
    {
      "epoch": 5.281162261962891e-05,
      "step": 55377,
      "training_step_time": 7.516143798828125
    },
    {
      "epoch": 5.281257629394531e-05,
      "model_forward_time": 1.6988725662231445,
      "step": 55378
    },
    {
      "epoch": 5.281257629394531e-05,
      "step": 55378,
      "training_step_time": 7.500036001205444
    },
    {
      "epoch": 5.281352996826172e-05,
      "model_forward_time": 1.7072458267211914,
      "step": 55379
    },
    {
      "epoch": 5.281352996826172e-05,
      "step": 55379,
      "training_step_time": 8.01480770111084
    },
    {
      "epoch": 5.281448364257813e-05,
      "grad_norm": 0.08763626217842102,
      "learning_rate": 4.5246848687045745e-06,
      "loss": 0.058,
      "step": 55380
    },
    {
      "epoch": 5.281448364257813e-05,
      "model_forward_time": 1.7020387649536133,
      "step": 55380
    },
    {
      "dynamics_loss_avg": 0.06662508025765419,
      "epoch": 5.281448364257813e-05,
      "step": 55380
    },
    {
      "action_loss_avg": 0.0017418009170796721,
      "epoch": 5.281448364257813e-05,
      "step": 55380
    },
    {
      "epoch": 5.281448364257813e-05,
      "step": 55380,
      "training_step_time": 7.677173137664795
    },
    {
      "epoch": 5.281543731689453e-05,
      "model_forward_time": 2.238598108291626,
      "step": 55381
    },
    {
      "epoch": 5.281543731689453e-05,
      "step": 55381,
      "training_step_time": 8.575358390808105
    },
    {
      "epoch": 5.2816390991210935e-05,
      "model_forward_time": 1.6989831924438477,
      "step": 55382
    },
    {
      "epoch": 5.2816390991210935e-05,
      "step": 55382,
      "training_step_time": 9.912360429763794
    },
    {
      "epoch": 5.2817344665527346e-05,
      "model_forward_time": 1.7015721797943115,
      "step": 55383
    },
    {
      "epoch": 5.2817344665527346e-05,
      "step": 55383,
      "training_step_time": 8.921910524368286
    },
    {
      "epoch": 5.281829833984375e-05,
      "model_forward_time": 1.7269034385681152,
      "step": 55384
    },
    {
      "epoch": 5.281829833984375e-05,
      "step": 55384,
      "training_step_time": 7.698574066162109
    },
    {
      "epoch": 5.2819252014160154e-05,
      "model_forward_time": 1.6997311115264893,
      "step": 55385
    },
    {
      "epoch": 5.2819252014160154e-05,
      "step": 55385,
      "training_step_time": 7.499938249588013
    },
    {
      "epoch": 5.2820205688476565e-05,
      "model_forward_time": 1.7128746509552002,
      "step": 55386
    },
    {
      "epoch": 5.2820205688476565e-05,
      "step": 55386,
      "training_step_time": 7.337040901184082
    },
    {
      "epoch": 5.282115936279297e-05,
      "model_forward_time": 1.7017595767974854,
      "step": 55387
    },
    {
      "epoch": 5.282115936279297e-05,
      "step": 55387,
      "training_step_time": 7.505145788192749
    },
    {
      "epoch": 5.282211303710937e-05,
      "model_forward_time": 1.7083871364593506,
      "step": 55388
    },
    {
      "epoch": 5.282211303710937e-05,
      "step": 55388,
      "training_step_time": 8.075710535049438
    },
    {
      "epoch": 5.2823066711425784e-05,
      "model_forward_time": 1.7048170566558838,
      "step": 55389
    },
    {
      "epoch": 5.2823066711425784e-05,
      "step": 55389,
      "training_step_time": 8.66940450668335
    },
    {
      "epoch": 5.282402038574219e-05,
      "grad_norm": 0.08027864247560501,
      "learning_rate": 4.523038913048688e-06,
      "loss": 0.0624,
      "step": 55390
    },
    {
      "epoch": 5.282402038574219e-05,
      "model_forward_time": 1.7038893699645996,
      "step": 55390
    },
    {
      "dynamics_loss_avg": 0.08378296159207821,
      "epoch": 5.282402038574219e-05,
      "step": 55390
    },
    {
      "action_loss_avg": 0.0018911038467194885,
      "epoch": 5.282402038574219e-05,
      "step": 55390
    },
    {
      "epoch": 5.282402038574219e-05,
      "step": 55390,
      "training_step_time": 8.414237022399902
    },
    {
      "epoch": 5.282497406005859e-05,
      "model_forward_time": 2.241684675216675,
      "step": 55391
    },
    {
      "epoch": 5.282497406005859e-05,
      "step": 55391,
      "training_step_time": 7.749955654144287
    },
    {
      "epoch": 5.2825927734375e-05,
      "model_forward_time": 1.3278038501739502,
      "step": 55392
    },
    {
      "epoch": 5.2825927734375e-05,
      "step": 55392,
      "training_step_time": 7.680158615112305
    },
    {
      "epoch": 5.282688140869141e-05,
      "model_forward_time": 1.7018120288848877,
      "step": 55393
    },
    {
      "epoch": 5.282688140869141e-05,
      "step": 55393,
      "training_step_time": 7.620059967041016
    },
    {
      "epoch": 5.282783508300781e-05,
      "model_forward_time": 1.7488796710968018,
      "step": 55394
    },
    {
      "epoch": 5.282783508300781e-05,
      "step": 55394,
      "training_step_time": 7.4178948402404785
    },
    {
      "epoch": 5.282878875732422e-05,
      "model_forward_time": 1.6852679252624512,
      "step": 55395
    },
    {
      "epoch": 5.282878875732422e-05,
      "step": 55395,
      "training_step_time": 7.175144672393799
    },
    {
      "epoch": 5.2829742431640626e-05,
      "model_forward_time": 1.6907577514648438,
      "step": 55396
    },
    {
      "epoch": 5.2829742431640626e-05,
      "step": 55396,
      "training_step_time": 7.3503804206848145
    },
    {
      "epoch": 5.283069610595703e-05,
      "model_forward_time": 1.6869053840637207,
      "step": 55397
    },
    {
      "epoch": 5.283069610595703e-05,
      "step": 55397,
      "training_step_time": 7.886492013931274
    },
    {
      "epoch": 5.283164978027344e-05,
      "model_forward_time": 1.6863391399383545,
      "step": 55398
    },
    {
      "epoch": 5.283164978027344e-05,
      "step": 55398,
      "training_step_time": 8.487515926361084
    },
    {
      "epoch": 5.2832603454589845e-05,
      "model_forward_time": 1.6814367771148682,
      "step": 55399
    },
    {
      "epoch": 5.2832603454589845e-05,
      "step": 55399,
      "training_step_time": 9.41891884803772
    },
    {
      "epoch": 5.283355712890625e-05,
      "grad_norm": 0.11498718708753586,
      "learning_rate": 4.521393009552549e-06,
      "loss": 0.0607,
      "step": 55400
    },
    {
      "epoch": 5.283355712890625e-05,
      "model_forward_time": 2.2543210983276367,
      "step": 55400
    },
    {
      "dynamics_loss_avg": 0.06378297321498394,
      "epoch": 5.283355712890625e-05,
      "step": 55400
    },
    {
      "action_loss_avg": 0.001388415027759038,
      "epoch": 5.283355712890625e-05,
      "step": 55400
    },
    {
      "epoch": 5.283355712890625e-05,
      "step": 55400,
      "training_step_time": 9.099923372268677
    },
    {
      "epoch": 5.283451080322266e-05,
      "model_forward_time": 1.6723682880401611,
      "step": 55401
    },
    {
      "epoch": 5.283451080322266e-05,
      "step": 55401,
      "training_step_time": 9.24644422531128
    },
    {
      "epoch": 5.2835464477539063e-05,
      "model_forward_time": 1.6883158683776855,
      "step": 55402
    },
    {
      "epoch": 5.2835464477539063e-05,
      "step": 55402,
      "training_step_time": 7.641366481781006
    },
    {
      "epoch": 5.283641815185547e-05,
      "model_forward_time": 1.702791452407837,
      "step": 55403
    },
    {
      "epoch": 5.283641815185547e-05,
      "step": 55403,
      "training_step_time": 7.541433572769165
    },
    {
      "epoch": 5.283737182617188e-05,
      "model_forward_time": 1.6913373470306396,
      "step": 55404
    },
    {
      "epoch": 5.283737182617188e-05,
      "step": 55404,
      "training_step_time": 7.171077251434326
    },
    {
      "epoch": 5.283832550048828e-05,
      "model_forward_time": 1.6888442039489746,
      "step": 55405
    },
    {
      "epoch": 5.283832550048828e-05,
      "step": 55405,
      "training_step_time": 7.324208736419678
    },
    {
      "epoch": 5.2839279174804686e-05,
      "model_forward_time": 1.690962553024292,
      "step": 55406
    },
    {
      "epoch": 5.2839279174804686e-05,
      "step": 55406,
      "training_step_time": 8.596563816070557
    },
    {
      "epoch": 5.284023284912109e-05,
      "model_forward_time": 1.684053897857666,
      "step": 55407
    },
    {
      "epoch": 5.284023284912109e-05,
      "step": 55407,
      "training_step_time": 7.313750982284546
    },
    {
      "epoch": 5.28411865234375e-05,
      "model_forward_time": 1.6857388019561768,
      "step": 55408
    },
    {
      "epoch": 5.28411865234375e-05,
      "step": 55408,
      "training_step_time": 9.271331310272217
    },
    {
      "epoch": 5.2842140197753905e-05,
      "model_forward_time": 2.2369332313537598,
      "step": 55409
    },
    {
      "epoch": 5.2842140197753905e-05,
      "step": 55409,
      "training_step_time": 8.277498245239258
    },
    {
      "epoch": 5.284309387207031e-05,
      "grad_norm": 0.08765092492103577,
      "learning_rate": 4.519747158396149e-06,
      "loss": 0.0589,
      "step": 55410
    },
    {
      "epoch": 5.284309387207031e-05,
      "model_forward_time": 1.6868255138397217,
      "step": 55410
    },
    {
      "dynamics_loss_avg": 0.0664722215384245,
      "epoch": 5.284309387207031e-05,
      "step": 55410
    },
    {
      "action_loss_avg": 0.0015131875814404338,
      "epoch": 5.284309387207031e-05,
      "step": 55410
    },
    {
      "epoch": 5.284309387207031e-05,
      "step": 55410,
      "training_step_time": 7.68864631652832
    },
    {
      "epoch": 5.284404754638672e-05,
      "model_forward_time": 1.6921453475952148,
      "step": 55411
    },
    {
      "epoch": 5.284404754638672e-05,
      "step": 55411,
      "training_step_time": 8.79568600654602
    },
    {
      "epoch": 5.2845001220703124e-05,
      "model_forward_time": 1.7009665966033936,
      "step": 55412
    },
    {
      "epoch": 5.2845001220703124e-05,
      "step": 55412,
      "training_step_time": 8.539441108703613
    },
    {
      "epoch": 5.284595489501953e-05,
      "model_forward_time": 1.693145513534546,
      "step": 55413
    },
    {
      "epoch": 5.284595489501953e-05,
      "step": 55413,
      "training_step_time": 7.041563987731934
    },
    {
      "epoch": 5.284690856933594e-05,
      "model_forward_time": 1.333953619003296,
      "step": 55414
    },
    {
      "epoch": 5.284690856933594e-05,
      "step": 55414,
      "training_step_time": 7.321699142456055
    },
    {
      "epoch": 5.284786224365234e-05,
      "model_forward_time": 1.7135975360870361,
      "step": 55415
    },
    {
      "epoch": 5.284786224365234e-05,
      "step": 55415,
      "training_step_time": 7.48413872718811
    },
    {
      "epoch": 5.284881591796875e-05,
      "model_forward_time": 1.7157518863677979,
      "step": 55416
    },
    {
      "epoch": 5.284881591796875e-05,
      "step": 55416,
      "training_step_time": 7.738850116729736
    },
    {
      "epoch": 5.284976959228516e-05,
      "model_forward_time": 1.6962788105010986,
      "step": 55417
    },
    {
      "epoch": 5.284976959228516e-05,
      "step": 55417,
      "training_step_time": 7.628415107727051
    },
    {
      "epoch": 5.285072326660156e-05,
      "model_forward_time": 2.274594306945801,
      "step": 55418
    },
    {
      "epoch": 5.285072326660156e-05,
      "step": 55418,
      "training_step_time": 7.751791715621948
    },
    {
      "epoch": 5.2851676940917966e-05,
      "model_forward_time": 1.7782361507415771,
      "step": 55419
    },
    {
      "epoch": 5.2851676940917966e-05,
      "step": 55419,
      "training_step_time": 8.81377124786377
    },
    {
      "epoch": 5.285263061523438e-05,
      "grad_norm": 0.11219283193349838,
      "learning_rate": 4.51810135975948e-06,
      "loss": 0.057,
      "step": 55420
    },
    {
      "epoch": 5.285263061523438e-05,
      "model_forward_time": 1.7101149559020996,
      "step": 55420
    },
    {
      "dynamics_loss_avg": 0.0546059463173151,
      "epoch": 5.285263061523438e-05,
      "step": 55420
    },
    {
      "action_loss_avg": 0.0016843042219989001,
      "epoch": 5.285263061523438e-05,
      "step": 55420
    },
    {
      "epoch": 5.285263061523438e-05,
      "step": 55420,
      "training_step_time": 7.664325714111328
    },
    {
      "epoch": 5.285358428955078e-05,
      "model_forward_time": 1.70723557472229,
      "step": 55421
    },
    {
      "epoch": 5.285358428955078e-05,
      "step": 55421,
      "training_step_time": 7.580106496810913
    },
    {
      "epoch": 5.2854537963867185e-05,
      "model_forward_time": 1.71181321144104,
      "step": 55422
    },
    {
      "epoch": 5.2854537963867185e-05,
      "step": 55422,
      "training_step_time": 7.539768218994141
    },
    {
      "epoch": 5.2855491638183595e-05,
      "model_forward_time": 1.6952693462371826,
      "step": 55423
    },
    {
      "epoch": 5.2855491638183595e-05,
      "step": 55423,
      "training_step_time": 7.2866857051849365
    },
    {
      "epoch": 5.28564453125e-05,
      "model_forward_time": 1.7123315334320068,
      "step": 55424
    },
    {
      "epoch": 5.28564453125e-05,
      "step": 55424,
      "training_step_time": 7.40017294883728
    },
    {
      "epoch": 5.2857398986816404e-05,
      "model_forward_time": 1.7121539115905762,
      "step": 55425
    },
    {
      "epoch": 5.2857398986816404e-05,
      "step": 55425,
      "training_step_time": 8.965088129043579
    },
    {
      "epoch": 5.2858352661132814e-05,
      "model_forward_time": 1.714909553527832,
      "step": 55426
    },
    {
      "epoch": 5.2858352661132814e-05,
      "step": 55426,
      "training_step_time": 9.462839603424072
    },
    {
      "epoch": 5.285930633544922e-05,
      "model_forward_time": 1.7089765071868896,
      "step": 55427
    },
    {
      "epoch": 5.285930633544922e-05,
      "step": 55427,
      "training_step_time": 9.089409351348877
    },
    {
      "epoch": 5.286026000976562e-05,
      "model_forward_time": 1.7629892826080322,
      "step": 55428
    },
    {
      "epoch": 5.286026000976562e-05,
      "step": 55428,
      "training_step_time": 7.732324838638306
    },
    {
      "epoch": 5.286121368408203e-05,
      "model_forward_time": 1.7090251445770264,
      "step": 55429
    },
    {
      "epoch": 5.286121368408203e-05,
      "step": 55429,
      "training_step_time": 7.762723445892334
    },
    {
      "epoch": 5.286216735839844e-05,
      "grad_norm": 0.09946390241384506,
      "learning_rate": 4.516455613822521e-06,
      "loss": 0.0617,
      "step": 55430
    },
    {
      "epoch": 5.286216735839844e-05,
      "model_forward_time": 1.710428237915039,
      "step": 55430
    },
    {
      "dynamics_loss_avg": 0.050566270388662814,
      "epoch": 5.286216735839844e-05,
      "step": 55430
    },
    {
      "action_loss_avg": 0.0009192483557853848,
      "epoch": 5.286216735839844e-05,
      "step": 55430
    },
    {
      "epoch": 5.286216735839844e-05,
      "step": 55430,
      "training_step_time": 7.648141860961914
    },
    {
      "epoch": 5.286312103271484e-05,
      "model_forward_time": 1.000476360321045,
      "step": 55431
    },
    {
      "epoch": 5.286312103271484e-05,
      "step": 55431,
      "training_step_time": 7.6168811321258545
    },
    {
      "epoch": 5.286407470703125e-05,
      "model_forward_time": 1.714331865310669,
      "step": 55432
    },
    {
      "epoch": 5.286407470703125e-05,
      "step": 55432,
      "training_step_time": 7.213203430175781
    },
    {
      "epoch": 5.2865028381347656e-05,
      "model_forward_time": 1.7130718231201172,
      "step": 55433
    },
    {
      "epoch": 5.2865028381347656e-05,
      "step": 55433,
      "training_step_time": 6.825622797012329
    },
    {
      "epoch": 5.286598205566406e-05,
      "model_forward_time": 1.697047472000122,
      "step": 55434
    },
    {
      "epoch": 5.286598205566406e-05,
      "step": 55434,
      "training_step_time": 8.059438943862915
    },
    {
      "epoch": 5.286693572998047e-05,
      "model_forward_time": 1.7067646980285645,
      "step": 55435
    },
    {
      "epoch": 5.286693572998047e-05,
      "step": 55435,
      "training_step_time": 8.651830196380615
    },
    {
      "epoch": 5.2867889404296875e-05,
      "model_forward_time": 1.7196292877197266,
      "step": 55436
    },
    {
      "epoch": 5.2867889404296875e-05,
      "step": 55436,
      "training_step_time": 8.661194562911987
    },
    {
      "epoch": 5.286884307861328e-05,
      "model_forward_time": 2.293588638305664,
      "step": 55437
    },
    {
      "epoch": 5.286884307861328e-05,
      "step": 55437,
      "training_step_time": 7.655726194381714
    },
    {
      "epoch": 5.286979675292969e-05,
      "model_forward_time": 1.8064601421356201,
      "step": 55438
    },
    {
      "epoch": 5.286979675292969e-05,
      "step": 55438,
      "training_step_time": 7.808652400970459
    },
    {
      "epoch": 5.2870750427246094e-05,
      "model_forward_time": 1.8076815605163574,
      "step": 55439
    },
    {
      "epoch": 5.2870750427246094e-05,
      "step": 55439,
      "training_step_time": 8.702355861663818
    },
    {
      "epoch": 5.28717041015625e-05,
      "grad_norm": 0.09805741161108017,
      "learning_rate": 4.514809920765249e-06,
      "loss": 0.0597,
      "step": 55440
    },
    {
      "epoch": 5.28717041015625e-05,
      "model_forward_time": 1.8090274333953857,
      "step": 55440
    },
    {
      "dynamics_loss_avg": 0.06647445019334555,
      "epoch": 5.28717041015625e-05,
      "step": 55440
    },
    {
      "action_loss_avg": 0.0013234897283837198,
      "epoch": 5.28717041015625e-05,
      "step": 55440
    },
    {
      "epoch": 5.28717041015625e-05,
      "step": 55440,
      "training_step_time": 7.590185880661011
    },
    {
      "epoch": 5.287265777587891e-05,
      "model_forward_time": 1.8052504062652588,
      "step": 55441
    },
    {
      "epoch": 5.287265777587891e-05,
      "step": 55441,
      "training_step_time": 7.363874197006226
    },
    {
      "epoch": 5.287361145019531e-05,
      "model_forward_time": 1.8033032417297363,
      "step": 55442
    },
    {
      "epoch": 5.287361145019531e-05,
      "step": 55442,
      "training_step_time": 6.784668445587158
    },
    {
      "epoch": 5.287456512451172e-05,
      "model_forward_time": 1.8040738105773926,
      "step": 55443
    },
    {
      "epoch": 5.287456512451172e-05,
      "step": 55443,
      "training_step_time": 7.879836320877075
    },
    {
      "epoch": 5.287551879882813e-05,
      "model_forward_time": 1.8069047927856445,
      "step": 55444
    },
    {
      "epoch": 5.287551879882813e-05,
      "step": 55444,
      "training_step_time": 8.406720638275146
    },
    {
      "epoch": 5.287647247314453e-05,
      "model_forward_time": 1.804905652999878,
      "step": 55445
    },
    {
      "epoch": 5.287647247314453e-05,
      "step": 55445,
      "training_step_time": 10.487407207489014
    },
    {
      "epoch": 5.2877426147460936e-05,
      "model_forward_time": 2.3431413173675537,
      "step": 55446
    },
    {
      "epoch": 5.2877426147460936e-05,
      "step": 55446,
      "training_step_time": 7.615137815475464
    },
    {
      "epoch": 5.2878379821777346e-05,
      "model_forward_time": 1.8056516647338867,
      "step": 55447
    },
    {
      "epoch": 5.2878379821777346e-05,
      "step": 55447,
      "training_step_time": 7.703545808792114
    },
    {
      "epoch": 5.287933349609375e-05,
      "model_forward_time": 1.807105302810669,
      "step": 55448
    },
    {
      "epoch": 5.287933349609375e-05,
      "step": 55448,
      "training_step_time": 7.719573497772217
    },
    {
      "epoch": 5.2880287170410154e-05,
      "model_forward_time": 1.8104281425476074,
      "step": 55449
    },
    {
      "epoch": 5.2880287170410154e-05,
      "step": 55449,
      "training_step_time": 7.601192951202393
    },
    {
      "epoch": 5.2881240844726565e-05,
      "grad_norm": 0.09178579598665237,
      "learning_rate": 4.513164280767636e-06,
      "loss": 0.0566,
      "step": 55450
    },
    {
      "epoch": 5.2881240844726565e-05,
      "model_forward_time": 1.8095147609710693,
      "step": 55450
    },
    {
      "dynamics_loss_avg": 0.048766665160655975,
      "epoch": 5.2881240844726565e-05,
      "step": 55450
    },
    {
      "action_loss_avg": 0.0011916340474272147,
      "epoch": 5.2881240844726565e-05,
      "step": 55450
    },
    {
      "epoch": 5.2881240844726565e-05,
      "step": 55450,
      "training_step_time": 7.340774297714233
    },
    {
      "epoch": 5.288219451904297e-05,
      "model_forward_time": 1.8094046115875244,
      "step": 55451
    },
    {
      "epoch": 5.288219451904297e-05,
      "step": 55451,
      "training_step_time": 7.2851972579956055
    },
    {
      "epoch": 5.288314819335937e-05,
      "model_forward_time": 1.7937324047088623,
      "step": 55452
    },
    {
      "epoch": 5.288314819335937e-05,
      "step": 55452,
      "training_step_time": 8.171842098236084
    },
    {
      "epoch": 5.2884101867675784e-05,
      "model_forward_time": 1.8184123039245605,
      "step": 55453
    },
    {
      "epoch": 5.2884101867675784e-05,
      "step": 55453,
      "training_step_time": 8.314753293991089
    },
    {
      "epoch": 5.288505554199219e-05,
      "model_forward_time": 1.8061699867248535,
      "step": 55454
    },
    {
      "epoch": 5.288505554199219e-05,
      "step": 55454,
      "training_step_time": 8.822045803070068
    },
    {
      "epoch": 5.288600921630859e-05,
      "model_forward_time": 2.332731246948242,
      "step": 55455
    },
    {
      "epoch": 5.288600921630859e-05,
      "step": 55455,
      "training_step_time": 8.655429124832153
    },
    {
      "epoch": 5.2886962890625e-05,
      "model_forward_time": 1.7971785068511963,
      "step": 55456
    },
    {
      "epoch": 5.2886962890625e-05,
      "step": 55456,
      "training_step_time": 7.780627965927124
    },
    {
      "epoch": 5.288791656494141e-05,
      "model_forward_time": 1.8029701709747314,
      "step": 55457
    },
    {
      "epoch": 5.288791656494141e-05,
      "step": 55457,
      "training_step_time": 7.63131856918335
    },
    {
      "epoch": 5.288887023925781e-05,
      "model_forward_time": 1.8104205131530762,
      "step": 55458
    },
    {
      "epoch": 5.288887023925781e-05,
      "step": 55458,
      "training_step_time": 8.167712688446045
    },
    {
      "epoch": 5.288982391357422e-05,
      "model_forward_time": 1.8008458614349365,
      "step": 55459
    },
    {
      "epoch": 5.288982391357422e-05,
      "step": 55459,
      "training_step_time": 7.53408408164978
    },
    {
      "epoch": 5.2890777587890626e-05,
      "grad_norm": 0.1006813645362854,
      "learning_rate": 4.5115186940096445e-06,
      "loss": 0.0591,
      "step": 55460
    },
    {
      "epoch": 5.2890777587890626e-05,
      "model_forward_time": 1.8001163005828857,
      "step": 55460
    },
    {
      "dynamics_loss_avg": 0.05047006942331791,
      "epoch": 5.2890777587890626e-05,
      "step": 55460
    },
    {
      "action_loss_avg": 0.0010701743973186239,
      "epoch": 5.2890777587890626e-05,
      "step": 55460
    },
    {
      "epoch": 5.2890777587890626e-05,
      "step": 55460,
      "training_step_time": 7.153007745742798
    },
    {
      "epoch": 5.289173126220703e-05,
      "model_forward_time": 1.4425017833709717,
      "step": 55461
    },
    {
      "epoch": 5.289173126220703e-05,
      "step": 55461,
      "training_step_time": 7.72555136680603
    },
    {
      "epoch": 5.289268493652344e-05,
      "model_forward_time": 1.8051056861877441,
      "step": 55462
    },
    {
      "epoch": 5.289268493652344e-05,
      "step": 55462,
      "training_step_time": 8.099039554595947
    },
    {
      "epoch": 5.2893638610839845e-05,
      "model_forward_time": 1.8071932792663574,
      "step": 55463
    },
    {
      "epoch": 5.2893638610839845e-05,
      "step": 55463,
      "training_step_time": 8.872272968292236
    },
    {
      "epoch": 5.289459228515625e-05,
      "model_forward_time": 1.8090708255767822,
      "step": 55464
    },
    {
      "epoch": 5.289459228515625e-05,
      "step": 55464,
      "training_step_time": 7.703097581863403
    },
    {
      "epoch": 5.289554595947266e-05,
      "model_forward_time": 2.346982955932617,
      "step": 55465
    },
    {
      "epoch": 5.289554595947266e-05,
      "step": 55465,
      "training_step_time": 9.223860502243042
    },
    {
      "epoch": 5.2896499633789064e-05,
      "model_forward_time": 1.805210828781128,
      "step": 55466
    },
    {
      "epoch": 5.2896499633789064e-05,
      "step": 55466,
      "training_step_time": 7.663339138031006
    },
    {
      "epoch": 5.289745330810547e-05,
      "model_forward_time": 1.8027050495147705,
      "step": 55467
    },
    {
      "epoch": 5.289745330810547e-05,
      "step": 55467,
      "training_step_time": 7.627756118774414
    },
    {
      "epoch": 5.289840698242188e-05,
      "model_forward_time": 1.8067402839660645,
      "step": 55468
    },
    {
      "epoch": 5.289840698242188e-05,
      "step": 55468,
      "training_step_time": 8.29601788520813
    },
    {
      "epoch": 5.289936065673828e-05,
      "model_forward_time": 1.8057126998901367,
      "step": 55469
    },
    {
      "epoch": 5.289936065673828e-05,
      "step": 55469,
      "training_step_time": 7.2869837284088135
    },
    {
      "epoch": 5.2900314331054686e-05,
      "grad_norm": 0.11264436691999435,
      "learning_rate": 4.509873160671231e-06,
      "loss": 0.0552,
      "step": 55470
    },
    {
      "epoch": 5.2900314331054686e-05,
      "model_forward_time": 1.7950425148010254,
      "step": 55470
    },
    {
      "dynamics_loss_avg": 0.048273021122440696,
      "epoch": 5.2900314331054686e-05,
      "step": 55470
    },
    {
      "action_loss_avg": 0.0013389788859058172,
      "epoch": 5.2900314331054686e-05,
      "step": 55470
    },
    {
      "epoch": 5.2900314331054686e-05,
      "step": 55470,
      "training_step_time": 7.879194974899292
    },
    {
      "epoch": 5.290126800537109e-05,
      "model_forward_time": 1.810293436050415,
      "step": 55471
    },
    {
      "epoch": 5.290126800537109e-05,
      "step": 55471,
      "training_step_time": 7.809662580490112
    },
    {
      "epoch": 5.29022216796875e-05,
      "model_forward_time": 1.8055694103240967,
      "step": 55472
    },
    {
      "epoch": 5.29022216796875e-05,
      "step": 55472,
      "training_step_time": 7.538760185241699
    },
    {
      "epoch": 5.2903175354003905e-05,
      "model_forward_time": 1.8030731678009033,
      "step": 55473
    },
    {
      "epoch": 5.2903175354003905e-05,
      "step": 55473,
      "training_step_time": 8.322577714920044
    },
    {
      "epoch": 5.290412902832031e-05,
      "model_forward_time": 2.3366482257843018,
      "step": 55474
    },
    {
      "epoch": 5.290412902832031e-05,
      "step": 55474,
      "training_step_time": 7.783306360244751
    },
    {
      "epoch": 5.290508270263672e-05,
      "model_forward_time": 1.8057026863098145,
      "step": 55475
    },
    {
      "epoch": 5.290508270263672e-05,
      "step": 55475,
      "training_step_time": 8.58132529258728
    },
    {
      "epoch": 5.2906036376953124e-05,
      "model_forward_time": 1.7900338172912598,
      "step": 55476
    },
    {
      "epoch": 5.2906036376953124e-05,
      "step": 55476,
      "training_step_time": 7.733744144439697
    },
    {
      "epoch": 5.290699005126953e-05,
      "model_forward_time": 1.8081588745117188,
      "step": 55477
    },
    {
      "epoch": 5.290699005126953e-05,
      "step": 55477,
      "training_step_time": 7.646359920501709
    },
    {
      "epoch": 5.290794372558594e-05,
      "model_forward_time": 1.8061821460723877,
      "step": 55478
    },
    {
      "epoch": 5.290794372558594e-05,
      "step": 55478,
      "training_step_time": 7.592379331588745
    },
    {
      "epoch": 5.290889739990234e-05,
      "model_forward_time": 1.8096401691436768,
      "step": 55479
    },
    {
      "epoch": 5.290889739990234e-05,
      "step": 55479,
      "training_step_time": 7.336300611495972
    },
    {
      "epoch": 5.290985107421875e-05,
      "grad_norm": 0.1621754914522171,
      "learning_rate": 4.508227680932355e-06,
      "loss": 0.058,
      "step": 55480
    },
    {
      "epoch": 5.290985107421875e-05,
      "model_forward_time": 1.7990260124206543,
      "step": 55480
    },
    {
      "dynamics_loss_avg": 0.054932639189064505,
      "epoch": 5.290985107421875e-05,
      "step": 55480
    },
    {
      "action_loss_avg": 0.0014613046660088003,
      "epoch": 5.290985107421875e-05,
      "step": 55480
    },
    {
      "epoch": 5.290985107421875e-05,
      "step": 55480,
      "training_step_time": 8.449689865112305
    },
    {
      "epoch": 5.291080474853516e-05,
      "model_forward_time": 1.8054795265197754,
      "step": 55481
    },
    {
      "epoch": 5.291080474853516e-05,
      "step": 55481,
      "training_step_time": 7.554353475570679
    },
    {
      "epoch": 5.291175842285156e-05,
      "model_forward_time": 1.8035461902618408,
      "step": 55482
    },
    {
      "epoch": 5.291175842285156e-05,
      "step": 55482,
      "training_step_time": 8.381177425384521
    },
    {
      "epoch": 5.2912712097167966e-05,
      "model_forward_time": 2.327197551727295,
      "step": 55483
    },
    {
      "epoch": 5.2912712097167966e-05,
      "step": 55483,
      "training_step_time": 8.815353155136108
    },
    {
      "epoch": 5.291366577148438e-05,
      "model_forward_time": 1.8036949634552002,
      "step": 55484
    },
    {
      "epoch": 5.291366577148438e-05,
      "step": 55484,
      "training_step_time": 7.716352701187134
    },
    {
      "epoch": 5.291461944580078e-05,
      "model_forward_time": 1.8045265674591064,
      "step": 55485
    },
    {
      "epoch": 5.291461944580078e-05,
      "step": 55485,
      "training_step_time": 7.778264760971069
    },
    {
      "epoch": 5.2915573120117185e-05,
      "model_forward_time": 1.785569429397583,
      "step": 55486
    },
    {
      "epoch": 5.2915573120117185e-05,
      "step": 55486,
      "training_step_time": 7.5549046993255615
    },
    {
      "epoch": 5.2916526794433596e-05,
      "model_forward_time": 1.8060290813446045,
      "step": 55487
    },
    {
      "epoch": 5.2916526794433596e-05,
      "step": 55487,
      "training_step_time": 7.65736985206604
    },
    {
      "epoch": 5.291748046875e-05,
      "model_forward_time": 1.8087656497955322,
      "step": 55488
    },
    {
      "epoch": 5.291748046875e-05,
      "step": 55488,
      "training_step_time": 7.2220165729522705
    },
    {
      "epoch": 5.2918434143066404e-05,
      "model_forward_time": 1.8119330406188965,
      "step": 55489
    },
    {
      "epoch": 5.2918434143066404e-05,
      "step": 55489,
      "training_step_time": 8.45658016204834
    },
    {
      "epoch": 5.2919387817382814e-05,
      "grad_norm": 0.16027571260929108,
      "learning_rate": 4.506582254972961e-06,
      "loss": 0.0532,
      "step": 55490
    },
    {
      "epoch": 5.2919387817382814e-05,
      "model_forward_time": 1.8050174713134766,
      "step": 55490
    },
    {
      "dynamics_loss_avg": 0.04304512795060873,
      "epoch": 5.2919387817382814e-05,
      "step": 55490
    },
    {
      "action_loss_avg": 0.0017379920231178404,
      "epoch": 5.2919387817382814e-05,
      "step": 55490
    },
    {
      "epoch": 5.2919387817382814e-05,
      "step": 55490,
      "training_step_time": 8.394192457199097
    },
    {
      "epoch": 5.292034149169922e-05,
      "model_forward_time": 1.8076128959655762,
      "step": 55491
    },
    {
      "epoch": 5.292034149169922e-05,
      "step": 55491,
      "training_step_time": 7.678267002105713
    },
    {
      "epoch": 5.292129516601562e-05,
      "model_forward_time": 2.323946952819824,
      "step": 55492
    },
    {
      "epoch": 5.292129516601562e-05,
      "step": 55492,
      "training_step_time": 7.628380298614502
    },
    {
      "epoch": 5.292224884033203e-05,
      "model_forward_time": 1.8001489639282227,
      "step": 55493
    },
    {
      "epoch": 5.292224884033203e-05,
      "step": 55493,
      "training_step_time": 8.40431261062622
    },
    {
      "epoch": 5.292320251464844e-05,
      "model_forward_time": 1.8088865280151367,
      "step": 55494
    },
    {
      "epoch": 5.292320251464844e-05,
      "step": 55494,
      "training_step_time": 8.655232191085815
    },
    {
      "epoch": 5.292415618896484e-05,
      "model_forward_time": 1.4289884567260742,
      "step": 55495
    },
    {
      "epoch": 5.292415618896484e-05,
      "step": 55495,
      "training_step_time": 7.526976585388184
    },
    {
      "epoch": 5.292510986328125e-05,
      "model_forward_time": 1.8038387298583984,
      "step": 55496
    },
    {
      "epoch": 5.292510986328125e-05,
      "step": 55496,
      "training_step_time": 8.55113172531128
    },
    {
      "epoch": 5.2926063537597656e-05,
      "model_forward_time": 1.7979395389556885,
      "step": 55497
    },
    {
      "epoch": 5.2926063537597656e-05,
      "step": 55497,
      "training_step_time": 7.138576507568359
    },
    {
      "epoch": 5.292701721191406e-05,
      "model_forward_time": 1.8072638511657715,
      "step": 55498
    },
    {
      "epoch": 5.292701721191406e-05,
      "step": 55498,
      "training_step_time": 8.619232654571533
    },
    {
      "epoch": 5.292797088623047e-05,
      "model_forward_time": 1.8077750205993652,
      "step": 55499
    },
    {
      "epoch": 5.292797088623047e-05,
      "step": 55499,
      "training_step_time": 7.608462572097778
    },
    {
      "epoch": 5.2928924560546875e-05,
      "grad_norm": 0.11460673063993454,
      "learning_rate": 4.504936882972989e-06,
      "loss": 0.0584,
      "step": 55500
    },
    {
      "epoch": 5.2928924560546875e-05,
      "model_forward_time": 1.7949903011322021,
      "step": 55500
    },
    {
      "dynamics_loss_avg": 0.06846294878050685,
      "epoch": 5.2928924560546875e-05,
      "step": 55500
    },
    {
      "action_loss_avg": 0.0013058542434009723,
      "epoch": 5.2928924560546875e-05,
      "step": 55500
    },
    {
      "epoch": 5.2928924560546875e-05,
      "step": 55500,
      "training_step_time": 8.595767259597778
    },
    {
      "epoch": 5.292987823486328e-05,
      "model_forward_time": 1.8090999126434326,
      "step": 55501
    },
    {
      "epoch": 5.292987823486328e-05,
      "step": 55501,
      "training_step_time": 9.675727128982544
    },
    {
      "epoch": 5.293083190917969e-05,
      "model_forward_time": 1.808330774307251,
      "step": 55502
    },
    {
      "epoch": 5.293083190917969e-05,
      "step": 55502,
      "training_step_time": 7.693167686462402
    },
    {
      "epoch": 5.2931785583496094e-05,
      "model_forward_time": 1.8020448684692383,
      "step": 55503
    },
    {
      "epoch": 5.2931785583496094e-05,
      "step": 55503,
      "training_step_time": 7.6673524379730225
    },
    {
      "epoch": 5.29327392578125e-05,
      "model_forward_time": 1.8105008602142334,
      "step": 55504
    },
    {
      "epoch": 5.29327392578125e-05,
      "step": 55504,
      "training_step_time": 7.5642619132995605
    },
    {
      "epoch": 5.293369293212891e-05,
      "model_forward_time": 1.8095245361328125,
      "step": 55505
    },
    {
      "epoch": 5.293369293212891e-05,
      "step": 55505,
      "training_step_time": 8.111053705215454
    },
    {
      "epoch": 5.293464660644531e-05,
      "model_forward_time": 1.8082835674285889,
      "step": 55506
    },
    {
      "epoch": 5.293464660644531e-05,
      "step": 55506,
      "training_step_time": 7.478041887283325
    },
    {
      "epoch": 5.293560028076172e-05,
      "model_forward_time": 1.7793805599212646,
      "step": 55507
    },
    {
      "epoch": 5.293560028076172e-05,
      "step": 55507,
      "training_step_time": 8.222076177597046
    },
    {
      "epoch": 5.293655395507813e-05,
      "model_forward_time": 1.70937180519104,
      "step": 55508
    },
    {
      "epoch": 5.293655395507813e-05,
      "step": 55508,
      "training_step_time": 7.83484148979187
    },
    {
      "epoch": 5.293750762939453e-05,
      "model_forward_time": 1.70930814743042,
      "step": 55509
    },
    {
      "epoch": 5.293750762939453e-05,
      "step": 55509,
      "training_step_time": 10.44607663154602
    },
    {
      "epoch": 5.2938461303710936e-05,
      "grad_norm": 0.11507055908441544,
      "learning_rate": 4.5032915651123745e-06,
      "loss": 0.0522,
      "step": 55510
    },
    {
      "epoch": 5.2938461303710936e-05,
      "model_forward_time": 1.7030737400054932,
      "step": 55510
    },
    {
      "dynamics_loss_avg": 0.04219758082181215,
      "epoch": 5.2938461303710936e-05,
      "step": 55510
    },
    {
      "action_loss_avg": 0.0012441207189112901,
      "epoch": 5.2938461303710936e-05,
      "step": 55510
    },
    {
      "epoch": 5.2938461303710936e-05,
      "step": 55510,
      "training_step_time": 7.678999185562134
    },
    {
      "epoch": 5.2939414978027346e-05,
      "model_forward_time": 2.329935073852539,
      "step": 55511
    },
    {
      "epoch": 5.2939414978027346e-05,
      "step": 55511,
      "training_step_time": 8.283388137817383
    },
    {
      "epoch": 5.294036865234375e-05,
      "model_forward_time": 1.7164099216461182,
      "step": 55512
    },
    {
      "epoch": 5.294036865234375e-05,
      "step": 55512,
      "training_step_time": 7.64344596862793
    },
    {
      "epoch": 5.2941322326660154e-05,
      "model_forward_time": 1.7099568843841553,
      "step": 55513
    },
    {
      "epoch": 5.2941322326660154e-05,
      "step": 55513,
      "training_step_time": 8.22781252861023
    },
    {
      "epoch": 5.2942276000976565e-05,
      "model_forward_time": 1.7064266204833984,
      "step": 55514
    },
    {
      "epoch": 5.2942276000976565e-05,
      "step": 55514,
      "training_step_time": 7.5573930740356445
    },
    {
      "epoch": 5.294322967529297e-05,
      "model_forward_time": 1.7004375457763672,
      "step": 55515
    },
    {
      "epoch": 5.294322967529297e-05,
      "step": 55515,
      "training_step_time": 7.5079264640808105
    },
    {
      "epoch": 5.294418334960937e-05,
      "model_forward_time": 1.7149608135223389,
      "step": 55516
    },
    {
      "epoch": 5.294418334960937e-05,
      "step": 55516,
      "training_step_time": 7.734891653060913
    },
    {
      "epoch": 5.2945137023925784e-05,
      "model_forward_time": 1.7138445377349854,
      "step": 55517
    },
    {
      "epoch": 5.2945137023925784e-05,
      "step": 55517,
      "training_step_time": 8.949502944946289
    },
    {
      "epoch": 5.294609069824219e-05,
      "model_forward_time": 1.7144763469696045,
      "step": 55518
    },
    {
      "epoch": 5.294609069824219e-05,
      "step": 55518,
      "training_step_time": 7.620440244674683
    },
    {
      "epoch": 5.294704437255859e-05,
      "model_forward_time": 1.711071491241455,
      "step": 55519
    },
    {
      "epoch": 5.294704437255859e-05,
      "step": 55519,
      "training_step_time": 7.716914653778076
    },
    {
      "epoch": 5.2947998046875e-05,
      "grad_norm": 0.1311834454536438,
      "learning_rate": 4.501646301571048e-06,
      "loss": 0.0549,
      "step": 55520
    },
    {
      "epoch": 5.2947998046875e-05,
      "model_forward_time": 2.3309924602508545,
      "step": 55520
    },
    {
      "dynamics_loss_avg": 0.07262176349759102,
      "epoch": 5.2947998046875e-05,
      "step": 55520
    },
    {
      "action_loss_avg": 0.0017975042486796156,
      "epoch": 5.2947998046875e-05,
      "step": 55520
    },
    {
      "epoch": 5.2947998046875e-05,
      "step": 55520,
      "training_step_time": 7.7579004764556885
    },
    {
      "epoch": 5.294895172119141e-05,
      "model_forward_time": 1.7124218940734863,
      "step": 55521
    },
    {
      "epoch": 5.294895172119141e-05,
      "step": 55521,
      "training_step_time": 7.646116733551025
    },
    {
      "epoch": 5.294990539550781e-05,
      "model_forward_time": 1.715482234954834,
      "step": 55522
    },
    {
      "epoch": 5.294990539550781e-05,
      "step": 55522,
      "training_step_time": 9.318604230880737
    },
    {
      "epoch": 5.295085906982422e-05,
      "model_forward_time": 1.7098987102508545,
      "step": 55523
    },
    {
      "epoch": 5.295085906982422e-05,
      "step": 55523,
      "training_step_time": 8.129184007644653
    },
    {
      "epoch": 5.2951812744140626e-05,
      "model_forward_time": 1.710364580154419,
      "step": 55524
    },
    {
      "epoch": 5.2951812744140626e-05,
      "step": 55524,
      "training_step_time": 7.554727077484131
    },
    {
      "epoch": 5.295276641845703e-05,
      "model_forward_time": 1.723059892654419,
      "step": 55525
    },
    {
      "epoch": 5.295276641845703e-05,
      "step": 55525,
      "training_step_time": 7.7869532108306885
    },
    {
      "epoch": 5.295372009277344e-05,
      "model_forward_time": 1.7203104496002197,
      "step": 55526
    },
    {
      "epoch": 5.295372009277344e-05,
      "step": 55526,
      "training_step_time": 8.411960124969482
    },
    {
      "epoch": 5.2954673767089845e-05,
      "model_forward_time": 1.7191801071166992,
      "step": 55527
    },
    {
      "epoch": 5.2954673767089845e-05,
      "step": 55527,
      "training_step_time": 9.841696739196777
    },
    {
      "epoch": 5.295562744140625e-05,
      "model_forward_time": 1.7135443687438965,
      "step": 55528
    },
    {
      "epoch": 5.295562744140625e-05,
      "step": 55528,
      "training_step_time": 8.722026109695435
    },
    {
      "epoch": 5.295658111572266e-05,
      "model_forward_time": 2.335306167602539,
      "step": 55529
    },
    {
      "epoch": 5.295658111572266e-05,
      "step": 55529,
      "training_step_time": 7.767439365386963
    },
    {
      "epoch": 5.2957534790039064e-05,
      "grad_norm": 0.1107720136642456,
      "learning_rate": 4.500001092528933e-06,
      "loss": 0.0562,
      "step": 55530
    },
    {
      "epoch": 5.2957534790039064e-05,
      "model_forward_time": 1.712712287902832,
      "step": 55530
    },
    {
      "dynamics_loss_avg": 0.04499526796862483,
      "epoch": 5.2957534790039064e-05,
      "step": 55530
    },
    {
      "action_loss_avg": 0.001739272364648059,
      "epoch": 5.2957534790039064e-05,
      "step": 55530
    },
    {
      "epoch": 5.2957534790039064e-05,
      "step": 55530,
      "training_step_time": 7.566356897354126
    },
    {
      "epoch": 5.295848846435547e-05,
      "model_forward_time": 1.7145864963531494,
      "step": 55531
    },
    {
      "epoch": 5.295848846435547e-05,
      "step": 55531,
      "training_step_time": 7.682380676269531
    },
    {
      "epoch": 5.295944213867187e-05,
      "model_forward_time": 1.7100732326507568,
      "step": 55532
    },
    {
      "epoch": 5.295944213867187e-05,
      "step": 55532,
      "training_step_time": 7.424571752548218
    },
    {
      "epoch": 5.296039581298828e-05,
      "model_forward_time": 1.7153337001800537,
      "step": 55533
    },
    {
      "epoch": 5.296039581298828e-05,
      "step": 55533,
      "training_step_time": 9.20955491065979
    },
    {
      "epoch": 5.2961349487304686e-05,
      "model_forward_time": 1.7127048969268799,
      "step": 55534
    },
    {
      "epoch": 5.2961349487304686e-05,
      "step": 55534,
      "training_step_time": 7.668407440185547
    },
    {
      "epoch": 5.296230316162109e-05,
      "model_forward_time": 1.714951515197754,
      "step": 55535
    },
    {
      "epoch": 5.296230316162109e-05,
      "step": 55535,
      "training_step_time": 8.601837158203125
    },
    {
      "epoch": 5.29632568359375e-05,
      "model_forward_time": 1.710533618927002,
      "step": 55536
    },
    {
      "epoch": 5.29632568359375e-05,
      "step": 55536,
      "training_step_time": 8.396698713302612
    },
    {
      "epoch": 5.2964210510253905e-05,
      "model_forward_time": 1.7208333015441895,
      "step": 55537
    },
    {
      "epoch": 5.2964210510253905e-05,
      "step": 55537,
      "training_step_time": 7.582169532775879
    },
    {
      "epoch": 5.296516418457031e-05,
      "model_forward_time": 2.2791454792022705,
      "step": 55538
    },
    {
      "epoch": 5.296516418457031e-05,
      "step": 55538,
      "training_step_time": 8.657627820968628
    },
    {
      "epoch": 5.296611785888672e-05,
      "model_forward_time": 1.7042171955108643,
      "step": 55539
    },
    {
      "epoch": 5.296611785888672e-05,
      "step": 55539,
      "training_step_time": 7.69837212562561
    },
    {
      "epoch": 5.2967071533203124e-05,
      "grad_norm": 0.1499287635087967,
      "learning_rate": 4.498355938165946e-06,
      "loss": 0.0534,
      "step": 55540
    },
    {
      "epoch": 5.2967071533203124e-05,
      "model_forward_time": 1.7123372554779053,
      "step": 55540
    },
    {
      "dynamics_loss_avg": 0.06138562746345997,
      "epoch": 5.2967071533203124e-05,
      "step": 55540
    },
    {
      "action_loss_avg": 0.0017578814236912876,
      "epoch": 5.2967071533203124e-05,
      "step": 55540
    },
    {
      "epoch": 5.2967071533203124e-05,
      "step": 55540,
      "training_step_time": 7.724129915237427
    },
    {
      "epoch": 5.296802520751953e-05,
      "model_forward_time": 1.710052728652954,
      "step": 55541
    },
    {
      "epoch": 5.296802520751953e-05,
      "step": 55541,
      "training_step_time": 7.63437819480896
    },
    {
      "epoch": 5.296897888183594e-05,
      "model_forward_time": 1.7056870460510254,
      "step": 55542
    },
    {
      "epoch": 5.296897888183594e-05,
      "step": 55542,
      "training_step_time": 7.57637882232666
    },
    {
      "epoch": 5.296993255615234e-05,
      "model_forward_time": 1.7180733680725098,
      "step": 55543
    },
    {
      "epoch": 5.296993255615234e-05,
      "step": 55543,
      "training_step_time": 7.612353324890137
    },
    {
      "epoch": 5.297088623046875e-05,
      "model_forward_time": 1.7131092548370361,
      "step": 55544
    },
    {
      "epoch": 5.297088623046875e-05,
      "step": 55544,
      "training_step_time": 7.740328550338745
    },
    {
      "epoch": 5.297183990478516e-05,
      "model_forward_time": 1.7056078910827637,
      "step": 55545
    },
    {
      "epoch": 5.297183990478516e-05,
      "step": 55545,
      "training_step_time": 7.5722291469573975
    },
    {
      "epoch": 5.297279357910156e-05,
      "model_forward_time": 1.7102906703948975,
      "step": 55546
    },
    {
      "epoch": 5.297279357910156e-05,
      "step": 55546,
      "training_step_time": 7.644595384597778
    },
    {
      "epoch": 5.2973747253417966e-05,
      "model_forward_time": 2.2947142124176025,
      "step": 55547
    },
    {
      "epoch": 5.2973747253417966e-05,
      "step": 55547,
      "training_step_time": 7.577028512954712
    },
    {
      "epoch": 5.297470092773438e-05,
      "model_forward_time": 1.7105252742767334,
      "step": 55548
    },
    {
      "epoch": 5.297470092773438e-05,
      "step": 55548,
      "training_step_time": 7.747908592224121
    },
    {
      "epoch": 5.297565460205078e-05,
      "model_forward_time": 1.7136225700378418,
      "step": 55549
    },
    {
      "epoch": 5.297565460205078e-05,
      "step": 55549,
      "training_step_time": 7.644351482391357
    },
    {
      "epoch": 5.2976608276367185e-05,
      "grad_norm": 0.08601127564907074,
      "learning_rate": 4.496710838661999e-06,
      "loss": 0.0531,
      "step": 55550
    },
    {
      "epoch": 5.2976608276367185e-05,
      "model_forward_time": 1.7145445346832275,
      "step": 55550
    },
    {
      "dynamics_loss_avg": 0.05650635352358222,
      "epoch": 5.2976608276367185e-05,
      "step": 55550
    },
    {
      "action_loss_avg": 0.0020911662810249253,
      "epoch": 5.2976608276367185e-05,
      "step": 55550
    },
    {
      "epoch": 5.2976608276367185e-05,
      "step": 55550,
      "training_step_time": 7.681758403778076
    },
    {
      "epoch": 5.2977561950683596e-05,
      "model_forward_time": 1.7234323024749756,
      "step": 55551
    },
    {
      "epoch": 5.2977561950683596e-05,
      "step": 55551,
      "training_step_time": 9.082628965377808
    },
    {
      "epoch": 5.2978515625e-05,
      "model_forward_time": 1.7270307540893555,
      "step": 55552
    },
    {
      "epoch": 5.2978515625e-05,
      "step": 55552,
      "training_step_time": 7.618057489395142
    },
    {
      "epoch": 5.2979469299316404e-05,
      "model_forward_time": 1.7170357704162598,
      "step": 55553
    },
    {
      "epoch": 5.2979469299316404e-05,
      "step": 55553,
      "training_step_time": 7.590994596481323
    },
    {
      "epoch": 5.2980422973632814e-05,
      "model_forward_time": 1.7139649391174316,
      "step": 55554
    },
    {
      "epoch": 5.2980422973632814e-05,
      "step": 55554,
      "training_step_time": 7.940701961517334
    },
    {
      "epoch": 5.298137664794922e-05,
      "model_forward_time": 1.7109367847442627,
      "step": 55555
    },
    {
      "epoch": 5.298137664794922e-05,
      "step": 55555,
      "training_step_time": 8.231054067611694
    },
    {
      "epoch": 5.298233032226562e-05,
      "model_forward_time": 1.7257022857666016,
      "step": 55556
    },
    {
      "epoch": 5.298233032226562e-05,
      "step": 55556,
      "training_step_time": 8.914130687713623
    },
    {
      "epoch": 5.298328399658203e-05,
      "model_forward_time": 1.7859070301055908,
      "step": 55557
    },
    {
      "epoch": 5.298328399658203e-05,
      "step": 55557,
      "training_step_time": 8.556883096694946
    },
    {
      "epoch": 5.298423767089844e-05,
      "model_forward_time": 1.7101781368255615,
      "step": 55558
    },
    {
      "epoch": 5.298423767089844e-05,
      "step": 55558,
      "training_step_time": 7.699025630950928
    },
    {
      "epoch": 5.298519134521484e-05,
      "model_forward_time": 1.7185022830963135,
      "step": 55559
    },
    {
      "epoch": 5.298519134521484e-05,
      "step": 55559,
      "training_step_time": 7.690345048904419
    },
    {
      "epoch": 5.298614501953125e-05,
      "grad_norm": 0.11265027523040771,
      "learning_rate": 4.495065794196999e-06,
      "loss": 0.0565,
      "step": 55560
    },
    {
      "epoch": 5.298614501953125e-05,
      "model_forward_time": 1.3331661224365234,
      "step": 55560
    },
    {
      "dynamics_loss_avg": 0.05657925764098763,
      "epoch": 5.298614501953125e-05,
      "step": 55560
    },
    {
      "action_loss_avg": 0.0019196609093341977,
      "epoch": 5.298614501953125e-05,
      "step": 55560
    },
    {
      "epoch": 5.298614501953125e-05,
      "step": 55560,
      "training_step_time": 7.462013483047485
    },
    {
      "epoch": 5.2987098693847656e-05,
      "model_forward_time": 1.7133338451385498,
      "step": 55561
    },
    {
      "epoch": 5.2987098693847656e-05,
      "step": 55561,
      "training_step_time": 7.65383243560791
    },
    {
      "epoch": 5.298805236816406e-05,
      "model_forward_time": 1.3322584629058838,
      "step": 55562
    },
    {
      "epoch": 5.298805236816406e-05,
      "step": 55562,
      "training_step_time": 8.149794816970825
    },
    {
      "epoch": 5.298900604248047e-05,
      "model_forward_time": 1.7243404388427734,
      "step": 55563
    },
    {
      "epoch": 5.298900604248047e-05,
      "step": 55563,
      "training_step_time": 8.865198135375977
    },
    {
      "epoch": 5.2989959716796875e-05,
      "model_forward_time": 1.7046914100646973,
      "step": 55564
    },
    {
      "epoch": 5.2989959716796875e-05,
      "step": 55564,
      "training_step_time": 8.633057117462158
    },
    {
      "epoch": 5.299091339111328e-05,
      "model_forward_time": 1.7232306003570557,
      "step": 55565
    },
    {
      "epoch": 5.299091339111328e-05,
      "step": 55565,
      "training_step_time": 7.873398065567017
    },
    {
      "epoch": 5.299186706542969e-05,
      "model_forward_time": 1.9268991947174072,
      "step": 55566
    },
    {
      "epoch": 5.299186706542969e-05,
      "step": 55566,
      "training_step_time": 7.7661943435668945
    },
    {
      "epoch": 5.2992820739746094e-05,
      "model_forward_time": 1.7178521156311035,
      "step": 55567
    },
    {
      "epoch": 5.2992820739746094e-05,
      "step": 55567,
      "training_step_time": 8.609363079071045
    },
    {
      "epoch": 5.29937744140625e-05,
      "model_forward_time": 1.6806020736694336,
      "step": 55568
    },
    {
      "epoch": 5.29937744140625e-05,
      "step": 55568,
      "training_step_time": 8.114217758178711
    },
    {
      "epoch": 5.299472808837891e-05,
      "model_forward_time": 1.722245693206787,
      "step": 55569
    },
    {
      "epoch": 5.299472808837891e-05,
      "step": 55569,
      "training_step_time": 7.560882568359375
    },
    {
      "epoch": 5.299568176269531e-05,
      "grad_norm": 0.10174877941608429,
      "learning_rate": 4.493420804950846e-06,
      "loss": 0.0546,
      "step": 55570
    },
    {
      "epoch": 5.299568176269531e-05,
      "model_forward_time": 1.7148704528808594,
      "step": 55570
    },
    {
      "dynamics_loss_avg": 0.04546823054552078,
      "epoch": 5.299568176269531e-05,
      "step": 55570
    },
    {
      "action_loss_avg": 0.0015468537050765008,
      "epoch": 5.299568176269531e-05,
      "step": 55570
    },
    {
      "epoch": 5.299568176269531e-05,
      "step": 55570,
      "training_step_time": 7.684136629104614
    },
    {
      "epoch": 5.299663543701172e-05,
      "model_forward_time": 1.7180697917938232,
      "step": 55571
    },
    {
      "epoch": 5.299663543701172e-05,
      "step": 55571,
      "training_step_time": 7.6397929191589355
    },
    {
      "epoch": 5.299758911132813e-05,
      "model_forward_time": 1.7185978889465332,
      "step": 55572
    },
    {
      "epoch": 5.299758911132813e-05,
      "step": 55572,
      "training_step_time": 7.89331579208374
    },
    {
      "epoch": 5.299854278564453e-05,
      "model_forward_time": 1.7336440086364746,
      "step": 55573
    },
    {
      "epoch": 5.299854278564453e-05,
      "step": 55573,
      "training_step_time": 9.480885744094849
    },
    {
      "epoch": 5.2999496459960936e-05,
      "model_forward_time": 1.3453080654144287,
      "step": 55574
    },
    {
      "epoch": 5.2999496459960936e-05,
      "step": 55574,
      "training_step_time": 7.826400995254517
    },
    {
      "epoch": 5.3000450134277346e-05,
      "model_forward_time": 2.289405584335327,
      "step": 55575
    },
    {
      "epoch": 5.3000450134277346e-05,
      "step": 55575,
      "training_step_time": 8.261956214904785
    },
    {
      "epoch": 5.300140380859375e-05,
      "model_forward_time": 1.7294378280639648,
      "step": 55576
    },
    {
      "epoch": 5.300140380859375e-05,
      "step": 55576,
      "training_step_time": 7.715545892715454
    },
    {
      "epoch": 5.3002357482910155e-05,
      "model_forward_time": 1.7330057621002197,
      "step": 55577
    },
    {
      "epoch": 5.3002357482910155e-05,
      "step": 55577,
      "training_step_time": 7.744828224182129
    },
    {
      "epoch": 5.3003311157226565e-05,
      "model_forward_time": 1.7366182804107666,
      "step": 55578
    },
    {
      "epoch": 5.3003311157226565e-05,
      "step": 55578,
      "training_step_time": 7.6218321323394775
    },
    {
      "epoch": 5.300426483154297e-05,
      "model_forward_time": 1.7272634506225586,
      "step": 55579
    },
    {
      "epoch": 5.300426483154297e-05,
      "step": 55579,
      "training_step_time": 7.736808776855469
    },
    {
      "epoch": 5.300521850585937e-05,
      "grad_norm": 0.0913713350892067,
      "learning_rate": 4.4917758711034305e-06,
      "loss": 0.0524,
      "step": 55580
    },
    {
      "epoch": 5.300521850585937e-05,
      "model_forward_time": 1.7168972492218018,
      "step": 55580
    },
    {
      "dynamics_loss_avg": 0.04650126341730356,
      "epoch": 5.300521850585937e-05,
      "step": 55580
    },
    {
      "action_loss_avg": 0.0013797141873510554,
      "epoch": 5.300521850585937e-05,
      "step": 55580
    },
    {
      "epoch": 5.300521850585937e-05,
      "step": 55580,
      "training_step_time": 8.315899133682251
    },
    {
      "epoch": 5.3006172180175784e-05,
      "model_forward_time": 1.7323524951934814,
      "step": 55581
    },
    {
      "epoch": 5.3006172180175784e-05,
      "step": 55581,
      "training_step_time": 7.8642637729644775
    },
    {
      "epoch": 5.300712585449219e-05,
      "model_forward_time": 1.7314627170562744,
      "step": 55582
    },
    {
      "epoch": 5.300712585449219e-05,
      "step": 55582,
      "training_step_time": 8.494173765182495
    },
    {
      "epoch": 5.300807952880859e-05,
      "model_forward_time": 1.7335164546966553,
      "step": 55583
    },
    {
      "epoch": 5.300807952880859e-05,
      "step": 55583,
      "training_step_time": 7.7641026973724365
    },
    {
      "epoch": 5.3009033203125e-05,
      "model_forward_time": 1.7349789142608643,
      "step": 55584
    },
    {
      "epoch": 5.3009033203125e-05,
      "step": 55584,
      "training_step_time": 8.29926609992981
    },
    {
      "epoch": 5.300998687744141e-05,
      "model_forward_time": 2.2758395671844482,
      "step": 55585
    },
    {
      "epoch": 5.300998687744141e-05,
      "step": 55585,
      "training_step_time": 8.375160455703735
    },
    {
      "epoch": 5.301094055175781e-05,
      "model_forward_time": 1.730053186416626,
      "step": 55586
    },
    {
      "epoch": 5.301094055175781e-05,
      "step": 55586,
      "training_step_time": 7.603666543960571
    },
    {
      "epoch": 5.301189422607422e-05,
      "model_forward_time": 1.7257728576660156,
      "step": 55587
    },
    {
      "epoch": 5.301189422607422e-05,
      "step": 55587,
      "training_step_time": 8.572231769561768
    },
    {
      "epoch": 5.3012847900390626e-05,
      "model_forward_time": 1.7280936241149902,
      "step": 55588
    },
    {
      "epoch": 5.3012847900390626e-05,
      "step": 55588,
      "training_step_time": 7.688408136367798
    },
    {
      "epoch": 5.301380157470703e-05,
      "model_forward_time": 1.7382981777191162,
      "step": 55589
    },
    {
      "epoch": 5.301380157470703e-05,
      "step": 55589,
      "training_step_time": 7.5658674240112305
    },
    {
      "epoch": 5.301475524902344e-05,
      "grad_norm": 0.11264978349208832,
      "learning_rate": 4.490130992834642e-06,
      "loss": 0.0582,
      "step": 55590
    },
    {
      "epoch": 5.301475524902344e-05,
      "model_forward_time": 1.72605562210083,
      "step": 55590
    },
    {
      "dynamics_loss_avg": 0.05871692076325417,
      "epoch": 5.301475524902344e-05,
      "step": 55590
    },
    {
      "action_loss_avg": 0.0015560311876470223,
      "epoch": 5.301475524902344e-05,
      "step": 55590
    },
    {
      "epoch": 5.301475524902344e-05,
      "step": 55590,
      "training_step_time": 8.000128507614136
    },
    {
      "epoch": 5.3015708923339845e-05,
      "model_forward_time": 1.7272961139678955,
      "step": 55591
    },
    {
      "epoch": 5.3015708923339845e-05,
      "step": 55591,
      "training_step_time": 7.622330904006958
    },
    {
      "epoch": 5.301666259765625e-05,
      "model_forward_time": 1.7307686805725098,
      "step": 55592
    },
    {
      "epoch": 5.301666259765625e-05,
      "step": 55592,
      "training_step_time": 9.162576675415039
    },
    {
      "epoch": 5.301761627197266e-05,
      "model_forward_time": 1.72735595703125,
      "step": 55593
    },
    {
      "epoch": 5.301761627197266e-05,
      "step": 55593,
      "training_step_time": 7.643100023269653
    },
    {
      "epoch": 5.3018569946289064e-05,
      "model_forward_time": 2.316873073577881,
      "step": 55594
    },
    {
      "epoch": 5.3018569946289064e-05,
      "step": 55594,
      "training_step_time": 8.786438941955566
    },
    {
      "epoch": 5.301952362060547e-05,
      "model_forward_time": 1.7306597232818604,
      "step": 55595
    },
    {
      "epoch": 5.301952362060547e-05,
      "step": 55595,
      "training_step_time": 8.075900554656982
    },
    {
      "epoch": 5.302047729492187e-05,
      "model_forward_time": 1.7345552444458008,
      "step": 55596
    },
    {
      "epoch": 5.302047729492187e-05,
      "step": 55596,
      "training_step_time": 7.746647357940674
    },
    {
      "epoch": 5.302143096923828e-05,
      "model_forward_time": 1.7203583717346191,
      "step": 55597
    },
    {
      "epoch": 5.302143096923828e-05,
      "step": 55597,
      "training_step_time": 7.674789190292358
    },
    {
      "epoch": 5.3022384643554687e-05,
      "model_forward_time": 1.7221028804779053,
      "step": 55598
    },
    {
      "epoch": 5.3022384643554687e-05,
      "step": 55598,
      "training_step_time": 7.559481382369995
    },
    {
      "epoch": 5.302333831787109e-05,
      "model_forward_time": 1.7160615921020508,
      "step": 55599
    },
    {
      "epoch": 5.302333831787109e-05,
      "step": 55599,
      "training_step_time": 7.9157257080078125
    },
    {
      "epoch": 5.30242919921875e-05,
      "grad_norm": 0.10295887291431427,
      "learning_rate": 4.488486170324359e-06,
      "loss": 0.058,
      "step": 55600
    },
    {
      "epoch": 5.30242919921875e-05,
      "model_forward_time": 1.7208266258239746,
      "step": 55600
    },
    {
      "dynamics_loss_avg": 0.05069848038256168,
      "epoch": 5.30242919921875e-05,
      "step": 55600
    },
    {
      "action_loss_avg": 0.001734859443968162,
      "epoch": 5.30242919921875e-05,
      "step": 55600
    },
    {
      "epoch": 5.30242919921875e-05,
      "step": 55600,
      "training_step_time": 7.575109481811523
    },
    {
      "epoch": 5.3025245666503905e-05,
      "model_forward_time": 1.7193851470947266,
      "step": 55601
    },
    {
      "epoch": 5.3025245666503905e-05,
      "step": 55601,
      "training_step_time": 8.108627080917358
    },
    {
      "epoch": 5.302619934082031e-05,
      "model_forward_time": 1.016115427017212,
      "step": 55602
    },
    {
      "epoch": 5.302619934082031e-05,
      "step": 55602,
      "training_step_time": 8.505512237548828
    },
    {
      "epoch": 5.302715301513672e-05,
      "model_forward_time": 2.313472032546997,
      "step": 55603
    },
    {
      "epoch": 5.302715301513672e-05,
      "step": 55603,
      "training_step_time": 7.876261472702026
    },
    {
      "epoch": 5.3028106689453124e-05,
      "model_forward_time": 1.7205755710601807,
      "step": 55604
    },
    {
      "epoch": 5.3028106689453124e-05,
      "step": 55604,
      "training_step_time": 7.578693389892578
    },
    {
      "epoch": 5.302906036376953e-05,
      "model_forward_time": 1.7219288349151611,
      "step": 55605
    },
    {
      "epoch": 5.302906036376953e-05,
      "step": 55605,
      "training_step_time": 8.591536521911621
    },
    {
      "epoch": 5.303001403808594e-05,
      "model_forward_time": 1.7212324142456055,
      "step": 55606
    },
    {
      "epoch": 5.303001403808594e-05,
      "step": 55606,
      "training_step_time": 7.670070648193359
    },
    {
      "epoch": 5.303096771240234e-05,
      "model_forward_time": 1.7262516021728516,
      "step": 55607
    },
    {
      "epoch": 5.303096771240234e-05,
      "step": 55607,
      "training_step_time": 7.211507081985474
    },
    {
      "epoch": 5.303192138671875e-05,
      "model_forward_time": 1.7219717502593994,
      "step": 55608
    },
    {
      "epoch": 5.303192138671875e-05,
      "step": 55608,
      "training_step_time": 7.803249359130859
    },
    {
      "epoch": 5.303287506103516e-05,
      "model_forward_time": 1.7205145359039307,
      "step": 55609
    },
    {
      "epoch": 5.303287506103516e-05,
      "step": 55609,
      "training_step_time": 7.486307144165039
    },
    {
      "epoch": 5.303382873535156e-05,
      "grad_norm": 0.09730857610702515,
      "learning_rate": 4.486841403752463e-06,
      "loss": 0.0587,
      "step": 55610
    },
    {
      "epoch": 5.303382873535156e-05,
      "model_forward_time": 1.712646722793579,
      "step": 55610
    },
    {
      "dynamics_loss_avg": 0.047820812091231346,
      "epoch": 5.303382873535156e-05,
      "step": 55610
    },
    {
      "action_loss_avg": 0.0008614419173682108,
      "epoch": 5.303382873535156e-05,
      "step": 55610
    },
    {
      "epoch": 5.303382873535156e-05,
      "step": 55610,
      "training_step_time": 9.440341711044312
    },
    {
      "epoch": 5.3034782409667966e-05,
      "model_forward_time": 1.709923267364502,
      "step": 55611
    },
    {
      "epoch": 5.3034782409667966e-05,
      "step": 55611,
      "training_step_time": 7.7366578578948975
    },
    {
      "epoch": 5.303573608398438e-05,
      "model_forward_time": 1.7930974960327148,
      "step": 55612
    },
    {
      "epoch": 5.303573608398438e-05,
      "step": 55612,
      "training_step_time": 9.105997562408447
    },
    {
      "epoch": 5.303668975830078e-05,
      "model_forward_time": 1.7328410148620605,
      "step": 55613
    },
    {
      "epoch": 5.303668975830078e-05,
      "step": 55613,
      "training_step_time": 7.62073540687561
    },
    {
      "epoch": 5.3037643432617185e-05,
      "model_forward_time": 1.7242257595062256,
      "step": 55614
    },
    {
      "epoch": 5.3037643432617185e-05,
      "step": 55614,
      "training_step_time": 8.275899171829224
    },
    {
      "epoch": 5.3038597106933596e-05,
      "model_forward_time": 1.7356071472167969,
      "step": 55615
    },
    {
      "epoch": 5.3038597106933596e-05,
      "step": 55615,
      "training_step_time": 8.773006439208984
    },
    {
      "epoch": 5.303955078125e-05,
      "model_forward_time": 1.723517894744873,
      "step": 55616
    },
    {
      "epoch": 5.303955078125e-05,
      "step": 55616,
      "training_step_time": 7.486046314239502
    },
    {
      "epoch": 5.3040504455566404e-05,
      "model_forward_time": 1.7227840423583984,
      "step": 55617
    },
    {
      "epoch": 5.3040504455566404e-05,
      "step": 55617,
      "training_step_time": 8.548651218414307
    },
    {
      "epoch": 5.3041458129882815e-05,
      "model_forward_time": 1.7191004753112793,
      "step": 55618
    },
    {
      "epoch": 5.3041458129882815e-05,
      "step": 55618,
      "training_step_time": 7.554277181625366
    },
    {
      "epoch": 5.304241180419922e-05,
      "model_forward_time": 1.72196364402771,
      "step": 55619
    },
    {
      "epoch": 5.304241180419922e-05,
      "step": 55619,
      "training_step_time": 7.4859161376953125
    },
    {
      "epoch": 5.304336547851562e-05,
      "grad_norm": 0.14186601340770721,
      "learning_rate": 4.485196693298818e-06,
      "loss": 0.0576,
      "step": 55620
    },
    {
      "epoch": 5.304336547851562e-05,
      "model_forward_time": 1.728050947189331,
      "step": 55620
    },
    {
      "dynamics_loss_avg": 0.060372337140142916,
      "epoch": 5.304336547851562e-05,
      "step": 55620
    },
    {
      "action_loss_avg": 0.0017154564033262432,
      "epoch": 5.304336547851562e-05,
      "step": 55620
    },
    {
      "epoch": 5.304336547851562e-05,
      "step": 55620,
      "training_step_time": 8.75636625289917
    },
    {
      "epoch": 5.304431915283203e-05,
      "model_forward_time": 1.7886919975280762,
      "step": 55621
    },
    {
      "epoch": 5.304431915283203e-05,
      "step": 55621,
      "training_step_time": 10.167356014251709
    },
    {
      "epoch": 5.304527282714844e-05,
      "model_forward_time": 2.2569453716278076,
      "step": 55622
    },
    {
      "epoch": 5.304527282714844e-05,
      "step": 55622,
      "training_step_time": 7.861274719238281
    },
    {
      "epoch": 5.304622650146484e-05,
      "model_forward_time": 1.7175374031066895,
      "step": 55623
    },
    {
      "epoch": 5.304622650146484e-05,
      "step": 55623,
      "training_step_time": 8.7246572971344
    },
    {
      "epoch": 5.304718017578125e-05,
      "model_forward_time": 1.7223539352416992,
      "step": 55624
    },
    {
      "epoch": 5.304718017578125e-05,
      "step": 55624,
      "training_step_time": 7.636923313140869
    },
    {
      "epoch": 5.3048133850097656e-05,
      "model_forward_time": 1.7183969020843506,
      "step": 55625
    },
    {
      "epoch": 5.3048133850097656e-05,
      "step": 55625,
      "training_step_time": 7.617064952850342
    },
    {
      "epoch": 5.304908752441406e-05,
      "model_forward_time": 1.7226002216339111,
      "step": 55626
    },
    {
      "epoch": 5.304908752441406e-05,
      "step": 55626,
      "training_step_time": 7.580910921096802
    },
    {
      "epoch": 5.305004119873047e-05,
      "model_forward_time": 1.7285854816436768,
      "step": 55627
    },
    {
      "epoch": 5.305004119873047e-05,
      "step": 55627,
      "training_step_time": 7.7672600746154785
    },
    {
      "epoch": 5.3050994873046875e-05,
      "model_forward_time": 1.7186079025268555,
      "step": 55628
    },
    {
      "epoch": 5.3050994873046875e-05,
      "step": 55628,
      "training_step_time": 7.5106189250946045
    },
    {
      "epoch": 5.305194854736328e-05,
      "model_forward_time": 1.7206554412841797,
      "step": 55629
    },
    {
      "epoch": 5.305194854736328e-05,
      "step": 55629,
      "training_step_time": 8.477545738220215
    },
    {
      "epoch": 5.305290222167969e-05,
      "grad_norm": 0.10220768302679062,
      "learning_rate": 4.48355203914329e-06,
      "loss": 0.0586,
      "step": 55630
    },
    {
      "epoch": 5.305290222167969e-05,
      "model_forward_time": 1.7905700206756592,
      "step": 55630
    },
    {
      "dynamics_loss_avg": 0.06004284173250198,
      "epoch": 5.305290222167969e-05,
      "step": 55630
    },
    {
      "action_loss_avg": 0.0015959539450705051,
      "epoch": 5.305290222167969e-05,
      "step": 55630
    },
    {
      "epoch": 5.305290222167969e-05,
      "step": 55630,
      "training_step_time": 7.690216064453125
    },
    {
      "epoch": 5.3053855895996094e-05,
      "model_forward_time": 2.2485055923461914,
      "step": 55631
    },
    {
      "epoch": 5.3053855895996094e-05,
      "step": 55631,
      "training_step_time": 8.748339414596558
    },
    {
      "epoch": 5.30548095703125e-05,
      "model_forward_time": 1.7249464988708496,
      "step": 55632
    },
    {
      "epoch": 5.30548095703125e-05,
      "step": 55632,
      "training_step_time": 7.659503698348999
    },
    {
      "epoch": 5.305576324462891e-05,
      "model_forward_time": 1.7263782024383545,
      "step": 55633
    },
    {
      "epoch": 5.305576324462891e-05,
      "step": 55633,
      "training_step_time": 7.6768975257873535
    },
    {
      "epoch": 5.305671691894531e-05,
      "model_forward_time": 1.7266583442687988,
      "step": 55634
    },
    {
      "epoch": 5.305671691894531e-05,
      "step": 55634,
      "training_step_time": 7.777574777603149
    },
    {
      "epoch": 5.305767059326172e-05,
      "model_forward_time": 1.7260017395019531,
      "step": 55635
    },
    {
      "epoch": 5.305767059326172e-05,
      "step": 55635,
      "training_step_time": 7.538572788238525
    },
    {
      "epoch": 5.305862426757813e-05,
      "model_forward_time": 1.7233893871307373,
      "step": 55636
    },
    {
      "epoch": 5.305862426757813e-05,
      "step": 55636,
      "training_step_time": 7.844215631484985
    },
    {
      "epoch": 5.305957794189453e-05,
      "model_forward_time": 1.723031759262085,
      "step": 55637
    },
    {
      "epoch": 5.305957794189453e-05,
      "step": 55637,
      "training_step_time": 7.596890211105347
    },
    {
      "epoch": 5.3060531616210936e-05,
      "model_forward_time": 1.724008560180664,
      "step": 55638
    },
    {
      "epoch": 5.3060531616210936e-05,
      "step": 55638,
      "training_step_time": 8.834625720977783
    },
    {
      "epoch": 5.3061485290527347e-05,
      "model_forward_time": 1.7764718532562256,
      "step": 55639
    },
    {
      "epoch": 5.3061485290527347e-05,
      "step": 55639,
      "training_step_time": 11.05512547492981
    },
    {
      "epoch": 5.306243896484375e-05,
      "grad_norm": 0.09006725251674652,
      "learning_rate": 4.481907441465733e-06,
      "loss": 0.0574,
      "step": 55640
    },
    {
      "epoch": 5.306243896484375e-05,
      "model_forward_time": 2.2510359287261963,
      "step": 55640
    },
    {
      "dynamics_loss_avg": 0.06324273869395255,
      "epoch": 5.306243896484375e-05,
      "step": 55640
    },
    {
      "action_loss_avg": 0.0013677830807864665,
      "epoch": 5.306243896484375e-05,
      "step": 55640
    },
    {
      "epoch": 5.306243896484375e-05,
      "step": 55640,
      "training_step_time": 8.38959813117981
    },
    {
      "epoch": 5.3063392639160155e-05,
      "model_forward_time": 1.720951795578003,
      "step": 55641
    },
    {
      "epoch": 5.3063392639160155e-05,
      "step": 55641,
      "training_step_time": 7.6378960609436035
    },
    {
      "epoch": 5.3064346313476565e-05,
      "model_forward_time": 1.730118751525879,
      "step": 55642
    },
    {
      "epoch": 5.3064346313476565e-05,
      "step": 55642,
      "training_step_time": 7.758633136749268
    },
    {
      "epoch": 5.306529998779297e-05,
      "model_forward_time": 1.346954107284546,
      "step": 55643
    },
    {
      "epoch": 5.306529998779297e-05,
      "step": 55643,
      "training_step_time": 8.45480751991272
    },
    {
      "epoch": 5.3066253662109373e-05,
      "model_forward_time": 1.711024522781372,
      "step": 55644
    },
    {
      "epoch": 5.3066253662109373e-05,
      "step": 55644,
      "training_step_time": 8.13136625289917
    },
    {
      "epoch": 5.3067207336425784e-05,
      "model_forward_time": 1.7216136455535889,
      "step": 55645
    },
    {
      "epoch": 5.3067207336425784e-05,
      "step": 55645,
      "training_step_time": 8.423626899719238
    },
    {
      "epoch": 5.306816101074219e-05,
      "model_forward_time": 1.7117280960083008,
      "step": 55646
    },
    {
      "epoch": 5.306816101074219e-05,
      "step": 55646,
      "training_step_time": 8.5859854221344
    },
    {
      "epoch": 5.306911468505859e-05,
      "model_forward_time": 1.72175931930542,
      "step": 55647
    },
    {
      "epoch": 5.306911468505859e-05,
      "step": 55647,
      "training_step_time": 8.535223960876465
    },
    {
      "epoch": 5.3070068359375e-05,
      "model_forward_time": 1.7944245338439941,
      "step": 55648
    },
    {
      "epoch": 5.3070068359375e-05,
      "step": 55648,
      "training_step_time": 9.320055484771729
    },
    {
      "epoch": 5.307102203369141e-05,
      "model_forward_time": 2.257570266723633,
      "step": 55649
    },
    {
      "epoch": 5.307102203369141e-05,
      "step": 55649,
      "training_step_time": 8.48813796043396
    },
    {
      "epoch": 5.307197570800781e-05,
      "grad_norm": 0.11051616817712784,
      "learning_rate": 4.480262900446e-06,
      "loss": 0.0572,
      "step": 55650
    },
    {
      "epoch": 5.307197570800781e-05,
      "model_forward_time": 1.7175569534301758,
      "step": 55650
    },
    {
      "dynamics_loss_avg": 0.055222836509346965,
      "epoch": 5.307197570800781e-05,
      "step": 55650
    },
    {
      "action_loss_avg": 0.0016760958911618217,
      "epoch": 5.307197570800781e-05,
      "step": 55650
    },
    {
      "epoch": 5.307197570800781e-05,
      "step": 55650,
      "training_step_time": 7.680669069290161
    },
    {
      "epoch": 5.307292938232422e-05,
      "model_forward_time": 1.7293901443481445,
      "step": 55651
    },
    {
      "epoch": 5.307292938232422e-05,
      "step": 55651,
      "training_step_time": 7.696100950241089
    },
    {
      "epoch": 5.3073883056640626e-05,
      "model_forward_time": 1.7257022857666016,
      "step": 55652
    },
    {
      "epoch": 5.3073883056640626e-05,
      "step": 55652,
      "training_step_time": 8.729449987411499
    },
    {
      "epoch": 5.307483673095703e-05,
      "model_forward_time": 1.7177009582519531,
      "step": 55653
    },
    {
      "epoch": 5.307483673095703e-05,
      "step": 55653,
      "training_step_time": 7.5786871910095215
    },
    {
      "epoch": 5.307579040527344e-05,
      "model_forward_time": 1.7270705699920654,
      "step": 55654
    },
    {
      "epoch": 5.307579040527344e-05,
      "step": 55654,
      "training_step_time": 8.405235528945923
    },
    {
      "epoch": 5.3076744079589845e-05,
      "model_forward_time": 1.0057144165039062,
      "step": 55655
    },
    {
      "epoch": 5.3076744079589845e-05,
      "step": 55655,
      "training_step_time": 7.685582876205444
    },
    {
      "epoch": 5.307769775390625e-05,
      "model_forward_time": 1.715486764907837,
      "step": 55656
    },
    {
      "epoch": 5.307769775390625e-05,
      "step": 55656,
      "training_step_time": 7.686977863311768
    },
    {
      "epoch": 5.307865142822265e-05,
      "model_forward_time": 1.786095380783081,
      "step": 55657
    },
    {
      "epoch": 5.307865142822265e-05,
      "step": 55657,
      "training_step_time": 8.794411420822144
    },
    {
      "epoch": 5.3079605102539064e-05,
      "model_forward_time": 2.283595323562622,
      "step": 55658
    },
    {
      "epoch": 5.3079605102539064e-05,
      "step": 55658,
      "training_step_time": 7.80821681022644
    },
    {
      "epoch": 5.308055877685547e-05,
      "model_forward_time": 1.7254345417022705,
      "step": 55659
    },
    {
      "epoch": 5.308055877685547e-05,
      "step": 55659,
      "training_step_time": 8.498748302459717
    },
    {
      "epoch": 5.308151245117187e-05,
      "grad_norm": 0.10356421768665314,
      "learning_rate": 4.478618416263933e-06,
      "loss": 0.0614,
      "step": 55660
    },
    {
      "epoch": 5.308151245117187e-05,
      "model_forward_time": 1.7232639789581299,
      "step": 55660
    },
    {
      "dynamics_loss_avg": 0.05961018707603216,
      "epoch": 5.308151245117187e-05,
      "step": 55660
    },
    {
      "action_loss_avg": 0.0025579089182429017,
      "epoch": 5.308151245117187e-05,
      "step": 55660
    },
    {
      "epoch": 5.308151245117187e-05,
      "step": 55660,
      "training_step_time": 8.389879703521729
    },
    {
      "epoch": 5.308246612548828e-05,
      "model_forward_time": 1.7253108024597168,
      "step": 55661
    },
    {
      "epoch": 5.308246612548828e-05,
      "step": 55661,
      "training_step_time": 7.441823244094849
    },
    {
      "epoch": 5.3083419799804687e-05,
      "model_forward_time": 1.7249865531921387,
      "step": 55662
    },
    {
      "epoch": 5.3083419799804687e-05,
      "step": 55662,
      "training_step_time": 7.626810312271118
    },
    {
      "epoch": 5.308437347412109e-05,
      "model_forward_time": 1.7201275825500488,
      "step": 55663
    },
    {
      "epoch": 5.308437347412109e-05,
      "step": 55663,
      "training_step_time": 8.753921270370483
    },
    {
      "epoch": 5.30853271484375e-05,
      "model_forward_time": 1.7095108032226562,
      "step": 55664
    },
    {
      "epoch": 5.30853271484375e-05,
      "step": 55664,
      "training_step_time": 7.564861059188843
    },
    {
      "epoch": 5.3086280822753905e-05,
      "model_forward_time": 1.705883502960205,
      "step": 55665
    },
    {
      "epoch": 5.3086280822753905e-05,
      "step": 55665,
      "training_step_time": 8.40630030632019
    },
    {
      "epoch": 5.308723449707031e-05,
      "model_forward_time": 1.7165124416351318,
      "step": 55666
    },
    {
      "epoch": 5.308723449707031e-05,
      "step": 55666,
      "training_step_time": 8.644235134124756
    },
    {
      "epoch": 5.308818817138672e-05,
      "model_forward_time": 2.250063896179199,
      "step": 55667
    },
    {
      "epoch": 5.308818817138672e-05,
      "step": 55667,
      "training_step_time": 7.5973052978515625
    },
    {
      "epoch": 5.3089141845703124e-05,
      "model_forward_time": 1.718122959136963,
      "step": 55668
    },
    {
      "epoch": 5.3089141845703124e-05,
      "step": 55668,
      "training_step_time": 8.543844223022461
    },
    {
      "epoch": 5.309009552001953e-05,
      "model_forward_time": 1.7294151782989502,
      "step": 55669
    },
    {
      "epoch": 5.309009552001953e-05,
      "step": 55669,
      "training_step_time": 7.665279150009155
    },
    {
      "epoch": 5.309104919433594e-05,
      "grad_norm": 0.09462182968854904,
      "learning_rate": 4.476973989099373e-06,
      "loss": 0.0577,
      "step": 55670
    },
    {
      "epoch": 5.309104919433594e-05,
      "model_forward_time": 1.712188720703125,
      "step": 55670
    },
    {
      "dynamics_loss_avg": 0.04580057170242071,
      "epoch": 5.309104919433594e-05,
      "step": 55670
    },
    {
      "action_loss_avg": 0.0018788428191328422,
      "epoch": 5.309104919433594e-05,
      "step": 55670
    },
    {
      "epoch": 5.309104919433594e-05,
      "step": 55670,
      "training_step_time": 7.422295093536377
    },
    {
      "epoch": 5.309200286865234e-05,
      "model_forward_time": 1.7165307998657227,
      "step": 55671
    },
    {
      "epoch": 5.309200286865234e-05,
      "step": 55671,
      "training_step_time": 7.767664670944214
    },
    {
      "epoch": 5.309295654296875e-05,
      "model_forward_time": 1.7323698997497559,
      "step": 55672
    },
    {
      "epoch": 5.309295654296875e-05,
      "step": 55672,
      "training_step_time": 7.664761781692505
    },
    {
      "epoch": 5.309391021728516e-05,
      "model_forward_time": 1.715348720550537,
      "step": 55673
    },
    {
      "epoch": 5.309391021728516e-05,
      "step": 55673,
      "training_step_time": 8.485539674758911
    },
    {
      "epoch": 5.309486389160156e-05,
      "model_forward_time": 1.7160167694091797,
      "step": 55674
    },
    {
      "epoch": 5.309486389160156e-05,
      "step": 55674,
      "training_step_time": 8.813538312911987
    },
    {
      "epoch": 5.3095817565917966e-05,
      "model_forward_time": 1.7651526927947998,
      "step": 55675
    },
    {
      "epoch": 5.3095817565917966e-05,
      "step": 55675,
      "training_step_time": 9.714853048324585
    },
    {
      "epoch": 5.309677124023438e-05,
      "model_forward_time": 1.741095781326294,
      "step": 55676
    },
    {
      "epoch": 5.309677124023438e-05,
      "step": 55676,
      "training_step_time": 8.224151849746704
    },
    {
      "epoch": 5.309772491455078e-05,
      "model_forward_time": 1.7271552085876465,
      "step": 55677
    },
    {
      "epoch": 5.309772491455078e-05,
      "step": 55677,
      "training_step_time": 7.811446905136108
    },
    {
      "epoch": 5.3098678588867185e-05,
      "model_forward_time": 1.7149877548217773,
      "step": 55678
    },
    {
      "epoch": 5.3098678588867185e-05,
      "step": 55678,
      "training_step_time": 8.809642791748047
    },
    {
      "epoch": 5.3099632263183596e-05,
      "model_forward_time": 1.7318086624145508,
      "step": 55679
    },
    {
      "epoch": 5.3099632263183596e-05,
      "step": 55679,
      "training_step_time": 7.873522520065308
    },
    {
      "epoch": 5.31005859375e-05,
      "grad_norm": 0.09485644102096558,
      "learning_rate": 4.4753296191321525e-06,
      "loss": 0.0549,
      "step": 55680
    },
    {
      "epoch": 5.31005859375e-05,
      "model_forward_time": 1.7259163856506348,
      "step": 55680
    },
    {
      "dynamics_loss_avg": 0.05809901542961597,
      "epoch": 5.31005859375e-05,
      "step": 55680
    },
    {
      "action_loss_avg": 0.0012905238050734623,
      "epoch": 5.31005859375e-05,
      "step": 55680
    },
    {
      "epoch": 5.31005859375e-05,
      "step": 55680,
      "training_step_time": 7.799391508102417
    },
    {
      "epoch": 5.3101539611816404e-05,
      "model_forward_time": 1.742523193359375,
      "step": 55681
    },
    {
      "epoch": 5.3101539611816404e-05,
      "step": 55681,
      "training_step_time": 8.230406999588013
    },
    {
      "epoch": 5.3102493286132815e-05,
      "model_forward_time": 1.7370264530181885,
      "step": 55682
    },
    {
      "epoch": 5.3102493286132815e-05,
      "step": 55682,
      "training_step_time": 8.508336067199707
    },
    {
      "epoch": 5.310344696044922e-05,
      "model_forward_time": 1.7361156940460205,
      "step": 55683
    },
    {
      "epoch": 5.310344696044922e-05,
      "step": 55683,
      "training_step_time": 7.690928936004639
    },
    {
      "epoch": 5.310440063476562e-05,
      "model_forward_time": 1.7607715129852295,
      "step": 55684
    },
    {
      "epoch": 5.310440063476562e-05,
      "step": 55684,
      "training_step_time": 8.736813068389893
    },
    {
      "epoch": 5.3105354309082033e-05,
      "model_forward_time": 1.7339797019958496,
      "step": 55685
    },
    {
      "epoch": 5.3105354309082033e-05,
      "step": 55685,
      "training_step_time": 7.835802793502808
    },
    {
      "epoch": 5.310630798339844e-05,
      "model_forward_time": 2.307440996170044,
      "step": 55686
    },
    {
      "epoch": 5.310630798339844e-05,
      "step": 55686,
      "training_step_time": 7.821111679077148
    },
    {
      "epoch": 5.310726165771484e-05,
      "model_forward_time": 1.731757402420044,
      "step": 55687
    },
    {
      "epoch": 5.310726165771484e-05,
      "step": 55687,
      "training_step_time": 7.70995306968689
    },
    {
      "epoch": 5.310821533203125e-05,
      "model_forward_time": 1.7334094047546387,
      "step": 55688
    },
    {
      "epoch": 5.310821533203125e-05,
      "step": 55688,
      "training_step_time": 8.820926427841187
    },
    {
      "epoch": 5.3109169006347656e-05,
      "model_forward_time": 1.7002534866333008,
      "step": 55689
    },
    {
      "epoch": 5.3109169006347656e-05,
      "step": 55689,
      "training_step_time": 7.825193166732788
    },
    {
      "epoch": 5.311012268066406e-05,
      "grad_norm": 0.09475536644458771,
      "learning_rate": 4.473685306542096e-06,
      "loss": 0.0559,
      "step": 55690
    },
    {
      "epoch": 5.311012268066406e-05,
      "model_forward_time": 1.7139475345611572,
      "step": 55690
    },
    {
      "dynamics_loss_avg": 0.060402679815888406,
      "epoch": 5.311012268066406e-05,
      "step": 55690
    },
    {
      "action_loss_avg": 0.0010986594214045908,
      "epoch": 5.311012268066406e-05,
      "step": 55690
    },
    {
      "epoch": 5.311012268066406e-05,
      "step": 55690,
      "training_step_time": 7.6889472007751465
    },
    {
      "epoch": 5.311107635498047e-05,
      "model_forward_time": 1.7183482646942139,
      "step": 55691
    },
    {
      "epoch": 5.311107635498047e-05,
      "step": 55691,
      "training_step_time": 7.646995306015015
    },
    {
      "epoch": 5.3112030029296875e-05,
      "model_forward_time": 1.7162539958953857,
      "step": 55692
    },
    {
      "epoch": 5.3112030029296875e-05,
      "step": 55692,
      "training_step_time": 7.6677165031433105
    },
    {
      "epoch": 5.311298370361328e-05,
      "model_forward_time": 1.7258718013763428,
      "step": 55693
    },
    {
      "epoch": 5.311298370361328e-05,
      "step": 55693,
      "training_step_time": 8.655458927154541
    },
    {
      "epoch": 5.311393737792969e-05,
      "model_forward_time": 1.723170518875122,
      "step": 55694
    },
    {
      "epoch": 5.311393737792969e-05,
      "step": 55694,
      "training_step_time": 8.770162105560303
    },
    {
      "epoch": 5.3114891052246094e-05,
      "model_forward_time": 2.251147508621216,
      "step": 55695
    },
    {
      "epoch": 5.3114891052246094e-05,
      "step": 55695,
      "training_step_time": 7.731333017349243
    },
    {
      "epoch": 5.31158447265625e-05,
      "model_forward_time": 1.7232279777526855,
      "step": 55696
    },
    {
      "epoch": 5.31158447265625e-05,
      "step": 55696,
      "training_step_time": 8.326019287109375
    },
    {
      "epoch": 5.311679840087891e-05,
      "model_forward_time": 1.716071605682373,
      "step": 55697
    },
    {
      "epoch": 5.311679840087891e-05,
      "step": 55697,
      "training_step_time": 7.7336859703063965
    },
    {
      "epoch": 5.311775207519531e-05,
      "model_forward_time": 1.713914394378662,
      "step": 55698
    },
    {
      "epoch": 5.311775207519531e-05,
      "step": 55698,
      "training_step_time": 7.5614142417907715
    },
    {
      "epoch": 5.311870574951172e-05,
      "model_forward_time": 1.7222504615783691,
      "step": 55699
    },
    {
      "epoch": 5.311870574951172e-05,
      "step": 55699,
      "training_step_time": 7.654989242553711
    },
    {
      "epoch": 5.311965942382813e-05,
      "grad_norm": 0.10700567811727524,
      "learning_rate": 4.472041051509023e-06,
      "loss": 0.0563,
      "step": 55700
    },
    {
      "epoch": 5.311965942382813e-05,
      "model_forward_time": 1.7290196418762207,
      "step": 55700
    },
    {
      "dynamics_loss_avg": 0.06984496563673019,
      "epoch": 5.311965942382813e-05,
      "step": 55700
    },
    {
      "action_loss_avg": 0.002003359980881214,
      "epoch": 5.311965942382813e-05,
      "step": 55700
    },
    {
      "epoch": 5.311965942382813e-05,
      "step": 55700,
      "training_step_time": 8.411072492599487
    },
    {
      "epoch": 5.312061309814453e-05,
      "model_forward_time": 1.721815586090088,
      "step": 55701
    },
    {
      "epoch": 5.312061309814453e-05,
      "step": 55701,
      "training_step_time": 8.4496328830719
    },
    {
      "epoch": 5.3121566772460936e-05,
      "model_forward_time": 1.721193790435791,
      "step": 55702
    },
    {
      "epoch": 5.3121566772460936e-05,
      "step": 55702,
      "training_step_time": 7.696745157241821
    },
    {
      "epoch": 5.3122520446777347e-05,
      "model_forward_time": 1.7213711738586426,
      "step": 55703
    },
    {
      "epoch": 5.3122520446777347e-05,
      "step": 55703,
      "training_step_time": 8.062639951705933
    },
    {
      "epoch": 5.312347412109375e-05,
      "model_forward_time": 2.2621121406555176,
      "step": 55704
    },
    {
      "epoch": 5.312347412109375e-05,
      "step": 55704,
      "training_step_time": 7.7589640617370605
    },
    {
      "epoch": 5.3124427795410155e-05,
      "model_forward_time": 2.804828643798828,
      "step": 55705
    },
    {
      "epoch": 5.3124427795410155e-05,
      "step": 55705,
      "training_step_time": 9.764648199081421
    },
    {
      "epoch": 5.3125381469726565e-05,
      "model_forward_time": 1.7167730331420898,
      "step": 55706
    },
    {
      "epoch": 5.3125381469726565e-05,
      "step": 55706,
      "training_step_time": 6.8565144538879395
    },
    {
      "epoch": 5.312633514404297e-05,
      "model_forward_time": 1.7244923114776611,
      "step": 55707
    },
    {
      "epoch": 5.312633514404297e-05,
      "step": 55707,
      "training_step_time": 6.817963361740112
    },
    {
      "epoch": 5.3127288818359373e-05,
      "model_forward_time": 1.6747708320617676,
      "step": 55708
    },
    {
      "epoch": 5.3127288818359373e-05,
      "step": 55708,
      "training_step_time": 6.820443153381348
    },
    {
      "epoch": 5.3128242492675784e-05,
      "model_forward_time": 1.682706356048584,
      "step": 55709
    },
    {
      "epoch": 5.3128242492675784e-05,
      "step": 55709,
      "training_step_time": 6.868695259094238
    },
    {
      "epoch": 5.312919616699219e-05,
      "grad_norm": 0.08851448446512222,
      "learning_rate": 4.470396854212746e-06,
      "loss": 0.0553,
      "step": 55710
    },
    {
      "epoch": 5.312919616699219e-05,
      "model_forward_time": 1.68660569190979,
      "step": 55710
    },
    {
      "dynamics_loss_avg": 0.06269725579768419,
      "epoch": 5.312919616699219e-05,
      "step": 55710
    },
    {
      "action_loss_avg": 0.001628712029196322,
      "epoch": 5.312919616699219e-05,
      "step": 55710
    },
    {
      "epoch": 5.312919616699219e-05,
      "step": 55710,
      "training_step_time": 6.869562864303589
    },
    {
      "epoch": 5.313014984130859e-05,
      "model_forward_time": 1.6991357803344727,
      "step": 55711
    },
    {
      "epoch": 5.313014984130859e-05,
      "step": 55711,
      "training_step_time": 6.858697414398193
    },
    {
      "epoch": 5.3131103515625e-05,
      "model_forward_time": 1.3278768062591553,
      "step": 55712
    },
    {
      "epoch": 5.3131103515625e-05,
      "step": 55712,
      "training_step_time": 6.899455308914185
    },
    {
      "epoch": 5.313205718994141e-05,
      "model_forward_time": 1.7043845653533936,
      "step": 55713
    },
    {
      "epoch": 5.313205718994141e-05,
      "step": 55713,
      "training_step_time": 7.7823874950408936
    },
    {
      "epoch": 5.313301086425781e-05,
      "model_forward_time": 2.2597317695617676,
      "step": 55714
    },
    {
      "epoch": 5.313301086425781e-05,
      "step": 55714,
      "training_step_time": 11.225533246994019
    },
    {
      "epoch": 5.313396453857422e-05,
      "model_forward_time": 1.696641445159912,
      "step": 55715
    },
    {
      "epoch": 5.313396453857422e-05,
      "step": 55715,
      "training_step_time": 7.967269420623779
    },
    {
      "epoch": 5.3134918212890626e-05,
      "model_forward_time": 1.6954755783081055,
      "step": 55716
    },
    {
      "epoch": 5.3134918212890626e-05,
      "step": 55716,
      "training_step_time": 6.785425186157227
    },
    {
      "epoch": 5.313587188720703e-05,
      "model_forward_time": 1.707169771194458,
      "step": 55717
    },
    {
      "epoch": 5.313587188720703e-05,
      "step": 55717,
      "training_step_time": 6.934551954269409
    },
    {
      "epoch": 5.313682556152344e-05,
      "model_forward_time": 1.7106363773345947,
      "step": 55718
    },
    {
      "epoch": 5.313682556152344e-05,
      "step": 55718,
      "training_step_time": 6.822147607803345
    },
    {
      "epoch": 5.3137779235839845e-05,
      "model_forward_time": 1.716614007949829,
      "step": 55719
    },
    {
      "epoch": 5.3137779235839845e-05,
      "step": 55719,
      "training_step_time": 6.796102523803711
    },
    {
      "epoch": 5.313873291015625e-05,
      "grad_norm": 0.07970862835645676,
      "learning_rate": 4.468752714833073e-06,
      "loss": 0.0564,
      "step": 55720
    },
    {
      "epoch": 5.313873291015625e-05,
      "model_forward_time": 1.7343683242797852,
      "step": 55720
    },
    {
      "dynamics_loss_avg": 0.06336250528693199,
      "epoch": 5.313873291015625e-05,
      "step": 55720
    },
    {
      "action_loss_avg": 0.0017558721970999613,
      "epoch": 5.313873291015625e-05,
      "step": 55720
    },
    {
      "epoch": 5.313873291015625e-05,
      "step": 55720,
      "training_step_time": 6.871536016464233
    },
    {
      "epoch": 5.313968658447265e-05,
      "model_forward_time": 1.7162184715270996,
      "step": 55721
    },
    {
      "epoch": 5.313968658447265e-05,
      "step": 55721,
      "training_step_time": 6.9055845737457275
    },
    {
      "epoch": 5.3140640258789064e-05,
      "model_forward_time": 1.349353551864624,
      "step": 55722
    },
    {
      "epoch": 5.3140640258789064e-05,
      "step": 55722,
      "training_step_time": 6.873709201812744
    },
    {
      "epoch": 5.314159393310547e-05,
      "model_forward_time": 2.3126673698425293,
      "step": 55723
    },
    {
      "epoch": 5.314159393310547e-05,
      "step": 55723,
      "training_step_time": 8.589338541030884
    },
    {
      "epoch": 5.314254760742187e-05,
      "model_forward_time": 1.714860200881958,
      "step": 55724
    },
    {
      "epoch": 5.314254760742187e-05,
      "step": 55724,
      "training_step_time": 8.011005878448486
    },
    {
      "epoch": 5.314350128173828e-05,
      "model_forward_time": 1.718338966369629,
      "step": 55725
    },
    {
      "epoch": 5.314350128173828e-05,
      "step": 55725,
      "training_step_time": 6.868658542633057
    },
    {
      "epoch": 5.314445495605469e-05,
      "model_forward_time": 1.7213082313537598,
      "step": 55726
    },
    {
      "epoch": 5.314445495605469e-05,
      "step": 55726,
      "training_step_time": 6.851678371429443
    },
    {
      "epoch": 5.314540863037109e-05,
      "model_forward_time": 1.7329771518707275,
      "step": 55727
    },
    {
      "epoch": 5.314540863037109e-05,
      "step": 55727,
      "training_step_time": 6.87561297416687
    },
    {
      "epoch": 5.31463623046875e-05,
      "model_forward_time": 1.716264247894287,
      "step": 55728
    },
    {
      "epoch": 5.31463623046875e-05,
      "step": 55728,
      "training_step_time": 6.95524525642395
    },
    {
      "epoch": 5.3147315979003905e-05,
      "model_forward_time": 1.7177424430847168,
      "step": 55729
    },
    {
      "epoch": 5.3147315979003905e-05,
      "step": 55729,
      "training_step_time": 6.844848871231079
    },
    {
      "epoch": 5.314826965332031e-05,
      "grad_norm": 0.10688824951648712,
      "learning_rate": 4.467108633549803e-06,
      "loss": 0.055,
      "step": 55730
    },
    {
      "epoch": 5.314826965332031e-05,
      "model_forward_time": 1.330141305923462,
      "step": 55730
    },
    {
      "dynamics_loss_avg": 0.053164546750485896,
      "epoch": 5.314826965332031e-05,
      "step": 55730
    },
    {
      "action_loss_avg": 0.001302381650020834,
      "epoch": 5.314826965332031e-05,
      "step": 55730
    },
    {
      "epoch": 5.314826965332031e-05,
      "step": 55730,
      "training_step_time": 6.8810248374938965
    },
    {
      "epoch": 5.314922332763672e-05,
      "model_forward_time": 1.7188615798950195,
      "step": 55731
    },
    {
      "epoch": 5.314922332763672e-05,
      "step": 55731,
      "training_step_time": 6.979429006576538
    },
    {
      "epoch": 5.3150177001953124e-05,
      "model_forward_time": 2.313840866088867,
      "step": 55732
    },
    {
      "epoch": 5.3150177001953124e-05,
      "step": 55732,
      "training_step_time": 8.487733364105225
    },
    {
      "epoch": 5.315113067626953e-05,
      "model_forward_time": 1.7160699367523193,
      "step": 55733
    },
    {
      "epoch": 5.315113067626953e-05,
      "step": 55733,
      "training_step_time": 8.507277011871338
    },
    {
      "epoch": 5.315208435058594e-05,
      "model_forward_time": 1.7051234245300293,
      "step": 55734
    },
    {
      "epoch": 5.315208435058594e-05,
      "step": 55734,
      "training_step_time": 6.887774229049683
    },
    {
      "epoch": 5.315303802490234e-05,
      "model_forward_time": 1.7080888748168945,
      "step": 55735
    },
    {
      "epoch": 5.315303802490234e-05,
      "step": 55735,
      "training_step_time": 6.939241409301758
    },
    {
      "epoch": 5.315399169921875e-05,
      "model_forward_time": 1.7225697040557861,
      "step": 55736
    },
    {
      "epoch": 5.315399169921875e-05,
      "step": 55736,
      "training_step_time": 6.962638854980469
    },
    {
      "epoch": 5.315494537353516e-05,
      "model_forward_time": 1.3430259227752686,
      "step": 55737
    },
    {
      "epoch": 5.315494537353516e-05,
      "step": 55737,
      "training_step_time": 6.885351657867432
    },
    {
      "epoch": 5.315589904785156e-05,
      "model_forward_time": 1.7115201950073242,
      "step": 55738
    },
    {
      "epoch": 5.315589904785156e-05,
      "step": 55738,
      "training_step_time": 6.856031179428101
    },
    {
      "epoch": 5.3156852722167966e-05,
      "model_forward_time": 1.7214388847351074,
      "step": 55739
    },
    {
      "epoch": 5.3156852722167966e-05,
      "step": 55739,
      "training_step_time": 6.9644598960876465
    },
    {
      "epoch": 5.315780639648438e-05,
      "grad_norm": 0.09549711644649506,
      "learning_rate": 4.465464610542734e-06,
      "loss": 0.0536,
      "step": 55740
    },
    {
      "epoch": 5.315780639648438e-05,
      "model_forward_time": 1.7281455993652344,
      "step": 55740
    },
    {
      "dynamics_loss_avg": 0.04422744922339916,
      "epoch": 5.315780639648438e-05,
      "step": 55740
    },
    {
      "action_loss_avg": 0.0013231994991656393,
      "epoch": 5.315780639648438e-05,
      "step": 55740
    },
    {
      "epoch": 5.315780639648438e-05,
      "step": 55740,
      "training_step_time": 6.8567187786102295
    },
    {
      "epoch": 5.315876007080078e-05,
      "model_forward_time": 1.3534064292907715,
      "step": 55741
    },
    {
      "epoch": 5.315876007080078e-05,
      "step": 55741,
      "training_step_time": 7.162269353866577
    },
    {
      "epoch": 5.3159713745117185e-05,
      "model_forward_time": 1.7583105564117432,
      "step": 55742
    },
    {
      "epoch": 5.3159713745117185e-05,
      "step": 55742,
      "training_step_time": 9.812002897262573
    },
    {
      "epoch": 5.3160667419433596e-05,
      "model_forward_time": 1.715362787246704,
      "step": 55743
    },
    {
      "epoch": 5.3160667419433596e-05,
      "step": 55743,
      "training_step_time": 7.96259069442749
    },
    {
      "epoch": 5.316162109375e-05,
      "model_forward_time": 1.710618257522583,
      "step": 55744
    },
    {
      "epoch": 5.316162109375e-05,
      "step": 55744,
      "training_step_time": 6.83473539352417
    },
    {
      "epoch": 5.3162574768066404e-05,
      "model_forward_time": 1.7145123481750488,
      "step": 55745
    },
    {
      "epoch": 5.3162574768066404e-05,
      "step": 55745,
      "training_step_time": 6.84647798538208
    },
    {
      "epoch": 5.3163528442382815e-05,
      "model_forward_time": 1.7227168083190918,
      "step": 55746
    },
    {
      "epoch": 5.3163528442382815e-05,
      "step": 55746,
      "training_step_time": 6.756574392318726
    },
    {
      "epoch": 5.316448211669922e-05,
      "model_forward_time": 1.7170255184173584,
      "step": 55747
    },
    {
      "epoch": 5.316448211669922e-05,
      "step": 55747,
      "training_step_time": 6.776069164276123
    },
    {
      "epoch": 5.316543579101562e-05,
      "model_forward_time": 1.7124311923980713,
      "step": 55748
    },
    {
      "epoch": 5.316543579101562e-05,
      "step": 55748,
      "training_step_time": 6.823575973510742
    },
    {
      "epoch": 5.3166389465332033e-05,
      "model_forward_time": 1.7169058322906494,
      "step": 55749
    },
    {
      "epoch": 5.3166389465332033e-05,
      "step": 55749,
      "training_step_time": 6.813791036605835
    },
    {
      "epoch": 5.316734313964844e-05,
      "grad_norm": 0.15132606029510498,
      "learning_rate": 4.463820645991651e-06,
      "loss": 0.0575,
      "step": 55750
    },
    {
      "epoch": 5.316734313964844e-05,
      "model_forward_time": 1.6988494396209717,
      "step": 55750
    },
    {
      "dynamics_loss_avg": 0.05553956013172865,
      "epoch": 5.316734313964844e-05,
      "step": 55750
    },
    {
      "action_loss_avg": 0.0017916820477694273,
      "epoch": 5.316734313964844e-05,
      "step": 55750
    },
    {
      "epoch": 5.316734313964844e-05,
      "step": 55750,
      "training_step_time": 7.095401287078857
    },
    {
      "epoch": 5.316829681396484e-05,
      "model_forward_time": 2.329735279083252,
      "step": 55751
    },
    {
      "epoch": 5.316829681396484e-05,
      "step": 55751,
      "training_step_time": 9.384576082229614
    },
    {
      "epoch": 5.316925048828125e-05,
      "model_forward_time": 1.7175226211547852,
      "step": 55752
    },
    {
      "epoch": 5.316925048828125e-05,
      "step": 55752,
      "training_step_time": 7.9379682540893555
    },
    {
      "epoch": 5.3170204162597656e-05,
      "model_forward_time": 1.7219250202178955,
      "step": 55753
    },
    {
      "epoch": 5.3170204162597656e-05,
      "step": 55753,
      "training_step_time": 6.789899587631226
    },
    {
      "epoch": 5.317115783691406e-05,
      "model_forward_time": 1.7120587825775146,
      "step": 55754
    },
    {
      "epoch": 5.317115783691406e-05,
      "step": 55754,
      "training_step_time": 6.906307220458984
    },
    {
      "epoch": 5.317211151123047e-05,
      "model_forward_time": 1.7106211185455322,
      "step": 55755
    },
    {
      "epoch": 5.317211151123047e-05,
      "step": 55755,
      "training_step_time": 6.833695650100708
    },
    {
      "epoch": 5.3173065185546875e-05,
      "model_forward_time": 1.7115445137023926,
      "step": 55756
    },
    {
      "epoch": 5.3173065185546875e-05,
      "step": 55756,
      "training_step_time": 6.817288160324097
    },
    {
      "epoch": 5.317401885986328e-05,
      "model_forward_time": 1.7488019466400146,
      "step": 55757
    },
    {
      "epoch": 5.317401885986328e-05,
      "step": 55757,
      "training_step_time": 6.816670179367065
    },
    {
      "epoch": 5.317497253417969e-05,
      "model_forward_time": 1.346092939376831,
      "step": 55758
    },
    {
      "epoch": 5.317497253417969e-05,
      "step": 55758,
      "training_step_time": 6.852371692657471
    },
    {
      "epoch": 5.3175926208496094e-05,
      "model_forward_time": 1.717027187347412,
      "step": 55759
    },
    {
      "epoch": 5.3175926208496094e-05,
      "step": 55759,
      "training_step_time": 7.012828826904297
    },
    {
      "epoch": 5.31768798828125e-05,
      "grad_norm": 0.10971467196941376,
      "learning_rate": 4.462176740076335e-06,
      "loss": 0.0602,
      "step": 55760
    },
    {
      "epoch": 5.31768798828125e-05,
      "model_forward_time": 2.2857728004455566,
      "step": 55760
    },
    {
      "dynamics_loss_avg": 0.056999125704169276,
      "epoch": 5.31768798828125e-05,
      "step": 55760
    },
    {
      "action_loss_avg": 0.001151698242756538,
      "epoch": 5.31768798828125e-05,
      "step": 55760
    },
    {
      "epoch": 5.31768798828125e-05,
      "step": 55760,
      "training_step_time": 8.439505577087402
    },
    {
      "epoch": 5.317783355712891e-05,
      "model_forward_time": 1.714637041091919,
      "step": 55761
    },
    {
      "epoch": 5.317783355712891e-05,
      "step": 55761,
      "training_step_time": 8.648542404174805
    },
    {
      "epoch": 5.317878723144531e-05,
      "model_forward_time": 1.3337576389312744,
      "step": 55762
    },
    {
      "epoch": 5.317878723144531e-05,
      "step": 55762,
      "training_step_time": 7.070292711257935
    },
    {
      "epoch": 5.317974090576172e-05,
      "model_forward_time": 1.7144420146942139,
      "step": 55763
    },
    {
      "epoch": 5.317974090576172e-05,
      "step": 55763,
      "training_step_time": 6.8117499351501465
    },
    {
      "epoch": 5.318069458007813e-05,
      "model_forward_time": 1.722712516784668,
      "step": 55764
    },
    {
      "epoch": 5.318069458007813e-05,
      "step": 55764,
      "training_step_time": 6.858989000320435
    },
    {
      "epoch": 5.318164825439453e-05,
      "model_forward_time": 1.714561939239502,
      "step": 55765
    },
    {
      "epoch": 5.318164825439453e-05,
      "step": 55765,
      "training_step_time": 6.892503976821899
    },
    {
      "epoch": 5.3182601928710936e-05,
      "model_forward_time": 1.7183332443237305,
      "step": 55766
    },
    {
      "epoch": 5.3182601928710936e-05,
      "step": 55766,
      "training_step_time": 6.845346212387085
    },
    {
      "epoch": 5.318355560302735e-05,
      "model_forward_time": 1.712158203125,
      "step": 55767
    },
    {
      "epoch": 5.318355560302735e-05,
      "step": 55767,
      "training_step_time": 6.765399694442749
    },
    {
      "epoch": 5.318450927734375e-05,
      "model_forward_time": 1.7127289772033691,
      "step": 55768
    },
    {
      "epoch": 5.318450927734375e-05,
      "step": 55768,
      "training_step_time": 6.957108497619629
    },
    {
      "epoch": 5.3185462951660155e-05,
      "model_forward_time": 2.254594564437866,
      "step": 55769
    },
    {
      "epoch": 5.3185462951660155e-05,
      "step": 55769,
      "training_step_time": 9.679034948348999
    },
    {
      "epoch": 5.3186416625976565e-05,
      "grad_norm": 0.12666794657707214,
      "learning_rate": 4.460532892976562e-06,
      "loss": 0.056,
      "step": 55770
    },
    {
      "epoch": 5.3186416625976565e-05,
      "model_forward_time": 1.7603974342346191,
      "step": 55770
    },
    {
      "dynamics_loss_avg": 0.05914129707962275,
      "epoch": 5.3186416625976565e-05,
      "step": 55770
    },
    {
      "action_loss_avg": 0.0015628515218850225,
      "epoch": 5.3186416625976565e-05,
      "step": 55770
    },
    {
      "epoch": 5.3186416625976565e-05,
      "step": 55770,
      "training_step_time": 8.056796073913574
    },
    {
      "epoch": 5.318737030029297e-05,
      "model_forward_time": 1.7178599834442139,
      "step": 55771
    },
    {
      "epoch": 5.318737030029297e-05,
      "step": 55771,
      "training_step_time": 7.590068101882935
    },
    {
      "epoch": 5.3188323974609374e-05,
      "model_forward_time": 1.7404415607452393,
      "step": 55772
    },
    {
      "epoch": 5.3188323974609374e-05,
      "step": 55772,
      "training_step_time": 6.821953535079956
    },
    {
      "epoch": 5.3189277648925784e-05,
      "model_forward_time": 1.7289643287658691,
      "step": 55773
    },
    {
      "epoch": 5.3189277648925784e-05,
      "step": 55773,
      "training_step_time": 6.944270133972168
    },
    {
      "epoch": 5.319023132324219e-05,
      "model_forward_time": 1.7290372848510742,
      "step": 55774
    },
    {
      "epoch": 5.319023132324219e-05,
      "step": 55774,
      "training_step_time": 6.866148948669434
    },
    {
      "epoch": 5.319118499755859e-05,
      "model_forward_time": 1.7251324653625488,
      "step": 55775
    },
    {
      "epoch": 5.319118499755859e-05,
      "step": 55775,
      "training_step_time": 6.852832794189453
    },
    {
      "epoch": 5.3192138671875e-05,
      "model_forward_time": 1.3511271476745605,
      "step": 55776
    },
    {
      "epoch": 5.3192138671875e-05,
      "step": 55776,
      "training_step_time": 6.976970434188843
    },
    {
      "epoch": 5.319309234619141e-05,
      "model_forward_time": 1.7293436527252197,
      "step": 55777
    },
    {
      "epoch": 5.319309234619141e-05,
      "step": 55777,
      "training_step_time": 6.878333806991577
    },
    {
      "epoch": 5.319404602050781e-05,
      "model_forward_time": 1.723762035369873,
      "step": 55778
    },
    {
      "epoch": 5.319404602050781e-05,
      "step": 55778,
      "training_step_time": 8.428023099899292
    },
    {
      "epoch": 5.319499969482422e-05,
      "model_forward_time": 1.767134189605713,
      "step": 55779
    },
    {
      "epoch": 5.319499969482422e-05,
      "step": 55779,
      "training_step_time": 7.928056716918945
    },
    {
      "epoch": 5.3195953369140626e-05,
      "grad_norm": 0.13195578753948212,
      "learning_rate": 4.458889104872103e-06,
      "loss": 0.0527,
      "step": 55780
    },
    {
      "epoch": 5.3195953369140626e-05,
      "model_forward_time": 1.7338368892669678,
      "step": 55780
    },
    {
      "dynamics_loss_avg": 0.048537743650376795,
      "epoch": 5.3195953369140626e-05,
      "step": 55780
    },
    {
      "action_loss_avg": 0.000982939181267284,
      "epoch": 5.3195953369140626e-05,
      "step": 55780
    },
    {
      "epoch": 5.3195953369140626e-05,
      "step": 55780,
      "training_step_time": 7.827670574188232
    },
    {
      "epoch": 5.319690704345703e-05,
      "model_forward_time": 1.3671743869781494,
      "step": 55781
    },
    {
      "epoch": 5.319690704345703e-05,
      "step": 55781,
      "training_step_time": 6.858054161071777
    },
    {
      "epoch": 5.3197860717773434e-05,
      "model_forward_time": 1.740684986114502,
      "step": 55782
    },
    {
      "epoch": 5.3197860717773434e-05,
      "step": 55782,
      "training_step_time": 6.806228876113892
    },
    {
      "epoch": 5.3198814392089845e-05,
      "model_forward_time": 1.7404072284698486,
      "step": 55783
    },
    {
      "epoch": 5.3198814392089845e-05,
      "step": 55783,
      "training_step_time": 6.822484970092773
    },
    {
      "epoch": 5.319976806640625e-05,
      "model_forward_time": 1.7290558815002441,
      "step": 55784
    },
    {
      "epoch": 5.319976806640625e-05,
      "step": 55784,
      "training_step_time": 6.94059419631958
    },
    {
      "epoch": 5.320072174072265e-05,
      "model_forward_time": 1.7416367530822754,
      "step": 55785
    },
    {
      "epoch": 5.320072174072265e-05,
      "step": 55785,
      "training_step_time": 6.854989767074585
    },
    {
      "epoch": 5.3201675415039064e-05,
      "model_forward_time": 1.740630865097046,
      "step": 55786
    },
    {
      "epoch": 5.3201675415039064e-05,
      "step": 55786,
      "training_step_time": 6.804646968841553
    },
    {
      "epoch": 5.320262908935547e-05,
      "model_forward_time": 1.7312285900115967,
      "step": 55787
    },
    {
      "epoch": 5.320262908935547e-05,
      "step": 55787,
      "training_step_time": 7.70632529258728
    },
    {
      "epoch": 5.320358276367187e-05,
      "model_forward_time": 2.3543708324432373,
      "step": 55788
    },
    {
      "epoch": 5.320358276367187e-05,
      "step": 55788,
      "training_step_time": 9.275884866714478
    },
    {
      "epoch": 5.320453643798828e-05,
      "model_forward_time": 1.730316400527954,
      "step": 55789
    },
    {
      "epoch": 5.320453643798828e-05,
      "step": 55789,
      "training_step_time": 7.978262186050415
    },
    {
      "epoch": 5.320549011230469e-05,
      "grad_norm": 0.07820026576519012,
      "learning_rate": 4.457245375942715e-06,
      "loss": 0.0546,
      "step": 55790
    },
    {
      "epoch": 5.320549011230469e-05,
      "model_forward_time": 1.7172987461090088,
      "step": 55790
    },
    {
      "dynamics_loss_avg": 0.05057342108339071,
      "epoch": 5.320549011230469e-05,
      "step": 55790
    },
    {
      "action_loss_avg": 0.0009082548291189596,
      "epoch": 5.320549011230469e-05,
      "step": 55790
    },
    {
      "epoch": 5.320549011230469e-05,
      "step": 55790,
      "training_step_time": 6.8209686279296875
    },
    {
      "epoch": 5.320644378662109e-05,
      "model_forward_time": 1.7395057678222656,
      "step": 55791
    },
    {
      "epoch": 5.320644378662109e-05,
      "step": 55791,
      "training_step_time": 6.85159969329834
    },
    {
      "epoch": 5.32073974609375e-05,
      "model_forward_time": 1.7469892501831055,
      "step": 55792
    },
    {
      "epoch": 5.32073974609375e-05,
      "step": 55792,
      "training_step_time": 6.935969352722168
    },
    {
      "epoch": 5.3208351135253906e-05,
      "model_forward_time": 1.7265379428863525,
      "step": 55793
    },
    {
      "epoch": 5.3208351135253906e-05,
      "step": 55793,
      "training_step_time": 6.957338809967041
    },
    {
      "epoch": 5.320930480957031e-05,
      "model_forward_time": 1.730982780456543,
      "step": 55794
    },
    {
      "epoch": 5.320930480957031e-05,
      "step": 55794,
      "training_step_time": 6.888920783996582
    },
    {
      "epoch": 5.321025848388672e-05,
      "model_forward_time": 1.7325303554534912,
      "step": 55795
    },
    {
      "epoch": 5.321025848388672e-05,
      "step": 55795,
      "training_step_time": 6.920057773590088
    },
    {
      "epoch": 5.3211212158203124e-05,
      "model_forward_time": 1.3653898239135742,
      "step": 55796
    },
    {
      "epoch": 5.3211212158203124e-05,
      "step": 55796,
      "training_step_time": 7.726754665374756
    },
    {
      "epoch": 5.321216583251953e-05,
      "model_forward_time": 2.2877345085144043,
      "step": 55797
    },
    {
      "epoch": 5.321216583251953e-05,
      "step": 55797,
      "training_step_time": 11.002535104751587
    },
    {
      "epoch": 5.321311950683594e-05,
      "model_forward_time": 1.7742369174957275,
      "step": 55798
    },
    {
      "epoch": 5.321311950683594e-05,
      "step": 55798,
      "training_step_time": 8.565890312194824
    },
    {
      "epoch": 5.321407318115234e-05,
      "model_forward_time": 1.7223520278930664,
      "step": 55799
    },
    {
      "epoch": 5.321407318115234e-05,
      "step": 55799,
      "training_step_time": 7.014275789260864
    },
    {
      "epoch": 5.321502685546875e-05,
      "grad_norm": 0.10986874252557755,
      "learning_rate": 4.455601706368156e-06,
      "loss": 0.0558,
      "step": 55800
    },
    {
      "epoch": 5.321502685546875e-05,
      "model_forward_time": 1.735356330871582,
      "step": 55800
    },
    {
      "dynamics_loss_avg": 0.040816800110042094,
      "epoch": 5.321502685546875e-05,
      "step": 55800
    },
    {
      "action_loss_avg": 0.0009433149039978162,
      "epoch": 5.321502685546875e-05,
      "step": 55800
    },
    {
      "epoch": 5.321502685546875e-05,
      "step": 55800,
      "training_step_time": 6.957653760910034
    },
    {
      "epoch": 5.321598052978516e-05,
      "model_forward_time": 1.7368817329406738,
      "step": 55801
    },
    {
      "epoch": 5.321598052978516e-05,
      "step": 55801,
      "training_step_time": 6.771380424499512
    },
    {
      "epoch": 5.321693420410156e-05,
      "model_forward_time": 1.7408499717712402,
      "step": 55802
    },
    {
      "epoch": 5.321693420410156e-05,
      "step": 55802,
      "training_step_time": 6.869896173477173
    },
    {
      "epoch": 5.3217887878417966e-05,
      "model_forward_time": 1.7404191493988037,
      "step": 55803
    },
    {
      "epoch": 5.3217887878417966e-05,
      "step": 55803,
      "training_step_time": 6.91511344909668
    },
    {
      "epoch": 5.321884155273438e-05,
      "model_forward_time": 1.7409143447875977,
      "step": 55804
    },
    {
      "epoch": 5.321884155273438e-05,
      "step": 55804,
      "training_step_time": 6.957119464874268
    },
    {
      "epoch": 5.321979522705078e-05,
      "model_forward_time": 1.748189926147461,
      "step": 55805
    },
    {
      "epoch": 5.321979522705078e-05,
      "step": 55805,
      "training_step_time": 7.774752378463745
    },
    {
      "epoch": 5.3220748901367185e-05,
      "model_forward_time": 2.843679189682007,
      "step": 55806
    },
    {
      "epoch": 5.3220748901367185e-05,
      "step": 55806,
      "training_step_time": 12.311016321182251
    },
    {
      "epoch": 5.3221702575683596e-05,
      "model_forward_time": 1.7413790225982666,
      "step": 55807
    },
    {
      "epoch": 5.3221702575683596e-05,
      "step": 55807,
      "training_step_time": 8.050658702850342
    },
    {
      "epoch": 5.322265625e-05,
      "model_forward_time": 1.7383887767791748,
      "step": 55808
    },
    {
      "epoch": 5.322265625e-05,
      "step": 55808,
      "training_step_time": 7.42368483543396
    },
    {
      "epoch": 5.3223609924316404e-05,
      "model_forward_time": 1.7345013618469238,
      "step": 55809
    },
    {
      "epoch": 5.3223609924316404e-05,
      "step": 55809,
      "training_step_time": 6.880682706832886
    },
    {
      "epoch": 5.3224563598632815e-05,
      "grad_norm": 0.15108007192611694,
      "learning_rate": 4.453958096328179e-06,
      "loss": 0.0568,
      "step": 55810
    },
    {
      "epoch": 5.3224563598632815e-05,
      "model_forward_time": 1.7356464862823486,
      "step": 55810
    },
    {
      "dynamics_loss_avg": 0.05515204891562462,
      "epoch": 5.3224563598632815e-05,
      "step": 55810
    },
    {
      "action_loss_avg": 0.0013273589778691531,
      "epoch": 5.3224563598632815e-05,
      "step": 55810
    },
    {
      "epoch": 5.3224563598632815e-05,
      "step": 55810,
      "training_step_time": 6.879663944244385
    },
    {
      "epoch": 5.322551727294922e-05,
      "model_forward_time": 1.0255560874938965,
      "step": 55811
    },
    {
      "epoch": 5.322551727294922e-05,
      "step": 55811,
      "training_step_time": 6.795713186264038
    },
    {
      "epoch": 5.322647094726562e-05,
      "model_forward_time": 1.7309942245483398,
      "step": 55812
    },
    {
      "epoch": 5.322647094726562e-05,
      "step": 55812,
      "training_step_time": 6.844950914382935
    },
    {
      "epoch": 5.3227424621582034e-05,
      "model_forward_time": 1.7363855838775635,
      "step": 55813
    },
    {
      "epoch": 5.3227424621582034e-05,
      "step": 55813,
      "training_step_time": 6.908296823501587
    },
    {
      "epoch": 5.322837829589844e-05,
      "model_forward_time": 1.74106764793396,
      "step": 55814
    },
    {
      "epoch": 5.322837829589844e-05,
      "step": 55814,
      "training_step_time": 8.71001648902893
    },
    {
      "epoch": 5.322933197021484e-05,
      "model_forward_time": 1.8123571872711182,
      "step": 55815
    },
    {
      "epoch": 5.322933197021484e-05,
      "step": 55815,
      "training_step_time": 11.16099238395691
    },
    {
      "epoch": 5.323028564453125e-05,
      "model_forward_time": 2.8769478797912598,
      "step": 55816
    },
    {
      "epoch": 5.323028564453125e-05,
      "step": 55816,
      "training_step_time": 8.431958436965942
    },
    {
      "epoch": 5.3231239318847656e-05,
      "model_forward_time": 1.7301685810089111,
      "step": 55817
    },
    {
      "epoch": 5.3231239318847656e-05,
      "step": 55817,
      "training_step_time": 7.667104482650757
    },
    {
      "epoch": 5.323219299316406e-05,
      "model_forward_time": 1.7305879592895508,
      "step": 55818
    },
    {
      "epoch": 5.323219299316406e-05,
      "step": 55818,
      "training_step_time": 6.833867788314819
    },
    {
      "epoch": 5.323314666748047e-05,
      "model_forward_time": 1.736520767211914,
      "step": 55819
    },
    {
      "epoch": 5.323314666748047e-05,
      "step": 55819,
      "training_step_time": 6.843855857849121
    },
    {
      "epoch": 5.3234100341796875e-05,
      "grad_norm": 0.1015160009264946,
      "learning_rate": 4.452314546002523e-06,
      "loss": 0.0553,
      "step": 55820
    },
    {
      "epoch": 5.3234100341796875e-05,
      "model_forward_time": 1.7428853511810303,
      "step": 55820
    },
    {
      "dynamics_loss_avg": 0.05100030247122049,
      "epoch": 5.3234100341796875e-05,
      "step": 55820
    },
    {
      "action_loss_avg": 0.0018703454028582201,
      "epoch": 5.3234100341796875e-05,
      "step": 55820
    },
    {
      "epoch": 5.3234100341796875e-05,
      "step": 55820,
      "training_step_time": 6.876603841781616
    },
    {
      "epoch": 5.323505401611328e-05,
      "model_forward_time": 1.3677496910095215,
      "step": 55821
    },
    {
      "epoch": 5.323505401611328e-05,
      "step": 55821,
      "training_step_time": 6.87536358833313
    },
    {
      "epoch": 5.323600769042969e-05,
      "model_forward_time": 1.7393639087677002,
      "step": 55822
    },
    {
      "epoch": 5.323600769042969e-05,
      "step": 55822,
      "training_step_time": 6.944660663604736
    },
    {
      "epoch": 5.3236961364746094e-05,
      "model_forward_time": 1.7381353378295898,
      "step": 55823
    },
    {
      "epoch": 5.3236961364746094e-05,
      "step": 55823,
      "training_step_time": 7.665912628173828
    },
    {
      "epoch": 5.32379150390625e-05,
      "model_forward_time": 1.8160288333892822,
      "step": 55824
    },
    {
      "epoch": 5.32379150390625e-05,
      "step": 55824,
      "training_step_time": 11.596977949142456
    },
    {
      "epoch": 5.323886871337891e-05,
      "model_forward_time": 2.842623710632324,
      "step": 55825
    },
    {
      "epoch": 5.323886871337891e-05,
      "step": 55825,
      "training_step_time": 7.838181257247925
    },
    {
      "epoch": 5.323982238769531e-05,
      "model_forward_time": 1.7393553256988525,
      "step": 55826
    },
    {
      "epoch": 5.323982238769531e-05,
      "step": 55826,
      "training_step_time": 7.933361291885376
    },
    {
      "epoch": 5.324077606201172e-05,
      "model_forward_time": 1.734928846359253,
      "step": 55827
    },
    {
      "epoch": 5.324077606201172e-05,
      "step": 55827,
      "training_step_time": 7.362140655517578
    },
    {
      "epoch": 5.324172973632813e-05,
      "model_forward_time": 1.722466230392456,
      "step": 55828
    },
    {
      "epoch": 5.324172973632813e-05,
      "step": 55828,
      "training_step_time": 6.892239332199097
    },
    {
      "epoch": 5.324268341064453e-05,
      "model_forward_time": 1.361076831817627,
      "step": 55829
    },
    {
      "epoch": 5.324268341064453e-05,
      "step": 55829,
      "training_step_time": 6.8486411571502686
    },
    {
      "epoch": 5.3243637084960936e-05,
      "grad_norm": 0.10703232884407043,
      "learning_rate": 4.450671055570925e-06,
      "loss": 0.0564,
      "step": 55830
    },
    {
      "epoch": 5.3243637084960936e-05,
      "model_forward_time": 1.7409358024597168,
      "step": 55830
    },
    {
      "dynamics_loss_avg": 0.04036871362477541,
      "epoch": 5.3243637084960936e-05,
      "step": 55830
    },
    {
      "action_loss_avg": 0.0011376501352060586,
      "epoch": 5.3243637084960936e-05,
      "step": 55830
    },
    {
      "epoch": 5.3243637084960936e-05,
      "step": 55830,
      "training_step_time": 6.878249168395996
    },
    {
      "epoch": 5.324459075927735e-05,
      "model_forward_time": 1.7271919250488281,
      "step": 55831
    },
    {
      "epoch": 5.324459075927735e-05,
      "step": 55831,
      "training_step_time": 6.944064378738403
    },
    {
      "epoch": 5.324554443359375e-05,
      "model_forward_time": 1.7268955707550049,
      "step": 55832
    },
    {
      "epoch": 5.324554443359375e-05,
      "step": 55832,
      "training_step_time": 7.672271728515625
    },
    {
      "epoch": 5.3246498107910155e-05,
      "model_forward_time": 1.8142752647399902,
      "step": 55833
    },
    {
      "epoch": 5.3246498107910155e-05,
      "step": 55833,
      "training_step_time": 7.751195430755615
    },
    {
      "epoch": 5.3247451782226566e-05,
      "model_forward_time": 1.7416818141937256,
      "step": 55834
    },
    {
      "epoch": 5.3247451782226566e-05,
      "step": 55834,
      "training_step_time": 10.89959454536438
    },
    {
      "epoch": 5.324840545654297e-05,
      "model_forward_time": 2.8952436447143555,
      "step": 55835
    },
    {
      "epoch": 5.324840545654297e-05,
      "step": 55835,
      "training_step_time": 7.998335361480713
    },
    {
      "epoch": 5.3249359130859374e-05,
      "model_forward_time": 1.743093490600586,
      "step": 55836
    },
    {
      "epoch": 5.3249359130859374e-05,
      "step": 55836,
      "training_step_time": 7.42674708366394
    },
    {
      "epoch": 5.3250312805175784e-05,
      "model_forward_time": 1.7343571186065674,
      "step": 55837
    },
    {
      "epoch": 5.3250312805175784e-05,
      "step": 55837,
      "training_step_time": 6.784268140792847
    },
    {
      "epoch": 5.325126647949219e-05,
      "model_forward_time": 1.3637526035308838,
      "step": 55838
    },
    {
      "epoch": 5.325126647949219e-05,
      "step": 55838,
      "training_step_time": 6.792504549026489
    },
    {
      "epoch": 5.325222015380859e-05,
      "model_forward_time": 1.369136095046997,
      "step": 55839
    },
    {
      "epoch": 5.325222015380859e-05,
      "step": 55839,
      "training_step_time": 6.865300178527832
    },
    {
      "epoch": 5.3253173828125e-05,
      "grad_norm": 0.10277873277664185,
      "learning_rate": 4.449027625213113e-06,
      "loss": 0.0578,
      "step": 55840
    },
    {
      "epoch": 5.3253173828125e-05,
      "model_forward_time": 1.7441701889038086,
      "step": 55840
    },
    {
      "dynamics_loss_avg": 0.04144457634538412,
      "epoch": 5.3253173828125e-05,
      "step": 55840
    },
    {
      "action_loss_avg": 0.0011536352016264574,
      "epoch": 5.3253173828125e-05,
      "step": 55840
    },
    {
      "epoch": 5.3253173828125e-05,
      "step": 55840,
      "training_step_time": 6.927063941955566
    },
    {
      "epoch": 5.325412750244141e-05,
      "model_forward_time": 1.730337381362915,
      "step": 55841
    },
    {
      "epoch": 5.325412750244141e-05,
      "step": 55841,
      "training_step_time": 7.676034450531006
    },
    {
      "epoch": 5.325508117675781e-05,
      "model_forward_time": 1.8095412254333496,
      "step": 55842
    },
    {
      "epoch": 5.325508117675781e-05,
      "step": 55842,
      "training_step_time": 9.675888061523438
    },
    {
      "epoch": 5.325603485107422e-05,
      "model_forward_time": 1.7337138652801514,
      "step": 55843
    },
    {
      "epoch": 5.325603485107422e-05,
      "step": 55843,
      "training_step_time": 9.793156147003174
    },
    {
      "epoch": 5.3256988525390626e-05,
      "model_forward_time": 2.888437032699585,
      "step": 55844
    },
    {
      "epoch": 5.3256988525390626e-05,
      "step": 55844,
      "training_step_time": 7.922262668609619
    },
    {
      "epoch": 5.325794219970703e-05,
      "model_forward_time": 1.3705987930297852,
      "step": 55845
    },
    {
      "epoch": 5.325794219970703e-05,
      "step": 55845,
      "training_step_time": 7.371264457702637
    },
    {
      "epoch": 5.3258895874023434e-05,
      "model_forward_time": 1.7375836372375488,
      "step": 55846
    },
    {
      "epoch": 5.3258895874023434e-05,
      "step": 55846,
      "training_step_time": 6.857040882110596
    },
    {
      "epoch": 5.3259849548339845e-05,
      "model_forward_time": 1.7336785793304443,
      "step": 55847
    },
    {
      "epoch": 5.3259849548339845e-05,
      "step": 55847,
      "training_step_time": 6.793361186981201
    },
    {
      "epoch": 5.326080322265625e-05,
      "model_forward_time": 1.7368700504302979,
      "step": 55848
    },
    {
      "epoch": 5.326080322265625e-05,
      "step": 55848,
      "training_step_time": 6.843739032745361
    },
    {
      "epoch": 5.326175689697265e-05,
      "model_forward_time": 1.72882080078125,
      "step": 55849
    },
    {
      "epoch": 5.326175689697265e-05,
      "step": 55849,
      "training_step_time": 6.865753650665283
    },
    {
      "epoch": 5.3262710571289064e-05,
      "grad_norm": 0.11506913602352142,
      "learning_rate": 4.447384255108813e-06,
      "loss": 0.0581,
      "step": 55850
    },
    {
      "epoch": 5.3262710571289064e-05,
      "model_forward_time": 1.7419939041137695,
      "step": 55850
    },
    {
      "dynamics_loss_avg": 0.03933810777962208,
      "epoch": 5.3262710571289064e-05,
      "step": 55850
    },
    {
      "action_loss_avg": 0.001417074163327925,
      "epoch": 5.3262710571289064e-05,
      "step": 55850
    },
    {
      "epoch": 5.3262710571289064e-05,
      "step": 55850,
      "training_step_time": 7.68334436416626
    },
    {
      "epoch": 5.326366424560547e-05,
      "model_forward_time": 1.8103890419006348,
      "step": 55851
    },
    {
      "epoch": 5.326366424560547e-05,
      "step": 55851,
      "training_step_time": 7.712651968002319
    },
    {
      "epoch": 5.326461791992187e-05,
      "model_forward_time": 1.7301838397979736,
      "step": 55852
    },
    {
      "epoch": 5.326461791992187e-05,
      "step": 55852,
      "training_step_time": 9.498028993606567
    },
    {
      "epoch": 5.326557159423828e-05,
      "model_forward_time": 2.8673181533813477,
      "step": 55853
    },
    {
      "epoch": 5.326557159423828e-05,
      "step": 55853,
      "training_step_time": 8.970629215240479
    },
    {
      "epoch": 5.326652526855469e-05,
      "model_forward_time": 1.7338781356811523,
      "step": 55854
    },
    {
      "epoch": 5.326652526855469e-05,
      "step": 55854,
      "training_step_time": 7.655261993408203
    },
    {
      "epoch": 5.326747894287109e-05,
      "model_forward_time": 1.7611641883850098,
      "step": 55855
    },
    {
      "epoch": 5.326747894287109e-05,
      "step": 55855,
      "training_step_time": 6.867653846740723
    },
    {
      "epoch": 5.32684326171875e-05,
      "model_forward_time": 1.7350361347198486,
      "step": 55856
    },
    {
      "epoch": 5.32684326171875e-05,
      "step": 55856,
      "training_step_time": 6.853725910186768
    },
    {
      "epoch": 5.3269386291503906e-05,
      "model_forward_time": 1.7338249683380127,
      "step": 55857
    },
    {
      "epoch": 5.3269386291503906e-05,
      "step": 55857,
      "training_step_time": 6.8401548862457275
    },
    {
      "epoch": 5.327033996582031e-05,
      "model_forward_time": 1.7332570552825928,
      "step": 55858
    },
    {
      "epoch": 5.327033996582031e-05,
      "step": 55858,
      "training_step_time": 6.825834512710571
    },
    {
      "epoch": 5.327129364013672e-05,
      "model_forward_time": 1.736689805984497,
      "step": 55859
    },
    {
      "epoch": 5.327129364013672e-05,
      "step": 55859,
      "training_step_time": 7.62607741355896
    },
    {
      "epoch": 5.3272247314453124e-05,
      "grad_norm": 0.11169467866420746,
      "learning_rate": 4.445740945437738e-06,
      "loss": 0.055,
      "step": 55860
    },
    {
      "epoch": 5.3272247314453124e-05,
      "model_forward_time": 1.8180162906646729,
      "step": 55860
    },
    {
      "dynamics_loss_avg": 0.04713951582089067,
      "epoch": 5.3272247314453124e-05,
      "step": 55860
    },
    {
      "action_loss_avg": 0.001427257584873587,
      "epoch": 5.3272247314453124e-05,
      "step": 55860
    },
    {
      "epoch": 5.3272247314453124e-05,
      "step": 55860,
      "training_step_time": 8.292864561080933
    },
    {
      "epoch": 5.327320098876953e-05,
      "model_forward_time": 1.7390904426574707,
      "step": 55861
    },
    {
      "epoch": 5.327320098876953e-05,
      "step": 55861,
      "training_step_time": 8.791572570800781
    },
    {
      "epoch": 5.327415466308594e-05,
      "model_forward_time": 1.734041690826416,
      "step": 55862
    },
    {
      "epoch": 5.327415466308594e-05,
      "step": 55862,
      "training_step_time": 8.595171213150024
    },
    {
      "epoch": 5.327510833740234e-05,
      "model_forward_time": 2.8718667030334473,
      "step": 55863
    },
    {
      "epoch": 5.327510833740234e-05,
      "step": 55863,
      "training_step_time": 7.737395763397217
    },
    {
      "epoch": 5.327606201171875e-05,
      "model_forward_time": 1.7426540851593018,
      "step": 55864
    },
    {
      "epoch": 5.327606201171875e-05,
      "step": 55864,
      "training_step_time": 6.825332164764404
    },
    {
      "epoch": 5.327701568603516e-05,
      "model_forward_time": 1.7330389022827148,
      "step": 55865
    },
    {
      "epoch": 5.327701568603516e-05,
      "step": 55865,
      "training_step_time": 6.799076318740845
    },
    {
      "epoch": 5.327796936035156e-05,
      "model_forward_time": 1.7347779273986816,
      "step": 55866
    },
    {
      "epoch": 5.327796936035156e-05,
      "step": 55866,
      "training_step_time": 6.814323425292969
    },
    {
      "epoch": 5.3278923034667966e-05,
      "model_forward_time": 1.3686308860778809,
      "step": 55867
    },
    {
      "epoch": 5.3278923034667966e-05,
      "step": 55867,
      "training_step_time": 6.9030139446258545
    },
    {
      "epoch": 5.327987670898438e-05,
      "model_forward_time": 1.7370853424072266,
      "step": 55868
    },
    {
      "epoch": 5.327987670898438e-05,
      "step": 55868,
      "training_step_time": 7.7384397983551025
    },
    {
      "epoch": 5.328083038330078e-05,
      "model_forward_time": 1.8105182647705078,
      "step": 55869
    },
    {
      "epoch": 5.328083038330078e-05,
      "step": 55869,
      "training_step_time": 7.758526563644409
    },
    {
      "epoch": 5.3281784057617185e-05,
      "grad_norm": 0.15524545311927795,
      "learning_rate": 4.444097696379602e-06,
      "loss": 0.0593,
      "step": 55870
    },
    {
      "epoch": 5.3281784057617185e-05,
      "model_forward_time": 1.7459611892700195,
      "step": 55870
    },
    {
      "dynamics_loss_avg": 0.0790985558182001,
      "epoch": 5.3281784057617185e-05,
      "step": 55870
    },
    {
      "action_loss_avg": 0.0016244943166384474,
      "epoch": 5.3281784057617185e-05,
      "step": 55870
    },
    {
      "epoch": 5.3281784057617185e-05,
      "step": 55870,
      "training_step_time": 9.643279552459717
    },
    {
      "epoch": 5.3282737731933596e-05,
      "model_forward_time": 1.3666820526123047,
      "step": 55871
    },
    {
      "epoch": 5.3282737731933596e-05,
      "step": 55871,
      "training_step_time": 8.718638896942139
    },
    {
      "epoch": 5.328369140625e-05,
      "model_forward_time": 2.908557415008545,
      "step": 55872
    },
    {
      "epoch": 5.328369140625e-05,
      "step": 55872,
      "training_step_time": 8.65959620475769
    },
    {
      "epoch": 5.3284645080566404e-05,
      "model_forward_time": 1.748072624206543,
      "step": 55873
    },
    {
      "epoch": 5.3284645080566404e-05,
      "step": 55873,
      "training_step_time": 7.331057548522949
    },
    {
      "epoch": 5.3285598754882815e-05,
      "model_forward_time": 1.3782079219818115,
      "step": 55874
    },
    {
      "epoch": 5.3285598754882815e-05,
      "step": 55874,
      "training_step_time": 6.8868584632873535
    },
    {
      "epoch": 5.328655242919922e-05,
      "model_forward_time": 1.7353301048278809,
      "step": 55875
    },
    {
      "epoch": 5.328655242919922e-05,
      "step": 55875,
      "training_step_time": 6.802220344543457
    },
    {
      "epoch": 5.328750610351562e-05,
      "model_forward_time": 1.7371129989624023,
      "step": 55876
    },
    {
      "epoch": 5.328750610351562e-05,
      "step": 55876,
      "training_step_time": 6.816396713256836
    },
    {
      "epoch": 5.3288459777832034e-05,
      "model_forward_time": 1.7346761226654053,
      "step": 55877
    },
    {
      "epoch": 5.3288459777832034e-05,
      "step": 55877,
      "training_step_time": 7.7445228099823
    },
    {
      "epoch": 5.328941345214844e-05,
      "model_forward_time": 1.7977685928344727,
      "step": 55878
    },
    {
      "epoch": 5.328941345214844e-05,
      "step": 55878,
      "training_step_time": 7.758938550949097
    },
    {
      "epoch": 5.329036712646484e-05,
      "model_forward_time": 1.73193359375,
      "step": 55879
    },
    {
      "epoch": 5.329036712646484e-05,
      "step": 55879,
      "training_step_time": 8.844528198242188
    },
    {
      "epoch": 5.329132080078125e-05,
      "grad_norm": 0.12309965491294861,
      "learning_rate": 4.4424545081141055e-06,
      "loss": 0.0579,
      "step": 55880
    },
    {
      "epoch": 5.329132080078125e-05,
      "model_forward_time": 1.7418134212493896,
      "step": 55880
    },
    {
      "dynamics_loss_avg": 0.05362343546003103,
      "epoch": 5.329132080078125e-05,
      "step": 55880
    },
    {
      "action_loss_avg": 0.0023439322598278523,
      "epoch": 5.329132080078125e-05,
      "step": 55880
    },
    {
      "epoch": 5.329132080078125e-05,
      "step": 55880,
      "training_step_time": 9.692287921905518
    },
    {
      "epoch": 5.3292274475097656e-05,
      "model_forward_time": 2.837275743484497,
      "step": 55881
    },
    {
      "epoch": 5.3292274475097656e-05,
      "step": 55881,
      "training_step_time": 7.8076934814453125
    },
    {
      "epoch": 5.329322814941406e-05,
      "model_forward_time": 1.7453413009643555,
      "step": 55882
    },
    {
      "epoch": 5.329322814941406e-05,
      "step": 55882,
      "training_step_time": 7.355421543121338
    },
    {
      "epoch": 5.329418182373047e-05,
      "model_forward_time": 1.7456090450286865,
      "step": 55883
    },
    {
      "epoch": 5.329418182373047e-05,
      "step": 55883,
      "training_step_time": 6.840914726257324
    },
    {
      "epoch": 5.3295135498046875e-05,
      "model_forward_time": 1.7386796474456787,
      "step": 55884
    },
    {
      "epoch": 5.3295135498046875e-05,
      "step": 55884,
      "training_step_time": 6.85078763961792
    },
    {
      "epoch": 5.329608917236328e-05,
      "model_forward_time": 1.7470357418060303,
      "step": 55885
    },
    {
      "epoch": 5.329608917236328e-05,
      "step": 55885,
      "training_step_time": 6.777441740036011
    },
    {
      "epoch": 5.329704284667969e-05,
      "model_forward_time": 1.7380144596099854,
      "step": 55886
    },
    {
      "epoch": 5.329704284667969e-05,
      "step": 55886,
      "training_step_time": 7.693644285202026
    },
    {
      "epoch": 5.3297996520996094e-05,
      "model_forward_time": 1.7677152156829834,
      "step": 55887
    },
    {
      "epoch": 5.3297996520996094e-05,
      "step": 55887,
      "training_step_time": 7.646230697631836
    },
    {
      "epoch": 5.32989501953125e-05,
      "model_forward_time": 1.733698844909668,
      "step": 55888
    },
    {
      "epoch": 5.32989501953125e-05,
      "step": 55888,
      "training_step_time": 7.786099672317505
    },
    {
      "epoch": 5.329990386962891e-05,
      "model_forward_time": 1.7337005138397217,
      "step": 55889
    },
    {
      "epoch": 5.329990386962891e-05,
      "step": 55889,
      "training_step_time": 8.315929651260376
    },
    {
      "epoch": 5.330085754394531e-05,
      "grad_norm": 0.2942589819431305,
      "learning_rate": 4.440811380820946e-06,
      "loss": 0.0562,
      "step": 55890
    },
    {
      "epoch": 5.330085754394531e-05,
      "model_forward_time": 1.7378418445587158,
      "step": 55890
    },
    {
      "dynamics_loss_avg": 0.07112706545740366,
      "epoch": 5.330085754394531e-05,
      "step": 55890
    },
    {
      "action_loss_avg": 0.002697150711901486,
      "epoch": 5.330085754394531e-05,
      "step": 55890
    },
    {
      "epoch": 5.330085754394531e-05,
      "step": 55890,
      "training_step_time": 7.790493011474609
    },
    {
      "epoch": 5.330181121826172e-05,
      "model_forward_time": 2.897141695022583,
      "step": 55891
    },
    {
      "epoch": 5.330181121826172e-05,
      "step": 55891,
      "training_step_time": 7.746103048324585
    },
    {
      "epoch": 5.330276489257813e-05,
      "model_forward_time": 1.734426498413086,
      "step": 55892
    },
    {
      "epoch": 5.330276489257813e-05,
      "step": 55892,
      "training_step_time": 6.786708354949951
    },
    {
      "epoch": 5.330371856689453e-05,
      "model_forward_time": 1.7315208911895752,
      "step": 55893
    },
    {
      "epoch": 5.330371856689453e-05,
      "step": 55893,
      "training_step_time": 6.811026096343994
    },
    {
      "epoch": 5.3304672241210936e-05,
      "model_forward_time": 1.7302603721618652,
      "step": 55894
    },
    {
      "epoch": 5.3304672241210936e-05,
      "step": 55894,
      "training_step_time": 6.776949644088745
    },
    {
      "epoch": 5.330562591552735e-05,
      "model_forward_time": 1.7341320514678955,
      "step": 55895
    },
    {
      "epoch": 5.330562591552735e-05,
      "step": 55895,
      "training_step_time": 7.699566602706909
    },
    {
      "epoch": 5.330657958984375e-05,
      "model_forward_time": 1.7345917224884033,
      "step": 55896
    },
    {
      "epoch": 5.330657958984375e-05,
      "step": 55896,
      "training_step_time": 8.135881185531616
    },
    {
      "epoch": 5.3307533264160155e-05,
      "model_forward_time": 1.7414963245391846,
      "step": 55897
    },
    {
      "epoch": 5.3307533264160155e-05,
      "step": 55897,
      "training_step_time": 8.648370265960693
    },
    {
      "epoch": 5.3308486938476566e-05,
      "model_forward_time": 1.7358677387237549,
      "step": 55898
    },
    {
      "epoch": 5.3308486938476566e-05,
      "step": 55898,
      "training_step_time": 7.725588321685791
    },
    {
      "epoch": 5.330944061279297e-05,
      "model_forward_time": 1.7380201816558838,
      "step": 55899
    },
    {
      "epoch": 5.330944061279297e-05,
      "step": 55899,
      "training_step_time": 9.216876745223999
    },
    {
      "epoch": 5.3310394287109374e-05,
      "grad_norm": 0.09174707531929016,
      "learning_rate": 4.439168314679813e-06,
      "loss": 0.057,
      "step": 55900
    },
    {
      "epoch": 5.3310394287109374e-05,
      "model_forward_time": 2.9402551651000977,
      "step": 55900
    },
    {
      "dynamics_loss_avg": 0.06688584387302399,
      "epoch": 5.3310394287109374e-05,
      "step": 55900
    },
    {
      "action_loss_avg": 0.0015507160598644987,
      "epoch": 5.3310394287109374e-05,
      "step": 55900
    },
    {
      "epoch": 5.3310394287109374e-05,
      "step": 55900,
      "training_step_time": 7.784245491027832
    },
    {
      "epoch": 5.3311347961425784e-05,
      "model_forward_time": 1.7326750755310059,
      "step": 55901
    },
    {
      "epoch": 5.3311347961425784e-05,
      "step": 55901,
      "training_step_time": 6.812944412231445
    },
    {
      "epoch": 5.331230163574219e-05,
      "model_forward_time": 1.7322943210601807,
      "step": 55902
    },
    {
      "epoch": 5.331230163574219e-05,
      "step": 55902,
      "training_step_time": 6.734091758728027
    },
    {
      "epoch": 5.331325531005859e-05,
      "model_forward_time": 1.7365577220916748,
      "step": 55903
    },
    {
      "epoch": 5.331325531005859e-05,
      "step": 55903,
      "training_step_time": 6.790006637573242
    },
    {
      "epoch": 5.3314208984375e-05,
      "model_forward_time": 1.7346937656402588,
      "step": 55904
    },
    {
      "epoch": 5.3314208984375e-05,
      "step": 55904,
      "training_step_time": 7.703623294830322
    },
    {
      "epoch": 5.331516265869141e-05,
      "model_forward_time": 1.735435962677002,
      "step": 55905
    },
    {
      "epoch": 5.331516265869141e-05,
      "step": 55905,
      "training_step_time": 8.448763370513916
    },
    {
      "epoch": 5.331611633300781e-05,
      "model_forward_time": 1.808722972869873,
      "step": 55906
    },
    {
      "epoch": 5.331611633300781e-05,
      "step": 55906,
      "training_step_time": 7.739076614379883
    },
    {
      "epoch": 5.3317070007324215e-05,
      "model_forward_time": 1.7349777221679688,
      "step": 55907
    },
    {
      "epoch": 5.3317070007324215e-05,
      "step": 55907,
      "training_step_time": 8.713465929031372
    },
    {
      "epoch": 5.3318023681640626e-05,
      "model_forward_time": 1.7316107749938965,
      "step": 55908
    },
    {
      "epoch": 5.3318023681640626e-05,
      "step": 55908,
      "training_step_time": 7.805887699127197
    },
    {
      "epoch": 5.331897735595703e-05,
      "model_forward_time": 2.9095795154571533,
      "step": 55909
    },
    {
      "epoch": 5.331897735595703e-05,
      "step": 55909,
      "training_step_time": 7.794398069381714
    },
    {
      "epoch": 5.3319931030273434e-05,
      "grad_norm": 0.10730241984128952,
      "learning_rate": 4.43752530987039e-06,
      "loss": 0.0583,
      "step": 55910
    },
    {
      "epoch": 5.3319931030273434e-05,
      "model_forward_time": 1.735724925994873,
      "step": 55910
    },
    {
      "dynamics_loss_avg": 0.07125223856419324,
      "epoch": 5.3319931030273434e-05,
      "step": 55910
    },
    {
      "action_loss_avg": 0.0020205461361911146,
      "epoch": 5.3319931030273434e-05,
      "step": 55910
    },
    {
      "epoch": 5.3319931030273434e-05,
      "step": 55910,
      "training_step_time": 7.4543774127960205
    },
    {
      "epoch": 5.3320884704589845e-05,
      "model_forward_time": 1.7378592491149902,
      "step": 55911
    },
    {
      "epoch": 5.3320884704589845e-05,
      "step": 55911,
      "training_step_time": 6.847855806350708
    },
    {
      "epoch": 5.332183837890625e-05,
      "model_forward_time": 1.7301480770111084,
      "step": 55912
    },
    {
      "epoch": 5.332183837890625e-05,
      "step": 55912,
      "training_step_time": 6.866937637329102
    },
    {
      "epoch": 5.332279205322265e-05,
      "model_forward_time": 1.7342207431793213,
      "step": 55913
    },
    {
      "epoch": 5.332279205322265e-05,
      "step": 55913,
      "training_step_time": 7.678240060806274
    },
    {
      "epoch": 5.3323745727539064e-05,
      "model_forward_time": 1.738692045211792,
      "step": 55914
    },
    {
      "epoch": 5.3323745727539064e-05,
      "step": 55914,
      "training_step_time": 7.3896331787109375
    },
    {
      "epoch": 5.332469940185547e-05,
      "model_forward_time": 1.8083081245422363,
      "step": 55915
    },
    {
      "epoch": 5.332469940185547e-05,
      "step": 55915,
      "training_step_time": 8.833556175231934
    },
    {
      "epoch": 5.332565307617187e-05,
      "model_forward_time": 1.737278699874878,
      "step": 55916
    },
    {
      "epoch": 5.332565307617187e-05,
      "step": 55916,
      "training_step_time": 8.52510952949524
    },
    {
      "epoch": 5.332660675048828e-05,
      "model_forward_time": 1.7347185611724854,
      "step": 55917
    },
    {
      "epoch": 5.332660675048828e-05,
      "step": 55917,
      "training_step_time": 9.27049970626831
    },
    {
      "epoch": 5.332756042480469e-05,
      "model_forward_time": 2.8920352458953857,
      "step": 55918
    },
    {
      "epoch": 5.332756042480469e-05,
      "step": 55918,
      "training_step_time": 8.869223594665527
    },
    {
      "epoch": 5.332851409912109e-05,
      "model_forward_time": 1.7738635540008545,
      "step": 55919
    },
    {
      "epoch": 5.332851409912109e-05,
      "step": 55919,
      "training_step_time": 7.3653929233551025
    },
    {
      "epoch": 5.33294677734375e-05,
      "grad_norm": 0.12687179446220398,
      "learning_rate": 4.435882366572352e-06,
      "loss": 0.0538,
      "step": 55920
    },
    {
      "epoch": 5.33294677734375e-05,
      "model_forward_time": 1.7436761856079102,
      "step": 55920
    },
    {
      "dynamics_loss_avg": 0.05945324394851923,
      "epoch": 5.33294677734375e-05,
      "step": 55920
    },
    {
      "action_loss_avg": 0.0016018762093153783,
      "epoch": 5.33294677734375e-05,
      "step": 55920
    },
    {
      "epoch": 5.33294677734375e-05,
      "step": 55920,
      "training_step_time": 6.827356338500977
    },
    {
      "epoch": 5.3330421447753906e-05,
      "model_forward_time": 1.7359826564788818,
      "step": 55921
    },
    {
      "epoch": 5.3330421447753906e-05,
      "step": 55921,
      "training_step_time": 6.78849458694458
    },
    {
      "epoch": 5.333137512207031e-05,
      "model_forward_time": 1.7439370155334473,
      "step": 55922
    },
    {
      "epoch": 5.333137512207031e-05,
      "step": 55922,
      "training_step_time": 7.610053300857544
    },
    {
      "epoch": 5.333232879638672e-05,
      "model_forward_time": 1.7374114990234375,
      "step": 55923
    },
    {
      "epoch": 5.333232879638672e-05,
      "step": 55923,
      "training_step_time": 7.096033334732056
    },
    {
      "epoch": 5.3333282470703125e-05,
      "model_forward_time": 1.8152530193328857,
      "step": 55924
    },
    {
      "epoch": 5.3333282470703125e-05,
      "step": 55924,
      "training_step_time": 7.7900779247283936
    },
    {
      "epoch": 5.333423614501953e-05,
      "model_forward_time": 1.736950397491455,
      "step": 55925
    },
    {
      "epoch": 5.333423614501953e-05,
      "step": 55925,
      "training_step_time": 7.775382041931152
    },
    {
      "epoch": 5.333518981933594e-05,
      "model_forward_time": 1.7329881191253662,
      "step": 55926
    },
    {
      "epoch": 5.333518981933594e-05,
      "step": 55926,
      "training_step_time": 11.267742395401001
    },
    {
      "epoch": 5.333614349365234e-05,
      "model_forward_time": 2.862858533859253,
      "step": 55927
    },
    {
      "epoch": 5.333614349365234e-05,
      "step": 55927,
      "training_step_time": 7.839195966720581
    },
    {
      "epoch": 5.333709716796875e-05,
      "model_forward_time": 1.7480080127716064,
      "step": 55928
    },
    {
      "epoch": 5.333709716796875e-05,
      "step": 55928,
      "training_step_time": 7.688264846801758
    },
    {
      "epoch": 5.333805084228516e-05,
      "model_forward_time": 1.3676784038543701,
      "step": 55929
    },
    {
      "epoch": 5.333805084228516e-05,
      "step": 55929,
      "training_step_time": 6.868775367736816
    },
    {
      "epoch": 5.333900451660156e-05,
      "grad_norm": 0.11739522218704224,
      "learning_rate": 4.434239484965371e-06,
      "loss": 0.0578,
      "step": 55930
    },
    {
      "epoch": 5.333900451660156e-05,
      "model_forward_time": 1.7410953044891357,
      "step": 55930
    },
    {
      "dynamics_loss_avg": 0.0607191413640976,
      "epoch": 5.333900451660156e-05,
      "step": 55930
    },
    {
      "action_loss_avg": 0.001455059548607096,
      "epoch": 5.333900451660156e-05,
      "step": 55930
    },
    {
      "epoch": 5.333900451660156e-05,
      "step": 55930,
      "training_step_time": 6.887815475463867
    },
    {
      "epoch": 5.3339958190917966e-05,
      "model_forward_time": 1.7398977279663086,
      "step": 55931
    },
    {
      "epoch": 5.3339958190917966e-05,
      "step": 55931,
      "training_step_time": 7.164075136184692
    },
    {
      "epoch": 5.334091186523438e-05,
      "model_forward_time": 1.7403998374938965,
      "step": 55932
    },
    {
      "epoch": 5.334091186523438e-05,
      "step": 55932,
      "training_step_time": 7.734150171279907
    },
    {
      "epoch": 5.334186553955078e-05,
      "model_forward_time": 1.7745516300201416,
      "step": 55933
    },
    {
      "epoch": 5.334186553955078e-05,
      "step": 55933,
      "training_step_time": 8.29774284362793
    },
    {
      "epoch": 5.3342819213867185e-05,
      "model_forward_time": 1.7346546649932861,
      "step": 55934
    },
    {
      "epoch": 5.3342819213867185e-05,
      "step": 55934,
      "training_step_time": 7.707982301712036
    },
    {
      "epoch": 5.3343772888183596e-05,
      "model_forward_time": 1.7413716316223145,
      "step": 55935
    },
    {
      "epoch": 5.3343772888183596e-05,
      "step": 55935,
      "training_step_time": 8.635814666748047
    },
    {
      "epoch": 5.33447265625e-05,
      "model_forward_time": 1.7363791465759277,
      "step": 55936
    },
    {
      "epoch": 5.33447265625e-05,
      "step": 55936,
      "training_step_time": 7.87592077255249
    },
    {
      "epoch": 5.3345680236816404e-05,
      "model_forward_time": 2.924614429473877,
      "step": 55937
    },
    {
      "epoch": 5.3345680236816404e-05,
      "step": 55937,
      "training_step_time": 7.829347372055054
    },
    {
      "epoch": 5.3346633911132815e-05,
      "model_forward_time": 1.7386987209320068,
      "step": 55938
    },
    {
      "epoch": 5.3346633911132815e-05,
      "step": 55938,
      "training_step_time": 6.856067419052124
    },
    {
      "epoch": 5.334758758544922e-05,
      "model_forward_time": 1.7359099388122559,
      "step": 55939
    },
    {
      "epoch": 5.334758758544922e-05,
      "step": 55939,
      "training_step_time": 6.845582485198975
    },
    {
      "epoch": 5.334854125976562e-05,
      "grad_norm": 0.11256914585828781,
      "learning_rate": 4.43259666522911e-06,
      "loss": 0.0557,
      "step": 55940
    },
    {
      "epoch": 5.334854125976562e-05,
      "model_forward_time": 1.7376179695129395,
      "step": 55940
    },
    {
      "dynamics_loss_avg": 0.04769921489059925,
      "epoch": 5.334854125976562e-05,
      "step": 55940
    },
    {
      "action_loss_avg": 0.0012412139622028917,
      "epoch": 5.334854125976562e-05,
      "step": 55940
    },
    {
      "epoch": 5.334854125976562e-05,
      "step": 55940,
      "training_step_time": 7.133035898208618
    },
    {
      "epoch": 5.3349494934082034e-05,
      "model_forward_time": 1.7368426322937012,
      "step": 55941
    },
    {
      "epoch": 5.3349494934082034e-05,
      "step": 55941,
      "training_step_time": 7.667023658752441
    },
    {
      "epoch": 5.335044860839844e-05,
      "model_forward_time": 1.7544670104980469,
      "step": 55942
    },
    {
      "epoch": 5.335044860839844e-05,
      "step": 55942,
      "training_step_time": 8.589001417160034
    },
    {
      "epoch": 5.335140228271484e-05,
      "model_forward_time": 1.7393379211425781,
      "step": 55943
    },
    {
      "epoch": 5.335140228271484e-05,
      "step": 55943,
      "training_step_time": 8.695837259292603
    },
    {
      "epoch": 5.335235595703125e-05,
      "model_forward_time": 1.7596774101257324,
      "step": 55944
    },
    {
      "epoch": 5.335235595703125e-05,
      "step": 55944,
      "training_step_time": 7.749887466430664
    },
    {
      "epoch": 5.3353309631347657e-05,
      "model_forward_time": 1.3638317584991455,
      "step": 55945
    },
    {
      "epoch": 5.3353309631347657e-05,
      "step": 55945,
      "training_step_time": 8.468263149261475
    },
    {
      "epoch": 5.335426330566406e-05,
      "model_forward_time": 2.9324445724487305,
      "step": 55946
    },
    {
      "epoch": 5.335426330566406e-05,
      "step": 55946,
      "training_step_time": 7.78262734413147
    },
    {
      "epoch": 5.335521697998047e-05,
      "model_forward_time": 1.7431716918945312,
      "step": 55947
    },
    {
      "epoch": 5.335521697998047e-05,
      "step": 55947,
      "training_step_time": 7.246598958969116
    },
    {
      "epoch": 5.3356170654296875e-05,
      "model_forward_time": 1.7190825939178467,
      "step": 55948
    },
    {
      "epoch": 5.3356170654296875e-05,
      "step": 55948,
      "training_step_time": 6.760764122009277
    },
    {
      "epoch": 5.335712432861328e-05,
      "model_forward_time": 1.7279024124145508,
      "step": 55949
    },
    {
      "epoch": 5.335712432861328e-05,
      "step": 55949,
      "training_step_time": 7.067506790161133
    },
    {
      "epoch": 5.335807800292969e-05,
      "grad_norm": 0.10711784660816193,
      "learning_rate": 4.430953907543225e-06,
      "loss": 0.0561,
      "step": 55950
    },
    {
      "epoch": 5.335807800292969e-05,
      "model_forward_time": 1.737706184387207,
      "step": 55950
    },
    {
      "dynamics_loss_avg": 0.043760023452341555,
      "epoch": 5.335807800292969e-05,
      "step": 55950
    },
    {
      "action_loss_avg": 0.0010368782008299605,
      "epoch": 5.335807800292969e-05,
      "step": 55950
    },
    {
      "epoch": 5.335807800292969e-05,
      "step": 55950,
      "training_step_time": 7.673498153686523
    },
    {
      "epoch": 5.3359031677246094e-05,
      "model_forward_time": 1.7465729713439941,
      "step": 55951
    },
    {
      "epoch": 5.3359031677246094e-05,
      "step": 55951,
      "training_step_time": 7.691080808639526
    },
    {
      "epoch": 5.33599853515625e-05,
      "model_forward_time": 1.8113312721252441,
      "step": 55952
    },
    {
      "epoch": 5.33599853515625e-05,
      "step": 55952,
      "training_step_time": 7.698407888412476
    },
    {
      "epoch": 5.336093902587891e-05,
      "model_forward_time": 1.7305116653442383,
      "step": 55953
    },
    {
      "epoch": 5.336093902587891e-05,
      "step": 55953,
      "training_step_time": 8.697922229766846
    },
    {
      "epoch": 5.336189270019531e-05,
      "model_forward_time": 1.732104778289795,
      "step": 55954
    },
    {
      "epoch": 5.336189270019531e-05,
      "step": 55954,
      "training_step_time": 8.941226243972778
    },
    {
      "epoch": 5.336284637451172e-05,
      "model_forward_time": 2.8880560398101807,
      "step": 55955
    },
    {
      "epoch": 5.336284637451172e-05,
      "step": 55955,
      "training_step_time": 8.320417881011963
    },
    {
      "epoch": 5.336380004882813e-05,
      "model_forward_time": 1.7427806854248047,
      "step": 55956
    },
    {
      "epoch": 5.336380004882813e-05,
      "step": 55956,
      "training_step_time": 7.360146760940552
    },
    {
      "epoch": 5.336475372314453e-05,
      "model_forward_time": 1.7320034503936768,
      "step": 55957
    },
    {
      "epoch": 5.336475372314453e-05,
      "step": 55957,
      "training_step_time": 6.848048686981201
    },
    {
      "epoch": 5.3365707397460936e-05,
      "model_forward_time": 1.73541259765625,
      "step": 55958
    },
    {
      "epoch": 5.3365707397460936e-05,
      "step": 55958,
      "training_step_time": 7.139955997467041
    },
    {
      "epoch": 5.336666107177735e-05,
      "model_forward_time": 1.7220940589904785,
      "step": 55959
    },
    {
      "epoch": 5.336666107177735e-05,
      "step": 55959,
      "training_step_time": 7.725103378295898
    },
    {
      "epoch": 5.336761474609375e-05,
      "grad_norm": 0.08643414825201035,
      "learning_rate": 4.429311212087365e-06,
      "loss": 0.0546,
      "step": 55960
    },
    {
      "epoch": 5.336761474609375e-05,
      "model_forward_time": 1.7473454475402832,
      "step": 55960
    },
    {
      "dynamics_loss_avg": 0.0469632763415575,
      "epoch": 5.336761474609375e-05,
      "step": 55960
    },
    {
      "action_loss_avg": 0.0013320903992280364,
      "epoch": 5.336761474609375e-05,
      "step": 55960
    },
    {
      "epoch": 5.336761474609375e-05,
      "step": 55960,
      "training_step_time": 7.775072813034058
    },
    {
      "epoch": 5.3368568420410155e-05,
      "model_forward_time": 1.8096601963043213,
      "step": 55961
    },
    {
      "epoch": 5.3368568420410155e-05,
      "step": 55961,
      "training_step_time": 8.745840549468994
    },
    {
      "epoch": 5.3369522094726566e-05,
      "model_forward_time": 1.7509942054748535,
      "step": 55962
    },
    {
      "epoch": 5.3369522094726566e-05,
      "step": 55962,
      "training_step_time": 10.407325506210327
    },
    {
      "epoch": 5.337047576904297e-05,
      "model_forward_time": 1.3649446964263916,
      "step": 55963
    },
    {
      "epoch": 5.337047576904297e-05,
      "step": 55963,
      "training_step_time": 10.561106204986572
    },
    {
      "epoch": 5.3371429443359374e-05,
      "model_forward_time": 2.8386342525482178,
      "step": 55964
    },
    {
      "epoch": 5.3371429443359374e-05,
      "step": 55964,
      "training_step_time": 7.758201599121094
    },
    {
      "epoch": 5.3372383117675785e-05,
      "model_forward_time": 1.7415204048156738,
      "step": 55965
    },
    {
      "epoch": 5.3372383117675785e-05,
      "step": 55965,
      "training_step_time": 7.73881196975708
    },
    {
      "epoch": 5.337333679199219e-05,
      "model_forward_time": 1.7360796928405762,
      "step": 55966
    },
    {
      "epoch": 5.337333679199219e-05,
      "step": 55966,
      "training_step_time": 6.817591428756714
    },
    {
      "epoch": 5.337429046630859e-05,
      "model_forward_time": 1.735764980316162,
      "step": 55967
    },
    {
      "epoch": 5.337429046630859e-05,
      "step": 55967,
      "training_step_time": 7.107171297073364
    },
    {
      "epoch": 5.3375244140625e-05,
      "model_forward_time": 1.7337372303009033,
      "step": 55968
    },
    {
      "epoch": 5.3375244140625e-05,
      "step": 55968,
      "training_step_time": 7.643369436264038
    },
    {
      "epoch": 5.337619781494141e-05,
      "model_forward_time": 1.7465476989746094,
      "step": 55969
    },
    {
      "epoch": 5.337619781494141e-05,
      "step": 55969,
      "training_step_time": 7.724822044372559
    },
    {
      "epoch": 5.337715148925781e-05,
      "grad_norm": 0.09771555662155151,
      "learning_rate": 4.427668579041171e-06,
      "loss": 0.0533,
      "step": 55970
    },
    {
      "epoch": 5.337715148925781e-05,
      "model_forward_time": 1.8167710304260254,
      "step": 55970
    },
    {
      "dynamics_loss_avg": 0.045658152736723426,
      "epoch": 5.337715148925781e-05,
      "step": 55970
    },
    {
      "action_loss_avg": 0.0014267152029788122,
      "epoch": 5.337715148925781e-05,
      "step": 55970
    },
    {
      "epoch": 5.337715148925781e-05,
      "step": 55970,
      "training_step_time": 7.690987586975098
    },
    {
      "epoch": 5.3378105163574215e-05,
      "model_forward_time": 1.7297813892364502,
      "step": 55971
    },
    {
      "epoch": 5.3378105163574215e-05,
      "step": 55971,
      "training_step_time": 9.307222127914429
    },
    {
      "epoch": 5.3379058837890626e-05,
      "model_forward_time": 1.7385532855987549,
      "step": 55972
    },
    {
      "epoch": 5.3379058837890626e-05,
      "step": 55972,
      "training_step_time": 8.947028636932373
    },
    {
      "epoch": 5.338001251220703e-05,
      "model_forward_time": 1.7292759418487549,
      "step": 55973
    },
    {
      "epoch": 5.338001251220703e-05,
      "step": 55973,
      "training_step_time": 7.837197542190552
    },
    {
      "epoch": 5.3380966186523434e-05,
      "model_forward_time": 2.930232048034668,
      "step": 55974
    },
    {
      "epoch": 5.3380966186523434e-05,
      "step": 55974,
      "training_step_time": 7.797087669372559
    },
    {
      "epoch": 5.3381919860839845e-05,
      "model_forward_time": 1.7293801307678223,
      "step": 55975
    },
    {
      "epoch": 5.3381919860839845e-05,
      "step": 55975,
      "training_step_time": 7.19870662689209
    },
    {
      "epoch": 5.338287353515625e-05,
      "model_forward_time": 1.7340772151947021,
      "step": 55976
    },
    {
      "epoch": 5.338287353515625e-05,
      "step": 55976,
      "training_step_time": 7.076747894287109
    },
    {
      "epoch": 5.338382720947265e-05,
      "model_forward_time": 1.7324035167694092,
      "step": 55977
    },
    {
      "epoch": 5.338382720947265e-05,
      "step": 55977,
      "training_step_time": 7.1335132122039795
    },
    {
      "epoch": 5.3384780883789064e-05,
      "model_forward_time": 1.7369663715362549,
      "step": 55978
    },
    {
      "epoch": 5.3384780883789064e-05,
      "step": 55978,
      "training_step_time": 7.849139928817749
    },
    {
      "epoch": 5.338573455810547e-05,
      "model_forward_time": 1.8050017356872559,
      "step": 55979
    },
    {
      "epoch": 5.338573455810547e-05,
      "step": 55979,
      "training_step_time": 7.699446439743042
    },
    {
      "epoch": 5.338668823242187e-05,
      "grad_norm": 0.12404605001211166,
      "learning_rate": 4.426026008584282e-06,
      "loss": 0.0579,
      "step": 55980
    },
    {
      "epoch": 5.338668823242187e-05,
      "model_forward_time": 1.7352404594421387,
      "step": 55980
    },
    {
      "dynamics_loss_avg": 0.06176605559885502,
      "epoch": 5.338668823242187e-05,
      "step": 55980
    },
    {
      "action_loss_avg": 0.0013645415659993888,
      "epoch": 5.338668823242187e-05,
      "step": 55980
    },
    {
      "epoch": 5.338668823242187e-05,
      "step": 55980,
      "training_step_time": 8.791407823562622
    },
    {
      "epoch": 5.338764190673828e-05,
      "model_forward_time": 1.3685622215270996,
      "step": 55981
    },
    {
      "epoch": 5.338764190673828e-05,
      "step": 55981,
      "training_step_time": 9.620233058929443
    },
    {
      "epoch": 5.338859558105469e-05,
      "model_forward_time": 1.7403302192687988,
      "step": 55982
    },
    {
      "epoch": 5.338859558105469e-05,
      "step": 55982,
      "training_step_time": 7.856621742248535
    },
    {
      "epoch": 5.338954925537109e-05,
      "model_forward_time": 2.8966102600097656,
      "step": 55983
    },
    {
      "epoch": 5.338954925537109e-05,
      "step": 55983,
      "training_step_time": 7.875406503677368
    },
    {
      "epoch": 5.33905029296875e-05,
      "model_forward_time": 1.7349622249603271,
      "step": 55984
    },
    {
      "epoch": 5.33905029296875e-05,
      "step": 55984,
      "training_step_time": 7.543587923049927
    },
    {
      "epoch": 5.3391456604003906e-05,
      "model_forward_time": 1.7424964904785156,
      "step": 55985
    },
    {
      "epoch": 5.3391456604003906e-05,
      "step": 55985,
      "training_step_time": 7.063033819198608
    },
    {
      "epoch": 5.339241027832031e-05,
      "model_forward_time": 1.7402887344360352,
      "step": 55986
    },
    {
      "epoch": 5.339241027832031e-05,
      "step": 55986,
      "training_step_time": 6.760332107543945
    },
    {
      "epoch": 5.339336395263672e-05,
      "model_forward_time": 1.7426636219024658,
      "step": 55987
    },
    {
      "epoch": 5.339336395263672e-05,
      "step": 55987,
      "training_step_time": 7.72115159034729
    },
    {
      "epoch": 5.3394317626953125e-05,
      "model_forward_time": 1.8122777938842773,
      "step": 55988
    },
    {
      "epoch": 5.3394317626953125e-05,
      "step": 55988,
      "training_step_time": 7.7107954025268555
    },
    {
      "epoch": 5.339527130126953e-05,
      "model_forward_time": 1.7460625171661377,
      "step": 55989
    },
    {
      "epoch": 5.339527130126953e-05,
      "step": 55989,
      "training_step_time": 8.736687421798706
    },
    {
      "epoch": 5.339622497558594e-05,
      "grad_norm": 0.09306638687849045,
      "learning_rate": 4.424383500896326e-06,
      "loss": 0.0556,
      "step": 55990
    },
    {
      "epoch": 5.339622497558594e-05,
      "model_forward_time": 1.7392997741699219,
      "step": 55990
    },
    {
      "dynamics_loss_avg": 0.06605487931519746,
      "epoch": 5.339622497558594e-05,
      "step": 55990
    },
    {
      "action_loss_avg": 0.0022158433916047216,
      "epoch": 5.339622497558594e-05,
      "step": 55990
    },
    {
      "epoch": 5.339622497558594e-05,
      "step": 55990,
      "training_step_time": 9.646589040756226
    },
    {
      "epoch": 5.3397178649902343e-05,
      "model_forward_time": 1.7444095611572266,
      "step": 55991
    },
    {
      "epoch": 5.3397178649902343e-05,
      "step": 55991,
      "training_step_time": 10.31053638458252
    },
    {
      "epoch": 5.339813232421875e-05,
      "model_forward_time": 2.9420478343963623,
      "step": 55992
    },
    {
      "epoch": 5.339813232421875e-05,
      "step": 55992,
      "training_step_time": 7.830857515335083
    },
    {
      "epoch": 5.339908599853516e-05,
      "model_forward_time": 1.7496223449707031,
      "step": 55993
    },
    {
      "epoch": 5.339908599853516e-05,
      "step": 55993,
      "training_step_time": 7.516960382461548
    },
    {
      "epoch": 5.340003967285156e-05,
      "model_forward_time": 1.7253446578979492,
      "step": 55994
    },
    {
      "epoch": 5.340003967285156e-05,
      "step": 55994,
      "training_step_time": 7.063352108001709
    },
    {
      "epoch": 5.3400993347167966e-05,
      "model_forward_time": 1.337048053741455,
      "step": 55995
    },
    {
      "epoch": 5.3400993347167966e-05,
      "step": 55995,
      "training_step_time": 6.788039684295654
    },
    {
      "epoch": 5.340194702148438e-05,
      "model_forward_time": 1.7330873012542725,
      "step": 55996
    },
    {
      "epoch": 5.340194702148438e-05,
      "step": 55996,
      "training_step_time": 7.725337982177734
    },
    {
      "epoch": 5.340290069580078e-05,
      "model_forward_time": 1.8308510780334473,
      "step": 55997
    },
    {
      "epoch": 5.340290069580078e-05,
      "step": 55997,
      "training_step_time": 7.800235986709595
    },
    {
      "epoch": 5.3403854370117185e-05,
      "model_forward_time": 1.7648580074310303,
      "step": 55998
    },
    {
      "epoch": 5.3403854370117185e-05,
      "step": 55998,
      "training_step_time": 7.998621463775635
    },
    {
      "epoch": 5.3404808044433596e-05,
      "model_forward_time": 1.7466931343078613,
      "step": 55999
    },
    {
      "epoch": 5.3404808044433596e-05,
      "step": 55999,
      "training_step_time": 8.748426675796509
    },
    {
      "epoch": 5.340576171875e-05,
      "grad_norm": 0.10558366030454636,
      "learning_rate": 4.4227410561569254e-06,
      "loss": 0.0536,
      "step": 56000
    },
    {
      "epoch": 5.340576171875e-05,
      "model_forward_time": 1.7762987613677979,
      "step": 56000
    },
    {
      "dynamics_loss_avg": 0.04122835900634527,
      "epoch": 5.340576171875e-05,
      "step": 56000
    },
    {
      "action_loss_avg": 0.0013832495315000415,
      "epoch": 5.340576171875e-05,
      "step": 56000
    },
    {
      "epoch": 5.340576171875e-05,
      "step": 56000,
      "training_step_time": 11.382107734680176
    },
    {
      "epoch": 5.3406715393066404e-05,
      "model_forward_time": 2.845088481903076,
      "step": 56001
    },
    {
      "epoch": 5.3406715393066404e-05,
      "step": 56001,
      "training_step_time": 10.524787664413452
    },
    {
      "epoch": 5.3407669067382815e-05,
      "model_forward_time": 1.3738913536071777,
      "step": 56002
    },
    {
      "epoch": 5.3407669067382815e-05,
      "step": 56002,
      "training_step_time": 7.369394063949585
    },
    {
      "epoch": 5.340862274169922e-05,
      "model_forward_time": 1.7241277694702148,
      "step": 56003
    },
    {
      "epoch": 5.340862274169922e-05,
      "step": 56003,
      "training_step_time": 7.089558839797974
    },
    {
      "epoch": 5.340957641601562e-05,
      "model_forward_time": 1.7347304821014404,
      "step": 56004
    },
    {
      "epoch": 5.340957641601562e-05,
      "step": 56004,
      "training_step_time": 7.738737106323242
    },
    {
      "epoch": 5.3410530090332034e-05,
      "model_forward_time": 1.7534313201904297,
      "step": 56005
    },
    {
      "epoch": 5.3410530090332034e-05,
      "step": 56005,
      "training_step_time": 7.706128835678101
    },
    {
      "epoch": 5.341148376464844e-05,
      "model_forward_time": 1.7239902019500732,
      "step": 56006
    },
    {
      "epoch": 5.341148376464844e-05,
      "step": 56006,
      "training_step_time": 8.706782817840576
    },
    {
      "epoch": 5.341243743896484e-05,
      "model_forward_time": 1.7178988456726074,
      "step": 56007
    },
    {
      "epoch": 5.341243743896484e-05,
      "step": 56007,
      "training_step_time": 8.222388505935669
    },
    {
      "epoch": 5.341339111328125e-05,
      "model_forward_time": 1.7340342998504639,
      "step": 56008
    },
    {
      "epoch": 5.341339111328125e-05,
      "step": 56008,
      "training_step_time": 10.963302850723267
    },
    {
      "epoch": 5.3414344787597657e-05,
      "model_forward_time": 1.742098331451416,
      "step": 56009
    },
    {
      "epoch": 5.3414344787597657e-05,
      "step": 56009,
      "training_step_time": 10.322453260421753
    },
    {
      "epoch": 5.341529846191406e-05,
      "grad_norm": 0.10860647261142731,
      "learning_rate": 4.421098674545693e-06,
      "loss": 0.0574,
      "step": 56010
    },
    {
      "epoch": 5.341529846191406e-05,
      "model_forward_time": 2.895984649658203,
      "step": 56010
    },
    {
      "dynamics_loss_avg": 0.05388868767768144,
      "epoch": 5.341529846191406e-05,
      "step": 56010
    },
    {
      "action_loss_avg": 0.0012638825239264407,
      "epoch": 5.341529846191406e-05,
      "step": 56010
    },
    {
      "epoch": 5.341529846191406e-05,
      "step": 56010,
      "training_step_time": 7.801786422729492
    },
    {
      "epoch": 5.341625213623047e-05,
      "model_forward_time": 1.7380638122558594,
      "step": 56011
    },
    {
      "epoch": 5.341625213623047e-05,
      "step": 56011,
      "training_step_time": 7.649758338928223
    },
    {
      "epoch": 5.3417205810546875e-05,
      "model_forward_time": 1.7387309074401855,
      "step": 56012
    },
    {
      "epoch": 5.3417205810546875e-05,
      "step": 56012,
      "training_step_time": 7.083539009094238
    },
    {
      "epoch": 5.341815948486328e-05,
      "model_forward_time": 1.7387802600860596,
      "step": 56013
    },
    {
      "epoch": 5.341815948486328e-05,
      "step": 56013,
      "training_step_time": 7.756451606750488
    },
    {
      "epoch": 5.341911315917969e-05,
      "model_forward_time": 1.741398811340332,
      "step": 56014
    },
    {
      "epoch": 5.341911315917969e-05,
      "step": 56014,
      "training_step_time": 7.744800090789795
    },
    {
      "epoch": 5.3420066833496094e-05,
      "model_forward_time": 1.8061408996582031,
      "step": 56015
    },
    {
      "epoch": 5.3420066833496094e-05,
      "step": 56015,
      "training_step_time": 9.535308361053467
    },
    {
      "epoch": 5.34210205078125e-05,
      "model_forward_time": 1.747351884841919,
      "step": 56016
    },
    {
      "epoch": 5.34210205078125e-05,
      "step": 56016,
      "training_step_time": 8.184158086776733
    },
    {
      "epoch": 5.342197418212891e-05,
      "model_forward_time": 1.7400727272033691,
      "step": 56017
    },
    {
      "epoch": 5.342197418212891e-05,
      "step": 56017,
      "training_step_time": 10.773755311965942
    },
    {
      "epoch": 5.342292785644531e-05,
      "model_forward_time": 1.7180280685424805,
      "step": 56018
    },
    {
      "epoch": 5.342292785644531e-05,
      "step": 56018,
      "training_step_time": 8.617807388305664
    },
    {
      "epoch": 5.342388153076172e-05,
      "model_forward_time": 2.8982603549957275,
      "step": 56019
    },
    {
      "epoch": 5.342388153076172e-05,
      "step": 56019,
      "training_step_time": 9.308008909225464
    },
    {
      "epoch": 5.342483520507813e-05,
      "grad_norm": 0.0808892697095871,
      "learning_rate": 4.419456356242241e-06,
      "loss": 0.0584,
      "step": 56020
    },
    {
      "epoch": 5.342483520507813e-05,
      "model_forward_time": 1.7346031665802002,
      "step": 56020
    },
    {
      "dynamics_loss_avg": 0.06149564925581217,
      "epoch": 5.342483520507813e-05,
      "step": 56020
    },
    {
      "action_loss_avg": 0.001545488223200664,
      "epoch": 5.342483520507813e-05,
      "step": 56020
    },
    {
      "epoch": 5.342483520507813e-05,
      "step": 56020,
      "training_step_time": 7.671312570571899
    },
    {
      "epoch": 5.342578887939453e-05,
      "model_forward_time": 1.733107089996338,
      "step": 56021
    },
    {
      "epoch": 5.342578887939453e-05,
      "step": 56021,
      "training_step_time": 7.108980655670166
    },
    {
      "epoch": 5.3426742553710936e-05,
      "model_forward_time": 1.7482972145080566,
      "step": 56022
    },
    {
      "epoch": 5.3426742553710936e-05,
      "step": 56022,
      "training_step_time": 6.741368770599365
    },
    {
      "epoch": 5.342769622802735e-05,
      "model_forward_time": 1.741438865661621,
      "step": 56023
    },
    {
      "epoch": 5.342769622802735e-05,
      "step": 56023,
      "training_step_time": 7.71841835975647
    },
    {
      "epoch": 5.342864990234375e-05,
      "model_forward_time": 1.8020122051239014,
      "step": 56024
    },
    {
      "epoch": 5.342864990234375e-05,
      "step": 56024,
      "training_step_time": 8.163000583648682
    },
    {
      "epoch": 5.3429603576660155e-05,
      "model_forward_time": 1.734410047531128,
      "step": 56025
    },
    {
      "epoch": 5.3429603576660155e-05,
      "step": 56025,
      "training_step_time": 8.48361086845398
    },
    {
      "epoch": 5.3430557250976566e-05,
      "model_forward_time": 1.7378454208374023,
      "step": 56026
    },
    {
      "epoch": 5.3430557250976566e-05,
      "step": 56026,
      "training_step_time": 10.093017101287842
    },
    {
      "epoch": 5.343151092529297e-05,
      "model_forward_time": 1.7460613250732422,
      "step": 56027
    },
    {
      "epoch": 5.343151092529297e-05,
      "step": 56027,
      "training_step_time": 8.301009178161621
    },
    {
      "epoch": 5.3432464599609374e-05,
      "model_forward_time": 1.734588384628296,
      "step": 56028
    },
    {
      "epoch": 5.3432464599609374e-05,
      "step": 56028,
      "training_step_time": 7.764044523239136
    },
    {
      "epoch": 5.3433418273925785e-05,
      "model_forward_time": 2.8706109523773193,
      "step": 56029
    },
    {
      "epoch": 5.3433418273925785e-05,
      "step": 56029,
      "training_step_time": 8.756539821624756
    },
    {
      "epoch": 5.343437194824219e-05,
      "grad_norm": 0.11818587779998779,
      "learning_rate": 4.417814101426169e-06,
      "loss": 0.059,
      "step": 56030
    },
    {
      "epoch": 5.343437194824219e-05,
      "model_forward_time": 1.7326724529266357,
      "step": 56030
    },
    {
      "dynamics_loss_avg": 0.052701893635094164,
      "epoch": 5.343437194824219e-05,
      "step": 56030
    },
    {
      "action_loss_avg": 0.0010734545503510162,
      "epoch": 5.343437194824219e-05,
      "step": 56030
    },
    {
      "epoch": 5.343437194824219e-05,
      "step": 56030,
      "training_step_time": 6.678077459335327
    },
    {
      "epoch": 5.343532562255859e-05,
      "model_forward_time": 1.3635694980621338,
      "step": 56031
    },
    {
      "epoch": 5.343532562255859e-05,
      "step": 56031,
      "training_step_time": 7.0460615158081055
    },
    {
      "epoch": 5.3436279296875e-05,
      "model_forward_time": 1.7360551357269287,
      "step": 56032
    },
    {
      "epoch": 5.3436279296875e-05,
      "step": 56032,
      "training_step_time": 7.722258806228638
    },
    {
      "epoch": 5.343723297119141e-05,
      "model_forward_time": 1.8091721534729004,
      "step": 56033
    },
    {
      "epoch": 5.343723297119141e-05,
      "step": 56033,
      "training_step_time": 6.984151363372803
    },
    {
      "epoch": 5.343818664550781e-05,
      "model_forward_time": 1.7446532249450684,
      "step": 56034
    },
    {
      "epoch": 5.343818664550781e-05,
      "step": 56034,
      "training_step_time": 8.201829433441162
    },
    {
      "epoch": 5.3439140319824215e-05,
      "model_forward_time": 1.7354817390441895,
      "step": 56035
    },
    {
      "epoch": 5.3439140319824215e-05,
      "step": 56035,
      "training_step_time": 7.651048183441162
    },
    {
      "epoch": 5.3440093994140626e-05,
      "model_forward_time": 1.741140604019165,
      "step": 56036
    },
    {
      "epoch": 5.3440093994140626e-05,
      "step": 56036,
      "training_step_time": 8.791197538375854
    },
    {
      "epoch": 5.344104766845703e-05,
      "model_forward_time": 1.734410047531128,
      "step": 56037
    },
    {
      "epoch": 5.344104766845703e-05,
      "step": 56037,
      "training_step_time": 10.240050315856934
    },
    {
      "epoch": 5.3442001342773434e-05,
      "model_forward_time": 2.9659087657928467,
      "step": 56038
    },
    {
      "epoch": 5.3442001342773434e-05,
      "step": 56038,
      "training_step_time": 7.765082359313965
    },
    {
      "epoch": 5.3442955017089845e-05,
      "model_forward_time": 1.7329535484313965,
      "step": 56039
    },
    {
      "epoch": 5.3442955017089845e-05,
      "step": 56039,
      "training_step_time": 7.559188604354858
    },
    {
      "epoch": 5.344390869140625e-05,
      "grad_norm": 0.1109001636505127,
      "learning_rate": 4.416171910277071e-06,
      "loss": 0.0535,
      "step": 56040
    },
    {
      "epoch": 5.344390869140625e-05,
      "model_forward_time": 1.7401342391967773,
      "step": 56040
    },
    {
      "dynamics_loss_avg": 0.04533042050898075,
      "epoch": 5.344390869140625e-05,
      "step": 56040
    },
    {
      "action_loss_avg": 0.0017671122273895889,
      "epoch": 5.344390869140625e-05,
      "step": 56040
    },
    {
      "epoch": 5.344390869140625e-05,
      "step": 56040,
      "training_step_time": 7.0311503410339355
    },
    {
      "epoch": 5.344486236572265e-05,
      "model_forward_time": 1.7354471683502197,
      "step": 56041
    },
    {
      "epoch": 5.344486236572265e-05,
      "step": 56041,
      "training_step_time": 7.632919073104858
    },
    {
      "epoch": 5.3445816040039064e-05,
      "model_forward_time": 1.726985216140747,
      "step": 56042
    },
    {
      "epoch": 5.3445816040039064e-05,
      "step": 56042,
      "training_step_time": 7.5685436725616455
    },
    {
      "epoch": 5.344676971435547e-05,
      "model_forward_time": 1.744652271270752,
      "step": 56043
    },
    {
      "epoch": 5.344676971435547e-05,
      "step": 56043,
      "training_step_time": 8.187775135040283
    },
    {
      "epoch": 5.344772338867187e-05,
      "model_forward_time": 1.7492218017578125,
      "step": 56044
    },
    {
      "epoch": 5.344772338867187e-05,
      "step": 56044,
      "training_step_time": 8.983602046966553
    },
    {
      "epoch": 5.344867706298828e-05,
      "model_forward_time": 1.7328367233276367,
      "step": 56045
    },
    {
      "epoch": 5.344867706298828e-05,
      "step": 56045,
      "training_step_time": 8.811599254608154
    },
    {
      "epoch": 5.344963073730469e-05,
      "model_forward_time": 1.7401392459869385,
      "step": 56046
    },
    {
      "epoch": 5.344963073730469e-05,
      "step": 56046,
      "training_step_time": 7.706021547317505
    },
    {
      "epoch": 5.345058441162109e-05,
      "model_forward_time": 2.910857677459717,
      "step": 56047
    },
    {
      "epoch": 5.345058441162109e-05,
      "step": 56047,
      "training_step_time": 8.289182901382446
    },
    {
      "epoch": 5.34515380859375e-05,
      "model_forward_time": 1.7263867855072021,
      "step": 56048
    },
    {
      "epoch": 5.34515380859375e-05,
      "step": 56048,
      "training_step_time": 7.589507341384888
    },
    {
      "epoch": 5.3452491760253906e-05,
      "model_forward_time": 1.7361533641815186,
      "step": 56049
    },
    {
      "epoch": 5.3452491760253906e-05,
      "step": 56049,
      "training_step_time": 7.037877082824707
    },
    {
      "epoch": 5.345344543457031e-05,
      "grad_norm": 0.12395588308572769,
      "learning_rate": 4.414529782974533e-06,
      "loss": 0.0587,
      "step": 56050
    },
    {
      "epoch": 5.345344543457031e-05,
      "model_forward_time": 1.7286818027496338,
      "step": 56050
    },
    {
      "dynamics_loss_avg": 0.057808294892311096,
      "epoch": 5.345344543457031e-05,
      "step": 56050
    },
    {
      "action_loss_avg": 0.0018356177082750947,
      "epoch": 5.345344543457031e-05,
      "step": 56050
    },
    {
      "epoch": 5.345344543457031e-05,
      "step": 56050,
      "training_step_time": 7.035438776016235
    },
    {
      "epoch": 5.345439910888672e-05,
      "model_forward_time": 1.7307543754577637,
      "step": 56051
    },
    {
      "epoch": 5.345439910888672e-05,
      "step": 56051,
      "training_step_time": 7.6509034633636475
    },
    {
      "epoch": 5.3455352783203125e-05,
      "model_forward_time": 1.7457275390625,
      "step": 56052
    },
    {
      "epoch": 5.3455352783203125e-05,
      "step": 56052,
      "training_step_time": 8.428776741027832
    },
    {
      "epoch": 5.345630645751953e-05,
      "model_forward_time": 1.7280335426330566,
      "step": 56053
    },
    {
      "epoch": 5.345630645751953e-05,
      "step": 56053,
      "training_step_time": 7.56344199180603
    },
    {
      "epoch": 5.345726013183594e-05,
      "model_forward_time": 1.7336652278900146,
      "step": 56054
    },
    {
      "epoch": 5.345726013183594e-05,
      "step": 56054,
      "training_step_time": 8.556981325149536
    },
    {
      "epoch": 5.3458213806152343e-05,
      "model_forward_time": 1.7565035820007324,
      "step": 56055
    },
    {
      "epoch": 5.3458213806152343e-05,
      "step": 56055,
      "training_step_time": 7.730000257492065
    },
    {
      "epoch": 5.345916748046875e-05,
      "model_forward_time": 2.9018619060516357,
      "step": 56056
    },
    {
      "epoch": 5.345916748046875e-05,
      "step": 56056,
      "training_step_time": 8.266865968704224
    },
    {
      "epoch": 5.346012115478516e-05,
      "model_forward_time": 1.7325918674468994,
      "step": 56057
    },
    {
      "epoch": 5.346012115478516e-05,
      "step": 56057,
      "training_step_time": 7.633896350860596
    },
    {
      "epoch": 5.346107482910156e-05,
      "model_forward_time": 1.7353429794311523,
      "step": 56058
    },
    {
      "epoch": 5.346107482910156e-05,
      "step": 56058,
      "training_step_time": 7.06340479850769
    },
    {
      "epoch": 5.3462028503417966e-05,
      "model_forward_time": 1.7355010509490967,
      "step": 56059
    },
    {
      "epoch": 5.3462028503417966e-05,
      "step": 56059,
      "training_step_time": 6.7106099128723145
    },
    {
      "epoch": 5.346298217773438e-05,
      "grad_norm": 0.09914976358413696,
      "learning_rate": 4.41288771969814e-06,
      "loss": 0.0576,
      "step": 56060
    },
    {
      "epoch": 5.346298217773438e-05,
      "model_forward_time": 1.7595694065093994,
      "step": 56060
    },
    {
      "dynamics_loss_avg": 0.05845730118453503,
      "epoch": 5.346298217773438e-05,
      "step": 56060
    },
    {
      "action_loss_avg": 0.002028127096127719,
      "epoch": 5.346298217773438e-05,
      "step": 56060
    },
    {
      "epoch": 5.346298217773438e-05,
      "step": 56060,
      "training_step_time": 7.654686212539673
    },
    {
      "epoch": 5.346393585205078e-05,
      "model_forward_time": 1.7500977516174316,
      "step": 56061
    },
    {
      "epoch": 5.346393585205078e-05,
      "step": 56061,
      "training_step_time": 8.17276382446289
    },
    {
      "epoch": 5.3464889526367185e-05,
      "model_forward_time": 1.7358684539794922,
      "step": 56062
    },
    {
      "epoch": 5.3464889526367185e-05,
      "step": 56062,
      "training_step_time": 7.420426845550537
    },
    {
      "epoch": 5.3465843200683596e-05,
      "model_forward_time": 1.7318072319030762,
      "step": 56063
    },
    {
      "epoch": 5.3465843200683596e-05,
      "step": 56063,
      "training_step_time": 8.288034915924072
    },
    {
      "epoch": 5.3466796875e-05,
      "model_forward_time": 1.7411065101623535,
      "step": 56064
    },
    {
      "epoch": 5.3466796875e-05,
      "step": 56064,
      "training_step_time": 7.604177474975586
    },
    {
      "epoch": 5.3467750549316404e-05,
      "model_forward_time": 2.869000196456909,
      "step": 56065
    },
    {
      "epoch": 5.3467750549316404e-05,
      "step": 56065,
      "training_step_time": 7.729799032211304
    },
    {
      "epoch": 5.3468704223632815e-05,
      "model_forward_time": 1.727722406387329,
      "step": 56066
    },
    {
      "epoch": 5.3468704223632815e-05,
      "step": 56066,
      "training_step_time": 7.60832142829895
    },
    {
      "epoch": 5.346965789794922e-05,
      "model_forward_time": 1.7258069515228271,
      "step": 56067
    },
    {
      "epoch": 5.346965789794922e-05,
      "step": 56067,
      "training_step_time": 7.082315921783447
    },
    {
      "epoch": 5.347061157226562e-05,
      "model_forward_time": 1.7349636554718018,
      "step": 56068
    },
    {
      "epoch": 5.347061157226562e-05,
      "step": 56068,
      "training_step_time": 6.728176832199097
    },
    {
      "epoch": 5.3471565246582034e-05,
      "model_forward_time": 1.7428326606750488,
      "step": 56069
    },
    {
      "epoch": 5.3471565246582034e-05,
      "step": 56069,
      "training_step_time": 7.657548427581787
    },
    {
      "epoch": 5.347251892089844e-05,
      "grad_norm": 0.09487634897232056,
      "learning_rate": 4.411245720627464e-06,
      "loss": 0.0551,
      "step": 56070
    },
    {
      "epoch": 5.347251892089844e-05,
      "model_forward_time": 1.7413039207458496,
      "step": 56070
    },
    {
      "dynamics_loss_avg": 0.05666047837585211,
      "epoch": 5.347251892089844e-05,
      "step": 56070
    },
    {
      "action_loss_avg": 0.0019420792756136506,
      "epoch": 5.347251892089844e-05,
      "step": 56070
    },
    {
      "epoch": 5.347251892089844e-05,
      "step": 56070,
      "training_step_time": 7.657350540161133
    },
    {
      "epoch": 5.347347259521484e-05,
      "model_forward_time": 1.7312822341918945,
      "step": 56071
    },
    {
      "epoch": 5.347347259521484e-05,
      "step": 56071,
      "training_step_time": 7.608714580535889
    },
    {
      "epoch": 5.347442626953125e-05,
      "model_forward_time": 1.7448594570159912,
      "step": 56072
    },
    {
      "epoch": 5.347442626953125e-05,
      "step": 56072,
      "training_step_time": 7.767068386077881
    },
    {
      "epoch": 5.347537994384766e-05,
      "model_forward_time": 1.752859354019165,
      "step": 56073
    },
    {
      "epoch": 5.347537994384766e-05,
      "step": 56073,
      "training_step_time": 8.249495029449463
    },
    {
      "epoch": 5.347633361816406e-05,
      "model_forward_time": 2.8969573974609375,
      "step": 56074
    },
    {
      "epoch": 5.347633361816406e-05,
      "step": 56074,
      "training_step_time": 7.759697675704956
    },
    {
      "epoch": 5.347728729248047e-05,
      "model_forward_time": 1.7444689273834229,
      "step": 56075
    },
    {
      "epoch": 5.347728729248047e-05,
      "step": 56075,
      "training_step_time": 7.749709129333496
    },
    {
      "epoch": 5.3478240966796875e-05,
      "model_forward_time": 1.743535041809082,
      "step": 56076
    },
    {
      "epoch": 5.3478240966796875e-05,
      "step": 56076,
      "training_step_time": 7.119266510009766
    },
    {
      "epoch": 5.347919464111328e-05,
      "model_forward_time": 1.7491517066955566,
      "step": 56077
    },
    {
      "epoch": 5.347919464111328e-05,
      "step": 56077,
      "training_step_time": 7.144113779067993
    },
    {
      "epoch": 5.348014831542969e-05,
      "model_forward_time": 1.7404682636260986,
      "step": 56078
    },
    {
      "epoch": 5.348014831542969e-05,
      "step": 56078,
      "training_step_time": 7.703772306442261
    },
    {
      "epoch": 5.3481101989746094e-05,
      "model_forward_time": 1.760239601135254,
      "step": 56079
    },
    {
      "epoch": 5.3481101989746094e-05,
      "step": 56079,
      "training_step_time": 7.53385591506958
    },
    {
      "epoch": 5.34820556640625e-05,
      "grad_norm": 0.0821024551987648,
      "learning_rate": 4.409603785942071e-06,
      "loss": 0.0583,
      "step": 56080
    },
    {
      "epoch": 5.34820556640625e-05,
      "model_forward_time": 1.7477140426635742,
      "step": 56080
    },
    {
      "dynamics_loss_avg": 0.06181788127869368,
      "epoch": 5.34820556640625e-05,
      "step": 56080
    },
    {
      "action_loss_avg": 0.001468976031173952,
      "epoch": 5.34820556640625e-05,
      "step": 56080
    },
    {
      "epoch": 5.34820556640625e-05,
      "step": 56080,
      "training_step_time": 7.666479825973511
    },
    {
      "epoch": 5.348300933837891e-05,
      "model_forward_time": 1.7443621158599854,
      "step": 56081
    },
    {
      "epoch": 5.348300933837891e-05,
      "step": 56081,
      "training_step_time": 7.723124980926514
    },
    {
      "epoch": 5.348396301269531e-05,
      "model_forward_time": 1.7493994235992432,
      "step": 56082
    },
    {
      "epoch": 5.348396301269531e-05,
      "step": 56082,
      "training_step_time": 8.585792064666748
    },
    {
      "epoch": 5.348491668701172e-05,
      "model_forward_time": 1.7717783451080322,
      "step": 56083
    },
    {
      "epoch": 5.348491668701172e-05,
      "step": 56083,
      "training_step_time": 9.594765901565552
    },
    {
      "epoch": 5.348587036132813e-05,
      "model_forward_time": 2.9661448001861572,
      "step": 56084
    },
    {
      "epoch": 5.348587036132813e-05,
      "step": 56084,
      "training_step_time": 7.743376970291138
    },
    {
      "epoch": 5.348682403564453e-05,
      "model_forward_time": 1.7427399158477783,
      "step": 56085
    },
    {
      "epoch": 5.348682403564453e-05,
      "step": 56085,
      "training_step_time": 7.581149578094482
    },
    {
      "epoch": 5.3487777709960936e-05,
      "model_forward_time": 1.7452247142791748,
      "step": 56086
    },
    {
      "epoch": 5.3487777709960936e-05,
      "step": 56086,
      "training_step_time": 7.129711151123047
    },
    {
      "epoch": 5.348873138427735e-05,
      "model_forward_time": 1.7527530193328857,
      "step": 56087
    },
    {
      "epoch": 5.348873138427735e-05,
      "step": 56087,
      "training_step_time": 7.616194486618042
    },
    {
      "epoch": 5.348968505859375e-05,
      "model_forward_time": 1.7358922958374023,
      "step": 56088
    },
    {
      "epoch": 5.348968505859375e-05,
      "step": 56088,
      "training_step_time": 6.753850221633911
    },
    {
      "epoch": 5.3490638732910155e-05,
      "model_forward_time": 1.3996508121490479,
      "step": 56089
    },
    {
      "epoch": 5.3490638732910155e-05,
      "step": 56089,
      "training_step_time": 7.58808445930481
    },
    {
      "epoch": 5.3491592407226566e-05,
      "grad_norm": 0.10525039583444595,
      "learning_rate": 4.4079619158215195e-06,
      "loss": 0.0577,
      "step": 56090
    },
    {
      "epoch": 5.3491592407226566e-05,
      "model_forward_time": 1.378610610961914,
      "step": 56090
    },
    {
      "dynamics_loss_avg": 0.05335639677941799,
      "epoch": 5.3491592407226566e-05,
      "step": 56090
    },
    {
      "action_loss_avg": 0.0024170080199837685,
      "epoch": 5.3491592407226566e-05,
      "step": 56090
    },
    {
      "epoch": 5.3491592407226566e-05,
      "step": 56090,
      "training_step_time": 8.427115678787231
    },
    {
      "epoch": 5.349254608154297e-05,
      "model_forward_time": 1.7434062957763672,
      "step": 56091
    },
    {
      "epoch": 5.349254608154297e-05,
      "step": 56091,
      "training_step_time": 9.326844930648804
    },
    {
      "epoch": 5.3493499755859374e-05,
      "model_forward_time": 1.7777881622314453,
      "step": 56092
    },
    {
      "epoch": 5.3493499755859374e-05,
      "step": 56092,
      "training_step_time": 7.747464179992676
    },
    {
      "epoch": 5.3494453430175785e-05,
      "model_forward_time": 2.874887228012085,
      "step": 56093
    },
    {
      "epoch": 5.3494453430175785e-05,
      "step": 56093,
      "training_step_time": 7.653401851654053
    },
    {
      "epoch": 5.349540710449219e-05,
      "model_forward_time": 1.740631341934204,
      "step": 56094
    },
    {
      "epoch": 5.349540710449219e-05,
      "step": 56094,
      "training_step_time": 7.552516937255859
    },
    {
      "epoch": 5.349636077880859e-05,
      "model_forward_time": 1.7433500289916992,
      "step": 56095
    },
    {
      "epoch": 5.349636077880859e-05,
      "step": 56095,
      "training_step_time": 7.144912242889404
    },
    {
      "epoch": 5.3497314453125e-05,
      "model_forward_time": 1.7579078674316406,
      "step": 56096
    },
    {
      "epoch": 5.3497314453125e-05,
      "step": 56096,
      "training_step_time": 7.608482837677002
    },
    {
      "epoch": 5.349826812744141e-05,
      "model_forward_time": 1.7419154644012451,
      "step": 56097
    },
    {
      "epoch": 5.349826812744141e-05,
      "step": 56097,
      "training_step_time": 7.79343056678772
    },
    {
      "epoch": 5.349922180175781e-05,
      "model_forward_time": 1.7576308250427246,
      "step": 56098
    },
    {
      "epoch": 5.349922180175781e-05,
      "step": 56098,
      "training_step_time": 7.680337905883789
    },
    {
      "epoch": 5.3500175476074216e-05,
      "model_forward_time": 1.7507219314575195,
      "step": 56099
    },
    {
      "epoch": 5.3500175476074216e-05,
      "step": 56099,
      "training_step_time": 8.691309452056885
    },
    {
      "epoch": 5.3501129150390626e-05,
      "grad_norm": 0.11037842184305191,
      "learning_rate": 4.406320110445362e-06,
      "loss": 0.0571,
      "step": 56100
    },
    {
      "epoch": 5.3501129150390626e-05,
      "model_forward_time": 1.746474027633667,
      "step": 56100
    },
    {
      "dynamics_loss_avg": 0.043836557678878306,
      "epoch": 5.3501129150390626e-05,
      "step": 56100
    },
    {
      "action_loss_avg": 0.0020262479898519813,
      "epoch": 5.3501129150390626e-05,
      "step": 56100
    },
    {
      "epoch": 5.3501129150390626e-05,
      "step": 56100,
      "training_step_time": 7.864171743392944
    },
    {
      "epoch": 5.350208282470703e-05,
      "model_forward_time": 1.7616546154022217,
      "step": 56101
    },
    {
      "epoch": 5.350208282470703e-05,
      "step": 56101,
      "training_step_time": 7.787508249282837
    },
    {
      "epoch": 5.3503036499023434e-05,
      "model_forward_time": 2.8608815670013428,
      "step": 56102
    },
    {
      "epoch": 5.3503036499023434e-05,
      "step": 56102,
      "training_step_time": 8.617098331451416
    },
    {
      "epoch": 5.3503990173339845e-05,
      "model_forward_time": 1.742072582244873,
      "step": 56103
    },
    {
      "epoch": 5.3503990173339845e-05,
      "step": 56103,
      "training_step_time": 7.636563062667847
    },
    {
      "epoch": 5.350494384765625e-05,
      "model_forward_time": 1.7323777675628662,
      "step": 56104
    },
    {
      "epoch": 5.350494384765625e-05,
      "step": 56104,
      "training_step_time": 7.079419851303101
    },
    {
      "epoch": 5.350589752197265e-05,
      "model_forward_time": 1.7441051006317139,
      "step": 56105
    },
    {
      "epoch": 5.350589752197265e-05,
      "step": 56105,
      "training_step_time": 7.137340307235718
    },
    {
      "epoch": 5.3506851196289064e-05,
      "model_forward_time": 1.7260653972625732,
      "step": 56106
    },
    {
      "epoch": 5.3506851196289064e-05,
      "step": 56106,
      "training_step_time": 7.7228758335113525
    },
    {
      "epoch": 5.350780487060547e-05,
      "model_forward_time": 1.761735439300537,
      "step": 56107
    },
    {
      "epoch": 5.350780487060547e-05,
      "step": 56107,
      "training_step_time": 8.129737854003906
    },
    {
      "epoch": 5.350875854492187e-05,
      "model_forward_time": 1.7385377883911133,
      "step": 56108
    },
    {
      "epoch": 5.350875854492187e-05,
      "step": 56108,
      "training_step_time": 8.563390254974365
    },
    {
      "epoch": 5.350971221923828e-05,
      "model_forward_time": 1.7416179180145264,
      "step": 56109
    },
    {
      "epoch": 5.350971221923828e-05,
      "step": 56109,
      "training_step_time": 9.589813947677612
    },
    {
      "epoch": 5.351066589355469e-05,
      "grad_norm": 0.11122553050518036,
      "learning_rate": 4.404678369993144e-06,
      "loss": 0.0566,
      "step": 56110
    },
    {
      "epoch": 5.351066589355469e-05,
      "model_forward_time": 1.736490249633789,
      "step": 56110
    },
    {
      "dynamics_loss_avg": 0.061348700895905495,
      "epoch": 5.351066589355469e-05,
      "step": 56110
    },
    {
      "action_loss_avg": 0.0013027070905081927,
      "epoch": 5.351066589355469e-05,
      "step": 56110
    },
    {
      "epoch": 5.351066589355469e-05,
      "step": 56110,
      "training_step_time": 8.656016111373901
    },
    {
      "epoch": 5.351161956787109e-05,
      "model_forward_time": 1.7387917041778564,
      "step": 56111
    },
    {
      "epoch": 5.351161956787109e-05,
      "step": 56111,
      "training_step_time": 8.804283380508423
    },
    {
      "epoch": 5.35125732421875e-05,
      "model_forward_time": 1.7399630546569824,
      "step": 56112
    },
    {
      "epoch": 5.35125732421875e-05,
      "step": 56112,
      "training_step_time": 8.756460666656494
    },
    {
      "epoch": 5.3513526916503906e-05,
      "model_forward_time": 1.7400927543640137,
      "step": 56113
    },
    {
      "epoch": 5.3513526916503906e-05,
      "step": 56113,
      "training_step_time": 7.217107057571411
    },
    {
      "epoch": 5.351448059082031e-05,
      "model_forward_time": 1.7380571365356445,
      "step": 56114
    },
    {
      "epoch": 5.351448059082031e-05,
      "step": 56114,
      "training_step_time": 8.161184549331665
    },
    {
      "epoch": 5.351543426513672e-05,
      "model_forward_time": 1.7395987510681152,
      "step": 56115
    },
    {
      "epoch": 5.351543426513672e-05,
      "step": 56115,
      "training_step_time": 7.773165941238403
    },
    {
      "epoch": 5.3516387939453125e-05,
      "model_forward_time": 1.7432756423950195,
      "step": 56116
    },
    {
      "epoch": 5.3516387939453125e-05,
      "step": 56116,
      "training_step_time": 8.235089302062988
    },
    {
      "epoch": 5.351734161376953e-05,
      "model_forward_time": 1.7492256164550781,
      "step": 56117
    },
    {
      "epoch": 5.351734161376953e-05,
      "step": 56117,
      "training_step_time": 7.717559337615967
    },
    {
      "epoch": 5.351829528808594e-05,
      "model_forward_time": 1.741457223892212,
      "step": 56118
    },
    {
      "epoch": 5.351829528808594e-05,
      "step": 56118,
      "training_step_time": 7.649882555007935
    },
    {
      "epoch": 5.3519248962402344e-05,
      "model_forward_time": 1.7398748397827148,
      "step": 56119
    },
    {
      "epoch": 5.3519248962402344e-05,
      "step": 56119,
      "training_step_time": 7.789834260940552
    },
    {
      "epoch": 5.352020263671875e-05,
      "grad_norm": 0.12311337888240814,
      "learning_rate": 4.403036694644406e-06,
      "loss": 0.0562,
      "step": 56120
    },
    {
      "epoch": 5.352020263671875e-05,
      "model_forward_time": 1.7531318664550781,
      "step": 56120
    },
    {
      "dynamics_loss_avg": 0.061644263379275796,
      "epoch": 5.352020263671875e-05,
      "step": 56120
    },
    {
      "action_loss_avg": 0.002082744843210094,
      "epoch": 5.352020263671875e-05,
      "step": 56120
    },
    {
      "epoch": 5.352020263671875e-05,
      "step": 56120,
      "training_step_time": 7.684555292129517
    },
    {
      "epoch": 5.352115631103516e-05,
      "model_forward_time": 2.9255001544952393,
      "step": 56121
    },
    {
      "epoch": 5.352115631103516e-05,
      "step": 56121,
      "training_step_time": 7.735388278961182
    },
    {
      "epoch": 5.352210998535156e-05,
      "model_forward_time": 1.740870714187622,
      "step": 56122
    },
    {
      "epoch": 5.352210998535156e-05,
      "step": 56122,
      "training_step_time": 7.587198495864868
    },
    {
      "epoch": 5.3523063659667966e-05,
      "model_forward_time": 1.7437283992767334,
      "step": 56123
    },
    {
      "epoch": 5.3523063659667966e-05,
      "step": 56123,
      "training_step_time": 7.029227018356323
    },
    {
      "epoch": 5.352401733398438e-05,
      "model_forward_time": 1.738255500793457,
      "step": 56124
    },
    {
      "epoch": 5.352401733398438e-05,
      "step": 56124,
      "training_step_time": 7.665371656417847
    },
    {
      "epoch": 5.352497100830078e-05,
      "model_forward_time": 1.7378971576690674,
      "step": 56125
    },
    {
      "epoch": 5.352497100830078e-05,
      "step": 56125,
      "training_step_time": 7.627285003662109
    },
    {
      "epoch": 5.3525924682617185e-05,
      "model_forward_time": 1.754032850265503,
      "step": 56126
    },
    {
      "epoch": 5.3525924682617185e-05,
      "step": 56126,
      "training_step_time": 8.353957176208496
    },
    {
      "epoch": 5.3526878356933596e-05,
      "model_forward_time": 1.7359919548034668,
      "step": 56127
    },
    {
      "epoch": 5.3526878356933596e-05,
      "step": 56127,
      "training_step_time": 9.23346471786499
    },
    {
      "epoch": 5.352783203125e-05,
      "model_forward_time": 1.740687370300293,
      "step": 56128
    },
    {
      "epoch": 5.352783203125e-05,
      "step": 56128,
      "training_step_time": 8.35187840461731
    },
    {
      "epoch": 5.3528785705566404e-05,
      "model_forward_time": 1.7668404579162598,
      "step": 56129
    },
    {
      "epoch": 5.3528785705566404e-05,
      "step": 56129,
      "training_step_time": 7.722783803939819
    },
    {
      "epoch": 5.3529739379882815e-05,
      "grad_norm": 0.15272736549377441,
      "learning_rate": 4.401395084578677e-06,
      "loss": 0.0566,
      "step": 56130
    },
    {
      "epoch": 5.3529739379882815e-05,
      "model_forward_time": 2.8890931606292725,
      "step": 56130
    },
    {
      "dynamics_loss_avg": 0.06988836377859116,
      "epoch": 5.3529739379882815e-05,
      "step": 56130
    },
    {
      "action_loss_avg": 0.0018633196712471544,
      "epoch": 5.3529739379882815e-05,
      "step": 56130
    },
    {
      "epoch": 5.3529739379882815e-05,
      "step": 56130,
      "training_step_time": 7.719943284988403
    },
    {
      "epoch": 5.353069305419922e-05,
      "model_forward_time": 1.7388546466827393,
      "step": 56131
    },
    {
      "epoch": 5.353069305419922e-05,
      "step": 56131,
      "training_step_time": 7.660193920135498
    },
    {
      "epoch": 5.353164672851562e-05,
      "model_forward_time": 1.7420763969421387,
      "step": 56132
    },
    {
      "epoch": 5.353164672851562e-05,
      "step": 56132,
      "training_step_time": 6.797139406204224
    },
    {
      "epoch": 5.3532600402832034e-05,
      "model_forward_time": 1.7408103942871094,
      "step": 56133
    },
    {
      "epoch": 5.3532600402832034e-05,
      "step": 56133,
      "training_step_time": 8.217785120010376
    },
    {
      "epoch": 5.353355407714844e-05,
      "model_forward_time": 1.7427561283111572,
      "step": 56134
    },
    {
      "epoch": 5.353355407714844e-05,
      "step": 56134,
      "training_step_time": 7.676580429077148
    },
    {
      "epoch": 5.353450775146484e-05,
      "model_forward_time": 1.8105547428131104,
      "step": 56135
    },
    {
      "epoch": 5.353450775146484e-05,
      "step": 56135,
      "training_step_time": 7.733584642410278
    },
    {
      "epoch": 5.353546142578125e-05,
      "model_forward_time": 1.7480058670043945,
      "step": 56136
    },
    {
      "epoch": 5.353546142578125e-05,
      "step": 56136,
      "training_step_time": 8.342242002487183
    },
    {
      "epoch": 5.353641510009766e-05,
      "model_forward_time": 1.7449285984039307,
      "step": 56137
    },
    {
      "epoch": 5.353641510009766e-05,
      "step": 56137,
      "training_step_time": 8.561774253845215
    },
    {
      "epoch": 5.353736877441406e-05,
      "model_forward_time": 1.7430815696716309,
      "step": 56138
    },
    {
      "epoch": 5.353736877441406e-05,
      "step": 56138,
      "training_step_time": 9.444626569747925
    },
    {
      "epoch": 5.353832244873047e-05,
      "model_forward_time": 2.93992018699646,
      "step": 56139
    },
    {
      "epoch": 5.353832244873047e-05,
      "step": 56139,
      "training_step_time": 8.764901399612427
    },
    {
      "epoch": 5.3539276123046876e-05,
      "grad_norm": 0.1191585436463356,
      "learning_rate": 4.399753539975482e-06,
      "loss": 0.0558,
      "step": 56140
    },
    {
      "epoch": 5.3539276123046876e-05,
      "model_forward_time": 1.7414300441741943,
      "step": 56140
    },
    {
      "dynamics_loss_avg": 0.04521079398691654,
      "epoch": 5.3539276123046876e-05,
      "step": 56140
    },
    {
      "action_loss_avg": 0.0016357101238099858,
      "epoch": 5.3539276123046876e-05,
      "step": 56140
    },
    {
      "epoch": 5.3539276123046876e-05,
      "step": 56140,
      "training_step_time": 7.705938816070557
    },
    {
      "epoch": 5.354022979736328e-05,
      "model_forward_time": 1.7476696968078613,
      "step": 56141
    },
    {
      "epoch": 5.354022979736328e-05,
      "step": 56141,
      "training_step_time": 7.020970821380615
    },
    {
      "epoch": 5.354118347167969e-05,
      "model_forward_time": 1.750952959060669,
      "step": 56142
    },
    {
      "epoch": 5.354118347167969e-05,
      "step": 56142,
      "training_step_time": 7.1266560554504395
    },
    {
      "epoch": 5.3542137145996094e-05,
      "model_forward_time": 1.7489392757415771,
      "step": 56143
    },
    {
      "epoch": 5.3542137145996094e-05,
      "step": 56143,
      "training_step_time": 7.623156309127808
    },
    {
      "epoch": 5.35430908203125e-05,
      "model_forward_time": 1.7683897018432617,
      "step": 56144
    },
    {
      "epoch": 5.35430908203125e-05,
      "step": 56144,
      "training_step_time": 7.697547435760498
    },
    {
      "epoch": 5.354404449462891e-05,
      "model_forward_time": 1.7479584217071533,
      "step": 56145
    },
    {
      "epoch": 5.354404449462891e-05,
      "step": 56145,
      "training_step_time": 9.203541278839111
    },
    {
      "epoch": 5.354499816894531e-05,
      "model_forward_time": 1.7490942478179932,
      "step": 56146
    },
    {
      "epoch": 5.354499816894531e-05,
      "step": 56146,
      "training_step_time": 8.302149772644043
    },
    {
      "epoch": 5.354595184326172e-05,
      "model_forward_time": 1.7493164539337158,
      "step": 56147
    },
    {
      "epoch": 5.354595184326172e-05,
      "step": 56147,
      "training_step_time": 9.510880947113037
    },
    {
      "epoch": 5.354690551757813e-05,
      "model_forward_time": 2.9398155212402344,
      "step": 56148
    },
    {
      "epoch": 5.354690551757813e-05,
      "step": 56148,
      "training_step_time": 7.7461206912994385
    },
    {
      "epoch": 5.354785919189453e-05,
      "model_forward_time": 1.7448492050170898,
      "step": 56149
    },
    {
      "epoch": 5.354785919189453e-05,
      "step": 56149,
      "training_step_time": 8.248421907424927
    },
    {
      "epoch": 5.3548812866210936e-05,
      "grad_norm": 0.15800151228904724,
      "learning_rate": 4.398112061014338e-06,
      "loss": 0.0573,
      "step": 56150
    },
    {
      "epoch": 5.3548812866210936e-05,
      "model_forward_time": 1.7575244903564453,
      "step": 56150
    },
    {
      "dynamics_loss_avg": 0.06482945643365383,
      "epoch": 5.3548812866210936e-05,
      "step": 56150
    },
    {
      "action_loss_avg": 0.0017953039437998085,
      "epoch": 5.3548812866210936e-05,
      "step": 56150
    },
    {
      "epoch": 5.3548812866210936e-05,
      "step": 56150,
      "training_step_time": 7.598998546600342
    },
    {
      "epoch": 5.354976654052735e-05,
      "model_forward_time": 1.7436034679412842,
      "step": 56151
    },
    {
      "epoch": 5.354976654052735e-05,
      "step": 56151,
      "training_step_time": 7.200292110443115
    },
    {
      "epoch": 5.355072021484375e-05,
      "model_forward_time": 1.7539598941802979,
      "step": 56152
    },
    {
      "epoch": 5.355072021484375e-05,
      "step": 56152,
      "training_step_time": 7.7183122634887695
    },
    {
      "epoch": 5.3551673889160155e-05,
      "model_forward_time": 1.749133825302124,
      "step": 56153
    },
    {
      "epoch": 5.3551673889160155e-05,
      "step": 56153,
      "training_step_time": 7.740664720535278
    },
    {
      "epoch": 5.3552627563476566e-05,
      "model_forward_time": 1.8111422061920166,
      "step": 56154
    },
    {
      "epoch": 5.3552627563476566e-05,
      "step": 56154,
      "training_step_time": 7.6706154346466064
    },
    {
      "epoch": 5.355358123779297e-05,
      "model_forward_time": 1.7476391792297363,
      "step": 56155
    },
    {
      "epoch": 5.355358123779297e-05,
      "step": 56155,
      "training_step_time": 7.783357858657837
    },
    {
      "epoch": 5.3554534912109374e-05,
      "model_forward_time": 1.7498266696929932,
      "step": 56156
    },
    {
      "epoch": 5.3554534912109374e-05,
      "step": 56156,
      "training_step_time": 8.58370566368103
    },
    {
      "epoch": 5.355548858642578e-05,
      "model_forward_time": 2.9178872108459473,
      "step": 56157
    },
    {
      "epoch": 5.355548858642578e-05,
      "step": 56157,
      "training_step_time": 8.605648517608643
    },
    {
      "epoch": 5.355644226074219e-05,
      "model_forward_time": 1.7486481666564941,
      "step": 56158
    },
    {
      "epoch": 5.355644226074219e-05,
      "step": 56158,
      "training_step_time": 7.730508804321289
    },
    {
      "epoch": 5.355739593505859e-05,
      "model_forward_time": 1.7419772148132324,
      "step": 56159
    },
    {
      "epoch": 5.355739593505859e-05,
      "step": 56159,
      "training_step_time": 7.644418716430664
    },
    {
      "epoch": 5.3558349609375e-05,
      "grad_norm": 0.13506251573562622,
      "learning_rate": 4.396470647874753e-06,
      "loss": 0.0621,
      "step": 56160
    },
    {
      "epoch": 5.3558349609375e-05,
      "model_forward_time": 1.757779836654663,
      "step": 56160
    },
    {
      "dynamics_loss_avg": 0.06430505774915218,
      "epoch": 5.3558349609375e-05,
      "step": 56160
    },
    {
      "action_loss_avg": 0.001871684571960941,
      "epoch": 5.3558349609375e-05,
      "step": 56160
    },
    {
      "epoch": 5.3558349609375e-05,
      "step": 56160,
      "training_step_time": 7.234426021575928
    },
    {
      "epoch": 5.355930328369141e-05,
      "model_forward_time": 1.7481138706207275,
      "step": 56161
    },
    {
      "epoch": 5.355930328369141e-05,
      "step": 56161,
      "training_step_time": 7.786346435546875
    },
    {
      "epoch": 5.356025695800781e-05,
      "model_forward_time": 1.3858273029327393,
      "step": 56162
    },
    {
      "epoch": 5.356025695800781e-05,
      "step": 56162,
      "training_step_time": 7.799809217453003
    },
    {
      "epoch": 5.3561210632324216e-05,
      "model_forward_time": 1.8200507164001465,
      "step": 56163
    },
    {
      "epoch": 5.3561210632324216e-05,
      "step": 56163,
      "training_step_time": 9.422202587127686
    },
    {
      "epoch": 5.3562164306640626e-05,
      "model_forward_time": 1.7436885833740234,
      "step": 56164
    },
    {
      "epoch": 5.3562164306640626e-05,
      "step": 56164,
      "training_step_time": 7.771622896194458
    },
    {
      "epoch": 5.356311798095703e-05,
      "model_forward_time": 1.7453680038452148,
      "step": 56165
    },
    {
      "epoch": 5.356311798095703e-05,
      "step": 56165,
      "training_step_time": 8.386585235595703
    },
    {
      "epoch": 5.3564071655273434e-05,
      "model_forward_time": 2.9248573780059814,
      "step": 56166
    },
    {
      "epoch": 5.3564071655273434e-05,
      "step": 56166,
      "training_step_time": 9.71257472038269
    },
    {
      "epoch": 5.3565025329589845e-05,
      "model_forward_time": 1.7453689575195312,
      "step": 56167
    },
    {
      "epoch": 5.3565025329589845e-05,
      "step": 56167,
      "training_step_time": 8.345906019210815
    },
    {
      "epoch": 5.356597900390625e-05,
      "model_forward_time": 1.0606107711791992,
      "step": 56168
    },
    {
      "epoch": 5.356597900390625e-05,
      "step": 56168,
      "training_step_time": 7.754741191864014
    },
    {
      "epoch": 5.356693267822265e-05,
      "model_forward_time": 1.7465875148773193,
      "step": 56169
    },
    {
      "epoch": 5.356693267822265e-05,
      "step": 56169,
      "training_step_time": 7.214468002319336
    },
    {
      "epoch": 5.3567886352539064e-05,
      "grad_norm": 0.09774714708328247,
      "learning_rate": 4.394829300736229e-06,
      "loss": 0.061,
      "step": 56170
    },
    {
      "epoch": 5.3567886352539064e-05,
      "model_forward_time": 1.75150465965271,
      "step": 56170
    },
    {
      "dynamics_loss_avg": 0.06795690655708313,
      "epoch": 5.3567886352539064e-05,
      "step": 56170
    },
    {
      "action_loss_avg": 0.0022109902900410818,
      "epoch": 5.3567886352539064e-05,
      "step": 56170
    },
    {
      "epoch": 5.3567886352539064e-05,
      "step": 56170,
      "training_step_time": 7.76400351524353
    },
    {
      "epoch": 5.356884002685547e-05,
      "model_forward_time": 1.7445967197418213,
      "step": 56171
    },
    {
      "epoch": 5.356884002685547e-05,
      "step": 56171,
      "training_step_time": 7.774745464324951
    },
    {
      "epoch": 5.356979370117187e-05,
      "model_forward_time": 1.820087194442749,
      "step": 56172
    },
    {
      "epoch": 5.356979370117187e-05,
      "step": 56172,
      "training_step_time": 7.741751670837402
    },
    {
      "epoch": 5.357074737548828e-05,
      "model_forward_time": 1.7497103214263916,
      "step": 56173
    },
    {
      "epoch": 5.357074737548828e-05,
      "step": 56173,
      "training_step_time": 8.473432540893555
    },
    {
      "epoch": 5.357170104980469e-05,
      "model_forward_time": 1.7468883991241455,
      "step": 56174
    },
    {
      "epoch": 5.357170104980469e-05,
      "step": 56174,
      "training_step_time": 8.243077754974365
    },
    {
      "epoch": 5.357265472412109e-05,
      "model_forward_time": 2.884256362915039,
      "step": 56175
    },
    {
      "epoch": 5.357265472412109e-05,
      "step": 56175,
      "training_step_time": 8.729148149490356
    },
    {
      "epoch": 5.35736083984375e-05,
      "model_forward_time": 1.7387206554412842,
      "step": 56176
    },
    {
      "epoch": 5.35736083984375e-05,
      "step": 56176,
      "training_step_time": 8.55424952507019
    },
    {
      "epoch": 5.3574562072753906e-05,
      "model_forward_time": 1.7483808994293213,
      "step": 56177
    },
    {
      "epoch": 5.3574562072753906e-05,
      "step": 56177,
      "training_step_time": 7.6025073528289795
    },
    {
      "epoch": 5.357551574707031e-05,
      "model_forward_time": 1.749913215637207,
      "step": 56178
    },
    {
      "epoch": 5.357551574707031e-05,
      "step": 56178,
      "training_step_time": 7.1234776973724365
    },
    {
      "epoch": 5.357646942138672e-05,
      "model_forward_time": 1.7456998825073242,
      "step": 56179
    },
    {
      "epoch": 5.357646942138672e-05,
      "step": 56179,
      "training_step_time": 7.679960489273071
    },
    {
      "epoch": 5.3577423095703125e-05,
      "grad_norm": 0.10046139359474182,
      "learning_rate": 4.393188019778265e-06,
      "loss": 0.0585,
      "step": 56180
    },
    {
      "epoch": 5.3577423095703125e-05,
      "model_forward_time": 1.7569019794464111,
      "step": 56180
    },
    {
      "dynamics_loss_avg": 0.047613486833870414,
      "epoch": 5.3577423095703125e-05,
      "step": 56180
    },
    {
      "action_loss_avg": 0.0013401295349467547,
      "epoch": 5.3577423095703125e-05,
      "step": 56180
    },
    {
      "epoch": 5.3577423095703125e-05,
      "step": 56180,
      "training_step_time": 7.686268091201782
    },
    {
      "epoch": 5.357837677001953e-05,
      "model_forward_time": 1.8207194805145264,
      "step": 56181
    },
    {
      "epoch": 5.357837677001953e-05,
      "step": 56181,
      "training_step_time": 8.229619026184082
    },
    {
      "epoch": 5.357933044433594e-05,
      "model_forward_time": 1.7469375133514404,
      "step": 56182
    },
    {
      "epoch": 5.357933044433594e-05,
      "step": 56182,
      "training_step_time": 9.970661878585815
    },
    {
      "epoch": 5.3580284118652344e-05,
      "model_forward_time": 1.7542448043823242,
      "step": 56183
    },
    {
      "epoch": 5.3580284118652344e-05,
      "step": 56183,
      "training_step_time": 7.776521444320679
    },
    {
      "epoch": 5.358123779296875e-05,
      "model_forward_time": 1.7504768371582031,
      "step": 56184
    },
    {
      "epoch": 5.358123779296875e-05,
      "step": 56184,
      "training_step_time": 10.974362850189209
    },
    {
      "epoch": 5.358219146728516e-05,
      "model_forward_time": 1.7524738311767578,
      "step": 56185
    },
    {
      "epoch": 5.358219146728516e-05,
      "step": 56185,
      "training_step_time": 7.708257675170898
    },
    {
      "epoch": 5.358314514160156e-05,
      "model_forward_time": 1.7457377910614014,
      "step": 56186
    },
    {
      "epoch": 5.358314514160156e-05,
      "step": 56186,
      "training_step_time": 7.7330121994018555
    },
    {
      "epoch": 5.3584098815917966e-05,
      "model_forward_time": 1.7418763637542725,
      "step": 56187
    },
    {
      "epoch": 5.3584098815917966e-05,
      "step": 56187,
      "training_step_time": 7.600161552429199
    },
    {
      "epoch": 5.358505249023438e-05,
      "model_forward_time": 1.747905969619751,
      "step": 56188
    },
    {
      "epoch": 5.358505249023438e-05,
      "step": 56188,
      "training_step_time": 7.715203762054443
    },
    {
      "epoch": 5.358600616455078e-05,
      "model_forward_time": 1.748323678970337,
      "step": 56189
    },
    {
      "epoch": 5.358600616455078e-05,
      "step": 56189,
      "training_step_time": 7.7592926025390625
    },
    {
      "epoch": 5.3586959838867185e-05,
      "grad_norm": 0.09854228794574738,
      "learning_rate": 4.3915468051803465e-06,
      "loss": 0.0597,
      "step": 56190
    },
    {
      "epoch": 5.3586959838867185e-05,
      "model_forward_time": 1.8155264854431152,
      "step": 56190
    },
    {
      "dynamics_loss_avg": 0.052546660788357256,
      "epoch": 5.3586959838867185e-05,
      "step": 56190
    },
    {
      "action_loss_avg": 0.0016276768335956148,
      "epoch": 5.3586959838867185e-05,
      "step": 56190
    },
    {
      "epoch": 5.3586959838867185e-05,
      "step": 56190,
      "training_step_time": 7.710202217102051
    },
    {
      "epoch": 5.3587913513183596e-05,
      "model_forward_time": 1.7465126514434814,
      "step": 56191
    },
    {
      "epoch": 5.3587913513183596e-05,
      "step": 56191,
      "training_step_time": 9.36817216873169
    },
    {
      "epoch": 5.35888671875e-05,
      "model_forward_time": 1.7435178756713867,
      "step": 56192
    },
    {
      "epoch": 5.35888671875e-05,
      "step": 56192,
      "training_step_time": 8.280835390090942
    },
    {
      "epoch": 5.3589820861816404e-05,
      "model_forward_time": 1.7431256771087646,
      "step": 56193
    },
    {
      "epoch": 5.3589820861816404e-05,
      "step": 56193,
      "training_step_time": 7.706536293029785
    },
    {
      "epoch": 5.3590774536132815e-05,
      "model_forward_time": 2.914625883102417,
      "step": 56194
    },
    {
      "epoch": 5.3590774536132815e-05,
      "step": 56194,
      "training_step_time": 7.722599983215332
    },
    {
      "epoch": 5.359172821044922e-05,
      "model_forward_time": 1.752338171005249,
      "step": 56195
    },
    {
      "epoch": 5.359172821044922e-05,
      "step": 56195,
      "training_step_time": 8.229286670684814
    },
    {
      "epoch": 5.359268188476562e-05,
      "model_forward_time": 1.745570421218872,
      "step": 56196
    },
    {
      "epoch": 5.359268188476562e-05,
      "step": 56196,
      "training_step_time": 7.572945833206177
    },
    {
      "epoch": 5.3593635559082034e-05,
      "model_forward_time": 1.7778096199035645,
      "step": 56197
    },
    {
      "epoch": 5.3593635559082034e-05,
      "step": 56197,
      "training_step_time": 7.743586778640747
    },
    {
      "epoch": 5.359458923339844e-05,
      "model_forward_time": 1.7390882968902588,
      "step": 56198
    },
    {
      "epoch": 5.359458923339844e-05,
      "step": 56198,
      "training_step_time": 7.705981254577637
    },
    {
      "epoch": 5.359554290771484e-05,
      "model_forward_time": 1.7566239833831787,
      "step": 56199
    },
    {
      "epoch": 5.359554290771484e-05,
      "step": 56199,
      "training_step_time": 7.727727890014648
    },
    {
      "epoch": 5.359649658203125e-05,
      "grad_norm": 0.08995529264211655,
      "learning_rate": 4.389905657121955e-06,
      "loss": 0.0635,
      "step": 56200
    },
    {
      "epoch": 5.359649658203125e-05,
      "model_forward_time": 1.752737283706665,
      "step": 56200
    },
    {
      "dynamics_loss_avg": 0.06687194984406233,
      "epoch": 5.359649658203125e-05,
      "step": 56200
    },
    {
      "action_loss_avg": 0.0018788711080560462,
      "epoch": 5.359649658203125e-05,
      "step": 56200
    },
    {
      "epoch": 5.359649658203125e-05,
      "step": 56200,
      "training_step_time": 7.685892343521118
    },
    {
      "epoch": 5.359745025634766e-05,
      "model_forward_time": 1.7446489334106445,
      "step": 56201
    },
    {
      "epoch": 5.359745025634766e-05,
      "step": 56201,
      "training_step_time": 9.189644813537598
    },
    {
      "epoch": 5.359840393066406e-05,
      "model_forward_time": 1.7450485229492188,
      "step": 56202
    },
    {
      "epoch": 5.359840393066406e-05,
      "step": 56202,
      "training_step_time": 7.763065814971924
    },
    {
      "epoch": 5.359935760498047e-05,
      "model_forward_time": 2.9133753776550293,
      "step": 56203
    },
    {
      "epoch": 5.359935760498047e-05,
      "step": 56203,
      "training_step_time": 8.730881690979004
    },
    {
      "epoch": 5.3600311279296876e-05,
      "model_forward_time": 1.7415812015533447,
      "step": 56204
    },
    {
      "epoch": 5.3600311279296876e-05,
      "step": 56204,
      "training_step_time": 8.61261773109436
    },
    {
      "epoch": 5.360126495361328e-05,
      "model_forward_time": 1.7376635074615479,
      "step": 56205
    },
    {
      "epoch": 5.360126495361328e-05,
      "step": 56205,
      "training_step_time": 7.3591468334198
    },
    {
      "epoch": 5.360221862792969e-05,
      "model_forward_time": 1.7483601570129395,
      "step": 56206
    },
    {
      "epoch": 5.360221862792969e-05,
      "step": 56206,
      "training_step_time": 7.682467937469482
    },
    {
      "epoch": 5.3603172302246094e-05,
      "model_forward_time": 1.7422089576721191,
      "step": 56207
    },
    {
      "epoch": 5.3603172302246094e-05,
      "step": 56207,
      "training_step_time": 8.415935039520264
    },
    {
      "epoch": 5.36041259765625e-05,
      "model_forward_time": 1.7431118488311768,
      "step": 56208
    },
    {
      "epoch": 5.36041259765625e-05,
      "step": 56208,
      "training_step_time": 8.110017776489258
    },
    {
      "epoch": 5.360507965087891e-05,
      "model_forward_time": 1.817319631576538,
      "step": 56209
    },
    {
      "epoch": 5.360507965087891e-05,
      "step": 56209,
      "training_step_time": 7.676119565963745
    },
    {
      "epoch": 5.360603332519531e-05,
      "grad_norm": 0.08679419010877609,
      "learning_rate": 4.3882645757825635e-06,
      "loss": 0.0572,
      "step": 56210
    },
    {
      "epoch": 5.360603332519531e-05,
      "model_forward_time": 1.7560925483703613,
      "step": 56210
    },
    {
      "dynamics_loss_avg": 0.05133868847042322,
      "epoch": 5.360603332519531e-05,
      "step": 56210
    },
    {
      "action_loss_avg": 0.0016087912255898118,
      "epoch": 5.360603332519531e-05,
      "step": 56210
    },
    {
      "epoch": 5.360603332519531e-05,
      "step": 56210,
      "training_step_time": 8.352966547012329
    },
    {
      "epoch": 5.360698699951172e-05,
      "model_forward_time": 1.7518956661224365,
      "step": 56211
    },
    {
      "epoch": 5.360698699951172e-05,
      "step": 56211,
      "training_step_time": 8.463328123092651
    },
    {
      "epoch": 5.360794067382813e-05,
      "model_forward_time": 2.920966625213623,
      "step": 56212
    },
    {
      "epoch": 5.360794067382813e-05,
      "step": 56212,
      "training_step_time": 7.712501525878906
    },
    {
      "epoch": 5.360889434814453e-05,
      "model_forward_time": 1.7453439235687256,
      "step": 56213
    },
    {
      "epoch": 5.360889434814453e-05,
      "step": 56213,
      "training_step_time": 7.645057439804077
    },
    {
      "epoch": 5.3609848022460936e-05,
      "model_forward_time": 1.752685546875,
      "step": 56214
    },
    {
      "epoch": 5.3609848022460936e-05,
      "step": 56214,
      "training_step_time": 7.639932632446289
    },
    {
      "epoch": 5.361080169677735e-05,
      "model_forward_time": 1.7496685981750488,
      "step": 56215
    },
    {
      "epoch": 5.361080169677735e-05,
      "step": 56215,
      "training_step_time": 8.813432931900024
    },
    {
      "epoch": 5.361175537109375e-05,
      "model_forward_time": 1.7540309429168701,
      "step": 56216
    },
    {
      "epoch": 5.361175537109375e-05,
      "step": 56216,
      "training_step_time": 7.743689775466919
    },
    {
      "epoch": 5.3612709045410155e-05,
      "model_forward_time": 1.7431530952453613,
      "step": 56217
    },
    {
      "epoch": 5.3612709045410155e-05,
      "step": 56217,
      "training_step_time": 8.167858362197876
    },
    {
      "epoch": 5.3613662719726566e-05,
      "model_forward_time": 1.8103148937225342,
      "step": 56218
    },
    {
      "epoch": 5.3613662719726566e-05,
      "step": 56218,
      "training_step_time": 7.7180702686309814
    },
    {
      "epoch": 5.361461639404297e-05,
      "model_forward_time": 1.7479076385498047,
      "step": 56219
    },
    {
      "epoch": 5.361461639404297e-05,
      "step": 56219,
      "training_step_time": 7.724058389663696
    },
    {
      "epoch": 5.3615570068359374e-05,
      "grad_norm": 0.14246498048305511,
      "learning_rate": 4.386623561341637e-06,
      "loss": 0.0632,
      "step": 56220
    },
    {
      "epoch": 5.3615570068359374e-05,
      "model_forward_time": 1.7511177062988281,
      "step": 56220
    },
    {
      "dynamics_loss_avg": 0.0769372072070837,
      "epoch": 5.3615570068359374e-05,
      "step": 56220
    },
    {
      "action_loss_avg": 0.001963665973744355,
      "epoch": 5.3615570068359374e-05,
      "step": 56220
    },
    {
      "epoch": 5.3615570068359374e-05,
      "step": 56220,
      "training_step_time": 8.227401733398438
    },
    {
      "epoch": 5.361652374267578e-05,
      "model_forward_time": 2.886620044708252,
      "step": 56221
    },
    {
      "epoch": 5.361652374267578e-05,
      "step": 56221,
      "training_step_time": 8.275829076766968
    },
    {
      "epoch": 5.361747741699219e-05,
      "model_forward_time": 1.7494261264801025,
      "step": 56222
    },
    {
      "epoch": 5.361747741699219e-05,
      "step": 56222,
      "training_step_time": 7.638085603713989
    },
    {
      "epoch": 5.361843109130859e-05,
      "model_forward_time": 1.746467113494873,
      "step": 56223
    },
    {
      "epoch": 5.361843109130859e-05,
      "step": 56223,
      "training_step_time": 7.580541133880615
    },
    {
      "epoch": 5.3619384765625e-05,
      "model_forward_time": 1.7467548847198486,
      "step": 56224
    },
    {
      "epoch": 5.3619384765625e-05,
      "step": 56224,
      "training_step_time": 7.652127742767334
    },
    {
      "epoch": 5.362033843994141e-05,
      "model_forward_time": 1.7484610080718994,
      "step": 56225
    },
    {
      "epoch": 5.362033843994141e-05,
      "step": 56225,
      "training_step_time": 7.700561285018921
    },
    {
      "epoch": 5.362129211425781e-05,
      "model_forward_time": 1.7447881698608398,
      "step": 56226
    },
    {
      "epoch": 5.362129211425781e-05,
      "step": 56226,
      "training_step_time": 7.880784749984741
    },
    {
      "epoch": 5.3622245788574216e-05,
      "model_forward_time": 1.4305083751678467,
      "step": 56227
    },
    {
      "epoch": 5.3622245788574216e-05,
      "step": 56227,
      "training_step_time": 7.687281370162964
    },
    {
      "epoch": 5.3623199462890626e-05,
      "model_forward_time": 1.7510771751403809,
      "step": 56228
    },
    {
      "epoch": 5.3623199462890626e-05,
      "step": 56228,
      "training_step_time": 8.495302677154541
    },
    {
      "epoch": 5.362415313720703e-05,
      "model_forward_time": 1.7453320026397705,
      "step": 56229
    },
    {
      "epoch": 5.362415313720703e-05,
      "step": 56229,
      "training_step_time": 8.64744257926941
    },
    {
      "epoch": 5.3625106811523435e-05,
      "grad_norm": 0.12637357413768768,
      "learning_rate": 4.384982613978636e-06,
      "loss": 0.0597,
      "step": 56230
    },
    {
      "epoch": 5.3625106811523435e-05,
      "model_forward_time": 2.8415167331695557,
      "step": 56230
    },
    {
      "dynamics_loss_avg": 0.06078592203557491,
      "epoch": 5.3625106811523435e-05,
      "step": 56230
    },
    {
      "action_loss_avg": 0.0014949891599826514,
      "epoch": 5.3625106811523435e-05,
      "step": 56230
    },
    {
      "epoch": 5.3625106811523435e-05,
      "step": 56230,
      "training_step_time": 9.279225826263428
    },
    {
      "epoch": 5.3626060485839845e-05,
      "model_forward_time": 1.7470743656158447,
      "step": 56231
    },
    {
      "epoch": 5.3626060485839845e-05,
      "step": 56231,
      "training_step_time": 8.365903854370117
    },
    {
      "epoch": 5.362701416015625e-05,
      "model_forward_time": 1.745055913925171,
      "step": 56232
    },
    {
      "epoch": 5.362701416015625e-05,
      "step": 56232,
      "training_step_time": 7.207704305648804
    },
    {
      "epoch": 5.362796783447265e-05,
      "model_forward_time": 1.7424960136413574,
      "step": 56233
    },
    {
      "epoch": 5.362796783447265e-05,
      "step": 56233,
      "training_step_time": 7.616312503814697
    },
    {
      "epoch": 5.3628921508789064e-05,
      "model_forward_time": 1.7466959953308105,
      "step": 56234
    },
    {
      "epoch": 5.3628921508789064e-05,
      "step": 56234,
      "training_step_time": 8.780460119247437
    },
    {
      "epoch": 5.362987518310547e-05,
      "model_forward_time": 1.7434477806091309,
      "step": 56235
    },
    {
      "epoch": 5.362987518310547e-05,
      "step": 56235,
      "training_step_time": 7.304527997970581
    },
    {
      "epoch": 5.363082885742187e-05,
      "model_forward_time": 1.7959165573120117,
      "step": 56236
    },
    {
      "epoch": 5.363082885742187e-05,
      "step": 56236,
      "training_step_time": 7.657607555389404
    },
    {
      "epoch": 5.363178253173828e-05,
      "model_forward_time": 1.7522220611572266,
      "step": 56237
    },
    {
      "epoch": 5.363178253173828e-05,
      "step": 56237,
      "training_step_time": 7.658432245254517
    },
    {
      "epoch": 5.363273620605469e-05,
      "model_forward_time": 1.7447013854980469,
      "step": 56238
    },
    {
      "epoch": 5.363273620605469e-05,
      "step": 56238,
      "training_step_time": 8.463029146194458
    },
    {
      "epoch": 5.363368988037109e-05,
      "model_forward_time": 1.745039701461792,
      "step": 56239
    },
    {
      "epoch": 5.363368988037109e-05,
      "step": 56239,
      "training_step_time": 7.832179307937622
    },
    {
      "epoch": 5.36346435546875e-05,
      "grad_norm": 0.11946170777082443,
      "learning_rate": 4.383341733873012e-06,
      "loss": 0.0614,
      "step": 56240
    },
    {
      "epoch": 5.36346435546875e-05,
      "model_forward_time": 2.9392950534820557,
      "step": 56240
    },
    {
      "dynamics_loss_avg": 0.053976803086698055,
      "epoch": 5.36346435546875e-05,
      "step": 56240
    },
    {
      "action_loss_avg": 0.0013881751539884136,
      "epoch": 5.36346435546875e-05,
      "step": 56240
    },
    {
      "epoch": 5.36346435546875e-05,
      "step": 56240,
      "training_step_time": 8.32715129852295
    },
    {
      "epoch": 5.3635597229003906e-05,
      "model_forward_time": 1.7522952556610107,
      "step": 56241
    },
    {
      "epoch": 5.3635597229003906e-05,
      "step": 56241,
      "training_step_time": 7.538971662521362
    },
    {
      "epoch": 5.363655090332031e-05,
      "model_forward_time": 1.7425060272216797,
      "step": 56242
    },
    {
      "epoch": 5.363655090332031e-05,
      "step": 56242,
      "training_step_time": 7.70592188835144
    },
    {
      "epoch": 5.363750457763672e-05,
      "model_forward_time": 1.748128890991211,
      "step": 56243
    },
    {
      "epoch": 5.363750457763672e-05,
      "step": 56243,
      "training_step_time": 7.75008225440979
    },
    {
      "epoch": 5.3638458251953125e-05,
      "model_forward_time": 1.7482309341430664,
      "step": 56244
    },
    {
      "epoch": 5.3638458251953125e-05,
      "step": 56244,
      "training_step_time": 7.751232862472534
    },
    {
      "epoch": 5.363941192626953e-05,
      "model_forward_time": 1.384488582611084,
      "step": 56245
    },
    {
      "epoch": 5.363941192626953e-05,
      "step": 56245,
      "training_step_time": 7.663096189498901
    },
    {
      "epoch": 5.364036560058594e-05,
      "model_forward_time": 1.7524511814117432,
      "step": 56246
    },
    {
      "epoch": 5.364036560058594e-05,
      "step": 56246,
      "training_step_time": 9.644155740737915
    },
    {
      "epoch": 5.3641319274902344e-05,
      "model_forward_time": 1.745356559753418,
      "step": 56247
    },
    {
      "epoch": 5.3641319274902344e-05,
      "step": 56247,
      "training_step_time": 8.485267639160156
    },
    {
      "epoch": 5.364227294921875e-05,
      "model_forward_time": 1.7441596984863281,
      "step": 56248
    },
    {
      "epoch": 5.364227294921875e-05,
      "step": 56248,
      "training_step_time": 8.26844048500061
    },
    {
      "epoch": 5.364322662353516e-05,
      "model_forward_time": 2.9246833324432373,
      "step": 56249
    },
    {
      "epoch": 5.364322662353516e-05,
      "step": 56249,
      "training_step_time": 7.710619926452637
    },
    {
      "epoch": 5.364418029785156e-05,
      "grad_norm": 0.1049620732665062,
      "learning_rate": 4.38170092120421e-06,
      "loss": 0.0625,
      "step": 56250
    },
    {
      "epoch": 5.364418029785156e-05,
      "model_forward_time": 1.3861360549926758,
      "step": 56250
    },
    {
      "dynamics_loss_avg": 0.061204304359853266,
      "epoch": 5.364418029785156e-05,
      "step": 56250
    },
    {
      "action_loss_avg": 0.001352282101288438,
      "epoch": 5.364418029785156e-05,
      "step": 56250
    },
    {
      "epoch": 5.364418029785156e-05,
      "step": 56250,
      "training_step_time": 7.494876384735107
    },
    {
      "epoch": 5.3645133972167967e-05,
      "model_forward_time": 1.7478134632110596,
      "step": 56251
    },
    {
      "epoch": 5.3645133972167967e-05,
      "step": 56251,
      "training_step_time": 8.826507329940796
    },
    {
      "epoch": 5.364608764648438e-05,
      "model_forward_time": 1.75360107421875,
      "step": 56252
    },
    {
      "epoch": 5.364608764648438e-05,
      "step": 56252,
      "training_step_time": 7.764028072357178
    },
    {
      "epoch": 5.364704132080078e-05,
      "model_forward_time": 1.742727518081665,
      "step": 56253
    },
    {
      "epoch": 5.364704132080078e-05,
      "step": 56253,
      "training_step_time": 7.733988285064697
    },
    {
      "epoch": 5.3647994995117185e-05,
      "model_forward_time": 1.7518165111541748,
      "step": 56254
    },
    {
      "epoch": 5.3647994995117185e-05,
      "step": 56254,
      "training_step_time": 7.695359945297241
    },
    {
      "epoch": 5.3648948669433596e-05,
      "model_forward_time": 1.8249397277832031,
      "step": 56255
    },
    {
      "epoch": 5.3648948669433596e-05,
      "step": 56255,
      "training_step_time": 9.059620380401611
    },
    {
      "epoch": 5.364990234375e-05,
      "model_forward_time": 1.7603411674499512,
      "step": 56256
    },
    {
      "epoch": 5.364990234375e-05,
      "step": 56256,
      "training_step_time": 7.714441776275635
    },
    {
      "epoch": 5.3650856018066404e-05,
      "model_forward_time": 1.752918004989624,
      "step": 56257
    },
    {
      "epoch": 5.3650856018066404e-05,
      "step": 56257,
      "training_step_time": 8.210719585418701
    },
    {
      "epoch": 5.3651809692382815e-05,
      "model_forward_time": 2.8607125282287598,
      "step": 56258
    },
    {
      "epoch": 5.3651809692382815e-05,
      "step": 56258,
      "training_step_time": 9.734673738479614
    },
    {
      "epoch": 5.365276336669922e-05,
      "model_forward_time": 1.749931812286377,
      "step": 56259
    },
    {
      "epoch": 5.365276336669922e-05,
      "step": 56259,
      "training_step_time": 8.073795557022095
    },
    {
      "epoch": 5.365371704101562e-05,
      "grad_norm": 0.12206485867500305,
      "learning_rate": 4.380060176151663e-06,
      "loss": 0.0612,
      "step": 56260
    },
    {
      "epoch": 5.365371704101562e-05,
      "model_forward_time": 1.7451696395874023,
      "step": 56260
    },
    {
      "dynamics_loss_avg": 0.06398549247533084,
      "epoch": 5.365371704101562e-05,
      "step": 56260
    },
    {
      "action_loss_avg": 0.0031830974156036975,
      "epoch": 5.365371704101562e-05,
      "step": 56260
    },
    {
      "epoch": 5.365371704101562e-05,
      "step": 56260,
      "training_step_time": 7.813936948776245
    },
    {
      "epoch": 5.3654670715332034e-05,
      "model_forward_time": 1.7482283115386963,
      "step": 56261
    },
    {
      "epoch": 5.3654670715332034e-05,
      "step": 56261,
      "training_step_time": 7.656559467315674
    },
    {
      "epoch": 5.365562438964844e-05,
      "model_forward_time": 1.7840726375579834,
      "step": 56262
    },
    {
      "epoch": 5.365562438964844e-05,
      "step": 56262,
      "training_step_time": 7.682100296020508
    },
    {
      "epoch": 5.365657806396484e-05,
      "model_forward_time": 1.759314775466919,
      "step": 56263
    },
    {
      "epoch": 5.365657806396484e-05,
      "step": 56263,
      "training_step_time": 8.22665023803711
    },
    {
      "epoch": 5.365753173828125e-05,
      "model_forward_time": 1.8056998252868652,
      "step": 56264
    },
    {
      "epoch": 5.365753173828125e-05,
      "step": 56264,
      "training_step_time": 7.6334826946258545
    },
    {
      "epoch": 5.365848541259766e-05,
      "model_forward_time": 1.7500357627868652,
      "step": 56265
    },
    {
      "epoch": 5.365848541259766e-05,
      "step": 56265,
      "training_step_time": 8.281205654144287
    },
    {
      "epoch": 5.365943908691406e-05,
      "model_forward_time": 1.7550716400146484,
      "step": 56266
    },
    {
      "epoch": 5.365943908691406e-05,
      "step": 56266,
      "training_step_time": 7.635909795761108
    },
    {
      "epoch": 5.366039276123047e-05,
      "model_forward_time": 2.8126635551452637,
      "step": 56267
    },
    {
      "epoch": 5.366039276123047e-05,
      "step": 56267,
      "training_step_time": 8.287185907363892
    },
    {
      "epoch": 5.3661346435546876e-05,
      "model_forward_time": 1.7585399150848389,
      "step": 56268
    },
    {
      "epoch": 5.3661346435546876e-05,
      "step": 56268,
      "training_step_time": 7.64115047454834
    },
    {
      "epoch": 5.366230010986328e-05,
      "model_forward_time": 1.7558894157409668,
      "step": 56269
    },
    {
      "epoch": 5.366230010986328e-05,
      "step": 56269,
      "training_step_time": 7.78119158744812
    },
    {
      "epoch": 5.366325378417969e-05,
      "grad_norm": 0.10974030941724777,
      "learning_rate": 4.378419498894805e-06,
      "loss": 0.0623,
      "step": 56270
    },
    {
      "epoch": 5.366325378417969e-05,
      "model_forward_time": 1.7529211044311523,
      "step": 56270
    },
    {
      "dynamics_loss_avg": 0.066298907995224,
      "epoch": 5.366325378417969e-05,
      "step": 56270
    },
    {
      "action_loss_avg": 0.0015022771825897508,
      "epoch": 5.366325378417969e-05,
      "step": 56270
    },
    {
      "epoch": 5.366325378417969e-05,
      "step": 56270,
      "training_step_time": 8.311466932296753
    },
    {
      "epoch": 5.3664207458496095e-05,
      "model_forward_time": 1.3912179470062256,
      "step": 56271
    },
    {
      "epoch": 5.3664207458496095e-05,
      "step": 56271,
      "training_step_time": 7.3013551235198975
    },
    {
      "epoch": 5.36651611328125e-05,
      "model_forward_time": 1.7300574779510498,
      "step": 56272
    },
    {
      "epoch": 5.36651611328125e-05,
      "step": 56272,
      "training_step_time": 7.766221284866333
    },
    {
      "epoch": 5.366611480712891e-05,
      "model_forward_time": 1.8262734413146973,
      "step": 56273
    },
    {
      "epoch": 5.366611480712891e-05,
      "step": 56273,
      "training_step_time": 7.497851371765137
    },
    {
      "epoch": 5.366706848144531e-05,
      "model_forward_time": 1.7515254020690918,
      "step": 56274
    },
    {
      "epoch": 5.366706848144531e-05,
      "step": 56274,
      "training_step_time": 8.537798881530762
    },
    {
      "epoch": 5.366802215576172e-05,
      "model_forward_time": 1.7465543746948242,
      "step": 56275
    },
    {
      "epoch": 5.366802215576172e-05,
      "step": 56275,
      "training_step_time": 8.535800218582153
    },
    {
      "epoch": 5.366897583007813e-05,
      "model_forward_time": 1.7505033016204834,
      "step": 56276
    },
    {
      "epoch": 5.366897583007813e-05,
      "step": 56276,
      "training_step_time": 9.355515241622925
    },
    {
      "epoch": 5.366992950439453e-05,
      "model_forward_time": 2.9526867866516113,
      "step": 56277
    },
    {
      "epoch": 5.366992950439453e-05,
      "step": 56277,
      "training_step_time": 8.58668851852417
    },
    {
      "epoch": 5.3670883178710936e-05,
      "model_forward_time": 1.7562568187713623,
      "step": 56278
    },
    {
      "epoch": 5.3670883178710936e-05,
      "step": 56278,
      "training_step_time": 7.698201894760132
    },
    {
      "epoch": 5.367183685302735e-05,
      "model_forward_time": 1.7516982555389404,
      "step": 56279
    },
    {
      "epoch": 5.367183685302735e-05,
      "step": 56279,
      "training_step_time": 7.734665393829346
    },
    {
      "epoch": 5.367279052734375e-05,
      "grad_norm": 0.12984523177146912,
      "learning_rate": 4.376778889613056e-06,
      "loss": 0.0577,
      "step": 56280
    },
    {
      "epoch": 5.367279052734375e-05,
      "model_forward_time": 1.7562518119812012,
      "step": 56280
    },
    {
      "dynamics_loss_avg": 0.06418796014040709,
      "epoch": 5.367279052734375e-05,
      "step": 56280
    },
    {
      "action_loss_avg": 0.0022601214790483938,
      "epoch": 5.367279052734375e-05,
      "step": 56280
    },
    {
      "epoch": 5.367279052734375e-05,
      "step": 56280,
      "training_step_time": 7.2802770137786865
    },
    {
      "epoch": 5.3673744201660155e-05,
      "model_forward_time": 1.758073091506958,
      "step": 56281
    },
    {
      "epoch": 5.3673744201660155e-05,
      "step": 56281,
      "training_step_time": 8.244949579238892
    },
    {
      "epoch": 5.367469787597656e-05,
      "model_forward_time": 1.795128345489502,
      "step": 56282
    },
    {
      "epoch": 5.367469787597656e-05,
      "step": 56282,
      "training_step_time": 8.291894674301147
    },
    {
      "epoch": 5.367565155029297e-05,
      "model_forward_time": 1.7525532245635986,
      "step": 56283
    },
    {
      "epoch": 5.367565155029297e-05,
      "step": 56283,
      "training_step_time": 7.642056941986084
    },
    {
      "epoch": 5.3676605224609374e-05,
      "model_forward_time": 1.754955530166626,
      "step": 56284
    },
    {
      "epoch": 5.3676605224609374e-05,
      "step": 56284,
      "training_step_time": 10.38822340965271
    },
    {
      "epoch": 5.367755889892578e-05,
      "model_forward_time": 1.7497727870941162,
      "step": 56285
    },
    {
      "epoch": 5.367755889892578e-05,
      "step": 56285,
      "training_step_time": 8.116940021514893
    },
    {
      "epoch": 5.367851257324219e-05,
      "model_forward_time": 2.9262003898620605,
      "step": 56286
    },
    {
      "epoch": 5.367851257324219e-05,
      "step": 56286,
      "training_step_time": 7.66589617729187
    },
    {
      "epoch": 5.367946624755859e-05,
      "model_forward_time": 1.7531917095184326,
      "step": 56287
    },
    {
      "epoch": 5.367946624755859e-05,
      "step": 56287,
      "training_step_time": 7.419341802597046
    },
    {
      "epoch": 5.3680419921875e-05,
      "model_forward_time": 1.750115156173706,
      "step": 56288
    },
    {
      "epoch": 5.3680419921875e-05,
      "step": 56288,
      "training_step_time": 7.680935859680176
    },
    {
      "epoch": 5.368137359619141e-05,
      "model_forward_time": 1.8345866203308105,
      "step": 56289
    },
    {
      "epoch": 5.368137359619141e-05,
      "step": 56289,
      "training_step_time": 7.238328218460083
    },
    {
      "epoch": 5.368232727050781e-05,
      "grad_norm": 0.11042453348636627,
      "learning_rate": 4.3751383484858314e-06,
      "loss": 0.0628,
      "step": 56290
    },
    {
      "epoch": 5.368232727050781e-05,
      "model_forward_time": 1.7547125816345215,
      "step": 56290
    },
    {
      "dynamics_loss_avg": 0.062016815319657324,
      "epoch": 5.368232727050781e-05,
      "step": 56290
    },
    {
      "action_loss_avg": 0.0029534685134422036,
      "epoch": 5.368232727050781e-05,
      "step": 56290
    },
    {
      "epoch": 5.368232727050781e-05,
      "step": 56290,
      "training_step_time": 7.7077906131744385
    },
    {
      "epoch": 5.3683280944824216e-05,
      "model_forward_time": 1.7478036880493164,
      "step": 56291
    },
    {
      "epoch": 5.3683280944824216e-05,
      "step": 56291,
      "training_step_time": 7.5049285888671875
    },
    {
      "epoch": 5.3684234619140627e-05,
      "model_forward_time": 1.8225889205932617,
      "step": 56292
    },
    {
      "epoch": 5.3684234619140627e-05,
      "step": 56292,
      "training_step_time": 7.665163040161133
    },
    {
      "epoch": 5.368518829345703e-05,
      "model_forward_time": 1.7543420791625977,
      "step": 56293
    },
    {
      "epoch": 5.368518829345703e-05,
      "step": 56293,
      "training_step_time": 7.754992485046387
    },
    {
      "epoch": 5.3686141967773435e-05,
      "model_forward_time": 1.7397453784942627,
      "step": 56294
    },
    {
      "epoch": 5.3686141967773435e-05,
      "step": 56294,
      "training_step_time": 8.641012907028198
    },
    {
      "epoch": 5.3687095642089845e-05,
      "model_forward_time": 2.928204298019409,
      "step": 56295
    },
    {
      "epoch": 5.3687095642089845e-05,
      "step": 56295,
      "training_step_time": 8.542740821838379
    },
    {
      "epoch": 5.368804931640625e-05,
      "model_forward_time": 1.7573084831237793,
      "step": 56296
    },
    {
      "epoch": 5.368804931640625e-05,
      "step": 56296,
      "training_step_time": 7.602693796157837
    },
    {
      "epoch": 5.3689002990722653e-05,
      "model_forward_time": 1.754807472229004,
      "step": 56297
    },
    {
      "epoch": 5.3689002990722653e-05,
      "step": 56297,
      "training_step_time": 7.642361402511597
    },
    {
      "epoch": 5.3689956665039064e-05,
      "model_forward_time": 1.7551920413970947,
      "step": 56298
    },
    {
      "epoch": 5.3689956665039064e-05,
      "step": 56298,
      "training_step_time": 7.251589775085449
    },
    {
      "epoch": 5.369091033935547e-05,
      "model_forward_time": 1.75071382522583,
      "step": 56299
    },
    {
      "epoch": 5.369091033935547e-05,
      "step": 56299,
      "training_step_time": 8.313287019729614
    },
    {
      "epoch": 5.369186401367187e-05,
      "grad_norm": 0.1207689717411995,
      "learning_rate": 4.373497875692535e-06,
      "loss": 0.0573,
      "step": 56300
    },
    {
      "epoch": 5.369186401367187e-05,
      "model_forward_time": 1.7469403743743896,
      "step": 56300
    },
    {
      "dynamics_loss_avg": 0.056537631805986165,
      "epoch": 5.369186401367187e-05,
      "step": 56300
    },
    {
      "action_loss_avg": 0.0018050824961392208,
      "epoch": 5.369186401367187e-05,
      "step": 56300
    },
    {
      "epoch": 5.369186401367187e-05,
      "step": 56300,
      "training_step_time": 7.802605152130127
    },
    {
      "epoch": 5.369281768798828e-05,
      "model_forward_time": 1.8189013004302979,
      "step": 56301
    },
    {
      "epoch": 5.369281768798828e-05,
      "step": 56301,
      "training_step_time": 9.5091712474823
    },
    {
      "epoch": 5.369377136230469e-05,
      "model_forward_time": 1.7534797191619873,
      "step": 56302
    },
    {
      "epoch": 5.369377136230469e-05,
      "step": 56302,
      "training_step_time": 8.150510311126709
    },
    {
      "epoch": 5.369472503662109e-05,
      "model_forward_time": 1.7524785995483398,
      "step": 56303
    },
    {
      "epoch": 5.369472503662109e-05,
      "step": 56303,
      "training_step_time": 7.561204195022583
    },
    {
      "epoch": 5.36956787109375e-05,
      "model_forward_time": 2.897653579711914,
      "step": 56304
    },
    {
      "epoch": 5.36956787109375e-05,
      "step": 56304,
      "training_step_time": 7.699824094772339
    },
    {
      "epoch": 5.3696632385253906e-05,
      "model_forward_time": 1.7567551136016846,
      "step": 56305
    },
    {
      "epoch": 5.3696632385253906e-05,
      "step": 56305,
      "training_step_time": 7.584576368331909
    },
    {
      "epoch": 5.369758605957031e-05,
      "model_forward_time": 1.7529230117797852,
      "step": 56306
    },
    {
      "epoch": 5.369758605957031e-05,
      "step": 56306,
      "training_step_time": 7.575006484985352
    },
    {
      "epoch": 5.369853973388672e-05,
      "model_forward_time": 1.7219316959381104,
      "step": 56307
    },
    {
      "epoch": 5.369853973388672e-05,
      "step": 56307,
      "training_step_time": 7.255994558334351
    },
    {
      "epoch": 5.3699493408203125e-05,
      "model_forward_time": 1.75262451171875,
      "step": 56308
    },
    {
      "epoch": 5.3699493408203125e-05,
      "step": 56308,
      "training_step_time": 7.629566431045532
    },
    {
      "epoch": 5.370044708251953e-05,
      "model_forward_time": 1.7474148273468018,
      "step": 56309
    },
    {
      "epoch": 5.370044708251953e-05,
      "step": 56309,
      "training_step_time": 7.725359678268433
    },
    {
      "epoch": 5.370140075683594e-05,
      "grad_norm": 0.08333493024110794,
      "learning_rate": 4.37185747141257e-06,
      "loss": 0.0613,
      "step": 56310
    },
    {
      "epoch": 5.370140075683594e-05,
      "model_forward_time": 1.7905633449554443,
      "step": 56310
    },
    {
      "dynamics_loss_avg": 0.06836697366088629,
      "epoch": 5.370140075683594e-05,
      "step": 56310
    },
    {
      "action_loss_avg": 0.001895242161117494,
      "epoch": 5.370140075683594e-05,
      "step": 56310
    },
    {
      "epoch": 5.370140075683594e-05,
      "step": 56310,
      "training_step_time": 9.119039535522461
    },
    {
      "epoch": 5.3702354431152344e-05,
      "model_forward_time": 1.3785009384155273,
      "step": 56311
    },
    {
      "epoch": 5.3702354431152344e-05,
      "step": 56311,
      "training_step_time": 7.672374963760376
    },
    {
      "epoch": 5.370330810546875e-05,
      "model_forward_time": 1.772826910018921,
      "step": 56312
    },
    {
      "epoch": 5.370330810546875e-05,
      "step": 56312,
      "training_step_time": 10.324760675430298
    },
    {
      "epoch": 5.370426177978516e-05,
      "model_forward_time": 1.7488000392913818,
      "step": 56313
    },
    {
      "epoch": 5.370426177978516e-05,
      "step": 56313,
      "training_step_time": 9.647666454315186
    },
    {
      "epoch": 5.370521545410156e-05,
      "model_forward_time": 2.8974416255950928,
      "step": 56314
    },
    {
      "epoch": 5.370521545410156e-05,
      "step": 56314,
      "training_step_time": 7.5931315422058105
    },
    {
      "epoch": 5.3706169128417967e-05,
      "model_forward_time": 1.757479190826416,
      "step": 56315
    },
    {
      "epoch": 5.3706169128417967e-05,
      "step": 56315,
      "training_step_time": 7.582597732543945
    },
    {
      "epoch": 5.370712280273438e-05,
      "model_forward_time": 1.7491300106048584,
      "step": 56316
    },
    {
      "epoch": 5.370712280273438e-05,
      "step": 56316,
      "training_step_time": 8.253883600234985
    },
    {
      "epoch": 5.370807647705078e-05,
      "model_forward_time": 1.7462849617004395,
      "step": 56317
    },
    {
      "epoch": 5.370807647705078e-05,
      "step": 56317,
      "training_step_time": 7.580432415008545
    },
    {
      "epoch": 5.3709030151367185e-05,
      "model_forward_time": 1.749758243560791,
      "step": 56318
    },
    {
      "epoch": 5.3709030151367185e-05,
      "step": 56318,
      "training_step_time": 7.573070526123047
    },
    {
      "epoch": 5.3709983825683596e-05,
      "model_forward_time": 1.757300615310669,
      "step": 56319
    },
    {
      "epoch": 5.3709983825683596e-05,
      "step": 56319,
      "training_step_time": 7.55679988861084
    },
    {
      "epoch": 5.37109375e-05,
      "grad_norm": 0.08992724120616913,
      "learning_rate": 4.370217135825329e-06,
      "loss": 0.0566,
      "step": 56320
    },
    {
      "epoch": 5.37109375e-05,
      "model_forward_time": 1.7558207511901855,
      "step": 56320
    },
    {
      "dynamics_loss_avg": 0.06726050730794668,
      "epoch": 5.37109375e-05,
      "step": 56320
    },
    {
      "action_loss_avg": 0.002230659150518477,
      "epoch": 5.37109375e-05,
      "step": 56320
    },
    {
      "epoch": 5.37109375e-05,
      "step": 56320,
      "training_step_time": 8.272039651870728
    },
    {
      "epoch": 5.3711891174316404e-05,
      "model_forward_time": 1.7497162818908691,
      "step": 56321
    },
    {
      "epoch": 5.3711891174316404e-05,
      "step": 56321,
      "training_step_time": 7.623239994049072
    },
    {
      "epoch": 5.3712844848632815e-05,
      "model_forward_time": 1.7515997886657715,
      "step": 56322
    },
    {
      "epoch": 5.3712844848632815e-05,
      "step": 56322,
      "training_step_time": 8.199036359786987
    },
    {
      "epoch": 5.371379852294922e-05,
      "model_forward_time": 2.9427225589752197,
      "step": 56323
    },
    {
      "epoch": 5.371379852294922e-05,
      "step": 56323,
      "training_step_time": 7.7000579833984375
    },
    {
      "epoch": 5.371475219726562e-05,
      "model_forward_time": 1.746267557144165,
      "step": 56324
    },
    {
      "epoch": 5.371475219726562e-05,
      "step": 56324,
      "training_step_time": 7.639421224594116
    },
    {
      "epoch": 5.3715705871582034e-05,
      "model_forward_time": 1.0456180572509766,
      "step": 56325
    },
    {
      "epoch": 5.3715705871582034e-05,
      "step": 56325,
      "training_step_time": 7.749746322631836
    },
    {
      "epoch": 5.371665954589844e-05,
      "model_forward_time": 1.3843388557434082,
      "step": 56326
    },
    {
      "epoch": 5.371665954589844e-05,
      "step": 56326,
      "training_step_time": 7.617321491241455
    },
    {
      "epoch": 5.371761322021484e-05,
      "model_forward_time": 1.751168966293335,
      "step": 56327
    },
    {
      "epoch": 5.371761322021484e-05,
      "step": 56327,
      "training_step_time": 7.404780387878418
    },
    {
      "epoch": 5.371856689453125e-05,
      "model_forward_time": 1.7476191520690918,
      "step": 56328
    },
    {
      "epoch": 5.371856689453125e-05,
      "step": 56328,
      "training_step_time": 7.571000814437866
    },
    {
      "epoch": 5.371952056884766e-05,
      "model_forward_time": 1.812978982925415,
      "step": 56329
    },
    {
      "epoch": 5.371952056884766e-05,
      "step": 56329,
      "training_step_time": 8.729056119918823
    },
    {
      "epoch": 5.372047424316406e-05,
      "grad_norm": 0.10739266127347946,
      "learning_rate": 4.368576869110195e-06,
      "loss": 0.0535,
      "step": 56330
    },
    {
      "epoch": 5.372047424316406e-05,
      "model_forward_time": 1.7369670867919922,
      "step": 56330
    },
    {
      "dynamics_loss_avg": 0.05920823216438294,
      "epoch": 5.372047424316406e-05,
      "step": 56330
    },
    {
      "action_loss_avg": 0.0029924051486887038,
      "epoch": 5.372047424316406e-05,
      "step": 56330
    },
    {
      "epoch": 5.372047424316406e-05,
      "step": 56330,
      "training_step_time": 8.349251985549927
    },
    {
      "epoch": 5.372142791748047e-05,
      "model_forward_time": 1.7389309406280518,
      "step": 56331
    },
    {
      "epoch": 5.372142791748047e-05,
      "step": 56331,
      "training_step_time": 7.681647539138794
    },
    {
      "epoch": 5.3722381591796876e-05,
      "model_forward_time": 1.7353055477142334,
      "step": 56332
    },
    {
      "epoch": 5.3722381591796876e-05,
      "step": 56332,
      "training_step_time": 8.70926833152771
    },
    {
      "epoch": 5.372333526611328e-05,
      "model_forward_time": 1.7458076477050781,
      "step": 56333
    },
    {
      "epoch": 5.372333526611328e-05,
      "step": 56333,
      "training_step_time": 7.686112403869629
    },
    {
      "epoch": 5.372428894042969e-05,
      "model_forward_time": 1.7557713985443115,
      "step": 56334
    },
    {
      "epoch": 5.372428894042969e-05,
      "step": 56334,
      "training_step_time": 8.718022584915161
    },
    {
      "epoch": 5.3725242614746095e-05,
      "model_forward_time": 1.7550678253173828,
      "step": 56335
    },
    {
      "epoch": 5.3725242614746095e-05,
      "step": 56335,
      "training_step_time": 8.24169135093689
    },
    {
      "epoch": 5.37261962890625e-05,
      "model_forward_time": 1.7534635066986084,
      "step": 56336
    },
    {
      "epoch": 5.37261962890625e-05,
      "step": 56336,
      "training_step_time": 7.416671276092529
    },
    {
      "epoch": 5.372714996337891e-05,
      "model_forward_time": 1.72489595413208,
      "step": 56337
    },
    {
      "epoch": 5.372714996337891e-05,
      "step": 56337,
      "training_step_time": 8.503782987594604
    },
    {
      "epoch": 5.3728103637695313e-05,
      "model_forward_time": 1.8189995288848877,
      "step": 56338
    },
    {
      "epoch": 5.3728103637695313e-05,
      "step": 56338,
      "training_step_time": 8.344289779663086
    },
    {
      "epoch": 5.372905731201172e-05,
      "model_forward_time": 1.7529761791229248,
      "step": 56339
    },
    {
      "epoch": 5.372905731201172e-05,
      "step": 56339,
      "training_step_time": 7.687623500823975
    },
    {
      "epoch": 5.373001098632813e-05,
      "grad_norm": 0.10901100933551788,
      "learning_rate": 4.366936671446544e-06,
      "loss": 0.0586,
      "step": 56340
    },
    {
      "epoch": 5.373001098632813e-05,
      "model_forward_time": 1.7616002559661865,
      "step": 56340
    },
    {
      "dynamics_loss_avg": 0.07174521591514349,
      "epoch": 5.373001098632813e-05,
      "step": 56340
    },
    {
      "action_loss_avg": 0.0022462407272541896,
      "epoch": 5.373001098632813e-05,
      "step": 56340
    },
    {
      "epoch": 5.373001098632813e-05,
      "step": 56340,
      "training_step_time": 8.491221189498901
    },
    {
      "epoch": 5.373096466064453e-05,
      "model_forward_time": 1.7470788955688477,
      "step": 56341
    },
    {
      "epoch": 5.373096466064453e-05,
      "step": 56341,
      "training_step_time": 7.678954839706421
    },
    {
      "epoch": 5.3731918334960936e-05,
      "model_forward_time": 2.937438488006592,
      "step": 56342
    },
    {
      "epoch": 5.3731918334960936e-05,
      "step": 56342,
      "training_step_time": 7.73919415473938
    },
    {
      "epoch": 5.373287200927735e-05,
      "model_forward_time": 1.7469584941864014,
      "step": 56343
    },
    {
      "epoch": 5.373287200927735e-05,
      "step": 56343,
      "training_step_time": 8.717596054077148
    },
    {
      "epoch": 5.373382568359375e-05,
      "model_forward_time": 1.7516987323760986,
      "step": 56344
    },
    {
      "epoch": 5.373382568359375e-05,
      "step": 56344,
      "training_step_time": 8.296980619430542
    },
    {
      "epoch": 5.3734779357910155e-05,
      "model_forward_time": 1.7497344017028809,
      "step": 56345
    },
    {
      "epoch": 5.3734779357910155e-05,
      "step": 56345,
      "training_step_time": 7.1898205280303955
    },
    {
      "epoch": 5.373573303222656e-05,
      "model_forward_time": 1.7479197978973389,
      "step": 56346
    },
    {
      "epoch": 5.373573303222656e-05,
      "step": 56346,
      "training_step_time": 7.858959674835205
    },
    {
      "epoch": 5.373668670654297e-05,
      "model_forward_time": 1.781984567642212,
      "step": 56347
    },
    {
      "epoch": 5.373668670654297e-05,
      "step": 56347,
      "training_step_time": 7.815313339233398
    },
    {
      "epoch": 5.3737640380859374e-05,
      "model_forward_time": 1.7501256465911865,
      "step": 56348
    },
    {
      "epoch": 5.3737640380859374e-05,
      "step": 56348,
      "training_step_time": 8.504997491836548
    },
    {
      "epoch": 5.373859405517578e-05,
      "model_forward_time": 1.753936529159546,
      "step": 56349
    },
    {
      "epoch": 5.373859405517578e-05,
      "step": 56349,
      "training_step_time": 8.505001783370972
    },
    {
      "epoch": 5.373954772949219e-05,
      "grad_norm": 0.1739376336336136,
      "learning_rate": 4.3652965430137476e-06,
      "loss": 0.0582,
      "step": 56350
    },
    {
      "epoch": 5.373954772949219e-05,
      "model_forward_time": 1.7450463771820068,
      "step": 56350
    },
    {
      "dynamics_loss_avg": 0.07676457613706589,
      "epoch": 5.373954772949219e-05,
      "step": 56350
    },
    {
      "action_loss_avg": 0.0018442066852003337,
      "epoch": 5.373954772949219e-05,
      "step": 56350
    },
    {
      "epoch": 5.373954772949219e-05,
      "step": 56350,
      "training_step_time": 7.630499601364136
    },
    {
      "epoch": 5.374050140380859e-05,
      "model_forward_time": 2.9160358905792236,
      "step": 56351
    },
    {
      "epoch": 5.374050140380859e-05,
      "step": 56351,
      "training_step_time": 7.655276775360107
    },
    {
      "epoch": 5.3741455078125e-05,
      "model_forward_time": 1.750415325164795,
      "step": 56352
    },
    {
      "epoch": 5.3741455078125e-05,
      "step": 56352,
      "training_step_time": 7.845140695571899
    },
    {
      "epoch": 5.374240875244141e-05,
      "model_forward_time": 1.7440881729125977,
      "step": 56353
    },
    {
      "epoch": 5.374240875244141e-05,
      "step": 56353,
      "training_step_time": 7.682663679122925
    },
    {
      "epoch": 5.374336242675781e-05,
      "model_forward_time": 1.7529878616333008,
      "step": 56354
    },
    {
      "epoch": 5.374336242675781e-05,
      "step": 56354,
      "training_step_time": 8.150379657745361
    },
    {
      "epoch": 5.3744316101074216e-05,
      "model_forward_time": 1.755239486694336,
      "step": 56355
    },
    {
      "epoch": 5.3744316101074216e-05,
      "step": 56355,
      "training_step_time": 8.43248438835144
    },
    {
      "epoch": 5.3745269775390627e-05,
      "model_forward_time": 1.7674586772918701,
      "step": 56356
    },
    {
      "epoch": 5.3745269775390627e-05,
      "step": 56356,
      "training_step_time": 7.709118604660034
    },
    {
      "epoch": 5.374622344970703e-05,
      "model_forward_time": 1.744119644165039,
      "step": 56357
    },
    {
      "epoch": 5.374622344970703e-05,
      "step": 56357,
      "training_step_time": 9.024670124053955
    },
    {
      "epoch": 5.3747177124023435e-05,
      "model_forward_time": 1.749955177307129,
      "step": 56358
    },
    {
      "epoch": 5.3747177124023435e-05,
      "step": 56358,
      "training_step_time": 8.265390157699585
    },
    {
      "epoch": 5.3748130798339845e-05,
      "model_forward_time": 1.7496294975280762,
      "step": 56359
    },
    {
      "epoch": 5.3748130798339845e-05,
      "step": 56359,
      "training_step_time": 8.143916130065918
    },
    {
      "epoch": 5.374908447265625e-05,
      "grad_norm": 0.10504743456840515,
      "learning_rate": 4.363656483991165e-06,
      "loss": 0.059,
      "step": 56360
    },
    {
      "epoch": 5.374908447265625e-05,
      "model_forward_time": 2.9323742389678955,
      "step": 56360
    },
    {
      "dynamics_loss_avg": 0.05094317933544516,
      "epoch": 5.374908447265625e-05,
      "step": 56360
    },
    {
      "action_loss_avg": 0.001909265259746462,
      "epoch": 5.374908447265625e-05,
      "step": 56360
    },
    {
      "epoch": 5.374908447265625e-05,
      "step": 56360,
      "training_step_time": 7.705849885940552
    },
    {
      "epoch": 5.3750038146972653e-05,
      "model_forward_time": 1.7542893886566162,
      "step": 56361
    },
    {
      "epoch": 5.3750038146972653e-05,
      "step": 56361,
      "training_step_time": 7.637293577194214
    },
    {
      "epoch": 5.3750991821289064e-05,
      "model_forward_time": 1.7504444122314453,
      "step": 56362
    },
    {
      "epoch": 5.3750991821289064e-05,
      "step": 56362,
      "training_step_time": 7.658624887466431
    },
    {
      "epoch": 5.375194549560547e-05,
      "model_forward_time": 1.3996500968933105,
      "step": 56363
    },
    {
      "epoch": 5.375194549560547e-05,
      "step": 56363,
      "training_step_time": 7.671196222305298
    },
    {
      "epoch": 5.375289916992187e-05,
      "model_forward_time": 1.740166187286377,
      "step": 56364
    },
    {
      "epoch": 5.375289916992187e-05,
      "step": 56364,
      "training_step_time": 7.593039512634277
    },
    {
      "epoch": 5.375385284423828e-05,
      "model_forward_time": 1.7258343696594238,
      "step": 56365
    },
    {
      "epoch": 5.375385284423828e-05,
      "step": 56365,
      "training_step_time": 8.240273714065552
    },
    {
      "epoch": 5.375480651855469e-05,
      "model_forward_time": 1.8167402744293213,
      "step": 56366
    },
    {
      "epoch": 5.375480651855469e-05,
      "step": 56366,
      "training_step_time": 9.147967100143433
    },
    {
      "epoch": 5.375576019287109e-05,
      "model_forward_time": 1.7418639659881592,
      "step": 56367
    },
    {
      "epoch": 5.375576019287109e-05,
      "step": 56367,
      "training_step_time": 9.308626651763916
    },
    {
      "epoch": 5.37567138671875e-05,
      "model_forward_time": 1.7549586296081543,
      "step": 56368
    },
    {
      "epoch": 5.37567138671875e-05,
      "step": 56368,
      "training_step_time": 7.566056728363037
    },
    {
      "epoch": 5.3757667541503906e-05,
      "model_forward_time": 4.089136123657227,
      "step": 56369
    },
    {
      "epoch": 5.3757667541503906e-05,
      "step": 56369,
      "training_step_time": 10.797350883483887
    },
    {
      "epoch": 5.375862121582031e-05,
      "grad_norm": 0.11558056622743607,
      "learning_rate": 4.362016494558155e-06,
      "loss": 0.059,
      "step": 56370
    },
    {
      "epoch": 5.375862121582031e-05,
      "model_forward_time": 1.7342207431793213,
      "step": 56370
    },
    {
      "dynamics_loss_avg": 0.0522579874843359,
      "epoch": 5.375862121582031e-05,
      "step": 56370
    },
    {
      "action_loss_avg": 0.0014017868437804283,
      "epoch": 5.375862121582031e-05,
      "step": 56370
    },
    {
      "epoch": 5.375862121582031e-05,
      "step": 56370,
      "training_step_time": 6.719346523284912
    },
    {
      "epoch": 5.375957489013672e-05,
      "model_forward_time": 1.7319738864898682,
      "step": 56371
    },
    {
      "epoch": 5.375957489013672e-05,
      "step": 56371,
      "training_step_time": 6.697720766067505
    },
    {
      "epoch": 5.3760528564453125e-05,
      "model_forward_time": 1.735840082168579,
      "step": 56372
    },
    {
      "epoch": 5.3760528564453125e-05,
      "step": 56372,
      "training_step_time": 6.722548246383667
    },
    {
      "epoch": 5.376148223876953e-05,
      "model_forward_time": 1.7301790714263916,
      "step": 56373
    },
    {
      "epoch": 5.376148223876953e-05,
      "step": 56373,
      "training_step_time": 6.761440277099609
    },
    {
      "epoch": 5.376243591308594e-05,
      "model_forward_time": 1.7346651554107666,
      "step": 56374
    },
    {
      "epoch": 5.376243591308594e-05,
      "step": 56374,
      "training_step_time": 6.657802581787109
    },
    {
      "epoch": 5.3763389587402344e-05,
      "model_forward_time": 1.7315948009490967,
      "step": 56375
    },
    {
      "epoch": 5.3763389587402344e-05,
      "step": 56375,
      "training_step_time": 6.7267844676971436
    },
    {
      "epoch": 5.376434326171875e-05,
      "model_forward_time": 1.736335277557373,
      "step": 56376
    },
    {
      "epoch": 5.376434326171875e-05,
      "step": 56376,
      "training_step_time": 6.709080457687378
    },
    {
      "epoch": 5.376529693603516e-05,
      "model_forward_time": 1.7361047267913818,
      "step": 56377
    },
    {
      "epoch": 5.376529693603516e-05,
      "step": 56377,
      "training_step_time": 6.6993489265441895
    },
    {
      "epoch": 5.376625061035156e-05,
      "model_forward_time": 2.9449820518493652,
      "step": 56378
    },
    {
      "epoch": 5.376625061035156e-05,
      "step": 56378,
      "training_step_time": 10.498883247375488
    },
    {
      "epoch": 5.376720428466797e-05,
      "model_forward_time": 1.7330639362335205,
      "step": 56379
    },
    {
      "epoch": 5.376720428466797e-05,
      "step": 56379,
      "training_step_time": 7.66007399559021
    },
    {
      "epoch": 5.376815795898438e-05,
      "grad_norm": 0.09390266239643097,
      "learning_rate": 4.360376574894062e-06,
      "loss": 0.0594,
      "step": 56380
    },
    {
      "epoch": 5.376815795898438e-05,
      "model_forward_time": 1.7351834774017334,
      "step": 56380
    },
    {
      "dynamics_loss_avg": 0.05042936690151691,
      "epoch": 5.376815795898438e-05,
      "step": 56380
    },
    {
      "action_loss_avg": 0.0014965541107812897,
      "epoch": 5.376815795898438e-05,
      "step": 56380
    },
    {
      "epoch": 5.376815795898438e-05,
      "step": 56380,
      "training_step_time": 6.667421817779541
    },
    {
      "epoch": 5.376911163330078e-05,
      "model_forward_time": 1.7337021827697754,
      "step": 56381
    },
    {
      "epoch": 5.376911163330078e-05,
      "step": 56381,
      "training_step_time": 6.6972739696502686
    },
    {
      "epoch": 5.3770065307617185e-05,
      "model_forward_time": 1.7327771186828613,
      "step": 56382
    },
    {
      "epoch": 5.3770065307617185e-05,
      "step": 56382,
      "training_step_time": 6.685335397720337
    },
    {
      "epoch": 5.3771018981933596e-05,
      "model_forward_time": 1.7327852249145508,
      "step": 56383
    },
    {
      "epoch": 5.3771018981933596e-05,
      "step": 56383,
      "training_step_time": 6.665876388549805
    },
    {
      "epoch": 5.377197265625e-05,
      "model_forward_time": 1.730119228363037,
      "step": 56384
    },
    {
      "epoch": 5.377197265625e-05,
      "step": 56384,
      "training_step_time": 6.695561408996582
    },
    {
      "epoch": 5.3772926330566404e-05,
      "model_forward_time": 1.730475664138794,
      "step": 56385
    },
    {
      "epoch": 5.3772926330566404e-05,
      "step": 56385,
      "training_step_time": 6.659147024154663
    },
    {
      "epoch": 5.3773880004882815e-05,
      "model_forward_time": 1.7342503070831299,
      "step": 56386
    },
    {
      "epoch": 5.3773880004882815e-05,
      "step": 56386,
      "training_step_time": 6.699149131774902
    },
    {
      "epoch": 5.377483367919922e-05,
      "model_forward_time": 2.9437942504882812,
      "step": 56387
    },
    {
      "epoch": 5.377483367919922e-05,
      "step": 56387,
      "training_step_time": 8.594444274902344
    },
    {
      "epoch": 5.377578735351562e-05,
      "model_forward_time": 1.732450246810913,
      "step": 56388
    },
    {
      "epoch": 5.377578735351562e-05,
      "step": 56388,
      "training_step_time": 7.740139484405518
    },
    {
      "epoch": 5.3776741027832034e-05,
      "model_forward_time": 1.7322545051574707,
      "step": 56389
    },
    {
      "epoch": 5.3776741027832034e-05,
      "step": 56389,
      "training_step_time": 6.69255805015564
    },
    {
      "epoch": 5.377769470214844e-05,
      "grad_norm": 0.08868101984262466,
      "learning_rate": 4.358736725178225e-06,
      "loss": 0.0589,
      "step": 56390
    },
    {
      "epoch": 5.377769470214844e-05,
      "model_forward_time": 1.731820821762085,
      "step": 56390
    },
    {
      "dynamics_loss_avg": 0.046692838333547114,
      "epoch": 5.377769470214844e-05,
      "step": 56390
    },
    {
      "action_loss_avg": 0.0013512340956367552,
      "epoch": 5.377769470214844e-05,
      "step": 56390
    },
    {
      "epoch": 5.377769470214844e-05,
      "step": 56390,
      "training_step_time": 6.700136661529541
    },
    {
      "epoch": 5.377864837646484e-05,
      "model_forward_time": 1.7321724891662598,
      "step": 56391
    },
    {
      "epoch": 5.377864837646484e-05,
      "step": 56391,
      "training_step_time": 6.719480991363525
    },
    {
      "epoch": 5.377960205078125e-05,
      "model_forward_time": 1.7348227500915527,
      "step": 56392
    },
    {
      "epoch": 5.377960205078125e-05,
      "step": 56392,
      "training_step_time": 6.681420564651489
    },
    {
      "epoch": 5.378055572509766e-05,
      "model_forward_time": 1.73160719871521,
      "step": 56393
    },
    {
      "epoch": 5.378055572509766e-05,
      "step": 56393,
      "training_step_time": 6.679503917694092
    },
    {
      "epoch": 5.378150939941406e-05,
      "model_forward_time": 1.7344157695770264,
      "step": 56394
    },
    {
      "epoch": 5.378150939941406e-05,
      "step": 56394,
      "training_step_time": 6.729717016220093
    },
    {
      "epoch": 5.378246307373047e-05,
      "model_forward_time": 1.734678030014038,
      "step": 56395
    },
    {
      "epoch": 5.378246307373047e-05,
      "step": 56395,
      "training_step_time": 6.701635122299194
    },
    {
      "epoch": 5.3783416748046876e-05,
      "model_forward_time": 2.903775691986084,
      "step": 56396
    },
    {
      "epoch": 5.3783416748046876e-05,
      "step": 56396,
      "training_step_time": 8.527693510055542
    },
    {
      "epoch": 5.378437042236328e-05,
      "model_forward_time": 1.8196687698364258,
      "step": 56397
    },
    {
      "epoch": 5.378437042236328e-05,
      "step": 56397,
      "training_step_time": 7.688851833343506
    },
    {
      "epoch": 5.378532409667969e-05,
      "model_forward_time": 1.366478681564331,
      "step": 56398
    },
    {
      "epoch": 5.378532409667969e-05,
      "step": 56398,
      "training_step_time": 7.580410718917847
    },
    {
      "epoch": 5.3786277770996095e-05,
      "model_forward_time": 1.7423899173736572,
      "step": 56399
    },
    {
      "epoch": 5.3786277770996095e-05,
      "step": 56399,
      "training_step_time": 6.726303577423096
    },
    {
      "epoch": 5.37872314453125e-05,
      "grad_norm": 0.0952262282371521,
      "learning_rate": 4.357096945589974e-06,
      "loss": 0.0579,
      "step": 56400
    },
    {
      "epoch": 5.37872314453125e-05,
      "model_forward_time": 1.7354683876037598,
      "step": 56400
    },
    {
      "dynamics_loss_avg": 0.04707135818898678,
      "epoch": 5.37872314453125e-05,
      "step": 56400
    },
    {
      "action_loss_avg": 0.0014160638849716634,
      "epoch": 5.37872314453125e-05,
      "step": 56400
    },
    {
      "epoch": 5.37872314453125e-05,
      "step": 56400,
      "training_step_time": 6.715271234512329
    },
    {
      "epoch": 5.378818511962891e-05,
      "model_forward_time": 1.7337055206298828,
      "step": 56401
    },
    {
      "epoch": 5.378818511962891e-05,
      "step": 56401,
      "training_step_time": 6.727737188339233
    },
    {
      "epoch": 5.3789138793945313e-05,
      "model_forward_time": 1.7348873615264893,
      "step": 56402
    },
    {
      "epoch": 5.3789138793945313e-05,
      "step": 56402,
      "training_step_time": 6.742262363433838
    },
    {
      "epoch": 5.379009246826172e-05,
      "model_forward_time": 1.7319657802581787,
      "step": 56403
    },
    {
      "epoch": 5.379009246826172e-05,
      "step": 56403,
      "training_step_time": 6.682644367218018
    },
    {
      "epoch": 5.379104614257813e-05,
      "model_forward_time": 1.7330772876739502,
      "step": 56404
    },
    {
      "epoch": 5.379104614257813e-05,
      "step": 56404,
      "training_step_time": 6.708566904067993
    },
    {
      "epoch": 5.379199981689453e-05,
      "model_forward_time": 1.7315211296081543,
      "step": 56405
    },
    {
      "epoch": 5.379199981689453e-05,
      "step": 56405,
      "training_step_time": 6.6834611892700195
    },
    {
      "epoch": 5.3792953491210936e-05,
      "model_forward_time": 2.9357807636260986,
      "step": 56406
    },
    {
      "epoch": 5.3792953491210936e-05,
      "step": 56406,
      "training_step_time": 8.29590129852295
    },
    {
      "epoch": 5.379390716552734e-05,
      "model_forward_time": 1.7252440452575684,
      "step": 56407
    },
    {
      "epoch": 5.379390716552734e-05,
      "step": 56407,
      "training_step_time": 7.630537986755371
    },
    {
      "epoch": 5.379486083984375e-05,
      "model_forward_time": 1.7480037212371826,
      "step": 56408
    },
    {
      "epoch": 5.379486083984375e-05,
      "step": 56408,
      "training_step_time": 6.744633436203003
    },
    {
      "epoch": 5.3795814514160155e-05,
      "model_forward_time": 1.7465648651123047,
      "step": 56409
    },
    {
      "epoch": 5.3795814514160155e-05,
      "step": 56409,
      "training_step_time": 6.726048469543457
    },
    {
      "epoch": 5.379676818847656e-05,
      "grad_norm": 0.08501104265451431,
      "learning_rate": 4.355457236308636e-06,
      "loss": 0.0582,
      "step": 56410
    },
    {
      "epoch": 5.379676818847656e-05,
      "model_forward_time": 1.7484614849090576,
      "step": 56410
    },
    {
      "dynamics_loss_avg": 0.07239742912352085,
      "epoch": 5.379676818847656e-05,
      "step": 56410
    },
    {
      "action_loss_avg": 0.002579827161389403,
      "epoch": 5.379676818847656e-05,
      "step": 56410
    },
    {
      "epoch": 5.379676818847656e-05,
      "step": 56410,
      "training_step_time": 6.743661642074585
    },
    {
      "epoch": 5.379772186279297e-05,
      "model_forward_time": 1.749758005142212,
      "step": 56411
    },
    {
      "epoch": 5.379772186279297e-05,
      "step": 56411,
      "training_step_time": 6.758756160736084
    },
    {
      "epoch": 5.3798675537109374e-05,
      "model_forward_time": 1.7501935958862305,
      "step": 56412
    },
    {
      "epoch": 5.3798675537109374e-05,
      "step": 56412,
      "training_step_time": 6.6869261264801025
    },
    {
      "epoch": 5.379962921142578e-05,
      "model_forward_time": 1.7471075057983398,
      "step": 56413
    },
    {
      "epoch": 5.379962921142578e-05,
      "step": 56413,
      "training_step_time": 6.7071239948272705
    },
    {
      "epoch": 5.380058288574219e-05,
      "model_forward_time": 1.7466161251068115,
      "step": 56414
    },
    {
      "epoch": 5.380058288574219e-05,
      "step": 56414,
      "training_step_time": 6.6953370571136475
    },
    {
      "epoch": 5.380153656005859e-05,
      "model_forward_time": 2.940797805786133,
      "step": 56415
    },
    {
      "epoch": 5.380153656005859e-05,
      "step": 56415,
      "training_step_time": 10.114198684692383
    },
    {
      "epoch": 5.3802490234375e-05,
      "model_forward_time": 1.7481555938720703,
      "step": 56416
    },
    {
      "epoch": 5.3802490234375e-05,
      "step": 56416,
      "training_step_time": 9.510912656784058
    },
    {
      "epoch": 5.380344390869141e-05,
      "model_forward_time": 1.7475500106811523,
      "step": 56417
    },
    {
      "epoch": 5.380344390869141e-05,
      "step": 56417,
      "training_step_time": 7.311850070953369
    },
    {
      "epoch": 5.380439758300781e-05,
      "model_forward_time": 1.7478110790252686,
      "step": 56418
    },
    {
      "epoch": 5.380439758300781e-05,
      "step": 56418,
      "training_step_time": 6.741680383682251
    },
    {
      "epoch": 5.3805351257324216e-05,
      "model_forward_time": 1.741436243057251,
      "step": 56419
    },
    {
      "epoch": 5.3805351257324216e-05,
      "step": 56419,
      "training_step_time": 6.751614332199097
    },
    {
      "epoch": 5.380630493164063e-05,
      "grad_norm": 0.13121290504932404,
      "learning_rate": 4.3538175975135255e-06,
      "loss": 0.0594,
      "step": 56420
    },
    {
      "epoch": 5.380630493164063e-05,
      "model_forward_time": 1.7413561344146729,
      "step": 56420
    },
    {
      "dynamics_loss_avg": 0.06075434163212776,
      "epoch": 5.380630493164063e-05,
      "step": 56420
    },
    {
      "action_loss_avg": 0.0016260013624560087,
      "epoch": 5.380630493164063e-05,
      "step": 56420
    },
    {
      "epoch": 5.380630493164063e-05,
      "step": 56420,
      "training_step_time": 6.7604687213897705
    },
    {
      "epoch": 5.380725860595703e-05,
      "model_forward_time": 1.7485268115997314,
      "step": 56421
    },
    {
      "epoch": 5.380725860595703e-05,
      "step": 56421,
      "training_step_time": 6.781866788864136
    },
    {
      "epoch": 5.3808212280273435e-05,
      "model_forward_time": 1.740706443786621,
      "step": 56422
    },
    {
      "epoch": 5.3808212280273435e-05,
      "step": 56422,
      "training_step_time": 6.789997339248657
    },
    {
      "epoch": 5.3809165954589845e-05,
      "model_forward_time": 1.7411754131317139,
      "step": 56423
    },
    {
      "epoch": 5.3809165954589845e-05,
      "step": 56423,
      "training_step_time": 6.813535213470459
    },
    {
      "epoch": 5.381011962890625e-05,
      "model_forward_time": 2.950453042984009,
      "step": 56424
    },
    {
      "epoch": 5.381011962890625e-05,
      "step": 56424,
      "training_step_time": 8.869497299194336
    },
    {
      "epoch": 5.3811073303222654e-05,
      "model_forward_time": 1.72857666015625,
      "step": 56425
    },
    {
      "epoch": 5.3811073303222654e-05,
      "step": 56425,
      "training_step_time": 7.793875694274902
    },
    {
      "epoch": 5.3812026977539064e-05,
      "model_forward_time": 1.7365031242370605,
      "step": 56426
    },
    {
      "epoch": 5.3812026977539064e-05,
      "step": 56426,
      "training_step_time": 7.655348062515259
    },
    {
      "epoch": 5.381298065185547e-05,
      "model_forward_time": 1.736989974975586,
      "step": 56427
    },
    {
      "epoch": 5.381298065185547e-05,
      "step": 56427,
      "training_step_time": 7.404210567474365
    },
    {
      "epoch": 5.381393432617187e-05,
      "model_forward_time": 1.7324726581573486,
      "step": 56428
    },
    {
      "epoch": 5.381393432617187e-05,
      "step": 56428,
      "training_step_time": 6.8017988204956055
    },
    {
      "epoch": 5.381488800048828e-05,
      "model_forward_time": 1.733168363571167,
      "step": 56429
    },
    {
      "epoch": 5.381488800048828e-05,
      "step": 56429,
      "training_step_time": 6.800715208053589
    },
    {
      "epoch": 5.381584167480469e-05,
      "grad_norm": 0.06556398421525955,
      "learning_rate": 4.3521780293839484e-06,
      "loss": 0.058,
      "step": 56430
    },
    {
      "epoch": 5.381584167480469e-05,
      "model_forward_time": 1.7338004112243652,
      "step": 56430
    },
    {
      "dynamics_loss_avg": 0.053192059695720675,
      "epoch": 5.381584167480469e-05,
      "step": 56430
    },
    {
      "action_loss_avg": 0.0018606826604809612,
      "epoch": 5.381584167480469e-05,
      "step": 56430
    },
    {
      "epoch": 5.381584167480469e-05,
      "step": 56430,
      "training_step_time": 6.775809049606323
    },
    {
      "epoch": 5.381679534912109e-05,
      "model_forward_time": 1.7364561557769775,
      "step": 56431
    },
    {
      "epoch": 5.381679534912109e-05,
      "step": 56431,
      "training_step_time": 6.781659841537476
    },
    {
      "epoch": 5.38177490234375e-05,
      "model_forward_time": 1.7364137172698975,
      "step": 56432
    },
    {
      "epoch": 5.38177490234375e-05,
      "step": 56432,
      "training_step_time": 6.781048774719238
    },
    {
      "epoch": 5.3818702697753906e-05,
      "model_forward_time": 2.893317937850952,
      "step": 56433
    },
    {
      "epoch": 5.3818702697753906e-05,
      "step": 56433,
      "training_step_time": 11.167875051498413
    },
    {
      "epoch": 5.381965637207031e-05,
      "model_forward_time": 1.739480972290039,
      "step": 56434
    },
    {
      "epoch": 5.381965637207031e-05,
      "step": 56434,
      "training_step_time": 7.816895246505737
    },
    {
      "epoch": 5.382061004638672e-05,
      "model_forward_time": 1.7294788360595703,
      "step": 56435
    },
    {
      "epoch": 5.382061004638672e-05,
      "step": 56435,
      "training_step_time": 7.681634902954102
    },
    {
      "epoch": 5.3821563720703125e-05,
      "model_forward_time": 1.7341492176055908,
      "step": 56436
    },
    {
      "epoch": 5.3821563720703125e-05,
      "step": 56436,
      "training_step_time": 7.3688201904296875
    },
    {
      "epoch": 5.382251739501953e-05,
      "model_forward_time": 1.7258844375610352,
      "step": 56437
    },
    {
      "epoch": 5.382251739501953e-05,
      "step": 56437,
      "training_step_time": 6.661692380905151
    },
    {
      "epoch": 5.382347106933594e-05,
      "model_forward_time": 1.729015588760376,
      "step": 56438
    },
    {
      "epoch": 5.382347106933594e-05,
      "step": 56438,
      "training_step_time": 6.661562204360962
    },
    {
      "epoch": 5.3824424743652344e-05,
      "model_forward_time": 1.7387700080871582,
      "step": 56439
    },
    {
      "epoch": 5.3824424743652344e-05,
      "step": 56439,
      "training_step_time": 6.715864658355713
    },
    {
      "epoch": 5.382537841796875e-05,
      "grad_norm": 0.13045980036258698,
      "learning_rate": 4.35053853209921e-06,
      "loss": 0.0568,
      "step": 56440
    },
    {
      "epoch": 5.382537841796875e-05,
      "model_forward_time": 1.7380986213684082,
      "step": 56440
    },
    {
      "dynamics_loss_avg": 0.05470512509346008,
      "epoch": 5.382537841796875e-05,
      "step": 56440
    },
    {
      "action_loss_avg": 0.0028311849455349147,
      "epoch": 5.382537841796875e-05,
      "step": 56440
    },
    {
      "epoch": 5.382537841796875e-05,
      "step": 56440,
      "training_step_time": 6.707750558853149
    },
    {
      "epoch": 5.382633209228516e-05,
      "model_forward_time": 1.7471272945404053,
      "step": 56441
    },
    {
      "epoch": 5.382633209228516e-05,
      "step": 56441,
      "training_step_time": 6.716888904571533
    },
    {
      "epoch": 5.382728576660156e-05,
      "model_forward_time": 2.901118755340576,
      "step": 56442
    },
    {
      "epoch": 5.382728576660156e-05,
      "step": 56442,
      "training_step_time": 8.766333103179932
    },
    {
      "epoch": 5.382823944091797e-05,
      "model_forward_time": 1.787665605545044,
      "step": 56443
    },
    {
      "epoch": 5.382823944091797e-05,
      "step": 56443,
      "training_step_time": 8.303903102874756
    },
    {
      "epoch": 5.382919311523438e-05,
      "model_forward_time": 1.7458555698394775,
      "step": 56444
    },
    {
      "epoch": 5.382919311523438e-05,
      "step": 56444,
      "training_step_time": 7.597280979156494
    },
    {
      "epoch": 5.383014678955078e-05,
      "model_forward_time": 1.7467031478881836,
      "step": 56445
    },
    {
      "epoch": 5.383014678955078e-05,
      "step": 56445,
      "training_step_time": 7.320265293121338
    },
    {
      "epoch": 5.3831100463867186e-05,
      "model_forward_time": 1.7387518882751465,
      "step": 56446
    },
    {
      "epoch": 5.3831100463867186e-05,
      "step": 56446,
      "training_step_time": 6.773347854614258
    },
    {
      "epoch": 5.3832054138183596e-05,
      "model_forward_time": 1.742967128753662,
      "step": 56447
    },
    {
      "epoch": 5.3832054138183596e-05,
      "step": 56447,
      "training_step_time": 6.752065658569336
    },
    {
      "epoch": 5.38330078125e-05,
      "model_forward_time": 1.7480511665344238,
      "step": 56448
    },
    {
      "epoch": 5.38330078125e-05,
      "step": 56448,
      "training_step_time": 6.758472442626953
    },
    {
      "epoch": 5.3833961486816404e-05,
      "model_forward_time": 1.7483251094818115,
      "step": 56449
    },
    {
      "epoch": 5.3833961486816404e-05,
      "step": 56449,
      "training_step_time": 6.743441820144653
    },
    {
      "epoch": 5.3834915161132815e-05,
      "grad_norm": 0.13115888833999634,
      "learning_rate": 4.348899105838602e-06,
      "loss": 0.0641,
      "step": 56450
    },
    {
      "epoch": 5.3834915161132815e-05,
      "model_forward_time": 1.7448854446411133,
      "step": 56450
    },
    {
      "dynamics_loss_avg": 0.04721247162669897,
      "epoch": 5.3834915161132815e-05,
      "step": 56450
    },
    {
      "action_loss_avg": 0.002104603365296498,
      "epoch": 5.3834915161132815e-05,
      "step": 56450
    },
    {
      "epoch": 5.3834915161132815e-05,
      "step": 56450,
      "training_step_time": 6.777876377105713
    },
    {
      "epoch": 5.383586883544922e-05,
      "model_forward_time": 1.7627429962158203,
      "step": 56451
    },
    {
      "epoch": 5.383586883544922e-05,
      "step": 56451,
      "training_step_time": 9.808582782745361
    },
    {
      "epoch": 5.383682250976562e-05,
      "model_forward_time": 1.7922923564910889,
      "step": 56452
    },
    {
      "epoch": 5.383682250976562e-05,
      "step": 56452,
      "training_step_time": 8.577430486679077
    },
    {
      "epoch": 5.3837776184082034e-05,
      "model_forward_time": 1.7576260566711426,
      "step": 56453
    },
    {
      "epoch": 5.3837776184082034e-05,
      "step": 56453,
      "training_step_time": 7.729203462600708
    },
    {
      "epoch": 5.383872985839844e-05,
      "model_forward_time": 1.758833408355713,
      "step": 56454
    },
    {
      "epoch": 5.383872985839844e-05,
      "step": 56454,
      "training_step_time": 7.4832446575164795
    },
    {
      "epoch": 5.383968353271484e-05,
      "model_forward_time": 1.7564449310302734,
      "step": 56455
    },
    {
      "epoch": 5.383968353271484e-05,
      "step": 56455,
      "training_step_time": 6.82504940032959
    },
    {
      "epoch": 5.384063720703125e-05,
      "model_forward_time": 1.7552640438079834,
      "step": 56456
    },
    {
      "epoch": 5.384063720703125e-05,
      "step": 56456,
      "training_step_time": 6.745295286178589
    },
    {
      "epoch": 5.384159088134766e-05,
      "model_forward_time": 1.7564198970794678,
      "step": 56457
    },
    {
      "epoch": 5.384159088134766e-05,
      "step": 56457,
      "training_step_time": 6.729346036911011
    },
    {
      "epoch": 5.384254455566406e-05,
      "model_forward_time": 1.7538964748382568,
      "step": 56458
    },
    {
      "epoch": 5.384254455566406e-05,
      "step": 56458,
      "training_step_time": 6.784505367279053
    },
    {
      "epoch": 5.384349822998047e-05,
      "model_forward_time": 1.7622630596160889,
      "step": 56459
    },
    {
      "epoch": 5.384349822998047e-05,
      "step": 56459,
      "training_step_time": 6.734247446060181
    },
    {
      "epoch": 5.3844451904296876e-05,
      "grad_norm": 0.11750732362270355,
      "learning_rate": 4.347259750781409e-06,
      "loss": 0.0636,
      "step": 56460
    },
    {
      "epoch": 5.3844451904296876e-05,
      "model_forward_time": 1.7542564868927002,
      "step": 56460
    },
    {
      "dynamics_loss_avg": 0.06061064526438713,
      "epoch": 5.3844451904296876e-05,
      "step": 56460
    },
    {
      "action_loss_avg": 0.0012461477017495781,
      "epoch": 5.3844451904296876e-05,
      "step": 56460
    },
    {
      "epoch": 5.3844451904296876e-05,
      "step": 56460,
      "training_step_time": 7.708778142929077
    },
    {
      "epoch": 5.384540557861328e-05,
      "model_forward_time": 2.9439804553985596,
      "step": 56461
    },
    {
      "epoch": 5.384540557861328e-05,
      "step": 56461,
      "training_step_time": 8.800437927246094
    },
    {
      "epoch": 5.384635925292969e-05,
      "model_forward_time": 1.756580114364624,
      "step": 56462
    },
    {
      "epoch": 5.384635925292969e-05,
      "step": 56462,
      "training_step_time": 8.558241844177246
    },
    {
      "epoch": 5.3847312927246095e-05,
      "model_forward_time": 1.7141320705413818,
      "step": 56463
    },
    {
      "epoch": 5.3847312927246095e-05,
      "step": 56463,
      "training_step_time": 7.547496557235718
    },
    {
      "epoch": 5.38482666015625e-05,
      "model_forward_time": 1.7506613731384277,
      "step": 56464
    },
    {
      "epoch": 5.38482666015625e-05,
      "step": 56464,
      "training_step_time": 7.3830037117004395
    },
    {
      "epoch": 5.384922027587891e-05,
      "model_forward_time": 1.7278594970703125,
      "step": 56465
    },
    {
      "epoch": 5.384922027587891e-05,
      "step": 56465,
      "training_step_time": 6.866931200027466
    },
    {
      "epoch": 5.3850173950195314e-05,
      "model_forward_time": 1.7308082580566406,
      "step": 56466
    },
    {
      "epoch": 5.3850173950195314e-05,
      "step": 56466,
      "training_step_time": 6.771111011505127
    },
    {
      "epoch": 5.385112762451172e-05,
      "model_forward_time": 1.7493245601654053,
      "step": 56467
    },
    {
      "epoch": 5.385112762451172e-05,
      "step": 56467,
      "training_step_time": 6.7574782371521
    },
    {
      "epoch": 5.385208129882813e-05,
      "model_forward_time": 1.7349812984466553,
      "step": 56468
    },
    {
      "epoch": 5.385208129882813e-05,
      "step": 56468,
      "training_step_time": 6.742330074310303
    },
    {
      "epoch": 5.385303497314453e-05,
      "model_forward_time": 1.7465851306915283,
      "step": 56469
    },
    {
      "epoch": 5.385303497314453e-05,
      "step": 56469,
      "training_step_time": 7.6409406661987305
    },
    {
      "epoch": 5.3853988647460936e-05,
      "grad_norm": 0.11633643507957458,
      "learning_rate": 4.345620467106907e-06,
      "loss": 0.0638,
      "step": 56470
    },
    {
      "epoch": 5.3853988647460936e-05,
      "model_forward_time": 2.909698963165283,
      "step": 56470
    },
    {
      "dynamics_loss_avg": 0.0646413067355752,
      "epoch": 5.3853988647460936e-05,
      "step": 56470
    },
    {
      "action_loss_avg": 0.0015950160450302064,
      "epoch": 5.3853988647460936e-05,
      "step": 56470
    },
    {
      "epoch": 5.3853988647460936e-05,
      "step": 56470,
      "training_step_time": 10.163539409637451
    },
    {
      "epoch": 5.385494232177734e-05,
      "model_forward_time": 1.8388333320617676,
      "step": 56471
    },
    {
      "epoch": 5.385494232177734e-05,
      "step": 56471,
      "training_step_time": 7.799659729003906
    },
    {
      "epoch": 5.385589599609375e-05,
      "model_forward_time": 1.7361056804656982,
      "step": 56472
    },
    {
      "epoch": 5.385589599609375e-05,
      "step": 56472,
      "training_step_time": 7.596529006958008
    },
    {
      "epoch": 5.3856849670410155e-05,
      "model_forward_time": 1.7504820823669434,
      "step": 56473
    },
    {
      "epoch": 5.3856849670410155e-05,
      "step": 56473,
      "training_step_time": 7.239787340164185
    },
    {
      "epoch": 5.385780334472656e-05,
      "model_forward_time": 1.7477483749389648,
      "step": 56474
    },
    {
      "epoch": 5.385780334472656e-05,
      "step": 56474,
      "training_step_time": 6.753112554550171
    },
    {
      "epoch": 5.385875701904297e-05,
      "model_forward_time": 1.7430908679962158,
      "step": 56475
    },
    {
      "epoch": 5.385875701904297e-05,
      "step": 56475,
      "training_step_time": 6.736144304275513
    },
    {
      "epoch": 5.3859710693359374e-05,
      "model_forward_time": 1.7477614879608154,
      "step": 56476
    },
    {
      "epoch": 5.3859710693359374e-05,
      "step": 56476,
      "training_step_time": 6.713121652603149
    },
    {
      "epoch": 5.386066436767578e-05,
      "model_forward_time": 1.7478513717651367,
      "step": 56477
    },
    {
      "epoch": 5.386066436767578e-05,
      "step": 56477,
      "training_step_time": 6.709097385406494
    },
    {
      "epoch": 5.386161804199219e-05,
      "model_forward_time": 1.741875410079956,
      "step": 56478
    },
    {
      "epoch": 5.386161804199219e-05,
      "step": 56478,
      "training_step_time": 7.604071855545044
    },
    {
      "epoch": 5.386257171630859e-05,
      "model_forward_time": 2.9195406436920166,
      "step": 56479
    },
    {
      "epoch": 5.386257171630859e-05,
      "step": 56479,
      "training_step_time": 7.676875829696655
    },
    {
      "epoch": 5.3863525390625e-05,
      "grad_norm": 0.1521500200033188,
      "learning_rate": 4.343981254994367e-06,
      "loss": 0.0635,
      "step": 56480
    },
    {
      "epoch": 5.3863525390625e-05,
      "model_forward_time": 1.8398401737213135,
      "step": 56480
    },
    {
      "dynamics_loss_avg": 0.0593153165653348,
      "epoch": 5.3863525390625e-05,
      "step": 56480
    },
    {
      "action_loss_avg": 0.0018865675025153904,
      "epoch": 5.3863525390625e-05,
      "step": 56480
    },
    {
      "epoch": 5.3863525390625e-05,
      "step": 56480,
      "training_step_time": 8.891515016555786
    },
    {
      "epoch": 5.386447906494141e-05,
      "model_forward_time": 1.7356667518615723,
      "step": 56481
    },
    {
      "epoch": 5.386447906494141e-05,
      "step": 56481,
      "training_step_time": 7.612586736679077
    },
    {
      "epoch": 5.386543273925781e-05,
      "model_forward_time": 1.7449445724487305,
      "step": 56482
    },
    {
      "epoch": 5.386543273925781e-05,
      "step": 56482,
      "training_step_time": 6.957365989685059
    },
    {
      "epoch": 5.3866386413574216e-05,
      "model_forward_time": 1.7471723556518555,
      "step": 56483
    },
    {
      "epoch": 5.3866386413574216e-05,
      "step": 56483,
      "training_step_time": 6.754108190536499
    },
    {
      "epoch": 5.386734008789063e-05,
      "model_forward_time": 1.7378239631652832,
      "step": 56484
    },
    {
      "epoch": 5.386734008789063e-05,
      "step": 56484,
      "training_step_time": 6.7426722049713135
    },
    {
      "epoch": 5.386829376220703e-05,
      "model_forward_time": 1.7491354942321777,
      "step": 56485
    },
    {
      "epoch": 5.386829376220703e-05,
      "step": 56485,
      "training_step_time": 6.746578693389893
    },
    {
      "epoch": 5.3869247436523435e-05,
      "model_forward_time": 1.7350819110870361,
      "step": 56486
    },
    {
      "epoch": 5.3869247436523435e-05,
      "step": 56486,
      "training_step_time": 6.757192134857178
    },
    {
      "epoch": 5.3870201110839846e-05,
      "model_forward_time": 1.7450532913208008,
      "step": 56487
    },
    {
      "epoch": 5.3870201110839846e-05,
      "step": 56487,
      "training_step_time": 7.594325065612793
    },
    {
      "epoch": 5.387115478515625e-05,
      "model_forward_time": 2.8710639476776123,
      "step": 56488
    },
    {
      "epoch": 5.387115478515625e-05,
      "step": 56488,
      "training_step_time": 8.861494541168213
    },
    {
      "epoch": 5.3872108459472654e-05,
      "model_forward_time": 1.8468565940856934,
      "step": 56489
    },
    {
      "epoch": 5.3872108459472654e-05,
      "step": 56489,
      "training_step_time": 7.67106294631958
    },
    {
      "epoch": 5.3873062133789064e-05,
      "grad_norm": 0.1355382651090622,
      "learning_rate": 4.34234211462305e-06,
      "loss": 0.0571,
      "step": 56490
    },
    {
      "epoch": 5.3873062133789064e-05,
      "model_forward_time": 1.7342493534088135,
      "step": 56490
    },
    {
      "dynamics_loss_avg": 0.05967889837920666,
      "epoch": 5.3873062133789064e-05,
      "step": 56490
    },
    {
      "action_loss_avg": 0.0017398197145666928,
      "epoch": 5.3873062133789064e-05,
      "step": 56490
    },
    {
      "epoch": 5.3873062133789064e-05,
      "step": 56490,
      "training_step_time": 7.660475492477417
    },
    {
      "epoch": 5.387401580810547e-05,
      "model_forward_time": 1.735048532485962,
      "step": 56491
    },
    {
      "epoch": 5.387401580810547e-05,
      "step": 56491,
      "training_step_time": 7.297587156295776
    },
    {
      "epoch": 5.387496948242187e-05,
      "model_forward_time": 1.7339789867401123,
      "step": 56492
    },
    {
      "epoch": 5.387496948242187e-05,
      "step": 56492,
      "training_step_time": 6.709437131881714
    },
    {
      "epoch": 5.387592315673828e-05,
      "model_forward_time": 1.7422029972076416,
      "step": 56493
    },
    {
      "epoch": 5.387592315673828e-05,
      "step": 56493,
      "training_step_time": 6.677976369857788
    },
    {
      "epoch": 5.387687683105469e-05,
      "model_forward_time": 1.739647626876831,
      "step": 56494
    },
    {
      "epoch": 5.387687683105469e-05,
      "step": 56494,
      "training_step_time": 6.666775226593018
    },
    {
      "epoch": 5.387783050537109e-05,
      "model_forward_time": 1.7532012462615967,
      "step": 56495
    },
    {
      "epoch": 5.387783050537109e-05,
      "step": 56495,
      "training_step_time": 6.686520099639893
    },
    {
      "epoch": 5.38787841796875e-05,
      "model_forward_time": 1.7335882186889648,
      "step": 56496
    },
    {
      "epoch": 5.38787841796875e-05,
      "step": 56496,
      "training_step_time": 7.547075986862183
    },
    {
      "epoch": 5.3879737854003906e-05,
      "model_forward_time": 2.833813190460205,
      "step": 56497
    },
    {
      "epoch": 5.3879737854003906e-05,
      "step": 56497,
      "training_step_time": 7.649472951889038
    },
    {
      "epoch": 5.388069152832031e-05,
      "model_forward_time": 1.8444321155548096,
      "step": 56498
    },
    {
      "epoch": 5.388069152832031e-05,
      "step": 56498,
      "training_step_time": 9.359111309051514
    },
    {
      "epoch": 5.388164520263672e-05,
      "model_forward_time": 1.7637600898742676,
      "step": 56499
    },
    {
      "epoch": 5.388164520263672e-05,
      "step": 56499,
      "training_step_time": 8.634104490280151
    },
    {
      "epoch": 5.3882598876953125e-05,
      "grad_norm": 0.11338862776756287,
      "learning_rate": 4.340703046172213e-06,
      "loss": 0.0608,
      "step": 56500
    },
    {
      "epoch": 5.3882598876953125e-05,
      "model_forward_time": 1.759840726852417,
      "step": 56500
    },
    {
      "dynamics_loss_avg": 0.038574786856770515,
      "epoch": 5.3882598876953125e-05,
      "step": 56500
    },
    {
      "action_loss_avg": 0.000745841723983176,
      "epoch": 5.3882598876953125e-05,
      "step": 56500
    },
    {
      "epoch": 5.3882598876953125e-05,
      "step": 56500,
      "training_step_time": 7.4384925365448
    },
    {
      "epoch": 5.388355255126953e-05,
      "model_forward_time": 1.7596027851104736,
      "step": 56501
    },
    {
      "epoch": 5.388355255126953e-05,
      "step": 56501,
      "training_step_time": 6.677248477935791
    },
    {
      "epoch": 5.388450622558594e-05,
      "model_forward_time": 1.7612347602844238,
      "step": 56502
    },
    {
      "epoch": 5.388450622558594e-05,
      "step": 56502,
      "training_step_time": 6.6775782108306885
    },
    {
      "epoch": 5.3885459899902344e-05,
      "model_forward_time": 1.7639012336730957,
      "step": 56503
    },
    {
      "epoch": 5.3885459899902344e-05,
      "step": 56503,
      "training_step_time": 6.657254219055176
    },
    {
      "epoch": 5.388641357421875e-05,
      "model_forward_time": 1.7546107769012451,
      "step": 56504
    },
    {
      "epoch": 5.388641357421875e-05,
      "step": 56504,
      "training_step_time": 6.647767782211304
    },
    {
      "epoch": 5.388736724853516e-05,
      "model_forward_time": 1.7603492736816406,
      "step": 56505
    },
    {
      "epoch": 5.388736724853516e-05,
      "step": 56505,
      "training_step_time": 7.548644542694092
    },
    {
      "epoch": 5.388832092285156e-05,
      "model_forward_time": 1.7573821544647217,
      "step": 56506
    },
    {
      "epoch": 5.388832092285156e-05,
      "step": 56506,
      "training_step_time": 7.614701986312866
    },
    {
      "epoch": 5.388927459716797e-05,
      "model_forward_time": 2.890194892883301,
      "step": 56507
    },
    {
      "epoch": 5.388927459716797e-05,
      "step": 56507,
      "training_step_time": 10.176539897918701
    },
    {
      "epoch": 5.389022827148438e-05,
      "model_forward_time": 1.763218641281128,
      "step": 56508
    },
    {
      "epoch": 5.389022827148438e-05,
      "step": 56508,
      "training_step_time": 9.368224143981934
    },
    {
      "epoch": 5.389118194580078e-05,
      "model_forward_time": 1.759080171585083,
      "step": 56509
    },
    {
      "epoch": 5.389118194580078e-05,
      "step": 56509,
      "training_step_time": 7.487554311752319
    },
    {
      "epoch": 5.3892135620117186e-05,
      "grad_norm": 0.10334867984056473,
      "learning_rate": 4.339064049821098e-06,
      "loss": 0.0598,
      "step": 56510
    },
    {
      "epoch": 5.3892135620117186e-05,
      "model_forward_time": 1.7567341327667236,
      "step": 56510
    },
    {
      "dynamics_loss_avg": 0.04763017725199461,
      "epoch": 5.3892135620117186e-05,
      "step": 56510
    },
    {
      "action_loss_avg": 0.0009641315991757438,
      "epoch": 5.3892135620117186e-05,
      "step": 56510
    },
    {
      "epoch": 5.3892135620117186e-05,
      "step": 56510,
      "training_step_time": 7.267448663711548
    },
    {
      "epoch": 5.3893089294433596e-05,
      "model_forward_time": 1.7777643203735352,
      "step": 56511
    },
    {
      "epoch": 5.3893089294433596e-05,
      "step": 56511,
      "training_step_time": 6.727599620819092
    },
    {
      "epoch": 5.389404296875e-05,
      "model_forward_time": 1.7548644542694092,
      "step": 56512
    },
    {
      "epoch": 5.389404296875e-05,
      "step": 56512,
      "training_step_time": 6.680101633071899
    },
    {
      "epoch": 5.3894996643066404e-05,
      "model_forward_time": 1.7525990009307861,
      "step": 56513
    },
    {
      "epoch": 5.3894996643066404e-05,
      "step": 56513,
      "training_step_time": 6.676804304122925
    },
    {
      "epoch": 5.3895950317382815e-05,
      "model_forward_time": 1.75685453414917,
      "step": 56514
    },
    {
      "epoch": 5.3895950317382815e-05,
      "step": 56514,
      "training_step_time": 7.552760601043701
    },
    {
      "epoch": 5.389690399169922e-05,
      "model_forward_time": 1.759467601776123,
      "step": 56515
    },
    {
      "epoch": 5.389690399169922e-05,
      "step": 56515,
      "training_step_time": 7.671432256698608
    },
    {
      "epoch": 5.389785766601562e-05,
      "model_forward_time": 2.875936269760132,
      "step": 56516
    },
    {
      "epoch": 5.389785766601562e-05,
      "step": 56516,
      "training_step_time": 9.518150091171265
    },
    {
      "epoch": 5.3898811340332034e-05,
      "model_forward_time": 1.7656688690185547,
      "step": 56517
    },
    {
      "epoch": 5.3898811340332034e-05,
      "step": 56517,
      "training_step_time": 9.504561185836792
    },
    {
      "epoch": 5.389976501464844e-05,
      "model_forward_time": 1.7491681575775146,
      "step": 56518
    },
    {
      "epoch": 5.389976501464844e-05,
      "step": 56518,
      "training_step_time": 7.715419769287109
    },
    {
      "epoch": 5.390071868896484e-05,
      "model_forward_time": 1.7565619945526123,
      "step": 56519
    },
    {
      "epoch": 5.390071868896484e-05,
      "step": 56519,
      "training_step_time": 7.315932512283325
    },
    {
      "epoch": 5.390167236328125e-05,
      "grad_norm": 0.10386236011981964,
      "learning_rate": 4.3374251257489446e-06,
      "loss": 0.0565,
      "step": 56520
    },
    {
      "epoch": 5.390167236328125e-05,
      "model_forward_time": 1.754537582397461,
      "step": 56520
    },
    {
      "dynamics_loss_avg": 0.06454994305968284,
      "epoch": 5.390167236328125e-05,
      "step": 56520
    },
    {
      "action_loss_avg": 0.0016137601633090526,
      "epoch": 5.390167236328125e-05,
      "step": 56520
    },
    {
      "epoch": 5.390167236328125e-05,
      "step": 56520,
      "training_step_time": 6.719557762145996
    },
    {
      "epoch": 5.390262603759766e-05,
      "model_forward_time": 1.7588791847229004,
      "step": 56521
    },
    {
      "epoch": 5.390262603759766e-05,
      "step": 56521,
      "training_step_time": 6.661970376968384
    },
    {
      "epoch": 5.390357971191406e-05,
      "model_forward_time": 1.7507531642913818,
      "step": 56522
    },
    {
      "epoch": 5.390357971191406e-05,
      "step": 56522,
      "training_step_time": 6.695972919464111
    },
    {
      "epoch": 5.390453338623047e-05,
      "model_forward_time": 1.7556607723236084,
      "step": 56523
    },
    {
      "epoch": 5.390453338623047e-05,
      "step": 56523,
      "training_step_time": 7.590426206588745
    },
    {
      "epoch": 5.3905487060546876e-05,
      "model_forward_time": 1.758695363998413,
      "step": 56524
    },
    {
      "epoch": 5.3905487060546876e-05,
      "step": 56524,
      "training_step_time": 7.588399171829224
    },
    {
      "epoch": 5.390644073486328e-05,
      "model_forward_time": 2.8754026889801025,
      "step": 56525
    },
    {
      "epoch": 5.390644073486328e-05,
      "step": 56525,
      "training_step_time": 7.69365382194519
    },
    {
      "epoch": 5.390739440917969e-05,
      "model_forward_time": 1.748650074005127,
      "step": 56526
    },
    {
      "epoch": 5.390739440917969e-05,
      "step": 56526,
      "training_step_time": 9.452016830444336
    },
    {
      "epoch": 5.3908348083496095e-05,
      "model_forward_time": 1.752389669418335,
      "step": 56527
    },
    {
      "epoch": 5.3908348083496095e-05,
      "step": 56527,
      "training_step_time": 8.473603963851929
    },
    {
      "epoch": 5.39093017578125e-05,
      "model_forward_time": 1.7560005187988281,
      "step": 56528
    },
    {
      "epoch": 5.39093017578125e-05,
      "step": 56528,
      "training_step_time": 7.300729990005493
    },
    {
      "epoch": 5.391025543212891e-05,
      "model_forward_time": 1.7569301128387451,
      "step": 56529
    },
    {
      "epoch": 5.391025543212891e-05,
      "step": 56529,
      "training_step_time": 6.72865629196167
    },
    {
      "epoch": 5.3911209106445314e-05,
      "grad_norm": 0.09116580337285995,
      "learning_rate": 4.335786274134984e-06,
      "loss": 0.0563,
      "step": 56530
    },
    {
      "epoch": 5.3911209106445314e-05,
      "model_forward_time": 1.7579638957977295,
      "step": 56530
    },
    {
      "dynamics_loss_avg": 0.045773614570498465,
      "epoch": 5.3911209106445314e-05,
      "step": 56530
    },
    {
      "action_loss_avg": 0.0014745892549399287,
      "epoch": 5.3911209106445314e-05,
      "step": 56530
    },
    {
      "epoch": 5.3911209106445314e-05,
      "step": 56530,
      "training_step_time": 6.711091995239258
    },
    {
      "epoch": 5.391216278076172e-05,
      "model_forward_time": 1.7555007934570312,
      "step": 56531
    },
    {
      "epoch": 5.391216278076172e-05,
      "step": 56531,
      "training_step_time": 6.68790602684021
    },
    {
      "epoch": 5.391311645507812e-05,
      "model_forward_time": 1.7606582641601562,
      "step": 56532
    },
    {
      "epoch": 5.391311645507812e-05,
      "step": 56532,
      "training_step_time": 7.596372365951538
    },
    {
      "epoch": 5.391407012939453e-05,
      "model_forward_time": 1.7517220973968506,
      "step": 56533
    },
    {
      "epoch": 5.391407012939453e-05,
      "step": 56533,
      "training_step_time": 7.55508828163147
    },
    {
      "epoch": 5.3915023803710936e-05,
      "model_forward_time": 2.891396999359131,
      "step": 56534
    },
    {
      "epoch": 5.3915023803710936e-05,
      "step": 56534,
      "training_step_time": 9.620386123657227
    },
    {
      "epoch": 5.391597747802734e-05,
      "model_forward_time": 1.7532000541687012,
      "step": 56535
    },
    {
      "epoch": 5.391597747802734e-05,
      "step": 56535,
      "training_step_time": 10.658118963241577
    },
    {
      "epoch": 5.391693115234375e-05,
      "model_forward_time": 1.7543120384216309,
      "step": 56536
    },
    {
      "epoch": 5.391693115234375e-05,
      "step": 56536,
      "training_step_time": 8.409785985946655
    },
    {
      "epoch": 5.3917884826660155e-05,
      "model_forward_time": 1.7389230728149414,
      "step": 56537
    },
    {
      "epoch": 5.3917884826660155e-05,
      "step": 56537,
      "training_step_time": 7.385656833648682
    },
    {
      "epoch": 5.391883850097656e-05,
      "model_forward_time": 1.7379176616668701,
      "step": 56538
    },
    {
      "epoch": 5.391883850097656e-05,
      "step": 56538,
      "training_step_time": 7.320513486862183
    },
    {
      "epoch": 5.391979217529297e-05,
      "model_forward_time": 1.7412960529327393,
      "step": 56539
    },
    {
      "epoch": 5.391979217529297e-05,
      "step": 56539,
      "training_step_time": 6.6730263233184814
    },
    {
      "epoch": 5.3920745849609374e-05,
      "grad_norm": 0.08694598078727722,
      "learning_rate": 4.334147495158438e-06,
      "loss": 0.059,
      "step": 56540
    },
    {
      "epoch": 5.3920745849609374e-05,
      "model_forward_time": 1.745558738708496,
      "step": 56540
    },
    {
      "dynamics_loss_avg": 0.06218988113105297,
      "epoch": 5.3920745849609374e-05,
      "step": 56540
    },
    {
      "action_loss_avg": 0.002040293259778991,
      "epoch": 5.3920745849609374e-05,
      "step": 56540
    },
    {
      "epoch": 5.3920745849609374e-05,
      "step": 56540,
      "training_step_time": 6.696345806121826
    },
    {
      "epoch": 5.392169952392578e-05,
      "model_forward_time": 1.7408745288848877,
      "step": 56541
    },
    {
      "epoch": 5.392169952392578e-05,
      "step": 56541,
      "training_step_time": 8.61007046699524
    },
    {
      "epoch": 5.392265319824219e-05,
      "model_forward_time": 1.7445919513702393,
      "step": 56542
    },
    {
      "epoch": 5.392265319824219e-05,
      "step": 56542,
      "training_step_time": 7.583052396774292
    },
    {
      "epoch": 5.392360687255859e-05,
      "model_forward_time": 2.8750104904174805,
      "step": 56543
    },
    {
      "epoch": 5.392360687255859e-05,
      "step": 56543,
      "training_step_time": 8.195842981338501
    },
    {
      "epoch": 5.3924560546875e-05,
      "model_forward_time": 1.7419648170471191,
      "step": 56544
    },
    {
      "epoch": 5.3924560546875e-05,
      "step": 56544,
      "training_step_time": 8.491685152053833
    },
    {
      "epoch": 5.392551422119141e-05,
      "model_forward_time": 1.7340004444122314,
      "step": 56545
    },
    {
      "epoch": 5.392551422119141e-05,
      "step": 56545,
      "training_step_time": 7.580802917480469
    },
    {
      "epoch": 5.392646789550781e-05,
      "model_forward_time": 1.7441112995147705,
      "step": 56546
    },
    {
      "epoch": 5.392646789550781e-05,
      "step": 56546,
      "training_step_time": 8.166444778442383
    },
    {
      "epoch": 5.3927421569824216e-05,
      "model_forward_time": 1.7761621475219727,
      "step": 56547
    },
    {
      "epoch": 5.3927421569824216e-05,
      "step": 56547,
      "training_step_time": 7.335738897323608
    },
    {
      "epoch": 5.392837524414063e-05,
      "model_forward_time": 1.75022554397583,
      "step": 56548
    },
    {
      "epoch": 5.392837524414063e-05,
      "step": 56548,
      "training_step_time": 6.726955890655518
    },
    {
      "epoch": 5.392932891845703e-05,
      "model_forward_time": 1.7485442161560059,
      "step": 56549
    },
    {
      "epoch": 5.392932891845703e-05,
      "step": 56549,
      "training_step_time": 6.754356861114502
    },
    {
      "epoch": 5.3930282592773435e-05,
      "grad_norm": 0.10162502527236938,
      "learning_rate": 4.33250878899852e-06,
      "loss": 0.057,
      "step": 56550
    },
    {
      "epoch": 5.3930282592773435e-05,
      "model_forward_time": 1.7494122982025146,
      "step": 56550
    },
    {
      "dynamics_loss_avg": 0.0595499511808157,
      "epoch": 5.3930282592773435e-05,
      "step": 56550
    },
    {
      "action_loss_avg": 0.0013525884976843372,
      "epoch": 5.3930282592773435e-05,
      "step": 56550
    },
    {
      "epoch": 5.3930282592773435e-05,
      "step": 56550,
      "training_step_time": 7.586463451385498
    },
    {
      "epoch": 5.3931236267089846e-05,
      "model_forward_time": 1.752328872680664,
      "step": 56551
    },
    {
      "epoch": 5.3931236267089846e-05,
      "step": 56551,
      "training_step_time": 8.356331586837769
    },
    {
      "epoch": 5.393218994140625e-05,
      "model_forward_time": 1.7453365325927734,
      "step": 56552
    },
    {
      "epoch": 5.393218994140625e-05,
      "step": 56552,
      "training_step_time": 9.322581052780151
    },
    {
      "epoch": 5.3933143615722654e-05,
      "model_forward_time": 1.8605215549468994,
      "step": 56553
    },
    {
      "epoch": 5.3933143615722654e-05,
      "step": 56553,
      "training_step_time": 8.511047840118408
    },
    {
      "epoch": 5.3934097290039064e-05,
      "model_forward_time": 1.747222661972046,
      "step": 56554
    },
    {
      "epoch": 5.3934097290039064e-05,
      "step": 56554,
      "training_step_time": 7.636324405670166
    },
    {
      "epoch": 5.393505096435547e-05,
      "model_forward_time": 1.7536756992340088,
      "step": 56555
    },
    {
      "epoch": 5.393505096435547e-05,
      "step": 56555,
      "training_step_time": 7.593393325805664
    },
    {
      "epoch": 5.393600463867187e-05,
      "model_forward_time": 1.744588851928711,
      "step": 56556
    },
    {
      "epoch": 5.393600463867187e-05,
      "step": 56556,
      "training_step_time": 7.724831819534302
    },
    {
      "epoch": 5.393695831298828e-05,
      "model_forward_time": 1.753713846206665,
      "step": 56557
    },
    {
      "epoch": 5.393695831298828e-05,
      "step": 56557,
      "training_step_time": 6.820440769195557
    },
    {
      "epoch": 5.393791198730469e-05,
      "model_forward_time": 1.7562298774719238,
      "step": 56558
    },
    {
      "epoch": 5.393791198730469e-05,
      "step": 56558,
      "training_step_time": 6.822183847427368
    },
    {
      "epoch": 5.393886566162109e-05,
      "model_forward_time": 1.7470097541809082,
      "step": 56559
    },
    {
      "epoch": 5.393886566162109e-05,
      "step": 56559,
      "training_step_time": 7.633828639984131
    },
    {
      "epoch": 5.39398193359375e-05,
      "grad_norm": 0.11295910179615021,
      "learning_rate": 4.3308701558344365e-06,
      "loss": 0.0602,
      "step": 56560
    },
    {
      "epoch": 5.39398193359375e-05,
      "model_forward_time": 1.7476212978363037,
      "step": 56560
    },
    {
      "dynamics_loss_avg": 0.0406718235462904,
      "epoch": 5.39398193359375e-05,
      "step": 56560
    },
    {
      "action_loss_avg": 0.0011393773223971949,
      "epoch": 5.39398193359375e-05,
      "step": 56560
    },
    {
      "epoch": 5.39398193359375e-05,
      "step": 56560,
      "training_step_time": 8.116682767868042
    },
    {
      "epoch": 5.3940773010253906e-05,
      "model_forward_time": 1.7473959922790527,
      "step": 56561
    },
    {
      "epoch": 5.3940773010253906e-05,
      "step": 56561,
      "training_step_time": 7.660434246063232
    },
    {
      "epoch": 5.394172668457031e-05,
      "model_forward_time": 2.961460590362549,
      "step": 56562
    },
    {
      "epoch": 5.394172668457031e-05,
      "step": 56562,
      "training_step_time": 9.663642168045044
    },
    {
      "epoch": 5.394268035888672e-05,
      "model_forward_time": 1.7345633506774902,
      "step": 56563
    },
    {
      "epoch": 5.394268035888672e-05,
      "step": 56563,
      "training_step_time": 9.991297960281372
    },
    {
      "epoch": 5.3943634033203125e-05,
      "model_forward_time": 1.7598919868469238,
      "step": 56564
    },
    {
      "epoch": 5.3943634033203125e-05,
      "step": 56564,
      "training_step_time": 7.802509784698486
    },
    {
      "epoch": 5.394458770751953e-05,
      "model_forward_time": 1.7475659847259521,
      "step": 56565
    },
    {
      "epoch": 5.394458770751953e-05,
      "step": 56565,
      "training_step_time": 7.738092422485352
    },
    {
      "epoch": 5.394554138183594e-05,
      "model_forward_time": 1.7531254291534424,
      "step": 56566
    },
    {
      "epoch": 5.394554138183594e-05,
      "step": 56566,
      "training_step_time": 7.428691148757935
    },
    {
      "epoch": 5.3946495056152344e-05,
      "model_forward_time": 1.7486696243286133,
      "step": 56567
    },
    {
      "epoch": 5.3946495056152344e-05,
      "step": 56567,
      "training_step_time": 6.782198667526245
    },
    {
      "epoch": 5.394744873046875e-05,
      "model_forward_time": 1.7450835704803467,
      "step": 56568
    },
    {
      "epoch": 5.394744873046875e-05,
      "step": 56568,
      "training_step_time": 7.631108283996582
    },
    {
      "epoch": 5.394840240478516e-05,
      "model_forward_time": 1.7545862197875977,
      "step": 56569
    },
    {
      "epoch": 5.394840240478516e-05,
      "step": 56569,
      "training_step_time": 7.8283915519714355
    },
    {
      "epoch": 5.394935607910156e-05,
      "grad_norm": 0.08870957791805267,
      "learning_rate": 4.329231595845388e-06,
      "loss": 0.0558,
      "step": 56570
    },
    {
      "epoch": 5.394935607910156e-05,
      "model_forward_time": 1.7326724529266357,
      "step": 56570
    },
    {
      "dynamics_loss_avg": 0.06085351724177599,
      "epoch": 5.394935607910156e-05,
      "step": 56570
    },
    {
      "action_loss_avg": 0.0020715448394184933,
      "epoch": 5.394935607910156e-05,
      "step": 56570
    },
    {
      "epoch": 5.394935607910156e-05,
      "step": 56570,
      "training_step_time": 7.685891151428223
    },
    {
      "epoch": 5.395030975341797e-05,
      "model_forward_time": 2.9960036277770996,
      "step": 56571
    },
    {
      "epoch": 5.395030975341797e-05,
      "step": 56571,
      "training_step_time": 7.786420583724976
    },
    {
      "epoch": 5.395126342773438e-05,
      "model_forward_time": 1.7469677925109863,
      "step": 56572
    },
    {
      "epoch": 5.395126342773438e-05,
      "step": 56572,
      "training_step_time": 9.714282512664795
    },
    {
      "epoch": 5.395221710205078e-05,
      "model_forward_time": 1.741633415222168,
      "step": 56573
    },
    {
      "epoch": 5.395221710205078e-05,
      "step": 56573,
      "training_step_time": 8.138976573944092
    },
    {
      "epoch": 5.3953170776367186e-05,
      "model_forward_time": 1.7462458610534668,
      "step": 56574
    },
    {
      "epoch": 5.3953170776367186e-05,
      "step": 56574,
      "training_step_time": 7.6338889598846436
    },
    {
      "epoch": 5.3954124450683596e-05,
      "model_forward_time": 1.747910499572754,
      "step": 56575
    },
    {
      "epoch": 5.3954124450683596e-05,
      "step": 56575,
      "training_step_time": 7.325782299041748
    },
    {
      "epoch": 5.3955078125e-05,
      "model_forward_time": 1.7402679920196533,
      "step": 56576
    },
    {
      "epoch": 5.3955078125e-05,
      "step": 56576,
      "training_step_time": 6.718186616897583
    },
    {
      "epoch": 5.3956031799316405e-05,
      "model_forward_time": 1.7414462566375732,
      "step": 56577
    },
    {
      "epoch": 5.3956031799316405e-05,
      "step": 56577,
      "training_step_time": 7.568763256072998
    },
    {
      "epoch": 5.3956985473632815e-05,
      "model_forward_time": 1.7541954517364502,
      "step": 56578
    },
    {
      "epoch": 5.3956985473632815e-05,
      "step": 56578,
      "training_step_time": 7.546105623245239
    },
    {
      "epoch": 5.395793914794922e-05,
      "model_forward_time": 1.7486035823822021,
      "step": 56579
    },
    {
      "epoch": 5.395793914794922e-05,
      "step": 56579,
      "training_step_time": 7.829381942749023
    },
    {
      "epoch": 5.395889282226562e-05,
      "grad_norm": 0.09276290982961655,
      "learning_rate": 4.327593109210564e-06,
      "loss": 0.0583,
      "step": 56580
    },
    {
      "epoch": 5.395889282226562e-05,
      "model_forward_time": 2.978961229324341,
      "step": 56580
    },
    {
      "dynamics_loss_avg": 0.06876913644373417,
      "epoch": 5.395889282226562e-05,
      "step": 56580
    },
    {
      "action_loss_avg": 0.0016319000074872746,
      "epoch": 5.395889282226562e-05,
      "step": 56580
    },
    {
      "epoch": 5.395889282226562e-05,
      "step": 56580,
      "training_step_time": 9.362636804580688
    },
    {
      "epoch": 5.3959846496582034e-05,
      "model_forward_time": 1.7446494102478027,
      "step": 56581
    },
    {
      "epoch": 5.3959846496582034e-05,
      "step": 56581,
      "training_step_time": 8.344182968139648
    },
    {
      "epoch": 5.396080017089844e-05,
      "model_forward_time": 1.7468397617340088,
      "step": 56582
    },
    {
      "epoch": 5.396080017089844e-05,
      "step": 56582,
      "training_step_time": 8.175652980804443
    },
    {
      "epoch": 5.396175384521484e-05,
      "model_forward_time": 1.804821491241455,
      "step": 56583
    },
    {
      "epoch": 5.396175384521484e-05,
      "step": 56583,
      "training_step_time": 7.695075511932373
    },
    {
      "epoch": 5.396270751953125e-05,
      "model_forward_time": 1.7438685894012451,
      "step": 56584
    },
    {
      "epoch": 5.396270751953125e-05,
      "step": 56584,
      "training_step_time": 7.329096555709839
    },
    {
      "epoch": 5.396366119384766e-05,
      "model_forward_time": 1.749159574508667,
      "step": 56585
    },
    {
      "epoch": 5.396366119384766e-05,
      "step": 56585,
      "training_step_time": 6.739363431930542
    },
    {
      "epoch": 5.396461486816406e-05,
      "model_forward_time": 1.7569584846496582,
      "step": 56586
    },
    {
      "epoch": 5.396461486816406e-05,
      "step": 56586,
      "training_step_time": 7.60658073425293
    },
    {
      "epoch": 5.396556854248047e-05,
      "model_forward_time": 1.7573025226593018,
      "step": 56587
    },
    {
      "epoch": 5.396556854248047e-05,
      "step": 56587,
      "training_step_time": 7.592512130737305
    },
    {
      "epoch": 5.3966522216796876e-05,
      "model_forward_time": 1.7495074272155762,
      "step": 56588
    },
    {
      "epoch": 5.3966522216796876e-05,
      "step": 56588,
      "training_step_time": 7.8111913204193115
    },
    {
      "epoch": 5.396747589111328e-05,
      "model_forward_time": 2.9078590869903564,
      "step": 56589
    },
    {
      "epoch": 5.396747589111328e-05,
      "step": 56589,
      "training_step_time": 7.671214818954468
    },
    {
      "epoch": 5.396842956542969e-05,
      "grad_norm": 0.094749316573143,
      "learning_rate": 4.325954696109147e-06,
      "loss": 0.0583,
      "step": 56590
    },
    {
      "epoch": 5.396842956542969e-05,
      "model_forward_time": 1.8631784915924072,
      "step": 56590
    },
    {
      "dynamics_loss_avg": 0.06114075351506472,
      "epoch": 5.396842956542969e-05,
      "step": 56590
    },
    {
      "action_loss_avg": 0.0012715657067019492,
      "epoch": 5.396842956542969e-05,
      "step": 56590
    },
    {
      "epoch": 5.396842956542969e-05,
      "step": 56590,
      "training_step_time": 8.436832189559937
    },
    {
      "epoch": 5.3969383239746095e-05,
      "model_forward_time": 1.7508459091186523,
      "step": 56591
    },
    {
      "epoch": 5.3969383239746095e-05,
      "step": 56591,
      "training_step_time": 8.530508518218994
    },
    {
      "epoch": 5.39703369140625e-05,
      "model_forward_time": 1.7473952770233154,
      "step": 56592
    },
    {
      "epoch": 5.39703369140625e-05,
      "step": 56592,
      "training_step_time": 8.179953813552856
    },
    {
      "epoch": 5.397129058837891e-05,
      "model_forward_time": 1.747086763381958,
      "step": 56593
    },
    {
      "epoch": 5.397129058837891e-05,
      "step": 56593,
      "training_step_time": 7.328258275985718
    },
    {
      "epoch": 5.3972244262695314e-05,
      "model_forward_time": 1.746880292892456,
      "step": 56594
    },
    {
      "epoch": 5.3972244262695314e-05,
      "step": 56594,
      "training_step_time": 6.7514660358428955
    },
    {
      "epoch": 5.397319793701172e-05,
      "model_forward_time": 1.7547924518585205,
      "step": 56595
    },
    {
      "epoch": 5.397319793701172e-05,
      "step": 56595,
      "training_step_time": 7.659131288528442
    },
    {
      "epoch": 5.397415161132812e-05,
      "model_forward_time": 1.7442314624786377,
      "step": 56596
    },
    {
      "epoch": 5.397415161132812e-05,
      "step": 56596,
      "training_step_time": 7.6192686557769775
    },
    {
      "epoch": 5.397510528564453e-05,
      "model_forward_time": 1.7444944381713867,
      "step": 56597
    },
    {
      "epoch": 5.397510528564453e-05,
      "step": 56597,
      "training_step_time": 8.032910346984863
    },
    {
      "epoch": 5.3976058959960937e-05,
      "model_forward_time": 2.99543833732605,
      "step": 56598
    },
    {
      "epoch": 5.3976058959960937e-05,
      "step": 56598,
      "training_step_time": 9.15605878829956
    },
    {
      "epoch": 5.397701263427734e-05,
      "model_forward_time": 1.7996101379394531,
      "step": 56599
    },
    {
      "epoch": 5.397701263427734e-05,
      "step": 56599,
      "training_step_time": 8.734226942062378
    },
    {
      "epoch": 5.397796630859375e-05,
      "grad_norm": 0.13209401071071625,
      "learning_rate": 4.324316356720311e-06,
      "loss": 0.0562,
      "step": 56600
    },
    {
      "epoch": 5.397796630859375e-05,
      "model_forward_time": 1.7308382987976074,
      "step": 56600
    },
    {
      "dynamics_loss_avg": 0.05895968973636627,
      "epoch": 5.397796630859375e-05,
      "step": 56600
    },
    {
      "action_loss_avg": 0.0011933242611121386,
      "epoch": 5.397796630859375e-05,
      "step": 56600
    },
    {
      "epoch": 5.397796630859375e-05,
      "step": 56600,
      "training_step_time": 9.531352758407593
    },
    {
      "epoch": 5.3978919982910155e-05,
      "model_forward_time": 1.744922399520874,
      "step": 56601
    },
    {
      "epoch": 5.3978919982910155e-05,
      "step": 56601,
      "training_step_time": 7.597574710845947
    },
    {
      "epoch": 5.397987365722656e-05,
      "model_forward_time": 1.7456507682800293,
      "step": 56602
    },
    {
      "epoch": 5.397987365722656e-05,
      "step": 56602,
      "training_step_time": 7.57685661315918
    },
    {
      "epoch": 5.398082733154297e-05,
      "model_forward_time": 1.7508418560028076,
      "step": 56603
    },
    {
      "epoch": 5.398082733154297e-05,
      "step": 56603,
      "training_step_time": 6.7253944873809814
    },
    {
      "epoch": 5.3981781005859374e-05,
      "model_forward_time": 1.7510199546813965,
      "step": 56604
    },
    {
      "epoch": 5.3981781005859374e-05,
      "step": 56604,
      "training_step_time": 7.59222674369812
    },
    {
      "epoch": 5.398273468017578e-05,
      "model_forward_time": 1.749605655670166,
      "step": 56605
    },
    {
      "epoch": 5.398273468017578e-05,
      "step": 56605,
      "training_step_time": 7.569228887557983
    },
    {
      "epoch": 5.398368835449219e-05,
      "model_forward_time": 1.7438147068023682,
      "step": 56606
    },
    {
      "epoch": 5.398368835449219e-05,
      "step": 56606,
      "training_step_time": 9.323948621749878
    },
    {
      "epoch": 5.398464202880859e-05,
      "model_forward_time": 2.8787200450897217,
      "step": 56607
    },
    {
      "epoch": 5.398464202880859e-05,
      "step": 56607,
      "training_step_time": 8.345502376556396
    },
    {
      "epoch": 5.3985595703125e-05,
      "model_forward_time": 1.7763159275054932,
      "step": 56608
    },
    {
      "epoch": 5.3985595703125e-05,
      "step": 56608,
      "training_step_time": 8.326245546340942
    },
    {
      "epoch": 5.398654937744141e-05,
      "model_forward_time": 1.7603182792663574,
      "step": 56609
    },
    {
      "epoch": 5.398654937744141e-05,
      "step": 56609,
      "training_step_time": 8.999328136444092
    },
    {
      "epoch": 5.398750305175781e-05,
      "grad_norm": 0.08662160485982895,
      "learning_rate": 4.3226780912232215e-06,
      "loss": 0.0584,
      "step": 56610
    },
    {
      "epoch": 5.398750305175781e-05,
      "model_forward_time": 1.7659010887145996,
      "step": 56610
    },
    {
      "dynamics_loss_avg": 0.061673957854509354,
      "epoch": 5.398750305175781e-05,
      "step": 56610
    },
    {
      "action_loss_avg": 0.0013447977253235877,
      "epoch": 5.398750305175781e-05,
      "step": 56610
    },
    {
      "epoch": 5.398750305175781e-05,
      "step": 56610,
      "training_step_time": 7.5430991649627686
    },
    {
      "epoch": 5.3988456726074216e-05,
      "model_forward_time": 1.7213280200958252,
      "step": 56611
    },
    {
      "epoch": 5.3988456726074216e-05,
      "step": 56611,
      "training_step_time": 7.58234977722168
    },
    {
      "epoch": 5.398941040039063e-05,
      "model_forward_time": 1.74733567237854,
      "step": 56612
    },
    {
      "epoch": 5.398941040039063e-05,
      "step": 56612,
      "training_step_time": 7.256139516830444
    },
    {
      "epoch": 5.399036407470703e-05,
      "model_forward_time": 1.3804728984832764,
      "step": 56613
    },
    {
      "epoch": 5.399036407470703e-05,
      "step": 56613,
      "training_step_time": 7.51288628578186
    },
    {
      "epoch": 5.3991317749023435e-05,
      "model_forward_time": 1.7378385066986084,
      "step": 56614
    },
    {
      "epoch": 5.3991317749023435e-05,
      "step": 56614,
      "training_step_time": 7.559025764465332
    },
    {
      "epoch": 5.3992271423339846e-05,
      "model_forward_time": 1.7457890510559082,
      "step": 56615
    },
    {
      "epoch": 5.3992271423339846e-05,
      "step": 56615,
      "training_step_time": 7.397535562515259
    },
    {
      "epoch": 5.399322509765625e-05,
      "model_forward_time": 2.91867733001709,
      "step": 56616
    },
    {
      "epoch": 5.399322509765625e-05,
      "step": 56616,
      "training_step_time": 8.281584024429321
    },
    {
      "epoch": 5.3994178771972654e-05,
      "model_forward_time": 1.7384819984436035,
      "step": 56617
    },
    {
      "epoch": 5.3994178771972654e-05,
      "step": 56617,
      "training_step_time": 8.446815490722656
    },
    {
      "epoch": 5.3995132446289065e-05,
      "model_forward_time": 1.8411412239074707,
      "step": 56618
    },
    {
      "epoch": 5.3995132446289065e-05,
      "step": 56618,
      "training_step_time": 8.735095024108887
    },
    {
      "epoch": 5.399608612060547e-05,
      "model_forward_time": 1.7397117614746094,
      "step": 56619
    },
    {
      "epoch": 5.399608612060547e-05,
      "step": 56619,
      "training_step_time": 8.256336688995361
    },
    {
      "epoch": 5.399703979492187e-05,
      "grad_norm": 0.07067230343818665,
      "learning_rate": 4.321039899797036e-06,
      "loss": 0.0597,
      "step": 56620
    },
    {
      "epoch": 5.399703979492187e-05,
      "model_forward_time": 1.7484889030456543,
      "step": 56620
    },
    {
      "dynamics_loss_avg": 0.04380342587828636,
      "epoch": 5.399703979492187e-05,
      "step": 56620
    },
    {
      "action_loss_avg": 0.0013000024831853808,
      "epoch": 5.399703979492187e-05,
      "step": 56620
    },
    {
      "epoch": 5.399703979492187e-05,
      "step": 56620,
      "training_step_time": 7.713157892227173
    },
    {
      "epoch": 5.399799346923828e-05,
      "model_forward_time": 1.7419099807739258,
      "step": 56621
    },
    {
      "epoch": 5.399799346923828e-05,
      "step": 56621,
      "training_step_time": 7.308876276016235
    },
    {
      "epoch": 5.399894714355469e-05,
      "model_forward_time": 1.74540376663208,
      "step": 56622
    },
    {
      "epoch": 5.399894714355469e-05,
      "step": 56622,
      "training_step_time": 7.595111131668091
    },
    {
      "epoch": 5.399990081787109e-05,
      "model_forward_time": 1.7513835430145264,
      "step": 56623
    },
    {
      "epoch": 5.399990081787109e-05,
      "step": 56623,
      "training_step_time": 7.671973943710327
    },
    {
      "epoch": 5.40008544921875e-05,
      "model_forward_time": 1.7533702850341797,
      "step": 56624
    },
    {
      "epoch": 5.40008544921875e-05,
      "step": 56624,
      "training_step_time": 7.553103685379028
    },
    {
      "epoch": 5.4001808166503906e-05,
      "model_forward_time": 2.9371395111083984,
      "step": 56625
    },
    {
      "epoch": 5.4001808166503906e-05,
      "step": 56625,
      "training_step_time": 8.497012853622437
    },
    {
      "epoch": 5.400276184082031e-05,
      "model_forward_time": 1.7440009117126465,
      "step": 56626
    },
    {
      "epoch": 5.400276184082031e-05,
      "step": 56626,
      "training_step_time": 9.412813901901245
    },
    {
      "epoch": 5.400371551513672e-05,
      "model_forward_time": 1.8396542072296143,
      "step": 56627
    },
    {
      "epoch": 5.400371551513672e-05,
      "step": 56627,
      "training_step_time": 8.66729998588562
    },
    {
      "epoch": 5.4004669189453125e-05,
      "model_forward_time": 1.7433617115020752,
      "step": 56628
    },
    {
      "epoch": 5.4004669189453125e-05,
      "step": 56628,
      "training_step_time": 8.239295721054077
    },
    {
      "epoch": 5.400562286376953e-05,
      "model_forward_time": 1.7128169536590576,
      "step": 56629
    },
    {
      "epoch": 5.400562286376953e-05,
      "step": 56629,
      "training_step_time": 7.682469606399536
    },
    {
      "epoch": 5.400657653808594e-05,
      "grad_norm": 0.0914803147315979,
      "learning_rate": 4.319401782620909e-06,
      "loss": 0.059,
      "step": 56630
    },
    {
      "epoch": 5.400657653808594e-05,
      "model_forward_time": 1.7467706203460693,
      "step": 56630
    },
    {
      "dynamics_loss_avg": 0.05720955319702625,
      "epoch": 5.400657653808594e-05,
      "step": 56630
    },
    {
      "action_loss_avg": 0.002108164952369407,
      "epoch": 5.400657653808594e-05,
      "step": 56630
    },
    {
      "epoch": 5.400657653808594e-05,
      "step": 56630,
      "training_step_time": 7.2775352001190186
    },
    {
      "epoch": 5.4007530212402344e-05,
      "model_forward_time": 1.7582671642303467,
      "step": 56631
    },
    {
      "epoch": 5.4007530212402344e-05,
      "step": 56631,
      "training_step_time": 7.620838403701782
    },
    {
      "epoch": 5.400848388671875e-05,
      "model_forward_time": 1.759192943572998,
      "step": 56632
    },
    {
      "epoch": 5.400848388671875e-05,
      "step": 56632,
      "training_step_time": 7.657862186431885
    },
    {
      "epoch": 5.400943756103516e-05,
      "model_forward_time": 1.749202013015747,
      "step": 56633
    },
    {
      "epoch": 5.400943756103516e-05,
      "step": 56633,
      "training_step_time": 7.4217445850372314
    },
    {
      "epoch": 5.401039123535156e-05,
      "model_forward_time": 2.9205915927886963,
      "step": 56634
    },
    {
      "epoch": 5.401039123535156e-05,
      "step": 56634,
      "training_step_time": 8.373877763748169
    },
    {
      "epoch": 5.401134490966797e-05,
      "model_forward_time": 1.7645552158355713,
      "step": 56635
    },
    {
      "epoch": 5.401134490966797e-05,
      "step": 56635,
      "training_step_time": 8.215829849243164
    },
    {
      "epoch": 5.401229858398438e-05,
      "model_forward_time": 1.8002431392669678,
      "step": 56636
    },
    {
      "epoch": 5.401229858398438e-05,
      "step": 56636,
      "training_step_time": 8.752894639968872
    },
    {
      "epoch": 5.401325225830078e-05,
      "model_forward_time": 1.7462737560272217,
      "step": 56637
    },
    {
      "epoch": 5.401325225830078e-05,
      "step": 56637,
      "training_step_time": 7.6774067878723145
    },
    {
      "epoch": 5.4014205932617186e-05,
      "model_forward_time": 1.3949761390686035,
      "step": 56638
    },
    {
      "epoch": 5.4014205932617186e-05,
      "step": 56638,
      "training_step_time": 8.95749306678772
    },
    {
      "epoch": 5.4015159606933597e-05,
      "model_forward_time": 1.7477340698242188,
      "step": 56639
    },
    {
      "epoch": 5.4015159606933597e-05,
      "step": 56639,
      "training_step_time": 7.3144447803497314
    },
    {
      "epoch": 5.401611328125e-05,
      "grad_norm": 0.17595186829566956,
      "learning_rate": 4.317763739873978e-06,
      "loss": 0.0524,
      "step": 56640
    },
    {
      "epoch": 5.401611328125e-05,
      "model_forward_time": 1.7417161464691162,
      "step": 56640
    },
    {
      "dynamics_loss_avg": 0.042632554098963736,
      "epoch": 5.401611328125e-05,
      "step": 56640
    },
    {
      "action_loss_avg": 0.0019817402469925584,
      "epoch": 5.401611328125e-05,
      "step": 56640
    },
    {
      "epoch": 5.401611328125e-05,
      "step": 56640,
      "training_step_time": 7.5454652309417725
    },
    {
      "epoch": 5.4017066955566405e-05,
      "model_forward_time": 1.7561473846435547,
      "step": 56641
    },
    {
      "epoch": 5.4017066955566405e-05,
      "step": 56641,
      "training_step_time": 7.64545464515686
    },
    {
      "epoch": 5.4018020629882815e-05,
      "model_forward_time": 1.755676031112671,
      "step": 56642
    },
    {
      "epoch": 5.4018020629882815e-05,
      "step": 56642,
      "training_step_time": 7.445265769958496
    },
    {
      "epoch": 5.401897430419922e-05,
      "model_forward_time": 2.8457319736480713,
      "step": 56643
    },
    {
      "epoch": 5.401897430419922e-05,
      "step": 56643,
      "training_step_time": 7.977179288864136
    },
    {
      "epoch": 5.4019927978515623e-05,
      "model_forward_time": 1.7541015148162842,
      "step": 56644
    },
    {
      "epoch": 5.4019927978515623e-05,
      "step": 56644,
      "training_step_time": 7.756066083908081
    },
    {
      "epoch": 5.4020881652832034e-05,
      "model_forward_time": 1.7663466930389404,
      "step": 56645
    },
    {
      "epoch": 5.4020881652832034e-05,
      "step": 56645,
      "training_step_time": 8.59914493560791
    },
    {
      "epoch": 5.402183532714844e-05,
      "model_forward_time": 1.7542593479156494,
      "step": 56646
    },
    {
      "epoch": 5.402183532714844e-05,
      "step": 56646,
      "training_step_time": 7.639437675476074
    },
    {
      "epoch": 5.402278900146484e-05,
      "model_forward_time": 1.7539842128753662,
      "step": 56647
    },
    {
      "epoch": 5.402278900146484e-05,
      "step": 56647,
      "training_step_time": 7.603403329849243
    },
    {
      "epoch": 5.402374267578125e-05,
      "model_forward_time": 1.7570791244506836,
      "step": 56648
    },
    {
      "epoch": 5.402374267578125e-05,
      "step": 56648,
      "training_step_time": 7.589192628860474
    },
    {
      "epoch": 5.402469635009766e-05,
      "model_forward_time": 1.7548668384552002,
      "step": 56649
    },
    {
      "epoch": 5.402469635009766e-05,
      "step": 56649,
      "training_step_time": 7.505509376525879
    },
    {
      "epoch": 5.402565002441406e-05,
      "grad_norm": 0.13799114525318146,
      "learning_rate": 4.31612577173538e-06,
      "loss": 0.0562,
      "step": 56650
    },
    {
      "epoch": 5.402565002441406e-05,
      "model_forward_time": 1.7531158924102783,
      "step": 56650
    },
    {
      "dynamics_loss_avg": 0.05498954150825739,
      "epoch": 5.402565002441406e-05,
      "step": 56650
    },
    {
      "action_loss_avg": 0.002712531416909769,
      "epoch": 5.402565002441406e-05,
      "step": 56650
    },
    {
      "epoch": 5.402565002441406e-05,
      "step": 56650,
      "training_step_time": 7.589876651763916
    },
    {
      "epoch": 5.402660369873047e-05,
      "model_forward_time": 1.743999719619751,
      "step": 56651
    },
    {
      "epoch": 5.402660369873047e-05,
      "step": 56651,
      "training_step_time": 7.604683876037598
    },
    {
      "epoch": 5.4027557373046876e-05,
      "model_forward_time": 1.7571680545806885,
      "step": 56652
    },
    {
      "epoch": 5.4027557373046876e-05,
      "step": 56652,
      "training_step_time": 8.151551246643066
    },
    {
      "epoch": 5.402851104736328e-05,
      "model_forward_time": 2.8703672885894775,
      "step": 56653
    },
    {
      "epoch": 5.402851104736328e-05,
      "step": 56653,
      "training_step_time": 7.873601675033569
    },
    {
      "epoch": 5.402946472167969e-05,
      "model_forward_time": 1.7495312690734863,
      "step": 56654
    },
    {
      "epoch": 5.402946472167969e-05,
      "step": 56654,
      "training_step_time": 7.871901035308838
    },
    {
      "epoch": 5.4030418395996095e-05,
      "model_forward_time": 1.861922025680542,
      "step": 56655
    },
    {
      "epoch": 5.4030418395996095e-05,
      "step": 56655,
      "training_step_time": 8.609404563903809
    },
    {
      "epoch": 5.40313720703125e-05,
      "model_forward_time": 1.7368035316467285,
      "step": 56656
    },
    {
      "epoch": 5.40313720703125e-05,
      "step": 56656,
      "training_step_time": 7.932124376296997
    },
    {
      "epoch": 5.40323257446289e-05,
      "model_forward_time": 1.7497098445892334,
      "step": 56657
    },
    {
      "epoch": 5.40323257446289e-05,
      "step": 56657,
      "training_step_time": 8.432629823684692
    },
    {
      "epoch": 5.4033279418945314e-05,
      "model_forward_time": 1.8037304878234863,
      "step": 56658
    },
    {
      "epoch": 5.4033279418945314e-05,
      "step": 56658,
      "training_step_time": 7.649425268173218
    },
    {
      "epoch": 5.403423309326172e-05,
      "model_forward_time": 1.7440907955169678,
      "step": 56659
    },
    {
      "epoch": 5.403423309326172e-05,
      "step": 56659,
      "training_step_time": 7.871110916137695
    },
    {
      "epoch": 5.403518676757812e-05,
      "grad_norm": 0.09069357812404633,
      "learning_rate": 4.314487878384239e-06,
      "loss": 0.0529,
      "step": 56660
    },
    {
      "epoch": 5.403518676757812e-05,
      "model_forward_time": 1.7531390190124512,
      "step": 56660
    },
    {
      "dynamics_loss_avg": 0.05586508326232433,
      "epoch": 5.403518676757812e-05,
      "step": 56660
    },
    {
      "action_loss_avg": 0.0015844430192373693,
      "epoch": 5.403518676757812e-05,
      "step": 56660
    },
    {
      "epoch": 5.403518676757812e-05,
      "step": 56660,
      "training_step_time": 7.840732097625732
    },
    {
      "epoch": 5.403614044189453e-05,
      "model_forward_time": 1.747089147567749,
      "step": 56661
    },
    {
      "epoch": 5.403614044189453e-05,
      "step": 56661,
      "training_step_time": 7.942983388900757
    },
    {
      "epoch": 5.4037094116210937e-05,
      "model_forward_time": 2.930860757827759,
      "step": 56662
    },
    {
      "epoch": 5.4037094116210937e-05,
      "step": 56662,
      "training_step_time": 8.942105054855347
    },
    {
      "epoch": 5.403804779052734e-05,
      "model_forward_time": 1.7505717277526855,
      "step": 56663
    },
    {
      "epoch": 5.403804779052734e-05,
      "step": 56663,
      "training_step_time": 8.726148843765259
    },
    {
      "epoch": 5.403900146484375e-05,
      "model_forward_time": 1.805138111114502,
      "step": 56664
    },
    {
      "epoch": 5.403900146484375e-05,
      "step": 56664,
      "training_step_time": 9.022975206375122
    },
    {
      "epoch": 5.4039955139160155e-05,
      "model_forward_time": 1.7490055561065674,
      "step": 56665
    },
    {
      "epoch": 5.4039955139160155e-05,
      "step": 56665,
      "training_step_time": 7.631277322769165
    },
    {
      "epoch": 5.404090881347656e-05,
      "model_forward_time": 1.7525432109832764,
      "step": 56666
    },
    {
      "epoch": 5.404090881347656e-05,
      "step": 56666,
      "training_step_time": 8.704999685287476
    },
    {
      "epoch": 5.404186248779297e-05,
      "model_forward_time": 1.7572169303894043,
      "step": 56667
    },
    {
      "epoch": 5.404186248779297e-05,
      "step": 56667,
      "training_step_time": 7.528089284896851
    },
    {
      "epoch": 5.4042816162109374e-05,
      "model_forward_time": 1.756591558456421,
      "step": 56668
    },
    {
      "epoch": 5.4042816162109374e-05,
      "step": 56668,
      "training_step_time": 7.926292181015015
    },
    {
      "epoch": 5.404376983642578e-05,
      "model_forward_time": 1.7564635276794434,
      "step": 56669
    },
    {
      "epoch": 5.404376983642578e-05,
      "step": 56669,
      "training_step_time": 7.864573001861572
    },
    {
      "epoch": 5.404472351074219e-05,
      "grad_norm": 0.10374082624912262,
      "learning_rate": 4.312850059999673e-06,
      "loss": 0.0565,
      "step": 56670
    },
    {
      "epoch": 5.404472351074219e-05,
      "model_forward_time": 1.751189947128296,
      "step": 56670
    },
    {
      "dynamics_loss_avg": 0.0669687146320939,
      "epoch": 5.404472351074219e-05,
      "step": 56670
    },
    {
      "action_loss_avg": 0.001508213847409934,
      "epoch": 5.404472351074219e-05,
      "step": 56670
    },
    {
      "epoch": 5.404472351074219e-05,
      "step": 56670,
      "training_step_time": 7.756112337112427
    },
    {
      "epoch": 5.404567718505859e-05,
      "model_forward_time": 2.9340405464172363,
      "step": 56671
    },
    {
      "epoch": 5.404567718505859e-05,
      "step": 56671,
      "training_step_time": 8.533560752868652
    },
    {
      "epoch": 5.4046630859375e-05,
      "model_forward_time": 1.7587549686431885,
      "step": 56672
    },
    {
      "epoch": 5.4046630859375e-05,
      "step": 56672,
      "training_step_time": 7.873751163482666
    },
    {
      "epoch": 5.404758453369141e-05,
      "model_forward_time": 1.7592616081237793,
      "step": 56673
    },
    {
      "epoch": 5.404758453369141e-05,
      "step": 56673,
      "training_step_time": 8.715704679489136
    },
    {
      "epoch": 5.404853820800781e-05,
      "model_forward_time": 1.7604334354400635,
      "step": 56674
    },
    {
      "epoch": 5.404853820800781e-05,
      "step": 56674,
      "training_step_time": 8.688453912734985
    },
    {
      "epoch": 5.4049491882324216e-05,
      "model_forward_time": 1.7627878189086914,
      "step": 56675
    },
    {
      "epoch": 5.4049491882324216e-05,
      "step": 56675,
      "training_step_time": 7.897190809249878
    },
    {
      "epoch": 5.405044555664063e-05,
      "model_forward_time": 1.746241569519043,
      "step": 56676
    },
    {
      "epoch": 5.405044555664063e-05,
      "step": 56676,
      "training_step_time": 7.753286600112915
    },
    {
      "epoch": 5.405139923095703e-05,
      "model_forward_time": 1.7636523246765137,
      "step": 56677
    },
    {
      "epoch": 5.405139923095703e-05,
      "step": 56677,
      "training_step_time": 8.002742528915405
    },
    {
      "epoch": 5.4052352905273435e-05,
      "model_forward_time": 1.762589931488037,
      "step": 56678
    },
    {
      "epoch": 5.4052352905273435e-05,
      "step": 56678,
      "training_step_time": 7.83537745475769
    },
    {
      "epoch": 5.4053306579589846e-05,
      "model_forward_time": 1.7671666145324707,
      "step": 56679
    },
    {
      "epoch": 5.4053306579589846e-05,
      "step": 56679,
      "training_step_time": 7.733457326889038
    },
    {
      "epoch": 5.405426025390625e-05,
      "grad_norm": 0.11061395704746246,
      "learning_rate": 4.31121231676079e-06,
      "loss": 0.0546,
      "step": 56680
    },
    {
      "epoch": 5.405426025390625e-05,
      "model_forward_time": 2.884082555770874,
      "step": 56680
    },
    {
      "dynamics_loss_avg": 0.046881253086030486,
      "epoch": 5.405426025390625e-05,
      "step": 56680
    },
    {
      "action_loss_avg": 0.001594006217783317,
      "epoch": 5.405426025390625e-05,
      "step": 56680
    },
    {
      "epoch": 5.405426025390625e-05,
      "step": 56680,
      "training_step_time": 8.02676796913147
    },
    {
      "epoch": 5.4055213928222654e-05,
      "model_forward_time": 1.7553930282592773,
      "step": 56681
    },
    {
      "epoch": 5.4055213928222654e-05,
      "step": 56681,
      "training_step_time": 8.535572528839111
    },
    {
      "epoch": 5.4056167602539065e-05,
      "model_forward_time": 1.7397046089172363,
      "step": 56682
    },
    {
      "epoch": 5.4056167602539065e-05,
      "step": 56682,
      "training_step_time": 8.711249828338623
    },
    {
      "epoch": 5.405712127685547e-05,
      "model_forward_time": 1.8683829307556152,
      "step": 56683
    },
    {
      "epoch": 5.405712127685547e-05,
      "step": 56683,
      "training_step_time": 10.436829090118408
    },
    {
      "epoch": 5.405807495117187e-05,
      "model_forward_time": 1.7538444995880127,
      "step": 56684
    },
    {
      "epoch": 5.405807495117187e-05,
      "step": 56684,
      "training_step_time": 8.644424438476562
    },
    {
      "epoch": 5.4059028625488283e-05,
      "model_forward_time": 1.7430410385131836,
      "step": 56685
    },
    {
      "epoch": 5.4059028625488283e-05,
      "step": 56685,
      "training_step_time": 7.682307481765747
    },
    {
      "epoch": 5.405998229980469e-05,
      "model_forward_time": 1.8100802898406982,
      "step": 56686
    },
    {
      "epoch": 5.405998229980469e-05,
      "step": 56686,
      "training_step_time": 7.818507194519043
    },
    {
      "epoch": 5.406093597412109e-05,
      "model_forward_time": 1.7436323165893555,
      "step": 56687
    },
    {
      "epoch": 5.406093597412109e-05,
      "step": 56687,
      "training_step_time": 8.085883617401123
    },
    {
      "epoch": 5.40618896484375e-05,
      "model_forward_time": 1.7446279525756836,
      "step": 56688
    },
    {
      "epoch": 5.40618896484375e-05,
      "step": 56688,
      "training_step_time": 7.820403814315796
    },
    {
      "epoch": 5.4062843322753906e-05,
      "model_forward_time": 2.871333599090576,
      "step": 56689
    },
    {
      "epoch": 5.4062843322753906e-05,
      "step": 56689,
      "training_step_time": 8.08957290649414
    },
    {
      "epoch": 5.406379699707031e-05,
      "grad_norm": 0.11055714637041092,
      "learning_rate": 4.309574648846694e-06,
      "loss": 0.0552,
      "step": 56690
    },
    {
      "epoch": 5.406379699707031e-05,
      "model_forward_time": 1.7996230125427246,
      "step": 56690
    },
    {
      "dynamics_loss_avg": 0.058526120521128176,
      "epoch": 5.406379699707031e-05,
      "step": 56690
    },
    {
      "action_loss_avg": 0.0019002600776730106,
      "epoch": 5.406379699707031e-05,
      "step": 56690
    },
    {
      "epoch": 5.406379699707031e-05,
      "step": 56690,
      "training_step_time": 7.886219263076782
    },
    {
      "epoch": 5.406475067138672e-05,
      "model_forward_time": 1.7461962699890137,
      "step": 56691
    },
    {
      "epoch": 5.406475067138672e-05,
      "step": 56691,
      "training_step_time": 9.423794269561768
    },
    {
      "epoch": 5.4065704345703125e-05,
      "model_forward_time": 1.7529809474945068,
      "step": 56692
    },
    {
      "epoch": 5.4065704345703125e-05,
      "step": 56692,
      "training_step_time": 11.697089433670044
    },
    {
      "epoch": 5.406665802001953e-05,
      "model_forward_time": 1.7495439052581787,
      "step": 56693
    },
    {
      "epoch": 5.406665802001953e-05,
      "step": 56693,
      "training_step_time": 8.957730531692505
    },
    {
      "epoch": 5.406761169433594e-05,
      "model_forward_time": 1.7542316913604736,
      "step": 56694
    },
    {
      "epoch": 5.406761169433594e-05,
      "step": 56694,
      "training_step_time": 7.884843349456787
    },
    {
      "epoch": 5.4068565368652344e-05,
      "model_forward_time": 1.7489233016967773,
      "step": 56695
    },
    {
      "epoch": 5.4068565368652344e-05,
      "step": 56695,
      "training_step_time": 7.881013870239258
    },
    {
      "epoch": 5.406951904296875e-05,
      "model_forward_time": 1.750230312347412,
      "step": 56696
    },
    {
      "epoch": 5.406951904296875e-05,
      "step": 56696,
      "training_step_time": 7.212079286575317
    },
    {
      "epoch": 5.407047271728516e-05,
      "model_forward_time": 1.7520654201507568,
      "step": 56697
    },
    {
      "epoch": 5.407047271728516e-05,
      "step": 56697,
      "training_step_time": 7.84906005859375
    },
    {
      "epoch": 5.407142639160156e-05,
      "model_forward_time": 1.7468836307525635,
      "step": 56698
    },
    {
      "epoch": 5.407142639160156e-05,
      "step": 56698,
      "training_step_time": 9.098999500274658
    },
    {
      "epoch": 5.407238006591797e-05,
      "model_forward_time": 1.7519121170043945,
      "step": 56699
    },
    {
      "epoch": 5.407238006591797e-05,
      "step": 56699,
      "training_step_time": 8.408950567245483
    },
    {
      "epoch": 5.407333374023438e-05,
      "grad_norm": 0.11797254532575607,
      "learning_rate": 4.307937056436476e-06,
      "loss": 0.0584,
      "step": 56700
    },
    {
      "epoch": 5.407333374023438e-05,
      "model_forward_time": 1.7579865455627441,
      "step": 56700
    },
    {
      "dynamics_loss_avg": 0.062158720754086974,
      "epoch": 5.407333374023438e-05,
      "step": 56700
    },
    {
      "action_loss_avg": 0.0017922239494509995,
      "epoch": 5.407333374023438e-05,
      "step": 56700
    },
    {
      "epoch": 5.407333374023438e-05,
      "step": 56700,
      "training_step_time": 10.04326343536377
    },
    {
      "epoch": 5.407428741455078e-05,
      "model_forward_time": 1.7597379684448242,
      "step": 56701
    },
    {
      "epoch": 5.407428741455078e-05,
      "step": 56701,
      "training_step_time": 7.83219838142395
    },
    {
      "epoch": 5.4075241088867186e-05,
      "model_forward_time": 1.8499982357025146,
      "step": 56702
    },
    {
      "epoch": 5.4075241088867186e-05,
      "step": 56702,
      "training_step_time": 8.557822942733765
    },
    {
      "epoch": 5.4076194763183597e-05,
      "model_forward_time": 1.7581720352172852,
      "step": 56703
    },
    {
      "epoch": 5.4076194763183597e-05,
      "step": 56703,
      "training_step_time": 7.829771041870117
    },
    {
      "epoch": 5.40771484375e-05,
      "model_forward_time": 1.7569780349731445,
      "step": 56704
    },
    {
      "epoch": 5.40771484375e-05,
      "step": 56704,
      "training_step_time": 7.804274320602417
    },
    {
      "epoch": 5.4078102111816405e-05,
      "model_forward_time": 1.7511746883392334,
      "step": 56705
    },
    {
      "epoch": 5.4078102111816405e-05,
      "step": 56705,
      "training_step_time": 7.396899461746216
    },
    {
      "epoch": 5.4079055786132815e-05,
      "model_forward_time": 1.7264750003814697,
      "step": 56706
    },
    {
      "epoch": 5.4079055786132815e-05,
      "step": 56706,
      "training_step_time": 7.873964071273804
    },
    {
      "epoch": 5.408000946044922e-05,
      "model_forward_time": 1.7484192848205566,
      "step": 56707
    },
    {
      "epoch": 5.408000946044922e-05,
      "step": 56707,
      "training_step_time": 7.984551429748535
    },
    {
      "epoch": 5.4080963134765623e-05,
      "model_forward_time": 2.8950345516204834,
      "step": 56708
    },
    {
      "epoch": 5.4080963134765623e-05,
      "step": 56708,
      "training_step_time": 7.877136707305908
    },
    {
      "epoch": 5.4081916809082034e-05,
      "model_forward_time": 1.7451353073120117,
      "step": 56709
    },
    {
      "epoch": 5.4081916809082034e-05,
      "step": 56709,
      "training_step_time": 8.883159875869751
    },
    {
      "epoch": 5.408287048339844e-05,
      "grad_norm": 0.10434693098068237,
      "learning_rate": 4.306299539709221e-06,
      "loss": 0.059,
      "step": 56710
    },
    {
      "epoch": 5.408287048339844e-05,
      "model_forward_time": 1.753267765045166,
      "step": 56710
    },
    {
      "dynamics_loss_avg": 0.05358493737876415,
      "epoch": 5.408287048339844e-05,
      "step": 56710
    },
    {
      "action_loss_avg": 0.0012600064772414043,
      "epoch": 5.408287048339844e-05,
      "step": 56710
    },
    {
      "epoch": 5.408287048339844e-05,
      "step": 56710,
      "training_step_time": 8.746751308441162
    },
    {
      "epoch": 5.408382415771484e-05,
      "model_forward_time": 1.840142011642456,
      "step": 56711
    },
    {
      "epoch": 5.408382415771484e-05,
      "step": 56711,
      "training_step_time": 8.394764423370361
    },
    {
      "epoch": 5.408477783203125e-05,
      "model_forward_time": 1.7512702941894531,
      "step": 56712
    },
    {
      "epoch": 5.408477783203125e-05,
      "step": 56712,
      "training_step_time": 7.816185235977173
    },
    {
      "epoch": 5.408573150634766e-05,
      "model_forward_time": 1.750436782836914,
      "step": 56713
    },
    {
      "epoch": 5.408573150634766e-05,
      "step": 56713,
      "training_step_time": 8.408048629760742
    },
    {
      "epoch": 5.408668518066406e-05,
      "model_forward_time": 1.7417206764221191,
      "step": 56714
    },
    {
      "epoch": 5.408668518066406e-05,
      "step": 56714,
      "training_step_time": 7.838679552078247
    },
    {
      "epoch": 5.408763885498047e-05,
      "model_forward_time": 1.7529380321502686,
      "step": 56715
    },
    {
      "epoch": 5.408763885498047e-05,
      "step": 56715,
      "training_step_time": 8.606165885925293
    },
    {
      "epoch": 5.4088592529296876e-05,
      "model_forward_time": 1.7451066970825195,
      "step": 56716
    },
    {
      "epoch": 5.4088592529296876e-05,
      "step": 56716,
      "training_step_time": 8.715583562850952
    },
    {
      "epoch": 5.408954620361328e-05,
      "model_forward_time": 2.9207959175109863,
      "step": 56717
    },
    {
      "epoch": 5.408954620361328e-05,
      "step": 56717,
      "training_step_time": 8.920431137084961
    },
    {
      "epoch": 5.409049987792969e-05,
      "model_forward_time": 1.748516321182251,
      "step": 56718
    },
    {
      "epoch": 5.409049987792969e-05,
      "step": 56718,
      "training_step_time": 7.852423667907715
    },
    {
      "epoch": 5.4091453552246095e-05,
      "model_forward_time": 1.7559306621551514,
      "step": 56719
    },
    {
      "epoch": 5.4091453552246095e-05,
      "step": 56719,
      "training_step_time": 8.732541799545288
    },
    {
      "epoch": 5.40924072265625e-05,
      "grad_norm": 0.06992463767528534,
      "learning_rate": 4.304662098844005e-06,
      "loss": 0.0569,
      "step": 56720
    },
    {
      "epoch": 5.40924072265625e-05,
      "model_forward_time": 1.808164119720459,
      "step": 56720
    },
    {
      "dynamics_loss_avg": 0.04873987725004554,
      "epoch": 5.40924072265625e-05,
      "step": 56720
    },
    {
      "action_loss_avg": 0.0013478634733473881,
      "epoch": 5.40924072265625e-05,
      "step": 56720
    },
    {
      "epoch": 5.40924072265625e-05,
      "step": 56720,
      "training_step_time": 8.798503875732422
    },
    {
      "epoch": 5.40933609008789e-05,
      "model_forward_time": 1.7487587928771973,
      "step": 56721
    },
    {
      "epoch": 5.40933609008789e-05,
      "step": 56721,
      "training_step_time": 7.803474426269531
    },
    {
      "epoch": 5.4094314575195314e-05,
      "model_forward_time": 1.7468457221984863,
      "step": 56722
    },
    {
      "epoch": 5.4094314575195314e-05,
      "step": 56722,
      "training_step_time": 7.833360195159912
    },
    {
      "epoch": 5.409526824951172e-05,
      "model_forward_time": 1.7477052211761475,
      "step": 56723
    },
    {
      "epoch": 5.409526824951172e-05,
      "step": 56723,
      "training_step_time": 7.900436878204346
    },
    {
      "epoch": 5.409622192382812e-05,
      "model_forward_time": 1.7433421611785889,
      "step": 56724
    },
    {
      "epoch": 5.409622192382812e-05,
      "step": 56724,
      "training_step_time": 7.531716585159302
    },
    {
      "epoch": 5.409717559814453e-05,
      "model_forward_time": 1.7500216960906982,
      "step": 56725
    },
    {
      "epoch": 5.409717559814453e-05,
      "step": 56725,
      "training_step_time": 7.74726939201355
    },
    {
      "epoch": 5.409812927246094e-05,
      "model_forward_time": 2.8774516582489014,
      "step": 56726
    },
    {
      "epoch": 5.409812927246094e-05,
      "step": 56726,
      "training_step_time": 8.055740594863892
    },
    {
      "epoch": 5.409908294677734e-05,
      "model_forward_time": 1.7544684410095215,
      "step": 56727
    },
    {
      "epoch": 5.409908294677734e-05,
      "step": 56727,
      "training_step_time": 7.823531150817871
    },
    {
      "epoch": 5.410003662109375e-05,
      "model_forward_time": 1.7554025650024414,
      "step": 56728
    },
    {
      "epoch": 5.410003662109375e-05,
      "step": 56728,
      "training_step_time": 10.342241525650024
    },
    {
      "epoch": 5.4100990295410155e-05,
      "model_forward_time": 1.7863023281097412,
      "step": 56729
    },
    {
      "epoch": 5.4100990295410155e-05,
      "step": 56729,
      "training_step_time": 8.941100597381592
    },
    {
      "epoch": 5.410194396972656e-05,
      "grad_norm": 0.11533305048942566,
      "learning_rate": 4.303024734019897e-06,
      "loss": 0.0575,
      "step": 56730
    },
    {
      "epoch": 5.410194396972656e-05,
      "model_forward_time": 1.739185094833374,
      "step": 56730
    },
    {
      "dynamics_loss_avg": 0.05458314241841435,
      "epoch": 5.410194396972656e-05,
      "step": 56730
    },
    {
      "action_loss_avg": 0.00221407585486304,
      "epoch": 5.410194396972656e-05,
      "step": 56730
    },
    {
      "epoch": 5.410194396972656e-05,
      "step": 56730,
      "training_step_time": 8.752941608428955
    },
    {
      "epoch": 5.410289764404297e-05,
      "model_forward_time": 1.7476627826690674,
      "step": 56731
    },
    {
      "epoch": 5.410289764404297e-05,
      "step": 56731,
      "training_step_time": 7.840430021286011
    },
    {
      "epoch": 5.4103851318359374e-05,
      "model_forward_time": 1.772162914276123,
      "step": 56732
    },
    {
      "epoch": 5.4103851318359374e-05,
      "step": 56732,
      "training_step_time": 7.906715393066406
    },
    {
      "epoch": 5.410480499267578e-05,
      "model_forward_time": 1.7527432441711426,
      "step": 56733
    },
    {
      "epoch": 5.410480499267578e-05,
      "step": 56733,
      "training_step_time": 8.133078575134277
    },
    {
      "epoch": 5.410575866699219e-05,
      "model_forward_time": 1.741631269454956,
      "step": 56734
    },
    {
      "epoch": 5.410575866699219e-05,
      "step": 56734,
      "training_step_time": 7.550999402999878
    },
    {
      "epoch": 5.410671234130859e-05,
      "model_forward_time": 2.844892978668213,
      "step": 56735
    },
    {
      "epoch": 5.410671234130859e-05,
      "step": 56735,
      "training_step_time": 8.030068397521973
    },
    {
      "epoch": 5.4107666015625e-05,
      "model_forward_time": 1.7513816356658936,
      "step": 56736
    },
    {
      "epoch": 5.4107666015625e-05,
      "step": 56736,
      "training_step_time": 7.78488564491272
    },
    {
      "epoch": 5.410861968994141e-05,
      "model_forward_time": 1.7474994659423828,
      "step": 56737
    },
    {
      "epoch": 5.410861968994141e-05,
      "step": 56737,
      "training_step_time": 8.608054161071777
    },
    {
      "epoch": 5.410957336425781e-05,
      "model_forward_time": 1.752457857131958,
      "step": 56738
    },
    {
      "epoch": 5.410957336425781e-05,
      "step": 56738,
      "training_step_time": 8.547137260437012
    },
    {
      "epoch": 5.4110527038574216e-05,
      "model_forward_time": 1.8576266765594482,
      "step": 56739
    },
    {
      "epoch": 5.4110527038574216e-05,
      "step": 56739,
      "training_step_time": 7.87869930267334
    },
    {
      "epoch": 5.411148071289063e-05,
      "grad_norm": 0.08180271089076996,
      "learning_rate": 4.301387445415957e-06,
      "loss": 0.0586,
      "step": 56740
    },
    {
      "epoch": 5.411148071289063e-05,
      "model_forward_time": 1.7480816841125488,
      "step": 56740
    },
    {
      "dynamics_loss_avg": 0.061496279947459695,
      "epoch": 5.411148071289063e-05,
      "step": 56740
    },
    {
      "action_loss_avg": 0.002145023288903758,
      "epoch": 5.411148071289063e-05,
      "step": 56740
    },
    {
      "epoch": 5.411148071289063e-05,
      "step": 56740,
      "training_step_time": 10.07060194015503
    },
    {
      "epoch": 5.411243438720703e-05,
      "model_forward_time": 1.7512214183807373,
      "step": 56741
    },
    {
      "epoch": 5.411243438720703e-05,
      "step": 56741,
      "training_step_time": 7.894997835159302
    },
    {
      "epoch": 5.4113388061523435e-05,
      "model_forward_time": 1.7509958744049072,
      "step": 56742
    },
    {
      "epoch": 5.4113388061523435e-05,
      "step": 56742,
      "training_step_time": 6.991289377212524
    },
    {
      "epoch": 5.4114341735839846e-05,
      "model_forward_time": 1.7431297302246094,
      "step": 56743
    },
    {
      "epoch": 5.4114341735839846e-05,
      "step": 56743,
      "training_step_time": 7.821900844573975
    },
    {
      "epoch": 5.411529541015625e-05,
      "model_forward_time": 2.8795127868652344,
      "step": 56744
    },
    {
      "epoch": 5.411529541015625e-05,
      "step": 56744,
      "training_step_time": 7.980020046234131
    },
    {
      "epoch": 5.4116249084472654e-05,
      "model_forward_time": 1.3910651206970215,
      "step": 56745
    },
    {
      "epoch": 5.4116249084472654e-05,
      "step": 56745,
      "training_step_time": 7.771193265914917
    },
    {
      "epoch": 5.4117202758789065e-05,
      "model_forward_time": 1.7592158317565918,
      "step": 56746
    },
    {
      "epoch": 5.4117202758789065e-05,
      "step": 56746,
      "training_step_time": 7.842145204544067
    },
    {
      "epoch": 5.411815643310547e-05,
      "model_forward_time": 1.757965326309204,
      "step": 56747
    },
    {
      "epoch": 5.411815643310547e-05,
      "step": 56747,
      "training_step_time": 8.526328802108765
    },
    {
      "epoch": 5.411911010742187e-05,
      "model_forward_time": 1.8434910774230957,
      "step": 56748
    },
    {
      "epoch": 5.411911010742187e-05,
      "step": 56748,
      "training_step_time": 8.504745244979858
    },
    {
      "epoch": 5.4120063781738283e-05,
      "model_forward_time": 1.7555019855499268,
      "step": 56749
    },
    {
      "epoch": 5.4120063781738283e-05,
      "step": 56749,
      "training_step_time": 8.747485399246216
    },
    {
      "epoch": 5.412101745605469e-05,
      "grad_norm": 0.09392653405666351,
      "learning_rate": 4.299750233211233e-06,
      "loss": 0.055,
      "step": 56750
    },
    {
      "epoch": 5.412101745605469e-05,
      "model_forward_time": 1.7498114109039307,
      "step": 56750
    },
    {
      "dynamics_loss_avg": 0.05221590641885996,
      "epoch": 5.412101745605469e-05,
      "step": 56750
    },
    {
      "action_loss_avg": 0.001540104509331286,
      "epoch": 5.412101745605469e-05,
      "step": 56750
    },
    {
      "epoch": 5.412101745605469e-05,
      "step": 56750,
      "training_step_time": 7.749757289886475
    },
    {
      "epoch": 5.412197113037109e-05,
      "model_forward_time": 1.7517709732055664,
      "step": 56751
    },
    {
      "epoch": 5.412197113037109e-05,
      "step": 56751,
      "training_step_time": 7.94531774520874
    },
    {
      "epoch": 5.41229248046875e-05,
      "model_forward_time": 1.7479982376098633,
      "step": 56752
    },
    {
      "epoch": 5.41229248046875e-05,
      "step": 56752,
      "training_step_time": 7.604479551315308
    },
    {
      "epoch": 5.4123878479003906e-05,
      "model_forward_time": 1.7486670017242432,
      "step": 56753
    },
    {
      "epoch": 5.4123878479003906e-05,
      "step": 56753,
      "training_step_time": 8.04146695137024
    },
    {
      "epoch": 5.412483215332031e-05,
      "model_forward_time": 2.9438416957855225,
      "step": 56754
    },
    {
      "epoch": 5.412483215332031e-05,
      "step": 56754,
      "training_step_time": 7.97603440284729
    },
    {
      "epoch": 5.412578582763672e-05,
      "model_forward_time": 1.7469761371612549,
      "step": 56755
    },
    {
      "epoch": 5.412578582763672e-05,
      "step": 56755,
      "training_step_time": 7.893489837646484
    },
    {
      "epoch": 5.4126739501953125e-05,
      "model_forward_time": 1.758030652999878,
      "step": 56756
    },
    {
      "epoch": 5.4126739501953125e-05,
      "step": 56756,
      "training_step_time": 7.93530797958374
    },
    {
      "epoch": 5.412769317626953e-05,
      "model_forward_time": 1.8204607963562012,
      "step": 56757
    },
    {
      "epoch": 5.412769317626953e-05,
      "step": 56757,
      "training_step_time": 7.861985921859741
    },
    {
      "epoch": 5.412864685058594e-05,
      "model_forward_time": 1.7492046356201172,
      "step": 56758
    },
    {
      "epoch": 5.412864685058594e-05,
      "step": 56758,
      "training_step_time": 8.497820377349854
    },
    {
      "epoch": 5.4129600524902344e-05,
      "model_forward_time": 1.7455358505249023,
      "step": 56759
    },
    {
      "epoch": 5.4129600524902344e-05,
      "step": 56759,
      "training_step_time": 7.7857911586761475
    },
    {
      "epoch": 5.413055419921875e-05,
      "grad_norm": 0.09047982096672058,
      "learning_rate": 4.298113097584772e-06,
      "loss": 0.0587,
      "step": 56760
    },
    {
      "epoch": 5.413055419921875e-05,
      "model_forward_time": 1.754326581954956,
      "step": 56760
    },
    {
      "dynamics_loss_avg": 0.05703103002160788,
      "epoch": 5.413055419921875e-05,
      "step": 56760
    },
    {
      "action_loss_avg": 0.0016267533879727124,
      "epoch": 5.413055419921875e-05,
      "step": 56760
    },
    {
      "epoch": 5.413055419921875e-05,
      "step": 56760,
      "training_step_time": 7.901895761489868
    },
    {
      "epoch": 5.413150787353516e-05,
      "model_forward_time": 1.3906300067901611,
      "step": 56761
    },
    {
      "epoch": 5.413150787353516e-05,
      "step": 56761,
      "training_step_time": 7.808526992797852
    },
    {
      "epoch": 5.413246154785156e-05,
      "model_forward_time": 1.736325740814209,
      "step": 56762
    },
    {
      "epoch": 5.413246154785156e-05,
      "step": 56762,
      "training_step_time": 7.785599946975708
    },
    {
      "epoch": 5.413341522216797e-05,
      "model_forward_time": 2.8914926052093506,
      "step": 56763
    },
    {
      "epoch": 5.413341522216797e-05,
      "step": 56763,
      "training_step_time": 8.530222415924072
    },
    {
      "epoch": 5.413436889648438e-05,
      "model_forward_time": 1.7641539573669434,
      "step": 56764
    },
    {
      "epoch": 5.413436889648438e-05,
      "step": 56764,
      "training_step_time": 7.81366229057312
    },
    {
      "epoch": 5.413532257080078e-05,
      "model_forward_time": 1.38370680809021,
      "step": 56765
    },
    {
      "epoch": 5.413532257080078e-05,
      "step": 56765,
      "training_step_time": 9.164768934249878
    },
    {
      "epoch": 5.4136276245117186e-05,
      "model_forward_time": 1.7897975444793701,
      "step": 56766
    },
    {
      "epoch": 5.4136276245117186e-05,
      "step": 56766,
      "training_step_time": 8.875694274902344
    },
    {
      "epoch": 5.41372299194336e-05,
      "model_forward_time": 1.7589821815490723,
      "step": 56767
    },
    {
      "epoch": 5.41372299194336e-05,
      "step": 56767,
      "training_step_time": 7.842682838439941
    },
    {
      "epoch": 5.413818359375e-05,
      "model_forward_time": 1.7603693008422852,
      "step": 56768
    },
    {
      "epoch": 5.413818359375e-05,
      "step": 56768,
      "training_step_time": 7.776621341705322
    },
    {
      "epoch": 5.4139137268066405e-05,
      "model_forward_time": 1.758568286895752,
      "step": 56769
    },
    {
      "epoch": 5.4139137268066405e-05,
      "step": 56769,
      "training_step_time": 7.882089376449585
    },
    {
      "epoch": 5.4140090942382815e-05,
      "grad_norm": 0.14355790615081787,
      "learning_rate": 4.296476038715608e-06,
      "loss": 0.0571,
      "step": 56770
    },
    {
      "epoch": 5.4140090942382815e-05,
      "model_forward_time": 1.7540605068206787,
      "step": 56770
    },
    {
      "dynamics_loss_avg": 0.04212055765092373,
      "epoch": 5.4140090942382815e-05,
      "step": 56770
    },
    {
      "action_loss_avg": 0.0015272291842848063,
      "epoch": 5.4140090942382815e-05,
      "step": 56770
    },
    {
      "epoch": 5.4140090942382815e-05,
      "step": 56770,
      "training_step_time": 7.835149526596069
    },
    {
      "epoch": 5.414104461669922e-05,
      "model_forward_time": 1.7514684200286865,
      "step": 56771
    },
    {
      "epoch": 5.414104461669922e-05,
      "step": 56771,
      "training_step_time": 7.829754829406738
    },
    {
      "epoch": 5.4141998291015624e-05,
      "model_forward_time": 2.8490781784057617,
      "step": 56772
    },
    {
      "epoch": 5.4141998291015624e-05,
      "step": 56772,
      "training_step_time": 9.771862268447876
    },
    {
      "epoch": 5.4142951965332034e-05,
      "model_forward_time": 1.7551233768463135,
      "step": 56773
    },
    {
      "epoch": 5.4142951965332034e-05,
      "step": 56773,
      "training_step_time": 7.8744730949401855
    },
    {
      "epoch": 5.414390563964844e-05,
      "model_forward_time": 1.720881462097168,
      "step": 56774
    },
    {
      "epoch": 5.414390563964844e-05,
      "step": 56774,
      "training_step_time": 9.557974815368652
    },
    {
      "epoch": 5.414485931396484e-05,
      "model_forward_time": 1.7764182090759277,
      "step": 56775
    },
    {
      "epoch": 5.414485931396484e-05,
      "step": 56775,
      "training_step_time": 8.3908371925354
    },
    {
      "epoch": 5.414581298828125e-05,
      "model_forward_time": 1.3865549564361572,
      "step": 56776
    },
    {
      "epoch": 5.414581298828125e-05,
      "step": 56776,
      "training_step_time": 8.611226320266724
    },
    {
      "epoch": 5.414676666259766e-05,
      "model_forward_time": 1.7562310695648193,
      "step": 56777
    },
    {
      "epoch": 5.414676666259766e-05,
      "step": 56777,
      "training_step_time": 7.870990514755249
    },
    {
      "epoch": 5.414772033691406e-05,
      "model_forward_time": 1.755016565322876,
      "step": 56778
    },
    {
      "epoch": 5.414772033691406e-05,
      "step": 56778,
      "training_step_time": 8.804105520248413
    },
    {
      "epoch": 5.414867401123047e-05,
      "model_forward_time": 1.7567694187164307,
      "step": 56779
    },
    {
      "epoch": 5.414867401123047e-05,
      "step": 56779,
      "training_step_time": 7.818190574645996
    },
    {
      "epoch": 5.4149627685546876e-05,
      "grad_norm": 0.08345390856266022,
      "learning_rate": 4.294839056782767e-06,
      "loss": 0.0571,
      "step": 56780
    },
    {
      "epoch": 5.4149627685546876e-05,
      "model_forward_time": 1.7479665279388428,
      "step": 56780
    },
    {
      "dynamics_loss_avg": 0.06353536173701287,
      "epoch": 5.4149627685546876e-05,
      "step": 56780
    },
    {
      "action_loss_avg": 0.0022938026959309354,
      "epoch": 5.4149627685546876e-05,
      "step": 56780
    },
    {
      "epoch": 5.4149627685546876e-05,
      "step": 56780,
      "training_step_time": 8.608200073242188
    },
    {
      "epoch": 5.415058135986328e-05,
      "model_forward_time": 1.381089448928833,
      "step": 56781
    },
    {
      "epoch": 5.415058135986328e-05,
      "step": 56781,
      "training_step_time": 8.161353826522827
    },
    {
      "epoch": 5.4151535034179684e-05,
      "model_forward_time": 2.9305806159973145,
      "step": 56782
    },
    {
      "epoch": 5.4151535034179684e-05,
      "step": 56782,
      "training_step_time": 7.820952653884888
    },
    {
      "epoch": 5.4152488708496095e-05,
      "model_forward_time": 1.399918556213379,
      "step": 56783
    },
    {
      "epoch": 5.4152488708496095e-05,
      "step": 56783,
      "training_step_time": 9.672779560089111
    },
    {
      "epoch": 5.41534423828125e-05,
      "model_forward_time": 1.7559480667114258,
      "step": 56784
    },
    {
      "epoch": 5.41534423828125e-05,
      "step": 56784,
      "training_step_time": 9.463821649551392
    },
    {
      "epoch": 5.41543960571289e-05,
      "model_forward_time": 1.8665771484375,
      "step": 56785
    },
    {
      "epoch": 5.41543960571289e-05,
      "step": 56785,
      "training_step_time": 7.978727102279663
    },
    {
      "epoch": 5.4155349731445314e-05,
      "model_forward_time": 1.7648615837097168,
      "step": 56786
    },
    {
      "epoch": 5.4155349731445314e-05,
      "step": 56786,
      "training_step_time": 9.208565711975098
    },
    {
      "epoch": 5.415630340576172e-05,
      "model_forward_time": 1.764747142791748,
      "step": 56787
    },
    {
      "epoch": 5.415630340576172e-05,
      "step": 56787,
      "training_step_time": 7.84580659866333
    },
    {
      "epoch": 5.415725708007812e-05,
      "model_forward_time": 1.764918327331543,
      "step": 56788
    },
    {
      "epoch": 5.415725708007812e-05,
      "step": 56788,
      "training_step_time": 7.878392457962036
    },
    {
      "epoch": 5.415821075439453e-05,
      "model_forward_time": 1.7427051067352295,
      "step": 56789
    },
    {
      "epoch": 5.415821075439453e-05,
      "step": 56789,
      "training_step_time": 7.84032940864563
    },
    {
      "epoch": 5.415916442871094e-05,
      "grad_norm": 0.09827131778001785,
      "learning_rate": 4.293202151965266e-06,
      "loss": 0.0522,
      "step": 56790
    },
    {
      "epoch": 5.415916442871094e-05,
      "model_forward_time": 1.7669117450714111,
      "step": 56790
    },
    {
      "dynamics_loss_avg": 0.04572466993704438,
      "epoch": 5.415916442871094e-05,
      "step": 56790
    },
    {
      "action_loss_avg": 0.0011847193527501076,
      "epoch": 5.415916442871094e-05,
      "step": 56790
    },
    {
      "epoch": 5.415916442871094e-05,
      "step": 56790,
      "training_step_time": 9.127332925796509
    },
    {
      "epoch": 5.416011810302734e-05,
      "model_forward_time": 2.881260633468628,
      "step": 56791
    },
    {
      "epoch": 5.416011810302734e-05,
      "step": 56791,
      "training_step_time": 7.966116666793823
    },
    {
      "epoch": 5.416107177734375e-05,
      "model_forward_time": 1.7572121620178223,
      "step": 56792
    },
    {
      "epoch": 5.416107177734375e-05,
      "step": 56792,
      "training_step_time": 7.896097183227539
    },
    {
      "epoch": 5.4162025451660156e-05,
      "model_forward_time": 1.7411587238311768,
      "step": 56793
    },
    {
      "epoch": 5.4162025451660156e-05,
      "step": 56793,
      "training_step_time": 8.587712049484253
    },
    {
      "epoch": 5.416297912597656e-05,
      "model_forward_time": 1.8284025192260742,
      "step": 56794
    },
    {
      "epoch": 5.416297912597656e-05,
      "step": 56794,
      "training_step_time": 9.439902305603027
    },
    {
      "epoch": 5.416393280029297e-05,
      "model_forward_time": 1.7576396465301514,
      "step": 56795
    },
    {
      "epoch": 5.416393280029297e-05,
      "step": 56795,
      "training_step_time": 8.561402559280396
    },
    {
      "epoch": 5.4164886474609374e-05,
      "model_forward_time": 1.7550108432769775,
      "step": 56796
    },
    {
      "epoch": 5.4164886474609374e-05,
      "step": 56796,
      "training_step_time": 7.848439693450928
    },
    {
      "epoch": 5.416584014892578e-05,
      "model_forward_time": 1.7575528621673584,
      "step": 56797
    },
    {
      "epoch": 5.416584014892578e-05,
      "step": 56797,
      "training_step_time": 7.8885178565979
    },
    {
      "epoch": 5.416679382324219e-05,
      "model_forward_time": 1.7594943046569824,
      "step": 56798
    },
    {
      "epoch": 5.416679382324219e-05,
      "step": 56798,
      "training_step_time": 7.86199688911438
    },
    {
      "epoch": 5.416774749755859e-05,
      "model_forward_time": 1.7545037269592285,
      "step": 56799
    },
    {
      "epoch": 5.416774749755859e-05,
      "step": 56799,
      "training_step_time": 8.377155065536499
    },
    {
      "epoch": 5.4168701171875e-05,
      "grad_norm": 0.12291208654642105,
      "learning_rate": 4.291565324442115e-06,
      "loss": 0.0554,
      "step": 56800
    },
    {
      "epoch": 5.4168701171875e-05,
      "model_forward_time": 2.841245174407959,
      "step": 56800
    },
    {
      "dynamics_loss_avg": 0.0638862770050764,
      "epoch": 5.4168701171875e-05,
      "step": 56800
    },
    {
      "action_loss_avg": 0.0012398383987601846,
      "epoch": 5.4168701171875e-05,
      "step": 56800
    },
    {
      "epoch": 5.4168701171875e-05,
      "step": 56800,
      "training_step_time": 8.076741933822632
    },
    {
      "epoch": 5.416965484619141e-05,
      "model_forward_time": 1.7554950714111328,
      "step": 56801
    },
    {
      "epoch": 5.416965484619141e-05,
      "step": 56801,
      "training_step_time": 8.534247398376465
    },
    {
      "epoch": 5.417060852050781e-05,
      "model_forward_time": 1.745408296585083,
      "step": 56802
    },
    {
      "epoch": 5.417060852050781e-05,
      "step": 56802,
      "training_step_time": 9.469217777252197
    },
    {
      "epoch": 5.4171562194824216e-05,
      "model_forward_time": 1.8733885288238525,
      "step": 56803
    },
    {
      "epoch": 5.4171562194824216e-05,
      "step": 56803,
      "training_step_time": 8.653927326202393
    },
    {
      "epoch": 5.417251586914063e-05,
      "model_forward_time": 1.7637386322021484,
      "step": 56804
    },
    {
      "epoch": 5.417251586914063e-05,
      "step": 56804,
      "training_step_time": 7.968089580535889
    },
    {
      "epoch": 5.417346954345703e-05,
      "model_forward_time": 1.762274980545044,
      "step": 56805
    },
    {
      "epoch": 5.417346954345703e-05,
      "step": 56805,
      "training_step_time": 8.603517532348633
    },
    {
      "epoch": 5.4174423217773435e-05,
      "model_forward_time": 1.7544779777526855,
      "step": 56806
    },
    {
      "epoch": 5.4174423217773435e-05,
      "step": 56806,
      "training_step_time": 7.92099404335022
    },
    {
      "epoch": 5.4175376892089846e-05,
      "model_forward_time": 1.7592875957489014,
      "step": 56807
    },
    {
      "epoch": 5.4175376892089846e-05,
      "step": 56807,
      "training_step_time": 7.92582631111145
    },
    {
      "epoch": 5.417633056640625e-05,
      "model_forward_time": 1.402801513671875,
      "step": 56808
    },
    {
      "epoch": 5.417633056640625e-05,
      "step": 56808,
      "training_step_time": 8.882323503494263
    },
    {
      "epoch": 5.4177284240722654e-05,
      "model_forward_time": 1.3951585292816162,
      "step": 56809
    },
    {
      "epoch": 5.4177284240722654e-05,
      "step": 56809,
      "training_step_time": 8.737918376922607
    },
    {
      "epoch": 5.4178237915039065e-05,
      "grad_norm": 0.1052132174372673,
      "learning_rate": 4.289928574392313e-06,
      "loss": 0.0563,
      "step": 56810
    },
    {
      "epoch": 5.4178237915039065e-05,
      "model_forward_time": 2.8878607749938965,
      "step": 56810
    },
    {
      "dynamics_loss_avg": 0.059610212594270705,
      "epoch": 5.4178237915039065e-05,
      "step": 56810
    },
    {
      "action_loss_avg": 0.0012999998405575752,
      "epoch": 5.4178237915039065e-05,
      "step": 56810
    },
    {
      "epoch": 5.4178237915039065e-05,
      "step": 56810,
      "training_step_time": 8.429274559020996
    },
    {
      "epoch": 5.417919158935547e-05,
      "model_forward_time": 1.7618277072906494,
      "step": 56811
    },
    {
      "epoch": 5.417919158935547e-05,
      "step": 56811,
      "training_step_time": 8.524580478668213
    },
    {
      "epoch": 5.418014526367187e-05,
      "model_forward_time": 1.8703348636627197,
      "step": 56812
    },
    {
      "epoch": 5.418014526367187e-05,
      "step": 56812,
      "training_step_time": 7.807137966156006
    },
    {
      "epoch": 5.4181098937988284e-05,
      "model_forward_time": 1.758500099182129,
      "step": 56813
    },
    {
      "epoch": 5.4181098937988284e-05,
      "step": 56813,
      "training_step_time": 8.03739595413208
    },
    {
      "epoch": 5.418205261230469e-05,
      "model_forward_time": 1.7572858333587646,
      "step": 56814
    },
    {
      "epoch": 5.418205261230469e-05,
      "step": 56814,
      "training_step_time": 7.892727613449097
    },
    {
      "epoch": 5.418300628662109e-05,
      "model_forward_time": 1.7624762058258057,
      "step": 56815
    },
    {
      "epoch": 5.418300628662109e-05,
      "step": 56815,
      "training_step_time": 7.940052509307861
    },
    {
      "epoch": 5.41839599609375e-05,
      "model_forward_time": 1.7642881870269775,
      "step": 56816
    },
    {
      "epoch": 5.41839599609375e-05,
      "step": 56816,
      "training_step_time": 8.709029197692871
    },
    {
      "epoch": 5.4184913635253906e-05,
      "model_forward_time": 1.7507877349853516,
      "step": 56817
    },
    {
      "epoch": 5.4184913635253906e-05,
      "step": 56817,
      "training_step_time": 7.8262951374053955
    },
    {
      "epoch": 5.418586730957031e-05,
      "model_forward_time": 1.7642230987548828,
      "step": 56818
    },
    {
      "epoch": 5.418586730957031e-05,
      "step": 56818,
      "training_step_time": 8.703319072723389
    },
    {
      "epoch": 5.418682098388672e-05,
      "model_forward_time": 2.9324145317077637,
      "step": 56819
    },
    {
      "epoch": 5.418682098388672e-05,
      "step": 56819,
      "training_step_time": 7.927637100219727
    },
    {
      "epoch": 5.4187774658203125e-05,
      "grad_norm": 0.11604784429073334,
      "learning_rate": 4.288291901994857e-06,
      "loss": 0.0532,
      "step": 56820
    },
    {
      "epoch": 5.4187774658203125e-05,
      "model_forward_time": 1.7650344371795654,
      "step": 56820
    },
    {
      "dynamics_loss_avg": 0.05558628123253584,
      "epoch": 5.4187774658203125e-05,
      "step": 56820
    },
    {
      "action_loss_avg": 0.001864205754827708,
      "epoch": 5.4187774658203125e-05,
      "step": 56820
    },
    {
      "epoch": 5.4187774658203125e-05,
      "step": 56820,
      "training_step_time": 10.158262729644775
    },
    {
      "epoch": 5.418872833251953e-05,
      "model_forward_time": 1.8659837245941162,
      "step": 56821
    },
    {
      "epoch": 5.418872833251953e-05,
      "step": 56821,
      "training_step_time": 7.783308982849121
    },
    {
      "epoch": 5.418968200683594e-05,
      "model_forward_time": 1.7572693824768066,
      "step": 56822
    },
    {
      "epoch": 5.418968200683594e-05,
      "step": 56822,
      "training_step_time": 8.395953178405762
    },
    {
      "epoch": 5.4190635681152344e-05,
      "model_forward_time": 1.7578425407409668,
      "step": 56823
    },
    {
      "epoch": 5.4190635681152344e-05,
      "step": 56823,
      "training_step_time": 8.632794857025146
    },
    {
      "epoch": 5.419158935546875e-05,
      "model_forward_time": 1.752342939376831,
      "step": 56824
    },
    {
      "epoch": 5.419158935546875e-05,
      "step": 56824,
      "training_step_time": 7.817713499069214
    },
    {
      "epoch": 5.419254302978516e-05,
      "model_forward_time": 1.7405104637145996,
      "step": 56825
    },
    {
      "epoch": 5.419254302978516e-05,
      "step": 56825,
      "training_step_time": 7.8528101444244385
    },
    {
      "epoch": 5.419349670410156e-05,
      "model_forward_time": 1.764784574508667,
      "step": 56826
    },
    {
      "epoch": 5.419349670410156e-05,
      "step": 56826,
      "training_step_time": 8.789833784103394
    },
    {
      "epoch": 5.419445037841797e-05,
      "model_forward_time": 1.7247285842895508,
      "step": 56827
    },
    {
      "epoch": 5.419445037841797e-05,
      "step": 56827,
      "training_step_time": 8.964928150177002
    },
    {
      "epoch": 5.419540405273438e-05,
      "model_forward_time": 2.8794541358947754,
      "step": 56828
    },
    {
      "epoch": 5.419540405273438e-05,
      "step": 56828,
      "training_step_time": 9.914363861083984
    },
    {
      "epoch": 5.419635772705078e-05,
      "model_forward_time": 1.7475574016571045,
      "step": 56829
    },
    {
      "epoch": 5.419635772705078e-05,
      "step": 56829,
      "training_step_time": 7.927295684814453
    },
    {
      "epoch": 5.4197311401367186e-05,
      "grad_norm": 0.10528332740068436,
      "learning_rate": 4.286655307428729e-06,
      "loss": 0.0548,
      "step": 56830
    },
    {
      "epoch": 5.4197311401367186e-05,
      "model_forward_time": 1.858842372894287,
      "step": 56830
    },
    {
      "dynamics_loss_avg": 0.04413394159637392,
      "epoch": 5.4197311401367186e-05,
      "step": 56830
    },
    {
      "action_loss_avg": 0.001726885317475535,
      "epoch": 5.4197311401367186e-05,
      "step": 56830
    },
    {
      "epoch": 5.4197311401367186e-05,
      "step": 56830,
      "training_step_time": 8.726828575134277
    },
    {
      "epoch": 5.41982650756836e-05,
      "model_forward_time": 1.7500619888305664,
      "step": 56831
    },
    {
      "epoch": 5.41982650756836e-05,
      "step": 56831,
      "training_step_time": 7.911284446716309
    },
    {
      "epoch": 5.419921875e-05,
      "model_forward_time": 1.7672512531280518,
      "step": 56832
    },
    {
      "epoch": 5.419921875e-05,
      "step": 56832,
      "training_step_time": 7.884946823120117
    },
    {
      "epoch": 5.4200172424316405e-05,
      "model_forward_time": 1.7549378871917725,
      "step": 56833
    },
    {
      "epoch": 5.4200172424316405e-05,
      "step": 56833,
      "training_step_time": 7.942230939865112
    },
    {
      "epoch": 5.4201126098632816e-05,
      "model_forward_time": 1.732407569885254,
      "step": 56834
    },
    {
      "epoch": 5.4201126098632816e-05,
      "step": 56834,
      "training_step_time": 7.900801181793213
    },
    {
      "epoch": 5.420207977294922e-05,
      "model_forward_time": 1.7686755657196045,
      "step": 56835
    },
    {
      "epoch": 5.420207977294922e-05,
      "step": 56835,
      "training_step_time": 7.770678281784058
    },
    {
      "epoch": 5.4203033447265624e-05,
      "model_forward_time": 1.7655925750732422,
      "step": 56836
    },
    {
      "epoch": 5.4203033447265624e-05,
      "step": 56836,
      "training_step_time": 8.569546699523926
    },
    {
      "epoch": 5.4203987121582034e-05,
      "model_forward_time": 2.8262150287628174,
      "step": 56837
    },
    {
      "epoch": 5.4203987121582034e-05,
      "step": 56837,
      "training_step_time": 7.985624313354492
    },
    {
      "epoch": 5.420494079589844e-05,
      "model_forward_time": 1.7461340427398682,
      "step": 56838
    },
    {
      "epoch": 5.420494079589844e-05,
      "step": 56838,
      "training_step_time": 7.864906311035156
    },
    {
      "epoch": 5.420589447021484e-05,
      "model_forward_time": 1.8562531471252441,
      "step": 56839
    },
    {
      "epoch": 5.420589447021484e-05,
      "step": 56839,
      "training_step_time": 8.56687045097351
    },
    {
      "epoch": 5.420684814453125e-05,
      "grad_norm": 0.08956367522478104,
      "learning_rate": 4.285018790872903e-06,
      "loss": 0.0538,
      "step": 56840
    },
    {
      "epoch": 5.420684814453125e-05,
      "model_forward_time": 1.3936166763305664,
      "step": 56840
    },
    {
      "dynamics_loss_avg": 0.0463618490844965,
      "epoch": 5.420684814453125e-05,
      "step": 56840
    },
    {
      "action_loss_avg": 0.0010909164149779827,
      "epoch": 5.420684814453125e-05,
      "step": 56840
    },
    {
      "epoch": 5.420684814453125e-05,
      "step": 56840,
      "training_step_time": 7.811973571777344
    },
    {
      "epoch": 5.420780181884766e-05,
      "model_forward_time": 1.737396001815796,
      "step": 56841
    },
    {
      "epoch": 5.420780181884766e-05,
      "step": 56841,
      "training_step_time": 8.432034492492676
    },
    {
      "epoch": 5.420875549316406e-05,
      "model_forward_time": 1.747892141342163,
      "step": 56842
    },
    {
      "epoch": 5.420875549316406e-05,
      "step": 56842,
      "training_step_time": 7.891505002975464
    },
    {
      "epoch": 5.420970916748047e-05,
      "model_forward_time": 1.747251272201538,
      "step": 56843
    },
    {
      "epoch": 5.420970916748047e-05,
      "step": 56843,
      "training_step_time": 7.859668731689453
    },
    {
      "epoch": 5.4210662841796876e-05,
      "model_forward_time": 1.3959970474243164,
      "step": 56844
    },
    {
      "epoch": 5.4210662841796876e-05,
      "step": 56844,
      "training_step_time": 8.645013332366943
    },
    {
      "epoch": 5.421161651611328e-05,
      "model_forward_time": 1.7403910160064697,
      "step": 56845
    },
    {
      "epoch": 5.421161651611328e-05,
      "step": 56845,
      "training_step_time": 7.845186233520508
    },
    {
      "epoch": 5.4212570190429684e-05,
      "model_forward_time": 1.7370216846466064,
      "step": 56846
    },
    {
      "epoch": 5.4212570190429684e-05,
      "step": 56846,
      "training_step_time": 8.98082447052002
    },
    {
      "epoch": 5.4213523864746095e-05,
      "model_forward_time": 2.892768144607544,
      "step": 56847
    },
    {
      "epoch": 5.4213523864746095e-05,
      "step": 56847,
      "training_step_time": 8.853314399719238
    },
    {
      "epoch": 5.42144775390625e-05,
      "model_forward_time": 1.8644938468933105,
      "step": 56848
    },
    {
      "epoch": 5.42144775390625e-05,
      "step": 56848,
      "training_step_time": 8.735377550125122
    },
    {
      "epoch": 5.42154312133789e-05,
      "model_forward_time": 1.7485301494598389,
      "step": 56849
    },
    {
      "epoch": 5.42154312133789e-05,
      "step": 56849,
      "training_step_time": 8.630721807479858
    },
    {
      "epoch": 5.4216384887695314e-05,
      "grad_norm": 0.08598410338163376,
      "learning_rate": 4.283382352506347e-06,
      "loss": 0.056,
      "step": 56850
    },
    {
      "epoch": 5.4216384887695314e-05,
      "model_forward_time": 1.745882272720337,
      "step": 56850
    },
    {
      "dynamics_loss_avg": 0.050332344323396686,
      "epoch": 5.4216384887695314e-05,
      "step": 56850
    },
    {
      "action_loss_avg": 0.0017605116445338354,
      "epoch": 5.4216384887695314e-05,
      "step": 56850
    },
    {
      "epoch": 5.4216384887695314e-05,
      "step": 56850,
      "training_step_time": 8.467386245727539
    },
    {
      "epoch": 5.421733856201172e-05,
      "model_forward_time": 1.7477681636810303,
      "step": 56851
    },
    {
      "epoch": 5.421733856201172e-05,
      "step": 56851,
      "training_step_time": 8.39090347290039
    },
    {
      "epoch": 5.421829223632812e-05,
      "model_forward_time": 1.7536916732788086,
      "step": 56852
    },
    {
      "epoch": 5.421829223632812e-05,
      "step": 56852,
      "training_step_time": 7.659918308258057
    },
    {
      "epoch": 5.421924591064453e-05,
      "model_forward_time": 1.757258653640747,
      "step": 56853
    },
    {
      "epoch": 5.421924591064453e-05,
      "step": 56853,
      "training_step_time": 7.50443172454834
    },
    {
      "epoch": 5.422019958496094e-05,
      "model_forward_time": 1.7509393692016602,
      "step": 56854
    },
    {
      "epoch": 5.422019958496094e-05,
      "step": 56854,
      "training_step_time": 7.539151906967163
    },
    {
      "epoch": 5.422115325927734e-05,
      "model_forward_time": 1.7620697021484375,
      "step": 56855
    },
    {
      "epoch": 5.422115325927734e-05,
      "step": 56855,
      "training_step_time": 7.820815086364746
    },
    {
      "epoch": 5.422210693359375e-05,
      "model_forward_time": 2.896754741668701,
      "step": 56856
    },
    {
      "epoch": 5.422210693359375e-05,
      "step": 56856,
      "training_step_time": 8.613752365112305
    },
    {
      "epoch": 5.4223060607910156e-05,
      "model_forward_time": 1.8666164875030518,
      "step": 56857
    },
    {
      "epoch": 5.4223060607910156e-05,
      "step": 56857,
      "training_step_time": 8.550713777542114
    },
    {
      "epoch": 5.422401428222656e-05,
      "model_forward_time": 1.755465030670166,
      "step": 56858
    },
    {
      "epoch": 5.422401428222656e-05,
      "step": 56858,
      "training_step_time": 8.08831262588501
    },
    {
      "epoch": 5.422496795654297e-05,
      "model_forward_time": 1.726738691329956,
      "step": 56859
    },
    {
      "epoch": 5.422496795654297e-05,
      "step": 56859,
      "training_step_time": 7.605650901794434
    },
    {
      "epoch": 5.4225921630859374e-05,
      "grad_norm": 0.09301777929067612,
      "learning_rate": 4.281745992508021e-06,
      "loss": 0.0586,
      "step": 56860
    },
    {
      "epoch": 5.4225921630859374e-05,
      "model_forward_time": 1.7445456981658936,
      "step": 56860
    },
    {
      "dynamics_loss_avg": 0.07191207334399223,
      "epoch": 5.4225921630859374e-05,
      "step": 56860
    },
    {
      "action_loss_avg": 0.0020254541770555077,
      "epoch": 5.4225921630859374e-05,
      "step": 56860
    },
    {
      "epoch": 5.4225921630859374e-05,
      "step": 56860,
      "training_step_time": 7.642556190490723
    },
    {
      "epoch": 5.422687530517578e-05,
      "model_forward_time": 1.750868558883667,
      "step": 56861
    },
    {
      "epoch": 5.422687530517578e-05,
      "step": 56861,
      "training_step_time": 7.6969263553619385
    },
    {
      "epoch": 5.422782897949219e-05,
      "model_forward_time": 1.7645270824432373,
      "step": 56862
    },
    {
      "epoch": 5.422782897949219e-05,
      "step": 56862,
      "training_step_time": 7.531618118286133
    },
    {
      "epoch": 5.422878265380859e-05,
      "model_forward_time": 1.7457067966461182,
      "step": 56863
    },
    {
      "epoch": 5.422878265380859e-05,
      "step": 56863,
      "training_step_time": 7.5314905643463135
    },
    {
      "epoch": 5.4229736328125e-05,
      "model_forward_time": 1.758589267730713,
      "step": 56864
    },
    {
      "epoch": 5.4229736328125e-05,
      "step": 56864,
      "training_step_time": 8.12662124633789
    },
    {
      "epoch": 5.423069000244141e-05,
      "model_forward_time": 2.88592529296875,
      "step": 56865
    },
    {
      "epoch": 5.423069000244141e-05,
      "step": 56865,
      "training_step_time": 7.625900030136108
    },
    {
      "epoch": 5.423164367675781e-05,
      "model_forward_time": 1.852903127670288,
      "step": 56866
    },
    {
      "epoch": 5.423164367675781e-05,
      "step": 56866,
      "training_step_time": 7.70259428024292
    },
    {
      "epoch": 5.4232597351074216e-05,
      "model_forward_time": 1.766061544418335,
      "step": 56867
    },
    {
      "epoch": 5.4232597351074216e-05,
      "step": 56867,
      "training_step_time": 8.340368270874023
    },
    {
      "epoch": 5.423355102539063e-05,
      "model_forward_time": 1.7596592903137207,
      "step": 56868
    },
    {
      "epoch": 5.423355102539063e-05,
      "step": 56868,
      "training_step_time": 7.555202960968018
    },
    {
      "epoch": 5.423450469970703e-05,
      "model_forward_time": 1.7407925128936768,
      "step": 56869
    },
    {
      "epoch": 5.423450469970703e-05,
      "step": 56869,
      "training_step_time": 7.671788692474365
    },
    {
      "epoch": 5.4235458374023435e-05,
      "grad_norm": 0.09792248904705048,
      "learning_rate": 4.28010971105687e-06,
      "loss": 0.0602,
      "step": 56870
    },
    {
      "epoch": 5.4235458374023435e-05,
      "model_forward_time": 1.7368597984313965,
      "step": 56870
    },
    {
      "dynamics_loss_avg": 0.05503920903429389,
      "epoch": 5.4235458374023435e-05,
      "step": 56870
    },
    {
      "action_loss_avg": 0.0019907647045329215,
      "epoch": 5.4235458374023435e-05,
      "step": 56870
    },
    {
      "epoch": 5.4235458374023435e-05,
      "step": 56870,
      "training_step_time": 8.538343667984009
    },
    {
      "epoch": 5.4236412048339846e-05,
      "model_forward_time": 1.7642323970794678,
      "step": 56871
    },
    {
      "epoch": 5.4236412048339846e-05,
      "step": 56871,
      "training_step_time": 7.58098030090332
    },
    {
      "epoch": 5.423736572265625e-05,
      "model_forward_time": 1.3897359371185303,
      "step": 56872
    },
    {
      "epoch": 5.423736572265625e-05,
      "step": 56872,
      "training_step_time": 8.045403242111206
    },
    {
      "epoch": 5.4238319396972654e-05,
      "model_forward_time": 1.7553927898406982,
      "step": 56873
    },
    {
      "epoch": 5.4238319396972654e-05,
      "step": 56873,
      "training_step_time": 9.072511911392212
    },
    {
      "epoch": 5.4239273071289065e-05,
      "model_forward_time": 1.7569591999053955,
      "step": 56874
    },
    {
      "epoch": 5.4239273071289065e-05,
      "step": 56874,
      "training_step_time": 7.763933897018433
    },
    {
      "epoch": 5.424022674560547e-05,
      "model_forward_time": 2.976104974746704,
      "step": 56875
    },
    {
      "epoch": 5.424022674560547e-05,
      "step": 56875,
      "training_step_time": 8.7340726852417
    },
    {
      "epoch": 5.424118041992187e-05,
      "model_forward_time": 1.7583911418914795,
      "step": 56876
    },
    {
      "epoch": 5.424118041992187e-05,
      "step": 56876,
      "training_step_time": 7.7133684158325195
    },
    {
      "epoch": 5.4242134094238284e-05,
      "model_forward_time": 1.766564130783081,
      "step": 56877
    },
    {
      "epoch": 5.4242134094238284e-05,
      "step": 56877,
      "training_step_time": 7.711210489273071
    },
    {
      "epoch": 5.424308776855469e-05,
      "model_forward_time": 1.7651143074035645,
      "step": 56878
    },
    {
      "epoch": 5.424308776855469e-05,
      "step": 56878,
      "training_step_time": 7.643948793411255
    },
    {
      "epoch": 5.424404144287109e-05,
      "model_forward_time": 1.7499721050262451,
      "step": 56879
    },
    {
      "epoch": 5.424404144287109e-05,
      "step": 56879,
      "training_step_time": 7.6323723793029785
    },
    {
      "epoch": 5.42449951171875e-05,
      "grad_norm": 0.07830923050642014,
      "learning_rate": 4.278473508331841e-06,
      "loss": 0.0605,
      "step": 56880
    },
    {
      "epoch": 5.42449951171875e-05,
      "model_forward_time": 1.7684433460235596,
      "step": 56880
    },
    {
      "dynamics_loss_avg": 0.05392270646989346,
      "epoch": 5.42449951171875e-05,
      "step": 56880
    },
    {
      "action_loss_avg": 0.0012113850971218199,
      "epoch": 5.42449951171875e-05,
      "step": 56880
    },
    {
      "epoch": 5.42449951171875e-05,
      "step": 56880,
      "training_step_time": 7.591136693954468
    },
    {
      "epoch": 5.4245948791503906e-05,
      "model_forward_time": 1.7793755531311035,
      "step": 56881
    },
    {
      "epoch": 5.4245948791503906e-05,
      "step": 56881,
      "training_step_time": 8.530221223831177
    },
    {
      "epoch": 5.424690246582031e-05,
      "model_forward_time": 1.747866153717041,
      "step": 56882
    },
    {
      "epoch": 5.424690246582031e-05,
      "step": 56882,
      "training_step_time": 7.8490540981292725
    },
    {
      "epoch": 5.424785614013672e-05,
      "model_forward_time": 1.771796464920044,
      "step": 56883
    },
    {
      "epoch": 5.424785614013672e-05,
      "step": 56883,
      "training_step_time": 9.014944553375244
    },
    {
      "epoch": 5.4248809814453125e-05,
      "model_forward_time": 3.0479238033294678,
      "step": 56884
    },
    {
      "epoch": 5.4248809814453125e-05,
      "step": 56884,
      "training_step_time": 7.969625473022461
    },
    {
      "epoch": 5.424976348876953e-05,
      "model_forward_time": 1.750166654586792,
      "step": 56885
    },
    {
      "epoch": 5.424976348876953e-05,
      "step": 56885,
      "training_step_time": 8.56972599029541
    },
    {
      "epoch": 5.425071716308594e-05,
      "model_forward_time": 1.7441391944885254,
      "step": 56886
    },
    {
      "epoch": 5.425071716308594e-05,
      "step": 56886,
      "training_step_time": 7.597887992858887
    },
    {
      "epoch": 5.4251670837402344e-05,
      "model_forward_time": 1.7638168334960938,
      "step": 56887
    },
    {
      "epoch": 5.4251670837402344e-05,
      "step": 56887,
      "training_step_time": 8.456803321838379
    },
    {
      "epoch": 5.425262451171875e-05,
      "model_forward_time": 1.7896392345428467,
      "step": 56888
    },
    {
      "epoch": 5.425262451171875e-05,
      "step": 56888,
      "training_step_time": 7.704317092895508
    },
    {
      "epoch": 5.425357818603516e-05,
      "model_forward_time": 1.768846035003662,
      "step": 56889
    },
    {
      "epoch": 5.425357818603516e-05,
      "step": 56889,
      "training_step_time": 7.644227981567383
    },
    {
      "epoch": 5.425453186035156e-05,
      "grad_norm": 0.1359395831823349,
      "learning_rate": 4.276837384511864e-06,
      "loss": 0.0547,
      "step": 56890
    },
    {
      "epoch": 5.425453186035156e-05,
      "model_forward_time": 1.752610683441162,
      "step": 56890
    },
    {
      "dynamics_loss_avg": 0.058280462957918645,
      "epoch": 5.425453186035156e-05,
      "step": 56890
    },
    {
      "action_loss_avg": 0.0025531202089041473,
      "epoch": 5.425453186035156e-05,
      "step": 56890
    },
    {
      "epoch": 5.425453186035156e-05,
      "step": 56890,
      "training_step_time": 7.561325788497925
    },
    {
      "epoch": 5.425548553466797e-05,
      "model_forward_time": 1.762343168258667,
      "step": 56891
    },
    {
      "epoch": 5.425548553466797e-05,
      "step": 56891,
      "training_step_time": 8.487471580505371
    },
    {
      "epoch": 5.425643920898438e-05,
      "model_forward_time": 1.7550103664398193,
      "step": 56892
    },
    {
      "epoch": 5.425643920898438e-05,
      "step": 56892,
      "training_step_time": 7.954507827758789
    },
    {
      "epoch": 5.425739288330078e-05,
      "model_forward_time": 2.9185545444488525,
      "step": 56893
    },
    {
      "epoch": 5.425739288330078e-05,
      "step": 56893,
      "training_step_time": 8.652039766311646
    },
    {
      "epoch": 5.4258346557617186e-05,
      "model_forward_time": 1.8754656314849854,
      "step": 56894
    },
    {
      "epoch": 5.4258346557617186e-05,
      "step": 56894,
      "training_step_time": 8.593203067779541
    },
    {
      "epoch": 5.42593002319336e-05,
      "model_forward_time": 1.386544942855835,
      "step": 56895
    },
    {
      "epoch": 5.42593002319336e-05,
      "step": 56895,
      "training_step_time": 7.599840402603149
    },
    {
      "epoch": 5.426025390625e-05,
      "model_forward_time": 1.7470946311950684,
      "step": 56896
    },
    {
      "epoch": 5.426025390625e-05,
      "step": 56896,
      "training_step_time": 7.651675462722778
    },
    {
      "epoch": 5.4261207580566405e-05,
      "model_forward_time": 1.3993396759033203,
      "step": 56897
    },
    {
      "epoch": 5.4261207580566405e-05,
      "step": 56897,
      "training_step_time": 7.647645950317383
    },
    {
      "epoch": 5.4262161254882816e-05,
      "model_forward_time": 1.404270887374878,
      "step": 56898
    },
    {
      "epoch": 5.4262161254882816e-05,
      "step": 56898,
      "training_step_time": 7.676129579544067
    },
    {
      "epoch": 5.426311492919922e-05,
      "model_forward_time": 1.7497830390930176,
      "step": 56899
    },
    {
      "epoch": 5.426311492919922e-05,
      "step": 56899,
      "training_step_time": 7.466693639755249
    },
    {
      "epoch": 5.4264068603515624e-05,
      "grad_norm": 0.11050645262002945,
      "learning_rate": 4.2752013397758644e-06,
      "loss": 0.0557,
      "step": 56900
    },
    {
      "epoch": 5.4264068603515624e-05,
      "model_forward_time": 1.7572929859161377,
      "step": 56900
    },
    {
      "dynamics_loss_avg": 0.07618542686104775,
      "epoch": 5.4264068603515624e-05,
      "step": 56900
    },
    {
      "action_loss_avg": 0.001776226394576952,
      "epoch": 5.4264068603515624e-05,
      "step": 56900
    },
    {
      "epoch": 5.4264068603515624e-05,
      "step": 56900,
      "training_step_time": 7.635776042938232
    },
    {
      "epoch": 5.4265022277832034e-05,
      "model_forward_time": 1.7649359703063965,
      "step": 56901
    },
    {
      "epoch": 5.4265022277832034e-05,
      "step": 56901,
      "training_step_time": 8.360298871994019
    },
    {
      "epoch": 5.426597595214844e-05,
      "model_forward_time": 1.770521640777588,
      "step": 56902
    },
    {
      "epoch": 5.426597595214844e-05,
      "step": 56902,
      "training_step_time": 7.676445245742798
    },
    {
      "epoch": 5.426692962646484e-05,
      "model_forward_time": 2.9676055908203125,
      "step": 56903
    },
    {
      "epoch": 5.426692962646484e-05,
      "step": 56903,
      "training_step_time": 7.762583017349243
    },
    {
      "epoch": 5.426788330078125e-05,
      "model_forward_time": 1.7584068775177002,
      "step": 56904
    },
    {
      "epoch": 5.426788330078125e-05,
      "step": 56904,
      "training_step_time": 7.698385953903198
    },
    {
      "epoch": 5.426883697509766e-05,
      "model_forward_time": 1.7586522102355957,
      "step": 56905
    },
    {
      "epoch": 5.426883697509766e-05,
      "step": 56905,
      "training_step_time": 8.168734788894653
    },
    {
      "epoch": 5.426979064941406e-05,
      "model_forward_time": 1.7503764629364014,
      "step": 56906
    },
    {
      "epoch": 5.426979064941406e-05,
      "step": 56906,
      "training_step_time": 7.701785564422607
    },
    {
      "epoch": 5.4270744323730465e-05,
      "model_forward_time": 1.7602007389068604,
      "step": 56907
    },
    {
      "epoch": 5.4270744323730465e-05,
      "step": 56907,
      "training_step_time": 7.596683502197266
    },
    {
      "epoch": 5.4271697998046876e-05,
      "model_forward_time": 1.7643184661865234,
      "step": 56908
    },
    {
      "epoch": 5.4271697998046876e-05,
      "step": 56908,
      "training_step_time": 7.460397481918335
    },
    {
      "epoch": 5.427265167236328e-05,
      "model_forward_time": 1.7551124095916748,
      "step": 56909
    },
    {
      "epoch": 5.427265167236328e-05,
      "step": 56909,
      "training_step_time": 7.862269401550293
    },
    {
      "epoch": 5.4273605346679684e-05,
      "grad_norm": 0.12821675837039948,
      "learning_rate": 4.273565374302756e-06,
      "loss": 0.056,
      "step": 56910
    },
    {
      "epoch": 5.4273605346679684e-05,
      "model_forward_time": 1.7537260055541992,
      "step": 56910
    },
    {
      "dynamics_loss_avg": 0.0572906531393528,
      "epoch": 5.4273605346679684e-05,
      "step": 56910
    },
    {
      "action_loss_avg": 0.0017865483852801845,
      "epoch": 5.4273605346679684e-05,
      "step": 56910
    },
    {
      "epoch": 5.4273605346679684e-05,
      "step": 56910,
      "training_step_time": 7.910407543182373
    },
    {
      "epoch": 5.4274559020996095e-05,
      "model_forward_time": 1.7600023746490479,
      "step": 56911
    },
    {
      "epoch": 5.4274559020996095e-05,
      "step": 56911,
      "training_step_time": 10.188228368759155
    },
    {
      "epoch": 5.42755126953125e-05,
      "model_forward_time": 2.94625186920166,
      "step": 56912
    },
    {
      "epoch": 5.42755126953125e-05,
      "step": 56912,
      "training_step_time": 7.752655982971191
    },
    {
      "epoch": 5.42764663696289e-05,
      "model_forward_time": 1.7474820613861084,
      "step": 56913
    },
    {
      "epoch": 5.42764663696289e-05,
      "step": 56913,
      "training_step_time": 8.769242525100708
    },
    {
      "epoch": 5.4277420043945314e-05,
      "model_forward_time": 1.757997989654541,
      "step": 56914
    },
    {
      "epoch": 5.4277420043945314e-05,
      "step": 56914,
      "training_step_time": 8.395802736282349
    },
    {
      "epoch": 5.427837371826172e-05,
      "model_forward_time": 1.7592339515686035,
      "step": 56915
    },
    {
      "epoch": 5.427837371826172e-05,
      "step": 56915,
      "training_step_time": 8.795667886734009
    },
    {
      "epoch": 5.427932739257812e-05,
      "model_forward_time": 1.7626245021820068,
      "step": 56916
    },
    {
      "epoch": 5.427932739257812e-05,
      "step": 56916,
      "training_step_time": 7.510200500488281
    },
    {
      "epoch": 5.428028106689453e-05,
      "model_forward_time": 1.756328821182251,
      "step": 56917
    },
    {
      "epoch": 5.428028106689453e-05,
      "step": 56917,
      "training_step_time": 8.298419713973999
    },
    {
      "epoch": 5.428123474121094e-05,
      "model_forward_time": 1.7658777236938477,
      "step": 56918
    },
    {
      "epoch": 5.428123474121094e-05,
      "step": 56918,
      "training_step_time": 7.694053649902344
    },
    {
      "epoch": 5.428218841552734e-05,
      "model_forward_time": 1.7543697357177734,
      "step": 56919
    },
    {
      "epoch": 5.428218841552734e-05,
      "step": 56919,
      "training_step_time": 8.349437475204468
    },
    {
      "epoch": 5.428314208984375e-05,
      "grad_norm": 0.09141821414232254,
      "learning_rate": 4.271929488271446e-06,
      "loss": 0.0531,
      "step": 56920
    },
    {
      "epoch": 5.428314208984375e-05,
      "model_forward_time": 1.7449100017547607,
      "step": 56920
    },
    {
      "dynamics_loss_avg": 0.0632227260619402,
      "epoch": 5.428314208984375e-05,
      "step": 56920
    },
    {
      "action_loss_avg": 0.0019083527178736404,
      "epoch": 5.428314208984375e-05,
      "step": 56920
    },
    {
      "epoch": 5.428314208984375e-05,
      "step": 56920,
      "training_step_time": 8.854698419570923
    },
    {
      "epoch": 5.4284095764160156e-05,
      "model_forward_time": 2.9698903560638428,
      "step": 56921
    },
    {
      "epoch": 5.4284095764160156e-05,
      "step": 56921,
      "training_step_time": 8.976609468460083
    },
    {
      "epoch": 5.428504943847656e-05,
      "model_forward_time": 1.7380197048187256,
      "step": 56922
    },
    {
      "epoch": 5.428504943847656e-05,
      "step": 56922,
      "training_step_time": 7.765473365783691
    },
    {
      "epoch": 5.428600311279297e-05,
      "model_forward_time": 1.7798871994018555,
      "step": 56923
    },
    {
      "epoch": 5.428600311279297e-05,
      "step": 56923,
      "training_step_time": 7.649072885513306
    },
    {
      "epoch": 5.4286956787109375e-05,
      "model_forward_time": 1.7633213996887207,
      "step": 56924
    },
    {
      "epoch": 5.4286956787109375e-05,
      "step": 56924,
      "training_step_time": 8.211815118789673
    },
    {
      "epoch": 5.428791046142578e-05,
      "model_forward_time": 1.7620694637298584,
      "step": 56925
    },
    {
      "epoch": 5.428791046142578e-05,
      "step": 56925,
      "training_step_time": 7.610859632492065
    },
    {
      "epoch": 5.428886413574219e-05,
      "model_forward_time": 1.766589879989624,
      "step": 56926
    },
    {
      "epoch": 5.428886413574219e-05,
      "step": 56926,
      "training_step_time": 7.562093734741211
    },
    {
      "epoch": 5.428981781005859e-05,
      "model_forward_time": 1.7493882179260254,
      "step": 56927
    },
    {
      "epoch": 5.428981781005859e-05,
      "step": 56927,
      "training_step_time": 8.624892473220825
    },
    {
      "epoch": 5.4290771484375e-05,
      "model_forward_time": 1.7368648052215576,
      "step": 56928
    },
    {
      "epoch": 5.4290771484375e-05,
      "step": 56928,
      "training_step_time": 8.670205116271973
    },
    {
      "epoch": 5.429172515869141e-05,
      "model_forward_time": 1.7654976844787598,
      "step": 56929
    },
    {
      "epoch": 5.429172515869141e-05,
      "step": 56929,
      "training_step_time": 7.834087610244751
    },
    {
      "epoch": 5.429267883300781e-05,
      "grad_norm": 0.14389212429523468,
      "learning_rate": 4.270293681860831e-06,
      "loss": 0.0549,
      "step": 56930
    },
    {
      "epoch": 5.429267883300781e-05,
      "model_forward_time": 2.9335787296295166,
      "step": 56930
    },
    {
      "dynamics_loss_avg": 0.05553268678486347,
      "epoch": 5.429267883300781e-05,
      "step": 56930
    },
    {
      "action_loss_avg": 0.0017130108608398587,
      "epoch": 5.429267883300781e-05,
      "step": 56930
    },
    {
      "epoch": 5.429267883300781e-05,
      "step": 56930,
      "training_step_time": 7.771838188171387
    },
    {
      "epoch": 5.4293632507324216e-05,
      "model_forward_time": 1.8777217864990234,
      "step": 56931
    },
    {
      "epoch": 5.4293632507324216e-05,
      "step": 56931,
      "training_step_time": 8.673547267913818
    },
    {
      "epoch": 5.429458618164063e-05,
      "model_forward_time": 1.7700152397155762,
      "step": 56932
    },
    {
      "epoch": 5.429458618164063e-05,
      "step": 56932,
      "training_step_time": 7.799489974975586
    },
    {
      "epoch": 5.429553985595703e-05,
      "model_forward_time": 1.7652065753936768,
      "step": 56933
    },
    {
      "epoch": 5.429553985595703e-05,
      "step": 56933,
      "training_step_time": 7.673508167266846
    },
    {
      "epoch": 5.4296493530273435e-05,
      "model_forward_time": 1.7673892974853516,
      "step": 56934
    },
    {
      "epoch": 5.4296493530273435e-05,
      "step": 56934,
      "training_step_time": 7.560890197753906
    },
    {
      "epoch": 5.4297447204589846e-05,
      "model_forward_time": 1.7652268409729004,
      "step": 56935
    },
    {
      "epoch": 5.4297447204589846e-05,
      "step": 56935,
      "training_step_time": 7.594593048095703
    },
    {
      "epoch": 5.429840087890625e-05,
      "model_forward_time": 1.7703895568847656,
      "step": 56936
    },
    {
      "epoch": 5.429840087890625e-05,
      "step": 56936,
      "training_step_time": 7.7741858959198
    },
    {
      "epoch": 5.4299354553222654e-05,
      "model_forward_time": 1.763608455657959,
      "step": 56937
    },
    {
      "epoch": 5.4299354553222654e-05,
      "step": 56937,
      "training_step_time": 7.641754627227783
    },
    {
      "epoch": 5.4300308227539065e-05,
      "model_forward_time": 1.77070951461792,
      "step": 56938
    },
    {
      "epoch": 5.4300308227539065e-05,
      "step": 56938,
      "training_step_time": 8.851540327072144
    },
    {
      "epoch": 5.430126190185547e-05,
      "model_forward_time": 2.876121997833252,
      "step": 56939
    },
    {
      "epoch": 5.430126190185547e-05,
      "step": 56939,
      "training_step_time": 7.696204423904419
    },
    {
      "epoch": 5.430221557617187e-05,
      "grad_norm": 0.10269426554441452,
      "learning_rate": 4.268657955249802e-06,
      "loss": 0.0565,
      "step": 56940
    },
    {
      "epoch": 5.430221557617187e-05,
      "model_forward_time": 1.8679635524749756,
      "step": 56940
    },
    {
      "dynamics_loss_avg": 0.05795628130435944,
      "epoch": 5.430221557617187e-05,
      "step": 56940
    },
    {
      "action_loss_avg": 0.0015002770640421658,
      "epoch": 5.430221557617187e-05,
      "step": 56940
    },
    {
      "epoch": 5.430221557617187e-05,
      "step": 56940,
      "training_step_time": 7.614443302154541
    },
    {
      "epoch": 5.4303169250488284e-05,
      "model_forward_time": 1.7868695259094238,
      "step": 56941
    },
    {
      "epoch": 5.4303169250488284e-05,
      "step": 56941,
      "training_step_time": 7.67693567276001
    },
    {
      "epoch": 5.430412292480469e-05,
      "model_forward_time": 1.762237310409546,
      "step": 56942
    },
    {
      "epoch": 5.430412292480469e-05,
      "step": 56942,
      "training_step_time": 7.7025861740112305
    },
    {
      "epoch": 5.430507659912109e-05,
      "model_forward_time": 1.7618541717529297,
      "step": 56943
    },
    {
      "epoch": 5.430507659912109e-05,
      "step": 56943,
      "training_step_time": 7.652072906494141
    },
    {
      "epoch": 5.43060302734375e-05,
      "model_forward_time": 1.7772574424743652,
      "step": 56944
    },
    {
      "epoch": 5.43060302734375e-05,
      "step": 56944,
      "training_step_time": 7.500509023666382
    },
    {
      "epoch": 5.4306983947753907e-05,
      "model_forward_time": 1.7666804790496826,
      "step": 56945
    },
    {
      "epoch": 5.4306983947753907e-05,
      "step": 56945,
      "training_step_time": 7.7238850593566895
    },
    {
      "epoch": 5.430793762207031e-05,
      "model_forward_time": 1.7685317993164062,
      "step": 56946
    },
    {
      "epoch": 5.430793762207031e-05,
      "step": 56946,
      "training_step_time": 7.585025310516357
    },
    {
      "epoch": 5.430889129638672e-05,
      "model_forward_time": 1.7651340961456299,
      "step": 56947
    },
    {
      "epoch": 5.430889129638672e-05,
      "step": 56947,
      "training_step_time": 8.496050119400024
    },
    {
      "epoch": 5.4309844970703125e-05,
      "model_forward_time": 1.7683062553405762,
      "step": 56948
    },
    {
      "epoch": 5.4309844970703125e-05,
      "step": 56948,
      "training_step_time": 9.33325743675232
    },
    {
      "epoch": 5.431079864501953e-05,
      "model_forward_time": 3.0314388275146484,
      "step": 56949
    },
    {
      "epoch": 5.431079864501953e-05,
      "step": 56949,
      "training_step_time": 8.772660255432129
    },
    {
      "epoch": 5.431175231933594e-05,
      "grad_norm": 0.07368490844964981,
      "learning_rate": 4.267022308617241e-06,
      "loss": 0.0533,
      "step": 56950
    },
    {
      "epoch": 5.431175231933594e-05,
      "model_forward_time": 1.7560861110687256,
      "step": 56950
    },
    {
      "dynamics_loss_avg": 0.06744899991899729,
      "epoch": 5.431175231933594e-05,
      "step": 56950
    },
    {
      "action_loss_avg": 0.0024658118141815065,
      "epoch": 5.431175231933594e-05,
      "step": 56950
    },
    {
      "epoch": 5.431175231933594e-05,
      "step": 56950,
      "training_step_time": 7.621485948562622
    },
    {
      "epoch": 5.4312705993652344e-05,
      "model_forward_time": 1.7643072605133057,
      "step": 56951
    },
    {
      "epoch": 5.4312705993652344e-05,
      "step": 56951,
      "training_step_time": 7.6341233253479
    },
    {
      "epoch": 5.431365966796875e-05,
      "model_forward_time": 1.7897694110870361,
      "step": 56952
    },
    {
      "epoch": 5.431365966796875e-05,
      "step": 56952,
      "training_step_time": 7.676425218582153
    },
    {
      "epoch": 5.431461334228516e-05,
      "model_forward_time": 1.7683591842651367,
      "step": 56953
    },
    {
      "epoch": 5.431461334228516e-05,
      "step": 56953,
      "training_step_time": 7.497723579406738
    },
    {
      "epoch": 5.431556701660156e-05,
      "model_forward_time": 1.7438673973083496,
      "step": 56954
    },
    {
      "epoch": 5.431556701660156e-05,
      "step": 56954,
      "training_step_time": 7.572176456451416
    },
    {
      "epoch": 5.431652069091797e-05,
      "model_forward_time": 1.7617697715759277,
      "step": 56955
    },
    {
      "epoch": 5.431652069091797e-05,
      "step": 56955,
      "training_step_time": 7.679348707199097
    },
    {
      "epoch": 5.431747436523438e-05,
      "model_forward_time": 1.775618314743042,
      "step": 56956
    },
    {
      "epoch": 5.431747436523438e-05,
      "step": 56956,
      "training_step_time": 8.670722723007202
    },
    {
      "epoch": 5.431842803955078e-05,
      "model_forward_time": 1.7628772258758545,
      "step": 56957
    },
    {
      "epoch": 5.431842803955078e-05,
      "step": 56957,
      "training_step_time": 7.656654357910156
    },
    {
      "epoch": 5.4319381713867186e-05,
      "model_forward_time": 2.994814157485962,
      "step": 56958
    },
    {
      "epoch": 5.4319381713867186e-05,
      "step": 56958,
      "training_step_time": 7.741879224777222
    },
    {
      "epoch": 5.43203353881836e-05,
      "model_forward_time": 1.767554521560669,
      "step": 56959
    },
    {
      "epoch": 5.43203353881836e-05,
      "step": 56959,
      "training_step_time": 7.7256059646606445
    },
    {
      "epoch": 5.43212890625e-05,
      "grad_norm": 0.12147349119186401,
      "learning_rate": 4.265386742142018e-06,
      "loss": 0.0562,
      "step": 56960
    },
    {
      "epoch": 5.43212890625e-05,
      "model_forward_time": 1.7423810958862305,
      "step": 56960
    },
    {
      "dynamics_loss_avg": 0.055111030116677284,
      "epoch": 5.43212890625e-05,
      "step": 56960
    },
    {
      "action_loss_avg": 0.002547416964080185,
      "epoch": 5.43212890625e-05,
      "step": 56960
    },
    {
      "epoch": 5.43212890625e-05,
      "step": 56960,
      "training_step_time": 7.761094093322754
    },
    {
      "epoch": 5.4322242736816405e-05,
      "model_forward_time": 1.7759859561920166,
      "step": 56961
    },
    {
      "epoch": 5.4322242736816405e-05,
      "step": 56961,
      "training_step_time": 7.61665153503418
    },
    {
      "epoch": 5.4323196411132816e-05,
      "model_forward_time": 1.7503743171691895,
      "step": 56962
    },
    {
      "epoch": 5.4323196411132816e-05,
      "step": 56962,
      "training_step_time": 8.840606689453125
    },
    {
      "epoch": 5.432415008544922e-05,
      "model_forward_time": 1.7675209045410156,
      "step": 56963
    },
    {
      "epoch": 5.432415008544922e-05,
      "step": 56963,
      "training_step_time": 7.572020053863525
    },
    {
      "epoch": 5.4325103759765624e-05,
      "model_forward_time": 1.7548589706420898,
      "step": 56964
    },
    {
      "epoch": 5.4325103759765624e-05,
      "step": 56964,
      "training_step_time": 7.658318996429443
    },
    {
      "epoch": 5.4326057434082035e-05,
      "model_forward_time": 1.7521061897277832,
      "step": 56965
    },
    {
      "epoch": 5.4326057434082035e-05,
      "step": 56965,
      "training_step_time": 7.536332607269287
    },
    {
      "epoch": 5.432701110839844e-05,
      "model_forward_time": 1.7550292015075684,
      "step": 56966
    },
    {
      "epoch": 5.432701110839844e-05,
      "step": 56966,
      "training_step_time": 8.336877822875977
    },
    {
      "epoch": 5.432796478271484e-05,
      "model_forward_time": 2.924541473388672,
      "step": 56967
    },
    {
      "epoch": 5.432796478271484e-05,
      "step": 56967,
      "training_step_time": 9.186686992645264
    },
    {
      "epoch": 5.432891845703125e-05,
      "model_forward_time": 1.8573527336120605,
      "step": 56968
    },
    {
      "epoch": 5.432891845703125e-05,
      "step": 56968,
      "training_step_time": 8.39460277557373
    },
    {
      "epoch": 5.432987213134766e-05,
      "model_forward_time": 1.7761037349700928,
      "step": 56969
    },
    {
      "epoch": 5.432987213134766e-05,
      "step": 56969,
      "training_step_time": 7.77651047706604
    },
    {
      "epoch": 5.433082580566406e-05,
      "grad_norm": 0.12871496379375458,
      "learning_rate": 4.263751256002996e-06,
      "loss": 0.0546,
      "step": 56970
    },
    {
      "epoch": 5.433082580566406e-05,
      "model_forward_time": 1.742917776107788,
      "step": 56970
    },
    {
      "dynamics_loss_avg": 0.04887573421001434,
      "epoch": 5.433082580566406e-05,
      "step": 56970
    },
    {
      "action_loss_avg": 0.00174848887254484,
      "epoch": 5.433082580566406e-05,
      "step": 56970
    },
    {
      "epoch": 5.433082580566406e-05,
      "step": 56970,
      "training_step_time": 8.299219369888306
    },
    {
      "epoch": 5.4331779479980465e-05,
      "model_forward_time": 1.769397497177124,
      "step": 56971
    },
    {
      "epoch": 5.4331779479980465e-05,
      "step": 56971,
      "training_step_time": 7.595514535903931
    },
    {
      "epoch": 5.4332733154296876e-05,
      "model_forward_time": 1.7601902484893799,
      "step": 56972
    },
    {
      "epoch": 5.4332733154296876e-05,
      "step": 56972,
      "training_step_time": 7.544156789779663
    },
    {
      "epoch": 5.433368682861328e-05,
      "model_forward_time": 1.7656216621398926,
      "step": 56973
    },
    {
      "epoch": 5.433368682861328e-05,
      "step": 56973,
      "training_step_time": 7.787267684936523
    },
    {
      "epoch": 5.4334640502929684e-05,
      "model_forward_time": 1.7560746669769287,
      "step": 56974
    },
    {
      "epoch": 5.4334640502929684e-05,
      "step": 56974,
      "training_step_time": 8.362859964370728
    },
    {
      "epoch": 5.4335594177246095e-05,
      "model_forward_time": 1.7605488300323486,
      "step": 56975
    },
    {
      "epoch": 5.4335594177246095e-05,
      "step": 56975,
      "training_step_time": 8.291897296905518
    },
    {
      "epoch": 5.43365478515625e-05,
      "model_forward_time": 2.9081759452819824,
      "step": 56976
    },
    {
      "epoch": 5.43365478515625e-05,
      "step": 56976,
      "training_step_time": 7.6587233543396
    },
    {
      "epoch": 5.43375015258789e-05,
      "model_forward_time": 1.8576853275299072,
      "step": 56977
    },
    {
      "epoch": 5.43375015258789e-05,
      "step": 56977,
      "training_step_time": 8.53925633430481
    },
    {
      "epoch": 5.4338455200195314e-05,
      "model_forward_time": 1.3918099403381348,
      "step": 56978
    },
    {
      "epoch": 5.4338455200195314e-05,
      "step": 56978,
      "training_step_time": 8.569950342178345
    },
    {
      "epoch": 5.433940887451172e-05,
      "model_forward_time": 1.7465338706970215,
      "step": 56979
    },
    {
      "epoch": 5.433940887451172e-05,
      "step": 56979,
      "training_step_time": 7.6913042068481445
    },
    {
      "epoch": 5.434036254882812e-05,
      "grad_norm": 0.1122710332274437,
      "learning_rate": 4.26211585037903e-06,
      "loss": 0.0517,
      "step": 56980
    },
    {
      "epoch": 5.434036254882812e-05,
      "model_forward_time": 1.7459816932678223,
      "step": 56980
    },
    {
      "dynamics_loss_avg": 0.05764550864696503,
      "epoch": 5.434036254882812e-05,
      "step": 56980
    },
    {
      "action_loss_avg": 0.001651831524213776,
      "epoch": 5.434036254882812e-05,
      "step": 56980
    },
    {
      "epoch": 5.434036254882812e-05,
      "step": 56980,
      "training_step_time": 7.536349773406982
    },
    {
      "epoch": 5.434131622314453e-05,
      "model_forward_time": 1.763279676437378,
      "step": 56981
    },
    {
      "epoch": 5.434131622314453e-05,
      "step": 56981,
      "training_step_time": 7.581458806991577
    },
    {
      "epoch": 5.434226989746094e-05,
      "model_forward_time": 1.7619905471801758,
      "step": 56982
    },
    {
      "epoch": 5.434226989746094e-05,
      "step": 56982,
      "training_step_time": 8.78107500076294
    },
    {
      "epoch": 5.434322357177734e-05,
      "model_forward_time": 1.7598984241485596,
      "step": 56983
    },
    {
      "epoch": 5.434322357177734e-05,
      "step": 56983,
      "training_step_time": 7.967352867126465
    },
    {
      "epoch": 5.434417724609375e-05,
      "model_forward_time": 1.7642123699188232,
      "step": 56984
    },
    {
      "epoch": 5.434417724609375e-05,
      "step": 56984,
      "training_step_time": 9.113089561462402
    },
    {
      "epoch": 5.4345130920410156e-05,
      "model_forward_time": 2.8443119525909424,
      "step": 56985
    },
    {
      "epoch": 5.4345130920410156e-05,
      "step": 56985,
      "training_step_time": 8.652682781219482
    },
    {
      "epoch": 5.434608459472656e-05,
      "model_forward_time": 1.8285107612609863,
      "step": 56986
    },
    {
      "epoch": 5.434608459472656e-05,
      "step": 56986,
      "training_step_time": 8.653257846832275
    },
    {
      "epoch": 5.434703826904297e-05,
      "model_forward_time": 1.7574527263641357,
      "step": 56987
    },
    {
      "epoch": 5.434703826904297e-05,
      "step": 56987,
      "training_step_time": 8.30985164642334
    },
    {
      "epoch": 5.4347991943359375e-05,
      "model_forward_time": 1.7586274147033691,
      "step": 56988
    },
    {
      "epoch": 5.4347991943359375e-05,
      "step": 56988,
      "training_step_time": 7.652150630950928
    },
    {
      "epoch": 5.434894561767578e-05,
      "model_forward_time": 1.7682418823242188,
      "step": 56989
    },
    {
      "epoch": 5.434894561767578e-05,
      "step": 56989,
      "training_step_time": 7.599541664123535
    },
    {
      "epoch": 5.434989929199219e-05,
      "grad_norm": 0.09577743709087372,
      "learning_rate": 4.260480525448965e-06,
      "loss": 0.0575,
      "step": 56990
    },
    {
      "epoch": 5.434989929199219e-05,
      "model_forward_time": 1.7618696689605713,
      "step": 56990
    },
    {
      "dynamics_loss_avg": 0.06116467602550983,
      "epoch": 5.434989929199219e-05,
      "step": 56990
    },
    {
      "action_loss_avg": 0.001765534924925305,
      "epoch": 5.434989929199219e-05,
      "step": 56990
    },
    {
      "epoch": 5.434989929199219e-05,
      "step": 56990,
      "training_step_time": 7.566826820373535
    },
    {
      "epoch": 5.4350852966308593e-05,
      "model_forward_time": 1.7638068199157715,
      "step": 56991
    },
    {
      "epoch": 5.4350852966308593e-05,
      "step": 56991,
      "training_step_time": 7.7626636028289795
    },
    {
      "epoch": 5.4351806640625e-05,
      "model_forward_time": 1.7662718296051025,
      "step": 56992
    },
    {
      "epoch": 5.4351806640625e-05,
      "step": 56992,
      "training_step_time": 7.499910116195679
    },
    {
      "epoch": 5.435276031494141e-05,
      "model_forward_time": 1.760873794555664,
      "step": 56993
    },
    {
      "epoch": 5.435276031494141e-05,
      "step": 56993,
      "training_step_time": 7.792071342468262
    },
    {
      "epoch": 5.435371398925781e-05,
      "model_forward_time": 1.7550296783447266,
      "step": 56994
    },
    {
      "epoch": 5.435371398925781e-05,
      "step": 56994,
      "training_step_time": 9.10923981666565
    },
    {
      "epoch": 5.4354667663574216e-05,
      "model_forward_time": 3.0432112216949463,
      "step": 56995
    },
    {
      "epoch": 5.4354667663574216e-05,
      "step": 56995,
      "training_step_time": 7.754916429519653
    },
    {
      "epoch": 5.435562133789063e-05,
      "model_forward_time": 1.7685308456420898,
      "step": 56996
    },
    {
      "epoch": 5.435562133789063e-05,
      "step": 56996,
      "training_step_time": 8.500844478607178
    },
    {
      "epoch": 5.435657501220703e-05,
      "model_forward_time": 1.7624690532684326,
      "step": 56997
    },
    {
      "epoch": 5.435657501220703e-05,
      "step": 56997,
      "training_step_time": 8.810328960418701
    },
    {
      "epoch": 5.4357528686523435e-05,
      "model_forward_time": 1.7769782543182373,
      "step": 56998
    },
    {
      "epoch": 5.4357528686523435e-05,
      "step": 56998,
      "training_step_time": 8.896735191345215
    },
    {
      "epoch": 5.4358482360839846e-05,
      "model_forward_time": 1.7699787616729736,
      "step": 56999
    },
    {
      "epoch": 5.4358482360839846e-05,
      "step": 56999,
      "training_step_time": 7.598890542984009
    },
    {
      "epoch": 5.435943603515625e-05,
      "grad_norm": 0.09563866257667542,
      "learning_rate": 4.258845281391638e-06,
      "loss": 0.0543,
      "step": 57000
    },
    {
      "epoch": 5.435943603515625e-05,
      "model_forward_time": 1.8065950870513916,
      "step": 57000
    },
    {
      "dynamics_loss_avg": 0.06106234937906265,
      "epoch": 5.435943603515625e-05,
      "step": 57000
    },
    {
      "action_loss_avg": 0.0012483057391364128,
      "epoch": 5.435943603515625e-05,
      "step": 57000
    },
    {
      "epoch": 5.435943603515625e-05,
      "step": 57000,
      "training_step_time": 7.148311376571655
    },
    {
      "epoch": 5.4360389709472654e-05,
      "model_forward_time": 1.767467737197876,
      "step": 57001
    },
    {
      "epoch": 5.4360389709472654e-05,
      "step": 57001,
      "training_step_time": 8.513654947280884
    },
    {
      "epoch": 5.4361343383789065e-05,
      "model_forward_time": 1.777796745300293,
      "step": 57002
    },
    {
      "epoch": 5.4361343383789065e-05,
      "step": 57002,
      "training_step_time": 8.64621090888977
    },
    {
      "epoch": 5.436229705810547e-05,
      "model_forward_time": 3.459897518157959,
      "step": 57003
    },
    {
      "epoch": 5.436229705810547e-05,
      "step": 57003,
      "training_step_time": 8.926113367080688
    },
    {
      "epoch": 5.436325073242187e-05,
      "model_forward_time": 1.7987878322601318,
      "step": 57004
    },
    {
      "epoch": 5.436325073242187e-05,
      "step": 57004,
      "training_step_time": 9.687600374221802
    },
    {
      "epoch": 5.4364204406738284e-05,
      "model_forward_time": 1.7797892093658447,
      "step": 57005
    },
    {
      "epoch": 5.4364204406738284e-05,
      "step": 57005,
      "training_step_time": 8.406635761260986
    },
    {
      "epoch": 5.436515808105469e-05,
      "model_forward_time": 1.7276239395141602,
      "step": 57006
    },
    {
      "epoch": 5.436515808105469e-05,
      "step": 57006,
      "training_step_time": 11.151252746582031
    },
    {
      "epoch": 5.436611175537109e-05,
      "model_forward_time": 1.7540860176086426,
      "step": 57007
    },
    {
      "epoch": 5.436611175537109e-05,
      "step": 57007,
      "training_step_time": 10.22445034980774
    },
    {
      "epoch": 5.43670654296875e-05,
      "model_forward_time": 1.765268087387085,
      "step": 57008
    },
    {
      "epoch": 5.43670654296875e-05,
      "step": 57008,
      "training_step_time": 8.64426064491272
    },
    {
      "epoch": 5.4368019104003907e-05,
      "model_forward_time": 1.7576158046722412,
      "step": 57009
    },
    {
      "epoch": 5.4368019104003907e-05,
      "step": 57009,
      "training_step_time": 7.653435945510864
    },
    {
      "epoch": 5.436897277832031e-05,
      "grad_norm": 0.09580344706773758,
      "learning_rate": 4.257210118385876e-06,
      "loss": 0.0544,
      "step": 57010
    },
    {
      "epoch": 5.436897277832031e-05,
      "model_forward_time": 1.7583796977996826,
      "step": 57010
    },
    {
      "dynamics_loss_avg": 0.07212508786469699,
      "epoch": 5.436897277832031e-05,
      "step": 57010
    },
    {
      "action_loss_avg": 0.002123381005367264,
      "epoch": 5.436897277832031e-05,
      "step": 57010
    },
    {
      "epoch": 5.436897277832031e-05,
      "step": 57010,
      "training_step_time": 7.794878721237183
    },
    {
      "epoch": 5.436992645263672e-05,
      "model_forward_time": 1.750061273574829,
      "step": 57011
    },
    {
      "epoch": 5.436992645263672e-05,
      "step": 57011,
      "training_step_time": 8.775578022003174
    },
    {
      "epoch": 5.4370880126953125e-05,
      "model_forward_time": 2.866377830505371,
      "step": 57012
    },
    {
      "epoch": 5.4370880126953125e-05,
      "step": 57012,
      "training_step_time": 8.715111494064331
    },
    {
      "epoch": 5.437183380126953e-05,
      "model_forward_time": 1.3963954448699951,
      "step": 57013
    },
    {
      "epoch": 5.437183380126953e-05,
      "step": 57013,
      "training_step_time": 9.286845207214355
    },
    {
      "epoch": 5.437278747558594e-05,
      "model_forward_time": 1.8419737815856934,
      "step": 57014
    },
    {
      "epoch": 5.437278747558594e-05,
      "step": 57014,
      "training_step_time": 8.255252361297607
    },
    {
      "epoch": 5.4373741149902344e-05,
      "model_forward_time": 1.759019374847412,
      "step": 57015
    },
    {
      "epoch": 5.4373741149902344e-05,
      "step": 57015,
      "training_step_time": 8.234999418258667
    },
    {
      "epoch": 5.437469482421875e-05,
      "model_forward_time": 1.7591161727905273,
      "step": 57016
    },
    {
      "epoch": 5.437469482421875e-05,
      "step": 57016,
      "training_step_time": 9.26997423171997
    },
    {
      "epoch": 5.437564849853516e-05,
      "model_forward_time": 1.7595248222351074,
      "step": 57017
    },
    {
      "epoch": 5.437564849853516e-05,
      "step": 57017,
      "training_step_time": 7.642865896224976
    },
    {
      "epoch": 5.437660217285156e-05,
      "model_forward_time": 1.7429327964782715,
      "step": 57018
    },
    {
      "epoch": 5.437660217285156e-05,
      "step": 57018,
      "training_step_time": 7.618348598480225
    },
    {
      "epoch": 5.437755584716797e-05,
      "model_forward_time": 1.754180669784546,
      "step": 57019
    },
    {
      "epoch": 5.437755584716797e-05,
      "step": 57019,
      "training_step_time": 7.745666980743408
    },
    {
      "epoch": 5.437850952148438e-05,
      "grad_norm": 0.11087588965892792,
      "learning_rate": 4.2555750366105e-06,
      "loss": 0.0549,
      "step": 57020
    },
    {
      "epoch": 5.437850952148438e-05,
      "model_forward_time": 1.7540197372436523,
      "step": 57020
    },
    {
      "dynamics_loss_avg": 0.06280623190104961,
      "epoch": 5.437850952148438e-05,
      "step": 57020
    },
    {
      "action_loss_avg": 0.0022303837817162276,
      "epoch": 5.437850952148438e-05,
      "step": 57020
    },
    {
      "epoch": 5.437850952148438e-05,
      "step": 57020,
      "training_step_time": 8.71215558052063
    },
    {
      "epoch": 5.437946319580078e-05,
      "model_forward_time": 1.7602481842041016,
      "step": 57021
    },
    {
      "epoch": 5.437946319580078e-05,
      "step": 57021,
      "training_step_time": 7.777004957199097
    },
    {
      "epoch": 5.4380416870117186e-05,
      "model_forward_time": 3.000180721282959,
      "step": 57022
    },
    {
      "epoch": 5.4380416870117186e-05,
      "step": 57022,
      "training_step_time": 7.7317070960998535
    },
    {
      "epoch": 5.43813705444336e-05,
      "model_forward_time": 1.8490145206451416,
      "step": 57023
    },
    {
      "epoch": 5.43813705444336e-05,
      "step": 57023,
      "training_step_time": 7.55883002281189
    },
    {
      "epoch": 5.438232421875e-05,
      "model_forward_time": 1.7589538097381592,
      "step": 57024
    },
    {
      "epoch": 5.438232421875e-05,
      "step": 57024,
      "training_step_time": 9.506891012191772
    },
    {
      "epoch": 5.4383277893066405e-05,
      "model_forward_time": 1.7648310661315918,
      "step": 57025
    },
    {
      "epoch": 5.4383277893066405e-05,
      "step": 57025,
      "training_step_time": 8.226017951965332
    },
    {
      "epoch": 5.4384231567382816e-05,
      "model_forward_time": 1.7522432804107666,
      "step": 57026
    },
    {
      "epoch": 5.4384231567382816e-05,
      "step": 57026,
      "training_step_time": 7.677396297454834
    },
    {
      "epoch": 5.438518524169922e-05,
      "model_forward_time": 1.7561054229736328,
      "step": 57027
    },
    {
      "epoch": 5.438518524169922e-05,
      "step": 57027,
      "training_step_time": 8.801967144012451
    },
    {
      "epoch": 5.4386138916015624e-05,
      "model_forward_time": 1.7664573192596436,
      "step": 57028
    },
    {
      "epoch": 5.4386138916015624e-05,
      "step": 57028,
      "training_step_time": 9.742263078689575
    },
    {
      "epoch": 5.4387092590332035e-05,
      "model_forward_time": 1.7741081714630127,
      "step": 57029
    },
    {
      "epoch": 5.4387092590332035e-05,
      "step": 57029,
      "training_step_time": 6.853145360946655
    },
    {
      "epoch": 5.438804626464844e-05,
      "grad_norm": 0.10368124395608902,
      "learning_rate": 4.253940036244318e-06,
      "loss": 0.0575,
      "step": 57030
    },
    {
      "epoch": 5.438804626464844e-05,
      "model_forward_time": 1.766681432723999,
      "step": 57030
    },
    {
      "dynamics_loss_avg": 0.061847968772053716,
      "epoch": 5.438804626464844e-05,
      "step": 57030
    },
    {
      "action_loss_avg": 0.001352103688986972,
      "epoch": 5.438804626464844e-05,
      "step": 57030
    },
    {
      "epoch": 5.438804626464844e-05,
      "step": 57030,
      "training_step_time": 6.850092887878418
    },
    {
      "epoch": 5.438899993896484e-05,
      "model_forward_time": 1.7510309219360352,
      "step": 57031
    },
    {
      "epoch": 5.438899993896484e-05,
      "step": 57031,
      "training_step_time": 6.8540496826171875
    },
    {
      "epoch": 5.438995361328125e-05,
      "model_forward_time": 1.7523276805877686,
      "step": 57032
    },
    {
      "epoch": 5.438995361328125e-05,
      "step": 57032,
      "training_step_time": 6.882043838500977
    },
    {
      "epoch": 5.439090728759766e-05,
      "model_forward_time": 1.7470519542694092,
      "step": 57033
    },
    {
      "epoch": 5.439090728759766e-05,
      "step": 57033,
      "training_step_time": 6.909988880157471
    },
    {
      "epoch": 5.439186096191406e-05,
      "model_forward_time": 1.7711617946624756,
      "step": 57034
    },
    {
      "epoch": 5.439186096191406e-05,
      "step": 57034,
      "training_step_time": 6.888179063796997
    },
    {
      "epoch": 5.4392814636230465e-05,
      "model_forward_time": 1.7713277339935303,
      "step": 57035
    },
    {
      "epoch": 5.4392814636230465e-05,
      "step": 57035,
      "training_step_time": 6.901689529418945
    },
    {
      "epoch": 5.4393768310546876e-05,
      "model_forward_time": 1.765312671661377,
      "step": 57036
    },
    {
      "epoch": 5.4393768310546876e-05,
      "step": 57036,
      "training_step_time": 6.870296001434326
    },
    {
      "epoch": 5.439472198486328e-05,
      "model_forward_time": 2.87451171875,
      "step": 57037
    },
    {
      "epoch": 5.439472198486328e-05,
      "step": 57037,
      "training_step_time": 7.743476152420044
    },
    {
      "epoch": 5.4395675659179684e-05,
      "model_forward_time": 1.7604563236236572,
      "step": 57038
    },
    {
      "epoch": 5.4395675659179684e-05,
      "step": 57038,
      "training_step_time": 8.70436143875122
    },
    {
      "epoch": 5.4396629333496095e-05,
      "model_forward_time": 1.731513261795044,
      "step": 57039
    },
    {
      "epoch": 5.4396629333496095e-05,
      "step": 57039,
      "training_step_time": 6.848752021789551
    },
    {
      "epoch": 5.43975830078125e-05,
      "grad_norm": 0.14855068922042847,
      "learning_rate": 4.252305117466133e-06,
      "loss": 0.0557,
      "step": 57040
    },
    {
      "epoch": 5.43975830078125e-05,
      "model_forward_time": 1.772214651107788,
      "step": 57040
    },
    {
      "dynamics_loss_avg": 0.055416247248649596,
      "epoch": 5.43975830078125e-05,
      "step": 57040
    },
    {
      "action_loss_avg": 0.001400115608703345,
      "epoch": 5.43975830078125e-05,
      "step": 57040
    },
    {
      "epoch": 5.43975830078125e-05,
      "step": 57040,
      "training_step_time": 6.848996162414551
    },
    {
      "epoch": 5.43985366821289e-05,
      "model_forward_time": 1.7489862442016602,
      "step": 57041
    },
    {
      "epoch": 5.43985366821289e-05,
      "step": 57041,
      "training_step_time": 6.846296072006226
    },
    {
      "epoch": 5.4399490356445314e-05,
      "model_forward_time": 1.757068395614624,
      "step": 57042
    },
    {
      "epoch": 5.4399490356445314e-05,
      "step": 57042,
      "training_step_time": 6.822686433792114
    },
    {
      "epoch": 5.440044403076172e-05,
      "model_forward_time": 1.7906696796417236,
      "step": 57043
    },
    {
      "epoch": 5.440044403076172e-05,
      "step": 57043,
      "training_step_time": 6.868987560272217
    },
    {
      "epoch": 5.440139770507812e-05,
      "model_forward_time": 1.767155647277832,
      "step": 57044
    },
    {
      "epoch": 5.440139770507812e-05,
      "step": 57044,
      "training_step_time": 6.860013484954834
    },
    {
      "epoch": 5.440235137939453e-05,
      "model_forward_time": 1.7653493881225586,
      "step": 57045
    },
    {
      "epoch": 5.440235137939453e-05,
      "step": 57045,
      "training_step_time": 6.853270530700684
    },
    {
      "epoch": 5.440330505371094e-05,
      "model_forward_time": 2.884697675704956,
      "step": 57046
    },
    {
      "epoch": 5.440330505371094e-05,
      "step": 57046,
      "training_step_time": 7.739062070846558
    },
    {
      "epoch": 5.440425872802734e-05,
      "model_forward_time": 1.760911226272583,
      "step": 57047
    },
    {
      "epoch": 5.440425872802734e-05,
      "step": 57047,
      "training_step_time": 7.878448963165283
    },
    {
      "epoch": 5.440521240234375e-05,
      "model_forward_time": 1.770277738571167,
      "step": 57048
    },
    {
      "epoch": 5.440521240234375e-05,
      "step": 57048,
      "training_step_time": 7.666353940963745
    },
    {
      "epoch": 5.4406166076660156e-05,
      "model_forward_time": 1.7566425800323486,
      "step": 57049
    },
    {
      "epoch": 5.4406166076660156e-05,
      "step": 57049,
      "training_step_time": 6.833784103393555
    },
    {
      "epoch": 5.440711975097656e-05,
      "grad_norm": 0.09970200061798096,
      "learning_rate": 4.250670280454736e-06,
      "loss": 0.0579,
      "step": 57050
    },
    {
      "epoch": 5.440711975097656e-05,
      "model_forward_time": 1.757796049118042,
      "step": 57050
    },
    {
      "dynamics_loss_avg": 0.051623398438096046,
      "epoch": 5.440711975097656e-05,
      "step": 57050
    },
    {
      "action_loss_avg": 0.0014817960618529469,
      "epoch": 5.440711975097656e-05,
      "step": 57050
    },
    {
      "epoch": 5.440711975097656e-05,
      "step": 57050,
      "training_step_time": 6.849315166473389
    },
    {
      "epoch": 5.440807342529297e-05,
      "model_forward_time": 1.7599952220916748,
      "step": 57051
    },
    {
      "epoch": 5.440807342529297e-05,
      "step": 57051,
      "training_step_time": 6.847146034240723
    },
    {
      "epoch": 5.4409027099609375e-05,
      "model_forward_time": 1.770488977432251,
      "step": 57052
    },
    {
      "epoch": 5.4409027099609375e-05,
      "step": 57052,
      "training_step_time": 6.851966381072998
    },
    {
      "epoch": 5.440998077392578e-05,
      "model_forward_time": 1.7643537521362305,
      "step": 57053
    },
    {
      "epoch": 5.440998077392578e-05,
      "step": 57053,
      "training_step_time": 6.878640174865723
    },
    {
      "epoch": 5.441093444824219e-05,
      "model_forward_time": 1.7613072395324707,
      "step": 57054
    },
    {
      "epoch": 5.441093444824219e-05,
      "step": 57054,
      "training_step_time": 6.850572109222412
    },
    {
      "epoch": 5.4411888122558593e-05,
      "model_forward_time": 2.91666316986084,
      "step": 57055
    },
    {
      "epoch": 5.4411888122558593e-05,
      "step": 57055,
      "training_step_time": 7.763992786407471
    },
    {
      "epoch": 5.4412841796875e-05,
      "model_forward_time": 1.7681608200073242,
      "step": 57056
    },
    {
      "epoch": 5.4412841796875e-05,
      "step": 57056,
      "training_step_time": 7.9063560962677
    },
    {
      "epoch": 5.441379547119141e-05,
      "model_forward_time": 1.7767863273620605,
      "step": 57057
    },
    {
      "epoch": 5.441379547119141e-05,
      "step": 57057,
      "training_step_time": 7.685410261154175
    },
    {
      "epoch": 5.441474914550781e-05,
      "model_forward_time": 1.7600882053375244,
      "step": 57058
    },
    {
      "epoch": 5.441474914550781e-05,
      "step": 57058,
      "training_step_time": 6.7807416915893555
    },
    {
      "epoch": 5.4415702819824216e-05,
      "model_forward_time": 1.7610182762145996,
      "step": 57059
    },
    {
      "epoch": 5.4415702819824216e-05,
      "step": 57059,
      "training_step_time": 6.82741379737854
    },
    {
      "epoch": 5.441665649414063e-05,
      "grad_norm": 0.13205385208129883,
      "learning_rate": 4.249035525388908e-06,
      "loss": 0.0577,
      "step": 57060
    },
    {
      "epoch": 5.441665649414063e-05,
      "model_forward_time": 1.76409912109375,
      "step": 57060
    },
    {
      "dynamics_loss_avg": 0.04690532684326172,
      "epoch": 5.441665649414063e-05,
      "step": 57060
    },
    {
      "action_loss_avg": 0.0013614119408885017,
      "epoch": 5.441665649414063e-05,
      "step": 57060
    },
    {
      "epoch": 5.441665649414063e-05,
      "step": 57060,
      "training_step_time": 6.8755784034729
    },
    {
      "epoch": 5.441761016845703e-05,
      "model_forward_time": 1.7679378986358643,
      "step": 57061
    },
    {
      "epoch": 5.441761016845703e-05,
      "step": 57061,
      "training_step_time": 6.916862487792969
    },
    {
      "epoch": 5.4418563842773435e-05,
      "model_forward_time": 1.7626426219940186,
      "step": 57062
    },
    {
      "epoch": 5.4418563842773435e-05,
      "step": 57062,
      "training_step_time": 6.88987135887146
    },
    {
      "epoch": 5.4419517517089846e-05,
      "model_forward_time": 1.7189714908599854,
      "step": 57063
    },
    {
      "epoch": 5.4419517517089846e-05,
      "step": 57063,
      "training_step_time": 6.900296211242676
    },
    {
      "epoch": 5.442047119140625e-05,
      "model_forward_time": 2.88142728805542,
      "step": 57064
    },
    {
      "epoch": 5.442047119140625e-05,
      "step": 57064,
      "training_step_time": 7.745199203491211
    },
    {
      "epoch": 5.4421424865722654e-05,
      "model_forward_time": 1.3848659992218018,
      "step": 57065
    },
    {
      "epoch": 5.4421424865722654e-05,
      "step": 57065,
      "training_step_time": 8.691102266311646
    },
    {
      "epoch": 5.4422378540039065e-05,
      "model_forward_time": 1.7514467239379883,
      "step": 57066
    },
    {
      "epoch": 5.4422378540039065e-05,
      "step": 57066,
      "training_step_time": 7.696516752243042
    },
    {
      "epoch": 5.442333221435547e-05,
      "model_forward_time": 1.7406425476074219,
      "step": 57067
    },
    {
      "epoch": 5.442333221435547e-05,
      "step": 57067,
      "training_step_time": 6.874773740768433
    },
    {
      "epoch": 5.442428588867187e-05,
      "model_forward_time": 1.7578651905059814,
      "step": 57068
    },
    {
      "epoch": 5.442428588867187e-05,
      "step": 57068,
      "training_step_time": 6.8635663986206055
    },
    {
      "epoch": 5.4425239562988284e-05,
      "model_forward_time": 1.7626044750213623,
      "step": 57069
    },
    {
      "epoch": 5.4425239562988284e-05,
      "step": 57069,
      "training_step_time": 6.890383958816528
    },
    {
      "epoch": 5.442619323730469e-05,
      "grad_norm": 0.08716484904289246,
      "learning_rate": 4.247400852447428e-06,
      "loss": 0.0562,
      "step": 57070
    },
    {
      "epoch": 5.442619323730469e-05,
      "model_forward_time": 1.7383506298065186,
      "step": 57070
    },
    {
      "dynamics_loss_avg": 0.056377469841390845,
      "epoch": 5.442619323730469e-05,
      "step": 57070
    },
    {
      "action_loss_avg": 0.002272668675868772,
      "epoch": 5.442619323730469e-05,
      "step": 57070
    },
    {
      "epoch": 5.442619323730469e-05,
      "step": 57070,
      "training_step_time": 6.903524160385132
    },
    {
      "epoch": 5.442714691162109e-05,
      "model_forward_time": 1.7444195747375488,
      "step": 57071
    },
    {
      "epoch": 5.442714691162109e-05,
      "step": 57071,
      "training_step_time": 6.877478361129761
    },
    {
      "epoch": 5.44281005859375e-05,
      "model_forward_time": 1.7475576400756836,
      "step": 57072
    },
    {
      "epoch": 5.44281005859375e-05,
      "step": 57072,
      "training_step_time": 6.848025321960449
    },
    {
      "epoch": 5.442905426025391e-05,
      "model_forward_time": 2.903815507888794,
      "step": 57073
    },
    {
      "epoch": 5.442905426025391e-05,
      "step": 57073,
      "training_step_time": 7.793394088745117
    },
    {
      "epoch": 5.443000793457031e-05,
      "model_forward_time": 1.749544620513916,
      "step": 57074
    },
    {
      "epoch": 5.443000793457031e-05,
      "step": 57074,
      "training_step_time": 8.40494418144226
    },
    {
      "epoch": 5.443096160888672e-05,
      "model_forward_time": 1.7326781749725342,
      "step": 57075
    },
    {
      "epoch": 5.443096160888672e-05,
      "step": 57075,
      "training_step_time": 7.774874687194824
    },
    {
      "epoch": 5.4431915283203125e-05,
      "model_forward_time": 1.7569341659545898,
      "step": 57076
    },
    {
      "epoch": 5.4431915283203125e-05,
      "step": 57076,
      "training_step_time": 7.13128662109375
    },
    {
      "epoch": 5.443286895751953e-05,
      "model_forward_time": 1.7628514766693115,
      "step": 57077
    },
    {
      "epoch": 5.443286895751953e-05,
      "step": 57077,
      "training_step_time": 6.8513288497924805
    },
    {
      "epoch": 5.443382263183594e-05,
      "model_forward_time": 1.7674336433410645,
      "step": 57078
    },
    {
      "epoch": 5.443382263183594e-05,
      "step": 57078,
      "training_step_time": 6.8486058712005615
    },
    {
      "epoch": 5.4434776306152344e-05,
      "model_forward_time": 1.766902208328247,
      "step": 57079
    },
    {
      "epoch": 5.4434776306152344e-05,
      "step": 57079,
      "training_step_time": 6.877506256103516
    },
    {
      "epoch": 5.443572998046875e-05,
      "grad_norm": 0.1298864334821701,
      "learning_rate": 4.245766261809059e-06,
      "loss": 0.0571,
      "step": 57080
    },
    {
      "epoch": 5.443572998046875e-05,
      "model_forward_time": 1.7393534183502197,
      "step": 57080
    },
    {
      "dynamics_loss_avg": 0.0418202986009419,
      "epoch": 5.443572998046875e-05,
      "step": 57080
    },
    {
      "action_loss_avg": 0.0012022038950817659,
      "epoch": 5.443572998046875e-05,
      "step": 57080
    },
    {
      "epoch": 5.443572998046875e-05,
      "step": 57080,
      "training_step_time": 6.871392250061035
    },
    {
      "epoch": 5.443668365478516e-05,
      "model_forward_time": 1.7487831115722656,
      "step": 57081
    },
    {
      "epoch": 5.443668365478516e-05,
      "step": 57081,
      "training_step_time": 6.889941930770874
    },
    {
      "epoch": 5.443763732910156e-05,
      "model_forward_time": 1.7427587509155273,
      "step": 57082
    },
    {
      "epoch": 5.443763732910156e-05,
      "step": 57082,
      "training_step_time": 7.825670003890991
    },
    {
      "epoch": 5.443859100341797e-05,
      "model_forward_time": 2.919029951095581,
      "step": 57083
    },
    {
      "epoch": 5.443859100341797e-05,
      "step": 57083,
      "training_step_time": 8.508361577987671
    },
    {
      "epoch": 5.443954467773438e-05,
      "model_forward_time": 1.7396347522735596,
      "step": 57084
    },
    {
      "epoch": 5.443954467773438e-05,
      "step": 57084,
      "training_step_time": 7.871265649795532
    },
    {
      "epoch": 5.444049835205078e-05,
      "model_forward_time": 1.7413368225097656,
      "step": 57085
    },
    {
      "epoch": 5.444049835205078e-05,
      "step": 57085,
      "training_step_time": 7.448060989379883
    },
    {
      "epoch": 5.4441452026367186e-05,
      "model_forward_time": 1.74210786819458,
      "step": 57086
    },
    {
      "epoch": 5.4441452026367186e-05,
      "step": 57086,
      "training_step_time": 6.837386846542358
    },
    {
      "epoch": 5.44424057006836e-05,
      "model_forward_time": 1.7456047534942627,
      "step": 57087
    },
    {
      "epoch": 5.44424057006836e-05,
      "step": 57087,
      "training_step_time": 6.840714693069458
    },
    {
      "epoch": 5.4443359375e-05,
      "model_forward_time": 1.7282099723815918,
      "step": 57088
    },
    {
      "epoch": 5.4443359375e-05,
      "step": 57088,
      "training_step_time": 6.842981338500977
    },
    {
      "epoch": 5.4444313049316405e-05,
      "model_forward_time": 1.7312119007110596,
      "step": 57089
    },
    {
      "epoch": 5.4444313049316405e-05,
      "step": 57089,
      "training_step_time": 6.856714487075806
    },
    {
      "epoch": 5.4445266723632816e-05,
      "grad_norm": 0.1059473529458046,
      "learning_rate": 4.2441317536525585e-06,
      "loss": 0.0597,
      "step": 57090
    },
    {
      "epoch": 5.4445266723632816e-05,
      "model_forward_time": 1.7385315895080566,
      "step": 57090
    },
    {
      "dynamics_loss_avg": 0.0508299158886075,
      "epoch": 5.4445266723632816e-05,
      "step": 57090
    },
    {
      "action_loss_avg": 0.00133430203422904,
      "epoch": 5.4445266723632816e-05,
      "step": 57090
    },
    {
      "epoch": 5.4445266723632816e-05,
      "step": 57090,
      "training_step_time": 6.855482339859009
    },
    {
      "epoch": 5.444622039794922e-05,
      "model_forward_time": 1.7324445247650146,
      "step": 57091
    },
    {
      "epoch": 5.444622039794922e-05,
      "step": 57091,
      "training_step_time": 7.805729627609253
    },
    {
      "epoch": 5.4447174072265624e-05,
      "model_forward_time": 2.9613704681396484,
      "step": 57092
    },
    {
      "epoch": 5.4447174072265624e-05,
      "step": 57092,
      "training_step_time": 7.801012992858887
    },
    {
      "epoch": 5.4448127746582035e-05,
      "model_forward_time": 1.7426960468292236,
      "step": 57093
    },
    {
      "epoch": 5.4448127746582035e-05,
      "step": 57093,
      "training_step_time": 9.454894304275513
    },
    {
      "epoch": 5.444908142089844e-05,
      "model_forward_time": 1.7301616668701172,
      "step": 57094
    },
    {
      "epoch": 5.444908142089844e-05,
      "step": 57094,
      "training_step_time": 7.635640859603882
    },
    {
      "epoch": 5.445003509521484e-05,
      "model_forward_time": 1.739933967590332,
      "step": 57095
    },
    {
      "epoch": 5.445003509521484e-05,
      "step": 57095,
      "training_step_time": 7.120945692062378
    },
    {
      "epoch": 5.445098876953125e-05,
      "model_forward_time": 1.7365124225616455,
      "step": 57096
    },
    {
      "epoch": 5.445098876953125e-05,
      "step": 57096,
      "training_step_time": 6.845932722091675
    },
    {
      "epoch": 5.445194244384766e-05,
      "model_forward_time": 1.7363357543945312,
      "step": 57097
    },
    {
      "epoch": 5.445194244384766e-05,
      "step": 57097,
      "training_step_time": 6.854322910308838
    },
    {
      "epoch": 5.445289611816406e-05,
      "model_forward_time": 1.3594555854797363,
      "step": 57098
    },
    {
      "epoch": 5.445289611816406e-05,
      "step": 57098,
      "training_step_time": 6.858496427536011
    },
    {
      "epoch": 5.4453849792480466e-05,
      "model_forward_time": 1.7375476360321045,
      "step": 57099
    },
    {
      "epoch": 5.4453849792480466e-05,
      "step": 57099,
      "training_step_time": 6.835628032684326
    },
    {
      "epoch": 5.4454803466796876e-05,
      "grad_norm": 0.13637275993824005,
      "learning_rate": 4.242497328156672e-06,
      "loss": 0.0614,
      "step": 57100
    },
    {
      "epoch": 5.4454803466796876e-05,
      "model_forward_time": 1.7295053005218506,
      "step": 57100
    },
    {
      "dynamics_loss_avg": 0.05966695360839367,
      "epoch": 5.4454803466796876e-05,
      "step": 57100
    },
    {
      "action_loss_avg": 0.0019457226328086108,
      "epoch": 5.4454803466796876e-05,
      "step": 57100
    },
    {
      "epoch": 5.4454803466796876e-05,
      "step": 57100,
      "training_step_time": 7.7902610301971436
    },
    {
      "epoch": 5.445575714111328e-05,
      "model_forward_time": 2.93926739692688,
      "step": 57101
    },
    {
      "epoch": 5.445575714111328e-05,
      "step": 57101,
      "training_step_time": 7.86185359954834
    },
    {
      "epoch": 5.4456710815429684e-05,
      "model_forward_time": 1.7449817657470703,
      "step": 57102
    },
    {
      "epoch": 5.4456710815429684e-05,
      "step": 57102,
      "training_step_time": 8.595665693283081
    },
    {
      "epoch": 5.4457664489746095e-05,
      "model_forward_time": 1.7435111999511719,
      "step": 57103
    },
    {
      "epoch": 5.4457664489746095e-05,
      "step": 57103,
      "training_step_time": 7.731285810470581
    },
    {
      "epoch": 5.44586181640625e-05,
      "model_forward_time": 1.7356669902801514,
      "step": 57104
    },
    {
      "epoch": 5.44586181640625e-05,
      "step": 57104,
      "training_step_time": 7.169655799865723
    },
    {
      "epoch": 5.44595718383789e-05,
      "model_forward_time": 1.7342994213104248,
      "step": 57105
    },
    {
      "epoch": 5.44595718383789e-05,
      "step": 57105,
      "training_step_time": 6.8851165771484375
    },
    {
      "epoch": 5.4460525512695314e-05,
      "model_forward_time": 1.7266926765441895,
      "step": 57106
    },
    {
      "epoch": 5.4460525512695314e-05,
      "step": 57106,
      "training_step_time": 6.885693550109863
    },
    {
      "epoch": 5.446147918701172e-05,
      "model_forward_time": 1.7348086833953857,
      "step": 57107
    },
    {
      "epoch": 5.446147918701172e-05,
      "step": 57107,
      "training_step_time": 6.805471897125244
    },
    {
      "epoch": 5.446243286132812e-05,
      "model_forward_time": 1.7413876056671143,
      "step": 57108
    },
    {
      "epoch": 5.446243286132812e-05,
      "step": 57108,
      "training_step_time": 6.84184455871582
    },
    {
      "epoch": 5.446338653564453e-05,
      "model_forward_time": 1.7284448146820068,
      "step": 57109
    },
    {
      "epoch": 5.446338653564453e-05,
      "step": 57109,
      "training_step_time": 6.843072414398193
    },
    {
      "epoch": 5.446434020996094e-05,
      "grad_norm": 0.10335320979356766,
      "learning_rate": 4.240862985500139e-06,
      "loss": 0.0622,
      "step": 57110
    },
    {
      "epoch": 5.446434020996094e-05,
      "model_forward_time": 2.960725784301758,
      "step": 57110
    },
    {
      "dynamics_loss_avg": 0.06920483633875847,
      "epoch": 5.446434020996094e-05,
      "step": 57110
    },
    {
      "action_loss_avg": 0.0016351316211512313,
      "epoch": 5.446434020996094e-05,
      "step": 57110
    },
    {
      "epoch": 5.446434020996094e-05,
      "step": 57110,
      "training_step_time": 7.849773645401001
    },
    {
      "epoch": 5.446529388427734e-05,
      "model_forward_time": 1.7196776866912842,
      "step": 57111
    },
    {
      "epoch": 5.446529388427734e-05,
      "step": 57111,
      "training_step_time": 10.35035753250122
    },
    {
      "epoch": 5.446624755859375e-05,
      "model_forward_time": 1.7285656929016113,
      "step": 57112
    },
    {
      "epoch": 5.446624755859375e-05,
      "step": 57112,
      "training_step_time": 7.686474084854126
    },
    {
      "epoch": 5.4467201232910156e-05,
      "model_forward_time": 1.7398595809936523,
      "step": 57113
    },
    {
      "epoch": 5.4467201232910156e-05,
      "step": 57113,
      "training_step_time": 7.152189493179321
    },
    {
      "epoch": 5.446815490722656e-05,
      "model_forward_time": 1.3727517127990723,
      "step": 57114
    },
    {
      "epoch": 5.446815490722656e-05,
      "step": 57114,
      "training_step_time": 6.885682582855225
    },
    {
      "epoch": 5.446910858154297e-05,
      "model_forward_time": 1.7298240661621094,
      "step": 57115
    },
    {
      "epoch": 5.446910858154297e-05,
      "step": 57115,
      "training_step_time": 6.894520044326782
    },
    {
      "epoch": 5.4470062255859375e-05,
      "model_forward_time": 1.7318909168243408,
      "step": 57116
    },
    {
      "epoch": 5.4470062255859375e-05,
      "step": 57116,
      "training_step_time": 6.9347264766693115
    },
    {
      "epoch": 5.447101593017578e-05,
      "model_forward_time": 1.7314388751983643,
      "step": 57117
    },
    {
      "epoch": 5.447101593017578e-05,
      "step": 57117,
      "training_step_time": 6.882902383804321
    },
    {
      "epoch": 5.447196960449219e-05,
      "model_forward_time": 1.7332074642181396,
      "step": 57118
    },
    {
      "epoch": 5.447196960449219e-05,
      "step": 57118,
      "training_step_time": 6.85724139213562
    },
    {
      "epoch": 5.4472923278808594e-05,
      "model_forward_time": 2.8351337909698486,
      "step": 57119
    },
    {
      "epoch": 5.4472923278808594e-05,
      "step": 57119,
      "training_step_time": 7.8494932651519775
    },
    {
      "epoch": 5.4473876953125e-05,
      "grad_norm": 0.09242219477891922,
      "learning_rate": 4.239228725861689e-06,
      "loss": 0.0588,
      "step": 57120
    },
    {
      "epoch": 5.4473876953125e-05,
      "model_forward_time": 1.8694369792938232,
      "step": 57120
    },
    {
      "dynamics_loss_avg": 0.04707079529762268,
      "epoch": 5.4473876953125e-05,
      "step": 57120
    },
    {
      "action_loss_avg": 0.0013794634694932028,
      "epoch": 5.4473876953125e-05,
      "step": 57120
    },
    {
      "epoch": 5.4473876953125e-05,
      "step": 57120,
      "training_step_time": 10.622867822647095
    },
    {
      "epoch": 5.447483062744141e-05,
      "model_forward_time": 1.7363064289093018,
      "step": 57121
    },
    {
      "epoch": 5.447483062744141e-05,
      "step": 57121,
      "training_step_time": 10.31297779083252
    },
    {
      "epoch": 5.447578430175781e-05,
      "model_forward_time": 1.7310805320739746,
      "step": 57122
    },
    {
      "epoch": 5.447578430175781e-05,
      "step": 57122,
      "training_step_time": 7.508634805679321
    },
    {
      "epoch": 5.4476737976074216e-05,
      "model_forward_time": 1.7416601181030273,
      "step": 57123
    },
    {
      "epoch": 5.4476737976074216e-05,
      "step": 57123,
      "training_step_time": 7.127625942230225
    },
    {
      "epoch": 5.447769165039063e-05,
      "model_forward_time": 1.7352681159973145,
      "step": 57124
    },
    {
      "epoch": 5.447769165039063e-05,
      "step": 57124,
      "training_step_time": 6.843152046203613
    },
    {
      "epoch": 5.447864532470703e-05,
      "model_forward_time": 1.7392280101776123,
      "step": 57125
    },
    {
      "epoch": 5.447864532470703e-05,
      "step": 57125,
      "training_step_time": 6.879237174987793
    },
    {
      "epoch": 5.4479598999023435e-05,
      "model_forward_time": 1.7417662143707275,
      "step": 57126
    },
    {
      "epoch": 5.4479598999023435e-05,
      "step": 57126,
      "training_step_time": 6.915231943130493
    },
    {
      "epoch": 5.4480552673339846e-05,
      "model_forward_time": 1.7395637035369873,
      "step": 57127
    },
    {
      "epoch": 5.4480552673339846e-05,
      "step": 57127,
      "training_step_time": 6.871020793914795
    },
    {
      "epoch": 5.448150634765625e-05,
      "model_forward_time": 1.7412376403808594,
      "step": 57128
    },
    {
      "epoch": 5.448150634765625e-05,
      "step": 57128,
      "training_step_time": 7.872461318969727
    },
    {
      "epoch": 5.4482460021972654e-05,
      "model_forward_time": 2.953718900680542,
      "step": 57129
    },
    {
      "epoch": 5.4482460021972654e-05,
      "step": 57129,
      "training_step_time": 9.859968423843384
    },
    {
      "epoch": 5.4483413696289065e-05,
      "grad_norm": 0.14298446476459503,
      "learning_rate": 4.23759454942004e-06,
      "loss": 0.0566,
      "step": 57130
    },
    {
      "epoch": 5.4483413696289065e-05,
      "model_forward_time": 1.7434189319610596,
      "step": 57130
    },
    {
      "dynamics_loss_avg": 0.06720834784209728,
      "epoch": 5.4483413696289065e-05,
      "step": 57130
    },
    {
      "action_loss_avg": 0.0018827331601642071,
      "epoch": 5.4483413696289065e-05,
      "step": 57130
    },
    {
      "epoch": 5.4483413696289065e-05,
      "step": 57130,
      "training_step_time": 9.430695295333862
    },
    {
      "epoch": 5.448436737060547e-05,
      "model_forward_time": 1.7321763038635254,
      "step": 57131
    },
    {
      "epoch": 5.448436737060547e-05,
      "step": 57131,
      "training_step_time": 7.724643707275391
    },
    {
      "epoch": 5.448532104492187e-05,
      "model_forward_time": 1.7388367652893066,
      "step": 57132
    },
    {
      "epoch": 5.448532104492187e-05,
      "step": 57132,
      "training_step_time": 7.247203350067139
    },
    {
      "epoch": 5.4486274719238284e-05,
      "model_forward_time": 1.730980634689331,
      "step": 57133
    },
    {
      "epoch": 5.4486274719238284e-05,
      "step": 57133,
      "training_step_time": 6.9420166015625
    },
    {
      "epoch": 5.448722839355469e-05,
      "model_forward_time": 1.7338800430297852,
      "step": 57134
    },
    {
      "epoch": 5.448722839355469e-05,
      "step": 57134,
      "training_step_time": 6.971287727355957
    },
    {
      "epoch": 5.448818206787109e-05,
      "model_forward_time": 1.737736701965332,
      "step": 57135
    },
    {
      "epoch": 5.448818206787109e-05,
      "step": 57135,
      "training_step_time": 6.832580804824829
    },
    {
      "epoch": 5.44891357421875e-05,
      "model_forward_time": 1.7392494678497314,
      "step": 57136
    },
    {
      "epoch": 5.44891357421875e-05,
      "step": 57136,
      "training_step_time": 6.900734186172485
    },
    {
      "epoch": 5.449008941650391e-05,
      "model_forward_time": 1.7361540794372559,
      "step": 57137
    },
    {
      "epoch": 5.449008941650391e-05,
      "step": 57137,
      "training_step_time": 7.896967649459839
    },
    {
      "epoch": 5.449104309082031e-05,
      "model_forward_time": 2.930373191833496,
      "step": 57138
    },
    {
      "epoch": 5.449104309082031e-05,
      "step": 57138,
      "training_step_time": 9.745018720626831
    },
    {
      "epoch": 5.449199676513672e-05,
      "model_forward_time": 1.7577321529388428,
      "step": 57139
    },
    {
      "epoch": 5.449199676513672e-05,
      "step": 57139,
      "training_step_time": 9.791999340057373
    },
    {
      "epoch": 5.4492950439453126e-05,
      "grad_norm": 0.10900495201349258,
      "learning_rate": 4.235960456353906e-06,
      "loss": 0.0557,
      "step": 57140
    },
    {
      "epoch": 5.4492950439453126e-05,
      "model_forward_time": 1.731325387954712,
      "step": 57140
    },
    {
      "dynamics_loss_avg": 0.05538385882973671,
      "epoch": 5.4492950439453126e-05,
      "step": 57140
    },
    {
      "action_loss_avg": 0.0014629035373218357,
      "epoch": 5.4492950439453126e-05,
      "step": 57140
    },
    {
      "epoch": 5.4492950439453126e-05,
      "step": 57140,
      "training_step_time": 7.7223546504974365
    },
    {
      "epoch": 5.449390411376953e-05,
      "model_forward_time": 1.7425885200500488,
      "step": 57141
    },
    {
      "epoch": 5.449390411376953e-05,
      "step": 57141,
      "training_step_time": 7.885143995285034
    },
    {
      "epoch": 5.449485778808594e-05,
      "model_forward_time": 1.7355091571807861,
      "step": 57142
    },
    {
      "epoch": 5.449485778808594e-05,
      "step": 57142,
      "training_step_time": 6.968770503997803
    },
    {
      "epoch": 5.4495811462402344e-05,
      "model_forward_time": 1.7305939197540283,
      "step": 57143
    },
    {
      "epoch": 5.4495811462402344e-05,
      "step": 57143,
      "training_step_time": 7.029131889343262
    },
    {
      "epoch": 5.449676513671875e-05,
      "model_forward_time": 1.7344450950622559,
      "step": 57144
    },
    {
      "epoch": 5.449676513671875e-05,
      "step": 57144,
      "training_step_time": 6.994489908218384
    },
    {
      "epoch": 5.449771881103516e-05,
      "model_forward_time": 1.7502567768096924,
      "step": 57145
    },
    {
      "epoch": 5.449771881103516e-05,
      "step": 57145,
      "training_step_time": 6.943898439407349
    },
    {
      "epoch": 5.449867248535156e-05,
      "model_forward_time": 1.7317490577697754,
      "step": 57146
    },
    {
      "epoch": 5.449867248535156e-05,
      "step": 57146,
      "training_step_time": 7.872700929641724
    },
    {
      "epoch": 5.449962615966797e-05,
      "model_forward_time": 2.937422275543213,
      "step": 57147
    },
    {
      "epoch": 5.449962615966797e-05,
      "step": 57147,
      "training_step_time": 7.846418857574463
    },
    {
      "epoch": 5.450057983398438e-05,
      "model_forward_time": 1.7343964576721191,
      "step": 57148
    },
    {
      "epoch": 5.450057983398438e-05,
      "step": 57148,
      "training_step_time": 9.95717477798462
    },
    {
      "epoch": 5.450153350830078e-05,
      "model_forward_time": 1.7324509620666504,
      "step": 57149
    },
    {
      "epoch": 5.450153350830078e-05,
      "step": 57149,
      "training_step_time": 7.717896223068237
    },
    {
      "epoch": 5.4502487182617186e-05,
      "grad_norm": 0.13264362514019012,
      "learning_rate": 4.234326446841989e-06,
      "loss": 0.0576,
      "step": 57150
    },
    {
      "epoch": 5.4502487182617186e-05,
      "model_forward_time": 1.73655366897583,
      "step": 57150
    },
    {
      "dynamics_loss_avg": 0.04947585919871926,
      "epoch": 5.4502487182617186e-05,
      "step": 57150
    },
    {
      "action_loss_avg": 0.0012495250353822484,
      "epoch": 5.4502487182617186e-05,
      "step": 57150
    },
    {
      "epoch": 5.4502487182617186e-05,
      "step": 57150,
      "training_step_time": 7.782266616821289
    },
    {
      "epoch": 5.45034408569336e-05,
      "model_forward_time": 1.7262864112854004,
      "step": 57151
    },
    {
      "epoch": 5.45034408569336e-05,
      "step": 57151,
      "training_step_time": 6.7781360149383545
    },
    {
      "epoch": 5.450439453125e-05,
      "model_forward_time": 1.7420060634613037,
      "step": 57152
    },
    {
      "epoch": 5.450439453125e-05,
      "step": 57152,
      "training_step_time": 6.878603935241699
    },
    {
      "epoch": 5.4505348205566405e-05,
      "model_forward_time": 1.7462310791015625,
      "step": 57153
    },
    {
      "epoch": 5.4505348205566405e-05,
      "step": 57153,
      "training_step_time": 6.8697028160095215
    },
    {
      "epoch": 5.4506301879882816e-05,
      "model_forward_time": 1.7396652698516846,
      "step": 57154
    },
    {
      "epoch": 5.4506301879882816e-05,
      "step": 57154,
      "training_step_time": 6.8481879234313965
    },
    {
      "epoch": 5.450725555419922e-05,
      "model_forward_time": 1.7489345073699951,
      "step": 57155
    },
    {
      "epoch": 5.450725555419922e-05,
      "step": 57155,
      "training_step_time": 7.859923601150513
    },
    {
      "epoch": 5.4508209228515624e-05,
      "model_forward_time": 2.950411796569824,
      "step": 57156
    },
    {
      "epoch": 5.4508209228515624e-05,
      "step": 57156,
      "training_step_time": 9.915992975234985
    },
    {
      "epoch": 5.450916290283203e-05,
      "model_forward_time": 1.7586135864257812,
      "step": 57157
    },
    {
      "epoch": 5.450916290283203e-05,
      "step": 57157,
      "training_step_time": 7.878576278686523
    },
    {
      "epoch": 5.451011657714844e-05,
      "model_forward_time": 1.752397060394287,
      "step": 57158
    },
    {
      "epoch": 5.451011657714844e-05,
      "step": 57158,
      "training_step_time": 7.839078664779663
    },
    {
      "epoch": 5.451107025146484e-05,
      "model_forward_time": 1.7415354251861572,
      "step": 57159
    },
    {
      "epoch": 5.451107025146484e-05,
      "step": 57159,
      "training_step_time": 7.660140752792358
    },
    {
      "epoch": 5.451202392578125e-05,
      "grad_norm": 0.14354881644248962,
      "learning_rate": 4.232692521062979e-06,
      "loss": 0.056,
      "step": 57160
    },
    {
      "epoch": 5.451202392578125e-05,
      "model_forward_time": 1.7647089958190918,
      "step": 57160
    },
    {
      "dynamics_loss_avg": 0.05129391700029373,
      "epoch": 5.451202392578125e-05,
      "step": 57160
    },
    {
      "action_loss_avg": 0.0017868347000330686,
      "epoch": 5.451202392578125e-05,
      "step": 57160
    },
    {
      "epoch": 5.451202392578125e-05,
      "step": 57160,
      "training_step_time": 6.873711824417114
    },
    {
      "epoch": 5.451297760009766e-05,
      "model_forward_time": 1.7567424774169922,
      "step": 57161
    },
    {
      "epoch": 5.451297760009766e-05,
      "step": 57161,
      "training_step_time": 6.833495616912842
    },
    {
      "epoch": 5.451393127441406e-05,
      "model_forward_time": 1.7470641136169434,
      "step": 57162
    },
    {
      "epoch": 5.451393127441406e-05,
      "step": 57162,
      "training_step_time": 6.850466012954712
    },
    {
      "epoch": 5.4514884948730466e-05,
      "model_forward_time": 1.751140832901001,
      "step": 57163
    },
    {
      "epoch": 5.4514884948730466e-05,
      "step": 57163,
      "training_step_time": 6.810928106307983
    },
    {
      "epoch": 5.4515838623046876e-05,
      "model_forward_time": 1.379918098449707,
      "step": 57164
    },
    {
      "epoch": 5.4515838623046876e-05,
      "step": 57164,
      "training_step_time": 7.635399103164673
    },
    {
      "epoch": 5.451679229736328e-05,
      "model_forward_time": 2.8079020977020264,
      "step": 57165
    },
    {
      "epoch": 5.451679229736328e-05,
      "step": 57165,
      "training_step_time": 7.852572679519653
    },
    {
      "epoch": 5.4517745971679684e-05,
      "model_forward_time": 1.8662257194519043,
      "step": 57166
    },
    {
      "epoch": 5.4517745971679684e-05,
      "step": 57166,
      "training_step_time": 9.108917474746704
    },
    {
      "epoch": 5.4518699645996095e-05,
      "model_forward_time": 1.747375726699829,
      "step": 57167
    },
    {
      "epoch": 5.4518699645996095e-05,
      "step": 57167,
      "training_step_time": 7.809258222579956
    },
    {
      "epoch": 5.45196533203125e-05,
      "model_forward_time": 1.7558038234710693,
      "step": 57168
    },
    {
      "epoch": 5.45196533203125e-05,
      "step": 57168,
      "training_step_time": 7.672677993774414
    },
    {
      "epoch": 5.45206069946289e-05,
      "model_forward_time": 1.740246057510376,
      "step": 57169
    },
    {
      "epoch": 5.45206069946289e-05,
      "step": 57169,
      "training_step_time": 7.791980266571045
    },
    {
      "epoch": 5.4521560668945314e-05,
      "grad_norm": 0.14223028719425201,
      "learning_rate": 4.231058679195562e-06,
      "loss": 0.0561,
      "step": 57170
    },
    {
      "epoch": 5.4521560668945314e-05,
      "model_forward_time": 1.7518541812896729,
      "step": 57170
    },
    {
      "dynamics_loss_avg": 0.05065912753343582,
      "epoch": 5.4521560668945314e-05,
      "step": 57170
    },
    {
      "action_loss_avg": 0.0019003685301868245,
      "epoch": 5.4521560668945314e-05,
      "step": 57170
    },
    {
      "epoch": 5.4521560668945314e-05,
      "step": 57170,
      "training_step_time": 6.865623474121094
    },
    {
      "epoch": 5.452251434326172e-05,
      "model_forward_time": 1.7584631443023682,
      "step": 57171
    },
    {
      "epoch": 5.452251434326172e-05,
      "step": 57171,
      "training_step_time": 6.919540882110596
    },
    {
      "epoch": 5.452346801757812e-05,
      "model_forward_time": 1.7537364959716797,
      "step": 57172
    },
    {
      "epoch": 5.452346801757812e-05,
      "step": 57172,
      "training_step_time": 6.8660728931427
    },
    {
      "epoch": 5.452442169189453e-05,
      "model_forward_time": 1.7510380744934082,
      "step": 57173
    },
    {
      "epoch": 5.452442169189453e-05,
      "step": 57173,
      "training_step_time": 7.608400344848633
    },
    {
      "epoch": 5.452537536621094e-05,
      "model_forward_time": 1.7449991703033447,
      "step": 57174
    },
    {
      "epoch": 5.452537536621094e-05,
      "step": 57174,
      "training_step_time": 7.803254842758179
    },
    {
      "epoch": 5.452632904052734e-05,
      "model_forward_time": 2.90278959274292,
      "step": 57175
    },
    {
      "epoch": 5.452632904052734e-05,
      "step": 57175,
      "training_step_time": 11.34467601776123
    },
    {
      "epoch": 5.452728271484375e-05,
      "model_forward_time": 1.4057765007019043,
      "step": 57176
    },
    {
      "epoch": 5.452728271484375e-05,
      "step": 57176,
      "training_step_time": 8.84790325164795
    },
    {
      "epoch": 5.4528236389160156e-05,
      "model_forward_time": 1.7540068626403809,
      "step": 57177
    },
    {
      "epoch": 5.4528236389160156e-05,
      "step": 57177,
      "training_step_time": 7.748177528381348
    },
    {
      "epoch": 5.452919006347656e-05,
      "model_forward_time": 1.7587008476257324,
      "step": 57178
    },
    {
      "epoch": 5.452919006347656e-05,
      "step": 57178,
      "training_step_time": 7.7977447509765625
    },
    {
      "epoch": 5.453014373779297e-05,
      "model_forward_time": 1.7574398517608643,
      "step": 57179
    },
    {
      "epoch": 5.453014373779297e-05,
      "step": 57179,
      "training_step_time": 6.81348729133606
    },
    {
      "epoch": 5.4531097412109375e-05,
      "grad_norm": 0.16082847118377686,
      "learning_rate": 4.229424921418411e-06,
      "loss": 0.0584,
      "step": 57180
    },
    {
      "epoch": 5.4531097412109375e-05,
      "model_forward_time": 1.757098913192749,
      "step": 57180
    },
    {
      "dynamics_loss_avg": 0.05745828095823526,
      "epoch": 5.4531097412109375e-05,
      "step": 57180
    },
    {
      "action_loss_avg": 0.0018148133880458773,
      "epoch": 5.4531097412109375e-05,
      "step": 57180
    },
    {
      "epoch": 5.4531097412109375e-05,
      "step": 57180,
      "training_step_time": 6.854166269302368
    },
    {
      "epoch": 5.453205108642578e-05,
      "model_forward_time": 1.7506325244903564,
      "step": 57181
    },
    {
      "epoch": 5.453205108642578e-05,
      "step": 57181,
      "training_step_time": 6.85078501701355
    },
    {
      "epoch": 5.453300476074219e-05,
      "model_forward_time": 1.7520878314971924,
      "step": 57182
    },
    {
      "epoch": 5.453300476074219e-05,
      "step": 57182,
      "training_step_time": 7.416838884353638
    },
    {
      "epoch": 5.4533958435058594e-05,
      "model_forward_time": 1.764373779296875,
      "step": 57183
    },
    {
      "epoch": 5.4533958435058594e-05,
      "step": 57183,
      "training_step_time": 7.735550403594971
    },
    {
      "epoch": 5.4534912109375e-05,
      "model_forward_time": 2.8396189212799072,
      "step": 57184
    },
    {
      "epoch": 5.4534912109375e-05,
      "step": 57184,
      "training_step_time": 9.259574174880981
    },
    {
      "epoch": 5.453586578369141e-05,
      "model_forward_time": 1.7652044296264648,
      "step": 57185
    },
    {
      "epoch": 5.453586578369141e-05,
      "step": 57185,
      "training_step_time": 9.695273160934448
    },
    {
      "epoch": 5.453681945800781e-05,
      "model_forward_time": 1.375748634338379,
      "step": 57186
    },
    {
      "epoch": 5.453681945800781e-05,
      "step": 57186,
      "training_step_time": 7.698954105377197
    },
    {
      "epoch": 5.4537773132324216e-05,
      "model_forward_time": 1.75205659866333,
      "step": 57187
    },
    {
      "epoch": 5.4537773132324216e-05,
      "step": 57187,
      "training_step_time": 7.789775609970093
    },
    {
      "epoch": 5.453872680664063e-05,
      "model_forward_time": 1.7558856010437012,
      "step": 57188
    },
    {
      "epoch": 5.453872680664063e-05,
      "step": 57188,
      "training_step_time": 6.854328870773315
    },
    {
      "epoch": 5.453968048095703e-05,
      "model_forward_time": 1.7740976810455322,
      "step": 57189
    },
    {
      "epoch": 5.453968048095703e-05,
      "step": 57189,
      "training_step_time": 6.762977361679077
    },
    {
      "epoch": 5.4540634155273435e-05,
      "grad_norm": 0.10099905729293823,
      "learning_rate": 4.227791247910192e-06,
      "loss": 0.0623,
      "step": 57190
    },
    {
      "epoch": 5.4540634155273435e-05,
      "model_forward_time": 1.748342752456665,
      "step": 57190
    },
    {
      "dynamics_loss_avg": 0.06461284998804331,
      "epoch": 5.4540634155273435e-05,
      "step": 57190
    },
    {
      "action_loss_avg": 0.0021132287656655535,
      "epoch": 5.4540634155273435e-05,
      "step": 57190
    },
    {
      "epoch": 5.4540634155273435e-05,
      "step": 57190,
      "training_step_time": 6.853626728057861
    },
    {
      "epoch": 5.4541587829589846e-05,
      "model_forward_time": 1.752763032913208,
      "step": 57191
    },
    {
      "epoch": 5.4541587829589846e-05,
      "step": 57191,
      "training_step_time": 7.47232985496521
    },
    {
      "epoch": 5.454254150390625e-05,
      "model_forward_time": 1.7612009048461914,
      "step": 57192
    },
    {
      "epoch": 5.454254150390625e-05,
      "step": 57192,
      "training_step_time": 8.582622528076172
    },
    {
      "epoch": 5.4543495178222654e-05,
      "model_forward_time": 2.8619585037231445,
      "step": 57193
    },
    {
      "epoch": 5.4543495178222654e-05,
      "step": 57193,
      "training_step_time": 8.931418895721436
    },
    {
      "epoch": 5.4544448852539065e-05,
      "model_forward_time": 1.7586636543273926,
      "step": 57194
    },
    {
      "epoch": 5.4544448852539065e-05,
      "step": 57194,
      "training_step_time": 9.864417314529419
    },
    {
      "epoch": 5.454540252685547e-05,
      "model_forward_time": 1.7532556056976318,
      "step": 57195
    },
    {
      "epoch": 5.454540252685547e-05,
      "step": 57195,
      "training_step_time": 10.028380870819092
    },
    {
      "epoch": 5.454635620117187e-05,
      "model_forward_time": 1.7586863040924072,
      "step": 57196
    },
    {
      "epoch": 5.454635620117187e-05,
      "step": 57196,
      "training_step_time": 7.730894327163696
    },
    {
      "epoch": 5.4547309875488284e-05,
      "model_forward_time": 1.749527931213379,
      "step": 57197
    },
    {
      "epoch": 5.4547309875488284e-05,
      "step": 57197,
      "training_step_time": 6.841562271118164
    },
    {
      "epoch": 5.454826354980469e-05,
      "model_forward_time": 1.7590065002441406,
      "step": 57198
    },
    {
      "epoch": 5.454826354980469e-05,
      "step": 57198,
      "training_step_time": 6.858068227767944
    },
    {
      "epoch": 5.454921722412109e-05,
      "model_forward_time": 1.7623209953308105,
      "step": 57199
    },
    {
      "epoch": 5.454921722412109e-05,
      "step": 57199,
      "training_step_time": 6.846654891967773
    },
    {
      "epoch": 5.45501708984375e-05,
      "grad_norm": 0.11875118315219879,
      "learning_rate": 4.226157658849562e-06,
      "loss": 0.0589,
      "step": 57200
    },
    {
      "epoch": 5.45501708984375e-05,
      "model_forward_time": 1.758068561553955,
      "step": 57200
    },
    {
      "dynamics_loss_avg": 0.05479989126324654,
      "epoch": 5.45501708984375e-05,
      "step": 57200
    },
    {
      "action_loss_avg": 0.0017558977706357837,
      "epoch": 5.45501708984375e-05,
      "step": 57200
    },
    {
      "epoch": 5.45501708984375e-05,
      "step": 57200,
      "training_step_time": 7.425881624221802
    },
    {
      "epoch": 5.455112457275391e-05,
      "model_forward_time": 1.7476823329925537,
      "step": 57201
    },
    {
      "epoch": 5.455112457275391e-05,
      "step": 57201,
      "training_step_time": 7.613713026046753
    },
    {
      "epoch": 5.455207824707031e-05,
      "model_forward_time": 1.7644331455230713,
      "step": 57202
    },
    {
      "epoch": 5.455207824707031e-05,
      "step": 57202,
      "training_step_time": 7.804965257644653
    },
    {
      "epoch": 5.455303192138672e-05,
      "model_forward_time": 2.871488094329834,
      "step": 57203
    },
    {
      "epoch": 5.455303192138672e-05,
      "step": 57203,
      "training_step_time": 8.640645503997803
    },
    {
      "epoch": 5.4553985595703126e-05,
      "model_forward_time": 1.7505555152893066,
      "step": 57204
    },
    {
      "epoch": 5.4553985595703126e-05,
      "step": 57204,
      "training_step_time": 9.308585405349731
    },
    {
      "epoch": 5.455493927001953e-05,
      "model_forward_time": 1.7509286403656006,
      "step": 57205
    },
    {
      "epoch": 5.455493927001953e-05,
      "step": 57205,
      "training_step_time": 7.7732834815979
    },
    {
      "epoch": 5.455589294433594e-05,
      "model_forward_time": 1.7495968341827393,
      "step": 57206
    },
    {
      "epoch": 5.455589294433594e-05,
      "step": 57206,
      "training_step_time": 7.77778697013855
    },
    {
      "epoch": 5.4556846618652344e-05,
      "model_forward_time": 1.75718092918396,
      "step": 57207
    },
    {
      "epoch": 5.4556846618652344e-05,
      "step": 57207,
      "training_step_time": 6.868556261062622
    },
    {
      "epoch": 5.455780029296875e-05,
      "model_forward_time": 1.7475767135620117,
      "step": 57208
    },
    {
      "epoch": 5.455780029296875e-05,
      "step": 57208,
      "training_step_time": 6.834775924682617
    },
    {
      "epoch": 5.455875396728516e-05,
      "model_forward_time": 1.7559614181518555,
      "step": 57209
    },
    {
      "epoch": 5.455875396728516e-05,
      "step": 57209,
      "training_step_time": 7.419722318649292
    },
    {
      "epoch": 5.455970764160156e-05,
      "grad_norm": 0.07514319568872452,
      "learning_rate": 4.2245241544151665e-06,
      "loss": 0.0558,
      "step": 57210
    },
    {
      "epoch": 5.455970764160156e-05,
      "model_forward_time": 1.7801411151885986,
      "step": 57210
    },
    {
      "dynamics_loss_avg": 0.04670480769127607,
      "epoch": 5.455970764160156e-05,
      "step": 57210
    },
    {
      "action_loss_avg": 0.0016782900871476158,
      "epoch": 5.455970764160156e-05,
      "step": 57210
    },
    {
      "epoch": 5.455970764160156e-05,
      "step": 57210,
      "training_step_time": 8.2134850025177
    },
    {
      "epoch": 5.456066131591797e-05,
      "model_forward_time": 1.75593900680542,
      "step": 57211
    },
    {
      "epoch": 5.456066131591797e-05,
      "step": 57211,
      "training_step_time": 7.824703216552734
    },
    {
      "epoch": 5.456161499023438e-05,
      "model_forward_time": 2.9384145736694336,
      "step": 57212
    },
    {
      "epoch": 5.456161499023438e-05,
      "step": 57212,
      "training_step_time": 9.083016872406006
    },
    {
      "epoch": 5.456256866455078e-05,
      "model_forward_time": 1.7627456188201904,
      "step": 57213
    },
    {
      "epoch": 5.456256866455078e-05,
      "step": 57213,
      "training_step_time": 8.305039405822754
    },
    {
      "epoch": 5.4563522338867186e-05,
      "model_forward_time": 1.7263641357421875,
      "step": 57214
    },
    {
      "epoch": 5.4563522338867186e-05,
      "step": 57214,
      "training_step_time": 8.534140825271606
    },
    {
      "epoch": 5.45644760131836e-05,
      "model_forward_time": 1.749647855758667,
      "step": 57215
    },
    {
      "epoch": 5.45644760131836e-05,
      "step": 57215,
      "training_step_time": 7.682077884674072
    },
    {
      "epoch": 5.45654296875e-05,
      "model_forward_time": 1.7625548839569092,
      "step": 57216
    },
    {
      "epoch": 5.45654296875e-05,
      "step": 57216,
      "training_step_time": 6.831758975982666
    },
    {
      "epoch": 5.4566383361816405e-05,
      "model_forward_time": 1.7495357990264893,
      "step": 57217
    },
    {
      "epoch": 5.4566383361816405e-05,
      "step": 57217,
      "training_step_time": 6.835834741592407
    },
    {
      "epoch": 5.4567337036132816e-05,
      "model_forward_time": 1.7460062503814697,
      "step": 57218
    },
    {
      "epoch": 5.4567337036132816e-05,
      "step": 57218,
      "training_step_time": 7.4169909954071045
    },
    {
      "epoch": 5.456829071044922e-05,
      "model_forward_time": 1.74578857421875,
      "step": 57219
    },
    {
      "epoch": 5.456829071044922e-05,
      "step": 57219,
      "training_step_time": 7.5714027881622314
    },
    {
      "epoch": 5.4569244384765624e-05,
      "grad_norm": 0.0805889219045639,
      "learning_rate": 4.222890734785643e-06,
      "loss": 0.0611,
      "step": 57220
    },
    {
      "epoch": 5.4569244384765624e-05,
      "model_forward_time": 1.7680292129516602,
      "step": 57220
    },
    {
      "dynamics_loss_avg": 0.05781553853303194,
      "epoch": 5.4569244384765624e-05,
      "step": 57220
    },
    {
      "action_loss_avg": 0.0014883134106639772,
      "epoch": 5.4569244384765624e-05,
      "step": 57220
    },
    {
      "epoch": 5.4569244384765624e-05,
      "step": 57220,
      "training_step_time": 7.8375890254974365
    },
    {
      "epoch": 5.457019805908203e-05,
      "model_forward_time": 2.9282450675964355,
      "step": 57221
    },
    {
      "epoch": 5.457019805908203e-05,
      "step": 57221,
      "training_step_time": 9.34226655960083
    },
    {
      "epoch": 5.457115173339844e-05,
      "model_forward_time": 1.757944107055664,
      "step": 57222
    },
    {
      "epoch": 5.457115173339844e-05,
      "step": 57222,
      "training_step_time": 8.533531427383423
    },
    {
      "epoch": 5.457210540771484e-05,
      "model_forward_time": 1.7549822330474854,
      "step": 57223
    },
    {
      "epoch": 5.457210540771484e-05,
      "step": 57223,
      "training_step_time": 7.719066381454468
    },
    {
      "epoch": 5.457305908203125e-05,
      "model_forward_time": 1.7604048252105713,
      "step": 57224
    },
    {
      "epoch": 5.457305908203125e-05,
      "step": 57224,
      "training_step_time": 7.778947353363037
    },
    {
      "epoch": 5.457401275634766e-05,
      "model_forward_time": 1.757617473602295,
      "step": 57225
    },
    {
      "epoch": 5.457401275634766e-05,
      "step": 57225,
      "training_step_time": 6.870511770248413
    },
    {
      "epoch": 5.457496643066406e-05,
      "model_forward_time": 1.7560653686523438,
      "step": 57226
    },
    {
      "epoch": 5.457496643066406e-05,
      "step": 57226,
      "training_step_time": 6.872436285018921
    },
    {
      "epoch": 5.4575920104980466e-05,
      "model_forward_time": 1.7607519626617432,
      "step": 57227
    },
    {
      "epoch": 5.4575920104980466e-05,
      "step": 57227,
      "training_step_time": 7.483288526535034
    },
    {
      "epoch": 5.4576873779296876e-05,
      "model_forward_time": 1.7552845478057861,
      "step": 57228
    },
    {
      "epoch": 5.4576873779296876e-05,
      "step": 57228,
      "training_step_time": 7.451598882675171
    },
    {
      "epoch": 5.457782745361328e-05,
      "model_forward_time": 1.7654895782470703,
      "step": 57229
    },
    {
      "epoch": 5.457782745361328e-05,
      "step": 57229,
      "training_step_time": 9.286578178405762
    },
    {
      "epoch": 5.4578781127929685e-05,
      "grad_norm": 0.1444767415523529,
      "learning_rate": 4.221257400139622e-06,
      "loss": 0.0603,
      "step": 57230
    },
    {
      "epoch": 5.4578781127929685e-05,
      "model_forward_time": 2.8859589099884033,
      "step": 57230
    },
    {
      "dynamics_loss_avg": 0.06733378358185291,
      "epoch": 5.4578781127929685e-05,
      "step": 57230
    },
    {
      "action_loss_avg": 0.0014270941785071046,
      "epoch": 5.4578781127929685e-05,
      "step": 57230
    },
    {
      "epoch": 5.4578781127929685e-05,
      "step": 57230,
      "training_step_time": 9.726231575012207
    },
    {
      "epoch": 5.4579734802246095e-05,
      "model_forward_time": 1.8282675743103027,
      "step": 57231
    },
    {
      "epoch": 5.4579734802246095e-05,
      "step": 57231,
      "training_step_time": 10.664528369903564
    },
    {
      "epoch": 5.45806884765625e-05,
      "model_forward_time": 1.7537908554077148,
      "step": 57232
    },
    {
      "epoch": 5.45806884765625e-05,
      "step": 57232,
      "training_step_time": 7.752946376800537
    },
    {
      "epoch": 5.45816421508789e-05,
      "model_forward_time": 1.7505271434783936,
      "step": 57233
    },
    {
      "epoch": 5.45816421508789e-05,
      "step": 57233,
      "training_step_time": 7.753763675689697
    },
    {
      "epoch": 5.4582595825195314e-05,
      "model_forward_time": 1.7203702926635742,
      "step": 57234
    },
    {
      "epoch": 5.4582595825195314e-05,
      "step": 57234,
      "training_step_time": 6.843583822250366
    },
    {
      "epoch": 5.458354949951172e-05,
      "model_forward_time": 1.7763581275939941,
      "step": 57235
    },
    {
      "epoch": 5.458354949951172e-05,
      "step": 57235,
      "training_step_time": 6.789050817489624
    },
    {
      "epoch": 5.458450317382812e-05,
      "model_forward_time": 1.7594764232635498,
      "step": 57236
    },
    {
      "epoch": 5.458450317382812e-05,
      "step": 57236,
      "training_step_time": 7.464298963546753
    },
    {
      "epoch": 5.458545684814453e-05,
      "model_forward_time": 1.7552845478057861,
      "step": 57237
    },
    {
      "epoch": 5.458545684814453e-05,
      "step": 57237,
      "training_step_time": 7.579957962036133
    },
    {
      "epoch": 5.458641052246094e-05,
      "model_forward_time": 1.7620518207550049,
      "step": 57238
    },
    {
      "epoch": 5.458641052246094e-05,
      "step": 57238,
      "training_step_time": 7.741103887557983
    },
    {
      "epoch": 5.458736419677734e-05,
      "model_forward_time": 2.8953816890716553,
      "step": 57239
    },
    {
      "epoch": 5.458736419677734e-05,
      "step": 57239,
      "training_step_time": 8.569531440734863
    },
    {
      "epoch": 5.458831787109375e-05,
      "grad_norm": 0.08816048502922058,
      "learning_rate": 4.21962415065572e-06,
      "loss": 0.0624,
      "step": 57240
    },
    {
      "epoch": 5.458831787109375e-05,
      "model_forward_time": 1.851637840270996,
      "step": 57240
    },
    {
      "dynamics_loss_avg": 0.06664522271603346,
      "epoch": 5.458831787109375e-05,
      "step": 57240
    },
    {
      "action_loss_avg": 0.0013002199004404247,
      "epoch": 5.458831787109375e-05,
      "step": 57240
    },
    {
      "epoch": 5.458831787109375e-05,
      "step": 57240,
      "training_step_time": 7.851409673690796
    },
    {
      "epoch": 5.4589271545410156e-05,
      "model_forward_time": 1.7452819347381592,
      "step": 57241
    },
    {
      "epoch": 5.4589271545410156e-05,
      "step": 57241,
      "training_step_time": 8.627662420272827
    },
    {
      "epoch": 5.459022521972656e-05,
      "model_forward_time": 1.7496633529663086,
      "step": 57242
    },
    {
      "epoch": 5.459022521972656e-05,
      "step": 57242,
      "training_step_time": 7.766232013702393
    },
    {
      "epoch": 5.459117889404297e-05,
      "model_forward_time": 1.7434942722320557,
      "step": 57243
    },
    {
      "epoch": 5.459117889404297e-05,
      "step": 57243,
      "training_step_time": 6.870503902435303
    },
    {
      "epoch": 5.4592132568359375e-05,
      "model_forward_time": 1.7651147842407227,
      "step": 57244
    },
    {
      "epoch": 5.4592132568359375e-05,
      "step": 57244,
      "training_step_time": 6.873802423477173
    },
    {
      "epoch": 5.459308624267578e-05,
      "model_forward_time": 1.057685375213623,
      "step": 57245
    },
    {
      "epoch": 5.459308624267578e-05,
      "step": 57245,
      "training_step_time": 7.454104900360107
    },
    {
      "epoch": 5.459403991699219e-05,
      "model_forward_time": 1.7360777854919434,
      "step": 57246
    },
    {
      "epoch": 5.459403991699219e-05,
      "step": 57246,
      "training_step_time": 7.55832839012146
    },
    {
      "epoch": 5.4594993591308594e-05,
      "model_forward_time": 1.7536027431488037,
      "step": 57247
    },
    {
      "epoch": 5.4594993591308594e-05,
      "step": 57247,
      "training_step_time": 7.682522296905518
    },
    {
      "epoch": 5.4595947265625e-05,
      "model_forward_time": 1.7634766101837158,
      "step": 57248
    },
    {
      "epoch": 5.4595947265625e-05,
      "step": 57248,
      "training_step_time": 9.529431819915771
    },
    {
      "epoch": 5.459690093994141e-05,
      "model_forward_time": 2.8577749729156494,
      "step": 57249
    },
    {
      "epoch": 5.459690093994141e-05,
      "step": 57249,
      "training_step_time": 8.477184057235718
    },
    {
      "epoch": 5.459785461425781e-05,
      "grad_norm": 0.09684278815984726,
      "learning_rate": 4.2179909865125475e-06,
      "loss": 0.0616,
      "step": 57250
    },
    {
      "epoch": 5.459785461425781e-05,
      "model_forward_time": 1.7730660438537598,
      "step": 57250
    },
    {
      "dynamics_loss_avg": 0.07894473094493151,
      "epoch": 5.459785461425781e-05,
      "step": 57250
    },
    {
      "action_loss_avg": 0.0014987776172347366,
      "epoch": 5.459785461425781e-05,
      "step": 57250
    },
    {
      "epoch": 5.459785461425781e-05,
      "step": 57250,
      "training_step_time": 8.371929407119751
    },
    {
      "epoch": 5.4598808288574217e-05,
      "model_forward_time": 1.7548243999481201,
      "step": 57251
    },
    {
      "epoch": 5.4598808288574217e-05,
      "step": 57251,
      "training_step_time": 7.878349304199219
    },
    {
      "epoch": 5.459976196289063e-05,
      "model_forward_time": 1.756805181503296,
      "step": 57252
    },
    {
      "epoch": 5.459976196289063e-05,
      "step": 57252,
      "training_step_time": 7.804351806640625
    },
    {
      "epoch": 5.460071563720703e-05,
      "model_forward_time": 1.7725014686584473,
      "step": 57253
    },
    {
      "epoch": 5.460071563720703e-05,
      "step": 57253,
      "training_step_time": 6.828040361404419
    },
    {
      "epoch": 5.4601669311523435e-05,
      "model_forward_time": 1.754704475402832,
      "step": 57254
    },
    {
      "epoch": 5.4601669311523435e-05,
      "step": 57254,
      "training_step_time": 6.813817501068115
    },
    {
      "epoch": 5.4602622985839846e-05,
      "model_forward_time": 1.7542943954467773,
      "step": 57255
    },
    {
      "epoch": 5.4602622985839846e-05,
      "step": 57255,
      "training_step_time": 7.604567527770996
    },
    {
      "epoch": 5.460357666015625e-05,
      "model_forward_time": 1.762756586074829,
      "step": 57256
    },
    {
      "epoch": 5.460357666015625e-05,
      "step": 57256,
      "training_step_time": 7.677572965621948
    },
    {
      "epoch": 5.4604530334472654e-05,
      "model_forward_time": 1.7653882503509521,
      "step": 57257
    },
    {
      "epoch": 5.4604530334472654e-05,
      "step": 57257,
      "training_step_time": 7.9248223304748535
    },
    {
      "epoch": 5.4605484008789065e-05,
      "model_forward_time": 2.854719638824463,
      "step": 57258
    },
    {
      "epoch": 5.4605484008789065e-05,
      "step": 57258,
      "training_step_time": 8.628698110580444
    },
    {
      "epoch": 5.460643768310547e-05,
      "model_forward_time": 1.7686948776245117,
      "step": 57259
    },
    {
      "epoch": 5.460643768310547e-05,
      "step": 57259,
      "training_step_time": 7.918709754943848
    },
    {
      "epoch": 5.460739135742187e-05,
      "grad_norm": 0.09439550340175629,
      "learning_rate": 4.216357907888704e-06,
      "loss": 0.0619,
      "step": 57260
    },
    {
      "epoch": 5.460739135742187e-05,
      "model_forward_time": 1.747722864151001,
      "step": 57260
    },
    {
      "dynamics_loss_avg": 0.0705876361578703,
      "epoch": 5.460739135742187e-05,
      "step": 57260
    },
    {
      "action_loss_avg": 0.002535506308777258,
      "epoch": 5.460739135742187e-05,
      "step": 57260
    },
    {
      "epoch": 5.460739135742187e-05,
      "step": 57260,
      "training_step_time": 8.804799556732178
    },
    {
      "epoch": 5.4608345031738284e-05,
      "model_forward_time": 1.7321016788482666,
      "step": 57261
    },
    {
      "epoch": 5.4608345031738284e-05,
      "step": 57261,
      "training_step_time": 7.863117694854736
    },
    {
      "epoch": 5.460929870605469e-05,
      "model_forward_time": 1.7708792686462402,
      "step": 57262
    },
    {
      "epoch": 5.460929870605469e-05,
      "step": 57262,
      "training_step_time": 6.873847723007202
    },
    {
      "epoch": 5.461025238037109e-05,
      "model_forward_time": 1.7572903633117676,
      "step": 57263
    },
    {
      "epoch": 5.461025238037109e-05,
      "step": 57263,
      "training_step_time": 6.881105422973633
    },
    {
      "epoch": 5.46112060546875e-05,
      "model_forward_time": 1.7599375247955322,
      "step": 57264
    },
    {
      "epoch": 5.46112060546875e-05,
      "step": 57264,
      "training_step_time": 7.606103181838989
    },
    {
      "epoch": 5.461215972900391e-05,
      "model_forward_time": 1.753286600112915,
      "step": 57265
    },
    {
      "epoch": 5.461215972900391e-05,
      "step": 57265,
      "training_step_time": 7.745997667312622
    },
    {
      "epoch": 5.461311340332031e-05,
      "model_forward_time": 1.7638423442840576,
      "step": 57266
    },
    {
      "epoch": 5.461311340332031e-05,
      "step": 57266,
      "training_step_time": 9.949607610702515
    },
    {
      "epoch": 5.461406707763672e-05,
      "model_forward_time": 2.919051170349121,
      "step": 57267
    },
    {
      "epoch": 5.461406707763672e-05,
      "step": 57267,
      "training_step_time": 9.550011396408081
    },
    {
      "epoch": 5.4615020751953126e-05,
      "model_forward_time": 1.762216329574585,
      "step": 57268
    },
    {
      "epoch": 5.4615020751953126e-05,
      "step": 57268,
      "training_step_time": 7.821780681610107
    },
    {
      "epoch": 5.461597442626953e-05,
      "model_forward_time": 1.7584998607635498,
      "step": 57269
    },
    {
      "epoch": 5.461597442626953e-05,
      "step": 57269,
      "training_step_time": 7.817389011383057
    },
    {
      "epoch": 5.461692810058594e-05,
      "grad_norm": 0.10874715447425842,
      "learning_rate": 4.2147249149627826e-06,
      "loss": 0.0632,
      "step": 57270
    },
    {
      "epoch": 5.461692810058594e-05,
      "model_forward_time": 1.7733559608459473,
      "step": 57270
    },
    {
      "dynamics_loss_avg": 0.047425557114183906,
      "epoch": 5.461692810058594e-05,
      "step": 57270
    },
    {
      "action_loss_avg": 0.0011743569280952214,
      "epoch": 5.461692810058594e-05,
      "step": 57270
    },
    {
      "epoch": 5.461692810058594e-05,
      "step": 57270,
      "training_step_time": 7.799687385559082
    },
    {
      "epoch": 5.4617881774902345e-05,
      "model_forward_time": 1.7567038536071777,
      "step": 57271
    },
    {
      "epoch": 5.4617881774902345e-05,
      "step": 57271,
      "training_step_time": 6.888870477676392
    },
    {
      "epoch": 5.461883544921875e-05,
      "model_forward_time": 1.7666356563568115,
      "step": 57272
    },
    {
      "epoch": 5.461883544921875e-05,
      "step": 57272,
      "training_step_time": 6.797223329544067
    },
    {
      "epoch": 5.461978912353516e-05,
      "model_forward_time": 1.7675697803497314,
      "step": 57273
    },
    {
      "epoch": 5.461978912353516e-05,
      "step": 57273,
      "training_step_time": 7.6351237297058105
    },
    {
      "epoch": 5.462074279785156e-05,
      "model_forward_time": 1.7597506046295166,
      "step": 57274
    },
    {
      "epoch": 5.462074279785156e-05,
      "step": 57274,
      "training_step_time": 7.707261085510254
    },
    {
      "epoch": 5.462169647216797e-05,
      "model_forward_time": 1.7565834522247314,
      "step": 57275
    },
    {
      "epoch": 5.462169647216797e-05,
      "step": 57275,
      "training_step_time": 7.795547723770142
    },
    {
      "epoch": 5.462265014648438e-05,
      "model_forward_time": 2.9087178707122803,
      "step": 57276
    },
    {
      "epoch": 5.462265014648438e-05,
      "step": 57276,
      "training_step_time": 7.810203552246094
    },
    {
      "epoch": 5.462360382080078e-05,
      "model_forward_time": 1.7691044807434082,
      "step": 57277
    },
    {
      "epoch": 5.462360382080078e-05,
      "step": 57277,
      "training_step_time": 9.520728588104248
    },
    {
      "epoch": 5.4624557495117186e-05,
      "model_forward_time": 1.7590663433074951,
      "step": 57278
    },
    {
      "epoch": 5.4624557495117186e-05,
      "step": 57278,
      "training_step_time": 7.716793537139893
    },
    {
      "epoch": 5.46255111694336e-05,
      "model_forward_time": 1.7687177658081055,
      "step": 57279
    },
    {
      "epoch": 5.46255111694336e-05,
      "step": 57279,
      "training_step_time": 7.787876605987549
    },
    {
      "epoch": 5.462646484375e-05,
      "grad_norm": 0.11962119489908218,
      "learning_rate": 4.213092007913364e-06,
      "loss": 0.058,
      "step": 57280
    },
    {
      "epoch": 5.462646484375e-05,
      "model_forward_time": 1.7578022480010986,
      "step": 57280
    },
    {
      "dynamics_loss_avg": 0.055748830549418926,
      "epoch": 5.462646484375e-05,
      "step": 57280
    },
    {
      "action_loss_avg": 0.0021775327913928777,
      "epoch": 5.462646484375e-05,
      "step": 57280
    },
    {
      "epoch": 5.462646484375e-05,
      "step": 57280,
      "training_step_time": 6.820357799530029
    },
    {
      "epoch": 5.4627418518066405e-05,
      "model_forward_time": 1.751058578491211,
      "step": 57281
    },
    {
      "epoch": 5.4627418518066405e-05,
      "step": 57281,
      "training_step_time": 6.859637975692749
    },
    {
      "epoch": 5.462837219238281e-05,
      "model_forward_time": 1.7578277587890625,
      "step": 57282
    },
    {
      "epoch": 5.462837219238281e-05,
      "step": 57282,
      "training_step_time": 7.5951457023620605
    },
    {
      "epoch": 5.462932586669922e-05,
      "model_forward_time": 1.4016544818878174,
      "step": 57283
    },
    {
      "epoch": 5.462932586669922e-05,
      "step": 57283,
      "training_step_time": 7.007688760757446
    },
    {
      "epoch": 5.4630279541015624e-05,
      "model_forward_time": 1.7628865242004395,
      "step": 57284
    },
    {
      "epoch": 5.4630279541015624e-05,
      "step": 57284,
      "training_step_time": 7.8412184715271
    },
    {
      "epoch": 5.463123321533203e-05,
      "model_forward_time": 1.755664587020874,
      "step": 57285
    },
    {
      "epoch": 5.463123321533203e-05,
      "step": 57285,
      "training_step_time": 9.154139280319214
    },
    {
      "epoch": 5.463218688964844e-05,
      "model_forward_time": 2.9337425231933594,
      "step": 57286
    },
    {
      "epoch": 5.463218688964844e-05,
      "step": 57286,
      "training_step_time": 8.635236024856567
    },
    {
      "epoch": 5.463314056396484e-05,
      "model_forward_time": 1.7700517177581787,
      "step": 57287
    },
    {
      "epoch": 5.463314056396484e-05,
      "step": 57287,
      "training_step_time": 8.840210199356079
    },
    {
      "epoch": 5.463409423828125e-05,
      "model_forward_time": 1.7699172496795654,
      "step": 57288
    },
    {
      "epoch": 5.463409423828125e-05,
      "step": 57288,
      "training_step_time": 7.7964699268341064
    },
    {
      "epoch": 5.463504791259766e-05,
      "model_forward_time": 1.7542471885681152,
      "step": 57289
    },
    {
      "epoch": 5.463504791259766e-05,
      "step": 57289,
      "training_step_time": 7.7791054248809814
    },
    {
      "epoch": 5.463600158691406e-05,
      "grad_norm": 0.11172610521316528,
      "learning_rate": 4.211459186919021e-06,
      "loss": 0.0535,
      "step": 57290
    },
    {
      "epoch": 5.463600158691406e-05,
      "model_forward_time": 1.7618906497955322,
      "step": 57290
    },
    {
      "dynamics_loss_avg": 0.061572659946978095,
      "epoch": 5.463600158691406e-05,
      "step": 57290
    },
    {
      "action_loss_avg": 0.001799233420751989,
      "epoch": 5.463600158691406e-05,
      "step": 57290
    },
    {
      "epoch": 5.463600158691406e-05,
      "step": 57290,
      "training_step_time": 6.8648130893707275
    },
    {
      "epoch": 5.4636955261230466e-05,
      "model_forward_time": 1.7644104957580566,
      "step": 57291
    },
    {
      "epoch": 5.4636955261230466e-05,
      "step": 57291,
      "training_step_time": 7.485592603683472
    },
    {
      "epoch": 5.4637908935546877e-05,
      "model_forward_time": 1.7523479461669922,
      "step": 57292
    },
    {
      "epoch": 5.4637908935546877e-05,
      "step": 57292,
      "training_step_time": 7.611660480499268
    },
    {
      "epoch": 5.463886260986328e-05,
      "model_forward_time": 1.7599713802337646,
      "step": 57293
    },
    {
      "epoch": 5.463886260986328e-05,
      "step": 57293,
      "training_step_time": 7.811954975128174
    },
    {
      "epoch": 5.4639816284179685e-05,
      "model_forward_time": 1.770538330078125,
      "step": 57294
    },
    {
      "epoch": 5.4639816284179685e-05,
      "step": 57294,
      "training_step_time": 7.845085382461548
    },
    {
      "epoch": 5.4640769958496095e-05,
      "model_forward_time": 2.8938560485839844,
      "step": 57295
    },
    {
      "epoch": 5.4640769958496095e-05,
      "step": 57295,
      "training_step_time": 8.409554243087769
    },
    {
      "epoch": 5.46417236328125e-05,
      "model_forward_time": 1.7477970123291016,
      "step": 57296
    },
    {
      "epoch": 5.46417236328125e-05,
      "step": 57296,
      "training_step_time": 7.762431859970093
    },
    {
      "epoch": 5.4642677307128903e-05,
      "model_forward_time": 1.7604408264160156,
      "step": 57297
    },
    {
      "epoch": 5.4642677307128903e-05,
      "step": 57297,
      "training_step_time": 7.771406412124634
    },
    {
      "epoch": 5.4643630981445314e-05,
      "model_forward_time": 1.7607414722442627,
      "step": 57298
    },
    {
      "epoch": 5.4643630981445314e-05,
      "step": 57298,
      "training_step_time": 7.8374786376953125
    },
    {
      "epoch": 5.464458465576172e-05,
      "model_forward_time": 1.7583470344543457,
      "step": 57299
    },
    {
      "epoch": 5.464458465576172e-05,
      "step": 57299,
      "training_step_time": 6.9393227100372314
    },
    {
      "epoch": 5.464553833007812e-05,
      "grad_norm": 0.1559031456708908,
      "learning_rate": 4.209826452158315e-06,
      "loss": 0.0581,
      "step": 57300
    },
    {
      "epoch": 5.464553833007812e-05,
      "model_forward_time": 1.769749402999878,
      "step": 57300
    },
    {
      "dynamics_loss_avg": 0.07108354698866606,
      "epoch": 5.464553833007812e-05,
      "step": 57300
    },
    {
      "action_loss_avg": 0.002215189879643731,
      "epoch": 5.464553833007812e-05,
      "step": 57300
    },
    {
      "epoch": 5.464553833007812e-05,
      "step": 57300,
      "training_step_time": 6.927579164505005
    },
    {
      "epoch": 5.464649200439453e-05,
      "model_forward_time": 1.7684478759765625,
      "step": 57301
    },
    {
      "epoch": 5.464649200439453e-05,
      "step": 57301,
      "training_step_time": 7.6277711391448975
    },
    {
      "epoch": 5.464744567871094e-05,
      "model_forward_time": 1.7483398914337158,
      "step": 57302
    },
    {
      "epoch": 5.464744567871094e-05,
      "step": 57302,
      "training_step_time": 8.510255098342896
    },
    {
      "epoch": 5.464839935302734e-05,
      "model_forward_time": 1.7731523513793945,
      "step": 57303
    },
    {
      "epoch": 5.464839935302734e-05,
      "step": 57303,
      "training_step_time": 7.847060203552246
    },
    {
      "epoch": 5.464935302734375e-05,
      "model_forward_time": 2.5316717624664307,
      "step": 57304
    },
    {
      "epoch": 5.464935302734375e-05,
      "step": 57304,
      "training_step_time": 7.7577965259552
    },
    {
      "epoch": 5.4650306701660156e-05,
      "model_forward_time": 1.7588269710540771,
      "step": 57305
    },
    {
      "epoch": 5.4650306701660156e-05,
      "step": 57305,
      "training_step_time": 8.558996438980103
    },
    {
      "epoch": 5.465126037597656e-05,
      "model_forward_time": 1.771594524383545,
      "step": 57306
    },
    {
      "epoch": 5.465126037597656e-05,
      "step": 57306,
      "training_step_time": 7.783108711242676
    },
    {
      "epoch": 5.465221405029297e-05,
      "model_forward_time": 1.7527575492858887,
      "step": 57307
    },
    {
      "epoch": 5.465221405029297e-05,
      "step": 57307,
      "training_step_time": 7.757998704910278
    },
    {
      "epoch": 5.4653167724609375e-05,
      "model_forward_time": 1.7259371280670166,
      "step": 57308
    },
    {
      "epoch": 5.4653167724609375e-05,
      "step": 57308,
      "training_step_time": 6.851333856582642
    },
    {
      "epoch": 5.465412139892578e-05,
      "model_forward_time": 1.761385202407837,
      "step": 57309
    },
    {
      "epoch": 5.465412139892578e-05,
      "step": 57309,
      "training_step_time": 6.852855443954468
    },
    {
      "epoch": 5.465507507324219e-05,
      "grad_norm": 0.1112842783331871,
      "learning_rate": 4.2081938038098e-06,
      "loss": 0.0584,
      "step": 57310
    },
    {
      "epoch": 5.465507507324219e-05,
      "model_forward_time": 1.7486746311187744,
      "step": 57310
    },
    {
      "dynamics_loss_avg": 0.05465221554040909,
      "epoch": 5.465507507324219e-05,
      "step": 57310
    },
    {
      "action_loss_avg": 0.00209492010762915,
      "epoch": 5.465507507324219e-05,
      "step": 57310
    },
    {
      "epoch": 5.465507507324219e-05,
      "step": 57310,
      "training_step_time": 7.554339647293091
    },
    {
      "epoch": 5.4656028747558594e-05,
      "model_forward_time": 1.7596864700317383,
      "step": 57311
    },
    {
      "epoch": 5.4656028747558594e-05,
      "step": 57311,
      "training_step_time": 8.177673816680908
    },
    {
      "epoch": 5.4656982421875e-05,
      "model_forward_time": 1.059135913848877,
      "step": 57312
    },
    {
      "epoch": 5.4656982421875e-05,
      "step": 57312,
      "training_step_time": 9.578569650650024
    },
    {
      "epoch": 5.465793609619141e-05,
      "model_forward_time": 1.8092801570892334,
      "step": 57313
    },
    {
      "epoch": 5.465793609619141e-05,
      "step": 57313,
      "training_step_time": 9.334704399108887
    },
    {
      "epoch": 5.465888977050781e-05,
      "model_forward_time": 2.953113079071045,
      "step": 57314
    },
    {
      "epoch": 5.465888977050781e-05,
      "step": 57314,
      "training_step_time": 8.539620399475098
    },
    {
      "epoch": 5.4659843444824217e-05,
      "model_forward_time": 1.75164794921875,
      "step": 57315
    },
    {
      "epoch": 5.4659843444824217e-05,
      "step": 57315,
      "training_step_time": 8.253007888793945
    },
    {
      "epoch": 5.466079711914063e-05,
      "model_forward_time": 1.7954118251800537,
      "step": 57316
    },
    {
      "epoch": 5.466079711914063e-05,
      "step": 57316,
      "training_step_time": 8.784480333328247
    },
    {
      "epoch": 5.466175079345703e-05,
      "model_forward_time": 1.7561697959899902,
      "step": 57317
    },
    {
      "epoch": 5.466175079345703e-05,
      "step": 57317,
      "training_step_time": 6.828848123550415
    },
    {
      "epoch": 5.4662704467773435e-05,
      "model_forward_time": 1.754948377609253,
      "step": 57318
    },
    {
      "epoch": 5.4662704467773435e-05,
      "step": 57318,
      "training_step_time": 6.865034818649292
    },
    {
      "epoch": 5.4663658142089846e-05,
      "model_forward_time": 1.7606983184814453,
      "step": 57319
    },
    {
      "epoch": 5.4663658142089846e-05,
      "step": 57319,
      "training_step_time": 7.583940744400024
    },
    {
      "epoch": 5.466461181640625e-05,
      "grad_norm": 0.1440521627664566,
      "learning_rate": 4.206561242052018e-06,
      "loss": 0.057,
      "step": 57320
    },
    {
      "epoch": 5.466461181640625e-05,
      "model_forward_time": 1.7560632228851318,
      "step": 57320
    },
    {
      "dynamics_loss_avg": 0.057165725715458394,
      "epoch": 5.466461181640625e-05,
      "step": 57320
    },
    {
      "action_loss_avg": 0.0018374961509834975,
      "epoch": 5.466461181640625e-05,
      "step": 57320
    },
    {
      "epoch": 5.466461181640625e-05,
      "step": 57320,
      "training_step_time": 7.757604360580444
    },
    {
      "epoch": 5.4665565490722654e-05,
      "model_forward_time": 1.7575647830963135,
      "step": 57321
    },
    {
      "epoch": 5.4665565490722654e-05,
      "step": 57321,
      "training_step_time": 8.817785501480103
    },
    {
      "epoch": 5.4666519165039065e-05,
      "model_forward_time": 1.7892563343048096,
      "step": 57322
    },
    {
      "epoch": 5.4666519165039065e-05,
      "step": 57322,
      "training_step_time": 9.888623237609863
    },
    {
      "epoch": 5.466747283935547e-05,
      "model_forward_time": 2.914132595062256,
      "step": 57323
    },
    {
      "epoch": 5.466747283935547e-05,
      "step": 57323,
      "training_step_time": 7.834152936935425
    },
    {
      "epoch": 5.466842651367187e-05,
      "model_forward_time": 1.7645089626312256,
      "step": 57324
    },
    {
      "epoch": 5.466842651367187e-05,
      "step": 57324,
      "training_step_time": 7.797450065612793
    },
    {
      "epoch": 5.4669380187988284e-05,
      "model_forward_time": 1.7576513290405273,
      "step": 57325
    },
    {
      "epoch": 5.4669380187988284e-05,
      "step": 57325,
      "training_step_time": 7.697940111160278
    },
    {
      "epoch": 5.467033386230469e-05,
      "model_forward_time": 1.3882629871368408,
      "step": 57326
    },
    {
      "epoch": 5.467033386230469e-05,
      "step": 57326,
      "training_step_time": 7.246243476867676
    },
    {
      "epoch": 5.467128753662109e-05,
      "model_forward_time": 1.7357697486877441,
      "step": 57327
    },
    {
      "epoch": 5.467128753662109e-05,
      "step": 57327,
      "training_step_time": 6.854362726211548
    },
    {
      "epoch": 5.46722412109375e-05,
      "model_forward_time": 1.7640581130981445,
      "step": 57328
    },
    {
      "epoch": 5.46722412109375e-05,
      "step": 57328,
      "training_step_time": 7.530523300170898
    },
    {
      "epoch": 5.467319488525391e-05,
      "model_forward_time": 1.3891851902008057,
      "step": 57329
    },
    {
      "epoch": 5.467319488525391e-05,
      "step": 57329,
      "training_step_time": 7.735097169876099
    },
    {
      "epoch": 5.467414855957031e-05,
      "grad_norm": 0.09320701658725739,
      "learning_rate": 4.204928767063509e-06,
      "loss": 0.0589,
      "step": 57330
    },
    {
      "epoch": 5.467414855957031e-05,
      "model_forward_time": 1.7409098148345947,
      "step": 57330
    },
    {
      "dynamics_loss_avg": 0.07428706306964158,
      "epoch": 5.467414855957031e-05,
      "step": 57330
    },
    {
      "action_loss_avg": 0.0023435015871655198,
      "epoch": 5.467414855957031e-05,
      "step": 57330
    },
    {
      "epoch": 5.467414855957031e-05,
      "step": 57330,
      "training_step_time": 8.55371904373169
    },
    {
      "epoch": 5.467510223388672e-05,
      "model_forward_time": 1.7591638565063477,
      "step": 57331
    },
    {
      "epoch": 5.467510223388672e-05,
      "step": 57331,
      "training_step_time": 7.814353942871094
    },
    {
      "epoch": 5.4676055908203126e-05,
      "model_forward_time": 2.939218044281006,
      "step": 57332
    },
    {
      "epoch": 5.4676055908203126e-05,
      "step": 57332,
      "training_step_time": 9.298901081085205
    },
    {
      "epoch": 5.467700958251953e-05,
      "model_forward_time": 1.7624211311340332,
      "step": 57333
    },
    {
      "epoch": 5.467700958251953e-05,
      "step": 57333,
      "training_step_time": 10.436326503753662
    },
    {
      "epoch": 5.467796325683594e-05,
      "model_forward_time": 1.7557718753814697,
      "step": 57334
    },
    {
      "epoch": 5.467796325683594e-05,
      "step": 57334,
      "training_step_time": 7.742988109588623
    },
    {
      "epoch": 5.4678916931152345e-05,
      "model_forward_time": 1.7477006912231445,
      "step": 57335
    },
    {
      "epoch": 5.4678916931152345e-05,
      "step": 57335,
      "training_step_time": 7.753113031387329
    },
    {
      "epoch": 5.467987060546875e-05,
      "model_forward_time": 1.7352423667907715,
      "step": 57336
    },
    {
      "epoch": 5.467987060546875e-05,
      "step": 57336,
      "training_step_time": 6.858224391937256
    },
    {
      "epoch": 5.468082427978516e-05,
      "model_forward_time": 1.7502872943878174,
      "step": 57337
    },
    {
      "epoch": 5.468082427978516e-05,
      "step": 57337,
      "training_step_time": 7.4253740310668945
    },
    {
      "epoch": 5.4681777954101563e-05,
      "model_forward_time": 1.789191484451294,
      "step": 57338
    },
    {
      "epoch": 5.4681777954101563e-05,
      "step": 57338,
      "training_step_time": 7.749472141265869
    },
    {
      "epoch": 5.468273162841797e-05,
      "model_forward_time": 1.7494006156921387,
      "step": 57339
    },
    {
      "epoch": 5.468273162841797e-05,
      "step": 57339,
      "training_step_time": 7.892098665237427
    },
    {
      "epoch": 5.468368530273438e-05,
      "grad_norm": 0.09228967130184174,
      "learning_rate": 4.203296379022794e-06,
      "loss": 0.0604,
      "step": 57340
    },
    {
      "epoch": 5.468368530273438e-05,
      "model_forward_time": 1.7320702075958252,
      "step": 57340
    },
    {
      "dynamics_loss_avg": 0.06894242875277996,
      "epoch": 5.468368530273438e-05,
      "step": 57340
    },
    {
      "action_loss_avg": 0.002221169893164188,
      "epoch": 5.468368530273438e-05,
      "step": 57340
    },
    {
      "epoch": 5.468368530273438e-05,
      "step": 57340,
      "training_step_time": 9.93648886680603
    },
    {
      "epoch": 5.468463897705078e-05,
      "model_forward_time": 1.8531098365783691,
      "step": 57341
    },
    {
      "epoch": 5.468463897705078e-05,
      "step": 57341,
      "training_step_time": 8.491414308547974
    },
    {
      "epoch": 5.4685592651367186e-05,
      "model_forward_time": 2.94268536567688,
      "step": 57342
    },
    {
      "epoch": 5.4685592651367186e-05,
      "step": 57342,
      "training_step_time": 9.092167854309082
    },
    {
      "epoch": 5.46865463256836e-05,
      "model_forward_time": 1.719475507736206,
      "step": 57343
    },
    {
      "epoch": 5.46865463256836e-05,
      "step": 57343,
      "training_step_time": 7.730966329574585
    },
    {
      "epoch": 5.46875e-05,
      "model_forward_time": 1.7280607223510742,
      "step": 57344
    },
    {
      "epoch": 5.46875e-05,
      "step": 57344,
      "training_step_time": 7.76287055015564
    },
    {
      "epoch": 5.4688453674316405e-05,
      "model_forward_time": 1.7530746459960938,
      "step": 57345
    },
    {
      "epoch": 5.4688453674316405e-05,
      "step": 57345,
      "training_step_time": 6.853633165359497
    },
    {
      "epoch": 5.468940734863281e-05,
      "model_forward_time": 1.7301125526428223,
      "step": 57346
    },
    {
      "epoch": 5.468940734863281e-05,
      "step": 57346,
      "training_step_time": 7.451386451721191
    },
    {
      "epoch": 5.469036102294922e-05,
      "model_forward_time": 1.7432503700256348,
      "step": 57347
    },
    {
      "epoch": 5.469036102294922e-05,
      "step": 57347,
      "training_step_time": 7.719341516494751
    },
    {
      "epoch": 5.4691314697265624e-05,
      "model_forward_time": 1.3615329265594482,
      "step": 57348
    },
    {
      "epoch": 5.4691314697265624e-05,
      "step": 57348,
      "training_step_time": 7.702188491821289
    },
    {
      "epoch": 5.469226837158203e-05,
      "model_forward_time": 1.740508794784546,
      "step": 57349
    },
    {
      "epoch": 5.469226837158203e-05,
      "step": 57349,
      "training_step_time": 8.995564222335815
    },
    {
      "epoch": 5.469322204589844e-05,
      "grad_norm": 0.11241336911916733,
      "learning_rate": 4.201664078108389e-06,
      "loss": 0.0541,
      "step": 57350
    },
    {
      "epoch": 5.469322204589844e-05,
      "model_forward_time": 1.8278307914733887,
      "step": 57350
    },
    {
      "dynamics_loss_avg": 0.05590770998969674,
      "epoch": 5.469322204589844e-05,
      "step": 57350
    },
    {
      "action_loss_avg": 0.0015592534240568058,
      "epoch": 5.469322204589844e-05,
      "step": 57350
    },
    {
      "epoch": 5.469322204589844e-05,
      "step": 57350,
      "training_step_time": 8.589070081710815
    },
    {
      "epoch": 5.469417572021484e-05,
      "model_forward_time": 2.2459521293640137,
      "step": 57351
    },
    {
      "epoch": 5.469417572021484e-05,
      "step": 57351,
      "training_step_time": 7.804719924926758
    },
    {
      "epoch": 5.469512939453125e-05,
      "model_forward_time": 1.7164373397827148,
      "step": 57352
    },
    {
      "epoch": 5.469512939453125e-05,
      "step": 57352,
      "training_step_time": 7.7348856925964355
    },
    {
      "epoch": 5.469608306884766e-05,
      "model_forward_time": 1.7119929790496826,
      "step": 57353
    },
    {
      "epoch": 5.469608306884766e-05,
      "step": 57353,
      "training_step_time": 7.66441798210144
    },
    {
      "epoch": 5.469703674316406e-05,
      "model_forward_time": 1.7130646705627441,
      "step": 57354
    },
    {
      "epoch": 5.469703674316406e-05,
      "step": 57354,
      "training_step_time": 7.151959419250488
    },
    {
      "epoch": 5.4697990417480466e-05,
      "model_forward_time": 1.7072389125823975,
      "step": 57355
    },
    {
      "epoch": 5.4697990417480466e-05,
      "step": 57355,
      "training_step_time": 7.524842262268066
    },
    {
      "epoch": 5.4698944091796877e-05,
      "model_forward_time": 1.7139570713043213,
      "step": 57356
    },
    {
      "epoch": 5.4698944091796877e-05,
      "step": 57356,
      "training_step_time": 7.656965970993042
    },
    {
      "epoch": 5.469989776611328e-05,
      "model_forward_time": 1.7100844383239746,
      "step": 57357
    },
    {
      "epoch": 5.469989776611328e-05,
      "step": 57357,
      "training_step_time": 8.46924638748169
    },
    {
      "epoch": 5.4700851440429685e-05,
      "model_forward_time": 1.715559959411621,
      "step": 57358
    },
    {
      "epoch": 5.4700851440429685e-05,
      "step": 57358,
      "training_step_time": 8.990585088729858
    },
    {
      "epoch": 5.4701805114746095e-05,
      "model_forward_time": 1.7250993251800537,
      "step": 57359
    },
    {
      "epoch": 5.4701805114746095e-05,
      "step": 57359,
      "training_step_time": 8.76993203163147
    },
    {
      "epoch": 5.47027587890625e-05,
      "grad_norm": 0.1042611300945282,
      "learning_rate": 4.2000318644988e-06,
      "loss": 0.0588,
      "step": 57360
    },
    {
      "epoch": 5.47027587890625e-05,
      "model_forward_time": 2.254507064819336,
      "step": 57360
    },
    {
      "dynamics_loss_avg": 0.040336463041603564,
      "epoch": 5.47027587890625e-05,
      "step": 57360
    },
    {
      "action_loss_avg": 0.0012190661509521306,
      "epoch": 5.47027587890625e-05,
      "step": 57360
    },
    {
      "epoch": 5.47027587890625e-05,
      "step": 57360,
      "training_step_time": 9.349087238311768
    },
    {
      "epoch": 5.4703712463378903e-05,
      "model_forward_time": 1.7169034481048584,
      "step": 57361
    },
    {
      "epoch": 5.4703712463378903e-05,
      "step": 57361,
      "training_step_time": 8.705915927886963
    },
    {
      "epoch": 5.4704666137695314e-05,
      "model_forward_time": 1.727257251739502,
      "step": 57362
    },
    {
      "epoch": 5.4704666137695314e-05,
      "step": 57362,
      "training_step_time": 7.549501419067383
    },
    {
      "epoch": 5.470561981201172e-05,
      "model_forward_time": 1.7084710597991943,
      "step": 57363
    },
    {
      "epoch": 5.470561981201172e-05,
      "step": 57363,
      "training_step_time": 7.780261516571045
    },
    {
      "epoch": 5.470657348632812e-05,
      "model_forward_time": 1.715315818786621,
      "step": 57364
    },
    {
      "epoch": 5.470657348632812e-05,
      "step": 57364,
      "training_step_time": 7.46797513961792
    },
    {
      "epoch": 5.470752716064453e-05,
      "model_forward_time": 1.7100446224212646,
      "step": 57365
    },
    {
      "epoch": 5.470752716064453e-05,
      "step": 57365,
      "training_step_time": 7.67860746383667
    },
    {
      "epoch": 5.470848083496094e-05,
      "model_forward_time": 1.7169816493988037,
      "step": 57366
    },
    {
      "epoch": 5.470848083496094e-05,
      "step": 57366,
      "training_step_time": 7.664992809295654
    },
    {
      "epoch": 5.470943450927734e-05,
      "model_forward_time": 1.7193422317504883,
      "step": 57367
    },
    {
      "epoch": 5.470943450927734e-05,
      "step": 57367,
      "training_step_time": 8.489438533782959
    },
    {
      "epoch": 5.471038818359375e-05,
      "model_forward_time": 1.7253906726837158,
      "step": 57368
    },
    {
      "epoch": 5.471038818359375e-05,
      "step": 57368,
      "training_step_time": 8.327510118484497
    },
    {
      "epoch": 5.4711341857910156e-05,
      "model_forward_time": 2.2695178985595703,
      "step": 57369
    },
    {
      "epoch": 5.4711341857910156e-05,
      "step": 57369,
      "training_step_time": 8.5464928150177
    },
    {
      "epoch": 5.471229553222656e-05,
      "grad_norm": 0.0950024202466011,
      "learning_rate": 4.198399738372524e-06,
      "loss": 0.0594,
      "step": 57370
    },
    {
      "epoch": 5.471229553222656e-05,
      "model_forward_time": 1.7116615772247314,
      "step": 57370
    },
    {
      "dynamics_loss_avg": 0.05995592977851629,
      "epoch": 5.471229553222656e-05,
      "step": 57370
    },
    {
      "action_loss_avg": 0.0012854432774474845,
      "epoch": 5.471229553222656e-05,
      "step": 57370
    },
    {
      "epoch": 5.471229553222656e-05,
      "step": 57370,
      "training_step_time": 8.186049461364746
    },
    {
      "epoch": 5.471324920654297e-05,
      "model_forward_time": 1.726759433746338,
      "step": 57371
    },
    {
      "epoch": 5.471324920654297e-05,
      "step": 57371,
      "training_step_time": 7.442287445068359
    },
    {
      "epoch": 5.4714202880859375e-05,
      "model_forward_time": 1.717733383178711,
      "step": 57372
    },
    {
      "epoch": 5.4714202880859375e-05,
      "step": 57372,
      "training_step_time": 7.6859166622161865
    },
    {
      "epoch": 5.471515655517578e-05,
      "model_forward_time": 1.715012788772583,
      "step": 57373
    },
    {
      "epoch": 5.471515655517578e-05,
      "step": 57373,
      "training_step_time": 7.425410747528076
    },
    {
      "epoch": 5.471611022949219e-05,
      "model_forward_time": 1.7099287509918213,
      "step": 57374
    },
    {
      "epoch": 5.471611022949219e-05,
      "step": 57374,
      "training_step_time": 7.61747670173645
    },
    {
      "epoch": 5.4717063903808594e-05,
      "model_forward_time": 1.7119309902191162,
      "step": 57375
    },
    {
      "epoch": 5.4717063903808594e-05,
      "step": 57375,
      "training_step_time": 8.437355279922485
    },
    {
      "epoch": 5.4718017578125e-05,
      "model_forward_time": 1.7147815227508545,
      "step": 57376
    },
    {
      "epoch": 5.4718017578125e-05,
      "step": 57376,
      "training_step_time": 9.55933690071106
    },
    {
      "epoch": 5.471897125244141e-05,
      "model_forward_time": 1.7295751571655273,
      "step": 57377
    },
    {
      "epoch": 5.471897125244141e-05,
      "step": 57377,
      "training_step_time": 8.640914916992188
    },
    {
      "epoch": 5.471992492675781e-05,
      "model_forward_time": 2.2952897548675537,
      "step": 57378
    },
    {
      "epoch": 5.471992492675781e-05,
      "step": 57378,
      "training_step_time": 9.56404733657837
    },
    {
      "epoch": 5.472087860107422e-05,
      "model_forward_time": 1.7250051498413086,
      "step": 57379
    },
    {
      "epoch": 5.472087860107422e-05,
      "step": 57379,
      "training_step_time": 7.817399263381958
    },
    {
      "epoch": 5.472183227539063e-05,
      "grad_norm": 0.1460379660129547,
      "learning_rate": 4.196767699908046e-06,
      "loss": 0.0567,
      "step": 57380
    },
    {
      "epoch": 5.472183227539063e-05,
      "model_forward_time": 1.7315082550048828,
      "step": 57380
    },
    {
      "dynamics_loss_avg": 0.07617581933736801,
      "epoch": 5.472183227539063e-05,
      "step": 57380
    },
    {
      "action_loss_avg": 0.0021871571079827845,
      "epoch": 5.472183227539063e-05,
      "step": 57380
    },
    {
      "epoch": 5.472183227539063e-05,
      "step": 57380,
      "training_step_time": 7.558087587356567
    },
    {
      "epoch": 5.472278594970703e-05,
      "model_forward_time": 1.733738660812378,
      "step": 57381
    },
    {
      "epoch": 5.472278594970703e-05,
      "step": 57381,
      "training_step_time": 7.292048931121826
    },
    {
      "epoch": 5.4723739624023435e-05,
      "model_forward_time": 1.7314350605010986,
      "step": 57382
    },
    {
      "epoch": 5.4723739624023435e-05,
      "step": 57382,
      "training_step_time": 7.879299640655518
    },
    {
      "epoch": 5.4724693298339846e-05,
      "model_forward_time": 1.7173402309417725,
      "step": 57383
    },
    {
      "epoch": 5.4724693298339846e-05,
      "step": 57383,
      "training_step_time": 7.834742546081543
    },
    {
      "epoch": 5.472564697265625e-05,
      "model_forward_time": 1.7293095588684082,
      "step": 57384
    },
    {
      "epoch": 5.472564697265625e-05,
      "step": 57384,
      "training_step_time": 7.790191411972046
    },
    {
      "epoch": 5.4726600646972654e-05,
      "model_forward_time": 1.3614513874053955,
      "step": 57385
    },
    {
      "epoch": 5.4726600646972654e-05,
      "step": 57385,
      "training_step_time": 7.917749404907227
    },
    {
      "epoch": 5.4727554321289065e-05,
      "model_forward_time": 1.727182149887085,
      "step": 57386
    },
    {
      "epoch": 5.4727554321289065e-05,
      "step": 57386,
      "training_step_time": 7.812502384185791
    },
    {
      "epoch": 5.472850799560547e-05,
      "model_forward_time": 1.7170202732086182,
      "step": 57387
    },
    {
      "epoch": 5.472850799560547e-05,
      "step": 57387,
      "training_step_time": 8.955774068832397
    },
    {
      "epoch": 5.472946166992187e-05,
      "model_forward_time": 2.2740581035614014,
      "step": 57388
    },
    {
      "epoch": 5.472946166992187e-05,
      "step": 57388,
      "training_step_time": 8.779201745986938
    },
    {
      "epoch": 5.4730415344238284e-05,
      "model_forward_time": 1.725001573562622,
      "step": 57389
    },
    {
      "epoch": 5.4730415344238284e-05,
      "step": 57389,
      "training_step_time": 7.7954182624816895
    },
    {
      "epoch": 5.473136901855469e-05,
      "grad_norm": 0.09943076968193054,
      "learning_rate": 4.195135749283845e-06,
      "loss": 0.0557,
      "step": 57390
    },
    {
      "epoch": 5.473136901855469e-05,
      "model_forward_time": 1.7248928546905518,
      "step": 57390
    },
    {
      "dynamics_loss_avg": 0.057688531652092936,
      "epoch": 5.473136901855469e-05,
      "step": 57390
    },
    {
      "action_loss_avg": 0.001954675780143589,
      "epoch": 5.473136901855469e-05,
      "step": 57390
    },
    {
      "epoch": 5.473136901855469e-05,
      "step": 57390,
      "training_step_time": 7.480011701583862
    },
    {
      "epoch": 5.473232269287109e-05,
      "model_forward_time": 1.7233302593231201,
      "step": 57391
    },
    {
      "epoch": 5.473232269287109e-05,
      "step": 57391,
      "training_step_time": 7.543720245361328
    },
    {
      "epoch": 5.47332763671875e-05,
      "model_forward_time": 1.7254350185394287,
      "step": 57392
    },
    {
      "epoch": 5.47332763671875e-05,
      "step": 57392,
      "training_step_time": 7.598405361175537
    },
    {
      "epoch": 5.473423004150391e-05,
      "model_forward_time": 1.7213573455810547,
      "step": 57393
    },
    {
      "epoch": 5.473423004150391e-05,
      "step": 57393,
      "training_step_time": 7.530433654785156
    },
    {
      "epoch": 5.473518371582031e-05,
      "model_forward_time": 1.7238569259643555,
      "step": 57394
    },
    {
      "epoch": 5.473518371582031e-05,
      "step": 57394,
      "training_step_time": 7.602858781814575
    },
    {
      "epoch": 5.473613739013672e-05,
      "model_forward_time": 1.7303626537322998,
      "step": 57395
    },
    {
      "epoch": 5.473613739013672e-05,
      "step": 57395,
      "training_step_time": 7.790271520614624
    },
    {
      "epoch": 5.4737091064453126e-05,
      "model_forward_time": 1.716602087020874,
      "step": 57396
    },
    {
      "epoch": 5.4737091064453126e-05,
      "step": 57396,
      "training_step_time": 8.748029947280884
    },
    {
      "epoch": 5.473804473876953e-05,
      "model_forward_time": 2.2735400199890137,
      "step": 57397
    },
    {
      "epoch": 5.473804473876953e-05,
      "step": 57397,
      "training_step_time": 7.698385238647461
    },
    {
      "epoch": 5.473899841308594e-05,
      "model_forward_time": 1.7184021472930908,
      "step": 57398
    },
    {
      "epoch": 5.473899841308594e-05,
      "step": 57398,
      "training_step_time": 8.14405083656311
    },
    {
      "epoch": 5.4739952087402345e-05,
      "model_forward_time": 1.7194957733154297,
      "step": 57399
    },
    {
      "epoch": 5.4739952087402345e-05,
      "step": 57399,
      "training_step_time": 7.356340408325195
    },
    {
      "epoch": 5.474090576171875e-05,
      "grad_norm": 0.10390487313270569,
      "learning_rate": 4.193503886678389e-06,
      "loss": 0.055,
      "step": 57400
    },
    {
      "epoch": 5.474090576171875e-05,
      "model_forward_time": 1.7247114181518555,
      "step": 57400
    },
    {
      "dynamics_loss_avg": 0.0563453733921051,
      "epoch": 5.474090576171875e-05,
      "step": 57400
    },
    {
      "action_loss_avg": 0.001897807852947153,
      "epoch": 5.474090576171875e-05,
      "step": 57400
    },
    {
      "epoch": 5.474090576171875e-05,
      "step": 57400,
      "training_step_time": 6.633684396743774
    },
    {
      "epoch": 5.474185943603516e-05,
      "model_forward_time": 1.7194240093231201,
      "step": 57401
    },
    {
      "epoch": 5.474185943603516e-05,
      "step": 57401,
      "training_step_time": 8.581924438476562
    },
    {
      "epoch": 5.4742813110351563e-05,
      "model_forward_time": 1.721165418624878,
      "step": 57402
    },
    {
      "epoch": 5.4742813110351563e-05,
      "step": 57402,
      "training_step_time": 7.469149827957153
    },
    {
      "epoch": 5.474376678466797e-05,
      "model_forward_time": 1.7157495021820068,
      "step": 57403
    },
    {
      "epoch": 5.474376678466797e-05,
      "step": 57403,
      "training_step_time": 7.464125394821167
    },
    {
      "epoch": 5.474472045898438e-05,
      "model_forward_time": 1.7155144214630127,
      "step": 57404
    },
    {
      "epoch": 5.474472045898438e-05,
      "step": 57404,
      "training_step_time": 8.469951391220093
    },
    {
      "epoch": 5.474567413330078e-05,
      "model_forward_time": 1.7384202480316162,
      "step": 57405
    },
    {
      "epoch": 5.474567413330078e-05,
      "step": 57405,
      "training_step_time": 8.27578067779541
    },
    {
      "epoch": 5.4746627807617186e-05,
      "model_forward_time": 2.268287420272827,
      "step": 57406
    },
    {
      "epoch": 5.4746627807617186e-05,
      "step": 57406,
      "training_step_time": 7.619938373565674
    },
    {
      "epoch": 5.474758148193359e-05,
      "model_forward_time": 1.7187366485595703,
      "step": 57407
    },
    {
      "epoch": 5.474758148193359e-05,
      "step": 57407,
      "training_step_time": 7.548966646194458
    },
    {
      "epoch": 5.474853515625e-05,
      "model_forward_time": 1.7326645851135254,
      "step": 57408
    },
    {
      "epoch": 5.474853515625e-05,
      "step": 57408,
      "training_step_time": 7.506715297698975
    },
    {
      "epoch": 5.4749488830566405e-05,
      "model_forward_time": 1.7236711978912354,
      "step": 57409
    },
    {
      "epoch": 5.4749488830566405e-05,
      "step": 57409,
      "training_step_time": 6.647599935531616
    },
    {
      "epoch": 5.475044250488281e-05,
      "grad_norm": 0.08942526578903198,
      "learning_rate": 4.191872112270136e-06,
      "loss": 0.0587,
      "step": 57410
    },
    {
      "epoch": 5.475044250488281e-05,
      "model_forward_time": 1.7181899547576904,
      "step": 57410
    },
    {
      "dynamics_loss_avg": 0.05147118307650089,
      "epoch": 5.475044250488281e-05,
      "step": 57410
    },
    {
      "action_loss_avg": 0.002203730173641816,
      "epoch": 5.475044250488281e-05,
      "step": 57410
    },
    {
      "epoch": 5.475044250488281e-05,
      "step": 57410,
      "training_step_time": 7.598282098770142
    },
    {
      "epoch": 5.475139617919922e-05,
      "model_forward_time": 1.719062328338623,
      "step": 57411
    },
    {
      "epoch": 5.475139617919922e-05,
      "step": 57411,
      "training_step_time": 7.57604193687439
    },
    {
      "epoch": 5.4752349853515624e-05,
      "model_forward_time": 1.7125372886657715,
      "step": 57412
    },
    {
      "epoch": 5.4752349853515624e-05,
      "step": 57412,
      "training_step_time": 7.436608552932739
    },
    {
      "epoch": 5.475330352783203e-05,
      "model_forward_time": 1.7186620235443115,
      "step": 57413
    },
    {
      "epoch": 5.475330352783203e-05,
      "step": 57413,
      "training_step_time": 7.8725292682647705
    },
    {
      "epoch": 5.475425720214844e-05,
      "model_forward_time": 1.7317521572113037,
      "step": 57414
    },
    {
      "epoch": 5.475425720214844e-05,
      "step": 57414,
      "training_step_time": 8.326720952987671
    },
    {
      "epoch": 5.475521087646484e-05,
      "model_forward_time": 2.2697913646698,
      "step": 57415
    },
    {
      "epoch": 5.475521087646484e-05,
      "step": 57415,
      "training_step_time": 7.701441049575806
    },
    {
      "epoch": 5.475616455078125e-05,
      "model_forward_time": 1.720109462738037,
      "step": 57416
    },
    {
      "epoch": 5.475616455078125e-05,
      "step": 57416,
      "training_step_time": 7.680677652359009
    },
    {
      "epoch": 5.475711822509766e-05,
      "model_forward_time": 1.7469570636749268,
      "step": 57417
    },
    {
      "epoch": 5.475711822509766e-05,
      "step": 57417,
      "training_step_time": 7.656167030334473
    },
    {
      "epoch": 5.475807189941406e-05,
      "model_forward_time": 1.71659255027771,
      "step": 57418
    },
    {
      "epoch": 5.475807189941406e-05,
      "step": 57418,
      "training_step_time": 7.7890236377716064
    },
    {
      "epoch": 5.4759025573730466e-05,
      "model_forward_time": 1.7185685634613037,
      "step": 57419
    },
    {
      "epoch": 5.4759025573730466e-05,
      "step": 57419,
      "training_step_time": 7.699630498886108
    },
    {
      "epoch": 5.475997924804688e-05,
      "grad_norm": 0.09852124750614166,
      "learning_rate": 4.1902404262375335e-06,
      "loss": 0.0541,
      "step": 57420
    },
    {
      "epoch": 5.475997924804688e-05,
      "model_forward_time": 1.718862771987915,
      "step": 57420
    },
    {
      "dynamics_loss_avg": 0.061718062963336706,
      "epoch": 5.475997924804688e-05,
      "step": 57420
    },
    {
      "action_loss_avg": 0.0024242828221758828,
      "epoch": 5.475997924804688e-05,
      "step": 57420
    },
    {
      "epoch": 5.475997924804688e-05,
      "step": 57420,
      "training_step_time": 7.609925031661987
    },
    {
      "epoch": 5.476093292236328e-05,
      "model_forward_time": 1.7272846698760986,
      "step": 57421
    },
    {
      "epoch": 5.476093292236328e-05,
      "step": 57421,
      "training_step_time": 7.517009735107422
    },
    {
      "epoch": 5.4761886596679685e-05,
      "model_forward_time": 1.720276117324829,
      "step": 57422
    },
    {
      "epoch": 5.4761886596679685e-05,
      "step": 57422,
      "training_step_time": 7.862029790878296
    },
    {
      "epoch": 5.4762840270996095e-05,
      "model_forward_time": 1.7249267101287842,
      "step": 57423
    },
    {
      "epoch": 5.4762840270996095e-05,
      "step": 57423,
      "training_step_time": 7.6636693477630615
    },
    {
      "epoch": 5.47637939453125e-05,
      "model_forward_time": 2.2568979263305664,
      "step": 57424
    },
    {
      "epoch": 5.47637939453125e-05,
      "step": 57424,
      "training_step_time": 7.707075357437134
    },
    {
      "epoch": 5.4764747619628904e-05,
      "model_forward_time": 1.7248971462249756,
      "step": 57425
    },
    {
      "epoch": 5.4764747619628904e-05,
      "step": 57425,
      "training_step_time": 7.629160165786743
    },
    {
      "epoch": 5.4765701293945314e-05,
      "model_forward_time": 1.7159392833709717,
      "step": 57426
    },
    {
      "epoch": 5.4765701293945314e-05,
      "step": 57426,
      "training_step_time": 9.103456258773804
    },
    {
      "epoch": 5.476665496826172e-05,
      "model_forward_time": 1.7171175479888916,
      "step": 57427
    },
    {
      "epoch": 5.476665496826172e-05,
      "step": 57427,
      "training_step_time": 7.487017631530762
    },
    {
      "epoch": 5.476760864257812e-05,
      "model_forward_time": 1.7206699848175049,
      "step": 57428
    },
    {
      "epoch": 5.476760864257812e-05,
      "step": 57428,
      "training_step_time": 7.73380970954895
    },
    {
      "epoch": 5.476856231689453e-05,
      "model_forward_time": 1.3526952266693115,
      "step": 57429
    },
    {
      "epoch": 5.476856231689453e-05,
      "step": 57429,
      "training_step_time": 7.6883320808410645
    },
    {
      "epoch": 5.476951599121094e-05,
      "grad_norm": 0.1112547218799591,
      "learning_rate": 4.18860882875902e-06,
      "loss": 0.057,
      "step": 57430
    },
    {
      "epoch": 5.476951599121094e-05,
      "model_forward_time": 1.7270264625549316,
      "step": 57430
    },
    {
      "dynamics_loss_avg": 0.06513651832938194,
      "epoch": 5.476951599121094e-05,
      "step": 57430
    },
    {
      "action_loss_avg": 0.0026757297688163816,
      "epoch": 5.476951599121094e-05,
      "step": 57430
    },
    {
      "epoch": 5.476951599121094e-05,
      "step": 57430,
      "training_step_time": 7.445115089416504
    },
    {
      "epoch": 5.477046966552734e-05,
      "model_forward_time": 1.719689130783081,
      "step": 57431
    },
    {
      "epoch": 5.477046966552734e-05,
      "step": 57431,
      "training_step_time": 8.936018228530884
    },
    {
      "epoch": 5.477142333984375e-05,
      "model_forward_time": 1.723435401916504,
      "step": 57432
    },
    {
      "epoch": 5.477142333984375e-05,
      "step": 57432,
      "training_step_time": 10.61156177520752
    },
    {
      "epoch": 5.4772377014160156e-05,
      "model_forward_time": 2.266850709915161,
      "step": 57433
    },
    {
      "epoch": 5.4772377014160156e-05,
      "step": 57433,
      "training_step_time": 8.50768256187439
    },
    {
      "epoch": 5.477333068847656e-05,
      "model_forward_time": 1.7218401432037354,
      "step": 57434
    },
    {
      "epoch": 5.477333068847656e-05,
      "step": 57434,
      "training_step_time": 7.610553979873657
    },
    {
      "epoch": 5.477428436279297e-05,
      "model_forward_time": 1.7181124687194824,
      "step": 57435
    },
    {
      "epoch": 5.477428436279297e-05,
      "step": 57435,
      "training_step_time": 7.592428207397461
    },
    {
      "epoch": 5.4775238037109375e-05,
      "model_forward_time": 1.7374687194824219,
      "step": 57436
    },
    {
      "epoch": 5.4775238037109375e-05,
      "step": 57436,
      "training_step_time": 7.3967156410217285
    },
    {
      "epoch": 5.477619171142578e-05,
      "model_forward_time": 1.7241356372833252,
      "step": 57437
    },
    {
      "epoch": 5.477619171142578e-05,
      "step": 57437,
      "training_step_time": 7.515990495681763
    },
    {
      "epoch": 5.477714538574219e-05,
      "model_forward_time": 1.7189054489135742,
      "step": 57438
    },
    {
      "epoch": 5.477714538574219e-05,
      "step": 57438,
      "training_step_time": 7.580220699310303
    },
    {
      "epoch": 5.4778099060058594e-05,
      "model_forward_time": 1.7203571796417236,
      "step": 57439
    },
    {
      "epoch": 5.4778099060058594e-05,
      "step": 57439,
      "training_step_time": 7.3027496337890625
    },
    {
      "epoch": 5.4779052734375e-05,
      "grad_norm": 0.10500848293304443,
      "learning_rate": 4.186977320013025e-06,
      "loss": 0.0558,
      "step": 57440
    },
    {
      "epoch": 5.4779052734375e-05,
      "model_forward_time": 1.7146947383880615,
      "step": 57440
    },
    {
      "dynamics_loss_avg": 0.06256624944508075,
      "epoch": 5.4779052734375e-05,
      "step": 57440
    },
    {
      "action_loss_avg": 0.002898014447418973,
      "epoch": 5.4779052734375e-05,
      "step": 57440
    },
    {
      "epoch": 5.4779052734375e-05,
      "step": 57440,
      "training_step_time": 7.514794588088989
    },
    {
      "epoch": 5.478000640869141e-05,
      "model_forward_time": 1.7157886028289795,
      "step": 57441
    },
    {
      "epoch": 5.478000640869141e-05,
      "step": 57441,
      "training_step_time": 10.288271427154541
    },
    {
      "epoch": 5.478096008300781e-05,
      "model_forward_time": 2.2938363552093506,
      "step": 57442
    },
    {
      "epoch": 5.478096008300781e-05,
      "step": 57442,
      "training_step_time": 9.024182081222534
    },
    {
      "epoch": 5.478191375732422e-05,
      "model_forward_time": 1.7129123210906982,
      "step": 57443
    },
    {
      "epoch": 5.478191375732422e-05,
      "step": 57443,
      "training_step_time": 8.32033920288086
    },
    {
      "epoch": 5.478286743164063e-05,
      "model_forward_time": 1.7252521514892578,
      "step": 57444
    },
    {
      "epoch": 5.478286743164063e-05,
      "step": 57444,
      "training_step_time": 8.343566179275513
    },
    {
      "epoch": 5.478382110595703e-05,
      "model_forward_time": 1.738664150238037,
      "step": 57445
    },
    {
      "epoch": 5.478382110595703e-05,
      "step": 57445,
      "training_step_time": 7.677216053009033
    },
    {
      "epoch": 5.4784774780273436e-05,
      "model_forward_time": 1.7158589363098145,
      "step": 57446
    },
    {
      "epoch": 5.4784774780273436e-05,
      "step": 57446,
      "training_step_time": 7.604395627975464
    },
    {
      "epoch": 5.4785728454589846e-05,
      "model_forward_time": 1.7200398445129395,
      "step": 57447
    },
    {
      "epoch": 5.4785728454589846e-05,
      "step": 57447,
      "training_step_time": 8.604278564453125
    },
    {
      "epoch": 5.478668212890625e-05,
      "model_forward_time": 1.718876838684082,
      "step": 57448
    },
    {
      "epoch": 5.478668212890625e-05,
      "step": 57448,
      "training_step_time": 7.434014081954956
    },
    {
      "epoch": 5.4787635803222654e-05,
      "model_forward_time": 1.7184348106384277,
      "step": 57449
    },
    {
      "epoch": 5.4787635803222654e-05,
      "step": 57449,
      "training_step_time": 7.494072675704956
    },
    {
      "epoch": 5.4788589477539065e-05,
      "grad_norm": 0.08526445180177689,
      "learning_rate": 4.185345900177967e-06,
      "loss": 0.054,
      "step": 57450
    },
    {
      "epoch": 5.4788589477539065e-05,
      "model_forward_time": 1.721667766571045,
      "step": 57450
    },
    {
      "dynamics_loss_avg": 0.0508611349388957,
      "epoch": 5.4788589477539065e-05,
      "step": 57450
    },
    {
      "action_loss_avg": 0.0019861327367834748,
      "epoch": 5.4788589477539065e-05,
      "step": 57450
    },
    {
      "epoch": 5.4788589477539065e-05,
      "step": 57450,
      "training_step_time": 8.34327745437622
    },
    {
      "epoch": 5.478954315185547e-05,
      "model_forward_time": 2.291994094848633,
      "step": 57451
    },
    {
      "epoch": 5.478954315185547e-05,
      "step": 57451,
      "training_step_time": 9.50374150276184
    },
    {
      "epoch": 5.479049682617187e-05,
      "model_forward_time": 1.7158098220825195,
      "step": 57452
    },
    {
      "epoch": 5.479049682617187e-05,
      "step": 57452,
      "training_step_time": 8.305137157440186
    },
    {
      "epoch": 5.4791450500488284e-05,
      "model_forward_time": 1.7164061069488525,
      "step": 57453
    },
    {
      "epoch": 5.4791450500488284e-05,
      "step": 57453,
      "training_step_time": 7.6094160079956055
    },
    {
      "epoch": 5.479240417480469e-05,
      "model_forward_time": 1.7409937381744385,
      "step": 57454
    },
    {
      "epoch": 5.479240417480469e-05,
      "step": 57454,
      "training_step_time": 8.640851497650146
    },
    {
      "epoch": 5.479335784912109e-05,
      "model_forward_time": 1.7211549282073975,
      "step": 57455
    },
    {
      "epoch": 5.479335784912109e-05,
      "step": 57455,
      "training_step_time": 7.3776984214782715
    },
    {
      "epoch": 5.47943115234375e-05,
      "model_forward_time": 1.722515344619751,
      "step": 57456
    },
    {
      "epoch": 5.47943115234375e-05,
      "step": 57456,
      "training_step_time": 7.278781890869141
    },
    {
      "epoch": 5.479526519775391e-05,
      "model_forward_time": 1.72039794921875,
      "step": 57457
    },
    {
      "epoch": 5.479526519775391e-05,
      "step": 57457,
      "training_step_time": 7.552337646484375
    },
    {
      "epoch": 5.479621887207031e-05,
      "model_forward_time": 1.7213304042816162,
      "step": 57458
    },
    {
      "epoch": 5.479621887207031e-05,
      "step": 57458,
      "training_step_time": 7.633659839630127
    },
    {
      "epoch": 5.479717254638672e-05,
      "model_forward_time": 1.7145400047302246,
      "step": 57459
    },
    {
      "epoch": 5.479717254638672e-05,
      "step": 57459,
      "training_step_time": 7.854928731918335
    },
    {
      "epoch": 5.4798126220703126e-05,
      "grad_norm": 0.08429422229528427,
      "learning_rate": 4.183714569432259e-06,
      "loss": 0.0562,
      "step": 57460
    },
    {
      "epoch": 5.4798126220703126e-05,
      "model_forward_time": 1.352745771408081,
      "step": 57460
    },
    {
      "dynamics_loss_avg": 0.05667585339397192,
      "epoch": 5.4798126220703126e-05,
      "step": 57460
    },
    {
      "action_loss_avg": 0.0019680494908243416,
      "epoch": 5.4798126220703126e-05,
      "step": 57460
    },
    {
      "epoch": 5.4798126220703126e-05,
      "step": 57460,
      "training_step_time": 7.605609178543091
    },
    {
      "epoch": 5.479907989501953e-05,
      "model_forward_time": 2.2695133686065674,
      "step": 57461
    },
    {
      "epoch": 5.479907989501953e-05,
      "step": 57461,
      "training_step_time": 8.673280239105225
    },
    {
      "epoch": 5.480003356933594e-05,
      "model_forward_time": 1.3520910739898682,
      "step": 57462
    },
    {
      "epoch": 5.480003356933594e-05,
      "step": 57462,
      "training_step_time": 7.56193208694458
    },
    {
      "epoch": 5.4800987243652345e-05,
      "model_forward_time": 1.3514554500579834,
      "step": 57463
    },
    {
      "epoch": 5.4800987243652345e-05,
      "step": 57463,
      "training_step_time": 8.157221794128418
    },
    {
      "epoch": 5.480194091796875e-05,
      "model_forward_time": 1.720588207244873,
      "step": 57464
    },
    {
      "epoch": 5.480194091796875e-05,
      "step": 57464,
      "training_step_time": 7.379086971282959
    },
    {
      "epoch": 5.480289459228516e-05,
      "model_forward_time": 1.7204837799072266,
      "step": 57465
    },
    {
      "epoch": 5.480289459228516e-05,
      "step": 57465,
      "training_step_time": 7.624878644943237
    },
    {
      "epoch": 5.4803848266601564e-05,
      "model_forward_time": 1.7196671962738037,
      "step": 57466
    },
    {
      "epoch": 5.4803848266601564e-05,
      "step": 57466,
      "training_step_time": 7.523342847824097
    },
    {
      "epoch": 5.480480194091797e-05,
      "model_forward_time": 1.7188951969146729,
      "step": 57467
    },
    {
      "epoch": 5.480480194091797e-05,
      "step": 57467,
      "training_step_time": 8.15602159500122
    },
    {
      "epoch": 5.480575561523438e-05,
      "model_forward_time": 1.7185242176055908,
      "step": 57468
    },
    {
      "epoch": 5.480575561523438e-05,
      "step": 57468,
      "training_step_time": 7.815123081207275
    },
    {
      "epoch": 5.480670928955078e-05,
      "model_forward_time": 1.724548101425171,
      "step": 57469
    },
    {
      "epoch": 5.480670928955078e-05,
      "step": 57469,
      "training_step_time": 8.951537609100342
    },
    {
      "epoch": 5.4807662963867186e-05,
      "grad_norm": 0.105923593044281,
      "learning_rate": 4.182083327954296e-06,
      "loss": 0.056,
      "step": 57470
    },
    {
      "epoch": 5.4807662963867186e-05,
      "model_forward_time": 2.2638518810272217,
      "step": 57470
    },
    {
      "dynamics_loss_avg": 0.06131302416324615,
      "epoch": 5.4807662963867186e-05,
      "step": 57470
    },
    {
      "action_loss_avg": 0.0017439938616007566,
      "epoch": 5.4807662963867186e-05,
      "step": 57470
    },
    {
      "epoch": 5.4807662963867186e-05,
      "step": 57470,
      "training_step_time": 9.662556171417236
    },
    {
      "epoch": 5.480861663818359e-05,
      "model_forward_time": 1.726707935333252,
      "step": 57471
    },
    {
      "epoch": 5.480861663818359e-05,
      "step": 57471,
      "training_step_time": 9.207585096359253
    },
    {
      "epoch": 5.48095703125e-05,
      "model_forward_time": 1.7181105613708496,
      "step": 57472
    },
    {
      "epoch": 5.48095703125e-05,
      "step": 57472,
      "training_step_time": 7.584508657455444
    },
    {
      "epoch": 5.4810523986816405e-05,
      "model_forward_time": 1.7361395359039307,
      "step": 57473
    },
    {
      "epoch": 5.4810523986816405e-05,
      "step": 57473,
      "training_step_time": 7.374665021896362
    },
    {
      "epoch": 5.481147766113281e-05,
      "model_forward_time": 1.719275712966919,
      "step": 57474
    },
    {
      "epoch": 5.481147766113281e-05,
      "step": 57474,
      "training_step_time": 7.5576159954071045
    },
    {
      "epoch": 5.481243133544922e-05,
      "model_forward_time": 1.7190911769866943,
      "step": 57475
    },
    {
      "epoch": 5.481243133544922e-05,
      "step": 57475,
      "training_step_time": 7.524274110794067
    },
    {
      "epoch": 5.4813385009765624e-05,
      "model_forward_time": 1.7308270931243896,
      "step": 57476
    },
    {
      "epoch": 5.4813385009765624e-05,
      "step": 57476,
      "training_step_time": 8.355713605880737
    },
    {
      "epoch": 5.481433868408203e-05,
      "model_forward_time": 1.7257874011993408,
      "step": 57477
    },
    {
      "epoch": 5.481433868408203e-05,
      "step": 57477,
      "training_step_time": 8.475634098052979
    },
    {
      "epoch": 5.481529235839844e-05,
      "model_forward_time": 1.715442419052124,
      "step": 57478
    },
    {
      "epoch": 5.481529235839844e-05,
      "step": 57478,
      "training_step_time": 8.494612216949463
    },
    {
      "epoch": 5.481624603271484e-05,
      "model_forward_time": 2.2525405883789062,
      "step": 57479
    },
    {
      "epoch": 5.481624603271484e-05,
      "step": 57479,
      "training_step_time": 8.172860860824585
    },
    {
      "epoch": 5.481719970703125e-05,
      "grad_norm": 0.09012016654014587,
      "learning_rate": 4.180452175922471e-06,
      "loss": 0.0548,
      "step": 57480
    },
    {
      "epoch": 5.481719970703125e-05,
      "model_forward_time": 1.701805591583252,
      "step": 57480
    },
    {
      "dynamics_loss_avg": 0.05878457315266132,
      "epoch": 5.481719970703125e-05,
      "step": 57480
    },
    {
      "action_loss_avg": 0.0018670486693736165,
      "epoch": 5.481719970703125e-05,
      "step": 57480
    },
    {
      "epoch": 5.481719970703125e-05,
      "step": 57480,
      "training_step_time": 8.154352903366089
    },
    {
      "epoch": 5.481815338134766e-05,
      "model_forward_time": 1.707383394241333,
      "step": 57481
    },
    {
      "epoch": 5.481815338134766e-05,
      "step": 57481,
      "training_step_time": 7.533574104309082
    },
    {
      "epoch": 5.481910705566406e-05,
      "model_forward_time": 1.7183609008789062,
      "step": 57482
    },
    {
      "epoch": 5.481910705566406e-05,
      "step": 57482,
      "training_step_time": 7.530515909194946
    },
    {
      "epoch": 5.4820060729980466e-05,
      "model_forward_time": 1.7059249877929688,
      "step": 57483
    },
    {
      "epoch": 5.4820060729980466e-05,
      "step": 57483,
      "training_step_time": 7.5401575565338135
    },
    {
      "epoch": 5.482101440429688e-05,
      "model_forward_time": 1.709209680557251,
      "step": 57484
    },
    {
      "epoch": 5.482101440429688e-05,
      "step": 57484,
      "training_step_time": 7.522622346878052
    },
    {
      "epoch": 5.482196807861328e-05,
      "model_forward_time": 1.705040454864502,
      "step": 57485
    },
    {
      "epoch": 5.482196807861328e-05,
      "step": 57485,
      "training_step_time": 7.423218250274658
    },
    {
      "epoch": 5.4822921752929685e-05,
      "model_forward_time": 1.7180180549621582,
      "step": 57486
    },
    {
      "epoch": 5.4822921752929685e-05,
      "step": 57486,
      "training_step_time": 8.39841628074646
    },
    {
      "epoch": 5.4823875427246096e-05,
      "model_forward_time": 1.7178668975830078,
      "step": 57487
    },
    {
      "epoch": 5.4823875427246096e-05,
      "step": 57487,
      "training_step_time": 8.535158395767212
    },
    {
      "epoch": 5.48248291015625e-05,
      "model_forward_time": 2.266977071762085,
      "step": 57488
    },
    {
      "epoch": 5.48248291015625e-05,
      "step": 57488,
      "training_step_time": 8.503716230392456
    },
    {
      "epoch": 5.4825782775878904e-05,
      "model_forward_time": 1.719226360321045,
      "step": 57489
    },
    {
      "epoch": 5.4825782775878904e-05,
      "step": 57489,
      "training_step_time": 8.407999753952026
    },
    {
      "epoch": 5.4826736450195314e-05,
      "grad_norm": 0.08234884589910507,
      "learning_rate": 4.178821113515166e-06,
      "loss": 0.0541,
      "step": 57490
    },
    {
      "epoch": 5.4826736450195314e-05,
      "model_forward_time": 1.7249302864074707,
      "step": 57490
    },
    {
      "dynamics_loss_avg": 0.05322163784876466,
      "epoch": 5.4826736450195314e-05,
      "step": 57490
    },
    {
      "action_loss_avg": 0.0013677384791662916,
      "epoch": 5.4826736450195314e-05,
      "step": 57490
    },
    {
      "epoch": 5.4826736450195314e-05,
      "step": 57490,
      "training_step_time": 8.168113946914673
    },
    {
      "epoch": 5.482769012451172e-05,
      "model_forward_time": 1.72011399269104,
      "step": 57491
    },
    {
      "epoch": 5.482769012451172e-05,
      "step": 57491,
      "training_step_time": 7.556733131408691
    },
    {
      "epoch": 5.482864379882812e-05,
      "model_forward_time": 1.7367346286773682,
      "step": 57492
    },
    {
      "epoch": 5.482864379882812e-05,
      "step": 57492,
      "training_step_time": 7.125852584838867
    },
    {
      "epoch": 5.482959747314453e-05,
      "model_forward_time": 1.7192473411560059,
      "step": 57493
    },
    {
      "epoch": 5.482959747314453e-05,
      "step": 57493,
      "training_step_time": 7.653979539871216
    },
    {
      "epoch": 5.483055114746094e-05,
      "model_forward_time": 1.717757225036621,
      "step": 57494
    },
    {
      "epoch": 5.483055114746094e-05,
      "step": 57494,
      "training_step_time": 7.644705057144165
    },
    {
      "epoch": 5.483150482177734e-05,
      "model_forward_time": 1.7189881801605225,
      "step": 57495
    },
    {
      "epoch": 5.483150482177734e-05,
      "step": 57495,
      "training_step_time": 8.465028762817383
    },
    {
      "epoch": 5.483245849609375e-05,
      "model_forward_time": 1.724665880203247,
      "step": 57496
    },
    {
      "epoch": 5.483245849609375e-05,
      "step": 57496,
      "training_step_time": 8.417340993881226
    },
    {
      "epoch": 5.4833412170410156e-05,
      "model_forward_time": 2.2569680213928223,
      "step": 57497
    },
    {
      "epoch": 5.4833412170410156e-05,
      "step": 57497,
      "training_step_time": 7.707557201385498
    },
    {
      "epoch": 5.483436584472656e-05,
      "model_forward_time": 1.7196321487426758,
      "step": 57498
    },
    {
      "epoch": 5.483436584472656e-05,
      "step": 57498,
      "training_step_time": 8.619486570358276
    },
    {
      "epoch": 5.483531951904297e-05,
      "model_forward_time": 1.7159397602081299,
      "step": 57499
    },
    {
      "epoch": 5.483531951904297e-05,
      "step": 57499,
      "training_step_time": 8.192686080932617
    },
    {
      "epoch": 5.4836273193359375e-05,
      "grad_norm": 0.08686693012714386,
      "learning_rate": 4.177190140910748e-06,
      "loss": 0.0568,
      "step": 57500
    },
    {
      "epoch": 5.4836273193359375e-05,
      "model_forward_time": 1.7191929817199707,
      "step": 57500
    },
    {
      "dynamics_loss_avg": 0.07629280984401703,
      "epoch": 5.4836273193359375e-05,
      "step": 57500
    },
    {
      "action_loss_avg": 0.002704331133281812,
      "epoch": 5.4836273193359375e-05,
      "step": 57500
    },
    {
      "epoch": 5.4836273193359375e-05,
      "step": 57500,
      "training_step_time": 7.711062669754028
    },
    {
      "epoch": 5.483722686767578e-05,
      "model_forward_time": 1.3690779209136963,
      "step": 57501
    },
    {
      "epoch": 5.483722686767578e-05,
      "step": 57501,
      "training_step_time": 7.4002814292907715
    },
    {
      "epoch": 5.483818054199219e-05,
      "model_forward_time": 1.7243037223815918,
      "step": 57502
    },
    {
      "epoch": 5.483818054199219e-05,
      "step": 57502,
      "training_step_time": 7.704802989959717
    },
    {
      "epoch": 5.4839134216308594e-05,
      "model_forward_time": 1.7212355136871338,
      "step": 57503
    },
    {
      "epoch": 5.4839134216308594e-05,
      "step": 57503,
      "training_step_time": 8.293143510818481
    },
    {
      "epoch": 5.4840087890625e-05,
      "model_forward_time": 1.7176544666290283,
      "step": 57504
    },
    {
      "epoch": 5.4840087890625e-05,
      "step": 57504,
      "training_step_time": 7.602085113525391
    },
    {
      "epoch": 5.484104156494141e-05,
      "model_forward_time": 1.7207648754119873,
      "step": 57505
    },
    {
      "epoch": 5.484104156494141e-05,
      "step": 57505,
      "training_step_time": 7.736114025115967
    },
    {
      "epoch": 5.484199523925781e-05,
      "model_forward_time": 1.3531582355499268,
      "step": 57506
    },
    {
      "epoch": 5.484199523925781e-05,
      "step": 57506,
      "training_step_time": 9.174326419830322
    },
    {
      "epoch": 5.484294891357422e-05,
      "model_forward_time": 1.554795503616333,
      "step": 57507
    },
    {
      "epoch": 5.484294891357422e-05,
      "step": 57507,
      "training_step_time": 8.20323133468628
    },
    {
      "epoch": 5.484390258789063e-05,
      "model_forward_time": 1.7185609340667725,
      "step": 57508
    },
    {
      "epoch": 5.484390258789063e-05,
      "step": 57508,
      "training_step_time": 7.568560838699341
    },
    {
      "epoch": 5.484485626220703e-05,
      "model_forward_time": 1.7230637073516846,
      "step": 57509
    },
    {
      "epoch": 5.484485626220703e-05,
      "step": 57509,
      "training_step_time": 7.6370627880096436
    },
    {
      "epoch": 5.4845809936523436e-05,
      "grad_norm": 0.10715232789516449,
      "learning_rate": 4.175559258287577e-06,
      "loss": 0.0568,
      "step": 57510
    },
    {
      "epoch": 5.4845809936523436e-05,
      "model_forward_time": 1.7329978942871094,
      "step": 57510
    },
    {
      "dynamics_loss_avg": 0.05518887061625719,
      "epoch": 5.4845809936523436e-05,
      "step": 57510
    },
    {
      "action_loss_avg": 0.001541020511649549,
      "epoch": 5.4845809936523436e-05,
      "step": 57510
    },
    {
      "epoch": 5.4845809936523436e-05,
      "step": 57510,
      "training_step_time": 7.697941064834595
    },
    {
      "epoch": 5.4846763610839846e-05,
      "model_forward_time": 1.723081350326538,
      "step": 57511
    },
    {
      "epoch": 5.4846763610839846e-05,
      "step": 57511,
      "training_step_time": 7.926822185516357
    },
    {
      "epoch": 5.484771728515625e-05,
      "model_forward_time": 1.719672441482544,
      "step": 57512
    },
    {
      "epoch": 5.484771728515625e-05,
      "step": 57512,
      "training_step_time": 7.821143865585327
    },
    {
      "epoch": 5.4848670959472654e-05,
      "model_forward_time": 1.7160918712615967,
      "step": 57513
    },
    {
      "epoch": 5.4848670959472654e-05,
      "step": 57513,
      "training_step_time": 8.980678796768188
    },
    {
      "epoch": 5.4849624633789065e-05,
      "model_forward_time": 1.7169358730316162,
      "step": 57514
    },
    {
      "epoch": 5.4849624633789065e-05,
      "step": 57514,
      "training_step_time": 8.050698280334473
    },
    {
      "epoch": 5.485057830810547e-05,
      "model_forward_time": 1.7289857864379883,
      "step": 57515
    },
    {
      "epoch": 5.485057830810547e-05,
      "step": 57515,
      "training_step_time": 10.33584475517273
    },
    {
      "epoch": 5.485153198242187e-05,
      "model_forward_time": 1.0156421661376953,
      "step": 57516
    },
    {
      "epoch": 5.485153198242187e-05,
      "step": 57516,
      "training_step_time": 7.809880971908569
    },
    {
      "epoch": 5.4852485656738284e-05,
      "model_forward_time": 2.2695815563201904,
      "step": 57517
    },
    {
      "epoch": 5.4852485656738284e-05,
      "step": 57517,
      "training_step_time": 7.753881216049194
    },
    {
      "epoch": 5.485343933105469e-05,
      "model_forward_time": 1.721804141998291,
      "step": 57518
    },
    {
      "epoch": 5.485343933105469e-05,
      "step": 57518,
      "training_step_time": 7.620043754577637
    },
    {
      "epoch": 5.485439300537109e-05,
      "model_forward_time": 1.732530117034912,
      "step": 57519
    },
    {
      "epoch": 5.485439300537109e-05,
      "step": 57519,
      "training_step_time": 7.586101293563843
    },
    {
      "epoch": 5.48553466796875e-05,
      "grad_norm": 0.1636420637369156,
      "learning_rate": 4.173928465824009e-06,
      "loss": 0.0555,
      "step": 57520
    },
    {
      "epoch": 5.48553466796875e-05,
      "model_forward_time": 1.7158589363098145,
      "step": 57520
    },
    {
      "dynamics_loss_avg": 0.04096545958891511,
      "epoch": 5.48553466796875e-05,
      "step": 57520
    },
    {
      "action_loss_avg": 0.0013778607331914826,
      "epoch": 5.48553466796875e-05,
      "step": 57520
    },
    {
      "epoch": 5.48553466796875e-05,
      "step": 57520,
      "training_step_time": 7.62798810005188
    },
    {
      "epoch": 5.485630035400391e-05,
      "model_forward_time": 1.7253239154815674,
      "step": 57521
    },
    {
      "epoch": 5.485630035400391e-05,
      "step": 57521,
      "training_step_time": 8.395719051361084
    },
    {
      "epoch": 5.485725402832031e-05,
      "model_forward_time": 1.7369873523712158,
      "step": 57522
    },
    {
      "epoch": 5.485725402832031e-05,
      "step": 57522,
      "training_step_time": 8.079700231552124
    },
    {
      "epoch": 5.485820770263672e-05,
      "model_forward_time": 1.7223789691925049,
      "step": 57523
    },
    {
      "epoch": 5.485820770263672e-05,
      "step": 57523,
      "training_step_time": 7.78938102722168
    },
    {
      "epoch": 5.4859161376953126e-05,
      "model_forward_time": 1.7173149585723877,
      "step": 57524
    },
    {
      "epoch": 5.4859161376953126e-05,
      "step": 57524,
      "training_step_time": 9.146110773086548
    },
    {
      "epoch": 5.486011505126953e-05,
      "model_forward_time": 1.7177479267120361,
      "step": 57525
    },
    {
      "epoch": 5.486011505126953e-05,
      "step": 57525,
      "training_step_time": 8.68680477142334
    },
    {
      "epoch": 5.486106872558594e-05,
      "model_forward_time": 2.2776968479156494,
      "step": 57526
    },
    {
      "epoch": 5.486106872558594e-05,
      "step": 57526,
      "training_step_time": 7.640424013137817
    },
    {
      "epoch": 5.4862022399902345e-05,
      "model_forward_time": 1.7178068161010742,
      "step": 57527
    },
    {
      "epoch": 5.4862022399902345e-05,
      "step": 57527,
      "training_step_time": 8.13230848312378
    },
    {
      "epoch": 5.486297607421875e-05,
      "model_forward_time": 1.7216856479644775,
      "step": 57528
    },
    {
      "epoch": 5.486297607421875e-05,
      "step": 57528,
      "training_step_time": 7.554297208786011
    },
    {
      "epoch": 5.486392974853516e-05,
      "model_forward_time": 1.7354958057403564,
      "step": 57529
    },
    {
      "epoch": 5.486392974853516e-05,
      "step": 57529,
      "training_step_time": 7.466344833374023
    },
    {
      "epoch": 5.4864883422851564e-05,
      "grad_norm": 0.08565287292003632,
      "learning_rate": 4.172297763698381e-06,
      "loss": 0.0568,
      "step": 57530
    },
    {
      "epoch": 5.4864883422851564e-05,
      "model_forward_time": 1.7138636112213135,
      "step": 57530
    },
    {
      "dynamics_loss_avg": 0.05580514464527368,
      "epoch": 5.4864883422851564e-05,
      "step": 57530
    },
    {
      "action_loss_avg": 0.002131115924566984,
      "epoch": 5.4864883422851564e-05,
      "step": 57530
    },
    {
      "epoch": 5.4864883422851564e-05,
      "step": 57530,
      "training_step_time": 7.297956228256226
    },
    {
      "epoch": 5.486583709716797e-05,
      "model_forward_time": 1.7193574905395508,
      "step": 57531
    },
    {
      "epoch": 5.486583709716797e-05,
      "step": 57531,
      "training_step_time": 7.4892191886901855
    },
    {
      "epoch": 5.486679077148437e-05,
      "model_forward_time": 1.7251780033111572,
      "step": 57532
    },
    {
      "epoch": 5.486679077148437e-05,
      "step": 57532,
      "training_step_time": 7.588605165481567
    },
    {
      "epoch": 5.486774444580078e-05,
      "model_forward_time": 1.7170507907867432,
      "step": 57533
    },
    {
      "epoch": 5.486774444580078e-05,
      "step": 57533,
      "training_step_time": 7.611132621765137
    },
    {
      "epoch": 5.4868698120117186e-05,
      "model_forward_time": 1.7360780239105225,
      "step": 57534
    },
    {
      "epoch": 5.4868698120117186e-05,
      "step": 57534,
      "training_step_time": 7.808877468109131
    },
    {
      "epoch": 5.486965179443359e-05,
      "model_forward_time": 2.3087332248687744,
      "step": 57535
    },
    {
      "epoch": 5.486965179443359e-05,
      "step": 57535,
      "training_step_time": 8.377130031585693
    },
    {
      "epoch": 5.487060546875e-05,
      "model_forward_time": 1.7223436832427979,
      "step": 57536
    },
    {
      "epoch": 5.487060546875e-05,
      "step": 57536,
      "training_step_time": 8.083644151687622
    },
    {
      "epoch": 5.4871559143066405e-05,
      "model_forward_time": 1.7202649116516113,
      "step": 57537
    },
    {
      "epoch": 5.4871559143066405e-05,
      "step": 57537,
      "training_step_time": 7.578031063079834
    },
    {
      "epoch": 5.487251281738281e-05,
      "model_forward_time": 1.744825839996338,
      "step": 57538
    },
    {
      "epoch": 5.487251281738281e-05,
      "step": 57538,
      "training_step_time": 7.617102146148682
    },
    {
      "epoch": 5.487346649169922e-05,
      "model_forward_time": 1.7240078449249268,
      "step": 57539
    },
    {
      "epoch": 5.487346649169922e-05,
      "step": 57539,
      "training_step_time": 7.587372303009033
    },
    {
      "epoch": 5.4874420166015624e-05,
      "grad_norm": 0.11121740192174911,
      "learning_rate": 4.170667152089025e-06,
      "loss": 0.0541,
      "step": 57540
    },
    {
      "epoch": 5.4874420166015624e-05,
      "model_forward_time": 1.724654197692871,
      "step": 57540
    },
    {
      "dynamics_loss_avg": 0.06062263697385788,
      "epoch": 5.4874420166015624e-05,
      "step": 57540
    },
    {
      "action_loss_avg": 0.0020442319219000638,
      "epoch": 5.4874420166015624e-05,
      "step": 57540
    },
    {
      "epoch": 5.4874420166015624e-05,
      "step": 57540,
      "training_step_time": 8.479673385620117
    },
    {
      "epoch": 5.487537384033203e-05,
      "model_forward_time": 1.7239677906036377,
      "step": 57541
    },
    {
      "epoch": 5.487537384033203e-05,
      "step": 57541,
      "training_step_time": 7.635908365249634
    },
    {
      "epoch": 5.487632751464844e-05,
      "model_forward_time": 1.7263448238372803,
      "step": 57542
    },
    {
      "epoch": 5.487632751464844e-05,
      "step": 57542,
      "training_step_time": 7.6573286056518555
    },
    {
      "epoch": 5.487728118896484e-05,
      "model_forward_time": 1.7713029384613037,
      "step": 57543
    },
    {
      "epoch": 5.487728118896484e-05,
      "step": 57543,
      "training_step_time": 9.124519109725952
    },
    {
      "epoch": 5.487823486328125e-05,
      "model_forward_time": 2.899920701980591,
      "step": 57544
    },
    {
      "epoch": 5.487823486328125e-05,
      "step": 57544,
      "training_step_time": 9.92181944847107
    },
    {
      "epoch": 5.487918853759766e-05,
      "model_forward_time": 1.746912956237793,
      "step": 57545
    },
    {
      "epoch": 5.487918853759766e-05,
      "step": 57545,
      "training_step_time": 7.5133819580078125
    },
    {
      "epoch": 5.488014221191406e-05,
      "model_forward_time": 1.755126714706421,
      "step": 57546
    },
    {
      "epoch": 5.488014221191406e-05,
      "step": 57546,
      "training_step_time": 8.622358083724976
    },
    {
      "epoch": 5.4881095886230466e-05,
      "model_forward_time": 1.757843017578125,
      "step": 57547
    },
    {
      "epoch": 5.4881095886230466e-05,
      "step": 57547,
      "training_step_time": 7.534301519393921
    },
    {
      "epoch": 5.488204956054688e-05,
      "model_forward_time": 1.7234375476837158,
      "step": 57548
    },
    {
      "epoch": 5.488204956054688e-05,
      "step": 57548,
      "training_step_time": 8.404005289077759
    },
    {
      "epoch": 5.488300323486328e-05,
      "model_forward_time": 1.7434470653533936,
      "step": 57549
    },
    {
      "epoch": 5.488300323486328e-05,
      "step": 57549,
      "training_step_time": 7.284055709838867
    },
    {
      "epoch": 5.4883956909179685e-05,
      "grad_norm": 0.07584313303232193,
      "learning_rate": 4.169036631174263e-06,
      "loss": 0.0536,
      "step": 57550
    },
    {
      "epoch": 5.4883956909179685e-05,
      "model_forward_time": 1.726426601409912,
      "step": 57550
    },
    {
      "dynamics_loss_avg": 0.045560807920992374,
      "epoch": 5.4883956909179685e-05,
      "step": 57550
    },
    {
      "action_loss_avg": 0.0014861724455840885,
      "epoch": 5.4883956909179685e-05,
      "step": 57550
    },
    {
      "epoch": 5.4883956909179685e-05,
      "step": 57550,
      "training_step_time": 7.584182024002075
    },
    {
      "epoch": 5.4884910583496096e-05,
      "model_forward_time": 1.7318990230560303,
      "step": 57551
    },
    {
      "epoch": 5.4884910583496096e-05,
      "step": 57551,
      "training_step_time": 8.225202083587646
    },
    {
      "epoch": 5.48858642578125e-05,
      "model_forward_time": 1.746201515197754,
      "step": 57552
    },
    {
      "epoch": 5.48858642578125e-05,
      "step": 57552,
      "training_step_time": 7.583065748214722
    },
    {
      "epoch": 5.4886817932128904e-05,
      "model_forward_time": 1.8399741649627686,
      "step": 57553
    },
    {
      "epoch": 5.4886817932128904e-05,
      "step": 57553,
      "training_step_time": 8.41472840309143
    },
    {
      "epoch": 5.4887771606445314e-05,
      "model_forward_time": 2.9106621742248535,
      "step": 57554
    },
    {
      "epoch": 5.4887771606445314e-05,
      "step": 57554,
      "training_step_time": 7.622185707092285
    },
    {
      "epoch": 5.488872528076172e-05,
      "model_forward_time": 1.736285924911499,
      "step": 57555
    },
    {
      "epoch": 5.488872528076172e-05,
      "step": 57555,
      "training_step_time": 7.5905327796936035
    },
    {
      "epoch": 5.488967895507812e-05,
      "model_forward_time": 1.7341556549072266,
      "step": 57556
    },
    {
      "epoch": 5.488967895507812e-05,
      "step": 57556,
      "training_step_time": 7.577854871749878
    },
    {
      "epoch": 5.489063262939453e-05,
      "model_forward_time": 1.7362010478973389,
      "step": 57557
    },
    {
      "epoch": 5.489063262939453e-05,
      "step": 57557,
      "training_step_time": 7.573743581771851
    },
    {
      "epoch": 5.489158630371094e-05,
      "model_forward_time": 1.7498395442962646,
      "step": 57558
    },
    {
      "epoch": 5.489158630371094e-05,
      "step": 57558,
      "training_step_time": 7.628150463104248
    },
    {
      "epoch": 5.489253997802734e-05,
      "model_forward_time": 1.731027364730835,
      "step": 57559
    },
    {
      "epoch": 5.489253997802734e-05,
      "step": 57559,
      "training_step_time": 8.131536483764648
    },
    {
      "epoch": 5.489349365234375e-05,
      "grad_norm": 0.10750817507505417,
      "learning_rate": 4.167406201132406e-06,
      "loss": 0.0546,
      "step": 57560
    },
    {
      "epoch": 5.489349365234375e-05,
      "model_forward_time": 1.715388298034668,
      "step": 57560
    },
    {
      "dynamics_loss_avg": 0.060138840414583686,
      "epoch": 5.489349365234375e-05,
      "step": 57560
    },
    {
      "action_loss_avg": 0.0019853673758916555,
      "epoch": 5.489349365234375e-05,
      "step": 57560
    },
    {
      "epoch": 5.489349365234375e-05,
      "step": 57560,
      "training_step_time": 7.6796019077301025
    },
    {
      "epoch": 5.4894447326660156e-05,
      "model_forward_time": 1.7455158233642578,
      "step": 57561
    },
    {
      "epoch": 5.4894447326660156e-05,
      "step": 57561,
      "training_step_time": 10.03828740119934
    },
    {
      "epoch": 5.489540100097656e-05,
      "model_forward_time": 1.4907631874084473,
      "step": 57562
    },
    {
      "epoch": 5.489540100097656e-05,
      "step": 57562,
      "training_step_time": 10.1399986743927
    },
    {
      "epoch": 5.489635467529297e-05,
      "model_forward_time": 2.8814775943756104,
      "step": 57563
    },
    {
      "epoch": 5.489635467529297e-05,
      "step": 57563,
      "training_step_time": 7.776631832122803
    },
    {
      "epoch": 5.4897308349609375e-05,
      "model_forward_time": 1.7252967357635498,
      "step": 57564
    },
    {
      "epoch": 5.4897308349609375e-05,
      "step": 57564,
      "training_step_time": 7.6333699226379395
    },
    {
      "epoch": 5.489826202392578e-05,
      "model_forward_time": 1.8222780227661133,
      "step": 57565
    },
    {
      "epoch": 5.489826202392578e-05,
      "step": 57565,
      "training_step_time": 7.608189105987549
    },
    {
      "epoch": 5.489921569824219e-05,
      "model_forward_time": 1.837223768234253,
      "step": 57566
    },
    {
      "epoch": 5.489921569824219e-05,
      "step": 57566,
      "training_step_time": 7.618424892425537
    },
    {
      "epoch": 5.4900169372558594e-05,
      "model_forward_time": 1.8440592288970947,
      "step": 57567
    },
    {
      "epoch": 5.4900169372558594e-05,
      "step": 57567,
      "training_step_time": 7.5781331062316895
    },
    {
      "epoch": 5.4901123046875e-05,
      "model_forward_time": 1.828953742980957,
      "step": 57568
    },
    {
      "epoch": 5.4901123046875e-05,
      "step": 57568,
      "training_step_time": 7.6272172927856445
    },
    {
      "epoch": 5.490207672119141e-05,
      "model_forward_time": 1.8409111499786377,
      "step": 57569
    },
    {
      "epoch": 5.490207672119141e-05,
      "step": 57569,
      "training_step_time": 9.313911199569702
    },
    {
      "epoch": 5.490303039550781e-05,
      "grad_norm": 0.10871250927448273,
      "learning_rate": 4.165775862141752e-06,
      "loss": 0.0551,
      "step": 57570
    },
    {
      "epoch": 5.490303039550781e-05,
      "model_forward_time": 1.8424174785614014,
      "step": 57570
    },
    {
      "dynamics_loss_avg": 0.054999133385717866,
      "epoch": 5.490303039550781e-05,
      "step": 57570
    },
    {
      "action_loss_avg": 0.0013928706932347268,
      "epoch": 5.490303039550781e-05,
      "step": 57570
    },
    {
      "epoch": 5.490303039550781e-05,
      "step": 57570,
      "training_step_time": 7.632217645645142
    },
    {
      "epoch": 5.490398406982422e-05,
      "model_forward_time": 1.8235993385314941,
      "step": 57571
    },
    {
      "epoch": 5.490398406982422e-05,
      "step": 57571,
      "training_step_time": 8.268624305725098
    },
    {
      "epoch": 5.490493774414063e-05,
      "model_forward_time": 2.3463690280914307,
      "step": 57572
    },
    {
      "epoch": 5.490493774414063e-05,
      "step": 57572,
      "training_step_time": 8.285597562789917
    },
    {
      "epoch": 5.490589141845703e-05,
      "model_forward_time": 1.8253824710845947,
      "step": 57573
    },
    {
      "epoch": 5.490589141845703e-05,
      "step": 57573,
      "training_step_time": 7.622222661972046
    },
    {
      "epoch": 5.4906845092773436e-05,
      "model_forward_time": 1.8248872756958008,
      "step": 57574
    },
    {
      "epoch": 5.4906845092773436e-05,
      "step": 57574,
      "training_step_time": 7.644405364990234
    },
    {
      "epoch": 5.4907798767089846e-05,
      "model_forward_time": 1.8357634544372559,
      "step": 57575
    },
    {
      "epoch": 5.4907798767089846e-05,
      "step": 57575,
      "training_step_time": 8.142628908157349
    },
    {
      "epoch": 5.490875244140625e-05,
      "model_forward_time": 1.8353815078735352,
      "step": 57576
    },
    {
      "epoch": 5.490875244140625e-05,
      "step": 57576,
      "training_step_time": 7.539355039596558
    },
    {
      "epoch": 5.4909706115722655e-05,
      "model_forward_time": 1.8306920528411865,
      "step": 57577
    },
    {
      "epoch": 5.4909706115722655e-05,
      "step": 57577,
      "training_step_time": 7.758445501327515
    },
    {
      "epoch": 5.4910659790039065e-05,
      "model_forward_time": 1.8267662525177002,
      "step": 57578
    },
    {
      "epoch": 5.4910659790039065e-05,
      "step": 57578,
      "training_step_time": 8.73808741569519
    },
    {
      "epoch": 5.491161346435547e-05,
      "model_forward_time": 1.8292851448059082,
      "step": 57579
    },
    {
      "epoch": 5.491161346435547e-05,
      "step": 57579,
      "training_step_time": 9.357477903366089
    },
    {
      "epoch": 5.491256713867187e-05,
      "grad_norm": 0.1364421248435974,
      "learning_rate": 4.164145614380598e-06,
      "loss": 0.0547,
      "step": 57580
    },
    {
      "epoch": 5.491256713867187e-05,
      "model_forward_time": 1.455005407333374,
      "step": 57580
    },
    {
      "dynamics_loss_avg": 0.0529652688652277,
      "epoch": 5.491256713867187e-05,
      "step": 57580
    },
    {
      "action_loss_avg": 0.0017661141406279056,
      "epoch": 5.491256713867187e-05,
      "step": 57580
    },
    {
      "epoch": 5.491256713867187e-05,
      "step": 57580,
      "training_step_time": 7.803186893463135
    },
    {
      "epoch": 5.4913520812988284e-05,
      "model_forward_time": 2.385976552963257,
      "step": 57581
    },
    {
      "epoch": 5.4913520812988284e-05,
      "step": 57581,
      "training_step_time": 7.82350754737854
    },
    {
      "epoch": 5.491447448730469e-05,
      "model_forward_time": 1.831355094909668,
      "step": 57582
    },
    {
      "epoch": 5.491447448730469e-05,
      "step": 57582,
      "training_step_time": 8.979148626327515
    },
    {
      "epoch": 5.491542816162109e-05,
      "model_forward_time": 1.8334112167358398,
      "step": 57583
    },
    {
      "epoch": 5.491542816162109e-05,
      "step": 57583,
      "training_step_time": 7.754145383834839
    },
    {
      "epoch": 5.49163818359375e-05,
      "model_forward_time": 1.8370096683502197,
      "step": 57584
    },
    {
      "epoch": 5.49163818359375e-05,
      "step": 57584,
      "training_step_time": 7.602222204208374
    },
    {
      "epoch": 5.491733551025391e-05,
      "model_forward_time": 1.829902172088623,
      "step": 57585
    },
    {
      "epoch": 5.491733551025391e-05,
      "step": 57585,
      "training_step_time": 7.617461442947388
    },
    {
      "epoch": 5.491828918457031e-05,
      "model_forward_time": 1.833763599395752,
      "step": 57586
    },
    {
      "epoch": 5.491828918457031e-05,
      "step": 57586,
      "training_step_time": 7.616469383239746
    },
    {
      "epoch": 5.491924285888672e-05,
      "model_forward_time": 1.821516513824463,
      "step": 57587
    },
    {
      "epoch": 5.491924285888672e-05,
      "step": 57587,
      "training_step_time": 8.571604490280151
    },
    {
      "epoch": 5.4920196533203126e-05,
      "model_forward_time": 1.8350207805633545,
      "step": 57588
    },
    {
      "epoch": 5.4920196533203126e-05,
      "step": 57588,
      "training_step_time": 7.618639945983887
    },
    {
      "epoch": 5.492115020751953e-05,
      "model_forward_time": 1.830012559890747,
      "step": 57589
    },
    {
      "epoch": 5.492115020751953e-05,
      "step": 57589,
      "training_step_time": 7.830930471420288
    },
    {
      "epoch": 5.492210388183594e-05,
      "grad_norm": 0.1266159564256668,
      "learning_rate": 4.162515458027223e-06,
      "loss": 0.0567,
      "step": 57590
    },
    {
      "epoch": 5.492210388183594e-05,
      "model_forward_time": 2.3577945232391357,
      "step": 57590
    },
    {
      "dynamics_loss_avg": 0.041031552478671074,
      "epoch": 5.492210388183594e-05,
      "step": 57590
    },
    {
      "action_loss_avg": 0.0011298951285425574,
      "epoch": 5.492210388183594e-05,
      "step": 57590
    },
    {
      "epoch": 5.492210388183594e-05,
      "step": 57590,
      "training_step_time": 7.878732442855835
    },
    {
      "epoch": 5.4923057556152345e-05,
      "model_forward_time": 1.816253662109375,
      "step": 57591
    },
    {
      "epoch": 5.4923057556152345e-05,
      "step": 57591,
      "training_step_time": 8.839807271957397
    },
    {
      "epoch": 5.492401123046875e-05,
      "model_forward_time": 1.4597764015197754,
      "step": 57592
    },
    {
      "epoch": 5.492401123046875e-05,
      "step": 57592,
      "training_step_time": 8.393030166625977
    },
    {
      "epoch": 5.492496490478516e-05,
      "model_forward_time": 1.8238167762756348,
      "step": 57593
    },
    {
      "epoch": 5.492496490478516e-05,
      "step": 57593,
      "training_step_time": 7.725670576095581
    },
    {
      "epoch": 5.4925918579101564e-05,
      "model_forward_time": 1.8284876346588135,
      "step": 57594
    },
    {
      "epoch": 5.4925918579101564e-05,
      "step": 57594,
      "training_step_time": 7.718409776687622
    },
    {
      "epoch": 5.492687225341797e-05,
      "model_forward_time": 1.829103946685791,
      "step": 57595
    },
    {
      "epoch": 5.492687225341797e-05,
      "step": 57595,
      "training_step_time": 8.75097918510437
    },
    {
      "epoch": 5.492782592773437e-05,
      "model_forward_time": 1.8272223472595215,
      "step": 57596
    },
    {
      "epoch": 5.492782592773437e-05,
      "step": 57596,
      "training_step_time": 7.840870141983032
    },
    {
      "epoch": 5.492877960205078e-05,
      "model_forward_time": 1.8395075798034668,
      "step": 57597
    },
    {
      "epoch": 5.492877960205078e-05,
      "step": 57597,
      "training_step_time": 7.666184186935425
    },
    {
      "epoch": 5.4929733276367187e-05,
      "model_forward_time": 1.4716129302978516,
      "step": 57598
    },
    {
      "epoch": 5.4929733276367187e-05,
      "step": 57598,
      "training_step_time": 8.65015721321106
    },
    {
      "epoch": 5.493068695068359e-05,
      "model_forward_time": 1.8280611038208008,
      "step": 57599
    },
    {
      "epoch": 5.493068695068359e-05,
      "step": 57599,
      "training_step_time": 8.29520559310913
    },
    {
      "epoch": 5.4931640625e-05,
      "grad_norm": 0.07415061444044113,
      "learning_rate": 4.160885393259899e-06,
      "loss": 0.0522,
      "step": 57600
    },
    {
      "epoch": 5.4931640625e-05,
      "model_forward_time": 2.3594577312469482,
      "step": 57600
    },
    {
      "dynamics_loss_avg": 0.06349169332534074,
      "epoch": 5.4931640625e-05,
      "step": 57600
    },
    {
      "action_loss_avg": 0.0016064643976278603,
      "epoch": 5.4931640625e-05,
      "step": 57600
    },
    {
      "epoch": 5.4931640625e-05,
      "step": 57600,
      "training_step_time": 7.673857927322388
    },
    {
      "epoch": 5.4932594299316405e-05,
      "model_forward_time": 1.8375191688537598,
      "step": 57601
    },
    {
      "epoch": 5.4932594299316405e-05,
      "step": 57601,
      "training_step_time": 7.699244499206543
    },
    {
      "epoch": 5.493354797363281e-05,
      "model_forward_time": 1.8266637325286865,
      "step": 57602
    },
    {
      "epoch": 5.493354797363281e-05,
      "step": 57602,
      "training_step_time": 7.718966484069824
    },
    {
      "epoch": 5.493450164794922e-05,
      "model_forward_time": 1.831237554550171,
      "step": 57603
    },
    {
      "epoch": 5.493450164794922e-05,
      "step": 57603,
      "training_step_time": 7.8209497928619385
    },
    {
      "epoch": 5.4935455322265624e-05,
      "model_forward_time": 1.8391449451446533,
      "step": 57604
    },
    {
      "epoch": 5.4935455322265624e-05,
      "step": 57604,
      "training_step_time": 7.7858123779296875
    },
    {
      "epoch": 5.493640899658203e-05,
      "model_forward_time": 1.8194026947021484,
      "step": 57605
    },
    {
      "epoch": 5.493640899658203e-05,
      "step": 57605,
      "training_step_time": 7.836784362792969
    },
    {
      "epoch": 5.493736267089844e-05,
      "model_forward_time": 1.8201441764831543,
      "step": 57606
    },
    {
      "epoch": 5.493736267089844e-05,
      "step": 57606,
      "training_step_time": 7.6469151973724365
    },
    {
      "epoch": 5.493831634521484e-05,
      "model_forward_time": 1.823817491531372,
      "step": 57607
    },
    {
      "epoch": 5.493831634521484e-05,
      "step": 57607,
      "training_step_time": 9.215035676956177
    },
    {
      "epoch": 5.493927001953125e-05,
      "model_forward_time": 1.822676181793213,
      "step": 57608
    },
    {
      "epoch": 5.493927001953125e-05,
      "step": 57608,
      "training_step_time": 7.81895637512207
    },
    {
      "epoch": 5.494022369384766e-05,
      "model_forward_time": 2.396373987197876,
      "step": 57609
    },
    {
      "epoch": 5.494022369384766e-05,
      "step": 57609,
      "training_step_time": 8.727056741714478
    },
    {
      "epoch": 5.494117736816406e-05,
      "grad_norm": 0.08897177129983902,
      "learning_rate": 4.159255420256885e-06,
      "loss": 0.0569,
      "step": 57610
    },
    {
      "epoch": 5.494117736816406e-05,
      "model_forward_time": 1.826246976852417,
      "step": 57610
    },
    {
      "dynamics_loss_avg": 0.07195723820477724,
      "epoch": 5.494117736816406e-05,
      "step": 57610
    },
    {
      "action_loss_avg": 0.001882027235114947,
      "epoch": 5.494117736816406e-05,
      "step": 57610
    },
    {
      "epoch": 5.494117736816406e-05,
      "step": 57610,
      "training_step_time": 7.836931467056274
    },
    {
      "epoch": 5.4942131042480466e-05,
      "model_forward_time": 1.8276405334472656,
      "step": 57611
    },
    {
      "epoch": 5.4942131042480466e-05,
      "step": 57611,
      "training_step_time": 7.715622901916504
    },
    {
      "epoch": 5.494308471679688e-05,
      "model_forward_time": 1.823312759399414,
      "step": 57612
    },
    {
      "epoch": 5.494308471679688e-05,
      "step": 57612,
      "training_step_time": 7.7207863330841064
    },
    {
      "epoch": 5.494403839111328e-05,
      "model_forward_time": 1.8332505226135254,
      "step": 57613
    },
    {
      "epoch": 5.494403839111328e-05,
      "step": 57613,
      "training_step_time": 7.7208147048950195
    },
    {
      "epoch": 5.4944992065429685e-05,
      "model_forward_time": 1.8318636417388916,
      "step": 57614
    },
    {
      "epoch": 5.4944992065429685e-05,
      "step": 57614,
      "training_step_time": 8.392368078231812
    },
    {
      "epoch": 5.4945945739746096e-05,
      "model_forward_time": 1.8226900100708008,
      "step": 57615
    },
    {
      "epoch": 5.4945945739746096e-05,
      "step": 57615,
      "training_step_time": 7.69080662727356
    },
    {
      "epoch": 5.49468994140625e-05,
      "model_forward_time": 1.8016924858093262,
      "step": 57616
    },
    {
      "epoch": 5.49468994140625e-05,
      "step": 57616,
      "training_step_time": 7.655561447143555
    },
    {
      "epoch": 5.4947853088378904e-05,
      "model_forward_time": 1.8171427249908447,
      "step": 57617
    },
    {
      "epoch": 5.4947853088378904e-05,
      "step": 57617,
      "training_step_time": 7.767217397689819
    },
    {
      "epoch": 5.4948806762695315e-05,
      "model_forward_time": 2.3537657260894775,
      "step": 57618
    },
    {
      "epoch": 5.4948806762695315e-05,
      "step": 57618,
      "training_step_time": 7.773483037948608
    },
    {
      "epoch": 5.494976043701172e-05,
      "model_forward_time": 1.8304719924926758,
      "step": 57619
    },
    {
      "epoch": 5.494976043701172e-05,
      "step": 57619,
      "training_step_time": 7.640427112579346
    },
    {
      "epoch": 5.495071411132812e-05,
      "grad_norm": 0.09010418504476547,
      "learning_rate": 4.157625539196436e-06,
      "loss": 0.0553,
      "step": 57620
    },
    {
      "epoch": 5.495071411132812e-05,
      "model_forward_time": 1.4618103504180908,
      "step": 57620
    },
    {
      "dynamics_loss_avg": 0.05879303850233555,
      "epoch": 5.495071411132812e-05,
      "step": 57620
    },
    {
      "action_loss_avg": 0.002572846339899115,
      "epoch": 5.495071411132812e-05,
      "step": 57620
    },
    {
      "epoch": 5.495071411132812e-05,
      "step": 57620,
      "training_step_time": 8.354822158813477
    },
    {
      "epoch": 5.495166778564453e-05,
      "model_forward_time": 1.8039469718933105,
      "step": 57621
    },
    {
      "epoch": 5.495166778564453e-05,
      "step": 57621,
      "training_step_time": 7.606405019760132
    },
    {
      "epoch": 5.495262145996094e-05,
      "model_forward_time": 1.824195146560669,
      "step": 57622
    },
    {
      "epoch": 5.495262145996094e-05,
      "step": 57622,
      "training_step_time": 8.660846471786499
    },
    {
      "epoch": 5.495357513427734e-05,
      "model_forward_time": 1.8190255165100098,
      "step": 57623
    },
    {
      "epoch": 5.495357513427734e-05,
      "step": 57623,
      "training_step_time": 7.679902076721191
    },
    {
      "epoch": 5.495452880859375e-05,
      "model_forward_time": 1.8331294059753418,
      "step": 57624
    },
    {
      "epoch": 5.495452880859375e-05,
      "step": 57624,
      "training_step_time": 8.941778421401978
    },
    {
      "epoch": 5.4955482482910156e-05,
      "model_forward_time": 1.8107850551605225,
      "step": 57625
    },
    {
      "epoch": 5.4955482482910156e-05,
      "step": 57625,
      "training_step_time": 7.6583251953125
    },
    {
      "epoch": 5.495643615722656e-05,
      "model_forward_time": 1.8362255096435547,
      "step": 57626
    },
    {
      "epoch": 5.495643615722656e-05,
      "step": 57626,
      "training_step_time": 9.14031982421875
    },
    {
      "epoch": 5.495738983154297e-05,
      "model_forward_time": 1.8217196464538574,
      "step": 57627
    },
    {
      "epoch": 5.495738983154297e-05,
      "step": 57627,
      "training_step_time": 8.45391583442688
    },
    {
      "epoch": 5.4958343505859375e-05,
      "model_forward_time": 2.343209981918335,
      "step": 57628
    },
    {
      "epoch": 5.4958343505859375e-05,
      "step": 57628,
      "training_step_time": 7.533358335494995
    },
    {
      "epoch": 5.495929718017578e-05,
      "model_forward_time": 1.8353397846221924,
      "step": 57629
    },
    {
      "epoch": 5.495929718017578e-05,
      "step": 57629,
      "training_step_time": 8.068104267120361
    },
    {
      "epoch": 5.496025085449219e-05,
      "grad_norm": 0.08325463533401489,
      "learning_rate": 4.155995750256791e-06,
      "loss": 0.0553,
      "step": 57630
    },
    {
      "epoch": 5.496025085449219e-05,
      "model_forward_time": 1.8274359703063965,
      "step": 57630
    },
    {
      "dynamics_loss_avg": 0.06073316782712936,
      "epoch": 5.496025085449219e-05,
      "step": 57630
    },
    {
      "action_loss_avg": 0.0014488688961137087,
      "epoch": 5.496025085449219e-05,
      "step": 57630
    },
    {
      "epoch": 5.496025085449219e-05,
      "step": 57630,
      "training_step_time": 7.557736158370972
    },
    {
      "epoch": 5.4961204528808594e-05,
      "model_forward_time": 1.8243534564971924,
      "step": 57631
    },
    {
      "epoch": 5.4961204528808594e-05,
      "step": 57631,
      "training_step_time": 7.5693676471710205
    },
    {
      "epoch": 5.4962158203125e-05,
      "model_forward_time": 1.7345848083496094,
      "step": 57632
    },
    {
      "epoch": 5.4962158203125e-05,
      "step": 57632,
      "training_step_time": 7.61807918548584
    },
    {
      "epoch": 5.496311187744141e-05,
      "model_forward_time": 1.7345569133758545,
      "step": 57633
    },
    {
      "epoch": 5.496311187744141e-05,
      "step": 57633,
      "training_step_time": 8.273418188095093
    },
    {
      "epoch": 5.496406555175781e-05,
      "model_forward_time": 1.7342185974121094,
      "step": 57634
    },
    {
      "epoch": 5.496406555175781e-05,
      "step": 57634,
      "training_step_time": 8.375670433044434
    },
    {
      "epoch": 5.496501922607422e-05,
      "model_forward_time": 1.728914499282837,
      "step": 57635
    },
    {
      "epoch": 5.496501922607422e-05,
      "step": 57635,
      "training_step_time": 8.71354627609253
    },
    {
      "epoch": 5.496597290039063e-05,
      "model_forward_time": 1.36476731300354,
      "step": 57636
    },
    {
      "epoch": 5.496597290039063e-05,
      "step": 57636,
      "training_step_time": 7.804558753967285
    },
    {
      "epoch": 5.496692657470703e-05,
      "model_forward_time": 2.260143995285034,
      "step": 57637
    },
    {
      "epoch": 5.496692657470703e-05,
      "step": 57637,
      "training_step_time": 8.775974035263062
    },
    {
      "epoch": 5.4967880249023436e-05,
      "model_forward_time": 1.7355291843414307,
      "step": 57638
    },
    {
      "epoch": 5.4967880249023436e-05,
      "step": 57638,
      "training_step_time": 8.85372805595398
    },
    {
      "epoch": 5.4968833923339847e-05,
      "model_forward_time": 1.746117353439331,
      "step": 57639
    },
    {
      "epoch": 5.4968833923339847e-05,
      "step": 57639,
      "training_step_time": 8.425906896591187
    },
    {
      "epoch": 5.496978759765625e-05,
      "grad_norm": 0.09263627976179123,
      "learning_rate": 4.154366053616179e-06,
      "loss": 0.0537,
      "step": 57640
    },
    {
      "epoch": 5.496978759765625e-05,
      "model_forward_time": 1.7819395065307617,
      "step": 57640
    },
    {
      "dynamics_loss_avg": 0.0495597830042243,
      "epoch": 5.496978759765625e-05,
      "step": 57640
    },
    {
      "action_loss_avg": 0.0014422206964809447,
      "epoch": 5.496978759765625e-05,
      "step": 57640
    },
    {
      "epoch": 5.496978759765625e-05,
      "step": 57640,
      "training_step_time": 7.8296284675598145
    },
    {
      "epoch": 5.4970741271972655e-05,
      "model_forward_time": 1.7297568321228027,
      "step": 57641
    },
    {
      "epoch": 5.4970741271972655e-05,
      "step": 57641,
      "training_step_time": 7.779510974884033
    },
    {
      "epoch": 5.4971694946289065e-05,
      "model_forward_time": 1.7111451625823975,
      "step": 57642
    },
    {
      "epoch": 5.4971694946289065e-05,
      "step": 57642,
      "training_step_time": 7.731478691101074
    },
    {
      "epoch": 5.497264862060547e-05,
      "model_forward_time": 1.73600172996521,
      "step": 57643
    },
    {
      "epoch": 5.497264862060547e-05,
      "step": 57643,
      "training_step_time": 9.32644510269165
    },
    {
      "epoch": 5.4973602294921873e-05,
      "model_forward_time": 1.7322955131530762,
      "step": 57644
    },
    {
      "epoch": 5.4973602294921873e-05,
      "step": 57644,
      "training_step_time": 7.8547258377075195
    },
    {
      "epoch": 5.4974555969238284e-05,
      "model_forward_time": 1.70253586769104,
      "step": 57645
    },
    {
      "epoch": 5.4974555969238284e-05,
      "step": 57645,
      "training_step_time": 8.486735105514526
    },
    {
      "epoch": 5.497550964355469e-05,
      "model_forward_time": 2.2920446395874023,
      "step": 57646
    },
    {
      "epoch": 5.497550964355469e-05,
      "step": 57646,
      "training_step_time": 7.739635944366455
    },
    {
      "epoch": 5.497646331787109e-05,
      "model_forward_time": 1.76869535446167,
      "step": 57647
    },
    {
      "epoch": 5.497646331787109e-05,
      "step": 57647,
      "training_step_time": 7.807982444763184
    },
    {
      "epoch": 5.49774169921875e-05,
      "model_forward_time": 1.7399795055389404,
      "step": 57648
    },
    {
      "epoch": 5.49774169921875e-05,
      "step": 57648,
      "training_step_time": 8.287777423858643
    },
    {
      "epoch": 5.497837066650391e-05,
      "model_forward_time": 1.7453358173370361,
      "step": 57649
    },
    {
      "epoch": 5.497837066650391e-05,
      "step": 57649,
      "training_step_time": 8.68619680404663
    },
    {
      "epoch": 5.497932434082031e-05,
      "grad_norm": 0.1471094787120819,
      "learning_rate": 4.152736449452827e-06,
      "loss": 0.0523,
      "step": 57650
    },
    {
      "epoch": 5.497932434082031e-05,
      "model_forward_time": 1.737133502960205,
      "step": 57650
    },
    {
      "dynamics_loss_avg": 0.0672343647107482,
      "epoch": 5.497932434082031e-05,
      "step": 57650
    },
    {
      "action_loss_avg": 0.0017084987455746159,
      "epoch": 5.497932434082031e-05,
      "step": 57650
    },
    {
      "epoch": 5.497932434082031e-05,
      "step": 57650,
      "training_step_time": 8.339887857437134
    },
    {
      "epoch": 5.498027801513672e-05,
      "model_forward_time": 1.7444298267364502,
      "step": 57651
    },
    {
      "epoch": 5.498027801513672e-05,
      "step": 57651,
      "training_step_time": 8.401952981948853
    },
    {
      "epoch": 5.4981231689453126e-05,
      "model_forward_time": 1.7352509498596191,
      "step": 57652
    },
    {
      "epoch": 5.4981231689453126e-05,
      "step": 57652,
      "training_step_time": 7.948945999145508
    },
    {
      "epoch": 5.498218536376953e-05,
      "model_forward_time": 1.7396173477172852,
      "step": 57653
    },
    {
      "epoch": 5.498218536376953e-05,
      "step": 57653,
      "training_step_time": 7.876190662384033
    },
    {
      "epoch": 5.498313903808594e-05,
      "model_forward_time": 1.737445592880249,
      "step": 57654
    },
    {
      "epoch": 5.498313903808594e-05,
      "step": 57654,
      "training_step_time": 8.358248472213745
    },
    {
      "epoch": 5.4984092712402345e-05,
      "model_forward_time": 2.2957711219787598,
      "step": 57655
    },
    {
      "epoch": 5.4984092712402345e-05,
      "step": 57655,
      "training_step_time": 7.798269033432007
    },
    {
      "epoch": 5.498504638671875e-05,
      "model_forward_time": 1.7376701831817627,
      "step": 57656
    },
    {
      "epoch": 5.498504638671875e-05,
      "step": 57656,
      "training_step_time": 7.621212482452393
    },
    {
      "epoch": 5.498600006103515e-05,
      "model_forward_time": 1.7033798694610596,
      "step": 57657
    },
    {
      "epoch": 5.498600006103515e-05,
      "step": 57657,
      "training_step_time": 7.522239923477173
    },
    {
      "epoch": 5.4986953735351564e-05,
      "model_forward_time": 1.7406206130981445,
      "step": 57658
    },
    {
      "epoch": 5.4986953735351564e-05,
      "step": 57658,
      "training_step_time": 7.607374668121338
    },
    {
      "epoch": 5.498790740966797e-05,
      "model_forward_time": 1.777123212814331,
      "step": 57659
    },
    {
      "epoch": 5.498790740966797e-05,
      "step": 57659,
      "training_step_time": 7.552962779998779
    },
    {
      "epoch": 5.498886108398437e-05,
      "grad_norm": 0.07763855904340744,
      "learning_rate": 4.151106937944943e-06,
      "loss": 0.0538,
      "step": 57660
    },
    {
      "epoch": 5.498886108398437e-05,
      "model_forward_time": 1.7480833530426025,
      "step": 57660
    },
    {
      "dynamics_loss_avg": 0.051946503669023515,
      "epoch": 5.498886108398437e-05,
      "step": 57660
    },
    {
      "action_loss_avg": 0.0015327741333749146,
      "epoch": 5.498886108398437e-05,
      "step": 57660
    },
    {
      "epoch": 5.498886108398437e-05,
      "step": 57660,
      "training_step_time": 8.409335374832153
    },
    {
      "epoch": 5.498981475830078e-05,
      "model_forward_time": 1.748901605606079,
      "step": 57661
    },
    {
      "epoch": 5.498981475830078e-05,
      "step": 57661,
      "training_step_time": 7.598830461502075
    },
    {
      "epoch": 5.4990768432617187e-05,
      "model_forward_time": 1.7380566596984863,
      "step": 57662
    },
    {
      "epoch": 5.4990768432617187e-05,
      "step": 57662,
      "training_step_time": 8.535547018051147
    },
    {
      "epoch": 5.499172210693359e-05,
      "model_forward_time": 1.7409651279449463,
      "step": 57663
    },
    {
      "epoch": 5.499172210693359e-05,
      "step": 57663,
      "training_step_time": 9.294246196746826
    },
    {
      "epoch": 5.499267578125e-05,
      "model_forward_time": 2.3066294193267822,
      "step": 57664
    },
    {
      "epoch": 5.499267578125e-05,
      "step": 57664,
      "training_step_time": 7.59946608543396
    },
    {
      "epoch": 5.4993629455566405e-05,
      "model_forward_time": 1.7493548393249512,
      "step": 57665
    },
    {
      "epoch": 5.4993629455566405e-05,
      "step": 57665,
      "training_step_time": 8.127052545547485
    },
    {
      "epoch": 5.499458312988281e-05,
      "model_forward_time": 1.3474183082580566,
      "step": 57666
    },
    {
      "epoch": 5.499458312988281e-05,
      "step": 57666,
      "training_step_time": 8.541693449020386
    },
    {
      "epoch": 5.499553680419922e-05,
      "model_forward_time": 1.7635586261749268,
      "step": 57667
    },
    {
      "epoch": 5.499553680419922e-05,
      "step": 57667,
      "training_step_time": 7.616608142852783
    },
    {
      "epoch": 5.4996490478515624e-05,
      "model_forward_time": 1.790886402130127,
      "step": 57668
    },
    {
      "epoch": 5.4996490478515624e-05,
      "step": 57668,
      "training_step_time": 8.037217855453491
    },
    {
      "epoch": 5.499744415283203e-05,
      "model_forward_time": 1.7414820194244385,
      "step": 57669
    },
    {
      "epoch": 5.499744415283203e-05,
      "step": 57669,
      "training_step_time": 8.666472434997559
    },
    {
      "epoch": 5.499839782714844e-05,
      "grad_norm": 0.08500989526510239,
      "learning_rate": 4.149477519270728e-06,
      "loss": 0.0557,
      "step": 57670
    },
    {
      "epoch": 5.499839782714844e-05,
      "model_forward_time": 1.7424187660217285,
      "step": 57670
    },
    {
      "dynamics_loss_avg": 0.05477660223841667,
      "epoch": 5.499839782714844e-05,
      "step": 57670
    },
    {
      "action_loss_avg": 0.0020161228079814465,
      "epoch": 5.499839782714844e-05,
      "step": 57670
    },
    {
      "epoch": 5.499839782714844e-05,
      "step": 57670,
      "training_step_time": 9.34835696220398
    },
    {
      "epoch": 5.499935150146484e-05,
      "model_forward_time": 1.7440824508666992,
      "step": 57671
    },
    {
      "epoch": 5.499935150146484e-05,
      "step": 57671,
      "training_step_time": 7.616962194442749
    },
    {
      "epoch": 5.500030517578125e-05,
      "model_forward_time": 1.7434217929840088,
      "step": 57672
    },
    {
      "epoch": 5.500030517578125e-05,
      "step": 57672,
      "training_step_time": 8.357178449630737
    },
    {
      "epoch": 5.500125885009766e-05,
      "model_forward_time": 1.751016616821289,
      "step": 57673
    },
    {
      "epoch": 5.500125885009766e-05,
      "step": 57673,
      "training_step_time": 9.21608567237854
    },
    {
      "epoch": 5.500221252441406e-05,
      "model_forward_time": 2.3009564876556396,
      "step": 57674
    },
    {
      "epoch": 5.500221252441406e-05,
      "step": 57674,
      "training_step_time": 7.777323246002197
    },
    {
      "epoch": 5.5003166198730466e-05,
      "model_forward_time": 1.7583355903625488,
      "step": 57675
    },
    {
      "epoch": 5.5003166198730466e-05,
      "step": 57675,
      "training_step_time": 8.625806331634521
    },
    {
      "epoch": 5.500411987304688e-05,
      "model_forward_time": 1.7448151111602783,
      "step": 57676
    },
    {
      "epoch": 5.500411987304688e-05,
      "step": 57676,
      "training_step_time": 7.539910316467285
    },
    {
      "epoch": 5.500507354736328e-05,
      "model_forward_time": 1.7762086391448975,
      "step": 57677
    },
    {
      "epoch": 5.500507354736328e-05,
      "step": 57677,
      "training_step_time": 7.640849828720093
    },
    {
      "epoch": 5.5006027221679685e-05,
      "model_forward_time": 1.8441576957702637,
      "step": 57678
    },
    {
      "epoch": 5.5006027221679685e-05,
      "step": 57678,
      "training_step_time": 7.6792566776275635
    },
    {
      "epoch": 5.5006980895996096e-05,
      "model_forward_time": 1.8432536125183105,
      "step": 57679
    },
    {
      "epoch": 5.5006980895996096e-05,
      "step": 57679,
      "training_step_time": 8.383315086364746
    },
    {
      "epoch": 5.50079345703125e-05,
      "grad_norm": 0.10706180334091187,
      "learning_rate": 4.1478481936083705e-06,
      "loss": 0.058,
      "step": 57680
    },
    {
      "epoch": 5.50079345703125e-05,
      "model_forward_time": 1.8355350494384766,
      "step": 57680
    },
    {
      "dynamics_loss_avg": 0.08000043835490941,
      "epoch": 5.50079345703125e-05,
      "step": 57680
    },
    {
      "action_loss_avg": 0.0031675850739702584,
      "epoch": 5.50079345703125e-05,
      "step": 57680
    },
    {
      "epoch": 5.50079345703125e-05,
      "step": 57680,
      "training_step_time": 7.635931730270386
    },
    {
      "epoch": 5.5008888244628904e-05,
      "model_forward_time": 1.845043659210205,
      "step": 57681
    },
    {
      "epoch": 5.5008888244628904e-05,
      "step": 57681,
      "training_step_time": 8.587613821029663
    },
    {
      "epoch": 5.5009841918945315e-05,
      "model_forward_time": 1.8387620449066162,
      "step": 57682
    },
    {
      "epoch": 5.5009841918945315e-05,
      "step": 57682,
      "training_step_time": 7.852553606033325
    },
    {
      "epoch": 5.501079559326172e-05,
      "model_forward_time": 2.380091905593872,
      "step": 57683
    },
    {
      "epoch": 5.501079559326172e-05,
      "step": 57683,
      "training_step_time": 8.780637741088867
    },
    {
      "epoch": 5.501174926757812e-05,
      "model_forward_time": 1.8450982570648193,
      "step": 57684
    },
    {
      "epoch": 5.501174926757812e-05,
      "step": 57684,
      "training_step_time": 7.539245367050171
    },
    {
      "epoch": 5.5012702941894533e-05,
      "model_forward_time": 1.837247610092163,
      "step": 57685
    },
    {
      "epoch": 5.5012702941894533e-05,
      "step": 57685,
      "training_step_time": 7.503965139389038
    },
    {
      "epoch": 5.501365661621094e-05,
      "model_forward_time": 1.8381545543670654,
      "step": 57686
    },
    {
      "epoch": 5.501365661621094e-05,
      "step": 57686,
      "training_step_time": 7.635042428970337
    },
    {
      "epoch": 5.501461029052734e-05,
      "model_forward_time": 1.8389294147491455,
      "step": 57687
    },
    {
      "epoch": 5.501461029052734e-05,
      "step": 57687,
      "training_step_time": 7.628211259841919
    },
    {
      "epoch": 5.501556396484375e-05,
      "model_forward_time": 1.8412461280822754,
      "step": 57688
    },
    {
      "epoch": 5.501556396484375e-05,
      "step": 57688,
      "training_step_time": 7.513627767562866
    },
    {
      "epoch": 5.5016517639160156e-05,
      "model_forward_time": 1.8460121154785156,
      "step": 57689
    },
    {
      "epoch": 5.5016517639160156e-05,
      "step": 57689,
      "training_step_time": 8.704162836074829
    },
    {
      "epoch": 5.501747131347656e-05,
      "grad_norm": 0.12074621021747589,
      "learning_rate": 4.146218961136056e-06,
      "loss": 0.0535,
      "step": 57690
    },
    {
      "epoch": 5.501747131347656e-05,
      "model_forward_time": 1.839738368988037,
      "step": 57690
    },
    {
      "dynamics_loss_avg": 0.051788732316344975,
      "epoch": 5.501747131347656e-05,
      "step": 57690
    },
    {
      "action_loss_avg": 0.0018583918601507321,
      "epoch": 5.501747131347656e-05,
      "step": 57690
    },
    {
      "epoch": 5.501747131347656e-05,
      "step": 57690,
      "training_step_time": 8.480071306228638
    },
    {
      "epoch": 5.501842498779297e-05,
      "model_forward_time": 1.8462870121002197,
      "step": 57691
    },
    {
      "epoch": 5.501842498779297e-05,
      "step": 57691,
      "training_step_time": 7.614574432373047
    },
    {
      "epoch": 5.5019378662109375e-05,
      "model_forward_time": 2.3769075870513916,
      "step": 57692
    },
    {
      "epoch": 5.5019378662109375e-05,
      "step": 57692,
      "training_step_time": 8.287214279174805
    },
    {
      "epoch": 5.502033233642578e-05,
      "model_forward_time": 1.850633144378662,
      "step": 57693
    },
    {
      "epoch": 5.502033233642578e-05,
      "step": 57693,
      "training_step_time": 7.684184789657593
    },
    {
      "epoch": 5.502128601074219e-05,
      "model_forward_time": 1.8501412868499756,
      "step": 57694
    },
    {
      "epoch": 5.502128601074219e-05,
      "step": 57694,
      "training_step_time": 7.593157768249512
    },
    {
      "epoch": 5.5022239685058594e-05,
      "model_forward_time": 1.8370485305786133,
      "step": 57695
    },
    {
      "epoch": 5.5022239685058594e-05,
      "step": 57695,
      "training_step_time": 7.580864906311035
    },
    {
      "epoch": 5.5023193359375e-05,
      "model_forward_time": 1.8585355281829834,
      "step": 57696
    },
    {
      "epoch": 5.5023193359375e-05,
      "step": 57696,
      "training_step_time": 7.644251346588135
    },
    {
      "epoch": 5.502414703369141e-05,
      "model_forward_time": 1.8458271026611328,
      "step": 57697
    },
    {
      "epoch": 5.502414703369141e-05,
      "step": 57697,
      "training_step_time": 8.362782716751099
    },
    {
      "epoch": 5.502510070800781e-05,
      "model_forward_time": 1.8472919464111328,
      "step": 57698
    },
    {
      "epoch": 5.502510070800781e-05,
      "step": 57698,
      "training_step_time": 9.018786191940308
    },
    {
      "epoch": 5.502605438232422e-05,
      "model_forward_time": 1.8368597030639648,
      "step": 57699
    },
    {
      "epoch": 5.502605438232422e-05,
      "step": 57699,
      "training_step_time": 7.648257255554199
    },
    {
      "epoch": 5.502700805664063e-05,
      "grad_norm": 0.17093543708324432,
      "learning_rate": 4.14458982203195e-06,
      "loss": 0.0567,
      "step": 57700
    },
    {
      "epoch": 5.502700805664063e-05,
      "model_forward_time": 1.8517894744873047,
      "step": 57700
    },
    {
      "dynamics_loss_avg": 0.06686257757246494,
      "epoch": 5.502700805664063e-05,
      "step": 57700
    },
    {
      "action_loss_avg": 0.0016896200919291004,
      "epoch": 5.502700805664063e-05,
      "step": 57700
    },
    {
      "epoch": 5.502700805664063e-05,
      "step": 57700,
      "training_step_time": 7.650451898574829
    },
    {
      "epoch": 5.502796173095703e-05,
      "model_forward_time": 2.4124839305877686,
      "step": 57701
    },
    {
      "epoch": 5.502796173095703e-05,
      "step": 57701,
      "training_step_time": 7.443437576293945
    },
    {
      "epoch": 5.5028915405273436e-05,
      "model_forward_time": 1.8455193042755127,
      "step": 57702
    },
    {
      "epoch": 5.5028915405273436e-05,
      "step": 57702,
      "training_step_time": 7.6900646686553955
    },
    {
      "epoch": 5.5029869079589847e-05,
      "model_forward_time": 1.8415896892547607,
      "step": 57703
    },
    {
      "epoch": 5.5029869079589847e-05,
      "step": 57703,
      "training_step_time": 7.587252616882324
    },
    {
      "epoch": 5.503082275390625e-05,
      "model_forward_time": 1.848510980606079,
      "step": 57704
    },
    {
      "epoch": 5.503082275390625e-05,
      "step": 57704,
      "training_step_time": 7.642411947250366
    },
    {
      "epoch": 5.5031776428222655e-05,
      "model_forward_time": 1.8501582145690918,
      "step": 57705
    },
    {
      "epoch": 5.5031776428222655e-05,
      "step": 57705,
      "training_step_time": 8.243446111679077
    },
    {
      "epoch": 5.5032730102539065e-05,
      "model_forward_time": 1.863405704498291,
      "step": 57706
    },
    {
      "epoch": 5.5032730102539065e-05,
      "step": 57706,
      "training_step_time": 7.372810363769531
    },
    {
      "epoch": 5.503368377685547e-05,
      "model_forward_time": 1.8605775833129883,
      "step": 57707
    },
    {
      "epoch": 5.503368377685547e-05,
      "step": 57707,
      "training_step_time": 8.414234161376953
    },
    {
      "epoch": 5.5034637451171873e-05,
      "model_forward_time": 1.4696621894836426,
      "step": 57708
    },
    {
      "epoch": 5.5034637451171873e-05,
      "step": 57708,
      "training_step_time": 7.614584922790527
    },
    {
      "epoch": 5.5035591125488284e-05,
      "model_forward_time": 1.490875005722046,
      "step": 57709
    },
    {
      "epoch": 5.5035591125488284e-05,
      "step": 57709,
      "training_step_time": 8.587860822677612
    },
    {
      "epoch": 5.503654479980469e-05,
      "grad_norm": 0.09224960952997208,
      "learning_rate": 4.142960776474217e-06,
      "loss": 0.0563,
      "step": 57710
    },
    {
      "epoch": 5.503654479980469e-05,
      "model_forward_time": 1.8173096179962158,
      "step": 57710
    },
    {
      "dynamics_loss_avg": 0.0523113907314837,
      "epoch": 5.503654479980469e-05,
      "step": 57710
    },
    {
      "action_loss_avg": 0.0016753920208429918,
      "epoch": 5.503654479980469e-05,
      "step": 57710
    },
    {
      "epoch": 5.503654479980469e-05,
      "step": 57710,
      "training_step_time": 9.47254490852356
    },
    {
      "epoch": 5.503749847412109e-05,
      "model_forward_time": 1.8421175479888916,
      "step": 57711
    },
    {
      "epoch": 5.503749847412109e-05,
      "step": 57711,
      "training_step_time": 9.647637844085693
    },
    {
      "epoch": 5.50384521484375e-05,
      "model_forward_time": 1.8468382358551025,
      "step": 57712
    },
    {
      "epoch": 5.50384521484375e-05,
      "step": 57712,
      "training_step_time": 6.787985324859619
    },
    {
      "epoch": 5.503940582275391e-05,
      "model_forward_time": 1.840804100036621,
      "step": 57713
    },
    {
      "epoch": 5.503940582275391e-05,
      "step": 57713,
      "training_step_time": 6.788064002990723
    },
    {
      "epoch": 5.504035949707031e-05,
      "model_forward_time": 1.8450555801391602,
      "step": 57714
    },
    {
      "epoch": 5.504035949707031e-05,
      "step": 57714,
      "training_step_time": 6.795062303543091
    },
    {
      "epoch": 5.504131317138672e-05,
      "model_forward_time": 1.8387925624847412,
      "step": 57715
    },
    {
      "epoch": 5.504131317138672e-05,
      "step": 57715,
      "training_step_time": 6.835062026977539
    },
    {
      "epoch": 5.5042266845703126e-05,
      "model_forward_time": 1.8515372276306152,
      "step": 57716
    },
    {
      "epoch": 5.5042266845703126e-05,
      "step": 57716,
      "training_step_time": 6.747898578643799
    },
    {
      "epoch": 5.504322052001953e-05,
      "model_forward_time": 1.8370754718780518,
      "step": 57717
    },
    {
      "epoch": 5.504322052001953e-05,
      "step": 57717,
      "training_step_time": 6.756257772445679
    },
    {
      "epoch": 5.504417419433594e-05,
      "model_forward_time": 1.8311998844146729,
      "step": 57718
    },
    {
      "epoch": 5.504417419433594e-05,
      "step": 57718,
      "training_step_time": 6.752875566482544
    },
    {
      "epoch": 5.5045127868652345e-05,
      "model_forward_time": 1.8395214080810547,
      "step": 57719
    },
    {
      "epoch": 5.5045127868652345e-05,
      "step": 57719,
      "training_step_time": 6.7458930015563965
    },
    {
      "epoch": 5.504608154296875e-05,
      "grad_norm": 0.11053988337516785,
      "learning_rate": 4.141331824641003e-06,
      "loss": 0.054,
      "step": 57720
    },
    {
      "epoch": 5.504608154296875e-05,
      "model_forward_time": 1.9980077743530273,
      "step": 57720
    },
    {
      "dynamics_loss_avg": 0.038130782172083856,
      "epoch": 5.504608154296875e-05,
      "step": 57720
    },
    {
      "action_loss_avg": 0.001079164061229676,
      "epoch": 5.504608154296875e-05,
      "step": 57720
    },
    {
      "epoch": 5.504608154296875e-05,
      "step": 57720,
      "training_step_time": 7.567729234695435
    },
    {
      "epoch": 5.504703521728515e-05,
      "model_forward_time": 1.8518009185791016,
      "step": 57721
    },
    {
      "epoch": 5.504703521728515e-05,
      "step": 57721,
      "training_step_time": 8.587280988693237
    },
    {
      "epoch": 5.5047988891601564e-05,
      "model_forward_time": 1.842719554901123,
      "step": 57722
    },
    {
      "epoch": 5.5047988891601564e-05,
      "step": 57722,
      "training_step_time": 6.785660743713379
    },
    {
      "epoch": 5.504894256591797e-05,
      "model_forward_time": 1.8513808250427246,
      "step": 57723
    },
    {
      "epoch": 5.504894256591797e-05,
      "step": 57723,
      "training_step_time": 6.74997615814209
    },
    {
      "epoch": 5.504989624023437e-05,
      "model_forward_time": 1.8435475826263428,
      "step": 57724
    },
    {
      "epoch": 5.504989624023437e-05,
      "step": 57724,
      "training_step_time": 6.77120041847229
    },
    {
      "epoch": 5.505084991455078e-05,
      "model_forward_time": 1.8410375118255615,
      "step": 57725
    },
    {
      "epoch": 5.505084991455078e-05,
      "step": 57725,
      "training_step_time": 6.818187713623047
    },
    {
      "epoch": 5.505180358886719e-05,
      "model_forward_time": 1.8502378463745117,
      "step": 57726
    },
    {
      "epoch": 5.505180358886719e-05,
      "step": 57726,
      "training_step_time": 6.765536785125732
    },
    {
      "epoch": 5.505275726318359e-05,
      "model_forward_time": 1.8393595218658447,
      "step": 57727
    },
    {
      "epoch": 5.505275726318359e-05,
      "step": 57727,
      "training_step_time": 6.683075428009033
    },
    {
      "epoch": 5.50537109375e-05,
      "model_forward_time": 1.841252326965332,
      "step": 57728
    },
    {
      "epoch": 5.50537109375e-05,
      "step": 57728,
      "training_step_time": 6.793513536453247
    },
    {
      "epoch": 5.5054664611816405e-05,
      "model_forward_time": 2.385819435119629,
      "step": 57729
    },
    {
      "epoch": 5.5054664611816405e-05,
      "step": 57729,
      "training_step_time": 7.6132800579071045
    },
    {
      "epoch": 5.505561828613281e-05,
      "grad_norm": 0.10878315567970276,
      "learning_rate": 4.139702966710452e-06,
      "loss": 0.0589,
      "step": 57730
    },
    {
      "epoch": 5.505561828613281e-05,
      "model_forward_time": 1.842658519744873,
      "step": 57730
    },
    {
      "dynamics_loss_avg": 0.07021721731871367,
      "epoch": 5.505561828613281e-05,
      "step": 57730
    },
    {
      "action_loss_avg": 0.0019304182380437851,
      "epoch": 5.505561828613281e-05,
      "step": 57730
    },
    {
      "epoch": 5.505561828613281e-05,
      "step": 57730,
      "training_step_time": 8.759602069854736
    },
    {
      "epoch": 5.505657196044922e-05,
      "model_forward_time": 1.0977153778076172,
      "step": 57731
    },
    {
      "epoch": 5.505657196044922e-05,
      "step": 57731,
      "training_step_time": 7.51108455657959
    },
    {
      "epoch": 5.5057525634765624e-05,
      "model_forward_time": 1.8377716541290283,
      "step": 57732
    },
    {
      "epoch": 5.5057525634765624e-05,
      "step": 57732,
      "training_step_time": 6.820922374725342
    },
    {
      "epoch": 5.505847930908203e-05,
      "model_forward_time": 1.8485758304595947,
      "step": 57733
    },
    {
      "epoch": 5.505847930908203e-05,
      "step": 57733,
      "training_step_time": 6.7409045696258545
    },
    {
      "epoch": 5.505943298339844e-05,
      "model_forward_time": 1.846806526184082,
      "step": 57734
    },
    {
      "epoch": 5.505943298339844e-05,
      "step": 57734,
      "training_step_time": 6.779841423034668
    },
    {
      "epoch": 5.506038665771484e-05,
      "model_forward_time": 1.8449854850769043,
      "step": 57735
    },
    {
      "epoch": 5.506038665771484e-05,
      "step": 57735,
      "training_step_time": 6.767921209335327
    },
    {
      "epoch": 5.506134033203125e-05,
      "model_forward_time": 1.846531867980957,
      "step": 57736
    },
    {
      "epoch": 5.506134033203125e-05,
      "step": 57736,
      "training_step_time": 6.74309778213501
    },
    {
      "epoch": 5.506229400634766e-05,
      "model_forward_time": 1.8495514392852783,
      "step": 57737
    },
    {
      "epoch": 5.506229400634766e-05,
      "step": 57737,
      "training_step_time": 6.712685585021973
    },
    {
      "epoch": 5.506324768066406e-05,
      "model_forward_time": 2.372907876968384,
      "step": 57738
    },
    {
      "epoch": 5.506324768066406e-05,
      "step": 57738,
      "training_step_time": 7.570585012435913
    },
    {
      "epoch": 5.5064201354980466e-05,
      "model_forward_time": 1.8481671810150146,
      "step": 57739
    },
    {
      "epoch": 5.5064201354980466e-05,
      "step": 57739,
      "training_step_time": 10.985859155654907
    },
    {
      "epoch": 5.506515502929688e-05,
      "grad_norm": 0.0925218015909195,
      "learning_rate": 4.13807420286069e-06,
      "loss": 0.0551,
      "step": 57740
    },
    {
      "epoch": 5.506515502929688e-05,
      "model_forward_time": 1.8346576690673828,
      "step": 57740
    },
    {
      "dynamics_loss_avg": 0.05738886948674917,
      "epoch": 5.506515502929688e-05,
      "step": 57740
    },
    {
      "action_loss_avg": 0.0016104265698231757,
      "epoch": 5.506515502929688e-05,
      "step": 57740
    },
    {
      "epoch": 5.506515502929688e-05,
      "step": 57740,
      "training_step_time": 7.794720649719238
    },
    {
      "epoch": 5.506610870361328e-05,
      "model_forward_time": 1.8408079147338867,
      "step": 57741
    },
    {
      "epoch": 5.506610870361328e-05,
      "step": 57741,
      "training_step_time": 6.709699630737305
    },
    {
      "epoch": 5.5067062377929685e-05,
      "model_forward_time": 1.843311071395874,
      "step": 57742
    },
    {
      "epoch": 5.5067062377929685e-05,
      "step": 57742,
      "training_step_time": 6.814693927764893
    },
    {
      "epoch": 5.5068016052246096e-05,
      "model_forward_time": 1.8503022193908691,
      "step": 57743
    },
    {
      "epoch": 5.5068016052246096e-05,
      "step": 57743,
      "training_step_time": 6.784001111984253
    },
    {
      "epoch": 5.50689697265625e-05,
      "model_forward_time": 1.4854495525360107,
      "step": 57744
    },
    {
      "epoch": 5.50689697265625e-05,
      "step": 57744,
      "training_step_time": 6.748091459274292
    },
    {
      "epoch": 5.5069923400878904e-05,
      "model_forward_time": 1.835693359375,
      "step": 57745
    },
    {
      "epoch": 5.5069923400878904e-05,
      "step": 57745,
      "training_step_time": 6.7743141651153564
    },
    {
      "epoch": 5.5070877075195315e-05,
      "model_forward_time": 1.842559814453125,
      "step": 57746
    },
    {
      "epoch": 5.5070877075195315e-05,
      "step": 57746,
      "training_step_time": 6.796799898147583
    },
    {
      "epoch": 5.507183074951172e-05,
      "model_forward_time": 1.8451359272003174,
      "step": 57747
    },
    {
      "epoch": 5.507183074951172e-05,
      "step": 57747,
      "training_step_time": 7.181442499160767
    },
    {
      "epoch": 5.507278442382812e-05,
      "model_forward_time": 2.3774852752685547,
      "step": 57748
    },
    {
      "epoch": 5.507278442382812e-05,
      "step": 57748,
      "training_step_time": 10.280162334442139
    },
    {
      "epoch": 5.5073738098144533e-05,
      "model_forward_time": 1.8556692600250244,
      "step": 57749
    },
    {
      "epoch": 5.5073738098144533e-05,
      "step": 57749,
      "training_step_time": 7.725068092346191
    },
    {
      "epoch": 5.507469177246094e-05,
      "grad_norm": 0.11324845999479294,
      "learning_rate": 4.136445533269837e-06,
      "loss": 0.0566,
      "step": 57750
    },
    {
      "epoch": 5.507469177246094e-05,
      "model_forward_time": 1.841233253479004,
      "step": 57750
    },
    {
      "dynamics_loss_avg": 0.067211982794106,
      "epoch": 5.507469177246094e-05,
      "step": 57750
    },
    {
      "action_loss_avg": 0.0013881113642128184,
      "epoch": 5.507469177246094e-05,
      "step": 57750
    },
    {
      "epoch": 5.507469177246094e-05,
      "step": 57750,
      "training_step_time": 6.781857967376709
    },
    {
      "epoch": 5.507564544677734e-05,
      "model_forward_time": 1.8422932624816895,
      "step": 57751
    },
    {
      "epoch": 5.507564544677734e-05,
      "step": 57751,
      "training_step_time": 6.830483436584473
    },
    {
      "epoch": 5.507659912109375e-05,
      "model_forward_time": 1.8501100540161133,
      "step": 57752
    },
    {
      "epoch": 5.507659912109375e-05,
      "step": 57752,
      "training_step_time": 6.760003089904785
    },
    {
      "epoch": 5.5077552795410156e-05,
      "model_forward_time": 1.8515944480895996,
      "step": 57753
    },
    {
      "epoch": 5.5077552795410156e-05,
      "step": 57753,
      "training_step_time": 6.738489627838135
    },
    {
      "epoch": 5.507850646972656e-05,
      "model_forward_time": 1.8528804779052734,
      "step": 57754
    },
    {
      "epoch": 5.507850646972656e-05,
      "step": 57754,
      "training_step_time": 6.831169843673706
    },
    {
      "epoch": 5.507946014404297e-05,
      "model_forward_time": 1.8463573455810547,
      "step": 57755
    },
    {
      "epoch": 5.507946014404297e-05,
      "step": 57755,
      "training_step_time": 6.846570730209351
    },
    {
      "epoch": 5.5080413818359375e-05,
      "model_forward_time": 1.8443636894226074,
      "step": 57756
    },
    {
      "epoch": 5.5080413818359375e-05,
      "step": 57756,
      "training_step_time": 7.60376238822937
    },
    {
      "epoch": 5.508136749267578e-05,
      "model_forward_time": 2.3547303676605225,
      "step": 57757
    },
    {
      "epoch": 5.508136749267578e-05,
      "step": 57757,
      "training_step_time": 8.843817234039307
    },
    {
      "epoch": 5.508232116699219e-05,
      "model_forward_time": 1.848048448562622,
      "step": 57758
    },
    {
      "epoch": 5.508232116699219e-05,
      "step": 57758,
      "training_step_time": 7.791509389877319
    },
    {
      "epoch": 5.5083274841308594e-05,
      "model_forward_time": 1.8374671936035156,
      "step": 57759
    },
    {
      "epoch": 5.5083274841308594e-05,
      "step": 57759,
      "training_step_time": 6.882449626922607
    },
    {
      "epoch": 5.5084228515625e-05,
      "grad_norm": 0.09079939872026443,
      "learning_rate": 4.134816958116003e-06,
      "loss": 0.0575,
      "step": 57760
    },
    {
      "epoch": 5.5084228515625e-05,
      "model_forward_time": 1.8436481952667236,
      "step": 57760
    },
    {
      "dynamics_loss_avg": 0.06786629278212786,
      "epoch": 5.5084228515625e-05,
      "step": 57760
    },
    {
      "action_loss_avg": 0.0012079450651071965,
      "epoch": 5.5084228515625e-05,
      "step": 57760
    },
    {
      "epoch": 5.5084228515625e-05,
      "step": 57760,
      "training_step_time": 6.8688740730285645
    },
    {
      "epoch": 5.508518218994141e-05,
      "model_forward_time": 1.844348430633545,
      "step": 57761
    },
    {
      "epoch": 5.508518218994141e-05,
      "step": 57761,
      "training_step_time": 6.869612693786621
    },
    {
      "epoch": 5.508613586425781e-05,
      "model_forward_time": 1.8402900695800781,
      "step": 57762
    },
    {
      "epoch": 5.508613586425781e-05,
      "step": 57762,
      "training_step_time": 6.804974794387817
    },
    {
      "epoch": 5.508708953857422e-05,
      "model_forward_time": 1.8440706729888916,
      "step": 57763
    },
    {
      "epoch": 5.508708953857422e-05,
      "step": 57763,
      "training_step_time": 6.80650520324707
    },
    {
      "epoch": 5.508804321289063e-05,
      "model_forward_time": 1.8496968746185303,
      "step": 57764
    },
    {
      "epoch": 5.508804321289063e-05,
      "step": 57764,
      "training_step_time": 6.824692249298096
    },
    {
      "epoch": 5.508899688720703e-05,
      "model_forward_time": 1.835310935974121,
      "step": 57765
    },
    {
      "epoch": 5.508899688720703e-05,
      "step": 57765,
      "training_step_time": 7.5061399936676025
    },
    {
      "epoch": 5.5089950561523436e-05,
      "model_forward_time": 2.42098069190979,
      "step": 57766
    },
    {
      "epoch": 5.5089950561523436e-05,
      "step": 57766,
      "training_step_time": 7.660459280014038
    },
    {
      "epoch": 5.509090423583985e-05,
      "model_forward_time": 1.844865322113037,
      "step": 57767
    },
    {
      "epoch": 5.509090423583985e-05,
      "step": 57767,
      "training_step_time": 7.868825435638428
    },
    {
      "epoch": 5.509185791015625e-05,
      "model_forward_time": 1.8573050498962402,
      "step": 57768
    },
    {
      "epoch": 5.509185791015625e-05,
      "step": 57768,
      "training_step_time": 7.475729942321777
    },
    {
      "epoch": 5.5092811584472655e-05,
      "model_forward_time": 1.8584184646606445,
      "step": 57769
    },
    {
      "epoch": 5.5092811584472655e-05,
      "step": 57769,
      "training_step_time": 6.71035361289978
    },
    {
      "epoch": 5.5093765258789065e-05,
      "grad_norm": 0.10048005729913712,
      "learning_rate": 4.1331884775772836e-06,
      "loss": 0.0591,
      "step": 57770
    },
    {
      "epoch": 5.5093765258789065e-05,
      "model_forward_time": 1.4975616931915283,
      "step": 57770
    },
    {
      "dynamics_loss_avg": 0.04939836049452424,
      "epoch": 5.5093765258789065e-05,
      "step": 57770
    },
    {
      "action_loss_avg": 0.0015310114424210042,
      "epoch": 5.5093765258789065e-05,
      "step": 57770
    },
    {
      "epoch": 5.5093765258789065e-05,
      "step": 57770,
      "training_step_time": 6.732813835144043
    },
    {
      "epoch": 5.509471893310547e-05,
      "model_forward_time": 1.8412587642669678,
      "step": 57771
    },
    {
      "epoch": 5.509471893310547e-05,
      "step": 57771,
      "training_step_time": 6.771658420562744
    },
    {
      "epoch": 5.5095672607421874e-05,
      "model_forward_time": 1.8425681591033936,
      "step": 57772
    },
    {
      "epoch": 5.5095672607421874e-05,
      "step": 57772,
      "training_step_time": 6.736888885498047
    },
    {
      "epoch": 5.5096626281738284e-05,
      "model_forward_time": 1.8594412803649902,
      "step": 57773
    },
    {
      "epoch": 5.5096626281738284e-05,
      "step": 57773,
      "training_step_time": 6.746626615524292
    },
    {
      "epoch": 5.509757995605469e-05,
      "model_forward_time": 1.8545441627502441,
      "step": 57774
    },
    {
      "epoch": 5.509757995605469e-05,
      "step": 57774,
      "training_step_time": 7.3670759201049805
    },
    {
      "epoch": 5.509853363037109e-05,
      "model_forward_time": 2.3996856212615967,
      "step": 57775
    },
    {
      "epoch": 5.509853363037109e-05,
      "step": 57775,
      "training_step_time": 7.6329169273376465
    },
    {
      "epoch": 5.50994873046875e-05,
      "model_forward_time": 1.8398878574371338,
      "step": 57776
    },
    {
      "epoch": 5.50994873046875e-05,
      "step": 57776,
      "training_step_time": 9.533004760742188
    },
    {
      "epoch": 5.510044097900391e-05,
      "model_forward_time": 1.8464086055755615,
      "step": 57777
    },
    {
      "epoch": 5.510044097900391e-05,
      "step": 57777,
      "training_step_time": 7.5926673412323
    },
    {
      "epoch": 5.510139465332031e-05,
      "model_forward_time": 1.854555606842041,
      "step": 57778
    },
    {
      "epoch": 5.510139465332031e-05,
      "step": 57778,
      "training_step_time": 6.768540143966675
    },
    {
      "epoch": 5.510234832763672e-05,
      "model_forward_time": 1.85502290725708,
      "step": 57779
    },
    {
      "epoch": 5.510234832763672e-05,
      "step": 57779,
      "training_step_time": 6.706689357757568
    },
    {
      "epoch": 5.5103302001953126e-05,
      "grad_norm": 0.10795319080352783,
      "learning_rate": 4.131560091831771e-06,
      "loss": 0.06,
      "step": 57780
    },
    {
      "epoch": 5.5103302001953126e-05,
      "model_forward_time": 1.8492894172668457,
      "step": 57780
    },
    {
      "dynamics_loss_avg": 0.0620322210714221,
      "epoch": 5.5103302001953126e-05,
      "step": 57780
    },
    {
      "action_loss_avg": 0.001629156176932156,
      "epoch": 5.5103302001953126e-05,
      "step": 57780
    },
    {
      "epoch": 5.5103302001953126e-05,
      "step": 57780,
      "training_step_time": 6.6303839683532715
    },
    {
      "epoch": 5.510425567626953e-05,
      "model_forward_time": 1.8396790027618408,
      "step": 57781
    },
    {
      "epoch": 5.510425567626953e-05,
      "step": 57781,
      "training_step_time": 6.790489673614502
    },
    {
      "epoch": 5.5105209350585934e-05,
      "model_forward_time": 1.8460562229156494,
      "step": 57782
    },
    {
      "epoch": 5.5105209350585934e-05,
      "step": 57782,
      "training_step_time": 6.766644239425659
    },
    {
      "epoch": 5.5106163024902345e-05,
      "model_forward_time": 1.8549728393554688,
      "step": 57783
    },
    {
      "epoch": 5.5106163024902345e-05,
      "step": 57783,
      "training_step_time": 6.775287628173828
    },
    {
      "epoch": 5.510711669921875e-05,
      "model_forward_time": 1.8462450504302979,
      "step": 57784
    },
    {
      "epoch": 5.510711669921875e-05,
      "step": 57784,
      "training_step_time": 7.545009136199951
    },
    {
      "epoch": 5.510807037353515e-05,
      "model_forward_time": 2.3987672328948975,
      "step": 57785
    },
    {
      "epoch": 5.510807037353515e-05,
      "step": 57785,
      "training_step_time": 9.419262647628784
    },
    {
      "epoch": 5.5109024047851564e-05,
      "model_forward_time": 1.8662822246551514,
      "step": 57786
    },
    {
      "epoch": 5.5109024047851564e-05,
      "step": 57786,
      "training_step_time": 8.73668360710144
    },
    {
      "epoch": 5.510997772216797e-05,
      "model_forward_time": 1.8472373485565186,
      "step": 57787
    },
    {
      "epoch": 5.510997772216797e-05,
      "step": 57787,
      "training_step_time": 6.775179147720337
    },
    {
      "epoch": 5.511093139648437e-05,
      "model_forward_time": 1.8631629943847656,
      "step": 57788
    },
    {
      "epoch": 5.511093139648437e-05,
      "step": 57788,
      "training_step_time": 6.733625173568726
    },
    {
      "epoch": 5.511188507080078e-05,
      "model_forward_time": 1.8373644351959229,
      "step": 57789
    },
    {
      "epoch": 5.511188507080078e-05,
      "step": 57789,
      "training_step_time": 6.738433599472046
    },
    {
      "epoch": 5.511283874511719e-05,
      "grad_norm": 0.08092562109231949,
      "learning_rate": 4.12993180105754e-06,
      "loss": 0.0541,
      "step": 57790
    },
    {
      "epoch": 5.511283874511719e-05,
      "model_forward_time": 1.8560233116149902,
      "step": 57790
    },
    {
      "dynamics_loss_avg": 0.056408295221626756,
      "epoch": 5.511283874511719e-05,
      "step": 57790
    },
    {
      "action_loss_avg": 0.0016066865297034382,
      "epoch": 5.511283874511719e-05,
      "step": 57790
    },
    {
      "epoch": 5.511283874511719e-05,
      "step": 57790,
      "training_step_time": 6.673036336898804
    },
    {
      "epoch": 5.511379241943359e-05,
      "model_forward_time": 1.8757450580596924,
      "step": 57791
    },
    {
      "epoch": 5.511379241943359e-05,
      "step": 57791,
      "training_step_time": 6.753646373748779
    },
    {
      "epoch": 5.511474609375e-05,
      "model_forward_time": 1.8348467350006104,
      "step": 57792
    },
    {
      "epoch": 5.511474609375e-05,
      "step": 57792,
      "training_step_time": 6.719692707061768
    },
    {
      "epoch": 5.5115699768066406e-05,
      "model_forward_time": 1.859501600265503,
      "step": 57793
    },
    {
      "epoch": 5.5115699768066406e-05,
      "step": 57793,
      "training_step_time": 8.592139482498169
    },
    {
      "epoch": 5.511665344238281e-05,
      "model_forward_time": 2.3970272541046143,
      "step": 57794
    },
    {
      "epoch": 5.511665344238281e-05,
      "step": 57794,
      "training_step_time": 10.490790605545044
    },
    {
      "epoch": 5.511760711669922e-05,
      "model_forward_time": 1.857015609741211,
      "step": 57795
    },
    {
      "epoch": 5.511760711669922e-05,
      "step": 57795,
      "training_step_time": 7.686691999435425
    },
    {
      "epoch": 5.5118560791015624e-05,
      "model_forward_time": 1.8520028591156006,
      "step": 57796
    },
    {
      "epoch": 5.5118560791015624e-05,
      "step": 57796,
      "training_step_time": 7.436683177947998
    },
    {
      "epoch": 5.511951446533203e-05,
      "model_forward_time": 1.8566417694091797,
      "step": 57797
    },
    {
      "epoch": 5.511951446533203e-05,
      "step": 57797,
      "training_step_time": 6.7416465282440186
    },
    {
      "epoch": 5.512046813964844e-05,
      "model_forward_time": 1.8597242832183838,
      "step": 57798
    },
    {
      "epoch": 5.512046813964844e-05,
      "step": 57798,
      "training_step_time": 6.706767559051514
    },
    {
      "epoch": 5.512142181396484e-05,
      "model_forward_time": 1.8562376499176025,
      "step": 57799
    },
    {
      "epoch": 5.512142181396484e-05,
      "step": 57799,
      "training_step_time": 6.700696706771851
    },
    {
      "epoch": 5.512237548828125e-05,
      "grad_norm": 0.1072864904999733,
      "learning_rate": 4.128303605432661e-06,
      "loss": 0.0578,
      "step": 57800
    },
    {
      "epoch": 5.512237548828125e-05,
      "model_forward_time": 1.8540918827056885,
      "step": 57800
    },
    {
      "dynamics_loss_avg": 0.05749005936086178,
      "epoch": 5.512237548828125e-05,
      "step": 57800
    },
    {
      "action_loss_avg": 0.0016160439001396299,
      "epoch": 5.512237548828125e-05,
      "step": 57800
    },
    {
      "epoch": 5.512237548828125e-05,
      "step": 57800,
      "training_step_time": 6.716925382614136
    },
    {
      "epoch": 5.512332916259766e-05,
      "model_forward_time": 1.8606500625610352,
      "step": 57801
    },
    {
      "epoch": 5.512332916259766e-05,
      "step": 57801,
      "training_step_time": 6.7200541496276855
    },
    {
      "epoch": 5.512428283691406e-05,
      "model_forward_time": 1.8538429737091064,
      "step": 57802
    },
    {
      "epoch": 5.512428283691406e-05,
      "step": 57802,
      "training_step_time": 7.526005744934082
    },
    {
      "epoch": 5.5125236511230466e-05,
      "model_forward_time": 2.423798084259033,
      "step": 57803
    },
    {
      "epoch": 5.5125236511230466e-05,
      "step": 57803,
      "training_step_time": 9.453470468521118
    },
    {
      "epoch": 5.512619018554688e-05,
      "model_forward_time": 1.849381685256958,
      "step": 57804
    },
    {
      "epoch": 5.512619018554688e-05,
      "step": 57804,
      "training_step_time": 7.731945037841797
    },
    {
      "epoch": 5.512714385986328e-05,
      "model_forward_time": 1.8340325355529785,
      "step": 57805
    },
    {
      "epoch": 5.512714385986328e-05,
      "step": 57805,
      "training_step_time": 7.50698447227478
    },
    {
      "epoch": 5.5128097534179685e-05,
      "model_forward_time": 1.860142469406128,
      "step": 57806
    },
    {
      "epoch": 5.5128097534179685e-05,
      "step": 57806,
      "training_step_time": 6.613651990890503
    },
    {
      "epoch": 5.5129051208496096e-05,
      "model_forward_time": 1.855778694152832,
      "step": 57807
    },
    {
      "epoch": 5.5129051208496096e-05,
      "step": 57807,
      "training_step_time": 6.715208053588867
    },
    {
      "epoch": 5.51300048828125e-05,
      "model_forward_time": 1.8672869205474854,
      "step": 57808
    },
    {
      "epoch": 5.51300048828125e-05,
      "step": 57808,
      "training_step_time": 6.709179401397705
    },
    {
      "epoch": 5.5130958557128904e-05,
      "model_forward_time": 1.8623526096343994,
      "step": 57809
    },
    {
      "epoch": 5.5130958557128904e-05,
      "step": 57809,
      "training_step_time": 6.710052013397217
    },
    {
      "epoch": 5.5131912231445315e-05,
      "grad_norm": 0.10784849524497986,
      "learning_rate": 4.126675505135188e-06,
      "loss": 0.0589,
      "step": 57810
    },
    {
      "epoch": 5.5131912231445315e-05,
      "model_forward_time": 1.8605329990386963,
      "step": 57810
    },
    {
      "dynamics_loss_avg": 0.07521379627287388,
      "epoch": 5.5131912231445315e-05,
      "step": 57810
    },
    {
      "action_loss_avg": 0.0018754182499833405,
      "epoch": 5.5131912231445315e-05,
      "step": 57810
    },
    {
      "epoch": 5.5131912231445315e-05,
      "step": 57810,
      "training_step_time": 6.688477516174316
    },
    {
      "epoch": 5.513286590576172e-05,
      "model_forward_time": 1.861250877380371,
      "step": 57811
    },
    {
      "epoch": 5.513286590576172e-05,
      "step": 57811,
      "training_step_time": 7.458645820617676
    },
    {
      "epoch": 5.513381958007812e-05,
      "model_forward_time": 2.431363105773926,
      "step": 57812
    },
    {
      "epoch": 5.513381958007812e-05,
      "step": 57812,
      "training_step_time": 7.646308183670044
    },
    {
      "epoch": 5.5134773254394534e-05,
      "model_forward_time": 1.874523639678955,
      "step": 57813
    },
    {
      "epoch": 5.5134773254394534e-05,
      "step": 57813,
      "training_step_time": 8.282031297683716
    },
    {
      "epoch": 5.513572692871094e-05,
      "model_forward_time": 1.8650801181793213,
      "step": 57814
    },
    {
      "epoch": 5.513572692871094e-05,
      "step": 57814,
      "training_step_time": 7.617103815078735
    },
    {
      "epoch": 5.513668060302734e-05,
      "model_forward_time": 1.8607087135314941,
      "step": 57815
    },
    {
      "epoch": 5.513668060302734e-05,
      "step": 57815,
      "training_step_time": 6.74078631401062
    },
    {
      "epoch": 5.513763427734375e-05,
      "model_forward_time": 1.8595590591430664,
      "step": 57816
    },
    {
      "epoch": 5.513763427734375e-05,
      "step": 57816,
      "training_step_time": 6.696985483169556
    },
    {
      "epoch": 5.5138587951660156e-05,
      "model_forward_time": 1.8578228950500488,
      "step": 57817
    },
    {
      "epoch": 5.5138587951660156e-05,
      "step": 57817,
      "training_step_time": 6.634442567825317
    },
    {
      "epoch": 5.513954162597656e-05,
      "model_forward_time": 1.8559727668762207,
      "step": 57818
    },
    {
      "epoch": 5.513954162597656e-05,
      "step": 57818,
      "training_step_time": 6.7140953540802
    },
    {
      "epoch": 5.514049530029297e-05,
      "model_forward_time": 1.5039682388305664,
      "step": 57819
    },
    {
      "epoch": 5.514049530029297e-05,
      "step": 57819,
      "training_step_time": 6.720822334289551
    },
    {
      "epoch": 5.5141448974609375e-05,
      "grad_norm": 0.13383285701274872,
      "learning_rate": 4.125047500343169e-06,
      "loss": 0.056,
      "step": 57820
    },
    {
      "epoch": 5.5141448974609375e-05,
      "model_forward_time": 1.864593267440796,
      "step": 57820
    },
    {
      "dynamics_loss_avg": 0.061937588080763814,
      "epoch": 5.5141448974609375e-05,
      "step": 57820
    },
    {
      "action_loss_avg": 0.0028037925221724436,
      "epoch": 5.5141448974609375e-05,
      "step": 57820
    },
    {
      "epoch": 5.5141448974609375e-05,
      "step": 57820,
      "training_step_time": 7.518432140350342
    },
    {
      "epoch": 5.514240264892578e-05,
      "model_forward_time": 1.876471757888794,
      "step": 57821
    },
    {
      "epoch": 5.514240264892578e-05,
      "step": 57821,
      "training_step_time": 8.335853338241577
    },
    {
      "epoch": 5.514335632324219e-05,
      "model_forward_time": 2.410043716430664,
      "step": 57822
    },
    {
      "epoch": 5.514335632324219e-05,
      "step": 57822,
      "training_step_time": 9.791670799255371
    },
    {
      "epoch": 5.5144309997558594e-05,
      "model_forward_time": 1.8560998439788818,
      "step": 57823
    },
    {
      "epoch": 5.5144309997558594e-05,
      "step": 57823,
      "training_step_time": 7.652190685272217
    },
    {
      "epoch": 5.5145263671875e-05,
      "model_forward_time": 1.849034070968628,
      "step": 57824
    },
    {
      "epoch": 5.5145263671875e-05,
      "step": 57824,
      "training_step_time": 6.671388387680054
    },
    {
      "epoch": 5.514621734619141e-05,
      "model_forward_time": 1.4530065059661865,
      "step": 57825
    },
    {
      "epoch": 5.514621734619141e-05,
      "step": 57825,
      "training_step_time": 6.729746341705322
    },
    {
      "epoch": 5.514717102050781e-05,
      "model_forward_time": 1.848785638809204,
      "step": 57826
    },
    {
      "epoch": 5.514717102050781e-05,
      "step": 57826,
      "training_step_time": 6.759974002838135
    },
    {
      "epoch": 5.514812469482422e-05,
      "model_forward_time": 1.8566665649414062,
      "step": 57827
    },
    {
      "epoch": 5.514812469482422e-05,
      "step": 57827,
      "training_step_time": 6.764922618865967
    },
    {
      "epoch": 5.514907836914063e-05,
      "model_forward_time": 1.8571734428405762,
      "step": 57828
    },
    {
      "epoch": 5.514907836914063e-05,
      "step": 57828,
      "training_step_time": 6.686756610870361
    },
    {
      "epoch": 5.515003204345703e-05,
      "model_forward_time": 1.8593599796295166,
      "step": 57829
    },
    {
      "epoch": 5.515003204345703e-05,
      "step": 57829,
      "training_step_time": 8.399968385696411
    },
    {
      "epoch": 5.5150985717773436e-05,
      "grad_norm": 0.09462771564722061,
      "learning_rate": 4.123419591234637e-06,
      "loss": 0.0551,
      "step": 57830
    },
    {
      "epoch": 5.5150985717773436e-05,
      "model_forward_time": 1.8521716594696045,
      "step": 57830
    },
    {
      "dynamics_loss_avg": 0.05424698255956173,
      "epoch": 5.5150985717773436e-05,
      "step": 57830
    },
    {
      "action_loss_avg": 0.001575319393305108,
      "epoch": 5.5150985717773436e-05,
      "step": 57830
    },
    {
      "epoch": 5.5150985717773436e-05,
      "step": 57830,
      "training_step_time": 8.747716426849365
    },
    {
      "epoch": 5.515193939208985e-05,
      "model_forward_time": 2.4037787914276123,
      "step": 57831
    },
    {
      "epoch": 5.515193939208985e-05,
      "step": 57831,
      "training_step_time": 8.657936096191406
    },
    {
      "epoch": 5.515289306640625e-05,
      "model_forward_time": 1.852278470993042,
      "step": 57832
    },
    {
      "epoch": 5.515289306640625e-05,
      "step": 57832,
      "training_step_time": 7.704752683639526
    },
    {
      "epoch": 5.5153846740722655e-05,
      "model_forward_time": 1.8431432247161865,
      "step": 57833
    },
    {
      "epoch": 5.5153846740722655e-05,
      "step": 57833,
      "training_step_time": 6.70459246635437
    },
    {
      "epoch": 5.5154800415039066e-05,
      "model_forward_time": 1.8502001762390137,
      "step": 57834
    },
    {
      "epoch": 5.5154800415039066e-05,
      "step": 57834,
      "training_step_time": 6.697986125946045
    },
    {
      "epoch": 5.515575408935547e-05,
      "model_forward_time": 1.8574888706207275,
      "step": 57835
    },
    {
      "epoch": 5.515575408935547e-05,
      "step": 57835,
      "training_step_time": 6.73676609992981
    },
    {
      "epoch": 5.5156707763671874e-05,
      "model_forward_time": 1.503875494003296,
      "step": 57836
    },
    {
      "epoch": 5.5156707763671874e-05,
      "step": 57836,
      "training_step_time": 6.722824335098267
    },
    {
      "epoch": 5.5157661437988284e-05,
      "model_forward_time": 1.8608245849609375,
      "step": 57837
    },
    {
      "epoch": 5.5157661437988284e-05,
      "step": 57837,
      "training_step_time": 6.7214484214782715
    },
    {
      "epoch": 5.515861511230469e-05,
      "model_forward_time": 1.859982967376709,
      "step": 57838
    },
    {
      "epoch": 5.515861511230469e-05,
      "step": 57838,
      "training_step_time": 7.597988128662109
    },
    {
      "epoch": 5.515956878662109e-05,
      "model_forward_time": 1.8440654277801514,
      "step": 57839
    },
    {
      "epoch": 5.515956878662109e-05,
      "step": 57839,
      "training_step_time": 7.6641905307769775
    },
    {
      "epoch": 5.51605224609375e-05,
      "grad_norm": 0.0963740348815918,
      "learning_rate": 4.121791777987624e-06,
      "loss": 0.0572,
      "step": 57840
    },
    {
      "epoch": 5.51605224609375e-05,
      "model_forward_time": 2.409069061279297,
      "step": 57840
    },
    {
      "dynamics_loss_avg": 0.055639990232884885,
      "epoch": 5.51605224609375e-05,
      "step": 57840
    },
    {
      "action_loss_avg": 0.00136639601259958,
      "epoch": 5.51605224609375e-05,
      "step": 57840
    },
    {
      "epoch": 5.51605224609375e-05,
      "step": 57840,
      "training_step_time": 10.403166055679321
    },
    {
      "epoch": 5.516147613525391e-05,
      "model_forward_time": 1.8395509719848633,
      "step": 57841
    },
    {
      "epoch": 5.516147613525391e-05,
      "step": 57841,
      "training_step_time": 7.705446004867554
    },
    {
      "epoch": 5.516242980957031e-05,
      "model_forward_time": 1.848252773284912,
      "step": 57842
    },
    {
      "epoch": 5.516242980957031e-05,
      "step": 57842,
      "training_step_time": 7.371835708618164
    },
    {
      "epoch": 5.516338348388672e-05,
      "model_forward_time": 1.8571624755859375,
      "step": 57843
    },
    {
      "epoch": 5.516338348388672e-05,
      "step": 57843,
      "training_step_time": 6.669867753982544
    },
    {
      "epoch": 5.5164337158203126e-05,
      "model_forward_time": 1.8499724864959717,
      "step": 57844
    },
    {
      "epoch": 5.5164337158203126e-05,
      "step": 57844,
      "training_step_time": 6.743354797363281
    },
    {
      "epoch": 5.516529083251953e-05,
      "model_forward_time": 1.852081298828125,
      "step": 57845
    },
    {
      "epoch": 5.516529083251953e-05,
      "step": 57845,
      "training_step_time": 6.700902223587036
    },
    {
      "epoch": 5.5166244506835934e-05,
      "model_forward_time": 1.8437774181365967,
      "step": 57846
    },
    {
      "epoch": 5.5166244506835934e-05,
      "step": 57846,
      "training_step_time": 6.719600200653076
    },
    {
      "epoch": 5.5167198181152345e-05,
      "model_forward_time": 1.8394317626953125,
      "step": 57847
    },
    {
      "epoch": 5.5167198181152345e-05,
      "step": 57847,
      "training_step_time": 7.536935806274414
    },
    {
      "epoch": 5.516815185546875e-05,
      "model_forward_time": 1.8410747051239014,
      "step": 57848
    },
    {
      "epoch": 5.516815185546875e-05,
      "step": 57848,
      "training_step_time": 7.719338417053223
    },
    {
      "epoch": 5.516910552978515e-05,
      "model_forward_time": 2.408381462097168,
      "step": 57849
    },
    {
      "epoch": 5.516910552978515e-05,
      "step": 57849,
      "training_step_time": 8.84717082977295
    },
    {
      "epoch": 5.5170059204101564e-05,
      "grad_norm": 0.1409551054239273,
      "learning_rate": 4.1201640607801415e-06,
      "loss": 0.0562,
      "step": 57850
    },
    {
      "epoch": 5.5170059204101564e-05,
      "model_forward_time": 1.830885648727417,
      "step": 57850
    },
    {
      "dynamics_loss_avg": 0.048069835919886825,
      "epoch": 5.5170059204101564e-05,
      "step": 57850
    },
    {
      "action_loss_avg": 0.0010143031657207757,
      "epoch": 5.5170059204101564e-05,
      "step": 57850
    },
    {
      "epoch": 5.5170059204101564e-05,
      "step": 57850,
      "training_step_time": 8.413571834564209
    },
    {
      "epoch": 5.517101287841797e-05,
      "model_forward_time": 1.852320909500122,
      "step": 57851
    },
    {
      "epoch": 5.517101287841797e-05,
      "step": 57851,
      "training_step_time": 7.433522701263428
    },
    {
      "epoch": 5.517196655273437e-05,
      "model_forward_time": 1.8491909503936768,
      "step": 57852
    },
    {
      "epoch": 5.517196655273437e-05,
      "step": 57852,
      "training_step_time": 6.744719505310059
    },
    {
      "epoch": 5.517292022705078e-05,
      "model_forward_time": 1.8526649475097656,
      "step": 57853
    },
    {
      "epoch": 5.517292022705078e-05,
      "step": 57853,
      "training_step_time": 6.6739842891693115
    },
    {
      "epoch": 5.517387390136719e-05,
      "model_forward_time": 1.8370676040649414,
      "step": 57854
    },
    {
      "epoch": 5.517387390136719e-05,
      "step": 57854,
      "training_step_time": 6.715852737426758
    },
    {
      "epoch": 5.517482757568359e-05,
      "model_forward_time": 1.851301670074463,
      "step": 57855
    },
    {
      "epoch": 5.517482757568359e-05,
      "step": 57855,
      "training_step_time": 6.683420181274414
    },
    {
      "epoch": 5.517578125e-05,
      "model_forward_time": 1.8317921161651611,
      "step": 57856
    },
    {
      "epoch": 5.517578125e-05,
      "step": 57856,
      "training_step_time": 7.670950174331665
    },
    {
      "epoch": 5.5176734924316406e-05,
      "model_forward_time": 1.842585563659668,
      "step": 57857
    },
    {
      "epoch": 5.5176734924316406e-05,
      "step": 57857,
      "training_step_time": 8.539840459823608
    },
    {
      "epoch": 5.517768859863281e-05,
      "model_forward_time": 1.137108325958252,
      "step": 57858
    },
    {
      "epoch": 5.517768859863281e-05,
      "step": 57858,
      "training_step_time": 8.776916027069092
    },
    {
      "epoch": 5.517864227294922e-05,
      "model_forward_time": 2.3993918895721436,
      "step": 57859
    },
    {
      "epoch": 5.517864227294922e-05,
      "step": 57859,
      "training_step_time": 8.784080982208252
    },
    {
      "epoch": 5.5179595947265624e-05,
      "grad_norm": 0.11357665061950684,
      "learning_rate": 4.118536439790194e-06,
      "loss": 0.0547,
      "step": 57860
    },
    {
      "epoch": 5.5179595947265624e-05,
      "model_forward_time": 1.8405418395996094,
      "step": 57860
    },
    {
      "dynamics_loss_avg": 0.06936238091439009,
      "epoch": 5.5179595947265624e-05,
      "step": 57860
    },
    {
      "action_loss_avg": 0.0017297092359513043,
      "epoch": 5.5179595947265624e-05,
      "step": 57860
    },
    {
      "epoch": 5.5179595947265624e-05,
      "step": 57860,
      "training_step_time": 7.598587512969971
    },
    {
      "epoch": 5.518054962158203e-05,
      "model_forward_time": 1.862663745880127,
      "step": 57861
    },
    {
      "epoch": 5.518054962158203e-05,
      "step": 57861,
      "training_step_time": 6.671069145202637
    },
    {
      "epoch": 5.518150329589844e-05,
      "model_forward_time": 1.868866205215454,
      "step": 57862
    },
    {
      "epoch": 5.518150329589844e-05,
      "step": 57862,
      "training_step_time": 6.722877740859985
    },
    {
      "epoch": 5.518245697021484e-05,
      "model_forward_time": 1.8644628524780273,
      "step": 57863
    },
    {
      "epoch": 5.518245697021484e-05,
      "step": 57863,
      "training_step_time": 6.707259178161621
    },
    {
      "epoch": 5.518341064453125e-05,
      "model_forward_time": 1.849609613418579,
      "step": 57864
    },
    {
      "epoch": 5.518341064453125e-05,
      "step": 57864,
      "training_step_time": 6.662686109542847
    },
    {
      "epoch": 5.518436431884766e-05,
      "model_forward_time": 1.8559455871582031,
      "step": 57865
    },
    {
      "epoch": 5.518436431884766e-05,
      "step": 57865,
      "training_step_time": 7.534862518310547
    },
    {
      "epoch": 5.518531799316406e-05,
      "model_forward_time": 1.8574190139770508,
      "step": 57866
    },
    {
      "epoch": 5.518531799316406e-05,
      "step": 57866,
      "training_step_time": 9.736514329910278
    },
    {
      "epoch": 5.5186271667480466e-05,
      "model_forward_time": 1.8189120292663574,
      "step": 57867
    },
    {
      "epoch": 5.5186271667480466e-05,
      "step": 57867,
      "training_step_time": 8.172993183135986
    },
    {
      "epoch": 5.518722534179688e-05,
      "model_forward_time": 2.3711740970611572,
      "step": 57868
    },
    {
      "epoch": 5.518722534179688e-05,
      "step": 57868,
      "training_step_time": 11.846054315567017
    },
    {
      "epoch": 5.518817901611328e-05,
      "model_forward_time": 1.8505122661590576,
      "step": 57869
    },
    {
      "epoch": 5.518817901611328e-05,
      "step": 57869,
      "training_step_time": 8.51348876953125
    },
    {
      "epoch": 5.5189132690429685e-05,
      "grad_norm": 0.10238807648420334,
      "learning_rate": 4.116908915195776e-06,
      "loss": 0.0601,
      "step": 57870
    },
    {
      "epoch": 5.5189132690429685e-05,
      "model_forward_time": 1.8476998805999756,
      "step": 57870
    },
    {
      "dynamics_loss_avg": 0.06025797985494137,
      "epoch": 5.5189132690429685e-05,
      "step": 57870
    },
    {
      "action_loss_avg": 0.0018530968227423728,
      "epoch": 5.5189132690429685e-05,
      "step": 57870
    },
    {
      "epoch": 5.5189132690429685e-05,
      "step": 57870,
      "training_step_time": 6.672595739364624
    },
    {
      "epoch": 5.5190086364746096e-05,
      "model_forward_time": 1.8245952129364014,
      "step": 57871
    },
    {
      "epoch": 5.5190086364746096e-05,
      "step": 57871,
      "training_step_time": 6.718531370162964
    },
    {
      "epoch": 5.51910400390625e-05,
      "model_forward_time": 1.8492498397827148,
      "step": 57872
    },
    {
      "epoch": 5.51910400390625e-05,
      "step": 57872,
      "training_step_time": 6.72260594367981
    },
    {
      "epoch": 5.5191993713378904e-05,
      "model_forward_time": 1.8486509323120117,
      "step": 57873
    },
    {
      "epoch": 5.5191993713378904e-05,
      "step": 57873,
      "training_step_time": 6.703698396682739
    },
    {
      "epoch": 5.5192947387695315e-05,
      "model_forward_time": 1.8279271125793457,
      "step": 57874
    },
    {
      "epoch": 5.5192947387695315e-05,
      "step": 57874,
      "training_step_time": 7.599202394485474
    },
    {
      "epoch": 5.519390106201172e-05,
      "model_forward_time": 1.8320505619049072,
      "step": 57875
    },
    {
      "epoch": 5.519390106201172e-05,
      "step": 57875,
      "training_step_time": 7.606732606887817
    },
    {
      "epoch": 5.519485473632812e-05,
      "model_forward_time": 1.8007924556732178,
      "step": 57876
    },
    {
      "epoch": 5.519485473632812e-05,
      "step": 57876,
      "training_step_time": 8.219504356384277
    },
    {
      "epoch": 5.5195808410644534e-05,
      "model_forward_time": 2.356569290161133,
      "step": 57877
    },
    {
      "epoch": 5.5195808410644534e-05,
      "step": 57877,
      "training_step_time": 9.691313743591309
    },
    {
      "epoch": 5.519676208496094e-05,
      "model_forward_time": 1.8168938159942627,
      "step": 57878
    },
    {
      "epoch": 5.519676208496094e-05,
      "step": 57878,
      "training_step_time": 8.883548259735107
    },
    {
      "epoch": 5.519771575927734e-05,
      "model_forward_time": 1.8148527145385742,
      "step": 57879
    },
    {
      "epoch": 5.519771575927734e-05,
      "step": 57879,
      "training_step_time": 7.596304416656494
    },
    {
      "epoch": 5.519866943359375e-05,
      "grad_norm": 0.09710336476564407,
      "learning_rate": 4.115281487174872e-06,
      "loss": 0.0582,
      "step": 57880
    },
    {
      "epoch": 5.519866943359375e-05,
      "model_forward_time": 1.8201181888580322,
      "step": 57880
    },
    {
      "dynamics_loss_avg": 0.053295851312577726,
      "epoch": 5.519866943359375e-05,
      "step": 57880
    },
    {
      "action_loss_avg": 0.001810716520412825,
      "epoch": 5.519866943359375e-05,
      "step": 57880
    },
    {
      "epoch": 5.519866943359375e-05,
      "step": 57880,
      "training_step_time": 6.716472148895264
    },
    {
      "epoch": 5.5199623107910156e-05,
      "model_forward_time": 1.8104937076568604,
      "step": 57881
    },
    {
      "epoch": 5.5199623107910156e-05,
      "step": 57881,
      "training_step_time": 6.73272967338562
    },
    {
      "epoch": 5.520057678222656e-05,
      "model_forward_time": 1.8191885948181152,
      "step": 57882
    },
    {
      "epoch": 5.520057678222656e-05,
      "step": 57882,
      "training_step_time": 6.71262788772583
    },
    {
      "epoch": 5.520153045654297e-05,
      "model_forward_time": 1.81508469581604,
      "step": 57883
    },
    {
      "epoch": 5.520153045654297e-05,
      "step": 57883,
      "training_step_time": 7.532324314117432
    },
    {
      "epoch": 5.5202484130859375e-05,
      "model_forward_time": 1.8185248374938965,
      "step": 57884
    },
    {
      "epoch": 5.5202484130859375e-05,
      "step": 57884,
      "training_step_time": 7.771619558334351
    },
    {
      "epoch": 5.520343780517578e-05,
      "model_forward_time": 1.813145637512207,
      "step": 57885
    },
    {
      "epoch": 5.520343780517578e-05,
      "step": 57885,
      "training_step_time": 8.825358867645264
    },
    {
      "epoch": 5.520439147949219e-05,
      "model_forward_time": 2.3653616905212402,
      "step": 57886
    },
    {
      "epoch": 5.520439147949219e-05,
      "step": 57886,
      "training_step_time": 10.234862089157104
    },
    {
      "epoch": 5.5205345153808594e-05,
      "model_forward_time": 1.8193058967590332,
      "step": 57887
    },
    {
      "epoch": 5.5205345153808594e-05,
      "step": 57887,
      "training_step_time": 7.820505380630493
    },
    {
      "epoch": 5.5206298828125e-05,
      "model_forward_time": 1.8113386631011963,
      "step": 57888
    },
    {
      "epoch": 5.5206298828125e-05,
      "step": 57888,
      "training_step_time": 7.596812725067139
    },
    {
      "epoch": 5.520725250244141e-05,
      "model_forward_time": 1.8034648895263672,
      "step": 57889
    },
    {
      "epoch": 5.520725250244141e-05,
      "step": 57889,
      "training_step_time": 6.7617785930633545
    },
    {
      "epoch": 5.520820617675781e-05,
      "grad_norm": 0.13388361036777496,
      "learning_rate": 4.113654155905452e-06,
      "loss": 0.0549,
      "step": 57890
    },
    {
      "epoch": 5.520820617675781e-05,
      "model_forward_time": 1.823124885559082,
      "step": 57890
    },
    {
      "dynamics_loss_avg": 0.05402093045413494,
      "epoch": 5.520820617675781e-05,
      "step": 57890
    },
    {
      "action_loss_avg": 0.0016981547290924937,
      "epoch": 5.520820617675781e-05,
      "step": 57890
    },
    {
      "epoch": 5.520820617675781e-05,
      "step": 57890,
      "training_step_time": 6.706216096878052
    },
    {
      "epoch": 5.520915985107422e-05,
      "model_forward_time": 1.8211023807525635,
      "step": 57891
    },
    {
      "epoch": 5.520915985107422e-05,
      "step": 57891,
      "training_step_time": 6.708444595336914
    },
    {
      "epoch": 5.521011352539063e-05,
      "model_forward_time": 1.8235299587249756,
      "step": 57892
    },
    {
      "epoch": 5.521011352539063e-05,
      "step": 57892,
      "training_step_time": 6.718831539154053
    },
    {
      "epoch": 5.521106719970703e-05,
      "model_forward_time": 1.8248167037963867,
      "step": 57893
    },
    {
      "epoch": 5.521106719970703e-05,
      "step": 57893,
      "training_step_time": 7.5378851890563965
    },
    {
      "epoch": 5.5212020874023436e-05,
      "model_forward_time": 1.815725326538086,
      "step": 57894
    },
    {
      "epoch": 5.5212020874023436e-05,
      "step": 57894,
      "training_step_time": 7.8016626834869385
    },
    {
      "epoch": 5.521297454833985e-05,
      "model_forward_time": 2.367955207824707,
      "step": 57895
    },
    {
      "epoch": 5.521297454833985e-05,
      "step": 57895,
      "training_step_time": 9.357332468032837
    },
    {
      "epoch": 5.521392822265625e-05,
      "model_forward_time": 1.8202152252197266,
      "step": 57896
    },
    {
      "epoch": 5.521392822265625e-05,
      "step": 57896,
      "training_step_time": 7.7809367179870605
    },
    {
      "epoch": 5.5214881896972655e-05,
      "model_forward_time": 1.8224525451660156,
      "step": 57897
    },
    {
      "epoch": 5.5214881896972655e-05,
      "step": 57897,
      "training_step_time": 7.615268230438232
    },
    {
      "epoch": 5.5215835571289066e-05,
      "model_forward_time": 1.8152961730957031,
      "step": 57898
    },
    {
      "epoch": 5.5215835571289066e-05,
      "step": 57898,
      "training_step_time": 7.4037184715271
    },
    {
      "epoch": 5.521678924560547e-05,
      "model_forward_time": 1.8188183307647705,
      "step": 57899
    },
    {
      "epoch": 5.521678924560547e-05,
      "step": 57899,
      "training_step_time": 6.711028337478638
    },
    {
      "epoch": 5.5217742919921874e-05,
      "grad_norm": 0.11305928230285645,
      "learning_rate": 4.112026921565483e-06,
      "loss": 0.0567,
      "step": 57900
    },
    {
      "epoch": 5.5217742919921874e-05,
      "model_forward_time": 1.800374984741211,
      "step": 57900
    },
    {
      "dynamics_loss_avg": 0.048506688233464955,
      "epoch": 5.5217742919921874e-05,
      "step": 57900
    },
    {
      "action_loss_avg": 0.0014826065627858043,
      "epoch": 5.5217742919921874e-05,
      "step": 57900
    },
    {
      "epoch": 5.5217742919921874e-05,
      "step": 57900,
      "training_step_time": 6.704560995101929
    },
    {
      "epoch": 5.5218696594238284e-05,
      "model_forward_time": 1.8172860145568848,
      "step": 57901
    },
    {
      "epoch": 5.5218696594238284e-05,
      "step": 57901,
      "training_step_time": 6.762114524841309
    },
    {
      "epoch": 5.521965026855469e-05,
      "model_forward_time": 1.8135182857513428,
      "step": 57902
    },
    {
      "epoch": 5.521965026855469e-05,
      "step": 57902,
      "training_step_time": 7.582108974456787
    },
    {
      "epoch": 5.522060394287109e-05,
      "model_forward_time": 1.8024547100067139,
      "step": 57903
    },
    {
      "epoch": 5.522060394287109e-05,
      "step": 57903,
      "training_step_time": 9.579895257949829
    },
    {
      "epoch": 5.52215576171875e-05,
      "model_forward_time": 1.8094713687896729,
      "step": 57904
    },
    {
      "epoch": 5.52215576171875e-05,
      "step": 57904,
      "training_step_time": 9.124107122421265
    },
    {
      "epoch": 5.522251129150391e-05,
      "model_forward_time": 1.829491376876831,
      "step": 57905
    },
    {
      "epoch": 5.522251129150391e-05,
      "step": 57905,
      "training_step_time": 8.349043846130371
    },
    {
      "epoch": 5.522346496582031e-05,
      "model_forward_time": 1.8164405822753906,
      "step": 57906
    },
    {
      "epoch": 5.522346496582031e-05,
      "step": 57906,
      "training_step_time": 7.704177618026733
    },
    {
      "epoch": 5.5224418640136715e-05,
      "model_forward_time": 1.817720651626587,
      "step": 57907
    },
    {
      "epoch": 5.5224418640136715e-05,
      "step": 57907,
      "training_step_time": 7.370347738265991
    },
    {
      "epoch": 5.5225372314453126e-05,
      "model_forward_time": 1.8132014274597168,
      "step": 57908
    },
    {
      "epoch": 5.5225372314453126e-05,
      "step": 57908,
      "training_step_time": 6.744657039642334
    },
    {
      "epoch": 5.522632598876953e-05,
      "model_forward_time": 1.8189387321472168,
      "step": 57909
    },
    {
      "epoch": 5.522632598876953e-05,
      "step": 57909,
      "training_step_time": 6.760637283325195
    },
    {
      "epoch": 5.5227279663085934e-05,
      "grad_norm": 0.08382158726453781,
      "learning_rate": 4.110399784332915e-06,
      "loss": 0.0582,
      "step": 57910
    },
    {
      "epoch": 5.5227279663085934e-05,
      "model_forward_time": 1.8166024684906006,
      "step": 57910
    },
    {
      "dynamics_loss_avg": 0.058031470887362956,
      "epoch": 5.5227279663085934e-05,
      "step": 57910
    },
    {
      "action_loss_avg": 0.0015780238310981077,
      "epoch": 5.5227279663085934e-05,
      "step": 57910
    },
    {
      "epoch": 5.5227279663085934e-05,
      "step": 57910,
      "training_step_time": 6.772178649902344
    },
    {
      "epoch": 5.5228233337402345e-05,
      "model_forward_time": 1.8123741149902344,
      "step": 57911
    },
    {
      "epoch": 5.5228233337402345e-05,
      "step": 57911,
      "training_step_time": 7.599571466445923
    },
    {
      "epoch": 5.522918701171875e-05,
      "model_forward_time": 1.8193249702453613,
      "step": 57912
    },
    {
      "epoch": 5.522918701171875e-05,
      "step": 57912,
      "training_step_time": 7.745269536972046
    },
    {
      "epoch": 5.523014068603515e-05,
      "model_forward_time": 1.8094396591186523,
      "step": 57913
    },
    {
      "epoch": 5.523014068603515e-05,
      "step": 57913,
      "training_step_time": 9.45299243927002
    },
    {
      "epoch": 5.5231094360351564e-05,
      "model_forward_time": 2.3498194217681885,
      "step": 57914
    },
    {
      "epoch": 5.5231094360351564e-05,
      "step": 57914,
      "training_step_time": 8.665450096130371
    },
    {
      "epoch": 5.523204803466797e-05,
      "model_forward_time": 1.8005077838897705,
      "step": 57915
    },
    {
      "epoch": 5.523204803466797e-05,
      "step": 57915,
      "training_step_time": 8.812872171401978
    },
    {
      "epoch": 5.523300170898437e-05,
      "model_forward_time": 1.8061585426330566,
      "step": 57916
    },
    {
      "epoch": 5.523300170898437e-05,
      "step": 57916,
      "training_step_time": 7.322434425354004
    },
    {
      "epoch": 5.523395538330078e-05,
      "model_forward_time": 1.8240270614624023,
      "step": 57917
    },
    {
      "epoch": 5.523395538330078e-05,
      "step": 57917,
      "training_step_time": 6.7511677742004395
    },
    {
      "epoch": 5.523490905761719e-05,
      "model_forward_time": 1.8233017921447754,
      "step": 57918
    },
    {
      "epoch": 5.523490905761719e-05,
      "step": 57918,
      "training_step_time": 6.745605945587158
    },
    {
      "epoch": 5.523586273193359e-05,
      "model_forward_time": 1.818228006362915,
      "step": 57919
    },
    {
      "epoch": 5.523586273193359e-05,
      "step": 57919,
      "training_step_time": 6.791245460510254
    },
    {
      "epoch": 5.523681640625e-05,
      "grad_norm": 0.1165890246629715,
      "learning_rate": 4.108772744385689e-06,
      "loss": 0.0567,
      "step": 57920
    },
    {
      "epoch": 5.523681640625e-05,
      "model_forward_time": 1.8231019973754883,
      "step": 57920
    },
    {
      "dynamics_loss_avg": 0.05134894959628582,
      "epoch": 5.523681640625e-05,
      "step": 57920
    },
    {
      "action_loss_avg": 0.0014565327379386871,
      "epoch": 5.523681640625e-05,
      "step": 57920
    },
    {
      "epoch": 5.523681640625e-05,
      "step": 57920,
      "training_step_time": 7.287936449050903
    },
    {
      "epoch": 5.5237770080566406e-05,
      "model_forward_time": 1.8240375518798828,
      "step": 57921
    },
    {
      "epoch": 5.5237770080566406e-05,
      "step": 57921,
      "training_step_time": 7.645294904708862
    },
    {
      "epoch": 5.523872375488281e-05,
      "model_forward_time": 1.8216617107391357,
      "step": 57922
    },
    {
      "epoch": 5.523872375488281e-05,
      "step": 57922,
      "training_step_time": 8.551986694335938
    },
    {
      "epoch": 5.523967742919922e-05,
      "model_forward_time": 2.347351551055908,
      "step": 57923
    },
    {
      "epoch": 5.523967742919922e-05,
      "step": 57923,
      "training_step_time": 8.35143780708313
    },
    {
      "epoch": 5.5240631103515625e-05,
      "model_forward_time": 1.8215649127960205,
      "step": 57924
    },
    {
      "epoch": 5.5240631103515625e-05,
      "step": 57924,
      "training_step_time": 10.36297345161438
    },
    {
      "epoch": 5.524158477783203e-05,
      "model_forward_time": 1.823106050491333,
      "step": 57925
    },
    {
      "epoch": 5.524158477783203e-05,
      "step": 57925,
      "training_step_time": 7.705620765686035
    },
    {
      "epoch": 5.524253845214844e-05,
      "model_forward_time": 1.8202178478240967,
      "step": 57926
    },
    {
      "epoch": 5.524253845214844e-05,
      "step": 57926,
      "training_step_time": 7.312215805053711
    },
    {
      "epoch": 5.524349212646484e-05,
      "model_forward_time": 1.7924246788024902,
      "step": 57927
    },
    {
      "epoch": 5.524349212646484e-05,
      "step": 57927,
      "training_step_time": 6.648311138153076
    },
    {
      "epoch": 5.524444580078125e-05,
      "model_forward_time": 1.8160271644592285,
      "step": 57928
    },
    {
      "epoch": 5.524444580078125e-05,
      "step": 57928,
      "training_step_time": 6.76199197769165
    },
    {
      "epoch": 5.524539947509766e-05,
      "model_forward_time": 1.8197078704833984,
      "step": 57929
    },
    {
      "epoch": 5.524539947509766e-05,
      "step": 57929,
      "training_step_time": 6.749336004257202
    },
    {
      "epoch": 5.524635314941406e-05,
      "grad_norm": 0.1528315395116806,
      "learning_rate": 4.107145801901735e-06,
      "loss": 0.0569,
      "step": 57930
    },
    {
      "epoch": 5.524635314941406e-05,
      "model_forward_time": 1.817793607711792,
      "step": 57930
    },
    {
      "dynamics_loss_avg": 0.04814685378223658,
      "epoch": 5.524635314941406e-05,
      "step": 57930
    },
    {
      "action_loss_avg": 0.0011663200828479602,
      "epoch": 5.524635314941406e-05,
      "step": 57930
    },
    {
      "epoch": 5.524635314941406e-05,
      "step": 57930,
      "training_step_time": 8.726306676864624
    },
    {
      "epoch": 5.5247306823730466e-05,
      "model_forward_time": 1.810086727142334,
      "step": 57931
    },
    {
      "epoch": 5.5247306823730466e-05,
      "step": 57931,
      "training_step_time": 8.454720973968506
    },
    {
      "epoch": 5.524826049804688e-05,
      "model_forward_time": 2.328998327255249,
      "step": 57932
    },
    {
      "epoch": 5.524826049804688e-05,
      "step": 57932,
      "training_step_time": 10.560427904129028
    },
    {
      "epoch": 5.524921417236328e-05,
      "model_forward_time": 1.8142311573028564,
      "step": 57933
    },
    {
      "epoch": 5.524921417236328e-05,
      "step": 57933,
      "training_step_time": 8.576521158218384
    },
    {
      "epoch": 5.5250167846679685e-05,
      "model_forward_time": 1.8082549571990967,
      "step": 57934
    },
    {
      "epoch": 5.5250167846679685e-05,
      "step": 57934,
      "training_step_time": 7.661462306976318
    },
    {
      "epoch": 5.5251121520996096e-05,
      "model_forward_time": 1.8015408515930176,
      "step": 57935
    },
    {
      "epoch": 5.5251121520996096e-05,
      "step": 57935,
      "training_step_time": 7.492237091064453
    },
    {
      "epoch": 5.52520751953125e-05,
      "model_forward_time": 1.823258638381958,
      "step": 57936
    },
    {
      "epoch": 5.52520751953125e-05,
      "step": 57936,
      "training_step_time": 6.752742767333984
    },
    {
      "epoch": 5.5253028869628904e-05,
      "model_forward_time": 1.8157012462615967,
      "step": 57937
    },
    {
      "epoch": 5.5253028869628904e-05,
      "step": 57937,
      "training_step_time": 6.741456031799316
    },
    {
      "epoch": 5.5253982543945315e-05,
      "model_forward_time": 1.8207197189331055,
      "step": 57938
    },
    {
      "epoch": 5.5253982543945315e-05,
      "step": 57938,
      "training_step_time": 7.314326524734497
    },
    {
      "epoch": 5.525493621826172e-05,
      "model_forward_time": 1.8284871578216553,
      "step": 57939
    },
    {
      "epoch": 5.525493621826172e-05,
      "step": 57939,
      "training_step_time": 7.630898952484131
    },
    {
      "epoch": 5.525588989257812e-05,
      "grad_norm": 0.17625823616981506,
      "learning_rate": 4.105518957058974e-06,
      "loss": 0.0528,
      "step": 57940
    },
    {
      "epoch": 5.525588989257812e-05,
      "model_forward_time": 1.8190312385559082,
      "step": 57940
    },
    {
      "dynamics_loss_avg": 0.0681918328627944,
      "epoch": 5.525588989257812e-05,
      "step": 57940
    },
    {
      "action_loss_avg": 0.0018604970071464777,
      "epoch": 5.525588989257812e-05,
      "step": 57940
    },
    {
      "epoch": 5.525588989257812e-05,
      "step": 57940,
      "training_step_time": 8.391843318939209
    },
    {
      "epoch": 5.5256843566894534e-05,
      "model_forward_time": 2.345292091369629,
      "step": 57941
    },
    {
      "epoch": 5.5256843566894534e-05,
      "step": 57941,
      "training_step_time": 8.236386775970459
    },
    {
      "epoch": 5.525779724121094e-05,
      "model_forward_time": 1.8073344230651855,
      "step": 57942
    },
    {
      "epoch": 5.525779724121094e-05,
      "step": 57942,
      "training_step_time": 10.10281491279602
    },
    {
      "epoch": 5.525875091552734e-05,
      "model_forward_time": 1.8188440799713135,
      "step": 57943
    },
    {
      "epoch": 5.525875091552734e-05,
      "step": 57943,
      "training_step_time": 7.836362838745117
    },
    {
      "epoch": 5.525970458984375e-05,
      "model_forward_time": 1.8252534866333008,
      "step": 57944
    },
    {
      "epoch": 5.525970458984375e-05,
      "step": 57944,
      "training_step_time": 7.733458042144775
    },
    {
      "epoch": 5.5260658264160157e-05,
      "model_forward_time": 1.8159065246582031,
      "step": 57945
    },
    {
      "epoch": 5.5260658264160157e-05,
      "step": 57945,
      "training_step_time": 6.798320531845093
    },
    {
      "epoch": 5.526161193847656e-05,
      "model_forward_time": 1.8109509944915771,
      "step": 57946
    },
    {
      "epoch": 5.526161193847656e-05,
      "step": 57946,
      "training_step_time": 6.829336881637573
    },
    {
      "epoch": 5.526256561279297e-05,
      "model_forward_time": 1.7958040237426758,
      "step": 57947
    },
    {
      "epoch": 5.526256561279297e-05,
      "step": 57947,
      "training_step_time": 7.352570295333862
    },
    {
      "epoch": 5.5263519287109375e-05,
      "model_forward_time": 1.813000202178955,
      "step": 57948
    },
    {
      "epoch": 5.5263519287109375e-05,
      "step": 57948,
      "training_step_time": 7.681376695632935
    },
    {
      "epoch": 5.526447296142578e-05,
      "model_forward_time": 1.8061823844909668,
      "step": 57949
    },
    {
      "epoch": 5.526447296142578e-05,
      "step": 57949,
      "training_step_time": 8.36657452583313
    },
    {
      "epoch": 5.526542663574219e-05,
      "grad_norm": 0.15614111721515656,
      "learning_rate": 4.1038922100353155e-06,
      "loss": 0.0577,
      "step": 57950
    },
    {
      "epoch": 5.526542663574219e-05,
      "model_forward_time": 2.360057830810547,
      "step": 57950
    },
    {
      "dynamics_loss_avg": 0.06928184106945992,
      "epoch": 5.526542663574219e-05,
      "step": 57950
    },
    {
      "action_loss_avg": 0.001835112611297518,
      "epoch": 5.526542663574219e-05,
      "step": 57950
    },
    {
      "epoch": 5.526542663574219e-05,
      "step": 57950,
      "training_step_time": 8.363626956939697
    },
    {
      "epoch": 5.5266380310058594e-05,
      "model_forward_time": 1.8123114109039307,
      "step": 57951
    },
    {
      "epoch": 5.5266380310058594e-05,
      "step": 57951,
      "training_step_time": 8.714970350265503
    },
    {
      "epoch": 5.5267333984375e-05,
      "model_forward_time": 1.8124525547027588,
      "step": 57952
    },
    {
      "epoch": 5.5267333984375e-05,
      "step": 57952,
      "training_step_time": 7.745070219039917
    },
    {
      "epoch": 5.526828765869141e-05,
      "model_forward_time": 1.8123979568481445,
      "step": 57953
    },
    {
      "epoch": 5.526828765869141e-05,
      "step": 57953,
      "training_step_time": 7.713515758514404
    },
    {
      "epoch": 5.526924133300781e-05,
      "model_forward_time": 1.8084065914154053,
      "step": 57954
    },
    {
      "epoch": 5.526924133300781e-05,
      "step": 57954,
      "training_step_time": 7.30253529548645
    },
    {
      "epoch": 5.527019500732422e-05,
      "model_forward_time": 1.8111042976379395,
      "step": 57955
    },
    {
      "epoch": 5.527019500732422e-05,
      "step": 57955,
      "training_step_time": 6.749845266342163
    },
    {
      "epoch": 5.527114868164063e-05,
      "model_forward_time": 1.821382999420166,
      "step": 57956
    },
    {
      "epoch": 5.527114868164063e-05,
      "step": 57956,
      "training_step_time": 7.503625392913818
    },
    {
      "epoch": 5.527210235595703e-05,
      "model_forward_time": 1.8095521926879883,
      "step": 57957
    },
    {
      "epoch": 5.527210235595703e-05,
      "step": 57957,
      "training_step_time": 7.683220863342285
    },
    {
      "epoch": 5.5273056030273436e-05,
      "model_forward_time": 1.809002161026001,
      "step": 57958
    },
    {
      "epoch": 5.5273056030273436e-05,
      "step": 57958,
      "training_step_time": 8.511131286621094
    },
    {
      "epoch": 5.527400970458985e-05,
      "model_forward_time": 1.8189401626586914,
      "step": 57959
    },
    {
      "epoch": 5.527400970458985e-05,
      "step": 57959,
      "training_step_time": 8.854766607284546
    },
    {
      "epoch": 5.527496337890625e-05,
      "grad_norm": 0.09121127426624298,
      "learning_rate": 4.102265561008656e-06,
      "loss": 0.0538,
      "step": 57960
    },
    {
      "epoch": 5.527496337890625e-05,
      "model_forward_time": 2.3546552658081055,
      "step": 57960
    },
    {
      "dynamics_loss_avg": 0.06235205661505461,
      "epoch": 5.527496337890625e-05,
      "step": 57960
    },
    {
      "action_loss_avg": 0.0026911242515780033,
      "epoch": 5.527496337890625e-05,
      "step": 57960
    },
    {
      "epoch": 5.527496337890625e-05,
      "step": 57960,
      "training_step_time": 8.868759870529175
    },
    {
      "epoch": 5.5275917053222655e-05,
      "model_forward_time": 1.8198456764221191,
      "step": 57961
    },
    {
      "epoch": 5.5275917053222655e-05,
      "step": 57961,
      "training_step_time": 7.820437669754028
    },
    {
      "epoch": 5.5276870727539066e-05,
      "model_forward_time": 1.807227373123169,
      "step": 57962
    },
    {
      "epoch": 5.5276870727539066e-05,
      "step": 57962,
      "training_step_time": 7.718626022338867
    },
    {
      "epoch": 5.527782440185547e-05,
      "model_forward_time": 1.8162140846252441,
      "step": 57963
    },
    {
      "epoch": 5.527782440185547e-05,
      "step": 57963,
      "training_step_time": 7.567479133605957
    },
    {
      "epoch": 5.5278778076171874e-05,
      "model_forward_time": 1.8211827278137207,
      "step": 57964
    },
    {
      "epoch": 5.5278778076171874e-05,
      "step": 57964,
      "training_step_time": 6.821012496948242
    },
    {
      "epoch": 5.5279731750488285e-05,
      "model_forward_time": 1.811741590499878,
      "step": 57965
    },
    {
      "epoch": 5.5279731750488285e-05,
      "step": 57965,
      "training_step_time": 6.847141265869141
    },
    {
      "epoch": 5.528068542480469e-05,
      "model_forward_time": 1.8149793148040771,
      "step": 57966
    },
    {
      "epoch": 5.528068542480469e-05,
      "step": 57966,
      "training_step_time": 7.768066644668579
    },
    {
      "epoch": 5.528163909912109e-05,
      "model_forward_time": 1.816263198852539,
      "step": 57967
    },
    {
      "epoch": 5.528163909912109e-05,
      "step": 57967,
      "training_step_time": 8.390100955963135
    },
    {
      "epoch": 5.52825927734375e-05,
      "model_forward_time": 1.815622329711914,
      "step": 57968
    },
    {
      "epoch": 5.52825927734375e-05,
      "step": 57968,
      "training_step_time": 8.957707166671753
    },
    {
      "epoch": 5.528354644775391e-05,
      "model_forward_time": 2.3545942306518555,
      "step": 57969
    },
    {
      "epoch": 5.528354644775391e-05,
      "step": 57969,
      "training_step_time": 10.655259609222412
    },
    {
      "epoch": 5.528450012207031e-05,
      "grad_norm": 0.12603206932544708,
      "learning_rate": 4.100639010156886e-06,
      "loss": 0.0579,
      "step": 57970
    },
    {
      "epoch": 5.528450012207031e-05,
      "model_forward_time": 1.813518762588501,
      "step": 57970
    },
    {
      "dynamics_loss_avg": 0.06131819430738687,
      "epoch": 5.528450012207031e-05,
      "step": 57970
    },
    {
      "action_loss_avg": 0.0014458195248153061,
      "epoch": 5.528450012207031e-05,
      "step": 57970
    },
    {
      "epoch": 5.528450012207031e-05,
      "step": 57970,
      "training_step_time": 8.896883487701416
    },
    {
      "epoch": 5.5285453796386715e-05,
      "model_forward_time": 1.7969410419464111,
      "step": 57971
    },
    {
      "epoch": 5.5285453796386715e-05,
      "step": 57971,
      "training_step_time": 7.687765598297119
    },
    {
      "epoch": 5.5286407470703126e-05,
      "model_forward_time": 1.8153979778289795,
      "step": 57972
    },
    {
      "epoch": 5.5286407470703126e-05,
      "step": 57972,
      "training_step_time": 7.513646364212036
    },
    {
      "epoch": 5.528736114501953e-05,
      "model_forward_time": 1.8124547004699707,
      "step": 57973
    },
    {
      "epoch": 5.528736114501953e-05,
      "step": 57973,
      "training_step_time": 6.780620098114014
    },
    {
      "epoch": 5.5288314819335934e-05,
      "model_forward_time": 1.8137845993041992,
      "step": 57974
    },
    {
      "epoch": 5.5288314819335934e-05,
      "step": 57974,
      "training_step_time": 6.751102685928345
    },
    {
      "epoch": 5.5289268493652345e-05,
      "model_forward_time": 1.816523551940918,
      "step": 57975
    },
    {
      "epoch": 5.5289268493652345e-05,
      "step": 57975,
      "training_step_time": 7.68351149559021
    },
    {
      "epoch": 5.529022216796875e-05,
      "model_forward_time": 1.8217823505401611,
      "step": 57976
    },
    {
      "epoch": 5.529022216796875e-05,
      "step": 57976,
      "training_step_time": 7.616224050521851
    },
    {
      "epoch": 5.529117584228515e-05,
      "model_forward_time": 1.818230152130127,
      "step": 57977
    },
    {
      "epoch": 5.529117584228515e-05,
      "step": 57977,
      "training_step_time": 9.680558919906616
    },
    {
      "epoch": 5.5292129516601564e-05,
      "model_forward_time": 2.341444253921509,
      "step": 57978
    },
    {
      "epoch": 5.5292129516601564e-05,
      "step": 57978,
      "training_step_time": 10.47738528251648
    },
    {
      "epoch": 5.529308319091797e-05,
      "model_forward_time": 1.8229930400848389,
      "step": 57979
    },
    {
      "epoch": 5.529308319091797e-05,
      "step": 57979,
      "training_step_time": 9.385666370391846
    },
    {
      "epoch": 5.529403686523437e-05,
      "grad_norm": 0.13555961847305298,
      "learning_rate": 4.099012557657882e-06,
      "loss": 0.0539,
      "step": 57980
    },
    {
      "epoch": 5.529403686523437e-05,
      "model_forward_time": 1.8151521682739258,
      "step": 57980
    },
    {
      "dynamics_loss_avg": 0.04050679244101048,
      "epoch": 5.529403686523437e-05,
      "step": 57980
    },
    {
      "action_loss_avg": 0.0013145654083928093,
      "epoch": 5.529403686523437e-05,
      "step": 57980
    },
    {
      "epoch": 5.529403686523437e-05,
      "step": 57980,
      "training_step_time": 8.441402912139893
    },
    {
      "epoch": 5.529499053955078e-05,
      "model_forward_time": 1.8233418464660645,
      "step": 57981
    },
    {
      "epoch": 5.529499053955078e-05,
      "step": 57981,
      "training_step_time": 7.709463119506836
    },
    {
      "epoch": 5.529594421386719e-05,
      "model_forward_time": 1.8187892436981201,
      "step": 57982
    },
    {
      "epoch": 5.529594421386719e-05,
      "step": 57982,
      "training_step_time": 7.45574426651001
    },
    {
      "epoch": 5.529689788818359e-05,
      "model_forward_time": 1.8144431114196777,
      "step": 57983
    },
    {
      "epoch": 5.529689788818359e-05,
      "step": 57983,
      "training_step_time": 6.755150556564331
    },
    {
      "epoch": 5.52978515625e-05,
      "model_forward_time": 1.8177566528320312,
      "step": 57984
    },
    {
      "epoch": 5.52978515625e-05,
      "step": 57984,
      "training_step_time": 7.666754245758057
    },
    {
      "epoch": 5.5298805236816406e-05,
      "model_forward_time": 1.823845386505127,
      "step": 57985
    },
    {
      "epoch": 5.5298805236816406e-05,
      "step": 57985,
      "training_step_time": 8.683733224868774
    },
    {
      "epoch": 5.529975891113281e-05,
      "model_forward_time": 1.8237674236297607,
      "step": 57986
    },
    {
      "epoch": 5.529975891113281e-05,
      "step": 57986,
      "training_step_time": 8.404964208602905
    },
    {
      "epoch": 5.530071258544922e-05,
      "model_forward_time": 2.363816022872925,
      "step": 57987
    },
    {
      "epoch": 5.530071258544922e-05,
      "step": 57987,
      "training_step_time": 7.763385057449341
    },
    {
      "epoch": 5.5301666259765625e-05,
      "model_forward_time": 1.8169944286346436,
      "step": 57988
    },
    {
      "epoch": 5.5301666259765625e-05,
      "step": 57988,
      "training_step_time": 8.599955558776855
    },
    {
      "epoch": 5.530261993408203e-05,
      "model_forward_time": 1.8184967041015625,
      "step": 57989
    },
    {
      "epoch": 5.530261993408203e-05,
      "step": 57989,
      "training_step_time": 7.738938570022583
    },
    {
      "epoch": 5.530357360839844e-05,
      "grad_norm": 0.1382981240749359,
      "learning_rate": 4.09738620368951e-06,
      "loss": 0.0573,
      "step": 57990
    },
    {
      "epoch": 5.530357360839844e-05,
      "model_forward_time": 1.8109030723571777,
      "step": 57990
    },
    {
      "dynamics_loss_avg": 0.06107879346236587,
      "epoch": 5.530357360839844e-05,
      "step": 57990
    },
    {
      "action_loss_avg": 0.0024504065047949554,
      "epoch": 5.530357360839844e-05,
      "step": 57990
    },
    {
      "epoch": 5.530357360839844e-05,
      "step": 57990,
      "training_step_time": 7.683358669281006
    },
    {
      "epoch": 5.5304527282714843e-05,
      "model_forward_time": 1.4596154689788818,
      "step": 57991
    },
    {
      "epoch": 5.5304527282714843e-05,
      "step": 57991,
      "training_step_time": 7.446377277374268
    },
    {
      "epoch": 5.530548095703125e-05,
      "model_forward_time": 1.817117691040039,
      "step": 57992
    },
    {
      "epoch": 5.530548095703125e-05,
      "step": 57992,
      "training_step_time": 6.7635743618011475
    },
    {
      "epoch": 5.530643463134766e-05,
      "model_forward_time": 1.8200006484985352,
      "step": 57993
    },
    {
      "epoch": 5.530643463134766e-05,
      "step": 57993,
      "training_step_time": 7.65423583984375
    },
    {
      "epoch": 5.530738830566406e-05,
      "model_forward_time": 1.8188207149505615,
      "step": 57994
    },
    {
      "epoch": 5.530738830566406e-05,
      "step": 57994,
      "training_step_time": 7.62701416015625
    },
    {
      "epoch": 5.5308341979980466e-05,
      "model_forward_time": 1.8158268928527832,
      "step": 57995
    },
    {
      "epoch": 5.5308341979980466e-05,
      "step": 57995,
      "training_step_time": 7.743326902389526
    },
    {
      "epoch": 5.530929565429688e-05,
      "model_forward_time": 2.3429946899414062,
      "step": 57996
    },
    {
      "epoch": 5.530929565429688e-05,
      "step": 57996,
      "training_step_time": 9.098839521408081
    },
    {
      "epoch": 5.531024932861328e-05,
      "model_forward_time": 1.8301401138305664,
      "step": 57997
    },
    {
      "epoch": 5.531024932861328e-05,
      "step": 57997,
      "training_step_time": 7.808980941772461
    },
    {
      "epoch": 5.5311203002929685e-05,
      "model_forward_time": 1.832557201385498,
      "step": 57998
    },
    {
      "epoch": 5.5311203002929685e-05,
      "step": 57998,
      "training_step_time": 8.975711822509766
    },
    {
      "epoch": 5.5312156677246096e-05,
      "model_forward_time": 1.8104894161224365,
      "step": 57999
    },
    {
      "epoch": 5.5312156677246096e-05,
      "step": 57999,
      "training_step_time": 8.237995147705078
    },
    {
      "epoch": 5.53131103515625e-05,
      "grad_norm": 0.14224255084991455,
      "learning_rate": 4.095759948429626e-06,
      "loss": 0.0564,
      "step": 58000
    },
    {
      "epoch": 5.53131103515625e-05,
      "model_forward_time": 1.7426941394805908,
      "step": 58000
    },
    {
      "dynamics_loss_avg": 0.051400622725486754,
      "epoch": 5.53131103515625e-05,
      "step": 58000
    },
    {
      "action_loss_avg": 0.0015755921340314671,
      "epoch": 5.53131103515625e-05,
      "step": 58000
    },
    {
      "epoch": 5.53131103515625e-05,
      "step": 58000,
      "training_step_time": 6.605308532714844
    },
    {
      "epoch": 5.5314064025878904e-05,
      "model_forward_time": 1.7138218879699707,
      "step": 58001
    },
    {
      "epoch": 5.5314064025878904e-05,
      "step": 58001,
      "training_step_time": 6.7787065505981445
    },
    {
      "epoch": 5.5315017700195315e-05,
      "model_forward_time": 1.7244102954864502,
      "step": 58002
    },
    {
      "epoch": 5.5315017700195315e-05,
      "step": 58002,
      "training_step_time": 8.410107612609863
    },
    {
      "epoch": 5.531597137451172e-05,
      "model_forward_time": 1.727332353591919,
      "step": 58003
    },
    {
      "epoch": 5.531597137451172e-05,
      "step": 58003,
      "training_step_time": 8.734898567199707
    },
    {
      "epoch": 5.531692504882812e-05,
      "model_forward_time": 1.014728307723999,
      "step": 58004
    },
    {
      "epoch": 5.531692504882812e-05,
      "step": 58004,
      "training_step_time": 8.65314531326294
    },
    {
      "epoch": 5.5317878723144534e-05,
      "model_forward_time": 3.4272842407226562,
      "step": 58005
    },
    {
      "epoch": 5.5317878723144534e-05,
      "step": 58005,
      "training_step_time": 11.61815619468689
    },
    {
      "epoch": 5.531883239746094e-05,
      "model_forward_time": 1.7306885719299316,
      "step": 58006
    },
    {
      "epoch": 5.531883239746094e-05,
      "step": 58006,
      "training_step_time": 10.736593246459961
    },
    {
      "epoch": 5.531978607177734e-05,
      "model_forward_time": 1.7521514892578125,
      "step": 58007
    },
    {
      "epoch": 5.531978607177734e-05,
      "step": 58007,
      "training_step_time": 11.350217342376709
    },
    {
      "epoch": 5.532073974609375e-05,
      "model_forward_time": 1.723463535308838,
      "step": 58008
    },
    {
      "epoch": 5.532073974609375e-05,
      "step": 58008,
      "training_step_time": 7.486180067062378
    },
    {
      "epoch": 5.5321693420410157e-05,
      "model_forward_time": 1.732316017150879,
      "step": 58009
    },
    {
      "epoch": 5.5321693420410157e-05,
      "step": 58009,
      "training_step_time": 7.580008268356323
    },
    {
      "epoch": 5.532264709472656e-05,
      "grad_norm": 0.16010993719100952,
      "learning_rate": 4.0941337920560745e-06,
      "loss": 0.0541,
      "step": 58010
    },
    {
      "epoch": 5.532264709472656e-05,
      "model_forward_time": 1.7187914848327637,
      "step": 58010
    },
    {
      "dynamics_loss_avg": 0.041507047787308696,
      "epoch": 5.532264709472656e-05,
      "step": 58010
    },
    {
      "action_loss_avg": 0.0017216494656167925,
      "epoch": 5.532264709472656e-05,
      "step": 58010
    },
    {
      "epoch": 5.532264709472656e-05,
      "step": 58010,
      "training_step_time": 6.723003387451172
    },
    {
      "epoch": 5.532360076904297e-05,
      "model_forward_time": 1.7267580032348633,
      "step": 58011
    },
    {
      "epoch": 5.532360076904297e-05,
      "step": 58011,
      "training_step_time": 7.600556373596191
    },
    {
      "epoch": 5.5324554443359375e-05,
      "model_forward_time": 1.7208468914031982,
      "step": 58012
    },
    {
      "epoch": 5.5324554443359375e-05,
      "step": 58012,
      "training_step_time": 7.7746031284332275
    },
    {
      "epoch": 5.532550811767578e-05,
      "model_forward_time": 1.7301044464111328,
      "step": 58013
    },
    {
      "epoch": 5.532550811767578e-05,
      "step": 58013,
      "training_step_time": 7.32201886177063
    },
    {
      "epoch": 5.532646179199219e-05,
      "model_forward_time": 1.732680082321167,
      "step": 58014
    },
    {
      "epoch": 5.532646179199219e-05,
      "step": 58014,
      "training_step_time": 7.7074596881866455
    },
    {
      "epoch": 5.5327415466308594e-05,
      "model_forward_time": 2.8911197185516357,
      "step": 58015
    },
    {
      "epoch": 5.5327415466308594e-05,
      "step": 58015,
      "training_step_time": 8.744434118270874
    },
    {
      "epoch": 5.5328369140625e-05,
      "model_forward_time": 1.7530148029327393,
      "step": 58016
    },
    {
      "epoch": 5.5328369140625e-05,
      "step": 58016,
      "training_step_time": 9.28849458694458
    },
    {
      "epoch": 5.532932281494141e-05,
      "model_forward_time": 1.7270231246948242,
      "step": 58017
    },
    {
      "epoch": 5.532932281494141e-05,
      "step": 58017,
      "training_step_time": 7.691753149032593
    },
    {
      "epoch": 5.533027648925781e-05,
      "model_forward_time": 1.7284319400787354,
      "step": 58018
    },
    {
      "epoch": 5.533027648925781e-05,
      "step": 58018,
      "training_step_time": 7.608914375305176
    },
    {
      "epoch": 5.533123016357422e-05,
      "model_forward_time": 1.7189440727233887,
      "step": 58019
    },
    {
      "epoch": 5.533123016357422e-05,
      "step": 58019,
      "training_step_time": 7.100944757461548
    },
    {
      "epoch": 5.533218383789063e-05,
      "grad_norm": 0.10060063749551773,
      "learning_rate": 4.092507734746688e-06,
      "loss": 0.0586,
      "step": 58020
    },
    {
      "epoch": 5.533218383789063e-05,
      "model_forward_time": 1.7244467735290527,
      "step": 58020
    },
    {
      "dynamics_loss_avg": 0.05913688652217388,
      "epoch": 5.533218383789063e-05,
      "step": 58020
    },
    {
      "action_loss_avg": 0.0013144627358997242,
      "epoch": 5.533218383789063e-05,
      "step": 58020
    },
    {
      "epoch": 5.533218383789063e-05,
      "step": 58020,
      "training_step_time": 6.72121524810791
    },
    {
      "epoch": 5.533313751220703e-05,
      "model_forward_time": 1.7251367568969727,
      "step": 58021
    },
    {
      "epoch": 5.533313751220703e-05,
      "step": 58021,
      "training_step_time": 7.685857534408569
    },
    {
      "epoch": 5.5334091186523436e-05,
      "model_forward_time": 1.7251477241516113,
      "step": 58022
    },
    {
      "epoch": 5.5334091186523436e-05,
      "step": 58022,
      "training_step_time": 8.043920040130615
    },
    {
      "epoch": 5.533504486083985e-05,
      "model_forward_time": 1.7279136180877686,
      "step": 58023
    },
    {
      "epoch": 5.533504486083985e-05,
      "step": 58023,
      "training_step_time": 8.591009616851807
    },
    {
      "epoch": 5.533599853515625e-05,
      "model_forward_time": 2.9430999755859375,
      "step": 58024
    },
    {
      "epoch": 5.533599853515625e-05,
      "step": 58024,
      "training_step_time": 7.731399059295654
    },
    {
      "epoch": 5.5336952209472655e-05,
      "model_forward_time": 1.7417101860046387,
      "step": 58025
    },
    {
      "epoch": 5.5336952209472655e-05,
      "step": 58025,
      "training_step_time": 7.786103248596191
    },
    {
      "epoch": 5.5337905883789066e-05,
      "model_forward_time": 1.7243714332580566,
      "step": 58026
    },
    {
      "epoch": 5.5337905883789066e-05,
      "step": 58026,
      "training_step_time": 7.763216972351074
    },
    {
      "epoch": 5.533885955810547e-05,
      "model_forward_time": 1.7219603061676025,
      "step": 58027
    },
    {
      "epoch": 5.533885955810547e-05,
      "step": 58027,
      "training_step_time": 8.333218812942505
    },
    {
      "epoch": 5.5339813232421874e-05,
      "model_forward_time": 1.72475266456604,
      "step": 58028
    },
    {
      "epoch": 5.5339813232421874e-05,
      "step": 58028,
      "training_step_time": 7.037014722824097
    },
    {
      "epoch": 5.5340766906738285e-05,
      "model_forward_time": 1.7240784168243408,
      "step": 58029
    },
    {
      "epoch": 5.5340766906738285e-05,
      "step": 58029,
      "training_step_time": 6.72294020652771
    },
    {
      "epoch": 5.534172058105469e-05,
      "grad_norm": 0.08975289762020111,
      "learning_rate": 4.090881776679293e-06,
      "loss": 0.0555,
      "step": 58030
    },
    {
      "epoch": 5.534172058105469e-05,
      "model_forward_time": 1.7229163646697998,
      "step": 58030
    },
    {
      "dynamics_loss_avg": 0.05763321202248335,
      "epoch": 5.534172058105469e-05,
      "step": 58030
    },
    {
      "action_loss_avg": 0.00208793071215041,
      "epoch": 5.534172058105469e-05,
      "step": 58030
    },
    {
      "epoch": 5.534172058105469e-05,
      "step": 58030,
      "training_step_time": 7.772754192352295
    },
    {
      "epoch": 5.534267425537109e-05,
      "model_forward_time": 1.7273080348968506,
      "step": 58031
    },
    {
      "epoch": 5.534267425537109e-05,
      "step": 58031,
      "training_step_time": 7.386334657669067
    },
    {
      "epoch": 5.53436279296875e-05,
      "model_forward_time": 1.7309582233428955,
      "step": 58032
    },
    {
      "epoch": 5.53436279296875e-05,
      "step": 58032,
      "training_step_time": 7.730989933013916
    },
    {
      "epoch": 5.534458160400391e-05,
      "model_forward_time": 2.861356019973755,
      "step": 58033
    },
    {
      "epoch": 5.534458160400391e-05,
      "step": 58033,
      "training_step_time": 9.045315265655518
    },
    {
      "epoch": 5.534553527832031e-05,
      "model_forward_time": 1.7340991497039795,
      "step": 58034
    },
    {
      "epoch": 5.534553527832031e-05,
      "step": 58034,
      "training_step_time": 8.778306722640991
    },
    {
      "epoch": 5.5346488952636715e-05,
      "model_forward_time": 1.7274043560028076,
      "step": 58035
    },
    {
      "epoch": 5.5346488952636715e-05,
      "step": 58035,
      "training_step_time": 7.70558500289917
    },
    {
      "epoch": 5.5347442626953126e-05,
      "model_forward_time": 1.7290196418762207,
      "step": 58036
    },
    {
      "epoch": 5.5347442626953126e-05,
      "step": 58036,
      "training_step_time": 7.553252696990967
    },
    {
      "epoch": 5.534839630126953e-05,
      "model_forward_time": 1.725266456604004,
      "step": 58037
    },
    {
      "epoch": 5.534839630126953e-05,
      "step": 58037,
      "training_step_time": 7.440113306045532
    },
    {
      "epoch": 5.5349349975585934e-05,
      "model_forward_time": 1.7243952751159668,
      "step": 58038
    },
    {
      "epoch": 5.5349349975585934e-05,
      "step": 58038,
      "training_step_time": 6.720606803894043
    },
    {
      "epoch": 5.5350303649902345e-05,
      "model_forward_time": 1.3520739078521729,
      "step": 58039
    },
    {
      "epoch": 5.5350303649902345e-05,
      "step": 58039,
      "training_step_time": 7.6949851512908936
    },
    {
      "epoch": 5.535125732421875e-05,
      "grad_norm": 0.08636646717786789,
      "learning_rate": 4.0892559180317e-06,
      "loss": 0.0575,
      "step": 58040
    },
    {
      "epoch": 5.535125732421875e-05,
      "model_forward_time": 1.720106840133667,
      "step": 58040
    },
    {
      "dynamics_loss_avg": 0.05151063371449709,
      "epoch": 5.535125732421875e-05,
      "step": 58040
    },
    {
      "action_loss_avg": 0.0018580661155283452,
      "epoch": 5.535125732421875e-05,
      "step": 58040
    },
    {
      "epoch": 5.535125732421875e-05,
      "step": 58040,
      "training_step_time": 7.169046640396118
    },
    {
      "epoch": 5.535221099853515e-05,
      "model_forward_time": 1.7280945777893066,
      "step": 58041
    },
    {
      "epoch": 5.535221099853515e-05,
      "step": 58041,
      "training_step_time": 7.7277302742004395
    },
    {
      "epoch": 5.5353164672851564e-05,
      "model_forward_time": 2.833251476287842,
      "step": 58042
    },
    {
      "epoch": 5.5353164672851564e-05,
      "step": 58042,
      "training_step_time": 8.795267105102539
    },
    {
      "epoch": 5.535411834716797e-05,
      "model_forward_time": 1.7279112339019775,
      "step": 58043
    },
    {
      "epoch": 5.535411834716797e-05,
      "step": 58043,
      "training_step_time": 8.557426452636719
    },
    {
      "epoch": 5.535507202148437e-05,
      "model_forward_time": 1.7212646007537842,
      "step": 58044
    },
    {
      "epoch": 5.535507202148437e-05,
      "step": 58044,
      "training_step_time": 7.73398494720459
    },
    {
      "epoch": 5.535602569580078e-05,
      "model_forward_time": 1.7354116439819336,
      "step": 58045
    },
    {
      "epoch": 5.535602569580078e-05,
      "step": 58045,
      "training_step_time": 8.611701488494873
    },
    {
      "epoch": 5.535697937011719e-05,
      "model_forward_time": 1.7258117198944092,
      "step": 58046
    },
    {
      "epoch": 5.535697937011719e-05,
      "step": 58046,
      "training_step_time": 7.469313144683838
    },
    {
      "epoch": 5.535793304443359e-05,
      "model_forward_time": 1.7236156463623047,
      "step": 58047
    },
    {
      "epoch": 5.535793304443359e-05,
      "step": 58047,
      "training_step_time": 7.120846748352051
    },
    {
      "epoch": 5.535888671875e-05,
      "model_forward_time": 1.725998878479004,
      "step": 58048
    },
    {
      "epoch": 5.535888671875e-05,
      "step": 58048,
      "training_step_time": 7.670783758163452
    },
    {
      "epoch": 5.5359840393066406e-05,
      "model_forward_time": 1.7245988845825195,
      "step": 58049
    },
    {
      "epoch": 5.5359840393066406e-05,
      "step": 58049,
      "training_step_time": 7.750536918640137
    },
    {
      "epoch": 5.536079406738281e-05,
      "grad_norm": 0.10898900777101517,
      "learning_rate": 4.08763015898171e-06,
      "loss": 0.0561,
      "step": 58050
    },
    {
      "epoch": 5.536079406738281e-05,
      "model_forward_time": 1.7278175354003906,
      "step": 58050
    },
    {
      "dynamics_loss_avg": 0.06548427324742079,
      "epoch": 5.536079406738281e-05,
      "step": 58050
    },
    {
      "action_loss_avg": 0.0017907865083543583,
      "epoch": 5.536079406738281e-05,
      "step": 58050
    },
    {
      "epoch": 5.536079406738281e-05,
      "step": 58050,
      "training_step_time": 7.66103458404541
    },
    {
      "epoch": 5.536174774169922e-05,
      "model_forward_time": 1.7280895709991455,
      "step": 58051
    },
    {
      "epoch": 5.536174774169922e-05,
      "step": 58051,
      "training_step_time": 10.168217182159424
    },
    {
      "epoch": 5.5362701416015625e-05,
      "model_forward_time": 1.3779339790344238,
      "step": 58052
    },
    {
      "epoch": 5.5362701416015625e-05,
      "step": 58052,
      "training_step_time": 8.866577863693237
    },
    {
      "epoch": 5.536365509033203e-05,
      "model_forward_time": 1.735058307647705,
      "step": 58053
    },
    {
      "epoch": 5.536365509033203e-05,
      "step": 58053,
      "training_step_time": 9.524891138076782
    },
    {
      "epoch": 5.536460876464844e-05,
      "model_forward_time": 1.7245314121246338,
      "step": 58054
    },
    {
      "epoch": 5.536460876464844e-05,
      "step": 58054,
      "training_step_time": 7.651359796524048
    },
    {
      "epoch": 5.5365562438964843e-05,
      "model_forward_time": 1.723710060119629,
      "step": 58055
    },
    {
      "epoch": 5.5365562438964843e-05,
      "step": 58055,
      "training_step_time": 7.585511684417725
    },
    {
      "epoch": 5.536651611328125e-05,
      "model_forward_time": 1.3525340557098389,
      "step": 58056
    },
    {
      "epoch": 5.536651611328125e-05,
      "step": 58056,
      "training_step_time": 7.060925245285034
    },
    {
      "epoch": 5.536746978759766e-05,
      "model_forward_time": 1.722388744354248,
      "step": 58057
    },
    {
      "epoch": 5.536746978759766e-05,
      "step": 58057,
      "training_step_time": 7.568383693695068
    },
    {
      "epoch": 5.536842346191406e-05,
      "model_forward_time": 1.717729091644287,
      "step": 58058
    },
    {
      "epoch": 5.536842346191406e-05,
      "step": 58058,
      "training_step_time": 7.768319845199585
    },
    {
      "epoch": 5.5369377136230466e-05,
      "model_forward_time": 1.7228920459747314,
      "step": 58059
    },
    {
      "epoch": 5.5369377136230466e-05,
      "step": 58059,
      "training_step_time": 8.095357894897461
    },
    {
      "epoch": 5.537033081054688e-05,
      "grad_norm": 0.10446537286043167,
      "learning_rate": 4.086004499707115e-06,
      "loss": 0.0594,
      "step": 58060
    },
    {
      "epoch": 5.537033081054688e-05,
      "model_forward_time": 1.0109038352966309,
      "step": 58060
    },
    {
      "dynamics_loss_avg": 0.04469895306974649,
      "epoch": 5.537033081054688e-05,
      "step": 58060
    },
    {
      "action_loss_avg": 0.001998065543011762,
      "epoch": 5.537033081054688e-05,
      "step": 58060
    },
    {
      "epoch": 5.537033081054688e-05,
      "step": 58060,
      "training_step_time": 9.318111419677734
    },
    {
      "epoch": 5.537128448486328e-05,
      "model_forward_time": 2.8373610973358154,
      "step": 58061
    },
    {
      "epoch": 5.537128448486328e-05,
      "step": 58061,
      "training_step_time": 9.05338168144226
    },
    {
      "epoch": 5.5372238159179685e-05,
      "model_forward_time": 1.7455997467041016,
      "step": 58062
    },
    {
      "epoch": 5.5372238159179685e-05,
      "step": 58062,
      "training_step_time": 9.932557582855225
    },
    {
      "epoch": 5.5373191833496096e-05,
      "model_forward_time": 1.7268438339233398,
      "step": 58063
    },
    {
      "epoch": 5.5373191833496096e-05,
      "step": 58063,
      "training_step_time": 7.719127893447876
    },
    {
      "epoch": 5.53741455078125e-05,
      "model_forward_time": 1.7213420867919922,
      "step": 58064
    },
    {
      "epoch": 5.53741455078125e-05,
      "step": 58064,
      "training_step_time": 7.5887839794158936
    },
    {
      "epoch": 5.5375099182128904e-05,
      "model_forward_time": 1.725414752960205,
      "step": 58065
    },
    {
      "epoch": 5.5375099182128904e-05,
      "step": 58065,
      "training_step_time": 7.082858324050903
    },
    {
      "epoch": 5.5376052856445315e-05,
      "model_forward_time": 1.7285513877868652,
      "step": 58066
    },
    {
      "epoch": 5.5376052856445315e-05,
      "step": 58066,
      "training_step_time": 6.687676429748535
    },
    {
      "epoch": 5.537700653076172e-05,
      "model_forward_time": 1.725475788116455,
      "step": 58067
    },
    {
      "epoch": 5.537700653076172e-05,
      "step": 58067,
      "training_step_time": 7.738408327102661
    },
    {
      "epoch": 5.537796020507812e-05,
      "model_forward_time": 1.729325294494629,
      "step": 58068
    },
    {
      "epoch": 5.537796020507812e-05,
      "step": 58068,
      "training_step_time": 7.323230266571045
    },
    {
      "epoch": 5.5378913879394534e-05,
      "model_forward_time": 1.7126810550689697,
      "step": 58069
    },
    {
      "epoch": 5.5378913879394534e-05,
      "step": 58069,
      "training_step_time": 9.839008808135986
    },
    {
      "epoch": 5.537986755371094e-05,
      "grad_norm": 0.11219976097345352,
      "learning_rate": 4.084378940385693e-06,
      "loss": 0.0536,
      "step": 58070
    },
    {
      "epoch": 5.537986755371094e-05,
      "model_forward_time": 1.7396514415740967,
      "step": 58070
    },
    {
      "dynamics_loss_avg": 0.04483294058591127,
      "epoch": 5.537986755371094e-05,
      "step": 58070
    },
    {
      "action_loss_avg": 0.0012099118495825679,
      "epoch": 5.537986755371094e-05,
      "step": 58070
    },
    {
      "epoch": 5.537986755371094e-05,
      "step": 58070,
      "training_step_time": 10.5739164352417
    },
    {
      "epoch": 5.538082122802734e-05,
      "model_forward_time": 1.7316994667053223,
      "step": 58071
    },
    {
      "epoch": 5.538082122802734e-05,
      "step": 58071,
      "training_step_time": 9.424531698226929
    },
    {
      "epoch": 5.538177490234375e-05,
      "model_forward_time": 1.7249259948730469,
      "step": 58072
    },
    {
      "epoch": 5.538177490234375e-05,
      "step": 58072,
      "training_step_time": 7.6063902378082275
    },
    {
      "epoch": 5.538272857666016e-05,
      "model_forward_time": 1.7230441570281982,
      "step": 58073
    },
    {
      "epoch": 5.538272857666016e-05,
      "step": 58073,
      "training_step_time": 7.587881803512573
    },
    {
      "epoch": 5.538368225097656e-05,
      "model_forward_time": 1.7370071411132812,
      "step": 58074
    },
    {
      "epoch": 5.538368225097656e-05,
      "step": 58074,
      "training_step_time": 7.044611930847168
    },
    {
      "epoch": 5.538463592529297e-05,
      "model_forward_time": 1.7180664539337158,
      "step": 58075
    },
    {
      "epoch": 5.538463592529297e-05,
      "step": 58075,
      "training_step_time": 6.707308530807495
    },
    {
      "epoch": 5.5385589599609375e-05,
      "model_forward_time": 1.7255229949951172,
      "step": 58076
    },
    {
      "epoch": 5.5385589599609375e-05,
      "step": 58076,
      "training_step_time": 7.72713565826416
    },
    {
      "epoch": 5.538654327392578e-05,
      "model_forward_time": 1.7145307064056396,
      "step": 58077
    },
    {
      "epoch": 5.538654327392578e-05,
      "step": 58077,
      "training_step_time": 8.3620023727417
    },
    {
      "epoch": 5.538749694824219e-05,
      "model_forward_time": 1.7251994609832764,
      "step": 58078
    },
    {
      "epoch": 5.538749694824219e-05,
      "step": 58078,
      "training_step_time": 8.849784135818481
    },
    {
      "epoch": 5.5388450622558594e-05,
      "model_forward_time": 1.7368834018707275,
      "step": 58079
    },
    {
      "epoch": 5.5388450622558594e-05,
      "step": 58079,
      "training_step_time": 8.22451901435852
    },
    {
      "epoch": 5.5389404296875e-05,
      "grad_norm": 0.10648923367261887,
      "learning_rate": 4.082753481195212e-06,
      "loss": 0.0558,
      "step": 58080
    },
    {
      "epoch": 5.5389404296875e-05,
      "model_forward_time": 2.963010311126709,
      "step": 58080
    },
    {
      "dynamics_loss_avg": 0.03625005353242159,
      "epoch": 5.5389404296875e-05,
      "step": 58080
    },
    {
      "action_loss_avg": 0.001485523849260062,
      "epoch": 5.5389404296875e-05,
      "step": 58080
    },
    {
      "epoch": 5.5389404296875e-05,
      "step": 58080,
      "training_step_time": 7.8470299243927
    },
    {
      "epoch": 5.539035797119141e-05,
      "model_forward_time": 1.726719617843628,
      "step": 58081
    },
    {
      "epoch": 5.539035797119141e-05,
      "step": 58081,
      "training_step_time": 8.829954862594604
    },
    {
      "epoch": 5.539131164550781e-05,
      "model_forward_time": 1.7224345207214355,
      "step": 58082
    },
    {
      "epoch": 5.539131164550781e-05,
      "step": 58082,
      "training_step_time": 7.721625089645386
    },
    {
      "epoch": 5.539226531982422e-05,
      "model_forward_time": 1.7262907028198242,
      "step": 58083
    },
    {
      "epoch": 5.539226531982422e-05,
      "step": 58083,
      "training_step_time": 7.158634424209595
    },
    {
      "epoch": 5.539321899414063e-05,
      "model_forward_time": 1.7310574054718018,
      "step": 58084
    },
    {
      "epoch": 5.539321899414063e-05,
      "step": 58084,
      "training_step_time": 6.718791246414185
    },
    {
      "epoch": 5.539417266845703e-05,
      "model_forward_time": 1.728269338607788,
      "step": 58085
    },
    {
      "epoch": 5.539417266845703e-05,
      "step": 58085,
      "training_step_time": 7.765048027038574
    },
    {
      "epoch": 5.5395126342773436e-05,
      "model_forward_time": 1.7195029258728027,
      "step": 58086
    },
    {
      "epoch": 5.5395126342773436e-05,
      "step": 58086,
      "training_step_time": 7.280698776245117
    },
    {
      "epoch": 5.539608001708985e-05,
      "model_forward_time": 1.7302711009979248,
      "step": 58087
    },
    {
      "epoch": 5.539608001708985e-05,
      "step": 58087,
      "training_step_time": 7.755299091339111
    },
    {
      "epoch": 5.539703369140625e-05,
      "model_forward_time": 1.7277426719665527,
      "step": 58088
    },
    {
      "epoch": 5.539703369140625e-05,
      "step": 58088,
      "training_step_time": 10.537309646606445
    },
    {
      "epoch": 5.5397987365722655e-05,
      "model_forward_time": 2.8850631713867188,
      "step": 58089
    },
    {
      "epoch": 5.5397987365722655e-05,
      "step": 58089,
      "training_step_time": 8.815860509872437
    },
    {
      "epoch": 5.5398941040039066e-05,
      "grad_norm": 0.11642248183488846,
      "learning_rate": 4.081128122313433e-06,
      "loss": 0.0579,
      "step": 58090
    },
    {
      "epoch": 5.5398941040039066e-05,
      "model_forward_time": 1.7577908039093018,
      "step": 58090
    },
    {
      "dynamics_loss_avg": 0.04737353352829814,
      "epoch": 5.5398941040039066e-05,
      "step": 58090
    },
    {
      "action_loss_avg": 0.0021794076776131986,
      "epoch": 5.5398941040039066e-05,
      "step": 58090
    },
    {
      "epoch": 5.5398941040039066e-05,
      "step": 58090,
      "training_step_time": 8.631731510162354
    },
    {
      "epoch": 5.539989471435547e-05,
      "model_forward_time": 1.726231336593628,
      "step": 58091
    },
    {
      "epoch": 5.539989471435547e-05,
      "step": 58091,
      "training_step_time": 7.725720167160034
    },
    {
      "epoch": 5.5400848388671874e-05,
      "model_forward_time": 1.7213168144226074,
      "step": 58092
    },
    {
      "epoch": 5.5400848388671874e-05,
      "step": 58092,
      "training_step_time": 7.500389814376831
    },
    {
      "epoch": 5.5401802062988285e-05,
      "model_forward_time": 1.7264995574951172,
      "step": 58093
    },
    {
      "epoch": 5.5401802062988285e-05,
      "step": 58093,
      "training_step_time": 6.69673752784729
    },
    {
      "epoch": 5.540275573730469e-05,
      "model_forward_time": 1.7217071056365967,
      "step": 58094
    },
    {
      "epoch": 5.540275573730469e-05,
      "step": 58094,
      "training_step_time": 7.781397104263306
    },
    {
      "epoch": 5.540370941162109e-05,
      "model_forward_time": 1.7221593856811523,
      "step": 58095
    },
    {
      "epoch": 5.540370941162109e-05,
      "step": 58095,
      "training_step_time": 7.319847822189331
    },
    {
      "epoch": 5.54046630859375e-05,
      "model_forward_time": 1.7263267040252686,
      "step": 58096
    },
    {
      "epoch": 5.54046630859375e-05,
      "step": 58096,
      "training_step_time": 7.803204298019409
    },
    {
      "epoch": 5.540561676025391e-05,
      "model_forward_time": 1.724794626235962,
      "step": 58097
    },
    {
      "epoch": 5.540561676025391e-05,
      "step": 58097,
      "training_step_time": 8.35813856124878
    },
    {
      "epoch": 5.540657043457031e-05,
      "model_forward_time": 2.916700839996338,
      "step": 58098
    },
    {
      "epoch": 5.540657043457031e-05,
      "step": 58098,
      "training_step_time": 8.768921852111816
    },
    {
      "epoch": 5.5407524108886716e-05,
      "model_forward_time": 1.7481157779693604,
      "step": 58099
    },
    {
      "epoch": 5.5407524108886716e-05,
      "step": 58099,
      "training_step_time": 8.255798578262329
    },
    {
      "epoch": 5.5408477783203126e-05,
      "grad_norm": 0.09523897618055344,
      "learning_rate": 4.0795028639181e-06,
      "loss": 0.0563,
      "step": 58100
    },
    {
      "epoch": 5.5408477783203126e-05,
      "model_forward_time": 1.7179286479949951,
      "step": 58100
    },
    {
      "dynamics_loss_avg": 0.0460072347894311,
      "epoch": 5.5408477783203126e-05,
      "step": 58100
    },
    {
      "action_loss_avg": 0.0011844985710922629,
      "epoch": 5.5408477783203126e-05,
      "step": 58100
    },
    {
      "epoch": 5.5408477783203126e-05,
      "step": 58100,
      "training_step_time": 7.699878454208374
    },
    {
      "epoch": 5.540943145751953e-05,
      "model_forward_time": 1.7280263900756836,
      "step": 58101
    },
    {
      "epoch": 5.540943145751953e-05,
      "step": 58101,
      "training_step_time": 7.5742387771606445
    },
    {
      "epoch": 5.5410385131835934e-05,
      "model_forward_time": 1.729403018951416,
      "step": 58102
    },
    {
      "epoch": 5.5410385131835934e-05,
      "step": 58102,
      "training_step_time": 7.084916114807129
    },
    {
      "epoch": 5.5411338806152345e-05,
      "model_forward_time": 1.7266347408294678,
      "step": 58103
    },
    {
      "epoch": 5.5411338806152345e-05,
      "step": 58103,
      "training_step_time": 7.5878143310546875
    },
    {
      "epoch": 5.541229248046875e-05,
      "model_forward_time": 1.7218947410583496,
      "step": 58104
    },
    {
      "epoch": 5.541229248046875e-05,
      "step": 58104,
      "training_step_time": 7.712269306182861
    },
    {
      "epoch": 5.541324615478515e-05,
      "model_forward_time": 1.7267868518829346,
      "step": 58105
    },
    {
      "epoch": 5.541324615478515e-05,
      "step": 58105,
      "training_step_time": 7.744717836380005
    },
    {
      "epoch": 5.5414199829101564e-05,
      "model_forward_time": 1.7321126461029053,
      "step": 58106
    },
    {
      "epoch": 5.5414199829101564e-05,
      "step": 58106,
      "training_step_time": 9.133102893829346
    },
    {
      "epoch": 5.541515350341797e-05,
      "model_forward_time": 2.880540132522583,
      "step": 58107
    },
    {
      "epoch": 5.541515350341797e-05,
      "step": 58107,
      "training_step_time": 10.526856899261475
    },
    {
      "epoch": 5.541610717773437e-05,
      "model_forward_time": 1.7434678077697754,
      "step": 58108
    },
    {
      "epoch": 5.541610717773437e-05,
      "step": 58108,
      "training_step_time": 7.76478910446167
    },
    {
      "epoch": 5.541706085205078e-05,
      "model_forward_time": 1.727663278579712,
      "step": 58109
    },
    {
      "epoch": 5.541706085205078e-05,
      "step": 58109,
      "training_step_time": 7.73580527305603
    },
    {
      "epoch": 5.541801452636719e-05,
      "grad_norm": 0.10825520008802414,
      "learning_rate": 4.077877706186951e-06,
      "loss": 0.0575,
      "step": 58110
    },
    {
      "epoch": 5.541801452636719e-05,
      "model_forward_time": 1.7227635383605957,
      "step": 58110
    },
    {
      "dynamics_loss_avg": 0.03975706398487091,
      "epoch": 5.541801452636719e-05,
      "step": 58110
    },
    {
      "action_loss_avg": 0.0010208868567133321,
      "epoch": 5.541801452636719e-05,
      "step": 58110
    },
    {
      "epoch": 5.541801452636719e-05,
      "step": 58110,
      "training_step_time": 7.672200918197632
    },
    {
      "epoch": 5.541896820068359e-05,
      "model_forward_time": 1.727938175201416,
      "step": 58111
    },
    {
      "epoch": 5.541896820068359e-05,
      "step": 58111,
      "training_step_time": 7.080671548843384
    },
    {
      "epoch": 5.5419921875e-05,
      "model_forward_time": 1.7242567539215088,
      "step": 58112
    },
    {
      "epoch": 5.5419921875e-05,
      "step": 58112,
      "training_step_time": 7.532914638519287
    },
    {
      "epoch": 5.5420875549316406e-05,
      "model_forward_time": 1.7232606410980225,
      "step": 58113
    },
    {
      "epoch": 5.5420875549316406e-05,
      "step": 58113,
      "training_step_time": 7.748035669326782
    },
    {
      "epoch": 5.542182922363281e-05,
      "model_forward_time": 1.7272298336029053,
      "step": 58114
    },
    {
      "epoch": 5.542182922363281e-05,
      "step": 58114,
      "training_step_time": 7.739837884902954
    },
    {
      "epoch": 5.542278289794922e-05,
      "model_forward_time": 1.3534328937530518,
      "step": 58115
    },
    {
      "epoch": 5.542278289794922e-05,
      "step": 58115,
      "training_step_time": 8.479028701782227
    },
    {
      "epoch": 5.5423736572265625e-05,
      "model_forward_time": 2.848832130432129,
      "step": 58116
    },
    {
      "epoch": 5.5423736572265625e-05,
      "step": 58116,
      "training_step_time": 7.752089262008667
    },
    {
      "epoch": 5.542469024658203e-05,
      "model_forward_time": 1.736189842224121,
      "step": 58117
    },
    {
      "epoch": 5.542469024658203e-05,
      "step": 58117,
      "training_step_time": 7.77960467338562
    },
    {
      "epoch": 5.542564392089844e-05,
      "model_forward_time": 1.724081039428711,
      "step": 58118
    },
    {
      "epoch": 5.542564392089844e-05,
      "step": 58118,
      "training_step_time": 7.735378265380859
    },
    {
      "epoch": 5.5426597595214844e-05,
      "model_forward_time": 1.7312111854553223,
      "step": 58119
    },
    {
      "epoch": 5.5426597595214844e-05,
      "step": 58119,
      "training_step_time": 7.624024868011475
    },
    {
      "epoch": 5.542755126953125e-05,
      "grad_norm": 0.10146534442901611,
      "learning_rate": 4.076252649297709e-06,
      "loss": 0.0576,
      "step": 58120
    },
    {
      "epoch": 5.542755126953125e-05,
      "model_forward_time": 1.724224328994751,
      "step": 58120
    },
    {
      "dynamics_loss_avg": 0.04720450825989246,
      "epoch": 5.542755126953125e-05,
      "step": 58120
    },
    {
      "action_loss_avg": 0.0015932596434140578,
      "epoch": 5.542755126953125e-05,
      "step": 58120
    },
    {
      "epoch": 5.542755126953125e-05,
      "step": 58120,
      "training_step_time": 7.091242551803589
    },
    {
      "epoch": 5.542850494384766e-05,
      "model_forward_time": 1.7352557182312012,
      "step": 58121
    },
    {
      "epoch": 5.542850494384766e-05,
      "step": 58121,
      "training_step_time": 7.1869635581970215
    },
    {
      "epoch": 5.542945861816406e-05,
      "model_forward_time": 1.7230799198150635,
      "step": 58122
    },
    {
      "epoch": 5.542945861816406e-05,
      "step": 58122,
      "training_step_time": 7.773622751235962
    },
    {
      "epoch": 5.5430412292480466e-05,
      "model_forward_time": 1.730276346206665,
      "step": 58123
    },
    {
      "epoch": 5.5430412292480466e-05,
      "step": 58123,
      "training_step_time": 7.758213996887207
    },
    {
      "epoch": 5.543136596679688e-05,
      "model_forward_time": 1.7352912425994873,
      "step": 58124
    },
    {
      "epoch": 5.543136596679688e-05,
      "step": 58124,
      "training_step_time": 8.610266208648682
    },
    {
      "epoch": 5.543231964111328e-05,
      "model_forward_time": 2.8602983951568604,
      "step": 58125
    },
    {
      "epoch": 5.543231964111328e-05,
      "step": 58125,
      "training_step_time": 7.721074342727661
    },
    {
      "epoch": 5.5433273315429685e-05,
      "model_forward_time": 1.7418537139892578,
      "step": 58126
    },
    {
      "epoch": 5.5433273315429685e-05,
      "step": 58126,
      "training_step_time": 8.461931228637695
    },
    {
      "epoch": 5.5434226989746096e-05,
      "model_forward_time": 1.7375226020812988,
      "step": 58127
    },
    {
      "epoch": 5.5434226989746096e-05,
      "step": 58127,
      "training_step_time": 7.696395397186279
    },
    {
      "epoch": 5.54351806640625e-05,
      "model_forward_time": 1.7354755401611328,
      "step": 58128
    },
    {
      "epoch": 5.54351806640625e-05,
      "step": 58128,
      "training_step_time": 8.68297791481018
    },
    {
      "epoch": 5.5436134338378904e-05,
      "model_forward_time": 1.7365436553955078,
      "step": 58129
    },
    {
      "epoch": 5.5436134338378904e-05,
      "step": 58129,
      "training_step_time": 7.04708456993103
    },
    {
      "epoch": 5.5437088012695315e-05,
      "grad_norm": 0.12615585327148438,
      "learning_rate": 4.07462769342809e-06,
      "loss": 0.0605,
      "step": 58130
    },
    {
      "epoch": 5.5437088012695315e-05,
      "model_forward_time": 1.7335808277130127,
      "step": 58130
    },
    {
      "dynamics_loss_avg": 0.04717306299135089,
      "epoch": 5.5437088012695315e-05,
      "step": 58130
    },
    {
      "action_loss_avg": 0.0009741178932017647,
      "epoch": 5.5437088012695315e-05,
      "step": 58130
    },
    {
      "epoch": 5.5437088012695315e-05,
      "step": 58130,
      "training_step_time": 7.28282618522644
    },
    {
      "epoch": 5.543804168701172e-05,
      "model_forward_time": 1.7399792671203613,
      "step": 58131
    },
    {
      "epoch": 5.543804168701172e-05,
      "step": 58131,
      "training_step_time": 7.807786226272583
    },
    {
      "epoch": 5.543899536132812e-05,
      "model_forward_time": 1.7276806831359863,
      "step": 58132
    },
    {
      "epoch": 5.543899536132812e-05,
      "step": 58132,
      "training_step_time": 7.691011428833008
    },
    {
      "epoch": 5.5439949035644534e-05,
      "model_forward_time": 1.7402780055999756,
      "step": 58133
    },
    {
      "epoch": 5.5439949035644534e-05,
      "step": 58133,
      "training_step_time": 7.904150485992432
    },
    {
      "epoch": 5.544090270996094e-05,
      "model_forward_time": 2.853696823120117,
      "step": 58134
    },
    {
      "epoch": 5.544090270996094e-05,
      "step": 58134,
      "training_step_time": 9.66701626777649
    },
    {
      "epoch": 5.544185638427734e-05,
      "model_forward_time": 1.7430624961853027,
      "step": 58135
    },
    {
      "epoch": 5.544185638427734e-05,
      "step": 58135,
      "training_step_time": 8.92989730834961
    },
    {
      "epoch": 5.544281005859375e-05,
      "model_forward_time": 1.753610610961914,
      "step": 58136
    },
    {
      "epoch": 5.544281005859375e-05,
      "step": 58136,
      "training_step_time": 8.714629888534546
    },
    {
      "epoch": 5.544376373291016e-05,
      "model_forward_time": 1.7347426414489746,
      "step": 58137
    },
    {
      "epoch": 5.544376373291016e-05,
      "step": 58137,
      "training_step_time": 7.716804027557373
    },
    {
      "epoch": 5.544471740722656e-05,
      "model_forward_time": 1.7358591556549072,
      "step": 58138
    },
    {
      "epoch": 5.544471740722656e-05,
      "step": 58138,
      "training_step_time": 7.568963527679443
    },
    {
      "epoch": 5.544567108154297e-05,
      "model_forward_time": 1.7396893501281738,
      "step": 58139
    },
    {
      "epoch": 5.544567108154297e-05,
      "step": 58139,
      "training_step_time": 7.182405948638916
    },
    {
      "epoch": 5.5446624755859376e-05,
      "grad_norm": 0.1305118054151535,
      "learning_rate": 4.073002838755793e-06,
      "loss": 0.058,
      "step": 58140
    },
    {
      "epoch": 5.5446624755859376e-05,
      "model_forward_time": 1.734727144241333,
      "step": 58140
    },
    {
      "dynamics_loss_avg": 0.0680821431800723,
      "epoch": 5.5446624755859376e-05,
      "step": 58140
    },
    {
      "action_loss_avg": 0.0020446007663849743,
      "epoch": 5.5446624755859376e-05,
      "step": 58140
    },
    {
      "epoch": 5.5446624755859376e-05,
      "step": 58140,
      "training_step_time": 8.685415744781494
    },
    {
      "epoch": 5.544757843017578e-05,
      "model_forward_time": 1.7390930652618408,
      "step": 58141
    },
    {
      "epoch": 5.544757843017578e-05,
      "step": 58141,
      "training_step_time": 7.7316858768463135
    },
    {
      "epoch": 5.544853210449219e-05,
      "model_forward_time": 1.7307469844818115,
      "step": 58142
    },
    {
      "epoch": 5.544853210449219e-05,
      "step": 58142,
      "training_step_time": 7.762430191040039
    },
    {
      "epoch": 5.5449485778808594e-05,
      "model_forward_time": 2.8428761959075928,
      "step": 58143
    },
    {
      "epoch": 5.5449485778808594e-05,
      "step": 58143,
      "training_step_time": 7.81989049911499
    },
    {
      "epoch": 5.5450439453125e-05,
      "model_forward_time": 1.7691786289215088,
      "step": 58144
    },
    {
      "epoch": 5.5450439453125e-05,
      "step": 58144,
      "training_step_time": 8.339376211166382
    },
    {
      "epoch": 5.545139312744141e-05,
      "model_forward_time": 1.745875597000122,
      "step": 58145
    },
    {
      "epoch": 5.545139312744141e-05,
      "step": 58145,
      "training_step_time": 8.330776453018188
    },
    {
      "epoch": 5.545234680175781e-05,
      "model_forward_time": 1.741516351699829,
      "step": 58146
    },
    {
      "epoch": 5.545234680175781e-05,
      "step": 58146,
      "training_step_time": 7.667020797729492
    },
    {
      "epoch": 5.545330047607422e-05,
      "model_forward_time": 1.7419507503509521,
      "step": 58147
    },
    {
      "epoch": 5.545330047607422e-05,
      "step": 58147,
      "training_step_time": 7.684327840805054
    },
    {
      "epoch": 5.545425415039063e-05,
      "model_forward_time": 1.744171142578125,
      "step": 58148
    },
    {
      "epoch": 5.545425415039063e-05,
      "step": 58148,
      "training_step_time": 7.1612772941589355
    },
    {
      "epoch": 5.545520782470703e-05,
      "model_forward_time": 1.7399518489837646,
      "step": 58149
    },
    {
      "epoch": 5.545520782470703e-05,
      "step": 58149,
      "training_step_time": 7.5975446701049805
    },
    {
      "epoch": 5.5456161499023436e-05,
      "grad_norm": 0.11918769776821136,
      "learning_rate": 4.071378085458511e-06,
      "loss": 0.0558,
      "step": 58150
    },
    {
      "epoch": 5.5456161499023436e-05,
      "model_forward_time": 1.7316884994506836,
      "step": 58150
    },
    {
      "dynamics_loss_avg": 0.04088397230952978,
      "epoch": 5.5456161499023436e-05,
      "step": 58150
    },
    {
      "action_loss_avg": 0.0011498730949824677,
      "epoch": 5.5456161499023436e-05,
      "step": 58150
    },
    {
      "epoch": 5.5456161499023436e-05,
      "step": 58150,
      "training_step_time": 7.808405160903931
    },
    {
      "epoch": 5.545711517333985e-05,
      "model_forward_time": 1.7371134757995605,
      "step": 58151
    },
    {
      "epoch": 5.545711517333985e-05,
      "step": 58151,
      "training_step_time": 8.435782670974731
    },
    {
      "epoch": 5.545806884765625e-05,
      "model_forward_time": 1.7339556217193604,
      "step": 58152
    },
    {
      "epoch": 5.545806884765625e-05,
      "step": 58152,
      "training_step_time": 8.922661542892456
    },
    {
      "epoch": 5.5459022521972655e-05,
      "model_forward_time": 2.904107093811035,
      "step": 58153
    },
    {
      "epoch": 5.5459022521972655e-05,
      "step": 58153,
      "training_step_time": 9.60090684890747
    },
    {
      "epoch": 5.5459976196289066e-05,
      "model_forward_time": 1.7337734699249268,
      "step": 58154
    },
    {
      "epoch": 5.5459976196289066e-05,
      "step": 58154,
      "training_step_time": 8.897292613983154
    },
    {
      "epoch": 5.546092987060547e-05,
      "model_forward_time": 1.7467377185821533,
      "step": 58155
    },
    {
      "epoch": 5.546092987060547e-05,
      "step": 58155,
      "training_step_time": 7.806595087051392
    },
    {
      "epoch": 5.5461883544921874e-05,
      "model_forward_time": 1.7377779483795166,
      "step": 58156
    },
    {
      "epoch": 5.5461883544921874e-05,
      "step": 58156,
      "training_step_time": 7.787600517272949
    },
    {
      "epoch": 5.546283721923828e-05,
      "model_forward_time": 1.735149621963501,
      "step": 58157
    },
    {
      "epoch": 5.546283721923828e-05,
      "step": 58157,
      "training_step_time": 7.340649366378784
    },
    {
      "epoch": 5.546379089355469e-05,
      "model_forward_time": 1.7435622215270996,
      "step": 58158
    },
    {
      "epoch": 5.546379089355469e-05,
      "step": 58158,
      "training_step_time": 6.871397495269775
    },
    {
      "epoch": 5.546474456787109e-05,
      "model_forward_time": 1.7600536346435547,
      "step": 58159
    },
    {
      "epoch": 5.546474456787109e-05,
      "step": 58159,
      "training_step_time": 7.912163734436035
    },
    {
      "epoch": 5.54656982421875e-05,
      "grad_norm": 0.10972002893686295,
      "learning_rate": 4.069753433713928e-06,
      "loss": 0.055,
      "step": 58160
    },
    {
      "epoch": 5.54656982421875e-05,
      "model_forward_time": 1.7473559379577637,
      "step": 58160
    },
    {
      "dynamics_loss_avg": 0.0646342808380723,
      "epoch": 5.54656982421875e-05,
      "step": 58160
    },
    {
      "action_loss_avg": 0.0013367535313591361,
      "epoch": 5.54656982421875e-05,
      "step": 58160
    },
    {
      "epoch": 5.54656982421875e-05,
      "step": 58160,
      "training_step_time": 8.411592960357666
    },
    {
      "epoch": 5.546665191650391e-05,
      "model_forward_time": 1.7514863014221191,
      "step": 58161
    },
    {
      "epoch": 5.546665191650391e-05,
      "step": 58161,
      "training_step_time": 9.692968845367432
    },
    {
      "epoch": 5.546760559082031e-05,
      "model_forward_time": 2.949901819229126,
      "step": 58162
    },
    {
      "epoch": 5.546760559082031e-05,
      "step": 58162,
      "training_step_time": 8.623852729797363
    },
    {
      "epoch": 5.5468559265136716e-05,
      "model_forward_time": 1.7196919918060303,
      "step": 58163
    },
    {
      "epoch": 5.5468559265136716e-05,
      "step": 58163,
      "training_step_time": 7.939331769943237
    },
    {
      "epoch": 5.5469512939453126e-05,
      "model_forward_time": 1.7405271530151367,
      "step": 58164
    },
    {
      "epoch": 5.5469512939453126e-05,
      "step": 58164,
      "training_step_time": 8.34952998161316
    },
    {
      "epoch": 5.547046661376953e-05,
      "model_forward_time": 1.7175581455230713,
      "step": 58165
    },
    {
      "epoch": 5.547046661376953e-05,
      "step": 58165,
      "training_step_time": 7.811298131942749
    },
    {
      "epoch": 5.5471420288085934e-05,
      "model_forward_time": 1.719161033630371,
      "step": 58166
    },
    {
      "epoch": 5.5471420288085934e-05,
      "step": 58166,
      "training_step_time": 7.711194276809692
    },
    {
      "epoch": 5.5472373962402345e-05,
      "model_forward_time": 1.7311604022979736,
      "step": 58167
    },
    {
      "epoch": 5.5472373962402345e-05,
      "step": 58167,
      "training_step_time": 6.899254560470581
    },
    {
      "epoch": 5.547332763671875e-05,
      "model_forward_time": 1.7257139682769775,
      "step": 58168
    },
    {
      "epoch": 5.547332763671875e-05,
      "step": 58168,
      "training_step_time": 7.780995607376099
    },
    {
      "epoch": 5.547428131103515e-05,
      "model_forward_time": 1.714231014251709,
      "step": 58169
    },
    {
      "epoch": 5.547428131103515e-05,
      "step": 58169,
      "training_step_time": 8.745199918746948
    },
    {
      "epoch": 5.5475234985351564e-05,
      "grad_norm": 0.10828954726457596,
      "learning_rate": 4.068128883699709e-06,
      "loss": 0.0565,
      "step": 58170
    },
    {
      "epoch": 5.5475234985351564e-05,
      "model_forward_time": 1.7130558490753174,
      "step": 58170
    },
    {
      "dynamics_loss_avg": 0.06836117338389158,
      "epoch": 5.5475234985351564e-05,
      "step": 58170
    },
    {
      "action_loss_avg": 0.0015403310651890933,
      "epoch": 5.5475234985351564e-05,
      "step": 58170
    },
    {
      "epoch": 5.5475234985351564e-05,
      "step": 58170,
      "training_step_time": 7.7501490116119385
    },
    {
      "epoch": 5.547618865966797e-05,
      "model_forward_time": 2.9091720581054688,
      "step": 58171
    },
    {
      "epoch": 5.547618865966797e-05,
      "step": 58171,
      "training_step_time": 7.703320026397705
    },
    {
      "epoch": 5.547714233398437e-05,
      "model_forward_time": 1.7317557334899902,
      "step": 58172
    },
    {
      "epoch": 5.547714233398437e-05,
      "step": 58172,
      "training_step_time": 7.719839334487915
    },
    {
      "epoch": 5.547809600830078e-05,
      "model_forward_time": 1.7657363414764404,
      "step": 58173
    },
    {
      "epoch": 5.547809600830078e-05,
      "step": 58173,
      "training_step_time": 7.770156621932983
    },
    {
      "epoch": 5.547904968261719e-05,
      "model_forward_time": 1.3737866878509521,
      "step": 58174
    },
    {
      "epoch": 5.547904968261719e-05,
      "step": 58174,
      "training_step_time": 7.591217041015625
    },
    {
      "epoch": 5.548000335693359e-05,
      "model_forward_time": 1.7400343418121338,
      "step": 58175
    },
    {
      "epoch": 5.548000335693359e-05,
      "step": 58175,
      "training_step_time": 7.586893558502197
    },
    {
      "epoch": 5.548095703125e-05,
      "model_forward_time": 1.7472507953643799,
      "step": 58176
    },
    {
      "epoch": 5.548095703125e-05,
      "step": 58176,
      "training_step_time": 6.767828464508057
    },
    {
      "epoch": 5.5481910705566406e-05,
      "model_forward_time": 1.7376682758331299,
      "step": 58177
    },
    {
      "epoch": 5.5481910705566406e-05,
      "step": 58177,
      "training_step_time": 8.804100036621094
    },
    {
      "epoch": 5.548286437988281e-05,
      "model_forward_time": 1.7448561191558838,
      "step": 58178
    },
    {
      "epoch": 5.548286437988281e-05,
      "step": 58178,
      "training_step_time": 7.732104301452637
    },
    {
      "epoch": 5.548381805419922e-05,
      "model_forward_time": 1.7453389167785645,
      "step": 58179
    },
    {
      "epoch": 5.548381805419922e-05,
      "step": 58179,
      "training_step_time": 9.780440092086792
    },
    {
      "epoch": 5.5484771728515625e-05,
      "grad_norm": 0.1122266948223114,
      "learning_rate": 4.066504435593516e-06,
      "loss": 0.0562,
      "step": 58180
    },
    {
      "epoch": 5.5484771728515625e-05,
      "model_forward_time": 2.8606348037719727,
      "step": 58180
    },
    {
      "dynamics_loss_avg": 0.03252729466184974,
      "epoch": 5.5484771728515625e-05,
      "step": 58180
    },
    {
      "action_loss_avg": 0.0010048965457826852,
      "epoch": 5.5484771728515625e-05,
      "step": 58180
    },
    {
      "epoch": 5.5484771728515625e-05,
      "step": 58180,
      "training_step_time": 7.668334484100342
    },
    {
      "epoch": 5.548572540283203e-05,
      "model_forward_time": 1.7379887104034424,
      "step": 58181
    },
    {
      "epoch": 5.548572540283203e-05,
      "step": 58181,
      "training_step_time": 7.727665662765503
    },
    {
      "epoch": 5.548667907714844e-05,
      "model_forward_time": 1.7561218738555908,
      "step": 58182
    },
    {
      "epoch": 5.548667907714844e-05,
      "step": 58182,
      "training_step_time": 8.84227466583252
    },
    {
      "epoch": 5.5487632751464844e-05,
      "model_forward_time": 1.7429401874542236,
      "step": 58183
    },
    {
      "epoch": 5.5487632751464844e-05,
      "step": 58183,
      "training_step_time": 7.528491020202637
    },
    {
      "epoch": 5.548858642578125e-05,
      "model_forward_time": 1.7381458282470703,
      "step": 58184
    },
    {
      "epoch": 5.548858642578125e-05,
      "step": 58184,
      "training_step_time": 7.562305927276611
    },
    {
      "epoch": 5.548954010009766e-05,
      "model_forward_time": 1.749436855316162,
      "step": 58185
    },
    {
      "epoch": 5.548954010009766e-05,
      "step": 58185,
      "training_step_time": 7.057073593139648
    },
    {
      "epoch": 5.549049377441406e-05,
      "model_forward_time": 1.715989351272583,
      "step": 58186
    },
    {
      "epoch": 5.549049377441406e-05,
      "step": 58186,
      "training_step_time": 7.792039632797241
    },
    {
      "epoch": 5.5491447448730466e-05,
      "model_forward_time": 1.722858190536499,
      "step": 58187
    },
    {
      "epoch": 5.5491447448730466e-05,
      "step": 58187,
      "training_step_time": 7.719472885131836
    },
    {
      "epoch": 5.549240112304688e-05,
      "model_forward_time": 1.7197163105010986,
      "step": 58188
    },
    {
      "epoch": 5.549240112304688e-05,
      "step": 58188,
      "training_step_time": 7.662735223770142
    },
    {
      "epoch": 5.549335479736328e-05,
      "model_forward_time": 1.0240309238433838,
      "step": 58189
    },
    {
      "epoch": 5.549335479736328e-05,
      "step": 58189,
      "training_step_time": 11.062578439712524
    },
    {
      "epoch": 5.5494308471679685e-05,
      "grad_norm": 0.12070833146572113,
      "learning_rate": 4.064880089572993e-06,
      "loss": 0.0566,
      "step": 58190
    },
    {
      "epoch": 5.5494308471679685e-05,
      "model_forward_time": 2.9501562118530273,
      "step": 58190
    },
    {
      "dynamics_loss_avg": 0.0630704652518034,
      "epoch": 5.5494308471679685e-05,
      "step": 58190
    },
    {
      "action_loss_avg": 0.0017392194597050548,
      "epoch": 5.5494308471679685e-05,
      "step": 58190
    },
    {
      "epoch": 5.5494308471679685e-05,
      "step": 58190,
      "training_step_time": 10.59372091293335
    },
    {
      "epoch": 5.5495262145996096e-05,
      "model_forward_time": 1.7261502742767334,
      "step": 58191
    },
    {
      "epoch": 5.5495262145996096e-05,
      "step": 58191,
      "training_step_time": 8.829196453094482
    },
    {
      "epoch": 5.54962158203125e-05,
      "model_forward_time": 1.7140889167785645,
      "step": 58192
    },
    {
      "epoch": 5.54962158203125e-05,
      "step": 58192,
      "training_step_time": 7.5244622230529785
    },
    {
      "epoch": 5.5497169494628904e-05,
      "model_forward_time": 1.720750331878662,
      "step": 58193
    },
    {
      "epoch": 5.5497169494628904e-05,
      "step": 58193,
      "training_step_time": 7.678659200668335
    },
    {
      "epoch": 5.5498123168945315e-05,
      "model_forward_time": 1.7190678119659424,
      "step": 58194
    },
    {
      "epoch": 5.5498123168945315e-05,
      "step": 58194,
      "training_step_time": 7.10667872428894
    },
    {
      "epoch": 5.549907684326172e-05,
      "model_forward_time": 1.730680227279663,
      "step": 58195
    },
    {
      "epoch": 5.549907684326172e-05,
      "step": 58195,
      "training_step_time": 7.134813547134399
    },
    {
      "epoch": 5.550003051757812e-05,
      "model_forward_time": 1.7353966236114502,
      "step": 58196
    },
    {
      "epoch": 5.550003051757812e-05,
      "step": 58196,
      "training_step_time": 8.432701349258423
    },
    {
      "epoch": 5.5500984191894534e-05,
      "model_forward_time": 1.719658613204956,
      "step": 58197
    },
    {
      "epoch": 5.5500984191894534e-05,
      "step": 58197,
      "training_step_time": 7.749539375305176
    },
    {
      "epoch": 5.550193786621094e-05,
      "model_forward_time": 1.7179620265960693,
      "step": 58198
    },
    {
      "epoch": 5.550193786621094e-05,
      "step": 58198,
      "training_step_time": 7.795765161514282
    },
    {
      "epoch": 5.550289154052734e-05,
      "model_forward_time": 2.870907783508301,
      "step": 58199
    },
    {
      "epoch": 5.550289154052734e-05,
      "step": 58199,
      "training_step_time": 8.287478923797607
    },
    {
      "epoch": 5.550384521484375e-05,
      "grad_norm": 0.10760965198278427,
      "learning_rate": 4.063255845815779e-06,
      "loss": 0.0587,
      "step": 58200
    },
    {
      "epoch": 5.550384521484375e-05,
      "model_forward_time": 1.7208681106567383,
      "step": 58200
    },
    {
      "dynamics_loss_avg": 0.05619935635477304,
      "epoch": 5.550384521484375e-05,
      "step": 58200
    },
    {
      "action_loss_avg": 0.0018182150641223415,
      "epoch": 5.550384521484375e-05,
      "step": 58200
    },
    {
      "epoch": 5.550384521484375e-05,
      "step": 58200,
      "training_step_time": 7.730420827865601
    },
    {
      "epoch": 5.550479888916016e-05,
      "model_forward_time": 1.7426981925964355,
      "step": 58201
    },
    {
      "epoch": 5.550479888916016e-05,
      "step": 58201,
      "training_step_time": 7.603288650512695
    },
    {
      "epoch": 5.550575256347656e-05,
      "model_forward_time": 1.7225418090820312,
      "step": 58202
    },
    {
      "epoch": 5.550575256347656e-05,
      "step": 58202,
      "training_step_time": 7.761595964431763
    },
    {
      "epoch": 5.550670623779297e-05,
      "model_forward_time": 1.7206110954284668,
      "step": 58203
    },
    {
      "epoch": 5.550670623779297e-05,
      "step": 58203,
      "training_step_time": 7.612342834472656
    },
    {
      "epoch": 5.5507659912109376e-05,
      "model_forward_time": 1.7245042324066162,
      "step": 58204
    },
    {
      "epoch": 5.5507659912109376e-05,
      "step": 58204,
      "training_step_time": 6.752464056015015
    },
    {
      "epoch": 5.550861358642578e-05,
      "model_forward_time": 1.7442619800567627,
      "step": 58205
    },
    {
      "epoch": 5.550861358642578e-05,
      "step": 58205,
      "training_step_time": 7.5272157192230225
    },
    {
      "epoch": 5.550956726074219e-05,
      "model_forward_time": 1.7204174995422363,
      "step": 58206
    },
    {
      "epoch": 5.550956726074219e-05,
      "step": 58206,
      "training_step_time": 8.999834537506104
    },
    {
      "epoch": 5.5510520935058594e-05,
      "model_forward_time": 1.7208783626556396,
      "step": 58207
    },
    {
      "epoch": 5.5510520935058594e-05,
      "step": 58207,
      "training_step_time": 8.587368726730347
    },
    {
      "epoch": 5.5511474609375e-05,
      "model_forward_time": 2.8322596549987793,
      "step": 58208
    },
    {
      "epoch": 5.5511474609375e-05,
      "step": 58208,
      "training_step_time": 9.15369963645935
    },
    {
      "epoch": 5.551242828369141e-05,
      "model_forward_time": 1.7360663414001465,
      "step": 58209
    },
    {
      "epoch": 5.551242828369141e-05,
      "step": 58209,
      "training_step_time": 8.274381399154663
    },
    {
      "epoch": 5.551338195800781e-05,
      "grad_norm": 0.1217438280582428,
      "learning_rate": 4.0616317044994945e-06,
      "loss": 0.0578,
      "step": 58210
    },
    {
      "epoch": 5.551338195800781e-05,
      "model_forward_time": 1.7187228202819824,
      "step": 58210
    },
    {
      "dynamics_loss_avg": 0.053221136610955,
      "epoch": 5.551338195800781e-05,
      "step": 58210
    },
    {
      "action_loss_avg": 0.0014207805128535257,
      "epoch": 5.551338195800781e-05,
      "step": 58210
    },
    {
      "epoch": 5.551338195800781e-05,
      "step": 58210,
      "training_step_time": 8.325269937515259
    },
    {
      "epoch": 5.551433563232422e-05,
      "model_forward_time": 1.7489500045776367,
      "step": 58211
    },
    {
      "epoch": 5.551433563232422e-05,
      "step": 58211,
      "training_step_time": 7.584444761276245
    },
    {
      "epoch": 5.551528930664063e-05,
      "model_forward_time": 1.740248441696167,
      "step": 58212
    },
    {
      "epoch": 5.551528930664063e-05,
      "step": 58212,
      "training_step_time": 7.551034450531006
    },
    {
      "epoch": 5.551624298095703e-05,
      "model_forward_time": 1.7301197052001953,
      "step": 58213
    },
    {
      "epoch": 5.551624298095703e-05,
      "step": 58213,
      "training_step_time": 7.031736135482788
    },
    {
      "epoch": 5.5517196655273436e-05,
      "model_forward_time": 1.7953970432281494,
      "step": 58214
    },
    {
      "epoch": 5.5517196655273436e-05,
      "step": 58214,
      "training_step_time": 7.621397018432617
    },
    {
      "epoch": 5.551815032958985e-05,
      "model_forward_time": 1.7457764148712158,
      "step": 58215
    },
    {
      "epoch": 5.551815032958985e-05,
      "step": 58215,
      "training_step_time": 7.73279881477356
    },
    {
      "epoch": 5.551910400390625e-05,
      "model_forward_time": 1.7456562519073486,
      "step": 58216
    },
    {
      "epoch": 5.551910400390625e-05,
      "step": 58216,
      "training_step_time": 7.735153913497925
    },
    {
      "epoch": 5.5520057678222655e-05,
      "model_forward_time": 2.8854122161865234,
      "step": 58217
    },
    {
      "epoch": 5.5520057678222655e-05,
      "step": 58217,
      "training_step_time": 9.421961307525635
    },
    {
      "epoch": 5.5521011352539066e-05,
      "model_forward_time": 1.7363002300262451,
      "step": 58218
    },
    {
      "epoch": 5.5521011352539066e-05,
      "step": 58218,
      "training_step_time": 9.644344329833984
    },
    {
      "epoch": 5.552196502685547e-05,
      "model_forward_time": 1.7456638813018799,
      "step": 58219
    },
    {
      "epoch": 5.552196502685547e-05,
      "step": 58219,
      "training_step_time": 8.308509349822998
    },
    {
      "epoch": 5.5522918701171874e-05,
      "grad_norm": 0.08065126091241837,
      "learning_rate": 4.0600076658017585e-06,
      "loss": 0.0564,
      "step": 58220
    },
    {
      "epoch": 5.5522918701171874e-05,
      "model_forward_time": 1.7203750610351562,
      "step": 58220
    },
    {
      "dynamics_loss_avg": 0.062180429324507715,
      "epoch": 5.5522918701171874e-05,
      "step": 58220
    },
    {
      "action_loss_avg": 0.0016052402439527213,
      "epoch": 5.5522918701171874e-05,
      "step": 58220
    },
    {
      "epoch": 5.5522918701171874e-05,
      "step": 58220,
      "training_step_time": 7.591080665588379
    },
    {
      "epoch": 5.552387237548828e-05,
      "model_forward_time": 1.7539727687835693,
      "step": 58221
    },
    {
      "epoch": 5.552387237548828e-05,
      "step": 58221,
      "training_step_time": 7.655098915100098
    },
    {
      "epoch": 5.552482604980469e-05,
      "model_forward_time": 1.7553715705871582,
      "step": 58222
    },
    {
      "epoch": 5.552482604980469e-05,
      "step": 58222,
      "training_step_time": 6.72965931892395
    },
    {
      "epoch": 5.552577972412109e-05,
      "model_forward_time": 1.7783663272857666,
      "step": 58223
    },
    {
      "epoch": 5.552577972412109e-05,
      "step": 58223,
      "training_step_time": 7.621635913848877
    },
    {
      "epoch": 5.55267333984375e-05,
      "model_forward_time": 1.7449641227722168,
      "step": 58224
    },
    {
      "epoch": 5.55267333984375e-05,
      "step": 58224,
      "training_step_time": 7.732547044754028
    },
    {
      "epoch": 5.552768707275391e-05,
      "model_forward_time": 1.7563562393188477,
      "step": 58225
    },
    {
      "epoch": 5.552768707275391e-05,
      "step": 58225,
      "training_step_time": 7.674093008041382
    },
    {
      "epoch": 5.552864074707031e-05,
      "model_forward_time": 1.7354326248168945,
      "step": 58226
    },
    {
      "epoch": 5.552864074707031e-05,
      "step": 58226,
      "training_step_time": 8.555860996246338
    },
    {
      "epoch": 5.5529594421386716e-05,
      "model_forward_time": 2.865119457244873,
      "step": 58227
    },
    {
      "epoch": 5.5529594421386716e-05,
      "step": 58227,
      "training_step_time": 8.278824090957642
    },
    {
      "epoch": 5.5530548095703126e-05,
      "model_forward_time": 1.745671272277832,
      "step": 58228
    },
    {
      "epoch": 5.5530548095703126e-05,
      "step": 58228,
      "training_step_time": 7.783987283706665
    },
    {
      "epoch": 5.553150177001953e-05,
      "model_forward_time": 1.731097936630249,
      "step": 58229
    },
    {
      "epoch": 5.553150177001953e-05,
      "step": 58229,
      "training_step_time": 7.536296844482422
    },
    {
      "epoch": 5.5532455444335935e-05,
      "grad_norm": 0.13874155282974243,
      "learning_rate": 4.05838372990017e-06,
      "loss": 0.056,
      "step": 58230
    },
    {
      "epoch": 5.5532455444335935e-05,
      "model_forward_time": 1.7229697704315186,
      "step": 58230
    },
    {
      "dynamics_loss_avg": 0.03972723316401243,
      "epoch": 5.5532455444335935e-05,
      "step": 58230
    },
    {
      "action_loss_avg": 0.00170047108840663,
      "epoch": 5.5532455444335935e-05,
      "step": 58230
    },
    {
      "epoch": 5.5532455444335935e-05,
      "step": 58230,
      "training_step_time": 7.522595167160034
    },
    {
      "epoch": 5.5533409118652345e-05,
      "model_forward_time": 1.746959924697876,
      "step": 58231
    },
    {
      "epoch": 5.5533409118652345e-05,
      "step": 58231,
      "training_step_time": 7.063151836395264
    },
    {
      "epoch": 5.553436279296875e-05,
      "model_forward_time": 1.7420053482055664,
      "step": 58232
    },
    {
      "epoch": 5.553436279296875e-05,
      "step": 58232,
      "training_step_time": 7.669989109039307
    },
    {
      "epoch": 5.553531646728515e-05,
      "model_forward_time": 1.7323734760284424,
      "step": 58233
    },
    {
      "epoch": 5.553531646728515e-05,
      "step": 58233,
      "training_step_time": 9.23178219795227
    },
    {
      "epoch": 5.5536270141601564e-05,
      "model_forward_time": 1.7390177249908447,
      "step": 58234
    },
    {
      "epoch": 5.5536270141601564e-05,
      "step": 58234,
      "training_step_time": 7.687225580215454
    },
    {
      "epoch": 5.553722381591797e-05,
      "model_forward_time": 1.7467665672302246,
      "step": 58235
    },
    {
      "epoch": 5.553722381591797e-05,
      "step": 58235,
      "training_step_time": 7.750725746154785
    },
    {
      "epoch": 5.553817749023437e-05,
      "model_forward_time": 2.8699018955230713,
      "step": 58236
    },
    {
      "epoch": 5.553817749023437e-05,
      "step": 58236,
      "training_step_time": 7.790905714035034
    },
    {
      "epoch": 5.553913116455078e-05,
      "model_forward_time": 1.735379934310913,
      "step": 58237
    },
    {
      "epoch": 5.553913116455078e-05,
      "step": 58237,
      "training_step_time": 8.881211757659912
    },
    {
      "epoch": 5.554008483886719e-05,
      "model_forward_time": 1.7339367866516113,
      "step": 58238
    },
    {
      "epoch": 5.554008483886719e-05,
      "step": 58238,
      "training_step_time": 7.585164546966553
    },
    {
      "epoch": 5.554103851318359e-05,
      "model_forward_time": 1.7371220588684082,
      "step": 58239
    },
    {
      "epoch": 5.554103851318359e-05,
      "step": 58239,
      "training_step_time": 7.647662878036499
    },
    {
      "epoch": 5.55419921875e-05,
      "grad_norm": 0.11148043721914291,
      "learning_rate": 4.056759896972322e-06,
      "loss": 0.0579,
      "step": 58240
    },
    {
      "epoch": 5.55419921875e-05,
      "model_forward_time": 1.7346701622009277,
      "step": 58240
    },
    {
      "dynamics_loss_avg": 0.0484898442402482,
      "epoch": 5.55419921875e-05,
      "step": 58240
    },
    {
      "action_loss_avg": 0.0015242081484757365,
      "epoch": 5.55419921875e-05,
      "step": 58240
    },
    {
      "epoch": 5.55419921875e-05,
      "step": 58240,
      "training_step_time": 7.10373592376709
    },
    {
      "epoch": 5.5542945861816406e-05,
      "model_forward_time": 1.7477762699127197,
      "step": 58241
    },
    {
      "epoch": 5.5542945861816406e-05,
      "step": 58241,
      "training_step_time": 8.177947759628296
    },
    {
      "epoch": 5.554389953613281e-05,
      "model_forward_time": 1.8041930198669434,
      "step": 58242
    },
    {
      "epoch": 5.554389953613281e-05,
      "step": 58242,
      "training_step_time": 7.7841527462005615
    },
    {
      "epoch": 5.554485321044922e-05,
      "model_forward_time": 1.734642744064331,
      "step": 58243
    },
    {
      "epoch": 5.554485321044922e-05,
      "step": 58243,
      "training_step_time": 8.754534006118774
    },
    {
      "epoch": 5.5545806884765625e-05,
      "model_forward_time": 1.7349915504455566,
      "step": 58244
    },
    {
      "epoch": 5.5545806884765625e-05,
      "step": 58244,
      "training_step_time": 7.75452995300293
    },
    {
      "epoch": 5.554676055908203e-05,
      "model_forward_time": 2.939201593399048,
      "step": 58245
    },
    {
      "epoch": 5.554676055908203e-05,
      "step": 58245,
      "training_step_time": 9.344454050064087
    },
    {
      "epoch": 5.554771423339844e-05,
      "model_forward_time": 1.7411162853240967,
      "step": 58246
    },
    {
      "epoch": 5.554771423339844e-05,
      "step": 58246,
      "training_step_time": 8.326889991760254
    },
    {
      "epoch": 5.5548667907714844e-05,
      "model_forward_time": 1.746415138244629,
      "step": 58247
    },
    {
      "epoch": 5.5548667907714844e-05,
      "step": 58247,
      "training_step_time": 8.200679063796997
    },
    {
      "epoch": 5.554962158203125e-05,
      "model_forward_time": 1.7385504245758057,
      "step": 58248
    },
    {
      "epoch": 5.554962158203125e-05,
      "step": 58248,
      "training_step_time": 7.636763095855713
    },
    {
      "epoch": 5.555057525634766e-05,
      "model_forward_time": 1.73185133934021,
      "step": 58249
    },
    {
      "epoch": 5.555057525634766e-05,
      "step": 58249,
      "training_step_time": 8.579994916915894
    },
    {
      "epoch": 5.555152893066406e-05,
      "grad_norm": 0.13202513754367828,
      "learning_rate": 4.055136167195793e-06,
      "loss": 0.0588,
      "step": 58250
    },
    {
      "epoch": 5.555152893066406e-05,
      "model_forward_time": 1.759070634841919,
      "step": 58250
    },
    {
      "dynamics_loss_avg": 0.05568266659975052,
      "epoch": 5.555152893066406e-05,
      "step": 58250
    },
    {
      "action_loss_avg": 0.0011741286114556714,
      "epoch": 5.555152893066406e-05,
      "step": 58250
    },
    {
      "epoch": 5.555152893066406e-05,
      "step": 58250,
      "training_step_time": 7.561392068862915
    },
    {
      "epoch": 5.5552482604980467e-05,
      "model_forward_time": 1.7994630336761475,
      "step": 58251
    },
    {
      "epoch": 5.5552482604980467e-05,
      "step": 58251,
      "training_step_time": 7.730332136154175
    },
    {
      "epoch": 5.555343627929688e-05,
      "model_forward_time": 1.7533786296844482,
      "step": 58252
    },
    {
      "epoch": 5.555343627929688e-05,
      "step": 58252,
      "training_step_time": 7.647558689117432
    },
    {
      "epoch": 5.555438995361328e-05,
      "model_forward_time": 1.7408547401428223,
      "step": 58253
    },
    {
      "epoch": 5.555438995361328e-05,
      "step": 58253,
      "training_step_time": 7.657144546508789
    },
    {
      "epoch": 5.5555343627929685e-05,
      "model_forward_time": 2.88325834274292,
      "step": 58254
    },
    {
      "epoch": 5.5555343627929685e-05,
      "step": 58254,
      "training_step_time": 8.771701097488403
    },
    {
      "epoch": 5.5556297302246096e-05,
      "model_forward_time": 1.3948922157287598,
      "step": 58255
    },
    {
      "epoch": 5.5556297302246096e-05,
      "step": 58255,
      "training_step_time": 9.6476571559906
    },
    {
      "epoch": 5.55572509765625e-05,
      "model_forward_time": 1.753220796585083,
      "step": 58256
    },
    {
      "epoch": 5.55572509765625e-05,
      "step": 58256,
      "training_step_time": 9.444565534591675
    },
    {
      "epoch": 5.5558204650878904e-05,
      "model_forward_time": 1.757490873336792,
      "step": 58257
    },
    {
      "epoch": 5.5558204650878904e-05,
      "step": 58257,
      "training_step_time": 7.593622922897339
    },
    {
      "epoch": 5.5559158325195315e-05,
      "model_forward_time": 1.755383014678955,
      "step": 58258
    },
    {
      "epoch": 5.5559158325195315e-05,
      "step": 58258,
      "training_step_time": 7.615680456161499
    },
    {
      "epoch": 5.556011199951172e-05,
      "model_forward_time": 1.7483632564544678,
      "step": 58259
    },
    {
      "epoch": 5.556011199951172e-05,
      "step": 58259,
      "training_step_time": 7.584558010101318
    },
    {
      "epoch": 5.556106567382812e-05,
      "grad_norm": 0.09384462237358093,
      "learning_rate": 4.0535125407481524e-06,
      "loss": 0.059,
      "step": 58260
    },
    {
      "epoch": 5.556106567382812e-05,
      "model_forward_time": 1.7594764232635498,
      "step": 58260
    },
    {
      "dynamics_loss_avg": 0.04554842803627253,
      "epoch": 5.556106567382812e-05,
      "step": 58260
    },
    {
      "action_loss_avg": 0.0013180190027924255,
      "epoch": 5.556106567382812e-05,
      "step": 58260
    },
    {
      "epoch": 5.556106567382812e-05,
      "step": 58260,
      "training_step_time": 7.762992858886719
    },
    {
      "epoch": 5.5562019348144534e-05,
      "model_forward_time": 1.7379658222198486,
      "step": 58261
    },
    {
      "epoch": 5.5562019348144534e-05,
      "step": 58261,
      "training_step_time": 8.320050477981567
    },
    {
      "epoch": 5.556297302246094e-05,
      "model_forward_time": 1.8351941108703613,
      "step": 58262
    },
    {
      "epoch": 5.556297302246094e-05,
      "step": 58262,
      "training_step_time": 7.728997707366943
    },
    {
      "epoch": 5.556392669677734e-05,
      "model_forward_time": 2.914947748184204,
      "step": 58263
    },
    {
      "epoch": 5.556392669677734e-05,
      "step": 58263,
      "training_step_time": 9.479273080825806
    },
    {
      "epoch": 5.556488037109375e-05,
      "model_forward_time": 1.7489440441131592,
      "step": 58264
    },
    {
      "epoch": 5.556488037109375e-05,
      "step": 58264,
      "training_step_time": 7.686676502227783
    },
    {
      "epoch": 5.556583404541016e-05,
      "model_forward_time": 1.754166841506958,
      "step": 58265
    },
    {
      "epoch": 5.556583404541016e-05,
      "step": 58265,
      "training_step_time": 7.618746519088745
    },
    {
      "epoch": 5.556678771972656e-05,
      "model_forward_time": 1.7528741359710693,
      "step": 58266
    },
    {
      "epoch": 5.556678771972656e-05,
      "step": 58266,
      "training_step_time": 7.6200644969940186
    },
    {
      "epoch": 5.556774139404297e-05,
      "model_forward_time": 1.7404606342315674,
      "step": 58267
    },
    {
      "epoch": 5.556774139404297e-05,
      "step": 58267,
      "training_step_time": 8.32775092124939
    },
    {
      "epoch": 5.5568695068359376e-05,
      "model_forward_time": 1.737919569015503,
      "step": 58268
    },
    {
      "epoch": 5.5568695068359376e-05,
      "step": 58268,
      "training_step_time": 7.234583139419556
    },
    {
      "epoch": 5.556964874267578e-05,
      "model_forward_time": 1.743180751800537,
      "step": 58269
    },
    {
      "epoch": 5.556964874267578e-05,
      "step": 58269,
      "training_step_time": 7.762326717376709
    },
    {
      "epoch": 5.557060241699219e-05,
      "grad_norm": 0.09162399172782898,
      "learning_rate": 4.051889017806955e-06,
      "loss": 0.056,
      "step": 58270
    },
    {
      "epoch": 5.557060241699219e-05,
      "model_forward_time": 1.7912638187408447,
      "step": 58270
    },
    {
      "dynamics_loss_avg": 0.05047491900622845,
      "epoch": 5.557060241699219e-05,
      "step": 58270
    },
    {
      "action_loss_avg": 0.001534891442861408,
      "epoch": 5.557060241699219e-05,
      "step": 58270
    },
    {
      "epoch": 5.557060241699219e-05,
      "step": 58270,
      "training_step_time": 8.285292148590088
    },
    {
      "epoch": 5.5571556091308595e-05,
      "model_forward_time": 1.7379872798919678,
      "step": 58271
    },
    {
      "epoch": 5.5571556091308595e-05,
      "step": 58271,
      "training_step_time": 8.187074184417725
    },
    {
      "epoch": 5.5572509765625e-05,
      "model_forward_time": 1.7403106689453125,
      "step": 58272
    },
    {
      "epoch": 5.5572509765625e-05,
      "step": 58272,
      "training_step_time": 8.503798484802246
    },
    {
      "epoch": 5.557346343994141e-05,
      "model_forward_time": 2.88948917388916,
      "step": 58273
    },
    {
      "epoch": 5.557346343994141e-05,
      "step": 58273,
      "training_step_time": 8.76084852218628
    },
    {
      "epoch": 5.557441711425781e-05,
      "model_forward_time": 1.7396628856658936,
      "step": 58274
    },
    {
      "epoch": 5.557441711425781e-05,
      "step": 58274,
      "training_step_time": 8.73349642753601
    },
    {
      "epoch": 5.557537078857422e-05,
      "model_forward_time": 1.7295429706573486,
      "step": 58275
    },
    {
      "epoch": 5.557537078857422e-05,
      "step": 58275,
      "training_step_time": 7.508091449737549
    },
    {
      "epoch": 5.557632446289063e-05,
      "model_forward_time": 1.738675832748413,
      "step": 58276
    },
    {
      "epoch": 5.557632446289063e-05,
      "step": 58276,
      "training_step_time": 7.53009033203125
    },
    {
      "epoch": 5.557727813720703e-05,
      "model_forward_time": 1.7521047592163086,
      "step": 58277
    },
    {
      "epoch": 5.557727813720703e-05,
      "step": 58277,
      "training_step_time": 7.5630433559417725
    },
    {
      "epoch": 5.5578231811523436e-05,
      "model_forward_time": 1.7600553035736084,
      "step": 58278
    },
    {
      "epoch": 5.5578231811523436e-05,
      "step": 58278,
      "training_step_time": 7.713070869445801
    },
    {
      "epoch": 5.557918548583985e-05,
      "model_forward_time": 1.817800521850586,
      "step": 58279
    },
    {
      "epoch": 5.557918548583985e-05,
      "step": 58279,
      "training_step_time": 8.069224119186401
    },
    {
      "epoch": 5.558013916015625e-05,
      "grad_norm": 0.1102442517876625,
      "learning_rate": 4.050265598549752e-06,
      "loss": 0.057,
      "step": 58280
    },
    {
      "epoch": 5.558013916015625e-05,
      "model_forward_time": 1.7623488903045654,
      "step": 58280
    },
    {
      "dynamics_loss_avg": 0.05274657718837261,
      "epoch": 5.558013916015625e-05,
      "step": 58280
    },
    {
      "action_loss_avg": 0.0013707340287510306,
      "epoch": 5.558013916015625e-05,
      "step": 58280
    },
    {
      "epoch": 5.558013916015625e-05,
      "step": 58280,
      "training_step_time": 7.585747957229614
    },
    {
      "epoch": 5.5581092834472655e-05,
      "model_forward_time": 1.7604618072509766,
      "step": 58281
    },
    {
      "epoch": 5.5581092834472655e-05,
      "step": 58281,
      "training_step_time": 8.388116359710693
    },
    {
      "epoch": 5.558204650878906e-05,
      "model_forward_time": 2.9466280937194824,
      "step": 58282
    },
    {
      "epoch": 5.558204650878906e-05,
      "step": 58282,
      "training_step_time": 7.739989995956421
    },
    {
      "epoch": 5.558300018310547e-05,
      "model_forward_time": 1.7597622871398926,
      "step": 58283
    },
    {
      "epoch": 5.558300018310547e-05,
      "step": 58283,
      "training_step_time": 7.662170886993408
    },
    {
      "epoch": 5.5583953857421874e-05,
      "model_forward_time": 1.7571799755096436,
      "step": 58284
    },
    {
      "epoch": 5.5583953857421874e-05,
      "step": 58284,
      "training_step_time": 8.401063442230225
    },
    {
      "epoch": 5.558490753173828e-05,
      "model_forward_time": 1.7626783847808838,
      "step": 58285
    },
    {
      "epoch": 5.558490753173828e-05,
      "step": 58285,
      "training_step_time": 7.587588548660278
    },
    {
      "epoch": 5.558586120605469e-05,
      "model_forward_time": 1.7599225044250488,
      "step": 58286
    },
    {
      "epoch": 5.558586120605469e-05,
      "step": 58286,
      "training_step_time": 7.574699878692627
    },
    {
      "epoch": 5.558681488037109e-05,
      "model_forward_time": 1.759941816329956,
      "step": 58287
    },
    {
      "epoch": 5.558681488037109e-05,
      "step": 58287,
      "training_step_time": 7.745233774185181
    },
    {
      "epoch": 5.55877685546875e-05,
      "model_forward_time": 1.7851893901824951,
      "step": 58288
    },
    {
      "epoch": 5.55877685546875e-05,
      "step": 58288,
      "training_step_time": 8.147771835327148
    },
    {
      "epoch": 5.558872222900391e-05,
      "model_forward_time": 1.7585108280181885,
      "step": 58289
    },
    {
      "epoch": 5.558872222900391e-05,
      "step": 58289,
      "training_step_time": 7.645199775695801
    },
    {
      "epoch": 5.558967590332031e-05,
      "grad_norm": 0.09635154902935028,
      "learning_rate": 4.048642283154074e-06,
      "loss": 0.054,
      "step": 58290
    },
    {
      "epoch": 5.558967590332031e-05,
      "model_forward_time": 1.7667341232299805,
      "step": 58290
    },
    {
      "dynamics_loss_avg": 0.059302822686731815,
      "epoch": 5.558967590332031e-05,
      "step": 58290
    },
    {
      "action_loss_avg": 0.001286686956882477,
      "epoch": 5.558967590332031e-05,
      "step": 58290
    },
    {
      "epoch": 5.558967590332031e-05,
      "step": 58290,
      "training_step_time": 7.635149240493774
    },
    {
      "epoch": 5.5590629577636716e-05,
      "model_forward_time": 3.014888286590576,
      "step": 58291
    },
    {
      "epoch": 5.5590629577636716e-05,
      "step": 58291,
      "training_step_time": 10.232272148132324
    },
    {
      "epoch": 5.5591583251953127e-05,
      "model_forward_time": 1.933743953704834,
      "step": 58292
    },
    {
      "epoch": 5.5591583251953127e-05,
      "step": 58292,
      "training_step_time": 8.387672662734985
    },
    {
      "epoch": 5.559253692626953e-05,
      "model_forward_time": 1.9265661239624023,
      "step": 58293
    },
    {
      "epoch": 5.559253692626953e-05,
      "step": 58293,
      "training_step_time": 7.606443405151367
    },
    {
      "epoch": 5.5593490600585935e-05,
      "model_forward_time": 1.749678134918213,
      "step": 58294
    },
    {
      "epoch": 5.5593490600585935e-05,
      "step": 58294,
      "training_step_time": 7.639551877975464
    },
    {
      "epoch": 5.5594444274902345e-05,
      "model_forward_time": 1.7514986991882324,
      "step": 58295
    },
    {
      "epoch": 5.5594444274902345e-05,
      "step": 58295,
      "training_step_time": 7.571558237075806
    },
    {
      "epoch": 5.559539794921875e-05,
      "model_forward_time": 1.74949312210083,
      "step": 58296
    },
    {
      "epoch": 5.559539794921875e-05,
      "step": 58296,
      "training_step_time": 7.7125749588012695
    },
    {
      "epoch": 5.5596351623535153e-05,
      "model_forward_time": 1.746253490447998,
      "step": 58297
    },
    {
      "epoch": 5.5596351623535153e-05,
      "step": 58297,
      "training_step_time": 8.508634567260742
    },
    {
      "epoch": 5.5597305297851564e-05,
      "model_forward_time": 1.753371000289917,
      "step": 58298
    },
    {
      "epoch": 5.5597305297851564e-05,
      "step": 58298,
      "training_step_time": 8.655560970306396
    },
    {
      "epoch": 5.559825897216797e-05,
      "model_forward_time": 1.7476961612701416,
      "step": 58299
    },
    {
      "epoch": 5.559825897216797e-05,
      "step": 58299,
      "training_step_time": 8.516207933425903
    },
    {
      "epoch": 5.559921264648437e-05,
      "grad_norm": 0.11126532405614853,
      "learning_rate": 4.047019071797447e-06,
      "loss": 0.0561,
      "step": 58300
    },
    {
      "epoch": 5.559921264648437e-05,
      "model_forward_time": 2.8760807514190674,
      "step": 58300
    },
    {
      "dynamics_loss_avg": 0.05047548804432154,
      "epoch": 5.559921264648437e-05,
      "step": 58300
    },
    {
      "action_loss_avg": 0.0017530565208289772,
      "epoch": 5.559921264648437e-05,
      "step": 58300
    },
    {
      "epoch": 5.559921264648437e-05,
      "step": 58300,
      "training_step_time": 8.737284898757935
    },
    {
      "epoch": 5.560016632080078e-05,
      "model_forward_time": 1.3904821872711182,
      "step": 58301
    },
    {
      "epoch": 5.560016632080078e-05,
      "step": 58301,
      "training_step_time": 8.534250974655151
    },
    {
      "epoch": 5.560111999511719e-05,
      "model_forward_time": 1.8934555053710938,
      "step": 58302
    },
    {
      "epoch": 5.560111999511719e-05,
      "step": 58302,
      "training_step_time": 8.513781785964966
    },
    {
      "epoch": 5.560207366943359e-05,
      "model_forward_time": 1.761824369430542,
      "step": 58303
    },
    {
      "epoch": 5.560207366943359e-05,
      "step": 58303,
      "training_step_time": 7.5476624965667725
    },
    {
      "epoch": 5.560302734375e-05,
      "model_forward_time": 1.7579569816589355,
      "step": 58304
    },
    {
      "epoch": 5.560302734375e-05,
      "step": 58304,
      "training_step_time": 8.504072427749634
    },
    {
      "epoch": 5.5603981018066406e-05,
      "model_forward_time": 1.753671407699585,
      "step": 58305
    },
    {
      "epoch": 5.5603981018066406e-05,
      "step": 58305,
      "training_step_time": 8.209671020507812
    },
    {
      "epoch": 5.560493469238281e-05,
      "model_forward_time": 1.756545066833496,
      "step": 58306
    },
    {
      "epoch": 5.560493469238281e-05,
      "step": 58306,
      "training_step_time": 8.324263334274292
    },
    {
      "epoch": 5.560588836669922e-05,
      "model_forward_time": 1.7504348754882812,
      "step": 58307
    },
    {
      "epoch": 5.560588836669922e-05,
      "step": 58307,
      "training_step_time": 7.756455183029175
    },
    {
      "epoch": 5.5606842041015625e-05,
      "model_forward_time": 1.7565815448760986,
      "step": 58308
    },
    {
      "epoch": 5.5606842041015625e-05,
      "step": 58308,
      "training_step_time": 9.529701471328735
    },
    {
      "epoch": 5.560779571533203e-05,
      "model_forward_time": 1.7497169971466064,
      "step": 58309
    },
    {
      "epoch": 5.560779571533203e-05,
      "step": 58309,
      "training_step_time": 9.361894845962524
    },
    {
      "epoch": 5.560874938964844e-05,
      "grad_norm": 0.08775785565376282,
      "learning_rate": 4.045395964657381e-06,
      "loss": 0.0564,
      "step": 58310
    },
    {
      "epoch": 5.560874938964844e-05,
      "model_forward_time": 1.7205500602722168,
      "step": 58310
    },
    {
      "dynamics_loss_avg": 0.07282075062394142,
      "epoch": 5.560874938964844e-05,
      "step": 58310
    },
    {
      "action_loss_avg": 0.0015966202219715342,
      "epoch": 5.560874938964844e-05,
      "step": 58310
    },
    {
      "epoch": 5.560874938964844e-05,
      "step": 58310,
      "training_step_time": 8.466745376586914
    },
    {
      "epoch": 5.5609703063964844e-05,
      "model_forward_time": 1.867938756942749,
      "step": 58311
    },
    {
      "epoch": 5.5609703063964844e-05,
      "step": 58311,
      "training_step_time": 8.583620309829712
    },
    {
      "epoch": 5.561065673828125e-05,
      "model_forward_time": 1.753342866897583,
      "step": 58312
    },
    {
      "epoch": 5.561065673828125e-05,
      "step": 58312,
      "training_step_time": 7.676274061203003
    },
    {
      "epoch": 5.561161041259766e-05,
      "model_forward_time": 1.7434277534484863,
      "step": 58313
    },
    {
      "epoch": 5.561161041259766e-05,
      "step": 58313,
      "training_step_time": 8.313860416412354
    },
    {
      "epoch": 5.561256408691406e-05,
      "model_forward_time": 1.7519574165344238,
      "step": 58314
    },
    {
      "epoch": 5.561256408691406e-05,
      "step": 58314,
      "training_step_time": 7.719241380691528
    },
    {
      "epoch": 5.5613517761230467e-05,
      "model_forward_time": 1.746086597442627,
      "step": 58315
    },
    {
      "epoch": 5.5613517761230467e-05,
      "step": 58315,
      "training_step_time": 7.663646936416626
    },
    {
      "epoch": 5.561447143554688e-05,
      "model_forward_time": 1.743457317352295,
      "step": 58316
    },
    {
      "epoch": 5.561447143554688e-05,
      "step": 58316,
      "training_step_time": 7.769536733627319
    },
    {
      "epoch": 5.561542510986328e-05,
      "model_forward_time": 1.7418267726898193,
      "step": 58317
    },
    {
      "epoch": 5.561542510986328e-05,
      "step": 58317,
      "training_step_time": 7.639831304550171
    },
    {
      "epoch": 5.5616378784179685e-05,
      "model_forward_time": 1.7412374019622803,
      "step": 58318
    },
    {
      "epoch": 5.5616378784179685e-05,
      "step": 58318,
      "training_step_time": 9.377969741821289
    },
    {
      "epoch": 5.5617332458496096e-05,
      "model_forward_time": 2.8912432193756104,
      "step": 58319
    },
    {
      "epoch": 5.5617332458496096e-05,
      "step": 58319,
      "training_step_time": 7.795278549194336
    },
    {
      "epoch": 5.56182861328125e-05,
      "grad_norm": 0.10363036394119263,
      "learning_rate": 4.043772961911377e-06,
      "loss": 0.0558,
      "step": 58320
    },
    {
      "epoch": 5.56182861328125e-05,
      "model_forward_time": 1.8557379245758057,
      "step": 58320
    },
    {
      "dynamics_loss_avg": 0.047688052989542484,
      "epoch": 5.56182861328125e-05,
      "step": 58320
    },
    {
      "action_loss_avg": 0.0014377143437741325,
      "epoch": 5.56182861328125e-05,
      "step": 58320
    },
    {
      "epoch": 5.56182861328125e-05,
      "step": 58320,
      "training_step_time": 7.664422035217285
    },
    {
      "epoch": 5.5619239807128904e-05,
      "model_forward_time": 1.738572359085083,
      "step": 58321
    },
    {
      "epoch": 5.5619239807128904e-05,
      "step": 58321,
      "training_step_time": 8.684842109680176
    },
    {
      "epoch": 5.5620193481445315e-05,
      "model_forward_time": 1.7534582614898682,
      "step": 58322
    },
    {
      "epoch": 5.5620193481445315e-05,
      "step": 58322,
      "training_step_time": 7.293360948562622
    },
    {
      "epoch": 5.562114715576172e-05,
      "model_forward_time": 1.7401845455169678,
      "step": 58323
    },
    {
      "epoch": 5.562114715576172e-05,
      "step": 58323,
      "training_step_time": 7.693262815475464
    },
    {
      "epoch": 5.562210083007812e-05,
      "model_forward_time": 1.742506504058838,
      "step": 58324
    },
    {
      "epoch": 5.562210083007812e-05,
      "step": 58324,
      "training_step_time": 7.621779441833496
    },
    {
      "epoch": 5.5623054504394534e-05,
      "model_forward_time": 1.7368130683898926,
      "step": 58325
    },
    {
      "epoch": 5.5623054504394534e-05,
      "step": 58325,
      "training_step_time": 7.6890037059783936
    },
    {
      "epoch": 5.562400817871094e-05,
      "model_forward_time": 1.7490456104278564,
      "step": 58326
    },
    {
      "epoch": 5.562400817871094e-05,
      "step": 58326,
      "training_step_time": 8.70084023475647
    },
    {
      "epoch": 5.562496185302734e-05,
      "model_forward_time": 1.7438106536865234,
      "step": 58327
    },
    {
      "epoch": 5.562496185302734e-05,
      "step": 58327,
      "training_step_time": 9.031973123550415
    },
    {
      "epoch": 5.562591552734375e-05,
      "model_forward_time": 2.882539987564087,
      "step": 58328
    },
    {
      "epoch": 5.562591552734375e-05,
      "step": 58328,
      "training_step_time": 8.322019815444946
    },
    {
      "epoch": 5.562686920166016e-05,
      "model_forward_time": 1.868513822555542,
      "step": 58329
    },
    {
      "epoch": 5.562686920166016e-05,
      "step": 58329,
      "training_step_time": 10.175813913345337
    },
    {
      "epoch": 5.562782287597656e-05,
      "grad_norm": 0.10017499327659607,
      "learning_rate": 4.0421500637369236e-06,
      "loss": 0.0535,
      "step": 58330
    },
    {
      "epoch": 5.562782287597656e-05,
      "model_forward_time": 1.7254152297973633,
      "step": 58330
    },
    {
      "dynamics_loss_avg": 0.051588792353868484,
      "epoch": 5.562782287597656e-05,
      "step": 58330
    },
    {
      "action_loss_avg": 0.0014472319453489035,
      "epoch": 5.562782287597656e-05,
      "step": 58330
    },
    {
      "epoch": 5.562782287597656e-05,
      "step": 58330,
      "training_step_time": 8.780295610427856
    },
    {
      "epoch": 5.562877655029297e-05,
      "model_forward_time": 1.7451868057250977,
      "step": 58331
    },
    {
      "epoch": 5.562877655029297e-05,
      "step": 58331,
      "training_step_time": 7.543212413787842
    },
    {
      "epoch": 5.5629730224609376e-05,
      "model_forward_time": 1.7386224269866943,
      "step": 58332
    },
    {
      "epoch": 5.5629730224609376e-05,
      "step": 58332,
      "training_step_time": 7.683192729949951
    },
    {
      "epoch": 5.563068389892578e-05,
      "model_forward_time": 1.7405283451080322,
      "step": 58333
    },
    {
      "epoch": 5.563068389892578e-05,
      "step": 58333,
      "training_step_time": 7.589933156967163
    },
    {
      "epoch": 5.563163757324219e-05,
      "model_forward_time": 1.7392709255218506,
      "step": 58334
    },
    {
      "epoch": 5.563163757324219e-05,
      "step": 58334,
      "training_step_time": 7.712869882583618
    },
    {
      "epoch": 5.5632591247558595e-05,
      "model_forward_time": 1.7368793487548828,
      "step": 58335
    },
    {
      "epoch": 5.5632591247558595e-05,
      "step": 58335,
      "training_step_time": 9.30118465423584
    },
    {
      "epoch": 5.5633544921875e-05,
      "model_forward_time": 1.737185001373291,
      "step": 58336
    },
    {
      "epoch": 5.5633544921875e-05,
      "step": 58336,
      "training_step_time": 7.593632936477661
    },
    {
      "epoch": 5.563449859619141e-05,
      "model_forward_time": 2.8682308197021484,
      "step": 58337
    },
    {
      "epoch": 5.563449859619141e-05,
      "step": 58337,
      "training_step_time": 8.51970362663269
    },
    {
      "epoch": 5.5635452270507813e-05,
      "model_forward_time": 1.8589720726013184,
      "step": 58338
    },
    {
      "epoch": 5.5635452270507813e-05,
      "step": 58338,
      "training_step_time": 7.771197319030762
    },
    {
      "epoch": 5.563640594482422e-05,
      "model_forward_time": 1.743971347808838,
      "step": 58339
    },
    {
      "epoch": 5.563640594482422e-05,
      "step": 58339,
      "training_step_time": 8.648758172988892
    },
    {
      "epoch": 5.563735961914063e-05,
      "grad_norm": 0.11115063726902008,
      "learning_rate": 4.040527270311498e-06,
      "loss": 0.0545,
      "step": 58340
    },
    {
      "epoch": 5.563735961914063e-05,
      "model_forward_time": 1.7296977043151855,
      "step": 58340
    },
    {
      "dynamics_loss_avg": 0.039798713847994806,
      "epoch": 5.563735961914063e-05,
      "step": 58340
    },
    {
      "action_loss_avg": 0.0007714617648161947,
      "epoch": 5.563735961914063e-05,
      "step": 58340
    },
    {
      "epoch": 5.563735961914063e-05,
      "step": 58340,
      "training_step_time": 7.712121248245239
    },
    {
      "epoch": 5.563831329345703e-05,
      "model_forward_time": 1.7369096279144287,
      "step": 58341
    },
    {
      "epoch": 5.563831329345703e-05,
      "step": 58341,
      "training_step_time": 7.693315029144287
    },
    {
      "epoch": 5.5639266967773436e-05,
      "model_forward_time": 1.7454273700714111,
      "step": 58342
    },
    {
      "epoch": 5.5639266967773436e-05,
      "step": 58342,
      "training_step_time": 7.630740165710449
    },
    {
      "epoch": 5.564022064208985e-05,
      "model_forward_time": 1.3734853267669678,
      "step": 58343
    },
    {
      "epoch": 5.564022064208985e-05,
      "step": 58343,
      "training_step_time": 7.710660696029663
    },
    {
      "epoch": 5.564117431640625e-05,
      "model_forward_time": 1.7427754402160645,
      "step": 58344
    },
    {
      "epoch": 5.564117431640625e-05,
      "step": 58344,
      "training_step_time": 8.520867586135864
    },
    {
      "epoch": 5.5642127990722655e-05,
      "model_forward_time": 1.7434968948364258,
      "step": 58345
    },
    {
      "epoch": 5.5642127990722655e-05,
      "step": 58345,
      "training_step_time": 7.810932397842407
    },
    {
      "epoch": 5.564308166503906e-05,
      "model_forward_time": 2.8178675174713135,
      "step": 58346
    },
    {
      "epoch": 5.564308166503906e-05,
      "step": 58346,
      "training_step_time": 8.512802362442017
    },
    {
      "epoch": 5.564403533935547e-05,
      "model_forward_time": 1.8447449207305908,
      "step": 58347
    },
    {
      "epoch": 5.564403533935547e-05,
      "step": 58347,
      "training_step_time": 9.28914475440979
    },
    {
      "epoch": 5.5644989013671874e-05,
      "model_forward_time": 1.748037576675415,
      "step": 58348
    },
    {
      "epoch": 5.5644989013671874e-05,
      "step": 58348,
      "training_step_time": 7.693387508392334
    },
    {
      "epoch": 5.564594268798828e-05,
      "model_forward_time": 1.3869476318359375,
      "step": 58349
    },
    {
      "epoch": 5.564594268798828e-05,
      "step": 58349,
      "training_step_time": 8.258767366409302
    },
    {
      "epoch": 5.564689636230469e-05,
      "grad_norm": 0.13647359609603882,
      "learning_rate": 4.038904581812569e-06,
      "loss": 0.0557,
      "step": 58350
    },
    {
      "epoch": 5.564689636230469e-05,
      "model_forward_time": 1.7366371154785156,
      "step": 58350
    },
    {
      "dynamics_loss_avg": 0.0298040471971035,
      "epoch": 5.564689636230469e-05,
      "step": 58350
    },
    {
      "action_loss_avg": 0.001071162094012834,
      "epoch": 5.564689636230469e-05,
      "step": 58350
    },
    {
      "epoch": 5.564689636230469e-05,
      "step": 58350,
      "training_step_time": 7.765470504760742
    },
    {
      "epoch": 5.564785003662109e-05,
      "model_forward_time": 1.7389576435089111,
      "step": 58351
    },
    {
      "epoch": 5.564785003662109e-05,
      "step": 58351,
      "training_step_time": 8.625192403793335
    },
    {
      "epoch": 5.56488037109375e-05,
      "model_forward_time": 1.742049217224121,
      "step": 58352
    },
    {
      "epoch": 5.56488037109375e-05,
      "step": 58352,
      "training_step_time": 7.636224031448364
    },
    {
      "epoch": 5.564975738525391e-05,
      "model_forward_time": 1.7408015727996826,
      "step": 58353
    },
    {
      "epoch": 5.564975738525391e-05,
      "step": 58353,
      "training_step_time": 7.807180166244507
    },
    {
      "epoch": 5.565071105957031e-05,
      "model_forward_time": 1.7525975704193115,
      "step": 58354
    },
    {
      "epoch": 5.565071105957031e-05,
      "step": 58354,
      "training_step_time": 8.733672380447388
    },
    {
      "epoch": 5.5651664733886716e-05,
      "model_forward_time": 1.7440519332885742,
      "step": 58355
    },
    {
      "epoch": 5.5651664733886716e-05,
      "step": 58355,
      "training_step_time": 8.91474962234497
    },
    {
      "epoch": 5.5652618408203127e-05,
      "model_forward_time": 2.9217686653137207,
      "step": 58356
    },
    {
      "epoch": 5.5652618408203127e-05,
      "step": 58356,
      "training_step_time": 7.688093662261963
    },
    {
      "epoch": 5.565357208251953e-05,
      "model_forward_time": 1.7364206314086914,
      "step": 58357
    },
    {
      "epoch": 5.565357208251953e-05,
      "step": 58357,
      "training_step_time": 8.597741603851318
    },
    {
      "epoch": 5.5654525756835935e-05,
      "model_forward_time": 1.735213041305542,
      "step": 58358
    },
    {
      "epoch": 5.5654525756835935e-05,
      "step": 58358,
      "training_step_time": 7.682031869888306
    },
    {
      "epoch": 5.5655479431152345e-05,
      "model_forward_time": 1.738664150238037,
      "step": 58359
    },
    {
      "epoch": 5.5655479431152345e-05,
      "step": 58359,
      "training_step_time": 8.470841407775879
    },
    {
      "epoch": 5.565643310546875e-05,
      "grad_norm": 0.08101014792919159,
      "learning_rate": 4.03728199841759e-06,
      "loss": 0.0544,
      "step": 58360
    },
    {
      "epoch": 5.565643310546875e-05,
      "model_forward_time": 1.7434146404266357,
      "step": 58360
    },
    {
      "dynamics_loss_avg": 0.05664877668023109,
      "epoch": 5.565643310546875e-05,
      "step": 58360
    },
    {
      "action_loss_avg": 0.0012288682046346366,
      "epoch": 5.565643310546875e-05,
      "step": 58360
    },
    {
      "epoch": 5.565643310546875e-05,
      "step": 58360,
      "training_step_time": 7.62400221824646
    },
    {
      "epoch": 5.5657386779785153e-05,
      "model_forward_time": 1.7380716800689697,
      "step": 58361
    },
    {
      "epoch": 5.5657386779785153e-05,
      "step": 58361,
      "training_step_time": 7.6590821743011475
    },
    {
      "epoch": 5.5658340454101564e-05,
      "model_forward_time": 1.7450928688049316,
      "step": 58362
    },
    {
      "epoch": 5.5658340454101564e-05,
      "step": 58362,
      "training_step_time": 8.83482813835144
    },
    {
      "epoch": 5.565929412841797e-05,
      "model_forward_time": 1.7449345588684082,
      "step": 58363
    },
    {
      "epoch": 5.565929412841797e-05,
      "step": 58363,
      "training_step_time": 7.754045486450195
    },
    {
      "epoch": 5.566024780273437e-05,
      "model_forward_time": 1.750565528869629,
      "step": 58364
    },
    {
      "epoch": 5.566024780273437e-05,
      "step": 58364,
      "training_step_time": 8.493539810180664
    },
    {
      "epoch": 5.566120147705078e-05,
      "model_forward_time": 2.9525887966156006,
      "step": 58365
    },
    {
      "epoch": 5.566120147705078e-05,
      "step": 58365,
      "training_step_time": 9.167551040649414
    },
    {
      "epoch": 5.566215515136719e-05,
      "model_forward_time": 1.7566375732421875,
      "step": 58366
    },
    {
      "epoch": 5.566215515136719e-05,
      "step": 58366,
      "training_step_time": 8.550895929336548
    },
    {
      "epoch": 5.566310882568359e-05,
      "model_forward_time": 1.7569465637207031,
      "step": 58367
    },
    {
      "epoch": 5.566310882568359e-05,
      "step": 58367,
      "training_step_time": 8.702479362487793
    },
    {
      "epoch": 5.56640625e-05,
      "model_forward_time": 1.7425618171691895,
      "step": 58368
    },
    {
      "epoch": 5.56640625e-05,
      "step": 58368,
      "training_step_time": 7.7548744678497314
    },
    {
      "epoch": 5.5665016174316406e-05,
      "model_forward_time": 1.749330997467041,
      "step": 58369
    },
    {
      "epoch": 5.5665016174316406e-05,
      "step": 58369,
      "training_step_time": 7.659738779067993
    },
    {
      "epoch": 5.566596984863281e-05,
      "grad_norm": 0.09692459553480148,
      "learning_rate": 4.035659520304004e-06,
      "loss": 0.0574,
      "step": 58370
    },
    {
      "epoch": 5.566596984863281e-05,
      "model_forward_time": 1.7581167221069336,
      "step": 58370
    },
    {
      "dynamics_loss_avg": 0.050994066335260865,
      "epoch": 5.566596984863281e-05,
      "step": 58370
    },
    {
      "action_loss_avg": 0.0016670051321852953,
      "epoch": 5.566596984863281e-05,
      "step": 58370
    },
    {
      "epoch": 5.566596984863281e-05,
      "step": 58370,
      "training_step_time": 7.691391706466675
    },
    {
      "epoch": 5.566692352294922e-05,
      "model_forward_time": 1.804694414138794,
      "step": 58371
    },
    {
      "epoch": 5.566692352294922e-05,
      "step": 58371,
      "training_step_time": 8.807051181793213
    },
    {
      "epoch": 5.5667877197265625e-05,
      "model_forward_time": 1.758070945739746,
      "step": 58372
    },
    {
      "epoch": 5.5667877197265625e-05,
      "step": 58372,
      "training_step_time": 8.446032762527466
    },
    {
      "epoch": 5.566883087158203e-05,
      "model_forward_time": 1.749434471130371,
      "step": 58373
    },
    {
      "epoch": 5.566883087158203e-05,
      "step": 58373,
      "training_step_time": 7.726147651672363
    },
    {
      "epoch": 5.566978454589844e-05,
      "model_forward_time": 2.92508864402771,
      "step": 58374
    },
    {
      "epoch": 5.566978454589844e-05,
      "step": 58374,
      "training_step_time": 8.813695430755615
    },
    {
      "epoch": 5.5670738220214844e-05,
      "model_forward_time": 1.8767650127410889,
      "step": 58375
    },
    {
      "epoch": 5.5670738220214844e-05,
      "step": 58375,
      "training_step_time": 8.598153352737427
    },
    {
      "epoch": 5.567169189453125e-05,
      "model_forward_time": 1.7568252086639404,
      "step": 58376
    },
    {
      "epoch": 5.567169189453125e-05,
      "step": 58376,
      "training_step_time": 9.278051614761353
    },
    {
      "epoch": 5.567264556884766e-05,
      "model_forward_time": 1.7445547580718994,
      "step": 58377
    },
    {
      "epoch": 5.567264556884766e-05,
      "step": 58377,
      "training_step_time": 8.679168224334717
    },
    {
      "epoch": 5.567359924316406e-05,
      "model_forward_time": 1.7472410202026367,
      "step": 58378
    },
    {
      "epoch": 5.567359924316406e-05,
      "step": 58378,
      "training_step_time": 6.739576816558838
    },
    {
      "epoch": 5.567455291748047e-05,
      "model_forward_time": 1.7448318004608154,
      "step": 58379
    },
    {
      "epoch": 5.567455291748047e-05,
      "step": 58379,
      "training_step_time": 6.74702787399292
    },
    {
      "epoch": 5.567550659179688e-05,
      "grad_norm": 0.11738920211791992,
      "learning_rate": 4.034037147649243e-06,
      "loss": 0.056,
      "step": 58380
    },
    {
      "epoch": 5.567550659179688e-05,
      "model_forward_time": 1.7500946521759033,
      "step": 58380
    },
    {
      "dynamics_loss_avg": 0.03709889519959688,
      "epoch": 5.567550659179688e-05,
      "step": 58380
    },
    {
      "action_loss_avg": 0.0010224514815490692,
      "epoch": 5.567550659179688e-05,
      "step": 58380
    },
    {
      "epoch": 5.567550659179688e-05,
      "step": 58380,
      "training_step_time": 6.746460676193237
    },
    {
      "epoch": 5.567646026611328e-05,
      "model_forward_time": 1.7499449253082275,
      "step": 58381
    },
    {
      "epoch": 5.567646026611328e-05,
      "step": 58381,
      "training_step_time": 6.706432580947876
    },
    {
      "epoch": 5.5677413940429685e-05,
      "model_forward_time": 1.751272439956665,
      "step": 58382
    },
    {
      "epoch": 5.5677413940429685e-05,
      "step": 58382,
      "training_step_time": 6.738101482391357
    },
    {
      "epoch": 5.5678367614746096e-05,
      "model_forward_time": 1.753960371017456,
      "step": 58383
    },
    {
      "epoch": 5.5678367614746096e-05,
      "step": 58383,
      "training_step_time": 6.727936506271362
    },
    {
      "epoch": 5.56793212890625e-05,
      "model_forward_time": 1.752497911453247,
      "step": 58384
    },
    {
      "epoch": 5.56793212890625e-05,
      "step": 58384,
      "training_step_time": 6.724788188934326
    },
    {
      "epoch": 5.5680274963378904e-05,
      "model_forward_time": 1.75459623336792,
      "step": 58385
    },
    {
      "epoch": 5.5680274963378904e-05,
      "step": 58385,
      "training_step_time": 8.855225324630737
    },
    {
      "epoch": 5.5681228637695315e-05,
      "model_forward_time": 2.895672559738159,
      "step": 58386
    },
    {
      "epoch": 5.5681228637695315e-05,
      "step": 58386,
      "training_step_time": 7.7084009647369385
    },
    {
      "epoch": 5.568218231201172e-05,
      "model_forward_time": 1.7466282844543457,
      "step": 58387
    },
    {
      "epoch": 5.568218231201172e-05,
      "step": 58387,
      "training_step_time": 6.713695526123047
    },
    {
      "epoch": 5.568313598632812e-05,
      "model_forward_time": 1.7449407577514648,
      "step": 58388
    },
    {
      "epoch": 5.568313598632812e-05,
      "step": 58388,
      "training_step_time": 6.709074974060059
    },
    {
      "epoch": 5.5684089660644534e-05,
      "model_forward_time": 1.7491745948791504,
      "step": 58389
    },
    {
      "epoch": 5.5684089660644534e-05,
      "step": 58389,
      "training_step_time": 6.769386053085327
    },
    {
      "epoch": 5.568504333496094e-05,
      "grad_norm": 0.10092557966709137,
      "learning_rate": 4.032414880630727e-06,
      "loss": 0.056,
      "step": 58390
    },
    {
      "epoch": 5.568504333496094e-05,
      "model_forward_time": 1.753896713256836,
      "step": 58390
    },
    {
      "dynamics_loss_avg": 0.030816507525742054,
      "epoch": 5.568504333496094e-05,
      "step": 58390
    },
    {
      "action_loss_avg": 0.0008691243041539565,
      "epoch": 5.568504333496094e-05,
      "step": 58390
    },
    {
      "epoch": 5.568504333496094e-05,
      "step": 58390,
      "training_step_time": 6.747111082077026
    },
    {
      "epoch": 5.568599700927734e-05,
      "model_forward_time": 1.7430710792541504,
      "step": 58391
    },
    {
      "epoch": 5.568599700927734e-05,
      "step": 58391,
      "training_step_time": 6.7010862827301025
    },
    {
      "epoch": 5.568695068359375e-05,
      "model_forward_time": 1.7467906475067139,
      "step": 58392
    },
    {
      "epoch": 5.568695068359375e-05,
      "step": 58392,
      "training_step_time": 6.702900409698486
    },
    {
      "epoch": 5.568790435791016e-05,
      "model_forward_time": 1.3897290229797363,
      "step": 58393
    },
    {
      "epoch": 5.568790435791016e-05,
      "step": 58393,
      "training_step_time": 6.6774678230285645
    },
    {
      "epoch": 5.568885803222656e-05,
      "model_forward_time": 1.757657527923584,
      "step": 58394
    },
    {
      "epoch": 5.568885803222656e-05,
      "step": 58394,
      "training_step_time": 6.802222728729248
    },
    {
      "epoch": 5.568981170654297e-05,
      "model_forward_time": 2.8814823627471924,
      "step": 58395
    },
    {
      "epoch": 5.568981170654297e-05,
      "step": 58395,
      "training_step_time": 7.7281293869018555
    },
    {
      "epoch": 5.5690765380859376e-05,
      "model_forward_time": 1.7534656524658203,
      "step": 58396
    },
    {
      "epoch": 5.5690765380859376e-05,
      "step": 58396,
      "training_step_time": 6.714695453643799
    },
    {
      "epoch": 5.569171905517578e-05,
      "model_forward_time": 1.7522857189178467,
      "step": 58397
    },
    {
      "epoch": 5.569171905517578e-05,
      "step": 58397,
      "training_step_time": 6.8028693199157715
    },
    {
      "epoch": 5.569267272949219e-05,
      "model_forward_time": 1.7468056678771973,
      "step": 58398
    },
    {
      "epoch": 5.569267272949219e-05,
      "step": 58398,
      "training_step_time": 6.770099401473999
    },
    {
      "epoch": 5.5693626403808595e-05,
      "model_forward_time": 1.7442071437835693,
      "step": 58399
    },
    {
      "epoch": 5.5693626403808595e-05,
      "step": 58399,
      "training_step_time": 6.726926803588867
    },
    {
      "epoch": 5.5694580078125e-05,
      "grad_norm": 0.09105764329433441,
      "learning_rate": 4.030792719425864e-06,
      "loss": 0.0581,
      "step": 58400
    },
    {
      "epoch": 5.5694580078125e-05,
      "model_forward_time": 1.7489771842956543,
      "step": 58400
    },
    {
      "dynamics_loss_avg": 0.05224366504698992,
      "epoch": 5.5694580078125e-05,
      "step": 58400
    },
    {
      "action_loss_avg": 0.0011449215890024789,
      "epoch": 5.5694580078125e-05,
      "step": 58400
    },
    {
      "epoch": 5.5694580078125e-05,
      "step": 58400,
      "training_step_time": 6.748759031295776
    },
    {
      "epoch": 5.569553375244141e-05,
      "model_forward_time": 1.7394113540649414,
      "step": 58401
    },
    {
      "epoch": 5.569553375244141e-05,
      "step": 58401,
      "training_step_time": 6.790055513381958
    },
    {
      "epoch": 5.5696487426757813e-05,
      "model_forward_time": 1.748523235321045,
      "step": 58402
    },
    {
      "epoch": 5.5696487426757813e-05,
      "step": 58402,
      "training_step_time": 6.7819647789001465
    },
    {
      "epoch": 5.569744110107422e-05,
      "model_forward_time": 1.7467103004455566,
      "step": 58403
    },
    {
      "epoch": 5.569744110107422e-05,
      "step": 58403,
      "training_step_time": 6.783457517623901
    },
    {
      "epoch": 5.569839477539063e-05,
      "model_forward_time": 2.8899478912353516,
      "step": 58404
    },
    {
      "epoch": 5.569839477539063e-05,
      "step": 58404,
      "training_step_time": 8.342489004135132
    },
    {
      "epoch": 5.569934844970703e-05,
      "model_forward_time": 1.755890130996704,
      "step": 58405
    },
    {
      "epoch": 5.569934844970703e-05,
      "step": 58405,
      "training_step_time": 7.669341325759888
    },
    {
      "epoch": 5.5700302124023436e-05,
      "model_forward_time": 1.745255947113037,
      "step": 58406
    },
    {
      "epoch": 5.5700302124023436e-05,
      "step": 58406,
      "training_step_time": 6.769701719284058
    },
    {
      "epoch": 5.570125579833984e-05,
      "model_forward_time": 1.7572016716003418,
      "step": 58407
    },
    {
      "epoch": 5.570125579833984e-05,
      "step": 58407,
      "training_step_time": 6.774382591247559
    },
    {
      "epoch": 5.570220947265625e-05,
      "model_forward_time": 1.752713680267334,
      "step": 58408
    },
    {
      "epoch": 5.570220947265625e-05,
      "step": 58408,
      "training_step_time": 6.765240430831909
    },
    {
      "epoch": 5.5703163146972655e-05,
      "model_forward_time": 1.746340036392212,
      "step": 58409
    },
    {
      "epoch": 5.5703163146972655e-05,
      "step": 58409,
      "training_step_time": 6.7610273361206055
    },
    {
      "epoch": 5.570411682128906e-05,
      "grad_norm": 0.0784105584025383,
      "learning_rate": 4.0291706642120545e-06,
      "loss": 0.0569,
      "step": 58410
    },
    {
      "epoch": 5.570411682128906e-05,
      "model_forward_time": 1.7530946731567383,
      "step": 58410
    },
    {
      "dynamics_loss_avg": 0.058628760278224945,
      "epoch": 5.570411682128906e-05,
      "step": 58410
    },
    {
      "action_loss_avg": 0.0018635726941283793,
      "epoch": 5.570411682128906e-05,
      "step": 58410
    },
    {
      "epoch": 5.570411682128906e-05,
      "step": 58410,
      "training_step_time": 6.771250247955322
    },
    {
      "epoch": 5.570507049560547e-05,
      "model_forward_time": 1.7455644607543945,
      "step": 58411
    },
    {
      "epoch": 5.570507049560547e-05,
      "step": 58411,
      "training_step_time": 6.746302366256714
    },
    {
      "epoch": 5.5706024169921874e-05,
      "model_forward_time": 1.7585818767547607,
      "step": 58412
    },
    {
      "epoch": 5.5706024169921874e-05,
      "step": 58412,
      "training_step_time": 6.79684591293335
    },
    {
      "epoch": 5.570697784423828e-05,
      "model_forward_time": 2.8883793354034424,
      "step": 58413
    },
    {
      "epoch": 5.570697784423828e-05,
      "step": 58413,
      "training_step_time": 10.498513460159302
    },
    {
      "epoch": 5.570793151855469e-05,
      "model_forward_time": 1.7457695007324219,
      "step": 58414
    },
    {
      "epoch": 5.570793151855469e-05,
      "step": 58414,
      "training_step_time": 7.635921239852905
    },
    {
      "epoch": 5.570888519287109e-05,
      "model_forward_time": 1.7480289936065674,
      "step": 58415
    },
    {
      "epoch": 5.570888519287109e-05,
      "step": 58415,
      "training_step_time": 7.097028493881226
    },
    {
      "epoch": 5.57098388671875e-05,
      "model_forward_time": 1.7471024990081787,
      "step": 58416
    },
    {
      "epoch": 5.57098388671875e-05,
      "step": 58416,
      "training_step_time": 6.810834884643555
    },
    {
      "epoch": 5.571079254150391e-05,
      "model_forward_time": 1.7494525909423828,
      "step": 58417
    },
    {
      "epoch": 5.571079254150391e-05,
      "step": 58417,
      "training_step_time": 6.7131524085998535
    },
    {
      "epoch": 5.571174621582031e-05,
      "model_forward_time": 1.73154616355896,
      "step": 58418
    },
    {
      "epoch": 5.571174621582031e-05,
      "step": 58418,
      "training_step_time": 6.682957887649536
    },
    {
      "epoch": 5.5712699890136716e-05,
      "model_forward_time": 1.7419967651367188,
      "step": 58419
    },
    {
      "epoch": 5.5712699890136716e-05,
      "step": 58419,
      "training_step_time": 6.752511978149414
    },
    {
      "epoch": 5.571365356445313e-05,
      "grad_norm": 0.12198344618082047,
      "learning_rate": 4.027548715166679e-06,
      "loss": 0.0543,
      "step": 58420
    },
    {
      "epoch": 5.571365356445313e-05,
      "model_forward_time": 1.7590022087097168,
      "step": 58420
    },
    {
      "dynamics_loss_avg": 0.05332741066813469,
      "epoch": 5.571365356445313e-05,
      "step": 58420
    },
    {
      "action_loss_avg": 0.00133662341395393,
      "epoch": 5.571365356445313e-05,
      "step": 58420
    },
    {
      "epoch": 5.571365356445313e-05,
      "step": 58420,
      "training_step_time": 6.781543493270874
    },
    {
      "epoch": 5.571460723876953e-05,
      "model_forward_time": 1.7439563274383545,
      "step": 58421
    },
    {
      "epoch": 5.571460723876953e-05,
      "step": 58421,
      "training_step_time": 6.780149936676025
    },
    {
      "epoch": 5.5715560913085935e-05,
      "model_forward_time": 1.7575433254241943,
      "step": 58422
    },
    {
      "epoch": 5.5715560913085935e-05,
      "step": 58422,
      "training_step_time": 9.568106412887573
    },
    {
      "epoch": 5.5716514587402345e-05,
      "model_forward_time": 2.9265730381011963,
      "step": 58423
    },
    {
      "epoch": 5.5716514587402345e-05,
      "step": 58423,
      "training_step_time": 7.748314619064331
    },
    {
      "epoch": 5.571746826171875e-05,
      "model_forward_time": 1.7515802383422852,
      "step": 58424
    },
    {
      "epoch": 5.571746826171875e-05,
      "step": 58424,
      "training_step_time": 7.124189138412476
    },
    {
      "epoch": 5.5718421936035154e-05,
      "model_forward_time": 1.7435345649719238,
      "step": 58425
    },
    {
      "epoch": 5.5718421936035154e-05,
      "step": 58425,
      "training_step_time": 6.733168840408325
    },
    {
      "epoch": 5.5719375610351564e-05,
      "model_forward_time": 1.3969032764434814,
      "step": 58426
    },
    {
      "epoch": 5.5719375610351564e-05,
      "step": 58426,
      "training_step_time": 6.696453094482422
    },
    {
      "epoch": 5.572032928466797e-05,
      "model_forward_time": 1.7597332000732422,
      "step": 58427
    },
    {
      "epoch": 5.572032928466797e-05,
      "step": 58427,
      "training_step_time": 6.725053548812866
    },
    {
      "epoch": 5.572128295898437e-05,
      "model_forward_time": 1.7527518272399902,
      "step": 58428
    },
    {
      "epoch": 5.572128295898437e-05,
      "step": 58428,
      "training_step_time": 6.760920286178589
    },
    {
      "epoch": 5.572223663330078e-05,
      "model_forward_time": 1.7457537651062012,
      "step": 58429
    },
    {
      "epoch": 5.572223663330078e-05,
      "step": 58429,
      "training_step_time": 6.775742292404175
    },
    {
      "epoch": 5.572319030761719e-05,
      "grad_norm": 0.1057390570640564,
      "learning_rate": 4.025926872467116e-06,
      "loss": 0.0569,
      "step": 58430
    },
    {
      "epoch": 5.572319030761719e-05,
      "model_forward_time": 1.7532110214233398,
      "step": 58430
    },
    {
      "dynamics_loss_avg": 0.05164332883432508,
      "epoch": 5.572319030761719e-05,
      "step": 58430
    },
    {
      "action_loss_avg": 0.001565570631646551,
      "epoch": 5.572319030761719e-05,
      "step": 58430
    },
    {
      "epoch": 5.572319030761719e-05,
      "step": 58430,
      "training_step_time": 6.731556177139282
    },
    {
      "epoch": 5.572414398193359e-05,
      "model_forward_time": 1.7469069957733154,
      "step": 58431
    },
    {
      "epoch": 5.572414398193359e-05,
      "step": 58431,
      "training_step_time": 12.380217552185059
    },
    {
      "epoch": 5.572509765625e-05,
      "model_forward_time": 2.878535032272339,
      "step": 58432
    },
    {
      "epoch": 5.572509765625e-05,
      "step": 58432,
      "training_step_time": 9.339935064315796
    },
    {
      "epoch": 5.5726051330566406e-05,
      "model_forward_time": 1.7483654022216797,
      "step": 58433
    },
    {
      "epoch": 5.5726051330566406e-05,
      "step": 58433,
      "training_step_time": 7.103022575378418
    },
    {
      "epoch": 5.572700500488281e-05,
      "model_forward_time": 1.745222806930542,
      "step": 58434
    },
    {
      "epoch": 5.572700500488281e-05,
      "step": 58434,
      "training_step_time": 6.739837646484375
    },
    {
      "epoch": 5.572795867919922e-05,
      "model_forward_time": 1.746225118637085,
      "step": 58435
    },
    {
      "epoch": 5.572795867919922e-05,
      "step": 58435,
      "training_step_time": 6.741631031036377
    },
    {
      "epoch": 5.5728912353515625e-05,
      "model_forward_time": 1.7445402145385742,
      "step": 58436
    },
    {
      "epoch": 5.5728912353515625e-05,
      "step": 58436,
      "training_step_time": 6.766895771026611
    },
    {
      "epoch": 5.572986602783203e-05,
      "model_forward_time": 1.7494919300079346,
      "step": 58437
    },
    {
      "epoch": 5.572986602783203e-05,
      "step": 58437,
      "training_step_time": 6.728360652923584
    },
    {
      "epoch": 5.573081970214844e-05,
      "model_forward_time": 1.7494094371795654,
      "step": 58438
    },
    {
      "epoch": 5.573081970214844e-05,
      "step": 58438,
      "training_step_time": 6.731907606124878
    },
    {
      "epoch": 5.5731773376464844e-05,
      "model_forward_time": 1.7560946941375732,
      "step": 58439
    },
    {
      "epoch": 5.5731773376464844e-05,
      "step": 58439,
      "training_step_time": 6.743769645690918
    },
    {
      "epoch": 5.573272705078125e-05,
      "grad_norm": 0.11492861062288284,
      "learning_rate": 4.024305136290725e-06,
      "loss": 0.0541,
      "step": 58440
    },
    {
      "epoch": 5.573272705078125e-05,
      "model_forward_time": 1.7806940078735352,
      "step": 58440
    },
    {
      "dynamics_loss_avg": 0.06338399387896061,
      "epoch": 5.573272705078125e-05,
      "step": 58440
    },
    {
      "action_loss_avg": 0.0019655460899230094,
      "epoch": 5.573272705078125e-05,
      "step": 58440
    },
    {
      "epoch": 5.573272705078125e-05,
      "step": 58440,
      "training_step_time": 9.280844926834106
    },
    {
      "epoch": 5.573368072509766e-05,
      "model_forward_time": 2.860309362411499,
      "step": 58441
    },
    {
      "epoch": 5.573368072509766e-05,
      "step": 58441,
      "training_step_time": 8.580560445785522
    },
    {
      "epoch": 5.573463439941406e-05,
      "model_forward_time": 1.7419030666351318,
      "step": 58442
    },
    {
      "epoch": 5.573463439941406e-05,
      "step": 58442,
      "training_step_time": 7.69692063331604
    },
    {
      "epoch": 5.573558807373047e-05,
      "model_forward_time": 1.7493395805358887,
      "step": 58443
    },
    {
      "epoch": 5.573558807373047e-05,
      "step": 58443,
      "training_step_time": 7.159470081329346
    },
    {
      "epoch": 5.573654174804688e-05,
      "model_forward_time": 1.7468361854553223,
      "step": 58444
    },
    {
      "epoch": 5.573654174804688e-05,
      "step": 58444,
      "training_step_time": 6.781526565551758
    },
    {
      "epoch": 5.573749542236328e-05,
      "model_forward_time": 1.7452583312988281,
      "step": 58445
    },
    {
      "epoch": 5.573749542236328e-05,
      "step": 58445,
      "training_step_time": 6.773763418197632
    },
    {
      "epoch": 5.5738449096679686e-05,
      "model_forward_time": 1.7507085800170898,
      "step": 58446
    },
    {
      "epoch": 5.5738449096679686e-05,
      "step": 58446,
      "training_step_time": 6.741647481918335
    },
    {
      "epoch": 5.5739402770996096e-05,
      "model_forward_time": 1.749455451965332,
      "step": 58447
    },
    {
      "epoch": 5.5739402770996096e-05,
      "step": 58447,
      "training_step_time": 6.733297109603882
    },
    {
      "epoch": 5.57403564453125e-05,
      "model_forward_time": 1.7492239475250244,
      "step": 58448
    },
    {
      "epoch": 5.57403564453125e-05,
      "step": 58448,
      "training_step_time": 6.800448656082153
    },
    {
      "epoch": 5.5741310119628904e-05,
      "model_forward_time": 1.7470922470092773,
      "step": 58449
    },
    {
      "epoch": 5.5741310119628904e-05,
      "step": 58449,
      "training_step_time": 7.719529151916504
    },
    {
      "epoch": 5.5742263793945315e-05,
      "grad_norm": 0.103451669216156,
      "learning_rate": 4.022683506814858e-06,
      "loss": 0.0531,
      "step": 58450
    },
    {
      "epoch": 5.5742263793945315e-05,
      "model_forward_time": 2.890038013458252,
      "step": 58450
    },
    {
      "dynamics_loss_avg": 0.04334895517677069,
      "epoch": 5.5742263793945315e-05,
      "step": 58450
    },
    {
      "action_loss_avg": 0.0011719681264366955,
      "epoch": 5.5742263793945315e-05,
      "step": 58450
    },
    {
      "epoch": 5.5742263793945315e-05,
      "step": 58450,
      "training_step_time": 7.912819862365723
    },
    {
      "epoch": 5.574321746826172e-05,
      "model_forward_time": 1.7550437450408936,
      "step": 58451
    },
    {
      "epoch": 5.574321746826172e-05,
      "step": 58451,
      "training_step_time": 7.742380619049072
    },
    {
      "epoch": 5.574417114257812e-05,
      "model_forward_time": 1.7519068717956543,
      "step": 58452
    },
    {
      "epoch": 5.574417114257812e-05,
      "step": 58452,
      "training_step_time": 7.126841068267822
    },
    {
      "epoch": 5.5745124816894534e-05,
      "model_forward_time": 1.7799427509307861,
      "step": 58453
    },
    {
      "epoch": 5.5745124816894534e-05,
      "step": 58453,
      "training_step_time": 6.82244348526001
    },
    {
      "epoch": 5.574607849121094e-05,
      "model_forward_time": 1.7410032749176025,
      "step": 58454
    },
    {
      "epoch": 5.574607849121094e-05,
      "step": 58454,
      "training_step_time": 6.85361909866333
    },
    {
      "epoch": 5.574703216552734e-05,
      "model_forward_time": 1.7454235553741455,
      "step": 58455
    },
    {
      "epoch": 5.574703216552734e-05,
      "step": 58455,
      "training_step_time": 6.803958892822266
    },
    {
      "epoch": 5.574798583984375e-05,
      "model_forward_time": 1.7521259784698486,
      "step": 58456
    },
    {
      "epoch": 5.574798583984375e-05,
      "step": 58456,
      "training_step_time": 6.8013997077941895
    },
    {
      "epoch": 5.574893951416016e-05,
      "model_forward_time": 1.7473008632659912,
      "step": 58457
    },
    {
      "epoch": 5.574893951416016e-05,
      "step": 58457,
      "training_step_time": 6.786707878112793
    },
    {
      "epoch": 5.574989318847656e-05,
      "model_forward_time": 1.7453160285949707,
      "step": 58458
    },
    {
      "epoch": 5.574989318847656e-05,
      "step": 58458,
      "training_step_time": 7.693813800811768
    },
    {
      "epoch": 5.575084686279297e-05,
      "model_forward_time": 1.740476131439209,
      "step": 58459
    },
    {
      "epoch": 5.575084686279297e-05,
      "step": 58459,
      "training_step_time": 11.450685739517212
    },
    {
      "epoch": 5.5751800537109376e-05,
      "grad_norm": 0.10444071888923645,
      "learning_rate": 4.021061984216853e-06,
      "loss": 0.0532,
      "step": 58460
    },
    {
      "epoch": 5.5751800537109376e-05,
      "model_forward_time": 1.7534668445587158,
      "step": 58460
    },
    {
      "dynamics_loss_avg": 0.04378266967833042,
      "epoch": 5.5751800537109376e-05,
      "step": 58460
    },
    {
      "action_loss_avg": 0.0016067858145106585,
      "epoch": 5.5751800537109376e-05,
      "step": 58460
    },
    {
      "epoch": 5.5751800537109376e-05,
      "step": 58460,
      "training_step_time": 7.786494255065918
    },
    {
      "epoch": 5.575275421142578e-05,
      "model_forward_time": 1.7406904697418213,
      "step": 58461
    },
    {
      "epoch": 5.575275421142578e-05,
      "step": 58461,
      "training_step_time": 7.751354932785034
    },
    {
      "epoch": 5.575370788574219e-05,
      "model_forward_time": 1.7513856887817383,
      "step": 58462
    },
    {
      "epoch": 5.575370788574219e-05,
      "step": 58462,
      "training_step_time": 6.760443687438965
    },
    {
      "epoch": 5.5754661560058595e-05,
      "model_forward_time": 1.7473320960998535,
      "step": 58463
    },
    {
      "epoch": 5.5754661560058595e-05,
      "step": 58463,
      "training_step_time": 6.762239933013916
    },
    {
      "epoch": 5.5755615234375e-05,
      "model_forward_time": 1.747922658920288,
      "step": 58464
    },
    {
      "epoch": 5.5755615234375e-05,
      "step": 58464,
      "training_step_time": 6.780068874359131
    },
    {
      "epoch": 5.575656890869141e-05,
      "model_forward_time": 1.7518675327301025,
      "step": 58465
    },
    {
      "epoch": 5.575656890869141e-05,
      "step": 58465,
      "training_step_time": 6.771113395690918
    },
    {
      "epoch": 5.5757522583007814e-05,
      "model_forward_time": 1.7548184394836426,
      "step": 58466
    },
    {
      "epoch": 5.5757522583007814e-05,
      "step": 58466,
      "training_step_time": 6.777397155761719
    },
    {
      "epoch": 5.575847625732422e-05,
      "model_forward_time": 1.7441916465759277,
      "step": 58467
    },
    {
      "epoch": 5.575847625732422e-05,
      "step": 58467,
      "training_step_time": 7.778843402862549
    },
    {
      "epoch": 5.575942993164063e-05,
      "model_forward_time": 1.7404475212097168,
      "step": 58468
    },
    {
      "epoch": 5.575942993164063e-05,
      "step": 58468,
      "training_step_time": 10.763487815856934
    },
    {
      "epoch": 5.576038360595703e-05,
      "model_forward_time": 2.8989973068237305,
      "step": 58469
    },
    {
      "epoch": 5.576038360595703e-05,
      "step": 58469,
      "training_step_time": 7.814758539199829
    },
    {
      "epoch": 5.5761337280273436e-05,
      "grad_norm": 0.09601820260286331,
      "learning_rate": 4.019440568674035e-06,
      "loss": 0.0536,
      "step": 58470
    },
    {
      "epoch": 5.5761337280273436e-05,
      "model_forward_time": 1.756927251815796,
      "step": 58470
    },
    {
      "dynamics_loss_avg": 0.05246690167114139,
      "epoch": 5.5761337280273436e-05,
      "step": 58470
    },
    {
      "action_loss_avg": 0.0010409937181975692,
      "epoch": 5.5761337280273436e-05,
      "step": 58470
    },
    {
      "epoch": 5.5761337280273436e-05,
      "step": 58470,
      "training_step_time": 7.697371959686279
    },
    {
      "epoch": 5.576229095458984e-05,
      "model_forward_time": 1.7468416690826416,
      "step": 58471
    },
    {
      "epoch": 5.576229095458984e-05,
      "step": 58471,
      "training_step_time": 6.702191591262817
    },
    {
      "epoch": 5.576324462890625e-05,
      "model_forward_time": 1.745619535446167,
      "step": 58472
    },
    {
      "epoch": 5.576324462890625e-05,
      "step": 58472,
      "training_step_time": 6.70004677772522
    },
    {
      "epoch": 5.5764198303222655e-05,
      "model_forward_time": 1.7513737678527832,
      "step": 58473
    },
    {
      "epoch": 5.5764198303222655e-05,
      "step": 58473,
      "training_step_time": 6.638177394866943
    },
    {
      "epoch": 5.576515197753906e-05,
      "model_forward_time": 1.7456347942352295,
      "step": 58474
    },
    {
      "epoch": 5.576515197753906e-05,
      "step": 58474,
      "training_step_time": 6.627649545669556
    },
    {
      "epoch": 5.576610565185547e-05,
      "model_forward_time": 1.7524962425231934,
      "step": 58475
    },
    {
      "epoch": 5.576610565185547e-05,
      "step": 58475,
      "training_step_time": 6.628064155578613
    },
    {
      "epoch": 5.5767059326171874e-05,
      "model_forward_time": 1.743678092956543,
      "step": 58476
    },
    {
      "epoch": 5.5767059326171874e-05,
      "step": 58476,
      "training_step_time": 8.463665962219238
    },
    {
      "epoch": 5.576801300048828e-05,
      "model_forward_time": 1.747633457183838,
      "step": 58477
    },
    {
      "epoch": 5.576801300048828e-05,
      "step": 58477,
      "training_step_time": 8.782735109329224
    },
    {
      "epoch": 5.576896667480469e-05,
      "model_forward_time": 2.8861985206604004,
      "step": 58478
    },
    {
      "epoch": 5.576896667480469e-05,
      "step": 58478,
      "training_step_time": 8.427015542984009
    },
    {
      "epoch": 5.576992034912109e-05,
      "model_forward_time": 1.739614725112915,
      "step": 58479
    },
    {
      "epoch": 5.576992034912109e-05,
      "step": 58479,
      "training_step_time": 8.656902074813843
    },
    {
      "epoch": 5.57708740234375e-05,
      "grad_norm": 0.09441874176263809,
      "learning_rate": 4.017819260363724e-06,
      "loss": 0.0537,
      "step": 58480
    },
    {
      "epoch": 5.57708740234375e-05,
      "model_forward_time": 1.81406831741333,
      "step": 58480
    },
    {
      "dynamics_loss_avg": 0.04761443557217717,
      "epoch": 5.57708740234375e-05,
      "step": 58480
    },
    {
      "action_loss_avg": 0.0009561923012370244,
      "epoch": 5.57708740234375e-05,
      "step": 58480
    },
    {
      "epoch": 5.57708740234375e-05,
      "step": 58480,
      "training_step_time": 6.6890869140625
    },
    {
      "epoch": 5.577182769775391e-05,
      "model_forward_time": 1.4012436866760254,
      "step": 58481
    },
    {
      "epoch": 5.577182769775391e-05,
      "step": 58481,
      "training_step_time": 6.653234004974365
    },
    {
      "epoch": 5.577278137207031e-05,
      "model_forward_time": 1.7658379077911377,
      "step": 58482
    },
    {
      "epoch": 5.577278137207031e-05,
      "step": 58482,
      "training_step_time": 6.706500291824341
    },
    {
      "epoch": 5.5773735046386716e-05,
      "model_forward_time": 1.7650580406188965,
      "step": 58483
    },
    {
      "epoch": 5.5773735046386716e-05,
      "step": 58483,
      "training_step_time": 6.662999391555786
    },
    {
      "epoch": 5.577468872070313e-05,
      "model_forward_time": 1.7561345100402832,
      "step": 58484
    },
    {
      "epoch": 5.577468872070313e-05,
      "step": 58484,
      "training_step_time": 6.673195838928223
    },
    {
      "epoch": 5.577564239501953e-05,
      "model_forward_time": 1.7493011951446533,
      "step": 58485
    },
    {
      "epoch": 5.577564239501953e-05,
      "step": 58485,
      "training_step_time": 7.608257293701172
    },
    {
      "epoch": 5.5776596069335935e-05,
      "model_forward_time": 1.7634685039520264,
      "step": 58486
    },
    {
      "epoch": 5.5776596069335935e-05,
      "step": 58486,
      "training_step_time": 8.52135944366455
    },
    {
      "epoch": 5.5777549743652346e-05,
      "model_forward_time": 2.9355640411376953,
      "step": 58487
    },
    {
      "epoch": 5.5777549743652346e-05,
      "step": 58487,
      "training_step_time": 8.78577184677124
    },
    {
      "epoch": 5.577850341796875e-05,
      "model_forward_time": 1.7589776515960693,
      "step": 58488
    },
    {
      "epoch": 5.577850341796875e-05,
      "step": 58488,
      "training_step_time": 8.678926467895508
    },
    {
      "epoch": 5.5779457092285154e-05,
      "model_forward_time": 1.758751392364502,
      "step": 58489
    },
    {
      "epoch": 5.5779457092285154e-05,
      "step": 58489,
      "training_step_time": 7.000927209854126
    },
    {
      "epoch": 5.5780410766601564e-05,
      "grad_norm": 0.08617914468050003,
      "learning_rate": 4.016198059463223e-06,
      "loss": 0.0548,
      "step": 58490
    },
    {
      "epoch": 5.5780410766601564e-05,
      "model_forward_time": 1.7693226337432861,
      "step": 58490
    },
    {
      "dynamics_loss_avg": 0.049049248546361925,
      "epoch": 5.5780410766601564e-05,
      "step": 58490
    },
    {
      "action_loss_avg": 0.000879029743373394,
      "epoch": 5.5780410766601564e-05,
      "step": 58490
    },
    {
      "epoch": 5.5780410766601564e-05,
      "step": 58490,
      "training_step_time": 6.665089845657349
    },
    {
      "epoch": 5.578136444091797e-05,
      "model_forward_time": 1.7606689929962158,
      "step": 58491
    },
    {
      "epoch": 5.578136444091797e-05,
      "step": 58491,
      "training_step_time": 6.642701625823975
    },
    {
      "epoch": 5.578231811523437e-05,
      "model_forward_time": 1.7541706562042236,
      "step": 58492
    },
    {
      "epoch": 5.578231811523437e-05,
      "step": 58492,
      "training_step_time": 6.652027130126953
    },
    {
      "epoch": 5.578327178955078e-05,
      "model_forward_time": 1.7603020668029785,
      "step": 58493
    },
    {
      "epoch": 5.578327178955078e-05,
      "step": 58493,
      "training_step_time": 6.633890151977539
    },
    {
      "epoch": 5.578422546386719e-05,
      "model_forward_time": 1.7511427402496338,
      "step": 58494
    },
    {
      "epoch": 5.578422546386719e-05,
      "step": 58494,
      "training_step_time": 6.866333961486816
    },
    {
      "epoch": 5.578517913818359e-05,
      "model_forward_time": 1.756152868270874,
      "step": 58495
    },
    {
      "epoch": 5.578517913818359e-05,
      "step": 58495,
      "training_step_time": 8.711213827133179
    },
    {
      "epoch": 5.57861328125e-05,
      "model_forward_time": 1.3973510265350342,
      "step": 58496
    },
    {
      "epoch": 5.57861328125e-05,
      "step": 58496,
      "training_step_time": 8.343164443969727
    },
    {
      "epoch": 5.5787086486816406e-05,
      "model_forward_time": 1.8571667671203613,
      "step": 58497
    },
    {
      "epoch": 5.5787086486816406e-05,
      "step": 58497,
      "training_step_time": 8.691635131835938
    },
    {
      "epoch": 5.578804016113281e-05,
      "model_forward_time": 1.7389698028564453,
      "step": 58498
    },
    {
      "epoch": 5.578804016113281e-05,
      "step": 58498,
      "training_step_time": 7.530002593994141
    },
    {
      "epoch": 5.578899383544922e-05,
      "model_forward_time": 1.7564327716827393,
      "step": 58499
    },
    {
      "epoch": 5.578899383544922e-05,
      "step": 58499,
      "training_step_time": 6.657484769821167
    },
    {
      "epoch": 5.5789947509765625e-05,
      "grad_norm": 0.08585289865732193,
      "learning_rate": 4.014576966149823e-06,
      "loss": 0.0593,
      "step": 58500
    },
    {
      "epoch": 5.5789947509765625e-05,
      "model_forward_time": 1.7682995796203613,
      "step": 58500
    },
    {
      "dynamics_loss_avg": 0.054561006091535094,
      "epoch": 5.5789947509765625e-05,
      "step": 58500
    },
    {
      "action_loss_avg": 0.001129244311596267,
      "epoch": 5.5789947509765625e-05,
      "step": 58500
    },
    {
      "epoch": 5.5789947509765625e-05,
      "step": 58500,
      "training_step_time": 6.6821887493133545
    },
    {
      "epoch": 5.579090118408203e-05,
      "model_forward_time": 1.748018503189087,
      "step": 58501
    },
    {
      "epoch": 5.579090118408203e-05,
      "step": 58501,
      "training_step_time": 6.631299018859863
    },
    {
      "epoch": 5.579185485839844e-05,
      "model_forward_time": 1.7476069927215576,
      "step": 58502
    },
    {
      "epoch": 5.579185485839844e-05,
      "step": 58502,
      "training_step_time": 6.712429761886597
    },
    {
      "epoch": 5.5792808532714844e-05,
      "model_forward_time": 1.3951516151428223,
      "step": 58503
    },
    {
      "epoch": 5.5792808532714844e-05,
      "step": 58503,
      "training_step_time": 6.688417434692383
    },
    {
      "epoch": 5.579376220703125e-05,
      "model_forward_time": 1.4031708240509033,
      "step": 58504
    },
    {
      "epoch": 5.579376220703125e-05,
      "step": 58504,
      "training_step_time": 9.30417251586914
    },
    {
      "epoch": 5.579471588134766e-05,
      "model_forward_time": 1.7564961910247803,
      "step": 58505
    },
    {
      "epoch": 5.579471588134766e-05,
      "step": 58505,
      "training_step_time": 9.790164709091187
    },
    {
      "epoch": 5.579566955566406e-05,
      "model_forward_time": 2.927114963531494,
      "step": 58506
    },
    {
      "epoch": 5.579566955566406e-05,
      "step": 58506,
      "training_step_time": 8.747745752334595
    },
    {
      "epoch": 5.579662322998047e-05,
      "model_forward_time": 1.7546906471252441,
      "step": 58507
    },
    {
      "epoch": 5.579662322998047e-05,
      "step": 58507,
      "training_step_time": 7.625399589538574
    },
    {
      "epoch": 5.579757690429688e-05,
      "model_forward_time": 1.7564215660095215,
      "step": 58508
    },
    {
      "epoch": 5.579757690429688e-05,
      "step": 58508,
      "training_step_time": 6.707144260406494
    },
    {
      "epoch": 5.579853057861328e-05,
      "model_forward_time": 1.7645292282104492,
      "step": 58509
    },
    {
      "epoch": 5.579853057861328e-05,
      "step": 58509,
      "training_step_time": 6.633163928985596
    },
    {
      "epoch": 5.5799484252929686e-05,
      "grad_norm": 0.1109868511557579,
      "learning_rate": 4.012955980600804e-06,
      "loss": 0.0558,
      "step": 58510
    },
    {
      "epoch": 5.5799484252929686e-05,
      "model_forward_time": 1.765303373336792,
      "step": 58510
    },
    {
      "dynamics_loss_avg": 0.060330701805651185,
      "epoch": 5.5799484252929686e-05,
      "step": 58510
    },
    {
      "action_loss_avg": 0.001588631863705814,
      "epoch": 5.5799484252929686e-05,
      "step": 58510
    },
    {
      "epoch": 5.5799484252929686e-05,
      "step": 58510,
      "training_step_time": 6.683319568634033
    },
    {
      "epoch": 5.5800437927246096e-05,
      "model_forward_time": 1.7576136589050293,
      "step": 58511
    },
    {
      "epoch": 5.5800437927246096e-05,
      "step": 58511,
      "training_step_time": 6.72219443321228
    },
    {
      "epoch": 5.58013916015625e-05,
      "model_forward_time": 1.7502169609069824,
      "step": 58512
    },
    {
      "epoch": 5.58013916015625e-05,
      "step": 58512,
      "training_step_time": 6.654947996139526
    },
    {
      "epoch": 5.5802345275878904e-05,
      "model_forward_time": 1.7567028999328613,
      "step": 58513
    },
    {
      "epoch": 5.5802345275878904e-05,
      "step": 58513,
      "training_step_time": 8.536811590194702
    },
    {
      "epoch": 5.5803298950195315e-05,
      "model_forward_time": 1.7520692348480225,
      "step": 58514
    },
    {
      "epoch": 5.5803298950195315e-05,
      "step": 58514,
      "training_step_time": 8.814044713973999
    },
    {
      "epoch": 5.580425262451172e-05,
      "model_forward_time": 2.9354264736175537,
      "step": 58515
    },
    {
      "epoch": 5.580425262451172e-05,
      "step": 58515,
      "training_step_time": 7.839189767837524
    },
    {
      "epoch": 5.580520629882812e-05,
      "model_forward_time": 1.7609620094299316,
      "step": 58516
    },
    {
      "epoch": 5.580520629882812e-05,
      "step": 58516,
      "training_step_time": 7.685639142990112
    },
    {
      "epoch": 5.5806159973144534e-05,
      "model_forward_time": 1.7621726989746094,
      "step": 58517
    },
    {
      "epoch": 5.5806159973144534e-05,
      "step": 58517,
      "training_step_time": 7.10021185874939
    },
    {
      "epoch": 5.580711364746094e-05,
      "model_forward_time": 1.7468364238739014,
      "step": 58518
    },
    {
      "epoch": 5.580711364746094e-05,
      "step": 58518,
      "training_step_time": 6.635585069656372
    },
    {
      "epoch": 5.580806732177734e-05,
      "model_forward_time": 1.7572240829467773,
      "step": 58519
    },
    {
      "epoch": 5.580806732177734e-05,
      "step": 58519,
      "training_step_time": 6.676769018173218
    },
    {
      "epoch": 5.580902099609375e-05,
      "grad_norm": 0.09398950636386871,
      "learning_rate": 4.0113351029934355e-06,
      "loss": 0.0574,
      "step": 58520
    },
    {
      "epoch": 5.580902099609375e-05,
      "model_forward_time": 1.7625348567962646,
      "step": 58520
    },
    {
      "dynamics_loss_avg": 0.0623475244268775,
      "epoch": 5.580902099609375e-05,
      "step": 58520
    },
    {
      "action_loss_avg": 0.0015464888565475121,
      "epoch": 5.580902099609375e-05,
      "step": 58520
    },
    {
      "epoch": 5.580902099609375e-05,
      "step": 58520,
      "training_step_time": 6.70525050163269
    },
    {
      "epoch": 5.580997467041016e-05,
      "model_forward_time": 1.7532033920288086,
      "step": 58521
    },
    {
      "epoch": 5.580997467041016e-05,
      "step": 58521,
      "training_step_time": 6.816091299057007
    },
    {
      "epoch": 5.581092834472656e-05,
      "model_forward_time": 1.755284070968628,
      "step": 58522
    },
    {
      "epoch": 5.581092834472656e-05,
      "step": 58522,
      "training_step_time": 7.78364896774292
    },
    {
      "epoch": 5.581188201904297e-05,
      "model_forward_time": 1.7476081848144531,
      "step": 58523
    },
    {
      "epoch": 5.581188201904297e-05,
      "step": 58523,
      "training_step_time": 10.476356506347656
    },
    {
      "epoch": 5.5812835693359376e-05,
      "model_forward_time": 2.909304618835449,
      "step": 58524
    },
    {
      "epoch": 5.5812835693359376e-05,
      "step": 58524,
      "training_step_time": 7.818983554840088
    },
    {
      "epoch": 5.581378936767578e-05,
      "model_forward_time": 1.4010746479034424,
      "step": 58525
    },
    {
      "epoch": 5.581378936767578e-05,
      "step": 58525,
      "training_step_time": 7.754851818084717
    },
    {
      "epoch": 5.581474304199219e-05,
      "model_forward_time": 1.7585012912750244,
      "step": 58526
    },
    {
      "epoch": 5.581474304199219e-05,
      "step": 58526,
      "training_step_time": 7.11900782585144
    },
    {
      "epoch": 5.5815696716308595e-05,
      "model_forward_time": 1.7573974132537842,
      "step": 58527
    },
    {
      "epoch": 5.5815696716308595e-05,
      "step": 58527,
      "training_step_time": 6.8096113204956055
    },
    {
      "epoch": 5.5816650390625e-05,
      "model_forward_time": 1.7599585056304932,
      "step": 58528
    },
    {
      "epoch": 5.5816650390625e-05,
      "step": 58528,
      "training_step_time": 6.819932222366333
    },
    {
      "epoch": 5.581760406494141e-05,
      "model_forward_time": 1.750504732131958,
      "step": 58529
    },
    {
      "epoch": 5.581760406494141e-05,
      "step": 58529,
      "training_step_time": 6.747078895568848
    },
    {
      "epoch": 5.5818557739257814e-05,
      "grad_norm": 0.09089475870132446,
      "learning_rate": 4.0097143335049715e-06,
      "loss": 0.0602,
      "step": 58530
    },
    {
      "epoch": 5.5818557739257814e-05,
      "model_forward_time": 1.7539026737213135,
      "step": 58530
    },
    {
      "dynamics_loss_avg": 0.037960534542799,
      "epoch": 5.5818557739257814e-05,
      "step": 58530
    },
    {
      "action_loss_avg": 0.0007856422860641032,
      "epoch": 5.5818557739257814e-05,
      "step": 58530
    },
    {
      "epoch": 5.5818557739257814e-05,
      "step": 58530,
      "training_step_time": 6.785863399505615
    },
    {
      "epoch": 5.581951141357422e-05,
      "model_forward_time": 1.7577800750732422,
      "step": 58531
    },
    {
      "epoch": 5.581951141357422e-05,
      "step": 58531,
      "training_step_time": 7.603902578353882
    },
    {
      "epoch": 5.582046508789062e-05,
      "model_forward_time": 1.7508368492126465,
      "step": 58532
    },
    {
      "epoch": 5.582046508789062e-05,
      "step": 58532,
      "training_step_time": 9.391749382019043
    },
    {
      "epoch": 5.582141876220703e-05,
      "model_forward_time": 1.7742950916290283,
      "step": 58533
    },
    {
      "epoch": 5.582141876220703e-05,
      "step": 58533,
      "training_step_time": 9.593725442886353
    },
    {
      "epoch": 5.5822372436523436e-05,
      "model_forward_time": 2.9900918006896973,
      "step": 58534
    },
    {
      "epoch": 5.5822372436523436e-05,
      "step": 58534,
      "training_step_time": 8.275286436080933
    },
    {
      "epoch": 5.582332611083984e-05,
      "model_forward_time": 1.7259414196014404,
      "step": 58535
    },
    {
      "epoch": 5.582332611083984e-05,
      "step": 58535,
      "training_step_time": 7.646275520324707
    },
    {
      "epoch": 5.582427978515625e-05,
      "model_forward_time": 1.7524023056030273,
      "step": 58536
    },
    {
      "epoch": 5.582427978515625e-05,
      "step": 58536,
      "training_step_time": 6.715619087219238
    },
    {
      "epoch": 5.5825233459472655e-05,
      "model_forward_time": 1.7498960494995117,
      "step": 58537
    },
    {
      "epoch": 5.5825233459472655e-05,
      "step": 58537,
      "training_step_time": 6.7170798778533936
    },
    {
      "epoch": 5.582618713378906e-05,
      "model_forward_time": 1.7382314205169678,
      "step": 58538
    },
    {
      "epoch": 5.582618713378906e-05,
      "step": 58538,
      "training_step_time": 6.720296621322632
    },
    {
      "epoch": 5.582714080810547e-05,
      "model_forward_time": 1.7564351558685303,
      "step": 58539
    },
    {
      "epoch": 5.582714080810547e-05,
      "step": 58539,
      "training_step_time": 6.766038417816162
    },
    {
      "epoch": 5.5828094482421874e-05,
      "grad_norm": 0.1365165114402771,
      "learning_rate": 4.008093672312661e-06,
      "loss": 0.0588,
      "step": 58540
    },
    {
      "epoch": 5.5828094482421874e-05,
      "model_forward_time": 1.3856923580169678,
      "step": 58540
    },
    {
      "dynamics_loss_avg": 0.0465686809271574,
      "epoch": 5.5828094482421874e-05,
      "step": 58540
    },
    {
      "action_loss_avg": 0.0006494335830211639,
      "epoch": 5.5828094482421874e-05,
      "step": 58540
    },
    {
      "epoch": 5.5828094482421874e-05,
      "step": 58540,
      "training_step_time": 7.4256510734558105
    },
    {
      "epoch": 5.582904815673828e-05,
      "model_forward_time": 1.75010085105896,
      "step": 58541
    },
    {
      "epoch": 5.582904815673828e-05,
      "step": 58541,
      "training_step_time": 7.722736597061157
    },
    {
      "epoch": 5.583000183105469e-05,
      "model_forward_time": 1.7442662715911865,
      "step": 58542
    },
    {
      "epoch": 5.583000183105469e-05,
      "step": 58542,
      "training_step_time": 8.384130239486694
    },
    {
      "epoch": 5.583095550537109e-05,
      "model_forward_time": 3.018263101577759,
      "step": 58543
    },
    {
      "epoch": 5.583095550537109e-05,
      "step": 58543,
      "training_step_time": 7.814016819000244
    },
    {
      "epoch": 5.58319091796875e-05,
      "model_forward_time": 1.3941304683685303,
      "step": 58544
    },
    {
      "epoch": 5.58319091796875e-05,
      "step": 58544,
      "training_step_time": 7.652279853820801
    },
    {
      "epoch": 5.583286285400391e-05,
      "model_forward_time": 1.761831521987915,
      "step": 58545
    },
    {
      "epoch": 5.583286285400391e-05,
      "step": 58545,
      "training_step_time": 6.726743221282959
    },
    {
      "epoch": 5.583381652832031e-05,
      "model_forward_time": 1.753544807434082,
      "step": 58546
    },
    {
      "epoch": 5.583381652832031e-05,
      "step": 58546,
      "training_step_time": 6.736107587814331
    },
    {
      "epoch": 5.5834770202636716e-05,
      "model_forward_time": 1.847529411315918,
      "step": 58547
    },
    {
      "epoch": 5.5834770202636716e-05,
      "step": 58547,
      "training_step_time": 6.776196241378784
    },
    {
      "epoch": 5.583572387695313e-05,
      "model_forward_time": 1.7570149898529053,
      "step": 58548
    },
    {
      "epoch": 5.583572387695313e-05,
      "step": 58548,
      "training_step_time": 6.780848503112793
    },
    {
      "epoch": 5.583667755126953e-05,
      "model_forward_time": 1.756948709487915,
      "step": 58549
    },
    {
      "epoch": 5.583667755126953e-05,
      "step": 58549,
      "training_step_time": 7.449689865112305
    },
    {
      "epoch": 5.5837631225585935e-05,
      "grad_norm": 0.07234694063663483,
      "learning_rate": 4.0064731195937355e-06,
      "loss": 0.059,
      "step": 58550
    },
    {
      "epoch": 5.5837631225585935e-05,
      "model_forward_time": 1.7647769451141357,
      "step": 58550
    },
    {
      "dynamics_loss_avg": 0.04022971279919148,
      "epoch": 5.5837631225585935e-05,
      "step": 58550
    },
    {
      "action_loss_avg": 0.0018607743753818796,
      "epoch": 5.5837631225585935e-05,
      "step": 58550
    },
    {
      "epoch": 5.5837631225585935e-05,
      "step": 58550,
      "training_step_time": 7.6963982582092285
    },
    {
      "epoch": 5.5838584899902346e-05,
      "model_forward_time": 1.758744239807129,
      "step": 58551
    },
    {
      "epoch": 5.5838584899902346e-05,
      "step": 58551,
      "training_step_time": 8.73160982131958
    },
    {
      "epoch": 5.583953857421875e-05,
      "model_forward_time": 1.7596683502197266,
      "step": 58552
    },
    {
      "epoch": 5.583953857421875e-05,
      "step": 58552,
      "training_step_time": 8.92929196357727
    },
    {
      "epoch": 5.5840492248535154e-05,
      "model_forward_time": 1.8823468685150146,
      "step": 58553
    },
    {
      "epoch": 5.5840492248535154e-05,
      "step": 58553,
      "training_step_time": 7.640085458755493
    },
    {
      "epoch": 5.5841445922851564e-05,
      "model_forward_time": 1.7660977840423584,
      "step": 58554
    },
    {
      "epoch": 5.5841445922851564e-05,
      "step": 58554,
      "training_step_time": 6.731391191482544
    },
    {
      "epoch": 5.584239959716797e-05,
      "model_forward_time": 1.7527892589569092,
      "step": 58555
    },
    {
      "epoch": 5.584239959716797e-05,
      "step": 58555,
      "training_step_time": 6.717384576797485
    },
    {
      "epoch": 5.584335327148437e-05,
      "model_forward_time": 1.7584950923919678,
      "step": 58556
    },
    {
      "epoch": 5.584335327148437e-05,
      "step": 58556,
      "training_step_time": 6.723816394805908
    },
    {
      "epoch": 5.584430694580078e-05,
      "model_forward_time": 1.7539942264556885,
      "step": 58557
    },
    {
      "epoch": 5.584430694580078e-05,
      "step": 58557,
      "training_step_time": 6.714648485183716
    },
    {
      "epoch": 5.584526062011719e-05,
      "model_forward_time": 1.759512186050415,
      "step": 58558
    },
    {
      "epoch": 5.584526062011719e-05,
      "step": 58558,
      "training_step_time": 7.435538053512573
    },
    {
      "epoch": 5.584621429443359e-05,
      "model_forward_time": 1.7580351829528809,
      "step": 58559
    },
    {
      "epoch": 5.584621429443359e-05,
      "step": 58559,
      "training_step_time": 8.689950704574585
    },
    {
      "epoch": 5.584716796875e-05,
      "grad_norm": 0.08522475510835648,
      "learning_rate": 4.004852675525415e-06,
      "loss": 0.056,
      "step": 58560
    },
    {
      "epoch": 5.584716796875e-05,
      "model_forward_time": 1.773787021636963,
      "step": 58560
    },
    {
      "dynamics_loss_avg": 0.043133429158478974,
      "epoch": 5.584716796875e-05,
      "step": 58560
    },
    {
      "action_loss_avg": 0.0011326588690280914,
      "epoch": 5.584716796875e-05,
      "step": 58560
    },
    {
      "epoch": 5.584716796875e-05,
      "step": 58560,
      "training_step_time": 8.609709024429321
    },
    {
      "epoch": 5.5848121643066406e-05,
      "model_forward_time": 1.7586729526519775,
      "step": 58561
    },
    {
      "epoch": 5.5848121643066406e-05,
      "step": 58561,
      "training_step_time": 8.505801439285278
    },
    {
      "epoch": 5.584907531738281e-05,
      "model_forward_time": 2.9106011390686035,
      "step": 58562
    },
    {
      "epoch": 5.584907531738281e-05,
      "step": 58562,
      "training_step_time": 7.698568105697632
    },
    {
      "epoch": 5.585002899169922e-05,
      "model_forward_time": 1.7564523220062256,
      "step": 58563
    },
    {
      "epoch": 5.585002899169922e-05,
      "step": 58563,
      "training_step_time": 7.350054979324341
    },
    {
      "epoch": 5.5850982666015625e-05,
      "model_forward_time": 1.7632453441619873,
      "step": 58564
    },
    {
      "epoch": 5.5850982666015625e-05,
      "step": 58564,
      "training_step_time": 6.695109844207764
    },
    {
      "epoch": 5.585193634033203e-05,
      "model_forward_time": 1.7604122161865234,
      "step": 58565
    },
    {
      "epoch": 5.585193634033203e-05,
      "step": 58565,
      "training_step_time": 6.692914724349976
    },
    {
      "epoch": 5.585289001464844e-05,
      "model_forward_time": 1.7582976818084717,
      "step": 58566
    },
    {
      "epoch": 5.585289001464844e-05,
      "step": 58566,
      "training_step_time": 6.649316072463989
    },
    {
      "epoch": 5.5853843688964844e-05,
      "model_forward_time": 1.7777421474456787,
      "step": 58567
    },
    {
      "epoch": 5.5853843688964844e-05,
      "step": 58567,
      "training_step_time": 7.395134925842285
    },
    {
      "epoch": 5.585479736328125e-05,
      "model_forward_time": 1.7511992454528809,
      "step": 58568
    },
    {
      "epoch": 5.585479736328125e-05,
      "step": 58568,
      "training_step_time": 8.25036096572876
    },
    {
      "epoch": 5.585575103759766e-05,
      "model_forward_time": 1.7509868144989014,
      "step": 58569
    },
    {
      "epoch": 5.585575103759766e-05,
      "step": 58569,
      "training_step_time": 7.73831033706665
    },
    {
      "epoch": 5.585670471191406e-05,
      "grad_norm": 0.10400307178497314,
      "learning_rate": 4.003232340284911e-06,
      "loss": 0.0563,
      "step": 58570
    },
    {
      "epoch": 5.585670471191406e-05,
      "model_forward_time": 1.7487235069274902,
      "step": 58570
    },
    {
      "dynamics_loss_avg": 0.03610541485249996,
      "epoch": 5.585670471191406e-05,
      "step": 58570
    },
    {
      "action_loss_avg": 0.001142096213879995,
      "epoch": 5.585670471191406e-05,
      "step": 58570
    },
    {
      "epoch": 5.585670471191406e-05,
      "step": 58570,
      "training_step_time": 8.799838781356812
    },
    {
      "epoch": 5.585765838623047e-05,
      "model_forward_time": 2.9019501209259033,
      "step": 58571
    },
    {
      "epoch": 5.585765838623047e-05,
      "step": 58571,
      "training_step_time": 8.469792366027832
    },
    {
      "epoch": 5.585861206054688e-05,
      "model_forward_time": 1.7537281513214111,
      "step": 58572
    },
    {
      "epoch": 5.585861206054688e-05,
      "step": 58572,
      "training_step_time": 7.474835634231567
    },
    {
      "epoch": 5.585956573486328e-05,
      "model_forward_time": 1.7560596466064453,
      "step": 58573
    },
    {
      "epoch": 5.585956573486328e-05,
      "step": 58573,
      "training_step_time": 6.711275100708008
    },
    {
      "epoch": 5.5860519409179686e-05,
      "model_forward_time": 1.7548983097076416,
      "step": 58574
    },
    {
      "epoch": 5.5860519409179686e-05,
      "step": 58574,
      "training_step_time": 6.720710039138794
    },
    {
      "epoch": 5.5861473083496096e-05,
      "model_forward_time": 1.7565250396728516,
      "step": 58575
    },
    {
      "epoch": 5.5861473083496096e-05,
      "step": 58575,
      "training_step_time": 6.731752634048462
    },
    {
      "epoch": 5.58624267578125e-05,
      "model_forward_time": 1.7621054649353027,
      "step": 58576
    },
    {
      "epoch": 5.58624267578125e-05,
      "step": 58576,
      "training_step_time": 8.461278676986694
    },
    {
      "epoch": 5.5863380432128905e-05,
      "model_forward_time": 1.7590012550354004,
      "step": 58577
    },
    {
      "epoch": 5.5863380432128905e-05,
      "step": 58577,
      "training_step_time": 8.414650440216064
    },
    {
      "epoch": 5.5864334106445315e-05,
      "model_forward_time": 1.7636215686798096,
      "step": 58578
    },
    {
      "epoch": 5.5864334106445315e-05,
      "step": 58578,
      "training_step_time": 8.367799043655396
    },
    {
      "epoch": 5.586528778076172e-05,
      "model_forward_time": 1.7609925270080566,
      "step": 58579
    },
    {
      "epoch": 5.586528778076172e-05,
      "step": 58579,
      "training_step_time": 7.8024609088897705
    },
    {
      "epoch": 5.586624145507812e-05,
      "grad_norm": 0.08981408923864365,
      "learning_rate": 4.001612114049419e-06,
      "loss": 0.0555,
      "step": 58580
    },
    {
      "epoch": 5.586624145507812e-05,
      "model_forward_time": 2.907114028930664,
      "step": 58580
    },
    {
      "dynamics_loss_avg": 0.04287213021889329,
      "epoch": 5.586624145507812e-05,
      "step": 58580
    },
    {
      "action_loss_avg": 0.001053916843375191,
      "epoch": 5.586624145507812e-05,
      "step": 58580
    },
    {
      "epoch": 5.586624145507812e-05,
      "step": 58580,
      "training_step_time": 8.464983463287354
    },
    {
      "epoch": 5.5867195129394534e-05,
      "model_forward_time": 1.7604739665985107,
      "step": 58581
    },
    {
      "epoch": 5.5867195129394534e-05,
      "step": 58581,
      "training_step_time": 7.737919569015503
    },
    {
      "epoch": 5.586814880371094e-05,
      "model_forward_time": 1.764620304107666,
      "step": 58582
    },
    {
      "epoch": 5.586814880371094e-05,
      "step": 58582,
      "training_step_time": 6.788206100463867
    },
    {
      "epoch": 5.586910247802734e-05,
      "model_forward_time": 1.7558395862579346,
      "step": 58583
    },
    {
      "epoch": 5.586910247802734e-05,
      "step": 58583,
      "training_step_time": 6.818448066711426
    },
    {
      "epoch": 5.587005615234375e-05,
      "model_forward_time": 1.7621874809265137,
      "step": 58584
    },
    {
      "epoch": 5.587005615234375e-05,
      "step": 58584,
      "training_step_time": 6.784671783447266
    },
    {
      "epoch": 5.587100982666016e-05,
      "model_forward_time": 1.7572028636932373,
      "step": 58585
    },
    {
      "epoch": 5.587100982666016e-05,
      "step": 58585,
      "training_step_time": 7.432637453079224
    },
    {
      "epoch": 5.587196350097656e-05,
      "model_forward_time": 1.7592103481292725,
      "step": 58586
    },
    {
      "epoch": 5.587196350097656e-05,
      "step": 58586,
      "training_step_time": 8.432580471038818
    },
    {
      "epoch": 5.587291717529297e-05,
      "model_forward_time": 1.7546062469482422,
      "step": 58587
    },
    {
      "epoch": 5.587291717529297e-05,
      "step": 58587,
      "training_step_time": 9.7219557762146
    },
    {
      "epoch": 5.5873870849609376e-05,
      "model_forward_time": 1.755894422531128,
      "step": 58588
    },
    {
      "epoch": 5.5873870849609376e-05,
      "step": 58588,
      "training_step_time": 8.357656717300415
    },
    {
      "epoch": 5.587482452392578e-05,
      "model_forward_time": 2.9474377632141113,
      "step": 58589
    },
    {
      "epoch": 5.587482452392578e-05,
      "step": 58589,
      "training_step_time": 7.759617805480957
    },
    {
      "epoch": 5.587577819824219e-05,
      "grad_norm": 0.10673817247152328,
      "learning_rate": 3.999991996996122e-06,
      "loss": 0.0585,
      "step": 58590
    },
    {
      "epoch": 5.587577819824219e-05,
      "model_forward_time": 1.75718092918396,
      "step": 58590
    },
    {
      "dynamics_loss_avg": 0.057485864497721195,
      "epoch": 5.587577819824219e-05,
      "step": 58590
    },
    {
      "action_loss_avg": 0.0015903543040622025,
      "epoch": 5.587577819824219e-05,
      "step": 58590
    },
    {
      "epoch": 5.587577819824219e-05,
      "step": 58590,
      "training_step_time": 7.688287258148193
    },
    {
      "epoch": 5.5876731872558595e-05,
      "model_forward_time": 1.7596967220306396,
      "step": 58591
    },
    {
      "epoch": 5.5876731872558595e-05,
      "step": 58591,
      "training_step_time": 6.795775651931763
    },
    {
      "epoch": 5.5877685546875e-05,
      "model_forward_time": 1.7517366409301758,
      "step": 58592
    },
    {
      "epoch": 5.5877685546875e-05,
      "step": 58592,
      "training_step_time": 6.758611440658569
    },
    {
      "epoch": 5.587863922119141e-05,
      "model_forward_time": 1.762073040008545,
      "step": 58593
    },
    {
      "epoch": 5.587863922119141e-05,
      "step": 58593,
      "training_step_time": 6.785125255584717
    },
    {
      "epoch": 5.5879592895507814e-05,
      "model_forward_time": 1.761012077331543,
      "step": 58594
    },
    {
      "epoch": 5.5879592895507814e-05,
      "step": 58594,
      "training_step_time": 7.379694223403931
    },
    {
      "epoch": 5.588054656982422e-05,
      "model_forward_time": 1.756291151046753,
      "step": 58595
    },
    {
      "epoch": 5.588054656982422e-05,
      "step": 58595,
      "training_step_time": 7.658702373504639
    },
    {
      "epoch": 5.588150024414062e-05,
      "model_forward_time": 1.7554078102111816,
      "step": 58596
    },
    {
      "epoch": 5.588150024414062e-05,
      "step": 58596,
      "training_step_time": 8.525882482528687
    },
    {
      "epoch": 5.588245391845703e-05,
      "model_forward_time": 1.7518162727355957,
      "step": 58597
    },
    {
      "epoch": 5.588245391845703e-05,
      "step": 58597,
      "training_step_time": 10.013930559158325
    },
    {
      "epoch": 5.5883407592773437e-05,
      "model_forward_time": 2.961325168609619,
      "step": 58598
    },
    {
      "epoch": 5.5883407592773437e-05,
      "step": 58598,
      "training_step_time": 7.824428558349609
    },
    {
      "epoch": 5.588436126708984e-05,
      "model_forward_time": 1.761728048324585,
      "step": 58599
    },
    {
      "epoch": 5.588436126708984e-05,
      "step": 58599,
      "training_step_time": 9.08186960220337
    },
    {
      "epoch": 5.588531494140625e-05,
      "grad_norm": 0.13368995487689972,
      "learning_rate": 3.9983719893022e-06,
      "loss": 0.0593,
      "step": 58600
    },
    {
      "epoch": 5.588531494140625e-05,
      "model_forward_time": 1.7657713890075684,
      "step": 58600
    },
    {
      "dynamics_loss_avg": 0.05585278887301683,
      "epoch": 5.588531494140625e-05,
      "step": 58600
    },
    {
      "action_loss_avg": 0.0012912700651213527,
      "epoch": 5.588531494140625e-05,
      "step": 58600
    },
    {
      "epoch": 5.588531494140625e-05,
      "step": 58600,
      "training_step_time": 7.728536367416382
    },
    {
      "epoch": 5.5886268615722655e-05,
      "model_forward_time": 1.75809645652771,
      "step": 58601
    },
    {
      "epoch": 5.5886268615722655e-05,
      "step": 58601,
      "training_step_time": 6.748712778091431
    },
    {
      "epoch": 5.588722229003906e-05,
      "model_forward_time": 1.7604622840881348,
      "step": 58602
    },
    {
      "epoch": 5.588722229003906e-05,
      "step": 58602,
      "training_step_time": 6.730333089828491
    },
    {
      "epoch": 5.588817596435547e-05,
      "model_forward_time": 1.7513749599456787,
      "step": 58603
    },
    {
      "epoch": 5.588817596435547e-05,
      "step": 58603,
      "training_step_time": 7.371299743652344
    },
    {
      "epoch": 5.5889129638671874e-05,
      "model_forward_time": 1.764714002609253,
      "step": 58604
    },
    {
      "epoch": 5.5889129638671874e-05,
      "step": 58604,
      "training_step_time": 7.68446946144104
    },
    {
      "epoch": 5.589008331298828e-05,
      "model_forward_time": 1.8116891384124756,
      "step": 58605
    },
    {
      "epoch": 5.589008331298828e-05,
      "step": 58605,
      "training_step_time": 8.182212591171265
    },
    {
      "epoch": 5.589103698730469e-05,
      "model_forward_time": 1.7572705745697021,
      "step": 58606
    },
    {
      "epoch": 5.589103698730469e-05,
      "step": 58606,
      "training_step_time": 9.867264032363892
    },
    {
      "epoch": 5.589199066162109e-05,
      "model_forward_time": 1.4398090839385986,
      "step": 58607
    },
    {
      "epoch": 5.589199066162109e-05,
      "step": 58607,
      "training_step_time": 8.259774446487427
    },
    {
      "epoch": 5.58929443359375e-05,
      "model_forward_time": 2.944105386734009,
      "step": 58608
    },
    {
      "epoch": 5.58929443359375e-05,
      "step": 58608,
      "training_step_time": 8.68670916557312
    },
    {
      "epoch": 5.589389801025391e-05,
      "model_forward_time": 1.7576086521148682,
      "step": 58609
    },
    {
      "epoch": 5.589389801025391e-05,
      "step": 58609,
      "training_step_time": 7.745619297027588
    },
    {
      "epoch": 5.589485168457031e-05,
      "grad_norm": 0.08666357398033142,
      "learning_rate": 3.996752091144812e-06,
      "loss": 0.0573,
      "step": 58610
    },
    {
      "epoch": 5.589485168457031e-05,
      "model_forward_time": 1.7599883079528809,
      "step": 58610
    },
    {
      "dynamics_loss_avg": 0.0581302922219038,
      "epoch": 5.589485168457031e-05,
      "step": 58610
    },
    {
      "action_loss_avg": 0.0018831395223969593,
      "epoch": 5.589485168457031e-05,
      "step": 58610
    },
    {
      "epoch": 5.589485168457031e-05,
      "step": 58610,
      "training_step_time": 6.816936254501343
    },
    {
      "epoch": 5.5895805358886716e-05,
      "model_forward_time": 1.7504851818084717,
      "step": 58611
    },
    {
      "epoch": 5.5895805358886716e-05,
      "step": 58611,
      "training_step_time": 6.810605525970459
    },
    {
      "epoch": 5.589675903320313e-05,
      "model_forward_time": 1.7488667964935303,
      "step": 58612
    },
    {
      "epoch": 5.589675903320313e-05,
      "step": 58612,
      "training_step_time": 7.38563084602356
    },
    {
      "epoch": 5.589771270751953e-05,
      "model_forward_time": 1.7547197341918945,
      "step": 58613
    },
    {
      "epoch": 5.589771270751953e-05,
      "step": 58613,
      "training_step_time": 7.738843679428101
    },
    {
      "epoch": 5.5898666381835935e-05,
      "model_forward_time": 1.7544784545898438,
      "step": 58614
    },
    {
      "epoch": 5.5898666381835935e-05,
      "step": 58614,
      "training_step_time": 8.6161048412323
    },
    {
      "epoch": 5.5899620056152346e-05,
      "model_forward_time": 1.7640659809112549,
      "step": 58615
    },
    {
      "epoch": 5.5899620056152346e-05,
      "step": 58615,
      "training_step_time": 8.334335327148438
    },
    {
      "epoch": 5.590057373046875e-05,
      "model_forward_time": 1.7968313694000244,
      "step": 58616
    },
    {
      "epoch": 5.590057373046875e-05,
      "step": 58616,
      "training_step_time": 10.449106454849243
    },
    {
      "epoch": 5.5901527404785154e-05,
      "model_forward_time": 2.9671690464019775,
      "step": 58617
    },
    {
      "epoch": 5.5901527404785154e-05,
      "step": 58617,
      "training_step_time": 7.796588659286499
    },
    {
      "epoch": 5.5902481079101565e-05,
      "model_forward_time": 1.7623767852783203,
      "step": 58618
    },
    {
      "epoch": 5.5902481079101565e-05,
      "step": 58618,
      "training_step_time": 7.728910684585571
    },
    {
      "epoch": 5.590343475341797e-05,
      "model_forward_time": 1.7614927291870117,
      "step": 58619
    },
    {
      "epoch": 5.590343475341797e-05,
      "step": 58619,
      "training_step_time": 6.762011289596558
    },
    {
      "epoch": 5.590438842773437e-05,
      "grad_norm": 0.08600953221321106,
      "learning_rate": 3.995132302701106e-06,
      "loss": 0.0612,
      "step": 58620
    },
    {
      "epoch": 5.590438842773437e-05,
      "model_forward_time": 1.7700004577636719,
      "step": 58620
    },
    {
      "dynamics_loss_avg": 0.061871078424155714,
      "epoch": 5.590438842773437e-05,
      "step": 58620
    },
    {
      "action_loss_avg": 0.0025172730209305884,
      "epoch": 5.590438842773437e-05,
      "step": 58620
    },
    {
      "epoch": 5.590438842773437e-05,
      "step": 58620,
      "training_step_time": 6.770978689193726
    },
    {
      "epoch": 5.590534210205078e-05,
      "model_forward_time": 1.7585065364837646,
      "step": 58621
    },
    {
      "epoch": 5.590534210205078e-05,
      "step": 58621,
      "training_step_time": 7.392421245574951
    },
    {
      "epoch": 5.590629577636719e-05,
      "model_forward_time": 1.7630538940429688,
      "step": 58622
    },
    {
      "epoch": 5.590629577636719e-05,
      "step": 58622,
      "training_step_time": 7.719578504562378
    },
    {
      "epoch": 5.590724945068359e-05,
      "model_forward_time": 1.7675724029541016,
      "step": 58623
    },
    {
      "epoch": 5.590724945068359e-05,
      "step": 58623,
      "training_step_time": 7.594508171081543
    },
    {
      "epoch": 5.5908203125e-05,
      "model_forward_time": 1.7626585960388184,
      "step": 58624
    },
    {
      "epoch": 5.5908203125e-05,
      "step": 58624,
      "training_step_time": 9.344156265258789
    },
    {
      "epoch": 5.5909156799316406e-05,
      "model_forward_time": 1.7922673225402832,
      "step": 58625
    },
    {
      "epoch": 5.5909156799316406e-05,
      "step": 58625,
      "training_step_time": 9.864742040634155
    },
    {
      "epoch": 5.591011047363281e-05,
      "model_forward_time": 2.9163706302642822,
      "step": 58626
    },
    {
      "epoch": 5.591011047363281e-05,
      "step": 58626,
      "training_step_time": 7.676899194717407
    },
    {
      "epoch": 5.591106414794922e-05,
      "model_forward_time": 1.758286952972412,
      "step": 58627
    },
    {
      "epoch": 5.591106414794922e-05,
      "step": 58627,
      "training_step_time": 8.207730293273926
    },
    {
      "epoch": 5.5912017822265625e-05,
      "model_forward_time": 1.7583673000335693,
      "step": 58628
    },
    {
      "epoch": 5.5912017822265625e-05,
      "step": 58628,
      "training_step_time": 7.665745735168457
    },
    {
      "epoch": 5.591297149658203e-05,
      "model_forward_time": 1.7629327774047852,
      "step": 58629
    },
    {
      "epoch": 5.591297149658203e-05,
      "step": 58629,
      "training_step_time": 6.654181957244873
    },
    {
      "epoch": 5.591392517089844e-05,
      "grad_norm": 0.09383652359247208,
      "learning_rate": 3.993512624148219e-06,
      "loss": 0.0586,
      "step": 58630
    },
    {
      "epoch": 5.591392517089844e-05,
      "model_forward_time": 1.7765860557556152,
      "step": 58630
    },
    {
      "dynamics_loss_avg": 0.05862895324826241,
      "epoch": 5.591392517089844e-05,
      "step": 58630
    },
    {
      "action_loss_avg": 0.0018325860699405893,
      "epoch": 5.591392517089844e-05,
      "step": 58630
    },
    {
      "epoch": 5.591392517089844e-05,
      "step": 58630,
      "training_step_time": 7.335620880126953
    },
    {
      "epoch": 5.5914878845214844e-05,
      "model_forward_time": 1.7645437717437744,
      "step": 58631
    },
    {
      "epoch": 5.5914878845214844e-05,
      "step": 58631,
      "training_step_time": 7.616941213607788
    },
    {
      "epoch": 5.591583251953125e-05,
      "model_forward_time": 1.762171745300293,
      "step": 58632
    },
    {
      "epoch": 5.591583251953125e-05,
      "step": 58632,
      "training_step_time": 7.535122871398926
    },
    {
      "epoch": 5.591678619384766e-05,
      "model_forward_time": 1.753241777420044,
      "step": 58633
    },
    {
      "epoch": 5.591678619384766e-05,
      "step": 58633,
      "training_step_time": 8.406131744384766
    },
    {
      "epoch": 5.591773986816406e-05,
      "model_forward_time": 1.7864024639129639,
      "step": 58634
    },
    {
      "epoch": 5.591773986816406e-05,
      "step": 58634,
      "training_step_time": 9.507150173187256
    },
    {
      "epoch": 5.591869354248047e-05,
      "model_forward_time": 2.9203908443450928,
      "step": 58635
    },
    {
      "epoch": 5.591869354248047e-05,
      "step": 58635,
      "training_step_time": 8.78740930557251
    },
    {
      "epoch": 5.591964721679688e-05,
      "model_forward_time": 1.775698184967041,
      "step": 58636
    },
    {
      "epoch": 5.591964721679688e-05,
      "step": 58636,
      "training_step_time": 8.718817234039307
    },
    {
      "epoch": 5.592060089111328e-05,
      "model_forward_time": 1.763472557067871,
      "step": 58637
    },
    {
      "epoch": 5.592060089111328e-05,
      "step": 58637,
      "training_step_time": 7.73021936416626
    },
    {
      "epoch": 5.5921554565429686e-05,
      "model_forward_time": 1.7522990703582764,
      "step": 58638
    },
    {
      "epoch": 5.5921554565429686e-05,
      "step": 58638,
      "training_step_time": 6.743964195251465
    },
    {
      "epoch": 5.5922508239746097e-05,
      "model_forward_time": 1.7559888362884521,
      "step": 58639
    },
    {
      "epoch": 5.5922508239746097e-05,
      "step": 58639,
      "training_step_time": 7.433244228363037
    },
    {
      "epoch": 5.59234619140625e-05,
      "grad_norm": 0.08298715204000473,
      "learning_rate": 3.99189305566328e-06,
      "loss": 0.0594,
      "step": 58640
    },
    {
      "epoch": 5.59234619140625e-05,
      "model_forward_time": 1.7398736476898193,
      "step": 58640
    },
    {
      "dynamics_loss_avg": 0.05494586732238531,
      "epoch": 5.59234619140625e-05,
      "step": 58640
    },
    {
      "action_loss_avg": 0.001751257007708773,
      "epoch": 5.59234619140625e-05,
      "step": 58640
    },
    {
      "epoch": 5.59234619140625e-05,
      "step": 58640,
      "training_step_time": 7.620866775512695
    },
    {
      "epoch": 5.5924415588378905e-05,
      "model_forward_time": 1.3954558372497559,
      "step": 58641
    },
    {
      "epoch": 5.5924415588378905e-05,
      "step": 58641,
      "training_step_time": 7.61771821975708
    },
    {
      "epoch": 5.5925369262695315e-05,
      "model_forward_time": 1.7590420246124268,
      "step": 58642
    },
    {
      "epoch": 5.5925369262695315e-05,
      "step": 58642,
      "training_step_time": 8.830620050430298
    },
    {
      "epoch": 5.592632293701172e-05,
      "model_forward_time": 1.7481639385223389,
      "step": 58643
    },
    {
      "epoch": 5.592632293701172e-05,
      "step": 58643,
      "training_step_time": 9.137187480926514
    },
    {
      "epoch": 5.5927276611328123e-05,
      "model_forward_time": 2.9417483806610107,
      "step": 58644
    },
    {
      "epoch": 5.5927276611328123e-05,
      "step": 58644,
      "training_step_time": 7.664306163787842
    },
    {
      "epoch": 5.5928230285644534e-05,
      "model_forward_time": 1.7614035606384277,
      "step": 58645
    },
    {
      "epoch": 5.5928230285644534e-05,
      "step": 58645,
      "training_step_time": 7.617438554763794
    },
    {
      "epoch": 5.592918395996094e-05,
      "model_forward_time": 1.7540698051452637,
      "step": 58646
    },
    {
      "epoch": 5.592918395996094e-05,
      "step": 58646,
      "training_step_time": 7.624003171920776
    },
    {
      "epoch": 5.593013763427734e-05,
      "model_forward_time": 1.7496473789215088,
      "step": 58647
    },
    {
      "epoch": 5.593013763427734e-05,
      "step": 58647,
      "training_step_time": 6.664117336273193
    },
    {
      "epoch": 5.593109130859375e-05,
      "model_forward_time": 1.7612345218658447,
      "step": 58648
    },
    {
      "epoch": 5.593109130859375e-05,
      "step": 58648,
      "training_step_time": 7.368206024169922
    },
    {
      "epoch": 5.593204498291016e-05,
      "model_forward_time": 1.407956600189209,
      "step": 58649
    },
    {
      "epoch": 5.593204498291016e-05,
      "step": 58649,
      "training_step_time": 7.022953748703003
    },
    {
      "epoch": 5.593299865722656e-05,
      "grad_norm": 0.0746258869767189,
      "learning_rate": 3.990273597423401e-06,
      "loss": 0.056,
      "step": 58650
    },
    {
      "epoch": 5.593299865722656e-05,
      "model_forward_time": 1.7617418766021729,
      "step": 58650
    },
    {
      "dynamics_loss_avg": 0.04938525166362524,
      "epoch": 5.593299865722656e-05,
      "step": 58650
    },
    {
      "action_loss_avg": 0.0017520164939924144,
      "epoch": 5.593299865722656e-05,
      "step": 58650
    },
    {
      "epoch": 5.593299865722656e-05,
      "step": 58650,
      "training_step_time": 7.622003078460693
    },
    {
      "epoch": 5.593395233154297e-05,
      "model_forward_time": 1.7540953159332275,
      "step": 58651
    },
    {
      "epoch": 5.593395233154297e-05,
      "step": 58651,
      "training_step_time": 8.567881345748901
    },
    {
      "epoch": 5.5934906005859376e-05,
      "model_forward_time": 1.7553813457489014,
      "step": 58652
    },
    {
      "epoch": 5.5934906005859376e-05,
      "step": 58652,
      "training_step_time": 9.591695785522461
    },
    {
      "epoch": 5.593585968017578e-05,
      "model_forward_time": 1.8057761192321777,
      "step": 58653
    },
    {
      "epoch": 5.593585968017578e-05,
      "step": 58653,
      "training_step_time": 9.156837224960327
    },
    {
      "epoch": 5.593681335449219e-05,
      "model_forward_time": 2.9411566257476807,
      "step": 58654
    },
    {
      "epoch": 5.593681335449219e-05,
      "step": 58654,
      "training_step_time": 7.747239828109741
    },
    {
      "epoch": 5.5937767028808595e-05,
      "model_forward_time": 1.7545137405395508,
      "step": 58655
    },
    {
      "epoch": 5.5937767028808595e-05,
      "step": 58655,
      "training_step_time": 7.734252452850342
    },
    {
      "epoch": 5.5938720703125e-05,
      "model_forward_time": 1.4072906970977783,
      "step": 58656
    },
    {
      "epoch": 5.5938720703125e-05,
      "step": 58656,
      "training_step_time": 6.773649454116821
    },
    {
      "epoch": 5.59396743774414e-05,
      "model_forward_time": 1.7585203647613525,
      "step": 58657
    },
    {
      "epoch": 5.59396743774414e-05,
      "step": 58657,
      "training_step_time": 7.42075514793396
    },
    {
      "epoch": 5.5940628051757814e-05,
      "model_forward_time": 1.7606229782104492,
      "step": 58658
    },
    {
      "epoch": 5.5940628051757814e-05,
      "step": 58658,
      "training_step_time": 7.10278582572937
    },
    {
      "epoch": 5.594158172607422e-05,
      "model_forward_time": 1.7624478340148926,
      "step": 58659
    },
    {
      "epoch": 5.594158172607422e-05,
      "step": 58659,
      "training_step_time": 7.700500726699829
    },
    {
      "epoch": 5.594253540039062e-05,
      "grad_norm": 0.0699467882514,
      "learning_rate": 3.988654249605681e-06,
      "loss": 0.0552,
      "step": 58660
    },
    {
      "epoch": 5.594253540039062e-05,
      "model_forward_time": 1.777388334274292,
      "step": 58660
    },
    {
      "dynamics_loss_avg": 0.06629210077226162,
      "epoch": 5.594253540039062e-05,
      "step": 58660
    },
    {
      "action_loss_avg": 0.00129369708156446,
      "epoch": 5.594253540039062e-05,
      "step": 58660
    },
    {
      "epoch": 5.594253540039062e-05,
      "step": 58660,
      "training_step_time": 7.890752077102661
    },
    {
      "epoch": 5.594348907470703e-05,
      "model_forward_time": 1.7481696605682373,
      "step": 58661
    },
    {
      "epoch": 5.594348907470703e-05,
      "step": 58661,
      "training_step_time": 11.12390947341919
    },
    {
      "epoch": 5.5944442749023437e-05,
      "model_forward_time": 1.7844879627227783,
      "step": 58662
    },
    {
      "epoch": 5.5944442749023437e-05,
      "step": 58662,
      "training_step_time": 8.232002973556519
    },
    {
      "epoch": 5.594539642333984e-05,
      "model_forward_time": 2.8948912620544434,
      "step": 58663
    },
    {
      "epoch": 5.594539642333984e-05,
      "step": 58663,
      "training_step_time": 7.688976526260376
    },
    {
      "epoch": 5.594635009765625e-05,
      "model_forward_time": 1.7714488506317139,
      "step": 58664
    },
    {
      "epoch": 5.594635009765625e-05,
      "step": 58664,
      "training_step_time": 7.757902383804321
    },
    {
      "epoch": 5.5947303771972655e-05,
      "model_forward_time": 1.7609443664550781,
      "step": 58665
    },
    {
      "epoch": 5.5947303771972655e-05,
      "step": 58665,
      "training_step_time": 7.279376029968262
    },
    {
      "epoch": 5.594825744628906e-05,
      "model_forward_time": 1.753979206085205,
      "step": 58666
    },
    {
      "epoch": 5.594825744628906e-05,
      "step": 58666,
      "training_step_time": 7.3846659660339355
    },
    {
      "epoch": 5.594921112060547e-05,
      "model_forward_time": 1.7561538219451904,
      "step": 58667
    },
    {
      "epoch": 5.594921112060547e-05,
      "step": 58667,
      "training_step_time": 6.78590726852417
    },
    {
      "epoch": 5.5950164794921874e-05,
      "model_forward_time": 1.7596492767333984,
      "step": 58668
    },
    {
      "epoch": 5.5950164794921874e-05,
      "step": 58668,
      "training_step_time": 7.598389148712158
    },
    {
      "epoch": 5.595111846923828e-05,
      "model_forward_time": 1.7621324062347412,
      "step": 58669
    },
    {
      "epoch": 5.595111846923828e-05,
      "step": 58669,
      "training_step_time": 8.720046520233154
    },
    {
      "epoch": 5.595207214355469e-05,
      "grad_norm": 0.09311560541391373,
      "learning_rate": 3.987035012387213e-06,
      "loss": 0.0569,
      "step": 58670
    },
    {
      "epoch": 5.595207214355469e-05,
      "model_forward_time": 1.751955509185791,
      "step": 58670
    },
    {
      "dynamics_loss_avg": 0.04815509356558323,
      "epoch": 5.595207214355469e-05,
      "step": 58670
    },
    {
      "action_loss_avg": 0.001982243629754521,
      "epoch": 5.595207214355469e-05,
      "step": 58670
    },
    {
      "epoch": 5.595207214355469e-05,
      "step": 58670,
      "training_step_time": 7.71599006652832
    },
    {
      "epoch": 5.595302581787109e-05,
      "model_forward_time": 1.76289963722229,
      "step": 58671
    },
    {
      "epoch": 5.595302581787109e-05,
      "step": 58671,
      "training_step_time": 8.731688976287842
    },
    {
      "epoch": 5.59539794921875e-05,
      "model_forward_time": 2.863053798675537,
      "step": 58672
    },
    {
      "epoch": 5.59539794921875e-05,
      "step": 58672,
      "training_step_time": 7.661238670349121
    },
    {
      "epoch": 5.595493316650391e-05,
      "model_forward_time": 1.757863998413086,
      "step": 58673
    },
    {
      "epoch": 5.595493316650391e-05,
      "step": 58673,
      "training_step_time": 7.777141332626343
    },
    {
      "epoch": 5.595588684082031e-05,
      "model_forward_time": 1.7639482021331787,
      "step": 58674
    },
    {
      "epoch": 5.595588684082031e-05,
      "step": 58674,
      "training_step_time": 7.456981658935547
    },
    {
      "epoch": 5.5956840515136716e-05,
      "model_forward_time": 1.7563190460205078,
      "step": 58675
    },
    {
      "epoch": 5.5956840515136716e-05,
      "step": 58675,
      "training_step_time": 7.378721237182617
    },
    {
      "epoch": 5.595779418945313e-05,
      "model_forward_time": 1.7559494972229004,
      "step": 58676
    },
    {
      "epoch": 5.595779418945313e-05,
      "step": 58676,
      "training_step_time": 6.793696165084839
    },
    {
      "epoch": 5.595874786376953e-05,
      "model_forward_time": 1.7479863166809082,
      "step": 58677
    },
    {
      "epoch": 5.595874786376953e-05,
      "step": 58677,
      "training_step_time": 8.608663320541382
    },
    {
      "epoch": 5.5959701538085935e-05,
      "model_forward_time": 1.761925458908081,
      "step": 58678
    },
    {
      "epoch": 5.5959701538085935e-05,
      "step": 58678,
      "training_step_time": 7.879380464553833
    },
    {
      "epoch": 5.5960655212402346e-05,
      "model_forward_time": 1.7566375732421875,
      "step": 58679
    },
    {
      "epoch": 5.5960655212402346e-05,
      "step": 58679,
      "training_step_time": 7.613709449768066
    },
    {
      "epoch": 5.596160888671875e-05,
      "grad_norm": 0.11962217837572098,
      "learning_rate": 3.9854158859450735e-06,
      "loss": 0.0587,
      "step": 58680
    },
    {
      "epoch": 5.596160888671875e-05,
      "model_forward_time": 1.765364170074463,
      "step": 58680
    },
    {
      "dynamics_loss_avg": 0.060310326889157294,
      "epoch": 5.596160888671875e-05,
      "step": 58680
    },
    {
      "action_loss_avg": 0.0013205469062086196,
      "epoch": 5.596160888671875e-05,
      "step": 58680
    },
    {
      "epoch": 5.596160888671875e-05,
      "step": 58680,
      "training_step_time": 9.351789951324463
    },
    {
      "epoch": 5.5962562561035154e-05,
      "model_forward_time": 1.8492426872253418,
      "step": 58681
    },
    {
      "epoch": 5.5962562561035154e-05,
      "step": 58681,
      "training_step_time": 8.139481782913208
    },
    {
      "epoch": 5.5963516235351565e-05,
      "model_forward_time": 1.7644803524017334,
      "step": 58682
    },
    {
      "epoch": 5.5963516235351565e-05,
      "step": 58682,
      "training_step_time": 7.7219696044921875
    },
    {
      "epoch": 5.596446990966797e-05,
      "model_forward_time": 1.7608990669250488,
      "step": 58683
    },
    {
      "epoch": 5.596446990966797e-05,
      "step": 58683,
      "training_step_time": 7.689560651779175
    },
    {
      "epoch": 5.596542358398437e-05,
      "model_forward_time": 1.7562875747680664,
      "step": 58684
    },
    {
      "epoch": 5.596542358398437e-05,
      "step": 58684,
      "training_step_time": 6.73804235458374
    },
    {
      "epoch": 5.5966377258300783e-05,
      "model_forward_time": 1.749551773071289,
      "step": 58685
    },
    {
      "epoch": 5.5966377258300783e-05,
      "step": 58685,
      "training_step_time": 7.334901809692383
    },
    {
      "epoch": 5.596733093261719e-05,
      "model_forward_time": 1.756685495376587,
      "step": 58686
    },
    {
      "epoch": 5.596733093261719e-05,
      "step": 58686,
      "training_step_time": 7.245494842529297
    },
    {
      "epoch": 5.596828460693359e-05,
      "model_forward_time": 1.7606358528137207,
      "step": 58687
    },
    {
      "epoch": 5.596828460693359e-05,
      "step": 58687,
      "training_step_time": 7.5802905559539795
    },
    {
      "epoch": 5.596923828125e-05,
      "model_forward_time": 1.7541155815124512,
      "step": 58688
    },
    {
      "epoch": 5.596923828125e-05,
      "step": 58688,
      "training_step_time": 7.8783392906188965
    },
    {
      "epoch": 5.5970191955566406e-05,
      "model_forward_time": 1.7550809383392334,
      "step": 58689
    },
    {
      "epoch": 5.5970191955566406e-05,
      "step": 58689,
      "training_step_time": 8.763386726379395
    },
    {
      "epoch": 5.597114562988281e-05,
      "grad_norm": 0.09860914945602417,
      "learning_rate": 3.983796870456328e-06,
      "loss": 0.0551,
      "step": 58690
    },
    {
      "epoch": 5.597114562988281e-05,
      "model_forward_time": 1.861264705657959,
      "step": 58690
    },
    {
      "dynamics_loss_avg": 0.04169824691489339,
      "epoch": 5.597114562988281e-05,
      "step": 58690
    },
    {
      "action_loss_avg": 0.0015897701741778292,
      "epoch": 5.597114562988281e-05,
      "step": 58690
    },
    {
      "epoch": 5.597114562988281e-05,
      "step": 58690,
      "training_step_time": 8.630447387695312
    },
    {
      "epoch": 5.597209930419922e-05,
      "model_forward_time": 1.7546594142913818,
      "step": 58691
    },
    {
      "epoch": 5.597209930419922e-05,
      "step": 58691,
      "training_step_time": 8.299725770950317
    },
    {
      "epoch": 5.5973052978515625e-05,
      "model_forward_time": 1.755599021911621,
      "step": 58692
    },
    {
      "epoch": 5.5973052978515625e-05,
      "step": 58692,
      "training_step_time": 7.66006875038147
    },
    {
      "epoch": 5.597400665283203e-05,
      "model_forward_time": 1.7622911930084229,
      "step": 58693
    },
    {
      "epoch": 5.597400665283203e-05,
      "step": 58693,
      "training_step_time": 6.718241214752197
    },
    {
      "epoch": 5.597496032714844e-05,
      "model_forward_time": 1.756119966506958,
      "step": 58694
    },
    {
      "epoch": 5.597496032714844e-05,
      "step": 58694,
      "training_step_time": 7.352942943572998
    },
    {
      "epoch": 5.5975914001464844e-05,
      "model_forward_time": 1.7629258632659912,
      "step": 58695
    },
    {
      "epoch": 5.5975914001464844e-05,
      "step": 58695,
      "training_step_time": 7.29235053062439
    },
    {
      "epoch": 5.597686767578125e-05,
      "model_forward_time": 1.7498512268066406,
      "step": 58696
    },
    {
      "epoch": 5.597686767578125e-05,
      "step": 58696,
      "training_step_time": 7.544088363647461
    },
    {
      "epoch": 5.597782135009766e-05,
      "model_forward_time": 1.760263204574585,
      "step": 58697
    },
    {
      "epoch": 5.597782135009766e-05,
      "step": 58697,
      "training_step_time": 8.842002391815186
    },
    {
      "epoch": 5.597877502441406e-05,
      "model_forward_time": 1.4013993740081787,
      "step": 58698
    },
    {
      "epoch": 5.597877502441406e-05,
      "step": 58698,
      "training_step_time": 8.569845914840698
    },
    {
      "epoch": 5.597972869873047e-05,
      "model_forward_time": 1.8778951168060303,
      "step": 58699
    },
    {
      "epoch": 5.597972869873047e-05,
      "step": 58699,
      "training_step_time": 9.984904289245605
    },
    {
      "epoch": 5.598068237304688e-05,
      "grad_norm": 0.08241584151983261,
      "learning_rate": 3.9821779660980285e-06,
      "loss": 0.0567,
      "step": 58700
    },
    {
      "epoch": 5.598068237304688e-05,
      "model_forward_time": 2.855574131011963,
      "step": 58700
    },
    {
      "dynamics_loss_avg": 0.036030438262969255,
      "epoch": 5.598068237304688e-05,
      "step": 58700
    },
    {
      "action_loss_avg": 0.0017026979505317286,
      "epoch": 5.598068237304688e-05,
      "step": 58700
    },
    {
      "epoch": 5.598068237304688e-05,
      "step": 58700,
      "training_step_time": 7.644885540008545
    },
    {
      "epoch": 5.598163604736328e-05,
      "model_forward_time": 1.7572681903839111,
      "step": 58701
    },
    {
      "epoch": 5.598163604736328e-05,
      "step": 58701,
      "training_step_time": 7.676024913787842
    },
    {
      "epoch": 5.5982589721679686e-05,
      "model_forward_time": 1.756376028060913,
      "step": 58702
    },
    {
      "epoch": 5.5982589721679686e-05,
      "step": 58702,
      "training_step_time": 7.637443780899048
    },
    {
      "epoch": 5.5983543395996097e-05,
      "model_forward_time": 1.7670674324035645,
      "step": 58703
    },
    {
      "epoch": 5.5983543395996097e-05,
      "step": 58703,
      "training_step_time": 7.401432991027832
    },
    {
      "epoch": 5.59844970703125e-05,
      "model_forward_time": 1.755096435546875,
      "step": 58704
    },
    {
      "epoch": 5.59844970703125e-05,
      "step": 58704,
      "training_step_time": 7.364752292633057
    },
    {
      "epoch": 5.5985450744628905e-05,
      "model_forward_time": 1.7616686820983887,
      "step": 58705
    },
    {
      "epoch": 5.5985450744628905e-05,
      "step": 58705,
      "training_step_time": 7.699082612991333
    },
    {
      "epoch": 5.5986404418945315e-05,
      "model_forward_time": 1.7639789581298828,
      "step": 58706
    },
    {
      "epoch": 5.5986404418945315e-05,
      "step": 58706,
      "training_step_time": 9.839429140090942
    },
    {
      "epoch": 5.598735809326172e-05,
      "model_forward_time": 1.758164882659912,
      "step": 58707
    },
    {
      "epoch": 5.598735809326172e-05,
      "step": 58707,
      "training_step_time": 8.404174566268921
    },
    {
      "epoch": 5.5988311767578123e-05,
      "model_forward_time": 1.8950095176696777,
      "step": 58708
    },
    {
      "epoch": 5.5988311767578123e-05,
      "step": 58708,
      "training_step_time": 7.902364253997803
    },
    {
      "epoch": 5.5989265441894534e-05,
      "model_forward_time": 2.9836361408233643,
      "step": 58709
    },
    {
      "epoch": 5.5989265441894534e-05,
      "step": 58709,
      "training_step_time": 7.816127061843872
    },
    {
      "epoch": 5.599021911621094e-05,
      "grad_norm": 0.0908145010471344,
      "learning_rate": 3.980559173047216e-06,
      "loss": 0.0546,
      "step": 58710
    },
    {
      "epoch": 5.599021911621094e-05,
      "model_forward_time": 1.766721487045288,
      "step": 58710
    },
    {
      "dynamics_loss_avg": 0.04098582472652197,
      "epoch": 5.599021911621094e-05,
      "step": 58710
    },
    {
      "action_loss_avg": 0.0011843996355310083,
      "epoch": 5.599021911621094e-05,
      "step": 58710
    },
    {
      "epoch": 5.599021911621094e-05,
      "step": 58710,
      "training_step_time": 7.670376777648926
    },
    {
      "epoch": 5.599117279052734e-05,
      "model_forward_time": 1.7560844421386719,
      "step": 58711
    },
    {
      "epoch": 5.599117279052734e-05,
      "step": 58711,
      "training_step_time": 7.7154951095581055
    },
    {
      "epoch": 5.599212646484375e-05,
      "model_forward_time": 1.7878074645996094,
      "step": 58712
    },
    {
      "epoch": 5.599212646484375e-05,
      "step": 58712,
      "training_step_time": 7.394687652587891
    },
    {
      "epoch": 5.599308013916016e-05,
      "model_forward_time": 1.7538483142852783,
      "step": 58713
    },
    {
      "epoch": 5.599308013916016e-05,
      "step": 58713,
      "training_step_time": 7.33149790763855
    },
    {
      "epoch": 5.599403381347656e-05,
      "model_forward_time": 1.7577235698699951,
      "step": 58714
    },
    {
      "epoch": 5.599403381347656e-05,
      "step": 58714,
      "training_step_time": 7.674539804458618
    },
    {
      "epoch": 5.599498748779297e-05,
      "model_forward_time": 1.7561547756195068,
      "step": 58715
    },
    {
      "epoch": 5.599498748779297e-05,
      "step": 58715,
      "training_step_time": 7.9776771068573
    },
    {
      "epoch": 5.5995941162109376e-05,
      "model_forward_time": 1.7552289962768555,
      "step": 58716
    },
    {
      "epoch": 5.5995941162109376e-05,
      "step": 58716,
      "training_step_time": 8.529036521911621
    },
    {
      "epoch": 5.599689483642578e-05,
      "model_forward_time": 1.8616912364959717,
      "step": 58717
    },
    {
      "epoch": 5.599689483642578e-05,
      "step": 58717,
      "training_step_time": 7.759633779525757
    },
    {
      "epoch": 5.599784851074219e-05,
      "model_forward_time": 2.943580150604248,
      "step": 58718
    },
    {
      "epoch": 5.599784851074219e-05,
      "step": 58718,
      "training_step_time": 7.771777868270874
    },
    {
      "epoch": 5.5998802185058595e-05,
      "model_forward_time": 1.7699923515319824,
      "step": 58719
    },
    {
      "epoch": 5.5998802185058595e-05,
      "step": 58719,
      "training_step_time": 7.642038345336914
    },
    {
      "epoch": 5.5999755859375e-05,
      "grad_norm": 0.08761756122112274,
      "learning_rate": 3.978940491480918e-06,
      "loss": 0.0599,
      "step": 58720
    },
    {
      "epoch": 5.5999755859375e-05,
      "model_forward_time": 1.7683043479919434,
      "step": 58720
    },
    {
      "dynamics_loss_avg": 0.05660678297281265,
      "epoch": 5.5999755859375e-05,
      "step": 58720
    },
    {
      "action_loss_avg": 0.0009028219181345775,
      "epoch": 5.5999755859375e-05,
      "step": 58720
    },
    {
      "epoch": 5.5999755859375e-05,
      "step": 58720,
      "training_step_time": 7.761559009552002
    },
    {
      "epoch": 5.60007095336914e-05,
      "model_forward_time": 1.755535364151001,
      "step": 58721
    },
    {
      "epoch": 5.60007095336914e-05,
      "step": 58721,
      "training_step_time": 7.404228925704956
    },
    {
      "epoch": 5.6001663208007814e-05,
      "model_forward_time": 1.760335922241211,
      "step": 58722
    },
    {
      "epoch": 5.6001663208007814e-05,
      "step": 58722,
      "training_step_time": 7.368376970291138
    },
    {
      "epoch": 5.600261688232422e-05,
      "model_forward_time": 1.754885196685791,
      "step": 58723
    },
    {
      "epoch": 5.600261688232422e-05,
      "step": 58723,
      "training_step_time": 7.659217596054077
    },
    {
      "epoch": 5.600357055664062e-05,
      "model_forward_time": 1.8292813301086426,
      "step": 58724
    },
    {
      "epoch": 5.600357055664062e-05,
      "step": 58724,
      "training_step_time": 7.681853294372559
    },
    {
      "epoch": 5.600452423095703e-05,
      "model_forward_time": 1.7532563209533691,
      "step": 58725
    },
    {
      "epoch": 5.600452423095703e-05,
      "step": 58725,
      "training_step_time": 10.210070610046387
    },
    {
      "epoch": 5.600547790527344e-05,
      "model_forward_time": 1.7702231407165527,
      "step": 58726
    },
    {
      "epoch": 5.600547790527344e-05,
      "step": 58726,
      "training_step_time": 9.634944915771484
    },
    {
      "epoch": 5.600643157958984e-05,
      "model_forward_time": 2.9498372077941895,
      "step": 58727
    },
    {
      "epoch": 5.600643157958984e-05,
      "step": 58727,
      "training_step_time": 9.452273845672607
    },
    {
      "epoch": 5.600738525390625e-05,
      "model_forward_time": 1.7595295906066895,
      "step": 58728
    },
    {
      "epoch": 5.600738525390625e-05,
      "step": 58728,
      "training_step_time": 7.675501108169556
    },
    {
      "epoch": 5.6008338928222655e-05,
      "model_forward_time": 1.754307508468628,
      "step": 58729
    },
    {
      "epoch": 5.6008338928222655e-05,
      "step": 58729,
      "training_step_time": 9.502387762069702
    },
    {
      "epoch": 5.600929260253906e-05,
      "grad_norm": 0.12452031672000885,
      "learning_rate": 3.977321921576155e-06,
      "loss": 0.0539,
      "step": 58730
    },
    {
      "epoch": 5.600929260253906e-05,
      "model_forward_time": 1.7626659870147705,
      "step": 58730
    },
    {
      "dynamics_loss_avg": 0.06383186914026737,
      "epoch": 5.600929260253906e-05,
      "step": 58730
    },
    {
      "action_loss_avg": 0.0016072955797426403,
      "epoch": 5.600929260253906e-05,
      "step": 58730
    },
    {
      "epoch": 5.600929260253906e-05,
      "step": 58730,
      "training_step_time": 7.534748792648315
    },
    {
      "epoch": 5.601024627685547e-05,
      "model_forward_time": 1.7614986896514893,
      "step": 58731
    },
    {
      "epoch": 5.601024627685547e-05,
      "step": 58731,
      "training_step_time": 7.36238431930542
    },
    {
      "epoch": 5.6011199951171874e-05,
      "model_forward_time": 1.7592155933380127,
      "step": 58732
    },
    {
      "epoch": 5.6011199951171874e-05,
      "step": 58732,
      "training_step_time": 7.698497295379639
    },
    {
      "epoch": 5.601215362548828e-05,
      "model_forward_time": 1.760185956954956,
      "step": 58733
    },
    {
      "epoch": 5.601215362548828e-05,
      "step": 58733,
      "training_step_time": 7.65316367149353
    },
    {
      "epoch": 5.601310729980469e-05,
      "model_forward_time": 1.7624421119689941,
      "step": 58734
    },
    {
      "epoch": 5.601310729980469e-05,
      "step": 58734,
      "training_step_time": 8.451720952987671
    },
    {
      "epoch": 5.601406097412109e-05,
      "model_forward_time": 1.743110179901123,
      "step": 58735
    },
    {
      "epoch": 5.601406097412109e-05,
      "step": 58735,
      "training_step_time": 10.07868504524231
    },
    {
      "epoch": 5.60150146484375e-05,
      "model_forward_time": 2.966536045074463,
      "step": 58736
    },
    {
      "epoch": 5.60150146484375e-05,
      "step": 58736,
      "training_step_time": 7.8088788986206055
    },
    {
      "epoch": 5.601596832275391e-05,
      "model_forward_time": 1.7654292583465576,
      "step": 58737
    },
    {
      "epoch": 5.601596832275391e-05,
      "step": 58737,
      "training_step_time": 9.73820447921753
    },
    {
      "epoch": 5.601692199707031e-05,
      "model_forward_time": 1.7546501159667969,
      "step": 58738
    },
    {
      "epoch": 5.601692199707031e-05,
      "step": 58738,
      "training_step_time": 7.76580548286438
    },
    {
      "epoch": 5.6017875671386716e-05,
      "model_forward_time": 1.757406234741211,
      "step": 58739
    },
    {
      "epoch": 5.6017875671386716e-05,
      "step": 58739,
      "training_step_time": 7.66439414024353
    },
    {
      "epoch": 5.601882934570313e-05,
      "grad_norm": 0.14193189144134521,
      "learning_rate": 3.975703463509928e-06,
      "loss": 0.0555,
      "step": 58740
    },
    {
      "epoch": 5.601882934570313e-05,
      "model_forward_time": 1.7605113983154297,
      "step": 58740
    },
    {
      "dynamics_loss_avg": 0.042173738684505226,
      "epoch": 5.601882934570313e-05,
      "step": 58740
    },
    {
      "action_loss_avg": 0.0011586498818360268,
      "epoch": 5.601882934570313e-05,
      "step": 58740
    },
    {
      "epoch": 5.601882934570313e-05,
      "step": 58740,
      "training_step_time": 7.379422664642334
    },
    {
      "epoch": 5.601978302001953e-05,
      "model_forward_time": 1.7560560703277588,
      "step": 58741
    },
    {
      "epoch": 5.601978302001953e-05,
      "step": 58741,
      "training_step_time": 7.674615383148193
    },
    {
      "epoch": 5.6020736694335935e-05,
      "model_forward_time": 1.7515876293182373,
      "step": 58742
    },
    {
      "epoch": 5.6020736694335935e-05,
      "step": 58742,
      "training_step_time": 7.507815361022949
    },
    {
      "epoch": 5.6021690368652346e-05,
      "model_forward_time": 1.7610054016113281,
      "step": 58743
    },
    {
      "epoch": 5.6021690368652346e-05,
      "step": 58743,
      "training_step_time": 9.40869426727295
    },
    {
      "epoch": 5.602264404296875e-05,
      "model_forward_time": 1.7482001781463623,
      "step": 58744
    },
    {
      "epoch": 5.602264404296875e-05,
      "step": 58744,
      "training_step_time": 8.628339052200317
    },
    {
      "epoch": 5.6023597717285154e-05,
      "model_forward_time": 2.886486053466797,
      "step": 58745
    },
    {
      "epoch": 5.6023597717285154e-05,
      "step": 58745,
      "training_step_time": 8.769988536834717
    },
    {
      "epoch": 5.6024551391601565e-05,
      "model_forward_time": 1.7610976696014404,
      "step": 58746
    },
    {
      "epoch": 5.6024551391601565e-05,
      "step": 58746,
      "training_step_time": 7.67339301109314
    },
    {
      "epoch": 5.602550506591797e-05,
      "model_forward_time": 1.7436203956604004,
      "step": 58747
    },
    {
      "epoch": 5.602550506591797e-05,
      "step": 58747,
      "training_step_time": 9.271007537841797
    },
    {
      "epoch": 5.602645874023437e-05,
      "model_forward_time": 1.7367587089538574,
      "step": 58748
    },
    {
      "epoch": 5.602645874023437e-05,
      "step": 58748,
      "training_step_time": 7.701471567153931
    },
    {
      "epoch": 5.6027412414550783e-05,
      "model_forward_time": 1.7449333667755127,
      "step": 58749
    },
    {
      "epoch": 5.6027412414550783e-05,
      "step": 58749,
      "training_step_time": 7.362224578857422
    },
    {
      "epoch": 5.602836608886719e-05,
      "grad_norm": 0.07587428390979767,
      "learning_rate": 3.974085117459231e-06,
      "loss": 0.0538,
      "step": 58750
    },
    {
      "epoch": 5.602836608886719e-05,
      "model_forward_time": 1.3662502765655518,
      "step": 58750
    },
    {
      "dynamics_loss_avg": 0.05823839101940394,
      "epoch": 5.602836608886719e-05,
      "step": 58750
    },
    {
      "action_loss_avg": 0.001332474494120106,
      "epoch": 5.602836608886719e-05,
      "step": 58750
    },
    {
      "epoch": 5.602836608886719e-05,
      "step": 58750,
      "training_step_time": 7.591892719268799
    },
    {
      "epoch": 5.602931976318359e-05,
      "model_forward_time": 1.751418113708496,
      "step": 58751
    },
    {
      "epoch": 5.602931976318359e-05,
      "step": 58751,
      "training_step_time": 7.619143724441528
    },
    {
      "epoch": 5.60302734375e-05,
      "model_forward_time": 1.7631475925445557,
      "step": 58752
    },
    {
      "epoch": 5.60302734375e-05,
      "step": 58752,
      "training_step_time": 8.718419313430786
    },
    {
      "epoch": 5.6031227111816406e-05,
      "model_forward_time": 1.752657413482666,
      "step": 58753
    },
    {
      "epoch": 5.6031227111816406e-05,
      "step": 58753,
      "training_step_time": 7.705028533935547
    },
    {
      "epoch": 5.603218078613281e-05,
      "model_forward_time": 1.7858757972717285,
      "step": 58754
    },
    {
      "epoch": 5.603218078613281e-05,
      "step": 58754,
      "training_step_time": 8.706837892532349
    },
    {
      "epoch": 5.603313446044922e-05,
      "model_forward_time": 2.9367382526397705,
      "step": 58755
    },
    {
      "epoch": 5.603313446044922e-05,
      "step": 58755,
      "training_step_time": 7.712981224060059
    },
    {
      "epoch": 5.6034088134765625e-05,
      "model_forward_time": 1.7454826831817627,
      "step": 58756
    },
    {
      "epoch": 5.6034088134765625e-05,
      "step": 58756,
      "training_step_time": 7.585267782211304
    },
    {
      "epoch": 5.603504180908203e-05,
      "model_forward_time": 1.7390286922454834,
      "step": 58757
    },
    {
      "epoch": 5.603504180908203e-05,
      "step": 58757,
      "training_step_time": 7.724450349807739
    },
    {
      "epoch": 5.603599548339844e-05,
      "model_forward_time": 1.7462503910064697,
      "step": 58758
    },
    {
      "epoch": 5.603599548339844e-05,
      "step": 58758,
      "training_step_time": 7.303137540817261
    },
    {
      "epoch": 5.6036949157714844e-05,
      "model_forward_time": 1.7609984874725342,
      "step": 58759
    },
    {
      "epoch": 5.6036949157714844e-05,
      "step": 58759,
      "training_step_time": 6.691148996353149
    },
    {
      "epoch": 5.603790283203125e-05,
      "grad_norm": 0.10912350565195084,
      "learning_rate": 3.9724668836010435e-06,
      "loss": 0.0551,
      "step": 58760
    },
    {
      "epoch": 5.603790283203125e-05,
      "model_forward_time": 1.7687575817108154,
      "step": 58760
    },
    {
      "dynamics_loss_avg": 0.04447262939065695,
      "epoch": 5.603790283203125e-05,
      "step": 58760
    },
    {
      "action_loss_avg": 0.001223656296497211,
      "epoch": 5.603790283203125e-05,
      "step": 58760
    },
    {
      "epoch": 5.603790283203125e-05,
      "step": 58760,
      "training_step_time": 7.603666305541992
    },
    {
      "epoch": 5.603885650634766e-05,
      "model_forward_time": 1.763866901397705,
      "step": 58761
    },
    {
      "epoch": 5.603885650634766e-05,
      "step": 58761,
      "training_step_time": 8.623183727264404
    },
    {
      "epoch": 5.603981018066406e-05,
      "model_forward_time": 1.7590196132659912,
      "step": 58762
    },
    {
      "epoch": 5.603981018066406e-05,
      "step": 58762,
      "training_step_time": 7.6493754386901855
    },
    {
      "epoch": 5.604076385498047e-05,
      "model_forward_time": 1.8903748989105225,
      "step": 58763
    },
    {
      "epoch": 5.604076385498047e-05,
      "step": 58763,
      "training_step_time": 8.247092008590698
    },
    {
      "epoch": 5.604171752929688e-05,
      "model_forward_time": 2.933659553527832,
      "step": 58764
    },
    {
      "epoch": 5.604171752929688e-05,
      "step": 58764,
      "training_step_time": 7.671954870223999
    },
    {
      "epoch": 5.604267120361328e-05,
      "model_forward_time": 1.7680275440216064,
      "step": 58765
    },
    {
      "epoch": 5.604267120361328e-05,
      "step": 58765,
      "training_step_time": 7.558840990066528
    },
    {
      "epoch": 5.6043624877929686e-05,
      "model_forward_time": 1.769906759262085,
      "step": 58766
    },
    {
      "epoch": 5.6043624877929686e-05,
      "step": 58766,
      "training_step_time": 7.6836159229278564
    },
    {
      "epoch": 5.60445785522461e-05,
      "model_forward_time": 1.7531113624572754,
      "step": 58767
    },
    {
      "epoch": 5.60445785522461e-05,
      "step": 58767,
      "training_step_time": 7.57424259185791
    },
    {
      "epoch": 5.60455322265625e-05,
      "model_forward_time": 1.778050184249878,
      "step": 58768
    },
    {
      "epoch": 5.60455322265625e-05,
      "step": 58768,
      "training_step_time": 6.697666168212891
    },
    {
      "epoch": 5.6046485900878905e-05,
      "model_forward_time": 1.7670316696166992,
      "step": 58769
    },
    {
      "epoch": 5.6046485900878905e-05,
      "step": 58769,
      "training_step_time": 7.602295875549316
    },
    {
      "epoch": 5.6047439575195315e-05,
      "grad_norm": 0.09790648519992828,
      "learning_rate": 3.9708487621123325e-06,
      "loss": 0.053,
      "step": 58770
    },
    {
      "epoch": 5.6047439575195315e-05,
      "model_forward_time": 1.7630586624145508,
      "step": 58770
    },
    {
      "dynamics_loss_avg": 0.05935778059065342,
      "epoch": 5.6047439575195315e-05,
      "step": 58770
    },
    {
      "action_loss_avg": 0.0017699736112263055,
      "epoch": 5.6047439575195315e-05,
      "step": 58770
    },
    {
      "epoch": 5.6047439575195315e-05,
      "step": 58770,
      "training_step_time": 8.806021690368652
    },
    {
      "epoch": 5.604839324951172e-05,
      "model_forward_time": 1.7538940906524658,
      "step": 58771
    },
    {
      "epoch": 5.604839324951172e-05,
      "step": 58771,
      "training_step_time": 9.714687824249268
    },
    {
      "epoch": 5.6049346923828124e-05,
      "model_forward_time": 1.8586456775665283,
      "step": 58772
    },
    {
      "epoch": 5.6049346923828124e-05,
      "step": 58772,
      "training_step_time": 7.776782274246216
    },
    {
      "epoch": 5.6050300598144534e-05,
      "model_forward_time": 2.879418134689331,
      "step": 58773
    },
    {
      "epoch": 5.6050300598144534e-05,
      "step": 58773,
      "training_step_time": 7.662789344787598
    },
    {
      "epoch": 5.605125427246094e-05,
      "model_forward_time": 1.4062602519989014,
      "step": 58774
    },
    {
      "epoch": 5.605125427246094e-05,
      "step": 58774,
      "training_step_time": 7.549687385559082
    },
    {
      "epoch": 5.605220794677734e-05,
      "model_forward_time": 1.7632408142089844,
      "step": 58775
    },
    {
      "epoch": 5.605220794677734e-05,
      "step": 58775,
      "training_step_time": 7.734801292419434
    },
    {
      "epoch": 5.605316162109375e-05,
      "model_forward_time": 1.7637474536895752,
      "step": 58776
    },
    {
      "epoch": 5.605316162109375e-05,
      "step": 58776,
      "training_step_time": 7.590560436248779
    },
    {
      "epoch": 5.605411529541016e-05,
      "model_forward_time": 1.7595727443695068,
      "step": 58777
    },
    {
      "epoch": 5.605411529541016e-05,
      "step": 58777,
      "training_step_time": 7.296299457550049
    },
    {
      "epoch": 5.605506896972656e-05,
      "model_forward_time": 1.759887933731079,
      "step": 58778
    },
    {
      "epoch": 5.605506896972656e-05,
      "step": 58778,
      "training_step_time": 7.594608306884766
    },
    {
      "epoch": 5.605602264404297e-05,
      "model_forward_time": 1.7735347747802734,
      "step": 58779
    },
    {
      "epoch": 5.605602264404297e-05,
      "step": 58779,
      "training_step_time": 7.812629222869873
    },
    {
      "epoch": 5.6056976318359376e-05,
      "grad_norm": 0.08037800341844559,
      "learning_rate": 3.9692307531700504e-06,
      "loss": 0.0568,
      "step": 58780
    },
    {
      "epoch": 5.6056976318359376e-05,
      "model_forward_time": 1.7572345733642578,
      "step": 58780
    },
    {
      "dynamics_loss_avg": 0.046080716978758575,
      "epoch": 5.6056976318359376e-05,
      "step": 58780
    },
    {
      "action_loss_avg": 0.0015099774551345035,
      "epoch": 5.6056976318359376e-05,
      "step": 58780
    },
    {
      "epoch": 5.6056976318359376e-05,
      "step": 58780,
      "training_step_time": 7.681819915771484
    },
    {
      "epoch": 5.605792999267578e-05,
      "model_forward_time": 1.8307249546051025,
      "step": 58781
    },
    {
      "epoch": 5.605792999267578e-05,
      "step": 58781,
      "training_step_time": 7.672036170959473
    },
    {
      "epoch": 5.6058883666992184e-05,
      "model_forward_time": 2.868238687515259,
      "step": 58782
    },
    {
      "epoch": 5.6058883666992184e-05,
      "step": 58782,
      "training_step_time": 7.679194688796997
    },
    {
      "epoch": 5.6059837341308595e-05,
      "model_forward_time": 1.745270013809204,
      "step": 58783
    },
    {
      "epoch": 5.6059837341308595e-05,
      "step": 58783,
      "training_step_time": 7.52824068069458
    },
    {
      "epoch": 5.6060791015625e-05,
      "model_forward_time": 1.7559964656829834,
      "step": 58784
    },
    {
      "epoch": 5.6060791015625e-05,
      "step": 58784,
      "training_step_time": 8.461319923400879
    },
    {
      "epoch": 5.60617446899414e-05,
      "model_forward_time": 1.7593114376068115,
      "step": 58785
    },
    {
      "epoch": 5.60617446899414e-05,
      "step": 58785,
      "training_step_time": 7.574088096618652
    },
    {
      "epoch": 5.6062698364257814e-05,
      "model_forward_time": 1.7638256549835205,
      "step": 58786
    },
    {
      "epoch": 5.6062698364257814e-05,
      "step": 58786,
      "training_step_time": 7.37885594367981
    },
    {
      "epoch": 5.606365203857422e-05,
      "model_forward_time": 1.7604777812957764,
      "step": 58787
    },
    {
      "epoch": 5.606365203857422e-05,
      "step": 58787,
      "training_step_time": 7.646741628646851
    },
    {
      "epoch": 5.606460571289062e-05,
      "model_forward_time": 1.754086971282959,
      "step": 58788
    },
    {
      "epoch": 5.606460571289062e-05,
      "step": 58788,
      "training_step_time": 8.516350030899048
    },
    {
      "epoch": 5.606555938720703e-05,
      "model_forward_time": 1.7554428577423096,
      "step": 58789
    },
    {
      "epoch": 5.606555938720703e-05,
      "step": 58789,
      "training_step_time": 8.755512714385986
    },
    {
      "epoch": 5.606651306152344e-05,
      "grad_norm": 0.1412266343832016,
      "learning_rate": 3.967612856951146e-06,
      "loss": 0.0543,
      "step": 58790
    },
    {
      "epoch": 5.606651306152344e-05,
      "model_forward_time": 1.7913024425506592,
      "step": 58790
    },
    {
      "dynamics_loss_avg": 0.050215219147503376,
      "epoch": 5.606651306152344e-05,
      "step": 58790
    },
    {
      "action_loss_avg": 0.001146533849532716,
      "epoch": 5.606651306152344e-05,
      "step": 58790
    },
    {
      "epoch": 5.606651306152344e-05,
      "step": 58790,
      "training_step_time": 8.424481391906738
    },
    {
      "epoch": 5.606746673583984e-05,
      "model_forward_time": 1.7630999088287354,
      "step": 58791
    },
    {
      "epoch": 5.606746673583984e-05,
      "step": 58791,
      "training_step_time": 7.748357534408569
    },
    {
      "epoch": 5.606842041015625e-05,
      "model_forward_time": 3.002176523208618,
      "step": 58792
    },
    {
      "epoch": 5.606842041015625e-05,
      "step": 58792,
      "training_step_time": 7.745034217834473
    },
    {
      "epoch": 5.6069374084472656e-05,
      "model_forward_time": 1.759573221206665,
      "step": 58793
    },
    {
      "epoch": 5.6069374084472656e-05,
      "step": 58793,
      "training_step_time": 7.6594648361206055
    },
    {
      "epoch": 5.607032775878906e-05,
      "model_forward_time": 1.7597620487213135,
      "step": 58794
    },
    {
      "epoch": 5.607032775878906e-05,
      "step": 58794,
      "training_step_time": 7.685220241546631
    },
    {
      "epoch": 5.607128143310547e-05,
      "model_forward_time": 1.7600915431976318,
      "step": 58795
    },
    {
      "epoch": 5.607128143310547e-05,
      "step": 58795,
      "training_step_time": 7.575148105621338
    },
    {
      "epoch": 5.6072235107421874e-05,
      "model_forward_time": 1.757249116897583,
      "step": 58796
    },
    {
      "epoch": 5.6072235107421874e-05,
      "step": 58796,
      "training_step_time": 7.229203939437866
    },
    {
      "epoch": 5.607318878173828e-05,
      "model_forward_time": 1.7621607780456543,
      "step": 58797
    },
    {
      "epoch": 5.607318878173828e-05,
      "step": 58797,
      "training_step_time": 7.6682188510894775
    },
    {
      "epoch": 5.607414245605469e-05,
      "model_forward_time": 1.7590007781982422,
      "step": 58798
    },
    {
      "epoch": 5.607414245605469e-05,
      "step": 58798,
      "training_step_time": 7.802795648574829
    },
    {
      "epoch": 5.607509613037109e-05,
      "model_forward_time": 1.7756924629211426,
      "step": 58799
    },
    {
      "epoch": 5.607509613037109e-05,
      "step": 58799,
      "training_step_time": 8.343803405761719
    },
    {
      "epoch": 5.60760498046875e-05,
      "grad_norm": 0.09490452706813812,
      "learning_rate": 3.965995073632548e-06,
      "loss": 0.054,
      "step": 58800
    },
    {
      "epoch": 5.60760498046875e-05,
      "model_forward_time": 1.7641146183013916,
      "step": 58800
    },
    {
      "dynamics_loss_avg": 0.05130712203681469,
      "epoch": 5.60760498046875e-05,
      "step": 58800
    },
    {
      "action_loss_avg": 0.0015173097344813868,
      "epoch": 5.60760498046875e-05,
      "step": 58800
    },
    {
      "epoch": 5.60760498046875e-05,
      "step": 58800,
      "training_step_time": 8.562012195587158
    },
    {
      "epoch": 5.607700347900391e-05,
      "model_forward_time": 2.9284827709198,
      "step": 58801
    },
    {
      "epoch": 5.607700347900391e-05,
      "step": 58801,
      "training_step_time": 7.695190668106079
    },
    {
      "epoch": 5.607795715332031e-05,
      "model_forward_time": 1.7839229106903076,
      "step": 58802
    },
    {
      "epoch": 5.607795715332031e-05,
      "step": 58802,
      "training_step_time": 7.633740663528442
    },
    {
      "epoch": 5.6078910827636716e-05,
      "model_forward_time": 1.7543284893035889,
      "step": 58803
    },
    {
      "epoch": 5.6078910827636716e-05,
      "step": 58803,
      "training_step_time": 7.689038038253784
    },
    {
      "epoch": 5.607986450195313e-05,
      "model_forward_time": 1.7716197967529297,
      "step": 58804
    },
    {
      "epoch": 5.607986450195313e-05,
      "step": 58804,
      "training_step_time": 7.586777687072754
    },
    {
      "epoch": 5.608081817626953e-05,
      "model_forward_time": 1.7538378238677979,
      "step": 58805
    },
    {
      "epoch": 5.608081817626953e-05,
      "step": 58805,
      "training_step_time": 7.845793008804321
    },
    {
      "epoch": 5.6081771850585935e-05,
      "model_forward_time": 1.7579169273376465,
      "step": 58806
    },
    {
      "epoch": 5.6081771850585935e-05,
      "step": 58806,
      "training_step_time": 7.597141265869141
    },
    {
      "epoch": 5.6082725524902346e-05,
      "model_forward_time": 1.7667818069458008,
      "step": 58807
    },
    {
      "epoch": 5.6082725524902346e-05,
      "step": 58807,
      "training_step_time": 7.843461036682129
    },
    {
      "epoch": 5.608367919921875e-05,
      "model_forward_time": 1.7540137767791748,
      "step": 58808
    },
    {
      "epoch": 5.608367919921875e-05,
      "step": 58808,
      "training_step_time": 8.808544397354126
    },
    {
      "epoch": 5.6084632873535154e-05,
      "model_forward_time": 1.8592734336853027,
      "step": 58809
    },
    {
      "epoch": 5.6084632873535154e-05,
      "step": 58809,
      "training_step_time": 8.45231580734253
    },
    {
      "epoch": 5.6085586547851565e-05,
      "grad_norm": 0.15141521394252777,
      "learning_rate": 3.964377403391174e-06,
      "loss": 0.0595,
      "step": 58810
    },
    {
      "epoch": 5.6085586547851565e-05,
      "model_forward_time": 2.8581318855285645,
      "step": 58810
    },
    {
      "dynamics_loss_avg": 0.043675878271460536,
      "epoch": 5.6085586547851565e-05,
      "step": 58810
    },
    {
      "action_loss_avg": 0.0009203645560774021,
      "epoch": 5.6085586547851565e-05,
      "step": 58810
    },
    {
      "epoch": 5.6085586547851565e-05,
      "step": 58810,
      "training_step_time": 7.662561416625977
    },
    {
      "epoch": 5.608654022216797e-05,
      "model_forward_time": 1.7608587741851807,
      "step": 58811
    },
    {
      "epoch": 5.608654022216797e-05,
      "step": 58811,
      "training_step_time": 7.670163631439209
    },
    {
      "epoch": 5.608749389648437e-05,
      "model_forward_time": 1.7591259479522705,
      "step": 58812
    },
    {
      "epoch": 5.608749389648437e-05,
      "step": 58812,
      "training_step_time": 7.654765844345093
    },
    {
      "epoch": 5.6088447570800784e-05,
      "model_forward_time": 1.7525296211242676,
      "step": 58813
    },
    {
      "epoch": 5.6088447570800784e-05,
      "step": 58813,
      "training_step_time": 7.599662780761719
    },
    {
      "epoch": 5.608940124511719e-05,
      "model_forward_time": 1.7576463222503662,
      "step": 58814
    },
    {
      "epoch": 5.608940124511719e-05,
      "step": 58814,
      "training_step_time": 7.274702548980713
    },
    {
      "epoch": 5.609035491943359e-05,
      "model_forward_time": 1.7505719661712646,
      "step": 58815
    },
    {
      "epoch": 5.609035491943359e-05,
      "step": 58815,
      "training_step_time": 7.57865571975708
    },
    {
      "epoch": 5.609130859375e-05,
      "model_forward_time": 1.3961186408996582,
      "step": 58816
    },
    {
      "epoch": 5.609130859375e-05,
      "step": 58816,
      "training_step_time": 7.829977750778198
    },
    {
      "epoch": 5.6092262268066406e-05,
      "model_forward_time": 1.7498447895050049,
      "step": 58817
    },
    {
      "epoch": 5.6092262268066406e-05,
      "step": 58817,
      "training_step_time": 9.80105996131897
    },
    {
      "epoch": 5.609321594238281e-05,
      "model_forward_time": 1.813403606414795,
      "step": 58818
    },
    {
      "epoch": 5.609321594238281e-05,
      "step": 58818,
      "training_step_time": 7.915320158004761
    },
    {
      "epoch": 5.609416961669922e-05,
      "model_forward_time": 2.8500213623046875,
      "step": 58819
    },
    {
      "epoch": 5.609416961669922e-05,
      "step": 58819,
      "training_step_time": 9.165180444717407
    },
    {
      "epoch": 5.6095123291015625e-05,
      "grad_norm": 0.07479961961507797,
      "learning_rate": 3.96275984640393e-06,
      "loss": 0.0562,
      "step": 58820
    },
    {
      "epoch": 5.6095123291015625e-05,
      "model_forward_time": 1.7523655891418457,
      "step": 58820
    },
    {
      "dynamics_loss_avg": 0.06058642156422138,
      "epoch": 5.6095123291015625e-05,
      "step": 58820
    },
    {
      "action_loss_avg": 0.0015386798360850662,
      "epoch": 5.6095123291015625e-05,
      "step": 58820
    },
    {
      "epoch": 5.6095123291015625e-05,
      "step": 58820,
      "training_step_time": 8.466248512268066
    },
    {
      "epoch": 5.609607696533203e-05,
      "model_forward_time": 1.7529563903808594,
      "step": 58821
    },
    {
      "epoch": 5.609607696533203e-05,
      "step": 58821,
      "training_step_time": 7.614629030227661
    },
    {
      "epoch": 5.609703063964844e-05,
      "model_forward_time": 1.7491145133972168,
      "step": 58822
    },
    {
      "epoch": 5.609703063964844e-05,
      "step": 58822,
      "training_step_time": 7.66497278213501
    },
    {
      "epoch": 5.6097984313964844e-05,
      "model_forward_time": 1.7566282749176025,
      "step": 58823
    },
    {
      "epoch": 5.6097984313964844e-05,
      "step": 58823,
      "training_step_time": 7.529660701751709
    },
    {
      "epoch": 5.609893798828125e-05,
      "model_forward_time": 1.746692180633545,
      "step": 58824
    },
    {
      "epoch": 5.609893798828125e-05,
      "step": 58824,
      "training_step_time": 7.607390403747559
    },
    {
      "epoch": 5.609989166259766e-05,
      "model_forward_time": 1.7522666454315186,
      "step": 58825
    },
    {
      "epoch": 5.609989166259766e-05,
      "step": 58825,
      "training_step_time": 7.850954532623291
    },
    {
      "epoch": 5.610084533691406e-05,
      "model_forward_time": 1.7470877170562744,
      "step": 58826
    },
    {
      "epoch": 5.610084533691406e-05,
      "step": 58826,
      "training_step_time": 8.350998640060425
    },
    {
      "epoch": 5.610179901123047e-05,
      "model_forward_time": 1.812208652496338,
      "step": 58827
    },
    {
      "epoch": 5.610179901123047e-05,
      "step": 58827,
      "training_step_time": 9.500134944915771
    },
    {
      "epoch": 5.610275268554688e-05,
      "model_forward_time": 1.7533543109893799,
      "step": 58828
    },
    {
      "epoch": 5.610275268554688e-05,
      "step": 58828,
      "training_step_time": 8.737956047058105
    },
    {
      "epoch": 5.610370635986328e-05,
      "model_forward_time": 2.9463157653808594,
      "step": 58829
    },
    {
      "epoch": 5.610370635986328e-05,
      "step": 58829,
      "training_step_time": 7.665838241577148
    },
    {
      "epoch": 5.6104660034179686e-05,
      "grad_norm": 0.10741101205348969,
      "learning_rate": 3.96114240284771e-06,
      "loss": 0.0564,
      "step": 58830
    },
    {
      "epoch": 5.6104660034179686e-05,
      "model_forward_time": 1.7759957313537598,
      "step": 58830
    },
    {
      "dynamics_loss_avg": 0.04733044747263193,
      "epoch": 5.6104660034179686e-05,
      "step": 58830
    },
    {
      "action_loss_avg": 0.0012036085070576518,
      "epoch": 5.6104660034179686e-05,
      "step": 58830
    },
    {
      "epoch": 5.6104660034179686e-05,
      "step": 58830,
      "training_step_time": 7.63047456741333
    },
    {
      "epoch": 5.61056137084961e-05,
      "model_forward_time": 1.7456648349761963,
      "step": 58831
    },
    {
      "epoch": 5.61056137084961e-05,
      "step": 58831,
      "training_step_time": 7.6247169971466064
    },
    {
      "epoch": 5.61065673828125e-05,
      "model_forward_time": 1.753617286682129,
      "step": 58832
    },
    {
      "epoch": 5.61065673828125e-05,
      "step": 58832,
      "training_step_time": 7.501568794250488
    },
    {
      "epoch": 5.6107521057128905e-05,
      "model_forward_time": 1.7620232105255127,
      "step": 58833
    },
    {
      "epoch": 5.6107521057128905e-05,
      "step": 58833,
      "training_step_time": 7.561120986938477
    },
    {
      "epoch": 5.6108474731445316e-05,
      "model_forward_time": 1.7539763450622559,
      "step": 58834
    },
    {
      "epoch": 5.6108474731445316e-05,
      "step": 58834,
      "training_step_time": 8.345162391662598
    },
    {
      "epoch": 5.610942840576172e-05,
      "model_forward_time": 1.7509005069732666,
      "step": 58835
    },
    {
      "epoch": 5.610942840576172e-05,
      "step": 58835,
      "training_step_time": 8.543683290481567
    },
    {
      "epoch": 5.6110382080078124e-05,
      "model_forward_time": 1.7588763236999512,
      "step": 58836
    },
    {
      "epoch": 5.6110382080078124e-05,
      "step": 58836,
      "training_step_time": 8.345835447311401
    },
    {
      "epoch": 5.6111335754394534e-05,
      "model_forward_time": 1.763551950454712,
      "step": 58837
    },
    {
      "epoch": 5.6111335754394534e-05,
      "step": 58837,
      "training_step_time": 8.312625408172607
    },
    {
      "epoch": 5.611228942871094e-05,
      "model_forward_time": 2.949491262435913,
      "step": 58838
    },
    {
      "epoch": 5.611228942871094e-05,
      "step": 58838,
      "training_step_time": 7.639264106750488
    },
    {
      "epoch": 5.611324310302734e-05,
      "model_forward_time": 1.756091833114624,
      "step": 58839
    },
    {
      "epoch": 5.611324310302734e-05,
      "step": 58839,
      "training_step_time": 7.504483461380005
    },
    {
      "epoch": 5.611419677734375e-05,
      "grad_norm": 0.12352333962917328,
      "learning_rate": 3.959525072899394e-06,
      "loss": 0.0587,
      "step": 58840
    },
    {
      "epoch": 5.611419677734375e-05,
      "model_forward_time": 1.754483938217163,
      "step": 58840
    },
    {
      "dynamics_loss_avg": 0.059570855461061,
      "epoch": 5.611419677734375e-05,
      "step": 58840
    },
    {
      "action_loss_avg": 0.0012784681632183493,
      "epoch": 5.611419677734375e-05,
      "step": 58840
    },
    {
      "epoch": 5.611419677734375e-05,
      "step": 58840,
      "training_step_time": 7.631579637527466
    },
    {
      "epoch": 5.611515045166016e-05,
      "model_forward_time": 1.753962755203247,
      "step": 58841
    },
    {
      "epoch": 5.611515045166016e-05,
      "step": 58841,
      "training_step_time": 7.416511058807373
    },
    {
      "epoch": 5.611610412597656e-05,
      "model_forward_time": 1.7527718544006348,
      "step": 58842
    },
    {
      "epoch": 5.611610412597656e-05,
      "step": 58842,
      "training_step_time": 8.332107067108154
    },
    {
      "epoch": 5.611705780029297e-05,
      "model_forward_time": 1.7676112651824951,
      "step": 58843
    },
    {
      "epoch": 5.611705780029297e-05,
      "step": 58843,
      "training_step_time": 7.746906995773315
    },
    {
      "epoch": 5.6118011474609376e-05,
      "model_forward_time": 1.7515857219696045,
      "step": 58844
    },
    {
      "epoch": 5.6118011474609376e-05,
      "step": 58844,
      "training_step_time": 8.209458112716675
    },
    {
      "epoch": 5.611896514892578e-05,
      "model_forward_time": 1.7582993507385254,
      "step": 58845
    },
    {
      "epoch": 5.611896514892578e-05,
      "step": 58845,
      "training_step_time": 7.623919725418091
    },
    {
      "epoch": 5.6119918823242184e-05,
      "model_forward_time": 1.866468906402588,
      "step": 58846
    },
    {
      "epoch": 5.6119918823242184e-05,
      "step": 58846,
      "training_step_time": 9.319056987762451
    },
    {
      "epoch": 5.6120872497558595e-05,
      "model_forward_time": 2.937621593475342,
      "step": 58847
    },
    {
      "epoch": 5.6120872497558595e-05,
      "step": 58847,
      "training_step_time": 8.650970458984375
    },
    {
      "epoch": 5.6121826171875e-05,
      "model_forward_time": 1.7499678134918213,
      "step": 58848
    },
    {
      "epoch": 5.6121826171875e-05,
      "step": 58848,
      "training_step_time": 9.28610110282898
    },
    {
      "epoch": 5.61227798461914e-05,
      "model_forward_time": 1.7533583641052246,
      "step": 58849
    },
    {
      "epoch": 5.61227798461914e-05,
      "step": 58849,
      "training_step_time": 7.655014276504517
    },
    {
      "epoch": 5.6123733520507814e-05,
      "grad_norm": 0.10447805374860764,
      "learning_rate": 3.957907856735851e-06,
      "loss": 0.0573,
      "step": 58850
    },
    {
      "epoch": 5.6123733520507814e-05,
      "model_forward_time": 1.7561335563659668,
      "step": 58850
    },
    {
      "dynamics_loss_avg": 0.045896032080054286,
      "epoch": 5.6123733520507814e-05,
      "step": 58850
    },
    {
      "action_loss_avg": 0.0015553863951936365,
      "epoch": 5.6123733520507814e-05,
      "step": 58850
    },
    {
      "epoch": 5.6123733520507814e-05,
      "step": 58850,
      "training_step_time": 7.356381893157959
    },
    {
      "epoch": 5.612468719482422e-05,
      "model_forward_time": 1.752661943435669,
      "step": 58851
    },
    {
      "epoch": 5.612468719482422e-05,
      "step": 58851,
      "training_step_time": 7.659423351287842
    },
    {
      "epoch": 5.612564086914062e-05,
      "model_forward_time": 1.7622337341308594,
      "step": 58852
    },
    {
      "epoch": 5.612564086914062e-05,
      "step": 58852,
      "training_step_time": 7.793337106704712
    },
    {
      "epoch": 5.612659454345703e-05,
      "model_forward_time": 1.7562522888183594,
      "step": 58853
    },
    {
      "epoch": 5.612659454345703e-05,
      "step": 58853,
      "training_step_time": 7.615585088729858
    },
    {
      "epoch": 5.612754821777344e-05,
      "model_forward_time": 1.7645680904388428,
      "step": 58854
    },
    {
      "epoch": 5.612754821777344e-05,
      "step": 58854,
      "training_step_time": 7.683368921279907
    },
    {
      "epoch": 5.612850189208984e-05,
      "model_forward_time": 1.8442091941833496,
      "step": 58855
    },
    {
      "epoch": 5.612850189208984e-05,
      "step": 58855,
      "training_step_time": 7.626899480819702
    },
    {
      "epoch": 5.612945556640625e-05,
      "model_forward_time": 2.933885335922241,
      "step": 58856
    },
    {
      "epoch": 5.612945556640625e-05,
      "step": 58856,
      "training_step_time": 7.735937595367432
    },
    {
      "epoch": 5.6130409240722656e-05,
      "model_forward_time": 1.754455327987671,
      "step": 58857
    },
    {
      "epoch": 5.6130409240722656e-05,
      "step": 58857,
      "training_step_time": 7.5895373821258545
    },
    {
      "epoch": 5.613136291503906e-05,
      "model_forward_time": 1.7616462707519531,
      "step": 58858
    },
    {
      "epoch": 5.613136291503906e-05,
      "step": 58858,
      "training_step_time": 7.605525732040405
    },
    {
      "epoch": 5.613231658935547e-05,
      "model_forward_time": 1.7548918724060059,
      "step": 58859
    },
    {
      "epoch": 5.613231658935547e-05,
      "step": 58859,
      "training_step_time": 7.706403017044067
    },
    {
      "epoch": 5.6133270263671874e-05,
      "grad_norm": 0.10434748977422714,
      "learning_rate": 3.956290754533939e-06,
      "loss": 0.0582,
      "step": 58860
    },
    {
      "epoch": 5.6133270263671874e-05,
      "model_forward_time": 1.749509334564209,
      "step": 58860
    },
    {
      "dynamics_loss_avg": 0.043784981779754165,
      "epoch": 5.6133270263671874e-05,
      "step": 58860
    },
    {
      "action_loss_avg": 0.0007941777701489627,
      "epoch": 5.6133270263671874e-05,
      "step": 58860
    },
    {
      "epoch": 5.6133270263671874e-05,
      "step": 58860,
      "training_step_time": 7.652327060699463
    },
    {
      "epoch": 5.613422393798828e-05,
      "model_forward_time": 1.75156831741333,
      "step": 58861
    },
    {
      "epoch": 5.613422393798828e-05,
      "step": 58861,
      "training_step_time": 7.583315134048462
    },
    {
      "epoch": 5.613517761230469e-05,
      "model_forward_time": 1.7598109245300293,
      "step": 58862
    },
    {
      "epoch": 5.613517761230469e-05,
      "step": 58862,
      "training_step_time": 9.529585599899292
    },
    {
      "epoch": 5.613613128662109e-05,
      "model_forward_time": 1.7574572563171387,
      "step": 58863
    },
    {
      "epoch": 5.613613128662109e-05,
      "step": 58863,
      "training_step_time": 8.591840028762817
    },
    {
      "epoch": 5.61370849609375e-05,
      "model_forward_time": 1.8180663585662842,
      "step": 58864
    },
    {
      "epoch": 5.61370849609375e-05,
      "step": 58864,
      "training_step_time": 9.081015586853027
    },
    {
      "epoch": 5.613803863525391e-05,
      "model_forward_time": 2.9411778450012207,
      "step": 58865
    },
    {
      "epoch": 5.613803863525391e-05,
      "step": 58865,
      "training_step_time": 7.708876848220825
    },
    {
      "epoch": 5.613899230957031e-05,
      "model_forward_time": 1.7533719539642334,
      "step": 58866
    },
    {
      "epoch": 5.613899230957031e-05,
      "step": 58866,
      "training_step_time": 7.6152989864349365
    },
    {
      "epoch": 5.6139945983886716e-05,
      "model_forward_time": 1.740710973739624,
      "step": 58867
    },
    {
      "epoch": 5.6139945983886716e-05,
      "step": 58867,
      "training_step_time": 7.546363830566406
    },
    {
      "epoch": 5.614089965820313e-05,
      "model_forward_time": 1.7412753105163574,
      "step": 58868
    },
    {
      "epoch": 5.614089965820313e-05,
      "step": 58868,
      "training_step_time": 7.650463581085205
    },
    {
      "epoch": 5.614185333251953e-05,
      "model_forward_time": 1.394465446472168,
      "step": 58869
    },
    {
      "epoch": 5.614185333251953e-05,
      "step": 58869,
      "training_step_time": 7.656371593475342
    },
    {
      "epoch": 5.6142807006835935e-05,
      "grad_norm": 0.10302697122097015,
      "learning_rate": 3.954673766470502e-06,
      "loss": 0.058,
      "step": 58870
    },
    {
      "epoch": 5.6142807006835935e-05,
      "model_forward_time": 1.7529101371765137,
      "step": 58870
    },
    {
      "dynamics_loss_avg": 0.044301533699035646,
      "epoch": 5.6142807006835935e-05,
      "step": 58870
    },
    {
      "action_loss_avg": 0.0013052661844994873,
      "epoch": 5.6142807006835935e-05,
      "step": 58870
    },
    {
      "epoch": 5.6142807006835935e-05,
      "step": 58870,
      "training_step_time": 7.590884447097778
    },
    {
      "epoch": 5.6143760681152346e-05,
      "model_forward_time": 1.7567546367645264,
      "step": 58871
    },
    {
      "epoch": 5.6143760681152346e-05,
      "step": 58871,
      "training_step_time": 7.7940192222595215
    },
    {
      "epoch": 5.614471435546875e-05,
      "model_forward_time": 1.7617144584655762,
      "step": 58872
    },
    {
      "epoch": 5.614471435546875e-05,
      "step": 58872,
      "training_step_time": 8.167384147644043
    },
    {
      "epoch": 5.6145668029785154e-05,
      "model_forward_time": 1.8053553104400635,
      "step": 58873
    },
    {
      "epoch": 5.6145668029785154e-05,
      "step": 58873,
      "training_step_time": 8.986819505691528
    },
    {
      "epoch": 5.6146621704101565e-05,
      "model_forward_time": 2.93267822265625,
      "step": 58874
    },
    {
      "epoch": 5.6146621704101565e-05,
      "step": 58874,
      "training_step_time": 7.735238790512085
    },
    {
      "epoch": 5.614757537841797e-05,
      "model_forward_time": 1.7567591667175293,
      "step": 58875
    },
    {
      "epoch": 5.614757537841797e-05,
      "step": 58875,
      "training_step_time": 8.610066652297974
    },
    {
      "epoch": 5.614852905273437e-05,
      "model_forward_time": 1.7560539245605469,
      "step": 58876
    },
    {
      "epoch": 5.614852905273437e-05,
      "step": 58876,
      "training_step_time": 7.651595592498779
    },
    {
      "epoch": 5.6149482727050784e-05,
      "model_forward_time": 1.7562286853790283,
      "step": 58877
    },
    {
      "epoch": 5.6149482727050784e-05,
      "step": 58877,
      "training_step_time": 7.647420406341553
    },
    {
      "epoch": 5.615043640136719e-05,
      "model_forward_time": 1.759592056274414,
      "step": 58878
    },
    {
      "epoch": 5.615043640136719e-05,
      "step": 58878,
      "training_step_time": 7.5484747886657715
    },
    {
      "epoch": 5.615139007568359e-05,
      "model_forward_time": 1.7582459449768066,
      "step": 58879
    },
    {
      "epoch": 5.615139007568359e-05,
      "step": 58879,
      "training_step_time": 7.542806148529053
    },
    {
      "epoch": 5.615234375e-05,
      "grad_norm": 0.10461971163749695,
      "learning_rate": 3.953056892722368e-06,
      "loss": 0.0565,
      "step": 58880
    },
    {
      "epoch": 5.615234375e-05,
      "model_forward_time": 1.754077434539795,
      "step": 58880
    },
    {
      "dynamics_loss_avg": 0.052066237293183805,
      "epoch": 5.615234375e-05,
      "step": 58880
    },
    {
      "action_loss_avg": 0.001095050921139773,
      "epoch": 5.615234375e-05,
      "step": 58880
    },
    {
      "epoch": 5.615234375e-05,
      "step": 58880,
      "training_step_time": 8.693516492843628
    },
    {
      "epoch": 5.6153297424316406e-05,
      "model_forward_time": 1.7605695724487305,
      "step": 58881
    },
    {
      "epoch": 5.6153297424316406e-05,
      "step": 58881,
      "training_step_time": 7.653635740280151
    },
    {
      "epoch": 5.615425109863281e-05,
      "model_forward_time": 1.390897512435913,
      "step": 58882
    },
    {
      "epoch": 5.615425109863281e-05,
      "step": 58882,
      "training_step_time": 8.627827405929565
    },
    {
      "epoch": 5.615520477294922e-05,
      "model_forward_time": 2.942599296569824,
      "step": 58883
    },
    {
      "epoch": 5.615520477294922e-05,
      "step": 58883,
      "training_step_time": 8.502274990081787
    },
    {
      "epoch": 5.6156158447265625e-05,
      "model_forward_time": 1.7678053379058838,
      "step": 58884
    },
    {
      "epoch": 5.6156158447265625e-05,
      "step": 58884,
      "training_step_time": 7.621911287307739
    },
    {
      "epoch": 5.615711212158203e-05,
      "model_forward_time": 1.757901668548584,
      "step": 58885
    },
    {
      "epoch": 5.615711212158203e-05,
      "step": 58885,
      "training_step_time": 8.262406826019287
    },
    {
      "epoch": 5.615806579589844e-05,
      "model_forward_time": 1.766024112701416,
      "step": 58886
    },
    {
      "epoch": 5.615806579589844e-05,
      "step": 58886,
      "training_step_time": 8.30565071105957
    },
    {
      "epoch": 5.6159019470214844e-05,
      "model_forward_time": 1.7616360187530518,
      "step": 58887
    },
    {
      "epoch": 5.6159019470214844e-05,
      "step": 58887,
      "training_step_time": 7.5786967277526855
    },
    {
      "epoch": 5.615997314453125e-05,
      "model_forward_time": 1.7626936435699463,
      "step": 58888
    },
    {
      "epoch": 5.615997314453125e-05,
      "step": 58888,
      "training_step_time": 7.487664222717285
    },
    {
      "epoch": 5.616092681884766e-05,
      "model_forward_time": 1.7562451362609863,
      "step": 58889
    },
    {
      "epoch": 5.616092681884766e-05,
      "step": 58889,
      "training_step_time": 7.771641731262207
    },
    {
      "epoch": 5.616188049316406e-05,
      "grad_norm": 0.10916198790073395,
      "learning_rate": 3.95144013346636e-06,
      "loss": 0.0581,
      "step": 58890
    },
    {
      "epoch": 5.616188049316406e-05,
      "model_forward_time": 1.75968337059021,
      "step": 58890
    },
    {
      "dynamics_loss_avg": 0.056211380660533904,
      "epoch": 5.616188049316406e-05,
      "step": 58890
    },
    {
      "action_loss_avg": 0.0009897277952404693,
      "epoch": 5.616188049316406e-05,
      "step": 58890
    },
    {
      "epoch": 5.616188049316406e-05,
      "step": 58890,
      "training_step_time": 7.604210138320923
    },
    {
      "epoch": 5.616283416748047e-05,
      "model_forward_time": 1.756197452545166,
      "step": 58891
    },
    {
      "epoch": 5.616283416748047e-05,
      "step": 58891,
      "training_step_time": 8.385508298873901
    },
    {
      "epoch": 5.616378784179688e-05,
      "model_forward_time": 2.9852967262268066,
      "step": 58892
    },
    {
      "epoch": 5.616378784179688e-05,
      "step": 58892,
      "training_step_time": 9.096884727478027
    },
    {
      "epoch": 5.616474151611328e-05,
      "model_forward_time": 1.7598292827606201,
      "step": 58893
    },
    {
      "epoch": 5.616474151611328e-05,
      "step": 58893,
      "training_step_time": 7.616816997528076
    },
    {
      "epoch": 5.6165695190429686e-05,
      "model_forward_time": 1.7648909091949463,
      "step": 58894
    },
    {
      "epoch": 5.6165695190429686e-05,
      "step": 58894,
      "training_step_time": 8.256285190582275
    },
    {
      "epoch": 5.61666488647461e-05,
      "model_forward_time": 1.762535810470581,
      "step": 58895
    },
    {
      "epoch": 5.61666488647461e-05,
      "step": 58895,
      "training_step_time": 8.232807874679565
    },
    {
      "epoch": 5.61676025390625e-05,
      "model_forward_time": 1.7672162055969238,
      "step": 58896
    },
    {
      "epoch": 5.61676025390625e-05,
      "step": 58896,
      "training_step_time": 7.651028156280518
    },
    {
      "epoch": 5.6168556213378905e-05,
      "model_forward_time": 1.7646222114562988,
      "step": 58897
    },
    {
      "epoch": 5.6168556213378905e-05,
      "step": 58897,
      "training_step_time": 7.369623899459839
    },
    {
      "epoch": 5.6169509887695316e-05,
      "model_forward_time": 1.7580156326293945,
      "step": 58898
    },
    {
      "epoch": 5.6169509887695316e-05,
      "step": 58898,
      "training_step_time": 8.36332392692566
    },
    {
      "epoch": 5.617046356201172e-05,
      "model_forward_time": 1.76434326171875,
      "step": 58899
    },
    {
      "epoch": 5.617046356201172e-05,
      "step": 58899,
      "training_step_time": 8.175330877304077
    },
    {
      "epoch": 5.6171417236328124e-05,
      "grad_norm": 0.09191547334194183,
      "learning_rate": 3.9498234888792815e-06,
      "loss": 0.0554,
      "step": 58900
    },
    {
      "epoch": 5.6171417236328124e-05,
      "model_forward_time": 1.750112533569336,
      "step": 58900
    },
    {
      "dynamics_loss_avg": 0.05918895583599806,
      "epoch": 5.6171417236328124e-05,
      "step": 58900
    },
    {
      "action_loss_avg": 0.002307650278089568,
      "epoch": 5.6171417236328124e-05,
      "step": 58900
    },
    {
      "epoch": 5.6171417236328124e-05,
      "step": 58900,
      "training_step_time": 8.509588956832886
    },
    {
      "epoch": 5.6172370910644534e-05,
      "model_forward_time": 1.865682601928711,
      "step": 58901
    },
    {
      "epoch": 5.6172370910644534e-05,
      "step": 58901,
      "training_step_time": 9.299927711486816
    },
    {
      "epoch": 5.617332458496094e-05,
      "model_forward_time": 1.7481915950775146,
      "step": 58902
    },
    {
      "epoch": 5.617332458496094e-05,
      "step": 58902,
      "training_step_time": 7.617712497711182
    },
    {
      "epoch": 5.617427825927734e-05,
      "model_forward_time": 1.7560596466064453,
      "step": 58903
    },
    {
      "epoch": 5.617427825927734e-05,
      "step": 58903,
      "training_step_time": 8.55077314376831
    },
    {
      "epoch": 5.617523193359375e-05,
      "model_forward_time": 1.7554759979248047,
      "step": 58904
    },
    {
      "epoch": 5.617523193359375e-05,
      "step": 58904,
      "training_step_time": 7.482038497924805
    },
    {
      "epoch": 5.617618560791016e-05,
      "model_forward_time": 1.758392095565796,
      "step": 58905
    },
    {
      "epoch": 5.617618560791016e-05,
      "step": 58905,
      "training_step_time": 8.63937497138977
    },
    {
      "epoch": 5.617713928222656e-05,
      "model_forward_time": 1.7576696872711182,
      "step": 58906
    },
    {
      "epoch": 5.617713928222656e-05,
      "step": 58906,
      "training_step_time": 7.4596147537231445
    },
    {
      "epoch": 5.6178092956542965e-05,
      "model_forward_time": 1.7528791427612305,
      "step": 58907
    },
    {
      "epoch": 5.6178092956542965e-05,
      "step": 58907,
      "training_step_time": 7.798807382583618
    },
    {
      "epoch": 5.6179046630859376e-05,
      "model_forward_time": 1.9064624309539795,
      "step": 58908
    },
    {
      "epoch": 5.6179046630859376e-05,
      "step": 58908,
      "training_step_time": 8.283963203430176
    },
    {
      "epoch": 5.618000030517578e-05,
      "model_forward_time": 1.9423081874847412,
      "step": 58909
    },
    {
      "epoch": 5.618000030517578e-05,
      "step": 58909,
      "training_step_time": 8.656556129455566
    },
    {
      "epoch": 5.6180953979492184e-05,
      "grad_norm": 0.08824387192726135,
      "learning_rate": 3.9482069591379276e-06,
      "loss": 0.0535,
      "step": 58910
    },
    {
      "epoch": 5.6180953979492184e-05,
      "model_forward_time": 1.9268035888671875,
      "step": 58910
    },
    {
      "dynamics_loss_avg": 0.04191121030598879,
      "epoch": 5.6180953979492184e-05,
      "step": 58910
    },
    {
      "action_loss_avg": 0.0013361237943172455,
      "epoch": 5.6180953979492184e-05,
      "step": 58910
    },
    {
      "epoch": 5.6180953979492184e-05,
      "step": 58910,
      "training_step_time": 8.129882335662842
    },
    {
      "epoch": 5.6181907653808595e-05,
      "model_forward_time": 3.049821138381958,
      "step": 58911
    },
    {
      "epoch": 5.6181907653808595e-05,
      "step": 58911,
      "training_step_time": 9.968108177185059
    },
    {
      "epoch": 5.6182861328125e-05,
      "model_forward_time": 1.9057371616363525,
      "step": 58912
    },
    {
      "epoch": 5.6182861328125e-05,
      "step": 58912,
      "training_step_time": 7.563535928726196
    },
    {
      "epoch": 5.61838150024414e-05,
      "model_forward_time": 1.899559497833252,
      "step": 58913
    },
    {
      "epoch": 5.61838150024414e-05,
      "step": 58913,
      "training_step_time": 9.025770664215088
    },
    {
      "epoch": 5.6184768676757814e-05,
      "model_forward_time": 1.8931891918182373,
      "step": 58914
    },
    {
      "epoch": 5.6184768676757814e-05,
      "step": 58914,
      "training_step_time": 7.638771295547485
    },
    {
      "epoch": 5.618572235107422e-05,
      "model_forward_time": 1.928884744644165,
      "step": 58915
    },
    {
      "epoch": 5.618572235107422e-05,
      "step": 58915,
      "training_step_time": 7.554373741149902
    },
    {
      "epoch": 5.618667602539062e-05,
      "model_forward_time": 1.9291572570800781,
      "step": 58916
    },
    {
      "epoch": 5.618667602539062e-05,
      "step": 58916,
      "training_step_time": 7.5607359409332275
    },
    {
      "epoch": 5.618762969970703e-05,
      "model_forward_time": 1.9032230377197266,
      "step": 58917
    },
    {
      "epoch": 5.618762969970703e-05,
      "step": 58917,
      "training_step_time": 7.783825874328613
    },
    {
      "epoch": 5.618858337402344e-05,
      "model_forward_time": 1.907275676727295,
      "step": 58918
    },
    {
      "epoch": 5.618858337402344e-05,
      "step": 58918,
      "training_step_time": 8.282931804656982
    },
    {
      "epoch": 5.618953704833984e-05,
      "model_forward_time": 1.9068937301635742,
      "step": 58919
    },
    {
      "epoch": 5.618953704833984e-05,
      "step": 58919,
      "training_step_time": 10.273147344589233
    },
    {
      "epoch": 5.619049072265625e-05,
      "grad_norm": 0.09826448559761047,
      "learning_rate": 3.946590544419079e-06,
      "loss": 0.0577,
      "step": 58920
    },
    {
      "epoch": 5.619049072265625e-05,
      "model_forward_time": 2.9949123859405518,
      "step": 58920
    },
    {
      "dynamics_loss_avg": 0.05348857864737511,
      "epoch": 5.619049072265625e-05,
      "step": 58920
    },
    {
      "action_loss_avg": 0.0015730354003608226,
      "epoch": 5.619049072265625e-05,
      "step": 58920
    },
    {
      "epoch": 5.619049072265625e-05,
      "step": 58920,
      "training_step_time": 8.44817328453064
    },
    {
      "epoch": 5.6191444396972656e-05,
      "model_forward_time": 1.9220428466796875,
      "step": 58921
    },
    {
      "epoch": 5.6191444396972656e-05,
      "step": 58921,
      "training_step_time": 8.199680805206299
    },
    {
      "epoch": 5.619239807128906e-05,
      "model_forward_time": 1.8960182666778564,
      "step": 58922
    },
    {
      "epoch": 5.619239807128906e-05,
      "step": 58922,
      "training_step_time": 8.629150152206421
    },
    {
      "epoch": 5.619335174560547e-05,
      "model_forward_time": 1.9283688068389893,
      "step": 58923
    },
    {
      "epoch": 5.619335174560547e-05,
      "step": 58923,
      "training_step_time": 7.6429150104522705
    },
    {
      "epoch": 5.6194305419921875e-05,
      "model_forward_time": 1.9064579010009766,
      "step": 58924
    },
    {
      "epoch": 5.6194305419921875e-05,
      "step": 58924,
      "training_step_time": 7.465161561965942
    },
    {
      "epoch": 5.619525909423828e-05,
      "model_forward_time": 1.8989787101745605,
      "step": 58925
    },
    {
      "epoch": 5.619525909423828e-05,
      "step": 58925,
      "training_step_time": 7.543444633483887
    },
    {
      "epoch": 5.619621276855469e-05,
      "model_forward_time": 1.9187378883361816,
      "step": 58926
    },
    {
      "epoch": 5.619621276855469e-05,
      "step": 58926,
      "training_step_time": 7.726794004440308
    },
    {
      "epoch": 5.619716644287109e-05,
      "model_forward_time": 1.9398188591003418,
      "step": 58927
    },
    {
      "epoch": 5.619716644287109e-05,
      "step": 58927,
      "training_step_time": 8.176246643066406
    },
    {
      "epoch": 5.61981201171875e-05,
      "model_forward_time": 1.9022142887115479,
      "step": 58928
    },
    {
      "epoch": 5.61981201171875e-05,
      "step": 58928,
      "training_step_time": 7.634260892868042
    },
    {
      "epoch": 5.619907379150391e-05,
      "model_forward_time": 2.9625139236450195,
      "step": 58929
    },
    {
      "epoch": 5.619907379150391e-05,
      "step": 58929,
      "training_step_time": 7.7000133991241455
    },
    {
      "epoch": 5.620002746582031e-05,
      "grad_norm": 0.09028896689414978,
      "learning_rate": 3.9449742448995055e-06,
      "loss": 0.0569,
      "step": 58930
    },
    {
      "epoch": 5.620002746582031e-05,
      "model_forward_time": 1.9197371006011963,
      "step": 58930
    },
    {
      "dynamics_loss_avg": 0.05119416806846857,
      "epoch": 5.620002746582031e-05,
      "step": 58930
    },
    {
      "action_loss_avg": 0.0016446415596874432,
      "epoch": 5.620002746582031e-05,
      "step": 58930
    },
    {
      "epoch": 5.620002746582031e-05,
      "step": 58930,
      "training_step_time": 8.149826049804688
    },
    {
      "epoch": 5.6200981140136716e-05,
      "model_forward_time": 1.9079360961914062,
      "step": 58931
    },
    {
      "epoch": 5.6200981140136716e-05,
      "step": 58931,
      "training_step_time": 7.625601768493652
    },
    {
      "epoch": 5.620193481445313e-05,
      "model_forward_time": 1.919348955154419,
      "step": 58932
    },
    {
      "epoch": 5.620193481445313e-05,
      "step": 58932,
      "training_step_time": 7.621636390686035
    },
    {
      "epoch": 5.620288848876953e-05,
      "model_forward_time": 1.9214470386505127,
      "step": 58933
    },
    {
      "epoch": 5.620288848876953e-05,
      "step": 58933,
      "training_step_time": 8.655597925186157
    },
    {
      "epoch": 5.6203842163085935e-05,
      "model_forward_time": 1.9202179908752441,
      "step": 58934
    },
    {
      "epoch": 5.6203842163085935e-05,
      "step": 58934,
      "training_step_time": 8.240797519683838
    },
    {
      "epoch": 5.6204795837402346e-05,
      "model_forward_time": 1.8857519626617432,
      "step": 58935
    },
    {
      "epoch": 5.6204795837402346e-05,
      "step": 58935,
      "training_step_time": 7.698740482330322
    },
    {
      "epoch": 5.620574951171875e-05,
      "model_forward_time": 1.550567388534546,
      "step": 58936
    },
    {
      "epoch": 5.620574951171875e-05,
      "step": 58936,
      "training_step_time": 7.733549356460571
    },
    {
      "epoch": 5.6206703186035154e-05,
      "model_forward_time": 1.942903995513916,
      "step": 58937
    },
    {
      "epoch": 5.6206703186035154e-05,
      "step": 58937,
      "training_step_time": 8.216336488723755
    },
    {
      "epoch": 5.6207656860351565e-05,
      "model_forward_time": 3.0427095890045166,
      "step": 58938
    },
    {
      "epoch": 5.6207656860351565e-05,
      "step": 58938,
      "training_step_time": 8.339405298233032
    },
    {
      "epoch": 5.620861053466797e-05,
      "model_forward_time": 1.9291326999664307,
      "step": 58939
    },
    {
      "epoch": 5.620861053466797e-05,
      "step": 58939,
      "training_step_time": 8.669443130493164
    },
    {
      "epoch": 5.620956420898437e-05,
      "grad_norm": 0.0798756405711174,
      "learning_rate": 3.943358060755962e-06,
      "loss": 0.0552,
      "step": 58940
    },
    {
      "epoch": 5.620956420898437e-05,
      "model_forward_time": 1.903045892715454,
      "step": 58940
    },
    {
      "dynamics_loss_avg": 0.046090795285999776,
      "epoch": 5.620956420898437e-05,
      "step": 58940
    },
    {
      "action_loss_avg": 0.0015991920983651653,
      "epoch": 5.620956420898437e-05,
      "step": 58940
    },
    {
      "epoch": 5.620956420898437e-05,
      "step": 58940,
      "training_step_time": 9.332987546920776
    },
    {
      "epoch": 5.6210517883300784e-05,
      "model_forward_time": 1.889237880706787,
      "step": 58941
    },
    {
      "epoch": 5.6210517883300784e-05,
      "step": 58941,
      "training_step_time": 8.495977401733398
    },
    {
      "epoch": 5.621147155761719e-05,
      "model_forward_time": 1.8876473903656006,
      "step": 58942
    },
    {
      "epoch": 5.621147155761719e-05,
      "step": 58942,
      "training_step_time": 7.63386344909668
    },
    {
      "epoch": 5.621242523193359e-05,
      "model_forward_time": 1.7631964683532715,
      "step": 58943
    },
    {
      "epoch": 5.621242523193359e-05,
      "step": 58943,
      "training_step_time": 7.602726936340332
    },
    {
      "epoch": 5.621337890625e-05,
      "model_forward_time": 1.7613890171051025,
      "step": 58944
    },
    {
      "epoch": 5.621337890625e-05,
      "step": 58944,
      "training_step_time": 8.867340326309204
    },
    {
      "epoch": 5.6214332580566407e-05,
      "model_forward_time": 1.7474007606506348,
      "step": 58945
    },
    {
      "epoch": 5.6214332580566407e-05,
      "step": 58945,
      "training_step_time": 8.352728843688965
    },
    {
      "epoch": 5.621528625488281e-05,
      "model_forward_time": 1.774561882019043,
      "step": 58946
    },
    {
      "epoch": 5.621528625488281e-05,
      "step": 58946,
      "training_step_time": 7.612247467041016
    },
    {
      "epoch": 5.621623992919922e-05,
      "model_forward_time": 1.8761746883392334,
      "step": 58947
    },
    {
      "epoch": 5.621623992919922e-05,
      "step": 58947,
      "training_step_time": 10.395253419876099
    },
    {
      "epoch": 5.6217193603515625e-05,
      "model_forward_time": 1.7533743381500244,
      "step": 58948
    },
    {
      "epoch": 5.6217193603515625e-05,
      "step": 58948,
      "training_step_time": 9.283130645751953
    },
    {
      "epoch": 5.621814727783203e-05,
      "model_forward_time": 1.7645378112792969,
      "step": 58949
    },
    {
      "epoch": 5.621814727783203e-05,
      "step": 58949,
      "training_step_time": 7.6001036167144775
    },
    {
      "epoch": 5.621910095214844e-05,
      "grad_norm": 0.12249651551246643,
      "learning_rate": 3.941741992165193e-06,
      "loss": 0.058,
      "step": 58950
    },
    {
      "epoch": 5.621910095214844e-05,
      "model_forward_time": 1.394991397857666,
      "step": 58950
    },
    {
      "dynamics_loss_avg": 0.059830735065042974,
      "epoch": 5.621910095214844e-05,
      "step": 58950
    },
    {
      "action_loss_avg": 0.0018270269414642827,
      "epoch": 5.621910095214844e-05,
      "step": 58950
    },
    {
      "epoch": 5.621910095214844e-05,
      "step": 58950,
      "training_step_time": 8.130094528198242
    },
    {
      "epoch": 5.6220054626464844e-05,
      "model_forward_time": 1.7480056285858154,
      "step": 58951
    },
    {
      "epoch": 5.6220054626464844e-05,
      "step": 58951,
      "training_step_time": 7.6371910572052
    },
    {
      "epoch": 5.622100830078125e-05,
      "model_forward_time": 1.7625010013580322,
      "step": 58952
    },
    {
      "epoch": 5.622100830078125e-05,
      "step": 58952,
      "training_step_time": 7.566072940826416
    },
    {
      "epoch": 5.622196197509766e-05,
      "model_forward_time": 1.7578890323638916,
      "step": 58953
    },
    {
      "epoch": 5.622196197509766e-05,
      "step": 58953,
      "training_step_time": 7.866898775100708
    },
    {
      "epoch": 5.622291564941406e-05,
      "model_forward_time": 1.7503445148468018,
      "step": 58954
    },
    {
      "epoch": 5.622291564941406e-05,
      "step": 58954,
      "training_step_time": 7.5518786907196045
    },
    {
      "epoch": 5.622386932373047e-05,
      "model_forward_time": 1.7659516334533691,
      "step": 58955
    },
    {
      "epoch": 5.622386932373047e-05,
      "step": 58955,
      "training_step_time": 7.613229513168335
    },
    {
      "epoch": 5.622482299804688e-05,
      "model_forward_time": 1.8448748588562012,
      "step": 58956
    },
    {
      "epoch": 5.622482299804688e-05,
      "step": 58956,
      "training_step_time": 7.913017272949219
    },
    {
      "epoch": 5.622577667236328e-05,
      "model_forward_time": 2.9233386516571045,
      "step": 58957
    },
    {
      "epoch": 5.622577667236328e-05,
      "step": 58957,
      "training_step_time": 8.642905712127686
    },
    {
      "epoch": 5.6226730346679686e-05,
      "model_forward_time": 1.7578237056732178,
      "step": 58958
    },
    {
      "epoch": 5.6226730346679686e-05,
      "step": 58958,
      "training_step_time": 8.281249046325684
    },
    {
      "epoch": 5.62276840209961e-05,
      "model_forward_time": 1.751206874847412,
      "step": 58959
    },
    {
      "epoch": 5.62276840209961e-05,
      "step": 58959,
      "training_step_time": 7.625786542892456
    },
    {
      "epoch": 5.62286376953125e-05,
      "grad_norm": 0.11715030670166016,
      "learning_rate": 3.940126039303928e-06,
      "loss": 0.0553,
      "step": 58960
    },
    {
      "epoch": 5.62286376953125e-05,
      "model_forward_time": 1.4069263935089111,
      "step": 58960
    },
    {
      "dynamics_loss_avg": 0.058313990198075774,
      "epoch": 5.62286376953125e-05,
      "step": 58960
    },
    {
      "action_loss_avg": 0.0014616309592383914,
      "epoch": 5.62286376953125e-05,
      "step": 58960
    },
    {
      "epoch": 5.62286376953125e-05,
      "step": 58960,
      "training_step_time": 7.640176296234131
    },
    {
      "epoch": 5.6229591369628905e-05,
      "model_forward_time": 1.7531299591064453,
      "step": 58961
    },
    {
      "epoch": 5.6229591369628905e-05,
      "step": 58961,
      "training_step_time": 8.697962522506714
    },
    {
      "epoch": 5.6230545043945316e-05,
      "model_forward_time": 1.750347375869751,
      "step": 58962
    },
    {
      "epoch": 5.6230545043945316e-05,
      "step": 58962,
      "training_step_time": 7.9287707805633545
    },
    {
      "epoch": 5.623149871826172e-05,
      "model_forward_time": 1.7493531703948975,
      "step": 58963
    },
    {
      "epoch": 5.623149871826172e-05,
      "step": 58963,
      "training_step_time": 7.766176462173462
    },
    {
      "epoch": 5.6232452392578124e-05,
      "model_forward_time": 1.759610652923584,
      "step": 58964
    },
    {
      "epoch": 5.6232452392578124e-05,
      "step": 58964,
      "training_step_time": 7.762096643447876
    },
    {
      "epoch": 5.6233406066894535e-05,
      "model_forward_time": 1.821122169494629,
      "step": 58965
    },
    {
      "epoch": 5.6233406066894535e-05,
      "step": 58965,
      "training_step_time": 8.742188453674316
    },
    {
      "epoch": 5.623435974121094e-05,
      "model_forward_time": 2.9145467281341553,
      "step": 58966
    },
    {
      "epoch": 5.623435974121094e-05,
      "step": 58966,
      "training_step_time": 8.862901210784912
    },
    {
      "epoch": 5.623531341552734e-05,
      "model_forward_time": 1.7683689594268799,
      "step": 58967
    },
    {
      "epoch": 5.623531341552734e-05,
      "step": 58967,
      "training_step_time": 8.672656297683716
    },
    {
      "epoch": 5.623626708984375e-05,
      "model_forward_time": 1.7615180015563965,
      "step": 58968
    },
    {
      "epoch": 5.623626708984375e-05,
      "step": 58968,
      "training_step_time": 8.734381914138794
    },
    {
      "epoch": 5.623722076416016e-05,
      "model_forward_time": 1.760577917098999,
      "step": 58969
    },
    {
      "epoch": 5.623722076416016e-05,
      "step": 58969,
      "training_step_time": 7.727774620056152
    },
    {
      "epoch": 5.623817443847656e-05,
      "grad_norm": 0.0843866690993309,
      "learning_rate": 3.938510202348886e-06,
      "loss": 0.0558,
      "step": 58970
    },
    {
      "epoch": 5.623817443847656e-05,
      "model_forward_time": 1.7505316734313965,
      "step": 58970
    },
    {
      "dynamics_loss_avg": 0.044110262766480446,
      "epoch": 5.623817443847656e-05,
      "step": 58970
    },
    {
      "action_loss_avg": 0.001006283052265644,
      "epoch": 5.623817443847656e-05,
      "step": 58970
    },
    {
      "epoch": 5.623817443847656e-05,
      "step": 58970,
      "training_step_time": 7.648963212966919
    },
    {
      "epoch": 5.6239128112792965e-05,
      "model_forward_time": 1.7621381282806396,
      "step": 58971
    },
    {
      "epoch": 5.6239128112792965e-05,
      "step": 58971,
      "training_step_time": 8.378102779388428
    },
    {
      "epoch": 5.6240081787109376e-05,
      "model_forward_time": 1.757817029953003,
      "step": 58972
    },
    {
      "epoch": 5.6240081787109376e-05,
      "step": 58972,
      "training_step_time": 8.604571104049683
    },
    {
      "epoch": 5.624103546142578e-05,
      "model_forward_time": 1.7431981563568115,
      "step": 58973
    },
    {
      "epoch": 5.624103546142578e-05,
      "step": 58973,
      "training_step_time": 8.554918050765991
    },
    {
      "epoch": 5.6241989135742184e-05,
      "model_forward_time": 1.7803187370300293,
      "step": 58974
    },
    {
      "epoch": 5.6241989135742184e-05,
      "step": 58974,
      "training_step_time": 8.648799180984497
    },
    {
      "epoch": 5.6242942810058595e-05,
      "model_forward_time": 2.875627040863037,
      "step": 58975
    },
    {
      "epoch": 5.6242942810058595e-05,
      "step": 58975,
      "training_step_time": 9.131038665771484
    },
    {
      "epoch": 5.6243896484375e-05,
      "model_forward_time": 1.7648015022277832,
      "step": 58976
    },
    {
      "epoch": 5.6243896484375e-05,
      "step": 58976,
      "training_step_time": 7.670461893081665
    },
    {
      "epoch": 5.62448501586914e-05,
      "model_forward_time": 1.7699096202850342,
      "step": 58977
    },
    {
      "epoch": 5.62448501586914e-05,
      "step": 58977,
      "training_step_time": 8.166267395019531
    },
    {
      "epoch": 5.6245803833007814e-05,
      "model_forward_time": 1.75260329246521,
      "step": 58978
    },
    {
      "epoch": 5.6245803833007814e-05,
      "step": 58978,
      "training_step_time": 7.5969579219818115
    },
    {
      "epoch": 5.624675750732422e-05,
      "model_forward_time": 1.7690227031707764,
      "step": 58979
    },
    {
      "epoch": 5.624675750732422e-05,
      "step": 58979,
      "training_step_time": 7.539168119430542
    },
    {
      "epoch": 5.624771118164062e-05,
      "grad_norm": 0.1339944303035736,
      "learning_rate": 3.9368944814767704e-06,
      "loss": 0.0564,
      "step": 58980
    },
    {
      "epoch": 5.624771118164062e-05,
      "model_forward_time": 1.7537546157836914,
      "step": 58980
    },
    {
      "dynamics_loss_avg": 0.04616710059344768,
      "epoch": 5.624771118164062e-05,
      "step": 58980
    },
    {
      "action_loss_avg": 0.0009280923928599805,
      "epoch": 5.624771118164062e-05,
      "step": 58980
    },
    {
      "epoch": 5.624771118164062e-05,
      "step": 58980,
      "training_step_time": 8.650930643081665
    },
    {
      "epoch": 5.624866485595703e-05,
      "model_forward_time": 1.742563009262085,
      "step": 58981
    },
    {
      "epoch": 5.624866485595703e-05,
      "step": 58981,
      "training_step_time": 7.719483137130737
    },
    {
      "epoch": 5.624961853027344e-05,
      "model_forward_time": 1.7293322086334229,
      "step": 58982
    },
    {
      "epoch": 5.624961853027344e-05,
      "step": 58982,
      "training_step_time": 7.7063257694244385
    },
    {
      "epoch": 5.625057220458984e-05,
      "model_forward_time": 1.7606074810028076,
      "step": 58983
    },
    {
      "epoch": 5.625057220458984e-05,
      "step": 58983,
      "training_step_time": 8.615444421768188
    },
    {
      "epoch": 5.625152587890625e-05,
      "model_forward_time": 1.766225814819336,
      "step": 58984
    },
    {
      "epoch": 5.625152587890625e-05,
      "step": 58984,
      "training_step_time": 8.92093014717102
    },
    {
      "epoch": 5.6252479553222656e-05,
      "model_forward_time": 2.9482626914978027,
      "step": 58985
    },
    {
      "epoch": 5.6252479553222656e-05,
      "step": 58985,
      "training_step_time": 7.697892665863037
    },
    {
      "epoch": 5.625343322753906e-05,
      "model_forward_time": 1.7563097476959229,
      "step": 58986
    },
    {
      "epoch": 5.625343322753906e-05,
      "step": 58986,
      "training_step_time": 7.589057207107544
    },
    {
      "epoch": 5.625438690185547e-05,
      "model_forward_time": 1.402799367904663,
      "step": 58987
    },
    {
      "epoch": 5.625438690185547e-05,
      "step": 58987,
      "training_step_time": 9.341015100479126
    },
    {
      "epoch": 5.6255340576171875e-05,
      "model_forward_time": 1.7482256889343262,
      "step": 58988
    },
    {
      "epoch": 5.6255340576171875e-05,
      "step": 58988,
      "training_step_time": 7.507882118225098
    },
    {
      "epoch": 5.625629425048828e-05,
      "model_forward_time": 1.7494440078735352,
      "step": 58989
    },
    {
      "epoch": 5.625629425048828e-05,
      "step": 58989,
      "training_step_time": 7.614446640014648
    },
    {
      "epoch": 5.625724792480469e-05,
      "grad_norm": 0.10547293722629547,
      "learning_rate": 3.935278876864278e-06,
      "loss": 0.0546,
      "step": 58990
    },
    {
      "epoch": 5.625724792480469e-05,
      "model_forward_time": 1.7434542179107666,
      "step": 58990
    },
    {
      "dynamics_loss_avg": 0.051061782985925674,
      "epoch": 5.625724792480469e-05,
      "step": 58990
    },
    {
      "action_loss_avg": 0.0016679476713761688,
      "epoch": 5.625724792480469e-05,
      "step": 58990
    },
    {
      "epoch": 5.625724792480469e-05,
      "step": 58990,
      "training_step_time": 7.709123849868774
    },
    {
      "epoch": 5.6258201599121093e-05,
      "model_forward_time": 1.750216007232666,
      "step": 58991
    },
    {
      "epoch": 5.6258201599121093e-05,
      "step": 58991,
      "training_step_time": 7.631283521652222
    },
    {
      "epoch": 5.62591552734375e-05,
      "model_forward_time": 1.7443695068359375,
      "step": 58992
    },
    {
      "epoch": 5.62591552734375e-05,
      "step": 58992,
      "training_step_time": 7.637767553329468
    },
    {
      "epoch": 5.626010894775391e-05,
      "model_forward_time": 1.8802096843719482,
      "step": 58993
    },
    {
      "epoch": 5.626010894775391e-05,
      "step": 58993,
      "training_step_time": 7.809493064880371
    },
    {
      "epoch": 5.626106262207031e-05,
      "model_forward_time": 2.908984422683716,
      "step": 58994
    },
    {
      "epoch": 5.626106262207031e-05,
      "step": 58994,
      "training_step_time": 9.603808164596558
    },
    {
      "epoch": 5.6262016296386716e-05,
      "model_forward_time": 1.762061357498169,
      "step": 58995
    },
    {
      "epoch": 5.6262016296386716e-05,
      "step": 58995,
      "training_step_time": 8.596283197402954
    },
    {
      "epoch": 5.626296997070313e-05,
      "model_forward_time": 1.745581865310669,
      "step": 58996
    },
    {
      "epoch": 5.626296997070313e-05,
      "step": 58996,
      "training_step_time": 7.709903955459595
    },
    {
      "epoch": 5.626392364501953e-05,
      "model_forward_time": 1.7337689399719238,
      "step": 58997
    },
    {
      "epoch": 5.626392364501953e-05,
      "step": 58997,
      "training_step_time": 7.537214756011963
    },
    {
      "epoch": 5.6264877319335935e-05,
      "model_forward_time": 1.4454011917114258,
      "step": 58998
    },
    {
      "epoch": 5.6264877319335935e-05,
      "step": 58998,
      "training_step_time": 9.911810874938965
    },
    {
      "epoch": 5.6265830993652346e-05,
      "model_forward_time": 1.7651207447052002,
      "step": 58999
    },
    {
      "epoch": 5.6265830993652346e-05,
      "step": 58999,
      "training_step_time": 7.988439083099365
    },
    {
      "epoch": 5.626678466796875e-05,
      "grad_norm": 0.08520246297121048,
      "learning_rate": 3.933663388688087e-06,
      "loss": 0.053,
      "step": 59000
    },
    {
      "epoch": 5.626678466796875e-05,
      "model_forward_time": 1.8604071140289307,
      "step": 59000
    },
    {
      "dynamics_loss_avg": 0.048146112635731694,
      "epoch": 5.626678466796875e-05,
      "step": 59000
    },
    {
      "action_loss_avg": 0.0011747537850169466,
      "epoch": 5.626678466796875e-05,
      "step": 59000
    },
    {
      "epoch": 5.626678466796875e-05,
      "step": 59000,
      "training_step_time": 6.778822660446167
    },
    {
      "epoch": 5.6267738342285154e-05,
      "model_forward_time": 1.7477443218231201,
      "step": 59001
    },
    {
      "epoch": 5.6267738342285154e-05,
      "step": 59001,
      "training_step_time": 8.866186618804932
    },
    {
      "epoch": 5.6268692016601565e-05,
      "model_forward_time": 3.4611074924468994,
      "step": 59002
    },
    {
      "epoch": 5.6268692016601565e-05,
      "step": 59002,
      "training_step_time": 8.412379503250122
    },
    {
      "epoch": 5.626964569091797e-05,
      "model_forward_time": 1.7491700649261475,
      "step": 59003
    },
    {
      "epoch": 5.626964569091797e-05,
      "step": 59003,
      "training_step_time": 10.478508472442627
    },
    {
      "epoch": 5.627059936523437e-05,
      "model_forward_time": 1.7541742324829102,
      "step": 59004
    },
    {
      "epoch": 5.627059936523437e-05,
      "step": 59004,
      "training_step_time": 8.364166975021362
    },
    {
      "epoch": 5.6271553039550784e-05,
      "model_forward_time": 1.741011381149292,
      "step": 59005
    },
    {
      "epoch": 5.6271553039550784e-05,
      "step": 59005,
      "training_step_time": 9.478493452072144
    },
    {
      "epoch": 5.627250671386719e-05,
      "model_forward_time": 1.7584476470947266,
      "step": 59006
    },
    {
      "epoch": 5.627250671386719e-05,
      "step": 59006,
      "training_step_time": 7.716057062149048
    },
    {
      "epoch": 5.627346038818359e-05,
      "model_forward_time": 1.7539024353027344,
      "step": 59007
    },
    {
      "epoch": 5.627346038818359e-05,
      "step": 59007,
      "training_step_time": 7.758732080459595
    },
    {
      "epoch": 5.62744140625e-05,
      "model_forward_time": 1.7538807392120361,
      "step": 59008
    },
    {
      "epoch": 5.62744140625e-05,
      "step": 59008,
      "training_step_time": 9.064188957214355
    },
    {
      "epoch": 5.6275367736816407e-05,
      "model_forward_time": 1.7481355667114258,
      "step": 59009
    },
    {
      "epoch": 5.6275367736816407e-05,
      "step": 59009,
      "training_step_time": 7.762596845626831
    },
    {
      "epoch": 5.627632141113281e-05,
      "grad_norm": 0.10209394991397858,
      "learning_rate": 3.932048017124865e-06,
      "loss": 0.0585,
      "step": 59010
    },
    {
      "epoch": 5.627632141113281e-05,
      "model_forward_time": 1.7351582050323486,
      "step": 59010
    },
    {
      "dynamics_loss_avg": 0.043835073709487915,
      "epoch": 5.627632141113281e-05,
      "step": 59010
    },
    {
      "action_loss_avg": 0.0012932227400597185,
      "epoch": 5.627632141113281e-05,
      "step": 59010
    },
    {
      "epoch": 5.627632141113281e-05,
      "step": 59010,
      "training_step_time": 7.92968487739563
    },
    {
      "epoch": 5.627727508544922e-05,
      "model_forward_time": 1.8853709697723389,
      "step": 59011
    },
    {
      "epoch": 5.627727508544922e-05,
      "step": 59011,
      "training_step_time": 7.93772292137146
    },
    {
      "epoch": 5.6278228759765625e-05,
      "model_forward_time": 2.9780008792877197,
      "step": 59012
    },
    {
      "epoch": 5.6278228759765625e-05,
      "step": 59012,
      "training_step_time": 7.923423528671265
    },
    {
      "epoch": 5.627918243408203e-05,
      "model_forward_time": 1.7494239807128906,
      "step": 59013
    },
    {
      "epoch": 5.627918243408203e-05,
      "step": 59013,
      "training_step_time": 7.817652940750122
    },
    {
      "epoch": 5.628013610839844e-05,
      "model_forward_time": 1.7464640140533447,
      "step": 59014
    },
    {
      "epoch": 5.628013610839844e-05,
      "step": 59014,
      "training_step_time": 7.770848751068115
    },
    {
      "epoch": 5.6281089782714844e-05,
      "model_forward_time": 1.7500195503234863,
      "step": 59015
    },
    {
      "epoch": 5.6281089782714844e-05,
      "step": 59015,
      "training_step_time": 7.844054460525513
    },
    {
      "epoch": 5.628204345703125e-05,
      "model_forward_time": 1.746894359588623,
      "step": 59016
    },
    {
      "epoch": 5.628204345703125e-05,
      "step": 59016,
      "training_step_time": 7.731643915176392
    },
    {
      "epoch": 5.628299713134766e-05,
      "model_forward_time": 1.7504229545593262,
      "step": 59017
    },
    {
      "epoch": 5.628299713134766e-05,
      "step": 59017,
      "training_step_time": 8.734649181365967
    },
    {
      "epoch": 5.628395080566406e-05,
      "model_forward_time": 1.7414453029632568,
      "step": 59018
    },
    {
      "epoch": 5.628395080566406e-05,
      "step": 59018,
      "training_step_time": 8.521438360214233
    },
    {
      "epoch": 5.628490447998047e-05,
      "model_forward_time": 1.750931978225708,
      "step": 59019
    },
    {
      "epoch": 5.628490447998047e-05,
      "step": 59019,
      "training_step_time": 8.921006441116333
    },
    {
      "epoch": 5.628585815429688e-05,
      "grad_norm": 0.1311468631029129,
      "learning_rate": 3.930432762351266e-06,
      "loss": 0.0582,
      "step": 59020
    },
    {
      "epoch": 5.628585815429688e-05,
      "model_forward_time": 1.8678479194641113,
      "step": 59020
    },
    {
      "dynamics_loss_avg": 0.05795518448576331,
      "epoch": 5.628585815429688e-05,
      "step": 59020
    },
    {
      "action_loss_avg": 0.0018222728569526226,
      "epoch": 5.628585815429688e-05,
      "step": 59020
    },
    {
      "epoch": 5.628585815429688e-05,
      "step": 59020,
      "training_step_time": 7.924895524978638
    },
    {
      "epoch": 5.628681182861328e-05,
      "model_forward_time": 2.9308629035949707,
      "step": 59021
    },
    {
      "epoch": 5.628681182861328e-05,
      "step": 59021,
      "training_step_time": 8.918268918991089
    },
    {
      "epoch": 5.6287765502929686e-05,
      "model_forward_time": 1.7509963512420654,
      "step": 59022
    },
    {
      "epoch": 5.6287765502929686e-05,
      "step": 59022,
      "training_step_time": 7.867305040359497
    },
    {
      "epoch": 5.62887191772461e-05,
      "model_forward_time": 1.7461082935333252,
      "step": 59023
    },
    {
      "epoch": 5.62887191772461e-05,
      "step": 59023,
      "training_step_time": 7.868130922317505
    },
    {
      "epoch": 5.62896728515625e-05,
      "model_forward_time": 1.750598669052124,
      "step": 59024
    },
    {
      "epoch": 5.62896728515625e-05,
      "step": 59024,
      "training_step_time": 7.870197057723999
    },
    {
      "epoch": 5.6290626525878905e-05,
      "model_forward_time": 1.7392020225524902,
      "step": 59025
    },
    {
      "epoch": 5.6290626525878905e-05,
      "step": 59025,
      "training_step_time": 9.06701135635376
    },
    {
      "epoch": 5.6291580200195316e-05,
      "model_forward_time": 1.7492625713348389,
      "step": 59026
    },
    {
      "epoch": 5.6291580200195316e-05,
      "step": 59026,
      "training_step_time": 7.860727548599243
    },
    {
      "epoch": 5.629253387451172e-05,
      "model_forward_time": 1.7538795471191406,
      "step": 59027
    },
    {
      "epoch": 5.629253387451172e-05,
      "step": 59027,
      "training_step_time": 7.937437534332275
    },
    {
      "epoch": 5.6293487548828124e-05,
      "model_forward_time": 1.7535960674285889,
      "step": 59028
    },
    {
      "epoch": 5.6293487548828124e-05,
      "step": 59028,
      "training_step_time": 7.916325092315674
    },
    {
      "epoch": 5.6294441223144535e-05,
      "model_forward_time": 1.8141224384307861,
      "step": 59029
    },
    {
      "epoch": 5.6294441223144535e-05,
      "step": 59029,
      "training_step_time": 7.811650276184082
    },
    {
      "epoch": 5.629539489746094e-05,
      "grad_norm": 0.09370801597833633,
      "learning_rate": 3.9288176245439315e-06,
      "loss": 0.0624,
      "step": 59030
    },
    {
      "epoch": 5.629539489746094e-05,
      "model_forward_time": 2.9100182056427,
      "step": 59030
    },
    {
      "dynamics_loss_avg": 0.05983286201953888,
      "epoch": 5.629539489746094e-05,
      "step": 59030
    },
    {
      "action_loss_avg": 0.0018292901280801744,
      "epoch": 5.629539489746094e-05,
      "step": 59030
    },
    {
      "epoch": 5.629539489746094e-05,
      "step": 59030,
      "training_step_time": 8.637013673782349
    },
    {
      "epoch": 5.629634857177734e-05,
      "model_forward_time": 1.7510993480682373,
      "step": 59031
    },
    {
      "epoch": 5.629634857177734e-05,
      "step": 59031,
      "training_step_time": 8.386393308639526
    },
    {
      "epoch": 5.629730224609375e-05,
      "model_forward_time": 1.7735354900360107,
      "step": 59032
    },
    {
      "epoch": 5.629730224609375e-05,
      "step": 59032,
      "training_step_time": 7.694877862930298
    },
    {
      "epoch": 5.629825592041016e-05,
      "model_forward_time": 1.7461051940917969,
      "step": 59033
    },
    {
      "epoch": 5.629825592041016e-05,
      "step": 59033,
      "training_step_time": 8.363197565078735
    },
    {
      "epoch": 5.629920959472656e-05,
      "model_forward_time": 1.3823168277740479,
      "step": 59034
    },
    {
      "epoch": 5.629920959472656e-05,
      "step": 59034,
      "training_step_time": 7.379542589187622
    },
    {
      "epoch": 5.6300163269042965e-05,
      "model_forward_time": 1.7404401302337646,
      "step": 59035
    },
    {
      "epoch": 5.6300163269042965e-05,
      "step": 59035,
      "training_step_time": 8.658643007278442
    },
    {
      "epoch": 5.6301116943359376e-05,
      "model_forward_time": 1.7460367679595947,
      "step": 59036
    },
    {
      "epoch": 5.6301116943359376e-05,
      "step": 59036,
      "training_step_time": 7.9619035720825195
    },
    {
      "epoch": 5.630207061767578e-05,
      "model_forward_time": 1.7509841918945312,
      "step": 59037
    },
    {
      "epoch": 5.630207061767578e-05,
      "step": 59037,
      "training_step_time": 7.917715311050415
    },
    {
      "epoch": 5.6303024291992184e-05,
      "model_forward_time": 1.8729872703552246,
      "step": 59038
    },
    {
      "epoch": 5.6303024291992184e-05,
      "step": 59038,
      "training_step_time": 8.952235698699951
    },
    {
      "epoch": 5.6303977966308595e-05,
      "model_forward_time": 2.897296667098999,
      "step": 59039
    },
    {
      "epoch": 5.6303977966308595e-05,
      "step": 59039,
      "training_step_time": 9.749424457550049
    },
    {
      "epoch": 5.6304931640625e-05,
      "grad_norm": 0.08774956315755844,
      "learning_rate": 3.927202603879488e-06,
      "loss": 0.0554,
      "step": 59040
    },
    {
      "epoch": 5.6304931640625e-05,
      "model_forward_time": 1.737877368927002,
      "step": 59040
    },
    {
      "dynamics_loss_avg": 0.04697376303374767,
      "epoch": 5.6304931640625e-05,
      "step": 59040
    },
    {
      "action_loss_avg": 0.0010655568446964025,
      "epoch": 5.6304931640625e-05,
      "step": 59040
    },
    {
      "epoch": 5.6304931640625e-05,
      "step": 59040,
      "training_step_time": 7.921815395355225
    },
    {
      "epoch": 5.63058853149414e-05,
      "model_forward_time": 1.746143102645874,
      "step": 59041
    },
    {
      "epoch": 5.63058853149414e-05,
      "step": 59041,
      "training_step_time": 7.79843807220459
    },
    {
      "epoch": 5.6306838989257814e-05,
      "model_forward_time": 1.7499732971191406,
      "step": 59042
    },
    {
      "epoch": 5.6306838989257814e-05,
      "step": 59042,
      "training_step_time": 7.850640535354614
    },
    {
      "epoch": 5.630779266357422e-05,
      "model_forward_time": 1.7517874240875244,
      "step": 59043
    },
    {
      "epoch": 5.630779266357422e-05,
      "step": 59043,
      "training_step_time": 8.637756824493408
    },
    {
      "epoch": 5.630874633789062e-05,
      "model_forward_time": 1.7466223239898682,
      "step": 59044
    },
    {
      "epoch": 5.630874633789062e-05,
      "step": 59044,
      "training_step_time": 8.621169328689575
    },
    {
      "epoch": 5.630970001220703e-05,
      "model_forward_time": 1.7483179569244385,
      "step": 59045
    },
    {
      "epoch": 5.630970001220703e-05,
      "step": 59045,
      "training_step_time": 7.036449909210205
    },
    {
      "epoch": 5.631065368652344e-05,
      "model_forward_time": 1.7466719150543213,
      "step": 59046
    },
    {
      "epoch": 5.631065368652344e-05,
      "step": 59046,
      "training_step_time": 7.03353476524353
    },
    {
      "epoch": 5.631160736083984e-05,
      "model_forward_time": 1.7410078048706055,
      "step": 59047
    },
    {
      "epoch": 5.631160736083984e-05,
      "step": 59047,
      "training_step_time": 6.818020343780518
    },
    {
      "epoch": 5.631256103515625e-05,
      "model_forward_time": 1.7438867092132568,
      "step": 59048
    },
    {
      "epoch": 5.631256103515625e-05,
      "step": 59048,
      "training_step_time": 7.058413028717041
    },
    {
      "epoch": 5.6313514709472656e-05,
      "model_forward_time": 1.745767593383789,
      "step": 59049
    },
    {
      "epoch": 5.6313514709472656e-05,
      "step": 59049,
      "training_step_time": 7.040716886520386
    },
    {
      "epoch": 5.631446838378906e-05,
      "grad_norm": 0.07429113984107971,
      "learning_rate": 3.925587700534559e-06,
      "loss": 0.0591,
      "step": 59050
    },
    {
      "epoch": 5.631446838378906e-05,
      "model_forward_time": 1.7434046268463135,
      "step": 59050
    },
    {
      "dynamics_loss_avg": 0.06912087760865689,
      "epoch": 5.631446838378906e-05,
      "step": 59050
    },
    {
      "action_loss_avg": 0.0017043535015545785,
      "epoch": 5.631446838378906e-05,
      "step": 59050
    },
    {
      "epoch": 5.631446838378906e-05,
      "step": 59050,
      "training_step_time": 7.039259195327759
    },
    {
      "epoch": 5.631542205810547e-05,
      "model_forward_time": 1.7487459182739258,
      "step": 59051
    },
    {
      "epoch": 5.631542205810547e-05,
      "step": 59051,
      "training_step_time": 7.086935520172119
    },
    {
      "epoch": 5.6316375732421875e-05,
      "model_forward_time": 1.7455418109893799,
      "step": 59052
    },
    {
      "epoch": 5.6316375732421875e-05,
      "step": 59052,
      "training_step_time": 7.07438850402832
    },
    {
      "epoch": 5.631732940673828e-05,
      "model_forward_time": 2.8831253051757812,
      "step": 59053
    },
    {
      "epoch": 5.631732940673828e-05,
      "step": 59053,
      "training_step_time": 11.369545459747314
    },
    {
      "epoch": 5.631828308105469e-05,
      "model_forward_time": 1.758554458618164,
      "step": 59054
    },
    {
      "epoch": 5.631828308105469e-05,
      "step": 59054,
      "training_step_time": 7.88516902923584
    },
    {
      "epoch": 5.6319236755371093e-05,
      "model_forward_time": 1.7618510723114014,
      "step": 59055
    },
    {
      "epoch": 5.6319236755371093e-05,
      "step": 59055,
      "training_step_time": 7.069239854812622
    },
    {
      "epoch": 5.63201904296875e-05,
      "model_forward_time": 1.7443888187408447,
      "step": 59056
    },
    {
      "epoch": 5.63201904296875e-05,
      "step": 59056,
      "training_step_time": 7.040688753128052
    },
    {
      "epoch": 5.632114410400391e-05,
      "model_forward_time": 1.7433280944824219,
      "step": 59057
    },
    {
      "epoch": 5.632114410400391e-05,
      "step": 59057,
      "training_step_time": 7.065896511077881
    },
    {
      "epoch": 5.632209777832031e-05,
      "model_forward_time": 1.7403168678283691,
      "step": 59058
    },
    {
      "epoch": 5.632209777832031e-05,
      "step": 59058,
      "training_step_time": 7.070205211639404
    },
    {
      "epoch": 5.6323051452636716e-05,
      "model_forward_time": 1.7423624992370605,
      "step": 59059
    },
    {
      "epoch": 5.6323051452636716e-05,
      "step": 59059,
      "training_step_time": 7.031510591506958
    },
    {
      "epoch": 5.632400512695313e-05,
      "grad_norm": 0.23167790472507477,
      "learning_rate": 3.923972914685743e-06,
      "loss": 0.0582,
      "step": 59060
    },
    {
      "epoch": 5.632400512695313e-05,
      "model_forward_time": 1.748978614807129,
      "step": 59060
    },
    {
      "dynamics_loss_avg": 0.05922743659466505,
      "epoch": 5.632400512695313e-05,
      "step": 59060
    },
    {
      "action_loss_avg": 0.0020247695210855454,
      "epoch": 5.632400512695313e-05,
      "step": 59060
    },
    {
      "epoch": 5.632400512695313e-05,
      "step": 59060,
      "training_step_time": 7.030491590499878
    },
    {
      "epoch": 5.632495880126953e-05,
      "model_forward_time": 1.7511370182037354,
      "step": 59061
    },
    {
      "epoch": 5.632495880126953e-05,
      "step": 59061,
      "training_step_time": 7.007844686508179
    },
    {
      "epoch": 5.6325912475585935e-05,
      "model_forward_time": 2.9000585079193115,
      "step": 59062
    },
    {
      "epoch": 5.6325912475585935e-05,
      "step": 59062,
      "training_step_time": 10.136029720306396
    },
    {
      "epoch": 5.6326866149902346e-05,
      "model_forward_time": 1.7536771297454834,
      "step": 59063
    },
    {
      "epoch": 5.6326866149902346e-05,
      "step": 59063,
      "training_step_time": 7.899582862854004
    },
    {
      "epoch": 5.632781982421875e-05,
      "model_forward_time": 1.75406813621521,
      "step": 59064
    },
    {
      "epoch": 5.632781982421875e-05,
      "step": 59064,
      "training_step_time": 7.73396372795105
    },
    {
      "epoch": 5.6328773498535154e-05,
      "model_forward_time": 1.7494375705718994,
      "step": 59065
    },
    {
      "epoch": 5.6328773498535154e-05,
      "step": 59065,
      "training_step_time": 6.986127853393555
    },
    {
      "epoch": 5.6329727172851565e-05,
      "model_forward_time": 1.7418413162231445,
      "step": 59066
    },
    {
      "epoch": 5.6329727172851565e-05,
      "step": 59066,
      "training_step_time": 6.866189241409302
    },
    {
      "epoch": 5.633068084716797e-05,
      "model_forward_time": 1.7466418743133545,
      "step": 59067
    },
    {
      "epoch": 5.633068084716797e-05,
      "step": 59067,
      "training_step_time": 6.960264205932617
    },
    {
      "epoch": 5.633163452148437e-05,
      "model_forward_time": 1.7448103427886963,
      "step": 59068
    },
    {
      "epoch": 5.633163452148437e-05,
      "step": 59068,
      "training_step_time": 6.978601932525635
    },
    {
      "epoch": 5.6332588195800784e-05,
      "model_forward_time": 1.7479426860809326,
      "step": 59069
    },
    {
      "epoch": 5.6332588195800784e-05,
      "step": 59069,
      "training_step_time": 6.992686033248901
    },
    {
      "epoch": 5.633354187011719e-05,
      "grad_norm": 0.09700281172990799,
      "learning_rate": 3.922358246509631e-06,
      "loss": 0.0567,
      "step": 59070
    },
    {
      "epoch": 5.633354187011719e-05,
      "model_forward_time": 1.746225118637085,
      "step": 59070
    },
    {
      "dynamics_loss_avg": 0.05399429779499769,
      "epoch": 5.633354187011719e-05,
      "step": 59070
    },
    {
      "action_loss_avg": 0.0012721005245111884,
      "epoch": 5.633354187011719e-05,
      "step": 59070
    },
    {
      "epoch": 5.633354187011719e-05,
      "step": 59070,
      "training_step_time": 7.016319990158081
    },
    {
      "epoch": 5.633449554443359e-05,
      "model_forward_time": 2.882765531539917,
      "step": 59071
    },
    {
      "epoch": 5.633449554443359e-05,
      "step": 59071,
      "training_step_time": 9.82288408279419
    },
    {
      "epoch": 5.633544921875e-05,
      "model_forward_time": 1.745591402053833,
      "step": 59072
    },
    {
      "epoch": 5.633544921875e-05,
      "step": 59072,
      "training_step_time": 8.019931316375732
    },
    {
      "epoch": 5.633640289306641e-05,
      "model_forward_time": 1.741403341293335,
      "step": 59073
    },
    {
      "epoch": 5.633640289306641e-05,
      "step": 59073,
      "training_step_time": 7.773951053619385
    },
    {
      "epoch": 5.633735656738281e-05,
      "model_forward_time": 1.7527198791503906,
      "step": 59074
    },
    {
      "epoch": 5.633735656738281e-05,
      "step": 59074,
      "training_step_time": 6.94314432144165
    },
    {
      "epoch": 5.633831024169922e-05,
      "model_forward_time": 1.75411057472229,
      "step": 59075
    },
    {
      "epoch": 5.633831024169922e-05,
      "step": 59075,
      "training_step_time": 7.010210037231445
    },
    {
      "epoch": 5.6339263916015625e-05,
      "model_forward_time": 1.7402830123901367,
      "step": 59076
    },
    {
      "epoch": 5.6339263916015625e-05,
      "step": 59076,
      "training_step_time": 6.852836608886719
    },
    {
      "epoch": 5.634021759033203e-05,
      "model_forward_time": 1.746917724609375,
      "step": 59077
    },
    {
      "epoch": 5.634021759033203e-05,
      "step": 59077,
      "training_step_time": 7.013615846633911
    },
    {
      "epoch": 5.634117126464844e-05,
      "model_forward_time": 1.7451462745666504,
      "step": 59078
    },
    {
      "epoch": 5.634117126464844e-05,
      "step": 59078,
      "training_step_time": 7.030112028121948
    },
    {
      "epoch": 5.6342124938964844e-05,
      "model_forward_time": 1.744354009628296,
      "step": 59079
    },
    {
      "epoch": 5.6342124938964844e-05,
      "step": 59079,
      "training_step_time": 6.980712175369263
    },
    {
      "epoch": 5.634307861328125e-05,
      "grad_norm": 0.09081488102674484,
      "learning_rate": 3.920743696182801e-06,
      "loss": 0.0539,
      "step": 59080
    },
    {
      "epoch": 5.634307861328125e-05,
      "model_forward_time": 2.8923394680023193,
      "step": 59080
    },
    {
      "dynamics_loss_avg": 0.05333498939871788,
      "epoch": 5.634307861328125e-05,
      "step": 59080
    },
    {
      "action_loss_avg": 0.0018409115553367883,
      "epoch": 5.634307861328125e-05,
      "step": 59080
    },
    {
      "epoch": 5.634307861328125e-05,
      "step": 59080,
      "training_step_time": 8.812289237976074
    },
    {
      "epoch": 5.634403228759766e-05,
      "model_forward_time": 1.746415138244629,
      "step": 59081
    },
    {
      "epoch": 5.634403228759766e-05,
      "step": 59081,
      "training_step_time": 8.036818504333496
    },
    {
      "epoch": 5.634498596191406e-05,
      "model_forward_time": 1.741729974746704,
      "step": 59082
    },
    {
      "epoch": 5.634498596191406e-05,
      "step": 59082,
      "training_step_time": 7.727491617202759
    },
    {
      "epoch": 5.634593963623047e-05,
      "model_forward_time": 1.7440505027770996,
      "step": 59083
    },
    {
      "epoch": 5.634593963623047e-05,
      "step": 59083,
      "training_step_time": 7.011165142059326
    },
    {
      "epoch": 5.634689331054688e-05,
      "model_forward_time": 1.7369358539581299,
      "step": 59084
    },
    {
      "epoch": 5.634689331054688e-05,
      "step": 59084,
      "training_step_time": 6.978434085845947
    },
    {
      "epoch": 5.634784698486328e-05,
      "model_forward_time": 1.7433302402496338,
      "step": 59085
    },
    {
      "epoch": 5.634784698486328e-05,
      "step": 59085,
      "training_step_time": 6.984304666519165
    },
    {
      "epoch": 5.6348800659179686e-05,
      "model_forward_time": 1.3835639953613281,
      "step": 59086
    },
    {
      "epoch": 5.6348800659179686e-05,
      "step": 59086,
      "training_step_time": 6.970656633377075
    },
    {
      "epoch": 5.63497543334961e-05,
      "model_forward_time": 1.7388279438018799,
      "step": 59087
    },
    {
      "epoch": 5.63497543334961e-05,
      "step": 59087,
      "training_step_time": 6.953947067260742
    },
    {
      "epoch": 5.63507080078125e-05,
      "model_forward_time": 1.7434852123260498,
      "step": 59088
    },
    {
      "epoch": 5.63507080078125e-05,
      "step": 59088,
      "training_step_time": 6.96419882774353
    },
    {
      "epoch": 5.6351661682128905e-05,
      "model_forward_time": 1.7406363487243652,
      "step": 59089
    },
    {
      "epoch": 5.6351661682128905e-05,
      "step": 59089,
      "training_step_time": 9.326049327850342
    },
    {
      "epoch": 5.6352615356445316e-05,
      "grad_norm": 0.0881032720208168,
      "learning_rate": 3.919129263881817e-06,
      "loss": 0.059,
      "step": 59090
    },
    {
      "epoch": 5.6352615356445316e-05,
      "model_forward_time": 2.89743709564209,
      "step": 59090
    },
    {
      "dynamics_loss_avg": 0.06728538554161786,
      "epoch": 5.6352615356445316e-05,
      "step": 59090
    },
    {
      "action_loss_avg": 0.001602486928459257,
      "epoch": 5.6352615356445316e-05,
      "step": 59090
    },
    {
      "epoch": 5.6352615356445316e-05,
      "step": 59090,
      "training_step_time": 9.526060581207275
    },
    {
      "epoch": 5.635356903076172e-05,
      "model_forward_time": 1.766775369644165,
      "step": 59091
    },
    {
      "epoch": 5.635356903076172e-05,
      "step": 59091,
      "training_step_time": 8.886193752288818
    },
    {
      "epoch": 5.6354522705078124e-05,
      "model_forward_time": 1.7397761344909668,
      "step": 59092
    },
    {
      "epoch": 5.6354522705078124e-05,
      "step": 59092,
      "training_step_time": 6.964592695236206
    },
    {
      "epoch": 5.6355476379394535e-05,
      "model_forward_time": 1.745377779006958,
      "step": 59093
    },
    {
      "epoch": 5.6355476379394535e-05,
      "step": 59093,
      "training_step_time": 6.973375558853149
    },
    {
      "epoch": 5.635643005371094e-05,
      "model_forward_time": 1.757509469985962,
      "step": 59094
    },
    {
      "epoch": 5.635643005371094e-05,
      "step": 59094,
      "training_step_time": 7.014451742172241
    },
    {
      "epoch": 5.635738372802734e-05,
      "model_forward_time": 1.75472092628479,
      "step": 59095
    },
    {
      "epoch": 5.635738372802734e-05,
      "step": 59095,
      "training_step_time": 7.0009613037109375
    },
    {
      "epoch": 5.635833740234375e-05,
      "model_forward_time": 1.7390358448028564,
      "step": 59096
    },
    {
      "epoch": 5.635833740234375e-05,
      "step": 59096,
      "training_step_time": 6.975997447967529
    },
    {
      "epoch": 5.635929107666016e-05,
      "model_forward_time": 1.7445290088653564,
      "step": 59097
    },
    {
      "epoch": 5.635929107666016e-05,
      "step": 59097,
      "training_step_time": 7.026297569274902
    },
    {
      "epoch": 5.636024475097656e-05,
      "model_forward_time": 1.7310986518859863,
      "step": 59098
    },
    {
      "epoch": 5.636024475097656e-05,
      "step": 59098,
      "training_step_time": 7.406277418136597
    },
    {
      "epoch": 5.6361198425292966e-05,
      "model_forward_time": 2.88204026222229,
      "step": 59099
    },
    {
      "epoch": 5.6361198425292966e-05,
      "step": 59099,
      "training_step_time": 10.767334461212158
    },
    {
      "epoch": 5.6362152099609376e-05,
      "grad_norm": 0.09265711158514023,
      "learning_rate": 3.917514949783231e-06,
      "loss": 0.0593,
      "step": 59100
    },
    {
      "epoch": 5.6362152099609376e-05,
      "model_forward_time": 1.7444357872009277,
      "step": 59100
    },
    {
      "dynamics_loss_avg": 0.05482649840414524,
      "epoch": 5.6362152099609376e-05,
      "step": 59100
    },
    {
      "action_loss_avg": 0.0021433798770885914,
      "epoch": 5.6362152099609376e-05,
      "step": 59100
    },
    {
      "epoch": 5.6362152099609376e-05,
      "step": 59100,
      "training_step_time": 7.823141098022461
    },
    {
      "epoch": 5.636310577392578e-05,
      "model_forward_time": 1.7419013977050781,
      "step": 59101
    },
    {
      "epoch": 5.636310577392578e-05,
      "step": 59101,
      "training_step_time": 7.292343378067017
    },
    {
      "epoch": 5.6364059448242184e-05,
      "model_forward_time": 1.7459616661071777,
      "step": 59102
    },
    {
      "epoch": 5.6364059448242184e-05,
      "step": 59102,
      "training_step_time": 6.862792015075684
    },
    {
      "epoch": 5.6365013122558595e-05,
      "model_forward_time": 1.743764877319336,
      "step": 59103
    },
    {
      "epoch": 5.6365013122558595e-05,
      "step": 59103,
      "training_step_time": 6.947267532348633
    },
    {
      "epoch": 5.6365966796875e-05,
      "model_forward_time": 1.745849847793579,
      "step": 59104
    },
    {
      "epoch": 5.6365966796875e-05,
      "step": 59104,
      "training_step_time": 7.004211187362671
    },
    {
      "epoch": 5.63669204711914e-05,
      "model_forward_time": 1.7442941665649414,
      "step": 59105
    },
    {
      "epoch": 5.63669204711914e-05,
      "step": 59105,
      "training_step_time": 7.0372772216796875
    },
    {
      "epoch": 5.6367874145507814e-05,
      "model_forward_time": 1.7410507202148438,
      "step": 59106
    },
    {
      "epoch": 5.6367874145507814e-05,
      "step": 59106,
      "training_step_time": 6.977678060531616
    },
    {
      "epoch": 5.636882781982422e-05,
      "model_forward_time": 1.7486937046051025,
      "step": 59107
    },
    {
      "epoch": 5.636882781982422e-05,
      "step": 59107,
      "training_step_time": 7.423938751220703
    },
    {
      "epoch": 5.636978149414062e-05,
      "model_forward_time": 2.8814899921417236,
      "step": 59108
    },
    {
      "epoch": 5.636978149414062e-05,
      "step": 59108,
      "training_step_time": 10.387609481811523
    },
    {
      "epoch": 5.637073516845703e-05,
      "model_forward_time": 1.7437028884887695,
      "step": 59109
    },
    {
      "epoch": 5.637073516845703e-05,
      "step": 59109,
      "training_step_time": 7.94771409034729
    },
    {
      "epoch": 5.637168884277344e-05,
      "grad_norm": 0.11529471725225449,
      "learning_rate": 3.915900754063584e-06,
      "loss": 0.0579,
      "step": 59110
    },
    {
      "epoch": 5.637168884277344e-05,
      "model_forward_time": 1.7388300895690918,
      "step": 59110
    },
    {
      "dynamics_loss_avg": 0.07175861671566963,
      "epoch": 5.637168884277344e-05,
      "step": 59110
    },
    {
      "action_loss_avg": 0.001538185152458027,
      "epoch": 5.637168884277344e-05,
      "step": 59110
    },
    {
      "epoch": 5.637168884277344e-05,
      "step": 59110,
      "training_step_time": 7.65653395652771
    },
    {
      "epoch": 5.637264251708984e-05,
      "model_forward_time": 1.7439048290252686,
      "step": 59111
    },
    {
      "epoch": 5.637264251708984e-05,
      "step": 59111,
      "training_step_time": 7.325868606567383
    },
    {
      "epoch": 5.637359619140625e-05,
      "model_forward_time": 1.7416775226593018,
      "step": 59112
    },
    {
      "epoch": 5.637359619140625e-05,
      "step": 59112,
      "training_step_time": 7.009974718093872
    },
    {
      "epoch": 5.6374549865722656e-05,
      "model_forward_time": 1.746227741241455,
      "step": 59113
    },
    {
      "epoch": 5.6374549865722656e-05,
      "step": 59113,
      "training_step_time": 6.929711103439331
    },
    {
      "epoch": 5.637550354003906e-05,
      "model_forward_time": 1.7407431602478027,
      "step": 59114
    },
    {
      "epoch": 5.637550354003906e-05,
      "step": 59114,
      "training_step_time": 6.9433159828186035
    },
    {
      "epoch": 5.637645721435547e-05,
      "model_forward_time": 1.738649606704712,
      "step": 59115
    },
    {
      "epoch": 5.637645721435547e-05,
      "step": 59115,
      "training_step_time": 6.962682247161865
    },
    {
      "epoch": 5.6377410888671875e-05,
      "model_forward_time": 1.7515430450439453,
      "step": 59116
    },
    {
      "epoch": 5.6377410888671875e-05,
      "step": 59116,
      "training_step_time": 7.403510332107544
    },
    {
      "epoch": 5.637836456298828e-05,
      "model_forward_time": 2.8544609546661377,
      "step": 59117
    },
    {
      "epoch": 5.637836456298828e-05,
      "step": 59117,
      "training_step_time": 9.476823568344116
    },
    {
      "epoch": 5.637931823730469e-05,
      "model_forward_time": 1.738966941833496,
      "step": 59118
    },
    {
      "epoch": 5.637931823730469e-05,
      "step": 59118,
      "training_step_time": 10.957792043685913
    },
    {
      "epoch": 5.6380271911621094e-05,
      "model_forward_time": 1.7408380508422852,
      "step": 59119
    },
    {
      "epoch": 5.6380271911621094e-05,
      "step": 59119,
      "training_step_time": 7.7828593254089355
    },
    {
      "epoch": 5.63812255859375e-05,
      "grad_norm": 0.08894424885511398,
      "learning_rate": 3.914286676899401e-06,
      "loss": 0.057,
      "step": 59120
    },
    {
      "epoch": 5.63812255859375e-05,
      "model_forward_time": 1.7430005073547363,
      "step": 59120
    },
    {
      "dynamics_loss_avg": 0.04577594809234142,
      "epoch": 5.63812255859375e-05,
      "step": 59120
    },
    {
      "action_loss_avg": 0.0012546518730232493,
      "epoch": 5.63812255859375e-05,
      "step": 59120
    },
    {
      "epoch": 5.63812255859375e-05,
      "step": 59120,
      "training_step_time": 7.278397560119629
    },
    {
      "epoch": 5.638217926025391e-05,
      "model_forward_time": 1.7453131675720215,
      "step": 59121
    },
    {
      "epoch": 5.638217926025391e-05,
      "step": 59121,
      "training_step_time": 6.958104133605957
    },
    {
      "epoch": 5.638313293457031e-05,
      "model_forward_time": 1.7501530647277832,
      "step": 59122
    },
    {
      "epoch": 5.638313293457031e-05,
      "step": 59122,
      "training_step_time": 6.969961404800415
    },
    {
      "epoch": 5.6384086608886716e-05,
      "model_forward_time": 1.755763053894043,
      "step": 59123
    },
    {
      "epoch": 5.6384086608886716e-05,
      "step": 59123,
      "training_step_time": 6.94351601600647
    },
    {
      "epoch": 5.638504028320313e-05,
      "model_forward_time": 1.7419021129608154,
      "step": 59124
    },
    {
      "epoch": 5.638504028320313e-05,
      "step": 59124,
      "training_step_time": 6.885489225387573
    },
    {
      "epoch": 5.638599395751953e-05,
      "model_forward_time": 1.7540762424468994,
      "step": 59125
    },
    {
      "epoch": 5.638599395751953e-05,
      "step": 59125,
      "training_step_time": 7.369772911071777
    },
    {
      "epoch": 5.6386947631835935e-05,
      "model_forward_time": 2.881983995437622,
      "step": 59126
    },
    {
      "epoch": 5.6386947631835935e-05,
      "step": 59126,
      "training_step_time": 8.903824806213379
    },
    {
      "epoch": 5.6387901306152346e-05,
      "model_forward_time": 1.731799840927124,
      "step": 59127
    },
    {
      "epoch": 5.6387901306152346e-05,
      "step": 59127,
      "training_step_time": 9.909869909286499
    },
    {
      "epoch": 5.638885498046875e-05,
      "model_forward_time": 1.7327206134796143,
      "step": 59128
    },
    {
      "epoch": 5.638885498046875e-05,
      "step": 59128,
      "training_step_time": 7.701667547225952
    },
    {
      "epoch": 5.6389808654785154e-05,
      "model_forward_time": 1.0281801223754883,
      "step": 59129
    },
    {
      "epoch": 5.6389808654785154e-05,
      "step": 59129,
      "training_step_time": 7.158443927764893
    },
    {
      "epoch": 5.6390762329101565e-05,
      "grad_norm": 0.12197022140026093,
      "learning_rate": 3.912672718467194e-06,
      "loss": 0.0562,
      "step": 59130
    },
    {
      "epoch": 5.6390762329101565e-05,
      "model_forward_time": 1.7384793758392334,
      "step": 59130
    },
    {
      "dynamics_loss_avg": 0.07013271898031234,
      "epoch": 5.6390762329101565e-05,
      "step": 59130
    },
    {
      "action_loss_avg": 0.0012797114584827795,
      "epoch": 5.6390762329101565e-05,
      "step": 59130
    },
    {
      "epoch": 5.6390762329101565e-05,
      "step": 59130,
      "training_step_time": 6.946630239486694
    },
    {
      "epoch": 5.639171600341797e-05,
      "model_forward_time": 1.7394120693206787,
      "step": 59131
    },
    {
      "epoch": 5.639171600341797e-05,
      "step": 59131,
      "training_step_time": 6.89080023765564
    },
    {
      "epoch": 5.639266967773437e-05,
      "model_forward_time": 1.7348923683166504,
      "step": 59132
    },
    {
      "epoch": 5.639266967773437e-05,
      "step": 59132,
      "training_step_time": 6.844168424606323
    },
    {
      "epoch": 5.6393623352050784e-05,
      "model_forward_time": 1.733529806137085,
      "step": 59133
    },
    {
      "epoch": 5.6393623352050784e-05,
      "step": 59133,
      "training_step_time": 6.937593936920166
    },
    {
      "epoch": 5.639457702636719e-05,
      "model_forward_time": 1.7302653789520264,
      "step": 59134
    },
    {
      "epoch": 5.639457702636719e-05,
      "step": 59134,
      "training_step_time": 7.489648342132568
    },
    {
      "epoch": 5.639553070068359e-05,
      "model_forward_time": 1.7301678657531738,
      "step": 59135
    },
    {
      "epoch": 5.639553070068359e-05,
      "step": 59135,
      "training_step_time": 7.8542702198028564
    },
    {
      "epoch": 5.6396484375e-05,
      "model_forward_time": 2.920192241668701,
      "step": 59136
    },
    {
      "epoch": 5.6396484375e-05,
      "step": 59136,
      "training_step_time": 9.189616441726685
    },
    {
      "epoch": 5.639743804931641e-05,
      "model_forward_time": 1.73704195022583,
      "step": 59137
    },
    {
      "epoch": 5.639743804931641e-05,
      "step": 59137,
      "training_step_time": 7.8009209632873535
    },
    {
      "epoch": 5.639839172363281e-05,
      "model_forward_time": 1.7272520065307617,
      "step": 59138
    },
    {
      "epoch": 5.639839172363281e-05,
      "step": 59138,
      "training_step_time": 7.732872247695923
    },
    {
      "epoch": 5.639934539794922e-05,
      "model_forward_time": 1.729935884475708,
      "step": 59139
    },
    {
      "epoch": 5.639934539794922e-05,
      "step": 59139,
      "training_step_time": 7.216871500015259
    },
    {
      "epoch": 5.6400299072265626e-05,
      "grad_norm": 0.14627616107463837,
      "learning_rate": 3.911058878943463e-06,
      "loss": 0.0567,
      "step": 59140
    },
    {
      "epoch": 5.6400299072265626e-05,
      "model_forward_time": 1.733280897140503,
      "step": 59140
    },
    {
      "dynamics_loss_avg": 0.06189983263611794,
      "epoch": 5.6400299072265626e-05,
      "step": 59140
    },
    {
      "action_loss_avg": 0.001537671301048249,
      "epoch": 5.6400299072265626e-05,
      "step": 59140
    },
    {
      "epoch": 5.6400299072265626e-05,
      "step": 59140,
      "training_step_time": 6.968701601028442
    },
    {
      "epoch": 5.640125274658203e-05,
      "model_forward_time": 1.7342901229858398,
      "step": 59141
    },
    {
      "epoch": 5.640125274658203e-05,
      "step": 59141,
      "training_step_time": 6.9705281257629395
    },
    {
      "epoch": 5.640220642089844e-05,
      "model_forward_time": 1.7329661846160889,
      "step": 59142
    },
    {
      "epoch": 5.640220642089844e-05,
      "step": 59142,
      "training_step_time": 6.9665985107421875
    },
    {
      "epoch": 5.6403160095214844e-05,
      "model_forward_time": 1.731640338897705,
      "step": 59143
    },
    {
      "epoch": 5.6403160095214844e-05,
      "step": 59143,
      "training_step_time": 7.445610284805298
    },
    {
      "epoch": 5.640411376953125e-05,
      "model_forward_time": 1.7284109592437744,
      "step": 59144
    },
    {
      "epoch": 5.640411376953125e-05,
      "step": 59144,
      "training_step_time": 7.91809344291687
    },
    {
      "epoch": 5.640506744384766e-05,
      "model_forward_time": 2.8500688076019287,
      "step": 59145
    },
    {
      "epoch": 5.640506744384766e-05,
      "step": 59145,
      "training_step_time": 11.840184688568115
    },
    {
      "epoch": 5.640602111816406e-05,
      "model_forward_time": 1.7481706142425537,
      "step": 59146
    },
    {
      "epoch": 5.640602111816406e-05,
      "step": 59146,
      "training_step_time": 7.8348870277404785
    },
    {
      "epoch": 5.640697479248047e-05,
      "model_forward_time": 1.7270731925964355,
      "step": 59147
    },
    {
      "epoch": 5.640697479248047e-05,
      "step": 59147,
      "training_step_time": 7.742490291595459
    },
    {
      "epoch": 5.640792846679688e-05,
      "model_forward_time": 1.728506326675415,
      "step": 59148
    },
    {
      "epoch": 5.640792846679688e-05,
      "step": 59148,
      "training_step_time": 7.137598991394043
    },
    {
      "epoch": 5.640888214111328e-05,
      "model_forward_time": 1.7327604293823242,
      "step": 59149
    },
    {
      "epoch": 5.640888214111328e-05,
      "step": 59149,
      "training_step_time": 6.894833087921143
    },
    {
      "epoch": 5.6409835815429686e-05,
      "grad_norm": 0.10689884424209595,
      "learning_rate": 3.909445158504698e-06,
      "loss": 0.059,
      "step": 59150
    },
    {
      "epoch": 5.6409835815429686e-05,
      "model_forward_time": 1.3595309257507324,
      "step": 59150
    },
    {
      "dynamics_loss_avg": 0.07909627482295037,
      "epoch": 5.6409835815429686e-05,
      "step": 59150
    },
    {
      "action_loss_avg": 0.001806391286663711,
      "epoch": 5.6409835815429686e-05,
      "step": 59150
    },
    {
      "epoch": 5.6409835815429686e-05,
      "step": 59150,
      "training_step_time": 6.938086986541748
    },
    {
      "epoch": 5.64107894897461e-05,
      "model_forward_time": 1.730938196182251,
      "step": 59151
    },
    {
      "epoch": 5.64107894897461e-05,
      "step": 59151,
      "training_step_time": 6.872580289840698
    },
    {
      "epoch": 5.64117431640625e-05,
      "model_forward_time": 1.7303695678710938,
      "step": 59152
    },
    {
      "epoch": 5.64117431640625e-05,
      "step": 59152,
      "training_step_time": 7.228055715560913
    },
    {
      "epoch": 5.6412696838378905e-05,
      "model_forward_time": 1.751952886581421,
      "step": 59153
    },
    {
      "epoch": 5.6412696838378905e-05,
      "step": 59153,
      "training_step_time": 7.86767053604126
    },
    {
      "epoch": 5.6413650512695316e-05,
      "model_forward_time": 2.9264779090881348,
      "step": 59154
    },
    {
      "epoch": 5.6413650512695316e-05,
      "step": 59154,
      "training_step_time": 10.232433795928955
    },
    {
      "epoch": 5.641460418701172e-05,
      "model_forward_time": 1.7459769248962402,
      "step": 59155
    },
    {
      "epoch": 5.641460418701172e-05,
      "step": 59155,
      "training_step_time": 9.59845519065857
    },
    {
      "epoch": 5.6415557861328124e-05,
      "model_forward_time": 1.7393574714660645,
      "step": 59156
    },
    {
      "epoch": 5.6415557861328124e-05,
      "step": 59156,
      "training_step_time": 7.659519910812378
    },
    {
      "epoch": 5.641651153564453e-05,
      "model_forward_time": 1.7474215030670166,
      "step": 59157
    },
    {
      "epoch": 5.641651153564453e-05,
      "step": 59157,
      "training_step_time": 7.093002557754517
    },
    {
      "epoch": 5.641746520996094e-05,
      "model_forward_time": 1.3905880451202393,
      "step": 59158
    },
    {
      "epoch": 5.641746520996094e-05,
      "step": 59158,
      "training_step_time": 6.828149318695068
    },
    {
      "epoch": 5.641841888427734e-05,
      "model_forward_time": 1.7485191822052002,
      "step": 59159
    },
    {
      "epoch": 5.641841888427734e-05,
      "step": 59159,
      "training_step_time": 6.849205493927002
    },
    {
      "epoch": 5.641937255859375e-05,
      "grad_norm": 0.10207214206457138,
      "learning_rate": 3.90783155732737e-06,
      "loss": 0.0575,
      "step": 59160
    },
    {
      "epoch": 5.641937255859375e-05,
      "model_forward_time": 1.7535395622253418,
      "step": 59160
    },
    {
      "dynamics_loss_avg": 0.04483958911150694,
      "epoch": 5.641937255859375e-05,
      "step": 59160
    },
    {
      "action_loss_avg": 0.0011919511656742544,
      "epoch": 5.641937255859375e-05,
      "step": 59160
    },
    {
      "epoch": 5.641937255859375e-05,
      "step": 59160,
      "training_step_time": 6.8623809814453125
    },
    {
      "epoch": 5.642032623291016e-05,
      "model_forward_time": 1.7531895637512207,
      "step": 59161
    },
    {
      "epoch": 5.642032623291016e-05,
      "step": 59161,
      "training_step_time": 7.141657590866089
    },
    {
      "epoch": 5.642127990722656e-05,
      "model_forward_time": 1.745133399963379,
      "step": 59162
    },
    {
      "epoch": 5.642127990722656e-05,
      "step": 59162,
      "training_step_time": 7.962226390838623
    },
    {
      "epoch": 5.6422233581542966e-05,
      "model_forward_time": 1.844219446182251,
      "step": 59163
    },
    {
      "epoch": 5.6422233581542966e-05,
      "step": 59163,
      "training_step_time": 9.753736972808838
    },
    {
      "epoch": 5.6423187255859376e-05,
      "model_forward_time": 2.882347583770752,
      "step": 59164
    },
    {
      "epoch": 5.6423187255859376e-05,
      "step": 59164,
      "training_step_time": 8.514632225036621
    },
    {
      "epoch": 5.642414093017578e-05,
      "model_forward_time": 1.7542684078216553,
      "step": 59165
    },
    {
      "epoch": 5.642414093017578e-05,
      "step": 59165,
      "training_step_time": 7.578350782394409
    },
    {
      "epoch": 5.6425094604492184e-05,
      "model_forward_time": 1.7528886795043945,
      "step": 59166
    },
    {
      "epoch": 5.6425094604492184e-05,
      "step": 59166,
      "training_step_time": 7.09161639213562
    },
    {
      "epoch": 5.6426048278808595e-05,
      "model_forward_time": 1.753316879272461,
      "step": 59167
    },
    {
      "epoch": 5.6426048278808595e-05,
      "step": 59167,
      "training_step_time": 6.80376124382019
    },
    {
      "epoch": 5.6427001953125e-05,
      "model_forward_time": 1.7571954727172852,
      "step": 59168
    },
    {
      "epoch": 5.6427001953125e-05,
      "step": 59168,
      "training_step_time": 6.810427665710449
    },
    {
      "epoch": 5.64279556274414e-05,
      "model_forward_time": 1.755155324935913,
      "step": 59169
    },
    {
      "epoch": 5.64279556274414e-05,
      "step": 59169,
      "training_step_time": 6.765392303466797
    },
    {
      "epoch": 5.6428909301757814e-05,
      "grad_norm": 0.095771923661232,
      "learning_rate": 3.906218075587938e-06,
      "loss": 0.0535,
      "step": 59170
    },
    {
      "epoch": 5.6428909301757814e-05,
      "model_forward_time": 1.748091459274292,
      "step": 59170
    },
    {
      "dynamics_loss_avg": 0.05424580657854676,
      "epoch": 5.6428909301757814e-05,
      "step": 59170
    },
    {
      "action_loss_avg": 0.0015895433287369087,
      "epoch": 5.6428909301757814e-05,
      "step": 59170
    },
    {
      "epoch": 5.6428909301757814e-05,
      "step": 59170,
      "training_step_time": 7.1576714515686035
    },
    {
      "epoch": 5.642986297607422e-05,
      "model_forward_time": 1.7556054592132568,
      "step": 59171
    },
    {
      "epoch": 5.642986297607422e-05,
      "step": 59171,
      "training_step_time": 7.968302488327026
    },
    {
      "epoch": 5.643081665039062e-05,
      "model_forward_time": 1.81508469581604,
      "step": 59172
    },
    {
      "epoch": 5.643081665039062e-05,
      "step": 59172,
      "training_step_time": 9.465931415557861
    },
    {
      "epoch": 5.643177032470703e-05,
      "model_forward_time": 2.902158737182617,
      "step": 59173
    },
    {
      "epoch": 5.643177032470703e-05,
      "step": 59173,
      "training_step_time": 9.323400497436523
    },
    {
      "epoch": 5.643272399902344e-05,
      "model_forward_time": 1.7544186115264893,
      "step": 59174
    },
    {
      "epoch": 5.643272399902344e-05,
      "step": 59174,
      "training_step_time": 7.848073244094849
    },
    {
      "epoch": 5.643367767333984e-05,
      "model_forward_time": 1.7602992057800293,
      "step": 59175
    },
    {
      "epoch": 5.643367767333984e-05,
      "step": 59175,
      "training_step_time": 8.63820743560791
    },
    {
      "epoch": 5.643463134765625e-05,
      "model_forward_time": 1.7556567192077637,
      "step": 59176
    },
    {
      "epoch": 5.643463134765625e-05,
      "step": 59176,
      "training_step_time": 6.862064361572266
    },
    {
      "epoch": 5.6435585021972656e-05,
      "model_forward_time": 1.7560203075408936,
      "step": 59177
    },
    {
      "epoch": 5.6435585021972656e-05,
      "step": 59177,
      "training_step_time": 6.84946084022522
    },
    {
      "epoch": 5.643653869628906e-05,
      "model_forward_time": 1.7540266513824463,
      "step": 59178
    },
    {
      "epoch": 5.643653869628906e-05,
      "step": 59178,
      "training_step_time": 6.818166017532349
    },
    {
      "epoch": 5.643749237060547e-05,
      "model_forward_time": 1.7501018047332764,
      "step": 59179
    },
    {
      "epoch": 5.643749237060547e-05,
      "step": 59179,
      "training_step_time": 7.131721019744873
    },
    {
      "epoch": 5.6438446044921875e-05,
      "grad_norm": 0.11631680279970169,
      "learning_rate": 3.904604713462855e-06,
      "loss": 0.0568,
      "step": 59180
    },
    {
      "epoch": 5.6438446044921875e-05,
      "model_forward_time": 1.7628483772277832,
      "step": 59180
    },
    {
      "dynamics_loss_avg": 0.05411186572164297,
      "epoch": 5.6438446044921875e-05,
      "step": 59180
    },
    {
      "action_loss_avg": 0.0018743472406640648,
      "epoch": 5.6438446044921875e-05,
      "step": 59180
    },
    {
      "epoch": 5.6438446044921875e-05,
      "step": 59180,
      "training_step_time": 7.86781120300293
    },
    {
      "epoch": 5.643939971923828e-05,
      "model_forward_time": 1.7799386978149414,
      "step": 59181
    },
    {
      "epoch": 5.643939971923828e-05,
      "step": 59181,
      "training_step_time": 8.278328895568848
    },
    {
      "epoch": 5.644035339355469e-05,
      "model_forward_time": 2.901703119277954,
      "step": 59182
    },
    {
      "epoch": 5.644035339355469e-05,
      "step": 59182,
      "training_step_time": 10.142735719680786
    },
    {
      "epoch": 5.6441307067871094e-05,
      "model_forward_time": 1.7544763088226318,
      "step": 59183
    },
    {
      "epoch": 5.6441307067871094e-05,
      "step": 59183,
      "training_step_time": 8.291704177856445
    },
    {
      "epoch": 5.64422607421875e-05,
      "model_forward_time": 1.4025189876556396,
      "step": 59184
    },
    {
      "epoch": 5.64422607421875e-05,
      "step": 59184,
      "training_step_time": 7.5554039478302
    },
    {
      "epoch": 5.644321441650391e-05,
      "model_forward_time": 1.7544448375701904,
      "step": 59185
    },
    {
      "epoch": 5.644321441650391e-05,
      "step": 59185,
      "training_step_time": 7.576836585998535
    },
    {
      "epoch": 5.644416809082031e-05,
      "model_forward_time": 1.750018835067749,
      "step": 59186
    },
    {
      "epoch": 5.644416809082031e-05,
      "step": 59186,
      "training_step_time": 6.84301233291626
    },
    {
      "epoch": 5.6445121765136716e-05,
      "model_forward_time": 1.7609248161315918,
      "step": 59187
    },
    {
      "epoch": 5.6445121765136716e-05,
      "step": 59187,
      "training_step_time": 6.793299674987793
    },
    {
      "epoch": 5.644607543945313e-05,
      "model_forward_time": 1.7560904026031494,
      "step": 59188
    },
    {
      "epoch": 5.644607543945313e-05,
      "step": 59188,
      "training_step_time": 7.130469799041748
    },
    {
      "epoch": 5.644702911376953e-05,
      "model_forward_time": 1.7538886070251465,
      "step": 59189
    },
    {
      "epoch": 5.644702911376953e-05,
      "step": 59189,
      "training_step_time": 7.264062404632568
    },
    {
      "epoch": 5.6447982788085935e-05,
      "grad_norm": 0.12149491906166077,
      "learning_rate": 3.902991471128554e-06,
      "loss": 0.0565,
      "step": 59190
    },
    {
      "epoch": 5.6447982788085935e-05,
      "model_forward_time": 1.7637197971343994,
      "step": 59190
    },
    {
      "dynamics_loss_avg": 0.05802160603925586,
      "epoch": 5.6447982788085935e-05,
      "step": 59190
    },
    {
      "action_loss_avg": 0.0029687337868381292,
      "epoch": 5.6447982788085935e-05,
      "step": 59190
    },
    {
      "epoch": 5.6447982788085935e-05,
      "step": 59190,
      "training_step_time": 7.893523454666138
    },
    {
      "epoch": 5.6448936462402346e-05,
      "model_forward_time": 2.8460183143615723,
      "step": 59191
    },
    {
      "epoch": 5.6448936462402346e-05,
      "step": 59191,
      "training_step_time": 9.168794631958008
    },
    {
      "epoch": 5.644989013671875e-05,
      "model_forward_time": 1.7466967105865479,
      "step": 59192
    },
    {
      "epoch": 5.644989013671875e-05,
      "step": 59192,
      "training_step_time": 8.429883480072021
    },
    {
      "epoch": 5.6450843811035154e-05,
      "model_forward_time": 1.764672040939331,
      "step": 59193
    },
    {
      "epoch": 5.6450843811035154e-05,
      "step": 59193,
      "training_step_time": 7.584975004196167
    },
    {
      "epoch": 5.6451797485351565e-05,
      "model_forward_time": 1.7500622272491455,
      "step": 59194
    },
    {
      "epoch": 5.6451797485351565e-05,
      "step": 59194,
      "training_step_time": 7.496506690979004
    },
    {
      "epoch": 5.645275115966797e-05,
      "model_forward_time": 1.763690710067749,
      "step": 59195
    },
    {
      "epoch": 5.645275115966797e-05,
      "step": 59195,
      "training_step_time": 6.829207897186279
    },
    {
      "epoch": 5.645370483398437e-05,
      "model_forward_time": 1.753208875656128,
      "step": 59196
    },
    {
      "epoch": 5.645370483398437e-05,
      "step": 59196,
      "training_step_time": 6.818864583969116
    },
    {
      "epoch": 5.6454658508300784e-05,
      "model_forward_time": 1.749868392944336,
      "step": 59197
    },
    {
      "epoch": 5.6454658508300784e-05,
      "step": 59197,
      "training_step_time": 7.124165773391724
    },
    {
      "epoch": 5.645561218261719e-05,
      "model_forward_time": 1.749046802520752,
      "step": 59198
    },
    {
      "epoch": 5.645561218261719e-05,
      "step": 59198,
      "training_step_time": 7.084280729293823
    },
    {
      "epoch": 5.645656585693359e-05,
      "model_forward_time": 1.761887550354004,
      "step": 59199
    },
    {
      "epoch": 5.645656585693359e-05,
      "step": 59199,
      "training_step_time": 8.683284044265747
    },
    {
      "epoch": 5.645751953125e-05,
      "grad_norm": 0.09921860694885254,
      "learning_rate": 3.901378348761457e-06,
      "loss": 0.0572,
      "step": 59200
    },
    {
      "epoch": 5.645751953125e-05,
      "model_forward_time": 1.865351676940918,
      "step": 59200
    },
    {
      "dynamics_loss_avg": 0.05307105798274279,
      "epoch": 5.645751953125e-05,
      "step": 59200
    },
    {
      "action_loss_avg": 0.001869143490330316,
      "epoch": 5.645751953125e-05,
      "step": 59200
    },
    {
      "epoch": 5.645751953125e-05,
      "step": 59200,
      "training_step_time": 8.507365942001343
    },
    {
      "epoch": 5.645847320556641e-05,
      "model_forward_time": 2.941380262374878,
      "step": 59201
    },
    {
      "epoch": 5.645847320556641e-05,
      "step": 59201,
      "training_step_time": 8.814141035079956
    },
    {
      "epoch": 5.645942687988281e-05,
      "model_forward_time": 1.7608394622802734,
      "step": 59202
    },
    {
      "epoch": 5.645942687988281e-05,
      "step": 59202,
      "training_step_time": 7.6451709270477295
    },
    {
      "epoch": 5.646038055419922e-05,
      "model_forward_time": 1.760740041732788,
      "step": 59203
    },
    {
      "epoch": 5.646038055419922e-05,
      "step": 59203,
      "training_step_time": 7.109637022018433
    },
    {
      "epoch": 5.6461334228515626e-05,
      "model_forward_time": 1.7461106777191162,
      "step": 59204
    },
    {
      "epoch": 5.6461334228515626e-05,
      "step": 59204,
      "training_step_time": 7.567587852478027
    },
    {
      "epoch": 5.646228790283203e-05,
      "model_forward_time": 1.7532291412353516,
      "step": 59205
    },
    {
      "epoch": 5.646228790283203e-05,
      "step": 59205,
      "training_step_time": 6.822117805480957
    },
    {
      "epoch": 5.646324157714844e-05,
      "model_forward_time": 1.7627665996551514,
      "step": 59206
    },
    {
      "epoch": 5.646324157714844e-05,
      "step": 59206,
      "training_step_time": 7.213014602661133
    },
    {
      "epoch": 5.6464195251464844e-05,
      "model_forward_time": 1.75522780418396,
      "step": 59207
    },
    {
      "epoch": 5.6464195251464844e-05,
      "step": 59207,
      "training_step_time": 6.911695241928101
    },
    {
      "epoch": 5.646514892578125e-05,
      "model_forward_time": 1.7527570724487305,
      "step": 59208
    },
    {
      "epoch": 5.646514892578125e-05,
      "step": 59208,
      "training_step_time": 8.96603775024414
    },
    {
      "epoch": 5.646610260009766e-05,
      "model_forward_time": 1.8630647659301758,
      "step": 59209
    },
    {
      "epoch": 5.646610260009766e-05,
      "step": 59209,
      "training_step_time": 8.862546443939209
    },
    {
      "epoch": 5.646705627441406e-05,
      "grad_norm": 0.09726280719041824,
      "learning_rate": 3.899765346537971e-06,
      "loss": 0.0557,
      "step": 59210
    },
    {
      "epoch": 5.646705627441406e-05,
      "model_forward_time": 2.9259793758392334,
      "step": 59210
    },
    {
      "dynamics_loss_avg": 0.04989780243486166,
      "epoch": 5.646705627441406e-05,
      "step": 59210
    },
    {
      "action_loss_avg": 0.0014801534212892874,
      "epoch": 5.646705627441406e-05,
      "step": 59210
    },
    {
      "epoch": 5.646705627441406e-05,
      "step": 59210,
      "training_step_time": 8.559227466583252
    },
    {
      "epoch": 5.646800994873047e-05,
      "model_forward_time": 1.7618250846862793,
      "step": 59211
    },
    {
      "epoch": 5.646800994873047e-05,
      "step": 59211,
      "training_step_time": 7.636327028274536
    },
    {
      "epoch": 5.646896362304688e-05,
      "model_forward_time": 1.7626147270202637,
      "step": 59212
    },
    {
      "epoch": 5.646896362304688e-05,
      "step": 59212,
      "training_step_time": 7.107628107070923
    },
    {
      "epoch": 5.646991729736328e-05,
      "model_forward_time": 1.7464308738708496,
      "step": 59213
    },
    {
      "epoch": 5.646991729736328e-05,
      "step": 59213,
      "training_step_time": 7.537890911102295
    },
    {
      "epoch": 5.6470870971679686e-05,
      "model_forward_time": 1.7482614517211914,
      "step": 59214
    },
    {
      "epoch": 5.6470870971679686e-05,
      "step": 59214,
      "training_step_time": 6.881056785583496
    },
    {
      "epoch": 5.64718246459961e-05,
      "model_forward_time": 1.7593867778778076,
      "step": 59215
    },
    {
      "epoch": 5.64718246459961e-05,
      "step": 59215,
      "training_step_time": 7.222044467926025
    },
    {
      "epoch": 5.64727783203125e-05,
      "model_forward_time": 1.4048943519592285,
      "step": 59216
    },
    {
      "epoch": 5.64727783203125e-05,
      "step": 59216,
      "training_step_time": 6.849567890167236
    },
    {
      "epoch": 5.6473731994628905e-05,
      "model_forward_time": 1.7575335502624512,
      "step": 59217
    },
    {
      "epoch": 5.6473731994628905e-05,
      "step": 59217,
      "training_step_time": 8.9696946144104
    },
    {
      "epoch": 5.6474685668945316e-05,
      "model_forward_time": 1.8271043300628662,
      "step": 59218
    },
    {
      "epoch": 5.6474685668945316e-05,
      "step": 59218,
      "training_step_time": 8.255512237548828
    },
    {
      "epoch": 5.647563934326172e-05,
      "model_forward_time": 2.9091670513153076,
      "step": 59219
    },
    {
      "epoch": 5.647563934326172e-05,
      "step": 59219,
      "training_step_time": 10.551065444946289
    },
    {
      "epoch": 5.6476593017578124e-05,
      "grad_norm": 0.13770323991775513,
      "learning_rate": 3.898152464634493e-06,
      "loss": 0.061,
      "step": 59220
    },
    {
      "epoch": 5.6476593017578124e-05,
      "model_forward_time": 1.7746233940124512,
      "step": 59220
    },
    {
      "dynamics_loss_avg": 0.054812602885067466,
      "epoch": 5.6476593017578124e-05,
      "step": 59220
    },
    {
      "action_loss_avg": 0.001931532277376391,
      "epoch": 5.6476593017578124e-05,
      "step": 59220
    },
    {
      "epoch": 5.6476593017578124e-05,
      "step": 59220,
      "training_step_time": 7.677535057067871
    },
    {
      "epoch": 5.647754669189453e-05,
      "model_forward_time": 1.389033317565918,
      "step": 59221
    },
    {
      "epoch": 5.647754669189453e-05,
      "step": 59221,
      "training_step_time": 8.249810218811035
    },
    {
      "epoch": 5.647850036621094e-05,
      "model_forward_time": 1.7638218402862549,
      "step": 59222
    },
    {
      "epoch": 5.647850036621094e-05,
      "step": 59222,
      "training_step_time": 7.539738655090332
    },
    {
      "epoch": 5.647945404052734e-05,
      "model_forward_time": 1.7630078792572021,
      "step": 59223
    },
    {
      "epoch": 5.647945404052734e-05,
      "step": 59223,
      "training_step_time": 6.81018328666687
    },
    {
      "epoch": 5.648040771484375e-05,
      "model_forward_time": 1.7589409351348877,
      "step": 59224
    },
    {
      "epoch": 5.648040771484375e-05,
      "step": 59224,
      "training_step_time": 7.1289942264556885
    },
    {
      "epoch": 5.648136138916016e-05,
      "model_forward_time": 1.76008939743042,
      "step": 59225
    },
    {
      "epoch": 5.648136138916016e-05,
      "step": 59225,
      "training_step_time": 6.793309450149536
    },
    {
      "epoch": 5.648231506347656e-05,
      "model_forward_time": 1.7609305381774902,
      "step": 59226
    },
    {
      "epoch": 5.648231506347656e-05,
      "step": 59226,
      "training_step_time": 8.567297220230103
    },
    {
      "epoch": 5.6483268737792966e-05,
      "model_forward_time": 1.7996299266815186,
      "step": 59227
    },
    {
      "epoch": 5.6483268737792966e-05,
      "step": 59227,
      "training_step_time": 8.451807975769043
    },
    {
      "epoch": 5.6484222412109376e-05,
      "model_forward_time": 2.823941946029663,
      "step": 59228
    },
    {
      "epoch": 5.6484222412109376e-05,
      "step": 59228,
      "training_step_time": 10.833109617233276
    },
    {
      "epoch": 5.648517608642578e-05,
      "model_forward_time": 1.7595973014831543,
      "step": 59229
    },
    {
      "epoch": 5.648517608642578e-05,
      "step": 59229,
      "training_step_time": 8.272683143615723
    },
    {
      "epoch": 5.6486129760742185e-05,
      "grad_norm": 0.10029257088899612,
      "learning_rate": 3.8965397032274025e-06,
      "loss": 0.061,
      "step": 59230
    },
    {
      "epoch": 5.6486129760742185e-05,
      "model_forward_time": 1.7481298446655273,
      "step": 59230
    },
    {
      "dynamics_loss_avg": 0.06240043845027685,
      "epoch": 5.6486129760742185e-05,
      "step": 59230
    },
    {
      "action_loss_avg": 0.0019085042004007846,
      "epoch": 5.6486129760742185e-05,
      "step": 59230
    },
    {
      "epoch": 5.6486129760742185e-05,
      "step": 59230,
      "training_step_time": 7.719824552536011
    },
    {
      "epoch": 5.6487083435058595e-05,
      "model_forward_time": 1.7553746700286865,
      "step": 59231
    },
    {
      "epoch": 5.6487083435058595e-05,
      "step": 59231,
      "training_step_time": 7.098416566848755
    },
    {
      "epoch": 5.6488037109375e-05,
      "model_forward_time": 1.755187749862671,
      "step": 59232
    },
    {
      "epoch": 5.6488037109375e-05,
      "step": 59232,
      "training_step_time": 7.613995790481567
    },
    {
      "epoch": 5.64889907836914e-05,
      "model_forward_time": 1.7731478214263916,
      "step": 59233
    },
    {
      "epoch": 5.64889907836914e-05,
      "step": 59233,
      "training_step_time": 6.877689599990845
    },
    {
      "epoch": 5.6489944458007814e-05,
      "model_forward_time": 1.7675340175628662,
      "step": 59234
    },
    {
      "epoch": 5.6489944458007814e-05,
      "step": 59234,
      "training_step_time": 7.261270523071289
    },
    {
      "epoch": 5.649089813232422e-05,
      "model_forward_time": 1.7525899410247803,
      "step": 59235
    },
    {
      "epoch": 5.649089813232422e-05,
      "step": 59235,
      "training_step_time": 7.333253622055054
    },
    {
      "epoch": 5.649185180664062e-05,
      "model_forward_time": 1.7699329853057861,
      "step": 59236
    },
    {
      "epoch": 5.649185180664062e-05,
      "step": 59236,
      "training_step_time": 8.919515609741211
    },
    {
      "epoch": 5.649280548095703e-05,
      "model_forward_time": 1.772007942199707,
      "step": 59237
    },
    {
      "epoch": 5.649280548095703e-05,
      "step": 59237,
      "training_step_time": 8.84573745727539
    },
    {
      "epoch": 5.649375915527344e-05,
      "model_forward_time": 2.3303961753845215,
      "step": 59238
    },
    {
      "epoch": 5.649375915527344e-05,
      "step": 59238,
      "training_step_time": 7.8800458908081055
    },
    {
      "epoch": 5.649471282958984e-05,
      "model_forward_time": 1.7120637893676758,
      "step": 59239
    },
    {
      "epoch": 5.649471282958984e-05,
      "step": 59239,
      "training_step_time": 7.722424745559692
    },
    {
      "epoch": 5.649566650390625e-05,
      "grad_norm": 0.1283227950334549,
      "learning_rate": 3.894927062493074e-06,
      "loss": 0.0583,
      "step": 59240
    },
    {
      "epoch": 5.649566650390625e-05,
      "model_forward_time": 1.7292332649230957,
      "step": 59240
    },
    {
      "dynamics_loss_avg": 0.053976122103631494,
      "epoch": 5.649566650390625e-05,
      "step": 59240
    },
    {
      "action_loss_avg": 0.001094876290881075,
      "epoch": 5.649566650390625e-05,
      "step": 59240
    },
    {
      "epoch": 5.649566650390625e-05,
      "step": 59240,
      "training_step_time": 7.106546640396118
    },
    {
      "epoch": 5.6496620178222656e-05,
      "model_forward_time": 1.7311248779296875,
      "step": 59241
    },
    {
      "epoch": 5.6496620178222656e-05,
      "step": 59241,
      "training_step_time": 7.6189751625061035
    },
    {
      "epoch": 5.649757385253906e-05,
      "model_forward_time": 1.726426362991333,
      "step": 59242
    },
    {
      "epoch": 5.649757385253906e-05,
      "step": 59242,
      "training_step_time": 6.875810861587524
    },
    {
      "epoch": 5.649852752685547e-05,
      "model_forward_time": 1.7238993644714355,
      "step": 59243
    },
    {
      "epoch": 5.649852752685547e-05,
      "step": 59243,
      "training_step_time": 7.250092029571533
    },
    {
      "epoch": 5.6499481201171875e-05,
      "model_forward_time": 1.7289175987243652,
      "step": 59244
    },
    {
      "epoch": 5.6499481201171875e-05,
      "step": 59244,
      "training_step_time": 7.31211519241333
    },
    {
      "epoch": 5.650043487548828e-05,
      "model_forward_time": 1.7222583293914795,
      "step": 59245
    },
    {
      "epoch": 5.650043487548828e-05,
      "step": 59245,
      "training_step_time": 8.524672031402588
    },
    {
      "epoch": 5.650138854980469e-05,
      "model_forward_time": 1.737941026687622,
      "step": 59246
    },
    {
      "epoch": 5.650138854980469e-05,
      "step": 59246,
      "training_step_time": 8.389293670654297
    },
    {
      "epoch": 5.6502342224121094e-05,
      "model_forward_time": 2.337184429168701,
      "step": 59247
    },
    {
      "epoch": 5.6502342224121094e-05,
      "step": 59247,
      "training_step_time": 8.268812656402588
    },
    {
      "epoch": 5.65032958984375e-05,
      "model_forward_time": 1.7486672401428223,
      "step": 59248
    },
    {
      "epoch": 5.65032958984375e-05,
      "step": 59248,
      "training_step_time": 7.820389032363892
    },
    {
      "epoch": 5.650424957275391e-05,
      "model_forward_time": 1.762105941772461,
      "step": 59249
    },
    {
      "epoch": 5.650424957275391e-05,
      "step": 59249,
      "training_step_time": 7.467876672744751
    },
    {
      "epoch": 5.650520324707031e-05,
      "grad_norm": 0.09860927611589432,
      "learning_rate": 3.893314542607859e-06,
      "loss": 0.0602,
      "step": 59250
    },
    {
      "epoch": 5.650520324707031e-05,
      "model_forward_time": 1.7469706535339355,
      "step": 59250
    },
    {
      "dynamics_loss_avg": 0.05493233092129231,
      "epoch": 5.650520324707031e-05,
      "step": 59250
    },
    {
      "action_loss_avg": 0.0014646406460087746,
      "epoch": 5.650520324707031e-05,
      "step": 59250
    },
    {
      "epoch": 5.650520324707031e-05,
      "step": 59250,
      "training_step_time": 7.553739786148071
    },
    {
      "epoch": 5.6506156921386717e-05,
      "model_forward_time": 1.761167287826538,
      "step": 59251
    },
    {
      "epoch": 5.6506156921386717e-05,
      "step": 59251,
      "training_step_time": 6.818840980529785
    },
    {
      "epoch": 5.650711059570313e-05,
      "model_forward_time": 1.757420539855957,
      "step": 59252
    },
    {
      "epoch": 5.650711059570313e-05,
      "step": 59252,
      "training_step_time": 7.238617658615112
    },
    {
      "epoch": 5.650806427001953e-05,
      "model_forward_time": 1.7630517482757568,
      "step": 59253
    },
    {
      "epoch": 5.650806427001953e-05,
      "step": 59253,
      "training_step_time": 7.306213855743408
    },
    {
      "epoch": 5.6509017944335935e-05,
      "model_forward_time": 1.7571330070495605,
      "step": 59254
    },
    {
      "epoch": 5.6509017944335935e-05,
      "step": 59254,
      "training_step_time": 7.860098600387573
    },
    {
      "epoch": 5.6509971618652346e-05,
      "model_forward_time": 1.7629046440124512,
      "step": 59255
    },
    {
      "epoch": 5.6509971618652346e-05,
      "step": 59255,
      "training_step_time": 7.81596565246582
    },
    {
      "epoch": 5.651092529296875e-05,
      "model_forward_time": 2.3443639278411865,
      "step": 59256
    },
    {
      "epoch": 5.651092529296875e-05,
      "step": 59256,
      "training_step_time": 8.31681227684021
    },
    {
      "epoch": 5.6511878967285154e-05,
      "model_forward_time": 1.7572402954101562,
      "step": 59257
    },
    {
      "epoch": 5.6511878967285154e-05,
      "step": 59257,
      "training_step_time": 7.768390893936157
    },
    {
      "epoch": 5.6512832641601565e-05,
      "model_forward_time": 1.7440533638000488,
      "step": 59258
    },
    {
      "epoch": 5.6512832641601565e-05,
      "step": 59258,
      "training_step_time": 7.7380595207214355
    },
    {
      "epoch": 5.651378631591797e-05,
      "model_forward_time": 1.7561957836151123,
      "step": 59259
    },
    {
      "epoch": 5.651378631591797e-05,
      "step": 59259,
      "training_step_time": 7.079495191574097
    },
    {
      "epoch": 5.651473999023437e-05,
      "grad_norm": 0.37336447834968567,
      "learning_rate": 3.891702143748103e-06,
      "loss": 0.0598,
      "step": 59260
    },
    {
      "epoch": 5.651473999023437e-05,
      "model_forward_time": 1.74515962600708,
      "step": 59260
    },
    {
      "dynamics_loss_avg": 0.07533830590546131,
      "epoch": 5.651473999023437e-05,
      "step": 59260
    },
    {
      "action_loss_avg": 0.0017383422586135567,
      "epoch": 5.651473999023437e-05,
      "step": 59260
    },
    {
      "epoch": 5.651473999023437e-05,
      "step": 59260,
      "training_step_time": 7.570847749710083
    },
    {
      "epoch": 5.6515693664550784e-05,
      "model_forward_time": 1.750802755355835,
      "step": 59261
    },
    {
      "epoch": 5.6515693664550784e-05,
      "step": 59261,
      "training_step_time": 6.873293161392212
    },
    {
      "epoch": 5.651664733886719e-05,
      "model_forward_time": 1.7594444751739502,
      "step": 59262
    },
    {
      "epoch": 5.651664733886719e-05,
      "step": 59262,
      "training_step_time": 7.367610692977905
    },
    {
      "epoch": 5.651760101318359e-05,
      "model_forward_time": 1.7611932754516602,
      "step": 59263
    },
    {
      "epoch": 5.651760101318359e-05,
      "step": 59263,
      "training_step_time": 7.9992899894714355
    },
    {
      "epoch": 5.65185546875e-05,
      "model_forward_time": 1.7458271980285645,
      "step": 59264
    },
    {
      "epoch": 5.65185546875e-05,
      "step": 59264,
      "training_step_time": 8.896536827087402
    },
    {
      "epoch": 5.651950836181641e-05,
      "model_forward_time": 2.3727879524230957,
      "step": 59265
    },
    {
      "epoch": 5.651950836181641e-05,
      "step": 59265,
      "training_step_time": 11.713512659072876
    },
    {
      "epoch": 5.652046203613281e-05,
      "model_forward_time": 1.756420612335205,
      "step": 59266
    },
    {
      "epoch": 5.652046203613281e-05,
      "step": 59266,
      "training_step_time": 9.45051622390747
    },
    {
      "epoch": 5.652141571044922e-05,
      "model_forward_time": 1.75057053565979,
      "step": 59267
    },
    {
      "epoch": 5.652141571044922e-05,
      "step": 59267,
      "training_step_time": 7.735277891159058
    },
    {
      "epoch": 5.6522369384765626e-05,
      "model_forward_time": 1.766084909439087,
      "step": 59268
    },
    {
      "epoch": 5.6522369384765626e-05,
      "step": 59268,
      "training_step_time": 7.170264005661011
    },
    {
      "epoch": 5.652332305908203e-05,
      "model_forward_time": 1.7577426433563232,
      "step": 59269
    },
    {
      "epoch": 5.652332305908203e-05,
      "step": 59269,
      "training_step_time": 7.582862138748169
    },
    {
      "epoch": 5.652427673339844e-05,
      "grad_norm": 0.10206124186515808,
      "learning_rate": 3.890089866090133e-06,
      "loss": 0.0587,
      "step": 59270
    },
    {
      "epoch": 5.652427673339844e-05,
      "model_forward_time": 1.7555210590362549,
      "step": 59270
    },
    {
      "dynamics_loss_avg": 0.057883730996400116,
      "epoch": 5.652427673339844e-05,
      "step": 59270
    },
    {
      "action_loss_avg": 0.0017216173815540969,
      "epoch": 5.652427673339844e-05,
      "step": 59270
    },
    {
      "epoch": 5.652427673339844e-05,
      "step": 59270,
      "training_step_time": 6.906717300415039
    },
    {
      "epoch": 5.6525230407714845e-05,
      "model_forward_time": 1.762913703918457,
      "step": 59271
    },
    {
      "epoch": 5.6525230407714845e-05,
      "step": 59271,
      "training_step_time": 7.342695951461792
    },
    {
      "epoch": 5.652618408203125e-05,
      "model_forward_time": 1.7721898555755615,
      "step": 59272
    },
    {
      "epoch": 5.652618408203125e-05,
      "step": 59272,
      "training_step_time": 8.477761507034302
    },
    {
      "epoch": 5.652713775634766e-05,
      "model_forward_time": 1.75974440574646,
      "step": 59273
    },
    {
      "epoch": 5.652713775634766e-05,
      "step": 59273,
      "training_step_time": 8.666297435760498
    },
    {
      "epoch": 5.652809143066406e-05,
      "model_forward_time": 2.3882765769958496,
      "step": 59274
    },
    {
      "epoch": 5.652809143066406e-05,
      "step": 59274,
      "training_step_time": 9.409655570983887
    },
    {
      "epoch": 5.652904510498047e-05,
      "model_forward_time": 1.7670562267303467,
      "step": 59275
    },
    {
      "epoch": 5.652904510498047e-05,
      "step": 59275,
      "training_step_time": 7.691269397735596
    },
    {
      "epoch": 5.652999877929688e-05,
      "model_forward_time": 1.7694709300994873,
      "step": 59276
    },
    {
      "epoch": 5.652999877929688e-05,
      "step": 59276,
      "training_step_time": 7.7908594608306885
    },
    {
      "epoch": 5.653095245361328e-05,
      "model_forward_time": 1.7580759525299072,
      "step": 59277
    },
    {
      "epoch": 5.653095245361328e-05,
      "step": 59277,
      "training_step_time": 7.176998138427734
    },
    {
      "epoch": 5.6531906127929686e-05,
      "model_forward_time": 1.762415885925293,
      "step": 59278
    },
    {
      "epoch": 5.6531906127929686e-05,
      "step": 59278,
      "training_step_time": 7.4274187088012695
    },
    {
      "epoch": 5.65328598022461e-05,
      "model_forward_time": 1.7626020908355713,
      "step": 59279
    },
    {
      "epoch": 5.65328598022461e-05,
      "step": 59279,
      "training_step_time": 6.8622517585754395
    },
    {
      "epoch": 5.65338134765625e-05,
      "grad_norm": 0.11226596683263779,
      "learning_rate": 3.888477709810268e-06,
      "loss": 0.0576,
      "step": 59280
    },
    {
      "epoch": 5.65338134765625e-05,
      "model_forward_time": 1.7577128410339355,
      "step": 59280
    },
    {
      "dynamics_loss_avg": 0.039465104974806306,
      "epoch": 5.65338134765625e-05,
      "step": 59280
    },
    {
      "action_loss_avg": 0.001512761703634169,
      "epoch": 5.65338134765625e-05,
      "step": 59280
    },
    {
      "epoch": 5.65338134765625e-05,
      "step": 59280,
      "training_step_time": 7.289759159088135
    },
    {
      "epoch": 5.6534767150878905e-05,
      "model_forward_time": 1.7616336345672607,
      "step": 59281
    },
    {
      "epoch": 5.6534767150878905e-05,
      "step": 59281,
      "training_step_time": 7.538248538970947
    },
    {
      "epoch": 5.653572082519531e-05,
      "model_forward_time": 1.7606780529022217,
      "step": 59282
    },
    {
      "epoch": 5.653572082519531e-05,
      "step": 59282,
      "training_step_time": 9.441929578781128
    },
    {
      "epoch": 5.653667449951172e-05,
      "model_forward_time": 1.8096222877502441,
      "step": 59283
    },
    {
      "epoch": 5.653667449951172e-05,
      "step": 59283,
      "training_step_time": 8.648253679275513
    },
    {
      "epoch": 5.6537628173828124e-05,
      "model_forward_time": 2.3300135135650635,
      "step": 59284
    },
    {
      "epoch": 5.6537628173828124e-05,
      "step": 59284,
      "training_step_time": 8.458710670471191
    },
    {
      "epoch": 5.653858184814453e-05,
      "model_forward_time": 1.765028715133667,
      "step": 59285
    },
    {
      "epoch": 5.653858184814453e-05,
      "step": 59285,
      "training_step_time": 7.776434421539307
    },
    {
      "epoch": 5.653953552246094e-05,
      "model_forward_time": 1.756371021270752,
      "step": 59286
    },
    {
      "epoch": 5.653953552246094e-05,
      "step": 59286,
      "training_step_time": 7.155121088027954
    },
    {
      "epoch": 5.654048919677734e-05,
      "model_forward_time": 1.758385419845581,
      "step": 59287
    },
    {
      "epoch": 5.654048919677734e-05,
      "step": 59287,
      "training_step_time": 6.865693807601929
    },
    {
      "epoch": 5.654144287109375e-05,
      "model_forward_time": 1.761594295501709,
      "step": 59288
    },
    {
      "epoch": 5.654144287109375e-05,
      "step": 59288,
      "training_step_time": 7.479719877243042
    },
    {
      "epoch": 5.654239654541016e-05,
      "model_forward_time": 1.765162467956543,
      "step": 59289
    },
    {
      "epoch": 5.654239654541016e-05,
      "step": 59289,
      "training_step_time": 7.281959772109985
    },
    {
      "epoch": 5.654335021972656e-05,
      "grad_norm": 0.11164862662553787,
      "learning_rate": 3.886865675084807e-06,
      "loss": 0.0604,
      "step": 59290
    },
    {
      "epoch": 5.654335021972656e-05,
      "model_forward_time": 1.699625015258789,
      "step": 59290
    },
    {
      "dynamics_loss_avg": 0.0605190172791481,
      "epoch": 5.654335021972656e-05,
      "step": 59290
    },
    {
      "action_loss_avg": 0.0017661146470345558,
      "epoch": 5.654335021972656e-05,
      "step": 59290
    },
    {
      "epoch": 5.654335021972656e-05,
      "step": 59290,
      "training_step_time": 7.2543535232543945
    },
    {
      "epoch": 5.6544303894042966e-05,
      "model_forward_time": 1.721360445022583,
      "step": 59291
    },
    {
      "epoch": 5.6544303894042966e-05,
      "step": 59291,
      "training_step_time": 7.922751188278198
    },
    {
      "epoch": 5.6545257568359377e-05,
      "model_forward_time": 1.737295150756836,
      "step": 59292
    },
    {
      "epoch": 5.6545257568359377e-05,
      "step": 59292,
      "training_step_time": 9.233335018157959
    },
    {
      "epoch": 5.654621124267578e-05,
      "model_forward_time": 2.258897066116333,
      "step": 59293
    },
    {
      "epoch": 5.654621124267578e-05,
      "step": 59293,
      "training_step_time": 7.72144627571106
    },
    {
      "epoch": 5.6547164916992185e-05,
      "model_forward_time": 1.726579189300537,
      "step": 59294
    },
    {
      "epoch": 5.6547164916992185e-05,
      "step": 59294,
      "training_step_time": 8.470672607421875
    },
    {
      "epoch": 5.6548118591308595e-05,
      "model_forward_time": 1.7194383144378662,
      "step": 59295
    },
    {
      "epoch": 5.6548118591308595e-05,
      "step": 59295,
      "training_step_time": 8.670695304870605
    },
    {
      "epoch": 5.6549072265625e-05,
      "model_forward_time": 1.7321197986602783,
      "step": 59296
    },
    {
      "epoch": 5.6549072265625e-05,
      "step": 59296,
      "training_step_time": 6.889745712280273
    },
    {
      "epoch": 5.6550025939941403e-05,
      "model_forward_time": 1.717280387878418,
      "step": 59297
    },
    {
      "epoch": 5.6550025939941403e-05,
      "step": 59297,
      "training_step_time": 7.546461343765259
    },
    {
      "epoch": 5.6550979614257814e-05,
      "model_forward_time": 1.7242577075958252,
      "step": 59298
    },
    {
      "epoch": 5.6550979614257814e-05,
      "step": 59298,
      "training_step_time": 7.311851739883423
    },
    {
      "epoch": 5.655193328857422e-05,
      "model_forward_time": 1.3434665203094482,
      "step": 59299
    },
    {
      "epoch": 5.655193328857422e-05,
      "step": 59299,
      "training_step_time": 7.288186550140381
    },
    {
      "epoch": 5.655288696289062e-05,
      "grad_norm": 0.15210852026939392,
      "learning_rate": 3.8852537620900435e-06,
      "loss": 0.0613,
      "step": 59300
    },
    {
      "epoch": 5.655288696289062e-05,
      "model_forward_time": 1.725451946258545,
      "step": 59300
    },
    {
      "dynamics_loss_avg": 0.05145794842392206,
      "epoch": 5.655288696289062e-05,
      "step": 59300
    },
    {
      "action_loss_avg": 0.001421694119926542,
      "epoch": 5.655288696289062e-05,
      "step": 59300
    },
    {
      "epoch": 5.655288696289062e-05,
      "step": 59300,
      "training_step_time": 8.660122871398926
    },
    {
      "epoch": 5.655384063720703e-05,
      "model_forward_time": 1.7189967632293701,
      "step": 59301
    },
    {
      "epoch": 5.655384063720703e-05,
      "step": 59301,
      "training_step_time": 8.459308385848999
    },
    {
      "epoch": 5.655479431152344e-05,
      "model_forward_time": 2.3068015575408936,
      "step": 59302
    },
    {
      "epoch": 5.655479431152344e-05,
      "step": 59302,
      "training_step_time": 9.655662775039673
    },
    {
      "epoch": 5.655574798583984e-05,
      "model_forward_time": 1.7221267223358154,
      "step": 59303
    },
    {
      "epoch": 5.655574798583984e-05,
      "step": 59303,
      "training_step_time": 7.779993772506714
    },
    {
      "epoch": 5.655670166015625e-05,
      "model_forward_time": 1.7143774032592773,
      "step": 59304
    },
    {
      "epoch": 5.655670166015625e-05,
      "step": 59304,
      "training_step_time": 7.635044813156128
    },
    {
      "epoch": 5.6557655334472656e-05,
      "model_forward_time": 1.7286148071289062,
      "step": 59305
    },
    {
      "epoch": 5.6557655334472656e-05,
      "step": 59305,
      "training_step_time": 7.114288091659546
    },
    {
      "epoch": 5.655860900878906e-05,
      "model_forward_time": 1.7219243049621582,
      "step": 59306
    },
    {
      "epoch": 5.655860900878906e-05,
      "step": 59306,
      "training_step_time": 7.549720287322998
    },
    {
      "epoch": 5.655956268310547e-05,
      "model_forward_time": 1.722884178161621,
      "step": 59307
    },
    {
      "epoch": 5.655956268310547e-05,
      "step": 59307,
      "training_step_time": 8.149586915969849
    },
    {
      "epoch": 5.6560516357421875e-05,
      "model_forward_time": 1.3491308689117432,
      "step": 59308
    },
    {
      "epoch": 5.6560516357421875e-05,
      "step": 59308,
      "training_step_time": 7.272463083267212
    },
    {
      "epoch": 5.656147003173828e-05,
      "model_forward_time": 1.725541114807129,
      "step": 59309
    },
    {
      "epoch": 5.656147003173828e-05,
      "step": 59309,
      "training_step_time": 8.141748189926147
    },
    {
      "epoch": 5.656242370605469e-05,
      "grad_norm": 0.0983969196677208,
      "learning_rate": 3.883641971002253e-06,
      "loss": 0.0594,
      "step": 59310
    },
    {
      "epoch": 5.656242370605469e-05,
      "model_forward_time": 1.722426176071167,
      "step": 59310
    },
    {
      "dynamics_loss_avg": 0.06279399245977402,
      "epoch": 5.656242370605469e-05,
      "step": 59310
    },
    {
      "action_loss_avg": 0.0016291297099087388,
      "epoch": 5.656242370605469e-05,
      "step": 59310
    },
    {
      "epoch": 5.656242370605469e-05,
      "step": 59310,
      "training_step_time": 9.872128009796143
    },
    {
      "epoch": 5.6563377380371094e-05,
      "model_forward_time": 2.333629608154297,
      "step": 59311
    },
    {
      "epoch": 5.6563377380371094e-05,
      "step": 59311,
      "training_step_time": 8.82079291343689
    },
    {
      "epoch": 5.65643310546875e-05,
      "model_forward_time": 1.7315483093261719,
      "step": 59312
    },
    {
      "epoch": 5.65643310546875e-05,
      "step": 59312,
      "training_step_time": 8.61964464187622
    },
    {
      "epoch": 5.656528472900391e-05,
      "model_forward_time": 1.7179756164550781,
      "step": 59313
    },
    {
      "epoch": 5.656528472900391e-05,
      "step": 59313,
      "training_step_time": 7.7747790813446045
    },
    {
      "epoch": 5.656623840332031e-05,
      "model_forward_time": 1.740788221359253,
      "step": 59314
    },
    {
      "epoch": 5.656623840332031e-05,
      "step": 59314,
      "training_step_time": 7.707211256027222
    },
    {
      "epoch": 5.6567192077636717e-05,
      "model_forward_time": 1.729236364364624,
      "step": 59315
    },
    {
      "epoch": 5.6567192077636717e-05,
      "step": 59315,
      "training_step_time": 6.8218324184417725
    },
    {
      "epoch": 5.656814575195313e-05,
      "model_forward_time": 1.7289888858795166,
      "step": 59316
    },
    {
      "epoch": 5.656814575195313e-05,
      "step": 59316,
      "training_step_time": 7.588346719741821
    },
    {
      "epoch": 5.656909942626953e-05,
      "model_forward_time": 1.7271456718444824,
      "step": 59317
    },
    {
      "epoch": 5.656909942626953e-05,
      "step": 59317,
      "training_step_time": 7.299888849258423
    },
    {
      "epoch": 5.6570053100585935e-05,
      "model_forward_time": 1.726454734802246,
      "step": 59318
    },
    {
      "epoch": 5.6570053100585935e-05,
      "step": 59318,
      "training_step_time": 8.568212032318115
    },
    {
      "epoch": 5.6571006774902346e-05,
      "model_forward_time": 1.7234268188476562,
      "step": 59319
    },
    {
      "epoch": 5.6571006774902346e-05,
      "step": 59319,
      "training_step_time": 8.960363864898682
    },
    {
      "epoch": 5.657196044921875e-05,
      "grad_norm": 0.12565089762210846,
      "learning_rate": 3.882030301997699e-06,
      "loss": 0.0597,
      "step": 59320
    },
    {
      "epoch": 5.657196044921875e-05,
      "model_forward_time": 1.7983324527740479,
      "step": 59320
    },
    {
      "dynamics_loss_avg": 0.06682295501232147,
      "epoch": 5.657196044921875e-05,
      "step": 59320
    },
    {
      "action_loss_avg": 0.001768814556999132,
      "epoch": 5.657196044921875e-05,
      "step": 59320
    },
    {
      "epoch": 5.657196044921875e-05,
      "step": 59320,
      "training_step_time": 8.62612795829773
    },
    {
      "epoch": 5.6572914123535154e-05,
      "model_forward_time": 2.269860029220581,
      "step": 59321
    },
    {
      "epoch": 5.6572914123535154e-05,
      "step": 59321,
      "training_step_time": 7.72531795501709
    },
    {
      "epoch": 5.6573867797851565e-05,
      "model_forward_time": 1.7247607707977295,
      "step": 59322
    },
    {
      "epoch": 5.6573867797851565e-05,
      "step": 59322,
      "training_step_time": 7.779541254043579
    },
    {
      "epoch": 5.657482147216797e-05,
      "model_forward_time": 1.7374353408813477,
      "step": 59323
    },
    {
      "epoch": 5.657482147216797e-05,
      "step": 59323,
      "training_step_time": 7.11872673034668
    },
    {
      "epoch": 5.657577514648437e-05,
      "model_forward_time": 1.7336030006408691,
      "step": 59324
    },
    {
      "epoch": 5.657577514648437e-05,
      "step": 59324,
      "training_step_time": 7.1978631019592285
    },
    {
      "epoch": 5.6576728820800784e-05,
      "model_forward_time": 1.7311105728149414,
      "step": 59325
    },
    {
      "epoch": 5.6576728820800784e-05,
      "step": 59325,
      "training_step_time": 7.6616950035095215
    },
    {
      "epoch": 5.657768249511719e-05,
      "model_forward_time": 1.728485345840454,
      "step": 59326
    },
    {
      "epoch": 5.657768249511719e-05,
      "step": 59326,
      "training_step_time": 7.325637102127075
    },
    {
      "epoch": 5.657863616943359e-05,
      "model_forward_time": 1.7330377101898193,
      "step": 59327
    },
    {
      "epoch": 5.657863616943359e-05,
      "step": 59327,
      "training_step_time": 7.500385522842407
    },
    {
      "epoch": 5.657958984375e-05,
      "model_forward_time": 1.7270581722259521,
      "step": 59328
    },
    {
      "epoch": 5.657958984375e-05,
      "step": 59328,
      "training_step_time": 8.619142770767212
    },
    {
      "epoch": 5.658054351806641e-05,
      "model_forward_time": 1.7843549251556396,
      "step": 59329
    },
    {
      "epoch": 5.658054351806641e-05,
      "step": 59329,
      "training_step_time": 8.829113006591797
    },
    {
      "epoch": 5.658149719238281e-05,
      "grad_norm": 0.09603821486234665,
      "learning_rate": 3.880418755252629e-06,
      "loss": 0.0593,
      "step": 59330
    },
    {
      "epoch": 5.658149719238281e-05,
      "model_forward_time": 2.2831504344940186,
      "step": 59330
    },
    {
      "dynamics_loss_avg": 0.060761979408562185,
      "epoch": 5.658149719238281e-05,
      "step": 59330
    },
    {
      "action_loss_avg": 0.0013958928815554828,
      "epoch": 5.658149719238281e-05,
      "step": 59330
    },
    {
      "epoch": 5.658149719238281e-05,
      "step": 59330,
      "training_step_time": 8.23020052909851
    },
    {
      "epoch": 5.658245086669922e-05,
      "model_forward_time": 1.3411028385162354,
      "step": 59331
    },
    {
      "epoch": 5.658245086669922e-05,
      "step": 59331,
      "training_step_time": 7.840884208679199
    },
    {
      "epoch": 5.6583404541015626e-05,
      "model_forward_time": 1.7352440357208252,
      "step": 59332
    },
    {
      "epoch": 5.6583404541015626e-05,
      "step": 59332,
      "training_step_time": 7.081893682479858
    },
    {
      "epoch": 5.658435821533203e-05,
      "model_forward_time": 1.722731113433838,
      "step": 59333
    },
    {
      "epoch": 5.658435821533203e-05,
      "step": 59333,
      "training_step_time": 7.2143168449401855
    },
    {
      "epoch": 5.658531188964844e-05,
      "model_forward_time": 1.7273108959197998,
      "step": 59334
    },
    {
      "epoch": 5.658531188964844e-05,
      "step": 59334,
      "training_step_time": 7.54759955406189
    },
    {
      "epoch": 5.6586265563964845e-05,
      "model_forward_time": 1.723935604095459,
      "step": 59335
    },
    {
      "epoch": 5.6586265563964845e-05,
      "step": 59335,
      "training_step_time": 7.273552894592285
    },
    {
      "epoch": 5.658721923828125e-05,
      "model_forward_time": 1.7155654430389404,
      "step": 59336
    },
    {
      "epoch": 5.658721923828125e-05,
      "step": 59336,
      "training_step_time": 7.356838703155518
    },
    {
      "epoch": 5.658817291259766e-05,
      "model_forward_time": 1.7216651439666748,
      "step": 59337
    },
    {
      "epoch": 5.658817291259766e-05,
      "step": 59337,
      "training_step_time": 7.548438310623169
    },
    {
      "epoch": 5.6589126586914063e-05,
      "model_forward_time": 1.768042802810669,
      "step": 59338
    },
    {
      "epoch": 5.6589126586914063e-05,
      "step": 59338,
      "training_step_time": 8.355790615081787
    },
    {
      "epoch": 5.659008026123047e-05,
      "model_forward_time": 2.289952516555786,
      "step": 59339
    },
    {
      "epoch": 5.659008026123047e-05,
      "step": 59339,
      "training_step_time": 8.458173036575317
    },
    {
      "epoch": 5.659103393554688e-05,
      "grad_norm": 0.13929393887519836,
      "learning_rate": 3.87880733094328e-06,
      "loss": 0.059,
      "step": 59340
    },
    {
      "epoch": 5.659103393554688e-05,
      "model_forward_time": 1.720716953277588,
      "step": 59340
    },
    {
      "dynamics_loss_avg": 0.04968967828899622,
      "epoch": 5.659103393554688e-05,
      "step": 59340
    },
    {
      "action_loss_avg": 0.00128577365539968,
      "epoch": 5.659103393554688e-05,
      "step": 59340
    },
    {
      "epoch": 5.659103393554688e-05,
      "step": 59340,
      "training_step_time": 8.534871578216553
    },
    {
      "epoch": 5.659198760986328e-05,
      "model_forward_time": 1.736534833908081,
      "step": 59341
    },
    {
      "epoch": 5.659198760986328e-05,
      "step": 59341,
      "training_step_time": 7.069644212722778
    },
    {
      "epoch": 5.6592941284179686e-05,
      "model_forward_time": 1.7260138988494873,
      "step": 59342
    },
    {
      "epoch": 5.6592941284179686e-05,
      "step": 59342,
      "training_step_time": 7.135150909423828
    },
    {
      "epoch": 5.65938949584961e-05,
      "model_forward_time": 1.7211921215057373,
      "step": 59343
    },
    {
      "epoch": 5.65938949584961e-05,
      "step": 59343,
      "training_step_time": 6.848311185836792
    },
    {
      "epoch": 5.65948486328125e-05,
      "model_forward_time": 1.719423770904541,
      "step": 59344
    },
    {
      "epoch": 5.65948486328125e-05,
      "step": 59344,
      "training_step_time": 7.545329809188843
    },
    {
      "epoch": 5.6595802307128905e-05,
      "model_forward_time": 1.7317821979522705,
      "step": 59345
    },
    {
      "epoch": 5.6595802307128905e-05,
      "step": 59345,
      "training_step_time": 7.33138108253479
    },
    {
      "epoch": 5.659675598144531e-05,
      "model_forward_time": 1.3544809818267822,
      "step": 59346
    },
    {
      "epoch": 5.659675598144531e-05,
      "step": 59346,
      "training_step_time": 8.224921941757202
    },
    {
      "epoch": 5.659770965576172e-05,
      "model_forward_time": 1.7516772747039795,
      "step": 59347
    },
    {
      "epoch": 5.659770965576172e-05,
      "step": 59347,
      "training_step_time": 8.696379899978638
    },
    {
      "epoch": 5.6598663330078124e-05,
      "model_forward_time": 2.2634692192077637,
      "step": 59348
    },
    {
      "epoch": 5.6598663330078124e-05,
      "step": 59348,
      "training_step_time": 8.660143852233887
    },
    {
      "epoch": 5.659961700439453e-05,
      "model_forward_time": 1.7273643016815186,
      "step": 59349
    },
    {
      "epoch": 5.659961700439453e-05,
      "step": 59349,
      "training_step_time": 7.762099981307983
    },
    {
      "epoch": 5.660057067871094e-05,
      "grad_norm": 0.148593932390213,
      "learning_rate": 3.877196029245877e-06,
      "loss": 0.0618,
      "step": 59350
    },
    {
      "epoch": 5.660057067871094e-05,
      "model_forward_time": 1.7200732231140137,
      "step": 59350
    },
    {
      "dynamics_loss_avg": 0.05779646206647158,
      "epoch": 5.660057067871094e-05,
      "step": 59350
    },
    {
      "action_loss_avg": 0.001994796801591292,
      "epoch": 5.660057067871094e-05,
      "step": 59350
    },
    {
      "epoch": 5.660057067871094e-05,
      "step": 59350,
      "training_step_time": 7.511211633682251
    },
    {
      "epoch": 5.660152435302734e-05,
      "model_forward_time": 1.7270822525024414,
      "step": 59351
    },
    {
      "epoch": 5.660152435302734e-05,
      "step": 59351,
      "training_step_time": 7.161914825439453
    },
    {
      "epoch": 5.660247802734375e-05,
      "model_forward_time": 1.7234315872192383,
      "step": 59352
    },
    {
      "epoch": 5.660247802734375e-05,
      "step": 59352,
      "training_step_time": 6.874083995819092
    },
    {
      "epoch": 5.660343170166016e-05,
      "model_forward_time": 1.7262637615203857,
      "step": 59353
    },
    {
      "epoch": 5.660343170166016e-05,
      "step": 59353,
      "training_step_time": 7.584006309509277
    },
    {
      "epoch": 5.660438537597656e-05,
      "model_forward_time": 1.7368414402008057,
      "step": 59354
    },
    {
      "epoch": 5.660438537597656e-05,
      "step": 59354,
      "training_step_time": 7.394862174987793
    },
    {
      "epoch": 5.6605339050292966e-05,
      "model_forward_time": 1.7279021739959717,
      "step": 59355
    },
    {
      "epoch": 5.6605339050292966e-05,
      "step": 59355,
      "training_step_time": 7.692654371261597
    },
    {
      "epoch": 5.6606292724609377e-05,
      "model_forward_time": 1.7289726734161377,
      "step": 59356
    },
    {
      "epoch": 5.6606292724609377e-05,
      "step": 59356,
      "training_step_time": 8.606042385101318
    },
    {
      "epoch": 5.660724639892578e-05,
      "model_forward_time": 1.7240774631500244,
      "step": 59357
    },
    {
      "epoch": 5.660724639892578e-05,
      "step": 59357,
      "training_step_time": 8.716683387756348
    },
    {
      "epoch": 5.6608200073242185e-05,
      "model_forward_time": 2.2672226428985596,
      "step": 59358
    },
    {
      "epoch": 5.6608200073242185e-05,
      "step": 59358,
      "training_step_time": 12.321367740631104
    },
    {
      "epoch": 5.6609153747558595e-05,
      "model_forward_time": 1.717566728591919,
      "step": 59359
    },
    {
      "epoch": 5.6609153747558595e-05,
      "step": 59359,
      "training_step_time": 7.834261178970337
    },
    {
      "epoch": 5.6610107421875e-05,
      "grad_norm": 0.14603495597839355,
      "learning_rate": 3.875584850336627e-06,
      "loss": 0.0619,
      "step": 59360
    },
    {
      "epoch": 5.6610107421875e-05,
      "model_forward_time": 1.7207300662994385,
      "step": 59360
    },
    {
      "dynamics_loss_avg": 0.06389875765889882,
      "epoch": 5.6610107421875e-05,
      "step": 59360
    },
    {
      "action_loss_avg": 0.0013074790069367737,
      "epoch": 5.6610107421875e-05,
      "step": 59360
    },
    {
      "epoch": 5.6610107421875e-05,
      "step": 59360,
      "training_step_time": 6.844882488250732
    },
    {
      "epoch": 5.6611061096191403e-05,
      "model_forward_time": 1.7234694957733154,
      "step": 59361
    },
    {
      "epoch": 5.6611061096191403e-05,
      "step": 59361,
      "training_step_time": 7.174062252044678
    },
    {
      "epoch": 5.6612014770507814e-05,
      "model_forward_time": 1.326772928237915,
      "step": 59362
    },
    {
      "epoch": 5.6612014770507814e-05,
      "step": 59362,
      "training_step_time": 7.617873191833496
    },
    {
      "epoch": 5.661296844482422e-05,
      "model_forward_time": 1.710364818572998,
      "step": 59363
    },
    {
      "epoch": 5.661296844482422e-05,
      "step": 59363,
      "training_step_time": 7.411196708679199
    },
    {
      "epoch": 5.661392211914062e-05,
      "model_forward_time": 0.992032527923584,
      "step": 59364
    },
    {
      "epoch": 5.661392211914062e-05,
      "step": 59364,
      "training_step_time": 8.10120677947998
    },
    {
      "epoch": 5.661487579345703e-05,
      "model_forward_time": 1.7027828693389893,
      "step": 59365
    },
    {
      "epoch": 5.661487579345703e-05,
      "step": 59365,
      "training_step_time": 7.972098112106323
    },
    {
      "epoch": 5.661582946777344e-05,
      "model_forward_time": 1.7141199111938477,
      "step": 59366
    },
    {
      "epoch": 5.661582946777344e-05,
      "step": 59366,
      "training_step_time": 9.157793760299683
    },
    {
      "epoch": 5.661678314208984e-05,
      "model_forward_time": 2.270522356033325,
      "step": 59367
    },
    {
      "epoch": 5.661678314208984e-05,
      "step": 59367,
      "training_step_time": 7.864051342010498
    },
    {
      "epoch": 5.661773681640625e-05,
      "model_forward_time": 1.7052624225616455,
      "step": 59368
    },
    {
      "epoch": 5.661773681640625e-05,
      "step": 59368,
      "training_step_time": 9.022263526916504
    },
    {
      "epoch": 5.6618690490722656e-05,
      "model_forward_time": 1.7074859142303467,
      "step": 59369
    },
    {
      "epoch": 5.6618690490722656e-05,
      "step": 59369,
      "training_step_time": 6.822051048278809
    },
    {
      "epoch": 5.661964416503906e-05,
      "grad_norm": 0.0947551354765892,
      "learning_rate": 3.873973794391728e-06,
      "loss": 0.0628,
      "step": 59370
    },
    {
      "epoch": 5.661964416503906e-05,
      "model_forward_time": 1.3292934894561768,
      "step": 59370
    },
    {
      "dynamics_loss_avg": 0.07748649772256613,
      "epoch": 5.661964416503906e-05,
      "step": 59370
    },
    {
      "action_loss_avg": 0.0015820192958926782,
      "epoch": 5.661964416503906e-05,
      "step": 59370
    },
    {
      "epoch": 5.661964416503906e-05,
      "step": 59370,
      "training_step_time": 7.127088785171509
    },
    {
      "epoch": 5.662059783935547e-05,
      "model_forward_time": 1.7098891735076904,
      "step": 59371
    },
    {
      "epoch": 5.662059783935547e-05,
      "step": 59371,
      "training_step_time": 7.480835676193237
    },
    {
      "epoch": 5.6621551513671875e-05,
      "model_forward_time": 1.7217285633087158,
      "step": 59372
    },
    {
      "epoch": 5.6621551513671875e-05,
      "step": 59372,
      "training_step_time": 8.089447736740112
    },
    {
      "epoch": 5.662250518798828e-05,
      "model_forward_time": 1.7200825214385986,
      "step": 59373
    },
    {
      "epoch": 5.662250518798828e-05,
      "step": 59373,
      "training_step_time": 7.507245779037476
    },
    {
      "epoch": 5.662345886230469e-05,
      "model_forward_time": 1.7210354804992676,
      "step": 59374
    },
    {
      "epoch": 5.662345886230469e-05,
      "step": 59374,
      "training_step_time": 8.61053466796875
    },
    {
      "epoch": 5.6624412536621094e-05,
      "model_forward_time": 1.7940444946289062,
      "step": 59375
    },
    {
      "epoch": 5.6624412536621094e-05,
      "step": 59375,
      "training_step_time": 10.701977729797363
    },
    {
      "epoch": 5.66253662109375e-05,
      "model_forward_time": 1.7463469505310059,
      "step": 59376
    },
    {
      "epoch": 5.66253662109375e-05,
      "step": 59376,
      "training_step_time": 8.575073003768921
    },
    {
      "epoch": 5.662631988525391e-05,
      "model_forward_time": 2.929847240447998,
      "step": 59377
    },
    {
      "epoch": 5.662631988525391e-05,
      "step": 59377,
      "training_step_time": 7.8488688468933105
    },
    {
      "epoch": 5.662727355957031e-05,
      "model_forward_time": 1.7458360195159912,
      "step": 59378
    },
    {
      "epoch": 5.662727355957031e-05,
      "step": 59378,
      "training_step_time": 7.366234540939331
    },
    {
      "epoch": 5.662822723388672e-05,
      "model_forward_time": 1.7522950172424316,
      "step": 59379
    },
    {
      "epoch": 5.662822723388672e-05,
      "step": 59379,
      "training_step_time": 7.162542343139648
    },
    {
      "epoch": 5.662918090820313e-05,
      "grad_norm": 0.1231582760810852,
      "learning_rate": 3.872362861587361e-06,
      "loss": 0.0575,
      "step": 59380
    },
    {
      "epoch": 5.662918090820313e-05,
      "model_forward_time": 1.7482919692993164,
      "step": 59380
    },
    {
      "dynamics_loss_avg": 0.05678136395290494,
      "epoch": 5.662918090820313e-05,
      "step": 59380
    },
    {
      "action_loss_avg": 0.0016187231987714768,
      "epoch": 5.662918090820313e-05,
      "step": 59380
    },
    {
      "epoch": 5.662918090820313e-05,
      "step": 59380,
      "training_step_time": 7.420464754104614
    },
    {
      "epoch": 5.663013458251953e-05,
      "model_forward_time": 1.7539122104644775,
      "step": 59381
    },
    {
      "epoch": 5.663013458251953e-05,
      "step": 59381,
      "training_step_time": 7.550438642501831
    },
    {
      "epoch": 5.6631088256835935e-05,
      "model_forward_time": 1.7153499126434326,
      "step": 59382
    },
    {
      "epoch": 5.6631088256835935e-05,
      "step": 59382,
      "training_step_time": 7.399245500564575
    },
    {
      "epoch": 5.6632041931152346e-05,
      "model_forward_time": 1.7076971530914307,
      "step": 59383
    },
    {
      "epoch": 5.6632041931152346e-05,
      "step": 59383,
      "training_step_time": 8.867208480834961
    },
    {
      "epoch": 5.663299560546875e-05,
      "model_forward_time": 1.7119638919830322,
      "step": 59384
    },
    {
      "epoch": 5.663299560546875e-05,
      "step": 59384,
      "training_step_time": 7.821554899215698
    },
    {
      "epoch": 5.6633949279785154e-05,
      "model_forward_time": 1.7026526927947998,
      "step": 59385
    },
    {
      "epoch": 5.6633949279785154e-05,
      "step": 59385,
      "training_step_time": 7.73571252822876
    },
    {
      "epoch": 5.6634902954101565e-05,
      "model_forward_time": 2.241649866104126,
      "step": 59386
    },
    {
      "epoch": 5.6634902954101565e-05,
      "step": 59386,
      "training_step_time": 8.811778545379639
    },
    {
      "epoch": 5.663585662841797e-05,
      "model_forward_time": 1.7037827968597412,
      "step": 59387
    },
    {
      "epoch": 5.663585662841797e-05,
      "step": 59387,
      "training_step_time": 7.449496269226074
    },
    {
      "epoch": 5.663681030273437e-05,
      "model_forward_time": 1.7002143859863281,
      "step": 59388
    },
    {
      "epoch": 5.663681030273437e-05,
      "step": 59388,
      "training_step_time": 7.156793117523193
    },
    {
      "epoch": 5.6637763977050784e-05,
      "model_forward_time": 1.6944239139556885,
      "step": 59389
    },
    {
      "epoch": 5.6637763977050784e-05,
      "step": 59389,
      "training_step_time": 7.131577491760254
    },
    {
      "epoch": 5.663871765136719e-05,
      "grad_norm": 0.10033264756202698,
      "learning_rate": 3.870752052099698e-06,
      "loss": 0.0628,
      "step": 59390
    },
    {
      "epoch": 5.663871765136719e-05,
      "model_forward_time": 1.6941020488739014,
      "step": 59390
    },
    {
      "dynamics_loss_avg": 0.05502032246440649,
      "epoch": 5.663871765136719e-05,
      "step": 59390
    },
    {
      "action_loss_avg": 0.0023076861980371177,
      "epoch": 5.663871765136719e-05,
      "step": 59390
    },
    {
      "epoch": 5.663871765136719e-05,
      "step": 59390,
      "training_step_time": 7.4284772872924805
    },
    {
      "epoch": 5.663967132568359e-05,
      "model_forward_time": 1.698615550994873,
      "step": 59391
    },
    {
      "epoch": 5.663967132568359e-05,
      "step": 59391,
      "training_step_time": 7.347077131271362
    },
    {
      "epoch": 5.6640625e-05,
      "model_forward_time": 1.6932456493377686,
      "step": 59392
    },
    {
      "epoch": 5.6640625e-05,
      "step": 59392,
      "training_step_time": 9.383328914642334
    },
    {
      "epoch": 5.664157867431641e-05,
      "model_forward_time": 1.703211784362793,
      "step": 59393
    },
    {
      "epoch": 5.664157867431641e-05,
      "step": 59393,
      "training_step_time": 8.849854469299316
    },
    {
      "epoch": 5.664253234863281e-05,
      "model_forward_time": 1.6939351558685303,
      "step": 59394
    },
    {
      "epoch": 5.664253234863281e-05,
      "step": 59394,
      "training_step_time": 8.693976163864136
    },
    {
      "epoch": 5.664348602294922e-05,
      "model_forward_time": 2.244293451309204,
      "step": 59395
    },
    {
      "epoch": 5.664348602294922e-05,
      "step": 59395,
      "training_step_time": 7.857407808303833
    },
    {
      "epoch": 5.6644439697265626e-05,
      "model_forward_time": 1.700157642364502,
      "step": 59396
    },
    {
      "epoch": 5.6644439697265626e-05,
      "step": 59396,
      "training_step_time": 7.725865125656128
    },
    {
      "epoch": 5.664539337158203e-05,
      "model_forward_time": 1.3251957893371582,
      "step": 59397
    },
    {
      "epoch": 5.664539337158203e-05,
      "step": 59397,
      "training_step_time": 7.528798341751099
    },
    {
      "epoch": 5.664634704589844e-05,
      "model_forward_time": 1.329496145248413,
      "step": 59398
    },
    {
      "epoch": 5.664634704589844e-05,
      "step": 59398,
      "training_step_time": 7.136456251144409
    },
    {
      "epoch": 5.6647300720214845e-05,
      "model_forward_time": 1.3268237113952637,
      "step": 59399
    },
    {
      "epoch": 5.6647300720214845e-05,
      "step": 59399,
      "training_step_time": 7.566575527191162
    },
    {
      "epoch": 5.664825439453125e-05,
      "grad_norm": 0.08934067189693451,
      "learning_rate": 3.869141366104892e-06,
      "loss": 0.0607,
      "step": 59400
    },
    {
      "epoch": 5.664825439453125e-05,
      "model_forward_time": 1.692185640335083,
      "step": 59400
    },
    {
      "dynamics_loss_avg": 0.05825235024094581,
      "epoch": 5.664825439453125e-05,
      "step": 59400
    },
    {
      "action_loss_avg": 0.0013130076782545074,
      "epoch": 5.664825439453125e-05,
      "step": 59400
    },
    {
      "epoch": 5.664825439453125e-05,
      "step": 59400,
      "training_step_time": 7.391779661178589
    },
    {
      "epoch": 5.664920806884766e-05,
      "model_forward_time": 1.704967975616455,
      "step": 59401
    },
    {
      "epoch": 5.664920806884766e-05,
      "step": 59401,
      "training_step_time": 8.868969440460205
    },
    {
      "epoch": 5.6650161743164063e-05,
      "model_forward_time": 1.6994457244873047,
      "step": 59402
    },
    {
      "epoch": 5.6650161743164063e-05,
      "step": 59402,
      "training_step_time": 9.50944185256958
    },
    {
      "epoch": 5.665111541748047e-05,
      "model_forward_time": 1.699648380279541,
      "step": 59403
    },
    {
      "epoch": 5.665111541748047e-05,
      "step": 59403,
      "training_step_time": 10.191633224487305
    },
    {
      "epoch": 5.665206909179688e-05,
      "model_forward_time": 1.7011351585388184,
      "step": 59404
    },
    {
      "epoch": 5.665206909179688e-05,
      "step": 59404,
      "training_step_time": 7.840731620788574
    },
    {
      "epoch": 5.665302276611328e-05,
      "model_forward_time": 2.2270286083221436,
      "step": 59405
    },
    {
      "epoch": 5.665302276611328e-05,
      "step": 59405,
      "training_step_time": 7.783260822296143
    },
    {
      "epoch": 5.6653976440429686e-05,
      "model_forward_time": 1.716228723526001,
      "step": 59406
    },
    {
      "epoch": 5.6653976440429686e-05,
      "step": 59406,
      "training_step_time": 7.501581430435181
    },
    {
      "epoch": 5.665493011474609e-05,
      "model_forward_time": 1.7070577144622803,
      "step": 59407
    },
    {
      "epoch": 5.665493011474609e-05,
      "step": 59407,
      "training_step_time": 7.1369476318359375
    },
    {
      "epoch": 5.66558837890625e-05,
      "model_forward_time": 1.703829050064087,
      "step": 59408
    },
    {
      "epoch": 5.66558837890625e-05,
      "step": 59408,
      "training_step_time": 7.501525640487671
    },
    {
      "epoch": 5.6656837463378905e-05,
      "model_forward_time": 1.7054598331451416,
      "step": 59409
    },
    {
      "epoch": 5.6656837463378905e-05,
      "step": 59409,
      "training_step_time": 7.321396112442017
    },
    {
      "epoch": 5.665779113769531e-05,
      "grad_norm": 0.15941113233566284,
      "learning_rate": 3.8675308037790855e-06,
      "loss": 0.0594,
      "step": 59410
    },
    {
      "epoch": 5.665779113769531e-05,
      "model_forward_time": 1.702993392944336,
      "step": 59410
    },
    {
      "dynamics_loss_avg": 0.05652820914983749,
      "epoch": 5.665779113769531e-05,
      "step": 59410
    },
    {
      "action_loss_avg": 0.0012356707185972482,
      "epoch": 5.665779113769531e-05,
      "step": 59410
    },
    {
      "epoch": 5.665779113769531e-05,
      "step": 59410,
      "training_step_time": 8.730520963668823
    },
    {
      "epoch": 5.665874481201172e-05,
      "model_forward_time": 1.6988754272460938,
      "step": 59411
    },
    {
      "epoch": 5.665874481201172e-05,
      "step": 59411,
      "training_step_time": 9.412547588348389
    },
    {
      "epoch": 5.6659698486328124e-05,
      "model_forward_time": 1.6968121528625488,
      "step": 59412
    },
    {
      "epoch": 5.6659698486328124e-05,
      "step": 59412,
      "training_step_time": 8.42016339302063
    },
    {
      "epoch": 5.666065216064453e-05,
      "model_forward_time": 1.7002527713775635,
      "step": 59413
    },
    {
      "epoch": 5.666065216064453e-05,
      "step": 59413,
      "training_step_time": 8.35793137550354
    },
    {
      "epoch": 5.666160583496094e-05,
      "model_forward_time": 2.240915060043335,
      "step": 59414
    },
    {
      "epoch": 5.666160583496094e-05,
      "step": 59414,
      "training_step_time": 7.8229265213012695
    },
    {
      "epoch": 5.666255950927734e-05,
      "model_forward_time": 1.6931757926940918,
      "step": 59415
    },
    {
      "epoch": 5.666255950927734e-05,
      "step": 59415,
      "training_step_time": 7.730078220367432
    },
    {
      "epoch": 5.666351318359375e-05,
      "model_forward_time": 1.6980445384979248,
      "step": 59416
    },
    {
      "epoch": 5.666351318359375e-05,
      "step": 59416,
      "training_step_time": 7.16205358505249
    },
    {
      "epoch": 5.666446685791016e-05,
      "model_forward_time": 1.6932141780853271,
      "step": 59417
    },
    {
      "epoch": 5.666446685791016e-05,
      "step": 59417,
      "training_step_time": 7.343491315841675
    },
    {
      "epoch": 5.666542053222656e-05,
      "model_forward_time": 1.7013962268829346,
      "step": 59418
    },
    {
      "epoch": 5.666542053222656e-05,
      "step": 59418,
      "training_step_time": 8.412870168685913
    },
    {
      "epoch": 5.6666374206542966e-05,
      "model_forward_time": 1.7020764350891113,
      "step": 59419
    },
    {
      "epoch": 5.6666374206542966e-05,
      "step": 59419,
      "training_step_time": 8.272480726242065
    },
    {
      "epoch": 5.666732788085938e-05,
      "grad_norm": 0.10741636157035828,
      "learning_rate": 3.865920365298405e-06,
      "loss": 0.0579,
      "step": 59420
    },
    {
      "epoch": 5.666732788085938e-05,
      "model_forward_time": 1.6940317153930664,
      "step": 59420
    },
    {
      "dynamics_loss_avg": 0.060176140069961546,
      "epoch": 5.666732788085938e-05,
      "step": 59420
    },
    {
      "action_loss_avg": 0.001863386033801362,
      "epoch": 5.666732788085938e-05,
      "step": 59420
    },
    {
      "epoch": 5.666732788085938e-05,
      "step": 59420,
      "training_step_time": 7.89591383934021
    },
    {
      "epoch": 5.666828155517578e-05,
      "model_forward_time": 1.7063000202178955,
      "step": 59421
    },
    {
      "epoch": 5.666828155517578e-05,
      "step": 59421,
      "training_step_time": 7.825988292694092
    },
    {
      "epoch": 5.6669235229492185e-05,
      "model_forward_time": 1.7005157470703125,
      "step": 59422
    },
    {
      "epoch": 5.6669235229492185e-05,
      "step": 59422,
      "training_step_time": 8.815848112106323
    },
    {
      "epoch": 5.6670188903808595e-05,
      "model_forward_time": 2.2413976192474365,
      "step": 59423
    },
    {
      "epoch": 5.6670188903808595e-05,
      "step": 59423,
      "training_step_time": 7.840233087539673
    },
    {
      "epoch": 5.6671142578125e-05,
      "model_forward_time": 1.69636869430542,
      "step": 59424
    },
    {
      "epoch": 5.6671142578125e-05,
      "step": 59424,
      "training_step_time": 7.655639410018921
    },
    {
      "epoch": 5.6672096252441404e-05,
      "model_forward_time": 1.7144858837127686,
      "step": 59425
    },
    {
      "epoch": 5.6672096252441404e-05,
      "step": 59425,
      "training_step_time": 7.510502576828003
    },
    {
      "epoch": 5.6673049926757814e-05,
      "model_forward_time": 1.69944429397583,
      "step": 59426
    },
    {
      "epoch": 5.6673049926757814e-05,
      "step": 59426,
      "training_step_time": 7.337647199630737
    },
    {
      "epoch": 5.667400360107422e-05,
      "model_forward_time": 1.3299875259399414,
      "step": 59427
    },
    {
      "epoch": 5.667400360107422e-05,
      "step": 59427,
      "training_step_time": 8.0550217628479
    },
    {
      "epoch": 5.667495727539062e-05,
      "model_forward_time": 1.6983883380889893,
      "step": 59428
    },
    {
      "epoch": 5.667495727539062e-05,
      "step": 59428,
      "training_step_time": 7.640877962112427
    },
    {
      "epoch": 5.667591094970703e-05,
      "model_forward_time": 1.7032902240753174,
      "step": 59429
    },
    {
      "epoch": 5.667591094970703e-05,
      "step": 59429,
      "training_step_time": 7.951699495315552
    },
    {
      "epoch": 5.667686462402344e-05,
      "grad_norm": 0.09547349810600281,
      "learning_rate": 3.864310050838972e-06,
      "loss": 0.0584,
      "step": 59430
    },
    {
      "epoch": 5.667686462402344e-05,
      "model_forward_time": 1.706939458847046,
      "step": 59430
    },
    {
      "dynamics_loss_avg": 0.04723964259028435,
      "epoch": 5.667686462402344e-05,
      "step": 59430
    },
    {
      "action_loss_avg": 0.001230902819952462,
      "epoch": 5.667686462402344e-05,
      "step": 59430
    },
    {
      "epoch": 5.667686462402344e-05,
      "step": 59430,
      "training_step_time": 8.625613927841187
    },
    {
      "epoch": 5.667781829833984e-05,
      "model_forward_time": 1.701972246170044,
      "step": 59431
    },
    {
      "epoch": 5.667781829833984e-05,
      "step": 59431,
      "training_step_time": 10.265326023101807
    },
    {
      "epoch": 5.667877197265625e-05,
      "model_forward_time": 2.2776684761047363,
      "step": 59432
    },
    {
      "epoch": 5.667877197265625e-05,
      "step": 59432,
      "training_step_time": 8.77591061592102
    },
    {
      "epoch": 5.6679725646972656e-05,
      "model_forward_time": 1.7006804943084717,
      "step": 59433
    },
    {
      "epoch": 5.6679725646972656e-05,
      "step": 59433,
      "training_step_time": 7.866421222686768
    },
    {
      "epoch": 5.668067932128906e-05,
      "model_forward_time": 1.7606596946716309,
      "step": 59434
    },
    {
      "epoch": 5.668067932128906e-05,
      "step": 59434,
      "training_step_time": 7.926781415939331
    },
    {
      "epoch": 5.668163299560547e-05,
      "model_forward_time": 1.7054274082183838,
      "step": 59435
    },
    {
      "epoch": 5.668163299560547e-05,
      "step": 59435,
      "training_step_time": 7.540236234664917
    },
    {
      "epoch": 5.6682586669921875e-05,
      "model_forward_time": 1.7148196697235107,
      "step": 59436
    },
    {
      "epoch": 5.6682586669921875e-05,
      "step": 59436,
      "training_step_time": 7.763443231582642
    },
    {
      "epoch": 5.668354034423828e-05,
      "model_forward_time": 1.7145090103149414,
      "step": 59437
    },
    {
      "epoch": 5.668354034423828e-05,
      "step": 59437,
      "training_step_time": 8.540705442428589
    },
    {
      "epoch": 5.668449401855469e-05,
      "model_forward_time": 1.7257204055786133,
      "step": 59438
    },
    {
      "epoch": 5.668449401855469e-05,
      "step": 59438,
      "training_step_time": 8.529786825180054
    },
    {
      "epoch": 5.6685447692871094e-05,
      "model_forward_time": 1.6966497898101807,
      "step": 59439
    },
    {
      "epoch": 5.6685447692871094e-05,
      "step": 59439,
      "training_step_time": 7.865692853927612
    },
    {
      "epoch": 5.66864013671875e-05,
      "grad_norm": 0.110973060131073,
      "learning_rate": 3.862699860576884e-06,
      "loss": 0.0567,
      "step": 59440
    },
    {
      "epoch": 5.66864013671875e-05,
      "model_forward_time": 1.7090375423431396,
      "step": 59440
    },
    {
      "dynamics_loss_avg": 0.05740330684930086,
      "epoch": 5.66864013671875e-05,
      "step": 59440
    },
    {
      "action_loss_avg": 0.0013304119580425321,
      "epoch": 5.66864013671875e-05,
      "step": 59440
    },
    {
      "epoch": 5.66864013671875e-05,
      "step": 59440,
      "training_step_time": 7.974934816360474
    },
    {
      "epoch": 5.668735504150391e-05,
      "model_forward_time": 2.2677340507507324,
      "step": 59441
    },
    {
      "epoch": 5.668735504150391e-05,
      "step": 59441,
      "training_step_time": 8.917803287506104
    },
    {
      "epoch": 5.668830871582031e-05,
      "model_forward_time": 1.3519115447998047,
      "step": 59442
    },
    {
      "epoch": 5.668830871582031e-05,
      "step": 59442,
      "training_step_time": 7.845770835876465
    },
    {
      "epoch": 5.668926239013672e-05,
      "model_forward_time": 1.724048376083374,
      "step": 59443
    },
    {
      "epoch": 5.668926239013672e-05,
      "step": 59443,
      "training_step_time": 7.8234076499938965
    },
    {
      "epoch": 5.669021606445313e-05,
      "model_forward_time": 1.711963415145874,
      "step": 59444
    },
    {
      "epoch": 5.669021606445313e-05,
      "step": 59444,
      "training_step_time": 7.158679485321045
    },
    {
      "epoch": 5.669116973876953e-05,
      "model_forward_time": 1.7160894870758057,
      "step": 59445
    },
    {
      "epoch": 5.669116973876953e-05,
      "step": 59445,
      "training_step_time": 7.620142936706543
    },
    {
      "epoch": 5.6692123413085936e-05,
      "model_forward_time": 1.715918779373169,
      "step": 59446
    },
    {
      "epoch": 5.6692123413085936e-05,
      "step": 59446,
      "training_step_time": 7.593997955322266
    },
    {
      "epoch": 5.6693077087402346e-05,
      "model_forward_time": 1.7167041301727295,
      "step": 59447
    },
    {
      "epoch": 5.6693077087402346e-05,
      "step": 59447,
      "training_step_time": 9.551103353500366
    },
    {
      "epoch": 5.669403076171875e-05,
      "model_forward_time": 1.702646255493164,
      "step": 59448
    },
    {
      "epoch": 5.669403076171875e-05,
      "step": 59448,
      "training_step_time": 9.202017307281494
    },
    {
      "epoch": 5.6694984436035154e-05,
      "model_forward_time": 1.7039368152618408,
      "step": 59449
    },
    {
      "epoch": 5.6694984436035154e-05,
      "step": 59449,
      "training_step_time": 7.7097485065460205
    },
    {
      "epoch": 5.6695938110351565e-05,
      "grad_norm": 0.12288235872983932,
      "learning_rate": 3.861089794688229e-06,
      "loss": 0.0571,
      "step": 59450
    },
    {
      "epoch": 5.6695938110351565e-05,
      "model_forward_time": 1.7157645225524902,
      "step": 59450
    },
    {
      "dynamics_loss_avg": 0.0575835706666112,
      "epoch": 5.6695938110351565e-05,
      "step": 59450
    },
    {
      "action_loss_avg": 0.0017736603273078798,
      "epoch": 5.6695938110351565e-05,
      "step": 59450
    },
    {
      "epoch": 5.6695938110351565e-05,
      "step": 59450,
      "training_step_time": 7.92451548576355
    },
    {
      "epoch": 5.669689178466797e-05,
      "model_forward_time": 2.261941432952881,
      "step": 59451
    },
    {
      "epoch": 5.669689178466797e-05,
      "step": 59451,
      "training_step_time": 7.833110332489014
    },
    {
      "epoch": 5.669784545898437e-05,
      "model_forward_time": 1.343902826309204,
      "step": 59452
    },
    {
      "epoch": 5.669784545898437e-05,
      "step": 59452,
      "training_step_time": 8.682519912719727
    },
    {
      "epoch": 5.6698799133300784e-05,
      "model_forward_time": 1.7712430953979492,
      "step": 59453
    },
    {
      "epoch": 5.6698799133300784e-05,
      "step": 59453,
      "training_step_time": 7.360101222991943
    },
    {
      "epoch": 5.669975280761719e-05,
      "model_forward_time": 1.7085564136505127,
      "step": 59454
    },
    {
      "epoch": 5.669975280761719e-05,
      "step": 59454,
      "training_step_time": 7.616358757019043
    },
    {
      "epoch": 5.670070648193359e-05,
      "model_forward_time": 1.7098994255065918,
      "step": 59455
    },
    {
      "epoch": 5.670070648193359e-05,
      "step": 59455,
      "training_step_time": 7.602537155151367
    },
    {
      "epoch": 5.670166015625e-05,
      "model_forward_time": 1.7125074863433838,
      "step": 59456
    },
    {
      "epoch": 5.670166015625e-05,
      "step": 59456,
      "training_step_time": 7.831045150756836
    },
    {
      "epoch": 5.670261383056641e-05,
      "model_forward_time": 1.7080540657043457,
      "step": 59457
    },
    {
      "epoch": 5.670261383056641e-05,
      "step": 59457,
      "training_step_time": 7.9109272956848145
    },
    {
      "epoch": 5.670356750488281e-05,
      "model_forward_time": 1.7147181034088135,
      "step": 59458
    },
    {
      "epoch": 5.670356750488281e-05,
      "step": 59458,
      "training_step_time": 9.19683027267456
    },
    {
      "epoch": 5.670452117919922e-05,
      "model_forward_time": 1.729989767074585,
      "step": 59459
    },
    {
      "epoch": 5.670452117919922e-05,
      "step": 59459,
      "training_step_time": 8.851228475570679
    },
    {
      "epoch": 5.6705474853515626e-05,
      "grad_norm": 0.11842511594295502,
      "learning_rate": 3.8594798533490815e-06,
      "loss": 0.0587,
      "step": 59460
    },
    {
      "epoch": 5.6705474853515626e-05,
      "model_forward_time": 2.2509443759918213,
      "step": 59460
    },
    {
      "dynamics_loss_avg": 0.07384534627199173,
      "epoch": 5.6705474853515626e-05,
      "step": 59460
    },
    {
      "action_loss_avg": 0.001605321787064895,
      "epoch": 5.6705474853515626e-05,
      "step": 59460
    },
    {
      "epoch": 5.6705474853515626e-05,
      "step": 59460,
      "training_step_time": 8.776963949203491
    },
    {
      "epoch": 5.670642852783203e-05,
      "model_forward_time": 1.711731195449829,
      "step": 59461
    },
    {
      "epoch": 5.670642852783203e-05,
      "step": 59461,
      "training_step_time": 7.696274518966675
    },
    {
      "epoch": 5.670738220214844e-05,
      "model_forward_time": 1.774080753326416,
      "step": 59462
    },
    {
      "epoch": 5.670738220214844e-05,
      "step": 59462,
      "training_step_time": 7.506520748138428
    },
    {
      "epoch": 5.6708335876464845e-05,
      "model_forward_time": 1.7176027297973633,
      "step": 59463
    },
    {
      "epoch": 5.6708335876464845e-05,
      "step": 59463,
      "training_step_time": 8.208786964416504
    },
    {
      "epoch": 5.670928955078125e-05,
      "model_forward_time": 1.7189335823059082,
      "step": 59464
    },
    {
      "epoch": 5.670928955078125e-05,
      "step": 59464,
      "training_step_time": 7.513213396072388
    },
    {
      "epoch": 5.671024322509766e-05,
      "model_forward_time": 1.7215192317962646,
      "step": 59465
    },
    {
      "epoch": 5.671024322509766e-05,
      "step": 59465,
      "training_step_time": 7.753470420837402
    },
    {
      "epoch": 5.6711196899414064e-05,
      "model_forward_time": 1.7145800590515137,
      "step": 59466
    },
    {
      "epoch": 5.6711196899414064e-05,
      "step": 59466,
      "training_step_time": 8.716524839401245
    },
    {
      "epoch": 5.671215057373047e-05,
      "model_forward_time": 1.7110953330993652,
      "step": 59467
    },
    {
      "epoch": 5.671215057373047e-05,
      "step": 59467,
      "training_step_time": 8.70672607421875
    },
    {
      "epoch": 5.671310424804688e-05,
      "model_forward_time": 1.722719669342041,
      "step": 59468
    },
    {
      "epoch": 5.671310424804688e-05,
      "step": 59468,
      "training_step_time": 7.84172797203064
    },
    {
      "epoch": 5.671405792236328e-05,
      "model_forward_time": 2.2571401596069336,
      "step": 59469
    },
    {
      "epoch": 5.671405792236328e-05,
      "step": 59469,
      "training_step_time": 7.754981994628906
    },
    {
      "epoch": 5.6715011596679686e-05,
      "grad_norm": 0.09264323115348816,
      "learning_rate": 3.857870036735503e-06,
      "loss": 0.0538,
      "step": 59470
    },
    {
      "epoch": 5.6715011596679686e-05,
      "model_forward_time": 1.7204148769378662,
      "step": 59470
    },
    {
      "dynamics_loss_avg": 0.05950767230242491,
      "epoch": 5.6715011596679686e-05,
      "step": 59470
    },
    {
      "action_loss_avg": 0.0014063773909583689,
      "epoch": 5.6715011596679686e-05,
      "step": 59470
    },
    {
      "epoch": 5.6715011596679686e-05,
      "step": 59470,
      "training_step_time": 7.738508939743042
    },
    {
      "epoch": 5.671596527099609e-05,
      "model_forward_time": 1.770862340927124,
      "step": 59471
    },
    {
      "epoch": 5.671596527099609e-05,
      "step": 59471,
      "training_step_time": 7.791517496109009
    },
    {
      "epoch": 5.67169189453125e-05,
      "model_forward_time": 1.7024362087249756,
      "step": 59472
    },
    {
      "epoch": 5.67169189453125e-05,
      "step": 59472,
      "training_step_time": 7.3951966762542725
    },
    {
      "epoch": 5.6717872619628905e-05,
      "model_forward_time": 1.730475902557373,
      "step": 59473
    },
    {
      "epoch": 5.6717872619628905e-05,
      "step": 59473,
      "training_step_time": 7.62187647819519
    },
    {
      "epoch": 5.671882629394531e-05,
      "model_forward_time": 1.70833420753479,
      "step": 59474
    },
    {
      "epoch": 5.671882629394531e-05,
      "step": 59474,
      "training_step_time": 7.7685227394104
    },
    {
      "epoch": 5.671977996826172e-05,
      "model_forward_time": 1.724005937576294,
      "step": 59475
    },
    {
      "epoch": 5.671977996826172e-05,
      "step": 59475,
      "training_step_time": 8.779170751571655
    },
    {
      "epoch": 5.6720733642578124e-05,
      "model_forward_time": 1.7002983093261719,
      "step": 59476
    },
    {
      "epoch": 5.6720733642578124e-05,
      "step": 59476,
      "training_step_time": 7.75088357925415
    },
    {
      "epoch": 5.672168731689453e-05,
      "model_forward_time": 1.706176519393921,
      "step": 59477
    },
    {
      "epoch": 5.672168731689453e-05,
      "step": 59477,
      "training_step_time": 8.418760538101196
    },
    {
      "epoch": 5.672264099121094e-05,
      "model_forward_time": 2.291417360305786,
      "step": 59478
    },
    {
      "epoch": 5.672264099121094e-05,
      "step": 59478,
      "training_step_time": 8.695656061172485
    },
    {
      "epoch": 5.672359466552734e-05,
      "model_forward_time": 1.7059476375579834,
      "step": 59479
    },
    {
      "epoch": 5.672359466552734e-05,
      "step": 59479,
      "training_step_time": 7.752302646636963
    },
    {
      "epoch": 5.672454833984375e-05,
      "grad_norm": 0.08538349717855453,
      "learning_rate": 3.8562603450235385e-06,
      "loss": 0.057,
      "step": 59480
    },
    {
      "epoch": 5.672454833984375e-05,
      "model_forward_time": 1.7537751197814941,
      "step": 59480
    },
    {
      "dynamics_loss_avg": 0.06999931372702121,
      "epoch": 5.672454833984375e-05,
      "step": 59480
    },
    {
      "action_loss_avg": 0.0017454975692089646,
      "epoch": 5.672454833984375e-05,
      "step": 59480
    },
    {
      "epoch": 5.672454833984375e-05,
      "step": 59480,
      "training_step_time": 7.7179834842681885
    },
    {
      "epoch": 5.672550201416016e-05,
      "model_forward_time": 1.708099603652954,
      "step": 59481
    },
    {
      "epoch": 5.672550201416016e-05,
      "step": 59481,
      "training_step_time": 7.186843395233154
    },
    {
      "epoch": 5.672645568847656e-05,
      "model_forward_time": 1.7064673900604248,
      "step": 59482
    },
    {
      "epoch": 5.672645568847656e-05,
      "step": 59482,
      "training_step_time": 8.131317138671875
    },
    {
      "epoch": 5.6727409362792966e-05,
      "model_forward_time": 1.7088067531585693,
      "step": 59483
    },
    {
      "epoch": 5.6727409362792966e-05,
      "step": 59483,
      "training_step_time": 8.584534645080566
    },
    {
      "epoch": 5.672836303710938e-05,
      "model_forward_time": 1.733649730682373,
      "step": 59484
    },
    {
      "epoch": 5.672836303710938e-05,
      "step": 59484,
      "training_step_time": 7.698464870452881
    },
    {
      "epoch": 5.672931671142578e-05,
      "model_forward_time": 1.7004756927490234,
      "step": 59485
    },
    {
      "epoch": 5.672931671142578e-05,
      "step": 59485,
      "training_step_time": 8.849663019180298
    },
    {
      "epoch": 5.6730270385742185e-05,
      "model_forward_time": 1.7049014568328857,
      "step": 59486
    },
    {
      "epoch": 5.6730270385742185e-05,
      "step": 59486,
      "training_step_time": 8.54723596572876
    },
    {
      "epoch": 5.6731224060058596e-05,
      "model_forward_time": 2.2383687496185303,
      "step": 59487
    },
    {
      "epoch": 5.6731224060058596e-05,
      "step": 59487,
      "training_step_time": 9.662403106689453
    },
    {
      "epoch": 5.6732177734375e-05,
      "model_forward_time": 1.732095718383789,
      "step": 59488
    },
    {
      "epoch": 5.6732177734375e-05,
      "step": 59488,
      "training_step_time": 8.510721206665039
    },
    {
      "epoch": 5.6733131408691404e-05,
      "model_forward_time": 1.7337486743927002,
      "step": 59489
    },
    {
      "epoch": 5.6733131408691404e-05,
      "step": 59489,
      "training_step_time": 7.615863800048828
    },
    {
      "epoch": 5.6734085083007814e-05,
      "grad_norm": 0.12666741013526917,
      "learning_rate": 3.854650778389224e-06,
      "loss": 0.0579,
      "step": 59490
    },
    {
      "epoch": 5.6734085083007814e-05,
      "model_forward_time": 1.7085239887237549,
      "step": 59490
    },
    {
      "dynamics_loss_avg": 0.057567293383181097,
      "epoch": 5.6734085083007814e-05,
      "step": 59490
    },
    {
      "action_loss_avg": 0.0013996417168527842,
      "epoch": 5.6734085083007814e-05,
      "step": 59490
    },
    {
      "epoch": 5.6734085083007814e-05,
      "step": 59490,
      "training_step_time": 7.110476970672607
    },
    {
      "epoch": 5.673503875732422e-05,
      "model_forward_time": 1.7160923480987549,
      "step": 59491
    },
    {
      "epoch": 5.673503875732422e-05,
      "step": 59491,
      "training_step_time": 7.507155179977417
    },
    {
      "epoch": 5.673599243164062e-05,
      "model_forward_time": 1.7126593589782715,
      "step": 59492
    },
    {
      "epoch": 5.673599243164062e-05,
      "step": 59492,
      "training_step_time": 7.731605768203735
    },
    {
      "epoch": 5.673694610595703e-05,
      "model_forward_time": 1.707719087600708,
      "step": 59493
    },
    {
      "epoch": 5.673694610595703e-05,
      "step": 59493,
      "training_step_time": 7.641348838806152
    },
    {
      "epoch": 5.673789978027344e-05,
      "model_forward_time": 1.6974427700042725,
      "step": 59494
    },
    {
      "epoch": 5.673789978027344e-05,
      "step": 59494,
      "training_step_time": 8.394749879837036
    },
    {
      "epoch": 5.673885345458984e-05,
      "model_forward_time": 1.6934051513671875,
      "step": 59495
    },
    {
      "epoch": 5.673885345458984e-05,
      "step": 59495,
      "training_step_time": 9.244672060012817
    },
    {
      "epoch": 5.673980712890625e-05,
      "model_forward_time": 1.7318744659423828,
      "step": 59496
    },
    {
      "epoch": 5.673980712890625e-05,
      "step": 59496,
      "training_step_time": 7.836233854293823
    },
    {
      "epoch": 5.6740760803222656e-05,
      "model_forward_time": 2.2830984592437744,
      "step": 59497
    },
    {
      "epoch": 5.6740760803222656e-05,
      "step": 59497,
      "training_step_time": 8.578132629394531
    },
    {
      "epoch": 5.674171447753906e-05,
      "model_forward_time": 1.7615118026733398,
      "step": 59498
    },
    {
      "epoch": 5.674171447753906e-05,
      "step": 59498,
      "training_step_time": 8.350310325622559
    },
    {
      "epoch": 5.674266815185547e-05,
      "model_forward_time": 1.7448289394378662,
      "step": 59499
    },
    {
      "epoch": 5.674266815185547e-05,
      "step": 59499,
      "training_step_time": 7.1475770473480225
    },
    {
      "epoch": 5.6743621826171875e-05,
      "grad_norm": 0.10075437277555466,
      "learning_rate": 3.853041337008579e-06,
      "loss": 0.0573,
      "step": 59500
    },
    {
      "epoch": 5.6743621826171875e-05,
      "model_forward_time": 1.729323387145996,
      "step": 59500
    },
    {
      "dynamics_loss_avg": 0.06934887655079365,
      "epoch": 5.6743621826171875e-05,
      "step": 59500
    },
    {
      "action_loss_avg": 0.0017878736514830961,
      "epoch": 5.6743621826171875e-05,
      "step": 59500
    },
    {
      "epoch": 5.6743621826171875e-05,
      "step": 59500,
      "training_step_time": 7.631184101104736
    },
    {
      "epoch": 5.674457550048828e-05,
      "model_forward_time": 1.7461755275726318,
      "step": 59501
    },
    {
      "epoch": 5.674457550048828e-05,
      "step": 59501,
      "training_step_time": 7.781541347503662
    },
    {
      "epoch": 5.674552917480469e-05,
      "model_forward_time": 1.023918628692627,
      "step": 59502
    },
    {
      "epoch": 5.674552917480469e-05,
      "step": 59502,
      "training_step_time": 7.763073444366455
    },
    {
      "epoch": 5.6746482849121094e-05,
      "model_forward_time": 1.7316622734069824,
      "step": 59503
    },
    {
      "epoch": 5.6746482849121094e-05,
      "step": 59503,
      "training_step_time": 9.513452291488647
    },
    {
      "epoch": 5.67474365234375e-05,
      "model_forward_time": 1.7382221221923828,
      "step": 59504
    },
    {
      "epoch": 5.67474365234375e-05,
      "step": 59504,
      "training_step_time": 7.803972005844116
    },
    {
      "epoch": 5.674839019775391e-05,
      "model_forward_time": 1.7367935180664062,
      "step": 59505
    },
    {
      "epoch": 5.674839019775391e-05,
      "step": 59505,
      "training_step_time": 7.840349435806274
    },
    {
      "epoch": 5.674934387207031e-05,
      "model_forward_time": 2.304772138595581,
      "step": 59506
    },
    {
      "epoch": 5.674934387207031e-05,
      "step": 59506,
      "training_step_time": 7.7785844802856445
    },
    {
      "epoch": 5.675029754638672e-05,
      "model_forward_time": 1.736865520477295,
      "step": 59507
    },
    {
      "epoch": 5.675029754638672e-05,
      "step": 59507,
      "training_step_time": 7.55048942565918
    },
    {
      "epoch": 5.675125122070313e-05,
      "model_forward_time": 1.7959110736846924,
      "step": 59508
    },
    {
      "epoch": 5.675125122070313e-05,
      "step": 59508,
      "training_step_time": 7.790232419967651
    },
    {
      "epoch": 5.675220489501953e-05,
      "model_forward_time": 1.7493317127227783,
      "step": 59509
    },
    {
      "epoch": 5.675220489501953e-05,
      "step": 59509,
      "training_step_time": 7.635003089904785
    },
    {
      "epoch": 5.6753158569335936e-05,
      "grad_norm": 0.11174064129590988,
      "learning_rate": 3.8514320210576085e-06,
      "loss": 0.0542,
      "step": 59510
    },
    {
      "epoch": 5.6753158569335936e-05,
      "model_forward_time": 1.7374989986419678,
      "step": 59510
    },
    {
      "dynamics_loss_avg": 0.05834717564284801,
      "epoch": 5.6753158569335936e-05,
      "step": 59510
    },
    {
      "action_loss_avg": 0.002040034340461716,
      "epoch": 5.6753158569335936e-05,
      "step": 59510
    },
    {
      "epoch": 5.6753158569335936e-05,
      "step": 59510,
      "training_step_time": 8.741075038909912
    },
    {
      "epoch": 5.6754112243652346e-05,
      "model_forward_time": 1.7365751266479492,
      "step": 59511
    },
    {
      "epoch": 5.6754112243652346e-05,
      "step": 59511,
      "training_step_time": 7.752095460891724
    },
    {
      "epoch": 5.675506591796875e-05,
      "model_forward_time": 1.7013976573944092,
      "step": 59512
    },
    {
      "epoch": 5.675506591796875e-05,
      "step": 59512,
      "training_step_time": 8.526124477386475
    },
    {
      "epoch": 5.6756019592285154e-05,
      "model_forward_time": 1.7466397285461426,
      "step": 59513
    },
    {
      "epoch": 5.6756019592285154e-05,
      "step": 59513,
      "training_step_time": 9.443751096725464
    },
    {
      "epoch": 5.6756973266601565e-05,
      "model_forward_time": 1.7496795654296875,
      "step": 59514
    },
    {
      "epoch": 5.6756973266601565e-05,
      "step": 59514,
      "training_step_time": 9.324803113937378
    },
    {
      "epoch": 5.675792694091797e-05,
      "model_forward_time": 2.8512628078460693,
      "step": 59515
    },
    {
      "epoch": 5.675792694091797e-05,
      "step": 59515,
      "training_step_time": 7.8099284172058105
    },
    {
      "epoch": 5.675888061523437e-05,
      "model_forward_time": 1.7534456253051758,
      "step": 59516
    },
    {
      "epoch": 5.675888061523437e-05,
      "step": 59516,
      "training_step_time": 7.5268940925598145
    },
    {
      "epoch": 5.6759834289550784e-05,
      "model_forward_time": 1.7400147914886475,
      "step": 59517
    },
    {
      "epoch": 5.6759834289550784e-05,
      "step": 59517,
      "training_step_time": 7.767428398132324
    },
    {
      "epoch": 5.676078796386719e-05,
      "model_forward_time": 1.7543656826019287,
      "step": 59518
    },
    {
      "epoch": 5.676078796386719e-05,
      "step": 59518,
      "training_step_time": 7.740191459655762
    },
    {
      "epoch": 5.676174163818359e-05,
      "model_forward_time": 1.7533812522888184,
      "step": 59519
    },
    {
      "epoch": 5.676174163818359e-05,
      "step": 59519,
      "training_step_time": 7.836702823638916
    },
    {
      "epoch": 5.67626953125e-05,
      "grad_norm": 0.12007976323366165,
      "learning_rate": 3.8498228307123056e-06,
      "loss": 0.0543,
      "step": 59520
    },
    {
      "epoch": 5.67626953125e-05,
      "model_forward_time": 1.7528817653656006,
      "step": 59520
    },
    {
      "dynamics_loss_avg": 0.047055512107908724,
      "epoch": 5.67626953125e-05,
      "step": 59520
    },
    {
      "action_loss_avg": 0.0027839980728458615,
      "epoch": 5.67626953125e-05,
      "step": 59520
    },
    {
      "epoch": 5.67626953125e-05,
      "step": 59520,
      "training_step_time": 9.440272331237793
    },
    {
      "epoch": 5.676364898681641e-05,
      "model_forward_time": 1.7772963047027588,
      "step": 59521
    },
    {
      "epoch": 5.676364898681641e-05,
      "step": 59521,
      "training_step_time": 8.068408966064453
    },
    {
      "epoch": 5.676460266113281e-05,
      "model_forward_time": 1.755080223083496,
      "step": 59522
    },
    {
      "epoch": 5.676460266113281e-05,
      "step": 59522,
      "training_step_time": 8.97341513633728
    },
    {
      "epoch": 5.676555633544922e-05,
      "model_forward_time": 1.7530665397644043,
      "step": 59523
    },
    {
      "epoch": 5.676555633544922e-05,
      "step": 59523,
      "training_step_time": 7.763465166091919
    },
    {
      "epoch": 5.6766510009765626e-05,
      "model_forward_time": 2.850529193878174,
      "step": 59524
    },
    {
      "epoch": 5.6766510009765626e-05,
      "step": 59524,
      "training_step_time": 7.818500280380249
    },
    {
      "epoch": 5.676746368408203e-05,
      "model_forward_time": 1.7521636486053467,
      "step": 59525
    },
    {
      "epoch": 5.676746368408203e-05,
      "step": 59525,
      "training_step_time": 8.556213140487671
    },
    {
      "epoch": 5.676841735839844e-05,
      "model_forward_time": 1.7589640617370605,
      "step": 59526
    },
    {
      "epoch": 5.676841735839844e-05,
      "step": 59526,
      "training_step_time": 7.774931192398071
    },
    {
      "epoch": 5.6769371032714845e-05,
      "model_forward_time": 1.7567427158355713,
      "step": 59527
    },
    {
      "epoch": 5.6769371032714845e-05,
      "step": 59527,
      "training_step_time": 7.596410751342773
    },
    {
      "epoch": 5.677032470703125e-05,
      "model_forward_time": 1.743638515472412,
      "step": 59528
    },
    {
      "epoch": 5.677032470703125e-05,
      "step": 59528,
      "training_step_time": 7.7934253215789795
    },
    {
      "epoch": 5.677127838134766e-05,
      "model_forward_time": 1.756962776184082,
      "step": 59529
    },
    {
      "epoch": 5.677127838134766e-05,
      "step": 59529,
      "training_step_time": 7.7416300773620605
    },
    {
      "epoch": 5.6772232055664064e-05,
      "grad_norm": 0.13123489916324615,
      "learning_rate": 3.848213766148649e-06,
      "loss": 0.0579,
      "step": 59530
    },
    {
      "epoch": 5.6772232055664064e-05,
      "model_forward_time": 1.7497310638427734,
      "step": 59530
    },
    {
      "dynamics_loss_avg": 0.05346690453588963,
      "epoch": 5.6772232055664064e-05,
      "step": 59530
    },
    {
      "action_loss_avg": 0.0009709508914966137,
      "epoch": 5.6772232055664064e-05,
      "step": 59530
    },
    {
      "epoch": 5.6772232055664064e-05,
      "step": 59530,
      "training_step_time": 8.89168906211853
    },
    {
      "epoch": 5.677318572998047e-05,
      "model_forward_time": 1.8219528198242188,
      "step": 59531
    },
    {
      "epoch": 5.677318572998047e-05,
      "step": 59531,
      "training_step_time": 8.597726106643677
    },
    {
      "epoch": 5.677413940429687e-05,
      "model_forward_time": 1.7521514892578125,
      "step": 59532
    },
    {
      "epoch": 5.677413940429687e-05,
      "step": 59532,
      "training_step_time": 7.683302402496338
    },
    {
      "epoch": 5.677509307861328e-05,
      "model_forward_time": 2.827439069747925,
      "step": 59533
    },
    {
      "epoch": 5.677509307861328e-05,
      "step": 59533,
      "training_step_time": 9.340105295181274
    },
    {
      "epoch": 5.6776046752929686e-05,
      "model_forward_time": 1.747988224029541,
      "step": 59534
    },
    {
      "epoch": 5.6776046752929686e-05,
      "step": 59534,
      "training_step_time": 7.752852916717529
    },
    {
      "epoch": 5.677700042724609e-05,
      "model_forward_time": 1.7474501132965088,
      "step": 59535
    },
    {
      "epoch": 5.677700042724609e-05,
      "step": 59535,
      "training_step_time": 7.576293468475342
    },
    {
      "epoch": 5.67779541015625e-05,
      "model_forward_time": 1.7514498233795166,
      "step": 59536
    },
    {
      "epoch": 5.67779541015625e-05,
      "step": 59536,
      "training_step_time": 7.738001585006714
    },
    {
      "epoch": 5.6778907775878905e-05,
      "model_forward_time": 1.7526323795318604,
      "step": 59537
    },
    {
      "epoch": 5.6778907775878905e-05,
      "step": 59537,
      "training_step_time": 7.764410734176636
    },
    {
      "epoch": 5.677986145019531e-05,
      "model_forward_time": 1.7545726299285889,
      "step": 59538
    },
    {
      "epoch": 5.677986145019531e-05,
      "step": 59538,
      "training_step_time": 9.652254819869995
    },
    {
      "epoch": 5.678081512451172e-05,
      "model_forward_time": 1.76271390914917,
      "step": 59539
    },
    {
      "epoch": 5.678081512451172e-05,
      "step": 59539,
      "training_step_time": 7.776822805404663
    },
    {
      "epoch": 5.6781768798828124e-05,
      "grad_norm": 0.10491538047790527,
      "learning_rate": 3.846604827542603e-06,
      "loss": 0.0572,
      "step": 59540
    },
    {
      "epoch": 5.6781768798828124e-05,
      "model_forward_time": 1.8132684230804443,
      "step": 59540
    },
    {
      "dynamics_loss_avg": 0.05576195418834686,
      "epoch": 5.6781768798828124e-05,
      "step": 59540
    },
    {
      "action_loss_avg": 0.0012506227241829038,
      "epoch": 5.6781768798828124e-05,
      "step": 59540
    },
    {
      "epoch": 5.6781768798828124e-05,
      "step": 59540,
      "training_step_time": 7.997941970825195
    },
    {
      "epoch": 5.678272247314453e-05,
      "model_forward_time": 1.7413058280944824,
      "step": 59541
    },
    {
      "epoch": 5.678272247314453e-05,
      "step": 59541,
      "training_step_time": 7.746863842010498
    },
    {
      "epoch": 5.678367614746094e-05,
      "model_forward_time": 1.7490813732147217,
      "step": 59542
    },
    {
      "epoch": 5.678367614746094e-05,
      "step": 59542,
      "training_step_time": 9.399331092834473
    },
    {
      "epoch": 5.678462982177734e-05,
      "model_forward_time": 2.9196016788482666,
      "step": 59543
    },
    {
      "epoch": 5.678462982177734e-05,
      "step": 59543,
      "training_step_time": 7.8456504344940186
    },
    {
      "epoch": 5.678558349609375e-05,
      "model_forward_time": 1.7414770126342773,
      "step": 59544
    },
    {
      "epoch": 5.678558349609375e-05,
      "step": 59544,
      "training_step_time": 7.628119945526123
    },
    {
      "epoch": 5.678653717041016e-05,
      "model_forward_time": 1.7496168613433838,
      "step": 59545
    },
    {
      "epoch": 5.678653717041016e-05,
      "step": 59545,
      "training_step_time": 8.784058094024658
    },
    {
      "epoch": 5.678749084472656e-05,
      "model_forward_time": 1.7538797855377197,
      "step": 59546
    },
    {
      "epoch": 5.678749084472656e-05,
      "step": 59546,
      "training_step_time": 7.610194444656372
    },
    {
      "epoch": 5.6788444519042966e-05,
      "model_forward_time": 1.9209883213043213,
      "step": 59547
    },
    {
      "epoch": 5.6788444519042966e-05,
      "step": 59547,
      "training_step_time": 8.363576650619507
    },
    {
      "epoch": 5.678939819335938e-05,
      "model_forward_time": 1.9135830402374268,
      "step": 59548
    },
    {
      "epoch": 5.678939819335938e-05,
      "step": 59548,
      "training_step_time": 7.712606906890869
    },
    {
      "epoch": 5.679035186767578e-05,
      "model_forward_time": 1.9141960144042969,
      "step": 59549
    },
    {
      "epoch": 5.679035186767578e-05,
      "step": 59549,
      "training_step_time": 10.620673894882202
    },
    {
      "epoch": 5.6791305541992185e-05,
      "grad_norm": 0.09594980627298355,
      "learning_rate": 3.8449960150701175e-06,
      "loss": 0.057,
      "step": 59550
    },
    {
      "epoch": 5.6791305541992185e-05,
      "model_forward_time": 1.8994204998016357,
      "step": 59550
    },
    {
      "dynamics_loss_avg": 0.06134357266128063,
      "epoch": 5.6791305541992185e-05,
      "step": 59550
    },
    {
      "action_loss_avg": 0.0012796479335520417,
      "epoch": 5.6791305541992185e-05,
      "step": 59550
    },
    {
      "epoch": 5.6791305541992185e-05,
      "step": 59550,
      "training_step_time": 9.296773910522461
    },
    {
      "epoch": 5.6792259216308596e-05,
      "model_forward_time": 1.9003477096557617,
      "step": 59551
    },
    {
      "epoch": 5.6792259216308596e-05,
      "step": 59551,
      "training_step_time": 8.37973928451538
    },
    {
      "epoch": 5.6793212890625e-05,
      "model_forward_time": 2.964742422103882,
      "step": 59552
    },
    {
      "epoch": 5.6793212890625e-05,
      "step": 59552,
      "training_step_time": 7.841808795928955
    },
    {
      "epoch": 5.6794166564941404e-05,
      "model_forward_time": 1.7425413131713867,
      "step": 59553
    },
    {
      "epoch": 5.6794166564941404e-05,
      "step": 59553,
      "training_step_time": 7.626452207565308
    },
    {
      "epoch": 5.6795120239257814e-05,
      "model_forward_time": 1.7445602416992188,
      "step": 59554
    },
    {
      "epoch": 5.6795120239257814e-05,
      "step": 59554,
      "training_step_time": 7.705732583999634
    },
    {
      "epoch": 5.679607391357422e-05,
      "model_forward_time": 1.7439348697662354,
      "step": 59555
    },
    {
      "epoch": 5.679607391357422e-05,
      "step": 59555,
      "training_step_time": 7.575092315673828
    },
    {
      "epoch": 5.679702758789062e-05,
      "model_forward_time": 1.7466106414794922,
      "step": 59556
    },
    {
      "epoch": 5.679702758789062e-05,
      "step": 59556,
      "training_step_time": 7.763124227523804
    },
    {
      "epoch": 5.679798126220703e-05,
      "model_forward_time": 1.7446751594543457,
      "step": 59557
    },
    {
      "epoch": 5.679798126220703e-05,
      "step": 59557,
      "training_step_time": 8.494463920593262
    },
    {
      "epoch": 5.679893493652344e-05,
      "model_forward_time": 1.8145020008087158,
      "step": 59558
    },
    {
      "epoch": 5.679893493652344e-05,
      "step": 59558,
      "training_step_time": 8.60866379737854
    },
    {
      "epoch": 5.679988861083984e-05,
      "model_forward_time": 1.746211290359497,
      "step": 59559
    },
    {
      "epoch": 5.679988861083984e-05,
      "step": 59559,
      "training_step_time": 11.51999807357788
    },
    {
      "epoch": 5.680084228515625e-05,
      "grad_norm": 0.11323889344930649,
      "learning_rate": 3.843387328907133e-06,
      "loss": 0.0577,
      "step": 59560
    },
    {
      "epoch": 5.680084228515625e-05,
      "model_forward_time": 1.7369048595428467,
      "step": 59560
    },
    {
      "dynamics_loss_avg": 0.04707454554736614,
      "epoch": 5.680084228515625e-05,
      "step": 59560
    },
    {
      "action_loss_avg": 0.0011232043616473675,
      "epoch": 5.680084228515625e-05,
      "step": 59560
    },
    {
      "epoch": 5.680084228515625e-05,
      "step": 59560,
      "training_step_time": 8.552919864654541
    },
    {
      "epoch": 5.6801795959472656e-05,
      "model_forward_time": 2.895784378051758,
      "step": 59561
    },
    {
      "epoch": 5.6801795959472656e-05,
      "step": 59561,
      "training_step_time": 7.771518707275391
    },
    {
      "epoch": 5.680274963378906e-05,
      "model_forward_time": 1.7414796352386475,
      "step": 59562
    },
    {
      "epoch": 5.680274963378906e-05,
      "step": 59562,
      "training_step_time": 7.586026906967163
    },
    {
      "epoch": 5.680370330810547e-05,
      "model_forward_time": 1.7391133308410645,
      "step": 59563
    },
    {
      "epoch": 5.680370330810547e-05,
      "step": 59563,
      "training_step_time": 7.525047302246094
    },
    {
      "epoch": 5.6804656982421875e-05,
      "model_forward_time": 1.745140552520752,
      "step": 59564
    },
    {
      "epoch": 5.6804656982421875e-05,
      "step": 59564,
      "training_step_time": 7.7229413986206055
    },
    {
      "epoch": 5.680561065673828e-05,
      "model_forward_time": 1.7455308437347412,
      "step": 59565
    },
    {
      "epoch": 5.680561065673828e-05,
      "step": 59565,
      "training_step_time": 7.722567319869995
    },
    {
      "epoch": 5.680656433105469e-05,
      "model_forward_time": 1.7514047622680664,
      "step": 59566
    },
    {
      "epoch": 5.680656433105469e-05,
      "step": 59566,
      "training_step_time": 7.745994329452515
    },
    {
      "epoch": 5.6807518005371094e-05,
      "model_forward_time": 1.7946882247924805,
      "step": 59567
    },
    {
      "epoch": 5.6807518005371094e-05,
      "step": 59567,
      "training_step_time": 8.832140684127808
    },
    {
      "epoch": 5.68084716796875e-05,
      "model_forward_time": 1.7520668506622314,
      "step": 59568
    },
    {
      "epoch": 5.68084716796875e-05,
      "step": 59568,
      "training_step_time": 8.472055435180664
    },
    {
      "epoch": 5.680942535400391e-05,
      "model_forward_time": 1.7428958415985107,
      "step": 59569
    },
    {
      "epoch": 5.680942535400391e-05,
      "step": 59569,
      "training_step_time": 8.431326150894165
    },
    {
      "epoch": 5.681037902832031e-05,
      "grad_norm": 0.1815384030342102,
      "learning_rate": 3.84177876922957e-06,
      "loss": 0.0598,
      "step": 59570
    },
    {
      "epoch": 5.681037902832031e-05,
      "model_forward_time": 2.8723480701446533,
      "step": 59570
    },
    {
      "dynamics_loss_avg": 0.053605856373906136,
      "epoch": 5.681037902832031e-05,
      "step": 59570
    },
    {
      "action_loss_avg": 0.001155970519175753,
      "epoch": 5.681037902832031e-05,
      "step": 59570
    },
    {
      "epoch": 5.681037902832031e-05,
      "step": 59570,
      "training_step_time": 8.632243871688843
    },
    {
      "epoch": 5.681133270263672e-05,
      "model_forward_time": 1.7459135055541992,
      "step": 59571
    },
    {
      "epoch": 5.681133270263672e-05,
      "step": 59571,
      "training_step_time": 7.617690324783325
    },
    {
      "epoch": 5.681228637695313e-05,
      "model_forward_time": 1.7455918788909912,
      "step": 59572
    },
    {
      "epoch": 5.681228637695313e-05,
      "step": 59572,
      "training_step_time": 8.447524309158325
    },
    {
      "epoch": 5.681324005126953e-05,
      "model_forward_time": 1.7513887882232666,
      "step": 59573
    },
    {
      "epoch": 5.681324005126953e-05,
      "step": 59573,
      "training_step_time": 7.661368131637573
    },
    {
      "epoch": 5.6814193725585936e-05,
      "model_forward_time": 1.3874361515045166,
      "step": 59574
    },
    {
      "epoch": 5.6814193725585936e-05,
      "step": 59574,
      "training_step_time": 7.595985174179077
    },
    {
      "epoch": 5.6815147399902346e-05,
      "model_forward_time": 1.7424187660217285,
      "step": 59575
    },
    {
      "epoch": 5.6815147399902346e-05,
      "step": 59575,
      "training_step_time": 7.603021144866943
    },
    {
      "epoch": 5.681610107421875e-05,
      "model_forward_time": 1.7417991161346436,
      "step": 59576
    },
    {
      "epoch": 5.681610107421875e-05,
      "step": 59576,
      "training_step_time": 7.604572296142578
    },
    {
      "epoch": 5.6817054748535155e-05,
      "model_forward_time": 1.7405626773834229,
      "step": 59577
    },
    {
      "epoch": 5.6817054748535155e-05,
      "step": 59577,
      "training_step_time": 7.859246015548706
    },
    {
      "epoch": 5.6818008422851565e-05,
      "model_forward_time": 1.7483835220336914,
      "step": 59578
    },
    {
      "epoch": 5.6818008422851565e-05,
      "step": 59578,
      "training_step_time": 8.301462173461914
    },
    {
      "epoch": 5.681896209716797e-05,
      "model_forward_time": 1.7422003746032715,
      "step": 59579
    },
    {
      "epoch": 5.681896209716797e-05,
      "step": 59579,
      "training_step_time": 7.700249195098877
    },
    {
      "epoch": 5.681991577148437e-05,
      "grad_norm": 0.17753759026527405,
      "learning_rate": 3.8401703362133406e-06,
      "loss": 0.0606,
      "step": 59580
    },
    {
      "epoch": 5.681991577148437e-05,
      "model_forward_time": 1.7494513988494873,
      "step": 59580
    },
    {
      "dynamics_loss_avg": 0.06423003803938628,
      "epoch": 5.681991577148437e-05,
      "step": 59580
    },
    {
      "action_loss_avg": 0.001714953887858428,
      "epoch": 5.681991577148437e-05,
      "step": 59580
    },
    {
      "epoch": 5.681991577148437e-05,
      "step": 59580,
      "training_step_time": 7.590737819671631
    },
    {
      "epoch": 5.6820869445800784e-05,
      "model_forward_time": 1.7483890056610107,
      "step": 59581
    },
    {
      "epoch": 5.6820869445800784e-05,
      "step": 59581,
      "training_step_time": 7.475505113601685
    },
    {
      "epoch": 5.682182312011719e-05,
      "model_forward_time": 1.7444818019866943,
      "step": 59582
    },
    {
      "epoch": 5.682182312011719e-05,
      "step": 59582,
      "training_step_time": 8.653515577316284
    },
    {
      "epoch": 5.682277679443359e-05,
      "model_forward_time": 1.7389254570007324,
      "step": 59583
    },
    {
      "epoch": 5.682277679443359e-05,
      "step": 59583,
      "training_step_time": 7.830377578735352
    },
    {
      "epoch": 5.682373046875e-05,
      "model_forward_time": 1.7385656833648682,
      "step": 59584
    },
    {
      "epoch": 5.682373046875e-05,
      "step": 59584,
      "training_step_time": 7.551300764083862
    },
    {
      "epoch": 5.682468414306641e-05,
      "model_forward_time": 1.7402381896972656,
      "step": 59585
    },
    {
      "epoch": 5.682468414306641e-05,
      "step": 59585,
      "training_step_time": 7.5907652378082275
    },
    {
      "epoch": 5.682563781738281e-05,
      "model_forward_time": 1.8214306831359863,
      "step": 59586
    },
    {
      "epoch": 5.682563781738281e-05,
      "step": 59586,
      "training_step_time": 7.782029151916504
    },
    {
      "epoch": 5.682659149169922e-05,
      "model_forward_time": 1.751772403717041,
      "step": 59587
    },
    {
      "epoch": 5.682659149169922e-05,
      "step": 59587,
      "training_step_time": 8.302801609039307
    },
    {
      "epoch": 5.6827545166015626e-05,
      "model_forward_time": 1.7303731441497803,
      "step": 59588
    },
    {
      "epoch": 5.6827545166015626e-05,
      "step": 59588,
      "training_step_time": 9.757847547531128
    },
    {
      "epoch": 5.682849884033203e-05,
      "model_forward_time": 2.9239249229431152,
      "step": 59589
    },
    {
      "epoch": 5.682849884033203e-05,
      "step": 59589,
      "training_step_time": 7.6485395431518555
    },
    {
      "epoch": 5.682945251464844e-05,
      "grad_norm": 0.08716660737991333,
      "learning_rate": 3.8385620300343375e-06,
      "loss": 0.0588,
      "step": 59590
    },
    {
      "epoch": 5.682945251464844e-05,
      "model_forward_time": 1.7455532550811768,
      "step": 59590
    },
    {
      "dynamics_loss_avg": 0.059310225397348405,
      "epoch": 5.682945251464844e-05,
      "step": 59590
    },
    {
      "action_loss_avg": 0.0027350796619430183,
      "epoch": 5.682945251464844e-05,
      "step": 59590
    },
    {
      "epoch": 5.682945251464844e-05,
      "step": 59590,
      "training_step_time": 7.499281644821167
    },
    {
      "epoch": 5.6830406188964845e-05,
      "model_forward_time": 1.7383668422698975,
      "step": 59591
    },
    {
      "epoch": 5.6830406188964845e-05,
      "step": 59591,
      "training_step_time": 7.517238140106201
    },
    {
      "epoch": 5.683135986328125e-05,
      "model_forward_time": 1.7435526847839355,
      "step": 59592
    },
    {
      "epoch": 5.683135986328125e-05,
      "step": 59592,
      "training_step_time": 7.39369797706604
    },
    {
      "epoch": 5.683231353759766e-05,
      "model_forward_time": 1.7380037307739258,
      "step": 59593
    },
    {
      "epoch": 5.683231353759766e-05,
      "step": 59593,
      "training_step_time": 7.641178846359253
    },
    {
      "epoch": 5.6833267211914064e-05,
      "model_forward_time": 1.7396986484527588,
      "step": 59594
    },
    {
      "epoch": 5.6833267211914064e-05,
      "step": 59594,
      "training_step_time": 7.7855424880981445
    },
    {
      "epoch": 5.683422088623047e-05,
      "model_forward_time": 1.8235197067260742,
      "step": 59595
    },
    {
      "epoch": 5.683422088623047e-05,
      "step": 59595,
      "training_step_time": 7.888618469238281
    },
    {
      "epoch": 5.683517456054687e-05,
      "model_forward_time": 1.7465901374816895,
      "step": 59596
    },
    {
      "epoch": 5.683517456054687e-05,
      "step": 59596,
      "training_step_time": 8.956445455551147
    },
    {
      "epoch": 5.683612823486328e-05,
      "model_forward_time": 1.7386698722839355,
      "step": 59597
    },
    {
      "epoch": 5.683612823486328e-05,
      "step": 59597,
      "training_step_time": 8.155599594116211
    },
    {
      "epoch": 5.6837081909179687e-05,
      "model_forward_time": 2.887206792831421,
      "step": 59598
    },
    {
      "epoch": 5.6837081909179687e-05,
      "step": 59598,
      "training_step_time": 7.74314022064209
    },
    {
      "epoch": 5.683803558349609e-05,
      "model_forward_time": 1.7392516136169434,
      "step": 59599
    },
    {
      "epoch": 5.683803558349609e-05,
      "step": 59599,
      "training_step_time": 7.728720664978027
    },
    {
      "epoch": 5.68389892578125e-05,
      "grad_norm": 0.11260238289833069,
      "learning_rate": 3.836953850868447e-06,
      "loss": 0.0581,
      "step": 59600
    },
    {
      "epoch": 5.68389892578125e-05,
      "model_forward_time": 1.723874568939209,
      "step": 59600
    },
    {
      "dynamics_loss_avg": 0.06842308677732944,
      "epoch": 5.68389892578125e-05,
      "step": 59600
    },
    {
      "action_loss_avg": 0.0021367416018620135,
      "epoch": 5.68389892578125e-05,
      "step": 59600
    },
    {
      "epoch": 5.68389892578125e-05,
      "step": 59600,
      "training_step_time": 7.559174537658691
    },
    {
      "epoch": 5.6839942932128905e-05,
      "model_forward_time": 1.7452754974365234,
      "step": 59601
    },
    {
      "epoch": 5.6839942932128905e-05,
      "step": 59601,
      "training_step_time": 7.606127023696899
    },
    {
      "epoch": 5.684089660644531e-05,
      "model_forward_time": 1.7420709133148193,
      "step": 59602
    },
    {
      "epoch": 5.684089660644531e-05,
      "step": 59602,
      "training_step_time": 7.749713182449341
    },
    {
      "epoch": 5.684185028076172e-05,
      "model_forward_time": 1.7399711608886719,
      "step": 59603
    },
    {
      "epoch": 5.684185028076172e-05,
      "step": 59603,
      "training_step_time": 7.739559650421143
    },
    {
      "epoch": 5.6842803955078124e-05,
      "model_forward_time": 1.815147876739502,
      "step": 59604
    },
    {
      "epoch": 5.6842803955078124e-05,
      "step": 59604,
      "training_step_time": 7.794797897338867
    },
    {
      "epoch": 5.684375762939453e-05,
      "model_forward_time": 1.7426795959472656,
      "step": 59605
    },
    {
      "epoch": 5.684375762939453e-05,
      "step": 59605,
      "training_step_time": 8.452720403671265
    },
    {
      "epoch": 5.684471130371094e-05,
      "model_forward_time": 1.7439591884613037,
      "step": 59606
    },
    {
      "epoch": 5.684471130371094e-05,
      "step": 59606,
      "training_step_time": 8.628002643585205
    },
    {
      "epoch": 5.684566497802734e-05,
      "model_forward_time": 2.854767084121704,
      "step": 59607
    },
    {
      "epoch": 5.684566497802734e-05,
      "step": 59607,
      "training_step_time": 8.318679809570312
    },
    {
      "epoch": 5.684661865234375e-05,
      "model_forward_time": 1.7458155155181885,
      "step": 59608
    },
    {
      "epoch": 5.684661865234375e-05,
      "step": 59608,
      "training_step_time": 7.7830281257629395
    },
    {
      "epoch": 5.684757232666016e-05,
      "model_forward_time": 1.7761666774749756,
      "step": 59609
    },
    {
      "epoch": 5.684757232666016e-05,
      "step": 59609,
      "training_step_time": 7.5860395431518555
    },
    {
      "epoch": 5.684852600097656e-05,
      "grad_norm": 0.08803591132164001,
      "learning_rate": 3.835345798891533e-06,
      "loss": 0.0593,
      "step": 59610
    },
    {
      "epoch": 5.684852600097656e-05,
      "model_forward_time": 1.7462158203125,
      "step": 59610
    },
    {
      "dynamics_loss_avg": 0.060773126874119045,
      "epoch": 5.684852600097656e-05,
      "step": 59610
    },
    {
      "action_loss_avg": 0.0021773207816295327,
      "epoch": 5.684852600097656e-05,
      "step": 59610
    },
    {
      "epoch": 5.684852600097656e-05,
      "step": 59610,
      "training_step_time": 7.532974481582642
    },
    {
      "epoch": 5.6849479675292966e-05,
      "model_forward_time": 1.7501287460327148,
      "step": 59611
    },
    {
      "epoch": 5.6849479675292966e-05,
      "step": 59611,
      "training_step_time": 7.701064825057983
    },
    {
      "epoch": 5.685043334960938e-05,
      "model_forward_time": 1.754241704940796,
      "step": 59612
    },
    {
      "epoch": 5.685043334960938e-05,
      "step": 59612,
      "training_step_time": 9.321359634399414
    },
    {
      "epoch": 5.685138702392578e-05,
      "model_forward_time": 1.7680737972259521,
      "step": 59613
    },
    {
      "epoch": 5.685138702392578e-05,
      "step": 59613,
      "training_step_time": 9.285767555236816
    },
    {
      "epoch": 5.6852340698242185e-05,
      "model_forward_time": 1.7459766864776611,
      "step": 59614
    },
    {
      "epoch": 5.6852340698242185e-05,
      "step": 59614,
      "training_step_time": 7.9170756340026855
    },
    {
      "epoch": 5.6853294372558596e-05,
      "model_forward_time": 1.7429413795471191,
      "step": 59615
    },
    {
      "epoch": 5.6853294372558596e-05,
      "step": 59615,
      "training_step_time": 7.664249658584595
    },
    {
      "epoch": 5.6854248046875e-05,
      "model_forward_time": 2.868567943572998,
      "step": 59616
    },
    {
      "epoch": 5.6854248046875e-05,
      "step": 59616,
      "training_step_time": 8.737397909164429
    },
    {
      "epoch": 5.6855201721191404e-05,
      "model_forward_time": 1.7444086074829102,
      "step": 59617
    },
    {
      "epoch": 5.6855201721191404e-05,
      "step": 59617,
      "training_step_time": 8.291776418685913
    },
    {
      "epoch": 5.6856155395507815e-05,
      "model_forward_time": 1.7452504634857178,
      "step": 59618
    },
    {
      "epoch": 5.6856155395507815e-05,
      "step": 59618,
      "training_step_time": 7.737530708312988
    },
    {
      "epoch": 5.685710906982422e-05,
      "model_forward_time": 1.741248369216919,
      "step": 59619
    },
    {
      "epoch": 5.685710906982422e-05,
      "step": 59619,
      "training_step_time": 8.761322021484375
    },
    {
      "epoch": 5.685806274414062e-05,
      "grad_norm": 0.11176817119121552,
      "learning_rate": 3.8337378742794525e-06,
      "loss": 0.0574,
      "step": 59620
    },
    {
      "epoch": 5.685806274414062e-05,
      "model_forward_time": 1.7439730167388916,
      "step": 59620
    },
    {
      "dynamics_loss_avg": 0.06329616252332926,
      "epoch": 5.685806274414062e-05,
      "step": 59620
    },
    {
      "action_loss_avg": 0.0015489429235458374,
      "epoch": 5.685806274414062e-05,
      "step": 59620
    },
    {
      "epoch": 5.685806274414062e-05,
      "step": 59620,
      "training_step_time": 7.640005350112915
    },
    {
      "epoch": 5.685901641845703e-05,
      "model_forward_time": 1.7346091270446777,
      "step": 59621
    },
    {
      "epoch": 5.685901641845703e-05,
      "step": 59621,
      "training_step_time": 8.70754337310791
    },
    {
      "epoch": 5.685997009277344e-05,
      "model_forward_time": 1.736525535583496,
      "step": 59622
    },
    {
      "epoch": 5.685997009277344e-05,
      "step": 59622,
      "training_step_time": 8.27010440826416
    },
    {
      "epoch": 5.686092376708984e-05,
      "model_forward_time": 1.8273975849151611,
      "step": 59623
    },
    {
      "epoch": 5.686092376708984e-05,
      "step": 59623,
      "training_step_time": 7.886904716491699
    },
    {
      "epoch": 5.686187744140625e-05,
      "model_forward_time": 1.3736402988433838,
      "step": 59624
    },
    {
      "epoch": 5.686187744140625e-05,
      "step": 59624,
      "training_step_time": 7.6599602699279785
    },
    {
      "epoch": 5.6862831115722656e-05,
      "model_forward_time": 1.738126516342163,
      "step": 59625
    },
    {
      "epoch": 5.6862831115722656e-05,
      "step": 59625,
      "training_step_time": 7.646254777908325
    },
    {
      "epoch": 5.686378479003906e-05,
      "model_forward_time": 2.924661874771118,
      "step": 59626
    },
    {
      "epoch": 5.686378479003906e-05,
      "step": 59626,
      "training_step_time": 8.788497686386108
    },
    {
      "epoch": 5.686473846435547e-05,
      "model_forward_time": 1.3850421905517578,
      "step": 59627
    },
    {
      "epoch": 5.686473846435547e-05,
      "step": 59627,
      "training_step_time": 7.700883865356445
    },
    {
      "epoch": 5.6865692138671875e-05,
      "model_forward_time": 1.7463054656982422,
      "step": 59628
    },
    {
      "epoch": 5.6865692138671875e-05,
      "step": 59628,
      "training_step_time": 7.685616493225098
    },
    {
      "epoch": 5.686664581298828e-05,
      "model_forward_time": 1.7504186630249023,
      "step": 59629
    },
    {
      "epoch": 5.686664581298828e-05,
      "step": 59629,
      "training_step_time": 7.692784309387207
    },
    {
      "epoch": 5.686759948730469e-05,
      "grad_norm": 0.14843954145908356,
      "learning_rate": 3.832130077208043e-06,
      "loss": 0.06,
      "step": 59630
    },
    {
      "epoch": 5.686759948730469e-05,
      "model_forward_time": 1.7478935718536377,
      "step": 59630
    },
    {
      "dynamics_loss_avg": 0.0612932188436389,
      "epoch": 5.686759948730469e-05,
      "step": 59630
    },
    {
      "action_loss_avg": 0.0012459594930987806,
      "epoch": 5.686759948730469e-05,
      "step": 59630
    },
    {
      "epoch": 5.686759948730469e-05,
      "step": 59630,
      "training_step_time": 8.698687314987183
    },
    {
      "epoch": 5.6868553161621094e-05,
      "model_forward_time": 1.7390530109405518,
      "step": 59631
    },
    {
      "epoch": 5.6868553161621094e-05,
      "step": 59631,
      "training_step_time": 8.877182483673096
    },
    {
      "epoch": 5.68695068359375e-05,
      "model_forward_time": 1.8127517700195312,
      "step": 59632
    },
    {
      "epoch": 5.68695068359375e-05,
      "step": 59632,
      "training_step_time": 8.392306804656982
    },
    {
      "epoch": 5.687046051025391e-05,
      "model_forward_time": 1.7447125911712646,
      "step": 59633
    },
    {
      "epoch": 5.687046051025391e-05,
      "step": 59633,
      "training_step_time": 7.611153602600098
    },
    {
      "epoch": 5.687141418457031e-05,
      "model_forward_time": 1.3642945289611816,
      "step": 59634
    },
    {
      "epoch": 5.687141418457031e-05,
      "step": 59634,
      "training_step_time": 9.460330724716187
    },
    {
      "epoch": 5.687236785888672e-05,
      "model_forward_time": 2.882214069366455,
      "step": 59635
    },
    {
      "epoch": 5.687236785888672e-05,
      "step": 59635,
      "training_step_time": 7.798358678817749
    },
    {
      "epoch": 5.687332153320313e-05,
      "model_forward_time": 1.7541441917419434,
      "step": 59636
    },
    {
      "epoch": 5.687332153320313e-05,
      "step": 59636,
      "training_step_time": 7.759379625320435
    },
    {
      "epoch": 5.687427520751953e-05,
      "model_forward_time": 1.7467944622039795,
      "step": 59637
    },
    {
      "epoch": 5.687427520751953e-05,
      "step": 59637,
      "training_step_time": 7.644369602203369
    },
    {
      "epoch": 5.6875228881835936e-05,
      "model_forward_time": 1.7411625385284424,
      "step": 59638
    },
    {
      "epoch": 5.6875228881835936e-05,
      "step": 59638,
      "training_step_time": 8.164012670516968
    },
    {
      "epoch": 5.6876182556152347e-05,
      "model_forward_time": 1.751507043838501,
      "step": 59639
    },
    {
      "epoch": 5.6876182556152347e-05,
      "step": 59639,
      "training_step_time": 7.636772155761719
    },
    {
      "epoch": 5.687713623046875e-05,
      "grad_norm": 0.10269804298877716,
      "learning_rate": 3.830522407853135e-06,
      "loss": 0.0593,
      "step": 59640
    },
    {
      "epoch": 5.687713623046875e-05,
      "model_forward_time": 1.7495760917663574,
      "step": 59640
    },
    {
      "dynamics_loss_avg": 0.05247277338057756,
      "epoch": 5.687713623046875e-05,
      "step": 59640
    },
    {
      "action_loss_avg": 0.001207689760485664,
      "epoch": 5.687713623046875e-05,
      "step": 59640
    },
    {
      "epoch": 5.687713623046875e-05,
      "step": 59640,
      "training_step_time": 8.74982476234436
    },
    {
      "epoch": 5.6878089904785155e-05,
      "model_forward_time": 1.8081533908843994,
      "step": 59641
    },
    {
      "epoch": 5.6878089904785155e-05,
      "step": 59641,
      "training_step_time": 7.774582624435425
    },
    {
      "epoch": 5.6879043579101565e-05,
      "model_forward_time": 1.7505497932434082,
      "step": 59642
    },
    {
      "epoch": 5.6879043579101565e-05,
      "step": 59642,
      "training_step_time": 9.48210597038269
    },
    {
      "epoch": 5.687999725341797e-05,
      "model_forward_time": 1.7518582344055176,
      "step": 59643
    },
    {
      "epoch": 5.687999725341797e-05,
      "step": 59643,
      "training_step_time": 7.678303003311157
    },
    {
      "epoch": 5.6880950927734373e-05,
      "model_forward_time": 2.8559153079986572,
      "step": 59644
    },
    {
      "epoch": 5.6880950927734373e-05,
      "step": 59644,
      "training_step_time": 7.795708656311035
    },
    {
      "epoch": 5.6881904602050784e-05,
      "model_forward_time": 1.7441890239715576,
      "step": 59645
    },
    {
      "epoch": 5.6881904602050784e-05,
      "step": 59645,
      "training_step_time": 7.70892071723938
    },
    {
      "epoch": 5.688285827636719e-05,
      "model_forward_time": 1.7448863983154297,
      "step": 59646
    },
    {
      "epoch": 5.688285827636719e-05,
      "step": 59646,
      "training_step_time": 7.475703001022339
    },
    {
      "epoch": 5.688381195068359e-05,
      "model_forward_time": 1.371919870376587,
      "step": 59647
    },
    {
      "epoch": 5.688381195068359e-05,
      "step": 59647,
      "training_step_time": 7.7614710330963135
    },
    {
      "epoch": 5.6884765625e-05,
      "model_forward_time": 1.753659963607788,
      "step": 59648
    },
    {
      "epoch": 5.6884765625e-05,
      "step": 59648,
      "training_step_time": 7.700087785720825
    },
    {
      "epoch": 5.688571929931641e-05,
      "model_forward_time": 1.3926517963409424,
      "step": 59649
    },
    {
      "epoch": 5.688571929931641e-05,
      "step": 59649,
      "training_step_time": 7.751415252685547
    },
    {
      "epoch": 5.688667297363281e-05,
      "grad_norm": 0.07834824919700623,
      "learning_rate": 3.828914866390537e-06,
      "loss": 0.0562,
      "step": 59650
    },
    {
      "epoch": 5.688667297363281e-05,
      "model_forward_time": 1.7933955192565918,
      "step": 59650
    },
    {
      "dynamics_loss_avg": 0.05493287872523069,
      "epoch": 5.688667297363281e-05,
      "step": 59650
    },
    {
      "action_loss_avg": 0.0013850364339305088,
      "epoch": 5.688667297363281e-05,
      "step": 59650
    },
    {
      "epoch": 5.688667297363281e-05,
      "step": 59650,
      "training_step_time": 8.829521894454956
    },
    {
      "epoch": 5.688762664794922e-05,
      "model_forward_time": 1.760822057723999,
      "step": 59651
    },
    {
      "epoch": 5.688762664794922e-05,
      "step": 59651,
      "training_step_time": 7.840445280075073
    },
    {
      "epoch": 5.6888580322265626e-05,
      "model_forward_time": 1.7443547248840332,
      "step": 59652
    },
    {
      "epoch": 5.6888580322265626e-05,
      "step": 59652,
      "training_step_time": 7.55941367149353
    },
    {
      "epoch": 5.688953399658203e-05,
      "model_forward_time": 1.7460520267486572,
      "step": 59653
    },
    {
      "epoch": 5.688953399658203e-05,
      "step": 59653,
      "training_step_time": 8.26436471939087
    },
    {
      "epoch": 5.689048767089844e-05,
      "model_forward_time": 1.7465651035308838,
      "step": 59654
    },
    {
      "epoch": 5.689048767089844e-05,
      "step": 59654,
      "training_step_time": 8.781697988510132
    },
    {
      "epoch": 5.6891441345214845e-05,
      "model_forward_time": 1.7480251789093018,
      "step": 59655
    },
    {
      "epoch": 5.6891441345214845e-05,
      "step": 59655,
      "training_step_time": 7.573424339294434
    },
    {
      "epoch": 5.689239501953125e-05,
      "model_forward_time": 1.7399694919586182,
      "step": 59656
    },
    {
      "epoch": 5.689239501953125e-05,
      "step": 59656,
      "training_step_time": 7.715331792831421
    },
    {
      "epoch": 5.689334869384765e-05,
      "model_forward_time": 1.7468647956848145,
      "step": 59657
    },
    {
      "epoch": 5.689334869384765e-05,
      "step": 59657,
      "training_step_time": 7.717592477798462
    },
    {
      "epoch": 5.6894302368164064e-05,
      "model_forward_time": 1.7613742351531982,
      "step": 59658
    },
    {
      "epoch": 5.6894302368164064e-05,
      "step": 59658,
      "training_step_time": 9.412029027938843
    },
    {
      "epoch": 5.689525604248047e-05,
      "model_forward_time": 1.743750810623169,
      "step": 59659
    },
    {
      "epoch": 5.689525604248047e-05,
      "step": 59659,
      "training_step_time": 8.949341773986816
    },
    {
      "epoch": 5.689620971679687e-05,
      "grad_norm": 0.08994472771883011,
      "learning_rate": 3.827307452996048e-06,
      "loss": 0.058,
      "step": 59660
    },
    {
      "epoch": 5.689620971679687e-05,
      "model_forward_time": 1.7541987895965576,
      "step": 59660
    },
    {
      "dynamics_loss_avg": 0.058177345991134645,
      "epoch": 5.689620971679687e-05,
      "step": 59660
    },
    {
      "action_loss_avg": 0.0012877227942226454,
      "epoch": 5.689620971679687e-05,
      "step": 59660
    },
    {
      "epoch": 5.689620971679687e-05,
      "step": 59660,
      "training_step_time": 7.86805272102356
    },
    {
      "epoch": 5.689716339111328e-05,
      "model_forward_time": 1.7422969341278076,
      "step": 59661
    },
    {
      "epoch": 5.689716339111328e-05,
      "step": 59661,
      "training_step_time": 7.689110517501831
    },
    {
      "epoch": 5.6898117065429687e-05,
      "model_forward_time": 1.739211082458496,
      "step": 59662
    },
    {
      "epoch": 5.6898117065429687e-05,
      "step": 59662,
      "training_step_time": 8.726040363311768
    },
    {
      "epoch": 5.689907073974609e-05,
      "model_forward_time": 1.803016185760498,
      "step": 59663
    },
    {
      "epoch": 5.689907073974609e-05,
      "step": 59663,
      "training_step_time": 6.761049270629883
    },
    {
      "epoch": 5.69000244140625e-05,
      "model_forward_time": 1.7507812976837158,
      "step": 59664
    },
    {
      "epoch": 5.69000244140625e-05,
      "step": 59664,
      "training_step_time": 7.736064195632935
    },
    {
      "epoch": 5.6900978088378905e-05,
      "model_forward_time": 1.7474119663238525,
      "step": 59665
    },
    {
      "epoch": 5.6900978088378905e-05,
      "step": 59665,
      "training_step_time": 7.769676923751831
    },
    {
      "epoch": 5.690193176269531e-05,
      "model_forward_time": 1.7477607727050781,
      "step": 59666
    },
    {
      "epoch": 5.690193176269531e-05,
      "step": 59666,
      "training_step_time": 8.615482330322266
    },
    {
      "epoch": 5.690288543701172e-05,
      "model_forward_time": 1.7407572269439697,
      "step": 59667
    },
    {
      "epoch": 5.690288543701172e-05,
      "step": 59667,
      "training_step_time": 7.644212961196899
    },
    {
      "epoch": 5.6903839111328124e-05,
      "model_forward_time": 1.7476143836975098,
      "step": 59668
    },
    {
      "epoch": 5.6903839111328124e-05,
      "step": 59668,
      "training_step_time": 7.864823579788208
    },
    {
      "epoch": 5.690479278564453e-05,
      "model_forward_time": 1.829071283340454,
      "step": 59669
    },
    {
      "epoch": 5.690479278564453e-05,
      "step": 59669,
      "training_step_time": 9.572851419448853
    },
    {
      "epoch": 5.690574645996094e-05,
      "grad_norm": 0.1444646269083023,
      "learning_rate": 3.825700167845454e-06,
      "loss": 0.0556,
      "step": 59670
    },
    {
      "epoch": 5.690574645996094e-05,
      "model_forward_time": 1.7493913173675537,
      "step": 59670
    },
    {
      "dynamics_loss_avg": 0.0651270380243659,
      "epoch": 5.690574645996094e-05,
      "step": 59670
    },
    {
      "action_loss_avg": 0.0019399073353270069,
      "epoch": 5.690574645996094e-05,
      "step": 59670
    },
    {
      "epoch": 5.690574645996094e-05,
      "step": 59670,
      "training_step_time": 7.598780632019043
    },
    {
      "epoch": 5.690670013427734e-05,
      "model_forward_time": 1.7469029426574707,
      "step": 59671
    },
    {
      "epoch": 5.690670013427734e-05,
      "step": 59671,
      "training_step_time": 9.32587480545044
    },
    {
      "epoch": 5.690765380859375e-05,
      "model_forward_time": 1.7565090656280518,
      "step": 59672
    },
    {
      "epoch": 5.690765380859375e-05,
      "step": 59672,
      "training_step_time": 7.72801399230957
    },
    {
      "epoch": 5.690860748291016e-05,
      "model_forward_time": 1.7470829486846924,
      "step": 59673
    },
    {
      "epoch": 5.690860748291016e-05,
      "step": 59673,
      "training_step_time": 7.797260522842407
    },
    {
      "epoch": 5.690956115722656e-05,
      "model_forward_time": 1.7480964660644531,
      "step": 59674
    },
    {
      "epoch": 5.690956115722656e-05,
      "step": 59674,
      "training_step_time": 7.740870237350464
    },
    {
      "epoch": 5.6910514831542966e-05,
      "model_forward_time": 1.751084566116333,
      "step": 59675
    },
    {
      "epoch": 5.6910514831542966e-05,
      "step": 59675,
      "training_step_time": 8.752926349639893
    },
    {
      "epoch": 5.691146850585938e-05,
      "model_forward_time": 1.7508175373077393,
      "step": 59676
    },
    {
      "epoch": 5.691146850585938e-05,
      "step": 59676,
      "training_step_time": 7.798059463500977
    },
    {
      "epoch": 5.691242218017578e-05,
      "model_forward_time": 1.752746343612671,
      "step": 59677
    },
    {
      "epoch": 5.691242218017578e-05,
      "step": 59677,
      "training_step_time": 7.893207550048828
    },
    {
      "epoch": 5.6913375854492185e-05,
      "model_forward_time": 1.825723648071289,
      "step": 59678
    },
    {
      "epoch": 5.6913375854492185e-05,
      "step": 59678,
      "training_step_time": 7.959583282470703
    },
    {
      "epoch": 5.6914329528808596e-05,
      "model_forward_time": 1.7593967914581299,
      "step": 59679
    },
    {
      "epoch": 5.6914329528808596e-05,
      "step": 59679,
      "training_step_time": 9.104016065597534
    },
    {
      "epoch": 5.6915283203125e-05,
      "grad_norm": 0.08415551483631134,
      "learning_rate": 3.824093011114521e-06,
      "loss": 0.0564,
      "step": 59680
    },
    {
      "epoch": 5.6915283203125e-05,
      "model_forward_time": 1.7442004680633545,
      "step": 59680
    },
    {
      "dynamics_loss_avg": 0.058671345561742784,
      "epoch": 5.6915283203125e-05,
      "step": 59680
    },
    {
      "action_loss_avg": 0.0017516100546345116,
      "epoch": 5.6915283203125e-05,
      "step": 59680
    },
    {
      "epoch": 5.6915283203125e-05,
      "step": 59680,
      "training_step_time": 8.855285167694092
    },
    {
      "epoch": 5.6916236877441404e-05,
      "model_forward_time": 2.533782720565796,
      "step": 59681
    },
    {
      "epoch": 5.6916236877441404e-05,
      "step": 59681,
      "training_step_time": 8.628206253051758
    },
    {
      "epoch": 5.6917190551757815e-05,
      "model_forward_time": 1.7556109428405762,
      "step": 59682
    },
    {
      "epoch": 5.6917190551757815e-05,
      "step": 59682,
      "training_step_time": 7.905638217926025
    },
    {
      "epoch": 5.691814422607422e-05,
      "model_forward_time": 1.7458012104034424,
      "step": 59683
    },
    {
      "epoch": 5.691814422607422e-05,
      "step": 59683,
      "training_step_time": 7.889013290405273
    },
    {
      "epoch": 5.691909790039062e-05,
      "model_forward_time": 1.753070592880249,
      "step": 59684
    },
    {
      "epoch": 5.691909790039062e-05,
      "step": 59684,
      "training_step_time": 7.755710601806641
    },
    {
      "epoch": 5.6920051574707033e-05,
      "model_forward_time": 1.7556889057159424,
      "step": 59685
    },
    {
      "epoch": 5.6920051574707033e-05,
      "step": 59685,
      "training_step_time": 7.833262920379639
    },
    {
      "epoch": 5.692100524902344e-05,
      "model_forward_time": 1.7502636909484863,
      "step": 59686
    },
    {
      "epoch": 5.692100524902344e-05,
      "step": 59686,
      "training_step_time": 8.853646278381348
    },
    {
      "epoch": 5.692195892333984e-05,
      "model_forward_time": 1.8188495635986328,
      "step": 59687
    },
    {
      "epoch": 5.692195892333984e-05,
      "step": 59687,
      "training_step_time": 8.554468870162964
    },
    {
      "epoch": 5.692291259765625e-05,
      "model_forward_time": 1.7502326965332031,
      "step": 59688
    },
    {
      "epoch": 5.692291259765625e-05,
      "step": 59688,
      "training_step_time": 8.579079151153564
    },
    {
      "epoch": 5.6923866271972656e-05,
      "model_forward_time": 1.7462611198425293,
      "step": 59689
    },
    {
      "epoch": 5.6923866271972656e-05,
      "step": 59689,
      "training_step_time": 7.682364463806152
    },
    {
      "epoch": 5.692481994628906e-05,
      "grad_norm": 0.11543513089418411,
      "learning_rate": 3.822485982979012e-06,
      "loss": 0.0534,
      "step": 59690
    },
    {
      "epoch": 5.692481994628906e-05,
      "model_forward_time": 2.875562906265259,
      "step": 59690
    },
    {
      "dynamics_loss_avg": 0.052829941362142564,
      "epoch": 5.692481994628906e-05,
      "step": 59690
    },
    {
      "action_loss_avg": 0.0023057857877574863,
      "epoch": 5.692481994628906e-05,
      "step": 59690
    },
    {
      "epoch": 5.692481994628906e-05,
      "step": 59690,
      "training_step_time": 8.64870810508728
    },
    {
      "epoch": 5.692577362060547e-05,
      "model_forward_time": 1.754652738571167,
      "step": 59691
    },
    {
      "epoch": 5.692577362060547e-05,
      "step": 59691,
      "training_step_time": 7.785732746124268
    },
    {
      "epoch": 5.6926727294921875e-05,
      "model_forward_time": 1.7558717727661133,
      "step": 59692
    },
    {
      "epoch": 5.6926727294921875e-05,
      "step": 59692,
      "training_step_time": 8.569374799728394
    },
    {
      "epoch": 5.692768096923828e-05,
      "model_forward_time": 1.7472364902496338,
      "step": 59693
    },
    {
      "epoch": 5.692768096923828e-05,
      "step": 59693,
      "training_step_time": 7.68148946762085
    },
    {
      "epoch": 5.692863464355469e-05,
      "model_forward_time": 1.7513782978057861,
      "step": 59694
    },
    {
      "epoch": 5.692863464355469e-05,
      "step": 59694,
      "training_step_time": 8.314818143844604
    },
    {
      "epoch": 5.6929588317871094e-05,
      "model_forward_time": 1.7515814304351807,
      "step": 59695
    },
    {
      "epoch": 5.6929588317871094e-05,
      "step": 59695,
      "training_step_time": 7.731631755828857
    },
    {
      "epoch": 5.69305419921875e-05,
      "model_forward_time": 1.8207588195800781,
      "step": 59696
    },
    {
      "epoch": 5.69305419921875e-05,
      "step": 59696,
      "training_step_time": 7.840716123580933
    },
    {
      "epoch": 5.693149566650391e-05,
      "model_forward_time": 1.74924635887146,
      "step": 59697
    },
    {
      "epoch": 5.693149566650391e-05,
      "step": 59697,
      "training_step_time": 7.902301549911499
    },
    {
      "epoch": 5.693244934082031e-05,
      "model_forward_time": 1.7507216930389404,
      "step": 59698
    },
    {
      "epoch": 5.693244934082031e-05,
      "step": 59698,
      "training_step_time": 8.221785068511963
    },
    {
      "epoch": 5.693340301513672e-05,
      "model_forward_time": 2.8854992389678955,
      "step": 59699
    },
    {
      "epoch": 5.693340301513672e-05,
      "step": 59699,
      "training_step_time": 7.7061240673065186
    },
    {
      "epoch": 5.693435668945313e-05,
      "grad_norm": 0.09017456322908401,
      "learning_rate": 3.8208790836146644e-06,
      "loss": 0.0574,
      "step": 59700
    },
    {
      "epoch": 5.693435668945313e-05,
      "model_forward_time": 1.7569541931152344,
      "step": 59700
    },
    {
      "dynamics_loss_avg": 0.06644333731383086,
      "epoch": 5.693435668945313e-05,
      "step": 59700
    },
    {
      "action_loss_avg": 0.0019754817622015253,
      "epoch": 5.693435668945313e-05,
      "step": 59700
    },
    {
      "epoch": 5.693435668945313e-05,
      "step": 59700,
      "training_step_time": 8.280198335647583
    },
    {
      "epoch": 5.693531036376953e-05,
      "model_forward_time": 1.741520881652832,
      "step": 59701
    },
    {
      "epoch": 5.693531036376953e-05,
      "step": 59701,
      "training_step_time": 7.777783155441284
    },
    {
      "epoch": 5.6936264038085936e-05,
      "model_forward_time": 1.751699686050415,
      "step": 59702
    },
    {
      "epoch": 5.6936264038085936e-05,
      "step": 59702,
      "training_step_time": 7.683851718902588
    },
    {
      "epoch": 5.6937217712402347e-05,
      "model_forward_time": 1.7221052646636963,
      "step": 59703
    },
    {
      "epoch": 5.6937217712402347e-05,
      "step": 59703,
      "training_step_time": 7.697184801101685
    },
    {
      "epoch": 5.693817138671875e-05,
      "model_forward_time": 1.7443137168884277,
      "step": 59704
    },
    {
      "epoch": 5.693817138671875e-05,
      "step": 59704,
      "training_step_time": 7.724822998046875
    },
    {
      "epoch": 5.6939125061035155e-05,
      "model_forward_time": 1.8110706806182861,
      "step": 59705
    },
    {
      "epoch": 5.6939125061035155e-05,
      "step": 59705,
      "training_step_time": 7.881418228149414
    },
    {
      "epoch": 5.6940078735351565e-05,
      "model_forward_time": 1.7496905326843262,
      "step": 59706
    },
    {
      "epoch": 5.6940078735351565e-05,
      "step": 59706,
      "training_step_time": 8.854362487792969
    },
    {
      "epoch": 5.694103240966797e-05,
      "model_forward_time": 1.7491483688354492,
      "step": 59707
    },
    {
      "epoch": 5.694103240966797e-05,
      "step": 59707,
      "training_step_time": 8.337848424911499
    },
    {
      "epoch": 5.6941986083984373e-05,
      "model_forward_time": 2.852189540863037,
      "step": 59708
    },
    {
      "epoch": 5.6941986083984373e-05,
      "step": 59708,
      "training_step_time": 7.759098529815674
    },
    {
      "epoch": 5.6942939758300784e-05,
      "model_forward_time": 1.74623703956604,
      "step": 59709
    },
    {
      "epoch": 5.6942939758300784e-05,
      "step": 59709,
      "training_step_time": 8.492518424987793
    },
    {
      "epoch": 5.694389343261719e-05,
      "grad_norm": 0.14545121788978577,
      "learning_rate": 3.819272313197208e-06,
      "loss": 0.0528,
      "step": 59710
    },
    {
      "epoch": 5.694389343261719e-05,
      "model_forward_time": 1.7596526145935059,
      "step": 59710
    },
    {
      "dynamics_loss_avg": 0.0652320770546794,
      "epoch": 5.694389343261719e-05,
      "step": 59710
    },
    {
      "action_loss_avg": 0.0016337910084985197,
      "epoch": 5.694389343261719e-05,
      "step": 59710
    },
    {
      "epoch": 5.694389343261719e-05,
      "step": 59710,
      "training_step_time": 7.86102557182312
    },
    {
      "epoch": 5.694484710693359e-05,
      "model_forward_time": 1.7464947700500488,
      "step": 59711
    },
    {
      "epoch": 5.694484710693359e-05,
      "step": 59711,
      "training_step_time": 7.724717378616333
    },
    {
      "epoch": 5.694580078125e-05,
      "model_forward_time": 1.7447509765625,
      "step": 59712
    },
    {
      "epoch": 5.694580078125e-05,
      "step": 59712,
      "training_step_time": 7.754170894622803
    },
    {
      "epoch": 5.694675445556641e-05,
      "model_forward_time": 1.7404084205627441,
      "step": 59713
    },
    {
      "epoch": 5.694675445556641e-05,
      "step": 59713,
      "training_step_time": 8.373853206634521
    },
    {
      "epoch": 5.694770812988281e-05,
      "model_forward_time": 1.8186962604522705,
      "step": 59714
    },
    {
      "epoch": 5.694770812988281e-05,
      "step": 59714,
      "training_step_time": 7.780095338821411
    },
    {
      "epoch": 5.694866180419922e-05,
      "model_forward_time": 1.7535045146942139,
      "step": 59715
    },
    {
      "epoch": 5.694866180419922e-05,
      "step": 59715,
      "training_step_time": 8.835610628128052
    },
    {
      "epoch": 5.6949615478515626e-05,
      "model_forward_time": 1.751084327697754,
      "step": 59716
    },
    {
      "epoch": 5.6949615478515626e-05,
      "step": 59716,
      "training_step_time": 8.39911937713623
    },
    {
      "epoch": 5.695056915283203e-05,
      "model_forward_time": 1.740027666091919,
      "step": 59717
    },
    {
      "epoch": 5.695056915283203e-05,
      "step": 59717,
      "training_step_time": 9.314703702926636
    },
    {
      "epoch": 5.695152282714844e-05,
      "model_forward_time": 1.7428514957427979,
      "step": 59718
    },
    {
      "epoch": 5.695152282714844e-05,
      "step": 59718,
      "training_step_time": 8.588493585586548
    },
    {
      "epoch": 5.6952476501464845e-05,
      "model_forward_time": 1.7483220100402832,
      "step": 59719
    },
    {
      "epoch": 5.6952476501464845e-05,
      "step": 59719,
      "training_step_time": 7.891020059585571
    },
    {
      "epoch": 5.695343017578125e-05,
      "grad_norm": 0.07952869683504105,
      "learning_rate": 3.817665671902356e-06,
      "loss": 0.0518,
      "step": 59720
    },
    {
      "epoch": 5.695343017578125e-05,
      "model_forward_time": 1.7372581958770752,
      "step": 59720
    },
    {
      "dynamics_loss_avg": 0.04493429400026798,
      "epoch": 5.695343017578125e-05,
      "step": 59720
    },
    {
      "action_loss_avg": 0.0013625573075842113,
      "epoch": 5.695343017578125e-05,
      "step": 59720
    },
    {
      "epoch": 5.695343017578125e-05,
      "step": 59720,
      "training_step_time": 8.73839807510376
    },
    {
      "epoch": 5.695438385009765e-05,
      "model_forward_time": 1.7383460998535156,
      "step": 59721
    },
    {
      "epoch": 5.695438385009765e-05,
      "step": 59721,
      "training_step_time": 9.80792236328125
    },
    {
      "epoch": 5.6955337524414064e-05,
      "model_forward_time": 1.7450647354125977,
      "step": 59722
    },
    {
      "epoch": 5.6955337524414064e-05,
      "step": 59722,
      "training_step_time": 6.865253448486328
    },
    {
      "epoch": 5.695629119873047e-05,
      "model_forward_time": 1.7356317043304443,
      "step": 59723
    },
    {
      "epoch": 5.695629119873047e-05,
      "step": 59723,
      "training_step_time": 6.881587028503418
    },
    {
      "epoch": 5.695724487304687e-05,
      "model_forward_time": 1.747145414352417,
      "step": 59724
    },
    {
      "epoch": 5.695724487304687e-05,
      "step": 59724,
      "training_step_time": 6.885265350341797
    },
    {
      "epoch": 5.695819854736328e-05,
      "model_forward_time": 1.734907865524292,
      "step": 59725
    },
    {
      "epoch": 5.695819854736328e-05,
      "step": 59725,
      "training_step_time": 6.829582214355469
    },
    {
      "epoch": 5.695915222167969e-05,
      "model_forward_time": 1.7473926544189453,
      "step": 59726
    },
    {
      "epoch": 5.695915222167969e-05,
      "step": 59726,
      "training_step_time": 6.930457353591919
    },
    {
      "epoch": 5.696010589599609e-05,
      "model_forward_time": 1.7416682243347168,
      "step": 59727
    },
    {
      "epoch": 5.696010589599609e-05,
      "step": 59727,
      "training_step_time": 6.882745265960693
    },
    {
      "epoch": 5.69610595703125e-05,
      "model_forward_time": 1.7423453330993652,
      "step": 59728
    },
    {
      "epoch": 5.69610595703125e-05,
      "step": 59728,
      "training_step_time": 6.846896171569824
    },
    {
      "epoch": 5.6962013244628905e-05,
      "model_forward_time": 1.7537524700164795,
      "step": 59729
    },
    {
      "epoch": 5.6962013244628905e-05,
      "step": 59729,
      "training_step_time": 6.8212785720825195
    },
    {
      "epoch": 5.696296691894531e-05,
      "grad_norm": 0.09295535832643509,
      "learning_rate": 3.816059159905808e-06,
      "loss": 0.0528,
      "step": 59730
    },
    {
      "epoch": 5.696296691894531e-05,
      "model_forward_time": 2.838186502456665,
      "step": 59730
    },
    {
      "dynamics_loss_avg": 0.04879873879253864,
      "epoch": 5.696296691894531e-05,
      "step": 59730
    },
    {
      "action_loss_avg": 0.0016459119971841574,
      "epoch": 5.696296691894531e-05,
      "step": 59730
    },
    {
      "epoch": 5.696296691894531e-05,
      "step": 59730,
      "training_step_time": 7.693898677825928
    },
    {
      "epoch": 5.696392059326172e-05,
      "model_forward_time": 1.751378059387207,
      "step": 59731
    },
    {
      "epoch": 5.696392059326172e-05,
      "step": 59731,
      "training_step_time": 7.925784349441528
    },
    {
      "epoch": 5.6964874267578124e-05,
      "model_forward_time": 1.7603652477264404,
      "step": 59732
    },
    {
      "epoch": 5.6964874267578124e-05,
      "step": 59732,
      "training_step_time": 6.840271472930908
    },
    {
      "epoch": 5.696582794189453e-05,
      "model_forward_time": 1.7544033527374268,
      "step": 59733
    },
    {
      "epoch": 5.696582794189453e-05,
      "step": 59733,
      "training_step_time": 6.93962836265564
    },
    {
      "epoch": 5.696678161621094e-05,
      "model_forward_time": 1.7529680728912354,
      "step": 59734
    },
    {
      "epoch": 5.696678161621094e-05,
      "step": 59734,
      "training_step_time": 6.891522407531738
    },
    {
      "epoch": 5.696773529052734e-05,
      "model_forward_time": 1.7484512329101562,
      "step": 59735
    },
    {
      "epoch": 5.696773529052734e-05,
      "step": 59735,
      "training_step_time": 6.8695924282073975
    },
    {
      "epoch": 5.696868896484375e-05,
      "model_forward_time": 1.7627167701721191,
      "step": 59736
    },
    {
      "epoch": 5.696868896484375e-05,
      "step": 59736,
      "training_step_time": 6.922796010971069
    },
    {
      "epoch": 5.696964263916016e-05,
      "model_forward_time": 1.7545204162597656,
      "step": 59737
    },
    {
      "epoch": 5.696964263916016e-05,
      "step": 59737,
      "training_step_time": 6.895535945892334
    },
    {
      "epoch": 5.697059631347656e-05,
      "model_forward_time": 1.7587504386901855,
      "step": 59738
    },
    {
      "epoch": 5.697059631347656e-05,
      "step": 59738,
      "training_step_time": 6.824651718139648
    },
    {
      "epoch": 5.6971549987792966e-05,
      "model_forward_time": 2.9126923084259033,
      "step": 59739
    },
    {
      "epoch": 5.6971549987792966e-05,
      "step": 59739,
      "training_step_time": 7.760302305221558
    },
    {
      "epoch": 5.697250366210938e-05,
      "grad_norm": 0.0800856500864029,
      "learning_rate": 3.814452777383248e-06,
      "loss": 0.0552,
      "step": 59740
    },
    {
      "epoch": 5.697250366210938e-05,
      "model_forward_time": 1.7646026611328125,
      "step": 59740
    },
    {
      "dynamics_loss_avg": 0.0546183668076992,
      "epoch": 5.697250366210938e-05,
      "step": 59740
    },
    {
      "action_loss_avg": 0.0012994318443816156,
      "epoch": 5.697250366210938e-05,
      "step": 59740
    },
    {
      "epoch": 5.697250366210938e-05,
      "step": 59740,
      "training_step_time": 7.9007415771484375
    },
    {
      "epoch": 5.697345733642578e-05,
      "model_forward_time": 1.7536566257476807,
      "step": 59741
    },
    {
      "epoch": 5.697345733642578e-05,
      "step": 59741,
      "training_step_time": 6.889866352081299
    },
    {
      "epoch": 5.6974411010742185e-05,
      "model_forward_time": 1.7549858093261719,
      "step": 59742
    },
    {
      "epoch": 5.6974411010742185e-05,
      "step": 59742,
      "training_step_time": 6.885746479034424
    },
    {
      "epoch": 5.6975364685058596e-05,
      "model_forward_time": 1.7517905235290527,
      "step": 59743
    },
    {
      "epoch": 5.6975364685058596e-05,
      "step": 59743,
      "training_step_time": 6.963988304138184
    },
    {
      "epoch": 5.6976318359375e-05,
      "model_forward_time": 1.751913070678711,
      "step": 59744
    },
    {
      "epoch": 5.6976318359375e-05,
      "step": 59744,
      "training_step_time": 6.843541383743286
    },
    {
      "epoch": 5.6977272033691404e-05,
      "model_forward_time": 1.757394790649414,
      "step": 59745
    },
    {
      "epoch": 5.6977272033691404e-05,
      "step": 59745,
      "training_step_time": 6.947153806686401
    },
    {
      "epoch": 5.6978225708007815e-05,
      "model_forward_time": 1.7525274753570557,
      "step": 59746
    },
    {
      "epoch": 5.6978225708007815e-05,
      "step": 59746,
      "training_step_time": 6.838897705078125
    },
    {
      "epoch": 5.697917938232422e-05,
      "model_forward_time": 1.7526586055755615,
      "step": 59747
    },
    {
      "epoch": 5.697917938232422e-05,
      "step": 59747,
      "training_step_time": 6.8523101806640625
    },
    {
      "epoch": 5.698013305664062e-05,
      "model_forward_time": 2.9001963138580322,
      "step": 59748
    },
    {
      "epoch": 5.698013305664062e-05,
      "step": 59748,
      "training_step_time": 7.712123870849609
    },
    {
      "epoch": 5.6981086730957033e-05,
      "model_forward_time": 1.7600493431091309,
      "step": 59749
    },
    {
      "epoch": 5.6981086730957033e-05,
      "step": 59749,
      "training_step_time": 7.833844900131226
    },
    {
      "epoch": 5.698204040527344e-05,
      "grad_norm": 0.09868067502975464,
      "learning_rate": 3.8128465245103525e-06,
      "loss": 0.0544,
      "step": 59750
    },
    {
      "epoch": 5.698204040527344e-05,
      "model_forward_time": 1.766509771347046,
      "step": 59750
    },
    {
      "dynamics_loss_avg": 0.049766133353114125,
      "epoch": 5.698204040527344e-05,
      "step": 59750
    },
    {
      "action_loss_avg": 0.0014977297745645047,
      "epoch": 5.698204040527344e-05,
      "step": 59750
    },
    {
      "epoch": 5.698204040527344e-05,
      "step": 59750,
      "training_step_time": 7.883918285369873
    },
    {
      "epoch": 5.698299407958984e-05,
      "model_forward_time": 1.7600657939910889,
      "step": 59751
    },
    {
      "epoch": 5.698299407958984e-05,
      "step": 59751,
      "training_step_time": 6.886464834213257
    },
    {
      "epoch": 5.698394775390625e-05,
      "model_forward_time": 1.7544465065002441,
      "step": 59752
    },
    {
      "epoch": 5.698394775390625e-05,
      "step": 59752,
      "training_step_time": 6.899630546569824
    },
    {
      "epoch": 5.6984901428222656e-05,
      "model_forward_time": 1.7517142295837402,
      "step": 59753
    },
    {
      "epoch": 5.6984901428222656e-05,
      "step": 59753,
      "training_step_time": 6.862521409988403
    },
    {
      "epoch": 5.698585510253906e-05,
      "model_forward_time": 1.7628750801086426,
      "step": 59754
    },
    {
      "epoch": 5.698585510253906e-05,
      "step": 59754,
      "training_step_time": 6.779893398284912
    },
    {
      "epoch": 5.698680877685547e-05,
      "model_forward_time": 1.0678036212921143,
      "step": 59755
    },
    {
      "epoch": 5.698680877685547e-05,
      "step": 59755,
      "training_step_time": 6.873240947723389
    },
    {
      "epoch": 5.6987762451171875e-05,
      "model_forward_time": 1.7485408782958984,
      "step": 59756
    },
    {
      "epoch": 5.6987762451171875e-05,
      "step": 59756,
      "training_step_time": 6.820465564727783
    },
    {
      "epoch": 5.698871612548828e-05,
      "model_forward_time": 3.004408836364746,
      "step": 59757
    },
    {
      "epoch": 5.698871612548828e-05,
      "step": 59757,
      "training_step_time": 7.833337068557739
    },
    {
      "epoch": 5.698966979980469e-05,
      "model_forward_time": 1.7615406513214111,
      "step": 59758
    },
    {
      "epoch": 5.698966979980469e-05,
      "step": 59758,
      "training_step_time": 9.31595492362976
    },
    {
      "epoch": 5.6990623474121094e-05,
      "model_forward_time": 1.7686402797698975,
      "step": 59759
    },
    {
      "epoch": 5.6990623474121094e-05,
      "step": 59759,
      "training_step_time": 7.689202070236206
    },
    {
      "epoch": 5.69915771484375e-05,
      "grad_norm": 0.12303490936756134,
      "learning_rate": 3.8112404014627755e-06,
      "loss": 0.0545,
      "step": 59760
    },
    {
      "epoch": 5.69915771484375e-05,
      "model_forward_time": 1.7610106468200684,
      "step": 59760
    },
    {
      "dynamics_loss_avg": 0.05085650598630309,
      "epoch": 5.69915771484375e-05,
      "step": 59760
    },
    {
      "action_loss_avg": 0.0012152218580013141,
      "epoch": 5.69915771484375e-05,
      "step": 59760
    },
    {
      "epoch": 5.69915771484375e-05,
      "step": 59760,
      "training_step_time": 6.847646474838257
    },
    {
      "epoch": 5.699253082275391e-05,
      "model_forward_time": 1.7630987167358398,
      "step": 59761
    },
    {
      "epoch": 5.699253082275391e-05,
      "step": 59761,
      "training_step_time": 6.823160171508789
    },
    {
      "epoch": 5.699348449707031e-05,
      "model_forward_time": 1.7660191059112549,
      "step": 59762
    },
    {
      "epoch": 5.699348449707031e-05,
      "step": 59762,
      "training_step_time": 6.811097860336304
    },
    {
      "epoch": 5.699443817138672e-05,
      "model_forward_time": 1.7565572261810303,
      "step": 59763
    },
    {
      "epoch": 5.699443817138672e-05,
      "step": 59763,
      "training_step_time": 6.834078311920166
    },
    {
      "epoch": 5.699539184570313e-05,
      "model_forward_time": 1.7670152187347412,
      "step": 59764
    },
    {
      "epoch": 5.699539184570313e-05,
      "step": 59764,
      "training_step_time": 6.787454843521118
    },
    {
      "epoch": 5.699634552001953e-05,
      "model_forward_time": 1.7564396858215332,
      "step": 59765
    },
    {
      "epoch": 5.699634552001953e-05,
      "step": 59765,
      "training_step_time": 6.853102684020996
    },
    {
      "epoch": 5.6997299194335936e-05,
      "model_forward_time": 2.9500932693481445,
      "step": 59766
    },
    {
      "epoch": 5.6997299194335936e-05,
      "step": 59766,
      "training_step_time": 7.786516904830933
    },
    {
      "epoch": 5.699825286865235e-05,
      "model_forward_time": 1.766230821609497,
      "step": 59767
    },
    {
      "epoch": 5.699825286865235e-05,
      "step": 59767,
      "training_step_time": 8.932416200637817
    },
    {
      "epoch": 5.699920654296875e-05,
      "model_forward_time": 1.7604081630706787,
      "step": 59768
    },
    {
      "epoch": 5.699920654296875e-05,
      "step": 59768,
      "training_step_time": 7.817956447601318
    },
    {
      "epoch": 5.7000160217285155e-05,
      "model_forward_time": 1.7572510242462158,
      "step": 59769
    },
    {
      "epoch": 5.7000160217285155e-05,
      "step": 59769,
      "training_step_time": 6.806838750839233
    },
    {
      "epoch": 5.7001113891601565e-05,
      "grad_norm": 0.09092266857624054,
      "learning_rate": 3.809634408416162e-06,
      "loss": 0.0529,
      "step": 59770
    },
    {
      "epoch": 5.7001113891601565e-05,
      "model_forward_time": 1.7608659267425537,
      "step": 59770
    },
    {
      "dynamics_loss_avg": 0.03847537245601416,
      "epoch": 5.7001113891601565e-05,
      "step": 59770
    },
    {
      "action_loss_avg": 0.0013595796946901828,
      "epoch": 5.7001113891601565e-05,
      "step": 59770
    },
    {
      "epoch": 5.7001113891601565e-05,
      "step": 59770,
      "training_step_time": 6.8282082080841064
    },
    {
      "epoch": 5.700206756591797e-05,
      "model_forward_time": 1.7616360187530518,
      "step": 59771
    },
    {
      "epoch": 5.700206756591797e-05,
      "step": 59771,
      "training_step_time": 6.881341934204102
    },
    {
      "epoch": 5.7003021240234374e-05,
      "model_forward_time": 1.7643003463745117,
      "step": 59772
    },
    {
      "epoch": 5.7003021240234374e-05,
      "step": 59772,
      "training_step_time": 6.859568119049072
    },
    {
      "epoch": 5.7003974914550784e-05,
      "model_forward_time": 1.7568566799163818,
      "step": 59773
    },
    {
      "epoch": 5.7003974914550784e-05,
      "step": 59773,
      "training_step_time": 6.880508899688721
    },
    {
      "epoch": 5.700492858886719e-05,
      "model_forward_time": 1.7672710418701172,
      "step": 59774
    },
    {
      "epoch": 5.700492858886719e-05,
      "step": 59774,
      "training_step_time": 6.8871777057647705
    },
    {
      "epoch": 5.700588226318359e-05,
      "model_forward_time": 1.8358044624328613,
      "step": 59775
    },
    {
      "epoch": 5.700588226318359e-05,
      "step": 59775,
      "training_step_time": 7.115935325622559
    },
    {
      "epoch": 5.70068359375e-05,
      "model_forward_time": 2.9261035919189453,
      "step": 59776
    },
    {
      "epoch": 5.70068359375e-05,
      "step": 59776,
      "training_step_time": 8.82139277458191
    },
    {
      "epoch": 5.700778961181641e-05,
      "model_forward_time": 1.7564058303833008,
      "step": 59777
    },
    {
      "epoch": 5.700778961181641e-05,
      "step": 59777,
      "training_step_time": 8.646504163742065
    },
    {
      "epoch": 5.700874328613281e-05,
      "model_forward_time": 1.7734284400939941,
      "step": 59778
    },
    {
      "epoch": 5.700874328613281e-05,
      "step": 59778,
      "training_step_time": 6.777454853057861
    },
    {
      "epoch": 5.700969696044922e-05,
      "model_forward_time": 1.7626969814300537,
      "step": 59779
    },
    {
      "epoch": 5.700969696044922e-05,
      "step": 59779,
      "training_step_time": 6.819072961807251
    },
    {
      "epoch": 5.7010650634765626e-05,
      "grad_norm": 0.17673659324645996,
      "learning_rate": 3.8080285455461396e-06,
      "loss": 0.0547,
      "step": 59780
    },
    {
      "epoch": 5.7010650634765626e-05,
      "model_forward_time": 1.7768003940582275,
      "step": 59780
    },
    {
      "dynamics_loss_avg": 0.046255902573466304,
      "epoch": 5.7010650634765626e-05,
      "step": 59780
    },
    {
      "action_loss_avg": 0.0012567288358695805,
      "epoch": 5.7010650634765626e-05,
      "step": 59780
    },
    {
      "epoch": 5.7010650634765626e-05,
      "step": 59780,
      "training_step_time": 6.897508859634399
    },
    {
      "epoch": 5.701160430908203e-05,
      "model_forward_time": 1.7677459716796875,
      "step": 59781
    },
    {
      "epoch": 5.701160430908203e-05,
      "step": 59781,
      "training_step_time": 6.818731307983398
    },
    {
      "epoch": 5.7012557983398434e-05,
      "model_forward_time": 1.7665379047393799,
      "step": 59782
    },
    {
      "epoch": 5.7012557983398434e-05,
      "step": 59782,
      "training_step_time": 6.86785888671875
    },
    {
      "epoch": 5.7013511657714845e-05,
      "model_forward_time": 1.7515869140625,
      "step": 59783
    },
    {
      "epoch": 5.7013511657714845e-05,
      "step": 59783,
      "training_step_time": 6.869235992431641
    },
    {
      "epoch": 5.701446533203125e-05,
      "model_forward_time": 1.8057818412780762,
      "step": 59784
    },
    {
      "epoch": 5.701446533203125e-05,
      "step": 59784,
      "training_step_time": 7.097797632217407
    },
    {
      "epoch": 5.701541900634765e-05,
      "model_forward_time": 2.930480718612671,
      "step": 59785
    },
    {
      "epoch": 5.701541900634765e-05,
      "step": 59785,
      "training_step_time": 7.7826385498046875
    },
    {
      "epoch": 5.7016372680664064e-05,
      "model_forward_time": 1.7612478733062744,
      "step": 59786
    },
    {
      "epoch": 5.7016372680664064e-05,
      "step": 59786,
      "training_step_time": 9.455716848373413
    },
    {
      "epoch": 5.701732635498047e-05,
      "model_forward_time": 1.7521514892578125,
      "step": 59787
    },
    {
      "epoch": 5.701732635498047e-05,
      "step": 59787,
      "training_step_time": 7.668800592422485
    },
    {
      "epoch": 5.701828002929687e-05,
      "model_forward_time": 1.7714552879333496,
      "step": 59788
    },
    {
      "epoch": 5.701828002929687e-05,
      "step": 59788,
      "training_step_time": 6.8431689739227295
    },
    {
      "epoch": 5.701923370361328e-05,
      "model_forward_time": 1.7573556900024414,
      "step": 59789
    },
    {
      "epoch": 5.701923370361328e-05,
      "step": 59789,
      "training_step_time": 6.8137593269348145
    },
    {
      "epoch": 5.702018737792969e-05,
      "grad_norm": 0.1015859916806221,
      "learning_rate": 3.8064228130283232e-06,
      "loss": 0.0568,
      "step": 59790
    },
    {
      "epoch": 5.702018737792969e-05,
      "model_forward_time": 1.7575829029083252,
      "step": 59790
    },
    {
      "dynamics_loss_avg": 0.040880062058568,
      "epoch": 5.702018737792969e-05,
      "step": 59790
    },
    {
      "action_loss_avg": 0.001156715489923954,
      "epoch": 5.702018737792969e-05,
      "step": 59790
    },
    {
      "epoch": 5.702018737792969e-05,
      "step": 59790,
      "training_step_time": 6.804344415664673
    },
    {
      "epoch": 5.702114105224609e-05,
      "model_forward_time": 1.7491819858551025,
      "step": 59791
    },
    {
      "epoch": 5.702114105224609e-05,
      "step": 59791,
      "training_step_time": 6.821671009063721
    },
    {
      "epoch": 5.70220947265625e-05,
      "model_forward_time": 1.7490978240966797,
      "step": 59792
    },
    {
      "epoch": 5.70220947265625e-05,
      "step": 59792,
      "training_step_time": 6.824651718139648
    },
    {
      "epoch": 5.7023048400878906e-05,
      "model_forward_time": 1.8152878284454346,
      "step": 59793
    },
    {
      "epoch": 5.7023048400878906e-05,
      "step": 59793,
      "training_step_time": 7.08102822303772
    },
    {
      "epoch": 5.702400207519531e-05,
      "model_forward_time": 2.8968679904937744,
      "step": 59794
    },
    {
      "epoch": 5.702400207519531e-05,
      "step": 59794,
      "training_step_time": 7.705763339996338
    },
    {
      "epoch": 5.702495574951172e-05,
      "model_forward_time": 1.7487890720367432,
      "step": 59795
    },
    {
      "epoch": 5.702495574951172e-05,
      "step": 59795,
      "training_step_time": 8.971539735794067
    },
    {
      "epoch": 5.7025909423828124e-05,
      "model_forward_time": 1.7467854022979736,
      "step": 59796
    },
    {
      "epoch": 5.7025909423828124e-05,
      "step": 59796,
      "training_step_time": 7.762540817260742
    },
    {
      "epoch": 5.702686309814453e-05,
      "model_forward_time": 1.746384859085083,
      "step": 59797
    },
    {
      "epoch": 5.702686309814453e-05,
      "step": 59797,
      "training_step_time": 6.808727741241455
    },
    {
      "epoch": 5.702781677246094e-05,
      "model_forward_time": 1.745917797088623,
      "step": 59798
    },
    {
      "epoch": 5.702781677246094e-05,
      "step": 59798,
      "training_step_time": 6.82976508140564
    },
    {
      "epoch": 5.702877044677734e-05,
      "model_forward_time": 1.7554123401641846,
      "step": 59799
    },
    {
      "epoch": 5.702877044677734e-05,
      "step": 59799,
      "training_step_time": 6.885205030441284
    },
    {
      "epoch": 5.702972412109375e-05,
      "grad_norm": 0.11830472201108932,
      "learning_rate": 3.8048172110383116e-06,
      "loss": 0.0576,
      "step": 59800
    },
    {
      "epoch": 5.702972412109375e-05,
      "model_forward_time": 1.753511905670166,
      "step": 59800
    },
    {
      "dynamics_loss_avg": 0.059039054065942766,
      "epoch": 5.702972412109375e-05,
      "step": 59800
    },
    {
      "action_loss_avg": 0.0013397437141975388,
      "epoch": 5.702972412109375e-05,
      "step": 59800
    },
    {
      "epoch": 5.702972412109375e-05,
      "step": 59800,
      "training_step_time": 6.824244260787964
    },
    {
      "epoch": 5.703067779541016e-05,
      "model_forward_time": 1.7527027130126953,
      "step": 59801
    },
    {
      "epoch": 5.703067779541016e-05,
      "step": 59801,
      "training_step_time": 6.853788375854492
    },
    {
      "epoch": 5.703163146972656e-05,
      "model_forward_time": 1.8347198963165283,
      "step": 59802
    },
    {
      "epoch": 5.703163146972656e-05,
      "step": 59802,
      "training_step_time": 7.155707120895386
    },
    {
      "epoch": 5.7032585144042966e-05,
      "model_forward_time": 2.881629705429077,
      "step": 59803
    },
    {
      "epoch": 5.7032585144042966e-05,
      "step": 59803,
      "training_step_time": 8.72548794746399
    },
    {
      "epoch": 5.703353881835938e-05,
      "model_forward_time": 1.7463698387145996,
      "step": 59804
    },
    {
      "epoch": 5.703353881835938e-05,
      "step": 59804,
      "training_step_time": 9.62870478630066
    },
    {
      "epoch": 5.703449249267578e-05,
      "model_forward_time": 1.7510898113250732,
      "step": 59805
    },
    {
      "epoch": 5.703449249267578e-05,
      "step": 59805,
      "training_step_time": 7.840602159500122
    },
    {
      "epoch": 5.7035446166992185e-05,
      "model_forward_time": 1.7488999366760254,
      "step": 59806
    },
    {
      "epoch": 5.7035446166992185e-05,
      "step": 59806,
      "training_step_time": 6.759883880615234
    },
    {
      "epoch": 5.7036399841308596e-05,
      "model_forward_time": 1.7532436847686768,
      "step": 59807
    },
    {
      "epoch": 5.7036399841308596e-05,
      "step": 59807,
      "training_step_time": 6.768598794937134
    },
    {
      "epoch": 5.7037353515625e-05,
      "model_forward_time": 1.7385859489440918,
      "step": 59808
    },
    {
      "epoch": 5.7037353515625e-05,
      "step": 59808,
      "training_step_time": 6.798063278198242
    },
    {
      "epoch": 5.7038307189941404e-05,
      "model_forward_time": 1.7665607929229736,
      "step": 59809
    },
    {
      "epoch": 5.7038307189941404e-05,
      "step": 59809,
      "training_step_time": 6.789596080780029
    },
    {
      "epoch": 5.7039260864257815e-05,
      "grad_norm": 0.07875465601682663,
      "learning_rate": 3.8032117397516945e-06,
      "loss": 0.0631,
      "step": 59810
    },
    {
      "epoch": 5.7039260864257815e-05,
      "model_forward_time": 1.7684924602508545,
      "step": 59810
    },
    {
      "dynamics_loss_avg": 0.05944141652435064,
      "epoch": 5.7039260864257815e-05,
      "step": 59810
    },
    {
      "action_loss_avg": 0.0013028183311689645,
      "epoch": 5.7039260864257815e-05,
      "step": 59810
    },
    {
      "epoch": 5.7039260864257815e-05,
      "step": 59810,
      "training_step_time": 6.806830167770386
    },
    {
      "epoch": 5.704021453857422e-05,
      "model_forward_time": 1.8010451793670654,
      "step": 59811
    },
    {
      "epoch": 5.704021453857422e-05,
      "step": 59811,
      "training_step_time": 6.972013711929321
    },
    {
      "epoch": 5.704116821289062e-05,
      "model_forward_time": 2.8902676105499268,
      "step": 59812
    },
    {
      "epoch": 5.704116821289062e-05,
      "step": 59812,
      "training_step_time": 7.59266471862793
    },
    {
      "epoch": 5.7042121887207034e-05,
      "model_forward_time": 1.743255853652954,
      "step": 59813
    },
    {
      "epoch": 5.7042121887207034e-05,
      "step": 59813,
      "training_step_time": 12.503448247909546
    },
    {
      "epoch": 5.704307556152344e-05,
      "model_forward_time": 1.7440528869628906,
      "step": 59814
    },
    {
      "epoch": 5.704307556152344e-05,
      "step": 59814,
      "training_step_time": 8.48508596420288
    },
    {
      "epoch": 5.704402923583984e-05,
      "model_forward_time": 1.3919920921325684,
      "step": 59815
    },
    {
      "epoch": 5.704402923583984e-05,
      "step": 59815,
      "training_step_time": 7.571839809417725
    },
    {
      "epoch": 5.704498291015625e-05,
      "model_forward_time": 1.751969337463379,
      "step": 59816
    },
    {
      "epoch": 5.704498291015625e-05,
      "step": 59816,
      "training_step_time": 6.776778697967529
    },
    {
      "epoch": 5.7045936584472656e-05,
      "model_forward_time": 1.7457685470581055,
      "step": 59817
    },
    {
      "epoch": 5.7045936584472656e-05,
      "step": 59817,
      "training_step_time": 6.753726482391357
    },
    {
      "epoch": 5.704689025878906e-05,
      "model_forward_time": 1.7505648136138916,
      "step": 59818
    },
    {
      "epoch": 5.704689025878906e-05,
      "step": 59818,
      "training_step_time": 6.704474210739136
    },
    {
      "epoch": 5.704784393310547e-05,
      "model_forward_time": 1.7517485618591309,
      "step": 59819
    },
    {
      "epoch": 5.704784393310547e-05,
      "step": 59819,
      "training_step_time": 6.748993873596191
    },
    {
      "epoch": 5.7048797607421875e-05,
      "grad_norm": 0.13184285163879395,
      "learning_rate": 3.801606399344042e-06,
      "loss": 0.057,
      "step": 59820
    },
    {
      "epoch": 5.7048797607421875e-05,
      "model_forward_time": 1.8058018684387207,
      "step": 59820
    },
    {
      "dynamics_loss_avg": 0.05376632995903492,
      "epoch": 5.7048797607421875e-05,
      "step": 59820
    },
    {
      "action_loss_avg": 0.0012604274263139815,
      "epoch": 5.7048797607421875e-05,
      "step": 59820
    },
    {
      "epoch": 5.7048797607421875e-05,
      "step": 59820,
      "training_step_time": 6.955629587173462
    },
    {
      "epoch": 5.704975128173828e-05,
      "model_forward_time": 1.7539126873016357,
      "step": 59821
    },
    {
      "epoch": 5.704975128173828e-05,
      "step": 59821,
      "training_step_time": 6.7215516567230225
    },
    {
      "epoch": 5.705070495605469e-05,
      "model_forward_time": 2.9368319511413574,
      "step": 59822
    },
    {
      "epoch": 5.705070495605469e-05,
      "step": 59822,
      "training_step_time": 8.479521989822388
    },
    {
      "epoch": 5.7051658630371094e-05,
      "model_forward_time": 1.7448339462280273,
      "step": 59823
    },
    {
      "epoch": 5.7051658630371094e-05,
      "step": 59823,
      "training_step_time": 7.765366554260254
    },
    {
      "epoch": 5.70526123046875e-05,
      "model_forward_time": 1.7423255443572998,
      "step": 59824
    },
    {
      "epoch": 5.70526123046875e-05,
      "step": 59824,
      "training_step_time": 7.737159490585327
    },
    {
      "epoch": 5.705356597900391e-05,
      "model_forward_time": 1.7695419788360596,
      "step": 59825
    },
    {
      "epoch": 5.705356597900391e-05,
      "step": 59825,
      "training_step_time": 6.727981328964233
    },
    {
      "epoch": 5.705451965332031e-05,
      "model_forward_time": 1.7746410369873047,
      "step": 59826
    },
    {
      "epoch": 5.705451965332031e-05,
      "step": 59826,
      "training_step_time": 6.7132956981658936
    },
    {
      "epoch": 5.705547332763672e-05,
      "model_forward_time": 1.7574081420898438,
      "step": 59827
    },
    {
      "epoch": 5.705547332763672e-05,
      "step": 59827,
      "training_step_time": 6.7451488971710205
    },
    {
      "epoch": 5.705642700195313e-05,
      "model_forward_time": 1.7516825199127197,
      "step": 59828
    },
    {
      "epoch": 5.705642700195313e-05,
      "step": 59828,
      "training_step_time": 6.757617950439453
    },
    {
      "epoch": 5.705738067626953e-05,
      "model_forward_time": 1.8129239082336426,
      "step": 59829
    },
    {
      "epoch": 5.705738067626953e-05,
      "step": 59829,
      "training_step_time": 6.995415449142456
    },
    {
      "epoch": 5.7058334350585936e-05,
      "grad_norm": 0.11810764670372009,
      "learning_rate": 3.800001189990913e-06,
      "loss": 0.0615,
      "step": 59830
    },
    {
      "epoch": 5.7058334350585936e-05,
      "model_forward_time": 1.7681865692138672,
      "step": 59830
    },
    {
      "dynamics_loss_avg": 0.058456819877028464,
      "epoch": 5.7058334350585936e-05,
      "step": 59830
    },
    {
      "action_loss_avg": 0.0015123649965971709,
      "epoch": 5.7058334350585936e-05,
      "step": 59830
    },
    {
      "epoch": 5.7058334350585936e-05,
      "step": 59830,
      "training_step_time": 6.749478101730347
    },
    {
      "epoch": 5.705928802490235e-05,
      "model_forward_time": 2.5761091709136963,
      "step": 59831
    },
    {
      "epoch": 5.705928802490235e-05,
      "step": 59831,
      "training_step_time": 7.457839250564575
    },
    {
      "epoch": 5.706024169921875e-05,
      "model_forward_time": 1.7441024780273438,
      "step": 59832
    },
    {
      "epoch": 5.706024169921875e-05,
      "step": 59832,
      "training_step_time": 7.74838924407959
    },
    {
      "epoch": 5.7061195373535155e-05,
      "model_forward_time": 1.752206802368164,
      "step": 59833
    },
    {
      "epoch": 5.7061195373535155e-05,
      "step": 59833,
      "training_step_time": 7.731558322906494
    },
    {
      "epoch": 5.7062149047851566e-05,
      "model_forward_time": 1.377882719039917,
      "step": 59834
    },
    {
      "epoch": 5.7062149047851566e-05,
      "step": 59834,
      "training_step_time": 6.727897882461548
    },
    {
      "epoch": 5.706310272216797e-05,
      "model_forward_time": 1.7428982257843018,
      "step": 59835
    },
    {
      "epoch": 5.706310272216797e-05,
      "step": 59835,
      "training_step_time": 6.702728748321533
    },
    {
      "epoch": 5.7064056396484374e-05,
      "model_forward_time": 1.7558634281158447,
      "step": 59836
    },
    {
      "epoch": 5.7064056396484374e-05,
      "step": 59836,
      "training_step_time": 6.773044109344482
    },
    {
      "epoch": 5.7065010070800784e-05,
      "model_forward_time": 1.7483863830566406,
      "step": 59837
    },
    {
      "epoch": 5.7065010070800784e-05,
      "step": 59837,
      "training_step_time": 6.789141893386841
    },
    {
      "epoch": 5.706596374511719e-05,
      "model_forward_time": 1.8075134754180908,
      "step": 59838
    },
    {
      "epoch": 5.706596374511719e-05,
      "step": 59838,
      "training_step_time": 7.057939529418945
    },
    {
      "epoch": 5.706691741943359e-05,
      "model_forward_time": 1.7659709453582764,
      "step": 59839
    },
    {
      "epoch": 5.706691741943359e-05,
      "step": 59839,
      "training_step_time": 6.740222215652466
    },
    {
      "epoch": 5.706787109375e-05,
      "grad_norm": 0.09376523643732071,
      "learning_rate": 3.7983961118678474e-06,
      "loss": 0.0607,
      "step": 59840
    },
    {
      "epoch": 5.706787109375e-05,
      "model_forward_time": 2.861189842224121,
      "step": 59840
    },
    {
      "dynamics_loss_avg": 0.06433694623410702,
      "epoch": 5.706787109375e-05,
      "step": 59840
    },
    {
      "action_loss_avg": 0.0009606116655049846,
      "epoch": 5.706787109375e-05,
      "step": 59840
    },
    {
      "epoch": 5.706787109375e-05,
      "step": 59840,
      "training_step_time": 8.869500398635864
    },
    {
      "epoch": 5.706882476806641e-05,
      "model_forward_time": 1.7550415992736816,
      "step": 59841
    },
    {
      "epoch": 5.706882476806641e-05,
      "step": 59841,
      "training_step_time": 8.591222286224365
    },
    {
      "epoch": 5.706977844238281e-05,
      "model_forward_time": 1.7500436305999756,
      "step": 59842
    },
    {
      "epoch": 5.706977844238281e-05,
      "step": 59842,
      "training_step_time": 7.687733173370361
    },
    {
      "epoch": 5.707073211669922e-05,
      "model_forward_time": 1.7490787506103516,
      "step": 59843
    },
    {
      "epoch": 5.707073211669922e-05,
      "step": 59843,
      "training_step_time": 7.462910890579224
    },
    {
      "epoch": 5.7071685791015626e-05,
      "model_forward_time": 1.7596311569213867,
      "step": 59844
    },
    {
      "epoch": 5.7071685791015626e-05,
      "step": 59844,
      "training_step_time": 6.721552610397339
    },
    {
      "epoch": 5.707263946533203e-05,
      "model_forward_time": 1.768256425857544,
      "step": 59845
    },
    {
      "epoch": 5.707263946533203e-05,
      "step": 59845,
      "training_step_time": 6.781745433807373
    },
    {
      "epoch": 5.7073593139648434e-05,
      "model_forward_time": 1.748335361480713,
      "step": 59846
    },
    {
      "epoch": 5.7073593139648434e-05,
      "step": 59846,
      "training_step_time": 6.773630142211914
    },
    {
      "epoch": 5.7074546813964845e-05,
      "model_forward_time": 1.7856225967407227,
      "step": 59847
    },
    {
      "epoch": 5.7074546813964845e-05,
      "step": 59847,
      "training_step_time": 7.04929780960083
    },
    {
      "epoch": 5.707550048828125e-05,
      "model_forward_time": 1.7598447799682617,
      "step": 59848
    },
    {
      "epoch": 5.707550048828125e-05,
      "step": 59848,
      "training_step_time": 6.7646849155426025
    },
    {
      "epoch": 5.707645416259765e-05,
      "model_forward_time": 1.7510802745819092,
      "step": 59849
    },
    {
      "epoch": 5.707645416259765e-05,
      "step": 59849,
      "training_step_time": 7.993591070175171
    },
    {
      "epoch": 5.7077407836914064e-05,
      "grad_norm": 0.10588511824607849,
      "learning_rate": 3.7967911651503787e-06,
      "loss": 0.0561,
      "step": 59850
    },
    {
      "epoch": 5.7077407836914064e-05,
      "model_forward_time": 1.7509777545928955,
      "step": 59850
    },
    {
      "dynamics_loss_avg": 0.06179498732089996,
      "epoch": 5.7077407836914064e-05,
      "step": 59850
    },
    {
      "action_loss_avg": 0.0017303439119132236,
      "epoch": 5.7077407836914064e-05,
      "step": 59850
    },
    {
      "epoch": 5.7077407836914064e-05,
      "step": 59850,
      "training_step_time": 8.409728527069092
    },
    {
      "epoch": 5.707836151123047e-05,
      "model_forward_time": 1.7694296836853027,
      "step": 59851
    },
    {
      "epoch": 5.707836151123047e-05,
      "step": 59851,
      "training_step_time": 8.319692134857178
    },
    {
      "epoch": 5.707931518554687e-05,
      "model_forward_time": 1.7485108375549316,
      "step": 59852
    },
    {
      "epoch": 5.707931518554687e-05,
      "step": 59852,
      "training_step_time": 7.678548336029053
    },
    {
      "epoch": 5.708026885986328e-05,
      "model_forward_time": 1.7534205913543701,
      "step": 59853
    },
    {
      "epoch": 5.708026885986328e-05,
      "step": 59853,
      "training_step_time": 6.718946695327759
    },
    {
      "epoch": 5.708122253417969e-05,
      "model_forward_time": 1.7605268955230713,
      "step": 59854
    },
    {
      "epoch": 5.708122253417969e-05,
      "step": 59854,
      "training_step_time": 6.709869384765625
    },
    {
      "epoch": 5.708217620849609e-05,
      "model_forward_time": 1.7546067237854004,
      "step": 59855
    },
    {
      "epoch": 5.708217620849609e-05,
      "step": 59855,
      "training_step_time": 6.724881649017334
    },
    {
      "epoch": 5.70831298828125e-05,
      "model_forward_time": 1.7612667083740234,
      "step": 59856
    },
    {
      "epoch": 5.70831298828125e-05,
      "step": 59856,
      "training_step_time": 6.74422550201416
    },
    {
      "epoch": 5.7084083557128906e-05,
      "model_forward_time": 1.4672596454620361,
      "step": 59857
    },
    {
      "epoch": 5.7084083557128906e-05,
      "step": 59857,
      "training_step_time": 7.02654767036438
    },
    {
      "epoch": 5.708503723144531e-05,
      "model_forward_time": 1.7502310276031494,
      "step": 59858
    },
    {
      "epoch": 5.708503723144531e-05,
      "step": 59858,
      "training_step_time": 9.011221885681152
    },
    {
      "epoch": 5.708599090576172e-05,
      "model_forward_time": 2.8678297996520996,
      "step": 59859
    },
    {
      "epoch": 5.708599090576172e-05,
      "step": 59859,
      "training_step_time": 10.891831874847412
    },
    {
      "epoch": 5.7086944580078124e-05,
      "grad_norm": 0.07875929027795792,
      "learning_rate": 3.7951863500140183e-06,
      "loss": 0.0545,
      "step": 59860
    },
    {
      "epoch": 5.7086944580078124e-05,
      "model_forward_time": 1.7557241916656494,
      "step": 59860
    },
    {
      "dynamics_loss_avg": 0.04985304772853851,
      "epoch": 5.7086944580078124e-05,
      "step": 59860
    },
    {
      "action_loss_avg": 0.0017022406071191653,
      "epoch": 5.7086944580078124e-05,
      "step": 59860
    },
    {
      "epoch": 5.7086944580078124e-05,
      "step": 59860,
      "training_step_time": 9.325794696807861
    },
    {
      "epoch": 5.708789825439453e-05,
      "model_forward_time": 1.7461779117584229,
      "step": 59861
    },
    {
      "epoch": 5.708789825439453e-05,
      "step": 59861,
      "training_step_time": 7.629223823547363
    },
    {
      "epoch": 5.708885192871094e-05,
      "model_forward_time": 1.7365319728851318,
      "step": 59862
    },
    {
      "epoch": 5.708885192871094e-05,
      "step": 59862,
      "training_step_time": 6.74228310585022
    },
    {
      "epoch": 5.708980560302734e-05,
      "model_forward_time": 1.3751156330108643,
      "step": 59863
    },
    {
      "epoch": 5.708980560302734e-05,
      "step": 59863,
      "training_step_time": 6.757028102874756
    },
    {
      "epoch": 5.709075927734375e-05,
      "model_forward_time": 1.7425472736358643,
      "step": 59864
    },
    {
      "epoch": 5.709075927734375e-05,
      "step": 59864,
      "training_step_time": 6.738554000854492
    },
    {
      "epoch": 5.709171295166016e-05,
      "model_forward_time": 1.751274585723877,
      "step": 59865
    },
    {
      "epoch": 5.709171295166016e-05,
      "step": 59865,
      "training_step_time": 6.74640679359436
    },
    {
      "epoch": 5.709266662597656e-05,
      "model_forward_time": 1.8098537921905518,
      "step": 59866
    },
    {
      "epoch": 5.709266662597656e-05,
      "step": 59866,
      "training_step_time": 6.987715721130371
    },
    {
      "epoch": 5.7093620300292966e-05,
      "model_forward_time": 1.7430634498596191,
      "step": 59867
    },
    {
      "epoch": 5.7093620300292966e-05,
      "step": 59867,
      "training_step_time": 8.141010046005249
    },
    {
      "epoch": 5.709457397460938e-05,
      "model_forward_time": 2.8376352787017822,
      "step": 59868
    },
    {
      "epoch": 5.709457397460938e-05,
      "step": 59868,
      "training_step_time": 9.384380340576172
    },
    {
      "epoch": 5.709552764892578e-05,
      "model_forward_time": 1.7456445693969727,
      "step": 59869
    },
    {
      "epoch": 5.709552764892578e-05,
      "step": 59869,
      "training_step_time": 11.307590961456299
    },
    {
      "epoch": 5.7096481323242185e-05,
      "grad_norm": 0.1437976211309433,
      "learning_rate": 3.793581666634266e-06,
      "loss": 0.0589,
      "step": 59870
    },
    {
      "epoch": 5.7096481323242185e-05,
      "model_forward_time": 1.746708631515503,
      "step": 59870
    },
    {
      "dynamics_loss_avg": 0.04146191719919443,
      "epoch": 5.7096481323242185e-05,
      "step": 59870
    },
    {
      "action_loss_avg": 0.0010649358388036489,
      "epoch": 5.7096481323242185e-05,
      "step": 59870
    },
    {
      "epoch": 5.7096481323242185e-05,
      "step": 59870,
      "training_step_time": 7.661045789718628
    },
    {
      "epoch": 5.7097434997558596e-05,
      "model_forward_time": 1.7123055458068848,
      "step": 59871
    },
    {
      "epoch": 5.7097434997558596e-05,
      "step": 59871,
      "training_step_time": 7.582545518875122
    },
    {
      "epoch": 5.7098388671875e-05,
      "model_forward_time": 1.7020187377929688,
      "step": 59872
    },
    {
      "epoch": 5.7098388671875e-05,
      "step": 59872,
      "training_step_time": 6.75698709487915
    },
    {
      "epoch": 5.7099342346191404e-05,
      "model_forward_time": 1.688429355621338,
      "step": 59873
    },
    {
      "epoch": 5.7099342346191404e-05,
      "step": 59873,
      "training_step_time": 6.791626214981079
    },
    {
      "epoch": 5.7100296020507815e-05,
      "model_forward_time": 1.6996338367462158,
      "step": 59874
    },
    {
      "epoch": 5.7100296020507815e-05,
      "step": 59874,
      "training_step_time": 6.733930826187134
    },
    {
      "epoch": 5.710124969482422e-05,
      "model_forward_time": 1.6962969303131104,
      "step": 59875
    },
    {
      "epoch": 5.710124969482422e-05,
      "step": 59875,
      "training_step_time": 7.226010084152222
    },
    {
      "epoch": 5.710220336914062e-05,
      "model_forward_time": 1.7027552127838135,
      "step": 59876
    },
    {
      "epoch": 5.710220336914062e-05,
      "step": 59876,
      "training_step_time": 7.540853977203369
    },
    {
      "epoch": 5.7103157043457034e-05,
      "model_forward_time": 2.2479212284088135,
      "step": 59877
    },
    {
      "epoch": 5.7103157043457034e-05,
      "step": 59877,
      "training_step_time": 8.639739036560059
    },
    {
      "epoch": 5.710411071777344e-05,
      "model_forward_time": 1.6969084739685059,
      "step": 59878
    },
    {
      "epoch": 5.710411071777344e-05,
      "step": 59878,
      "training_step_time": 9.160027980804443
    },
    {
      "epoch": 5.710506439208984e-05,
      "model_forward_time": 1.6930499076843262,
      "step": 59879
    },
    {
      "epoch": 5.710506439208984e-05,
      "step": 59879,
      "training_step_time": 7.631590127944946
    },
    {
      "epoch": 5.710601806640625e-05,
      "grad_norm": 0.1350472867488861,
      "learning_rate": 3.7919771151866094e-06,
      "loss": 0.0572,
      "step": 59880
    },
    {
      "epoch": 5.710601806640625e-05,
      "model_forward_time": 1.695814847946167,
      "step": 59880
    },
    {
      "dynamics_loss_avg": 0.08060790896415711,
      "epoch": 5.710601806640625e-05,
      "step": 59880
    },
    {
      "action_loss_avg": 0.0018470711132977159,
      "epoch": 5.710601806640625e-05,
      "step": 59880
    },
    {
      "epoch": 5.710601806640625e-05,
      "step": 59880,
      "training_step_time": 7.552478313446045
    },
    {
      "epoch": 5.7106971740722656e-05,
      "model_forward_time": 1.7023332118988037,
      "step": 59881
    },
    {
      "epoch": 5.7106971740722656e-05,
      "step": 59881,
      "training_step_time": 6.73956561088562
    },
    {
      "epoch": 5.710792541503906e-05,
      "model_forward_time": 1.6996769905090332,
      "step": 59882
    },
    {
      "epoch": 5.710792541503906e-05,
      "step": 59882,
      "training_step_time": 6.724476337432861
    },
    {
      "epoch": 5.710887908935547e-05,
      "model_forward_time": 1.7105073928833008,
      "step": 59883
    },
    {
      "epoch": 5.710887908935547e-05,
      "step": 59883,
      "training_step_time": 6.77048134803772
    },
    {
      "epoch": 5.7109832763671875e-05,
      "model_forward_time": 1.7122480869293213,
      "step": 59884
    },
    {
      "epoch": 5.7109832763671875e-05,
      "step": 59884,
      "training_step_time": 7.065643787384033
    },
    {
      "epoch": 5.711078643798828e-05,
      "model_forward_time": 1.710881233215332,
      "step": 59885
    },
    {
      "epoch": 5.711078643798828e-05,
      "step": 59885,
      "training_step_time": 8.361979484558105
    },
    {
      "epoch": 5.711174011230469e-05,
      "model_forward_time": 1.7607874870300293,
      "step": 59886
    },
    {
      "epoch": 5.711174011230469e-05,
      "step": 59886,
      "training_step_time": 8.616869926452637
    },
    {
      "epoch": 5.7112693786621094e-05,
      "model_forward_time": 1.738945722579956,
      "step": 59887
    },
    {
      "epoch": 5.7112693786621094e-05,
      "step": 59887,
      "training_step_time": 9.308046102523804
    },
    {
      "epoch": 5.71136474609375e-05,
      "model_forward_time": 1.73301362991333,
      "step": 59888
    },
    {
      "epoch": 5.71136474609375e-05,
      "step": 59888,
      "training_step_time": 7.644444942474365
    },
    {
      "epoch": 5.711460113525391e-05,
      "model_forward_time": 1.7049314975738525,
      "step": 59889
    },
    {
      "epoch": 5.711460113525391e-05,
      "step": 59889,
      "training_step_time": 7.69293999671936
    },
    {
      "epoch": 5.711555480957031e-05,
      "grad_norm": 0.11898887157440186,
      "learning_rate": 3.7903726958465204e-06,
      "loss": 0.0566,
      "step": 59890
    },
    {
      "epoch": 5.711555480957031e-05,
      "model_forward_time": 1.7077960968017578,
      "step": 59890
    },
    {
      "dynamics_loss_avg": 0.052644484862685206,
      "epoch": 5.711555480957031e-05,
      "step": 59890
    },
    {
      "action_loss_avg": 0.00240893202717416,
      "epoch": 5.711555480957031e-05,
      "step": 59890
    },
    {
      "epoch": 5.711555480957031e-05,
      "step": 59890,
      "training_step_time": 6.7676215171813965
    },
    {
      "epoch": 5.711650848388672e-05,
      "model_forward_time": 1.7137551307678223,
      "step": 59891
    },
    {
      "epoch": 5.711650848388672e-05,
      "step": 59891,
      "training_step_time": 6.769731521606445
    },
    {
      "epoch": 5.711746215820313e-05,
      "model_forward_time": 1.713458776473999,
      "step": 59892
    },
    {
      "epoch": 5.711746215820313e-05,
      "step": 59892,
      "training_step_time": 6.771356821060181
    },
    {
      "epoch": 5.711841583251953e-05,
      "model_forward_time": 1.6981544494628906,
      "step": 59893
    },
    {
      "epoch": 5.711841583251953e-05,
      "step": 59893,
      "training_step_time": 6.976137161254883
    },
    {
      "epoch": 5.7119369506835936e-05,
      "model_forward_time": 1.7089893817901611,
      "step": 59894
    },
    {
      "epoch": 5.7119369506835936e-05,
      "step": 59894,
      "training_step_time": 7.181650400161743
    },
    {
      "epoch": 5.712032318115235e-05,
      "model_forward_time": 1.7307090759277344,
      "step": 59895
    },
    {
      "epoch": 5.712032318115235e-05,
      "step": 59895,
      "training_step_time": 8.885100364685059
    },
    {
      "epoch": 5.712127685546875e-05,
      "model_forward_time": 2.2519431114196777,
      "step": 59896
    },
    {
      "epoch": 5.712127685546875e-05,
      "step": 59896,
      "training_step_time": 8.74221420288086
    },
    {
      "epoch": 5.7122230529785155e-05,
      "model_forward_time": 1.702449083328247,
      "step": 59897
    },
    {
      "epoch": 5.7122230529785155e-05,
      "step": 59897,
      "training_step_time": 8.595088005065918
    },
    {
      "epoch": 5.7123184204101566e-05,
      "model_forward_time": 1.7391250133514404,
      "step": 59898
    },
    {
      "epoch": 5.7123184204101566e-05,
      "step": 59898,
      "training_step_time": 7.668845176696777
    },
    {
      "epoch": 5.712413787841797e-05,
      "model_forward_time": 1.730468511581421,
      "step": 59899
    },
    {
      "epoch": 5.712413787841797e-05,
      "step": 59899,
      "training_step_time": 6.783389091491699
    },
    {
      "epoch": 5.7125091552734374e-05,
      "grad_norm": 0.09819485247135162,
      "learning_rate": 3.7887684087894544e-06,
      "loss": 0.0571,
      "step": 59900
    },
    {
      "epoch": 5.7125091552734374e-05,
      "model_forward_time": 1.709632396697998,
      "step": 59900
    },
    {
      "dynamics_loss_avg": 0.06557619217783213,
      "epoch": 5.7125091552734374e-05,
      "step": 59900
    },
    {
      "action_loss_avg": 0.0024049426545389,
      "epoch": 5.7125091552734374e-05,
      "step": 59900
    },
    {
      "epoch": 5.7125091552734374e-05,
      "step": 59900,
      "training_step_time": 6.7719197273254395
    },
    {
      "epoch": 5.7126045227050784e-05,
      "model_forward_time": 1.7110645771026611,
      "step": 59901
    },
    {
      "epoch": 5.7126045227050784e-05,
      "step": 59901,
      "training_step_time": 6.7567853927612305
    },
    {
      "epoch": 5.712699890136719e-05,
      "model_forward_time": 1.7169206142425537,
      "step": 59902
    },
    {
      "epoch": 5.712699890136719e-05,
      "step": 59902,
      "training_step_time": 7.019233465194702
    },
    {
      "epoch": 5.712795257568359e-05,
      "model_forward_time": 1.7066383361816406,
      "step": 59903
    },
    {
      "epoch": 5.712795257568359e-05,
      "step": 59903,
      "training_step_time": 7.601860523223877
    },
    {
      "epoch": 5.712890625e-05,
      "model_forward_time": 1.704890251159668,
      "step": 59904
    },
    {
      "epoch": 5.712890625e-05,
      "step": 59904,
      "training_step_time": 7.5797131061553955
    },
    {
      "epoch": 5.712985992431641e-05,
      "model_forward_time": 2.280001163482666,
      "step": 59905
    },
    {
      "epoch": 5.712985992431641e-05,
      "step": 59905,
      "training_step_time": 8.661392211914062
    },
    {
      "epoch": 5.713081359863281e-05,
      "model_forward_time": 1.7145345211029053,
      "step": 59906
    },
    {
      "epoch": 5.713081359863281e-05,
      "step": 59906,
      "training_step_time": 9.549330472946167
    },
    {
      "epoch": 5.7131767272949215e-05,
      "model_forward_time": 1.7175400257110596,
      "step": 59907
    },
    {
      "epoch": 5.7131767272949215e-05,
      "step": 59907,
      "training_step_time": 7.639392137527466
    },
    {
      "epoch": 5.7132720947265626e-05,
      "model_forward_time": 1.7114741802215576,
      "step": 59908
    },
    {
      "epoch": 5.7132720947265626e-05,
      "step": 59908,
      "training_step_time": 6.707826375961304
    },
    {
      "epoch": 5.713367462158203e-05,
      "model_forward_time": 1.7091357707977295,
      "step": 59909
    },
    {
      "epoch": 5.713367462158203e-05,
      "step": 59909,
      "training_step_time": 6.709908485412598
    },
    {
      "epoch": 5.7134628295898434e-05,
      "grad_norm": 0.11787562817335129,
      "learning_rate": 3.7871642541908544e-06,
      "loss": 0.0573,
      "step": 59910
    },
    {
      "epoch": 5.7134628295898434e-05,
      "model_forward_time": 1.7072808742523193,
      "step": 59910
    },
    {
      "dynamics_loss_avg": 0.054768587648868564,
      "epoch": 5.7134628295898434e-05,
      "step": 59910
    },
    {
      "action_loss_avg": 0.0014291129889898002,
      "epoch": 5.7134628295898434e-05,
      "step": 59910
    },
    {
      "epoch": 5.7134628295898434e-05,
      "step": 59910,
      "training_step_time": 6.740458250045776
    },
    {
      "epoch": 5.7135581970214845e-05,
      "model_forward_time": 1.7971854209899902,
      "step": 59911
    },
    {
      "epoch": 5.7135581970214845e-05,
      "step": 59911,
      "training_step_time": 6.719451904296875
    },
    {
      "epoch": 5.713653564453125e-05,
      "model_forward_time": 1.7259972095489502,
      "step": 59912
    },
    {
      "epoch": 5.713653564453125e-05,
      "step": 59912,
      "training_step_time": 7.0044660568237305
    },
    {
      "epoch": 5.713748931884765e-05,
      "model_forward_time": 1.712676763534546,
      "step": 59913
    },
    {
      "epoch": 5.713748931884765e-05,
      "step": 59913,
      "training_step_time": 8.374842643737793
    },
    {
      "epoch": 5.7138442993164064e-05,
      "model_forward_time": 2.33254337310791,
      "step": 59914
    },
    {
      "epoch": 5.7138442993164064e-05,
      "step": 59914,
      "training_step_time": 8.801624059677124
    },
    {
      "epoch": 5.713939666748047e-05,
      "model_forward_time": 1.7131881713867188,
      "step": 59915
    },
    {
      "epoch": 5.713939666748047e-05,
      "step": 59915,
      "training_step_time": 8.857465028762817
    },
    {
      "epoch": 5.714035034179687e-05,
      "model_forward_time": 1.714012861251831,
      "step": 59916
    },
    {
      "epoch": 5.714035034179687e-05,
      "step": 59916,
      "training_step_time": 7.64423394203186
    },
    {
      "epoch": 5.714130401611328e-05,
      "model_forward_time": 1.7376670837402344,
      "step": 59917
    },
    {
      "epoch": 5.714130401611328e-05,
      "step": 59917,
      "training_step_time": 7.6989336013793945
    },
    {
      "epoch": 5.714225769042969e-05,
      "model_forward_time": 1.707942247390747,
      "step": 59918
    },
    {
      "epoch": 5.714225769042969e-05,
      "step": 59918,
      "training_step_time": 6.705189943313599
    },
    {
      "epoch": 5.714321136474609e-05,
      "model_forward_time": 1.7107563018798828,
      "step": 59919
    },
    {
      "epoch": 5.714321136474609e-05,
      "step": 59919,
      "training_step_time": 6.715121746063232
    },
    {
      "epoch": 5.71441650390625e-05,
      "grad_norm": 0.13614308834075928,
      "learning_rate": 3.7855602322261476e-06,
      "loss": 0.0569,
      "step": 59920
    },
    {
      "epoch": 5.71441650390625e-05,
      "model_forward_time": 1.7300326824188232,
      "step": 59920
    },
    {
      "dynamics_loss_avg": 0.05664515625685453,
      "epoch": 5.71441650390625e-05,
      "step": 59920
    },
    {
      "action_loss_avg": 0.0013841467327438295,
      "epoch": 5.71441650390625e-05,
      "step": 59920
    },
    {
      "epoch": 5.71441650390625e-05,
      "step": 59920,
      "training_step_time": 6.7051475048065186
    },
    {
      "epoch": 5.7145118713378906e-05,
      "model_forward_time": 1.7173364162445068,
      "step": 59921
    },
    {
      "epoch": 5.7145118713378906e-05,
      "step": 59921,
      "training_step_time": 7.025181531906128
    },
    {
      "epoch": 5.714607238769531e-05,
      "model_forward_time": 1.7104394435882568,
      "step": 59922
    },
    {
      "epoch": 5.714607238769531e-05,
      "step": 59922,
      "training_step_time": 7.182554721832275
    },
    {
      "epoch": 5.714702606201172e-05,
      "model_forward_time": 2.265671730041504,
      "step": 59923
    },
    {
      "epoch": 5.714702606201172e-05,
      "step": 59923,
      "training_step_time": 9.243672847747803
    },
    {
      "epoch": 5.7147979736328125e-05,
      "model_forward_time": 1.7729666233062744,
      "step": 59924
    },
    {
      "epoch": 5.7147979736328125e-05,
      "step": 59924,
      "training_step_time": 10.065136909484863
    },
    {
      "epoch": 5.714893341064453e-05,
      "model_forward_time": 1.7098579406738281,
      "step": 59925
    },
    {
      "epoch": 5.714893341064453e-05,
      "step": 59925,
      "training_step_time": 8.782837390899658
    },
    {
      "epoch": 5.714988708496094e-05,
      "model_forward_time": 1.7080154418945312,
      "step": 59926
    },
    {
      "epoch": 5.714988708496094e-05,
      "step": 59926,
      "training_step_time": 7.742476224899292
    },
    {
      "epoch": 5.715084075927734e-05,
      "model_forward_time": 1.7121989727020264,
      "step": 59927
    },
    {
      "epoch": 5.715084075927734e-05,
      "step": 59927,
      "training_step_time": 7.512293577194214
    },
    {
      "epoch": 5.715179443359375e-05,
      "model_forward_time": 1.7157700061798096,
      "step": 59928
    },
    {
      "epoch": 5.715179443359375e-05,
      "step": 59928,
      "training_step_time": 6.910375356674194
    },
    {
      "epoch": 5.715274810791016e-05,
      "model_forward_time": 1.7135400772094727,
      "step": 59929
    },
    {
      "epoch": 5.715274810791016e-05,
      "step": 59929,
      "training_step_time": 6.794689178466797
    },
    {
      "epoch": 5.715370178222656e-05,
      "grad_norm": 0.13757362961769104,
      "learning_rate": 3.783956343070746e-06,
      "loss": 0.0579,
      "step": 59930
    },
    {
      "epoch": 5.715370178222656e-05,
      "model_forward_time": 1.7276365756988525,
      "step": 59930
    },
    {
      "dynamics_loss_avg": 0.057538088597357276,
      "epoch": 5.715370178222656e-05,
      "step": 59930
    },
    {
      "action_loss_avg": 0.0012082992325304076,
      "epoch": 5.715370178222656e-05,
      "step": 59930
    },
    {
      "epoch": 5.715370178222656e-05,
      "step": 59930,
      "training_step_time": 7.113749027252197
    },
    {
      "epoch": 5.7154655456542966e-05,
      "model_forward_time": 1.7048850059509277,
      "step": 59931
    },
    {
      "epoch": 5.7154655456542966e-05,
      "step": 59931,
      "training_step_time": 7.359645128250122
    },
    {
      "epoch": 5.715560913085938e-05,
      "model_forward_time": 2.261409282684326,
      "step": 59932
    },
    {
      "epoch": 5.715560913085938e-05,
      "step": 59932,
      "training_step_time": 9.311991214752197
    },
    {
      "epoch": 5.715656280517578e-05,
      "model_forward_time": 1.7762022018432617,
      "step": 59933
    },
    {
      "epoch": 5.715656280517578e-05,
      "step": 59933,
      "training_step_time": 10.47953987121582
    },
    {
      "epoch": 5.7157516479492185e-05,
      "model_forward_time": 1.7088775634765625,
      "step": 59934
    },
    {
      "epoch": 5.7157516479492185e-05,
      "step": 59934,
      "training_step_time": 8.449082851409912
    },
    {
      "epoch": 5.7158470153808596e-05,
      "model_forward_time": 1.7010583877563477,
      "step": 59935
    },
    {
      "epoch": 5.7158470153808596e-05,
      "step": 59935,
      "training_step_time": 8.778349876403809
    },
    {
      "epoch": 5.7159423828125e-05,
      "model_forward_time": 1.7102293968200684,
      "step": 59936
    },
    {
      "epoch": 5.7159423828125e-05,
      "step": 59936,
      "training_step_time": 7.475410223007202
    },
    {
      "epoch": 5.7160377502441404e-05,
      "model_forward_time": 1.708890438079834,
      "step": 59937
    },
    {
      "epoch": 5.7160377502441404e-05,
      "step": 59937,
      "training_step_time": 6.765665769577026
    },
    {
      "epoch": 5.7161331176757815e-05,
      "model_forward_time": 1.713301658630371,
      "step": 59938
    },
    {
      "epoch": 5.7161331176757815e-05,
      "step": 59938,
      "training_step_time": 6.784851551055908
    },
    {
      "epoch": 5.716228485107422e-05,
      "model_forward_time": 1.7307171821594238,
      "step": 59939
    },
    {
      "epoch": 5.716228485107422e-05,
      "step": 59939,
      "training_step_time": 7.032770156860352
    },
    {
      "epoch": 5.716323852539062e-05,
      "grad_norm": 0.11167861521244049,
      "learning_rate": 3.7823525869000527e-06,
      "loss": 0.0584,
      "step": 59940
    },
    {
      "epoch": 5.716323852539062e-05,
      "model_forward_time": 1.7058823108673096,
      "step": 59940
    },
    {
      "dynamics_loss_avg": 0.06759864762425423,
      "epoch": 5.716323852539062e-05,
      "step": 59940
    },
    {
      "action_loss_avg": 0.001929472925257869,
      "epoch": 5.716323852539062e-05,
      "step": 59940
    },
    {
      "epoch": 5.716323852539062e-05,
      "step": 59940,
      "training_step_time": 7.98108172416687
    },
    {
      "epoch": 5.7164192199707034e-05,
      "model_forward_time": 1.7045416831970215,
      "step": 59941
    },
    {
      "epoch": 5.7164192199707034e-05,
      "step": 59941,
      "training_step_time": 8.695642709732056
    },
    {
      "epoch": 5.716514587402344e-05,
      "model_forward_time": 1.7308130264282227,
      "step": 59942
    },
    {
      "epoch": 5.716514587402344e-05,
      "step": 59942,
      "training_step_time": 9.192756414413452
    },
    {
      "epoch": 5.716609954833984e-05,
      "model_forward_time": 1.7061939239501953,
      "step": 59943
    },
    {
      "epoch": 5.716609954833984e-05,
      "step": 59943,
      "training_step_time": 7.7909955978393555
    },
    {
      "epoch": 5.716705322265625e-05,
      "model_forward_time": 1.7122657299041748,
      "step": 59944
    },
    {
      "epoch": 5.716705322265625e-05,
      "step": 59944,
      "training_step_time": 8.470914602279663
    },
    {
      "epoch": 5.7168006896972657e-05,
      "model_forward_time": 1.7034876346588135,
      "step": 59945
    },
    {
      "epoch": 5.7168006896972657e-05,
      "step": 59945,
      "training_step_time": 7.668900966644287
    },
    {
      "epoch": 5.716896057128906e-05,
      "model_forward_time": 1.720870018005371,
      "step": 59946
    },
    {
      "epoch": 5.716896057128906e-05,
      "step": 59946,
      "training_step_time": 6.784040927886963
    },
    {
      "epoch": 5.716991424560547e-05,
      "model_forward_time": 1.713031530380249,
      "step": 59947
    },
    {
      "epoch": 5.716991424560547e-05,
      "step": 59947,
      "training_step_time": 6.672223329544067
    },
    {
      "epoch": 5.7170867919921875e-05,
      "model_forward_time": 1.7028095722198486,
      "step": 59948
    },
    {
      "epoch": 5.7170867919921875e-05,
      "step": 59948,
      "training_step_time": 6.7859392166137695
    },
    {
      "epoch": 5.717182159423828e-05,
      "model_forward_time": 1.7029948234558105,
      "step": 59949
    },
    {
      "epoch": 5.717182159423828e-05,
      "step": 59949,
      "training_step_time": 7.250331163406372
    },
    {
      "epoch": 5.717277526855469e-05,
      "grad_norm": 0.08519944548606873,
      "learning_rate": 3.78074896388945e-06,
      "loss": 0.0571,
      "step": 59950
    },
    {
      "epoch": 5.717277526855469e-05,
      "model_forward_time": 1.7029170989990234,
      "step": 59950
    },
    {
      "dynamics_loss_avg": 0.048424556665122506,
      "epoch": 5.717277526855469e-05,
      "step": 59950
    },
    {
      "action_loss_avg": 0.0012952180113643408,
      "epoch": 5.717277526855469e-05,
      "step": 59950
    },
    {
      "epoch": 5.717277526855469e-05,
      "step": 59950,
      "training_step_time": 8.443701028823853
    },
    {
      "epoch": 5.7173728942871094e-05,
      "model_forward_time": 2.2732107639312744,
      "step": 59951
    },
    {
      "epoch": 5.7173728942871094e-05,
      "step": 59951,
      "training_step_time": 7.717854976654053
    },
    {
      "epoch": 5.71746826171875e-05,
      "model_forward_time": 1.702739953994751,
      "step": 59952
    },
    {
      "epoch": 5.71746826171875e-05,
      "step": 59952,
      "training_step_time": 9.212166547775269
    },
    {
      "epoch": 5.717563629150391e-05,
      "model_forward_time": 1.7023236751556396,
      "step": 59953
    },
    {
      "epoch": 5.717563629150391e-05,
      "step": 59953,
      "training_step_time": 7.7624125480651855
    },
    {
      "epoch": 5.717658996582031e-05,
      "model_forward_time": 1.704314947128296,
      "step": 59954
    },
    {
      "epoch": 5.717658996582031e-05,
      "step": 59954,
      "training_step_time": 7.696920871734619
    },
    {
      "epoch": 5.717754364013672e-05,
      "model_forward_time": 1.7282147407531738,
      "step": 59955
    },
    {
      "epoch": 5.717754364013672e-05,
      "step": 59955,
      "training_step_time": 6.717358112335205
    },
    {
      "epoch": 5.717849731445313e-05,
      "model_forward_time": 1.7063689231872559,
      "step": 59956
    },
    {
      "epoch": 5.717849731445313e-05,
      "step": 59956,
      "training_step_time": 6.662262201309204
    },
    {
      "epoch": 5.717945098876953e-05,
      "model_forward_time": 1.6988632678985596,
      "step": 59957
    },
    {
      "epoch": 5.717945098876953e-05,
      "step": 59957,
      "training_step_time": 6.718607187271118
    },
    {
      "epoch": 5.7180404663085936e-05,
      "model_forward_time": 1.7123947143554688,
      "step": 59958
    },
    {
      "epoch": 5.7180404663085936e-05,
      "step": 59958,
      "training_step_time": 7.148757219314575
    },
    {
      "epoch": 5.718135833740235e-05,
      "model_forward_time": 1.715221643447876,
      "step": 59959
    },
    {
      "epoch": 5.718135833740235e-05,
      "step": 59959,
      "training_step_time": 7.597504615783691
    },
    {
      "epoch": 5.718231201171875e-05,
      "grad_norm": 0.127386674284935,
      "learning_rate": 3.7791454742143073e-06,
      "loss": 0.0544,
      "step": 59960
    },
    {
      "epoch": 5.718231201171875e-05,
      "model_forward_time": 2.2657454013824463,
      "step": 59960
    },
    {
      "dynamics_loss_avg": 0.05115693565458059,
      "epoch": 5.718231201171875e-05,
      "step": 59960
    },
    {
      "action_loss_avg": 0.001436326265684329,
      "epoch": 5.718231201171875e-05,
      "step": 59960
    },
    {
      "epoch": 5.718231201171875e-05,
      "step": 59960,
      "training_step_time": 10.131145000457764
    },
    {
      "epoch": 5.7183265686035155e-05,
      "model_forward_time": 1.7735342979431152,
      "step": 59961
    },
    {
      "epoch": 5.7183265686035155e-05,
      "step": 59961,
      "training_step_time": 9.444775342941284
    },
    {
      "epoch": 5.7184219360351566e-05,
      "model_forward_time": 1.712085485458374,
      "step": 59962
    },
    {
      "epoch": 5.7184219360351566e-05,
      "step": 59962,
      "training_step_time": 7.7545552253723145
    },
    {
      "epoch": 5.718517303466797e-05,
      "model_forward_time": 1.7093183994293213,
      "step": 59963
    },
    {
      "epoch": 5.718517303466797e-05,
      "step": 59963,
      "training_step_time": 7.57288122177124
    },
    {
      "epoch": 5.7186126708984374e-05,
      "model_forward_time": 1.7356300354003906,
      "step": 59964
    },
    {
      "epoch": 5.7186126708984374e-05,
      "step": 59964,
      "training_step_time": 6.69675087928772
    },
    {
      "epoch": 5.7187080383300785e-05,
      "model_forward_time": 1.716400146484375,
      "step": 59965
    },
    {
      "epoch": 5.7187080383300785e-05,
      "step": 59965,
      "training_step_time": 6.657540559768677
    },
    {
      "epoch": 5.718803405761719e-05,
      "model_forward_time": 1.7083816528320312,
      "step": 59966
    },
    {
      "epoch": 5.718803405761719e-05,
      "step": 59966,
      "training_step_time": 6.675658941268921
    },
    {
      "epoch": 5.718898773193359e-05,
      "model_forward_time": 1.697110652923584,
      "step": 59967
    },
    {
      "epoch": 5.718898773193359e-05,
      "step": 59967,
      "training_step_time": 6.944691181182861
    },
    {
      "epoch": 5.718994140625e-05,
      "model_forward_time": 1.7041141986846924,
      "step": 59968
    },
    {
      "epoch": 5.718994140625e-05,
      "step": 59968,
      "training_step_time": 7.6413397789001465
    },
    {
      "epoch": 5.719089508056641e-05,
      "model_forward_time": 2.2541139125823975,
      "step": 59969
    },
    {
      "epoch": 5.719089508056641e-05,
      "step": 59969,
      "training_step_time": 8.631303787231445
    },
    {
      "epoch": 5.719184875488281e-05,
      "grad_norm": 0.11620039492845535,
      "learning_rate": 3.77754211804998e-06,
      "loss": 0.0553,
      "step": 59970
    },
    {
      "epoch": 5.719184875488281e-05,
      "model_forward_time": 1.7344012260437012,
      "step": 59970
    },
    {
      "dynamics_loss_avg": 0.05743435863405466,
      "epoch": 5.719184875488281e-05,
      "step": 59970
    },
    {
      "action_loss_avg": 0.0020657899003708735,
      "epoch": 5.719184875488281e-05,
      "step": 59970
    },
    {
      "epoch": 5.719184875488281e-05,
      "step": 59970,
      "training_step_time": 8.82167935371399
    },
    {
      "epoch": 5.7192802429199215e-05,
      "model_forward_time": 1.7103087902069092,
      "step": 59971
    },
    {
      "epoch": 5.7192802429199215e-05,
      "step": 59971,
      "training_step_time": 9.440926313400269
    },
    {
      "epoch": 5.7193756103515626e-05,
      "model_forward_time": 1.7160511016845703,
      "step": 59972
    },
    {
      "epoch": 5.7193756103515626e-05,
      "step": 59972,
      "training_step_time": 7.624346733093262
    },
    {
      "epoch": 5.719470977783203e-05,
      "model_forward_time": 1.7148127555847168,
      "step": 59973
    },
    {
      "epoch": 5.719470977783203e-05,
      "step": 59973,
      "training_step_time": 6.684460878372192
    },
    {
      "epoch": 5.7195663452148434e-05,
      "model_forward_time": 1.7129530906677246,
      "step": 59974
    },
    {
      "epoch": 5.7195663452148434e-05,
      "step": 59974,
      "training_step_time": 6.729688882827759
    },
    {
      "epoch": 5.7196617126464845e-05,
      "model_forward_time": 1.7203550338745117,
      "step": 59975
    },
    {
      "epoch": 5.7196617126464845e-05,
      "step": 59975,
      "training_step_time": 6.812493562698364
    },
    {
      "epoch": 5.719757080078125e-05,
      "model_forward_time": 1.715118646621704,
      "step": 59976
    },
    {
      "epoch": 5.719757080078125e-05,
      "step": 59976,
      "training_step_time": 7.03173565864563
    },
    {
      "epoch": 5.719852447509765e-05,
      "model_forward_time": 1.715437650680542,
      "step": 59977
    },
    {
      "epoch": 5.719852447509765e-05,
      "step": 59977,
      "training_step_time": 8.360697031021118
    },
    {
      "epoch": 5.7199478149414064e-05,
      "model_forward_time": 2.27549409866333,
      "step": 59978
    },
    {
      "epoch": 5.7199478149414064e-05,
      "step": 59978,
      "training_step_time": 9.389809370040894
    },
    {
      "epoch": 5.720043182373047e-05,
      "model_forward_time": 1.7184829711914062,
      "step": 59979
    },
    {
      "epoch": 5.720043182373047e-05,
      "step": 59979,
      "training_step_time": 9.186901569366455
    },
    {
      "epoch": 5.720138549804687e-05,
      "grad_norm": 0.10448842495679855,
      "learning_rate": 3.7759388955718085e-06,
      "loss": 0.0538,
      "step": 59980
    },
    {
      "epoch": 5.720138549804687e-05,
      "model_forward_time": 1.7172324657440186,
      "step": 59980
    },
    {
      "dynamics_loss_avg": 0.06805899366736412,
      "epoch": 5.720138549804687e-05,
      "step": 59980
    },
    {
      "action_loss_avg": 0.0016591626190347597,
      "epoch": 5.720138549804687e-05,
      "step": 59980
    },
    {
      "epoch": 5.720138549804687e-05,
      "step": 59980,
      "training_step_time": 7.809368848800659
    },
    {
      "epoch": 5.720233917236328e-05,
      "model_forward_time": 1.7028882503509521,
      "step": 59981
    },
    {
      "epoch": 5.720233917236328e-05,
      "step": 59981,
      "training_step_time": 7.731010437011719
    },
    {
      "epoch": 5.720329284667969e-05,
      "model_forward_time": 1.7049555778503418,
      "step": 59982
    },
    {
      "epoch": 5.720329284667969e-05,
      "step": 59982,
      "training_step_time": 7.094039440155029
    },
    {
      "epoch": 5.720424652099609e-05,
      "model_forward_time": 1.705019474029541,
      "step": 59983
    },
    {
      "epoch": 5.720424652099609e-05,
      "step": 59983,
      "training_step_time": 6.752772092819214
    },
    {
      "epoch": 5.72052001953125e-05,
      "model_forward_time": 1.708312749862671,
      "step": 59984
    },
    {
      "epoch": 5.72052001953125e-05,
      "step": 59984,
      "training_step_time": 6.707695722579956
    },
    {
      "epoch": 5.7206153869628906e-05,
      "model_forward_time": 1.7003157138824463,
      "step": 59985
    },
    {
      "epoch": 5.7206153869628906e-05,
      "step": 59985,
      "training_step_time": 7.0069191455841064
    },
    {
      "epoch": 5.720710754394531e-05,
      "model_forward_time": 1.6998612880706787,
      "step": 59986
    },
    {
      "epoch": 5.720710754394531e-05,
      "step": 59986,
      "training_step_time": 7.456171989440918
    },
    {
      "epoch": 5.720806121826172e-05,
      "model_forward_time": 2.2332231998443604,
      "step": 59987
    },
    {
      "epoch": 5.720806121826172e-05,
      "step": 59987,
      "training_step_time": 9.325116634368896
    },
    {
      "epoch": 5.7209014892578125e-05,
      "model_forward_time": 1.7025477886199951,
      "step": 59988
    },
    {
      "epoch": 5.7209014892578125e-05,
      "step": 59988,
      "training_step_time": 8.410168647766113
    },
    {
      "epoch": 5.720996856689453e-05,
      "model_forward_time": 1.7628545761108398,
      "step": 59989
    },
    {
      "epoch": 5.720996856689453e-05,
      "step": 59989,
      "training_step_time": 8.641597270965576
    },
    {
      "epoch": 5.721092224121094e-05,
      "grad_norm": 0.07026488333940506,
      "learning_rate": 3.7743358069551174e-06,
      "loss": 0.0529,
      "step": 59990
    },
    {
      "epoch": 5.721092224121094e-05,
      "model_forward_time": 1.707319736480713,
      "step": 59990
    },
    {
      "dynamics_loss_avg": 0.04369937460869551,
      "epoch": 5.721092224121094e-05,
      "step": 59990
    },
    {
      "action_loss_avg": 0.0016018175228964537,
      "epoch": 5.721092224121094e-05,
      "step": 59990
    },
    {
      "epoch": 5.721092224121094e-05,
      "step": 59990,
      "training_step_time": 7.816237688064575
    },
    {
      "epoch": 5.7211875915527343e-05,
      "model_forward_time": 1.701319694519043,
      "step": 59991
    },
    {
      "epoch": 5.7211875915527343e-05,
      "step": 59991,
      "training_step_time": 7.153449058532715
    },
    {
      "epoch": 5.721282958984375e-05,
      "model_forward_time": 1.7019908428192139,
      "step": 59992
    },
    {
      "epoch": 5.721282958984375e-05,
      "step": 59992,
      "training_step_time": 6.864201307296753
    },
    {
      "epoch": 5.721378326416016e-05,
      "model_forward_time": 1.7140605449676514,
      "step": 59993
    },
    {
      "epoch": 5.721378326416016e-05,
      "step": 59993,
      "training_step_time": 6.838456630706787
    },
    {
      "epoch": 5.721473693847656e-05,
      "model_forward_time": 1.7030551433563232,
      "step": 59994
    },
    {
      "epoch": 5.721473693847656e-05,
      "step": 59994,
      "training_step_time": 7.185113430023193
    },
    {
      "epoch": 5.7215690612792966e-05,
      "model_forward_time": 1.7050085067749023,
      "step": 59995
    },
    {
      "epoch": 5.7215690612792966e-05,
      "step": 59995,
      "training_step_time": 7.597851991653442
    },
    {
      "epoch": 5.721664428710938e-05,
      "model_forward_time": 1.7002482414245605,
      "step": 59996
    },
    {
      "epoch": 5.721664428710938e-05,
      "step": 59996,
      "training_step_time": 9.703162908554077
    },
    {
      "epoch": 5.721759796142578e-05,
      "model_forward_time": 2.2842788696289062,
      "step": 59997
    },
    {
      "epoch": 5.721759796142578e-05,
      "step": 59997,
      "training_step_time": 9.513678073883057
    },
    {
      "epoch": 5.7218551635742185e-05,
      "model_forward_time": 1.7903265953063965,
      "step": 59998
    },
    {
      "epoch": 5.7218551635742185e-05,
      "step": 59998,
      "training_step_time": 8.675601720809937
    },
    {
      "epoch": 5.7219505310058596e-05,
      "model_forward_time": 1.70296049118042,
      "step": 59999
    },
    {
      "epoch": 5.7219505310058596e-05,
      "step": 59999,
      "training_step_time": 7.832223176956177
    },
    {
      "epoch": 5.7220458984375e-05,
      "grad_norm": 0.07763783633708954,
      "learning_rate": 3.7727328523752217e-06,
      "loss": 0.0536,
      "step": 60000
    },
    {
      "epoch": 5.7220458984375e-05,
      "model_forward_time": 1.802520990371704,
      "step": 60000
    },
    {
      "dynamics_loss_avg": 0.061157713271677495,
      "epoch": 5.7220458984375e-05,
      "step": 60000
    },
    {
      "action_loss_avg": 0.0014425646484596654,
      "epoch": 5.7220458984375e-05,
      "step": 60000
    },
    {
      "epoch": 5.7220458984375e-05,
      "step": 60000,
      "training_step_time": 6.556708335876465
    },
    {
      "epoch": 5.7221412658691404e-05,
      "model_forward_time": 1.7654392719268799,
      "step": 60001
    },
    {
      "epoch": 5.7221412658691404e-05,
      "step": 60001,
      "training_step_time": 7.653853178024292
    },
    {
      "epoch": 5.7222366333007815e-05,
      "model_forward_time": 1.7382466793060303,
      "step": 60002
    },
    {
      "epoch": 5.7222366333007815e-05,
      "step": 60002,
      "training_step_time": 6.654547929763794
    },
    {
      "epoch": 5.722332000732422e-05,
      "model_forward_time": 1.7531521320343018,
      "step": 60003
    },
    {
      "epoch": 5.722332000732422e-05,
      "step": 60003,
      "training_step_time": 7.734828233718872
    },
    {
      "epoch": 5.722427368164062e-05,
      "model_forward_time": 1.0596063137054443,
      "step": 60004
    },
    {
      "epoch": 5.722427368164062e-05,
      "step": 60004,
      "training_step_time": 8.773843765258789
    },
    {
      "epoch": 5.7225227355957034e-05,
      "model_forward_time": 3.4703640937805176,
      "step": 60005
    },
    {
      "epoch": 5.7225227355957034e-05,
      "step": 60005,
      "training_step_time": 10.526052236557007
    },
    {
      "epoch": 5.722618103027344e-05,
      "model_forward_time": 1.7511394023895264,
      "step": 60006
    },
    {
      "epoch": 5.722618103027344e-05,
      "step": 60006,
      "training_step_time": 12.389121532440186
    },
    {
      "epoch": 5.722713470458984e-05,
      "model_forward_time": 1.7531039714813232,
      "step": 60007
    },
    {
      "epoch": 5.722713470458984e-05,
      "step": 60007,
      "training_step_time": 9.903530359268188
    },
    {
      "epoch": 5.722808837890625e-05,
      "model_forward_time": 1.7479116916656494,
      "step": 60008
    },
    {
      "epoch": 5.722808837890625e-05,
      "step": 60008,
      "training_step_time": 8.655080080032349
    },
    {
      "epoch": 5.7229042053222657e-05,
      "model_forward_time": 1.7548346519470215,
      "step": 60009
    },
    {
      "epoch": 5.7229042053222657e-05,
      "step": 60009,
      "training_step_time": 7.550395488739014
    },
    {
      "epoch": 5.722999572753906e-05,
      "grad_norm": 0.07973657548427582,
      "learning_rate": 3.7711300320074156e-06,
      "loss": 0.0545,
      "step": 60010
    },
    {
      "epoch": 5.722999572753906e-05,
      "model_forward_time": 1.7596185207366943,
      "step": 60010
    },
    {
      "dynamics_loss_avg": 0.049139338359236714,
      "epoch": 5.722999572753906e-05,
      "step": 60010
    },
    {
      "action_loss_avg": 0.001371814904268831,
      "epoch": 5.722999572753906e-05,
      "step": 60010
    },
    {
      "epoch": 5.722999572753906e-05,
      "step": 60010,
      "training_step_time": 7.68665075302124
    },
    {
      "epoch": 5.723094940185547e-05,
      "model_forward_time": 1.7810418605804443,
      "step": 60011
    },
    {
      "epoch": 5.723094940185547e-05,
      "step": 60011,
      "training_step_time": 6.686522483825684
    },
    {
      "epoch": 5.7231903076171875e-05,
      "model_forward_time": 1.7670860290527344,
      "step": 60012
    },
    {
      "epoch": 5.7231903076171875e-05,
      "step": 60012,
      "training_step_time": 7.5587382316589355
    },
    {
      "epoch": 5.723285675048828e-05,
      "model_forward_time": 1.756687879562378,
      "step": 60013
    },
    {
      "epoch": 5.723285675048828e-05,
      "step": 60013,
      "training_step_time": 7.732854604721069
    },
    {
      "epoch": 5.723381042480469e-05,
      "model_forward_time": 1.7573561668395996,
      "step": 60014
    },
    {
      "epoch": 5.723381042480469e-05,
      "step": 60014,
      "training_step_time": 7.728487253189087
    },
    {
      "epoch": 5.7234764099121094e-05,
      "model_forward_time": 2.9229679107666016,
      "step": 60015
    },
    {
      "epoch": 5.7234764099121094e-05,
      "step": 60015,
      "training_step_time": 10.4805588722229
    },
    {
      "epoch": 5.72357177734375e-05,
      "model_forward_time": 1.7551023960113525,
      "step": 60016
    },
    {
      "epoch": 5.72357177734375e-05,
      "step": 60016,
      "training_step_time": 11.358842849731445
    },
    {
      "epoch": 5.723667144775391e-05,
      "model_forward_time": 1.7577195167541504,
      "step": 60017
    },
    {
      "epoch": 5.723667144775391e-05,
      "step": 60017,
      "training_step_time": 7.882566213607788
    },
    {
      "epoch": 5.723762512207031e-05,
      "model_forward_time": 1.7540349960327148,
      "step": 60018
    },
    {
      "epoch": 5.723762512207031e-05,
      "step": 60018,
      "training_step_time": 7.780295133590698
    },
    {
      "epoch": 5.723857879638672e-05,
      "model_forward_time": 1.7509503364562988,
      "step": 60019
    },
    {
      "epoch": 5.723857879638672e-05,
      "step": 60019,
      "training_step_time": 7.722258806228638
    },
    {
      "epoch": 5.723953247070313e-05,
      "grad_norm": 0.09934864938259125,
      "learning_rate": 3.769527346026982e-06,
      "loss": 0.0552,
      "step": 60020
    },
    {
      "epoch": 5.723953247070313e-05,
      "model_forward_time": 1.7560791969299316,
      "step": 60020
    },
    {
      "dynamics_loss_avg": 0.05747096063569188,
      "epoch": 5.723953247070313e-05,
      "step": 60020
    },
    {
      "action_loss_avg": 0.0013066784042166546,
      "epoch": 5.723953247070313e-05,
      "step": 60020
    },
    {
      "epoch": 5.723953247070313e-05,
      "step": 60020,
      "training_step_time": 6.742749929428101
    },
    {
      "epoch": 5.724048614501953e-05,
      "model_forward_time": 1.760932445526123,
      "step": 60021
    },
    {
      "epoch": 5.724048614501953e-05,
      "step": 60021,
      "training_step_time": 6.68340277671814
    },
    {
      "epoch": 5.7241439819335936e-05,
      "model_forward_time": 1.75118088722229,
      "step": 60022
    },
    {
      "epoch": 5.7241439819335936e-05,
      "step": 60022,
      "training_step_time": 7.771591901779175
    },
    {
      "epoch": 5.724239349365235e-05,
      "model_forward_time": 1.7451257705688477,
      "step": 60023
    },
    {
      "epoch": 5.724239349365235e-05,
      "step": 60023,
      "training_step_time": 8.406853199005127
    },
    {
      "epoch": 5.724334716796875e-05,
      "model_forward_time": 2.915095329284668,
      "step": 60024
    },
    {
      "epoch": 5.724334716796875e-05,
      "step": 60024,
      "training_step_time": 8.244600057601929
    },
    {
      "epoch": 5.7244300842285155e-05,
      "model_forward_time": 1.745312213897705,
      "step": 60025
    },
    {
      "epoch": 5.7244300842285155e-05,
      "step": 60025,
      "training_step_time": 8.763999938964844
    },
    {
      "epoch": 5.7245254516601566e-05,
      "model_forward_time": 1.3794522285461426,
      "step": 60026
    },
    {
      "epoch": 5.7245254516601566e-05,
      "step": 60026,
      "training_step_time": 8.84378457069397
    },
    {
      "epoch": 5.724620819091797e-05,
      "model_forward_time": 1.7495028972625732,
      "step": 60027
    },
    {
      "epoch": 5.724620819091797e-05,
      "step": 60027,
      "training_step_time": 7.726773738861084
    },
    {
      "epoch": 5.7247161865234374e-05,
      "model_forward_time": 1.7603647708892822,
      "step": 60028
    },
    {
      "epoch": 5.7247161865234374e-05,
      "step": 60028,
      "training_step_time": 7.767998456954956
    },
    {
      "epoch": 5.7248115539550785e-05,
      "model_forward_time": 1.7475969791412354,
      "step": 60029
    },
    {
      "epoch": 5.7248115539550785e-05,
      "step": 60029,
      "training_step_time": 6.74087381362915
    },
    {
      "epoch": 5.724906921386719e-05,
      "grad_norm": 0.09428080171346664,
      "learning_rate": 3.767924794609188e-06,
      "loss": 0.057,
      "step": 60030
    },
    {
      "epoch": 5.724906921386719e-05,
      "model_forward_time": 1.7432918548583984,
      "step": 60030
    },
    {
      "dynamics_loss_avg": 0.05784994382411242,
      "epoch": 5.724906921386719e-05,
      "step": 60030
    },
    {
      "action_loss_avg": 0.0018884390970924868,
      "epoch": 5.724906921386719e-05,
      "step": 60030
    },
    {
      "epoch": 5.724906921386719e-05,
      "step": 60030,
      "training_step_time": 6.757067441940308
    },
    {
      "epoch": 5.725002288818359e-05,
      "model_forward_time": 1.7423810958862305,
      "step": 60031
    },
    {
      "epoch": 5.725002288818359e-05,
      "step": 60031,
      "training_step_time": 7.819421768188477
    },
    {
      "epoch": 5.72509765625e-05,
      "model_forward_time": 1.7569406032562256,
      "step": 60032
    },
    {
      "epoch": 5.72509765625e-05,
      "step": 60032,
      "training_step_time": 8.250272989273071
    },
    {
      "epoch": 5.725193023681641e-05,
      "model_forward_time": 2.8905434608459473,
      "step": 60033
    },
    {
      "epoch": 5.725193023681641e-05,
      "step": 60033,
      "training_step_time": 8.702702283859253
    },
    {
      "epoch": 5.725288391113281e-05,
      "model_forward_time": 1.7446751594543457,
      "step": 60034
    },
    {
      "epoch": 5.725288391113281e-05,
      "step": 60034,
      "training_step_time": 9.106850385665894
    },
    {
      "epoch": 5.7253837585449215e-05,
      "model_forward_time": 1.7461531162261963,
      "step": 60035
    },
    {
      "epoch": 5.7253837585449215e-05,
      "step": 60035,
      "training_step_time": 7.650698900222778
    },
    {
      "epoch": 5.7254791259765626e-05,
      "model_forward_time": 1.7414591312408447,
      "step": 60036
    },
    {
      "epoch": 5.7254791259765626e-05,
      "step": 60036,
      "training_step_time": 7.7646644115448
    },
    {
      "epoch": 5.725574493408203e-05,
      "model_forward_time": 1.7470519542694092,
      "step": 60037
    },
    {
      "epoch": 5.725574493408203e-05,
      "step": 60037,
      "training_step_time": 7.5660154819488525
    },
    {
      "epoch": 5.7256698608398434e-05,
      "model_forward_time": 1.7409496307373047,
      "step": 60038
    },
    {
      "epoch": 5.7256698608398434e-05,
      "step": 60038,
      "training_step_time": 6.683035135269165
    },
    {
      "epoch": 5.7257652282714845e-05,
      "model_forward_time": 1.7417285442352295,
      "step": 60039
    },
    {
      "epoch": 5.7257652282714845e-05,
      "step": 60039,
      "training_step_time": 6.674144983291626
    },
    {
      "epoch": 5.725860595703125e-05,
      "grad_norm": 0.11033880710601807,
      "learning_rate": 3.7663223779292858e-06,
      "loss": 0.0571,
      "step": 60040
    },
    {
      "epoch": 5.725860595703125e-05,
      "model_forward_time": 1.7513394355773926,
      "step": 60040
    },
    {
      "dynamics_loss_avg": 0.04317158907651901,
      "epoch": 5.725860595703125e-05,
      "step": 60040
    },
    {
      "action_loss_avg": 0.0018834990449249744,
      "epoch": 5.725860595703125e-05,
      "step": 60040
    },
    {
      "epoch": 5.725860595703125e-05,
      "step": 60040,
      "training_step_time": 8.489973068237305
    },
    {
      "epoch": 5.725955963134765e-05,
      "model_forward_time": 1.763037919998169,
      "step": 60041
    },
    {
      "epoch": 5.725955963134765e-05,
      "step": 60041,
      "training_step_time": 7.93818736076355
    },
    {
      "epoch": 5.7260513305664064e-05,
      "model_forward_time": 2.9125709533691406,
      "step": 60042
    },
    {
      "epoch": 5.7260513305664064e-05,
      "step": 60042,
      "training_step_time": 8.381959676742554
    },
    {
      "epoch": 5.726146697998047e-05,
      "model_forward_time": 1.7565383911132812,
      "step": 60043
    },
    {
      "epoch": 5.726146697998047e-05,
      "step": 60043,
      "training_step_time": 9.383681774139404
    },
    {
      "epoch": 5.726242065429687e-05,
      "model_forward_time": 1.751539945602417,
      "step": 60044
    },
    {
      "epoch": 5.726242065429687e-05,
      "step": 60044,
      "training_step_time": 7.6487085819244385
    },
    {
      "epoch": 5.726337432861328e-05,
      "model_forward_time": 1.758579969406128,
      "step": 60045
    },
    {
      "epoch": 5.726337432861328e-05,
      "step": 60045,
      "training_step_time": 7.814872741699219
    },
    {
      "epoch": 5.726432800292969e-05,
      "model_forward_time": 1.759495735168457,
      "step": 60046
    },
    {
      "epoch": 5.726432800292969e-05,
      "step": 60046,
      "training_step_time": 7.6150782108306885
    },
    {
      "epoch": 5.726528167724609e-05,
      "model_forward_time": 1.760634422302246,
      "step": 60047
    },
    {
      "epoch": 5.726528167724609e-05,
      "step": 60047,
      "training_step_time": 6.677626371383667
    },
    {
      "epoch": 5.72662353515625e-05,
      "model_forward_time": 1.7546594142913818,
      "step": 60048
    },
    {
      "epoch": 5.72662353515625e-05,
      "step": 60048,
      "training_step_time": 6.686840534210205
    },
    {
      "epoch": 5.7267189025878906e-05,
      "model_forward_time": 1.7580649852752686,
      "step": 60049
    },
    {
      "epoch": 5.7267189025878906e-05,
      "step": 60049,
      "training_step_time": 7.43030858039856
    },
    {
      "epoch": 5.726814270019531e-05,
      "grad_norm": 0.09804334491491318,
      "learning_rate": 3.7647200961625124e-06,
      "loss": 0.0547,
      "step": 60050
    },
    {
      "epoch": 5.726814270019531e-05,
      "model_forward_time": 1.7608506679534912,
      "step": 60050
    },
    {
      "dynamics_loss_avg": 0.06021001972258091,
      "epoch": 5.726814270019531e-05,
      "step": 60050
    },
    {
      "action_loss_avg": 0.001971022109501064,
      "epoch": 5.726814270019531e-05,
      "step": 60050
    },
    {
      "epoch": 5.726814270019531e-05,
      "step": 60050,
      "training_step_time": 7.762090682983398
    },
    {
      "epoch": 5.726909637451172e-05,
      "model_forward_time": 1.7720129489898682,
      "step": 60051
    },
    {
      "epoch": 5.726909637451172e-05,
      "step": 60051,
      "training_step_time": 9.66998839378357
    },
    {
      "epoch": 5.7270050048828125e-05,
      "model_forward_time": 1.7554371356964111,
      "step": 60052
    },
    {
      "epoch": 5.7270050048828125e-05,
      "step": 60052,
      "training_step_time": 8.258203268051147
    },
    {
      "epoch": 5.727100372314453e-05,
      "model_forward_time": 1.7536017894744873,
      "step": 60053
    },
    {
      "epoch": 5.727100372314453e-05,
      "step": 60053,
      "training_step_time": 9.314626216888428
    },
    {
      "epoch": 5.727195739746094e-05,
      "model_forward_time": 1.388772964477539,
      "step": 60054
    },
    {
      "epoch": 5.727195739746094e-05,
      "step": 60054,
      "training_step_time": 7.895965814590454
    },
    {
      "epoch": 5.7272911071777343e-05,
      "model_forward_time": 1.7528965473175049,
      "step": 60055
    },
    {
      "epoch": 5.7272911071777343e-05,
      "step": 60055,
      "training_step_time": 8.758444547653198
    },
    {
      "epoch": 5.727386474609375e-05,
      "model_forward_time": 1.7616076469421387,
      "step": 60056
    },
    {
      "epoch": 5.727386474609375e-05,
      "step": 60056,
      "training_step_time": 6.880075931549072
    },
    {
      "epoch": 5.727481842041016e-05,
      "model_forward_time": 1.760603666305542,
      "step": 60057
    },
    {
      "epoch": 5.727481842041016e-05,
      "step": 60057,
      "training_step_time": 6.825269460678101
    },
    {
      "epoch": 5.727577209472656e-05,
      "model_forward_time": 1.7636053562164307,
      "step": 60058
    },
    {
      "epoch": 5.727577209472656e-05,
      "step": 60058,
      "training_step_time": 7.120996952056885
    },
    {
      "epoch": 5.7276725769042966e-05,
      "model_forward_time": 1.7633259296417236,
      "step": 60059
    },
    {
      "epoch": 5.7276725769042966e-05,
      "step": 60059,
      "training_step_time": 7.794452428817749
    },
    {
      "epoch": 5.727767944335938e-05,
      "grad_norm": 0.08898717164993286,
      "learning_rate": 3.763117949484092e-06,
      "loss": 0.0557,
      "step": 60060
    },
    {
      "epoch": 5.727767944335938e-05,
      "model_forward_time": 1.7610538005828857,
      "step": 60060
    },
    {
      "dynamics_loss_avg": 0.049811975285410884,
      "epoch": 5.727767944335938e-05,
      "step": 60060
    },
    {
      "action_loss_avg": 0.0012302755261771382,
      "epoch": 5.727767944335938e-05,
      "step": 60060
    },
    {
      "epoch": 5.727767944335938e-05,
      "step": 60060,
      "training_step_time": 7.6507227420806885
    },
    {
      "epoch": 5.727863311767578e-05,
      "model_forward_time": 2.8964107036590576,
      "step": 60061
    },
    {
      "epoch": 5.727863311767578e-05,
      "step": 60061,
      "training_step_time": 7.665656805038452
    },
    {
      "epoch": 5.7279586791992185e-05,
      "model_forward_time": 1.7539713382720947,
      "step": 60062
    },
    {
      "epoch": 5.7279586791992185e-05,
      "step": 60062,
      "training_step_time": 8.31132984161377
    },
    {
      "epoch": 5.7280540466308596e-05,
      "model_forward_time": 1.758190631866455,
      "step": 60063
    },
    {
      "epoch": 5.7280540466308596e-05,
      "step": 60063,
      "training_step_time": 8.877285718917847
    },
    {
      "epoch": 5.7281494140625e-05,
      "model_forward_time": 1.7560083866119385,
      "step": 60064
    },
    {
      "epoch": 5.7281494140625e-05,
      "step": 60064,
      "training_step_time": 7.675991773605347
    },
    {
      "epoch": 5.7282447814941404e-05,
      "model_forward_time": 1.7574269771575928,
      "step": 60065
    },
    {
      "epoch": 5.7282447814941404e-05,
      "step": 60065,
      "training_step_time": 7.131864309310913
    },
    {
      "epoch": 5.7283401489257815e-05,
      "model_forward_time": 1.757903814315796,
      "step": 60066
    },
    {
      "epoch": 5.7283401489257815e-05,
      "step": 60066,
      "training_step_time": 6.686760425567627
    },
    {
      "epoch": 5.728435516357422e-05,
      "model_forward_time": 1.7557997703552246,
      "step": 60067
    },
    {
      "epoch": 5.728435516357422e-05,
      "step": 60067,
      "training_step_time": 7.387763738632202
    },
    {
      "epoch": 5.728530883789062e-05,
      "model_forward_time": 1.7585089206695557,
      "step": 60068
    },
    {
      "epoch": 5.728530883789062e-05,
      "step": 60068,
      "training_step_time": 7.707643032073975
    },
    {
      "epoch": 5.7286262512207034e-05,
      "model_forward_time": 1.7489945888519287,
      "step": 60069
    },
    {
      "epoch": 5.7286262512207034e-05,
      "step": 60069,
      "training_step_time": 8.479404211044312
    },
    {
      "epoch": 5.728721618652344e-05,
      "grad_norm": 0.09205850213766098,
      "learning_rate": 3.7615159380692347e-06,
      "loss": 0.0553,
      "step": 60070
    },
    {
      "epoch": 5.728721618652344e-05,
      "model_forward_time": 2.8979666233062744,
      "step": 60070
    },
    {
      "dynamics_loss_avg": 0.04195591686293483,
      "epoch": 5.728721618652344e-05,
      "step": 60070
    },
    {
      "action_loss_avg": 0.0013462285452988,
      "epoch": 5.728721618652344e-05,
      "step": 60070
    },
    {
      "epoch": 5.728721618652344e-05,
      "step": 60070,
      "training_step_time": 9.711253881454468
    },
    {
      "epoch": 5.728816986083984e-05,
      "model_forward_time": 1.7554550170898438,
      "step": 60071
    },
    {
      "epoch": 5.728816986083984e-05,
      "step": 60071,
      "training_step_time": 8.935545206069946
    },
    {
      "epoch": 5.728912353515625e-05,
      "model_forward_time": 1.7610411643981934,
      "step": 60072
    },
    {
      "epoch": 5.728912353515625e-05,
      "step": 60072,
      "training_step_time": 7.826855182647705
    },
    {
      "epoch": 5.729007720947266e-05,
      "model_forward_time": 1.754045009613037,
      "step": 60073
    },
    {
      "epoch": 5.729007720947266e-05,
      "step": 60073,
      "training_step_time": 7.788950204849243
    },
    {
      "epoch": 5.729103088378906e-05,
      "model_forward_time": 1.7432503700256348,
      "step": 60074
    },
    {
      "epoch": 5.729103088378906e-05,
      "step": 60074,
      "training_step_time": 7.148743391036987
    },
    {
      "epoch": 5.729198455810547e-05,
      "model_forward_time": 1.7297537326812744,
      "step": 60075
    },
    {
      "epoch": 5.729198455810547e-05,
      "step": 60075,
      "training_step_time": 6.753296136856079
    },
    {
      "epoch": 5.7292938232421875e-05,
      "model_forward_time": 1.7600350379943848,
      "step": 60076
    },
    {
      "epoch": 5.7292938232421875e-05,
      "step": 60076,
      "training_step_time": 6.807110786437988
    },
    {
      "epoch": 5.729389190673828e-05,
      "model_forward_time": 1.7545287609100342,
      "step": 60077
    },
    {
      "epoch": 5.729389190673828e-05,
      "step": 60077,
      "training_step_time": 7.7878382205963135
    },
    {
      "epoch": 5.729484558105469e-05,
      "model_forward_time": 1.7675321102142334,
      "step": 60078
    },
    {
      "epoch": 5.729484558105469e-05,
      "step": 60078,
      "training_step_time": 8.154456615447998
    },
    {
      "epoch": 5.7295799255371094e-05,
      "model_forward_time": 2.859545946121216,
      "step": 60079
    },
    {
      "epoch": 5.7295799255371094e-05,
      "step": 60079,
      "training_step_time": 7.7398521900177
    },
    {
      "epoch": 5.72967529296875e-05,
      "grad_norm": 0.10866863280534744,
      "learning_rate": 3.759914062093132e-06,
      "loss": 0.0594,
      "step": 60080
    },
    {
      "epoch": 5.72967529296875e-05,
      "model_forward_time": 1.7472660541534424,
      "step": 60080
    },
    {
      "dynamics_loss_avg": 0.04608449712395668,
      "epoch": 5.72967529296875e-05,
      "step": 60080
    },
    {
      "action_loss_avg": 0.00148391138354782,
      "epoch": 5.72967529296875e-05,
      "step": 60080
    },
    {
      "epoch": 5.72967529296875e-05,
      "step": 60080,
      "training_step_time": 10.075917720794678
    },
    {
      "epoch": 5.729770660400391e-05,
      "model_forward_time": 1.7394208908081055,
      "step": 60081
    },
    {
      "epoch": 5.729770660400391e-05,
      "step": 60081,
      "training_step_time": 7.803720712661743
    },
    {
      "epoch": 5.729866027832031e-05,
      "model_forward_time": 1.7402572631835938,
      "step": 60082
    },
    {
      "epoch": 5.729866027832031e-05,
      "step": 60082,
      "training_step_time": 8.494846105575562
    },
    {
      "epoch": 5.729961395263672e-05,
      "model_forward_time": 1.7419085502624512,
      "step": 60083
    },
    {
      "epoch": 5.729961395263672e-05,
      "step": 60083,
      "training_step_time": 7.662774085998535
    },
    {
      "epoch": 5.730056762695313e-05,
      "model_forward_time": 1.7480359077453613,
      "step": 60084
    },
    {
      "epoch": 5.730056762695313e-05,
      "step": 60084,
      "training_step_time": 6.778632164001465
    },
    {
      "epoch": 5.730152130126953e-05,
      "model_forward_time": 1.7350952625274658,
      "step": 60085
    },
    {
      "epoch": 5.730152130126953e-05,
      "step": 60085,
      "training_step_time": 6.783989429473877
    },
    {
      "epoch": 5.7302474975585936e-05,
      "model_forward_time": 1.741581678390503,
      "step": 60086
    },
    {
      "epoch": 5.7302474975585936e-05,
      "step": 60086,
      "training_step_time": 7.765599489212036
    },
    {
      "epoch": 5.730342864990235e-05,
      "model_forward_time": 1.7497332096099854,
      "step": 60087
    },
    {
      "epoch": 5.730342864990235e-05,
      "step": 60087,
      "training_step_time": 7.332504749298096
    },
    {
      "epoch": 5.730438232421875e-05,
      "model_forward_time": 2.8486428260803223,
      "step": 60088
    },
    {
      "epoch": 5.730438232421875e-05,
      "step": 60088,
      "training_step_time": 9.284894466400146
    },
    {
      "epoch": 5.7305335998535155e-05,
      "model_forward_time": 1.7526521682739258,
      "step": 60089
    },
    {
      "epoch": 5.7305335998535155e-05,
      "step": 60089,
      "training_step_time": 9.567194700241089
    },
    {
      "epoch": 5.7306289672851566e-05,
      "grad_norm": 0.07830165326595306,
      "learning_rate": 3.7583123217309613e-06,
      "loss": 0.0554,
      "step": 60090
    },
    {
      "epoch": 5.7306289672851566e-05,
      "model_forward_time": 1.743333339691162,
      "step": 60090
    },
    {
      "dynamics_loss_avg": 0.04568294072523713,
      "epoch": 5.7306289672851566e-05,
      "step": 60090
    },
    {
      "action_loss_avg": 0.0013778974476736038,
      "epoch": 5.7306289672851566e-05,
      "step": 60090
    },
    {
      "epoch": 5.7306289672851566e-05,
      "step": 60090,
      "training_step_time": 8.810214519500732
    },
    {
      "epoch": 5.730724334716797e-05,
      "model_forward_time": 1.7268223762512207,
      "step": 60091
    },
    {
      "epoch": 5.730724334716797e-05,
      "step": 60091,
      "training_step_time": 8.749773263931274
    },
    {
      "epoch": 5.7308197021484374e-05,
      "model_forward_time": 1.7401118278503418,
      "step": 60092
    },
    {
      "epoch": 5.7308197021484374e-05,
      "step": 60092,
      "training_step_time": 7.58951997756958
    },
    {
      "epoch": 5.7309150695800785e-05,
      "model_forward_time": 1.743330478668213,
      "step": 60093
    },
    {
      "epoch": 5.7309150695800785e-05,
      "step": 60093,
      "training_step_time": 6.762186527252197
    },
    {
      "epoch": 5.731010437011719e-05,
      "model_forward_time": 1.7350797653198242,
      "step": 60094
    },
    {
      "epoch": 5.731010437011719e-05,
      "step": 60094,
      "training_step_time": 6.769640207290649
    },
    {
      "epoch": 5.731105804443359e-05,
      "model_forward_time": 1.7384164333343506,
      "step": 60095
    },
    {
      "epoch": 5.731105804443359e-05,
      "step": 60095,
      "training_step_time": 7.7211644649505615
    },
    {
      "epoch": 5.731201171875e-05,
      "model_forward_time": 1.748255729675293,
      "step": 60096
    },
    {
      "epoch": 5.731201171875e-05,
      "step": 60096,
      "training_step_time": 7.0396716594696045
    },
    {
      "epoch": 5.731296539306641e-05,
      "model_forward_time": 1.7391531467437744,
      "step": 60097
    },
    {
      "epoch": 5.731296539306641e-05,
      "step": 60097,
      "training_step_time": 8.215105533599854
    },
    {
      "epoch": 5.731391906738281e-05,
      "model_forward_time": 2.923583984375,
      "step": 60098
    },
    {
      "epoch": 5.731391906738281e-05,
      "step": 60098,
      "training_step_time": 8.296728372573853
    },
    {
      "epoch": 5.7314872741699216e-05,
      "model_forward_time": 1.742933750152588,
      "step": 60099
    },
    {
      "epoch": 5.7314872741699216e-05,
      "step": 60099,
      "training_step_time": 11.284947395324707
    },
    {
      "epoch": 5.7315826416015626e-05,
      "grad_norm": 0.09322558343410492,
      "learning_rate": 3.7567107171578905e-06,
      "loss": 0.0587,
      "step": 60100
    },
    {
      "epoch": 5.7315826416015626e-05,
      "model_forward_time": 1.7354328632354736,
      "step": 60100
    },
    {
      "dynamics_loss_avg": 0.0642062658444047,
      "epoch": 5.7315826416015626e-05,
      "step": 60100
    },
    {
      "action_loss_avg": 0.0020969354431144894,
      "epoch": 5.7315826416015626e-05,
      "step": 60100
    },
    {
      "epoch": 5.7315826416015626e-05,
      "step": 60100,
      "training_step_time": 7.669465065002441
    },
    {
      "epoch": 5.731678009033203e-05,
      "model_forward_time": 1.7443628311157227,
      "step": 60101
    },
    {
      "epoch": 5.731678009033203e-05,
      "step": 60101,
      "training_step_time": 7.748594045639038
    },
    {
      "epoch": 5.7317733764648434e-05,
      "model_forward_time": 1.7459123134613037,
      "step": 60102
    },
    {
      "epoch": 5.7317733764648434e-05,
      "step": 60102,
      "training_step_time": 7.1203227043151855
    },
    {
      "epoch": 5.7318687438964845e-05,
      "model_forward_time": 1.736311435699463,
      "step": 60103
    },
    {
      "epoch": 5.7318687438964845e-05,
      "step": 60103,
      "training_step_time": 6.7718400955200195
    },
    {
      "epoch": 5.731964111328125e-05,
      "model_forward_time": 1.7395849227905273,
      "step": 60104
    },
    {
      "epoch": 5.731964111328125e-05,
      "step": 60104,
      "training_step_time": 7.589402675628662
    },
    {
      "epoch": 5.732059478759765e-05,
      "model_forward_time": 1.739473819732666,
      "step": 60105
    },
    {
      "epoch": 5.732059478759765e-05,
      "step": 60105,
      "training_step_time": 7.746034145355225
    },
    {
      "epoch": 5.7321548461914064e-05,
      "model_forward_time": 1.765230655670166,
      "step": 60106
    },
    {
      "epoch": 5.7321548461914064e-05,
      "step": 60106,
      "training_step_time": 7.630815267562866
    },
    {
      "epoch": 5.732250213623047e-05,
      "model_forward_time": 2.8603882789611816,
      "step": 60107
    },
    {
      "epoch": 5.732250213623047e-05,
      "step": 60107,
      "training_step_time": 7.707943677902222
    },
    {
      "epoch": 5.732345581054687e-05,
      "model_forward_time": 1.7291035652160645,
      "step": 60108
    },
    {
      "epoch": 5.732345581054687e-05,
      "step": 60108,
      "training_step_time": 10.162500619888306
    },
    {
      "epoch": 5.732440948486328e-05,
      "model_forward_time": 1.732818365097046,
      "step": 60109
    },
    {
      "epoch": 5.732440948486328e-05,
      "step": 60109,
      "training_step_time": 7.679349660873413
    },
    {
      "epoch": 5.732536315917969e-05,
      "grad_norm": 0.12139108031988144,
      "learning_rate": 3.7551092485490657e-06,
      "loss": 0.0592,
      "step": 60110
    },
    {
      "epoch": 5.732536315917969e-05,
      "model_forward_time": 1.7380001544952393,
      "step": 60110
    },
    {
      "dynamics_loss_avg": 0.0503142599016428,
      "epoch": 5.732536315917969e-05,
      "step": 60110
    },
    {
      "action_loss_avg": 0.00260924362228252,
      "epoch": 5.732536315917969e-05,
      "step": 60110
    },
    {
      "epoch": 5.732536315917969e-05,
      "step": 60110,
      "training_step_time": 7.706706285476685
    },
    {
      "epoch": 5.732631683349609e-05,
      "model_forward_time": 1.7375538349151611,
      "step": 60111
    },
    {
      "epoch": 5.732631683349609e-05,
      "step": 60111,
      "training_step_time": 7.494556665420532
    },
    {
      "epoch": 5.73272705078125e-05,
      "model_forward_time": 1.7384872436523438,
      "step": 60112
    },
    {
      "epoch": 5.73272705078125e-05,
      "step": 60112,
      "training_step_time": 6.737426519393921
    },
    {
      "epoch": 5.7328224182128906e-05,
      "model_forward_time": 1.7321486473083496,
      "step": 60113
    },
    {
      "epoch": 5.7328224182128906e-05,
      "step": 60113,
      "training_step_time": 7.345104455947876
    },
    {
      "epoch": 5.732917785644531e-05,
      "model_forward_time": 1.7326929569244385,
      "step": 60114
    },
    {
      "epoch": 5.732917785644531e-05,
      "step": 60114,
      "training_step_time": 7.759700059890747
    },
    {
      "epoch": 5.733013153076172e-05,
      "model_forward_time": 1.7675669193267822,
      "step": 60115
    },
    {
      "epoch": 5.733013153076172e-05,
      "step": 60115,
      "training_step_time": 7.551110744476318
    },
    {
      "epoch": 5.7331085205078125e-05,
      "model_forward_time": 2.8692374229431152,
      "step": 60116
    },
    {
      "epoch": 5.7331085205078125e-05,
      "step": 60116,
      "training_step_time": 8.766173124313354
    },
    {
      "epoch": 5.733203887939453e-05,
      "model_forward_time": 1.7513387203216553,
      "step": 60117
    },
    {
      "epoch": 5.733203887939453e-05,
      "step": 60117,
      "training_step_time": 8.53816819190979
    },
    {
      "epoch": 5.733299255371094e-05,
      "model_forward_time": 1.74226975440979,
      "step": 60118
    },
    {
      "epoch": 5.733299255371094e-05,
      "step": 60118,
      "training_step_time": 9.66703987121582
    },
    {
      "epoch": 5.7333946228027344e-05,
      "model_forward_time": 1.7452092170715332,
      "step": 60119
    },
    {
      "epoch": 5.7333946228027344e-05,
      "step": 60119,
      "training_step_time": 8.410962104797363
    },
    {
      "epoch": 5.733489990234375e-05,
      "grad_norm": 0.11238377541303635,
      "learning_rate": 3.7535079160796207e-06,
      "loss": 0.0582,
      "step": 60120
    },
    {
      "epoch": 5.733489990234375e-05,
      "model_forward_time": 1.752553939819336,
      "step": 60120
    },
    {
      "dynamics_loss_avg": 0.05654789470136166,
      "epoch": 5.733489990234375e-05,
      "step": 60120
    },
    {
      "action_loss_avg": 0.0015148662962019444,
      "epoch": 5.733489990234375e-05,
      "step": 60120
    },
    {
      "epoch": 5.733489990234375e-05,
      "step": 60120,
      "training_step_time": 7.5984930992126465
    },
    {
      "epoch": 5.733585357666016e-05,
      "model_forward_time": 1.7472419738769531,
      "step": 60121
    },
    {
      "epoch": 5.733585357666016e-05,
      "step": 60121,
      "training_step_time": 6.7591941356658936
    },
    {
      "epoch": 5.733680725097656e-05,
      "model_forward_time": 1.7468252182006836,
      "step": 60122
    },
    {
      "epoch": 5.733680725097656e-05,
      "step": 60122,
      "training_step_time": 6.744765043258667
    },
    {
      "epoch": 5.7337760925292966e-05,
      "model_forward_time": 1.7537710666656494,
      "step": 60123
    },
    {
      "epoch": 5.7337760925292966e-05,
      "step": 60123,
      "training_step_time": 7.725963115692139
    },
    {
      "epoch": 5.733871459960938e-05,
      "model_forward_time": 1.7480597496032715,
      "step": 60124
    },
    {
      "epoch": 5.733871459960938e-05,
      "step": 60124,
      "training_step_time": 7.674371242523193
    },
    {
      "epoch": 5.733966827392578e-05,
      "model_forward_time": 2.904590129852295,
      "step": 60125
    },
    {
      "epoch": 5.733966827392578e-05,
      "step": 60125,
      "training_step_time": 9.096695184707642
    },
    {
      "epoch": 5.7340621948242185e-05,
      "model_forward_time": 1.7529819011688232,
      "step": 60126
    },
    {
      "epoch": 5.7340621948242185e-05,
      "step": 60126,
      "training_step_time": 9.627246141433716
    },
    {
      "epoch": 5.7341575622558596e-05,
      "model_forward_time": 1.7573094367980957,
      "step": 60127
    },
    {
      "epoch": 5.7341575622558596e-05,
      "step": 60127,
      "training_step_time": 8.381320238113403
    },
    {
      "epoch": 5.7342529296875e-05,
      "model_forward_time": 1.7451848983764648,
      "step": 60128
    },
    {
      "epoch": 5.7342529296875e-05,
      "step": 60128,
      "training_step_time": 7.696739196777344
    },
    {
      "epoch": 5.7343482971191404e-05,
      "model_forward_time": 1.7511379718780518,
      "step": 60129
    },
    {
      "epoch": 5.7343482971191404e-05,
      "step": 60129,
      "training_step_time": 7.557259798049927
    },
    {
      "epoch": 5.7344436645507815e-05,
      "grad_norm": 0.08156536519527435,
      "learning_rate": 3.751906719924676e-06,
      "loss": 0.0568,
      "step": 60130
    },
    {
      "epoch": 5.7344436645507815e-05,
      "model_forward_time": 1.76228928565979,
      "step": 60130
    },
    {
      "dynamics_loss_avg": 0.042368566058576106,
      "epoch": 5.7344436645507815e-05,
      "step": 60130
    },
    {
      "action_loss_avg": 0.001032938330899924,
      "epoch": 5.7344436645507815e-05,
      "step": 60130
    },
    {
      "epoch": 5.7344436645507815e-05,
      "step": 60130,
      "training_step_time": 6.73598575592041
    },
    {
      "epoch": 5.734539031982422e-05,
      "model_forward_time": 1.740004062652588,
      "step": 60131
    },
    {
      "epoch": 5.734539031982422e-05,
      "step": 60131,
      "training_step_time": 6.761702299118042
    },
    {
      "epoch": 5.734634399414062e-05,
      "model_forward_time": 1.7586417198181152,
      "step": 60132
    },
    {
      "epoch": 5.734634399414062e-05,
      "step": 60132,
      "training_step_time": 7.70191216468811
    },
    {
      "epoch": 5.7347297668457034e-05,
      "model_forward_time": 1.7538824081420898,
      "step": 60133
    },
    {
      "epoch": 5.7347297668457034e-05,
      "step": 60133,
      "training_step_time": 7.570311069488525
    },
    {
      "epoch": 5.734825134277344e-05,
      "model_forward_time": 2.9543380737304688,
      "step": 60134
    },
    {
      "epoch": 5.734825134277344e-05,
      "step": 60134,
      "training_step_time": 7.690210819244385
    },
    {
      "epoch": 5.734920501708984e-05,
      "model_forward_time": 1.7571637630462646,
      "step": 60135
    },
    {
      "epoch": 5.734920501708984e-05,
      "step": 60135,
      "training_step_time": 7.790525913238525
    },
    {
      "epoch": 5.735015869140625e-05,
      "model_forward_time": 1.7466363906860352,
      "step": 60136
    },
    {
      "epoch": 5.735015869140625e-05,
      "step": 60136,
      "training_step_time": 8.810267210006714
    },
    {
      "epoch": 5.735111236572266e-05,
      "model_forward_time": 1.7497665882110596,
      "step": 60137
    },
    {
      "epoch": 5.735111236572266e-05,
      "step": 60137,
      "training_step_time": 7.596143484115601
    },
    {
      "epoch": 5.735206604003906e-05,
      "model_forward_time": 1.7613110542297363,
      "step": 60138
    },
    {
      "epoch": 5.735206604003906e-05,
      "step": 60138,
      "training_step_time": 7.611818075180054
    },
    {
      "epoch": 5.735301971435547e-05,
      "model_forward_time": 1.7508461475372314,
      "step": 60139
    },
    {
      "epoch": 5.735301971435547e-05,
      "step": 60139,
      "training_step_time": 6.806271076202393
    },
    {
      "epoch": 5.7353973388671876e-05,
      "grad_norm": 0.09313292801380157,
      "learning_rate": 3.7503056602593375e-06,
      "loss": 0.0555,
      "step": 60140
    },
    {
      "epoch": 5.7353973388671876e-05,
      "model_forward_time": 1.7535500526428223,
      "step": 60140
    },
    {
      "dynamics_loss_avg": 0.053247787058353424,
      "epoch": 5.7353973388671876e-05,
      "step": 60140
    },
    {
      "action_loss_avg": 0.00136901019141078,
      "epoch": 5.7353973388671876e-05,
      "step": 60140
    },
    {
      "epoch": 5.7353973388671876e-05,
      "step": 60140,
      "training_step_time": 6.764748811721802
    },
    {
      "epoch": 5.735492706298828e-05,
      "model_forward_time": 1.75807785987854,
      "step": 60141
    },
    {
      "epoch": 5.735492706298828e-05,
      "step": 60141,
      "training_step_time": 7.522315263748169
    },
    {
      "epoch": 5.735588073730469e-05,
      "model_forward_time": 1.769637107849121,
      "step": 60142
    },
    {
      "epoch": 5.735588073730469e-05,
      "step": 60142,
      "training_step_time": 7.657599925994873
    },
    {
      "epoch": 5.7356834411621094e-05,
      "model_forward_time": 1.742746353149414,
      "step": 60143
    },
    {
      "epoch": 5.7356834411621094e-05,
      "step": 60143,
      "training_step_time": 9.676361322402954
    },
    {
      "epoch": 5.73577880859375e-05,
      "model_forward_time": 1.7489628791809082,
      "step": 60144
    },
    {
      "epoch": 5.73577880859375e-05,
      "step": 60144,
      "training_step_time": 9.650090217590332
    },
    {
      "epoch": 5.735874176025391e-05,
      "model_forward_time": 1.7627265453338623,
      "step": 60145
    },
    {
      "epoch": 5.735874176025391e-05,
      "step": 60145,
      "training_step_time": 8.387993097305298
    },
    {
      "epoch": 5.735969543457031e-05,
      "model_forward_time": 1.7430031299591064,
      "step": 60146
    },
    {
      "epoch": 5.735969543457031e-05,
      "step": 60146,
      "training_step_time": 7.576946258544922
    },
    {
      "epoch": 5.736064910888672e-05,
      "model_forward_time": 1.753957748413086,
      "step": 60147
    },
    {
      "epoch": 5.736064910888672e-05,
      "step": 60147,
      "training_step_time": 8.67125153541565
    },
    {
      "epoch": 5.736160278320313e-05,
      "model_forward_time": 1.759885549545288,
      "step": 60148
    },
    {
      "epoch": 5.736160278320313e-05,
      "step": 60148,
      "training_step_time": 7.274247169494629
    },
    {
      "epoch": 5.736255645751953e-05,
      "model_forward_time": 1.7532415390014648,
      "step": 60149
    },
    {
      "epoch": 5.736255645751953e-05,
      "step": 60149,
      "training_step_time": 6.749398946762085
    },
    {
      "epoch": 5.7363510131835936e-05,
      "grad_norm": 0.09895500540733337,
      "learning_rate": 3.7487047372586927e-06,
      "loss": 0.0558,
      "step": 60150
    },
    {
      "epoch": 5.7363510131835936e-05,
      "model_forward_time": 1.7564775943756104,
      "step": 60150
    },
    {
      "dynamics_loss_avg": 0.05217934902757406,
      "epoch": 5.7363510131835936e-05,
      "step": 60150
    },
    {
      "action_loss_avg": 0.0016985536669380963,
      "epoch": 5.7363510131835936e-05,
      "step": 60150
    },
    {
      "epoch": 5.7363510131835936e-05,
      "step": 60150,
      "training_step_time": 7.45025110244751
    },
    {
      "epoch": 5.736446380615235e-05,
      "model_forward_time": 1.7396166324615479,
      "step": 60151
    },
    {
      "epoch": 5.736446380615235e-05,
      "step": 60151,
      "training_step_time": 7.576300382614136
    },
    {
      "epoch": 5.736541748046875e-05,
      "model_forward_time": 1.7464759349822998,
      "step": 60152
    },
    {
      "epoch": 5.736541748046875e-05,
      "step": 60152,
      "training_step_time": 8.532111644744873
    },
    {
      "epoch": 5.7366371154785155e-05,
      "model_forward_time": 2.8940320014953613,
      "step": 60153
    },
    {
      "epoch": 5.7366371154785155e-05,
      "step": 60153,
      "training_step_time": 9.280817747116089
    },
    {
      "epoch": 5.7367324829101566e-05,
      "model_forward_time": 1.7503814697265625,
      "step": 60154
    },
    {
      "epoch": 5.7367324829101566e-05,
      "step": 60154,
      "training_step_time": 7.761176586151123
    },
    {
      "epoch": 5.736827850341797e-05,
      "model_forward_time": 1.7440054416656494,
      "step": 60155
    },
    {
      "epoch": 5.736827850341797e-05,
      "step": 60155,
      "training_step_time": 9.298343896865845
    },
    {
      "epoch": 5.7369232177734374e-05,
      "model_forward_time": 1.747307538986206,
      "step": 60156
    },
    {
      "epoch": 5.7369232177734374e-05,
      "step": 60156,
      "training_step_time": 8.42819881439209
    },
    {
      "epoch": 5.737018585205078e-05,
      "model_forward_time": 1.748741865158081,
      "step": 60157
    },
    {
      "epoch": 5.737018585205078e-05,
      "step": 60157,
      "training_step_time": 7.547642469406128
    },
    {
      "epoch": 5.737113952636719e-05,
      "model_forward_time": 1.7457256317138672,
      "step": 60158
    },
    {
      "epoch": 5.737113952636719e-05,
      "step": 60158,
      "training_step_time": 6.715615272521973
    },
    {
      "epoch": 5.737209320068359e-05,
      "model_forward_time": 1.7377197742462158,
      "step": 60159
    },
    {
      "epoch": 5.737209320068359e-05,
      "step": 60159,
      "training_step_time": 6.773711204528809
    },
    {
      "epoch": 5.7373046875e-05,
      "grad_norm": 0.12641644477844238,
      "learning_rate": 3.7471039510978163e-06,
      "loss": 0.0553,
      "step": 60160
    },
    {
      "epoch": 5.7373046875e-05,
      "model_forward_time": 1.7445943355560303,
      "step": 60160
    },
    {
      "dynamics_loss_avg": 0.056585725490003826,
      "epoch": 5.7373046875e-05,
      "step": 60160
    },
    {
      "action_loss_avg": 0.0014909724006429315,
      "epoch": 5.7373046875e-05,
      "step": 60160
    },
    {
      "epoch": 5.7373046875e-05,
      "step": 60160,
      "training_step_time": 8.384183883666992
    },
    {
      "epoch": 5.737400054931641e-05,
      "model_forward_time": 1.7415378093719482,
      "step": 60161
    },
    {
      "epoch": 5.737400054931641e-05,
      "step": 60161,
      "training_step_time": 7.594686508178711
    },
    {
      "epoch": 5.737495422363281e-05,
      "model_forward_time": 2.9220330715179443,
      "step": 60162
    },
    {
      "epoch": 5.737495422363281e-05,
      "step": 60162,
      "training_step_time": 10.327415227890015
    },
    {
      "epoch": 5.7375907897949216e-05,
      "model_forward_time": 1.7426292896270752,
      "step": 60163
    },
    {
      "epoch": 5.7375907897949216e-05,
      "step": 60163,
      "training_step_time": 8.360763788223267
    },
    {
      "epoch": 5.7376861572265626e-05,
      "model_forward_time": 1.7485873699188232,
      "step": 60164
    },
    {
      "epoch": 5.7376861572265626e-05,
      "step": 60164,
      "training_step_time": 8.466009616851807
    },
    {
      "epoch": 5.737781524658203e-05,
      "model_forward_time": 1.743295669555664,
      "step": 60165
    },
    {
      "epoch": 5.737781524658203e-05,
      "step": 60165,
      "training_step_time": 7.630640506744385
    },
    {
      "epoch": 5.7378768920898434e-05,
      "model_forward_time": 1.7409038543701172,
      "step": 60166
    },
    {
      "epoch": 5.7378768920898434e-05,
      "step": 60166,
      "training_step_time": 7.4894187450408936
    },
    {
      "epoch": 5.7379722595214845e-05,
      "model_forward_time": 1.7450098991394043,
      "step": 60167
    },
    {
      "epoch": 5.7379722595214845e-05,
      "step": 60167,
      "training_step_time": 7.306296348571777
    },
    {
      "epoch": 5.738067626953125e-05,
      "model_forward_time": 1.7405734062194824,
      "step": 60168
    },
    {
      "epoch": 5.738067626953125e-05,
      "step": 60168,
      "training_step_time": 6.72884202003479
    },
    {
      "epoch": 5.738162994384765e-05,
      "model_forward_time": 1.8563249111175537,
      "step": 60169
    },
    {
      "epoch": 5.738162994384765e-05,
      "step": 60169,
      "training_step_time": 7.652307748794556
    },
    {
      "epoch": 5.7382583618164064e-05,
      "grad_norm": 0.09883227199316025,
      "learning_rate": 3.7455033019517685e-06,
      "loss": 0.0556,
      "step": 60170
    },
    {
      "epoch": 5.7382583618164064e-05,
      "model_forward_time": 1.746328353881836,
      "step": 60170
    },
    {
      "dynamics_loss_avg": 0.04440125692635775,
      "epoch": 5.7382583618164064e-05,
      "step": 60170
    },
    {
      "action_loss_avg": 0.0014163983811158686,
      "epoch": 5.7382583618164064e-05,
      "step": 60170
    },
    {
      "epoch": 5.7382583618164064e-05,
      "step": 60170,
      "training_step_time": 8.399080038070679
    },
    {
      "epoch": 5.738353729248047e-05,
      "model_forward_time": 2.301948308944702,
      "step": 60171
    },
    {
      "epoch": 5.738353729248047e-05,
      "step": 60171,
      "training_step_time": 7.717155694961548
    },
    {
      "epoch": 5.738449096679687e-05,
      "model_forward_time": 1.7406375408172607,
      "step": 60172
    },
    {
      "epoch": 5.738449096679687e-05,
      "step": 60172,
      "training_step_time": 8.22288179397583
    },
    {
      "epoch": 5.738544464111328e-05,
      "model_forward_time": 1.7415449619293213,
      "step": 60173
    },
    {
      "epoch": 5.738544464111328e-05,
      "step": 60173,
      "training_step_time": 7.608961582183838
    },
    {
      "epoch": 5.738639831542969e-05,
      "model_forward_time": 1.7445662021636963,
      "step": 60174
    },
    {
      "epoch": 5.738639831542969e-05,
      "step": 60174,
      "training_step_time": 10.183220148086548
    },
    {
      "epoch": 5.738735198974609e-05,
      "model_forward_time": 1.7429583072662354,
      "step": 60175
    },
    {
      "epoch": 5.738735198974609e-05,
      "step": 60175,
      "training_step_time": 7.583459377288818
    },
    {
      "epoch": 5.73883056640625e-05,
      "model_forward_time": 1.742617130279541,
      "step": 60176
    },
    {
      "epoch": 5.73883056640625e-05,
      "step": 60176,
      "training_step_time": 7.407027959823608
    },
    {
      "epoch": 5.7389259338378906e-05,
      "model_forward_time": 1.7418417930603027,
      "step": 60177
    },
    {
      "epoch": 5.7389259338378906e-05,
      "step": 60177,
      "training_step_time": 6.720991611480713
    },
    {
      "epoch": 5.739021301269531e-05,
      "model_forward_time": 1.7430105209350586,
      "step": 60178
    },
    {
      "epoch": 5.739021301269531e-05,
      "step": 60178,
      "training_step_time": 7.686740875244141
    },
    {
      "epoch": 5.739116668701172e-05,
      "model_forward_time": 1.709742784500122,
      "step": 60179
    },
    {
      "epoch": 5.739116668701172e-05,
      "step": 60179,
      "training_step_time": 7.485341310501099
    },
    {
      "epoch": 5.7392120361328125e-05,
      "grad_norm": 0.11093991249799728,
      "learning_rate": 3.7439027899955916e-06,
      "loss": 0.0557,
      "step": 60180
    },
    {
      "epoch": 5.7392120361328125e-05,
      "model_forward_time": 2.287868022918701,
      "step": 60180
    },
    {
      "dynamics_loss_avg": 0.057541387341916564,
      "epoch": 5.7392120361328125e-05,
      "step": 60180
    },
    {
      "action_loss_avg": 0.0019038933212868869,
      "epoch": 5.7392120361328125e-05,
      "step": 60180
    },
    {
      "epoch": 5.7392120361328125e-05,
      "step": 60180,
      "training_step_time": 7.768084764480591
    },
    {
      "epoch": 5.739307403564453e-05,
      "model_forward_time": 1.3666880130767822,
      "step": 60181
    },
    {
      "epoch": 5.739307403564453e-05,
      "step": 60181,
      "training_step_time": 8.666041851043701
    },
    {
      "epoch": 5.739402770996094e-05,
      "model_forward_time": 1.7251474857330322,
      "step": 60182
    },
    {
      "epoch": 5.739402770996094e-05,
      "step": 60182,
      "training_step_time": 9.824790477752686
    },
    {
      "epoch": 5.7394981384277344e-05,
      "model_forward_time": 1.7706096172332764,
      "step": 60183
    },
    {
      "epoch": 5.7394981384277344e-05,
      "step": 60183,
      "training_step_time": 7.618230819702148
    },
    {
      "epoch": 5.739593505859375e-05,
      "model_forward_time": 1.7265031337738037,
      "step": 60184
    },
    {
      "epoch": 5.739593505859375e-05,
      "step": 60184,
      "training_step_time": 7.741736650466919
    },
    {
      "epoch": 5.739688873291016e-05,
      "model_forward_time": 1.7116694450378418,
      "step": 60185
    },
    {
      "epoch": 5.739688873291016e-05,
      "step": 60185,
      "training_step_time": 7.417467355728149
    },
    {
      "epoch": 5.739784240722656e-05,
      "model_forward_time": 1.7117207050323486,
      "step": 60186
    },
    {
      "epoch": 5.739784240722656e-05,
      "step": 60186,
      "training_step_time": 6.744218826293945
    },
    {
      "epoch": 5.7398796081542966e-05,
      "model_forward_time": 1.7192931175231934,
      "step": 60187
    },
    {
      "epoch": 5.7398796081542966e-05,
      "step": 60187,
      "training_step_time": 7.433943510055542
    },
    {
      "epoch": 5.739974975585938e-05,
      "model_forward_time": 1.7253148555755615,
      "step": 60188
    },
    {
      "epoch": 5.739974975585938e-05,
      "step": 60188,
      "training_step_time": 7.7133519649505615
    },
    {
      "epoch": 5.740070343017578e-05,
      "model_forward_time": 2.2704122066497803,
      "step": 60189
    },
    {
      "epoch": 5.740070343017578e-05,
      "step": 60189,
      "training_step_time": 9.218958854675293
    },
    {
      "epoch": 5.7401657104492185e-05,
      "grad_norm": 0.10529132932424545,
      "learning_rate": 3.7423024154043186e-06,
      "loss": 0.0545,
      "step": 60190
    },
    {
      "epoch": 5.7401657104492185e-05,
      "model_forward_time": 1.7208216190338135,
      "step": 60190
    },
    {
      "dynamics_loss_avg": 0.05907844062894583,
      "epoch": 5.7401657104492185e-05,
      "step": 60190
    },
    {
      "action_loss_avg": 0.0019977478426881135,
      "epoch": 5.7401657104492185e-05,
      "step": 60190
    },
    {
      "epoch": 5.7401657104492185e-05,
      "step": 60190,
      "training_step_time": 8.153527021408081
    },
    {
      "epoch": 5.7402610778808596e-05,
      "model_forward_time": 1.7142579555511475,
      "step": 60191
    },
    {
      "epoch": 5.7402610778808596e-05,
      "step": 60191,
      "training_step_time": 7.71796178817749
    },
    {
      "epoch": 5.7403564453125e-05,
      "model_forward_time": 1.777660846710205,
      "step": 60192
    },
    {
      "epoch": 5.7403564453125e-05,
      "step": 60192,
      "training_step_time": 8.730899333953857
    },
    {
      "epoch": 5.7404518127441404e-05,
      "model_forward_time": 1.7244863510131836,
      "step": 60193
    },
    {
      "epoch": 5.7404518127441404e-05,
      "step": 60193,
      "training_step_time": 7.7749669551849365
    },
    {
      "epoch": 5.7405471801757815e-05,
      "model_forward_time": 1.7318229675292969,
      "step": 60194
    },
    {
      "epoch": 5.7405471801757815e-05,
      "step": 60194,
      "training_step_time": 7.062817573547363
    },
    {
      "epoch": 5.740642547607422e-05,
      "model_forward_time": 1.7154819965362549,
      "step": 60195
    },
    {
      "epoch": 5.740642547607422e-05,
      "step": 60195,
      "training_step_time": 7.3524861335754395
    },
    {
      "epoch": 5.740737915039062e-05,
      "model_forward_time": 1.7245845794677734,
      "step": 60196
    },
    {
      "epoch": 5.740737915039062e-05,
      "step": 60196,
      "training_step_time": 7.4330737590789795
    },
    {
      "epoch": 5.7408332824707034e-05,
      "model_forward_time": 1.7184832096099854,
      "step": 60197
    },
    {
      "epoch": 5.7408332824707034e-05,
      "step": 60197,
      "training_step_time": 7.645344257354736
    },
    {
      "epoch": 5.740928649902344e-05,
      "model_forward_time": 2.2639319896698,
      "step": 60198
    },
    {
      "epoch": 5.740928649902344e-05,
      "step": 60198,
      "training_step_time": 7.705425500869751
    },
    {
      "epoch": 5.741024017333984e-05,
      "model_forward_time": 1.7189109325408936,
      "step": 60199
    },
    {
      "epoch": 5.741024017333984e-05,
      "step": 60199,
      "training_step_time": 8.52297043800354
    },
    {
      "epoch": 5.741119384765625e-05,
      "grad_norm": 0.11126178503036499,
      "learning_rate": 3.740702178352963e-06,
      "loss": 0.0577,
      "step": 60200
    },
    {
      "epoch": 5.741119384765625e-05,
      "model_forward_time": 1.7138292789459229,
      "step": 60200
    },
    {
      "dynamics_loss_avg": 0.04827164616435766,
      "epoch": 5.741119384765625e-05,
      "step": 60200
    },
    {
      "action_loss_avg": 0.0012327854929026216,
      "epoch": 5.741119384765625e-05,
      "step": 60200
    },
    {
      "epoch": 5.741119384765625e-05,
      "step": 60200,
      "training_step_time": 10.284950733184814
    },
    {
      "epoch": 5.741214752197266e-05,
      "model_forward_time": 1.757047414779663,
      "step": 60201
    },
    {
      "epoch": 5.741214752197266e-05,
      "step": 60201,
      "training_step_time": 8.730067014694214
    },
    {
      "epoch": 5.741310119628906e-05,
      "model_forward_time": 1.7193794250488281,
      "step": 60202
    },
    {
      "epoch": 5.741310119628906e-05,
      "step": 60202,
      "training_step_time": 7.71022629737854
    },
    {
      "epoch": 5.741405487060547e-05,
      "model_forward_time": 1.7324228286743164,
      "step": 60203
    },
    {
      "epoch": 5.741405487060547e-05,
      "step": 60203,
      "training_step_time": 7.509769439697266
    },
    {
      "epoch": 5.7415008544921876e-05,
      "model_forward_time": 1.7256407737731934,
      "step": 60204
    },
    {
      "epoch": 5.7415008544921876e-05,
      "step": 60204,
      "training_step_time": 7.349359750747681
    },
    {
      "epoch": 5.741596221923828e-05,
      "model_forward_time": 1.7123627662658691,
      "step": 60205
    },
    {
      "epoch": 5.741596221923828e-05,
      "step": 60205,
      "training_step_time": 7.507050037384033
    },
    {
      "epoch": 5.741691589355469e-05,
      "model_forward_time": 1.7230870723724365,
      "step": 60206
    },
    {
      "epoch": 5.741691589355469e-05,
      "step": 60206,
      "training_step_time": 7.754451751708984
    },
    {
      "epoch": 5.7417869567871094e-05,
      "model_forward_time": 2.318635940551758,
      "step": 60207
    },
    {
      "epoch": 5.7417869567871094e-05,
      "step": 60207,
      "training_step_time": 7.761034250259399
    },
    {
      "epoch": 5.74188232421875e-05,
      "model_forward_time": 1.7211298942565918,
      "step": 60208
    },
    {
      "epoch": 5.74188232421875e-05,
      "step": 60208,
      "training_step_time": 8.788309812545776
    },
    {
      "epoch": 5.741977691650391e-05,
      "model_forward_time": 1.7342298030853271,
      "step": 60209
    },
    {
      "epoch": 5.741977691650391e-05,
      "step": 60209,
      "training_step_time": 9.0172438621521
    },
    {
      "epoch": 5.742073059082031e-05,
      "grad_norm": 0.08534462004899979,
      "learning_rate": 3.7391020790165234e-06,
      "loss": 0.0609,
      "step": 60210
    },
    {
      "epoch": 5.742073059082031e-05,
      "model_forward_time": 1.7117154598236084,
      "step": 60210
    },
    {
      "dynamics_loss_avg": 0.04767340067774058,
      "epoch": 5.742073059082031e-05,
      "step": 60210
    },
    {
      "action_loss_avg": 0.0011136936373077333,
      "epoch": 5.742073059082031e-05,
      "step": 60210
    },
    {
      "epoch": 5.742073059082031e-05,
      "step": 60210,
      "training_step_time": 8.341203689575195
    },
    {
      "epoch": 5.742168426513672e-05,
      "model_forward_time": 1.7803542613983154,
      "step": 60211
    },
    {
      "epoch": 5.742168426513672e-05,
      "step": 60211,
      "training_step_time": 8.70267391204834
    },
    {
      "epoch": 5.742263793945313e-05,
      "model_forward_time": 1.7294514179229736,
      "step": 60212
    },
    {
      "epoch": 5.742263793945313e-05,
      "step": 60212,
      "training_step_time": 7.518748044967651
    },
    {
      "epoch": 5.742359161376953e-05,
      "model_forward_time": 1.7195935249328613,
      "step": 60213
    },
    {
      "epoch": 5.742359161376953e-05,
      "step": 60213,
      "training_step_time": 7.41883659362793
    },
    {
      "epoch": 5.7424545288085936e-05,
      "model_forward_time": 1.7238197326660156,
      "step": 60214
    },
    {
      "epoch": 5.7424545288085936e-05,
      "step": 60214,
      "training_step_time": 7.435117959976196
    },
    {
      "epoch": 5.742549896240235e-05,
      "model_forward_time": 1.7285799980163574,
      "step": 60215
    },
    {
      "epoch": 5.742549896240235e-05,
      "step": 60215,
      "training_step_time": 8.338983058929443
    },
    {
      "epoch": 5.742645263671875e-05,
      "model_forward_time": 2.242156744003296,
      "step": 60216
    },
    {
      "epoch": 5.742645263671875e-05,
      "step": 60216,
      "training_step_time": 7.854366064071655
    },
    {
      "epoch": 5.7427406311035155e-05,
      "model_forward_time": 1.711961269378662,
      "step": 60217
    },
    {
      "epoch": 5.7427406311035155e-05,
      "step": 60217,
      "training_step_time": 8.313836812973022
    },
    {
      "epoch": 5.7428359985351566e-05,
      "model_forward_time": 1.3501648902893066,
      "step": 60218
    },
    {
      "epoch": 5.7428359985351566e-05,
      "step": 60218,
      "training_step_time": 9.829854249954224
    },
    {
      "epoch": 5.742931365966797e-05,
      "model_forward_time": 1.7092292308807373,
      "step": 60219
    },
    {
      "epoch": 5.742931365966797e-05,
      "step": 60219,
      "training_step_time": 9.097800016403198
    },
    {
      "epoch": 5.7430267333984374e-05,
      "grad_norm": 0.14127837121486664,
      "learning_rate": 3.7375021175699847e-06,
      "loss": 0.0602,
      "step": 60220
    },
    {
      "epoch": 5.7430267333984374e-05,
      "model_forward_time": 1.766650676727295,
      "step": 60220
    },
    {
      "dynamics_loss_avg": 0.05349302999675274,
      "epoch": 5.7430267333984374e-05,
      "step": 60220
    },
    {
      "action_loss_avg": 0.001991152059054002,
      "epoch": 5.7430267333984374e-05,
      "step": 60220
    },
    {
      "epoch": 5.7430267333984374e-05,
      "step": 60220,
      "training_step_time": 7.6004393100738525
    },
    {
      "epoch": 5.743122100830078e-05,
      "model_forward_time": 1.7051141262054443,
      "step": 60221
    },
    {
      "epoch": 5.743122100830078e-05,
      "step": 60221,
      "training_step_time": 7.637973308563232
    },
    {
      "epoch": 5.743217468261719e-05,
      "model_forward_time": 1.7099382877349854,
      "step": 60222
    },
    {
      "epoch": 5.743217468261719e-05,
      "step": 60222,
      "training_step_time": 7.432407855987549
    },
    {
      "epoch": 5.743312835693359e-05,
      "model_forward_time": 1.7241148948669434,
      "step": 60223
    },
    {
      "epoch": 5.743312835693359e-05,
      "step": 60223,
      "training_step_time": 7.411776304244995
    },
    {
      "epoch": 5.743408203125e-05,
      "model_forward_time": 1.7093555927276611,
      "step": 60224
    },
    {
      "epoch": 5.743408203125e-05,
      "step": 60224,
      "training_step_time": 7.580630779266357
    },
    {
      "epoch": 5.743503570556641e-05,
      "model_forward_time": 1.7053568363189697,
      "step": 60225
    },
    {
      "epoch": 5.743503570556641e-05,
      "step": 60225,
      "training_step_time": 7.796283960342407
    },
    {
      "epoch": 5.743598937988281e-05,
      "model_forward_time": 2.253736734390259,
      "step": 60226
    },
    {
      "epoch": 5.743598937988281e-05,
      "step": 60226,
      "training_step_time": 7.698186635971069
    },
    {
      "epoch": 5.7436943054199216e-05,
      "model_forward_time": 1.7079946994781494,
      "step": 60227
    },
    {
      "epoch": 5.7436943054199216e-05,
      "step": 60227,
      "training_step_time": 7.546554327011108
    },
    {
      "epoch": 5.7437896728515626e-05,
      "model_forward_time": 1.707200050354004,
      "step": 60228
    },
    {
      "epoch": 5.7437896728515626e-05,
      "step": 60228,
      "training_step_time": 7.913908958435059
    },
    {
      "epoch": 5.743885040283203e-05,
      "model_forward_time": 1.7438383102416992,
      "step": 60229
    },
    {
      "epoch": 5.743885040283203e-05,
      "step": 60229,
      "training_step_time": 7.804016828536987
    },
    {
      "epoch": 5.7439804077148435e-05,
      "grad_norm": 0.12807577848434448,
      "learning_rate": 3.7359022941883168e-06,
      "loss": 0.0589,
      "step": 60230
    },
    {
      "epoch": 5.7439804077148435e-05,
      "model_forward_time": 1.7042715549468994,
      "step": 60230
    },
    {
      "dynamics_loss_avg": 0.057447954453527926,
      "epoch": 5.7439804077148435e-05,
      "step": 60230
    },
    {
      "action_loss_avg": 0.0017618384677916764,
      "epoch": 5.7439804077148435e-05,
      "step": 60230
    },
    {
      "epoch": 5.7439804077148435e-05,
      "step": 60230,
      "training_step_time": 7.755589246749878
    },
    {
      "epoch": 5.7440757751464845e-05,
      "model_forward_time": 1.7057318687438965,
      "step": 60231
    },
    {
      "epoch": 5.7440757751464845e-05,
      "step": 60231,
      "training_step_time": 7.150381565093994
    },
    {
      "epoch": 5.744171142578125e-05,
      "model_forward_time": 1.3459594249725342,
      "step": 60232
    },
    {
      "epoch": 5.744171142578125e-05,
      "step": 60232,
      "training_step_time": 7.494778156280518
    },
    {
      "epoch": 5.744266510009765e-05,
      "model_forward_time": 1.7129974365234375,
      "step": 60233
    },
    {
      "epoch": 5.744266510009765e-05,
      "step": 60233,
      "training_step_time": 7.629109621047974
    },
    {
      "epoch": 5.7443618774414064e-05,
      "model_forward_time": 1.7048945426940918,
      "step": 60234
    },
    {
      "epoch": 5.7443618774414064e-05,
      "step": 60234,
      "training_step_time": 7.850674152374268
    },
    {
      "epoch": 5.744457244873047e-05,
      "model_forward_time": 2.2577247619628906,
      "step": 60235
    },
    {
      "epoch": 5.744457244873047e-05,
      "step": 60235,
      "training_step_time": 8.968481302261353
    },
    {
      "epoch": 5.744552612304687e-05,
      "model_forward_time": 1.7117583751678467,
      "step": 60236
    },
    {
      "epoch": 5.744552612304687e-05,
      "step": 60236,
      "training_step_time": 8.36244797706604
    },
    {
      "epoch": 5.744647979736328e-05,
      "model_forward_time": 1.723161220550537,
      "step": 60237
    },
    {
      "epoch": 5.744647979736328e-05,
      "step": 60237,
      "training_step_time": 8.729536533355713
    },
    {
      "epoch": 5.744743347167969e-05,
      "model_forward_time": 1.7055459022521973,
      "step": 60238
    },
    {
      "epoch": 5.744743347167969e-05,
      "step": 60238,
      "training_step_time": 8.377704381942749
    },
    {
      "epoch": 5.744838714599609e-05,
      "model_forward_time": 1.7041821479797363,
      "step": 60239
    },
    {
      "epoch": 5.744838714599609e-05,
      "step": 60239,
      "training_step_time": 7.521006345748901
    },
    {
      "epoch": 5.74493408203125e-05,
      "grad_norm": 0.12136300653219223,
      "learning_rate": 3.734302609046473e-06,
      "loss": 0.0562,
      "step": 60240
    },
    {
      "epoch": 5.74493408203125e-05,
      "model_forward_time": 1.7087457180023193,
      "step": 60240
    },
    {
      "dynamics_loss_avg": 0.0503611059859395,
      "epoch": 5.74493408203125e-05,
      "step": 60240
    },
    {
      "action_loss_avg": 0.0016106152324937284,
      "epoch": 5.74493408203125e-05,
      "step": 60240
    },
    {
      "epoch": 5.74493408203125e-05,
      "step": 60240,
      "training_step_time": 7.523211240768433
    },
    {
      "epoch": 5.7450294494628906e-05,
      "model_forward_time": 1.7091054916381836,
      "step": 60241
    },
    {
      "epoch": 5.7450294494628906e-05,
      "step": 60241,
      "training_step_time": 7.101267576217651
    },
    {
      "epoch": 5.745124816894531e-05,
      "model_forward_time": 1.7137556076049805,
      "step": 60242
    },
    {
      "epoch": 5.745124816894531e-05,
      "step": 60242,
      "training_step_time": 7.615662574768066
    },
    {
      "epoch": 5.745220184326172e-05,
      "model_forward_time": 1.7096798419952393,
      "step": 60243
    },
    {
      "epoch": 5.745220184326172e-05,
      "step": 60243,
      "training_step_time": 7.871344566345215
    },
    {
      "epoch": 5.7453155517578125e-05,
      "model_forward_time": 2.2751784324645996,
      "step": 60244
    },
    {
      "epoch": 5.7453155517578125e-05,
      "step": 60244,
      "training_step_time": 7.677910327911377
    },
    {
      "epoch": 5.745410919189453e-05,
      "model_forward_time": 1.7041006088256836,
      "step": 60245
    },
    {
      "epoch": 5.745410919189453e-05,
      "step": 60245,
      "training_step_time": 9.346771001815796
    },
    {
      "epoch": 5.745506286621094e-05,
      "model_forward_time": 1.7133326530456543,
      "step": 60246
    },
    {
      "epoch": 5.745506286621094e-05,
      "step": 60246,
      "training_step_time": 8.606624364852905
    },
    {
      "epoch": 5.7456016540527344e-05,
      "model_forward_time": 1.71730637550354,
      "step": 60247
    },
    {
      "epoch": 5.7456016540527344e-05,
      "step": 60247,
      "training_step_time": 7.697311639785767
    },
    {
      "epoch": 5.745697021484375e-05,
      "model_forward_time": 1.7700283527374268,
      "step": 60248
    },
    {
      "epoch": 5.745697021484375e-05,
      "step": 60248,
      "training_step_time": 7.666410207748413
    },
    {
      "epoch": 5.745792388916016e-05,
      "model_forward_time": 1.7102432250976562,
      "step": 60249
    },
    {
      "epoch": 5.745792388916016e-05,
      "step": 60249,
      "training_step_time": 7.54084038734436
    },
    {
      "epoch": 5.745887756347656e-05,
      "grad_norm": 0.09489156305789948,
      "learning_rate": 3.732703062319391e-06,
      "loss": 0.0561,
      "step": 60250
    },
    {
      "epoch": 5.745887756347656e-05,
      "model_forward_time": 1.7117161750793457,
      "step": 60250
    },
    {
      "dynamics_loss_avg": 0.04980780128389597,
      "epoch": 5.745887756347656e-05,
      "step": 60250
    },
    {
      "action_loss_avg": 0.0014917428401531652,
      "epoch": 5.745887756347656e-05,
      "step": 60250
    },
    {
      "epoch": 5.745887756347656e-05,
      "step": 60250,
      "training_step_time": 7.436202526092529
    },
    {
      "epoch": 5.7459831237792967e-05,
      "model_forward_time": 1.7113301753997803,
      "step": 60251
    },
    {
      "epoch": 5.7459831237792967e-05,
      "step": 60251,
      "training_step_time": 7.493869781494141
    },
    {
      "epoch": 5.746078491210938e-05,
      "model_forward_time": 1.7142596244812012,
      "step": 60252
    },
    {
      "epoch": 5.746078491210938e-05,
      "step": 60252,
      "training_step_time": 8.702139139175415
    },
    {
      "epoch": 5.746173858642578e-05,
      "model_forward_time": 2.2410686016082764,
      "step": 60253
    },
    {
      "epoch": 5.746173858642578e-05,
      "step": 60253,
      "training_step_time": 8.712663650512695
    },
    {
      "epoch": 5.7462692260742185e-05,
      "model_forward_time": 1.7083370685577393,
      "step": 60254
    },
    {
      "epoch": 5.7462692260742185e-05,
      "step": 60254,
      "training_step_time": 8.926382541656494
    },
    {
      "epoch": 5.7463645935058596e-05,
      "model_forward_time": 1.7085976600646973,
      "step": 60255
    },
    {
      "epoch": 5.7463645935058596e-05,
      "step": 60255,
      "training_step_time": 7.668374300003052
    },
    {
      "epoch": 5.7464599609375e-05,
      "model_forward_time": 1.7177209854125977,
      "step": 60256
    },
    {
      "epoch": 5.7464599609375e-05,
      "step": 60256,
      "training_step_time": 9.169347047805786
    },
    {
      "epoch": 5.7465553283691404e-05,
      "model_forward_time": 1.7711989879608154,
      "step": 60257
    },
    {
      "epoch": 5.7465553283691404e-05,
      "step": 60257,
      "training_step_time": 7.6259684562683105
    },
    {
      "epoch": 5.7466506958007815e-05,
      "model_forward_time": 1.713667392730713,
      "step": 60258
    },
    {
      "epoch": 5.7466506958007815e-05,
      "step": 60258,
      "training_step_time": 7.5359485149383545
    },
    {
      "epoch": 5.746746063232422e-05,
      "model_forward_time": 1.7142505645751953,
      "step": 60259
    },
    {
      "epoch": 5.746746063232422e-05,
      "step": 60259,
      "training_step_time": 7.137270927429199
    },
    {
      "epoch": 5.746841430664062e-05,
      "grad_norm": 0.07654532790184021,
      "learning_rate": 3.731103654181999e-06,
      "loss": 0.0583,
      "step": 60260
    },
    {
      "epoch": 5.746841430664062e-05,
      "model_forward_time": 1.7058570384979248,
      "step": 60260
    },
    {
      "dynamics_loss_avg": 0.07492345534265041,
      "epoch": 5.746841430664062e-05,
      "step": 60260
    },
    {
      "action_loss_avg": 0.0024072430212982,
      "epoch": 5.746841430664062e-05,
      "step": 60260
    },
    {
      "epoch": 5.746841430664062e-05,
      "step": 60260,
      "training_step_time": 7.427014350891113
    },
    {
      "epoch": 5.7469367980957034e-05,
      "model_forward_time": 1.7019636631011963,
      "step": 60261
    },
    {
      "epoch": 5.7469367980957034e-05,
      "step": 60261,
      "training_step_time": 7.772132873535156
    },
    {
      "epoch": 5.747032165527344e-05,
      "model_forward_time": 1.7092502117156982,
      "step": 60262
    },
    {
      "epoch": 5.747032165527344e-05,
      "step": 60262,
      "training_step_time": 8.492316007614136
    },
    {
      "epoch": 5.747127532958984e-05,
      "model_forward_time": 1.8766632080078125,
      "step": 60263
    },
    {
      "epoch": 5.747127532958984e-05,
      "step": 60263,
      "training_step_time": 8.33534574508667
    },
    {
      "epoch": 5.747222900390625e-05,
      "model_forward_time": 1.3386647701263428,
      "step": 60264
    },
    {
      "epoch": 5.747222900390625e-05,
      "step": 60264,
      "training_step_time": 7.887439727783203
    },
    {
      "epoch": 5.747318267822266e-05,
      "model_forward_time": 1.7131779193878174,
      "step": 60265
    },
    {
      "epoch": 5.747318267822266e-05,
      "step": 60265,
      "training_step_time": 7.680492877960205
    },
    {
      "epoch": 5.747413635253906e-05,
      "model_forward_time": 1.766721487045288,
      "step": 60266
    },
    {
      "epoch": 5.747413635253906e-05,
      "step": 60266,
      "training_step_time": 9.683342933654785
    },
    {
      "epoch": 5.747509002685547e-05,
      "model_forward_time": 1.6977715492248535,
      "step": 60267
    },
    {
      "epoch": 5.747509002685547e-05,
      "step": 60267,
      "training_step_time": 8.532311916351318
    },
    {
      "epoch": 5.7476043701171876e-05,
      "model_forward_time": 1.7115352153778076,
      "step": 60268
    },
    {
      "epoch": 5.7476043701171876e-05,
      "step": 60268,
      "training_step_time": 7.532160043716431
    },
    {
      "epoch": 5.747699737548828e-05,
      "model_forward_time": 1.7286429405212402,
      "step": 60269
    },
    {
      "epoch": 5.747699737548828e-05,
      "step": 60269,
      "training_step_time": 7.4368391036987305
    },
    {
      "epoch": 5.747795104980469e-05,
      "grad_norm": 0.09911668300628662,
      "learning_rate": 3.7295043848092035e-06,
      "loss": 0.0583,
      "step": 60270
    },
    {
      "epoch": 5.747795104980469e-05,
      "model_forward_time": 1.7170584201812744,
      "step": 60270
    },
    {
      "dynamics_loss_avg": 0.05860590441152454,
      "epoch": 5.747795104980469e-05,
      "step": 60270
    },
    {
      "action_loss_avg": 0.0019110629189526662,
      "epoch": 5.747795104980469e-05,
      "step": 60270
    },
    {
      "epoch": 5.747795104980469e-05,
      "step": 60270,
      "training_step_time": 7.539548873901367
    },
    {
      "epoch": 5.7478904724121095e-05,
      "model_forward_time": 1.713273048400879,
      "step": 60271
    },
    {
      "epoch": 5.7478904724121095e-05,
      "step": 60271,
      "training_step_time": 7.829841613769531
    },
    {
      "epoch": 5.74798583984375e-05,
      "model_forward_time": 2.2907350063323975,
      "step": 60272
    },
    {
      "epoch": 5.74798583984375e-05,
      "step": 60272,
      "training_step_time": 7.576266050338745
    },
    {
      "epoch": 5.748081207275391e-05,
      "model_forward_time": 1.7132134437561035,
      "step": 60273
    },
    {
      "epoch": 5.748081207275391e-05,
      "step": 60273,
      "training_step_time": 7.526491165161133
    },
    {
      "epoch": 5.748176574707031e-05,
      "model_forward_time": 1.7081429958343506,
      "step": 60274
    },
    {
      "epoch": 5.748176574707031e-05,
      "step": 60274,
      "training_step_time": 8.696703910827637
    },
    {
      "epoch": 5.748271942138672e-05,
      "model_forward_time": 1.742577075958252,
      "step": 60275
    },
    {
      "epoch": 5.748271942138672e-05,
      "step": 60275,
      "training_step_time": 8.573508262634277
    },
    {
      "epoch": 5.748367309570313e-05,
      "model_forward_time": 1.7087037563323975,
      "step": 60276
    },
    {
      "epoch": 5.748367309570313e-05,
      "step": 60276,
      "training_step_time": 7.582894563674927
    },
    {
      "epoch": 5.748462677001953e-05,
      "model_forward_time": 1.711951494216919,
      "step": 60277
    },
    {
      "epoch": 5.748462677001953e-05,
      "step": 60277,
      "training_step_time": 7.554020166397095
    },
    {
      "epoch": 5.7485580444335936e-05,
      "model_forward_time": 1.7151272296905518,
      "step": 60278
    },
    {
      "epoch": 5.7485580444335936e-05,
      "step": 60278,
      "training_step_time": 8.236884832382202
    },
    {
      "epoch": 5.748653411865235e-05,
      "model_forward_time": 1.7044517993927002,
      "step": 60279
    },
    {
      "epoch": 5.748653411865235e-05,
      "step": 60279,
      "training_step_time": 7.567013502120972
    },
    {
      "epoch": 5.748748779296875e-05,
      "grad_norm": 0.12660667300224304,
      "learning_rate": 3.727905254375898e-06,
      "loss": 0.0588,
      "step": 60280
    },
    {
      "epoch": 5.748748779296875e-05,
      "model_forward_time": 1.703129768371582,
      "step": 60280
    },
    {
      "dynamics_loss_avg": 0.06143013443797827,
      "epoch": 5.748748779296875e-05,
      "step": 60280
    },
    {
      "action_loss_avg": 0.0016428429808001965,
      "epoch": 5.748748779296875e-05,
      "step": 60280
    },
    {
      "epoch": 5.748748779296875e-05,
      "step": 60280,
      "training_step_time": 7.756346940994263
    },
    {
      "epoch": 5.7488441467285155e-05,
      "model_forward_time": 2.258474111557007,
      "step": 60281
    },
    {
      "epoch": 5.7488441467285155e-05,
      "step": 60281,
      "training_step_time": 7.610880374908447
    },
    {
      "epoch": 5.748939514160156e-05,
      "model_forward_time": 1.710618257522583,
      "step": 60282
    },
    {
      "epoch": 5.748939514160156e-05,
      "step": 60282,
      "training_step_time": 8.231285095214844
    },
    {
      "epoch": 5.749034881591797e-05,
      "model_forward_time": 1.7163820266723633,
      "step": 60283
    },
    {
      "epoch": 5.749034881591797e-05,
      "step": 60283,
      "training_step_time": 9.228323698043823
    },
    {
      "epoch": 5.7491302490234374e-05,
      "model_forward_time": 1.717780590057373,
      "step": 60284
    },
    {
      "epoch": 5.7491302490234374e-05,
      "step": 60284,
      "training_step_time": 8.18242883682251
    },
    {
      "epoch": 5.749225616455078e-05,
      "model_forward_time": 1.7134921550750732,
      "step": 60285
    },
    {
      "epoch": 5.749225616455078e-05,
      "step": 60285,
      "training_step_time": 9.159227132797241
    },
    {
      "epoch": 5.749320983886719e-05,
      "model_forward_time": 1.7049169540405273,
      "step": 60286
    },
    {
      "epoch": 5.749320983886719e-05,
      "step": 60286,
      "training_step_time": 7.552371025085449
    },
    {
      "epoch": 5.749416351318359e-05,
      "model_forward_time": 1.706944227218628,
      "step": 60287
    },
    {
      "epoch": 5.749416351318359e-05,
      "step": 60287,
      "training_step_time": 7.3630053997039795
    },
    {
      "epoch": 5.74951171875e-05,
      "model_forward_time": 1.710099697113037,
      "step": 60288
    },
    {
      "epoch": 5.74951171875e-05,
      "step": 60288,
      "training_step_time": 7.555121898651123
    },
    {
      "epoch": 5.749607086181641e-05,
      "model_forward_time": 1.7002453804016113,
      "step": 60289
    },
    {
      "epoch": 5.749607086181641e-05,
      "step": 60289,
      "training_step_time": 7.797678470611572
    },
    {
      "epoch": 5.749702453613281e-05,
      "grad_norm": 0.09330923855304718,
      "learning_rate": 3.7263062630569614e-06,
      "loss": 0.0561,
      "step": 60290
    },
    {
      "epoch": 5.749702453613281e-05,
      "model_forward_time": 2.272186279296875,
      "step": 60290
    },
    {
      "dynamics_loss_avg": 0.06510537061840296,
      "epoch": 5.749702453613281e-05,
      "step": 60290
    },
    {
      "action_loss_avg": 0.0011903933656867594,
      "epoch": 5.749702453613281e-05,
      "step": 60290
    },
    {
      "epoch": 5.749702453613281e-05,
      "step": 60290,
      "training_step_time": 7.5699286460876465
    },
    {
      "epoch": 5.7497978210449216e-05,
      "model_forward_time": 1.7050857543945312,
      "step": 60291
    },
    {
      "epoch": 5.7497978210449216e-05,
      "step": 60291,
      "training_step_time": 7.602108716964722
    },
    {
      "epoch": 5.7498931884765627e-05,
      "model_forward_time": 1.7028400897979736,
      "step": 60292
    },
    {
      "epoch": 5.7498931884765627e-05,
      "step": 60292,
      "training_step_time": 7.704015493392944
    },
    {
      "epoch": 5.749988555908203e-05,
      "model_forward_time": 1.7050690650939941,
      "step": 60293
    },
    {
      "epoch": 5.749988555908203e-05,
      "step": 60293,
      "training_step_time": 8.362892389297485
    },
    {
      "epoch": 5.7500839233398435e-05,
      "model_forward_time": 1.702071189880371,
      "step": 60294
    },
    {
      "epoch": 5.7500839233398435e-05,
      "step": 60294,
      "training_step_time": 7.611842155456543
    },
    {
      "epoch": 5.7501792907714845e-05,
      "model_forward_time": 1.7049968242645264,
      "step": 60295
    },
    {
      "epoch": 5.7501792907714845e-05,
      "step": 60295,
      "training_step_time": 7.537759304046631
    },
    {
      "epoch": 5.750274658203125e-05,
      "model_forward_time": 1.7050466537475586,
      "step": 60296
    },
    {
      "epoch": 5.750274658203125e-05,
      "step": 60296,
      "training_step_time": 7.378260612487793
    },
    {
      "epoch": 5.7503700256347653e-05,
      "model_forward_time": 1.7083454132080078,
      "step": 60297
    },
    {
      "epoch": 5.7503700256347653e-05,
      "step": 60297,
      "training_step_time": 7.463123559951782
    },
    {
      "epoch": 5.7504653930664064e-05,
      "model_forward_time": 1.7039642333984375,
      "step": 60298
    },
    {
      "epoch": 5.7504653930664064e-05,
      "step": 60298,
      "training_step_time": 7.84438419342041
    },
    {
      "epoch": 5.750560760498047e-05,
      "model_forward_time": 1.7068593502044678,
      "step": 60299
    },
    {
      "epoch": 5.750560760498047e-05,
      "step": 60299,
      "training_step_time": 8.892591714859009
    },
    {
      "epoch": 5.750656127929687e-05,
      "grad_norm": 0.08536943048238754,
      "learning_rate": 3.7247074110272562e-06,
      "loss": 0.0561,
      "step": 60300
    },
    {
      "epoch": 5.750656127929687e-05,
      "model_forward_time": 1.6994946002960205,
      "step": 60300
    },
    {
      "dynamics_loss_avg": 0.06775988526642322,
      "epoch": 5.750656127929687e-05,
      "step": 60300
    },
    {
      "action_loss_avg": 0.0028653466259129347,
      "epoch": 5.750656127929687e-05,
      "step": 60300
    },
    {
      "epoch": 5.750656127929687e-05,
      "step": 60300,
      "training_step_time": 8.117522954940796
    },
    {
      "epoch": 5.750751495361328e-05,
      "model_forward_time": 1.70485520362854,
      "step": 60301
    },
    {
      "epoch": 5.750751495361328e-05,
      "step": 60301,
      "training_step_time": 8.243464708328247
    },
    {
      "epoch": 5.750846862792969e-05,
      "model_forward_time": 1.7026681900024414,
      "step": 60302
    },
    {
      "epoch": 5.750846862792969e-05,
      "step": 60302,
      "training_step_time": 7.6927008628845215
    },
    {
      "epoch": 5.750942230224609e-05,
      "model_forward_time": 1.7553527355194092,
      "step": 60303
    },
    {
      "epoch": 5.750942230224609e-05,
      "step": 60303,
      "training_step_time": 8.64130163192749
    },
    {
      "epoch": 5.75103759765625e-05,
      "model_forward_time": 1.7215709686279297,
      "step": 60304
    },
    {
      "epoch": 5.75103759765625e-05,
      "step": 60304,
      "training_step_time": 7.589670181274414
    },
    {
      "epoch": 5.7511329650878906e-05,
      "model_forward_time": 1.7070708274841309,
      "step": 60305
    },
    {
      "epoch": 5.7511329650878906e-05,
      "step": 60305,
      "training_step_time": 7.372514486312866
    },
    {
      "epoch": 5.751228332519531e-05,
      "model_forward_time": 1.7002942562103271,
      "step": 60306
    },
    {
      "epoch": 5.751228332519531e-05,
      "step": 60306,
      "training_step_time": 7.492761850357056
    },
    {
      "epoch": 5.751323699951172e-05,
      "model_forward_time": 1.704178810119629,
      "step": 60307
    },
    {
      "epoch": 5.751323699951172e-05,
      "step": 60307,
      "training_step_time": 8.619327783584595
    },
    {
      "epoch": 5.7514190673828125e-05,
      "model_forward_time": 1.7046363353729248,
      "step": 60308
    },
    {
      "epoch": 5.7514190673828125e-05,
      "step": 60308,
      "training_step_time": 8.515040636062622
    },
    {
      "epoch": 5.751514434814453e-05,
      "model_forward_time": 2.2386269569396973,
      "step": 60309
    },
    {
      "epoch": 5.751514434814453e-05,
      "step": 60309,
      "training_step_time": 9.134262084960938
    },
    {
      "epoch": 5.751609802246094e-05,
      "grad_norm": 0.11221756041049957,
      "learning_rate": 3.7231086984616312e-06,
      "loss": 0.0605,
      "step": 60310
    },
    {
      "epoch": 5.751609802246094e-05,
      "model_forward_time": 1.7016980648040771,
      "step": 60310
    },
    {
      "dynamics_loss_avg": 0.051283915899693965,
      "epoch": 5.751609802246094e-05,
      "step": 60310
    },
    {
      "action_loss_avg": 0.001713520076009445,
      "epoch": 5.751609802246094e-05,
      "step": 60310
    },
    {
      "epoch": 5.751609802246094e-05,
      "step": 60310,
      "training_step_time": 8.157930374145508
    },
    {
      "epoch": 5.7517051696777344e-05,
      "model_forward_time": 1.7012810707092285,
      "step": 60311
    },
    {
      "epoch": 5.7517051696777344e-05,
      "step": 60311,
      "training_step_time": 10.195802688598633
    },
    {
      "epoch": 5.751800537109375e-05,
      "model_forward_time": 1.7590665817260742,
      "step": 60312
    },
    {
      "epoch": 5.751800537109375e-05,
      "step": 60312,
      "training_step_time": 7.612235069274902
    },
    {
      "epoch": 5.751895904541016e-05,
      "model_forward_time": 1.7035071849822998,
      "step": 60313
    },
    {
      "epoch": 5.751895904541016e-05,
      "step": 60313,
      "training_step_time": 7.595372915267944
    },
    {
      "epoch": 5.751991271972656e-05,
      "model_forward_time": 1.708371877670288,
      "step": 60314
    },
    {
      "epoch": 5.751991271972656e-05,
      "step": 60314,
      "training_step_time": 7.608323335647583
    },
    {
      "epoch": 5.7520866394042967e-05,
      "model_forward_time": 1.7058079242706299,
      "step": 60315
    },
    {
      "epoch": 5.7520866394042967e-05,
      "step": 60315,
      "training_step_time": 7.380592584609985
    },
    {
      "epoch": 5.752182006835938e-05,
      "model_forward_time": 1.7023320198059082,
      "step": 60316
    },
    {
      "epoch": 5.752182006835938e-05,
      "step": 60316,
      "training_step_time": 8.686990976333618
    },
    {
      "epoch": 5.752277374267578e-05,
      "model_forward_time": 1.728668451309204,
      "step": 60317
    },
    {
      "epoch": 5.752277374267578e-05,
      "step": 60317,
      "training_step_time": 7.814195871353149
    },
    {
      "epoch": 5.7523727416992185e-05,
      "model_forward_time": 2.2497572898864746,
      "step": 60318
    },
    {
      "epoch": 5.7523727416992185e-05,
      "step": 60318,
      "training_step_time": 7.662579298019409
    },
    {
      "epoch": 5.7524681091308596e-05,
      "model_forward_time": 1.6999056339263916,
      "step": 60319
    },
    {
      "epoch": 5.7524681091308596e-05,
      "step": 60319,
      "training_step_time": 8.664551973342896
    },
    {
      "epoch": 5.7525634765625e-05,
      "grad_norm": 0.10913544148206711,
      "learning_rate": 3.72151012553492e-06,
      "loss": 0.0554,
      "step": 60320
    },
    {
      "epoch": 5.7525634765625e-05,
      "model_forward_time": 1.7046408653259277,
      "step": 60320
    },
    {
      "dynamics_loss_avg": 0.03604571502655744,
      "epoch": 5.7525634765625e-05,
      "step": 60320
    },
    {
      "action_loss_avg": 0.0012543075776193292,
      "epoch": 5.7525634765625e-05,
      "step": 60320
    },
    {
      "epoch": 5.7525634765625e-05,
      "step": 60320,
      "training_step_time": 9.05048680305481
    },
    {
      "epoch": 5.7526588439941404e-05,
      "model_forward_time": 1.7482843399047852,
      "step": 60321
    },
    {
      "epoch": 5.7526588439941404e-05,
      "step": 60321,
      "training_step_time": 8.4666109085083
    },
    {
      "epoch": 5.7527542114257815e-05,
      "model_forward_time": 1.705249547958374,
      "step": 60322
    },
    {
      "epoch": 5.7527542114257815e-05,
      "step": 60322,
      "training_step_time": 7.606789588928223
    },
    {
      "epoch": 5.752849578857422e-05,
      "model_forward_time": 1.7068605422973633,
      "step": 60323
    },
    {
      "epoch": 5.752849578857422e-05,
      "step": 60323,
      "training_step_time": 7.630769968032837
    },
    {
      "epoch": 5.752944946289062e-05,
      "model_forward_time": 1.7036821842193604,
      "step": 60324
    },
    {
      "epoch": 5.752944946289062e-05,
      "step": 60324,
      "training_step_time": 7.347581148147583
    },
    {
      "epoch": 5.7530403137207034e-05,
      "model_forward_time": 1.7071382999420166,
      "step": 60325
    },
    {
      "epoch": 5.7530403137207034e-05,
      "step": 60325,
      "training_step_time": 8.452751159667969
    },
    {
      "epoch": 5.753135681152344e-05,
      "model_forward_time": 1.6980037689208984,
      "step": 60326
    },
    {
      "epoch": 5.753135681152344e-05,
      "step": 60326,
      "training_step_time": 8.538507223129272
    },
    {
      "epoch": 5.753231048583984e-05,
      "model_forward_time": 2.2401907444000244,
      "step": 60327
    },
    {
      "epoch": 5.753231048583984e-05,
      "step": 60327,
      "training_step_time": 7.609666585922241
    },
    {
      "epoch": 5.753326416015625e-05,
      "model_forward_time": 1.6997790336608887,
      "step": 60328
    },
    {
      "epoch": 5.753326416015625e-05,
      "step": 60328,
      "training_step_time": 7.581161260604858
    },
    {
      "epoch": 5.753421783447266e-05,
      "model_forward_time": 1.706989049911499,
      "step": 60329
    },
    {
      "epoch": 5.753421783447266e-05,
      "step": 60329,
      "training_step_time": 8.718448162078857
    },
    {
      "epoch": 5.753517150878906e-05,
      "grad_norm": 0.07176506519317627,
      "learning_rate": 3.719911692421939e-06,
      "loss": 0.0552,
      "step": 60330
    },
    {
      "epoch": 5.753517150878906e-05,
      "model_forward_time": 1.7080183029174805,
      "step": 60330
    },
    {
      "dynamics_loss_avg": 0.046623615082353354,
      "epoch": 5.753517150878906e-05,
      "step": 60330
    },
    {
      "action_loss_avg": 0.0013031045848038048,
      "epoch": 5.753517150878906e-05,
      "step": 60330
    },
    {
      "epoch": 5.753517150878906e-05,
      "step": 60330,
      "training_step_time": 7.666758060455322
    },
    {
      "epoch": 5.753612518310547e-05,
      "model_forward_time": 1.6908173561096191,
      "step": 60331
    },
    {
      "epoch": 5.753612518310547e-05,
      "step": 60331,
      "training_step_time": 7.581772565841675
    },
    {
      "epoch": 5.7537078857421876e-05,
      "model_forward_time": 1.6953227519989014,
      "step": 60332
    },
    {
      "epoch": 5.7537078857421876e-05,
      "step": 60332,
      "training_step_time": 7.604817628860474
    },
    {
      "epoch": 5.753803253173828e-05,
      "model_forward_time": 1.7090325355529785,
      "step": 60333
    },
    {
      "epoch": 5.753803253173828e-05,
      "step": 60333,
      "training_step_time": 7.3680033683776855
    },
    {
      "epoch": 5.753898620605469e-05,
      "model_forward_time": 1.6973285675048828,
      "step": 60334
    },
    {
      "epoch": 5.753898620605469e-05,
      "step": 60334,
      "training_step_time": 7.676807403564453
    },
    {
      "epoch": 5.7539939880371095e-05,
      "model_forward_time": 1.6872978210449219,
      "step": 60335
    },
    {
      "epoch": 5.7539939880371095e-05,
      "step": 60335,
      "training_step_time": 7.8261942863464355
    },
    {
      "epoch": 5.75408935546875e-05,
      "model_forward_time": 2.229778528213501,
      "step": 60336
    },
    {
      "epoch": 5.75408935546875e-05,
      "step": 60336,
      "training_step_time": 7.541117429733276
    },
    {
      "epoch": 5.754184722900391e-05,
      "model_forward_time": 1.6935744285583496,
      "step": 60337
    },
    {
      "epoch": 5.754184722900391e-05,
      "step": 60337,
      "training_step_time": 8.42424464225769
    },
    {
      "epoch": 5.7542800903320313e-05,
      "model_forward_time": 1.6900169849395752,
      "step": 60338
    },
    {
      "epoch": 5.7542800903320313e-05,
      "step": 60338,
      "training_step_time": 8.391874074935913
    },
    {
      "epoch": 5.754375457763672e-05,
      "model_forward_time": 1.6905932426452637,
      "step": 60339
    },
    {
      "epoch": 5.754375457763672e-05,
      "step": 60339,
      "training_step_time": 8.415626525878906
    },
    {
      "epoch": 5.754470825195313e-05,
      "grad_norm": 0.18920226395130157,
      "learning_rate": 3.7183133992974903e-06,
      "loss": 0.0555,
      "step": 60340
    },
    {
      "epoch": 5.754470825195313e-05,
      "model_forward_time": 1.6909537315368652,
      "step": 60340
    },
    {
      "dynamics_loss_avg": 0.05810922887176275,
      "epoch": 5.754470825195313e-05,
      "step": 60340
    },
    {
      "action_loss_avg": 0.001215460937237367,
      "epoch": 5.754470825195313e-05,
      "step": 60340
    },
    {
      "epoch": 5.754470825195313e-05,
      "step": 60340,
      "training_step_time": 7.591569662094116
    },
    {
      "epoch": 5.754566192626953e-05,
      "model_forward_time": 1.697160243988037,
      "step": 60341
    },
    {
      "epoch": 5.754566192626953e-05,
      "step": 60341,
      "training_step_time": 9.347543239593506
    },
    {
      "epoch": 5.7546615600585936e-05,
      "model_forward_time": 1.7034337520599365,
      "step": 60342
    },
    {
      "epoch": 5.7546615600585936e-05,
      "step": 60342,
      "training_step_time": 7.522374629974365
    },
    {
      "epoch": 5.754756927490235e-05,
      "model_forward_time": 1.702303171157837,
      "step": 60343
    },
    {
      "epoch": 5.754756927490235e-05,
      "step": 60343,
      "training_step_time": 7.686462640762329
    },
    {
      "epoch": 5.754852294921875e-05,
      "model_forward_time": 1.6993718147277832,
      "step": 60344
    },
    {
      "epoch": 5.754852294921875e-05,
      "step": 60344,
      "training_step_time": 7.7873618602752686
    },
    {
      "epoch": 5.7549476623535155e-05,
      "model_forward_time": 2.24636173248291,
      "step": 60345
    },
    {
      "epoch": 5.7549476623535155e-05,
      "step": 60345,
      "training_step_time": 8.398737907409668
    },
    {
      "epoch": 5.755043029785156e-05,
      "model_forward_time": 1.7023472785949707,
      "step": 60346
    },
    {
      "epoch": 5.755043029785156e-05,
      "step": 60346,
      "training_step_time": 8.638745307922363
    },
    {
      "epoch": 5.755138397216797e-05,
      "model_forward_time": 1.7048566341400146,
      "step": 60347
    },
    {
      "epoch": 5.755138397216797e-05,
      "step": 60347,
      "training_step_time": 8.726763248443604
    },
    {
      "epoch": 5.7552337646484374e-05,
      "model_forward_time": 1.7043728828430176,
      "step": 60348
    },
    {
      "epoch": 5.7552337646484374e-05,
      "step": 60348,
      "training_step_time": 8.378747701644897
    },
    {
      "epoch": 5.755329132080078e-05,
      "model_forward_time": 1.7009670734405518,
      "step": 60349
    },
    {
      "epoch": 5.755329132080078e-05,
      "step": 60349,
      "training_step_time": 7.624306917190552
    },
    {
      "epoch": 5.755424499511719e-05,
      "grad_norm": 0.11657989770174026,
      "learning_rate": 3.7167152463363622e-06,
      "loss": 0.0535,
      "step": 60350
    },
    {
      "epoch": 5.755424499511719e-05,
      "model_forward_time": 1.702798843383789,
      "step": 60350
    },
    {
      "dynamics_loss_avg": 0.04740156326442957,
      "epoch": 5.755424499511719e-05,
      "step": 60350
    },
    {
      "action_loss_avg": 0.0015831821219762787,
      "epoch": 5.755424499511719e-05,
      "step": 60350
    },
    {
      "epoch": 5.755424499511719e-05,
      "step": 60350,
      "training_step_time": 7.606173038482666
    },
    {
      "epoch": 5.755519866943359e-05,
      "model_forward_time": 1.6944751739501953,
      "step": 60351
    },
    {
      "epoch": 5.755519866943359e-05,
      "step": 60351,
      "training_step_time": 7.5459513664245605
    },
    {
      "epoch": 5.755615234375e-05,
      "model_forward_time": 1.7318706512451172,
      "step": 60352
    },
    {
      "epoch": 5.755615234375e-05,
      "step": 60352,
      "training_step_time": 7.64775824546814
    },
    {
      "epoch": 5.755710601806641e-05,
      "model_forward_time": 1.7103400230407715,
      "step": 60353
    },
    {
      "epoch": 5.755710601806641e-05,
      "step": 60353,
      "training_step_time": 8.260887145996094
    },
    {
      "epoch": 5.755805969238281e-05,
      "model_forward_time": 2.8984835147857666,
      "step": 60354
    },
    {
      "epoch": 5.755805969238281e-05,
      "step": 60354,
      "training_step_time": 7.6473002433776855
    },
    {
      "epoch": 5.7559013366699216e-05,
      "model_forward_time": 1.758622646331787,
      "step": 60355
    },
    {
      "epoch": 5.7559013366699216e-05,
      "step": 60355,
      "training_step_time": 7.617140769958496
    },
    {
      "epoch": 5.7559967041015627e-05,
      "model_forward_time": 1.752708911895752,
      "step": 60356
    },
    {
      "epoch": 5.7559967041015627e-05,
      "step": 60356,
      "training_step_time": 8.449355602264404
    },
    {
      "epoch": 5.756092071533203e-05,
      "model_forward_time": 1.7500677108764648,
      "step": 60357
    },
    {
      "epoch": 5.756092071533203e-05,
      "step": 60357,
      "training_step_time": 7.65650200843811
    },
    {
      "epoch": 5.7561874389648435e-05,
      "model_forward_time": 1.7437446117401123,
      "step": 60358
    },
    {
      "epoch": 5.7561874389648435e-05,
      "step": 60358,
      "training_step_time": 7.619761943817139
    },
    {
      "epoch": 5.7562828063964845e-05,
      "model_forward_time": 1.7557868957519531,
      "step": 60359
    },
    {
      "epoch": 5.7562828063964845e-05,
      "step": 60359,
      "training_step_time": 7.644784927368164
    },
    {
      "epoch": 5.756378173828125e-05,
      "grad_norm": 0.15661855041980743,
      "learning_rate": 3.7151172337133245e-06,
      "loss": 0.0547,
      "step": 60360
    },
    {
      "epoch": 5.756378173828125e-05,
      "model_forward_time": 1.7561445236206055,
      "step": 60360
    },
    {
      "dynamics_loss_avg": 0.051411422155797484,
      "epoch": 5.756378173828125e-05,
      "step": 60360
    },
    {
      "action_loss_avg": 0.0012608947115950286,
      "epoch": 5.756378173828125e-05,
      "step": 60360
    },
    {
      "epoch": 5.756378173828125e-05,
      "step": 60360,
      "training_step_time": 7.603949069976807
    },
    {
      "epoch": 5.7564735412597653e-05,
      "model_forward_time": 1.7486584186553955,
      "step": 60361
    },
    {
      "epoch": 5.7564735412597653e-05,
      "step": 60361,
      "training_step_time": 7.654345750808716
    },
    {
      "epoch": 5.7565689086914064e-05,
      "model_forward_time": 1.7604949474334717,
      "step": 60362
    },
    {
      "epoch": 5.7565689086914064e-05,
      "step": 60362,
      "training_step_time": 7.69850754737854
    },
    {
      "epoch": 5.756664276123047e-05,
      "model_forward_time": 2.8982138633728027,
      "step": 60363
    },
    {
      "epoch": 5.756664276123047e-05,
      "step": 60363,
      "training_step_time": 7.748986721038818
    },
    {
      "epoch": 5.756759643554687e-05,
      "model_forward_time": 1.7581536769866943,
      "step": 60364
    },
    {
      "epoch": 5.756759643554687e-05,
      "step": 60364,
      "training_step_time": 7.5781378746032715
    },
    {
      "epoch": 5.756855010986328e-05,
      "model_forward_time": 1.7440416812896729,
      "step": 60365
    },
    {
      "epoch": 5.756855010986328e-05,
      "step": 60365,
      "training_step_time": 7.767303705215454
    },
    {
      "epoch": 5.756950378417969e-05,
      "model_forward_time": 1.7415006160736084,
      "step": 60366
    },
    {
      "epoch": 5.756950378417969e-05,
      "step": 60366,
      "training_step_time": 9.519330024719238
    },
    {
      "epoch": 5.757045745849609e-05,
      "model_forward_time": 1.7511663436889648,
      "step": 60367
    },
    {
      "epoch": 5.757045745849609e-05,
      "step": 60367,
      "training_step_time": 8.587903499603271
    },
    {
      "epoch": 5.75714111328125e-05,
      "model_forward_time": 1.3940675258636475,
      "step": 60368
    },
    {
      "epoch": 5.75714111328125e-05,
      "step": 60368,
      "training_step_time": 7.545443534851074
    },
    {
      "epoch": 5.7572364807128906e-05,
      "model_forward_time": 1.7551133632659912,
      "step": 60369
    },
    {
      "epoch": 5.7572364807128906e-05,
      "step": 60369,
      "training_step_time": 7.592154502868652
    },
    {
      "epoch": 5.757331848144531e-05,
      "grad_norm": 0.12285915017127991,
      "learning_rate": 3.7135193616031352e-06,
      "loss": 0.0575,
      "step": 60370
    },
    {
      "epoch": 5.757331848144531e-05,
      "model_forward_time": 1.754434585571289,
      "step": 60370
    },
    {
      "dynamics_loss_avg": 0.06535197682678699,
      "epoch": 5.757331848144531e-05,
      "step": 60370
    },
    {
      "action_loss_avg": 0.00150049869553186,
      "epoch": 5.757331848144531e-05,
      "step": 60370
    },
    {
      "epoch": 5.757331848144531e-05,
      "step": 60370,
      "training_step_time": 7.590025186538696
    },
    {
      "epoch": 5.757427215576172e-05,
      "model_forward_time": 1.7505056858062744,
      "step": 60371
    },
    {
      "epoch": 5.757427215576172e-05,
      "step": 60371,
      "training_step_time": 7.704637050628662
    },
    {
      "epoch": 5.7575225830078125e-05,
      "model_forward_time": 2.8905959129333496,
      "step": 60372
    },
    {
      "epoch": 5.7575225830078125e-05,
      "step": 60372,
      "training_step_time": 8.77649974822998
    },
    {
      "epoch": 5.757617950439453e-05,
      "model_forward_time": 1.7542426586151123,
      "step": 60373
    },
    {
      "epoch": 5.757617950439453e-05,
      "step": 60373,
      "training_step_time": 7.564555883407593
    },
    {
      "epoch": 5.757713317871094e-05,
      "model_forward_time": 1.7508177757263184,
      "step": 60374
    },
    {
      "epoch": 5.757713317871094e-05,
      "step": 60374,
      "training_step_time": 9.249191761016846
    },
    {
      "epoch": 5.7578086853027344e-05,
      "model_forward_time": 1.7513294219970703,
      "step": 60375
    },
    {
      "epoch": 5.7578086853027344e-05,
      "step": 60375,
      "training_step_time": 10.475748062133789
    },
    {
      "epoch": 5.757904052734375e-05,
      "model_forward_time": 1.731321096420288,
      "step": 60376
    },
    {
      "epoch": 5.757904052734375e-05,
      "step": 60376,
      "training_step_time": 7.627788305282593
    },
    {
      "epoch": 5.757999420166016e-05,
      "model_forward_time": 1.7314538955688477,
      "step": 60377
    },
    {
      "epoch": 5.757999420166016e-05,
      "step": 60377,
      "training_step_time": 7.616725444793701
    },
    {
      "epoch": 5.758094787597656e-05,
      "model_forward_time": 1.7391512393951416,
      "step": 60378
    },
    {
      "epoch": 5.758094787597656e-05,
      "step": 60378,
      "training_step_time": 7.596397399902344
    },
    {
      "epoch": 5.758190155029297e-05,
      "model_forward_time": 1.735497236251831,
      "step": 60379
    },
    {
      "epoch": 5.758190155029297e-05,
      "step": 60379,
      "training_step_time": 7.654558420181274
    },
    {
      "epoch": 5.758285522460938e-05,
      "grad_norm": 0.10091974586248398,
      "learning_rate": 3.7119216301805323e-06,
      "loss": 0.0563,
      "step": 60380
    },
    {
      "epoch": 5.758285522460938e-05,
      "model_forward_time": 1.740556240081787,
      "step": 60380
    },
    {
      "dynamics_loss_avg": 0.06139450408518314,
      "epoch": 5.758285522460938e-05,
      "step": 60380
    },
    {
      "action_loss_avg": 0.0022860406956169755,
      "epoch": 5.758285522460938e-05,
      "step": 60380
    },
    {
      "epoch": 5.758285522460938e-05,
      "step": 60380,
      "training_step_time": 8.089733839035034
    },
    {
      "epoch": 5.758380889892578e-05,
      "model_forward_time": 2.889439105987549,
      "step": 60381
    },
    {
      "epoch": 5.758380889892578e-05,
      "step": 60381,
      "training_step_time": 7.662473440170288
    },
    {
      "epoch": 5.7584762573242185e-05,
      "model_forward_time": 1.7371816635131836,
      "step": 60382
    },
    {
      "epoch": 5.7584762573242185e-05,
      "step": 60382,
      "training_step_time": 7.70245099067688
    },
    {
      "epoch": 5.7585716247558596e-05,
      "model_forward_time": 1.7336087226867676,
      "step": 60383
    },
    {
      "epoch": 5.7585716247558596e-05,
      "step": 60383,
      "training_step_time": 7.704110622406006
    },
    {
      "epoch": 5.7586669921875e-05,
      "model_forward_time": 1.7340636253356934,
      "step": 60384
    },
    {
      "epoch": 5.7586669921875e-05,
      "step": 60384,
      "training_step_time": 7.5798585414886475
    },
    {
      "epoch": 5.7587623596191404e-05,
      "model_forward_time": 1.7414524555206299,
      "step": 60385
    },
    {
      "epoch": 5.7587623596191404e-05,
      "step": 60385,
      "training_step_time": 8.220447540283203
    },
    {
      "epoch": 5.7588577270507815e-05,
      "model_forward_time": 1.7400846481323242,
      "step": 60386
    },
    {
      "epoch": 5.7588577270507815e-05,
      "step": 60386,
      "training_step_time": 7.5791778564453125
    },
    {
      "epoch": 5.758953094482422e-05,
      "model_forward_time": 1.742912769317627,
      "step": 60387
    },
    {
      "epoch": 5.758953094482422e-05,
      "step": 60387,
      "training_step_time": 7.533651828765869
    },
    {
      "epoch": 5.759048461914062e-05,
      "model_forward_time": 1.7371511459350586,
      "step": 60388
    },
    {
      "epoch": 5.759048461914062e-05,
      "step": 60388,
      "training_step_time": 7.564552068710327
    },
    {
      "epoch": 5.7591438293457034e-05,
      "model_forward_time": 1.7424356937408447,
      "step": 60389
    },
    {
      "epoch": 5.7591438293457034e-05,
      "step": 60389,
      "training_step_time": 7.701468229293823
    },
    {
      "epoch": 5.759239196777344e-05,
      "grad_norm": 0.11963678896427155,
      "learning_rate": 3.710324039620245e-06,
      "loss": 0.0592,
      "step": 60390
    },
    {
      "epoch": 5.759239196777344e-05,
      "model_forward_time": 2.905735731124878,
      "step": 60390
    },
    {
      "dynamics_loss_avg": 0.06887397672981024,
      "epoch": 5.759239196777344e-05,
      "step": 60390
    },
    {
      "action_loss_avg": 0.0012695605517365038,
      "epoch": 5.759239196777344e-05,
      "step": 60390
    },
    {
      "epoch": 5.759239196777344e-05,
      "step": 60390,
      "training_step_time": 7.633764982223511
    },
    {
      "epoch": 5.759334564208984e-05,
      "model_forward_time": 1.734419584274292,
      "step": 60391
    },
    {
      "epoch": 5.759334564208984e-05,
      "step": 60391,
      "training_step_time": 7.773207902908325
    },
    {
      "epoch": 5.759429931640625e-05,
      "model_forward_time": 2.896852731704712,
      "step": 60392
    },
    {
      "epoch": 5.759429931640625e-05,
      "step": 60392,
      "training_step_time": 7.6174609661102295
    },
    {
      "epoch": 5.759525299072266e-05,
      "model_forward_time": 1.7335846424102783,
      "step": 60393
    },
    {
      "epoch": 5.759525299072266e-05,
      "step": 60393,
      "training_step_time": 8.744508028030396
    },
    {
      "epoch": 5.759620666503906e-05,
      "model_forward_time": 1.7393040657043457,
      "step": 60394
    },
    {
      "epoch": 5.759620666503906e-05,
      "step": 60394,
      "training_step_time": 6.6143341064453125
    },
    {
      "epoch": 5.759716033935547e-05,
      "model_forward_time": 1.7362093925476074,
      "step": 60395
    },
    {
      "epoch": 5.759716033935547e-05,
      "step": 60395,
      "training_step_time": 6.6573779582977295
    },
    {
      "epoch": 5.7598114013671876e-05,
      "model_forward_time": 1.7577018737792969,
      "step": 60396
    },
    {
      "epoch": 5.7598114013671876e-05,
      "step": 60396,
      "training_step_time": 6.702728271484375
    },
    {
      "epoch": 5.759906768798828e-05,
      "model_forward_time": 1.7350883483886719,
      "step": 60397
    },
    {
      "epoch": 5.759906768798828e-05,
      "step": 60397,
      "training_step_time": 6.681565999984741
    },
    {
      "epoch": 5.760002136230469e-05,
      "model_forward_time": 1.7384717464447021,
      "step": 60398
    },
    {
      "epoch": 5.760002136230469e-05,
      "step": 60398,
      "training_step_time": 6.654889345169067
    },
    {
      "epoch": 5.7600975036621095e-05,
      "model_forward_time": 1.7354228496551514,
      "step": 60399
    },
    {
      "epoch": 5.7600975036621095e-05,
      "step": 60399,
      "training_step_time": 6.6741814613342285
    },
    {
      "epoch": 5.76019287109375e-05,
      "grad_norm": 0.1263541579246521,
      "learning_rate": 3.708726590096982e-06,
      "loss": 0.0554,
      "step": 60400
    },
    {
      "epoch": 5.76019287109375e-05,
      "model_forward_time": 1.7192981243133545,
      "step": 60400
    },
    {
      "dynamics_loss_avg": 0.051169511303305625,
      "epoch": 5.76019287109375e-05,
      "step": 60400
    },
    {
      "action_loss_avg": 0.0022567725158296526,
      "epoch": 5.76019287109375e-05,
      "step": 60400
    },
    {
      "epoch": 5.76019287109375e-05,
      "step": 60400,
      "training_step_time": 6.645428657531738
    },
    {
      "epoch": 5.760288238525391e-05,
      "model_forward_time": 1.7396810054779053,
      "step": 60401
    },
    {
      "epoch": 5.760288238525391e-05,
      "step": 60401,
      "training_step_time": 8.632489919662476
    },
    {
      "epoch": 5.7603836059570313e-05,
      "model_forward_time": 1.8647143840789795,
      "step": 60402
    },
    {
      "epoch": 5.7603836059570313e-05,
      "step": 60402,
      "training_step_time": 10.593067169189453
    },
    {
      "epoch": 5.760478973388672e-05,
      "model_forward_time": 1.7436423301696777,
      "step": 60403
    },
    {
      "epoch": 5.760478973388672e-05,
      "step": 60403,
      "training_step_time": 7.575377464294434
    },
    {
      "epoch": 5.760574340820313e-05,
      "model_forward_time": 1.7125673294067383,
      "step": 60404
    },
    {
      "epoch": 5.760574340820313e-05,
      "step": 60404,
      "training_step_time": 6.7291319370269775
    },
    {
      "epoch": 5.760669708251953e-05,
      "model_forward_time": 1.7474937438964844,
      "step": 60405
    },
    {
      "epoch": 5.760669708251953e-05,
      "step": 60405,
      "training_step_time": 6.668987989425659
    },
    {
      "epoch": 5.7607650756835936e-05,
      "model_forward_time": 1.7356536388397217,
      "step": 60406
    },
    {
      "epoch": 5.7607650756835936e-05,
      "step": 60406,
      "training_step_time": 6.640318155288696
    },
    {
      "epoch": 5.760860443115234e-05,
      "model_forward_time": 1.729351282119751,
      "step": 60407
    },
    {
      "epoch": 5.760860443115234e-05,
      "step": 60407,
      "training_step_time": 6.66866135597229
    },
    {
      "epoch": 5.760955810546875e-05,
      "model_forward_time": 1.730010747909546,
      "step": 60408
    },
    {
      "epoch": 5.760955810546875e-05,
      "step": 60408,
      "training_step_time": 6.66220235824585
    },
    {
      "epoch": 5.7610511779785155e-05,
      "model_forward_time": 1.3664746284484863,
      "step": 60409
    },
    {
      "epoch": 5.7610511779785155e-05,
      "step": 60409,
      "training_step_time": 6.64679217338562
    },
    {
      "epoch": 5.761146545410156e-05,
      "grad_norm": 0.12416651099920273,
      "learning_rate": 3.707129281785438e-06,
      "loss": 0.0573,
      "step": 60410
    },
    {
      "epoch": 5.761146545410156e-05,
      "model_forward_time": 1.7295072078704834,
      "step": 60410
    },
    {
      "dynamics_loss_avg": 0.06314884200692177,
      "epoch": 5.761146545410156e-05,
      "step": 60410
    },
    {
      "action_loss_avg": 0.0022694127750582994,
      "epoch": 5.761146545410156e-05,
      "step": 60410
    },
    {
      "epoch": 5.761146545410156e-05,
      "step": 60410,
      "training_step_time": 6.665735483169556
    },
    {
      "epoch": 5.761241912841797e-05,
      "model_forward_time": 2.8859193325042725,
      "step": 60411
    },
    {
      "epoch": 5.761241912841797e-05,
      "step": 60411,
      "training_step_time": 11.502495527267456
    },
    {
      "epoch": 5.7613372802734374e-05,
      "model_forward_time": 1.7368175983428955,
      "step": 60412
    },
    {
      "epoch": 5.7613372802734374e-05,
      "step": 60412,
      "training_step_time": 7.638242244720459
    },
    {
      "epoch": 5.761432647705078e-05,
      "model_forward_time": 1.7318994998931885,
      "step": 60413
    },
    {
      "epoch": 5.761432647705078e-05,
      "step": 60413,
      "training_step_time": 6.666239976882935
    },
    {
      "epoch": 5.761528015136719e-05,
      "model_forward_time": 1.7388877868652344,
      "step": 60414
    },
    {
      "epoch": 5.761528015136719e-05,
      "step": 60414,
      "training_step_time": 6.6538403034210205
    },
    {
      "epoch": 5.761623382568359e-05,
      "model_forward_time": 1.7377347946166992,
      "step": 60415
    },
    {
      "epoch": 5.761623382568359e-05,
      "step": 60415,
      "training_step_time": 6.654431581497192
    },
    {
      "epoch": 5.76171875e-05,
      "model_forward_time": 1.734827995300293,
      "step": 60416
    },
    {
      "epoch": 5.76171875e-05,
      "step": 60416,
      "training_step_time": 6.65378212928772
    },
    {
      "epoch": 5.761814117431641e-05,
      "model_forward_time": 1.7397148609161377,
      "step": 60417
    },
    {
      "epoch": 5.761814117431641e-05,
      "step": 60417,
      "training_step_time": 6.660828113555908
    },
    {
      "epoch": 5.761909484863281e-05,
      "model_forward_time": 1.7345929145812988,
      "step": 60418
    },
    {
      "epoch": 5.761909484863281e-05,
      "step": 60418,
      "training_step_time": 6.684845209121704
    },
    {
      "epoch": 5.7620048522949216e-05,
      "model_forward_time": 1.7348113059997559,
      "step": 60419
    },
    {
      "epoch": 5.7620048522949216e-05,
      "step": 60419,
      "training_step_time": 6.6764609813690186
    },
    {
      "epoch": 5.762100219726563e-05,
      "grad_norm": 0.08387406915426254,
      "learning_rate": 3.7055321148602913e-06,
      "loss": 0.0539,
      "step": 60420
    },
    {
      "epoch": 5.762100219726563e-05,
      "model_forward_time": 2.8741683959960938,
      "step": 60420
    },
    {
      "dynamics_loss_avg": 0.050329935550689694,
      "epoch": 5.762100219726563e-05,
      "step": 60420
    },
    {
      "action_loss_avg": 0.0012074181533535012,
      "epoch": 5.762100219726563e-05,
      "step": 60420
    },
    {
      "epoch": 5.762100219726563e-05,
      "step": 60420,
      "training_step_time": 9.763245820999146
    },
    {
      "epoch": 5.762195587158203e-05,
      "model_forward_time": 1.7385876178741455,
      "step": 60421
    },
    {
      "epoch": 5.762195587158203e-05,
      "step": 60421,
      "training_step_time": 8.189646482467651
    },
    {
      "epoch": 5.7622909545898435e-05,
      "model_forward_time": 1.737513542175293,
      "step": 60422
    },
    {
      "epoch": 5.7622909545898435e-05,
      "step": 60422,
      "training_step_time": 6.7014453411102295
    },
    {
      "epoch": 5.7623863220214845e-05,
      "model_forward_time": 1.7333259582519531,
      "step": 60423
    },
    {
      "epoch": 5.7623863220214845e-05,
      "step": 60423,
      "training_step_time": 6.696783542633057
    },
    {
      "epoch": 5.762481689453125e-05,
      "model_forward_time": 1.7393343448638916,
      "step": 60424
    },
    {
      "epoch": 5.762481689453125e-05,
      "step": 60424,
      "training_step_time": 6.673585414886475
    },
    {
      "epoch": 5.7625770568847654e-05,
      "model_forward_time": 1.7376322746276855,
      "step": 60425
    },
    {
      "epoch": 5.7625770568847654e-05,
      "step": 60425,
      "training_step_time": 6.708539724349976
    },
    {
      "epoch": 5.7626724243164064e-05,
      "model_forward_time": 1.740539789199829,
      "step": 60426
    },
    {
      "epoch": 5.7626724243164064e-05,
      "step": 60426,
      "training_step_time": 6.689655303955078
    },
    {
      "epoch": 5.762767791748047e-05,
      "model_forward_time": 1.7409825325012207,
      "step": 60427
    },
    {
      "epoch": 5.762767791748047e-05,
      "step": 60427,
      "training_step_time": 6.69360089302063
    },
    {
      "epoch": 5.762863159179687e-05,
      "model_forward_time": 1.7224955558776855,
      "step": 60428
    },
    {
      "epoch": 5.762863159179687e-05,
      "step": 60428,
      "training_step_time": 6.653829097747803
    },
    {
      "epoch": 5.762958526611328e-05,
      "model_forward_time": 2.8857526779174805,
      "step": 60429
    },
    {
      "epoch": 5.762958526611328e-05,
      "step": 60429,
      "training_step_time": 11.966143131256104
    },
    {
      "epoch": 5.763053894042969e-05,
      "grad_norm": 0.11015655845403671,
      "learning_rate": 3.7039350894962067e-06,
      "loss": 0.0573,
      "step": 60430
    },
    {
      "epoch": 5.763053894042969e-05,
      "model_forward_time": 1.7247133255004883,
      "step": 60430
    },
    {
      "dynamics_loss_avg": 0.05391262080520391,
      "epoch": 5.763053894042969e-05,
      "step": 60430
    },
    {
      "action_loss_avg": 0.0012284471740713343,
      "epoch": 5.763053894042969e-05,
      "step": 60430
    },
    {
      "epoch": 5.763053894042969e-05,
      "step": 60430,
      "training_step_time": 7.7338547706604
    },
    {
      "epoch": 5.763149261474609e-05,
      "model_forward_time": 1.7391602993011475,
      "step": 60431
    },
    {
      "epoch": 5.763149261474609e-05,
      "step": 60431,
      "training_step_time": 7.612002372741699
    },
    {
      "epoch": 5.76324462890625e-05,
      "model_forward_time": 1.3710274696350098,
      "step": 60432
    },
    {
      "epoch": 5.76324462890625e-05,
      "step": 60432,
      "training_step_time": 6.694443941116333
    },
    {
      "epoch": 5.7633399963378906e-05,
      "model_forward_time": 1.7424252033233643,
      "step": 60433
    },
    {
      "epoch": 5.7633399963378906e-05,
      "step": 60433,
      "training_step_time": 6.663213491439819
    },
    {
      "epoch": 5.763435363769531e-05,
      "model_forward_time": 1.72279953956604,
      "step": 60434
    },
    {
      "epoch": 5.763435363769531e-05,
      "step": 60434,
      "training_step_time": 6.607377052307129
    },
    {
      "epoch": 5.763530731201172e-05,
      "model_forward_time": 1.7377192974090576,
      "step": 60435
    },
    {
      "epoch": 5.763530731201172e-05,
      "step": 60435,
      "training_step_time": 6.664544343948364
    },
    {
      "epoch": 5.7636260986328125e-05,
      "model_forward_time": 1.7364120483398438,
      "step": 60436
    },
    {
      "epoch": 5.7636260986328125e-05,
      "step": 60436,
      "training_step_time": 6.65339207649231
    },
    {
      "epoch": 5.763721466064453e-05,
      "model_forward_time": 1.735161304473877,
      "step": 60437
    },
    {
      "epoch": 5.763721466064453e-05,
      "step": 60437,
      "training_step_time": 6.685916423797607
    },
    {
      "epoch": 5.763816833496094e-05,
      "model_forward_time": 1.7636563777923584,
      "step": 60438
    },
    {
      "epoch": 5.763816833496094e-05,
      "step": 60438,
      "training_step_time": 8.531060457229614
    },
    {
      "epoch": 5.7639122009277344e-05,
      "model_forward_time": 1.7403576374053955,
      "step": 60439
    },
    {
      "epoch": 5.7639122009277344e-05,
      "step": 60439,
      "training_step_time": 7.71476936340332
    },
    {
      "epoch": 5.764007568359375e-05,
      "grad_norm": 0.11134634166955948,
      "learning_rate": 3.7023382058678294e-06,
      "loss": 0.0581,
      "step": 60440
    },
    {
      "epoch": 5.764007568359375e-05,
      "model_forward_time": 1.7342498302459717,
      "step": 60440
    },
    {
      "dynamics_loss_avg": 0.032078270241618155,
      "epoch": 5.764007568359375e-05,
      "step": 60440
    },
    {
      "action_loss_avg": 0.0013040756108239292,
      "epoch": 5.764007568359375e-05,
      "step": 60440
    },
    {
      "epoch": 5.764007568359375e-05,
      "step": 60440,
      "training_step_time": 7.542745351791382
    },
    {
      "epoch": 5.764102935791016e-05,
      "model_forward_time": 1.733950138092041,
      "step": 60441
    },
    {
      "epoch": 5.764102935791016e-05,
      "step": 60441,
      "training_step_time": 6.733750104904175
    },
    {
      "epoch": 5.764198303222656e-05,
      "model_forward_time": 1.73636794090271,
      "step": 60442
    },
    {
      "epoch": 5.764198303222656e-05,
      "step": 60442,
      "training_step_time": 6.6789772510528564
    },
    {
      "epoch": 5.764293670654297e-05,
      "model_forward_time": 1.7372589111328125,
      "step": 60443
    },
    {
      "epoch": 5.764293670654297e-05,
      "step": 60443,
      "training_step_time": 6.694090843200684
    },
    {
      "epoch": 5.764389038085938e-05,
      "model_forward_time": 1.7340235710144043,
      "step": 60444
    },
    {
      "epoch": 5.764389038085938e-05,
      "step": 60444,
      "training_step_time": 6.803552865982056
    },
    {
      "epoch": 5.764484405517578e-05,
      "model_forward_time": 1.7366409301757812,
      "step": 60445
    },
    {
      "epoch": 5.764484405517578e-05,
      "step": 60445,
      "training_step_time": 6.764861345291138
    },
    {
      "epoch": 5.7645797729492186e-05,
      "model_forward_time": 1.7362403869628906,
      "step": 60446
    },
    {
      "epoch": 5.7645797729492186e-05,
      "step": 60446,
      "training_step_time": 6.727930307388306
    },
    {
      "epoch": 5.7646751403808596e-05,
      "model_forward_time": 1.7489361763000488,
      "step": 60447
    },
    {
      "epoch": 5.7646751403808596e-05,
      "step": 60447,
      "training_step_time": 9.35857105255127
    },
    {
      "epoch": 5.7647705078125e-05,
      "model_forward_time": 1.7341606616973877,
      "step": 60448
    },
    {
      "epoch": 5.7647705078125e-05,
      "step": 60448,
      "training_step_time": 10.94435715675354
    },
    {
      "epoch": 5.7648658752441404e-05,
      "model_forward_time": 1.7325139045715332,
      "step": 60449
    },
    {
      "epoch": 5.7648658752441404e-05,
      "step": 60449,
      "training_step_time": 7.670677423477173
    },
    {
      "epoch": 5.7649612426757815e-05,
      "grad_norm": 0.14212743937969208,
      "learning_rate": 3.7007414641497974e-06,
      "loss": 0.0547,
      "step": 60450
    },
    {
      "epoch": 5.7649612426757815e-05,
      "model_forward_time": 1.726133108139038,
      "step": 60450
    },
    {
      "dynamics_loss_avg": 0.04467419078573585,
      "epoch": 5.7649612426757815e-05,
      "step": 60450
    },
    {
      "action_loss_avg": 0.0015420042211189866,
      "epoch": 5.7649612426757815e-05,
      "step": 60450
    },
    {
      "epoch": 5.7649612426757815e-05,
      "step": 60450,
      "training_step_time": 7.085942029953003
    },
    {
      "epoch": 5.765056610107422e-05,
      "model_forward_time": 1.7373318672180176,
      "step": 60451
    },
    {
      "epoch": 5.765056610107422e-05,
      "step": 60451,
      "training_step_time": 6.71496319770813
    },
    {
      "epoch": 5.765151977539062e-05,
      "model_forward_time": 1.742185115814209,
      "step": 60452
    },
    {
      "epoch": 5.765151977539062e-05,
      "step": 60452,
      "training_step_time": 6.67518949508667
    },
    {
      "epoch": 5.7652473449707034e-05,
      "model_forward_time": 1.7366423606872559,
      "step": 60453
    },
    {
      "epoch": 5.7652473449707034e-05,
      "step": 60453,
      "training_step_time": 6.702520847320557
    },
    {
      "epoch": 5.765342712402344e-05,
      "model_forward_time": 1.7371902465820312,
      "step": 60454
    },
    {
      "epoch": 5.765342712402344e-05,
      "step": 60454,
      "training_step_time": 6.708901405334473
    },
    {
      "epoch": 5.765438079833984e-05,
      "model_forward_time": 1.7320549488067627,
      "step": 60455
    },
    {
      "epoch": 5.765438079833984e-05,
      "step": 60455,
      "training_step_time": 6.752598524093628
    },
    {
      "epoch": 5.765533447265625e-05,
      "model_forward_time": 1.7349402904510498,
      "step": 60456
    },
    {
      "epoch": 5.765533447265625e-05,
      "step": 60456,
      "training_step_time": 7.559753894805908
    },
    {
      "epoch": 5.765628814697266e-05,
      "model_forward_time": 2.868960380554199,
      "step": 60457
    },
    {
      "epoch": 5.765628814697266e-05,
      "step": 60457,
      "training_step_time": 10.644307374954224
    },
    {
      "epoch": 5.765724182128906e-05,
      "model_forward_time": 1.7322044372558594,
      "step": 60458
    },
    {
      "epoch": 5.765724182128906e-05,
      "step": 60458,
      "training_step_time": 8.22392988204956
    },
    {
      "epoch": 5.765819549560547e-05,
      "model_forward_time": 1.7369120121002197,
      "step": 60459
    },
    {
      "epoch": 5.765819549560547e-05,
      "step": 60459,
      "training_step_time": 7.4525487422943115
    },
    {
      "epoch": 5.7659149169921876e-05,
      "grad_norm": 0.12141221016645432,
      "learning_rate": 3.699144864516726e-06,
      "loss": 0.0609,
      "step": 60460
    },
    {
      "epoch": 5.7659149169921876e-05,
      "model_forward_time": 1.7303507328033447,
      "step": 60460
    },
    {
      "dynamics_loss_avg": 0.06173704983666539,
      "epoch": 5.7659149169921876e-05,
      "step": 60460
    },
    {
      "action_loss_avg": 0.0010166735068196432,
      "epoch": 5.7659149169921876e-05,
      "step": 60460
    },
    {
      "epoch": 5.7659149169921876e-05,
      "step": 60460,
      "training_step_time": 6.6976025104522705
    },
    {
      "epoch": 5.766010284423828e-05,
      "model_forward_time": 1.7337799072265625,
      "step": 60461
    },
    {
      "epoch": 5.766010284423828e-05,
      "step": 60461,
      "training_step_time": 6.696535348892212
    },
    {
      "epoch": 5.766105651855469e-05,
      "model_forward_time": 1.734928846359253,
      "step": 60462
    },
    {
      "epoch": 5.766105651855469e-05,
      "step": 60462,
      "training_step_time": 6.735439300537109
    },
    {
      "epoch": 5.7662010192871095e-05,
      "model_forward_time": 1.7327821254730225,
      "step": 60463
    },
    {
      "epoch": 5.7662010192871095e-05,
      "step": 60463,
      "training_step_time": 6.6773786544799805
    },
    {
      "epoch": 5.76629638671875e-05,
      "model_forward_time": 1.7337403297424316,
      "step": 60464
    },
    {
      "epoch": 5.76629638671875e-05,
      "step": 60464,
      "training_step_time": 6.712864398956299
    },
    {
      "epoch": 5.766391754150391e-05,
      "model_forward_time": 1.7217819690704346,
      "step": 60465
    },
    {
      "epoch": 5.766391754150391e-05,
      "step": 60465,
      "training_step_time": 7.611067533493042
    },
    {
      "epoch": 5.7664871215820314e-05,
      "model_forward_time": 2.914248466491699,
      "step": 60466
    },
    {
      "epoch": 5.7664871215820314e-05,
      "step": 60466,
      "training_step_time": 10.1055006980896
    },
    {
      "epoch": 5.766582489013672e-05,
      "model_forward_time": 1.3734424114227295,
      "step": 60467
    },
    {
      "epoch": 5.766582489013672e-05,
      "step": 60467,
      "training_step_time": 7.801656246185303
    },
    {
      "epoch": 5.766677856445313e-05,
      "model_forward_time": 1.7370526790618896,
      "step": 60468
    },
    {
      "epoch": 5.766677856445313e-05,
      "step": 60468,
      "training_step_time": 7.610677242279053
    },
    {
      "epoch": 5.766773223876953e-05,
      "model_forward_time": 1.7371103763580322,
      "step": 60469
    },
    {
      "epoch": 5.766773223876953e-05,
      "step": 60469,
      "training_step_time": 6.726712703704834
    },
    {
      "epoch": 5.7668685913085936e-05,
      "grad_norm": 0.11131579428911209,
      "learning_rate": 3.6975484071432156e-06,
      "loss": 0.0555,
      "step": 60470
    },
    {
      "epoch": 5.7668685913085936e-05,
      "model_forward_time": 1.7220711708068848,
      "step": 60470
    },
    {
      "dynamics_loss_avg": 0.059380959905684,
      "epoch": 5.7668685913085936e-05,
      "step": 60470
    },
    {
      "action_loss_avg": 0.0011862094717798755,
      "epoch": 5.7668685913085936e-05,
      "step": 60470
    },
    {
      "epoch": 5.7668685913085936e-05,
      "step": 60470,
      "training_step_time": 6.7020111083984375
    },
    {
      "epoch": 5.766963958740234e-05,
      "model_forward_time": 1.7329692840576172,
      "step": 60471
    },
    {
      "epoch": 5.766963958740234e-05,
      "step": 60471,
      "training_step_time": 6.710206747055054
    },
    {
      "epoch": 5.767059326171875e-05,
      "model_forward_time": 1.7340319156646729,
      "step": 60472
    },
    {
      "epoch": 5.767059326171875e-05,
      "step": 60472,
      "training_step_time": 6.734951972961426
    },
    {
      "epoch": 5.7671546936035155e-05,
      "model_forward_time": 1.7266943454742432,
      "step": 60473
    },
    {
      "epoch": 5.7671546936035155e-05,
      "step": 60473,
      "training_step_time": 6.698123455047607
    },
    {
      "epoch": 5.767250061035156e-05,
      "model_forward_time": 1.7039406299591064,
      "step": 60474
    },
    {
      "epoch": 5.767250061035156e-05,
      "step": 60474,
      "training_step_time": 7.275990724563599
    },
    {
      "epoch": 5.767345428466797e-05,
      "model_forward_time": 2.9259045124053955,
      "step": 60475
    },
    {
      "epoch": 5.767345428466797e-05,
      "step": 60475,
      "training_step_time": 9.810238599777222
    },
    {
      "epoch": 5.7674407958984374e-05,
      "model_forward_time": 1.7181265354156494,
      "step": 60476
    },
    {
      "epoch": 5.7674407958984374e-05,
      "step": 60476,
      "training_step_time": 7.8212292194366455
    },
    {
      "epoch": 5.767536163330078e-05,
      "model_forward_time": 1.7138488292694092,
      "step": 60477
    },
    {
      "epoch": 5.767536163330078e-05,
      "step": 60477,
      "training_step_time": 7.623703479766846
    },
    {
      "epoch": 5.767631530761719e-05,
      "model_forward_time": 1.7168285846710205,
      "step": 60478
    },
    {
      "epoch": 5.767631530761719e-05,
      "step": 60478,
      "training_step_time": 6.68991231918335
    },
    {
      "epoch": 5.767726898193359e-05,
      "model_forward_time": 1.7186853885650635,
      "step": 60479
    },
    {
      "epoch": 5.767726898193359e-05,
      "step": 60479,
      "training_step_time": 6.667867183685303
    },
    {
      "epoch": 5.767822265625e-05,
      "grad_norm": 0.09250069409608841,
      "learning_rate": 3.695952092203854e-06,
      "loss": 0.0607,
      "step": 60480
    },
    {
      "epoch": 5.767822265625e-05,
      "model_forward_time": 1.7148728370666504,
      "step": 60480
    },
    {
      "dynamics_loss_avg": 0.05835921745747328,
      "epoch": 5.767822265625e-05,
      "step": 60480
    },
    {
      "action_loss_avg": 0.0015182946517597884,
      "epoch": 5.767822265625e-05,
      "step": 60480
    },
    {
      "epoch": 5.767822265625e-05,
      "step": 60480,
      "training_step_time": 6.720550537109375
    },
    {
      "epoch": 5.767917633056641e-05,
      "model_forward_time": 1.7240676879882812,
      "step": 60481
    },
    {
      "epoch": 5.767917633056641e-05,
      "step": 60481,
      "training_step_time": 6.709585189819336
    },
    {
      "epoch": 5.768013000488281e-05,
      "model_forward_time": 1.7289974689483643,
      "step": 60482
    },
    {
      "epoch": 5.768013000488281e-05,
      "step": 60482,
      "training_step_time": 6.690416097640991
    },
    {
      "epoch": 5.7681083679199216e-05,
      "model_forward_time": 1.7078676223754883,
      "step": 60483
    },
    {
      "epoch": 5.7681083679199216e-05,
      "step": 60483,
      "training_step_time": 6.631466388702393
    },
    {
      "epoch": 5.768203735351563e-05,
      "model_forward_time": 2.938007116317749,
      "step": 60484
    },
    {
      "epoch": 5.768203735351563e-05,
      "step": 60484,
      "training_step_time": 8.430521011352539
    },
    {
      "epoch": 5.768299102783203e-05,
      "model_forward_time": 1.723433017730713,
      "step": 60485
    },
    {
      "epoch": 5.768299102783203e-05,
      "step": 60485,
      "training_step_time": 8.630553007125854
    },
    {
      "epoch": 5.7683944702148435e-05,
      "model_forward_time": 1.716207504272461,
      "step": 60486
    },
    {
      "epoch": 5.7683944702148435e-05,
      "step": 60486,
      "training_step_time": 8.506019830703735
    },
    {
      "epoch": 5.7684898376464846e-05,
      "model_forward_time": 1.7159795761108398,
      "step": 60487
    },
    {
      "epoch": 5.7684898376464846e-05,
      "step": 60487,
      "training_step_time": 7.6278111934661865
    },
    {
      "epoch": 5.768585205078125e-05,
      "model_forward_time": 1.7274482250213623,
      "step": 60488
    },
    {
      "epoch": 5.768585205078125e-05,
      "step": 60488,
      "training_step_time": 6.691934823989868
    },
    {
      "epoch": 5.7686805725097654e-05,
      "model_forward_time": 1.7170827388763428,
      "step": 60489
    },
    {
      "epoch": 5.7686805725097654e-05,
      "step": 60489,
      "training_step_time": 6.660796165466309
    },
    {
      "epoch": 5.7687759399414064e-05,
      "grad_norm": 0.13632239401340485,
      "learning_rate": 3.6943559198732105e-06,
      "loss": 0.0577,
      "step": 60490
    },
    {
      "epoch": 5.7687759399414064e-05,
      "model_forward_time": 1.7133727073669434,
      "step": 60490
    },
    {
      "dynamics_loss_avg": 0.04868452791124582,
      "epoch": 5.7687759399414064e-05,
      "step": 60490
    },
    {
      "action_loss_avg": 0.0015306704910472036,
      "epoch": 5.7687759399414064e-05,
      "step": 60490
    },
    {
      "epoch": 5.7687759399414064e-05,
      "step": 60490,
      "training_step_time": 6.696350336074829
    },
    {
      "epoch": 5.768871307373047e-05,
      "model_forward_time": 1.7184381484985352,
      "step": 60491
    },
    {
      "epoch": 5.768871307373047e-05,
      "step": 60491,
      "training_step_time": 6.694147109985352
    },
    {
      "epoch": 5.768966674804687e-05,
      "model_forward_time": 1.7164862155914307,
      "step": 60492
    },
    {
      "epoch": 5.768966674804687e-05,
      "step": 60492,
      "training_step_time": 6.716846466064453
    },
    {
      "epoch": 5.769062042236328e-05,
      "model_forward_time": 2.9404122829437256,
      "step": 60493
    },
    {
      "epoch": 5.769062042236328e-05,
      "step": 60493,
      "training_step_time": 7.676170825958252
    },
    {
      "epoch": 5.769157409667969e-05,
      "model_forward_time": 1.7437493801116943,
      "step": 60494
    },
    {
      "epoch": 5.769157409667969e-05,
      "step": 60494,
      "training_step_time": 9.675042152404785
    },
    {
      "epoch": 5.769252777099609e-05,
      "model_forward_time": 1.7144172191619873,
      "step": 60495
    },
    {
      "epoch": 5.769252777099609e-05,
      "step": 60495,
      "training_step_time": 7.659222364425659
    },
    {
      "epoch": 5.76934814453125e-05,
      "model_forward_time": 1.7272489070892334,
      "step": 60496
    },
    {
      "epoch": 5.76934814453125e-05,
      "step": 60496,
      "training_step_time": 7.708107948303223
    },
    {
      "epoch": 5.7694435119628906e-05,
      "model_forward_time": 1.7190227508544922,
      "step": 60497
    },
    {
      "epoch": 5.7694435119628906e-05,
      "step": 60497,
      "training_step_time": 6.722299814224243
    },
    {
      "epoch": 5.769538879394531e-05,
      "model_forward_time": 1.7210278511047363,
      "step": 60498
    },
    {
      "epoch": 5.769538879394531e-05,
      "step": 60498,
      "training_step_time": 6.729981184005737
    },
    {
      "epoch": 5.769634246826172e-05,
      "model_forward_time": 1.7207844257354736,
      "step": 60499
    },
    {
      "epoch": 5.769634246826172e-05,
      "step": 60499,
      "training_step_time": 6.695970058441162
    },
    {
      "epoch": 5.7697296142578125e-05,
      "grad_norm": 0.10269788652658463,
      "learning_rate": 3.6927598903258375e-06,
      "loss": 0.0558,
      "step": 60500
    },
    {
      "epoch": 5.7697296142578125e-05,
      "model_forward_time": 1.7172186374664307,
      "step": 60500
    },
    {
      "dynamics_loss_avg": 0.05324892606586218,
      "epoch": 5.7697296142578125e-05,
      "step": 60500
    },
    {
      "action_loss_avg": 0.0017220778856426478,
      "epoch": 5.7697296142578125e-05,
      "step": 60500
    },
    {
      "epoch": 5.7697296142578125e-05,
      "step": 60500,
      "training_step_time": 6.694529294967651
    },
    {
      "epoch": 5.769824981689453e-05,
      "model_forward_time": 1.726349115371704,
      "step": 60501
    },
    {
      "epoch": 5.769824981689453e-05,
      "step": 60501,
      "training_step_time": 6.701395511627197
    },
    {
      "epoch": 5.769920349121094e-05,
      "model_forward_time": 2.846269369125366,
      "step": 60502
    },
    {
      "epoch": 5.769920349121094e-05,
      "step": 60502,
      "training_step_time": 8.63611364364624
    },
    {
      "epoch": 5.7700157165527344e-05,
      "model_forward_time": 1.7370171546936035,
      "step": 60503
    },
    {
      "epoch": 5.7700157165527344e-05,
      "step": 60503,
      "training_step_time": 8.40698528289795
    },
    {
      "epoch": 5.770111083984375e-05,
      "model_forward_time": 1.726579189300537,
      "step": 60504
    },
    {
      "epoch": 5.770111083984375e-05,
      "step": 60504,
      "training_step_time": 8.286364078521729
    },
    {
      "epoch": 5.770206451416016e-05,
      "model_forward_time": 1.7189085483551025,
      "step": 60505
    },
    {
      "epoch": 5.770206451416016e-05,
      "step": 60505,
      "training_step_time": 7.683604955673218
    },
    {
      "epoch": 5.770301818847656e-05,
      "model_forward_time": 1.7199513912200928,
      "step": 60506
    },
    {
      "epoch": 5.770301818847656e-05,
      "step": 60506,
      "training_step_time": 6.73004937171936
    },
    {
      "epoch": 5.770397186279297e-05,
      "model_forward_time": 1.7228662967681885,
      "step": 60507
    },
    {
      "epoch": 5.770397186279297e-05,
      "step": 60507,
      "training_step_time": 6.760867118835449
    },
    {
      "epoch": 5.770492553710938e-05,
      "model_forward_time": 1.724701166152954,
      "step": 60508
    },
    {
      "epoch": 5.770492553710938e-05,
      "step": 60508,
      "training_step_time": 6.769689083099365
    },
    {
      "epoch": 5.770587921142578e-05,
      "model_forward_time": 1.7178544998168945,
      "step": 60509
    },
    {
      "epoch": 5.770587921142578e-05,
      "step": 60509,
      "training_step_time": 6.705940008163452
    },
    {
      "epoch": 5.7706832885742186e-05,
      "grad_norm": 0.09428907185792923,
      "learning_rate": 3.691164003736281e-06,
      "loss": 0.0543,
      "step": 60510
    },
    {
      "epoch": 5.7706832885742186e-05,
      "model_forward_time": 1.7139039039611816,
      "step": 60510
    },
    {
      "dynamics_loss_avg": 0.062822219543159,
      "epoch": 5.7706832885742186e-05,
      "step": 60510
    },
    {
      "action_loss_avg": 0.001711608615005389,
      "epoch": 5.7706832885742186e-05,
      "step": 60510
    },
    {
      "epoch": 5.7706832885742186e-05,
      "step": 60510,
      "training_step_time": 6.691743612289429
    },
    {
      "epoch": 5.7707786560058596e-05,
      "model_forward_time": 1.714768648147583,
      "step": 60511
    },
    {
      "epoch": 5.7707786560058596e-05,
      "step": 60511,
      "training_step_time": 8.057142496109009
    },
    {
      "epoch": 5.7708740234375e-05,
      "model_forward_time": 1.026712417602539,
      "step": 60512
    },
    {
      "epoch": 5.7708740234375e-05,
      "step": 60512,
      "training_step_time": 9.316522598266602
    },
    {
      "epoch": 5.7709693908691404e-05,
      "model_forward_time": 1.7225675582885742,
      "step": 60513
    },
    {
      "epoch": 5.7709693908691404e-05,
      "step": 60513,
      "training_step_time": 8.553135871887207
    },
    {
      "epoch": 5.7710647583007815e-05,
      "model_forward_time": 1.7245690822601318,
      "step": 60514
    },
    {
      "epoch": 5.7710647583007815e-05,
      "step": 60514,
      "training_step_time": 7.6731345653533936
    },
    {
      "epoch": 5.771160125732422e-05,
      "model_forward_time": 1.7152557373046875,
      "step": 60515
    },
    {
      "epoch": 5.771160125732422e-05,
      "step": 60515,
      "training_step_time": 6.732382535934448
    },
    {
      "epoch": 5.771255493164062e-05,
      "model_forward_time": 1.71246337890625,
      "step": 60516
    },
    {
      "epoch": 5.771255493164062e-05,
      "step": 60516,
      "training_step_time": 6.747687101364136
    },
    {
      "epoch": 5.7713508605957034e-05,
      "model_forward_time": 1.714902639389038,
      "step": 60517
    },
    {
      "epoch": 5.7713508605957034e-05,
      "step": 60517,
      "training_step_time": 6.740077018737793
    },
    {
      "epoch": 5.771446228027344e-05,
      "model_forward_time": 1.3395168781280518,
      "step": 60518
    },
    {
      "epoch": 5.771446228027344e-05,
      "step": 60518,
      "training_step_time": 6.737792253494263
    },
    {
      "epoch": 5.771541595458984e-05,
      "model_forward_time": 1.7221755981445312,
      "step": 60519
    },
    {
      "epoch": 5.771541595458984e-05,
      "step": 60519,
      "training_step_time": 6.7205305099487305
    },
    {
      "epoch": 5.771636962890625e-05,
      "grad_norm": 0.1294325590133667,
      "learning_rate": 3.689568260279061e-06,
      "loss": 0.0552,
      "step": 60520
    },
    {
      "epoch": 5.771636962890625e-05,
      "model_forward_time": 1.7241134643554688,
      "step": 60520
    },
    {
      "dynamics_loss_avg": 0.06307478938251734,
      "epoch": 5.771636962890625e-05,
      "step": 60520
    },
    {
      "action_loss_avg": 0.0012948290444910527,
      "epoch": 5.771636962890625e-05,
      "step": 60520
    },
    {
      "epoch": 5.771636962890625e-05,
      "step": 60520,
      "training_step_time": 7.52587890625
    },
    {
      "epoch": 5.771732330322266e-05,
      "model_forward_time": 2.81266188621521,
      "step": 60521
    },
    {
      "epoch": 5.771732330322266e-05,
      "step": 60521,
      "training_step_time": 9.391989707946777
    },
    {
      "epoch": 5.771827697753906e-05,
      "model_forward_time": 1.7241313457489014,
      "step": 60522
    },
    {
      "epoch": 5.771827697753906e-05,
      "step": 60522,
      "training_step_time": 8.368235111236572
    },
    {
      "epoch": 5.771923065185547e-05,
      "model_forward_time": 1.7105519771575928,
      "step": 60523
    },
    {
      "epoch": 5.771923065185547e-05,
      "step": 60523,
      "training_step_time": 7.661381721496582
    },
    {
      "epoch": 5.7720184326171876e-05,
      "model_forward_time": 1.7186291217803955,
      "step": 60524
    },
    {
      "epoch": 5.7720184326171876e-05,
      "step": 60524,
      "training_step_time": 7.22427773475647
    },
    {
      "epoch": 5.772113800048828e-05,
      "model_forward_time": 1.7295336723327637,
      "step": 60525
    },
    {
      "epoch": 5.772113800048828e-05,
      "step": 60525,
      "training_step_time": 6.764924049377441
    },
    {
      "epoch": 5.772209167480469e-05,
      "model_forward_time": 1.7228500843048096,
      "step": 60526
    },
    {
      "epoch": 5.772209167480469e-05,
      "step": 60526,
      "training_step_time": 6.650313377380371
    },
    {
      "epoch": 5.7723045349121095e-05,
      "model_forward_time": 1.7161099910736084,
      "step": 60527
    },
    {
      "epoch": 5.7723045349121095e-05,
      "step": 60527,
      "training_step_time": 6.709185838699341
    },
    {
      "epoch": 5.77239990234375e-05,
      "model_forward_time": 1.7218017578125,
      "step": 60528
    },
    {
      "epoch": 5.77239990234375e-05,
      "step": 60528,
      "training_step_time": 6.670399188995361
    },
    {
      "epoch": 5.772495269775391e-05,
      "model_forward_time": 1.7223272323608398,
      "step": 60529
    },
    {
      "epoch": 5.772495269775391e-05,
      "step": 60529,
      "training_step_time": 7.470505714416504
    },
    {
      "epoch": 5.7725906372070314e-05,
      "grad_norm": 0.11765953153371811,
      "learning_rate": 3.687972660128686e-06,
      "loss": 0.0569,
      "step": 60530
    },
    {
      "epoch": 5.7725906372070314e-05,
      "model_forward_time": 2.833533763885498,
      "step": 60530
    },
    {
      "dynamics_loss_avg": 0.046803155541419984,
      "epoch": 5.7725906372070314e-05,
      "step": 60530
    },
    {
      "action_loss_avg": 0.0018251041969051585,
      "epoch": 5.7725906372070314e-05,
      "step": 60530
    },
    {
      "epoch": 5.7725906372070314e-05,
      "step": 60530,
      "training_step_time": 8.588013887405396
    },
    {
      "epoch": 5.772686004638672e-05,
      "model_forward_time": 1.83782958984375,
      "step": 60531
    },
    {
      "epoch": 5.772686004638672e-05,
      "step": 60531,
      "training_step_time": 8.675974369049072
    },
    {
      "epoch": 5.772781372070312e-05,
      "model_forward_time": 1.724665641784668,
      "step": 60532
    },
    {
      "epoch": 5.772781372070312e-05,
      "step": 60532,
      "training_step_time": 8.177452325820923
    },
    {
      "epoch": 5.772876739501953e-05,
      "model_forward_time": 1.730545997619629,
      "step": 60533
    },
    {
      "epoch": 5.772876739501953e-05,
      "step": 60533,
      "training_step_time": 7.5920164585113525
    },
    {
      "epoch": 5.7729721069335936e-05,
      "model_forward_time": 1.8121273517608643,
      "step": 60534
    },
    {
      "epoch": 5.7729721069335936e-05,
      "step": 60534,
      "training_step_time": 6.680370569229126
    },
    {
      "epoch": 5.773067474365234e-05,
      "model_forward_time": 1.7318196296691895,
      "step": 60535
    },
    {
      "epoch": 5.773067474365234e-05,
      "step": 60535,
      "training_step_time": 6.743337154388428
    },
    {
      "epoch": 5.773162841796875e-05,
      "model_forward_time": 1.7301273345947266,
      "step": 60536
    },
    {
      "epoch": 5.773162841796875e-05,
      "step": 60536,
      "training_step_time": 6.718837261199951
    },
    {
      "epoch": 5.7732582092285155e-05,
      "model_forward_time": 1.738832950592041,
      "step": 60537
    },
    {
      "epoch": 5.7732582092285155e-05,
      "step": 60537,
      "training_step_time": 6.66589879989624
    },
    {
      "epoch": 5.773353576660156e-05,
      "model_forward_time": 1.7300662994384766,
      "step": 60538
    },
    {
      "epoch": 5.773353576660156e-05,
      "step": 60538,
      "training_step_time": 7.4371726512908936
    },
    {
      "epoch": 5.773448944091797e-05,
      "model_forward_time": 1.7281019687652588,
      "step": 60539
    },
    {
      "epoch": 5.773448944091797e-05,
      "step": 60539,
      "training_step_time": 8.057910919189453
    },
    {
      "epoch": 5.7735443115234374e-05,
      "grad_norm": 0.11191410571336746,
      "learning_rate": 3.686377203459648e-06,
      "loss": 0.0562,
      "step": 60540
    },
    {
      "epoch": 5.7735443115234374e-05,
      "model_forward_time": 1.3663139343261719,
      "step": 60540
    },
    {
      "dynamics_loss_avg": 0.03892779499292374,
      "epoch": 5.7735443115234374e-05,
      "step": 60540
    },
    {
      "action_loss_avg": 0.0011860737344250084,
      "epoch": 5.7735443115234374e-05,
      "step": 60540
    },
    {
      "epoch": 5.7735443115234374e-05,
      "step": 60540,
      "training_step_time": 10.371277570724487
    },
    {
      "epoch": 5.773639678955078e-05,
      "model_forward_time": 1.729893445968628,
      "step": 60541
    },
    {
      "epoch": 5.773639678955078e-05,
      "step": 60541,
      "training_step_time": 9.616486549377441
    },
    {
      "epoch": 5.773735046386719e-05,
      "model_forward_time": 1.733649492263794,
      "step": 60542
    },
    {
      "epoch": 5.773735046386719e-05,
      "step": 60542,
      "training_step_time": 7.622318267822266
    },
    {
      "epoch": 5.773830413818359e-05,
      "model_forward_time": 1.731276273727417,
      "step": 60543
    },
    {
      "epoch": 5.773830413818359e-05,
      "step": 60543,
      "training_step_time": 6.968891382217407
    },
    {
      "epoch": 5.77392578125e-05,
      "model_forward_time": 1.730952501296997,
      "step": 60544
    },
    {
      "epoch": 5.77392578125e-05,
      "step": 60544,
      "training_step_time": 6.745204210281372
    },
    {
      "epoch": 5.774021148681641e-05,
      "model_forward_time": 1.7852990627288818,
      "step": 60545
    },
    {
      "epoch": 5.774021148681641e-05,
      "step": 60545,
      "training_step_time": 6.7014830112457275
    },
    {
      "epoch": 5.774116516113281e-05,
      "model_forward_time": 1.7376508712768555,
      "step": 60546
    },
    {
      "epoch": 5.774116516113281e-05,
      "step": 60546,
      "training_step_time": 6.731729984283447
    },
    {
      "epoch": 5.7742118835449216e-05,
      "model_forward_time": 1.733250617980957,
      "step": 60547
    },
    {
      "epoch": 5.7742118835449216e-05,
      "step": 60547,
      "training_step_time": 7.511581182479858
    },
    {
      "epoch": 5.774307250976563e-05,
      "model_forward_time": 1.3644194602966309,
      "step": 60548
    },
    {
      "epoch": 5.774307250976563e-05,
      "step": 60548,
      "training_step_time": 8.896250009536743
    },
    {
      "epoch": 5.774402618408203e-05,
      "model_forward_time": 2.9228904247283936,
      "step": 60549
    },
    {
      "epoch": 5.774402618408203e-05,
      "step": 60549,
      "training_step_time": 10.492174625396729
    },
    {
      "epoch": 5.7744979858398435e-05,
      "grad_norm": 0.09193909913301468,
      "learning_rate": 3.684781890446423e-06,
      "loss": 0.0582,
      "step": 60550
    },
    {
      "epoch": 5.7744979858398435e-05,
      "model_forward_time": 1.8477628231048584,
      "step": 60550
    },
    {
      "dynamics_loss_avg": 0.04818862751126289,
      "epoch": 5.7744979858398435e-05,
      "step": 60550
    },
    {
      "action_loss_avg": 0.0014161647413857281,
      "epoch": 5.7744979858398435e-05,
      "step": 60550
    },
    {
      "epoch": 5.7744979858398435e-05,
      "step": 60550,
      "training_step_time": 7.841696262359619
    },
    {
      "epoch": 5.7745933532714846e-05,
      "model_forward_time": 1.7450435161590576,
      "step": 60551
    },
    {
      "epoch": 5.7745933532714846e-05,
      "step": 60551,
      "training_step_time": 7.694695711135864
    },
    {
      "epoch": 5.774688720703125e-05,
      "model_forward_time": 1.72731351852417,
      "step": 60552
    },
    {
      "epoch": 5.774688720703125e-05,
      "step": 60552,
      "training_step_time": 7.3133604526519775
    },
    {
      "epoch": 5.7747840881347654e-05,
      "model_forward_time": 1.7298359870910645,
      "step": 60553
    },
    {
      "epoch": 5.7747840881347654e-05,
      "step": 60553,
      "training_step_time": 6.751551389694214
    },
    {
      "epoch": 5.7748794555664064e-05,
      "model_forward_time": 1.7294063568115234,
      "step": 60554
    },
    {
      "epoch": 5.7748794555664064e-05,
      "step": 60554,
      "training_step_time": 6.711265802383423
    },
    {
      "epoch": 5.774974822998047e-05,
      "model_forward_time": 1.7424335479736328,
      "step": 60555
    },
    {
      "epoch": 5.774974822998047e-05,
      "step": 60555,
      "training_step_time": 6.68005895614624
    },
    {
      "epoch": 5.775070190429687e-05,
      "model_forward_time": 1.760988712310791,
      "step": 60556
    },
    {
      "epoch": 5.775070190429687e-05,
      "step": 60556,
      "training_step_time": 7.509100675582886
    },
    {
      "epoch": 5.775165557861328e-05,
      "model_forward_time": 1.7307617664337158,
      "step": 60557
    },
    {
      "epoch": 5.775165557861328e-05,
      "step": 60557,
      "training_step_time": 7.518160104751587
    },
    {
      "epoch": 5.775260925292969e-05,
      "model_forward_time": 2.9229354858398438,
      "step": 60558
    },
    {
      "epoch": 5.775260925292969e-05,
      "step": 60558,
      "training_step_time": 8.698357582092285
    },
    {
      "epoch": 5.775356292724609e-05,
      "model_forward_time": 1.850823163986206,
      "step": 60559
    },
    {
      "epoch": 5.775356292724609e-05,
      "step": 60559,
      "training_step_time": 7.74196457862854
    },
    {
      "epoch": 5.77545166015625e-05,
      "grad_norm": 0.09811419248580933,
      "learning_rate": 3.683186721263474e-06,
      "loss": 0.055,
      "step": 60560
    },
    {
      "epoch": 5.77545166015625e-05,
      "model_forward_time": 1.7304391860961914,
      "step": 60560
    },
    {
      "dynamics_loss_avg": 0.05038648098707199,
      "epoch": 5.77545166015625e-05,
      "step": 60560
    },
    {
      "action_loss_avg": 0.002181147295050323,
      "epoch": 5.77545166015625e-05,
      "step": 60560
    },
    {
      "epoch": 5.77545166015625e-05,
      "step": 60560,
      "training_step_time": 7.687963008880615
    },
    {
      "epoch": 5.7755470275878906e-05,
      "model_forward_time": 1.7354109287261963,
      "step": 60561
    },
    {
      "epoch": 5.7755470275878906e-05,
      "step": 60561,
      "training_step_time": 7.365119457244873
    },
    {
      "epoch": 5.775642395019531e-05,
      "model_forward_time": 1.7355866432189941,
      "step": 60562
    },
    {
      "epoch": 5.775642395019531e-05,
      "step": 60562,
      "training_step_time": 6.677404880523682
    },
    {
      "epoch": 5.775737762451172e-05,
      "model_forward_time": 1.7192456722259521,
      "step": 60563
    },
    {
      "epoch": 5.775737762451172e-05,
      "step": 60563,
      "training_step_time": 6.685489892959595
    },
    {
      "epoch": 5.7758331298828125e-05,
      "model_forward_time": 1.7374608516693115,
      "step": 60564
    },
    {
      "epoch": 5.7758331298828125e-05,
      "step": 60564,
      "training_step_time": 6.6863954067230225
    },
    {
      "epoch": 5.775928497314453e-05,
      "model_forward_time": 1.3731834888458252,
      "step": 60565
    },
    {
      "epoch": 5.775928497314453e-05,
      "step": 60565,
      "training_step_time": 7.581370830535889
    },
    {
      "epoch": 5.776023864746094e-05,
      "model_forward_time": 1.7431371212005615,
      "step": 60566
    },
    {
      "epoch": 5.776023864746094e-05,
      "step": 60566,
      "training_step_time": 7.5806403160095215
    },
    {
      "epoch": 5.7761192321777344e-05,
      "model_forward_time": 1.7282371520996094,
      "step": 60567
    },
    {
      "epoch": 5.7761192321777344e-05,
      "step": 60567,
      "training_step_time": 9.406980276107788
    },
    {
      "epoch": 5.776214599609375e-05,
      "model_forward_time": 1.8359911441802979,
      "step": 60568
    },
    {
      "epoch": 5.776214599609375e-05,
      "step": 60568,
      "training_step_time": 8.45590090751648
    },
    {
      "epoch": 5.776309967041016e-05,
      "model_forward_time": 1.7271699905395508,
      "step": 60569
    },
    {
      "epoch": 5.776309967041016e-05,
      "step": 60569,
      "training_step_time": 7.617146968841553
    },
    {
      "epoch": 5.776405334472656e-05,
      "grad_norm": 0.1264875829219818,
      "learning_rate": 3.6815916960852437e-06,
      "loss": 0.0581,
      "step": 60570
    },
    {
      "epoch": 5.776405334472656e-05,
      "model_forward_time": 1.3557350635528564,
      "step": 60570
    },
    {
      "dynamics_loss_avg": 0.06544832922518254,
      "epoch": 5.776405334472656e-05,
      "step": 60570
    },
    {
      "action_loss_avg": 0.0013133340020431206,
      "epoch": 5.776405334472656e-05,
      "step": 60570
    },
    {
      "epoch": 5.776405334472656e-05,
      "step": 60570,
      "training_step_time": 7.806415557861328
    },
    {
      "epoch": 5.776500701904297e-05,
      "model_forward_time": 1.7349720001220703,
      "step": 60571
    },
    {
      "epoch": 5.776500701904297e-05,
      "step": 60571,
      "training_step_time": 6.6669020652771
    },
    {
      "epoch": 5.776596069335938e-05,
      "model_forward_time": 1.7421770095825195,
      "step": 60572
    },
    {
      "epoch": 5.776596069335938e-05,
      "step": 60572,
      "training_step_time": 6.66109299659729
    },
    {
      "epoch": 5.776691436767578e-05,
      "model_forward_time": 1.7320556640625,
      "step": 60573
    },
    {
      "epoch": 5.776691436767578e-05,
      "step": 60573,
      "training_step_time": 6.660008192062378
    },
    {
      "epoch": 5.7767868041992186e-05,
      "model_forward_time": 1.7397267818450928,
      "step": 60574
    },
    {
      "epoch": 5.7767868041992186e-05,
      "step": 60574,
      "training_step_time": 7.5315117835998535
    },
    {
      "epoch": 5.7768821716308596e-05,
      "model_forward_time": 1.7377276420593262,
      "step": 60575
    },
    {
      "epoch": 5.7768821716308596e-05,
      "step": 60575,
      "training_step_time": 8.804916143417358
    },
    {
      "epoch": 5.7769775390625e-05,
      "model_forward_time": 1.769233226776123,
      "step": 60576
    },
    {
      "epoch": 5.7769775390625e-05,
      "step": 60576,
      "training_step_time": 8.299591064453125
    },
    {
      "epoch": 5.7770729064941405e-05,
      "model_forward_time": 2.9153990745544434,
      "step": 60577
    },
    {
      "epoch": 5.7770729064941405e-05,
      "step": 60577,
      "training_step_time": 10.65047550201416
    },
    {
      "epoch": 5.7771682739257815e-05,
      "model_forward_time": 1.7309236526489258,
      "step": 60578
    },
    {
      "epoch": 5.7771682739257815e-05,
      "step": 60578,
      "training_step_time": 8.816688060760498
    },
    {
      "epoch": 5.777263641357422e-05,
      "model_forward_time": 1.7328143119812012,
      "step": 60579
    },
    {
      "epoch": 5.777263641357422e-05,
      "step": 60579,
      "training_step_time": 7.8621790409088135
    },
    {
      "epoch": 5.777359008789062e-05,
      "grad_norm": 0.11860758811235428,
      "learning_rate": 3.679996815086165e-06,
      "loss": 0.0601,
      "step": 60580
    },
    {
      "epoch": 5.777359008789062e-05,
      "model_forward_time": 1.730226755142212,
      "step": 60580
    },
    {
      "dynamics_loss_avg": 0.06305308192968369,
      "epoch": 5.777359008789062e-05,
      "step": 60580
    },
    {
      "action_loss_avg": 0.0018794636358506977,
      "epoch": 5.777359008789062e-05,
      "step": 60580
    },
    {
      "epoch": 5.777359008789062e-05,
      "step": 60580,
      "training_step_time": 7.299935340881348
    },
    {
      "epoch": 5.7774543762207034e-05,
      "model_forward_time": 1.7437973022460938,
      "step": 60581
    },
    {
      "epoch": 5.7774543762207034e-05,
      "step": 60581,
      "training_step_time": 6.709692716598511
    },
    {
      "epoch": 5.777549743652344e-05,
      "model_forward_time": 1.7780413627624512,
      "step": 60582
    },
    {
      "epoch": 5.777549743652344e-05,
      "step": 60582,
      "training_step_time": 6.685393571853638
    },
    {
      "epoch": 5.777645111083984e-05,
      "model_forward_time": 1.7768402099609375,
      "step": 60583
    },
    {
      "epoch": 5.777645111083984e-05,
      "step": 60583,
      "training_step_time": 7.550370931625366
    },
    {
      "epoch": 5.777740478515625e-05,
      "model_forward_time": 1.738044261932373,
      "step": 60584
    },
    {
      "epoch": 5.777740478515625e-05,
      "step": 60584,
      "training_step_time": 9.17336392402649
    },
    {
      "epoch": 5.777835845947266e-05,
      "model_forward_time": 1.7304575443267822,
      "step": 60585
    },
    {
      "epoch": 5.777835845947266e-05,
      "step": 60585,
      "training_step_time": 7.634007930755615
    },
    {
      "epoch": 5.777931213378906e-05,
      "model_forward_time": 2.891019821166992,
      "step": 60586
    },
    {
      "epoch": 5.777931213378906e-05,
      "step": 60586,
      "training_step_time": 10.212589502334595
    },
    {
      "epoch": 5.778026580810547e-05,
      "model_forward_time": 1.8311572074890137,
      "step": 60587
    },
    {
      "epoch": 5.778026580810547e-05,
      "step": 60587,
      "training_step_time": 8.724688291549683
    },
    {
      "epoch": 5.7781219482421876e-05,
      "model_forward_time": 1.734415054321289,
      "step": 60588
    },
    {
      "epoch": 5.7781219482421876e-05,
      "step": 60588,
      "training_step_time": 7.823408842086792
    },
    {
      "epoch": 5.778217315673828e-05,
      "model_forward_time": 1.7292683124542236,
      "step": 60589
    },
    {
      "epoch": 5.778217315673828e-05,
      "step": 60589,
      "training_step_time": 7.294755220413208
    },
    {
      "epoch": 5.778312683105469e-05,
      "grad_norm": 0.08097480982542038,
      "learning_rate": 3.6784020784406493e-06,
      "loss": 0.0556,
      "step": 60590
    },
    {
      "epoch": 5.778312683105469e-05,
      "model_forward_time": 1.7295992374420166,
      "step": 60590
    },
    {
      "dynamics_loss_avg": 0.052230176702141765,
      "epoch": 5.778312683105469e-05,
      "step": 60590
    },
    {
      "action_loss_avg": 0.0019333978940267115,
      "epoch": 5.778312683105469e-05,
      "step": 60590
    },
    {
      "epoch": 5.778312683105469e-05,
      "step": 60590,
      "training_step_time": 6.718755483627319
    },
    {
      "epoch": 5.7784080505371095e-05,
      "model_forward_time": 1.743431806564331,
      "step": 60591
    },
    {
      "epoch": 5.7784080505371095e-05,
      "step": 60591,
      "training_step_time": 6.734187126159668
    },
    {
      "epoch": 5.77850341796875e-05,
      "model_forward_time": 1.726224422454834,
      "step": 60592
    },
    {
      "epoch": 5.77850341796875e-05,
      "step": 60592,
      "training_step_time": 7.554880857467651
    },
    {
      "epoch": 5.778598785400391e-05,
      "model_forward_time": 1.7213521003723145,
      "step": 60593
    },
    {
      "epoch": 5.778598785400391e-05,
      "step": 60593,
      "training_step_time": 7.519233465194702
    },
    {
      "epoch": 5.7786941528320314e-05,
      "model_forward_time": 1.7669508457183838,
      "step": 60594
    },
    {
      "epoch": 5.7786941528320314e-05,
      "step": 60594,
      "training_step_time": 8.403550386428833
    },
    {
      "epoch": 5.778789520263672e-05,
      "model_forward_time": 2.8660802841186523,
      "step": 60595
    },
    {
      "epoch": 5.778789520263672e-05,
      "step": 60595,
      "training_step_time": 9.434926986694336
    },
    {
      "epoch": 5.778884887695312e-05,
      "model_forward_time": 1.7419030666351318,
      "step": 60596
    },
    {
      "epoch": 5.778884887695312e-05,
      "step": 60596,
      "training_step_time": 7.748141527175903
    },
    {
      "epoch": 5.778980255126953e-05,
      "model_forward_time": 1.7336246967315674,
      "step": 60597
    },
    {
      "epoch": 5.778980255126953e-05,
      "step": 60597,
      "training_step_time": 8.619637727737427
    },
    {
      "epoch": 5.7790756225585937e-05,
      "model_forward_time": 1.7267696857452393,
      "step": 60598
    },
    {
      "epoch": 5.7790756225585937e-05,
      "step": 60598,
      "training_step_time": 7.8642261028289795
    },
    {
      "epoch": 5.779170989990234e-05,
      "model_forward_time": 1.7172739505767822,
      "step": 60599
    },
    {
      "epoch": 5.779170989990234e-05,
      "step": 60599,
      "training_step_time": 6.679141998291016
    },
    {
      "epoch": 5.779266357421875e-05,
      "grad_norm": 0.10095221549272537,
      "learning_rate": 3.6768074863230964e-06,
      "loss": 0.0543,
      "step": 60600
    },
    {
      "epoch": 5.779266357421875e-05,
      "model_forward_time": 1.746558427810669,
      "step": 60600
    },
    {
      "dynamics_loss_avg": 0.03886777283623814,
      "epoch": 5.779266357421875e-05,
      "step": 60600
    },
    {
      "action_loss_avg": 0.002255518423044123,
      "epoch": 5.779266357421875e-05,
      "step": 60600
    },
    {
      "epoch": 5.779266357421875e-05,
      "step": 60600,
      "training_step_time": 6.696367979049683
    },
    {
      "epoch": 5.7793617248535155e-05,
      "model_forward_time": 1.855179786682129,
      "step": 60601
    },
    {
      "epoch": 5.7793617248535155e-05,
      "step": 60601,
      "training_step_time": 7.540977954864502
    },
    {
      "epoch": 5.779457092285156e-05,
      "model_forward_time": 1.7238731384277344,
      "step": 60602
    },
    {
      "epoch": 5.779457092285156e-05,
      "step": 60602,
      "training_step_time": 8.412804126739502
    },
    {
      "epoch": 5.779552459716797e-05,
      "model_forward_time": 1.7203786373138428,
      "step": 60603
    },
    {
      "epoch": 5.779552459716797e-05,
      "step": 60603,
      "training_step_time": 7.521648645401001
    },
    {
      "epoch": 5.7796478271484374e-05,
      "model_forward_time": 2.854915142059326,
      "step": 60604
    },
    {
      "epoch": 5.7796478271484374e-05,
      "step": 60604,
      "training_step_time": 7.827101230621338
    },
    {
      "epoch": 5.779743194580078e-05,
      "model_forward_time": 1.7196130752563477,
      "step": 60605
    },
    {
      "epoch": 5.779743194580078e-05,
      "step": 60605,
      "training_step_time": 8.620506525039673
    },
    {
      "epoch": 5.779838562011719e-05,
      "model_forward_time": 1.8125450611114502,
      "step": 60606
    },
    {
      "epoch": 5.779838562011719e-05,
      "step": 60606,
      "training_step_time": 7.730700254440308
    },
    {
      "epoch": 5.779933929443359e-05,
      "model_forward_time": 1.7225263118743896,
      "step": 60607
    },
    {
      "epoch": 5.779933929443359e-05,
      "step": 60607,
      "training_step_time": 7.852258920669556
    },
    {
      "epoch": 5.780029296875e-05,
      "model_forward_time": 1.7227914333343506,
      "step": 60608
    },
    {
      "epoch": 5.780029296875e-05,
      "step": 60608,
      "training_step_time": 7.106937408447266
    },
    {
      "epoch": 5.780124664306641e-05,
      "model_forward_time": 1.7221932411193848,
      "step": 60609
    },
    {
      "epoch": 5.780124664306641e-05,
      "step": 60609,
      "training_step_time": 6.823913335800171
    },
    {
      "epoch": 5.780220031738281e-05,
      "grad_norm": 0.10776527971029282,
      "learning_rate": 3.675213038907887e-06,
      "loss": 0.0574,
      "step": 60610
    },
    {
      "epoch": 5.780220031738281e-05,
      "model_forward_time": 1.7312045097351074,
      "step": 60610
    },
    {
      "dynamics_loss_avg": 0.051906400732696055,
      "epoch": 5.780220031738281e-05,
      "step": 60610
    },
    {
      "action_loss_avg": 0.0011515942984260618,
      "epoch": 5.780220031738281e-05,
      "step": 60610
    },
    {
      "epoch": 5.780220031738281e-05,
      "step": 60610,
      "training_step_time": 7.5123021602630615
    },
    {
      "epoch": 5.7803153991699216e-05,
      "model_forward_time": 1.721968173980713,
      "step": 60611
    },
    {
      "epoch": 5.7803153991699216e-05,
      "step": 60611,
      "training_step_time": 7.254741191864014
    },
    {
      "epoch": 5.780410766601563e-05,
      "model_forward_time": 1.7222907543182373,
      "step": 60612
    },
    {
      "epoch": 5.780410766601563e-05,
      "step": 60612,
      "training_step_time": 8.408920764923096
    },
    {
      "epoch": 5.780506134033203e-05,
      "model_forward_time": 1.737391471862793,
      "step": 60613
    },
    {
      "epoch": 5.780506134033203e-05,
      "step": 60613,
      "training_step_time": 9.459582090377808
    },
    {
      "epoch": 5.7806015014648435e-05,
      "model_forward_time": 1.7259292602539062,
      "step": 60614
    },
    {
      "epoch": 5.7806015014648435e-05,
      "step": 60614,
      "training_step_time": 9.428048849105835
    },
    {
      "epoch": 5.7806968688964846e-05,
      "model_forward_time": 1.789905071258545,
      "step": 60615
    },
    {
      "epoch": 5.7806968688964846e-05,
      "step": 60615,
      "training_step_time": 8.505141019821167
    },
    {
      "epoch": 5.780792236328125e-05,
      "model_forward_time": 1.717219352722168,
      "step": 60616
    },
    {
      "epoch": 5.780792236328125e-05,
      "step": 60616,
      "training_step_time": 7.746393918991089
    },
    {
      "epoch": 5.7808876037597654e-05,
      "model_forward_time": 1.7325046062469482,
      "step": 60617
    },
    {
      "epoch": 5.7808876037597654e-05,
      "step": 60617,
      "training_step_time": 7.071617126464844
    },
    {
      "epoch": 5.7809829711914065e-05,
      "model_forward_time": 1.8075623512268066,
      "step": 60618
    },
    {
      "epoch": 5.7809829711914065e-05,
      "step": 60618,
      "training_step_time": 6.747257232666016
    },
    {
      "epoch": 5.781078338623047e-05,
      "model_forward_time": 1.7363460063934326,
      "step": 60619
    },
    {
      "epoch": 5.781078338623047e-05,
      "step": 60619,
      "training_step_time": 7.590932607650757
    },
    {
      "epoch": 5.781173706054687e-05,
      "grad_norm": 0.10910142213106155,
      "learning_rate": 3.6736187363693877e-06,
      "loss": 0.0551,
      "step": 60620
    },
    {
      "epoch": 5.781173706054687e-05,
      "model_forward_time": 1.7218897342681885,
      "step": 60620
    },
    {
      "dynamics_loss_avg": 0.03957606144249439,
      "epoch": 5.781173706054687e-05,
      "step": 60620
    },
    {
      "action_loss_avg": 0.0012743334926199168,
      "epoch": 5.781173706054687e-05,
      "step": 60620
    },
    {
      "epoch": 5.781173706054687e-05,
      "step": 60620,
      "training_step_time": 7.286769151687622
    },
    {
      "epoch": 5.781269073486328e-05,
      "model_forward_time": 1.7297604084014893,
      "step": 60621
    },
    {
      "epoch": 5.781269073486328e-05,
      "step": 60621,
      "training_step_time": 7.5009965896606445
    },
    {
      "epoch": 5.781364440917969e-05,
      "model_forward_time": 1.722761631011963,
      "step": 60622
    },
    {
      "epoch": 5.781364440917969e-05,
      "step": 60622,
      "training_step_time": 8.914673328399658
    },
    {
      "epoch": 5.781459808349609e-05,
      "model_forward_time": 2.9078619480133057,
      "step": 60623
    },
    {
      "epoch": 5.781459808349609e-05,
      "step": 60623,
      "training_step_time": 7.796198606491089
    },
    {
      "epoch": 5.78155517578125e-05,
      "model_forward_time": 1.7272069454193115,
      "step": 60624
    },
    {
      "epoch": 5.78155517578125e-05,
      "step": 60624,
      "training_step_time": 8.991137504577637
    },
    {
      "epoch": 5.7816505432128906e-05,
      "model_forward_time": 1.8331570625305176,
      "step": 60625
    },
    {
      "epoch": 5.7816505432128906e-05,
      "step": 60625,
      "training_step_time": 7.691772937774658
    },
    {
      "epoch": 5.781745910644531e-05,
      "model_forward_time": 1.7361788749694824,
      "step": 60626
    },
    {
      "epoch": 5.781745910644531e-05,
      "step": 60626,
      "training_step_time": 7.439608812332153
    },
    {
      "epoch": 5.781841278076172e-05,
      "model_forward_time": 1.725282907485962,
      "step": 60627
    },
    {
      "epoch": 5.781841278076172e-05,
      "step": 60627,
      "training_step_time": 6.6771016120910645
    },
    {
      "epoch": 5.7819366455078125e-05,
      "model_forward_time": 1.7341785430908203,
      "step": 60628
    },
    {
      "epoch": 5.7819366455078125e-05,
      "step": 60628,
      "training_step_time": 7.521878957748413
    },
    {
      "epoch": 5.782032012939453e-05,
      "model_forward_time": 1.7180933952331543,
      "step": 60629
    },
    {
      "epoch": 5.782032012939453e-05,
      "step": 60629,
      "training_step_time": 7.996248245239258
    },
    {
      "epoch": 5.782127380371094e-05,
      "grad_norm": 0.1143936961889267,
      "learning_rate": 3.6720245788819474e-06,
      "loss": 0.0552,
      "step": 60630
    },
    {
      "epoch": 5.782127380371094e-05,
      "model_forward_time": 1.734510898590088,
      "step": 60630
    },
    {
      "dynamics_loss_avg": 0.0684721939265728,
      "epoch": 5.782127380371094e-05,
      "step": 60630
    },
    {
      "action_loss_avg": 0.0013969069026643411,
      "epoch": 5.782127380371094e-05,
      "step": 60630
    },
    {
      "epoch": 5.782127380371094e-05,
      "step": 60630,
      "training_step_time": 7.530973434448242
    },
    {
      "epoch": 5.7822227478027344e-05,
      "model_forward_time": 1.718400239944458,
      "step": 60631
    },
    {
      "epoch": 5.7822227478027344e-05,
      "step": 60631,
      "training_step_time": 9.789210319519043
    },
    {
      "epoch": 5.782318115234375e-05,
      "model_forward_time": 2.900012969970703,
      "step": 60632
    },
    {
      "epoch": 5.782318115234375e-05,
      "step": 60632,
      "training_step_time": 9.481659412384033
    },
    {
      "epoch": 5.782413482666016e-05,
      "model_forward_time": 1.7219748497009277,
      "step": 60633
    },
    {
      "epoch": 5.782413482666016e-05,
      "step": 60633,
      "training_step_time": 9.679915428161621
    },
    {
      "epoch": 5.782508850097656e-05,
      "model_forward_time": 1.8281755447387695,
      "step": 60634
    },
    {
      "epoch": 5.782508850097656e-05,
      "step": 60634,
      "training_step_time": 7.780438423156738
    },
    {
      "epoch": 5.782604217529297e-05,
      "model_forward_time": 1.7299866676330566,
      "step": 60635
    },
    {
      "epoch": 5.782604217529297e-05,
      "step": 60635,
      "training_step_time": 7.474283933639526
    },
    {
      "epoch": 5.782699584960938e-05,
      "model_forward_time": 1.7251133918762207,
      "step": 60636
    },
    {
      "epoch": 5.782699584960938e-05,
      "step": 60636,
      "training_step_time": 7.068348169326782
    },
    {
      "epoch": 5.782794952392578e-05,
      "model_forward_time": 1.7233600616455078,
      "step": 60637
    },
    {
      "epoch": 5.782794952392578e-05,
      "step": 60637,
      "training_step_time": 7.563951253890991
    },
    {
      "epoch": 5.7828903198242186e-05,
      "model_forward_time": 1.7227096557617188,
      "step": 60638
    },
    {
      "epoch": 5.7828903198242186e-05,
      "step": 60638,
      "training_step_time": 7.266826868057251
    },
    {
      "epoch": 5.7829856872558597e-05,
      "model_forward_time": 1.3649561405181885,
      "step": 60639
    },
    {
      "epoch": 5.7829856872558597e-05,
      "step": 60639,
      "training_step_time": 7.573932409286499
    },
    {
      "epoch": 5.7830810546875e-05,
      "grad_norm": 0.09907396137714386,
      "learning_rate": 3.6704305666199046e-06,
      "loss": 0.0547,
      "step": 60640
    },
    {
      "epoch": 5.7830810546875e-05,
      "model_forward_time": 1.7318015098571777,
      "step": 60640
    },
    {
      "dynamics_loss_avg": 0.06271616201847792,
      "epoch": 5.7830810546875e-05,
      "step": 60640
    },
    {
      "action_loss_avg": 0.0016067234391812235,
      "epoch": 5.7830810546875e-05,
      "step": 60640
    },
    {
      "epoch": 5.7830810546875e-05,
      "step": 60640,
      "training_step_time": 7.525890588760376
    },
    {
      "epoch": 5.7831764221191405e-05,
      "model_forward_time": 2.8887135982513428,
      "step": 60641
    },
    {
      "epoch": 5.7831764221191405e-05,
      "step": 60641,
      "training_step_time": 9.426154851913452
    },
    {
      "epoch": 5.7832717895507815e-05,
      "model_forward_time": 1.726135015487671,
      "step": 60642
    },
    {
      "epoch": 5.7832717895507815e-05,
      "step": 60642,
      "training_step_time": 7.710315704345703
    },
    {
      "epoch": 5.783367156982422e-05,
      "model_forward_time": 1.7985739707946777,
      "step": 60643
    },
    {
      "epoch": 5.783367156982422e-05,
      "step": 60643,
      "training_step_time": 8.727294445037842
    },
    {
      "epoch": 5.7834625244140623e-05,
      "model_forward_time": 1.7248377799987793,
      "step": 60644
    },
    {
      "epoch": 5.7834625244140623e-05,
      "step": 60644,
      "training_step_time": 8.057401418685913
    },
    {
      "epoch": 5.7835578918457034e-05,
      "model_forward_time": 1.7387146949768066,
      "step": 60645
    },
    {
      "epoch": 5.7835578918457034e-05,
      "step": 60645,
      "training_step_time": 6.687899351119995
    },
    {
      "epoch": 5.783653259277344e-05,
      "model_forward_time": 1.7287943363189697,
      "step": 60646
    },
    {
      "epoch": 5.783653259277344e-05,
      "step": 60646,
      "training_step_time": 7.3522703647613525
    },
    {
      "epoch": 5.783748626708984e-05,
      "model_forward_time": 1.7206976413726807,
      "step": 60647
    },
    {
      "epoch": 5.783748626708984e-05,
      "step": 60647,
      "training_step_time": 7.561132907867432
    },
    {
      "epoch": 5.783843994140625e-05,
      "model_forward_time": 1.728743076324463,
      "step": 60648
    },
    {
      "epoch": 5.783843994140625e-05,
      "step": 60648,
      "training_step_time": 7.526687383651733
    },
    {
      "epoch": 5.783939361572266e-05,
      "model_forward_time": 1.7326531410217285,
      "step": 60649
    },
    {
      "epoch": 5.783939361572266e-05,
      "step": 60649,
      "training_step_time": 8.513056993484497
    },
    {
      "epoch": 5.784034729003906e-05,
      "grad_norm": 0.10077004879713058,
      "learning_rate": 3.6688366997575763e-06,
      "loss": 0.0558,
      "step": 60650
    },
    {
      "epoch": 5.784034729003906e-05,
      "model_forward_time": 2.827131986618042,
      "step": 60650
    },
    {
      "dynamics_loss_avg": 0.05764598175883293,
      "epoch": 5.784034729003906e-05,
      "step": 60650
    },
    {
      "action_loss_avg": 0.0018487343360902742,
      "epoch": 5.784034729003906e-05,
      "step": 60650
    },
    {
      "epoch": 5.784034729003906e-05,
      "step": 60650,
      "training_step_time": 9.012247800827026
    },
    {
      "epoch": 5.784130096435547e-05,
      "model_forward_time": 1.7385151386260986,
      "step": 60651
    },
    {
      "epoch": 5.784130096435547e-05,
      "step": 60651,
      "training_step_time": 7.793947219848633
    },
    {
      "epoch": 5.7842254638671876e-05,
      "model_forward_time": 1.780599594116211,
      "step": 60652
    },
    {
      "epoch": 5.7842254638671876e-05,
      "step": 60652,
      "training_step_time": 8.388949394226074
    },
    {
      "epoch": 5.784320831298828e-05,
      "model_forward_time": 1.7358298301696777,
      "step": 60653
    },
    {
      "epoch": 5.784320831298828e-05,
      "step": 60653,
      "training_step_time": 7.6920247077941895
    },
    {
      "epoch": 5.784416198730469e-05,
      "model_forward_time": 1.7428500652313232,
      "step": 60654
    },
    {
      "epoch": 5.784416198730469e-05,
      "step": 60654,
      "training_step_time": 7.505337953567505
    },
    {
      "epoch": 5.7845115661621095e-05,
      "model_forward_time": 1.727179765701294,
      "step": 60655
    },
    {
      "epoch": 5.7845115661621095e-05,
      "step": 60655,
      "training_step_time": 7.378289222717285
    },
    {
      "epoch": 5.78460693359375e-05,
      "model_forward_time": 1.730172872543335,
      "step": 60656
    },
    {
      "epoch": 5.78460693359375e-05,
      "step": 60656,
      "training_step_time": 7.3678600788116455
    },
    {
      "epoch": 5.78470230102539e-05,
      "model_forward_time": 1.73626708984375,
      "step": 60657
    },
    {
      "epoch": 5.78470230102539e-05,
      "step": 60657,
      "training_step_time": 7.527576923370361
    },
    {
      "epoch": 5.7847976684570314e-05,
      "model_forward_time": 1.735201120376587,
      "step": 60658
    },
    {
      "epoch": 5.7847976684570314e-05,
      "step": 60658,
      "training_step_time": 7.5457422733306885
    },
    {
      "epoch": 5.784893035888672e-05,
      "model_forward_time": 1.7249116897583008,
      "step": 60659
    },
    {
      "epoch": 5.784893035888672e-05,
      "step": 60659,
      "training_step_time": 9.330935955047607
    },
    {
      "epoch": 5.784988403320312e-05,
      "grad_norm": 0.10284292697906494,
      "learning_rate": 3.6672429784692648e-06,
      "loss": 0.052,
      "step": 60660
    },
    {
      "epoch": 5.784988403320312e-05,
      "model_forward_time": 1.7408440113067627,
      "step": 60660
    },
    {
      "dynamics_loss_avg": 0.034912490658462046,
      "epoch": 5.784988403320312e-05,
      "step": 60660
    },
    {
      "action_loss_avg": 0.0014897788845701144,
      "epoch": 5.784988403320312e-05,
      "step": 60660
    },
    {
      "epoch": 5.784988403320312e-05,
      "step": 60660,
      "training_step_time": 7.83027458190918
    },
    {
      "epoch": 5.785083770751953e-05,
      "model_forward_time": 1.7430915832519531,
      "step": 60661
    },
    {
      "epoch": 5.785083770751953e-05,
      "step": 60661,
      "training_step_time": 9.557413339614868
    },
    {
      "epoch": 5.7851791381835937e-05,
      "model_forward_time": 1.751997947692871,
      "step": 60662
    },
    {
      "epoch": 5.7851791381835937e-05,
      "step": 60662,
      "training_step_time": 7.710688829421997
    },
    {
      "epoch": 5.785274505615234e-05,
      "model_forward_time": 1.7276709079742432,
      "step": 60663
    },
    {
      "epoch": 5.785274505615234e-05,
      "step": 60663,
      "training_step_time": 7.502937316894531
    },
    {
      "epoch": 5.785369873046875e-05,
      "model_forward_time": 1.7491908073425293,
      "step": 60664
    },
    {
      "epoch": 5.785369873046875e-05,
      "step": 60664,
      "training_step_time": 7.4829065799713135
    },
    {
      "epoch": 5.7854652404785155e-05,
      "model_forward_time": 1.7427730560302734,
      "step": 60665
    },
    {
      "epoch": 5.7854652404785155e-05,
      "step": 60665,
      "training_step_time": 7.161787509918213
    },
    {
      "epoch": 5.785560607910156e-05,
      "model_forward_time": 1.7304186820983887,
      "step": 60666
    },
    {
      "epoch": 5.785560607910156e-05,
      "step": 60666,
      "training_step_time": 7.266444683074951
    },
    {
      "epoch": 5.785655975341797e-05,
      "model_forward_time": 1.7330827713012695,
      "step": 60667
    },
    {
      "epoch": 5.785655975341797e-05,
      "step": 60667,
      "training_step_time": 8.49028730392456
    },
    {
      "epoch": 5.7857513427734374e-05,
      "model_forward_time": 1.7355623245239258,
      "step": 60668
    },
    {
      "epoch": 5.7857513427734374e-05,
      "step": 60668,
      "training_step_time": 9.31379747390747
    },
    {
      "epoch": 5.785846710205078e-05,
      "model_forward_time": 2.927583694458008,
      "step": 60669
    },
    {
      "epoch": 5.785846710205078e-05,
      "step": 60669,
      "training_step_time": 8.478480100631714
    },
    {
      "epoch": 5.785942077636719e-05,
      "grad_norm": 0.09465780109167099,
      "learning_rate": 3.6656494029292578e-06,
      "loss": 0.0587,
      "step": 60670
    },
    {
      "epoch": 5.785942077636719e-05,
      "model_forward_time": 1.740025520324707,
      "step": 60670
    },
    {
      "dynamics_loss_avg": 0.07166262622922659,
      "epoch": 5.785942077636719e-05,
      "step": 60670
    },
    {
      "action_loss_avg": 0.002115429047262296,
      "epoch": 5.785942077636719e-05,
      "step": 60670
    },
    {
      "epoch": 5.785942077636719e-05,
      "step": 60670,
      "training_step_time": 9.400288105010986
    },
    {
      "epoch": 5.786037445068359e-05,
      "model_forward_time": 1.849783182144165,
      "step": 60671
    },
    {
      "epoch": 5.786037445068359e-05,
      "step": 60671,
      "training_step_time": 7.810815095901489
    },
    {
      "epoch": 5.7861328125e-05,
      "model_forward_time": 1.7301287651062012,
      "step": 60672
    },
    {
      "epoch": 5.7861328125e-05,
      "step": 60672,
      "training_step_time": 7.3392109870910645
    },
    {
      "epoch": 5.786228179931641e-05,
      "model_forward_time": 1.7377991676330566,
      "step": 60673
    },
    {
      "epoch": 5.786228179931641e-05,
      "step": 60673,
      "training_step_time": 7.4622509479522705
    },
    {
      "epoch": 5.786323547363281e-05,
      "model_forward_time": 1.7375869750976562,
      "step": 60674
    },
    {
      "epoch": 5.786323547363281e-05,
      "step": 60674,
      "training_step_time": 7.224781513214111
    },
    {
      "epoch": 5.7864189147949216e-05,
      "model_forward_time": 1.7322802543640137,
      "step": 60675
    },
    {
      "epoch": 5.7864189147949216e-05,
      "step": 60675,
      "training_step_time": 7.30736780166626
    },
    {
      "epoch": 5.786514282226563e-05,
      "model_forward_time": 1.7324237823486328,
      "step": 60676
    },
    {
      "epoch": 5.786514282226563e-05,
      "step": 60676,
      "training_step_time": 7.613382577896118
    },
    {
      "epoch": 5.786609649658203e-05,
      "model_forward_time": 1.727720022201538,
      "step": 60677
    },
    {
      "epoch": 5.786609649658203e-05,
      "step": 60677,
      "training_step_time": 9.786228656768799
    },
    {
      "epoch": 5.7867050170898435e-05,
      "model_forward_time": 2.9419314861297607,
      "step": 60678
    },
    {
      "epoch": 5.7867050170898435e-05,
      "step": 60678,
      "training_step_time": 9.082171201705933
    },
    {
      "epoch": 5.7868003845214846e-05,
      "model_forward_time": 1.3797752857208252,
      "step": 60679
    },
    {
      "epoch": 5.7868003845214846e-05,
      "step": 60679,
      "training_step_time": 9.511320114135742
    },
    {
      "epoch": 5.786895751953125e-05,
      "grad_norm": 0.09049292653799057,
      "learning_rate": 3.664055973311825e-06,
      "loss": 0.0599,
      "step": 60680
    },
    {
      "epoch": 5.786895751953125e-05,
      "model_forward_time": 1.822005271911621,
      "step": 60680
    },
    {
      "dynamics_loss_avg": 0.03763099070638418,
      "epoch": 5.786895751953125e-05,
      "step": 60680
    },
    {
      "action_loss_avg": 0.0008184592821635306,
      "epoch": 5.786895751953125e-05,
      "step": 60680
    },
    {
      "epoch": 5.786895751953125e-05,
      "step": 60680,
      "training_step_time": 7.713111639022827
    },
    {
      "epoch": 5.7869911193847654e-05,
      "model_forward_time": 1.7127621173858643,
      "step": 60681
    },
    {
      "epoch": 5.7869911193847654e-05,
      "step": 60681,
      "training_step_time": 6.648405313491821
    },
    {
      "epoch": 5.7870864868164065e-05,
      "model_forward_time": 1.7163805961608887,
      "step": 60682
    },
    {
      "epoch": 5.7870864868164065e-05,
      "step": 60682,
      "training_step_time": 7.4339330196380615
    },
    {
      "epoch": 5.787181854248047e-05,
      "model_forward_time": 1.7182857990264893,
      "step": 60683
    },
    {
      "epoch": 5.787181854248047e-05,
      "step": 60683,
      "training_step_time": 7.098304748535156
    },
    {
      "epoch": 5.787277221679687e-05,
      "model_forward_time": 1.732024908065796,
      "step": 60684
    },
    {
      "epoch": 5.787277221679687e-05,
      "step": 60684,
      "training_step_time": 7.35773491859436
    },
    {
      "epoch": 5.7873725891113283e-05,
      "model_forward_time": 1.7173426151275635,
      "step": 60685
    },
    {
      "epoch": 5.7873725891113283e-05,
      "step": 60685,
      "training_step_time": 7.508002519607544
    },
    {
      "epoch": 5.787467956542969e-05,
      "model_forward_time": 1.717726707458496,
      "step": 60686
    },
    {
      "epoch": 5.787467956542969e-05,
      "step": 60686,
      "training_step_time": 7.598327159881592
    },
    {
      "epoch": 5.787563323974609e-05,
      "model_forward_time": 2.9345195293426514,
      "step": 60687
    },
    {
      "epoch": 5.787563323974609e-05,
      "step": 60687,
      "training_step_time": 9.678620338439941
    },
    {
      "epoch": 5.78765869140625e-05,
      "model_forward_time": 1.716336727142334,
      "step": 60688
    },
    {
      "epoch": 5.78765869140625e-05,
      "step": 60688,
      "training_step_time": 7.726685285568237
    },
    {
      "epoch": 5.7877540588378906e-05,
      "model_forward_time": 1.8046178817749023,
      "step": 60689
    },
    {
      "epoch": 5.7877540588378906e-05,
      "step": 60689,
      "training_step_time": 8.695571660995483
    },
    {
      "epoch": 5.787849426269531e-05,
      "grad_norm": 0.11360753327608109,
      "learning_rate": 3.6624626897912213e-06,
      "loss": 0.0561,
      "step": 60690
    },
    {
      "epoch": 5.787849426269531e-05,
      "model_forward_time": 1.711618423461914,
      "step": 60690
    },
    {
      "dynamics_loss_avg": 0.04599298276007176,
      "epoch": 5.787849426269531e-05,
      "step": 60690
    },
    {
      "action_loss_avg": 0.0018585896701551973,
      "epoch": 5.787849426269531e-05,
      "step": 60690
    },
    {
      "epoch": 5.787849426269531e-05,
      "step": 60690,
      "training_step_time": 7.591449737548828
    },
    {
      "epoch": 5.787944793701172e-05,
      "model_forward_time": 1.7226817607879639,
      "step": 60691
    },
    {
      "epoch": 5.787944793701172e-05,
      "step": 60691,
      "training_step_time": 7.400229215621948
    },
    {
      "epoch": 5.7880401611328125e-05,
      "model_forward_time": 1.3524165153503418,
      "step": 60692
    },
    {
      "epoch": 5.7880401611328125e-05,
      "step": 60692,
      "training_step_time": 7.253337860107422
    },
    {
      "epoch": 5.788135528564453e-05,
      "model_forward_time": 1.7269959449768066,
      "step": 60693
    },
    {
      "epoch": 5.788135528564453e-05,
      "step": 60693,
      "training_step_time": 7.014955043792725
    },
    {
      "epoch": 5.788230895996094e-05,
      "model_forward_time": 1.7213923931121826,
      "step": 60694
    },
    {
      "epoch": 5.788230895996094e-05,
      "step": 60694,
      "training_step_time": 7.477842807769775
    },
    {
      "epoch": 5.7883262634277344e-05,
      "model_forward_time": 1.7205159664154053,
      "step": 60695
    },
    {
      "epoch": 5.7883262634277344e-05,
      "step": 60695,
      "training_step_time": 9.206202507019043
    },
    {
      "epoch": 5.788421630859375e-05,
      "model_forward_time": 2.857508420944214,
      "step": 60696
    },
    {
      "epoch": 5.788421630859375e-05,
      "step": 60696,
      "training_step_time": 8.148092269897461
    },
    {
      "epoch": 5.788516998291016e-05,
      "model_forward_time": 1.7188191413879395,
      "step": 60697
    },
    {
      "epoch": 5.788516998291016e-05,
      "step": 60697,
      "training_step_time": 8.19032621383667
    },
    {
      "epoch": 5.788612365722656e-05,
      "model_forward_time": 1.7857277393341064,
      "step": 60698
    },
    {
      "epoch": 5.788612365722656e-05,
      "step": 60698,
      "training_step_time": 7.6924121379852295
    },
    {
      "epoch": 5.788707733154297e-05,
      "model_forward_time": 1.7251472473144531,
      "step": 60699
    },
    {
      "epoch": 5.788707733154297e-05,
      "step": 60699,
      "training_step_time": 7.606285810470581
    },
    {
      "epoch": 5.788803100585938e-05,
      "grad_norm": 0.09646358340978622,
      "learning_rate": 3.6608695525416886e-06,
      "loss": 0.057,
      "step": 60700
    },
    {
      "epoch": 5.788803100585938e-05,
      "model_forward_time": 1.717949628829956,
      "step": 60700
    },
    {
      "dynamics_loss_avg": 0.05872453935444355,
      "epoch": 5.788803100585938e-05,
      "step": 60700
    },
    {
      "action_loss_avg": 0.0015131416323129088,
      "epoch": 5.788803100585938e-05,
      "step": 60700
    },
    {
      "epoch": 5.788803100585938e-05,
      "step": 60700,
      "training_step_time": 7.358563184738159
    },
    {
      "epoch": 5.788898468017578e-05,
      "model_forward_time": 1.7191390991210938,
      "step": 60701
    },
    {
      "epoch": 5.788898468017578e-05,
      "step": 60701,
      "training_step_time": 6.7199835777282715
    },
    {
      "epoch": 5.7889938354492186e-05,
      "model_forward_time": 1.7316129207611084,
      "step": 60702
    },
    {
      "epoch": 5.7889938354492186e-05,
      "step": 60702,
      "training_step_time": 7.097437858581543
    },
    {
      "epoch": 5.7890892028808597e-05,
      "model_forward_time": 1.7326397895812988,
      "step": 60703
    },
    {
      "epoch": 5.7890892028808597e-05,
      "step": 60703,
      "training_step_time": 7.432205677032471
    },
    {
      "epoch": 5.7891845703125e-05,
      "model_forward_time": 1.7290446758270264,
      "step": 60704
    },
    {
      "epoch": 5.7891845703125e-05,
      "step": 60704,
      "training_step_time": 8.60606074333191
    },
    {
      "epoch": 5.7892799377441405e-05,
      "model_forward_time": 2.8445706367492676,
      "step": 60705
    },
    {
      "epoch": 5.7892799377441405e-05,
      "step": 60705,
      "training_step_time": 9.641930103302002
    },
    {
      "epoch": 5.7893753051757815e-05,
      "model_forward_time": 1.7244491577148438,
      "step": 60706
    },
    {
      "epoch": 5.7893753051757815e-05,
      "step": 60706,
      "training_step_time": 7.718034982681274
    },
    {
      "epoch": 5.789470672607422e-05,
      "model_forward_time": 1.7545416355133057,
      "step": 60707
    },
    {
      "epoch": 5.789470672607422e-05,
      "step": 60707,
      "training_step_time": 9.069247961044312
    },
    {
      "epoch": 5.7895660400390623e-05,
      "model_forward_time": 1.7246105670928955,
      "step": 60708
    },
    {
      "epoch": 5.7895660400390623e-05,
      "step": 60708,
      "training_step_time": 7.5439982414245605
    },
    {
      "epoch": 5.7896614074707034e-05,
      "model_forward_time": 1.7140676975250244,
      "step": 60709
    },
    {
      "epoch": 5.7896614074707034e-05,
      "step": 60709,
      "training_step_time": 8.445068359375
    },
    {
      "epoch": 5.789756774902344e-05,
      "grad_norm": 0.0944666862487793,
      "learning_rate": 3.6592765617374483e-06,
      "loss": 0.0576,
      "step": 60710
    },
    {
      "epoch": 5.789756774902344e-05,
      "model_forward_time": 1.719529151916504,
      "step": 60710
    },
    {
      "dynamics_loss_avg": 0.061016489192843436,
      "epoch": 5.789756774902344e-05,
      "step": 60710
    },
    {
      "action_loss_avg": 0.0014487560489214956,
      "epoch": 5.789756774902344e-05,
      "step": 60710
    },
    {
      "epoch": 5.789756774902344e-05,
      "step": 60710,
      "training_step_time": 7.570363998413086
    },
    {
      "epoch": 5.789852142333984e-05,
      "model_forward_time": 1.3491475582122803,
      "step": 60711
    },
    {
      "epoch": 5.789852142333984e-05,
      "step": 60711,
      "training_step_time": 7.252779245376587
    },
    {
      "epoch": 5.789947509765625e-05,
      "model_forward_time": 1.724951982498169,
      "step": 60712
    },
    {
      "epoch": 5.789947509765625e-05,
      "step": 60712,
      "training_step_time": 7.539727449417114
    },
    {
      "epoch": 5.790042877197266e-05,
      "model_forward_time": 1.7322030067443848,
      "step": 60713
    },
    {
      "epoch": 5.790042877197266e-05,
      "step": 60713,
      "training_step_time": 7.698517322540283
    },
    {
      "epoch": 5.790138244628906e-05,
      "model_forward_time": 1.7178683280944824,
      "step": 60714
    },
    {
      "epoch": 5.790138244628906e-05,
      "step": 60714,
      "training_step_time": 8.566869735717773
    },
    {
      "epoch": 5.790233612060547e-05,
      "model_forward_time": 2.9800782203674316,
      "step": 60715
    },
    {
      "epoch": 5.790233612060547e-05,
      "step": 60715,
      "training_step_time": 8.272560834884644
    },
    {
      "epoch": 5.7903289794921876e-05,
      "model_forward_time": 1.7220690250396729,
      "step": 60716
    },
    {
      "epoch": 5.7903289794921876e-05,
      "step": 60716,
      "training_step_time": 9.993728399276733
    },
    {
      "epoch": 5.790424346923828e-05,
      "model_forward_time": 1.729990005493164,
      "step": 60717
    },
    {
      "epoch": 5.790424346923828e-05,
      "step": 60717,
      "training_step_time": 8.788329362869263
    },
    {
      "epoch": 5.790519714355469e-05,
      "model_forward_time": 1.7233479022979736,
      "step": 60718
    },
    {
      "epoch": 5.790519714355469e-05,
      "step": 60718,
      "training_step_time": 7.411292314529419
    },
    {
      "epoch": 5.7906150817871095e-05,
      "model_forward_time": 1.731161117553711,
      "step": 60719
    },
    {
      "epoch": 5.7906150817871095e-05,
      "step": 60719,
      "training_step_time": 7.627845764160156
    },
    {
      "epoch": 5.79071044921875e-05,
      "grad_norm": 0.09478289633989334,
      "learning_rate": 3.657683717552708e-06,
      "loss": 0.0605,
      "step": 60720
    },
    {
      "epoch": 5.79071044921875e-05,
      "model_forward_time": 1.7285842895507812,
      "step": 60720
    },
    {
      "dynamics_loss_avg": 0.055021383427083494,
      "epoch": 5.79071044921875e-05,
      "step": 60720
    },
    {
      "action_loss_avg": 0.0011878962337505071,
      "epoch": 5.79071044921875e-05,
      "step": 60720
    },
    {
      "epoch": 5.79071044921875e-05,
      "step": 60720,
      "training_step_time": 7.37685227394104
    },
    {
      "epoch": 5.79080581665039e-05,
      "model_forward_time": 1.7209434509277344,
      "step": 60721
    },
    {
      "epoch": 5.79080581665039e-05,
      "step": 60721,
      "training_step_time": 7.468954086303711
    },
    {
      "epoch": 5.7909011840820314e-05,
      "model_forward_time": 1.7252025604248047,
      "step": 60722
    },
    {
      "epoch": 5.7909011840820314e-05,
      "step": 60722,
      "training_step_time": 8.396441459655762
    },
    {
      "epoch": 5.790996551513672e-05,
      "model_forward_time": 1.7317001819610596,
      "step": 60723
    },
    {
      "epoch": 5.790996551513672e-05,
      "step": 60723,
      "training_step_time": 8.429164409637451
    },
    {
      "epoch": 5.791091918945312e-05,
      "model_forward_time": 2.9235002994537354,
      "step": 60724
    },
    {
      "epoch": 5.791091918945312e-05,
      "step": 60724,
      "training_step_time": 8.605259418487549
    },
    {
      "epoch": 5.791187286376953e-05,
      "model_forward_time": 1.7314834594726562,
      "step": 60725
    },
    {
      "epoch": 5.791187286376953e-05,
      "step": 60725,
      "training_step_time": 9.950634241104126
    },
    {
      "epoch": 5.791282653808594e-05,
      "model_forward_time": 1.832733392715454,
      "step": 60726
    },
    {
      "epoch": 5.791282653808594e-05,
      "step": 60726,
      "training_step_time": 7.85654091835022
    },
    {
      "epoch": 5.791378021240234e-05,
      "model_forward_time": 1.732478141784668,
      "step": 60727
    },
    {
      "epoch": 5.791378021240234e-05,
      "step": 60727,
      "training_step_time": 7.754172086715698
    },
    {
      "epoch": 5.791473388671875e-05,
      "model_forward_time": 1.7248849868774414,
      "step": 60728
    },
    {
      "epoch": 5.791473388671875e-05,
      "step": 60728,
      "training_step_time": 7.810317516326904
    },
    {
      "epoch": 5.7915687561035155e-05,
      "model_forward_time": 1.7263789176940918,
      "step": 60729
    },
    {
      "epoch": 5.7915687561035155e-05,
      "step": 60729,
      "training_step_time": 7.340671539306641
    },
    {
      "epoch": 5.791664123535156e-05,
      "grad_norm": 0.10364627093076706,
      "learning_rate": 3.656091020161657e-06,
      "loss": 0.0574,
      "step": 60730
    },
    {
      "epoch": 5.791664123535156e-05,
      "model_forward_time": 1.7298529148101807,
      "step": 60730
    },
    {
      "dynamics_loss_avg": 0.05283350013196468,
      "epoch": 5.791664123535156e-05,
      "step": 60730
    },
    {
      "action_loss_avg": 0.00167404426028952,
      "epoch": 5.791664123535156e-05,
      "step": 60730
    },
    {
      "epoch": 5.791664123535156e-05,
      "step": 60730,
      "training_step_time": 7.4895689487457275
    },
    {
      "epoch": 5.791759490966797e-05,
      "model_forward_time": 1.7656359672546387,
      "step": 60731
    },
    {
      "epoch": 5.791759490966797e-05,
      "step": 60731,
      "training_step_time": 7.758600950241089
    },
    {
      "epoch": 5.7918548583984374e-05,
      "model_forward_time": 1.7155570983886719,
      "step": 60732
    },
    {
      "epoch": 5.7918548583984374e-05,
      "step": 60732,
      "training_step_time": 8.66806697845459
    },
    {
      "epoch": 5.791950225830078e-05,
      "model_forward_time": 2.9226224422454834,
      "step": 60733
    },
    {
      "epoch": 5.791950225830078e-05,
      "step": 60733,
      "training_step_time": 10.104513883590698
    },
    {
      "epoch": 5.792045593261719e-05,
      "model_forward_time": 1.727872610092163,
      "step": 60734
    },
    {
      "epoch": 5.792045593261719e-05,
      "step": 60734,
      "training_step_time": 7.947032928466797
    },
    {
      "epoch": 5.792140960693359e-05,
      "model_forward_time": 1.7961387634277344,
      "step": 60735
    },
    {
      "epoch": 5.792140960693359e-05,
      "step": 60735,
      "training_step_time": 7.76868462562561
    },
    {
      "epoch": 5.792236328125e-05,
      "model_forward_time": 1.7269318103790283,
      "step": 60736
    },
    {
      "epoch": 5.792236328125e-05,
      "step": 60736,
      "training_step_time": 7.8032660484313965
    },
    {
      "epoch": 5.792331695556641e-05,
      "model_forward_time": 1.3625743389129639,
      "step": 60737
    },
    {
      "epoch": 5.792331695556641e-05,
      "step": 60737,
      "training_step_time": 7.82755184173584
    },
    {
      "epoch": 5.792427062988281e-05,
      "model_forward_time": 1.73061203956604,
      "step": 60738
    },
    {
      "epoch": 5.792427062988281e-05,
      "step": 60738,
      "training_step_time": 7.621197700500488
    },
    {
      "epoch": 5.7925224304199216e-05,
      "model_forward_time": 1.7276065349578857,
      "step": 60739
    },
    {
      "epoch": 5.7925224304199216e-05,
      "step": 60739,
      "training_step_time": 7.423142433166504
    },
    {
      "epoch": 5.792617797851563e-05,
      "grad_norm": 0.14555887877941132,
      "learning_rate": 3.6544984697384723e-06,
      "loss": 0.0616,
      "step": 60740
    },
    {
      "epoch": 5.792617797851563e-05,
      "model_forward_time": 1.727717399597168,
      "step": 60740
    },
    {
      "dynamics_loss_avg": 0.07555565387010574,
      "epoch": 5.792617797851563e-05,
      "step": 60740
    },
    {
      "action_loss_avg": 0.001787902688374743,
      "epoch": 5.792617797851563e-05,
      "step": 60740
    },
    {
      "epoch": 5.792617797851563e-05,
      "step": 60740,
      "training_step_time": 7.791802644729614
    },
    {
      "epoch": 5.792713165283203e-05,
      "model_forward_time": 1.7234399318695068,
      "step": 60741
    },
    {
      "epoch": 5.792713165283203e-05,
      "step": 60741,
      "training_step_time": 9.3628408908844
    },
    {
      "epoch": 5.7928085327148435e-05,
      "model_forward_time": 2.923522472381592,
      "step": 60742
    },
    {
      "epoch": 5.7928085327148435e-05,
      "step": 60742,
      "training_step_time": 10.147104501724243
    },
    {
      "epoch": 5.7929039001464846e-05,
      "model_forward_time": 1.7191317081451416,
      "step": 60743
    },
    {
      "epoch": 5.7929039001464846e-05,
      "step": 60743,
      "training_step_time": 8.509601354598999
    },
    {
      "epoch": 5.792999267578125e-05,
      "model_forward_time": 1.781376600265503,
      "step": 60744
    },
    {
      "epoch": 5.792999267578125e-05,
      "step": 60744,
      "training_step_time": 7.879275321960449
    },
    {
      "epoch": 5.7930946350097654e-05,
      "model_forward_time": 1.3654074668884277,
      "step": 60745
    },
    {
      "epoch": 5.7930946350097654e-05,
      "step": 60745,
      "training_step_time": 7.744465351104736
    },
    {
      "epoch": 5.7931900024414065e-05,
      "model_forward_time": 1.7284111976623535,
      "step": 60746
    },
    {
      "epoch": 5.7931900024414065e-05,
      "step": 60746,
      "training_step_time": 7.556110143661499
    },
    {
      "epoch": 5.793285369873047e-05,
      "model_forward_time": 1.7349603176116943,
      "step": 60747
    },
    {
      "epoch": 5.793285369873047e-05,
      "step": 60747,
      "training_step_time": 7.454103708267212
    },
    {
      "epoch": 5.793380737304687e-05,
      "model_forward_time": 1.7312710285186768,
      "step": 60748
    },
    {
      "epoch": 5.793380737304687e-05,
      "step": 60748,
      "training_step_time": 7.426629066467285
    },
    {
      "epoch": 5.7934761047363283e-05,
      "model_forward_time": 1.7398290634155273,
      "step": 60749
    },
    {
      "epoch": 5.7934761047363283e-05,
      "step": 60749,
      "training_step_time": 7.520183324813843
    },
    {
      "epoch": 5.793571472167969e-05,
      "grad_norm": 0.10008423775434494,
      "learning_rate": 3.6529060664573122e-06,
      "loss": 0.0554,
      "step": 60750
    },
    {
      "epoch": 5.793571472167969e-05,
      "model_forward_time": 1.7330148220062256,
      "step": 60750
    },
    {
      "dynamics_loss_avg": 0.05446693263947964,
      "epoch": 5.793571472167969e-05,
      "step": 60750
    },
    {
      "action_loss_avg": 0.0013159722846467047,
      "epoch": 5.793571472167969e-05,
      "step": 60750
    },
    {
      "epoch": 5.793571472167969e-05,
      "step": 60750,
      "training_step_time": 8.606361865997314
    },
    {
      "epoch": 5.793666839599609e-05,
      "model_forward_time": 2.8627126216888428,
      "step": 60751
    },
    {
      "epoch": 5.793666839599609e-05,
      "step": 60751,
      "training_step_time": 9.872946739196777
    },
    {
      "epoch": 5.79376220703125e-05,
      "model_forward_time": 1.753864049911499,
      "step": 60752
    },
    {
      "epoch": 5.79376220703125e-05,
      "step": 60752,
      "training_step_time": 9.45872974395752
    },
    {
      "epoch": 5.7938575744628906e-05,
      "model_forward_time": 1.731156826019287,
      "step": 60753
    },
    {
      "epoch": 5.7938575744628906e-05,
      "step": 60753,
      "training_step_time": 7.921992540359497
    },
    {
      "epoch": 5.793952941894531e-05,
      "model_forward_time": 1.8527448177337646,
      "step": 60754
    },
    {
      "epoch": 5.793952941894531e-05,
      "step": 60754,
      "training_step_time": 7.739065885543823
    },
    {
      "epoch": 5.794048309326172e-05,
      "model_forward_time": 1.7366523742675781,
      "step": 60755
    },
    {
      "epoch": 5.794048309326172e-05,
      "step": 60755,
      "training_step_time": 7.742397308349609
    },
    {
      "epoch": 5.7941436767578125e-05,
      "model_forward_time": 1.726701021194458,
      "step": 60756
    },
    {
      "epoch": 5.7941436767578125e-05,
      "step": 60756,
      "training_step_time": 7.783507347106934
    },
    {
      "epoch": 5.794239044189453e-05,
      "model_forward_time": 1.7380213737487793,
      "step": 60757
    },
    {
      "epoch": 5.794239044189453e-05,
      "step": 60757,
      "training_step_time": 7.438382863998413
    },
    {
      "epoch": 5.794334411621094e-05,
      "model_forward_time": 1.734393835067749,
      "step": 60758
    },
    {
      "epoch": 5.794334411621094e-05,
      "step": 60758,
      "training_step_time": 8.812451124191284
    },
    {
      "epoch": 5.7944297790527344e-05,
      "model_forward_time": 1.7315313816070557,
      "step": 60759
    },
    {
      "epoch": 5.7944297790527344e-05,
      "step": 60759,
      "training_step_time": 8.789488792419434
    },
    {
      "epoch": 5.794525146484375e-05,
      "grad_norm": 0.1347808688879013,
      "learning_rate": 3.6513138104923176e-06,
      "loss": 0.0571,
      "step": 60760
    },
    {
      "epoch": 5.794525146484375e-05,
      "model_forward_time": 2.8883297443389893,
      "step": 60760
    },
    {
      "dynamics_loss_avg": 0.049491691030561925,
      "epoch": 5.794525146484375e-05,
      "step": 60760
    },
    {
      "action_loss_avg": 0.00167451994784642,
      "epoch": 5.794525146484375e-05,
      "step": 60760
    },
    {
      "epoch": 5.794525146484375e-05,
      "step": 60760,
      "training_step_time": 8.406770944595337
    },
    {
      "epoch": 5.794620513916016e-05,
      "model_forward_time": 1.7257907390594482,
      "step": 60761
    },
    {
      "epoch": 5.794620513916016e-05,
      "step": 60761,
      "training_step_time": 8.706924438476562
    },
    {
      "epoch": 5.794715881347656e-05,
      "model_forward_time": 1.7697131633758545,
      "step": 60762
    },
    {
      "epoch": 5.794715881347656e-05,
      "step": 60762,
      "training_step_time": 8.023998737335205
    },
    {
      "epoch": 5.794811248779297e-05,
      "model_forward_time": 1.852501392364502,
      "step": 60763
    },
    {
      "epoch": 5.794811248779297e-05,
      "step": 60763,
      "training_step_time": 7.762992858886719
    },
    {
      "epoch": 5.794906616210938e-05,
      "model_forward_time": 1.7297730445861816,
      "step": 60764
    },
    {
      "epoch": 5.794906616210938e-05,
      "step": 60764,
      "training_step_time": 7.798487424850464
    },
    {
      "epoch": 5.795001983642578e-05,
      "model_forward_time": 1.7360947132110596,
      "step": 60765
    },
    {
      "epoch": 5.795001983642578e-05,
      "step": 60765,
      "training_step_time": 7.853006362915039
    },
    {
      "epoch": 5.7950973510742186e-05,
      "model_forward_time": 1.7259917259216309,
      "step": 60766
    },
    {
      "epoch": 5.7950973510742186e-05,
      "step": 60766,
      "training_step_time": 7.586818218231201
    },
    {
      "epoch": 5.79519271850586e-05,
      "model_forward_time": 1.7305047512054443,
      "step": 60767
    },
    {
      "epoch": 5.79519271850586e-05,
      "step": 60767,
      "training_step_time": 7.735154867172241
    },
    {
      "epoch": 5.7952880859375e-05,
      "model_forward_time": 1.7273094654083252,
      "step": 60768
    },
    {
      "epoch": 5.7952880859375e-05,
      "step": 60768,
      "training_step_time": 7.769001245498657
    },
    {
      "epoch": 5.7953834533691405e-05,
      "model_forward_time": 1.7158801555633545,
      "step": 60769
    },
    {
      "epoch": 5.7953834533691405e-05,
      "step": 60769,
      "training_step_time": 7.837864637374878
    },
    {
      "epoch": 5.7954788208007815e-05,
      "grad_norm": 0.09655508399009705,
      "learning_rate": 3.649721702017618e-06,
      "loss": 0.0542,
      "step": 60770
    },
    {
      "epoch": 5.7954788208007815e-05,
      "model_forward_time": 2.8921098709106445,
      "step": 60770
    },
    {
      "dynamics_loss_avg": 0.07170995064079762,
      "epoch": 5.7954788208007815e-05,
      "step": 60770
    },
    {
      "action_loss_avg": 0.0020170023140963166,
      "epoch": 5.7954788208007815e-05,
      "step": 60770
    },
    {
      "epoch": 5.7954788208007815e-05,
      "step": 60770,
      "training_step_time": 10.385391235351562
    },
    {
      "epoch": 5.795574188232422e-05,
      "model_forward_time": 1.7201025485992432,
      "step": 60771
    },
    {
      "epoch": 5.795574188232422e-05,
      "step": 60771,
      "training_step_time": 9.087248086929321
    },
    {
      "epoch": 5.7956695556640624e-05,
      "model_forward_time": 1.814474105834961,
      "step": 60772
    },
    {
      "epoch": 5.7956695556640624e-05,
      "step": 60772,
      "training_step_time": 7.917682409286499
    },
    {
      "epoch": 5.7957649230957034e-05,
      "model_forward_time": 1.720393419265747,
      "step": 60773
    },
    {
      "epoch": 5.7957649230957034e-05,
      "step": 60773,
      "training_step_time": 7.6189165115356445
    },
    {
      "epoch": 5.795860290527344e-05,
      "model_forward_time": 1.7238984107971191,
      "step": 60774
    },
    {
      "epoch": 5.795860290527344e-05,
      "step": 60774,
      "training_step_time": 7.844364404678345
    },
    {
      "epoch": 5.795955657958984e-05,
      "model_forward_time": 1.7226781845092773,
      "step": 60775
    },
    {
      "epoch": 5.795955657958984e-05,
      "step": 60775,
      "training_step_time": 7.876754522323608
    },
    {
      "epoch": 5.796051025390625e-05,
      "model_forward_time": 1.715801477432251,
      "step": 60776
    },
    {
      "epoch": 5.796051025390625e-05,
      "step": 60776,
      "training_step_time": 8.292975187301636
    },
    {
      "epoch": 5.796146392822266e-05,
      "model_forward_time": 1.719261884689331,
      "step": 60777
    },
    {
      "epoch": 5.796146392822266e-05,
      "step": 60777,
      "training_step_time": 7.795841932296753
    },
    {
      "epoch": 5.796241760253906e-05,
      "model_forward_time": 1.73897385597229,
      "step": 60778
    },
    {
      "epoch": 5.796241760253906e-05,
      "step": 60778,
      "training_step_time": 7.827584505081177
    },
    {
      "epoch": 5.796337127685547e-05,
      "model_forward_time": 2.8952908515930176,
      "step": 60779
    },
    {
      "epoch": 5.796337127685547e-05,
      "step": 60779,
      "training_step_time": 8.838191509246826
    },
    {
      "epoch": 5.7964324951171876e-05,
      "grad_norm": 0.13811661303043365,
      "learning_rate": 3.6481297412073237e-06,
      "loss": 0.0547,
      "step": 60780
    },
    {
      "epoch": 5.7964324951171876e-05,
      "model_forward_time": 1.7316389083862305,
      "step": 60780
    },
    {
      "dynamics_loss_avg": 0.05358608216047287,
      "epoch": 5.7964324951171876e-05,
      "step": 60780
    },
    {
      "action_loss_avg": 0.0014538124611135571,
      "epoch": 5.7964324951171876e-05,
      "step": 60780
    },
    {
      "epoch": 5.7964324951171876e-05,
      "step": 60780,
      "training_step_time": 8.007762908935547
    },
    {
      "epoch": 5.796527862548828e-05,
      "model_forward_time": 1.7846012115478516,
      "step": 60781
    },
    {
      "epoch": 5.796527862548828e-05,
      "step": 60781,
      "training_step_time": 8.533240556716919
    },
    {
      "epoch": 5.7966232299804684e-05,
      "model_forward_time": 1.7200660705566406,
      "step": 60782
    },
    {
      "epoch": 5.7966232299804684e-05,
      "step": 60782,
      "training_step_time": 8.7333345413208
    },
    {
      "epoch": 5.7967185974121095e-05,
      "model_forward_time": 1.7157440185546875,
      "step": 60783
    },
    {
      "epoch": 5.7967185974121095e-05,
      "step": 60783,
      "training_step_time": 8.175321340560913
    },
    {
      "epoch": 5.79681396484375e-05,
      "model_forward_time": 1.7156062126159668,
      "step": 60784
    },
    {
      "epoch": 5.79681396484375e-05,
      "step": 60784,
      "training_step_time": 7.871232748031616
    },
    {
      "epoch": 5.79690933227539e-05,
      "model_forward_time": 1.7179503440856934,
      "step": 60785
    },
    {
      "epoch": 5.79690933227539e-05,
      "step": 60785,
      "training_step_time": 7.658898115158081
    },
    {
      "epoch": 5.7970046997070314e-05,
      "model_forward_time": 1.7192692756652832,
      "step": 60786
    },
    {
      "epoch": 5.7970046997070314e-05,
      "step": 60786,
      "training_step_time": 7.719355821609497
    },
    {
      "epoch": 5.797100067138672e-05,
      "model_forward_time": 1.718308687210083,
      "step": 60787
    },
    {
      "epoch": 5.797100067138672e-05,
      "step": 60787,
      "training_step_time": 8.865319967269897
    },
    {
      "epoch": 5.797195434570312e-05,
      "model_forward_time": 2.895228385925293,
      "step": 60788
    },
    {
      "epoch": 5.797195434570312e-05,
      "step": 60788,
      "training_step_time": 9.26250410079956
    },
    {
      "epoch": 5.797290802001953e-05,
      "model_forward_time": 1.7187659740447998,
      "step": 60789
    },
    {
      "epoch": 5.797290802001953e-05,
      "step": 60789,
      "training_step_time": 7.845280647277832
    },
    {
      "epoch": 5.797386169433594e-05,
      "grad_norm": 0.0866740345954895,
      "learning_rate": 3.6465379282355273e-06,
      "loss": 0.0569,
      "step": 60790
    },
    {
      "epoch": 5.797386169433594e-05,
      "model_forward_time": 1.7688934803009033,
      "step": 60790
    },
    {
      "dynamics_loss_avg": 0.047014438360929486,
      "epoch": 5.797386169433594e-05,
      "step": 60790
    },
    {
      "action_loss_avg": 0.001453914784360677,
      "epoch": 5.797386169433594e-05,
      "step": 60790
    },
    {
      "epoch": 5.797386169433594e-05,
      "step": 60790,
      "training_step_time": 8.060679912567139
    },
    {
      "epoch": 5.797481536865234e-05,
      "model_forward_time": 1.7161335945129395,
      "step": 60791
    },
    {
      "epoch": 5.797481536865234e-05,
      "step": 60791,
      "training_step_time": 7.81136417388916
    },
    {
      "epoch": 5.797576904296875e-05,
      "model_forward_time": 1.7226803302764893,
      "step": 60792
    },
    {
      "epoch": 5.797576904296875e-05,
      "step": 60792,
      "training_step_time": 7.687513828277588
    },
    {
      "epoch": 5.7976722717285156e-05,
      "model_forward_time": 1.7246830463409424,
      "step": 60793
    },
    {
      "epoch": 5.7976722717285156e-05,
      "step": 60793,
      "training_step_time": 7.877350568771362
    },
    {
      "epoch": 5.797767639160156e-05,
      "model_forward_time": 1.7190043926239014,
      "step": 60794
    },
    {
      "epoch": 5.797767639160156e-05,
      "step": 60794,
      "training_step_time": 7.684660196304321
    },
    {
      "epoch": 5.797863006591797e-05,
      "model_forward_time": 1.729161262512207,
      "step": 60795
    },
    {
      "epoch": 5.797863006591797e-05,
      "step": 60795,
      "training_step_time": 7.747281789779663
    },
    {
      "epoch": 5.7979583740234374e-05,
      "model_forward_time": 1.7277777194976807,
      "step": 60796
    },
    {
      "epoch": 5.7979583740234374e-05,
      "step": 60796,
      "training_step_time": 8.918998956680298
    },
    {
      "epoch": 5.798053741455078e-05,
      "model_forward_time": 2.938375949859619,
      "step": 60797
    },
    {
      "epoch": 5.798053741455078e-05,
      "step": 60797,
      "training_step_time": 7.933831453323364
    },
    {
      "epoch": 5.798149108886719e-05,
      "model_forward_time": 1.7271580696105957,
      "step": 60798
    },
    {
      "epoch": 5.798149108886719e-05,
      "step": 60798,
      "training_step_time": 7.764815330505371
    },
    {
      "epoch": 5.798244476318359e-05,
      "model_forward_time": 1.3441152572631836,
      "step": 60799
    },
    {
      "epoch": 5.798244476318359e-05,
      "step": 60799,
      "training_step_time": 8.933345317840576
    },
    {
      "epoch": 5.79833984375e-05,
      "grad_norm": 0.1569901406764984,
      "learning_rate": 3.644946263276308e-06,
      "loss": 0.0592,
      "step": 60800
    },
    {
      "epoch": 5.79833984375e-05,
      "model_forward_time": 1.7212464809417725,
      "step": 60800
    },
    {
      "dynamics_loss_avg": 0.06303895730525255,
      "epoch": 5.79833984375e-05,
      "step": 60800
    },
    {
      "action_loss_avg": 0.0016794757102616131,
      "epoch": 5.79833984375e-05,
      "step": 60800
    },
    {
      "epoch": 5.79833984375e-05,
      "step": 60800,
      "training_step_time": 7.866273880004883
    },
    {
      "epoch": 5.798435211181641e-05,
      "model_forward_time": 1.7186777591705322,
      "step": 60801
    },
    {
      "epoch": 5.798435211181641e-05,
      "step": 60801,
      "training_step_time": 7.9317731857299805
    },
    {
      "epoch": 5.798530578613281e-05,
      "model_forward_time": 1.7233679294586182,
      "step": 60802
    },
    {
      "epoch": 5.798530578613281e-05,
      "step": 60802,
      "training_step_time": 7.8650007247924805
    },
    {
      "epoch": 5.7986259460449216e-05,
      "model_forward_time": 1.7176837921142578,
      "step": 60803
    },
    {
      "epoch": 5.7986259460449216e-05,
      "step": 60803,
      "training_step_time": 7.652616262435913
    },
    {
      "epoch": 5.798721313476563e-05,
      "model_forward_time": 1.7132346630096436,
      "step": 60804
    },
    {
      "epoch": 5.798721313476563e-05,
      "step": 60804,
      "training_step_time": 7.698354959487915
    },
    {
      "epoch": 5.798816680908203e-05,
      "model_forward_time": 1.7183351516723633,
      "step": 60805
    },
    {
      "epoch": 5.798816680908203e-05,
      "step": 60805,
      "training_step_time": 8.247316360473633
    },
    {
      "epoch": 5.7989120483398435e-05,
      "model_forward_time": 2.8112268447875977,
      "step": 60806
    },
    {
      "epoch": 5.7989120483398435e-05,
      "step": 60806,
      "training_step_time": 7.819224834442139
    },
    {
      "epoch": 5.7990074157714846e-05,
      "model_forward_time": 1.7241237163543701,
      "step": 60807
    },
    {
      "epoch": 5.7990074157714846e-05,
      "step": 60807,
      "training_step_time": 8.827824354171753
    },
    {
      "epoch": 5.799102783203125e-05,
      "model_forward_time": 1.7204115390777588,
      "step": 60808
    },
    {
      "epoch": 5.799102783203125e-05,
      "step": 60808,
      "training_step_time": 8.639995813369751
    },
    {
      "epoch": 5.7991981506347654e-05,
      "model_forward_time": 1.8262951374053955,
      "step": 60809
    },
    {
      "epoch": 5.7991981506347654e-05,
      "step": 60809,
      "training_step_time": 7.876256942749023
    },
    {
      "epoch": 5.7992935180664065e-05,
      "grad_norm": 0.07210737466812134,
      "learning_rate": 3.643354746503729e-06,
      "loss": 0.0546,
      "step": 60810
    },
    {
      "epoch": 5.7992935180664065e-05,
      "model_forward_time": 1.7248694896697998,
      "step": 60810
    },
    {
      "dynamics_loss_avg": 0.047198088839650154,
      "epoch": 5.7992935180664065e-05,
      "step": 60810
    },
    {
      "action_loss_avg": 0.0012112222670111806,
      "epoch": 5.7992935180664065e-05,
      "step": 60810
    },
    {
      "epoch": 5.7992935180664065e-05,
      "step": 60810,
      "training_step_time": 7.670915365219116
    },
    {
      "epoch": 5.799388885498047e-05,
      "model_forward_time": 1.7201249599456787,
      "step": 60811
    },
    {
      "epoch": 5.799388885498047e-05,
      "step": 60811,
      "training_step_time": 7.661489486694336
    },
    {
      "epoch": 5.799484252929687e-05,
      "model_forward_time": 1.7184021472930908,
      "step": 60812
    },
    {
      "epoch": 5.799484252929687e-05,
      "step": 60812,
      "training_step_time": 7.4713709354400635
    },
    {
      "epoch": 5.7995796203613284e-05,
      "model_forward_time": 1.7148284912109375,
      "step": 60813
    },
    {
      "epoch": 5.7995796203613284e-05,
      "step": 60813,
      "training_step_time": 8.340593576431274
    },
    {
      "epoch": 5.799674987792969e-05,
      "model_forward_time": 1.722917079925537,
      "step": 60814
    },
    {
      "epoch": 5.799674987792969e-05,
      "step": 60814,
      "training_step_time": 7.608708381652832
    },
    {
      "epoch": 5.799770355224609e-05,
      "model_forward_time": 1.7156212329864502,
      "step": 60815
    },
    {
      "epoch": 5.799770355224609e-05,
      "step": 60815,
      "training_step_time": 8.24568772315979
    },
    {
      "epoch": 5.79986572265625e-05,
      "model_forward_time": 2.9707226753234863,
      "step": 60816
    },
    {
      "epoch": 5.79986572265625e-05,
      "step": 60816,
      "training_step_time": 8.803137063980103
    },
    {
      "epoch": 5.7999610900878906e-05,
      "model_forward_time": 1.7165515422821045,
      "step": 60817
    },
    {
      "epoch": 5.7999610900878906e-05,
      "step": 60817,
      "training_step_time": 9.248255014419556
    },
    {
      "epoch": 5.800056457519531e-05,
      "model_forward_time": 1.8476061820983887,
      "step": 60818
    },
    {
      "epoch": 5.800056457519531e-05,
      "step": 60818,
      "training_step_time": 7.876492738723755
    },
    {
      "epoch": 5.800151824951172e-05,
      "model_forward_time": 1.7269940376281738,
      "step": 60819
    },
    {
      "epoch": 5.800151824951172e-05,
      "step": 60819,
      "training_step_time": 7.627511262893677
    },
    {
      "epoch": 5.8002471923828125e-05,
      "grad_norm": 0.09695218503475189,
      "learning_rate": 3.6417633780918344e-06,
      "loss": 0.055,
      "step": 60820
    },
    {
      "epoch": 5.8002471923828125e-05,
      "model_forward_time": 1.7332587242126465,
      "step": 60820
    },
    {
      "dynamics_loss_avg": 0.04427198003977537,
      "epoch": 5.8002471923828125e-05,
      "step": 60820
    },
    {
      "action_loss_avg": 0.001555316272424534,
      "epoch": 5.8002471923828125e-05,
      "step": 60820
    },
    {
      "epoch": 5.8002471923828125e-05,
      "step": 60820,
      "training_step_time": 7.7034337520599365
    },
    {
      "epoch": 5.800342559814453e-05,
      "model_forward_time": 1.7414205074310303,
      "step": 60821
    },
    {
      "epoch": 5.800342559814453e-05,
      "step": 60821,
      "training_step_time": 8.33820366859436
    },
    {
      "epoch": 5.800437927246094e-05,
      "model_forward_time": 1.7280049324035645,
      "step": 60822
    },
    {
      "epoch": 5.800437927246094e-05,
      "step": 60822,
      "training_step_time": 7.512162208557129
    },
    {
      "epoch": 5.8005332946777344e-05,
      "model_forward_time": 1.735788345336914,
      "step": 60823
    },
    {
      "epoch": 5.8005332946777344e-05,
      "step": 60823,
      "training_step_time": 8.401582479476929
    },
    {
      "epoch": 5.800628662109375e-05,
      "model_forward_time": 1.375664234161377,
      "step": 60824
    },
    {
      "epoch": 5.800628662109375e-05,
      "step": 60824,
      "training_step_time": 8.656760454177856
    },
    {
      "epoch": 5.800724029541016e-05,
      "model_forward_time": 2.8859806060791016,
      "step": 60825
    },
    {
      "epoch": 5.800724029541016e-05,
      "step": 60825,
      "training_step_time": 8.856792449951172
    },
    {
      "epoch": 5.800819396972656e-05,
      "model_forward_time": 1.7389612197875977,
      "step": 60826
    },
    {
      "epoch": 5.800819396972656e-05,
      "step": 60826,
      "training_step_time": 8.435930013656616
    },
    {
      "epoch": 5.800914764404297e-05,
      "model_forward_time": 1.808919906616211,
      "step": 60827
    },
    {
      "epoch": 5.800914764404297e-05,
      "step": 60827,
      "training_step_time": 8.667214155197144
    },
    {
      "epoch": 5.801010131835938e-05,
      "model_forward_time": 1.740854263305664,
      "step": 60828
    },
    {
      "epoch": 5.801010131835938e-05,
      "step": 60828,
      "training_step_time": 8.771455764770508
    },
    {
      "epoch": 5.801105499267578e-05,
      "model_forward_time": 1.7322595119476318,
      "step": 60829
    },
    {
      "epoch": 5.801105499267578e-05,
      "step": 60829,
      "training_step_time": 8.242903232574463
    },
    {
      "epoch": 5.8012008666992186e-05,
      "grad_norm": 0.11370863765478134,
      "learning_rate": 3.6401721582146554e-06,
      "loss": 0.0554,
      "step": 60830
    },
    {
      "epoch": 5.8012008666992186e-05,
      "model_forward_time": 1.7454705238342285,
      "step": 60830
    },
    {
      "dynamics_loss_avg": 0.0494397034868598,
      "epoch": 5.8012008666992186e-05,
      "step": 60830
    },
    {
      "action_loss_avg": 0.0013455442123813555,
      "epoch": 5.8012008666992186e-05,
      "step": 60830
    },
    {
      "epoch": 5.8012008666992186e-05,
      "step": 60830,
      "training_step_time": 7.519444942474365
    },
    {
      "epoch": 5.80129623413086e-05,
      "model_forward_time": 1.716383934020996,
      "step": 60831
    },
    {
      "epoch": 5.80129623413086e-05,
      "step": 60831,
      "training_step_time": 8.082454681396484
    },
    {
      "epoch": 5.8013916015625e-05,
      "model_forward_time": 1.7191352844238281,
      "step": 60832
    },
    {
      "epoch": 5.8013916015625e-05,
      "step": 60832,
      "training_step_time": 7.599314212799072
    },
    {
      "epoch": 5.8014869689941405e-05,
      "model_forward_time": 1.726823091506958,
      "step": 60833
    },
    {
      "epoch": 5.8014869689941405e-05,
      "step": 60833,
      "training_step_time": 7.6655237674713135
    },
    {
      "epoch": 5.8015823364257816e-05,
      "model_forward_time": 2.870500087738037,
      "step": 60834
    },
    {
      "epoch": 5.8015823364257816e-05,
      "step": 60834,
      "training_step_time": 9.698365926742554
    },
    {
      "epoch": 5.801677703857422e-05,
      "model_forward_time": 1.7283871173858643,
      "step": 60835
    },
    {
      "epoch": 5.801677703857422e-05,
      "step": 60835,
      "training_step_time": 8.34479808807373
    },
    {
      "epoch": 5.8017730712890624e-05,
      "model_forward_time": 1.7664167881011963,
      "step": 60836
    },
    {
      "epoch": 5.8017730712890624e-05,
      "step": 60836,
      "training_step_time": 7.871314764022827
    },
    {
      "epoch": 5.8018684387207034e-05,
      "model_forward_time": 1.7241621017456055,
      "step": 60837
    },
    {
      "epoch": 5.8018684387207034e-05,
      "step": 60837,
      "training_step_time": 7.824124574661255
    },
    {
      "epoch": 5.801963806152344e-05,
      "model_forward_time": 1.728489875793457,
      "step": 60838
    },
    {
      "epoch": 5.801963806152344e-05,
      "step": 60838,
      "training_step_time": 7.4639904499053955
    },
    {
      "epoch": 5.802059173583984e-05,
      "model_forward_time": 1.7194221019744873,
      "step": 60839
    },
    {
      "epoch": 5.802059173583984e-05,
      "step": 60839,
      "training_step_time": 7.730430603027344
    },
    {
      "epoch": 5.802154541015625e-05,
      "grad_norm": 0.10182199627161026,
      "learning_rate": 3.6385810870462034e-06,
      "loss": 0.0566,
      "step": 60840
    },
    {
      "epoch": 5.802154541015625e-05,
      "model_forward_time": 1.7223210334777832,
      "step": 60840
    },
    {
      "dynamics_loss_avg": 0.05348282344639301,
      "epoch": 5.802154541015625e-05,
      "step": 60840
    },
    {
      "action_loss_avg": 0.001442707545356825,
      "epoch": 5.802154541015625e-05,
      "step": 60840
    },
    {
      "epoch": 5.802154541015625e-05,
      "step": 60840,
      "training_step_time": 7.3128502368927
    },
    {
      "epoch": 5.802249908447266e-05,
      "model_forward_time": 1.7263453006744385,
      "step": 60841
    },
    {
      "epoch": 5.802249908447266e-05,
      "step": 60841,
      "training_step_time": 7.570842027664185
    },
    {
      "epoch": 5.802345275878906e-05,
      "model_forward_time": 1.723384141921997,
      "step": 60842
    },
    {
      "epoch": 5.802345275878906e-05,
      "step": 60842,
      "training_step_time": 8.218265771865845
    },
    {
      "epoch": 5.802440643310547e-05,
      "model_forward_time": 2.841141700744629,
      "step": 60843
    },
    {
      "epoch": 5.802440643310547e-05,
      "step": 60843,
      "training_step_time": 7.735963821411133
    },
    {
      "epoch": 5.8025360107421876e-05,
      "model_forward_time": 1.7285802364349365,
      "step": 60844
    },
    {
      "epoch": 5.8025360107421876e-05,
      "step": 60844,
      "training_step_time": 8.244853734970093
    },
    {
      "epoch": 5.802631378173828e-05,
      "model_forward_time": 1.7209367752075195,
      "step": 60845
    },
    {
      "epoch": 5.802631378173828e-05,
      "step": 60845,
      "training_step_time": 9.530105113983154
    },
    {
      "epoch": 5.8027267456054684e-05,
      "model_forward_time": 1.728302240371704,
      "step": 60846
    },
    {
      "epoch": 5.8027267456054684e-05,
      "step": 60846,
      "training_step_time": 7.810900449752808
    },
    {
      "epoch": 5.8028221130371095e-05,
      "model_forward_time": 1.7228991985321045,
      "step": 60847
    },
    {
      "epoch": 5.8028221130371095e-05,
      "step": 60847,
      "training_step_time": 8.585989952087402
    },
    {
      "epoch": 5.80291748046875e-05,
      "model_forward_time": 1.711944341659546,
      "step": 60848
    },
    {
      "epoch": 5.80291748046875e-05,
      "step": 60848,
      "training_step_time": 8.57964038848877
    },
    {
      "epoch": 5.80301284790039e-05,
      "model_forward_time": 1.7161495685577393,
      "step": 60849
    },
    {
      "epoch": 5.80301284790039e-05,
      "step": 60849,
      "training_step_time": 7.114928722381592
    },
    {
      "epoch": 5.8031082153320314e-05,
      "grad_norm": 0.1470527946949005,
      "learning_rate": 3.636990164760479e-06,
      "loss": 0.0522,
      "step": 60850
    },
    {
      "epoch": 5.8031082153320314e-05,
      "model_forward_time": 1.7163965702056885,
      "step": 60850
    },
    {
      "dynamics_loss_avg": 0.03762850984930992,
      "epoch": 5.8031082153320314e-05,
      "step": 60850
    },
    {
      "action_loss_avg": 0.0011341492354404181,
      "epoch": 5.8031082153320314e-05,
      "step": 60850
    },
    {
      "epoch": 5.8031082153320314e-05,
      "step": 60850,
      "training_step_time": 8.561584234237671
    },
    {
      "epoch": 5.803203582763672e-05,
      "model_forward_time": 1.7314560413360596,
      "step": 60851
    },
    {
      "epoch": 5.803203582763672e-05,
      "step": 60851,
      "training_step_time": 7.630019664764404
    },
    {
      "epoch": 5.803298950195312e-05,
      "model_forward_time": 1.7247974872589111,
      "step": 60852
    },
    {
      "epoch": 5.803298950195312e-05,
      "step": 60852,
      "training_step_time": 8.987317323684692
    },
    {
      "epoch": 5.803394317626953e-05,
      "model_forward_time": 1.71291184425354,
      "step": 60853
    },
    {
      "epoch": 5.803394317626953e-05,
      "step": 60853,
      "training_step_time": 8.53257441520691
    },
    {
      "epoch": 5.803489685058594e-05,
      "model_forward_time": 1.7060036659240723,
      "step": 60854
    },
    {
      "epoch": 5.803489685058594e-05,
      "step": 60854,
      "training_step_time": 8.53173828125
    },
    {
      "epoch": 5.803585052490234e-05,
      "model_forward_time": 1.4453668594360352,
      "step": 60855
    },
    {
      "epoch": 5.803585052490234e-05,
      "step": 60855,
      "training_step_time": 7.7888343334198
    },
    {
      "epoch": 5.803680419921875e-05,
      "model_forward_time": 1.0160930156707764,
      "step": 60856
    },
    {
      "epoch": 5.803680419921875e-05,
      "step": 60856,
      "training_step_time": 7.6647937297821045
    },
    {
      "epoch": 5.8037757873535156e-05,
      "model_forward_time": 1.7323153018951416,
      "step": 60857
    },
    {
      "epoch": 5.8037757873535156e-05,
      "step": 60857,
      "training_step_time": 7.782795429229736
    },
    {
      "epoch": 5.803871154785156e-05,
      "model_forward_time": 1.7602784633636475,
      "step": 60858
    },
    {
      "epoch": 5.803871154785156e-05,
      "step": 60858,
      "training_step_time": 7.244341135025024
    },
    {
      "epoch": 5.803966522216797e-05,
      "model_forward_time": 1.711707353591919,
      "step": 60859
    },
    {
      "epoch": 5.803966522216797e-05,
      "step": 60859,
      "training_step_time": 7.5619215965271
    },
    {
      "epoch": 5.8040618896484374e-05,
      "grad_norm": 0.08850662410259247,
      "learning_rate": 3.63539939153146e-06,
      "loss": 0.052,
      "step": 60860
    },
    {
      "epoch": 5.8040618896484374e-05,
      "model_forward_time": 1.7210538387298584,
      "step": 60860
    },
    {
      "dynamics_loss_avg": 0.059273458831012246,
      "epoch": 5.8040618896484374e-05,
      "step": 60860
    },
    {
      "action_loss_avg": 0.0015911908994894475,
      "epoch": 5.8040618896484374e-05,
      "step": 60860
    },
    {
      "epoch": 5.8040618896484374e-05,
      "step": 60860,
      "training_step_time": 7.914254188537598
    },
    {
      "epoch": 5.804157257080078e-05,
      "model_forward_time": 1.0089962482452393,
      "step": 60861
    },
    {
      "epoch": 5.804157257080078e-05,
      "step": 60861,
      "training_step_time": 8.222612142562866
    },
    {
      "epoch": 5.804252624511719e-05,
      "model_forward_time": 1.7419366836547852,
      "step": 60862
    },
    {
      "epoch": 5.804252624511719e-05,
      "step": 60862,
      "training_step_time": 8.31218671798706
    },
    {
      "epoch": 5.804347991943359e-05,
      "model_forward_time": 1.7208950519561768,
      "step": 60863
    },
    {
      "epoch": 5.804347991943359e-05,
      "step": 60863,
      "training_step_time": 7.640620708465576
    },
    {
      "epoch": 5.804443359375e-05,
      "model_forward_time": 1.810617208480835,
      "step": 60864
    },
    {
      "epoch": 5.804443359375e-05,
      "step": 60864,
      "training_step_time": 8.732119798660278
    },
    {
      "epoch": 5.804538726806641e-05,
      "model_forward_time": 1.716257095336914,
      "step": 60865
    },
    {
      "epoch": 5.804538726806641e-05,
      "step": 60865,
      "training_step_time": 8.488351821899414
    },
    {
      "epoch": 5.804634094238281e-05,
      "model_forward_time": 1.7140169143676758,
      "step": 60866
    },
    {
      "epoch": 5.804634094238281e-05,
      "step": 60866,
      "training_step_time": 7.764326572418213
    },
    {
      "epoch": 5.8047294616699216e-05,
      "model_forward_time": 1.728351354598999,
      "step": 60867
    },
    {
      "epoch": 5.8047294616699216e-05,
      "step": 60867,
      "training_step_time": 7.280745267868042
    },
    {
      "epoch": 5.804824829101563e-05,
      "model_forward_time": 1.7180907726287842,
      "step": 60868
    },
    {
      "epoch": 5.804824829101563e-05,
      "step": 60868,
      "training_step_time": 8.55972695350647
    },
    {
      "epoch": 5.804920196533203e-05,
      "model_forward_time": 1.7229790687561035,
      "step": 60869
    },
    {
      "epoch": 5.804920196533203e-05,
      "step": 60869,
      "training_step_time": 7.77148699760437
    },
    {
      "epoch": 5.8050155639648435e-05,
      "grad_norm": 0.0867520123720169,
      "learning_rate": 3.6338087675331124e-06,
      "loss": 0.0512,
      "step": 60870
    },
    {
      "epoch": 5.8050155639648435e-05,
      "model_forward_time": 1.7273855209350586,
      "step": 60870
    },
    {
      "dynamics_loss_avg": 0.05287283528596163,
      "epoch": 5.8050155639648435e-05,
      "step": 60870
    },
    {
      "action_loss_avg": 0.002452221931889653,
      "epoch": 5.8050155639648435e-05,
      "step": 60870
    },
    {
      "epoch": 5.8050155639648435e-05,
      "step": 60870,
      "training_step_time": 7.676985502243042
    },
    {
      "epoch": 5.8051109313964846e-05,
      "model_forward_time": 1.7309355735778809,
      "step": 60871
    },
    {
      "epoch": 5.8051109313964846e-05,
      "step": 60871,
      "training_step_time": 9.895697832107544
    },
    {
      "epoch": 5.805206298828125e-05,
      "model_forward_time": 2.8985507488250732,
      "step": 60872
    },
    {
      "epoch": 5.805206298828125e-05,
      "step": 60872,
      "training_step_time": 9.77082371711731
    },
    {
      "epoch": 5.8053016662597654e-05,
      "model_forward_time": 1.7956037521362305,
      "step": 60873
    },
    {
      "epoch": 5.8053016662597654e-05,
      "step": 60873,
      "training_step_time": 7.7865869998931885
    },
    {
      "epoch": 5.8053970336914065e-05,
      "model_forward_time": 1.7816195487976074,
      "step": 60874
    },
    {
      "epoch": 5.8053970336914065e-05,
      "step": 60874,
      "training_step_time": 7.6383376121521
    },
    {
      "epoch": 5.805492401123047e-05,
      "model_forward_time": 1.7281389236450195,
      "step": 60875
    },
    {
      "epoch": 5.805492401123047e-05,
      "step": 60875,
      "training_step_time": 7.8707451820373535
    },
    {
      "epoch": 5.805587768554687e-05,
      "model_forward_time": 1.730686902999878,
      "step": 60876
    },
    {
      "epoch": 5.805587768554687e-05,
      "step": 60876,
      "training_step_time": 7.701125621795654
    },
    {
      "epoch": 5.8056831359863284e-05,
      "model_forward_time": 1.7381179332733154,
      "step": 60877
    },
    {
      "epoch": 5.8056831359863284e-05,
      "step": 60877,
      "training_step_time": 7.514760494232178
    },
    {
      "epoch": 5.805778503417969e-05,
      "model_forward_time": 1.733041763305664,
      "step": 60878
    },
    {
      "epoch": 5.805778503417969e-05,
      "step": 60878,
      "training_step_time": 7.834398508071899
    },
    {
      "epoch": 5.805873870849609e-05,
      "model_forward_time": 1.707402229309082,
      "step": 60879
    },
    {
      "epoch": 5.805873870849609e-05,
      "step": 60879,
      "training_step_time": 8.65087103843689
    },
    {
      "epoch": 5.80596923828125e-05,
      "grad_norm": 0.0839494988322258,
      "learning_rate": 3.6322182929393833e-06,
      "loss": 0.0526,
      "step": 60880
    },
    {
      "epoch": 5.80596923828125e-05,
      "model_forward_time": 1.7197003364562988,
      "step": 60880
    },
    {
      "dynamics_loss_avg": 0.041287950705736876,
      "epoch": 5.80596923828125e-05,
      "step": 60880
    },
    {
      "action_loss_avg": 0.001748933386988938,
      "epoch": 5.80596923828125e-05,
      "step": 60880
    },
    {
      "epoch": 5.80596923828125e-05,
      "step": 60880,
      "training_step_time": 8.864657640457153
    },
    {
      "epoch": 5.8060646057128906e-05,
      "model_forward_time": 2.889281988143921,
      "step": 60881
    },
    {
      "epoch": 5.8060646057128906e-05,
      "step": 60881,
      "training_step_time": 7.713782787322998
    },
    {
      "epoch": 5.806159973144531e-05,
      "model_forward_time": 1.734086036682129,
      "step": 60882
    },
    {
      "epoch": 5.806159973144531e-05,
      "step": 60882,
      "training_step_time": 7.663957118988037
    },
    {
      "epoch": 5.806255340576172e-05,
      "model_forward_time": 1.7238314151763916,
      "step": 60883
    },
    {
      "epoch": 5.806255340576172e-05,
      "step": 60883,
      "training_step_time": 7.7614781856536865
    },
    {
      "epoch": 5.8063507080078125e-05,
      "model_forward_time": 1.72352933883667,
      "step": 60884
    },
    {
      "epoch": 5.8063507080078125e-05,
      "step": 60884,
      "training_step_time": 7.925550222396851
    },
    {
      "epoch": 5.806446075439453e-05,
      "model_forward_time": 1.724827766418457,
      "step": 60885
    },
    {
      "epoch": 5.806446075439453e-05,
      "step": 60885,
      "training_step_time": 7.8030476570129395
    },
    {
      "epoch": 5.806541442871094e-05,
      "model_forward_time": 1.7206528186798096,
      "step": 60886
    },
    {
      "epoch": 5.806541442871094e-05,
      "step": 60886,
      "training_step_time": 7.656060457229614
    },
    {
      "epoch": 5.8066368103027344e-05,
      "model_forward_time": 1.716801404953003,
      "step": 60887
    },
    {
      "epoch": 5.8066368103027344e-05,
      "step": 60887,
      "training_step_time": 7.939633369445801
    },
    {
      "epoch": 5.806732177734375e-05,
      "model_forward_time": 1.7283267974853516,
      "step": 60888
    },
    {
      "epoch": 5.806732177734375e-05,
      "step": 60888,
      "training_step_time": 9.0189790725708
    },
    {
      "epoch": 5.806827545166016e-05,
      "model_forward_time": 1.720992088317871,
      "step": 60889
    },
    {
      "epoch": 5.806827545166016e-05,
      "step": 60889,
      "training_step_time": 7.836647033691406
    },
    {
      "epoch": 5.806922912597656e-05,
      "grad_norm": 0.09622617065906525,
      "learning_rate": 3.6306279679242035e-06,
      "loss": 0.0568,
      "step": 60890
    },
    {
      "epoch": 5.806922912597656e-05,
      "model_forward_time": 2.9120123386383057,
      "step": 60890
    },
    {
      "dynamics_loss_avg": 0.06204159092158079,
      "epoch": 5.806922912597656e-05,
      "step": 60890
    },
    {
      "action_loss_avg": 0.0015306731511373073,
      "epoch": 5.806922912597656e-05,
      "step": 60890
    },
    {
      "epoch": 5.806922912597656e-05,
      "step": 60890,
      "training_step_time": 8.588580846786499
    },
    {
      "epoch": 5.807018280029297e-05,
      "model_forward_time": 1.7191152572631836,
      "step": 60891
    },
    {
      "epoch": 5.807018280029297e-05,
      "step": 60891,
      "training_step_time": 7.803230285644531
    },
    {
      "epoch": 5.807113647460938e-05,
      "model_forward_time": 1.8271236419677734,
      "step": 60892
    },
    {
      "epoch": 5.807113647460938e-05,
      "step": 60892,
      "training_step_time": 7.731531381607056
    },
    {
      "epoch": 5.807209014892578e-05,
      "model_forward_time": 1.7266199588775635,
      "step": 60893
    },
    {
      "epoch": 5.807209014892578e-05,
      "step": 60893,
      "training_step_time": 7.842854976654053
    },
    {
      "epoch": 5.8073043823242186e-05,
      "model_forward_time": 1.7218570709228516,
      "step": 60894
    },
    {
      "epoch": 5.8073043823242186e-05,
      "step": 60894,
      "training_step_time": 8.53123426437378
    },
    {
      "epoch": 5.80739974975586e-05,
      "model_forward_time": 1.7321209907531738,
      "step": 60895
    },
    {
      "epoch": 5.80739974975586e-05,
      "step": 60895,
      "training_step_time": 8.384915590286255
    },
    {
      "epoch": 5.8074951171875e-05,
      "model_forward_time": 1.720113754272461,
      "step": 60896
    },
    {
      "epoch": 5.8074951171875e-05,
      "step": 60896,
      "training_step_time": 9.690323114395142
    },
    {
      "epoch": 5.8075904846191405e-05,
      "model_forward_time": 1.7328886985778809,
      "step": 60897
    },
    {
      "epoch": 5.8075904846191405e-05,
      "step": 60897,
      "training_step_time": 9.235754251480103
    },
    {
      "epoch": 5.8076858520507816e-05,
      "model_forward_time": 1.7193541526794434,
      "step": 60898
    },
    {
      "epoch": 5.8076858520507816e-05,
      "step": 60898,
      "training_step_time": 9.662394762039185
    },
    {
      "epoch": 5.807781219482422e-05,
      "model_forward_time": 2.8827316761016846,
      "step": 60899
    },
    {
      "epoch": 5.807781219482422e-05,
      "step": 60899,
      "training_step_time": 7.768928527832031
    },
    {
      "epoch": 5.8078765869140624e-05,
      "grad_norm": 0.11145127564668655,
      "learning_rate": 3.6290377926614923e-06,
      "loss": 0.0558,
      "step": 60900
    },
    {
      "epoch": 5.8078765869140624e-05,
      "model_forward_time": 1.726928472518921,
      "step": 60900
    },
    {
      "dynamics_loss_avg": 0.03892962262034416,
      "epoch": 5.8078765869140624e-05,
      "step": 60900
    },
    {
      "action_loss_avg": 0.0008071545555139892,
      "epoch": 5.8078765869140624e-05,
      "step": 60900
    },
    {
      "epoch": 5.8078765869140624e-05,
      "step": 60900,
      "training_step_time": 7.697213649749756
    },
    {
      "epoch": 5.8079719543457034e-05,
      "model_forward_time": 1.8496203422546387,
      "step": 60901
    },
    {
      "epoch": 5.8079719543457034e-05,
      "step": 60901,
      "training_step_time": 8.359426975250244
    },
    {
      "epoch": 5.808067321777344e-05,
      "model_forward_time": 1.7218961715698242,
      "step": 60902
    },
    {
      "epoch": 5.808067321777344e-05,
      "step": 60902,
      "training_step_time": 8.213245153427124
    },
    {
      "epoch": 5.808162689208984e-05,
      "model_forward_time": 1.7224624156951904,
      "step": 60903
    },
    {
      "epoch": 5.808162689208984e-05,
      "step": 60903,
      "training_step_time": 7.868513584136963
    },
    {
      "epoch": 5.808258056640625e-05,
      "model_forward_time": 1.7299423217773438,
      "step": 60904
    },
    {
      "epoch": 5.808258056640625e-05,
      "step": 60904,
      "training_step_time": 7.3150634765625
    },
    {
      "epoch": 5.808353424072266e-05,
      "model_forward_time": 1.0318076610565186,
      "step": 60905
    },
    {
      "epoch": 5.808353424072266e-05,
      "step": 60905,
      "training_step_time": 7.886795997619629
    },
    {
      "epoch": 5.808448791503906e-05,
      "model_forward_time": 1.746253252029419,
      "step": 60906
    },
    {
      "epoch": 5.808448791503906e-05,
      "step": 60906,
      "training_step_time": 8.57140064239502
    },
    {
      "epoch": 5.8085441589355465e-05,
      "model_forward_time": 1.7393546104431152,
      "step": 60907
    },
    {
      "epoch": 5.8085441589355465e-05,
      "step": 60907,
      "training_step_time": 9.780945301055908
    },
    {
      "epoch": 5.8086395263671876e-05,
      "model_forward_time": 2.953230381011963,
      "step": 60908
    },
    {
      "epoch": 5.8086395263671876e-05,
      "step": 60908,
      "training_step_time": 9.535179138183594
    },
    {
      "epoch": 5.808734893798828e-05,
      "model_forward_time": 1.727504014968872,
      "step": 60909
    },
    {
      "epoch": 5.808734893798828e-05,
      "step": 60909,
      "training_step_time": 7.713921785354614
    },
    {
      "epoch": 5.8088302612304684e-05,
      "grad_norm": 0.11125461012125015,
      "learning_rate": 3.6274477673251475e-06,
      "loss": 0.0552,
      "step": 60910
    },
    {
      "epoch": 5.8088302612304684e-05,
      "model_forward_time": 1.8338820934295654,
      "step": 60910
    },
    {
      "dynamics_loss_avg": 0.06141385268419981,
      "epoch": 5.8088302612304684e-05,
      "step": 60910
    },
    {
      "action_loss_avg": 0.0021308783456333914,
      "epoch": 5.8088302612304684e-05,
      "step": 60910
    },
    {
      "epoch": 5.8088302612304684e-05,
      "step": 60910,
      "training_step_time": 7.909378290176392
    },
    {
      "epoch": 5.8089256286621095e-05,
      "model_forward_time": 1.7288792133331299,
      "step": 60911
    },
    {
      "epoch": 5.8089256286621095e-05,
      "step": 60911,
      "training_step_time": 7.7436203956604
    },
    {
      "epoch": 5.80902099609375e-05,
      "model_forward_time": 1.7122232913970947,
      "step": 60912
    },
    {
      "epoch": 5.80902099609375e-05,
      "step": 60912,
      "training_step_time": 8.117341756820679
    },
    {
      "epoch": 5.80911636352539e-05,
      "model_forward_time": 1.719940423965454,
      "step": 60913
    },
    {
      "epoch": 5.80911636352539e-05,
      "step": 60913,
      "training_step_time": 7.978763818740845
    },
    {
      "epoch": 5.8092117309570314e-05,
      "model_forward_time": 1.7153208255767822,
      "step": 60914
    },
    {
      "epoch": 5.8092117309570314e-05,
      "step": 60914,
      "training_step_time": 7.915311336517334
    },
    {
      "epoch": 5.809307098388672e-05,
      "model_forward_time": 1.7112302780151367,
      "step": 60915
    },
    {
      "epoch": 5.809307098388672e-05,
      "step": 60915,
      "training_step_time": 7.863906145095825
    },
    {
      "epoch": 5.809402465820312e-05,
      "model_forward_time": 1.7244925498962402,
      "step": 60916
    },
    {
      "epoch": 5.809402465820312e-05,
      "step": 60916,
      "training_step_time": 7.888533592224121
    },
    {
      "epoch": 5.809497833251953e-05,
      "model_forward_time": 2.8399105072021484,
      "step": 60917
    },
    {
      "epoch": 5.809497833251953e-05,
      "step": 60917,
      "training_step_time": 9.548220157623291
    },
    {
      "epoch": 5.809593200683594e-05,
      "model_forward_time": 1.7238883972167969,
      "step": 60918
    },
    {
      "epoch": 5.809593200683594e-05,
      "step": 60918,
      "training_step_time": 8.56551742553711
    },
    {
      "epoch": 5.809688568115234e-05,
      "model_forward_time": 1.7948098182678223,
      "step": 60919
    },
    {
      "epoch": 5.809688568115234e-05,
      "step": 60919,
      "training_step_time": 7.887472867965698
    },
    {
      "epoch": 5.809783935546875e-05,
      "grad_norm": 0.11700523644685745,
      "learning_rate": 3.6258578920890503e-06,
      "loss": 0.0564,
      "step": 60920
    },
    {
      "epoch": 5.809783935546875e-05,
      "model_forward_time": 1.7115459442138672,
      "step": 60920
    },
    {
      "dynamics_loss_avg": 0.05170817915350199,
      "epoch": 5.809783935546875e-05,
      "step": 60920
    },
    {
      "action_loss_avg": 0.001560226024594158,
      "epoch": 5.809783935546875e-05,
      "step": 60920
    },
    {
      "epoch": 5.809783935546875e-05,
      "step": 60920,
      "training_step_time": 8.98721170425415
    },
    {
      "epoch": 5.8098793029785156e-05,
      "model_forward_time": 1.7119033336639404,
      "step": 60921
    },
    {
      "epoch": 5.8098793029785156e-05,
      "step": 60921,
      "training_step_time": 8.588990211486816
    },
    {
      "epoch": 5.809974670410156e-05,
      "model_forward_time": 1.713524341583252,
      "step": 60922
    },
    {
      "epoch": 5.809974670410156e-05,
      "step": 60922,
      "training_step_time": 7.8359198570251465
    },
    {
      "epoch": 5.810070037841797e-05,
      "model_forward_time": 1.7195312976837158,
      "step": 60923
    },
    {
      "epoch": 5.810070037841797e-05,
      "step": 60923,
      "training_step_time": 7.881760597229004
    },
    {
      "epoch": 5.8101654052734375e-05,
      "model_forward_time": 1.7132086753845215,
      "step": 60924
    },
    {
      "epoch": 5.8101654052734375e-05,
      "step": 60924,
      "training_step_time": 7.771749973297119
    },
    {
      "epoch": 5.810260772705078e-05,
      "model_forward_time": 1.7198331356048584,
      "step": 60925
    },
    {
      "epoch": 5.810260772705078e-05,
      "step": 60925,
      "training_step_time": 8.341052532196045
    },
    {
      "epoch": 5.810356140136719e-05,
      "model_forward_time": 2.8176674842834473,
      "step": 60926
    },
    {
      "epoch": 5.810356140136719e-05,
      "step": 60926,
      "training_step_time": 9.487997055053711
    },
    {
      "epoch": 5.810451507568359e-05,
      "model_forward_time": 1.719560146331787,
      "step": 60927
    },
    {
      "epoch": 5.810451507568359e-05,
      "step": 60927,
      "training_step_time": 8.78816843032837
    },
    {
      "epoch": 5.810546875e-05,
      "model_forward_time": 1.761364459991455,
      "step": 60928
    },
    {
      "epoch": 5.810546875e-05,
      "step": 60928,
      "training_step_time": 8.389925956726074
    },
    {
      "epoch": 5.810642242431641e-05,
      "model_forward_time": 1.7145812511444092,
      "step": 60929
    },
    {
      "epoch": 5.810642242431641e-05,
      "step": 60929,
      "training_step_time": 7.754290342330933
    },
    {
      "epoch": 5.810737609863281e-05,
      "grad_norm": 0.10358183830976486,
      "learning_rate": 3.6242681671270685e-06,
      "loss": 0.0536,
      "step": 60930
    },
    {
      "epoch": 5.810737609863281e-05,
      "model_forward_time": 1.7477855682373047,
      "step": 60930
    },
    {
      "dynamics_loss_avg": 0.048123284988105294,
      "epoch": 5.810737609863281e-05,
      "step": 60930
    },
    {
      "action_loss_avg": 0.0011758131411625072,
      "epoch": 5.810737609863281e-05,
      "step": 60930
    },
    {
      "epoch": 5.810737609863281e-05,
      "step": 60930,
      "training_step_time": 7.6761157512664795
    },
    {
      "epoch": 5.8108329772949216e-05,
      "model_forward_time": 1.7201223373413086,
      "step": 60931
    },
    {
      "epoch": 5.8108329772949216e-05,
      "step": 60931,
      "training_step_time": 7.742227554321289
    },
    {
      "epoch": 5.810928344726563e-05,
      "model_forward_time": 1.7149834632873535,
      "step": 60932
    },
    {
      "epoch": 5.810928344726563e-05,
      "step": 60932,
      "training_step_time": 8.414880275726318
    },
    {
      "epoch": 5.811023712158203e-05,
      "model_forward_time": 1.715639352798462,
      "step": 60933
    },
    {
      "epoch": 5.811023712158203e-05,
      "step": 60933,
      "training_step_time": 7.7067365646362305
    },
    {
      "epoch": 5.8111190795898435e-05,
      "model_forward_time": 1.718496561050415,
      "step": 60934
    },
    {
      "epoch": 5.8111190795898435e-05,
      "step": 60934,
      "training_step_time": 8.552898645401001
    },
    {
      "epoch": 5.8112144470214846e-05,
      "model_forward_time": 1.7234265804290771,
      "step": 60935
    },
    {
      "epoch": 5.8112144470214846e-05,
      "step": 60935,
      "training_step_time": 7.792523622512817
    },
    {
      "epoch": 5.811309814453125e-05,
      "model_forward_time": 2.9167792797088623,
      "step": 60936
    },
    {
      "epoch": 5.811309814453125e-05,
      "step": 60936,
      "training_step_time": 8.439764738082886
    },
    {
      "epoch": 5.8114051818847654e-05,
      "model_forward_time": 1.7259202003479004,
      "step": 60937
    },
    {
      "epoch": 5.8114051818847654e-05,
      "step": 60937,
      "training_step_time": 7.716287136077881
    },
    {
      "epoch": 5.8115005493164065e-05,
      "model_forward_time": 1.7173237800598145,
      "step": 60938
    },
    {
      "epoch": 5.8115005493164065e-05,
      "step": 60938,
      "training_step_time": 8.412904024124146
    },
    {
      "epoch": 5.811595916748047e-05,
      "model_forward_time": 1.7182376384735107,
      "step": 60939
    },
    {
      "epoch": 5.811595916748047e-05,
      "step": 60939,
      "training_step_time": 7.633384466171265
    },
    {
      "epoch": 5.811691284179687e-05,
      "grad_norm": 0.13865189254283905,
      "learning_rate": 3.6226785926130514e-06,
      "loss": 0.0547,
      "step": 60940
    },
    {
      "epoch": 5.811691284179687e-05,
      "model_forward_time": 1.726935863494873,
      "step": 60940
    },
    {
      "dynamics_loss_avg": 0.03866665046662092,
      "epoch": 5.811691284179687e-05,
      "step": 60940
    },
    {
      "action_loss_avg": 0.0011027690256014467,
      "epoch": 5.811691284179687e-05,
      "step": 60940
    },
    {
      "epoch": 5.811691284179687e-05,
      "step": 60940,
      "training_step_time": 8.540107011795044
    },
    {
      "epoch": 5.8117866516113284e-05,
      "model_forward_time": 1.724496603012085,
      "step": 60941
    },
    {
      "epoch": 5.8117866516113284e-05,
      "step": 60941,
      "training_step_time": 7.712992429733276
    },
    {
      "epoch": 5.811882019042969e-05,
      "model_forward_time": 1.716494083404541,
      "step": 60942
    },
    {
      "epoch": 5.811882019042969e-05,
      "step": 60942,
      "training_step_time": 9.353040218353271
    },
    {
      "epoch": 5.811977386474609e-05,
      "model_forward_time": 1.7178308963775635,
      "step": 60943
    },
    {
      "epoch": 5.811977386474609e-05,
      "step": 60943,
      "training_step_time": 7.698734283447266
    },
    {
      "epoch": 5.81207275390625e-05,
      "model_forward_time": 1.7138488292694092,
      "step": 60944
    },
    {
      "epoch": 5.81207275390625e-05,
      "step": 60944,
      "training_step_time": 7.728404521942139
    },
    {
      "epoch": 5.8121681213378907e-05,
      "model_forward_time": 2.9244065284729004,
      "step": 60945
    },
    {
      "epoch": 5.8121681213378907e-05,
      "step": 60945,
      "training_step_time": 9.435569047927856
    },
    {
      "epoch": 5.812263488769531e-05,
      "model_forward_time": 1.3543856143951416,
      "step": 60946
    },
    {
      "epoch": 5.812263488769531e-05,
      "step": 60946,
      "training_step_time": 7.6097657680511475
    },
    {
      "epoch": 5.812358856201172e-05,
      "model_forward_time": 1.8364109992980957,
      "step": 60947
    },
    {
      "epoch": 5.812358856201172e-05,
      "step": 60947,
      "training_step_time": 8.375697374343872
    },
    {
      "epoch": 5.8124542236328125e-05,
      "model_forward_time": 1.713318109512329,
      "step": 60948
    },
    {
      "epoch": 5.8124542236328125e-05,
      "step": 60948,
      "training_step_time": 7.535781383514404
    },
    {
      "epoch": 5.812549591064453e-05,
      "model_forward_time": 1.3442206382751465,
      "step": 60949
    },
    {
      "epoch": 5.812549591064453e-05,
      "step": 60949,
      "training_step_time": 7.585708379745483
    },
    {
      "epoch": 5.812644958496094e-05,
      "grad_norm": 0.10413817316293716,
      "learning_rate": 3.6210891687208306e-06,
      "loss": 0.0554,
      "step": 60950
    },
    {
      "epoch": 5.812644958496094e-05,
      "model_forward_time": 1.728935718536377,
      "step": 60950
    },
    {
      "dynamics_loss_avg": 0.05169565286487341,
      "epoch": 5.812644958496094e-05,
      "step": 60950
    },
    {
      "action_loss_avg": 0.0012773119495250284,
      "epoch": 5.812644958496094e-05,
      "step": 60950
    },
    {
      "epoch": 5.812644958496094e-05,
      "step": 60950,
      "training_step_time": 7.857670545578003
    },
    {
      "epoch": 5.8127403259277344e-05,
      "model_forward_time": 1.3510422706604004,
      "step": 60951
    },
    {
      "epoch": 5.8127403259277344e-05,
      "step": 60951,
      "training_step_time": 7.943031311035156
    },
    {
      "epoch": 5.812835693359375e-05,
      "model_forward_time": 1.7361059188842773,
      "step": 60952
    },
    {
      "epoch": 5.812835693359375e-05,
      "step": 60952,
      "training_step_time": 7.8349549770355225
    },
    {
      "epoch": 5.812931060791016e-05,
      "model_forward_time": 1.7319047451019287,
      "step": 60953
    },
    {
      "epoch": 5.812931060791016e-05,
      "step": 60953,
      "training_step_time": 7.893918752670288
    },
    {
      "epoch": 5.813026428222656e-05,
      "model_forward_time": 2.827615737915039,
      "step": 60954
    },
    {
      "epoch": 5.813026428222656e-05,
      "step": 60954,
      "training_step_time": 7.8930768966674805
    },
    {
      "epoch": 5.813121795654297e-05,
      "model_forward_time": 1.7192118167877197,
      "step": 60955
    },
    {
      "epoch": 5.813121795654297e-05,
      "step": 60955,
      "training_step_time": 7.637908220291138
    },
    {
      "epoch": 5.813217163085938e-05,
      "model_forward_time": 1.8150794506072998,
      "step": 60956
    },
    {
      "epoch": 5.813217163085938e-05,
      "step": 60956,
      "training_step_time": 8.329267501831055
    },
    {
      "epoch": 5.813312530517578e-05,
      "model_forward_time": 1.7202136516571045,
      "step": 60957
    },
    {
      "epoch": 5.813312530517578e-05,
      "step": 60957,
      "training_step_time": 7.50560188293457
    },
    {
      "epoch": 5.8134078979492186e-05,
      "model_forward_time": 1.7187106609344482,
      "step": 60958
    },
    {
      "epoch": 5.8134078979492186e-05,
      "step": 60958,
      "training_step_time": 7.774251937866211
    },
    {
      "epoch": 5.81350326538086e-05,
      "model_forward_time": 1.7296769618988037,
      "step": 60959
    },
    {
      "epoch": 5.81350326538086e-05,
      "step": 60959,
      "training_step_time": 7.784348726272583
    },
    {
      "epoch": 5.8135986328125e-05,
      "grad_norm": 0.12572507560253143,
      "learning_rate": 3.6194998956242277e-06,
      "loss": 0.0579,
      "step": 60960
    },
    {
      "epoch": 5.8135986328125e-05,
      "model_forward_time": 1.719764232635498,
      "step": 60960
    },
    {
      "dynamics_loss_avg": 0.05860292268916965,
      "epoch": 5.8135986328125e-05,
      "step": 60960
    },
    {
      "action_loss_avg": 0.00141478318546433,
      "epoch": 5.8135986328125e-05,
      "step": 60960
    },
    {
      "epoch": 5.8135986328125e-05,
      "step": 60960,
      "training_step_time": 7.731994867324829
    },
    {
      "epoch": 5.8136940002441405e-05,
      "model_forward_time": 1.7150166034698486,
      "step": 60961
    },
    {
      "epoch": 5.8136940002441405e-05,
      "step": 60961,
      "training_step_time": 8.202057123184204
    },
    {
      "epoch": 5.8137893676757816e-05,
      "model_forward_time": 1.7231426239013672,
      "step": 60962
    },
    {
      "epoch": 5.8137893676757816e-05,
      "step": 60962,
      "training_step_time": 9.942816972732544
    },
    {
      "epoch": 5.813884735107422e-05,
      "model_forward_time": 1.7212395668029785,
      "step": 60963
    },
    {
      "epoch": 5.813884735107422e-05,
      "step": 60963,
      "training_step_time": 7.751003742218018
    },
    {
      "epoch": 5.8139801025390624e-05,
      "model_forward_time": 2.874847412109375,
      "step": 60964
    },
    {
      "epoch": 5.8139801025390624e-05,
      "step": 60964,
      "training_step_time": 7.719839096069336
    },
    {
      "epoch": 5.8140754699707035e-05,
      "model_forward_time": 1.7821192741394043,
      "step": 60965
    },
    {
      "epoch": 5.8140754699707035e-05,
      "step": 60965,
      "training_step_time": 7.756689548492432
    },
    {
      "epoch": 5.814170837402344e-05,
      "model_forward_time": 1.7141759395599365,
      "step": 60966
    },
    {
      "epoch": 5.814170837402344e-05,
      "step": 60966,
      "training_step_time": 8.598976612091064
    },
    {
      "epoch": 5.814266204833984e-05,
      "model_forward_time": 1.7550570964813232,
      "step": 60967
    },
    {
      "epoch": 5.814266204833984e-05,
      "step": 60967,
      "training_step_time": 7.640289068222046
    },
    {
      "epoch": 5.814361572265625e-05,
      "model_forward_time": 1.7665085792541504,
      "step": 60968
    },
    {
      "epoch": 5.814361572265625e-05,
      "step": 60968,
      "training_step_time": 7.6741204261779785
    },
    {
      "epoch": 5.814456939697266e-05,
      "model_forward_time": 1.7631828784942627,
      "step": 60969
    },
    {
      "epoch": 5.814456939697266e-05,
      "step": 60969,
      "training_step_time": 7.5995728969573975
    },
    {
      "epoch": 5.814552307128906e-05,
      "grad_norm": 0.1086977869272232,
      "learning_rate": 3.6179107734970403e-06,
      "loss": 0.0586,
      "step": 60970
    },
    {
      "epoch": 5.814552307128906e-05,
      "model_forward_time": 1.7611050605773926,
      "step": 60970
    },
    {
      "dynamics_loss_avg": 0.06164070442318916,
      "epoch": 5.814552307128906e-05,
      "step": 60970
    },
    {
      "action_loss_avg": 0.0016827589890453965,
      "epoch": 5.814552307128906e-05,
      "step": 60970
    },
    {
      "epoch": 5.814552307128906e-05,
      "step": 60970,
      "training_step_time": 9.109151840209961
    },
    {
      "epoch": 5.8146476745605465e-05,
      "model_forward_time": 1.7589631080627441,
      "step": 60971
    },
    {
      "epoch": 5.8146476745605465e-05,
      "step": 60971,
      "training_step_time": 7.583214044570923
    },
    {
      "epoch": 5.8147430419921876e-05,
      "model_forward_time": 1.7311124801635742,
      "step": 60972
    },
    {
      "epoch": 5.8147430419921876e-05,
      "step": 60972,
      "training_step_time": 8.672823667526245
    },
    {
      "epoch": 5.814838409423828e-05,
      "model_forward_time": 2.1382696628570557,
      "step": 60973
    },
    {
      "epoch": 5.814838409423828e-05,
      "step": 60973,
      "training_step_time": 8.593055248260498
    },
    {
      "epoch": 5.8149337768554684e-05,
      "model_forward_time": 1.7455811500549316,
      "step": 60974
    },
    {
      "epoch": 5.8149337768554684e-05,
      "step": 60974,
      "training_step_time": 8.371423482894897
    },
    {
      "epoch": 5.8150291442871095e-05,
      "model_forward_time": 1.725254774093628,
      "step": 60975
    },
    {
      "epoch": 5.8150291442871095e-05,
      "step": 60975,
      "training_step_time": 7.641518592834473
    },
    {
      "epoch": 5.81512451171875e-05,
      "model_forward_time": 1.721968412399292,
      "step": 60976
    },
    {
      "epoch": 5.81512451171875e-05,
      "step": 60976,
      "training_step_time": 7.586370229721069
    },
    {
      "epoch": 5.81521987915039e-05,
      "model_forward_time": 1.7327885627746582,
      "step": 60977
    },
    {
      "epoch": 5.81521987915039e-05,
      "step": 60977,
      "training_step_time": 7.549111604690552
    },
    {
      "epoch": 5.8153152465820314e-05,
      "model_forward_time": 1.7247345447540283,
      "step": 60978
    },
    {
      "epoch": 5.8153152465820314e-05,
      "step": 60978,
      "training_step_time": 7.498307466506958
    },
    {
      "epoch": 5.815410614013672e-05,
      "model_forward_time": 1.8037300109863281,
      "step": 60979
    },
    {
      "epoch": 5.815410614013672e-05,
      "step": 60979,
      "training_step_time": 7.563265562057495
    },
    {
      "epoch": 5.815505981445312e-05,
      "grad_norm": 0.08926797658205032,
      "learning_rate": 3.616321802513053e-06,
      "loss": 0.0566,
      "step": 60980
    },
    {
      "epoch": 5.815505981445312e-05,
      "model_forward_time": 1.7340617179870605,
      "step": 60980
    },
    {
      "dynamics_loss_avg": 0.05616562571376562,
      "epoch": 5.815505981445312e-05,
      "step": 60980
    },
    {
      "action_loss_avg": 0.0022397588356398044,
      "epoch": 5.815505981445312e-05,
      "step": 60980
    },
    {
      "epoch": 5.815505981445312e-05,
      "step": 60980,
      "training_step_time": 7.6194446086883545
    },
    {
      "epoch": 5.815601348876953e-05,
      "model_forward_time": 1.7274532318115234,
      "step": 60981
    },
    {
      "epoch": 5.815601348876953e-05,
      "step": 60981,
      "training_step_time": 8.674866437911987
    },
    {
      "epoch": 5.815696716308594e-05,
      "model_forward_time": 2.9093141555786133,
      "step": 60982
    },
    {
      "epoch": 5.815696716308594e-05,
      "step": 60982,
      "training_step_time": 8.944290161132812
    },
    {
      "epoch": 5.815792083740234e-05,
      "model_forward_time": 1.728790521621704,
      "step": 60983
    },
    {
      "epoch": 5.815792083740234e-05,
      "step": 60983,
      "training_step_time": 8.334920644760132
    },
    {
      "epoch": 5.815887451171875e-05,
      "model_forward_time": 1.7254924774169922,
      "step": 60984
    },
    {
      "epoch": 5.815887451171875e-05,
      "step": 60984,
      "training_step_time": 7.615458726882935
    },
    {
      "epoch": 5.8159828186035156e-05,
      "model_forward_time": 1.7399797439575195,
      "step": 60985
    },
    {
      "epoch": 5.8159828186035156e-05,
      "step": 60985,
      "training_step_time": 8.162720918655396
    },
    {
      "epoch": 5.816078186035156e-05,
      "model_forward_time": 1.7277660369873047,
      "step": 60986
    },
    {
      "epoch": 5.816078186035156e-05,
      "step": 60986,
      "training_step_time": 7.572282552719116
    },
    {
      "epoch": 5.816173553466797e-05,
      "model_forward_time": 1.7277922630310059,
      "step": 60987
    },
    {
      "epoch": 5.816173553466797e-05,
      "step": 60987,
      "training_step_time": 7.659468650817871
    },
    {
      "epoch": 5.8162689208984375e-05,
      "model_forward_time": 1.7852494716644287,
      "step": 60988
    },
    {
      "epoch": 5.8162689208984375e-05,
      "step": 60988,
      "training_step_time": 8.351745367050171
    },
    {
      "epoch": 5.816364288330078e-05,
      "model_forward_time": 1.726426362991333,
      "step": 60989
    },
    {
      "epoch": 5.816364288330078e-05,
      "step": 60989,
      "training_step_time": 8.661404132843018
    },
    {
      "epoch": 5.816459655761719e-05,
      "grad_norm": 0.07486908882856369,
      "learning_rate": 3.6147329828460325e-06,
      "loss": 0.0566,
      "step": 60990
    },
    {
      "epoch": 5.816459655761719e-05,
      "model_forward_time": 1.7374732494354248,
      "step": 60990
    },
    {
      "dynamics_loss_avg": 0.054825997166335586,
      "epoch": 5.816459655761719e-05,
      "step": 60990
    },
    {
      "action_loss_avg": 0.0017777709348592908,
      "epoch": 5.816459655761719e-05,
      "step": 60990
    },
    {
      "epoch": 5.816459655761719e-05,
      "step": 60990,
      "training_step_time": 8.617736339569092
    },
    {
      "epoch": 5.8165550231933593e-05,
      "model_forward_time": 2.8068299293518066,
      "step": 60991
    },
    {
      "epoch": 5.8165550231933593e-05,
      "step": 60991,
      "training_step_time": 8.553248643875122
    },
    {
      "epoch": 5.816650390625e-05,
      "model_forward_time": 1.7360785007476807,
      "step": 60992
    },
    {
      "epoch": 5.816650390625e-05,
      "step": 60992,
      "training_step_time": 7.542745113372803
    },
    {
      "epoch": 5.816745758056641e-05,
      "model_forward_time": 1.7264020442962646,
      "step": 60993
    },
    {
      "epoch": 5.816745758056641e-05,
      "step": 60993,
      "training_step_time": 8.16178822517395
    },
    {
      "epoch": 5.816841125488281e-05,
      "model_forward_time": 1.7178664207458496,
      "step": 60994
    },
    {
      "epoch": 5.816841125488281e-05,
      "step": 60994,
      "training_step_time": 7.368577718734741
    },
    {
      "epoch": 5.8169364929199216e-05,
      "model_forward_time": 1.7284016609191895,
      "step": 60995
    },
    {
      "epoch": 5.8169364929199216e-05,
      "step": 60995,
      "training_step_time": 7.647289991378784
    },
    {
      "epoch": 5.817031860351563e-05,
      "model_forward_time": 1.0311174392700195,
      "step": 60996
    },
    {
      "epoch": 5.817031860351563e-05,
      "step": 60996,
      "training_step_time": 7.702866792678833
    },
    {
      "epoch": 5.817127227783203e-05,
      "model_forward_time": 1.7546026706695557,
      "step": 60997
    },
    {
      "epoch": 5.817127227783203e-05,
      "step": 60997,
      "training_step_time": 7.574016094207764
    },
    {
      "epoch": 5.8172225952148435e-05,
      "model_forward_time": 1.8143901824951172,
      "step": 60998
    },
    {
      "epoch": 5.8172225952148435e-05,
      "step": 60998,
      "training_step_time": 7.95009183883667
    },
    {
      "epoch": 5.8173179626464846e-05,
      "model_forward_time": 1.759612798690796,
      "step": 60999
    },
    {
      "epoch": 5.8173179626464846e-05,
      "step": 60999,
      "training_step_time": 9.58156156539917
    },
    {
      "epoch": 5.817413330078125e-05,
      "grad_norm": 0.12013895064592361,
      "learning_rate": 3.613144314669731e-06,
      "loss": 0.0547,
      "step": 61000
    },
    {
      "epoch": 5.817413330078125e-05,
      "model_forward_time": 3.4072959423065186,
      "step": 61000
    },
    {
      "dynamics_loss_avg": 0.05066862497478723,
      "epoch": 5.817413330078125e-05,
      "step": 61000
    },
    {
      "action_loss_avg": 0.0010725795262260362,
      "epoch": 5.817413330078125e-05,
      "step": 61000
    },
    {
      "epoch": 5.817413330078125e-05,
      "step": 61000,
      "training_step_time": 9.773720741271973
    },
    {
      "epoch": 5.8175086975097654e-05,
      "model_forward_time": 1.8856885433197021,
      "step": 61001
    },
    {
      "epoch": 5.8175086975097654e-05,
      "step": 61001,
      "training_step_time": 10.047481536865234
    },
    {
      "epoch": 5.8176040649414065e-05,
      "model_forward_time": 1.8743693828582764,
      "step": 61002
    },
    {
      "epoch": 5.8176040649414065e-05,
      "step": 61002,
      "training_step_time": 8.300939559936523
    },
    {
      "epoch": 5.817699432373047e-05,
      "model_forward_time": 1.8613488674163818,
      "step": 61003
    },
    {
      "epoch": 5.817699432373047e-05,
      "step": 61003,
      "training_step_time": 8.31698226928711
    },
    {
      "epoch": 5.817794799804687e-05,
      "model_forward_time": 1.9049298763275146,
      "step": 61004
    },
    {
      "epoch": 5.817794799804687e-05,
      "step": 61004,
      "training_step_time": 8.456414461135864
    },
    {
      "epoch": 5.8178901672363284e-05,
      "model_forward_time": 1.890864610671997,
      "step": 61005
    },
    {
      "epoch": 5.8178901672363284e-05,
      "step": 61005,
      "training_step_time": 7.713537216186523
    },
    {
      "epoch": 5.817985534667969e-05,
      "model_forward_time": 1.9168205261230469,
      "step": 61006
    },
    {
      "epoch": 5.817985534667969e-05,
      "step": 61006,
      "training_step_time": 9.41738224029541
    },
    {
      "epoch": 5.818080902099609e-05,
      "model_forward_time": 1.8928189277648926,
      "step": 61007
    },
    {
      "epoch": 5.818080902099609e-05,
      "step": 61007,
      "training_step_time": 8.170706510543823
    },
    {
      "epoch": 5.81817626953125e-05,
      "model_forward_time": 1.9089877605438232,
      "step": 61008
    },
    {
      "epoch": 5.81817626953125e-05,
      "step": 61008,
      "training_step_time": 11.936150312423706
    },
    {
      "epoch": 5.8182716369628907e-05,
      "model_forward_time": 1.8876850605010986,
      "step": 61009
    },
    {
      "epoch": 5.8182716369628907e-05,
      "step": 61009,
      "training_step_time": 7.822999477386475
    },
    {
      "epoch": 5.818367004394531e-05,
      "grad_norm": 0.0946710929274559,
      "learning_rate": 3.611555798157879e-06,
      "loss": 0.0551,
      "step": 61010
    },
    {
      "epoch": 5.818367004394531e-05,
      "model_forward_time": 2.93811297416687,
      "step": 61010
    },
    {
      "dynamics_loss_avg": 0.050266532227396965,
      "epoch": 5.818367004394531e-05,
      "step": 61010
    },
    {
      "action_loss_avg": 0.0022189141542185097,
      "epoch": 5.818367004394531e-05,
      "step": 61010
    },
    {
      "epoch": 5.818367004394531e-05,
      "step": 61010,
      "training_step_time": 9.636843204498291
    },
    {
      "epoch": 5.818462371826172e-05,
      "model_forward_time": 1.8971478939056396,
      "step": 61011
    },
    {
      "epoch": 5.818462371826172e-05,
      "step": 61011,
      "training_step_time": 7.71950626373291
    },
    {
      "epoch": 5.8185577392578125e-05,
      "model_forward_time": 1.9256904125213623,
      "step": 61012
    },
    {
      "epoch": 5.8185577392578125e-05,
      "step": 61012,
      "training_step_time": 7.6278746128082275
    },
    {
      "epoch": 5.818653106689453e-05,
      "model_forward_time": 1.9128670692443848,
      "step": 61013
    },
    {
      "epoch": 5.818653106689453e-05,
      "step": 61013,
      "training_step_time": 7.571398735046387
    },
    {
      "epoch": 5.818748474121094e-05,
      "model_forward_time": 1.8955073356628418,
      "step": 61014
    },
    {
      "epoch": 5.818748474121094e-05,
      "step": 61014,
      "training_step_time": 7.767313718795776
    },
    {
      "epoch": 5.8188438415527344e-05,
      "model_forward_time": 1.888909101486206,
      "step": 61015
    },
    {
      "epoch": 5.8188438415527344e-05,
      "step": 61015,
      "training_step_time": 8.424929857254028
    },
    {
      "epoch": 5.818939208984375e-05,
      "model_forward_time": 1.8972365856170654,
      "step": 61016
    },
    {
      "epoch": 5.818939208984375e-05,
      "step": 61016,
      "training_step_time": 7.600904703140259
    },
    {
      "epoch": 5.819034576416016e-05,
      "model_forward_time": 1.9194190502166748,
      "step": 61017
    },
    {
      "epoch": 5.819034576416016e-05,
      "step": 61017,
      "training_step_time": 7.749758720397949
    },
    {
      "epoch": 5.819129943847656e-05,
      "model_forward_time": 1.8588764667510986,
      "step": 61018
    },
    {
      "epoch": 5.819129943847656e-05,
      "step": 61018,
      "training_step_time": 7.722274541854858
    },
    {
      "epoch": 5.819225311279297e-05,
      "model_forward_time": 2.972116708755493,
      "step": 61019
    },
    {
      "epoch": 5.819225311279297e-05,
      "step": 61019,
      "training_step_time": 9.312011241912842
    },
    {
      "epoch": 5.819320678710938e-05,
      "grad_norm": 0.11920181661844254,
      "learning_rate": 3.6099674334842005e-06,
      "loss": 0.0557,
      "step": 61020
    },
    {
      "epoch": 5.819320678710938e-05,
      "model_forward_time": 1.7040092945098877,
      "step": 61020
    },
    {
      "dynamics_loss_avg": 0.04944935236126184,
      "epoch": 5.819320678710938e-05,
      "step": 61020
    },
    {
      "action_loss_avg": 0.0009525218047201633,
      "epoch": 5.819320678710938e-05,
      "step": 61020
    },
    {
      "epoch": 5.819320678710938e-05,
      "step": 61020,
      "training_step_time": 8.561800479888916
    },
    {
      "epoch": 5.819416046142578e-05,
      "model_forward_time": 1.7228877544403076,
      "step": 61021
    },
    {
      "epoch": 5.819416046142578e-05,
      "step": 61021,
      "training_step_time": 7.6047751903533936
    },
    {
      "epoch": 5.8195114135742186e-05,
      "model_forward_time": 1.3596131801605225,
      "step": 61022
    },
    {
      "epoch": 5.8195114135742186e-05,
      "step": 61022,
      "training_step_time": 7.701625823974609
    },
    {
      "epoch": 5.81960678100586e-05,
      "model_forward_time": 1.7288525104522705,
      "step": 61023
    },
    {
      "epoch": 5.81960678100586e-05,
      "step": 61023,
      "training_step_time": 7.8297507762908936
    },
    {
      "epoch": 5.8197021484375e-05,
      "model_forward_time": 1.7277181148529053,
      "step": 61024
    },
    {
      "epoch": 5.8197021484375e-05,
      "step": 61024,
      "training_step_time": 7.64990496635437
    },
    {
      "epoch": 5.8197975158691405e-05,
      "model_forward_time": 1.8107032775878906,
      "step": 61025
    },
    {
      "epoch": 5.8197975158691405e-05,
      "step": 61025,
      "training_step_time": 8.20387887954712
    },
    {
      "epoch": 5.8198928833007816e-05,
      "model_forward_time": 1.7299840450286865,
      "step": 61026
    },
    {
      "epoch": 5.8198928833007816e-05,
      "step": 61026,
      "training_step_time": 8.188791990280151
    },
    {
      "epoch": 5.819988250732422e-05,
      "model_forward_time": 1.7368590831756592,
      "step": 61027
    },
    {
      "epoch": 5.819988250732422e-05,
      "step": 61027,
      "training_step_time": 7.750728130340576
    },
    {
      "epoch": 5.8200836181640624e-05,
      "model_forward_time": 2.8425986766815186,
      "step": 61028
    },
    {
      "epoch": 5.8200836181640624e-05,
      "step": 61028,
      "training_step_time": 8.479848146438599
    },
    {
      "epoch": 5.8201789855957035e-05,
      "model_forward_time": 1.7140545845031738,
      "step": 61029
    },
    {
      "epoch": 5.8201789855957035e-05,
      "step": 61029,
      "training_step_time": 7.766668319702148
    },
    {
      "epoch": 5.820274353027344e-05,
      "grad_norm": 0.09859456866979599,
      "learning_rate": 3.6083792208223932e-06,
      "loss": 0.0591,
      "step": 61030
    },
    {
      "epoch": 5.820274353027344e-05,
      "model_forward_time": 1.7156412601470947,
      "step": 61030
    },
    {
      "dynamics_loss_avg": 0.048886610008776185,
      "epoch": 5.820274353027344e-05,
      "step": 61030
    },
    {
      "action_loss_avg": 0.0012034750776365398,
      "epoch": 5.820274353027344e-05,
      "step": 61030
    },
    {
      "epoch": 5.820274353027344e-05,
      "step": 61030,
      "training_step_time": 7.657592296600342
    },
    {
      "epoch": 5.820369720458984e-05,
      "model_forward_time": 1.7314841747283936,
      "step": 61031
    },
    {
      "epoch": 5.820369720458984e-05,
      "step": 61031,
      "training_step_time": 7.758317470550537
    },
    {
      "epoch": 5.820465087890625e-05,
      "model_forward_time": 1.720660924911499,
      "step": 61032
    },
    {
      "epoch": 5.820465087890625e-05,
      "step": 61032,
      "training_step_time": 8.448718070983887
    },
    {
      "epoch": 5.820560455322266e-05,
      "model_forward_time": 1.7237513065338135,
      "step": 61033
    },
    {
      "epoch": 5.820560455322266e-05,
      "step": 61033,
      "training_step_time": 7.868157148361206
    },
    {
      "epoch": 5.820655822753906e-05,
      "model_forward_time": 1.7736396789550781,
      "step": 61034
    },
    {
      "epoch": 5.820655822753906e-05,
      "step": 61034,
      "training_step_time": 7.652761936187744
    },
    {
      "epoch": 5.8207511901855465e-05,
      "model_forward_time": 1.712252140045166,
      "step": 61035
    },
    {
      "epoch": 5.8207511901855465e-05,
      "step": 61035,
      "training_step_time": 8.575289964675903
    },
    {
      "epoch": 5.8208465576171876e-05,
      "model_forward_time": 1.727682113647461,
      "step": 61036
    },
    {
      "epoch": 5.8208465576171876e-05,
      "step": 61036,
      "training_step_time": 8.247848987579346
    },
    {
      "epoch": 5.820941925048828e-05,
      "model_forward_time": 2.8655781745910645,
      "step": 61037
    },
    {
      "epoch": 5.820941925048828e-05,
      "step": 61037,
      "training_step_time": 9.184781789779663
    },
    {
      "epoch": 5.8210372924804684e-05,
      "model_forward_time": 1.7402820587158203,
      "step": 61038
    },
    {
      "epoch": 5.8210372924804684e-05,
      "step": 61038,
      "training_step_time": 7.728928327560425
    },
    {
      "epoch": 5.8211326599121095e-05,
      "model_forward_time": 1.7219557762145996,
      "step": 61039
    },
    {
      "epoch": 5.8211326599121095e-05,
      "step": 61039,
      "training_step_time": 8.524276733398438
    },
    {
      "epoch": 5.82122802734375e-05,
      "grad_norm": 0.10266860574483871,
      "learning_rate": 3.6067911603461418e-06,
      "loss": 0.0595,
      "step": 61040
    },
    {
      "epoch": 5.82122802734375e-05,
      "model_forward_time": 1.7266290187835693,
      "step": 61040
    },
    {
      "dynamics_loss_avg": 0.046070471778512,
      "epoch": 5.82122802734375e-05,
      "step": 61040
    },
    {
      "action_loss_avg": 0.0014531971886754037,
      "epoch": 5.82122802734375e-05,
      "step": 61040
    },
    {
      "epoch": 5.82122802734375e-05,
      "step": 61040,
      "training_step_time": 8.454400300979614
    },
    {
      "epoch": 5.82132339477539e-05,
      "model_forward_time": 1.7232866287231445,
      "step": 61041
    },
    {
      "epoch": 5.82132339477539e-05,
      "step": 61041,
      "training_step_time": 8.488280057907104
    },
    {
      "epoch": 5.8214187622070314e-05,
      "model_forward_time": 1.7333223819732666,
      "step": 61042
    },
    {
      "epoch": 5.8214187622070314e-05,
      "step": 61042,
      "training_step_time": 8.561978816986084
    },
    {
      "epoch": 5.821514129638672e-05,
      "model_forward_time": 1.7312660217285156,
      "step": 61043
    },
    {
      "epoch": 5.821514129638672e-05,
      "step": 61043,
      "training_step_time": 8.611297130584717
    },
    {
      "epoch": 5.821609497070312e-05,
      "model_forward_time": 1.723801851272583,
      "step": 61044
    },
    {
      "epoch": 5.821609497070312e-05,
      "step": 61044,
      "training_step_time": 8.52816915512085
    },
    {
      "epoch": 5.821704864501953e-05,
      "model_forward_time": 1.7209820747375488,
      "step": 61045
    },
    {
      "epoch": 5.821704864501953e-05,
      "step": 61045,
      "training_step_time": 8.879946231842041
    },
    {
      "epoch": 5.821800231933594e-05,
      "model_forward_time": 2.808678388595581,
      "step": 61046
    },
    {
      "epoch": 5.821800231933594e-05,
      "step": 61046,
      "training_step_time": 9.785957336425781
    },
    {
      "epoch": 5.821895599365234e-05,
      "model_forward_time": 1.728881597518921,
      "step": 61047
    },
    {
      "epoch": 5.821895599365234e-05,
      "step": 61047,
      "training_step_time": 7.639343976974487
    },
    {
      "epoch": 5.821990966796875e-05,
      "model_forward_time": 1.71932053565979,
      "step": 61048
    },
    {
      "epoch": 5.821990966796875e-05,
      "step": 61048,
      "training_step_time": 8.502569437026978
    },
    {
      "epoch": 5.8220863342285156e-05,
      "model_forward_time": 1.7234234809875488,
      "step": 61049
    },
    {
      "epoch": 5.8220863342285156e-05,
      "step": 61049,
      "training_step_time": 7.942239284515381
    },
    {
      "epoch": 5.822181701660156e-05,
      "grad_norm": 0.09328813850879669,
      "learning_rate": 3.6052032522291137e-06,
      "loss": 0.0584,
      "step": 61050
    },
    {
      "epoch": 5.822181701660156e-05,
      "model_forward_time": 1.71638822555542,
      "step": 61050
    },
    {
      "dynamics_loss_avg": 0.052271680906414986,
      "epoch": 5.822181701660156e-05,
      "step": 61050
    },
    {
      "action_loss_avg": 0.002203068498056382,
      "epoch": 5.822181701660156e-05,
      "step": 61050
    },
    {
      "epoch": 5.822181701660156e-05,
      "step": 61050,
      "training_step_time": 8.127775192260742
    },
    {
      "epoch": 5.822277069091797e-05,
      "model_forward_time": 1.7330372333526611,
      "step": 61051
    },
    {
      "epoch": 5.822277069091797e-05,
      "step": 61051,
      "training_step_time": 7.805521488189697
    },
    {
      "epoch": 5.8223724365234375e-05,
      "model_forward_time": 1.7238819599151611,
      "step": 61052
    },
    {
      "epoch": 5.8223724365234375e-05,
      "step": 61052,
      "training_step_time": 8.746991634368896
    },
    {
      "epoch": 5.822467803955078e-05,
      "model_forward_time": 1.7402100563049316,
      "step": 61053
    },
    {
      "epoch": 5.822467803955078e-05,
      "step": 61053,
      "training_step_time": 9.54019570350647
    },
    {
      "epoch": 5.822563171386719e-05,
      "model_forward_time": 1.7219676971435547,
      "step": 61054
    },
    {
      "epoch": 5.822563171386719e-05,
      "step": 61054,
      "training_step_time": 6.722510099411011
    },
    {
      "epoch": 5.8226585388183593e-05,
      "model_forward_time": 1.7216293811798096,
      "step": 61055
    },
    {
      "epoch": 5.8226585388183593e-05,
      "step": 61055,
      "training_step_time": 6.713907718658447
    },
    {
      "epoch": 5.82275390625e-05,
      "model_forward_time": 1.7306146621704102,
      "step": 61056
    },
    {
      "epoch": 5.82275390625e-05,
      "step": 61056,
      "training_step_time": 6.72126030921936
    },
    {
      "epoch": 5.822849273681641e-05,
      "model_forward_time": 1.7436909675598145,
      "step": 61057
    },
    {
      "epoch": 5.822849273681641e-05,
      "step": 61057,
      "training_step_time": 6.693263292312622
    },
    {
      "epoch": 5.822944641113281e-05,
      "model_forward_time": 1.7266979217529297,
      "step": 61058
    },
    {
      "epoch": 5.822944641113281e-05,
      "step": 61058,
      "training_step_time": 6.700347661972046
    },
    {
      "epoch": 5.8230400085449216e-05,
      "model_forward_time": 1.7258429527282715,
      "step": 61059
    },
    {
      "epoch": 5.8230400085449216e-05,
      "step": 61059,
      "training_step_time": 6.767223596572876
    },
    {
      "epoch": 5.823135375976563e-05,
      "grad_norm": 0.09726548939943314,
      "learning_rate": 3.6036154966449622e-06,
      "loss": 0.0593,
      "step": 61060
    },
    {
      "epoch": 5.823135375976563e-05,
      "model_forward_time": 1.7042863368988037,
      "step": 61060
    },
    {
      "dynamics_loss_avg": 0.046737225539982316,
      "epoch": 5.823135375976563e-05,
      "step": 61060
    },
    {
      "action_loss_avg": 0.0019116799026960506,
      "epoch": 5.823135375976563e-05,
      "step": 61060
    },
    {
      "epoch": 5.823135375976563e-05,
      "step": 61060,
      "training_step_time": 6.754323482513428
    },
    {
      "epoch": 5.823230743408203e-05,
      "model_forward_time": 1.7271666526794434,
      "step": 61061
    },
    {
      "epoch": 5.823230743408203e-05,
      "step": 61061,
      "training_step_time": 7.8048436641693115
    },
    {
      "epoch": 5.8233261108398435e-05,
      "model_forward_time": 2.9239001274108887,
      "step": 61062
    },
    {
      "epoch": 5.8233261108398435e-05,
      "step": 61062,
      "training_step_time": 7.713068008422852
    },
    {
      "epoch": 5.8234214782714846e-05,
      "model_forward_time": 1.7348742485046387,
      "step": 61063
    },
    {
      "epoch": 5.8234214782714846e-05,
      "step": 61063,
      "training_step_time": 7.912294864654541
    },
    {
      "epoch": 5.823516845703125e-05,
      "model_forward_time": 1.723963737487793,
      "step": 61064
    },
    {
      "epoch": 5.823516845703125e-05,
      "step": 61064,
      "training_step_time": 6.724562883377075
    },
    {
      "epoch": 5.8236122131347654e-05,
      "model_forward_time": 1.7432808876037598,
      "step": 61065
    },
    {
      "epoch": 5.8236122131347654e-05,
      "step": 61065,
      "training_step_time": 6.751811981201172
    },
    {
      "epoch": 5.8237075805664065e-05,
      "model_forward_time": 1.7334213256835938,
      "step": 61066
    },
    {
      "epoch": 5.8237075805664065e-05,
      "step": 61066,
      "training_step_time": 6.763572454452515
    },
    {
      "epoch": 5.823802947998047e-05,
      "model_forward_time": 1.7343673706054688,
      "step": 61067
    },
    {
      "epoch": 5.823802947998047e-05,
      "step": 61067,
      "training_step_time": 6.815231800079346
    },
    {
      "epoch": 5.823898315429687e-05,
      "model_forward_time": 1.7227110862731934,
      "step": 61068
    },
    {
      "epoch": 5.823898315429687e-05,
      "step": 61068,
      "training_step_time": 6.808617353439331
    },
    {
      "epoch": 5.8239936828613284e-05,
      "model_forward_time": 1.7283425331115723,
      "step": 61069
    },
    {
      "epoch": 5.8239936828613284e-05,
      "step": 61069,
      "training_step_time": 6.786744594573975
    },
    {
      "epoch": 5.824089050292969e-05,
      "grad_norm": 0.10113921016454697,
      "learning_rate": 3.6020278937673202e-06,
      "loss": 0.0582,
      "step": 61070
    },
    {
      "epoch": 5.824089050292969e-05,
      "model_forward_time": 1.7145156860351562,
      "step": 61070
    },
    {
      "dynamics_loss_avg": 0.055761652253568175,
      "epoch": 5.824089050292969e-05,
      "step": 61070
    },
    {
      "action_loss_avg": 0.0008622622728580609,
      "epoch": 5.824089050292969e-05,
      "step": 61070
    },
    {
      "epoch": 5.824089050292969e-05,
      "step": 61070,
      "training_step_time": 7.371476650238037
    },
    {
      "epoch": 5.824184417724609e-05,
      "model_forward_time": 2.887537717819214,
      "step": 61071
    },
    {
      "epoch": 5.824184417724609e-05,
      "step": 61071,
      "training_step_time": 7.65958833694458
    },
    {
      "epoch": 5.82427978515625e-05,
      "model_forward_time": 1.732398271560669,
      "step": 61072
    },
    {
      "epoch": 5.82427978515625e-05,
      "step": 61072,
      "training_step_time": 7.917423725128174
    },
    {
      "epoch": 5.824375152587891e-05,
      "model_forward_time": 1.7282540798187256,
      "step": 61073
    },
    {
      "epoch": 5.824375152587891e-05,
      "step": 61073,
      "training_step_time": 7.231356620788574
    },
    {
      "epoch": 5.824470520019531e-05,
      "model_forward_time": 1.7242591381072998,
      "step": 61074
    },
    {
      "epoch": 5.824470520019531e-05,
      "step": 61074,
      "training_step_time": 6.762829542160034
    },
    {
      "epoch": 5.824565887451172e-05,
      "model_forward_time": 1.7315809726715088,
      "step": 61075
    },
    {
      "epoch": 5.824565887451172e-05,
      "step": 61075,
      "training_step_time": 6.736280918121338
    },
    {
      "epoch": 5.8246612548828125e-05,
      "model_forward_time": 1.7373075485229492,
      "step": 61076
    },
    {
      "epoch": 5.8246612548828125e-05,
      "step": 61076,
      "training_step_time": 6.7601659297943115
    },
    {
      "epoch": 5.824756622314453e-05,
      "model_forward_time": 1.7279112339019775,
      "step": 61077
    },
    {
      "epoch": 5.824756622314453e-05,
      "step": 61077,
      "training_step_time": 6.689123630523682
    },
    {
      "epoch": 5.824851989746094e-05,
      "model_forward_time": 1.7325806617736816,
      "step": 61078
    },
    {
      "epoch": 5.824851989746094e-05,
      "step": 61078,
      "training_step_time": 6.723217725753784
    },
    {
      "epoch": 5.8249473571777344e-05,
      "model_forward_time": 1.7240922451019287,
      "step": 61079
    },
    {
      "epoch": 5.8249473571777344e-05,
      "step": 61079,
      "training_step_time": 6.739088296890259
    },
    {
      "epoch": 5.825042724609375e-05,
      "grad_norm": 0.12668722867965698,
      "learning_rate": 3.6004404437698036e-06,
      "loss": 0.0589,
      "step": 61080
    },
    {
      "epoch": 5.825042724609375e-05,
      "model_forward_time": 2.8493521213531494,
      "step": 61080
    },
    {
      "dynamics_loss_avg": 0.06681559886783361,
      "epoch": 5.825042724609375e-05,
      "step": 61080
    },
    {
      "action_loss_avg": 0.0018842593999579549,
      "epoch": 5.825042724609375e-05,
      "step": 61080
    },
    {
      "epoch": 5.825042724609375e-05,
      "step": 61080,
      "training_step_time": 7.7511866092681885
    },
    {
      "epoch": 5.825138092041016e-05,
      "model_forward_time": 1.727370262145996,
      "step": 61081
    },
    {
      "epoch": 5.825138092041016e-05,
      "step": 61081,
      "training_step_time": 8.882004499435425
    },
    {
      "epoch": 5.825233459472656e-05,
      "model_forward_time": 1.729710340499878,
      "step": 61082
    },
    {
      "epoch": 5.825233459472656e-05,
      "step": 61082,
      "training_step_time": 7.697936058044434
    },
    {
      "epoch": 5.825328826904297e-05,
      "model_forward_time": 1.728503942489624,
      "step": 61083
    },
    {
      "epoch": 5.825328826904297e-05,
      "step": 61083,
      "training_step_time": 6.7733190059661865
    },
    {
      "epoch": 5.825424194335938e-05,
      "model_forward_time": 1.7354719638824463,
      "step": 61084
    },
    {
      "epoch": 5.825424194335938e-05,
      "step": 61084,
      "training_step_time": 6.740537643432617
    },
    {
      "epoch": 5.825519561767578e-05,
      "model_forward_time": 1.72597336769104,
      "step": 61085
    },
    {
      "epoch": 5.825519561767578e-05,
      "step": 61085,
      "training_step_time": 6.7061285972595215
    },
    {
      "epoch": 5.8256149291992186e-05,
      "model_forward_time": 1.725104808807373,
      "step": 61086
    },
    {
      "epoch": 5.8256149291992186e-05,
      "step": 61086,
      "training_step_time": 6.744472503662109
    },
    {
      "epoch": 5.82571029663086e-05,
      "model_forward_time": 1.7271411418914795,
      "step": 61087
    },
    {
      "epoch": 5.82571029663086e-05,
      "step": 61087,
      "training_step_time": 6.735954761505127
    },
    {
      "epoch": 5.8258056640625e-05,
      "model_forward_time": 1.7234795093536377,
      "step": 61088
    },
    {
      "epoch": 5.8258056640625e-05,
      "step": 61088,
      "training_step_time": 6.682587146759033
    },
    {
      "epoch": 5.8259010314941405e-05,
      "model_forward_time": 2.865755796432495,
      "step": 61089
    },
    {
      "epoch": 5.8259010314941405e-05,
      "step": 61089,
      "training_step_time": 7.6976072788238525
    },
    {
      "epoch": 5.8259963989257816e-05,
      "grad_norm": 0.10800392180681229,
      "learning_rate": 3.5988531468260185e-06,
      "loss": 0.0612,
      "step": 61090
    },
    {
      "epoch": 5.8259963989257816e-05,
      "model_forward_time": 1.7234342098236084,
      "step": 61090
    },
    {
      "dynamics_loss_avg": 0.06185324527323246,
      "epoch": 5.8259963989257816e-05,
      "step": 61090
    },
    {
      "action_loss_avg": 0.0019905600522179155,
      "epoch": 5.8259963989257816e-05,
      "step": 61090
    },
    {
      "epoch": 5.8259963989257816e-05,
      "step": 61090,
      "training_step_time": 8.924827814102173
    },
    {
      "epoch": 5.826091766357422e-05,
      "model_forward_time": 1.7253103256225586,
      "step": 61091
    },
    {
      "epoch": 5.826091766357422e-05,
      "step": 61091,
      "training_step_time": 7.721132278442383
    },
    {
      "epoch": 5.8261871337890624e-05,
      "model_forward_time": 1.7405307292938232,
      "step": 61092
    },
    {
      "epoch": 5.8261871337890624e-05,
      "step": 61092,
      "training_step_time": 6.612557411193848
    },
    {
      "epoch": 5.8262825012207035e-05,
      "model_forward_time": 1.7251935005187988,
      "step": 61093
    },
    {
      "epoch": 5.8262825012207035e-05,
      "step": 61093,
      "training_step_time": 6.695934295654297
    },
    {
      "epoch": 5.826377868652344e-05,
      "model_forward_time": 1.728219747543335,
      "step": 61094
    },
    {
      "epoch": 5.826377868652344e-05,
      "step": 61094,
      "training_step_time": 6.734276294708252
    },
    {
      "epoch": 5.826473236083984e-05,
      "model_forward_time": 1.3599762916564941,
      "step": 61095
    },
    {
      "epoch": 5.826473236083984e-05,
      "step": 61095,
      "training_step_time": 6.7450666427612305
    },
    {
      "epoch": 5.826568603515625e-05,
      "model_forward_time": 1.7322349548339844,
      "step": 61096
    },
    {
      "epoch": 5.826568603515625e-05,
      "step": 61096,
      "training_step_time": 6.734875679016113
    },
    {
      "epoch": 5.826663970947266e-05,
      "model_forward_time": 1.7336020469665527,
      "step": 61097
    },
    {
      "epoch": 5.826663970947266e-05,
      "step": 61097,
      "training_step_time": 6.746345281600952
    },
    {
      "epoch": 5.826759338378906e-05,
      "model_forward_time": 2.893702268600464,
      "step": 61098
    },
    {
      "epoch": 5.826759338378906e-05,
      "step": 61098,
      "training_step_time": 7.630005359649658
    },
    {
      "epoch": 5.8268547058105466e-05,
      "model_forward_time": 1.7307686805725098,
      "step": 61099
    },
    {
      "epoch": 5.8268547058105466e-05,
      "step": 61099,
      "training_step_time": 10.189178466796875
    },
    {
      "epoch": 5.8269500732421876e-05,
      "grad_norm": 0.10425201058387756,
      "learning_rate": 3.5972660031095462e-06,
      "loss": 0.0576,
      "step": 61100
    },
    {
      "epoch": 5.8269500732421876e-05,
      "model_forward_time": 1.7210781574249268,
      "step": 61100
    },
    {
      "dynamics_loss_avg": 0.05546062663197517,
      "epoch": 5.8269500732421876e-05,
      "step": 61100
    },
    {
      "action_loss_avg": 0.0008364104840438813,
      "epoch": 5.8269500732421876e-05,
      "step": 61100
    },
    {
      "epoch": 5.8269500732421876e-05,
      "step": 61100,
      "training_step_time": 7.74950385093689
    },
    {
      "epoch": 5.827045440673828e-05,
      "model_forward_time": 1.7260899543762207,
      "step": 61101
    },
    {
      "epoch": 5.827045440673828e-05,
      "step": 61101,
      "training_step_time": 7.620765447616577
    },
    {
      "epoch": 5.8271408081054684e-05,
      "model_forward_time": 1.7329540252685547,
      "step": 61102
    },
    {
      "epoch": 5.8271408081054684e-05,
      "step": 61102,
      "training_step_time": 6.7484869956970215
    },
    {
      "epoch": 5.8272361755371095e-05,
      "model_forward_time": 1.7330775260925293,
      "step": 61103
    },
    {
      "epoch": 5.8272361755371095e-05,
      "step": 61103,
      "training_step_time": 6.6721062660217285
    },
    {
      "epoch": 5.82733154296875e-05,
      "model_forward_time": 1.7364695072174072,
      "step": 61104
    },
    {
      "epoch": 5.82733154296875e-05,
      "step": 61104,
      "training_step_time": 6.749739408493042
    },
    {
      "epoch": 5.82742691040039e-05,
      "model_forward_time": 1.736417531967163,
      "step": 61105
    },
    {
      "epoch": 5.82742691040039e-05,
      "step": 61105,
      "training_step_time": 6.754253149032593
    },
    {
      "epoch": 5.8275222778320314e-05,
      "model_forward_time": 1.7304883003234863,
      "step": 61106
    },
    {
      "epoch": 5.8275222778320314e-05,
      "step": 61106,
      "training_step_time": 6.731271266937256
    },
    {
      "epoch": 5.827617645263672e-05,
      "model_forward_time": 1.7852091789245605,
      "step": 61107
    },
    {
      "epoch": 5.827617645263672e-05,
      "step": 61107,
      "training_step_time": 7.665870189666748
    },
    {
      "epoch": 5.827713012695312e-05,
      "model_forward_time": 2.8746793270111084,
      "step": 61108
    },
    {
      "epoch": 5.827713012695312e-05,
      "step": 61108,
      "training_step_time": 8.746046781539917
    },
    {
      "epoch": 5.827808380126953e-05,
      "model_forward_time": 1.723416805267334,
      "step": 61109
    },
    {
      "epoch": 5.827808380126953e-05,
      "step": 61109,
      "training_step_time": 7.656561613082886
    },
    {
      "epoch": 5.827903747558594e-05,
      "grad_norm": 0.12249018251895905,
      "learning_rate": 3.5956790127939556e-06,
      "loss": 0.0578,
      "step": 61110
    },
    {
      "epoch": 5.827903747558594e-05,
      "model_forward_time": 1.7118732929229736,
      "step": 61110
    },
    {
      "dynamics_loss_avg": 0.0633932262659073,
      "epoch": 5.827903747558594e-05,
      "step": 61110
    },
    {
      "action_loss_avg": 0.0017903449654113502,
      "epoch": 5.827903747558594e-05,
      "step": 61110
    },
    {
      "epoch": 5.827903747558594e-05,
      "step": 61110,
      "training_step_time": 7.632766485214233
    },
    {
      "epoch": 5.827999114990234e-05,
      "model_forward_time": 1.7242701053619385,
      "step": 61111
    },
    {
      "epoch": 5.827999114990234e-05,
      "step": 61111,
      "training_step_time": 6.711002349853516
    },
    {
      "epoch": 5.828094482421875e-05,
      "model_forward_time": 1.7156860828399658,
      "step": 61112
    },
    {
      "epoch": 5.828094482421875e-05,
      "step": 61112,
      "training_step_time": 6.730313062667847
    },
    {
      "epoch": 5.8281898498535156e-05,
      "model_forward_time": 1.7340624332427979,
      "step": 61113
    },
    {
      "epoch": 5.8281898498535156e-05,
      "step": 61113,
      "training_step_time": 6.677494287490845
    },
    {
      "epoch": 5.828285217285156e-05,
      "model_forward_time": 1.722247838973999,
      "step": 61114
    },
    {
      "epoch": 5.828285217285156e-05,
      "step": 61114,
      "training_step_time": 6.7054314613342285
    },
    {
      "epoch": 5.828380584716797e-05,
      "model_forward_time": 1.7246861457824707,
      "step": 61115
    },
    {
      "epoch": 5.828380584716797e-05,
      "step": 61115,
      "training_step_time": 6.717579364776611
    },
    {
      "epoch": 5.8284759521484375e-05,
      "model_forward_time": 1.7646398544311523,
      "step": 61116
    },
    {
      "epoch": 5.8284759521484375e-05,
      "step": 61116,
      "training_step_time": 7.497308969497681
    },
    {
      "epoch": 5.828571319580078e-05,
      "model_forward_time": 3.0120887756347656,
      "step": 61117
    },
    {
      "epoch": 5.828571319580078e-05,
      "step": 61117,
      "training_step_time": 9.743317127227783
    },
    {
      "epoch": 5.828666687011719e-05,
      "model_forward_time": 1.7278101444244385,
      "step": 61118
    },
    {
      "epoch": 5.828666687011719e-05,
      "step": 61118,
      "training_step_time": 9.526480436325073
    },
    {
      "epoch": 5.8287620544433594e-05,
      "model_forward_time": 1.7307565212249756,
      "step": 61119
    },
    {
      "epoch": 5.8287620544433594e-05,
      "step": 61119,
      "training_step_time": 7.55593729019165
    },
    {
      "epoch": 5.828857421875e-05,
      "grad_norm": 0.11146660894155502,
      "learning_rate": 3.594092176052796e-06,
      "loss": 0.0572,
      "step": 61120
    },
    {
      "epoch": 5.828857421875e-05,
      "model_forward_time": 1.7522008419036865,
      "step": 61120
    },
    {
      "dynamics_loss_avg": 0.04647954944521189,
      "epoch": 5.828857421875e-05,
      "step": 61120
    },
    {
      "action_loss_avg": 0.001768023567274213,
      "epoch": 5.828857421875e-05,
      "step": 61120
    },
    {
      "epoch": 5.828857421875e-05,
      "step": 61120,
      "training_step_time": 6.702709674835205
    },
    {
      "epoch": 5.828952789306641e-05,
      "model_forward_time": 1.7229900360107422,
      "step": 61121
    },
    {
      "epoch": 5.828952789306641e-05,
      "step": 61121,
      "training_step_time": 6.598290681838989
    },
    {
      "epoch": 5.829048156738281e-05,
      "model_forward_time": 1.7123608589172363,
      "step": 61122
    },
    {
      "epoch": 5.829048156738281e-05,
      "step": 61122,
      "training_step_time": 6.609048366546631
    },
    {
      "epoch": 5.8291435241699216e-05,
      "model_forward_time": 1.7568466663360596,
      "step": 61123
    },
    {
      "epoch": 5.8291435241699216e-05,
      "step": 61123,
      "training_step_time": 6.596473932266235
    },
    {
      "epoch": 5.829238891601563e-05,
      "model_forward_time": 1.707643985748291,
      "step": 61124
    },
    {
      "epoch": 5.829238891601563e-05,
      "step": 61124,
      "training_step_time": 6.656182527542114
    },
    {
      "epoch": 5.829334259033203e-05,
      "model_forward_time": 1.7018749713897705,
      "step": 61125
    },
    {
      "epoch": 5.829334259033203e-05,
      "step": 61125,
      "training_step_time": 7.211069822311401
    },
    {
      "epoch": 5.8294296264648435e-05,
      "model_forward_time": 2.162078380584717,
      "step": 61126
    },
    {
      "epoch": 5.8294296264648435e-05,
      "step": 61126,
      "training_step_time": 8.222286462783813
    },
    {
      "epoch": 5.8295249938964846e-05,
      "model_forward_time": 1.7240314483642578,
      "step": 61127
    },
    {
      "epoch": 5.8295249938964846e-05,
      "step": 61127,
      "training_step_time": 8.635459661483765
    },
    {
      "epoch": 5.829620361328125e-05,
      "model_forward_time": 1.7289042472839355,
      "step": 61128
    },
    {
      "epoch": 5.829620361328125e-05,
      "step": 61128,
      "training_step_time": 7.476719617843628
    },
    {
      "epoch": 5.8297157287597654e-05,
      "model_forward_time": 1.7247600555419922,
      "step": 61129
    },
    {
      "epoch": 5.8297157287597654e-05,
      "step": 61129,
      "training_step_time": 6.6364099979400635
    },
    {
      "epoch": 5.8298110961914065e-05,
      "grad_norm": 0.09758839756250381,
      "learning_rate": 3.5925054930596026e-06,
      "loss": 0.0527,
      "step": 61130
    },
    {
      "epoch": 5.8298110961914065e-05,
      "model_forward_time": 1.7404332160949707,
      "step": 61130
    },
    {
      "dynamics_loss_avg": 0.044879255816340446,
      "epoch": 5.8298110961914065e-05,
      "step": 61130
    },
    {
      "action_loss_avg": 0.0011817576189059765,
      "epoch": 5.8298110961914065e-05,
      "step": 61130
    },
    {
      "epoch": 5.8298110961914065e-05,
      "step": 61130,
      "training_step_time": 6.6712446212768555
    },
    {
      "epoch": 5.829906463623047e-05,
      "model_forward_time": 1.7207484245300293,
      "step": 61131
    },
    {
      "epoch": 5.829906463623047e-05,
      "step": 61131,
      "training_step_time": 6.645612478256226
    },
    {
      "epoch": 5.830001831054687e-05,
      "model_forward_time": 1.7381398677825928,
      "step": 61132
    },
    {
      "epoch": 5.830001831054687e-05,
      "step": 61132,
      "training_step_time": 6.633098125457764
    },
    {
      "epoch": 5.8300971984863284e-05,
      "model_forward_time": 1.7254045009613037,
      "step": 61133
    },
    {
      "epoch": 5.8300971984863284e-05,
      "step": 61133,
      "training_step_time": 6.681088209152222
    },
    {
      "epoch": 5.830192565917969e-05,
      "model_forward_time": 1.720982551574707,
      "step": 61134
    },
    {
      "epoch": 5.830192565917969e-05,
      "step": 61134,
      "training_step_time": 7.126610279083252
    },
    {
      "epoch": 5.830287933349609e-05,
      "model_forward_time": 1.7795133590698242,
      "step": 61135
    },
    {
      "epoch": 5.830287933349609e-05,
      "step": 61135,
      "training_step_time": 9.718322277069092
    },
    {
      "epoch": 5.83038330078125e-05,
      "model_forward_time": 1.727036714553833,
      "step": 61136
    },
    {
      "epoch": 5.83038330078125e-05,
      "step": 61136,
      "training_step_time": 9.540916681289673
    },
    {
      "epoch": 5.830478668212891e-05,
      "model_forward_time": 1.7208433151245117,
      "step": 61137
    },
    {
      "epoch": 5.830478668212891e-05,
      "step": 61137,
      "training_step_time": 7.646603345870972
    },
    {
      "epoch": 5.830574035644531e-05,
      "model_forward_time": 1.7256078720092773,
      "step": 61138
    },
    {
      "epoch": 5.830574035644531e-05,
      "step": 61138,
      "training_step_time": 7.416158199310303
    },
    {
      "epoch": 5.830669403076172e-05,
      "model_forward_time": 1.727353572845459,
      "step": 61139
    },
    {
      "epoch": 5.830669403076172e-05,
      "step": 61139,
      "training_step_time": 6.659572124481201
    },
    {
      "epoch": 5.8307647705078126e-05,
      "grad_norm": 0.09095276892185211,
      "learning_rate": 3.5909189639878904e-06,
      "loss": 0.0605,
      "step": 61140
    },
    {
      "epoch": 5.8307647705078126e-05,
      "model_forward_time": 1.7195913791656494,
      "step": 61140
    },
    {
      "dynamics_loss_avg": 0.055010884813964364,
      "epoch": 5.8307647705078126e-05,
      "step": 61140
    },
    {
      "action_loss_avg": 0.0027173261740244927,
      "epoch": 5.8307647705078126e-05,
      "step": 61140
    },
    {
      "epoch": 5.8307647705078126e-05,
      "step": 61140,
      "training_step_time": 6.625135660171509
    },
    {
      "epoch": 5.830860137939453e-05,
      "model_forward_time": 1.7441039085388184,
      "step": 61141
    },
    {
      "epoch": 5.830860137939453e-05,
      "step": 61141,
      "training_step_time": 6.635495901107788
    },
    {
      "epoch": 5.830955505371094e-05,
      "model_forward_time": 1.7192485332489014,
      "step": 61142
    },
    {
      "epoch": 5.830955505371094e-05,
      "step": 61142,
      "training_step_time": 6.631542682647705
    },
    {
      "epoch": 5.8310508728027344e-05,
      "model_forward_time": 1.7324693202972412,
      "step": 61143
    },
    {
      "epoch": 5.8310508728027344e-05,
      "step": 61143,
      "training_step_time": 6.623297214508057
    },
    {
      "epoch": 5.831146240234375e-05,
      "model_forward_time": 1.7341995239257812,
      "step": 61144
    },
    {
      "epoch": 5.831146240234375e-05,
      "step": 61144,
      "training_step_time": 8.39194130897522
    },
    {
      "epoch": 5.831241607666016e-05,
      "model_forward_time": 2.85183048248291,
      "step": 61145
    },
    {
      "epoch": 5.831241607666016e-05,
      "step": 61145,
      "training_step_time": 10.31324291229248
    },
    {
      "epoch": 5.831336975097656e-05,
      "model_forward_time": 1.7397289276123047,
      "step": 61146
    },
    {
      "epoch": 5.831336975097656e-05,
      "step": 61146,
      "training_step_time": 7.760664701461792
    },
    {
      "epoch": 5.831432342529297e-05,
      "model_forward_time": 1.7223362922668457,
      "step": 61147
    },
    {
      "epoch": 5.831432342529297e-05,
      "step": 61147,
      "training_step_time": 7.604856967926025
    },
    {
      "epoch": 5.831527709960938e-05,
      "model_forward_time": 1.7310230731964111,
      "step": 61148
    },
    {
      "epoch": 5.831527709960938e-05,
      "step": 61148,
      "training_step_time": 6.612119913101196
    },
    {
      "epoch": 5.831623077392578e-05,
      "model_forward_time": 1.7043137550354004,
      "step": 61149
    },
    {
      "epoch": 5.831623077392578e-05,
      "step": 61149,
      "training_step_time": 6.743241310119629
    },
    {
      "epoch": 5.8317184448242186e-05,
      "grad_norm": 0.10487788170576096,
      "learning_rate": 3.589332589011163e-06,
      "loss": 0.0591,
      "step": 61150
    },
    {
      "epoch": 5.8317184448242186e-05,
      "model_forward_time": 1.7064650058746338,
      "step": 61150
    },
    {
      "dynamics_loss_avg": 0.057103398628532885,
      "epoch": 5.8317184448242186e-05,
      "step": 61150
    },
    {
      "action_loss_avg": 0.0014253615750931202,
      "epoch": 5.8317184448242186e-05,
      "step": 61150
    },
    {
      "epoch": 5.8317184448242186e-05,
      "step": 61150,
      "training_step_time": 6.662259101867676
    },
    {
      "epoch": 5.83181381225586e-05,
      "model_forward_time": 1.7286450862884521,
      "step": 61151
    },
    {
      "epoch": 5.83181381225586e-05,
      "step": 61151,
      "training_step_time": 6.660382509231567
    },
    {
      "epoch": 5.8319091796875e-05,
      "model_forward_time": 1.7267980575561523,
      "step": 61152
    },
    {
      "epoch": 5.8319091796875e-05,
      "step": 61152,
      "training_step_time": 6.643799304962158
    },
    {
      "epoch": 5.8320045471191405e-05,
      "model_forward_time": 1.7175567150115967,
      "step": 61153
    },
    {
      "epoch": 5.8320045471191405e-05,
      "step": 61153,
      "training_step_time": 10.079080820083618
    },
    {
      "epoch": 5.8320999145507816e-05,
      "model_forward_time": 2.8669044971466064,
      "step": 61154
    },
    {
      "epoch": 5.8320999145507816e-05,
      "step": 61154,
      "training_step_time": 7.775136709213257
    },
    {
      "epoch": 5.832195281982422e-05,
      "model_forward_time": 1.727097988128662,
      "step": 61155
    },
    {
      "epoch": 5.832195281982422e-05,
      "step": 61155,
      "training_step_time": 7.758912801742554
    },
    {
      "epoch": 5.8322906494140624e-05,
      "model_forward_time": 1.7183120250701904,
      "step": 61156
    },
    {
      "epoch": 5.8322906494140624e-05,
      "step": 61156,
      "training_step_time": 7.540595531463623
    },
    {
      "epoch": 5.832386016845703e-05,
      "model_forward_time": 1.7220065593719482,
      "step": 61157
    },
    {
      "epoch": 5.832386016845703e-05,
      "step": 61157,
      "training_step_time": 6.700147867202759
    },
    {
      "epoch": 5.832481384277344e-05,
      "model_forward_time": 1.712900161743164,
      "step": 61158
    },
    {
      "epoch": 5.832481384277344e-05,
      "step": 61158,
      "training_step_time": 6.645052194595337
    },
    {
      "epoch": 5.832576751708984e-05,
      "model_forward_time": 1.7175536155700684,
      "step": 61159
    },
    {
      "epoch": 5.832576751708984e-05,
      "step": 61159,
      "training_step_time": 6.669172525405884
    },
    {
      "epoch": 5.832672119140625e-05,
      "grad_norm": 0.08901266753673553,
      "learning_rate": 3.5877463683029033e-06,
      "loss": 0.0585,
      "step": 61160
    },
    {
      "epoch": 5.832672119140625e-05,
      "model_forward_time": 1.7055108547210693,
      "step": 61160
    },
    {
      "dynamics_loss_avg": 0.04290082566440105,
      "epoch": 5.832672119140625e-05,
      "step": 61160
    },
    {
      "action_loss_avg": 0.0014227561157895252,
      "epoch": 5.832672119140625e-05,
      "step": 61160
    },
    {
      "epoch": 5.832672119140625e-05,
      "step": 61160,
      "training_step_time": 6.659514427185059
    },
    {
      "epoch": 5.832767486572266e-05,
      "model_forward_time": 1.7353734970092773,
      "step": 61161
    },
    {
      "epoch": 5.832767486572266e-05,
      "step": 61161,
      "training_step_time": 6.680186748504639
    },
    {
      "epoch": 5.832862854003906e-05,
      "model_forward_time": 1.718489646911621,
      "step": 61162
    },
    {
      "epoch": 5.832862854003906e-05,
      "step": 61162,
      "training_step_time": 7.607341289520264
    },
    {
      "epoch": 5.8329582214355466e-05,
      "model_forward_time": 2.8696818351745605,
      "step": 61163
    },
    {
      "epoch": 5.8329582214355466e-05,
      "step": 61163,
      "training_step_time": 10.51366376876831
    },
    {
      "epoch": 5.8330535888671876e-05,
      "model_forward_time": 1.7159464359283447,
      "step": 61164
    },
    {
      "epoch": 5.8330535888671876e-05,
      "step": 61164,
      "training_step_time": 7.704730749130249
    },
    {
      "epoch": 5.833148956298828e-05,
      "model_forward_time": 1.7295606136322021,
      "step": 61165
    },
    {
      "epoch": 5.833148956298828e-05,
      "step": 61165,
      "training_step_time": 7.547134160995483
    },
    {
      "epoch": 5.8332443237304684e-05,
      "model_forward_time": 1.7305889129638672,
      "step": 61166
    },
    {
      "epoch": 5.8332443237304684e-05,
      "step": 61166,
      "training_step_time": 7.523194074630737
    },
    {
      "epoch": 5.8333396911621095e-05,
      "model_forward_time": 1.737661361694336,
      "step": 61167
    },
    {
      "epoch": 5.8333396911621095e-05,
      "step": 61167,
      "training_step_time": 6.712059020996094
    },
    {
      "epoch": 5.83343505859375e-05,
      "model_forward_time": 1.7179713249206543,
      "step": 61168
    },
    {
      "epoch": 5.83343505859375e-05,
      "step": 61168,
      "training_step_time": 6.753040552139282
    },
    {
      "epoch": 5.83353042602539e-05,
      "model_forward_time": 1.7240512371063232,
      "step": 61169
    },
    {
      "epoch": 5.83353042602539e-05,
      "step": 61169,
      "training_step_time": 6.761819124221802
    },
    {
      "epoch": 5.8336257934570314e-05,
      "grad_norm": 0.10427703708410263,
      "learning_rate": 3.586160302036578e-06,
      "loss": 0.0552,
      "step": 61170
    },
    {
      "epoch": 5.8336257934570314e-05,
      "model_forward_time": 1.719693660736084,
      "step": 61170
    },
    {
      "dynamics_loss_avg": 0.05338399615138769,
      "epoch": 5.8336257934570314e-05,
      "step": 61170
    },
    {
      "action_loss_avg": 0.0021876431885175405,
      "epoch": 5.8336257934570314e-05,
      "step": 61170
    },
    {
      "epoch": 5.8336257934570314e-05,
      "step": 61170,
      "training_step_time": 6.7082273960113525
    },
    {
      "epoch": 5.833721160888672e-05,
      "model_forward_time": 1.7317702770233154,
      "step": 61171
    },
    {
      "epoch": 5.833721160888672e-05,
      "step": 61171,
      "training_step_time": 9.18872332572937
    },
    {
      "epoch": 5.833816528320312e-05,
      "model_forward_time": 2.8951122760772705,
      "step": 61172
    },
    {
      "epoch": 5.833816528320312e-05,
      "step": 61172,
      "training_step_time": 9.525961637496948
    },
    {
      "epoch": 5.833911895751953e-05,
      "model_forward_time": 1.7413372993469238,
      "step": 61173
    },
    {
      "epoch": 5.833911895751953e-05,
      "step": 61173,
      "training_step_time": 8.962863445281982
    },
    {
      "epoch": 5.834007263183594e-05,
      "model_forward_time": 1.7320568561553955,
      "step": 61174
    },
    {
      "epoch": 5.834007263183594e-05,
      "step": 61174,
      "training_step_time": 8.73712158203125
    },
    {
      "epoch": 5.834102630615234e-05,
      "model_forward_time": 1.7404634952545166,
      "step": 61175
    },
    {
      "epoch": 5.834102630615234e-05,
      "step": 61175,
      "training_step_time": 7.593649625778198
    },
    {
      "epoch": 5.834197998046875e-05,
      "model_forward_time": 1.7366809844970703,
      "step": 61176
    },
    {
      "epoch": 5.834197998046875e-05,
      "step": 61176,
      "training_step_time": 6.729599475860596
    },
    {
      "epoch": 5.8342933654785156e-05,
      "model_forward_time": 1.7333545684814453,
      "step": 61177
    },
    {
      "epoch": 5.8342933654785156e-05,
      "step": 61177,
      "training_step_time": 6.76807427406311
    },
    {
      "epoch": 5.834388732910156e-05,
      "model_forward_time": 1.7294189929962158,
      "step": 61178
    },
    {
      "epoch": 5.834388732910156e-05,
      "step": 61178,
      "training_step_time": 6.757595777511597
    },
    {
      "epoch": 5.834484100341797e-05,
      "model_forward_time": 1.7311110496520996,
      "step": 61179
    },
    {
      "epoch": 5.834484100341797e-05,
      "step": 61179,
      "training_step_time": 6.739426612854004
    },
    {
      "epoch": 5.8345794677734375e-05,
      "grad_norm": 0.0818636491894722,
      "learning_rate": 3.5845743903856363e-06,
      "loss": 0.0537,
      "step": 61180
    },
    {
      "epoch": 5.8345794677734375e-05,
      "model_forward_time": 1.7571239471435547,
      "step": 61180
    },
    {
      "dynamics_loss_avg": 0.04340578522533178,
      "epoch": 5.8345794677734375e-05,
      "step": 61180
    },
    {
      "action_loss_avg": 0.002391600367263891,
      "epoch": 5.8345794677734375e-05,
      "step": 61180
    },
    {
      "epoch": 5.8345794677734375e-05,
      "step": 61180,
      "training_step_time": 7.717214345932007
    },
    {
      "epoch": 5.834674835205078e-05,
      "model_forward_time": 2.8858401775360107,
      "step": 61181
    },
    {
      "epoch": 5.834674835205078e-05,
      "step": 61181,
      "training_step_time": 8.915591478347778
    },
    {
      "epoch": 5.834770202636719e-05,
      "model_forward_time": 1.7261977195739746,
      "step": 61182
    },
    {
      "epoch": 5.834770202636719e-05,
      "step": 61182,
      "training_step_time": 11.409201860427856
    },
    {
      "epoch": 5.8348655700683594e-05,
      "model_forward_time": 1.7299301624298096,
      "step": 61183
    },
    {
      "epoch": 5.8348655700683594e-05,
      "step": 61183,
      "training_step_time": 8.592809677124023
    },
    {
      "epoch": 5.8349609375e-05,
      "model_forward_time": 1.7385900020599365,
      "step": 61184
    },
    {
      "epoch": 5.8349609375e-05,
      "step": 61184,
      "training_step_time": 7.682097911834717
    },
    {
      "epoch": 5.835056304931641e-05,
      "model_forward_time": 1.7289097309112549,
      "step": 61185
    },
    {
      "epoch": 5.835056304931641e-05,
      "step": 61185,
      "training_step_time": 6.722829580307007
    },
    {
      "epoch": 5.835151672363281e-05,
      "model_forward_time": 1.718747854232788,
      "step": 61186
    },
    {
      "epoch": 5.835151672363281e-05,
      "step": 61186,
      "training_step_time": 6.732945919036865
    },
    {
      "epoch": 5.8352470397949216e-05,
      "model_forward_time": 1.7282359600067139,
      "step": 61187
    },
    {
      "epoch": 5.8352470397949216e-05,
      "step": 61187,
      "training_step_time": 6.743907928466797
    },
    {
      "epoch": 5.835342407226563e-05,
      "model_forward_time": 1.7310092449188232,
      "step": 61188
    },
    {
      "epoch": 5.835342407226563e-05,
      "step": 61188,
      "training_step_time": 6.742581367492676
    },
    {
      "epoch": 5.835437774658203e-05,
      "model_forward_time": 1.7316620349884033,
      "step": 61189
    },
    {
      "epoch": 5.835437774658203e-05,
      "step": 61189,
      "training_step_time": 8.164017915725708
    },
    {
      "epoch": 5.8355331420898435e-05,
      "grad_norm": 0.11393698304891586,
      "learning_rate": 3.582988633523511e-06,
      "loss": 0.0583,
      "step": 61190
    },
    {
      "epoch": 5.8355331420898435e-05,
      "model_forward_time": 1.6981408596038818,
      "step": 61190
    },
    {
      "dynamics_loss_avg": 0.06477347537875175,
      "epoch": 5.8355331420898435e-05,
      "step": 61190
    },
    {
      "action_loss_avg": 0.0026019910757895557,
      "epoch": 5.8355331420898435e-05,
      "step": 61190
    },
    {
      "epoch": 5.8355331420898435e-05,
      "step": 61190,
      "training_step_time": 7.774030447006226
    },
    {
      "epoch": 5.8356285095214846e-05,
      "model_forward_time": 2.8863139152526855,
      "step": 61191
    },
    {
      "epoch": 5.8356285095214846e-05,
      "step": 61191,
      "training_step_time": 9.807397365570068
    },
    {
      "epoch": 5.835723876953125e-05,
      "model_forward_time": 1.0054633617401123,
      "step": 61192
    },
    {
      "epoch": 5.835723876953125e-05,
      "step": 61192,
      "training_step_time": 8.576387882232666
    },
    {
      "epoch": 5.8358192443847654e-05,
      "model_forward_time": 1.7233843803405762,
      "step": 61193
    },
    {
      "epoch": 5.8358192443847654e-05,
      "step": 61193,
      "training_step_time": 7.67415976524353
    },
    {
      "epoch": 5.8359146118164065e-05,
      "model_forward_time": 1.727543830871582,
      "step": 61194
    },
    {
      "epoch": 5.8359146118164065e-05,
      "step": 61194,
      "training_step_time": 7.544076681137085
    },
    {
      "epoch": 5.836009979248047e-05,
      "model_forward_time": 1.7179856300354004,
      "step": 61195
    },
    {
      "epoch": 5.836009979248047e-05,
      "step": 61195,
      "training_step_time": 6.753704309463501
    },
    {
      "epoch": 5.836105346679687e-05,
      "model_forward_time": 1.7226903438568115,
      "step": 61196
    },
    {
      "epoch": 5.836105346679687e-05,
      "step": 61196,
      "training_step_time": 6.7347211837768555
    },
    {
      "epoch": 5.8362007141113284e-05,
      "model_forward_time": 1.7341928482055664,
      "step": 61197
    },
    {
      "epoch": 5.8362007141113284e-05,
      "step": 61197,
      "training_step_time": 6.72525954246521
    },
    {
      "epoch": 5.836296081542969e-05,
      "model_forward_time": 1.7302498817443848,
      "step": 61198
    },
    {
      "epoch": 5.836296081542969e-05,
      "step": 61198,
      "training_step_time": 8.660077333450317
    },
    {
      "epoch": 5.836391448974609e-05,
      "model_forward_time": 1.724522352218628,
      "step": 61199
    },
    {
      "epoch": 5.836391448974609e-05,
      "step": 61199,
      "training_step_time": 9.209573984146118
    },
    {
      "epoch": 5.83648681640625e-05,
      "grad_norm": 0.0792735293507576,
      "learning_rate": 3.581403031623616e-06,
      "loss": 0.0587,
      "step": 61200
    },
    {
      "epoch": 5.83648681640625e-05,
      "model_forward_time": 2.867502450942993,
      "step": 61200
    },
    {
      "dynamics_loss_avg": 0.05428789621219039,
      "epoch": 5.83648681640625e-05,
      "step": 61200
    },
    {
      "action_loss_avg": 0.0011164382507558913,
      "epoch": 5.83648681640625e-05,
      "step": 61200
    },
    {
      "epoch": 5.83648681640625e-05,
      "step": 61200,
      "training_step_time": 10.019162654876709
    },
    {
      "epoch": 5.836582183837891e-05,
      "model_forward_time": 1.731313943862915,
      "step": 61201
    },
    {
      "epoch": 5.836582183837891e-05,
      "step": 61201,
      "training_step_time": 7.8359596729278564
    },
    {
      "epoch": 5.836677551269531e-05,
      "model_forward_time": 1.7432520389556885,
      "step": 61202
    },
    {
      "epoch": 5.836677551269531e-05,
      "step": 61202,
      "training_step_time": 7.66126823425293
    },
    {
      "epoch": 5.836772918701172e-05,
      "model_forward_time": 1.7300424575805664,
      "step": 61203
    },
    {
      "epoch": 5.836772918701172e-05,
      "step": 61203,
      "training_step_time": 7.61780571937561
    },
    {
      "epoch": 5.8368682861328126e-05,
      "model_forward_time": 1.7297780513763428,
      "step": 61204
    },
    {
      "epoch": 5.8368682861328126e-05,
      "step": 61204,
      "training_step_time": 6.736664295196533
    },
    {
      "epoch": 5.836963653564453e-05,
      "model_forward_time": 1.7331037521362305,
      "step": 61205
    },
    {
      "epoch": 5.836963653564453e-05,
      "step": 61205,
      "training_step_time": 6.740860939025879
    },
    {
      "epoch": 5.837059020996094e-05,
      "model_forward_time": 1.7270417213439941,
      "step": 61206
    },
    {
      "epoch": 5.837059020996094e-05,
      "step": 61206,
      "training_step_time": 6.7720561027526855
    },
    {
      "epoch": 5.8371543884277344e-05,
      "model_forward_time": 1.724569320678711,
      "step": 61207
    },
    {
      "epoch": 5.8371543884277344e-05,
      "step": 61207,
      "training_step_time": 7.701809406280518
    },
    {
      "epoch": 5.837249755859375e-05,
      "model_forward_time": 1.727959394454956,
      "step": 61208
    },
    {
      "epoch": 5.837249755859375e-05,
      "step": 61208,
      "training_step_time": 7.781919479370117
    },
    {
      "epoch": 5.837345123291016e-05,
      "model_forward_time": 2.8807895183563232,
      "step": 61209
    },
    {
      "epoch": 5.837345123291016e-05,
      "step": 61209,
      "training_step_time": 9.792749881744385
    },
    {
      "epoch": 5.837440490722656e-05,
      "grad_norm": 0.12072773277759552,
      "learning_rate": 3.579817584859354e-06,
      "loss": 0.0535,
      "step": 61210
    },
    {
      "epoch": 5.837440490722656e-05,
      "model_forward_time": 1.7191102504730225,
      "step": 61210
    },
    {
      "dynamics_loss_avg": 0.05078001376241446,
      "epoch": 5.837440490722656e-05,
      "step": 61210
    },
    {
      "action_loss_avg": 0.0016433227661764248,
      "epoch": 5.837440490722656e-05,
      "step": 61210
    },
    {
      "epoch": 5.837440490722656e-05,
      "step": 61210,
      "training_step_time": 7.774848461151123
    },
    {
      "epoch": 5.837535858154297e-05,
      "model_forward_time": 1.7371339797973633,
      "step": 61211
    },
    {
      "epoch": 5.837535858154297e-05,
      "step": 61211,
      "training_step_time": 7.700976848602295
    },
    {
      "epoch": 5.837631225585938e-05,
      "model_forward_time": 1.7295260429382324,
      "step": 61212
    },
    {
      "epoch": 5.837631225585938e-05,
      "step": 61212,
      "training_step_time": 7.7374513149261475
    },
    {
      "epoch": 5.837726593017578e-05,
      "model_forward_time": 1.7327163219451904,
      "step": 61213
    },
    {
      "epoch": 5.837726593017578e-05,
      "step": 61213,
      "training_step_time": 6.7590062618255615
    },
    {
      "epoch": 5.8378219604492186e-05,
      "model_forward_time": 1.762237548828125,
      "step": 61214
    },
    {
      "epoch": 5.8378219604492186e-05,
      "step": 61214,
      "training_step_time": 6.765518665313721
    },
    {
      "epoch": 5.83791732788086e-05,
      "model_forward_time": 1.721736192703247,
      "step": 61215
    },
    {
      "epoch": 5.83791732788086e-05,
      "step": 61215,
      "training_step_time": 6.720271348953247
    },
    {
      "epoch": 5.8380126953125e-05,
      "model_forward_time": 1.7277486324310303,
      "step": 61216
    },
    {
      "epoch": 5.8380126953125e-05,
      "step": 61216,
      "training_step_time": 7.585486650466919
    },
    {
      "epoch": 5.8381080627441405e-05,
      "model_forward_time": 1.7405462265014648,
      "step": 61217
    },
    {
      "epoch": 5.8381080627441405e-05,
      "step": 61217,
      "training_step_time": 7.733470439910889
    },
    {
      "epoch": 5.8382034301757816e-05,
      "model_forward_time": 1.7526602745056152,
      "step": 61218
    },
    {
      "epoch": 5.8382034301757816e-05,
      "step": 61218,
      "training_step_time": 10.035293579101562
    },
    {
      "epoch": 5.838298797607422e-05,
      "model_forward_time": 1.745743751525879,
      "step": 61219
    },
    {
      "epoch": 5.838298797607422e-05,
      "step": 61219,
      "training_step_time": 8.929556369781494
    },
    {
      "epoch": 5.8383941650390624e-05,
      "grad_norm": 0.08402395993471146,
      "learning_rate": 3.5782322934041068e-06,
      "loss": 0.0568,
      "step": 61220
    },
    {
      "epoch": 5.8383941650390624e-05,
      "model_forward_time": 1.7502171993255615,
      "step": 61220
    },
    {
      "dynamics_loss_avg": 0.05371938794851303,
      "epoch": 5.8383941650390624e-05,
      "step": 61220
    },
    {
      "action_loss_avg": 0.0022872298548463734,
      "epoch": 5.8383941650390624e-05,
      "step": 61220
    },
    {
      "epoch": 5.8383941650390624e-05,
      "step": 61220,
      "training_step_time": 8.613873720169067
    },
    {
      "epoch": 5.838489532470703e-05,
      "model_forward_time": 1.741652011871338,
      "step": 61221
    },
    {
      "epoch": 5.838489532470703e-05,
      "step": 61221,
      "training_step_time": 7.278057813644409
    },
    {
      "epoch": 5.838584899902344e-05,
      "model_forward_time": 1.7436537742614746,
      "step": 61222
    },
    {
      "epoch": 5.838584899902344e-05,
      "step": 61222,
      "training_step_time": 7.573405981063843
    },
    {
      "epoch": 5.838680267333984e-05,
      "model_forward_time": 1.748115062713623,
      "step": 61223
    },
    {
      "epoch": 5.838680267333984e-05,
      "step": 61223,
      "training_step_time": 6.7666709423065186
    },
    {
      "epoch": 5.838775634765625e-05,
      "model_forward_time": 1.7447435855865479,
      "step": 61224
    },
    {
      "epoch": 5.838775634765625e-05,
      "step": 61224,
      "training_step_time": 6.762631177902222
    },
    {
      "epoch": 5.838871002197266e-05,
      "model_forward_time": 1.75394868850708,
      "step": 61225
    },
    {
      "epoch": 5.838871002197266e-05,
      "step": 61225,
      "training_step_time": 7.602474689483643
    },
    {
      "epoch": 5.838966369628906e-05,
      "model_forward_time": 1.7542901039123535,
      "step": 61226
    },
    {
      "epoch": 5.838966369628906e-05,
      "step": 61226,
      "training_step_time": 7.684788942337036
    },
    {
      "epoch": 5.8390617370605466e-05,
      "model_forward_time": 1.754997968673706,
      "step": 61227
    },
    {
      "epoch": 5.8390617370605466e-05,
      "step": 61227,
      "training_step_time": 9.67799711227417
    },
    {
      "epoch": 5.8391571044921876e-05,
      "model_forward_time": 2.371239185333252,
      "step": 61228
    },
    {
      "epoch": 5.8391571044921876e-05,
      "step": 61228,
      "training_step_time": 9.860265970230103
    },
    {
      "epoch": 5.839252471923828e-05,
      "model_forward_time": 1.7549536228179932,
      "step": 61229
    },
    {
      "epoch": 5.839252471923828e-05,
      "step": 61229,
      "training_step_time": 7.846769571304321
    },
    {
      "epoch": 5.8393478393554685e-05,
      "grad_norm": 0.11594049632549286,
      "learning_rate": 3.5766471574312385e-06,
      "loss": 0.0608,
      "step": 61230
    },
    {
      "epoch": 5.8393478393554685e-05,
      "model_forward_time": 1.7582659721374512,
      "step": 61230
    },
    {
      "dynamics_loss_avg": 0.06654874309897423,
      "epoch": 5.8393478393554685e-05,
      "step": 61230
    },
    {
      "action_loss_avg": 0.0014598300389479847,
      "epoch": 5.8393478393554685e-05,
      "step": 61230
    },
    {
      "epoch": 5.8393478393554685e-05,
      "step": 61230,
      "training_step_time": 7.681466341018677
    },
    {
      "epoch": 5.8394432067871095e-05,
      "model_forward_time": 1.7566232681274414,
      "step": 61231
    },
    {
      "epoch": 5.8394432067871095e-05,
      "step": 61231,
      "training_step_time": 7.693354606628418
    },
    {
      "epoch": 5.83953857421875e-05,
      "model_forward_time": 1.7588741779327393,
      "step": 61232
    },
    {
      "epoch": 5.83953857421875e-05,
      "step": 61232,
      "training_step_time": 6.820111036300659
    },
    {
      "epoch": 5.83963394165039e-05,
      "model_forward_time": 1.7595796585083008,
      "step": 61233
    },
    {
      "epoch": 5.83963394165039e-05,
      "step": 61233,
      "training_step_time": 6.78725528717041
    },
    {
      "epoch": 5.8397293090820314e-05,
      "model_forward_time": 1.7644410133361816,
      "step": 61234
    },
    {
      "epoch": 5.8397293090820314e-05,
      "step": 61234,
      "training_step_time": 7.633234977722168
    },
    {
      "epoch": 5.839824676513672e-05,
      "model_forward_time": 1.7560980319976807,
      "step": 61235
    },
    {
      "epoch": 5.839824676513672e-05,
      "step": 61235,
      "training_step_time": 7.432939052581787
    },
    {
      "epoch": 5.839920043945312e-05,
      "model_forward_time": 1.7540338039398193,
      "step": 61236
    },
    {
      "epoch": 5.839920043945312e-05,
      "step": 61236,
      "training_step_time": 11.0173499584198
    },
    {
      "epoch": 5.840015411376953e-05,
      "model_forward_time": 2.3271923065185547,
      "step": 61237
    },
    {
      "epoch": 5.840015411376953e-05,
      "step": 61237,
      "training_step_time": 8.682812929153442
    },
    {
      "epoch": 5.840110778808594e-05,
      "model_forward_time": 1.7477409839630127,
      "step": 61238
    },
    {
      "epoch": 5.840110778808594e-05,
      "step": 61238,
      "training_step_time": 8.526556015014648
    },
    {
      "epoch": 5.840206146240234e-05,
      "model_forward_time": 1.7569079399108887,
      "step": 61239
    },
    {
      "epoch": 5.840206146240234e-05,
      "step": 61239,
      "training_step_time": 7.677699089050293
    },
    {
      "epoch": 5.840301513671875e-05,
      "grad_norm": 0.10285737365484238,
      "learning_rate": 3.5750621771140964e-06,
      "loss": 0.0577,
      "step": 61240
    },
    {
      "epoch": 5.840301513671875e-05,
      "model_forward_time": 1.7401399612426758,
      "step": 61240
    },
    {
      "dynamics_loss_avg": 0.049286668002605435,
      "epoch": 5.840301513671875e-05,
      "step": 61240
    },
    {
      "action_loss_avg": 0.001875048442161642,
      "epoch": 5.840301513671875e-05,
      "step": 61240
    },
    {
      "epoch": 5.840301513671875e-05,
      "step": 61240,
      "training_step_time": 7.61941385269165
    },
    {
      "epoch": 5.8403968811035156e-05,
      "model_forward_time": 1.7460193634033203,
      "step": 61241
    },
    {
      "epoch": 5.8403968811035156e-05,
      "step": 61241,
      "training_step_time": 6.748747110366821
    },
    {
      "epoch": 5.840492248535156e-05,
      "model_forward_time": 1.7483775615692139,
      "step": 61242
    },
    {
      "epoch": 5.840492248535156e-05,
      "step": 61242,
      "training_step_time": 6.803608655929565
    },
    {
      "epoch": 5.840587615966797e-05,
      "model_forward_time": 1.751047134399414,
      "step": 61243
    },
    {
      "epoch": 5.840587615966797e-05,
      "step": 61243,
      "training_step_time": 7.261870861053467
    },
    {
      "epoch": 5.8406829833984375e-05,
      "model_forward_time": 1.7531013488769531,
      "step": 61244
    },
    {
      "epoch": 5.8406829833984375e-05,
      "step": 61244,
      "training_step_time": 7.684809923171997
    },
    {
      "epoch": 5.840778350830078e-05,
      "model_forward_time": 1.755065679550171,
      "step": 61245
    },
    {
      "epoch": 5.840778350830078e-05,
      "step": 61245,
      "training_step_time": 8.615517616271973
    },
    {
      "epoch": 5.840873718261719e-05,
      "model_forward_time": 2.353975772857666,
      "step": 61246
    },
    {
      "epoch": 5.840873718261719e-05,
      "step": 61246,
      "training_step_time": 8.533241987228394
    },
    {
      "epoch": 5.8409690856933594e-05,
      "model_forward_time": 1.7560155391693115,
      "step": 61247
    },
    {
      "epoch": 5.8409690856933594e-05,
      "step": 61247,
      "training_step_time": 9.716554403305054
    },
    {
      "epoch": 5.841064453125e-05,
      "model_forward_time": 1.7542593479156494,
      "step": 61248
    },
    {
      "epoch": 5.841064453125e-05,
      "step": 61248,
      "training_step_time": 7.707765102386475
    },
    {
      "epoch": 5.841159820556641e-05,
      "model_forward_time": 1.7361583709716797,
      "step": 61249
    },
    {
      "epoch": 5.841159820556641e-05,
      "step": 61249,
      "training_step_time": 7.128369331359863
    },
    {
      "epoch": 5.841255187988281e-05,
      "grad_norm": 0.10568077862262726,
      "learning_rate": 3.5734773526260113e-06,
      "loss": 0.0638,
      "step": 61250
    },
    {
      "epoch": 5.841255187988281e-05,
      "model_forward_time": 1.7445824146270752,
      "step": 61250
    },
    {
      "dynamics_loss_avg": 0.064143006503582,
      "epoch": 5.841255187988281e-05,
      "step": 61250
    },
    {
      "action_loss_avg": 0.0016214476840104909,
      "epoch": 5.841255187988281e-05,
      "step": 61250
    },
    {
      "epoch": 5.841255187988281e-05,
      "step": 61250,
      "training_step_time": 6.792023658752441
    },
    {
      "epoch": 5.8413505554199217e-05,
      "model_forward_time": 1.7533679008483887,
      "step": 61251
    },
    {
      "epoch": 5.8413505554199217e-05,
      "step": 61251,
      "training_step_time": 6.748690128326416
    },
    {
      "epoch": 5.841445922851563e-05,
      "model_forward_time": 1.7541017532348633,
      "step": 61252
    },
    {
      "epoch": 5.841445922851563e-05,
      "step": 61252,
      "training_step_time": 7.221822738647461
    },
    {
      "epoch": 5.841541290283203e-05,
      "model_forward_time": 1.7267467975616455,
      "step": 61253
    },
    {
      "epoch": 5.841541290283203e-05,
      "step": 61253,
      "training_step_time": 7.6270363330841064
    },
    {
      "epoch": 5.8416366577148435e-05,
      "model_forward_time": 1.7375128269195557,
      "step": 61254
    },
    {
      "epoch": 5.8416366577148435e-05,
      "step": 61254,
      "training_step_time": 8.74156141281128
    },
    {
      "epoch": 5.8417320251464846e-05,
      "model_forward_time": 2.358630418777466,
      "step": 61255
    },
    {
      "epoch": 5.8417320251464846e-05,
      "step": 61255,
      "training_step_time": 9.123855829238892
    },
    {
      "epoch": 5.841827392578125e-05,
      "model_forward_time": 1.7483580112457275,
      "step": 61256
    },
    {
      "epoch": 5.841827392578125e-05,
      "step": 61256,
      "training_step_time": 9.321085214614868
    },
    {
      "epoch": 5.8419227600097654e-05,
      "model_forward_time": 1.757091999053955,
      "step": 61257
    },
    {
      "epoch": 5.8419227600097654e-05,
      "step": 61257,
      "training_step_time": 7.81952166557312
    },
    {
      "epoch": 5.8420181274414065e-05,
      "model_forward_time": 1.7557549476623535,
      "step": 61258
    },
    {
      "epoch": 5.8420181274414065e-05,
      "step": 61258,
      "training_step_time": 7.24038553237915
    },
    {
      "epoch": 5.842113494873047e-05,
      "model_forward_time": 1.7530467510223389,
      "step": 61259
    },
    {
      "epoch": 5.842113494873047e-05,
      "step": 61259,
      "training_step_time": 7.6060850620269775
    },
    {
      "epoch": 5.842208862304687e-05,
      "grad_norm": 0.08157980442047119,
      "learning_rate": 3.5718926841402993e-06,
      "loss": 0.0595,
      "step": 61260
    },
    {
      "epoch": 5.842208862304687e-05,
      "model_forward_time": 1.7263391017913818,
      "step": 61260
    },
    {
      "dynamics_loss_avg": 0.037824433855712415,
      "epoch": 5.842208862304687e-05,
      "step": 61260
    },
    {
      "action_loss_avg": 0.001668775049620308,
      "epoch": 5.842208862304687e-05,
      "step": 61260
    },
    {
      "epoch": 5.842208862304687e-05,
      "step": 61260,
      "training_step_time": 6.785285949707031
    },
    {
      "epoch": 5.8423042297363284e-05,
      "model_forward_time": 1.7439954280853271,
      "step": 61261
    },
    {
      "epoch": 5.8423042297363284e-05,
      "step": 61261,
      "training_step_time": 7.248393535614014
    },
    {
      "epoch": 5.842399597167969e-05,
      "model_forward_time": 1.726442575454712,
      "step": 61262
    },
    {
      "epoch": 5.842399597167969e-05,
      "step": 61262,
      "training_step_time": 7.605034589767456
    },
    {
      "epoch": 5.842494964599609e-05,
      "model_forward_time": 1.7418310642242432,
      "step": 61263
    },
    {
      "epoch": 5.842494964599609e-05,
      "step": 61263,
      "training_step_time": 7.644663572311401
    },
    {
      "epoch": 5.84259033203125e-05,
      "model_forward_time": 2.352400064468384,
      "step": 61264
    },
    {
      "epoch": 5.84259033203125e-05,
      "step": 61264,
      "training_step_time": 7.7294721603393555
    },
    {
      "epoch": 5.842685699462891e-05,
      "model_forward_time": 1.7477660179138184,
      "step": 61265
    },
    {
      "epoch": 5.842685699462891e-05,
      "step": 61265,
      "training_step_time": 7.756529092788696
    },
    {
      "epoch": 5.842781066894531e-05,
      "model_forward_time": 1.7267165184020996,
      "step": 61266
    },
    {
      "epoch": 5.842781066894531e-05,
      "step": 61266,
      "training_step_time": 9.629053592681885
    },
    {
      "epoch": 5.842876434326172e-05,
      "model_forward_time": 1.7412574291229248,
      "step": 61267
    },
    {
      "epoch": 5.842876434326172e-05,
      "step": 61267,
      "training_step_time": 7.6321635246276855
    },
    {
      "epoch": 5.8429718017578126e-05,
      "model_forward_time": 1.7373101711273193,
      "step": 61268
    },
    {
      "epoch": 5.8429718017578126e-05,
      "step": 61268,
      "training_step_time": 7.186285018920898
    },
    {
      "epoch": 5.843067169189453e-05,
      "model_forward_time": 1.7364771366119385,
      "step": 61269
    },
    {
      "epoch": 5.843067169189453e-05,
      "step": 61269,
      "training_step_time": 6.783545017242432
    },
    {
      "epoch": 5.843162536621094e-05,
      "grad_norm": 0.12519918382167816,
      "learning_rate": 3.570308171830256e-06,
      "loss": 0.0583,
      "step": 61270
    },
    {
      "epoch": 5.843162536621094e-05,
      "model_forward_time": 1.7269237041473389,
      "step": 61270
    },
    {
      "dynamics_loss_avg": 0.0515482222661376,
      "epoch": 5.843162536621094e-05,
      "step": 61270
    },
    {
      "action_loss_avg": 0.0015337872318923473,
      "epoch": 5.843162536621094e-05,
      "step": 61270
    },
    {
      "epoch": 5.843162536621094e-05,
      "step": 61270,
      "training_step_time": 7.256896018981934
    },
    {
      "epoch": 5.8432579040527345e-05,
      "model_forward_time": 1.7352139949798584,
      "step": 61271
    },
    {
      "epoch": 5.8432579040527345e-05,
      "step": 61271,
      "training_step_time": 7.5792765617370605
    },
    {
      "epoch": 5.843353271484375e-05,
      "model_forward_time": 1.7395155429840088,
      "step": 61272
    },
    {
      "epoch": 5.843353271484375e-05,
      "step": 61272,
      "training_step_time": 8.72436261177063
    },
    {
      "epoch": 5.843448638916016e-05,
      "model_forward_time": 1.7401700019836426,
      "step": 61273
    },
    {
      "epoch": 5.843448638916016e-05,
      "step": 61273,
      "training_step_time": 9.082424640655518
    },
    {
      "epoch": 5.843544006347656e-05,
      "model_forward_time": 1.7156586647033691,
      "step": 61274
    },
    {
      "epoch": 5.843544006347656e-05,
      "step": 61274,
      "training_step_time": 8.894203424453735
    },
    {
      "epoch": 5.843639373779297e-05,
      "model_forward_time": 1.7218973636627197,
      "step": 61275
    },
    {
      "epoch": 5.843639373779297e-05,
      "step": 61275,
      "training_step_time": 8.635664939880371
    },
    {
      "epoch": 5.843734741210938e-05,
      "model_forward_time": 1.7385399341583252,
      "step": 61276
    },
    {
      "epoch": 5.843734741210938e-05,
      "step": 61276,
      "training_step_time": 7.698581218719482
    },
    {
      "epoch": 5.843830108642578e-05,
      "model_forward_time": 1.7404272556304932,
      "step": 61277
    },
    {
      "epoch": 5.843830108642578e-05,
      "step": 61277,
      "training_step_time": 7.625666618347168
    },
    {
      "epoch": 5.8439254760742186e-05,
      "model_forward_time": 1.7400634288787842,
      "step": 61278
    },
    {
      "epoch": 5.8439254760742186e-05,
      "step": 61278,
      "training_step_time": 6.777702331542969
    },
    {
      "epoch": 5.84402084350586e-05,
      "model_forward_time": 1.7402076721191406,
      "step": 61279
    },
    {
      "epoch": 5.84402084350586e-05,
      "step": 61279,
      "training_step_time": 7.260611057281494
    },
    {
      "epoch": 5.8441162109375e-05,
      "grad_norm": 0.08600772172212601,
      "learning_rate": 3.5687238158691616e-06,
      "loss": 0.0607,
      "step": 61280
    },
    {
      "epoch": 5.8441162109375e-05,
      "model_forward_time": 1.7338297367095947,
      "step": 61280
    },
    {
      "dynamics_loss_avg": 0.06077010203152895,
      "epoch": 5.8441162109375e-05,
      "step": 61280
    },
    {
      "action_loss_avg": 0.0017323381180176511,
      "epoch": 5.8441162109375e-05,
      "step": 61280
    },
    {
      "epoch": 5.8441162109375e-05,
      "step": 61280,
      "training_step_time": 8.40074872970581
    },
    {
      "epoch": 5.8442115783691405e-05,
      "model_forward_time": 1.7368247509002686,
      "step": 61281
    },
    {
      "epoch": 5.8442115783691405e-05,
      "step": 61281,
      "training_step_time": 7.681875944137573
    },
    {
      "epoch": 5.844306945800781e-05,
      "model_forward_time": 1.7366983890533447,
      "step": 61282
    },
    {
      "epoch": 5.844306945800781e-05,
      "step": 61282,
      "training_step_time": 8.577250480651855
    },
    {
      "epoch": 5.844402313232422e-05,
      "model_forward_time": 2.322666883468628,
      "step": 61283
    },
    {
      "epoch": 5.844402313232422e-05,
      "step": 61283,
      "training_step_time": 7.775233030319214
    },
    {
      "epoch": 5.8444976806640624e-05,
      "model_forward_time": 1.7353107929229736,
      "step": 61284
    },
    {
      "epoch": 5.8444976806640624e-05,
      "step": 61284,
      "training_step_time": 7.7140326499938965
    },
    {
      "epoch": 5.844593048095703e-05,
      "model_forward_time": 1.7395832538604736,
      "step": 61285
    },
    {
      "epoch": 5.844593048095703e-05,
      "step": 61285,
      "training_step_time": 7.647240877151489
    },
    {
      "epoch": 5.844688415527344e-05,
      "model_forward_time": 1.7502822875976562,
      "step": 61286
    },
    {
      "epoch": 5.844688415527344e-05,
      "step": 61286,
      "training_step_time": 7.4933085441589355
    },
    {
      "epoch": 5.844783782958984e-05,
      "model_forward_time": 1.7452099323272705,
      "step": 61287
    },
    {
      "epoch": 5.844783782958984e-05,
      "step": 61287,
      "training_step_time": 6.716529369354248
    },
    {
      "epoch": 5.844879150390625e-05,
      "model_forward_time": 1.7484633922576904,
      "step": 61288
    },
    {
      "epoch": 5.844879150390625e-05,
      "step": 61288,
      "training_step_time": 7.230430841445923
    },
    {
      "epoch": 5.844974517822266e-05,
      "model_forward_time": 1.7266831398010254,
      "step": 61289
    },
    {
      "epoch": 5.844974517822266e-05,
      "step": 61289,
      "training_step_time": 7.0939271450042725
    },
    {
      "epoch": 5.845069885253906e-05,
      "grad_norm": 0.09912745654582977,
      "learning_rate": 3.5671396164302777e-06,
      "loss": 0.0602,
      "step": 61290
    },
    {
      "epoch": 5.845069885253906e-05,
      "model_forward_time": 1.7212145328521729,
      "step": 61290
    },
    {
      "dynamics_loss_avg": 0.05486501418054104,
      "epoch": 5.845069885253906e-05,
      "step": 61290
    },
    {
      "action_loss_avg": 0.0010702174622565508,
      "epoch": 5.845069885253906e-05,
      "step": 61290
    },
    {
      "epoch": 5.845069885253906e-05,
      "step": 61290,
      "training_step_time": 7.700828790664673
    },
    {
      "epoch": 5.8451652526855466e-05,
      "model_forward_time": 1.7415876388549805,
      "step": 61291
    },
    {
      "epoch": 5.8451652526855466e-05,
      "step": 61291,
      "training_step_time": 9.04564905166626
    },
    {
      "epoch": 5.8452606201171877e-05,
      "model_forward_time": 2.3620834350585938,
      "step": 61292
    },
    {
      "epoch": 5.8452606201171877e-05,
      "step": 61292,
      "training_step_time": 8.560336112976074
    },
    {
      "epoch": 5.845355987548828e-05,
      "model_forward_time": 1.735515832901001,
      "step": 61293
    },
    {
      "epoch": 5.845355987548828e-05,
      "step": 61293,
      "training_step_time": 7.714030742645264
    },
    {
      "epoch": 5.8454513549804685e-05,
      "model_forward_time": 1.7435932159423828,
      "step": 61294
    },
    {
      "epoch": 5.8454513549804685e-05,
      "step": 61294,
      "training_step_time": 7.828238248825073
    },
    {
      "epoch": 5.8455467224121095e-05,
      "model_forward_time": 1.7421188354492188,
      "step": 61295
    },
    {
      "epoch": 5.8455467224121095e-05,
      "step": 61295,
      "training_step_time": 7.296305418014526
    },
    {
      "epoch": 5.84564208984375e-05,
      "model_forward_time": 1.7425322532653809,
      "step": 61296
    },
    {
      "epoch": 5.84564208984375e-05,
      "step": 61296,
      "training_step_time": 7.506701231002808
    },
    {
      "epoch": 5.8457374572753903e-05,
      "model_forward_time": 1.3681890964508057,
      "step": 61297
    },
    {
      "epoch": 5.8457374572753903e-05,
      "step": 61297,
      "training_step_time": 7.286264657974243
    },
    {
      "epoch": 5.8458328247070314e-05,
      "model_forward_time": 1.7469561100006104,
      "step": 61298
    },
    {
      "epoch": 5.8458328247070314e-05,
      "step": 61298,
      "training_step_time": 7.842324733734131
    },
    {
      "epoch": 5.845928192138672e-05,
      "model_forward_time": 1.746302843093872,
      "step": 61299
    },
    {
      "epoch": 5.845928192138672e-05,
      "step": 61299,
      "training_step_time": 8.227858781814575
    },
    {
      "epoch": 5.846023559570312e-05,
      "grad_norm": 0.11163157224655151,
      "learning_rate": 3.565555573686853e-06,
      "loss": 0.0587,
      "step": 61300
    },
    {
      "epoch": 5.846023559570312e-05,
      "model_forward_time": 1.7217633724212646,
      "step": 61300
    },
    {
      "dynamics_loss_avg": 0.05855099558830261,
      "epoch": 5.846023559570312e-05,
      "step": 61300
    },
    {
      "action_loss_avg": 0.002268709265626967,
      "epoch": 5.846023559570312e-05,
      "step": 61300
    },
    {
      "epoch": 5.846023559570312e-05,
      "step": 61300,
      "training_step_time": 8.252352714538574
    },
    {
      "epoch": 5.846118927001953e-05,
      "model_forward_time": 2.3363325595855713,
      "step": 61301
    },
    {
      "epoch": 5.846118927001953e-05,
      "step": 61301,
      "training_step_time": 9.38227128982544
    },
    {
      "epoch": 5.846214294433594e-05,
      "model_forward_time": 1.7339537143707275,
      "step": 61302
    },
    {
      "epoch": 5.846214294433594e-05,
      "step": 61302,
      "training_step_time": 8.43975305557251
    },
    {
      "epoch": 5.846309661865234e-05,
      "model_forward_time": 1.738204002380371,
      "step": 61303
    },
    {
      "epoch": 5.846309661865234e-05,
      "step": 61303,
      "training_step_time": 7.765087127685547
    },
    {
      "epoch": 5.846405029296875e-05,
      "model_forward_time": 1.738950490951538,
      "step": 61304
    },
    {
      "epoch": 5.846405029296875e-05,
      "step": 61304,
      "training_step_time": 7.6314496994018555
    },
    {
      "epoch": 5.8465003967285156e-05,
      "model_forward_time": 1.7422356605529785,
      "step": 61305
    },
    {
      "epoch": 5.8465003967285156e-05,
      "step": 61305,
      "training_step_time": 7.635193109512329
    },
    {
      "epoch": 5.846595764160156e-05,
      "model_forward_time": 1.739243507385254,
      "step": 61306
    },
    {
      "epoch": 5.846595764160156e-05,
      "step": 61306,
      "training_step_time": 7.287609100341797
    },
    {
      "epoch": 5.846691131591797e-05,
      "model_forward_time": 1.7349519729614258,
      "step": 61307
    },
    {
      "epoch": 5.846691131591797e-05,
      "step": 61307,
      "training_step_time": 6.816978931427002
    },
    {
      "epoch": 5.8467864990234375e-05,
      "model_forward_time": 1.7496469020843506,
      "step": 61308
    },
    {
      "epoch": 5.8467864990234375e-05,
      "step": 61308,
      "training_step_time": 7.582616806030273
    },
    {
      "epoch": 5.846881866455078e-05,
      "model_forward_time": 1.7517828941345215,
      "step": 61309
    },
    {
      "epoch": 5.846881866455078e-05,
      "step": 61309,
      "training_step_time": 8.742127180099487
    },
    {
      "epoch": 5.846977233886719e-05,
      "grad_norm": 0.11624009907245636,
      "learning_rate": 3.563971687812113e-06,
      "loss": 0.0582,
      "step": 61310
    },
    {
      "epoch": 5.846977233886719e-05,
      "model_forward_time": 2.339167356491089,
      "step": 61310
    },
    {
      "dynamics_loss_avg": 0.0630022183060646,
      "epoch": 5.846977233886719e-05,
      "step": 61310
    },
    {
      "action_loss_avg": 0.0015257837512763217,
      "epoch": 5.846977233886719e-05,
      "step": 61310
    },
    {
      "epoch": 5.846977233886719e-05,
      "step": 61310,
      "training_step_time": 7.744370460510254
    },
    {
      "epoch": 5.8470726013183594e-05,
      "model_forward_time": 1.7476584911346436,
      "step": 61311
    },
    {
      "epoch": 5.8470726013183594e-05,
      "step": 61311,
      "training_step_time": 9.161435842514038
    },
    {
      "epoch": 5.84716796875e-05,
      "model_forward_time": 1.7431139945983887,
      "step": 61312
    },
    {
      "epoch": 5.84716796875e-05,
      "step": 61312,
      "training_step_time": 7.762721538543701
    },
    {
      "epoch": 5.847263336181641e-05,
      "model_forward_time": 1.748689889907837,
      "step": 61313
    },
    {
      "epoch": 5.847263336181641e-05,
      "step": 61313,
      "training_step_time": 7.627230644226074
    },
    {
      "epoch": 5.847358703613281e-05,
      "model_forward_time": 1.753671646118164,
      "step": 61314
    },
    {
      "epoch": 5.847358703613281e-05,
      "step": 61314,
      "training_step_time": 7.575816631317139
    },
    {
      "epoch": 5.8474540710449217e-05,
      "model_forward_time": 1.75449800491333,
      "step": 61315
    },
    {
      "epoch": 5.8474540710449217e-05,
      "step": 61315,
      "training_step_time": 7.277418613433838
    },
    {
      "epoch": 5.847549438476563e-05,
      "model_forward_time": 1.7518587112426758,
      "step": 61316
    },
    {
      "epoch": 5.847549438476563e-05,
      "step": 61316,
      "training_step_time": 6.8616883754730225
    },
    {
      "epoch": 5.847644805908203e-05,
      "model_forward_time": 1.7514877319335938,
      "step": 61317
    },
    {
      "epoch": 5.847644805908203e-05,
      "step": 61317,
      "training_step_time": 7.574286460876465
    },
    {
      "epoch": 5.8477401733398435e-05,
      "model_forward_time": 1.7477099895477295,
      "step": 61318
    },
    {
      "epoch": 5.8477401733398435e-05,
      "step": 61318,
      "training_step_time": 7.673429250717163
    },
    {
      "epoch": 5.8478355407714846e-05,
      "model_forward_time": 1.7630949020385742,
      "step": 61319
    },
    {
      "epoch": 5.8478355407714846e-05,
      "step": 61319,
      "training_step_time": 9.292131185531616
    },
    {
      "epoch": 5.847930908203125e-05,
      "grad_norm": 0.12779320776462555,
      "learning_rate": 3.562387958979271e-06,
      "loss": 0.0563,
      "step": 61320
    },
    {
      "epoch": 5.847930908203125e-05,
      "model_forward_time": 1.7415473461151123,
      "step": 61320
    },
    {
      "dynamics_loss_avg": 0.04362907186150551,
      "epoch": 5.847930908203125e-05,
      "step": 61320
    },
    {
      "action_loss_avg": 0.0010808979684952646,
      "epoch": 5.847930908203125e-05,
      "step": 61320
    },
    {
      "epoch": 5.847930908203125e-05,
      "step": 61320,
      "training_step_time": 11.379763126373291
    },
    {
      "epoch": 5.8480262756347654e-05,
      "model_forward_time": 1.751488447189331,
      "step": 61321
    },
    {
      "epoch": 5.8480262756347654e-05,
      "step": 61321,
      "training_step_time": 8.182987213134766
    },
    {
      "epoch": 5.8481216430664065e-05,
      "model_forward_time": 1.7553081512451172,
      "step": 61322
    },
    {
      "epoch": 5.8481216430664065e-05,
      "step": 61322,
      "training_step_time": 7.854580879211426
    },
    {
      "epoch": 5.848217010498047e-05,
      "model_forward_time": 1.7504973411560059,
      "step": 61323
    },
    {
      "epoch": 5.848217010498047e-05,
      "step": 61323,
      "training_step_time": 7.232873916625977
    },
    {
      "epoch": 5.848312377929687e-05,
      "model_forward_time": 1.7366702556610107,
      "step": 61324
    },
    {
      "epoch": 5.848312377929687e-05,
      "step": 61324,
      "training_step_time": 7.566843748092651
    },
    {
      "epoch": 5.8484077453613284e-05,
      "model_forward_time": 1.7433979511260986,
      "step": 61325
    },
    {
      "epoch": 5.8484077453613284e-05,
      "step": 61325,
      "training_step_time": 6.907190561294556
    },
    {
      "epoch": 5.848503112792969e-05,
      "model_forward_time": 1.752488136291504,
      "step": 61326
    },
    {
      "epoch": 5.848503112792969e-05,
      "step": 61326,
      "training_step_time": 7.290998697280884
    },
    {
      "epoch": 5.848598480224609e-05,
      "model_forward_time": 1.7483372688293457,
      "step": 61327
    },
    {
      "epoch": 5.848598480224609e-05,
      "step": 61327,
      "training_step_time": 8.763086557388306
    },
    {
      "epoch": 5.84869384765625e-05,
      "model_forward_time": 1.7474653720855713,
      "step": 61328
    },
    {
      "epoch": 5.84869384765625e-05,
      "step": 61328,
      "training_step_time": 9.20255994796753
    },
    {
      "epoch": 5.848789215087891e-05,
      "model_forward_time": 2.28421950340271,
      "step": 61329
    },
    {
      "epoch": 5.848789215087891e-05,
      "step": 61329,
      "training_step_time": 8.935689449310303
    },
    {
      "epoch": 5.848884582519531e-05,
      "grad_norm": 0.10524975508451462,
      "learning_rate": 3.5608043873615185e-06,
      "loss": 0.0545,
      "step": 61330
    },
    {
      "epoch": 5.848884582519531e-05,
      "model_forward_time": 1.733266830444336,
      "step": 61330
    },
    {
      "dynamics_loss_avg": 0.04456894434988499,
      "epoch": 5.848884582519531e-05,
      "step": 61330
    },
    {
      "action_loss_avg": 0.0017529852397274227,
      "epoch": 5.848884582519531e-05,
      "step": 61330
    },
    {
      "epoch": 5.848884582519531e-05,
      "step": 61330,
      "training_step_time": 7.775594234466553
    },
    {
      "epoch": 5.848979949951172e-05,
      "model_forward_time": 1.7480225563049316,
      "step": 61331
    },
    {
      "epoch": 5.848979949951172e-05,
      "step": 61331,
      "training_step_time": 7.889861822128296
    },
    {
      "epoch": 5.8490753173828126e-05,
      "model_forward_time": 1.748931646347046,
      "step": 61332
    },
    {
      "epoch": 5.8490753173828126e-05,
      "step": 61332,
      "training_step_time": 7.344122648239136
    },
    {
      "epoch": 5.849170684814453e-05,
      "model_forward_time": 1.7480978965759277,
      "step": 61333
    },
    {
      "epoch": 5.849170684814453e-05,
      "step": 61333,
      "training_step_time": 7.6937031745910645
    },
    {
      "epoch": 5.849266052246094e-05,
      "model_forward_time": 1.7415616512298584,
      "step": 61334
    },
    {
      "epoch": 5.849266052246094e-05,
      "step": 61334,
      "training_step_time": 6.882143974304199
    },
    {
      "epoch": 5.8493614196777345e-05,
      "model_forward_time": 1.3695461750030518,
      "step": 61335
    },
    {
      "epoch": 5.8493614196777345e-05,
      "step": 61335,
      "training_step_time": 7.184087038040161
    },
    {
      "epoch": 5.849456787109375e-05,
      "model_forward_time": 1.749295711517334,
      "step": 61336
    },
    {
      "epoch": 5.849456787109375e-05,
      "step": 61336,
      "training_step_time": 7.6765687465667725
    },
    {
      "epoch": 5.849552154541016e-05,
      "model_forward_time": 1.755406141281128,
      "step": 61337
    },
    {
      "epoch": 5.849552154541016e-05,
      "step": 61337,
      "training_step_time": 9.603153705596924
    },
    {
      "epoch": 5.8496475219726563e-05,
      "model_forward_time": 2.3680994510650635,
      "step": 61338
    },
    {
      "epoch": 5.8496475219726563e-05,
      "step": 61338,
      "training_step_time": 9.134669303894043
    },
    {
      "epoch": 5.849742889404297e-05,
      "model_forward_time": 1.749643087387085,
      "step": 61339
    },
    {
      "epoch": 5.849742889404297e-05,
      "step": 61339,
      "training_step_time": 7.702056646347046
    },
    {
      "epoch": 5.849838256835938e-05,
      "grad_norm": 0.09088139981031418,
      "learning_rate": 3.5592209731320354e-06,
      "loss": 0.0585,
      "step": 61340
    },
    {
      "epoch": 5.849838256835938e-05,
      "model_forward_time": 1.724428653717041,
      "step": 61340
    },
    {
      "dynamics_loss_avg": 0.056684701889753344,
      "epoch": 5.849838256835938e-05,
      "step": 61340
    },
    {
      "action_loss_avg": 0.0019555603328626605,
      "epoch": 5.849838256835938e-05,
      "step": 61340
    },
    {
      "epoch": 5.849838256835938e-05,
      "step": 61340,
      "training_step_time": 7.870927572250366
    },
    {
      "epoch": 5.849933624267578e-05,
      "model_forward_time": 1.7510735988616943,
      "step": 61341
    },
    {
      "epoch": 5.849933624267578e-05,
      "step": 61341,
      "training_step_time": 7.719722509384155
    },
    {
      "epoch": 5.8500289916992186e-05,
      "model_forward_time": 1.741948127746582,
      "step": 61342
    },
    {
      "epoch": 5.8500289916992186e-05,
      "step": 61342,
      "training_step_time": 7.624017953872681
    },
    {
      "epoch": 5.85012435913086e-05,
      "model_forward_time": 1.7423639297485352,
      "step": 61343
    },
    {
      "epoch": 5.85012435913086e-05,
      "step": 61343,
      "training_step_time": 7.2935569286346436
    },
    {
      "epoch": 5.8502197265625e-05,
      "model_forward_time": 1.7388215065002441,
      "step": 61344
    },
    {
      "epoch": 5.8502197265625e-05,
      "step": 61344,
      "training_step_time": 7.0999321937561035
    },
    {
      "epoch": 5.8503150939941405e-05,
      "model_forward_time": 1.7400763034820557,
      "step": 61345
    },
    {
      "epoch": 5.8503150939941405e-05,
      "step": 61345,
      "training_step_time": 8.134117603302002
    },
    {
      "epoch": 5.850410461425781e-05,
      "model_forward_time": 1.7460441589355469,
      "step": 61346
    },
    {
      "epoch": 5.850410461425781e-05,
      "step": 61346,
      "training_step_time": 8.435863494873047
    },
    {
      "epoch": 5.850505828857422e-05,
      "model_forward_time": 2.353285789489746,
      "step": 61347
    },
    {
      "epoch": 5.850505828857422e-05,
      "step": 61347,
      "training_step_time": 8.607396602630615
    },
    {
      "epoch": 5.8506011962890624e-05,
      "model_forward_time": 1.7542576789855957,
      "step": 61348
    },
    {
      "epoch": 5.8506011962890624e-05,
      "step": 61348,
      "training_step_time": 7.698712110519409
    },
    {
      "epoch": 5.850696563720703e-05,
      "model_forward_time": 1.7472283840179443,
      "step": 61349
    },
    {
      "epoch": 5.850696563720703e-05,
      "step": 61349,
      "training_step_time": 7.8003034591674805
    },
    {
      "epoch": 5.850791931152344e-05,
      "grad_norm": 0.10931513458490372,
      "learning_rate": 3.5576377164639814e-06,
      "loss": 0.0605,
      "step": 61350
    },
    {
      "epoch": 5.850791931152344e-05,
      "model_forward_time": 1.7300794124603271,
      "step": 61350
    },
    {
      "dynamics_loss_avg": 0.05352478642016649,
      "epoch": 5.850791931152344e-05,
      "step": 61350
    },
    {
      "action_loss_avg": 0.001607581548159942,
      "epoch": 5.850791931152344e-05,
      "step": 61350
    },
    {
      "epoch": 5.850791931152344e-05,
      "step": 61350,
      "training_step_time": 7.50116229057312
    },
    {
      "epoch": 5.850887298583984e-05,
      "model_forward_time": 1.7481465339660645,
      "step": 61351
    },
    {
      "epoch": 5.850887298583984e-05,
      "step": 61351,
      "training_step_time": 7.224007844924927
    },
    {
      "epoch": 5.850982666015625e-05,
      "model_forward_time": 1.7437865734100342,
      "step": 61352
    },
    {
      "epoch": 5.850982666015625e-05,
      "step": 61352,
      "training_step_time": 7.541268825531006
    },
    {
      "epoch": 5.851078033447266e-05,
      "model_forward_time": 1.746551275253296,
      "step": 61353
    },
    {
      "epoch": 5.851078033447266e-05,
      "step": 61353,
      "training_step_time": 6.813972473144531
    },
    {
      "epoch": 5.851173400878906e-05,
      "model_forward_time": 1.7421910762786865,
      "step": 61354
    },
    {
      "epoch": 5.851173400878906e-05,
      "step": 61354,
      "training_step_time": 7.551835298538208
    },
    {
      "epoch": 5.8512687683105466e-05,
      "model_forward_time": 1.7507917881011963,
      "step": 61355
    },
    {
      "epoch": 5.8512687683105466e-05,
      "step": 61355,
      "training_step_time": 8.441650390625
    },
    {
      "epoch": 5.8513641357421877e-05,
      "model_forward_time": 2.271502733230591,
      "step": 61356
    },
    {
      "epoch": 5.8513641357421877e-05,
      "step": 61356,
      "training_step_time": 8.722869157791138
    },
    {
      "epoch": 5.851459503173828e-05,
      "model_forward_time": 1.7370316982269287,
      "step": 61357
    },
    {
      "epoch": 5.851459503173828e-05,
      "step": 61357,
      "training_step_time": 9.549659013748169
    },
    {
      "epoch": 5.8515548706054685e-05,
      "model_forward_time": 1.7492871284484863,
      "step": 61358
    },
    {
      "epoch": 5.8515548706054685e-05,
      "step": 61358,
      "training_step_time": 7.665438890457153
    },
    {
      "epoch": 5.8516502380371095e-05,
      "model_forward_time": 1.7547953128814697,
      "step": 61359
    },
    {
      "epoch": 5.8516502380371095e-05,
      "step": 61359,
      "training_step_time": 8.887723684310913
    },
    {
      "epoch": 5.85174560546875e-05,
      "grad_norm": 0.1428706794977188,
      "learning_rate": 3.5560546175304977e-06,
      "loss": 0.0581,
      "step": 61360
    },
    {
      "epoch": 5.85174560546875e-05,
      "model_forward_time": 1.740424633026123,
      "step": 61360
    },
    {
      "dynamics_loss_avg": 0.067007677257061,
      "epoch": 5.85174560546875e-05,
      "step": 61360
    },
    {
      "action_loss_avg": 0.0021189747494645417,
      "epoch": 5.85174560546875e-05,
      "step": 61360
    },
    {
      "epoch": 5.85174560546875e-05,
      "step": 61360,
      "training_step_time": 7.23627495765686
    },
    {
      "epoch": 5.8518409729003903e-05,
      "model_forward_time": 1.7494685649871826,
      "step": 61361
    },
    {
      "epoch": 5.8518409729003903e-05,
      "step": 61361,
      "training_step_time": 7.704211235046387
    },
    {
      "epoch": 5.8519363403320314e-05,
      "model_forward_time": 1.7585158348083496,
      "step": 61362
    },
    {
      "epoch": 5.8519363403320314e-05,
      "step": 61362,
      "training_step_time": 6.783007383346558
    },
    {
      "epoch": 5.852031707763672e-05,
      "model_forward_time": 1.378051996231079,
      "step": 61363
    },
    {
      "epoch": 5.852031707763672e-05,
      "step": 61363,
      "training_step_time": 7.196215629577637
    },
    {
      "epoch": 5.852127075195312e-05,
      "model_forward_time": 1.7520389556884766,
      "step": 61364
    },
    {
      "epoch": 5.852127075195312e-05,
      "step": 61364,
      "training_step_time": 8.626871824264526
    },
    {
      "epoch": 5.852222442626953e-05,
      "model_forward_time": 2.31842041015625,
      "step": 61365
    },
    {
      "epoch": 5.852222442626953e-05,
      "step": 61365,
      "training_step_time": 8.840529680252075
    },
    {
      "epoch": 5.852317810058594e-05,
      "model_forward_time": 1.802504539489746,
      "step": 61366
    },
    {
      "epoch": 5.852317810058594e-05,
      "step": 61366,
      "training_step_time": 9.492167949676514
    },
    {
      "epoch": 5.852413177490234e-05,
      "model_forward_time": 1.7400686740875244,
      "step": 61367
    },
    {
      "epoch": 5.852413177490234e-05,
      "step": 61367,
      "training_step_time": 8.659791469573975
    },
    {
      "epoch": 5.852508544921875e-05,
      "model_forward_time": 1.751190423965454,
      "step": 61368
    },
    {
      "epoch": 5.852508544921875e-05,
      "step": 61368,
      "training_step_time": 7.892118453979492
    },
    {
      "epoch": 5.8526039123535156e-05,
      "model_forward_time": 1.7282395362854004,
      "step": 61369
    },
    {
      "epoch": 5.8526039123535156e-05,
      "step": 61369,
      "training_step_time": 7.6150994300842285
    },
    {
      "epoch": 5.852699279785156e-05,
      "grad_norm": 0.10399957001209259,
      "learning_rate": 3.5544716765047103e-06,
      "loss": 0.0569,
      "step": 61370
    },
    {
      "epoch": 5.852699279785156e-05,
      "model_forward_time": 1.772674798965454,
      "step": 61370
    },
    {
      "dynamics_loss_avg": 0.057594836503267285,
      "epoch": 5.852699279785156e-05,
      "step": 61370
    },
    {
      "action_loss_avg": 0.001541894196998328,
      "epoch": 5.852699279785156e-05,
      "step": 61370
    },
    {
      "epoch": 5.852699279785156e-05,
      "step": 61370,
      "training_step_time": 7.653167009353638
    },
    {
      "epoch": 5.852794647216797e-05,
      "model_forward_time": 1.7416884899139404,
      "step": 61371
    },
    {
      "epoch": 5.852794647216797e-05,
      "step": 61371,
      "training_step_time": 6.75841212272644
    },
    {
      "epoch": 5.8528900146484375e-05,
      "model_forward_time": 1.745764970779419,
      "step": 61372
    },
    {
      "epoch": 5.8528900146484375e-05,
      "step": 61372,
      "training_step_time": 7.991055965423584
    },
    {
      "epoch": 5.852985382080078e-05,
      "model_forward_time": 1.7447257041931152,
      "step": 61373
    },
    {
      "epoch": 5.852985382080078e-05,
      "step": 61373,
      "training_step_time": 7.628438711166382
    },
    {
      "epoch": 5.853080749511719e-05,
      "model_forward_time": 1.7424888610839844,
      "step": 61374
    },
    {
      "epoch": 5.853080749511719e-05,
      "step": 61374,
      "training_step_time": 8.431668281555176
    },
    {
      "epoch": 5.8531761169433594e-05,
      "model_forward_time": 2.3028159141540527,
      "step": 61375
    },
    {
      "epoch": 5.8531761169433594e-05,
      "step": 61375,
      "training_step_time": 8.340545654296875
    },
    {
      "epoch": 5.853271484375e-05,
      "model_forward_time": 1.3659422397613525,
      "step": 61376
    },
    {
      "epoch": 5.853271484375e-05,
      "step": 61376,
      "training_step_time": 7.828466415405273
    },
    {
      "epoch": 5.853366851806641e-05,
      "model_forward_time": 1.7564303874969482,
      "step": 61377
    },
    {
      "epoch": 5.853366851806641e-05,
      "step": 61377,
      "training_step_time": 7.747233629226685
    },
    {
      "epoch": 5.853462219238281e-05,
      "model_forward_time": 1.7501440048217773,
      "step": 61378
    },
    {
      "epoch": 5.853462219238281e-05,
      "step": 61378,
      "training_step_time": 7.981462001800537
    },
    {
      "epoch": 5.853557586669922e-05,
      "model_forward_time": 1.7485675811767578,
      "step": 61379
    },
    {
      "epoch": 5.853557586669922e-05,
      "step": 61379,
      "training_step_time": 8.467180967330933
    },
    {
      "epoch": 5.853652954101563e-05,
      "grad_norm": 0.10363932698965073,
      "learning_rate": 3.552888893559727e-06,
      "loss": 0.0568,
      "step": 61380
    },
    {
      "epoch": 5.853652954101563e-05,
      "model_forward_time": 1.7386906147003174,
      "step": 61380
    },
    {
      "dynamics_loss_avg": 0.04271744154393673,
      "epoch": 5.853652954101563e-05,
      "step": 61380
    },
    {
      "action_loss_avg": 0.0012498535215854645,
      "epoch": 5.853652954101563e-05,
      "step": 61380
    },
    {
      "epoch": 5.853652954101563e-05,
      "step": 61380,
      "training_step_time": 6.756281137466431
    },
    {
      "epoch": 5.853748321533203e-05,
      "model_forward_time": 1.746713638305664,
      "step": 61381
    },
    {
      "epoch": 5.853748321533203e-05,
      "step": 61381,
      "training_step_time": 7.587041139602661
    },
    {
      "epoch": 5.8538436889648435e-05,
      "model_forward_time": 1.7476274967193604,
      "step": 61382
    },
    {
      "epoch": 5.8538436889648435e-05,
      "step": 61382,
      "training_step_time": 8.313002109527588
    },
    {
      "epoch": 5.8539390563964846e-05,
      "model_forward_time": 1.7494261264801025,
      "step": 61383
    },
    {
      "epoch": 5.8539390563964846e-05,
      "step": 61383,
      "training_step_time": 9.119738817214966
    },
    {
      "epoch": 5.854034423828125e-05,
      "model_forward_time": 2.28006649017334,
      "step": 61384
    },
    {
      "epoch": 5.854034423828125e-05,
      "step": 61384,
      "training_step_time": 10.292657852172852
    },
    {
      "epoch": 5.8541297912597654e-05,
      "model_forward_time": 1.8132100105285645,
      "step": 61385
    },
    {
      "epoch": 5.8541297912597654e-05,
      "step": 61385,
      "training_step_time": 7.625285387039185
    },
    {
      "epoch": 5.8542251586914065e-05,
      "model_forward_time": 1.7372112274169922,
      "step": 61386
    },
    {
      "epoch": 5.8542251586914065e-05,
      "step": 61386,
      "training_step_time": 7.722313642501831
    },
    {
      "epoch": 5.854320526123047e-05,
      "model_forward_time": 1.7281899452209473,
      "step": 61387
    },
    {
      "epoch": 5.854320526123047e-05,
      "step": 61387,
      "training_step_time": 8.62408709526062
    },
    {
      "epoch": 5.854415893554687e-05,
      "model_forward_time": 1.7418112754821777,
      "step": 61388
    },
    {
      "epoch": 5.854415893554687e-05,
      "step": 61388,
      "training_step_time": 7.208555221557617
    },
    {
      "epoch": 5.8545112609863284e-05,
      "model_forward_time": 1.383690595626831,
      "step": 61389
    },
    {
      "epoch": 5.8545112609863284e-05,
      "step": 61389,
      "training_step_time": 7.581773042678833
    },
    {
      "epoch": 5.854606628417969e-05,
      "grad_norm": 0.12209765613079071,
      "learning_rate": 3.551306268868636e-06,
      "loss": 0.0568,
      "step": 61390
    },
    {
      "epoch": 5.854606628417969e-05,
      "model_forward_time": 1.7533140182495117,
      "step": 61390
    },
    {
      "dynamics_loss_avg": 0.05920075327157974,
      "epoch": 5.854606628417969e-05,
      "step": 61390
    },
    {
      "action_loss_avg": 0.001903519331244752,
      "epoch": 5.854606628417969e-05,
      "step": 61390
    },
    {
      "epoch": 5.854606628417969e-05,
      "step": 61390,
      "training_step_time": 7.7281036376953125
    },
    {
      "epoch": 5.854701995849609e-05,
      "model_forward_time": 1.726797342300415,
      "step": 61391
    },
    {
      "epoch": 5.854701995849609e-05,
      "step": 61391,
      "training_step_time": 7.562101125717163
    },
    {
      "epoch": 5.85479736328125e-05,
      "model_forward_time": 1.7435283660888672,
      "step": 61392
    },
    {
      "epoch": 5.85479736328125e-05,
      "step": 61392,
      "training_step_time": 7.684635639190674
    },
    {
      "epoch": 5.854892730712891e-05,
      "model_forward_time": 2.3049404621124268,
      "step": 61393
    },
    {
      "epoch": 5.854892730712891e-05,
      "step": 61393,
      "training_step_time": 9.413975715637207
    },
    {
      "epoch": 5.854988098144531e-05,
      "model_forward_time": 1.7486324310302734,
      "step": 61394
    },
    {
      "epoch": 5.854988098144531e-05,
      "step": 61394,
      "training_step_time": 7.741371393203735
    },
    {
      "epoch": 5.855083465576172e-05,
      "model_forward_time": 1.7512357234954834,
      "step": 61395
    },
    {
      "epoch": 5.855083465576172e-05,
      "step": 61395,
      "training_step_time": 7.71721339225769
    },
    {
      "epoch": 5.8551788330078126e-05,
      "model_forward_time": 1.7505238056182861,
      "step": 61396
    },
    {
      "epoch": 5.8551788330078126e-05,
      "step": 61396,
      "training_step_time": 7.779417276382446
    },
    {
      "epoch": 5.855274200439453e-05,
      "model_forward_time": 1.746488332748413,
      "step": 61397
    },
    {
      "epoch": 5.855274200439453e-05,
      "step": 61397,
      "training_step_time": 7.705943822860718
    },
    {
      "epoch": 5.855369567871094e-05,
      "model_forward_time": 1.7546820640563965,
      "step": 61398
    },
    {
      "epoch": 5.855369567871094e-05,
      "step": 61398,
      "training_step_time": 7.630636930465698
    },
    {
      "epoch": 5.8554649353027345e-05,
      "model_forward_time": 1.744704246520996,
      "step": 61399
    },
    {
      "epoch": 5.8554649353027345e-05,
      "step": 61399,
      "training_step_time": 7.664854526519775
    },
    {
      "epoch": 5.855560302734375e-05,
      "grad_norm": 0.11877647042274475,
      "learning_rate": 3.5497238026045166e-06,
      "loss": 0.0547,
      "step": 61400
    },
    {
      "epoch": 5.855560302734375e-05,
      "model_forward_time": 1.7374088764190674,
      "step": 61400
    },
    {
      "dynamics_loss_avg": 0.05676418114453554,
      "epoch": 5.855560302734375e-05,
      "step": 61400
    },
    {
      "action_loss_avg": 0.0018403466674499214,
      "epoch": 5.855560302734375e-05,
      "step": 61400
    },
    {
      "epoch": 5.855560302734375e-05,
      "step": 61400,
      "training_step_time": 7.605207681655884
    },
    {
      "epoch": 5.855655670166016e-05,
      "model_forward_time": 1.7367260456085205,
      "step": 61401
    },
    {
      "epoch": 5.855655670166016e-05,
      "step": 61401,
      "training_step_time": 7.771218538284302
    },
    {
      "epoch": 5.8557510375976563e-05,
      "model_forward_time": 2.3225855827331543,
      "step": 61402
    },
    {
      "epoch": 5.8557510375976563e-05,
      "step": 61402,
      "training_step_time": 8.414062976837158
    },
    {
      "epoch": 5.855846405029297e-05,
      "model_forward_time": 1.030592679977417,
      "step": 61403
    },
    {
      "epoch": 5.855846405029297e-05,
      "step": 61403,
      "training_step_time": 9.322062253952026
    },
    {
      "epoch": 5.855941772460938e-05,
      "model_forward_time": 1.7483797073364258,
      "step": 61404
    },
    {
      "epoch": 5.855941772460938e-05,
      "step": 61404,
      "training_step_time": 7.624388933181763
    },
    {
      "epoch": 5.856037139892578e-05,
      "model_forward_time": 1.7482588291168213,
      "step": 61405
    },
    {
      "epoch": 5.856037139892578e-05,
      "step": 61405,
      "training_step_time": 8.71225094795227
    },
    {
      "epoch": 5.8561325073242186e-05,
      "model_forward_time": 1.750859260559082,
      "step": 61406
    },
    {
      "epoch": 5.8561325073242186e-05,
      "step": 61406,
      "training_step_time": 7.639353275299072
    },
    {
      "epoch": 5.856227874755859e-05,
      "model_forward_time": 1.7497365474700928,
      "step": 61407
    },
    {
      "epoch": 5.856227874755859e-05,
      "step": 61407,
      "training_step_time": 7.654744625091553
    },
    {
      "epoch": 5.8563232421875e-05,
      "model_forward_time": 1.7515625953674316,
      "step": 61408
    },
    {
      "epoch": 5.8563232421875e-05,
      "step": 61408,
      "training_step_time": 7.5391011238098145
    },
    {
      "epoch": 5.8564186096191405e-05,
      "model_forward_time": 1.746516227722168,
      "step": 61409
    },
    {
      "epoch": 5.8564186096191405e-05,
      "step": 61409,
      "training_step_time": 7.589215517044067
    },
    {
      "epoch": 5.856513977050781e-05,
      "grad_norm": 0.09327128529548645,
      "learning_rate": 3.5481414949404212e-06,
      "loss": 0.056,
      "step": 61410
    },
    {
      "epoch": 5.856513977050781e-05,
      "model_forward_time": 1.7364611625671387,
      "step": 61410
    },
    {
      "dynamics_loss_avg": 0.04266819376498461,
      "epoch": 5.856513977050781e-05,
      "step": 61410
    },
    {
      "action_loss_avg": 0.0013982920674607158,
      "epoch": 5.856513977050781e-05,
      "step": 61410
    },
    {
      "epoch": 5.856513977050781e-05,
      "step": 61410,
      "training_step_time": 7.815824270248413
    },
    {
      "epoch": 5.856609344482422e-05,
      "model_forward_time": 1.7525324821472168,
      "step": 61411
    },
    {
      "epoch": 5.856609344482422e-05,
      "step": 61411,
      "training_step_time": 7.636102199554443
    },
    {
      "epoch": 5.8567047119140624e-05,
      "model_forward_time": 2.3307602405548096,
      "step": 61412
    },
    {
      "epoch": 5.8567047119140624e-05,
      "step": 61412,
      "training_step_time": 9.246058940887451
    },
    {
      "epoch": 5.856800079345703e-05,
      "model_forward_time": 1.7524449825286865,
      "step": 61413
    },
    {
      "epoch": 5.856800079345703e-05,
      "step": 61413,
      "training_step_time": 7.5987207889556885
    },
    {
      "epoch": 5.856895446777344e-05,
      "model_forward_time": 1.7419865131378174,
      "step": 61414
    },
    {
      "epoch": 5.856895446777344e-05,
      "step": 61414,
      "training_step_time": 8.430789709091187
    },
    {
      "epoch": 5.856990814208984e-05,
      "model_forward_time": 1.7518310546875,
      "step": 61415
    },
    {
      "epoch": 5.856990814208984e-05,
      "step": 61415,
      "training_step_time": 7.609131574630737
    },
    {
      "epoch": 5.857086181640625e-05,
      "model_forward_time": 1.7187862396240234,
      "step": 61416
    },
    {
      "epoch": 5.857086181640625e-05,
      "step": 61416,
      "training_step_time": 7.523987531661987
    },
    {
      "epoch": 5.857181549072266e-05,
      "model_forward_time": 1.7342236042022705,
      "step": 61417
    },
    {
      "epoch": 5.857181549072266e-05,
      "step": 61417,
      "training_step_time": 6.766473770141602
    },
    {
      "epoch": 5.857276916503906e-05,
      "model_forward_time": 1.7361047267913818,
      "step": 61418
    },
    {
      "epoch": 5.857276916503906e-05,
      "step": 61418,
      "training_step_time": 7.708037376403809
    },
    {
      "epoch": 5.8573722839355466e-05,
      "model_forward_time": 1.73451566696167,
      "step": 61419
    },
    {
      "epoch": 5.8573722839355466e-05,
      "step": 61419,
      "training_step_time": 7.804668426513672
    },
    {
      "epoch": 5.857467651367188e-05,
      "grad_norm": 0.1030995100736618,
      "learning_rate": 3.5465593460493886e-06,
      "loss": 0.0573,
      "step": 61420
    },
    {
      "epoch": 5.857467651367188e-05,
      "model_forward_time": 1.7244138717651367,
      "step": 61420
    },
    {
      "dynamics_loss_avg": 0.06229244824498892,
      "epoch": 5.857467651367188e-05,
      "step": 61420
    },
    {
      "action_loss_avg": 0.0023571740835905074,
      "epoch": 5.857467651367188e-05,
      "step": 61420
    },
    {
      "epoch": 5.857467651367188e-05,
      "step": 61420,
      "training_step_time": 8.683306217193604
    },
    {
      "epoch": 5.857563018798828e-05,
      "model_forward_time": 2.2930421829223633,
      "step": 61421
    },
    {
      "epoch": 5.857563018798828e-05,
      "step": 61421,
      "training_step_time": 8.982940673828125
    },
    {
      "epoch": 5.8576583862304685e-05,
      "model_forward_time": 1.7133216857910156,
      "step": 61422
    },
    {
      "epoch": 5.8576583862304685e-05,
      "step": 61422,
      "training_step_time": 8.984512567520142
    },
    {
      "epoch": 5.8577537536621095e-05,
      "model_forward_time": 1.7280421257019043,
      "step": 61423
    },
    {
      "epoch": 5.8577537536621095e-05,
      "step": 61423,
      "training_step_time": 7.725692987442017
    },
    {
      "epoch": 5.85784912109375e-05,
      "model_forward_time": 1.732344150543213,
      "step": 61424
    },
    {
      "epoch": 5.85784912109375e-05,
      "step": 61424,
      "training_step_time": 8.710346460342407
    },
    {
      "epoch": 5.8579444885253904e-05,
      "model_forward_time": 1.7362725734710693,
      "step": 61425
    },
    {
      "epoch": 5.8579444885253904e-05,
      "step": 61425,
      "training_step_time": 7.321184158325195
    },
    {
      "epoch": 5.8580398559570314e-05,
      "model_forward_time": 1.7318108081817627,
      "step": 61426
    },
    {
      "epoch": 5.8580398559570314e-05,
      "step": 61426,
      "training_step_time": 7.629359483718872
    },
    {
      "epoch": 5.858135223388672e-05,
      "model_forward_time": 1.7352721691131592,
      "step": 61427
    },
    {
      "epoch": 5.858135223388672e-05,
      "step": 61427,
      "training_step_time": 7.744458913803101
    },
    {
      "epoch": 5.858230590820312e-05,
      "model_forward_time": 1.7284038066864014,
      "step": 61428
    },
    {
      "epoch": 5.858230590820312e-05,
      "step": 61428,
      "training_step_time": 9.369846105575562
    },
    {
      "epoch": 5.858325958251953e-05,
      "model_forward_time": 1.7352886199951172,
      "step": 61429
    },
    {
      "epoch": 5.858325958251953e-05,
      "step": 61429,
      "training_step_time": 7.673078775405884
    },
    {
      "epoch": 5.858421325683594e-05,
      "grad_norm": 0.11246387660503387,
      "learning_rate": 3.544977356104441e-06,
      "loss": 0.0559,
      "step": 61430
    },
    {
      "epoch": 5.858421325683594e-05,
      "model_forward_time": 2.275813102722168,
      "step": 61430
    },
    {
      "dynamics_loss_avg": 0.05981778092682362,
      "epoch": 5.858421325683594e-05,
      "step": 61430
    },
    {
      "action_loss_avg": 0.002031829732004553,
      "epoch": 5.858421325683594e-05,
      "step": 61430
    },
    {
      "epoch": 5.858421325683594e-05,
      "step": 61430,
      "training_step_time": 7.728265762329102
    },
    {
      "epoch": 5.858516693115234e-05,
      "model_forward_time": 1.7912089824676514,
      "step": 61431
    },
    {
      "epoch": 5.858516693115234e-05,
      "step": 61431,
      "training_step_time": 9.159795999526978
    },
    {
      "epoch": 5.858612060546875e-05,
      "model_forward_time": 1.733508825302124,
      "step": 61432
    },
    {
      "epoch": 5.858612060546875e-05,
      "step": 61432,
      "training_step_time": 7.64629602432251
    },
    {
      "epoch": 5.8587074279785156e-05,
      "model_forward_time": 1.7320809364318848,
      "step": 61433
    },
    {
      "epoch": 5.8587074279785156e-05,
      "step": 61433,
      "training_step_time": 7.6874916553497314
    },
    {
      "epoch": 5.858802795410156e-05,
      "model_forward_time": 1.7361905574798584,
      "step": 61434
    },
    {
      "epoch": 5.858802795410156e-05,
      "step": 61434,
      "training_step_time": 7.611244440078735
    },
    {
      "epoch": 5.858898162841797e-05,
      "model_forward_time": 1.720982551574707,
      "step": 61435
    },
    {
      "epoch": 5.858898162841797e-05,
      "step": 61435,
      "training_step_time": 7.575055122375488
    },
    {
      "epoch": 5.8589935302734375e-05,
      "model_forward_time": 1.7353065013885498,
      "step": 61436
    },
    {
      "epoch": 5.8589935302734375e-05,
      "step": 61436,
      "training_step_time": 7.597195386886597
    },
    {
      "epoch": 5.859088897705078e-05,
      "model_forward_time": 1.7245633602142334,
      "step": 61437
    },
    {
      "epoch": 5.859088897705078e-05,
      "step": 61437,
      "training_step_time": 7.488964796066284
    },
    {
      "epoch": 5.859184265136719e-05,
      "model_forward_time": 1.7336030006408691,
      "step": 61438
    },
    {
      "epoch": 5.859184265136719e-05,
      "step": 61438,
      "training_step_time": 7.730988264083862
    },
    {
      "epoch": 5.8592796325683594e-05,
      "model_forward_time": 2.2936289310455322,
      "step": 61439
    },
    {
      "epoch": 5.8592796325683594e-05,
      "step": 61439,
      "training_step_time": 9.644777059555054
    },
    {
      "epoch": 5.859375e-05,
      "grad_norm": 0.12097277492284775,
      "learning_rate": 3.5433955252785824e-06,
      "loss": 0.0563,
      "step": 61440
    },
    {
      "epoch": 5.859375e-05,
      "model_forward_time": 1.7521061897277832,
      "step": 61440
    },
    {
      "dynamics_loss_avg": 0.06054056398570538,
      "epoch": 5.859375e-05,
      "step": 61440
    },
    {
      "action_loss_avg": 0.0018563146033557132,
      "epoch": 5.859375e-05,
      "step": 61440
    },
    {
      "epoch": 5.859375e-05,
      "step": 61440,
      "training_step_time": 10.28914761543274
    },
    {
      "epoch": 5.859470367431641e-05,
      "model_forward_time": 1.7273964881896973,
      "step": 61441
    },
    {
      "epoch": 5.859470367431641e-05,
      "step": 61441,
      "training_step_time": 9.648594379425049
    },
    {
      "epoch": 5.859565734863281e-05,
      "model_forward_time": 1.724827766418457,
      "step": 61442
    },
    {
      "epoch": 5.859565734863281e-05,
      "step": 61442,
      "training_step_time": 9.286332607269287
    },
    {
      "epoch": 5.859661102294922e-05,
      "model_forward_time": 1.7200841903686523,
      "step": 61443
    },
    {
      "epoch": 5.859661102294922e-05,
      "step": 61443,
      "training_step_time": 7.588149547576904
    },
    {
      "epoch": 5.859756469726563e-05,
      "model_forward_time": 1.7346007823944092,
      "step": 61444
    },
    {
      "epoch": 5.859756469726563e-05,
      "step": 61444,
      "training_step_time": 7.608100175857544
    },
    {
      "epoch": 5.859851837158203e-05,
      "model_forward_time": 1.7296676635742188,
      "step": 61445
    },
    {
      "epoch": 5.859851837158203e-05,
      "step": 61445,
      "training_step_time": 7.694082498550415
    },
    {
      "epoch": 5.8599472045898436e-05,
      "model_forward_time": 1.7328343391418457,
      "step": 61446
    },
    {
      "epoch": 5.8599472045898436e-05,
      "step": 61446,
      "training_step_time": 7.574012517929077
    },
    {
      "epoch": 5.8600425720214846e-05,
      "model_forward_time": 1.7326726913452148,
      "step": 61447
    },
    {
      "epoch": 5.8600425720214846e-05,
      "step": 61447,
      "training_step_time": 8.816544771194458
    },
    {
      "epoch": 5.860137939453125e-05,
      "model_forward_time": 1.7206528186798096,
      "step": 61448
    },
    {
      "epoch": 5.860137939453125e-05,
      "step": 61448,
      "training_step_time": 9.29922366142273
    },
    {
      "epoch": 5.8602333068847654e-05,
      "model_forward_time": 1.7290866374969482,
      "step": 61449
    },
    {
      "epoch": 5.8602333068847654e-05,
      "step": 61449,
      "training_step_time": 8.477731227874756
    },
    {
      "epoch": 5.8603286743164065e-05,
      "grad_norm": 0.12004341930150986,
      "learning_rate": 3.541813853744799e-06,
      "loss": 0.0565,
      "step": 61450
    },
    {
      "epoch": 5.8603286743164065e-05,
      "model_forward_time": 1.7280247211456299,
      "step": 61450
    },
    {
      "dynamics_loss_avg": 0.04596904926002025,
      "epoch": 5.8603286743164065e-05,
      "step": 61450
    },
    {
      "action_loss_avg": 0.000949405221035704,
      "epoch": 5.8603286743164065e-05,
      "step": 61450
    },
    {
      "epoch": 5.8603286743164065e-05,
      "step": 61450,
      "training_step_time": 8.662258625030518
    },
    {
      "epoch": 5.860424041748047e-05,
      "model_forward_time": 1.7298803329467773,
      "step": 61451
    },
    {
      "epoch": 5.860424041748047e-05,
      "step": 61451,
      "training_step_time": 8.755512237548828
    },
    {
      "epoch": 5.860519409179687e-05,
      "model_forward_time": 1.7208950519561768,
      "step": 61452
    },
    {
      "epoch": 5.860519409179687e-05,
      "step": 61452,
      "training_step_time": 7.604700565338135
    },
    {
      "epoch": 5.8606147766113284e-05,
      "model_forward_time": 1.7253787517547607,
      "step": 61453
    },
    {
      "epoch": 5.8606147766113284e-05,
      "step": 61453,
      "training_step_time": 7.7103025913238525
    },
    {
      "epoch": 5.860710144042969e-05,
      "model_forward_time": 1.3519456386566162,
      "step": 61454
    },
    {
      "epoch": 5.860710144042969e-05,
      "step": 61454,
      "training_step_time": 7.625777006149292
    },
    {
      "epoch": 5.860805511474609e-05,
      "model_forward_time": 1.740363597869873,
      "step": 61455
    },
    {
      "epoch": 5.860805511474609e-05,
      "step": 61455,
      "training_step_time": 8.808150291442871
    },
    {
      "epoch": 5.86090087890625e-05,
      "model_forward_time": 1.7252366542816162,
      "step": 61456
    },
    {
      "epoch": 5.86090087890625e-05,
      "step": 61456,
      "training_step_time": 7.793806552886963
    },
    {
      "epoch": 5.860996246337891e-05,
      "model_forward_time": 1.7243311405181885,
      "step": 61457
    },
    {
      "epoch": 5.860996246337891e-05,
      "step": 61457,
      "training_step_time": 8.352930784225464
    },
    {
      "epoch": 5.861091613769531e-05,
      "model_forward_time": 2.274421453475952,
      "step": 61458
    },
    {
      "epoch": 5.861091613769531e-05,
      "step": 61458,
      "training_step_time": 8.54324746131897
    },
    {
      "epoch": 5.861186981201172e-05,
      "model_forward_time": 1.7935709953308105,
      "step": 61459
    },
    {
      "epoch": 5.861186981201172e-05,
      "step": 61459,
      "training_step_time": 9.74333930015564
    },
    {
      "epoch": 5.8612823486328126e-05,
      "grad_norm": 0.10201551020145416,
      "learning_rate": 3.5402323416760576e-06,
      "loss": 0.0568,
      "step": 61460
    },
    {
      "epoch": 5.8612823486328126e-05,
      "model_forward_time": 1.719792127609253,
      "step": 61460
    },
    {
      "dynamics_loss_avg": 0.04819300062954426,
      "epoch": 5.8612823486328126e-05,
      "step": 61460
    },
    {
      "action_loss_avg": 0.0016474886040668934,
      "epoch": 5.8612823486328126e-05,
      "step": 61460
    },
    {
      "epoch": 5.8612823486328126e-05,
      "step": 61460,
      "training_step_time": 7.714769601821899
    },
    {
      "epoch": 5.861377716064453e-05,
      "model_forward_time": 1.732254981994629,
      "step": 61461
    },
    {
      "epoch": 5.861377716064453e-05,
      "step": 61461,
      "training_step_time": 7.738196611404419
    },
    {
      "epoch": 5.861473083496094e-05,
      "model_forward_time": 1.7258944511413574,
      "step": 61462
    },
    {
      "epoch": 5.861473083496094e-05,
      "step": 61462,
      "training_step_time": 7.561187982559204
    },
    {
      "epoch": 5.8615684509277345e-05,
      "model_forward_time": 1.7359247207641602,
      "step": 61463
    },
    {
      "epoch": 5.8615684509277345e-05,
      "step": 61463,
      "training_step_time": 7.694333553314209
    },
    {
      "epoch": 5.861663818359375e-05,
      "model_forward_time": 1.735410451889038,
      "step": 61464
    },
    {
      "epoch": 5.861663818359375e-05,
      "step": 61464,
      "training_step_time": 7.7443528175354
    },
    {
      "epoch": 5.861759185791016e-05,
      "model_forward_time": 1.7337615489959717,
      "step": 61465
    },
    {
      "epoch": 5.861759185791016e-05,
      "step": 61465,
      "training_step_time": 8.837932348251343
    },
    {
      "epoch": 5.8618545532226564e-05,
      "model_forward_time": 1.7209508419036865,
      "step": 61466
    },
    {
      "epoch": 5.8618545532226564e-05,
      "step": 61466,
      "training_step_time": 8.810121297836304
    },
    {
      "epoch": 5.861949920654297e-05,
      "model_forward_time": 2.2837564945220947,
      "step": 61467
    },
    {
      "epoch": 5.861949920654297e-05,
      "step": 61467,
      "training_step_time": 8.850926876068115
    },
    {
      "epoch": 5.862045288085938e-05,
      "model_forward_time": 1.7812836170196533,
      "step": 61468
    },
    {
      "epoch": 5.862045288085938e-05,
      "step": 61468,
      "training_step_time": 8.610264301300049
    },
    {
      "epoch": 5.862140655517578e-05,
      "model_forward_time": 1.7363038063049316,
      "step": 61469
    },
    {
      "epoch": 5.862140655517578e-05,
      "step": 61469,
      "training_step_time": 7.691213607788086
    },
    {
      "epoch": 5.8622360229492186e-05,
      "grad_norm": 0.09761475771665573,
      "learning_rate": 3.5386509892453157e-06,
      "loss": 0.0598,
      "step": 61470
    },
    {
      "epoch": 5.8622360229492186e-05,
      "model_forward_time": 1.7353568077087402,
      "step": 61470
    },
    {
      "dynamics_loss_avg": 0.04349631257355213,
      "epoch": 5.8622360229492186e-05,
      "step": 61470
    },
    {
      "action_loss_avg": 0.0008878874505171552,
      "epoch": 5.8622360229492186e-05,
      "step": 61470
    },
    {
      "epoch": 5.8622360229492186e-05,
      "step": 61470,
      "training_step_time": 8.638262510299683
    },
    {
      "epoch": 5.862331390380859e-05,
      "model_forward_time": 1.7346892356872559,
      "step": 61471
    },
    {
      "epoch": 5.862331390380859e-05,
      "step": 61471,
      "training_step_time": 7.607306241989136
    },
    {
      "epoch": 5.8624267578125e-05,
      "model_forward_time": 1.7324309349060059,
      "step": 61472
    },
    {
      "epoch": 5.8624267578125e-05,
      "step": 61472,
      "training_step_time": 7.59609317779541
    },
    {
      "epoch": 5.8625221252441405e-05,
      "model_forward_time": 1.7295258045196533,
      "step": 61473
    },
    {
      "epoch": 5.8625221252441405e-05,
      "step": 61473,
      "training_step_time": 7.690741777420044
    },
    {
      "epoch": 5.862617492675781e-05,
      "model_forward_time": 1.725501298904419,
      "step": 61474
    },
    {
      "epoch": 5.862617492675781e-05,
      "step": 61474,
      "training_step_time": 8.106460332870483
    },
    {
      "epoch": 5.862712860107422e-05,
      "model_forward_time": 1.726461410522461,
      "step": 61475
    },
    {
      "epoch": 5.862712860107422e-05,
      "step": 61475,
      "training_step_time": 7.759655714035034
    },
    {
      "epoch": 5.8628082275390624e-05,
      "model_forward_time": 1.8798809051513672,
      "step": 61476
    },
    {
      "epoch": 5.8628082275390624e-05,
      "step": 61476,
      "training_step_time": 7.733152627944946
    },
    {
      "epoch": 5.862903594970703e-05,
      "model_forward_time": 1.7631556987762451,
      "step": 61477
    },
    {
      "epoch": 5.862903594970703e-05,
      "step": 61477,
      "training_step_time": 7.80920934677124
    },
    {
      "epoch": 5.862998962402344e-05,
      "model_forward_time": 1.7297406196594238,
      "step": 61478
    },
    {
      "epoch": 5.862998962402344e-05,
      "step": 61478,
      "training_step_time": 8.153029918670654
    },
    {
      "epoch": 5.863094329833984e-05,
      "model_forward_time": 1.7253694534301758,
      "step": 61479
    },
    {
      "epoch": 5.863094329833984e-05,
      "step": 61479,
      "training_step_time": 7.756853342056274
    },
    {
      "epoch": 5.863189697265625e-05,
      "grad_norm": 0.09247856587171555,
      "learning_rate": 3.5370697966255043e-06,
      "loss": 0.0574,
      "step": 61480
    },
    {
      "epoch": 5.863189697265625e-05,
      "model_forward_time": 1.7205045223236084,
      "step": 61480
    },
    {
      "dynamics_loss_avg": 0.06250445879995822,
      "epoch": 5.863189697265625e-05,
      "step": 61480
    },
    {
      "action_loss_avg": 0.001989943711669184,
      "epoch": 5.863189697265625e-05,
      "step": 61480
    },
    {
      "epoch": 5.863189697265625e-05,
      "step": 61480,
      "training_step_time": 7.682717561721802
    },
    {
      "epoch": 5.863285064697266e-05,
      "model_forward_time": 1.7298154830932617,
      "step": 61481
    },
    {
      "epoch": 5.863285064697266e-05,
      "step": 61481,
      "training_step_time": 7.564502477645874
    },
    {
      "epoch": 5.863380432128906e-05,
      "model_forward_time": 1.7091798782348633,
      "step": 61482
    },
    {
      "epoch": 5.863380432128906e-05,
      "step": 61482,
      "training_step_time": 8.699524879455566
    },
    {
      "epoch": 5.8634757995605466e-05,
      "model_forward_time": 1.700167179107666,
      "step": 61483
    },
    {
      "epoch": 5.8634757995605466e-05,
      "step": 61483,
      "training_step_time": 8.133975982666016
    },
    {
      "epoch": 5.863571166992188e-05,
      "model_forward_time": 1.6974196434020996,
      "step": 61484
    },
    {
      "epoch": 5.863571166992188e-05,
      "step": 61484,
      "training_step_time": 8.80785322189331
    },
    {
      "epoch": 5.863666534423828e-05,
      "model_forward_time": 1.7068707942962646,
      "step": 61485
    },
    {
      "epoch": 5.863666534423828e-05,
      "step": 61485,
      "training_step_time": 9.132154941558838
    },
    {
      "epoch": 5.8637619018554685e-05,
      "model_forward_time": 1.7107548713684082,
      "step": 61486
    },
    {
      "epoch": 5.8637619018554685e-05,
      "step": 61486,
      "training_step_time": 8.881488800048828
    },
    {
      "epoch": 5.8638572692871096e-05,
      "model_forward_time": 1.7074344158172607,
      "step": 61487
    },
    {
      "epoch": 5.8638572692871096e-05,
      "step": 61487,
      "training_step_time": 8.468785047531128
    },
    {
      "epoch": 5.86395263671875e-05,
      "model_forward_time": 1.6991429328918457,
      "step": 61488
    },
    {
      "epoch": 5.86395263671875e-05,
      "step": 61488,
      "training_step_time": 7.830759286880493
    },
    {
      "epoch": 5.8640480041503904e-05,
      "model_forward_time": 1.7094073295593262,
      "step": 61489
    },
    {
      "epoch": 5.8640480041503904e-05,
      "step": 61489,
      "training_step_time": 7.662121057510376
    },
    {
      "epoch": 5.8641433715820314e-05,
      "grad_norm": 0.09203900396823883,
      "learning_rate": 3.5354887639895398e-06,
      "loss": 0.0555,
      "step": 61490
    },
    {
      "epoch": 5.8641433715820314e-05,
      "model_forward_time": 1.705801010131836,
      "step": 61490
    },
    {
      "dynamics_loss_avg": 0.05926077757030725,
      "epoch": 5.8641433715820314e-05,
      "step": 61490
    },
    {
      "action_loss_avg": 0.0020330286351963878,
      "epoch": 5.8641433715820314e-05,
      "step": 61490
    },
    {
      "epoch": 5.8641433715820314e-05,
      "step": 61490,
      "training_step_time": 8.081135034561157
    },
    {
      "epoch": 5.864238739013672e-05,
      "model_forward_time": 1.7071599960327148,
      "step": 61491
    },
    {
      "epoch": 5.864238739013672e-05,
      "step": 61491,
      "training_step_time": 8.60669732093811
    },
    {
      "epoch": 5.864334106445312e-05,
      "model_forward_time": 1.7020695209503174,
      "step": 61492
    },
    {
      "epoch": 5.864334106445312e-05,
      "step": 61492,
      "training_step_time": 7.397920608520508
    },
    {
      "epoch": 5.864429473876953e-05,
      "model_forward_time": 1.6994810104370117,
      "step": 61493
    },
    {
      "epoch": 5.864429473876953e-05,
      "step": 61493,
      "training_step_time": 7.749166250228882
    },
    {
      "epoch": 5.864524841308594e-05,
      "model_forward_time": 1.7134103775024414,
      "step": 61494
    },
    {
      "epoch": 5.864524841308594e-05,
      "step": 61494,
      "training_step_time": 7.592860221862793
    },
    {
      "epoch": 5.864620208740234e-05,
      "model_forward_time": 2.2574336528778076,
      "step": 61495
    },
    {
      "epoch": 5.864620208740234e-05,
      "step": 61495,
      "training_step_time": 9.32093858718872
    },
    {
      "epoch": 5.864715576171875e-05,
      "model_forward_time": 1.7049927711486816,
      "step": 61496
    },
    {
      "epoch": 5.864715576171875e-05,
      "step": 61496,
      "training_step_time": 9.239311456680298
    },
    {
      "epoch": 5.8648109436035156e-05,
      "model_forward_time": 1.7038226127624512,
      "step": 61497
    },
    {
      "epoch": 5.8648109436035156e-05,
      "step": 61497,
      "training_step_time": 8.7452552318573
    },
    {
      "epoch": 5.864906311035156e-05,
      "model_forward_time": 1.7072374820709229,
      "step": 61498
    },
    {
      "epoch": 5.864906311035156e-05,
      "step": 61498,
      "training_step_time": 8.532492637634277
    },
    {
      "epoch": 5.865001678466797e-05,
      "model_forward_time": 1.702622413635254,
      "step": 61499
    },
    {
      "epoch": 5.865001678466797e-05,
      "step": 61499,
      "training_step_time": 7.6787707805633545
    },
    {
      "epoch": 5.8650970458984375e-05,
      "grad_norm": 0.12256317585706711,
      "learning_rate": 3.5339078915103214e-06,
      "loss": 0.0593,
      "step": 61500
    },
    {
      "epoch": 5.8650970458984375e-05,
      "model_forward_time": 1.7007410526275635,
      "step": 61500
    },
    {
      "dynamics_loss_avg": 0.05668761096894741,
      "epoch": 5.8650970458984375e-05,
      "step": 61500
    },
    {
      "action_loss_avg": 0.0014977073878981173,
      "epoch": 5.8650970458984375e-05,
      "step": 61500
    },
    {
      "epoch": 5.8650970458984375e-05,
      "step": 61500,
      "training_step_time": 7.613823175430298
    },
    {
      "epoch": 5.865192413330078e-05,
      "model_forward_time": 1.703583002090454,
      "step": 61501
    },
    {
      "epoch": 5.865192413330078e-05,
      "step": 61501,
      "training_step_time": 7.7075276374816895
    },
    {
      "epoch": 5.865287780761719e-05,
      "model_forward_time": 1.6924612522125244,
      "step": 61502
    },
    {
      "epoch": 5.865287780761719e-05,
      "step": 61502,
      "training_step_time": 8.870120763778687
    },
    {
      "epoch": 5.8653831481933594e-05,
      "model_forward_time": 1.7093374729156494,
      "step": 61503
    },
    {
      "epoch": 5.8653831481933594e-05,
      "step": 61503,
      "training_step_time": 8.767274856567383
    },
    {
      "epoch": 5.865478515625e-05,
      "model_forward_time": 2.2623682022094727,
      "step": 61504
    },
    {
      "epoch": 5.865478515625e-05,
      "step": 61504,
      "training_step_time": 9.335599184036255
    },
    {
      "epoch": 5.865573883056641e-05,
      "model_forward_time": 1.7299823760986328,
      "step": 61505
    },
    {
      "epoch": 5.865573883056641e-05,
      "step": 61505,
      "training_step_time": 7.891016721725464
    },
    {
      "epoch": 5.865669250488281e-05,
      "model_forward_time": 1.7276949882507324,
      "step": 61506
    },
    {
      "epoch": 5.865669250488281e-05,
      "step": 61506,
      "training_step_time": 8.861963272094727
    },
    {
      "epoch": 5.865764617919922e-05,
      "model_forward_time": 1.742811679840088,
      "step": 61507
    },
    {
      "epoch": 5.865764617919922e-05,
      "step": 61507,
      "training_step_time": 7.747710227966309
    },
    {
      "epoch": 5.865859985351563e-05,
      "model_forward_time": 1.3457119464874268,
      "step": 61508
    },
    {
      "epoch": 5.865859985351563e-05,
      "step": 61508,
      "training_step_time": 7.667443037033081
    },
    {
      "epoch": 5.865955352783203e-05,
      "model_forward_time": 1.7284317016601562,
      "step": 61509
    },
    {
      "epoch": 5.865955352783203e-05,
      "step": 61509,
      "training_step_time": 7.561917066574097
    },
    {
      "epoch": 5.8660507202148436e-05,
      "grad_norm": 0.08825549483299255,
      "learning_rate": 3.5323271793607328e-06,
      "loss": 0.0595,
      "step": 61510
    },
    {
      "epoch": 5.8660507202148436e-05,
      "model_forward_time": 1.7424874305725098,
      "step": 61510
    },
    {
      "dynamics_loss_avg": 0.04207316860556602,
      "epoch": 5.8660507202148436e-05,
      "step": 61510
    },
    {
      "action_loss_avg": 0.0006782873999327422,
      "epoch": 5.8660507202148436e-05,
      "step": 61510
    },
    {
      "epoch": 5.8660507202148436e-05,
      "step": 61510,
      "training_step_time": 7.671941041946411
    },
    {
      "epoch": 5.8661460876464846e-05,
      "model_forward_time": 1.7688801288604736,
      "step": 61511
    },
    {
      "epoch": 5.8661460876464846e-05,
      "step": 61511,
      "training_step_time": 8.424297332763672
    },
    {
      "epoch": 5.866241455078125e-05,
      "model_forward_time": 1.3769102096557617,
      "step": 61512
    },
    {
      "epoch": 5.866241455078125e-05,
      "step": 61512,
      "training_step_time": 9.411713361740112
    },
    {
      "epoch": 5.8663368225097654e-05,
      "model_forward_time": 2.2604823112487793,
      "step": 61513
    },
    {
      "epoch": 5.8663368225097654e-05,
      "step": 61513,
      "training_step_time": 8.827221155166626
    },
    {
      "epoch": 5.8664321899414065e-05,
      "model_forward_time": 1.728832483291626,
      "step": 61514
    },
    {
      "epoch": 5.8664321899414065e-05,
      "step": 61514,
      "training_step_time": 8.930371761322021
    },
    {
      "epoch": 5.866527557373047e-05,
      "model_forward_time": 1.715355634689331,
      "step": 61515
    },
    {
      "epoch": 5.866527557373047e-05,
      "step": 61515,
      "training_step_time": 7.63951563835144
    },
    {
      "epoch": 5.866622924804687e-05,
      "model_forward_time": 1.7128872871398926,
      "step": 61516
    },
    {
      "epoch": 5.866622924804687e-05,
      "step": 61516,
      "training_step_time": 9.710338592529297
    },
    {
      "epoch": 5.8667182922363284e-05,
      "model_forward_time": 1.718735694885254,
      "step": 61517
    },
    {
      "epoch": 5.8667182922363284e-05,
      "step": 61517,
      "training_step_time": 7.70324969291687
    },
    {
      "epoch": 5.866813659667969e-05,
      "model_forward_time": 1.6909480094909668,
      "step": 61518
    },
    {
      "epoch": 5.866813659667969e-05,
      "step": 61518,
      "training_step_time": 7.647863388061523
    },
    {
      "epoch": 5.866909027099609e-05,
      "model_forward_time": 1.7184381484985352,
      "step": 61519
    },
    {
      "epoch": 5.866909027099609e-05,
      "step": 61519,
      "training_step_time": 7.6481099128723145
    },
    {
      "epoch": 5.86700439453125e-05,
      "grad_norm": 0.12468737363815308,
      "learning_rate": 3.530746627713636e-06,
      "loss": 0.0619,
      "step": 61520
    },
    {
      "epoch": 5.86700439453125e-05,
      "model_forward_time": 1.7287538051605225,
      "step": 61520
    },
    {
      "dynamics_loss_avg": 0.0593375981785357,
      "epoch": 5.86700439453125e-05,
      "step": 61520
    },
    {
      "action_loss_avg": 0.001274635616573505,
      "epoch": 5.86700439453125e-05,
      "step": 61520
    },
    {
      "epoch": 5.86700439453125e-05,
      "step": 61520,
      "training_step_time": 7.642449617385864
    },
    {
      "epoch": 5.867099761962891e-05,
      "model_forward_time": 1.7143583297729492,
      "step": 61521
    },
    {
      "epoch": 5.867099761962891e-05,
      "step": 61521,
      "training_step_time": 10.191648483276367
    },
    {
      "epoch": 5.867195129394531e-05,
      "model_forward_time": 1.741593599319458,
      "step": 61522
    },
    {
      "epoch": 5.867195129394531e-05,
      "step": 61522,
      "training_step_time": 8.268538475036621
    },
    {
      "epoch": 5.867290496826172e-05,
      "model_forward_time": 2.47048020362854,
      "step": 61523
    },
    {
      "epoch": 5.867290496826172e-05,
      "step": 61523,
      "training_step_time": 9.31930661201477
    },
    {
      "epoch": 5.8673858642578126e-05,
      "model_forward_time": 1.7271125316619873,
      "step": 61524
    },
    {
      "epoch": 5.8673858642578126e-05,
      "step": 61524,
      "training_step_time": 7.635550022125244
    },
    {
      "epoch": 5.867481231689453e-05,
      "model_forward_time": 1.7290291786193848,
      "step": 61525
    },
    {
      "epoch": 5.867481231689453e-05,
      "step": 61525,
      "training_step_time": 8.813047409057617
    },
    {
      "epoch": 5.867576599121094e-05,
      "model_forward_time": 1.7339937686920166,
      "step": 61526
    },
    {
      "epoch": 5.867576599121094e-05,
      "step": 61526,
      "training_step_time": 7.7027428150177
    },
    {
      "epoch": 5.8676719665527345e-05,
      "model_forward_time": 1.7226645946502686,
      "step": 61527
    },
    {
      "epoch": 5.8676719665527345e-05,
      "step": 61527,
      "training_step_time": 7.522514820098877
    },
    {
      "epoch": 5.867767333984375e-05,
      "model_forward_time": 1.7368686199188232,
      "step": 61528
    },
    {
      "epoch": 5.867767333984375e-05,
      "step": 61528,
      "training_step_time": 7.645267486572266
    },
    {
      "epoch": 5.867862701416016e-05,
      "model_forward_time": 1.7424304485321045,
      "step": 61529
    },
    {
      "epoch": 5.867862701416016e-05,
      "step": 61529,
      "training_step_time": 8.524510145187378
    },
    {
      "epoch": 5.8679580688476564e-05,
      "grad_norm": 0.10193782299757004,
      "learning_rate": 3.5291662367418805e-06,
      "loss": 0.0608,
      "step": 61530
    },
    {
      "epoch": 5.8679580688476564e-05,
      "model_forward_time": 1.8508758544921875,
      "step": 61530
    },
    {
      "dynamics_loss_avg": 0.04314579926431179,
      "epoch": 5.8679580688476564e-05,
      "step": 61530
    },
    {
      "action_loss_avg": 0.0019174340122845024,
      "epoch": 5.8679580688476564e-05,
      "step": 61530
    },
    {
      "epoch": 5.8679580688476564e-05,
      "step": 61530,
      "training_step_time": 9.599486827850342
    },
    {
      "epoch": 5.868053436279297e-05,
      "model_forward_time": 1.7396228313446045,
      "step": 61531
    },
    {
      "epoch": 5.868053436279297e-05,
      "step": 61531,
      "training_step_time": 8.509878396987915
    },
    {
      "epoch": 5.868148803710937e-05,
      "model_forward_time": 2.8598990440368652,
      "step": 61532
    },
    {
      "epoch": 5.868148803710937e-05,
      "step": 61532,
      "training_step_time": 9.491238355636597
    },
    {
      "epoch": 5.868244171142578e-05,
      "model_forward_time": 1.732180118560791,
      "step": 61533
    },
    {
      "epoch": 5.868244171142578e-05,
      "step": 61533,
      "training_step_time": 9.543495655059814
    },
    {
      "epoch": 5.8683395385742186e-05,
      "model_forward_time": 1.0094847679138184,
      "step": 61534
    },
    {
      "epoch": 5.8683395385742186e-05,
      "step": 61534,
      "training_step_time": 8.61197829246521
    },
    {
      "epoch": 5.868434906005859e-05,
      "model_forward_time": 1.7328016757965088,
      "step": 61535
    },
    {
      "epoch": 5.868434906005859e-05,
      "step": 61535,
      "training_step_time": 7.726176023483276
    },
    {
      "epoch": 5.8685302734375e-05,
      "model_forward_time": 1.7336297035217285,
      "step": 61536
    },
    {
      "epoch": 5.8685302734375e-05,
      "step": 61536,
      "training_step_time": 7.128569602966309
    },
    {
      "epoch": 5.8686256408691405e-05,
      "model_forward_time": 1.7426857948303223,
      "step": 61537
    },
    {
      "epoch": 5.8686256408691405e-05,
      "step": 61537,
      "training_step_time": 7.641890525817871
    },
    {
      "epoch": 5.868721008300781e-05,
      "model_forward_time": 1.7272157669067383,
      "step": 61538
    },
    {
      "epoch": 5.868721008300781e-05,
      "step": 61538,
      "training_step_time": 7.4022743701934814
    },
    {
      "epoch": 5.868816375732422e-05,
      "model_forward_time": 1.8139374256134033,
      "step": 61539
    },
    {
      "epoch": 5.868816375732422e-05,
      "step": 61539,
      "training_step_time": 8.97314977645874
    },
    {
      "epoch": 5.8689117431640624e-05,
      "grad_norm": 0.10336869955062866,
      "learning_rate": 3.527586006618293e-06,
      "loss": 0.0586,
      "step": 61540
    },
    {
      "epoch": 5.8689117431640624e-05,
      "model_forward_time": 1.746436595916748,
      "step": 61540
    },
    {
      "dynamics_loss_avg": 0.06450391858816147,
      "epoch": 5.8689117431640624e-05,
      "step": 61540
    },
    {
      "action_loss_avg": 0.0021005289920140056,
      "epoch": 5.8689117431640624e-05,
      "step": 61540
    },
    {
      "epoch": 5.8689117431640624e-05,
      "step": 61540,
      "training_step_time": 8.384748458862305
    },
    {
      "epoch": 5.869007110595703e-05,
      "model_forward_time": 1.7129170894622803,
      "step": 61541
    },
    {
      "epoch": 5.869007110595703e-05,
      "step": 61541,
      "training_step_time": 9.354512929916382
    },
    {
      "epoch": 5.869102478027344e-05,
      "model_forward_time": 1.733936071395874,
      "step": 61542
    },
    {
      "epoch": 5.869102478027344e-05,
      "step": 61542,
      "training_step_time": 7.735334157943726
    },
    {
      "epoch": 5.869197845458984e-05,
      "model_forward_time": 1.7225713729858398,
      "step": 61543
    },
    {
      "epoch": 5.869197845458984e-05,
      "step": 61543,
      "training_step_time": 8.70091962814331
    },
    {
      "epoch": 5.869293212890625e-05,
      "model_forward_time": 1.7344071865081787,
      "step": 61544
    },
    {
      "epoch": 5.869293212890625e-05,
      "step": 61544,
      "training_step_time": 7.869367599487305
    },
    {
      "epoch": 5.869388580322266e-05,
      "model_forward_time": 1.7230360507965088,
      "step": 61545
    },
    {
      "epoch": 5.869388580322266e-05,
      "step": 61545,
      "training_step_time": 7.746731758117676
    },
    {
      "epoch": 5.869483947753906e-05,
      "model_forward_time": 1.726379632949829,
      "step": 61546
    },
    {
      "epoch": 5.869483947753906e-05,
      "step": 61546,
      "training_step_time": 7.635218620300293
    },
    {
      "epoch": 5.8695793151855466e-05,
      "model_forward_time": 1.7288568019866943,
      "step": 61547
    },
    {
      "epoch": 5.8695793151855466e-05,
      "step": 61547,
      "training_step_time": 7.7264649868011475
    },
    {
      "epoch": 5.869674682617188e-05,
      "model_forward_time": 1.7340202331542969,
      "step": 61548
    },
    {
      "epoch": 5.869674682617188e-05,
      "step": 61548,
      "training_step_time": 7.719747543334961
    },
    {
      "epoch": 5.869770050048828e-05,
      "model_forward_time": 1.7192537784576416,
      "step": 61549
    },
    {
      "epoch": 5.869770050048828e-05,
      "step": 61549,
      "training_step_time": 8.868247747421265
    },
    {
      "epoch": 5.8698654174804685e-05,
      "grad_norm": 0.10282210260629654,
      "learning_rate": 3.526005937515687e-06,
      "loss": 0.0587,
      "step": 61550
    },
    {
      "epoch": 5.8698654174804685e-05,
      "model_forward_time": 1.728365182876587,
      "step": 61550
    },
    {
      "dynamics_loss_avg": 0.055680517479777335,
      "epoch": 5.8698654174804685e-05,
      "step": 61550
    },
    {
      "action_loss_avg": 0.0013502437330316752,
      "epoch": 5.8698654174804685e-05,
      "step": 61550
    },
    {
      "epoch": 5.8698654174804685e-05,
      "step": 61550,
      "training_step_time": 7.827523469924927
    },
    {
      "epoch": 5.8699607849121096e-05,
      "model_forward_time": 2.8760592937469482,
      "step": 61551
    },
    {
      "epoch": 5.8699607849121096e-05,
      "step": 61551,
      "training_step_time": 7.794053554534912
    },
    {
      "epoch": 5.87005615234375e-05,
      "model_forward_time": 1.7156009674072266,
      "step": 61552
    },
    {
      "epoch": 5.87005615234375e-05,
      "step": 61552,
      "training_step_time": 7.673492193222046
    },
    {
      "epoch": 5.8701515197753904e-05,
      "model_forward_time": 1.7325701713562012,
      "step": 61553
    },
    {
      "epoch": 5.8701515197753904e-05,
      "step": 61553,
      "training_step_time": 8.39082670211792
    },
    {
      "epoch": 5.8702468872070314e-05,
      "model_forward_time": 1.7124252319335938,
      "step": 61554
    },
    {
      "epoch": 5.8702468872070314e-05,
      "step": 61554,
      "training_step_time": 7.735507965087891
    },
    {
      "epoch": 5.870342254638672e-05,
      "model_forward_time": 1.7256338596343994,
      "step": 61555
    },
    {
      "epoch": 5.870342254638672e-05,
      "step": 61555,
      "training_step_time": 7.609212636947632
    },
    {
      "epoch": 5.870437622070312e-05,
      "model_forward_time": 1.7309365272521973,
      "step": 61556
    },
    {
      "epoch": 5.870437622070312e-05,
      "step": 61556,
      "training_step_time": 7.7254462242126465
    },
    {
      "epoch": 5.870532989501953e-05,
      "model_forward_time": 1.7196176052093506,
      "step": 61557
    },
    {
      "epoch": 5.870532989501953e-05,
      "step": 61557,
      "training_step_time": 8.160995960235596
    },
    {
      "epoch": 5.870628356933594e-05,
      "model_forward_time": 1.7396667003631592,
      "step": 61558
    },
    {
      "epoch": 5.870628356933594e-05,
      "step": 61558,
      "training_step_time": 9.47165846824646
    },
    {
      "epoch": 5.870723724365234e-05,
      "model_forward_time": 1.7168595790863037,
      "step": 61559
    },
    {
      "epoch": 5.870723724365234e-05,
      "step": 61559,
      "training_step_time": 7.736849069595337
    },
    {
      "epoch": 5.870819091796875e-05,
      "grad_norm": 0.08593675494194031,
      "learning_rate": 3.5244260296068554e-06,
      "loss": 0.0591,
      "step": 61560
    },
    {
      "epoch": 5.870819091796875e-05,
      "model_forward_time": 2.920193672180176,
      "step": 61560
    },
    {
      "dynamics_loss_avg": 0.06035015322268009,
      "epoch": 5.870819091796875e-05,
      "step": 61560
    },
    {
      "action_loss_avg": 0.0021023665816755966,
      "epoch": 5.870819091796875e-05,
      "step": 61560
    },
    {
      "epoch": 5.870819091796875e-05,
      "step": 61560,
      "training_step_time": 7.858412742614746
    },
    {
      "epoch": 5.8709144592285156e-05,
      "model_forward_time": 1.7274103164672852,
      "step": 61561
    },
    {
      "epoch": 5.8709144592285156e-05,
      "step": 61561,
      "training_step_time": 8.525819778442383
    },
    {
      "epoch": 5.871009826660156e-05,
      "model_forward_time": 1.7259886264801025,
      "step": 61562
    },
    {
      "epoch": 5.871009826660156e-05,
      "step": 61562,
      "training_step_time": 7.813932418823242
    },
    {
      "epoch": 5.871105194091797e-05,
      "model_forward_time": 1.729792594909668,
      "step": 61563
    },
    {
      "epoch": 5.871105194091797e-05,
      "step": 61563,
      "training_step_time": 7.802196502685547
    },
    {
      "epoch": 5.8712005615234375e-05,
      "model_forward_time": 1.720388412475586,
      "step": 61564
    },
    {
      "epoch": 5.8712005615234375e-05,
      "step": 61564,
      "training_step_time": 7.656969308853149
    },
    {
      "epoch": 5.871295928955078e-05,
      "model_forward_time": 1.720372200012207,
      "step": 61565
    },
    {
      "epoch": 5.871295928955078e-05,
      "step": 61565,
      "training_step_time": 7.7565271854400635
    },
    {
      "epoch": 5.871391296386719e-05,
      "model_forward_time": 1.7293567657470703,
      "step": 61566
    },
    {
      "epoch": 5.871391296386719e-05,
      "step": 61566,
      "training_step_time": 8.234034776687622
    },
    {
      "epoch": 5.8714866638183594e-05,
      "model_forward_time": 1.731215476989746,
      "step": 61567
    },
    {
      "epoch": 5.8714866638183594e-05,
      "step": 61567,
      "training_step_time": 8.525862216949463
    },
    {
      "epoch": 5.87158203125e-05,
      "model_forward_time": 1.721463680267334,
      "step": 61568
    },
    {
      "epoch": 5.87158203125e-05,
      "step": 61568,
      "training_step_time": 7.761744022369385
    },
    {
      "epoch": 5.871677398681641e-05,
      "model_forward_time": 2.8850719928741455,
      "step": 61569
    },
    {
      "epoch": 5.871677398681641e-05,
      "step": 61569,
      "training_step_time": 8.927223443984985
    },
    {
      "epoch": 5.871772766113281e-05,
      "grad_norm": 0.1081656664609909,
      "learning_rate": 3.522846283064575e-06,
      "loss": 0.0524,
      "step": 61570
    },
    {
      "epoch": 5.871772766113281e-05,
      "model_forward_time": 1.7190306186676025,
      "step": 61570
    },
    {
      "dynamics_loss_avg": 0.05068236142396927,
      "epoch": 5.871772766113281e-05,
      "step": 61570
    },
    {
      "action_loss_avg": 0.0022567359381355347,
      "epoch": 5.871772766113281e-05,
      "step": 61570
    },
    {
      "epoch": 5.871772766113281e-05,
      "step": 61570,
      "training_step_time": 8.410096406936646
    },
    {
      "epoch": 5.871868133544922e-05,
      "model_forward_time": 1.730931282043457,
      "step": 61571
    },
    {
      "epoch": 5.871868133544922e-05,
      "step": 61571,
      "training_step_time": 8.38347315788269
    },
    {
      "epoch": 5.871963500976563e-05,
      "model_forward_time": 1.716440200805664,
      "step": 61572
    },
    {
      "epoch": 5.871963500976563e-05,
      "step": 61572,
      "training_step_time": 7.765127658843994
    },
    {
      "epoch": 5.872058868408203e-05,
      "model_forward_time": 1.7260799407958984,
      "step": 61573
    },
    {
      "epoch": 5.872058868408203e-05,
      "step": 61573,
      "training_step_time": 8.590308666229248
    },
    {
      "epoch": 5.8721542358398436e-05,
      "model_forward_time": 1.720369815826416,
      "step": 61574
    },
    {
      "epoch": 5.8721542358398436e-05,
      "step": 61574,
      "training_step_time": 7.697918653488159
    },
    {
      "epoch": 5.8722496032714846e-05,
      "model_forward_time": 1.7227199077606201,
      "step": 61575
    },
    {
      "epoch": 5.8722496032714846e-05,
      "step": 61575,
      "training_step_time": 8.129531621932983
    },
    {
      "epoch": 5.872344970703125e-05,
      "model_forward_time": 1.7355146408081055,
      "step": 61576
    },
    {
      "epoch": 5.872344970703125e-05,
      "step": 61576,
      "training_step_time": 7.747345209121704
    },
    {
      "epoch": 5.8724403381347655e-05,
      "model_forward_time": 1.723235845565796,
      "step": 61577
    },
    {
      "epoch": 5.8724403381347655e-05,
      "step": 61577,
      "training_step_time": 9.653171300888062
    },
    {
      "epoch": 5.8725357055664065e-05,
      "model_forward_time": 2.935347557067871,
      "step": 61578
    },
    {
      "epoch": 5.8725357055664065e-05,
      "step": 61578,
      "training_step_time": 8.890448808670044
    },
    {
      "epoch": 5.872631072998047e-05,
      "model_forward_time": 1.7227325439453125,
      "step": 61579
    },
    {
      "epoch": 5.872631072998047e-05,
      "step": 61579,
      "training_step_time": 7.747372150421143
    },
    {
      "epoch": 5.872726440429687e-05,
      "grad_norm": 0.10690339654684067,
      "learning_rate": 3.521266698061604e-06,
      "loss": 0.0588,
      "step": 61580
    },
    {
      "epoch": 5.872726440429687e-05,
      "model_forward_time": 1.7200157642364502,
      "step": 61580
    },
    {
      "dynamics_loss_avg": 0.0710260147228837,
      "epoch": 5.872726440429687e-05,
      "step": 61580
    },
    {
      "action_loss_avg": 0.0013629195134853944,
      "epoch": 5.872726440429687e-05,
      "step": 61580
    },
    {
      "epoch": 5.872726440429687e-05,
      "step": 61580,
      "training_step_time": 7.778735399246216
    },
    {
      "epoch": 5.8728218078613284e-05,
      "model_forward_time": 1.7292664051055908,
      "step": 61581
    },
    {
      "epoch": 5.8728218078613284e-05,
      "step": 61581,
      "training_step_time": 7.759109973907471
    },
    {
      "epoch": 5.872917175292969e-05,
      "model_forward_time": 1.7207801342010498,
      "step": 61582
    },
    {
      "epoch": 5.872917175292969e-05,
      "step": 61582,
      "training_step_time": 7.483285665512085
    },
    {
      "epoch": 5.873012542724609e-05,
      "model_forward_time": 1.7166879177093506,
      "step": 61583
    },
    {
      "epoch": 5.873012542724609e-05,
      "step": 61583,
      "training_step_time": 8.401975154876709
    },
    {
      "epoch": 5.87310791015625e-05,
      "model_forward_time": 1.7288744449615479,
      "step": 61584
    },
    {
      "epoch": 5.87310791015625e-05,
      "step": 61584,
      "training_step_time": 8.013662338256836
    },
    {
      "epoch": 5.873203277587891e-05,
      "model_forward_time": 1.7140095233917236,
      "step": 61585
    },
    {
      "epoch": 5.873203277587891e-05,
      "step": 61585,
      "training_step_time": 7.766551494598389
    },
    {
      "epoch": 5.873298645019531e-05,
      "model_forward_time": 1.7715015411376953,
      "step": 61586
    },
    {
      "epoch": 5.873298645019531e-05,
      "step": 61586,
      "training_step_time": 7.757466554641724
    },
    {
      "epoch": 5.873394012451172e-05,
      "model_forward_time": 1.0642099380493164,
      "step": 61587
    },
    {
      "epoch": 5.873394012451172e-05,
      "step": 61587,
      "training_step_time": 7.823858022689819
    },
    {
      "epoch": 5.8734893798828126e-05,
      "model_forward_time": 2.312134027481079,
      "step": 61588
    },
    {
      "epoch": 5.8734893798828126e-05,
      "step": 61588,
      "training_step_time": 8.469220161437988
    },
    {
      "epoch": 5.873584747314453e-05,
      "model_forward_time": 1.7704107761383057,
      "step": 61589
    },
    {
      "epoch": 5.873584747314453e-05,
      "step": 61589,
      "training_step_time": 7.737701654434204
    },
    {
      "epoch": 5.873680114746094e-05,
      "grad_norm": 0.10937817394733429,
      "learning_rate": 3.519687274770681e-06,
      "loss": 0.0561,
      "step": 61590
    },
    {
      "epoch": 5.873680114746094e-05,
      "model_forward_time": 1.7597997188568115,
      "step": 61590
    },
    {
      "dynamics_loss_avg": 0.05588987711817026,
      "epoch": 5.873680114746094e-05,
      "step": 61590
    },
    {
      "action_loss_avg": 0.0021436119626741857,
      "epoch": 5.873680114746094e-05,
      "step": 61590
    },
    {
      "epoch": 5.873680114746094e-05,
      "step": 61590,
      "training_step_time": 7.759289741516113
    },
    {
      "epoch": 5.8737754821777345e-05,
      "model_forward_time": 1.7564351558685303,
      "step": 61591
    },
    {
      "epoch": 5.8737754821777345e-05,
      "step": 61591,
      "training_step_time": 7.7953574657440186
    },
    {
      "epoch": 5.873870849609375e-05,
      "model_forward_time": 1.7209208011627197,
      "step": 61592
    },
    {
      "epoch": 5.873870849609375e-05,
      "step": 61592,
      "training_step_time": 7.663041591644287
    },
    {
      "epoch": 5.873966217041016e-05,
      "model_forward_time": 1.7253518104553223,
      "step": 61593
    },
    {
      "epoch": 5.873966217041016e-05,
      "step": 61593,
      "training_step_time": 7.652813196182251
    },
    {
      "epoch": 5.8740615844726564e-05,
      "model_forward_time": 1.716238260269165,
      "step": 61594
    },
    {
      "epoch": 5.8740615844726564e-05,
      "step": 61594,
      "training_step_time": 7.711331129074097
    },
    {
      "epoch": 5.874156951904297e-05,
      "model_forward_time": 1.70815110206604,
      "step": 61595
    },
    {
      "epoch": 5.874156951904297e-05,
      "step": 61595,
      "training_step_time": 7.751795291900635
    },
    {
      "epoch": 5.874252319335937e-05,
      "model_forward_time": 1.7416713237762451,
      "step": 61596
    },
    {
      "epoch": 5.874252319335937e-05,
      "step": 61596,
      "training_step_time": 7.719432592391968
    },
    {
      "epoch": 5.874347686767578e-05,
      "model_forward_time": 1.8841307163238525,
      "step": 61597
    },
    {
      "epoch": 5.874347686767578e-05,
      "step": 61597,
      "training_step_time": 8.282482385635376
    },
    {
      "epoch": 5.8744430541992187e-05,
      "model_forward_time": 1.7257006168365479,
      "step": 61598
    },
    {
      "epoch": 5.8744430541992187e-05,
      "step": 61598,
      "training_step_time": 8.37466311454773
    },
    {
      "epoch": 5.874538421630859e-05,
      "model_forward_time": 1.7215962409973145,
      "step": 61599
    },
    {
      "epoch": 5.874538421630859e-05,
      "step": 61599,
      "training_step_time": 8.753283500671387
    },
    {
      "epoch": 5.8746337890625e-05,
      "grad_norm": 0.08575163036584854,
      "learning_rate": 3.518108013364535e-06,
      "loss": 0.0541,
      "step": 61600
    },
    {
      "epoch": 5.8746337890625e-05,
      "model_forward_time": 1.7160191535949707,
      "step": 61600
    },
    {
      "dynamics_loss_avg": 0.05942355021834374,
      "epoch": 5.8746337890625e-05,
      "step": 61600
    },
    {
      "action_loss_avg": 0.0024748300085775552,
      "epoch": 5.8746337890625e-05,
      "step": 61600
    },
    {
      "epoch": 5.8746337890625e-05,
      "step": 61600,
      "training_step_time": 7.7147767543792725
    },
    {
      "epoch": 5.8747291564941405e-05,
      "model_forward_time": 1.7318546772003174,
      "step": 61601
    },
    {
      "epoch": 5.8747291564941405e-05,
      "step": 61601,
      "training_step_time": 8.164886951446533
    },
    {
      "epoch": 5.874824523925781e-05,
      "model_forward_time": 1.710118293762207,
      "step": 61602
    },
    {
      "epoch": 5.874824523925781e-05,
      "step": 61602,
      "training_step_time": 7.798252582550049
    },
    {
      "epoch": 5.874919891357422e-05,
      "model_forward_time": 1.7297511100769043,
      "step": 61603
    },
    {
      "epoch": 5.874919891357422e-05,
      "step": 61603,
      "training_step_time": 8.626847743988037
    },
    {
      "epoch": 5.8750152587890624e-05,
      "model_forward_time": 1.7166669368743896,
      "step": 61604
    },
    {
      "epoch": 5.8750152587890624e-05,
      "step": 61604,
      "training_step_time": 10.698205471038818
    },
    {
      "epoch": 5.875110626220703e-05,
      "model_forward_time": 1.7110342979431152,
      "step": 61605
    },
    {
      "epoch": 5.875110626220703e-05,
      "step": 61605,
      "training_step_time": 9.847431659698486
    },
    {
      "epoch": 5.875205993652344e-05,
      "model_forward_time": 2.304097890853882,
      "step": 61606
    },
    {
      "epoch": 5.875205993652344e-05,
      "step": 61606,
      "training_step_time": 9.59591293334961
    },
    {
      "epoch": 5.875301361083984e-05,
      "model_forward_time": 1.7321405410766602,
      "step": 61607
    },
    {
      "epoch": 5.875301361083984e-05,
      "step": 61607,
      "training_step_time": 8.801947593688965
    },
    {
      "epoch": 5.875396728515625e-05,
      "model_forward_time": 1.731442928314209,
      "step": 61608
    },
    {
      "epoch": 5.875396728515625e-05,
      "step": 61608,
      "training_step_time": 9.459211587905884
    },
    {
      "epoch": 5.875492095947266e-05,
      "model_forward_time": 1.7216928005218506,
      "step": 61609
    },
    {
      "epoch": 5.875492095947266e-05,
      "step": 61609,
      "training_step_time": 7.826356887817383
    },
    {
      "epoch": 5.875587463378906e-05,
      "grad_norm": 0.10523516684770584,
      "learning_rate": 3.516528914015868e-06,
      "loss": 0.0536,
      "step": 61610
    },
    {
      "epoch": 5.875587463378906e-05,
      "model_forward_time": 1.7288777828216553,
      "step": 61610
    },
    {
      "dynamics_loss_avg": 0.054177391529083255,
      "epoch": 5.875587463378906e-05,
      "step": 61610
    },
    {
      "action_loss_avg": 0.002349566057091579,
      "epoch": 5.875587463378906e-05,
      "step": 61610
    },
    {
      "epoch": 5.875587463378906e-05,
      "step": 61610,
      "training_step_time": 7.708248138427734
    },
    {
      "epoch": 5.8756828308105466e-05,
      "model_forward_time": 1.7093791961669922,
      "step": 61611
    },
    {
      "epoch": 5.8756828308105466e-05,
      "step": 61611,
      "training_step_time": 7.821832656860352
    },
    {
      "epoch": 5.875778198242188e-05,
      "model_forward_time": 1.7122392654418945,
      "step": 61612
    },
    {
      "epoch": 5.875778198242188e-05,
      "step": 61612,
      "training_step_time": 8.45481276512146
    },
    {
      "epoch": 5.875873565673828e-05,
      "model_forward_time": 1.712188720703125,
      "step": 61613
    },
    {
      "epoch": 5.875873565673828e-05,
      "step": 61613,
      "training_step_time": 7.834362030029297
    },
    {
      "epoch": 5.8759689331054685e-05,
      "model_forward_time": 1.7140514850616455,
      "step": 61614
    },
    {
      "epoch": 5.8759689331054685e-05,
      "step": 61614,
      "training_step_time": 7.977218389511108
    },
    {
      "epoch": 5.8760643005371096e-05,
      "model_forward_time": 2.3080172538757324,
      "step": 61615
    },
    {
      "epoch": 5.8760643005371096e-05,
      "step": 61615,
      "training_step_time": 8.815698146820068
    },
    {
      "epoch": 5.87615966796875e-05,
      "model_forward_time": 1.7189996242523193,
      "step": 61616
    },
    {
      "epoch": 5.87615966796875e-05,
      "step": 61616,
      "training_step_time": 7.862941741943359
    },
    {
      "epoch": 5.8762550354003904e-05,
      "model_forward_time": 1.719031810760498,
      "step": 61617
    },
    {
      "epoch": 5.8762550354003904e-05,
      "step": 61617,
      "training_step_time": 7.852524518966675
    },
    {
      "epoch": 5.8763504028320315e-05,
      "model_forward_time": 1.718106985092163,
      "step": 61618
    },
    {
      "epoch": 5.8763504028320315e-05,
      "step": 61618,
      "training_step_time": 7.767524242401123
    },
    {
      "epoch": 5.876445770263672e-05,
      "model_forward_time": 1.7295055389404297,
      "step": 61619
    },
    {
      "epoch": 5.876445770263672e-05,
      "step": 61619,
      "training_step_time": 7.796350479125977
    },
    {
      "epoch": 5.876541137695312e-05,
      "grad_norm": 0.09504920244216919,
      "learning_rate": 3.514949976897369e-06,
      "loss": 0.0533,
      "step": 61620
    },
    {
      "epoch": 5.876541137695312e-05,
      "model_forward_time": 1.7227354049682617,
      "step": 61620
    },
    {
      "dynamics_loss_avg": 0.04947839118540287,
      "epoch": 5.876541137695312e-05,
      "step": 61620
    },
    {
      "action_loss_avg": 0.0013519805652322248,
      "epoch": 5.876541137695312e-05,
      "step": 61620
    },
    {
      "epoch": 5.876541137695312e-05,
      "step": 61620,
      "training_step_time": 7.78050971031189
    },
    {
      "epoch": 5.876636505126953e-05,
      "model_forward_time": 1.7333307266235352,
      "step": 61621
    },
    {
      "epoch": 5.876636505126953e-05,
      "step": 61621,
      "training_step_time": 8.67848253250122
    },
    {
      "epoch": 5.876731872558594e-05,
      "model_forward_time": 1.7116267681121826,
      "step": 61622
    },
    {
      "epoch": 5.876731872558594e-05,
      "step": 61622,
      "training_step_time": 10.320897579193115
    },
    {
      "epoch": 5.876827239990234e-05,
      "model_forward_time": 1.711841344833374,
      "step": 61623
    },
    {
      "epoch": 5.876827239990234e-05,
      "step": 61623,
      "training_step_time": 8.430996894836426
    },
    {
      "epoch": 5.876922607421875e-05,
      "model_forward_time": 1.7665507793426514,
      "step": 61624
    },
    {
      "epoch": 5.876922607421875e-05,
      "step": 61624,
      "training_step_time": 9.526540279388428
    },
    {
      "epoch": 5.8770179748535156e-05,
      "model_forward_time": 1.7101998329162598,
      "step": 61625
    },
    {
      "epoch": 5.8770179748535156e-05,
      "step": 61625,
      "training_step_time": 7.7625110149383545
    },
    {
      "epoch": 5.877113342285156e-05,
      "model_forward_time": 1.7212893962860107,
      "step": 61626
    },
    {
      "epoch": 5.877113342285156e-05,
      "step": 61626,
      "training_step_time": 7.7373082637786865
    },
    {
      "epoch": 5.877208709716797e-05,
      "model_forward_time": 1.7260031700134277,
      "step": 61627
    },
    {
      "epoch": 5.877208709716797e-05,
      "step": 61627,
      "training_step_time": 8.67197561264038
    },
    {
      "epoch": 5.8773040771484375e-05,
      "model_forward_time": 1.7136614322662354,
      "step": 61628
    },
    {
      "epoch": 5.8773040771484375e-05,
      "step": 61628,
      "training_step_time": 7.738156795501709
    },
    {
      "epoch": 5.877399444580078e-05,
      "model_forward_time": 1.7249233722686768,
      "step": 61629
    },
    {
      "epoch": 5.877399444580078e-05,
      "step": 61629,
      "training_step_time": 7.684130430221558
    },
    {
      "epoch": 5.877494812011719e-05,
      "grad_norm": 0.09463417530059814,
      "learning_rate": 3.513371202181708e-06,
      "loss": 0.0531,
      "step": 61630
    },
    {
      "epoch": 5.877494812011719e-05,
      "model_forward_time": 1.7253549098968506,
      "step": 61630
    },
    {
      "dynamics_loss_avg": 0.05634052623063326,
      "epoch": 5.877494812011719e-05,
      "step": 61630
    },
    {
      "action_loss_avg": 0.0017531229299493135,
      "epoch": 5.877494812011719e-05,
      "step": 61630
    },
    {
      "epoch": 5.877494812011719e-05,
      "step": 61630,
      "training_step_time": 7.749543190002441
    },
    {
      "epoch": 5.8775901794433594e-05,
      "model_forward_time": 1.706251621246338,
      "step": 61631
    },
    {
      "epoch": 5.8775901794433594e-05,
      "step": 61631,
      "training_step_time": 7.721135854721069
    },
    {
      "epoch": 5.877685546875e-05,
      "model_forward_time": 1.7137889862060547,
      "step": 61632
    },
    {
      "epoch": 5.877685546875e-05,
      "step": 61632,
      "training_step_time": 8.3433518409729
    },
    {
      "epoch": 5.877780914306641e-05,
      "model_forward_time": 1.7627806663513184,
      "step": 61633
    },
    {
      "epoch": 5.877780914306641e-05,
      "step": 61633,
      "training_step_time": 8.018897294998169
    },
    {
      "epoch": 5.877876281738281e-05,
      "model_forward_time": 2.2656123638153076,
      "step": 61634
    },
    {
      "epoch": 5.877876281738281e-05,
      "step": 61634,
      "training_step_time": 7.72393274307251
    },
    {
      "epoch": 5.877971649169922e-05,
      "model_forward_time": 1.7141988277435303,
      "step": 61635
    },
    {
      "epoch": 5.877971649169922e-05,
      "step": 61635,
      "training_step_time": 8.334368467330933
    },
    {
      "epoch": 5.878067016601563e-05,
      "model_forward_time": 1.74320387840271,
      "step": 61636
    },
    {
      "epoch": 5.878067016601563e-05,
      "step": 61636,
      "training_step_time": 8.615922212600708
    },
    {
      "epoch": 5.878162384033203e-05,
      "model_forward_time": 1.7321207523345947,
      "step": 61637
    },
    {
      "epoch": 5.878162384033203e-05,
      "step": 61637,
      "training_step_time": 8.404464960098267
    },
    {
      "epoch": 5.8782577514648436e-05,
      "model_forward_time": 1.7257928848266602,
      "step": 61638
    },
    {
      "epoch": 5.8782577514648436e-05,
      "step": 61638,
      "training_step_time": 8.655619859695435
    },
    {
      "epoch": 5.8783531188964847e-05,
      "model_forward_time": 1.711390733718872,
      "step": 61639
    },
    {
      "epoch": 5.8783531188964847e-05,
      "step": 61639,
      "training_step_time": 7.464804649353027
    },
    {
      "epoch": 5.878448486328125e-05,
      "grad_norm": 0.10141357779502869,
      "learning_rate": 3.511792590041537e-06,
      "loss": 0.0538,
      "step": 61640
    },
    {
      "epoch": 5.878448486328125e-05,
      "model_forward_time": 1.7339322566986084,
      "step": 61640
    },
    {
      "dynamics_loss_avg": 0.048218070343136785,
      "epoch": 5.878448486328125e-05,
      "step": 61640
    },
    {
      "action_loss_avg": 0.0013017159973969683,
      "epoch": 5.878448486328125e-05,
      "step": 61640
    },
    {
      "epoch": 5.878448486328125e-05,
      "step": 61640,
      "training_step_time": 9.314789295196533
    },
    {
      "epoch": 5.8785438537597655e-05,
      "model_forward_time": 1.7038764953613281,
      "step": 61641
    },
    {
      "epoch": 5.8785438537597655e-05,
      "step": 61641,
      "training_step_time": 8.632756471633911
    },
    {
      "epoch": 5.8786392211914065e-05,
      "model_forward_time": 1.7322142124176025,
      "step": 61642
    },
    {
      "epoch": 5.8786392211914065e-05,
      "step": 61642,
      "training_step_time": 7.935808420181274
    },
    {
      "epoch": 5.878734588623047e-05,
      "model_forward_time": 1.9611289501190186,
      "step": 61643
    },
    {
      "epoch": 5.878734588623047e-05,
      "step": 61643,
      "training_step_time": 7.792528390884399
    },
    {
      "epoch": 5.8788299560546873e-05,
      "model_forward_time": 1.6984474658966064,
      "step": 61644
    },
    {
      "epoch": 5.8788299560546873e-05,
      "step": 61644,
      "training_step_time": 7.767303705215454
    },
    {
      "epoch": 5.8789253234863284e-05,
      "model_forward_time": 1.7190093994140625,
      "step": 61645
    },
    {
      "epoch": 5.8789253234863284e-05,
      "step": 61645,
      "training_step_time": 8.749849319458008
    },
    {
      "epoch": 5.879020690917969e-05,
      "model_forward_time": 1.7125153541564941,
      "step": 61646
    },
    {
      "epoch": 5.879020690917969e-05,
      "step": 61646,
      "training_step_time": 7.7967000007629395
    },
    {
      "epoch": 5.879116058349609e-05,
      "model_forward_time": 1.3634779453277588,
      "step": 61647
    },
    {
      "epoch": 5.879116058349609e-05,
      "step": 61647,
      "training_step_time": 8.737369537353516
    },
    {
      "epoch": 5.87921142578125e-05,
      "model_forward_time": 1.738067865371704,
      "step": 61648
    },
    {
      "epoch": 5.87921142578125e-05,
      "step": 61648,
      "training_step_time": 7.751374959945679
    },
    {
      "epoch": 5.879306793212891e-05,
      "model_forward_time": 1.735243558883667,
      "step": 61649
    },
    {
      "epoch": 5.879306793212891e-05,
      "step": 61649,
      "training_step_time": 8.778106451034546
    },
    {
      "epoch": 5.879402160644531e-05,
      "grad_norm": 0.11210024356842041,
      "learning_rate": 3.5102141406494883e-06,
      "loss": 0.056,
      "step": 61650
    },
    {
      "epoch": 5.879402160644531e-05,
      "model_forward_time": 1.7402832508087158,
      "step": 61650
    },
    {
      "dynamics_loss_avg": 0.048506351001560685,
      "epoch": 5.879402160644531e-05,
      "step": 61650
    },
    {
      "action_loss_avg": 0.0031253504828782754,
      "epoch": 5.879402160644531e-05,
      "step": 61650
    },
    {
      "epoch": 5.879402160644531e-05,
      "step": 61650,
      "training_step_time": 8.30018162727356
    },
    {
      "epoch": 5.879497528076172e-05,
      "model_forward_time": 1.7371406555175781,
      "step": 61651
    },
    {
      "epoch": 5.879497528076172e-05,
      "step": 61651,
      "training_step_time": 7.860209703445435
    },
    {
      "epoch": 5.8795928955078126e-05,
      "model_forward_time": 2.323765754699707,
      "step": 61652
    },
    {
      "epoch": 5.8795928955078126e-05,
      "step": 61652,
      "training_step_time": 8.337955713272095
    },
    {
      "epoch": 5.879688262939453e-05,
      "model_forward_time": 1.7361078262329102,
      "step": 61653
    },
    {
      "epoch": 5.879688262939453e-05,
      "step": 61653,
      "training_step_time": 7.729972839355469
    },
    {
      "epoch": 5.879783630371094e-05,
      "model_forward_time": 1.7403173446655273,
      "step": 61654
    },
    {
      "epoch": 5.879783630371094e-05,
      "step": 61654,
      "training_step_time": 7.7608373165130615
    },
    {
      "epoch": 5.8798789978027345e-05,
      "model_forward_time": 1.740539312362671,
      "step": 61655
    },
    {
      "epoch": 5.8798789978027345e-05,
      "step": 61655,
      "training_step_time": 7.695093393325806
    },
    {
      "epoch": 5.879974365234375e-05,
      "model_forward_time": 1.7386257648468018,
      "step": 61656
    },
    {
      "epoch": 5.879974365234375e-05,
      "step": 61656,
      "training_step_time": 8.517319440841675
    },
    {
      "epoch": 5.880069732666015e-05,
      "model_forward_time": 1.7456755638122559,
      "step": 61657
    },
    {
      "epoch": 5.880069732666015e-05,
      "step": 61657,
      "training_step_time": 7.694957971572876
    },
    {
      "epoch": 5.8801651000976564e-05,
      "model_forward_time": 1.3713045120239258,
      "step": 61658
    },
    {
      "epoch": 5.8801651000976564e-05,
      "step": 61658,
      "training_step_time": 7.72067403793335
    },
    {
      "epoch": 5.880260467529297e-05,
      "model_forward_time": 1.7199456691741943,
      "step": 61659
    },
    {
      "epoch": 5.880260467529297e-05,
      "step": 61659,
      "training_step_time": 8.543133974075317
    },
    {
      "epoch": 5.880355834960937e-05,
      "grad_norm": 0.14272058010101318,
      "learning_rate": 3.508635854178184e-06,
      "loss": 0.0552,
      "step": 61660
    },
    {
      "epoch": 5.880355834960937e-05,
      "model_forward_time": 1.7138853073120117,
      "step": 61660
    },
    {
      "dynamics_loss_avg": 0.04877273878082633,
      "epoch": 5.880355834960937e-05,
      "step": 61660
    },
    {
      "action_loss_avg": 0.0016013544285669924,
      "epoch": 5.880355834960937e-05,
      "step": 61660
    },
    {
      "epoch": 5.880355834960937e-05,
      "step": 61660,
      "training_step_time": 8.790368556976318
    },
    {
      "epoch": 5.880451202392578e-05,
      "model_forward_time": 2.414860486984253,
      "step": 61661
    },
    {
      "epoch": 5.880451202392578e-05,
      "step": 61661,
      "training_step_time": 9.989407062530518
    },
    {
      "epoch": 5.8805465698242187e-05,
      "model_forward_time": 1.721083164215088,
      "step": 61662
    },
    {
      "epoch": 5.8805465698242187e-05,
      "step": 61662,
      "training_step_time": 7.626962661743164
    },
    {
      "epoch": 5.880641937255859e-05,
      "model_forward_time": 1.728058099746704,
      "step": 61663
    },
    {
      "epoch": 5.880641937255859e-05,
      "step": 61663,
      "training_step_time": 8.64827036857605
    },
    {
      "epoch": 5.8807373046875e-05,
      "model_forward_time": 1.718419075012207,
      "step": 61664
    },
    {
      "epoch": 5.8807373046875e-05,
      "step": 61664,
      "training_step_time": 8.33933711051941
    },
    {
      "epoch": 5.8808326721191405e-05,
      "model_forward_time": 1.7161693572998047,
      "step": 61665
    },
    {
      "epoch": 5.8808326721191405e-05,
      "step": 61665,
      "training_step_time": 7.685824394226074
    },
    {
      "epoch": 5.880928039550781e-05,
      "model_forward_time": 1.7321982383728027,
      "step": 61666
    },
    {
      "epoch": 5.880928039550781e-05,
      "step": 61666,
      "training_step_time": 7.668829679489136
    },
    {
      "epoch": 5.881023406982422e-05,
      "model_forward_time": 1.7164812088012695,
      "step": 61667
    },
    {
      "epoch": 5.881023406982422e-05,
      "step": 61667,
      "training_step_time": 9.55693531036377
    },
    {
      "epoch": 5.8811187744140624e-05,
      "model_forward_time": 1.7316927909851074,
      "step": 61668
    },
    {
      "epoch": 5.8811187744140624e-05,
      "step": 61668,
      "training_step_time": 9.37242841720581
    },
    {
      "epoch": 5.881214141845703e-05,
      "model_forward_time": 1.7285304069519043,
      "step": 61669
    },
    {
      "epoch": 5.881214141845703e-05,
      "step": 61669,
      "training_step_time": 7.744401216506958
    },
    {
      "epoch": 5.881309509277344e-05,
      "grad_norm": 0.13957948982715607,
      "learning_rate": 3.50705773080022e-06,
      "loss": 0.057,
      "step": 61670
    },
    {
      "epoch": 5.881309509277344e-05,
      "model_forward_time": 1.7160353660583496,
      "step": 61670
    },
    {
      "dynamics_loss_avg": 0.060728549025952816,
      "epoch": 5.881309509277344e-05,
      "step": 61670
    },
    {
      "action_loss_avg": 0.0014409255440114067,
      "epoch": 5.881309509277344e-05,
      "step": 61670
    },
    {
      "epoch": 5.881309509277344e-05,
      "step": 61670,
      "training_step_time": 8.559383392333984
    },
    {
      "epoch": 5.881404876708984e-05,
      "model_forward_time": 2.920362949371338,
      "step": 61671
    },
    {
      "epoch": 5.881404876708984e-05,
      "step": 61671,
      "training_step_time": 7.630416631698608
    },
    {
      "epoch": 5.881500244140625e-05,
      "model_forward_time": 1.7404098510742188,
      "step": 61672
    },
    {
      "epoch": 5.881500244140625e-05,
      "step": 61672,
      "training_step_time": 7.610309839248657
    },
    {
      "epoch": 5.881595611572266e-05,
      "model_forward_time": 1.7348098754882812,
      "step": 61673
    },
    {
      "epoch": 5.881595611572266e-05,
      "step": 61673,
      "training_step_time": 7.6185033321380615
    },
    {
      "epoch": 5.881690979003906e-05,
      "model_forward_time": 1.7235500812530518,
      "step": 61674
    },
    {
      "epoch": 5.881690979003906e-05,
      "step": 61674,
      "training_step_time": 7.580737829208374
    },
    {
      "epoch": 5.8817863464355466e-05,
      "model_forward_time": 1.7202043533325195,
      "step": 61675
    },
    {
      "epoch": 5.8817863464355466e-05,
      "step": 61675,
      "training_step_time": 7.573270320892334
    },
    {
      "epoch": 5.881881713867188e-05,
      "model_forward_time": 1.727799892425537,
      "step": 61676
    },
    {
      "epoch": 5.881881713867188e-05,
      "step": 61676,
      "training_step_time": 7.54646635055542
    },
    {
      "epoch": 5.881977081298828e-05,
      "model_forward_time": 1.7301068305969238,
      "step": 61677
    },
    {
      "epoch": 5.881977081298828e-05,
      "step": 61677,
      "training_step_time": 7.582524061203003
    },
    {
      "epoch": 5.8820724487304685e-05,
      "model_forward_time": 1.7400403022766113,
      "step": 61678
    },
    {
      "epoch": 5.8820724487304685e-05,
      "step": 61678,
      "training_step_time": 9.6858811378479
    },
    {
      "epoch": 5.8821678161621096e-05,
      "model_forward_time": 1.7298619747161865,
      "step": 61679
    },
    {
      "epoch": 5.8821678161621096e-05,
      "step": 61679,
      "training_step_time": 7.78022575378418
    },
    {
      "epoch": 5.88226318359375e-05,
      "grad_norm": 0.11032012104988098,
      "learning_rate": 3.505479770688178e-06,
      "loss": 0.0578,
      "step": 61680
    },
    {
      "epoch": 5.88226318359375e-05,
      "model_forward_time": 2.880300760269165,
      "step": 61680
    },
    {
      "dynamics_loss_avg": 0.05071900337934494,
      "epoch": 5.88226318359375e-05,
      "step": 61680
    },
    {
      "action_loss_avg": 0.0017068130837287754,
      "epoch": 5.88226318359375e-05,
      "step": 61680
    },
    {
      "epoch": 5.88226318359375e-05,
      "step": 61680,
      "training_step_time": 9.138912200927734
    },
    {
      "epoch": 5.8823585510253904e-05,
      "model_forward_time": 1.7516651153564453,
      "step": 61681
    },
    {
      "epoch": 5.8823585510253904e-05,
      "step": 61681,
      "training_step_time": 7.592975378036499
    },
    {
      "epoch": 5.8824539184570315e-05,
      "model_forward_time": 1.718449592590332,
      "step": 61682
    },
    {
      "epoch": 5.8824539184570315e-05,
      "step": 61682,
      "training_step_time": 7.6372034549713135
    },
    {
      "epoch": 5.882549285888672e-05,
      "model_forward_time": 1.71478271484375,
      "step": 61683
    },
    {
      "epoch": 5.882549285888672e-05,
      "step": 61683,
      "training_step_time": 7.7428929805755615
    },
    {
      "epoch": 5.882644653320312e-05,
      "model_forward_time": 1.7308802604675293,
      "step": 61684
    },
    {
      "epoch": 5.882644653320312e-05,
      "step": 61684,
      "training_step_time": 7.660726547241211
    },
    {
      "epoch": 5.8827400207519533e-05,
      "model_forward_time": 1.7085492610931396,
      "step": 61685
    },
    {
      "epoch": 5.8827400207519533e-05,
      "step": 61685,
      "training_step_time": 7.730408191680908
    },
    {
      "epoch": 5.882835388183594e-05,
      "model_forward_time": 1.7191352844238281,
      "step": 61686
    },
    {
      "epoch": 5.882835388183594e-05,
      "step": 61686,
      "training_step_time": 8.549782276153564
    },
    {
      "epoch": 5.882930755615234e-05,
      "model_forward_time": 1.756422758102417,
      "step": 61687
    },
    {
      "epoch": 5.882930755615234e-05,
      "step": 61687,
      "training_step_time": 8.504008531570435
    },
    {
      "epoch": 5.883026123046875e-05,
      "model_forward_time": 1.7229597568511963,
      "step": 61688
    },
    {
      "epoch": 5.883026123046875e-05,
      "step": 61688,
      "training_step_time": 9.128384351730347
    },
    {
      "epoch": 5.8831214904785156e-05,
      "model_forward_time": 2.893397569656372,
      "step": 61689
    },
    {
      "epoch": 5.8831214904785156e-05,
      "step": 61689,
      "training_step_time": 7.753915548324585
    },
    {
      "epoch": 5.883216857910156e-05,
      "grad_norm": 0.09452708065509796,
      "learning_rate": 3.5039019740146217e-06,
      "loss": 0.0553,
      "step": 61690
    },
    {
      "epoch": 5.883216857910156e-05,
      "model_forward_time": 1.735630989074707,
      "step": 61690
    },
    {
      "dynamics_loss_avg": 0.04203155897557735,
      "epoch": 5.883216857910156e-05,
      "step": 61690
    },
    {
      "action_loss_avg": 0.0016310567836626433,
      "epoch": 5.883216857910156e-05,
      "step": 61690
    },
    {
      "epoch": 5.883216857910156e-05,
      "step": 61690,
      "training_step_time": 7.721238136291504
    },
    {
      "epoch": 5.883312225341797e-05,
      "model_forward_time": 1.7183620929718018,
      "step": 61691
    },
    {
      "epoch": 5.883312225341797e-05,
      "step": 61691,
      "training_step_time": 7.87654185295105
    },
    {
      "epoch": 5.8834075927734375e-05,
      "model_forward_time": 1.7323648929595947,
      "step": 61692
    },
    {
      "epoch": 5.8834075927734375e-05,
      "step": 61692,
      "training_step_time": 7.737633466720581
    },
    {
      "epoch": 5.883502960205078e-05,
      "model_forward_time": 1.7187321186065674,
      "step": 61693
    },
    {
      "epoch": 5.883502960205078e-05,
      "step": 61693,
      "training_step_time": 7.7969090938568115
    },
    {
      "epoch": 5.883598327636719e-05,
      "model_forward_time": 1.7251853942871094,
      "step": 61694
    },
    {
      "epoch": 5.883598327636719e-05,
      "step": 61694,
      "training_step_time": 7.780030727386475
    },
    {
      "epoch": 5.8836936950683594e-05,
      "model_forward_time": 1.7216172218322754,
      "step": 61695
    },
    {
      "epoch": 5.8836936950683594e-05,
      "step": 61695,
      "training_step_time": 7.722834825515747
    },
    {
      "epoch": 5.8837890625e-05,
      "model_forward_time": 1.737818717956543,
      "step": 61696
    },
    {
      "epoch": 5.8837890625e-05,
      "step": 61696,
      "training_step_time": 7.628871917724609
    },
    {
      "epoch": 5.883884429931641e-05,
      "model_forward_time": 1.7139959335327148,
      "step": 61697
    },
    {
      "epoch": 5.883884429931641e-05,
      "step": 61697,
      "training_step_time": 7.818869352340698
    },
    {
      "epoch": 5.883979797363281e-05,
      "model_forward_time": 2.839409112930298,
      "step": 61698
    },
    {
      "epoch": 5.883979797363281e-05,
      "step": 61698,
      "training_step_time": 9.602024793624878
    },
    {
      "epoch": 5.884075164794922e-05,
      "model_forward_time": 1.7110004425048828,
      "step": 61699
    },
    {
      "epoch": 5.884075164794922e-05,
      "step": 61699,
      "training_step_time": 7.519470453262329
    },
    {
      "epoch": 5.884170532226563e-05,
      "grad_norm": 0.10560338944196701,
      "learning_rate": 3.5023243409520956e-06,
      "loss": 0.0569,
      "step": 61700
    },
    {
      "epoch": 5.884170532226563e-05,
      "model_forward_time": 1.7176718711853027,
      "step": 61700
    },
    {
      "dynamics_loss_avg": 0.06312343422323466,
      "epoch": 5.884170532226563e-05,
      "step": 61700
    },
    {
      "action_loss_avg": 0.0021470296196639538,
      "epoch": 5.884170532226563e-05,
      "step": 61700
    },
    {
      "epoch": 5.884170532226563e-05,
      "step": 61700,
      "training_step_time": 7.645323991775513
    },
    {
      "epoch": 5.884265899658203e-05,
      "model_forward_time": 1.6987018585205078,
      "step": 61701
    },
    {
      "epoch": 5.884265899658203e-05,
      "step": 61701,
      "training_step_time": 7.577562093734741
    },
    {
      "epoch": 5.8843612670898436e-05,
      "model_forward_time": 1.698361873626709,
      "step": 61702
    },
    {
      "epoch": 5.8843612670898436e-05,
      "step": 61702,
      "training_step_time": 8.28429365158081
    },
    {
      "epoch": 5.8844566345214847e-05,
      "model_forward_time": 1.713421106338501,
      "step": 61703
    },
    {
      "epoch": 5.8844566345214847e-05,
      "step": 61703,
      "training_step_time": 8.288048028945923
    },
    {
      "epoch": 5.884552001953125e-05,
      "model_forward_time": 1.7016711235046387,
      "step": 61704
    },
    {
      "epoch": 5.884552001953125e-05,
      "step": 61704,
      "training_step_time": 7.591197729110718
    },
    {
      "epoch": 5.8846473693847655e-05,
      "model_forward_time": 1.7261927127838135,
      "step": 61705
    },
    {
      "epoch": 5.8846473693847655e-05,
      "step": 61705,
      "training_step_time": 7.663786172866821
    },
    {
      "epoch": 5.8847427368164065e-05,
      "model_forward_time": 1.7100613117218018,
      "step": 61706
    },
    {
      "epoch": 5.8847427368164065e-05,
      "step": 61706,
      "training_step_time": 9.328834533691406
    },
    {
      "epoch": 5.884838104248047e-05,
      "model_forward_time": 2.8219566345214844,
      "step": 61707
    },
    {
      "epoch": 5.884838104248047e-05,
      "step": 61707,
      "training_step_time": 8.47591757774353
    },
    {
      "epoch": 5.8849334716796873e-05,
      "model_forward_time": 1.7165751457214355,
      "step": 61708
    },
    {
      "epoch": 5.8849334716796873e-05,
      "step": 61708,
      "training_step_time": 7.660974740982056
    },
    {
      "epoch": 5.8850288391113284e-05,
      "model_forward_time": 1.7172763347625732,
      "step": 61709
    },
    {
      "epoch": 5.8850288391113284e-05,
      "step": 61709,
      "training_step_time": 7.647167921066284
    },
    {
      "epoch": 5.885124206542969e-05,
      "grad_norm": 0.14319080114364624,
      "learning_rate": 3.500746871673126e-06,
      "loss": 0.0595,
      "step": 61710
    },
    {
      "epoch": 5.885124206542969e-05,
      "model_forward_time": 1.7037582397460938,
      "step": 61710
    },
    {
      "dynamics_loss_avg": 0.0467730475589633,
      "epoch": 5.885124206542969e-05,
      "step": 61710
    },
    {
      "action_loss_avg": 0.0012076250917743892,
      "epoch": 5.885124206542969e-05,
      "step": 61710
    },
    {
      "epoch": 5.885124206542969e-05,
      "step": 61710,
      "training_step_time": 7.579606533050537
    },
    {
      "epoch": 5.885219573974609e-05,
      "model_forward_time": 1.715684413909912,
      "step": 61711
    },
    {
      "epoch": 5.885219573974609e-05,
      "step": 61711,
      "training_step_time": 7.598383665084839
    },
    {
      "epoch": 5.88531494140625e-05,
      "model_forward_time": 1.6989030838012695,
      "step": 61712
    },
    {
      "epoch": 5.88531494140625e-05,
      "step": 61712,
      "training_step_time": 7.65238881111145
    },
    {
      "epoch": 5.885410308837891e-05,
      "model_forward_time": 1.701200246810913,
      "step": 61713
    },
    {
      "epoch": 5.885410308837891e-05,
      "step": 61713,
      "training_step_time": 8.094738960266113
    },
    {
      "epoch": 5.885505676269531e-05,
      "model_forward_time": 1.739800214767456,
      "step": 61714
    },
    {
      "epoch": 5.885505676269531e-05,
      "step": 61714,
      "training_step_time": 9.199445962905884
    },
    {
      "epoch": 5.885601043701172e-05,
      "model_forward_time": 1.701796054840088,
      "step": 61715
    },
    {
      "epoch": 5.885601043701172e-05,
      "step": 61715,
      "training_step_time": 7.524120092391968
    },
    {
      "epoch": 5.8856964111328126e-05,
      "model_forward_time": 1.703073501586914,
      "step": 61716
    },
    {
      "epoch": 5.8856964111328126e-05,
      "step": 61716,
      "training_step_time": 9.197244882583618
    },
    {
      "epoch": 5.885791778564453e-05,
      "model_forward_time": 2.8941476345062256,
      "step": 61717
    },
    {
      "epoch": 5.885791778564453e-05,
      "step": 61717,
      "training_step_time": 7.587966442108154
    },
    {
      "epoch": 5.885887145996094e-05,
      "model_forward_time": 1.7154736518859863,
      "step": 61718
    },
    {
      "epoch": 5.885887145996094e-05,
      "step": 61718,
      "training_step_time": 7.614410877227783
    },
    {
      "epoch": 5.8859825134277345e-05,
      "model_forward_time": 1.7042763233184814,
      "step": 61719
    },
    {
      "epoch": 5.8859825134277345e-05,
      "step": 61719,
      "training_step_time": 7.734868049621582
    },
    {
      "epoch": 5.886077880859375e-05,
      "grad_norm": 0.09142117202281952,
      "learning_rate": 3.499169566350227e-06,
      "loss": 0.0568,
      "step": 61720
    },
    {
      "epoch": 5.886077880859375e-05,
      "model_forward_time": 1.709165334701538,
      "step": 61720
    },
    {
      "dynamics_loss_avg": 0.052990635950118305,
      "epoch": 5.886077880859375e-05,
      "step": 61720
    },
    {
      "action_loss_avg": 0.001610086920845788,
      "epoch": 5.886077880859375e-05,
      "step": 61720
    },
    {
      "epoch": 5.886077880859375e-05,
      "step": 61720,
      "training_step_time": 7.612084150314331
    },
    {
      "epoch": 5.886173248291015e-05,
      "model_forward_time": 1.6969349384307861,
      "step": 61721
    },
    {
      "epoch": 5.886173248291015e-05,
      "step": 61721,
      "training_step_time": 7.657799243927002
    },
    {
      "epoch": 5.8862686157226564e-05,
      "model_forward_time": 1.7199900150299072,
      "step": 61722
    },
    {
      "epoch": 5.8862686157226564e-05,
      "step": 61722,
      "training_step_time": 7.446026802062988
    },
    {
      "epoch": 5.886363983154297e-05,
      "model_forward_time": 1.72548508644104,
      "step": 61723
    },
    {
      "epoch": 5.886363983154297e-05,
      "step": 61723,
      "training_step_time": 7.5853588581085205
    },
    {
      "epoch": 5.886459350585937e-05,
      "model_forward_time": 1.699629545211792,
      "step": 61724
    },
    {
      "epoch": 5.886459350585937e-05,
      "step": 61724,
      "training_step_time": 8.618004322052002
    },
    {
      "epoch": 5.886554718017578e-05,
      "model_forward_time": 1.7094056606292725,
      "step": 61725
    },
    {
      "epoch": 5.886554718017578e-05,
      "step": 61725,
      "training_step_time": 8.327848434448242
    },
    {
      "epoch": 5.886650085449219e-05,
      "model_forward_time": 2.8670010566711426,
      "step": 61726
    },
    {
      "epoch": 5.886650085449219e-05,
      "step": 61726,
      "training_step_time": 7.6012420654296875
    },
    {
      "epoch": 5.886745452880859e-05,
      "model_forward_time": 1.71372652053833,
      "step": 61727
    },
    {
      "epoch": 5.886745452880859e-05,
      "step": 61727,
      "training_step_time": 7.547943115234375
    },
    {
      "epoch": 5.8868408203125e-05,
      "model_forward_time": 1.70534348487854,
      "step": 61728
    },
    {
      "epoch": 5.8868408203125e-05,
      "step": 61728,
      "training_step_time": 8.476440668106079
    },
    {
      "epoch": 5.8869361877441405e-05,
      "model_forward_time": 1.6994736194610596,
      "step": 61729
    },
    {
      "epoch": 5.8869361877441405e-05,
      "step": 61729,
      "training_step_time": 7.559634447097778
    },
    {
      "epoch": 5.887031555175781e-05,
      "grad_norm": 0.08363346010446548,
      "learning_rate": 3.497592425155888e-06,
      "loss": 0.0573,
      "step": 61730
    },
    {
      "epoch": 5.887031555175781e-05,
      "model_forward_time": 1.7026755809783936,
      "step": 61730
    },
    {
      "dynamics_loss_avg": 0.04880290050059557,
      "epoch": 5.887031555175781e-05,
      "step": 61730
    },
    {
      "action_loss_avg": 0.0011329007567837835,
      "epoch": 5.887031555175781e-05,
      "step": 61730
    },
    {
      "epoch": 5.887031555175781e-05,
      "step": 61730,
      "training_step_time": 9.733467817306519
    },
    {
      "epoch": 5.887126922607422e-05,
      "model_forward_time": 1.6975634098052979,
      "step": 61731
    },
    {
      "epoch": 5.887126922607422e-05,
      "step": 61731,
      "training_step_time": 9.509103059768677
    },
    {
      "epoch": 5.8872222900390624e-05,
      "model_forward_time": 1.7059059143066406,
      "step": 61732
    },
    {
      "epoch": 5.8872222900390624e-05,
      "step": 61732,
      "training_step_time": 6.716994762420654
    },
    {
      "epoch": 5.887317657470703e-05,
      "model_forward_time": 1.6944913864135742,
      "step": 61733
    },
    {
      "epoch": 5.887317657470703e-05,
      "step": 61733,
      "training_step_time": 6.742162227630615
    },
    {
      "epoch": 5.887413024902344e-05,
      "model_forward_time": 1.7017185688018799,
      "step": 61734
    },
    {
      "epoch": 5.887413024902344e-05,
      "step": 61734,
      "training_step_time": 6.72691011428833
    },
    {
      "epoch": 5.887508392333984e-05,
      "model_forward_time": 1.696605920791626,
      "step": 61735
    },
    {
      "epoch": 5.887508392333984e-05,
      "step": 61735,
      "training_step_time": 6.725390672683716
    },
    {
      "epoch": 5.887603759765625e-05,
      "model_forward_time": 1.6980290412902832,
      "step": 61736
    },
    {
      "epoch": 5.887603759765625e-05,
      "step": 61736,
      "training_step_time": 6.724673748016357
    },
    {
      "epoch": 5.887699127197266e-05,
      "model_forward_time": 1.7037475109100342,
      "step": 61737
    },
    {
      "epoch": 5.887699127197266e-05,
      "step": 61737,
      "training_step_time": 6.720968723297119
    },
    {
      "epoch": 5.887794494628906e-05,
      "model_forward_time": 1.7037322521209717,
      "step": 61738
    },
    {
      "epoch": 5.887794494628906e-05,
      "step": 61738,
      "training_step_time": 6.761373996734619
    },
    {
      "epoch": 5.8878898620605466e-05,
      "model_forward_time": 1.7114319801330566,
      "step": 61739
    },
    {
      "epoch": 5.8878898620605466e-05,
      "step": 61739,
      "training_step_time": 6.74300742149353
    },
    {
      "epoch": 5.887985229492188e-05,
      "grad_norm": 0.08225754648447037,
      "learning_rate": 3.4960154482625817e-06,
      "loss": 0.0561,
      "step": 61740
    },
    {
      "epoch": 5.887985229492188e-05,
      "model_forward_time": 2.846182107925415,
      "step": 61740
    },
    {
      "dynamics_loss_avg": 0.045082521811127665,
      "epoch": 5.887985229492188e-05,
      "step": 61740
    },
    {
      "action_loss_avg": 0.0010499699070351197,
      "epoch": 5.887985229492188e-05,
      "step": 61740
    },
    {
      "epoch": 5.887985229492188e-05,
      "step": 61740,
      "training_step_time": 7.544717788696289
    },
    {
      "epoch": 5.888080596923828e-05,
      "model_forward_time": 1.6987113952636719,
      "step": 61741
    },
    {
      "epoch": 5.888080596923828e-05,
      "step": 61741,
      "training_step_time": 7.699800491333008
    },
    {
      "epoch": 5.8881759643554685e-05,
      "model_forward_time": 1.7014470100402832,
      "step": 61742
    },
    {
      "epoch": 5.8881759643554685e-05,
      "step": 61742,
      "training_step_time": 6.778838396072388
    },
    {
      "epoch": 5.8882713317871096e-05,
      "model_forward_time": 1.7044763565063477,
      "step": 61743
    },
    {
      "epoch": 5.8882713317871096e-05,
      "step": 61743,
      "training_step_time": 6.747559070587158
    },
    {
      "epoch": 5.88836669921875e-05,
      "model_forward_time": 1.7521090507507324,
      "step": 61744
    },
    {
      "epoch": 5.88836669921875e-05,
      "step": 61744,
      "training_step_time": 6.753622531890869
    },
    {
      "epoch": 5.8884620666503904e-05,
      "model_forward_time": 1.7217493057250977,
      "step": 61745
    },
    {
      "epoch": 5.8884620666503904e-05,
      "step": 61745,
      "training_step_time": 6.817778587341309
    },
    {
      "epoch": 5.8885574340820315e-05,
      "model_forward_time": 1.7168736457824707,
      "step": 61746
    },
    {
      "epoch": 5.8885574340820315e-05,
      "step": 61746,
      "training_step_time": 6.741160154342651
    },
    {
      "epoch": 5.888652801513672e-05,
      "model_forward_time": 1.7155680656433105,
      "step": 61747
    },
    {
      "epoch": 5.888652801513672e-05,
      "step": 61747,
      "training_step_time": 6.726102828979492
    },
    {
      "epoch": 5.888748168945312e-05,
      "model_forward_time": 1.7058250904083252,
      "step": 61748
    },
    {
      "epoch": 5.888748168945312e-05,
      "step": 61748,
      "training_step_time": 6.749124050140381
    },
    {
      "epoch": 5.8888435363769533e-05,
      "model_forward_time": 2.2628018856048584,
      "step": 61749
    },
    {
      "epoch": 5.8888435363769533e-05,
      "step": 61749,
      "training_step_time": 7.499643325805664
    },
    {
      "epoch": 5.888938903808594e-05,
      "grad_norm": 0.09442983567714691,
      "learning_rate": 3.494438635842764e-06,
      "loss": 0.0613,
      "step": 61750
    },
    {
      "epoch": 5.888938903808594e-05,
      "model_forward_time": 1.716510534286499,
      "step": 61750
    },
    {
      "dynamics_loss_avg": 0.03960388442501426,
      "epoch": 5.888938903808594e-05,
      "step": 61750
    },
    {
      "action_loss_avg": 0.0014095484773861244,
      "epoch": 5.888938903808594e-05,
      "step": 61750
    },
    {
      "epoch": 5.888938903808594e-05,
      "step": 61750,
      "training_step_time": 8.595452070236206
    },
    {
      "epoch": 5.889034271240234e-05,
      "model_forward_time": 1.7149536609649658,
      "step": 61751
    },
    {
      "epoch": 5.889034271240234e-05,
      "step": 61751,
      "training_step_time": 6.799600839614868
    },
    {
      "epoch": 5.889129638671875e-05,
      "model_forward_time": 1.7202324867248535,
      "step": 61752
    },
    {
      "epoch": 5.889129638671875e-05,
      "step": 61752,
      "training_step_time": 6.8063671588897705
    },
    {
      "epoch": 5.8892250061035156e-05,
      "model_forward_time": 1.7190821170806885,
      "step": 61753
    },
    {
      "epoch": 5.8892250061035156e-05,
      "step": 61753,
      "training_step_time": 6.750084400177002
    },
    {
      "epoch": 5.889320373535156e-05,
      "model_forward_time": 1.714233160018921,
      "step": 61754
    },
    {
      "epoch": 5.889320373535156e-05,
      "step": 61754,
      "training_step_time": 6.787318468093872
    },
    {
      "epoch": 5.889415740966797e-05,
      "model_forward_time": 1.7109870910644531,
      "step": 61755
    },
    {
      "epoch": 5.889415740966797e-05,
      "step": 61755,
      "training_step_time": 6.7821362018585205
    },
    {
      "epoch": 5.8895111083984375e-05,
      "model_forward_time": 1.7148373126983643,
      "step": 61756
    },
    {
      "epoch": 5.8895111083984375e-05,
      "step": 61756,
      "training_step_time": 6.780142307281494
    },
    {
      "epoch": 5.889606475830078e-05,
      "model_forward_time": 1.7189557552337646,
      "step": 61757
    },
    {
      "epoch": 5.889606475830078e-05,
      "step": 61757,
      "training_step_time": 6.8120481967926025
    },
    {
      "epoch": 5.889701843261719e-05,
      "model_forward_time": 2.2597126960754395,
      "step": 61758
    },
    {
      "epoch": 5.889701843261719e-05,
      "step": 61758,
      "training_step_time": 7.631987810134888
    },
    {
      "epoch": 5.8897972106933594e-05,
      "model_forward_time": 1.7108190059661865,
      "step": 61759
    },
    {
      "epoch": 5.8897972106933594e-05,
      "step": 61759,
      "training_step_time": 8.53279733657837
    },
    {
      "epoch": 5.889892578125e-05,
      "grad_norm": 0.10941246151924133,
      "learning_rate": 3.492861988068874e-06,
      "loss": 0.0566,
      "step": 61760
    },
    {
      "epoch": 5.889892578125e-05,
      "model_forward_time": 1.714247226715088,
      "step": 61760
    },
    {
      "dynamics_loss_avg": 0.04461826793849468,
      "epoch": 5.889892578125e-05,
      "step": 61760
    },
    {
      "action_loss_avg": 0.0018944597541121766,
      "epoch": 5.889892578125e-05,
      "step": 61760
    },
    {
      "epoch": 5.889892578125e-05,
      "step": 61760,
      "training_step_time": 7.641133546829224
    },
    {
      "epoch": 5.889987945556641e-05,
      "model_forward_time": 1.703862190246582,
      "step": 61761
    },
    {
      "epoch": 5.889987945556641e-05,
      "step": 61761,
      "training_step_time": 6.8199098110198975
    },
    {
      "epoch": 5.890083312988281e-05,
      "model_forward_time": 1.7194483280181885,
      "step": 61762
    },
    {
      "epoch": 5.890083312988281e-05,
      "step": 61762,
      "training_step_time": 6.80396580696106
    },
    {
      "epoch": 5.890178680419922e-05,
      "model_forward_time": 1.7153184413909912,
      "step": 61763
    },
    {
      "epoch": 5.890178680419922e-05,
      "step": 61763,
      "training_step_time": 6.755428791046143
    },
    {
      "epoch": 5.890274047851563e-05,
      "model_forward_time": 1.7160468101501465,
      "step": 61764
    },
    {
      "epoch": 5.890274047851563e-05,
      "step": 61764,
      "training_step_time": 6.782095432281494
    },
    {
      "epoch": 5.890369415283203e-05,
      "model_forward_time": 1.7213704586029053,
      "step": 61765
    },
    {
      "epoch": 5.890369415283203e-05,
      "step": 61765,
      "training_step_time": 6.775716543197632
    },
    {
      "epoch": 5.8904647827148436e-05,
      "model_forward_time": 1.7194719314575195,
      "step": 61766
    },
    {
      "epoch": 5.8904647827148436e-05,
      "step": 61766,
      "training_step_time": 6.855525016784668
    },
    {
      "epoch": 5.890560150146485e-05,
      "model_forward_time": 2.277188301086426,
      "step": 61767
    },
    {
      "epoch": 5.890560150146485e-05,
      "step": 61767,
      "training_step_time": 7.6844658851623535
    },
    {
      "epoch": 5.890655517578125e-05,
      "model_forward_time": 1.7075152397155762,
      "step": 61768
    },
    {
      "epoch": 5.890655517578125e-05,
      "step": 61768,
      "training_step_time": 10.617368936538696
    },
    {
      "epoch": 5.8907508850097655e-05,
      "model_forward_time": 1.7134606838226318,
      "step": 61769
    },
    {
      "epoch": 5.8907508850097655e-05,
      "step": 61769,
      "training_step_time": 7.785390853881836
    },
    {
      "epoch": 5.8908462524414065e-05,
      "grad_norm": 0.10305173695087433,
      "learning_rate": 3.491285505113332e-06,
      "loss": 0.0569,
      "step": 61770
    },
    {
      "epoch": 5.8908462524414065e-05,
      "model_forward_time": 1.027552843093872,
      "step": 61770
    },
    {
      "dynamics_loss_avg": 0.05499185044318437,
      "epoch": 5.8908462524414065e-05,
      "step": 61770
    },
    {
      "action_loss_avg": 0.001899463712470606,
      "epoch": 5.8908462524414065e-05,
      "step": 61770
    },
    {
      "epoch": 5.8908462524414065e-05,
      "step": 61770,
      "training_step_time": 6.885960817337036
    },
    {
      "epoch": 5.890941619873047e-05,
      "model_forward_time": 1.714306116104126,
      "step": 61771
    },
    {
      "epoch": 5.890941619873047e-05,
      "step": 61771,
      "training_step_time": 6.926722526550293
    },
    {
      "epoch": 5.8910369873046874e-05,
      "model_forward_time": 1.7152082920074463,
      "step": 61772
    },
    {
      "epoch": 5.8910369873046874e-05,
      "step": 61772,
      "training_step_time": 6.953171968460083
    },
    {
      "epoch": 5.8911323547363284e-05,
      "model_forward_time": 1.7107038497924805,
      "step": 61773
    },
    {
      "epoch": 5.8911323547363284e-05,
      "step": 61773,
      "training_step_time": 6.953214645385742
    },
    {
      "epoch": 5.891227722167969e-05,
      "model_forward_time": 1.7074716091156006,
      "step": 61774
    },
    {
      "epoch": 5.891227722167969e-05,
      "step": 61774,
      "training_step_time": 6.906057596206665
    },
    {
      "epoch": 5.891323089599609e-05,
      "model_forward_time": 1.7015478610992432,
      "step": 61775
    },
    {
      "epoch": 5.891323089599609e-05,
      "step": 61775,
      "training_step_time": 6.955575227737427
    },
    {
      "epoch": 5.89141845703125e-05,
      "model_forward_time": 1.720409631729126,
      "step": 61776
    },
    {
      "epoch": 5.89141845703125e-05,
      "step": 61776,
      "training_step_time": 7.713886022567749
    },
    {
      "epoch": 5.891513824462891e-05,
      "model_forward_time": 2.256143093109131,
      "step": 61777
    },
    {
      "epoch": 5.891513824462891e-05,
      "step": 61777,
      "training_step_time": 8.96007251739502
    },
    {
      "epoch": 5.891609191894531e-05,
      "model_forward_time": 1.711014986038208,
      "step": 61778
    },
    {
      "epoch": 5.891609191894531e-05,
      "step": 61778,
      "training_step_time": 8.47318720817566
    },
    {
      "epoch": 5.891704559326172e-05,
      "model_forward_time": 1.7338275909423828,
      "step": 61779
    },
    {
      "epoch": 5.891704559326172e-05,
      "step": 61779,
      "training_step_time": 6.863497018814087
    },
    {
      "epoch": 5.8917999267578126e-05,
      "grad_norm": 0.09397917985916138,
      "learning_rate": 3.4897091871485376e-06,
      "loss": 0.0531,
      "step": 61780
    },
    {
      "epoch": 5.8917999267578126e-05,
      "model_forward_time": 1.7189784049987793,
      "step": 61780
    },
    {
      "dynamics_loss_avg": 0.04557697549462318,
      "epoch": 5.8917999267578126e-05,
      "step": 61780
    },
    {
      "action_loss_avg": 0.0019710133317857982,
      "epoch": 5.8917999267578126e-05,
      "step": 61780
    },
    {
      "epoch": 5.8917999267578126e-05,
      "step": 61780,
      "training_step_time": 6.891045093536377
    },
    {
      "epoch": 5.891895294189453e-05,
      "model_forward_time": 1.7066302299499512,
      "step": 61781
    },
    {
      "epoch": 5.891895294189453e-05,
      "step": 61781,
      "training_step_time": 6.874690294265747
    },
    {
      "epoch": 5.8919906616210934e-05,
      "model_forward_time": 1.7166883945465088,
      "step": 61782
    },
    {
      "epoch": 5.8919906616210934e-05,
      "step": 61782,
      "training_step_time": 6.941211938858032
    },
    {
      "epoch": 5.8920860290527345e-05,
      "model_forward_time": 1.720064640045166,
      "step": 61783
    },
    {
      "epoch": 5.8920860290527345e-05,
      "step": 61783,
      "training_step_time": 6.8406147956848145
    },
    {
      "epoch": 5.892181396484375e-05,
      "model_forward_time": 1.7211239337921143,
      "step": 61784
    },
    {
      "epoch": 5.892181396484375e-05,
      "step": 61784,
      "training_step_time": 6.849732875823975
    },
    {
      "epoch": 5.892276763916015e-05,
      "model_forward_time": 1.7069385051727295,
      "step": 61785
    },
    {
      "epoch": 5.892276763916015e-05,
      "step": 61785,
      "training_step_time": 8.408499956130981
    },
    {
      "epoch": 5.8923721313476564e-05,
      "model_forward_time": 2.2552382946014404,
      "step": 61786
    },
    {
      "epoch": 5.8923721313476564e-05,
      "step": 61786,
      "training_step_time": 10.225172519683838
    },
    {
      "epoch": 5.892467498779297e-05,
      "model_forward_time": 1.7008602619171143,
      "step": 61787
    },
    {
      "epoch": 5.892467498779297e-05,
      "step": 61787,
      "training_step_time": 8.817699432373047
    },
    {
      "epoch": 5.892562866210937e-05,
      "model_forward_time": 1.7305333614349365,
      "step": 61788
    },
    {
      "epoch": 5.892562866210937e-05,
      "step": 61788,
      "training_step_time": 7.6767051219940186
    },
    {
      "epoch": 5.892658233642578e-05,
      "model_forward_time": 1.3599119186401367,
      "step": 61789
    },
    {
      "epoch": 5.892658233642578e-05,
      "step": 61789,
      "training_step_time": 6.8677356243133545
    },
    {
      "epoch": 5.892753601074219e-05,
      "grad_norm": 0.11952999234199524,
      "learning_rate": 3.4881330343468755e-06,
      "loss": 0.0547,
      "step": 61790
    },
    {
      "epoch": 5.892753601074219e-05,
      "model_forward_time": 1.7159838676452637,
      "step": 61790
    },
    {
      "dynamics_loss_avg": 0.06282332092523575,
      "epoch": 5.892753601074219e-05,
      "step": 61790
    },
    {
      "action_loss_avg": 0.0019656931748613716,
      "epoch": 5.892753601074219e-05,
      "step": 61790
    },
    {
      "epoch": 5.892753601074219e-05,
      "step": 61790,
      "training_step_time": 6.848480701446533
    },
    {
      "epoch": 5.892848968505859e-05,
      "model_forward_time": 1.712003469467163,
      "step": 61791
    },
    {
      "epoch": 5.892848968505859e-05,
      "step": 61791,
      "training_step_time": 6.918933629989624
    },
    {
      "epoch": 5.8929443359375e-05,
      "model_forward_time": 1.7187094688415527,
      "step": 61792
    },
    {
      "epoch": 5.8929443359375e-05,
      "step": 61792,
      "training_step_time": 6.886094808578491
    },
    {
      "epoch": 5.8930397033691406e-05,
      "model_forward_time": 1.7166614532470703,
      "step": 61793
    },
    {
      "epoch": 5.8930397033691406e-05,
      "step": 61793,
      "training_step_time": 6.876401662826538
    },
    {
      "epoch": 5.893135070800781e-05,
      "model_forward_time": 1.7118360996246338,
      "step": 61794
    },
    {
      "epoch": 5.893135070800781e-05,
      "step": 61794,
      "training_step_time": 7.767191171646118
    },
    {
      "epoch": 5.893230438232422e-05,
      "model_forward_time": 2.2742161750793457,
      "step": 61795
    },
    {
      "epoch": 5.893230438232422e-05,
      "step": 61795,
      "training_step_time": 9.983280897140503
    },
    {
      "epoch": 5.8933258056640624e-05,
      "model_forward_time": 1.7191946506500244,
      "step": 61796
    },
    {
      "epoch": 5.8933258056640624e-05,
      "step": 61796,
      "training_step_time": 7.858251094818115
    },
    {
      "epoch": 5.893421173095703e-05,
      "model_forward_time": 1.709578037261963,
      "step": 61797
    },
    {
      "epoch": 5.893421173095703e-05,
      "step": 61797,
      "training_step_time": 7.724425554275513
    },
    {
      "epoch": 5.893516540527344e-05,
      "model_forward_time": 1.71441650390625,
      "step": 61798
    },
    {
      "epoch": 5.893516540527344e-05,
      "step": 61798,
      "training_step_time": 6.876415729522705
    },
    {
      "epoch": 5.893611907958984e-05,
      "model_forward_time": 1.7162363529205322,
      "step": 61799
    },
    {
      "epoch": 5.893611907958984e-05,
      "step": 61799,
      "training_step_time": 6.851331949234009
    },
    {
      "epoch": 5.893707275390625e-05,
      "grad_norm": 0.09458843618631363,
      "learning_rate": 3.486557046880713e-06,
      "loss": 0.0528,
      "step": 61800
    },
    {
      "epoch": 5.893707275390625e-05,
      "model_forward_time": 1.7216906547546387,
      "step": 61800
    },
    {
      "dynamics_loss_avg": 0.0660904448479414,
      "epoch": 5.893707275390625e-05,
      "step": 61800
    },
    {
      "action_loss_avg": 0.0014743340521818026,
      "epoch": 5.893707275390625e-05,
      "step": 61800
    },
    {
      "epoch": 5.893707275390625e-05,
      "step": 61800,
      "training_step_time": 6.75787091255188
    },
    {
      "epoch": 5.893802642822266e-05,
      "model_forward_time": 1.7036559581756592,
      "step": 61801
    },
    {
      "epoch": 5.893802642822266e-05,
      "step": 61801,
      "training_step_time": 6.691210508346558
    },
    {
      "epoch": 5.893898010253906e-05,
      "model_forward_time": 1.7125520706176758,
      "step": 61802
    },
    {
      "epoch": 5.893898010253906e-05,
      "step": 61802,
      "training_step_time": 6.754861354827881
    },
    {
      "epoch": 5.8939933776855466e-05,
      "model_forward_time": 1.711663007736206,
      "step": 61803
    },
    {
      "epoch": 5.8939933776855466e-05,
      "step": 61803,
      "training_step_time": 7.4360246658325195
    },
    {
      "epoch": 5.894088745117188e-05,
      "model_forward_time": 2.294524908065796,
      "step": 61804
    },
    {
      "epoch": 5.894088745117188e-05,
      "step": 61804,
      "training_step_time": 8.627564907073975
    },
    {
      "epoch": 5.894184112548828e-05,
      "model_forward_time": 1.7054331302642822,
      "step": 61805
    },
    {
      "epoch": 5.894184112548828e-05,
      "step": 61805,
      "training_step_time": 9.16924500465393
    },
    {
      "epoch": 5.8942794799804685e-05,
      "model_forward_time": 1.7115395069122314,
      "step": 61806
    },
    {
      "epoch": 5.8942794799804685e-05,
      "step": 61806,
      "training_step_time": 7.590670108795166
    },
    {
      "epoch": 5.8943748474121096e-05,
      "model_forward_time": 1.7034311294555664,
      "step": 61807
    },
    {
      "epoch": 5.8943748474121096e-05,
      "step": 61807,
      "training_step_time": 6.7162628173828125
    },
    {
      "epoch": 5.89447021484375e-05,
      "model_forward_time": 1.7145214080810547,
      "step": 61808
    },
    {
      "epoch": 5.89447021484375e-05,
      "step": 61808,
      "training_step_time": 6.7355852127075195
    },
    {
      "epoch": 5.8945655822753904e-05,
      "model_forward_time": 1.718501329421997,
      "step": 61809
    },
    {
      "epoch": 5.8945655822753904e-05,
      "step": 61809,
      "training_step_time": 6.74351692199707
    },
    {
      "epoch": 5.8946609497070315e-05,
      "grad_norm": 0.10154638439416885,
      "learning_rate": 3.4849812249223958e-06,
      "loss": 0.0555,
      "step": 61810
    },
    {
      "epoch": 5.8946609497070315e-05,
      "model_forward_time": 1.7244744300842285,
      "step": 61810
    },
    {
      "dynamics_loss_avg": 0.061157426610589025,
      "epoch": 5.8946609497070315e-05,
      "step": 61810
    },
    {
      "action_loss_avg": 0.002239967614877969,
      "epoch": 5.8946609497070315e-05,
      "step": 61810
    },
    {
      "epoch": 5.8946609497070315e-05,
      "step": 61810,
      "training_step_time": 6.721548318862915
    },
    {
      "epoch": 5.894756317138672e-05,
      "model_forward_time": 1.7132554054260254,
      "step": 61811
    },
    {
      "epoch": 5.894756317138672e-05,
      "step": 61811,
      "training_step_time": 6.709529161453247
    },
    {
      "epoch": 5.894851684570312e-05,
      "model_forward_time": 1.7149896621704102,
      "step": 61812
    },
    {
      "epoch": 5.894851684570312e-05,
      "step": 61812,
      "training_step_time": 7.483264207839966
    },
    {
      "epoch": 5.8949470520019534e-05,
      "model_forward_time": 1.7698700428009033,
      "step": 61813
    },
    {
      "epoch": 5.8949470520019534e-05,
      "step": 61813,
      "training_step_time": 8.632329225540161
    },
    {
      "epoch": 5.895042419433594e-05,
      "model_forward_time": 1.71852445602417,
      "step": 61814
    },
    {
      "epoch": 5.895042419433594e-05,
      "step": 61814,
      "training_step_time": 11.055935382843018
    },
    {
      "epoch": 5.895137786865234e-05,
      "model_forward_time": 1.7171225547790527,
      "step": 61815
    },
    {
      "epoch": 5.895137786865234e-05,
      "step": 61815,
      "training_step_time": 7.656065940856934
    },
    {
      "epoch": 5.895233154296875e-05,
      "model_forward_time": 1.7354083061218262,
      "step": 61816
    },
    {
      "epoch": 5.895233154296875e-05,
      "step": 61816,
      "training_step_time": 6.7420361042022705
    },
    {
      "epoch": 5.8953285217285156e-05,
      "model_forward_time": 1.7036454677581787,
      "step": 61817
    },
    {
      "epoch": 5.8953285217285156e-05,
      "step": 61817,
      "training_step_time": 6.701136112213135
    },
    {
      "epoch": 5.895423889160156e-05,
      "model_forward_time": 1.7094683647155762,
      "step": 61818
    },
    {
      "epoch": 5.895423889160156e-05,
      "step": 61818,
      "training_step_time": 6.686751127243042
    },
    {
      "epoch": 5.895519256591797e-05,
      "model_forward_time": 1.3509912490844727,
      "step": 61819
    },
    {
      "epoch": 5.895519256591797e-05,
      "step": 61819,
      "training_step_time": 6.714759588241577
    },
    {
      "epoch": 5.8956146240234375e-05,
      "grad_norm": 0.11575336009263992,
      "learning_rate": 3.483405568644255e-06,
      "loss": 0.0554,
      "step": 61820
    },
    {
      "epoch": 5.8956146240234375e-05,
      "model_forward_time": 1.7094814777374268,
      "step": 61820
    },
    {
      "dynamics_loss_avg": 0.05759907942265272,
      "epoch": 5.8956146240234375e-05,
      "step": 61820
    },
    {
      "action_loss_avg": 0.0019311822077725082,
      "epoch": 5.8956146240234375e-05,
      "step": 61820
    },
    {
      "epoch": 5.8956146240234375e-05,
      "step": 61820,
      "training_step_time": 6.677276134490967
    },
    {
      "epoch": 5.895709991455078e-05,
      "model_forward_time": 1.7131876945495605,
      "step": 61821
    },
    {
      "epoch": 5.895709991455078e-05,
      "step": 61821,
      "training_step_time": 7.701119899749756
    },
    {
      "epoch": 5.895805358886719e-05,
      "model_forward_time": 1.770305871963501,
      "step": 61822
    },
    {
      "epoch": 5.895805358886719e-05,
      "step": 61822,
      "training_step_time": 7.622005939483643
    },
    {
      "epoch": 5.8959007263183594e-05,
      "model_forward_time": 2.251957416534424,
      "step": 61823
    },
    {
      "epoch": 5.8959007263183594e-05,
      "step": 61823,
      "training_step_time": 9.45551586151123
    },
    {
      "epoch": 5.89599609375e-05,
      "model_forward_time": 1.7188565731048584,
      "step": 61824
    },
    {
      "epoch": 5.89599609375e-05,
      "step": 61824,
      "training_step_time": 7.628858804702759
    },
    {
      "epoch": 5.896091461181641e-05,
      "model_forward_time": 1.721601963043213,
      "step": 61825
    },
    {
      "epoch": 5.896091461181641e-05,
      "step": 61825,
      "training_step_time": 7.282233953475952
    },
    {
      "epoch": 5.896186828613281e-05,
      "model_forward_time": 1.7134203910827637,
      "step": 61826
    },
    {
      "epoch": 5.896186828613281e-05,
      "step": 61826,
      "training_step_time": 6.712235927581787
    },
    {
      "epoch": 5.896282196044922e-05,
      "model_forward_time": 1.714951992034912,
      "step": 61827
    },
    {
      "epoch": 5.896282196044922e-05,
      "step": 61827,
      "training_step_time": 6.6845293045043945
    },
    {
      "epoch": 5.896377563476563e-05,
      "model_forward_time": 1.7149133682250977,
      "step": 61828
    },
    {
      "epoch": 5.896377563476563e-05,
      "step": 61828,
      "training_step_time": 6.7166900634765625
    },
    {
      "epoch": 5.896472930908203e-05,
      "model_forward_time": 1.72432541847229,
      "step": 61829
    },
    {
      "epoch": 5.896472930908203e-05,
      "step": 61829,
      "training_step_time": 6.719628810882568
    },
    {
      "epoch": 5.8965682983398436e-05,
      "grad_norm": 0.11937312036752701,
      "learning_rate": 3.4818300782186e-06,
      "loss": 0.054,
      "step": 61830
    },
    {
      "epoch": 5.8965682983398436e-05,
      "model_forward_time": 1.7214648723602295,
      "step": 61830
    },
    {
      "dynamics_loss_avg": 0.04459101296961308,
      "epoch": 5.8965682983398436e-05,
      "step": 61830
    },
    {
      "action_loss_avg": 0.001478743675397709,
      "epoch": 5.8965682983398436e-05,
      "step": 61830
    },
    {
      "epoch": 5.8965682983398436e-05,
      "step": 61830,
      "training_step_time": 6.7130396366119385
    },
    {
      "epoch": 5.896663665771485e-05,
      "model_forward_time": 1.761732816696167,
      "step": 61831
    },
    {
      "epoch": 5.896663665771485e-05,
      "step": 61831,
      "training_step_time": 8.797223329544067
    },
    {
      "epoch": 5.896759033203125e-05,
      "model_forward_time": 2.2614095211029053,
      "step": 61832
    },
    {
      "epoch": 5.896759033203125e-05,
      "step": 61832,
      "training_step_time": 10.417148351669312
    },
    {
      "epoch": 5.8968544006347655e-05,
      "model_forward_time": 1.7103664875030518,
      "step": 61833
    },
    {
      "epoch": 5.8968544006347655e-05,
      "step": 61833,
      "training_step_time": 7.6894145011901855
    },
    {
      "epoch": 5.8969497680664066e-05,
      "model_forward_time": 1.7152378559112549,
      "step": 61834
    },
    {
      "epoch": 5.8969497680664066e-05,
      "step": 61834,
      "training_step_time": 7.521658897399902
    },
    {
      "epoch": 5.897045135498047e-05,
      "model_forward_time": 1.713545322418213,
      "step": 61835
    },
    {
      "epoch": 5.897045135498047e-05,
      "step": 61835,
      "training_step_time": 6.754648208618164
    },
    {
      "epoch": 5.8971405029296874e-05,
      "model_forward_time": 1.7096171379089355,
      "step": 61836
    },
    {
      "epoch": 5.8971405029296874e-05,
      "step": 61836,
      "training_step_time": 6.687729120254517
    },
    {
      "epoch": 5.8972358703613284e-05,
      "model_forward_time": 1.710838794708252,
      "step": 61837
    },
    {
      "epoch": 5.8972358703613284e-05,
      "step": 61837,
      "training_step_time": 6.686185598373413
    },
    {
      "epoch": 5.897331237792969e-05,
      "model_forward_time": 1.723442554473877,
      "step": 61838
    },
    {
      "epoch": 5.897331237792969e-05,
      "step": 61838,
      "training_step_time": 6.721768140792847
    },
    {
      "epoch": 5.897426605224609e-05,
      "model_forward_time": 1.357017993927002,
      "step": 61839
    },
    {
      "epoch": 5.897426605224609e-05,
      "step": 61839,
      "training_step_time": 6.694068431854248
    },
    {
      "epoch": 5.89752197265625e-05,
      "grad_norm": 0.1057182252407074,
      "learning_rate": 3.480254753817724e-06,
      "loss": 0.0562,
      "step": 61840
    },
    {
      "epoch": 5.89752197265625e-05,
      "model_forward_time": 1.7763135433197021,
      "step": 61840
    },
    {
      "dynamics_loss_avg": 0.054776782542467116,
      "epoch": 5.89752197265625e-05,
      "step": 61840
    },
    {
      "action_loss_avg": 0.0022458959836512806,
      "epoch": 5.89752197265625e-05,
      "step": 61840
    },
    {
      "epoch": 5.89752197265625e-05,
      "step": 61840,
      "training_step_time": 8.26837158203125
    },
    {
      "epoch": 5.897617340087891e-05,
      "model_forward_time": 2.2747457027435303,
      "step": 61841
    },
    {
      "epoch": 5.897617340087891e-05,
      "step": 61841,
      "training_step_time": 9.548009157180786
    },
    {
      "epoch": 5.897712707519531e-05,
      "model_forward_time": 1.7282512187957764,
      "step": 61842
    },
    {
      "epoch": 5.897712707519531e-05,
      "step": 61842,
      "training_step_time": 8.730872392654419
    },
    {
      "epoch": 5.897808074951172e-05,
      "model_forward_time": 1.7362914085388184,
      "step": 61843
    },
    {
      "epoch": 5.897808074951172e-05,
      "step": 61843,
      "training_step_time": 7.70017147064209
    },
    {
      "epoch": 5.8979034423828126e-05,
      "model_forward_time": 1.71952486038208,
      "step": 61844
    },
    {
      "epoch": 5.8979034423828126e-05,
      "step": 61844,
      "training_step_time": 6.666287899017334
    },
    {
      "epoch": 5.897998809814453e-05,
      "model_forward_time": 1.7245218753814697,
      "step": 61845
    },
    {
      "epoch": 5.897998809814453e-05,
      "step": 61845,
      "training_step_time": 6.706521034240723
    },
    {
      "epoch": 5.8980941772460934e-05,
      "model_forward_time": 1.7293479442596436,
      "step": 61846
    },
    {
      "epoch": 5.8980941772460934e-05,
      "step": 61846,
      "training_step_time": 6.765274524688721
    },
    {
      "epoch": 5.8981895446777345e-05,
      "model_forward_time": 1.7132847309112549,
      "step": 61847
    },
    {
      "epoch": 5.8981895446777345e-05,
      "step": 61847,
      "training_step_time": 6.696614980697632
    },
    {
      "epoch": 5.898284912109375e-05,
      "model_forward_time": 1.726473093032837,
      "step": 61848
    },
    {
      "epoch": 5.898284912109375e-05,
      "step": 61848,
      "training_step_time": 6.685350179672241
    },
    {
      "epoch": 5.898380279541015e-05,
      "model_forward_time": 1.7460756301879883,
      "step": 61849
    },
    {
      "epoch": 5.898380279541015e-05,
      "step": 61849,
      "training_step_time": 8.436968564987183
    },
    {
      "epoch": 5.8984756469726564e-05,
      "grad_norm": 0.4211907684803009,
      "learning_rate": 3.4786795956139045e-06,
      "loss": 0.0529,
      "step": 61850
    },
    {
      "epoch": 5.8984756469726564e-05,
      "model_forward_time": 2.2663402557373047,
      "step": 61850
    },
    {
      "dynamics_loss_avg": 0.05954748336225748,
      "epoch": 5.8984756469726564e-05,
      "step": 61850
    },
    {
      "action_loss_avg": 0.003248139377683401,
      "epoch": 5.8984756469726564e-05,
      "step": 61850
    },
    {
      "epoch": 5.8984756469726564e-05,
      "step": 61850,
      "training_step_time": 8.73480486869812
    },
    {
      "epoch": 5.898571014404297e-05,
      "model_forward_time": 1.7238147258758545,
      "step": 61851
    },
    {
      "epoch": 5.898571014404297e-05,
      "step": 61851,
      "training_step_time": 9.749958992004395
    },
    {
      "epoch": 5.898666381835937e-05,
      "model_forward_time": 1.725818157196045,
      "step": 61852
    },
    {
      "epoch": 5.898666381835937e-05,
      "step": 61852,
      "training_step_time": 7.7690558433532715
    },
    {
      "epoch": 5.898761749267578e-05,
      "model_forward_time": 1.7268850803375244,
      "step": 61853
    },
    {
      "epoch": 5.898761749267578e-05,
      "step": 61853,
      "training_step_time": 6.740744113922119
    },
    {
      "epoch": 5.898857116699219e-05,
      "model_forward_time": 1.7141542434692383,
      "step": 61854
    },
    {
      "epoch": 5.898857116699219e-05,
      "step": 61854,
      "training_step_time": 6.74893593788147
    },
    {
      "epoch": 5.898952484130859e-05,
      "model_forward_time": 1.7330303192138672,
      "step": 61855
    },
    {
      "epoch": 5.898952484130859e-05,
      "step": 61855,
      "training_step_time": 6.7689125537872314
    },
    {
      "epoch": 5.8990478515625e-05,
      "model_forward_time": 1.7302262783050537,
      "step": 61856
    },
    {
      "epoch": 5.8990478515625e-05,
      "step": 61856,
      "training_step_time": 6.691105365753174
    },
    {
      "epoch": 5.8991432189941406e-05,
      "model_forward_time": 1.7147564888000488,
      "step": 61857
    },
    {
      "epoch": 5.8991432189941406e-05,
      "step": 61857,
      "training_step_time": 6.662957668304443
    },
    {
      "epoch": 5.899238586425781e-05,
      "model_forward_time": 1.7254643440246582,
      "step": 61858
    },
    {
      "epoch": 5.899238586425781e-05,
      "step": 61858,
      "training_step_time": 7.237822532653809
    },
    {
      "epoch": 5.899333953857422e-05,
      "model_forward_time": 1.7132573127746582,
      "step": 61859
    },
    {
      "epoch": 5.899333953857422e-05,
      "step": 61859,
      "training_step_time": 7.615677118301392
    },
    {
      "epoch": 5.8994293212890624e-05,
      "grad_norm": 0.06997007876634598,
      "learning_rate": 3.4771046037793978e-06,
      "loss": 0.0567,
      "step": 61860
    },
    {
      "epoch": 5.8994293212890624e-05,
      "model_forward_time": 2.269256830215454,
      "step": 61860
    },
    {
      "dynamics_loss_avg": 0.04473572988063097,
      "epoch": 5.8994293212890624e-05,
      "step": 61860
    },
    {
      "action_loss_avg": 0.0012978464015759527,
      "epoch": 5.8994293212890624e-05,
      "step": 61860
    },
    {
      "epoch": 5.8994293212890624e-05,
      "step": 61860,
      "training_step_time": 9.294286489486694
    },
    {
      "epoch": 5.899524688720703e-05,
      "model_forward_time": 1.7230603694915771,
      "step": 61861
    },
    {
      "epoch": 5.899524688720703e-05,
      "step": 61861,
      "training_step_time": 7.7737507820129395
    },
    {
      "epoch": 5.899620056152344e-05,
      "model_forward_time": 1.732203722000122,
      "step": 61862
    },
    {
      "epoch": 5.899620056152344e-05,
      "step": 61862,
      "training_step_time": 7.578548908233643
    },
    {
      "epoch": 5.899715423583984e-05,
      "model_forward_time": 1.7242512702941895,
      "step": 61863
    },
    {
      "epoch": 5.899715423583984e-05,
      "step": 61863,
      "training_step_time": 6.757259845733643
    },
    {
      "epoch": 5.899810791015625e-05,
      "model_forward_time": 1.7157039642333984,
      "step": 61864
    },
    {
      "epoch": 5.899810791015625e-05,
      "step": 61864,
      "training_step_time": 6.705427169799805
    },
    {
      "epoch": 5.899906158447266e-05,
      "model_forward_time": 1.3600478172302246,
      "step": 61865
    },
    {
      "epoch": 5.899906158447266e-05,
      "step": 61865,
      "training_step_time": 6.717980861663818
    },
    {
      "epoch": 5.900001525878906e-05,
      "model_forward_time": 1.7305383682250977,
      "step": 61866
    },
    {
      "epoch": 5.900001525878906e-05,
      "step": 61866,
      "training_step_time": 6.747973203659058
    },
    {
      "epoch": 5.9000968933105466e-05,
      "model_forward_time": 1.7228026390075684,
      "step": 61867
    },
    {
      "epoch": 5.9000968933105466e-05,
      "step": 61867,
      "training_step_time": 7.197756290435791
    },
    {
      "epoch": 5.900192260742188e-05,
      "model_forward_time": 1.7756860256195068,
      "step": 61868
    },
    {
      "epoch": 5.900192260742188e-05,
      "step": 61868,
      "training_step_time": 7.6455769538879395
    },
    {
      "epoch": 5.900287628173828e-05,
      "model_forward_time": 2.255925178527832,
      "step": 61869
    },
    {
      "epoch": 5.900287628173828e-05,
      "step": 61869,
      "training_step_time": 7.8019702434539795
    },
    {
      "epoch": 5.9003829956054685e-05,
      "grad_norm": 0.08520283550024033,
      "learning_rate": 3.4755297784864415e-06,
      "loss": 0.0585,
      "step": 61870
    },
    {
      "epoch": 5.9003829956054685e-05,
      "model_forward_time": 1.7172164916992188,
      "step": 61870
    },
    {
      "dynamics_loss_avg": 0.05486204484477639,
      "epoch": 5.9003829956054685e-05,
      "step": 61870
    },
    {
      "action_loss_avg": 0.0015023084633867256,
      "epoch": 5.9003829956054685e-05,
      "step": 61870
    },
    {
      "epoch": 5.9003829956054685e-05,
      "step": 61870,
      "training_step_time": 8.682279825210571
    },
    {
      "epoch": 5.9004783630371096e-05,
      "model_forward_time": 1.724238395690918,
      "step": 61871
    },
    {
      "epoch": 5.9004783630371096e-05,
      "step": 61871,
      "training_step_time": 7.639775037765503
    },
    {
      "epoch": 5.90057373046875e-05,
      "model_forward_time": 1.723174810409546,
      "step": 61872
    },
    {
      "epoch": 5.90057373046875e-05,
      "step": 61872,
      "training_step_time": 6.7179179191589355
    },
    {
      "epoch": 5.9006690979003904e-05,
      "model_forward_time": 1.7293949127197266,
      "step": 61873
    },
    {
      "epoch": 5.9006690979003904e-05,
      "step": 61873,
      "training_step_time": 6.730522632598877
    },
    {
      "epoch": 5.9007644653320315e-05,
      "model_forward_time": 1.7239000797271729,
      "step": 61874
    },
    {
      "epoch": 5.9007644653320315e-05,
      "step": 61874,
      "training_step_time": 6.739778995513916
    },
    {
      "epoch": 5.900859832763672e-05,
      "model_forward_time": 1.7397372722625732,
      "step": 61875
    },
    {
      "epoch": 5.900859832763672e-05,
      "step": 61875,
      "training_step_time": 6.694769620895386
    },
    {
      "epoch": 5.900955200195312e-05,
      "model_forward_time": 1.7294938564300537,
      "step": 61876
    },
    {
      "epoch": 5.900955200195312e-05,
      "step": 61876,
      "training_step_time": 7.1287031173706055
    },
    {
      "epoch": 5.9010505676269534e-05,
      "model_forward_time": 1.7712812423706055,
      "step": 61877
    },
    {
      "epoch": 5.9010505676269534e-05,
      "step": 61877,
      "training_step_time": 8.402950763702393
    },
    {
      "epoch": 5.901145935058594e-05,
      "model_forward_time": 2.2704532146453857,
      "step": 61878
    },
    {
      "epoch": 5.901145935058594e-05,
      "step": 61878,
      "training_step_time": 9.585933923721313
    },
    {
      "epoch": 5.901241302490234e-05,
      "model_forward_time": 1.7325079441070557,
      "step": 61879
    },
    {
      "epoch": 5.901241302490234e-05,
      "step": 61879,
      "training_step_time": 9.160795450210571
    },
    {
      "epoch": 5.901336669921875e-05,
      "grad_norm": 0.12457912415266037,
      "learning_rate": 3.4739551199072575e-06,
      "loss": 0.059,
      "step": 61880
    },
    {
      "epoch": 5.901336669921875e-05,
      "model_forward_time": 1.726550579071045,
      "step": 61880
    },
    {
      "dynamics_loss_avg": 0.058021338656544685,
      "epoch": 5.901336669921875e-05,
      "step": 61880
    },
    {
      "action_loss_avg": 0.0024073278589639814,
      "epoch": 5.901336669921875e-05,
      "step": 61880
    },
    {
      "epoch": 5.901336669921875e-05,
      "step": 61880,
      "training_step_time": 8.547862768173218
    },
    {
      "epoch": 5.9014320373535156e-05,
      "model_forward_time": 1.7266476154327393,
      "step": 61881
    },
    {
      "epoch": 5.9014320373535156e-05,
      "step": 61881,
      "training_step_time": 6.674880743026733
    },
    {
      "epoch": 5.901527404785156e-05,
      "model_forward_time": 1.7211012840270996,
      "step": 61882
    },
    {
      "epoch": 5.901527404785156e-05,
      "step": 61882,
      "training_step_time": 6.693192481994629
    },
    {
      "epoch": 5.901622772216797e-05,
      "model_forward_time": 1.7216143608093262,
      "step": 61883
    },
    {
      "epoch": 5.901622772216797e-05,
      "step": 61883,
      "training_step_time": 6.775769472122192
    },
    {
      "epoch": 5.9017181396484375e-05,
      "model_forward_time": 1.7269196510314941,
      "step": 61884
    },
    {
      "epoch": 5.9017181396484375e-05,
      "step": 61884,
      "training_step_time": 6.610898971557617
    },
    {
      "epoch": 5.901813507080078e-05,
      "model_forward_time": 1.7175352573394775,
      "step": 61885
    },
    {
      "epoch": 5.901813507080078e-05,
      "step": 61885,
      "training_step_time": 7.081812858581543
    },
    {
      "epoch": 5.901908874511719e-05,
      "model_forward_time": 1.7607812881469727,
      "step": 61886
    },
    {
      "epoch": 5.901908874511719e-05,
      "step": 61886,
      "training_step_time": 8.181382656097412
    },
    {
      "epoch": 5.9020042419433594e-05,
      "model_forward_time": 2.2997217178344727,
      "step": 61887
    },
    {
      "epoch": 5.9020042419433594e-05,
      "step": 61887,
      "training_step_time": 7.733683109283447
    },
    {
      "epoch": 5.902099609375e-05,
      "model_forward_time": 1.7468807697296143,
      "step": 61888
    },
    {
      "epoch": 5.902099609375e-05,
      "step": 61888,
      "training_step_time": 8.647135972976685
    },
    {
      "epoch": 5.902194976806641e-05,
      "model_forward_time": 1.716268539428711,
      "step": 61889
    },
    {
      "epoch": 5.902194976806641e-05,
      "step": 61889,
      "training_step_time": 7.835122585296631
    },
    {
      "epoch": 5.902290344238281e-05,
      "grad_norm": 0.10826131701469421,
      "learning_rate": 3.472380628214046e-06,
      "loss": 0.0556,
      "step": 61890
    },
    {
      "epoch": 5.902290344238281e-05,
      "model_forward_time": 1.705690622329712,
      "step": 61890
    },
    {
      "dynamics_loss_avg": 0.05865730196237564,
      "epoch": 5.902290344238281e-05,
      "step": 61890
    },
    {
      "action_loss_avg": 0.0014573658350855111,
      "epoch": 5.902290344238281e-05,
      "step": 61890
    },
    {
      "epoch": 5.902290344238281e-05,
      "step": 61890,
      "training_step_time": 7.5847907066345215
    },
    {
      "epoch": 5.902385711669922e-05,
      "model_forward_time": 1.7240300178527832,
      "step": 61891
    },
    {
      "epoch": 5.902385711669922e-05,
      "step": 61891,
      "training_step_time": 6.780450105667114
    },
    {
      "epoch": 5.902481079101563e-05,
      "model_forward_time": 1.7114293575286865,
      "step": 61892
    },
    {
      "epoch": 5.902481079101563e-05,
      "step": 61892,
      "training_step_time": 6.7350382804870605
    },
    {
      "epoch": 5.902576446533203e-05,
      "model_forward_time": 1.7141430377960205,
      "step": 61893
    },
    {
      "epoch": 5.902576446533203e-05,
      "step": 61893,
      "training_step_time": 6.809547185897827
    },
    {
      "epoch": 5.9026718139648436e-05,
      "model_forward_time": 1.7145791053771973,
      "step": 61894
    },
    {
      "epoch": 5.9026718139648436e-05,
      "step": 61894,
      "training_step_time": 7.051452875137329
    },
    {
      "epoch": 5.902767181396485e-05,
      "model_forward_time": 1.7315316200256348,
      "step": 61895
    },
    {
      "epoch": 5.902767181396485e-05,
      "step": 61895,
      "training_step_time": 7.5823142528533936
    },
    {
      "epoch": 5.902862548828125e-05,
      "model_forward_time": 2.2723309993743896,
      "step": 61896
    },
    {
      "epoch": 5.902862548828125e-05,
      "step": 61896,
      "training_step_time": 8.71328616142273
    },
    {
      "epoch": 5.9029579162597655e-05,
      "model_forward_time": 1.7059905529022217,
      "step": 61897
    },
    {
      "epoch": 5.9029579162597655e-05,
      "step": 61897,
      "training_step_time": 8.759195566177368
    },
    {
      "epoch": 5.9030532836914066e-05,
      "model_forward_time": 1.7168059349060059,
      "step": 61898
    },
    {
      "epoch": 5.9030532836914066e-05,
      "step": 61898,
      "training_step_time": 7.74052357673645
    },
    {
      "epoch": 5.903148651123047e-05,
      "model_forward_time": 1.714242696762085,
      "step": 61899
    },
    {
      "epoch": 5.903148651123047e-05,
      "step": 61899,
      "training_step_time": 7.613927125930786
    },
    {
      "epoch": 5.9032440185546874e-05,
      "grad_norm": 0.1104646697640419,
      "learning_rate": 3.4708063035789906e-06,
      "loss": 0.0569,
      "step": 61900
    },
    {
      "epoch": 5.9032440185546874e-05,
      "model_forward_time": 1.7010071277618408,
      "step": 61900
    },
    {
      "dynamics_loss_avg": 0.0496498960070312,
      "epoch": 5.9032440185546874e-05,
      "step": 61900
    },
    {
      "action_loss_avg": 0.0012415262899594381,
      "epoch": 5.9032440185546874e-05,
      "step": 61900
    },
    {
      "epoch": 5.9032440185546874e-05,
      "step": 61900,
      "training_step_time": 6.693763971328735
    },
    {
      "epoch": 5.9033393859863284e-05,
      "model_forward_time": 1.7167506217956543,
      "step": 61901
    },
    {
      "epoch": 5.9033393859863284e-05,
      "step": 61901,
      "training_step_time": 6.729224443435669
    },
    {
      "epoch": 5.903434753417969e-05,
      "model_forward_time": 1.7230372428894043,
      "step": 61902
    },
    {
      "epoch": 5.903434753417969e-05,
      "step": 61902,
      "training_step_time": 6.752725839614868
    },
    {
      "epoch": 5.903530120849609e-05,
      "model_forward_time": 1.7181792259216309,
      "step": 61903
    },
    {
      "epoch": 5.903530120849609e-05,
      "step": 61903,
      "training_step_time": 7.14110803604126
    },
    {
      "epoch": 5.90362548828125e-05,
      "model_forward_time": 1.7159464359283447,
      "step": 61904
    },
    {
      "epoch": 5.90362548828125e-05,
      "step": 61904,
      "training_step_time": 7.6192896366119385
    },
    {
      "epoch": 5.903720855712891e-05,
      "model_forward_time": 2.2420239448547363,
      "step": 61905
    },
    {
      "epoch": 5.903720855712891e-05,
      "step": 61905,
      "training_step_time": 9.624450445175171
    },
    {
      "epoch": 5.903816223144531e-05,
      "model_forward_time": 1.7148194313049316,
      "step": 61906
    },
    {
      "epoch": 5.903816223144531e-05,
      "step": 61906,
      "training_step_time": 8.474621534347534
    },
    {
      "epoch": 5.9039115905761715e-05,
      "model_forward_time": 1.7099497318267822,
      "step": 61907
    },
    {
      "epoch": 5.9039115905761715e-05,
      "step": 61907,
      "training_step_time": 7.906794548034668
    },
    {
      "epoch": 5.9040069580078126e-05,
      "model_forward_time": 1.7199177742004395,
      "step": 61908
    },
    {
      "epoch": 5.9040069580078126e-05,
      "step": 61908,
      "training_step_time": 7.796513795852661
    },
    {
      "epoch": 5.904102325439453e-05,
      "model_forward_time": 1.7187840938568115,
      "step": 61909
    },
    {
      "epoch": 5.904102325439453e-05,
      "step": 61909,
      "training_step_time": 6.888118505477905
    },
    {
      "epoch": 5.9041976928710934e-05,
      "grad_norm": 0.1517791450023651,
      "learning_rate": 3.469232146174261e-06,
      "loss": 0.0581,
      "step": 61910
    },
    {
      "epoch": 5.9041976928710934e-05,
      "model_forward_time": 1.709519624710083,
      "step": 61910
    },
    {
      "dynamics_loss_avg": 0.060510692000389096,
      "epoch": 5.9041976928710934e-05,
      "step": 61910
    },
    {
      "action_loss_avg": 0.0014473141811322421,
      "epoch": 5.9041976928710934e-05,
      "step": 61910
    },
    {
      "epoch": 5.9041976928710934e-05,
      "step": 61910,
      "training_step_time": 6.842176198959351
    },
    {
      "epoch": 5.9042930603027345e-05,
      "model_forward_time": 1.7184462547302246,
      "step": 61911
    },
    {
      "epoch": 5.9042930603027345e-05,
      "step": 61911,
      "training_step_time": 6.839755296707153
    },
    {
      "epoch": 5.904388427734375e-05,
      "model_forward_time": 1.7204763889312744,
      "step": 61912
    },
    {
      "epoch": 5.904388427734375e-05,
      "step": 61912,
      "training_step_time": 7.209302186965942
    },
    {
      "epoch": 5.904483795166015e-05,
      "model_forward_time": 1.7065973281860352,
      "step": 61913
    },
    {
      "epoch": 5.904483795166015e-05,
      "step": 61913,
      "training_step_time": 7.638451337814331
    },
    {
      "epoch": 5.9045791625976564e-05,
      "model_forward_time": 1.7580053806304932,
      "step": 61914
    },
    {
      "epoch": 5.9045791625976564e-05,
      "step": 61914,
      "training_step_time": 9.846263408660889
    },
    {
      "epoch": 5.904674530029297e-05,
      "model_forward_time": 2.258082151412964,
      "step": 61915
    },
    {
      "epoch": 5.904674530029297e-05,
      "step": 61915,
      "training_step_time": 8.353001594543457
    },
    {
      "epoch": 5.904769897460937e-05,
      "model_forward_time": 1.701270341873169,
      "step": 61916
    },
    {
      "epoch": 5.904769897460937e-05,
      "step": 61916,
      "training_step_time": 9.66930341720581
    },
    {
      "epoch": 5.904865264892578e-05,
      "model_forward_time": 1.735830545425415,
      "step": 61917
    },
    {
      "epoch": 5.904865264892578e-05,
      "step": 61917,
      "training_step_time": 7.774382829666138
    },
    {
      "epoch": 5.904960632324219e-05,
      "model_forward_time": 1.7240986824035645,
      "step": 61918
    },
    {
      "epoch": 5.904960632324219e-05,
      "step": 61918,
      "training_step_time": 6.927961826324463
    },
    {
      "epoch": 5.905055999755859e-05,
      "model_forward_time": 1.7353978157043457,
      "step": 61919
    },
    {
      "epoch": 5.905055999755859e-05,
      "step": 61919,
      "training_step_time": 6.862066745758057
    },
    {
      "epoch": 5.9051513671875e-05,
      "grad_norm": 0.08534973114728928,
      "learning_rate": 3.4676581561720024e-06,
      "loss": 0.0569,
      "step": 61920
    },
    {
      "epoch": 5.9051513671875e-05,
      "model_forward_time": 1.7080762386322021,
      "step": 61920
    },
    {
      "dynamics_loss_avg": 0.05544060692191124,
      "epoch": 5.9051513671875e-05,
      "step": 61920
    },
    {
      "action_loss_avg": 0.0022755662153940647,
      "epoch": 5.9051513671875e-05,
      "step": 61920
    },
    {
      "epoch": 5.9051513671875e-05,
      "step": 61920,
      "training_step_time": 6.838881015777588
    },
    {
      "epoch": 5.9052467346191406e-05,
      "model_forward_time": 1.711899995803833,
      "step": 61921
    },
    {
      "epoch": 5.9052467346191406e-05,
      "step": 61921,
      "training_step_time": 6.923401832580566
    },
    {
      "epoch": 5.905342102050781e-05,
      "model_forward_time": 1.7061700820922852,
      "step": 61922
    },
    {
      "epoch": 5.905342102050781e-05,
      "step": 61922,
      "training_step_time": 7.72536563873291
    },
    {
      "epoch": 5.905437469482422e-05,
      "model_forward_time": 1.751577615737915,
      "step": 61923
    },
    {
      "epoch": 5.905437469482422e-05,
      "step": 61923,
      "training_step_time": 9.314295291900635
    },
    {
      "epoch": 5.9055328369140625e-05,
      "model_forward_time": 2.2597036361694336,
      "step": 61924
    },
    {
      "epoch": 5.9055328369140625e-05,
      "step": 61924,
      "training_step_time": 8.838312149047852
    },
    {
      "epoch": 5.905628204345703e-05,
      "model_forward_time": 1.357051134109497,
      "step": 61925
    },
    {
      "epoch": 5.905628204345703e-05,
      "step": 61925,
      "training_step_time": 8.91261339187622
    },
    {
      "epoch": 5.905723571777344e-05,
      "model_forward_time": 1.709782600402832,
      "step": 61926
    },
    {
      "epoch": 5.905723571777344e-05,
      "step": 61926,
      "training_step_time": 9.41570520401001
    },
    {
      "epoch": 5.905818939208984e-05,
      "model_forward_time": 1.718580961227417,
      "step": 61927
    },
    {
      "epoch": 5.905818939208984e-05,
      "step": 61927,
      "training_step_time": 6.899003028869629
    },
    {
      "epoch": 5.905914306640625e-05,
      "model_forward_time": 1.7222437858581543,
      "step": 61928
    },
    {
      "epoch": 5.905914306640625e-05,
      "step": 61928,
      "training_step_time": 6.790833234786987
    },
    {
      "epoch": 5.906009674072266e-05,
      "model_forward_time": 1.7273426055908203,
      "step": 61929
    },
    {
      "epoch": 5.906009674072266e-05,
      "step": 61929,
      "training_step_time": 6.857674837112427
    },
    {
      "epoch": 5.906105041503906e-05,
      "grad_norm": 0.09542260318994522,
      "learning_rate": 3.466084333744344e-06,
      "loss": 0.0558,
      "step": 61930
    },
    {
      "epoch": 5.906105041503906e-05,
      "model_forward_time": 1.7177987098693848,
      "step": 61930
    },
    {
      "dynamics_loss_avg": 0.05832006148993969,
      "epoch": 5.906105041503906e-05,
      "step": 61930
    },
    {
      "action_loss_avg": 0.0015267888666130602,
      "epoch": 5.906105041503906e-05,
      "step": 61930
    },
    {
      "epoch": 5.906105041503906e-05,
      "step": 61930,
      "training_step_time": 6.888383865356445
    },
    {
      "epoch": 5.9062004089355466e-05,
      "model_forward_time": 1.7178716659545898,
      "step": 61931
    },
    {
      "epoch": 5.9062004089355466e-05,
      "step": 61931,
      "training_step_time": 7.684154033660889
    },
    {
      "epoch": 5.906295776367188e-05,
      "model_forward_time": 1.7787244319915771,
      "step": 61932
    },
    {
      "epoch": 5.906295776367188e-05,
      "step": 61932,
      "training_step_time": 7.693694829940796
    },
    {
      "epoch": 5.906391143798828e-05,
      "model_forward_time": 2.2809927463531494,
      "step": 61933
    },
    {
      "epoch": 5.906391143798828e-05,
      "step": 61933,
      "training_step_time": 8.795565366744995
    },
    {
      "epoch": 5.9064865112304685e-05,
      "model_forward_time": 1.7267417907714844,
      "step": 61934
    },
    {
      "epoch": 5.9064865112304685e-05,
      "step": 61934,
      "training_step_time": 9.516169786453247
    },
    {
      "epoch": 5.9065818786621096e-05,
      "model_forward_time": 1.715867519378662,
      "step": 61935
    },
    {
      "epoch": 5.9065818786621096e-05,
      "step": 61935,
      "training_step_time": 9.61339020729065
    },
    {
      "epoch": 5.90667724609375e-05,
      "model_forward_time": 1.72672700881958,
      "step": 61936
    },
    {
      "epoch": 5.90667724609375e-05,
      "step": 61936,
      "training_step_time": 7.3901472091674805
    },
    {
      "epoch": 5.9067726135253904e-05,
      "model_forward_time": 1.719092845916748,
      "step": 61937
    },
    {
      "epoch": 5.9067726135253904e-05,
      "step": 61937,
      "training_step_time": 6.906783580780029
    },
    {
      "epoch": 5.9068679809570315e-05,
      "model_forward_time": 1.726492166519165,
      "step": 61938
    },
    {
      "epoch": 5.9068679809570315e-05,
      "step": 61938,
      "training_step_time": 6.8946757316589355
    },
    {
      "epoch": 5.906963348388672e-05,
      "model_forward_time": 1.7187707424163818,
      "step": 61939
    },
    {
      "epoch": 5.906963348388672e-05,
      "step": 61939,
      "training_step_time": 6.846560716629028
    },
    {
      "epoch": 5.907058715820312e-05,
      "grad_norm": 0.08272390067577362,
      "learning_rate": 3.4645106790633965e-06,
      "loss": 0.0557,
      "step": 61940
    },
    {
      "epoch": 5.907058715820312e-05,
      "model_forward_time": 1.7315447330474854,
      "step": 61940
    },
    {
      "dynamics_loss_avg": 0.05630453620105982,
      "epoch": 5.907058715820312e-05,
      "step": 61940
    },
    {
      "action_loss_avg": 0.0016119788197102025,
      "epoch": 5.907058715820312e-05,
      "step": 61940
    },
    {
      "epoch": 5.907058715820312e-05,
      "step": 61940,
      "training_step_time": 7.636483430862427
    },
    {
      "epoch": 5.9071540832519534e-05,
      "model_forward_time": 1.7635371685028076,
      "step": 61941
    },
    {
      "epoch": 5.9071540832519534e-05,
      "step": 61941,
      "training_step_time": 9.128829002380371
    },
    {
      "epoch": 5.907249450683594e-05,
      "model_forward_time": 2.287848949432373,
      "step": 61942
    },
    {
      "epoch": 5.907249450683594e-05,
      "step": 61942,
      "training_step_time": 8.406510829925537
    },
    {
      "epoch": 5.907344818115234e-05,
      "model_forward_time": 1.748283863067627,
      "step": 61943
    },
    {
      "epoch": 5.907344818115234e-05,
      "step": 61943,
      "training_step_time": 8.603092432022095
    },
    {
      "epoch": 5.907440185546875e-05,
      "model_forward_time": 1.7370834350585938,
      "step": 61944
    },
    {
      "epoch": 5.907440185546875e-05,
      "step": 61944,
      "training_step_time": 9.777430295944214
    },
    {
      "epoch": 5.9075355529785157e-05,
      "model_forward_time": 1.7283203601837158,
      "step": 61945
    },
    {
      "epoch": 5.9075355529785157e-05,
      "step": 61945,
      "training_step_time": 7.751560926437378
    },
    {
      "epoch": 5.907630920410156e-05,
      "model_forward_time": 1.7330162525177002,
      "step": 61946
    },
    {
      "epoch": 5.907630920410156e-05,
      "step": 61946,
      "training_step_time": 6.874293565750122
    },
    {
      "epoch": 5.907726287841797e-05,
      "model_forward_time": 1.747896671295166,
      "step": 61947
    },
    {
      "epoch": 5.907726287841797e-05,
      "step": 61947,
      "training_step_time": 6.913072109222412
    },
    {
      "epoch": 5.9078216552734375e-05,
      "model_forward_time": 1.3782761096954346,
      "step": 61948
    },
    {
      "epoch": 5.9078216552734375e-05,
      "step": 61948,
      "training_step_time": 6.922386169433594
    },
    {
      "epoch": 5.907917022705078e-05,
      "model_forward_time": 1.743551254272461,
      "step": 61949
    },
    {
      "epoch": 5.907917022705078e-05,
      "step": 61949,
      "training_step_time": 7.628103971481323
    },
    {
      "epoch": 5.908012390136719e-05,
      "grad_norm": 0.08246757090091705,
      "learning_rate": 3.462937192301253e-06,
      "loss": 0.0584,
      "step": 61950
    },
    {
      "epoch": 5.908012390136719e-05,
      "model_forward_time": 1.752659559249878,
      "step": 61950
    },
    {
      "dynamics_loss_avg": 0.06837061867117881,
      "epoch": 5.908012390136719e-05,
      "step": 61950
    },
    {
      "action_loss_avg": 0.0018558656389359384,
      "epoch": 5.908012390136719e-05,
      "step": 61950
    },
    {
      "epoch": 5.908012390136719e-05,
      "step": 61950,
      "training_step_time": 7.6528160572052
    },
    {
      "epoch": 5.9081077575683594e-05,
      "model_forward_time": 1.7463359832763672,
      "step": 61951
    },
    {
      "epoch": 5.9081077575683594e-05,
      "step": 61951,
      "training_step_time": 8.610755681991577
    },
    {
      "epoch": 5.908203125e-05,
      "model_forward_time": 2.343188524246216,
      "step": 61952
    },
    {
      "epoch": 5.908203125e-05,
      "step": 61952,
      "training_step_time": 9.532227754592896
    },
    {
      "epoch": 5.908298492431641e-05,
      "model_forward_time": 1.8154892921447754,
      "step": 61953
    },
    {
      "epoch": 5.908298492431641e-05,
      "step": 61953,
      "training_step_time": 8.77477216720581
    },
    {
      "epoch": 5.908393859863281e-05,
      "model_forward_time": 1.8293044567108154,
      "step": 61954
    },
    {
      "epoch": 5.908393859863281e-05,
      "step": 61954,
      "training_step_time": 7.573142051696777
    },
    {
      "epoch": 5.908489227294922e-05,
      "model_forward_time": 1.8411743640899658,
      "step": 61955
    },
    {
      "epoch": 5.908489227294922e-05,
      "step": 61955,
      "training_step_time": 6.907615661621094
    },
    {
      "epoch": 5.908584594726563e-05,
      "model_forward_time": 1.8479251861572266,
      "step": 61956
    },
    {
      "epoch": 5.908584594726563e-05,
      "step": 61956,
      "training_step_time": 6.879724740982056
    },
    {
      "epoch": 5.908679962158203e-05,
      "model_forward_time": 1.8316268920898438,
      "step": 61957
    },
    {
      "epoch": 5.908679962158203e-05,
      "step": 61957,
      "training_step_time": 6.830859184265137
    },
    {
      "epoch": 5.9087753295898436e-05,
      "model_forward_time": 1.8308606147766113,
      "step": 61958
    },
    {
      "epoch": 5.9087753295898436e-05,
      "step": 61958,
      "training_step_time": 7.260702610015869
    },
    {
      "epoch": 5.908870697021485e-05,
      "model_forward_time": 1.8024883270263672,
      "step": 61959
    },
    {
      "epoch": 5.908870697021485e-05,
      "step": 61959,
      "training_step_time": 8.208539485931396
    },
    {
      "epoch": 5.908966064453125e-05,
      "grad_norm": 0.09026409685611725,
      "learning_rate": 3.461363873629986e-06,
      "loss": 0.0577,
      "step": 61960
    },
    {
      "epoch": 5.908966064453125e-05,
      "model_forward_time": 1.8191280364990234,
      "step": 61960
    },
    {
      "dynamics_loss_avg": 0.0591543834656477,
      "epoch": 5.908966064453125e-05,
      "step": 61960
    },
    {
      "action_loss_avg": 0.0013494474085746333,
      "epoch": 5.908966064453125e-05,
      "step": 61960
    },
    {
      "epoch": 5.908966064453125e-05,
      "step": 61960,
      "training_step_time": 7.660213470458984
    },
    {
      "epoch": 5.9090614318847655e-05,
      "model_forward_time": 2.334263563156128,
      "step": 61961
    },
    {
      "epoch": 5.9090614318847655e-05,
      "step": 61961,
      "training_step_time": 8.7643301486969
    },
    {
      "epoch": 5.9091567993164066e-05,
      "model_forward_time": 1.8150553703308105,
      "step": 61962
    },
    {
      "epoch": 5.9091567993164066e-05,
      "step": 61962,
      "training_step_time": 8.690959453582764
    },
    {
      "epoch": 5.909252166748047e-05,
      "model_forward_time": 1.8099415302276611,
      "step": 61963
    },
    {
      "epoch": 5.909252166748047e-05,
      "step": 61963,
      "training_step_time": 7.731725692749023
    },
    {
      "epoch": 5.9093475341796874e-05,
      "model_forward_time": 1.807199239730835,
      "step": 61964
    },
    {
      "epoch": 5.9093475341796874e-05,
      "step": 61964,
      "training_step_time": 6.876277923583984
    },
    {
      "epoch": 5.9094429016113285e-05,
      "model_forward_time": 1.8122856616973877,
      "step": 61965
    },
    {
      "epoch": 5.9094429016113285e-05,
      "step": 61965,
      "training_step_time": 6.881896018981934
    },
    {
      "epoch": 5.909538269042969e-05,
      "model_forward_time": 1.812098503112793,
      "step": 61966
    },
    {
      "epoch": 5.909538269042969e-05,
      "step": 61966,
      "training_step_time": 6.902825832366943
    },
    {
      "epoch": 5.909633636474609e-05,
      "model_forward_time": 1.7949271202087402,
      "step": 61967
    },
    {
      "epoch": 5.909633636474609e-05,
      "step": 61967,
      "training_step_time": 7.294326066970825
    },
    {
      "epoch": 5.90972900390625e-05,
      "model_forward_time": 1.8136374950408936,
      "step": 61968
    },
    {
      "epoch": 5.90972900390625e-05,
      "step": 61968,
      "training_step_time": 7.654340982437134
    },
    {
      "epoch": 5.909824371337891e-05,
      "model_forward_time": 1.809041976928711,
      "step": 61969
    },
    {
      "epoch": 5.909824371337891e-05,
      "step": 61969,
      "training_step_time": 7.657820463180542
    },
    {
      "epoch": 5.909919738769531e-05,
      "grad_norm": 0.09031803160905838,
      "learning_rate": 3.4597907232216522e-06,
      "loss": 0.0557,
      "step": 61970
    },
    {
      "epoch": 5.909919738769531e-05,
      "model_forward_time": 2.3200833797454834,
      "step": 61970
    },
    {
      "dynamics_loss_avg": 0.05670187342911959,
      "epoch": 5.909919738769531e-05,
      "step": 61970
    },
    {
      "action_loss_avg": 0.0026107856596354396,
      "epoch": 5.909919738769531e-05,
      "step": 61970
    },
    {
      "epoch": 5.909919738769531e-05,
      "step": 61970,
      "training_step_time": 8.732325792312622
    },
    {
      "epoch": 5.9100151062011715e-05,
      "model_forward_time": 1.7994987964630127,
      "step": 61971
    },
    {
      "epoch": 5.9100151062011715e-05,
      "step": 61971,
      "training_step_time": 7.735753536224365
    },
    {
      "epoch": 5.9101104736328126e-05,
      "model_forward_time": 1.8203516006469727,
      "step": 61972
    },
    {
      "epoch": 5.9101104736328126e-05,
      "step": 61972,
      "training_step_time": 7.973353385925293
    },
    {
      "epoch": 5.910205841064453e-05,
      "model_forward_time": 1.8135435581207275,
      "step": 61973
    },
    {
      "epoch": 5.910205841064453e-05,
      "step": 61973,
      "training_step_time": 7.616612434387207
    },
    {
      "epoch": 5.9103012084960934e-05,
      "model_forward_time": 1.8032505512237549,
      "step": 61974
    },
    {
      "epoch": 5.9103012084960934e-05,
      "step": 61974,
      "training_step_time": 6.89192271232605
    },
    {
      "epoch": 5.9103965759277345e-05,
      "model_forward_time": 1.7883479595184326,
      "step": 61975
    },
    {
      "epoch": 5.9103965759277345e-05,
      "step": 61975,
      "training_step_time": 6.880498647689819
    },
    {
      "epoch": 5.910491943359375e-05,
      "model_forward_time": 1.7236788272857666,
      "step": 61976
    },
    {
      "epoch": 5.910491943359375e-05,
      "step": 61976,
      "training_step_time": 7.314001560211182
    },
    {
      "epoch": 5.910587310791015e-05,
      "model_forward_time": 1.7867915630340576,
      "step": 61977
    },
    {
      "epoch": 5.910587310791015e-05,
      "step": 61977,
      "training_step_time": 7.6432435512542725
    },
    {
      "epoch": 5.9106826782226564e-05,
      "model_forward_time": 1.8025012016296387,
      "step": 61978
    },
    {
      "epoch": 5.9106826782226564e-05,
      "step": 61978,
      "training_step_time": 8.16511082649231
    },
    {
      "epoch": 5.910778045654297e-05,
      "model_forward_time": 2.3308873176574707,
      "step": 61979
    },
    {
      "epoch": 5.910778045654297e-05,
      "step": 61979,
      "training_step_time": 10.843690156936646
    },
    {
      "epoch": 5.910873413085937e-05,
      "grad_norm": 0.09568366408348083,
      "learning_rate": 3.4582177412482904e-06,
      "loss": 0.0561,
      "step": 61980
    },
    {
      "epoch": 5.910873413085937e-05,
      "model_forward_time": 1.7954003810882568,
      "step": 61980
    },
    {
      "dynamics_loss_avg": 0.06019533164799214,
      "epoch": 5.910873413085937e-05,
      "step": 61980
    },
    {
      "action_loss_avg": 0.0013110656815115362,
      "epoch": 5.910873413085937e-05,
      "step": 61980
    },
    {
      "epoch": 5.910873413085937e-05,
      "step": 61980,
      "training_step_time": 7.746943712234497
    },
    {
      "epoch": 5.910968780517578e-05,
      "model_forward_time": 1.8141734600067139,
      "step": 61981
    },
    {
      "epoch": 5.910968780517578e-05,
      "step": 61981,
      "training_step_time": 7.870980501174927
    },
    {
      "epoch": 5.911064147949219e-05,
      "model_forward_time": 1.4356215000152588,
      "step": 61982
    },
    {
      "epoch": 5.911064147949219e-05,
      "step": 61982,
      "training_step_time": 7.7289814949035645
    },
    {
      "epoch": 5.911159515380859e-05,
      "model_forward_time": 1.804335117340088,
      "step": 61983
    },
    {
      "epoch": 5.911159515380859e-05,
      "step": 61983,
      "training_step_time": 6.851034164428711
    },
    {
      "epoch": 5.9112548828125e-05,
      "model_forward_time": 1.827653169631958,
      "step": 61984
    },
    {
      "epoch": 5.9112548828125e-05,
      "step": 61984,
      "training_step_time": 6.850706338882446
    },
    {
      "epoch": 5.9113502502441406e-05,
      "model_forward_time": 1.8245317935943604,
      "step": 61985
    },
    {
      "epoch": 5.9113502502441406e-05,
      "step": 61985,
      "training_step_time": 7.201760292053223
    },
    {
      "epoch": 5.911445617675781e-05,
      "model_forward_time": 1.8311700820922852,
      "step": 61986
    },
    {
      "epoch": 5.911445617675781e-05,
      "step": 61986,
      "training_step_time": 8.278134107589722
    },
    {
      "epoch": 5.911540985107422e-05,
      "model_forward_time": 1.820620059967041,
      "step": 61987
    },
    {
      "epoch": 5.911540985107422e-05,
      "step": 61987,
      "training_step_time": 7.593740224838257
    },
    {
      "epoch": 5.9116363525390625e-05,
      "model_forward_time": 1.824596643447876,
      "step": 61988
    },
    {
      "epoch": 5.9116363525390625e-05,
      "step": 61988,
      "training_step_time": 9.22343111038208
    },
    {
      "epoch": 5.911731719970703e-05,
      "model_forward_time": 1.829148769378662,
      "step": 61989
    },
    {
      "epoch": 5.911731719970703e-05,
      "step": 61989,
      "training_step_time": 8.704240560531616
    },
    {
      "epoch": 5.911827087402344e-05,
      "grad_norm": 0.08839419484138489,
      "learning_rate": 3.456644927881918e-06,
      "loss": 0.0588,
      "step": 61990
    },
    {
      "epoch": 5.911827087402344e-05,
      "model_forward_time": 1.8236877918243408,
      "step": 61990
    },
    {
      "dynamics_loss_avg": 0.053030490316450594,
      "epoch": 5.911827087402344e-05,
      "step": 61990
    },
    {
      "action_loss_avg": 0.002356167632387951,
      "epoch": 5.911827087402344e-05,
      "step": 61990
    },
    {
      "epoch": 5.911827087402344e-05,
      "step": 61990,
      "training_step_time": 7.886664628982544
    },
    {
      "epoch": 5.9119224548339843e-05,
      "model_forward_time": 1.8160943984985352,
      "step": 61991
    },
    {
      "epoch": 5.9119224548339843e-05,
      "step": 61991,
      "training_step_time": 7.776195287704468
    },
    {
      "epoch": 5.912017822265625e-05,
      "model_forward_time": 1.8422152996063232,
      "step": 61992
    },
    {
      "epoch": 5.912017822265625e-05,
      "step": 61992,
      "training_step_time": 6.912129878997803
    },
    {
      "epoch": 5.912113189697266e-05,
      "model_forward_time": 1.834608554840088,
      "step": 61993
    },
    {
      "epoch": 5.912113189697266e-05,
      "step": 61993,
      "training_step_time": 6.9294047355651855
    },
    {
      "epoch": 5.912208557128906e-05,
      "model_forward_time": 1.8219225406646729,
      "step": 61994
    },
    {
      "epoch": 5.912208557128906e-05,
      "step": 61994,
      "training_step_time": 7.295999050140381
    },
    {
      "epoch": 5.9123039245605466e-05,
      "model_forward_time": 1.8342981338500977,
      "step": 61995
    },
    {
      "epoch": 5.9123039245605466e-05,
      "step": 61995,
      "training_step_time": 7.202049732208252
    },
    {
      "epoch": 5.912399291992188e-05,
      "model_forward_time": 1.7734971046447754,
      "step": 61996
    },
    {
      "epoch": 5.912399291992188e-05,
      "step": 61996,
      "training_step_time": 7.680732250213623
    },
    {
      "epoch": 5.912494659423828e-05,
      "model_forward_time": 1.805110216140747,
      "step": 61997
    },
    {
      "epoch": 5.912494659423828e-05,
      "step": 61997,
      "training_step_time": 9.36227536201477
    },
    {
      "epoch": 5.9125900268554685e-05,
      "model_forward_time": 1.8095252513885498,
      "step": 61998
    },
    {
      "epoch": 5.9125900268554685e-05,
      "step": 61998,
      "training_step_time": 8.616199254989624
    },
    {
      "epoch": 5.9126853942871096e-05,
      "model_forward_time": 1.8169701099395752,
      "step": 61999
    },
    {
      "epoch": 5.9126853942871096e-05,
      "step": 61999,
      "training_step_time": 9.07158088684082
    },
    {
      "epoch": 5.91278076171875e-05,
      "grad_norm": 0.07813797891139984,
      "learning_rate": 3.4550722832945355e-06,
      "loss": 0.0576,
      "step": 62000
    },
    {
      "epoch": 5.91278076171875e-05,
      "model_forward_time": 1.7239620685577393,
      "step": 62000
    },
    {
      "dynamics_loss_avg": 0.06021380815654993,
      "epoch": 5.91278076171875e-05,
      "step": 62000
    },
    {
      "action_loss_avg": 0.0023756009817589075,
      "epoch": 5.91278076171875e-05,
      "step": 62000
    },
    {
      "epoch": 5.91278076171875e-05,
      "step": 62000,
      "training_step_time": 7.253293514251709
    },
    {
      "epoch": 5.9128761291503904e-05,
      "model_forward_time": 1.7229743003845215,
      "step": 62001
    },
    {
      "epoch": 5.9128761291503904e-05,
      "step": 62001,
      "training_step_time": 6.87313985824585
    },
    {
      "epoch": 5.9129714965820315e-05,
      "model_forward_time": 1.717790126800537,
      "step": 62002
    },
    {
      "epoch": 5.9129714965820315e-05,
      "step": 62002,
      "training_step_time": 6.912562131881714
    },
    {
      "epoch": 5.913066864013672e-05,
      "model_forward_time": 1.7260234355926514,
      "step": 62003
    },
    {
      "epoch": 5.913066864013672e-05,
      "step": 62003,
      "training_step_time": 7.6722376346588135
    },
    {
      "epoch": 5.913162231445312e-05,
      "model_forward_time": 1.7169115543365479,
      "step": 62004
    },
    {
      "epoch": 5.913162231445312e-05,
      "step": 62004,
      "training_step_time": 7.2901623249053955
    },
    {
      "epoch": 5.9132575988769534e-05,
      "model_forward_time": 1.7692451477050781,
      "step": 62005
    },
    {
      "epoch": 5.9132575988769534e-05,
      "step": 62005,
      "training_step_time": 8.328933238983154
    },
    {
      "epoch": 5.913352966308594e-05,
      "model_forward_time": 2.7100019454956055,
      "step": 62006
    },
    {
      "epoch": 5.913352966308594e-05,
      "step": 62006,
      "training_step_time": 10.828257083892822
    },
    {
      "epoch": 5.913448333740234e-05,
      "model_forward_time": 1.7107653617858887,
      "step": 62007
    },
    {
      "epoch": 5.913448333740234e-05,
      "step": 62007,
      "training_step_time": 11.85682487487793
    },
    {
      "epoch": 5.913543701171875e-05,
      "model_forward_time": 1.719433307647705,
      "step": 62008
    },
    {
      "epoch": 5.913543701171875e-05,
      "step": 62008,
      "training_step_time": 10.072479486465454
    },
    {
      "epoch": 5.9136390686035157e-05,
      "model_forward_time": 1.7250583171844482,
      "step": 62009
    },
    {
      "epoch": 5.9136390686035157e-05,
      "step": 62009,
      "training_step_time": 8.78062653541565
    },
    {
      "epoch": 5.913734436035156e-05,
      "grad_norm": 0.10055886954069138,
      "learning_rate": 3.4534998076581263e-06,
      "loss": 0.0612,
      "step": 62010
    },
    {
      "epoch": 5.913734436035156e-05,
      "model_forward_time": 1.716477632522583,
      "step": 62010
    },
    {
      "dynamics_loss_avg": 0.04325167639181018,
      "epoch": 5.913734436035156e-05,
      "step": 62010
    },
    {
      "action_loss_avg": 0.0012888286757515744,
      "epoch": 5.913734436035156e-05,
      "step": 62010
    },
    {
      "epoch": 5.913734436035156e-05,
      "step": 62010,
      "training_step_time": 7.3315534591674805
    },
    {
      "epoch": 5.913829803466797e-05,
      "model_forward_time": 1.7193520069122314,
      "step": 62011
    },
    {
      "epoch": 5.913829803466797e-05,
      "step": 62011,
      "training_step_time": 6.8722944259643555
    },
    {
      "epoch": 5.9139251708984375e-05,
      "model_forward_time": 1.7316863536834717,
      "step": 62012
    },
    {
      "epoch": 5.9139251708984375e-05,
      "step": 62012,
      "training_step_time": 7.69301176071167
    },
    {
      "epoch": 5.914020538330078e-05,
      "model_forward_time": 1.7187538146972656,
      "step": 62013
    },
    {
      "epoch": 5.914020538330078e-05,
      "step": 62013,
      "training_step_time": 6.808025360107422
    },
    {
      "epoch": 5.914115905761719e-05,
      "model_forward_time": 1.7278404235839844,
      "step": 62014
    },
    {
      "epoch": 5.914115905761719e-05,
      "step": 62014,
      "training_step_time": 7.514625072479248
    },
    {
      "epoch": 5.9142112731933594e-05,
      "model_forward_time": 2.2918484210968018,
      "step": 62015
    },
    {
      "epoch": 5.9142112731933594e-05,
      "step": 62015,
      "training_step_time": 9.269831418991089
    },
    {
      "epoch": 5.914306640625e-05,
      "model_forward_time": 1.7230045795440674,
      "step": 62016
    },
    {
      "epoch": 5.914306640625e-05,
      "step": 62016,
      "training_step_time": 8.544169187545776
    },
    {
      "epoch": 5.914402008056641e-05,
      "model_forward_time": 1.7204675674438477,
      "step": 62017
    },
    {
      "epoch": 5.914402008056641e-05,
      "step": 62017,
      "training_step_time": 8.269707441329956
    },
    {
      "epoch": 5.914497375488281e-05,
      "model_forward_time": 1.7279455661773682,
      "step": 62018
    },
    {
      "epoch": 5.914497375488281e-05,
      "step": 62018,
      "training_step_time": 8.509881496429443
    },
    {
      "epoch": 5.914592742919922e-05,
      "model_forward_time": 1.7183310985565186,
      "step": 62019
    },
    {
      "epoch": 5.914592742919922e-05,
      "step": 62019,
      "training_step_time": 7.634132146835327
    },
    {
      "epoch": 5.914688110351563e-05,
      "grad_norm": 0.0957798957824707,
      "learning_rate": 3.451927501144653e-06,
      "loss": 0.0601,
      "step": 62020
    },
    {
      "epoch": 5.914688110351563e-05,
      "model_forward_time": 1.7271184921264648,
      "step": 62020
    },
    {
      "dynamics_loss_avg": 0.059200245700776574,
      "epoch": 5.914688110351563e-05,
      "step": 62020
    },
    {
      "action_loss_avg": 0.001970398717094213,
      "epoch": 5.914688110351563e-05,
      "step": 62020
    },
    {
      "epoch": 5.914688110351563e-05,
      "step": 62020,
      "training_step_time": 6.9143900871276855
    },
    {
      "epoch": 5.914783477783203e-05,
      "model_forward_time": 1.7319042682647705,
      "step": 62021
    },
    {
      "epoch": 5.914783477783203e-05,
      "step": 62021,
      "training_step_time": 7.641897201538086
    },
    {
      "epoch": 5.9148788452148436e-05,
      "model_forward_time": 1.7203993797302246,
      "step": 62022
    },
    {
      "epoch": 5.9148788452148436e-05,
      "step": 62022,
      "training_step_time": 6.858364820480347
    },
    {
      "epoch": 5.914974212646485e-05,
      "model_forward_time": 1.708956003189087,
      "step": 62023
    },
    {
      "epoch": 5.914974212646485e-05,
      "step": 62023,
      "training_step_time": 7.652291774749756
    },
    {
      "epoch": 5.915069580078125e-05,
      "model_forward_time": 2.2719345092773438,
      "step": 62024
    },
    {
      "epoch": 5.915069580078125e-05,
      "step": 62024,
      "training_step_time": 8.150014877319336
    },
    {
      "epoch": 5.9151649475097655e-05,
      "model_forward_time": 1.7174394130706787,
      "step": 62025
    },
    {
      "epoch": 5.9151649475097655e-05,
      "step": 62025,
      "training_step_time": 8.614816427230835
    },
    {
      "epoch": 5.9152603149414066e-05,
      "model_forward_time": 1.7199983596801758,
      "step": 62026
    },
    {
      "epoch": 5.9152603149414066e-05,
      "step": 62026,
      "training_step_time": 10.297179460525513
    },
    {
      "epoch": 5.915355682373047e-05,
      "model_forward_time": 1.7195611000061035,
      "step": 62027
    },
    {
      "epoch": 5.915355682373047e-05,
      "step": 62027,
      "training_step_time": 8.613581657409668
    },
    {
      "epoch": 5.9154510498046874e-05,
      "model_forward_time": 1.7322072982788086,
      "step": 62028
    },
    {
      "epoch": 5.9154510498046874e-05,
      "step": 62028,
      "training_step_time": 7.702535152435303
    },
    {
      "epoch": 5.9155464172363285e-05,
      "model_forward_time": 1.7888715267181396,
      "step": 62029
    },
    {
      "epoch": 5.9155464172363285e-05,
      "step": 62029,
      "training_step_time": 6.825043439865112
    },
    {
      "epoch": 5.915641784667969e-05,
      "grad_norm": 0.36057916283607483,
      "learning_rate": 3.450355363926059e-06,
      "loss": 0.0608,
      "step": 62030
    },
    {
      "epoch": 5.915641784667969e-05,
      "model_forward_time": 1.713716983795166,
      "step": 62030
    },
    {
      "dynamics_loss_avg": 0.0691294252872467,
      "epoch": 5.915641784667969e-05,
      "step": 62030
    },
    {
      "action_loss_avg": 0.002204097842331976,
      "epoch": 5.915641784667969e-05,
      "step": 62030
    },
    {
      "epoch": 5.915641784667969e-05,
      "step": 62030,
      "training_step_time": 7.577062129974365
    },
    {
      "epoch": 5.915737152099609e-05,
      "model_forward_time": 1.7224211692810059,
      "step": 62031
    },
    {
      "epoch": 5.915737152099609e-05,
      "step": 62031,
      "training_step_time": 6.771726369857788
    },
    {
      "epoch": 5.91583251953125e-05,
      "model_forward_time": 1.727006196975708,
      "step": 62032
    },
    {
      "epoch": 5.91583251953125e-05,
      "step": 62032,
      "training_step_time": 7.4974799156188965
    },
    {
      "epoch": 5.915927886962891e-05,
      "model_forward_time": 1.7636840343475342,
      "step": 62033
    },
    {
      "epoch": 5.915927886962891e-05,
      "step": 62033,
      "training_step_time": 7.477707147598267
    },
    {
      "epoch": 5.916023254394531e-05,
      "model_forward_time": 2.2454335689544678,
      "step": 62034
    },
    {
      "epoch": 5.916023254394531e-05,
      "step": 62034,
      "training_step_time": 9.554646492004395
    },
    {
      "epoch": 5.9161186218261715e-05,
      "model_forward_time": 1.7245287895202637,
      "step": 62035
    },
    {
      "epoch": 5.9161186218261715e-05,
      "step": 62035,
      "training_step_time": 8.19629693031311
    },
    {
      "epoch": 5.9162139892578126e-05,
      "model_forward_time": 1.7186462879180908,
      "step": 62036
    },
    {
      "epoch": 5.9162139892578126e-05,
      "step": 62036,
      "training_step_time": 8.437136888504028
    },
    {
      "epoch": 5.916309356689453e-05,
      "model_forward_time": 1.7345449924468994,
      "step": 62037
    },
    {
      "epoch": 5.916309356689453e-05,
      "step": 62037,
      "training_step_time": 7.634953498840332
    },
    {
      "epoch": 5.9164047241210934e-05,
      "model_forward_time": 1.7212944030761719,
      "step": 62038
    },
    {
      "epoch": 5.9164047241210934e-05,
      "step": 62038,
      "training_step_time": 7.233304023742676
    },
    {
      "epoch": 5.9165000915527345e-05,
      "model_forward_time": 1.7165162563323975,
      "step": 62039
    },
    {
      "epoch": 5.9165000915527345e-05,
      "step": 62039,
      "training_step_time": 7.606098651885986
    },
    {
      "epoch": 5.916595458984375e-05,
      "grad_norm": 0.12362967431545258,
      "learning_rate": 3.4487833961742735e-06,
      "loss": 0.0587,
      "step": 62040
    },
    {
      "epoch": 5.916595458984375e-05,
      "model_forward_time": 1.7223124504089355,
      "step": 62040
    },
    {
      "dynamics_loss_avg": 0.0624126810580492,
      "epoch": 5.916595458984375e-05,
      "step": 62040
    },
    {
      "action_loss_avg": 0.001270478923106566,
      "epoch": 5.916595458984375e-05,
      "step": 62040
    },
    {
      "epoch": 5.916595458984375e-05,
      "step": 62040,
      "training_step_time": 6.822235345840454
    },
    {
      "epoch": 5.916690826416015e-05,
      "model_forward_time": 1.7309529781341553,
      "step": 62041
    },
    {
      "epoch": 5.916690826416015e-05,
      "step": 62041,
      "training_step_time": 7.50295877456665
    },
    {
      "epoch": 5.9167861938476564e-05,
      "model_forward_time": 1.76277756690979,
      "step": 62042
    },
    {
      "epoch": 5.9167861938476564e-05,
      "step": 62042,
      "training_step_time": 7.960387706756592
    },
    {
      "epoch": 5.916881561279297e-05,
      "model_forward_time": 2.250978469848633,
      "step": 62043
    },
    {
      "epoch": 5.916881561279297e-05,
      "step": 62043,
      "training_step_time": 9.254621267318726
    },
    {
      "epoch": 5.916976928710937e-05,
      "model_forward_time": 1.7209293842315674,
      "step": 62044
    },
    {
      "epoch": 5.916976928710937e-05,
      "step": 62044,
      "training_step_time": 8.717377185821533
    },
    {
      "epoch": 5.917072296142578e-05,
      "model_forward_time": 1.7148821353912354,
      "step": 62045
    },
    {
      "epoch": 5.917072296142578e-05,
      "step": 62045,
      "training_step_time": 7.7134599685668945
    },
    {
      "epoch": 5.917167663574219e-05,
      "model_forward_time": 1.7308008670806885,
      "step": 62046
    },
    {
      "epoch": 5.917167663574219e-05,
      "step": 62046,
      "training_step_time": 7.666530132293701
    },
    {
      "epoch": 5.917263031005859e-05,
      "model_forward_time": 1.714808464050293,
      "step": 62047
    },
    {
      "epoch": 5.917263031005859e-05,
      "step": 62047,
      "training_step_time": 7.638264417648315
    },
    {
      "epoch": 5.9173583984375e-05,
      "model_forward_time": 1.7204811573028564,
      "step": 62048
    },
    {
      "epoch": 5.9173583984375e-05,
      "step": 62048,
      "training_step_time": 7.619719505310059
    },
    {
      "epoch": 5.9174537658691406e-05,
      "model_forward_time": 1.7238123416900635,
      "step": 62049
    },
    {
      "epoch": 5.9174537658691406e-05,
      "step": 62049,
      "training_step_time": 7.171884775161743
    },
    {
      "epoch": 5.917549133300781e-05,
      "grad_norm": 0.09442611783742905,
      "learning_rate": 3.4472115980612038e-06,
      "loss": 0.0584,
      "step": 62050
    },
    {
      "epoch": 5.917549133300781e-05,
      "model_forward_time": 1.725825548171997,
      "step": 62050
    },
    {
      "dynamics_loss_avg": 0.057242426089942455,
      "epoch": 5.917549133300781e-05,
      "step": 62050
    },
    {
      "action_loss_avg": 0.0022422356996685266,
      "epoch": 5.917549133300781e-05,
      "step": 62050
    },
    {
      "epoch": 5.917549133300781e-05,
      "step": 62050,
      "training_step_time": 7.306492567062378
    },
    {
      "epoch": 5.917644500732422e-05,
      "model_forward_time": 1.761134386062622,
      "step": 62051
    },
    {
      "epoch": 5.917644500732422e-05,
      "step": 62051,
      "training_step_time": 7.470403432846069
    },
    {
      "epoch": 5.9177398681640625e-05,
      "model_forward_time": 2.2651431560516357,
      "step": 62052
    },
    {
      "epoch": 5.9177398681640625e-05,
      "step": 62052,
      "training_step_time": 9.0749032497406
    },
    {
      "epoch": 5.917835235595703e-05,
      "model_forward_time": 1.7230000495910645,
      "step": 62053
    },
    {
      "epoch": 5.917835235595703e-05,
      "step": 62053,
      "training_step_time": 9.57723093032837
    },
    {
      "epoch": 5.917930603027344e-05,
      "model_forward_time": 1.7093582153320312,
      "step": 62054
    },
    {
      "epoch": 5.917930603027344e-05,
      "step": 62054,
      "training_step_time": 8.665096998214722
    },
    {
      "epoch": 5.9180259704589843e-05,
      "model_forward_time": 1.718510389328003,
      "step": 62055
    },
    {
      "epoch": 5.9180259704589843e-05,
      "step": 62055,
      "training_step_time": 7.7095277309417725
    },
    {
      "epoch": 5.918121337890625e-05,
      "model_forward_time": 1.7176978588104248,
      "step": 62056
    },
    {
      "epoch": 5.918121337890625e-05,
      "step": 62056,
      "training_step_time": 7.6518590450286865
    },
    {
      "epoch": 5.918216705322266e-05,
      "model_forward_time": 1.7227783203125,
      "step": 62057
    },
    {
      "epoch": 5.918216705322266e-05,
      "step": 62057,
      "training_step_time": 7.566500186920166
    },
    {
      "epoch": 5.918312072753906e-05,
      "model_forward_time": 1.7175025939941406,
      "step": 62058
    },
    {
      "epoch": 5.918312072753906e-05,
      "step": 62058,
      "training_step_time": 7.185702323913574
    },
    {
      "epoch": 5.9184074401855466e-05,
      "model_forward_time": 1.7141270637512207,
      "step": 62059
    },
    {
      "epoch": 5.9184074401855466e-05,
      "step": 62059,
      "training_step_time": 7.210623025894165
    },
    {
      "epoch": 5.918502807617188e-05,
      "grad_norm": 0.09635032713413239,
      "learning_rate": 3.4456399697587396e-06,
      "loss": 0.0599,
      "step": 62060
    },
    {
      "epoch": 5.918502807617188e-05,
      "model_forward_time": 1.76473069190979,
      "step": 62060
    },
    {
      "dynamics_loss_avg": 0.06168922558426857,
      "epoch": 5.918502807617188e-05,
      "step": 62060
    },
    {
      "action_loss_avg": 0.0018511778180254623,
      "epoch": 5.918502807617188e-05,
      "step": 62060
    },
    {
      "epoch": 5.918502807617188e-05,
      "step": 62060,
      "training_step_time": 7.482565402984619
    },
    {
      "epoch": 5.918598175048828e-05,
      "model_forward_time": 2.284816265106201,
      "step": 62061
    },
    {
      "epoch": 5.918598175048828e-05,
      "step": 62061,
      "training_step_time": 9.03865361213684
    },
    {
      "epoch": 5.9186935424804685e-05,
      "model_forward_time": 1.7133848667144775,
      "step": 62062
    },
    {
      "epoch": 5.9186935424804685e-05,
      "step": 62062,
      "training_step_time": 9.475408792495728
    },
    {
      "epoch": 5.9187889099121096e-05,
      "model_forward_time": 1.727280855178833,
      "step": 62063
    },
    {
      "epoch": 5.9187889099121096e-05,
      "step": 62063,
      "training_step_time": 8.486554622650146
    },
    {
      "epoch": 5.91888427734375e-05,
      "model_forward_time": 1.7279677391052246,
      "step": 62064
    },
    {
      "epoch": 5.91888427734375e-05,
      "step": 62064,
      "training_step_time": 8.6326265335083
    },
    {
      "epoch": 5.9189796447753904e-05,
      "model_forward_time": 1.7201237678527832,
      "step": 62065
    },
    {
      "epoch": 5.9189796447753904e-05,
      "step": 62065,
      "training_step_time": 7.516418218612671
    },
    {
      "epoch": 5.9190750122070315e-05,
      "model_forward_time": 1.711963415145874,
      "step": 62066
    },
    {
      "epoch": 5.9190750122070315e-05,
      "step": 62066,
      "training_step_time": 7.471415996551514
    },
    {
      "epoch": 5.919170379638672e-05,
      "model_forward_time": 1.7351150512695312,
      "step": 62067
    },
    {
      "epoch": 5.919170379638672e-05,
      "step": 62067,
      "training_step_time": 7.441291809082031
    },
    {
      "epoch": 5.919265747070312e-05,
      "model_forward_time": 1.702558994293213,
      "step": 62068
    },
    {
      "epoch": 5.919265747070312e-05,
      "step": 62068,
      "training_step_time": 7.217723369598389
    },
    {
      "epoch": 5.9193611145019534e-05,
      "model_forward_time": 1.7111282348632812,
      "step": 62069
    },
    {
      "epoch": 5.9193611145019534e-05,
      "step": 62069,
      "training_step_time": 7.460304498672485
    },
    {
      "epoch": 5.919456481933594e-05,
      "grad_norm": 0.11841540038585663,
      "learning_rate": 3.4440685114387508e-06,
      "loss": 0.0557,
      "step": 62070
    },
    {
      "epoch": 5.919456481933594e-05,
      "model_forward_time": 2.8366942405700684,
      "step": 62070
    },
    {
      "dynamics_loss_avg": 0.05492200180888176,
      "epoch": 5.919456481933594e-05,
      "step": 62070
    },
    {
      "action_loss_avg": 0.0018015192326856777,
      "epoch": 5.919456481933594e-05,
      "step": 62070
    },
    {
      "epoch": 5.919456481933594e-05,
      "step": 62070,
      "training_step_time": 7.555345296859741
    },
    {
      "epoch": 5.919551849365234e-05,
      "model_forward_time": 1.7365920543670654,
      "step": 62071
    },
    {
      "epoch": 5.919551849365234e-05,
      "step": 62071,
      "training_step_time": 9.19388461112976
    },
    {
      "epoch": 5.919647216796875e-05,
      "model_forward_time": 1.7133164405822754,
      "step": 62072
    },
    {
      "epoch": 5.919647216796875e-05,
      "step": 62072,
      "training_step_time": 7.761865854263306
    },
    {
      "epoch": 5.919742584228516e-05,
      "model_forward_time": 1.722008228302002,
      "step": 62073
    },
    {
      "epoch": 5.919742584228516e-05,
      "step": 62073,
      "training_step_time": 8.247246742248535
    },
    {
      "epoch": 5.919837951660156e-05,
      "model_forward_time": 1.7134673595428467,
      "step": 62074
    },
    {
      "epoch": 5.919837951660156e-05,
      "step": 62074,
      "training_step_time": 7.622202634811401
    },
    {
      "epoch": 5.919933319091797e-05,
      "model_forward_time": 1.7097959518432617,
      "step": 62075
    },
    {
      "epoch": 5.919933319091797e-05,
      "step": 62075,
      "training_step_time": 8.558558702468872
    },
    {
      "epoch": 5.9200286865234375e-05,
      "model_forward_time": 1.7135796546936035,
      "step": 62076
    },
    {
      "epoch": 5.9200286865234375e-05,
      "step": 62076,
      "training_step_time": 6.749347448348999
    },
    {
      "epoch": 5.920124053955078e-05,
      "model_forward_time": 1.708082675933838,
      "step": 62077
    },
    {
      "epoch": 5.920124053955078e-05,
      "step": 62077,
      "training_step_time": 7.2234721183776855
    },
    {
      "epoch": 5.920219421386719e-05,
      "model_forward_time": 1.7101759910583496,
      "step": 62078
    },
    {
      "epoch": 5.920219421386719e-05,
      "step": 62078,
      "training_step_time": 7.360997438430786
    },
    {
      "epoch": 5.9203147888183594e-05,
      "model_forward_time": 1.7336812019348145,
      "step": 62079
    },
    {
      "epoch": 5.9203147888183594e-05,
      "step": 62079,
      "training_step_time": 8.891821384429932
    },
    {
      "epoch": 5.92041015625e-05,
      "grad_norm": 0.12319796532392502,
      "learning_rate": 3.442497223273091e-06,
      "loss": 0.0517,
      "step": 62080
    },
    {
      "epoch": 5.92041015625e-05,
      "model_forward_time": 1.7318007946014404,
      "step": 62080
    },
    {
      "dynamics_loss_avg": 0.057143077999353406,
      "epoch": 5.92041015625e-05,
      "step": 62080
    },
    {
      "action_loss_avg": 0.002423979656305164,
      "epoch": 5.92041015625e-05,
      "step": 62080
    },
    {
      "epoch": 5.92041015625e-05,
      "step": 62080,
      "training_step_time": 8.990005254745483
    },
    {
      "epoch": 5.920505523681641e-05,
      "model_forward_time": 1.730616807937622,
      "step": 62081
    },
    {
      "epoch": 5.920505523681641e-05,
      "step": 62081,
      "training_step_time": 8.915859460830688
    },
    {
      "epoch": 5.920600891113281e-05,
      "model_forward_time": 1.7146637439727783,
      "step": 62082
    },
    {
      "epoch": 5.920600891113281e-05,
      "step": 62082,
      "training_step_time": 7.670085430145264
    },
    {
      "epoch": 5.920696258544922e-05,
      "model_forward_time": 1.7211661338806152,
      "step": 62083
    },
    {
      "epoch": 5.920696258544922e-05,
      "step": 62083,
      "training_step_time": 7.669119834899902
    },
    {
      "epoch": 5.920791625976563e-05,
      "model_forward_time": 1.7309303283691406,
      "step": 62084
    },
    {
      "epoch": 5.920791625976563e-05,
      "step": 62084,
      "training_step_time": 7.56231164932251
    },
    {
      "epoch": 5.920886993408203e-05,
      "model_forward_time": 1.7347772121429443,
      "step": 62085
    },
    {
      "epoch": 5.920886993408203e-05,
      "step": 62085,
      "training_step_time": 7.534831285476685
    },
    {
      "epoch": 5.9209823608398436e-05,
      "model_forward_time": 1.7305352687835693,
      "step": 62086
    },
    {
      "epoch": 5.9209823608398436e-05,
      "step": 62086,
      "training_step_time": 7.1716063022613525
    },
    {
      "epoch": 5.921077728271485e-05,
      "model_forward_time": 1.737276554107666,
      "step": 62087
    },
    {
      "epoch": 5.921077728271485e-05,
      "step": 62087,
      "training_step_time": 8.360949754714966
    },
    {
      "epoch": 5.921173095703125e-05,
      "model_forward_time": 1.7333731651306152,
      "step": 62088
    },
    {
      "epoch": 5.921173095703125e-05,
      "step": 62088,
      "training_step_time": 8.393096208572388
    },
    {
      "epoch": 5.9212684631347655e-05,
      "model_forward_time": 2.923638105392456,
      "step": 62089
    },
    {
      "epoch": 5.9212684631347655e-05,
      "step": 62089,
      "training_step_time": 7.646392583847046
    },
    {
      "epoch": 5.9213638305664066e-05,
      "grad_norm": 0.1015724316239357,
      "learning_rate": 3.440926105433593e-06,
      "loss": 0.0545,
      "step": 62090
    },
    {
      "epoch": 5.9213638305664066e-05,
      "model_forward_time": 1.7257280349731445,
      "step": 62090
    },
    {
      "dynamics_loss_avg": 0.05723569057881832,
      "epoch": 5.9213638305664066e-05,
      "step": 62090
    },
    {
      "action_loss_avg": 0.0017495210631750524,
      "epoch": 5.9213638305664066e-05,
      "step": 62090
    },
    {
      "epoch": 5.9213638305664066e-05,
      "step": 62090,
      "training_step_time": 9.331212997436523
    },
    {
      "epoch": 5.921459197998047e-05,
      "model_forward_time": 1.7324187755584717,
      "step": 62091
    },
    {
      "epoch": 5.921459197998047e-05,
      "step": 62091,
      "training_step_time": 7.613418340682983
    },
    {
      "epoch": 5.9215545654296874e-05,
      "model_forward_time": 1.737036943435669,
      "step": 62092
    },
    {
      "epoch": 5.9215545654296874e-05,
      "step": 62092,
      "training_step_time": 7.614382028579712
    },
    {
      "epoch": 5.9216499328613285e-05,
      "model_forward_time": 1.7388126850128174,
      "step": 62093
    },
    {
      "epoch": 5.9216499328613285e-05,
      "step": 62093,
      "training_step_time": 7.5760438442230225
    },
    {
      "epoch": 5.921745300292969e-05,
      "model_forward_time": 1.7342398166656494,
      "step": 62094
    },
    {
      "epoch": 5.921745300292969e-05,
      "step": 62094,
      "training_step_time": 7.5114359855651855
    },
    {
      "epoch": 5.921840667724609e-05,
      "model_forward_time": 1.7341928482055664,
      "step": 62095
    },
    {
      "epoch": 5.921840667724609e-05,
      "step": 62095,
      "training_step_time": 7.202853679656982
    },
    {
      "epoch": 5.92193603515625e-05,
      "model_forward_time": 1.7370116710662842,
      "step": 62096
    },
    {
      "epoch": 5.92193603515625e-05,
      "step": 62096,
      "training_step_time": 8.529293060302734
    },
    {
      "epoch": 5.922031402587891e-05,
      "model_forward_time": 1.733367681503296,
      "step": 62097
    },
    {
      "epoch": 5.922031402587891e-05,
      "step": 62097,
      "training_step_time": 8.097567796707153
    },
    {
      "epoch": 5.922126770019531e-05,
      "model_forward_time": 2.9126365184783936,
      "step": 62098
    },
    {
      "epoch": 5.922126770019531e-05,
      "step": 62098,
      "training_step_time": 7.658081769943237
    },
    {
      "epoch": 5.9222221374511716e-05,
      "model_forward_time": 1.7334070205688477,
      "step": 62099
    },
    {
      "epoch": 5.9222221374511716e-05,
      "step": 62099,
      "training_step_time": 8.949444055557251
    },
    {
      "epoch": 5.9223175048828126e-05,
      "grad_norm": 0.07708246260881424,
      "learning_rate": 3.4393551580920693e-06,
      "loss": 0.0509,
      "step": 62100
    },
    {
      "epoch": 5.9223175048828126e-05,
      "model_forward_time": 1.7408158779144287,
      "step": 62100
    },
    {
      "dynamics_loss_avg": 0.04365251585841179,
      "epoch": 5.9223175048828126e-05,
      "step": 62100
    },
    {
      "action_loss_avg": 0.0014726543915458024,
      "epoch": 5.9223175048828126e-05,
      "step": 62100
    },
    {
      "epoch": 5.9223175048828126e-05,
      "step": 62100,
      "training_step_time": 8.302550315856934
    },
    {
      "epoch": 5.922412872314453e-05,
      "model_forward_time": 1.7244648933410645,
      "step": 62101
    },
    {
      "epoch": 5.922412872314453e-05,
      "step": 62101,
      "training_step_time": 7.583032131195068
    },
    {
      "epoch": 5.9225082397460934e-05,
      "model_forward_time": 1.7302486896514893,
      "step": 62102
    },
    {
      "epoch": 5.9225082397460934e-05,
      "step": 62102,
      "training_step_time": 7.558628797531128
    },
    {
      "epoch": 5.9226036071777345e-05,
      "model_forward_time": 1.786832571029663,
      "step": 62103
    },
    {
      "epoch": 5.9226036071777345e-05,
      "step": 62103,
      "training_step_time": 7.597454071044922
    },
    {
      "epoch": 5.922698974609375e-05,
      "model_forward_time": 1.7322816848754883,
      "step": 62104
    },
    {
      "epoch": 5.922698974609375e-05,
      "step": 62104,
      "training_step_time": 7.051401853561401
    },
    {
      "epoch": 5.922794342041015e-05,
      "model_forward_time": 1.734393835067749,
      "step": 62105
    },
    {
      "epoch": 5.922794342041015e-05,
      "step": 62105,
      "training_step_time": 7.919225454330444
    },
    {
      "epoch": 5.9228897094726564e-05,
      "model_forward_time": 1.7341182231903076,
      "step": 62106
    },
    {
      "epoch": 5.9228897094726564e-05,
      "step": 62106,
      "training_step_time": 7.547394275665283
    },
    {
      "epoch": 5.922985076904297e-05,
      "model_forward_time": 2.8584377765655518,
      "step": 62107
    },
    {
      "epoch": 5.922985076904297e-05,
      "step": 62107,
      "training_step_time": 7.581512451171875
    },
    {
      "epoch": 5.923080444335937e-05,
      "model_forward_time": 1.7476081848144531,
      "step": 62108
    },
    {
      "epoch": 5.923080444335937e-05,
      "step": 62108,
      "training_step_time": 8.68477988243103
    },
    {
      "epoch": 5.923175811767578e-05,
      "model_forward_time": 1.7525672912597656,
      "step": 62109
    },
    {
      "epoch": 5.923175811767578e-05,
      "step": 62109,
      "training_step_time": 8.68137264251709
    },
    {
      "epoch": 5.923271179199219e-05,
      "grad_norm": 0.12052140384912491,
      "learning_rate": 3.4377843814203222e-06,
      "loss": 0.0531,
      "step": 62110
    },
    {
      "epoch": 5.923271179199219e-05,
      "model_forward_time": 1.7311878204345703,
      "step": 62110
    },
    {
      "dynamics_loss_avg": 0.06380999898537994,
      "epoch": 5.923271179199219e-05,
      "step": 62110
    },
    {
      "action_loss_avg": 0.0026493418728932737,
      "epoch": 5.923271179199219e-05,
      "step": 62110
    },
    {
      "epoch": 5.923271179199219e-05,
      "step": 62110,
      "training_step_time": 7.634511947631836
    },
    {
      "epoch": 5.923366546630859e-05,
      "model_forward_time": 1.7279372215270996,
      "step": 62111
    },
    {
      "epoch": 5.923366546630859e-05,
      "step": 62111,
      "training_step_time": 7.629844665527344
    },
    {
      "epoch": 5.9234619140625e-05,
      "model_forward_time": 1.7405498027801514,
      "step": 62112
    },
    {
      "epoch": 5.9234619140625e-05,
      "step": 62112,
      "training_step_time": 7.522243022918701
    },
    {
      "epoch": 5.9235572814941406e-05,
      "model_forward_time": 1.7293481826782227,
      "step": 62113
    },
    {
      "epoch": 5.9235572814941406e-05,
      "step": 62113,
      "training_step_time": 7.1735899448394775
    },
    {
      "epoch": 5.923652648925781e-05,
      "model_forward_time": 1.732015609741211,
      "step": 62114
    },
    {
      "epoch": 5.923652648925781e-05,
      "step": 62114,
      "training_step_time": 7.240229606628418
    },
    {
      "epoch": 5.923748016357422e-05,
      "model_forward_time": 1.7278270721435547,
      "step": 62115
    },
    {
      "epoch": 5.923748016357422e-05,
      "step": 62115,
      "training_step_time": 7.529942989349365
    },
    {
      "epoch": 5.9238433837890625e-05,
      "model_forward_time": 2.871389627456665,
      "step": 62116
    },
    {
      "epoch": 5.9238433837890625e-05,
      "step": 62116,
      "training_step_time": 7.5768749713897705
    },
    {
      "epoch": 5.923938751220703e-05,
      "model_forward_time": 1.7514901161193848,
      "step": 62117
    },
    {
      "epoch": 5.923938751220703e-05,
      "step": 62117,
      "training_step_time": 9.612928867340088
    },
    {
      "epoch": 5.924034118652344e-05,
      "model_forward_time": 1.7430634498596191,
      "step": 62118
    },
    {
      "epoch": 5.924034118652344e-05,
      "step": 62118,
      "training_step_time": 7.754772901535034
    },
    {
      "epoch": 5.9241294860839844e-05,
      "model_forward_time": 1.718123435974121,
      "step": 62119
    },
    {
      "epoch": 5.9241294860839844e-05,
      "step": 62119,
      "training_step_time": 7.668963432312012
    },
    {
      "epoch": 5.924224853515625e-05,
      "grad_norm": 0.09067117422819138,
      "learning_rate": 3.436213775590126e-06,
      "loss": 0.0553,
      "step": 62120
    },
    {
      "epoch": 5.924224853515625e-05,
      "model_forward_time": 1.733950138092041,
      "step": 62120
    },
    {
      "dynamics_loss_avg": 0.054305733367800715,
      "epoch": 5.924224853515625e-05,
      "step": 62120
    },
    {
      "action_loss_avg": 0.0012432856136001646,
      "epoch": 5.924224853515625e-05,
      "step": 62120
    },
    {
      "epoch": 5.924224853515625e-05,
      "step": 62120,
      "training_step_time": 7.689153671264648
    },
    {
      "epoch": 5.924320220947266e-05,
      "model_forward_time": 1.7376959323883057,
      "step": 62121
    },
    {
      "epoch": 5.924320220947266e-05,
      "step": 62121,
      "training_step_time": 7.688320636749268
    },
    {
      "epoch": 5.924415588378906e-05,
      "model_forward_time": 1.7314562797546387,
      "step": 62122
    },
    {
      "epoch": 5.924415588378906e-05,
      "step": 62122,
      "training_step_time": 7.219076871871948
    },
    {
      "epoch": 5.9245109558105466e-05,
      "model_forward_time": 1.7379841804504395,
      "step": 62123
    },
    {
      "epoch": 5.9245109558105466e-05,
      "step": 62123,
      "training_step_time": 7.281899929046631
    },
    {
      "epoch": 5.924606323242188e-05,
      "model_forward_time": 1.7281701564788818,
      "step": 62124
    },
    {
      "epoch": 5.924606323242188e-05,
      "step": 62124,
      "training_step_time": 8.264890670776367
    },
    {
      "epoch": 5.924701690673828e-05,
      "model_forward_time": 2.8605904579162598,
      "step": 62125
    },
    {
      "epoch": 5.924701690673828e-05,
      "step": 62125,
      "training_step_time": 8.487016677856445
    },
    {
      "epoch": 5.9247970581054685e-05,
      "model_forward_time": 1.739818811416626,
      "step": 62126
    },
    {
      "epoch": 5.9247970581054685e-05,
      "step": 62126,
      "training_step_time": 10.209306478500366
    },
    {
      "epoch": 5.9248924255371096e-05,
      "model_forward_time": 1.7287092208862305,
      "step": 62127
    },
    {
      "epoch": 5.9248924255371096e-05,
      "step": 62127,
      "training_step_time": 10.059362173080444
    },
    {
      "epoch": 5.92498779296875e-05,
      "model_forward_time": 1.7364952564239502,
      "step": 62128
    },
    {
      "epoch": 5.92498779296875e-05,
      "step": 62128,
      "training_step_time": 7.627091884613037
    },
    {
      "epoch": 5.9250831604003904e-05,
      "model_forward_time": 1.7324740886688232,
      "step": 62129
    },
    {
      "epoch": 5.9250831604003904e-05,
      "step": 62129,
      "training_step_time": 9.491068601608276
    },
    {
      "epoch": 5.9251785278320315e-05,
      "grad_norm": 0.11818332225084305,
      "learning_rate": 3.4346433407732405e-06,
      "loss": 0.0546,
      "step": 62130
    },
    {
      "epoch": 5.9251785278320315e-05,
      "model_forward_time": 1.730738878250122,
      "step": 62130
    },
    {
      "dynamics_loss_avg": 0.04658906329423189,
      "epoch": 5.9251785278320315e-05,
      "step": 62130
    },
    {
      "action_loss_avg": 0.0011913309281226248,
      "epoch": 5.9251785278320315e-05,
      "step": 62130
    },
    {
      "epoch": 5.9251785278320315e-05,
      "step": 62130,
      "training_step_time": 7.573844909667969
    },
    {
      "epoch": 5.925273895263672e-05,
      "model_forward_time": 1.7272312641143799,
      "step": 62131
    },
    {
      "epoch": 5.925273895263672e-05,
      "step": 62131,
      "training_step_time": 7.203627347946167
    },
    {
      "epoch": 5.925369262695312e-05,
      "model_forward_time": 1.7308151721954346,
      "step": 62132
    },
    {
      "epoch": 5.925369262695312e-05,
      "step": 62132,
      "training_step_time": 7.210021734237671
    },
    {
      "epoch": 5.9254646301269534e-05,
      "model_forward_time": 1.7401084899902344,
      "step": 62133
    },
    {
      "epoch": 5.9254646301269534e-05,
      "step": 62133,
      "training_step_time": 7.585516452789307
    },
    {
      "epoch": 5.925559997558594e-05,
      "model_forward_time": 1.7333247661590576,
      "step": 62134
    },
    {
      "epoch": 5.925559997558594e-05,
      "step": 62134,
      "training_step_time": 8.244648456573486
    },
    {
      "epoch": 5.925655364990234e-05,
      "model_forward_time": 2.979893207550049,
      "step": 62135
    },
    {
      "epoch": 5.925655364990234e-05,
      "step": 62135,
      "training_step_time": 9.006269693374634
    },
    {
      "epoch": 5.925750732421875e-05,
      "model_forward_time": 1.733185052871704,
      "step": 62136
    },
    {
      "epoch": 5.925750732421875e-05,
      "step": 62136,
      "training_step_time": 7.8319292068481445
    },
    {
      "epoch": 5.925846099853516e-05,
      "model_forward_time": 1.7279353141784668,
      "step": 62137
    },
    {
      "epoch": 5.925846099853516e-05,
      "step": 62137,
      "training_step_time": 7.64851188659668
    },
    {
      "epoch": 5.925941467285156e-05,
      "model_forward_time": 1.7365226745605469,
      "step": 62138
    },
    {
      "epoch": 5.925941467285156e-05,
      "step": 62138,
      "training_step_time": 7.673234224319458
    },
    {
      "epoch": 5.926036834716797e-05,
      "model_forward_time": 1.7427685260772705,
      "step": 62139
    },
    {
      "epoch": 5.926036834716797e-05,
      "step": 62139,
      "training_step_time": 7.610145568847656
    },
    {
      "epoch": 5.9261322021484376e-05,
      "grad_norm": 0.09958494454622269,
      "learning_rate": 3.4330730771414052e-06,
      "loss": 0.056,
      "step": 62140
    },
    {
      "epoch": 5.9261322021484376e-05,
      "model_forward_time": 1.731476068496704,
      "step": 62140
    },
    {
      "dynamics_loss_avg": 0.04274330688640475,
      "epoch": 5.9261322021484376e-05,
      "step": 62140
    },
    {
      "action_loss_avg": 0.0010082836262881756,
      "epoch": 5.9261322021484376e-05,
      "step": 62140
    },
    {
      "epoch": 5.9261322021484376e-05,
      "step": 62140,
      "training_step_time": 7.531310319900513
    },
    {
      "epoch": 5.926227569580078e-05,
      "model_forward_time": 1.7322137355804443,
      "step": 62141
    },
    {
      "epoch": 5.926227569580078e-05,
      "step": 62141,
      "training_step_time": 6.796450138092041
    },
    {
      "epoch": 5.926322937011719e-05,
      "model_forward_time": 1.7422475814819336,
      "step": 62142
    },
    {
      "epoch": 5.926322937011719e-05,
      "step": 62142,
      "training_step_time": 7.579391717910767
    },
    {
      "epoch": 5.9264183044433594e-05,
      "model_forward_time": 1.3617784976959229,
      "step": 62143
    },
    {
      "epoch": 5.9264183044433594e-05,
      "step": 62143,
      "training_step_time": 7.512510538101196
    },
    {
      "epoch": 5.926513671875e-05,
      "model_forward_time": 2.5032873153686523,
      "step": 62144
    },
    {
      "epoch": 5.926513671875e-05,
      "step": 62144,
      "training_step_time": 8.563663482666016
    },
    {
      "epoch": 5.926609039306641e-05,
      "model_forward_time": 1.7589781284332275,
      "step": 62145
    },
    {
      "epoch": 5.926609039306641e-05,
      "step": 62145,
      "training_step_time": 8.747880935668945
    },
    {
      "epoch": 5.926704406738281e-05,
      "model_forward_time": 1.7334191799163818,
      "step": 62146
    },
    {
      "epoch": 5.926704406738281e-05,
      "step": 62146,
      "training_step_time": 7.592171669006348
    },
    {
      "epoch": 5.926799774169922e-05,
      "model_forward_time": 1.7332501411437988,
      "step": 62147
    },
    {
      "epoch": 5.926799774169922e-05,
      "step": 62147,
      "training_step_time": 7.629629135131836
    },
    {
      "epoch": 5.926895141601563e-05,
      "model_forward_time": 1.7367751598358154,
      "step": 62148
    },
    {
      "epoch": 5.926895141601563e-05,
      "step": 62148,
      "training_step_time": 7.5764000415802
    },
    {
      "epoch": 5.926990509033203e-05,
      "model_forward_time": 1.7360632419586182,
      "step": 62149
    },
    {
      "epoch": 5.926990509033203e-05,
      "step": 62149,
      "training_step_time": 7.517035722732544
    },
    {
      "epoch": 5.9270858764648436e-05,
      "grad_norm": 0.10670985281467438,
      "learning_rate": 3.4315029848663427e-06,
      "loss": 0.0561,
      "step": 62150
    },
    {
      "epoch": 5.9270858764648436e-05,
      "model_forward_time": 1.7307262420654297,
      "step": 62150
    },
    {
      "dynamics_loss_avg": 0.05304839573800564,
      "epoch": 5.9270858764648436e-05,
      "step": 62150
    },
    {
      "action_loss_avg": 0.0016820042044855654,
      "epoch": 5.9270858764648436e-05,
      "step": 62150
    },
    {
      "epoch": 5.9270858764648436e-05,
      "step": 62150,
      "training_step_time": 7.790163040161133
    },
    {
      "epoch": 5.927181243896485e-05,
      "model_forward_time": 1.742553472518921,
      "step": 62151
    },
    {
      "epoch": 5.927181243896485e-05,
      "step": 62151,
      "training_step_time": 7.555656671524048
    },
    {
      "epoch": 5.927276611328125e-05,
      "model_forward_time": 1.7349774837493896,
      "step": 62152
    },
    {
      "epoch": 5.927276611328125e-05,
      "step": 62152,
      "training_step_time": 7.47685170173645
    },
    {
      "epoch": 5.9273719787597655e-05,
      "model_forward_time": 2.879063367843628,
      "step": 62153
    },
    {
      "epoch": 5.9273719787597655e-05,
      "step": 62153,
      "training_step_time": 8.359246969223022
    },
    {
      "epoch": 5.9274673461914066e-05,
      "model_forward_time": 1.7616918087005615,
      "step": 62154
    },
    {
      "epoch": 5.9274673461914066e-05,
      "step": 62154,
      "training_step_time": 7.852757453918457
    },
    {
      "epoch": 5.927562713623047e-05,
      "model_forward_time": 1.7315270900726318,
      "step": 62155
    },
    {
      "epoch": 5.927562713623047e-05,
      "step": 62155,
      "training_step_time": 9.302007913589478
    },
    {
      "epoch": 5.9276580810546874e-05,
      "model_forward_time": 1.7405829429626465,
      "step": 62156
    },
    {
      "epoch": 5.9276580810546874e-05,
      "step": 62156,
      "training_step_time": 7.632591485977173
    },
    {
      "epoch": 5.927753448486328e-05,
      "model_forward_time": 1.7351200580596924,
      "step": 62157
    },
    {
      "epoch": 5.927753448486328e-05,
      "step": 62157,
      "training_step_time": 7.549566745758057
    },
    {
      "epoch": 5.927848815917969e-05,
      "model_forward_time": 1.368438482284546,
      "step": 62158
    },
    {
      "epoch": 5.927848815917969e-05,
      "step": 62158,
      "training_step_time": 7.549373149871826
    },
    {
      "epoch": 5.927944183349609e-05,
      "model_forward_time": 1.7368121147155762,
      "step": 62159
    },
    {
      "epoch": 5.927944183349609e-05,
      "step": 62159,
      "training_step_time": 7.14795708656311
    },
    {
      "epoch": 5.92803955078125e-05,
      "grad_norm": 0.10693848133087158,
      "learning_rate": 3.429933064119753e-06,
      "loss": 0.0576,
      "step": 62160
    },
    {
      "epoch": 5.92803955078125e-05,
      "model_forward_time": 1.741309404373169,
      "step": 62160
    },
    {
      "dynamics_loss_avg": 0.055057065188884736,
      "epoch": 5.92803955078125e-05,
      "step": 62160
    },
    {
      "action_loss_avg": 0.0018178021971834823,
      "epoch": 5.92803955078125e-05,
      "step": 62160
    },
    {
      "epoch": 5.92803955078125e-05,
      "step": 62160,
      "training_step_time": 7.5210394859313965
    },
    {
      "epoch": 5.928134918212891e-05,
      "model_forward_time": 1.7280795574188232,
      "step": 62161
    },
    {
      "epoch": 5.928134918212891e-05,
      "step": 62161,
      "training_step_time": 7.479611873626709
    },
    {
      "epoch": 5.928230285644531e-05,
      "model_forward_time": 1.3759026527404785,
      "step": 62162
    },
    {
      "epoch": 5.928230285644531e-05,
      "step": 62162,
      "training_step_time": 8.848512649536133
    },
    {
      "epoch": 5.9283256530761716e-05,
      "model_forward_time": 1.7490501403808594,
      "step": 62163
    },
    {
      "epoch": 5.9283256530761716e-05,
      "step": 62163,
      "training_step_time": 7.840281963348389
    },
    {
      "epoch": 5.9284210205078126e-05,
      "model_forward_time": 1.733778953552246,
      "step": 62164
    },
    {
      "epoch": 5.9284210205078126e-05,
      "step": 62164,
      "training_step_time": 7.5987982749938965
    },
    {
      "epoch": 5.928516387939453e-05,
      "model_forward_time": 1.731504201889038,
      "step": 62165
    },
    {
      "epoch": 5.928516387939453e-05,
      "step": 62165,
      "training_step_time": 8.471032857894897
    },
    {
      "epoch": 5.9286117553710934e-05,
      "model_forward_time": 1.7351112365722656,
      "step": 62166
    },
    {
      "epoch": 5.9286117553710934e-05,
      "step": 62166,
      "training_step_time": 7.614837646484375
    },
    {
      "epoch": 5.9287071228027345e-05,
      "model_forward_time": 1.7331786155700684,
      "step": 62167
    },
    {
      "epoch": 5.9287071228027345e-05,
      "step": 62167,
      "training_step_time": 7.623985290527344
    },
    {
      "epoch": 5.928802490234375e-05,
      "model_forward_time": 1.7498478889465332,
      "step": 62168
    },
    {
      "epoch": 5.928802490234375e-05,
      "step": 62168,
      "training_step_time": 7.191329479217529
    },
    {
      "epoch": 5.928897857666015e-05,
      "model_forward_time": 1.756103754043579,
      "step": 62169
    },
    {
      "epoch": 5.928897857666015e-05,
      "step": 62169,
      "training_step_time": 7.268174171447754
    },
    {
      "epoch": 5.9289932250976564e-05,
      "grad_norm": 0.12491665035486221,
      "learning_rate": 3.428363315073325e-06,
      "loss": 0.0581,
      "step": 62170
    },
    {
      "epoch": 5.9289932250976564e-05,
      "model_forward_time": 1.7511496543884277,
      "step": 62170
    },
    {
      "dynamics_loss_avg": 0.06732509825378656,
      "epoch": 5.9289932250976564e-05,
      "step": 62170
    },
    {
      "action_loss_avg": 0.0013969226798508315,
      "epoch": 5.9289932250976564e-05,
      "step": 62170
    },
    {
      "epoch": 5.9289932250976564e-05,
      "step": 62170,
      "training_step_time": 7.601762056350708
    },
    {
      "epoch": 5.929088592529297e-05,
      "model_forward_time": 1.7636313438415527,
      "step": 62171
    },
    {
      "epoch": 5.929088592529297e-05,
      "step": 62171,
      "training_step_time": 7.527657747268677
    },
    {
      "epoch": 5.929183959960937e-05,
      "model_forward_time": 2.997591972351074,
      "step": 62172
    },
    {
      "epoch": 5.929183959960937e-05,
      "step": 62172,
      "training_step_time": 8.868053913116455
    },
    {
      "epoch": 5.929279327392578e-05,
      "model_forward_time": 1.749643325805664,
      "step": 62173
    },
    {
      "epoch": 5.929279327392578e-05,
      "step": 62173,
      "training_step_time": 8.251619100570679
    },
    {
      "epoch": 5.929374694824219e-05,
      "model_forward_time": 1.7522680759429932,
      "step": 62174
    },
    {
      "epoch": 5.929374694824219e-05,
      "step": 62174,
      "training_step_time": 8.49501085281372
    },
    {
      "epoch": 5.929470062255859e-05,
      "model_forward_time": 1.7450041770935059,
      "step": 62175
    },
    {
      "epoch": 5.929470062255859e-05,
      "step": 62175,
      "training_step_time": 7.657581806182861
    },
    {
      "epoch": 5.9295654296875e-05,
      "model_forward_time": 1.7441678047180176,
      "step": 62176
    },
    {
      "epoch": 5.9295654296875e-05,
      "step": 62176,
      "training_step_time": 7.68093204498291
    },
    {
      "epoch": 5.9296607971191406e-05,
      "model_forward_time": 1.7493009567260742,
      "step": 62177
    },
    {
      "epoch": 5.9296607971191406e-05,
      "step": 62177,
      "training_step_time": 7.488446950912476
    },
    {
      "epoch": 5.929756164550781e-05,
      "model_forward_time": 1.7435355186462402,
      "step": 62178
    },
    {
      "epoch": 5.929756164550781e-05,
      "step": 62178,
      "training_step_time": 7.410556316375732
    },
    {
      "epoch": 5.929851531982422e-05,
      "model_forward_time": 1.7519598007202148,
      "step": 62179
    },
    {
      "epoch": 5.929851531982422e-05,
      "step": 62179,
      "training_step_time": 7.578389644622803
    },
    {
      "epoch": 5.9299468994140625e-05,
      "grad_norm": 0.10823912918567657,
      "learning_rate": 3.4267937378987227e-06,
      "loss": 0.0581,
      "step": 62180
    },
    {
      "epoch": 5.9299468994140625e-05,
      "model_forward_time": 1.7461845874786377,
      "step": 62180
    },
    {
      "dynamics_loss_avg": 0.04993289122357965,
      "epoch": 5.9299468994140625e-05,
      "step": 62180
    },
    {
      "action_loss_avg": 0.0015437796770129352,
      "epoch": 5.9299468994140625e-05,
      "step": 62180
    },
    {
      "epoch": 5.9299468994140625e-05,
      "step": 62180,
      "training_step_time": 8.071708679199219
    },
    {
      "epoch": 5.930042266845703e-05,
      "model_forward_time": 2.9436123371124268,
      "step": 62181
    },
    {
      "epoch": 5.930042266845703e-05,
      "step": 62181,
      "training_step_time": 10.454410552978516
    },
    {
      "epoch": 5.930137634277344e-05,
      "model_forward_time": 1.7617840766906738,
      "step": 62182
    },
    {
      "epoch": 5.930137634277344e-05,
      "step": 62182,
      "training_step_time": 7.676202058792114
    },
    {
      "epoch": 5.9302330017089844e-05,
      "model_forward_time": 1.7473552227020264,
      "step": 62183
    },
    {
      "epoch": 5.9302330017089844e-05,
      "step": 62183,
      "training_step_time": 8.597804069519043
    },
    {
      "epoch": 5.930328369140625e-05,
      "model_forward_time": 1.7484633922576904,
      "step": 62184
    },
    {
      "epoch": 5.930328369140625e-05,
      "step": 62184,
      "training_step_time": 7.724980115890503
    },
    {
      "epoch": 5.930423736572266e-05,
      "model_forward_time": 1.7597010135650635,
      "step": 62185
    },
    {
      "epoch": 5.930423736572266e-05,
      "step": 62185,
      "training_step_time": 7.626852989196777
    },
    {
      "epoch": 5.930519104003906e-05,
      "model_forward_time": 1.739666223526001,
      "step": 62186
    },
    {
      "epoch": 5.930519104003906e-05,
      "step": 62186,
      "training_step_time": 7.5148749351501465
    },
    {
      "epoch": 5.9306144714355466e-05,
      "model_forward_time": 1.7489383220672607,
      "step": 62187
    },
    {
      "epoch": 5.9306144714355466e-05,
      "step": 62187,
      "training_step_time": 7.206053256988525
    },
    {
      "epoch": 5.930709838867188e-05,
      "model_forward_time": 1.7676262855529785,
      "step": 62188
    },
    {
      "epoch": 5.930709838867188e-05,
      "step": 62188,
      "training_step_time": 7.60226035118103
    },
    {
      "epoch": 5.930805206298828e-05,
      "model_forward_time": 1.7600691318511963,
      "step": 62189
    },
    {
      "epoch": 5.930805206298828e-05,
      "step": 62189,
      "training_step_time": 7.53011155128479
    },
    {
      "epoch": 5.9309005737304685e-05,
      "grad_norm": 0.0864066481590271,
      "learning_rate": 3.4252243327675914e-06,
      "loss": 0.0542,
      "step": 62190
    },
    {
      "epoch": 5.9309005737304685e-05,
      "model_forward_time": 2.9033498764038086,
      "step": 62190
    },
    {
      "dynamics_loss_avg": 0.04613465443253517,
      "epoch": 5.9309005737304685e-05,
      "step": 62190
    },
    {
      "action_loss_avg": 0.0016861932454048656,
      "epoch": 5.9309005737304685e-05,
      "step": 62190
    },
    {
      "epoch": 5.9309005737304685e-05,
      "step": 62190,
      "training_step_time": 7.670215606689453
    },
    {
      "epoch": 5.9309959411621096e-05,
      "model_forward_time": 1.7568094730377197,
      "step": 62191
    },
    {
      "epoch": 5.9309959411621096e-05,
      "step": 62191,
      "training_step_time": 8.594504117965698
    },
    {
      "epoch": 5.93109130859375e-05,
      "model_forward_time": 1.7357113361358643,
      "step": 62192
    },
    {
      "epoch": 5.93109130859375e-05,
      "step": 62192,
      "training_step_time": 9.443968772888184
    },
    {
      "epoch": 5.9311866760253904e-05,
      "model_forward_time": 1.7583904266357422,
      "step": 62193
    },
    {
      "epoch": 5.9311866760253904e-05,
      "step": 62193,
      "training_step_time": 8.370535373687744
    },
    {
      "epoch": 5.9312820434570315e-05,
      "model_forward_time": 1.7474327087402344,
      "step": 62194
    },
    {
      "epoch": 5.9312820434570315e-05,
      "step": 62194,
      "training_step_time": 7.547349452972412
    },
    {
      "epoch": 5.931377410888672e-05,
      "model_forward_time": 1.7380287647247314,
      "step": 62195
    },
    {
      "epoch": 5.931377410888672e-05,
      "step": 62195,
      "training_step_time": 7.653853416442871
    },
    {
      "epoch": 5.931472778320312e-05,
      "model_forward_time": 1.7408552169799805,
      "step": 62196
    },
    {
      "epoch": 5.931472778320312e-05,
      "step": 62196,
      "training_step_time": 7.184083938598633
    },
    {
      "epoch": 5.9315681457519534e-05,
      "model_forward_time": 1.7474966049194336,
      "step": 62197
    },
    {
      "epoch": 5.9315681457519534e-05,
      "step": 62197,
      "training_step_time": 7.553643703460693
    },
    {
      "epoch": 5.931663513183594e-05,
      "model_forward_time": 1.7538187503814697,
      "step": 62198
    },
    {
      "epoch": 5.931663513183594e-05,
      "step": 62198,
      "training_step_time": 7.5368123054504395
    },
    {
      "epoch": 5.931758880615234e-05,
      "model_forward_time": 2.9821114540100098,
      "step": 62199
    },
    {
      "epoch": 5.931758880615234e-05,
      "step": 62199,
      "training_step_time": 7.707611799240112
    },
    {
      "epoch": 5.931854248046875e-05,
      "grad_norm": 0.08437655866146088,
      "learning_rate": 3.4236550998515593e-06,
      "loss": 0.0569,
      "step": 62200
    },
    {
      "epoch": 5.931854248046875e-05,
      "model_forward_time": 1.8306622505187988,
      "step": 62200
    },
    {
      "dynamics_loss_avg": 0.07137101106345653,
      "epoch": 5.931854248046875e-05,
      "step": 62200
    },
    {
      "action_loss_avg": 0.00178024583728984,
      "epoch": 5.931854248046875e-05,
      "step": 62200
    },
    {
      "epoch": 5.931854248046875e-05,
      "step": 62200,
      "training_step_time": 10.03225302696228
    },
    {
      "epoch": 5.931949615478516e-05,
      "model_forward_time": 1.7543952465057373,
      "step": 62201
    },
    {
      "epoch": 5.931949615478516e-05,
      "step": 62201,
      "training_step_time": 9.204959630966187
    },
    {
      "epoch": 5.932044982910156e-05,
      "model_forward_time": 1.7555108070373535,
      "step": 62202
    },
    {
      "epoch": 5.932044982910156e-05,
      "step": 62202,
      "training_step_time": 9.53057050704956
    },
    {
      "epoch": 5.932140350341797e-05,
      "model_forward_time": 1.770890235900879,
      "step": 62203
    },
    {
      "epoch": 5.932140350341797e-05,
      "step": 62203,
      "training_step_time": 7.542049169540405
    },
    {
      "epoch": 5.9322357177734376e-05,
      "model_forward_time": 1.752899408340454,
      "step": 62204
    },
    {
      "epoch": 5.9322357177734376e-05,
      "step": 62204,
      "training_step_time": 7.666740894317627
    },
    {
      "epoch": 5.932331085205078e-05,
      "model_forward_time": 1.7567496299743652,
      "step": 62205
    },
    {
      "epoch": 5.932331085205078e-05,
      "step": 62205,
      "training_step_time": 7.186730623245239
    },
    {
      "epoch": 5.932426452636719e-05,
      "model_forward_time": 1.7561500072479248,
      "step": 62206
    },
    {
      "epoch": 5.932426452636719e-05,
      "step": 62206,
      "training_step_time": 7.510178089141846
    },
    {
      "epoch": 5.9325218200683594e-05,
      "model_forward_time": 1.7430729866027832,
      "step": 62207
    },
    {
      "epoch": 5.9325218200683594e-05,
      "step": 62207,
      "training_step_time": 7.511547803878784
    },
    {
      "epoch": 5.9326171875e-05,
      "model_forward_time": 2.876152276992798,
      "step": 62208
    },
    {
      "epoch": 5.9326171875e-05,
      "step": 62208,
      "training_step_time": 8.44140887260437
    },
    {
      "epoch": 5.932712554931641e-05,
      "model_forward_time": 1.4118165969848633,
      "step": 62209
    },
    {
      "epoch": 5.932712554931641e-05,
      "step": 62209,
      "training_step_time": 8.746169805526733
    },
    {
      "epoch": 5.932807922363281e-05,
      "grad_norm": 0.12227486819028854,
      "learning_rate": 3.4220860393222347e-06,
      "loss": 0.0575,
      "step": 62210
    },
    {
      "epoch": 5.932807922363281e-05,
      "model_forward_time": 1.7478504180908203,
      "step": 62210
    },
    {
      "dynamics_loss_avg": 0.039987528789788486,
      "epoch": 5.932807922363281e-05,
      "step": 62210
    },
    {
      "action_loss_avg": 0.0017813435290008784,
      "epoch": 5.932807922363281e-05,
      "step": 62210
    },
    {
      "epoch": 5.932807922363281e-05,
      "step": 62210,
      "training_step_time": 8.715488910675049
    },
    {
      "epoch": 5.932903289794922e-05,
      "model_forward_time": 1.755469560623169,
      "step": 62211
    },
    {
      "epoch": 5.932903289794922e-05,
      "step": 62211,
      "training_step_time": 7.592148303985596
    },
    {
      "epoch": 5.932998657226563e-05,
      "model_forward_time": 1.7727115154266357,
      "step": 62212
    },
    {
      "epoch": 5.932998657226563e-05,
      "step": 62212,
      "training_step_time": 7.738933324813843
    },
    {
      "epoch": 5.933094024658203e-05,
      "model_forward_time": 1.7459619045257568,
      "step": 62213
    },
    {
      "epoch": 5.933094024658203e-05,
      "step": 62213,
      "training_step_time": 7.607398748397827
    },
    {
      "epoch": 5.9331893920898436e-05,
      "model_forward_time": 1.7443993091583252,
      "step": 62214
    },
    {
      "epoch": 5.9331893920898436e-05,
      "step": 62214,
      "training_step_time": 7.135760545730591
    },
    {
      "epoch": 5.933284759521485e-05,
      "model_forward_time": 1.7540204524993896,
      "step": 62215
    },
    {
      "epoch": 5.933284759521485e-05,
      "step": 62215,
      "training_step_time": 7.520666837692261
    },
    {
      "epoch": 5.933380126953125e-05,
      "model_forward_time": 1.377431869506836,
      "step": 62216
    },
    {
      "epoch": 5.933380126953125e-05,
      "step": 62216,
      "training_step_time": 8.019339084625244
    },
    {
      "epoch": 5.9334754943847655e-05,
      "model_forward_time": 1.7337028980255127,
      "step": 62217
    },
    {
      "epoch": 5.9334754943847655e-05,
      "step": 62217,
      "training_step_time": 7.5602333545684814
    },
    {
      "epoch": 5.9335708618164066e-05,
      "model_forward_time": 2.988704204559326,
      "step": 62218
    },
    {
      "epoch": 5.9335708618164066e-05,
      "step": 62218,
      "training_step_time": 7.723324775695801
    },
    {
      "epoch": 5.933666229248047e-05,
      "model_forward_time": 1.3797485828399658,
      "step": 62219
    },
    {
      "epoch": 5.933666229248047e-05,
      "step": 62219,
      "training_step_time": 9.388776063919067
    },
    {
      "epoch": 5.9337615966796874e-05,
      "grad_norm": 0.13034912943840027,
      "learning_rate": 3.420517151351209e-06,
      "loss": 0.0588,
      "step": 62220
    },
    {
      "epoch": 5.9337615966796874e-05,
      "model_forward_time": 1.7505905628204346,
      "step": 62220
    },
    {
      "dynamics_loss_avg": 0.08764588721096515,
      "epoch": 5.9337615966796874e-05,
      "step": 62220
    },
    {
      "action_loss_avg": 0.0034520160872489212,
      "epoch": 5.9337615966796874e-05,
      "step": 62220
    },
    {
      "epoch": 5.9337615966796874e-05,
      "step": 62220,
      "training_step_time": 7.660087823867798
    },
    {
      "epoch": 5.933856964111328e-05,
      "model_forward_time": 1.739562749862671,
      "step": 62221
    },
    {
      "epoch": 5.933856964111328e-05,
      "step": 62221,
      "training_step_time": 8.28694462776184
    },
    {
      "epoch": 5.933952331542969e-05,
      "model_forward_time": 1.7381138801574707,
      "step": 62222
    },
    {
      "epoch": 5.933952331542969e-05,
      "step": 62222,
      "training_step_time": 7.58269476890564
    },
    {
      "epoch": 5.934047698974609e-05,
      "model_forward_time": 1.788217306137085,
      "step": 62223
    },
    {
      "epoch": 5.934047698974609e-05,
      "step": 62223,
      "training_step_time": 7.6113433837890625
    },
    {
      "epoch": 5.93414306640625e-05,
      "model_forward_time": 1.375964641571045,
      "step": 62224
    },
    {
      "epoch": 5.93414306640625e-05,
      "step": 62224,
      "training_step_time": 7.497374534606934
    },
    {
      "epoch": 5.934238433837891e-05,
      "model_forward_time": 1.742182731628418,
      "step": 62225
    },
    {
      "epoch": 5.934238433837891e-05,
      "step": 62225,
      "training_step_time": 7.437063932418823
    },
    {
      "epoch": 5.934333801269531e-05,
      "model_forward_time": 1.7434375286102295,
      "step": 62226
    },
    {
      "epoch": 5.934333801269531e-05,
      "step": 62226,
      "training_step_time": 8.121227741241455
    },
    {
      "epoch": 5.9344291687011716e-05,
      "model_forward_time": 2.874223470687866,
      "step": 62227
    },
    {
      "epoch": 5.9344291687011716e-05,
      "step": 62227,
      "training_step_time": 8.707244396209717
    },
    {
      "epoch": 5.9345245361328126e-05,
      "model_forward_time": 1.812143325805664,
      "step": 62228
    },
    {
      "epoch": 5.9345245361328126e-05,
      "step": 62228,
      "training_step_time": 8.358022928237915
    },
    {
      "epoch": 5.934619903564453e-05,
      "model_forward_time": 1.7484917640686035,
      "step": 62229
    },
    {
      "epoch": 5.934619903564453e-05,
      "step": 62229,
      "training_step_time": 8.667770624160767
    },
    {
      "epoch": 5.9347152709960935e-05,
      "grad_norm": 0.1081320270895958,
      "learning_rate": 3.4189484361100532e-06,
      "loss": 0.0584,
      "step": 62230
    },
    {
      "epoch": 5.9347152709960935e-05,
      "model_forward_time": 1.3787167072296143,
      "step": 62230
    },
    {
      "dynamics_loss_avg": 0.05957252401858568,
      "epoch": 5.9347152709960935e-05,
      "step": 62230
    },
    {
      "action_loss_avg": 0.00166773822857067,
      "epoch": 5.9347152709960935e-05,
      "step": 62230
    },
    {
      "epoch": 5.9347152709960935e-05,
      "step": 62230,
      "training_step_time": 7.642111539840698
    },
    {
      "epoch": 5.9348106384277345e-05,
      "model_forward_time": 1.7539408206939697,
      "step": 62231
    },
    {
      "epoch": 5.9348106384277345e-05,
      "step": 62231,
      "training_step_time": 7.50626802444458
    },
    {
      "epoch": 5.934906005859375e-05,
      "model_forward_time": 1.740715503692627,
      "step": 62232
    },
    {
      "epoch": 5.934906005859375e-05,
      "step": 62232,
      "training_step_time": 7.594893217086792
    },
    {
      "epoch": 5.935001373291015e-05,
      "model_forward_time": 1.7407186031341553,
      "step": 62233
    },
    {
      "epoch": 5.935001373291015e-05,
      "step": 62233,
      "training_step_time": 7.507359504699707
    },
    {
      "epoch": 5.9350967407226564e-05,
      "model_forward_time": 1.7477397918701172,
      "step": 62234
    },
    {
      "epoch": 5.9350967407226564e-05,
      "step": 62234,
      "training_step_time": 7.457136392593384
    },
    {
      "epoch": 5.935192108154297e-05,
      "model_forward_time": 1.7446095943450928,
      "step": 62235
    },
    {
      "epoch": 5.935192108154297e-05,
      "step": 62235,
      "training_step_time": 7.413020372390747
    },
    {
      "epoch": 5.935287475585937e-05,
      "model_forward_time": 2.867661237716675,
      "step": 62236
    },
    {
      "epoch": 5.935287475585937e-05,
      "step": 62236,
      "training_step_time": 7.63092827796936
    },
    {
      "epoch": 5.935382843017578e-05,
      "model_forward_time": 1.8152656555175781,
      "step": 62237
    },
    {
      "epoch": 5.935382843017578e-05,
      "step": 62237,
      "training_step_time": 9.039034366607666
    },
    {
      "epoch": 5.935478210449219e-05,
      "model_forward_time": 1.7458784580230713,
      "step": 62238
    },
    {
      "epoch": 5.935478210449219e-05,
      "step": 62238,
      "training_step_time": 8.714192867279053
    },
    {
      "epoch": 5.935573577880859e-05,
      "model_forward_time": 1.7494404315948486,
      "step": 62239
    },
    {
      "epoch": 5.935573577880859e-05,
      "step": 62239,
      "training_step_time": 7.609961986541748
    },
    {
      "epoch": 5.9356689453125e-05,
      "grad_norm": 0.09441603720188141,
      "learning_rate": 3.4173798937703196e-06,
      "loss": 0.0565,
      "step": 62240
    },
    {
      "epoch": 5.9356689453125e-05,
      "model_forward_time": 1.7562780380249023,
      "step": 62240
    },
    {
      "dynamics_loss_avg": 0.05839750319719315,
      "epoch": 5.9356689453125e-05,
      "step": 62240
    },
    {
      "action_loss_avg": 0.002215989094111137,
      "epoch": 5.9356689453125e-05,
      "step": 62240
    },
    {
      "epoch": 5.9356689453125e-05,
      "step": 62240,
      "training_step_time": 7.457306385040283
    },
    {
      "epoch": 5.9357643127441406e-05,
      "model_forward_time": 1.7443993091583252,
      "step": 62241
    },
    {
      "epoch": 5.9357643127441406e-05,
      "step": 62241,
      "training_step_time": 7.627016067504883
    },
    {
      "epoch": 5.935859680175781e-05,
      "model_forward_time": 1.7398898601531982,
      "step": 62242
    },
    {
      "epoch": 5.935859680175781e-05,
      "step": 62242,
      "training_step_time": 6.82942271232605
    },
    {
      "epoch": 5.935955047607422e-05,
      "model_forward_time": 1.7418339252471924,
      "step": 62243
    },
    {
      "epoch": 5.935955047607422e-05,
      "step": 62243,
      "training_step_time": 8.238271951675415
    },
    {
      "epoch": 5.9360504150390625e-05,
      "model_forward_time": 1.7124171257019043,
      "step": 62244
    },
    {
      "epoch": 5.9360504150390625e-05,
      "step": 62244,
      "training_step_time": 7.543957710266113
    },
    {
      "epoch": 5.936145782470703e-05,
      "model_forward_time": 2.2729804515838623,
      "step": 62245
    },
    {
      "epoch": 5.936145782470703e-05,
      "step": 62245,
      "training_step_time": 7.799784898757935
    },
    {
      "epoch": 5.936241149902344e-05,
      "model_forward_time": 1.7193655967712402,
      "step": 62246
    },
    {
      "epoch": 5.936241149902344e-05,
      "step": 62246,
      "training_step_time": 7.667801856994629
    },
    {
      "epoch": 5.9363365173339844e-05,
      "model_forward_time": 1.7259538173675537,
      "step": 62247
    },
    {
      "epoch": 5.9363365173339844e-05,
      "step": 62247,
      "training_step_time": 8.623485326766968
    },
    {
      "epoch": 5.936431884765625e-05,
      "model_forward_time": 1.7308475971221924,
      "step": 62248
    },
    {
      "epoch": 5.936431884765625e-05,
      "step": 62248,
      "training_step_time": 7.557356119155884
    },
    {
      "epoch": 5.936527252197266e-05,
      "model_forward_time": 1.722292423248291,
      "step": 62249
    },
    {
      "epoch": 5.936527252197266e-05,
      "step": 62249,
      "training_step_time": 7.416740894317627
    },
    {
      "epoch": 5.936622619628906e-05,
      "grad_norm": 0.08663647621870041,
      "learning_rate": 3.41581152450354e-06,
      "loss": 0.0568,
      "step": 62250
    },
    {
      "epoch": 5.936622619628906e-05,
      "model_forward_time": 1.7018706798553467,
      "step": 62250
    },
    {
      "dynamics_loss_avg": 0.05323224347084761,
      "epoch": 5.936622619628906e-05,
      "step": 62250
    },
    {
      "action_loss_avg": 0.0021181123156566173,
      "epoch": 5.936622619628906e-05,
      "step": 62250
    },
    {
      "epoch": 5.936622619628906e-05,
      "step": 62250,
      "training_step_time": 7.562351226806641
    },
    {
      "epoch": 5.9367179870605467e-05,
      "model_forward_time": 1.7236945629119873,
      "step": 62251
    },
    {
      "epoch": 5.9367179870605467e-05,
      "step": 62251,
      "training_step_time": 7.197707653045654
    },
    {
      "epoch": 5.936813354492188e-05,
      "model_forward_time": 1.7227747440338135,
      "step": 62252
    },
    {
      "epoch": 5.936813354492188e-05,
      "step": 62252,
      "training_step_time": 7.612959146499634
    },
    {
      "epoch": 5.936908721923828e-05,
      "model_forward_time": 1.7276005744934082,
      "step": 62253
    },
    {
      "epoch": 5.936908721923828e-05,
      "step": 62253,
      "training_step_time": 8.067354202270508
    },
    {
      "epoch": 5.9370040893554685e-05,
      "model_forward_time": 1.7163004875183105,
      "step": 62254
    },
    {
      "epoch": 5.9370040893554685e-05,
      "step": 62254,
      "training_step_time": 7.630795001983643
    },
    {
      "epoch": 5.9370994567871096e-05,
      "model_forward_time": 2.2600855827331543,
      "step": 62255
    },
    {
      "epoch": 5.9370994567871096e-05,
      "step": 62255,
      "training_step_time": 8.192650318145752
    },
    {
      "epoch": 5.93719482421875e-05,
      "model_forward_time": 1.7270214557647705,
      "step": 62256
    },
    {
      "epoch": 5.93719482421875e-05,
      "step": 62256,
      "training_step_time": 9.237625122070312
    },
    {
      "epoch": 5.9372901916503904e-05,
      "model_forward_time": 1.7291224002838135,
      "step": 62257
    },
    {
      "epoch": 5.9372901916503904e-05,
      "step": 62257,
      "training_step_time": 8.524251461029053
    },
    {
      "epoch": 5.9373855590820315e-05,
      "model_forward_time": 1.7199270725250244,
      "step": 62258
    },
    {
      "epoch": 5.9373855590820315e-05,
      "step": 62258,
      "training_step_time": 8.116300344467163
    },
    {
      "epoch": 5.937480926513672e-05,
      "model_forward_time": 1.7211194038391113,
      "step": 62259
    },
    {
      "epoch": 5.937480926513672e-05,
      "step": 62259,
      "training_step_time": 7.593578338623047
    },
    {
      "epoch": 5.937576293945312e-05,
      "grad_norm": 0.09107521921396255,
      "learning_rate": 3.4142433284812315e-06,
      "loss": 0.0655,
      "step": 62260
    },
    {
      "epoch": 5.937576293945312e-05,
      "model_forward_time": 1.7165617942810059,
      "step": 62260
    },
    {
      "dynamics_loss_avg": 0.06166213843971491,
      "epoch": 5.937576293945312e-05,
      "step": 62260
    },
    {
      "action_loss_avg": 0.0016632882645353675,
      "epoch": 5.937576293945312e-05,
      "step": 62260
    },
    {
      "epoch": 5.937576293945312e-05,
      "step": 62260,
      "training_step_time": 6.741986274719238
    },
    {
      "epoch": 5.9376716613769534e-05,
      "model_forward_time": 1.353912353515625,
      "step": 62261
    },
    {
      "epoch": 5.9376716613769534e-05,
      "step": 62261,
      "training_step_time": 7.545447826385498
    },
    {
      "epoch": 5.937767028808594e-05,
      "model_forward_time": 1.7270092964172363,
      "step": 62262
    },
    {
      "epoch": 5.937767028808594e-05,
      "step": 62262,
      "training_step_time": 8.06807565689087
    },
    {
      "epoch": 5.937862396240234e-05,
      "model_forward_time": 1.7255029678344727,
      "step": 62263
    },
    {
      "epoch": 5.937862396240234e-05,
      "step": 62263,
      "training_step_time": 7.7933433055877686
    },
    {
      "epoch": 5.937957763671875e-05,
      "model_forward_time": 2.2746992111206055,
      "step": 62264
    },
    {
      "epoch": 5.937957763671875e-05,
      "step": 62264,
      "training_step_time": 9.383165836334229
    },
    {
      "epoch": 5.938053131103516e-05,
      "model_forward_time": 1.7287440299987793,
      "step": 62265
    },
    {
      "epoch": 5.938053131103516e-05,
      "step": 62265,
      "training_step_time": 7.787074327468872
    },
    {
      "epoch": 5.938148498535156e-05,
      "model_forward_time": 1.730790138244629,
      "step": 62266
    },
    {
      "epoch": 5.938148498535156e-05,
      "step": 62266,
      "training_step_time": 7.621983766555786
    },
    {
      "epoch": 5.938243865966797e-05,
      "model_forward_time": 1.7257933616638184,
      "step": 62267
    },
    {
      "epoch": 5.938243865966797e-05,
      "step": 62267,
      "training_step_time": 8.268464088439941
    },
    {
      "epoch": 5.9383392333984376e-05,
      "model_forward_time": 1.7227842807769775,
      "step": 62268
    },
    {
      "epoch": 5.9383392333984376e-05,
      "step": 62268,
      "training_step_time": 7.6799421310424805
    },
    {
      "epoch": 5.938434600830078e-05,
      "model_forward_time": 1.7341961860656738,
      "step": 62269
    },
    {
      "epoch": 5.938434600830078e-05,
      "step": 62269,
      "training_step_time": 7.585490942001343
    },
    {
      "epoch": 5.938529968261719e-05,
      "grad_norm": 0.10662653297185898,
      "learning_rate": 3.4126753058748885e-06,
      "loss": 0.0585,
      "step": 62270
    },
    {
      "epoch": 5.938529968261719e-05,
      "model_forward_time": 1.3508105278015137,
      "step": 62270
    },
    {
      "dynamics_loss_avg": 0.06005180533975363,
      "epoch": 5.938529968261719e-05,
      "step": 62270
    },
    {
      "action_loss_avg": 0.0018184413202106953,
      "epoch": 5.938529968261719e-05,
      "step": 62270
    },
    {
      "epoch": 5.938529968261719e-05,
      "step": 62270,
      "training_step_time": 8.484825849533081
    },
    {
      "epoch": 5.9386253356933595e-05,
      "model_forward_time": 1.722278356552124,
      "step": 62271
    },
    {
      "epoch": 5.9386253356933595e-05,
      "step": 62271,
      "training_step_time": 8.622019290924072
    },
    {
      "epoch": 5.938720703125e-05,
      "model_forward_time": 1.7290630340576172,
      "step": 62272
    },
    {
      "epoch": 5.938720703125e-05,
      "step": 62272,
      "training_step_time": 7.896977186203003
    },
    {
      "epoch": 5.938816070556641e-05,
      "model_forward_time": 1.8972618579864502,
      "step": 62273
    },
    {
      "epoch": 5.938816070556641e-05,
      "step": 62273,
      "training_step_time": 8.65508222579956
    },
    {
      "epoch": 5.938911437988281e-05,
      "model_forward_time": 1.7259631156921387,
      "step": 62274
    },
    {
      "epoch": 5.938911437988281e-05,
      "step": 62274,
      "training_step_time": 8.737234115600586
    },
    {
      "epoch": 5.939006805419922e-05,
      "model_forward_time": 1.7249953746795654,
      "step": 62275
    },
    {
      "epoch": 5.939006805419922e-05,
      "step": 62275,
      "training_step_time": 9.405726194381714
    },
    {
      "epoch": 5.939102172851563e-05,
      "model_forward_time": 1.7258281707763672,
      "step": 62276
    },
    {
      "epoch": 5.939102172851563e-05,
      "step": 62276,
      "training_step_time": 8.439580202102661
    },
    {
      "epoch": 5.939197540283203e-05,
      "model_forward_time": 1.7273318767547607,
      "step": 62277
    },
    {
      "epoch": 5.939197540283203e-05,
      "step": 62277,
      "training_step_time": 7.2889015674591064
    },
    {
      "epoch": 5.9392929077148436e-05,
      "model_forward_time": 1.7106506824493408,
      "step": 62278
    },
    {
      "epoch": 5.9392929077148436e-05,
      "step": 62278,
      "training_step_time": 7.6308135986328125
    },
    {
      "epoch": 5.939388275146485e-05,
      "model_forward_time": 1.7250757217407227,
      "step": 62279
    },
    {
      "epoch": 5.939388275146485e-05,
      "step": 62279,
      "training_step_time": 8.663346290588379
    },
    {
      "epoch": 5.939483642578125e-05,
      "grad_norm": 0.07250822335481644,
      "learning_rate": 3.4111074568559876e-06,
      "loss": 0.0577,
      "step": 62280
    },
    {
      "epoch": 5.939483642578125e-05,
      "model_forward_time": 1.7188830375671387,
      "step": 62280
    },
    {
      "dynamics_loss_avg": 0.06380742955952882,
      "epoch": 5.939483642578125e-05,
      "step": 62280
    },
    {
      "action_loss_avg": 0.0018826625600922853,
      "epoch": 5.939483642578125e-05,
      "step": 62280
    },
    {
      "epoch": 5.939483642578125e-05,
      "step": 62280,
      "training_step_time": 8.51763129234314
    },
    {
      "epoch": 5.9395790100097655e-05,
      "model_forward_time": 1.6928048133850098,
      "step": 62281
    },
    {
      "epoch": 5.9395790100097655e-05,
      "step": 62281,
      "training_step_time": 9.303455829620361
    },
    {
      "epoch": 5.939674377441406e-05,
      "model_forward_time": 1.7498185634613037,
      "step": 62282
    },
    {
      "epoch": 5.939674377441406e-05,
      "step": 62282,
      "training_step_time": 7.566836833953857
    },
    {
      "epoch": 5.939769744873047e-05,
      "model_forward_time": 2.2639975547790527,
      "step": 62283
    },
    {
      "epoch": 5.939769744873047e-05,
      "step": 62283,
      "training_step_time": 8.519994497299194
    },
    {
      "epoch": 5.9398651123046874e-05,
      "model_forward_time": 1.7263414859771729,
      "step": 62284
    },
    {
      "epoch": 5.9398651123046874e-05,
      "step": 62284,
      "training_step_time": 8.609272003173828
    },
    {
      "epoch": 5.939960479736328e-05,
      "model_forward_time": 1.7336609363555908,
      "step": 62285
    },
    {
      "epoch": 5.939960479736328e-05,
      "step": 62285,
      "training_step_time": 7.692131996154785
    },
    {
      "epoch": 5.940055847167969e-05,
      "model_forward_time": 1.7274436950683594,
      "step": 62286
    },
    {
      "epoch": 5.940055847167969e-05,
      "step": 62286,
      "training_step_time": 7.74047589302063
    },
    {
      "epoch": 5.940151214599609e-05,
      "model_forward_time": 1.7241454124450684,
      "step": 62287
    },
    {
      "epoch": 5.940151214599609e-05,
      "step": 62287,
      "training_step_time": 7.632240056991577
    },
    {
      "epoch": 5.94024658203125e-05,
      "model_forward_time": 1.7291972637176514,
      "step": 62288
    },
    {
      "epoch": 5.94024658203125e-05,
      "step": 62288,
      "training_step_time": 7.512655735015869
    },
    {
      "epoch": 5.940341949462891e-05,
      "model_forward_time": 1.735288143157959,
      "step": 62289
    },
    {
      "epoch": 5.940341949462891e-05,
      "step": 62289,
      "training_step_time": 7.564481496810913
    },
    {
      "epoch": 5.940437316894531e-05,
      "grad_norm": 0.1276194155216217,
      "learning_rate": 3.4095397815959846e-06,
      "loss": 0.0605,
      "step": 62290
    },
    {
      "epoch": 5.940437316894531e-05,
      "model_forward_time": 1.8181259632110596,
      "step": 62290
    },
    {
      "dynamics_loss_avg": 0.06391555704176426,
      "epoch": 5.940437316894531e-05,
      "step": 62290
    },
    {
      "action_loss_avg": 0.0014454051619395615,
      "epoch": 5.940437316894531e-05,
      "step": 62290
    },
    {
      "epoch": 5.940437316894531e-05,
      "step": 62290,
      "training_step_time": 8.250597476959229
    },
    {
      "epoch": 5.9405326843261716e-05,
      "model_forward_time": 1.8171162605285645,
      "step": 62291
    },
    {
      "epoch": 5.9405326843261716e-05,
      "step": 62291,
      "training_step_time": 7.652243375778198
    },
    {
      "epoch": 5.9406280517578127e-05,
      "model_forward_time": 2.351925849914551,
      "step": 62292
    },
    {
      "epoch": 5.9406280517578127e-05,
      "step": 62292,
      "training_step_time": 8.178956985473633
    },
    {
      "epoch": 5.940723419189453e-05,
      "model_forward_time": 1.8156204223632812,
      "step": 62293
    },
    {
      "epoch": 5.940723419189453e-05,
      "step": 62293,
      "training_step_time": 8.269859552383423
    },
    {
      "epoch": 5.9408187866210935e-05,
      "model_forward_time": 1.8202667236328125,
      "step": 62294
    },
    {
      "epoch": 5.9408187866210935e-05,
      "step": 62294,
      "training_step_time": 7.713815450668335
    },
    {
      "epoch": 5.9409141540527345e-05,
      "model_forward_time": 1.8116569519042969,
      "step": 62295
    },
    {
      "epoch": 5.9409141540527345e-05,
      "step": 62295,
      "training_step_time": 7.691033601760864
    },
    {
      "epoch": 5.941009521484375e-05,
      "model_forward_time": 1.8170428276062012,
      "step": 62296
    },
    {
      "epoch": 5.941009521484375e-05,
      "step": 62296,
      "training_step_time": 7.674517631530762
    },
    {
      "epoch": 5.9411048889160153e-05,
      "model_forward_time": 1.8227026462554932,
      "step": 62297
    },
    {
      "epoch": 5.9411048889160153e-05,
      "step": 62297,
      "training_step_time": 7.489518642425537
    },
    {
      "epoch": 5.9412002563476564e-05,
      "model_forward_time": 1.8089401721954346,
      "step": 62298
    },
    {
      "epoch": 5.9412002563476564e-05,
      "step": 62298,
      "training_step_time": 8.555479049682617
    },
    {
      "epoch": 5.941295623779297e-05,
      "model_forward_time": 1.8101108074188232,
      "step": 62299
    },
    {
      "epoch": 5.941295623779297e-05,
      "step": 62299,
      "training_step_time": 7.507843494415283
    },
    {
      "epoch": 5.941390991210937e-05,
      "grad_norm": 0.14317025244235992,
      "learning_rate": 3.4079722802663216e-06,
      "loss": 0.0584,
      "step": 62300
    },
    {
      "epoch": 5.941390991210937e-05,
      "model_forward_time": 1.8129558563232422,
      "step": 62300
    },
    {
      "dynamics_loss_avg": 0.05183042585849762,
      "epoch": 5.941390991210937e-05,
      "step": 62300
    },
    {
      "action_loss_avg": 0.0016651410143822432,
      "epoch": 5.941390991210937e-05,
      "step": 62300
    },
    {
      "epoch": 5.941390991210937e-05,
      "step": 62300,
      "training_step_time": 7.69561767578125
    },
    {
      "epoch": 5.941486358642578e-05,
      "model_forward_time": 2.3329524993896484,
      "step": 62301
    },
    {
      "epoch": 5.941486358642578e-05,
      "step": 62301,
      "training_step_time": 9.604504585266113
    },
    {
      "epoch": 5.941581726074219e-05,
      "model_forward_time": 1.8103253841400146,
      "step": 62302
    },
    {
      "epoch": 5.941581726074219e-05,
      "step": 62302,
      "training_step_time": 8.33689832687378
    },
    {
      "epoch": 5.941677093505859e-05,
      "model_forward_time": 1.4467461109161377,
      "step": 62303
    },
    {
      "epoch": 5.941677093505859e-05,
      "step": 62303,
      "training_step_time": 8.68669319152832
    },
    {
      "epoch": 5.9417724609375e-05,
      "model_forward_time": 1.8121147155761719,
      "step": 62304
    },
    {
      "epoch": 5.9417724609375e-05,
      "step": 62304,
      "training_step_time": 7.703802824020386
    },
    {
      "epoch": 5.9418678283691406e-05,
      "model_forward_time": 1.8169939517974854,
      "step": 62305
    },
    {
      "epoch": 5.9418678283691406e-05,
      "step": 62305,
      "training_step_time": 7.723539352416992
    },
    {
      "epoch": 5.941963195800781e-05,
      "model_forward_time": 1.8063173294067383,
      "step": 62306
    },
    {
      "epoch": 5.941963195800781e-05,
      "step": 62306,
      "training_step_time": 7.567172288894653
    },
    {
      "epoch": 5.942058563232422e-05,
      "model_forward_time": 1.8123128414154053,
      "step": 62307
    },
    {
      "epoch": 5.942058563232422e-05,
      "step": 62307,
      "training_step_time": 7.570598363876343
    },
    {
      "epoch": 5.9421539306640625e-05,
      "model_forward_time": 1.8078536987304688,
      "step": 62308
    },
    {
      "epoch": 5.9421539306640625e-05,
      "step": 62308,
      "training_step_time": 8.830476999282837
    },
    {
      "epoch": 5.942249298095703e-05,
      "model_forward_time": 1.8116416931152344,
      "step": 62309
    },
    {
      "epoch": 5.942249298095703e-05,
      "step": 62309,
      "training_step_time": 7.6800994873046875
    },
    {
      "epoch": 5.942344665527344e-05,
      "grad_norm": 0.09997226297855377,
      "learning_rate": 3.406404953038417e-06,
      "loss": 0.0607,
      "step": 62310
    },
    {
      "epoch": 5.942344665527344e-05,
      "model_forward_time": 2.3367528915405273,
      "step": 62310
    },
    {
      "dynamics_loss_avg": 0.06423737164586782,
      "epoch": 5.942344665527344e-05,
      "step": 62310
    },
    {
      "action_loss_avg": 0.0025222594500519334,
      "epoch": 5.942344665527344e-05,
      "step": 62310
    },
    {
      "epoch": 5.942344665527344e-05,
      "step": 62310,
      "training_step_time": 9.475986957550049
    },
    {
      "epoch": 5.9424400329589844e-05,
      "model_forward_time": 1.8159620761871338,
      "step": 62311
    },
    {
      "epoch": 5.9424400329589844e-05,
      "step": 62311,
      "training_step_time": 9.56447696685791
    },
    {
      "epoch": 5.942535400390625e-05,
      "model_forward_time": 1.8151328563690186,
      "step": 62312
    },
    {
      "epoch": 5.942535400390625e-05,
      "step": 62312,
      "training_step_time": 8.659383296966553
    },
    {
      "epoch": 5.942630767822266e-05,
      "model_forward_time": 1.8155906200408936,
      "step": 62313
    },
    {
      "epoch": 5.942630767822266e-05,
      "step": 62313,
      "training_step_time": 7.817450523376465
    },
    {
      "epoch": 5.942726135253906e-05,
      "model_forward_time": 1.8120341300964355,
      "step": 62314
    },
    {
      "epoch": 5.942726135253906e-05,
      "step": 62314,
      "training_step_time": 7.6805195808410645
    },
    {
      "epoch": 5.9428215026855467e-05,
      "model_forward_time": 1.8133108615875244,
      "step": 62315
    },
    {
      "epoch": 5.9428215026855467e-05,
      "step": 62315,
      "training_step_time": 7.517140626907349
    },
    {
      "epoch": 5.942916870117188e-05,
      "model_forward_time": 1.8206605911254883,
      "step": 62316
    },
    {
      "epoch": 5.942916870117188e-05,
      "step": 62316,
      "training_step_time": 7.576660633087158
    },
    {
      "epoch": 5.943012237548828e-05,
      "model_forward_time": 1.8077893257141113,
      "step": 62317
    },
    {
      "epoch": 5.943012237548828e-05,
      "step": 62317,
      "training_step_time": 7.505331993103027
    },
    {
      "epoch": 5.9431076049804685e-05,
      "model_forward_time": 1.8131146430969238,
      "step": 62318
    },
    {
      "epoch": 5.9431076049804685e-05,
      "step": 62318,
      "training_step_time": 8.174155235290527
    },
    {
      "epoch": 5.9432029724121096e-05,
      "model_forward_time": 1.8050198554992676,
      "step": 62319
    },
    {
      "epoch": 5.9432029724121096e-05,
      "step": 62319,
      "training_step_time": 9.052336931228638
    },
    {
      "epoch": 5.94329833984375e-05,
      "grad_norm": 0.11210442334413528,
      "learning_rate": 3.4048378000836702e-06,
      "loss": 0.0581,
      "step": 62320
    },
    {
      "epoch": 5.94329833984375e-05,
      "model_forward_time": 1.8049633502960205,
      "step": 62320
    },
    {
      "dynamics_loss_avg": 0.06711363866925239,
      "epoch": 5.94329833984375e-05,
      "step": 62320
    },
    {
      "action_loss_avg": 0.002033903251867741,
      "epoch": 5.94329833984375e-05,
      "step": 62320
    },
    {
      "epoch": 5.94329833984375e-05,
      "step": 62320,
      "training_step_time": 8.736480951309204
    },
    {
      "epoch": 5.9433937072753904e-05,
      "model_forward_time": 1.8093395233154297,
      "step": 62321
    },
    {
      "epoch": 5.9433937072753904e-05,
      "step": 62321,
      "training_step_time": 7.633537530899048
    },
    {
      "epoch": 5.9434890747070315e-05,
      "model_forward_time": 1.8020668029785156,
      "step": 62322
    },
    {
      "epoch": 5.9434890747070315e-05,
      "step": 62322,
      "training_step_time": 7.687579393386841
    },
    {
      "epoch": 5.943584442138672e-05,
      "model_forward_time": 1.81266450881958,
      "step": 62323
    },
    {
      "epoch": 5.943584442138672e-05,
      "step": 62323,
      "training_step_time": 8.289652824401855
    },
    {
      "epoch": 5.943679809570312e-05,
      "model_forward_time": 1.8196682929992676,
      "step": 62324
    },
    {
      "epoch": 5.943679809570312e-05,
      "step": 62324,
      "training_step_time": 7.54950475692749
    },
    {
      "epoch": 5.9437751770019534e-05,
      "model_forward_time": 1.8186593055725098,
      "step": 62325
    },
    {
      "epoch": 5.9437751770019534e-05,
      "step": 62325,
      "training_step_time": 7.522660970687866
    },
    {
      "epoch": 5.943870544433594e-05,
      "model_forward_time": 1.8256909847259521,
      "step": 62326
    },
    {
      "epoch": 5.943870544433594e-05,
      "step": 62326,
      "training_step_time": 7.429673433303833
    },
    {
      "epoch": 5.943965911865234e-05,
      "model_forward_time": 1.8153116703033447,
      "step": 62327
    },
    {
      "epoch": 5.943965911865234e-05,
      "step": 62327,
      "training_step_time": 7.539209604263306
    },
    {
      "epoch": 5.944061279296875e-05,
      "model_forward_time": 1.8261325359344482,
      "step": 62328
    },
    {
      "epoch": 5.944061279296875e-05,
      "step": 62328,
      "training_step_time": 9.18652057647705
    },
    {
      "epoch": 5.944156646728516e-05,
      "model_forward_time": 2.35744309425354,
      "step": 62329
    },
    {
      "epoch": 5.944156646728516e-05,
      "step": 62329,
      "training_step_time": 8.393612623214722
    },
    {
      "epoch": 5.944252014160156e-05,
      "grad_norm": 0.07916783541440964,
      "learning_rate": 3.4032708215734633e-06,
      "loss": 0.0602,
      "step": 62330
    },
    {
      "epoch": 5.944252014160156e-05,
      "model_forward_time": 1.7499322891235352,
      "step": 62330
    },
    {
      "dynamics_loss_avg": 0.051402812078595164,
      "epoch": 5.944252014160156e-05,
      "step": 62330
    },
    {
      "action_loss_avg": 0.0018178259197156876,
      "epoch": 5.944252014160156e-05,
      "step": 62330
    },
    {
      "epoch": 5.944252014160156e-05,
      "step": 62330,
      "training_step_time": 7.677659034729004
    },
    {
      "epoch": 5.944347381591797e-05,
      "model_forward_time": 1.7192192077636719,
      "step": 62331
    },
    {
      "epoch": 5.944347381591797e-05,
      "step": 62331,
      "training_step_time": 9.461190223693848
    },
    {
      "epoch": 5.9444427490234376e-05,
      "model_forward_time": 1.7227609157562256,
      "step": 62332
    },
    {
      "epoch": 5.9444427490234376e-05,
      "step": 62332,
      "training_step_time": 8.404324531555176
    },
    {
      "epoch": 5.944538116455078e-05,
      "model_forward_time": 1.718925952911377,
      "step": 62333
    },
    {
      "epoch": 5.944538116455078e-05,
      "step": 62333,
      "training_step_time": 7.72996187210083
    },
    {
      "epoch": 5.944633483886719e-05,
      "model_forward_time": 1.714721918106079,
      "step": 62334
    },
    {
      "epoch": 5.944633483886719e-05,
      "step": 62334,
      "training_step_time": 7.6084558963775635
    },
    {
      "epoch": 5.9447288513183595e-05,
      "model_forward_time": 1.6998462677001953,
      "step": 62335
    },
    {
      "epoch": 5.9447288513183595e-05,
      "step": 62335,
      "training_step_time": 9.5130136013031
    },
    {
      "epoch": 5.94482421875e-05,
      "model_forward_time": 1.719573736190796,
      "step": 62336
    },
    {
      "epoch": 5.94482421875e-05,
      "step": 62336,
      "training_step_time": 8.276165962219238
    },
    {
      "epoch": 5.944919586181641e-05,
      "model_forward_time": 1.3415601253509521,
      "step": 62337
    },
    {
      "epoch": 5.944919586181641e-05,
      "step": 62337,
      "training_step_time": 8.461150407791138
    },
    {
      "epoch": 5.9450149536132813e-05,
      "model_forward_time": 2.267716884613037,
      "step": 62338
    },
    {
      "epoch": 5.9450149536132813e-05,
      "step": 62338,
      "training_step_time": 8.504018306732178
    },
    {
      "epoch": 5.945110321044922e-05,
      "model_forward_time": 1.7157673835754395,
      "step": 62339
    },
    {
      "epoch": 5.945110321044922e-05,
      "step": 62339,
      "training_step_time": 7.582369327545166
    },
    {
      "epoch": 5.945205688476563e-05,
      "grad_norm": 0.0933171734213829,
      "learning_rate": 3.4017040176791583e-06,
      "loss": 0.0543,
      "step": 62340
    },
    {
      "epoch": 5.945205688476563e-05,
      "model_forward_time": 1.7145626544952393,
      "step": 62340
    },
    {
      "dynamics_loss_avg": 0.05143225197680294,
      "epoch": 5.945205688476563e-05,
      "step": 62340
    },
    {
      "action_loss_avg": 0.0018696710263611748,
      "epoch": 5.945205688476563e-05,
      "step": 62340
    },
    {
      "epoch": 5.945205688476563e-05,
      "step": 62340,
      "training_step_time": 7.6212499141693115
    },
    {
      "epoch": 5.945301055908203e-05,
      "model_forward_time": 1.7209644317626953,
      "step": 62341
    },
    {
      "epoch": 5.945301055908203e-05,
      "step": 62341,
      "training_step_time": 7.631929874420166
    },
    {
      "epoch": 5.9453964233398436e-05,
      "model_forward_time": 1.7176146507263184,
      "step": 62342
    },
    {
      "epoch": 5.9453964233398436e-05,
      "step": 62342,
      "training_step_time": 7.6957244873046875
    },
    {
      "epoch": 5.945491790771485e-05,
      "model_forward_time": 1.7138218879699707,
      "step": 62343
    },
    {
      "epoch": 5.945491790771485e-05,
      "step": 62343,
      "training_step_time": 7.55370306968689
    },
    {
      "epoch": 5.945587158203125e-05,
      "model_forward_time": 1.7153937816619873,
      "step": 62344
    },
    {
      "epoch": 5.945587158203125e-05,
      "step": 62344,
      "training_step_time": 7.54318642616272
    },
    {
      "epoch": 5.9456825256347655e-05,
      "model_forward_time": 1.405754566192627,
      "step": 62345
    },
    {
      "epoch": 5.9456825256347655e-05,
      "step": 62345,
      "training_step_time": 8.23968243598938
    },
    {
      "epoch": 5.945777893066406e-05,
      "model_forward_time": 1.7113261222839355,
      "step": 62346
    },
    {
      "epoch": 5.945777893066406e-05,
      "step": 62346,
      "training_step_time": 9.239222049713135
    },
    {
      "epoch": 5.945873260498047e-05,
      "model_forward_time": 2.2593796253204346,
      "step": 62347
    },
    {
      "epoch": 5.945873260498047e-05,
      "step": 62347,
      "training_step_time": 8.377320528030396
    },
    {
      "epoch": 5.9459686279296874e-05,
      "model_forward_time": 1.7226624488830566,
      "step": 62348
    },
    {
      "epoch": 5.9459686279296874e-05,
      "step": 62348,
      "training_step_time": 8.553889274597168
    },
    {
      "epoch": 5.946063995361328e-05,
      "model_forward_time": 1.7194433212280273,
      "step": 62349
    },
    {
      "epoch": 5.946063995361328e-05,
      "step": 62349,
      "training_step_time": 7.726224422454834
    },
    {
      "epoch": 5.946159362792969e-05,
      "grad_norm": 0.1161152645945549,
      "learning_rate": 3.400137388572098e-06,
      "loss": 0.0576,
      "step": 62350
    },
    {
      "epoch": 5.946159362792969e-05,
      "model_forward_time": 1.7188372611999512,
      "step": 62350
    },
    {
      "dynamics_loss_avg": 0.055485562048852445,
      "epoch": 5.946159362792969e-05,
      "step": 62350
    },
    {
      "action_loss_avg": 0.0014841366210021079,
      "epoch": 5.946159362792969e-05,
      "step": 62350
    },
    {
      "epoch": 5.946159362792969e-05,
      "step": 62350,
      "training_step_time": 8.536168098449707
    },
    {
      "epoch": 5.946254730224609e-05,
      "model_forward_time": 1.7088713645935059,
      "step": 62351
    },
    {
      "epoch": 5.946254730224609e-05,
      "step": 62351,
      "training_step_time": 7.643532991409302
    },
    {
      "epoch": 5.94635009765625e-05,
      "model_forward_time": 1.7073891162872314,
      "step": 62352
    },
    {
      "epoch": 5.94635009765625e-05,
      "step": 62352,
      "training_step_time": 7.485811471939087
    },
    {
      "epoch": 5.946445465087891e-05,
      "model_forward_time": 1.711291790008545,
      "step": 62353
    },
    {
      "epoch": 5.946445465087891e-05,
      "step": 62353,
      "training_step_time": 7.474937438964844
    },
    {
      "epoch": 5.946540832519531e-05,
      "model_forward_time": 1.7578554153442383,
      "step": 62354
    },
    {
      "epoch": 5.946540832519531e-05,
      "step": 62354,
      "training_step_time": 7.658514499664307
    },
    {
      "epoch": 5.9466361999511716e-05,
      "model_forward_time": 1.7122466564178467,
      "step": 62355
    },
    {
      "epoch": 5.9466361999511716e-05,
      "step": 62355,
      "training_step_time": 7.788009405136108
    },
    {
      "epoch": 5.9467315673828127e-05,
      "model_forward_time": 1.715780258178711,
      "step": 62356
    },
    {
      "epoch": 5.9467315673828127e-05,
      "step": 62356,
      "training_step_time": 11.293456792831421
    },
    {
      "epoch": 5.946826934814453e-05,
      "model_forward_time": 2.2686867713928223,
      "step": 62357
    },
    {
      "epoch": 5.946826934814453e-05,
      "step": 62357,
      "training_step_time": 8.862586736679077
    },
    {
      "epoch": 5.9469223022460935e-05,
      "model_forward_time": 1.7125017642974854,
      "step": 62358
    },
    {
      "epoch": 5.9469223022460935e-05,
      "step": 62358,
      "training_step_time": 7.694118022918701
    },
    {
      "epoch": 5.9470176696777345e-05,
      "model_forward_time": 1.7124509811401367,
      "step": 62359
    },
    {
      "epoch": 5.9470176696777345e-05,
      "step": 62359,
      "training_step_time": 8.554704666137695
    },
    {
      "epoch": 5.947113037109375e-05,
      "grad_norm": 0.1348039209842682,
      "learning_rate": 3.3985709344236086e-06,
      "loss": 0.0571,
      "step": 62360
    },
    {
      "epoch": 5.947113037109375e-05,
      "model_forward_time": 1.7136945724487305,
      "step": 62360
    },
    {
      "dynamics_loss_avg": 0.05918114446103573,
      "epoch": 5.947113037109375e-05,
      "step": 62360
    },
    {
      "action_loss_avg": 0.002316470380173996,
      "epoch": 5.947113037109375e-05,
      "step": 62360
    },
    {
      "epoch": 5.947113037109375e-05,
      "step": 62360,
      "training_step_time": 7.672764301300049
    },
    {
      "epoch": 5.9472084045410153e-05,
      "model_forward_time": 1.7114546298980713,
      "step": 62361
    },
    {
      "epoch": 5.9472084045410153e-05,
      "step": 62361,
      "training_step_time": 7.573458194732666
    },
    {
      "epoch": 5.9473037719726564e-05,
      "model_forward_time": 1.717646837234497,
      "step": 62362
    },
    {
      "epoch": 5.9473037719726564e-05,
      "step": 62362,
      "training_step_time": 8.54173231124878
    },
    {
      "epoch": 5.947399139404297e-05,
      "model_forward_time": 1.4004366397857666,
      "step": 62363
    },
    {
      "epoch": 5.947399139404297e-05,
      "step": 62363,
      "training_step_time": 8.33598518371582
    },
    {
      "epoch": 5.947494506835937e-05,
      "model_forward_time": 1.7180535793304443,
      "step": 62364
    },
    {
      "epoch": 5.947494506835937e-05,
      "step": 62364,
      "training_step_time": 7.694181442260742
    },
    {
      "epoch": 5.947589874267578e-05,
      "model_forward_time": 1.7152361869812012,
      "step": 62365
    },
    {
      "epoch": 5.947589874267578e-05,
      "step": 62365,
      "training_step_time": 8.29848337173462
    },
    {
      "epoch": 5.947685241699219e-05,
      "model_forward_time": 2.251190423965454,
      "step": 62366
    },
    {
      "epoch": 5.947685241699219e-05,
      "step": 62366,
      "training_step_time": 9.8056321144104
    },
    {
      "epoch": 5.947780609130859e-05,
      "model_forward_time": 1.719172477722168,
      "step": 62367
    },
    {
      "epoch": 5.947780609130859e-05,
      "step": 62367,
      "training_step_time": 7.659063339233398
    },
    {
      "epoch": 5.9478759765625e-05,
      "model_forward_time": 1.71500825881958,
      "step": 62368
    },
    {
      "epoch": 5.9478759765625e-05,
      "step": 62368,
      "training_step_time": 7.695157527923584
    },
    {
      "epoch": 5.9479713439941406e-05,
      "model_forward_time": 1.7118189334869385,
      "step": 62369
    },
    {
      "epoch": 5.9479713439941406e-05,
      "step": 62369,
      "training_step_time": 7.640263319015503
    },
    {
      "epoch": 5.948066711425781e-05,
      "grad_norm": 0.10131201893091202,
      "learning_rate": 3.397004655404994e-06,
      "loss": 0.0596,
      "step": 62370
    },
    {
      "epoch": 5.948066711425781e-05,
      "model_forward_time": 1.7199451923370361,
      "step": 62370
    },
    {
      "dynamics_loss_avg": 0.08289723116904497,
      "epoch": 5.948066711425781e-05,
      "step": 62370
    },
    {
      "action_loss_avg": 0.002016064996132627,
      "epoch": 5.948066711425781e-05,
      "step": 62370
    },
    {
      "epoch": 5.948066711425781e-05,
      "step": 62370,
      "training_step_time": 7.6856303215026855
    },
    {
      "epoch": 5.948162078857422e-05,
      "model_forward_time": 1.71756911277771,
      "step": 62371
    },
    {
      "epoch": 5.948162078857422e-05,
      "step": 62371,
      "training_step_time": 7.612942695617676
    },
    {
      "epoch": 5.9482574462890625e-05,
      "model_forward_time": 1.7715709209442139,
      "step": 62372
    },
    {
      "epoch": 5.9482574462890625e-05,
      "step": 62372,
      "training_step_time": 7.626915216445923
    },
    {
      "epoch": 5.948352813720703e-05,
      "model_forward_time": 1.7201452255249023,
      "step": 62373
    },
    {
      "epoch": 5.948352813720703e-05,
      "step": 62373,
      "training_step_time": 7.729847431182861
    },
    {
      "epoch": 5.948448181152344e-05,
      "model_forward_time": 1.7164111137390137,
      "step": 62374
    },
    {
      "epoch": 5.948448181152344e-05,
      "step": 62374,
      "training_step_time": 8.413032531738281
    },
    {
      "epoch": 5.9485435485839844e-05,
      "model_forward_time": 2.264483690261841,
      "step": 62375
    },
    {
      "epoch": 5.9485435485839844e-05,
      "step": 62375,
      "training_step_time": 7.713982105255127
    },
    {
      "epoch": 5.948638916015625e-05,
      "model_forward_time": 1.716677188873291,
      "step": 62376
    },
    {
      "epoch": 5.948638916015625e-05,
      "step": 62376,
      "training_step_time": 9.862129926681519
    },
    {
      "epoch": 5.948734283447266e-05,
      "model_forward_time": 1.3397376537322998,
      "step": 62377
    },
    {
      "epoch": 5.948734283447266e-05,
      "step": 62377,
      "training_step_time": 7.70150351524353
    },
    {
      "epoch": 5.948829650878906e-05,
      "model_forward_time": 1.7149674892425537,
      "step": 62378
    },
    {
      "epoch": 5.948829650878906e-05,
      "step": 62378,
      "training_step_time": 7.742609024047852
    },
    {
      "epoch": 5.948925018310547e-05,
      "model_forward_time": 1.714015007019043,
      "step": 62379
    },
    {
      "epoch": 5.948925018310547e-05,
      "step": 62379,
      "training_step_time": 7.702740907669067
    },
    {
      "epoch": 5.949020385742188e-05,
      "grad_norm": 0.22520272433757782,
      "learning_rate": 3.395438551687541e-06,
      "loss": 0.0545,
      "step": 62380
    },
    {
      "epoch": 5.949020385742188e-05,
      "model_forward_time": 1.7137196063995361,
      "step": 62380
    },
    {
      "dynamics_loss_avg": 0.04314016429707408,
      "epoch": 5.949020385742188e-05,
      "step": 62380
    },
    {
      "action_loss_avg": 0.001484182698186487,
      "epoch": 5.949020385742188e-05,
      "step": 62380
    },
    {
      "epoch": 5.949020385742188e-05,
      "step": 62380,
      "training_step_time": 7.625216960906982
    },
    {
      "epoch": 5.949115753173828e-05,
      "model_forward_time": 1.771923303604126,
      "step": 62381
    },
    {
      "epoch": 5.949115753173828e-05,
      "step": 62381,
      "training_step_time": 8.675324201583862
    },
    {
      "epoch": 5.9492111206054685e-05,
      "model_forward_time": 1.7109472751617432,
      "step": 62382
    },
    {
      "epoch": 5.9492111206054685e-05,
      "step": 62382,
      "training_step_time": 7.79121470451355
    },
    {
      "epoch": 5.9493064880371096e-05,
      "model_forward_time": 1.7151856422424316,
      "step": 62383
    },
    {
      "epoch": 5.9493064880371096e-05,
      "step": 62383,
      "training_step_time": 9.184168100357056
    },
    {
      "epoch": 5.94940185546875e-05,
      "model_forward_time": 2.253575325012207,
      "step": 62384
    },
    {
      "epoch": 5.94940185546875e-05,
      "step": 62384,
      "training_step_time": 9.098886966705322
    },
    {
      "epoch": 5.9494972229003904e-05,
      "model_forward_time": 1.7109897136688232,
      "step": 62385
    },
    {
      "epoch": 5.9494972229003904e-05,
      "step": 62385,
      "training_step_time": 8.5676589012146
    },
    {
      "epoch": 5.9495925903320315e-05,
      "model_forward_time": 1.709918737411499,
      "step": 62386
    },
    {
      "epoch": 5.9495925903320315e-05,
      "step": 62386,
      "training_step_time": 8.877743244171143
    },
    {
      "epoch": 5.949687957763672e-05,
      "model_forward_time": 1.7115604877471924,
      "step": 62387
    },
    {
      "epoch": 5.949687957763672e-05,
      "step": 62387,
      "training_step_time": 7.576016426086426
    },
    {
      "epoch": 5.949783325195312e-05,
      "model_forward_time": 1.7121331691741943,
      "step": 62388
    },
    {
      "epoch": 5.949783325195312e-05,
      "step": 62388,
      "training_step_time": 7.727754354476929
    },
    {
      "epoch": 5.9498786926269534e-05,
      "model_forward_time": 1.7138075828552246,
      "step": 62389
    },
    {
      "epoch": 5.9498786926269534e-05,
      "step": 62389,
      "training_step_time": 7.583590507507324
    },
    {
      "epoch": 5.949974060058594e-05,
      "grad_norm": 0.09331860393285751,
      "learning_rate": 3.393872623442515e-06,
      "loss": 0.0558,
      "step": 62390
    },
    {
      "epoch": 5.949974060058594e-05,
      "model_forward_time": 1.7638065814971924,
      "step": 62390
    },
    {
      "dynamics_loss_avg": 0.052627327665686605,
      "epoch": 5.949974060058594e-05,
      "step": 62390
    },
    {
      "action_loss_avg": 0.0020523125829640778,
      "epoch": 5.949974060058594e-05,
      "step": 62390
    },
    {
      "epoch": 5.949974060058594e-05,
      "step": 62390,
      "training_step_time": 7.673901557922363
    },
    {
      "epoch": 5.950069427490234e-05,
      "model_forward_time": 1.714510440826416,
      "step": 62391
    },
    {
      "epoch": 5.950069427490234e-05,
      "step": 62391,
      "training_step_time": 7.707714557647705
    },
    {
      "epoch": 5.950164794921875e-05,
      "model_forward_time": 1.7125887870788574,
      "step": 62392
    },
    {
      "epoch": 5.950164794921875e-05,
      "step": 62392,
      "training_step_time": 8.574267864227295
    },
    {
      "epoch": 5.950260162353516e-05,
      "model_forward_time": 2.252624273300171,
      "step": 62393
    },
    {
      "epoch": 5.950260162353516e-05,
      "step": 62393,
      "training_step_time": 8.484688758850098
    },
    {
      "epoch": 5.950355529785156e-05,
      "model_forward_time": 1.714101791381836,
      "step": 62394
    },
    {
      "epoch": 5.950355529785156e-05,
      "step": 62394,
      "training_step_time": 7.6500256061553955
    },
    {
      "epoch": 5.950450897216797e-05,
      "model_forward_time": 1.7135734558105469,
      "step": 62395
    },
    {
      "epoch": 5.950450897216797e-05,
      "step": 62395,
      "training_step_time": 7.7100207805633545
    },
    {
      "epoch": 5.9505462646484376e-05,
      "model_forward_time": 1.7038590908050537,
      "step": 62396
    },
    {
      "epoch": 5.9505462646484376e-05,
      "step": 62396,
      "training_step_time": 7.721731662750244
    },
    {
      "epoch": 5.950641632080078e-05,
      "model_forward_time": 1.713104248046875,
      "step": 62397
    },
    {
      "epoch": 5.950641632080078e-05,
      "step": 62397,
      "training_step_time": 8.737749338150024
    },
    {
      "epoch": 5.950736999511719e-05,
      "model_forward_time": 1.8085150718688965,
      "step": 62398
    },
    {
      "epoch": 5.950736999511719e-05,
      "step": 62398,
      "training_step_time": 8.696496725082397
    },
    {
      "epoch": 5.9508323669433595e-05,
      "model_forward_time": 1.8060691356658936,
      "step": 62399
    },
    {
      "epoch": 5.9508323669433595e-05,
      "step": 62399,
      "training_step_time": 10.20861268043518
    },
    {
      "epoch": 5.950927734375e-05,
      "grad_norm": 0.07567717880010605,
      "learning_rate": 3.3923068708411645e-06,
      "loss": 0.0535,
      "step": 62400
    },
    {
      "epoch": 5.950927734375e-05,
      "model_forward_time": 1.8169338703155518,
      "step": 62400
    },
    {
      "dynamics_loss_avg": 0.03921852745115757,
      "epoch": 5.950927734375e-05,
      "step": 62400
    },
    {
      "action_loss_avg": 0.0014936186838895083,
      "epoch": 5.950927734375e-05,
      "step": 62400
    },
    {
      "epoch": 5.950927734375e-05,
      "step": 62400,
      "training_step_time": 6.833952903747559
    },
    {
      "epoch": 5.951023101806641e-05,
      "model_forward_time": 1.8127241134643555,
      "step": 62401
    },
    {
      "epoch": 5.951023101806641e-05,
      "step": 62401,
      "training_step_time": 6.825082302093506
    },
    {
      "epoch": 5.9511184692382813e-05,
      "model_forward_time": 1.8140246868133545,
      "step": 62402
    },
    {
      "epoch": 5.9511184692382813e-05,
      "step": 62402,
      "training_step_time": 6.7822582721710205
    },
    {
      "epoch": 5.951213836669922e-05,
      "model_forward_time": 1.8065648078918457,
      "step": 62403
    },
    {
      "epoch": 5.951213836669922e-05,
      "step": 62403,
      "training_step_time": 6.807348728179932
    },
    {
      "epoch": 5.951309204101563e-05,
      "model_forward_time": 1.8186554908752441,
      "step": 62404
    },
    {
      "epoch": 5.951309204101563e-05,
      "step": 62404,
      "training_step_time": 6.817429780960083
    },
    {
      "epoch": 5.951404571533203e-05,
      "model_forward_time": 1.8200910091400146,
      "step": 62405
    },
    {
      "epoch": 5.951404571533203e-05,
      "step": 62405,
      "training_step_time": 6.808331251144409
    },
    {
      "epoch": 5.9514999389648436e-05,
      "model_forward_time": 1.8183684349060059,
      "step": 62406
    },
    {
      "epoch": 5.9514999389648436e-05,
      "step": 62406,
      "training_step_time": 6.794499158859253
    },
    {
      "epoch": 5.951595306396484e-05,
      "model_forward_time": 1.817230463027954,
      "step": 62407
    },
    {
      "epoch": 5.951595306396484e-05,
      "step": 62407,
      "training_step_time": 8.182835102081299
    },
    {
      "epoch": 5.951690673828125e-05,
      "model_forward_time": 1.8124265670776367,
      "step": 62408
    },
    {
      "epoch": 5.951690673828125e-05,
      "step": 62408,
      "training_step_time": 7.800369739532471
    },
    {
      "epoch": 5.9517860412597655e-05,
      "model_forward_time": 1.8167839050292969,
      "step": 62409
    },
    {
      "epoch": 5.9517860412597655e-05,
      "step": 62409,
      "training_step_time": 7.958857536315918
    },
    {
      "epoch": 5.951881408691406e-05,
      "grad_norm": 0.0880691409111023,
      "learning_rate": 3.390741294054715e-06,
      "loss": 0.0558,
      "step": 62410
    },
    {
      "epoch": 5.951881408691406e-05,
      "model_forward_time": 1.8156013488769531,
      "step": 62410
    },
    {
      "dynamics_loss_avg": 0.06611521877348422,
      "epoch": 5.951881408691406e-05,
      "step": 62410
    },
    {
      "action_loss_avg": 0.002414216537727043,
      "epoch": 5.951881408691406e-05,
      "step": 62410
    },
    {
      "epoch": 5.951881408691406e-05,
      "step": 62410,
      "training_step_time": 6.969543695449829
    },
    {
      "epoch": 5.951976776123047e-05,
      "model_forward_time": 1.814558744430542,
      "step": 62411
    },
    {
      "epoch": 5.951976776123047e-05,
      "step": 62411,
      "training_step_time": 7.010918617248535
    },
    {
      "epoch": 5.9520721435546874e-05,
      "model_forward_time": 1.82094407081604,
      "step": 62412
    },
    {
      "epoch": 5.9520721435546874e-05,
      "step": 62412,
      "training_step_time": 6.970614433288574
    },
    {
      "epoch": 5.952167510986328e-05,
      "model_forward_time": 1.810307264328003,
      "step": 62413
    },
    {
      "epoch": 5.952167510986328e-05,
      "step": 62413,
      "training_step_time": 6.990745306015015
    },
    {
      "epoch": 5.952262878417969e-05,
      "model_forward_time": 1.805948257446289,
      "step": 62414
    },
    {
      "epoch": 5.952262878417969e-05,
      "step": 62414,
      "training_step_time": 6.989314317703247
    },
    {
      "epoch": 5.952358245849609e-05,
      "model_forward_time": 1.8183846473693848,
      "step": 62415
    },
    {
      "epoch": 5.952358245849609e-05,
      "step": 62415,
      "training_step_time": 7.055885553359985
    },
    {
      "epoch": 5.95245361328125e-05,
      "model_forward_time": 1.4401054382324219,
      "step": 62416
    },
    {
      "epoch": 5.95245361328125e-05,
      "step": 62416,
      "training_step_time": 7.0111284255981445
    },
    {
      "epoch": 5.952548980712891e-05,
      "model_forward_time": 2.3785364627838135,
      "step": 62417
    },
    {
      "epoch": 5.952548980712891e-05,
      "step": 62417,
      "training_step_time": 7.7824554443359375
    },
    {
      "epoch": 5.952644348144531e-05,
      "model_forward_time": 1.8253018856048584,
      "step": 62418
    },
    {
      "epoch": 5.952644348144531e-05,
      "step": 62418,
      "training_step_time": 8.034599304199219
    },
    {
      "epoch": 5.9527397155761716e-05,
      "model_forward_time": 1.8162128925323486,
      "step": 62419
    },
    {
      "epoch": 5.9527397155761716e-05,
      "step": 62419,
      "training_step_time": 7.038928270339966
    },
    {
      "epoch": 5.952835083007813e-05,
      "grad_norm": 0.09834470599889755,
      "learning_rate": 3.3891758932543807e-06,
      "loss": 0.0573,
      "step": 62420
    },
    {
      "epoch": 5.952835083007813e-05,
      "model_forward_time": 1.8197486400604248,
      "step": 62420
    },
    {
      "dynamics_loss_avg": 0.05409039575606585,
      "epoch": 5.952835083007813e-05,
      "step": 62420
    },
    {
      "action_loss_avg": 0.002015163586474955,
      "epoch": 5.952835083007813e-05,
      "step": 62420
    },
    {
      "epoch": 5.952835083007813e-05,
      "step": 62420,
      "training_step_time": 6.958997011184692
    },
    {
      "epoch": 5.952930450439453e-05,
      "model_forward_time": 1.8266825675964355,
      "step": 62421
    },
    {
      "epoch": 5.952930450439453e-05,
      "step": 62421,
      "training_step_time": 6.963386297225952
    },
    {
      "epoch": 5.9530258178710935e-05,
      "model_forward_time": 1.8157975673675537,
      "step": 62422
    },
    {
      "epoch": 5.9530258178710935e-05,
      "step": 62422,
      "training_step_time": 7.021312236785889
    },
    {
      "epoch": 5.9531211853027345e-05,
      "model_forward_time": 1.8157258033752441,
      "step": 62423
    },
    {
      "epoch": 5.9531211853027345e-05,
      "step": 62423,
      "training_step_time": 6.996776103973389
    },
    {
      "epoch": 5.953216552734375e-05,
      "model_forward_time": 1.8158726692199707,
      "step": 62424
    },
    {
      "epoch": 5.953216552734375e-05,
      "step": 62424,
      "training_step_time": 7.024494171142578
    },
    {
      "epoch": 5.9533119201660154e-05,
      "model_forward_time": 1.4329664707183838,
      "step": 62425
    },
    {
      "epoch": 5.9533119201660154e-05,
      "step": 62425,
      "training_step_time": 7.037001848220825
    },
    {
      "epoch": 5.9534072875976564e-05,
      "model_forward_time": 2.296555757522583,
      "step": 62426
    },
    {
      "epoch": 5.9534072875976564e-05,
      "step": 62426,
      "training_step_time": 7.849273681640625
    },
    {
      "epoch": 5.953502655029297e-05,
      "model_forward_time": 1.7115874290466309,
      "step": 62427
    },
    {
      "epoch": 5.953502655029297e-05,
      "step": 62427,
      "training_step_time": 8.085422992706299
    },
    {
      "epoch": 5.953598022460937e-05,
      "model_forward_time": 1.7128913402557373,
      "step": 62428
    },
    {
      "epoch": 5.953598022460937e-05,
      "step": 62428,
      "training_step_time": 7.691264629364014
    },
    {
      "epoch": 5.953693389892578e-05,
      "model_forward_time": 1.7127141952514648,
      "step": 62429
    },
    {
      "epoch": 5.953693389892578e-05,
      "step": 62429,
      "training_step_time": 7.070652484893799
    },
    {
      "epoch": 5.953788757324219e-05,
      "grad_norm": 0.07664859294891357,
      "learning_rate": 3.3876106686113487e-06,
      "loss": 0.0562,
      "step": 62430
    },
    {
      "epoch": 5.953788757324219e-05,
      "model_forward_time": 1.7116432189941406,
      "step": 62430
    },
    {
      "dynamics_loss_avg": 0.05156483631581068,
      "epoch": 5.953788757324219e-05,
      "step": 62430
    },
    {
      "action_loss_avg": 0.0016468278598040342,
      "epoch": 5.953788757324219e-05,
      "step": 62430
    },
    {
      "epoch": 5.953788757324219e-05,
      "step": 62430,
      "training_step_time": 7.043397426605225
    },
    {
      "epoch": 5.953884124755859e-05,
      "model_forward_time": 1.7162973880767822,
      "step": 62431
    },
    {
      "epoch": 5.953884124755859e-05,
      "step": 62431,
      "training_step_time": 7.059024810791016
    },
    {
      "epoch": 5.9539794921875e-05,
      "model_forward_time": 1.713709831237793,
      "step": 62432
    },
    {
      "epoch": 5.9539794921875e-05,
      "step": 62432,
      "training_step_time": 7.032400608062744
    },
    {
      "epoch": 5.9540748596191406e-05,
      "model_forward_time": 1.7096726894378662,
      "step": 62433
    },
    {
      "epoch": 5.9540748596191406e-05,
      "step": 62433,
      "training_step_time": 7.019818544387817
    },
    {
      "epoch": 5.954170227050781e-05,
      "model_forward_time": 1.7125468254089355,
      "step": 62434
    },
    {
      "epoch": 5.954170227050781e-05,
      "step": 62434,
      "training_step_time": 6.9984681606292725
    },
    {
      "epoch": 5.954265594482422e-05,
      "model_forward_time": 2.2827343940734863,
      "step": 62435
    },
    {
      "epoch": 5.954265594482422e-05,
      "step": 62435,
      "training_step_time": 7.510289907455444
    },
    {
      "epoch": 5.9543609619140625e-05,
      "model_forward_time": 1.7087984085083008,
      "step": 62436
    },
    {
      "epoch": 5.9543609619140625e-05,
      "step": 62436,
      "training_step_time": 10.157072067260742
    },
    {
      "epoch": 5.954456329345703e-05,
      "model_forward_time": 1.7129931449890137,
      "step": 62437
    },
    {
      "epoch": 5.954456329345703e-05,
      "step": 62437,
      "training_step_time": 8.022063732147217
    },
    {
      "epoch": 5.954551696777344e-05,
      "model_forward_time": 1.7099463939666748,
      "step": 62438
    },
    {
      "epoch": 5.954551696777344e-05,
      "step": 62438,
      "training_step_time": 7.063410758972168
    },
    {
      "epoch": 5.9546470642089844e-05,
      "model_forward_time": 1.7140312194824219,
      "step": 62439
    },
    {
      "epoch": 5.9546470642089844e-05,
      "step": 62439,
      "training_step_time": 7.029749393463135
    },
    {
      "epoch": 5.954742431640625e-05,
      "grad_norm": 0.12477591633796692,
      "learning_rate": 3.3860456202967895e-06,
      "loss": 0.0548,
      "step": 62440
    },
    {
      "epoch": 5.954742431640625e-05,
      "model_forward_time": 1.697913646697998,
      "step": 62440
    },
    {
      "dynamics_loss_avg": 0.05689553767442703,
      "epoch": 5.954742431640625e-05,
      "step": 62440
    },
    {
      "action_loss_avg": 0.002654195262584835,
      "epoch": 5.954742431640625e-05,
      "step": 62440
    },
    {
      "epoch": 5.954742431640625e-05,
      "step": 62440,
      "training_step_time": 7.030782699584961
    },
    {
      "epoch": 5.954837799072266e-05,
      "model_forward_time": 1.710052728652954,
      "step": 62441
    },
    {
      "epoch": 5.954837799072266e-05,
      "step": 62441,
      "training_step_time": 7.031804323196411
    },
    {
      "epoch": 5.954933166503906e-05,
      "model_forward_time": 1.7097370624542236,
      "step": 62442
    },
    {
      "epoch": 5.954933166503906e-05,
      "step": 62442,
      "training_step_time": 7.049575567245483
    },
    {
      "epoch": 5.955028533935547e-05,
      "model_forward_time": 1.7129185199737549,
      "step": 62443
    },
    {
      "epoch": 5.955028533935547e-05,
      "step": 62443,
      "training_step_time": 6.993150949478149
    },
    {
      "epoch": 5.955123901367188e-05,
      "model_forward_time": 2.2910289764404297,
      "step": 62444
    },
    {
      "epoch": 5.955123901367188e-05,
      "step": 62444,
      "training_step_time": 7.428287506103516
    },
    {
      "epoch": 5.955219268798828e-05,
      "model_forward_time": 1.7091283798217773,
      "step": 62445
    },
    {
      "epoch": 5.955219268798828e-05,
      "step": 62445,
      "training_step_time": 8.692858219146729
    },
    {
      "epoch": 5.9553146362304686e-05,
      "model_forward_time": 1.7238056659698486,
      "step": 62446
    },
    {
      "epoch": 5.9553146362304686e-05,
      "step": 62446,
      "training_step_time": 7.993983030319214
    },
    {
      "epoch": 5.9554100036621096e-05,
      "model_forward_time": 1.7209432125091553,
      "step": 62447
    },
    {
      "epoch": 5.9554100036621096e-05,
      "step": 62447,
      "training_step_time": 7.391441345214844
    },
    {
      "epoch": 5.95550537109375e-05,
      "model_forward_time": 1.708711862564087,
      "step": 62448
    },
    {
      "epoch": 5.95550537109375e-05,
      "step": 62448,
      "training_step_time": 6.967072486877441
    },
    {
      "epoch": 5.9556007385253904e-05,
      "model_forward_time": 1.7204065322875977,
      "step": 62449
    },
    {
      "epoch": 5.9556007385253904e-05,
      "step": 62449,
      "training_step_time": 6.9807469844818115
    },
    {
      "epoch": 5.9556961059570315e-05,
      "grad_norm": 0.11084552854299545,
      "learning_rate": 3.384480748481855e-06,
      "loss": 0.0589,
      "step": 62450
    },
    {
      "epoch": 5.9556961059570315e-05,
      "model_forward_time": 1.7106163501739502,
      "step": 62450
    },
    {
      "dynamics_loss_avg": 0.07536484804004431,
      "epoch": 5.9556961059570315e-05,
      "step": 62450
    },
    {
      "action_loss_avg": 0.001757469042786397,
      "epoch": 5.9556961059570315e-05,
      "step": 62450
    },
    {
      "epoch": 5.9556961059570315e-05,
      "step": 62450,
      "training_step_time": 7.029329538345337
    },
    {
      "epoch": 5.955791473388672e-05,
      "model_forward_time": 1.7183587551116943,
      "step": 62451
    },
    {
      "epoch": 5.955791473388672e-05,
      "step": 62451,
      "training_step_time": 7.0105485916137695
    },
    {
      "epoch": 5.955886840820312e-05,
      "model_forward_time": 1.719482660293579,
      "step": 62452
    },
    {
      "epoch": 5.955886840820312e-05,
      "step": 62452,
      "training_step_time": 6.9949421882629395
    },
    {
      "epoch": 5.9559822082519534e-05,
      "model_forward_time": 2.291764974594116,
      "step": 62453
    },
    {
      "epoch": 5.9559822082519534e-05,
      "step": 62453,
      "training_step_time": 7.395040988922119
    },
    {
      "epoch": 5.956077575683594e-05,
      "model_forward_time": 1.71171236038208,
      "step": 62454
    },
    {
      "epoch": 5.956077575683594e-05,
      "step": 62454,
      "training_step_time": 7.985084533691406
    },
    {
      "epoch": 5.956172943115234e-05,
      "model_forward_time": 1.7226755619049072,
      "step": 62455
    },
    {
      "epoch": 5.956172943115234e-05,
      "step": 62455,
      "training_step_time": 8.105567932128906
    },
    {
      "epoch": 5.956268310546875e-05,
      "model_forward_time": 1.7184860706329346,
      "step": 62456
    },
    {
      "epoch": 5.956268310546875e-05,
      "step": 62456,
      "training_step_time": 7.801599740982056
    },
    {
      "epoch": 5.956363677978516e-05,
      "model_forward_time": 1.7178871631622314,
      "step": 62457
    },
    {
      "epoch": 5.956363677978516e-05,
      "step": 62457,
      "training_step_time": 7.410598039627075
    },
    {
      "epoch": 5.956459045410156e-05,
      "model_forward_time": 1.7167038917541504,
      "step": 62458
    },
    {
      "epoch": 5.956459045410156e-05,
      "step": 62458,
      "training_step_time": 7.000126361846924
    },
    {
      "epoch": 5.956554412841797e-05,
      "model_forward_time": 1.7123305797576904,
      "step": 62459
    },
    {
      "epoch": 5.956554412841797e-05,
      "step": 62459,
      "training_step_time": 6.981712818145752
    },
    {
      "epoch": 5.9566497802734376e-05,
      "grad_norm": 0.1292172074317932,
      "learning_rate": 3.382916053337676e-06,
      "loss": 0.0539,
      "step": 62460
    },
    {
      "epoch": 5.9566497802734376e-05,
      "model_forward_time": 1.7148988246917725,
      "step": 62460
    },
    {
      "dynamics_loss_avg": 0.05203941334038973,
      "epoch": 5.9566497802734376e-05,
      "step": 62460
    },
    {
      "action_loss_avg": 0.0014840967924101278,
      "epoch": 5.9566497802734376e-05,
      "step": 62460
    },
    {
      "epoch": 5.9566497802734376e-05,
      "step": 62460,
      "training_step_time": 6.989636659622192
    },
    {
      "epoch": 5.956745147705078e-05,
      "model_forward_time": 1.7085204124450684,
      "step": 62461
    },
    {
      "epoch": 5.956745147705078e-05,
      "step": 62461,
      "training_step_time": 6.999963760375977
    },
    {
      "epoch": 5.956840515136719e-05,
      "model_forward_time": 1.725818395614624,
      "step": 62462
    },
    {
      "epoch": 5.956840515136719e-05,
      "step": 62462,
      "training_step_time": 7.378075361251831
    },
    {
      "epoch": 5.9569358825683595e-05,
      "model_forward_time": 2.2718446254730225,
      "step": 62463
    },
    {
      "epoch": 5.9569358825683595e-05,
      "step": 62463,
      "training_step_time": 8.011150360107422
    },
    {
      "epoch": 5.95703125e-05,
      "model_forward_time": 1.7284269332885742,
      "step": 62464
    },
    {
      "epoch": 5.95703125e-05,
      "step": 62464,
      "training_step_time": 9.445560216903687
    },
    {
      "epoch": 5.957126617431641e-05,
      "model_forward_time": 1.7072129249572754,
      "step": 62465
    },
    {
      "epoch": 5.957126617431641e-05,
      "step": 62465,
      "training_step_time": 7.9561073780059814
    },
    {
      "epoch": 5.9572219848632814e-05,
      "model_forward_time": 1.712303876876831,
      "step": 62466
    },
    {
      "epoch": 5.9572219848632814e-05,
      "step": 62466,
      "training_step_time": 7.421105146408081
    },
    {
      "epoch": 5.957317352294922e-05,
      "model_forward_time": 1.718947410583496,
      "step": 62467
    },
    {
      "epoch": 5.957317352294922e-05,
      "step": 62467,
      "training_step_time": 7.013402462005615
    },
    {
      "epoch": 5.957412719726563e-05,
      "model_forward_time": 1.7211003303527832,
      "step": 62468
    },
    {
      "epoch": 5.957412719726563e-05,
      "step": 62468,
      "training_step_time": 6.949275493621826
    },
    {
      "epoch": 5.957508087158203e-05,
      "model_forward_time": 1.7390685081481934,
      "step": 62469
    },
    {
      "epoch": 5.957508087158203e-05,
      "step": 62469,
      "training_step_time": 6.955003499984741
    },
    {
      "epoch": 5.9576034545898436e-05,
      "grad_norm": 0.095122329890728,
      "learning_rate": 3.3813515350353666e-06,
      "loss": 0.06,
      "step": 62470
    },
    {
      "epoch": 5.9576034545898436e-05,
      "model_forward_time": 1.7217316627502441,
      "step": 62470
    },
    {
      "dynamics_loss_avg": 0.07328924238681793,
      "epoch": 5.9576034545898436e-05,
      "step": 62470
    },
    {
      "action_loss_avg": 0.0023713645932730287,
      "epoch": 5.9576034545898436e-05,
      "step": 62470
    },
    {
      "epoch": 5.9576034545898436e-05,
      "step": 62470,
      "training_step_time": 7.000261545181274
    },
    {
      "epoch": 5.957698822021484e-05,
      "model_forward_time": 1.7247192859649658,
      "step": 62471
    },
    {
      "epoch": 5.957698822021484e-05,
      "step": 62471,
      "training_step_time": 6.984556436538696
    },
    {
      "epoch": 5.957794189453125e-05,
      "model_forward_time": 2.3025290966033936,
      "step": 62472
    },
    {
      "epoch": 5.957794189453125e-05,
      "step": 62472,
      "training_step_time": 8.021093606948853
    },
    {
      "epoch": 5.9578895568847655e-05,
      "model_forward_time": 1.718611478805542,
      "step": 62473
    },
    {
      "epoch": 5.9578895568847655e-05,
      "step": 62473,
      "training_step_time": 9.938308477401733
    },
    {
      "epoch": 5.957984924316406e-05,
      "model_forward_time": 1.720266580581665,
      "step": 62474
    },
    {
      "epoch": 5.957984924316406e-05,
      "step": 62474,
      "training_step_time": 8.104170083999634
    },
    {
      "epoch": 5.958080291748047e-05,
      "model_forward_time": 1.7189369201660156,
      "step": 62475
    },
    {
      "epoch": 5.958080291748047e-05,
      "step": 62475,
      "training_step_time": 7.533339023590088
    },
    {
      "epoch": 5.9581756591796874e-05,
      "model_forward_time": 1.7142174243927002,
      "step": 62476
    },
    {
      "epoch": 5.9581756591796874e-05,
      "step": 62476,
      "training_step_time": 7.011154651641846
    },
    {
      "epoch": 5.958271026611328e-05,
      "model_forward_time": 1.7180085182189941,
      "step": 62477
    },
    {
      "epoch": 5.958271026611328e-05,
      "step": 62477,
      "training_step_time": 7.024760484695435
    },
    {
      "epoch": 5.958366394042969e-05,
      "model_forward_time": 1.7154898643493652,
      "step": 62478
    },
    {
      "epoch": 5.958366394042969e-05,
      "step": 62478,
      "training_step_time": 7.060838937759399
    },
    {
      "epoch": 5.958461761474609e-05,
      "model_forward_time": 1.7142088413238525,
      "step": 62479
    },
    {
      "epoch": 5.958461761474609e-05,
      "step": 62479,
      "training_step_time": 7.006733417510986
    },
    {
      "epoch": 5.95855712890625e-05,
      "grad_norm": 0.11216054111719131,
      "learning_rate": 3.379787193746019e-06,
      "loss": 0.0556,
      "step": 62480
    },
    {
      "epoch": 5.95855712890625e-05,
      "model_forward_time": 1.7199983596801758,
      "step": 62480
    },
    {
      "dynamics_loss_avg": 0.05936085972934961,
      "epoch": 5.95855712890625e-05,
      "step": 62480
    },
    {
      "action_loss_avg": 0.003007193372468464,
      "epoch": 5.95855712890625e-05,
      "step": 62480
    },
    {
      "epoch": 5.95855712890625e-05,
      "step": 62480,
      "training_step_time": 7.023371696472168
    },
    {
      "epoch": 5.958652496337891e-05,
      "model_forward_time": 2.2988386154174805,
      "step": 62481
    },
    {
      "epoch": 5.958652496337891e-05,
      "step": 62481,
      "training_step_time": 7.950493574142456
    },
    {
      "epoch": 5.958747863769531e-05,
      "model_forward_time": 1.712308645248413,
      "step": 62482
    },
    {
      "epoch": 5.958747863769531e-05,
      "step": 62482,
      "training_step_time": 10.503931522369385
    },
    {
      "epoch": 5.9588432312011716e-05,
      "model_forward_time": 1.7179632186889648,
      "step": 62483
    },
    {
      "epoch": 5.9588432312011716e-05,
      "step": 62483,
      "training_step_time": 8.553436040878296
    },
    {
      "epoch": 5.958938598632813e-05,
      "model_forward_time": 1.72084641456604,
      "step": 62484
    },
    {
      "epoch": 5.958938598632813e-05,
      "step": 62484,
      "training_step_time": 8.019296646118164
    },
    {
      "epoch": 5.959033966064453e-05,
      "model_forward_time": 1.714111089706421,
      "step": 62485
    },
    {
      "epoch": 5.959033966064453e-05,
      "step": 62485,
      "training_step_time": 7.014952182769775
    },
    {
      "epoch": 5.9591293334960935e-05,
      "model_forward_time": 1.7167868614196777,
      "step": 62486
    },
    {
      "epoch": 5.9591293334960935e-05,
      "step": 62486,
      "training_step_time": 6.9894654750823975
    },
    {
      "epoch": 5.9592247009277346e-05,
      "model_forward_time": 1.7180960178375244,
      "step": 62487
    },
    {
      "epoch": 5.9592247009277346e-05,
      "step": 62487,
      "training_step_time": 6.9861671924591064
    },
    {
      "epoch": 5.959320068359375e-05,
      "model_forward_time": 1.722823143005371,
      "step": 62488
    },
    {
      "epoch": 5.959320068359375e-05,
      "step": 62488,
      "training_step_time": 6.956955909729004
    },
    {
      "epoch": 5.9594154357910154e-05,
      "model_forward_time": 1.7131495475769043,
      "step": 62489
    },
    {
      "epoch": 5.9594154357910154e-05,
      "step": 62489,
      "training_step_time": 6.99395751953125
    },
    {
      "epoch": 5.9595108032226564e-05,
      "grad_norm": 0.1316588819026947,
      "learning_rate": 3.3782230296407094e-06,
      "loss": 0.0594,
      "step": 62490
    },
    {
      "epoch": 5.9595108032226564e-05,
      "model_forward_time": 2.3057544231414795,
      "step": 62490
    },
    {
      "dynamics_loss_avg": 0.04432883206754923,
      "epoch": 5.9595108032226564e-05,
      "step": 62490
    },
    {
      "action_loss_avg": 0.001191890392510686,
      "epoch": 5.9595108032226564e-05,
      "step": 62490
    },
    {
      "epoch": 5.9595108032226564e-05,
      "step": 62490,
      "training_step_time": 7.986954927444458
    },
    {
      "epoch": 5.959606170654297e-05,
      "model_forward_time": 1.7111859321594238,
      "step": 62491
    },
    {
      "epoch": 5.959606170654297e-05,
      "step": 62491,
      "training_step_time": 10.40410304069519
    },
    {
      "epoch": 5.959701538085937e-05,
      "model_forward_time": 1.7180438041687012,
      "step": 62492
    },
    {
      "epoch": 5.959701538085937e-05,
      "step": 62492,
      "training_step_time": 8.000695943832397
    },
    {
      "epoch": 5.959796905517578e-05,
      "model_forward_time": 1.7309951782226562,
      "step": 62493
    },
    {
      "epoch": 5.959796905517578e-05,
      "step": 62493,
      "training_step_time": 7.774860382080078
    },
    {
      "epoch": 5.959892272949219e-05,
      "model_forward_time": 1.7181496620178223,
      "step": 62494
    },
    {
      "epoch": 5.959892272949219e-05,
      "step": 62494,
      "training_step_time": 7.433802366256714
    },
    {
      "epoch": 5.959987640380859e-05,
      "model_forward_time": 1.7200605869293213,
      "step": 62495
    },
    {
      "epoch": 5.959987640380859e-05,
      "step": 62495,
      "training_step_time": 7.000311851501465
    },
    {
      "epoch": 5.9600830078125e-05,
      "model_forward_time": 1.7139546871185303,
      "step": 62496
    },
    {
      "epoch": 5.9600830078125e-05,
      "step": 62496,
      "training_step_time": 7.007461786270142
    },
    {
      "epoch": 5.9601783752441406e-05,
      "model_forward_time": 1.723001480102539,
      "step": 62497
    },
    {
      "epoch": 5.9601783752441406e-05,
      "step": 62497,
      "training_step_time": 7.021291971206665
    },
    {
      "epoch": 5.960273742675781e-05,
      "model_forward_time": 1.6904850006103516,
      "step": 62498
    },
    {
      "epoch": 5.960273742675781e-05,
      "step": 62498,
      "training_step_time": 7.01625394821167
    },
    {
      "epoch": 5.960369110107422e-05,
      "model_forward_time": 2.275557041168213,
      "step": 62499
    },
    {
      "epoch": 5.960369110107422e-05,
      "step": 62499,
      "training_step_time": 8.00058913230896
    },
    {
      "epoch": 5.9604644775390625e-05,
      "grad_norm": 0.0911899134516716,
      "learning_rate": 3.376659042890489e-06,
      "loss": 0.0545,
      "step": 62500
    },
    {
      "epoch": 5.9604644775390625e-05,
      "model_forward_time": 1.693284034729004,
      "step": 62500
    },
    {
      "dynamics_loss_avg": 0.07076261416077614,
      "epoch": 5.9604644775390625e-05,
      "step": 62500
    },
    {
      "action_loss_avg": 0.0017992923385463655,
      "epoch": 5.9604644775390625e-05,
      "step": 62500
    },
    {
      "epoch": 5.9604644775390625e-05,
      "step": 62500,
      "training_step_time": 10.41130518913269
    },
    {
      "epoch": 5.960559844970703e-05,
      "model_forward_time": 1.6992552280426025,
      "step": 62501
    },
    {
      "epoch": 5.960559844970703e-05,
      "step": 62501,
      "training_step_time": 8.953392028808594
    },
    {
      "epoch": 5.960655212402344e-05,
      "model_forward_time": 1.7109076976776123,
      "step": 62502
    },
    {
      "epoch": 5.960655212402344e-05,
      "step": 62502,
      "training_step_time": 7.4054460525512695
    },
    {
      "epoch": 5.9607505798339844e-05,
      "model_forward_time": 1.7004799842834473,
      "step": 62503
    },
    {
      "epoch": 5.9607505798339844e-05,
      "step": 62503,
      "training_step_time": 7.440974950790405
    },
    {
      "epoch": 5.960845947265625e-05,
      "model_forward_time": 1.6965796947479248,
      "step": 62504
    },
    {
      "epoch": 5.960845947265625e-05,
      "step": 62504,
      "training_step_time": 6.967226505279541
    },
    {
      "epoch": 5.960941314697266e-05,
      "model_forward_time": 1.695070505142212,
      "step": 62505
    },
    {
      "epoch": 5.960941314697266e-05,
      "step": 62505,
      "training_step_time": 7.022966384887695
    },
    {
      "epoch": 5.961036682128906e-05,
      "model_forward_time": 1.713014841079712,
      "step": 62506
    },
    {
      "epoch": 5.961036682128906e-05,
      "step": 62506,
      "training_step_time": 7.0176637172698975
    },
    {
      "epoch": 5.961132049560547e-05,
      "model_forward_time": 1.698951005935669,
      "step": 62507
    },
    {
      "epoch": 5.961132049560547e-05,
      "step": 62507,
      "training_step_time": 7.019099235534668
    },
    {
      "epoch": 5.961227416992188e-05,
      "model_forward_time": 1.707132339477539,
      "step": 62508
    },
    {
      "epoch": 5.961227416992188e-05,
      "step": 62508,
      "training_step_time": 8.4556565284729
    },
    {
      "epoch": 5.961322784423828e-05,
      "model_forward_time": 1.6990864276885986,
      "step": 62509
    },
    {
      "epoch": 5.961322784423828e-05,
      "step": 62509,
      "training_step_time": 10.168261528015137
    },
    {
      "epoch": 5.9614181518554686e-05,
      "grad_norm": 0.10983508080244064,
      "learning_rate": 3.3750952336663973e-06,
      "loss": 0.0569,
      "step": 62510
    },
    {
      "epoch": 5.9614181518554686e-05,
      "model_forward_time": 1.6975789070129395,
      "step": 62510
    },
    {
      "dynamics_loss_avg": 0.06120766382664442,
      "epoch": 5.9614181518554686e-05,
      "step": 62510
    },
    {
      "action_loss_avg": 0.0017450359868234956,
      "epoch": 5.9614181518554686e-05,
      "step": 62510
    },
    {
      "epoch": 5.9614181518554686e-05,
      "step": 62510,
      "training_step_time": 8.975586414337158
    },
    {
      "epoch": 5.9615135192871096e-05,
      "model_forward_time": 1.692190170288086,
      "step": 62511
    },
    {
      "epoch": 5.9615135192871096e-05,
      "step": 62511,
      "training_step_time": 7.994761943817139
    },
    {
      "epoch": 5.96160888671875e-05,
      "model_forward_time": 1.6975834369659424,
      "step": 62512
    },
    {
      "epoch": 5.96160888671875e-05,
      "step": 62512,
      "training_step_time": 7.408936500549316
    },
    {
      "epoch": 5.9617042541503904e-05,
      "model_forward_time": 1.3295321464538574,
      "step": 62513
    },
    {
      "epoch": 5.9617042541503904e-05,
      "step": 62513,
      "training_step_time": 6.9660484790802
    },
    {
      "epoch": 5.9617996215820315e-05,
      "model_forward_time": 1.716428518295288,
      "step": 62514
    },
    {
      "epoch": 5.9617996215820315e-05,
      "step": 62514,
      "training_step_time": 7.039318084716797
    },
    {
      "epoch": 5.961894989013672e-05,
      "model_forward_time": 1.6972579956054688,
      "step": 62515
    },
    {
      "epoch": 5.961894989013672e-05,
      "step": 62515,
      "training_step_time": 7.0558576583862305
    },
    {
      "epoch": 5.961990356445312e-05,
      "model_forward_time": 1.7101573944091797,
      "step": 62516
    },
    {
      "epoch": 5.961990356445312e-05,
      "step": 62516,
      "training_step_time": 7.011353015899658
    },
    {
      "epoch": 5.9620857238769534e-05,
      "model_forward_time": 1.7140564918518066,
      "step": 62517
    },
    {
      "epoch": 5.9620857238769534e-05,
      "step": 62517,
      "training_step_time": 7.746959209442139
    },
    {
      "epoch": 5.962181091308594e-05,
      "model_forward_time": 2.284236431121826,
      "step": 62518
    },
    {
      "epoch": 5.962181091308594e-05,
      "step": 62518,
      "training_step_time": 9.625816345214844
    },
    {
      "epoch": 5.962276458740234e-05,
      "model_forward_time": 1.7080967426300049,
      "step": 62519
    },
    {
      "epoch": 5.962276458740234e-05,
      "step": 62519,
      "training_step_time": 7.972550868988037
    },
    {
      "epoch": 5.962371826171875e-05,
      "grad_norm": 0.08224979788064957,
      "learning_rate": 3.373531602139447e-06,
      "loss": 0.059,
      "step": 62520
    },
    {
      "epoch": 5.962371826171875e-05,
      "model_forward_time": 1.717360019683838,
      "step": 62520
    },
    {
      "dynamics_loss_avg": 0.05926649980247021,
      "epoch": 5.962371826171875e-05,
      "step": 62520
    },
    {
      "action_loss_avg": 0.0016203314473386854,
      "epoch": 5.962371826171875e-05,
      "step": 62520
    },
    {
      "epoch": 5.962371826171875e-05,
      "step": 62520,
      "training_step_time": 8.00085735321045
    },
    {
      "epoch": 5.962467193603516e-05,
      "model_forward_time": 1.7148075103759766,
      "step": 62521
    },
    {
      "epoch": 5.962467193603516e-05,
      "step": 62521,
      "training_step_time": 7.799198150634766
    },
    {
      "epoch": 5.962562561035156e-05,
      "model_forward_time": 1.7073428630828857,
      "step": 62522
    },
    {
      "epoch": 5.962562561035156e-05,
      "step": 62522,
      "training_step_time": 7.0075390338897705
    },
    {
      "epoch": 5.962657928466797e-05,
      "model_forward_time": 1.7195301055908203,
      "step": 62523
    },
    {
      "epoch": 5.962657928466797e-05,
      "step": 62523,
      "training_step_time": 7.0276947021484375
    },
    {
      "epoch": 5.9627532958984376e-05,
      "model_forward_time": 1.7294714450836182,
      "step": 62524
    },
    {
      "epoch": 5.9627532958984376e-05,
      "step": 62524,
      "training_step_time": 7.0227673053741455
    },
    {
      "epoch": 5.962848663330078e-05,
      "model_forward_time": 1.715855360031128,
      "step": 62525
    },
    {
      "epoch": 5.962848663330078e-05,
      "step": 62525,
      "training_step_time": 7.019008159637451
    },
    {
      "epoch": 5.962944030761719e-05,
      "model_forward_time": 1.356567621231079,
      "step": 62526
    },
    {
      "epoch": 5.962944030761719e-05,
      "step": 62526,
      "training_step_time": 8.822332620620728
    },
    {
      "epoch": 5.9630393981933595e-05,
      "model_forward_time": 2.283357858657837,
      "step": 62527
    },
    {
      "epoch": 5.9630393981933595e-05,
      "step": 62527,
      "training_step_time": 8.976936101913452
    },
    {
      "epoch": 5.963134765625e-05,
      "model_forward_time": 1.7177720069885254,
      "step": 62528
    },
    {
      "epoch": 5.963134765625e-05,
      "step": 62528,
      "training_step_time": 9.082121849060059
    },
    {
      "epoch": 5.963230133056641e-05,
      "model_forward_time": 1.7151329517364502,
      "step": 62529
    },
    {
      "epoch": 5.963230133056641e-05,
      "step": 62529,
      "training_step_time": 8.605656147003174
    },
    {
      "epoch": 5.9633255004882814e-05,
      "grad_norm": 0.10652414709329605,
      "learning_rate": 3.3719681484806354e-06,
      "loss": 0.0577,
      "step": 62530
    },
    {
      "epoch": 5.9633255004882814e-05,
      "model_forward_time": 1.7221570014953613,
      "step": 62530
    },
    {
      "dynamics_loss_avg": 0.04261172469705343,
      "epoch": 5.9633255004882814e-05,
      "step": 62530
    },
    {
      "action_loss_avg": 0.001746851921780035,
      "epoch": 5.9633255004882814e-05,
      "step": 62530
    },
    {
      "epoch": 5.9633255004882814e-05,
      "step": 62530,
      "training_step_time": 7.816658973693848
    },
    {
      "epoch": 5.963420867919922e-05,
      "model_forward_time": 1.7278196811676025,
      "step": 62531
    },
    {
      "epoch": 5.963420867919922e-05,
      "step": 62531,
      "training_step_time": 7.0039472579956055
    },
    {
      "epoch": 5.963516235351562e-05,
      "model_forward_time": 1.7190899848937988,
      "step": 62532
    },
    {
      "epoch": 5.963516235351562e-05,
      "step": 62532,
      "training_step_time": 6.958875894546509
    },
    {
      "epoch": 5.963611602783203e-05,
      "model_forward_time": 1.718073844909668,
      "step": 62533
    },
    {
      "epoch": 5.963611602783203e-05,
      "step": 62533,
      "training_step_time": 6.9719438552856445
    },
    {
      "epoch": 5.9637069702148436e-05,
      "model_forward_time": 1.7034120559692383,
      "step": 62534
    },
    {
      "epoch": 5.9637069702148436e-05,
      "step": 62534,
      "training_step_time": 6.973041772842407
    },
    {
      "epoch": 5.963802337646484e-05,
      "model_forward_time": 1.704341173171997,
      "step": 62535
    },
    {
      "epoch": 5.963802337646484e-05,
      "step": 62535,
      "training_step_time": 7.785333633422852
    },
    {
      "epoch": 5.963897705078125e-05,
      "model_forward_time": 2.279391288757324,
      "step": 62536
    },
    {
      "epoch": 5.963897705078125e-05,
      "step": 62536,
      "training_step_time": 8.903367042541504
    },
    {
      "epoch": 5.9639930725097655e-05,
      "model_forward_time": 1.7016313076019287,
      "step": 62537
    },
    {
      "epoch": 5.9639930725097655e-05,
      "step": 62537,
      "training_step_time": 8.74101734161377
    },
    {
      "epoch": 5.964088439941406e-05,
      "model_forward_time": 1.7073004245758057,
      "step": 62538
    },
    {
      "epoch": 5.964088439941406e-05,
      "step": 62538,
      "training_step_time": 8.992103338241577
    },
    {
      "epoch": 5.964183807373047e-05,
      "model_forward_time": 1.3437132835388184,
      "step": 62539
    },
    {
      "epoch": 5.964183807373047e-05,
      "step": 62539,
      "training_step_time": 7.714995384216309
    },
    {
      "epoch": 5.9642791748046874e-05,
      "grad_norm": 0.09653091430664062,
      "learning_rate": 3.3704048728609373e-06,
      "loss": 0.0551,
      "step": 62540
    },
    {
      "epoch": 5.9642791748046874e-05,
      "model_forward_time": 1.716336727142334,
      "step": 62540
    },
    {
      "dynamics_loss_avg": 0.05349244624376297,
      "epoch": 5.9642791748046874e-05,
      "step": 62540
    },
    {
      "action_loss_avg": 0.0016372538695577532,
      "epoch": 5.9642791748046874e-05,
      "step": 62540
    },
    {
      "epoch": 5.9642791748046874e-05,
      "step": 62540,
      "training_step_time": 7.012671947479248
    },
    {
      "epoch": 5.964374542236328e-05,
      "model_forward_time": 1.711679220199585,
      "step": 62541
    },
    {
      "epoch": 5.964374542236328e-05,
      "step": 62541,
      "training_step_time": 6.999581575393677
    },
    {
      "epoch": 5.964469909667969e-05,
      "model_forward_time": 1.7061166763305664,
      "step": 62542
    },
    {
      "epoch": 5.964469909667969e-05,
      "step": 62542,
      "training_step_time": 7.023943185806274
    },
    {
      "epoch": 5.964565277099609e-05,
      "model_forward_time": 1.7036218643188477,
      "step": 62543
    },
    {
      "epoch": 5.964565277099609e-05,
      "step": 62543,
      "training_step_time": 7.038945913314819
    },
    {
      "epoch": 5.96466064453125e-05,
      "model_forward_time": 1.7118794918060303,
      "step": 62544
    },
    {
      "epoch": 5.96466064453125e-05,
      "step": 62544,
      "training_step_time": 7.801083326339722
    },
    {
      "epoch": 5.964756011962891e-05,
      "model_forward_time": 1.734297513961792,
      "step": 62545
    },
    {
      "epoch": 5.964756011962891e-05,
      "step": 62545,
      "training_step_time": 7.81852912902832
    },
    {
      "epoch": 5.964851379394531e-05,
      "model_forward_time": 2.250967264175415,
      "step": 62546
    },
    {
      "epoch": 5.964851379394531e-05,
      "step": 62546,
      "training_step_time": 9.071937322616577
    },
    {
      "epoch": 5.9649467468261716e-05,
      "model_forward_time": 1.7144169807434082,
      "step": 62547
    },
    {
      "epoch": 5.9649467468261716e-05,
      "step": 62547,
      "training_step_time": 8.51485824584961
    },
    {
      "epoch": 5.965042114257813e-05,
      "model_forward_time": 1.7099039554595947,
      "step": 62548
    },
    {
      "epoch": 5.965042114257813e-05,
      "step": 62548,
      "training_step_time": 7.822754621505737
    },
    {
      "epoch": 5.965137481689453e-05,
      "model_forward_time": 1.7082202434539795,
      "step": 62549
    },
    {
      "epoch": 5.965137481689453e-05,
      "step": 62549,
      "training_step_time": 6.937670469284058
    },
    {
      "epoch": 5.9652328491210935e-05,
      "grad_norm": 0.14300298690795898,
      "learning_rate": 3.3688417754513147e-06,
      "loss": 0.0568,
      "step": 62550
    },
    {
      "epoch": 5.9652328491210935e-05,
      "model_forward_time": 1.7151978015899658,
      "step": 62550
    },
    {
      "dynamics_loss_avg": 0.058588418364524844,
      "epoch": 5.9652328491210935e-05,
      "step": 62550
    },
    {
      "action_loss_avg": 0.001702269894303754,
      "epoch": 5.9652328491210935e-05,
      "step": 62550
    },
    {
      "epoch": 5.9652328491210935e-05,
      "step": 62550,
      "training_step_time": 7.026895046234131
    },
    {
      "epoch": 5.9653282165527346e-05,
      "model_forward_time": 1.7089321613311768,
      "step": 62551
    },
    {
      "epoch": 5.9653282165527346e-05,
      "step": 62551,
      "training_step_time": 7.023555040359497
    },
    {
      "epoch": 5.965423583984375e-05,
      "model_forward_time": 1.7082927227020264,
      "step": 62552
    },
    {
      "epoch": 5.965423583984375e-05,
      "step": 62552,
      "training_step_time": 6.973857402801514
    },
    {
      "epoch": 5.9655189514160154e-05,
      "model_forward_time": 1.7112674713134766,
      "step": 62553
    },
    {
      "epoch": 5.9655189514160154e-05,
      "step": 62553,
      "training_step_time": 7.7663414478302
    },
    {
      "epoch": 5.9656143188476564e-05,
      "model_forward_time": 1.7468483448028564,
      "step": 62554
    },
    {
      "epoch": 5.9656143188476564e-05,
      "step": 62554,
      "training_step_time": 7.846647024154663
    },
    {
      "epoch": 5.965709686279297e-05,
      "model_forward_time": 2.3126957416534424,
      "step": 62555
    },
    {
      "epoch": 5.965709686279297e-05,
      "step": 62555,
      "training_step_time": 7.929225444793701
    },
    {
      "epoch": 5.965805053710937e-05,
      "model_forward_time": 1.708453893661499,
      "step": 62556
    },
    {
      "epoch": 5.965805053710937e-05,
      "step": 62556,
      "training_step_time": 9.43660020828247
    },
    {
      "epoch": 5.965900421142578e-05,
      "model_forward_time": 1.7097923755645752,
      "step": 62557
    },
    {
      "epoch": 5.965900421142578e-05,
      "step": 62557,
      "training_step_time": 7.982752084732056
    },
    {
      "epoch": 5.965995788574219e-05,
      "model_forward_time": 1.7071590423583984,
      "step": 62558
    },
    {
      "epoch": 5.965995788574219e-05,
      "step": 62558,
      "training_step_time": 7.002222299575806
    },
    {
      "epoch": 5.966091156005859e-05,
      "model_forward_time": 1.713292121887207,
      "step": 62559
    },
    {
      "epoch": 5.966091156005859e-05,
      "step": 62559,
      "training_step_time": 6.963423490524292
    },
    {
      "epoch": 5.9661865234375e-05,
      "grad_norm": 0.11161120235919952,
      "learning_rate": 3.367278856422703e-06,
      "loss": 0.0561,
      "step": 62560
    },
    {
      "epoch": 5.9661865234375e-05,
      "model_forward_time": 1.71097731590271,
      "step": 62560
    },
    {
      "dynamics_loss_avg": 0.06966361720114947,
      "epoch": 5.9661865234375e-05,
      "step": 62560
    },
    {
      "action_loss_avg": 0.0018860390089685097,
      "epoch": 5.9661865234375e-05,
      "step": 62560
    },
    {
      "epoch": 5.9661865234375e-05,
      "step": 62560,
      "training_step_time": 6.9975762367248535
    },
    {
      "epoch": 5.9662818908691406e-05,
      "model_forward_time": 1.708432912826538,
      "step": 62561
    },
    {
      "epoch": 5.9662818908691406e-05,
      "step": 62561,
      "training_step_time": 7.003997564315796
    },
    {
      "epoch": 5.966377258300781e-05,
      "model_forward_time": 1.7121517658233643,
      "step": 62562
    },
    {
      "epoch": 5.966377258300781e-05,
      "step": 62562,
      "training_step_time": 7.735642910003662
    },
    {
      "epoch": 5.966472625732422e-05,
      "model_forward_time": 1.7075436115264893,
      "step": 62563
    },
    {
      "epoch": 5.966472625732422e-05,
      "step": 62563,
      "training_step_time": 9.94870400428772
    },
    {
      "epoch": 5.9665679931640625e-05,
      "model_forward_time": 2.34989333152771,
      "step": 62564
    },
    {
      "epoch": 5.9665679931640625e-05,
      "step": 62564,
      "training_step_time": 8.780683755874634
    },
    {
      "epoch": 5.966663360595703e-05,
      "model_forward_time": 1.7356553077697754,
      "step": 62565
    },
    {
      "epoch": 5.966663360595703e-05,
      "step": 62565,
      "training_step_time": 9.655158996582031
    },
    {
      "epoch": 5.966758728027344e-05,
      "model_forward_time": 1.7111992835998535,
      "step": 62566
    },
    {
      "epoch": 5.966758728027344e-05,
      "step": 62566,
      "training_step_time": 8.664884567260742
    },
    {
      "epoch": 5.9668540954589844e-05,
      "model_forward_time": 1.7060251235961914,
      "step": 62567
    },
    {
      "epoch": 5.9668540954589844e-05,
      "step": 62567,
      "training_step_time": 7.637867212295532
    },
    {
      "epoch": 5.966949462890625e-05,
      "model_forward_time": 1.706550121307373,
      "step": 62568
    },
    {
      "epoch": 5.966949462890625e-05,
      "step": 62568,
      "training_step_time": 7.002347707748413
    },
    {
      "epoch": 5.967044830322266e-05,
      "model_forward_time": 1.706751823425293,
      "step": 62569
    },
    {
      "epoch": 5.967044830322266e-05,
      "step": 62569,
      "training_step_time": 6.950705289840698
    },
    {
      "epoch": 5.967140197753906e-05,
      "grad_norm": 0.07455913722515106,
      "learning_rate": 3.3657161159460205e-06,
      "loss": 0.0565,
      "step": 62570
    },
    {
      "epoch": 5.967140197753906e-05,
      "model_forward_time": 1.7036185264587402,
      "step": 62570
    },
    {
      "dynamics_loss_avg": 0.05962648000568151,
      "epoch": 5.967140197753906e-05,
      "step": 62570
    },
    {
      "action_loss_avg": 0.0017831152479629963,
      "epoch": 5.967140197753906e-05,
      "step": 62570
    },
    {
      "epoch": 5.967140197753906e-05,
      "step": 62570,
      "training_step_time": 7.022510051727295
    },
    {
      "epoch": 5.967235565185547e-05,
      "model_forward_time": 1.7116355895996094,
      "step": 62571
    },
    {
      "epoch": 5.967235565185547e-05,
      "step": 62571,
      "training_step_time": 7.70851731300354
    },
    {
      "epoch": 5.967330932617188e-05,
      "model_forward_time": 1.7068781852722168,
      "step": 62572
    },
    {
      "epoch": 5.967330932617188e-05,
      "step": 62572,
      "training_step_time": 8.794052124023438
    },
    {
      "epoch": 5.967426300048828e-05,
      "model_forward_time": 2.276547908782959,
      "step": 62573
    },
    {
      "epoch": 5.967426300048828e-05,
      "step": 62573,
      "training_step_time": 8.755402326583862
    },
    {
      "epoch": 5.9675216674804686e-05,
      "model_forward_time": 1.6959705352783203,
      "step": 62574
    },
    {
      "epoch": 5.9675216674804686e-05,
      "step": 62574,
      "training_step_time": 8.854821681976318
    },
    {
      "epoch": 5.9676170349121096e-05,
      "model_forward_time": 1.6943504810333252,
      "step": 62575
    },
    {
      "epoch": 5.9676170349121096e-05,
      "step": 62575,
      "training_step_time": 7.934175729751587
    },
    {
      "epoch": 5.96771240234375e-05,
      "model_forward_time": 1.70086669921875,
      "step": 62576
    },
    {
      "epoch": 5.96771240234375e-05,
      "step": 62576,
      "training_step_time": 7.790934801101685
    },
    {
      "epoch": 5.9678077697753905e-05,
      "model_forward_time": 1.702294111251831,
      "step": 62577
    },
    {
      "epoch": 5.9678077697753905e-05,
      "step": 62577,
      "training_step_time": 7.045255184173584
    },
    {
      "epoch": 5.9679031372070315e-05,
      "model_forward_time": 1.6991229057312012,
      "step": 62578
    },
    {
      "epoch": 5.9679031372070315e-05,
      "step": 62578,
      "training_step_time": 7.109347820281982
    },
    {
      "epoch": 5.967998504638672e-05,
      "model_forward_time": 1.7015902996063232,
      "step": 62579
    },
    {
      "epoch": 5.967998504638672e-05,
      "step": 62579,
      "training_step_time": 7.104451417922974
    },
    {
      "epoch": 5.968093872070312e-05,
      "grad_norm": 0.08844759315252304,
      "learning_rate": 3.3641535541921666e-06,
      "loss": 0.0563,
      "step": 62580
    },
    {
      "epoch": 5.968093872070312e-05,
      "model_forward_time": 1.6991028785705566,
      "step": 62580
    },
    {
      "dynamics_loss_avg": 0.05856726448982954,
      "epoch": 5.968093872070312e-05,
      "step": 62580
    },
    {
      "action_loss_avg": 0.002635581768117845,
      "epoch": 5.968093872070312e-05,
      "step": 62580
    },
    {
      "epoch": 5.968093872070312e-05,
      "step": 62580,
      "training_step_time": 7.803981304168701
    },
    {
      "epoch": 5.9681892395019534e-05,
      "model_forward_time": 1.6983342170715332,
      "step": 62581
    },
    {
      "epoch": 5.9681892395019534e-05,
      "step": 62581,
      "training_step_time": 8.828655242919922
    },
    {
      "epoch": 5.968284606933594e-05,
      "model_forward_time": 2.2793543338775635,
      "step": 62582
    },
    {
      "epoch": 5.968284606933594e-05,
      "step": 62582,
      "training_step_time": 8.039760828018188
    },
    {
      "epoch": 5.968379974365234e-05,
      "model_forward_time": 1.7008087635040283,
      "step": 62583
    },
    {
      "epoch": 5.968379974365234e-05,
      "step": 62583,
      "training_step_time": 8.139328479766846
    },
    {
      "epoch": 5.968475341796875e-05,
      "model_forward_time": 1.7004191875457764,
      "step": 62584
    },
    {
      "epoch": 5.968475341796875e-05,
      "step": 62584,
      "training_step_time": 8.782670974731445
    },
    {
      "epoch": 5.968570709228516e-05,
      "model_forward_time": 1.7004013061523438,
      "step": 62585
    },
    {
      "epoch": 5.968570709228516e-05,
      "step": 62585,
      "training_step_time": 7.852272033691406
    },
    {
      "epoch": 5.968666076660156e-05,
      "model_forward_time": 1.7097575664520264,
      "step": 62586
    },
    {
      "epoch": 5.968666076660156e-05,
      "step": 62586,
      "training_step_time": 7.662004709243774
    },
    {
      "epoch": 5.968761444091797e-05,
      "model_forward_time": 1.7108969688415527,
      "step": 62587
    },
    {
      "epoch": 5.968761444091797e-05,
      "step": 62587,
      "training_step_time": 7.019733190536499
    },
    {
      "epoch": 5.9688568115234376e-05,
      "model_forward_time": 1.7091259956359863,
      "step": 62588
    },
    {
      "epoch": 5.9688568115234376e-05,
      "step": 62588,
      "training_step_time": 7.039421081542969
    },
    {
      "epoch": 5.968952178955078e-05,
      "model_forward_time": 1.7303063869476318,
      "step": 62589
    },
    {
      "epoch": 5.968952178955078e-05,
      "step": 62589,
      "training_step_time": 7.835925579071045
    },
    {
      "epoch": 5.969047546386719e-05,
      "grad_norm": 0.06606701016426086,
      "learning_rate": 3.3625911713320205e-06,
      "loss": 0.053,
      "step": 62590
    },
    {
      "epoch": 5.969047546386719e-05,
      "model_forward_time": 1.7039711475372314,
      "step": 62590
    },
    {
      "dynamics_loss_avg": 0.057009998708963394,
      "epoch": 5.969047546386719e-05,
      "step": 62590
    },
    {
      "action_loss_avg": 0.0014417890459299087,
      "epoch": 5.969047546386719e-05,
      "step": 62590
    },
    {
      "epoch": 5.969047546386719e-05,
      "step": 62590,
      "training_step_time": 7.831169605255127
    },
    {
      "epoch": 5.9691429138183595e-05,
      "model_forward_time": 2.276553153991699,
      "step": 62591
    },
    {
      "epoch": 5.9691429138183595e-05,
      "step": 62591,
      "training_step_time": 9.857598304748535
    },
    {
      "epoch": 5.96923828125e-05,
      "model_forward_time": 1.7709825038909912,
      "step": 62592
    },
    {
      "epoch": 5.96923828125e-05,
      "step": 62592,
      "training_step_time": 8.199915170669556
    },
    {
      "epoch": 5.969333648681641e-05,
      "model_forward_time": 1.6998891830444336,
      "step": 62593
    },
    {
      "epoch": 5.969333648681641e-05,
      "step": 62593,
      "training_step_time": 9.911462306976318
    },
    {
      "epoch": 5.9694290161132814e-05,
      "model_forward_time": 1.707139253616333,
      "step": 62594
    },
    {
      "epoch": 5.9694290161132814e-05,
      "step": 62594,
      "training_step_time": 7.881067752838135
    },
    {
      "epoch": 5.969524383544922e-05,
      "model_forward_time": 1.7074196338653564,
      "step": 62595
    },
    {
      "epoch": 5.969524383544922e-05,
      "step": 62595,
      "training_step_time": 7.535327434539795
    },
    {
      "epoch": 5.969619750976562e-05,
      "model_forward_time": 1.7114784717559814,
      "step": 62596
    },
    {
      "epoch": 5.969619750976562e-05,
      "step": 62596,
      "training_step_time": 7.032593011856079
    },
    {
      "epoch": 5.969715118408203e-05,
      "model_forward_time": 1.7100939750671387,
      "step": 62597
    },
    {
      "epoch": 5.969715118408203e-05,
      "step": 62597,
      "training_step_time": 7.014625310897827
    },
    {
      "epoch": 5.9698104858398437e-05,
      "model_forward_time": 1.7158029079437256,
      "step": 62598
    },
    {
      "epoch": 5.9698104858398437e-05,
      "step": 62598,
      "training_step_time": 7.806578159332275
    },
    {
      "epoch": 5.969905853271484e-05,
      "model_forward_time": 1.7032008171081543,
      "step": 62599
    },
    {
      "epoch": 5.969905853271484e-05,
      "step": 62599,
      "training_step_time": 8.521491289138794
    },
    {
      "epoch": 5.970001220703125e-05,
      "grad_norm": 0.10966403037309647,
      "learning_rate": 3.3610289675364406e-06,
      "loss": 0.0555,
      "step": 62600
    },
    {
      "epoch": 5.970001220703125e-05,
      "model_forward_time": 2.2924251556396484,
      "step": 62600
    },
    {
      "dynamics_loss_avg": 0.0588787516579032,
      "epoch": 5.970001220703125e-05,
      "step": 62600
    },
    {
      "action_loss_avg": 0.002066786104114726,
      "epoch": 5.970001220703125e-05,
      "step": 62600
    },
    {
      "epoch": 5.970001220703125e-05,
      "step": 62600,
      "training_step_time": 7.831432104110718
    },
    {
      "epoch": 5.9700965881347655e-05,
      "model_forward_time": 1.7523846626281738,
      "step": 62601
    },
    {
      "epoch": 5.9700965881347655e-05,
      "step": 62601,
      "training_step_time": 10.931768655776978
    },
    {
      "epoch": 5.970191955566406e-05,
      "model_forward_time": 1.7231674194335938,
      "step": 62602
    },
    {
      "epoch": 5.970191955566406e-05,
      "step": 62602,
      "training_step_time": 9.237043380737305
    },
    {
      "epoch": 5.970287322998047e-05,
      "model_forward_time": 1.7128982543945312,
      "step": 62603
    },
    {
      "epoch": 5.970287322998047e-05,
      "step": 62603,
      "training_step_time": 9.978044986724854
    },
    {
      "epoch": 5.9703826904296874e-05,
      "model_forward_time": 1.7207567691802979,
      "step": 62604
    },
    {
      "epoch": 5.9703826904296874e-05,
      "step": 62604,
      "training_step_time": 7.223890066146851
    },
    {
      "epoch": 5.970478057861328e-05,
      "model_forward_time": 1.7188372611999512,
      "step": 62605
    },
    {
      "epoch": 5.970478057861328e-05,
      "step": 62605,
      "training_step_time": 7.634811878204346
    },
    {
      "epoch": 5.970573425292969e-05,
      "model_forward_time": 1.7329199314117432,
      "step": 62606
    },
    {
      "epoch": 5.970573425292969e-05,
      "step": 62606,
      "training_step_time": 7.025349855422974
    },
    {
      "epoch": 5.970668792724609e-05,
      "model_forward_time": 1.7288281917572021,
      "step": 62607
    },
    {
      "epoch": 5.970668792724609e-05,
      "step": 62607,
      "training_step_time": 7.883163690567017
    },
    {
      "epoch": 5.97076416015625e-05,
      "model_forward_time": 1.717766284942627,
      "step": 62608
    },
    {
      "epoch": 5.97076416015625e-05,
      "step": 62608,
      "training_step_time": 7.805354118347168
    },
    {
      "epoch": 5.970859527587891e-05,
      "model_forward_time": 2.26824951171875,
      "step": 62609
    },
    {
      "epoch": 5.970859527587891e-05,
      "step": 62609,
      "training_step_time": 7.950174808502197
    },
    {
      "epoch": 5.970954895019531e-05,
      "grad_norm": 0.0917530283331871,
      "learning_rate": 3.3594669429762698e-06,
      "loss": 0.0538,
      "step": 62610
    },
    {
      "epoch": 5.970954895019531e-05,
      "model_forward_time": 1.7536139488220215,
      "step": 62610
    },
    {
      "dynamics_loss_avg": 0.08004489094018936,
      "epoch": 5.970954895019531e-05,
      "step": 62610
    },
    {
      "action_loss_avg": 0.0020823738363105805,
      "epoch": 5.970954895019531e-05,
      "step": 62610
    },
    {
      "epoch": 5.970954895019531e-05,
      "step": 62610,
      "training_step_time": 10.853489637374878
    },
    {
      "epoch": 5.9710502624511716e-05,
      "model_forward_time": 1.706181287765503,
      "step": 62611
    },
    {
      "epoch": 5.9710502624511716e-05,
      "step": 62611,
      "training_step_time": 8.81205415725708
    },
    {
      "epoch": 5.971145629882813e-05,
      "model_forward_time": 1.7094650268554688,
      "step": 62612
    },
    {
      "epoch": 5.971145629882813e-05,
      "step": 62612,
      "training_step_time": 8.000746965408325
    },
    {
      "epoch": 5.971240997314453e-05,
      "model_forward_time": 1.7116143703460693,
      "step": 62613
    },
    {
      "epoch": 5.971240997314453e-05,
      "step": 62613,
      "training_step_time": 7.86429500579834
    },
    {
      "epoch": 5.9713363647460935e-05,
      "model_forward_time": 1.7051656246185303,
      "step": 62614
    },
    {
      "epoch": 5.9713363647460935e-05,
      "step": 62614,
      "training_step_time": 7.649710416793823
    },
    {
      "epoch": 5.9714317321777346e-05,
      "model_forward_time": 1.7054717540740967,
      "step": 62615
    },
    {
      "epoch": 5.9714317321777346e-05,
      "step": 62615,
      "training_step_time": 7.028952598571777
    },
    {
      "epoch": 5.971527099609375e-05,
      "model_forward_time": 1.3252034187316895,
      "step": 62616
    },
    {
      "epoch": 5.971527099609375e-05,
      "step": 62616,
      "training_step_time": 7.793901205062866
    },
    {
      "epoch": 5.9716224670410154e-05,
      "model_forward_time": 1.704042673110962,
      "step": 62617
    },
    {
      "epoch": 5.9716224670410154e-05,
      "step": 62617,
      "training_step_time": 7.764890670776367
    },
    {
      "epoch": 5.9717178344726565e-05,
      "model_forward_time": 2.2497200965881348,
      "step": 62618
    },
    {
      "epoch": 5.9717178344726565e-05,
      "step": 62618,
      "training_step_time": 7.874133586883545
    },
    {
      "epoch": 5.971813201904297e-05,
      "model_forward_time": 1.730247974395752,
      "step": 62619
    },
    {
      "epoch": 5.971813201904297e-05,
      "step": 62619,
      "training_step_time": 9.90348505973816
    },
    {
      "epoch": 5.971908569335937e-05,
      "grad_norm": 0.08644479513168335,
      "learning_rate": 3.357905097822328e-06,
      "loss": 0.0525,
      "step": 62620
    },
    {
      "epoch": 5.971908569335937e-05,
      "model_forward_time": 1.702899694442749,
      "step": 62620
    },
    {
      "dynamics_loss_avg": 0.06137549355626106,
      "epoch": 5.971908569335937e-05,
      "step": 62620
    },
    {
      "action_loss_avg": 0.0014517059520585463,
      "epoch": 5.971908569335937e-05,
      "step": 62620
    },
    {
      "epoch": 5.971908569335937e-05,
      "step": 62620,
      "training_step_time": 8.682024478912354
    },
    {
      "epoch": 5.972003936767578e-05,
      "model_forward_time": 1.7006661891937256,
      "step": 62621
    },
    {
      "epoch": 5.972003936767578e-05,
      "step": 62621,
      "training_step_time": 7.862976312637329
    },
    {
      "epoch": 5.972099304199219e-05,
      "model_forward_time": 1.7035515308380127,
      "step": 62622
    },
    {
      "epoch": 5.972099304199219e-05,
      "step": 62622,
      "training_step_time": 7.470297574996948
    },
    {
      "epoch": 5.972194671630859e-05,
      "model_forward_time": 1.7069432735443115,
      "step": 62623
    },
    {
      "epoch": 5.972194671630859e-05,
      "step": 62623,
      "training_step_time": 7.778755187988281
    },
    {
      "epoch": 5.9722900390625e-05,
      "model_forward_time": 1.704472303390503,
      "step": 62624
    },
    {
      "epoch": 5.9722900390625e-05,
      "step": 62624,
      "training_step_time": 7.033457517623901
    },
    {
      "epoch": 5.9723854064941406e-05,
      "model_forward_time": 1.7043328285217285,
      "step": 62625
    },
    {
      "epoch": 5.9723854064941406e-05,
      "step": 62625,
      "training_step_time": 7.792139053344727
    },
    {
      "epoch": 5.972480773925781e-05,
      "model_forward_time": 1.701401948928833,
      "step": 62626
    },
    {
      "epoch": 5.972480773925781e-05,
      "step": 62626,
      "training_step_time": 7.777957916259766
    },
    {
      "epoch": 5.972576141357422e-05,
      "model_forward_time": 2.2724862098693848,
      "step": 62627
    },
    {
      "epoch": 5.972576141357422e-05,
      "step": 62627,
      "training_step_time": 7.878643989562988
    },
    {
      "epoch": 5.9726715087890625e-05,
      "model_forward_time": 1.7101385593414307,
      "step": 62628
    },
    {
      "epoch": 5.9726715087890625e-05,
      "step": 62628,
      "training_step_time": 9.063559770584106
    },
    {
      "epoch": 5.972766876220703e-05,
      "model_forward_time": 1.7103703022003174,
      "step": 62629
    },
    {
      "epoch": 5.972766876220703e-05,
      "step": 62629,
      "training_step_time": 9.72449016571045
    },
    {
      "epoch": 5.972862243652344e-05,
      "grad_norm": 0.08288279175758362,
      "learning_rate": 3.356343432245416e-06,
      "loss": 0.0549,
      "step": 62630
    },
    {
      "epoch": 5.972862243652344e-05,
      "model_forward_time": 1.7063498497009277,
      "step": 62630
    },
    {
      "dynamics_loss_avg": 0.04956036014482379,
      "epoch": 5.972862243652344e-05,
      "step": 62630
    },
    {
      "action_loss_avg": 0.001820576298632659,
      "epoch": 5.972862243652344e-05,
      "step": 62630
    },
    {
      "epoch": 5.972862243652344e-05,
      "step": 62630,
      "training_step_time": 8.895082712173462
    },
    {
      "epoch": 5.9729576110839844e-05,
      "model_forward_time": 1.7111232280731201,
      "step": 62631
    },
    {
      "epoch": 5.9729576110839844e-05,
      "step": 62631,
      "training_step_time": 7.959300756454468
    },
    {
      "epoch": 5.973052978515625e-05,
      "model_forward_time": 1.6991755962371826,
      "step": 62632
    },
    {
      "epoch": 5.973052978515625e-05,
      "step": 62632,
      "training_step_time": 7.877481460571289
    },
    {
      "epoch": 5.973148345947266e-05,
      "model_forward_time": 1.7432904243469238,
      "step": 62633
    },
    {
      "epoch": 5.973148345947266e-05,
      "step": 62633,
      "training_step_time": 6.9404261112213135
    },
    {
      "epoch": 5.973243713378906e-05,
      "model_forward_time": 1.712822437286377,
      "step": 62634
    },
    {
      "epoch": 5.973243713378906e-05,
      "step": 62634,
      "training_step_time": 7.735579967498779
    },
    {
      "epoch": 5.973339080810547e-05,
      "model_forward_time": 1.7301599979400635,
      "step": 62635
    },
    {
      "epoch": 5.973339080810547e-05,
      "step": 62635,
      "training_step_time": 7.694211721420288
    },
    {
      "epoch": 5.973434448242188e-05,
      "model_forward_time": 2.313260078430176,
      "step": 62636
    },
    {
      "epoch": 5.973434448242188e-05,
      "step": 62636,
      "training_step_time": 7.809759855270386
    },
    {
      "epoch": 5.973529815673828e-05,
      "model_forward_time": 1.7324442863464355,
      "step": 62637
    },
    {
      "epoch": 5.973529815673828e-05,
      "step": 62637,
      "training_step_time": 9.651740312576294
    },
    {
      "epoch": 5.9736251831054686e-05,
      "model_forward_time": 1.7824885845184326,
      "step": 62638
    },
    {
      "epoch": 5.9736251831054686e-05,
      "step": 62638,
      "training_step_time": 7.890584945678711
    },
    {
      "epoch": 5.9737205505371097e-05,
      "model_forward_time": 1.7229197025299072,
      "step": 62639
    },
    {
      "epoch": 5.9737205505371097e-05,
      "step": 62639,
      "training_step_time": 7.945600271224976
    },
    {
      "epoch": 5.97381591796875e-05,
      "grad_norm": 0.1326802521944046,
      "learning_rate": 3.354781946416315e-06,
      "loss": 0.0513,
      "step": 62640
    },
    {
      "epoch": 5.97381591796875e-05,
      "model_forward_time": 1.730348825454712,
      "step": 62640
    },
    {
      "dynamics_loss_avg": 0.05142560247331858,
      "epoch": 5.97381591796875e-05,
      "step": 62640
    },
    {
      "action_loss_avg": 0.001960358954966068,
      "epoch": 5.97381591796875e-05,
      "step": 62640
    },
    {
      "epoch": 5.97381591796875e-05,
      "step": 62640,
      "training_step_time": 8.013424158096313
    },
    {
      "epoch": 5.9739112854003905e-05,
      "model_forward_time": 1.7281932830810547,
      "step": 62641
    },
    {
      "epoch": 5.9739112854003905e-05,
      "step": 62641,
      "training_step_time": 8.752402305603027
    },
    {
      "epoch": 5.9740066528320315e-05,
      "model_forward_time": 1.7266104221343994,
      "step": 62642
    },
    {
      "epoch": 5.9740066528320315e-05,
      "step": 62642,
      "training_step_time": 7.7237138748168945
    },
    {
      "epoch": 5.974102020263672e-05,
      "model_forward_time": 1.7212870121002197,
      "step": 62643
    },
    {
      "epoch": 5.974102020263672e-05,
      "step": 62643,
      "training_step_time": 7.799323320388794
    },
    {
      "epoch": 5.9741973876953123e-05,
      "model_forward_time": 1.6939036846160889,
      "step": 62644
    },
    {
      "epoch": 5.9741973876953123e-05,
      "step": 62644,
      "training_step_time": 7.755122900009155
    },
    {
      "epoch": 5.9742927551269534e-05,
      "model_forward_time": 2.258868455886841,
      "step": 62645
    },
    {
      "epoch": 5.9742927551269534e-05,
      "step": 62645,
      "training_step_time": 7.853304862976074
    },
    {
      "epoch": 5.974388122558594e-05,
      "model_forward_time": 1.7061307430267334,
      "step": 62646
    },
    {
      "epoch": 5.974388122558594e-05,
      "step": 62646,
      "training_step_time": 8.942615032196045
    },
    {
      "epoch": 5.974483489990234e-05,
      "model_forward_time": 1.7321150302886963,
      "step": 62647
    },
    {
      "epoch": 5.974483489990234e-05,
      "step": 62647,
      "training_step_time": 9.507655143737793
    },
    {
      "epoch": 5.974578857421875e-05,
      "model_forward_time": 1.686781883239746,
      "step": 62648
    },
    {
      "epoch": 5.974578857421875e-05,
      "step": 62648,
      "training_step_time": 9.602729320526123
    },
    {
      "epoch": 5.974674224853516e-05,
      "model_forward_time": 1.6958134174346924,
      "step": 62649
    },
    {
      "epoch": 5.974674224853516e-05,
      "step": 62649,
      "training_step_time": 7.95087194442749
    },
    {
      "epoch": 5.974769592285156e-05,
      "grad_norm": 0.10491082817316055,
      "learning_rate": 3.3532206405057865e-06,
      "loss": 0.0556,
      "step": 62650
    },
    {
      "epoch": 5.974769592285156e-05,
      "model_forward_time": 1.7001159191131592,
      "step": 62650
    },
    {
      "dynamics_loss_avg": 0.034371128957718614,
      "epoch": 5.974769592285156e-05,
      "step": 62650
    },
    {
      "action_loss_avg": 0.0007924179604742676,
      "epoch": 5.974769592285156e-05,
      "step": 62650
    },
    {
      "epoch": 5.974769592285156e-05,
      "step": 62650,
      "training_step_time": 7.861032247543335
    },
    {
      "epoch": 5.974864959716797e-05,
      "model_forward_time": 1.7010157108306885,
      "step": 62651
    },
    {
      "epoch": 5.974864959716797e-05,
      "step": 62651,
      "training_step_time": 7.623553276062012
    },
    {
      "epoch": 5.9749603271484376e-05,
      "model_forward_time": 1.6951680183410645,
      "step": 62652
    },
    {
      "epoch": 5.9749603271484376e-05,
      "step": 62652,
      "training_step_time": 7.838357925415039
    },
    {
      "epoch": 5.975055694580078e-05,
      "model_forward_time": 1.7022650241851807,
      "step": 62653
    },
    {
      "epoch": 5.975055694580078e-05,
      "step": 62653,
      "training_step_time": 7.761116027832031
    },
    {
      "epoch": 5.975151062011719e-05,
      "model_forward_time": 1.6950085163116455,
      "step": 62654
    },
    {
      "epoch": 5.975151062011719e-05,
      "step": 62654,
      "training_step_time": 8.591498136520386
    },
    {
      "epoch": 5.9752464294433595e-05,
      "model_forward_time": 2.2456820011138916,
      "step": 62655
    },
    {
      "epoch": 5.9752464294433595e-05,
      "step": 62655,
      "training_step_time": 7.9495604038238525
    },
    {
      "epoch": 5.975341796875e-05,
      "model_forward_time": 1.714235544204712,
      "step": 62656
    },
    {
      "epoch": 5.975341796875e-05,
      "step": 62656,
      "training_step_time": 9.963874340057373
    },
    {
      "epoch": 5.97543716430664e-05,
      "model_forward_time": 1.6901960372924805,
      "step": 62657
    },
    {
      "epoch": 5.97543716430664e-05,
      "step": 62657,
      "training_step_time": 9.51302170753479
    },
    {
      "epoch": 5.9755325317382814e-05,
      "model_forward_time": 1.6984913349151611,
      "step": 62658
    },
    {
      "epoch": 5.9755325317382814e-05,
      "step": 62658,
      "training_step_time": 7.8813230991363525
    },
    {
      "epoch": 5.975627899169922e-05,
      "model_forward_time": 1.68965482711792,
      "step": 62659
    },
    {
      "epoch": 5.975627899169922e-05,
      "step": 62659,
      "training_step_time": 8.939391851425171
    },
    {
      "epoch": 5.975723266601562e-05,
      "grad_norm": 0.10093265026807785,
      "learning_rate": 3.3516595146845732e-06,
      "loss": 0.0557,
      "step": 62660
    },
    {
      "epoch": 5.975723266601562e-05,
      "model_forward_time": 1.6889519691467285,
      "step": 62660
    },
    {
      "dynamics_loss_avg": 0.05775912720710039,
      "epoch": 5.975723266601562e-05,
      "step": 62660
    },
    {
      "action_loss_avg": 0.0014798531774431467,
      "epoch": 5.975723266601562e-05,
      "step": 62660
    },
    {
      "epoch": 5.975723266601562e-05,
      "step": 62660,
      "training_step_time": 7.580881357192993
    },
    {
      "epoch": 5.975818634033203e-05,
      "model_forward_time": 1.6861989498138428,
      "step": 62661
    },
    {
      "epoch": 5.975818634033203e-05,
      "step": 62661,
      "training_step_time": 7.853536367416382
    },
    {
      "epoch": 5.9759140014648437e-05,
      "model_forward_time": 1.7029626369476318,
      "step": 62662
    },
    {
      "epoch": 5.9759140014648437e-05,
      "step": 62662,
      "training_step_time": 7.728866815567017
    },
    {
      "epoch": 5.976009368896484e-05,
      "model_forward_time": 1.7101755142211914,
      "step": 62663
    },
    {
      "epoch": 5.976009368896484e-05,
      "step": 62663,
      "training_step_time": 7.485924482345581
    },
    {
      "epoch": 5.976104736328125e-05,
      "model_forward_time": 2.267583131790161,
      "step": 62664
    },
    {
      "epoch": 5.976104736328125e-05,
      "step": 62664,
      "training_step_time": 8.840147256851196
    },
    {
      "epoch": 5.9762001037597655e-05,
      "model_forward_time": 1.7052268981933594,
      "step": 62665
    },
    {
      "epoch": 5.9762001037597655e-05,
      "step": 62665,
      "training_step_time": 8.730355024337769
    },
    {
      "epoch": 5.976295471191406e-05,
      "model_forward_time": 1.6962201595306396,
      "step": 62666
    },
    {
      "epoch": 5.976295471191406e-05,
      "step": 62666,
      "training_step_time": 7.990862607955933
    },
    {
      "epoch": 5.976390838623047e-05,
      "model_forward_time": 1.727036952972412,
      "step": 62667
    },
    {
      "epoch": 5.976390838623047e-05,
      "step": 62667,
      "training_step_time": 9.057701587677002
    },
    {
      "epoch": 5.9764862060546874e-05,
      "model_forward_time": 1.3362488746643066,
      "step": 62668
    },
    {
      "epoch": 5.9764862060546874e-05,
      "step": 62668,
      "training_step_time": 8.009230613708496
    },
    {
      "epoch": 5.976581573486328e-05,
      "model_forward_time": 1.7290940284729004,
      "step": 62669
    },
    {
      "epoch": 5.976581573486328e-05,
      "step": 62669,
      "training_step_time": 7.893348932266235
    },
    {
      "epoch": 5.976676940917969e-05,
      "grad_norm": 0.11585596948862076,
      "learning_rate": 3.350098569123395e-06,
      "loss": 0.0589,
      "step": 62670
    },
    {
      "epoch": 5.976676940917969e-05,
      "model_forward_time": 1.7177283763885498,
      "step": 62670
    },
    {
      "dynamics_loss_avg": 0.053517929278314114,
      "epoch": 5.976676940917969e-05,
      "step": 62670
    },
    {
      "action_loss_avg": 0.0015078091702889651,
      "epoch": 5.976676940917969e-05,
      "step": 62670
    },
    {
      "epoch": 5.976676940917969e-05,
      "step": 62670,
      "training_step_time": 7.90740966796875
    },
    {
      "epoch": 5.976772308349609e-05,
      "model_forward_time": 1.719965934753418,
      "step": 62671
    },
    {
      "epoch": 5.976772308349609e-05,
      "step": 62671,
      "training_step_time": 7.592681169509888
    },
    {
      "epoch": 5.97686767578125e-05,
      "model_forward_time": 1.7215867042541504,
      "step": 62672
    },
    {
      "epoch": 5.97686767578125e-05,
      "step": 62672,
      "training_step_time": 7.803276062011719
    },
    {
      "epoch": 5.976963043212891e-05,
      "model_forward_time": 2.3104050159454346,
      "step": 62673
    },
    {
      "epoch": 5.976963043212891e-05,
      "step": 62673,
      "training_step_time": 7.95289421081543
    },
    {
      "epoch": 5.977058410644531e-05,
      "model_forward_time": 1.7139623165130615,
      "step": 62674
    },
    {
      "epoch": 5.977058410644531e-05,
      "step": 62674,
      "training_step_time": 9.093023538589478
    },
    {
      "epoch": 5.9771537780761716e-05,
      "model_forward_time": 1.7739620208740234,
      "step": 62675
    },
    {
      "epoch": 5.9771537780761716e-05,
      "step": 62675,
      "training_step_time": 7.89797306060791
    },
    {
      "epoch": 5.977249145507813e-05,
      "model_forward_time": 1.7256393432617188,
      "step": 62676
    },
    {
      "epoch": 5.977249145507813e-05,
      "step": 62676,
      "training_step_time": 8.689752101898193
    },
    {
      "epoch": 5.977344512939453e-05,
      "model_forward_time": 1.7185275554656982,
      "step": 62677
    },
    {
      "epoch": 5.977344512939453e-05,
      "step": 62677,
      "training_step_time": 7.948490381240845
    },
    {
      "epoch": 5.9774398803710935e-05,
      "model_forward_time": 1.7197868824005127,
      "step": 62678
    },
    {
      "epoch": 5.9774398803710935e-05,
      "step": 62678,
      "training_step_time": 8.048663854598999
    },
    {
      "epoch": 5.9775352478027346e-05,
      "model_forward_time": 1.7233989238739014,
      "step": 62679
    },
    {
      "epoch": 5.9775352478027346e-05,
      "step": 62679,
      "training_step_time": 7.858646631240845
    },
    {
      "epoch": 5.977630615234375e-05,
      "grad_norm": 0.13203956186771393,
      "learning_rate": 3.3485378039929583e-06,
      "loss": 0.0611,
      "step": 62680
    },
    {
      "epoch": 5.977630615234375e-05,
      "model_forward_time": 1.7107932567596436,
      "step": 62680
    },
    {
      "dynamics_loss_avg": 0.06866226028650999,
      "epoch": 5.977630615234375e-05,
      "step": 62680
    },
    {
      "action_loss_avg": 0.0017365033316309563,
      "epoch": 5.977630615234375e-05,
      "step": 62680
    },
    {
      "epoch": 5.977630615234375e-05,
      "step": 62680,
      "training_step_time": 7.499088525772095
    },
    {
      "epoch": 5.9777259826660154e-05,
      "model_forward_time": 1.7204430103302002,
      "step": 62681
    },
    {
      "epoch": 5.9777259826660154e-05,
      "step": 62681,
      "training_step_time": 8.226413488388062
    },
    {
      "epoch": 5.9778213500976565e-05,
      "model_forward_time": 2.297947883605957,
      "step": 62682
    },
    {
      "epoch": 5.9778213500976565e-05,
      "step": 62682,
      "training_step_time": 8.060998678207397
    },
    {
      "epoch": 5.977916717529297e-05,
      "model_forward_time": 1.7150931358337402,
      "step": 62683
    },
    {
      "epoch": 5.977916717529297e-05,
      "step": 62683,
      "training_step_time": 7.981966257095337
    },
    {
      "epoch": 5.978012084960937e-05,
      "model_forward_time": 1.7744081020355225,
      "step": 62684
    },
    {
      "epoch": 5.978012084960937e-05,
      "step": 62684,
      "training_step_time": 7.757776260375977
    },
    {
      "epoch": 5.9781074523925783e-05,
      "model_forward_time": 1.7163617610931396,
      "step": 62685
    },
    {
      "epoch": 5.9781074523925783e-05,
      "step": 62685,
      "training_step_time": 7.970986604690552
    },
    {
      "epoch": 5.978202819824219e-05,
      "model_forward_time": 1.728569507598877,
      "step": 62686
    },
    {
      "epoch": 5.978202819824219e-05,
      "step": 62686,
      "training_step_time": 7.9573211669921875
    },
    {
      "epoch": 5.978298187255859e-05,
      "model_forward_time": 1.714735746383667,
      "step": 62687
    },
    {
      "epoch": 5.978298187255859e-05,
      "step": 62687,
      "training_step_time": 8.000123739242554
    },
    {
      "epoch": 5.9783935546875e-05,
      "model_forward_time": 1.7211723327636719,
      "step": 62688
    },
    {
      "epoch": 5.9783935546875e-05,
      "step": 62688,
      "training_step_time": 7.976572275161743
    },
    {
      "epoch": 5.9784889221191406e-05,
      "model_forward_time": 1.723332166671753,
      "step": 62689
    },
    {
      "epoch": 5.9784889221191406e-05,
      "step": 62689,
      "training_step_time": 7.527301788330078
    },
    {
      "epoch": 5.978584289550781e-05,
      "grad_norm": 0.10616806894540787,
      "learning_rate": 3.346977219463945e-06,
      "loss": 0.0569,
      "step": 62690
    },
    {
      "epoch": 5.978584289550781e-05,
      "model_forward_time": 1.7209811210632324,
      "step": 62690
    },
    {
      "dynamics_loss_avg": 0.06133933197706938,
      "epoch": 5.978584289550781e-05,
      "step": 62690
    },
    {
      "action_loss_avg": 0.0015811771707376465,
      "epoch": 5.978584289550781e-05,
      "step": 62690
    },
    {
      "epoch": 5.978584289550781e-05,
      "step": 62690,
      "training_step_time": 7.733518838882446
    },
    {
      "epoch": 5.978679656982422e-05,
      "model_forward_time": 1.7201833724975586,
      "step": 62691
    },
    {
      "epoch": 5.978679656982422e-05,
      "step": 62691,
      "training_step_time": 8.44877004623413
    },
    {
      "epoch": 5.9787750244140625e-05,
      "model_forward_time": 1.7178776264190674,
      "step": 62692
    },
    {
      "epoch": 5.9787750244140625e-05,
      "step": 62692,
      "training_step_time": 8.94969367980957
    },
    {
      "epoch": 5.978870391845703e-05,
      "model_forward_time": 1.3507449626922607,
      "step": 62693
    },
    {
      "epoch": 5.978870391845703e-05,
      "step": 62693,
      "training_step_time": 8.0766921043396
    },
    {
      "epoch": 5.978965759277344e-05,
      "model_forward_time": 1.706458330154419,
      "step": 62694
    },
    {
      "epoch": 5.978965759277344e-05,
      "step": 62694,
      "training_step_time": 8.971322774887085
    },
    {
      "epoch": 5.9790611267089844e-05,
      "model_forward_time": 1.7303590774536133,
      "step": 62695
    },
    {
      "epoch": 5.9790611267089844e-05,
      "step": 62695,
      "training_step_time": 7.941330671310425
    },
    {
      "epoch": 5.979156494140625e-05,
      "model_forward_time": 1.7216203212738037,
      "step": 62696
    },
    {
      "epoch": 5.979156494140625e-05,
      "step": 62696,
      "training_step_time": 7.959134817123413
    },
    {
      "epoch": 5.979251861572266e-05,
      "model_forward_time": 1.7289445400238037,
      "step": 62697
    },
    {
      "epoch": 5.979251861572266e-05,
      "step": 62697,
      "training_step_time": 8.598433494567871
    },
    {
      "epoch": 5.979347229003906e-05,
      "model_forward_time": 1.70298171043396,
      "step": 62698
    },
    {
      "epoch": 5.979347229003906e-05,
      "step": 62698,
      "training_step_time": 7.9662926197052
    },
    {
      "epoch": 5.979442596435547e-05,
      "model_forward_time": 1.7143936157226562,
      "step": 62699
    },
    {
      "epoch": 5.979442596435547e-05,
      "step": 62699,
      "training_step_time": 7.789394855499268
    },
    {
      "epoch": 5.979537963867188e-05,
      "grad_norm": 0.09048641473054886,
      "learning_rate": 3.345416815707018e-06,
      "loss": 0.0573,
      "step": 62700
    },
    {
      "epoch": 5.979537963867188e-05,
      "model_forward_time": 1.729814052581787,
      "step": 62700
    },
    {
      "dynamics_loss_avg": 0.043322419933974746,
      "epoch": 5.979537963867188e-05,
      "step": 62700
    },
    {
      "action_loss_avg": 0.002074485836783424,
      "epoch": 5.979537963867188e-05,
      "step": 62700
    },
    {
      "epoch": 5.979537963867188e-05,
      "step": 62700,
      "training_step_time": 8.927301406860352
    },
    {
      "epoch": 5.979633331298828e-05,
      "model_forward_time": 1.885859489440918,
      "step": 62701
    },
    {
      "epoch": 5.979633331298828e-05,
      "step": 62701,
      "training_step_time": 9.628461122512817
    },
    {
      "epoch": 5.9797286987304686e-05,
      "model_forward_time": 1.7110581398010254,
      "step": 62702
    },
    {
      "epoch": 5.9797286987304686e-05,
      "step": 62702,
      "training_step_time": 9.080646991729736
    },
    {
      "epoch": 5.9798240661621097e-05,
      "model_forward_time": 1.7036314010620117,
      "step": 62703
    },
    {
      "epoch": 5.9798240661621097e-05,
      "step": 62703,
      "training_step_time": 9.544797658920288
    },
    {
      "epoch": 5.97991943359375e-05,
      "model_forward_time": 1.71484375,
      "step": 62704
    },
    {
      "epoch": 5.97991943359375e-05,
      "step": 62704,
      "training_step_time": 8.021246433258057
    },
    {
      "epoch": 5.9800148010253905e-05,
      "model_forward_time": 1.706876516342163,
      "step": 62705
    },
    {
      "epoch": 5.9800148010253905e-05,
      "step": 62705,
      "training_step_time": 8.018016815185547
    },
    {
      "epoch": 5.9801101684570315e-05,
      "model_forward_time": 1.3527281284332275,
      "step": 62706
    },
    {
      "epoch": 5.9801101684570315e-05,
      "step": 62706,
      "training_step_time": 8.007339715957642
    },
    {
      "epoch": 5.980205535888672e-05,
      "model_forward_time": 1.705728530883789,
      "step": 62707
    },
    {
      "epoch": 5.980205535888672e-05,
      "step": 62707,
      "training_step_time": 7.879442453384399
    },
    {
      "epoch": 5.9803009033203123e-05,
      "model_forward_time": 1.7109830379486084,
      "step": 62708
    },
    {
      "epoch": 5.9803009033203123e-05,
      "step": 62708,
      "training_step_time": 7.767283916473389
    },
    {
      "epoch": 5.9803962707519534e-05,
      "model_forward_time": 1.347702980041504,
      "step": 62709
    },
    {
      "epoch": 5.9803962707519534e-05,
      "step": 62709,
      "training_step_time": 7.95989727973938
    },
    {
      "epoch": 5.980491638183594e-05,
      "grad_norm": 0.1289418339729309,
      "learning_rate": 3.3438565928928184e-06,
      "loss": 0.0567,
      "step": 62710
    },
    {
      "epoch": 5.980491638183594e-05,
      "model_forward_time": 2.249999761581421,
      "step": 62710
    },
    {
      "dynamics_loss_avg": 0.0642002485692501,
      "epoch": 5.980491638183594e-05,
      "step": 62710
    },
    {
      "action_loss_avg": 0.0019811128091532736,
      "epoch": 5.980491638183594e-05,
      "step": 62710
    },
    {
      "epoch": 5.980491638183594e-05,
      "step": 62710,
      "training_step_time": 7.836221694946289
    },
    {
      "epoch": 5.980587005615234e-05,
      "model_forward_time": 1.711869478225708,
      "step": 62711
    },
    {
      "epoch": 5.980587005615234e-05,
      "step": 62711,
      "training_step_time": 8.723588943481445
    },
    {
      "epoch": 5.980682373046875e-05,
      "model_forward_time": 1.767336368560791,
      "step": 62712
    },
    {
      "epoch": 5.980682373046875e-05,
      "step": 62712,
      "training_step_time": 10.081841945648193
    },
    {
      "epoch": 5.980777740478516e-05,
      "model_forward_time": 1.7136740684509277,
      "step": 62713
    },
    {
      "epoch": 5.980777740478516e-05,
      "step": 62713,
      "training_step_time": 7.948121547698975
    },
    {
      "epoch": 5.980873107910156e-05,
      "model_forward_time": 1.7169978618621826,
      "step": 62714
    },
    {
      "epoch": 5.980873107910156e-05,
      "step": 62714,
      "training_step_time": 7.902271747589111
    },
    {
      "epoch": 5.980968475341797e-05,
      "model_forward_time": 1.7204113006591797,
      "step": 62715
    },
    {
      "epoch": 5.980968475341797e-05,
      "step": 62715,
      "training_step_time": 7.974563360214233
    },
    {
      "epoch": 5.9810638427734376e-05,
      "model_forward_time": 1.710603952407837,
      "step": 62716
    },
    {
      "epoch": 5.9810638427734376e-05,
      "step": 62716,
      "training_step_time": 7.813459157943726
    },
    {
      "epoch": 5.981159210205078e-05,
      "model_forward_time": 1.7170491218566895,
      "step": 62717
    },
    {
      "epoch": 5.981159210205078e-05,
      "step": 62717,
      "training_step_time": 7.761602163314819
    },
    {
      "epoch": 5.981254577636719e-05,
      "model_forward_time": 1.7192106246948242,
      "step": 62718
    },
    {
      "epoch": 5.981254577636719e-05,
      "step": 62718,
      "training_step_time": 9.457042455673218
    },
    {
      "epoch": 5.9813499450683595e-05,
      "model_forward_time": 1.7060976028442383,
      "step": 62719
    },
    {
      "epoch": 5.9813499450683595e-05,
      "step": 62719,
      "training_step_time": 7.836925983428955
    },
    {
      "epoch": 5.9814453125e-05,
      "grad_norm": 0.11711715161800385,
      "learning_rate": 3.3422965511919737e-06,
      "loss": 0.0555,
      "step": 62720
    },
    {
      "epoch": 5.9814453125e-05,
      "model_forward_time": 2.294081926345825,
      "step": 62720
    },
    {
      "dynamics_loss_avg": 0.05105836093425751,
      "epoch": 5.9814453125e-05,
      "step": 62720
    },
    {
      "action_loss_avg": 0.0015583980362862348,
      "epoch": 5.9814453125e-05,
      "step": 62720
    },
    {
      "epoch": 5.9814453125e-05,
      "step": 62720,
      "training_step_time": 8.496453523635864
    },
    {
      "epoch": 5.98154067993164e-05,
      "model_forward_time": 1.764753818511963,
      "step": 62721
    },
    {
      "epoch": 5.98154067993164e-05,
      "step": 62721,
      "training_step_time": 9.81162428855896
    },
    {
      "epoch": 5.9816360473632814e-05,
      "model_forward_time": 1.7206041812896729,
      "step": 62722
    },
    {
      "epoch": 5.9816360473632814e-05,
      "step": 62722,
      "training_step_time": 7.875528573989868
    },
    {
      "epoch": 5.981731414794922e-05,
      "model_forward_time": 1.7149028778076172,
      "step": 62723
    },
    {
      "epoch": 5.981731414794922e-05,
      "step": 62723,
      "training_step_time": 9.262528657913208
    },
    {
      "epoch": 5.981826782226562e-05,
      "model_forward_time": 1.720658540725708,
      "step": 62724
    },
    {
      "epoch": 5.981826782226562e-05,
      "step": 62724,
      "training_step_time": 7.912121057510376
    },
    {
      "epoch": 5.981922149658203e-05,
      "model_forward_time": 1.70829439163208,
      "step": 62725
    },
    {
      "epoch": 5.981922149658203e-05,
      "step": 62725,
      "training_step_time": 7.5942628383636475
    },
    {
      "epoch": 5.982017517089844e-05,
      "model_forward_time": 1.7134461402893066,
      "step": 62726
    },
    {
      "epoch": 5.982017517089844e-05,
      "step": 62726,
      "training_step_time": 7.738064527511597
    },
    {
      "epoch": 5.982112884521484e-05,
      "model_forward_time": 1.7161509990692139,
      "step": 62727
    },
    {
      "epoch": 5.982112884521484e-05,
      "step": 62727,
      "training_step_time": 7.887012958526611
    },
    {
      "epoch": 5.982208251953125e-05,
      "model_forward_time": 1.716137409210205,
      "step": 62728
    },
    {
      "epoch": 5.982208251953125e-05,
      "step": 62728,
      "training_step_time": 7.797159671783447
    },
    {
      "epoch": 5.9823036193847655e-05,
      "model_forward_time": 1.8764817714691162,
      "step": 62729
    },
    {
      "epoch": 5.9823036193847655e-05,
      "step": 62729,
      "training_step_time": 8.490882158279419
    },
    {
      "epoch": 5.982398986816406e-05,
      "grad_norm": 0.08911801874637604,
      "learning_rate": 3.3407366907750842e-06,
      "loss": 0.0531,
      "step": 62730
    },
    {
      "epoch": 5.982398986816406e-05,
      "model_forward_time": 1.76918363571167,
      "step": 62730
    },
    {
      "dynamics_loss_avg": 0.05379938520491123,
      "epoch": 5.982398986816406e-05,
      "step": 62730
    },
    {
      "action_loss_avg": 0.002195894956821576,
      "epoch": 5.982398986816406e-05,
      "step": 62730
    },
    {
      "epoch": 5.982398986816406e-05,
      "step": 62730,
      "training_step_time": 7.945035934448242
    },
    {
      "epoch": 5.982494354248047e-05,
      "model_forward_time": 1.712202787399292,
      "step": 62731
    },
    {
      "epoch": 5.982494354248047e-05,
      "step": 62731,
      "training_step_time": 7.880640983581543
    },
    {
      "epoch": 5.9825897216796874e-05,
      "model_forward_time": 1.7158701419830322,
      "step": 62732
    },
    {
      "epoch": 5.9825897216796874e-05,
      "step": 62732,
      "training_step_time": 7.85438084602356
    },
    {
      "epoch": 5.982685089111328e-05,
      "model_forward_time": 1.7104244232177734,
      "step": 62733
    },
    {
      "epoch": 5.982685089111328e-05,
      "step": 62733,
      "training_step_time": 7.8998777866363525
    },
    {
      "epoch": 5.982780456542969e-05,
      "model_forward_time": 1.7035942077636719,
      "step": 62734
    },
    {
      "epoch": 5.982780456542969e-05,
      "step": 62734,
      "training_step_time": 7.912318229675293
    },
    {
      "epoch": 5.982875823974609e-05,
      "model_forward_time": 1.7131593227386475,
      "step": 62735
    },
    {
      "epoch": 5.982875823974609e-05,
      "step": 62735,
      "training_step_time": 7.86478066444397
    },
    {
      "epoch": 5.98297119140625e-05,
      "model_forward_time": 1.7125120162963867,
      "step": 62736
    },
    {
      "epoch": 5.98297119140625e-05,
      "step": 62736,
      "training_step_time": 7.874457597732544
    },
    {
      "epoch": 5.983066558837891e-05,
      "model_forward_time": 1.7167611122131348,
      "step": 62737
    },
    {
      "epoch": 5.983066558837891e-05,
      "step": 62737,
      "training_step_time": 7.8461503982543945
    },
    {
      "epoch": 5.983161926269531e-05,
      "model_forward_time": 2.312082290649414,
      "step": 62738
    },
    {
      "epoch": 5.983161926269531e-05,
      "step": 62738,
      "training_step_time": 7.864075183868408
    },
    {
      "epoch": 5.9832572937011716e-05,
      "model_forward_time": 1.7411444187164307,
      "step": 62739
    },
    {
      "epoch": 5.9832572937011716e-05,
      "step": 62739,
      "training_step_time": 7.996363401412964
    },
    {
      "epoch": 5.983352661132813e-05,
      "grad_norm": 0.16631588339805603,
      "learning_rate": 3.339177011812737e-06,
      "loss": 0.0589,
      "step": 62740
    },
    {
      "epoch": 5.983352661132813e-05,
      "model_forward_time": 1.713322639465332,
      "step": 62740
    },
    {
      "dynamics_loss_avg": 0.05288461726158857,
      "epoch": 5.983352661132813e-05,
      "step": 62740
    },
    {
      "action_loss_avg": 0.001450654654763639,
      "epoch": 5.983352661132813e-05,
      "step": 62740
    },
    {
      "epoch": 5.983352661132813e-05,
      "step": 62740,
      "training_step_time": 8.91486930847168
    },
    {
      "epoch": 5.983448028564453e-05,
      "model_forward_time": 1.7093801498413086,
      "step": 62741
    },
    {
      "epoch": 5.983448028564453e-05,
      "step": 62741,
      "training_step_time": 7.938379287719727
    },
    {
      "epoch": 5.9835433959960935e-05,
      "model_forward_time": 1.7042796611785889,
      "step": 62742
    },
    {
      "epoch": 5.9835433959960935e-05,
      "step": 62742,
      "training_step_time": 7.901135444641113
    },
    {
      "epoch": 5.9836387634277346e-05,
      "model_forward_time": 1.7133910655975342,
      "step": 62743
    },
    {
      "epoch": 5.9836387634277346e-05,
      "step": 62743,
      "training_step_time": 7.9567530155181885
    },
    {
      "epoch": 5.983734130859375e-05,
      "model_forward_time": 1.7192461490631104,
      "step": 62744
    },
    {
      "epoch": 5.983734130859375e-05,
      "step": 62744,
      "training_step_time": 7.849314451217651
    },
    {
      "epoch": 5.9838294982910154e-05,
      "model_forward_time": 1.704822063446045,
      "step": 62745
    },
    {
      "epoch": 5.9838294982910154e-05,
      "step": 62745,
      "training_step_time": 9.17652416229248
    },
    {
      "epoch": 5.9839248657226565e-05,
      "model_forward_time": 1.706221580505371,
      "step": 62746
    },
    {
      "epoch": 5.9839248657226565e-05,
      "step": 62746,
      "training_step_time": 7.830659627914429
    },
    {
      "epoch": 5.984020233154297e-05,
      "model_forward_time": 2.2769129276275635,
      "step": 62747
    },
    {
      "epoch": 5.984020233154297e-05,
      "step": 62747,
      "training_step_time": 8.860138177871704
    },
    {
      "epoch": 5.984115600585937e-05,
      "model_forward_time": 1.7205889225006104,
      "step": 62748
    },
    {
      "epoch": 5.984115600585937e-05,
      "step": 62748,
      "training_step_time": 7.9686360359191895
    },
    {
      "epoch": 5.9842109680175783e-05,
      "model_forward_time": 1.7062408924102783,
      "step": 62749
    },
    {
      "epoch": 5.9842109680175783e-05,
      "step": 62749,
      "training_step_time": 7.884420156478882
    },
    {
      "epoch": 5.984306335449219e-05,
      "grad_norm": 0.14623764157295227,
      "learning_rate": 3.3376175144754936e-06,
      "loss": 0.0549,
      "step": 62750
    },
    {
      "epoch": 5.984306335449219e-05,
      "model_forward_time": 1.7210392951965332,
      "step": 62750
    },
    {
      "dynamics_loss_avg": 0.06631287317723036,
      "epoch": 5.984306335449219e-05,
      "step": 62750
    },
    {
      "action_loss_avg": 0.003610647248569876,
      "epoch": 5.984306335449219e-05,
      "step": 62750
    },
    {
      "epoch": 5.984306335449219e-05,
      "step": 62750,
      "training_step_time": 8.802604675292969
    },
    {
      "epoch": 5.984401702880859e-05,
      "model_forward_time": 1.7036218643188477,
      "step": 62751
    },
    {
      "epoch": 5.984401702880859e-05,
      "step": 62751,
      "training_step_time": 7.959851980209351
    },
    {
      "epoch": 5.9844970703125e-05,
      "model_forward_time": 1.7195513248443604,
      "step": 62752
    },
    {
      "epoch": 5.9844970703125e-05,
      "step": 62752,
      "training_step_time": 7.993263483047485
    },
    {
      "epoch": 5.9845924377441406e-05,
      "model_forward_time": 1.7058086395263672,
      "step": 62753
    },
    {
      "epoch": 5.9845924377441406e-05,
      "step": 62753,
      "training_step_time": 7.795219659805298
    },
    {
      "epoch": 5.984687805175781e-05,
      "model_forward_time": 1.7054290771484375,
      "step": 62754
    },
    {
      "epoch": 5.984687805175781e-05,
      "step": 62754,
      "training_step_time": 7.722451210021973
    },
    {
      "epoch": 5.984783172607422e-05,
      "model_forward_time": 1.7160916328430176,
      "step": 62755
    },
    {
      "epoch": 5.984783172607422e-05,
      "step": 62755,
      "training_step_time": 7.990534543991089
    },
    {
      "epoch": 5.9848785400390625e-05,
      "model_forward_time": 1.7102360725402832,
      "step": 62756
    },
    {
      "epoch": 5.9848785400390625e-05,
      "step": 62756,
      "training_step_time": 9.204519987106323
    },
    {
      "epoch": 5.984973907470703e-05,
      "model_forward_time": 1.7096571922302246,
      "step": 62757
    },
    {
      "epoch": 5.984973907470703e-05,
      "step": 62757,
      "training_step_time": 8.740137815475464
    },
    {
      "epoch": 5.985069274902344e-05,
      "model_forward_time": 1.7104477882385254,
      "step": 62758
    },
    {
      "epoch": 5.985069274902344e-05,
      "step": 62758,
      "training_step_time": 8.361421346664429
    },
    {
      "epoch": 5.9851646423339844e-05,
      "model_forward_time": 1.7167489528656006,
      "step": 62759
    },
    {
      "epoch": 5.9851646423339844e-05,
      "step": 62759,
      "training_step_time": 7.920008897781372
    },
    {
      "epoch": 5.985260009765625e-05,
      "grad_norm": 0.11408743262290955,
      "learning_rate": 3.3360581989339003e-06,
      "loss": 0.0554,
      "step": 62760
    },
    {
      "epoch": 5.985260009765625e-05,
      "model_forward_time": 1.7081124782562256,
      "step": 62760
    },
    {
      "dynamics_loss_avg": 0.07945596612989902,
      "epoch": 5.985260009765625e-05,
      "step": 62760
    },
    {
      "action_loss_avg": 0.0018605529505293817,
      "epoch": 5.985260009765625e-05,
      "step": 62760
    },
    {
      "epoch": 5.985260009765625e-05,
      "step": 62760,
      "training_step_time": 7.815075397491455
    },
    {
      "epoch": 5.985355377197266e-05,
      "model_forward_time": 1.7085797786712646,
      "step": 62761
    },
    {
      "epoch": 5.985355377197266e-05,
      "step": 62761,
      "training_step_time": 7.889068603515625
    },
    {
      "epoch": 5.985450744628906e-05,
      "model_forward_time": 1.71293306350708,
      "step": 62762
    },
    {
      "epoch": 5.985450744628906e-05,
      "step": 62762,
      "training_step_time": 8.317802906036377
    },
    {
      "epoch": 5.985546112060547e-05,
      "model_forward_time": 1.7128427028656006,
      "step": 62763
    },
    {
      "epoch": 5.985546112060547e-05,
      "step": 62763,
      "training_step_time": 7.841926097869873
    },
    {
      "epoch": 5.985641479492188e-05,
      "model_forward_time": 1.7095327377319336,
      "step": 62764
    },
    {
      "epoch": 5.985641479492188e-05,
      "step": 62764,
      "training_step_time": 8.007169246673584
    },
    {
      "epoch": 5.985736846923828e-05,
      "model_forward_time": 1.7077503204345703,
      "step": 62765
    },
    {
      "epoch": 5.985736846923828e-05,
      "step": 62765,
      "training_step_time": 7.875908136367798
    },
    {
      "epoch": 5.9858322143554686e-05,
      "model_forward_time": 2.2674977779388428,
      "step": 62766
    },
    {
      "epoch": 5.9858322143554686e-05,
      "step": 62766,
      "training_step_time": 9.014925003051758
    },
    {
      "epoch": 5.98592758178711e-05,
      "model_forward_time": 1.771596908569336,
      "step": 62767
    },
    {
      "epoch": 5.98592758178711e-05,
      "step": 62767,
      "training_step_time": 7.903841495513916
    },
    {
      "epoch": 5.98602294921875e-05,
      "model_forward_time": 1.7047083377838135,
      "step": 62768
    },
    {
      "epoch": 5.98602294921875e-05,
      "step": 62768,
      "training_step_time": 9.49048137664795
    },
    {
      "epoch": 5.9861183166503905e-05,
      "model_forward_time": 1.7147443294525146,
      "step": 62769
    },
    {
      "epoch": 5.9861183166503905e-05,
      "step": 62769,
      "training_step_time": 7.901110887527466
    },
    {
      "epoch": 5.9862136840820315e-05,
      "grad_norm": 0.10468597710132599,
      "learning_rate": 3.3344990653584808e-06,
      "loss": 0.0542,
      "step": 62770
    },
    {
      "epoch": 5.9862136840820315e-05,
      "model_forward_time": 1.7241060733795166,
      "step": 62770
    },
    {
      "dynamics_loss_avg": 0.04746521823108196,
      "epoch": 5.9862136840820315e-05,
      "step": 62770
    },
    {
      "action_loss_avg": 0.0014043075090739876,
      "epoch": 5.9862136840820315e-05,
      "step": 62770
    },
    {
      "epoch": 5.9862136840820315e-05,
      "step": 62770,
      "training_step_time": 7.96860408782959
    },
    {
      "epoch": 5.986309051513672e-05,
      "model_forward_time": 1.7062036991119385,
      "step": 62771
    },
    {
      "epoch": 5.986309051513672e-05,
      "step": 62771,
      "training_step_time": 7.952667236328125
    },
    {
      "epoch": 5.9864044189453124e-05,
      "model_forward_time": 1.7069573402404785,
      "step": 62772
    },
    {
      "epoch": 5.9864044189453124e-05,
      "step": 62772,
      "training_step_time": 7.965616703033447
    },
    {
      "epoch": 5.9864997863769534e-05,
      "model_forward_time": 1.3458805084228516,
      "step": 62773
    },
    {
      "epoch": 5.9864997863769534e-05,
      "step": 62773,
      "training_step_time": 8.92976689338684
    },
    {
      "epoch": 5.986595153808594e-05,
      "model_forward_time": 1.7102344036102295,
      "step": 62774
    },
    {
      "epoch": 5.986595153808594e-05,
      "step": 62774,
      "training_step_time": 7.926776170730591
    },
    {
      "epoch": 5.986690521240234e-05,
      "model_forward_time": 2.278573751449585,
      "step": 62775
    },
    {
      "epoch": 5.986690521240234e-05,
      "step": 62775,
      "training_step_time": 9.38594126701355
    },
    {
      "epoch": 5.986785888671875e-05,
      "model_forward_time": 1.4287583827972412,
      "step": 62776
    },
    {
      "epoch": 5.986785888671875e-05,
      "step": 62776,
      "training_step_time": 9.008403062820435
    },
    {
      "epoch": 5.986881256103516e-05,
      "model_forward_time": 1.699683666229248,
      "step": 62777
    },
    {
      "epoch": 5.986881256103516e-05,
      "step": 62777,
      "training_step_time": 8.408200025558472
    },
    {
      "epoch": 5.986976623535156e-05,
      "model_forward_time": 1.710045337677002,
      "step": 62778
    },
    {
      "epoch": 5.986976623535156e-05,
      "step": 62778,
      "training_step_time": 9.572345972061157
    },
    {
      "epoch": 5.987071990966797e-05,
      "model_forward_time": 1.707150936126709,
      "step": 62779
    },
    {
      "epoch": 5.987071990966797e-05,
      "step": 62779,
      "training_step_time": 7.956955194473267
    },
    {
      "epoch": 5.9871673583984376e-05,
      "grad_norm": 0.06953033804893494,
      "learning_rate": 3.3329401139197393e-06,
      "loss": 0.0563,
      "step": 62780
    },
    {
      "epoch": 5.9871673583984376e-05,
      "model_forward_time": 1.7177960872650146,
      "step": 62780
    },
    {
      "dynamics_loss_avg": 0.05557030476629734,
      "epoch": 5.9871673583984376e-05,
      "step": 62780
    },
    {
      "action_loss_avg": 0.001149180711945519,
      "epoch": 5.9871673583984376e-05,
      "step": 62780
    },
    {
      "epoch": 5.9871673583984376e-05,
      "step": 62780,
      "training_step_time": 7.943199872970581
    },
    {
      "epoch": 5.987262725830078e-05,
      "model_forward_time": 1.3411214351654053,
      "step": 62781
    },
    {
      "epoch": 5.987262725830078e-05,
      "step": 62781,
      "training_step_time": 7.960374116897583
    },
    {
      "epoch": 5.9873580932617184e-05,
      "model_forward_time": 1.7048225402832031,
      "step": 62782
    },
    {
      "epoch": 5.9873580932617184e-05,
      "step": 62782,
      "training_step_time": 8.541479349136353
    },
    {
      "epoch": 5.9874534606933595e-05,
      "model_forward_time": 1.7097065448760986,
      "step": 62783
    },
    {
      "epoch": 5.9874534606933595e-05,
      "step": 62783,
      "training_step_time": 9.657385349273682
    },
    {
      "epoch": 5.987548828125e-05,
      "model_forward_time": 2.2472875118255615,
      "step": 62784
    },
    {
      "epoch": 5.987548828125e-05,
      "step": 62784,
      "training_step_time": 10.039070129394531
    },
    {
      "epoch": 5.98764419555664e-05,
      "model_forward_time": 1.7520246505737305,
      "step": 62785
    },
    {
      "epoch": 5.98764419555664e-05,
      "step": 62785,
      "training_step_time": 11.147035837173462
    },
    {
      "epoch": 5.9877395629882814e-05,
      "model_forward_time": 1.6920857429504395,
      "step": 62786
    },
    {
      "epoch": 5.9877395629882814e-05,
      "step": 62786,
      "training_step_time": 9.114765882492065
    },
    {
      "epoch": 5.987834930419922e-05,
      "model_forward_time": 1.6967642307281494,
      "step": 62787
    },
    {
      "epoch": 5.987834930419922e-05,
      "step": 62787,
      "training_step_time": 8.892822504043579
    },
    {
      "epoch": 5.987930297851562e-05,
      "model_forward_time": 1.6865260601043701,
      "step": 62788
    },
    {
      "epoch": 5.987930297851562e-05,
      "step": 62788,
      "training_step_time": 7.978665590286255
    },
    {
      "epoch": 5.988025665283203e-05,
      "model_forward_time": 1.6936330795288086,
      "step": 62789
    },
    {
      "epoch": 5.988025665283203e-05,
      "step": 62789,
      "training_step_time": 7.9349143505096436
    },
    {
      "epoch": 5.988121032714844e-05,
      "grad_norm": 0.08253342658281326,
      "learning_rate": 3.331381344788161e-06,
      "loss": 0.0572,
      "step": 62790
    },
    {
      "epoch": 5.988121032714844e-05,
      "model_forward_time": 1.6890134811401367,
      "step": 62790
    },
    {
      "dynamics_loss_avg": 0.04686333229765296,
      "epoch": 5.988121032714844e-05,
      "step": 62790
    },
    {
      "action_loss_avg": 0.0012597710738191381,
      "epoch": 5.988121032714844e-05,
      "step": 62790
    },
    {
      "epoch": 5.988121032714844e-05,
      "step": 62790,
      "training_step_time": 7.92863917350769
    },
    {
      "epoch": 5.988216400146484e-05,
      "model_forward_time": 1.7335569858551025,
      "step": 62791
    },
    {
      "epoch": 5.988216400146484e-05,
      "step": 62791,
      "training_step_time": 8.930086612701416
    },
    {
      "epoch": 5.988311767578125e-05,
      "model_forward_time": 1.6892173290252686,
      "step": 62792
    },
    {
      "epoch": 5.988311767578125e-05,
      "step": 62792,
      "training_step_time": 7.884600639343262
    },
    {
      "epoch": 5.9884071350097656e-05,
      "model_forward_time": 1.6914863586425781,
      "step": 62793
    },
    {
      "epoch": 5.9884071350097656e-05,
      "step": 62793,
      "training_step_time": 8.411652088165283
    },
    {
      "epoch": 5.988502502441406e-05,
      "model_forward_time": 1.7292797565460205,
      "step": 62794
    },
    {
      "epoch": 5.988502502441406e-05,
      "step": 62794,
      "training_step_time": 9.055221557617188
    },
    {
      "epoch": 5.988597869873047e-05,
      "model_forward_time": 1.6917922496795654,
      "step": 62795
    },
    {
      "epoch": 5.988597869873047e-05,
      "step": 62795,
      "training_step_time": 7.876384258270264
    },
    {
      "epoch": 5.9886932373046874e-05,
      "model_forward_time": 1.6974248886108398,
      "step": 62796
    },
    {
      "epoch": 5.9886932373046874e-05,
      "step": 62796,
      "training_step_time": 8.453547239303589
    },
    {
      "epoch": 5.988788604736328e-05,
      "model_forward_time": 1.6920139789581299,
      "step": 62797
    },
    {
      "epoch": 5.988788604736328e-05,
      "step": 62797,
      "training_step_time": 8.765278577804565
    },
    {
      "epoch": 5.988883972167969e-05,
      "model_forward_time": 1.6961603164672852,
      "step": 62798
    },
    {
      "epoch": 5.988883972167969e-05,
      "step": 62798,
      "training_step_time": 8.81104826927185
    },
    {
      "epoch": 5.988979339599609e-05,
      "model_forward_time": 1.7053132057189941,
      "step": 62799
    },
    {
      "epoch": 5.988979339599609e-05,
      "step": 62799,
      "training_step_time": 7.872626543045044
    },
    {
      "epoch": 5.98907470703125e-05,
      "grad_norm": 0.09325921535491943,
      "learning_rate": 3.3298227581342078e-06,
      "loss": 0.0543,
      "step": 62800
    },
    {
      "epoch": 5.98907470703125e-05,
      "model_forward_time": 1.706482172012329,
      "step": 62800
    },
    {
      "dynamics_loss_avg": 0.06925974488258362,
      "epoch": 5.98907470703125e-05,
      "step": 62800
    },
    {
      "action_loss_avg": 0.0018245270010083913,
      "epoch": 5.98907470703125e-05,
      "step": 62800
    },
    {
      "epoch": 5.98907470703125e-05,
      "step": 62800,
      "training_step_time": 7.906750679016113
    },
    {
      "epoch": 5.989170074462891e-05,
      "model_forward_time": 1.7138376235961914,
      "step": 62801
    },
    {
      "epoch": 5.989170074462891e-05,
      "step": 62801,
      "training_step_time": 9.908690929412842
    },
    {
      "epoch": 5.989265441894531e-05,
      "model_forward_time": 1.708400011062622,
      "step": 62802
    },
    {
      "epoch": 5.989265441894531e-05,
      "step": 62802,
      "training_step_time": 7.913224935531616
    },
    {
      "epoch": 5.9893608093261716e-05,
      "model_forward_time": 2.299459218978882,
      "step": 62803
    },
    {
      "epoch": 5.9893608093261716e-05,
      "step": 62803,
      "training_step_time": 8.958629369735718
    },
    {
      "epoch": 5.989456176757813e-05,
      "model_forward_time": 1.7067441940307617,
      "step": 62804
    },
    {
      "epoch": 5.989456176757813e-05,
      "step": 62804,
      "training_step_time": 9.57023310661316
    },
    {
      "epoch": 5.989551544189453e-05,
      "model_forward_time": 1.715484619140625,
      "step": 62805
    },
    {
      "epoch": 5.989551544189453e-05,
      "step": 62805,
      "training_step_time": 7.981005668640137
    },
    {
      "epoch": 5.9896469116210935e-05,
      "model_forward_time": 1.7018494606018066,
      "step": 62806
    },
    {
      "epoch": 5.9896469116210935e-05,
      "step": 62806,
      "training_step_time": 7.91135573387146
    },
    {
      "epoch": 5.9897422790527346e-05,
      "model_forward_time": 1.7056119441986084,
      "step": 62807
    },
    {
      "epoch": 5.9897422790527346e-05,
      "step": 62807,
      "training_step_time": 7.959202766418457
    },
    {
      "epoch": 5.989837646484375e-05,
      "model_forward_time": 1.7028446197509766,
      "step": 62808
    },
    {
      "epoch": 5.989837646484375e-05,
      "step": 62808,
      "training_step_time": 7.87421727180481
    },
    {
      "epoch": 5.9899330139160154e-05,
      "model_forward_time": 1.708665370941162,
      "step": 62809
    },
    {
      "epoch": 5.9899330139160154e-05,
      "step": 62809,
      "training_step_time": 7.834360122680664
    },
    {
      "epoch": 5.9900283813476565e-05,
      "grad_norm": 0.13064594566822052,
      "learning_rate": 3.328264354128329e-06,
      "loss": 0.0534,
      "step": 62810
    },
    {
      "epoch": 5.9900283813476565e-05,
      "model_forward_time": 1.7051730155944824,
      "step": 62810
    },
    {
      "dynamics_loss_avg": 0.041870835423469546,
      "epoch": 5.9900283813476565e-05,
      "step": 62810
    },
    {
      "action_loss_avg": 0.002354084080434404,
      "epoch": 5.9900283813476565e-05,
      "step": 62810
    },
    {
      "epoch": 5.9900283813476565e-05,
      "step": 62810,
      "training_step_time": 8.742958545684814
    },
    {
      "epoch": 5.990123748779297e-05,
      "model_forward_time": 1.7048277854919434,
      "step": 62811
    },
    {
      "epoch": 5.990123748779297e-05,
      "step": 62811,
      "training_step_time": 7.8601789474487305
    },
    {
      "epoch": 5.990219116210937e-05,
      "model_forward_time": 2.2855753898620605,
      "step": 62812
    },
    {
      "epoch": 5.990219116210937e-05,
      "step": 62812,
      "training_step_time": 7.946398973464966
    },
    {
      "epoch": 5.9903144836425784e-05,
      "model_forward_time": 1.7014963626861572,
      "step": 62813
    },
    {
      "epoch": 5.9903144836425784e-05,
      "step": 62813,
      "training_step_time": 7.818318605422974
    },
    {
      "epoch": 5.990409851074219e-05,
      "model_forward_time": 1.7042796611785889,
      "step": 62814
    },
    {
      "epoch": 5.990409851074219e-05,
      "step": 62814,
      "training_step_time": 7.876814365386963
    },
    {
      "epoch": 5.990505218505859e-05,
      "model_forward_time": 1.7024898529052734,
      "step": 62815
    },
    {
      "epoch": 5.990505218505859e-05,
      "step": 62815,
      "training_step_time": 9.41911005973816
    },
    {
      "epoch": 5.9906005859375e-05,
      "model_forward_time": 1.7340779304504395,
      "step": 62816
    },
    {
      "epoch": 5.9906005859375e-05,
      "step": 62816,
      "training_step_time": 7.944669961929321
    },
    {
      "epoch": 5.9906959533691406e-05,
      "model_forward_time": 1.7028720378875732,
      "step": 62817
    },
    {
      "epoch": 5.9906959533691406e-05,
      "step": 62817,
      "training_step_time": 7.958574056625366
    },
    {
      "epoch": 5.990791320800781e-05,
      "model_forward_time": 1.7013838291168213,
      "step": 62818
    },
    {
      "epoch": 5.990791320800781e-05,
      "step": 62818,
      "training_step_time": 7.862808465957642
    },
    {
      "epoch": 5.990886688232422e-05,
      "model_forward_time": 1.3486919403076172,
      "step": 62819
    },
    {
      "epoch": 5.990886688232422e-05,
      "step": 62819,
      "training_step_time": 8.034205675125122
    },
    {
      "epoch": 5.9909820556640625e-05,
      "grad_norm": 0.09154988080263138,
      "learning_rate": 3.3267061329409465e-06,
      "loss": 0.0559,
      "step": 62820
    },
    {
      "epoch": 5.9909820556640625e-05,
      "model_forward_time": 1.7054808139801025,
      "step": 62820
    },
    {
      "dynamics_loss_avg": 0.0622321929782629,
      "epoch": 5.9909820556640625e-05,
      "step": 62820
    },
    {
      "action_loss_avg": 0.0015405641868710517,
      "epoch": 5.9909820556640625e-05,
      "step": 62820
    },
    {
      "epoch": 5.9909820556640625e-05,
      "step": 62820,
      "training_step_time": 8.424625873565674
    },
    {
      "epoch": 5.991077423095703e-05,
      "model_forward_time": 2.271588087081909,
      "step": 62821
    },
    {
      "epoch": 5.991077423095703e-05,
      "step": 62821,
      "training_step_time": 8.481858015060425
    },
    {
      "epoch": 5.991172790527344e-05,
      "model_forward_time": 1.760038137435913,
      "step": 62822
    },
    {
      "epoch": 5.991172790527344e-05,
      "step": 62822,
      "training_step_time": 9.426780700683594
    },
    {
      "epoch": 5.9912681579589844e-05,
      "model_forward_time": 1.706815242767334,
      "step": 62823
    },
    {
      "epoch": 5.9912681579589844e-05,
      "step": 62823,
      "training_step_time": 7.829903602600098
    },
    {
      "epoch": 5.991363525390625e-05,
      "model_forward_time": 1.6970703601837158,
      "step": 62824
    },
    {
      "epoch": 5.991363525390625e-05,
      "step": 62824,
      "training_step_time": 7.830785274505615
    },
    {
      "epoch": 5.991458892822266e-05,
      "model_forward_time": 1.7057442665100098,
      "step": 62825
    },
    {
      "epoch": 5.991458892822266e-05,
      "step": 62825,
      "training_step_time": 7.968422889709473
    },
    {
      "epoch": 5.991554260253906e-05,
      "model_forward_time": 1.7083523273468018,
      "step": 62826
    },
    {
      "epoch": 5.991554260253906e-05,
      "step": 62826,
      "training_step_time": 7.903167486190796
    },
    {
      "epoch": 5.991649627685547e-05,
      "model_forward_time": 1.7144129276275635,
      "step": 62827
    },
    {
      "epoch": 5.991649627685547e-05,
      "step": 62827,
      "training_step_time": 7.987963914871216
    },
    {
      "epoch": 5.991744995117188e-05,
      "model_forward_time": 1.7048046588897705,
      "step": 62828
    },
    {
      "epoch": 5.991744995117188e-05,
      "step": 62828,
      "training_step_time": 8.033239126205444
    },
    {
      "epoch": 5.991840362548828e-05,
      "model_forward_time": 1.7036521434783936,
      "step": 62829
    },
    {
      "epoch": 5.991840362548828e-05,
      "step": 62829,
      "training_step_time": 9.608277797698975
    },
    {
      "epoch": 5.9919357299804686e-05,
      "grad_norm": 0.1004529520869255,
      "learning_rate": 3.325148094742466e-06,
      "loss": 0.0555,
      "step": 62830
    },
    {
      "epoch": 5.9919357299804686e-05,
      "model_forward_time": 2.2632548809051514,
      "step": 62830
    },
    {
      "dynamics_loss_avg": 0.04052440859377384,
      "epoch": 5.9919357299804686e-05,
      "step": 62830
    },
    {
      "action_loss_avg": 0.0011464670635177754,
      "epoch": 5.9919357299804686e-05,
      "step": 62830
    },
    {
      "epoch": 5.9919357299804686e-05,
      "step": 62830,
      "training_step_time": 7.916194438934326
    },
    {
      "epoch": 5.99203109741211e-05,
      "model_forward_time": 1.7579472064971924,
      "step": 62831
    },
    {
      "epoch": 5.99203109741211e-05,
      "step": 62831,
      "training_step_time": 7.938432693481445
    },
    {
      "epoch": 5.99212646484375e-05,
      "model_forward_time": 1.702657699584961,
      "step": 62832
    },
    {
      "epoch": 5.99212646484375e-05,
      "step": 62832,
      "training_step_time": 8.905118465423584
    },
    {
      "epoch": 5.9922218322753905e-05,
      "model_forward_time": 1.7073752880096436,
      "step": 62833
    },
    {
      "epoch": 5.9922218322753905e-05,
      "step": 62833,
      "training_step_time": 8.331402778625488
    },
    {
      "epoch": 5.9923171997070316e-05,
      "model_forward_time": 1.7036449909210205,
      "step": 62834
    },
    {
      "epoch": 5.9923171997070316e-05,
      "step": 62834,
      "training_step_time": 7.852751731872559
    },
    {
      "epoch": 5.992412567138672e-05,
      "model_forward_time": 1.702761173248291,
      "step": 62835
    },
    {
      "epoch": 5.992412567138672e-05,
      "step": 62835,
      "training_step_time": 8.841536521911621
    },
    {
      "epoch": 5.9925079345703124e-05,
      "model_forward_time": 1.7060341835021973,
      "step": 62836
    },
    {
      "epoch": 5.9925079345703124e-05,
      "step": 62836,
      "training_step_time": 7.698652267456055
    },
    {
      "epoch": 5.9926033020019534e-05,
      "model_forward_time": 1.7046709060668945,
      "step": 62837
    },
    {
      "epoch": 5.9926033020019534e-05,
      "step": 62837,
      "training_step_time": 8.048152208328247
    },
    {
      "epoch": 5.992698669433594e-05,
      "model_forward_time": 1.7048752307891846,
      "step": 62838
    },
    {
      "epoch": 5.992698669433594e-05,
      "step": 62838,
      "training_step_time": 7.815765380859375
    },
    {
      "epoch": 5.992794036865234e-05,
      "model_forward_time": 2.273735523223877,
      "step": 62839
    },
    {
      "epoch": 5.992794036865234e-05,
      "step": 62839,
      "training_step_time": 9.792965650558472
    },
    {
      "epoch": 5.992889404296875e-05,
      "grad_norm": 0.081194207072258,
      "learning_rate": 3.3235902397032716e-06,
      "loss": 0.0557,
      "step": 62840
    },
    {
      "epoch": 5.992889404296875e-05,
      "model_forward_time": 1.7411634922027588,
      "step": 62840
    },
    {
      "dynamics_loss_avg": 0.0670343678444624,
      "epoch": 5.992889404296875e-05,
      "step": 62840
    },
    {
      "action_loss_avg": 0.00166267299791798,
      "epoch": 5.992889404296875e-05,
      "step": 62840
    },
    {
      "epoch": 5.992889404296875e-05,
      "step": 62840,
      "training_step_time": 7.944615602493286
    },
    {
      "epoch": 5.992984771728516e-05,
      "model_forward_time": 1.7006864547729492,
      "step": 62841
    },
    {
      "epoch": 5.992984771728516e-05,
      "step": 62841,
      "training_step_time": 9.388478517532349
    },
    {
      "epoch": 5.993080139160156e-05,
      "model_forward_time": 1.7075917720794678,
      "step": 62842
    },
    {
      "epoch": 5.993080139160156e-05,
      "step": 62842,
      "training_step_time": 7.919506072998047
    },
    {
      "epoch": 5.993175506591797e-05,
      "model_forward_time": 1.7043826580047607,
      "step": 62843
    },
    {
      "epoch": 5.993175506591797e-05,
      "step": 62843,
      "training_step_time": 7.816884279251099
    },
    {
      "epoch": 5.9932708740234376e-05,
      "model_forward_time": 1.7113418579101562,
      "step": 62844
    },
    {
      "epoch": 5.9932708740234376e-05,
      "step": 62844,
      "training_step_time": 8.112128257751465
    },
    {
      "epoch": 5.993366241455078e-05,
      "model_forward_time": 1.710883378982544,
      "step": 62845
    },
    {
      "epoch": 5.993366241455078e-05,
      "step": 62845,
      "training_step_time": 8.457963705062866
    },
    {
      "epoch": 5.9934616088867184e-05,
      "model_forward_time": 1.7184324264526367,
      "step": 62846
    },
    {
      "epoch": 5.9934616088867184e-05,
      "step": 62846,
      "training_step_time": 7.643934488296509
    },
    {
      "epoch": 5.9935569763183595e-05,
      "model_forward_time": 1.7182486057281494,
      "step": 62847
    },
    {
      "epoch": 5.9935569763183595e-05,
      "step": 62847,
      "training_step_time": 8.012035846710205
    },
    {
      "epoch": 5.99365234375e-05,
      "model_forward_time": 2.2710204124450684,
      "step": 62848
    },
    {
      "epoch": 5.99365234375e-05,
      "step": 62848,
      "training_step_time": 8.326266527175903
    },
    {
      "epoch": 5.99374771118164e-05,
      "model_forward_time": 1.7282359600067139,
      "step": 62849
    },
    {
      "epoch": 5.99374771118164e-05,
      "step": 62849,
      "training_step_time": 9.863920211791992
    },
    {
      "epoch": 5.9938430786132814e-05,
      "grad_norm": 0.10922718793153763,
      "learning_rate": 3.3220325679937283e-06,
      "loss": 0.056,
      "step": 62850
    },
    {
      "epoch": 5.9938430786132814e-05,
      "model_forward_time": 1.7114894390106201,
      "step": 62850
    },
    {
      "dynamics_loss_avg": 0.059478705190122126,
      "epoch": 5.9938430786132814e-05,
      "step": 62850
    },
    {
      "action_loss_avg": 0.0015681863704230636,
      "epoch": 5.9938430786132814e-05,
      "step": 62850
    },
    {
      "epoch": 5.9938430786132814e-05,
      "step": 62850,
      "training_step_time": 7.8349769115448
    },
    {
      "epoch": 5.993938446044922e-05,
      "model_forward_time": 1.71390962600708,
      "step": 62851
    },
    {
      "epoch": 5.993938446044922e-05,
      "step": 62851,
      "training_step_time": 8.94370412826538
    },
    {
      "epoch": 5.994033813476562e-05,
      "model_forward_time": 1.7126100063323975,
      "step": 62852
    },
    {
      "epoch": 5.994033813476562e-05,
      "step": 62852,
      "training_step_time": 8.880319356918335
    },
    {
      "epoch": 5.994129180908203e-05,
      "model_forward_time": 1.7214345932006836,
      "step": 62853
    },
    {
      "epoch": 5.994129180908203e-05,
      "step": 62853,
      "training_step_time": 7.916946649551392
    },
    {
      "epoch": 5.994224548339844e-05,
      "model_forward_time": 1.7111170291900635,
      "step": 62854
    },
    {
      "epoch": 5.994224548339844e-05,
      "step": 62854,
      "training_step_time": 7.929327487945557
    },
    {
      "epoch": 5.994319915771484e-05,
      "model_forward_time": 1.7169992923736572,
      "step": 62855
    },
    {
      "epoch": 5.994319915771484e-05,
      "step": 62855,
      "training_step_time": 7.938178062438965
    },
    {
      "epoch": 5.994415283203125e-05,
      "model_forward_time": 1.7194106578826904,
      "step": 62856
    },
    {
      "epoch": 5.994415283203125e-05,
      "step": 62856,
      "training_step_time": 7.917656183242798
    },
    {
      "epoch": 5.9945106506347656e-05,
      "model_forward_time": 1.714799404144287,
      "step": 62857
    },
    {
      "epoch": 5.9945106506347656e-05,
      "step": 62857,
      "training_step_time": 7.743150949478149
    },
    {
      "epoch": 5.994606018066406e-05,
      "model_forward_time": 2.268249034881592,
      "step": 62858
    },
    {
      "epoch": 5.994606018066406e-05,
      "step": 62858,
      "training_step_time": 8.83640456199646
    },
    {
      "epoch": 5.994701385498047e-05,
      "model_forward_time": 1.777198314666748,
      "step": 62859
    },
    {
      "epoch": 5.994701385498047e-05,
      "step": 62859,
      "training_step_time": 9.002379417419434
    },
    {
      "epoch": 5.9947967529296874e-05,
      "grad_norm": 0.09795322269201279,
      "learning_rate": 3.3204750797841777e-06,
      "loss": 0.0547,
      "step": 62860
    },
    {
      "epoch": 5.9947967529296874e-05,
      "model_forward_time": 1.701512098312378,
      "step": 62860
    },
    {
      "dynamics_loss_avg": 0.05623653708025813,
      "epoch": 5.9947967529296874e-05,
      "step": 62860
    },
    {
      "action_loss_avg": 0.001631335448473692,
      "epoch": 5.9947967529296874e-05,
      "step": 62860
    },
    {
      "epoch": 5.9947967529296874e-05,
      "step": 62860,
      "training_step_time": 7.8220908641815186
    },
    {
      "epoch": 5.994892120361328e-05,
      "model_forward_time": 1.7265141010284424,
      "step": 62861
    },
    {
      "epoch": 5.994892120361328e-05,
      "step": 62861,
      "training_step_time": 8.587203025817871
    },
    {
      "epoch": 5.994987487792969e-05,
      "model_forward_time": 1.7062606811523438,
      "step": 62862
    },
    {
      "epoch": 5.994987487792969e-05,
      "step": 62862,
      "training_step_time": 7.816234827041626
    },
    {
      "epoch": 5.995082855224609e-05,
      "model_forward_time": 1.7098004817962646,
      "step": 62863
    },
    {
      "epoch": 5.995082855224609e-05,
      "step": 62863,
      "training_step_time": 7.9147984981536865
    },
    {
      "epoch": 5.99517822265625e-05,
      "model_forward_time": 1.7016897201538086,
      "step": 62864
    },
    {
      "epoch": 5.99517822265625e-05,
      "step": 62864,
      "training_step_time": 9.572532892227173
    },
    {
      "epoch": 5.995273590087891e-05,
      "model_forward_time": 1.7093684673309326,
      "step": 62865
    },
    {
      "epoch": 5.995273590087891e-05,
      "step": 62865,
      "training_step_time": 8.836910963058472
    },
    {
      "epoch": 5.995368957519531e-05,
      "model_forward_time": 1.7035191059112549,
      "step": 62866
    },
    {
      "epoch": 5.995368957519531e-05,
      "step": 62866,
      "training_step_time": 7.830326080322266
    },
    {
      "epoch": 5.9954643249511716e-05,
      "model_forward_time": 2.267656087875366,
      "step": 62867
    },
    {
      "epoch": 5.9954643249511716e-05,
      "step": 62867,
      "training_step_time": 8.012421607971191
    },
    {
      "epoch": 5.995559692382813e-05,
      "model_forward_time": 1.7675292491912842,
      "step": 62868
    },
    {
      "epoch": 5.995559692382813e-05,
      "step": 62868,
      "training_step_time": 8.00528073310852
    },
    {
      "epoch": 5.995655059814453e-05,
      "model_forward_time": 1.701667308807373,
      "step": 62869
    },
    {
      "epoch": 5.995655059814453e-05,
      "step": 62869,
      "training_step_time": 7.8358495235443115
    },
    {
      "epoch": 5.9957504272460935e-05,
      "grad_norm": 0.08901068568229675,
      "learning_rate": 3.3189177752449504e-06,
      "loss": 0.0541,
      "step": 62870
    },
    {
      "epoch": 5.9957504272460935e-05,
      "model_forward_time": 1.7095074653625488,
      "step": 62870
    },
    {
      "dynamics_loss_avg": 0.059522542357444766,
      "epoch": 5.9957504272460935e-05,
      "step": 62870
    },
    {
      "action_loss_avg": 0.002442580187926069,
      "epoch": 5.9957504272460935e-05,
      "step": 62870
    },
    {
      "epoch": 5.9957504272460935e-05,
      "step": 62870,
      "training_step_time": 7.865004062652588
    },
    {
      "epoch": 5.9958457946777346e-05,
      "model_forward_time": 1.7068593502044678,
      "step": 62871
    },
    {
      "epoch": 5.9958457946777346e-05,
      "step": 62871,
      "training_step_time": 8.52372145652771
    },
    {
      "epoch": 5.995941162109375e-05,
      "model_forward_time": 1.7048020362854004,
      "step": 62872
    },
    {
      "epoch": 5.995941162109375e-05,
      "step": 62872,
      "training_step_time": 7.959717273712158
    },
    {
      "epoch": 5.9960365295410154e-05,
      "model_forward_time": 1.7132365703582764,
      "step": 62873
    },
    {
      "epoch": 5.9960365295410154e-05,
      "step": 62873,
      "training_step_time": 7.880178451538086
    },
    {
      "epoch": 5.9961318969726565e-05,
      "model_forward_time": 1.7052888870239258,
      "step": 62874
    },
    {
      "epoch": 5.9961318969726565e-05,
      "step": 62874,
      "training_step_time": 7.921042203903198
    },
    {
      "epoch": 5.996227264404297e-05,
      "model_forward_time": 1.6976492404937744,
      "step": 62875
    },
    {
      "epoch": 5.996227264404297e-05,
      "step": 62875,
      "training_step_time": 8.62269377708435
    },
    {
      "epoch": 5.996322631835937e-05,
      "model_forward_time": 2.2363314628601074,
      "step": 62876
    },
    {
      "epoch": 5.996322631835937e-05,
      "step": 62876,
      "training_step_time": 8.906708002090454
    },
    {
      "epoch": 5.9964179992675784e-05,
      "model_forward_time": 1.7490994930267334,
      "step": 62877
    },
    {
      "epoch": 5.9964179992675784e-05,
      "step": 62877,
      "training_step_time": 8.572109937667847
    },
    {
      "epoch": 5.996513366699219e-05,
      "model_forward_time": 1.6978731155395508,
      "step": 62878
    },
    {
      "epoch": 5.996513366699219e-05,
      "step": 62878,
      "training_step_time": 9.640937089920044
    },
    {
      "epoch": 5.996608734130859e-05,
      "model_forward_time": 1.7042739391326904,
      "step": 62879
    },
    {
      "epoch": 5.996608734130859e-05,
      "step": 62879,
      "training_step_time": 7.871158599853516
    },
    {
      "epoch": 5.9967041015625e-05,
      "grad_norm": 0.10333611816167831,
      "learning_rate": 3.3173606545463473e-06,
      "loss": 0.0568,
      "step": 62880
    },
    {
      "epoch": 5.9967041015625e-05,
      "model_forward_time": 1.6898419857025146,
      "step": 62880
    },
    {
      "dynamics_loss_avg": 0.04586432240903378,
      "epoch": 5.9967041015625e-05,
      "step": 62880
    },
    {
      "action_loss_avg": 0.0018064890609821305,
      "epoch": 5.9967041015625e-05,
      "step": 62880
    },
    {
      "epoch": 5.9967041015625e-05,
      "step": 62880,
      "training_step_time": 7.694577693939209
    },
    {
      "epoch": 5.9967994689941406e-05,
      "model_forward_time": 1.6937012672424316,
      "step": 62881
    },
    {
      "epoch": 5.9967994689941406e-05,
      "step": 62881,
      "training_step_time": 7.921682834625244
    },
    {
      "epoch": 5.996894836425781e-05,
      "model_forward_time": 1.692429542541504,
      "step": 62882
    },
    {
      "epoch": 5.996894836425781e-05,
      "step": 62882,
      "training_step_time": 8.25057077407837
    },
    {
      "epoch": 5.996990203857422e-05,
      "model_forward_time": 1.6922922134399414,
      "step": 62883
    },
    {
      "epoch": 5.996990203857422e-05,
      "step": 62883,
      "training_step_time": 7.957090616226196
    },
    {
      "epoch": 5.9970855712890625e-05,
      "model_forward_time": 1.7028489112854004,
      "step": 62884
    },
    {
      "epoch": 5.9970855712890625e-05,
      "step": 62884,
      "training_step_time": 8.331409931182861
    },
    {
      "epoch": 5.997180938720703e-05,
      "model_forward_time": 2.2507145404815674,
      "step": 62885
    },
    {
      "epoch": 5.997180938720703e-05,
      "step": 62885,
      "training_step_time": 8.8463716506958
    },
    {
      "epoch": 5.997276306152344e-05,
      "model_forward_time": 1.737112045288086,
      "step": 62886
    },
    {
      "epoch": 5.997276306152344e-05,
      "step": 62886,
      "training_step_time": 8.48945164680481
    },
    {
      "epoch": 5.9973716735839844e-05,
      "model_forward_time": 1.7042007446289062,
      "step": 62887
    },
    {
      "epoch": 5.9973716735839844e-05,
      "step": 62887,
      "training_step_time": 9.08896517753601
    },
    {
      "epoch": 5.997467041015625e-05,
      "model_forward_time": 1.687286615371704,
      "step": 62888
    },
    {
      "epoch": 5.997467041015625e-05,
      "step": 62888,
      "training_step_time": 7.834820747375488
    },
    {
      "epoch": 5.997562408447266e-05,
      "model_forward_time": 1.6855084896087646,
      "step": 62889
    },
    {
      "epoch": 5.997562408447266e-05,
      "step": 62889,
      "training_step_time": 8.705269813537598
    },
    {
      "epoch": 5.997657775878906e-05,
      "grad_norm": 0.11769858747720718,
      "learning_rate": 3.3158037178586533e-06,
      "loss": 0.0564,
      "step": 62890
    },
    {
      "epoch": 5.997657775878906e-05,
      "model_forward_time": 1.6957030296325684,
      "step": 62890
    },
    {
      "dynamics_loss_avg": 0.06535814898088574,
      "epoch": 5.997657775878906e-05,
      "step": 62890
    },
    {
      "action_loss_avg": 0.001224618601554539,
      "epoch": 5.997657775878906e-05,
      "step": 62890
    },
    {
      "epoch": 5.997657775878906e-05,
      "step": 62890,
      "training_step_time": 7.836285352706909
    },
    {
      "epoch": 5.997753143310547e-05,
      "model_forward_time": 1.7077374458312988,
      "step": 62891
    },
    {
      "epoch": 5.997753143310547e-05,
      "step": 62891,
      "training_step_time": 7.702068567276001
    },
    {
      "epoch": 5.997848510742188e-05,
      "model_forward_time": 1.7050514221191406,
      "step": 62892
    },
    {
      "epoch": 5.997848510742188e-05,
      "step": 62892,
      "training_step_time": 8.088135242462158
    },
    {
      "epoch": 5.997943878173828e-05,
      "model_forward_time": 1.7079124450683594,
      "step": 62893
    },
    {
      "epoch": 5.997943878173828e-05,
      "step": 62893,
      "training_step_time": 7.77403998374939
    },
    {
      "epoch": 5.9980392456054686e-05,
      "model_forward_time": 2.256617784500122,
      "step": 62894
    },
    {
      "epoch": 5.9980392456054686e-05,
      "step": 62894,
      "training_step_time": 9.174904346466064
    },
    {
      "epoch": 5.99813461303711e-05,
      "model_forward_time": 1.7248284816741943,
      "step": 62895
    },
    {
      "epoch": 5.99813461303711e-05,
      "step": 62895,
      "training_step_time": 8.991808891296387
    },
    {
      "epoch": 5.99822998046875e-05,
      "model_forward_time": 1.6997201442718506,
      "step": 62896
    },
    {
      "epoch": 5.99822998046875e-05,
      "step": 62896,
      "training_step_time": 8.867435693740845
    },
    {
      "epoch": 5.9983253479003905e-05,
      "model_forward_time": 1.7036066055297852,
      "step": 62897
    },
    {
      "epoch": 5.9983253479003905e-05,
      "step": 62897,
      "training_step_time": 8.486218690872192
    },
    {
      "epoch": 5.9984207153320316e-05,
      "model_forward_time": 1.713430643081665,
      "step": 62898
    },
    {
      "epoch": 5.9984207153320316e-05,
      "step": 62898,
      "training_step_time": 8.666651964187622
    },
    {
      "epoch": 5.998516082763672e-05,
      "model_forward_time": 1.3560526371002197,
      "step": 62899
    },
    {
      "epoch": 5.998516082763672e-05,
      "step": 62899,
      "training_step_time": 7.77525520324707
    },
    {
      "epoch": 5.9986114501953124e-05,
      "grad_norm": 0.11702943593263626,
      "learning_rate": 3.314246965352132e-06,
      "loss": 0.0605,
      "step": 62900
    },
    {
      "epoch": 5.9986114501953124e-05,
      "model_forward_time": 1.3527212142944336,
      "step": 62900
    },
    {
      "dynamics_loss_avg": 0.0657019334845245,
      "epoch": 5.9986114501953124e-05,
      "step": 62900
    },
    {
      "action_loss_avg": 0.002269912196788937,
      "epoch": 5.9986114501953124e-05,
      "step": 62900
    },
    {
      "epoch": 5.9986114501953124e-05,
      "step": 62900,
      "training_step_time": 7.857267379760742
    },
    {
      "epoch": 5.9987068176269534e-05,
      "model_forward_time": 1.7197928428649902,
      "step": 62901
    },
    {
      "epoch": 5.9987068176269534e-05,
      "step": 62901,
      "training_step_time": 7.965711832046509
    },
    {
      "epoch": 5.998802185058594e-05,
      "model_forward_time": 1.7086296081542969,
      "step": 62902
    },
    {
      "epoch": 5.998802185058594e-05,
      "step": 62902,
      "training_step_time": 8.690736293792725
    },
    {
      "epoch": 5.998897552490234e-05,
      "model_forward_time": 1.7164201736450195,
      "step": 62903
    },
    {
      "epoch": 5.998897552490234e-05,
      "step": 62903,
      "training_step_time": 8.873956441879272
    },
    {
      "epoch": 5.998992919921875e-05,
      "model_forward_time": 2.2898402214050293,
      "step": 62904
    },
    {
      "epoch": 5.998992919921875e-05,
      "step": 62904,
      "training_step_time": 8.713834047317505
    },
    {
      "epoch": 5.999088287353516e-05,
      "model_forward_time": 1.7172701358795166,
      "step": 62905
    },
    {
      "epoch": 5.999088287353516e-05,
      "step": 62905,
      "training_step_time": 9.79001784324646
    },
    {
      "epoch": 5.999183654785156e-05,
      "model_forward_time": 1.7227318286895752,
      "step": 62906
    },
    {
      "epoch": 5.999183654785156e-05,
      "step": 62906,
      "training_step_time": 8.592349290847778
    },
    {
      "epoch": 5.9992790222167965e-05,
      "model_forward_time": 1.7109735012054443,
      "step": 62907
    },
    {
      "epoch": 5.9992790222167965e-05,
      "step": 62907,
      "training_step_time": 8.671905994415283
    },
    {
      "epoch": 5.9993743896484376e-05,
      "model_forward_time": 1.7316322326660156,
      "step": 62908
    },
    {
      "epoch": 5.9993743896484376e-05,
      "step": 62908,
      "training_step_time": 7.747970819473267
    },
    {
      "epoch": 5.999469757080078e-05,
      "model_forward_time": 1.7132713794708252,
      "step": 62909
    },
    {
      "epoch": 5.999469757080078e-05,
      "step": 62909,
      "training_step_time": 8.00395131111145
    },
    {
      "epoch": 5.9995651245117184e-05,
      "grad_norm": 0.08323812484741211,
      "learning_rate": 3.3126903971970286e-06,
      "loss": 0.0562,
      "step": 62910
    },
    {
      "epoch": 5.9995651245117184e-05,
      "model_forward_time": 1.7090022563934326,
      "step": 62910
    },
    {
      "dynamics_loss_avg": 0.05096307955682278,
      "epoch": 5.9995651245117184e-05,
      "step": 62910
    },
    {
      "action_loss_avg": 0.001841971048270352,
      "epoch": 5.9995651245117184e-05,
      "step": 62910
    },
    {
      "epoch": 5.9995651245117184e-05,
      "step": 62910,
      "training_step_time": 7.926756381988525
    },
    {
      "epoch": 5.9996604919433595e-05,
      "model_forward_time": 1.7174184322357178,
      "step": 62911
    },
    {
      "epoch": 5.9996604919433595e-05,
      "step": 62911,
      "training_step_time": 7.8732593059539795
    },
    {
      "epoch": 5.999755859375e-05,
      "model_forward_time": 1.7170612812042236,
      "step": 62912
    },
    {
      "epoch": 5.999755859375e-05,
      "step": 62912,
      "training_step_time": 7.830447673797607
    },
    {
      "epoch": 5.99985122680664e-05,
      "model_forward_time": 2.2592852115631104,
      "step": 62913
    },
    {
      "epoch": 5.99985122680664e-05,
      "step": 62913,
      "training_step_time": 7.903681516647339
    },
    {
      "epoch": 5.9999465942382814e-05,
      "model_forward_time": 1.7658698558807373,
      "step": 62914
    },
    {
      "epoch": 5.9999465942382814e-05,
      "step": 62914,
      "training_step_time": 7.887175559997559
    },
    {
      "epoch": 6.000041961669922e-05,
      "model_forward_time": 1.7073957920074463,
      "step": 62915
    },
    {
      "epoch": 6.000041961669922e-05,
      "step": 62915,
      "training_step_time": 8.502418518066406
    },
    {
      "epoch": 6.000137329101562e-05,
      "model_forward_time": 1.7104804515838623,
      "step": 62916
    },
    {
      "epoch": 6.000137329101562e-05,
      "step": 62916,
      "training_step_time": 7.871134996414185
    },
    {
      "epoch": 6.000232696533203e-05,
      "model_forward_time": 1.7106328010559082,
      "step": 62917
    },
    {
      "epoch": 6.000232696533203e-05,
      "step": 62917,
      "training_step_time": 7.750079154968262
    },
    {
      "epoch": 6.000328063964844e-05,
      "model_forward_time": 1.709064245223999,
      "step": 62918
    },
    {
      "epoch": 6.000328063964844e-05,
      "step": 62918,
      "training_step_time": 8.958755254745483
    },
    {
      "epoch": 6.000423431396484e-05,
      "model_forward_time": 1.6948153972625732,
      "step": 62919
    },
    {
      "epoch": 6.000423431396484e-05,
      "step": 62919,
      "training_step_time": 7.933894634246826
    },
    {
      "epoch": 6.000518798828125e-05,
      "grad_norm": 0.08595035970211029,
      "learning_rate": 3.3111340135635645e-06,
      "loss": 0.0563,
      "step": 62920
    },
    {
      "epoch": 6.000518798828125e-05,
      "model_forward_time": 1.7031941413879395,
      "step": 62920
    },
    {
      "dynamics_loss_avg": 0.057007538713514806,
      "epoch": 6.000518798828125e-05,
      "step": 62920
    },
    {
      "action_loss_avg": 0.0015309975831769406,
      "epoch": 6.000518798828125e-05,
      "step": 62920
    },
    {
      "epoch": 6.000518798828125e-05,
      "step": 62920,
      "training_step_time": 7.802191257476807
    },
    {
      "epoch": 6.0006141662597656e-05,
      "model_forward_time": 1.7102208137512207,
      "step": 62921
    },
    {
      "epoch": 6.0006141662597656e-05,
      "step": 62921,
      "training_step_time": 7.792570352554321
    },
    {
      "epoch": 6.000709533691406e-05,
      "model_forward_time": 2.29146671295166,
      "step": 62922
    },
    {
      "epoch": 6.000709533691406e-05,
      "step": 62922,
      "training_step_time": 8.597690105438232
    },
    {
      "epoch": 6.000804901123047e-05,
      "model_forward_time": 1.7632944583892822,
      "step": 62923
    },
    {
      "epoch": 6.000804901123047e-05,
      "step": 62923,
      "training_step_time": 9.953145980834961
    },
    {
      "epoch": 6.0009002685546875e-05,
      "model_forward_time": 1.701061725616455,
      "step": 62924
    },
    {
      "epoch": 6.0009002685546875e-05,
      "step": 62924,
      "training_step_time": 7.964964389801025
    },
    {
      "epoch": 6.000995635986328e-05,
      "model_forward_time": 1.7067234516143799,
      "step": 62925
    },
    {
      "epoch": 6.000995635986328e-05,
      "step": 62925,
      "training_step_time": 7.635157823562622
    },
    {
      "epoch": 6.001091003417969e-05,
      "model_forward_time": 1.700124979019165,
      "step": 62926
    },
    {
      "epoch": 6.001091003417969e-05,
      "step": 62926,
      "training_step_time": 7.721272230148315
    },
    {
      "epoch": 6.001186370849609e-05,
      "model_forward_time": 1.702585220336914,
      "step": 62927
    },
    {
      "epoch": 6.001186370849609e-05,
      "step": 62927,
      "training_step_time": 8.663453578948975
    },
    {
      "epoch": 6.00128173828125e-05,
      "model_forward_time": 1.706444501876831,
      "step": 62928
    },
    {
      "epoch": 6.00128173828125e-05,
      "step": 62928,
      "training_step_time": 7.94921350479126
    },
    {
      "epoch": 6.001377105712891e-05,
      "model_forward_time": 1.7006752490997314,
      "step": 62929
    },
    {
      "epoch": 6.001377105712891e-05,
      "step": 62929,
      "training_step_time": 7.962702989578247
    },
    {
      "epoch": 6.001472473144531e-05,
      "grad_norm": 0.11528909206390381,
      "learning_rate": 3.309577814621947e-06,
      "loss": 0.0573,
      "step": 62930
    },
    {
      "epoch": 6.001472473144531e-05,
      "model_forward_time": 1.7060151100158691,
      "step": 62930
    },
    {
      "dynamics_loss_avg": 0.061184655968099834,
      "epoch": 6.001472473144531e-05,
      "step": 62930
    },
    {
      "action_loss_avg": 0.0011511446151416748,
      "epoch": 6.001472473144531e-05,
      "step": 62930
    },
    {
      "epoch": 6.001472473144531e-05,
      "step": 62930,
      "training_step_time": 8.778931379318237
    },
    {
      "epoch": 6.0015678405761716e-05,
      "model_forward_time": 2.2800891399383545,
      "step": 62931
    },
    {
      "epoch": 6.0015678405761716e-05,
      "step": 62931,
      "training_step_time": 8.020381450653076
    },
    {
      "epoch": 6.001663208007813e-05,
      "model_forward_time": 1.7424075603485107,
      "step": 62932
    },
    {
      "epoch": 6.001663208007813e-05,
      "step": 62932,
      "training_step_time": 8.868067741394043
    },
    {
      "epoch": 6.001758575439453e-05,
      "model_forward_time": 1.7089595794677734,
      "step": 62933
    },
    {
      "epoch": 6.001758575439453e-05,
      "step": 62933,
      "training_step_time": 9.858841180801392
    },
    {
      "epoch": 6.0018539428710935e-05,
      "model_forward_time": 1.7120301723480225,
      "step": 62934
    },
    {
      "epoch": 6.0018539428710935e-05,
      "step": 62934,
      "training_step_time": 9.19020938873291
    },
    {
      "epoch": 6.0019493103027346e-05,
      "model_forward_time": 1.6994948387145996,
      "step": 62935
    },
    {
      "epoch": 6.0019493103027346e-05,
      "step": 62935,
      "training_step_time": 8.689653396606445
    },
    {
      "epoch": 6.002044677734375e-05,
      "model_forward_time": 1.7106950283050537,
      "step": 62936
    },
    {
      "epoch": 6.002044677734375e-05,
      "step": 62936,
      "training_step_time": 7.817102432250977
    },
    {
      "epoch": 6.0021400451660154e-05,
      "model_forward_time": 1.6966187953948975,
      "step": 62937
    },
    {
      "epoch": 6.0021400451660154e-05,
      "step": 62937,
      "training_step_time": 7.870442628860474
    },
    {
      "epoch": 6.0022354125976565e-05,
      "model_forward_time": 1.734713077545166,
      "step": 62938
    },
    {
      "epoch": 6.0022354125976565e-05,
      "step": 62938,
      "training_step_time": 7.976722478866577
    },
    {
      "epoch": 6.002330780029297e-05,
      "model_forward_time": 1.7455463409423828,
      "step": 62939
    },
    {
      "epoch": 6.002330780029297e-05,
      "step": 62939,
      "training_step_time": 8.832165479660034
    },
    {
      "epoch": 6.002426147460937e-05,
      "grad_norm": 0.11301985383033752,
      "learning_rate": 3.3080218005423592e-06,
      "loss": 0.0565,
      "step": 62940
    },
    {
      "epoch": 6.002426147460937e-05,
      "model_forward_time": 1.7508342266082764,
      "step": 62940
    },
    {
      "dynamics_loss_avg": 0.053591635264456275,
      "epoch": 6.002426147460937e-05,
      "step": 62940
    },
    {
      "action_loss_avg": 0.001560972142033279,
      "epoch": 6.002426147460937e-05,
      "step": 62940
    },
    {
      "epoch": 6.002426147460937e-05,
      "step": 62940,
      "training_step_time": 10.21416449546814
    },
    {
      "epoch": 6.0025215148925784e-05,
      "model_forward_time": 1.748110055923462,
      "step": 62941
    },
    {
      "epoch": 6.0025215148925784e-05,
      "step": 62941,
      "training_step_time": 7.978355646133423
    },
    {
      "epoch": 6.002616882324219e-05,
      "model_forward_time": 1.747431993484497,
      "step": 62942
    },
    {
      "epoch": 6.002616882324219e-05,
      "step": 62942,
      "training_step_time": 7.909308910369873
    },
    {
      "epoch": 6.002712249755859e-05,
      "model_forward_time": 1.7941250801086426,
      "step": 62943
    },
    {
      "epoch": 6.002712249755859e-05,
      "step": 62943,
      "training_step_time": 7.940809726715088
    },
    {
      "epoch": 6.0028076171875e-05,
      "model_forward_time": 1.7435328960418701,
      "step": 62944
    },
    {
      "epoch": 6.0028076171875e-05,
      "step": 62944,
      "training_step_time": 7.915380954742432
    },
    {
      "epoch": 6.0029029846191407e-05,
      "model_forward_time": 1.7409100532531738,
      "step": 62945
    },
    {
      "epoch": 6.0029029846191407e-05,
      "step": 62945,
      "training_step_time": 7.7659752368927
    },
    {
      "epoch": 6.002998352050781e-05,
      "model_forward_time": 1.7405173778533936,
      "step": 62946
    },
    {
      "epoch": 6.002998352050781e-05,
      "step": 62946,
      "training_step_time": 7.8923304080963135
    },
    {
      "epoch": 6.003093719482422e-05,
      "model_forward_time": 1.7446479797363281,
      "step": 62947
    },
    {
      "epoch": 6.003093719482422e-05,
      "step": 62947,
      "training_step_time": 7.974047422409058
    },
    {
      "epoch": 6.0031890869140625e-05,
      "model_forward_time": 1.7419805526733398,
      "step": 62948
    },
    {
      "epoch": 6.0031890869140625e-05,
      "step": 62948,
      "training_step_time": 7.8743743896484375
    },
    {
      "epoch": 6.003284454345703e-05,
      "model_forward_time": 1.7503764629364014,
      "step": 62949
    },
    {
      "epoch": 6.003284454345703e-05,
      "step": 62949,
      "training_step_time": 7.934877634048462
    },
    {
      "epoch": 6.003379821777344e-05,
      "grad_norm": 0.07353062182664871,
      "learning_rate": 3.3064659714949628e-06,
      "loss": 0.0558,
      "step": 62950
    },
    {
      "epoch": 6.003379821777344e-05,
      "model_forward_time": 2.9182450771331787,
      "step": 62950
    },
    {
      "dynamics_loss_avg": 0.0497759960591793,
      "epoch": 6.003379821777344e-05,
      "step": 62950
    },
    {
      "action_loss_avg": 0.0018222659651655703,
      "epoch": 6.003379821777344e-05,
      "step": 62950
    },
    {
      "epoch": 6.003379821777344e-05,
      "step": 62950,
      "training_step_time": 7.974090576171875
    },
    {
      "epoch": 6.0034751892089844e-05,
      "model_forward_time": 1.741919755935669,
      "step": 62951
    },
    {
      "epoch": 6.0034751892089844e-05,
      "step": 62951,
      "training_step_time": 8.679365158081055
    },
    {
      "epoch": 6.003570556640625e-05,
      "model_forward_time": 1.7904257774353027,
      "step": 62952
    },
    {
      "epoch": 6.003570556640625e-05,
      "step": 62952,
      "training_step_time": 9.671945095062256
    },
    {
      "epoch": 6.003665924072266e-05,
      "model_forward_time": 1.7496662139892578,
      "step": 62953
    },
    {
      "epoch": 6.003665924072266e-05,
      "step": 62953,
      "training_step_time": 8.324403524398804
    },
    {
      "epoch": 6.003761291503906e-05,
      "model_forward_time": 1.7491743564605713,
      "step": 62954
    },
    {
      "epoch": 6.003761291503906e-05,
      "step": 62954,
      "training_step_time": 7.825846433639526
    },
    {
      "epoch": 6.003856658935547e-05,
      "model_forward_time": 1.7128713130950928,
      "step": 62955
    },
    {
      "epoch": 6.003856658935547e-05,
      "step": 62955,
      "training_step_time": 7.912654876708984
    },
    {
      "epoch": 6.003952026367188e-05,
      "model_forward_time": 1.7355632781982422,
      "step": 62956
    },
    {
      "epoch": 6.003952026367188e-05,
      "step": 62956,
      "training_step_time": 7.98900580406189
    },
    {
      "epoch": 6.004047393798828e-05,
      "model_forward_time": 1.72763991355896,
      "step": 62957
    },
    {
      "epoch": 6.004047393798828e-05,
      "step": 62957,
      "training_step_time": 8.504590272903442
    },
    {
      "epoch": 6.0041427612304686e-05,
      "model_forward_time": 1.7282147407531738,
      "step": 62958
    },
    {
      "epoch": 6.0041427612304686e-05,
      "step": 62958,
      "training_step_time": 7.866559982299805
    },
    {
      "epoch": 6.00423812866211e-05,
      "model_forward_time": 2.914421319961548,
      "step": 62959
    },
    {
      "epoch": 6.00423812866211e-05,
      "step": 62959,
      "training_step_time": 8.660847663879395
    },
    {
      "epoch": 6.00433349609375e-05,
      "grad_norm": 0.09539619833230972,
      "learning_rate": 3.3049103276499006e-06,
      "loss": 0.062,
      "step": 62960
    },
    {
      "epoch": 6.00433349609375e-05,
      "model_forward_time": 1.7195701599121094,
      "step": 62960
    },
    {
      "dynamics_loss_avg": 0.06454421114176512,
      "epoch": 6.00433349609375e-05,
      "step": 62960
    },
    {
      "action_loss_avg": 0.0026833837386220694,
      "epoch": 6.00433349609375e-05,
      "step": 62960
    },
    {
      "epoch": 6.00433349609375e-05,
      "step": 62960,
      "training_step_time": 8.793622970581055
    },
    {
      "epoch": 6.0044288635253905e-05,
      "model_forward_time": 1.7335906028747559,
      "step": 62961
    },
    {
      "epoch": 6.0044288635253905e-05,
      "step": 62961,
      "training_step_time": 9.672209739685059
    },
    {
      "epoch": 6.0045242309570316e-05,
      "model_forward_time": 1.7368109226226807,
      "step": 62962
    },
    {
      "epoch": 6.0045242309570316e-05,
      "step": 62962,
      "training_step_time": 7.869454383850098
    },
    {
      "epoch": 6.004619598388672e-05,
      "model_forward_time": 1.729189157485962,
      "step": 62963
    },
    {
      "epoch": 6.004619598388672e-05,
      "step": 62963,
      "training_step_time": 7.697751998901367
    },
    {
      "epoch": 6.0047149658203124e-05,
      "model_forward_time": 1.7244482040405273,
      "step": 62964
    },
    {
      "epoch": 6.0047149658203124e-05,
      "step": 62964,
      "training_step_time": 7.8834240436553955
    },
    {
      "epoch": 6.0048103332519535e-05,
      "model_forward_time": 1.7381811141967773,
      "step": 62965
    },
    {
      "epoch": 6.0048103332519535e-05,
      "step": 62965,
      "training_step_time": 8.546702146530151
    },
    {
      "epoch": 6.004905700683594e-05,
      "model_forward_time": 1.7344424724578857,
      "step": 62966
    },
    {
      "epoch": 6.004905700683594e-05,
      "step": 62966,
      "training_step_time": 8.770081520080566
    },
    {
      "epoch": 6.005001068115234e-05,
      "model_forward_time": 1.7341902256011963,
      "step": 62967
    },
    {
      "epoch": 6.005001068115234e-05,
      "step": 62967,
      "training_step_time": 8.593989610671997
    },
    {
      "epoch": 6.005096435546875e-05,
      "model_forward_time": 2.8261189460754395,
      "step": 62968
    },
    {
      "epoch": 6.005096435546875e-05,
      "step": 62968,
      "training_step_time": 9.04947304725647
    },
    {
      "epoch": 6.005191802978516e-05,
      "model_forward_time": 1.7248613834381104,
      "step": 62969
    },
    {
      "epoch": 6.005191802978516e-05,
      "step": 62969,
      "training_step_time": 8.038412094116211
    },
    {
      "epoch": 6.005287170410156e-05,
      "grad_norm": 0.09598482400178909,
      "learning_rate": 3.3033548691772976e-06,
      "loss": 0.0565,
      "step": 62970
    },
    {
      "epoch": 6.005287170410156e-05,
      "model_forward_time": 1.727384090423584,
      "step": 62970
    },
    {
      "dynamics_loss_avg": 0.047985462658107284,
      "epoch": 6.005287170410156e-05,
      "step": 62970
    },
    {
      "action_loss_avg": 0.0011372882523573934,
      "epoch": 6.005287170410156e-05,
      "step": 62970
    },
    {
      "epoch": 6.005287170410156e-05,
      "step": 62970,
      "training_step_time": 8.615983247756958
    },
    {
      "epoch": 6.0053825378417965e-05,
      "model_forward_time": 1.7744698524475098,
      "step": 62971
    },
    {
      "epoch": 6.0053825378417965e-05,
      "step": 62971,
      "training_step_time": 7.883599042892456
    },
    {
      "epoch": 6.0054779052734376e-05,
      "model_forward_time": 1.7357258796691895,
      "step": 62972
    },
    {
      "epoch": 6.0054779052734376e-05,
      "step": 62972,
      "training_step_time": 7.729377508163452
    },
    {
      "epoch": 6.005573272705078e-05,
      "model_forward_time": 1.7282216548919678,
      "step": 62973
    },
    {
      "epoch": 6.005573272705078e-05,
      "step": 62973,
      "training_step_time": 8.930661916732788
    },
    {
      "epoch": 6.0056686401367184e-05,
      "model_forward_time": 1.7257921695709229,
      "step": 62974
    },
    {
      "epoch": 6.0056686401367184e-05,
      "step": 62974,
      "training_step_time": 7.985293626785278
    },
    {
      "epoch": 6.0057640075683595e-05,
      "model_forward_time": 1.7286763191223145,
      "step": 62975
    },
    {
      "epoch": 6.0057640075683595e-05,
      "step": 62975,
      "training_step_time": 8.818998098373413
    },
    {
      "epoch": 6.005859375e-05,
      "model_forward_time": 1.7303693294525146,
      "step": 62976
    },
    {
      "epoch": 6.005859375e-05,
      "step": 62976,
      "training_step_time": 9.106261253356934
    },
    {
      "epoch": 6.00595474243164e-05,
      "model_forward_time": 2.85444974899292,
      "step": 62977
    },
    {
      "epoch": 6.00595474243164e-05,
      "step": 62977,
      "training_step_time": 7.912373065948486
    },
    {
      "epoch": 6.0060501098632814e-05,
      "model_forward_time": 1.7301356792449951,
      "step": 62978
    },
    {
      "epoch": 6.0060501098632814e-05,
      "step": 62978,
      "training_step_time": 8.04027247428894
    },
    {
      "epoch": 6.006145477294922e-05,
      "model_forward_time": 1.7322194576263428,
      "step": 62979
    },
    {
      "epoch": 6.006145477294922e-05,
      "step": 62979,
      "training_step_time": 8.033562898635864
    },
    {
      "epoch": 6.006240844726562e-05,
      "grad_norm": 0.10768201947212219,
      "learning_rate": 3.301799596247255e-06,
      "loss": 0.0568,
      "step": 62980
    },
    {
      "epoch": 6.006240844726562e-05,
      "model_forward_time": 1.7553391456604004,
      "step": 62980
    },
    {
      "dynamics_loss_avg": 0.05355647131800652,
      "epoch": 6.006240844726562e-05,
      "step": 62980
    },
    {
      "action_loss_avg": 0.0012685004010563716,
      "epoch": 6.006240844726562e-05,
      "step": 62980
    },
    {
      "epoch": 6.006240844726562e-05,
      "step": 62980,
      "training_step_time": 7.809648036956787
    },
    {
      "epoch": 6.006336212158203e-05,
      "model_forward_time": 1.7362253665924072,
      "step": 62981
    },
    {
      "epoch": 6.006336212158203e-05,
      "step": 62981,
      "training_step_time": 7.837890863418579
    },
    {
      "epoch": 6.006431579589844e-05,
      "model_forward_time": 1.7260372638702393,
      "step": 62982
    },
    {
      "epoch": 6.006431579589844e-05,
      "step": 62982,
      "training_step_time": 7.909941911697388
    },
    {
      "epoch": 6.006526947021484e-05,
      "model_forward_time": 1.73164963722229,
      "step": 62983
    },
    {
      "epoch": 6.006526947021484e-05,
      "step": 62983,
      "training_step_time": 7.78895902633667
    },
    {
      "epoch": 6.006622314453125e-05,
      "model_forward_time": 1.7334744930267334,
      "step": 62984
    },
    {
      "epoch": 6.006622314453125e-05,
      "step": 62984,
      "training_step_time": 7.9691572189331055
    },
    {
      "epoch": 6.0067176818847656e-05,
      "model_forward_time": 1.7570624351501465,
      "step": 62985
    },
    {
      "epoch": 6.0067176818847656e-05,
      "step": 62985,
      "training_step_time": 8.357642412185669
    },
    {
      "epoch": 6.006813049316406e-05,
      "model_forward_time": 2.8958144187927246,
      "step": 62986
    },
    {
      "epoch": 6.006813049316406e-05,
      "step": 62986,
      "training_step_time": 10.245298624038696
    },
    {
      "epoch": 6.006908416748047e-05,
      "model_forward_time": 1.746952772140503,
      "step": 62987
    },
    {
      "epoch": 6.006908416748047e-05,
      "step": 62987,
      "training_step_time": 7.971156597137451
    },
    {
      "epoch": 6.0070037841796875e-05,
      "model_forward_time": 1.7457377910614014,
      "step": 62988
    },
    {
      "epoch": 6.0070037841796875e-05,
      "step": 62988,
      "training_step_time": 7.7508463859558105
    },
    {
      "epoch": 6.007099151611328e-05,
      "model_forward_time": 1.755220890045166,
      "step": 62989
    },
    {
      "epoch": 6.007099151611328e-05,
      "step": 62989,
      "training_step_time": 7.948971748352051
    },
    {
      "epoch": 6.007194519042969e-05,
      "grad_norm": 0.1418815553188324,
      "learning_rate": 3.300244509029854e-06,
      "loss": 0.0538,
      "step": 62990
    },
    {
      "epoch": 6.007194519042969e-05,
      "model_forward_time": 1.7617547512054443,
      "step": 62990
    },
    {
      "dynamics_loss_avg": 0.045810509286820886,
      "epoch": 6.007194519042969e-05,
      "step": 62990
    },
    {
      "action_loss_avg": 0.0017390094639267773,
      "epoch": 6.007194519042969e-05,
      "step": 62990
    },
    {
      "epoch": 6.007194519042969e-05,
      "step": 62990,
      "training_step_time": 7.825658559799194
    },
    {
      "epoch": 6.0072898864746093e-05,
      "model_forward_time": 1.7478418350219727,
      "step": 62991
    },
    {
      "epoch": 6.0072898864746093e-05,
      "step": 62991,
      "training_step_time": 7.8960347175598145
    },
    {
      "epoch": 6.00738525390625e-05,
      "model_forward_time": 1.75370454788208,
      "step": 62992
    },
    {
      "epoch": 6.00738525390625e-05,
      "step": 62992,
      "training_step_time": 7.780990839004517
    },
    {
      "epoch": 6.007480621337891e-05,
      "model_forward_time": 1.7483115196228027,
      "step": 62993
    },
    {
      "epoch": 6.007480621337891e-05,
      "step": 62993,
      "training_step_time": 7.943360090255737
    },
    {
      "epoch": 6.007575988769531e-05,
      "model_forward_time": 1.7382709980010986,
      "step": 62994
    },
    {
      "epoch": 6.007575988769531e-05,
      "step": 62994,
      "training_step_time": 8.665061473846436
    },
    {
      "epoch": 6.0076713562011716e-05,
      "model_forward_time": 1.740647792816162,
      "step": 62995
    },
    {
      "epoch": 6.0076713562011716e-05,
      "step": 62995,
      "training_step_time": 8.308944463729858
    },
    {
      "epoch": 6.007766723632813e-05,
      "model_forward_time": 1.7528252601623535,
      "step": 62996
    },
    {
      "epoch": 6.007766723632813e-05,
      "step": 62996,
      "training_step_time": 8.93113923072815
    },
    {
      "epoch": 6.007862091064453e-05,
      "model_forward_time": 1.7432804107666016,
      "step": 62997
    },
    {
      "epoch": 6.007862091064453e-05,
      "step": 62997,
      "training_step_time": 8.723820686340332
    },
    {
      "epoch": 6.0079574584960935e-05,
      "model_forward_time": 1.7999382019042969,
      "step": 62998
    },
    {
      "epoch": 6.0079574584960935e-05,
      "step": 62998,
      "training_step_time": 8.158217191696167
    },
    {
      "epoch": 6.0080528259277346e-05,
      "model_forward_time": 1.8658969402313232,
      "step": 62999
    },
    {
      "epoch": 6.0080528259277346e-05,
      "step": 62999,
      "training_step_time": 8.83198070526123
    },
    {
      "epoch": 6.008148193359375e-05,
      "grad_norm": 0.0836556926369667,
      "learning_rate": 3.2986896076951586e-06,
      "loss": 0.0567,
      "step": 63000
    },
    {
      "epoch": 6.008148193359375e-05,
      "model_forward_time": 1.7674570083618164,
      "step": 63000
    },
    {
      "dynamics_loss_avg": 0.05569224171340466,
      "epoch": 6.008148193359375e-05,
      "step": 63000
    },
    {
      "action_loss_avg": 0.0012469907815102488,
      "epoch": 6.008148193359375e-05,
      "step": 63000
    },
    {
      "epoch": 6.008148193359375e-05,
      "step": 63000,
      "training_step_time": 7.6203484535217285
    },
    {
      "epoch": 6.0082435607910154e-05,
      "model_forward_time": 1.7005045413970947,
      "step": 63001
    },
    {
      "epoch": 6.0082435607910154e-05,
      "step": 63001,
      "training_step_time": 8.709568500518799
    },
    {
      "epoch": 6.0083389282226565e-05,
      "model_forward_time": 1.705251693725586,
      "step": 63002
    },
    {
      "epoch": 6.0083389282226565e-05,
      "step": 63002,
      "training_step_time": 8.971708536148071
    },
    {
      "epoch": 6.008434295654297e-05,
      "model_forward_time": 1.699751377105713,
      "step": 63003
    },
    {
      "epoch": 6.008434295654297e-05,
      "step": 63003,
      "training_step_time": 9.070922136306763
    },
    {
      "epoch": 6.008529663085937e-05,
      "model_forward_time": 2.742927312850952,
      "step": 63004
    },
    {
      "epoch": 6.008529663085937e-05,
      "step": 63004,
      "training_step_time": 10.182333946228027
    },
    {
      "epoch": 6.0086250305175784e-05,
      "model_forward_time": 1.7082409858703613,
      "step": 63005
    },
    {
      "epoch": 6.0086250305175784e-05,
      "step": 63005,
      "training_step_time": 8.373466491699219
    },
    {
      "epoch": 6.008720397949219e-05,
      "model_forward_time": 1.6959669589996338,
      "step": 63006
    },
    {
      "epoch": 6.008720397949219e-05,
      "step": 63006,
      "training_step_time": 8.749916791915894
    },
    {
      "epoch": 6.008815765380859e-05,
      "model_forward_time": 1.6935462951660156,
      "step": 63007
    },
    {
      "epoch": 6.008815765380859e-05,
      "step": 63007,
      "training_step_time": 11.482887983322144
    },
    {
      "epoch": 6.0089111328125e-05,
      "model_forward_time": 1.6962287425994873,
      "step": 63008
    },
    {
      "epoch": 6.0089111328125e-05,
      "step": 63008,
      "training_step_time": 10.353850364685059
    },
    {
      "epoch": 6.0090065002441407e-05,
      "model_forward_time": 1.6958603858947754,
      "step": 63009
    },
    {
      "epoch": 6.0090065002441407e-05,
      "step": 63009,
      "training_step_time": 7.814859628677368
    },
    {
      "epoch": 6.009101867675781e-05,
      "grad_norm": 1.128883719444275,
      "learning_rate": 3.2971348924132107e-06,
      "loss": 0.053,
      "step": 63010
    },
    {
      "epoch": 6.009101867675781e-05,
      "model_forward_time": 1.6947212219238281,
      "step": 63010
    },
    {
      "dynamics_loss_avg": 0.04697207137942314,
      "epoch": 6.009101867675781e-05,
      "step": 63010
    },
    {
      "action_loss_avg": 0.0014894701540470122,
      "epoch": 6.009101867675781e-05,
      "step": 63010
    },
    {
      "epoch": 6.009101867675781e-05,
      "step": 63010,
      "training_step_time": 7.618507623672485
    },
    {
      "epoch": 6.009197235107422e-05,
      "model_forward_time": 1.695744514465332,
      "step": 63011
    },
    {
      "epoch": 6.009197235107422e-05,
      "step": 63011,
      "training_step_time": 7.832446575164795
    },
    {
      "epoch": 6.0092926025390625e-05,
      "model_forward_time": 1.6970245838165283,
      "step": 63012
    },
    {
      "epoch": 6.0092926025390625e-05,
      "step": 63012,
      "training_step_time": 8.427634954452515
    },
    {
      "epoch": 6.009387969970703e-05,
      "model_forward_time": 2.2700719833374023,
      "step": 63013
    },
    {
      "epoch": 6.009387969970703e-05,
      "step": 63013,
      "training_step_time": 8.525773048400879
    },
    {
      "epoch": 6.009483337402344e-05,
      "model_forward_time": 1.6941542625427246,
      "step": 63014
    },
    {
      "epoch": 6.009483337402344e-05,
      "step": 63014,
      "training_step_time": 10.550074577331543
    },
    {
      "epoch": 6.0095787048339844e-05,
      "model_forward_time": 1.7543160915374756,
      "step": 63015
    },
    {
      "epoch": 6.0095787048339844e-05,
      "step": 63015,
      "training_step_time": 7.783675909042358
    },
    {
      "epoch": 6.009674072265625e-05,
      "model_forward_time": 1.6961100101470947,
      "step": 63016
    },
    {
      "epoch": 6.009674072265625e-05,
      "step": 63016,
      "training_step_time": 9.41735577583313
    },
    {
      "epoch": 6.009769439697266e-05,
      "model_forward_time": 1.7071285247802734,
      "step": 63017
    },
    {
      "epoch": 6.009769439697266e-05,
      "step": 63017,
      "training_step_time": 8.77757477760315
    },
    {
      "epoch": 6.009864807128906e-05,
      "model_forward_time": 1.6956384181976318,
      "step": 63018
    },
    {
      "epoch": 6.009864807128906e-05,
      "step": 63018,
      "training_step_time": 7.693946123123169
    },
    {
      "epoch": 6.009960174560547e-05,
      "model_forward_time": 1.6992688179016113,
      "step": 63019
    },
    {
      "epoch": 6.009960174560547e-05,
      "step": 63019,
      "training_step_time": 7.748617649078369
    },
    {
      "epoch": 6.010055541992188e-05,
      "grad_norm": 0.10575306415557861,
      "learning_rate": 3.295580363354031e-06,
      "loss": 0.057,
      "step": 63020
    },
    {
      "epoch": 6.010055541992188e-05,
      "model_forward_time": 1.6906390190124512,
      "step": 63020
    },
    {
      "dynamics_loss_avg": 0.06252388283610344,
      "epoch": 6.010055541992188e-05,
      "step": 63020
    },
    {
      "action_loss_avg": 0.0019337197300046682,
      "epoch": 6.010055541992188e-05,
      "step": 63020
    },
    {
      "epoch": 6.010055541992188e-05,
      "step": 63020,
      "training_step_time": 8.583555459976196
    },
    {
      "epoch": 6.010150909423828e-05,
      "model_forward_time": 1.694188117980957,
      "step": 63021
    },
    {
      "epoch": 6.010150909423828e-05,
      "step": 63021,
      "training_step_time": 7.71719217300415
    },
    {
      "epoch": 6.0102462768554686e-05,
      "model_forward_time": 2.2333734035491943,
      "step": 63022
    },
    {
      "epoch": 6.0102462768554686e-05,
      "step": 63022,
      "training_step_time": 8.161251783370972
    },
    {
      "epoch": 6.01034164428711e-05,
      "model_forward_time": 1.6944336891174316,
      "step": 63023
    },
    {
      "epoch": 6.01034164428711e-05,
      "step": 63023,
      "training_step_time": 8.667691946029663
    },
    {
      "epoch": 6.01043701171875e-05,
      "model_forward_time": 1.7594175338745117,
      "step": 63024
    },
    {
      "epoch": 6.01043701171875e-05,
      "step": 63024,
      "training_step_time": 8.565672159194946
    },
    {
      "epoch": 6.0105323791503905e-05,
      "model_forward_time": 1.6986138820648193,
      "step": 63025
    },
    {
      "epoch": 6.0105323791503905e-05,
      "step": 63025,
      "training_step_time": 8.28081464767456
    },
    {
      "epoch": 6.0106277465820316e-05,
      "model_forward_time": 1.6956787109375,
      "step": 63026
    },
    {
      "epoch": 6.0106277465820316e-05,
      "step": 63026,
      "training_step_time": 8.184899806976318
    },
    {
      "epoch": 6.010723114013672e-05,
      "model_forward_time": 1.697873592376709,
      "step": 63027
    },
    {
      "epoch": 6.010723114013672e-05,
      "step": 63027,
      "training_step_time": 8.675553321838379
    },
    {
      "epoch": 6.0108184814453124e-05,
      "model_forward_time": 1.7031364440917969,
      "step": 63028
    },
    {
      "epoch": 6.0108184814453124e-05,
      "step": 63028,
      "training_step_time": 7.748389720916748
    },
    {
      "epoch": 6.0109138488769535e-05,
      "model_forward_time": 1.7109146118164062,
      "step": 63029
    },
    {
      "epoch": 6.0109138488769535e-05,
      "step": 63029,
      "training_step_time": 7.756678104400635
    },
    {
      "epoch": 6.011009216308594e-05,
      "grad_norm": 0.1692650467157364,
      "learning_rate": 3.2940260206876197e-06,
      "loss": 0.0542,
      "step": 63030
    },
    {
      "epoch": 6.011009216308594e-05,
      "model_forward_time": 1.7045671939849854,
      "step": 63030
    },
    {
      "dynamics_loss_avg": 0.06629493478685618,
      "epoch": 6.011009216308594e-05,
      "step": 63030
    },
    {
      "action_loss_avg": 0.0022064491698984058,
      "epoch": 6.011009216308594e-05,
      "step": 63030
    },
    {
      "epoch": 6.011009216308594e-05,
      "step": 63030,
      "training_step_time": 7.715576887130737
    },
    {
      "epoch": 6.011104583740234e-05,
      "model_forward_time": 1.7139697074890137,
      "step": 63031
    },
    {
      "epoch": 6.011104583740234e-05,
      "step": 63031,
      "training_step_time": 8.346841812133789
    },
    {
      "epoch": 6.011199951171875e-05,
      "model_forward_time": 2.2342584133148193,
      "step": 63032
    },
    {
      "epoch": 6.011199951171875e-05,
      "step": 63032,
      "training_step_time": 8.995479583740234
    },
    {
      "epoch": 6.011295318603516e-05,
      "model_forward_time": 1.7538061141967773,
      "step": 63033
    },
    {
      "epoch": 6.011295318603516e-05,
      "step": 63033,
      "training_step_time": 8.73505973815918
    },
    {
      "epoch": 6.011390686035156e-05,
      "model_forward_time": 1.7008686065673828,
      "step": 63034
    },
    {
      "epoch": 6.011390686035156e-05,
      "step": 63034,
      "training_step_time": 8.635813236236572
    },
    {
      "epoch": 6.0114860534667965e-05,
      "model_forward_time": 1.7051970958709717,
      "step": 63035
    },
    {
      "epoch": 6.0114860534667965e-05,
      "step": 63035,
      "training_step_time": 7.560989141464233
    },
    {
      "epoch": 6.0115814208984376e-05,
      "model_forward_time": 1.7049217224121094,
      "step": 63036
    },
    {
      "epoch": 6.0115814208984376e-05,
      "step": 63036,
      "training_step_time": 8.475741147994995
    },
    {
      "epoch": 6.011676788330078e-05,
      "model_forward_time": 1.7145066261291504,
      "step": 63037
    },
    {
      "epoch": 6.011676788330078e-05,
      "step": 63037,
      "training_step_time": 7.781139373779297
    },
    {
      "epoch": 6.0117721557617184e-05,
      "model_forward_time": 1.7153689861297607,
      "step": 63038
    },
    {
      "epoch": 6.0117721557617184e-05,
      "step": 63038,
      "training_step_time": 7.518698692321777
    },
    {
      "epoch": 6.0118675231933595e-05,
      "model_forward_time": 1.7087047100067139,
      "step": 63039
    },
    {
      "epoch": 6.0118675231933595e-05,
      "step": 63039,
      "training_step_time": 8.508519172668457
    },
    {
      "epoch": 6.011962890625e-05,
      "grad_norm": 0.15369883179664612,
      "learning_rate": 3.2924718645839583e-06,
      "loss": 0.0548,
      "step": 63040
    },
    {
      "epoch": 6.011962890625e-05,
      "model_forward_time": 1.7060000896453857,
      "step": 63040
    },
    {
      "dynamics_loss_avg": 0.07345395274460316,
      "epoch": 6.011962890625e-05,
      "step": 63040
    },
    {
      "action_loss_avg": 0.0021253097918815913,
      "epoch": 6.011962890625e-05,
      "step": 63040
    },
    {
      "epoch": 6.011962890625e-05,
      "step": 63040,
      "training_step_time": 7.7514564990997314
    },
    {
      "epoch": 6.01205825805664e-05,
      "model_forward_time": 2.2519922256469727,
      "step": 63041
    },
    {
      "epoch": 6.01205825805664e-05,
      "step": 63041,
      "training_step_time": 8.817200422286987
    },
    {
      "epoch": 6.0121536254882814e-05,
      "model_forward_time": 1.7432582378387451,
      "step": 63042
    },
    {
      "epoch": 6.0121536254882814e-05,
      "step": 63042,
      "training_step_time": 9.313245058059692
    },
    {
      "epoch": 6.012248992919922e-05,
      "model_forward_time": 1.3404450416564941,
      "step": 63043
    },
    {
      "epoch": 6.012248992919922e-05,
      "step": 63043,
      "training_step_time": 9.627933263778687
    },
    {
      "epoch": 6.012344360351562e-05,
      "model_forward_time": 1.7118782997131348,
      "step": 63044
    },
    {
      "epoch": 6.012344360351562e-05,
      "step": 63044,
      "training_step_time": 6.782831430435181
    },
    {
      "epoch": 6.012439727783203e-05,
      "model_forward_time": 1.718540906906128,
      "step": 63045
    },
    {
      "epoch": 6.012439727783203e-05,
      "step": 63045,
      "training_step_time": 6.839378595352173
    },
    {
      "epoch": 6.012535095214844e-05,
      "model_forward_time": 1.7089509963989258,
      "step": 63046
    },
    {
      "epoch": 6.012535095214844e-05,
      "step": 63046,
      "training_step_time": 6.779959201812744
    },
    {
      "epoch": 6.012630462646484e-05,
      "model_forward_time": 1.7191100120544434,
      "step": 63047
    },
    {
      "epoch": 6.012630462646484e-05,
      "step": 63047,
      "training_step_time": 6.746779918670654
    },
    {
      "epoch": 6.012725830078125e-05,
      "model_forward_time": 1.7065656185150146,
      "step": 63048
    },
    {
      "epoch": 6.012725830078125e-05,
      "step": 63048,
      "training_step_time": 6.848137140274048
    },
    {
      "epoch": 6.0128211975097656e-05,
      "model_forward_time": 1.7118346691131592,
      "step": 63049
    },
    {
      "epoch": 6.0128211975097656e-05,
      "step": 63049,
      "training_step_time": 6.7920496463775635
    },
    {
      "epoch": 6.012916564941406e-05,
      "grad_norm": 0.10142911225557327,
      "learning_rate": 3.2909178952130037e-06,
      "loss": 0.0577,
      "step": 63050
    },
    {
      "epoch": 6.012916564941406e-05,
      "model_forward_time": 1.7160584926605225,
      "step": 63050
    },
    {
      "dynamics_loss_avg": 0.05772150158882141,
      "epoch": 6.012916564941406e-05,
      "step": 63050
    },
    {
      "action_loss_avg": 0.0010722815641202033,
      "epoch": 6.012916564941406e-05,
      "step": 63050
    },
    {
      "epoch": 6.012916564941406e-05,
      "step": 63050,
      "training_step_time": 6.785689830780029
    },
    {
      "epoch": 6.013011932373047e-05,
      "model_forward_time": 1.7175168991088867,
      "step": 63051
    },
    {
      "epoch": 6.013011932373047e-05,
      "step": 63051,
      "training_step_time": 6.811445713043213
    },
    {
      "epoch": 6.0131072998046875e-05,
      "model_forward_time": 2.2688705921173096,
      "step": 63052
    },
    {
      "epoch": 6.0131072998046875e-05,
      "step": 63052,
      "training_step_time": 7.601792573928833
    },
    {
      "epoch": 6.013202667236328e-05,
      "model_forward_time": 1.7203831672668457,
      "step": 63053
    },
    {
      "epoch": 6.013202667236328e-05,
      "step": 63053,
      "training_step_time": 7.819178104400635
    },
    {
      "epoch": 6.013298034667969e-05,
      "model_forward_time": 1.7008333206176758,
      "step": 63054
    },
    {
      "epoch": 6.013298034667969e-05,
      "step": 63054,
      "training_step_time": 6.739550590515137
    },
    {
      "epoch": 6.0133934020996093e-05,
      "model_forward_time": 1.70505952835083,
      "step": 63055
    },
    {
      "epoch": 6.0133934020996093e-05,
      "step": 63055,
      "training_step_time": 6.823728322982788
    },
    {
      "epoch": 6.01348876953125e-05,
      "model_forward_time": 1.7137658596038818,
      "step": 63056
    },
    {
      "epoch": 6.01348876953125e-05,
      "step": 63056,
      "training_step_time": 6.85930871963501
    },
    {
      "epoch": 6.013584136962891e-05,
      "model_forward_time": 1.7091763019561768,
      "step": 63057
    },
    {
      "epoch": 6.013584136962891e-05,
      "step": 63057,
      "training_step_time": 6.844872713088989
    },
    {
      "epoch": 6.013679504394531e-05,
      "model_forward_time": 1.7088282108306885,
      "step": 63058
    },
    {
      "epoch": 6.013679504394531e-05,
      "step": 63058,
      "training_step_time": 6.884446144104004
    },
    {
      "epoch": 6.0137748718261716e-05,
      "model_forward_time": 1.7089648246765137,
      "step": 63059
    },
    {
      "epoch": 6.0137748718261716e-05,
      "step": 63059,
      "training_step_time": 6.866539478302002
    },
    {
      "epoch": 6.013870239257813e-05,
      "grad_norm": 0.11562447249889374,
      "learning_rate": 3.2893641127447013e-06,
      "loss": 0.0577,
      "step": 63060
    },
    {
      "epoch": 6.013870239257813e-05,
      "model_forward_time": 1.7182235717773438,
      "step": 63060
    },
    {
      "dynamics_loss_avg": 0.04737679967656731,
      "epoch": 6.013870239257813e-05,
      "step": 63060
    },
    {
      "action_loss_avg": 0.0014517385832732543,
      "epoch": 6.013870239257813e-05,
      "step": 63060
    },
    {
      "epoch": 6.013870239257813e-05,
      "step": 63060,
      "training_step_time": 6.816898822784424
    },
    {
      "epoch": 6.013965606689453e-05,
      "model_forward_time": 2.3032219409942627,
      "step": 63061
    },
    {
      "epoch": 6.013965606689453e-05,
      "step": 63061,
      "training_step_time": 7.637538433074951
    },
    {
      "epoch": 6.0140609741210935e-05,
      "model_forward_time": 1.6901931762695312,
      "step": 63062
    },
    {
      "epoch": 6.0140609741210935e-05,
      "step": 63062,
      "training_step_time": 7.938937664031982
    },
    {
      "epoch": 6.0141563415527346e-05,
      "model_forward_time": 1.7068397998809814,
      "step": 63063
    },
    {
      "epoch": 6.0141563415527346e-05,
      "step": 63063,
      "training_step_time": 6.888866662979126
    },
    {
      "epoch": 6.014251708984375e-05,
      "model_forward_time": 1.7261030673980713,
      "step": 63064
    },
    {
      "epoch": 6.014251708984375e-05,
      "step": 63064,
      "training_step_time": 6.859639644622803
    },
    {
      "epoch": 6.0143470764160154e-05,
      "model_forward_time": 1.708977222442627,
      "step": 63065
    },
    {
      "epoch": 6.0143470764160154e-05,
      "step": 63065,
      "training_step_time": 6.903996706008911
    },
    {
      "epoch": 6.0144424438476565e-05,
      "model_forward_time": 1.7131547927856445,
      "step": 63066
    },
    {
      "epoch": 6.0144424438476565e-05,
      "step": 63066,
      "training_step_time": 6.796561241149902
    },
    {
      "epoch": 6.014537811279297e-05,
      "model_forward_time": 1.705909013748169,
      "step": 63067
    },
    {
      "epoch": 6.014537811279297e-05,
      "step": 63067,
      "training_step_time": 6.798475980758667
    },
    {
      "epoch": 6.014633178710937e-05,
      "model_forward_time": 1.68874192237854,
      "step": 63068
    },
    {
      "epoch": 6.014633178710937e-05,
      "step": 63068,
      "training_step_time": 6.863513469696045
    },
    {
      "epoch": 6.0147285461425784e-05,
      "model_forward_time": 1.7128942012786865,
      "step": 63069
    },
    {
      "epoch": 6.0147285461425784e-05,
      "step": 63069,
      "training_step_time": 6.8234031200408936
    },
    {
      "epoch": 6.014823913574219e-05,
      "grad_norm": 0.08030153810977936,
      "learning_rate": 3.287810517348967e-06,
      "loss": 0.0548,
      "step": 63070
    },
    {
      "epoch": 6.014823913574219e-05,
      "model_forward_time": 2.2974846363067627,
      "step": 63070
    },
    {
      "dynamics_loss_avg": 0.050179934687912466,
      "epoch": 6.014823913574219e-05,
      "step": 63070
    },
    {
      "action_loss_avg": 0.0022538035991601646,
      "epoch": 6.014823913574219e-05,
      "step": 63070
    },
    {
      "epoch": 6.014823913574219e-05,
      "step": 63070,
      "training_step_time": 7.62352442741394
    },
    {
      "epoch": 6.014919281005859e-05,
      "model_forward_time": 1.7199640274047852,
      "step": 63071
    },
    {
      "epoch": 6.014919281005859e-05,
      "step": 63071,
      "training_step_time": 8.928503036499023
    },
    {
      "epoch": 6.0150146484375e-05,
      "model_forward_time": 1.7220666408538818,
      "step": 63072
    },
    {
      "epoch": 6.0150146484375e-05,
      "step": 63072,
      "training_step_time": 7.34961462020874
    },
    {
      "epoch": 6.015110015869141e-05,
      "model_forward_time": 1.7244088649749756,
      "step": 63073
    },
    {
      "epoch": 6.015110015869141e-05,
      "step": 63073,
      "training_step_time": 6.89980411529541
    },
    {
      "epoch": 6.015205383300781e-05,
      "model_forward_time": 1.7099850177764893,
      "step": 63074
    },
    {
      "epoch": 6.015205383300781e-05,
      "step": 63074,
      "training_step_time": 6.8508217334747314
    },
    {
      "epoch": 6.015300750732422e-05,
      "model_forward_time": 1.7348203659057617,
      "step": 63075
    },
    {
      "epoch": 6.015300750732422e-05,
      "step": 63075,
      "training_step_time": 6.799077987670898
    },
    {
      "epoch": 6.0153961181640625e-05,
      "model_forward_time": 1.7227764129638672,
      "step": 63076
    },
    {
      "epoch": 6.0153961181640625e-05,
      "step": 63076,
      "training_step_time": 6.851571083068848
    },
    {
      "epoch": 6.015491485595703e-05,
      "model_forward_time": 1.7145140171051025,
      "step": 63077
    },
    {
      "epoch": 6.015491485595703e-05,
      "step": 63077,
      "training_step_time": 6.824950456619263
    },
    {
      "epoch": 6.015586853027344e-05,
      "model_forward_time": 1.7377665042877197,
      "step": 63078
    },
    {
      "epoch": 6.015586853027344e-05,
      "step": 63078,
      "training_step_time": 6.81840705871582
    },
    {
      "epoch": 6.0156822204589844e-05,
      "model_forward_time": 2.30499267578125,
      "step": 63079
    },
    {
      "epoch": 6.0156822204589844e-05,
      "step": 63079,
      "training_step_time": 7.636486530303955
    },
    {
      "epoch": 6.015777587890625e-05,
      "grad_norm": 0.14004434645175934,
      "learning_rate": 3.2862571091957014e-06,
      "loss": 0.0579,
      "step": 63080
    },
    {
      "epoch": 6.015777587890625e-05,
      "model_forward_time": 1.7225492000579834,
      "step": 63080
    },
    {
      "dynamics_loss_avg": 0.053229326568543914,
      "epoch": 6.015777587890625e-05,
      "step": 63080
    },
    {
      "action_loss_avg": 0.0011271440831478684,
      "epoch": 6.015777587890625e-05,
      "step": 63080
    },
    {
      "epoch": 6.015777587890625e-05,
      "step": 63080,
      "training_step_time": 9.312459230422974
    },
    {
      "epoch": 6.015872955322266e-05,
      "model_forward_time": 1.7208454608917236,
      "step": 63081
    },
    {
      "epoch": 6.015872955322266e-05,
      "step": 63081,
      "training_step_time": 7.738130331039429
    },
    {
      "epoch": 6.015968322753906e-05,
      "model_forward_time": 1.7173871994018555,
      "step": 63082
    },
    {
      "epoch": 6.015968322753906e-05,
      "step": 63082,
      "training_step_time": 6.808473110198975
    },
    {
      "epoch": 6.016063690185547e-05,
      "model_forward_time": 1.700688362121582,
      "step": 63083
    },
    {
      "epoch": 6.016063690185547e-05,
      "step": 63083,
      "training_step_time": 6.849433660507202
    },
    {
      "epoch": 6.016159057617188e-05,
      "model_forward_time": 1.7107152938842773,
      "step": 63084
    },
    {
      "epoch": 6.016159057617188e-05,
      "step": 63084,
      "training_step_time": 6.827995300292969
    },
    {
      "epoch": 6.016254425048828e-05,
      "model_forward_time": 1.7135748863220215,
      "step": 63085
    },
    {
      "epoch": 6.016254425048828e-05,
      "step": 63085,
      "training_step_time": 6.820282220840454
    },
    {
      "epoch": 6.0163497924804686e-05,
      "model_forward_time": 1.7140934467315674,
      "step": 63086
    },
    {
      "epoch": 6.0163497924804686e-05,
      "step": 63086,
      "training_step_time": 6.7729315757751465
    },
    {
      "epoch": 6.01644515991211e-05,
      "model_forward_time": 1.7097351551055908,
      "step": 63087
    },
    {
      "epoch": 6.01644515991211e-05,
      "step": 63087,
      "training_step_time": 6.761484146118164
    },
    {
      "epoch": 6.01654052734375e-05,
      "model_forward_time": 2.2844626903533936,
      "step": 63088
    },
    {
      "epoch": 6.01654052734375e-05,
      "step": 63088,
      "training_step_time": 7.238131284713745
    },
    {
      "epoch": 6.0166358947753905e-05,
      "model_forward_time": 1.7290198802947998,
      "step": 63089
    },
    {
      "epoch": 6.0166358947753905e-05,
      "step": 63089,
      "training_step_time": 9.745005130767822
    },
    {
      "epoch": 6.0167312622070316e-05,
      "grad_norm": 0.10049878060817719,
      "learning_rate": 3.284703888454781e-06,
      "loss": 0.0562,
      "step": 63090
    },
    {
      "epoch": 6.0167312622070316e-05,
      "model_forward_time": 1.7269635200500488,
      "step": 63090
    },
    {
      "dynamics_loss_avg": 0.05743412710726261,
      "epoch": 6.0167312622070316e-05,
      "step": 63090
    },
    {
      "action_loss_avg": 0.0018541557394200936,
      "epoch": 6.0167312622070316e-05,
      "step": 63090
    },
    {
      "epoch": 6.0167312622070316e-05,
      "step": 63090,
      "training_step_time": 7.840356826782227
    },
    {
      "epoch": 6.016826629638672e-05,
      "model_forward_time": 1.7185852527618408,
      "step": 63091
    },
    {
      "epoch": 6.016826629638672e-05,
      "step": 63091,
      "training_step_time": 6.816531419754028
    },
    {
      "epoch": 6.0169219970703124e-05,
      "model_forward_time": 1.73530912399292,
      "step": 63092
    },
    {
      "epoch": 6.0169219970703124e-05,
      "step": 63092,
      "training_step_time": 6.774587631225586
    },
    {
      "epoch": 6.0170173645019535e-05,
      "model_forward_time": 1.7313899993896484,
      "step": 63093
    },
    {
      "epoch": 6.0170173645019535e-05,
      "step": 63093,
      "training_step_time": 6.7336907386779785
    },
    {
      "epoch": 6.017112731933594e-05,
      "model_forward_time": 1.7215189933776855,
      "step": 63094
    },
    {
      "epoch": 6.017112731933594e-05,
      "step": 63094,
      "training_step_time": 6.783676624298096
    },
    {
      "epoch": 6.017208099365234e-05,
      "model_forward_time": 1.7184793949127197,
      "step": 63095
    },
    {
      "epoch": 6.017208099365234e-05,
      "step": 63095,
      "training_step_time": 6.767791032791138
    },
    {
      "epoch": 6.017303466796875e-05,
      "model_forward_time": 1.7196741104125977,
      "step": 63096
    },
    {
      "epoch": 6.017303466796875e-05,
      "step": 63096,
      "training_step_time": 6.732820987701416
    },
    {
      "epoch": 6.017398834228516e-05,
      "model_forward_time": 1.721186876296997,
      "step": 63097
    },
    {
      "epoch": 6.017398834228516e-05,
      "step": 63097,
      "training_step_time": 7.074692487716675
    },
    {
      "epoch": 6.017494201660156e-05,
      "model_forward_time": 2.3076791763305664,
      "step": 63098
    },
    {
      "epoch": 6.017494201660156e-05,
      "step": 63098,
      "training_step_time": 9.515460014343262
    },
    {
      "epoch": 6.0175895690917966e-05,
      "model_forward_time": 1.7204921245574951,
      "step": 63099
    },
    {
      "epoch": 6.0175895690917966e-05,
      "step": 63099,
      "training_step_time": 8.360563516616821
    },
    {
      "epoch": 6.0176849365234376e-05,
      "grad_norm": 0.09357861429452896,
      "learning_rate": 3.2831508552960646e-06,
      "loss": 0.059,
      "step": 63100
    },
    {
      "epoch": 6.0176849365234376e-05,
      "model_forward_time": 1.726947546005249,
      "step": 63100
    },
    {
      "dynamics_loss_avg": 0.07807044833898544,
      "epoch": 6.0176849365234376e-05,
      "step": 63100
    },
    {
      "action_loss_avg": 0.0018334704218432307,
      "epoch": 6.0176849365234376e-05,
      "step": 63100
    },
    {
      "epoch": 6.0176849365234376e-05,
      "step": 63100,
      "training_step_time": 7.051413059234619
    },
    {
      "epoch": 6.017780303955078e-05,
      "model_forward_time": 1.7353787422180176,
      "step": 63101
    },
    {
      "epoch": 6.017780303955078e-05,
      "step": 63101,
      "training_step_time": 6.781846046447754
    },
    {
      "epoch": 6.0178756713867184e-05,
      "model_forward_time": 1.7011823654174805,
      "step": 63102
    },
    {
      "epoch": 6.0178756713867184e-05,
      "step": 63102,
      "training_step_time": 6.809549331665039
    },
    {
      "epoch": 6.0179710388183595e-05,
      "model_forward_time": 1.7112758159637451,
      "step": 63103
    },
    {
      "epoch": 6.0179710388183595e-05,
      "step": 63103,
      "training_step_time": 6.780161380767822
    },
    {
      "epoch": 6.01806640625e-05,
      "model_forward_time": 1.716052770614624,
      "step": 63104
    },
    {
      "epoch": 6.01806640625e-05,
      "step": 63104,
      "training_step_time": 6.793778896331787
    },
    {
      "epoch": 6.01816177368164e-05,
      "model_forward_time": 1.7251441478729248,
      "step": 63105
    },
    {
      "epoch": 6.01816177368164e-05,
      "step": 63105,
      "training_step_time": 6.842457294464111
    },
    {
      "epoch": 6.0182571411132814e-05,
      "model_forward_time": 1.706305980682373,
      "step": 63106
    },
    {
      "epoch": 6.0182571411132814e-05,
      "step": 63106,
      "training_step_time": 7.074706554412842
    },
    {
      "epoch": 6.018352508544922e-05,
      "model_forward_time": 2.2723731994628906,
      "step": 63107
    },
    {
      "epoch": 6.018352508544922e-05,
      "step": 63107,
      "training_step_time": 7.572186470031738
    },
    {
      "epoch": 6.018447875976562e-05,
      "model_forward_time": 1.7041265964508057,
      "step": 63108
    },
    {
      "epoch": 6.018447875976562e-05,
      "step": 63108,
      "training_step_time": 8.782965183258057
    },
    {
      "epoch": 6.018543243408203e-05,
      "model_forward_time": 1.7061684131622314,
      "step": 63109
    },
    {
      "epoch": 6.018543243408203e-05,
      "step": 63109,
      "training_step_time": 7.6831488609313965
    },
    {
      "epoch": 6.018638610839844e-05,
      "grad_norm": 0.12384338676929474,
      "learning_rate": 3.2815980098893875e-06,
      "loss": 0.0581,
      "step": 63110
    },
    {
      "epoch": 6.018638610839844e-05,
      "model_forward_time": 1.6968412399291992,
      "step": 63110
    },
    {
      "dynamics_loss_avg": 0.04858011566102505,
      "epoch": 6.018638610839844e-05,
      "step": 63110
    },
    {
      "action_loss_avg": 0.0014364167291205376,
      "epoch": 6.018638610839844e-05,
      "step": 63110
    },
    {
      "epoch": 6.018638610839844e-05,
      "step": 63110,
      "training_step_time": 6.817302703857422
    },
    {
      "epoch": 6.018733978271484e-05,
      "model_forward_time": 1.7018516063690186,
      "step": 63111
    },
    {
      "epoch": 6.018733978271484e-05,
      "step": 63111,
      "training_step_time": 6.7729363441467285
    },
    {
      "epoch": 6.018829345703125e-05,
      "model_forward_time": 1.7049603462219238,
      "step": 63112
    },
    {
      "epoch": 6.018829345703125e-05,
      "step": 63112,
      "training_step_time": 6.796788692474365
    },
    {
      "epoch": 6.0189247131347656e-05,
      "model_forward_time": 1.7178258895874023,
      "step": 63113
    },
    {
      "epoch": 6.0189247131347656e-05,
      "step": 63113,
      "training_step_time": 6.848323583602905
    },
    {
      "epoch": 6.019020080566406e-05,
      "model_forward_time": 1.6994469165802002,
      "step": 63114
    },
    {
      "epoch": 6.019020080566406e-05,
      "step": 63114,
      "training_step_time": 6.773099184036255
    },
    {
      "epoch": 6.019115447998047e-05,
      "model_forward_time": 1.7145988941192627,
      "step": 63115
    },
    {
      "epoch": 6.019115447998047e-05,
      "step": 63115,
      "training_step_time": 6.797613859176636
    },
    {
      "epoch": 6.0192108154296875e-05,
      "model_forward_time": 2.2835192680358887,
      "step": 63116
    },
    {
      "epoch": 6.0192108154296875e-05,
      "step": 63116,
      "training_step_time": 7.634593486785889
    },
    {
      "epoch": 6.019306182861328e-05,
      "model_forward_time": 1.704817295074463,
      "step": 63117
    },
    {
      "epoch": 6.019306182861328e-05,
      "step": 63117,
      "training_step_time": 11.163799524307251
    },
    {
      "epoch": 6.019401550292969e-05,
      "model_forward_time": 1.7166643142700195,
      "step": 63118
    },
    {
      "epoch": 6.019401550292969e-05,
      "step": 63118,
      "training_step_time": 7.745049476623535
    },
    {
      "epoch": 6.0194969177246094e-05,
      "model_forward_time": 1.715522289276123,
      "step": 63119
    },
    {
      "epoch": 6.0194969177246094e-05,
      "step": 63119,
      "training_step_time": 6.735820293426514
    },
    {
      "epoch": 6.01959228515625e-05,
      "grad_norm": 0.0856810063123703,
      "learning_rate": 3.28004535240457e-06,
      "loss": 0.0608,
      "step": 63120
    },
    {
      "epoch": 6.01959228515625e-05,
      "model_forward_time": 1.7126948833465576,
      "step": 63120
    },
    {
      "dynamics_loss_avg": 0.06241302080452442,
      "epoch": 6.01959228515625e-05,
      "step": 63120
    },
    {
      "action_loss_avg": 0.0018867621081881225,
      "epoch": 6.01959228515625e-05,
      "step": 63120
    },
    {
      "epoch": 6.01959228515625e-05,
      "step": 63120,
      "training_step_time": 6.760015249252319
    },
    {
      "epoch": 6.019687652587891e-05,
      "model_forward_time": 1.7154736518859863,
      "step": 63121
    },
    {
      "epoch": 6.019687652587891e-05,
      "step": 63121,
      "training_step_time": 6.8351523876190186
    },
    {
      "epoch": 6.019783020019531e-05,
      "model_forward_time": 1.7060701847076416,
      "step": 63122
    },
    {
      "epoch": 6.019783020019531e-05,
      "step": 63122,
      "training_step_time": 6.741142511367798
    },
    {
      "epoch": 6.0198783874511716e-05,
      "model_forward_time": 1.6985297203063965,
      "step": 63123
    },
    {
      "epoch": 6.0198783874511716e-05,
      "step": 63123,
      "training_step_time": 6.736588954925537
    },
    {
      "epoch": 6.019973754882813e-05,
      "model_forward_time": 1.345015525817871,
      "step": 63124
    },
    {
      "epoch": 6.019973754882813e-05,
      "step": 63124,
      "training_step_time": 6.773611307144165
    },
    {
      "epoch": 6.020069122314453e-05,
      "model_forward_time": 2.300990581512451,
      "step": 63125
    },
    {
      "epoch": 6.020069122314453e-05,
      "step": 63125,
      "training_step_time": 7.591866970062256
    },
    {
      "epoch": 6.0201644897460935e-05,
      "model_forward_time": 1.7127854824066162,
      "step": 63126
    },
    {
      "epoch": 6.0201644897460935e-05,
      "step": 63126,
      "training_step_time": 11.398282289505005
    },
    {
      "epoch": 6.0202598571777346e-05,
      "model_forward_time": 1.7122595310211182,
      "step": 63127
    },
    {
      "epoch": 6.0202598571777346e-05,
      "step": 63127,
      "training_step_time": 7.816650867462158
    },
    {
      "epoch": 6.020355224609375e-05,
      "model_forward_time": 1.7128756046295166,
      "step": 63128
    },
    {
      "epoch": 6.020355224609375e-05,
      "step": 63128,
      "training_step_time": 7.056307315826416
    },
    {
      "epoch": 6.0204505920410154e-05,
      "model_forward_time": 1.717564344406128,
      "step": 63129
    },
    {
      "epoch": 6.0204505920410154e-05,
      "step": 63129,
      "training_step_time": 6.843863248825073
    },
    {
      "epoch": 6.0205459594726565e-05,
      "grad_norm": 0.0888819545507431,
      "learning_rate": 3.278492883011407e-06,
      "loss": 0.0577,
      "step": 63130
    },
    {
      "epoch": 6.0205459594726565e-05,
      "model_forward_time": 1.7160704135894775,
      "step": 63130
    },
    {
      "dynamics_loss_avg": 0.059150451049208644,
      "epoch": 6.0205459594726565e-05,
      "step": 63130
    },
    {
      "action_loss_avg": 0.00137642179033719,
      "epoch": 6.0205459594726565e-05,
      "step": 63130
    },
    {
      "epoch": 6.0205459594726565e-05,
      "step": 63130,
      "training_step_time": 6.836308479309082
    },
    {
      "epoch": 6.020641326904297e-05,
      "model_forward_time": 1.7151763439178467,
      "step": 63131
    },
    {
      "epoch": 6.020641326904297e-05,
      "step": 63131,
      "training_step_time": 6.764143705368042
    },
    {
      "epoch": 6.020736694335937e-05,
      "model_forward_time": 1.7013885974884033,
      "step": 63132
    },
    {
      "epoch": 6.020736694335937e-05,
      "step": 63132,
      "training_step_time": 6.724287509918213
    },
    {
      "epoch": 6.0208320617675784e-05,
      "model_forward_time": 1.7007100582122803,
      "step": 63133
    },
    {
      "epoch": 6.0208320617675784e-05,
      "step": 63133,
      "training_step_time": 6.748575687408447
    },
    {
      "epoch": 6.020927429199219e-05,
      "model_forward_time": 2.2661476135253906,
      "step": 63134
    },
    {
      "epoch": 6.020927429199219e-05,
      "step": 63134,
      "training_step_time": 7.144045352935791
    },
    {
      "epoch": 6.021022796630859e-05,
      "model_forward_time": 1.7112054824829102,
      "step": 63135
    },
    {
      "epoch": 6.021022796630859e-05,
      "step": 63135,
      "training_step_time": 10.507535219192505
    },
    {
      "epoch": 6.0211181640625e-05,
      "model_forward_time": 1.7011651992797852,
      "step": 63136
    },
    {
      "epoch": 6.0211181640625e-05,
      "step": 63136,
      "training_step_time": 8.785674810409546
    },
    {
      "epoch": 6.021213531494141e-05,
      "model_forward_time": 1.7106831073760986,
      "step": 63137
    },
    {
      "epoch": 6.021213531494141e-05,
      "step": 63137,
      "training_step_time": 7.14444375038147
    },
    {
      "epoch": 6.021308898925781e-05,
      "model_forward_time": 1.7047765254974365,
      "step": 63138
    },
    {
      "epoch": 6.021308898925781e-05,
      "step": 63138,
      "training_step_time": 6.753634214401245
    },
    {
      "epoch": 6.021404266357422e-05,
      "model_forward_time": 1.6991946697235107,
      "step": 63139
    },
    {
      "epoch": 6.021404266357422e-05,
      "step": 63139,
      "training_step_time": 6.739161014556885
    },
    {
      "epoch": 6.0214996337890626e-05,
      "grad_norm": 0.1131039410829544,
      "learning_rate": 3.2769406018796748e-06,
      "loss": 0.0595,
      "step": 63140
    },
    {
      "epoch": 6.0214996337890626e-05,
      "model_forward_time": 1.704103946685791,
      "step": 63140
    },
    {
      "dynamics_loss_avg": 0.057539939507842065,
      "epoch": 6.0214996337890626e-05,
      "step": 63140
    },
    {
      "action_loss_avg": 0.001173643025686033,
      "epoch": 6.0214996337890626e-05,
      "step": 63140
    },
    {
      "epoch": 6.0214996337890626e-05,
      "step": 63140,
      "training_step_time": 6.767843961715698
    },
    {
      "epoch": 6.021595001220703e-05,
      "model_forward_time": 1.7101309299468994,
      "step": 63141
    },
    {
      "epoch": 6.021595001220703e-05,
      "step": 63141,
      "training_step_time": 6.750621795654297
    },
    {
      "epoch": 6.021690368652344e-05,
      "model_forward_time": 1.728508710861206,
      "step": 63142
    },
    {
      "epoch": 6.021690368652344e-05,
      "step": 63142,
      "training_step_time": 6.768096685409546
    },
    {
      "epoch": 6.0217857360839844e-05,
      "model_forward_time": 1.7026443481445312,
      "step": 63143
    },
    {
      "epoch": 6.0217857360839844e-05,
      "step": 63143,
      "training_step_time": 7.740645408630371
    },
    {
      "epoch": 6.021881103515625e-05,
      "model_forward_time": 2.255103588104248,
      "step": 63144
    },
    {
      "epoch": 6.021881103515625e-05,
      "step": 63144,
      "training_step_time": 10.648943901062012
    },
    {
      "epoch": 6.021976470947266e-05,
      "model_forward_time": 1.7008459568023682,
      "step": 63145
    },
    {
      "epoch": 6.021976470947266e-05,
      "step": 63145,
      "training_step_time": 8.600542545318604
    },
    {
      "epoch": 6.022071838378906e-05,
      "model_forward_time": 1.7054636478424072,
      "step": 63146
    },
    {
      "epoch": 6.022071838378906e-05,
      "step": 63146,
      "training_step_time": 7.621579885482788
    },
    {
      "epoch": 6.022167205810547e-05,
      "model_forward_time": 1.702643632888794,
      "step": 63147
    },
    {
      "epoch": 6.022167205810547e-05,
      "step": 63147,
      "training_step_time": 6.741958379745483
    },
    {
      "epoch": 6.022262573242188e-05,
      "model_forward_time": 1.700671911239624,
      "step": 63148
    },
    {
      "epoch": 6.022262573242188e-05,
      "step": 63148,
      "training_step_time": 6.733843088150024
    },
    {
      "epoch": 6.022357940673828e-05,
      "model_forward_time": 1.7121739387512207,
      "step": 63149
    },
    {
      "epoch": 6.022357940673828e-05,
      "step": 63149,
      "training_step_time": 6.773263216018677
    },
    {
      "epoch": 6.0224533081054686e-05,
      "grad_norm": 0.12809713184833527,
      "learning_rate": 3.2753885091791273e-06,
      "loss": 0.0609,
      "step": 63150
    },
    {
      "epoch": 6.0224533081054686e-05,
      "model_forward_time": 1.7090740203857422,
      "step": 63150
    },
    {
      "dynamics_loss_avg": 0.057251288555562496,
      "epoch": 6.0224533081054686e-05,
      "step": 63150
    },
    {
      "action_loss_avg": 0.002421762963058427,
      "epoch": 6.0224533081054686e-05,
      "step": 63150
    },
    {
      "epoch": 6.0224533081054686e-05,
      "step": 63150,
      "training_step_time": 6.721877574920654
    },
    {
      "epoch": 6.02254867553711e-05,
      "model_forward_time": 1.7171893119812012,
      "step": 63151
    },
    {
      "epoch": 6.02254867553711e-05,
      "step": 63151,
      "training_step_time": 6.754060983657837
    },
    {
      "epoch": 6.02264404296875e-05,
      "model_forward_time": 1.726783037185669,
      "step": 63152
    },
    {
      "epoch": 6.02264404296875e-05,
      "step": 63152,
      "training_step_time": 6.7584381103515625
    },
    {
      "epoch": 6.0227394104003905e-05,
      "model_forward_time": 2.2788264751434326,
      "step": 63153
    },
    {
      "epoch": 6.0227394104003905e-05,
      "step": 63153,
      "training_step_time": 7.542854070663452
    },
    {
      "epoch": 6.0228347778320316e-05,
      "model_forward_time": 1.7117691040039062,
      "step": 63154
    },
    {
      "epoch": 6.0228347778320316e-05,
      "step": 63154,
      "training_step_time": 9.18483304977417
    },
    {
      "epoch": 6.022930145263672e-05,
      "model_forward_time": 1.714306116104126,
      "step": 63155
    },
    {
      "epoch": 6.022930145263672e-05,
      "step": 63155,
      "training_step_time": 7.697265386581421
    },
    {
      "epoch": 6.0230255126953124e-05,
      "model_forward_time": 1.7111773490905762,
      "step": 63156
    },
    {
      "epoch": 6.0230255126953124e-05,
      "step": 63156,
      "training_step_time": 7.011351823806763
    },
    {
      "epoch": 6.023120880126953e-05,
      "model_forward_time": 1.7080397605895996,
      "step": 63157
    },
    {
      "epoch": 6.023120880126953e-05,
      "step": 63157,
      "training_step_time": 6.77161431312561
    },
    {
      "epoch": 6.023216247558594e-05,
      "model_forward_time": 1.7137038707733154,
      "step": 63158
    },
    {
      "epoch": 6.023216247558594e-05,
      "step": 63158,
      "training_step_time": 6.766965866088867
    },
    {
      "epoch": 6.023311614990234e-05,
      "model_forward_time": 1.708965539932251,
      "step": 63159
    },
    {
      "epoch": 6.023311614990234e-05,
      "step": 63159,
      "training_step_time": 6.732020139694214
    },
    {
      "epoch": 6.023406982421875e-05,
      "grad_norm": 0.14001579582691193,
      "learning_rate": 3.273836605079499e-06,
      "loss": 0.0602,
      "step": 63160
    },
    {
      "epoch": 6.023406982421875e-05,
      "model_forward_time": 1.707892894744873,
      "step": 63160
    },
    {
      "dynamics_loss_avg": 0.062317054439336064,
      "epoch": 6.023406982421875e-05,
      "step": 63160
    },
    {
      "action_loss_avg": 0.0017688938707578928,
      "epoch": 6.023406982421875e-05,
      "step": 63160
    },
    {
      "epoch": 6.023406982421875e-05,
      "step": 63160,
      "training_step_time": 6.767419099807739
    },
    {
      "epoch": 6.023502349853516e-05,
      "model_forward_time": 1.7099854946136475,
      "step": 63161
    },
    {
      "epoch": 6.023502349853516e-05,
      "step": 63161,
      "training_step_time": 6.750414848327637
    },
    {
      "epoch": 6.023597717285156e-05,
      "model_forward_time": 2.289902448654175,
      "step": 63162
    },
    {
      "epoch": 6.023597717285156e-05,
      "step": 63162,
      "training_step_time": 8.613992691040039
    },
    {
      "epoch": 6.0236930847167966e-05,
      "model_forward_time": 1.7183842658996582,
      "step": 63163
    },
    {
      "epoch": 6.0236930847167966e-05,
      "step": 63163,
      "training_step_time": 7.860299587249756
    },
    {
      "epoch": 6.0237884521484376e-05,
      "model_forward_time": 1.7095167636871338,
      "step": 63164
    },
    {
      "epoch": 6.0237884521484376e-05,
      "step": 63164,
      "training_step_time": 7.734330415725708
    },
    {
      "epoch": 6.023883819580078e-05,
      "model_forward_time": 1.7106330394744873,
      "step": 63165
    },
    {
      "epoch": 6.023883819580078e-05,
      "step": 63165,
      "training_step_time": 7.039258003234863
    },
    {
      "epoch": 6.0239791870117184e-05,
      "model_forward_time": 1.7149031162261963,
      "step": 63166
    },
    {
      "epoch": 6.0239791870117184e-05,
      "step": 63166,
      "training_step_time": 6.719170331954956
    },
    {
      "epoch": 6.0240745544433595e-05,
      "model_forward_time": 1.7244369983673096,
      "step": 63167
    },
    {
      "epoch": 6.0240745544433595e-05,
      "step": 63167,
      "training_step_time": 6.75441837310791
    },
    {
      "epoch": 6.024169921875e-05,
      "model_forward_time": 1.7116165161132812,
      "step": 63168
    },
    {
      "epoch": 6.024169921875e-05,
      "step": 63168,
      "training_step_time": 6.778981447219849
    },
    {
      "epoch": 6.02426528930664e-05,
      "model_forward_time": 1.704155683517456,
      "step": 63169
    },
    {
      "epoch": 6.02426528930664e-05,
      "step": 63169,
      "training_step_time": 6.7980122566223145
    },
    {
      "epoch": 6.0243606567382814e-05,
      "grad_norm": 0.13887396454811096,
      "learning_rate": 3.272284889750504e-06,
      "loss": 0.0602,
      "step": 63170
    },
    {
      "epoch": 6.0243606567382814e-05,
      "model_forward_time": 1.7195122241973877,
      "step": 63170
    },
    {
      "dynamics_loss_avg": 0.07440528478473425,
      "epoch": 6.0243606567382814e-05,
      "step": 63170
    },
    {
      "action_loss_avg": 0.002240051201079041,
      "epoch": 6.0243606567382814e-05,
      "step": 63170
    },
    {
      "epoch": 6.0243606567382814e-05,
      "step": 63170,
      "training_step_time": 6.796077013015747
    },
    {
      "epoch": 6.024456024169922e-05,
      "model_forward_time": 2.2952027320861816,
      "step": 63171
    },
    {
      "epoch": 6.024456024169922e-05,
      "step": 63171,
      "training_step_time": 7.622225761413574
    },
    {
      "epoch": 6.024551391601562e-05,
      "model_forward_time": 1.7168283462524414,
      "step": 63172
    },
    {
      "epoch": 6.024551391601562e-05,
      "step": 63172,
      "training_step_time": 8.224945306777954
    },
    {
      "epoch": 6.024646759033203e-05,
      "model_forward_time": 1.7142915725708008,
      "step": 63173
    },
    {
      "epoch": 6.024646759033203e-05,
      "step": 63173,
      "training_step_time": 7.850156307220459
    },
    {
      "epoch": 6.024742126464844e-05,
      "model_forward_time": 1.7126939296722412,
      "step": 63174
    },
    {
      "epoch": 6.024742126464844e-05,
      "step": 63174,
      "training_step_time": 7.221670627593994
    },
    {
      "epoch": 6.024837493896484e-05,
      "model_forward_time": 1.716559886932373,
      "step": 63175
    },
    {
      "epoch": 6.024837493896484e-05,
      "step": 63175,
      "training_step_time": 6.794941663742065
    },
    {
      "epoch": 6.024932861328125e-05,
      "model_forward_time": 1.7193565368652344,
      "step": 63176
    },
    {
      "epoch": 6.024932861328125e-05,
      "step": 63176,
      "training_step_time": 6.795794248580933
    },
    {
      "epoch": 6.0250282287597656e-05,
      "model_forward_time": 1.7149198055267334,
      "step": 63177
    },
    {
      "epoch": 6.0250282287597656e-05,
      "step": 63177,
      "training_step_time": 6.746373891830444
    },
    {
      "epoch": 6.025123596191406e-05,
      "model_forward_time": 1.7213373184204102,
      "step": 63178
    },
    {
      "epoch": 6.025123596191406e-05,
      "step": 63178,
      "training_step_time": 6.782106399536133
    },
    {
      "epoch": 6.025218963623047e-05,
      "model_forward_time": 1.7228784561157227,
      "step": 63179
    },
    {
      "epoch": 6.025218963623047e-05,
      "step": 63179,
      "training_step_time": 6.756956100463867
    },
    {
      "epoch": 6.0253143310546875e-05,
      "grad_norm": 0.10207521915435791,
      "learning_rate": 3.2707333633618354e-06,
      "loss": 0.0586,
      "step": 63180
    },
    {
      "epoch": 6.0253143310546875e-05,
      "model_forward_time": 2.316572427749634,
      "step": 63180
    },
    {
      "dynamics_loss_avg": 0.060736515745520595,
      "epoch": 6.0253143310546875e-05,
      "step": 63180
    },
    {
      "action_loss_avg": 0.002335118595510721,
      "epoch": 6.0253143310546875e-05,
      "step": 63180
    },
    {
      "epoch": 6.0253143310546875e-05,
      "step": 63180,
      "training_step_time": 7.675462961196899
    },
    {
      "epoch": 6.025409698486328e-05,
      "model_forward_time": 1.7264044284820557,
      "step": 63181
    },
    {
      "epoch": 6.025409698486328e-05,
      "step": 63181,
      "training_step_time": 9.53903865814209
    },
    {
      "epoch": 6.025505065917969e-05,
      "model_forward_time": 1.724336862564087,
      "step": 63182
    },
    {
      "epoch": 6.025505065917969e-05,
      "step": 63182,
      "training_step_time": 8.552163124084473
    },
    {
      "epoch": 6.0256004333496094e-05,
      "model_forward_time": 1.7129108905792236,
      "step": 63183
    },
    {
      "epoch": 6.0256004333496094e-05,
      "step": 63183,
      "training_step_time": 7.743778467178345
    },
    {
      "epoch": 6.02569580078125e-05,
      "model_forward_time": 1.7161846160888672,
      "step": 63184
    },
    {
      "epoch": 6.02569580078125e-05,
      "step": 63184,
      "training_step_time": 6.757824182510376
    },
    {
      "epoch": 6.025791168212891e-05,
      "model_forward_time": 1.723524570465088,
      "step": 63185
    },
    {
      "epoch": 6.025791168212891e-05,
      "step": 63185,
      "training_step_time": 6.754360675811768
    },
    {
      "epoch": 6.025886535644531e-05,
      "model_forward_time": 1.7331750392913818,
      "step": 63186
    },
    {
      "epoch": 6.025886535644531e-05,
      "step": 63186,
      "training_step_time": 6.791125059127808
    },
    {
      "epoch": 6.0259819030761716e-05,
      "model_forward_time": 1.736161470413208,
      "step": 63187
    },
    {
      "epoch": 6.0259819030761716e-05,
      "step": 63187,
      "training_step_time": 6.7961883544921875
    },
    {
      "epoch": 6.026077270507813e-05,
      "model_forward_time": 1.7185640335083008,
      "step": 63188
    },
    {
      "epoch": 6.026077270507813e-05,
      "step": 63188,
      "training_step_time": 6.765855550765991
    },
    {
      "epoch": 6.026172637939453e-05,
      "model_forward_time": 1.72357177734375,
      "step": 63189
    },
    {
      "epoch": 6.026172637939453e-05,
      "step": 63189,
      "training_step_time": 7.637840986251831
    },
    {
      "epoch": 6.0262680053710935e-05,
      "grad_norm": 0.0967974066734314,
      "learning_rate": 3.2691820260831674e-06,
      "loss": 0.0551,
      "step": 63190
    },
    {
      "epoch": 6.0262680053710935e-05,
      "model_forward_time": 2.2694311141967773,
      "step": 63190
    },
    {
      "dynamics_loss_avg": 0.05951965097337961,
      "epoch": 6.0262680053710935e-05,
      "step": 63190
    },
    {
      "action_loss_avg": 0.0018634654756169767,
      "epoch": 6.0262680053710935e-05,
      "step": 63190
    },
    {
      "epoch": 6.0262680053710935e-05,
      "step": 63190,
      "training_step_time": 11.14574384689331
    },
    {
      "epoch": 6.0263633728027346e-05,
      "model_forward_time": 1.718609094619751,
      "step": 63191
    },
    {
      "epoch": 6.0263633728027346e-05,
      "step": 63191,
      "training_step_time": 9.805500507354736
    },
    {
      "epoch": 6.026458740234375e-05,
      "model_forward_time": 1.723590612411499,
      "step": 63192
    },
    {
      "epoch": 6.026458740234375e-05,
      "step": 63192,
      "training_step_time": 7.650951623916626
    },
    {
      "epoch": 6.0265541076660154e-05,
      "model_forward_time": 1.738175392150879,
      "step": 63193
    },
    {
      "epoch": 6.0265541076660154e-05,
      "step": 63193,
      "training_step_time": 7.065757989883423
    },
    {
      "epoch": 6.0266494750976565e-05,
      "model_forward_time": 1.72597336769104,
      "step": 63194
    },
    {
      "epoch": 6.0266494750976565e-05,
      "step": 63194,
      "training_step_time": 6.824130535125732
    },
    {
      "epoch": 6.026744842529297e-05,
      "model_forward_time": 1.724475622177124,
      "step": 63195
    },
    {
      "epoch": 6.026744842529297e-05,
      "step": 63195,
      "training_step_time": 6.803971529006958
    },
    {
      "epoch": 6.026840209960937e-05,
      "model_forward_time": 1.7195875644683838,
      "step": 63196
    },
    {
      "epoch": 6.026840209960937e-05,
      "step": 63196,
      "training_step_time": 6.82314658164978
    },
    {
      "epoch": 6.0269355773925784e-05,
      "model_forward_time": 1.7213032245635986,
      "step": 63197
    },
    {
      "epoch": 6.0269355773925784e-05,
      "step": 63197,
      "training_step_time": 6.778715133666992
    },
    {
      "epoch": 6.027030944824219e-05,
      "model_forward_time": 1.7219007015228271,
      "step": 63198
    },
    {
      "epoch": 6.027030944824219e-05,
      "step": 63198,
      "training_step_time": 7.5881805419921875
    },
    {
      "epoch": 6.027126312255859e-05,
      "model_forward_time": 2.3285372257232666,
      "step": 63199
    },
    {
      "epoch": 6.027126312255859e-05,
      "step": 63199,
      "training_step_time": 9.364250898361206
    },
    {
      "epoch": 6.0272216796875e-05,
      "grad_norm": 0.10400364547967911,
      "learning_rate": 3.2676308780841513e-06,
      "loss": 0.0583,
      "step": 63200
    },
    {
      "epoch": 6.0272216796875e-05,
      "model_forward_time": 1.7098188400268555,
      "step": 63200
    },
    {
      "dynamics_loss_avg": 0.07308728117495775,
      "epoch": 6.0272216796875e-05,
      "step": 63200
    },
    {
      "action_loss_avg": 0.0024899022188037633,
      "epoch": 6.0272216796875e-05,
      "step": 63200
    },
    {
      "epoch": 6.0272216796875e-05,
      "step": 63200,
      "training_step_time": 11.03079867362976
    },
    {
      "epoch": 6.027317047119141e-05,
      "model_forward_time": 1.7186176776885986,
      "step": 63201
    },
    {
      "epoch": 6.027317047119141e-05,
      "step": 63201,
      "training_step_time": 7.800041437149048
    },
    {
      "epoch": 6.027412414550781e-05,
      "model_forward_time": 1.724470853805542,
      "step": 63202
    },
    {
      "epoch": 6.027412414550781e-05,
      "step": 63202,
      "training_step_time": 7.060191869735718
    },
    {
      "epoch": 6.027507781982422e-05,
      "model_forward_time": 1.728055715560913,
      "step": 63203
    },
    {
      "epoch": 6.027507781982422e-05,
      "step": 63203,
      "training_step_time": 6.726583480834961
    },
    {
      "epoch": 6.0276031494140626e-05,
      "model_forward_time": 1.7266395092010498,
      "step": 63204
    },
    {
      "epoch": 6.0276031494140626e-05,
      "step": 63204,
      "training_step_time": 6.777703285217285
    },
    {
      "epoch": 6.027698516845703e-05,
      "model_forward_time": 1.725944995880127,
      "step": 63205
    },
    {
      "epoch": 6.027698516845703e-05,
      "step": 63205,
      "training_step_time": 6.77332878112793
    },
    {
      "epoch": 6.027793884277344e-05,
      "model_forward_time": 1.725409746170044,
      "step": 63206
    },
    {
      "epoch": 6.027793884277344e-05,
      "step": 63206,
      "training_step_time": 6.723788738250732
    },
    {
      "epoch": 6.0278892517089844e-05,
      "model_forward_time": 1.729144811630249,
      "step": 63207
    },
    {
      "epoch": 6.0278892517089844e-05,
      "step": 63207,
      "training_step_time": 7.604273557662964
    },
    {
      "epoch": 6.027984619140625e-05,
      "model_forward_time": 2.331285238265991,
      "step": 63208
    },
    {
      "epoch": 6.027984619140625e-05,
      "step": 63208,
      "training_step_time": 8.452730894088745
    },
    {
      "epoch": 6.028079986572266e-05,
      "model_forward_time": 1.720289707183838,
      "step": 63209
    },
    {
      "epoch": 6.028079986572266e-05,
      "step": 63209,
      "training_step_time": 9.28996467590332
    },
    {
      "epoch": 6.028175354003906e-05,
      "grad_norm": 0.08961696922779083,
      "learning_rate": 3.266079919534416e-06,
      "loss": 0.0588,
      "step": 63210
    },
    {
      "epoch": 6.028175354003906e-05,
      "model_forward_time": 1.7283763885498047,
      "step": 63210
    },
    {
      "dynamics_loss_avg": 0.04506937675178051,
      "epoch": 6.028175354003906e-05,
      "step": 63210
    },
    {
      "action_loss_avg": 0.0010485849168617278,
      "epoch": 6.028175354003906e-05,
      "step": 63210
    },
    {
      "epoch": 6.028175354003906e-05,
      "step": 63210,
      "training_step_time": 7.812191486358643
    },
    {
      "epoch": 6.028270721435547e-05,
      "model_forward_time": 1.7242798805236816,
      "step": 63211
    },
    {
      "epoch": 6.028270721435547e-05,
      "step": 63211,
      "training_step_time": 7.706586122512817
    },
    {
      "epoch": 6.028366088867188e-05,
      "model_forward_time": 1.7254681587219238,
      "step": 63212
    },
    {
      "epoch": 6.028366088867188e-05,
      "step": 63212,
      "training_step_time": 6.7586610317230225
    },
    {
      "epoch": 6.028461456298828e-05,
      "model_forward_time": 0.7078671455383301,
      "step": 63213
    },
    {
      "epoch": 6.028461456298828e-05,
      "step": 63213,
      "training_step_time": 6.793787240982056
    },
    {
      "epoch": 6.0285568237304686e-05,
      "model_forward_time": 1.740166425704956,
      "step": 63214
    },
    {
      "epoch": 6.0285568237304686e-05,
      "step": 63214,
      "training_step_time": 6.795508146286011
    },
    {
      "epoch": 6.02865219116211e-05,
      "model_forward_time": 1.7463922500610352,
      "step": 63215
    },
    {
      "epoch": 6.02865219116211e-05,
      "step": 63215,
      "training_step_time": 6.8091514110565186
    },
    {
      "epoch": 6.02874755859375e-05,
      "model_forward_time": 1.7276611328125,
      "step": 63216
    },
    {
      "epoch": 6.02874755859375e-05,
      "step": 63216,
      "training_step_time": 7.649040460586548
    },
    {
      "epoch": 6.0288429260253905e-05,
      "model_forward_time": 1.7496051788330078,
      "step": 63217
    },
    {
      "epoch": 6.0288429260253905e-05,
      "step": 63217,
      "training_step_time": 8.125479698181152
    },
    {
      "epoch": 6.0289382934570316e-05,
      "model_forward_time": 1.7266871929168701,
      "step": 63218
    },
    {
      "epoch": 6.0289382934570316e-05,
      "step": 63218,
      "training_step_time": 7.74370002746582
    },
    {
      "epoch": 6.029033660888672e-05,
      "model_forward_time": 1.7238819599151611,
      "step": 63219
    },
    {
      "epoch": 6.029033660888672e-05,
      "step": 63219,
      "training_step_time": 7.738544464111328
    },
    {
      "epoch": 6.0291290283203124e-05,
      "grad_norm": 0.0968865379691124,
      "learning_rate": 3.2645291506035753e-06,
      "loss": 0.06,
      "step": 63220
    },
    {
      "epoch": 6.0291290283203124e-05,
      "model_forward_time": 1.717292308807373,
      "step": 63220
    },
    {
      "dynamics_loss_avg": 0.07792858202010393,
      "epoch": 6.0291290283203124e-05,
      "step": 63220
    },
    {
      "action_loss_avg": 0.0024287266103783624,
      "epoch": 6.0291290283203124e-05,
      "step": 63220
    },
    {
      "epoch": 6.0291290283203124e-05,
      "step": 63220,
      "training_step_time": 7.71325945854187
    },
    {
      "epoch": 6.029224395751953e-05,
      "model_forward_time": 1.7242505550384521,
      "step": 63221
    },
    {
      "epoch": 6.029224395751953e-05,
      "step": 63221,
      "training_step_time": 6.803699254989624
    },
    {
      "epoch": 6.029319763183594e-05,
      "model_forward_time": 1.7256073951721191,
      "step": 63222
    },
    {
      "epoch": 6.029319763183594e-05,
      "step": 63222,
      "training_step_time": 6.71867036819458
    },
    {
      "epoch": 6.029415130615234e-05,
      "model_forward_time": 1.7246167659759521,
      "step": 63223
    },
    {
      "epoch": 6.029415130615234e-05,
      "step": 63223,
      "training_step_time": 6.749026775360107
    },
    {
      "epoch": 6.029510498046875e-05,
      "model_forward_time": 1.720855951309204,
      "step": 63224
    },
    {
      "epoch": 6.029510498046875e-05,
      "step": 63224,
      "training_step_time": 6.813405752182007
    },
    {
      "epoch": 6.029605865478516e-05,
      "model_forward_time": 1.7254362106323242,
      "step": 63225
    },
    {
      "epoch": 6.029605865478516e-05,
      "step": 63225,
      "training_step_time": 7.625683069229126
    },
    {
      "epoch": 6.029701232910156e-05,
      "model_forward_time": 1.7322301864624023,
      "step": 63226
    },
    {
      "epoch": 6.029701232910156e-05,
      "step": 63226,
      "training_step_time": 7.618985414505005
    },
    {
      "epoch": 6.0297966003417966e-05,
      "model_forward_time": 2.2610344886779785,
      "step": 63227
    },
    {
      "epoch": 6.0297966003417966e-05,
      "step": 63227,
      "training_step_time": 7.642955780029297
    },
    {
      "epoch": 6.0298919677734376e-05,
      "model_forward_time": 1.7221393585205078,
      "step": 63228
    },
    {
      "epoch": 6.0298919677734376e-05,
      "step": 63228,
      "training_step_time": 10.286443948745728
    },
    {
      "epoch": 6.029987335205078e-05,
      "model_forward_time": 1.722965955734253,
      "step": 63229
    },
    {
      "epoch": 6.029987335205078e-05,
      "step": 63229,
      "training_step_time": 8.26533555984497
    },
    {
      "epoch": 6.0300827026367185e-05,
      "grad_norm": 0.16049937903881073,
      "learning_rate": 3.2629785714612173e-06,
      "loss": 0.0541,
      "step": 63230
    },
    {
      "epoch": 6.0300827026367185e-05,
      "model_forward_time": 1.7280373573303223,
      "step": 63230
    },
    {
      "dynamics_loss_avg": 0.05816262988373637,
      "epoch": 6.0300827026367185e-05,
      "step": 63230
    },
    {
      "action_loss_avg": 0.0013849342241883278,
      "epoch": 6.0300827026367185e-05,
      "step": 63230
    },
    {
      "epoch": 6.0300827026367185e-05,
      "step": 63230,
      "training_step_time": 7.119722843170166
    },
    {
      "epoch": 6.0301780700683595e-05,
      "model_forward_time": 1.7292344570159912,
      "step": 63231
    },
    {
      "epoch": 6.0301780700683595e-05,
      "step": 63231,
      "training_step_time": 6.810626983642578
    },
    {
      "epoch": 6.0302734375e-05,
      "model_forward_time": 1.7256906032562256,
      "step": 63232
    },
    {
      "epoch": 6.0302734375e-05,
      "step": 63232,
      "training_step_time": 6.805638790130615
    },
    {
      "epoch": 6.03036880493164e-05,
      "model_forward_time": 1.7283239364624023,
      "step": 63233
    },
    {
      "epoch": 6.03036880493164e-05,
      "step": 63233,
      "training_step_time": 6.827919244766235
    },
    {
      "epoch": 6.0304641723632814e-05,
      "model_forward_time": 1.7263505458831787,
      "step": 63234
    },
    {
      "epoch": 6.0304641723632814e-05,
      "step": 63234,
      "training_step_time": 7.786421060562134
    },
    {
      "epoch": 6.030559539794922e-05,
      "model_forward_time": 1.733893871307373,
      "step": 63235
    },
    {
      "epoch": 6.030559539794922e-05,
      "step": 63235,
      "training_step_time": 7.630010604858398
    },
    {
      "epoch": 6.030654907226562e-05,
      "model_forward_time": 2.338820695877075,
      "step": 63236
    },
    {
      "epoch": 6.030654907226562e-05,
      "step": 63236,
      "training_step_time": 8.456174850463867
    },
    {
      "epoch": 6.030750274658203e-05,
      "model_forward_time": 1.7212932109832764,
      "step": 63237
    },
    {
      "epoch": 6.030750274658203e-05,
      "step": 63237,
      "training_step_time": 7.775274038314819
    },
    {
      "epoch": 6.030845642089844e-05,
      "model_forward_time": 1.722656011581421,
      "step": 63238
    },
    {
      "epoch": 6.030845642089844e-05,
      "step": 63238,
      "training_step_time": 7.879133462905884
    },
    {
      "epoch": 6.030941009521484e-05,
      "model_forward_time": 1.7298164367675781,
      "step": 63239
    },
    {
      "epoch": 6.030941009521484e-05,
      "step": 63239,
      "training_step_time": 7.156131029129028
    },
    {
      "epoch": 6.031036376953125e-05,
      "grad_norm": 0.10872577875852585,
      "learning_rate": 3.26142818227691e-06,
      "loss": 0.0523,
      "step": 63240
    },
    {
      "epoch": 6.031036376953125e-05,
      "model_forward_time": 1.7270545959472656,
      "step": 63240
    },
    {
      "dynamics_loss_avg": 0.0632925933226943,
      "epoch": 6.031036376953125e-05,
      "step": 63240
    },
    {
      "action_loss_avg": 0.0016636098094750195,
      "epoch": 6.031036376953125e-05,
      "step": 63240
    },
    {
      "epoch": 6.031036376953125e-05,
      "step": 63240,
      "training_step_time": 6.837468862533569
    },
    {
      "epoch": 6.0311317443847656e-05,
      "model_forward_time": 1.722062110900879,
      "step": 63241
    },
    {
      "epoch": 6.0311317443847656e-05,
      "step": 63241,
      "training_step_time": 6.76276969909668
    },
    {
      "epoch": 6.031227111816406e-05,
      "model_forward_time": 1.7332258224487305,
      "step": 63242
    },
    {
      "epoch": 6.031227111816406e-05,
      "step": 63242,
      "training_step_time": 6.763340473175049
    },
    {
      "epoch": 6.031322479248047e-05,
      "model_forward_time": 1.7240140438079834,
      "step": 63243
    },
    {
      "epoch": 6.031322479248047e-05,
      "step": 63243,
      "training_step_time": 7.707618713378906
    },
    {
      "epoch": 6.0314178466796875e-05,
      "model_forward_time": 1.725778579711914,
      "step": 63244
    },
    {
      "epoch": 6.0314178466796875e-05,
      "step": 63244,
      "training_step_time": 7.580732107162476
    },
    {
      "epoch": 6.031513214111328e-05,
      "model_forward_time": 2.348306894302368,
      "step": 63245
    },
    {
      "epoch": 6.031513214111328e-05,
      "step": 63245,
      "training_step_time": 8.320108652114868
    },
    {
      "epoch": 6.031608581542969e-05,
      "model_forward_time": 1.7294044494628906,
      "step": 63246
    },
    {
      "epoch": 6.031608581542969e-05,
      "step": 63246,
      "training_step_time": 11.027771472930908
    },
    {
      "epoch": 6.0317039489746094e-05,
      "model_forward_time": 1.7941093444824219,
      "step": 63247
    },
    {
      "epoch": 6.0317039489746094e-05,
      "step": 63247,
      "training_step_time": 7.805452346801758
    },
    {
      "epoch": 6.03179931640625e-05,
      "model_forward_time": 1.8342692852020264,
      "step": 63248
    },
    {
      "epoch": 6.03179931640625e-05,
      "step": 63248,
      "training_step_time": 7.524681568145752
    },
    {
      "epoch": 6.031894683837891e-05,
      "model_forward_time": 1.8338377475738525,
      "step": 63249
    },
    {
      "epoch": 6.031894683837891e-05,
      "step": 63249,
      "training_step_time": 6.748570680618286
    },
    {
      "epoch": 6.031990051269531e-05,
      "grad_norm": 0.12354999035596848,
      "learning_rate": 3.2598779832202033e-06,
      "loss": 0.0533,
      "step": 63250
    },
    {
      "epoch": 6.031990051269531e-05,
      "model_forward_time": 1.8328206539154053,
      "step": 63250
    },
    {
      "dynamics_loss_avg": 0.056272910349071024,
      "epoch": 6.031990051269531e-05,
      "step": 63250
    },
    {
      "action_loss_avg": 0.0016129233175888658,
      "epoch": 6.031990051269531e-05,
      "step": 63250
    },
    {
      "epoch": 6.031990051269531e-05,
      "step": 63250,
      "training_step_time": 6.807945251464844
    },
    {
      "epoch": 6.0320854187011717e-05,
      "model_forward_time": 1.839953899383545,
      "step": 63251
    },
    {
      "epoch": 6.0320854187011717e-05,
      "step": 63251,
      "training_step_time": 6.7246479988098145
    },
    {
      "epoch": 6.032180786132813e-05,
      "model_forward_time": 1.824575662612915,
      "step": 63252
    },
    {
      "epoch": 6.032180786132813e-05,
      "step": 63252,
      "training_step_time": 7.640344619750977
    },
    {
      "epoch": 6.032276153564453e-05,
      "model_forward_time": 1.8270649909973145,
      "step": 63253
    },
    {
      "epoch": 6.032276153564453e-05,
      "step": 63253,
      "training_step_time": 8.356245517730713
    },
    {
      "epoch": 6.0323715209960935e-05,
      "model_forward_time": 2.3984696865081787,
      "step": 63254
    },
    {
      "epoch": 6.0323715209960935e-05,
      "step": 63254,
      "training_step_time": 9.908668756484985
    },
    {
      "epoch": 6.0324668884277346e-05,
      "model_forward_time": 1.8231632709503174,
      "step": 63255
    },
    {
      "epoch": 6.0324668884277346e-05,
      "step": 63255,
      "training_step_time": 9.691018342971802
    },
    {
      "epoch": 6.032562255859375e-05,
      "model_forward_time": 1.8189756870269775,
      "step": 63256
    },
    {
      "epoch": 6.032562255859375e-05,
      "step": 63256,
      "training_step_time": 9.237386465072632
    },
    {
      "epoch": 6.0326576232910154e-05,
      "model_forward_time": 1.8252801895141602,
      "step": 63257
    },
    {
      "epoch": 6.0326576232910154e-05,
      "step": 63257,
      "training_step_time": 7.6684722900390625
    },
    {
      "epoch": 6.0327529907226565e-05,
      "model_forward_time": 1.823190450668335,
      "step": 63258
    },
    {
      "epoch": 6.0327529907226565e-05,
      "step": 63258,
      "training_step_time": 6.760974645614624
    },
    {
      "epoch": 6.032848358154297e-05,
      "model_forward_time": 1.816551923751831,
      "step": 63259
    },
    {
      "epoch": 6.032848358154297e-05,
      "step": 63259,
      "training_step_time": 6.785022735595703
    },
    {
      "epoch": 6.032943725585937e-05,
      "grad_norm": 0.12843453884124756,
      "learning_rate": 3.258327974460625e-06,
      "loss": 0.0502,
      "step": 63260
    },
    {
      "epoch": 6.032943725585937e-05,
      "model_forward_time": 1.8143739700317383,
      "step": 63260
    },
    {
      "dynamics_loss_avg": 0.056971481442451476,
      "epoch": 6.032943725585937e-05,
      "step": 63260
    },
    {
      "action_loss_avg": 0.0016185903310542925,
      "epoch": 6.032943725585937e-05,
      "step": 63260
    },
    {
      "epoch": 6.032943725585937e-05,
      "step": 63260,
      "training_step_time": 6.8320839405059814
    },
    {
      "epoch": 6.0330390930175784e-05,
      "model_forward_time": 1.815443515777588,
      "step": 63261
    },
    {
      "epoch": 6.0330390930175784e-05,
      "step": 63261,
      "training_step_time": 7.592798709869385
    },
    {
      "epoch": 6.033134460449219e-05,
      "model_forward_time": 1.8158888816833496,
      "step": 63262
    },
    {
      "epoch": 6.033134460449219e-05,
      "step": 63262,
      "training_step_time": 7.428004741668701
    },
    {
      "epoch": 6.033229827880859e-05,
      "model_forward_time": 2.3667151927948,
      "step": 63263
    },
    {
      "epoch": 6.033229827880859e-05,
      "step": 63263,
      "training_step_time": 8.449037313461304
    },
    {
      "epoch": 6.0333251953125e-05,
      "model_forward_time": 1.8185555934906006,
      "step": 63264
    },
    {
      "epoch": 6.0333251953125e-05,
      "step": 63264,
      "training_step_time": 8.79796314239502
    },
    {
      "epoch": 6.033420562744141e-05,
      "model_forward_time": 1.8152961730957031,
      "step": 63265
    },
    {
      "epoch": 6.033420562744141e-05,
      "step": 63265,
      "training_step_time": 9.856432437896729
    },
    {
      "epoch": 6.033515930175781e-05,
      "model_forward_time": 1.8135771751403809,
      "step": 63266
    },
    {
      "epoch": 6.033515930175781e-05,
      "step": 63266,
      "training_step_time": 7.618853807449341
    },
    {
      "epoch": 6.033611297607422e-05,
      "model_forward_time": 1.8163092136383057,
      "step": 63267
    },
    {
      "epoch": 6.033611297607422e-05,
      "step": 63267,
      "training_step_time": 7.128710746765137
    },
    {
      "epoch": 6.0337066650390626e-05,
      "model_forward_time": 1.8184266090393066,
      "step": 63268
    },
    {
      "epoch": 6.0337066650390626e-05,
      "step": 63268,
      "training_step_time": 6.795335531234741
    },
    {
      "epoch": 6.033802032470703e-05,
      "model_forward_time": 1.8139495849609375,
      "step": 63269
    },
    {
      "epoch": 6.033802032470703e-05,
      "step": 63269,
      "training_step_time": 6.833860874176025
    },
    {
      "epoch": 6.033897399902344e-05,
      "grad_norm": 0.10954473912715912,
      "learning_rate": 3.25677815616768e-06,
      "loss": 0.0537,
      "step": 63270
    },
    {
      "epoch": 6.033897399902344e-05,
      "model_forward_time": 1.8182384967803955,
      "step": 63270
    },
    {
      "dynamics_loss_avg": 0.05239821411669254,
      "epoch": 6.033897399902344e-05,
      "step": 63270
    },
    {
      "action_loss_avg": 0.0017679903336102142,
      "epoch": 6.033897399902344e-05,
      "step": 63270
    },
    {
      "epoch": 6.033897399902344e-05,
      "step": 63270,
      "training_step_time": 7.610719919204712
    },
    {
      "epoch": 6.0339927673339845e-05,
      "model_forward_time": 1.819960117340088,
      "step": 63271
    },
    {
      "epoch": 6.0339927673339845e-05,
      "step": 63271,
      "training_step_time": 7.754762411117554
    },
    {
      "epoch": 6.034088134765625e-05,
      "model_forward_time": 2.3682568073272705,
      "step": 63272
    },
    {
      "epoch": 6.034088134765625e-05,
      "step": 63272,
      "training_step_time": 8.89166784286499
    },
    {
      "epoch": 6.034183502197266e-05,
      "model_forward_time": 1.8208305835723877,
      "step": 63273
    },
    {
      "epoch": 6.034183502197266e-05,
      "step": 63273,
      "training_step_time": 8.973633050918579
    },
    {
      "epoch": 6.034278869628906e-05,
      "model_forward_time": 1.8183722496032715,
      "step": 63274
    },
    {
      "epoch": 6.034278869628906e-05,
      "step": 63274,
      "training_step_time": 9.448181629180908
    },
    {
      "epoch": 6.034374237060547e-05,
      "model_forward_time": 1.814265489578247,
      "step": 63275
    },
    {
      "epoch": 6.034374237060547e-05,
      "step": 63275,
      "training_step_time": 7.724264144897461
    },
    {
      "epoch": 6.034469604492188e-05,
      "model_forward_time": 1.8107833862304688,
      "step": 63276
    },
    {
      "epoch": 6.034469604492188e-05,
      "step": 63276,
      "training_step_time": 7.598196506500244
    },
    {
      "epoch": 6.034564971923828e-05,
      "model_forward_time": 1.822601318359375,
      "step": 63277
    },
    {
      "epoch": 6.034564971923828e-05,
      "step": 63277,
      "training_step_time": 6.77115273475647
    },
    {
      "epoch": 6.0346603393554686e-05,
      "model_forward_time": 1.8153533935546875,
      "step": 63278
    },
    {
      "epoch": 6.0346603393554686e-05,
      "step": 63278,
      "training_step_time": 6.737929105758667
    },
    {
      "epoch": 6.03475570678711e-05,
      "model_forward_time": 1.8179881572723389,
      "step": 63279
    },
    {
      "epoch": 6.03475570678711e-05,
      "step": 63279,
      "training_step_time": 7.618685007095337
    },
    {
      "epoch": 6.03485107421875e-05,
      "grad_norm": 0.09968076646327972,
      "learning_rate": 3.2552285285108565e-06,
      "loss": 0.0572,
      "step": 63280
    },
    {
      "epoch": 6.03485107421875e-05,
      "model_forward_time": 1.8208551406860352,
      "step": 63280
    },
    {
      "dynamics_loss_avg": 0.07616833485662937,
      "epoch": 6.03485107421875e-05,
      "step": 63280
    },
    {
      "action_loss_avg": 0.002333879959769547,
      "epoch": 6.03485107421875e-05,
      "step": 63280
    },
    {
      "epoch": 6.03485107421875e-05,
      "step": 63280,
      "training_step_time": 7.7547383308410645
    },
    {
      "epoch": 6.0349464416503905e-05,
      "model_forward_time": 2.3625998497009277,
      "step": 63281
    },
    {
      "epoch": 6.0349464416503905e-05,
      "step": 63281,
      "training_step_time": 9.240670919418335
    },
    {
      "epoch": 6.035041809082031e-05,
      "model_forward_time": 1.8239989280700684,
      "step": 63282
    },
    {
      "epoch": 6.035041809082031e-05,
      "step": 63282,
      "training_step_time": 9.331979990005493
    },
    {
      "epoch": 6.035137176513672e-05,
      "model_forward_time": 1.820192813873291,
      "step": 63283
    },
    {
      "epoch": 6.035137176513672e-05,
      "step": 63283,
      "training_step_time": 10.497673511505127
    },
    {
      "epoch": 6.0352325439453124e-05,
      "model_forward_time": 1.818338394165039,
      "step": 63284
    },
    {
      "epoch": 6.0352325439453124e-05,
      "step": 63284,
      "training_step_time": 9.537323713302612
    },
    {
      "epoch": 6.035327911376953e-05,
      "model_forward_time": 1.8175926208496094,
      "step": 63285
    },
    {
      "epoch": 6.035327911376953e-05,
      "step": 63285,
      "training_step_time": 7.567200422286987
    },
    {
      "epoch": 6.035423278808594e-05,
      "model_forward_time": 1.8166747093200684,
      "step": 63286
    },
    {
      "epoch": 6.035423278808594e-05,
      "step": 63286,
      "training_step_time": 6.7744460105896
    },
    {
      "epoch": 6.035518646240234e-05,
      "model_forward_time": 1.8224363327026367,
      "step": 63287
    },
    {
      "epoch": 6.035518646240234e-05,
      "step": 63287,
      "training_step_time": 6.757065773010254
    },
    {
      "epoch": 6.035614013671875e-05,
      "model_forward_time": 1.8174703121185303,
      "step": 63288
    },
    {
      "epoch": 6.035614013671875e-05,
      "step": 63288,
      "training_step_time": 7.5634565353393555
    },
    {
      "epoch": 6.035709381103516e-05,
      "model_forward_time": 1.8184003829956055,
      "step": 63289
    },
    {
      "epoch": 6.035709381103516e-05,
      "step": 63289,
      "training_step_time": 8.12068772315979
    },
    {
      "epoch": 6.035804748535156e-05,
      "grad_norm": 0.09850078076124191,
      "learning_rate": 3.2536790916596167e-06,
      "loss": 0.0548,
      "step": 63290
    },
    {
      "epoch": 6.035804748535156e-05,
      "model_forward_time": 2.3841826915740967,
      "step": 63290
    },
    {
      "dynamics_loss_avg": 0.05359354224056005,
      "epoch": 6.035804748535156e-05,
      "step": 63290
    },
    {
      "action_loss_avg": 0.0018495985365007072,
      "epoch": 6.035804748535156e-05,
      "step": 63290
    },
    {
      "epoch": 6.035804748535156e-05,
      "step": 63290,
      "training_step_time": 8.80748200416565
    },
    {
      "epoch": 6.0359001159667966e-05,
      "model_forward_time": 1.815840482711792,
      "step": 63291
    },
    {
      "epoch": 6.0359001159667966e-05,
      "step": 63291,
      "training_step_time": 10.672711610794067
    },
    {
      "epoch": 6.0359954833984377e-05,
      "model_forward_time": 1.8170273303985596,
      "step": 63292
    },
    {
      "epoch": 6.0359954833984377e-05,
      "step": 63292,
      "training_step_time": 8.264972686767578
    },
    {
      "epoch": 6.036090850830078e-05,
      "model_forward_time": 1.8263537883758545,
      "step": 63293
    },
    {
      "epoch": 6.036090850830078e-05,
      "step": 63293,
      "training_step_time": 7.801472902297974
    },
    {
      "epoch": 6.0361862182617185e-05,
      "model_forward_time": 1.8238017559051514,
      "step": 63294
    },
    {
      "epoch": 6.0361862182617185e-05,
      "step": 63294,
      "training_step_time": 7.61581015586853
    },
    {
      "epoch": 6.0362815856933595e-05,
      "model_forward_time": 1.8220465183258057,
      "step": 63295
    },
    {
      "epoch": 6.0362815856933595e-05,
      "step": 63295,
      "training_step_time": 7.05230188369751
    },
    {
      "epoch": 6.036376953125e-05,
      "model_forward_time": 1.8258998394012451,
      "step": 63296
    },
    {
      "epoch": 6.036376953125e-05,
      "step": 63296,
      "training_step_time": 6.79559063911438
    },
    {
      "epoch": 6.0364723205566403e-05,
      "model_forward_time": 1.826395034790039,
      "step": 63297
    },
    {
      "epoch": 6.0364723205566403e-05,
      "step": 63297,
      "training_step_time": 7.566769599914551
    },
    {
      "epoch": 6.0365676879882814e-05,
      "model_forward_time": 1.8299720287322998,
      "step": 63298
    },
    {
      "epoch": 6.0365676879882814e-05,
      "step": 63298,
      "training_step_time": 7.451941251754761
    },
    {
      "epoch": 6.036663055419922e-05,
      "model_forward_time": 2.3616878986358643,
      "step": 63299
    },
    {
      "epoch": 6.036663055419922e-05,
      "step": 63299,
      "training_step_time": 7.632573843002319
    },
    {
      "epoch": 6.036758422851562e-05,
      "grad_norm": 0.10256480425596237,
      "learning_rate": 3.2521298457834037e-06,
      "loss": 0.0549,
      "step": 63300
    },
    {
      "epoch": 6.036758422851562e-05,
      "model_forward_time": 1.8198223114013672,
      "step": 63300
    },
    {
      "dynamics_loss_avg": 0.04424290526658296,
      "epoch": 6.036758422851562e-05,
      "step": 63300
    },
    {
      "action_loss_avg": 0.0011473390302853658,
      "epoch": 6.036758422851562e-05,
      "step": 63300
    },
    {
      "epoch": 6.036758422851562e-05,
      "step": 63300,
      "training_step_time": 8.428589582443237
    },
    {
      "epoch": 6.036853790283203e-05,
      "model_forward_time": 1.8190736770629883,
      "step": 63301
    },
    {
      "epoch": 6.036853790283203e-05,
      "step": 63301,
      "training_step_time": 8.476195096969604
    },
    {
      "epoch": 6.036949157714844e-05,
      "model_forward_time": 1.811614751815796,
      "step": 63302
    },
    {
      "epoch": 6.036949157714844e-05,
      "step": 63302,
      "training_step_time": 7.8039679527282715
    },
    {
      "epoch": 6.037044525146484e-05,
      "model_forward_time": 1.8183939456939697,
      "step": 63303
    },
    {
      "epoch": 6.037044525146484e-05,
      "step": 63303,
      "training_step_time": 7.6737847328186035
    },
    {
      "epoch": 6.037139892578125e-05,
      "model_forward_time": 1.8103439807891846,
      "step": 63304
    },
    {
      "epoch": 6.037139892578125e-05,
      "step": 63304,
      "training_step_time": 7.0627570152282715
    },
    {
      "epoch": 6.0372352600097656e-05,
      "model_forward_time": 1.802262306213379,
      "step": 63305
    },
    {
      "epoch": 6.0372352600097656e-05,
      "step": 63305,
      "training_step_time": 6.756638288497925
    },
    {
      "epoch": 6.037330627441406e-05,
      "model_forward_time": 1.8145537376403809,
      "step": 63306
    },
    {
      "epoch": 6.037330627441406e-05,
      "step": 63306,
      "training_step_time": 7.52575421333313
    },
    {
      "epoch": 6.037425994873047e-05,
      "model_forward_time": 1.8212008476257324,
      "step": 63307
    },
    {
      "epoch": 6.037425994873047e-05,
      "step": 63307,
      "training_step_time": 6.7177135944366455
    },
    {
      "epoch": 6.0375213623046875e-05,
      "model_forward_time": 2.3654181957244873,
      "step": 63308
    },
    {
      "epoch": 6.0375213623046875e-05,
      "step": 63308,
      "training_step_time": 8.728097438812256
    },
    {
      "epoch": 6.037616729736328e-05,
      "model_forward_time": 1.8076999187469482,
      "step": 63309
    },
    {
      "epoch": 6.037616729736328e-05,
      "step": 63309,
      "training_step_time": 8.759129047393799
    },
    {
      "epoch": 6.037712097167969e-05,
      "grad_norm": 0.09726539254188538,
      "learning_rate": 3.250580791051646e-06,
      "loss": 0.0561,
      "step": 63310
    },
    {
      "epoch": 6.037712097167969e-05,
      "model_forward_time": 1.8169119358062744,
      "step": 63310
    },
    {
      "dynamics_loss_avg": 0.04936813283711672,
      "epoch": 6.037712097167969e-05,
      "step": 63310
    },
    {
      "action_loss_avg": 0.0016178080171812327,
      "epoch": 6.037712097167969e-05,
      "step": 63310
    },
    {
      "epoch": 6.037712097167969e-05,
      "step": 63310,
      "training_step_time": 8.321502208709717
    },
    {
      "epoch": 6.0378074645996094e-05,
      "model_forward_time": 1.8099093437194824,
      "step": 63311
    },
    {
      "epoch": 6.0378074645996094e-05,
      "step": 63311,
      "training_step_time": 7.808922529220581
    },
    {
      "epoch": 6.03790283203125e-05,
      "model_forward_time": 1.8178129196166992,
      "step": 63312
    },
    {
      "epoch": 6.03790283203125e-05,
      "step": 63312,
      "training_step_time": 7.714268445968628
    },
    {
      "epoch": 6.037998199462891e-05,
      "model_forward_time": 1.8205363750457764,
      "step": 63313
    },
    {
      "epoch": 6.037998199462891e-05,
      "step": 63313,
      "training_step_time": 7.529258489608765
    },
    {
      "epoch": 6.038093566894531e-05,
      "model_forward_time": 1.821376085281372,
      "step": 63314
    },
    {
      "epoch": 6.038093566894531e-05,
      "step": 63314,
      "training_step_time": 7.021129608154297
    },
    {
      "epoch": 6.0381889343261717e-05,
      "model_forward_time": 1.8207371234893799,
      "step": 63315
    },
    {
      "epoch": 6.0381889343261717e-05,
      "step": 63315,
      "training_step_time": 6.734976291656494
    },
    {
      "epoch": 6.038284301757813e-05,
      "model_forward_time": 1.8247425556182861,
      "step": 63316
    },
    {
      "epoch": 6.038284301757813e-05,
      "step": 63316,
      "training_step_time": 6.693797826766968
    },
    {
      "epoch": 6.038379669189453e-05,
      "model_forward_time": 2.3739736080169678,
      "step": 63317
    },
    {
      "epoch": 6.038379669189453e-05,
      "step": 63317,
      "training_step_time": 7.701240062713623
    },
    {
      "epoch": 6.0384750366210935e-05,
      "model_forward_time": 1.8159773349761963,
      "step": 63318
    },
    {
      "epoch": 6.0384750366210935e-05,
      "step": 63318,
      "training_step_time": 7.5998618602752686
    },
    {
      "epoch": 6.0385704040527346e-05,
      "model_forward_time": 1.8151345252990723,
      "step": 63319
    },
    {
      "epoch": 6.0385704040527346e-05,
      "step": 63319,
      "training_step_time": 9.36379337310791
    },
    {
      "epoch": 6.038665771484375e-05,
      "grad_norm": 0.08527597039937973,
      "learning_rate": 3.2490319276337416e-06,
      "loss": 0.0586,
      "step": 63320
    },
    {
      "epoch": 6.038665771484375e-05,
      "model_forward_time": 1.8112008571624756,
      "step": 63320
    },
    {
      "dynamics_loss_avg": 0.04642011746764183,
      "epoch": 6.038665771484375e-05,
      "step": 63320
    },
    {
      "action_loss_avg": 0.0014284430828411133,
      "epoch": 6.038665771484375e-05,
      "step": 63320
    },
    {
      "epoch": 6.038665771484375e-05,
      "step": 63320,
      "training_step_time": 11.572483777999878
    },
    {
      "epoch": 6.0387611389160154e-05,
      "model_forward_time": 1.8200175762176514,
      "step": 63321
    },
    {
      "epoch": 6.0387611389160154e-05,
      "step": 63321,
      "training_step_time": 9.096769571304321
    },
    {
      "epoch": 6.0388565063476565e-05,
      "model_forward_time": 1.8194091320037842,
      "step": 63322
    },
    {
      "epoch": 6.0388565063476565e-05,
      "step": 63322,
      "training_step_time": 7.374425411224365
    },
    {
      "epoch": 6.038951873779297e-05,
      "model_forward_time": 1.8166203498840332,
      "step": 63323
    },
    {
      "epoch": 6.038951873779297e-05,
      "step": 63323,
      "training_step_time": 7.075670957565308
    },
    {
      "epoch": 6.039047241210937e-05,
      "model_forward_time": 1.8170404434204102,
      "step": 63324
    },
    {
      "epoch": 6.039047241210937e-05,
      "step": 63324,
      "training_step_time": 6.786347389221191
    },
    {
      "epoch": 6.0391426086425784e-05,
      "model_forward_time": 1.8152246475219727,
      "step": 63325
    },
    {
      "epoch": 6.0391426086425784e-05,
      "step": 63325,
      "training_step_time": 6.777513742446899
    },
    {
      "epoch": 6.039237976074219e-05,
      "model_forward_time": 2.3632569313049316,
      "step": 63326
    },
    {
      "epoch": 6.039237976074219e-05,
      "step": 63326,
      "training_step_time": 7.629944801330566
    },
    {
      "epoch": 6.039333343505859e-05,
      "model_forward_time": 1.8221063613891602,
      "step": 63327
    },
    {
      "epoch": 6.039333343505859e-05,
      "step": 63327,
      "training_step_time": 8.137615442276001
    },
    {
      "epoch": 6.0394287109375e-05,
      "model_forward_time": 1.826545238494873,
      "step": 63328
    },
    {
      "epoch": 6.0394287109375e-05,
      "step": 63328,
      "training_step_time": 10.239648342132568
    },
    {
      "epoch": 6.039524078369141e-05,
      "model_forward_time": 1.8256511688232422,
      "step": 63329
    },
    {
      "epoch": 6.039524078369141e-05,
      "step": 63329,
      "training_step_time": 8.835794448852539
    },
    {
      "epoch": 6.039619445800781e-05,
      "grad_norm": 0.09932488948106766,
      "learning_rate": 3.247483255699074e-06,
      "loss": 0.0574,
      "step": 63330
    },
    {
      "epoch": 6.039619445800781e-05,
      "model_forward_time": 1.8322880268096924,
      "step": 63330
    },
    {
      "dynamics_loss_avg": 0.07275722026824952,
      "epoch": 6.039619445800781e-05,
      "step": 63330
    },
    {
      "action_loss_avg": 0.002150417375378311,
      "epoch": 6.039619445800781e-05,
      "step": 63330
    },
    {
      "epoch": 6.039619445800781e-05,
      "step": 63330,
      "training_step_time": 7.770294904708862
    },
    {
      "epoch": 6.039714813232422e-05,
      "model_forward_time": 1.8280959129333496,
      "step": 63331
    },
    {
      "epoch": 6.039714813232422e-05,
      "step": 63331,
      "training_step_time": 7.5312535762786865
    },
    {
      "epoch": 6.0398101806640626e-05,
      "model_forward_time": 1.803091287612915,
      "step": 63332
    },
    {
      "epoch": 6.0398101806640626e-05,
      "step": 63332,
      "training_step_time": 7.0536394119262695
    },
    {
      "epoch": 6.039905548095703e-05,
      "model_forward_time": 1.8254401683807373,
      "step": 63333
    },
    {
      "epoch": 6.039905548095703e-05,
      "step": 63333,
      "training_step_time": 7.0440354347229
    },
    {
      "epoch": 6.040000915527344e-05,
      "model_forward_time": 1.810457706451416,
      "step": 63334
    },
    {
      "epoch": 6.040000915527344e-05,
      "step": 63334,
      "training_step_time": 6.743010759353638
    },
    {
      "epoch": 6.0400962829589845e-05,
      "model_forward_time": 2.392946243286133,
      "step": 63335
    },
    {
      "epoch": 6.0400962829589845e-05,
      "step": 63335,
      "training_step_time": 7.604639053344727
    },
    {
      "epoch": 6.040191650390625e-05,
      "model_forward_time": 1.8289012908935547,
      "step": 63336
    },
    {
      "epoch": 6.040191650390625e-05,
      "step": 63336,
      "training_step_time": 8.922375679016113
    },
    {
      "epoch": 6.040287017822266e-05,
      "model_forward_time": 1.8239269256591797,
      "step": 63337
    },
    {
      "epoch": 6.040287017822266e-05,
      "step": 63337,
      "training_step_time": 8.297523498535156
    },
    {
      "epoch": 6.0403823852539063e-05,
      "model_forward_time": 1.828449010848999,
      "step": 63338
    },
    {
      "epoch": 6.0403823852539063e-05,
      "step": 63338,
      "training_step_time": 7.621300458908081
    },
    {
      "epoch": 6.040477752685547e-05,
      "model_forward_time": 1.8269984722137451,
      "step": 63339
    },
    {
      "epoch": 6.040477752685547e-05,
      "step": 63339,
      "training_step_time": 8.848802328109741
    },
    {
      "epoch": 6.040573120117188e-05,
      "grad_norm": 0.15020914375782013,
      "learning_rate": 3.245934775417002e-06,
      "loss": 0.0562,
      "step": 63340
    },
    {
      "epoch": 6.040573120117188e-05,
      "model_forward_time": 1.8291704654693604,
      "step": 63340
    },
    {
      "dynamics_loss_avg": 0.047095400653779505,
      "epoch": 6.040573120117188e-05,
      "step": 63340
    },
    {
      "action_loss_avg": 0.0018881717522162944,
      "epoch": 6.040573120117188e-05,
      "step": 63340
    },
    {
      "epoch": 6.040573120117188e-05,
      "step": 63340,
      "training_step_time": 7.487797021865845
    },
    {
      "epoch": 6.040668487548828e-05,
      "model_forward_time": 1.828718662261963,
      "step": 63341
    },
    {
      "epoch": 6.040668487548828e-05,
      "step": 63341,
      "training_step_time": 7.523068904876709
    },
    {
      "epoch": 6.0407638549804686e-05,
      "model_forward_time": 1.8294165134429932,
      "step": 63342
    },
    {
      "epoch": 6.0407638549804686e-05,
      "step": 63342,
      "training_step_time": 7.054387807846069
    },
    {
      "epoch": 6.04085922241211e-05,
      "model_forward_time": 1.8271501064300537,
      "step": 63343
    },
    {
      "epoch": 6.04085922241211e-05,
      "step": 63343,
      "training_step_time": 6.78643012046814
    },
    {
      "epoch": 6.04095458984375e-05,
      "model_forward_time": 2.3851425647735596,
      "step": 63344
    },
    {
      "epoch": 6.04095458984375e-05,
      "step": 63344,
      "training_step_time": 8.406602621078491
    },
    {
      "epoch": 6.0410499572753905e-05,
      "model_forward_time": 1.8255219459533691,
      "step": 63345
    },
    {
      "epoch": 6.0410499572753905e-05,
      "step": 63345,
      "training_step_time": 8.548906326293945
    },
    {
      "epoch": 6.041145324707031e-05,
      "model_forward_time": 1.8311951160430908,
      "step": 63346
    },
    {
      "epoch": 6.041145324707031e-05,
      "step": 63346,
      "training_step_time": 9.141624212265015
    },
    {
      "epoch": 6.041240692138672e-05,
      "model_forward_time": 1.8300251960754395,
      "step": 63347
    },
    {
      "epoch": 6.041240692138672e-05,
      "step": 63347,
      "training_step_time": 9.64102578163147
    },
    {
      "epoch": 6.0413360595703124e-05,
      "model_forward_time": 1.7214441299438477,
      "step": 63348
    },
    {
      "epoch": 6.0413360595703124e-05,
      "step": 63348,
      "training_step_time": 8.239696741104126
    },
    {
      "epoch": 6.041431427001953e-05,
      "model_forward_time": 1.7138383388519287,
      "step": 63349
    },
    {
      "epoch": 6.041431427001953e-05,
      "step": 63349,
      "training_step_time": 7.676448106765747
    },
    {
      "epoch": 6.041526794433594e-05,
      "grad_norm": 0.10297064483165741,
      "learning_rate": 3.2443864869568666e-06,
      "loss": 0.0582,
      "step": 63350
    },
    {
      "epoch": 6.041526794433594e-05,
      "model_forward_time": 1.726759433746338,
      "step": 63350
    },
    {
      "dynamics_loss_avg": 0.06853548306971788,
      "epoch": 6.041526794433594e-05,
      "step": 63350
    },
    {
      "action_loss_avg": 0.0026502868393436076,
      "epoch": 6.041526794433594e-05,
      "step": 63350
    },
    {
      "epoch": 6.041526794433594e-05,
      "step": 63350,
      "training_step_time": 8.073392868041992
    },
    {
      "epoch": 6.041622161865234e-05,
      "model_forward_time": 1.719726800918579,
      "step": 63351
    },
    {
      "epoch": 6.041622161865234e-05,
      "step": 63351,
      "training_step_time": 7.053684711456299
    },
    {
      "epoch": 6.041717529296875e-05,
      "model_forward_time": 1.7081325054168701,
      "step": 63352
    },
    {
      "epoch": 6.041717529296875e-05,
      "step": 63352,
      "training_step_time": 6.775081396102905
    },
    {
      "epoch": 6.041812896728516e-05,
      "model_forward_time": 2.266467571258545,
      "step": 63353
    },
    {
      "epoch": 6.041812896728516e-05,
      "step": 63353,
      "training_step_time": 7.352910280227661
    },
    {
      "epoch": 6.041908264160156e-05,
      "model_forward_time": 1.7238693237304688,
      "step": 63354
    },
    {
      "epoch": 6.041908264160156e-05,
      "step": 63354,
      "training_step_time": 7.723690509796143
    },
    {
      "epoch": 6.0420036315917966e-05,
      "model_forward_time": 1.7706029415130615,
      "step": 63355
    },
    {
      "epoch": 6.0420036315917966e-05,
      "step": 63355,
      "training_step_time": 7.584396839141846
    },
    {
      "epoch": 6.0420989990234377e-05,
      "model_forward_time": 1.7227060794830322,
      "step": 63356
    },
    {
      "epoch": 6.0420989990234377e-05,
      "step": 63356,
      "training_step_time": 8.436266899108887
    },
    {
      "epoch": 6.042194366455078e-05,
      "model_forward_time": 1.7193543910980225,
      "step": 63357
    },
    {
      "epoch": 6.042194366455078e-05,
      "step": 63357,
      "training_step_time": 9.211147546768188
    },
    {
      "epoch": 6.0422897338867185e-05,
      "model_forward_time": 1.714920997619629,
      "step": 63358
    },
    {
      "epoch": 6.0422897338867185e-05,
      "step": 63358,
      "training_step_time": 8.308779001235962
    },
    {
      "epoch": 6.0423851013183595e-05,
      "model_forward_time": 1.7279846668243408,
      "step": 63359
    },
    {
      "epoch": 6.0423851013183595e-05,
      "step": 63359,
      "training_step_time": 7.463446855545044
    },
    {
      "epoch": 6.04248046875e-05,
      "grad_norm": 0.08425074070692062,
      "learning_rate": 3.2428383904879844e-06,
      "loss": 0.0594,
      "step": 63360
    },
    {
      "epoch": 6.04248046875e-05,
      "model_forward_time": 1.7183260917663574,
      "step": 63360
    },
    {
      "dynamics_loss_avg": 0.06769299786537886,
      "epoch": 6.04248046875e-05,
      "step": 63360
    },
    {
      "action_loss_avg": 0.0016059481451520697,
      "epoch": 6.04248046875e-05,
      "step": 63360
    },
    {
      "epoch": 6.04248046875e-05,
      "step": 63360,
      "training_step_time": 7.053626775741577
    },
    {
      "epoch": 6.0425758361816403e-05,
      "model_forward_time": 1.716825008392334,
      "step": 63361
    },
    {
      "epoch": 6.0425758361816403e-05,
      "step": 63361,
      "training_step_time": 7.077566385269165
    },
    {
      "epoch": 6.0426712036132814e-05,
      "model_forward_time": 2.291928291320801,
      "step": 63362
    },
    {
      "epoch": 6.0426712036132814e-05,
      "step": 63362,
      "training_step_time": 7.342529058456421
    },
    {
      "epoch": 6.042766571044922e-05,
      "model_forward_time": 1.7177681922912598,
      "step": 63363
    },
    {
      "epoch": 6.042766571044922e-05,
      "step": 63363,
      "training_step_time": 7.635175704956055
    },
    {
      "epoch": 6.042861938476562e-05,
      "model_forward_time": 1.7514235973358154,
      "step": 63364
    },
    {
      "epoch": 6.042861938476562e-05,
      "step": 63364,
      "training_step_time": 8.101706743240356
    },
    {
      "epoch": 6.042957305908203e-05,
      "model_forward_time": 1.7211759090423584,
      "step": 63365
    },
    {
      "epoch": 6.042957305908203e-05,
      "step": 63365,
      "training_step_time": 8.093729019165039
    },
    {
      "epoch": 6.043052673339844e-05,
      "model_forward_time": 1.7291984558105469,
      "step": 63366
    },
    {
      "epoch": 6.043052673339844e-05,
      "step": 63366,
      "training_step_time": 7.667981863021851
    },
    {
      "epoch": 6.043148040771484e-05,
      "model_forward_time": 1.7185368537902832,
      "step": 63367
    },
    {
      "epoch": 6.043148040771484e-05,
      "step": 63367,
      "training_step_time": 7.677319288253784
    },
    {
      "epoch": 6.043243408203125e-05,
      "model_forward_time": 1.7164452075958252,
      "step": 63368
    },
    {
      "epoch": 6.043243408203125e-05,
      "step": 63368,
      "training_step_time": 7.430134534835815
    },
    {
      "epoch": 6.0433387756347656e-05,
      "model_forward_time": 1.715944766998291,
      "step": 63369
    },
    {
      "epoch": 6.0433387756347656e-05,
      "step": 63369,
      "training_step_time": 7.550516843795776
    },
    {
      "epoch": 6.043434143066406e-05,
      "grad_norm": 0.09863881021738052,
      "learning_rate": 3.2412904861796525e-06,
      "loss": 0.0587,
      "step": 63370
    },
    {
      "epoch": 6.043434143066406e-05,
      "model_forward_time": 1.717374324798584,
      "step": 63370
    },
    {
      "dynamics_loss_avg": 0.05502604078501463,
      "epoch": 6.043434143066406e-05,
      "step": 63370
    },
    {
      "action_loss_avg": 0.0018267232284415514,
      "epoch": 6.043434143066406e-05,
      "step": 63370
    },
    {
      "epoch": 6.043434143066406e-05,
      "step": 63370,
      "training_step_time": 7.052269220352173
    },
    {
      "epoch": 6.043529510498047e-05,
      "model_forward_time": 2.280179500579834,
      "step": 63371
    },
    {
      "epoch": 6.043529510498047e-05,
      "step": 63371,
      "training_step_time": 7.933851718902588
    },
    {
      "epoch": 6.0436248779296875e-05,
      "model_forward_time": 1.753140926361084,
      "step": 63372
    },
    {
      "epoch": 6.0436248779296875e-05,
      "step": 63372,
      "training_step_time": 7.715458154678345
    },
    {
      "epoch": 6.043720245361328e-05,
      "model_forward_time": 1.7291193008422852,
      "step": 63373
    },
    {
      "epoch": 6.043720245361328e-05,
      "step": 63373,
      "training_step_time": 8.598839282989502
    },
    {
      "epoch": 6.043815612792969e-05,
      "model_forward_time": 1.7184975147247314,
      "step": 63374
    },
    {
      "epoch": 6.043815612792969e-05,
      "step": 63374,
      "training_step_time": 9.369745016098022
    },
    {
      "epoch": 6.0439109802246094e-05,
      "model_forward_time": 1.7221601009368896,
      "step": 63375
    },
    {
      "epoch": 6.0439109802246094e-05,
      "step": 63375,
      "training_step_time": 8.431391477584839
    },
    {
      "epoch": 6.04400634765625e-05,
      "model_forward_time": 1.7229044437408447,
      "step": 63376
    },
    {
      "epoch": 6.04400634765625e-05,
      "step": 63376,
      "training_step_time": 7.695882081985474
    },
    {
      "epoch": 6.044101715087891e-05,
      "model_forward_time": 1.7231991291046143,
      "step": 63377
    },
    {
      "epoch": 6.044101715087891e-05,
      "step": 63377,
      "training_step_time": 7.416757822036743
    },
    {
      "epoch": 6.044197082519531e-05,
      "model_forward_time": 1.7247216701507568,
      "step": 63378
    },
    {
      "epoch": 6.044197082519531e-05,
      "step": 63378,
      "training_step_time": 8.077550649642944
    },
    {
      "epoch": 6.044292449951172e-05,
      "model_forward_time": 1.7048802375793457,
      "step": 63379
    },
    {
      "epoch": 6.044292449951172e-05,
      "step": 63379,
      "training_step_time": 7.059398412704468
    },
    {
      "epoch": 6.044387817382813e-05,
      "grad_norm": 0.10283565521240234,
      "learning_rate": 3.2397427742011513e-06,
      "loss": 0.0577,
      "step": 63380
    },
    {
      "epoch": 6.044387817382813e-05,
      "model_forward_time": 2.266810655593872,
      "step": 63380
    },
    {
      "dynamics_loss_avg": 0.05693661160767079,
      "epoch": 6.044387817382813e-05,
      "step": 63380
    },
    {
      "action_loss_avg": 0.0011329343658871948,
      "epoch": 6.044387817382813e-05,
      "step": 63380
    },
    {
      "epoch": 6.044387817382813e-05,
      "step": 63380,
      "training_step_time": 7.348918199539185
    },
    {
      "epoch": 6.044483184814453e-05,
      "model_forward_time": 1.7108538150787354,
      "step": 63381
    },
    {
      "epoch": 6.044483184814453e-05,
      "step": 63381,
      "training_step_time": 7.594547986984253
    },
    {
      "epoch": 6.0445785522460935e-05,
      "model_forward_time": 1.7086000442504883,
      "step": 63382
    },
    {
      "epoch": 6.0445785522460935e-05,
      "step": 63382,
      "training_step_time": 7.550820827484131
    },
    {
      "epoch": 6.0446739196777346e-05,
      "model_forward_time": 1.7662889957427979,
      "step": 63383
    },
    {
      "epoch": 6.0446739196777346e-05,
      "step": 63383,
      "training_step_time": 7.674401760101318
    },
    {
      "epoch": 6.044769287109375e-05,
      "model_forward_time": 1.7078526020050049,
      "step": 63384
    },
    {
      "epoch": 6.044769287109375e-05,
      "step": 63384,
      "training_step_time": 9.538496732711792
    },
    {
      "epoch": 6.0448646545410154e-05,
      "model_forward_time": 1.7180585861206055,
      "step": 63385
    },
    {
      "epoch": 6.0448646545410154e-05,
      "step": 63385,
      "training_step_time": 9.391796588897705
    },
    {
      "epoch": 6.0449600219726565e-05,
      "model_forward_time": 1.7120471000671387,
      "step": 63386
    },
    {
      "epoch": 6.0449600219726565e-05,
      "step": 63386,
      "training_step_time": 7.613855600357056
    },
    {
      "epoch": 6.045055389404297e-05,
      "model_forward_time": 1.7066092491149902,
      "step": 63387
    },
    {
      "epoch": 6.045055389404297e-05,
      "step": 63387,
      "training_step_time": 7.442762851715088
    },
    {
      "epoch": 6.045150756835937e-05,
      "model_forward_time": 1.7071435451507568,
      "step": 63388
    },
    {
      "epoch": 6.045150756835937e-05,
      "step": 63388,
      "training_step_time": 7.53922700881958
    },
    {
      "epoch": 6.0452461242675784e-05,
      "model_forward_time": 2.262922525405884,
      "step": 63389
    },
    {
      "epoch": 6.0452461242675784e-05,
      "step": 63389,
      "training_step_time": 7.367671728134155
    },
    {
      "epoch": 6.045341491699219e-05,
      "grad_norm": 0.10027138143777847,
      "learning_rate": 3.238195254721733e-06,
      "loss": 0.0578,
      "step": 63390
    },
    {
      "epoch": 6.045341491699219e-05,
      "model_forward_time": 1.7147109508514404,
      "step": 63390
    },
    {
      "dynamics_loss_avg": 0.05781135577708483,
      "epoch": 6.045341491699219e-05,
      "step": 63390
    },
    {
      "action_loss_avg": 0.0009679971466539427,
      "epoch": 6.045341491699219e-05,
      "step": 63390
    },
    {
      "epoch": 6.045341491699219e-05,
      "step": 63390,
      "training_step_time": 7.539230108261108
    },
    {
      "epoch": 6.045436859130859e-05,
      "model_forward_time": 1.7151153087615967,
      "step": 63391
    },
    {
      "epoch": 6.045436859130859e-05,
      "step": 63391,
      "training_step_time": 8.557674169540405
    },
    {
      "epoch": 6.0455322265625e-05,
      "model_forward_time": 1.7611610889434814,
      "step": 63392
    },
    {
      "epoch": 6.0455322265625e-05,
      "step": 63392,
      "training_step_time": 7.597737789154053
    },
    {
      "epoch": 6.045627593994141e-05,
      "model_forward_time": 1.7093653678894043,
      "step": 63393
    },
    {
      "epoch": 6.045627593994141e-05,
      "step": 63393,
      "training_step_time": 10.414040565490723
    },
    {
      "epoch": 6.045722961425781e-05,
      "model_forward_time": 0.6948146820068359,
      "step": 63394
    },
    {
      "epoch": 6.045722961425781e-05,
      "step": 63394,
      "training_step_time": 9.759177684783936
    },
    {
      "epoch": 6.045818328857422e-05,
      "model_forward_time": 1.721207857131958,
      "step": 63395
    },
    {
      "epoch": 6.045818328857422e-05,
      "step": 63395,
      "training_step_time": 7.58368444442749
    },
    {
      "epoch": 6.0459136962890626e-05,
      "model_forward_time": 1.703794240951538,
      "step": 63396
    },
    {
      "epoch": 6.0459136962890626e-05,
      "step": 63396,
      "training_step_time": 7.541807174682617
    },
    {
      "epoch": 6.046009063720703e-05,
      "model_forward_time": 1.712615728378296,
      "step": 63397
    },
    {
      "epoch": 6.046009063720703e-05,
      "step": 63397,
      "training_step_time": 7.602298021316528
    },
    {
      "epoch": 6.046104431152344e-05,
      "model_forward_time": 2.270792007446289,
      "step": 63398
    },
    {
      "epoch": 6.046104431152344e-05,
      "step": 63398,
      "training_step_time": 7.405835390090942
    },
    {
      "epoch": 6.0461997985839845e-05,
      "model_forward_time": 1.7078328132629395,
      "step": 63399
    },
    {
      "epoch": 6.0461997985839845e-05,
      "step": 63399,
      "training_step_time": 7.449923753738403
    },
    {
      "epoch": 6.046295166015625e-05,
      "grad_norm": 0.11304407566785812,
      "learning_rate": 3.2366479279106332e-06,
      "loss": 0.0602,
      "step": 63400
    },
    {
      "epoch": 6.046295166015625e-05,
      "model_forward_time": 1.708927869796753,
      "step": 63400
    },
    {
      "dynamics_loss_avg": 0.07304870299994945,
      "epoch": 6.046295166015625e-05,
      "step": 63400
    },
    {
      "action_loss_avg": 0.0015980711847078054,
      "epoch": 6.046295166015625e-05,
      "step": 63400
    },
    {
      "epoch": 6.046295166015625e-05,
      "step": 63400,
      "training_step_time": 7.670045375823975
    },
    {
      "epoch": 6.046390533447266e-05,
      "model_forward_time": 1.7585515975952148,
      "step": 63401
    },
    {
      "epoch": 6.046390533447266e-05,
      "step": 63401,
      "training_step_time": 8.43118166923523
    },
    {
      "epoch": 6.0464859008789063e-05,
      "model_forward_time": 1.7024028301239014,
      "step": 63402
    },
    {
      "epoch": 6.0464859008789063e-05,
      "step": 63402,
      "training_step_time": 7.692843198776245
    },
    {
      "epoch": 6.046581268310547e-05,
      "model_forward_time": 1.719001054763794,
      "step": 63403
    },
    {
      "epoch": 6.046581268310547e-05,
      "step": 63403,
      "training_step_time": 8.296818256378174
    },
    {
      "epoch": 6.046676635742188e-05,
      "model_forward_time": 1.7162158489227295,
      "step": 63404
    },
    {
      "epoch": 6.046676635742188e-05,
      "step": 63404,
      "training_step_time": 7.737563848495483
    },
    {
      "epoch": 6.046772003173828e-05,
      "model_forward_time": 1.7209875583648682,
      "step": 63405
    },
    {
      "epoch": 6.046772003173828e-05,
      "step": 63405,
      "training_step_time": 7.4415810108184814
    },
    {
      "epoch": 6.0468673706054686e-05,
      "model_forward_time": 1.7187392711639404,
      "step": 63406
    },
    {
      "epoch": 6.0468673706054686e-05,
      "step": 63406,
      "training_step_time": 7.490999460220337
    },
    {
      "epoch": 6.046962738037109e-05,
      "model_forward_time": 2.2822346687316895,
      "step": 63407
    },
    {
      "epoch": 6.046962738037109e-05,
      "step": 63407,
      "training_step_time": 7.346538305282593
    },
    {
      "epoch": 6.04705810546875e-05,
      "model_forward_time": 1.720597505569458,
      "step": 63408
    },
    {
      "epoch": 6.04705810546875e-05,
      "step": 63408,
      "training_step_time": 7.359689950942993
    },
    {
      "epoch": 6.0471534729003905e-05,
      "model_forward_time": 1.7224266529083252,
      "step": 63409
    },
    {
      "epoch": 6.0471534729003905e-05,
      "step": 63409,
      "training_step_time": 8.101125478744507
    },
    {
      "epoch": 6.047248840332031e-05,
      "grad_norm": 0.11957167088985443,
      "learning_rate": 3.235100793937065e-06,
      "loss": 0.058,
      "step": 63410
    },
    {
      "epoch": 6.047248840332031e-05,
      "model_forward_time": 1.7470650672912598,
      "step": 63410
    },
    {
      "dynamics_loss_avg": 0.05613377168774605,
      "epoch": 6.047248840332031e-05,
      "step": 63410
    },
    {
      "action_loss_avg": 0.001566184271359816,
      "epoch": 6.047248840332031e-05,
      "step": 63410
    },
    {
      "epoch": 6.047248840332031e-05,
      "step": 63410,
      "training_step_time": 9.60469651222229
    },
    {
      "epoch": 6.047344207763672e-05,
      "model_forward_time": 1.7255873680114746,
      "step": 63411
    },
    {
      "epoch": 6.047344207763672e-05,
      "step": 63411,
      "training_step_time": 8.313685417175293
    },
    {
      "epoch": 6.0474395751953124e-05,
      "model_forward_time": 1.7230255603790283,
      "step": 63412
    },
    {
      "epoch": 6.0474395751953124e-05,
      "step": 63412,
      "training_step_time": 7.690897464752197
    },
    {
      "epoch": 6.047534942626953e-05,
      "model_forward_time": 1.7056000232696533,
      "step": 63413
    },
    {
      "epoch": 6.047534942626953e-05,
      "step": 63413,
      "training_step_time": 9.575014352798462
    },
    {
      "epoch": 6.047630310058594e-05,
      "model_forward_time": 1.714249849319458,
      "step": 63414
    },
    {
      "epoch": 6.047630310058594e-05,
      "step": 63414,
      "training_step_time": 7.436415910720825
    },
    {
      "epoch": 6.047725677490234e-05,
      "model_forward_time": 1.7058465480804443,
      "step": 63415
    },
    {
      "epoch": 6.047725677490234e-05,
      "step": 63415,
      "training_step_time": 7.08717942237854
    },
    {
      "epoch": 6.047821044921875e-05,
      "model_forward_time": 1.7054917812347412,
      "step": 63416
    },
    {
      "epoch": 6.047821044921875e-05,
      "step": 63416,
      "training_step_time": 8.101293087005615
    },
    {
      "epoch": 6.047916412353516e-05,
      "model_forward_time": 1.7270622253417969,
      "step": 63417
    },
    {
      "epoch": 6.047916412353516e-05,
      "step": 63417,
      "training_step_time": 7.385444164276123
    },
    {
      "epoch": 6.048011779785156e-05,
      "model_forward_time": 1.7225632667541504,
      "step": 63418
    },
    {
      "epoch": 6.048011779785156e-05,
      "step": 63418,
      "training_step_time": 8.170624017715454
    },
    {
      "epoch": 6.0481071472167966e-05,
      "model_forward_time": 1.722726583480835,
      "step": 63419
    },
    {
      "epoch": 6.0481071472167966e-05,
      "step": 63419,
      "training_step_time": 8.106070756912231
    },
    {
      "epoch": 6.048202514648438e-05,
      "grad_norm": 0.1018427386879921,
      "learning_rate": 3.233553852970219e-06,
      "loss": 0.0583,
      "step": 63420
    },
    {
      "epoch": 6.048202514648438e-05,
      "model_forward_time": 1.733201026916504,
      "step": 63420
    },
    {
      "dynamics_loss_avg": 0.06974854543805123,
      "epoch": 6.048202514648438e-05,
      "step": 63420
    },
    {
      "action_loss_avg": 0.0018611346778925509,
      "epoch": 6.048202514648438e-05,
      "step": 63420
    },
    {
      "epoch": 6.048202514648438e-05,
      "step": 63420,
      "training_step_time": 8.479901790618896
    },
    {
      "epoch": 6.048297882080078e-05,
      "model_forward_time": 1.7038311958312988,
      "step": 63421
    },
    {
      "epoch": 6.048297882080078e-05,
      "step": 63421,
      "training_step_time": 8.4047532081604
    },
    {
      "epoch": 6.0483932495117185e-05,
      "model_forward_time": 1.740516185760498,
      "step": 63422
    },
    {
      "epoch": 6.0483932495117185e-05,
      "step": 63422,
      "training_step_time": 7.653059482574463
    },
    {
      "epoch": 6.0484886169433595e-05,
      "model_forward_time": 1.7088439464569092,
      "step": 63423
    },
    {
      "epoch": 6.0484886169433595e-05,
      "step": 63423,
      "training_step_time": 7.491106986999512
    },
    {
      "epoch": 6.048583984375e-05,
      "model_forward_time": 1.7106523513793945,
      "step": 63424
    },
    {
      "epoch": 6.048583984375e-05,
      "step": 63424,
      "training_step_time": 7.49200963973999
    },
    {
      "epoch": 6.0486793518066404e-05,
      "model_forward_time": 1.7157948017120361,
      "step": 63425
    },
    {
      "epoch": 6.0486793518066404e-05,
      "step": 63425,
      "training_step_time": 7.503591299057007
    },
    {
      "epoch": 6.0487747192382814e-05,
      "model_forward_time": 2.265120506286621,
      "step": 63426
    },
    {
      "epoch": 6.0487747192382814e-05,
      "step": 63426,
      "training_step_time": 7.348766326904297
    },
    {
      "epoch": 6.048870086669922e-05,
      "model_forward_time": 1.7135331630706787,
      "step": 63427
    },
    {
      "epoch": 6.048870086669922e-05,
      "step": 63427,
      "training_step_time": 7.633902549743652
    },
    {
      "epoch": 6.048965454101562e-05,
      "model_forward_time": 1.7134275436401367,
      "step": 63428
    },
    {
      "epoch": 6.048965454101562e-05,
      "step": 63428,
      "training_step_time": 8.43447995185852
    },
    {
      "epoch": 6.049060821533203e-05,
      "model_forward_time": 1.7618982791900635,
      "step": 63429
    },
    {
      "epoch": 6.049060821533203e-05,
      "step": 63429,
      "training_step_time": 7.612628221511841
    },
    {
      "epoch": 6.049156188964844e-05,
      "grad_norm": 0.09699663519859314,
      "learning_rate": 3.2320071051792677e-06,
      "loss": 0.0575,
      "step": 63430
    },
    {
      "epoch": 6.049156188964844e-05,
      "model_forward_time": 1.724013090133667,
      "step": 63430
    },
    {
      "dynamics_loss_avg": 0.058066203445196155,
      "epoch": 6.049156188964844e-05,
      "step": 63430
    },
    {
      "action_loss_avg": 0.0014538164134137333,
      "epoch": 6.049156188964844e-05,
      "step": 63430
    },
    {
      "epoch": 6.049156188964844e-05,
      "step": 63430,
      "training_step_time": 7.713610887527466
    },
    {
      "epoch": 6.049251556396484e-05,
      "model_forward_time": 1.7196266651153564,
      "step": 63431
    },
    {
      "epoch": 6.049251556396484e-05,
      "step": 63431,
      "training_step_time": 7.688955783843994
    },
    {
      "epoch": 6.049346923828125e-05,
      "model_forward_time": 1.7252049446105957,
      "step": 63432
    },
    {
      "epoch": 6.049346923828125e-05,
      "step": 63432,
      "training_step_time": 7.4102699756622314
    },
    {
      "epoch": 6.0494422912597656e-05,
      "model_forward_time": 1.7309014797210693,
      "step": 63433
    },
    {
      "epoch": 6.0494422912597656e-05,
      "step": 63433,
      "training_step_time": 7.626269102096558
    },
    {
      "epoch": 6.049537658691406e-05,
      "model_forward_time": 1.7244539260864258,
      "step": 63434
    },
    {
      "epoch": 6.049537658691406e-05,
      "step": 63434,
      "training_step_time": 7.564628601074219
    },
    {
      "epoch": 6.049633026123047e-05,
      "model_forward_time": 2.895618200302124,
      "step": 63435
    },
    {
      "epoch": 6.049633026123047e-05,
      "step": 63435,
      "training_step_time": 8.620899438858032
    },
    {
      "epoch": 6.0497283935546875e-05,
      "model_forward_time": 1.72477126121521,
      "step": 63436
    },
    {
      "epoch": 6.0497283935546875e-05,
      "step": 63436,
      "training_step_time": 9.020671367645264
    },
    {
      "epoch": 6.049823760986328e-05,
      "model_forward_time": 1.7246880531311035,
      "step": 63437
    },
    {
      "epoch": 6.049823760986328e-05,
      "step": 63437,
      "training_step_time": 7.5540876388549805
    },
    {
      "epoch": 6.049919128417969e-05,
      "model_forward_time": 1.7445425987243652,
      "step": 63438
    },
    {
      "epoch": 6.049919128417969e-05,
      "step": 63438,
      "training_step_time": 8.681244134902954
    },
    {
      "epoch": 6.0500144958496094e-05,
      "model_forward_time": 1.7238717079162598,
      "step": 63439
    },
    {
      "epoch": 6.0500144958496094e-05,
      "step": 63439,
      "training_step_time": 7.734645843505859
    },
    {
      "epoch": 6.05010986328125e-05,
      "grad_norm": 0.10843618959188461,
      "learning_rate": 3.230460550733363e-06,
      "loss": 0.0591,
      "step": 63440
    },
    {
      "epoch": 6.05010986328125e-05,
      "model_forward_time": 1.757962703704834,
      "step": 63440
    },
    {
      "dynamics_loss_avg": 0.05756259225308895,
      "epoch": 6.05010986328125e-05,
      "step": 63440
    },
    {
      "action_loss_avg": 0.0026757485116831957,
      "epoch": 6.05010986328125e-05,
      "step": 63440
    },
    {
      "epoch": 6.05010986328125e-05,
      "step": 63440,
      "training_step_time": 7.712120532989502
    },
    {
      "epoch": 6.050205230712891e-05,
      "model_forward_time": 1.7333343029022217,
      "step": 63441
    },
    {
      "epoch": 6.050205230712891e-05,
      "step": 63441,
      "training_step_time": 7.72817325592041
    },
    {
      "epoch": 6.050300598144531e-05,
      "model_forward_time": 1.731651782989502,
      "step": 63442
    },
    {
      "epoch": 6.050300598144531e-05,
      "step": 63442,
      "training_step_time": 7.5382020473480225
    },
    {
      "epoch": 6.050395965576172e-05,
      "model_forward_time": 1.7235362529754639,
      "step": 63443
    },
    {
      "epoch": 6.050395965576172e-05,
      "step": 63443,
      "training_step_time": 8.053306102752686
    },
    {
      "epoch": 6.050491333007813e-05,
      "model_forward_time": 2.932098627090454,
      "step": 63444
    },
    {
      "epoch": 6.050491333007813e-05,
      "step": 63444,
      "training_step_time": 7.662474632263184
    },
    {
      "epoch": 6.050586700439453e-05,
      "model_forward_time": 1.7325422763824463,
      "step": 63445
    },
    {
      "epoch": 6.050586700439453e-05,
      "step": 63445,
      "training_step_time": 7.667303085327148
    },
    {
      "epoch": 6.0506820678710936e-05,
      "model_forward_time": 1.7509896755218506,
      "step": 63446
    },
    {
      "epoch": 6.0506820678710936e-05,
      "step": 63446,
      "training_step_time": 8.125454187393188
    },
    {
      "epoch": 6.0507774353027346e-05,
      "model_forward_time": 1.7459533214569092,
      "step": 63447
    },
    {
      "epoch": 6.0507774353027346e-05,
      "step": 63447,
      "training_step_time": 8.415342330932617
    },
    {
      "epoch": 6.050872802734375e-05,
      "model_forward_time": 1.7393124103546143,
      "step": 63448
    },
    {
      "epoch": 6.050872802734375e-05,
      "step": 63448,
      "training_step_time": 7.665235280990601
    },
    {
      "epoch": 6.0509681701660154e-05,
      "model_forward_time": 1.4948530197143555,
      "step": 63449
    },
    {
      "epoch": 6.0509681701660154e-05,
      "step": 63449,
      "training_step_time": 8.515962362289429
    },
    {
      "epoch": 6.0510635375976565e-05,
      "grad_norm": 0.10880113393068314,
      "learning_rate": 3.228914189801632e-06,
      "loss": 0.0602,
      "step": 63450
    },
    {
      "epoch": 6.0510635375976565e-05,
      "model_forward_time": 1.7466108798980713,
      "step": 63450
    },
    {
      "dynamics_loss_avg": 0.07927352469414473,
      "epoch": 6.0510635375976565e-05,
      "step": 63450
    },
    {
      "action_loss_avg": 0.001949443417834118,
      "epoch": 6.0510635375976565e-05,
      "step": 63450
    },
    {
      "epoch": 6.0510635375976565e-05,
      "step": 63450,
      "training_step_time": 8.647877931594849
    },
    {
      "epoch": 6.051158905029297e-05,
      "model_forward_time": 1.7435235977172852,
      "step": 63451
    },
    {
      "epoch": 6.051158905029297e-05,
      "step": 63451,
      "training_step_time": 7.519491672515869
    },
    {
      "epoch": 6.051254272460937e-05,
      "model_forward_time": 1.7249174118041992,
      "step": 63452
    },
    {
      "epoch": 6.051254272460937e-05,
      "step": 63452,
      "training_step_time": 7.09049916267395
    },
    {
      "epoch": 6.0513496398925784e-05,
      "model_forward_time": 2.887840986251831,
      "step": 63453
    },
    {
      "epoch": 6.0513496398925784e-05,
      "step": 63453,
      "training_step_time": 7.654356956481934
    },
    {
      "epoch": 6.051445007324219e-05,
      "model_forward_time": 1.7427325248718262,
      "step": 63454
    },
    {
      "epoch": 6.051445007324219e-05,
      "step": 63454,
      "training_step_time": 7.574176549911499
    },
    {
      "epoch": 6.051540374755859e-05,
      "model_forward_time": 1.7373850345611572,
      "step": 63455
    },
    {
      "epoch": 6.051540374755859e-05,
      "step": 63455,
      "training_step_time": 9.373615264892578
    },
    {
      "epoch": 6.0516357421875e-05,
      "model_forward_time": 1.7341480255126953,
      "step": 63456
    },
    {
      "epoch": 6.0516357421875e-05,
      "step": 63456,
      "training_step_time": 8.755038738250732
    },
    {
      "epoch": 6.051731109619141e-05,
      "model_forward_time": 1.7515153884887695,
      "step": 63457
    },
    {
      "epoch": 6.051731109619141e-05,
      "step": 63457,
      "training_step_time": 8.350547552108765
    },
    {
      "epoch": 6.051826477050781e-05,
      "model_forward_time": 1.7818851470947266,
      "step": 63458
    },
    {
      "epoch": 6.051826477050781e-05,
      "step": 63458,
      "training_step_time": 8.723742485046387
    },
    {
      "epoch": 6.051921844482422e-05,
      "model_forward_time": 1.740619421005249,
      "step": 63459
    },
    {
      "epoch": 6.051921844482422e-05,
      "step": 63459,
      "training_step_time": 7.7061848640441895
    },
    {
      "epoch": 6.0520172119140626e-05,
      "grad_norm": 0.12500856816768646,
      "learning_rate": 3.2273680225531808e-06,
      "loss": 0.0592,
      "step": 63460
    },
    {
      "epoch": 6.0520172119140626e-05,
      "model_forward_time": 1.7210958003997803,
      "step": 63460
    },
    {
      "dynamics_loss_avg": 0.0466285090893507,
      "epoch": 6.0520172119140626e-05,
      "step": 63460
    },
    {
      "action_loss_avg": 0.0013653389643877745,
      "epoch": 6.0520172119140626e-05,
      "step": 63460
    },
    {
      "epoch": 6.0520172119140626e-05,
      "step": 63460,
      "training_step_time": 7.645143270492554
    },
    {
      "epoch": 6.052112579345703e-05,
      "model_forward_time": 1.7268564701080322,
      "step": 63461
    },
    {
      "epoch": 6.052112579345703e-05,
      "step": 63461,
      "training_step_time": 7.1497578620910645
    },
    {
      "epoch": 6.052207946777344e-05,
      "model_forward_time": 2.882544755935669,
      "step": 63462
    },
    {
      "epoch": 6.052207946777344e-05,
      "step": 63462,
      "training_step_time": 7.583564758300781
    },
    {
      "epoch": 6.0523033142089845e-05,
      "model_forward_time": 1.7201952934265137,
      "step": 63463
    },
    {
      "epoch": 6.0523033142089845e-05,
      "step": 63463,
      "training_step_time": 7.397709608078003
    },
    {
      "epoch": 6.052398681640625e-05,
      "model_forward_time": 1.7207615375518799,
      "step": 63464
    },
    {
      "epoch": 6.052398681640625e-05,
      "step": 63464,
      "training_step_time": 7.63340163230896
    },
    {
      "epoch": 6.052494049072266e-05,
      "model_forward_time": 1.3616349697113037,
      "step": 63465
    },
    {
      "epoch": 6.052494049072266e-05,
      "step": 63465,
      "training_step_time": 8.196779489517212
    },
    {
      "epoch": 6.0525894165039064e-05,
      "model_forward_time": 1.7242631912231445,
      "step": 63466
    },
    {
      "epoch": 6.0525894165039064e-05,
      "step": 63466,
      "training_step_time": 8.688001155853271
    },
    {
      "epoch": 6.052684783935547e-05,
      "model_forward_time": 1.7619373798370361,
      "step": 63467
    },
    {
      "epoch": 6.052684783935547e-05,
      "step": 63467,
      "training_step_time": 11.430784702301025
    },
    {
      "epoch": 6.052780151367188e-05,
      "model_forward_time": 1.7284770011901855,
      "step": 63468
    },
    {
      "epoch": 6.052780151367188e-05,
      "step": 63468,
      "training_step_time": 8.677165746688843
    },
    {
      "epoch": 6.052875518798828e-05,
      "model_forward_time": 1.727066993713379,
      "step": 63469
    },
    {
      "epoch": 6.052875518798828e-05,
      "step": 63469,
      "training_step_time": 7.740342378616333
    },
    {
      "epoch": 6.0529708862304686e-05,
      "grad_norm": 0.08821732550859451,
      "learning_rate": 3.2258220491570987e-06,
      "loss": 0.0546,
      "step": 63470
    },
    {
      "epoch": 6.0529708862304686e-05,
      "model_forward_time": 1.716932773590088,
      "step": 63470
    },
    {
      "dynamics_loss_avg": 0.06020891033113003,
      "epoch": 6.0529708862304686e-05,
      "step": 63470
    },
    {
      "action_loss_avg": 0.0023338334547588603,
      "epoch": 6.0529708862304686e-05,
      "step": 63470
    },
    {
      "epoch": 6.0529708862304686e-05,
      "step": 63470,
      "training_step_time": 7.631129026412964
    },
    {
      "epoch": 6.053066253662109e-05,
      "model_forward_time": 2.844752788543701,
      "step": 63471
    },
    {
      "epoch": 6.053066253662109e-05,
      "step": 63471,
      "training_step_time": 7.620969295501709
    },
    {
      "epoch": 6.05316162109375e-05,
      "model_forward_time": 1.7333457469940186,
      "step": 63472
    },
    {
      "epoch": 6.05316162109375e-05,
      "step": 63472,
      "training_step_time": 7.485326290130615
    },
    {
      "epoch": 6.0532569885253905e-05,
      "model_forward_time": 1.7343146800994873,
      "step": 63473
    },
    {
      "epoch": 6.0532569885253905e-05,
      "step": 63473,
      "training_step_time": 7.727601528167725
    },
    {
      "epoch": 6.053352355957031e-05,
      "model_forward_time": 1.7209022045135498,
      "step": 63474
    },
    {
      "epoch": 6.053352355957031e-05,
      "step": 63474,
      "training_step_time": 8.283551931381226
    },
    {
      "epoch": 6.053447723388672e-05,
      "model_forward_time": 1.7418415546417236,
      "step": 63475
    },
    {
      "epoch": 6.053447723388672e-05,
      "step": 63475,
      "training_step_time": 9.443689823150635
    },
    {
      "epoch": 6.0535430908203124e-05,
      "model_forward_time": 1.7339351177215576,
      "step": 63476
    },
    {
      "epoch": 6.0535430908203124e-05,
      "step": 63476,
      "training_step_time": 8.201493263244629
    },
    {
      "epoch": 6.053638458251953e-05,
      "model_forward_time": 1.7432036399841309,
      "step": 63477
    },
    {
      "epoch": 6.053638458251953e-05,
      "step": 63477,
      "training_step_time": 8.850697755813599
    },
    {
      "epoch": 6.053733825683594e-05,
      "model_forward_time": 1.739058017730713,
      "step": 63478
    },
    {
      "epoch": 6.053733825683594e-05,
      "step": 63478,
      "training_step_time": 7.7197105884552
    },
    {
      "epoch": 6.053829193115234e-05,
      "model_forward_time": 1.7374811172485352,
      "step": 63479
    },
    {
      "epoch": 6.053829193115234e-05,
      "step": 63479,
      "training_step_time": 7.641937255859375
    },
    {
      "epoch": 6.053924560546875e-05,
      "grad_norm": 0.24149413406848907,
      "learning_rate": 3.2242762697824496e-06,
      "loss": 0.057,
      "step": 63480
    },
    {
      "epoch": 6.053924560546875e-05,
      "model_forward_time": 1.7341318130493164,
      "step": 63480
    },
    {
      "dynamics_loss_avg": 0.060293405689299105,
      "epoch": 6.053924560546875e-05,
      "step": 63480
    },
    {
      "action_loss_avg": 0.0021575645892880856,
      "epoch": 6.053924560546875e-05,
      "step": 63480
    },
    {
      "epoch": 6.053924560546875e-05,
      "step": 63480,
      "training_step_time": 8.33841609954834
    },
    {
      "epoch": 6.054019927978516e-05,
      "model_forward_time": 1.737635850906372,
      "step": 63481
    },
    {
      "epoch": 6.054019927978516e-05,
      "step": 63481,
      "training_step_time": 7.457239627838135
    },
    {
      "epoch": 6.054115295410156e-05,
      "model_forward_time": 1.74456787109375,
      "step": 63482
    },
    {
      "epoch": 6.054115295410156e-05,
      "step": 63482,
      "training_step_time": 8.511452436447144
    },
    {
      "epoch": 6.0542106628417966e-05,
      "model_forward_time": 1.731539011001587,
      "step": 63483
    },
    {
      "epoch": 6.0542106628417966e-05,
      "step": 63483,
      "training_step_time": 7.580732107162476
    },
    {
      "epoch": 6.054306030273438e-05,
      "model_forward_time": 1.7356908321380615,
      "step": 63484
    },
    {
      "epoch": 6.054306030273438e-05,
      "step": 63484,
      "training_step_time": 7.857709884643555
    },
    {
      "epoch": 6.054401397705078e-05,
      "model_forward_time": 1.8311774730682373,
      "step": 63485
    },
    {
      "epoch": 6.054401397705078e-05,
      "step": 63485,
      "training_step_time": 9.522460460662842
    },
    {
      "epoch": 6.0544967651367185e-05,
      "model_forward_time": 1.734694480895996,
      "step": 63486
    },
    {
      "epoch": 6.0544967651367185e-05,
      "step": 63486,
      "training_step_time": 10.336918354034424
    },
    {
      "epoch": 6.0545921325683596e-05,
      "model_forward_time": 1.740487813949585,
      "step": 63487
    },
    {
      "epoch": 6.0545921325683596e-05,
      "step": 63487,
      "training_step_time": 7.633310794830322
    },
    {
      "epoch": 6.0546875e-05,
      "model_forward_time": 1.7323248386383057,
      "step": 63488
    },
    {
      "epoch": 6.0546875e-05,
      "step": 63488,
      "training_step_time": 7.59171724319458
    },
    {
      "epoch": 6.0547828674316404e-05,
      "model_forward_time": 1.735964059829712,
      "step": 63489
    },
    {
      "epoch": 6.0547828674316404e-05,
      "step": 63489,
      "training_step_time": 7.1343607902526855
    },
    {
      "epoch": 6.0548782348632814e-05,
      "grad_norm": 0.10853694379329681,
      "learning_rate": 3.222730684598279e-06,
      "loss": 0.0575,
      "step": 63490
    },
    {
      "epoch": 6.0548782348632814e-05,
      "model_forward_time": 2.9442222118377686,
      "step": 63490
    },
    {
      "dynamics_loss_avg": 0.05638785902410746,
      "epoch": 6.0548782348632814e-05,
      "step": 63490
    },
    {
      "action_loss_avg": 0.0021901235886616632,
      "epoch": 6.0548782348632814e-05,
      "step": 63490
    },
    {
      "epoch": 6.0548782348632814e-05,
      "step": 63490,
      "training_step_time": 8.566695928573608
    },
    {
      "epoch": 6.054973602294922e-05,
      "model_forward_time": 1.7302660942077637,
      "step": 63491
    },
    {
      "epoch": 6.054973602294922e-05,
      "step": 63491,
      "training_step_time": 7.367722272872925
    },
    {
      "epoch": 6.055068969726562e-05,
      "model_forward_time": 1.7368824481964111,
      "step": 63492
    },
    {
      "epoch": 6.055068969726562e-05,
      "step": 63492,
      "training_step_time": 7.530767917633057
    },
    {
      "epoch": 6.055164337158203e-05,
      "model_forward_time": 1.7407758235931396,
      "step": 63493
    },
    {
      "epoch": 6.055164337158203e-05,
      "step": 63493,
      "training_step_time": 8.485928297042847
    },
    {
      "epoch": 6.055259704589844e-05,
      "model_forward_time": 1.7699899673461914,
      "step": 63494
    },
    {
      "epoch": 6.055259704589844e-05,
      "step": 63494,
      "training_step_time": 7.539218425750732
    },
    {
      "epoch": 6.055355072021484e-05,
      "model_forward_time": 1.7340247631072998,
      "step": 63495
    },
    {
      "epoch": 6.055355072021484e-05,
      "step": 63495,
      "training_step_time": 7.738184452056885
    },
    {
      "epoch": 6.055450439453125e-05,
      "model_forward_time": 1.7362966537475586,
      "step": 63496
    },
    {
      "epoch": 6.055450439453125e-05,
      "step": 63496,
      "training_step_time": 8.228803396224976
    },
    {
      "epoch": 6.0555458068847656e-05,
      "model_forward_time": 1.7398440837860107,
      "step": 63497
    },
    {
      "epoch": 6.0555458068847656e-05,
      "step": 63497,
      "training_step_time": 7.561859369277954
    },
    {
      "epoch": 6.055641174316406e-05,
      "model_forward_time": 1.741325855255127,
      "step": 63498
    },
    {
      "epoch": 6.055641174316406e-05,
      "step": 63498,
      "training_step_time": 7.609272241592407
    },
    {
      "epoch": 6.055736541748047e-05,
      "model_forward_time": 2.8993799686431885,
      "step": 63499
    },
    {
      "epoch": 6.055736541748047e-05,
      "step": 63499,
      "training_step_time": 7.701450824737549
    },
    {
      "epoch": 6.0558319091796875e-05,
      "grad_norm": 0.09063122421503067,
      "learning_rate": 3.2211852937736054e-06,
      "loss": 0.0564,
      "step": 63500
    },
    {
      "epoch": 6.0558319091796875e-05,
      "model_forward_time": 1.738325834274292,
      "step": 63500
    },
    {
      "dynamics_loss_avg": 0.05961896125227213,
      "epoch": 6.0558319091796875e-05,
      "step": 63500
    },
    {
      "action_loss_avg": 0.0015628542751073837,
      "epoch": 6.0558319091796875e-05,
      "step": 63500
    },
    {
      "epoch": 6.0558319091796875e-05,
      "step": 63500,
      "training_step_time": 7.380478858947754
    },
    {
      "epoch": 6.055927276611328e-05,
      "model_forward_time": 1.7279636859893799,
      "step": 63501
    },
    {
      "epoch": 6.055927276611328e-05,
      "step": 63501,
      "training_step_time": 7.564517498016357
    },
    {
      "epoch": 6.056022644042969e-05,
      "model_forward_time": 1.7312719821929932,
      "step": 63502
    },
    {
      "epoch": 6.056022644042969e-05,
      "step": 63502,
      "training_step_time": 8.437805414199829
    },
    {
      "epoch": 6.0561180114746094e-05,
      "model_forward_time": 1.7745296955108643,
      "step": 63503
    },
    {
      "epoch": 6.0561180114746094e-05,
      "step": 63503,
      "training_step_time": 7.676940202713013
    },
    {
      "epoch": 6.05621337890625e-05,
      "model_forward_time": 1.7298390865325928,
      "step": 63504
    },
    {
      "epoch": 6.05621337890625e-05,
      "step": 63504,
      "training_step_time": 9.383192539215088
    },
    {
      "epoch": 6.056308746337891e-05,
      "model_forward_time": 1.7218925952911377,
      "step": 63505
    },
    {
      "epoch": 6.056308746337891e-05,
      "step": 63505,
      "training_step_time": 8.440792083740234
    },
    {
      "epoch": 6.056404113769531e-05,
      "model_forward_time": 1.7353322505950928,
      "step": 63506
    },
    {
      "epoch": 6.056404113769531e-05,
      "step": 63506,
      "training_step_time": 7.509415149688721
    },
    {
      "epoch": 6.056499481201172e-05,
      "model_forward_time": 1.7290716171264648,
      "step": 63507
    },
    {
      "epoch": 6.056499481201172e-05,
      "step": 63507,
      "training_step_time": 7.546875476837158
    },
    {
      "epoch": 6.056594848632813e-05,
      "model_forward_time": 2.8330037593841553,
      "step": 63508
    },
    {
      "epoch": 6.056594848632813e-05,
      "step": 63508,
      "training_step_time": 8.113768577575684
    },
    {
      "epoch": 6.056690216064453e-05,
      "model_forward_time": 1.7288503646850586,
      "step": 63509
    },
    {
      "epoch": 6.056690216064453e-05,
      "step": 63509,
      "training_step_time": 7.39238166809082
    },
    {
      "epoch": 6.0567855834960936e-05,
      "grad_norm": 0.10737551748752594,
      "learning_rate": 3.2196400974774358e-06,
      "loss": 0.0544,
      "step": 63510
    },
    {
      "epoch": 6.0567855834960936e-05,
      "model_forward_time": 1.733865737915039,
      "step": 63510
    },
    {
      "dynamics_loss_avg": 0.05156174190342426,
      "epoch": 6.0567855834960936e-05,
      "step": 63510
    },
    {
      "action_loss_avg": 0.0019077994104009122,
      "epoch": 6.0567855834960936e-05,
      "step": 63510
    },
    {
      "epoch": 6.0567855834960936e-05,
      "step": 63510,
      "training_step_time": 7.545966386795044
    },
    {
      "epoch": 6.0568809509277346e-05,
      "model_forward_time": 1.7273402214050293,
      "step": 63511
    },
    {
      "epoch": 6.0568809509277346e-05,
      "step": 63511,
      "training_step_time": 8.355668306350708
    },
    {
      "epoch": 6.056976318359375e-05,
      "model_forward_time": 1.7475039958953857,
      "step": 63512
    },
    {
      "epoch": 6.056976318359375e-05,
      "step": 63512,
      "training_step_time": 7.6747658252716064
    },
    {
      "epoch": 6.0570716857910154e-05,
      "model_forward_time": 1.727919101715088,
      "step": 63513
    },
    {
      "epoch": 6.0570716857910154e-05,
      "step": 63513,
      "training_step_time": 8.485741376876831
    },
    {
      "epoch": 6.0571670532226565e-05,
      "model_forward_time": 1.7150249481201172,
      "step": 63514
    },
    {
      "epoch": 6.0571670532226565e-05,
      "step": 63514,
      "training_step_time": 8.44410753250122
    },
    {
      "epoch": 6.057262420654297e-05,
      "model_forward_time": 1.7273211479187012,
      "step": 63515
    },
    {
      "epoch": 6.057262420654297e-05,
      "step": 63515,
      "training_step_time": 7.679418325424194
    },
    {
      "epoch": 6.057357788085937e-05,
      "model_forward_time": 1.7312989234924316,
      "step": 63516
    },
    {
      "epoch": 6.057357788085937e-05,
      "step": 63516,
      "training_step_time": 7.535054445266724
    },
    {
      "epoch": 6.0574531555175784e-05,
      "model_forward_time": 2.861375093460083,
      "step": 63517
    },
    {
      "epoch": 6.0574531555175784e-05,
      "step": 63517,
      "training_step_time": 8.324739933013916
    },
    {
      "epoch": 6.057548522949219e-05,
      "model_forward_time": 1.7287518978118896,
      "step": 63518
    },
    {
      "epoch": 6.057548522949219e-05,
      "step": 63518,
      "training_step_time": 7.49895167350769
    },
    {
      "epoch": 6.057643890380859e-05,
      "model_forward_time": 1.7269837856292725,
      "step": 63519
    },
    {
      "epoch": 6.057643890380859e-05,
      "step": 63519,
      "training_step_time": 7.5127949714660645
    },
    {
      "epoch": 6.0577392578125e-05,
      "grad_norm": 0.1263752430677414,
      "learning_rate": 3.2180950958787487e-06,
      "loss": 0.0589,
      "step": 63520
    },
    {
      "epoch": 6.0577392578125e-05,
      "model_forward_time": 1.7334082126617432,
      "step": 63520
    },
    {
      "dynamics_loss_avg": 0.05188954584300518,
      "epoch": 6.0577392578125e-05,
      "step": 63520
    },
    {
      "action_loss_avg": 0.0015491959580685942,
      "epoch": 6.0577392578125e-05,
      "step": 63520
    },
    {
      "epoch": 6.0577392578125e-05,
      "step": 63520,
      "training_step_time": 8.125239372253418
    },
    {
      "epoch": 6.057834625244141e-05,
      "model_forward_time": 1.7440359592437744,
      "step": 63521
    },
    {
      "epoch": 6.057834625244141e-05,
      "step": 63521,
      "training_step_time": 9.544209718704224
    },
    {
      "epoch": 6.057929992675781e-05,
      "model_forward_time": 1.7624213695526123,
      "step": 63522
    },
    {
      "epoch": 6.057929992675781e-05,
      "step": 63522,
      "training_step_time": 8.31847357749939
    },
    {
      "epoch": 6.058025360107422e-05,
      "model_forward_time": 1.7236912250518799,
      "step": 63523
    },
    {
      "epoch": 6.058025360107422e-05,
      "step": 63523,
      "training_step_time": 8.348653554916382
    },
    {
      "epoch": 6.0581207275390626e-05,
      "model_forward_time": 1.7262382507324219,
      "step": 63524
    },
    {
      "epoch": 6.0581207275390626e-05,
      "step": 63524,
      "training_step_time": 8.352028608322144
    },
    {
      "epoch": 6.058216094970703e-05,
      "model_forward_time": 1.721426010131836,
      "step": 63525
    },
    {
      "epoch": 6.058216094970703e-05,
      "step": 63525,
      "training_step_time": 7.656046152114868
    },
    {
      "epoch": 6.058311462402344e-05,
      "model_forward_time": 2.8026490211486816,
      "step": 63526
    },
    {
      "epoch": 6.058311462402344e-05,
      "step": 63526,
      "training_step_time": 7.565727472305298
    },
    {
      "epoch": 6.0584068298339845e-05,
      "model_forward_time": 1.7227060794830322,
      "step": 63527
    },
    {
      "epoch": 6.0584068298339845e-05,
      "step": 63527,
      "training_step_time": 7.555983543395996
    },
    {
      "epoch": 6.058502197265625e-05,
      "model_forward_time": 1.7165968418121338,
      "step": 63528
    },
    {
      "epoch": 6.058502197265625e-05,
      "step": 63528,
      "training_step_time": 7.46752142906189
    },
    {
      "epoch": 6.058597564697266e-05,
      "model_forward_time": 1.7308378219604492,
      "step": 63529
    },
    {
      "epoch": 6.058597564697266e-05,
      "step": 63529,
      "training_step_time": 7.578469276428223
    },
    {
      "epoch": 6.0586929321289064e-05,
      "grad_norm": 0.08435006439685822,
      "learning_rate": 3.2165502891465026e-06,
      "loss": 0.0567,
      "step": 63530
    },
    {
      "epoch": 6.0586929321289064e-05,
      "model_forward_time": 1.7848238945007324,
      "step": 63530
    },
    {
      "dynamics_loss_avg": 0.05502966782078147,
      "epoch": 6.0586929321289064e-05,
      "step": 63530
    },
    {
      "action_loss_avg": 0.002113705745432526,
      "epoch": 6.0586929321289064e-05,
      "step": 63530
    },
    {
      "epoch": 6.0586929321289064e-05,
      "step": 63530,
      "training_step_time": 7.811661720275879
    },
    {
      "epoch": 6.058788299560547e-05,
      "model_forward_time": 1.7480647563934326,
      "step": 63531
    },
    {
      "epoch": 6.058788299560547e-05,
      "step": 63531,
      "training_step_time": 7.671854019165039
    },
    {
      "epoch": 6.058883666992187e-05,
      "model_forward_time": 1.7254703044891357,
      "step": 63532
    },
    {
      "epoch": 6.058883666992187e-05,
      "step": 63532,
      "training_step_time": 7.729177474975586
    },
    {
      "epoch": 6.058979034423828e-05,
      "model_forward_time": 1.733640432357788,
      "step": 63533
    },
    {
      "epoch": 6.058979034423828e-05,
      "step": 63533,
      "training_step_time": 8.611377477645874
    },
    {
      "epoch": 6.0590744018554686e-05,
      "model_forward_time": 1.7253496646881104,
      "step": 63534
    },
    {
      "epoch": 6.0590744018554686e-05,
      "step": 63534,
      "training_step_time": 7.549880266189575
    },
    {
      "epoch": 6.059169769287109e-05,
      "model_forward_time": 1.73225736618042,
      "step": 63535
    },
    {
      "epoch": 6.059169769287109e-05,
      "step": 63535,
      "training_step_time": 7.394967317581177
    },
    {
      "epoch": 6.05926513671875e-05,
      "model_forward_time": 2.932466506958008,
      "step": 63536
    },
    {
      "epoch": 6.05926513671875e-05,
      "step": 63536,
      "training_step_time": 7.74229621887207
    },
    {
      "epoch": 6.0593605041503905e-05,
      "model_forward_time": 1.7171647548675537,
      "step": 63537
    },
    {
      "epoch": 6.0593605041503905e-05,
      "step": 63537,
      "training_step_time": 7.439655065536499
    },
    {
      "epoch": 6.059455871582031e-05,
      "model_forward_time": 1.7327618598937988,
      "step": 63538
    },
    {
      "epoch": 6.059455871582031e-05,
      "step": 63538,
      "training_step_time": 7.578920841217041
    },
    {
      "epoch": 6.059551239013672e-05,
      "model_forward_time": 1.7303507328033447,
      "step": 63539
    },
    {
      "epoch": 6.059551239013672e-05,
      "step": 63539,
      "training_step_time": 7.7217395305633545
    },
    {
      "epoch": 6.0596466064453124e-05,
      "grad_norm": 0.0940505713224411,
      "learning_rate": 3.2150056774496363e-06,
      "loss": 0.0564,
      "step": 63540
    },
    {
      "epoch": 6.0596466064453124e-05,
      "model_forward_time": 1.7468321323394775,
      "step": 63540
    },
    {
      "dynamics_loss_avg": 0.04786136951297522,
      "epoch": 6.0596466064453124e-05,
      "step": 63540
    },
    {
      "action_loss_avg": 0.0014328077610116451,
      "epoch": 6.0596466064453124e-05,
      "step": 63540
    },
    {
      "epoch": 6.0596466064453124e-05,
      "step": 63540,
      "training_step_time": 8.456572771072388
    },
    {
      "epoch": 6.059741973876953e-05,
      "model_forward_time": 1.729527473449707,
      "step": 63541
    },
    {
      "epoch": 6.059741973876953e-05,
      "step": 63541,
      "training_step_time": 8.637023687362671
    },
    {
      "epoch": 6.059837341308594e-05,
      "model_forward_time": 1.726161003112793,
      "step": 63542
    },
    {
      "epoch": 6.059837341308594e-05,
      "step": 63542,
      "training_step_time": 7.698086261749268
    },
    {
      "epoch": 6.059932708740234e-05,
      "model_forward_time": 1.7257452011108398,
      "step": 63543
    },
    {
      "epoch": 6.059932708740234e-05,
      "step": 63543,
      "training_step_time": 7.597951650619507
    },
    {
      "epoch": 6.060028076171875e-05,
      "model_forward_time": 1.727592945098877,
      "step": 63544
    },
    {
      "epoch": 6.060028076171875e-05,
      "step": 63544,
      "training_step_time": 7.460821866989136
    },
    {
      "epoch": 6.060123443603516e-05,
      "model_forward_time": 2.8912761211395264,
      "step": 63545
    },
    {
      "epoch": 6.060123443603516e-05,
      "step": 63545,
      "training_step_time": 7.740330219268799
    },
    {
      "epoch": 6.060218811035156e-05,
      "model_forward_time": 1.7195820808410645,
      "step": 63546
    },
    {
      "epoch": 6.060218811035156e-05,
      "step": 63546,
      "training_step_time": 7.507748603820801
    },
    {
      "epoch": 6.0603141784667966e-05,
      "model_forward_time": 1.729539155960083,
      "step": 63547
    },
    {
      "epoch": 6.0603141784667966e-05,
      "step": 63547,
      "training_step_time": 7.635152578353882
    },
    {
      "epoch": 6.060409545898438e-05,
      "model_forward_time": 1.749680757522583,
      "step": 63548
    },
    {
      "epoch": 6.060409545898438e-05,
      "step": 63548,
      "training_step_time": 8.606347799301147
    },
    {
      "epoch": 6.060504913330078e-05,
      "model_forward_time": 1.722029447555542,
      "step": 63549
    },
    {
      "epoch": 6.060504913330078e-05,
      "step": 63549,
      "training_step_time": 8.687345027923584
    },
    {
      "epoch": 6.0606002807617185e-05,
      "grad_norm": 0.10672316700220108,
      "learning_rate": 3.213461260957065e-06,
      "loss": 0.0572,
      "step": 63550
    },
    {
      "epoch": 6.0606002807617185e-05,
      "model_forward_time": 1.7609350681304932,
      "step": 63550
    },
    {
      "dynamics_loss_avg": 0.06370199061930179,
      "epoch": 6.0606002807617185e-05,
      "step": 63550
    },
    {
      "action_loss_avg": 0.0018017239170148968,
      "epoch": 6.0606002807617185e-05,
      "step": 63550
    },
    {
      "epoch": 6.0606002807617185e-05,
      "step": 63550,
      "training_step_time": 7.670978307723999
    },
    {
      "epoch": 6.0606956481933596e-05,
      "model_forward_time": 1.7181508541107178,
      "step": 63551
    },
    {
      "epoch": 6.0606956481933596e-05,
      "step": 63551,
      "training_step_time": 7.6694910526275635
    },
    {
      "epoch": 6.060791015625e-05,
      "model_forward_time": 1.7371983528137207,
      "step": 63552
    },
    {
      "epoch": 6.060791015625e-05,
      "step": 63552,
      "training_step_time": 7.575109481811523
    },
    {
      "epoch": 6.0608863830566404e-05,
      "model_forward_time": 1.7239627838134766,
      "step": 63553
    },
    {
      "epoch": 6.0608863830566404e-05,
      "step": 63553,
      "training_step_time": 7.301882266998291
    },
    {
      "epoch": 6.0609817504882814e-05,
      "model_forward_time": 2.242466449737549,
      "step": 63554
    },
    {
      "epoch": 6.0609817504882814e-05,
      "step": 63554,
      "training_step_time": 7.486253261566162
    },
    {
      "epoch": 6.061077117919922e-05,
      "model_forward_time": 1.7290754318237305,
      "step": 63555
    },
    {
      "epoch": 6.061077117919922e-05,
      "step": 63555,
      "training_step_time": 7.5120930671691895
    },
    {
      "epoch": 6.061172485351562e-05,
      "model_forward_time": 1.7317750453948975,
      "step": 63556
    },
    {
      "epoch": 6.061172485351562e-05,
      "step": 63556,
      "training_step_time": 7.4538421630859375
    },
    {
      "epoch": 6.061267852783203e-05,
      "model_forward_time": 1.7301998138427734,
      "step": 63557
    },
    {
      "epoch": 6.061267852783203e-05,
      "step": 63557,
      "training_step_time": 8.882482290267944
    },
    {
      "epoch": 6.061363220214844e-05,
      "model_forward_time": 1.7543792724609375,
      "step": 63558
    },
    {
      "epoch": 6.061363220214844e-05,
      "step": 63558,
      "training_step_time": 8.292093276977539
    },
    {
      "epoch": 6.061458587646484e-05,
      "model_forward_time": 1.7497248649597168,
      "step": 63559
    },
    {
      "epoch": 6.061458587646484e-05,
      "step": 63559,
      "training_step_time": 8.682036876678467
    },
    {
      "epoch": 6.061553955078125e-05,
      "grad_norm": 0.1038978099822998,
      "learning_rate": 3.211917039837682e-06,
      "loss": 0.059,
      "step": 63560
    },
    {
      "epoch": 6.061553955078125e-05,
      "model_forward_time": 1.7213199138641357,
      "step": 63560
    },
    {
      "dynamics_loss_avg": 0.0702348243445158,
      "epoch": 6.061553955078125e-05,
      "step": 63560
    },
    {
      "action_loss_avg": 0.0021475327579537407,
      "epoch": 6.061553955078125e-05,
      "step": 63560
    },
    {
      "epoch": 6.061553955078125e-05,
      "step": 63560,
      "training_step_time": 7.741875410079956
    },
    {
      "epoch": 6.0616493225097656e-05,
      "model_forward_time": 1.7323193550109863,
      "step": 63561
    },
    {
      "epoch": 6.0616493225097656e-05,
      "step": 63561,
      "training_step_time": 8.738811016082764
    },
    {
      "epoch": 6.061744689941406e-05,
      "model_forward_time": 1.7291913032531738,
      "step": 63562
    },
    {
      "epoch": 6.061744689941406e-05,
      "step": 63562,
      "training_step_time": 7.544628381729126
    },
    {
      "epoch": 6.061840057373047e-05,
      "model_forward_time": 2.8469598293304443,
      "step": 63563
    },
    {
      "epoch": 6.061840057373047e-05,
      "step": 63563,
      "training_step_time": 7.604648590087891
    },
    {
      "epoch": 6.0619354248046875e-05,
      "model_forward_time": 1.7273974418640137,
      "step": 63564
    },
    {
      "epoch": 6.0619354248046875e-05,
      "step": 63564,
      "training_step_time": 7.4478325843811035
    },
    {
      "epoch": 6.062030792236328e-05,
      "model_forward_time": 1.7343909740447998,
      "step": 63565
    },
    {
      "epoch": 6.062030792236328e-05,
      "step": 63565,
      "training_step_time": 7.485043525695801
    },
    {
      "epoch": 6.062126159667969e-05,
      "model_forward_time": 1.7414531707763672,
      "step": 63566
    },
    {
      "epoch": 6.062126159667969e-05,
      "step": 63566,
      "training_step_time": 9.790403842926025
    },
    {
      "epoch": 6.0622215270996094e-05,
      "model_forward_time": 1.7523441314697266,
      "step": 63567
    },
    {
      "epoch": 6.0622215270996094e-05,
      "step": 63567,
      "training_step_time": 8.570642471313477
    },
    {
      "epoch": 6.06231689453125e-05,
      "model_forward_time": 1.74284029006958,
      "step": 63568
    },
    {
      "epoch": 6.06231689453125e-05,
      "step": 63568,
      "training_step_time": 7.663438558578491
    },
    {
      "epoch": 6.062412261962891e-05,
      "model_forward_time": 1.7348251342773438,
      "step": 63569
    },
    {
      "epoch": 6.062412261962891e-05,
      "step": 63569,
      "training_step_time": 7.741065502166748
    },
    {
      "epoch": 6.062507629394531e-05,
      "grad_norm": 0.11147576570510864,
      "learning_rate": 3.210373014260365e-06,
      "loss": 0.0594,
      "step": 63570
    },
    {
      "epoch": 6.062507629394531e-05,
      "model_forward_time": 1.731916904449463,
      "step": 63570
    },
    {
      "dynamics_loss_avg": 0.05767444726079703,
      "epoch": 6.062507629394531e-05,
      "step": 63570
    },
    {
      "action_loss_avg": 0.0017003651591949165,
      "epoch": 6.062507629394531e-05,
      "step": 63570
    },
    {
      "epoch": 6.062507629394531e-05,
      "step": 63570,
      "training_step_time": 7.564759969711304
    },
    {
      "epoch": 6.062602996826172e-05,
      "model_forward_time": 1.7260932922363281,
      "step": 63571
    },
    {
      "epoch": 6.062602996826172e-05,
      "step": 63571,
      "training_step_time": 7.584094047546387
    },
    {
      "epoch": 6.062698364257813e-05,
      "model_forward_time": 2.8338541984558105,
      "step": 63572
    },
    {
      "epoch": 6.062698364257813e-05,
      "step": 63572,
      "training_step_time": 7.568538188934326
    },
    {
      "epoch": 6.062793731689453e-05,
      "model_forward_time": 1.7280535697937012,
      "step": 63573
    },
    {
      "epoch": 6.062793731689453e-05,
      "step": 63573,
      "training_step_time": 7.450181007385254
    },
    {
      "epoch": 6.0628890991210936e-05,
      "model_forward_time": 1.7338385581970215,
      "step": 63574
    },
    {
      "epoch": 6.0628890991210936e-05,
      "step": 63574,
      "training_step_time": 8.130132675170898
    },
    {
      "epoch": 6.0629844665527346e-05,
      "model_forward_time": 1.726010799407959,
      "step": 63575
    },
    {
      "epoch": 6.0629844665527346e-05,
      "step": 63575,
      "training_step_time": 8.028450012207031
    },
    {
      "epoch": 6.063079833984375e-05,
      "model_forward_time": 1.738722801208496,
      "step": 63576
    },
    {
      "epoch": 6.063079833984375e-05,
      "step": 63576,
      "training_step_time": 9.154460668563843
    },
    {
      "epoch": 6.0631752014160155e-05,
      "model_forward_time": 1.8043055534362793,
      "step": 63577
    },
    {
      "epoch": 6.0631752014160155e-05,
      "step": 63577,
      "training_step_time": 8.172025918960571
    },
    {
      "epoch": 6.0632705688476565e-05,
      "model_forward_time": 1.8139050006866455,
      "step": 63578
    },
    {
      "epoch": 6.0632705688476565e-05,
      "step": 63578,
      "training_step_time": 8.670346021652222
    },
    {
      "epoch": 6.063365936279297e-05,
      "model_forward_time": 1.7422337532043457,
      "step": 63579
    },
    {
      "epoch": 6.063365936279297e-05,
      "step": 63579,
      "training_step_time": 7.73136305809021
    },
    {
      "epoch": 6.063461303710937e-05,
      "grad_norm": 0.09331351518630981,
      "learning_rate": 3.2088291843939653e-06,
      "loss": 0.0576,
      "step": 63580
    },
    {
      "epoch": 6.063461303710937e-05,
      "model_forward_time": 1.7541122436523438,
      "step": 63580
    },
    {
      "dynamics_loss_avg": 0.06556330882012844,
      "epoch": 6.063461303710937e-05,
      "step": 63580
    },
    {
      "action_loss_avg": 0.0014874582935590296,
      "epoch": 6.063461303710937e-05,
      "step": 63580
    },
    {
      "epoch": 6.063461303710937e-05,
      "step": 63580,
      "training_step_time": 7.562731027603149
    },
    {
      "epoch": 6.0635566711425784e-05,
      "model_forward_time": 1.740753412246704,
      "step": 63581
    },
    {
      "epoch": 6.0635566711425784e-05,
      "step": 63581,
      "training_step_time": 7.507716417312622
    },
    {
      "epoch": 6.063652038574219e-05,
      "model_forward_time": 2.899365186691284,
      "step": 63582
    },
    {
      "epoch": 6.063652038574219e-05,
      "step": 63582,
      "training_step_time": 7.623291015625
    },
    {
      "epoch": 6.063747406005859e-05,
      "model_forward_time": 1.7458739280700684,
      "step": 63583
    },
    {
      "epoch": 6.063747406005859e-05,
      "step": 63583,
      "training_step_time": 8.06348466873169
    },
    {
      "epoch": 6.0638427734375e-05,
      "model_forward_time": 1.7413923740386963,
      "step": 63584
    },
    {
      "epoch": 6.0638427734375e-05,
      "step": 63584,
      "training_step_time": 7.510028600692749
    },
    {
      "epoch": 6.063938140869141e-05,
      "model_forward_time": 1.7402284145355225,
      "step": 63585
    },
    {
      "epoch": 6.063938140869141e-05,
      "step": 63585,
      "training_step_time": 9.041094779968262
    },
    {
      "epoch": 6.064033508300781e-05,
      "model_forward_time": 1.7347009181976318,
      "step": 63586
    },
    {
      "epoch": 6.064033508300781e-05,
      "step": 63586,
      "training_step_time": 7.9044294357299805
    },
    {
      "epoch": 6.064128875732422e-05,
      "model_forward_time": 1.8085711002349854,
      "step": 63587
    },
    {
      "epoch": 6.064128875732422e-05,
      "step": 63587,
      "training_step_time": 7.811459302902222
    },
    {
      "epoch": 6.0642242431640626e-05,
      "model_forward_time": 1.7387042045593262,
      "step": 63588
    },
    {
      "epoch": 6.0642242431640626e-05,
      "step": 63588,
      "training_step_time": 7.9783689975738525
    },
    {
      "epoch": 6.064319610595703e-05,
      "model_forward_time": 1.741584062576294,
      "step": 63589
    },
    {
      "epoch": 6.064319610595703e-05,
      "step": 63589,
      "training_step_time": 8.443063735961914
    },
    {
      "epoch": 6.064414978027344e-05,
      "grad_norm": 0.1274036169052124,
      "learning_rate": 3.2072855504073124e-06,
      "loss": 0.0549,
      "step": 63590
    },
    {
      "epoch": 6.064414978027344e-05,
      "model_forward_time": 1.7444686889648438,
      "step": 63590
    },
    {
      "dynamics_loss_avg": 0.05162314865738153,
      "epoch": 6.064414978027344e-05,
      "step": 63590
    },
    {
      "action_loss_avg": 0.0012131415220210328,
      "epoch": 6.064414978027344e-05,
      "step": 63590
    },
    {
      "epoch": 6.064414978027344e-05,
      "step": 63590,
      "training_step_time": 7.759486198425293
    },
    {
      "epoch": 6.0645103454589845e-05,
      "model_forward_time": 2.9220597743988037,
      "step": 63591
    },
    {
      "epoch": 6.0645103454589845e-05,
      "step": 63591,
      "training_step_time": 7.882838726043701
    },
    {
      "epoch": 6.064605712890625e-05,
      "model_forward_time": 1.7470808029174805,
      "step": 63592
    },
    {
      "epoch": 6.064605712890625e-05,
      "step": 63592,
      "training_step_time": 8.333645105361938
    },
    {
      "epoch": 6.064701080322266e-05,
      "model_forward_time": 1.7430756092071533,
      "step": 63593
    },
    {
      "epoch": 6.064701080322266e-05,
      "step": 63593,
      "training_step_time": 7.657816410064697
    },
    {
      "epoch": 6.0647964477539064e-05,
      "model_forward_time": 1.7316765785217285,
      "step": 63594
    },
    {
      "epoch": 6.0647964477539064e-05,
      "step": 63594,
      "training_step_time": 7.997403621673584
    },
    {
      "epoch": 6.064891815185547e-05,
      "model_forward_time": 1.7469305992126465,
      "step": 63595
    },
    {
      "epoch": 6.064891815185547e-05,
      "step": 63595,
      "training_step_time": 9.638741493225098
    },
    {
      "epoch": 6.064987182617187e-05,
      "model_forward_time": 1.7444560527801514,
      "step": 63596
    },
    {
      "epoch": 6.064987182617187e-05,
      "step": 63596,
      "training_step_time": 7.798485040664673
    },
    {
      "epoch": 6.065082550048828e-05,
      "model_forward_time": 1.7362022399902344,
      "step": 63597
    },
    {
      "epoch": 6.065082550048828e-05,
      "step": 63597,
      "training_step_time": 7.898542642593384
    },
    {
      "epoch": 6.0651779174804687e-05,
      "model_forward_time": 1.741255283355713,
      "step": 63598
    },
    {
      "epoch": 6.0651779174804687e-05,
      "step": 63598,
      "training_step_time": 7.908753156661987
    },
    {
      "epoch": 6.065273284912109e-05,
      "model_forward_time": 1.7422518730163574,
      "step": 63599
    },
    {
      "epoch": 6.065273284912109e-05,
      "step": 63599,
      "training_step_time": 7.768580198287964
    },
    {
      "epoch": 6.06536865234375e-05,
      "grad_norm": 0.0979086309671402,
      "learning_rate": 3.205742112469217e-06,
      "loss": 0.0562,
      "step": 63600
    },
    {
      "epoch": 6.06536865234375e-05,
      "model_forward_time": 2.88804030418396,
      "step": 63600
    },
    {
      "dynamics_loss_avg": 0.07321617100387812,
      "epoch": 6.06536865234375e-05,
      "step": 63600
    },
    {
      "action_loss_avg": 0.0022538484598044308,
      "epoch": 6.06536865234375e-05,
      "step": 63600
    },
    {
      "epoch": 6.06536865234375e-05,
      "step": 63600,
      "training_step_time": 8.58455491065979
    },
    {
      "epoch": 6.0654640197753905e-05,
      "model_forward_time": 1.7442445755004883,
      "step": 63601
    },
    {
      "epoch": 6.0654640197753905e-05,
      "step": 63601,
      "training_step_time": 7.552804470062256
    },
    {
      "epoch": 6.065559387207031e-05,
      "model_forward_time": 1.7474944591522217,
      "step": 63602
    },
    {
      "epoch": 6.065559387207031e-05,
      "step": 63602,
      "training_step_time": 7.7246413230896
    },
    {
      "epoch": 6.065654754638672e-05,
      "model_forward_time": 1.7453727722167969,
      "step": 63603
    },
    {
      "epoch": 6.065654754638672e-05,
      "step": 63603,
      "training_step_time": 7.916772842407227
    },
    {
      "epoch": 6.0657501220703124e-05,
      "model_forward_time": 1.8218739032745361,
      "step": 63604
    },
    {
      "epoch": 6.0657501220703124e-05,
      "step": 63604,
      "training_step_time": 7.953183174133301
    },
    {
      "epoch": 6.065845489501953e-05,
      "model_forward_time": 1.748809814453125,
      "step": 63605
    },
    {
      "epoch": 6.065845489501953e-05,
      "step": 63605,
      "training_step_time": 7.822197198867798
    },
    {
      "epoch": 6.065940856933594e-05,
      "model_forward_time": 1.756871223449707,
      "step": 63606
    },
    {
      "epoch": 6.065940856933594e-05,
      "step": 63606,
      "training_step_time": 8.977794408798218
    },
    {
      "epoch": 6.066036224365234e-05,
      "model_forward_time": 1.7529847621917725,
      "step": 63607
    },
    {
      "epoch": 6.066036224365234e-05,
      "step": 63607,
      "training_step_time": 7.899816989898682
    },
    {
      "epoch": 6.066131591796875e-05,
      "model_forward_time": 1.7639219760894775,
      "step": 63608
    },
    {
      "epoch": 6.066131591796875e-05,
      "step": 63608,
      "training_step_time": 7.824342250823975
    },
    {
      "epoch": 6.066226959228516e-05,
      "model_forward_time": 2.8973617553710938,
      "step": 63609
    },
    {
      "epoch": 6.066226959228516e-05,
      "step": 63609,
      "training_step_time": 7.895922899246216
    },
    {
      "epoch": 6.066322326660156e-05,
      "grad_norm": 0.09252657741308212,
      "learning_rate": 3.2041988707484655e-06,
      "loss": 0.054,
      "step": 63610
    },
    {
      "epoch": 6.066322326660156e-05,
      "model_forward_time": 1.7947652339935303,
      "step": 63610
    },
    {
      "dynamics_loss_avg": 0.05487470123916864,
      "epoch": 6.066322326660156e-05,
      "step": 63610
    },
    {
      "action_loss_avg": 0.0016354754334315658,
      "epoch": 6.066322326660156e-05,
      "step": 63610
    },
    {
      "epoch": 6.066322326660156e-05,
      "step": 63610,
      "training_step_time": 7.53808856010437
    },
    {
      "epoch": 6.0664176940917966e-05,
      "model_forward_time": 1.7455062866210938,
      "step": 63611
    },
    {
      "epoch": 6.0664176940917966e-05,
      "step": 63611,
      "training_step_time": 7.749569416046143
    },
    {
      "epoch": 6.066513061523438e-05,
      "model_forward_time": 1.748408317565918,
      "step": 63612
    },
    {
      "epoch": 6.066513061523438e-05,
      "step": 63612,
      "training_step_time": 7.853474140167236
    },
    {
      "epoch": 6.066608428955078e-05,
      "model_forward_time": 1.7660472393035889,
      "step": 63613
    },
    {
      "epoch": 6.066608428955078e-05,
      "step": 63613,
      "training_step_time": 8.03450322151184
    },
    {
      "epoch": 6.0667037963867185e-05,
      "model_forward_time": 1.7264719009399414,
      "step": 63614
    },
    {
      "epoch": 6.0667037963867185e-05,
      "step": 63614,
      "training_step_time": 8.043449878692627
    },
    {
      "epoch": 6.0667991638183596e-05,
      "model_forward_time": 1.7265844345092773,
      "step": 63615
    },
    {
      "epoch": 6.0667991638183596e-05,
      "step": 63615,
      "training_step_time": 8.515182733535767
    },
    {
      "epoch": 6.06689453125e-05,
      "model_forward_time": 1.7033329010009766,
      "step": 63616
    },
    {
      "epoch": 6.06689453125e-05,
      "step": 63616,
      "training_step_time": 7.861109018325806
    },
    {
      "epoch": 6.0669898986816404e-05,
      "model_forward_time": 1.7280516624450684,
      "step": 63617
    },
    {
      "epoch": 6.0669898986816404e-05,
      "step": 63617,
      "training_step_time": 7.865784645080566
    },
    {
      "epoch": 6.0670852661132815e-05,
      "model_forward_time": 2.312655448913574,
      "step": 63618
    },
    {
      "epoch": 6.0670852661132815e-05,
      "step": 63618,
      "training_step_time": 8.684396982192993
    },
    {
      "epoch": 6.067180633544922e-05,
      "model_forward_time": 1.722151279449463,
      "step": 63619
    },
    {
      "epoch": 6.067180633544922e-05,
      "step": 63619,
      "training_step_time": 7.7833757400512695
    },
    {
      "epoch": 6.067276000976562e-05,
      "grad_norm": 0.11865115165710449,
      "learning_rate": 3.202655825413824e-06,
      "loss": 0.0553,
      "step": 63620
    },
    {
      "epoch": 6.067276000976562e-05,
      "model_forward_time": 1.7202284336090088,
      "step": 63620
    },
    {
      "dynamics_loss_avg": 0.06138049717992544,
      "epoch": 6.067276000976562e-05,
      "step": 63620
    },
    {
      "action_loss_avg": 0.001688788816682063,
      "epoch": 6.067276000976562e-05,
      "step": 63620
    },
    {
      "epoch": 6.067276000976562e-05,
      "step": 63620,
      "training_step_time": 7.777503967285156
    },
    {
      "epoch": 6.067371368408203e-05,
      "model_forward_time": 1.73858642578125,
      "step": 63621
    },
    {
      "epoch": 6.067371368408203e-05,
      "step": 63621,
      "training_step_time": 8.5496244430542
    },
    {
      "epoch": 6.067466735839844e-05,
      "model_forward_time": 1.7262814044952393,
      "step": 63622
    },
    {
      "epoch": 6.067466735839844e-05,
      "step": 63622,
      "training_step_time": 8.893579483032227
    },
    {
      "epoch": 6.067562103271484e-05,
      "model_forward_time": 1.7651612758636475,
      "step": 63623
    },
    {
      "epoch": 6.067562103271484e-05,
      "step": 63623,
      "training_step_time": 9.0628080368042
    },
    {
      "epoch": 6.067657470703125e-05,
      "model_forward_time": 1.7138216495513916,
      "step": 63624
    },
    {
      "epoch": 6.067657470703125e-05,
      "step": 63624,
      "training_step_time": 9.086992502212524
    },
    {
      "epoch": 6.0677528381347656e-05,
      "model_forward_time": 1.6960334777832031,
      "step": 63625
    },
    {
      "epoch": 6.0677528381347656e-05,
      "step": 63625,
      "training_step_time": 7.719189167022705
    },
    {
      "epoch": 6.067848205566406e-05,
      "model_forward_time": 1.7205226421356201,
      "step": 63626
    },
    {
      "epoch": 6.067848205566406e-05,
      "step": 63626,
      "training_step_time": 7.581012725830078
    },
    {
      "epoch": 6.067943572998047e-05,
      "model_forward_time": 1.7221653461456299,
      "step": 63627
    },
    {
      "epoch": 6.067943572998047e-05,
      "step": 63627,
      "training_step_time": 8.057945489883423
    },
    {
      "epoch": 6.0680389404296875e-05,
      "model_forward_time": 1.7184932231903076,
      "step": 63628
    },
    {
      "epoch": 6.0680389404296875e-05,
      "step": 63628,
      "training_step_time": 7.523826599121094
    },
    {
      "epoch": 6.068134307861328e-05,
      "model_forward_time": 1.7223539352416992,
      "step": 63629
    },
    {
      "epoch": 6.068134307861328e-05,
      "step": 63629,
      "training_step_time": 7.4722137451171875
    },
    {
      "epoch": 6.068229675292969e-05,
      "grad_norm": 0.07874546945095062,
      "learning_rate": 3.2011129766340416e-06,
      "loss": 0.0568,
      "step": 63630
    },
    {
      "epoch": 6.068229675292969e-05,
      "model_forward_time": 1.7142713069915771,
      "step": 63630
    },
    {
      "dynamics_loss_avg": 0.07150879800319672,
      "epoch": 6.068229675292969e-05,
      "step": 63630
    },
    {
      "action_loss_avg": 0.0017165966331958772,
      "epoch": 6.068229675292969e-05,
      "step": 63630
    },
    {
      "epoch": 6.068229675292969e-05,
      "step": 63630,
      "training_step_time": 7.5785229206085205
    },
    {
      "epoch": 6.0683250427246094e-05,
      "model_forward_time": 1.7203352451324463,
      "step": 63631
    },
    {
      "epoch": 6.0683250427246094e-05,
      "step": 63631,
      "training_step_time": 9.253690004348755
    },
    {
      "epoch": 6.06842041015625e-05,
      "model_forward_time": 1.7596168518066406,
      "step": 63632
    },
    {
      "epoch": 6.06842041015625e-05,
      "step": 63632,
      "training_step_time": 7.71396017074585
    },
    {
      "epoch": 6.068515777587891e-05,
      "model_forward_time": 1.7305457592010498,
      "step": 63633
    },
    {
      "epoch": 6.068515777587891e-05,
      "step": 63633,
      "training_step_time": 8.776881217956543
    },
    {
      "epoch": 6.068611145019531e-05,
      "model_forward_time": 1.7225422859191895,
      "step": 63634
    },
    {
      "epoch": 6.068611145019531e-05,
      "step": 63634,
      "training_step_time": 8.538709163665771
    },
    {
      "epoch": 6.068706512451172e-05,
      "model_forward_time": 1.7322185039520264,
      "step": 63635
    },
    {
      "epoch": 6.068706512451172e-05,
      "step": 63635,
      "training_step_time": 7.757640600204468
    },
    {
      "epoch": 6.068801879882813e-05,
      "model_forward_time": 1.7212071418762207,
      "step": 63636
    },
    {
      "epoch": 6.068801879882813e-05,
      "step": 63636,
      "training_step_time": 7.721389055252075
    },
    {
      "epoch": 6.068897247314453e-05,
      "model_forward_time": 2.2650060653686523,
      "step": 63637
    },
    {
      "epoch": 6.068897247314453e-05,
      "step": 63637,
      "training_step_time": 7.595783710479736
    },
    {
      "epoch": 6.0689926147460936e-05,
      "model_forward_time": 1.7265419960021973,
      "step": 63638
    },
    {
      "epoch": 6.0689926147460936e-05,
      "step": 63638,
      "training_step_time": 8.1438729763031
    },
    {
      "epoch": 6.0690879821777347e-05,
      "model_forward_time": 1.7182059288024902,
      "step": 63639
    },
    {
      "epoch": 6.0690879821777347e-05,
      "step": 63639,
      "training_step_time": 8.42212176322937
    },
    {
      "epoch": 6.069183349609375e-05,
      "grad_norm": 0.30288586020469666,
      "learning_rate": 3.199570324577839e-06,
      "loss": 0.0529,
      "step": 63640
    },
    {
      "epoch": 6.069183349609375e-05,
      "model_forward_time": 1.7204415798187256,
      "step": 63640
    },
    {
      "dynamics_loss_avg": 0.06572226770222186,
      "epoch": 6.069183349609375e-05,
      "step": 63640
    },
    {
      "action_loss_avg": 0.0021214592328760774,
      "epoch": 6.069183349609375e-05,
      "step": 63640
    },
    {
      "epoch": 6.069183349609375e-05,
      "step": 63640,
      "training_step_time": 8.919968605041504
    },
    {
      "epoch": 6.0692787170410155e-05,
      "model_forward_time": 1.7693853378295898,
      "step": 63641
    },
    {
      "epoch": 6.0692787170410155e-05,
      "step": 63641,
      "training_step_time": 7.784654378890991
    },
    {
      "epoch": 6.0693740844726565e-05,
      "model_forward_time": 1.7239398956298828,
      "step": 63642
    },
    {
      "epoch": 6.0693740844726565e-05,
      "step": 63642,
      "training_step_time": 9.288732767105103
    },
    {
      "epoch": 6.069469451904297e-05,
      "model_forward_time": 1.7246737480163574,
      "step": 63643
    },
    {
      "epoch": 6.069469451904297e-05,
      "step": 63643,
      "training_step_time": 7.910011053085327
    },
    {
      "epoch": 6.0695648193359373e-05,
      "model_forward_time": 1.7254579067230225,
      "step": 63644
    },
    {
      "epoch": 6.0695648193359373e-05,
      "step": 63644,
      "training_step_time": 8.521303176879883
    },
    {
      "epoch": 6.0696601867675784e-05,
      "model_forward_time": 1.726701021194458,
      "step": 63645
    },
    {
      "epoch": 6.0696601867675784e-05,
      "step": 63645,
      "training_step_time": 7.754419565200806
    },
    {
      "epoch": 6.069755554199219e-05,
      "model_forward_time": 2.280252456665039,
      "step": 63646
    },
    {
      "epoch": 6.069755554199219e-05,
      "step": 63646,
      "training_step_time": 7.6088807582855225
    },
    {
      "epoch": 6.069850921630859e-05,
      "model_forward_time": 1.7221543788909912,
      "step": 63647
    },
    {
      "epoch": 6.069850921630859e-05,
      "step": 63647,
      "training_step_time": 8.580339908599854
    },
    {
      "epoch": 6.0699462890625e-05,
      "model_forward_time": 1.7208170890808105,
      "step": 63648
    },
    {
      "epoch": 6.0699462890625e-05,
      "step": 63648,
      "training_step_time": 7.57235050201416
    },
    {
      "epoch": 6.070041656494141e-05,
      "model_forward_time": 1.7266504764556885,
      "step": 63649
    },
    {
      "epoch": 6.070041656494141e-05,
      "step": 63649,
      "training_step_time": 7.771405935287476
    },
    {
      "epoch": 6.070137023925781e-05,
      "grad_norm": 0.09734687209129333,
      "learning_rate": 3.1980278694139186e-06,
      "loss": 0.0558,
      "step": 63650
    },
    {
      "epoch": 6.070137023925781e-05,
      "model_forward_time": 1.7641668319702148,
      "step": 63650
    },
    {
      "dynamics_loss_avg": 0.06730213416740298,
      "epoch": 6.070137023925781e-05,
      "step": 63650
    },
    {
      "action_loss_avg": 0.0021751197637058794,
      "epoch": 6.070137023925781e-05,
      "step": 63650
    },
    {
      "epoch": 6.070137023925781e-05,
      "step": 63650,
      "training_step_time": 7.79596734046936
    },
    {
      "epoch": 6.070232391357422e-05,
      "model_forward_time": 1.7216706275939941,
      "step": 63651
    },
    {
      "epoch": 6.070232391357422e-05,
      "step": 63651,
      "training_step_time": 7.689629793167114
    },
    {
      "epoch": 6.0703277587890626e-05,
      "model_forward_time": 1.722717046737671,
      "step": 63652
    },
    {
      "epoch": 6.0703277587890626e-05,
      "step": 63652,
      "training_step_time": 9.30280327796936
    },
    {
      "epoch": 6.070423126220703e-05,
      "model_forward_time": 1.7207672595977783,
      "step": 63653
    },
    {
      "epoch": 6.070423126220703e-05,
      "step": 63653,
      "training_step_time": 7.616473913192749
    },
    {
      "epoch": 6.070518493652344e-05,
      "model_forward_time": 1.7239255905151367,
      "step": 63654
    },
    {
      "epoch": 6.070518493652344e-05,
      "step": 63654,
      "training_step_time": 7.392762899398804
    },
    {
      "epoch": 6.0706138610839845e-05,
      "model_forward_time": 2.2876453399658203,
      "step": 63655
    },
    {
      "epoch": 6.0706138610839845e-05,
      "step": 63655,
      "training_step_time": 7.570042133331299
    },
    {
      "epoch": 6.070709228515625e-05,
      "model_forward_time": 1.7121648788452148,
      "step": 63656
    },
    {
      "epoch": 6.070709228515625e-05,
      "step": 63656,
      "training_step_time": 7.467360973358154
    },
    {
      "epoch": 6.070804595947265e-05,
      "model_forward_time": 1.7283973693847656,
      "step": 63657
    },
    {
      "epoch": 6.070804595947265e-05,
      "step": 63657,
      "training_step_time": 7.511094570159912
    },
    {
      "epoch": 6.0708999633789064e-05,
      "model_forward_time": 1.726961612701416,
      "step": 63658
    },
    {
      "epoch": 6.0708999633789064e-05,
      "step": 63658,
      "training_step_time": 7.695528745651245
    },
    {
      "epoch": 6.070995330810547e-05,
      "model_forward_time": 1.776585340499878,
      "step": 63659
    },
    {
      "epoch": 6.070995330810547e-05,
      "step": 63659,
      "training_step_time": 8.561874866485596
    },
    {
      "epoch": 6.071090698242187e-05,
      "grad_norm": 0.07983153313398361,
      "learning_rate": 3.196485611310962e-06,
      "loss": 0.0516,
      "step": 63660
    },
    {
      "epoch": 6.071090698242187e-05,
      "model_forward_time": 1.71364164352417,
      "step": 63660
    },
    {
      "dynamics_loss_avg": 0.04105050163343549,
      "epoch": 6.071090698242187e-05,
      "step": 63660
    },
    {
      "action_loss_avg": 0.0010338949767174199,
      "epoch": 6.071090698242187e-05,
      "step": 63660
    },
    {
      "epoch": 6.071090698242187e-05,
      "step": 63660,
      "training_step_time": 7.6481664180755615
    },
    {
      "epoch": 6.071186065673828e-05,
      "model_forward_time": 1.7277295589447021,
      "step": 63661
    },
    {
      "epoch": 6.071186065673828e-05,
      "step": 63661,
      "training_step_time": 8.726448774337769
    },
    {
      "epoch": 6.0712814331054687e-05,
      "model_forward_time": 1.7076044082641602,
      "step": 63662
    },
    {
      "epoch": 6.0712814331054687e-05,
      "step": 63662,
      "training_step_time": 7.783404111862183
    },
    {
      "epoch": 6.071376800537109e-05,
      "model_forward_time": 1.7285921573638916,
      "step": 63663
    },
    {
      "epoch": 6.071376800537109e-05,
      "step": 63663,
      "training_step_time": 7.493733644485474
    },
    {
      "epoch": 6.07147216796875e-05,
      "model_forward_time": 2.299741744995117,
      "step": 63664
    },
    {
      "epoch": 6.07147216796875e-05,
      "step": 63664,
      "training_step_time": 7.612191915512085
    },
    {
      "epoch": 6.0715675354003905e-05,
      "model_forward_time": 1.7225122451782227,
      "step": 63665
    },
    {
      "epoch": 6.0715675354003905e-05,
      "step": 63665,
      "training_step_time": 7.555242538452148
    },
    {
      "epoch": 6.071662902832031e-05,
      "model_forward_time": 1.715766429901123,
      "step": 63666
    },
    {
      "epoch": 6.071662902832031e-05,
      "step": 63666,
      "training_step_time": 8.16469955444336
    },
    {
      "epoch": 6.071758270263672e-05,
      "model_forward_time": 1.730461597442627,
      "step": 63667
    },
    {
      "epoch": 6.071758270263672e-05,
      "step": 63667,
      "training_step_time": 9.997920036315918
    },
    {
      "epoch": 6.0718536376953124e-05,
      "model_forward_time": 1.7677593231201172,
      "step": 63668
    },
    {
      "epoch": 6.0718536376953124e-05,
      "step": 63668,
      "training_step_time": 10.808211088180542
    },
    {
      "epoch": 6.071949005126953e-05,
      "model_forward_time": 1.722107172012329,
      "step": 63669
    },
    {
      "epoch": 6.071949005126953e-05,
      "step": 63669,
      "training_step_time": 7.675893306732178
    },
    {
      "epoch": 6.072044372558594e-05,
      "grad_norm": 0.08852306008338928,
      "learning_rate": 3.194943550437626e-06,
      "loss": 0.0553,
      "step": 63670
    },
    {
      "epoch": 6.072044372558594e-05,
      "model_forward_time": 1.7159380912780762,
      "step": 63670
    },
    {
      "dynamics_loss_avg": 0.044766866229474546,
      "epoch": 6.072044372558594e-05,
      "step": 63670
    },
    {
      "action_loss_avg": 0.001396259036846459,
      "epoch": 6.072044372558594e-05,
      "step": 63670
    },
    {
      "epoch": 6.072044372558594e-05,
      "step": 63670,
      "training_step_time": 8.583693265914917
    },
    {
      "epoch": 6.072139739990234e-05,
      "model_forward_time": 1.7269377708435059,
      "step": 63671
    },
    {
      "epoch": 6.072139739990234e-05,
      "step": 63671,
      "training_step_time": 9.398713111877441
    },
    {
      "epoch": 6.072235107421875e-05,
      "model_forward_time": 1.7278554439544678,
      "step": 63672
    },
    {
      "epoch": 6.072235107421875e-05,
      "step": 63672,
      "training_step_time": 8.59375262260437
    },
    {
      "epoch": 6.072330474853516e-05,
      "model_forward_time": 2.2714898586273193,
      "step": 63673
    },
    {
      "epoch": 6.072330474853516e-05,
      "step": 63673,
      "training_step_time": 7.624650001525879
    },
    {
      "epoch": 6.072425842285156e-05,
      "model_forward_time": 1.7226006984710693,
      "step": 63674
    },
    {
      "epoch": 6.072425842285156e-05,
      "step": 63674,
      "training_step_time": 7.670270204544067
    },
    {
      "epoch": 6.0725212097167966e-05,
      "model_forward_time": 1.7297422885894775,
      "step": 63675
    },
    {
      "epoch": 6.0725212097167966e-05,
      "step": 63675,
      "training_step_time": 8.482393980026245
    },
    {
      "epoch": 6.072616577148438e-05,
      "model_forward_time": 1.7265424728393555,
      "step": 63676
    },
    {
      "epoch": 6.072616577148438e-05,
      "step": 63676,
      "training_step_time": 7.591829061508179
    },
    {
      "epoch": 6.072711944580078e-05,
      "model_forward_time": 1.7755272388458252,
      "step": 63677
    },
    {
      "epoch": 6.072711944580078e-05,
      "step": 63677,
      "training_step_time": 8.254042863845825
    },
    {
      "epoch": 6.0728073120117185e-05,
      "model_forward_time": 1.7151744365692139,
      "step": 63678
    },
    {
      "epoch": 6.0728073120117185e-05,
      "step": 63678,
      "training_step_time": 7.736558198928833
    },
    {
      "epoch": 6.0729026794433596e-05,
      "model_forward_time": 1.7297463417053223,
      "step": 63679
    },
    {
      "epoch": 6.0729026794433596e-05,
      "step": 63679,
      "training_step_time": 9.826102256774902
    },
    {
      "epoch": 6.072998046875e-05,
      "grad_norm": 0.11500471085309982,
      "learning_rate": 3.1934016869625506e-06,
      "loss": 0.0542,
      "step": 63680
    },
    {
      "epoch": 6.072998046875e-05,
      "model_forward_time": 1.7167155742645264,
      "step": 63680
    },
    {
      "dynamics_loss_avg": 0.07033413238823413,
      "epoch": 6.072998046875e-05,
      "step": 63680
    },
    {
      "action_loss_avg": 0.001592734025325626,
      "epoch": 6.072998046875e-05,
      "step": 63680
    },
    {
      "epoch": 6.072998046875e-05,
      "step": 63680,
      "training_step_time": 8.54215145111084
    },
    {
      "epoch": 6.0730934143066404e-05,
      "model_forward_time": 1.7258939743041992,
      "step": 63681
    },
    {
      "epoch": 6.0730934143066404e-05,
      "step": 63681,
      "training_step_time": 7.750205993652344
    },
    {
      "epoch": 6.0731887817382815e-05,
      "model_forward_time": 1.7407264709472656,
      "step": 63682
    },
    {
      "epoch": 6.0731887817382815e-05,
      "step": 63682,
      "training_step_time": 7.374107837677002
    },
    {
      "epoch": 6.073284149169922e-05,
      "model_forward_time": 2.283665657043457,
      "step": 63683
    },
    {
      "epoch": 6.073284149169922e-05,
      "step": 63683,
      "training_step_time": 9.21616816520691
    },
    {
      "epoch": 6.073379516601562e-05,
      "model_forward_time": 1.7386219501495361,
      "step": 63684
    },
    {
      "epoch": 6.073379516601562e-05,
      "step": 63684,
      "training_step_time": 7.244189262390137
    },
    {
      "epoch": 6.0734748840332033e-05,
      "model_forward_time": 1.7357456684112549,
      "step": 63685
    },
    {
      "epoch": 6.0734748840332033e-05,
      "step": 63685,
      "training_step_time": 8.977900266647339
    },
    {
      "epoch": 6.073570251464844e-05,
      "model_forward_time": 1.7751994132995605,
      "step": 63686
    },
    {
      "epoch": 6.073570251464844e-05,
      "step": 63686,
      "training_step_time": 7.8188817501068115
    },
    {
      "epoch": 6.073665618896484e-05,
      "model_forward_time": 1.7358336448669434,
      "step": 63687
    },
    {
      "epoch": 6.073665618896484e-05,
      "step": 63687,
      "training_step_time": 7.824660778045654
    },
    {
      "epoch": 6.073760986328125e-05,
      "model_forward_time": 1.7352790832519531,
      "step": 63688
    },
    {
      "epoch": 6.073760986328125e-05,
      "step": 63688,
      "training_step_time": 7.722994804382324
    },
    {
      "epoch": 6.0738563537597656e-05,
      "model_forward_time": 1.7290880680084229,
      "step": 63689
    },
    {
      "epoch": 6.0738563537597656e-05,
      "step": 63689,
      "training_step_time": 8.715607643127441
    },
    {
      "epoch": 6.073951721191406e-05,
      "grad_norm": 0.08522401750087738,
      "learning_rate": 3.1918600210543495e-06,
      "loss": 0.0519,
      "step": 63690
    },
    {
      "epoch": 6.073951721191406e-05,
      "model_forward_time": 1.7225291728973389,
      "step": 63690
    },
    {
      "dynamics_loss_avg": 0.056957945600152014,
      "epoch": 6.073951721191406e-05,
      "step": 63690
    },
    {
      "action_loss_avg": 0.0014578226982848719,
      "epoch": 6.073951721191406e-05,
      "step": 63690
    },
    {
      "epoch": 6.073951721191406e-05,
      "step": 63690,
      "training_step_time": 7.760808229446411
    },
    {
      "epoch": 6.074047088623047e-05,
      "model_forward_time": 1.7263894081115723,
      "step": 63691
    },
    {
      "epoch": 6.074047088623047e-05,
      "step": 63691,
      "training_step_time": 8.039942026138306
    },
    {
      "epoch": 6.0741424560546875e-05,
      "model_forward_time": 2.300265073776245,
      "step": 63692
    },
    {
      "epoch": 6.0741424560546875e-05,
      "step": 63692,
      "training_step_time": 7.590660095214844
    },
    {
      "epoch": 6.074237823486328e-05,
      "model_forward_time": 1.723564863204956,
      "step": 63693
    },
    {
      "epoch": 6.074237823486328e-05,
      "step": 63693,
      "training_step_time": 7.532235145568848
    },
    {
      "epoch": 6.074333190917969e-05,
      "model_forward_time": 1.7218201160430908,
      "step": 63694
    },
    {
      "epoch": 6.074333190917969e-05,
      "step": 63694,
      "training_step_time": 8.144004106521606
    },
    {
      "epoch": 6.0744285583496094e-05,
      "model_forward_time": 1.7689933776855469,
      "step": 63695
    },
    {
      "epoch": 6.0744285583496094e-05,
      "step": 63695,
      "training_step_time": 7.657102584838867
    },
    {
      "epoch": 6.07452392578125e-05,
      "model_forward_time": 1.7200872898101807,
      "step": 63696
    },
    {
      "epoch": 6.07452392578125e-05,
      "step": 63696,
      "training_step_time": 8.716588020324707
    },
    {
      "epoch": 6.074619293212891e-05,
      "model_forward_time": 1.7103161811828613,
      "step": 63697
    },
    {
      "epoch": 6.074619293212891e-05,
      "step": 63697,
      "training_step_time": 8.988615274429321
    },
    {
      "epoch": 6.074714660644531e-05,
      "model_forward_time": 1.7207458019256592,
      "step": 63698
    },
    {
      "epoch": 6.074714660644531e-05,
      "step": 63698,
      "training_step_time": 7.666644811630249
    },
    {
      "epoch": 6.074810028076172e-05,
      "model_forward_time": 1.721796989440918,
      "step": 63699
    },
    {
      "epoch": 6.074810028076172e-05,
      "step": 63699,
      "training_step_time": 8.627491474151611
    },
    {
      "epoch": 6.074905395507813e-05,
      "grad_norm": 0.09137201309204102,
      "learning_rate": 3.1903185528816183e-06,
      "loss": 0.056,
      "step": 63700
    },
    {
      "epoch": 6.074905395507813e-05,
      "model_forward_time": 1.7221336364746094,
      "step": 63700
    },
    {
      "dynamics_loss_avg": 0.06230578999966383,
      "epoch": 6.074905395507813e-05,
      "step": 63700
    },
    {
      "action_loss_avg": 0.0012066730472724886,
      "epoch": 6.074905395507813e-05,
      "step": 63700
    },
    {
      "epoch": 6.074905395507813e-05,
      "step": 63700,
      "training_step_time": 7.47596287727356
    },
    {
      "epoch": 6.075000762939453e-05,
      "model_forward_time": 2.2821435928344727,
      "step": 63701
    },
    {
      "epoch": 6.075000762939453e-05,
      "step": 63701,
      "training_step_time": 7.602358818054199
    },
    {
      "epoch": 6.0750961303710936e-05,
      "model_forward_time": 1.7239089012145996,
      "step": 63702
    },
    {
      "epoch": 6.0750961303710936e-05,
      "step": 63702,
      "training_step_time": 7.527455806732178
    },
    {
      "epoch": 6.0751914978027347e-05,
      "model_forward_time": 1.7305877208709717,
      "step": 63703
    },
    {
      "epoch": 6.0751914978027347e-05,
      "step": 63703,
      "training_step_time": 8.411922693252563
    },
    {
      "epoch": 6.075286865234375e-05,
      "model_forward_time": 1.7665045261383057,
      "step": 63704
    },
    {
      "epoch": 6.075286865234375e-05,
      "step": 63704,
      "training_step_time": 8.598886728286743
    },
    {
      "epoch": 6.0753822326660155e-05,
      "model_forward_time": 1.7336766719818115,
      "step": 63705
    },
    {
      "epoch": 6.0753822326660155e-05,
      "step": 63705,
      "training_step_time": 8.70843243598938
    },
    {
      "epoch": 6.0754776000976565e-05,
      "model_forward_time": 1.7283833026885986,
      "step": 63706
    },
    {
      "epoch": 6.0754776000976565e-05,
      "step": 63706,
      "training_step_time": 8.63474440574646
    },
    {
      "epoch": 6.075572967529297e-05,
      "model_forward_time": 1.7287395000457764,
      "step": 63707
    },
    {
      "epoch": 6.075572967529297e-05,
      "step": 63707,
      "training_step_time": 9.828427791595459
    },
    {
      "epoch": 6.0756683349609373e-05,
      "model_forward_time": 1.723947525024414,
      "step": 63708
    },
    {
      "epoch": 6.0756683349609373e-05,
      "step": 63708,
      "training_step_time": 8.755879878997803
    },
    {
      "epoch": 6.0757637023925784e-05,
      "model_forward_time": 1.7306513786315918,
      "step": 63709
    },
    {
      "epoch": 6.0757637023925784e-05,
      "step": 63709,
      "training_step_time": 7.507680654525757
    },
    {
      "epoch": 6.075859069824219e-05,
      "grad_norm": 0.09198317676782608,
      "learning_rate": 3.1887772826129285e-06,
      "loss": 0.0568,
      "step": 63710
    },
    {
      "epoch": 6.075859069824219e-05,
      "model_forward_time": 2.3132615089416504,
      "step": 63710
    },
    {
      "dynamics_loss_avg": 0.04866240937262774,
      "epoch": 6.075859069824219e-05,
      "step": 63710
    },
    {
      "action_loss_avg": 0.0015262728091329336,
      "epoch": 6.075859069824219e-05,
      "step": 63710
    },
    {
      "epoch": 6.075859069824219e-05,
      "step": 63710,
      "training_step_time": 8.183868646621704
    },
    {
      "epoch": 6.075954437255859e-05,
      "model_forward_time": 1.7270803451538086,
      "step": 63711
    },
    {
      "epoch": 6.075954437255859e-05,
      "step": 63711,
      "training_step_time": 7.640316963195801
    },
    {
      "epoch": 6.0760498046875e-05,
      "model_forward_time": 1.7142114639282227,
      "step": 63712
    },
    {
      "epoch": 6.0760498046875e-05,
      "step": 63712,
      "training_step_time": 7.519618272781372
    },
    {
      "epoch": 6.076145172119141e-05,
      "model_forward_time": 1.7107353210449219,
      "step": 63713
    },
    {
      "epoch": 6.076145172119141e-05,
      "step": 63713,
      "training_step_time": 7.504633665084839
    },
    {
      "epoch": 6.076240539550781e-05,
      "model_forward_time": 1.7226028442382812,
      "step": 63714
    },
    {
      "epoch": 6.076240539550781e-05,
      "step": 63714,
      "training_step_time": 7.573900461196899
    },
    {
      "epoch": 6.076335906982422e-05,
      "model_forward_time": 1.7168076038360596,
      "step": 63715
    },
    {
      "epoch": 6.076335906982422e-05,
      "step": 63715,
      "training_step_time": 8.535913705825806
    },
    {
      "epoch": 6.0764312744140626e-05,
      "model_forward_time": 1.7187175750732422,
      "step": 63716
    },
    {
      "epoch": 6.0764312744140626e-05,
      "step": 63716,
      "training_step_time": 8.2836172580719
    },
    {
      "epoch": 6.076526641845703e-05,
      "model_forward_time": 1.7134885787963867,
      "step": 63717
    },
    {
      "epoch": 6.076526641845703e-05,
      "step": 63717,
      "training_step_time": 7.721231460571289
    },
    {
      "epoch": 6.076622009277344e-05,
      "model_forward_time": 1.7100083827972412,
      "step": 63718
    },
    {
      "epoch": 6.076622009277344e-05,
      "step": 63718,
      "training_step_time": 7.631580114364624
    },
    {
      "epoch": 6.0767173767089845e-05,
      "model_forward_time": 1.8715333938598633,
      "step": 63719
    },
    {
      "epoch": 6.0767173767089845e-05,
      "step": 63719,
      "training_step_time": 7.381076812744141
    },
    {
      "epoch": 6.076812744140625e-05,
      "grad_norm": 0.07906951755285263,
      "learning_rate": 3.187236210416834e-06,
      "loss": 0.0573,
      "step": 63720
    },
    {
      "epoch": 6.076812744140625e-05,
      "model_forward_time": 1.7103855609893799,
      "step": 63720
    },
    {
      "dynamics_loss_avg": 0.058120181784033775,
      "epoch": 6.076812744140625e-05,
      "step": 63720
    },
    {
      "action_loss_avg": 0.0012985520850634202,
      "epoch": 6.076812744140625e-05,
      "step": 63720
    },
    {
      "epoch": 6.076812744140625e-05,
      "step": 63720,
      "training_step_time": 7.632288694381714
    },
    {
      "epoch": 6.076908111572265e-05,
      "model_forward_time": 1.7094390392303467,
      "step": 63721
    },
    {
      "epoch": 6.076908111572265e-05,
      "step": 63721,
      "training_step_time": 7.521095037460327
    },
    {
      "epoch": 6.0770034790039064e-05,
      "model_forward_time": 1.7252578735351562,
      "step": 63722
    },
    {
      "epoch": 6.0770034790039064e-05,
      "step": 63722,
      "training_step_time": 7.533482074737549
    },
    {
      "epoch": 6.077098846435547e-05,
      "model_forward_time": 1.7400076389312744,
      "step": 63723
    },
    {
      "epoch": 6.077098846435547e-05,
      "step": 63723,
      "training_step_time": 7.6415205001831055
    },
    {
      "epoch": 6.077194213867187e-05,
      "model_forward_time": 1.7102534770965576,
      "step": 63724
    },
    {
      "epoch": 6.077194213867187e-05,
      "step": 63724,
      "training_step_time": 7.798856496810913
    },
    {
      "epoch": 6.077289581298828e-05,
      "model_forward_time": 1.7080745697021484,
      "step": 63725
    },
    {
      "epoch": 6.077289581298828e-05,
      "step": 63725,
      "training_step_time": 9.344299793243408
    },
    {
      "epoch": 6.077384948730469e-05,
      "model_forward_time": 1.694169044494629,
      "step": 63726
    },
    {
      "epoch": 6.077384948730469e-05,
      "step": 63726,
      "training_step_time": 7.694258451461792
    },
    {
      "epoch": 6.077480316162109e-05,
      "model_forward_time": 1.6963684558868408,
      "step": 63727
    },
    {
      "epoch": 6.077480316162109e-05,
      "step": 63727,
      "training_step_time": 7.580028772354126
    },
    {
      "epoch": 6.07757568359375e-05,
      "model_forward_time": 1.6869196891784668,
      "step": 63728
    },
    {
      "epoch": 6.07757568359375e-05,
      "step": 63728,
      "training_step_time": 7.2929301261901855
    },
    {
      "epoch": 6.0776710510253905e-05,
      "model_forward_time": 2.2557387351989746,
      "step": 63729
    },
    {
      "epoch": 6.0776710510253905e-05,
      "step": 63729,
      "training_step_time": 7.561731338500977
    },
    {
      "epoch": 6.077766418457031e-05,
      "grad_norm": 0.07491879910230637,
      "learning_rate": 3.185695336461861e-06,
      "loss": 0.0581,
      "step": 63730
    },
    {
      "epoch": 6.077766418457031e-05,
      "model_forward_time": 1.6934733390808105,
      "step": 63730
    },
    {
      "dynamics_loss_avg": 0.06859538294374942,
      "epoch": 6.077766418457031e-05,
      "step": 63730
    },
    {
      "action_loss_avg": 0.0019230468315072358,
      "epoch": 6.077766418457031e-05,
      "step": 63730
    },
    {
      "epoch": 6.077766418457031e-05,
      "step": 63730,
      "training_step_time": 8.662605285644531
    },
    {
      "epoch": 6.077861785888672e-05,
      "model_forward_time": 1.696993350982666,
      "step": 63731
    },
    {
      "epoch": 6.077861785888672e-05,
      "step": 63731,
      "training_step_time": 9.778676748275757
    },
    {
      "epoch": 6.0779571533203124e-05,
      "model_forward_time": 1.7106356620788574,
      "step": 63732
    },
    {
      "epoch": 6.0779571533203124e-05,
      "step": 63732,
      "training_step_time": 6.996678590774536
    },
    {
      "epoch": 6.078052520751953e-05,
      "model_forward_time": 1.713061809539795,
      "step": 63733
    },
    {
      "epoch": 6.078052520751953e-05,
      "step": 63733,
      "training_step_time": 6.93898868560791
    },
    {
      "epoch": 6.078147888183594e-05,
      "model_forward_time": 1.7224092483520508,
      "step": 63734
    },
    {
      "epoch": 6.078147888183594e-05,
      "step": 63734,
      "training_step_time": 7.036403656005859
    },
    {
      "epoch": 6.078243255615234e-05,
      "model_forward_time": 1.7201294898986816,
      "step": 63735
    },
    {
      "epoch": 6.078243255615234e-05,
      "step": 63735,
      "training_step_time": 6.748345613479614
    },
    {
      "epoch": 6.078338623046875e-05,
      "model_forward_time": 1.7141163349151611,
      "step": 63736
    },
    {
      "epoch": 6.078338623046875e-05,
      "step": 63736,
      "training_step_time": 6.9902708530426025
    },
    {
      "epoch": 6.078433990478516e-05,
      "model_forward_time": 1.7147693634033203,
      "step": 63737
    },
    {
      "epoch": 6.078433990478516e-05,
      "step": 63737,
      "training_step_time": 6.953449726104736
    },
    {
      "epoch": 6.078529357910156e-05,
      "model_forward_time": 1.7169594764709473,
      "step": 63738
    },
    {
      "epoch": 6.078529357910156e-05,
      "step": 63738,
      "training_step_time": 6.938703536987305
    },
    {
      "epoch": 6.0786247253417966e-05,
      "model_forward_time": 1.68855619430542,
      "step": 63739
    },
    {
      "epoch": 6.0786247253417966e-05,
      "step": 63739,
      "training_step_time": 6.963691473007202
    },
    {
      "epoch": 6.078720092773438e-05,
      "grad_norm": 0.10563087463378906,
      "learning_rate": 3.1841546609165187e-06,
      "loss": 0.0595,
      "step": 63740
    },
    {
      "epoch": 6.078720092773438e-05,
      "model_forward_time": 2.262596845626831,
      "step": 63740
    },
    {
      "dynamics_loss_avg": 0.04623911529779434,
      "epoch": 6.078720092773438e-05,
      "step": 63740
    },
    {
      "action_loss_avg": 0.0013973142602480948,
      "epoch": 6.078720092773438e-05,
      "step": 63740
    },
    {
      "epoch": 6.078720092773438e-05,
      "step": 63740,
      "training_step_time": 7.406611204147339
    },
    {
      "epoch": 6.078815460205078e-05,
      "model_forward_time": 1.7021243572235107,
      "step": 63741
    },
    {
      "epoch": 6.078815460205078e-05,
      "step": 63741,
      "training_step_time": 7.9366843700408936
    },
    {
      "epoch": 6.0789108276367185e-05,
      "model_forward_time": 1.6954827308654785,
      "step": 63742
    },
    {
      "epoch": 6.0789108276367185e-05,
      "step": 63742,
      "training_step_time": 6.926208257675171
    },
    {
      "epoch": 6.0790061950683596e-05,
      "model_forward_time": 1.7051725387573242,
      "step": 63743
    },
    {
      "epoch": 6.0790061950683596e-05,
      "step": 63743,
      "training_step_time": 6.947681188583374
    },
    {
      "epoch": 6.0791015625e-05,
      "model_forward_time": 1.7061538696289062,
      "step": 63744
    },
    {
      "epoch": 6.0791015625e-05,
      "step": 63744,
      "training_step_time": 6.917484760284424
    },
    {
      "epoch": 6.0791969299316404e-05,
      "model_forward_time": 1.694939374923706,
      "step": 63745
    },
    {
      "epoch": 6.0791969299316404e-05,
      "step": 63745,
      "training_step_time": 6.95624041557312
    },
    {
      "epoch": 6.0792922973632815e-05,
      "model_forward_time": 1.6951792240142822,
      "step": 63746
    },
    {
      "epoch": 6.0792922973632815e-05,
      "step": 63746,
      "training_step_time": 6.960753440856934
    },
    {
      "epoch": 6.079387664794922e-05,
      "model_forward_time": 1.6929481029510498,
      "step": 63747
    },
    {
      "epoch": 6.079387664794922e-05,
      "step": 63747,
      "training_step_time": 6.923201560974121
    },
    {
      "epoch": 6.079483032226562e-05,
      "model_forward_time": 1.7007598876953125,
      "step": 63748
    },
    {
      "epoch": 6.079483032226562e-05,
      "step": 63748,
      "training_step_time": 6.968871593475342
    },
    {
      "epoch": 6.0795783996582033e-05,
      "model_forward_time": 2.257927179336548,
      "step": 63749
    },
    {
      "epoch": 6.0795783996582033e-05,
      "step": 63749,
      "training_step_time": 7.385993242263794
    },
    {
      "epoch": 6.079673767089844e-05,
      "grad_norm": 0.08740647882223129,
      "learning_rate": 3.1826141839492907e-06,
      "loss": 0.0576,
      "step": 63750
    },
    {
      "epoch": 6.079673767089844e-05,
      "model_forward_time": 1.6957762241363525,
      "step": 63750
    },
    {
      "dynamics_loss_avg": 0.048933788202703,
      "epoch": 6.079673767089844e-05,
      "step": 63750
    },
    {
      "action_loss_avg": 0.0010175383999012412,
      "epoch": 6.079673767089844e-05,
      "step": 63750
    },
    {
      "epoch": 6.079673767089844e-05,
      "step": 63750,
      "training_step_time": 7.850625276565552
    },
    {
      "epoch": 6.079769134521484e-05,
      "model_forward_time": 1.7033181190490723,
      "step": 63751
    },
    {
      "epoch": 6.079769134521484e-05,
      "step": 63751,
      "training_step_time": 6.982798337936401
    },
    {
      "epoch": 6.079864501953125e-05,
      "model_forward_time": 1.7072019577026367,
      "step": 63752
    },
    {
      "epoch": 6.079864501953125e-05,
      "step": 63752,
      "training_step_time": 7.017904996871948
    },
    {
      "epoch": 6.0799598693847656e-05,
      "model_forward_time": 1.7258684635162354,
      "step": 63753
    },
    {
      "epoch": 6.0799598693847656e-05,
      "step": 63753,
      "training_step_time": 7.032495021820068
    },
    {
      "epoch": 6.080055236816406e-05,
      "model_forward_time": 1.7153489589691162,
      "step": 63754
    },
    {
      "epoch": 6.080055236816406e-05,
      "step": 63754,
      "training_step_time": 7.0382726192474365
    },
    {
      "epoch": 6.080150604248047e-05,
      "model_forward_time": 1.710536241531372,
      "step": 63755
    },
    {
      "epoch": 6.080150604248047e-05,
      "step": 63755,
      "training_step_time": 7.031194686889648
    },
    {
      "epoch": 6.0802459716796875e-05,
      "model_forward_time": 1.713712215423584,
      "step": 63756
    },
    {
      "epoch": 6.0802459716796875e-05,
      "step": 63756,
      "training_step_time": 7.021195411682129
    },
    {
      "epoch": 6.080341339111328e-05,
      "model_forward_time": 1.7148816585540771,
      "step": 63757
    },
    {
      "epoch": 6.080341339111328e-05,
      "step": 63757,
      "training_step_time": 7.0471296310424805
    },
    {
      "epoch": 6.080436706542969e-05,
      "model_forward_time": 2.292829751968384,
      "step": 63758
    },
    {
      "epoch": 6.080436706542969e-05,
      "step": 63758,
      "training_step_time": 7.901588201522827
    },
    {
      "epoch": 6.0805320739746094e-05,
      "model_forward_time": 1.7119126319885254,
      "step": 63759
    },
    {
      "epoch": 6.0805320739746094e-05,
      "step": 63759,
      "training_step_time": 8.448129415512085
    },
    {
      "epoch": 6.08062744140625e-05,
      "grad_norm": 0.08592503517866135,
      "learning_rate": 3.1810739057286444e-06,
      "loss": 0.0583,
      "step": 63760
    },
    {
      "epoch": 6.08062744140625e-05,
      "model_forward_time": 1.7117435932159424,
      "step": 63760
    },
    {
      "dynamics_loss_avg": 0.0512420903891325,
      "epoch": 6.08062744140625e-05,
      "step": 63760
    },
    {
      "action_loss_avg": 0.0019230474368669092,
      "epoch": 6.08062744140625e-05,
      "step": 63760
    },
    {
      "epoch": 6.08062744140625e-05,
      "step": 63760,
      "training_step_time": 7.846457242965698
    },
    {
      "epoch": 6.080722808837891e-05,
      "model_forward_time": 1.716141939163208,
      "step": 63761
    },
    {
      "epoch": 6.080722808837891e-05,
      "step": 63761,
      "training_step_time": 7.049432992935181
    },
    {
      "epoch": 6.080818176269531e-05,
      "model_forward_time": 1.7061259746551514,
      "step": 63762
    },
    {
      "epoch": 6.080818176269531e-05,
      "step": 63762,
      "training_step_time": 6.958556413650513
    },
    {
      "epoch": 6.080913543701172e-05,
      "model_forward_time": 1.705439805984497,
      "step": 63763
    },
    {
      "epoch": 6.080913543701172e-05,
      "step": 63763,
      "training_step_time": 7.00001859664917
    },
    {
      "epoch": 6.081008911132813e-05,
      "model_forward_time": 1.3486499786376953,
      "step": 63764
    },
    {
      "epoch": 6.081008911132813e-05,
      "step": 63764,
      "training_step_time": 7.0204479694366455
    },
    {
      "epoch": 6.081104278564453e-05,
      "model_forward_time": 1.7111473083496094,
      "step": 63765
    },
    {
      "epoch": 6.081104278564453e-05,
      "step": 63765,
      "training_step_time": 6.96098518371582
    },
    {
      "epoch": 6.0811996459960936e-05,
      "model_forward_time": 1.7156877517700195,
      "step": 63766
    },
    {
      "epoch": 6.0811996459960936e-05,
      "step": 63766,
      "training_step_time": 6.9818644523620605
    },
    {
      "epoch": 6.081295013427735e-05,
      "model_forward_time": 2.300105333328247,
      "step": 63767
    },
    {
      "epoch": 6.081295013427735e-05,
      "step": 63767,
      "training_step_time": 7.833417177200317
    },
    {
      "epoch": 6.081390380859375e-05,
      "model_forward_time": 1.7189629077911377,
      "step": 63768
    },
    {
      "epoch": 6.081390380859375e-05,
      "step": 63768,
      "training_step_time": 10.507717847824097
    },
    {
      "epoch": 6.0814857482910155e-05,
      "model_forward_time": 1.710582971572876,
      "step": 63769
    },
    {
      "epoch": 6.0814857482910155e-05,
      "step": 63769,
      "training_step_time": 7.897026538848877
    },
    {
      "epoch": 6.0815811157226565e-05,
      "grad_norm": 0.09664043039083481,
      "learning_rate": 3.1795338264230217e-06,
      "loss": 0.0596,
      "step": 63770
    },
    {
      "epoch": 6.0815811157226565e-05,
      "model_forward_time": 1.726511001586914,
      "step": 63770
    },
    {
      "dynamics_loss_avg": 0.05154590886086226,
      "epoch": 6.0815811157226565e-05,
      "step": 63770
    },
    {
      "action_loss_avg": 0.0016214968287386,
      "epoch": 6.0815811157226565e-05,
      "step": 63770
    },
    {
      "epoch": 6.0815811157226565e-05,
      "step": 63770,
      "training_step_time": 7.203023672103882
    },
    {
      "epoch": 6.081676483154297e-05,
      "model_forward_time": 1.7269198894500732,
      "step": 63771
    },
    {
      "epoch": 6.081676483154297e-05,
      "step": 63771,
      "training_step_time": 6.992190599441528
    },
    {
      "epoch": 6.0817718505859374e-05,
      "model_forward_time": 1.7294862270355225,
      "step": 63772
    },
    {
      "epoch": 6.0817718505859374e-05,
      "step": 63772,
      "training_step_time": 7.031132221221924
    },
    {
      "epoch": 6.0818672180175784e-05,
      "model_forward_time": 1.7298710346221924,
      "step": 63773
    },
    {
      "epoch": 6.0818672180175784e-05,
      "step": 63773,
      "training_step_time": 6.981973171234131
    },
    {
      "epoch": 6.081962585449219e-05,
      "model_forward_time": 1.7381086349487305,
      "step": 63774
    },
    {
      "epoch": 6.081962585449219e-05,
      "step": 63774,
      "training_step_time": 6.976650714874268
    },
    {
      "epoch": 6.082057952880859e-05,
      "model_forward_time": 1.7294864654541016,
      "step": 63775
    },
    {
      "epoch": 6.082057952880859e-05,
      "step": 63775,
      "training_step_time": 6.944033622741699
    },
    {
      "epoch": 6.0821533203125e-05,
      "model_forward_time": 2.339198112487793,
      "step": 63776
    },
    {
      "epoch": 6.0821533203125e-05,
      "step": 63776,
      "training_step_time": 7.72478985786438
    },
    {
      "epoch": 6.082248687744141e-05,
      "model_forward_time": 1.727238655090332,
      "step": 63777
    },
    {
      "epoch": 6.082248687744141e-05,
      "step": 63777,
      "training_step_time": 11.038662433624268
    },
    {
      "epoch": 6.082344055175781e-05,
      "model_forward_time": 1.7252187728881836,
      "step": 63778
    },
    {
      "epoch": 6.082344055175781e-05,
      "step": 63778,
      "training_step_time": 8.495670080184937
    },
    {
      "epoch": 6.082439422607422e-05,
      "model_forward_time": 1.7378623485565186,
      "step": 63779
    },
    {
      "epoch": 6.082439422607422e-05,
      "step": 63779,
      "training_step_time": 7.191006898880005
    },
    {
      "epoch": 6.0825347900390626e-05,
      "grad_norm": 0.07687978446483612,
      "learning_rate": 3.177993946200843e-06,
      "loss": 0.0569,
      "step": 63780
    },
    {
      "epoch": 6.0825347900390626e-05,
      "model_forward_time": 1.7268457412719727,
      "step": 63780
    },
    {
      "dynamics_loss_avg": 0.07269912045449019,
      "epoch": 6.0825347900390626e-05,
      "step": 63780
    },
    {
      "action_loss_avg": 0.003127991978544742,
      "epoch": 6.0825347900390626e-05,
      "step": 63780
    },
    {
      "epoch": 6.0825347900390626e-05,
      "step": 63780,
      "training_step_time": 6.955923318862915
    },
    {
      "epoch": 6.082630157470703e-05,
      "model_forward_time": 1.728865146636963,
      "step": 63781
    },
    {
      "epoch": 6.082630157470703e-05,
      "step": 63781,
      "training_step_time": 6.946730375289917
    },
    {
      "epoch": 6.0827255249023434e-05,
      "model_forward_time": 1.7318971157073975,
      "step": 63782
    },
    {
      "epoch": 6.0827255249023434e-05,
      "step": 63782,
      "training_step_time": 7.044485569000244
    },
    {
      "epoch": 6.0828208923339845e-05,
      "model_forward_time": 1.7431397438049316,
      "step": 63783
    },
    {
      "epoch": 6.0828208923339845e-05,
      "step": 63783,
      "training_step_time": 6.998707294464111
    },
    {
      "epoch": 6.082916259765625e-05,
      "model_forward_time": 1.7283406257629395,
      "step": 63784
    },
    {
      "epoch": 6.082916259765625e-05,
      "step": 63784,
      "training_step_time": 7.019483327865601
    },
    {
      "epoch": 6.083011627197265e-05,
      "model_forward_time": 2.3397743701934814,
      "step": 63785
    },
    {
      "epoch": 6.083011627197265e-05,
      "step": 63785,
      "training_step_time": 7.8320746421813965
    },
    {
      "epoch": 6.0831069946289064e-05,
      "model_forward_time": 1.7279512882232666,
      "step": 63786
    },
    {
      "epoch": 6.0831069946289064e-05,
      "step": 63786,
      "training_step_time": 8.919084548950195
    },
    {
      "epoch": 6.083202362060547e-05,
      "model_forward_time": 1.7149555683135986,
      "step": 63787
    },
    {
      "epoch": 6.083202362060547e-05,
      "step": 63787,
      "training_step_time": 7.880953311920166
    },
    {
      "epoch": 6.083297729492187e-05,
      "model_forward_time": 1.7042479515075684,
      "step": 63788
    },
    {
      "epoch": 6.083297729492187e-05,
      "step": 63788,
      "training_step_time": 7.841517925262451
    },
    {
      "epoch": 6.083393096923828e-05,
      "model_forward_time": 1.714001178741455,
      "step": 63789
    },
    {
      "epoch": 6.083393096923828e-05,
      "step": 63789,
      "training_step_time": 6.905431747436523
    },
    {
      "epoch": 6.083488464355469e-05,
      "grad_norm": 0.22299785912036896,
      "learning_rate": 3.1764542652305065e-06,
      "loss": 0.0589,
      "step": 63790
    },
    {
      "epoch": 6.083488464355469e-05,
      "model_forward_time": 1.7332448959350586,
      "step": 63790
    },
    {
      "dynamics_loss_avg": 0.061086995992809534,
      "epoch": 6.083488464355469e-05,
      "step": 63790
    },
    {
      "action_loss_avg": 0.0020874378707958384,
      "epoch": 6.083488464355469e-05,
      "step": 63790
    },
    {
      "epoch": 6.083488464355469e-05,
      "step": 63790,
      "training_step_time": 6.914303541183472
    },
    {
      "epoch": 6.083583831787109e-05,
      "model_forward_time": 1.7058613300323486,
      "step": 63791
    },
    {
      "epoch": 6.083583831787109e-05,
      "step": 63791,
      "training_step_time": 6.9295737743377686
    },
    {
      "epoch": 6.08367919921875e-05,
      "model_forward_time": 1.7123234272003174,
      "step": 63792
    },
    {
      "epoch": 6.08367919921875e-05,
      "step": 63792,
      "training_step_time": 6.92930793762207
    },
    {
      "epoch": 6.0837745666503906e-05,
      "model_forward_time": 1.7147305011749268,
      "step": 63793
    },
    {
      "epoch": 6.0837745666503906e-05,
      "step": 63793,
      "training_step_time": 6.916747331619263
    },
    {
      "epoch": 6.083869934082031e-05,
      "model_forward_time": 2.2976133823394775,
      "step": 63794
    },
    {
      "epoch": 6.083869934082031e-05,
      "step": 63794,
      "training_step_time": 7.69898247718811
    },
    {
      "epoch": 6.083965301513672e-05,
      "model_forward_time": 1.7513518333435059,
      "step": 63795
    },
    {
      "epoch": 6.083965301513672e-05,
      "step": 63795,
      "training_step_time": 7.26818060874939
    },
    {
      "epoch": 6.0840606689453124e-05,
      "model_forward_time": 1.7099215984344482,
      "step": 63796
    },
    {
      "epoch": 6.0840606689453124e-05,
      "step": 63796,
      "training_step_time": 9.520750045776367
    },
    {
      "epoch": 6.084156036376953e-05,
      "model_forward_time": 1.710479736328125,
      "step": 63797
    },
    {
      "epoch": 6.084156036376953e-05,
      "step": 63797,
      "training_step_time": 7.89379096031189
    },
    {
      "epoch": 6.084251403808594e-05,
      "model_forward_time": 1.7052948474884033,
      "step": 63798
    },
    {
      "epoch": 6.084251403808594e-05,
      "step": 63798,
      "training_step_time": 7.133734941482544
    },
    {
      "epoch": 6.084346771240234e-05,
      "model_forward_time": 1.7135899066925049,
      "step": 63799
    },
    {
      "epoch": 6.084346771240234e-05,
      "step": 63799,
      "training_step_time": 6.8943963050842285
    },
    {
      "epoch": 6.084442138671875e-05,
      "grad_norm": 0.12022173404693604,
      "learning_rate": 3.17491478368039e-06,
      "loss": 0.0612,
      "step": 63800
    },
    {
      "epoch": 6.084442138671875e-05,
      "model_forward_time": 1.7065761089324951,
      "step": 63800
    },
    {
      "dynamics_loss_avg": 0.0743807964026928,
      "epoch": 6.084442138671875e-05,
      "step": 63800
    },
    {
      "action_loss_avg": 0.0022985321353189646,
      "epoch": 6.084442138671875e-05,
      "step": 63800
    },
    {
      "epoch": 6.084442138671875e-05,
      "step": 63800,
      "training_step_time": 6.933416366577148
    },
    {
      "epoch": 6.084537506103516e-05,
      "model_forward_time": 1.7243871688842773,
      "step": 63801
    },
    {
      "epoch": 6.084537506103516e-05,
      "step": 63801,
      "training_step_time": 6.8967249393463135
    },
    {
      "epoch": 6.084632873535156e-05,
      "model_forward_time": 1.7085151672363281,
      "step": 63802
    },
    {
      "epoch": 6.084632873535156e-05,
      "step": 63802,
      "training_step_time": 6.868593454360962
    },
    {
      "epoch": 6.0847282409667966e-05,
      "model_forward_time": 2.2534472942352295,
      "step": 63803
    },
    {
      "epoch": 6.0847282409667966e-05,
      "step": 63803,
      "training_step_time": 7.717499256134033
    },
    {
      "epoch": 6.084823608398438e-05,
      "model_forward_time": 1.7683231830596924,
      "step": 63804
    },
    {
      "epoch": 6.084823608398438e-05,
      "step": 63804,
      "training_step_time": 7.351537704467773
    },
    {
      "epoch": 6.084918975830078e-05,
      "model_forward_time": 1.7172620296478271,
      "step": 63805
    },
    {
      "epoch": 6.084918975830078e-05,
      "step": 63805,
      "training_step_time": 10.054356098175049
    },
    {
      "epoch": 6.0850143432617185e-05,
      "model_forward_time": 1.7129228115081787,
      "step": 63806
    },
    {
      "epoch": 6.0850143432617185e-05,
      "step": 63806,
      "training_step_time": 7.866095542907715
    },
    {
      "epoch": 6.0851097106933596e-05,
      "model_forward_time": 1.711876630783081,
      "step": 63807
    },
    {
      "epoch": 6.0851097106933596e-05,
      "step": 63807,
      "training_step_time": 7.69857382774353
    },
    {
      "epoch": 6.085205078125e-05,
      "model_forward_time": 1.7036640644073486,
      "step": 63808
    },
    {
      "epoch": 6.085205078125e-05,
      "step": 63808,
      "training_step_time": 6.906242847442627
    },
    {
      "epoch": 6.0853004455566404e-05,
      "model_forward_time": 1.7073912620544434,
      "step": 63809
    },
    {
      "epoch": 6.0853004455566404e-05,
      "step": 63809,
      "training_step_time": 6.894508123397827
    },
    {
      "epoch": 6.0853958129882815e-05,
      "grad_norm": 0.13018758594989777,
      "learning_rate": 3.173375501718847e-06,
      "loss": 0.0577,
      "step": 63810
    },
    {
      "epoch": 6.0853958129882815e-05,
      "model_forward_time": 1.7287352085113525,
      "step": 63810
    },
    {
      "dynamics_loss_avg": 0.04811999602243304,
      "epoch": 6.0853958129882815e-05,
      "step": 63810
    },
    {
      "action_loss_avg": 0.001615336473332718,
      "epoch": 6.0853958129882815e-05,
      "step": 63810
    },
    {
      "epoch": 6.0853958129882815e-05,
      "step": 63810,
      "training_step_time": 6.950426816940308
    },
    {
      "epoch": 6.085491180419922e-05,
      "model_forward_time": 1.7146618366241455,
      "step": 63811
    },
    {
      "epoch": 6.085491180419922e-05,
      "step": 63811,
      "training_step_time": 6.873070955276489
    },
    {
      "epoch": 6.085586547851562e-05,
      "model_forward_time": 1.7073144912719727,
      "step": 63812
    },
    {
      "epoch": 6.085586547851562e-05,
      "step": 63812,
      "training_step_time": 7.684830904006958
    },
    {
      "epoch": 6.0856819152832034e-05,
      "model_forward_time": 2.298377275466919,
      "step": 63813
    },
    {
      "epoch": 6.0856819152832034e-05,
      "step": 63813,
      "training_step_time": 7.343975067138672
    },
    {
      "epoch": 6.085777282714844e-05,
      "model_forward_time": 1.702850580215454,
      "step": 63814
    },
    {
      "epoch": 6.085777282714844e-05,
      "step": 63814,
      "training_step_time": 11.728092670440674
    },
    {
      "epoch": 6.085872650146484e-05,
      "model_forward_time": 1.7112858295440674,
      "step": 63815
    },
    {
      "epoch": 6.085872650146484e-05,
      "step": 63815,
      "training_step_time": 7.8106889724731445
    },
    {
      "epoch": 6.085968017578125e-05,
      "model_forward_time": 1.7051281929016113,
      "step": 63816
    },
    {
      "epoch": 6.085968017578125e-05,
      "step": 63816,
      "training_step_time": 7.733130931854248
    },
    {
      "epoch": 6.0860633850097656e-05,
      "model_forward_time": 1.7136452198028564,
      "step": 63817
    },
    {
      "epoch": 6.0860633850097656e-05,
      "step": 63817,
      "training_step_time": 6.896539926528931
    },
    {
      "epoch": 6.086158752441406e-05,
      "model_forward_time": 1.7089478969573975,
      "step": 63818
    },
    {
      "epoch": 6.086158752441406e-05,
      "step": 63818,
      "training_step_time": 6.9519617557525635
    },
    {
      "epoch": 6.086254119873047e-05,
      "model_forward_time": 1.716468095779419,
      "step": 63819
    },
    {
      "epoch": 6.086254119873047e-05,
      "step": 63819,
      "training_step_time": 6.964954376220703
    },
    {
      "epoch": 6.0863494873046875e-05,
      "grad_norm": 0.09527389705181122,
      "learning_rate": 3.171836419514215e-06,
      "loss": 0.0602,
      "step": 63820
    },
    {
      "epoch": 6.0863494873046875e-05,
      "model_forward_time": 1.7101316452026367,
      "step": 63820
    },
    {
      "dynamics_loss_avg": 0.058848537504673004,
      "epoch": 6.0863494873046875e-05,
      "step": 63820
    },
    {
      "action_loss_avg": 0.002389918017433956,
      "epoch": 6.0863494873046875e-05,
      "step": 63820
    },
    {
      "epoch": 6.0863494873046875e-05,
      "step": 63820,
      "training_step_time": 6.8195884227752686
    },
    {
      "epoch": 6.086444854736328e-05,
      "model_forward_time": 1.7089664936065674,
      "step": 63821
    },
    {
      "epoch": 6.086444854736328e-05,
      "step": 63821,
      "training_step_time": 6.877975225448608
    },
    {
      "epoch": 6.086540222167969e-05,
      "model_forward_time": 2.296644926071167,
      "step": 63822
    },
    {
      "epoch": 6.086540222167969e-05,
      "step": 63822,
      "training_step_time": 7.666594505310059
    },
    {
      "epoch": 6.0866355895996094e-05,
      "model_forward_time": 1.7103345394134521,
      "step": 63823
    },
    {
      "epoch": 6.0866355895996094e-05,
      "step": 63823,
      "training_step_time": 8.691128015518188
    },
    {
      "epoch": 6.08673095703125e-05,
      "model_forward_time": 1.7092547416687012,
      "step": 63824
    },
    {
      "epoch": 6.08673095703125e-05,
      "step": 63824,
      "training_step_time": 9.363848686218262
    },
    {
      "epoch": 6.086826324462891e-05,
      "model_forward_time": 1.7180547714233398,
      "step": 63825
    },
    {
      "epoch": 6.086826324462891e-05,
      "step": 63825,
      "training_step_time": 7.702586889266968
    },
    {
      "epoch": 6.086921691894531e-05,
      "model_forward_time": 1.7089660167694092,
      "step": 63826
    },
    {
      "epoch": 6.086921691894531e-05,
      "step": 63826,
      "training_step_time": 6.887966871261597
    },
    {
      "epoch": 6.087017059326172e-05,
      "model_forward_time": 1.7167418003082275,
      "step": 63827
    },
    {
      "epoch": 6.087017059326172e-05,
      "step": 63827,
      "training_step_time": 6.892003774642944
    },
    {
      "epoch": 6.087112426757813e-05,
      "model_forward_time": 1.7122938632965088,
      "step": 63828
    },
    {
      "epoch": 6.087112426757813e-05,
      "step": 63828,
      "training_step_time": 6.873612642288208
    },
    {
      "epoch": 6.087207794189453e-05,
      "model_forward_time": 1.705054521560669,
      "step": 63829
    },
    {
      "epoch": 6.087207794189453e-05,
      "step": 63829,
      "training_step_time": 6.918551445007324
    },
    {
      "epoch": 6.0873031616210936e-05,
      "grad_norm": 0.09097979217767715,
      "learning_rate": 3.170297537234803e-06,
      "loss": 0.0566,
      "step": 63830
    },
    {
      "epoch": 6.0873031616210936e-05,
      "model_forward_time": 1.7124507427215576,
      "step": 63830
    },
    {
      "dynamics_loss_avg": 0.07162588592618704,
      "epoch": 6.0873031616210936e-05,
      "step": 63830
    },
    {
      "action_loss_avg": 0.0020208231318974866,
      "epoch": 6.0873031616210936e-05,
      "step": 63830
    },
    {
      "epoch": 6.0873031616210936e-05,
      "step": 63830,
      "training_step_time": 6.889755010604858
    },
    {
      "epoch": 6.087398529052735e-05,
      "model_forward_time": 2.2601826190948486,
      "step": 63831
    },
    {
      "epoch": 6.087398529052735e-05,
      "step": 63831,
      "training_step_time": 7.64240574836731
    },
    {
      "epoch": 6.087493896484375e-05,
      "model_forward_time": 1.7650260925292969,
      "step": 63832
    },
    {
      "epoch": 6.087493896484375e-05,
      "step": 63832,
      "training_step_time": 7.72955584526062
    },
    {
      "epoch": 6.0875892639160155e-05,
      "model_forward_time": 1.7105882167816162,
      "step": 63833
    },
    {
      "epoch": 6.0875892639160155e-05,
      "step": 63833,
      "training_step_time": 8.742411851882935
    },
    {
      "epoch": 6.0876846313476566e-05,
      "model_forward_time": 1.7110862731933594,
      "step": 63834
    },
    {
      "epoch": 6.0876846313476566e-05,
      "step": 63834,
      "training_step_time": 7.863880157470703
    },
    {
      "epoch": 6.087779998779297e-05,
      "model_forward_time": 1.7237248420715332,
      "step": 63835
    },
    {
      "epoch": 6.087779998779297e-05,
      "step": 63835,
      "training_step_time": 6.894075870513916
    },
    {
      "epoch": 6.0878753662109374e-05,
      "model_forward_time": 1.7373201847076416,
      "step": 63836
    },
    {
      "epoch": 6.0878753662109374e-05,
      "step": 63836,
      "training_step_time": 6.838732957839966
    },
    {
      "epoch": 6.0879707336425784e-05,
      "model_forward_time": 1.7376205921173096,
      "step": 63837
    },
    {
      "epoch": 6.0879707336425784e-05,
      "step": 63837,
      "training_step_time": 6.861908435821533
    },
    {
      "epoch": 6.088066101074219e-05,
      "model_forward_time": 1.7315213680267334,
      "step": 63838
    },
    {
      "epoch": 6.088066101074219e-05,
      "step": 63838,
      "training_step_time": 6.882140874862671
    },
    {
      "epoch": 6.088161468505859e-05,
      "model_forward_time": 1.7336180210113525,
      "step": 63839
    },
    {
      "epoch": 6.088161468505859e-05,
      "step": 63839,
      "training_step_time": 6.871975421905518
    },
    {
      "epoch": 6.0882568359375e-05,
      "grad_norm": 0.10322652757167816,
      "learning_rate": 3.1687588550489013e-06,
      "loss": 0.0564,
      "step": 63840
    },
    {
      "epoch": 6.0882568359375e-05,
      "model_forward_time": 2.2728357315063477,
      "step": 63840
    },
    {
      "dynamics_loss_avg": 0.054990568570792674,
      "epoch": 6.0882568359375e-05,
      "step": 63840
    },
    {
      "action_loss_avg": 0.0015993192791938782,
      "epoch": 6.0882568359375e-05,
      "step": 63840
    },
    {
      "epoch": 6.0882568359375e-05,
      "step": 63840,
      "training_step_time": 7.659557580947876
    },
    {
      "epoch": 6.088352203369141e-05,
      "model_forward_time": 1.7720532417297363,
      "step": 63841
    },
    {
      "epoch": 6.088352203369141e-05,
      "step": 63841,
      "training_step_time": 9.95547103881836
    },
    {
      "epoch": 6.088447570800781e-05,
      "model_forward_time": 1.7080423831939697,
      "step": 63842
    },
    {
      "epoch": 6.088447570800781e-05,
      "step": 63842,
      "training_step_time": 8.876288652420044
    },
    {
      "epoch": 6.088542938232422e-05,
      "model_forward_time": 1.711031436920166,
      "step": 63843
    },
    {
      "epoch": 6.088542938232422e-05,
      "step": 63843,
      "training_step_time": 8.764424562454224
    },
    {
      "epoch": 6.0886383056640626e-05,
      "model_forward_time": 1.7137725353240967,
      "step": 63844
    },
    {
      "epoch": 6.0886383056640626e-05,
      "step": 63844,
      "training_step_time": 6.829029560089111
    },
    {
      "epoch": 6.088733673095703e-05,
      "model_forward_time": 1.7281599044799805,
      "step": 63845
    },
    {
      "epoch": 6.088733673095703e-05,
      "step": 63845,
      "training_step_time": 6.845565319061279
    },
    {
      "epoch": 6.0888290405273434e-05,
      "model_forward_time": 1.7362980842590332,
      "step": 63846
    },
    {
      "epoch": 6.0888290405273434e-05,
      "step": 63846,
      "training_step_time": 6.844817161560059
    },
    {
      "epoch": 6.0889244079589845e-05,
      "model_forward_time": 1.730787754058838,
      "step": 63847
    },
    {
      "epoch": 6.0889244079589845e-05,
      "step": 63847,
      "training_step_time": 6.87749981880188
    },
    {
      "epoch": 6.089019775390625e-05,
      "model_forward_time": 1.7327098846435547,
      "step": 63848
    },
    {
      "epoch": 6.089019775390625e-05,
      "step": 63848,
      "training_step_time": 6.910548686981201
    },
    {
      "epoch": 6.089115142822265e-05,
      "model_forward_time": 2.2895634174346924,
      "step": 63849
    },
    {
      "epoch": 6.089115142822265e-05,
      "step": 63849,
      "training_step_time": 7.644992828369141
    },
    {
      "epoch": 6.0892105102539064e-05,
      "grad_norm": 0.0888175442814827,
      "learning_rate": 3.1672203731247786e-06,
      "loss": 0.0559,
      "step": 63850
    },
    {
      "epoch": 6.0892105102539064e-05,
      "model_forward_time": 1.776505470275879,
      "step": 63850
    },
    {
      "dynamics_loss_avg": 0.05469165490940213,
      "epoch": 6.0892105102539064e-05,
      "step": 63850
    },
    {
      "action_loss_avg": 0.0013237820705398918,
      "epoch": 6.0892105102539064e-05,
      "step": 63850
    },
    {
      "epoch": 6.0892105102539064e-05,
      "step": 63850,
      "training_step_time": 8.6598961353302
    },
    {
      "epoch": 6.089305877685547e-05,
      "model_forward_time": 1.7271311283111572,
      "step": 63851
    },
    {
      "epoch": 6.089305877685547e-05,
      "step": 63851,
      "training_step_time": 9.597958326339722
    },
    {
      "epoch": 6.089401245117187e-05,
      "model_forward_time": 1.7304046154022217,
      "step": 63852
    },
    {
      "epoch": 6.089401245117187e-05,
      "step": 63852,
      "training_step_time": 7.766427755355835
    },
    {
      "epoch": 6.089496612548828e-05,
      "model_forward_time": 1.7371244430541992,
      "step": 63853
    },
    {
      "epoch": 6.089496612548828e-05,
      "step": 63853,
      "training_step_time": 7.782414197921753
    },
    {
      "epoch": 6.089591979980469e-05,
      "model_forward_time": 1.354895830154419,
      "step": 63854
    },
    {
      "epoch": 6.089591979980469e-05,
      "step": 63854,
      "training_step_time": 6.87700343132019
    },
    {
      "epoch": 6.089687347412109e-05,
      "model_forward_time": 1.7401750087738037,
      "step": 63855
    },
    {
      "epoch": 6.089687347412109e-05,
      "step": 63855,
      "training_step_time": 6.904827833175659
    },
    {
      "epoch": 6.08978271484375e-05,
      "model_forward_time": 1.7394187450408936,
      "step": 63856
    },
    {
      "epoch": 6.08978271484375e-05,
      "step": 63856,
      "training_step_time": 6.855436086654663
    },
    {
      "epoch": 6.0898780822753906e-05,
      "model_forward_time": 1.7296643257141113,
      "step": 63857
    },
    {
      "epoch": 6.0898780822753906e-05,
      "step": 63857,
      "training_step_time": 6.929482936859131
    },
    {
      "epoch": 6.089973449707031e-05,
      "model_forward_time": 1.7302136421203613,
      "step": 63858
    },
    {
      "epoch": 6.089973449707031e-05,
      "step": 63858,
      "training_step_time": 7.636772394180298
    },
    {
      "epoch": 6.090068817138672e-05,
      "model_forward_time": 2.3513758182525635,
      "step": 63859
    },
    {
      "epoch": 6.090068817138672e-05,
      "step": 63859,
      "training_step_time": 8.708025217056274
    },
    {
      "epoch": 6.0901641845703124e-05,
      "grad_norm": 0.10643534362316132,
      "learning_rate": 3.1656820916306795e-06,
      "loss": 0.0566,
      "step": 63860
    },
    {
      "epoch": 6.0901641845703124e-05,
      "model_forward_time": 1.7298054695129395,
      "step": 63860
    },
    {
      "dynamics_loss_avg": 0.051560019887983796,
      "epoch": 6.0901641845703124e-05,
      "step": 63860
    },
    {
      "action_loss_avg": 0.002143699859152548,
      "epoch": 6.0901641845703124e-05,
      "step": 63860
    },
    {
      "epoch": 6.0901641845703124e-05,
      "step": 63860,
      "training_step_time": 9.249143838882446
    },
    {
      "epoch": 6.090259552001953e-05,
      "model_forward_time": 1.7414543628692627,
      "step": 63861
    },
    {
      "epoch": 6.090259552001953e-05,
      "step": 63861,
      "training_step_time": 7.8857035636901855
    },
    {
      "epoch": 6.090354919433594e-05,
      "model_forward_time": 1.725057601928711,
      "step": 63862
    },
    {
      "epoch": 6.090354919433594e-05,
      "step": 63862,
      "training_step_time": 7.826769113540649
    },
    {
      "epoch": 6.090450286865234e-05,
      "model_forward_time": 1.6938543319702148,
      "step": 63863
    },
    {
      "epoch": 6.090450286865234e-05,
      "step": 63863,
      "training_step_time": 6.90682053565979
    },
    {
      "epoch": 6.090545654296875e-05,
      "model_forward_time": 1.706099271774292,
      "step": 63864
    },
    {
      "epoch": 6.090545654296875e-05,
      "step": 63864,
      "training_step_time": 6.954277515411377
    },
    {
      "epoch": 6.090641021728516e-05,
      "model_forward_time": 1.7118871212005615,
      "step": 63865
    },
    {
      "epoch": 6.090641021728516e-05,
      "step": 63865,
      "training_step_time": 6.901744604110718
    },
    {
      "epoch": 6.090736389160156e-05,
      "model_forward_time": 1.706122875213623,
      "step": 63866
    },
    {
      "epoch": 6.090736389160156e-05,
      "step": 63866,
      "training_step_time": 6.917351722717285
    },
    {
      "epoch": 6.0908317565917966e-05,
      "model_forward_time": 1.703484296798706,
      "step": 63867
    },
    {
      "epoch": 6.0908317565917966e-05,
      "step": 63867,
      "training_step_time": 6.927594423294067
    },
    {
      "epoch": 6.090927124023438e-05,
      "model_forward_time": 2.301870584487915,
      "step": 63868
    },
    {
      "epoch": 6.090927124023438e-05,
      "step": 63868,
      "training_step_time": 7.725682497024536
    },
    {
      "epoch": 6.091022491455078e-05,
      "model_forward_time": 1.7128159999847412,
      "step": 63869
    },
    {
      "epoch": 6.091022491455078e-05,
      "step": 63869,
      "training_step_time": 10.080575704574585
    },
    {
      "epoch": 6.0911178588867185e-05,
      "grad_norm": 0.144328773021698,
      "learning_rate": 3.16414401073483e-06,
      "loss": 0.0528,
      "step": 63870
    },
    {
      "epoch": 6.0911178588867185e-05,
      "model_forward_time": 1.6987383365631104,
      "step": 63870
    },
    {
      "dynamics_loss_avg": 0.034371384792029856,
      "epoch": 6.0911178588867185e-05,
      "step": 63870
    },
    {
      "action_loss_avg": 0.0012240212818142027,
      "epoch": 6.0911178588867185e-05,
      "step": 63870
    },
    {
      "epoch": 6.0911178588867185e-05,
      "step": 63870,
      "training_step_time": 9.115816354751587
    },
    {
      "epoch": 6.0912132263183596e-05,
      "model_forward_time": 1.7029576301574707,
      "step": 63871
    },
    {
      "epoch": 6.0912132263183596e-05,
      "step": 63871,
      "training_step_time": 8.043910264968872
    },
    {
      "epoch": 6.09130859375e-05,
      "model_forward_time": 1.7083609104156494,
      "step": 63872
    },
    {
      "epoch": 6.09130859375e-05,
      "step": 63872,
      "training_step_time": 7.822808504104614
    },
    {
      "epoch": 6.0914039611816404e-05,
      "model_forward_time": 1.7151904106140137,
      "step": 63873
    },
    {
      "epoch": 6.0914039611816404e-05,
      "step": 63873,
      "training_step_time": 7.062793254852295
    },
    {
      "epoch": 6.0914993286132815e-05,
      "model_forward_time": 1.6960346698760986,
      "step": 63874
    },
    {
      "epoch": 6.0914993286132815e-05,
      "step": 63874,
      "training_step_time": 6.989086627960205
    },
    {
      "epoch": 6.091594696044922e-05,
      "model_forward_time": 1.716623067855835,
      "step": 63875
    },
    {
      "epoch": 6.091594696044922e-05,
      "step": 63875,
      "training_step_time": 7.106456518173218
    },
    {
      "epoch": 6.091690063476562e-05,
      "model_forward_time": 1.7218055725097656,
      "step": 63876
    },
    {
      "epoch": 6.091690063476562e-05,
      "step": 63876,
      "training_step_time": 7.144060134887695
    },
    {
      "epoch": 6.0917854309082034e-05,
      "model_forward_time": 2.331037759780884,
      "step": 63877
    },
    {
      "epoch": 6.0917854309082034e-05,
      "step": 63877,
      "training_step_time": 8.837585926055908
    },
    {
      "epoch": 6.091880798339844e-05,
      "model_forward_time": 1.7220685482025146,
      "step": 63878
    },
    {
      "epoch": 6.091880798339844e-05,
      "step": 63878,
      "training_step_time": 11.152692556381226
    },
    {
      "epoch": 6.091976165771484e-05,
      "model_forward_time": 1.7133162021636963,
      "step": 63879
    },
    {
      "epoch": 6.091976165771484e-05,
      "step": 63879,
      "training_step_time": 9.73842978477478
    },
    {
      "epoch": 6.092071533203125e-05,
      "grad_norm": 0.08813251554965973,
      "learning_rate": 3.162606130605429e-06,
      "loss": 0.0565,
      "step": 63880
    },
    {
      "epoch": 6.092071533203125e-05,
      "model_forward_time": 1.7231507301330566,
      "step": 63880
    },
    {
      "dynamics_loss_avg": 0.07073151916265488,
      "epoch": 6.092071533203125e-05,
      "step": 63880
    },
    {
      "action_loss_avg": 0.0021734982263296844,
      "epoch": 6.092071533203125e-05,
      "step": 63880
    },
    {
      "epoch": 6.092071533203125e-05,
      "step": 63880,
      "training_step_time": 9.106082201004028
    },
    {
      "epoch": 6.0921669006347656e-05,
      "model_forward_time": 1.729947566986084,
      "step": 63881
    },
    {
      "epoch": 6.0921669006347656e-05,
      "step": 63881,
      "training_step_time": 7.831238746643066
    },
    {
      "epoch": 6.092262268066406e-05,
      "model_forward_time": 1.7335689067840576,
      "step": 63882
    },
    {
      "epoch": 6.092262268066406e-05,
      "step": 63882,
      "training_step_time": 7.130437135696411
    },
    {
      "epoch": 6.092357635498047e-05,
      "model_forward_time": 1.7286832332611084,
      "step": 63883
    },
    {
      "epoch": 6.092357635498047e-05,
      "step": 63883,
      "training_step_time": 7.044228553771973
    },
    {
      "epoch": 6.0924530029296875e-05,
      "model_forward_time": 1.7285792827606201,
      "step": 63884
    },
    {
      "epoch": 6.0924530029296875e-05,
      "step": 63884,
      "training_step_time": 7.035569667816162
    },
    {
      "epoch": 6.092548370361328e-05,
      "model_forward_time": 1.7292826175689697,
      "step": 63885
    },
    {
      "epoch": 6.092548370361328e-05,
      "step": 63885,
      "training_step_time": 6.97977352142334
    },
    {
      "epoch": 6.092643737792969e-05,
      "model_forward_time": 2.3190627098083496,
      "step": 63886
    },
    {
      "epoch": 6.092643737792969e-05,
      "step": 63886,
      "training_step_time": 8.930082321166992
    },
    {
      "epoch": 6.0927391052246094e-05,
      "model_forward_time": 1.7807135581970215,
      "step": 63887
    },
    {
      "epoch": 6.0927391052246094e-05,
      "step": 63887,
      "training_step_time": 8.783909320831299
    },
    {
      "epoch": 6.09283447265625e-05,
      "model_forward_time": 1.7234060764312744,
      "step": 63888
    },
    {
      "epoch": 6.09283447265625e-05,
      "step": 63888,
      "training_step_time": 8.77865481376648
    },
    {
      "epoch": 6.092929840087891e-05,
      "model_forward_time": 1.7235066890716553,
      "step": 63889
    },
    {
      "epoch": 6.092929840087891e-05,
      "step": 63889,
      "training_step_time": 8.488119840621948
    },
    {
      "epoch": 6.093025207519531e-05,
      "grad_norm": 0.08938837051391602,
      "learning_rate": 3.1610684514106603e-06,
      "loss": 0.0566,
      "step": 63890
    },
    {
      "epoch": 6.093025207519531e-05,
      "model_forward_time": 1.717909812927246,
      "step": 63890
    },
    {
      "dynamics_loss_avg": 0.061342825181782246,
      "epoch": 6.093025207519531e-05,
      "step": 63890
    },
    {
      "action_loss_avg": 0.0015780601184815168,
      "epoch": 6.093025207519531e-05,
      "step": 63890
    },
    {
      "epoch": 6.093025207519531e-05,
      "step": 63890,
      "training_step_time": 7.922718048095703
    },
    {
      "epoch": 6.093120574951172e-05,
      "model_forward_time": 1.7273190021514893,
      "step": 63891
    },
    {
      "epoch": 6.093120574951172e-05,
      "step": 63891,
      "training_step_time": 7.7750701904296875
    },
    {
      "epoch": 6.093215942382813e-05,
      "model_forward_time": 1.7515788078308105,
      "step": 63892
    },
    {
      "epoch": 6.093215942382813e-05,
      "step": 63892,
      "training_step_time": 7.040197849273682
    },
    {
      "epoch": 6.093311309814453e-05,
      "model_forward_time": 1.7190310955047607,
      "step": 63893
    },
    {
      "epoch": 6.093311309814453e-05,
      "step": 63893,
      "training_step_time": 6.845509767532349
    },
    {
      "epoch": 6.0934066772460936e-05,
      "model_forward_time": 1.7257368564605713,
      "step": 63894
    },
    {
      "epoch": 6.0934066772460936e-05,
      "step": 63894,
      "training_step_time": 6.996256589889526
    },
    {
      "epoch": 6.093502044677735e-05,
      "model_forward_time": 2.27470064163208,
      "step": 63895
    },
    {
      "epoch": 6.093502044677735e-05,
      "step": 63895,
      "training_step_time": 7.848829984664917
    },
    {
      "epoch": 6.093597412109375e-05,
      "model_forward_time": 1.409358263015747,
      "step": 63896
    },
    {
      "epoch": 6.093597412109375e-05,
      "step": 63896,
      "training_step_time": 7.834815502166748
    },
    {
      "epoch": 6.0936927795410155e-05,
      "model_forward_time": 1.7258636951446533,
      "step": 63897
    },
    {
      "epoch": 6.0936927795410155e-05,
      "step": 63897,
      "training_step_time": 11.260005950927734
    },
    {
      "epoch": 6.0937881469726566e-05,
      "model_forward_time": 1.722139596939087,
      "step": 63898
    },
    {
      "epoch": 6.0937881469726566e-05,
      "step": 63898,
      "training_step_time": 8.053467273712158
    },
    {
      "epoch": 6.093883514404297e-05,
      "model_forward_time": 1.724297285079956,
      "step": 63899
    },
    {
      "epoch": 6.093883514404297e-05,
      "step": 63899,
      "training_step_time": 7.980898380279541
    },
    {
      "epoch": 6.0939788818359374e-05,
      "grad_norm": 0.1114044040441513,
      "learning_rate": 3.1595309733186818e-06,
      "loss": 0.057,
      "step": 63900
    },
    {
      "epoch": 6.0939788818359374e-05,
      "model_forward_time": 1.7220630645751953,
      "step": 63900
    },
    {
      "dynamics_loss_avg": 0.08292827680706978,
      "epoch": 6.0939788818359374e-05,
      "step": 63900
    },
    {
      "action_loss_avg": 0.0023285143717657773,
      "epoch": 6.0939788818359374e-05,
      "step": 63900
    },
    {
      "epoch": 6.0939788818359374e-05,
      "step": 63900,
      "training_step_time": 7.789804697036743
    },
    {
      "epoch": 6.0940742492675784e-05,
      "model_forward_time": 1.7277538776397705,
      "step": 63901
    },
    {
      "epoch": 6.0940742492675784e-05,
      "step": 63901,
      "training_step_time": 6.979161500930786
    },
    {
      "epoch": 6.094169616699219e-05,
      "model_forward_time": 1.7296814918518066,
      "step": 63902
    },
    {
      "epoch": 6.094169616699219e-05,
      "step": 63902,
      "training_step_time": 7.045799970626831
    },
    {
      "epoch": 6.094264984130859e-05,
      "model_forward_time": 1.7086055278778076,
      "step": 63903
    },
    {
      "epoch": 6.094264984130859e-05,
      "step": 63903,
      "training_step_time": 6.994606971740723
    },
    {
      "epoch": 6.0943603515625e-05,
      "model_forward_time": 1.7032058238983154,
      "step": 63904
    },
    {
      "epoch": 6.0943603515625e-05,
      "step": 63904,
      "training_step_time": 7.774227857589722
    },
    {
      "epoch": 6.094455718994141e-05,
      "model_forward_time": 2.257066011428833,
      "step": 63905
    },
    {
      "epoch": 6.094455718994141e-05,
      "step": 63905,
      "training_step_time": 9.234296560287476
    },
    {
      "epoch": 6.094551086425781e-05,
      "model_forward_time": 1.6934480667114258,
      "step": 63906
    },
    {
      "epoch": 6.094551086425781e-05,
      "step": 63906,
      "training_step_time": 9.296391487121582
    },
    {
      "epoch": 6.0946464538574215e-05,
      "model_forward_time": 1.7027993202209473,
      "step": 63907
    },
    {
      "epoch": 6.0946464538574215e-05,
      "step": 63907,
      "training_step_time": 7.9276275634765625
    },
    {
      "epoch": 6.0947418212890626e-05,
      "model_forward_time": 1.6980643272399902,
      "step": 63908
    },
    {
      "epoch": 6.0947418212890626e-05,
      "step": 63908,
      "training_step_time": 7.7222301959991455
    },
    {
      "epoch": 6.094837188720703e-05,
      "model_forward_time": 1.6950700283050537,
      "step": 63909
    },
    {
      "epoch": 6.094837188720703e-05,
      "step": 63909,
      "training_step_time": 7.712698936462402
    },
    {
      "epoch": 6.0949325561523434e-05,
      "grad_norm": 0.11416523903608322,
      "learning_rate": 3.1579936964976286e-06,
      "loss": 0.0563,
      "step": 63910
    },
    {
      "epoch": 6.0949325561523434e-05,
      "model_forward_time": 1.6933729648590088,
      "step": 63910
    },
    {
      "dynamics_loss_avg": 0.06031183917075396,
      "epoch": 6.0949325561523434e-05,
      "step": 63910
    },
    {
      "action_loss_avg": 0.0012947187409736217,
      "epoch": 6.0949325561523434e-05,
      "step": 63910
    },
    {
      "epoch": 6.0949325561523434e-05,
      "step": 63910,
      "training_step_time": 6.945689678192139
    },
    {
      "epoch": 6.0950279235839845e-05,
      "model_forward_time": 1.6939787864685059,
      "step": 63911
    },
    {
      "epoch": 6.0950279235839845e-05,
      "step": 63911,
      "training_step_time": 7.008647441864014
    },
    {
      "epoch": 6.095123291015625e-05,
      "model_forward_time": 1.6972875595092773,
      "step": 63912
    },
    {
      "epoch": 6.095123291015625e-05,
      "step": 63912,
      "training_step_time": 6.968676328659058
    },
    {
      "epoch": 6.095218658447265e-05,
      "model_forward_time": 1.6983380317687988,
      "step": 63913
    },
    {
      "epoch": 6.095218658447265e-05,
      "step": 63913,
      "training_step_time": 7.126457214355469
    },
    {
      "epoch": 6.0953140258789064e-05,
      "model_forward_time": 2.2505316734313965,
      "step": 63914
    },
    {
      "epoch": 6.0953140258789064e-05,
      "step": 63914,
      "training_step_time": 7.787479639053345
    },
    {
      "epoch": 6.095409393310547e-05,
      "model_forward_time": 1.6969187259674072,
      "step": 63915
    },
    {
      "epoch": 6.095409393310547e-05,
      "step": 63915,
      "training_step_time": 9.98997950553894
    },
    {
      "epoch": 6.095504760742187e-05,
      "model_forward_time": 1.6986494064331055,
      "step": 63916
    },
    {
      "epoch": 6.095504760742187e-05,
      "step": 63916,
      "training_step_time": 7.890042304992676
    },
    {
      "epoch": 6.095600128173828e-05,
      "model_forward_time": 1.6966018676757812,
      "step": 63917
    },
    {
      "epoch": 6.095600128173828e-05,
      "step": 63917,
      "training_step_time": 8.924725770950317
    },
    {
      "epoch": 6.095695495605469e-05,
      "model_forward_time": 1.6954786777496338,
      "step": 63918
    },
    {
      "epoch": 6.095695495605469e-05,
      "step": 63918,
      "training_step_time": 7.755746603012085
    },
    {
      "epoch": 6.095790863037109e-05,
      "model_forward_time": 1.7007067203521729,
      "step": 63919
    },
    {
      "epoch": 6.095790863037109e-05,
      "step": 63919,
      "training_step_time": 7.004885196685791
    },
    {
      "epoch": 6.09588623046875e-05,
      "grad_norm": 0.10055670887231827,
      "learning_rate": 3.156456621115615e-06,
      "loss": 0.0581,
      "step": 63920
    },
    {
      "epoch": 6.09588623046875e-05,
      "model_forward_time": 1.694916009902954,
      "step": 63920
    },
    {
      "dynamics_loss_avg": 0.0506289528682828,
      "epoch": 6.09588623046875e-05,
      "step": 63920
    },
    {
      "action_loss_avg": 0.0016345094656571746,
      "epoch": 6.09588623046875e-05,
      "step": 63920
    },
    {
      "epoch": 6.09588623046875e-05,
      "step": 63920,
      "training_step_time": 7.086679935455322
    },
    {
      "epoch": 6.0959815979003906e-05,
      "model_forward_time": 1.7115435600280762,
      "step": 63921
    },
    {
      "epoch": 6.0959815979003906e-05,
      "step": 63921,
      "training_step_time": 7.010497093200684
    },
    {
      "epoch": 6.096076965332031e-05,
      "model_forward_time": 1.70798921585083,
      "step": 63922
    },
    {
      "epoch": 6.096076965332031e-05,
      "step": 63922,
      "training_step_time": 6.992588758468628
    },
    {
      "epoch": 6.096172332763672e-05,
      "model_forward_time": 2.2975683212280273,
      "step": 63923
    },
    {
      "epoch": 6.096172332763672e-05,
      "step": 63923,
      "training_step_time": 7.736347436904907
    },
    {
      "epoch": 6.0962677001953125e-05,
      "model_forward_time": 1.7124519348144531,
      "step": 63924
    },
    {
      "epoch": 6.0962677001953125e-05,
      "step": 63924,
      "training_step_time": 9.815617084503174
    },
    {
      "epoch": 6.096363067626953e-05,
      "model_forward_time": 1.711655855178833,
      "step": 63925
    },
    {
      "epoch": 6.096363067626953e-05,
      "step": 63925,
      "training_step_time": 7.9590373039245605
    },
    {
      "epoch": 6.096458435058594e-05,
      "model_forward_time": 1.7060978412628174,
      "step": 63926
    },
    {
      "epoch": 6.096458435058594e-05,
      "step": 63926,
      "training_step_time": 7.950619697570801
    },
    {
      "epoch": 6.096553802490234e-05,
      "model_forward_time": 1.7157750129699707,
      "step": 63927
    },
    {
      "epoch": 6.096553802490234e-05,
      "step": 63927,
      "training_step_time": 7.840075254440308
    },
    {
      "epoch": 6.096649169921875e-05,
      "model_forward_time": 1.6840996742248535,
      "step": 63928
    },
    {
      "epoch": 6.096649169921875e-05,
      "step": 63928,
      "training_step_time": 7.6839423179626465
    },
    {
      "epoch": 6.096744537353516e-05,
      "model_forward_time": 1.719613790512085,
      "step": 63929
    },
    {
      "epoch": 6.096744537353516e-05,
      "step": 63929,
      "training_step_time": 7.031689643859863
    },
    {
      "epoch": 6.096839904785156e-05,
      "grad_norm": 0.13391214609146118,
      "learning_rate": 3.1549197473407345e-06,
      "loss": 0.0547,
      "step": 63930
    },
    {
      "epoch": 6.096839904785156e-05,
      "model_forward_time": 1.7083854675292969,
      "step": 63930
    },
    {
      "dynamics_loss_avg": 0.04601370319724083,
      "epoch": 6.096839904785156e-05,
      "step": 63930
    },
    {
      "action_loss_avg": 0.0013486697134794667,
      "epoch": 6.096839904785156e-05,
      "step": 63930
    },
    {
      "epoch": 6.096839904785156e-05,
      "step": 63930,
      "training_step_time": 7.023204326629639
    },
    {
      "epoch": 6.0969352722167966e-05,
      "model_forward_time": 1.7214348316192627,
      "step": 63931
    },
    {
      "epoch": 6.0969352722167966e-05,
      "step": 63931,
      "training_step_time": 6.979456186294556
    },
    {
      "epoch": 6.097030639648438e-05,
      "model_forward_time": 2.2819766998291016,
      "step": 63932
    },
    {
      "epoch": 6.097030639648438e-05,
      "step": 63932,
      "training_step_time": 8.760304927825928
    },
    {
      "epoch": 6.097126007080078e-05,
      "model_forward_time": 1.7102906703948975,
      "step": 63933
    },
    {
      "epoch": 6.097126007080078e-05,
      "step": 63933,
      "training_step_time": 8.90406847000122
    },
    {
      "epoch": 6.0972213745117185e-05,
      "model_forward_time": 1.7116951942443848,
      "step": 63934
    },
    {
      "epoch": 6.0972213745117185e-05,
      "step": 63934,
      "training_step_time": 8.878614902496338
    },
    {
      "epoch": 6.0973167419433596e-05,
      "model_forward_time": 1.7152862548828125,
      "step": 63935
    },
    {
      "epoch": 6.0973167419433596e-05,
      "step": 63935,
      "training_step_time": 8.993666648864746
    },
    {
      "epoch": 6.097412109375e-05,
      "model_forward_time": 1.7139010429382324,
      "step": 63936
    },
    {
      "epoch": 6.097412109375e-05,
      "step": 63936,
      "training_step_time": 7.8665361404418945
    },
    {
      "epoch": 6.0975074768066404e-05,
      "model_forward_time": 1.712247371673584,
      "step": 63937
    },
    {
      "epoch": 6.0975074768066404e-05,
      "step": 63937,
      "training_step_time": 7.761492729187012
    },
    {
      "epoch": 6.0976028442382815e-05,
      "model_forward_time": 1.7267072200775146,
      "step": 63938
    },
    {
      "epoch": 6.0976028442382815e-05,
      "step": 63938,
      "training_step_time": 6.822380065917969
    },
    {
      "epoch": 6.097698211669922e-05,
      "model_forward_time": 1.7104666233062744,
      "step": 63939
    },
    {
      "epoch": 6.097698211669922e-05,
      "step": 63939,
      "training_step_time": 7.088118076324463
    },
    {
      "epoch": 6.097793579101562e-05,
      "grad_norm": 0.09063579142093658,
      "learning_rate": 3.1533830753410543e-06,
      "loss": 0.0534,
      "step": 63940
    },
    {
      "epoch": 6.097793579101562e-05,
      "model_forward_time": 1.710784912109375,
      "step": 63940
    },
    {
      "dynamics_loss_avg": 0.0642258339561522,
      "epoch": 6.097793579101562e-05,
      "step": 63940
    },
    {
      "action_loss_avg": 0.001653416600311175,
      "epoch": 6.097793579101562e-05,
      "step": 63940
    },
    {
      "epoch": 6.097793579101562e-05,
      "step": 63940,
      "training_step_time": 7.026230573654175
    },
    {
      "epoch": 6.0978889465332034e-05,
      "model_forward_time": 2.251960277557373,
      "step": 63941
    },
    {
      "epoch": 6.0978889465332034e-05,
      "step": 63941,
      "training_step_time": 7.810749769210815
    },
    {
      "epoch": 6.097984313964844e-05,
      "model_forward_time": 1.7767696380615234,
      "step": 63942
    },
    {
      "epoch": 6.097984313964844e-05,
      "step": 63942,
      "training_step_time": 9.464748620986938
    },
    {
      "epoch": 6.098079681396484e-05,
      "model_forward_time": 1.7177722454071045,
      "step": 63943
    },
    {
      "epoch": 6.098079681396484e-05,
      "step": 63943,
      "training_step_time": 9.948855638504028
    },
    {
      "epoch": 6.098175048828125e-05,
      "model_forward_time": 1.72200345993042,
      "step": 63944
    },
    {
      "epoch": 6.098175048828125e-05,
      "step": 63944,
      "training_step_time": 7.861970901489258
    },
    {
      "epoch": 6.0982704162597657e-05,
      "model_forward_time": 1.7244672775268555,
      "step": 63945
    },
    {
      "epoch": 6.0982704162597657e-05,
      "step": 63945,
      "training_step_time": 7.8832621574401855
    },
    {
      "epoch": 6.098365783691406e-05,
      "model_forward_time": 1.7104058265686035,
      "step": 63946
    },
    {
      "epoch": 6.098365783691406e-05,
      "step": 63946,
      "training_step_time": 7.72326135635376
    },
    {
      "epoch": 6.098461151123047e-05,
      "model_forward_time": 1.7172367572784424,
      "step": 63947
    },
    {
      "epoch": 6.098461151123047e-05,
      "step": 63947,
      "training_step_time": 6.981487035751343
    },
    {
      "epoch": 6.0985565185546875e-05,
      "model_forward_time": 1.725348711013794,
      "step": 63948
    },
    {
      "epoch": 6.0985565185546875e-05,
      "step": 63948,
      "training_step_time": 7.01181173324585
    },
    {
      "epoch": 6.098651885986328e-05,
      "model_forward_time": 1.348963975906372,
      "step": 63949
    },
    {
      "epoch": 6.098651885986328e-05,
      "step": 63949,
      "training_step_time": 7.0129735469818115
    },
    {
      "epoch": 6.098747253417969e-05,
      "grad_norm": 0.10120629519224167,
      "learning_rate": 3.1518466052846264e-06,
      "loss": 0.0523,
      "step": 63950
    },
    {
      "epoch": 6.098747253417969e-05,
      "model_forward_time": 2.279815435409546,
      "step": 63950
    },
    {
      "dynamics_loss_avg": 0.055175235867500304,
      "epoch": 6.098747253417969e-05,
      "step": 63950
    },
    {
      "action_loss_avg": 0.001506408880231902,
      "epoch": 6.098747253417969e-05,
      "step": 63950
    },
    {
      "epoch": 6.098747253417969e-05,
      "step": 63950,
      "training_step_time": 7.765783309936523
    },
    {
      "epoch": 6.0988426208496094e-05,
      "model_forward_time": 1.7807023525238037,
      "step": 63951
    },
    {
      "epoch": 6.0988426208496094e-05,
      "step": 63951,
      "training_step_time": 8.999099254608154
    },
    {
      "epoch": 6.09893798828125e-05,
      "model_forward_time": 1.7188441753387451,
      "step": 63952
    },
    {
      "epoch": 6.09893798828125e-05,
      "step": 63952,
      "training_step_time": 9.627899408340454
    },
    {
      "epoch": 6.099033355712891e-05,
      "model_forward_time": 1.7247910499572754,
      "step": 63953
    },
    {
      "epoch": 6.099033355712891e-05,
      "step": 63953,
      "training_step_time": 8.618815183639526
    },
    {
      "epoch": 6.099128723144531e-05,
      "model_forward_time": 1.7302942276000977,
      "step": 63954
    },
    {
      "epoch": 6.099128723144531e-05,
      "step": 63954,
      "training_step_time": 8.015188694000244
    },
    {
      "epoch": 6.099224090576172e-05,
      "model_forward_time": 1.7211201190948486,
      "step": 63955
    },
    {
      "epoch": 6.099224090576172e-05,
      "step": 63955,
      "training_step_time": 7.894711017608643
    },
    {
      "epoch": 6.099319458007813e-05,
      "model_forward_time": 1.719092607498169,
      "step": 63956
    },
    {
      "epoch": 6.099319458007813e-05,
      "step": 63956,
      "training_step_time": 7.088474750518799
    },
    {
      "epoch": 6.099414825439453e-05,
      "model_forward_time": 1.729630470275879,
      "step": 63957
    },
    {
      "epoch": 6.099414825439453e-05,
      "step": 63957,
      "training_step_time": 7.015502214431763
    },
    {
      "epoch": 6.0995101928710936e-05,
      "model_forward_time": 1.720334529876709,
      "step": 63958
    },
    {
      "epoch": 6.0995101928710936e-05,
      "step": 63958,
      "training_step_time": 7.0435709953308105
    },
    {
      "epoch": 6.099605560302735e-05,
      "model_forward_time": 1.8987531661987305,
      "step": 63959
    },
    {
      "epoch": 6.099605560302735e-05,
      "step": 63959,
      "training_step_time": 7.770734071731567
    },
    {
      "epoch": 6.099700927734375e-05,
      "grad_norm": 0.24182197451591492,
      "learning_rate": 3.1503103373394737e-06,
      "loss": 0.056,
      "step": 63960
    },
    {
      "epoch": 6.099700927734375e-05,
      "model_forward_time": 1.7711844444274902,
      "step": 63960
    },
    {
      "dynamics_loss_avg": 0.07564789187163115,
      "epoch": 6.099700927734375e-05,
      "step": 63960
    },
    {
      "action_loss_avg": 0.0023122394719393923,
      "epoch": 6.099700927734375e-05,
      "step": 63960
    },
    {
      "epoch": 6.099700927734375e-05,
      "step": 63960,
      "training_step_time": 8.436131238937378
    },
    {
      "epoch": 6.0997962951660155e-05,
      "model_forward_time": 1.715069055557251,
      "step": 63961
    },
    {
      "epoch": 6.0997962951660155e-05,
      "step": 63961,
      "training_step_time": 10.701109170913696
    },
    {
      "epoch": 6.0998916625976566e-05,
      "model_forward_time": 1.730611801147461,
      "step": 63962
    },
    {
      "epoch": 6.0998916625976566e-05,
      "step": 63962,
      "training_step_time": 8.070506811141968
    },
    {
      "epoch": 6.099987030029297e-05,
      "model_forward_time": 1.7155897617340088,
      "step": 63963
    },
    {
      "epoch": 6.099987030029297e-05,
      "step": 63963,
      "training_step_time": 8.00611162185669
    },
    {
      "epoch": 6.1000823974609374e-05,
      "model_forward_time": 1.7192981243133545,
      "step": 63964
    },
    {
      "epoch": 6.1000823974609374e-05,
      "step": 63964,
      "training_step_time": 7.940608024597168
    },
    {
      "epoch": 6.1001777648925785e-05,
      "model_forward_time": 1.720574140548706,
      "step": 63965
    },
    {
      "epoch": 6.1001777648925785e-05,
      "step": 63965,
      "training_step_time": 7.776750087738037
    },
    {
      "epoch": 6.100273132324219e-05,
      "model_forward_time": 1.7186906337738037,
      "step": 63966
    },
    {
      "epoch": 6.100273132324219e-05,
      "step": 63966,
      "training_step_time": 7.041658163070679
    },
    {
      "epoch": 6.100368499755859e-05,
      "model_forward_time": 1.721228837966919,
      "step": 63967
    },
    {
      "epoch": 6.100368499755859e-05,
      "step": 63967,
      "training_step_time": 7.042325973510742
    },
    {
      "epoch": 6.1004638671875e-05,
      "model_forward_time": 2.2848129272460938,
      "step": 63968
    },
    {
      "epoch": 6.1004638671875e-05,
      "step": 63968,
      "training_step_time": 7.231223106384277
    },
    {
      "epoch": 6.100559234619141e-05,
      "model_forward_time": 1.7777831554412842,
      "step": 63969
    },
    {
      "epoch": 6.100559234619141e-05,
      "step": 63969,
      "training_step_time": 7.81662130355835
    },
    {
      "epoch": 6.100654602050781e-05,
      "grad_norm": 0.1468595415353775,
      "learning_rate": 3.1487742716736036e-06,
      "loss": 0.0575,
      "step": 63970
    },
    {
      "epoch": 6.100654602050781e-05,
      "model_forward_time": 1.7122714519500732,
      "step": 63970
    },
    {
      "dynamics_loss_avg": 0.0586030931212008,
      "epoch": 6.100654602050781e-05,
      "step": 63970
    },
    {
      "action_loss_avg": 0.0013280657411087304,
      "epoch": 6.100654602050781e-05,
      "step": 63970
    },
    {
      "epoch": 6.100654602050781e-05,
      "step": 63970,
      "training_step_time": 10.052213191986084
    },
    {
      "epoch": 6.1007499694824215e-05,
      "model_forward_time": 1.7324590682983398,
      "step": 63971
    },
    {
      "epoch": 6.1007499694824215e-05,
      "step": 63971,
      "training_step_time": 7.966973066329956
    },
    {
      "epoch": 6.1008453369140626e-05,
      "model_forward_time": 1.7209961414337158,
      "step": 63972
    },
    {
      "epoch": 6.1008453369140626e-05,
      "step": 63972,
      "training_step_time": 7.950366497039795
    },
    {
      "epoch": 6.100940704345703e-05,
      "model_forward_time": 1.7178261280059814,
      "step": 63973
    },
    {
      "epoch": 6.100940704345703e-05,
      "step": 63973,
      "training_step_time": 7.888864278793335
    },
    {
      "epoch": 6.1010360717773434e-05,
      "model_forward_time": 1.7129344940185547,
      "step": 63974
    },
    {
      "epoch": 6.1010360717773434e-05,
      "step": 63974,
      "training_step_time": 7.724254608154297
    },
    {
      "epoch": 6.1011314392089845e-05,
      "model_forward_time": 1.7198231220245361,
      "step": 63975
    },
    {
      "epoch": 6.1011314392089845e-05,
      "step": 63975,
      "training_step_time": 7.003952980041504
    },
    {
      "epoch": 6.101226806640625e-05,
      "model_forward_time": 1.7222800254821777,
      "step": 63976
    },
    {
      "epoch": 6.101226806640625e-05,
      "step": 63976,
      "training_step_time": 7.052335977554321
    },
    {
      "epoch": 6.101322174072265e-05,
      "model_forward_time": 1.7295863628387451,
      "step": 63977
    },
    {
      "epoch": 6.101322174072265e-05,
      "step": 63977,
      "training_step_time": 7.774414300918579
    },
    {
      "epoch": 6.1014175415039064e-05,
      "model_forward_time": 1.7796964645385742,
      "step": 63978
    },
    {
      "epoch": 6.1014175415039064e-05,
      "step": 63978,
      "training_step_time": 7.962252378463745
    },
    {
      "epoch": 6.101512908935547e-05,
      "model_forward_time": 1.7306926250457764,
      "step": 63979
    },
    {
      "epoch": 6.101512908935547e-05,
      "step": 63979,
      "training_step_time": 9.039688348770142
    },
    {
      "epoch": 6.101608276367187e-05,
      "grad_norm": 0.09366324543952942,
      "learning_rate": 3.147238408454995e-06,
      "loss": 0.0559,
      "step": 63980
    },
    {
      "epoch": 6.101608276367187e-05,
      "model_forward_time": 1.731558084487915,
      "step": 63980
    },
    {
      "dynamics_loss_avg": 0.054464191943407056,
      "epoch": 6.101608276367187e-05,
      "step": 63980
    },
    {
      "action_loss_avg": 0.0019336464174557477,
      "epoch": 6.101608276367187e-05,
      "step": 63980
    },
    {
      "epoch": 6.101608276367187e-05,
      "step": 63980,
      "training_step_time": 8.771034717559814
    },
    {
      "epoch": 6.101703643798828e-05,
      "model_forward_time": 1.697373628616333,
      "step": 63981
    },
    {
      "epoch": 6.101703643798828e-05,
      "step": 63981,
      "training_step_time": 9.722745656967163
    },
    {
      "epoch": 6.101799011230469e-05,
      "model_forward_time": 1.7068138122558594,
      "step": 63982
    },
    {
      "epoch": 6.101799011230469e-05,
      "step": 63982,
      "training_step_time": 7.989814281463623
    },
    {
      "epoch": 6.101894378662109e-05,
      "model_forward_time": 1.721836805343628,
      "step": 63983
    },
    {
      "epoch": 6.101894378662109e-05,
      "step": 63983,
      "training_step_time": 7.777080297470093
    },
    {
      "epoch": 6.10198974609375e-05,
      "model_forward_time": 0.6994051933288574,
      "step": 63984
    },
    {
      "epoch": 6.10198974609375e-05,
      "step": 63984,
      "training_step_time": 7.043408632278442
    },
    {
      "epoch": 6.1020851135253906e-05,
      "model_forward_time": 1.7227041721343994,
      "step": 63985
    },
    {
      "epoch": 6.1020851135253906e-05,
      "step": 63985,
      "training_step_time": 7.079193830490112
    },
    {
      "epoch": 6.102180480957031e-05,
      "model_forward_time": 1.7142674922943115,
      "step": 63986
    },
    {
      "epoch": 6.102180480957031e-05,
      "step": 63986,
      "training_step_time": 7.296996355056763
    },
    {
      "epoch": 6.102275848388672e-05,
      "model_forward_time": 2.289668083190918,
      "step": 63987
    },
    {
      "epoch": 6.102275848388672e-05,
      "step": 63987,
      "training_step_time": 9.01404595375061
    },
    {
      "epoch": 6.1023712158203125e-05,
      "model_forward_time": 1.7080392837524414,
      "step": 63988
    },
    {
      "epoch": 6.1023712158203125e-05,
      "step": 63988,
      "training_step_time": 8.840816259384155
    },
    {
      "epoch": 6.102466583251953e-05,
      "model_forward_time": 1.7089226245880127,
      "step": 63989
    },
    {
      "epoch": 6.102466583251953e-05,
      "step": 63989,
      "training_step_time": 8.688840866088867
    },
    {
      "epoch": 6.102561950683594e-05,
      "grad_norm": 0.08539582043886185,
      "learning_rate": 3.1457027478516087e-06,
      "loss": 0.0588,
      "step": 63990
    },
    {
      "epoch": 6.102561950683594e-05,
      "model_forward_time": 1.7058300971984863,
      "step": 63990
    },
    {
      "dynamics_loss_avg": 0.0423016925342381,
      "epoch": 6.102561950683594e-05,
      "step": 63990
    },
    {
      "action_loss_avg": 0.0011223127876291983,
      "epoch": 6.102561950683594e-05,
      "step": 63990
    },
    {
      "epoch": 6.102561950683594e-05,
      "step": 63990,
      "training_step_time": 8.769949197769165
    },
    {
      "epoch": 6.1026573181152343e-05,
      "model_forward_time": 1.7034544944763184,
      "step": 63991
    },
    {
      "epoch": 6.1026573181152343e-05,
      "step": 63991,
      "training_step_time": 7.7457661628723145
    },
    {
      "epoch": 6.102752685546875e-05,
      "model_forward_time": 1.7316672801971436,
      "step": 63992
    },
    {
      "epoch": 6.102752685546875e-05,
      "step": 63992,
      "training_step_time": 7.895921230316162
    },
    {
      "epoch": 6.102848052978516e-05,
      "model_forward_time": 1.7069337368011475,
      "step": 63993
    },
    {
      "epoch": 6.102848052978516e-05,
      "step": 63993,
      "training_step_time": 6.997513055801392
    },
    {
      "epoch": 6.102943420410156e-05,
      "model_forward_time": 1.710089921951294,
      "step": 63994
    },
    {
      "epoch": 6.102943420410156e-05,
      "step": 63994,
      "training_step_time": 7.013838291168213
    },
    {
      "epoch": 6.1030387878417966e-05,
      "model_forward_time": 1.7085187435150146,
      "step": 63995
    },
    {
      "epoch": 6.1030387878417966e-05,
      "step": 63995,
      "training_step_time": 7.322188138961792
    },
    {
      "epoch": 6.103134155273438e-05,
      "model_forward_time": 2.295395851135254,
      "step": 63996
    },
    {
      "epoch": 6.103134155273438e-05,
      "step": 63996,
      "training_step_time": 7.8431715965271
    },
    {
      "epoch": 6.103229522705078e-05,
      "model_forward_time": 1.7234835624694824,
      "step": 63997
    },
    {
      "epoch": 6.103229522705078e-05,
      "step": 63997,
      "training_step_time": 8.72469711303711
    },
    {
      "epoch": 6.1033248901367185e-05,
      "model_forward_time": 1.7438805103302002,
      "step": 63998
    },
    {
      "epoch": 6.1033248901367185e-05,
      "step": 63998,
      "training_step_time": 8.899898290634155
    },
    {
      "epoch": 6.1034202575683596e-05,
      "model_forward_time": 1.710392951965332,
      "step": 63999
    },
    {
      "epoch": 6.1034202575683596e-05,
      "step": 63999,
      "training_step_time": 7.944591760635376
    },
    {
      "epoch": 6.103515625e-05,
      "grad_norm": 0.09972784668207169,
      "learning_rate": 3.1441672900313818e-06,
      "loss": 0.0566,
      "step": 64000
    },
    {
      "epoch": 6.103515625e-05,
      "model_forward_time": 1.776672124862671,
      "step": 64000
    },
    {
      "dynamics_loss_avg": 0.06289575956761836,
      "epoch": 6.103515625e-05,
      "step": 64000
    },
    {
      "action_loss_avg": 0.001634404232027009,
      "epoch": 6.103515625e-05,
      "step": 64000
    },
    {
      "epoch": 6.103515625e-05,
      "step": 64000,
      "training_step_time": 6.509232044219971
    },
    {
      "epoch": 6.103610992431641e-05,
      "model_forward_time": 1.6985313892364502,
      "step": 64001
    },
    {
      "epoch": 6.103610992431641e-05,
      "step": 64001,
      "training_step_time": 8.39359164237976
    },
    {
      "epoch": 6.103706359863281e-05,
      "model_forward_time": 1.7023797035217285,
      "step": 64002
    },
    {
      "epoch": 6.103706359863281e-05,
      "step": 64002,
      "training_step_time": 6.898303031921387
    },
    {
      "epoch": 6.103801727294922e-05,
      "model_forward_time": 1.7008185386657715,
      "step": 64003
    },
    {
      "epoch": 6.103801727294922e-05,
      "step": 64003,
      "training_step_time": 6.949098110198975
    },
    {
      "epoch": 6.103897094726563e-05,
      "model_forward_time": 1.6877868175506592,
      "step": 64004
    },
    {
      "epoch": 6.103897094726563e-05,
      "step": 64004,
      "training_step_time": 8.293667554855347
    },
    {
      "epoch": 6.103992462158203e-05,
      "model_forward_time": 2.729271173477173,
      "step": 64005
    },
    {
      "epoch": 6.103992462158203e-05,
      "step": 64005,
      "training_step_time": 8.424590110778809
    },
    {
      "epoch": 6.104087829589844e-05,
      "model_forward_time": 1.7659401893615723,
      "step": 64006
    },
    {
      "epoch": 6.104087829589844e-05,
      "step": 64006,
      "training_step_time": 8.6153244972229
    },
    {
      "epoch": 6.104183197021485e-05,
      "model_forward_time": 1.7132339477539062,
      "step": 64007
    },
    {
      "epoch": 6.104183197021485e-05,
      "step": 64007,
      "training_step_time": 13.458249568939209
    },
    {
      "epoch": 6.104278564453125e-05,
      "model_forward_time": 1.7159640789031982,
      "step": 64008
    },
    {
      "epoch": 6.104278564453125e-05,
      "step": 64008,
      "training_step_time": 12.496856212615967
    },
    {
      "epoch": 6.104373931884766e-05,
      "model_forward_time": 1.7091238498687744,
      "step": 64009
    },
    {
      "epoch": 6.104373931884766e-05,
      "step": 64009,
      "training_step_time": 7.644674777984619
    },
    {
      "epoch": 6.104469299316407e-05,
      "grad_norm": 0.1035885289311409,
      "learning_rate": 3.1426320351622276e-06,
      "loss": 0.0616,
      "step": 64010
    },
    {
      "epoch": 6.104469299316407e-05,
      "model_forward_time": 1.7044241428375244,
      "step": 64010
    },
    {
      "dynamics_loss_avg": 0.05871641505509615,
      "epoch": 6.104469299316407e-05,
      "step": 64010
    },
    {
      "action_loss_avg": 0.0012162481056293474,
      "epoch": 6.104469299316407e-05,
      "step": 64010
    },
    {
      "epoch": 6.104469299316407e-05,
      "step": 64010,
      "training_step_time": 7.605072498321533
    },
    {
      "epoch": 6.104564666748046e-05,
      "model_forward_time": 1.7193472385406494,
      "step": 64011
    },
    {
      "epoch": 6.104564666748046e-05,
      "step": 64011,
      "training_step_time": 7.581773281097412
    },
    {
      "epoch": 6.104660034179688e-05,
      "model_forward_time": 1.7103722095489502,
      "step": 64012
    },
    {
      "epoch": 6.104660034179688e-05,
      "step": 64012,
      "training_step_time": 6.557149410247803
    },
    {
      "epoch": 6.104755401611329e-05,
      "model_forward_time": 1.7165772914886475,
      "step": 64013
    },
    {
      "epoch": 6.104755401611329e-05,
      "step": 64013,
      "training_step_time": 6.571728944778442
    },
    {
      "epoch": 6.104850769042968e-05,
      "model_forward_time": 2.2777597904205322,
      "step": 64014
    },
    {
      "epoch": 6.104850769042968e-05,
      "step": 64014,
      "training_step_time": 7.575357437133789
    },
    {
      "epoch": 6.10494613647461e-05,
      "model_forward_time": 1.773791790008545,
      "step": 64015
    },
    {
      "epoch": 6.10494613647461e-05,
      "step": 64015,
      "training_step_time": 10.223031520843506
    },
    {
      "epoch": 6.10504150390625e-05,
      "model_forward_time": 1.7312521934509277,
      "step": 64016
    },
    {
      "epoch": 6.10504150390625e-05,
      "step": 64016,
      "training_step_time": 11.4185311794281
    },
    {
      "epoch": 6.10513687133789e-05,
      "model_forward_time": 1.738684892654419,
      "step": 64017
    },
    {
      "epoch": 6.10513687133789e-05,
      "step": 64017,
      "training_step_time": 10.325589179992676
    },
    {
      "epoch": 6.105232238769531e-05,
      "model_forward_time": 1.7158925533294678,
      "step": 64018
    },
    {
      "epoch": 6.105232238769531e-05,
      "step": 64018,
      "training_step_time": 7.575981378555298
    },
    {
      "epoch": 6.105327606201172e-05,
      "model_forward_time": 1.726942539215088,
      "step": 64019
    },
    {
      "epoch": 6.105327606201172e-05,
      "step": 64019,
      "training_step_time": 7.730512619018555
    },
    {
      "epoch": 6.105422973632812e-05,
      "grad_norm": 0.09082289785146713,
      "learning_rate": 3.1410969834120443e-06,
      "loss": 0.0586,
      "step": 64020
    },
    {
      "epoch": 6.105422973632812e-05,
      "model_forward_time": 1.7191050052642822,
      "step": 64020
    },
    {
      "dynamics_loss_avg": 0.06656916476786137,
      "epoch": 6.105422973632812e-05,
      "step": 64020
    },
    {
      "action_loss_avg": 0.0019464185897959396,
      "epoch": 6.105422973632812e-05,
      "step": 64020
    },
    {
      "epoch": 6.105422973632812e-05,
      "step": 64020,
      "training_step_time": 7.651339769363403
    },
    {
      "epoch": 6.105518341064453e-05,
      "model_forward_time": 1.7256865501403809,
      "step": 64021
    },
    {
      "epoch": 6.105518341064453e-05,
      "step": 64021,
      "training_step_time": 6.557119607925415
    },
    {
      "epoch": 6.105613708496094e-05,
      "model_forward_time": 1.7177603244781494,
      "step": 64022
    },
    {
      "epoch": 6.105613708496094e-05,
      "step": 64022,
      "training_step_time": 6.545760154724121
    },
    {
      "epoch": 6.105709075927734e-05,
      "model_forward_time": 2.3004162311553955,
      "step": 64023
    },
    {
      "epoch": 6.105709075927734e-05,
      "step": 64023,
      "training_step_time": 7.437309265136719
    },
    {
      "epoch": 6.105804443359375e-05,
      "model_forward_time": 1.7412662506103516,
      "step": 64024
    },
    {
      "epoch": 6.105804443359375e-05,
      "step": 64024,
      "training_step_time": 7.618790149688721
    },
    {
      "epoch": 6.105899810791016e-05,
      "model_forward_time": 1.7398667335510254,
      "step": 64025
    },
    {
      "epoch": 6.105899810791016e-05,
      "step": 64025,
      "training_step_time": 7.721332311630249
    },
    {
      "epoch": 6.105995178222656e-05,
      "model_forward_time": 1.7399482727050781,
      "step": 64026
    },
    {
      "epoch": 6.105995178222656e-05,
      "step": 64026,
      "training_step_time": 10.309126377105713
    },
    {
      "epoch": 6.106090545654297e-05,
      "model_forward_time": 1.7346208095550537,
      "step": 64027
    },
    {
      "epoch": 6.106090545654297e-05,
      "step": 64027,
      "training_step_time": 7.696185111999512
    },
    {
      "epoch": 6.106185913085938e-05,
      "model_forward_time": 1.7534966468811035,
      "step": 64028
    },
    {
      "epoch": 6.106185913085938e-05,
      "step": 64028,
      "training_step_time": 8.393590688705444
    },
    {
      "epoch": 6.106281280517578e-05,
      "model_forward_time": 1.7273859977722168,
      "step": 64029
    },
    {
      "epoch": 6.106281280517578e-05,
      "step": 64029,
      "training_step_time": 7.70428204536438
    },
    {
      "epoch": 6.106376647949219e-05,
      "grad_norm": 0.09401815384626389,
      "learning_rate": 3.1395621349486994e-06,
      "loss": 0.0613,
      "step": 64030
    },
    {
      "epoch": 6.106376647949219e-05,
      "model_forward_time": 1.7146122455596924,
      "step": 64030
    },
    {
      "dynamics_loss_avg": 0.0595978120341897,
      "epoch": 6.106376647949219e-05,
      "step": 64030
    },
    {
      "action_loss_avg": 0.001240122693707235,
      "epoch": 6.106376647949219e-05,
      "step": 64030
    },
    {
      "epoch": 6.106376647949219e-05,
      "step": 64030,
      "training_step_time": 6.624887228012085
    },
    {
      "epoch": 6.10647201538086e-05,
      "model_forward_time": 1.723243236541748,
      "step": 64031
    },
    {
      "epoch": 6.10647201538086e-05,
      "step": 64031,
      "training_step_time": 6.619102716445923
    },
    {
      "epoch": 6.1065673828125e-05,
      "model_forward_time": 2.2886054515838623,
      "step": 64032
    },
    {
      "epoch": 6.1065673828125e-05,
      "step": 64032,
      "training_step_time": 7.570284128189087
    },
    {
      "epoch": 6.106662750244141e-05,
      "model_forward_time": 1.719820499420166,
      "step": 64033
    },
    {
      "epoch": 6.106662750244141e-05,
      "step": 64033,
      "training_step_time": 7.619955778121948
    },
    {
      "epoch": 6.106758117675782e-05,
      "model_forward_time": 1.7670576572418213,
      "step": 64034
    },
    {
      "epoch": 6.106758117675782e-05,
      "step": 64034,
      "training_step_time": 7.771073818206787
    },
    {
      "epoch": 6.106853485107422e-05,
      "model_forward_time": 1.7288904190063477,
      "step": 64035
    },
    {
      "epoch": 6.106853485107422e-05,
      "step": 64035,
      "training_step_time": 7.6777026653289795
    },
    {
      "epoch": 6.106948852539063e-05,
      "model_forward_time": 1.7230806350708008,
      "step": 64036
    },
    {
      "epoch": 6.106948852539063e-05,
      "step": 64036,
      "training_step_time": 7.700854063034058
    },
    {
      "epoch": 6.107044219970704e-05,
      "model_forward_time": 1.7210800647735596,
      "step": 64037
    },
    {
      "epoch": 6.107044219970704e-05,
      "step": 64037,
      "training_step_time": 7.786301136016846
    },
    {
      "epoch": 6.107139587402343e-05,
      "model_forward_time": 1.7182912826538086,
      "step": 64038
    },
    {
      "epoch": 6.107139587402343e-05,
      "step": 64038,
      "training_step_time": 7.622903108596802
    },
    {
      "epoch": 6.107234954833985e-05,
      "model_forward_time": 1.7200016975402832,
      "step": 64039
    },
    {
      "epoch": 6.107234954833985e-05,
      "step": 64039,
      "training_step_time": 7.661142826080322
    },
    {
      "epoch": 6.107330322265626e-05,
      "grad_norm": 0.08435936272144318,
      "learning_rate": 3.1380274899400426e-06,
      "loss": 0.059,
      "step": 64040
    },
    {
      "epoch": 6.107330322265626e-05,
      "model_forward_time": 1.7162349224090576,
      "step": 64040
    },
    {
      "dynamics_loss_avg": 0.05517579633742571,
      "epoch": 6.107330322265626e-05,
      "step": 64040
    },
    {
      "action_loss_avg": 0.0014935817249352112,
      "epoch": 6.107330322265626e-05,
      "step": 64040
    },
    {
      "epoch": 6.107330322265626e-05,
      "step": 64040,
      "training_step_time": 6.81899619102478
    },
    {
      "epoch": 6.107425689697265e-05,
      "model_forward_time": 2.297301769256592,
      "step": 64041
    },
    {
      "epoch": 6.107425689697265e-05,
      "step": 64041,
      "training_step_time": 7.559188365936279
    },
    {
      "epoch": 6.107521057128906e-05,
      "model_forward_time": 1.72935152053833,
      "step": 64042
    },
    {
      "epoch": 6.107521057128906e-05,
      "step": 64042,
      "training_step_time": 7.65151309967041
    },
    {
      "epoch": 6.107616424560547e-05,
      "model_forward_time": 1.7629859447479248,
      "step": 64043
    },
    {
      "epoch": 6.107616424560547e-05,
      "step": 64043,
      "training_step_time": 7.740894556045532
    },
    {
      "epoch": 6.107711791992187e-05,
      "model_forward_time": 1.721540927886963,
      "step": 64044
    },
    {
      "epoch": 6.107711791992187e-05,
      "step": 64044,
      "training_step_time": 7.674715280532837
    },
    {
      "epoch": 6.107807159423828e-05,
      "model_forward_time": 1.7280540466308594,
      "step": 64045
    },
    {
      "epoch": 6.107807159423828e-05,
      "step": 64045,
      "training_step_time": 7.761744260787964
    },
    {
      "epoch": 6.10790252685547e-05,
      "model_forward_time": 1.7189216613769531,
      "step": 64046
    },
    {
      "epoch": 6.10790252685547e-05,
      "step": 64046,
      "training_step_time": 7.67832612991333
    },
    {
      "epoch": 6.107997894287109e-05,
      "model_forward_time": 1.7273480892181396,
      "step": 64047
    },
    {
      "epoch": 6.107997894287109e-05,
      "step": 64047,
      "training_step_time": 7.591712713241577
    },
    {
      "epoch": 6.10809326171875e-05,
      "model_forward_time": 1.7260560989379883,
      "step": 64048
    },
    {
      "epoch": 6.10809326171875e-05,
      "step": 64048,
      "training_step_time": 7.6761603355407715
    },
    {
      "epoch": 6.108188629150391e-05,
      "model_forward_time": 1.7217788696289062,
      "step": 64049
    },
    {
      "epoch": 6.108188629150391e-05,
      "step": 64049,
      "training_step_time": 6.672829627990723
    },
    {
      "epoch": 6.108283996582031e-05,
      "grad_norm": 0.13811005651950836,
      "learning_rate": 3.1364930485539004e-06,
      "loss": 0.0584,
      "step": 64050
    },
    {
      "epoch": 6.108283996582031e-05,
      "model_forward_time": 1.3440570831298828,
      "step": 64050
    },
    {
      "dynamics_loss_avg": 0.05854591652750969,
      "epoch": 6.108283996582031e-05,
      "step": 64050
    },
    {
      "action_loss_avg": 0.0019431863562203943,
      "epoch": 6.108283996582031e-05,
      "step": 64050
    },
    {
      "epoch": 6.108283996582031e-05,
      "step": 64050,
      "training_step_time": 6.614840745925903
    },
    {
      "epoch": 6.108379364013672e-05,
      "model_forward_time": 2.280019521713257,
      "step": 64051
    },
    {
      "epoch": 6.108379364013672e-05,
      "step": 64051,
      "training_step_time": 7.633172988891602
    },
    {
      "epoch": 6.108474731445313e-05,
      "model_forward_time": 1.7655820846557617,
      "step": 64052
    },
    {
      "epoch": 6.108474731445313e-05,
      "step": 64052,
      "training_step_time": 10.009166479110718
    },
    {
      "epoch": 6.108570098876953e-05,
      "model_forward_time": 1.7216598987579346,
      "step": 64053
    },
    {
      "epoch": 6.108570098876953e-05,
      "step": 64053,
      "training_step_time": 7.740142107009888
    },
    {
      "epoch": 6.108665466308594e-05,
      "model_forward_time": 1.729058027267456,
      "step": 64054
    },
    {
      "epoch": 6.108665466308594e-05,
      "step": 64054,
      "training_step_time": 9.686750888824463
    },
    {
      "epoch": 6.108760833740235e-05,
      "model_forward_time": 1.7322957515716553,
      "step": 64055
    },
    {
      "epoch": 6.108760833740235e-05,
      "step": 64055,
      "training_step_time": 7.613731384277344
    },
    {
      "epoch": 6.108856201171875e-05,
      "model_forward_time": 1.7313692569732666,
      "step": 64056
    },
    {
      "epoch": 6.108856201171875e-05,
      "step": 64056,
      "training_step_time": 7.634663105010986
    },
    {
      "epoch": 6.108951568603516e-05,
      "model_forward_time": 1.7351646423339844,
      "step": 64057
    },
    {
      "epoch": 6.108951568603516e-05,
      "step": 64057,
      "training_step_time": 7.621907711029053
    },
    {
      "epoch": 6.109046936035157e-05,
      "model_forward_time": 1.7418663501739502,
      "step": 64058
    },
    {
      "epoch": 6.109046936035157e-05,
      "step": 64058,
      "training_step_time": 6.669891595840454
    },
    {
      "epoch": 6.109142303466797e-05,
      "model_forward_time": 1.7286109924316406,
      "step": 64059
    },
    {
      "epoch": 6.109142303466797e-05,
      "step": 64059,
      "training_step_time": 6.600735902786255
    },
    {
      "epoch": 6.109237670898438e-05,
      "grad_norm": 0.10882080346345901,
      "learning_rate": 3.134958810958076e-06,
      "loss": 0.0579,
      "step": 64060
    },
    {
      "epoch": 6.109237670898438e-05,
      "model_forward_time": 2.294613838195801,
      "step": 64060
    },
    {
      "dynamics_loss_avg": 0.042727816849946976,
      "epoch": 6.109237670898438e-05,
      "step": 64060
    },
    {
      "action_loss_avg": 0.0011352926841937006,
      "epoch": 6.109237670898438e-05,
      "step": 64060
    },
    {
      "epoch": 6.109237670898438e-05,
      "step": 64060,
      "training_step_time": 8.232262372970581
    },
    {
      "epoch": 6.109333038330079e-05,
      "model_forward_time": 1.7748939990997314,
      "step": 64061
    },
    {
      "epoch": 6.109333038330079e-05,
      "step": 64061,
      "training_step_time": 7.653498888015747
    },
    {
      "epoch": 6.109428405761719e-05,
      "model_forward_time": 1.7381253242492676,
      "step": 64062
    },
    {
      "epoch": 6.109428405761719e-05,
      "step": 64062,
      "training_step_time": 9.950929164886475
    },
    {
      "epoch": 6.10952377319336e-05,
      "model_forward_time": 1.7411658763885498,
      "step": 64063
    },
    {
      "epoch": 6.10952377319336e-05,
      "step": 64063,
      "training_step_time": 8.639754295349121
    },
    {
      "epoch": 6.109619140625e-05,
      "model_forward_time": 1.7416810989379883,
      "step": 64064
    },
    {
      "epoch": 6.109619140625e-05,
      "step": 64064,
      "training_step_time": 8.24980354309082
    },
    {
      "epoch": 6.10971450805664e-05,
      "model_forward_time": 1.742323637008667,
      "step": 64065
    },
    {
      "epoch": 6.10971450805664e-05,
      "step": 64065,
      "training_step_time": 7.757473707199097
    },
    {
      "epoch": 6.109809875488281e-05,
      "model_forward_time": 1.7448453903198242,
      "step": 64066
    },
    {
      "epoch": 6.109809875488281e-05,
      "step": 64066,
      "training_step_time": 8.740877151489258
    },
    {
      "epoch": 6.109905242919921e-05,
      "model_forward_time": 1.7402307987213135,
      "step": 64067
    },
    {
      "epoch": 6.109905242919921e-05,
      "step": 64067,
      "training_step_time": 6.7442567348480225
    },
    {
      "epoch": 6.110000610351562e-05,
      "model_forward_time": 1.7517330646514893,
      "step": 64068
    },
    {
      "epoch": 6.110000610351562e-05,
      "step": 64068,
      "training_step_time": 6.753520250320435
    },
    {
      "epoch": 6.110095977783203e-05,
      "model_forward_time": 2.28338360786438,
      "step": 64069
    },
    {
      "epoch": 6.110095977783203e-05,
      "step": 64069,
      "training_step_time": 7.689765930175781
    },
    {
      "epoch": 6.110191345214843e-05,
      "grad_norm": 0.10321900993585587,
      "learning_rate": 3.1334247773203503e-06,
      "loss": 0.056,
      "step": 64070
    },
    {
      "epoch": 6.110191345214843e-05,
      "model_forward_time": 1.783595085144043,
      "step": 64070
    },
    {
      "dynamics_loss_avg": 0.05978514328598976,
      "epoch": 6.110191345214843e-05,
      "step": 64070
    },
    {
      "action_loss_avg": 0.002156677556922659,
      "epoch": 6.110191345214843e-05,
      "step": 64070
    },
    {
      "epoch": 6.110191345214843e-05,
      "step": 64070,
      "training_step_time": 7.558202266693115
    },
    {
      "epoch": 6.110286712646484e-05,
      "model_forward_time": 1.747957468032837,
      "step": 64071
    },
    {
      "epoch": 6.110286712646484e-05,
      "step": 64071,
      "training_step_time": 8.34001612663269
    },
    {
      "epoch": 6.110382080078125e-05,
      "model_forward_time": 1.728900671005249,
      "step": 64072
    },
    {
      "epoch": 6.110382080078125e-05,
      "step": 64072,
      "training_step_time": 8.591655254364014
    },
    {
      "epoch": 6.110477447509765e-05,
      "model_forward_time": 1.7513551712036133,
      "step": 64073
    },
    {
      "epoch": 6.110477447509765e-05,
      "step": 64073,
      "training_step_time": 8.735272884368896
    },
    {
      "epoch": 6.110572814941406e-05,
      "model_forward_time": 1.7418363094329834,
      "step": 64074
    },
    {
      "epoch": 6.110572814941406e-05,
      "step": 64074,
      "training_step_time": 7.708613395690918
    },
    {
      "epoch": 6.110668182373047e-05,
      "model_forward_time": 1.7554748058319092,
      "step": 64075
    },
    {
      "epoch": 6.110668182373047e-05,
      "step": 64075,
      "training_step_time": 7.557430744171143
    },
    {
      "epoch": 6.110763549804687e-05,
      "model_forward_time": 1.745476245880127,
      "step": 64076
    },
    {
      "epoch": 6.110763549804687e-05,
      "step": 64076,
      "training_step_time": 7.606561899185181
    },
    {
      "epoch": 6.110858917236328e-05,
      "model_forward_time": 1.736152172088623,
      "step": 64077
    },
    {
      "epoch": 6.110858917236328e-05,
      "step": 64077,
      "training_step_time": 6.634971857070923
    },
    {
      "epoch": 6.110954284667969e-05,
      "model_forward_time": 2.3192601203918457,
      "step": 64078
    },
    {
      "epoch": 6.110954284667969e-05,
      "step": 64078,
      "training_step_time": 7.567359209060669
    },
    {
      "epoch": 6.111049652099609e-05,
      "model_forward_time": 1.789522409439087,
      "step": 64079
    },
    {
      "epoch": 6.111049652099609e-05,
      "step": 64079,
      "training_step_time": 7.495373964309692
    },
    {
      "epoch": 6.11114501953125e-05,
      "grad_norm": 0.10613884776830673,
      "learning_rate": 3.131890947808487e-06,
      "loss": 0.06,
      "step": 64080
    },
    {
      "epoch": 6.11114501953125e-05,
      "model_forward_time": 1.741762638092041,
      "step": 64080
    },
    {
      "dynamics_loss_avg": 0.0599273944273591,
      "epoch": 6.11114501953125e-05,
      "step": 64080
    },
    {
      "action_loss_avg": 0.0011790744989411905,
      "epoch": 6.11114501953125e-05,
      "step": 64080
    },
    {
      "epoch": 6.11114501953125e-05,
      "step": 64080,
      "training_step_time": 9.448082208633423
    },
    {
      "epoch": 6.111240386962891e-05,
      "model_forward_time": 1.7189109325408936,
      "step": 64081
    },
    {
      "epoch": 6.111240386962891e-05,
      "step": 64081,
      "training_step_time": 7.746752023696899
    },
    {
      "epoch": 6.11133575439453e-05,
      "model_forward_time": 1.7295215129852295,
      "step": 64082
    },
    {
      "epoch": 6.11133575439453e-05,
      "step": 64082,
      "training_step_time": 8.903932332992554
    },
    {
      "epoch": 6.111431121826172e-05,
      "model_forward_time": 1.7287652492523193,
      "step": 64083
    },
    {
      "epoch": 6.111431121826172e-05,
      "step": 64083,
      "training_step_time": 9.385316133499146
    },
    {
      "epoch": 6.111526489257813e-05,
      "model_forward_time": 1.7359366416931152,
      "step": 64084
    },
    {
      "epoch": 6.111526489257813e-05,
      "step": 64084,
      "training_step_time": 7.406317949295044
    },
    {
      "epoch": 6.111621856689453e-05,
      "model_forward_time": 1.7224085330963135,
      "step": 64085
    },
    {
      "epoch": 6.111621856689453e-05,
      "step": 64085,
      "training_step_time": 7.671121835708618
    },
    {
      "epoch": 6.111717224121094e-05,
      "model_forward_time": 1.7277367115020752,
      "step": 64086
    },
    {
      "epoch": 6.111717224121094e-05,
      "step": 64086,
      "training_step_time": 6.705733060836792
    },
    {
      "epoch": 6.111812591552735e-05,
      "model_forward_time": 2.275432586669922,
      "step": 64087
    },
    {
      "epoch": 6.111812591552735e-05,
      "step": 64087,
      "training_step_time": 8.226669073104858
    },
    {
      "epoch": 6.111907958984374e-05,
      "model_forward_time": 1.7745020389556885,
      "step": 64088
    },
    {
      "epoch": 6.111907958984374e-05,
      "step": 64088,
      "training_step_time": 7.5521559715271
    },
    {
      "epoch": 6.112003326416015e-05,
      "model_forward_time": 1.734065055847168,
      "step": 64089
    },
    {
      "epoch": 6.112003326416015e-05,
      "step": 64089,
      "training_step_time": 7.843710660934448
    },
    {
      "epoch": 6.112098693847657e-05,
      "grad_norm": 0.09949607402086258,
      "learning_rate": 3.13035732259022e-06,
      "loss": 0.0588,
      "step": 64090
    },
    {
      "epoch": 6.112098693847657e-05,
      "model_forward_time": 1.718303918838501,
      "step": 64090
    },
    {
      "dynamics_loss_avg": 0.05448477733880282,
      "epoch": 6.112098693847657e-05,
      "step": 64090
    },
    {
      "action_loss_avg": 0.001382782935979776,
      "epoch": 6.112098693847657e-05,
      "step": 64090
    },
    {
      "epoch": 6.112098693847657e-05,
      "step": 64090,
      "training_step_time": 10.591399908065796
    },
    {
      "epoch": 6.112194061279296e-05,
      "model_forward_time": 1.7295153141021729,
      "step": 64091
    },
    {
      "epoch": 6.112194061279296e-05,
      "step": 64091,
      "training_step_time": 8.90671706199646
    },
    {
      "epoch": 6.112289428710937e-05,
      "model_forward_time": 1.7177393436431885,
      "step": 64092
    },
    {
      "epoch": 6.112289428710937e-05,
      "step": 64092,
      "training_step_time": 7.803385019302368
    },
    {
      "epoch": 6.112384796142578e-05,
      "model_forward_time": 1.3543047904968262,
      "step": 64093
    },
    {
      "epoch": 6.112384796142578e-05,
      "step": 64093,
      "training_step_time": 7.735587120056152
    },
    {
      "epoch": 6.112480163574218e-05,
      "model_forward_time": 1.7255020141601562,
      "step": 64094
    },
    {
      "epoch": 6.112480163574218e-05,
      "step": 64094,
      "training_step_time": 7.579110622406006
    },
    {
      "epoch": 6.112575531005859e-05,
      "model_forward_time": 1.7264809608459473,
      "step": 64095
    },
    {
      "epoch": 6.112575531005859e-05,
      "step": 64095,
      "training_step_time": 7.66120719909668
    },
    {
      "epoch": 6.1126708984375e-05,
      "model_forward_time": 1.7206127643585205,
      "step": 64096
    },
    {
      "epoch": 6.1126708984375e-05,
      "step": 64096,
      "training_step_time": 7.438257932662964
    },
    {
      "epoch": 6.11276626586914e-05,
      "model_forward_time": 2.30637264251709,
      "step": 64097
    },
    {
      "epoch": 6.11276626586914e-05,
      "step": 64097,
      "training_step_time": 7.583733320236206
    },
    {
      "epoch": 6.112861633300781e-05,
      "model_forward_time": 1.7180671691894531,
      "step": 64098
    },
    {
      "epoch": 6.112861633300781e-05,
      "step": 64098,
      "training_step_time": 8.725781440734863
    },
    {
      "epoch": 6.112957000732422e-05,
      "model_forward_time": 1.7208960056304932,
      "step": 64099
    },
    {
      "epoch": 6.112957000732422e-05,
      "step": 64099,
      "training_step_time": 10.345089197158813
    },
    {
      "epoch": 6.113052368164062e-05,
      "grad_norm": 0.16270500421524048,
      "learning_rate": 3.1288239018332655e-06,
      "loss": 0.0601,
      "step": 64100
    },
    {
      "epoch": 6.113052368164062e-05,
      "model_forward_time": 1.7412471771240234,
      "step": 64100
    },
    {
      "dynamics_loss_avg": 0.06693160180002451,
      "epoch": 6.113052368164062e-05,
      "step": 64100
    },
    {
      "action_loss_avg": 0.0017487533914390951,
      "epoch": 6.113052368164062e-05,
      "step": 64100
    },
    {
      "epoch": 6.113052368164062e-05,
      "step": 64100,
      "training_step_time": 9.776379346847534
    },
    {
      "epoch": 6.113147735595703e-05,
      "model_forward_time": 1.724003553390503,
      "step": 64101
    },
    {
      "epoch": 6.113147735595703e-05,
      "step": 64101,
      "training_step_time": 7.628506898880005
    },
    {
      "epoch": 6.113243103027344e-05,
      "model_forward_time": 1.7640790939331055,
      "step": 64102
    },
    {
      "epoch": 6.113243103027344e-05,
      "step": 64102,
      "training_step_time": 7.611128091812134
    },
    {
      "epoch": 6.113338470458984e-05,
      "model_forward_time": 1.7508869171142578,
      "step": 64103
    },
    {
      "epoch": 6.113338470458984e-05,
      "step": 64103,
      "training_step_time": 7.553715467453003
    },
    {
      "epoch": 6.113433837890625e-05,
      "model_forward_time": 1.711057424545288,
      "step": 64104
    },
    {
      "epoch": 6.113433837890625e-05,
      "step": 64104,
      "training_step_time": 7.619439363479614
    },
    {
      "epoch": 6.113529205322266e-05,
      "model_forward_time": 1.7165780067443848,
      "step": 64105
    },
    {
      "epoch": 6.113529205322266e-05,
      "step": 64105,
      "training_step_time": 7.448955059051514
    },
    {
      "epoch": 6.113624572753906e-05,
      "model_forward_time": 2.3155245780944824,
      "step": 64106
    },
    {
      "epoch": 6.113624572753906e-05,
      "step": 64106,
      "training_step_time": 7.591033458709717
    },
    {
      "epoch": 6.113719940185547e-05,
      "model_forward_time": 1.7177793979644775,
      "step": 64107
    },
    {
      "epoch": 6.113719940185547e-05,
      "step": 64107,
      "training_step_time": 7.584293603897095
    },
    {
      "epoch": 6.113815307617188e-05,
      "model_forward_time": 1.7233061790466309,
      "step": 64108
    },
    {
      "epoch": 6.113815307617188e-05,
      "step": 64108,
      "training_step_time": 9.310750246047974
    },
    {
      "epoch": 6.113910675048828e-05,
      "model_forward_time": 1.719907283782959,
      "step": 64109
    },
    {
      "epoch": 6.113910675048828e-05,
      "step": 64109,
      "training_step_time": 8.691847085952759
    },
    {
      "epoch": 6.114006042480469e-05,
      "grad_norm": 0.10350865125656128,
      "learning_rate": 3.1272906857053164e-06,
      "loss": 0.055,
      "step": 64110
    },
    {
      "epoch": 6.114006042480469e-05,
      "model_forward_time": 1.732400894165039,
      "step": 64110
    },
    {
      "dynamics_loss_avg": 0.052803620137274264,
      "epoch": 6.114006042480469e-05,
      "step": 64110
    },
    {
      "action_loss_avg": 0.0016740972379921005,
      "epoch": 6.114006042480469e-05,
      "step": 64110
    },
    {
      "epoch": 6.114006042480469e-05,
      "step": 64110,
      "training_step_time": 8.753939867019653
    },
    {
      "epoch": 6.11410140991211e-05,
      "model_forward_time": 1.7181978225708008,
      "step": 64111
    },
    {
      "epoch": 6.11410140991211e-05,
      "step": 64111,
      "training_step_time": 8.507302522659302
    },
    {
      "epoch": 6.11419677734375e-05,
      "model_forward_time": 1.7256426811218262,
      "step": 64112
    },
    {
      "epoch": 6.11419677734375e-05,
      "step": 64112,
      "training_step_time": 7.61995267868042
    },
    {
      "epoch": 6.11429214477539e-05,
      "model_forward_time": 1.7386384010314941,
      "step": 64113
    },
    {
      "epoch": 6.11429214477539e-05,
      "step": 64113,
      "training_step_time": 7.614037752151489
    },
    {
      "epoch": 6.114387512207032e-05,
      "model_forward_time": 1.7394049167633057,
      "step": 64114
    },
    {
      "epoch": 6.114387512207032e-05,
      "step": 64114,
      "training_step_time": 7.386465072631836
    },
    {
      "epoch": 6.114482879638671e-05,
      "model_forward_time": 2.3189964294433594,
      "step": 64115
    },
    {
      "epoch": 6.114482879638671e-05,
      "step": 64115,
      "training_step_time": 7.568394184112549
    },
    {
      "epoch": 6.114578247070312e-05,
      "model_forward_time": 1.7744357585906982,
      "step": 64116
    },
    {
      "epoch": 6.114578247070312e-05,
      "step": 64116,
      "training_step_time": 9.333767652511597
    },
    {
      "epoch": 6.114673614501954e-05,
      "model_forward_time": 1.7424707412719727,
      "step": 64117
    },
    {
      "epoch": 6.114673614501954e-05,
      "step": 64117,
      "training_step_time": 8.799873113632202
    },
    {
      "epoch": 6.114768981933593e-05,
      "model_forward_time": 1.7398622035980225,
      "step": 64118
    },
    {
      "epoch": 6.114768981933593e-05,
      "step": 64118,
      "training_step_time": 8.852102518081665
    },
    {
      "epoch": 6.114864349365234e-05,
      "model_forward_time": 1.7632617950439453,
      "step": 64119
    },
    {
      "epoch": 6.114864349365234e-05,
      "step": 64119,
      "training_step_time": 8.69217038154602
    },
    {
      "epoch": 6.114959716796875e-05,
      "grad_norm": 0.11336576193571091,
      "learning_rate": 3.1257576743740427e-06,
      "loss": 0.0583,
      "step": 64120
    },
    {
      "epoch": 6.114959716796875e-05,
      "model_forward_time": 1.7357358932495117,
      "step": 64120
    },
    {
      "dynamics_loss_avg": 0.04188582915812731,
      "epoch": 6.114959716796875e-05,
      "step": 64120
    },
    {
      "action_loss_avg": 0.0010625577589962632,
      "epoch": 6.114959716796875e-05,
      "step": 64120
    },
    {
      "epoch": 6.114959716796875e-05,
      "step": 64120,
      "training_step_time": 7.653188228607178
    },
    {
      "epoch": 6.115055084228515e-05,
      "model_forward_time": 1.7445566654205322,
      "step": 64121
    },
    {
      "epoch": 6.115055084228515e-05,
      "step": 64121,
      "training_step_time": 7.599818468093872
    },
    {
      "epoch": 6.115150451660156e-05,
      "model_forward_time": 1.7371485233306885,
      "step": 64122
    },
    {
      "epoch": 6.115150451660156e-05,
      "step": 64122,
      "training_step_time": 7.582234144210815
    },
    {
      "epoch": 6.115245819091797e-05,
      "model_forward_time": 1.7417147159576416,
      "step": 64123
    },
    {
      "epoch": 6.115245819091797e-05,
      "step": 64123,
      "training_step_time": 7.007976293563843
    },
    {
      "epoch": 6.115341186523437e-05,
      "model_forward_time": 2.332524299621582,
      "step": 64124
    },
    {
      "epoch": 6.115341186523437e-05,
      "step": 64124,
      "training_step_time": 7.6126697063446045
    },
    {
      "epoch": 6.115436553955078e-05,
      "model_forward_time": 1.801142692565918,
      "step": 64125
    },
    {
      "epoch": 6.115436553955078e-05,
      "step": 64125,
      "training_step_time": 7.45232629776001
    },
    {
      "epoch": 6.115531921386719e-05,
      "model_forward_time": 1.736335277557373,
      "step": 64126
    },
    {
      "epoch": 6.115531921386719e-05,
      "step": 64126,
      "training_step_time": 9.873335361480713
    },
    {
      "epoch": 6.115627288818359e-05,
      "model_forward_time": 1.754729986190796,
      "step": 64127
    },
    {
      "epoch": 6.115627288818359e-05,
      "step": 64127,
      "training_step_time": 8.703436851501465
    },
    {
      "epoch": 6.11572265625e-05,
      "model_forward_time": 1.7431347370147705,
      "step": 64128
    },
    {
      "epoch": 6.11572265625e-05,
      "step": 64128,
      "training_step_time": 8.351155281066895
    },
    {
      "epoch": 6.115818023681641e-05,
      "model_forward_time": 1.7448718547821045,
      "step": 64129
    },
    {
      "epoch": 6.115818023681641e-05,
      "step": 64129,
      "training_step_time": 7.634012460708618
    },
    {
      "epoch": 6.115913391113281e-05,
      "grad_norm": 0.1120055541396141,
      "learning_rate": 3.1242248680070885e-06,
      "loss": 0.0573,
      "step": 64130
    },
    {
      "epoch": 6.115913391113281e-05,
      "model_forward_time": 1.7434444427490234,
      "step": 64130
    },
    {
      "dynamics_loss_avg": 0.04855060949921608,
      "epoch": 6.115913391113281e-05,
      "step": 64130
    },
    {
      "action_loss_avg": 0.0018504951789509506,
      "epoch": 6.115913391113281e-05,
      "step": 64130
    },
    {
      "epoch": 6.115913391113281e-05,
      "step": 64130,
      "training_step_time": 8.107200860977173
    },
    {
      "epoch": 6.116008758544922e-05,
      "model_forward_time": 1.743046760559082,
      "step": 64131
    },
    {
      "epoch": 6.116008758544922e-05,
      "step": 64131,
      "training_step_time": 6.761699199676514
    },
    {
      "epoch": 6.116104125976563e-05,
      "model_forward_time": 1.7468068599700928,
      "step": 64132
    },
    {
      "epoch": 6.116104125976563e-05,
      "step": 64132,
      "training_step_time": 7.653968095779419
    },
    {
      "epoch": 6.116199493408203e-05,
      "model_forward_time": 2.304574966430664,
      "step": 64133
    },
    {
      "epoch": 6.116199493408203e-05,
      "step": 64133,
      "training_step_time": 7.576494216918945
    },
    {
      "epoch": 6.116294860839844e-05,
      "model_forward_time": 1.7876641750335693,
      "step": 64134
    },
    {
      "epoch": 6.116294860839844e-05,
      "step": 64134,
      "training_step_time": 8.131086826324463
    },
    {
      "epoch": 6.116390228271485e-05,
      "model_forward_time": 1.7123570442199707,
      "step": 64135
    },
    {
      "epoch": 6.116390228271485e-05,
      "step": 64135,
      "training_step_time": 8.397037029266357
    },
    {
      "epoch": 6.116485595703125e-05,
      "model_forward_time": 1.7194581031799316,
      "step": 64136
    },
    {
      "epoch": 6.116485595703125e-05,
      "step": 64136,
      "training_step_time": 8.760677576065063
    },
    {
      "epoch": 6.116580963134766e-05,
      "model_forward_time": 1.7254581451416016,
      "step": 64137
    },
    {
      "epoch": 6.116580963134766e-05,
      "step": 64137,
      "training_step_time": 8.41753888130188
    },
    {
      "epoch": 6.116676330566407e-05,
      "model_forward_time": 1.7373590469360352,
      "step": 64138
    },
    {
      "epoch": 6.116676330566407e-05,
      "step": 64138,
      "training_step_time": 8.746484279632568
    },
    {
      "epoch": 6.116771697998046e-05,
      "model_forward_time": 1.7219595909118652,
      "step": 64139
    },
    {
      "epoch": 6.116771697998046e-05,
      "step": 64139,
      "training_step_time": 7.961374282836914
    },
    {
      "epoch": 6.116867065429688e-05,
      "grad_norm": 0.09777196496725082,
      "learning_rate": 3.1226922667720857e-06,
      "loss": 0.0572,
      "step": 64140
    },
    {
      "epoch": 6.116867065429688e-05,
      "model_forward_time": 1.7293896675109863,
      "step": 64140
    },
    {
      "dynamics_loss_avg": 0.049690735340118405,
      "epoch": 6.116867065429688e-05,
      "step": 64140
    },
    {
      "action_loss_avg": 0.001341982209123671,
      "epoch": 6.116867065429688e-05,
      "step": 64140
    },
    {
      "epoch": 6.116867065429688e-05,
      "step": 64140,
      "training_step_time": 7.042367219924927
    },
    {
      "epoch": 6.116962432861329e-05,
      "model_forward_time": 1.7248361110687256,
      "step": 64141
    },
    {
      "epoch": 6.116962432861329e-05,
      "step": 64141,
      "training_step_time": 7.674361944198608
    },
    {
      "epoch": 6.117057800292968e-05,
      "model_forward_time": 1.7201595306396484,
      "step": 64142
    },
    {
      "epoch": 6.117057800292968e-05,
      "step": 64142,
      "training_step_time": 7.548258543014526
    },
    {
      "epoch": 6.11715316772461e-05,
      "model_forward_time": 2.3306422233581543,
      "step": 64143
    },
    {
      "epoch": 6.11715316772461e-05,
      "step": 64143,
      "training_step_time": 8.191049575805664
    },
    {
      "epoch": 6.11724853515625e-05,
      "model_forward_time": 1.724076509475708,
      "step": 64144
    },
    {
      "epoch": 6.11724853515625e-05,
      "step": 64144,
      "training_step_time": 9.274613857269287
    },
    {
      "epoch": 6.11734390258789e-05,
      "model_forward_time": 1.7222986221313477,
      "step": 64145
    },
    {
      "epoch": 6.11734390258789e-05,
      "step": 64145,
      "training_step_time": 7.721822261810303
    },
    {
      "epoch": 6.117439270019531e-05,
      "model_forward_time": 1.7238013744354248,
      "step": 64146
    },
    {
      "epoch": 6.117439270019531e-05,
      "step": 64146,
      "training_step_time": 8.736713886260986
    },
    {
      "epoch": 6.117534637451172e-05,
      "model_forward_time": 1.7239339351654053,
      "step": 64147
    },
    {
      "epoch": 6.117534637451172e-05,
      "step": 64147,
      "training_step_time": 8.7057044506073
    },
    {
      "epoch": 6.117630004882812e-05,
      "model_forward_time": 1.7842693328857422,
      "step": 64148
    },
    {
      "epoch": 6.117630004882812e-05,
      "step": 64148,
      "training_step_time": 7.444104433059692
    },
    {
      "epoch": 6.117725372314453e-05,
      "model_forward_time": 1.8182084560394287,
      "step": 64149
    },
    {
      "epoch": 6.117725372314453e-05,
      "step": 64149,
      "training_step_time": 7.61181378364563
    },
    {
      "epoch": 6.117820739746094e-05,
      "grad_norm": 0.08968524634838104,
      "learning_rate": 3.1211598708366333e-06,
      "loss": 0.0569,
      "step": 64150
    },
    {
      "epoch": 6.117820739746094e-05,
      "model_forward_time": 1.8203999996185303,
      "step": 64150
    },
    {
      "dynamics_loss_avg": 0.05454027634114027,
      "epoch": 6.117820739746094e-05,
      "step": 64150
    },
    {
      "action_loss_avg": 0.0016455245640827343,
      "epoch": 6.117820739746094e-05,
      "step": 64150
    },
    {
      "epoch": 6.117820739746094e-05,
      "step": 64150,
      "training_step_time": 7.655095338821411
    },
    {
      "epoch": 6.117916107177734e-05,
      "model_forward_time": 1.8236408233642578,
      "step": 64151
    },
    {
      "epoch": 6.117916107177734e-05,
      "step": 64151,
      "training_step_time": 7.46122670173645
    },
    {
      "epoch": 6.118011474609375e-05,
      "model_forward_time": 2.3622190952301025,
      "step": 64152
    },
    {
      "epoch": 6.118011474609375e-05,
      "step": 64152,
      "training_step_time": 8.352489233016968
    },
    {
      "epoch": 6.118106842041016e-05,
      "model_forward_time": 1.8172035217285156,
      "step": 64153
    },
    {
      "epoch": 6.118106842041016e-05,
      "step": 64153,
      "training_step_time": 7.830953121185303
    },
    {
      "epoch": 6.118202209472656e-05,
      "model_forward_time": 1.818453311920166,
      "step": 64154
    },
    {
      "epoch": 6.118202209472656e-05,
      "step": 64154,
      "training_step_time": 7.780574798583984
    },
    {
      "epoch": 6.118297576904297e-05,
      "model_forward_time": 1.8193140029907227,
      "step": 64155
    },
    {
      "epoch": 6.118297576904297e-05,
      "step": 64155,
      "training_step_time": 7.752851724624634
    },
    {
      "epoch": 6.118392944335938e-05,
      "model_forward_time": 1.8250854015350342,
      "step": 64156
    },
    {
      "epoch": 6.118392944335938e-05,
      "step": 64156,
      "training_step_time": 7.64792013168335
    },
    {
      "epoch": 6.118488311767578e-05,
      "model_forward_time": 1.8246040344238281,
      "step": 64157
    },
    {
      "epoch": 6.118488311767578e-05,
      "step": 64157,
      "training_step_time": 8.72550892829895
    },
    {
      "epoch": 6.118583679199219e-05,
      "model_forward_time": 1.8361940383911133,
      "step": 64158
    },
    {
      "epoch": 6.118583679199219e-05,
      "step": 64158,
      "training_step_time": 7.380522727966309
    },
    {
      "epoch": 6.11867904663086e-05,
      "model_forward_time": 1.844794750213623,
      "step": 64159
    },
    {
      "epoch": 6.11867904663086e-05,
      "step": 64159,
      "training_step_time": 7.656290769577026
    },
    {
      "epoch": 6.1187744140625e-05,
      "grad_norm": 0.09463424235582352,
      "learning_rate": 3.119627680368312e-06,
      "loss": 0.0565,
      "step": 64160
    },
    {
      "epoch": 6.1187744140625e-05,
      "model_forward_time": 1.8372318744659424,
      "step": 64160
    },
    {
      "dynamics_loss_avg": 0.05761296190321445,
      "epoch": 6.1187744140625e-05,
      "step": 64160
    },
    {
      "action_loss_avg": 0.002117017941782251,
      "epoch": 6.1187744140625e-05,
      "step": 64160
    },
    {
      "epoch": 6.1187744140625e-05,
      "step": 64160,
      "training_step_time": 7.437302350997925
    },
    {
      "epoch": 6.118869781494141e-05,
      "model_forward_time": 2.376030445098877,
      "step": 64161
    },
    {
      "epoch": 6.118869781494141e-05,
      "step": 64161,
      "training_step_time": 7.623256206512451
    },
    {
      "epoch": 6.118965148925782e-05,
      "model_forward_time": 1.8377699851989746,
      "step": 64162
    },
    {
      "epoch": 6.118965148925782e-05,
      "step": 64162,
      "training_step_time": 7.749934434890747
    },
    {
      "epoch": 6.119060516357422e-05,
      "model_forward_time": 1.8380672931671143,
      "step": 64163
    },
    {
      "epoch": 6.119060516357422e-05,
      "step": 64163,
      "training_step_time": 8.254364967346191
    },
    {
      "epoch": 6.119155883789063e-05,
      "model_forward_time": 1.8383467197418213,
      "step": 64164
    },
    {
      "epoch": 6.119155883789063e-05,
      "step": 64164,
      "training_step_time": 9.453612804412842
    },
    {
      "epoch": 6.119251251220704e-05,
      "model_forward_time": 1.8392305374145508,
      "step": 64165
    },
    {
      "epoch": 6.119251251220704e-05,
      "step": 64165,
      "training_step_time": 10.727851390838623
    },
    {
      "epoch": 6.119346618652343e-05,
      "model_forward_time": 1.8435046672821045,
      "step": 64166
    },
    {
      "epoch": 6.119346618652343e-05,
      "step": 64166,
      "training_step_time": 9.370550870895386
    },
    {
      "epoch": 6.119441986083985e-05,
      "model_forward_time": 1.821084976196289,
      "step": 64167
    },
    {
      "epoch": 6.119441986083985e-05,
      "step": 64167,
      "training_step_time": 7.326170921325684
    },
    {
      "epoch": 6.119537353515626e-05,
      "model_forward_time": 1.8167736530303955,
      "step": 64168
    },
    {
      "epoch": 6.119537353515626e-05,
      "step": 64168,
      "training_step_time": 7.629108905792236
    },
    {
      "epoch": 6.119632720947265e-05,
      "model_forward_time": 1.8118155002593994,
      "step": 64169
    },
    {
      "epoch": 6.119632720947265e-05,
      "step": 64169,
      "training_step_time": 7.356456995010376
    },
    {
      "epoch": 6.119728088378906e-05,
      "grad_norm": 0.08445089310407639,
      "learning_rate": 3.1180956955346815e-06,
      "loss": 0.0588,
      "step": 64170
    },
    {
      "epoch": 6.119728088378906e-05,
      "model_forward_time": 2.3387739658355713,
      "step": 64170
    },
    {
      "dynamics_loss_avg": 0.060720869526267054,
      "epoch": 6.119728088378906e-05,
      "step": 64170
    },
    {
      "action_loss_avg": 0.0019126358907669782,
      "epoch": 6.119728088378906e-05,
      "step": 64170
    },
    {
      "epoch": 6.119728088378906e-05,
      "step": 64170,
      "training_step_time": 7.529982805252075
    },
    {
      "epoch": 6.119823455810547e-05,
      "model_forward_time": 1.8143727779388428,
      "step": 64171
    },
    {
      "epoch": 6.119823455810547e-05,
      "step": 64171,
      "training_step_time": 7.644408941268921
    },
    {
      "epoch": 6.119918823242187e-05,
      "model_forward_time": 1.8182487487792969,
      "step": 64172
    },
    {
      "epoch": 6.119918823242187e-05,
      "step": 64172,
      "training_step_time": 9.199493885040283
    },
    {
      "epoch": 6.120014190673828e-05,
      "model_forward_time": 1.818756103515625,
      "step": 64173
    },
    {
      "epoch": 6.120014190673828e-05,
      "step": 64173,
      "training_step_time": 7.670068025588989
    },
    {
      "epoch": 6.12010955810547e-05,
      "model_forward_time": 1.8200135231018066,
      "step": 64174
    },
    {
      "epoch": 6.12010955810547e-05,
      "step": 64174,
      "training_step_time": 7.665046215057373
    },
    {
      "epoch": 6.120204925537109e-05,
      "model_forward_time": 1.8149135112762451,
      "step": 64175
    },
    {
      "epoch": 6.120204925537109e-05,
      "step": 64175,
      "training_step_time": 7.590611696243286
    },
    {
      "epoch": 6.12030029296875e-05,
      "model_forward_time": 1.82228422164917,
      "step": 64176
    },
    {
      "epoch": 6.12030029296875e-05,
      "step": 64176,
      "training_step_time": 7.324733018875122
    },
    {
      "epoch": 6.120395660400391e-05,
      "model_forward_time": 1.8190927505493164,
      "step": 64177
    },
    {
      "epoch": 6.120395660400391e-05,
      "step": 64177,
      "training_step_time": 7.100879669189453
    },
    {
      "epoch": 6.120491027832031e-05,
      "model_forward_time": 1.8207590579986572,
      "step": 64178
    },
    {
      "epoch": 6.120491027832031e-05,
      "step": 64178,
      "training_step_time": 7.630042791366577
    },
    {
      "epoch": 6.120586395263672e-05,
      "model_forward_time": 2.373138904571533,
      "step": 64179
    },
    {
      "epoch": 6.120586395263672e-05,
      "step": 64179,
      "training_step_time": 8.934770107269287
    },
    {
      "epoch": 6.120681762695313e-05,
      "grad_norm": 0.2565470039844513,
      "learning_rate": 3.1165639165032764e-06,
      "loss": 0.0559,
      "step": 64180
    },
    {
      "epoch": 6.120681762695313e-05,
      "model_forward_time": 1.844179391860962,
      "step": 64180
    },
    {
      "dynamics_loss_avg": 0.06738704666495324,
      "epoch": 6.120681762695313e-05,
      "step": 64180
    },
    {
      "action_loss_avg": 0.0019043534528464078,
      "epoch": 6.120681762695313e-05,
      "step": 64180
    },
    {
      "epoch": 6.120681762695313e-05,
      "step": 64180,
      "training_step_time": 7.5045061111450195
    },
    {
      "epoch": 6.120777130126953e-05,
      "model_forward_time": 1.8227674961090088,
      "step": 64181
    },
    {
      "epoch": 6.120777130126953e-05,
      "step": 64181,
      "training_step_time": 9.31823444366455
    },
    {
      "epoch": 6.120872497558594e-05,
      "model_forward_time": 1.8186180591583252,
      "step": 64182
    },
    {
      "epoch": 6.120872497558594e-05,
      "step": 64182,
      "training_step_time": 8.38992714881897
    },
    {
      "epoch": 6.120967864990235e-05,
      "model_forward_time": 1.8335809707641602,
      "step": 64183
    },
    {
      "epoch": 6.120967864990235e-05,
      "step": 64183,
      "training_step_time": 8.407996892929077
    },
    {
      "epoch": 6.121063232421875e-05,
      "model_forward_time": 1.8394932746887207,
      "step": 64184
    },
    {
      "epoch": 6.121063232421875e-05,
      "step": 64184,
      "training_step_time": 7.833470344543457
    },
    {
      "epoch": 6.121158599853516e-05,
      "model_forward_time": 1.8362417221069336,
      "step": 64185
    },
    {
      "epoch": 6.121158599853516e-05,
      "step": 64185,
      "training_step_time": 8.00807523727417
    },
    {
      "epoch": 6.121253967285157e-05,
      "model_forward_time": 1.8489115238189697,
      "step": 64186
    },
    {
      "epoch": 6.121253967285157e-05,
      "step": 64186,
      "training_step_time": 7.237390756607056
    },
    {
      "epoch": 6.121349334716797e-05,
      "model_forward_time": 1.8400487899780273,
      "step": 64187
    },
    {
      "epoch": 6.121349334716797e-05,
      "step": 64187,
      "training_step_time": 7.762264013290405
    },
    {
      "epoch": 6.121444702148438e-05,
      "model_forward_time": 1.8386139869689941,
      "step": 64188
    },
    {
      "epoch": 6.121444702148438e-05,
      "step": 64188,
      "training_step_time": 7.979712724685669
    },
    {
      "epoch": 6.121540069580077e-05,
      "model_forward_time": 1.819345474243164,
      "step": 64189
    },
    {
      "epoch": 6.121540069580077e-05,
      "step": 64189,
      "training_step_time": 7.647104024887085
    },
    {
      "epoch": 6.121635437011719e-05,
      "grad_norm": 0.1334211230278015,
      "learning_rate": 3.1150323434416095e-06,
      "loss": 0.0561,
      "step": 64190
    },
    {
      "epoch": 6.121635437011719e-05,
      "model_forward_time": 1.4700286388397217,
      "step": 64190
    },
    {
      "dynamics_loss_avg": 0.045197596028447154,
      "epoch": 6.121635437011719e-05,
      "step": 64190
    },
    {
      "action_loss_avg": 0.0011375984351616353,
      "epoch": 6.121635437011719e-05,
      "step": 64190
    },
    {
      "epoch": 6.121635437011719e-05,
      "step": 64190,
      "training_step_time": 8.77784776687622
    },
    {
      "epoch": 6.12173080444336e-05,
      "model_forward_time": 1.8356175422668457,
      "step": 64191
    },
    {
      "epoch": 6.12173080444336e-05,
      "step": 64191,
      "training_step_time": 8.343443155288696
    },
    {
      "epoch": 6.121826171875e-05,
      "model_forward_time": 1.835050344467163,
      "step": 64192
    },
    {
      "epoch": 6.121826171875e-05,
      "step": 64192,
      "training_step_time": 7.759228467941284
    },
    {
      "epoch": 6.12192153930664e-05,
      "model_forward_time": 1.8424320220947266,
      "step": 64193
    },
    {
      "epoch": 6.12192153930664e-05,
      "step": 64193,
      "training_step_time": 8.72376799583435
    },
    {
      "epoch": 6.122016906738281e-05,
      "model_forward_time": 1.8439860343933105,
      "step": 64194
    },
    {
      "epoch": 6.122016906738281e-05,
      "step": 64194,
      "training_step_time": 7.636653661727905
    },
    {
      "epoch": 6.122112274169921e-05,
      "model_forward_time": 1.8383545875549316,
      "step": 64195
    },
    {
      "epoch": 6.122112274169921e-05,
      "step": 64195,
      "training_step_time": 7.391850471496582
    },
    {
      "epoch": 6.122207641601562e-05,
      "model_forward_time": 1.8428823947906494,
      "step": 64196
    },
    {
      "epoch": 6.122207641601562e-05,
      "step": 64196,
      "training_step_time": 7.702072381973267
    },
    {
      "epoch": 6.122303009033203e-05,
      "model_forward_time": 1.7968823909759521,
      "step": 64197
    },
    {
      "epoch": 6.122303009033203e-05,
      "step": 64197,
      "training_step_time": 7.419535160064697
    },
    {
      "epoch": 6.122398376464843e-05,
      "model_forward_time": 2.376636266708374,
      "step": 64198
    },
    {
      "epoch": 6.122398376464843e-05,
      "step": 64198,
      "training_step_time": 7.643099069595337
    },
    {
      "epoch": 6.122493743896484e-05,
      "model_forward_time": 1.8230490684509277,
      "step": 64199
    },
    {
      "epoch": 6.122493743896484e-05,
      "step": 64199,
      "training_step_time": 7.879151821136475
    },
    {
      "epoch": 6.122589111328125e-05,
      "grad_norm": 0.0945221558213234,
      "learning_rate": 3.1135009765171696e-06,
      "loss": 0.0582,
      "step": 64200
    },
    {
      "epoch": 6.122589111328125e-05,
      "model_forward_time": 1.8119168281555176,
      "step": 64200
    },
    {
      "dynamics_loss_avg": 0.06293901838362218,
      "epoch": 6.122589111328125e-05,
      "step": 64200
    },
    {
      "action_loss_avg": 0.0014554212626535445,
      "epoch": 6.122589111328125e-05,
      "step": 64200
    },
    {
      "epoch": 6.122589111328125e-05,
      "step": 64200,
      "training_step_time": 7.696339845657349
    },
    {
      "epoch": 6.122684478759765e-05,
      "model_forward_time": 1.8126723766326904,
      "step": 64201
    },
    {
      "epoch": 6.122684478759765e-05,
      "step": 64201,
      "training_step_time": 9.229630470275879
    },
    {
      "epoch": 6.122779846191406e-05,
      "model_forward_time": 1.8147807121276855,
      "step": 64202
    },
    {
      "epoch": 6.122779846191406e-05,
      "step": 64202,
      "training_step_time": 7.685395240783691
    },
    {
      "epoch": 6.122875213623047e-05,
      "model_forward_time": 1.813901662826538,
      "step": 64203
    },
    {
      "epoch": 6.122875213623047e-05,
      "step": 64203,
      "training_step_time": 8.651067733764648
    },
    {
      "epoch": 6.122970581054687e-05,
      "model_forward_time": 1.8173820972442627,
      "step": 64204
    },
    {
      "epoch": 6.122970581054687e-05,
      "step": 64204,
      "training_step_time": 7.5530846118927
    },
    {
      "epoch": 6.123065948486328e-05,
      "model_forward_time": 1.8139386177062988,
      "step": 64205
    },
    {
      "epoch": 6.123065948486328e-05,
      "step": 64205,
      "training_step_time": 7.6664276123046875
    },
    {
      "epoch": 6.123161315917969e-05,
      "model_forward_time": 1.8128783702850342,
      "step": 64206
    },
    {
      "epoch": 6.123161315917969e-05,
      "step": 64206,
      "training_step_time": 7.404394626617432
    },
    {
      "epoch": 6.123256683349609e-05,
      "model_forward_time": 2.393226146697998,
      "step": 64207
    },
    {
      "epoch": 6.123256683349609e-05,
      "step": 64207,
      "training_step_time": 7.61424446105957
    },
    {
      "epoch": 6.12335205078125e-05,
      "model_forward_time": 1.8168001174926758,
      "step": 64208
    },
    {
      "epoch": 6.12335205078125e-05,
      "step": 64208,
      "training_step_time": 7.715378522872925
    },
    {
      "epoch": 6.123447418212891e-05,
      "model_forward_time": 1.8167424201965332,
      "step": 64209
    },
    {
      "epoch": 6.123447418212891e-05,
      "step": 64209,
      "training_step_time": 8.272111415863037
    },
    {
      "epoch": 6.12354278564453e-05,
      "grad_norm": 0.07634785771369934,
      "learning_rate": 3.1119698158974288e-06,
      "loss": 0.0562,
      "step": 64210
    },
    {
      "epoch": 6.12354278564453e-05,
      "model_forward_time": 1.8249809741973877,
      "step": 64210
    },
    {
      "dynamics_loss_avg": 0.05949138104915619,
      "epoch": 6.12354278564453e-05,
      "step": 64210
    },
    {
      "action_loss_avg": 0.0017565297312103212,
      "epoch": 6.12354278564453e-05,
      "step": 64210
    },
    {
      "epoch": 6.12354278564453e-05,
      "step": 64210,
      "training_step_time": 7.70221209526062
    },
    {
      "epoch": 6.123638153076172e-05,
      "model_forward_time": 1.8174147605895996,
      "step": 64211
    },
    {
      "epoch": 6.123638153076172e-05,
      "step": 64211,
      "training_step_time": 8.510493040084839
    },
    {
      "epoch": 6.123733520507813e-05,
      "model_forward_time": 1.8253867626190186,
      "step": 64212
    },
    {
      "epoch": 6.123733520507813e-05,
      "step": 64212,
      "training_step_time": 7.697645425796509
    },
    {
      "epoch": 6.123828887939453e-05,
      "model_forward_time": 1.810244083404541,
      "step": 64213
    },
    {
      "epoch": 6.123828887939453e-05,
      "step": 64213,
      "training_step_time": 7.546447992324829
    },
    {
      "epoch": 6.123924255371094e-05,
      "model_forward_time": 1.811635971069336,
      "step": 64214
    },
    {
      "epoch": 6.123924255371094e-05,
      "step": 64214,
      "training_step_time": 7.668998718261719
    },
    {
      "epoch": 6.124019622802735e-05,
      "model_forward_time": 1.81337571144104,
      "step": 64215
    },
    {
      "epoch": 6.124019622802735e-05,
      "step": 64215,
      "training_step_time": 7.35411810874939
    },
    {
      "epoch": 6.124114990234374e-05,
      "model_forward_time": 2.3608663082122803,
      "step": 64216
    },
    {
      "epoch": 6.124114990234374e-05,
      "step": 64216,
      "training_step_time": 7.602796316146851
    },
    {
      "epoch": 6.124210357666016e-05,
      "model_forward_time": 1.8127713203430176,
      "step": 64217
    },
    {
      "epoch": 6.124210357666016e-05,
      "step": 64217,
      "training_step_time": 7.648521900177002
    },
    {
      "epoch": 6.124305725097657e-05,
      "model_forward_time": 1.8016154766082764,
      "step": 64218
    },
    {
      "epoch": 6.124305725097657e-05,
      "step": 64218,
      "training_step_time": 9.082829475402832
    },
    {
      "epoch": 6.124401092529296e-05,
      "model_forward_time": 1.8101420402526855,
      "step": 64219
    },
    {
      "epoch": 6.124401092529296e-05,
      "step": 64219,
      "training_step_time": 7.649950265884399
    },
    {
      "epoch": 6.124496459960937e-05,
      "grad_norm": 0.12011043727397919,
      "learning_rate": 3.1104388617498297e-06,
      "loss": 0.0575,
      "step": 64220
    },
    {
      "epoch": 6.124496459960937e-05,
      "model_forward_time": 1.8143105506896973,
      "step": 64220
    },
    {
      "dynamics_loss_avg": 0.057245110906660554,
      "epoch": 6.124496459960937e-05,
      "step": 64220
    },
    {
      "action_loss_avg": 0.0018464387860149146,
      "epoch": 6.124496459960937e-05,
      "step": 64220
    },
    {
      "epoch": 6.124496459960937e-05,
      "step": 64220,
      "training_step_time": 9.116098165512085
    },
    {
      "epoch": 6.124591827392578e-05,
      "model_forward_time": 1.8138785362243652,
      "step": 64221
    },
    {
      "epoch": 6.124591827392578e-05,
      "step": 64221,
      "training_step_time": 7.710437059402466
    },
    {
      "epoch": 6.124687194824218e-05,
      "model_forward_time": 1.8154792785644531,
      "step": 64222
    },
    {
      "epoch": 6.124687194824218e-05,
      "step": 64222,
      "training_step_time": 7.660061836242676
    },
    {
      "epoch": 6.124782562255859e-05,
      "model_forward_time": 1.8158717155456543,
      "step": 64223
    },
    {
      "epoch": 6.124782562255859e-05,
      "step": 64223,
      "training_step_time": 7.569495916366577
    },
    {
      "epoch": 6.1248779296875e-05,
      "model_forward_time": 1.8140575885772705,
      "step": 64224
    },
    {
      "epoch": 6.1248779296875e-05,
      "step": 64224,
      "training_step_time": 7.624788522720337
    },
    {
      "epoch": 6.12497329711914e-05,
      "model_forward_time": 2.355009078979492,
      "step": 64225
    },
    {
      "epoch": 6.12497329711914e-05,
      "step": 64225,
      "training_step_time": 7.571467399597168
    },
    {
      "epoch": 6.125068664550781e-05,
      "model_forward_time": 1.8129339218139648,
      "step": 64226
    },
    {
      "epoch": 6.125068664550781e-05,
      "step": 64226,
      "training_step_time": 8.27235722541809
    },
    {
      "epoch": 6.125164031982422e-05,
      "model_forward_time": 1.81821608543396,
      "step": 64227
    },
    {
      "epoch": 6.125164031982422e-05,
      "step": 64227,
      "training_step_time": 8.704094171524048
    },
    {
      "epoch": 6.125259399414062e-05,
      "model_forward_time": 1.814121961593628,
      "step": 64228
    },
    {
      "epoch": 6.125259399414062e-05,
      "step": 64228,
      "training_step_time": 8.694836854934692
    },
    {
      "epoch": 6.125354766845703e-05,
      "model_forward_time": 1.8202736377716064,
      "step": 64229
    },
    {
      "epoch": 6.125354766845703e-05,
      "step": 64229,
      "training_step_time": 7.728490829467773
    },
    {
      "epoch": 6.125450134277344e-05,
      "grad_norm": 0.09482557326555252,
      "learning_rate": 3.1089081142417966e-06,
      "loss": 0.0597,
      "step": 64230
    },
    {
      "epoch": 6.125450134277344e-05,
      "model_forward_time": 1.8123779296875,
      "step": 64230
    },
    {
      "dynamics_loss_avg": 0.05036720484495163,
      "epoch": 6.125450134277344e-05,
      "step": 64230
    },
    {
      "action_loss_avg": 0.0015156763140112162,
      "epoch": 6.125450134277344e-05,
      "step": 64230
    },
    {
      "epoch": 6.125450134277344e-05,
      "step": 64230,
      "training_step_time": 9.742844820022583
    },
    {
      "epoch": 6.125545501708984e-05,
      "model_forward_time": 1.803633689880371,
      "step": 64231
    },
    {
      "epoch": 6.125545501708984e-05,
      "step": 64231,
      "training_step_time": 8.653846025466919
    },
    {
      "epoch": 6.125640869140625e-05,
      "model_forward_time": 1.8046340942382812,
      "step": 64232
    },
    {
      "epoch": 6.125640869140625e-05,
      "step": 64232,
      "training_step_time": 7.558436632156372
    },
    {
      "epoch": 6.125736236572266e-05,
      "model_forward_time": 1.8037855625152588,
      "step": 64233
    },
    {
      "epoch": 6.125736236572266e-05,
      "step": 64233,
      "training_step_time": 7.737157106399536
    },
    {
      "epoch": 6.125831604003906e-05,
      "model_forward_time": 1.8047428131103516,
      "step": 64234
    },
    {
      "epoch": 6.125831604003906e-05,
      "step": 64234,
      "training_step_time": 7.517025947570801
    },
    {
      "epoch": 6.125926971435547e-05,
      "model_forward_time": 2.3598697185516357,
      "step": 64235
    },
    {
      "epoch": 6.125926971435547e-05,
      "step": 64235,
      "training_step_time": 7.601457834243774
    },
    {
      "epoch": 6.126022338867188e-05,
      "model_forward_time": 1.8081519603729248,
      "step": 64236
    },
    {
      "epoch": 6.126022338867188e-05,
      "step": 64236,
      "training_step_time": 8.213911533355713
    },
    {
      "epoch": 6.126117706298828e-05,
      "model_forward_time": 1.794581413269043,
      "step": 64237
    },
    {
      "epoch": 6.126117706298828e-05,
      "step": 64237,
      "training_step_time": 7.651289463043213
    },
    {
      "epoch": 6.126213073730469e-05,
      "model_forward_time": 1.8101074695587158,
      "step": 64238
    },
    {
      "epoch": 6.126213073730469e-05,
      "step": 64238,
      "training_step_time": 9.30965518951416
    },
    {
      "epoch": 6.12630844116211e-05,
      "model_forward_time": 1.810356855392456,
      "step": 64239
    },
    {
      "epoch": 6.12630844116211e-05,
      "step": 64239,
      "training_step_time": 7.673434734344482
    },
    {
      "epoch": 6.12640380859375e-05,
      "grad_norm": 0.11659453064203262,
      "learning_rate": 3.1073775735407284e-06,
      "loss": 0.0583,
      "step": 64240
    },
    {
      "epoch": 6.12640380859375e-05,
      "model_forward_time": 1.8170969486236572,
      "step": 64240
    },
    {
      "dynamics_loss_avg": 0.0625486621633172,
      "epoch": 6.12640380859375e-05,
      "step": 64240
    },
    {
      "action_loss_avg": 0.0011118190450360999,
      "epoch": 6.12640380859375e-05,
      "step": 64240
    },
    {
      "epoch": 6.12640380859375e-05,
      "step": 64240,
      "training_step_time": 7.593708753585815
    },
    {
      "epoch": 6.12649917602539e-05,
      "model_forward_time": 1.8132245540618896,
      "step": 64241
    },
    {
      "epoch": 6.12649917602539e-05,
      "step": 64241,
      "training_step_time": 7.455634117126465
    },
    {
      "epoch": 6.126594543457032e-05,
      "model_forward_time": 1.8201816082000732,
      "step": 64242
    },
    {
      "epoch": 6.126594543457032e-05,
      "step": 64242,
      "training_step_time": 8.17792558670044
    },
    {
      "epoch": 6.126689910888671e-05,
      "model_forward_time": 1.7995030879974365,
      "step": 64243
    },
    {
      "epoch": 6.126689910888671e-05,
      "step": 64243,
      "training_step_time": 7.4722466468811035
    },
    {
      "epoch": 6.126785278320312e-05,
      "model_forward_time": 2.35265851020813,
      "step": 64244
    },
    {
      "epoch": 6.126785278320312e-05,
      "step": 64244,
      "training_step_time": 7.645796537399292
    },
    {
      "epoch": 6.126880645751954e-05,
      "model_forward_time": 1.8158457279205322,
      "step": 64245
    },
    {
      "epoch": 6.126880645751954e-05,
      "step": 64245,
      "training_step_time": 8.770190954208374
    },
    {
      "epoch": 6.126976013183593e-05,
      "model_forward_time": 1.8148236274719238,
      "step": 64246
    },
    {
      "epoch": 6.126976013183593e-05,
      "step": 64246,
      "training_step_time": 8.295470952987671
    },
    {
      "epoch": 6.127071380615234e-05,
      "model_forward_time": 1.817967176437378,
      "step": 64247
    },
    {
      "epoch": 6.127071380615234e-05,
      "step": 64247,
      "training_step_time": 7.705063581466675
    },
    {
      "epoch": 6.127166748046875e-05,
      "model_forward_time": 1.8130266666412354,
      "step": 64248
    },
    {
      "epoch": 6.127166748046875e-05,
      "step": 64248,
      "training_step_time": 8.742923498153687
    },
    {
      "epoch": 6.127262115478515e-05,
      "model_forward_time": 1.8192822933197021,
      "step": 64249
    },
    {
      "epoch": 6.127262115478515e-05,
      "step": 64249,
      "training_step_time": 8.831305027008057
    },
    {
      "epoch": 6.127357482910156e-05,
      "grad_norm": 0.07982555776834488,
      "learning_rate": 3.1058472398140032e-06,
      "loss": 0.0576,
      "step": 64250
    },
    {
      "epoch": 6.127357482910156e-05,
      "model_forward_time": 1.8134989738464355,
      "step": 64250
    },
    {
      "dynamics_loss_avg": 0.05519700068980456,
      "epoch": 6.127357482910156e-05,
      "step": 64250
    },
    {
      "action_loss_avg": 0.0017233581806067378,
      "epoch": 6.127357482910156e-05,
      "step": 64250
    },
    {
      "epoch": 6.127357482910156e-05,
      "step": 64250,
      "training_step_time": 7.699223279953003
    },
    {
      "epoch": 6.127452850341797e-05,
      "model_forward_time": 1.7908568382263184,
      "step": 64251
    },
    {
      "epoch": 6.127452850341797e-05,
      "step": 64251,
      "training_step_time": 7.6743199825286865
    },
    {
      "epoch": 6.127548217773437e-05,
      "model_forward_time": 1.80680251121521,
      "step": 64252
    },
    {
      "epoch": 6.127548217773437e-05,
      "step": 64252,
      "training_step_time": 7.509170055389404
    },
    {
      "epoch": 6.127643585205078e-05,
      "model_forward_time": 2.3289806842803955,
      "step": 64253
    },
    {
      "epoch": 6.127643585205078e-05,
      "step": 64253,
      "training_step_time": 7.652129650115967
    },
    {
      "epoch": 6.127738952636719e-05,
      "model_forward_time": 1.8077092170715332,
      "step": 64254
    },
    {
      "epoch": 6.127738952636719e-05,
      "step": 64254,
      "training_step_time": 7.691805362701416
    },
    {
      "epoch": 6.127834320068359e-05,
      "model_forward_time": 1.8107774257659912,
      "step": 64255
    },
    {
      "epoch": 6.127834320068359e-05,
      "step": 64255,
      "training_step_time": 8.216255187988281
    },
    {
      "epoch": 6.1279296875e-05,
      "model_forward_time": 1.8125085830688477,
      "step": 64256
    },
    {
      "epoch": 6.1279296875e-05,
      "step": 64256,
      "training_step_time": 8.987653970718384
    },
    {
      "epoch": 6.128025054931641e-05,
      "model_forward_time": 1.8120136260986328,
      "step": 64257
    },
    {
      "epoch": 6.128025054931641e-05,
      "step": 64257,
      "training_step_time": 7.702388286590576
    },
    {
      "epoch": 6.128120422363281e-05,
      "model_forward_time": 1.8113152980804443,
      "step": 64258
    },
    {
      "epoch": 6.128120422363281e-05,
      "step": 64258,
      "training_step_time": 7.744766712188721
    },
    {
      "epoch": 6.128215789794922e-05,
      "model_forward_time": 1.8080296516418457,
      "step": 64259
    },
    {
      "epoch": 6.128215789794922e-05,
      "step": 64259,
      "training_step_time": 8.657420873641968
    },
    {
      "epoch": 6.128311157226563e-05,
      "grad_norm": 0.15017923712730408,
      "learning_rate": 3.104317113228974e-06,
      "loss": 0.0592,
      "step": 64260
    },
    {
      "epoch": 6.128311157226563e-05,
      "model_forward_time": 1.79083251953125,
      "step": 64260
    },
    {
      "dynamics_loss_avg": 0.04870903864502907,
      "epoch": 6.128311157226563e-05,
      "step": 64260
    },
    {
      "action_loss_avg": 0.0009522199805360287,
      "epoch": 6.128311157226563e-05,
      "step": 64260
    },
    {
      "epoch": 6.128311157226563e-05,
      "step": 64260,
      "training_step_time": 8.577175378799438
    },
    {
      "epoch": 6.128406524658203e-05,
      "model_forward_time": 1.8078687191009521,
      "step": 64261
    },
    {
      "epoch": 6.128406524658203e-05,
      "step": 64261,
      "training_step_time": 7.366117238998413
    },
    {
      "epoch": 6.128501892089844e-05,
      "model_forward_time": 2.33917236328125,
      "step": 64262
    },
    {
      "epoch": 6.128501892089844e-05,
      "step": 64262,
      "training_step_time": 8.427324295043945
    },
    {
      "epoch": 6.128597259521485e-05,
      "model_forward_time": 1.7971856594085693,
      "step": 64263
    },
    {
      "epoch": 6.128597259521485e-05,
      "step": 64263,
      "training_step_time": 8.316664218902588
    },
    {
      "epoch": 6.128692626953125e-05,
      "model_forward_time": 1.817894697189331,
      "step": 64264
    },
    {
      "epoch": 6.128692626953125e-05,
      "step": 64264,
      "training_step_time": 8.675499677658081
    },
    {
      "epoch": 6.128787994384766e-05,
      "model_forward_time": 1.8110830783843994,
      "step": 64265
    },
    {
      "epoch": 6.128787994384766e-05,
      "step": 64265,
      "training_step_time": 8.893372058868408
    },
    {
      "epoch": 6.128883361816407e-05,
      "model_forward_time": 1.8131091594696045,
      "step": 64266
    },
    {
      "epoch": 6.128883361816407e-05,
      "step": 64266,
      "training_step_time": 8.413335800170898
    },
    {
      "epoch": 6.128978729248046e-05,
      "model_forward_time": 1.817701816558838,
      "step": 64267
    },
    {
      "epoch": 6.128978729248046e-05,
      "step": 64267,
      "training_step_time": 8.739513874053955
    },
    {
      "epoch": 6.129074096679688e-05,
      "model_forward_time": 1.8210301399230957,
      "step": 64268
    },
    {
      "epoch": 6.129074096679688e-05,
      "step": 64268,
      "training_step_time": 7.632397890090942
    },
    {
      "epoch": 6.129169464111329e-05,
      "model_forward_time": 1.8152422904968262,
      "step": 64269
    },
    {
      "epoch": 6.129169464111329e-05,
      "step": 64269,
      "training_step_time": 7.5568201541900635
    },
    {
      "epoch": 6.129264831542968e-05,
      "grad_norm": 0.11837500333786011,
      "learning_rate": 3.1027871939529775e-06,
      "loss": 0.0576,
      "step": 64270
    },
    {
      "epoch": 6.129264831542968e-05,
      "model_forward_time": 1.8142879009246826,
      "step": 64270
    },
    {
      "dynamics_loss_avg": 0.05776461400091648,
      "epoch": 6.129264831542968e-05,
      "step": 64270
    },
    {
      "action_loss_avg": 0.0010515405127080157,
      "epoch": 6.129264831542968e-05,
      "step": 64270
    },
    {
      "epoch": 6.129264831542968e-05,
      "step": 64270,
      "training_step_time": 7.570516586303711
    },
    {
      "epoch": 6.12936019897461e-05,
      "model_forward_time": 2.3590900897979736,
      "step": 64271
    },
    {
      "epoch": 6.12936019897461e-05,
      "step": 64271,
      "training_step_time": 8.142324209213257
    },
    {
      "epoch": 6.12945556640625e-05,
      "model_forward_time": 1.8134584426879883,
      "step": 64272
    },
    {
      "epoch": 6.12945556640625e-05,
      "step": 64272,
      "training_step_time": 8.665264368057251
    },
    {
      "epoch": 6.12955093383789e-05,
      "model_forward_time": 1.8143348693847656,
      "step": 64273
    },
    {
      "epoch": 6.12955093383789e-05,
      "step": 64273,
      "training_step_time": 9.228680849075317
    },
    {
      "epoch": 6.129646301269531e-05,
      "model_forward_time": 1.816455602645874,
      "step": 64274
    },
    {
      "epoch": 6.129646301269531e-05,
      "step": 64274,
      "training_step_time": 8.201856851577759
    },
    {
      "epoch": 6.129741668701172e-05,
      "model_forward_time": 1.7982189655303955,
      "step": 64275
    },
    {
      "epoch": 6.129741668701172e-05,
      "step": 64275,
      "training_step_time": 9.507802486419678
    },
    {
      "epoch": 6.129837036132812e-05,
      "model_forward_time": 1.8168506622314453,
      "step": 64276
    },
    {
      "epoch": 6.129837036132812e-05,
      "step": 64276,
      "training_step_time": 8.558281898498535
    },
    {
      "epoch": 6.129932403564453e-05,
      "model_forward_time": 1.8219356536865234,
      "step": 64277
    },
    {
      "epoch": 6.129932403564453e-05,
      "step": 64277,
      "training_step_time": 7.619171857833862
    },
    {
      "epoch": 6.130027770996094e-05,
      "model_forward_time": 1.8188114166259766,
      "step": 64278
    },
    {
      "epoch": 6.130027770996094e-05,
      "step": 64278,
      "training_step_time": 7.516763687133789
    },
    {
      "epoch": 6.130123138427734e-05,
      "model_forward_time": 1.8208537101745605,
      "step": 64279
    },
    {
      "epoch": 6.130123138427734e-05,
      "step": 64279,
      "training_step_time": 7.665123701095581
    },
    {
      "epoch": 6.130218505859375e-05,
      "grad_norm": 0.13033482432365417,
      "learning_rate": 3.1012574821533214e-06,
      "loss": 0.0605,
      "step": 64280
    },
    {
      "epoch": 6.130218505859375e-05,
      "model_forward_time": 1.8113279342651367,
      "step": 64280
    },
    {
      "dynamics_loss_avg": 0.061544974148273465,
      "epoch": 6.130218505859375e-05,
      "step": 64280
    },
    {
      "action_loss_avg": 0.002129737625364214,
      "epoch": 6.130218505859375e-05,
      "step": 64280
    },
    {
      "epoch": 6.130218505859375e-05,
      "step": 64280,
      "training_step_time": 8.015308380126953
    },
    {
      "epoch": 6.130313873291016e-05,
      "model_forward_time": 1.8196039199829102,
      "step": 64281
    },
    {
      "epoch": 6.130313873291016e-05,
      "step": 64281,
      "training_step_time": 7.6558144092559814
    },
    {
      "epoch": 6.130409240722656e-05,
      "model_forward_time": 1.8225457668304443,
      "step": 64282
    },
    {
      "epoch": 6.130409240722656e-05,
      "step": 64282,
      "training_step_time": 7.664110898971558
    },
    {
      "epoch": 6.130504608154297e-05,
      "model_forward_time": 1.8225083351135254,
      "step": 64283
    },
    {
      "epoch": 6.130504608154297e-05,
      "step": 64283,
      "training_step_time": 10.005707263946533
    },
    {
      "epoch": 6.130599975585938e-05,
      "model_forward_time": 1.8156366348266602,
      "step": 64284
    },
    {
      "epoch": 6.130599975585938e-05,
      "step": 64284,
      "training_step_time": 8.288355588912964
    },
    {
      "epoch": 6.130695343017578e-05,
      "model_forward_time": 1.8246958255767822,
      "step": 64285
    },
    {
      "epoch": 6.130695343017578e-05,
      "step": 64285,
      "training_step_time": 8.65985918045044
    },
    {
      "epoch": 6.130790710449219e-05,
      "model_forward_time": 1.816354751586914,
      "step": 64286
    },
    {
      "epoch": 6.130790710449219e-05,
      "step": 64286,
      "training_step_time": 7.633068561553955
    },
    {
      "epoch": 6.13088607788086e-05,
      "model_forward_time": 1.8204329013824463,
      "step": 64287
    },
    {
      "epoch": 6.13088607788086e-05,
      "step": 64287,
      "training_step_time": 7.592745780944824
    },
    {
      "epoch": 6.1309814453125e-05,
      "model_forward_time": 1.8179020881652832,
      "step": 64288
    },
    {
      "epoch": 6.1309814453125e-05,
      "step": 64288,
      "training_step_time": 7.612565517425537
    },
    {
      "epoch": 6.131076812744141e-05,
      "model_forward_time": 1.817798137664795,
      "step": 64289
    },
    {
      "epoch": 6.131076812744141e-05,
      "step": 64289,
      "training_step_time": 7.4259374141693115
    },
    {
      "epoch": 6.131172180175782e-05,
      "grad_norm": 0.07844699174165726,
      "learning_rate": 3.099727977997292e-06,
      "loss": 0.0536,
      "step": 64290
    },
    {
      "epoch": 6.131172180175782e-05,
      "model_forward_time": 2.3725805282592773,
      "step": 64290
    },
    {
      "dynamics_loss_avg": 0.052362740598618986,
      "epoch": 6.131172180175782e-05,
      "step": 64290
    },
    {
      "action_loss_avg": 0.0017695861955871806,
      "epoch": 6.131172180175782e-05,
      "step": 64290
    },
    {
      "epoch": 6.131172180175782e-05,
      "step": 64290,
      "training_step_time": 7.646064043045044
    },
    {
      "epoch": 6.131267547607422e-05,
      "model_forward_time": 1.8201501369476318,
      "step": 64291
    },
    {
      "epoch": 6.131267547607422e-05,
      "step": 64291,
      "training_step_time": 8.239443063735962
    },
    {
      "epoch": 6.131362915039063e-05,
      "model_forward_time": 1.8242371082305908,
      "step": 64292
    },
    {
      "epoch": 6.131362915039063e-05,
      "step": 64292,
      "training_step_time": 8.55137586593628
    },
    {
      "epoch": 6.131458282470704e-05,
      "model_forward_time": 1.8160743713378906,
      "step": 64293
    },
    {
      "epoch": 6.131458282470704e-05,
      "step": 64293,
      "training_step_time": 7.738195419311523
    },
    {
      "epoch": 6.131553649902343e-05,
      "model_forward_time": 1.8161654472351074,
      "step": 64294
    },
    {
      "epoch": 6.131553649902343e-05,
      "step": 64294,
      "training_step_time": 7.628377437591553
    },
    {
      "epoch": 6.131649017333985e-05,
      "model_forward_time": 1.8214647769927979,
      "step": 64295
    },
    {
      "epoch": 6.131649017333985e-05,
      "step": 64295,
      "training_step_time": 7.682823657989502
    },
    {
      "epoch": 6.131744384765626e-05,
      "model_forward_time": 1.8122193813323975,
      "step": 64296
    },
    {
      "epoch": 6.131744384765626e-05,
      "step": 64296,
      "training_step_time": 7.640767574310303
    },
    {
      "epoch": 6.131839752197265e-05,
      "model_forward_time": 1.8149499893188477,
      "step": 64297
    },
    {
      "epoch": 6.131839752197265e-05,
      "step": 64297,
      "training_step_time": 7.631027460098267
    },
    {
      "epoch": 6.131935119628906e-05,
      "model_forward_time": 1.794612169265747,
      "step": 64298
    },
    {
      "epoch": 6.131935119628906e-05,
      "step": 64298,
      "training_step_time": 7.517751932144165
    },
    {
      "epoch": 6.132030487060548e-05,
      "model_forward_time": 2.3590128421783447,
      "step": 64299
    },
    {
      "epoch": 6.132030487060548e-05,
      "step": 64299,
      "training_step_time": 7.678197860717773
    },
    {
      "epoch": 6.132125854492187e-05,
      "grad_norm": 0.0952773168683052,
      "learning_rate": 3.0981986816521536e-06,
      "loss": 0.0593,
      "step": 64300
    },
    {
      "epoch": 6.132125854492187e-05,
      "model_forward_time": 1.8155450820922852,
      "step": 64300
    },
    {
      "dynamics_loss_avg": 0.05644277594983578,
      "epoch": 6.132125854492187e-05,
      "step": 64300
    },
    {
      "action_loss_avg": 0.0011145427240990102,
      "epoch": 6.132125854492187e-05,
      "step": 64300
    },
    {
      "epoch": 6.132125854492187e-05,
      "step": 64300,
      "training_step_time": 7.68168306350708
    },
    {
      "epoch": 6.132221221923828e-05,
      "model_forward_time": 1.8208019733428955,
      "step": 64301
    },
    {
      "epoch": 6.132221221923828e-05,
      "step": 64301,
      "training_step_time": 7.707690954208374
    },
    {
      "epoch": 6.13231658935547e-05,
      "model_forward_time": 1.813704252243042,
      "step": 64302
    },
    {
      "epoch": 6.13231658935547e-05,
      "step": 64302,
      "training_step_time": 9.725452184677124
    },
    {
      "epoch": 6.132411956787109e-05,
      "model_forward_time": 1.823056697845459,
      "step": 64303
    },
    {
      "epoch": 6.132411956787109e-05,
      "step": 64303,
      "training_step_time": 8.530350685119629
    },
    {
      "epoch": 6.13250732421875e-05,
      "model_forward_time": 1.8115224838256836,
      "step": 64304
    },
    {
      "epoch": 6.13250732421875e-05,
      "step": 64304,
      "training_step_time": 7.818596839904785
    },
    {
      "epoch": 6.132602691650391e-05,
      "model_forward_time": 1.8188683986663818,
      "step": 64305
    },
    {
      "epoch": 6.132602691650391e-05,
      "step": 64305,
      "training_step_time": 7.669708967208862
    },
    {
      "epoch": 6.132698059082031e-05,
      "model_forward_time": 1.7916948795318604,
      "step": 64306
    },
    {
      "epoch": 6.132698059082031e-05,
      "step": 64306,
      "training_step_time": 7.599680662155151
    },
    {
      "epoch": 6.132793426513672e-05,
      "model_forward_time": 1.8092772960662842,
      "step": 64307
    },
    {
      "epoch": 6.132793426513672e-05,
      "step": 64307,
      "training_step_time": 7.590084552764893
    },
    {
      "epoch": 6.132888793945313e-05,
      "model_forward_time": 2.3731515407562256,
      "step": 64308
    },
    {
      "epoch": 6.132888793945313e-05,
      "step": 64308,
      "training_step_time": 7.635131120681763
    },
    {
      "epoch": 6.132984161376953e-05,
      "model_forward_time": 1.819561243057251,
      "step": 64309
    },
    {
      "epoch": 6.132984161376953e-05,
      "step": 64309,
      "training_step_time": 7.675949811935425
    },
    {
      "epoch": 6.133079528808594e-05,
      "grad_norm": 0.09133244305849075,
      "learning_rate": 3.0966695932851478e-06,
      "loss": 0.0594,
      "step": 64310
    },
    {
      "epoch": 6.133079528808594e-05,
      "model_forward_time": 1.8078372478485107,
      "step": 64310
    },
    {
      "dynamics_loss_avg": 0.05391392204910517,
      "epoch": 6.133079528808594e-05,
      "step": 64310
    },
    {
      "action_loss_avg": 0.0018552802328485995,
      "epoch": 6.133079528808594e-05,
      "step": 64310
    },
    {
      "epoch": 6.133079528808594e-05,
      "step": 64310,
      "training_step_time": 8.27754259109497
    },
    {
      "epoch": 6.133174896240235e-05,
      "model_forward_time": 1.8163962364196777,
      "step": 64311
    },
    {
      "epoch": 6.133174896240235e-05,
      "step": 64311,
      "training_step_time": 10.03722620010376
    },
    {
      "epoch": 6.133270263671875e-05,
      "model_forward_time": 1.809558391571045,
      "step": 64312
    },
    {
      "epoch": 6.133270263671875e-05,
      "step": 64312,
      "training_step_time": 9.01036024093628
    },
    {
      "epoch": 6.133365631103516e-05,
      "model_forward_time": 1.806778907775879,
      "step": 64313
    },
    {
      "epoch": 6.133365631103516e-05,
      "step": 64313,
      "training_step_time": 7.6436097621917725
    },
    {
      "epoch": 6.133460998535156e-05,
      "model_forward_time": 1.808152675628662,
      "step": 64314
    },
    {
      "epoch": 6.133460998535156e-05,
      "step": 64314,
      "training_step_time": 8.50309944152832
    },
    {
      "epoch": 6.133556365966797e-05,
      "model_forward_time": 1.8183720111846924,
      "step": 64315
    },
    {
      "epoch": 6.133556365966797e-05,
      "step": 64315,
      "training_step_time": 7.282562494277954
    },
    {
      "epoch": 6.133651733398438e-05,
      "model_forward_time": 1.8156421184539795,
      "step": 64316
    },
    {
      "epoch": 6.133651733398438e-05,
      "step": 64316,
      "training_step_time": 7.524741888046265
    },
    {
      "epoch": 6.133747100830077e-05,
      "model_forward_time": 2.346090793609619,
      "step": 64317
    },
    {
      "epoch": 6.133747100830077e-05,
      "step": 64317,
      "training_step_time": 8.192232131958008
    },
    {
      "epoch": 6.133842468261719e-05,
      "model_forward_time": 1.8217692375183105,
      "step": 64318
    },
    {
      "epoch": 6.133842468261719e-05,
      "step": 64318,
      "training_step_time": 7.633181095123291
    },
    {
      "epoch": 6.13393783569336e-05,
      "model_forward_time": 1.8121623992919922,
      "step": 64319
    },
    {
      "epoch": 6.13393783569336e-05,
      "step": 64319,
      "training_step_time": 7.758175373077393
    },
    {
      "epoch": 6.134033203125e-05,
      "grad_norm": 0.0798288956284523,
      "learning_rate": 3.0951407130634923e-06,
      "loss": 0.0602,
      "step": 64320
    },
    {
      "epoch": 6.134033203125e-05,
      "model_forward_time": 1.8090851306915283,
      "step": 64320
    },
    {
      "dynamics_loss_avg": 0.049952448904514314,
      "epoch": 6.134033203125e-05,
      "step": 64320
    },
    {
      "action_loss_avg": 0.0012080072687240318,
      "epoch": 6.134033203125e-05,
      "step": 64320
    },
    {
      "epoch": 6.134033203125e-05,
      "step": 64320,
      "training_step_time": 8.233148097991943
    },
    {
      "epoch": 6.13412857055664e-05,
      "model_forward_time": 1.8158717155456543,
      "step": 64321
    },
    {
      "epoch": 6.13412857055664e-05,
      "step": 64321,
      "training_step_time": 7.759482145309448
    },
    {
      "epoch": 6.134223937988282e-05,
      "model_forward_time": 1.8112003803253174,
      "step": 64322
    },
    {
      "epoch": 6.134223937988282e-05,
      "step": 64322,
      "training_step_time": 7.654510259628296
    },
    {
      "epoch": 6.134319305419921e-05,
      "model_forward_time": 1.8099391460418701,
      "step": 64323
    },
    {
      "epoch": 6.134319305419921e-05,
      "step": 64323,
      "training_step_time": 7.739913702011108
    },
    {
      "epoch": 6.134414672851562e-05,
      "model_forward_time": 1.8193838596343994,
      "step": 64324
    },
    {
      "epoch": 6.134414672851562e-05,
      "step": 64324,
      "training_step_time": 7.697420835494995
    },
    {
      "epoch": 6.134510040283203e-05,
      "model_forward_time": 1.8156957626342773,
      "step": 64325
    },
    {
      "epoch": 6.134510040283203e-05,
      "step": 64325,
      "training_step_time": 7.624630928039551
    },
    {
      "epoch": 6.134605407714843e-05,
      "model_forward_time": 2.3352298736572266,
      "step": 64326
    },
    {
      "epoch": 6.134605407714843e-05,
      "step": 64326,
      "training_step_time": 7.669312238693237
    },
    {
      "epoch": 6.134700775146484e-05,
      "model_forward_time": 1.8196091651916504,
      "step": 64327
    },
    {
      "epoch": 6.134700775146484e-05,
      "step": 64327,
      "training_step_time": 8.724241971969604
    },
    {
      "epoch": 6.134796142578125e-05,
      "model_forward_time": 1.8082451820373535,
      "step": 64328
    },
    {
      "epoch": 6.134796142578125e-05,
      "step": 64328,
      "training_step_time": 8.759132146835327
    },
    {
      "epoch": 6.134891510009765e-05,
      "model_forward_time": 1.8159046173095703,
      "step": 64329
    },
    {
      "epoch": 6.134891510009765e-05,
      "step": 64329,
      "training_step_time": 7.728180885314941
    },
    {
      "epoch": 6.134986877441406e-05,
      "grad_norm": 0.1920625865459442,
      "learning_rate": 3.0936120411543864e-06,
      "loss": 0.0572,
      "step": 64330
    },
    {
      "epoch": 6.134986877441406e-05,
      "model_forward_time": 1.814666509628296,
      "step": 64330
    },
    {
      "dynamics_loss_avg": 0.06347446627914906,
      "epoch": 6.134986877441406e-05,
      "step": 64330
    },
    {
      "action_loss_avg": 0.002029473811853677,
      "epoch": 6.134986877441406e-05,
      "step": 64330
    },
    {
      "epoch": 6.134986877441406e-05,
      "step": 64330,
      "training_step_time": 9.263601303100586
    },
    {
      "epoch": 6.135082244873047e-05,
      "model_forward_time": 1.820636510848999,
      "step": 64331
    },
    {
      "epoch": 6.135082244873047e-05,
      "step": 64331,
      "training_step_time": 8.208834171295166
    },
    {
      "epoch": 6.135177612304687e-05,
      "model_forward_time": 1.7461028099060059,
      "step": 64332
    },
    {
      "epoch": 6.135177612304687e-05,
      "step": 64332,
      "training_step_time": 8.284154891967773
    },
    {
      "epoch": 6.135272979736328e-05,
      "model_forward_time": 1.710805892944336,
      "step": 64333
    },
    {
      "epoch": 6.135272979736328e-05,
      "step": 64333,
      "training_step_time": 7.653477191925049
    },
    {
      "epoch": 6.135368347167969e-05,
      "model_forward_time": 1.7055373191833496,
      "step": 64334
    },
    {
      "epoch": 6.135368347167969e-05,
      "step": 64334,
      "training_step_time": 8.619081258773804
    },
    {
      "epoch": 6.135463714599609e-05,
      "model_forward_time": 2.2611351013183594,
      "step": 64335
    },
    {
      "epoch": 6.135463714599609e-05,
      "step": 64335,
      "training_step_time": 7.596364736557007
    },
    {
      "epoch": 6.13555908203125e-05,
      "model_forward_time": 1.8271911144256592,
      "step": 64336
    },
    {
      "epoch": 6.13555908203125e-05,
      "step": 64336,
      "training_step_time": 7.668452024459839
    },
    {
      "epoch": 6.135654449462891e-05,
      "model_forward_time": 1.810027837753296,
      "step": 64337
    },
    {
      "epoch": 6.135654449462891e-05,
      "step": 64337,
      "training_step_time": 7.612186670303345
    },
    {
      "epoch": 6.13574981689453e-05,
      "model_forward_time": 1.735058307647705,
      "step": 64338
    },
    {
      "epoch": 6.13574981689453e-05,
      "step": 64338,
      "training_step_time": 8.45166563987732
    },
    {
      "epoch": 6.135845184326172e-05,
      "model_forward_time": 1.7119286060333252,
      "step": 64339
    },
    {
      "epoch": 6.135845184326172e-05,
      "step": 64339,
      "training_step_time": 9.356621742248535
    },
    {
      "epoch": 6.135940551757813e-05,
      "grad_norm": 0.09862619638442993,
      "learning_rate": 3.0920835777250014e-06,
      "loss": 0.0577,
      "step": 64340
    },
    {
      "epoch": 6.135940551757813e-05,
      "model_forward_time": 1.6947247982025146,
      "step": 64340
    },
    {
      "dynamics_loss_avg": 0.0721598131582141,
      "epoch": 6.135940551757813e-05,
      "step": 64340
    },
    {
      "action_loss_avg": 0.001710655735223554,
      "epoch": 6.135940551757813e-05,
      "step": 64340
    },
    {
      "epoch": 6.135940551757813e-05,
      "step": 64340,
      "training_step_time": 7.6314263343811035
    },
    {
      "epoch": 6.136035919189453e-05,
      "model_forward_time": 1.7148516178131104,
      "step": 64341
    },
    {
      "epoch": 6.136035919189453e-05,
      "step": 64341,
      "training_step_time": 7.672904968261719
    },
    {
      "epoch": 6.136131286621094e-05,
      "model_forward_time": 1.7080490589141846,
      "step": 64342
    },
    {
      "epoch": 6.136131286621094e-05,
      "step": 64342,
      "training_step_time": 7.639260292053223
    },
    {
      "epoch": 6.136226654052735e-05,
      "model_forward_time": 1.6978247165679932,
      "step": 64343
    },
    {
      "epoch": 6.136226654052735e-05,
      "step": 64343,
      "training_step_time": 7.587404012680054
    },
    {
      "epoch": 6.136322021484374e-05,
      "model_forward_time": 2.2771270275115967,
      "step": 64344
    },
    {
      "epoch": 6.136322021484374e-05,
      "step": 64344,
      "training_step_time": 8.487738132476807
    },
    {
      "epoch": 6.136417388916016e-05,
      "model_forward_time": 1.769455909729004,
      "step": 64345
    },
    {
      "epoch": 6.136417388916016e-05,
      "step": 64345,
      "training_step_time": 7.71032977104187
    },
    {
      "epoch": 6.136512756347657e-05,
      "model_forward_time": 1.7187280654907227,
      "step": 64346
    },
    {
      "epoch": 6.136512756347657e-05,
      "step": 64346,
      "training_step_time": 7.692290782928467
    },
    {
      "epoch": 6.136608123779296e-05,
      "model_forward_time": 1.7043511867523193,
      "step": 64347
    },
    {
      "epoch": 6.136608123779296e-05,
      "step": 64347,
      "training_step_time": 7.737756967544556
    },
    {
      "epoch": 6.136703491210937e-05,
      "model_forward_time": 1.7154958248138428,
      "step": 64348
    },
    {
      "epoch": 6.136703491210937e-05,
      "step": 64348,
      "training_step_time": 7.826995611190796
    },
    {
      "epoch": 6.136798858642578e-05,
      "model_forward_time": 1.7144837379455566,
      "step": 64349
    },
    {
      "epoch": 6.136798858642578e-05,
      "step": 64349,
      "training_step_time": 8.30875825881958
    },
    {
      "epoch": 6.136894226074218e-05,
      "grad_norm": 0.1510753631591797,
      "learning_rate": 3.0905553229424877e-06,
      "loss": 0.0585,
      "step": 64350
    },
    {
      "epoch": 6.136894226074218e-05,
      "model_forward_time": 1.712195873260498,
      "step": 64350
    },
    {
      "dynamics_loss_avg": 0.04757336899638176,
      "epoch": 6.136894226074218e-05,
      "step": 64350
    },
    {
      "action_loss_avg": 0.0008803610748145729,
      "epoch": 6.136894226074218e-05,
      "step": 64350
    },
    {
      "epoch": 6.136894226074218e-05,
      "step": 64350,
      "training_step_time": 7.7442405223846436
    },
    {
      "epoch": 6.136989593505859e-05,
      "model_forward_time": 1.713129997253418,
      "step": 64351
    },
    {
      "epoch": 6.136989593505859e-05,
      "step": 64351,
      "training_step_time": 7.602860927581787
    },
    {
      "epoch": 6.1370849609375e-05,
      "model_forward_time": 1.6896202564239502,
      "step": 64352
    },
    {
      "epoch": 6.1370849609375e-05,
      "step": 64352,
      "training_step_time": 7.599727630615234
    },
    {
      "epoch": 6.13718032836914e-05,
      "model_forward_time": 2.246436357498169,
      "step": 64353
    },
    {
      "epoch": 6.13718032836914e-05,
      "step": 64353,
      "training_step_time": 7.645099639892578
    },
    {
      "epoch": 6.137275695800781e-05,
      "model_forward_time": 1.7177395820617676,
      "step": 64354
    },
    {
      "epoch": 6.137275695800781e-05,
      "step": 64354,
      "training_step_time": 7.67734432220459
    },
    {
      "epoch": 6.137371063232422e-05,
      "model_forward_time": 1.6938254833221436,
      "step": 64355
    },
    {
      "epoch": 6.137371063232422e-05,
      "step": 64355,
      "training_step_time": 8.477152347564697
    },
    {
      "epoch": 6.137466430664062e-05,
      "model_forward_time": 1.6927142143249512,
      "step": 64356
    },
    {
      "epoch": 6.137466430664062e-05,
      "step": 64356,
      "training_step_time": 7.725536108016968
    },
    {
      "epoch": 6.137561798095703e-05,
      "model_forward_time": 1.6952276229858398,
      "step": 64357
    },
    {
      "epoch": 6.137561798095703e-05,
      "step": 64357,
      "training_step_time": 8.523231267929077
    },
    {
      "epoch": 6.137657165527344e-05,
      "model_forward_time": 1.6976158618927002,
      "step": 64358
    },
    {
      "epoch": 6.137657165527344e-05,
      "step": 64358,
      "training_step_time": 8.498076677322388
    },
    {
      "epoch": 6.137752532958984e-05,
      "model_forward_time": 1.6923062801361084,
      "step": 64359
    },
    {
      "epoch": 6.137752532958984e-05,
      "step": 64359,
      "training_step_time": 8.843475103378296
    },
    {
      "epoch": 6.137847900390625e-05,
      "grad_norm": 0.13667000830173492,
      "learning_rate": 3.0890272769739734e-06,
      "loss": 0.0575,
      "step": 64360
    },
    {
      "epoch": 6.137847900390625e-05,
      "model_forward_time": 1.698483943939209,
      "step": 64360
    },
    {
      "dynamics_loss_avg": 0.0649403365328908,
      "epoch": 6.137847900390625e-05,
      "step": 64360
    },
    {
      "action_loss_avg": 0.0007837602635845542,
      "epoch": 6.137847900390625e-05,
      "step": 64360
    },
    {
      "epoch": 6.137847900390625e-05,
      "step": 64360,
      "training_step_time": 7.621253490447998
    },
    {
      "epoch": 6.137943267822266e-05,
      "model_forward_time": 1.692901611328125,
      "step": 64361
    },
    {
      "epoch": 6.137943267822266e-05,
      "step": 64361,
      "training_step_time": 7.425992250442505
    },
    {
      "epoch": 6.138038635253906e-05,
      "model_forward_time": 2.2563798427581787,
      "step": 64362
    },
    {
      "epoch": 6.138038635253906e-05,
      "step": 64362,
      "training_step_time": 7.6567370891571045
    },
    {
      "epoch": 6.138134002685547e-05,
      "model_forward_time": 1.7101097106933594,
      "step": 64363
    },
    {
      "epoch": 6.138134002685547e-05,
      "step": 64363,
      "training_step_time": 7.650505304336548
    },
    {
      "epoch": 6.138229370117188e-05,
      "model_forward_time": 1.711965799331665,
      "step": 64364
    },
    {
      "epoch": 6.138229370117188e-05,
      "step": 64364,
      "training_step_time": 7.7837841510772705
    },
    {
      "epoch": 6.138324737548828e-05,
      "model_forward_time": 1.6938049793243408,
      "step": 64365
    },
    {
      "epoch": 6.138324737548828e-05,
      "step": 64365,
      "training_step_time": 8.286067008972168
    },
    {
      "epoch": 6.138420104980469e-05,
      "model_forward_time": 1.6981472969055176,
      "step": 64366
    },
    {
      "epoch": 6.138420104980469e-05,
      "step": 64366,
      "training_step_time": 8.168740034103394
    },
    {
      "epoch": 6.13851547241211e-05,
      "model_forward_time": 1.6884982585906982,
      "step": 64367
    },
    {
      "epoch": 6.13851547241211e-05,
      "step": 64367,
      "training_step_time": 8.66610860824585
    },
    {
      "epoch": 6.13861083984375e-05,
      "model_forward_time": 1.687835693359375,
      "step": 64368
    },
    {
      "epoch": 6.13861083984375e-05,
      "step": 64368,
      "training_step_time": 8.63503623008728
    },
    {
      "epoch": 6.13870620727539e-05,
      "model_forward_time": 1.690333604812622,
      "step": 64369
    },
    {
      "epoch": 6.13870620727539e-05,
      "step": 64369,
      "training_step_time": 8.151697874069214
    },
    {
      "epoch": 6.138801574707032e-05,
      "grad_norm": 0.09804116189479828,
      "learning_rate": 3.0874994399865616e-06,
      "loss": 0.0534,
      "step": 64370
    },
    {
      "epoch": 6.138801574707032e-05,
      "model_forward_time": 1.6886324882507324,
      "step": 64370
    },
    {
      "dynamics_loss_avg": 0.05381989423185587,
      "epoch": 6.138801574707032e-05,
      "step": 64370
    },
    {
      "action_loss_avg": 0.0015642345300875605,
      "epoch": 6.138801574707032e-05,
      "step": 64370
    },
    {
      "epoch": 6.138801574707032e-05,
      "step": 64370,
      "training_step_time": 7.533173322677612
    },
    {
      "epoch": 6.138896942138671e-05,
      "model_forward_time": 2.2511708736419678,
      "step": 64371
    },
    {
      "epoch": 6.138896942138671e-05,
      "step": 64371,
      "training_step_time": 9.228333950042725
    },
    {
      "epoch": 6.138992309570312e-05,
      "model_forward_time": 1.699436902999878,
      "step": 64372
    },
    {
      "epoch": 6.138992309570312e-05,
      "step": 64372,
      "training_step_time": 8.390181541442871
    },
    {
      "epoch": 6.139087677001954e-05,
      "model_forward_time": 1.7071328163146973,
      "step": 64373
    },
    {
      "epoch": 6.139087677001954e-05,
      "step": 64373,
      "training_step_time": 7.892733097076416
    },
    {
      "epoch": 6.139183044433593e-05,
      "model_forward_time": 1.692523717880249,
      "step": 64374
    },
    {
      "epoch": 6.139183044433593e-05,
      "step": 64374,
      "training_step_time": 9.253963708877563
    },
    {
      "epoch": 6.139278411865234e-05,
      "model_forward_time": 1.3215973377227783,
      "step": 64375
    },
    {
      "epoch": 6.139278411865234e-05,
      "step": 64375,
      "training_step_time": 8.383616924285889
    },
    {
      "epoch": 6.139373779296875e-05,
      "model_forward_time": 1.6920320987701416,
      "step": 64376
    },
    {
      "epoch": 6.139373779296875e-05,
      "step": 64376,
      "training_step_time": 8.782596826553345
    },
    {
      "epoch": 6.139469146728515e-05,
      "model_forward_time": 1.6985299587249756,
      "step": 64377
    },
    {
      "epoch": 6.139469146728515e-05,
      "step": 64377,
      "training_step_time": 7.668527603149414
    },
    {
      "epoch": 6.139564514160156e-05,
      "model_forward_time": 1.6893761157989502,
      "step": 64378
    },
    {
      "epoch": 6.139564514160156e-05,
      "step": 64378,
      "training_step_time": 8.685783624649048
    },
    {
      "epoch": 6.139659881591797e-05,
      "model_forward_time": 1.6958067417144775,
      "step": 64379
    },
    {
      "epoch": 6.139659881591797e-05,
      "step": 64379,
      "training_step_time": 7.629385709762573
    },
    {
      "epoch": 6.139755249023437e-05,
      "grad_norm": 0.0677061676979065,
      "learning_rate": 3.085971812147337e-06,
      "loss": 0.0528,
      "step": 64380
    },
    {
      "epoch": 6.139755249023437e-05,
      "model_forward_time": 2.250929355621338,
      "step": 64380
    },
    {
      "dynamics_loss_avg": 0.06057635117322206,
      "epoch": 6.139755249023437e-05,
      "step": 64380
    },
    {
      "action_loss_avg": 0.0019243943723267875,
      "epoch": 6.139755249023437e-05,
      "step": 64380
    },
    {
      "epoch": 6.139755249023437e-05,
      "step": 64380,
      "training_step_time": 7.728039503097534
    },
    {
      "epoch": 6.139850616455078e-05,
      "model_forward_time": 1.6940340995788574,
      "step": 64381
    },
    {
      "epoch": 6.139850616455078e-05,
      "step": 64381,
      "training_step_time": 7.577224493026733
    },
    {
      "epoch": 6.139945983886719e-05,
      "model_forward_time": 1.7288179397583008,
      "step": 64382
    },
    {
      "epoch": 6.139945983886719e-05,
      "step": 64382,
      "training_step_time": 7.634670972824097
    },
    {
      "epoch": 6.140041351318359e-05,
      "model_forward_time": 1.693725347518921,
      "step": 64383
    },
    {
      "epoch": 6.140041351318359e-05,
      "step": 64383,
      "training_step_time": 8.330543041229248
    },
    {
      "epoch": 6.14013671875e-05,
      "model_forward_time": 1.679227352142334,
      "step": 64384
    },
    {
      "epoch": 6.14013671875e-05,
      "step": 64384,
      "training_step_time": 9.246675729751587
    },
    {
      "epoch": 6.140232086181641e-05,
      "model_forward_time": 1.6789400577545166,
      "step": 64385
    },
    {
      "epoch": 6.140232086181641e-05,
      "step": 64385,
      "training_step_time": 9.231191396713257
    },
    {
      "epoch": 6.140327453613281e-05,
      "model_forward_time": 1.6828515529632568,
      "step": 64386
    },
    {
      "epoch": 6.140327453613281e-05,
      "step": 64386,
      "training_step_time": 7.626567125320435
    },
    {
      "epoch": 6.140422821044922e-05,
      "model_forward_time": 1.6963586807250977,
      "step": 64387
    },
    {
      "epoch": 6.140422821044922e-05,
      "step": 64387,
      "training_step_time": 8.758642673492432
    },
    {
      "epoch": 6.140518188476563e-05,
      "model_forward_time": 1.6914451122283936,
      "step": 64388
    },
    {
      "epoch": 6.140518188476563e-05,
      "step": 64388,
      "training_step_time": 7.6422669887542725
    },
    {
      "epoch": 6.140613555908203e-05,
      "model_forward_time": 2.276677370071411,
      "step": 64389
    },
    {
      "epoch": 6.140613555908203e-05,
      "step": 64389,
      "training_step_time": 7.638404130935669
    },
    {
      "epoch": 6.140708923339844e-05,
      "grad_norm": 0.13484375178813934,
      "learning_rate": 3.0844443936233557e-06,
      "loss": 0.0546,
      "step": 64390
    },
    {
      "epoch": 6.140708923339844e-05,
      "model_forward_time": 1.690230131149292,
      "step": 64390
    },
    {
      "dynamics_loss_avg": 0.051481026224792004,
      "epoch": 6.140708923339844e-05,
      "step": 64390
    },
    {
      "action_loss_avg": 0.0012481036712415517,
      "epoch": 6.140708923339844e-05,
      "step": 64390
    },
    {
      "epoch": 6.140708923339844e-05,
      "step": 64390,
      "training_step_time": 7.423565626144409
    },
    {
      "epoch": 6.140804290771485e-05,
      "model_forward_time": 1.721743106842041,
      "step": 64391
    },
    {
      "epoch": 6.140804290771485e-05,
      "step": 64391,
      "training_step_time": 8.089961767196655
    },
    {
      "epoch": 6.140899658203125e-05,
      "model_forward_time": 1.6821603775024414,
      "step": 64392
    },
    {
      "epoch": 6.140899658203125e-05,
      "step": 64392,
      "training_step_time": 9.872888565063477
    },
    {
      "epoch": 6.140995025634766e-05,
      "model_forward_time": 1.691258192062378,
      "step": 64393
    },
    {
      "epoch": 6.140995025634766e-05,
      "step": 64393,
      "training_step_time": 6.76212215423584
    },
    {
      "epoch": 6.141090393066407e-05,
      "model_forward_time": 1.6919069290161133,
      "step": 64394
    },
    {
      "epoch": 6.141090393066407e-05,
      "step": 64394,
      "training_step_time": 6.76902174949646
    },
    {
      "epoch": 6.141185760498046e-05,
      "model_forward_time": 1.6960163116455078,
      "step": 64395
    },
    {
      "epoch": 6.141185760498046e-05,
      "step": 64395,
      "training_step_time": 6.749239444732666
    },
    {
      "epoch": 6.141281127929688e-05,
      "model_forward_time": 1.6954588890075684,
      "step": 64396
    },
    {
      "epoch": 6.141281127929688e-05,
      "step": 64396,
      "training_step_time": 6.751253604888916
    },
    {
      "epoch": 6.141376495361329e-05,
      "model_forward_time": 1.6885554790496826,
      "step": 64397
    },
    {
      "epoch": 6.141376495361329e-05,
      "step": 64397,
      "training_step_time": 6.750876426696777
    },
    {
      "epoch": 6.141471862792968e-05,
      "model_forward_time": 1.6894865036010742,
      "step": 64398
    },
    {
      "epoch": 6.141471862792968e-05,
      "step": 64398,
      "training_step_time": 6.751129150390625
    },
    {
      "epoch": 6.14156723022461e-05,
      "model_forward_time": 1.6881744861602783,
      "step": 64399
    },
    {
      "epoch": 6.14156723022461e-05,
      "step": 64399,
      "training_step_time": 6.728300333023071
    },
    {
      "epoch": 6.14166259765625e-05,
      "grad_norm": 0.09653836488723755,
      "learning_rate": 3.082917184581657e-06,
      "loss": 0.0545,
      "step": 64400
    },
    {
      "epoch": 6.14166259765625e-05,
      "model_forward_time": 1.6846165657043457,
      "step": 64400
    },
    {
      "dynamics_loss_avg": 0.04988335818052292,
      "epoch": 6.14166259765625e-05,
      "step": 64400
    },
    {
      "action_loss_avg": 0.002215759429964237,
      "epoch": 6.14166259765625e-05,
      "step": 64400
    },
    {
      "epoch": 6.14166259765625e-05,
      "step": 64400,
      "training_step_time": 8.14607858657837
    },
    {
      "epoch": 6.14175796508789e-05,
      "model_forward_time": 1.713561773300171,
      "step": 64401
    },
    {
      "epoch": 6.14175796508789e-05,
      "step": 64401,
      "training_step_time": 7.1284308433532715
    },
    {
      "epoch": 6.141853332519531e-05,
      "model_forward_time": 1.6913950443267822,
      "step": 64402
    },
    {
      "epoch": 6.141853332519531e-05,
      "step": 64402,
      "training_step_time": 7.753541946411133
    },
    {
      "epoch": 6.141948699951172e-05,
      "model_forward_time": 1.6896512508392334,
      "step": 64403
    },
    {
      "epoch": 6.141948699951172e-05,
      "step": 64403,
      "training_step_time": 6.7631940841674805
    },
    {
      "epoch": 6.142044067382812e-05,
      "model_forward_time": 1.6924586296081543,
      "step": 64404
    },
    {
      "epoch": 6.142044067382812e-05,
      "step": 64404,
      "training_step_time": 6.750953197479248
    },
    {
      "epoch": 6.142139434814453e-05,
      "model_forward_time": 1.6966304779052734,
      "step": 64405
    },
    {
      "epoch": 6.142139434814453e-05,
      "step": 64405,
      "training_step_time": 6.806440830230713
    },
    {
      "epoch": 6.142234802246094e-05,
      "model_forward_time": 1.6855323314666748,
      "step": 64406
    },
    {
      "epoch": 6.142234802246094e-05,
      "step": 64406,
      "training_step_time": 6.71746563911438
    },
    {
      "epoch": 6.142330169677734e-05,
      "model_forward_time": 1.6902878284454346,
      "step": 64407
    },
    {
      "epoch": 6.142330169677734e-05,
      "step": 64407,
      "training_step_time": 6.798352241516113
    },
    {
      "epoch": 6.142425537109375e-05,
      "model_forward_time": 1.690298318862915,
      "step": 64408
    },
    {
      "epoch": 6.142425537109375e-05,
      "step": 64408,
      "training_step_time": 6.787489175796509
    },
    {
      "epoch": 6.142520904541016e-05,
      "model_forward_time": 1.6892139911651611,
      "step": 64409
    },
    {
      "epoch": 6.142520904541016e-05,
      "step": 64409,
      "training_step_time": 6.860159397125244
    },
    {
      "epoch": 6.142616271972656e-05,
      "grad_norm": 0.12111055850982666,
      "learning_rate": 3.0813901851892514e-06,
      "loss": 0.0521,
      "step": 64410
    },
    {
      "epoch": 6.142616271972656e-05,
      "model_forward_time": 2.288257360458374,
      "step": 64410
    },
    {
      "dynamics_loss_avg": 0.05335398502647877,
      "epoch": 6.142616271972656e-05,
      "step": 64410
    },
    {
      "action_loss_avg": 0.0016826315782964229,
      "epoch": 6.142616271972656e-05,
      "step": 64410
    },
    {
      "epoch": 6.142616271972656e-05,
      "step": 64410,
      "training_step_time": 8.262887001037598
    },
    {
      "epoch": 6.142711639404297e-05,
      "model_forward_time": 1.6899309158325195,
      "step": 64411
    },
    {
      "epoch": 6.142711639404297e-05,
      "step": 64411,
      "training_step_time": 8.834229469299316
    },
    {
      "epoch": 6.142807006835938e-05,
      "model_forward_time": 1.689873218536377,
      "step": 64412
    },
    {
      "epoch": 6.142807006835938e-05,
      "step": 64412,
      "training_step_time": 6.806420803070068
    },
    {
      "epoch": 6.142902374267578e-05,
      "model_forward_time": 1.6979172229766846,
      "step": 64413
    },
    {
      "epoch": 6.142902374267578e-05,
      "step": 64413,
      "training_step_time": 6.7959160804748535
    },
    {
      "epoch": 6.142997741699219e-05,
      "model_forward_time": 1.6911277770996094,
      "step": 64414
    },
    {
      "epoch": 6.142997741699219e-05,
      "step": 64414,
      "training_step_time": 6.813573837280273
    },
    {
      "epoch": 6.14309310913086e-05,
      "model_forward_time": 1.6935210227966309,
      "step": 64415
    },
    {
      "epoch": 6.14309310913086e-05,
      "step": 64415,
      "training_step_time": 6.863921165466309
    },
    {
      "epoch": 6.1431884765625e-05,
      "model_forward_time": 1.689772605895996,
      "step": 64416
    },
    {
      "epoch": 6.1431884765625e-05,
      "step": 64416,
      "training_step_time": 6.786393404006958
    },
    {
      "epoch": 6.143283843994141e-05,
      "model_forward_time": 1.690744161605835,
      "step": 64417
    },
    {
      "epoch": 6.143283843994141e-05,
      "step": 64417,
      "training_step_time": 6.823826789855957
    },
    {
      "epoch": 6.143379211425782e-05,
      "model_forward_time": 1.6928141117095947,
      "step": 64418
    },
    {
      "epoch": 6.143379211425782e-05,
      "step": 64418,
      "training_step_time": 6.883899211883545
    },
    {
      "epoch": 6.143474578857422e-05,
      "model_forward_time": 2.281635046005249,
      "step": 64419
    },
    {
      "epoch": 6.143474578857422e-05,
      "step": 64419,
      "training_step_time": 7.266351222991943
    },
    {
      "epoch": 6.143569946289063e-05,
      "grad_norm": 0.0960984155535698,
      "learning_rate": 3.079863395613132e-06,
      "loss": 0.0536,
      "step": 64420
    },
    {
      "epoch": 6.143569946289063e-05,
      "model_forward_time": 1.688797950744629,
      "step": 64420
    },
    {
      "dynamics_loss_avg": 0.05401697289198637,
      "epoch": 6.143569946289063e-05,
      "step": 64420
    },
    {
      "action_loss_avg": 0.0012434431118890643,
      "epoch": 6.143569946289063e-05,
      "step": 64420
    },
    {
      "epoch": 6.143569946289063e-05,
      "step": 64420,
      "training_step_time": 7.853723049163818
    },
    {
      "epoch": 6.143665313720704e-05,
      "model_forward_time": 1.694105863571167,
      "step": 64421
    },
    {
      "epoch": 6.143665313720704e-05,
      "step": 64421,
      "training_step_time": 7.637390851974487
    },
    {
      "epoch": 6.143760681152343e-05,
      "model_forward_time": 1.6974895000457764,
      "step": 64422
    },
    {
      "epoch": 6.143760681152343e-05,
      "step": 64422,
      "training_step_time": 6.812061309814453
    },
    {
      "epoch": 6.143856048583985e-05,
      "model_forward_time": 1.6997265815734863,
      "step": 64423
    },
    {
      "epoch": 6.143856048583985e-05,
      "step": 64423,
      "training_step_time": 6.773603916168213
    },
    {
      "epoch": 6.143951416015626e-05,
      "model_forward_time": 1.6920795440673828,
      "step": 64424
    },
    {
      "epoch": 6.143951416015626e-05,
      "step": 64424,
      "training_step_time": 6.76775050163269
    },
    {
      "epoch": 6.144046783447265e-05,
      "model_forward_time": 1.6983633041381836,
      "step": 64425
    },
    {
      "epoch": 6.144046783447265e-05,
      "step": 64425,
      "training_step_time": 6.797941446304321
    },
    {
      "epoch": 6.144142150878906e-05,
      "model_forward_time": 1.6919374465942383,
      "step": 64426
    },
    {
      "epoch": 6.144142150878906e-05,
      "step": 64426,
      "training_step_time": 6.751107215881348
    },
    {
      "epoch": 6.144237518310548e-05,
      "model_forward_time": 1.691772222518921,
      "step": 64427
    },
    {
      "epoch": 6.144237518310548e-05,
      "step": 64427,
      "training_step_time": 6.762033939361572
    },
    {
      "epoch": 6.144332885742187e-05,
      "model_forward_time": 2.2886390686035156,
      "step": 64428
    },
    {
      "epoch": 6.144332885742187e-05,
      "step": 64428,
      "training_step_time": 7.218151330947876
    },
    {
      "epoch": 6.144428253173828e-05,
      "model_forward_time": 1.6950922012329102,
      "step": 64429
    },
    {
      "epoch": 6.144428253173828e-05,
      "step": 64429,
      "training_step_time": 9.28353214263916
    },
    {
      "epoch": 6.14452362060547e-05,
      "grad_norm": 0.10070029646158218,
      "learning_rate": 3.0783368160202644e-06,
      "loss": 0.0546,
      "step": 64430
    },
    {
      "epoch": 6.14452362060547e-05,
      "model_forward_time": 1.689692735671997,
      "step": 64430
    },
    {
      "dynamics_loss_avg": 0.048489436693489554,
      "epoch": 6.14452362060547e-05,
      "step": 64430
    },
    {
      "action_loss_avg": 0.001492077886359766,
      "epoch": 6.14452362060547e-05,
      "step": 64430
    },
    {
      "epoch": 6.14452362060547e-05,
      "step": 64430,
      "training_step_time": 7.562180042266846
    },
    {
      "epoch": 6.144618988037109e-05,
      "model_forward_time": 1.6955695152282715,
      "step": 64431
    },
    {
      "epoch": 6.144618988037109e-05,
      "step": 64431,
      "training_step_time": 6.7436535358428955
    },
    {
      "epoch": 6.14471435546875e-05,
      "model_forward_time": 1.6948702335357666,
      "step": 64432
    },
    {
      "epoch": 6.14471435546875e-05,
      "step": 64432,
      "training_step_time": 6.77078104019165
    },
    {
      "epoch": 6.144809722900391e-05,
      "model_forward_time": 1.6921725273132324,
      "step": 64433
    },
    {
      "epoch": 6.144809722900391e-05,
      "step": 64433,
      "training_step_time": 6.767702102661133
    },
    {
      "epoch": 6.144905090332031e-05,
      "model_forward_time": 1.697091817855835,
      "step": 64434
    },
    {
      "epoch": 6.144905090332031e-05,
      "step": 64434,
      "training_step_time": 6.796332120895386
    },
    {
      "epoch": 6.145000457763672e-05,
      "model_forward_time": 1.697535753250122,
      "step": 64435
    },
    {
      "epoch": 6.145000457763672e-05,
      "step": 64435,
      "training_step_time": 6.771094083786011
    },
    {
      "epoch": 6.145095825195313e-05,
      "model_forward_time": 1.702608346939087,
      "step": 64436
    },
    {
      "epoch": 6.145095825195313e-05,
      "step": 64436,
      "training_step_time": 6.804194211959839
    },
    {
      "epoch": 6.145191192626953e-05,
      "model_forward_time": 2.257017135620117,
      "step": 64437
    },
    {
      "epoch": 6.145191192626953e-05,
      "step": 64437,
      "training_step_time": 7.223270416259766
    },
    {
      "epoch": 6.145286560058594e-05,
      "model_forward_time": 1.7144737243652344,
      "step": 64438
    },
    {
      "epoch": 6.145286560058594e-05,
      "step": 64438,
      "training_step_time": 8.823302507400513
    },
    {
      "epoch": 6.145381927490234e-05,
      "model_forward_time": 1.6923048496246338,
      "step": 64439
    },
    {
      "epoch": 6.145381927490234e-05,
      "step": 64439,
      "training_step_time": 7.589505434036255
    },
    {
      "epoch": 6.145477294921875e-05,
      "grad_norm": 0.09611938893795013,
      "learning_rate": 3.076810446577594e-06,
      "loss": 0.0522,
      "step": 64440
    },
    {
      "epoch": 6.145477294921875e-05,
      "model_forward_time": 1.6903798580169678,
      "step": 64440
    },
    {
      "dynamics_loss_avg": 0.050022862106561664,
      "epoch": 6.145477294921875e-05,
      "step": 64440
    },
    {
      "action_loss_avg": 0.0015217735839542002,
      "epoch": 6.145477294921875e-05,
      "step": 64440
    },
    {
      "epoch": 6.145477294921875e-05,
      "step": 64440,
      "training_step_time": 7.005975246429443
    },
    {
      "epoch": 6.145572662353516e-05,
      "model_forward_time": 1.6937592029571533,
      "step": 64441
    },
    {
      "epoch": 6.145572662353516e-05,
      "step": 64441,
      "training_step_time": 6.7577197551727295
    },
    {
      "epoch": 6.145668029785156e-05,
      "model_forward_time": 1.6995139122009277,
      "step": 64442
    },
    {
      "epoch": 6.145668029785156e-05,
      "step": 64442,
      "training_step_time": 6.772643327713013
    },
    {
      "epoch": 6.145763397216797e-05,
      "model_forward_time": 1.7013881206512451,
      "step": 64443
    },
    {
      "epoch": 6.145763397216797e-05,
      "step": 64443,
      "training_step_time": 6.820799112319946
    },
    {
      "epoch": 6.145858764648438e-05,
      "model_forward_time": 1.7077784538269043,
      "step": 64444
    },
    {
      "epoch": 6.145858764648438e-05,
      "step": 64444,
      "training_step_time": 6.75506854057312
    },
    {
      "epoch": 6.145954132080077e-05,
      "model_forward_time": 1.7013919353485107,
      "step": 64445
    },
    {
      "epoch": 6.145954132080077e-05,
      "step": 64445,
      "training_step_time": 6.7857348918914795
    },
    {
      "epoch": 6.146049499511719e-05,
      "model_forward_time": 1.8659782409667969,
      "step": 64446
    },
    {
      "epoch": 6.146049499511719e-05,
      "step": 64446,
      "training_step_time": 7.198055267333984
    },
    {
      "epoch": 6.14614486694336e-05,
      "model_forward_time": 1.71193265914917,
      "step": 64447
    },
    {
      "epoch": 6.14614486694336e-05,
      "step": 64447,
      "training_step_time": 9.278050661087036
    },
    {
      "epoch": 6.146240234375e-05,
      "model_forward_time": 1.7037131786346436,
      "step": 64448
    },
    {
      "epoch": 6.146240234375e-05,
      "step": 64448,
      "training_step_time": 7.663250923156738
    },
    {
      "epoch": 6.14633560180664e-05,
      "model_forward_time": 1.6950042247772217,
      "step": 64449
    },
    {
      "epoch": 6.14633560180664e-05,
      "step": 64449,
      "training_step_time": 7.086091756820679
    },
    {
      "epoch": 6.146430969238282e-05,
      "grad_norm": 0.1007528230547905,
      "learning_rate": 3.0752842874520393e-06,
      "loss": 0.057,
      "step": 64450
    },
    {
      "epoch": 6.146430969238282e-05,
      "model_forward_time": 1.6961233615875244,
      "step": 64450
    },
    {
      "dynamics_loss_avg": 0.04339969772845507,
      "epoch": 6.146430969238282e-05,
      "step": 64450
    },
    {
      "action_loss_avg": 0.0014123571017989889,
      "epoch": 6.146430969238282e-05,
      "step": 64450
    },
    {
      "epoch": 6.146430969238282e-05,
      "step": 64450,
      "training_step_time": 6.741989374160767
    },
    {
      "epoch": 6.146526336669921e-05,
      "model_forward_time": 1.7021257877349854,
      "step": 64451
    },
    {
      "epoch": 6.146526336669921e-05,
      "step": 64451,
      "training_step_time": 6.7276060581207275
    },
    {
      "epoch": 6.146621704101562e-05,
      "model_forward_time": 1.704014539718628,
      "step": 64452
    },
    {
      "epoch": 6.146621704101562e-05,
      "step": 64452,
      "training_step_time": 6.743414640426636
    },
    {
      "epoch": 6.146717071533203e-05,
      "model_forward_time": 1.6979711055755615,
      "step": 64453
    },
    {
      "epoch": 6.146717071533203e-05,
      "step": 64453,
      "training_step_time": 6.754131317138672
    },
    {
      "epoch": 6.146812438964843e-05,
      "model_forward_time": 1.6983602046966553,
      "step": 64454
    },
    {
      "epoch": 6.146812438964843e-05,
      "step": 64454,
      "training_step_time": 6.757024526596069
    },
    {
      "epoch": 6.146907806396484e-05,
      "model_forward_time": 1.6990160942077637,
      "step": 64455
    },
    {
      "epoch": 6.146907806396484e-05,
      "step": 64455,
      "training_step_time": 7.279330253601074
    },
    {
      "epoch": 6.147003173828125e-05,
      "model_forward_time": 2.2936723232269287,
      "step": 64456
    },
    {
      "epoch": 6.147003173828125e-05,
      "step": 64456,
      "training_step_time": 8.564260721206665
    },
    {
      "epoch": 6.147098541259765e-05,
      "model_forward_time": 1.6982262134552002,
      "step": 64457
    },
    {
      "epoch": 6.147098541259765e-05,
      "step": 64457,
      "training_step_time": 8.271276712417603
    },
    {
      "epoch": 6.147193908691406e-05,
      "model_forward_time": 1.698796033859253,
      "step": 64458
    },
    {
      "epoch": 6.147193908691406e-05,
      "step": 64458,
      "training_step_time": 7.300914287567139
    },
    {
      "epoch": 6.147289276123047e-05,
      "model_forward_time": 1.7027990818023682,
      "step": 64459
    },
    {
      "epoch": 6.147289276123047e-05,
      "step": 64459,
      "training_step_time": 6.816378355026245
    },
    {
      "epoch": 6.147384643554687e-05,
      "grad_norm": 0.10090985149145126,
      "learning_rate": 3.073758338810503e-06,
      "loss": 0.0558,
      "step": 64460
    },
    {
      "epoch": 6.147384643554687e-05,
      "model_forward_time": 1.6986117362976074,
      "step": 64460
    },
    {
      "dynamics_loss_avg": 0.046662386320531365,
      "epoch": 6.147384643554687e-05,
      "step": 64460
    },
    {
      "action_loss_avg": 0.0010687859248719178,
      "epoch": 6.147384643554687e-05,
      "step": 64460
    },
    {
      "epoch": 6.147384643554687e-05,
      "step": 64460,
      "training_step_time": 6.873180866241455
    },
    {
      "epoch": 6.147480010986328e-05,
      "model_forward_time": 1.6940407752990723,
      "step": 64461
    },
    {
      "epoch": 6.147480010986328e-05,
      "step": 64461,
      "training_step_time": 6.789514541625977
    },
    {
      "epoch": 6.147575378417969e-05,
      "model_forward_time": 1.699054479598999,
      "step": 64462
    },
    {
      "epoch": 6.147575378417969e-05,
      "step": 64462,
      "training_step_time": 6.750544309616089
    },
    {
      "epoch": 6.147670745849609e-05,
      "model_forward_time": 0.9777510166168213,
      "step": 64463
    },
    {
      "epoch": 6.147670745849609e-05,
      "step": 64463,
      "training_step_time": 6.75528359413147
    },
    {
      "epoch": 6.14776611328125e-05,
      "model_forward_time": 1.703566074371338,
      "step": 64464
    },
    {
      "epoch": 6.14776611328125e-05,
      "step": 64464,
      "training_step_time": 6.774396181106567
    },
    {
      "epoch": 6.147861480712891e-05,
      "model_forward_time": 2.2813963890075684,
      "step": 64465
    },
    {
      "epoch": 6.147861480712891e-05,
      "step": 64465,
      "training_step_time": 8.75446367263794
    },
    {
      "epoch": 6.14795684814453e-05,
      "model_forward_time": 1.319439172744751,
      "step": 64466
    },
    {
      "epoch": 6.14795684814453e-05,
      "step": 64466,
      "training_step_time": 10.67090129852295
    },
    {
      "epoch": 6.148052215576172e-05,
      "model_forward_time": 1.6937158107757568,
      "step": 64467
    },
    {
      "epoch": 6.148052215576172e-05,
      "step": 64467,
      "training_step_time": 7.6411004066467285
    },
    {
      "epoch": 6.148147583007813e-05,
      "model_forward_time": 1.6964590549468994,
      "step": 64468
    },
    {
      "epoch": 6.148147583007813e-05,
      "step": 64468,
      "training_step_time": 6.773860216140747
    },
    {
      "epoch": 6.148242950439453e-05,
      "model_forward_time": 1.6964800357818604,
      "step": 64469
    },
    {
      "epoch": 6.148242950439453e-05,
      "step": 64469,
      "training_step_time": 6.756400108337402
    },
    {
      "epoch": 6.148338317871094e-05,
      "grad_norm": 0.08300059288740158,
      "learning_rate": 3.072232600819859e-06,
      "loss": 0.0555,
      "step": 64470
    },
    {
      "epoch": 6.148338317871094e-05,
      "model_forward_time": 1.700124979019165,
      "step": 64470
    },
    {
      "dynamics_loss_avg": 0.049174043163657186,
      "epoch": 6.148338317871094e-05,
      "step": 64470
    },
    {
      "action_loss_avg": 0.0018819097182131372,
      "epoch": 6.148338317871094e-05,
      "step": 64470
    },
    {
      "epoch": 6.148338317871094e-05,
      "step": 64470,
      "training_step_time": 6.752636671066284
    },
    {
      "epoch": 6.148433685302735e-05,
      "model_forward_time": 1.7016682624816895,
      "step": 64471
    },
    {
      "epoch": 6.148433685302735e-05,
      "step": 64471,
      "training_step_time": 6.761728048324585
    },
    {
      "epoch": 6.148529052734374e-05,
      "model_forward_time": 1.7008485794067383,
      "step": 64472
    },
    {
      "epoch": 6.148529052734374e-05,
      "step": 64472,
      "training_step_time": 6.77583646774292
    },
    {
      "epoch": 6.148624420166016e-05,
      "model_forward_time": 1.7053217887878418,
      "step": 64473
    },
    {
      "epoch": 6.148624420166016e-05,
      "step": 64473,
      "training_step_time": 6.788855075836182
    },
    {
      "epoch": 6.148719787597657e-05,
      "model_forward_time": 1.7678308486938477,
      "step": 64474
    },
    {
      "epoch": 6.148719787597657e-05,
      "step": 64474,
      "training_step_time": 8.772363662719727
    },
    {
      "epoch": 6.148815155029296e-05,
      "model_forward_time": 1.7107055187225342,
      "step": 64475
    },
    {
      "epoch": 6.148815155029296e-05,
      "step": 64475,
      "training_step_time": 11.039956092834473
    },
    {
      "epoch": 6.148910522460937e-05,
      "model_forward_time": 1.7176663875579834,
      "step": 64476
    },
    {
      "epoch": 6.148910522460937e-05,
      "step": 64476,
      "training_step_time": 7.6494140625
    },
    {
      "epoch": 6.149005889892578e-05,
      "model_forward_time": 1.7098379135131836,
      "step": 64477
    },
    {
      "epoch": 6.149005889892578e-05,
      "step": 64477,
      "training_step_time": 7.044613599777222
    },
    {
      "epoch": 6.149101257324218e-05,
      "model_forward_time": 1.7087712287902832,
      "step": 64478
    },
    {
      "epoch": 6.149101257324218e-05,
      "step": 64478,
      "training_step_time": 6.721434116363525
    },
    {
      "epoch": 6.149196624755859e-05,
      "model_forward_time": 1.7094638347625732,
      "step": 64479
    },
    {
      "epoch": 6.149196624755859e-05,
      "step": 64479,
      "training_step_time": 6.755277872085571
    },
    {
      "epoch": 6.1492919921875e-05,
      "grad_norm": 0.0891525074839592,
      "learning_rate": 3.070707073646959e-06,
      "loss": 0.0549,
      "step": 64480
    },
    {
      "epoch": 6.1492919921875e-05,
      "model_forward_time": 1.7167282104492188,
      "step": 64480
    },
    {
      "dynamics_loss_avg": 0.06399030312895775,
      "epoch": 6.1492919921875e-05,
      "step": 64480
    },
    {
      "action_loss_avg": 0.00162293967732694,
      "epoch": 6.1492919921875e-05,
      "step": 64480
    },
    {
      "epoch": 6.1492919921875e-05,
      "step": 64480,
      "training_step_time": 6.771767616271973
    },
    {
      "epoch": 6.14938735961914e-05,
      "model_forward_time": 1.7150192260742188,
      "step": 64481
    },
    {
      "epoch": 6.14938735961914e-05,
      "step": 64481,
      "training_step_time": 6.7722320556640625
    },
    {
      "epoch": 6.149482727050781e-05,
      "model_forward_time": 1.7130382061004639,
      "step": 64482
    },
    {
      "epoch": 6.149482727050781e-05,
      "step": 64482,
      "training_step_time": 6.782389879226685
    },
    {
      "epoch": 6.149578094482422e-05,
      "model_forward_time": 1.741224765777588,
      "step": 64483
    },
    {
      "epoch": 6.149578094482422e-05,
      "step": 64483,
      "training_step_time": 7.732337474822998
    },
    {
      "epoch": 6.149673461914062e-05,
      "model_forward_time": 2.2722673416137695,
      "step": 64484
    },
    {
      "epoch": 6.149673461914062e-05,
      "step": 64484,
      "training_step_time": 10.448239803314209
    },
    {
      "epoch": 6.149768829345703e-05,
      "model_forward_time": 1.707230806350708,
      "step": 64485
    },
    {
      "epoch": 6.149768829345703e-05,
      "step": 64485,
      "training_step_time": 7.737907648086548
    },
    {
      "epoch": 6.149864196777344e-05,
      "model_forward_time": 1.708509922027588,
      "step": 64486
    },
    {
      "epoch": 6.149864196777344e-05,
      "step": 64486,
      "training_step_time": 7.6062188148498535
    },
    {
      "epoch": 6.149959564208984e-05,
      "model_forward_time": 1.344123125076294,
      "step": 64487
    },
    {
      "epoch": 6.149959564208984e-05,
      "step": 64487,
      "training_step_time": 6.753493070602417
    },
    {
      "epoch": 6.150054931640625e-05,
      "model_forward_time": 1.7179677486419678,
      "step": 64488
    },
    {
      "epoch": 6.150054931640625e-05,
      "step": 64488,
      "training_step_time": 6.773281574249268
    },
    {
      "epoch": 6.150150299072266e-05,
      "model_forward_time": 1.7120225429534912,
      "step": 64489
    },
    {
      "epoch": 6.150150299072266e-05,
      "step": 64489,
      "training_step_time": 6.7507100105285645
    },
    {
      "epoch": 6.150245666503906e-05,
      "grad_norm": 0.11570395529270172,
      "learning_rate": 3.069181757458633e-06,
      "loss": 0.0587,
      "step": 64490
    },
    {
      "epoch": 6.150245666503906e-05,
      "model_forward_time": 1.7096619606018066,
      "step": 64490
    },
    {
      "dynamics_loss_avg": 0.046261406131088735,
      "epoch": 6.150245666503906e-05,
      "step": 64490
    },
    {
      "action_loss_avg": 0.0012036054511554539,
      "epoch": 6.150245666503906e-05,
      "step": 64490
    },
    {
      "epoch": 6.150245666503906e-05,
      "step": 64490,
      "training_step_time": 6.82270073890686
    },
    {
      "epoch": 6.150341033935547e-05,
      "model_forward_time": 1.7112698554992676,
      "step": 64491
    },
    {
      "epoch": 6.150341033935547e-05,
      "step": 64491,
      "training_step_time": 6.787376880645752
    },
    {
      "epoch": 6.150436401367188e-05,
      "model_forward_time": 1.7262399196624756,
      "step": 64492
    },
    {
      "epoch": 6.150436401367188e-05,
      "step": 64492,
      "training_step_time": 7.747370004653931
    },
    {
      "epoch": 6.150531768798828e-05,
      "model_forward_time": 2.285421133041382,
      "step": 64493
    },
    {
      "epoch": 6.150531768798828e-05,
      "step": 64493,
      "training_step_time": 9.600287199020386
    },
    {
      "epoch": 6.150627136230469e-05,
      "model_forward_time": 1.7207682132720947,
      "step": 64494
    },
    {
      "epoch": 6.150627136230469e-05,
      "step": 64494,
      "training_step_time": 8.44431471824646
    },
    {
      "epoch": 6.15072250366211e-05,
      "model_forward_time": 1.7115285396575928,
      "step": 64495
    },
    {
      "epoch": 6.15072250366211e-05,
      "step": 64495,
      "training_step_time": 7.583995342254639
    },
    {
      "epoch": 6.15081787109375e-05,
      "model_forward_time": 1.7166309356689453,
      "step": 64496
    },
    {
      "epoch": 6.15081787109375e-05,
      "step": 64496,
      "training_step_time": 6.786384105682373
    },
    {
      "epoch": 6.15091323852539e-05,
      "model_forward_time": 1.7135727405548096,
      "step": 64497
    },
    {
      "epoch": 6.15091323852539e-05,
      "step": 64497,
      "training_step_time": 6.779183387756348
    },
    {
      "epoch": 6.151008605957032e-05,
      "model_forward_time": 1.7204172611236572,
      "step": 64498
    },
    {
      "epoch": 6.151008605957032e-05,
      "step": 64498,
      "training_step_time": 6.804325580596924
    },
    {
      "epoch": 6.151103973388671e-05,
      "model_forward_time": 1.7148756980895996,
      "step": 64499
    },
    {
      "epoch": 6.151103973388671e-05,
      "step": 64499,
      "training_step_time": 6.791613817214966
    },
    {
      "epoch": 6.151199340820312e-05,
      "grad_norm": 0.09273481369018555,
      "learning_rate": 3.067656652421687e-06,
      "loss": 0.0563,
      "step": 64500
    },
    {
      "epoch": 6.151199340820312e-05,
      "model_forward_time": 1.7037367820739746,
      "step": 64500
    },
    {
      "dynamics_loss_avg": 0.05720072342082858,
      "epoch": 6.151199340820312e-05,
      "step": 64500
    },
    {
      "action_loss_avg": 0.0012788201798684894,
      "epoch": 6.151199340820312e-05,
      "step": 64500
    },
    {
      "epoch": 6.151199340820312e-05,
      "step": 64500,
      "training_step_time": 6.810081481933594
    },
    {
      "epoch": 6.151294708251954e-05,
      "model_forward_time": 1.7130930423736572,
      "step": 64501
    },
    {
      "epoch": 6.151294708251954e-05,
      "step": 64501,
      "training_step_time": 8.469570875167847
    },
    {
      "epoch": 6.151390075683593e-05,
      "model_forward_time": 2.309422731399536,
      "step": 64502
    },
    {
      "epoch": 6.151390075683593e-05,
      "step": 64502,
      "training_step_time": 8.833608150482178
    },
    {
      "epoch": 6.151485443115234e-05,
      "model_forward_time": 1.708662986755371,
      "step": 64503
    },
    {
      "epoch": 6.151485443115234e-05,
      "step": 64503,
      "training_step_time": 8.64741063117981
    },
    {
      "epoch": 6.151580810546875e-05,
      "model_forward_time": 1.7211740016937256,
      "step": 64504
    },
    {
      "epoch": 6.151580810546875e-05,
      "step": 64504,
      "training_step_time": 7.643446922302246
    },
    {
      "epoch": 6.151676177978515e-05,
      "model_forward_time": 1.7109785079956055,
      "step": 64505
    },
    {
      "epoch": 6.151676177978515e-05,
      "step": 64505,
      "training_step_time": 7.56547474861145
    },
    {
      "epoch": 6.151771545410156e-05,
      "model_forward_time": 1.7126638889312744,
      "step": 64506
    },
    {
      "epoch": 6.151771545410156e-05,
      "step": 64506,
      "training_step_time": 6.819725275039673
    },
    {
      "epoch": 6.151866912841797e-05,
      "model_forward_time": 1.7116496562957764,
      "step": 64507
    },
    {
      "epoch": 6.151866912841797e-05,
      "step": 64507,
      "training_step_time": 6.820621490478516
    },
    {
      "epoch": 6.151962280273437e-05,
      "model_forward_time": 1.720186471939087,
      "step": 64508
    },
    {
      "epoch": 6.151962280273437e-05,
      "step": 64508,
      "training_step_time": 6.825915336608887
    },
    {
      "epoch": 6.152057647705078e-05,
      "model_forward_time": 1.7158901691436768,
      "step": 64509
    },
    {
      "epoch": 6.152057647705078e-05,
      "step": 64509,
      "training_step_time": 6.757627248764038
    },
    {
      "epoch": 6.152153015136719e-05,
      "grad_norm": 0.10739470273256302,
      "learning_rate": 3.0661317587029026e-06,
      "loss": 0.0578,
      "step": 64510
    },
    {
      "epoch": 6.152153015136719e-05,
      "model_forward_time": 1.7080662250518799,
      "step": 64510
    },
    {
      "dynamics_loss_avg": 0.05782341100275516,
      "epoch": 6.152153015136719e-05,
      "step": 64510
    },
    {
      "action_loss_avg": 0.0013100193347781897,
      "epoch": 6.152153015136719e-05,
      "step": 64510
    },
    {
      "epoch": 6.152153015136719e-05,
      "step": 64510,
      "training_step_time": 7.720073461532593
    },
    {
      "epoch": 6.152248382568359e-05,
      "model_forward_time": 2.2977490425109863,
      "step": 64511
    },
    {
      "epoch": 6.152248382568359e-05,
      "step": 64511,
      "training_step_time": 8.84362268447876
    },
    {
      "epoch": 6.15234375e-05,
      "model_forward_time": 1.720308542251587,
      "step": 64512
    },
    {
      "epoch": 6.15234375e-05,
      "step": 64512,
      "training_step_time": 9.927166223526001
    },
    {
      "epoch": 6.152439117431641e-05,
      "model_forward_time": 1.7170815467834473,
      "step": 64513
    },
    {
      "epoch": 6.152439117431641e-05,
      "step": 64513,
      "training_step_time": 7.6142737865448
    },
    {
      "epoch": 6.152534484863281e-05,
      "model_forward_time": 1.7106707096099854,
      "step": 64514
    },
    {
      "epoch": 6.152534484863281e-05,
      "step": 64514,
      "training_step_time": 7.5228376388549805
    },
    {
      "epoch": 6.152629852294922e-05,
      "model_forward_time": 1.713068962097168,
      "step": 64515
    },
    {
      "epoch": 6.152629852294922e-05,
      "step": 64515,
      "training_step_time": 6.74652361869812
    },
    {
      "epoch": 6.152725219726563e-05,
      "model_forward_time": 1.71649169921875,
      "step": 64516
    },
    {
      "epoch": 6.152725219726563e-05,
      "step": 64516,
      "training_step_time": 6.716732740402222
    },
    {
      "epoch": 6.152820587158203e-05,
      "model_forward_time": 1.7174835205078125,
      "step": 64517
    },
    {
      "epoch": 6.152820587158203e-05,
      "step": 64517,
      "training_step_time": 6.747321844100952
    },
    {
      "epoch": 6.152915954589844e-05,
      "model_forward_time": 1.7226486206054688,
      "step": 64518
    },
    {
      "epoch": 6.152915954589844e-05,
      "step": 64518,
      "training_step_time": 6.72546911239624
    },
    {
      "epoch": 6.153011322021485e-05,
      "model_forward_time": 1.734140157699585,
      "step": 64519
    },
    {
      "epoch": 6.153011322021485e-05,
      "step": 64519,
      "training_step_time": 7.5696234703063965
    },
    {
      "epoch": 6.153106689453125e-05,
      "grad_norm": 0.09702835977077484,
      "learning_rate": 3.0646070764690433e-06,
      "loss": 0.0552,
      "step": 64520
    },
    {
      "epoch": 6.153106689453125e-05,
      "model_forward_time": 1.7675504684448242,
      "step": 64520
    },
    {
      "dynamics_loss_avg": 0.0632768353447318,
      "epoch": 6.153106689453125e-05,
      "step": 64520
    },
    {
      "action_loss_avg": 0.002237891952972859,
      "epoch": 6.153106689453125e-05,
      "step": 64520
    },
    {
      "epoch": 6.153106689453125e-05,
      "step": 64520,
      "training_step_time": 8.70723032951355
    },
    {
      "epoch": 6.153202056884766e-05,
      "model_forward_time": 2.3018136024475098,
      "step": 64521
    },
    {
      "epoch": 6.153202056884766e-05,
      "step": 64521,
      "training_step_time": 9.297791719436646
    },
    {
      "epoch": 6.153297424316407e-05,
      "model_forward_time": 1.7130634784698486,
      "step": 64522
    },
    {
      "epoch": 6.153297424316407e-05,
      "step": 64522,
      "training_step_time": 8.721761465072632
    },
    {
      "epoch": 6.153392791748047e-05,
      "model_forward_time": 1.721357822418213,
      "step": 64523
    },
    {
      "epoch": 6.153392791748047e-05,
      "step": 64523,
      "training_step_time": 8.388410806655884
    },
    {
      "epoch": 6.153488159179688e-05,
      "model_forward_time": 1.7146599292755127,
      "step": 64524
    },
    {
      "epoch": 6.153488159179688e-05,
      "step": 64524,
      "training_step_time": 6.739274978637695
    },
    {
      "epoch": 6.153583526611329e-05,
      "model_forward_time": 1.7187561988830566,
      "step": 64525
    },
    {
      "epoch": 6.153583526611329e-05,
      "step": 64525,
      "training_step_time": 6.703083753585815
    },
    {
      "epoch": 6.153678894042968e-05,
      "model_forward_time": 1.7110321521759033,
      "step": 64526
    },
    {
      "epoch": 6.153678894042968e-05,
      "step": 64526,
      "training_step_time": 6.690347194671631
    },
    {
      "epoch": 6.15377426147461e-05,
      "model_forward_time": 1.7208821773529053,
      "step": 64527
    },
    {
      "epoch": 6.15377426147461e-05,
      "step": 64527,
      "training_step_time": 6.715952157974243
    },
    {
      "epoch": 6.15386962890625e-05,
      "model_forward_time": 1.713775634765625,
      "step": 64528
    },
    {
      "epoch": 6.15386962890625e-05,
      "step": 64528,
      "training_step_time": 7.510499954223633
    },
    {
      "epoch": 6.15396499633789e-05,
      "model_forward_time": 1.7411904335021973,
      "step": 64529
    },
    {
      "epoch": 6.15396499633789e-05,
      "step": 64529,
      "training_step_time": 8.522709846496582
    },
    {
      "epoch": 6.154060363769531e-05,
      "grad_norm": 0.081319659948349,
      "learning_rate": 3.0630826058868446e-06,
      "loss": 0.0564,
      "step": 64530
    },
    {
      "epoch": 6.154060363769531e-05,
      "model_forward_time": 2.2906203269958496,
      "step": 64530
    },
    {
      "dynamics_loss_avg": 0.0435422345995903,
      "epoch": 6.154060363769531e-05,
      "step": 64530
    },
    {
      "action_loss_avg": 0.001242457370972261,
      "epoch": 6.154060363769531e-05,
      "step": 64530
    },
    {
      "epoch": 6.154060363769531e-05,
      "step": 64530,
      "training_step_time": 8.766159534454346
    },
    {
      "epoch": 6.154155731201172e-05,
      "model_forward_time": 1.7148151397705078,
      "step": 64531
    },
    {
      "epoch": 6.154155731201172e-05,
      "step": 64531,
      "training_step_time": 7.6247334480285645
    },
    {
      "epoch": 6.154251098632812e-05,
      "model_forward_time": 1.7120990753173828,
      "step": 64532
    },
    {
      "epoch": 6.154251098632812e-05,
      "step": 64532,
      "training_step_time": 7.583258867263794
    },
    {
      "epoch": 6.154346466064453e-05,
      "model_forward_time": 1.7161531448364258,
      "step": 64533
    },
    {
      "epoch": 6.154346466064453e-05,
      "step": 64533,
      "training_step_time": 7.422667980194092
    },
    {
      "epoch": 6.154441833496094e-05,
      "model_forward_time": 1.7148423194885254,
      "step": 64534
    },
    {
      "epoch": 6.154441833496094e-05,
      "step": 64534,
      "training_step_time": 6.729084014892578
    },
    {
      "epoch": 6.154537200927734e-05,
      "model_forward_time": 1.7155442237854004,
      "step": 64535
    },
    {
      "epoch": 6.154537200927734e-05,
      "step": 64535,
      "training_step_time": 6.721041679382324
    },
    {
      "epoch": 6.154632568359375e-05,
      "model_forward_time": 1.7191095352172852,
      "step": 64536
    },
    {
      "epoch": 6.154632568359375e-05,
      "step": 64536,
      "training_step_time": 6.732373952865601
    },
    {
      "epoch": 6.154727935791016e-05,
      "model_forward_time": 1.7126071453094482,
      "step": 64537
    },
    {
      "epoch": 6.154727935791016e-05,
      "step": 64537,
      "training_step_time": 8.603838205337524
    },
    {
      "epoch": 6.154823303222656e-05,
      "model_forward_time": 1.7252790927886963,
      "step": 64538
    },
    {
      "epoch": 6.154823303222656e-05,
      "step": 64538,
      "training_step_time": 8.770678043365479
    },
    {
      "epoch": 6.154918670654297e-05,
      "model_forward_time": 2.276211977005005,
      "step": 64539
    },
    {
      "epoch": 6.154918670654297e-05,
      "step": 64539,
      "training_step_time": 9.613988637924194
    },
    {
      "epoch": 6.155014038085938e-05,
      "grad_norm": 0.07740704715251923,
      "learning_rate": 3.061558347123019e-06,
      "loss": 0.0541,
      "step": 64540
    },
    {
      "epoch": 6.155014038085938e-05,
      "model_forward_time": 1.7178394794464111,
      "step": 64540
    },
    {
      "dynamics_loss_avg": 0.0578910980373621,
      "epoch": 6.155014038085938e-05,
      "step": 64540
    },
    {
      "action_loss_avg": 0.0014823832432739436,
      "epoch": 6.155014038085938e-05,
      "step": 64540
    },
    {
      "epoch": 6.155014038085938e-05,
      "step": 64540,
      "training_step_time": 8.379564046859741
    },
    {
      "epoch": 6.155109405517578e-05,
      "model_forward_time": 1.7170031070709229,
      "step": 64541
    },
    {
      "epoch": 6.155109405517578e-05,
      "step": 64541,
      "training_step_time": 7.5802319049835205
    },
    {
      "epoch": 6.155204772949219e-05,
      "model_forward_time": 1.7166855335235596,
      "step": 64542
    },
    {
      "epoch": 6.155204772949219e-05,
      "step": 64542,
      "training_step_time": 7.494253873825073
    },
    {
      "epoch": 6.15530014038086e-05,
      "model_forward_time": 1.7177398204803467,
      "step": 64543
    },
    {
      "epoch": 6.15530014038086e-05,
      "step": 64543,
      "training_step_time": 6.727753400802612
    },
    {
      "epoch": 6.1553955078125e-05,
      "model_forward_time": 1.7124898433685303,
      "step": 64544
    },
    {
      "epoch": 6.1553955078125e-05,
      "step": 64544,
      "training_step_time": 6.7074058055877686
    },
    {
      "epoch": 6.155490875244141e-05,
      "model_forward_time": 1.7161734104156494,
      "step": 64545
    },
    {
      "epoch": 6.155490875244141e-05,
      "step": 64545,
      "training_step_time": 6.7302000522613525
    },
    {
      "epoch": 6.155586242675782e-05,
      "model_forward_time": 1.7248353958129883,
      "step": 64546
    },
    {
      "epoch": 6.155586242675782e-05,
      "step": 64546,
      "training_step_time": 7.592752695083618
    },
    {
      "epoch": 6.155681610107422e-05,
      "model_forward_time": 1.717865228652954,
      "step": 64547
    },
    {
      "epoch": 6.155681610107422e-05,
      "step": 64547,
      "training_step_time": 8.519834518432617
    },
    {
      "epoch": 6.155776977539063e-05,
      "model_forward_time": 2.2575769424438477,
      "step": 64548
    },
    {
      "epoch": 6.155776977539063e-05,
      "step": 64548,
      "training_step_time": 7.732296943664551
    },
    {
      "epoch": 6.155872344970704e-05,
      "model_forward_time": 1.7109792232513428,
      "step": 64549
    },
    {
      "epoch": 6.155872344970704e-05,
      "step": 64549,
      "training_step_time": 9.497328281402588
    },
    {
      "epoch": 6.155967712402343e-05,
      "grad_norm": 0.16528022289276123,
      "learning_rate": 3.060034300344259e-06,
      "loss": 0.0559,
      "step": 64550
    },
    {
      "epoch": 6.155967712402343e-05,
      "model_forward_time": 1.7306411266326904,
      "step": 64550
    },
    {
      "dynamics_loss_avg": 0.04759309682995081,
      "epoch": 6.155967712402343e-05,
      "step": 64550
    },
    {
      "action_loss_avg": 0.0009525223984383046,
      "epoch": 6.155967712402343e-05,
      "step": 64550
    },
    {
      "epoch": 6.155967712402343e-05,
      "step": 64550,
      "training_step_time": 7.666503429412842
    },
    {
      "epoch": 6.156063079833985e-05,
      "model_forward_time": 1.7166974544525146,
      "step": 64551
    },
    {
      "epoch": 6.156063079833985e-05,
      "step": 64551,
      "training_step_time": 7.610051870346069
    },
    {
      "epoch": 6.156158447265626e-05,
      "model_forward_time": 1.7136766910552979,
      "step": 64552
    },
    {
      "epoch": 6.156158447265626e-05,
      "step": 64552,
      "training_step_time": 6.742472887039185
    },
    {
      "epoch": 6.156253814697265e-05,
      "model_forward_time": 1.7161622047424316,
      "step": 64553
    },
    {
      "epoch": 6.156253814697265e-05,
      "step": 64553,
      "training_step_time": 6.729110956192017
    },
    {
      "epoch": 6.156349182128906e-05,
      "model_forward_time": 1.7194554805755615,
      "step": 64554
    },
    {
      "epoch": 6.156349182128906e-05,
      "step": 64554,
      "training_step_time": 6.720368385314941
    },
    {
      "epoch": 6.156444549560548e-05,
      "model_forward_time": 1.7179360389709473,
      "step": 64555
    },
    {
      "epoch": 6.156444549560548e-05,
      "step": 64555,
      "training_step_time": 7.4815449714660645
    },
    {
      "epoch": 6.156539916992187e-05,
      "model_forward_time": 1.7221605777740479,
      "step": 64556
    },
    {
      "epoch": 6.156539916992187e-05,
      "step": 64556,
      "training_step_time": 7.636495351791382
    },
    {
      "epoch": 6.156635284423828e-05,
      "model_forward_time": 2.303239107131958,
      "step": 64557
    },
    {
      "epoch": 6.156635284423828e-05,
      "step": 64557,
      "training_step_time": 9.269382238388062
    },
    {
      "epoch": 6.15673065185547e-05,
      "model_forward_time": 1.726902723312378,
      "step": 64558
    },
    {
      "epoch": 6.15673065185547e-05,
      "step": 64558,
      "training_step_time": 9.869946956634521
    },
    {
      "epoch": 6.156826019287109e-05,
      "model_forward_time": 1.7155704498291016,
      "step": 64559
    },
    {
      "epoch": 6.156826019287109e-05,
      "step": 64559,
      "training_step_time": 7.696974992752075
    },
    {
      "epoch": 6.15692138671875e-05,
      "grad_norm": 0.14413250982761383,
      "learning_rate": 3.0585104657172317e-06,
      "loss": 0.0542,
      "step": 64560
    },
    {
      "epoch": 6.15692138671875e-05,
      "model_forward_time": 1.7192745208740234,
      "step": 64560
    },
    {
      "dynamics_loss_avg": 0.057679989747703074,
      "epoch": 6.15692138671875e-05,
      "step": 64560
    },
    {
      "action_loss_avg": 0.0014166603446938097,
      "epoch": 6.15692138671875e-05,
      "step": 64560
    },
    {
      "epoch": 6.15692138671875e-05,
      "step": 64560,
      "training_step_time": 7.582689523696899
    },
    {
      "epoch": 6.157016754150391e-05,
      "model_forward_time": 1.7196934223175049,
      "step": 64561
    },
    {
      "epoch": 6.157016754150391e-05,
      "step": 64561,
      "training_step_time": 6.773426532745361
    },
    {
      "epoch": 6.157112121582031e-05,
      "model_forward_time": 1.717010736465454,
      "step": 64562
    },
    {
      "epoch": 6.157112121582031e-05,
      "step": 64562,
      "training_step_time": 6.796556234359741
    },
    {
      "epoch": 6.157207489013672e-05,
      "model_forward_time": 1.7112329006195068,
      "step": 64563
    },
    {
      "epoch": 6.157207489013672e-05,
      "step": 64563,
      "training_step_time": 6.746412038803101
    },
    {
      "epoch": 6.157302856445312e-05,
      "model_forward_time": 1.722245693206787,
      "step": 64564
    },
    {
      "epoch": 6.157302856445312e-05,
      "step": 64564,
      "training_step_time": 6.748989582061768
    },
    {
      "epoch": 6.157398223876953e-05,
      "model_forward_time": 1.7461035251617432,
      "step": 64565
    },
    {
      "epoch": 6.157398223876953e-05,
      "step": 64565,
      "training_step_time": 8.201919317245483
    },
    {
      "epoch": 6.157493591308594e-05,
      "model_forward_time": 1.765279769897461,
      "step": 64566
    },
    {
      "epoch": 6.157493591308594e-05,
      "step": 64566,
      "training_step_time": 8.224961996078491
    },
    {
      "epoch": 6.157588958740234e-05,
      "model_forward_time": 1.7088141441345215,
      "step": 64567
    },
    {
      "epoch": 6.157588958740234e-05,
      "step": 64567,
      "training_step_time": 10.030844688415527
    },
    {
      "epoch": 6.157684326171875e-05,
      "model_forward_time": 1.7047252655029297,
      "step": 64568
    },
    {
      "epoch": 6.157684326171875e-05,
      "step": 64568,
      "training_step_time": 7.6473777294158936
    },
    {
      "epoch": 6.157779693603516e-05,
      "model_forward_time": 1.7097437381744385,
      "step": 64569
    },
    {
      "epoch": 6.157779693603516e-05,
      "step": 64569,
      "training_step_time": 7.571412563323975
    },
    {
      "epoch": 6.157875061035156e-05,
      "grad_norm": 0.08643289655447006,
      "learning_rate": 3.0569868434085803e-06,
      "loss": 0.0581,
      "step": 64570
    },
    {
      "epoch": 6.157875061035156e-05,
      "model_forward_time": 1.7015810012817383,
      "step": 64570
    },
    {
      "dynamics_loss_avg": 0.05793606229126454,
      "epoch": 6.157875061035156e-05,
      "step": 64570
    },
    {
      "action_loss_avg": 0.0018082140828482806,
      "epoch": 6.157875061035156e-05,
      "step": 64570
    },
    {
      "epoch": 6.157875061035156e-05,
      "step": 64570,
      "training_step_time": 7.410808801651001
    },
    {
      "epoch": 6.157970428466797e-05,
      "model_forward_time": 1.7042381763458252,
      "step": 64571
    },
    {
      "epoch": 6.157970428466797e-05,
      "step": 64571,
      "training_step_time": 6.761142253875732
    },
    {
      "epoch": 6.158065795898438e-05,
      "model_forward_time": 1.712144136428833,
      "step": 64572
    },
    {
      "epoch": 6.158065795898438e-05,
      "step": 64572,
      "training_step_time": 6.727726936340332
    },
    {
      "epoch": 6.158161163330077e-05,
      "model_forward_time": 1.7342476844787598,
      "step": 64573
    },
    {
      "epoch": 6.158161163330077e-05,
      "step": 64573,
      "training_step_time": 6.728389501571655
    },
    {
      "epoch": 6.158256530761719e-05,
      "model_forward_time": 1.7079472541809082,
      "step": 64574
    },
    {
      "epoch": 6.158256530761719e-05,
      "step": 64574,
      "training_step_time": 8.176881551742554
    },
    {
      "epoch": 6.15835189819336e-05,
      "model_forward_time": 1.7359695434570312,
      "step": 64575
    },
    {
      "epoch": 6.15835189819336e-05,
      "step": 64575,
      "training_step_time": 8.230480194091797
    },
    {
      "epoch": 6.158447265625e-05,
      "model_forward_time": 2.2695271968841553,
      "step": 64576
    },
    {
      "epoch": 6.158447265625e-05,
      "step": 64576,
      "training_step_time": 8.290557384490967
    },
    {
      "epoch": 6.15854263305664e-05,
      "model_forward_time": 1.706390619277954,
      "step": 64577
    },
    {
      "epoch": 6.15854263305664e-05,
      "step": 64577,
      "training_step_time": 9.45252776145935
    },
    {
      "epoch": 6.158638000488282e-05,
      "model_forward_time": 1.7305293083190918,
      "step": 64578
    },
    {
      "epoch": 6.158638000488282e-05,
      "step": 64578,
      "training_step_time": 8.47119665145874
    },
    {
      "epoch": 6.158733367919921e-05,
      "model_forward_time": 1.7033212184906006,
      "step": 64579
    },
    {
      "epoch": 6.158733367919921e-05,
      "step": 64579,
      "training_step_time": 7.606273651123047
    },
    {
      "epoch": 6.158828735351562e-05,
      "grad_norm": 0.10688362270593643,
      "learning_rate": 3.055463433584924e-06,
      "loss": 0.0597,
      "step": 64580
    },
    {
      "epoch": 6.158828735351562e-05,
      "model_forward_time": 1.7063627243041992,
      "step": 64580
    },
    {
      "dynamics_loss_avg": 0.05872550960630178,
      "epoch": 6.158828735351562e-05,
      "step": 64580
    },
    {
      "action_loss_avg": 0.001902851409977302,
      "epoch": 6.158828735351562e-05,
      "step": 64580
    },
    {
      "epoch": 6.158828735351562e-05,
      "step": 64580,
      "training_step_time": 6.747651100158691
    },
    {
      "epoch": 6.158924102783203e-05,
      "model_forward_time": 1.7090296745300293,
      "step": 64581
    },
    {
      "epoch": 6.158924102783203e-05,
      "step": 64581,
      "training_step_time": 6.75231409072876
    },
    {
      "epoch": 6.159019470214843e-05,
      "model_forward_time": 1.7220964431762695,
      "step": 64582
    },
    {
      "epoch": 6.159019470214843e-05,
      "step": 64582,
      "training_step_time": 6.772325038909912
    },
    {
      "epoch": 6.159114837646484e-05,
      "model_forward_time": 1.72243332862854,
      "step": 64583
    },
    {
      "epoch": 6.159114837646484e-05,
      "step": 64583,
      "training_step_time": 8.615793466567993
    },
    {
      "epoch": 6.159210205078125e-05,
      "model_forward_time": 1.710275411605835,
      "step": 64584
    },
    {
      "epoch": 6.159210205078125e-05,
      "step": 64584,
      "training_step_time": 8.60351037979126
    },
    {
      "epoch": 6.159305572509765e-05,
      "model_forward_time": 2.260547637939453,
      "step": 64585
    },
    {
      "epoch": 6.159305572509765e-05,
      "step": 64585,
      "training_step_time": 8.313454389572144
    },
    {
      "epoch": 6.159400939941406e-05,
      "model_forward_time": 1.7114884853363037,
      "step": 64586
    },
    {
      "epoch": 6.159400939941406e-05,
      "step": 64586,
      "training_step_time": 9.737327098846436
    },
    {
      "epoch": 6.159496307373047e-05,
      "model_forward_time": 1.7077231407165527,
      "step": 64587
    },
    {
      "epoch": 6.159496307373047e-05,
      "step": 64587,
      "training_step_time": 7.591891527175903
    },
    {
      "epoch": 6.159591674804687e-05,
      "model_forward_time": 1.7061491012573242,
      "step": 64588
    },
    {
      "epoch": 6.159591674804687e-05,
      "step": 64588,
      "training_step_time": 7.52147102355957
    },
    {
      "epoch": 6.159687042236328e-05,
      "model_forward_time": 1.7126615047454834,
      "step": 64589
    },
    {
      "epoch": 6.159687042236328e-05,
      "step": 64589,
      "training_step_time": 7.393681049346924
    },
    {
      "epoch": 6.159782409667969e-05,
      "grad_norm": 0.0982779935002327,
      "learning_rate": 3.053940236412866e-06,
      "loss": 0.0565,
      "step": 64590
    },
    {
      "epoch": 6.159782409667969e-05,
      "model_forward_time": 1.711158037185669,
      "step": 64590
    },
    {
      "dynamics_loss_avg": 0.049928102642297745,
      "epoch": 6.159782409667969e-05,
      "step": 64590
    },
    {
      "action_loss_avg": 0.00150684270192869,
      "epoch": 6.159782409667969e-05,
      "step": 64590
    },
    {
      "epoch": 6.159782409667969e-05,
      "step": 64590,
      "training_step_time": 6.751074552536011
    },
    {
      "epoch": 6.159877777099609e-05,
      "model_forward_time": 1.7060513496398926,
      "step": 64591
    },
    {
      "epoch": 6.159877777099609e-05,
      "step": 64591,
      "training_step_time": 6.744281053543091
    },
    {
      "epoch": 6.15997314453125e-05,
      "model_forward_time": 1.7075777053833008,
      "step": 64592
    },
    {
      "epoch": 6.15997314453125e-05,
      "step": 64592,
      "training_step_time": 7.547064542770386
    },
    {
      "epoch": 6.160068511962891e-05,
      "model_forward_time": 1.7105858325958252,
      "step": 64593
    },
    {
      "epoch": 6.160068511962891e-05,
      "step": 64593,
      "training_step_time": 7.6870341300964355
    },
    {
      "epoch": 6.160163879394531e-05,
      "model_forward_time": 2.3199825286865234,
      "step": 64594
    },
    {
      "epoch": 6.160163879394531e-05,
      "step": 64594,
      "training_step_time": 10.534419059753418
    },
    {
      "epoch": 6.160259246826172e-05,
      "model_forward_time": 1.7140507698059082,
      "step": 64595
    },
    {
      "epoch": 6.160259246826172e-05,
      "step": 64595,
      "training_step_time": 8.756114482879639
    },
    {
      "epoch": 6.160354614257813e-05,
      "model_forward_time": 1.6966965198516846,
      "step": 64596
    },
    {
      "epoch": 6.160354614257813e-05,
      "step": 64596,
      "training_step_time": 8.888273239135742
    },
    {
      "epoch": 6.160449981689453e-05,
      "model_forward_time": 1.7159433364868164,
      "step": 64597
    },
    {
      "epoch": 6.160449981689453e-05,
      "step": 64597,
      "training_step_time": 7.558855295181274
    },
    {
      "epoch": 6.160545349121094e-05,
      "model_forward_time": 1.709670066833496,
      "step": 64598
    },
    {
      "epoch": 6.160545349121094e-05,
      "step": 64598,
      "training_step_time": 7.4750590324401855
    },
    {
      "epoch": 6.160640716552735e-05,
      "model_forward_time": 1.7074596881866455,
      "step": 64599
    },
    {
      "epoch": 6.160640716552735e-05,
      "step": 64599,
      "training_step_time": 6.7722063064575195
    },
    {
      "epoch": 6.160736083984374e-05,
      "grad_norm": 0.10300242900848389,
      "learning_rate": 3.052417252058977e-06,
      "loss": 0.0553,
      "step": 64600
    },
    {
      "epoch": 6.160736083984374e-05,
      "model_forward_time": 1.6972057819366455,
      "step": 64600
    },
    {
      "dynamics_loss_avg": 0.046015945449471476,
      "epoch": 6.160736083984374e-05,
      "step": 64600
    },
    {
      "action_loss_avg": 0.0018606971687404438,
      "epoch": 6.160736083984374e-05,
      "step": 64600
    },
    {
      "epoch": 6.160736083984374e-05,
      "step": 64600,
      "training_step_time": 6.734412431716919
    },
    {
      "epoch": 6.160831451416016e-05,
      "model_forward_time": 1.7051916122436523,
      "step": 64601
    },
    {
      "epoch": 6.160831451416016e-05,
      "step": 64601,
      "training_step_time": 8.493006229400635
    },
    {
      "epoch": 6.160926818847657e-05,
      "model_forward_time": 1.3348143100738525,
      "step": 64602
    },
    {
      "epoch": 6.160926818847657e-05,
      "step": 64602,
      "training_step_time": 8.253235816955566
    },
    {
      "epoch": 6.161022186279296e-05,
      "model_forward_time": 2.2985384464263916,
      "step": 64603
    },
    {
      "epoch": 6.161022186279296e-05,
      "step": 64603,
      "training_step_time": 8.545249700546265
    },
    {
      "epoch": 6.161117553710937e-05,
      "model_forward_time": 1.7101855278015137,
      "step": 64604
    },
    {
      "epoch": 6.161117553710937e-05,
      "step": 64604,
      "training_step_time": 8.883833169937134
    },
    {
      "epoch": 6.161212921142579e-05,
      "model_forward_time": 1.7043957710266113,
      "step": 64605
    },
    {
      "epoch": 6.161212921142579e-05,
      "step": 64605,
      "training_step_time": 7.757267475128174
    },
    {
      "epoch": 6.161308288574218e-05,
      "model_forward_time": 1.7121105194091797,
      "step": 64606
    },
    {
      "epoch": 6.161308288574218e-05,
      "step": 64606,
      "training_step_time": 7.642619848251343
    },
    {
      "epoch": 6.161403656005859e-05,
      "model_forward_time": 1.7026073932647705,
      "step": 64607
    },
    {
      "epoch": 6.161403656005859e-05,
      "step": 64607,
      "training_step_time": 7.615912437438965
    },
    {
      "epoch": 6.1614990234375e-05,
      "model_forward_time": 1.708432674407959,
      "step": 64608
    },
    {
      "epoch": 6.1614990234375e-05,
      "step": 64608,
      "training_step_time": 6.801519870758057
    },
    {
      "epoch": 6.16159439086914e-05,
      "model_forward_time": 1.7056214809417725,
      "step": 64609
    },
    {
      "epoch": 6.16159439086914e-05,
      "step": 64609,
      "training_step_time": 6.786519527435303
    },
    {
      "epoch": 6.161689758300781e-05,
      "grad_norm": 0.15230818092823029,
      "learning_rate": 3.0508944806898094e-06,
      "loss": 0.0605,
      "step": 64610
    },
    {
      "epoch": 6.161689758300781e-05,
      "model_forward_time": 1.7125301361083984,
      "step": 64610
    },
    {
      "dynamics_loss_avg": 0.0618498619645834,
      "epoch": 6.161689758300781e-05,
      "step": 64610
    },
    {
      "action_loss_avg": 0.001768719867686741,
      "epoch": 6.161689758300781e-05,
      "step": 64610
    },
    {
      "epoch": 6.161689758300781e-05,
      "step": 64610,
      "training_step_time": 7.397875070571899
    },
    {
      "epoch": 6.161785125732422e-05,
      "model_forward_time": 1.7052736282348633,
      "step": 64611
    },
    {
      "epoch": 6.161785125732422e-05,
      "step": 64611,
      "training_step_time": 7.74861478805542
    },
    {
      "epoch": 6.161880493164062e-05,
      "model_forward_time": 2.2701516151428223,
      "step": 64612
    },
    {
      "epoch": 6.161880493164062e-05,
      "step": 64612,
      "training_step_time": 7.734968185424805
    },
    {
      "epoch": 6.161975860595703e-05,
      "model_forward_time": 1.339113712310791,
      "step": 64613
    },
    {
      "epoch": 6.161975860595703e-05,
      "step": 64613,
      "training_step_time": 10.693596124649048
    },
    {
      "epoch": 6.162071228027344e-05,
      "model_forward_time": 1.7044124603271484,
      "step": 64614
    },
    {
      "epoch": 6.162071228027344e-05,
      "step": 64614,
      "training_step_time": 7.788188695907593
    },
    {
      "epoch": 6.162166595458984e-05,
      "model_forward_time": 1.697072982788086,
      "step": 64615
    },
    {
      "epoch": 6.162166595458984e-05,
      "step": 64615,
      "training_step_time": 7.605006694793701
    },
    {
      "epoch": 6.162261962890625e-05,
      "model_forward_time": 1.7124783992767334,
      "step": 64616
    },
    {
      "epoch": 6.162261962890625e-05,
      "step": 64616,
      "training_step_time": 7.613853693008423
    },
    {
      "epoch": 6.162357330322266e-05,
      "model_forward_time": 1.7079603672027588,
      "step": 64617
    },
    {
      "epoch": 6.162357330322266e-05,
      "step": 64617,
      "training_step_time": 6.778083086013794
    },
    {
      "epoch": 6.162452697753906e-05,
      "model_forward_time": 1.7046184539794922,
      "step": 64618
    },
    {
      "epoch": 6.162452697753906e-05,
      "step": 64618,
      "training_step_time": 6.784937381744385
    },
    {
      "epoch": 6.162548065185547e-05,
      "model_forward_time": 1.7226123809814453,
      "step": 64619
    },
    {
      "epoch": 6.162548065185547e-05,
      "step": 64619,
      "training_step_time": 7.437124967575073
    },
    {
      "epoch": 6.162643432617188e-05,
      "grad_norm": 0.10076654702425003,
      "learning_rate": 3.0493719224718897e-06,
      "loss": 0.0567,
      "step": 64620
    },
    {
      "epoch": 6.162643432617188e-05,
      "model_forward_time": 1.7128684520721436,
      "step": 64620
    },
    {
      "dynamics_loss_avg": 0.045643591694533824,
      "epoch": 6.162643432617188e-05,
      "step": 64620
    },
    {
      "action_loss_avg": 0.0011164352588821202,
      "epoch": 6.162643432617188e-05,
      "step": 64620
    },
    {
      "epoch": 6.162643432617188e-05,
      "step": 64620,
      "training_step_time": 7.657165288925171
    },
    {
      "epoch": 6.162738800048828e-05,
      "model_forward_time": 1.7314231395721436,
      "step": 64621
    },
    {
      "epoch": 6.162738800048828e-05,
      "step": 64621,
      "training_step_time": 9.760610818862915
    },
    {
      "epoch": 6.162834167480469e-05,
      "model_forward_time": 1.7101049423217773,
      "step": 64622
    },
    {
      "epoch": 6.162834167480469e-05,
      "step": 64622,
      "training_step_time": 8.378210544586182
    },
    {
      "epoch": 6.16292953491211e-05,
      "model_forward_time": 1.7153387069702148,
      "step": 64623
    },
    {
      "epoch": 6.16292953491211e-05,
      "step": 64623,
      "training_step_time": 9.675569534301758
    },
    {
      "epoch": 6.16302490234375e-05,
      "model_forward_time": 1.7104496955871582,
      "step": 64624
    },
    {
      "epoch": 6.16302490234375e-05,
      "step": 64624,
      "training_step_time": 7.548957347869873
    },
    {
      "epoch": 6.16312026977539e-05,
      "model_forward_time": 1.7146661281585693,
      "step": 64625
    },
    {
      "epoch": 6.16312026977539e-05,
      "step": 64625,
      "training_step_time": 7.551908016204834
    },
    {
      "epoch": 6.163215637207032e-05,
      "model_forward_time": 1.723158836364746,
      "step": 64626
    },
    {
      "epoch": 6.163215637207032e-05,
      "step": 64626,
      "training_step_time": 7.442468166351318
    },
    {
      "epoch": 6.163311004638671e-05,
      "model_forward_time": 1.7144064903259277,
      "step": 64627
    },
    {
      "epoch": 6.163311004638671e-05,
      "step": 64627,
      "training_step_time": 6.73993992805481
    },
    {
      "epoch": 6.163406372070313e-05,
      "model_forward_time": 1.709273338317871,
      "step": 64628
    },
    {
      "epoch": 6.163406372070313e-05,
      "step": 64628,
      "training_step_time": 7.400966167449951
    },
    {
      "epoch": 6.163501739501954e-05,
      "model_forward_time": 1.7094485759735107,
      "step": 64629
    },
    {
      "epoch": 6.163501739501954e-05,
      "step": 64629,
      "training_step_time": 7.587881565093994
    },
    {
      "epoch": 6.163597106933593e-05,
      "grad_norm": 0.1520281583070755,
      "learning_rate": 3.0478495775717252e-06,
      "loss": 0.0562,
      "step": 64630
    },
    {
      "epoch": 6.163597106933593e-05,
      "model_forward_time": 1.7150061130523682,
      "step": 64630
    },
    {
      "dynamics_loss_avg": 0.04582329578697682,
      "epoch": 6.163597106933593e-05,
      "step": 64630
    },
    {
      "action_loss_avg": 0.0008885158924385906,
      "epoch": 6.163597106933593e-05,
      "step": 64630
    },
    {
      "epoch": 6.163597106933593e-05,
      "step": 64630,
      "training_step_time": 8.78054141998291
    },
    {
      "epoch": 6.163692474365234e-05,
      "model_forward_time": 2.264907121658325,
      "step": 64631
    },
    {
      "epoch": 6.163692474365234e-05,
      "step": 64631,
      "training_step_time": 7.780498504638672
    },
    {
      "epoch": 6.163787841796875e-05,
      "model_forward_time": 1.7145843505859375,
      "step": 64632
    },
    {
      "epoch": 6.163787841796875e-05,
      "step": 64632,
      "training_step_time": 8.273003101348877
    },
    {
      "epoch": 6.163883209228515e-05,
      "model_forward_time": 1.7137258052825928,
      "step": 64633
    },
    {
      "epoch": 6.163883209228515e-05,
      "step": 64633,
      "training_step_time": 7.729309797286987
    },
    {
      "epoch": 6.163978576660156e-05,
      "model_forward_time": 1.71260404586792,
      "step": 64634
    },
    {
      "epoch": 6.163978576660156e-05,
      "step": 64634,
      "training_step_time": 8.140640258789062
    },
    {
      "epoch": 6.164073944091797e-05,
      "model_forward_time": 1.7196598052978516,
      "step": 64635
    },
    {
      "epoch": 6.164073944091797e-05,
      "step": 64635,
      "training_step_time": 7.579402685165405
    },
    {
      "epoch": 6.164169311523437e-05,
      "model_forward_time": 1.7045912742614746,
      "step": 64636
    },
    {
      "epoch": 6.164169311523437e-05,
      "step": 64636,
      "training_step_time": 6.735921859741211
    },
    {
      "epoch": 6.164264678955078e-05,
      "model_forward_time": 1.713240623474121,
      "step": 64637
    },
    {
      "epoch": 6.164264678955078e-05,
      "step": 64637,
      "training_step_time": 7.398085355758667
    },
    {
      "epoch": 6.164360046386719e-05,
      "model_forward_time": 1.7076971530914307,
      "step": 64638
    },
    {
      "epoch": 6.164360046386719e-05,
      "step": 64638,
      "training_step_time": 7.539851903915405
    },
    {
      "epoch": 6.164455413818359e-05,
      "model_forward_time": 1.7224085330963135,
      "step": 64639
    },
    {
      "epoch": 6.164455413818359e-05,
      "step": 64639,
      "training_step_time": 8.277461051940918
    },
    {
      "epoch": 6.16455078125e-05,
      "grad_norm": 0.07353416085243225,
      "learning_rate": 3.046327446155795e-06,
      "loss": 0.0562,
      "step": 64640
    },
    {
      "epoch": 6.16455078125e-05,
      "model_forward_time": 2.2623472213745117,
      "step": 64640
    },
    {
      "dynamics_loss_avg": 0.05115824341773987,
      "epoch": 6.16455078125e-05,
      "step": 64640
    },
    {
      "action_loss_avg": 0.0011944524099817499,
      "epoch": 6.16455078125e-05,
      "step": 64640
    },
    {
      "epoch": 6.16455078125e-05,
      "step": 64640,
      "training_step_time": 8.712424516677856
    },
    {
      "epoch": 6.164646148681641e-05,
      "model_forward_time": 1.714435338973999,
      "step": 64641
    },
    {
      "epoch": 6.164646148681641e-05,
      "step": 64641,
      "training_step_time": 8.385933637619019
    },
    {
      "epoch": 6.164741516113281e-05,
      "model_forward_time": 1.7069087028503418,
      "step": 64642
    },
    {
      "epoch": 6.164741516113281e-05,
      "step": 64642,
      "training_step_time": 8.668733358383179
    },
    {
      "epoch": 6.164836883544922e-05,
      "model_forward_time": 1.704206943511963,
      "step": 64643
    },
    {
      "epoch": 6.164836883544922e-05,
      "step": 64643,
      "training_step_time": 7.894433975219727
    },
    {
      "epoch": 6.164932250976563e-05,
      "model_forward_time": 1.7096648216247559,
      "step": 64644
    },
    {
      "epoch": 6.164932250976563e-05,
      "step": 64644,
      "training_step_time": 7.54742431640625
    },
    {
      "epoch": 6.165027618408203e-05,
      "model_forward_time": 1.7125859260559082,
      "step": 64645
    },
    {
      "epoch": 6.165027618408203e-05,
      "step": 64645,
      "training_step_time": 6.704731702804565
    },
    {
      "epoch": 6.165122985839844e-05,
      "model_forward_time": 1.703439712524414,
      "step": 64646
    },
    {
      "epoch": 6.165122985839844e-05,
      "step": 64646,
      "training_step_time": 7.372726202011108
    },
    {
      "epoch": 6.165218353271485e-05,
      "model_forward_time": 1.713183879852295,
      "step": 64647
    },
    {
      "epoch": 6.165218353271485e-05,
      "step": 64647,
      "training_step_time": 7.5116941928863525
    },
    {
      "epoch": 6.165313720703125e-05,
      "model_forward_time": 1.7062866687774658,
      "step": 64648
    },
    {
      "epoch": 6.165313720703125e-05,
      "step": 64648,
      "training_step_time": 8.217297315597534
    },
    {
      "epoch": 6.165409088134766e-05,
      "model_forward_time": 2.2833189964294434,
      "step": 64649
    },
    {
      "epoch": 6.165409088134766e-05,
      "step": 64649,
      "training_step_time": 8.592988967895508
    },
    {
      "epoch": 6.165504455566407e-05,
      "grad_norm": 0.09570906311273575,
      "learning_rate": 3.0448055283905586e-06,
      "loss": 0.0578,
      "step": 64650
    },
    {
      "epoch": 6.165504455566407e-05,
      "model_forward_time": 1.710921287536621,
      "step": 64650
    },
    {
      "dynamics_loss_avg": 0.042716284096240995,
      "epoch": 6.165504455566407e-05,
      "step": 64650
    },
    {
      "action_loss_avg": 0.001264384260866791,
      "epoch": 6.165504455566407e-05,
      "step": 64650
    },
    {
      "epoch": 6.165504455566407e-05,
      "step": 64650,
      "training_step_time": 9.81134295463562
    },
    {
      "epoch": 6.165599822998047e-05,
      "model_forward_time": 1.705836296081543,
      "step": 64651
    },
    {
      "epoch": 6.165599822998047e-05,
      "step": 64651,
      "training_step_time": 7.642046928405762
    },
    {
      "epoch": 6.165695190429688e-05,
      "model_forward_time": 1.7210490703582764,
      "step": 64652
    },
    {
      "epoch": 6.165695190429688e-05,
      "step": 64652,
      "training_step_time": 7.360671043395996
    },
    {
      "epoch": 6.165790557861329e-05,
      "model_forward_time": 1.6959757804870605,
      "step": 64653
    },
    {
      "epoch": 6.165790557861329e-05,
      "step": 64653,
      "training_step_time": 7.559515714645386
    },
    {
      "epoch": 6.165885925292968e-05,
      "model_forward_time": 1.7180018424987793,
      "step": 64654
    },
    {
      "epoch": 6.165885925292968e-05,
      "step": 64654,
      "training_step_time": 7.4715306758880615
    },
    {
      "epoch": 6.16598129272461e-05,
      "model_forward_time": 1.7275028228759766,
      "step": 64655
    },
    {
      "epoch": 6.16598129272461e-05,
      "step": 64655,
      "training_step_time": 7.968269109725952
    },
    {
      "epoch": 6.16607666015625e-05,
      "model_forward_time": 1.713512659072876,
      "step": 64656
    },
    {
      "epoch": 6.16607666015625e-05,
      "step": 64656,
      "training_step_time": 8.12339997291565
    },
    {
      "epoch": 6.16617202758789e-05,
      "model_forward_time": 1.710860252380371,
      "step": 64657
    },
    {
      "epoch": 6.16617202758789e-05,
      "step": 64657,
      "training_step_time": 8.93994951248169
    },
    {
      "epoch": 6.166267395019531e-05,
      "model_forward_time": 1.9054782390594482,
      "step": 64658
    },
    {
      "epoch": 6.166267395019531e-05,
      "step": 64658,
      "training_step_time": 8.679771661758423
    },
    {
      "epoch": 6.166362762451172e-05,
      "model_forward_time": 1.7067770957946777,
      "step": 64659
    },
    {
      "epoch": 6.166362762451172e-05,
      "step": 64659,
      "training_step_time": 9.304884672164917
    },
    {
      "epoch": 6.166458129882812e-05,
      "grad_norm": 0.08372647315263748,
      "learning_rate": 3.0432838244424497e-06,
      "loss": 0.0582,
      "step": 64660
    },
    {
      "epoch": 6.166458129882812e-05,
      "model_forward_time": 1.3423120975494385,
      "step": 64660
    },
    {
      "dynamics_loss_avg": 0.07800241652876139,
      "epoch": 6.166458129882812e-05,
      "step": 64660
    },
    {
      "action_loss_avg": 0.002469527331413701,
      "epoch": 6.166458129882812e-05,
      "step": 64660
    },
    {
      "epoch": 6.166458129882812e-05,
      "step": 64660,
      "training_step_time": 7.5821545124053955
    },
    {
      "epoch": 6.166553497314453e-05,
      "model_forward_time": 1.7246334552764893,
      "step": 64661
    },
    {
      "epoch": 6.166553497314453e-05,
      "step": 64661,
      "training_step_time": 7.709630250930786
    },
    {
      "epoch": 6.166648864746094e-05,
      "model_forward_time": 1.731774091720581,
      "step": 64662
    },
    {
      "epoch": 6.166648864746094e-05,
      "step": 64662,
      "training_step_time": 7.5627923011779785
    },
    {
      "epoch": 6.166744232177734e-05,
      "model_forward_time": 1.7160892486572266,
      "step": 64663
    },
    {
      "epoch": 6.166744232177734e-05,
      "step": 64663,
      "training_step_time": 7.560074090957642
    },
    {
      "epoch": 6.166839599609375e-05,
      "model_forward_time": 1.72218656539917,
      "step": 64664
    },
    {
      "epoch": 6.166839599609375e-05,
      "step": 64664,
      "training_step_time": 7.230109930038452
    },
    {
      "epoch": 6.166934967041016e-05,
      "model_forward_time": 1.7152438163757324,
      "step": 64665
    },
    {
      "epoch": 6.166934967041016e-05,
      "step": 64665,
      "training_step_time": 7.516678094863892
    },
    {
      "epoch": 6.167030334472656e-05,
      "model_forward_time": 1.7216176986694336,
      "step": 64666
    },
    {
      "epoch": 6.167030334472656e-05,
      "step": 64666,
      "training_step_time": 8.15759563446045
    },
    {
      "epoch": 6.167125701904297e-05,
      "model_forward_time": 1.7420766353607178,
      "step": 64667
    },
    {
      "epoch": 6.167125701904297e-05,
      "step": 64667,
      "training_step_time": 9.780942678451538
    },
    {
      "epoch": 6.167221069335938e-05,
      "model_forward_time": 2.2801694869995117,
      "step": 64668
    },
    {
      "epoch": 6.167221069335938e-05,
      "step": 64668,
      "training_step_time": 10.327314376831055
    },
    {
      "epoch": 6.167316436767578e-05,
      "model_forward_time": 1.7286744117736816,
      "step": 64669
    },
    {
      "epoch": 6.167316436767578e-05,
      "step": 64669,
      "training_step_time": 8.231865644454956
    },
    {
      "epoch": 6.167411804199219e-05,
      "grad_norm": 0.09917271882295609,
      "learning_rate": 3.04176233447788e-06,
      "loss": 0.0568,
      "step": 64670
    },
    {
      "epoch": 6.167411804199219e-05,
      "model_forward_time": 1.7127561569213867,
      "step": 64670
    },
    {
      "dynamics_loss_avg": 0.043346074596047404,
      "epoch": 6.167411804199219e-05,
      "step": 64670
    },
    {
      "action_loss_avg": 0.00119698871858418,
      "epoch": 6.167411804199219e-05,
      "step": 64670
    },
    {
      "epoch": 6.167411804199219e-05,
      "step": 64670,
      "training_step_time": 7.6931445598602295
    },
    {
      "epoch": 6.16750717163086e-05,
      "model_forward_time": 1.7110633850097656,
      "step": 64671
    },
    {
      "epoch": 6.16750717163086e-05,
      "step": 64671,
      "training_step_time": 7.535405158996582
    },
    {
      "epoch": 6.1676025390625e-05,
      "model_forward_time": 1.7291104793548584,
      "step": 64672
    },
    {
      "epoch": 6.1676025390625e-05,
      "step": 64672,
      "training_step_time": 7.570524454116821
    },
    {
      "epoch": 6.167697906494141e-05,
      "model_forward_time": 1.7176475524902344,
      "step": 64673
    },
    {
      "epoch": 6.167697906494141e-05,
      "step": 64673,
      "training_step_time": 7.229297399520874
    },
    {
      "epoch": 6.167793273925782e-05,
      "model_forward_time": 1.722182273864746,
      "step": 64674
    },
    {
      "epoch": 6.167793273925782e-05,
      "step": 64674,
      "training_step_time": 7.582762718200684
    },
    {
      "epoch": 6.167888641357422e-05,
      "model_forward_time": 1.715010404586792,
      "step": 64675
    },
    {
      "epoch": 6.167888641357422e-05,
      "step": 64675,
      "training_step_time": 8.681963682174683
    },
    {
      "epoch": 6.167984008789063e-05,
      "model_forward_time": 1.7274329662322998,
      "step": 64676
    },
    {
      "epoch": 6.167984008789063e-05,
      "step": 64676,
      "training_step_time": 7.701571464538574
    },
    {
      "epoch": 6.168079376220704e-05,
      "model_forward_time": 2.268775701522827,
      "step": 64677
    },
    {
      "epoch": 6.168079376220704e-05,
      "step": 64677,
      "training_step_time": 9.716413259506226
    },
    {
      "epoch": 6.168174743652343e-05,
      "model_forward_time": 1.7185108661651611,
      "step": 64678
    },
    {
      "epoch": 6.168174743652343e-05,
      "step": 64678,
      "training_step_time": 7.731470108032227
    },
    {
      "epoch": 6.168270111083985e-05,
      "model_forward_time": 1.7183258533477783,
      "step": 64679
    },
    {
      "epoch": 6.168270111083985e-05,
      "step": 64679,
      "training_step_time": 7.716492176055908
    },
    {
      "epoch": 6.168365478515626e-05,
      "grad_norm": 0.09153240919113159,
      "learning_rate": 3.040241058663238e-06,
      "loss": 0.0558,
      "step": 64680
    },
    {
      "epoch": 6.168365478515626e-05,
      "model_forward_time": 1.6930654048919678,
      "step": 64680
    },
    {
      "dynamics_loss_avg": 0.04528876766562462,
      "epoch": 6.168365478515626e-05,
      "step": 64680
    },
    {
      "action_loss_avg": 0.00166228777525248,
      "epoch": 6.168365478515626e-05,
      "step": 64680
    },
    {
      "epoch": 6.168365478515626e-05,
      "step": 64680,
      "training_step_time": 7.309586048126221
    },
    {
      "epoch": 6.168460845947265e-05,
      "model_forward_time": 1.8093297481536865,
      "step": 64681
    },
    {
      "epoch": 6.168460845947265e-05,
      "step": 64681,
      "training_step_time": 7.612967252731323
    },
    {
      "epoch": 6.168556213378906e-05,
      "model_forward_time": 1.811680555343628,
      "step": 64682
    },
    {
      "epoch": 6.168556213378906e-05,
      "step": 64682,
      "training_step_time": 7.65550684928894
    },
    {
      "epoch": 6.168651580810548e-05,
      "model_forward_time": 1.8102211952209473,
      "step": 64683
    },
    {
      "epoch": 6.168651580810548e-05,
      "step": 64683,
      "training_step_time": 7.578886032104492
    },
    {
      "epoch": 6.168746948242187e-05,
      "model_forward_time": 1.810582160949707,
      "step": 64684
    },
    {
      "epoch": 6.168746948242187e-05,
      "step": 64684,
      "training_step_time": 8.139662265777588
    },
    {
      "epoch": 6.168842315673828e-05,
      "model_forward_time": 1.807281255722046,
      "step": 64685
    },
    {
      "epoch": 6.168842315673828e-05,
      "step": 64685,
      "training_step_time": 8.826106071472168
    },
    {
      "epoch": 6.16893768310547e-05,
      "model_forward_time": 2.3535642623901367,
      "step": 64686
    },
    {
      "epoch": 6.16893768310547e-05,
      "step": 64686,
      "training_step_time": 8.348093271255493
    },
    {
      "epoch": 6.169033050537109e-05,
      "model_forward_time": 1.8044450283050537,
      "step": 64687
    },
    {
      "epoch": 6.169033050537109e-05,
      "step": 64687,
      "training_step_time": 10.386713743209839
    },
    {
      "epoch": 6.16912841796875e-05,
      "model_forward_time": 1.797969102859497,
      "step": 64688
    },
    {
      "epoch": 6.16912841796875e-05,
      "step": 64688,
      "training_step_time": 8.101469993591309
    },
    {
      "epoch": 6.16922378540039e-05,
      "model_forward_time": 1.8151671886444092,
      "step": 64689
    },
    {
      "epoch": 6.16922378540039e-05,
      "step": 64689,
      "training_step_time": 7.653476715087891
    },
    {
      "epoch": 6.169319152832031e-05,
      "grad_norm": 0.09095317870378494,
      "learning_rate": 3.0387199971648875e-06,
      "loss": 0.0561,
      "step": 64690
    },
    {
      "epoch": 6.169319152832031e-05,
      "model_forward_time": 1.8190815448760986,
      "step": 64690
    },
    {
      "dynamics_loss_avg": 0.055284886807203296,
      "epoch": 6.169319152832031e-05,
      "step": 64690
    },
    {
      "action_loss_avg": 0.0015132125874515623,
      "epoch": 6.169319152832031e-05,
      "step": 64690
    },
    {
      "epoch": 6.169319152832031e-05,
      "step": 64690,
      "training_step_time": 7.344374656677246
    },
    {
      "epoch": 6.169414520263672e-05,
      "model_forward_time": 1.8042962551116943,
      "step": 64691
    },
    {
      "epoch": 6.169414520263672e-05,
      "step": 64691,
      "training_step_time": 7.617323637008667
    },
    {
      "epoch": 6.169509887695312e-05,
      "model_forward_time": 1.8209636211395264,
      "step": 64692
    },
    {
      "epoch": 6.169509887695312e-05,
      "step": 64692,
      "training_step_time": 7.548919916152954
    },
    {
      "epoch": 6.169605255126953e-05,
      "model_forward_time": 1.8111083507537842,
      "step": 64693
    },
    {
      "epoch": 6.169605255126953e-05,
      "step": 64693,
      "training_step_time": 7.777438402175903
    },
    {
      "epoch": 6.169700622558594e-05,
      "model_forward_time": 1.8080973625183105,
      "step": 64694
    },
    {
      "epoch": 6.169700622558594e-05,
      "step": 64694,
      "training_step_time": 7.736015319824219
    },
    {
      "epoch": 6.169795989990234e-05,
      "model_forward_time": 2.3536667823791504,
      "step": 64695
    },
    {
      "epoch": 6.169795989990234e-05,
      "step": 64695,
      "training_step_time": 9.56755805015564
    },
    {
      "epoch": 6.169891357421875e-05,
      "model_forward_time": 1.4322254657745361,
      "step": 64696
    },
    {
      "epoch": 6.169891357421875e-05,
      "step": 64696,
      "training_step_time": 8.625165224075317
    },
    {
      "epoch": 6.169986724853516e-05,
      "model_forward_time": 1.8189454078674316,
      "step": 64697
    },
    {
      "epoch": 6.169986724853516e-05,
      "step": 64697,
      "training_step_time": 8.605216026306152
    },
    {
      "epoch": 6.170082092285156e-05,
      "model_forward_time": 1.8276662826538086,
      "step": 64698
    },
    {
      "epoch": 6.170082092285156e-05,
      "step": 64698,
      "training_step_time": 7.6489152908325195
    },
    {
      "epoch": 6.170177459716797e-05,
      "model_forward_time": 1.8258693218231201,
      "step": 64699
    },
    {
      "epoch": 6.170177459716797e-05,
      "step": 64699,
      "training_step_time": 7.416318893432617
    },
    {
      "epoch": 6.170272827148438e-05,
      "grad_norm": 0.10096155107021332,
      "learning_rate": 3.037199150149168e-06,
      "loss": 0.0546,
      "step": 64700
    },
    {
      "epoch": 6.170272827148438e-05,
      "model_forward_time": 1.824488878250122,
      "step": 64700
    },
    {
      "dynamics_loss_avg": 0.034027916053310035,
      "epoch": 6.170272827148438e-05,
      "step": 64700
    },
    {
      "action_loss_avg": 0.0009058142197318376,
      "epoch": 6.170272827148438e-05,
      "step": 64700
    },
    {
      "epoch": 6.170272827148438e-05,
      "step": 64700,
      "training_step_time": 7.5469958782196045
    },
    {
      "epoch": 6.170368194580077e-05,
      "model_forward_time": 1.8287489414215088,
      "step": 64701
    },
    {
      "epoch": 6.170368194580077e-05,
      "step": 64701,
      "training_step_time": 7.524998188018799
    },
    {
      "epoch": 6.170463562011719e-05,
      "model_forward_time": 1.8293416500091553,
      "step": 64702
    },
    {
      "epoch": 6.170463562011719e-05,
      "step": 64702,
      "training_step_time": 6.748964309692383
    },
    {
      "epoch": 6.17055892944336e-05,
      "model_forward_time": 1.8180789947509766,
      "step": 64703
    },
    {
      "epoch": 6.17055892944336e-05,
      "step": 64703,
      "training_step_time": 8.705548286437988
    },
    {
      "epoch": 6.170654296875e-05,
      "model_forward_time": 1.827437400817871,
      "step": 64704
    },
    {
      "epoch": 6.170654296875e-05,
      "step": 64704,
      "training_step_time": 10.210431814193726
    },
    {
      "epoch": 6.17074966430664e-05,
      "model_forward_time": 2.3783881664276123,
      "step": 64705
    },
    {
      "epoch": 6.17074966430664e-05,
      "step": 64705,
      "training_step_time": 8.484173059463501
    },
    {
      "epoch": 6.170845031738282e-05,
      "model_forward_time": 1.8230431079864502,
      "step": 64706
    },
    {
      "epoch": 6.170845031738282e-05,
      "step": 64706,
      "training_step_time": 9.191473007202148
    },
    {
      "epoch": 6.170940399169921e-05,
      "model_forward_time": 1.823564052581787,
      "step": 64707
    },
    {
      "epoch": 6.170940399169921e-05,
      "step": 64707,
      "training_step_time": 7.6657326221466064
    },
    {
      "epoch": 6.171035766601562e-05,
      "model_forward_time": 1.819234848022461,
      "step": 64708
    },
    {
      "epoch": 6.171035766601562e-05,
      "step": 64708,
      "training_step_time": 7.4068214893341064
    },
    {
      "epoch": 6.171131134033203e-05,
      "model_forward_time": 1.8220009803771973,
      "step": 64709
    },
    {
      "epoch": 6.171131134033203e-05,
      "step": 64709,
      "training_step_time": 7.556437253952026
    },
    {
      "epoch": 6.171226501464843e-05,
      "grad_norm": 0.09291329979896545,
      "learning_rate": 3.035678517782398e-06,
      "loss": 0.0565,
      "step": 64710
    },
    {
      "epoch": 6.171226501464843e-05,
      "model_forward_time": 1.82574462890625,
      "step": 64710
    },
    {
      "dynamics_loss_avg": 0.0421813546679914,
      "epoch": 6.171226501464843e-05,
      "step": 64710
    },
    {
      "action_loss_avg": 0.0011015516618499532,
      "epoch": 6.171226501464843e-05,
      "step": 64710
    },
    {
      "epoch": 6.171226501464843e-05,
      "step": 64710,
      "training_step_time": 7.549652099609375
    },
    {
      "epoch": 6.171321868896484e-05,
      "model_forward_time": 1.8346827030181885,
      "step": 64711
    },
    {
      "epoch": 6.171321868896484e-05,
      "step": 64711,
      "training_step_time": 7.593111515045166
    },
    {
      "epoch": 6.171417236328125e-05,
      "model_forward_time": 1.8352665901184082,
      "step": 64712
    },
    {
      "epoch": 6.171417236328125e-05,
      "step": 64712,
      "training_step_time": 7.665607929229736
    },
    {
      "epoch": 6.171512603759765e-05,
      "model_forward_time": 1.840822696685791,
      "step": 64713
    },
    {
      "epoch": 6.171512603759765e-05,
      "step": 64713,
      "training_step_time": 9.358785152435303
    },
    {
      "epoch": 6.171607971191406e-05,
      "model_forward_time": 2.438201904296875,
      "step": 64714
    },
    {
      "epoch": 6.171607971191406e-05,
      "step": 64714,
      "training_step_time": 8.66674280166626
    },
    {
      "epoch": 6.171703338623047e-05,
      "model_forward_time": 1.865161418914795,
      "step": 64715
    },
    {
      "epoch": 6.171703338623047e-05,
      "step": 64715,
      "training_step_time": 7.756603717803955
    },
    {
      "epoch": 6.171798706054687e-05,
      "model_forward_time": 1.8648312091827393,
      "step": 64716
    },
    {
      "epoch": 6.171798706054687e-05,
      "step": 64716,
      "training_step_time": 7.733378171920776
    },
    {
      "epoch": 6.171894073486328e-05,
      "model_forward_time": 1.866330623626709,
      "step": 64717
    },
    {
      "epoch": 6.171894073486328e-05,
      "step": 64717,
      "training_step_time": 7.093168497085571
    },
    {
      "epoch": 6.171989440917969e-05,
      "model_forward_time": 1.8742690086364746,
      "step": 64718
    },
    {
      "epoch": 6.171989440917969e-05,
      "step": 64718,
      "training_step_time": 7.564303874969482
    },
    {
      "epoch": 6.172084808349609e-05,
      "model_forward_time": 1.8590998649597168,
      "step": 64719
    },
    {
      "epoch": 6.172084808349609e-05,
      "step": 64719,
      "training_step_time": 7.584671974182129
    },
    {
      "epoch": 6.17218017578125e-05,
      "grad_norm": 0.10031361877918243,
      "learning_rate": 3.034158100230873e-06,
      "loss": 0.0559,
      "step": 64720
    },
    {
      "epoch": 6.17218017578125e-05,
      "model_forward_time": 1.7879836559295654,
      "step": 64720
    },
    {
      "dynamics_loss_avg": 0.061734906025230886,
      "epoch": 6.17218017578125e-05,
      "step": 64720
    },
    {
      "action_loss_avg": 0.0014777372387470677,
      "epoch": 6.17218017578125e-05,
      "step": 64720
    },
    {
      "epoch": 6.17218017578125e-05,
      "step": 64720,
      "training_step_time": 7.568509340286255
    },
    {
      "epoch": 6.172275543212891e-05,
      "model_forward_time": 1.7964146137237549,
      "step": 64721
    },
    {
      "epoch": 6.172275543212891e-05,
      "step": 64721,
      "training_step_time": 8.18656039237976
    },
    {
      "epoch": 6.172370910644531e-05,
      "model_forward_time": 1.8061089515686035,
      "step": 64722
    },
    {
      "epoch": 6.172370910644531e-05,
      "step": 64722,
      "training_step_time": 7.798649549484253
    },
    {
      "epoch": 6.172466278076172e-05,
      "model_forward_time": 2.3176074028015137,
      "step": 64723
    },
    {
      "epoch": 6.172466278076172e-05,
      "step": 64723,
      "training_step_time": 8.729962587356567
    },
    {
      "epoch": 6.172561645507813e-05,
      "model_forward_time": 1.808112382888794,
      "step": 64724
    },
    {
      "epoch": 6.172561645507813e-05,
      "step": 64724,
      "training_step_time": 9.790159225463867
    },
    {
      "epoch": 6.172657012939453e-05,
      "model_forward_time": 1.7799451351165771,
      "step": 64725
    },
    {
      "epoch": 6.172657012939453e-05,
      "step": 64725,
      "training_step_time": 9.104036569595337
    },
    {
      "epoch": 6.172752380371094e-05,
      "model_forward_time": 1.8144536018371582,
      "step": 64726
    },
    {
      "epoch": 6.172752380371094e-05,
      "step": 64726,
      "training_step_time": 7.62691068649292
    },
    {
      "epoch": 6.172847747802735e-05,
      "model_forward_time": 1.8048412799835205,
      "step": 64727
    },
    {
      "epoch": 6.172847747802735e-05,
      "step": 64727,
      "training_step_time": 7.5384438037872314
    },
    {
      "epoch": 6.172943115234374e-05,
      "model_forward_time": 1.8113734722137451,
      "step": 64728
    },
    {
      "epoch": 6.172943115234374e-05,
      "step": 64728,
      "training_step_time": 7.573195457458496
    },
    {
      "epoch": 6.173038482666016e-05,
      "model_forward_time": 1.815201997756958,
      "step": 64729
    },
    {
      "epoch": 6.173038482666016e-05,
      "step": 64729,
      "training_step_time": 7.501024484634399
    },
    {
      "epoch": 6.173133850097657e-05,
      "grad_norm": 0.10440602898597717,
      "learning_rate": 3.032637897660863e-06,
      "loss": 0.0565,
      "step": 64730
    },
    {
      "epoch": 6.173133850097657e-05,
      "model_forward_time": 1.8094592094421387,
      "step": 64730
    },
    {
      "dynamics_loss_avg": 0.06041882317513227,
      "epoch": 6.173133850097657e-05,
      "step": 64730
    },
    {
      "action_loss_avg": 0.0013781133864540606,
      "epoch": 6.173133850097657e-05,
      "step": 64730
    },
    {
      "epoch": 6.173133850097657e-05,
      "step": 64730,
      "training_step_time": 7.593379974365234
    },
    {
      "epoch": 6.173229217529296e-05,
      "model_forward_time": 1.790252923965454,
      "step": 64731
    },
    {
      "epoch": 6.173229217529296e-05,
      "step": 64731,
      "training_step_time": 9.28682827949524
    },
    {
      "epoch": 6.173324584960937e-05,
      "model_forward_time": 2.3575050830841064,
      "step": 64732
    },
    {
      "epoch": 6.173324584960937e-05,
      "step": 64732,
      "training_step_time": 9.08311128616333
    },
    {
      "epoch": 6.173419952392579e-05,
      "model_forward_time": 1.434272289276123,
      "step": 64733
    },
    {
      "epoch": 6.173419952392579e-05,
      "step": 64733,
      "training_step_time": 9.137410640716553
    },
    {
      "epoch": 6.173515319824218e-05,
      "model_forward_time": 1.8104965686798096,
      "step": 64734
    },
    {
      "epoch": 6.173515319824218e-05,
      "step": 64734,
      "training_step_time": 7.666305303573608
    },
    {
      "epoch": 6.173610687255859e-05,
      "model_forward_time": 1.804173469543457,
      "step": 64735
    },
    {
      "epoch": 6.173610687255859e-05,
      "step": 64735,
      "training_step_time": 7.693725824356079
    },
    {
      "epoch": 6.1737060546875e-05,
      "model_forward_time": 1.8019134998321533,
      "step": 64736
    },
    {
      "epoch": 6.1737060546875e-05,
      "step": 64736,
      "training_step_time": 7.508413553237915
    },
    {
      "epoch": 6.17380142211914e-05,
      "model_forward_time": 1.8079719543457031,
      "step": 64737
    },
    {
      "epoch": 6.17380142211914e-05,
      "step": 64737,
      "training_step_time": 7.606522560119629
    },
    {
      "epoch": 6.173896789550781e-05,
      "model_forward_time": 1.807943344116211,
      "step": 64738
    },
    {
      "epoch": 6.173896789550781e-05,
      "step": 64738,
      "training_step_time": 7.554945468902588
    },
    {
      "epoch": 6.173992156982422e-05,
      "model_forward_time": 1.8056509494781494,
      "step": 64739
    },
    {
      "epoch": 6.173992156982422e-05,
      "step": 64739,
      "training_step_time": 7.5549116134643555
    },
    {
      "epoch": 6.174087524414062e-05,
      "grad_norm": 0.07327970862388611,
      "learning_rate": 3.0311179102386147e-06,
      "loss": 0.0575,
      "step": 64740
    },
    {
      "epoch": 6.174087524414062e-05,
      "model_forward_time": 1.8004937171936035,
      "step": 64740
    },
    {
      "dynamics_loss_avg": 0.060546718537807465,
      "epoch": 6.174087524414062e-05,
      "step": 64740
    },
    {
      "action_loss_avg": 0.0019375980453332886,
      "epoch": 6.174087524414062e-05,
      "step": 64740
    },
    {
      "epoch": 6.174087524414062e-05,
      "step": 64740,
      "training_step_time": 8.709952354431152
    },
    {
      "epoch": 6.174182891845703e-05,
      "model_forward_time": 1.8007526397705078,
      "step": 64741
    },
    {
      "epoch": 6.174182891845703e-05,
      "step": 64741,
      "training_step_time": 8.441254138946533
    },
    {
      "epoch": 6.174278259277344e-05,
      "model_forward_time": 2.409177303314209,
      "step": 64742
    },
    {
      "epoch": 6.174278259277344e-05,
      "step": 64742,
      "training_step_time": 7.69206166267395
    },
    {
      "epoch": 6.174373626708984e-05,
      "model_forward_time": 1.8832685947418213,
      "step": 64743
    },
    {
      "epoch": 6.174373626708984e-05,
      "step": 64743,
      "training_step_time": 8.341347694396973
    },
    {
      "epoch": 6.174468994140625e-05,
      "model_forward_time": 1.863851547241211,
      "step": 64744
    },
    {
      "epoch": 6.174468994140625e-05,
      "step": 64744,
      "training_step_time": 8.352640390396118
    },
    {
      "epoch": 6.174564361572266e-05,
      "model_forward_time": 1.8884665966033936,
      "step": 64745
    },
    {
      "epoch": 6.174564361572266e-05,
      "step": 64745,
      "training_step_time": 7.400362253189087
    },
    {
      "epoch": 6.174659729003906e-05,
      "model_forward_time": 1.8720638751983643,
      "step": 64746
    },
    {
      "epoch": 6.174659729003906e-05,
      "step": 64746,
      "training_step_time": 7.522454500198364
    },
    {
      "epoch": 6.174755096435547e-05,
      "model_forward_time": 1.8701307773590088,
      "step": 64747
    },
    {
      "epoch": 6.174755096435547e-05,
      "step": 64747,
      "training_step_time": 6.71740198135376
    },
    {
      "epoch": 6.174850463867188e-05,
      "model_forward_time": 1.870682716369629,
      "step": 64748
    },
    {
      "epoch": 6.174850463867188e-05,
      "step": 64748,
      "training_step_time": 7.513866662979126
    },
    {
      "epoch": 6.174945831298828e-05,
      "model_forward_time": 1.88045072555542,
      "step": 64749
    },
    {
      "epoch": 6.174945831298828e-05,
      "step": 64749,
      "training_step_time": 9.488488674163818
    },
    {
      "epoch": 6.175041198730469e-05,
      "grad_norm": 0.11377429217100143,
      "learning_rate": 3.029598138130351e-06,
      "loss": 0.0579,
      "step": 64750
    },
    {
      "epoch": 6.175041198730469e-05,
      "model_forward_time": 1.8621745109558105,
      "step": 64750
    },
    {
      "dynamics_loss_avg": 0.04602745361626148,
      "epoch": 6.175041198730469e-05,
      "step": 64750
    },
    {
      "action_loss_avg": 0.0009118689253227785,
      "epoch": 6.175041198730469e-05,
      "step": 64750
    },
    {
      "epoch": 6.175041198730469e-05,
      "step": 64750,
      "training_step_time": 8.326711416244507
    },
    {
      "epoch": 6.17513656616211e-05,
      "model_forward_time": 2.3977298736572266,
      "step": 64751
    },
    {
      "epoch": 6.17513656616211e-05,
      "step": 64751,
      "training_step_time": 9.677587747573853
    },
    {
      "epoch": 6.17523193359375e-05,
      "model_forward_time": 1.881648302078247,
      "step": 64752
    },
    {
      "epoch": 6.17523193359375e-05,
      "step": 64752,
      "training_step_time": 7.6815173625946045
    },
    {
      "epoch": 6.17532730102539e-05,
      "model_forward_time": 1.8708505630493164,
      "step": 64753
    },
    {
      "epoch": 6.17532730102539e-05,
      "step": 64753,
      "training_step_time": 7.611807107925415
    },
    {
      "epoch": 6.175422668457032e-05,
      "model_forward_time": 1.8742311000823975,
      "step": 64754
    },
    {
      "epoch": 6.175422668457032e-05,
      "step": 64754,
      "training_step_time": 7.328042507171631
    },
    {
      "epoch": 6.175518035888671e-05,
      "model_forward_time": 1.881809949874878,
      "step": 64755
    },
    {
      "epoch": 6.175518035888671e-05,
      "step": 64755,
      "training_step_time": 7.541527509689331
    },
    {
      "epoch": 6.175613403320313e-05,
      "model_forward_time": 1.8794970512390137,
      "step": 64756
    },
    {
      "epoch": 6.175613403320313e-05,
      "step": 64756,
      "training_step_time": 7.4597272872924805
    },
    {
      "epoch": 6.175708770751954e-05,
      "model_forward_time": 1.8691542148590088,
      "step": 64757
    },
    {
      "epoch": 6.175708770751954e-05,
      "step": 64757,
      "training_step_time": 7.5635809898376465
    },
    {
      "epoch": 6.175804138183593e-05,
      "model_forward_time": 1.871413230895996,
      "step": 64758
    },
    {
      "epoch": 6.175804138183593e-05,
      "step": 64758,
      "training_step_time": 7.66605019569397
    },
    {
      "epoch": 6.175899505615234e-05,
      "model_forward_time": 1.8814282417297363,
      "step": 64759
    },
    {
      "epoch": 6.175899505615234e-05,
      "step": 64759,
      "training_step_time": 8.709283590316772
    },
    {
      "epoch": 6.175994873046875e-05,
      "grad_norm": 0.09240137785673141,
      "learning_rate": 3.0280785815022727e-06,
      "loss": 0.0522,
      "step": 64760
    },
    {
      "epoch": 6.175994873046875e-05,
      "model_forward_time": 2.400118350982666,
      "step": 64760
    },
    {
      "dynamics_loss_avg": 0.0473077729344368,
      "epoch": 6.175994873046875e-05,
      "step": 64760
    },
    {
      "action_loss_avg": 0.001615535983000882,
      "epoch": 6.175994873046875e-05,
      "step": 64760
    },
    {
      "epoch": 6.175994873046875e-05,
      "step": 64760,
      "training_step_time": 8.604789733886719
    },
    {
      "epoch": 6.176090240478515e-05,
      "model_forward_time": 1.8830687999725342,
      "step": 64761
    },
    {
      "epoch": 6.176090240478515e-05,
      "step": 64761,
      "training_step_time": 9.522913694381714
    },
    {
      "epoch": 6.176185607910156e-05,
      "model_forward_time": 1.8723249435424805,
      "step": 64762
    },
    {
      "epoch": 6.176185607910156e-05,
      "step": 64762,
      "training_step_time": 7.649306297302246
    },
    {
      "epoch": 6.176280975341797e-05,
      "model_forward_time": 1.880239486694336,
      "step": 64763
    },
    {
      "epoch": 6.176280975341797e-05,
      "step": 64763,
      "training_step_time": 7.3946897983551025
    },
    {
      "epoch": 6.176376342773437e-05,
      "model_forward_time": 1.8884010314941406,
      "step": 64764
    },
    {
      "epoch": 6.176376342773437e-05,
      "step": 64764,
      "training_step_time": 8.263210535049438
    },
    {
      "epoch": 6.176471710205078e-05,
      "model_forward_time": 1.8765537738800049,
      "step": 64765
    },
    {
      "epoch": 6.176471710205078e-05,
      "step": 64765,
      "training_step_time": 7.524561643600464
    },
    {
      "epoch": 6.176567077636719e-05,
      "model_forward_time": 1.8816986083984375,
      "step": 64766
    },
    {
      "epoch": 6.176567077636719e-05,
      "step": 64766,
      "training_step_time": 7.550225257873535
    },
    {
      "epoch": 6.176662445068359e-05,
      "model_forward_time": 1.8759434223175049,
      "step": 64767
    },
    {
      "epoch": 6.176662445068359e-05,
      "step": 64767,
      "training_step_time": 8.7126624584198
    },
    {
      "epoch": 6.1767578125e-05,
      "model_forward_time": 1.867919683456421,
      "step": 64768
    },
    {
      "epoch": 6.1767578125e-05,
      "step": 64768,
      "training_step_time": 8.800077199935913
    },
    {
      "epoch": 6.176853179931641e-05,
      "model_forward_time": 2.4800212383270264,
      "step": 64769
    },
    {
      "epoch": 6.176853179931641e-05,
      "step": 64769,
      "training_step_time": 8.567385911941528
    },
    {
      "epoch": 6.176948547363281e-05,
      "grad_norm": 0.1338156908750534,
      "learning_rate": 3.0265592405205546e-06,
      "loss": 0.0595,
      "step": 64770
    },
    {
      "epoch": 6.176948547363281e-05,
      "model_forward_time": 1.8824121952056885,
      "step": 64770
    },
    {
      "dynamics_loss_avg": 0.0548942930996418,
      "epoch": 6.176948547363281e-05,
      "step": 64770
    },
    {
      "action_loss_avg": 0.0010259020287776365,
      "epoch": 6.176948547363281e-05,
      "step": 64770
    },
    {
      "epoch": 6.176948547363281e-05,
      "step": 64770,
      "training_step_time": 8.786429643630981
    },
    {
      "epoch": 6.177043914794922e-05,
      "model_forward_time": 1.8853859901428223,
      "step": 64771
    },
    {
      "epoch": 6.177043914794922e-05,
      "step": 64771,
      "training_step_time": 8.359020948410034
    },
    {
      "epoch": 6.177139282226563e-05,
      "model_forward_time": 1.4821476936340332,
      "step": 64772
    },
    {
      "epoch": 6.177139282226563e-05,
      "step": 64772,
      "training_step_time": 7.263154029846191
    },
    {
      "epoch": 6.177234649658203e-05,
      "model_forward_time": 1.8708076477050781,
      "step": 64773
    },
    {
      "epoch": 6.177234649658203e-05,
      "step": 64773,
      "training_step_time": 8.538537502288818
    },
    {
      "epoch": 6.177330017089844e-05,
      "model_forward_time": 1.8810420036315918,
      "step": 64774
    },
    {
      "epoch": 6.177330017089844e-05,
      "step": 64774,
      "training_step_time": 7.492533445358276
    },
    {
      "epoch": 6.177425384521485e-05,
      "model_forward_time": 1.882335901260376,
      "step": 64775
    },
    {
      "epoch": 6.177425384521485e-05,
      "step": 64775,
      "training_step_time": 7.567164421081543
    },
    {
      "epoch": 6.177520751953125e-05,
      "model_forward_time": 1.9064419269561768,
      "step": 64776
    },
    {
      "epoch": 6.177520751953125e-05,
      "step": 64776,
      "training_step_time": 7.624048233032227
    },
    {
      "epoch": 6.177616119384766e-05,
      "model_forward_time": 1.8580076694488525,
      "step": 64777
    },
    {
      "epoch": 6.177616119384766e-05,
      "step": 64777,
      "training_step_time": 7.564483642578125
    },
    {
      "epoch": 6.177711486816407e-05,
      "model_forward_time": 2.4282498359680176,
      "step": 64778
    },
    {
      "epoch": 6.177711486816407e-05,
      "step": 64778,
      "training_step_time": 8.258109331130981
    },
    {
      "epoch": 6.177806854248047e-05,
      "model_forward_time": 1.879504680633545,
      "step": 64779
    },
    {
      "epoch": 6.177806854248047e-05,
      "step": 64779,
      "training_step_time": 8.36523175239563
    },
    {
      "epoch": 6.177902221679688e-05,
      "grad_norm": 0.07400864362716675,
      "learning_rate": 3.025040115351352e-06,
      "loss": 0.0542,
      "step": 64780
    },
    {
      "epoch": 6.177902221679688e-05,
      "model_forward_time": 1.87148118019104,
      "step": 64780
    },
    {
      "dynamics_loss_avg": 0.05054635219275951,
      "epoch": 6.177902221679688e-05,
      "step": 64780
    },
    {
      "action_loss_avg": 0.001573647529585287,
      "epoch": 6.177902221679688e-05,
      "step": 64780
    },
    {
      "epoch": 6.177902221679688e-05,
      "step": 64780,
      "training_step_time": 8.762387752532959
    },
    {
      "epoch": 6.177997589111329e-05,
      "model_forward_time": 1.8852031230926514,
      "step": 64781
    },
    {
      "epoch": 6.177997589111329e-05,
      "step": 64781,
      "training_step_time": 8.140515327453613
    },
    {
      "epoch": 6.178092956542968e-05,
      "model_forward_time": 1.8872501850128174,
      "step": 64782
    },
    {
      "epoch": 6.178092956542968e-05,
      "step": 64782,
      "training_step_time": 7.499371767044067
    },
    {
      "epoch": 6.17818832397461e-05,
      "model_forward_time": 1.8536319732666016,
      "step": 64783
    },
    {
      "epoch": 6.17818832397461e-05,
      "step": 64783,
      "training_step_time": 7.597200155258179
    },
    {
      "epoch": 6.17828369140625e-05,
      "model_forward_time": 1.8802945613861084,
      "step": 64784
    },
    {
      "epoch": 6.17828369140625e-05,
      "step": 64784,
      "training_step_time": 8.37604022026062
    },
    {
      "epoch": 6.17837905883789e-05,
      "model_forward_time": 1.8898413181304932,
      "step": 64785
    },
    {
      "epoch": 6.17837905883789e-05,
      "step": 64785,
      "training_step_time": 7.721136808395386
    },
    {
      "epoch": 6.178474426269531e-05,
      "model_forward_time": 1.8651466369628906,
      "step": 64786
    },
    {
      "epoch": 6.178474426269531e-05,
      "step": 64786,
      "training_step_time": 7.703753232955933
    },
    {
      "epoch": 6.178569793701172e-05,
      "model_forward_time": 1.8832604885101318,
      "step": 64787
    },
    {
      "epoch": 6.178569793701172e-05,
      "step": 64787,
      "training_step_time": 8.416613340377808
    },
    {
      "epoch": 6.178665161132812e-05,
      "model_forward_time": 2.0496506690979004,
      "step": 64788
    },
    {
      "epoch": 6.178665161132812e-05,
      "step": 64788,
      "training_step_time": 9.708710432052612
    },
    {
      "epoch": 6.178760528564453e-05,
      "model_forward_time": 1.8791770935058594,
      "step": 64789
    },
    {
      "epoch": 6.178760528564453e-05,
      "step": 64789,
      "training_step_time": 8.557860136032104
    },
    {
      "epoch": 6.178855895996094e-05,
      "grad_norm": 0.08913881331682205,
      "learning_rate": 3.023521206160794e-06,
      "loss": 0.0554,
      "step": 64790
    },
    {
      "epoch": 6.178855895996094e-05,
      "model_forward_time": 1.4844567775726318,
      "step": 64790
    },
    {
      "dynamics_loss_avg": 0.05203918553888798,
      "epoch": 6.178855895996094e-05,
      "step": 64790
    },
    {
      "action_loss_avg": 0.0014593913074349985,
      "epoch": 6.178855895996094e-05,
      "step": 64790
    },
    {
      "epoch": 6.178855895996094e-05,
      "step": 64790,
      "training_step_time": 7.636881589889526
    },
    {
      "epoch": 6.178951263427734e-05,
      "model_forward_time": 1.8694851398468018,
      "step": 64791
    },
    {
      "epoch": 6.178951263427734e-05,
      "step": 64791,
      "training_step_time": 7.457665681838989
    },
    {
      "epoch": 6.179046630859375e-05,
      "model_forward_time": 1.878431797027588,
      "step": 64792
    },
    {
      "epoch": 6.179046630859375e-05,
      "step": 64792,
      "training_step_time": 7.589900493621826
    },
    {
      "epoch": 6.179141998291016e-05,
      "model_forward_time": 1.8864014148712158,
      "step": 64793
    },
    {
      "epoch": 6.179141998291016e-05,
      "step": 64793,
      "training_step_time": 7.479401588439941
    },
    {
      "epoch": 6.179237365722656e-05,
      "model_forward_time": 1.8656318187713623,
      "step": 64794
    },
    {
      "epoch": 6.179237365722656e-05,
      "step": 64794,
      "training_step_time": 7.655941963195801
    },
    {
      "epoch": 6.179332733154297e-05,
      "model_forward_time": 1.872492790222168,
      "step": 64795
    },
    {
      "epoch": 6.179332733154297e-05,
      "step": 64795,
      "training_step_time": 9.344848155975342
    },
    {
      "epoch": 6.179428100585938e-05,
      "model_forward_time": 1.8682680130004883,
      "step": 64796
    },
    {
      "epoch": 6.179428100585938e-05,
      "step": 64796,
      "training_step_time": 8.35824990272522
    },
    {
      "epoch": 6.179523468017578e-05,
      "model_forward_time": 2.414318561553955,
      "step": 64797
    },
    {
      "epoch": 6.179523468017578e-05,
      "step": 64797,
      "training_step_time": 10.059412956237793
    },
    {
      "epoch": 6.179618835449219e-05,
      "model_forward_time": 1.8848588466644287,
      "step": 64798
    },
    {
      "epoch": 6.179618835449219e-05,
      "step": 64798,
      "training_step_time": 10.169657468795776
    },
    {
      "epoch": 6.17971420288086e-05,
      "model_forward_time": 1.8756985664367676,
      "step": 64799
    },
    {
      "epoch": 6.17971420288086e-05,
      "step": 64799,
      "training_step_time": 7.583647012710571
    },
    {
      "epoch": 6.1798095703125e-05,
      "grad_norm": 0.07732463628053665,
      "learning_rate": 3.0220025131149845e-06,
      "loss": 0.0553,
      "step": 64800
    },
    {
      "epoch": 6.1798095703125e-05,
      "model_forward_time": 1.8694064617156982,
      "step": 64800
    },
    {
      "dynamics_loss_avg": 0.042015129886567595,
      "epoch": 6.1798095703125e-05,
      "step": 64800
    },
    {
      "action_loss_avg": 0.0010881145426537842,
      "epoch": 6.1798095703125e-05,
      "step": 64800
    },
    {
      "epoch": 6.1798095703125e-05,
      "step": 64800,
      "training_step_time": 7.376913785934448
    },
    {
      "epoch": 6.179904937744141e-05,
      "model_forward_time": 1.8769383430480957,
      "step": 64801
    },
    {
      "epoch": 6.179904937744141e-05,
      "step": 64801,
      "training_step_time": 7.597243547439575
    },
    {
      "epoch": 6.180000305175782e-05,
      "model_forward_time": 1.8810882568359375,
      "step": 64802
    },
    {
      "epoch": 6.180000305175782e-05,
      "step": 64802,
      "training_step_time": 7.497679233551025
    },
    {
      "epoch": 6.180095672607422e-05,
      "model_forward_time": 1.8629987239837646,
      "step": 64803
    },
    {
      "epoch": 6.180095672607422e-05,
      "step": 64803,
      "training_step_time": 8.159672737121582
    },
    {
      "epoch": 6.180191040039063e-05,
      "model_forward_time": 1.8861830234527588,
      "step": 64804
    },
    {
      "epoch": 6.180191040039063e-05,
      "step": 64804,
      "training_step_time": 7.678240537643433
    },
    {
      "epoch": 6.180286407470704e-05,
      "model_forward_time": 1.8685920238494873,
      "step": 64805
    },
    {
      "epoch": 6.180286407470704e-05,
      "step": 64805,
      "training_step_time": 7.804349660873413
    },
    {
      "epoch": 6.180381774902344e-05,
      "model_forward_time": 2.4564526081085205,
      "step": 64806
    },
    {
      "epoch": 6.180381774902344e-05,
      "step": 64806,
      "training_step_time": 7.70398473739624
    },
    {
      "epoch": 6.180477142333985e-05,
      "model_forward_time": 1.8779988288879395,
      "step": 64807
    },
    {
      "epoch": 6.180477142333985e-05,
      "step": 64807,
      "training_step_time": 7.701739549636841
    },
    {
      "epoch": 6.180572509765626e-05,
      "model_forward_time": 1.8753831386566162,
      "step": 64808
    },
    {
      "epoch": 6.180572509765626e-05,
      "step": 64808,
      "training_step_time": 7.695866823196411
    },
    {
      "epoch": 6.180667877197265e-05,
      "model_forward_time": 1.8702819347381592,
      "step": 64809
    },
    {
      "epoch": 6.180667877197265e-05,
      "step": 64809,
      "training_step_time": 7.559892416000366
    },
    {
      "epoch": 6.180763244628906e-05,
      "grad_norm": 0.09247591346502304,
      "learning_rate": 3.020484036380007e-06,
      "loss": 0.0553,
      "step": 64810
    },
    {
      "epoch": 6.180763244628906e-05,
      "model_forward_time": 1.8807156085968018,
      "step": 64810
    },
    {
      "dynamics_loss_avg": 0.05307861920446157,
      "epoch": 6.180763244628906e-05,
      "step": 64810
    },
    {
      "action_loss_avg": 0.0011969776591286064,
      "epoch": 6.180763244628906e-05,
      "step": 64810
    },
    {
      "epoch": 6.180763244628906e-05,
      "step": 64810,
      "training_step_time": 7.628186464309692
    },
    {
      "epoch": 6.180858612060548e-05,
      "model_forward_time": 1.8699116706848145,
      "step": 64811
    },
    {
      "epoch": 6.180858612060548e-05,
      "step": 64811,
      "training_step_time": 7.421569347381592
    },
    {
      "epoch": 6.180953979492187e-05,
      "model_forward_time": 1.8768219947814941,
      "step": 64812
    },
    {
      "epoch": 6.180953979492187e-05,
      "step": 64812,
      "training_step_time": 7.622834205627441
    },
    {
      "epoch": 6.181049346923828e-05,
      "model_forward_time": 1.8840000629425049,
      "step": 64813
    },
    {
      "epoch": 6.181049346923828e-05,
      "step": 64813,
      "training_step_time": 8.687727689743042
    },
    {
      "epoch": 6.181144714355468e-05,
      "model_forward_time": 1.868994951248169,
      "step": 64814
    },
    {
      "epoch": 6.181144714355468e-05,
      "step": 64814,
      "training_step_time": 7.760851144790649
    },
    {
      "epoch": 6.181240081787109e-05,
      "model_forward_time": 2.4469852447509766,
      "step": 64815
    },
    {
      "epoch": 6.181240081787109e-05,
      "step": 64815,
      "training_step_time": 9.5536367893219
    },
    {
      "epoch": 6.18133544921875e-05,
      "model_forward_time": 1.8666884899139404,
      "step": 64816
    },
    {
      "epoch": 6.18133544921875e-05,
      "step": 64816,
      "training_step_time": 8.422673463821411
    },
    {
      "epoch": 6.18143081665039e-05,
      "model_forward_time": 1.8765616416931152,
      "step": 64817
    },
    {
      "epoch": 6.18143081665039e-05,
      "step": 64817,
      "training_step_time": 7.586462497711182
    },
    {
      "epoch": 6.181526184082031e-05,
      "model_forward_time": 1.8767268657684326,
      "step": 64818
    },
    {
      "epoch": 6.181526184082031e-05,
      "step": 64818,
      "training_step_time": 7.580732107162476
    },
    {
      "epoch": 6.181621551513672e-05,
      "model_forward_time": 1.8689842224121094,
      "step": 64819
    },
    {
      "epoch": 6.181621551513672e-05,
      "step": 64819,
      "training_step_time": 7.487327337265015
    },
    {
      "epoch": 6.181716918945312e-05,
      "grad_norm": 0.08694548159837723,
      "learning_rate": 3.0189657761219187e-06,
      "loss": 0.0573,
      "step": 64820
    },
    {
      "epoch": 6.181716918945312e-05,
      "model_forward_time": 1.8740735054016113,
      "step": 64820
    },
    {
      "dynamics_loss_avg": 0.038042910024523736,
      "epoch": 6.181716918945312e-05,
      "step": 64820
    },
    {
      "action_loss_avg": 0.0017607207497349008,
      "epoch": 6.181716918945312e-05,
      "step": 64820
    },
    {
      "epoch": 6.181716918945312e-05,
      "step": 64820,
      "training_step_time": 7.174064874649048
    },
    {
      "epoch": 6.181812286376953e-05,
      "model_forward_time": 1.885251760482788,
      "step": 64821
    },
    {
      "epoch": 6.181812286376953e-05,
      "step": 64821,
      "training_step_time": 7.587998628616333
    },
    {
      "epoch": 6.181907653808594e-05,
      "model_forward_time": 1.867414951324463,
      "step": 64822
    },
    {
      "epoch": 6.181907653808594e-05,
      "step": 64822,
      "training_step_time": 8.683087587356567
    },
    {
      "epoch": 6.182003021240234e-05,
      "model_forward_time": 1.8773446083068848,
      "step": 64823
    },
    {
      "epoch": 6.182003021240234e-05,
      "step": 64823,
      "training_step_time": 8.520383596420288
    },
    {
      "epoch": 6.182098388671875e-05,
      "model_forward_time": 1.4749183654785156,
      "step": 64824
    },
    {
      "epoch": 6.182098388671875e-05,
      "step": 64824,
      "training_step_time": 7.699399709701538
    },
    {
      "epoch": 6.182193756103516e-05,
      "model_forward_time": 2.430140972137451,
      "step": 64825
    },
    {
      "epoch": 6.182193756103516e-05,
      "step": 64825,
      "training_step_time": 8.632261037826538
    },
    {
      "epoch": 6.182289123535156e-05,
      "model_forward_time": 1.888075828552246,
      "step": 64826
    },
    {
      "epoch": 6.182289123535156e-05,
      "step": 64826,
      "training_step_time": 7.701587438583374
    },
    {
      "epoch": 6.182384490966797e-05,
      "model_forward_time": 1.8666765689849854,
      "step": 64827
    },
    {
      "epoch": 6.182384490966797e-05,
      "step": 64827,
      "training_step_time": 7.632379770278931
    },
    {
      "epoch": 6.182479858398438e-05,
      "model_forward_time": 1.8813471794128418,
      "step": 64828
    },
    {
      "epoch": 6.182479858398438e-05,
      "step": 64828,
      "training_step_time": 7.97779393196106
    },
    {
      "epoch": 6.182575225830078e-05,
      "model_forward_time": 1.8741888999938965,
      "step": 64829
    },
    {
      "epoch": 6.182575225830078e-05,
      "step": 64829,
      "training_step_time": 7.556317567825317
    },
    {
      "epoch": 6.182670593261719e-05,
      "grad_norm": 0.09623166173696518,
      "learning_rate": 3.017447732506753e-06,
      "loss": 0.0577,
      "step": 64830
    },
    {
      "epoch": 6.182670593261719e-05,
      "model_forward_time": 1.8774521350860596,
      "step": 64830
    },
    {
      "dynamics_loss_avg": 0.04377350565046072,
      "epoch": 6.182670593261719e-05,
      "step": 64830
    },
    {
      "action_loss_avg": 0.0007602432044222951,
      "epoch": 6.182670593261719e-05,
      "step": 64830
    },
    {
      "epoch": 6.182670593261719e-05,
      "step": 64830,
      "training_step_time": 8.324576377868652
    },
    {
      "epoch": 6.18276596069336e-05,
      "model_forward_time": 1.8718090057373047,
      "step": 64831
    },
    {
      "epoch": 6.18276596069336e-05,
      "step": 64831,
      "training_step_time": 8.973167419433594
    },
    {
      "epoch": 6.182861328125e-05,
      "model_forward_time": 1.8696541786193848,
      "step": 64832
    },
    {
      "epoch": 6.182861328125e-05,
      "step": 64832,
      "training_step_time": 9.488611459732056
    },
    {
      "epoch": 6.18295669555664e-05,
      "model_forward_time": 1.8854176998138428,
      "step": 64833
    },
    {
      "epoch": 6.18295669555664e-05,
      "step": 64833,
      "training_step_time": 8.66961145401001
    },
    {
      "epoch": 6.183052062988282e-05,
      "model_forward_time": 2.4005191326141357,
      "step": 64834
    },
    {
      "epoch": 6.183052062988282e-05,
      "step": 64834,
      "training_step_time": 8.840527534484863
    },
    {
      "epoch": 6.183147430419921e-05,
      "model_forward_time": 1.8869690895080566,
      "step": 64835
    },
    {
      "epoch": 6.183147430419921e-05,
      "step": 64835,
      "training_step_time": 8.472516298294067
    },
    {
      "epoch": 6.183242797851562e-05,
      "model_forward_time": 1.8706820011138916,
      "step": 64836
    },
    {
      "epoch": 6.183242797851562e-05,
      "step": 64836,
      "training_step_time": 8.935162782669067
    },
    {
      "epoch": 6.183338165283203e-05,
      "model_forward_time": 1.8771841526031494,
      "step": 64837
    },
    {
      "epoch": 6.183338165283203e-05,
      "step": 64837,
      "training_step_time": 7.799297571182251
    },
    {
      "epoch": 6.183433532714843e-05,
      "model_forward_time": 1.8600292205810547,
      "step": 64838
    },
    {
      "epoch": 6.183433532714843e-05,
      "step": 64838,
      "training_step_time": 7.575518846511841
    },
    {
      "epoch": 6.183528900146484e-05,
      "model_forward_time": 1.8730762004852295,
      "step": 64839
    },
    {
      "epoch": 6.183528900146484e-05,
      "step": 64839,
      "training_step_time": 8.51642894744873
    },
    {
      "epoch": 6.183624267578125e-05,
      "grad_norm": 0.1669502556324005,
      "learning_rate": 3.0159299057005244e-06,
      "loss": 0.0566,
      "step": 64840
    },
    {
      "epoch": 6.183624267578125e-05,
      "model_forward_time": 1.8709356784820557,
      "step": 64840
    },
    {
      "dynamics_loss_avg": 0.04793724473565817,
      "epoch": 6.183624267578125e-05,
      "step": 64840
    },
    {
      "action_loss_avg": 0.0013698891911190004,
      "epoch": 6.183624267578125e-05,
      "step": 64840
    },
    {
      "epoch": 6.183624267578125e-05,
      "step": 64840,
      "training_step_time": 7.6503586769104
    },
    {
      "epoch": 6.183719635009765e-05,
      "model_forward_time": 1.8708958625793457,
      "step": 64841
    },
    {
      "epoch": 6.183719635009765e-05,
      "step": 64841,
      "training_step_time": 8.482882976531982
    },
    {
      "epoch": 6.183815002441406e-05,
      "model_forward_time": 1.8738954067230225,
      "step": 64842
    },
    {
      "epoch": 6.183815002441406e-05,
      "step": 64842,
      "training_step_time": 8.690375328063965
    },
    {
      "epoch": 6.183910369873047e-05,
      "model_forward_time": 2.4426653385162354,
      "step": 64843
    },
    {
      "epoch": 6.183910369873047e-05,
      "step": 64843,
      "training_step_time": 8.340821504592896
    },
    {
      "epoch": 6.184005737304687e-05,
      "model_forward_time": 1.8719532489776611,
      "step": 64844
    },
    {
      "epoch": 6.184005737304687e-05,
      "step": 64844,
      "training_step_time": 7.701368570327759
    },
    {
      "epoch": 6.184101104736328e-05,
      "model_forward_time": 1.8793833255767822,
      "step": 64845
    },
    {
      "epoch": 6.184101104736328e-05,
      "step": 64845,
      "training_step_time": 7.662067890167236
    },
    {
      "epoch": 6.184196472167969e-05,
      "model_forward_time": 1.8731558322906494,
      "step": 64846
    },
    {
      "epoch": 6.184196472167969e-05,
      "step": 64846,
      "training_step_time": 7.4858927726745605
    },
    {
      "epoch": 6.184291839599609e-05,
      "model_forward_time": 1.86948561668396,
      "step": 64847
    },
    {
      "epoch": 6.184291839599609e-05,
      "step": 64847,
      "training_step_time": 7.223546743392944
    },
    {
      "epoch": 6.18438720703125e-05,
      "model_forward_time": 1.8801438808441162,
      "step": 64848
    },
    {
      "epoch": 6.18438720703125e-05,
      "step": 64848,
      "training_step_time": 7.5756871700286865
    },
    {
      "epoch": 6.184482574462891e-05,
      "model_forward_time": 1.870366096496582,
      "step": 64849
    },
    {
      "epoch": 6.184482574462891e-05,
      "step": 64849,
      "training_step_time": 7.651456117630005
    },
    {
      "epoch": 6.184577941894531e-05,
      "grad_norm": 0.12089739739894867,
      "learning_rate": 3.0144122958692177e-06,
      "loss": 0.059,
      "step": 64850
    },
    {
      "epoch": 6.184577941894531e-05,
      "model_forward_time": 1.8766562938690186,
      "step": 64850
    },
    {
      "dynamics_loss_avg": 0.05883397255092859,
      "epoch": 6.184577941894531e-05,
      "step": 64850
    },
    {
      "action_loss_avg": 0.0019067487155552953,
      "epoch": 6.184577941894531e-05,
      "step": 64850
    },
    {
      "epoch": 6.184577941894531e-05,
      "step": 64850,
      "training_step_time": 7.651951789855957
    },
    {
      "epoch": 6.184673309326172e-05,
      "model_forward_time": 1.8706989288330078,
      "step": 64851
    },
    {
      "epoch": 6.184673309326172e-05,
      "step": 64851,
      "training_step_time": 9.112242460250854
    },
    {
      "epoch": 6.184768676757813e-05,
      "model_forward_time": 2.4443817138671875,
      "step": 64852
    },
    {
      "epoch": 6.184768676757813e-05,
      "step": 64852,
      "training_step_time": 8.610648155212402
    },
    {
      "epoch": 6.184864044189453e-05,
      "model_forward_time": 1.7554643154144287,
      "step": 64853
    },
    {
      "epoch": 6.184864044189453e-05,
      "step": 64853,
      "training_step_time": 9.41635537147522
    },
    {
      "epoch": 6.184959411621094e-05,
      "model_forward_time": 1.7598693370819092,
      "step": 64854
    },
    {
      "epoch": 6.184959411621094e-05,
      "step": 64854,
      "training_step_time": 7.664680480957031
    },
    {
      "epoch": 6.185054779052735e-05,
      "model_forward_time": 1.7655067443847656,
      "step": 64855
    },
    {
      "epoch": 6.185054779052735e-05,
      "step": 64855,
      "training_step_time": 7.558520555496216
    },
    {
      "epoch": 6.185150146484374e-05,
      "model_forward_time": 1.7641375064849854,
      "step": 64856
    },
    {
      "epoch": 6.185150146484374e-05,
      "step": 64856,
      "training_step_time": 7.260565996170044
    },
    {
      "epoch": 6.185245513916016e-05,
      "model_forward_time": 1.7589733600616455,
      "step": 64857
    },
    {
      "epoch": 6.185245513916016e-05,
      "step": 64857,
      "training_step_time": 7.615594387054443
    },
    {
      "epoch": 6.185340881347657e-05,
      "model_forward_time": 1.760434865951538,
      "step": 64858
    },
    {
      "epoch": 6.185340881347657e-05,
      "step": 64858,
      "training_step_time": 7.586225271224976
    },
    {
      "epoch": 6.185436248779296e-05,
      "model_forward_time": 1.7703797817230225,
      "step": 64859
    },
    {
      "epoch": 6.185436248779296e-05,
      "step": 64859,
      "training_step_time": 7.587996244430542
    },
    {
      "epoch": 6.185531616210937e-05,
      "grad_norm": 0.10484765470027924,
      "learning_rate": 3.012894903178798e-06,
      "loss": 0.0578,
      "step": 64860
    },
    {
      "epoch": 6.185531616210937e-05,
      "model_forward_time": 1.8227853775024414,
      "step": 64860
    },
    {
      "dynamics_loss_avg": 0.04865988940000534,
      "epoch": 6.185531616210937e-05,
      "step": 64860
    },
    {
      "action_loss_avg": 0.0013142146257450805,
      "epoch": 6.185531616210937e-05,
      "step": 64860
    },
    {
      "epoch": 6.185531616210937e-05,
      "step": 64860,
      "training_step_time": 8.55399489402771
    },
    {
      "epoch": 6.185626983642579e-05,
      "model_forward_time": 2.3412413597106934,
      "step": 64861
    },
    {
      "epoch": 6.185626983642579e-05,
      "step": 64861,
      "training_step_time": 8.440100908279419
    },
    {
      "epoch": 6.185722351074218e-05,
      "model_forward_time": 1.7536349296569824,
      "step": 64862
    },
    {
      "epoch": 6.185722351074218e-05,
      "step": 64862,
      "training_step_time": 7.745481252670288
    },
    {
      "epoch": 6.18581771850586e-05,
      "model_forward_time": 1.700439453125,
      "step": 64863
    },
    {
      "epoch": 6.18581771850586e-05,
      "step": 64863,
      "training_step_time": 8.197484254837036
    },
    {
      "epoch": 6.1859130859375e-05,
      "model_forward_time": 1.7186076641082764,
      "step": 64864
    },
    {
      "epoch": 6.1859130859375e-05,
      "step": 64864,
      "training_step_time": 7.631291151046753
    },
    {
      "epoch": 6.18600845336914e-05,
      "model_forward_time": 1.7046716213226318,
      "step": 64865
    },
    {
      "epoch": 6.18600845336914e-05,
      "step": 64865,
      "training_step_time": 7.258775472640991
    },
    {
      "epoch": 6.186103820800781e-05,
      "model_forward_time": 1.7068746089935303,
      "step": 64866
    },
    {
      "epoch": 6.186103820800781e-05,
      "step": 64866,
      "training_step_time": 7.5949413776397705
    },
    {
      "epoch": 6.186199188232422e-05,
      "model_forward_time": 1.716012954711914,
      "step": 64867
    },
    {
      "epoch": 6.186199188232422e-05,
      "step": 64867,
      "training_step_time": 7.6483659744262695
    },
    {
      "epoch": 6.186294555664062e-05,
      "model_forward_time": 1.7044529914855957,
      "step": 64868
    },
    {
      "epoch": 6.186294555664062e-05,
      "step": 64868,
      "training_step_time": 8.329608678817749
    },
    {
      "epoch": 6.186389923095703e-05,
      "model_forward_time": 1.7624726295471191,
      "step": 64869
    },
    {
      "epoch": 6.186389923095703e-05,
      "step": 64869,
      "training_step_time": 8.334700345993042
    },
    {
      "epoch": 6.186485290527344e-05,
      "grad_norm": 0.14263683557510376,
      "learning_rate": 3.011377727795202e-06,
      "loss": 0.0552,
      "step": 64870
    },
    {
      "epoch": 6.186485290527344e-05,
      "model_forward_time": 1.693176031112671,
      "step": 64870
    },
    {
      "dynamics_loss_avg": 0.06278252862393856,
      "epoch": 6.186485290527344e-05,
      "step": 64870
    },
    {
      "action_loss_avg": 0.0011195363651495427,
      "epoch": 6.186485290527344e-05,
      "step": 64870
    },
    {
      "epoch": 6.186485290527344e-05,
      "step": 64870,
      "training_step_time": 8.65534782409668
    },
    {
      "epoch": 6.186580657958984e-05,
      "model_forward_time": 2.2555911540985107,
      "step": 64871
    },
    {
      "epoch": 6.186580657958984e-05,
      "step": 64871,
      "training_step_time": 8.192551851272583
    },
    {
      "epoch": 6.186676025390625e-05,
      "model_forward_time": 1.7111504077911377,
      "step": 64872
    },
    {
      "epoch": 6.186676025390625e-05,
      "step": 64872,
      "training_step_time": 8.692119598388672
    },
    {
      "epoch": 6.186771392822266e-05,
      "model_forward_time": 1.7099292278289795,
      "step": 64873
    },
    {
      "epoch": 6.186771392822266e-05,
      "step": 64873,
      "training_step_time": 8.194783449172974
    },
    {
      "epoch": 6.186866760253906e-05,
      "model_forward_time": 1.3494257926940918,
      "step": 64874
    },
    {
      "epoch": 6.186866760253906e-05,
      "step": 64874,
      "training_step_time": 7.192776441574097
    },
    {
      "epoch": 6.186962127685547e-05,
      "model_forward_time": 1.7095398902893066,
      "step": 64875
    },
    {
      "epoch": 6.186962127685547e-05,
      "step": 64875,
      "training_step_time": 7.3054187297821045
    },
    {
      "epoch": 6.187057495117188e-05,
      "model_forward_time": 1.7069830894470215,
      "step": 64876
    },
    {
      "epoch": 6.187057495117188e-05,
      "step": 64876,
      "training_step_time": 7.542524576187134
    },
    {
      "epoch": 6.187152862548828e-05,
      "model_forward_time": 1.6894946098327637,
      "step": 64877
    },
    {
      "epoch": 6.187152862548828e-05,
      "step": 64877,
      "training_step_time": 7.577035427093506
    },
    {
      "epoch": 6.187248229980469e-05,
      "model_forward_time": 1.74993896484375,
      "step": 64878
    },
    {
      "epoch": 6.187248229980469e-05,
      "step": 64878,
      "training_step_time": 9.532135963439941
    },
    {
      "epoch": 6.18734359741211e-05,
      "model_forward_time": 1.7060449123382568,
      "step": 64879
    },
    {
      "epoch": 6.18734359741211e-05,
      "step": 64879,
      "training_step_time": 8.54564905166626
    },
    {
      "epoch": 6.18743896484375e-05,
      "grad_norm": 0.13724154233932495,
      "learning_rate": 3.0098607698843495e-06,
      "loss": 0.0572,
      "step": 64880
    },
    {
      "epoch": 6.18743896484375e-05,
      "model_forward_time": 2.232774496078491,
      "step": 64880
    },
    {
      "dynamics_loss_avg": 0.056547543685883286,
      "epoch": 6.18743896484375e-05,
      "step": 64880
    },
    {
      "action_loss_avg": 0.0015691645443439485,
      "epoch": 6.18743896484375e-05,
      "step": 64880
    },
    {
      "epoch": 6.18743896484375e-05,
      "step": 64880,
      "training_step_time": 7.703246593475342
    },
    {
      "epoch": 6.18753433227539e-05,
      "model_forward_time": 1.7145216464996338,
      "step": 64881
    },
    {
      "epoch": 6.18753433227539e-05,
      "step": 64881,
      "training_step_time": 10.209784269332886
    },
    {
      "epoch": 6.187629699707032e-05,
      "model_forward_time": 1.7091484069824219,
      "step": 64882
    },
    {
      "epoch": 6.187629699707032e-05,
      "step": 64882,
      "training_step_time": 7.5950610637664795
    },
    {
      "epoch": 6.187725067138671e-05,
      "model_forward_time": 1.7132630348205566,
      "step": 64883
    },
    {
      "epoch": 6.187725067138671e-05,
      "step": 64883,
      "training_step_time": 7.493812322616577
    },
    {
      "epoch": 6.187820434570313e-05,
      "model_forward_time": 1.71811842918396,
      "step": 64884
    },
    {
      "epoch": 6.187820434570313e-05,
      "step": 64884,
      "training_step_time": 7.240188121795654
    },
    {
      "epoch": 6.187915802001954e-05,
      "model_forward_time": 1.7100672721862793,
      "step": 64885
    },
    {
      "epoch": 6.187915802001954e-05,
      "step": 64885,
      "training_step_time": 8.559943437576294
    },
    {
      "epoch": 6.188011169433593e-05,
      "model_forward_time": 1.709409475326538,
      "step": 64886
    },
    {
      "epoch": 6.188011169433593e-05,
      "step": 64886,
      "training_step_time": 7.531091928482056
    },
    {
      "epoch": 6.188106536865234e-05,
      "model_forward_time": 1.7363495826721191,
      "step": 64887
    },
    {
      "epoch": 6.188106536865234e-05,
      "step": 64887,
      "training_step_time": 7.716555595397949
    },
    {
      "epoch": 6.188201904296875e-05,
      "model_forward_time": 1.7182111740112305,
      "step": 64888
    },
    {
      "epoch": 6.188201904296875e-05,
      "step": 64888,
      "training_step_time": 7.647616863250732
    },
    {
      "epoch": 6.188297271728515e-05,
      "model_forward_time": 2.266211986541748,
      "step": 64889
    },
    {
      "epoch": 6.188297271728515e-05,
      "step": 64889,
      "training_step_time": 8.302205324172974
    },
    {
      "epoch": 6.188392639160156e-05,
      "grad_norm": 0.15301582217216492,
      "learning_rate": 3.0083440296121312e-06,
      "loss": 0.0567,
      "step": 64890
    },
    {
      "epoch": 6.188392639160156e-05,
      "model_forward_time": 1.7131431102752686,
      "step": 64890
    },
    {
      "dynamics_loss_avg": 0.05778859164565801,
      "epoch": 6.188392639160156e-05,
      "step": 64890
    },
    {
      "action_loss_avg": 0.00159575728175696,
      "epoch": 6.188392639160156e-05,
      "step": 64890
    },
    {
      "epoch": 6.188392639160156e-05,
      "step": 64890,
      "training_step_time": 7.670785903930664
    },
    {
      "epoch": 6.188488006591797e-05,
      "model_forward_time": 1.7122647762298584,
      "step": 64891
    },
    {
      "epoch": 6.188488006591797e-05,
      "step": 64891,
      "training_step_time": 8.66188669204712
    },
    {
      "epoch": 6.188583374023437e-05,
      "model_forward_time": 1.7097184658050537,
      "step": 64892
    },
    {
      "epoch": 6.188583374023437e-05,
      "step": 64892,
      "training_step_time": 7.528710603713989
    },
    {
      "epoch": 6.188678741455078e-05,
      "model_forward_time": 1.713517189025879,
      "step": 64893
    },
    {
      "epoch": 6.188678741455078e-05,
      "step": 64893,
      "training_step_time": 7.253243684768677
    },
    {
      "epoch": 6.188774108886719e-05,
      "model_forward_time": 1.7113630771636963,
      "step": 64894
    },
    {
      "epoch": 6.188774108886719e-05,
      "step": 64894,
      "training_step_time": 7.605441331863403
    },
    {
      "epoch": 6.188869476318359e-05,
      "model_forward_time": 1.7059471607208252,
      "step": 64895
    },
    {
      "epoch": 6.188869476318359e-05,
      "step": 64895,
      "training_step_time": 7.515390157699585
    },
    {
      "epoch": 6.18896484375e-05,
      "model_forward_time": 1.7159485816955566,
      "step": 64896
    },
    {
      "epoch": 6.18896484375e-05,
      "step": 64896,
      "training_step_time": 7.7978434562683105
    },
    {
      "epoch": 6.189060211181641e-05,
      "model_forward_time": 1.7083466053009033,
      "step": 64897
    },
    {
      "epoch": 6.189060211181641e-05,
      "step": 64897,
      "training_step_time": 8.725396394729614
    },
    {
      "epoch": 6.189155578613281e-05,
      "model_forward_time": 2.2725088596343994,
      "step": 64898
    },
    {
      "epoch": 6.189155578613281e-05,
      "step": 64898,
      "training_step_time": 9.284448623657227
    },
    {
      "epoch": 6.189250946044922e-05,
      "model_forward_time": 1.7010853290557861,
      "step": 64899
    },
    {
      "epoch": 6.189250946044922e-05,
      "step": 64899,
      "training_step_time": 8.761895895004272
    },
    {
      "epoch": 6.189346313476563e-05,
      "grad_norm": 0.10841736942529678,
      "learning_rate": 3.006827507144414e-06,
      "loss": 0.0566,
      "step": 64900
    },
    {
      "epoch": 6.189346313476563e-05,
      "model_forward_time": 1.714930772781372,
      "step": 64900
    },
    {
      "dynamics_loss_avg": 0.04725339263677597,
      "epoch": 6.189346313476563e-05,
      "step": 64900
    },
    {
      "action_loss_avg": 0.0014216506097000092,
      "epoch": 6.189346313476563e-05,
      "step": 64900
    },
    {
      "epoch": 6.189346313476563e-05,
      "step": 64900,
      "training_step_time": 7.6166088581085205
    },
    {
      "epoch": 6.189441680908203e-05,
      "model_forward_time": 1.7077915668487549,
      "step": 64901
    },
    {
      "epoch": 6.189441680908203e-05,
      "step": 64901,
      "training_step_time": 7.599335193634033
    },
    {
      "epoch": 6.189537048339844e-05,
      "model_forward_time": 1.715052843093872,
      "step": 64902
    },
    {
      "epoch": 6.189537048339844e-05,
      "step": 64902,
      "training_step_time": 7.934294700622559
    },
    {
      "epoch": 6.189632415771485e-05,
      "model_forward_time": 1.7245030403137207,
      "step": 64903
    },
    {
      "epoch": 6.189632415771485e-05,
      "step": 64903,
      "training_step_time": 7.610094308853149
    },
    {
      "epoch": 6.189727783203125e-05,
      "model_forward_time": 1.7083470821380615,
      "step": 64904
    },
    {
      "epoch": 6.189727783203125e-05,
      "step": 64904,
      "training_step_time": 8.306059122085571
    },
    {
      "epoch": 6.189823150634766e-05,
      "model_forward_time": 1.7129192352294922,
      "step": 64905
    },
    {
      "epoch": 6.189823150634766e-05,
      "step": 64905,
      "training_step_time": 7.771518707275391
    },
    {
      "epoch": 6.189918518066407e-05,
      "model_forward_time": 1.7574660778045654,
      "step": 64906
    },
    {
      "epoch": 6.189918518066407e-05,
      "step": 64906,
      "training_step_time": 7.650655031204224
    },
    {
      "epoch": 6.190013885498047e-05,
      "model_forward_time": 2.2583425045013428,
      "step": 64907
    },
    {
      "epoch": 6.190013885498047e-05,
      "step": 64907,
      "training_step_time": 8.212084770202637
    },
    {
      "epoch": 6.190109252929688e-05,
      "model_forward_time": 1.7060954570770264,
      "step": 64908
    },
    {
      "epoch": 6.190109252929688e-05,
      "step": 64908,
      "training_step_time": 7.660116195678711
    },
    {
      "epoch": 6.190204620361329e-05,
      "model_forward_time": 1.7246909141540527,
      "step": 64909
    },
    {
      "epoch": 6.190204620361329e-05,
      "step": 64909,
      "training_step_time": 7.582989931106567
    },
    {
      "epoch": 6.190299987792968e-05,
      "grad_norm": 0.1205642968416214,
      "learning_rate": 3.0053112026470443e-06,
      "loss": 0.0523,
      "step": 64910
    },
    {
      "epoch": 6.190299987792968e-05,
      "model_forward_time": 1.721848964691162,
      "step": 64910
    },
    {
      "dynamics_loss_avg": 0.052328201942145824,
      "epoch": 6.190299987792968e-05,
      "step": 64910
    },
    {
      "action_loss_avg": 0.002238133572973311,
      "epoch": 6.190299987792968e-05,
      "step": 64910
    },
    {
      "epoch": 6.190299987792968e-05,
      "step": 64910,
      "training_step_time": 7.627845764160156
    },
    {
      "epoch": 6.19039535522461e-05,
      "model_forward_time": 1.7042038440704346,
      "step": 64911
    },
    {
      "epoch": 6.19039535522461e-05,
      "step": 64911,
      "training_step_time": 7.540257692337036
    },
    {
      "epoch": 6.19049072265625e-05,
      "model_forward_time": 1.7191016674041748,
      "step": 64912
    },
    {
      "epoch": 6.19049072265625e-05,
      "step": 64912,
      "training_step_time": 7.462795972824097
    },
    {
      "epoch": 6.19058609008789e-05,
      "model_forward_time": 1.7067227363586426,
      "step": 64913
    },
    {
      "epoch": 6.19058609008789e-05,
      "step": 64913,
      "training_step_time": 7.648470878601074
    },
    {
      "epoch": 6.190681457519531e-05,
      "model_forward_time": 1.7059903144836426,
      "step": 64914
    },
    {
      "epoch": 6.190681457519531e-05,
      "step": 64914,
      "training_step_time": 8.334749221801758
    },
    {
      "epoch": 6.190776824951172e-05,
      "model_forward_time": 1.7572150230407715,
      "step": 64915
    },
    {
      "epoch": 6.190776824951172e-05,
      "step": 64915,
      "training_step_time": 7.538729667663574
    },
    {
      "epoch": 6.190872192382812e-05,
      "model_forward_time": 1.7092006206512451,
      "step": 64916
    },
    {
      "epoch": 6.190872192382812e-05,
      "step": 64916,
      "training_step_time": 8.148104429244995
    },
    {
      "epoch": 6.190967559814453e-05,
      "model_forward_time": 1.703479528427124,
      "step": 64917
    },
    {
      "epoch": 6.190967559814453e-05,
      "step": 64917,
      "training_step_time": 8.6843900680542
    },
    {
      "epoch": 6.191062927246094e-05,
      "model_forward_time": 1.7051396369934082,
      "step": 64918
    },
    {
      "epoch": 6.191062927246094e-05,
      "step": 64918,
      "training_step_time": 9.703856706619263
    },
    {
      "epoch": 6.191158294677734e-05,
      "model_forward_time": 1.7236483097076416,
      "step": 64919
    },
    {
      "epoch": 6.191158294677734e-05,
      "step": 64919,
      "training_step_time": 7.673402547836304
    },
    {
      "epoch": 6.191253662109375e-05,
      "grad_norm": 0.11681776493787766,
      "learning_rate": 3.0037951162858436e-06,
      "loss": 0.0554,
      "step": 64920
    },
    {
      "epoch": 6.191253662109375e-05,
      "model_forward_time": 1.7055323123931885,
      "step": 64920
    },
    {
      "dynamics_loss_avg": 0.06517234835773707,
      "epoch": 6.191253662109375e-05,
      "step": 64920
    },
    {
      "action_loss_avg": 0.0019159579504048451,
      "epoch": 6.191253662109375e-05,
      "step": 64920
    },
    {
      "epoch": 6.191253662109375e-05,
      "step": 64920,
      "training_step_time": 7.527264833450317
    },
    {
      "epoch": 6.191349029541016e-05,
      "model_forward_time": 1.7064099311828613,
      "step": 64921
    },
    {
      "epoch": 6.191349029541016e-05,
      "step": 64921,
      "training_step_time": 7.383841037750244
    },
    {
      "epoch": 6.191444396972656e-05,
      "model_forward_time": 1.70939040184021,
      "step": 64922
    },
    {
      "epoch": 6.191444396972656e-05,
      "step": 64922,
      "training_step_time": 8.595658779144287
    },
    {
      "epoch": 6.191539764404297e-05,
      "model_forward_time": 1.6819696426391602,
      "step": 64923
    },
    {
      "epoch": 6.191539764404297e-05,
      "step": 64923,
      "training_step_time": 9.620301485061646
    },
    {
      "epoch": 6.191635131835938e-05,
      "model_forward_time": 1.7230873107910156,
      "step": 64924
    },
    {
      "epoch": 6.191635131835938e-05,
      "step": 64924,
      "training_step_time": 7.621366262435913
    },
    {
      "epoch": 6.191730499267578e-05,
      "model_forward_time": 1.72078537940979,
      "step": 64925
    },
    {
      "epoch": 6.191730499267578e-05,
      "step": 64925,
      "training_step_time": 7.825046062469482
    },
    {
      "epoch": 6.191825866699219e-05,
      "model_forward_time": 2.231846332550049,
      "step": 64926
    },
    {
      "epoch": 6.191825866699219e-05,
      "step": 64926,
      "training_step_time": 7.800215005874634
    },
    {
      "epoch": 6.19192123413086e-05,
      "model_forward_time": 1.7124276161193848,
      "step": 64927
    },
    {
      "epoch": 6.19192123413086e-05,
      "step": 64927,
      "training_step_time": 9.603384971618652
    },
    {
      "epoch": 6.1920166015625e-05,
      "model_forward_time": 1.706620454788208,
      "step": 64928
    },
    {
      "epoch": 6.1920166015625e-05,
      "step": 64928,
      "training_step_time": 7.661781549453735
    },
    {
      "epoch": 6.192111968994141e-05,
      "model_forward_time": 1.7110769748687744,
      "step": 64929
    },
    {
      "epoch": 6.192111968994141e-05,
      "step": 64929,
      "training_step_time": 7.6490654945373535
    },
    {
      "epoch": 6.192207336425782e-05,
      "grad_norm": 0.08911769092082977,
      "learning_rate": 3.0022792482266073e-06,
      "loss": 0.0561,
      "step": 64930
    },
    {
      "epoch": 6.192207336425782e-05,
      "model_forward_time": 1.7046122550964355,
      "step": 64930
    },
    {
      "dynamics_loss_avg": 0.053039741888642314,
      "epoch": 6.192207336425782e-05,
      "step": 64930
    },
    {
      "action_loss_avg": 0.001375527336494997,
      "epoch": 6.192207336425782e-05,
      "step": 64930
    },
    {
      "epoch": 6.192207336425782e-05,
      "step": 64930,
      "training_step_time": 7.309849500656128
    },
    {
      "epoch": 6.192302703857422e-05,
      "model_forward_time": 1.7080700397491455,
      "step": 64931
    },
    {
      "epoch": 6.192302703857422e-05,
      "step": 64931,
      "training_step_time": 7.724704742431641
    },
    {
      "epoch": 6.192398071289063e-05,
      "model_forward_time": 1.7062430381774902,
      "step": 64932
    },
    {
      "epoch": 6.192398071289063e-05,
      "step": 64932,
      "training_step_time": 7.96223783493042
    },
    {
      "epoch": 6.192493438720704e-05,
      "model_forward_time": 1.7081694602966309,
      "step": 64933
    },
    {
      "epoch": 6.192493438720704e-05,
      "step": 64933,
      "training_step_time": 8.606041193008423
    },
    {
      "epoch": 6.192588806152344e-05,
      "model_forward_time": 1.7035865783691406,
      "step": 64934
    },
    {
      "epoch": 6.192588806152344e-05,
      "step": 64934,
      "training_step_time": 7.735476970672607
    },
    {
      "epoch": 6.192684173583985e-05,
      "model_forward_time": 2.2765846252441406,
      "step": 64935
    },
    {
      "epoch": 6.192684173583985e-05,
      "step": 64935,
      "training_step_time": 7.744102478027344
    },
    {
      "epoch": 6.192779541015626e-05,
      "model_forward_time": 1.7032592296600342,
      "step": 64936
    },
    {
      "epoch": 6.192779541015626e-05,
      "step": 64936,
      "training_step_time": 9.197223901748657
    },
    {
      "epoch": 6.192874908447265e-05,
      "model_forward_time": 1.6957972049713135,
      "step": 64937
    },
    {
      "epoch": 6.192874908447265e-05,
      "step": 64937,
      "training_step_time": 8.544256210327148
    },
    {
      "epoch": 6.192970275878906e-05,
      "model_forward_time": 1.696209192276001,
      "step": 64938
    },
    {
      "epoch": 6.192970275878906e-05,
      "step": 64938,
      "training_step_time": 7.723571062088013
    },
    {
      "epoch": 6.193065643310546e-05,
      "model_forward_time": 1.7169280052185059,
      "step": 64939
    },
    {
      "epoch": 6.193065643310546e-05,
      "step": 64939,
      "training_step_time": 7.304281234741211
    },
    {
      "epoch": 6.193161010742187e-05,
      "grad_norm": 0.11453863978385925,
      "learning_rate": 3.000763598635109e-06,
      "loss": 0.0565,
      "step": 64940
    },
    {
      "epoch": 6.193161010742187e-05,
      "model_forward_time": 1.6972236633300781,
      "step": 64940
    },
    {
      "dynamics_loss_avg": 0.06788719706237316,
      "epoch": 6.193161010742187e-05,
      "step": 64940
    },
    {
      "action_loss_avg": 0.002231497553293593,
      "epoch": 6.193161010742187e-05,
      "step": 64940
    },
    {
      "epoch": 6.193161010742187e-05,
      "step": 64940,
      "training_step_time": 7.642582893371582
    },
    {
      "epoch": 6.193256378173828e-05,
      "model_forward_time": 1.7210688591003418,
      "step": 64941
    },
    {
      "epoch": 6.193256378173828e-05,
      "step": 64941,
      "training_step_time": 8.991414546966553
    },
    {
      "epoch": 6.193351745605468e-05,
      "model_forward_time": 1.6972377300262451,
      "step": 64942
    },
    {
      "epoch": 6.193351745605468e-05,
      "step": 64942,
      "training_step_time": 7.614872932434082
    },
    {
      "epoch": 6.193447113037109e-05,
      "model_forward_time": 1.758368730545044,
      "step": 64943
    },
    {
      "epoch": 6.193447113037109e-05,
      "step": 64943,
      "training_step_time": 8.435811519622803
    },
    {
      "epoch": 6.19354248046875e-05,
      "model_forward_time": 2.2503812313079834,
      "step": 64944
    },
    {
      "epoch": 6.19354248046875e-05,
      "step": 64944,
      "training_step_time": 10.452574968338013
    },
    {
      "epoch": 6.19363784790039e-05,
      "model_forward_time": 1.7039711475372314,
      "step": 64945
    },
    {
      "epoch": 6.19363784790039e-05,
      "step": 64945,
      "training_step_time": 8.599583625793457
    },
    {
      "epoch": 6.193733215332031e-05,
      "model_forward_time": 1.7124714851379395,
      "step": 64946
    },
    {
      "epoch": 6.193733215332031e-05,
      "step": 64946,
      "training_step_time": 7.665097236633301
    },
    {
      "epoch": 6.193828582763672e-05,
      "model_forward_time": 1.7164461612701416,
      "step": 64947
    },
    {
      "epoch": 6.193828582763672e-05,
      "step": 64947,
      "training_step_time": 8.751907348632812
    },
    {
      "epoch": 6.193923950195312e-05,
      "model_forward_time": 1.7048847675323486,
      "step": 64948
    },
    {
      "epoch": 6.193923950195312e-05,
      "step": 64948,
      "training_step_time": 7.582225799560547
    },
    {
      "epoch": 6.194019317626953e-05,
      "model_forward_time": 1.7076067924499512,
      "step": 64949
    },
    {
      "epoch": 6.194019317626953e-05,
      "step": 64949,
      "training_step_time": 8.057136297225952
    },
    {
      "epoch": 6.194114685058594e-05,
      "grad_norm": 0.07966411858797073,
      "learning_rate": 2.9992481676770977e-06,
      "loss": 0.0582,
      "step": 64950
    },
    {
      "epoch": 6.194114685058594e-05,
      "model_forward_time": 1.7155888080596924,
      "step": 64950
    },
    {
      "dynamics_loss_avg": 0.054399534687399866,
      "epoch": 6.194114685058594e-05,
      "step": 64950
    },
    {
      "action_loss_avg": 0.0010471599918673747,
      "epoch": 6.194114685058594e-05,
      "step": 64950
    },
    {
      "epoch": 6.194114685058594e-05,
      "step": 64950,
      "training_step_time": 7.627290487289429
    },
    {
      "epoch": 6.194210052490234e-05,
      "model_forward_time": 1.704540729522705,
      "step": 64951
    },
    {
      "epoch": 6.194210052490234e-05,
      "step": 64951,
      "training_step_time": 7.8539087772369385
    },
    {
      "epoch": 6.194305419921875e-05,
      "model_forward_time": 1.756633996963501,
      "step": 64952
    },
    {
      "epoch": 6.194305419921875e-05,
      "step": 64952,
      "training_step_time": 9.290986776351929
    },
    {
      "epoch": 6.194400787353516e-05,
      "model_forward_time": 2.2797904014587402,
      "step": 64953
    },
    {
      "epoch": 6.194400787353516e-05,
      "step": 64953,
      "training_step_time": 7.643283128738403
    },
    {
      "epoch": 6.194496154785156e-05,
      "model_forward_time": 1.6942713260650635,
      "step": 64954
    },
    {
      "epoch": 6.194496154785156e-05,
      "step": 64954,
      "training_step_time": 7.675448894500732
    },
    {
      "epoch": 6.194591522216797e-05,
      "model_forward_time": 1.707632064819336,
      "step": 64955
    },
    {
      "epoch": 6.194591522216797e-05,
      "step": 64955,
      "training_step_time": 9.779481172561646
    },
    {
      "epoch": 6.194686889648438e-05,
      "model_forward_time": 1.7089345455169678,
      "step": 64956
    },
    {
      "epoch": 6.194686889648438e-05,
      "step": 64956,
      "training_step_time": 7.660728693008423
    },
    {
      "epoch": 6.194782257080078e-05,
      "model_forward_time": 1.714470386505127,
      "step": 64957
    },
    {
      "epoch": 6.194782257080078e-05,
      "step": 64957,
      "training_step_time": 8.385422706604004
    },
    {
      "epoch": 6.194877624511719e-05,
      "model_forward_time": 1.7079246044158936,
      "step": 64958
    },
    {
      "epoch": 6.194877624511719e-05,
      "step": 64958,
      "training_step_time": 7.680139541625977
    },
    {
      "epoch": 6.19497299194336e-05,
      "model_forward_time": 1.706502914428711,
      "step": 64959
    },
    {
      "epoch": 6.19497299194336e-05,
      "step": 64959,
      "training_step_time": 7.840060234069824
    },
    {
      "epoch": 6.195068359375e-05,
      "grad_norm": 0.08595119416713715,
      "learning_rate": 2.9977329555182977e-06,
      "loss": 0.0536,
      "step": 64960
    },
    {
      "epoch": 6.195068359375e-05,
      "model_forward_time": 1.6935768127441406,
      "step": 64960
    },
    {
      "dynamics_loss_avg": 0.04542951267212629,
      "epoch": 6.195068359375e-05,
      "step": 64960
    },
    {
      "action_loss_avg": 0.0014072594640310854,
      "epoch": 6.195068359375e-05,
      "step": 64960
    },
    {
      "epoch": 6.195068359375e-05,
      "step": 64960,
      "training_step_time": 7.9634222984313965
    },
    {
      "epoch": 6.19516372680664e-05,
      "model_forward_time": 1.7428181171417236,
      "step": 64961
    },
    {
      "epoch": 6.19516372680664e-05,
      "step": 64961,
      "training_step_time": 8.324388980865479
    },
    {
      "epoch": 6.195259094238282e-05,
      "model_forward_time": 2.2661280632019043,
      "step": 64962
    },
    {
      "epoch": 6.195259094238282e-05,
      "step": 64962,
      "training_step_time": 7.691205263137817
    },
    {
      "epoch": 6.195354461669921e-05,
      "model_forward_time": 1.7038674354553223,
      "step": 64963
    },
    {
      "epoch": 6.195354461669921e-05,
      "step": 64963,
      "training_step_time": 8.408698558807373
    },
    {
      "epoch": 6.195449829101562e-05,
      "model_forward_time": 1.3507559299468994,
      "step": 64964
    },
    {
      "epoch": 6.195449829101562e-05,
      "step": 64964,
      "training_step_time": 8.57872486114502
    },
    {
      "epoch": 6.195545196533203e-05,
      "model_forward_time": 1.7034862041473389,
      "step": 64965
    },
    {
      "epoch": 6.195545196533203e-05,
      "step": 64965,
      "training_step_time": 7.851459980010986
    },
    {
      "epoch": 6.195640563964843e-05,
      "model_forward_time": 1.7147698402404785,
      "step": 64966
    },
    {
      "epoch": 6.195640563964843e-05,
      "step": 64966,
      "training_step_time": 7.705864429473877
    },
    {
      "epoch": 6.195735931396484e-05,
      "model_forward_time": 1.7182767391204834,
      "step": 64967
    },
    {
      "epoch": 6.195735931396484e-05,
      "step": 64967,
      "training_step_time": 7.575729846954346
    },
    {
      "epoch": 6.195831298828125e-05,
      "model_forward_time": 1.7239165306091309,
      "step": 64968
    },
    {
      "epoch": 6.195831298828125e-05,
      "step": 64968,
      "training_step_time": 7.818978548049927
    },
    {
      "epoch": 6.195926666259765e-05,
      "model_forward_time": 1.7085065841674805,
      "step": 64969
    },
    {
      "epoch": 6.195926666259765e-05,
      "step": 64969,
      "training_step_time": 8.036446571350098
    },
    {
      "epoch": 6.196022033691406e-05,
      "grad_norm": 0.1185280904173851,
      "learning_rate": 2.9962179623244127e-06,
      "loss": 0.0563,
      "step": 64970
    },
    {
      "epoch": 6.196022033691406e-05,
      "model_forward_time": 1.7208831310272217,
      "step": 64970
    },
    {
      "dynamics_loss_avg": 0.053383889980614184,
      "epoch": 6.196022033691406e-05,
      "step": 64970
    },
    {
      "action_loss_avg": 0.001797323254868388,
      "epoch": 6.196022033691406e-05,
      "step": 64970
    },
    {
      "epoch": 6.196022033691406e-05,
      "step": 64970,
      "training_step_time": 9.334700345993042
    },
    {
      "epoch": 6.196117401123047e-05,
      "model_forward_time": 1.352585792541504,
      "step": 64971
    },
    {
      "epoch": 6.196117401123047e-05,
      "step": 64971,
      "training_step_time": 9.824338436126709
    },
    {
      "epoch": 6.196212768554687e-05,
      "model_forward_time": 2.2539443969726562,
      "step": 64972
    },
    {
      "epoch": 6.196212768554687e-05,
      "step": 64972,
      "training_step_time": 7.811513185501099
    },
    {
      "epoch": 6.196308135986328e-05,
      "model_forward_time": 1.7151434421539307,
      "step": 64973
    },
    {
      "epoch": 6.196308135986328e-05,
      "step": 64973,
      "training_step_time": 7.748167037963867
    },
    {
      "epoch": 6.196403503417969e-05,
      "model_forward_time": 1.7243366241455078,
      "step": 64974
    },
    {
      "epoch": 6.196403503417969e-05,
      "step": 64974,
      "training_step_time": 7.796219348907471
    },
    {
      "epoch": 6.196498870849609e-05,
      "model_forward_time": 1.6919355392456055,
      "step": 64975
    },
    {
      "epoch": 6.196498870849609e-05,
      "step": 64975,
      "training_step_time": 7.924736976623535
    },
    {
      "epoch": 6.19659423828125e-05,
      "model_forward_time": 1.71244215965271,
      "step": 64976
    },
    {
      "epoch": 6.19659423828125e-05,
      "step": 64976,
      "training_step_time": 7.703182220458984
    },
    {
      "epoch": 6.196689605712891e-05,
      "model_forward_time": 1.72804594039917,
      "step": 64977
    },
    {
      "epoch": 6.196689605712891e-05,
      "step": 64977,
      "training_step_time": 8.635172843933105
    },
    {
      "epoch": 6.196784973144531e-05,
      "model_forward_time": 1.715524435043335,
      "step": 64978
    },
    {
      "epoch": 6.196784973144531e-05,
      "step": 64978,
      "training_step_time": 8.080948114395142
    },
    {
      "epoch": 6.196880340576172e-05,
      "model_forward_time": 1.7162044048309326,
      "step": 64979
    },
    {
      "epoch": 6.196880340576172e-05,
      "step": 64979,
      "training_step_time": 7.827916860580444
    },
    {
      "epoch": 6.196975708007813e-05,
      "grad_norm": 0.10603967308998108,
      "learning_rate": 2.994703188261118e-06,
      "loss": 0.0585,
      "step": 64980
    },
    {
      "epoch": 6.196975708007813e-05,
      "model_forward_time": 1.7637481689453125,
      "step": 64980
    },
    {
      "dynamics_loss_avg": 0.03840514030307531,
      "epoch": 6.196975708007813e-05,
      "step": 64980
    },
    {
      "action_loss_avg": 0.000662664047558792,
      "epoch": 6.196975708007813e-05,
      "step": 64980
    },
    {
      "epoch": 6.196975708007813e-05,
      "step": 64980,
      "training_step_time": 8.4991614818573
    },
    {
      "epoch": 6.197071075439453e-05,
      "model_forward_time": 2.247945547103882,
      "step": 64981
    },
    {
      "epoch": 6.197071075439453e-05,
      "step": 64981,
      "training_step_time": 8.35369086265564
    },
    {
      "epoch": 6.197166442871094e-05,
      "model_forward_time": 1.3492481708526611,
      "step": 64982
    },
    {
      "epoch": 6.197166442871094e-05,
      "step": 64982,
      "training_step_time": 7.855419874191284
    },
    {
      "epoch": 6.197261810302735e-05,
      "model_forward_time": 1.7086470127105713,
      "step": 64983
    },
    {
      "epoch": 6.197261810302735e-05,
      "step": 64983,
      "training_step_time": 8.742423057556152
    },
    {
      "epoch": 6.197357177734374e-05,
      "model_forward_time": 1.7187225818634033,
      "step": 64984
    },
    {
      "epoch": 6.197357177734374e-05,
      "step": 64984,
      "training_step_time": 7.763942241668701
    },
    {
      "epoch": 6.197452545166016e-05,
      "model_forward_time": 1.7147572040557861,
      "step": 64985
    },
    {
      "epoch": 6.197452545166016e-05,
      "step": 64985,
      "training_step_time": 7.768548488616943
    },
    {
      "epoch": 6.197547912597657e-05,
      "model_forward_time": 1.7167415618896484,
      "step": 64986
    },
    {
      "epoch": 6.197547912597657e-05,
      "step": 64986,
      "training_step_time": 7.881856679916382
    },
    {
      "epoch": 6.197643280029296e-05,
      "model_forward_time": 1.7116851806640625,
      "step": 64987
    },
    {
      "epoch": 6.197643280029296e-05,
      "step": 64987,
      "training_step_time": 7.79216742515564
    },
    {
      "epoch": 6.197738647460937e-05,
      "model_forward_time": 1.7154781818389893,
      "step": 64988
    },
    {
      "epoch": 6.197738647460937e-05,
      "step": 64988,
      "training_step_time": 8.868212938308716
    },
    {
      "epoch": 6.197834014892579e-05,
      "model_forward_time": 1.770629644393921,
      "step": 64989
    },
    {
      "epoch": 6.197834014892579e-05,
      "step": 64989,
      "training_step_time": 7.766527891159058
    },
    {
      "epoch": 6.197929382324218e-05,
      "grad_norm": 0.08943469822406769,
      "learning_rate": 2.9931886334940685e-06,
      "loss": 0.0572,
      "step": 64990
    },
    {
      "epoch": 6.197929382324218e-05,
      "model_forward_time": 2.2740299701690674,
      "step": 64990
    },
    {
      "dynamics_loss_avg": 0.054307257011532786,
      "epoch": 6.197929382324218e-05,
      "step": 64990
    },
    {
      "action_loss_avg": 0.001150120425154455,
      "epoch": 6.197929382324218e-05,
      "step": 64990
    },
    {
      "epoch": 6.197929382324218e-05,
      "step": 64990,
      "training_step_time": 7.652263879776001
    },
    {
      "epoch": 6.19802474975586e-05,
      "model_forward_time": 1.7099990844726562,
      "step": 64991
    },
    {
      "epoch": 6.19802474975586e-05,
      "step": 64991,
      "training_step_time": 8.313189029693604
    },
    {
      "epoch": 6.1981201171875e-05,
      "model_forward_time": 1.7290828227996826,
      "step": 64992
    },
    {
      "epoch": 6.1981201171875e-05,
      "step": 64992,
      "training_step_time": 8.726953983306885
    },
    {
      "epoch": 6.19821548461914e-05,
      "model_forward_time": 1.7131750583648682,
      "step": 64993
    },
    {
      "epoch": 6.19821548461914e-05,
      "step": 64993,
      "training_step_time": 7.660268068313599
    },
    {
      "epoch": 6.198310852050781e-05,
      "model_forward_time": 1.711989164352417,
      "step": 64994
    },
    {
      "epoch": 6.198310852050781e-05,
      "step": 64994,
      "training_step_time": 7.650040149688721
    },
    {
      "epoch": 6.198406219482422e-05,
      "model_forward_time": 1.7008683681488037,
      "step": 64995
    },
    {
      "epoch": 6.198406219482422e-05,
      "step": 64995,
      "training_step_time": 8.322249174118042
    },
    {
      "epoch": 6.198501586914062e-05,
      "model_forward_time": 1.7146596908569336,
      "step": 64996
    },
    {
      "epoch": 6.198501586914062e-05,
      "step": 64996,
      "training_step_time": 7.6122801303863525
    },
    {
      "epoch": 6.198596954345703e-05,
      "model_forward_time": 1.7078852653503418,
      "step": 64997
    },
    {
      "epoch": 6.198596954345703e-05,
      "step": 64997,
      "training_step_time": 7.800505638122559
    },
    {
      "epoch": 6.198692321777344e-05,
      "model_forward_time": 1.7881619930267334,
      "step": 64998
    },
    {
      "epoch": 6.198692321777344e-05,
      "step": 64998,
      "training_step_time": 7.906754493713379
    },
    {
      "epoch": 6.198787689208984e-05,
      "model_forward_time": 1.7076449394226074,
      "step": 64999
    },
    {
      "epoch": 6.198787689208984e-05,
      "step": 64999,
      "training_step_time": 10.334962368011475
    },
    {
      "epoch": 6.198883056640625e-05,
      "grad_norm": 0.16541336476802826,
      "learning_rate": 2.9916742981888924e-06,
      "loss": 0.058,
      "step": 65000
    },
    {
      "epoch": 6.198883056640625e-05,
      "model_forward_time": 1.7605156898498535,
      "step": 65000
    },
    {
      "dynamics_loss_avg": 0.050993979908525944,
      "epoch": 6.198883056640625e-05,
      "step": 65000
    },
    {
      "action_loss_avg": 0.0015199575282167643,
      "epoch": 6.198883056640625e-05,
      "step": 65000
    },
    {
      "epoch": 6.198883056640625e-05,
      "step": 65000,
      "training_step_time": 9.773294687271118
    },
    {
      "epoch": 6.198978424072266e-05,
      "model_forward_time": 1.7320997714996338,
      "step": 65001
    },
    {
      "epoch": 6.198978424072266e-05,
      "step": 65001,
      "training_step_time": 10.844568014144897
    },
    {
      "epoch": 6.199073791503906e-05,
      "model_forward_time": 1.737626075744629,
      "step": 65002
    },
    {
      "epoch": 6.199073791503906e-05,
      "step": 65002,
      "training_step_time": 9.272191524505615
    },
    {
      "epoch": 6.199169158935547e-05,
      "model_forward_time": 1.7155804634094238,
      "step": 65003
    },
    {
      "epoch": 6.199169158935547e-05,
      "step": 65003,
      "training_step_time": 7.7180869579315186
    },
    {
      "epoch": 6.199264526367188e-05,
      "model_forward_time": 1.7270517349243164,
      "step": 65004
    },
    {
      "epoch": 6.199264526367188e-05,
      "step": 65004,
      "training_step_time": 8.523487567901611
    },
    {
      "epoch": 6.199359893798828e-05,
      "model_forward_time": 1.7128818035125732,
      "step": 65005
    },
    {
      "epoch": 6.199359893798828e-05,
      "step": 65005,
      "training_step_time": 9.78832483291626
    },
    {
      "epoch": 6.199455261230469e-05,
      "model_forward_time": 1.734865427017212,
      "step": 65006
    },
    {
      "epoch": 6.199455261230469e-05,
      "step": 65006,
      "training_step_time": 9.713290691375732
    },
    {
      "epoch": 6.19955062866211e-05,
      "model_forward_time": 1.7686741352081299,
      "step": 65007
    },
    {
      "epoch": 6.19955062866211e-05,
      "step": 65007,
      "training_step_time": 10.366029262542725
    },
    {
      "epoch": 6.19964599609375e-05,
      "model_forward_time": 2.7398452758789062,
      "step": 65008
    },
    {
      "epoch": 6.19964599609375e-05,
      "step": 65008,
      "training_step_time": 8.359015703201294
    },
    {
      "epoch": 6.19974136352539e-05,
      "model_forward_time": 1.71842622756958,
      "step": 65009
    },
    {
      "epoch": 6.19974136352539e-05,
      "step": 65009,
      "training_step_time": 8.614513397216797
    },
    {
      "epoch": 6.199836730957032e-05,
      "grad_norm": 0.10492119193077087,
      "learning_rate": 2.9901601825111958e-06,
      "loss": 0.0592,
      "step": 65010
    },
    {
      "epoch": 6.199836730957032e-05,
      "model_forward_time": 1.7038686275482178,
      "step": 65010
    },
    {
      "dynamics_loss_avg": 0.06053667478263378,
      "epoch": 6.199836730957032e-05,
      "step": 65010
    },
    {
      "action_loss_avg": 0.001785313372965902,
      "epoch": 6.199836730957032e-05,
      "step": 65010
    },
    {
      "epoch": 6.199836730957032e-05,
      "step": 65010,
      "training_step_time": 9.362388610839844
    },
    {
      "epoch": 6.199932098388671e-05,
      "model_forward_time": 1.7105488777160645,
      "step": 65011
    },
    {
      "epoch": 6.199932098388671e-05,
      "step": 65011,
      "training_step_time": 7.7839648723602295
    },
    {
      "epoch": 6.200027465820313e-05,
      "model_forward_time": 1.698922872543335,
      "step": 65012
    },
    {
      "epoch": 6.200027465820313e-05,
      "step": 65012,
      "training_step_time": 7.778797626495361
    },
    {
      "epoch": 6.200122833251954e-05,
      "model_forward_time": 1.7061593532562256,
      "step": 65013
    },
    {
      "epoch": 6.200122833251954e-05,
      "step": 65013,
      "training_step_time": 7.531682014465332
    },
    {
      "epoch": 6.200218200683593e-05,
      "model_forward_time": 1.7136530876159668,
      "step": 65014
    },
    {
      "epoch": 6.200218200683593e-05,
      "step": 65014,
      "training_step_time": 8.456861019134521
    },
    {
      "epoch": 6.200313568115234e-05,
      "model_forward_time": 1.7030279636383057,
      "step": 65015
    },
    {
      "epoch": 6.200313568115234e-05,
      "step": 65015,
      "training_step_time": 8.729242324829102
    },
    {
      "epoch": 6.200408935546876e-05,
      "model_forward_time": 1.7347543239593506,
      "step": 65016
    },
    {
      "epoch": 6.200408935546876e-05,
      "step": 65016,
      "training_step_time": 7.723572015762329
    },
    {
      "epoch": 6.200504302978515e-05,
      "model_forward_time": 2.2541611194610596,
      "step": 65017
    },
    {
      "epoch": 6.200504302978515e-05,
      "step": 65017,
      "training_step_time": 9.638857126235962
    },
    {
      "epoch": 6.200599670410156e-05,
      "model_forward_time": 1.707214593887329,
      "step": 65018
    },
    {
      "epoch": 6.200599670410156e-05,
      "step": 65018,
      "training_step_time": 8.364779233932495
    },
    {
      "epoch": 6.200695037841797e-05,
      "model_forward_time": 1.7212374210357666,
      "step": 65019
    },
    {
      "epoch": 6.200695037841797e-05,
      "step": 65019,
      "training_step_time": 7.68921422958374
    },
    {
      "epoch": 6.200790405273437e-05,
      "grad_norm": 0.11922481656074524,
      "learning_rate": 2.9886462866265575e-06,
      "loss": 0.0576,
      "step": 65020
    },
    {
      "epoch": 6.200790405273437e-05,
      "model_forward_time": 1.7172787189483643,
      "step": 65020
    },
    {
      "dynamics_loss_avg": 0.06777059212327004,
      "epoch": 6.200790405273437e-05,
      "step": 65020
    },
    {
      "action_loss_avg": 0.0017949238244909793,
      "epoch": 6.200790405273437e-05,
      "step": 65020
    },
    {
      "epoch": 6.200790405273437e-05,
      "step": 65020,
      "training_step_time": 7.71319580078125
    },
    {
      "epoch": 6.200885772705078e-05,
      "model_forward_time": 1.7424070835113525,
      "step": 65021
    },
    {
      "epoch": 6.200885772705078e-05,
      "step": 65021,
      "training_step_time": 7.715436935424805
    },
    {
      "epoch": 6.200981140136719e-05,
      "model_forward_time": 1.7380917072296143,
      "step": 65022
    },
    {
      "epoch": 6.200981140136719e-05,
      "step": 65022,
      "training_step_time": 7.62652325630188
    },
    {
      "epoch": 6.201076507568359e-05,
      "model_forward_time": 1.71696138381958,
      "step": 65023
    },
    {
      "epoch": 6.201076507568359e-05,
      "step": 65023,
      "training_step_time": 7.725507497787476
    },
    {
      "epoch": 6.201171875e-05,
      "model_forward_time": 1.7304818630218506,
      "step": 65024
    },
    {
      "epoch": 6.201171875e-05,
      "step": 65024,
      "training_step_time": 7.775347709655762
    },
    {
      "epoch": 6.201267242431641e-05,
      "model_forward_time": 1.734205722808838,
      "step": 65025
    },
    {
      "epoch": 6.201267242431641e-05,
      "step": 65025,
      "training_step_time": 7.762963533401489
    },
    {
      "epoch": 6.201362609863281e-05,
      "model_forward_time": 2.2643651962280273,
      "step": 65026
    },
    {
      "epoch": 6.201362609863281e-05,
      "step": 65026,
      "training_step_time": 7.788949489593506
    },
    {
      "epoch": 6.201457977294922e-05,
      "model_forward_time": 1.7432146072387695,
      "step": 65027
    },
    {
      "epoch": 6.201457977294922e-05,
      "step": 65027,
      "training_step_time": 7.735661268234253
    },
    {
      "epoch": 6.201553344726563e-05,
      "model_forward_time": 1.6898343563079834,
      "step": 65028
    },
    {
      "epoch": 6.201553344726563e-05,
      "step": 65028,
      "training_step_time": 8.462750911712646
    },
    {
      "epoch": 6.201648712158203e-05,
      "model_forward_time": 1.7122306823730469,
      "step": 65029
    },
    {
      "epoch": 6.201648712158203e-05,
      "step": 65029,
      "training_step_time": 8.443345308303833
    },
    {
      "epoch": 6.201744079589844e-05,
      "grad_norm": 0.1260584443807602,
      "learning_rate": 2.9871326107005395e-06,
      "loss": 0.056,
      "step": 65030
    },
    {
      "epoch": 6.201744079589844e-05,
      "model_forward_time": 1.7095832824707031,
      "step": 65030
    },
    {
      "dynamics_loss_avg": 0.05075684655457735,
      "epoch": 6.201744079589844e-05,
      "step": 65030
    },
    {
      "action_loss_avg": 0.0034892216266598553,
      "epoch": 6.201744079589844e-05,
      "step": 65030
    },
    {
      "epoch": 6.201744079589844e-05,
      "step": 65030,
      "training_step_time": 7.6826441287994385
    },
    {
      "epoch": 6.201839447021485e-05,
      "model_forward_time": 1.731675386428833,
      "step": 65031
    },
    {
      "epoch": 6.201839447021485e-05,
      "step": 65031,
      "training_step_time": 8.452197790145874
    },
    {
      "epoch": 6.201934814453125e-05,
      "model_forward_time": 1.7199757099151611,
      "step": 65032
    },
    {
      "epoch": 6.201934814453125e-05,
      "step": 65032,
      "training_step_time": 7.6700615882873535
    },
    {
      "epoch": 6.202030181884766e-05,
      "model_forward_time": 1.718449592590332,
      "step": 65033
    },
    {
      "epoch": 6.202030181884766e-05,
      "step": 65033,
      "training_step_time": 8.704054355621338
    },
    {
      "epoch": 6.202125549316407e-05,
      "model_forward_time": 1.7119204998016357,
      "step": 65034
    },
    {
      "epoch": 6.202125549316407e-05,
      "step": 65034,
      "training_step_time": 8.745957374572754
    },
    {
      "epoch": 6.202220916748047e-05,
      "model_forward_time": 1.7708029747009277,
      "step": 65035
    },
    {
      "epoch": 6.202220916748047e-05,
      "step": 65035,
      "training_step_time": 8.51870059967041
    },
    {
      "epoch": 6.202316284179688e-05,
      "model_forward_time": 2.270841598510742,
      "step": 65036
    },
    {
      "epoch": 6.202316284179688e-05,
      "step": 65036,
      "training_step_time": 7.777616739273071
    },
    {
      "epoch": 6.202411651611329e-05,
      "model_forward_time": 1.7179789543151855,
      "step": 65037
    },
    {
      "epoch": 6.202411651611329e-05,
      "step": 65037,
      "training_step_time": 7.707226753234863
    },
    {
      "epoch": 6.202507019042968e-05,
      "model_forward_time": 1.70662260055542,
      "step": 65038
    },
    {
      "epoch": 6.202507019042968e-05,
      "step": 65038,
      "training_step_time": 7.531369686126709
    },
    {
      "epoch": 6.20260238647461e-05,
      "model_forward_time": 1.7049434185028076,
      "step": 65039
    },
    {
      "epoch": 6.20260238647461e-05,
      "step": 65039,
      "training_step_time": 7.596272945404053
    },
    {
      "epoch": 6.20269775390625e-05,
      "grad_norm": 0.15218764543533325,
      "learning_rate": 2.9856191548986737e-06,
      "loss": 0.0568,
      "step": 65040
    },
    {
      "epoch": 6.20269775390625e-05,
      "model_forward_time": 1.7094392776489258,
      "step": 65040
    },
    {
      "dynamics_loss_avg": 0.055900799296796325,
      "epoch": 6.20269775390625e-05,
      "step": 65040
    },
    {
      "action_loss_avg": 0.0019289888441562653,
      "epoch": 6.20269775390625e-05,
      "step": 65040
    },
    {
      "epoch": 6.20269775390625e-05,
      "step": 65040,
      "training_step_time": 7.693647384643555
    },
    {
      "epoch": 6.20279312133789e-05,
      "model_forward_time": 1.7084934711456299,
      "step": 65041
    },
    {
      "epoch": 6.20279312133789e-05,
      "step": 65041,
      "training_step_time": 7.521820306777954
    },
    {
      "epoch": 6.202888488769531e-05,
      "model_forward_time": 1.7123668193817139,
      "step": 65042
    },
    {
      "epoch": 6.202888488769531e-05,
      "step": 65042,
      "training_step_time": 9.649926662445068
    },
    {
      "epoch": 6.202983856201172e-05,
      "model_forward_time": 1.7124428749084473,
      "step": 65043
    },
    {
      "epoch": 6.202983856201172e-05,
      "step": 65043,
      "training_step_time": 8.670112371444702
    },
    {
      "epoch": 6.203079223632812e-05,
      "model_forward_time": 1.7399864196777344,
      "step": 65044
    },
    {
      "epoch": 6.203079223632812e-05,
      "step": 65044,
      "training_step_time": 8.629011154174805
    },
    {
      "epoch": 6.203174591064453e-05,
      "model_forward_time": 2.2513842582702637,
      "step": 65045
    },
    {
      "epoch": 6.203174591064453e-05,
      "step": 65045,
      "training_step_time": 8.647602558135986
    },
    {
      "epoch": 6.203269958496094e-05,
      "model_forward_time": 1.7108337879180908,
      "step": 65046
    },
    {
      "epoch": 6.203269958496094e-05,
      "step": 65046,
      "training_step_time": 8.22109341621399
    },
    {
      "epoch": 6.203365325927734e-05,
      "model_forward_time": 1.7234163284301758,
      "step": 65047
    },
    {
      "epoch": 6.203365325927734e-05,
      "step": 65047,
      "training_step_time": 8.366801023483276
    },
    {
      "epoch": 6.203460693359375e-05,
      "model_forward_time": 1.7093546390533447,
      "step": 65048
    },
    {
      "epoch": 6.203460693359375e-05,
      "step": 65048,
      "training_step_time": 7.4966881275177
    },
    {
      "epoch": 6.203556060791016e-05,
      "model_forward_time": 1.7031478881835938,
      "step": 65049
    },
    {
      "epoch": 6.203556060791016e-05,
      "step": 65049,
      "training_step_time": 7.521981477737427
    },
    {
      "epoch": 6.203651428222656e-05,
      "grad_norm": 0.07984000444412231,
      "learning_rate": 2.9841059193864676e-06,
      "loss": 0.0567,
      "step": 65050
    },
    {
      "epoch": 6.203651428222656e-05,
      "model_forward_time": 1.7044782638549805,
      "step": 65050
    },
    {
      "dynamics_loss_avg": 0.06794184129685163,
      "epoch": 6.203651428222656e-05,
      "step": 65050
    },
    {
      "action_loss_avg": 0.001410543665406294,
      "epoch": 6.203651428222656e-05,
      "step": 65050
    },
    {
      "epoch": 6.203651428222656e-05,
      "step": 65050,
      "training_step_time": 8.442964553833008
    },
    {
      "epoch": 6.203746795654297e-05,
      "model_forward_time": 1.7043695449829102,
      "step": 65051
    },
    {
      "epoch": 6.203746795654297e-05,
      "step": 65051,
      "training_step_time": 8.652849197387695
    },
    {
      "epoch": 6.203842163085938e-05,
      "model_forward_time": 1.7099199295043945,
      "step": 65052
    },
    {
      "epoch": 6.203842163085938e-05,
      "step": 65052,
      "training_step_time": 7.634192943572998
    },
    {
      "epoch": 6.203937530517578e-05,
      "model_forward_time": 1.7188327312469482,
      "step": 65053
    },
    {
      "epoch": 6.203937530517578e-05,
      "step": 65053,
      "training_step_time": 8.770000696182251
    },
    {
      "epoch": 6.204032897949219e-05,
      "model_forward_time": 2.775958299636841,
      "step": 65054
    },
    {
      "epoch": 6.204032897949219e-05,
      "step": 65054,
      "training_step_time": 10.821441173553467
    },
    {
      "epoch": 6.20412826538086e-05,
      "model_forward_time": 1.7144365310668945,
      "step": 65055
    },
    {
      "epoch": 6.20412826538086e-05,
      "step": 65055,
      "training_step_time": 6.62056040763855
    },
    {
      "epoch": 6.2042236328125e-05,
      "model_forward_time": 1.713505744934082,
      "step": 65056
    },
    {
      "epoch": 6.2042236328125e-05,
      "step": 65056,
      "training_step_time": 6.625004053115845
    },
    {
      "epoch": 6.204319000244141e-05,
      "model_forward_time": 1.7470498085021973,
      "step": 65057
    },
    {
      "epoch": 6.204319000244141e-05,
      "step": 65057,
      "training_step_time": 6.58989953994751
    },
    {
      "epoch": 6.204414367675782e-05,
      "model_forward_time": 1.7454566955566406,
      "step": 65058
    },
    {
      "epoch": 6.204414367675782e-05,
      "step": 65058,
      "training_step_time": 6.555221319198608
    },
    {
      "epoch": 6.204509735107422e-05,
      "model_forward_time": 1.743011236190796,
      "step": 65059
    },
    {
      "epoch": 6.204509735107422e-05,
      "step": 65059,
      "training_step_time": 6.577630043029785
    },
    {
      "epoch": 6.204605102539063e-05,
      "grad_norm": 0.09559936821460724,
      "learning_rate": 2.982592904329407e-06,
      "loss": 0.056,
      "step": 65060
    },
    {
      "epoch": 6.204605102539063e-05,
      "model_forward_time": 1.7434687614440918,
      "step": 65060
    },
    {
      "dynamics_loss_avg": 0.05808714013546705,
      "epoch": 6.204605102539063e-05,
      "step": 65060
    },
    {
      "action_loss_avg": 0.0017333762720227242,
      "epoch": 6.204605102539063e-05,
      "step": 65060
    },
    {
      "epoch": 6.204605102539063e-05,
      "step": 65060,
      "training_step_time": 6.5668275356292725
    },
    {
      "epoch": 6.204700469970704e-05,
      "model_forward_time": 1.7495672702789307,
      "step": 65061
    },
    {
      "epoch": 6.204700469970704e-05,
      "step": 65061,
      "training_step_time": 6.563889503479004
    },
    {
      "epoch": 6.204795837402344e-05,
      "model_forward_time": 1.7440428733825684,
      "step": 65062
    },
    {
      "epoch": 6.204795837402344e-05,
      "step": 65062,
      "training_step_time": 6.5929484367370605
    },
    {
      "epoch": 6.204891204833985e-05,
      "model_forward_time": 2.9080135822296143,
      "step": 65063
    },
    {
      "epoch": 6.204891204833985e-05,
      "step": 65063,
      "training_step_time": 10.033838748931885
    },
    {
      "epoch": 6.204986572265624e-05,
      "model_forward_time": 1.7479748725891113,
      "step": 65064
    },
    {
      "epoch": 6.204986572265624e-05,
      "step": 65064,
      "training_step_time": 7.681879758834839
    },
    {
      "epoch": 6.205081939697265e-05,
      "model_forward_time": 1.735809087753296,
      "step": 65065
    },
    {
      "epoch": 6.205081939697265e-05,
      "step": 65065,
      "training_step_time": 6.573345184326172
    },
    {
      "epoch": 6.205177307128906e-05,
      "model_forward_time": 1.7415175437927246,
      "step": 65066
    },
    {
      "epoch": 6.205177307128906e-05,
      "step": 65066,
      "training_step_time": 6.569172620773315
    },
    {
      "epoch": 6.205272674560546e-05,
      "model_forward_time": 1.7463562488555908,
      "step": 65067
    },
    {
      "epoch": 6.205272674560546e-05,
      "step": 65067,
      "training_step_time": 6.5511579513549805
    },
    {
      "epoch": 6.205368041992187e-05,
      "model_forward_time": 1.742069959640503,
      "step": 65068
    },
    {
      "epoch": 6.205368041992187e-05,
      "step": 65068,
      "training_step_time": 6.578487396240234
    },
    {
      "epoch": 6.205463409423828e-05,
      "model_forward_time": 1.7452435493469238,
      "step": 65069
    },
    {
      "epoch": 6.205463409423828e-05,
      "step": 65069,
      "training_step_time": 6.560176134109497
    },
    {
      "epoch": 6.205558776855468e-05,
      "grad_norm": 0.1169375330209732,
      "learning_rate": 2.9810801098929543e-06,
      "loss": 0.056,
      "step": 65070
    },
    {
      "epoch": 6.205558776855468e-05,
      "model_forward_time": 1.710878610610962,
      "step": 65070
    },
    {
      "dynamics_loss_avg": 0.05172856599092483,
      "epoch": 6.205558776855468e-05,
      "step": 65070
    },
    {
      "action_loss_avg": 0.0012194721362902782,
      "epoch": 6.205558776855468e-05,
      "step": 65070
    },
    {
      "epoch": 6.205558776855468e-05,
      "step": 65070,
      "training_step_time": 6.570772409439087
    },
    {
      "epoch": 6.205654144287109e-05,
      "model_forward_time": 1.726644515991211,
      "step": 65071
    },
    {
      "epoch": 6.205654144287109e-05,
      "step": 65071,
      "training_step_time": 6.571965932846069
    },
    {
      "epoch": 6.20574951171875e-05,
      "model_forward_time": 2.8754537105560303,
      "step": 65072
    },
    {
      "epoch": 6.20574951171875e-05,
      "step": 65072,
      "training_step_time": 8.334434509277344
    },
    {
      "epoch": 6.20584487915039e-05,
      "model_forward_time": 1.723994493484497,
      "step": 65073
    },
    {
      "epoch": 6.20584487915039e-05,
      "step": 65073,
      "training_step_time": 7.726407051086426
    },
    {
      "epoch": 6.205940246582031e-05,
      "model_forward_time": 1.7267990112304688,
      "step": 65074
    },
    {
      "epoch": 6.205940246582031e-05,
      "step": 65074,
      "training_step_time": 6.603684425354004
    },
    {
      "epoch": 6.206035614013672e-05,
      "model_forward_time": 1.7220969200134277,
      "step": 65075
    },
    {
      "epoch": 6.206035614013672e-05,
      "step": 65075,
      "training_step_time": 6.645238876342773
    },
    {
      "epoch": 6.206130981445312e-05,
      "model_forward_time": 1.7255990505218506,
      "step": 65076
    },
    {
      "epoch": 6.206130981445312e-05,
      "step": 65076,
      "training_step_time": 6.630380630493164
    },
    {
      "epoch": 6.206226348876953e-05,
      "model_forward_time": 1.7309696674346924,
      "step": 65077
    },
    {
      "epoch": 6.206226348876953e-05,
      "step": 65077,
      "training_step_time": 6.5634849071502686
    },
    {
      "epoch": 6.206321716308594e-05,
      "model_forward_time": 1.7268412113189697,
      "step": 65078
    },
    {
      "epoch": 6.206321716308594e-05,
      "step": 65078,
      "training_step_time": 6.622184991836548
    },
    {
      "epoch": 6.206417083740234e-05,
      "model_forward_time": 1.7255711555480957,
      "step": 65079
    },
    {
      "epoch": 6.206417083740234e-05,
      "step": 65079,
      "training_step_time": 6.5950353145599365
    },
    {
      "epoch": 6.206512451171875e-05,
      "grad_norm": 0.1053670346736908,
      "learning_rate": 2.9795675362425434e-06,
      "loss": 0.0523,
      "step": 65080
    },
    {
      "epoch": 6.206512451171875e-05,
      "model_forward_time": 1.727179765701294,
      "step": 65080
    },
    {
      "dynamics_loss_avg": 0.04445622432976961,
      "epoch": 6.206512451171875e-05,
      "step": 65080
    },
    {
      "action_loss_avg": 0.001635502331191674,
      "epoch": 6.206512451171875e-05,
      "step": 65080
    },
    {
      "epoch": 6.206512451171875e-05,
      "step": 65080,
      "training_step_time": 6.60846734046936
    },
    {
      "epoch": 6.206607818603516e-05,
      "model_forward_time": 1.768115758895874,
      "step": 65081
    },
    {
      "epoch": 6.206607818603516e-05,
      "step": 65081,
      "training_step_time": 8.217687368392944
    },
    {
      "epoch": 6.206703186035156e-05,
      "model_forward_time": 1.825183629989624,
      "step": 65082
    },
    {
      "epoch": 6.206703186035156e-05,
      "step": 65082,
      "training_step_time": 9.690879106521606
    },
    {
      "epoch": 6.206798553466797e-05,
      "model_forward_time": 1.7226536273956299,
      "step": 65083
    },
    {
      "epoch": 6.206798553466797e-05,
      "step": 65083,
      "training_step_time": 7.436195373535156
    },
    {
      "epoch": 6.206893920898438e-05,
      "model_forward_time": 1.724562168121338,
      "step": 65084
    },
    {
      "epoch": 6.206893920898438e-05,
      "step": 65084,
      "training_step_time": 6.5756449699401855
    },
    {
      "epoch": 6.206989288330078e-05,
      "model_forward_time": 1.7217462062835693,
      "step": 65085
    },
    {
      "epoch": 6.206989288330078e-05,
      "step": 65085,
      "training_step_time": 6.6005003452301025
    },
    {
      "epoch": 6.207084655761719e-05,
      "model_forward_time": 1.7300283908843994,
      "step": 65086
    },
    {
      "epoch": 6.207084655761719e-05,
      "step": 65086,
      "training_step_time": 6.583548545837402
    },
    {
      "epoch": 6.20718002319336e-05,
      "model_forward_time": 1.7420916557312012,
      "step": 65087
    },
    {
      "epoch": 6.20718002319336e-05,
      "step": 65087,
      "training_step_time": 6.554160833358765
    },
    {
      "epoch": 6.207275390625e-05,
      "model_forward_time": 1.7252039909362793,
      "step": 65088
    },
    {
      "epoch": 6.207275390625e-05,
      "step": 65088,
      "training_step_time": 6.549242734909058
    },
    {
      "epoch": 6.20737075805664e-05,
      "model_forward_time": 1.7285094261169434,
      "step": 65089
    },
    {
      "epoch": 6.20737075805664e-05,
      "step": 65089,
      "training_step_time": 6.550121784210205
    },
    {
      "epoch": 6.207466125488282e-05,
      "grad_norm": 0.1134626567363739,
      "learning_rate": 2.97805518354359e-06,
      "loss": 0.0543,
      "step": 65090
    },
    {
      "epoch": 6.207466125488282e-05,
      "model_forward_time": 1.7600677013397217,
      "step": 65090
    },
    {
      "dynamics_loss_avg": 0.05562407001852989,
      "epoch": 6.207466125488282e-05,
      "step": 65090
    },
    {
      "action_loss_avg": 0.0017966630897717551,
      "epoch": 6.207466125488282e-05,
      "step": 65090
    },
    {
      "epoch": 6.207466125488282e-05,
      "step": 65090,
      "training_step_time": 6.597844123840332
    },
    {
      "epoch": 6.207561492919921e-05,
      "model_forward_time": 3.0031485557556152,
      "step": 65091
    },
    {
      "epoch": 6.207561492919921e-05,
      "step": 65091,
      "training_step_time": 10.16498589515686
    },
    {
      "epoch": 6.207656860351562e-05,
      "model_forward_time": 1.7249858379364014,
      "step": 65092
    },
    {
      "epoch": 6.207656860351562e-05,
      "step": 65092,
      "training_step_time": 7.691493988037109
    },
    {
      "epoch": 6.207752227783203e-05,
      "model_forward_time": 1.7260725498199463,
      "step": 65093
    },
    {
      "epoch": 6.207752227783203e-05,
      "step": 65093,
      "training_step_time": 6.570813417434692
    },
    {
      "epoch": 6.207847595214843e-05,
      "model_forward_time": 1.7295050621032715,
      "step": 65094
    },
    {
      "epoch": 6.207847595214843e-05,
      "step": 65094,
      "training_step_time": 6.581649303436279
    },
    {
      "epoch": 6.207942962646484e-05,
      "model_forward_time": 1.7239444255828857,
      "step": 65095
    },
    {
      "epoch": 6.207942962646484e-05,
      "step": 65095,
      "training_step_time": 6.591387510299683
    },
    {
      "epoch": 6.208038330078125e-05,
      "model_forward_time": 1.7269494533538818,
      "step": 65096
    },
    {
      "epoch": 6.208038330078125e-05,
      "step": 65096,
      "training_step_time": 6.595286846160889
    },
    {
      "epoch": 6.208133697509765e-05,
      "model_forward_time": 1.7279894351959229,
      "step": 65097
    },
    {
      "epoch": 6.208133697509765e-05,
      "step": 65097,
      "training_step_time": 6.573375463485718
    },
    {
      "epoch": 6.208229064941406e-05,
      "model_forward_time": 1.7192306518554688,
      "step": 65098
    },
    {
      "epoch": 6.208229064941406e-05,
      "step": 65098,
      "training_step_time": 6.541345596313477
    },
    {
      "epoch": 6.208324432373047e-05,
      "model_forward_time": 1.7248389720916748,
      "step": 65099
    },
    {
      "epoch": 6.208324432373047e-05,
      "step": 65099,
      "training_step_time": 6.540279865264893
    },
    {
      "epoch": 6.208419799804687e-05,
      "grad_norm": 0.09425048530101776,
      "learning_rate": 2.9765430519614826e-06,
      "loss": 0.0539,
      "step": 65100
    },
    {
      "epoch": 6.208419799804687e-05,
      "model_forward_time": 2.9308981895446777,
      "step": 65100
    },
    {
      "dynamics_loss_avg": 0.042032504081726076,
      "epoch": 6.208419799804687e-05,
      "step": 65100
    },
    {
      "action_loss_avg": 0.002243767149047926,
      "epoch": 6.208419799804687e-05,
      "step": 65100
    },
    {
      "epoch": 6.208419799804687e-05,
      "step": 65100,
      "training_step_time": 9.435421466827393
    },
    {
      "epoch": 6.208515167236328e-05,
      "model_forward_time": 1.361907958984375,
      "step": 65101
    },
    {
      "epoch": 6.208515167236328e-05,
      "step": 65101,
      "training_step_time": 8.450999736785889
    },
    {
      "epoch": 6.208610534667969e-05,
      "model_forward_time": 1.7254741191864014,
      "step": 65102
    },
    {
      "epoch": 6.208610534667969e-05,
      "step": 65102,
      "training_step_time": 6.603815793991089
    },
    {
      "epoch": 6.208705902099609e-05,
      "model_forward_time": 1.7295567989349365,
      "step": 65103
    },
    {
      "epoch": 6.208705902099609e-05,
      "step": 65103,
      "training_step_time": 6.5543577671051025
    },
    {
      "epoch": 6.20880126953125e-05,
      "model_forward_time": 1.7265779972076416,
      "step": 65104
    },
    {
      "epoch": 6.20880126953125e-05,
      "step": 65104,
      "training_step_time": 6.564514636993408
    },
    {
      "epoch": 6.208896636962891e-05,
      "model_forward_time": 1.728710651397705,
      "step": 65105
    },
    {
      "epoch": 6.208896636962891e-05,
      "step": 65105,
      "training_step_time": 6.586015224456787
    },
    {
      "epoch": 6.208992004394531e-05,
      "model_forward_time": 1.7258634567260742,
      "step": 65106
    },
    {
      "epoch": 6.208992004394531e-05,
      "step": 65106,
      "training_step_time": 6.573728084564209
    },
    {
      "epoch": 6.209087371826172e-05,
      "model_forward_time": 1.7237987518310547,
      "step": 65107
    },
    {
      "epoch": 6.209087371826172e-05,
      "step": 65107,
      "training_step_time": 6.593637943267822
    },
    {
      "epoch": 6.209182739257813e-05,
      "model_forward_time": 1.7295076847076416,
      "step": 65108
    },
    {
      "epoch": 6.209182739257813e-05,
      "step": 65108,
      "training_step_time": 6.609259843826294
    },
    {
      "epoch": 6.209278106689453e-05,
      "model_forward_time": 2.820674419403076,
      "step": 65109
    },
    {
      "epoch": 6.209278106689453e-05,
      "step": 65109,
      "training_step_time": 8.942871570587158
    },
    {
      "epoch": 6.209373474121094e-05,
      "grad_norm": 0.15584814548492432,
      "learning_rate": 2.975031141661584e-06,
      "loss": 0.0561,
      "step": 65110
    },
    {
      "epoch": 6.209373474121094e-05,
      "model_forward_time": 1.8338096141815186,
      "step": 65110
    },
    {
      "dynamics_loss_avg": 0.05417402312159538,
      "epoch": 6.209373474121094e-05,
      "step": 65110
    },
    {
      "action_loss_avg": 0.00252348160429392,
      "epoch": 6.209373474121094e-05,
      "step": 65110
    },
    {
      "epoch": 6.209373474121094e-05,
      "step": 65110,
      "training_step_time": 7.649054765701294
    },
    {
      "epoch": 6.209468841552735e-05,
      "model_forward_time": 1.754704236984253,
      "step": 65111
    },
    {
      "epoch": 6.209468841552735e-05,
      "step": 65111,
      "training_step_time": 7.206976890563965
    },
    {
      "epoch": 6.209564208984375e-05,
      "model_forward_time": 1.732175350189209,
      "step": 65112
    },
    {
      "epoch": 6.209564208984375e-05,
      "step": 65112,
      "training_step_time": 6.575407028198242
    },
    {
      "epoch": 6.209659576416016e-05,
      "model_forward_time": 1.7250635623931885,
      "step": 65113
    },
    {
      "epoch": 6.209659576416016e-05,
      "step": 65113,
      "training_step_time": 6.597899436950684
    },
    {
      "epoch": 6.209754943847657e-05,
      "model_forward_time": 1.7225897312164307,
      "step": 65114
    },
    {
      "epoch": 6.209754943847657e-05,
      "step": 65114,
      "training_step_time": 6.5715248584747314
    },
    {
      "epoch": 6.209850311279296e-05,
      "model_forward_time": 1.727687120437622,
      "step": 65115
    },
    {
      "epoch": 6.209850311279296e-05,
      "step": 65115,
      "training_step_time": 6.584827899932861
    },
    {
      "epoch": 6.209945678710937e-05,
      "model_forward_time": 1.7240333557128906,
      "step": 65116
    },
    {
      "epoch": 6.209945678710937e-05,
      "step": 65116,
      "training_step_time": 6.583297252655029
    },
    {
      "epoch": 6.210041046142579e-05,
      "model_forward_time": 1.7243804931640625,
      "step": 65117
    },
    {
      "epoch": 6.210041046142579e-05,
      "step": 65117,
      "training_step_time": 6.540004730224609
    },
    {
      "epoch": 6.210136413574218e-05,
      "model_forward_time": 2.8453569412231445,
      "step": 65118
    },
    {
      "epoch": 6.210136413574218e-05,
      "step": 65118,
      "training_step_time": 8.80312991142273
    },
    {
      "epoch": 6.21023178100586e-05,
      "model_forward_time": 1.7717840671539307,
      "step": 65119
    },
    {
      "epoch": 6.21023178100586e-05,
      "step": 65119,
      "training_step_time": 8.565990686416626
    },
    {
      "epoch": 6.2103271484375e-05,
      "grad_norm": 0.11995933204889297,
      "learning_rate": 2.9735194528092348e-06,
      "loss": 0.0554,
      "step": 65120
    },
    {
      "epoch": 6.2103271484375e-05,
      "model_forward_time": 1.72572660446167,
      "step": 65120
    },
    {
      "dynamics_loss_avg": 0.045269657392054796,
      "epoch": 6.2103271484375e-05,
      "step": 65120
    },
    {
      "action_loss_avg": 0.002095502997690346,
      "epoch": 6.2103271484375e-05,
      "step": 65120
    },
    {
      "epoch": 6.2103271484375e-05,
      "step": 65120,
      "training_step_time": 7.345220565795898
    },
    {
      "epoch": 6.21042251586914e-05,
      "model_forward_time": 1.729203462600708,
      "step": 65121
    },
    {
      "epoch": 6.21042251586914e-05,
      "step": 65121,
      "training_step_time": 6.552187442779541
    },
    {
      "epoch": 6.210517883300781e-05,
      "model_forward_time": 1.7232630252838135,
      "step": 65122
    },
    {
      "epoch": 6.210517883300781e-05,
      "step": 65122,
      "training_step_time": 6.579229831695557
    },
    {
      "epoch": 6.210613250732422e-05,
      "model_forward_time": 1.7276737689971924,
      "step": 65123
    },
    {
      "epoch": 6.210613250732422e-05,
      "step": 65123,
      "training_step_time": 6.557021379470825
    },
    {
      "epoch": 6.210708618164062e-05,
      "model_forward_time": 1.365018606185913,
      "step": 65124
    },
    {
      "epoch": 6.210708618164062e-05,
      "step": 65124,
      "training_step_time": 6.573788404464722
    },
    {
      "epoch": 6.210803985595703e-05,
      "model_forward_time": 1.7628910541534424,
      "step": 65125
    },
    {
      "epoch": 6.210803985595703e-05,
      "step": 65125,
      "training_step_time": 6.733561754226685
    },
    {
      "epoch": 6.210899353027344e-05,
      "model_forward_time": 1.7715847492218018,
      "step": 65126
    },
    {
      "epoch": 6.210899353027344e-05,
      "step": 65126,
      "training_step_time": 6.7294230461120605
    },
    {
      "epoch": 6.210994720458984e-05,
      "model_forward_time": 1.7359693050384521,
      "step": 65127
    },
    {
      "epoch": 6.210994720458984e-05,
      "step": 65127,
      "training_step_time": 7.917527675628662
    },
    {
      "epoch": 6.211090087890625e-05,
      "model_forward_time": 3.004178762435913,
      "step": 65128
    },
    {
      "epoch": 6.211090087890625e-05,
      "step": 65128,
      "training_step_time": 8.784616470336914
    },
    {
      "epoch": 6.211185455322266e-05,
      "model_forward_time": 1.738309383392334,
      "step": 65129
    },
    {
      "epoch": 6.211185455322266e-05,
      "step": 65129,
      "training_step_time": 8.557153940200806
    },
    {
      "epoch": 6.211280822753906e-05,
      "grad_norm": 0.10955207049846649,
      "learning_rate": 2.9720079855697526e-06,
      "loss": 0.0572,
      "step": 65130
    },
    {
      "epoch": 6.211280822753906e-05,
      "model_forward_time": 1.742689609527588,
      "step": 65130
    },
    {
      "dynamics_loss_avg": 0.052144198678433895,
      "epoch": 6.211280822753906e-05,
      "step": 65130
    },
    {
      "action_loss_avg": 0.0029424812295474113,
      "epoch": 6.211280822753906e-05,
      "step": 65130
    },
    {
      "epoch": 6.211280822753906e-05,
      "step": 65130,
      "training_step_time": 6.773508310317993
    },
    {
      "epoch": 6.211376190185547e-05,
      "model_forward_time": 1.382908821105957,
      "step": 65131
    },
    {
      "epoch": 6.211376190185547e-05,
      "step": 65131,
      "training_step_time": 6.7086217403411865
    },
    {
      "epoch": 6.211471557617188e-05,
      "model_forward_time": 1.744999885559082,
      "step": 65132
    },
    {
      "epoch": 6.211471557617188e-05,
      "step": 65132,
      "training_step_time": 6.709399461746216
    },
    {
      "epoch": 6.211566925048828e-05,
      "model_forward_time": 1.7289845943450928,
      "step": 65133
    },
    {
      "epoch": 6.211566925048828e-05,
      "step": 65133,
      "training_step_time": 6.7497780323028564
    },
    {
      "epoch": 6.211662292480469e-05,
      "model_forward_time": 1.7269208431243896,
      "step": 65134
    },
    {
      "epoch": 6.211662292480469e-05,
      "step": 65134,
      "training_step_time": 6.769402265548706
    },
    {
      "epoch": 6.21175765991211e-05,
      "model_forward_time": 1.7222084999084473,
      "step": 65135
    },
    {
      "epoch": 6.21175765991211e-05,
      "step": 65135,
      "training_step_time": 6.728853464126587
    },
    {
      "epoch": 6.21185302734375e-05,
      "model_forward_time": 1.7251834869384766,
      "step": 65136
    },
    {
      "epoch": 6.21185302734375e-05,
      "step": 65136,
      "training_step_time": 8.602384567260742
    },
    {
      "epoch": 6.21194839477539e-05,
      "model_forward_time": 2.9401471614837646,
      "step": 65137
    },
    {
      "epoch": 6.21194839477539e-05,
      "step": 65137,
      "training_step_time": 8.289976358413696
    },
    {
      "epoch": 6.212043762207032e-05,
      "model_forward_time": 1.7257359027862549,
      "step": 65138
    },
    {
      "epoch": 6.212043762207032e-05,
      "step": 65138,
      "training_step_time": 7.662542343139648
    },
    {
      "epoch": 6.212139129638671e-05,
      "model_forward_time": 1.7355000972747803,
      "step": 65139
    },
    {
      "epoch": 6.212139129638671e-05,
      "step": 65139,
      "training_step_time": 7.351474761962891
    },
    {
      "epoch": 6.212234497070313e-05,
      "grad_norm": 0.1263587921857834,
      "learning_rate": 2.9704967401084272e-06,
      "loss": 0.0602,
      "step": 65140
    },
    {
      "epoch": 6.212234497070313e-05,
      "model_forward_time": 1.7263269424438477,
      "step": 65140
    },
    {
      "dynamics_loss_avg": 0.06087802071124315,
      "epoch": 6.212234497070313e-05,
      "step": 65140
    },
    {
      "action_loss_avg": 0.0015537228086031974,
      "epoch": 6.212234497070313e-05,
      "step": 65140
    },
    {
      "epoch": 6.212234497070313e-05,
      "step": 65140,
      "training_step_time": 6.741664886474609
    },
    {
      "epoch": 6.212329864501954e-05,
      "model_forward_time": 1.7290894985198975,
      "step": 65141
    },
    {
      "epoch": 6.212329864501954e-05,
      "step": 65141,
      "training_step_time": 6.770191431045532
    },
    {
      "epoch": 6.212425231933593e-05,
      "model_forward_time": 1.7263247966766357,
      "step": 65142
    },
    {
      "epoch": 6.212425231933593e-05,
      "step": 65142,
      "training_step_time": 6.725471019744873
    },
    {
      "epoch": 6.212520599365234e-05,
      "model_forward_time": 1.7239964008331299,
      "step": 65143
    },
    {
      "epoch": 6.212520599365234e-05,
      "step": 65143,
      "training_step_time": 6.740516185760498
    },
    {
      "epoch": 6.212615966796876e-05,
      "model_forward_time": 1.7244019508361816,
      "step": 65144
    },
    {
      "epoch": 6.212615966796876e-05,
      "step": 65144,
      "training_step_time": 6.746319770812988
    },
    {
      "epoch": 6.212711334228515e-05,
      "model_forward_time": 1.7266552448272705,
      "step": 65145
    },
    {
      "epoch": 6.212711334228515e-05,
      "step": 65145,
      "training_step_time": 8.939016580581665
    },
    {
      "epoch": 6.212806701660156e-05,
      "model_forward_time": 2.4906985759735107,
      "step": 65146
    },
    {
      "epoch": 6.212806701660156e-05,
      "step": 65146,
      "training_step_time": 8.249947786331177
    },
    {
      "epoch": 6.212902069091797e-05,
      "model_forward_time": 1.8318941593170166,
      "step": 65147
    },
    {
      "epoch": 6.212902069091797e-05,
      "step": 65147,
      "training_step_time": 7.648006916046143
    },
    {
      "epoch": 6.212997436523437e-05,
      "model_forward_time": 1.719980239868164,
      "step": 65148
    },
    {
      "epoch": 6.212997436523437e-05,
      "step": 65148,
      "training_step_time": 7.334140777587891
    },
    {
      "epoch": 6.213092803955078e-05,
      "model_forward_time": 1.7257182598114014,
      "step": 65149
    },
    {
      "epoch": 6.213092803955078e-05,
      "step": 65149,
      "training_step_time": 6.717675685882568
    },
    {
      "epoch": 6.213188171386719e-05,
      "grad_norm": 0.1298113316297531,
      "learning_rate": 2.9689857165905274e-06,
      "loss": 0.0568,
      "step": 65150
    },
    {
      "epoch": 6.213188171386719e-05,
      "model_forward_time": 1.71982741355896,
      "step": 65150
    },
    {
      "dynamics_loss_avg": 0.04999718349426985,
      "epoch": 6.213188171386719e-05,
      "step": 65150
    },
    {
      "action_loss_avg": 0.001763330832181964,
      "epoch": 6.213188171386719e-05,
      "step": 65150
    },
    {
      "epoch": 6.213188171386719e-05,
      "step": 65150,
      "training_step_time": 6.719751358032227
    },
    {
      "epoch": 6.213283538818359e-05,
      "model_forward_time": 1.72206711769104,
      "step": 65151
    },
    {
      "epoch": 6.213283538818359e-05,
      "step": 65151,
      "training_step_time": 6.712964296340942
    },
    {
      "epoch": 6.21337890625e-05,
      "model_forward_time": 1.7284386157989502,
      "step": 65152
    },
    {
      "epoch": 6.21337890625e-05,
      "step": 65152,
      "training_step_time": 6.737189769744873
    },
    {
      "epoch": 6.213474273681641e-05,
      "model_forward_time": 1.7193996906280518,
      "step": 65153
    },
    {
      "epoch": 6.213474273681641e-05,
      "step": 65153,
      "training_step_time": 6.709092617034912
    },
    {
      "epoch": 6.213569641113281e-05,
      "model_forward_time": 1.721998691558838,
      "step": 65154
    },
    {
      "epoch": 6.213569641113281e-05,
      "step": 65154,
      "training_step_time": 8.326037883758545
    },
    {
      "epoch": 6.213665008544922e-05,
      "model_forward_time": 1.7250442504882812,
      "step": 65155
    },
    {
      "epoch": 6.213665008544922e-05,
      "step": 65155,
      "training_step_time": 11.057578802108765
    },
    {
      "epoch": 6.213760375976563e-05,
      "model_forward_time": 1.7762250900268555,
      "step": 65156
    },
    {
      "epoch": 6.213760375976563e-05,
      "step": 65156,
      "training_step_time": 7.726187229156494
    },
    {
      "epoch": 6.213855743408203e-05,
      "model_forward_time": 1.7264249324798584,
      "step": 65157
    },
    {
      "epoch": 6.213855743408203e-05,
      "step": 65157,
      "training_step_time": 7.592709541320801
    },
    {
      "epoch": 6.213951110839844e-05,
      "model_forward_time": 1.719283103942871,
      "step": 65158
    },
    {
      "epoch": 6.213951110839844e-05,
      "step": 65158,
      "training_step_time": 6.705109596252441
    },
    {
      "epoch": 6.214046478271485e-05,
      "model_forward_time": 1.7244997024536133,
      "step": 65159
    },
    {
      "epoch": 6.214046478271485e-05,
      "step": 65159,
      "training_step_time": 6.731531858444214
    },
    {
      "epoch": 6.214141845703125e-05,
      "grad_norm": 0.14810515940189362,
      "learning_rate": 2.9674749151812953e-06,
      "loss": 0.0561,
      "step": 65160
    },
    {
      "epoch": 6.214141845703125e-05,
      "model_forward_time": 1.7268848419189453,
      "step": 65160
    },
    {
      "dynamics_loss_avg": 0.04613505303859711,
      "epoch": 6.214141845703125e-05,
      "step": 65160
    },
    {
      "action_loss_avg": 0.0012304376024985686,
      "epoch": 6.214141845703125e-05,
      "step": 65160
    },
    {
      "epoch": 6.214141845703125e-05,
      "step": 65160,
      "training_step_time": 6.644801378250122
    },
    {
      "epoch": 6.214237213134766e-05,
      "model_forward_time": 1.7157902717590332,
      "step": 65161
    },
    {
      "epoch": 6.214237213134766e-05,
      "step": 65161,
      "training_step_time": 6.696637153625488
    },
    {
      "epoch": 6.214332580566407e-05,
      "model_forward_time": 1.7211940288543701,
      "step": 65162
    },
    {
      "epoch": 6.214332580566407e-05,
      "step": 65162,
      "training_step_time": 6.691223382949829
    },
    {
      "epoch": 6.214427947998047e-05,
      "model_forward_time": 1.7247037887573242,
      "step": 65163
    },
    {
      "epoch": 6.214427947998047e-05,
      "step": 65163,
      "training_step_time": 7.537845611572266
    },
    {
      "epoch": 6.214523315429688e-05,
      "model_forward_time": 1.7221598625183105,
      "step": 65164
    },
    {
      "epoch": 6.214523315429688e-05,
      "step": 65164,
      "training_step_time": 10.534064769744873
    },
    {
      "epoch": 6.214618682861329e-05,
      "model_forward_time": 1.8174314498901367,
      "step": 65165
    },
    {
      "epoch": 6.214618682861329e-05,
      "step": 65165,
      "training_step_time": 6.745198965072632
    },
    {
      "epoch": 6.214714050292968e-05,
      "model_forward_time": 1.7280452251434326,
      "step": 65166
    },
    {
      "epoch": 6.214714050292968e-05,
      "step": 65166,
      "training_step_time": 7.61498498916626
    },
    {
      "epoch": 6.21480941772461e-05,
      "model_forward_time": 1.7210693359375,
      "step": 65167
    },
    {
      "epoch": 6.21480941772461e-05,
      "step": 65167,
      "training_step_time": 6.671329975128174
    },
    {
      "epoch": 6.21490478515625e-05,
      "model_forward_time": 1.7294974327087402,
      "step": 65168
    },
    {
      "epoch": 6.21490478515625e-05,
      "step": 65168,
      "training_step_time": 6.690166473388672
    },
    {
      "epoch": 6.21500015258789e-05,
      "model_forward_time": 1.72188401222229,
      "step": 65169
    },
    {
      "epoch": 6.21500015258789e-05,
      "step": 65169,
      "training_step_time": 6.666271686553955
    },
    {
      "epoch": 6.215095520019531e-05,
      "grad_norm": 0.17956654727458954,
      "learning_rate": 2.965964336045951e-06,
      "loss": 0.0592,
      "step": 65170
    },
    {
      "epoch": 6.215095520019531e-05,
      "model_forward_time": 1.7218143939971924,
      "step": 65170
    },
    {
      "dynamics_loss_avg": 0.045966864377260205,
      "epoch": 6.215095520019531e-05,
      "step": 65170
    },
    {
      "action_loss_avg": 0.0017249492797418498,
      "epoch": 6.215095520019531e-05,
      "step": 65170
    },
    {
      "epoch": 6.215095520019531e-05,
      "step": 65170,
      "training_step_time": 6.694272756576538
    },
    {
      "epoch": 6.215190887451172e-05,
      "model_forward_time": 1.723982334136963,
      "step": 65171
    },
    {
      "epoch": 6.215190887451172e-05,
      "step": 65171,
      "training_step_time": 6.618741512298584
    },
    {
      "epoch": 6.215286254882812e-05,
      "model_forward_time": 1.7289209365844727,
      "step": 65172
    },
    {
      "epoch": 6.215286254882812e-05,
      "step": 65172,
      "training_step_time": 7.524102449417114
    },
    {
      "epoch": 6.215381622314453e-05,
      "model_forward_time": 1.7181665897369385,
      "step": 65173
    },
    {
      "epoch": 6.215381622314453e-05,
      "step": 65173,
      "training_step_time": 9.871747493743896
    },
    {
      "epoch": 6.215476989746094e-05,
      "model_forward_time": 1.7830393314361572,
      "step": 65174
    },
    {
      "epoch": 6.215476989746094e-05,
      "step": 65174,
      "training_step_time": 8.848354578018188
    },
    {
      "epoch": 6.215572357177734e-05,
      "model_forward_time": 1.3651647567749023,
      "step": 65175
    },
    {
      "epoch": 6.215572357177734e-05,
      "step": 65175,
      "training_step_time": 7.614412069320679
    },
    {
      "epoch": 6.215667724609375e-05,
      "model_forward_time": 1.7287564277648926,
      "step": 65176
    },
    {
      "epoch": 6.215667724609375e-05,
      "step": 65176,
      "training_step_time": 7.352063417434692
    },
    {
      "epoch": 6.215763092041016e-05,
      "model_forward_time": 1.7266452312469482,
      "step": 65177
    },
    {
      "epoch": 6.215763092041016e-05,
      "step": 65177,
      "training_step_time": 6.681814908981323
    },
    {
      "epoch": 6.215858459472656e-05,
      "model_forward_time": 1.7248635292053223,
      "step": 65178
    },
    {
      "epoch": 6.215858459472656e-05,
      "step": 65178,
      "training_step_time": 6.661994457244873
    },
    {
      "epoch": 6.215953826904297e-05,
      "model_forward_time": 1.7281770706176758,
      "step": 65179
    },
    {
      "epoch": 6.215953826904297e-05,
      "step": 65179,
      "training_step_time": 6.75561785697937
    },
    {
      "epoch": 6.216049194335938e-05,
      "grad_norm": 0.10269856452941895,
      "learning_rate": 2.9644539793496897e-06,
      "loss": 0.0577,
      "step": 65180
    },
    {
      "epoch": 6.216049194335938e-05,
      "model_forward_time": 1.7214717864990234,
      "step": 65180
    },
    {
      "dynamics_loss_avg": 0.05144036430865526,
      "epoch": 6.216049194335938e-05,
      "step": 65180
    },
    {
      "action_loss_avg": 0.001332674769219011,
      "epoch": 6.216049194335938e-05,
      "step": 65180
    },
    {
      "epoch": 6.216049194335938e-05,
      "step": 65180,
      "training_step_time": 6.7086451053619385
    },
    {
      "epoch": 6.216144561767578e-05,
      "model_forward_time": 1.722475528717041,
      "step": 65181
    },
    {
      "epoch": 6.216144561767578e-05,
      "step": 65181,
      "training_step_time": 8.431588888168335
    },
    {
      "epoch": 6.216239929199219e-05,
      "model_forward_time": 1.726252555847168,
      "step": 65182
    },
    {
      "epoch": 6.216239929199219e-05,
      "step": 65182,
      "training_step_time": 9.638463497161865
    },
    {
      "epoch": 6.21633529663086e-05,
      "model_forward_time": 3.002704620361328,
      "step": 65183
    },
    {
      "epoch": 6.21633529663086e-05,
      "step": 65183,
      "training_step_time": 10.655219793319702
    },
    {
      "epoch": 6.2164306640625e-05,
      "model_forward_time": 1.7211956977844238,
      "step": 65184
    },
    {
      "epoch": 6.2164306640625e-05,
      "step": 65184,
      "training_step_time": 7.621082067489624
    },
    {
      "epoch": 6.216526031494141e-05,
      "model_forward_time": 1.7240233421325684,
      "step": 65185
    },
    {
      "epoch": 6.216526031494141e-05,
      "step": 65185,
      "training_step_time": 7.508771181106567
    },
    {
      "epoch": 6.216621398925782e-05,
      "model_forward_time": 1.724113941192627,
      "step": 65186
    },
    {
      "epoch": 6.216621398925782e-05,
      "step": 65186,
      "training_step_time": 6.668219804763794
    },
    {
      "epoch": 6.216716766357422e-05,
      "model_forward_time": 1.7289209365844727,
      "step": 65187
    },
    {
      "epoch": 6.216716766357422e-05,
      "step": 65187,
      "training_step_time": 6.689610481262207
    },
    {
      "epoch": 6.216812133789063e-05,
      "model_forward_time": 1.7256615161895752,
      "step": 65188
    },
    {
      "epoch": 6.216812133789063e-05,
      "step": 65188,
      "training_step_time": 6.6961705684661865
    },
    {
      "epoch": 6.216907501220702e-05,
      "model_forward_time": 1.7189562320709229,
      "step": 65189
    },
    {
      "epoch": 6.216907501220702e-05,
      "step": 65189,
      "training_step_time": 6.732098817825317
    },
    {
      "epoch": 6.217002868652344e-05,
      "grad_norm": 0.10300562530755997,
      "learning_rate": 2.9629438452576807e-06,
      "loss": 0.0557,
      "step": 65190
    },
    {
      "epoch": 6.217002868652344e-05,
      "model_forward_time": 1.729365348815918,
      "step": 65190
    },
    {
      "dynamics_loss_avg": 0.052361204754561186,
      "epoch": 6.217002868652344e-05,
      "step": 65190
    },
    {
      "action_loss_avg": 0.0030156124354107305,
      "epoch": 6.217002868652344e-05,
      "step": 65190
    },
    {
      "epoch": 6.217002868652344e-05,
      "step": 65190,
      "training_step_time": 7.54238224029541
    },
    {
      "epoch": 6.217098236083985e-05,
      "model_forward_time": 1.767094612121582,
      "step": 65191
    },
    {
      "epoch": 6.217098236083985e-05,
      "step": 65191,
      "training_step_time": 8.817782402038574
    },
    {
      "epoch": 6.217193603515624e-05,
      "model_forward_time": 2.8803532123565674,
      "step": 65192
    },
    {
      "epoch": 6.217193603515624e-05,
      "step": 65192,
      "training_step_time": 11.393441915512085
    },
    {
      "epoch": 6.217288970947265e-05,
      "model_forward_time": 1.8288445472717285,
      "step": 65193
    },
    {
      "epoch": 6.217288970947265e-05,
      "step": 65193,
      "training_step_time": 7.736563444137573
    },
    {
      "epoch": 6.217384338378906e-05,
      "model_forward_time": 1.7310123443603516,
      "step": 65194
    },
    {
      "epoch": 6.217384338378906e-05,
      "step": 65194,
      "training_step_time": 7.578459024429321
    },
    {
      "epoch": 6.217479705810546e-05,
      "model_forward_time": 1.7274160385131836,
      "step": 65195
    },
    {
      "epoch": 6.217479705810546e-05,
      "step": 65195,
      "training_step_time": 6.745031833648682
    },
    {
      "epoch": 6.217575073242187e-05,
      "model_forward_time": 1.7335643768310547,
      "step": 65196
    },
    {
      "epoch": 6.217575073242187e-05,
      "step": 65196,
      "training_step_time": 6.716567039489746
    },
    {
      "epoch": 6.217670440673828e-05,
      "model_forward_time": 1.7320709228515625,
      "step": 65197
    },
    {
      "epoch": 6.217670440673828e-05,
      "step": 65197,
      "training_step_time": 6.776172399520874
    },
    {
      "epoch": 6.217765808105468e-05,
      "model_forward_time": 1.730940580368042,
      "step": 65198
    },
    {
      "epoch": 6.217765808105468e-05,
      "step": 65198,
      "training_step_time": 6.7452712059021
    },
    {
      "epoch": 6.217861175537109e-05,
      "model_forward_time": 1.7267191410064697,
      "step": 65199
    },
    {
      "epoch": 6.217861175537109e-05,
      "step": 65199,
      "training_step_time": 7.594361066818237
    },
    {
      "epoch": 6.21795654296875e-05,
      "grad_norm": 0.09265566617250443,
      "learning_rate": 2.96143393393507e-06,
      "loss": 0.0582,
      "step": 65200
    },
    {
      "epoch": 6.21795654296875e-05,
      "model_forward_time": 1.7249224185943604,
      "step": 65200
    },
    {
      "dynamics_loss_avg": 0.05076213367283344,
      "epoch": 6.21795654296875e-05,
      "step": 65200
    },
    {
      "action_loss_avg": 0.002475202950881794,
      "epoch": 6.21795654296875e-05,
      "step": 65200
    },
    {
      "epoch": 6.21795654296875e-05,
      "step": 65200,
      "training_step_time": 7.661572694778442
    },
    {
      "epoch": 6.21805191040039e-05,
      "model_forward_time": 2.8517863750457764,
      "step": 65201
    },
    {
      "epoch": 6.21805191040039e-05,
      "step": 65201,
      "training_step_time": 8.830425500869751
    },
    {
      "epoch": 6.218147277832031e-05,
      "model_forward_time": 1.8447117805480957,
      "step": 65202
    },
    {
      "epoch": 6.218147277832031e-05,
      "step": 65202,
      "training_step_time": 8.322296857833862
    },
    {
      "epoch": 6.218242645263672e-05,
      "model_forward_time": 1.7309985160827637,
      "step": 65203
    },
    {
      "epoch": 6.218242645263672e-05,
      "step": 65203,
      "training_step_time": 7.7946062088012695
    },
    {
      "epoch": 6.218338012695312e-05,
      "model_forward_time": 1.7262470722198486,
      "step": 65204
    },
    {
      "epoch": 6.218338012695312e-05,
      "step": 65204,
      "training_step_time": 6.830693960189819
    },
    {
      "epoch": 6.218433380126953e-05,
      "model_forward_time": 1.7539145946502686,
      "step": 65205
    },
    {
      "epoch": 6.218433380126953e-05,
      "step": 65205,
      "training_step_time": 6.834891319274902
    },
    {
      "epoch": 6.218528747558594e-05,
      "model_forward_time": 1.7279863357543945,
      "step": 65206
    },
    {
      "epoch": 6.218528747558594e-05,
      "step": 65206,
      "training_step_time": 6.811278820037842
    },
    {
      "epoch": 6.218624114990234e-05,
      "model_forward_time": 1.7265598773956299,
      "step": 65207
    },
    {
      "epoch": 6.218624114990234e-05,
      "step": 65207,
      "training_step_time": 6.776164531707764
    },
    {
      "epoch": 6.218719482421875e-05,
      "model_forward_time": 1.7257921695709229,
      "step": 65208
    },
    {
      "epoch": 6.218719482421875e-05,
      "step": 65208,
      "training_step_time": 7.610492706298828
    },
    {
      "epoch": 6.218814849853516e-05,
      "model_forward_time": 1.7255349159240723,
      "step": 65209
    },
    {
      "epoch": 6.218814849853516e-05,
      "step": 65209,
      "training_step_time": 8.352611064910889
    },
    {
      "epoch": 6.218910217285156e-05,
      "grad_norm": 0.09775897115468979,
      "learning_rate": 2.959924245546979e-06,
      "loss": 0.0556,
      "step": 65210
    },
    {
      "epoch": 6.218910217285156e-05,
      "model_forward_time": 2.8624398708343506,
      "step": 65210
    },
    {
      "dynamics_loss_avg": 0.058550532162189486,
      "epoch": 6.218910217285156e-05,
      "step": 65210
    },
    {
      "action_loss_avg": 0.0014502864447422325,
      "epoch": 6.218910217285156e-05,
      "step": 65210
    },
    {
      "epoch": 6.218910217285156e-05,
      "step": 65210,
      "training_step_time": 9.718009233474731
    },
    {
      "epoch": 6.219005584716797e-05,
      "model_forward_time": 1.7996671199798584,
      "step": 65211
    },
    {
      "epoch": 6.219005584716797e-05,
      "step": 65211,
      "training_step_time": 7.770981073379517
    },
    {
      "epoch": 6.219100952148438e-05,
      "model_forward_time": 1.7268614768981934,
      "step": 65212
    },
    {
      "epoch": 6.219100952148438e-05,
      "step": 65212,
      "training_step_time": 7.7392072677612305
    },
    {
      "epoch": 6.219196319580078e-05,
      "model_forward_time": 1.7273938655853271,
      "step": 65213
    },
    {
      "epoch": 6.219196319580078e-05,
      "step": 65213,
      "training_step_time": 7.557352781295776
    },
    {
      "epoch": 6.219291687011719e-05,
      "model_forward_time": 1.7275078296661377,
      "step": 65214
    },
    {
      "epoch": 6.219291687011719e-05,
      "step": 65214,
      "training_step_time": 6.726219654083252
    },
    {
      "epoch": 6.21938705444336e-05,
      "model_forward_time": 1.7381939888000488,
      "step": 65215
    },
    {
      "epoch": 6.21938705444336e-05,
      "step": 65215,
      "training_step_time": 6.769274473190308
    },
    {
      "epoch": 6.219482421875e-05,
      "model_forward_time": 1.723846197128296,
      "step": 65216
    },
    {
      "epoch": 6.219482421875e-05,
      "step": 65216,
      "training_step_time": 6.741839408874512
    },
    {
      "epoch": 6.21957778930664e-05,
      "model_forward_time": 1.7248013019561768,
      "step": 65217
    },
    {
      "epoch": 6.21957778930664e-05,
      "step": 65217,
      "training_step_time": 7.511955976486206
    },
    {
      "epoch": 6.219673156738282e-05,
      "model_forward_time": 1.7278766632080078,
      "step": 65218
    },
    {
      "epoch": 6.219673156738282e-05,
      "step": 65218,
      "training_step_time": 8.644686222076416
    },
    {
      "epoch": 6.219768524169921e-05,
      "model_forward_time": 1.7262864112854004,
      "step": 65219
    },
    {
      "epoch": 6.219768524169921e-05,
      "step": 65219,
      "training_step_time": 8.703405857086182
    },
    {
      "epoch": 6.219863891601562e-05,
      "grad_norm": 0.10050475597381592,
      "learning_rate": 2.958414780258504e-06,
      "loss": 0.0585,
      "step": 65220
    },
    {
      "epoch": 6.219863891601562e-05,
      "model_forward_time": 1.7750325202941895,
      "step": 65220
    },
    {
      "dynamics_loss_avg": 0.05297771953046322,
      "epoch": 6.219863891601562e-05,
      "step": 65220
    },
    {
      "action_loss_avg": 0.0023940506420331077,
      "epoch": 6.219863891601562e-05,
      "step": 65220
    },
    {
      "epoch": 6.219863891601562e-05,
      "step": 65220,
      "training_step_time": 10.698614835739136
    },
    {
      "epoch": 6.219959259033203e-05,
      "model_forward_time": 1.7247672080993652,
      "step": 65221
    },
    {
      "epoch": 6.219959259033203e-05,
      "step": 65221,
      "training_step_time": 8.466564893722534
    },
    {
      "epoch": 6.220054626464843e-05,
      "model_forward_time": 1.7286880016326904,
      "step": 65222
    },
    {
      "epoch": 6.220054626464843e-05,
      "step": 65222,
      "training_step_time": 7.538286447525024
    },
    {
      "epoch": 6.220149993896484e-05,
      "model_forward_time": 1.7274324893951416,
      "step": 65223
    },
    {
      "epoch": 6.220149993896484e-05,
      "step": 65223,
      "training_step_time": 6.701141357421875
    },
    {
      "epoch": 6.220245361328125e-05,
      "model_forward_time": 1.7269833087921143,
      "step": 65224
    },
    {
      "epoch": 6.220245361328125e-05,
      "step": 65224,
      "training_step_time": 6.6901092529296875
    },
    {
      "epoch": 6.220340728759765e-05,
      "model_forward_time": 1.725950002670288,
      "step": 65225
    },
    {
      "epoch": 6.220340728759765e-05,
      "step": 65225,
      "training_step_time": 6.722473859786987
    },
    {
      "epoch": 6.220436096191406e-05,
      "model_forward_time": 1.7248446941375732,
      "step": 65226
    },
    {
      "epoch": 6.220436096191406e-05,
      "step": 65226,
      "training_step_time": 7.546973705291748
    },
    {
      "epoch": 6.220531463623047e-05,
      "model_forward_time": 1.7269954681396484,
      "step": 65227
    },
    {
      "epoch": 6.220531463623047e-05,
      "step": 65227,
      "training_step_time": 7.696199178695679
    },
    {
      "epoch": 6.220626831054687e-05,
      "model_forward_time": 1.737130880355835,
      "step": 65228
    },
    {
      "epoch": 6.220626831054687e-05,
      "step": 65228,
      "training_step_time": 7.7161643505096436
    },
    {
      "epoch": 6.220722198486328e-05,
      "model_forward_time": 2.9251458644866943,
      "step": 65229
    },
    {
      "epoch": 6.220722198486328e-05,
      "step": 65229,
      "training_step_time": 9.02934193611145
    },
    {
      "epoch": 6.220817565917969e-05,
      "grad_norm": 0.11951124668121338,
      "learning_rate": 2.9569055382347213e-06,
      "loss": 0.0579,
      "step": 65230
    },
    {
      "epoch": 6.220817565917969e-05,
      "model_forward_time": 1.7240073680877686,
      "step": 65230
    },
    {
      "dynamics_loss_avg": 0.056292242370545864,
      "epoch": 6.220817565917969e-05,
      "step": 65230
    },
    {
      "action_loss_avg": 0.002406581083778292,
      "epoch": 6.220817565917969e-05,
      "step": 65230
    },
    {
      "epoch": 6.220817565917969e-05,
      "step": 65230,
      "training_step_time": 7.6746110916137695
    },
    {
      "epoch": 6.220912933349609e-05,
      "model_forward_time": 1.735027551651001,
      "step": 65231
    },
    {
      "epoch": 6.220912933349609e-05,
      "step": 65231,
      "training_step_time": 7.549601793289185
    },
    {
      "epoch": 6.22100830078125e-05,
      "model_forward_time": 1.7197208404541016,
      "step": 65232
    },
    {
      "epoch": 6.22100830078125e-05,
      "step": 65232,
      "training_step_time": 6.716161727905273
    },
    {
      "epoch": 6.221103668212891e-05,
      "model_forward_time": 1.7407200336456299,
      "step": 65233
    },
    {
      "epoch": 6.221103668212891e-05,
      "step": 65233,
      "training_step_time": 6.708788633346558
    },
    {
      "epoch": 6.221199035644531e-05,
      "model_forward_time": 1.7239627838134766,
      "step": 65234
    },
    {
      "epoch": 6.221199035644531e-05,
      "step": 65234,
      "training_step_time": 6.71481990814209
    },
    {
      "epoch": 6.221294403076172e-05,
      "model_forward_time": 1.725597620010376,
      "step": 65235
    },
    {
      "epoch": 6.221294403076172e-05,
      "step": 65235,
      "training_step_time": 7.526087284088135
    },
    {
      "epoch": 6.221389770507813e-05,
      "model_forward_time": 1.7673718929290771,
      "step": 65236
    },
    {
      "epoch": 6.221389770507813e-05,
      "step": 65236,
      "training_step_time": 7.588894605636597
    },
    {
      "epoch": 6.221485137939453e-05,
      "model_forward_time": 1.741046667098999,
      "step": 65237
    },
    {
      "epoch": 6.221485137939453e-05,
      "step": 65237,
      "training_step_time": 9.041704416275024
    },
    {
      "epoch": 6.221580505371094e-05,
      "model_forward_time": 2.9317822456359863,
      "step": 65238
    },
    {
      "epoch": 6.221580505371094e-05,
      "step": 65238,
      "training_step_time": 10.101380825042725
    },
    {
      "epoch": 6.221675872802735e-05,
      "model_forward_time": 1.8362815380096436,
      "step": 65239
    },
    {
      "epoch": 6.221675872802735e-05,
      "step": 65239,
      "training_step_time": 8.270284175872803
    },
    {
      "epoch": 6.221771240234375e-05,
      "grad_norm": 0.08991904556751251,
      "learning_rate": 2.9553965196406765e-06,
      "loss": 0.0592,
      "step": 65240
    },
    {
      "epoch": 6.221771240234375e-05,
      "model_forward_time": 1.7248799800872803,
      "step": 65240
    },
    {
      "dynamics_loss_avg": 0.05747663490474224,
      "epoch": 6.221771240234375e-05,
      "step": 65240
    },
    {
      "action_loss_avg": 0.0019596913538407535,
      "epoch": 6.221771240234375e-05,
      "step": 65240
    },
    {
      "epoch": 6.221771240234375e-05,
      "step": 65240,
      "training_step_time": 7.594725131988525
    },
    {
      "epoch": 6.221866607666016e-05,
      "model_forward_time": 1.729072093963623,
      "step": 65241
    },
    {
      "epoch": 6.221866607666016e-05,
      "step": 65241,
      "training_step_time": 6.673948764801025
    },
    {
      "epoch": 6.221961975097657e-05,
      "model_forward_time": 1.7276806831359863,
      "step": 65242
    },
    {
      "epoch": 6.221961975097657e-05,
      "step": 65242,
      "training_step_time": 6.571341037750244
    },
    {
      "epoch": 6.222057342529296e-05,
      "model_forward_time": 1.7321994304656982,
      "step": 65243
    },
    {
      "epoch": 6.222057342529296e-05,
      "step": 65243,
      "training_step_time": 6.68886923789978
    },
    {
      "epoch": 6.222152709960937e-05,
      "model_forward_time": 1.7230072021484375,
      "step": 65244
    },
    {
      "epoch": 6.222152709960937e-05,
      "step": 65244,
      "training_step_time": 7.4983813762664795
    },
    {
      "epoch": 6.222248077392579e-05,
      "model_forward_time": 1.7292499542236328,
      "step": 65245
    },
    {
      "epoch": 6.222248077392579e-05,
      "step": 65245,
      "training_step_time": 7.58113694190979
    },
    {
      "epoch": 6.222343444824218e-05,
      "model_forward_time": 1.7323951721191406,
      "step": 65246
    },
    {
      "epoch": 6.222343444824218e-05,
      "step": 65246,
      "training_step_time": 7.623852252960205
    },
    {
      "epoch": 6.22243881225586e-05,
      "model_forward_time": 2.860703468322754,
      "step": 65247
    },
    {
      "epoch": 6.22243881225586e-05,
      "step": 65247,
      "training_step_time": 8.277984857559204
    },
    {
      "epoch": 6.2225341796875e-05,
      "model_forward_time": 1.81514310836792,
      "step": 65248
    },
    {
      "epoch": 6.2225341796875e-05,
      "step": 65248,
      "training_step_time": 8.966476917266846
    },
    {
      "epoch": 6.22262954711914e-05,
      "model_forward_time": 1.728140115737915,
      "step": 65249
    },
    {
      "epoch": 6.22262954711914e-05,
      "step": 65249,
      "training_step_time": 7.721492290496826
    },
    {
      "epoch": 6.222724914550781e-05,
      "grad_norm": 0.11760187149047852,
      "learning_rate": 2.9538877246413943e-06,
      "loss": 0.0581,
      "step": 65250
    },
    {
      "epoch": 6.222724914550781e-05,
      "model_forward_time": 1.7249195575714111,
      "step": 65250
    },
    {
      "dynamics_loss_avg": 0.0557730671018362,
      "epoch": 6.222724914550781e-05,
      "step": 65250
    },
    {
      "action_loss_avg": 0.0015475142310606316,
      "epoch": 6.222724914550781e-05,
      "step": 65250
    },
    {
      "epoch": 6.222724914550781e-05,
      "step": 65250,
      "training_step_time": 7.123861312866211
    },
    {
      "epoch": 6.222820281982422e-05,
      "model_forward_time": 1.7244765758514404,
      "step": 65251
    },
    {
      "epoch": 6.222820281982422e-05,
      "step": 65251,
      "training_step_time": 6.793687105178833
    },
    {
      "epoch": 6.222915649414062e-05,
      "model_forward_time": 1.7270891666412354,
      "step": 65252
    },
    {
      "epoch": 6.222915649414062e-05,
      "step": 65252,
      "training_step_time": 6.759479284286499
    },
    {
      "epoch": 6.223011016845703e-05,
      "model_forward_time": 1.719491958618164,
      "step": 65253
    },
    {
      "epoch": 6.223011016845703e-05,
      "step": 65253,
      "training_step_time": 7.504610538482666
    },
    {
      "epoch": 6.223106384277344e-05,
      "model_forward_time": 1.7261600494384766,
      "step": 65254
    },
    {
      "epoch": 6.223106384277344e-05,
      "step": 65254,
      "training_step_time": 7.58351993560791
    },
    {
      "epoch": 6.223201751708984e-05,
      "model_forward_time": 1.7670211791992188,
      "step": 65255
    },
    {
      "epoch": 6.223201751708984e-05,
      "step": 65255,
      "training_step_time": 8.661263227462769
    },
    {
      "epoch": 6.223297119140625e-05,
      "model_forward_time": 2.881207227706909,
      "step": 65256
    },
    {
      "epoch": 6.223297119140625e-05,
      "step": 65256,
      "training_step_time": 9.331695556640625
    },
    {
      "epoch": 6.223392486572266e-05,
      "model_forward_time": 1.7583043575286865,
      "step": 65257
    },
    {
      "epoch": 6.223392486572266e-05,
      "step": 65257,
      "training_step_time": 10.413408756256104
    },
    {
      "epoch": 6.223487854003906e-05,
      "model_forward_time": 1.7257964611053467,
      "step": 65258
    },
    {
      "epoch": 6.223487854003906e-05,
      "step": 65258,
      "training_step_time": 7.711611270904541
    },
    {
      "epoch": 6.223583221435547e-05,
      "model_forward_time": 1.7200732231140137,
      "step": 65259
    },
    {
      "epoch": 6.223583221435547e-05,
      "step": 65259,
      "training_step_time": 7.063372611999512
    },
    {
      "epoch": 6.223678588867188e-05,
      "grad_norm": 0.11870060116052628,
      "learning_rate": 2.9523791534018743e-06,
      "loss": 0.0589,
      "step": 65260
    },
    {
      "epoch": 6.223678588867188e-05,
      "model_forward_time": 1.728140115737915,
      "step": 65260
    },
    {
      "dynamics_loss_avg": 0.05403833352029323,
      "epoch": 6.223678588867188e-05,
      "step": 65260
    },
    {
      "action_loss_avg": 0.0030667831655591727,
      "epoch": 6.223678588867188e-05,
      "step": 65260
    },
    {
      "epoch": 6.223678588867188e-05,
      "step": 65260,
      "training_step_time": 6.841019630432129
    },
    {
      "epoch": 6.223773956298828e-05,
      "model_forward_time": 1.724412441253662,
      "step": 65261
    },
    {
      "epoch": 6.223773956298828e-05,
      "step": 65261,
      "training_step_time": 6.853521823883057
    },
    {
      "epoch": 6.223869323730469e-05,
      "model_forward_time": 1.7177166938781738,
      "step": 65262
    },
    {
      "epoch": 6.223869323730469e-05,
      "step": 65262,
      "training_step_time": 7.66628623008728
    },
    {
      "epoch": 6.22396469116211e-05,
      "model_forward_time": 1.7258241176605225,
      "step": 65263
    },
    {
      "epoch": 6.22396469116211e-05,
      "step": 65263,
      "training_step_time": 7.705439805984497
    },
    {
      "epoch": 6.22406005859375e-05,
      "model_forward_time": 1.7193715572357178,
      "step": 65264
    },
    {
      "epoch": 6.22406005859375e-05,
      "step": 65264,
      "training_step_time": 7.777207374572754
    },
    {
      "epoch": 6.224155426025391e-05,
      "model_forward_time": 2.834000825881958,
      "step": 65265
    },
    {
      "epoch": 6.224155426025391e-05,
      "step": 65265,
      "training_step_time": 8.411119222640991
    },
    {
      "epoch": 6.224250793457032e-05,
      "model_forward_time": 1.724841833114624,
      "step": 65266
    },
    {
      "epoch": 6.224250793457032e-05,
      "step": 65266,
      "training_step_time": 7.728532791137695
    },
    {
      "epoch": 6.224346160888671e-05,
      "model_forward_time": 1.7298882007598877,
      "step": 65267
    },
    {
      "epoch": 6.224346160888671e-05,
      "step": 65267,
      "training_step_time": 7.698651075363159
    },
    {
      "epoch": 6.224441528320313e-05,
      "model_forward_time": 1.7285935878753662,
      "step": 65268
    },
    {
      "epoch": 6.224441528320313e-05,
      "step": 65268,
      "training_step_time": 7.038725137710571
    },
    {
      "epoch": 6.224536895751954e-05,
      "model_forward_time": 1.7291455268859863,
      "step": 65269
    },
    {
      "epoch": 6.224536895751954e-05,
      "step": 65269,
      "training_step_time": 6.733568906784058
    },
    {
      "epoch": 6.224632263183593e-05,
      "grad_norm": 0.10599233210086823,
      "learning_rate": 2.9508708060870917e-06,
      "loss": 0.0607,
      "step": 65270
    },
    {
      "epoch": 6.224632263183593e-05,
      "model_forward_time": 1.727595329284668,
      "step": 65270
    },
    {
      "dynamics_loss_avg": 0.05357330348342657,
      "epoch": 6.224632263183593e-05,
      "step": 65270
    },
    {
      "action_loss_avg": 0.0020704006019514054,
      "epoch": 6.224632263183593e-05,
      "step": 65270
    },
    {
      "epoch": 6.224632263183593e-05,
      "step": 65270,
      "training_step_time": 6.728929281234741
    },
    {
      "epoch": 6.224727630615234e-05,
      "model_forward_time": 1.7246170043945312,
      "step": 65271
    },
    {
      "epoch": 6.224727630615234e-05,
      "step": 65271,
      "training_step_time": 7.381440877914429
    },
    {
      "epoch": 6.224822998046876e-05,
      "model_forward_time": 1.7270584106445312,
      "step": 65272
    },
    {
      "epoch": 6.224822998046876e-05,
      "step": 65272,
      "training_step_time": 7.515363931655884
    },
    {
      "epoch": 6.224918365478515e-05,
      "model_forward_time": 1.7226176261901855,
      "step": 65273
    },
    {
      "epoch": 6.224918365478515e-05,
      "step": 65273,
      "training_step_time": 7.617583513259888
    },
    {
      "epoch": 6.225013732910156e-05,
      "model_forward_time": 1.7259266376495361,
      "step": 65274
    },
    {
      "epoch": 6.225013732910156e-05,
      "step": 65274,
      "training_step_time": 8.329115390777588
    },
    {
      "epoch": 6.225109100341797e-05,
      "model_forward_time": 2.939110040664673,
      "step": 65275
    },
    {
      "epoch": 6.225109100341797e-05,
      "step": 65275,
      "training_step_time": 7.682274103164673
    },
    {
      "epoch": 6.225204467773437e-05,
      "model_forward_time": 1.8359763622283936,
      "step": 65276
    },
    {
      "epoch": 6.225204467773437e-05,
      "step": 65276,
      "training_step_time": 7.645571947097778
    },
    {
      "epoch": 6.225299835205078e-05,
      "model_forward_time": 1.3896260261535645,
      "step": 65277
    },
    {
      "epoch": 6.225299835205078e-05,
      "step": 65277,
      "training_step_time": 7.329564332962036
    },
    {
      "epoch": 6.225395202636719e-05,
      "model_forward_time": 1.7260820865631104,
      "step": 65278
    },
    {
      "epoch": 6.225395202636719e-05,
      "step": 65278,
      "training_step_time": 6.991680383682251
    },
    {
      "epoch": 6.225490570068359e-05,
      "model_forward_time": 1.7278838157653809,
      "step": 65279
    },
    {
      "epoch": 6.225490570068359e-05,
      "step": 65279,
      "training_step_time": 6.7172346115112305
    },
    {
      "epoch": 6.2255859375e-05,
      "grad_norm": 0.08748793601989746,
      "learning_rate": 2.949362682861995e-06,
      "loss": 0.0595,
      "step": 65280
    },
    {
      "epoch": 6.2255859375e-05,
      "model_forward_time": 1.7256262302398682,
      "step": 65280
    },
    {
      "dynamics_loss_avg": 0.04903568886220455,
      "epoch": 6.2255859375e-05,
      "step": 65280
    },
    {
      "action_loss_avg": 0.0014648928328824696,
      "epoch": 6.2255859375e-05,
      "step": 65280
    },
    {
      "epoch": 6.2255859375e-05,
      "step": 65280,
      "training_step_time": 7.4824464321136475
    },
    {
      "epoch": 6.225681304931641e-05,
      "model_forward_time": 1.7273333072662354,
      "step": 65281
    },
    {
      "epoch": 6.225681304931641e-05,
      "step": 65281,
      "training_step_time": 7.182267904281616
    },
    {
      "epoch": 6.225776672363281e-05,
      "model_forward_time": 1.7244508266448975,
      "step": 65282
    },
    {
      "epoch": 6.225776672363281e-05,
      "step": 65282,
      "training_step_time": 8.640116214752197
    },
    {
      "epoch": 6.225872039794922e-05,
      "model_forward_time": 1.73215651512146,
      "step": 65283
    },
    {
      "epoch": 6.225872039794922e-05,
      "step": 65283,
      "training_step_time": 8.643808841705322
    },
    {
      "epoch": 6.225967407226563e-05,
      "model_forward_time": 2.9513397216796875,
      "step": 65284
    },
    {
      "epoch": 6.225967407226563e-05,
      "step": 65284,
      "training_step_time": 10.995234727859497
    },
    {
      "epoch": 6.226062774658203e-05,
      "model_forward_time": 1.8328237533569336,
      "step": 65285
    },
    {
      "epoch": 6.226062774658203e-05,
      "step": 65285,
      "training_step_time": 7.760365009307861
    },
    {
      "epoch": 6.226158142089844e-05,
      "model_forward_time": 1.7220396995544434,
      "step": 65286
    },
    {
      "epoch": 6.226158142089844e-05,
      "step": 65286,
      "training_step_time": 7.414576768875122
    },
    {
      "epoch": 6.226253509521485e-05,
      "model_forward_time": 1.7215371131896973,
      "step": 65287
    },
    {
      "epoch": 6.226253509521485e-05,
      "step": 65287,
      "training_step_time": 7.0667641162872314
    },
    {
      "epoch": 6.226348876953125e-05,
      "model_forward_time": 1.7236313819885254,
      "step": 65288
    },
    {
      "epoch": 6.226348876953125e-05,
      "step": 65288,
      "training_step_time": 6.835281848907471
    },
    {
      "epoch": 6.226444244384766e-05,
      "model_forward_time": 1.7226479053497314,
      "step": 65289
    },
    {
      "epoch": 6.226444244384766e-05,
      "step": 65289,
      "training_step_time": 7.471843957901001
    },
    {
      "epoch": 6.226539611816407e-05,
      "grad_norm": 0.10383783280849457,
      "learning_rate": 2.9478547838915138e-06,
      "loss": 0.055,
      "step": 65290
    },
    {
      "epoch": 6.226539611816407e-05,
      "model_forward_time": 1.3596711158752441,
      "step": 65290
    },
    {
      "dynamics_loss_avg": 0.053077207878232,
      "epoch": 6.226539611816407e-05,
      "step": 65290
    },
    {
      "action_loss_avg": 0.0017639814934227616,
      "epoch": 6.226539611816407e-05,
      "step": 65290
    },
    {
      "epoch": 6.226539611816407e-05,
      "step": 65290,
      "training_step_time": 7.582714557647705
    },
    {
      "epoch": 6.226634979248047e-05,
      "model_forward_time": 1.7213184833526611,
      "step": 65291
    },
    {
      "epoch": 6.226634979248047e-05,
      "step": 65291,
      "training_step_time": 7.746906042098999
    },
    {
      "epoch": 6.226730346679688e-05,
      "model_forward_time": 1.7236871719360352,
      "step": 65292
    },
    {
      "epoch": 6.226730346679688e-05,
      "step": 65292,
      "training_step_time": 8.60743236541748
    },
    {
      "epoch": 6.226825714111329e-05,
      "model_forward_time": 2.8701975345611572,
      "step": 65293
    },
    {
      "epoch": 6.226825714111329e-05,
      "step": 65293,
      "training_step_time": 7.762417554855347
    },
    {
      "epoch": 6.226921081542968e-05,
      "model_forward_time": 1.7974812984466553,
      "step": 65294
    },
    {
      "epoch": 6.226921081542968e-05,
      "step": 65294,
      "training_step_time": 8.273302555084229
    },
    {
      "epoch": 6.22701644897461e-05,
      "model_forward_time": 1.7245309352874756,
      "step": 65295
    },
    {
      "epoch": 6.22701644897461e-05,
      "step": 65295,
      "training_step_time": 7.675983190536499
    },
    {
      "epoch": 6.22711181640625e-05,
      "model_forward_time": 1.7235281467437744,
      "step": 65296
    },
    {
      "epoch": 6.22711181640625e-05,
      "step": 65296,
      "training_step_time": 7.014898777008057
    },
    {
      "epoch": 6.22720718383789e-05,
      "model_forward_time": 1.7190873622894287,
      "step": 65297
    },
    {
      "epoch": 6.22720718383789e-05,
      "step": 65297,
      "training_step_time": 6.714074611663818
    },
    {
      "epoch": 6.227302551269531e-05,
      "model_forward_time": 1.7270312309265137,
      "step": 65298
    },
    {
      "epoch": 6.227302551269531e-05,
      "step": 65298,
      "training_step_time": 7.404320240020752
    },
    {
      "epoch": 6.227397918701173e-05,
      "model_forward_time": 1.8708581924438477,
      "step": 65299
    },
    {
      "epoch": 6.227397918701173e-05,
      "step": 65299,
      "training_step_time": 7.523228406906128
    },
    {
      "epoch": 6.227493286132812e-05,
      "grad_norm": 0.08113595098257065,
      "learning_rate": 2.946347109340548e-06,
      "loss": 0.0579,
      "step": 65300
    },
    {
      "epoch": 6.227493286132812e-05,
      "model_forward_time": 1.8628804683685303,
      "step": 65300
    },
    {
      "dynamics_loss_avg": 0.0559745168313384,
      "epoch": 6.227493286132812e-05,
      "step": 65300
    },
    {
      "action_loss_avg": 0.0017222532362211496,
      "epoch": 6.227493286132812e-05,
      "step": 65300
    },
    {
      "epoch": 6.227493286132812e-05,
      "step": 65300,
      "training_step_time": 8.533790349960327
    },
    {
      "epoch": 6.227588653564453e-05,
      "model_forward_time": 1.8933863639831543,
      "step": 65301
    },
    {
      "epoch": 6.227588653564453e-05,
      "step": 65301,
      "training_step_time": 9.783081293106079
    },
    {
      "epoch": 6.227684020996094e-05,
      "model_forward_time": 2.9809303283691406,
      "step": 65302
    },
    {
      "epoch": 6.227684020996094e-05,
      "step": 65302,
      "training_step_time": 9.28591775894165
    },
    {
      "epoch": 6.227779388427734e-05,
      "model_forward_time": 1.8844616413116455,
      "step": 65303
    },
    {
      "epoch": 6.227779388427734e-05,
      "step": 65303,
      "training_step_time": 8.787940502166748
    },
    {
      "epoch": 6.227874755859375e-05,
      "model_forward_time": 1.9056775569915771,
      "step": 65304
    },
    {
      "epoch": 6.227874755859375e-05,
      "step": 65304,
      "training_step_time": 7.702845573425293
    },
    {
      "epoch": 6.227970123291016e-05,
      "model_forward_time": 1.8975694179534912,
      "step": 65305
    },
    {
      "epoch": 6.227970123291016e-05,
      "step": 65305,
      "training_step_time": 7.075241565704346
    },
    {
      "epoch": 6.228065490722656e-05,
      "model_forward_time": 1.8939743041992188,
      "step": 65306
    },
    {
      "epoch": 6.228065490722656e-05,
      "step": 65306,
      "training_step_time": 6.810003995895386
    },
    {
      "epoch": 6.228160858154297e-05,
      "model_forward_time": 1.8774211406707764,
      "step": 65307
    },
    {
      "epoch": 6.228160858154297e-05,
      "step": 65307,
      "training_step_time": 7.421354532241821
    },
    {
      "epoch": 6.228256225585938e-05,
      "model_forward_time": 1.8952343463897705,
      "step": 65308
    },
    {
      "epoch": 6.228256225585938e-05,
      "step": 65308,
      "training_step_time": 7.6398231983184814
    },
    {
      "epoch": 6.228351593017578e-05,
      "model_forward_time": 1.9004578590393066,
      "step": 65309
    },
    {
      "epoch": 6.228351593017578e-05,
      "step": 65309,
      "training_step_time": 7.700466871261597
    },
    {
      "epoch": 6.228446960449219e-05,
      "grad_norm": 0.11480353772640228,
      "learning_rate": 2.9448396593739758e-06,
      "loss": 0.0584,
      "step": 65310
    },
    {
      "epoch": 6.228446960449219e-05,
      "model_forward_time": 1.8848483562469482,
      "step": 65310
    },
    {
      "dynamics_loss_avg": 0.045234722644090654,
      "epoch": 6.228446960449219e-05,
      "step": 65310
    },
    {
      "action_loss_avg": 0.001117222935135942,
      "epoch": 6.228446960449219e-05,
      "step": 65310
    },
    {
      "epoch": 6.228446960449219e-05,
      "step": 65310,
      "training_step_time": 8.170199155807495
    },
    {
      "epoch": 6.22854232788086e-05,
      "model_forward_time": 2.933736801147461,
      "step": 65311
    },
    {
      "epoch": 6.22854232788086e-05,
      "step": 65311,
      "training_step_time": 8.508382320404053
    },
    {
      "epoch": 6.2286376953125e-05,
      "model_forward_time": 1.7166802883148193,
      "step": 65312
    },
    {
      "epoch": 6.2286376953125e-05,
      "step": 65312,
      "training_step_time": 9.75688123703003
    },
    {
      "epoch": 6.228733062744141e-05,
      "model_forward_time": 1.7797813415527344,
      "step": 65313
    },
    {
      "epoch": 6.228733062744141e-05,
      "step": 65313,
      "training_step_time": 8.34827709197998
    },
    {
      "epoch": 6.22882843017578e-05,
      "model_forward_time": 1.71822190284729,
      "step": 65314
    },
    {
      "epoch": 6.22882843017578e-05,
      "step": 65314,
      "training_step_time": 7.339098930358887
    },
    {
      "epoch": 6.228923797607422e-05,
      "model_forward_time": 1.711972713470459,
      "step": 65315
    },
    {
      "epoch": 6.228923797607422e-05,
      "step": 65315,
      "training_step_time": 6.771821975708008
    },
    {
      "epoch": 6.229019165039063e-05,
      "model_forward_time": 1.707829475402832,
      "step": 65316
    },
    {
      "epoch": 6.229019165039063e-05,
      "step": 65316,
      "training_step_time": 7.386291980743408
    },
    {
      "epoch": 6.229114532470702e-05,
      "model_forward_time": 1.7301154136657715,
      "step": 65317
    },
    {
      "epoch": 6.229114532470702e-05,
      "step": 65317,
      "training_step_time": 7.568304538726807
    },
    {
      "epoch": 6.229209899902344e-05,
      "model_forward_time": 1.7126498222351074,
      "step": 65318
    },
    {
      "epoch": 6.229209899902344e-05,
      "step": 65318,
      "training_step_time": 7.521402835845947
    },
    {
      "epoch": 6.229305267333985e-05,
      "model_forward_time": 1.7303149700164795,
      "step": 65319
    },
    {
      "epoch": 6.229305267333985e-05,
      "step": 65319,
      "training_step_time": 8.25331163406372
    },
    {
      "epoch": 6.229400634765624e-05,
      "grad_norm": 0.10718725621700287,
      "learning_rate": 2.9433324341566473e-06,
      "loss": 0.057,
      "step": 65320
    },
    {
      "epoch": 6.229400634765624e-05,
      "model_forward_time": 1.711118221282959,
      "step": 65320
    },
    {
      "dynamics_loss_avg": 0.05699599664658308,
      "epoch": 6.229400634765624e-05,
      "step": 65320
    },
    {
      "action_loss_avg": 0.0027780914708273484,
      "epoch": 6.229400634765624e-05,
      "step": 65320
    },
    {
      "epoch": 6.229400634765624e-05,
      "step": 65320,
      "training_step_time": 8.294715166091919
    },
    {
      "epoch": 6.229496002197265e-05,
      "model_forward_time": 2.867002010345459,
      "step": 65321
    },
    {
      "epoch": 6.229496002197265e-05,
      "step": 65321,
      "training_step_time": 9.746296405792236
    },
    {
      "epoch": 6.229591369628907e-05,
      "model_forward_time": 1.7848584651947021,
      "step": 65322
    },
    {
      "epoch": 6.229591369628907e-05,
      "step": 65322,
      "training_step_time": 7.615241765975952
    },
    {
      "epoch": 6.229686737060546e-05,
      "model_forward_time": 1.7171647548675537,
      "step": 65323
    },
    {
      "epoch": 6.229686737060546e-05,
      "step": 65323,
      "training_step_time": 8.182621955871582
    },
    {
      "epoch": 6.229782104492187e-05,
      "model_forward_time": 1.7122514247894287,
      "step": 65324
    },
    {
      "epoch": 6.229782104492187e-05,
      "step": 65324,
      "training_step_time": 6.694617033004761
    },
    {
      "epoch": 6.229877471923828e-05,
      "model_forward_time": 1.714005947113037,
      "step": 65325
    },
    {
      "epoch": 6.229877471923828e-05,
      "step": 65325,
      "training_step_time": 7.368838310241699
    },
    {
      "epoch": 6.229972839355468e-05,
      "model_forward_time": 1.7576289176940918,
      "step": 65326
    },
    {
      "epoch": 6.229972839355468e-05,
      "step": 65326,
      "training_step_time": 7.470128059387207
    },
    {
      "epoch": 6.230068206787109e-05,
      "model_forward_time": 1.348052978515625,
      "step": 65327
    },
    {
      "epoch": 6.230068206787109e-05,
      "step": 65327,
      "training_step_time": 7.165898084640503
    },
    {
      "epoch": 6.23016357421875e-05,
      "model_forward_time": 1.7119226455688477,
      "step": 65328
    },
    {
      "epoch": 6.23016357421875e-05,
      "step": 65328,
      "training_step_time": 8.329962253570557
    },
    {
      "epoch": 6.23025894165039e-05,
      "model_forward_time": 1.7204434871673584,
      "step": 65329
    },
    {
      "epoch": 6.23025894165039e-05,
      "step": 65329,
      "training_step_time": 7.689175128936768
    },
    {
      "epoch": 6.230354309082031e-05,
      "grad_norm": 0.10095599293708801,
      "learning_rate": 2.9418254338533914e-06,
      "loss": 0.0569,
      "step": 65330
    },
    {
      "epoch": 6.230354309082031e-05,
      "model_forward_time": 2.8818068504333496,
      "step": 65330
    },
    {
      "dynamics_loss_avg": 0.06046391148120165,
      "epoch": 6.230354309082031e-05,
      "step": 65330
    },
    {
      "action_loss_avg": 0.0019347982772160322,
      "epoch": 6.230354309082031e-05,
      "step": 65330
    },
    {
      "epoch": 6.230354309082031e-05,
      "step": 65330,
      "training_step_time": 9.172704458236694
    },
    {
      "epoch": 6.230449676513672e-05,
      "model_forward_time": 1.7789349555969238,
      "step": 65331
    },
    {
      "epoch": 6.230449676513672e-05,
      "step": 65331,
      "training_step_time": 7.592187881469727
    },
    {
      "epoch": 6.230545043945312e-05,
      "model_forward_time": 1.708103895187378,
      "step": 65332
    },
    {
      "epoch": 6.230545043945312e-05,
      "step": 65332,
      "training_step_time": 7.619551420211792
    },
    {
      "epoch": 6.230640411376953e-05,
      "model_forward_time": 1.7120084762573242,
      "step": 65333
    },
    {
      "epoch": 6.230640411376953e-05,
      "step": 65333,
      "training_step_time": 7.343357801437378
    },
    {
      "epoch": 6.230735778808594e-05,
      "model_forward_time": 1.7201261520385742,
      "step": 65334
    },
    {
      "epoch": 6.230735778808594e-05,
      "step": 65334,
      "training_step_time": 7.369220733642578
    },
    {
      "epoch": 6.230831146240234e-05,
      "model_forward_time": 1.7205116748809814,
      "step": 65335
    },
    {
      "epoch": 6.230831146240234e-05,
      "step": 65335,
      "training_step_time": 8.324942827224731
    },
    {
      "epoch": 6.230926513671875e-05,
      "model_forward_time": 1.7086832523345947,
      "step": 65336
    },
    {
      "epoch": 6.230926513671875e-05,
      "step": 65336,
      "training_step_time": 7.150828123092651
    },
    {
      "epoch": 6.231021881103516e-05,
      "model_forward_time": 1.72981595993042,
      "step": 65337
    },
    {
      "epoch": 6.231021881103516e-05,
      "step": 65337,
      "training_step_time": 7.574571847915649
    },
    {
      "epoch": 6.231117248535156e-05,
      "model_forward_time": 1.7188177108764648,
      "step": 65338
    },
    {
      "epoch": 6.231117248535156e-05,
      "step": 65338,
      "training_step_time": 9.30683445930481
    },
    {
      "epoch": 6.231212615966797e-05,
      "model_forward_time": 2.830112934112549,
      "step": 65339
    },
    {
      "epoch": 6.231212615966797e-05,
      "step": 65339,
      "training_step_time": 8.441774368286133
    },
    {
      "epoch": 6.231307983398438e-05,
      "grad_norm": 0.11306215077638626,
      "learning_rate": 2.940318658629011e-06,
      "loss": 0.055,
      "step": 65340
    },
    {
      "epoch": 6.231307983398438e-05,
      "model_forward_time": 1.7750825881958008,
      "step": 65340
    },
    {
      "dynamics_loss_avg": 0.05591492652893067,
      "epoch": 6.231307983398438e-05,
      "step": 65340
    },
    {
      "action_loss_avg": 0.0031343803158961235,
      "epoch": 6.231307983398438e-05,
      "step": 65340
    },
    {
      "epoch": 6.231307983398438e-05,
      "step": 65340,
      "training_step_time": 8.520159721374512
    },
    {
      "epoch": 6.231403350830078e-05,
      "model_forward_time": 1.708503246307373,
      "step": 65341
    },
    {
      "epoch": 6.231403350830078e-05,
      "step": 65341,
      "training_step_time": 7.646782398223877
    },
    {
      "epoch": 6.231498718261719e-05,
      "model_forward_time": 1.7163796424865723,
      "step": 65342
    },
    {
      "epoch": 6.231498718261719e-05,
      "step": 65342,
      "training_step_time": 7.326735973358154
    },
    {
      "epoch": 6.23159408569336e-05,
      "model_forward_time": 1.7171237468719482,
      "step": 65343
    },
    {
      "epoch": 6.23159408569336e-05,
      "step": 65343,
      "training_step_time": 6.674144744873047
    },
    {
      "epoch": 6.231689453125e-05,
      "model_forward_time": 1.7199180126190186,
      "step": 65344
    },
    {
      "epoch": 6.231689453125e-05,
      "step": 65344,
      "training_step_time": 7.29894495010376
    },
    {
      "epoch": 6.23178482055664e-05,
      "model_forward_time": 1.715649127960205,
      "step": 65345
    },
    {
      "epoch": 6.23178482055664e-05,
      "step": 65345,
      "training_step_time": 7.500813961029053
    },
    {
      "epoch": 6.231880187988282e-05,
      "model_forward_time": 1.716773271560669,
      "step": 65346
    },
    {
      "epoch": 6.231880187988282e-05,
      "step": 65346,
      "training_step_time": 7.579294681549072
    },
    {
      "epoch": 6.231975555419921e-05,
      "model_forward_time": 1.7088468074798584,
      "step": 65347
    },
    {
      "epoch": 6.231975555419921e-05,
      "step": 65347,
      "training_step_time": 7.634414434432983
    },
    {
      "epoch": 6.232070922851562e-05,
      "model_forward_time": 2.8132457733154297,
      "step": 65348
    },
    {
      "epoch": 6.232070922851562e-05,
      "step": 65348,
      "training_step_time": 8.339149236679077
    },
    {
      "epoch": 6.232166290283203e-05,
      "model_forward_time": 1.78206205368042,
      "step": 65349
    },
    {
      "epoch": 6.232166290283203e-05,
      "step": 65349,
      "training_step_time": 7.68809175491333
    },
    {
      "epoch": 6.232261657714843e-05,
      "grad_norm": 0.27033552527427673,
      "learning_rate": 2.938812108648287e-06,
      "loss": 0.0596,
      "step": 65350
    },
    {
      "epoch": 6.232261657714843e-05,
      "model_forward_time": 1.7071785926818848,
      "step": 65350
    },
    {
      "dynamics_loss_avg": 0.05585596468299627,
      "epoch": 6.232261657714843e-05,
      "step": 65350
    },
    {
      "action_loss_avg": 0.0007921144613646902,
      "epoch": 6.232261657714843e-05,
      "step": 65350
    },
    {
      "epoch": 6.232261657714843e-05,
      "step": 65350,
      "training_step_time": 8.602094888687134
    },
    {
      "epoch": 6.232357025146484e-05,
      "model_forward_time": 1.722978115081787,
      "step": 65351
    },
    {
      "epoch": 6.232357025146484e-05,
      "step": 65351,
      "training_step_time": 7.450918674468994
    },
    {
      "epoch": 6.232452392578125e-05,
      "model_forward_time": 1.7209155559539795,
      "step": 65352
    },
    {
      "epoch": 6.232452392578125e-05,
      "step": 65352,
      "training_step_time": 6.711944341659546
    },
    {
      "epoch": 6.232547760009765e-05,
      "model_forward_time": 1.720475673675537,
      "step": 65353
    },
    {
      "epoch": 6.232547760009765e-05,
      "step": 65353,
      "training_step_time": 7.3486151695251465
    },
    {
      "epoch": 6.232643127441406e-05,
      "model_forward_time": 1.7103374004364014,
      "step": 65354
    },
    {
      "epoch": 6.232643127441406e-05,
      "step": 65354,
      "training_step_time": 7.571809768676758
    },
    {
      "epoch": 6.232738494873047e-05,
      "model_forward_time": 1.7114086151123047,
      "step": 65355
    },
    {
      "epoch": 6.232738494873047e-05,
      "step": 65355,
      "training_step_time": 8.840783596038818
    },
    {
      "epoch": 6.232833862304687e-05,
      "model_forward_time": 1.7294974327087402,
      "step": 65356
    },
    {
      "epoch": 6.232833862304687e-05,
      "step": 65356,
      "training_step_time": 7.6246442794799805
    },
    {
      "epoch": 6.232929229736328e-05,
      "model_forward_time": 1.7125110626220703,
      "step": 65357
    },
    {
      "epoch": 6.232929229736328e-05,
      "step": 65357,
      "training_step_time": 8.716671705245972
    },
    {
      "epoch": 6.233024597167969e-05,
      "model_forward_time": 1.7411985397338867,
      "step": 65358
    },
    {
      "epoch": 6.233024597167969e-05,
      "step": 65358,
      "training_step_time": 9.533491134643555
    },
    {
      "epoch": 6.233119964599609e-05,
      "model_forward_time": 1.7229843139648438,
      "step": 65359
    },
    {
      "epoch": 6.233119964599609e-05,
      "step": 65359,
      "training_step_time": 7.664906024932861
    },
    {
      "epoch": 6.23321533203125e-05,
      "grad_norm": 0.13559316098690033,
      "learning_rate": 2.937305784075972e-06,
      "loss": 0.0555,
      "step": 65360
    },
    {
      "epoch": 6.23321533203125e-05,
      "model_forward_time": 1.7048678398132324,
      "step": 65360
    },
    {
      "dynamics_loss_avg": 0.047168934065848586,
      "epoch": 6.23321533203125e-05,
      "step": 65360
    },
    {
      "action_loss_avg": 0.0010518503782805055,
      "epoch": 6.23321533203125e-05,
      "step": 65360
    },
    {
      "epoch": 6.23321533203125e-05,
      "step": 65360,
      "training_step_time": 7.645451307296753
    },
    {
      "epoch": 6.233310699462891e-05,
      "model_forward_time": 1.706049919128418,
      "step": 65361
    },
    {
      "epoch": 6.233310699462891e-05,
      "step": 65361,
      "training_step_time": 6.675001621246338
    },
    {
      "epoch": 6.233406066894531e-05,
      "model_forward_time": 1.7174546718597412,
      "step": 65362
    },
    {
      "epoch": 6.233406066894531e-05,
      "step": 65362,
      "training_step_time": 7.287112236022949
    },
    {
      "epoch": 6.233501434326172e-05,
      "model_forward_time": 1.7052500247955322,
      "step": 65363
    },
    {
      "epoch": 6.233501434326172e-05,
      "step": 65363,
      "training_step_time": 7.569917678833008
    },
    {
      "epoch": 6.233596801757813e-05,
      "model_forward_time": 1.7201697826385498,
      "step": 65364
    },
    {
      "epoch": 6.233596801757813e-05,
      "step": 65364,
      "training_step_time": 7.013336181640625
    },
    {
      "epoch": 6.233692169189453e-05,
      "model_forward_time": 1.7205889225006104,
      "step": 65365
    },
    {
      "epoch": 6.233692169189453e-05,
      "step": 65365,
      "training_step_time": 8.183483123779297
    },
    {
      "epoch": 6.233787536621094e-05,
      "model_forward_time": 1.7285091876983643,
      "step": 65366
    },
    {
      "epoch": 6.233787536621094e-05,
      "step": 65366,
      "training_step_time": 9.384010314941406
    },
    {
      "epoch": 6.233882904052735e-05,
      "model_forward_time": 1.7182295322418213,
      "step": 65367
    },
    {
      "epoch": 6.233882904052735e-05,
      "step": 65367,
      "training_step_time": 11.0283682346344
    },
    {
      "epoch": 6.233978271484375e-05,
      "model_forward_time": 1.7243330478668213,
      "step": 65368
    },
    {
      "epoch": 6.233978271484375e-05,
      "step": 65368,
      "training_step_time": 9.48021674156189
    },
    {
      "epoch": 6.234073638916016e-05,
      "model_forward_time": 1.71504545211792,
      "step": 65369
    },
    {
      "epoch": 6.234073638916016e-05,
      "step": 65369,
      "training_step_time": 7.650970458984375
    },
    {
      "epoch": 6.234169006347657e-05,
      "grad_norm": 0.130532905459404,
      "learning_rate": 2.935799685076794e-06,
      "loss": 0.0544,
      "step": 65370
    },
    {
      "epoch": 6.234169006347657e-05,
      "model_forward_time": 1.7076385021209717,
      "step": 65370
    },
    {
      "dynamics_loss_avg": 0.040011503454297784,
      "epoch": 6.234169006347657e-05,
      "step": 65370
    },
    {
      "action_loss_avg": 0.001698578443028964,
      "epoch": 6.234169006347657e-05,
      "step": 65370
    },
    {
      "epoch": 6.234169006347657e-05,
      "step": 65370,
      "training_step_time": 7.4783735275268555
    },
    {
      "epoch": 6.234264373779296e-05,
      "model_forward_time": 1.7102539539337158,
      "step": 65371
    },
    {
      "epoch": 6.234264373779296e-05,
      "step": 65371,
      "training_step_time": 7.361252546310425
    },
    {
      "epoch": 6.234359741210937e-05,
      "model_forward_time": 1.3420517444610596,
      "step": 65372
    },
    {
      "epoch": 6.234359741210937e-05,
      "step": 65372,
      "training_step_time": 7.631169557571411
    },
    {
      "epoch": 6.234455108642579e-05,
      "model_forward_time": 1.7210063934326172,
      "step": 65373
    },
    {
      "epoch": 6.234455108642579e-05,
      "step": 65373,
      "training_step_time": 6.700943946838379
    },
    {
      "epoch": 6.234550476074218e-05,
      "model_forward_time": 1.7213666439056396,
      "step": 65374
    },
    {
      "epoch": 6.234550476074218e-05,
      "step": 65374,
      "training_step_time": 7.531929969787598
    },
    {
      "epoch": 6.23464584350586e-05,
      "model_forward_time": 1.7198543548583984,
      "step": 65375
    },
    {
      "epoch": 6.23464584350586e-05,
      "step": 65375,
      "training_step_time": 8.190755128860474
    },
    {
      "epoch": 6.2347412109375e-05,
      "model_forward_time": 2.8605010509490967,
      "step": 65376
    },
    {
      "epoch": 6.2347412109375e-05,
      "step": 65376,
      "training_step_time": 9.958143472671509
    },
    {
      "epoch": 6.23483657836914e-05,
      "model_forward_time": 1.785337209701538,
      "step": 65377
    },
    {
      "epoch": 6.23483657836914e-05,
      "step": 65377,
      "training_step_time": 8.425837278366089
    },
    {
      "epoch": 6.234931945800781e-05,
      "model_forward_time": 1.7073519229888916,
      "step": 65378
    },
    {
      "epoch": 6.234931945800781e-05,
      "step": 65378,
      "training_step_time": 7.600275039672852
    },
    {
      "epoch": 6.235027313232422e-05,
      "model_forward_time": 1.7069823741912842,
      "step": 65379
    },
    {
      "epoch": 6.235027313232422e-05,
      "step": 65379,
      "training_step_time": 7.438031911849976
    },
    {
      "epoch": 6.235122680664062e-05,
      "grad_norm": 0.12460585683584213,
      "learning_rate": 2.9342938118154595e-06,
      "loss": 0.0539,
      "step": 65380
    },
    {
      "epoch": 6.235122680664062e-05,
      "model_forward_time": 1.7355365753173828,
      "step": 65380
    },
    {
      "dynamics_loss_avg": 0.04135533235967159,
      "epoch": 6.235122680664062e-05,
      "step": 65380
    },
    {
      "action_loss_avg": 0.0015389229782158508,
      "epoch": 6.235122680664062e-05,
      "step": 65380
    },
    {
      "epoch": 6.235122680664062e-05,
      "step": 65380,
      "training_step_time": 7.254552841186523
    },
    {
      "epoch": 6.235218048095703e-05,
      "model_forward_time": 1.7313258647918701,
      "step": 65381
    },
    {
      "epoch": 6.235218048095703e-05,
      "step": 65381,
      "training_step_time": 7.587921380996704
    },
    {
      "epoch": 6.235313415527344e-05,
      "model_forward_time": 1.736835241317749,
      "step": 65382
    },
    {
      "epoch": 6.235313415527344e-05,
      "step": 65382,
      "training_step_time": 6.802135229110718
    },
    {
      "epoch": 6.235408782958984e-05,
      "model_forward_time": 1.7180702686309814,
      "step": 65383
    },
    {
      "epoch": 6.235408782958984e-05,
      "step": 65383,
      "training_step_time": 7.618887186050415
    },
    {
      "epoch": 6.235504150390625e-05,
      "model_forward_time": 1.7700023651123047,
      "step": 65384
    },
    {
      "epoch": 6.235504150390625e-05,
      "step": 65384,
      "training_step_time": 7.650581359863281
    },
    {
      "epoch": 6.235599517822266e-05,
      "model_forward_time": 2.9375710487365723,
      "step": 65385
    },
    {
      "epoch": 6.235599517822266e-05,
      "step": 65385,
      "training_step_time": 9.559555768966675
    },
    {
      "epoch": 6.235694885253906e-05,
      "model_forward_time": 1.7871184349060059,
      "step": 65386
    },
    {
      "epoch": 6.235694885253906e-05,
      "step": 65386,
      "training_step_time": 7.725016117095947
    },
    {
      "epoch": 6.235790252685547e-05,
      "model_forward_time": 1.7388875484466553,
      "step": 65387
    },
    {
      "epoch": 6.235790252685547e-05,
      "step": 65387,
      "training_step_time": 7.711717367172241
    },
    {
      "epoch": 6.235885620117188e-05,
      "model_forward_time": 1.7326760292053223,
      "step": 65388
    },
    {
      "epoch": 6.235885620117188e-05,
      "step": 65388,
      "training_step_time": 7.527500629425049
    },
    {
      "epoch": 6.235980987548828e-05,
      "model_forward_time": 1.731294870376587,
      "step": 65389
    },
    {
      "epoch": 6.235980987548828e-05,
      "step": 65389,
      "training_step_time": 7.43913197517395
    },
    {
      "epoch": 6.236076354980469e-05,
      "grad_norm": 0.06782808154821396,
      "learning_rate": 2.9327881644566485e-06,
      "loss": 0.0538,
      "step": 65390
    },
    {
      "epoch": 6.236076354980469e-05,
      "model_forward_time": 1.7002041339874268,
      "step": 65390
    },
    {
      "dynamics_loss_avg": 0.059646681696176526,
      "epoch": 6.236076354980469e-05,
      "step": 65390
    },
    {
      "action_loss_avg": 0.0022145486436784267,
      "epoch": 6.236076354980469e-05,
      "step": 65390
    },
    {
      "epoch": 6.236076354980469e-05,
      "step": 65390,
      "training_step_time": 7.5179572105407715
    },
    {
      "epoch": 6.23617172241211e-05,
      "model_forward_time": 1.7257106304168701,
      "step": 65391
    },
    {
      "epoch": 6.23617172241211e-05,
      "step": 65391,
      "training_step_time": 6.765608549118042
    },
    {
      "epoch": 6.23626708984375e-05,
      "model_forward_time": 1.7125122547149658,
      "step": 65392
    },
    {
      "epoch": 6.23626708984375e-05,
      "step": 65392,
      "training_step_time": 8.511717319488525
    },
    {
      "epoch": 6.236362457275391e-05,
      "model_forward_time": 1.7217342853546143,
      "step": 65393
    },
    {
      "epoch": 6.236362457275391e-05,
      "step": 65393,
      "training_step_time": 7.774244070053101
    },
    {
      "epoch": 6.236457824707032e-05,
      "model_forward_time": 2.861936330795288,
      "step": 65394
    },
    {
      "epoch": 6.236457824707032e-05,
      "step": 65394,
      "training_step_time": 9.710472345352173
    },
    {
      "epoch": 6.236553192138671e-05,
      "model_forward_time": 1.7699999809265137,
      "step": 65395
    },
    {
      "epoch": 6.236553192138671e-05,
      "step": 65395,
      "training_step_time": 9.370402336120605
    },
    {
      "epoch": 6.236648559570313e-05,
      "model_forward_time": 1.7039873600006104,
      "step": 65396
    },
    {
      "epoch": 6.236648559570313e-05,
      "step": 65396,
      "training_step_time": 8.414170503616333
    },
    {
      "epoch": 6.236743927001954e-05,
      "model_forward_time": 1.7203233242034912,
      "step": 65397
    },
    {
      "epoch": 6.236743927001954e-05,
      "step": 65397,
      "training_step_time": 7.743142366409302
    },
    {
      "epoch": 6.236839294433593e-05,
      "model_forward_time": 1.3542497158050537,
      "step": 65398
    },
    {
      "epoch": 6.236839294433593e-05,
      "step": 65398,
      "training_step_time": 7.369598627090454
    },
    {
      "epoch": 6.236934661865234e-05,
      "model_forward_time": 1.701545000076294,
      "step": 65399
    },
    {
      "epoch": 6.236934661865234e-05,
      "step": 65399,
      "training_step_time": 7.346794128417969
    },
    {
      "epoch": 6.237030029296876e-05,
      "grad_norm": 0.07649680972099304,
      "learning_rate": 2.931282743165016e-06,
      "loss": 0.0561,
      "step": 65400
    },
    {
      "epoch": 6.237030029296876e-05,
      "model_forward_time": 1.7221755981445312,
      "step": 65400
    },
    {
      "dynamics_loss_avg": 0.057912289164960384,
      "epoch": 6.237030029296876e-05,
      "step": 65400
    },
    {
      "action_loss_avg": 0.0022169569798279553,
      "epoch": 6.237030029296876e-05,
      "step": 65400
    },
    {
      "epoch": 6.237030029296876e-05,
      "step": 65400,
      "training_step_time": 7.56066107749939
    },
    {
      "epoch": 6.237125396728515e-05,
      "model_forward_time": 1.7226548194885254,
      "step": 65401
    },
    {
      "epoch": 6.237125396728515e-05,
      "step": 65401,
      "training_step_time": 7.535658359527588
    },
    {
      "epoch": 6.237220764160156e-05,
      "model_forward_time": 1.7103066444396973,
      "step": 65402
    },
    {
      "epoch": 6.237220764160156e-05,
      "step": 65402,
      "training_step_time": 10.277729272842407
    },
    {
      "epoch": 6.237316131591797e-05,
      "model_forward_time": 2.8196635246276855,
      "step": 65403
    },
    {
      "epoch": 6.237316131591797e-05,
      "step": 65403,
      "training_step_time": 9.605833530426025
    },
    {
      "epoch": 6.237411499023437e-05,
      "model_forward_time": 1.7360203266143799,
      "step": 65404
    },
    {
      "epoch": 6.237411499023437e-05,
      "step": 65404,
      "training_step_time": 9.2161865234375
    },
    {
      "epoch": 6.237506866455078e-05,
      "model_forward_time": 1.7066075801849365,
      "step": 65405
    },
    {
      "epoch": 6.237506866455078e-05,
      "step": 65405,
      "training_step_time": 8.461925983428955
    },
    {
      "epoch": 6.237602233886719e-05,
      "model_forward_time": 1.7204184532165527,
      "step": 65406
    },
    {
      "epoch": 6.237602233886719e-05,
      "step": 65406,
      "training_step_time": 7.633355617523193
    },
    {
      "epoch": 6.237697601318359e-05,
      "model_forward_time": 1.7073938846588135,
      "step": 65407
    },
    {
      "epoch": 6.237697601318359e-05,
      "step": 65407,
      "training_step_time": 7.371975898742676
    },
    {
      "epoch": 6.23779296875e-05,
      "model_forward_time": 1.7204947471618652,
      "step": 65408
    },
    {
      "epoch": 6.23779296875e-05,
      "step": 65408,
      "training_step_time": 7.33555006980896
    },
    {
      "epoch": 6.237888336181641e-05,
      "model_forward_time": 1.7178375720977783,
      "step": 65409
    },
    {
      "epoch": 6.237888336181641e-05,
      "step": 65409,
      "training_step_time": 7.59539270401001
    },
    {
      "epoch": 6.237983703613281e-05,
      "grad_norm": 0.1042175218462944,
      "learning_rate": 2.92977754810519e-06,
      "loss": 0.0527,
      "step": 65410
    },
    {
      "epoch": 6.237983703613281e-05,
      "model_forward_time": 1.7292687892913818,
      "step": 65410
    },
    {
      "dynamics_loss_avg": 0.052495611272752284,
      "epoch": 6.237983703613281e-05,
      "step": 65410
    },
    {
      "action_loss_avg": 0.001577761975931935,
      "epoch": 6.237983703613281e-05,
      "step": 65410
    },
    {
      "epoch": 6.237983703613281e-05,
      "step": 65410,
      "training_step_time": 8.551311254501343
    },
    {
      "epoch": 6.238079071044922e-05,
      "model_forward_time": 1.7054507732391357,
      "step": 65411
    },
    {
      "epoch": 6.238079071044922e-05,
      "step": 65411,
      "training_step_time": 9.181207656860352
    },
    {
      "epoch": 6.238174438476563e-05,
      "model_forward_time": 2.811476230621338,
      "step": 65412
    },
    {
      "epoch": 6.238174438476563e-05,
      "step": 65412,
      "training_step_time": 9.377736330032349
    },
    {
      "epoch": 6.238269805908203e-05,
      "model_forward_time": 1.717839241027832,
      "step": 65413
    },
    {
      "epoch": 6.238269805908203e-05,
      "step": 65413,
      "training_step_time": 9.41268277168274
    },
    {
      "epoch": 6.238365173339844e-05,
      "model_forward_time": 1.7809512615203857,
      "step": 65414
    },
    {
      "epoch": 6.238365173339844e-05,
      "step": 65414,
      "training_step_time": 7.710808038711548
    },
    {
      "epoch": 6.238460540771485e-05,
      "model_forward_time": 1.762030839920044,
      "step": 65415
    },
    {
      "epoch": 6.238460540771485e-05,
      "step": 65415,
      "training_step_time": 7.578387260437012
    },
    {
      "epoch": 6.238555908203125e-05,
      "model_forward_time": 1.723405122756958,
      "step": 65416
    },
    {
      "epoch": 6.238555908203125e-05,
      "step": 65416,
      "training_step_time": 7.04032039642334
    },
    {
      "epoch": 6.238651275634766e-05,
      "model_forward_time": 1.7104334831237793,
      "step": 65417
    },
    {
      "epoch": 6.238651275634766e-05,
      "step": 65417,
      "training_step_time": 7.395474672317505
    },
    {
      "epoch": 6.238746643066407e-05,
      "model_forward_time": 1.7196810245513916,
      "step": 65418
    },
    {
      "epoch": 6.238746643066407e-05,
      "step": 65418,
      "training_step_time": 7.624927520751953
    },
    {
      "epoch": 6.238842010498047e-05,
      "model_forward_time": 1.7217333316802979,
      "step": 65419
    },
    {
      "epoch": 6.238842010498047e-05,
      "step": 65419,
      "training_step_time": 8.50492238998413
    },
    {
      "epoch": 6.238937377929688e-05,
      "grad_norm": 0.07674643397331238,
      "learning_rate": 2.928272579441781e-06,
      "loss": 0.0551,
      "step": 65420
    },
    {
      "epoch": 6.238937377929688e-05,
      "model_forward_time": 1.7100379467010498,
      "step": 65420
    },
    {
      "dynamics_loss_avg": 0.04405661411583424,
      "epoch": 6.238937377929688e-05,
      "step": 65420
    },
    {
      "action_loss_avg": 0.00206152236496564,
      "epoch": 6.238937377929688e-05,
      "step": 65420
    },
    {
      "epoch": 6.238937377929688e-05,
      "step": 65420,
      "training_step_time": 7.676026344299316
    },
    {
      "epoch": 6.239032745361329e-05,
      "model_forward_time": 1.725264072418213,
      "step": 65421
    },
    {
      "epoch": 6.239032745361329e-05,
      "step": 65421,
      "training_step_time": 9.304016351699829
    },
    {
      "epoch": 6.239128112792968e-05,
      "model_forward_time": 2.8978168964385986,
      "step": 65422
    },
    {
      "epoch": 6.239128112792968e-05,
      "step": 65422,
      "training_step_time": 7.726583242416382
    },
    {
      "epoch": 6.23922348022461e-05,
      "model_forward_time": 1.7841732501983643,
      "step": 65423
    },
    {
      "epoch": 6.23922348022461e-05,
      "step": 65423,
      "training_step_time": 7.705755710601807
    },
    {
      "epoch": 6.23931884765625e-05,
      "model_forward_time": 1.717024326324463,
      "step": 65424
    },
    {
      "epoch": 6.23931884765625e-05,
      "step": 65424,
      "training_step_time": 7.640130996704102
    },
    {
      "epoch": 6.23941421508789e-05,
      "model_forward_time": 1.3523509502410889,
      "step": 65425
    },
    {
      "epoch": 6.23941421508789e-05,
      "step": 65425,
      "training_step_time": 7.625400066375732
    },
    {
      "epoch": 6.239509582519531e-05,
      "model_forward_time": 1.7005348205566406,
      "step": 65426
    },
    {
      "epoch": 6.239509582519531e-05,
      "step": 65426,
      "training_step_time": 7.363337278366089
    },
    {
      "epoch": 6.239604949951173e-05,
      "model_forward_time": 1.7205040454864502,
      "step": 65427
    },
    {
      "epoch": 6.239604949951173e-05,
      "step": 65427,
      "training_step_time": 7.569480895996094
    },
    {
      "epoch": 6.239700317382812e-05,
      "model_forward_time": 1.7094740867614746,
      "step": 65428
    },
    {
      "epoch": 6.239700317382812e-05,
      "step": 65428,
      "training_step_time": 7.561776161193848
    },
    {
      "epoch": 6.239795684814453e-05,
      "model_forward_time": 1.7523984909057617,
      "step": 65429
    },
    {
      "epoch": 6.239795684814453e-05,
      "step": 65429,
      "training_step_time": 8.337218999862671
    },
    {
      "epoch": 6.239891052246094e-05,
      "grad_norm": 0.10809989273548126,
      "learning_rate": 2.9267678373393678e-06,
      "loss": 0.0542,
      "step": 65430
    },
    {
      "epoch": 6.239891052246094e-05,
      "model_forward_time": 1.722599983215332,
      "step": 65430
    },
    {
      "dynamics_loss_avg": 0.05190913639962673,
      "epoch": 6.239891052246094e-05,
      "step": 65430
    },
    {
      "action_loss_avg": 0.0012199643184430896,
      "epoch": 6.239891052246094e-05,
      "step": 65430
    },
    {
      "epoch": 6.239891052246094e-05,
      "step": 65430,
      "training_step_time": 8.727032899856567
    },
    {
      "epoch": 6.239986419677734e-05,
      "model_forward_time": 2.5249781608581543,
      "step": 65431
    },
    {
      "epoch": 6.239986419677734e-05,
      "step": 65431,
      "training_step_time": 8.701924085617065
    },
    {
      "epoch": 6.240081787109375e-05,
      "model_forward_time": 1.7910575866699219,
      "step": 65432
    },
    {
      "epoch": 6.240081787109375e-05,
      "step": 65432,
      "training_step_time": 7.741304397583008
    },
    {
      "epoch": 6.240177154541016e-05,
      "model_forward_time": 1.7180988788604736,
      "step": 65433
    },
    {
      "epoch": 6.240177154541016e-05,
      "step": 65433,
      "training_step_time": 7.54149055480957
    },
    {
      "epoch": 6.240272521972656e-05,
      "model_forward_time": 1.707026720046997,
      "step": 65434
    },
    {
      "epoch": 6.240272521972656e-05,
      "step": 65434,
      "training_step_time": 7.671088695526123
    },
    {
      "epoch": 6.240367889404297e-05,
      "model_forward_time": 1.7128889560699463,
      "step": 65435
    },
    {
      "epoch": 6.240367889404297e-05,
      "step": 65435,
      "training_step_time": 7.29679274559021
    },
    {
      "epoch": 6.240463256835938e-05,
      "model_forward_time": 1.700556755065918,
      "step": 65436
    },
    {
      "epoch": 6.240463256835938e-05,
      "step": 65436,
      "training_step_time": 7.468168258666992
    },
    {
      "epoch": 6.240558624267578e-05,
      "model_forward_time": 1.7055208683013916,
      "step": 65437
    },
    {
      "epoch": 6.240558624267578e-05,
      "step": 65437,
      "training_step_time": 7.5298240184783936
    },
    {
      "epoch": 6.240653991699219e-05,
      "model_forward_time": 1.7078981399536133,
      "step": 65438
    },
    {
      "epoch": 6.240653991699219e-05,
      "step": 65438,
      "training_step_time": 8.333995342254639
    },
    {
      "epoch": 6.240749359130859e-05,
      "model_forward_time": 1.8892831802368164,
      "step": 65439
    },
    {
      "epoch": 6.240749359130859e-05,
      "step": 65439,
      "training_step_time": 7.629702806472778
    },
    {
      "epoch": 6.2408447265625e-05,
      "grad_norm": 0.10474886745214462,
      "learning_rate": 2.9252633219625073e-06,
      "loss": 0.0562,
      "step": 65440
    },
    {
      "epoch": 6.2408447265625e-05,
      "model_forward_time": 2.9626004695892334,
      "step": 65440
    },
    {
      "dynamics_loss_avg": 0.06685064528137445,
      "epoch": 6.2408447265625e-05,
      "step": 65440
    },
    {
      "action_loss_avg": 0.0020959235516784247,
      "epoch": 6.2408447265625e-05,
      "step": 65440
    },
    {
      "epoch": 6.2408447265625e-05,
      "step": 65440,
      "training_step_time": 8.698153018951416
    },
    {
      "epoch": 6.240940093994141e-05,
      "model_forward_time": 1.718963623046875,
      "step": 65441
    },
    {
      "epoch": 6.240940093994141e-05,
      "step": 65441,
      "training_step_time": 8.851996183395386
    },
    {
      "epoch": 6.24103546142578e-05,
      "model_forward_time": 1.7251572608947754,
      "step": 65442
    },
    {
      "epoch": 6.24103546142578e-05,
      "step": 65442,
      "training_step_time": 7.754856586456299
    },
    {
      "epoch": 6.241130828857422e-05,
      "model_forward_time": 1.725325584411621,
      "step": 65443
    },
    {
      "epoch": 6.241130828857422e-05,
      "step": 65443,
      "training_step_time": 7.702049255371094
    },
    {
      "epoch": 6.241226196289063e-05,
      "model_forward_time": 1.722649097442627,
      "step": 65444
    },
    {
      "epoch": 6.241226196289063e-05,
      "step": 65444,
      "training_step_time": 7.615737199783325
    },
    {
      "epoch": 6.241321563720702e-05,
      "model_forward_time": 1.7156848907470703,
      "step": 65445
    },
    {
      "epoch": 6.241321563720702e-05,
      "step": 65445,
      "training_step_time": 7.337681293487549
    },
    {
      "epoch": 6.241416931152344e-05,
      "model_forward_time": 1.7151296138763428,
      "step": 65446
    },
    {
      "epoch": 6.241416931152344e-05,
      "step": 65446,
      "training_step_time": 7.5788421630859375
    },
    {
      "epoch": 6.241512298583985e-05,
      "model_forward_time": 1.71307373046875,
      "step": 65447
    },
    {
      "epoch": 6.241512298583985e-05,
      "step": 65447,
      "training_step_time": 8.560494422912598
    },
    {
      "epoch": 6.241607666015624e-05,
      "model_forward_time": 1.712630271911621,
      "step": 65448
    },
    {
      "epoch": 6.241607666015624e-05,
      "step": 65448,
      "training_step_time": 9.286186933517456
    },
    {
      "epoch": 6.241703033447265e-05,
      "model_forward_time": 1.7290573120117188,
      "step": 65449
    },
    {
      "epoch": 6.241703033447265e-05,
      "step": 65449,
      "training_step_time": 9.554277658462524
    },
    {
      "epoch": 6.241798400878907e-05,
      "grad_norm": 0.10113268345594406,
      "learning_rate": 2.9237590334757295e-06,
      "loss": 0.0567,
      "step": 65450
    },
    {
      "epoch": 6.241798400878907e-05,
      "model_forward_time": 1.7307326793670654,
      "step": 65450
    },
    {
      "dynamics_loss_avg": 0.06387867387384176,
      "epoch": 6.241798400878907e-05,
      "step": 65450
    },
    {
      "action_loss_avg": 0.0015998377406504006,
      "epoch": 6.241798400878907e-05,
      "step": 65450
    },
    {
      "epoch": 6.241798400878907e-05,
      "step": 65450,
      "training_step_time": 9.419740200042725
    },
    {
      "epoch": 6.241893768310546e-05,
      "model_forward_time": 1.729233741760254,
      "step": 65451
    },
    {
      "epoch": 6.241893768310546e-05,
      "step": 65451,
      "training_step_time": 7.651992321014404
    },
    {
      "epoch": 6.241989135742187e-05,
      "model_forward_time": 1.8426458835601807,
      "step": 65452
    },
    {
      "epoch": 6.241989135742187e-05,
      "step": 65452,
      "training_step_time": 7.626899480819702
    },
    {
      "epoch": 6.242084503173828e-05,
      "model_forward_time": 1.7249324321746826,
      "step": 65453
    },
    {
      "epoch": 6.242084503173828e-05,
      "step": 65453,
      "training_step_time": 7.577303886413574
    },
    {
      "epoch": 6.242179870605468e-05,
      "model_forward_time": 1.7265591621398926,
      "step": 65454
    },
    {
      "epoch": 6.242179870605468e-05,
      "step": 65454,
      "training_step_time": 7.363278150558472
    },
    {
      "epoch": 6.242275238037109e-05,
      "model_forward_time": 1.7238702774047852,
      "step": 65455
    },
    {
      "epoch": 6.242275238037109e-05,
      "step": 65455,
      "training_step_time": 7.583202600479126
    },
    {
      "epoch": 6.24237060546875e-05,
      "model_forward_time": 1.7385525703430176,
      "step": 65456
    },
    {
      "epoch": 6.24237060546875e-05,
      "step": 65456,
      "training_step_time": 7.6280741691589355
    },
    {
      "epoch": 6.24246597290039e-05,
      "model_forward_time": 1.730104923248291,
      "step": 65457
    },
    {
      "epoch": 6.24246597290039e-05,
      "step": 65457,
      "training_step_time": 8.763301849365234
    },
    {
      "epoch": 6.242561340332031e-05,
      "model_forward_time": 1.7272419929504395,
      "step": 65458
    },
    {
      "epoch": 6.242561340332031e-05,
      "step": 65458,
      "training_step_time": 8.781459093093872
    },
    {
      "epoch": 6.242656707763672e-05,
      "model_forward_time": 2.970097303390503,
      "step": 65459
    },
    {
      "epoch": 6.242656707763672e-05,
      "step": 65459,
      "training_step_time": 7.729756116867065
    },
    {
      "epoch": 6.242752075195312e-05,
      "grad_norm": 0.10918986797332764,
      "learning_rate": 2.9222549720435435e-06,
      "loss": 0.0547,
      "step": 65460
    },
    {
      "epoch": 6.242752075195312e-05,
      "model_forward_time": 1.372692346572876,
      "step": 65460
    },
    {
      "dynamics_loss_avg": 0.045595860294997695,
      "epoch": 6.242752075195312e-05,
      "step": 65460
    },
    {
      "action_loss_avg": 0.001348072913242504,
      "epoch": 6.242752075195312e-05,
      "step": 65460
    },
    {
      "epoch": 6.242752075195312e-05,
      "step": 65460,
      "training_step_time": 9.190251350402832
    },
    {
      "epoch": 6.242847442626953e-05,
      "model_forward_time": 1.844526767730713,
      "step": 65461
    },
    {
      "epoch": 6.242847442626953e-05,
      "step": 65461,
      "training_step_time": 7.575419902801514
    },
    {
      "epoch": 6.242942810058594e-05,
      "model_forward_time": 1.726705551147461,
      "step": 65462
    },
    {
      "epoch": 6.242942810058594e-05,
      "step": 65462,
      "training_step_time": 7.620610952377319
    },
    {
      "epoch": 6.243038177490234e-05,
      "model_forward_time": 1.7255980968475342,
      "step": 65463
    },
    {
      "epoch": 6.243038177490234e-05,
      "step": 65463,
      "training_step_time": 7.247826814651489
    },
    {
      "epoch": 6.243133544921875e-05,
      "model_forward_time": 1.732755184173584,
      "step": 65464
    },
    {
      "epoch": 6.243133544921875e-05,
      "step": 65464,
      "training_step_time": 7.6140806674957275
    },
    {
      "epoch": 6.243228912353516e-05,
      "model_forward_time": 1.7465388774871826,
      "step": 65465
    },
    {
      "epoch": 6.243228912353516e-05,
      "step": 65465,
      "training_step_time": 8.275049448013306
    },
    {
      "epoch": 6.243324279785156e-05,
      "model_forward_time": 1.7450792789459229,
      "step": 65466
    },
    {
      "epoch": 6.243324279785156e-05,
      "step": 65466,
      "training_step_time": 8.454340696334839
    },
    {
      "epoch": 6.243419647216797e-05,
      "model_forward_time": 1.7259972095489502,
      "step": 65467
    },
    {
      "epoch": 6.243419647216797e-05,
      "step": 65467,
      "training_step_time": 7.793986558914185
    },
    {
      "epoch": 6.243515014648438e-05,
      "model_forward_time": 2.874746799468994,
      "step": 65468
    },
    {
      "epoch": 6.243515014648438e-05,
      "step": 65468,
      "training_step_time": 9.748745679855347
    },
    {
      "epoch": 6.243610382080078e-05,
      "model_forward_time": 1.7270865440368652,
      "step": 65469
    },
    {
      "epoch": 6.243610382080078e-05,
      "step": 65469,
      "training_step_time": 8.774222135543823
    },
    {
      "epoch": 6.243705749511719e-05,
      "grad_norm": 0.11677345633506775,
      "learning_rate": 2.9207511378304266e-06,
      "loss": 0.0555,
      "step": 65470
    },
    {
      "epoch": 6.243705749511719e-05,
      "model_forward_time": 1.8363330364227295,
      "step": 65470
    },
    {
      "dynamics_loss_avg": 0.05189968384802342,
      "epoch": 6.243705749511719e-05,
      "step": 65470
    },
    {
      "action_loss_avg": 0.003554884434561245,
      "epoch": 6.243705749511719e-05,
      "step": 65470
    },
    {
      "epoch": 6.243705749511719e-05,
      "step": 65470,
      "training_step_time": 8.79894232749939
    },
    {
      "epoch": 6.24380111694336e-05,
      "model_forward_time": 1.7291173934936523,
      "step": 65471
    },
    {
      "epoch": 6.24380111694336e-05,
      "step": 65471,
      "training_step_time": 8.853021144866943
    },
    {
      "epoch": 6.243896484375e-05,
      "model_forward_time": 1.725475549697876,
      "step": 65472
    },
    {
      "epoch": 6.243896484375e-05,
      "step": 65472,
      "training_step_time": 7.411484718322754
    },
    {
      "epoch": 6.24399185180664e-05,
      "model_forward_time": 1.7154009342193604,
      "step": 65473
    },
    {
      "epoch": 6.24399185180664e-05,
      "step": 65473,
      "training_step_time": 7.623541831970215
    },
    {
      "epoch": 6.244087219238282e-05,
      "model_forward_time": 1.719825267791748,
      "step": 65474
    },
    {
      "epoch": 6.244087219238282e-05,
      "step": 65474,
      "training_step_time": 7.783084869384766
    },
    {
      "epoch": 6.244182586669921e-05,
      "model_forward_time": 1.7180109024047852,
      "step": 65475
    },
    {
      "epoch": 6.244182586669921e-05,
      "step": 65475,
      "training_step_time": 8.722601652145386
    },
    {
      "epoch": 6.244277954101562e-05,
      "model_forward_time": 1.7099201679229736,
      "step": 65476
    },
    {
      "epoch": 6.244277954101562e-05,
      "step": 65476,
      "training_step_time": 8.938360452651978
    },
    {
      "epoch": 6.244373321533203e-05,
      "model_forward_time": 2.8862431049346924,
      "step": 65477
    },
    {
      "epoch": 6.244373321533203e-05,
      "step": 65477,
      "training_step_time": 8.772409915924072
    },
    {
      "epoch": 6.244468688964843e-05,
      "model_forward_time": 1.3469932079315186,
      "step": 65478
    },
    {
      "epoch": 6.244468688964843e-05,
      "step": 65478,
      "training_step_time": 8.417060136795044
    },
    {
      "epoch": 6.244564056396484e-05,
      "model_forward_time": 1.8246793746948242,
      "step": 65479
    },
    {
      "epoch": 6.244564056396484e-05,
      "step": 65479,
      "training_step_time": 9.317087411880493
    },
    {
      "epoch": 6.244659423828125e-05,
      "grad_norm": 0.09892241656780243,
      "learning_rate": 2.919247531000843e-06,
      "loss": 0.0551,
      "step": 65480
    },
    {
      "epoch": 6.244659423828125e-05,
      "model_forward_time": 1.7087769508361816,
      "step": 65480
    },
    {
      "dynamics_loss_avg": 0.052485099900513886,
      "epoch": 6.244659423828125e-05,
      "step": 65480
    },
    {
      "action_loss_avg": 0.0020233591290889307,
      "epoch": 6.244659423828125e-05,
      "step": 65480
    },
    {
      "epoch": 6.244659423828125e-05,
      "step": 65480,
      "training_step_time": 8.821313619613647
    },
    {
      "epoch": 6.244754791259765e-05,
      "model_forward_time": 1.7403008937835693,
      "step": 65481
    },
    {
      "epoch": 6.244754791259765e-05,
      "step": 65481,
      "training_step_time": 7.423872470855713
    },
    {
      "epoch": 6.244850158691406e-05,
      "model_forward_time": 1.7128634452819824,
      "step": 65482
    },
    {
      "epoch": 6.244850158691406e-05,
      "step": 65482,
      "training_step_time": 7.605571269989014
    },
    {
      "epoch": 6.244945526123047e-05,
      "model_forward_time": 1.712892770767212,
      "step": 65483
    },
    {
      "epoch": 6.244945526123047e-05,
      "step": 65483,
      "training_step_time": 7.802168369293213
    },
    {
      "epoch": 6.245040893554687e-05,
      "model_forward_time": 1.7147083282470703,
      "step": 65484
    },
    {
      "epoch": 6.245040893554687e-05,
      "step": 65484,
      "training_step_time": 7.68587327003479
    },
    {
      "epoch": 6.245136260986328e-05,
      "model_forward_time": 1.7133445739746094,
      "step": 65485
    },
    {
      "epoch": 6.245136260986328e-05,
      "step": 65485,
      "training_step_time": 8.659775257110596
    },
    {
      "epoch": 6.245231628417969e-05,
      "model_forward_time": 2.818606376647949,
      "step": 65486
    },
    {
      "epoch": 6.245231628417969e-05,
      "step": 65486,
      "training_step_time": 8.633245468139648
    },
    {
      "epoch": 6.245326995849609e-05,
      "model_forward_time": 1.717930555343628,
      "step": 65487
    },
    {
      "epoch": 6.245326995849609e-05,
      "step": 65487,
      "training_step_time": 7.720734119415283
    },
    {
      "epoch": 6.24542236328125e-05,
      "model_forward_time": 1.8224568367004395,
      "step": 65488
    },
    {
      "epoch": 6.24542236328125e-05,
      "step": 65488,
      "training_step_time": 7.721532106399536
    },
    {
      "epoch": 6.245517730712891e-05,
      "model_forward_time": 1.7144136428833008,
      "step": 65489
    },
    {
      "epoch": 6.245517730712891e-05,
      "step": 65489,
      "training_step_time": 8.323446035385132
    },
    {
      "epoch": 6.245613098144531e-05,
      "grad_norm": 0.34806016087532043,
      "learning_rate": 2.917744151719219e-06,
      "loss": 0.0551,
      "step": 65490
    },
    {
      "epoch": 6.245613098144531e-05,
      "model_forward_time": 1.7116889953613281,
      "step": 65490
    },
    {
      "dynamics_loss_avg": 0.05127661358565092,
      "epoch": 6.245613098144531e-05,
      "step": 65490
    },
    {
      "action_loss_avg": 0.0016689146286807954,
      "epoch": 6.245613098144531e-05,
      "step": 65490
    },
    {
      "epoch": 6.245613098144531e-05,
      "step": 65490,
      "training_step_time": 7.605803966522217
    },
    {
      "epoch": 6.245708465576172e-05,
      "model_forward_time": 1.709766149520874,
      "step": 65491
    },
    {
      "epoch": 6.245708465576172e-05,
      "step": 65491,
      "training_step_time": 7.494682312011719
    },
    {
      "epoch": 6.245803833007813e-05,
      "model_forward_time": 1.7190186977386475,
      "step": 65492
    },
    {
      "epoch": 6.245803833007813e-05,
      "step": 65492,
      "training_step_time": 7.670143365859985
    },
    {
      "epoch": 6.245899200439453e-05,
      "model_forward_time": 1.710960865020752,
      "step": 65493
    },
    {
      "epoch": 6.245899200439453e-05,
      "step": 65493,
      "training_step_time": 7.608962059020996
    },
    {
      "epoch": 6.245994567871094e-05,
      "model_forward_time": 1.7133865356445312,
      "step": 65494
    },
    {
      "epoch": 6.245994567871094e-05,
      "step": 65494,
      "training_step_time": 9.251336097717285
    },
    {
      "epoch": 6.246089935302735e-05,
      "model_forward_time": 2.7883009910583496,
      "step": 65495
    },
    {
      "epoch": 6.246089935302735e-05,
      "step": 65495,
      "training_step_time": 9.380932331085205
    },
    {
      "epoch": 6.246185302734375e-05,
      "model_forward_time": 1.7127289772033691,
      "step": 65496
    },
    {
      "epoch": 6.246185302734375e-05,
      "step": 65496,
      "training_step_time": 8.804107427597046
    },
    {
      "epoch": 6.246280670166016e-05,
      "model_forward_time": 1.824720859527588,
      "step": 65497
    },
    {
      "epoch": 6.246280670166016e-05,
      "step": 65497,
      "training_step_time": 7.667700529098511
    },
    {
      "epoch": 6.246376037597657e-05,
      "model_forward_time": 1.7224080562591553,
      "step": 65498
    },
    {
      "epoch": 6.246376037597657e-05,
      "step": 65498,
      "training_step_time": 7.549010992050171
    },
    {
      "epoch": 6.246471405029296e-05,
      "model_forward_time": 1.7142088413238525,
      "step": 65499
    },
    {
      "epoch": 6.246471405029296e-05,
      "step": 65499,
      "training_step_time": 7.563216686248779
    },
    {
      "epoch": 6.246566772460938e-05,
      "grad_norm": 0.11872522532939911,
      "learning_rate": 2.916241000149964e-06,
      "loss": 0.0562,
      "step": 65500
    },
    {
      "epoch": 6.246566772460938e-05,
      "model_forward_time": 1.7101051807403564,
      "step": 65500
    },
    {
      "dynamics_loss_avg": 0.056980868056416514,
      "epoch": 6.246566772460938e-05,
      "step": 65500
    },
    {
      "action_loss_avg": 0.002406799863092601,
      "epoch": 6.246566772460938e-05,
      "step": 65500
    },
    {
      "epoch": 6.246566772460938e-05,
      "step": 65500,
      "training_step_time": 7.497360944747925
    },
    {
      "epoch": 6.246662139892579e-05,
      "model_forward_time": 1.7140672206878662,
      "step": 65501
    },
    {
      "epoch": 6.246662139892579e-05,
      "step": 65501,
      "training_step_time": 7.552849292755127
    },
    {
      "epoch": 6.246757507324218e-05,
      "model_forward_time": 1.7114107608795166,
      "step": 65502
    },
    {
      "epoch": 6.246757507324218e-05,
      "step": 65502,
      "training_step_time": 8.628792762756348
    },
    {
      "epoch": 6.24685287475586e-05,
      "model_forward_time": 1.7136833667755127,
      "step": 65503
    },
    {
      "epoch": 6.24685287475586e-05,
      "step": 65503,
      "training_step_time": 7.627078056335449
    },
    {
      "epoch": 6.2469482421875e-05,
      "model_forward_time": 1.7101964950561523,
      "step": 65504
    },
    {
      "epoch": 6.2469482421875e-05,
      "step": 65504,
      "training_step_time": 8.618857145309448
    },
    {
      "epoch": 6.24704360961914e-05,
      "model_forward_time": 2.8991386890411377,
      "step": 65505
    },
    {
      "epoch": 6.24704360961914e-05,
      "step": 65505,
      "training_step_time": 8.23669171333313
    },
    {
      "epoch": 6.247138977050781e-05,
      "model_forward_time": 1.8148643970489502,
      "step": 65506
    },
    {
      "epoch": 6.247138977050781e-05,
      "step": 65506,
      "training_step_time": 7.63431191444397
    },
    {
      "epoch": 6.247234344482422e-05,
      "model_forward_time": 1.7547883987426758,
      "step": 65507
    },
    {
      "epoch": 6.247234344482422e-05,
      "step": 65507,
      "training_step_time": 8.62231731414795
    },
    {
      "epoch": 6.247329711914062e-05,
      "model_forward_time": 1.722290277481079,
      "step": 65508
    },
    {
      "epoch": 6.247329711914062e-05,
      "step": 65508,
      "training_step_time": 7.532582998275757
    },
    {
      "epoch": 6.247425079345703e-05,
      "model_forward_time": 1.7065458297729492,
      "step": 65509
    },
    {
      "epoch": 6.247425079345703e-05,
      "step": 65509,
      "training_step_time": 7.586818695068359
    },
    {
      "epoch": 6.247520446777344e-05,
      "grad_norm": 0.08114975690841675,
      "learning_rate": 2.9147380764574625e-06,
      "loss": 0.06,
      "step": 65510
    },
    {
      "epoch": 6.247520446777344e-05,
      "model_forward_time": 1.7150864601135254,
      "step": 65510
    },
    {
      "dynamics_loss_avg": 0.04655127814039588,
      "epoch": 6.247520446777344e-05,
      "step": 65510
    },
    {
      "action_loss_avg": 0.000730167658184655,
      "epoch": 6.247520446777344e-05,
      "step": 65510
    },
    {
      "epoch": 6.247520446777344e-05,
      "step": 65510,
      "training_step_time": 7.475507974624634
    },
    {
      "epoch": 6.247615814208984e-05,
      "model_forward_time": 1.717238426208496,
      "step": 65511
    },
    {
      "epoch": 6.247615814208984e-05,
      "step": 65511,
      "training_step_time": 7.570607900619507
    },
    {
      "epoch": 6.247711181640625e-05,
      "model_forward_time": 1.7128040790557861,
      "step": 65512
    },
    {
      "epoch": 6.247711181640625e-05,
      "step": 65512,
      "training_step_time": 8.663799524307251
    },
    {
      "epoch": 6.247806549072266e-05,
      "model_forward_time": 1.7117891311645508,
      "step": 65513
    },
    {
      "epoch": 6.247806549072266e-05,
      "step": 65513,
      "training_step_time": 7.870310544967651
    },
    {
      "epoch": 6.247901916503906e-05,
      "model_forward_time": 2.9170963764190674,
      "step": 65514
    },
    {
      "epoch": 6.247901916503906e-05,
      "step": 65514,
      "training_step_time": 8.91146206855774
    },
    {
      "epoch": 6.247997283935547e-05,
      "model_forward_time": 1.8207769393920898,
      "step": 65515
    },
    {
      "epoch": 6.247997283935547e-05,
      "step": 65515,
      "training_step_time": 7.786957263946533
    },
    {
      "epoch": 6.248092651367188e-05,
      "model_forward_time": 1.7118709087371826,
      "step": 65516
    },
    {
      "epoch": 6.248092651367188e-05,
      "step": 65516,
      "training_step_time": 7.795944929122925
    },
    {
      "epoch": 6.248188018798828e-05,
      "model_forward_time": 1.7089905738830566,
      "step": 65517
    },
    {
      "epoch": 6.248188018798828e-05,
      "step": 65517,
      "training_step_time": 7.843714475631714
    },
    {
      "epoch": 6.248283386230469e-05,
      "model_forward_time": 1.7139256000518799,
      "step": 65518
    },
    {
      "epoch": 6.248283386230469e-05,
      "step": 65518,
      "training_step_time": 7.707438945770264
    },
    {
      "epoch": 6.24837875366211e-05,
      "model_forward_time": 1.7196035385131836,
      "step": 65519
    },
    {
      "epoch": 6.24837875366211e-05,
      "step": 65519,
      "training_step_time": 7.640415668487549
    },
    {
      "epoch": 6.24847412109375e-05,
      "grad_norm": 0.08201227337121964,
      "learning_rate": 2.9132353808060674e-06,
      "loss": 0.0541,
      "step": 65520
    },
    {
      "epoch": 6.24847412109375e-05,
      "model_forward_time": 1.7131555080413818,
      "step": 65520
    },
    {
      "dynamics_loss_avg": 0.051042836904525754,
      "epoch": 6.24847412109375e-05,
      "step": 65520
    },
    {
      "action_loss_avg": 0.0013254505000077188,
      "epoch": 6.24847412109375e-05,
      "step": 65520
    },
    {
      "epoch": 6.24847412109375e-05,
      "step": 65520,
      "training_step_time": 8.593295812606812
    },
    {
      "epoch": 6.248569488525391e-05,
      "model_forward_time": 1.8083064556121826,
      "step": 65521
    },
    {
      "epoch": 6.248569488525391e-05,
      "step": 65521,
      "training_step_time": 7.730345010757446
    },
    {
      "epoch": 6.248664855957032e-05,
      "model_forward_time": 1.7063939571380615,
      "step": 65522
    },
    {
      "epoch": 6.248664855957032e-05,
      "step": 65522,
      "training_step_time": 9.414344549179077
    },
    {
      "epoch": 6.248760223388672e-05,
      "model_forward_time": 2.886559247970581,
      "step": 65523
    },
    {
      "epoch": 6.248760223388672e-05,
      "step": 65523,
      "training_step_time": 7.9234349727630615
    },
    {
      "epoch": 6.248855590820313e-05,
      "model_forward_time": 1.7472405433654785,
      "step": 65524
    },
    {
      "epoch": 6.248855590820313e-05,
      "step": 65524,
      "training_step_time": 7.639682769775391
    },
    {
      "epoch": 6.248950958251954e-05,
      "model_forward_time": 1.7164287567138672,
      "step": 65525
    },
    {
      "epoch": 6.248950958251954e-05,
      "step": 65525,
      "training_step_time": 7.601144075393677
    },
    {
      "epoch": 6.249046325683593e-05,
      "model_forward_time": 1.3579022884368896,
      "step": 65526
    },
    {
      "epoch": 6.249046325683593e-05,
      "step": 65526,
      "training_step_time": 7.629544496536255
    },
    {
      "epoch": 6.249141693115234e-05,
      "model_forward_time": 1.712653636932373,
      "step": 65527
    },
    {
      "epoch": 6.249141693115234e-05,
      "step": 65527,
      "training_step_time": 7.661987066268921
    },
    {
      "epoch": 6.249237060546876e-05,
      "model_forward_time": 1.7163500785827637,
      "step": 65528
    },
    {
      "epoch": 6.249237060546876e-05,
      "step": 65528,
      "training_step_time": 7.571346282958984
    },
    {
      "epoch": 6.249332427978515e-05,
      "model_forward_time": 1.7195689678192139,
      "step": 65529
    },
    {
      "epoch": 6.249332427978515e-05,
      "step": 65529,
      "training_step_time": 8.094143390655518
    },
    {
      "epoch": 6.249427795410156e-05,
      "grad_norm": 0.0807950422167778,
      "learning_rate": 2.9117329133601134e-06,
      "loss": 0.0571,
      "step": 65530
    },
    {
      "epoch": 6.249427795410156e-05,
      "model_forward_time": 1.7137088775634766,
      "step": 65530
    },
    {
      "dynamics_loss_avg": 0.05575048662722111,
      "epoch": 6.249427795410156e-05,
      "step": 65530
    },
    {
      "action_loss_avg": 0.0013203428170527332,
      "epoch": 6.249427795410156e-05,
      "step": 65530
    },
    {
      "epoch": 6.249427795410156e-05,
      "step": 65530,
      "training_step_time": 7.874690532684326
    },
    {
      "epoch": 6.249523162841797e-05,
      "model_forward_time": 1.711230993270874,
      "step": 65531
    },
    {
      "epoch": 6.249523162841797e-05,
      "step": 65531,
      "training_step_time": 7.648113012313843
    },
    {
      "epoch": 6.249618530273437e-05,
      "model_forward_time": 2.8241372108459473,
      "step": 65532
    },
    {
      "epoch": 6.249618530273437e-05,
      "step": 65532,
      "training_step_time": 8.360708236694336
    },
    {
      "epoch": 6.249713897705078e-05,
      "model_forward_time": 1.7163221836090088,
      "step": 65533
    },
    {
      "epoch": 6.249713897705078e-05,
      "step": 65533,
      "training_step_time": 9.457072257995605
    },
    {
      "epoch": 6.24980926513672e-05,
      "model_forward_time": 1.7206008434295654,
      "step": 65534
    },
    {
      "epoch": 6.24980926513672e-05,
      "step": 65534,
      "training_step_time": 7.713429689407349
    },
    {
      "epoch": 6.249904632568359e-05,
      "model_forward_time": 1.711500644683838,
      "step": 65535
    },
    {
      "epoch": 6.249904632568359e-05,
      "step": 65535,
      "training_step_time": 7.770423889160156
    },
    {
      "epoch": 6.25e-05,
      "model_forward_time": 1.7179112434387207,
      "step": 65536
    },
    {
      "epoch": 6.25e-05,
      "step": 65536,
      "training_step_time": 7.654025554656982
    },
    {
      "epoch": 6.250095367431641e-05,
      "model_forward_time": 1.7175192832946777,
      "step": 65537
    },
    {
      "epoch": 6.250095367431641e-05,
      "step": 65537,
      "training_step_time": 7.4763453006744385
    },
    {
      "epoch": 6.250190734863281e-05,
      "model_forward_time": 1.7177214622497559,
      "step": 65538
    },
    {
      "epoch": 6.250190734863281e-05,
      "step": 65538,
      "training_step_time": 7.694898366928101
    },
    {
      "epoch": 6.250286102294922e-05,
      "model_forward_time": 1.7082340717315674,
      "step": 65539
    },
    {
      "epoch": 6.250286102294922e-05,
      "step": 65539,
      "training_step_time": 7.622599363327026
    },
    {
      "epoch": 6.250381469726563e-05,
      "grad_norm": 0.1069439947605133,
      "learning_rate": 2.910230674283909e-06,
      "loss": 0.0562,
      "step": 65540
    },
    {
      "epoch": 6.250381469726563e-05,
      "model_forward_time": 1.7145578861236572,
      "step": 65540
    },
    {
      "dynamics_loss_avg": 0.0606542544439435,
      "epoch": 6.250381469726563e-05,
      "step": 65540
    },
    {
      "action_loss_avg": 0.0019272938719950617,
      "epoch": 6.250381469726563e-05,
      "step": 65540
    },
    {
      "epoch": 6.250381469726563e-05,
      "step": 65540,
      "training_step_time": 7.693817853927612
    },
    {
      "epoch": 6.250476837158203e-05,
      "model_forward_time": 1.7169592380523682,
      "step": 65541
    },
    {
      "epoch": 6.250476837158203e-05,
      "step": 65541,
      "training_step_time": 9.548314571380615
    },
    {
      "epoch": 6.250572204589844e-05,
      "model_forward_time": 1.7117927074432373,
      "step": 65542
    },
    {
      "epoch": 6.250572204589844e-05,
      "step": 65542,
      "training_step_time": 9.835822343826294
    },
    {
      "epoch": 6.250667572021485e-05,
      "model_forward_time": 1.7273924350738525,
      "step": 65543
    },
    {
      "epoch": 6.250667572021485e-05,
      "step": 65543,
      "training_step_time": 7.66112756729126
    },
    {
      "epoch": 6.250762939453125e-05,
      "model_forward_time": 1.7104556560516357,
      "step": 65544
    },
    {
      "epoch": 6.250762939453125e-05,
      "step": 65544,
      "training_step_time": 8.877689599990845
    },
    {
      "epoch": 6.250858306884766e-05,
      "model_forward_time": 1.7092957496643066,
      "step": 65545
    },
    {
      "epoch": 6.250858306884766e-05,
      "step": 65545,
      "training_step_time": 7.564500093460083
    },
    {
      "epoch": 6.250953674316407e-05,
      "model_forward_time": 1.7122809886932373,
      "step": 65546
    },
    {
      "epoch": 6.250953674316407e-05,
      "step": 65546,
      "training_step_time": 7.588505983352661
    },
    {
      "epoch": 6.251049041748047e-05,
      "model_forward_time": 1.7141482830047607,
      "step": 65547
    },
    {
      "epoch": 6.251049041748047e-05,
      "step": 65547,
      "training_step_time": 7.609179496765137
    },
    {
      "epoch": 6.251144409179688e-05,
      "model_forward_time": 1.7066338062286377,
      "step": 65548
    },
    {
      "epoch": 6.251144409179688e-05,
      "step": 65548,
      "training_step_time": 7.786895990371704
    },
    {
      "epoch": 6.251239776611329e-05,
      "model_forward_time": 1.7226243019104004,
      "step": 65549
    },
    {
      "epoch": 6.251239776611329e-05,
      "step": 65549,
      "training_step_time": 8.222014427185059
    },
    {
      "epoch": 6.251335144042968e-05,
      "grad_norm": 0.1041184663772583,
      "learning_rate": 2.908728663741738e-06,
      "loss": 0.0577,
      "step": 65550
    },
    {
      "epoch": 6.251335144042968e-05,
      "model_forward_time": 1.707451581954956,
      "step": 65550
    },
    {
      "dynamics_loss_avg": 0.0629030417650938,
      "epoch": 6.251335144042968e-05,
      "step": 65550
    },
    {
      "action_loss_avg": 0.0017920882150065155,
      "epoch": 6.251335144042968e-05,
      "step": 65550
    },
    {
      "epoch": 6.251335144042968e-05,
      "step": 65550,
      "training_step_time": 7.814686059951782
    },
    {
      "epoch": 6.25143051147461e-05,
      "model_forward_time": 2.932377815246582,
      "step": 65551
    },
    {
      "epoch": 6.25143051147461e-05,
      "step": 65551,
      "training_step_time": 8.713131427764893
    },
    {
      "epoch": 6.25152587890625e-05,
      "model_forward_time": 1.7214930057525635,
      "step": 65552
    },
    {
      "epoch": 6.25152587890625e-05,
      "step": 65552,
      "training_step_time": 8.447365045547485
    },
    {
      "epoch": 6.25162124633789e-05,
      "model_forward_time": 1.7170696258544922,
      "step": 65553
    },
    {
      "epoch": 6.25162124633789e-05,
      "step": 65553,
      "training_step_time": 7.681077003479004
    },
    {
      "epoch": 6.251716613769531e-05,
      "model_forward_time": 1.718294620513916,
      "step": 65554
    },
    {
      "epoch": 6.251716613769531e-05,
      "step": 65554,
      "training_step_time": 7.771103858947754
    },
    {
      "epoch": 6.251811981201173e-05,
      "model_forward_time": 1.7095887660980225,
      "step": 65555
    },
    {
      "epoch": 6.251811981201173e-05,
      "step": 65555,
      "training_step_time": 7.53532338142395
    },
    {
      "epoch": 6.251907348632812e-05,
      "model_forward_time": 1.715691089630127,
      "step": 65556
    },
    {
      "epoch": 6.251907348632812e-05,
      "step": 65556,
      "training_step_time": 7.617517709732056
    },
    {
      "epoch": 6.252002716064453e-05,
      "model_forward_time": 1.7167248725891113,
      "step": 65557
    },
    {
      "epoch": 6.252002716064453e-05,
      "step": 65557,
      "training_step_time": 7.5976927280426025
    },
    {
      "epoch": 6.252098083496094e-05,
      "model_forward_time": 1.722153663635254,
      "step": 65558
    },
    {
      "epoch": 6.252098083496094e-05,
      "step": 65558,
      "training_step_time": 8.383457660675049
    },
    {
      "epoch": 6.252193450927734e-05,
      "model_forward_time": 1.7108819484710693,
      "step": 65559
    },
    {
      "epoch": 6.252193450927734e-05,
      "step": 65559,
      "training_step_time": 9.280023574829102
    },
    {
      "epoch": 6.252288818359375e-05,
      "grad_norm": 0.09019419550895691,
      "learning_rate": 2.907226881897854e-06,
      "loss": 0.0548,
      "step": 65560
    },
    {
      "epoch": 6.252288818359375e-05,
      "model_forward_time": 2.8980295658111572,
      "step": 65560
    },
    {
      "dynamics_loss_avg": 0.04407690055668354,
      "epoch": 6.252288818359375e-05,
      "step": 65560
    },
    {
      "action_loss_avg": 0.0028158324799733236,
      "epoch": 6.252288818359375e-05,
      "step": 65560
    },
    {
      "epoch": 6.252288818359375e-05,
      "step": 65560,
      "training_step_time": 8.78067421913147
    },
    {
      "epoch": 6.252384185791016e-05,
      "model_forward_time": 1.7158598899841309,
      "step": 65561
    },
    {
      "epoch": 6.252384185791016e-05,
      "step": 65561,
      "training_step_time": 7.760923385620117
    },
    {
      "epoch": 6.252479553222656e-05,
      "model_forward_time": 1.7068939208984375,
      "step": 65562
    },
    {
      "epoch": 6.252479553222656e-05,
      "step": 65562,
      "training_step_time": 7.573386192321777
    },
    {
      "epoch": 6.252574920654297e-05,
      "model_forward_time": 1.7065789699554443,
      "step": 65563
    },
    {
      "epoch": 6.252574920654297e-05,
      "step": 65563,
      "training_step_time": 8.458602666854858
    },
    {
      "epoch": 6.252670288085937e-05,
      "model_forward_time": 1.7070903778076172,
      "step": 65564
    },
    {
      "epoch": 6.252670288085937e-05,
      "step": 65564,
      "training_step_time": 7.654794216156006
    },
    {
      "epoch": 6.252765655517578e-05,
      "model_forward_time": 1.7153856754302979,
      "step": 65565
    },
    {
      "epoch": 6.252765655517578e-05,
      "step": 65565,
      "training_step_time": 7.593113422393799
    },
    {
      "epoch": 6.252861022949219e-05,
      "model_forward_time": 1.7089579105377197,
      "step": 65566
    },
    {
      "epoch": 6.252861022949219e-05,
      "step": 65566,
      "training_step_time": 7.612303733825684
    },
    {
      "epoch": 6.252956390380859e-05,
      "model_forward_time": 1.7225704193115234,
      "step": 65567
    },
    {
      "epoch": 6.252956390380859e-05,
      "step": 65567,
      "training_step_time": 8.358734369277954
    },
    {
      "epoch": 6.2530517578125e-05,
      "model_forward_time": 1.7127883434295654,
      "step": 65568
    },
    {
      "epoch": 6.2530517578125e-05,
      "step": 65568,
      "training_step_time": 9.761963129043579
    },
    {
      "epoch": 6.253147125244141e-05,
      "model_forward_time": 2.8315682411193848,
      "step": 65569
    },
    {
      "epoch": 6.253147125244141e-05,
      "step": 65569,
      "training_step_time": 8.979684352874756
    },
    {
      "epoch": 6.25324249267578e-05,
      "grad_norm": 0.10798411071300507,
      "learning_rate": 2.9057253289164934e-06,
      "loss": 0.0583,
      "step": 65570
    },
    {
      "epoch": 6.25324249267578e-05,
      "model_forward_time": 1.7056896686553955,
      "step": 65570
    },
    {
      "dynamics_loss_avg": 0.0524143778719008,
      "epoch": 6.25324249267578e-05,
      "step": 65570
    },
    {
      "action_loss_avg": 0.0015394061512779445,
      "epoch": 6.25324249267578e-05,
      "step": 65570
    },
    {
      "epoch": 6.25324249267578e-05,
      "step": 65570,
      "training_step_time": 9.754671335220337
    },
    {
      "epoch": 6.253337860107422e-05,
      "model_forward_time": 1.7103066444396973,
      "step": 65571
    },
    {
      "epoch": 6.253337860107422e-05,
      "step": 65571,
      "training_step_time": 7.656314849853516
    },
    {
      "epoch": 6.253433227539063e-05,
      "model_forward_time": 1.710097074508667,
      "step": 65572
    },
    {
      "epoch": 6.253433227539063e-05,
      "step": 65572,
      "training_step_time": 7.682172536849976
    },
    {
      "epoch": 6.253528594970702e-05,
      "model_forward_time": 1.7055377960205078,
      "step": 65573
    },
    {
      "epoch": 6.253528594970702e-05,
      "step": 65573,
      "training_step_time": 8.496315717697144
    },
    {
      "epoch": 6.253623962402344e-05,
      "model_forward_time": 1.722379446029663,
      "step": 65574
    },
    {
      "epoch": 6.253623962402344e-05,
      "step": 65574,
      "training_step_time": 8.016630411148071
    },
    {
      "epoch": 6.253719329833985e-05,
      "model_forward_time": 1.7066059112548828,
      "step": 65575
    },
    {
      "epoch": 6.253719329833985e-05,
      "step": 65575,
      "training_step_time": 7.528714179992676
    },
    {
      "epoch": 6.253814697265624e-05,
      "model_forward_time": 1.701988935470581,
      "step": 65576
    },
    {
      "epoch": 6.253814697265624e-05,
      "step": 65576,
      "training_step_time": 8.680323839187622
    },
    {
      "epoch": 6.253910064697265e-05,
      "model_forward_time": 1.7085402011871338,
      "step": 65577
    },
    {
      "epoch": 6.253910064697265e-05,
      "step": 65577,
      "training_step_time": 7.596877813339233
    },
    {
      "epoch": 6.254005432128907e-05,
      "model_forward_time": 2.8350582122802734,
      "step": 65578
    },
    {
      "epoch": 6.254005432128907e-05,
      "step": 65578,
      "training_step_time": 7.766584634780884
    },
    {
      "epoch": 6.254100799560546e-05,
      "model_forward_time": 1.7055096626281738,
      "step": 65579
    },
    {
      "epoch": 6.254100799560546e-05,
      "step": 65579,
      "training_step_time": 9.958149433135986
    },
    {
      "epoch": 6.254196166992187e-05,
      "grad_norm": 0.09158556163311005,
      "learning_rate": 2.904224004961861e-06,
      "loss": 0.0565,
      "step": 65580
    },
    {
      "epoch": 6.254196166992187e-05,
      "model_forward_time": 1.7040092945098877,
      "step": 65580
    },
    {
      "dynamics_loss_avg": 0.05694752298295498,
      "epoch": 6.254196166992187e-05,
      "step": 65580
    },
    {
      "action_loss_avg": 0.0013494899671059103,
      "epoch": 6.254196166992187e-05,
      "step": 65580
    },
    {
      "epoch": 6.254196166992187e-05,
      "step": 65580,
      "training_step_time": 8.622984647750854
    },
    {
      "epoch": 6.254291534423828e-05,
      "model_forward_time": 1.7126219272613525,
      "step": 65581
    },
    {
      "epoch": 6.254291534423828e-05,
      "step": 65581,
      "training_step_time": 8.609321355819702
    },
    {
      "epoch": 6.254386901855468e-05,
      "model_forward_time": 1.7054173946380615,
      "step": 65582
    },
    {
      "epoch": 6.254386901855468e-05,
      "step": 65582,
      "training_step_time": 7.735621929168701
    },
    {
      "epoch": 6.254482269287109e-05,
      "model_forward_time": 1.712362289428711,
      "step": 65583
    },
    {
      "epoch": 6.254482269287109e-05,
      "step": 65583,
      "training_step_time": 7.657501220703125
    },
    {
      "epoch": 6.25457763671875e-05,
      "model_forward_time": 1.7092299461364746,
      "step": 65584
    },
    {
      "epoch": 6.25457763671875e-05,
      "step": 65584,
      "training_step_time": 7.624873638153076
    },
    {
      "epoch": 6.25467300415039e-05,
      "model_forward_time": 1.720931053161621,
      "step": 65585
    },
    {
      "epoch": 6.25467300415039e-05,
      "step": 65585,
      "training_step_time": 8.805155992507935
    },
    {
      "epoch": 6.254768371582031e-05,
      "model_forward_time": 1.7307229042053223,
      "step": 65586
    },
    {
      "epoch": 6.254768371582031e-05,
      "step": 65586,
      "training_step_time": 8.234831809997559
    },
    {
      "epoch": 6.254863739013672e-05,
      "model_forward_time": 1.7120192050933838,
      "step": 65587
    },
    {
      "epoch": 6.254863739013672e-05,
      "step": 65587,
      "training_step_time": 10.407676219940186
    },
    {
      "epoch": 6.254959106445312e-05,
      "model_forward_time": 1.7087655067443848,
      "step": 65588
    },
    {
      "epoch": 6.254959106445312e-05,
      "step": 65588,
      "training_step_time": 7.729466199874878
    },
    {
      "epoch": 6.255054473876953e-05,
      "model_forward_time": 1.7299373149871826,
      "step": 65589
    },
    {
      "epoch": 6.255054473876953e-05,
      "step": 65589,
      "training_step_time": 7.615859270095825
    },
    {
      "epoch": 6.255149841308594e-05,
      "grad_norm": 0.09209324419498444,
      "learning_rate": 2.9027229101981414e-06,
      "loss": 0.0574,
      "step": 65590
    },
    {
      "epoch": 6.255149841308594e-05,
      "model_forward_time": 1.7104623317718506,
      "step": 65590
    },
    {
      "dynamics_loss_avg": 0.0532086668536067,
      "epoch": 6.255149841308594e-05,
      "step": 65590
    },
    {
      "action_loss_avg": 0.0012580465554492548,
      "epoch": 6.255149841308594e-05,
      "step": 65590
    },
    {
      "epoch": 6.255149841308594e-05,
      "step": 65590,
      "training_step_time": 8.196981430053711
    },
    {
      "epoch": 6.255245208740234e-05,
      "model_forward_time": 1.7093124389648438,
      "step": 65591
    },
    {
      "epoch": 6.255245208740234e-05,
      "step": 65591,
      "training_step_time": 7.665172576904297
    },
    {
      "epoch": 6.255340576171875e-05,
      "model_forward_time": 1.7053065299987793,
      "step": 65592
    },
    {
      "epoch": 6.255340576171875e-05,
      "step": 65592,
      "training_step_time": 7.177488088607788
    },
    {
      "epoch": 6.255435943603516e-05,
      "model_forward_time": 1.3373494148254395,
      "step": 65593
    },
    {
      "epoch": 6.255435943603516e-05,
      "step": 65593,
      "training_step_time": 7.572662115097046
    },
    {
      "epoch": 6.255531311035156e-05,
      "model_forward_time": 1.7096354961395264,
      "step": 65594
    },
    {
      "epoch": 6.255531311035156e-05,
      "step": 65594,
      "training_step_time": 8.30121374130249
    },
    {
      "epoch": 6.255626678466797e-05,
      "model_forward_time": 1.7295141220092773,
      "step": 65595
    },
    {
      "epoch": 6.255626678466797e-05,
      "step": 65595,
      "training_step_time": 8.714059352874756
    },
    {
      "epoch": 6.255722045898438e-05,
      "model_forward_time": 1.7152328491210938,
      "step": 65596
    },
    {
      "epoch": 6.255722045898438e-05,
      "step": 65596,
      "training_step_time": 8.347851276397705
    },
    {
      "epoch": 6.255817413330078e-05,
      "model_forward_time": 2.8526852130889893,
      "step": 65597
    },
    {
      "epoch": 6.255817413330078e-05,
      "step": 65597,
      "training_step_time": 7.6668620109558105
    },
    {
      "epoch": 6.255912780761719e-05,
      "model_forward_time": 1.7318120002746582,
      "step": 65598
    },
    {
      "epoch": 6.255912780761719e-05,
      "step": 65598,
      "training_step_time": 7.624944448471069
    },
    {
      "epoch": 6.25600814819336e-05,
      "model_forward_time": 1.7138514518737793,
      "step": 65599
    },
    {
      "epoch": 6.25600814819336e-05,
      "step": 65599,
      "training_step_time": 7.562745094299316
    },
    {
      "epoch": 6.256103515625e-05,
      "grad_norm": 0.27608516812324524,
      "learning_rate": 2.901222044789488e-06,
      "loss": 0.0558,
      "step": 65600
    },
    {
      "epoch": 6.256103515625e-05,
      "model_forward_time": 1.7071101665496826,
      "step": 65600
    },
    {
      "dynamics_loss_avg": 0.04996874425560236,
      "epoch": 6.256103515625e-05,
      "step": 65600
    },
    {
      "action_loss_avg": 0.0009971088671591132,
      "epoch": 6.256103515625e-05,
      "step": 65600
    },
    {
      "epoch": 6.256103515625e-05,
      "step": 65600,
      "training_step_time": 7.639680862426758
    },
    {
      "epoch": 6.25619888305664e-05,
      "model_forward_time": 1.7085072994232178,
      "step": 65601
    },
    {
      "epoch": 6.25619888305664e-05,
      "step": 65601,
      "training_step_time": 7.356565952301025
    },
    {
      "epoch": 6.256294250488282e-05,
      "model_forward_time": 1.70658540725708,
      "step": 65602
    },
    {
      "epoch": 6.256294250488282e-05,
      "step": 65602,
      "training_step_time": 7.544842720031738
    },
    {
      "epoch": 6.256389617919921e-05,
      "model_forward_time": 1.7157366275787354,
      "step": 65603
    },
    {
      "epoch": 6.256389617919921e-05,
      "step": 65603,
      "training_step_time": 7.604750394821167
    },
    {
      "epoch": 6.256484985351562e-05,
      "model_forward_time": 1.7300450801849365,
      "step": 65604
    },
    {
      "epoch": 6.256484985351562e-05,
      "step": 65604,
      "training_step_time": 7.680639743804932
    },
    {
      "epoch": 6.256580352783204e-05,
      "model_forward_time": 1.7223567962646484,
      "step": 65605
    },
    {
      "epoch": 6.256580352783204e-05,
      "step": 65605,
      "training_step_time": 9.285135984420776
    },
    {
      "epoch": 6.256675720214843e-05,
      "model_forward_time": 2.820509433746338,
      "step": 65606
    },
    {
      "epoch": 6.256675720214843e-05,
      "step": 65606,
      "training_step_time": 8.704708337783813
    },
    {
      "epoch": 6.256771087646484e-05,
      "model_forward_time": 1.7537071704864502,
      "step": 65607
    },
    {
      "epoch": 6.256771087646484e-05,
      "step": 65607,
      "training_step_time": 7.685645341873169
    },
    {
      "epoch": 6.256866455078125e-05,
      "model_forward_time": 1.712916612625122,
      "step": 65608
    },
    {
      "epoch": 6.256866455078125e-05,
      "step": 65608,
      "training_step_time": 7.682911157608032
    },
    {
      "epoch": 6.256961822509765e-05,
      "model_forward_time": 1.7187566757202148,
      "step": 65609
    },
    {
      "epoch": 6.256961822509765e-05,
      "step": 65609,
      "training_step_time": 8.41585898399353
    },
    {
      "epoch": 6.257057189941406e-05,
      "grad_norm": 0.10568831861019135,
      "learning_rate": 2.8997214089000404e-06,
      "loss": 0.0563,
      "step": 65610
    },
    {
      "epoch": 6.257057189941406e-05,
      "model_forward_time": 1.722301959991455,
      "step": 65610
    },
    {
      "dynamics_loss_avg": 0.05794894956052303,
      "epoch": 6.257057189941406e-05,
      "step": 65610
    },
    {
      "action_loss_avg": 0.002124023882788606,
      "epoch": 6.257057189941406e-05,
      "step": 65610
    },
    {
      "epoch": 6.257057189941406e-05,
      "step": 65610,
      "training_step_time": 7.483764886856079
    },
    {
      "epoch": 6.257152557373047e-05,
      "model_forward_time": 1.7186877727508545,
      "step": 65611
    },
    {
      "epoch": 6.257152557373047e-05,
      "step": 65611,
      "training_step_time": 7.677711725234985
    },
    {
      "epoch": 6.257247924804687e-05,
      "model_forward_time": 1.7216627597808838,
      "step": 65612
    },
    {
      "epoch": 6.257247924804687e-05,
      "step": 65612,
      "training_step_time": 8.567269086837769
    },
    {
      "epoch": 6.257343292236328e-05,
      "model_forward_time": 1.719677209854126,
      "step": 65613
    },
    {
      "epoch": 6.257343292236328e-05,
      "step": 65613,
      "training_step_time": 7.731014013290405
    },
    {
      "epoch": 6.257438659667969e-05,
      "model_forward_time": 1.7237985134124756,
      "step": 65614
    },
    {
      "epoch": 6.257438659667969e-05,
      "step": 65614,
      "training_step_time": 7.778002500534058
    },
    {
      "epoch": 6.257534027099609e-05,
      "model_forward_time": 2.8014321327209473,
      "step": 65615
    },
    {
      "epoch": 6.257534027099609e-05,
      "step": 65615,
      "training_step_time": 8.89725112915039
    },
    {
      "epoch": 6.25762939453125e-05,
      "model_forward_time": 1.731334924697876,
      "step": 65616
    },
    {
      "epoch": 6.25762939453125e-05,
      "step": 65616,
      "training_step_time": 7.739194869995117
    },
    {
      "epoch": 6.257724761962891e-05,
      "model_forward_time": 1.7643771171569824,
      "step": 65617
    },
    {
      "epoch": 6.257724761962891e-05,
      "step": 65617,
      "training_step_time": 8.27274489402771
    },
    {
      "epoch": 6.257820129394531e-05,
      "model_forward_time": 1.722799301147461,
      "step": 65618
    },
    {
      "epoch": 6.257820129394531e-05,
      "step": 65618,
      "training_step_time": 8.241811037063599
    },
    {
      "epoch": 6.257915496826172e-05,
      "model_forward_time": 1.7200250625610352,
      "step": 65619
    },
    {
      "epoch": 6.257915496826172e-05,
      "step": 65619,
      "training_step_time": 7.740519285202026
    },
    {
      "epoch": 6.258010864257813e-05,
      "grad_norm": 0.19232766330242157,
      "learning_rate": 2.898221002693899e-06,
      "loss": 0.0535,
      "step": 65620
    },
    {
      "epoch": 6.258010864257813e-05,
      "model_forward_time": 1.727005958557129,
      "step": 65620
    },
    {
      "dynamics_loss_avg": 0.04540296685881913,
      "epoch": 6.258010864257813e-05,
      "step": 65620
    },
    {
      "action_loss_avg": 0.001391521774348803,
      "epoch": 6.258010864257813e-05,
      "step": 65620
    },
    {
      "epoch": 6.258010864257813e-05,
      "step": 65620,
      "training_step_time": 7.688789367675781
    },
    {
      "epoch": 6.258106231689453e-05,
      "model_forward_time": 1.719686508178711,
      "step": 65621
    },
    {
      "epoch": 6.258106231689453e-05,
      "step": 65621,
      "training_step_time": 7.630283832550049
    },
    {
      "epoch": 6.258201599121094e-05,
      "model_forward_time": 1.7173206806182861,
      "step": 65622
    },
    {
      "epoch": 6.258201599121094e-05,
      "step": 65622,
      "training_step_time": 7.712756156921387
    },
    {
      "epoch": 6.258296966552735e-05,
      "model_forward_time": 1.7293996810913086,
      "step": 65623
    },
    {
      "epoch": 6.258296966552735e-05,
      "step": 65623,
      "training_step_time": 8.842832803726196
    },
    {
      "epoch": 6.258392333984375e-05,
      "model_forward_time": 2.825873851776123,
      "step": 65624
    },
    {
      "epoch": 6.258392333984375e-05,
      "step": 65624,
      "training_step_time": 8.488889217376709
    },
    {
      "epoch": 6.258487701416016e-05,
      "model_forward_time": 1.3584964275360107,
      "step": 65625
    },
    {
      "epoch": 6.258487701416016e-05,
      "step": 65625,
      "training_step_time": 9.908159017562866
    },
    {
      "epoch": 6.258583068847657e-05,
      "model_forward_time": 1.431114912033081,
      "step": 65626
    },
    {
      "epoch": 6.258583068847657e-05,
      "step": 65626,
      "training_step_time": 7.7491841316223145
    },
    {
      "epoch": 6.258678436279296e-05,
      "model_forward_time": 1.7335243225097656,
      "step": 65627
    },
    {
      "epoch": 6.258678436279296e-05,
      "step": 65627,
      "training_step_time": 7.6838219165802
    },
    {
      "epoch": 6.258773803710938e-05,
      "model_forward_time": 1.718721866607666,
      "step": 65628
    },
    {
      "epoch": 6.258773803710938e-05,
      "step": 65628,
      "training_step_time": 8.521179914474487
    },
    {
      "epoch": 6.258869171142579e-05,
      "model_forward_time": 1.7167506217956543,
      "step": 65629
    },
    {
      "epoch": 6.258869171142579e-05,
      "step": 65629,
      "training_step_time": 7.646718263626099
    },
    {
      "epoch": 6.258964538574218e-05,
      "grad_norm": 0.08729172497987747,
      "learning_rate": 2.896720826335151e-06,
      "loss": 0.0564,
      "step": 65630
    },
    {
      "epoch": 6.258964538574218e-05,
      "model_forward_time": 1.7254078388214111,
      "step": 65630
    },
    {
      "dynamics_loss_avg": 0.08024871926754713,
      "epoch": 6.258964538574218e-05,
      "step": 65630
    },
    {
      "action_loss_avg": 0.0019126022467389702,
      "epoch": 6.258964538574218e-05,
      "step": 65630
    },
    {
      "epoch": 6.258964538574218e-05,
      "step": 65630,
      "training_step_time": 7.645755767822266
    },
    {
      "epoch": 6.25905990600586e-05,
      "model_forward_time": 1.7180016040802002,
      "step": 65631
    },
    {
      "epoch": 6.25905990600586e-05,
      "step": 65631,
      "training_step_time": 7.740136384963989
    },
    {
      "epoch": 6.2591552734375e-05,
      "model_forward_time": 1.7218027114868164,
      "step": 65632
    },
    {
      "epoch": 6.2591552734375e-05,
      "step": 65632,
      "training_step_time": 8.72360897064209
    },
    {
      "epoch": 6.25925064086914e-05,
      "model_forward_time": 1.7253227233886719,
      "step": 65633
    },
    {
      "epoch": 6.25925064086914e-05,
      "step": 65633,
      "training_step_time": 7.719034671783447
    },
    {
      "epoch": 6.259346008300781e-05,
      "model_forward_time": 2.9251034259796143,
      "step": 65634
    },
    {
      "epoch": 6.259346008300781e-05,
      "step": 65634,
      "training_step_time": 8.597213745117188
    },
    {
      "epoch": 6.259441375732422e-05,
      "model_forward_time": 1.8001022338867188,
      "step": 65635
    },
    {
      "epoch": 6.259441375732422e-05,
      "step": 65635,
      "training_step_time": 9.391909837722778
    },
    {
      "epoch": 6.259536743164062e-05,
      "model_forward_time": 1.7164201736450195,
      "step": 65636
    },
    {
      "epoch": 6.259536743164062e-05,
      "step": 65636,
      "training_step_time": 7.661447286605835
    },
    {
      "epoch": 6.259632110595703e-05,
      "model_forward_time": 1.7300419807434082,
      "step": 65637
    },
    {
      "epoch": 6.259632110595703e-05,
      "step": 65637,
      "training_step_time": 7.802339553833008
    },
    {
      "epoch": 6.259727478027344e-05,
      "model_forward_time": 1.7284207344055176,
      "step": 65638
    },
    {
      "epoch": 6.259727478027344e-05,
      "step": 65638,
      "training_step_time": 8.439301490783691
    },
    {
      "epoch": 6.259822845458984e-05,
      "model_forward_time": 1.719890832901001,
      "step": 65639
    },
    {
      "epoch": 6.259822845458984e-05,
      "step": 65639,
      "training_step_time": 7.72049617767334
    },
    {
      "epoch": 6.259918212890625e-05,
      "grad_norm": 0.1250385195016861,
      "learning_rate": 2.8952208799878495e-06,
      "loss": 0.0562,
      "step": 65640
    },
    {
      "epoch": 6.259918212890625e-05,
      "model_forward_time": 1.7222931385040283,
      "step": 65640
    },
    {
      "dynamics_loss_avg": 0.052528434246778485,
      "epoch": 6.259918212890625e-05,
      "step": 65640
    },
    {
      "action_loss_avg": 0.0009713138308143244,
      "epoch": 6.259918212890625e-05,
      "step": 65640
    },
    {
      "epoch": 6.259918212890625e-05,
      "step": 65640,
      "training_step_time": 7.790310382843018
    },
    {
      "epoch": 6.260013580322266e-05,
      "model_forward_time": 1.7197153568267822,
      "step": 65641
    },
    {
      "epoch": 6.260013580322266e-05,
      "step": 65641,
      "training_step_time": 8.454514265060425
    },
    {
      "epoch": 6.260108947753906e-05,
      "model_forward_time": 1.719444751739502,
      "step": 65642
    },
    {
      "epoch": 6.260108947753906e-05,
      "step": 65642,
      "training_step_time": 8.883130550384521
    },
    {
      "epoch": 6.260204315185547e-05,
      "model_forward_time": 2.8632633686065674,
      "step": 65643
    },
    {
      "epoch": 6.260204315185547e-05,
      "step": 65643,
      "training_step_time": 8.518753290176392
    },
    {
      "epoch": 6.260299682617188e-05,
      "model_forward_time": 1.7772178649902344,
      "step": 65644
    },
    {
      "epoch": 6.260299682617188e-05,
      "step": 65644,
      "training_step_time": 7.763268232345581
    },
    {
      "epoch": 6.260395050048828e-05,
      "model_forward_time": 1.7443993091583252,
      "step": 65645
    },
    {
      "epoch": 6.260395050048828e-05,
      "step": 65645,
      "training_step_time": 8.346467971801758
    },
    {
      "epoch": 6.260490417480469e-05,
      "model_forward_time": 1.7133164405822754,
      "step": 65646
    },
    {
      "epoch": 6.260490417480469e-05,
      "step": 65646,
      "training_step_time": 7.79386830329895
    },
    {
      "epoch": 6.26058578491211e-05,
      "model_forward_time": 1.7284417152404785,
      "step": 65647
    },
    {
      "epoch": 6.26058578491211e-05,
      "step": 65647,
      "training_step_time": 7.628340005874634
    },
    {
      "epoch": 6.26068115234375e-05,
      "model_forward_time": 1.7170689105987549,
      "step": 65648
    },
    {
      "epoch": 6.26068115234375e-05,
      "step": 65648,
      "training_step_time": 8.476432085037231
    },
    {
      "epoch": 6.260776519775391e-05,
      "model_forward_time": 1.7220447063446045,
      "step": 65649
    },
    {
      "epoch": 6.260776519775391e-05,
      "step": 65649,
      "training_step_time": 7.760944843292236
    },
    {
      "epoch": 6.260871887207032e-05,
      "grad_norm": 0.08454785495996475,
      "learning_rate": 2.89372116381603e-06,
      "loss": 0.0557,
      "step": 65650
    },
    {
      "epoch": 6.260871887207032e-05,
      "model_forward_time": 1.7251582145690918,
      "step": 65650
    },
    {
      "dynamics_loss_avg": 0.054693184047937396,
      "epoch": 6.260871887207032e-05,
      "step": 65650
    },
    {
      "action_loss_avg": 0.0008946653688326478,
      "epoch": 6.260871887207032e-05,
      "step": 65650
    },
    {
      "epoch": 6.260871887207032e-05,
      "step": 65650,
      "training_step_time": 8.924131870269775
    },
    {
      "epoch": 6.260967254638672e-05,
      "model_forward_time": 1.726724624633789,
      "step": 65651
    },
    {
      "epoch": 6.260967254638672e-05,
      "step": 65651,
      "training_step_time": 9.54385232925415
    },
    {
      "epoch": 6.261062622070313e-05,
      "model_forward_time": 2.9000251293182373,
      "step": 65652
    },
    {
      "epoch": 6.261062622070313e-05,
      "step": 65652,
      "training_step_time": 7.753910303115845
    },
    {
      "epoch": 6.261157989501954e-05,
      "model_forward_time": 1.7436771392822266,
      "step": 65653
    },
    {
      "epoch": 6.261157989501954e-05,
      "step": 65653,
      "training_step_time": 7.783621072769165
    },
    {
      "epoch": 6.261253356933593e-05,
      "model_forward_time": 1.7217450141906738,
      "step": 65654
    },
    {
      "epoch": 6.261253356933593e-05,
      "step": 65654,
      "training_step_time": 7.746503114700317
    },
    {
      "epoch": 6.261348724365234e-05,
      "model_forward_time": 1.7294085025787354,
      "step": 65655
    },
    {
      "epoch": 6.261348724365234e-05,
      "step": 65655,
      "training_step_time": 8.827191829681396
    },
    {
      "epoch": 6.261444091796876e-05,
      "model_forward_time": 1.7236125469207764,
      "step": 65656
    },
    {
      "epoch": 6.261444091796876e-05,
      "step": 65656,
      "training_step_time": 7.73362135887146
    },
    {
      "epoch": 6.261539459228515e-05,
      "model_forward_time": 1.7233870029449463,
      "step": 65657
    },
    {
      "epoch": 6.261539459228515e-05,
      "step": 65657,
      "training_step_time": 7.6762003898620605
    },
    {
      "epoch": 6.261634826660156e-05,
      "model_forward_time": 1.71494722366333,
      "step": 65658
    },
    {
      "epoch": 6.261634826660156e-05,
      "step": 65658,
      "training_step_time": 8.662794828414917
    },
    {
      "epoch": 6.261730194091797e-05,
      "model_forward_time": 1.76170015335083,
      "step": 65659
    },
    {
      "epoch": 6.261730194091797e-05,
      "step": 65659,
      "training_step_time": 8.336451768875122
    },
    {
      "epoch": 6.261825561523437e-05,
      "grad_norm": 0.10502421855926514,
      "learning_rate": 2.892221677983693e-06,
      "loss": 0.056,
      "step": 65660
    },
    {
      "epoch": 6.261825561523437e-05,
      "model_forward_time": 1.7253100872039795,
      "step": 65660
    },
    {
      "dynamics_loss_avg": 0.061136084422469136,
      "epoch": 6.261825561523437e-05,
      "step": 65660
    },
    {
      "action_loss_avg": 0.0024252122035250067,
      "epoch": 6.261825561523437e-05,
      "step": 65660
    },
    {
      "epoch": 6.261825561523437e-05,
      "step": 65660,
      "training_step_time": 7.71130633354187
    },
    {
      "epoch": 6.261920928955078e-05,
      "model_forward_time": 2.814772844314575,
      "step": 65661
    },
    {
      "epoch": 6.261920928955078e-05,
      "step": 65661,
      "training_step_time": 7.756440162658691
    },
    {
      "epoch": 6.26201629638672e-05,
      "model_forward_time": 1.725898265838623,
      "step": 65662
    },
    {
      "epoch": 6.26201629638672e-05,
      "step": 65662,
      "training_step_time": 8.79720950126648
    },
    {
      "epoch": 6.262111663818359e-05,
      "model_forward_time": 1.8277044296264648,
      "step": 65663
    },
    {
      "epoch": 6.262111663818359e-05,
      "step": 65663,
      "training_step_time": 9.368587255477905
    },
    {
      "epoch": 6.26220703125e-05,
      "model_forward_time": 1.7305707931518555,
      "step": 65664
    },
    {
      "epoch": 6.26220703125e-05,
      "step": 65664,
      "training_step_time": 7.759351491928101
    },
    {
      "epoch": 6.262302398681641e-05,
      "model_forward_time": 1.7140967845916748,
      "step": 65665
    },
    {
      "epoch": 6.262302398681641e-05,
      "step": 65665,
      "training_step_time": 7.767160177230835
    },
    {
      "epoch": 6.262397766113281e-05,
      "model_forward_time": 1.7176010608673096,
      "step": 65666
    },
    {
      "epoch": 6.262397766113281e-05,
      "step": 65666,
      "training_step_time": 8.537465810775757
    },
    {
      "epoch": 6.262493133544922e-05,
      "model_forward_time": 1.7233002185821533,
      "step": 65667
    },
    {
      "epoch": 6.262493133544922e-05,
      "step": 65667,
      "training_step_time": 7.817129850387573
    },
    {
      "epoch": 6.262588500976563e-05,
      "model_forward_time": 1.7193400859832764,
      "step": 65668
    },
    {
      "epoch": 6.262588500976563e-05,
      "step": 65668,
      "training_step_time": 9.709764957427979
    },
    {
      "epoch": 6.262683868408203e-05,
      "model_forward_time": 1.7337806224822998,
      "step": 65669
    },
    {
      "epoch": 6.262683868408203e-05,
      "step": 65669,
      "training_step_time": 9.443999290466309
    },
    {
      "epoch": 6.262779235839844e-05,
      "grad_norm": 0.1039525717496872,
      "learning_rate": 2.8907224226548284e-06,
      "loss": 0.058,
      "step": 65670
    },
    {
      "epoch": 6.262779235839844e-05,
      "model_forward_time": 1.7095541954040527,
      "step": 65670
    },
    {
      "dynamics_loss_avg": 0.05023531131446361,
      "epoch": 6.262779235839844e-05,
      "step": 65670
    },
    {
      "action_loss_avg": 0.0013754477084148675,
      "epoch": 6.262779235839844e-05,
      "step": 65670
    },
    {
      "epoch": 6.262779235839844e-05,
      "step": 65670,
      "training_step_time": 7.7181174755096436
    },
    {
      "epoch": 6.262874603271485e-05,
      "model_forward_time": 2.885819435119629,
      "step": 65671
    },
    {
      "epoch": 6.262874603271485e-05,
      "step": 65671,
      "training_step_time": 9.635258674621582
    },
    {
      "epoch": 6.262969970703125e-05,
      "model_forward_time": 1.8085856437683105,
      "step": 65672
    },
    {
      "epoch": 6.262969970703125e-05,
      "step": 65672,
      "training_step_time": 8.48002290725708
    },
    {
      "epoch": 6.263065338134766e-05,
      "model_forward_time": 1.7297682762145996,
      "step": 65673
    },
    {
      "epoch": 6.263065338134766e-05,
      "step": 65673,
      "training_step_time": 7.7711029052734375
    },
    {
      "epoch": 6.263160705566407e-05,
      "model_forward_time": 1.7160818576812744,
      "step": 65674
    },
    {
      "epoch": 6.263160705566407e-05,
      "step": 65674,
      "training_step_time": 7.788020849227905
    },
    {
      "epoch": 6.263256072998047e-05,
      "model_forward_time": 1.7175006866455078,
      "step": 65675
    },
    {
      "epoch": 6.263256072998047e-05,
      "step": 65675,
      "training_step_time": 7.670734643936157
    },
    {
      "epoch": 6.263351440429688e-05,
      "model_forward_time": 1.741924524307251,
      "step": 65676
    },
    {
      "epoch": 6.263351440429688e-05,
      "step": 65676,
      "training_step_time": 7.728748321533203
    },
    {
      "epoch": 6.263446807861329e-05,
      "model_forward_time": 1.7179126739501953,
      "step": 65677
    },
    {
      "epoch": 6.263446807861329e-05,
      "step": 65677,
      "training_step_time": 7.7846620082855225
    },
    {
      "epoch": 6.263542175292968e-05,
      "model_forward_time": 1.7075626850128174,
      "step": 65678
    },
    {
      "epoch": 6.263542175292968e-05,
      "step": 65678,
      "training_step_time": 7.830032825469971
    },
    {
      "epoch": 6.26363754272461e-05,
      "model_forward_time": 1.7471449375152588,
      "step": 65679
    },
    {
      "epoch": 6.26363754272461e-05,
      "step": 65679,
      "training_step_time": 7.779404401779175
    },
    {
      "epoch": 6.26373291015625e-05,
      "grad_norm": 0.08998283743858337,
      "learning_rate": 2.889223397993387e-06,
      "loss": 0.0508,
      "step": 65680
    },
    {
      "epoch": 6.26373291015625e-05,
      "model_forward_time": 2.9239861965179443,
      "step": 65680
    },
    {
      "dynamics_loss_avg": 0.04160702675580978,
      "epoch": 6.26373291015625e-05,
      "step": 65680
    },
    {
      "action_loss_avg": 0.001312797493301332,
      "epoch": 6.26373291015625e-05,
      "step": 65680
    },
    {
      "epoch": 6.26373291015625e-05,
      "step": 65680,
      "training_step_time": 7.8534255027771
    },
    {
      "epoch": 6.26382827758789e-05,
      "model_forward_time": 1.7852439880371094,
      "step": 65681
    },
    {
      "epoch": 6.26382827758789e-05,
      "step": 65681,
      "training_step_time": 7.605957746505737
    },
    {
      "epoch": 6.263923645019531e-05,
      "model_forward_time": 1.724696397781372,
      "step": 65682
    },
    {
      "epoch": 6.263923645019531e-05,
      "step": 65682,
      "training_step_time": 8.607234716415405
    },
    {
      "epoch": 6.264019012451173e-05,
      "model_forward_time": 1.7132396697998047,
      "step": 65683
    },
    {
      "epoch": 6.264019012451173e-05,
      "step": 65683,
      "training_step_time": 7.616185665130615
    },
    {
      "epoch": 6.264114379882812e-05,
      "model_forward_time": 1.7283945083618164,
      "step": 65684
    },
    {
      "epoch": 6.264114379882812e-05,
      "step": 65684,
      "training_step_time": 7.516220331192017
    },
    {
      "epoch": 6.264209747314453e-05,
      "model_forward_time": 1.7216756343841553,
      "step": 65685
    },
    {
      "epoch": 6.264209747314453e-05,
      "step": 65685,
      "training_step_time": 8.175611019134521
    },
    {
      "epoch": 6.264305114746094e-05,
      "model_forward_time": 1.7148587703704834,
      "step": 65686
    },
    {
      "epoch": 6.264305114746094e-05,
      "step": 65686,
      "training_step_time": 8.833972454071045
    },
    {
      "epoch": 6.264400482177734e-05,
      "model_forward_time": 1.719236135482788,
      "step": 65687
    },
    {
      "epoch": 6.264400482177734e-05,
      "step": 65687,
      "training_step_time": 8.587363958358765
    },
    {
      "epoch": 6.264495849609375e-05,
      "model_forward_time": 1.7019379138946533,
      "step": 65688
    },
    {
      "epoch": 6.264495849609375e-05,
      "step": 65688,
      "training_step_time": 7.5070600509643555
    },
    {
      "epoch": 6.264591217041015e-05,
      "model_forward_time": 2.927243947982788,
      "step": 65689
    },
    {
      "epoch": 6.264591217041015e-05,
      "step": 65689,
      "training_step_time": 7.7100489139556885
    },
    {
      "epoch": 6.264686584472656e-05,
      "grad_norm": 0.1151895523071289,
      "learning_rate": 2.8877246041633027e-06,
      "loss": 0.0589,
      "step": 65690
    },
    {
      "epoch": 6.264686584472656e-05,
      "model_forward_time": 1.7534112930297852,
      "step": 65690
    },
    {
      "dynamics_loss_avg": 0.05204693581908941,
      "epoch": 6.264686584472656e-05,
      "step": 65690
    },
    {
      "action_loss_avg": 0.0025784832949284466,
      "epoch": 6.264686584472656e-05,
      "step": 65690
    },
    {
      "epoch": 6.264686584472656e-05,
      "step": 65690,
      "training_step_time": 9.144155025482178
    },
    {
      "epoch": 6.264781951904297e-05,
      "model_forward_time": 1.7209234237670898,
      "step": 65691
    },
    {
      "epoch": 6.264781951904297e-05,
      "step": 65691,
      "training_step_time": 7.5625646114349365
    },
    {
      "epoch": 6.264877319335937e-05,
      "model_forward_time": 1.7202789783477783,
      "step": 65692
    },
    {
      "epoch": 6.264877319335937e-05,
      "step": 65692,
      "training_step_time": 7.608995676040649
    },
    {
      "epoch": 6.264972686767578e-05,
      "model_forward_time": 1.7215089797973633,
      "step": 65693
    },
    {
      "epoch": 6.264972686767578e-05,
      "step": 65693,
      "training_step_time": 7.511693239212036
    },
    {
      "epoch": 6.265068054199219e-05,
      "model_forward_time": 1.7159724235534668,
      "step": 65694
    },
    {
      "epoch": 6.265068054199219e-05,
      "step": 65694,
      "training_step_time": 7.587994813919067
    },
    {
      "epoch": 6.265163421630859e-05,
      "model_forward_time": 1.7258350849151611,
      "step": 65695
    },
    {
      "epoch": 6.265163421630859e-05,
      "step": 65695,
      "training_step_time": 7.61760687828064
    },
    {
      "epoch": 6.2652587890625e-05,
      "model_forward_time": 1.709299087524414,
      "step": 65696
    },
    {
      "epoch": 6.2652587890625e-05,
      "step": 65696,
      "training_step_time": 7.7417151927948
    },
    {
      "epoch": 6.265354156494141e-05,
      "model_forward_time": 1.714888095855713,
      "step": 65697
    },
    {
      "epoch": 6.265354156494141e-05,
      "step": 65697,
      "training_step_time": 7.540771722793579
    },
    {
      "epoch": 6.26544952392578e-05,
      "model_forward_time": 2.889383316040039,
      "step": 65698
    },
    {
      "epoch": 6.26544952392578e-05,
      "step": 65698,
      "training_step_time": 7.6033265590667725
    },
    {
      "epoch": 6.265544891357422e-05,
      "model_forward_time": 1.7230346202850342,
      "step": 65699
    },
    {
      "epoch": 6.265544891357422e-05,
      "step": 65699,
      "training_step_time": 8.262723445892334
    },
    {
      "epoch": 6.265640258789063e-05,
      "grad_norm": 0.11193934082984924,
      "learning_rate": 2.886226041328478e-06,
      "loss": 0.058,
      "step": 65700
    },
    {
      "epoch": 6.265640258789063e-05,
      "model_forward_time": 1.8287427425384521,
      "step": 65700
    },
    {
      "dynamics_loss_avg": 0.04468913562595844,
      "epoch": 6.265640258789063e-05,
      "step": 65700
    },
    {
      "action_loss_avg": 0.0016035414271755144,
      "epoch": 6.265640258789063e-05,
      "step": 65700
    },
    {
      "epoch": 6.265640258789063e-05,
      "step": 65700,
      "training_step_time": 7.637180805206299
    },
    {
      "epoch": 6.265735626220702e-05,
      "model_forward_time": 1.7241740226745605,
      "step": 65701
    },
    {
      "epoch": 6.265735626220702e-05,
      "step": 65701,
      "training_step_time": 7.668262958526611
    },
    {
      "epoch": 6.265830993652344e-05,
      "model_forward_time": 1.7223646640777588,
      "step": 65702
    },
    {
      "epoch": 6.265830993652344e-05,
      "step": 65702,
      "training_step_time": 7.523227214813232
    },
    {
      "epoch": 6.265926361083985e-05,
      "model_forward_time": 1.7212493419647217,
      "step": 65703
    },
    {
      "epoch": 6.265926361083985e-05,
      "step": 65703,
      "training_step_time": 7.548011064529419
    },
    {
      "epoch": 6.266021728515624e-05,
      "model_forward_time": 1.7250845432281494,
      "step": 65704
    },
    {
      "epoch": 6.266021728515624e-05,
      "step": 65704,
      "training_step_time": 8.372636556625366
    },
    {
      "epoch": 6.266117095947265e-05,
      "model_forward_time": 1.7265324592590332,
      "step": 65705
    },
    {
      "epoch": 6.266117095947265e-05,
      "step": 65705,
      "training_step_time": 8.127557039260864
    },
    {
      "epoch": 6.266212463378907e-05,
      "model_forward_time": 1.7186341285705566,
      "step": 65706
    },
    {
      "epoch": 6.266212463378907e-05,
      "step": 65706,
      "training_step_time": 7.500044345855713
    },
    {
      "epoch": 6.266307830810546e-05,
      "model_forward_time": 2.8281710147857666,
      "step": 65707
    },
    {
      "epoch": 6.266307830810546e-05,
      "step": 65707,
      "training_step_time": 7.641391277313232
    },
    {
      "epoch": 6.266403198242187e-05,
      "model_forward_time": 1.7166221141815186,
      "step": 65708
    },
    {
      "epoch": 6.266403198242187e-05,
      "step": 65708,
      "training_step_time": 8.734504461288452
    },
    {
      "epoch": 6.266498565673828e-05,
      "model_forward_time": 1.8299190998077393,
      "step": 65709
    },
    {
      "epoch": 6.266498565673828e-05,
      "step": 65709,
      "training_step_time": 7.691427946090698
    },
    {
      "epoch": 6.266593933105468e-05,
      "grad_norm": 0.084034763276577,
      "learning_rate": 2.8847277096527957e-06,
      "loss": 0.0535,
      "step": 65710
    },
    {
      "epoch": 6.266593933105468e-05,
      "model_forward_time": 1.7306439876556396,
      "step": 65710
    },
    {
      "dynamics_loss_avg": 0.055912654101848605,
      "epoch": 6.266593933105468e-05,
      "step": 65710
    },
    {
      "action_loss_avg": 0.0032837602309882643,
      "epoch": 6.266593933105468e-05,
      "step": 65710
    },
    {
      "epoch": 6.266593933105468e-05,
      "step": 65710,
      "training_step_time": 7.6533825397491455
    },
    {
      "epoch": 6.266689300537109e-05,
      "model_forward_time": 1.7186627388000488,
      "step": 65711
    },
    {
      "epoch": 6.266689300537109e-05,
      "step": 65711,
      "training_step_time": 7.5762128829956055
    },
    {
      "epoch": 6.26678466796875e-05,
      "model_forward_time": 1.7191245555877686,
      "step": 65712
    },
    {
      "epoch": 6.26678466796875e-05,
      "step": 65712,
      "training_step_time": 7.577756404876709
    },
    {
      "epoch": 6.26688003540039e-05,
      "model_forward_time": 1.7246994972229004,
      "step": 65713
    },
    {
      "epoch": 6.26688003540039e-05,
      "step": 65713,
      "training_step_time": 7.670449733734131
    },
    {
      "epoch": 6.266975402832031e-05,
      "model_forward_time": 1.7193095684051514,
      "step": 65714
    },
    {
      "epoch": 6.266975402832031e-05,
      "step": 65714,
      "training_step_time": 8.364728212356567
    },
    {
      "epoch": 6.267070770263672e-05,
      "model_forward_time": 1.7200279235839844,
      "step": 65715
    },
    {
      "epoch": 6.267070770263672e-05,
      "step": 65715,
      "training_step_time": 7.602257490158081
    },
    {
      "epoch": 6.267166137695312e-05,
      "model_forward_time": 1.7182624340057373,
      "step": 65716
    },
    {
      "epoch": 6.267166137695312e-05,
      "step": 65716,
      "training_step_time": 8.7168550491333
    },
    {
      "epoch": 6.267261505126953e-05,
      "model_forward_time": 1.7136931419372559,
      "step": 65717
    },
    {
      "epoch": 6.267261505126953e-05,
      "step": 65717,
      "training_step_time": 7.742502450942993
    },
    {
      "epoch": 6.267356872558594e-05,
      "model_forward_time": 1.7959108352661133,
      "step": 65718
    },
    {
      "epoch": 6.267356872558594e-05,
      "step": 65718,
      "training_step_time": 7.704850196838379
    },
    {
      "epoch": 6.267452239990234e-05,
      "model_forward_time": 1.724071979522705,
      "step": 65719
    },
    {
      "epoch": 6.267452239990234e-05,
      "step": 65719,
      "training_step_time": 7.617297172546387
    },
    {
      "epoch": 6.267547607421875e-05,
      "grad_norm": 0.11342679709196091,
      "learning_rate": 2.8832296093001105e-06,
      "loss": 0.0565,
      "step": 65720
    },
    {
      "epoch": 6.267547607421875e-05,
      "model_forward_time": 1.7171063423156738,
      "step": 65720
    },
    {
      "dynamics_loss_avg": 0.04712866609916091,
      "epoch": 6.267547607421875e-05,
      "step": 65720
    },
    {
      "action_loss_avg": 0.0010319246706785635,
      "epoch": 6.267547607421875e-05,
      "step": 65720
    },
    {
      "epoch": 6.267547607421875e-05,
      "step": 65720,
      "training_step_time": 7.617625713348389
    },
    {
      "epoch": 6.267642974853516e-05,
      "model_forward_time": 1.7146074771881104,
      "step": 65721
    },
    {
      "epoch": 6.267642974853516e-05,
      "step": 65721,
      "training_step_time": 7.560789346694946
    },
    {
      "epoch": 6.267738342285156e-05,
      "model_forward_time": 1.7368996143341064,
      "step": 65722
    },
    {
      "epoch": 6.267738342285156e-05,
      "step": 65722,
      "training_step_time": 7.68206787109375
    },
    {
      "epoch": 6.267833709716797e-05,
      "model_forward_time": 1.71299409866333,
      "step": 65723
    },
    {
      "epoch": 6.267833709716797e-05,
      "step": 65723,
      "training_step_time": 7.689728498458862
    },
    {
      "epoch": 6.267929077148438e-05,
      "model_forward_time": 1.7157890796661377,
      "step": 65724
    },
    {
      "epoch": 6.267929077148438e-05,
      "step": 65724,
      "training_step_time": 7.681520700454712
    },
    {
      "epoch": 6.268024444580078e-05,
      "model_forward_time": 1.76521897315979,
      "step": 65725
    },
    {
      "epoch": 6.268024444580078e-05,
      "step": 65725,
      "training_step_time": 8.200975179672241
    },
    {
      "epoch": 6.268119812011719e-05,
      "model_forward_time": 2.871741533279419,
      "step": 65726
    },
    {
      "epoch": 6.268119812011719e-05,
      "step": 65726,
      "training_step_time": 7.711899518966675
    },
    {
      "epoch": 6.26821517944336e-05,
      "model_forward_time": 1.756765365600586,
      "step": 65727
    },
    {
      "epoch": 6.26821517944336e-05,
      "step": 65727,
      "training_step_time": 7.55431056022644
    },
    {
      "epoch": 6.268310546875e-05,
      "model_forward_time": 1.3704826831817627,
      "step": 65728
    },
    {
      "epoch": 6.268310546875e-05,
      "step": 65728,
      "training_step_time": 7.61202073097229
    },
    {
      "epoch": 6.26840591430664e-05,
      "model_forward_time": 1.7651355266571045,
      "step": 65729
    },
    {
      "epoch": 6.26840591430664e-05,
      "step": 65729,
      "training_step_time": 7.357834577560425
    },
    {
      "epoch": 6.268501281738282e-05,
      "grad_norm": 0.10613321512937546,
      "learning_rate": 2.881731740434255e-06,
      "loss": 0.0568,
      "step": 65730
    },
    {
      "epoch": 6.268501281738282e-05,
      "model_forward_time": 1.7228760719299316,
      "step": 65730
    },
    {
      "dynamics_loss_avg": 0.04683734457939863,
      "epoch": 6.268501281738282e-05,
      "step": 65730
    },
    {
      "action_loss_avg": 0.001661554689053446,
      "epoch": 6.268501281738282e-05,
      "step": 65730
    },
    {
      "epoch": 6.268501281738282e-05,
      "step": 65730,
      "training_step_time": 7.567758083343506
    },
    {
      "epoch": 6.268596649169921e-05,
      "model_forward_time": 1.7263178825378418,
      "step": 65731
    },
    {
      "epoch": 6.268596649169921e-05,
      "step": 65731,
      "training_step_time": 7.754645586013794
    },
    {
      "epoch": 6.268692016601562e-05,
      "model_forward_time": 1.7203493118286133,
      "step": 65732
    },
    {
      "epoch": 6.268692016601562e-05,
      "step": 65732,
      "training_step_time": 7.632369756698608
    },
    {
      "epoch": 6.268787384033204e-05,
      "model_forward_time": 1.710986614227295,
      "step": 65733
    },
    {
      "epoch": 6.268787384033204e-05,
      "step": 65733,
      "training_step_time": 8.565987586975098
    },
    {
      "epoch": 6.268882751464843e-05,
      "model_forward_time": 1.7250604629516602,
      "step": 65734
    },
    {
      "epoch": 6.268882751464843e-05,
      "step": 65734,
      "training_step_time": 7.728352069854736
    },
    {
      "epoch": 6.268978118896484e-05,
      "model_forward_time": 2.8813233375549316,
      "step": 65735
    },
    {
      "epoch": 6.268978118896484e-05,
      "step": 65735,
      "training_step_time": 9.278419256210327
    },
    {
      "epoch": 6.269073486328125e-05,
      "model_forward_time": 1.721876859664917,
      "step": 65736
    },
    {
      "epoch": 6.269073486328125e-05,
      "step": 65736,
      "training_step_time": 8.859215259552002
    },
    {
      "epoch": 6.269168853759765e-05,
      "model_forward_time": 1.8200647830963135,
      "step": 65737
    },
    {
      "epoch": 6.269168853759765e-05,
      "step": 65737,
      "training_step_time": 7.7238099575042725
    },
    {
      "epoch": 6.269264221191406e-05,
      "model_forward_time": 1.721101999282837,
      "step": 65738
    },
    {
      "epoch": 6.269264221191406e-05,
      "step": 65738,
      "training_step_time": 7.806468963623047
    },
    {
      "epoch": 6.269359588623047e-05,
      "model_forward_time": 1.7124736309051514,
      "step": 65739
    },
    {
      "epoch": 6.269359588623047e-05,
      "step": 65739,
      "training_step_time": 7.663044214248657
    },
    {
      "epoch": 6.269454956054687e-05,
      "grad_norm": 0.08948078751564026,
      "learning_rate": 2.880234103219029e-06,
      "loss": 0.0539,
      "step": 65740
    },
    {
      "epoch": 6.269454956054687e-05,
      "model_forward_time": 1.7110509872436523,
      "step": 65740
    },
    {
      "dynamics_loss_avg": 0.06040707305073738,
      "epoch": 6.269454956054687e-05,
      "step": 65740
    },
    {
      "action_loss_avg": 0.001759026641957462,
      "epoch": 6.269454956054687e-05,
      "step": 65740
    },
    {
      "epoch": 6.269454956054687e-05,
      "step": 65740,
      "training_step_time": 7.721059560775757
    },
    {
      "epoch": 6.269550323486328e-05,
      "model_forward_time": 1.7192206382751465,
      "step": 65741
    },
    {
      "epoch": 6.269550323486328e-05,
      "step": 65741,
      "training_step_time": 8.650558710098267
    },
    {
      "epoch": 6.269645690917969e-05,
      "model_forward_time": 1.7152867317199707,
      "step": 65742
    },
    {
      "epoch": 6.269645690917969e-05,
      "step": 65742,
      "training_step_time": 6.904867887496948
    },
    {
      "epoch": 6.269741058349609e-05,
      "model_forward_time": 1.7169060707092285,
      "step": 65743
    },
    {
      "epoch": 6.269741058349609e-05,
      "step": 65743,
      "training_step_time": 6.87383246421814
    },
    {
      "epoch": 6.26983642578125e-05,
      "model_forward_time": 1.713766098022461,
      "step": 65744
    },
    {
      "epoch": 6.26983642578125e-05,
      "step": 65744,
      "training_step_time": 6.833806753158569
    },
    {
      "epoch": 6.269931793212891e-05,
      "model_forward_time": 1.723167896270752,
      "step": 65745
    },
    {
      "epoch": 6.269931793212891e-05,
      "step": 65745,
      "training_step_time": 6.839028358459473
    },
    {
      "epoch": 6.270027160644531e-05,
      "model_forward_time": 1.7235894203186035,
      "step": 65746
    },
    {
      "epoch": 6.270027160644531e-05,
      "step": 65746,
      "training_step_time": 6.872791051864624
    },
    {
      "epoch": 6.270122528076172e-05,
      "model_forward_time": 1.7203595638275146,
      "step": 65747
    },
    {
      "epoch": 6.270122528076172e-05,
      "step": 65747,
      "training_step_time": 6.917647361755371
    },
    {
      "epoch": 6.270217895507813e-05,
      "model_forward_time": 1.7151179313659668,
      "step": 65748
    },
    {
      "epoch": 6.270217895507813e-05,
      "step": 65748,
      "training_step_time": 6.85133171081543
    },
    {
      "epoch": 6.270313262939453e-05,
      "model_forward_time": 1.7186059951782227,
      "step": 65749
    },
    {
      "epoch": 6.270313262939453e-05,
      "step": 65749,
      "training_step_time": 6.9080750942230225
    },
    {
      "epoch": 6.270408630371094e-05,
      "grad_norm": 0.09420711547136307,
      "learning_rate": 2.8787366978182153e-06,
      "loss": 0.0589,
      "step": 65750
    },
    {
      "epoch": 6.270408630371094e-05,
      "model_forward_time": 2.8551383018493652,
      "step": 65750
    },
    {
      "dynamics_loss_avg": 0.04321600012481212,
      "epoch": 6.270408630371094e-05,
      "step": 65750
    },
    {
      "action_loss_avg": 0.0011111977102700621,
      "epoch": 6.270408630371094e-05,
      "step": 65750
    },
    {
      "epoch": 6.270408630371094e-05,
      "step": 65750,
      "training_step_time": 10.569788694381714
    },
    {
      "epoch": 6.270503997802735e-05,
      "model_forward_time": 1.7146039009094238,
      "step": 65751
    },
    {
      "epoch": 6.270503997802735e-05,
      "step": 65751,
      "training_step_time": 7.941525220870972
    },
    {
      "epoch": 6.270599365234375e-05,
      "model_forward_time": 1.728480339050293,
      "step": 65752
    },
    {
      "epoch": 6.270599365234375e-05,
      "step": 65752,
      "training_step_time": 6.893298149108887
    },
    {
      "epoch": 6.270694732666016e-05,
      "model_forward_time": 1.7268128395080566,
      "step": 65753
    },
    {
      "epoch": 6.270694732666016e-05,
      "step": 65753,
      "training_step_time": 6.8415892124176025
    },
    {
      "epoch": 6.270790100097657e-05,
      "model_forward_time": 1.7233343124389648,
      "step": 65754
    },
    {
      "epoch": 6.270790100097657e-05,
      "step": 65754,
      "training_step_time": 6.864923000335693
    },
    {
      "epoch": 6.270885467529296e-05,
      "model_forward_time": 1.7147183418273926,
      "step": 65755
    },
    {
      "epoch": 6.270885467529296e-05,
      "step": 65755,
      "training_step_time": 6.8517467975616455
    },
    {
      "epoch": 6.270980834960938e-05,
      "model_forward_time": 1.71280837059021,
      "step": 65756
    },
    {
      "epoch": 6.270980834960938e-05,
      "step": 65756,
      "training_step_time": 6.877992153167725
    },
    {
      "epoch": 6.271076202392579e-05,
      "model_forward_time": 1.7231676578521729,
      "step": 65757
    },
    {
      "epoch": 6.271076202392579e-05,
      "step": 65757,
      "training_step_time": 6.8986687660217285
    },
    {
      "epoch": 6.271171569824218e-05,
      "model_forward_time": 1.7223145961761475,
      "step": 65758
    },
    {
      "epoch": 6.271171569824218e-05,
      "step": 65758,
      "training_step_time": 6.84749698638916
    },
    {
      "epoch": 6.27126693725586e-05,
      "model_forward_time": 2.8326985836029053,
      "step": 65759
    },
    {
      "epoch": 6.27126693725586e-05,
      "step": 65759,
      "training_step_time": 10.68535327911377
    },
    {
      "epoch": 6.2713623046875e-05,
      "grad_norm": 0.0955447405576706,
      "learning_rate": 2.877239524395568e-06,
      "loss": 0.0559,
      "step": 65760
    },
    {
      "epoch": 6.2713623046875e-05,
      "model_forward_time": 1.7247066497802734,
      "step": 65760
    },
    {
      "dynamics_loss_avg": 0.05741968136280775,
      "epoch": 6.2713623046875e-05,
      "step": 65760
    },
    {
      "action_loss_avg": 0.0015600217710016295,
      "epoch": 6.2713623046875e-05,
      "step": 65760
    },
    {
      "epoch": 6.2713623046875e-05,
      "step": 65760,
      "training_step_time": 7.901006698608398
    },
    {
      "epoch": 6.27145767211914e-05,
      "model_forward_time": 1.7167105674743652,
      "step": 65761
    },
    {
      "epoch": 6.27145767211914e-05,
      "step": 65761,
      "training_step_time": 6.912214994430542
    },
    {
      "epoch": 6.271553039550781e-05,
      "model_forward_time": 1.716604471206665,
      "step": 65762
    },
    {
      "epoch": 6.271553039550781e-05,
      "step": 65762,
      "training_step_time": 6.873514413833618
    },
    {
      "epoch": 6.271648406982422e-05,
      "model_forward_time": 1.7158236503601074,
      "step": 65763
    },
    {
      "epoch": 6.271648406982422e-05,
      "step": 65763,
      "training_step_time": 6.909279823303223
    },
    {
      "epoch": 6.271743774414062e-05,
      "model_forward_time": 1.7298479080200195,
      "step": 65764
    },
    {
      "epoch": 6.271743774414062e-05,
      "step": 65764,
      "training_step_time": 6.933921813964844
    },
    {
      "epoch": 6.271839141845703e-05,
      "model_forward_time": 1.7432613372802734,
      "step": 65765
    },
    {
      "epoch": 6.271839141845703e-05,
      "step": 65765,
      "training_step_time": 6.926028251647949
    },
    {
      "epoch": 6.271934509277344e-05,
      "model_forward_time": 1.712374210357666,
      "step": 65766
    },
    {
      "epoch": 6.271934509277344e-05,
      "step": 65766,
      "training_step_time": 6.913599729537964
    },
    {
      "epoch": 6.272029876708984e-05,
      "model_forward_time": 1.7220137119293213,
      "step": 65767
    },
    {
      "epoch": 6.272029876708984e-05,
      "step": 65767,
      "training_step_time": 6.913567066192627
    },
    {
      "epoch": 6.272125244140625e-05,
      "model_forward_time": 2.8620011806488037,
      "step": 65768
    },
    {
      "epoch": 6.272125244140625e-05,
      "step": 65768,
      "training_step_time": 10.547818899154663
    },
    {
      "epoch": 6.272220611572266e-05,
      "model_forward_time": 1.716768741607666,
      "step": 65769
    },
    {
      "epoch": 6.272220611572266e-05,
      "step": 65769,
      "training_step_time": 8.987122535705566
    },
    {
      "epoch": 6.272315979003906e-05,
      "grad_norm": 0.10077876597642899,
      "learning_rate": 2.8757425831148135e-06,
      "loss": 0.0547,
      "step": 65770
    },
    {
      "epoch": 6.272315979003906e-05,
      "model_forward_time": 1.7199199199676514,
      "step": 65770
    },
    {
      "dynamics_loss_avg": 0.05901904758065939,
      "epoch": 6.272315979003906e-05,
      "step": 65770
    },
    {
      "action_loss_avg": 0.0015419028524775058,
      "epoch": 6.272315979003906e-05,
      "step": 65770
    },
    {
      "epoch": 6.272315979003906e-05,
      "step": 65770,
      "training_step_time": 7.584060430526733
    },
    {
      "epoch": 6.272411346435547e-05,
      "model_forward_time": 1.712651252746582,
      "step": 65771
    },
    {
      "epoch": 6.272411346435547e-05,
      "step": 65771,
      "training_step_time": 6.84152626991272
    },
    {
      "epoch": 6.272506713867188e-05,
      "model_forward_time": 1.722041368484497,
      "step": 65772
    },
    {
      "epoch": 6.272506713867188e-05,
      "step": 65772,
      "training_step_time": 6.8077170848846436
    },
    {
      "epoch": 6.272602081298828e-05,
      "model_forward_time": 1.3592777252197266,
      "step": 65773
    },
    {
      "epoch": 6.272602081298828e-05,
      "step": 65773,
      "training_step_time": 6.83791971206665
    },
    {
      "epoch": 6.272697448730469e-05,
      "model_forward_time": 1.7263908386230469,
      "step": 65774
    },
    {
      "epoch": 6.272697448730469e-05,
      "step": 65774,
      "training_step_time": 6.876752853393555
    },
    {
      "epoch": 6.27279281616211e-05,
      "model_forward_time": 1.7307136058807373,
      "step": 65775
    },
    {
      "epoch": 6.27279281616211e-05,
      "step": 65775,
      "training_step_time": 6.815551996231079
    },
    {
      "epoch": 6.27288818359375e-05,
      "model_forward_time": 1.359163522720337,
      "step": 65776
    },
    {
      "epoch": 6.27288818359375e-05,
      "step": 65776,
      "training_step_time": 6.857801914215088
    },
    {
      "epoch": 6.272983551025391e-05,
      "model_forward_time": 1.7630250453948975,
      "step": 65777
    },
    {
      "epoch": 6.272983551025391e-05,
      "step": 65777,
      "training_step_time": 11.07051134109497
    },
    {
      "epoch": 6.273078918457032e-05,
      "model_forward_time": 1.4271965026855469,
      "step": 65778
    },
    {
      "epoch": 6.273078918457032e-05,
      "step": 65778,
      "training_step_time": 8.443392992019653
    },
    {
      "epoch": 6.273174285888672e-05,
      "model_forward_time": 1.7335617542266846,
      "step": 65779
    },
    {
      "epoch": 6.273174285888672e-05,
      "step": 65779,
      "training_step_time": 7.802398204803467
    },
    {
      "epoch": 6.273269653320313e-05,
      "grad_norm": 0.11651322990655899,
      "learning_rate": 2.8742458741396585e-06,
      "loss": 0.0565,
      "step": 65780
    },
    {
      "epoch": 6.273269653320313e-05,
      "model_forward_time": 1.7210328578948975,
      "step": 65780
    },
    {
      "dynamics_loss_avg": 0.04919427735731006,
      "epoch": 6.273269653320313e-05,
      "step": 65780
    },
    {
      "action_loss_avg": 0.0018660046422155573,
      "epoch": 6.273269653320313e-05,
      "step": 65780
    },
    {
      "epoch": 6.273269653320313e-05,
      "step": 65780,
      "training_step_time": 6.844470262527466
    },
    {
      "epoch": 6.273365020751954e-05,
      "model_forward_time": 1.7295465469360352,
      "step": 65781
    },
    {
      "epoch": 6.273365020751954e-05,
      "step": 65781,
      "training_step_time": 6.890454053878784
    },
    {
      "epoch": 6.273460388183593e-05,
      "model_forward_time": 1.726651906967163,
      "step": 65782
    },
    {
      "epoch": 6.273460388183593e-05,
      "step": 65782,
      "training_step_time": 6.86247181892395
    },
    {
      "epoch": 6.273555755615234e-05,
      "model_forward_time": 1.7392702102661133,
      "step": 65783
    },
    {
      "epoch": 6.273555755615234e-05,
      "step": 65783,
      "training_step_time": 6.832669258117676
    },
    {
      "epoch": 6.273651123046876e-05,
      "model_forward_time": 1.7319653034210205,
      "step": 65784
    },
    {
      "epoch": 6.273651123046876e-05,
      "step": 65784,
      "training_step_time": 6.801533460617065
    },
    {
      "epoch": 6.273746490478515e-05,
      "model_forward_time": 1.743865966796875,
      "step": 65785
    },
    {
      "epoch": 6.273746490478515e-05,
      "step": 65785,
      "training_step_time": 6.849496603012085
    },
    {
      "epoch": 6.273841857910156e-05,
      "model_forward_time": 1.736337661743164,
      "step": 65786
    },
    {
      "epoch": 6.273841857910156e-05,
      "step": 65786,
      "training_step_time": 7.679132699966431
    },
    {
      "epoch": 6.273937225341797e-05,
      "model_forward_time": 2.8491299152374268,
      "step": 65787
    },
    {
      "epoch": 6.273937225341797e-05,
      "step": 65787,
      "training_step_time": 9.054200649261475
    },
    {
      "epoch": 6.274032592773437e-05,
      "model_forward_time": 1.7465019226074219,
      "step": 65788
    },
    {
      "epoch": 6.274032592773437e-05,
      "step": 65788,
      "training_step_time": 8.798931121826172
    },
    {
      "epoch": 6.274127960205078e-05,
      "model_forward_time": 1.7265095710754395,
      "step": 65789
    },
    {
      "epoch": 6.274127960205078e-05,
      "step": 65789,
      "training_step_time": 6.858770370483398
    },
    {
      "epoch": 6.27422332763672e-05,
      "grad_norm": 0.09192339330911636,
      "learning_rate": 2.8727493976337737e-06,
      "loss": 0.0528,
      "step": 65790
    },
    {
      "epoch": 6.27422332763672e-05,
      "model_forward_time": 1.7158498764038086,
      "step": 65790
    },
    {
      "dynamics_loss_avg": 0.043746255524456504,
      "epoch": 6.27422332763672e-05,
      "step": 65790
    },
    {
      "action_loss_avg": 0.001149344301666133,
      "epoch": 6.27422332763672e-05,
      "step": 65790
    },
    {
      "epoch": 6.27422332763672e-05,
      "step": 65790,
      "training_step_time": 6.850621223449707
    },
    {
      "epoch": 6.274318695068359e-05,
      "model_forward_time": 1.7286884784698486,
      "step": 65791
    },
    {
      "epoch": 6.274318695068359e-05,
      "step": 65791,
      "training_step_time": 6.831778049468994
    },
    {
      "epoch": 6.2744140625e-05,
      "model_forward_time": 1.7355024814605713,
      "step": 65792
    },
    {
      "epoch": 6.2744140625e-05,
      "step": 65792,
      "training_step_time": 6.815365314483643
    },
    {
      "epoch": 6.274509429931641e-05,
      "model_forward_time": 1.7348034381866455,
      "step": 65793
    },
    {
      "epoch": 6.274509429931641e-05,
      "step": 65793,
      "training_step_time": 6.834073066711426
    },
    {
      "epoch": 6.274604797363281e-05,
      "model_forward_time": 1.7388553619384766,
      "step": 65794
    },
    {
      "epoch": 6.274604797363281e-05,
      "step": 65794,
      "training_step_time": 6.835824012756348
    },
    {
      "epoch": 6.274700164794922e-05,
      "model_forward_time": 1.7266371250152588,
      "step": 65795
    },
    {
      "epoch": 6.274700164794922e-05,
      "step": 65795,
      "training_step_time": 7.597189426422119
    },
    {
      "epoch": 6.274795532226563e-05,
      "model_forward_time": 2.8987908363342285,
      "step": 65796
    },
    {
      "epoch": 6.274795532226563e-05,
      "step": 65796,
      "training_step_time": 10.655456304550171
    },
    {
      "epoch": 6.274890899658203e-05,
      "model_forward_time": 1.7214460372924805,
      "step": 65797
    },
    {
      "epoch": 6.274890899658203e-05,
      "step": 65797,
      "training_step_time": 7.962947845458984
    },
    {
      "epoch": 6.274986267089844e-05,
      "model_forward_time": 1.7433953285217285,
      "step": 65798
    },
    {
      "epoch": 6.274986267089844e-05,
      "step": 65798,
      "training_step_time": 7.591960430145264
    },
    {
      "epoch": 6.275081634521485e-05,
      "model_forward_time": 1.737356424331665,
      "step": 65799
    },
    {
      "epoch": 6.275081634521485e-05,
      "step": 65799,
      "training_step_time": 6.812644004821777
    },
    {
      "epoch": 6.275177001953125e-05,
      "grad_norm": 0.11625562608242035,
      "learning_rate": 2.8712531537608217e-06,
      "loss": 0.0552,
      "step": 65800
    },
    {
      "epoch": 6.275177001953125e-05,
      "model_forward_time": 1.7238762378692627,
      "step": 65800
    },
    {
      "dynamics_loss_avg": 0.04953401070088148,
      "epoch": 6.275177001953125e-05,
      "step": 65800
    },
    {
      "action_loss_avg": 0.0019456466630799695,
      "epoch": 6.275177001953125e-05,
      "step": 65800
    },
    {
      "epoch": 6.275177001953125e-05,
      "step": 65800,
      "training_step_time": 6.827297925949097
    },
    {
      "epoch": 6.275272369384766e-05,
      "model_forward_time": 1.7260351181030273,
      "step": 65801
    },
    {
      "epoch": 6.275272369384766e-05,
      "step": 65801,
      "training_step_time": 6.8794426918029785
    },
    {
      "epoch": 6.275367736816407e-05,
      "model_forward_time": 1.7415716648101807,
      "step": 65802
    },
    {
      "epoch": 6.275367736816407e-05,
      "step": 65802,
      "training_step_time": 6.816052675247192
    },
    {
      "epoch": 6.275463104248047e-05,
      "model_forward_time": 1.7462399005889893,
      "step": 65803
    },
    {
      "epoch": 6.275463104248047e-05,
      "step": 65803,
      "training_step_time": 6.82511568069458
    },
    {
      "epoch": 6.275558471679688e-05,
      "model_forward_time": 1.7287251949310303,
      "step": 65804
    },
    {
      "epoch": 6.275558471679688e-05,
      "step": 65804,
      "training_step_time": 7.705904245376587
    },
    {
      "epoch": 6.275653839111329e-05,
      "model_forward_time": 2.878464460372925,
      "step": 65805
    },
    {
      "epoch": 6.275653839111329e-05,
      "step": 65805,
      "training_step_time": 11.604857683181763
    },
    {
      "epoch": 6.275749206542969e-05,
      "model_forward_time": 1.7342422008514404,
      "step": 65806
    },
    {
      "epoch": 6.275749206542969e-05,
      "step": 65806,
      "training_step_time": 8.447163820266724
    },
    {
      "epoch": 6.27584457397461e-05,
      "model_forward_time": 1.7830476760864258,
      "step": 65807
    },
    {
      "epoch": 6.27584457397461e-05,
      "step": 65807,
      "training_step_time": 7.509117603302002
    },
    {
      "epoch": 6.27593994140625e-05,
      "model_forward_time": 1.7341985702514648,
      "step": 65808
    },
    {
      "epoch": 6.27593994140625e-05,
      "step": 65808,
      "training_step_time": 6.804853439331055
    },
    {
      "epoch": 6.27603530883789e-05,
      "model_forward_time": 1.7388076782226562,
      "step": 65809
    },
    {
      "epoch": 6.27603530883789e-05,
      "step": 65809,
      "training_step_time": 6.79738450050354
    },
    {
      "epoch": 6.276130676269531e-05,
      "grad_norm": 0.08245823532342911,
      "learning_rate": 2.869757142684422e-06,
      "loss": 0.054,
      "step": 65810
    },
    {
      "epoch": 6.276130676269531e-05,
      "model_forward_time": 1.723761796951294,
      "step": 65810
    },
    {
      "dynamics_loss_avg": 0.0515044666826725,
      "epoch": 6.276130676269531e-05,
      "step": 65810
    },
    {
      "action_loss_avg": 0.0013928003405453636,
      "epoch": 6.276130676269531e-05,
      "step": 65810
    },
    {
      "epoch": 6.276130676269531e-05,
      "step": 65810,
      "training_step_time": 6.856012582778931
    },
    {
      "epoch": 6.276226043701173e-05,
      "model_forward_time": 1.7346539497375488,
      "step": 65811
    },
    {
      "epoch": 6.276226043701173e-05,
      "step": 65811,
      "training_step_time": 6.877932548522949
    },
    {
      "epoch": 6.276321411132812e-05,
      "model_forward_time": 1.7490062713623047,
      "step": 65812
    },
    {
      "epoch": 6.276321411132812e-05,
      "step": 65812,
      "training_step_time": 6.80132532119751
    },
    {
      "epoch": 6.276416778564453e-05,
      "model_forward_time": 1.3601477146148682,
      "step": 65813
    },
    {
      "epoch": 6.276416778564453e-05,
      "step": 65813,
      "training_step_time": 6.816741704940796
    },
    {
      "epoch": 6.276512145996093e-05,
      "model_forward_time": 2.841104030609131,
      "step": 65814
    },
    {
      "epoch": 6.276512145996093e-05,
      "step": 65814,
      "training_step_time": 9.922589778900146
    },
    {
      "epoch": 6.276607513427734e-05,
      "model_forward_time": 1.737029790878296,
      "step": 65815
    },
    {
      "epoch": 6.276607513427734e-05,
      "step": 65815,
      "training_step_time": 7.8694634437561035
    },
    {
      "epoch": 6.276702880859375e-05,
      "model_forward_time": 1.7347354888916016,
      "step": 65816
    },
    {
      "epoch": 6.276702880859375e-05,
      "step": 65816,
      "training_step_time": 7.925239562988281
    },
    {
      "epoch": 6.276798248291015e-05,
      "model_forward_time": 1.7215940952301025,
      "step": 65817
    },
    {
      "epoch": 6.276798248291015e-05,
      "step": 65817,
      "training_step_time": 7.518754005432129
    },
    {
      "epoch": 6.276893615722656e-05,
      "model_forward_time": 1.7395362854003906,
      "step": 65818
    },
    {
      "epoch": 6.276893615722656e-05,
      "step": 65818,
      "training_step_time": 6.877786636352539
    },
    {
      "epoch": 6.276988983154297e-05,
      "model_forward_time": 1.731635332107544,
      "step": 65819
    },
    {
      "epoch": 6.276988983154297e-05,
      "step": 65819,
      "training_step_time": 6.876598119735718
    },
    {
      "epoch": 6.277084350585937e-05,
      "grad_norm": 0.09876243770122528,
      "learning_rate": 2.86826136456818e-06,
      "loss": 0.0543,
      "step": 65820
    },
    {
      "epoch": 6.277084350585937e-05,
      "model_forward_time": 1.7206923961639404,
      "step": 65820
    },
    {
      "dynamics_loss_avg": 0.0621988607570529,
      "epoch": 6.277084350585937e-05,
      "step": 65820
    },
    {
      "action_loss_avg": 0.0018603204254759476,
      "epoch": 6.277084350585937e-05,
      "step": 65820
    },
    {
      "epoch": 6.277084350585937e-05,
      "step": 65820,
      "training_step_time": 6.812986135482788
    },
    {
      "epoch": 6.277179718017578e-05,
      "model_forward_time": 1.7284612655639648,
      "step": 65821
    },
    {
      "epoch": 6.277179718017578e-05,
      "step": 65821,
      "training_step_time": 6.844148874282837
    },
    {
      "epoch": 6.277275085449219e-05,
      "model_forward_time": 1.3584098815917969,
      "step": 65822
    },
    {
      "epoch": 6.277275085449219e-05,
      "step": 65822,
      "training_step_time": 6.843026399612427
    },
    {
      "epoch": 6.277370452880859e-05,
      "model_forward_time": 1.7252438068389893,
      "step": 65823
    },
    {
      "epoch": 6.277370452880859e-05,
      "step": 65823,
      "training_step_time": 9.451415061950684
    },
    {
      "epoch": 6.2774658203125e-05,
      "model_forward_time": 2.882734775543213,
      "step": 65824
    },
    {
      "epoch": 6.2774658203125e-05,
      "step": 65824,
      "training_step_time": 9.623522996902466
    },
    {
      "epoch": 6.277561187744141e-05,
      "model_forward_time": 1.7329683303833008,
      "step": 65825
    },
    {
      "epoch": 6.277561187744141e-05,
      "step": 65825,
      "training_step_time": 7.918482780456543
    },
    {
      "epoch": 6.27765655517578e-05,
      "model_forward_time": 1.7294292449951172,
      "step": 65826
    },
    {
      "epoch": 6.27765655517578e-05,
      "step": 65826,
      "training_step_time": 7.629605054855347
    },
    {
      "epoch": 6.277751922607422e-05,
      "model_forward_time": 1.7295753955841064,
      "step": 65827
    },
    {
      "epoch": 6.277751922607422e-05,
      "step": 65827,
      "training_step_time": 6.910925626754761
    },
    {
      "epoch": 6.277847290039063e-05,
      "model_forward_time": 1.7363386154174805,
      "step": 65828
    },
    {
      "epoch": 6.277847290039063e-05,
      "step": 65828,
      "training_step_time": 6.902828693389893
    },
    {
      "epoch": 6.277942657470703e-05,
      "model_forward_time": 1.7344281673431396,
      "step": 65829
    },
    {
      "epoch": 6.277942657470703e-05,
      "step": 65829,
      "training_step_time": 6.831804990768433
    },
    {
      "epoch": 6.278038024902344e-05,
      "grad_norm": 0.13129276037216187,
      "learning_rate": 2.8667658195756686e-06,
      "loss": 0.0553,
      "step": 65830
    },
    {
      "epoch": 6.278038024902344e-05,
      "model_forward_time": 1.7220726013183594,
      "step": 65830
    },
    {
      "dynamics_loss_avg": 0.05744073102250695,
      "epoch": 6.278038024902344e-05,
      "step": 65830
    },
    {
      "action_loss_avg": 0.003524574846960604,
      "epoch": 6.278038024902344e-05,
      "step": 65830
    },
    {
      "epoch": 6.278038024902344e-05,
      "step": 65830,
      "training_step_time": 6.861083984375
    },
    {
      "epoch": 6.278133392333985e-05,
      "model_forward_time": 1.7305541038513184,
      "step": 65831
    },
    {
      "epoch": 6.278133392333985e-05,
      "step": 65831,
      "training_step_time": 6.826495409011841
    },
    {
      "epoch": 6.278228759765624e-05,
      "model_forward_time": 1.7271251678466797,
      "step": 65832
    },
    {
      "epoch": 6.278228759765624e-05,
      "step": 65832,
      "training_step_time": 8.408658266067505
    },
    {
      "epoch": 6.278324127197265e-05,
      "model_forward_time": 2.89678955078125,
      "step": 65833
    },
    {
      "epoch": 6.278324127197265e-05,
      "step": 65833,
      "training_step_time": 8.834799766540527
    },
    {
      "epoch": 6.278419494628907e-05,
      "model_forward_time": 1.7237379550933838,
      "step": 65834
    },
    {
      "epoch": 6.278419494628907e-05,
      "step": 65834,
      "training_step_time": 7.916969537734985
    },
    {
      "epoch": 6.278514862060546e-05,
      "model_forward_time": 1.7282416820526123,
      "step": 65835
    },
    {
      "epoch": 6.278514862060546e-05,
      "step": 65835,
      "training_step_time": 7.629542827606201
    },
    {
      "epoch": 6.278610229492187e-05,
      "model_forward_time": 1.7311182022094727,
      "step": 65836
    },
    {
      "epoch": 6.278610229492187e-05,
      "step": 65836,
      "training_step_time": 6.810448884963989
    },
    {
      "epoch": 6.278705596923828e-05,
      "model_forward_time": 1.761444330215454,
      "step": 65837
    },
    {
      "epoch": 6.278705596923828e-05,
      "step": 65837,
      "training_step_time": 6.816730976104736
    },
    {
      "epoch": 6.278800964355468e-05,
      "model_forward_time": 1.7292473316192627,
      "step": 65838
    },
    {
      "epoch": 6.278800964355468e-05,
      "step": 65838,
      "training_step_time": 6.830351114273071
    },
    {
      "epoch": 6.278896331787109e-05,
      "model_forward_time": 1.7467553615570068,
      "step": 65839
    },
    {
      "epoch": 6.278896331787109e-05,
      "step": 65839,
      "training_step_time": 6.815905332565308
    },
    {
      "epoch": 6.27899169921875e-05,
      "grad_norm": 0.08812975138425827,
      "learning_rate": 2.8652705078704413e-06,
      "loss": 0.0535,
      "step": 65840
    },
    {
      "epoch": 6.27899169921875e-05,
      "model_forward_time": 1.7528252601623535,
      "step": 65840
    },
    {
      "dynamics_loss_avg": 0.0516038266941905,
      "epoch": 6.27899169921875e-05,
      "step": 65840
    },
    {
      "action_loss_avg": 0.001520533612347208,
      "epoch": 6.27899169921875e-05,
      "step": 65840
    },
    {
      "epoch": 6.27899169921875e-05,
      "step": 65840,
      "training_step_time": 6.796126365661621
    },
    {
      "epoch": 6.27908706665039e-05,
      "model_forward_time": 1.7493863105773926,
      "step": 65841
    },
    {
      "epoch": 6.27908706665039e-05,
      "step": 65841,
      "training_step_time": 7.574468612670898
    },
    {
      "epoch": 6.279182434082031e-05,
      "model_forward_time": 2.341930866241455,
      "step": 65842
    },
    {
      "epoch": 6.279182434082031e-05,
      "step": 65842,
      "training_step_time": 9.500424861907959
    },
    {
      "epoch": 6.279277801513672e-05,
      "model_forward_time": 1.74721097946167,
      "step": 65843
    },
    {
      "epoch": 6.279277801513672e-05,
      "step": 65843,
      "training_step_time": 7.8326826095581055
    },
    {
      "epoch": 6.279373168945312e-05,
      "model_forward_time": 1.754955768585205,
      "step": 65844
    },
    {
      "epoch": 6.279373168945312e-05,
      "step": 65844,
      "training_step_time": 7.822800159454346
    },
    {
      "epoch": 6.279468536376953e-05,
      "model_forward_time": 1.7545166015625,
      "step": 65845
    },
    {
      "epoch": 6.279468536376953e-05,
      "step": 65845,
      "training_step_time": 7.487020254135132
    },
    {
      "epoch": 6.279563903808594e-05,
      "model_forward_time": 1.7482335567474365,
      "step": 65846
    },
    {
      "epoch": 6.279563903808594e-05,
      "step": 65846,
      "training_step_time": 6.821319341659546
    },
    {
      "epoch": 6.279659271240234e-05,
      "model_forward_time": 1.7561910152435303,
      "step": 65847
    },
    {
      "epoch": 6.279659271240234e-05,
      "step": 65847,
      "training_step_time": 6.852221488952637
    },
    {
      "epoch": 6.279754638671875e-05,
      "model_forward_time": 1.7543771266937256,
      "step": 65848
    },
    {
      "epoch": 6.279754638671875e-05,
      "step": 65848,
      "training_step_time": 6.834804534912109
    },
    {
      "epoch": 6.279850006103516e-05,
      "model_forward_time": 1.760237455368042,
      "step": 65849
    },
    {
      "epoch": 6.279850006103516e-05,
      "step": 65849,
      "training_step_time": 6.913439035415649
    },
    {
      "epoch": 6.279945373535156e-05,
      "grad_norm": 0.08458996564149857,
      "learning_rate": 2.863775429616018e-06,
      "loss": 0.0565,
      "step": 65850
    },
    {
      "epoch": 6.279945373535156e-05,
      "model_forward_time": 1.7489538192749023,
      "step": 65850
    },
    {
      "dynamics_loss_avg": 0.045026546251028775,
      "epoch": 6.279945373535156e-05,
      "step": 65850
    },
    {
      "action_loss_avg": 0.001370379216677975,
      "epoch": 6.279945373535156e-05,
      "step": 65850
    },
    {
      "epoch": 6.279945373535156e-05,
      "step": 65850,
      "training_step_time": 7.548851013183594
    },
    {
      "epoch": 6.280040740966797e-05,
      "model_forward_time": 2.337125301361084,
      "step": 65851
    },
    {
      "epoch": 6.280040740966797e-05,
      "step": 65851,
      "training_step_time": 9.225884437561035
    },
    {
      "epoch": 6.280136108398438e-05,
      "model_forward_time": 1.7556581497192383,
      "step": 65852
    },
    {
      "epoch": 6.280136108398438e-05,
      "step": 65852,
      "training_step_time": 7.829455614089966
    },
    {
      "epoch": 6.280231475830078e-05,
      "model_forward_time": 1.753732442855835,
      "step": 65853
    },
    {
      "epoch": 6.280231475830078e-05,
      "step": 65853,
      "training_step_time": 7.866665363311768
    },
    {
      "epoch": 6.280326843261719e-05,
      "model_forward_time": 1.7567682266235352,
      "step": 65854
    },
    {
      "epoch": 6.280326843261719e-05,
      "step": 65854,
      "training_step_time": 7.553079605102539
    },
    {
      "epoch": 6.28042221069336e-05,
      "model_forward_time": 1.75968337059021,
      "step": 65855
    },
    {
      "epoch": 6.28042221069336e-05,
      "step": 65855,
      "training_step_time": 6.862738370895386
    },
    {
      "epoch": 6.280517578125e-05,
      "model_forward_time": 1.7572951316833496,
      "step": 65856
    },
    {
      "epoch": 6.280517578125e-05,
      "step": 65856,
      "training_step_time": 6.87396502494812
    },
    {
      "epoch": 6.28061294555664e-05,
      "model_forward_time": 1.7574563026428223,
      "step": 65857
    },
    {
      "epoch": 6.28061294555664e-05,
      "step": 65857,
      "training_step_time": 6.884806156158447
    },
    {
      "epoch": 6.280708312988282e-05,
      "model_forward_time": 1.7388477325439453,
      "step": 65858
    },
    {
      "epoch": 6.280708312988282e-05,
      "step": 65858,
      "training_step_time": 6.943811416625977
    },
    {
      "epoch": 6.280803680419921e-05,
      "model_forward_time": 1.7397058010101318,
      "step": 65859
    },
    {
      "epoch": 6.280803680419921e-05,
      "step": 65859,
      "training_step_time": 7.617512464523315
    },
    {
      "epoch": 6.280899047851562e-05,
      "grad_norm": 0.08149707317352295,
      "learning_rate": 2.862280584975906e-06,
      "loss": 0.0536,
      "step": 65860
    },
    {
      "epoch": 6.280899047851562e-05,
      "model_forward_time": 1.7409107685089111,
      "step": 65860
    },
    {
      "dynamics_loss_avg": 0.040675174444913864,
      "epoch": 6.280899047851562e-05,
      "step": 65860
    },
    {
      "action_loss_avg": 0.0013025447784457355,
      "epoch": 6.280899047851562e-05,
      "step": 65860
    },
    {
      "epoch": 6.280899047851562e-05,
      "step": 65860,
      "training_step_time": 10.125172853469849
    },
    {
      "epoch": 6.280994415283204e-05,
      "model_forward_time": 1.7482452392578125,
      "step": 65861
    },
    {
      "epoch": 6.280994415283204e-05,
      "step": 65861,
      "training_step_time": 10.477336406707764
    },
    {
      "epoch": 6.281089782714843e-05,
      "model_forward_time": 1.7428793907165527,
      "step": 65862
    },
    {
      "epoch": 6.281089782714843e-05,
      "step": 65862,
      "training_step_time": 7.893832206726074
    },
    {
      "epoch": 6.281185150146484e-05,
      "model_forward_time": 1.7528352737426758,
      "step": 65863
    },
    {
      "epoch": 6.281185150146484e-05,
      "step": 65863,
      "training_step_time": 7.114502191543579
    },
    {
      "epoch": 6.281280517578125e-05,
      "model_forward_time": 1.7458765506744385,
      "step": 65864
    },
    {
      "epoch": 6.281280517578125e-05,
      "step": 65864,
      "training_step_time": 6.893414258956909
    },
    {
      "epoch": 6.281375885009765e-05,
      "model_forward_time": 1.7462658882141113,
      "step": 65865
    },
    {
      "epoch": 6.281375885009765e-05,
      "step": 65865,
      "training_step_time": 6.921065330505371
    },
    {
      "epoch": 6.281471252441406e-05,
      "model_forward_time": 1.7743470668792725,
      "step": 65866
    },
    {
      "epoch": 6.281471252441406e-05,
      "step": 65866,
      "training_step_time": 6.922635316848755
    },
    {
      "epoch": 6.281566619873047e-05,
      "model_forward_time": 1.7364988327026367,
      "step": 65867
    },
    {
      "epoch": 6.281566619873047e-05,
      "step": 65867,
      "training_step_time": 6.9005279541015625
    },
    {
      "epoch": 6.281661987304687e-05,
      "model_forward_time": 1.7392833232879639,
      "step": 65868
    },
    {
      "epoch": 6.281661987304687e-05,
      "step": 65868,
      "training_step_time": 8.472350120544434
    },
    {
      "epoch": 6.281757354736328e-05,
      "model_forward_time": 1.7410011291503906,
      "step": 65869
    },
    {
      "epoch": 6.281757354736328e-05,
      "step": 65869,
      "training_step_time": 9.520232200622559
    },
    {
      "epoch": 6.281852722167969e-05,
      "grad_norm": 0.07500189542770386,
      "learning_rate": 2.8607859741135735e-06,
      "loss": 0.055,
      "step": 65870
    },
    {
      "epoch": 6.281852722167969e-05,
      "model_forward_time": 2.975944995880127,
      "step": 65870
    },
    {
      "dynamics_loss_avg": 0.04476289451122284,
      "epoch": 6.281852722167969e-05,
      "step": 65870
    },
    {
      "action_loss_avg": 0.0010698951198719441,
      "epoch": 6.281852722167969e-05,
      "step": 65870
    },
    {
      "epoch": 6.281852722167969e-05,
      "step": 65870,
      "training_step_time": 8.965630054473877
    },
    {
      "epoch": 6.281948089599609e-05,
      "model_forward_time": 1.7365283966064453,
      "step": 65871
    },
    {
      "epoch": 6.281948089599609e-05,
      "step": 65871,
      "training_step_time": 7.882450103759766
    },
    {
      "epoch": 6.28204345703125e-05,
      "model_forward_time": 1.7378661632537842,
      "step": 65872
    },
    {
      "epoch": 6.28204345703125e-05,
      "step": 65872,
      "training_step_time": 7.816235542297363
    },
    {
      "epoch": 6.282138824462891e-05,
      "model_forward_time": 1.7425944805145264,
      "step": 65873
    },
    {
      "epoch": 6.282138824462891e-05,
      "step": 65873,
      "training_step_time": 7.568955421447754
    },
    {
      "epoch": 6.282234191894531e-05,
      "model_forward_time": 1.728405475616455,
      "step": 65874
    },
    {
      "epoch": 6.282234191894531e-05,
      "step": 65874,
      "training_step_time": 6.873898983001709
    },
    {
      "epoch": 6.282329559326172e-05,
      "model_forward_time": 1.7260358333587646,
      "step": 65875
    },
    {
      "epoch": 6.282329559326172e-05,
      "step": 65875,
      "training_step_time": 6.885248184204102
    },
    {
      "epoch": 6.282424926757813e-05,
      "model_forward_time": 1.7304012775421143,
      "step": 65876
    },
    {
      "epoch": 6.282424926757813e-05,
      "step": 65876,
      "training_step_time": 6.879588842391968
    },
    {
      "epoch": 6.282520294189453e-05,
      "model_forward_time": 1.7309496402740479,
      "step": 65877
    },
    {
      "epoch": 6.282520294189453e-05,
      "step": 65877,
      "training_step_time": 8.330559492111206
    },
    {
      "epoch": 6.282615661621094e-05,
      "model_forward_time": 1.7317872047424316,
      "step": 65878
    },
    {
      "epoch": 6.282615661621094e-05,
      "step": 65878,
      "training_step_time": 7.844365119934082
    },
    {
      "epoch": 6.282711029052735e-05,
      "model_forward_time": 2.8913941383361816,
      "step": 65879
    },
    {
      "epoch": 6.282711029052735e-05,
      "step": 65879,
      "training_step_time": 8.966060876846313
    },
    {
      "epoch": 6.282806396484375e-05,
      "grad_norm": 0.08804300427436829,
      "learning_rate": 2.8592915971924724e-06,
      "loss": 0.0554,
      "step": 65880
    },
    {
      "epoch": 6.282806396484375e-05,
      "model_forward_time": 1.723362684249878,
      "step": 65880
    },
    {
      "dynamics_loss_avg": 0.054904050566256045,
      "epoch": 6.282806396484375e-05,
      "step": 65880
    },
    {
      "action_loss_avg": 0.0019358046993147582,
      "epoch": 6.282806396484375e-05,
      "step": 65880
    },
    {
      "epoch": 6.282806396484375e-05,
      "step": 65880,
      "training_step_time": 7.924809694290161
    },
    {
      "epoch": 6.282901763916016e-05,
      "model_forward_time": 1.735525131225586,
      "step": 65881
    },
    {
      "epoch": 6.282901763916016e-05,
      "step": 65881,
      "training_step_time": 7.752244234085083
    },
    {
      "epoch": 6.282997131347657e-05,
      "model_forward_time": 1.7298533916473389,
      "step": 65882
    },
    {
      "epoch": 6.282997131347657e-05,
      "step": 65882,
      "training_step_time": 7.607717752456665
    },
    {
      "epoch": 6.283092498779296e-05,
      "model_forward_time": 1.7332215309143066,
      "step": 65883
    },
    {
      "epoch": 6.283092498779296e-05,
      "step": 65883,
      "training_step_time": 6.835453987121582
    },
    {
      "epoch": 6.283187866210938e-05,
      "model_forward_time": 1.734206199645996,
      "step": 65884
    },
    {
      "epoch": 6.283187866210938e-05,
      "step": 65884,
      "training_step_time": 6.83602237701416
    },
    {
      "epoch": 6.283283233642579e-05,
      "model_forward_time": 1.7307910919189453,
      "step": 65885
    },
    {
      "epoch": 6.283283233642579e-05,
      "step": 65885,
      "training_step_time": 6.840412616729736
    },
    {
      "epoch": 6.283378601074218e-05,
      "model_forward_time": 1.736555814743042,
      "step": 65886
    },
    {
      "epoch": 6.283378601074218e-05,
      "step": 65886,
      "training_step_time": 8.665906429290771
    },
    {
      "epoch": 6.28347396850586e-05,
      "model_forward_time": 1.7463679313659668,
      "step": 65887
    },
    {
      "epoch": 6.28347396850586e-05,
      "step": 65887,
      "training_step_time": 8.570179462432861
    },
    {
      "epoch": 6.2835693359375e-05,
      "model_forward_time": 2.488790273666382,
      "step": 65888
    },
    {
      "epoch": 6.2835693359375e-05,
      "step": 65888,
      "training_step_time": 9.924370527267456
    },
    {
      "epoch": 6.28366470336914e-05,
      "model_forward_time": 1.732790231704712,
      "step": 65889
    },
    {
      "epoch": 6.28366470336914e-05,
      "step": 65889,
      "training_step_time": 8.541260719299316
    },
    {
      "epoch": 6.283760070800781e-05,
      "grad_norm": 0.07978099584579468,
      "learning_rate": 2.857797454376021e-06,
      "loss": 0.0575,
      "step": 65890
    },
    {
      "epoch": 6.283760070800781e-05,
      "model_forward_time": 1.7347259521484375,
      "step": 65890
    },
    {
      "dynamics_loss_avg": 0.06361996438354253,
      "epoch": 6.283760070800781e-05,
      "step": 65890
    },
    {
      "action_loss_avg": 0.001836198742967099,
      "epoch": 6.283760070800781e-05,
      "step": 65890
    },
    {
      "epoch": 6.283760070800781e-05,
      "step": 65890,
      "training_step_time": 7.794434547424316
    },
    {
      "epoch": 6.283855438232422e-05,
      "model_forward_time": 1.74644136428833,
      "step": 65891
    },
    {
      "epoch": 6.283855438232422e-05,
      "step": 65891,
      "training_step_time": 7.531615495681763
    },
    {
      "epoch": 6.283950805664062e-05,
      "model_forward_time": 1.7389957904815674,
      "step": 65892
    },
    {
      "epoch": 6.283950805664062e-05,
      "step": 65892,
      "training_step_time": 6.919266939163208
    },
    {
      "epoch": 6.284046173095703e-05,
      "model_forward_time": 1.7456834316253662,
      "step": 65893
    },
    {
      "epoch": 6.284046173095703e-05,
      "step": 65893,
      "training_step_time": 6.8607728481292725
    },
    {
      "epoch": 6.284141540527344e-05,
      "model_forward_time": 1.7396647930145264,
      "step": 65894
    },
    {
      "epoch": 6.284141540527344e-05,
      "step": 65894,
      "training_step_time": 6.877118825912476
    },
    {
      "epoch": 6.284236907958984e-05,
      "model_forward_time": 1.724285364151001,
      "step": 65895
    },
    {
      "epoch": 6.284236907958984e-05,
      "step": 65895,
      "training_step_time": 7.6318182945251465
    },
    {
      "epoch": 6.284332275390625e-05,
      "model_forward_time": 1.7416472434997559,
      "step": 65896
    },
    {
      "epoch": 6.284332275390625e-05,
      "step": 65896,
      "training_step_time": 7.839719533920288
    },
    {
      "epoch": 6.284427642822266e-05,
      "model_forward_time": 2.913241147994995,
      "step": 65897
    },
    {
      "epoch": 6.284427642822266e-05,
      "step": 65897,
      "training_step_time": 10.487821817398071
    },
    {
      "epoch": 6.284523010253906e-05,
      "model_forward_time": 1.7359685897827148,
      "step": 65898
    },
    {
      "epoch": 6.284523010253906e-05,
      "step": 65898,
      "training_step_time": 7.917423725128174
    },
    {
      "epoch": 6.284618377685547e-05,
      "model_forward_time": 1.7421371936798096,
      "step": 65899
    },
    {
      "epoch": 6.284618377685547e-05,
      "step": 65899,
      "training_step_time": 7.763598203659058
    },
    {
      "epoch": 6.284713745117188e-05,
      "grad_norm": 0.08977577090263367,
      "learning_rate": 2.8563035458276225e-06,
      "loss": 0.0586,
      "step": 65900
    },
    {
      "epoch": 6.284713745117188e-05,
      "model_forward_time": 1.724073886871338,
      "step": 65900
    },
    {
      "dynamics_loss_avg": 0.04654734283685684,
      "epoch": 6.284713745117188e-05,
      "step": 65900
    },
    {
      "action_loss_avg": 0.0019781107344897463,
      "epoch": 6.284713745117188e-05,
      "step": 65900
    },
    {
      "epoch": 6.284713745117188e-05,
      "step": 65900,
      "training_step_time": 7.386396646499634
    },
    {
      "epoch": 6.284809112548828e-05,
      "model_forward_time": 1.743011236190796,
      "step": 65901
    },
    {
      "epoch": 6.284809112548828e-05,
      "step": 65901,
      "training_step_time": 6.814226388931274
    },
    {
      "epoch": 6.284904479980469e-05,
      "model_forward_time": 1.737882137298584,
      "step": 65902
    },
    {
      "epoch": 6.284904479980469e-05,
      "step": 65902,
      "training_step_time": 6.795424938201904
    },
    {
      "epoch": 6.28499984741211e-05,
      "model_forward_time": 1.773371696472168,
      "step": 65903
    },
    {
      "epoch": 6.28499984741211e-05,
      "step": 65903,
      "training_step_time": 6.785874128341675
    },
    {
      "epoch": 6.28509521484375e-05,
      "model_forward_time": 1.746509313583374,
      "step": 65904
    },
    {
      "epoch": 6.28509521484375e-05,
      "step": 65904,
      "training_step_time": 7.556204795837402
    },
    {
      "epoch": 6.285190582275391e-05,
      "model_forward_time": 1.7456073760986328,
      "step": 65905
    },
    {
      "epoch": 6.285190582275391e-05,
      "step": 65905,
      "training_step_time": 7.753346681594849
    },
    {
      "epoch": 6.285285949707032e-05,
      "model_forward_time": 1.7557682991027832,
      "step": 65906
    },
    {
      "epoch": 6.285285949707032e-05,
      "step": 65906,
      "training_step_time": 8.428114652633667
    },
    {
      "epoch": 6.285381317138672e-05,
      "model_forward_time": 2.9303977489471436,
      "step": 65907
    },
    {
      "epoch": 6.285381317138672e-05,
      "step": 65907,
      "training_step_time": 10.60344123840332
    },
    {
      "epoch": 6.285476684570313e-05,
      "model_forward_time": 1.7411549091339111,
      "step": 65908
    },
    {
      "epoch": 6.285476684570313e-05,
      "step": 65908,
      "training_step_time": 7.731204509735107
    },
    {
      "epoch": 6.285572052001954e-05,
      "model_forward_time": 1.742706537246704,
      "step": 65909
    },
    {
      "epoch": 6.285572052001954e-05,
      "step": 65909,
      "training_step_time": 7.64044189453125
    },
    {
      "epoch": 6.285667419433593e-05,
      "grad_norm": 0.10481268167495728,
      "learning_rate": 2.8548098717106425e-06,
      "loss": 0.0544,
      "step": 65910
    },
    {
      "epoch": 6.285667419433593e-05,
      "model_forward_time": 1.7351112365722656,
      "step": 65910
    },
    {
      "dynamics_loss_avg": 0.07158815544098615,
      "epoch": 6.285667419433593e-05,
      "step": 65910
    },
    {
      "action_loss_avg": 0.002759600314311683,
      "epoch": 6.285667419433593e-05,
      "step": 65910
    },
    {
      "epoch": 6.285667419433593e-05,
      "step": 65910,
      "training_step_time": 7.552657127380371
    },
    {
      "epoch": 6.285762786865235e-05,
      "model_forward_time": 1.7375640869140625,
      "step": 65911
    },
    {
      "epoch": 6.285762786865235e-05,
      "step": 65911,
      "training_step_time": 6.841375112533569
    },
    {
      "epoch": 6.285858154296876e-05,
      "model_forward_time": 1.7500042915344238,
      "step": 65912
    },
    {
      "epoch": 6.285858154296876e-05,
      "step": 65912,
      "training_step_time": 6.851140975952148
    },
    {
      "epoch": 6.285953521728515e-05,
      "model_forward_time": 1.7455449104309082,
      "step": 65913
    },
    {
      "epoch": 6.285953521728515e-05,
      "step": 65913,
      "training_step_time": 7.612903118133545
    },
    {
      "epoch": 6.286048889160156e-05,
      "model_forward_time": 1.7374827861785889,
      "step": 65914
    },
    {
      "epoch": 6.286048889160156e-05,
      "step": 65914,
      "training_step_time": 7.524512052536011
    },
    {
      "epoch": 6.286144256591797e-05,
      "model_forward_time": 1.7350661754608154,
      "step": 65915
    },
    {
      "epoch": 6.286144256591797e-05,
      "step": 65915,
      "training_step_time": 8.792182207107544
    },
    {
      "epoch": 6.286239624023437e-05,
      "model_forward_time": 2.8678040504455566,
      "step": 65916
    },
    {
      "epoch": 6.286239624023437e-05,
      "step": 65916,
      "training_step_time": 8.931366920471191
    },
    {
      "epoch": 6.286334991455078e-05,
      "model_forward_time": 1.7313263416290283,
      "step": 65917
    },
    {
      "epoch": 6.286334991455078e-05,
      "step": 65917,
      "training_step_time": 7.699981689453125
    },
    {
      "epoch": 6.28643035888672e-05,
      "model_forward_time": 1.7418057918548584,
      "step": 65918
    },
    {
      "epoch": 6.28643035888672e-05,
      "step": 65918,
      "training_step_time": 7.68160080909729
    },
    {
      "epoch": 6.286525726318359e-05,
      "model_forward_time": 1.363102912902832,
      "step": 65919
    },
    {
      "epoch": 6.286525726318359e-05,
      "step": 65919,
      "training_step_time": 7.5601794719696045
    },
    {
      "epoch": 6.28662109375e-05,
      "grad_norm": 0.11834291368722916,
      "learning_rate": 2.85331643218843e-06,
      "loss": 0.0555,
      "step": 65920
    },
    {
      "epoch": 6.28662109375e-05,
      "model_forward_time": 1.729419231414795,
      "step": 65920
    },
    {
      "dynamics_loss_avg": 0.060666232276707886,
      "epoch": 6.28662109375e-05,
      "step": 65920
    },
    {
      "action_loss_avg": 0.0027054236037656666,
      "epoch": 6.28662109375e-05,
      "step": 65920
    },
    {
      "epoch": 6.28662109375e-05,
      "step": 65920,
      "training_step_time": 6.792064428329468
    },
    {
      "epoch": 6.286716461181641e-05,
      "model_forward_time": 1.7708370685577393,
      "step": 65921
    },
    {
      "epoch": 6.286716461181641e-05,
      "step": 65921,
      "training_step_time": 6.840180397033691
    },
    {
      "epoch": 6.286811828613281e-05,
      "model_forward_time": 1.7404253482818604,
      "step": 65922
    },
    {
      "epoch": 6.286811828613281e-05,
      "step": 65922,
      "training_step_time": 7.499996185302734
    },
    {
      "epoch": 6.286907196044922e-05,
      "model_forward_time": 1.7135796546936035,
      "step": 65923
    },
    {
      "epoch": 6.286907196044922e-05,
      "step": 65923,
      "training_step_time": 7.801663875579834
    },
    {
      "epoch": 6.287002563476563e-05,
      "model_forward_time": 1.7278246879577637,
      "step": 65924
    },
    {
      "epoch": 6.287002563476563e-05,
      "step": 65924,
      "training_step_time": 8.29195261001587
    },
    {
      "epoch": 6.287097930908203e-05,
      "model_forward_time": 2.9213340282440186,
      "step": 65925
    },
    {
      "epoch": 6.287097930908203e-05,
      "step": 65925,
      "training_step_time": 10.065284729003906
    },
    {
      "epoch": 6.287193298339844e-05,
      "model_forward_time": 1.7364165782928467,
      "step": 65926
    },
    {
      "epoch": 6.287193298339844e-05,
      "step": 65926,
      "training_step_time": 7.953555345535278
    },
    {
      "epoch": 6.287288665771485e-05,
      "model_forward_time": 1.7409160137176514,
      "step": 65927
    },
    {
      "epoch": 6.287288665771485e-05,
      "step": 65927,
      "training_step_time": 7.788438558578491
    },
    {
      "epoch": 6.287384033203125e-05,
      "model_forward_time": 1.7365076541900635,
      "step": 65928
    },
    {
      "epoch": 6.287384033203125e-05,
      "step": 65928,
      "training_step_time": 7.534006357192993
    },
    {
      "epoch": 6.287479400634766e-05,
      "model_forward_time": 1.7400507926940918,
      "step": 65929
    },
    {
      "epoch": 6.287479400634766e-05,
      "step": 65929,
      "training_step_time": 6.853248357772827
    },
    {
      "epoch": 6.287574768066407e-05,
      "grad_norm": 0.08352126181125641,
      "learning_rate": 2.8518232274243047e-06,
      "loss": 0.0572,
      "step": 65930
    },
    {
      "epoch": 6.287574768066407e-05,
      "model_forward_time": 1.755275011062622,
      "step": 65930
    },
    {
      "dynamics_loss_avg": 0.03813781756907701,
      "epoch": 6.287574768066407e-05,
      "step": 65930
    },
    {
      "action_loss_avg": 0.0010856205743039026,
      "epoch": 6.287574768066407e-05,
      "step": 65930
    },
    {
      "epoch": 6.287574768066407e-05,
      "step": 65930,
      "training_step_time": 6.852917909622192
    },
    {
      "epoch": 6.287670135498047e-05,
      "model_forward_time": 1.7244784832000732,
      "step": 65931
    },
    {
      "epoch": 6.287670135498047e-05,
      "step": 65931,
      "training_step_time": 7.460363388061523
    },
    {
      "epoch": 6.287765502929688e-05,
      "model_forward_time": 1.7370951175689697,
      "step": 65932
    },
    {
      "epoch": 6.287765502929688e-05,
      "step": 65932,
      "training_step_time": 7.726416110992432
    },
    {
      "epoch": 6.287860870361329e-05,
      "model_forward_time": 1.7319612503051758,
      "step": 65933
    },
    {
      "epoch": 6.287860870361329e-05,
      "step": 65933,
      "training_step_time": 8.869465351104736
    },
    {
      "epoch": 6.287956237792969e-05,
      "model_forward_time": 2.948763847351074,
      "step": 65934
    },
    {
      "epoch": 6.287956237792969e-05,
      "step": 65934,
      "training_step_time": 8.816616535186768
    },
    {
      "epoch": 6.28805160522461e-05,
      "model_forward_time": 1.7363831996917725,
      "step": 65935
    },
    {
      "epoch": 6.28805160522461e-05,
      "step": 65935,
      "training_step_time": 8.477329730987549
    },
    {
      "epoch": 6.28814697265625e-05,
      "model_forward_time": 1.7306218147277832,
      "step": 65936
    },
    {
      "epoch": 6.28814697265625e-05,
      "step": 65936,
      "training_step_time": 7.74893593788147
    },
    {
      "epoch": 6.28824234008789e-05,
      "model_forward_time": 1.7336130142211914,
      "step": 65937
    },
    {
      "epoch": 6.28824234008789e-05,
      "step": 65937,
      "training_step_time": 7.654617547988892
    },
    {
      "epoch": 6.288337707519531e-05,
      "model_forward_time": 1.743774175643921,
      "step": 65938
    },
    {
      "epoch": 6.288337707519531e-05,
      "step": 65938,
      "training_step_time": 6.830103397369385
    },
    {
      "epoch": 6.288433074951171e-05,
      "model_forward_time": 1.7430479526519775,
      "step": 65939
    },
    {
      "epoch": 6.288433074951171e-05,
      "step": 65939,
      "training_step_time": 6.772482633590698
    },
    {
      "epoch": 6.288528442382812e-05,
      "grad_norm": 0.08768629282712936,
      "learning_rate": 2.850330257581564e-06,
      "loss": 0.0597,
      "step": 65940
    },
    {
      "epoch": 6.288528442382812e-05,
      "model_forward_time": 1.7325115203857422,
      "step": 65940
    },
    {
      "dynamics_loss_avg": 0.061280657351016996,
      "epoch": 6.288528442382812e-05,
      "step": 65940
    },
    {
      "action_loss_avg": 0.0013503532769391312,
      "epoch": 6.288528442382812e-05,
      "step": 65940
    },
    {
      "epoch": 6.288528442382812e-05,
      "step": 65940,
      "training_step_time": 7.515088081359863
    },
    {
      "epoch": 6.288623809814453e-05,
      "model_forward_time": 1.7313261032104492,
      "step": 65941
    },
    {
      "epoch": 6.288623809814453e-05,
      "step": 65941,
      "training_step_time": 8.190628051757812
    },
    {
      "epoch": 6.288719177246093e-05,
      "model_forward_time": 1.7447845935821533,
      "step": 65942
    },
    {
      "epoch": 6.288719177246093e-05,
      "step": 65942,
      "training_step_time": 8.462075233459473
    },
    {
      "epoch": 6.288814544677734e-05,
      "model_forward_time": 1.7940332889556885,
      "step": 65943
    },
    {
      "epoch": 6.288814544677734e-05,
      "step": 65943,
      "training_step_time": 11.606543779373169
    },
    {
      "epoch": 6.288909912109375e-05,
      "model_forward_time": 1.7294139862060547,
      "step": 65944
    },
    {
      "epoch": 6.288909912109375e-05,
      "step": 65944,
      "training_step_time": 9.27269983291626
    },
    {
      "epoch": 6.289005279541015e-05,
      "model_forward_time": 1.744626760482788,
      "step": 65945
    },
    {
      "epoch": 6.289005279541015e-05,
      "step": 65945,
      "training_step_time": 7.813101291656494
    },
    {
      "epoch": 6.289100646972656e-05,
      "model_forward_time": 1.7319481372833252,
      "step": 65946
    },
    {
      "epoch": 6.289100646972656e-05,
      "step": 65946,
      "training_step_time": 7.714812755584717
    },
    {
      "epoch": 6.289196014404297e-05,
      "model_forward_time": 1.7342228889465332,
      "step": 65947
    },
    {
      "epoch": 6.289196014404297e-05,
      "step": 65947,
      "training_step_time": 7.427126169204712
    },
    {
      "epoch": 6.289291381835937e-05,
      "model_forward_time": 1.7406134605407715,
      "step": 65948
    },
    {
      "epoch": 6.289291381835937e-05,
      "step": 65948,
      "training_step_time": 6.78267240524292
    },
    {
      "epoch": 6.289386749267578e-05,
      "model_forward_time": 1.7423241138458252,
      "step": 65949
    },
    {
      "epoch": 6.289386749267578e-05,
      "step": 65949,
      "training_step_time": 7.552601337432861
    },
    {
      "epoch": 6.289482116699219e-05,
      "grad_norm": 0.0968523621559143,
      "learning_rate": 2.8488375228234716e-06,
      "loss": 0.0553,
      "step": 65950
    },
    {
      "epoch": 6.289482116699219e-05,
      "model_forward_time": 1.721482753753662,
      "step": 65950
    },
    {
      "dynamics_loss_avg": 0.05326573858037591,
      "epoch": 6.289482116699219e-05,
      "step": 65950
    },
    {
      "action_loss_avg": 0.0020866956561803817,
      "epoch": 6.289482116699219e-05,
      "step": 65950
    },
    {
      "epoch": 6.289482116699219e-05,
      "step": 65950,
      "training_step_time": 7.541777610778809
    },
    {
      "epoch": 6.289577484130859e-05,
      "model_forward_time": 1.7293834686279297,
      "step": 65951
    },
    {
      "epoch": 6.289577484130859e-05,
      "step": 65951,
      "training_step_time": 7.664933919906616
    },
    {
      "epoch": 6.2896728515625e-05,
      "model_forward_time": 1.77431058883667,
      "step": 65952
    },
    {
      "epoch": 6.2896728515625e-05,
      "step": 65952,
      "training_step_time": 9.209603548049927
    },
    {
      "epoch": 6.289768218994141e-05,
      "model_forward_time": 2.9180808067321777,
      "step": 65953
    },
    {
      "epoch": 6.289768218994141e-05,
      "step": 65953,
      "training_step_time": 9.960073471069336
    },
    {
      "epoch": 6.28986358642578e-05,
      "model_forward_time": 1.7280786037445068,
      "step": 65954
    },
    {
      "epoch": 6.28986358642578e-05,
      "step": 65954,
      "training_step_time": 7.76408052444458
    },
    {
      "epoch": 6.289958953857422e-05,
      "model_forward_time": 1.7282915115356445,
      "step": 65955
    },
    {
      "epoch": 6.289958953857422e-05,
      "step": 65955,
      "training_step_time": 7.639001131057739
    },
    {
      "epoch": 6.290054321289063e-05,
      "model_forward_time": 1.7457818984985352,
      "step": 65956
    },
    {
      "epoch": 6.290054321289063e-05,
      "step": 65956,
      "training_step_time": 7.666222810745239
    },
    {
      "epoch": 6.290149688720703e-05,
      "model_forward_time": 1.7301428318023682,
      "step": 65957
    },
    {
      "epoch": 6.290149688720703e-05,
      "step": 65957,
      "training_step_time": 6.7848124504089355
    },
    {
      "epoch": 6.290245056152344e-05,
      "model_forward_time": 1.7331469058990479,
      "step": 65958
    },
    {
      "epoch": 6.290245056152344e-05,
      "step": 65958,
      "training_step_time": 7.551449537277222
    },
    {
      "epoch": 6.290340423583985e-05,
      "model_forward_time": 1.73130464553833,
      "step": 65959
    },
    {
      "epoch": 6.290340423583985e-05,
      "step": 65959,
      "training_step_time": 7.510512590408325
    },
    {
      "epoch": 6.290435791015624e-05,
      "grad_norm": 0.09508451074361801,
      "learning_rate": 2.8473450233132736e-06,
      "loss": 0.058,
      "step": 65960
    },
    {
      "epoch": 6.290435791015624e-05,
      "model_forward_time": 1.734482765197754,
      "step": 65960
    },
    {
      "dynamics_loss_avg": 0.0628592062741518,
      "epoch": 6.290435791015624e-05,
      "step": 65960
    },
    {
      "action_loss_avg": 0.002241290983511135,
      "epoch": 6.290435791015624e-05,
      "step": 65960
    },
    {
      "epoch": 6.290435791015624e-05,
      "step": 65960,
      "training_step_time": 7.721698522567749
    },
    {
      "epoch": 6.290531158447265e-05,
      "model_forward_time": 1.7334332466125488,
      "step": 65961
    },
    {
      "epoch": 6.290531158447265e-05,
      "step": 65961,
      "training_step_time": 10.290955305099487
    },
    {
      "epoch": 6.290626525878907e-05,
      "model_forward_time": 2.924954414367676,
      "step": 65962
    },
    {
      "epoch": 6.290626525878907e-05,
      "step": 65962,
      "training_step_time": 10.766989946365356
    },
    {
      "epoch": 6.290721893310546e-05,
      "model_forward_time": 1.739203691482544,
      "step": 65963
    },
    {
      "epoch": 6.290721893310546e-05,
      "step": 65963,
      "training_step_time": 7.783181190490723
    },
    {
      "epoch": 6.290817260742187e-05,
      "model_forward_time": 1.7333786487579346,
      "step": 65964
    },
    {
      "epoch": 6.290817260742187e-05,
      "step": 65964,
      "training_step_time": 7.717392206192017
    },
    {
      "epoch": 6.290912628173828e-05,
      "model_forward_time": 1.7355375289916992,
      "step": 65965
    },
    {
      "epoch": 6.290912628173828e-05,
      "step": 65965,
      "training_step_time": 7.7265095710754395
    },
    {
      "epoch": 6.291007995605468e-05,
      "model_forward_time": 1.7503056526184082,
      "step": 65966
    },
    {
      "epoch": 6.291007995605468e-05,
      "step": 65966,
      "training_step_time": 6.825051546096802
    },
    {
      "epoch": 6.291103363037109e-05,
      "model_forward_time": 1.734802484512329,
      "step": 65967
    },
    {
      "epoch": 6.291103363037109e-05,
      "step": 65967,
      "training_step_time": 7.633918046951294
    },
    {
      "epoch": 6.29119873046875e-05,
      "model_forward_time": 1.732895851135254,
      "step": 65968
    },
    {
      "epoch": 6.29119873046875e-05,
      "step": 65968,
      "training_step_time": 7.376772880554199
    },
    {
      "epoch": 6.29129409790039e-05,
      "model_forward_time": 1.7285785675048828,
      "step": 65969
    },
    {
      "epoch": 6.29129409790039e-05,
      "step": 65969,
      "training_step_time": 7.766898155212402
    },
    {
      "epoch": 6.291389465332031e-05,
      "grad_norm": 0.1127140074968338,
      "learning_rate": 2.84585275921419e-06,
      "loss": 0.0562,
      "step": 65970
    },
    {
      "epoch": 6.291389465332031e-05,
      "model_forward_time": 1.7337007522583008,
      "step": 65970
    },
    {
      "dynamics_loss_avg": 0.04701412925496697,
      "epoch": 6.291389465332031e-05,
      "step": 65970
    },
    {
      "action_loss_avg": 0.0012906879914226011,
      "epoch": 6.291389465332031e-05,
      "step": 65970
    },
    {
      "epoch": 6.291389465332031e-05,
      "step": 65970,
      "training_step_time": 8.529109954833984
    },
    {
      "epoch": 6.291484832763672e-05,
      "model_forward_time": 2.950540065765381,
      "step": 65971
    },
    {
      "epoch": 6.291484832763672e-05,
      "step": 65971,
      "training_step_time": 7.821859359741211
    },
    {
      "epoch": 6.291580200195312e-05,
      "model_forward_time": 1.7384734153747559,
      "step": 65972
    },
    {
      "epoch": 6.291580200195312e-05,
      "step": 65972,
      "training_step_time": 8.842412948608398
    },
    {
      "epoch": 6.291675567626953e-05,
      "model_forward_time": 1.742600440979004,
      "step": 65973
    },
    {
      "epoch": 6.291675567626953e-05,
      "step": 65973,
      "training_step_time": 7.685933351516724
    },
    {
      "epoch": 6.291770935058594e-05,
      "model_forward_time": 1.7233715057373047,
      "step": 65974
    },
    {
      "epoch": 6.291770935058594e-05,
      "step": 65974,
      "training_step_time": 7.668808937072754
    },
    {
      "epoch": 6.291866302490234e-05,
      "model_forward_time": 1.7448875904083252,
      "step": 65975
    },
    {
      "epoch": 6.291866302490234e-05,
      "step": 65975,
      "training_step_time": 6.755682945251465
    },
    {
      "epoch": 6.291961669921875e-05,
      "model_forward_time": 1.7432618141174316,
      "step": 65976
    },
    {
      "epoch": 6.291961669921875e-05,
      "step": 65976,
      "training_step_time": 7.622087717056274
    },
    {
      "epoch": 6.292057037353516e-05,
      "model_forward_time": 1.736229658126831,
      "step": 65977
    },
    {
      "epoch": 6.292057037353516e-05,
      "step": 65977,
      "training_step_time": 7.386539936065674
    },
    {
      "epoch": 6.292152404785156e-05,
      "model_forward_time": 1.73319411277771,
      "step": 65978
    },
    {
      "epoch": 6.292152404785156e-05,
      "step": 65978,
      "training_step_time": 7.676144123077393
    },
    {
      "epoch": 6.292247772216797e-05,
      "model_forward_time": 1.7464807033538818,
      "step": 65979
    },
    {
      "epoch": 6.292247772216797e-05,
      "step": 65979,
      "training_step_time": 9.754591226577759
    },
    {
      "epoch": 6.292343139648438e-05,
      "grad_norm": 0.09715922921895981,
      "learning_rate": 2.844360730689407e-06,
      "loss": 0.055,
      "step": 65980
    },
    {
      "epoch": 6.292343139648438e-05,
      "model_forward_time": 3.003594398498535,
      "step": 65980
    },
    {
      "dynamics_loss_avg": 0.05314614418894052,
      "epoch": 6.292343139648438e-05,
      "step": 65980
    },
    {
      "action_loss_avg": 0.001152349144103937,
      "epoch": 6.292343139648438e-05,
      "step": 65980
    },
    {
      "epoch": 6.292343139648438e-05,
      "step": 65980,
      "training_step_time": 8.383733034133911
    },
    {
      "epoch": 6.292438507080078e-05,
      "model_forward_time": 1.7429046630859375,
      "step": 65981
    },
    {
      "epoch": 6.292438507080078e-05,
      "step": 65981,
      "training_step_time": 9.395141839981079
    },
    {
      "epoch": 6.292533874511719e-05,
      "model_forward_time": 1.7456316947937012,
      "step": 65982
    },
    {
      "epoch": 6.292533874511719e-05,
      "step": 65982,
      "training_step_time": 7.7283034324646
    },
    {
      "epoch": 6.29262924194336e-05,
      "model_forward_time": 1.739994764328003,
      "step": 65983
    },
    {
      "epoch": 6.29262924194336e-05,
      "step": 65983,
      "training_step_time": 8.425722599029541
    },
    {
      "epoch": 6.292724609375e-05,
      "model_forward_time": 1.7403957843780518,
      "step": 65984
    },
    {
      "epoch": 6.292724609375e-05,
      "step": 65984,
      "training_step_time": 6.740673303604126
    },
    {
      "epoch": 6.29281997680664e-05,
      "model_forward_time": 1.7338030338287354,
      "step": 65985
    },
    {
      "epoch": 6.29281997680664e-05,
      "step": 65985,
      "training_step_time": 7.661491632461548
    },
    {
      "epoch": 6.292915344238282e-05,
      "model_forward_time": 1.7432246208190918,
      "step": 65986
    },
    {
      "epoch": 6.292915344238282e-05,
      "step": 65986,
      "training_step_time": 7.434335947036743
    },
    {
      "epoch": 6.293010711669921e-05,
      "model_forward_time": 1.7339460849761963,
      "step": 65987
    },
    {
      "epoch": 6.293010711669921e-05,
      "step": 65987,
      "training_step_time": 8.614053726196289
    },
    {
      "epoch": 6.293106079101562e-05,
      "model_forward_time": 1.749612808227539,
      "step": 65988
    },
    {
      "epoch": 6.293106079101562e-05,
      "step": 65988,
      "training_step_time": 7.673108816146851
    },
    {
      "epoch": 6.293201446533204e-05,
      "model_forward_time": 2.942857503890991,
      "step": 65989
    },
    {
      "epoch": 6.293201446533204e-05,
      "step": 65989,
      "training_step_time": 9.300803422927856
    },
    {
      "epoch": 6.293296813964843e-05,
      "grad_norm": 0.0924677699804306,
      "learning_rate": 2.8428689379020945e-06,
      "loss": 0.0589,
      "step": 65990
    },
    {
      "epoch": 6.293296813964843e-05,
      "model_forward_time": 1.7412500381469727,
      "step": 65990
    },
    {
      "dynamics_loss_avg": 0.054028937593102455,
      "epoch": 6.293296813964843e-05,
      "step": 65990
    },
    {
      "action_loss_avg": 0.0013027253793552519,
      "epoch": 6.293296813964843e-05,
      "step": 65990
    },
    {
      "epoch": 6.293296813964843e-05,
      "step": 65990,
      "training_step_time": 8.493712902069092
    },
    {
      "epoch": 6.293392181396484e-05,
      "model_forward_time": 1.7405095100402832,
      "step": 65991
    },
    {
      "epoch": 6.293392181396484e-05,
      "step": 65991,
      "training_step_time": 7.808048963546753
    },
    {
      "epoch": 6.293487548828125e-05,
      "model_forward_time": 1.7424380779266357,
      "step": 65992
    },
    {
      "epoch": 6.293487548828125e-05,
      "step": 65992,
      "training_step_time": 7.666039228439331
    },
    {
      "epoch": 6.293582916259765e-05,
      "model_forward_time": 1.750972032546997,
      "step": 65993
    },
    {
      "epoch": 6.293582916259765e-05,
      "step": 65993,
      "training_step_time": 7.8149285316467285
    },
    {
      "epoch": 6.293678283691406e-05,
      "model_forward_time": 1.7388343811035156,
      "step": 65994
    },
    {
      "epoch": 6.293678283691406e-05,
      "step": 65994,
      "training_step_time": 7.612635135650635
    },
    {
      "epoch": 6.293773651123047e-05,
      "model_forward_time": 1.7119572162628174,
      "step": 65995
    },
    {
      "epoch": 6.293773651123047e-05,
      "step": 65995,
      "training_step_time": 7.478468179702759
    },
    {
      "epoch": 6.293869018554687e-05,
      "model_forward_time": 1.723569631576538,
      "step": 65996
    },
    {
      "epoch": 6.293869018554687e-05,
      "step": 65996,
      "training_step_time": 7.560137033462524
    },
    {
      "epoch": 6.293964385986328e-05,
      "model_forward_time": 1.7369513511657715,
      "step": 65997
    },
    {
      "epoch": 6.293964385986328e-05,
      "step": 65997,
      "training_step_time": 9.266436576843262
    },
    {
      "epoch": 6.294059753417969e-05,
      "model_forward_time": 1.8269891738891602,
      "step": 65998
    },
    {
      "epoch": 6.294059753417969e-05,
      "step": 65998,
      "training_step_time": 11.18568754196167
    },
    {
      "epoch": 6.294155120849609e-05,
      "model_forward_time": 1.7330222129821777,
      "step": 65999
    },
    {
      "epoch": 6.294155120849609e-05,
      "step": 65999,
      "training_step_time": 9.700810432434082
    },
    {
      "epoch": 6.29425048828125e-05,
      "grad_norm": 0.1854822188615799,
      "learning_rate": 2.841377381015391e-06,
      "loss": 0.0528,
      "step": 66000
    },
    {
      "epoch": 6.29425048828125e-05,
      "model_forward_time": 1.7279045581817627,
      "step": 66000
    },
    {
      "dynamics_loss_avg": 0.04866573326289654,
      "epoch": 6.29425048828125e-05,
      "step": 66000
    },
    {
      "action_loss_avg": 0.001378643623320386,
      "epoch": 6.29425048828125e-05,
      "step": 66000
    },
    {
      "epoch": 6.29425048828125e-05,
      "step": 66000,
      "training_step_time": 7.1518518924713135
    },
    {
      "epoch": 6.294345855712891e-05,
      "model_forward_time": 1.7447683811187744,
      "step": 66001
    },
    {
      "epoch": 6.294345855712891e-05,
      "step": 66001,
      "training_step_time": 7.860101222991943
    },
    {
      "epoch": 6.294441223144531e-05,
      "model_forward_time": 1.7405085563659668,
      "step": 66002
    },
    {
      "epoch": 6.294441223144531e-05,
      "step": 66002,
      "training_step_time": 7.698939561843872
    },
    {
      "epoch": 6.294536590576172e-05,
      "model_forward_time": 1.7520556449890137,
      "step": 66003
    },
    {
      "epoch": 6.294536590576172e-05,
      "step": 66003,
      "training_step_time": 8.324660062789917
    },
    {
      "epoch": 6.294631958007813e-05,
      "model_forward_time": 1.7458415031433105,
      "step": 66004
    },
    {
      "epoch": 6.294631958007813e-05,
      "step": 66004,
      "training_step_time": 7.36877703666687
    },
    {
      "epoch": 6.294727325439453e-05,
      "model_forward_time": 1.736752986907959,
      "step": 66005
    },
    {
      "epoch": 6.294727325439453e-05,
      "step": 66005,
      "training_step_time": 7.584270238876343
    },
    {
      "epoch": 6.294822692871094e-05,
      "model_forward_time": 1.7478759288787842,
      "step": 66006
    },
    {
      "epoch": 6.294822692871094e-05,
      "step": 66006,
      "training_step_time": 8.60374116897583
    },
    {
      "epoch": 6.294918060302735e-05,
      "model_forward_time": 3.439556360244751,
      "step": 66007
    },
    {
      "epoch": 6.294918060302735e-05,
      "step": 66007,
      "training_step_time": 12.554086685180664
    },
    {
      "epoch": 6.295013427734375e-05,
      "model_forward_time": 1.7398896217346191,
      "step": 66008
    },
    {
      "epoch": 6.295013427734375e-05,
      "step": 66008,
      "training_step_time": 9.472927570343018
    },
    {
      "epoch": 6.295108795166016e-05,
      "model_forward_time": 1.7413606643676758,
      "step": 66009
    },
    {
      "epoch": 6.295108795166016e-05,
      "step": 66009,
      "training_step_time": 8.4030921459198
    },
    {
      "epoch": 6.295204162597657e-05,
      "grad_norm": 0.08413301408290863,
      "learning_rate": 2.8398860601924145e-06,
      "loss": 0.0566,
      "step": 66010
    },
    {
      "epoch": 6.295204162597657e-05,
      "model_forward_time": 1.7392971515655518,
      "step": 66010
    },
    {
      "dynamics_loss_avg": 0.046880653314292434,
      "epoch": 6.295204162597657e-05,
      "step": 66010
    },
    {
      "action_loss_avg": 0.0018066469114273787,
      "epoch": 6.295204162597657e-05,
      "step": 66010
    },
    {
      "epoch": 6.295204162597657e-05,
      "step": 66010,
      "training_step_time": 7.817490339279175
    },
    {
      "epoch": 6.295299530029296e-05,
      "model_forward_time": 1.3641822338104248,
      "step": 66011
    },
    {
      "epoch": 6.295299530029296e-05,
      "step": 66011,
      "training_step_time": 7.8083555698394775
    },
    {
      "epoch": 6.295394897460938e-05,
      "model_forward_time": 1.716444492340088,
      "step": 66012
    },
    {
      "epoch": 6.295394897460938e-05,
      "step": 66012,
      "training_step_time": 8.473099946975708
    },
    {
      "epoch": 6.295490264892579e-05,
      "model_forward_time": 1.7390072345733643,
      "step": 66013
    },
    {
      "epoch": 6.295490264892579e-05,
      "step": 66013,
      "training_step_time": 7.393895864486694
    },
    {
      "epoch": 6.295585632324218e-05,
      "model_forward_time": 1.7503211498260498,
      "step": 66014
    },
    {
      "epoch": 6.295585632324218e-05,
      "step": 66014,
      "training_step_time": 7.593021392822266
    },
    {
      "epoch": 6.29568099975586e-05,
      "model_forward_time": 1.7396862506866455,
      "step": 66015
    },
    {
      "epoch": 6.29568099975586e-05,
      "step": 66015,
      "training_step_time": 8.479419946670532
    },
    {
      "epoch": 6.2957763671875e-05,
      "model_forward_time": 2.970156669616699,
      "step": 66016
    },
    {
      "epoch": 6.2957763671875e-05,
      "step": 66016,
      "training_step_time": 8.976124286651611
    },
    {
      "epoch": 6.29587173461914e-05,
      "model_forward_time": 1.7418746948242188,
      "step": 66017
    },
    {
      "epoch": 6.29587173461914e-05,
      "step": 66017,
      "training_step_time": 8.902558088302612
    },
    {
      "epoch": 6.295967102050781e-05,
      "model_forward_time": 1.7367699146270752,
      "step": 66018
    },
    {
      "epoch": 6.295967102050781e-05,
      "step": 66018,
      "training_step_time": 7.869510650634766
    },
    {
      "epoch": 6.296062469482422e-05,
      "model_forward_time": 1.3729252815246582,
      "step": 66019
    },
    {
      "epoch": 6.296062469482422e-05,
      "step": 66019,
      "training_step_time": 7.82740044593811
    },
    {
      "epoch": 6.296157836914062e-05,
      "grad_norm": 0.08560606837272644,
      "learning_rate": 2.838394975596248e-06,
      "loss": 0.0554,
      "step": 66020
    },
    {
      "epoch": 6.296157836914062e-05,
      "model_forward_time": 1.7348949909210205,
      "step": 66020
    },
    {
      "dynamics_loss_avg": 0.0621882339939475,
      "epoch": 6.296157836914062e-05,
      "step": 66020
    },
    {
      "action_loss_avg": 0.0017652701586484908,
      "epoch": 6.296157836914062e-05,
      "step": 66020
    },
    {
      "epoch": 6.296157836914062e-05,
      "step": 66020,
      "training_step_time": 7.841159105300903
    },
    {
      "epoch": 6.296253204345703e-05,
      "model_forward_time": 1.7406206130981445,
      "step": 66021
    },
    {
      "epoch": 6.296253204345703e-05,
      "step": 66021,
      "training_step_time": 7.699372053146362
    },
    {
      "epoch": 6.296348571777344e-05,
      "model_forward_time": 1.7420940399169922,
      "step": 66022
    },
    {
      "epoch": 6.296348571777344e-05,
      "step": 66022,
      "training_step_time": 7.448904991149902
    },
    {
      "epoch": 6.296443939208984e-05,
      "model_forward_time": 1.734276533126831,
      "step": 66023
    },
    {
      "epoch": 6.296443939208984e-05,
      "step": 66023,
      "training_step_time": 7.54072380065918
    },
    {
      "epoch": 6.296539306640625e-05,
      "model_forward_time": 0.7028965950012207,
      "step": 66024
    },
    {
      "epoch": 6.296539306640625e-05,
      "step": 66024,
      "training_step_time": 7.808240175247192
    },
    {
      "epoch": 6.296634674072266e-05,
      "model_forward_time": 1.8078227043151855,
      "step": 66025
    },
    {
      "epoch": 6.296634674072266e-05,
      "step": 66025,
      "training_step_time": 8.496994018554688
    },
    {
      "epoch": 6.296730041503906e-05,
      "model_forward_time": 2.868232011795044,
      "step": 66026
    },
    {
      "epoch": 6.296730041503906e-05,
      "step": 66026,
      "training_step_time": 8.48275637626648
    },
    {
      "epoch": 6.296825408935547e-05,
      "model_forward_time": 1.7519478797912598,
      "step": 66027
    },
    {
      "epoch": 6.296825408935547e-05,
      "step": 66027,
      "training_step_time": 8.38521957397461
    },
    {
      "epoch": 6.296920776367188e-05,
      "model_forward_time": 1.7397677898406982,
      "step": 66028
    },
    {
      "epoch": 6.296920776367188e-05,
      "step": 66028,
      "training_step_time": 7.890357732772827
    },
    {
      "epoch": 6.297016143798828e-05,
      "model_forward_time": 1.0223557949066162,
      "step": 66029
    },
    {
      "epoch": 6.297016143798828e-05,
      "step": 66029,
      "training_step_time": 7.799592018127441
    },
    {
      "epoch": 6.297111511230469e-05,
      "grad_norm": 0.103550985455513,
      "learning_rate": 2.83690412738996e-06,
      "loss": 0.0555,
      "step": 66030
    },
    {
      "epoch": 6.297111511230469e-05,
      "model_forward_time": 1.7426109313964844,
      "step": 66030
    },
    {
      "dynamics_loss_avg": 0.05809341371059418,
      "epoch": 6.297111511230469e-05,
      "step": 66030
    },
    {
      "action_loss_avg": 0.0014728313981322572,
      "epoch": 6.297111511230469e-05,
      "step": 66030
    },
    {
      "epoch": 6.297111511230469e-05,
      "step": 66030,
      "training_step_time": 7.592085361480713
    },
    {
      "epoch": 6.29720687866211e-05,
      "model_forward_time": 1.7399475574493408,
      "step": 66031
    },
    {
      "epoch": 6.29720687866211e-05,
      "step": 66031,
      "training_step_time": 7.482217788696289
    },
    {
      "epoch": 6.29730224609375e-05,
      "model_forward_time": 1.7348759174346924,
      "step": 66032
    },
    {
      "epoch": 6.29730224609375e-05,
      "step": 66032,
      "training_step_time": 7.523623466491699
    },
    {
      "epoch": 6.297397613525391e-05,
      "model_forward_time": 1.7535834312438965,
      "step": 66033
    },
    {
      "epoch": 6.297397613525391e-05,
      "step": 66033,
      "training_step_time": 8.707478761672974
    },
    {
      "epoch": 6.297492980957032e-05,
      "model_forward_time": 1.8013017177581787,
      "step": 66034
    },
    {
      "epoch": 6.297492980957032e-05,
      "step": 66034,
      "training_step_time": 8.735065460205078
    },
    {
      "epoch": 6.297588348388672e-05,
      "model_forward_time": 1.754504680633545,
      "step": 66035
    },
    {
      "epoch": 6.297588348388672e-05,
      "step": 66035,
      "training_step_time": 8.891849994659424
    },
    {
      "epoch": 6.297683715820313e-05,
      "model_forward_time": 2.935335159301758,
      "step": 66036
    },
    {
      "epoch": 6.297683715820313e-05,
      "step": 66036,
      "training_step_time": 8.847466468811035
    },
    {
      "epoch": 6.297779083251954e-05,
      "model_forward_time": 1.7498486042022705,
      "step": 66037
    },
    {
      "epoch": 6.297779083251954e-05,
      "step": 66037,
      "training_step_time": 8.504677534103394
    },
    {
      "epoch": 6.297874450683593e-05,
      "model_forward_time": 1.7411234378814697,
      "step": 66038
    },
    {
      "epoch": 6.297874450683593e-05,
      "step": 66038,
      "training_step_time": 7.902948617935181
    },
    {
      "epoch": 6.297969818115235e-05,
      "model_forward_time": 1.7486445903778076,
      "step": 66039
    },
    {
      "epoch": 6.297969818115235e-05,
      "step": 66039,
      "training_step_time": 7.355801582336426
    },
    {
      "epoch": 6.298065185546876e-05,
      "grad_norm": 0.09097936004400253,
      "learning_rate": 2.835413515736579e-06,
      "loss": 0.0569,
      "step": 66040
    },
    {
      "epoch": 6.298065185546876e-05,
      "model_forward_time": 1.7464678287506104,
      "step": 66040
    },
    {
      "dynamics_loss_avg": 0.06479139905422926,
      "epoch": 6.298065185546876e-05,
      "step": 66040
    },
    {
      "action_loss_avg": 0.0021247763477731498,
      "epoch": 6.298065185546876e-05,
      "step": 66040
    },
    {
      "epoch": 6.298065185546876e-05,
      "step": 66040,
      "training_step_time": 7.424110174179077
    },
    {
      "epoch": 6.298160552978515e-05,
      "model_forward_time": 1.7400603294372559,
      "step": 66041
    },
    {
      "epoch": 6.298160552978515e-05,
      "step": 66041,
      "training_step_time": 8.177024841308594
    },
    {
      "epoch": 6.298255920410156e-05,
      "model_forward_time": 1.7367503643035889,
      "step": 66042
    },
    {
      "epoch": 6.298255920410156e-05,
      "step": 66042,
      "training_step_time": 8.257518291473389
    },
    {
      "epoch": 6.298351287841797e-05,
      "model_forward_time": 1.033466100692749,
      "step": 66043
    },
    {
      "epoch": 6.298351287841797e-05,
      "step": 66043,
      "training_step_time": 8.483542442321777
    },
    {
      "epoch": 6.298446655273437e-05,
      "model_forward_time": 1.7322769165039062,
      "step": 66044
    },
    {
      "epoch": 6.298446655273437e-05,
      "step": 66044,
      "training_step_time": 9.273087978363037
    },
    {
      "epoch": 6.298542022705078e-05,
      "model_forward_time": 2.862762689590454,
      "step": 66045
    },
    {
      "epoch": 6.298542022705078e-05,
      "step": 66045,
      "training_step_time": 9.805204629898071
    },
    {
      "epoch": 6.29863739013672e-05,
      "model_forward_time": 1.7457268238067627,
      "step": 66046
    },
    {
      "epoch": 6.29863739013672e-05,
      "step": 66046,
      "training_step_time": 7.842895030975342
    },
    {
      "epoch": 6.298732757568359e-05,
      "model_forward_time": 1.7376015186309814,
      "step": 66047
    },
    {
      "epoch": 6.298732757568359e-05,
      "step": 66047,
      "training_step_time": 7.812723636627197
    },
    {
      "epoch": 6.298828125e-05,
      "model_forward_time": 1.7380192279815674,
      "step": 66048
    },
    {
      "epoch": 6.298828125e-05,
      "step": 66048,
      "training_step_time": 7.76346492767334
    },
    {
      "epoch": 6.298923492431641e-05,
      "model_forward_time": 1.7495107650756836,
      "step": 66049
    },
    {
      "epoch": 6.298923492431641e-05,
      "step": 66049,
      "training_step_time": 7.402312994003296
    },
    {
      "epoch": 6.299018859863281e-05,
      "grad_norm": 0.142248272895813,
      "learning_rate": 2.833923140799127e-06,
      "loss": 0.0561,
      "step": 66050
    },
    {
      "epoch": 6.299018859863281e-05,
      "model_forward_time": 1.7404415607452393,
      "step": 66050
    },
    {
      "dynamics_loss_avg": 0.06251633781939744,
      "epoch": 6.299018859863281e-05,
      "step": 66050
    },
    {
      "action_loss_avg": 0.0017712677596136927,
      "epoch": 6.299018859863281e-05,
      "step": 66050
    },
    {
      "epoch": 6.299018859863281e-05,
      "step": 66050,
      "training_step_time": 7.348193645477295
    },
    {
      "epoch": 6.299114227294922e-05,
      "model_forward_time": 1.7448546886444092,
      "step": 66051
    },
    {
      "epoch": 6.299114227294922e-05,
      "step": 66051,
      "training_step_time": 8.39907169342041
    },
    {
      "epoch": 6.299209594726563e-05,
      "model_forward_time": 1.7502741813659668,
      "step": 66052
    },
    {
      "epoch": 6.299209594726563e-05,
      "step": 66052,
      "training_step_time": 8.86737847328186
    },
    {
      "epoch": 6.299304962158203e-05,
      "model_forward_time": 1.7393617630004883,
      "step": 66053
    },
    {
      "epoch": 6.299304962158203e-05,
      "step": 66053,
      "training_step_time": 9.246659278869629
    },
    {
      "epoch": 6.299400329589844e-05,
      "model_forward_time": 1.749990701675415,
      "step": 66054
    },
    {
      "epoch": 6.299400329589844e-05,
      "step": 66054,
      "training_step_time": 10.58000135421753
    },
    {
      "epoch": 6.299495697021485e-05,
      "model_forward_time": 1.7433788776397705,
      "step": 66055
    },
    {
      "epoch": 6.299495697021485e-05,
      "step": 66055,
      "training_step_time": 7.804958820343018
    },
    {
      "epoch": 6.299591064453125e-05,
      "model_forward_time": 1.7522025108337402,
      "step": 66056
    },
    {
      "epoch": 6.299591064453125e-05,
      "step": 66056,
      "training_step_time": 7.830543279647827
    },
    {
      "epoch": 6.299686431884766e-05,
      "model_forward_time": 1.7241864204406738,
      "step": 66057
    },
    {
      "epoch": 6.299686431884766e-05,
      "step": 66057,
      "training_step_time": 7.791213035583496
    },
    {
      "epoch": 6.299781799316407e-05,
      "model_forward_time": 1.7260072231292725,
      "step": 66058
    },
    {
      "epoch": 6.299781799316407e-05,
      "step": 66058,
      "training_step_time": 7.400045394897461
    },
    {
      "epoch": 6.299877166748047e-05,
      "model_forward_time": 1.725224494934082,
      "step": 66059
    },
    {
      "epoch": 6.299877166748047e-05,
      "step": 66059,
      "training_step_time": 7.327606678009033
    },
    {
      "epoch": 6.299972534179688e-05,
      "grad_norm": 0.08587172627449036,
      "learning_rate": 2.832433002740581e-06,
      "loss": 0.0545,
      "step": 66060
    },
    {
      "epoch": 6.299972534179688e-05,
      "model_forward_time": 1.7056059837341309,
      "step": 66060
    },
    {
      "dynamics_loss_avg": 0.05059735402464867,
      "epoch": 6.299972534179688e-05,
      "step": 66060
    },
    {
      "action_loss_avg": 0.0022669302532449366,
      "epoch": 6.299972534179688e-05,
      "step": 66060
    },
    {
      "epoch": 6.299972534179688e-05,
      "step": 66060,
      "training_step_time": 7.476684093475342
    },
    {
      "epoch": 6.300067901611329e-05,
      "model_forward_time": 1.7088532447814941,
      "step": 66061
    },
    {
      "epoch": 6.300067901611329e-05,
      "step": 66061,
      "training_step_time": 7.734654188156128
    },
    {
      "epoch": 6.300163269042969e-05,
      "model_forward_time": 1.733832836151123,
      "step": 66062
    },
    {
      "epoch": 6.300163269042969e-05,
      "step": 66062,
      "training_step_time": 7.871042251586914
    },
    {
      "epoch": 6.30025863647461e-05,
      "model_forward_time": 1.7119240760803223,
      "step": 66063
    },
    {
      "epoch": 6.30025863647461e-05,
      "step": 66063,
      "training_step_time": 8.894641637802124
    },
    {
      "epoch": 6.30035400390625e-05,
      "model_forward_time": 2.9590227603912354,
      "step": 66064
    },
    {
      "epoch": 6.30035400390625e-05,
      "step": 66064,
      "training_step_time": 7.875211000442505
    },
    {
      "epoch": 6.30044937133789e-05,
      "model_forward_time": 1.7078955173492432,
      "step": 66065
    },
    {
      "epoch": 6.30044937133789e-05,
      "step": 66065,
      "training_step_time": 7.812291145324707
    },
    {
      "epoch": 6.300544738769531e-05,
      "model_forward_time": 1.7067229747772217,
      "step": 66066
    },
    {
      "epoch": 6.300544738769531e-05,
      "step": 66066,
      "training_step_time": 7.731255054473877
    },
    {
      "epoch": 6.300640106201171e-05,
      "model_forward_time": 1.709547519683838,
      "step": 66067
    },
    {
      "epoch": 6.300640106201171e-05,
      "step": 66067,
      "training_step_time": 7.444987535476685
    },
    {
      "epoch": 6.300735473632812e-05,
      "model_forward_time": 1.3347439765930176,
      "step": 66068
    },
    {
      "epoch": 6.300735473632812e-05,
      "step": 66068,
      "training_step_time": 7.463642358779907
    },
    {
      "epoch": 6.300830841064453e-05,
      "model_forward_time": 1.7310447692871094,
      "step": 66069
    },
    {
      "epoch": 6.300830841064453e-05,
      "step": 66069,
      "training_step_time": 7.5325727462768555
    },
    {
      "epoch": 6.300926208496093e-05,
      "grad_norm": 0.09231957048177719,
      "learning_rate": 2.8309431017239047e-06,
      "loss": 0.0565,
      "step": 66070
    },
    {
      "epoch": 6.300926208496093e-05,
      "model_forward_time": 1.7115886211395264,
      "step": 66070
    },
    {
      "dynamics_loss_avg": 0.0633759293705225,
      "epoch": 6.300926208496093e-05,
      "step": 66070
    },
    {
      "action_loss_avg": 0.0023811017046682535,
      "epoch": 6.300926208496093e-05,
      "step": 66070
    },
    {
      "epoch": 6.300926208496093e-05,
      "step": 66070,
      "training_step_time": 7.725223541259766
    },
    {
      "epoch": 6.301021575927734e-05,
      "model_forward_time": 1.0113046169281006,
      "step": 66071
    },
    {
      "epoch": 6.301021575927734e-05,
      "step": 66071,
      "training_step_time": 7.914717197418213
    },
    {
      "epoch": 6.301116943359375e-05,
      "model_forward_time": 1.7146451473236084,
      "step": 66072
    },
    {
      "epoch": 6.301116943359375e-05,
      "step": 66072,
      "training_step_time": 8.44555401802063
    },
    {
      "epoch": 6.301212310791015e-05,
      "model_forward_time": 1.73075270652771,
      "step": 66073
    },
    {
      "epoch": 6.301212310791015e-05,
      "step": 66073,
      "training_step_time": 9.450847148895264
    },
    {
      "epoch": 6.301307678222656e-05,
      "model_forward_time": 1.7243731021881104,
      "step": 66074
    },
    {
      "epoch": 6.301307678222656e-05,
      "step": 66074,
      "training_step_time": 7.762198209762573
    },
    {
      "epoch": 6.301403045654297e-05,
      "model_forward_time": 1.727684736251831,
      "step": 66075
    },
    {
      "epoch": 6.301403045654297e-05,
      "step": 66075,
      "training_step_time": 7.871622323989868
    },
    {
      "epoch": 6.301498413085937e-05,
      "model_forward_time": 1.0119397640228271,
      "step": 66076
    },
    {
      "epoch": 6.301498413085937e-05,
      "step": 66076,
      "training_step_time": 7.710264682769775
    },
    {
      "epoch": 6.301593780517578e-05,
      "model_forward_time": 1.7110600471496582,
      "step": 66077
    },
    {
      "epoch": 6.301593780517578e-05,
      "step": 66077,
      "training_step_time": 6.808774948120117
    },
    {
      "epoch": 6.301689147949219e-05,
      "model_forward_time": 1.724412441253662,
      "step": 66078
    },
    {
      "epoch": 6.301689147949219e-05,
      "step": 66078,
      "training_step_time": 7.470630645751953
    },
    {
      "epoch": 6.301784515380859e-05,
      "model_forward_time": 1.7192349433898926,
      "step": 66079
    },
    {
      "epoch": 6.301784515380859e-05,
      "step": 66079,
      "training_step_time": 7.593683481216431
    },
    {
      "epoch": 6.3018798828125e-05,
      "grad_norm": 0.11895949393510818,
      "learning_rate": 2.829453437912028e-06,
      "loss": 0.0523,
      "step": 66080
    },
    {
      "epoch": 6.3018798828125e-05,
      "model_forward_time": 1.7259125709533691,
      "step": 66080
    },
    {
      "dynamics_loss_avg": 0.052666636556386946,
      "epoch": 6.3018798828125e-05,
      "step": 66080
    },
    {
      "action_loss_avg": 0.0013794992817565798,
      "epoch": 6.3018798828125e-05,
      "step": 66080
    },
    {
      "epoch": 6.3018798828125e-05,
      "step": 66080,
      "training_step_time": 8.339032173156738
    },
    {
      "epoch": 6.301975250244141e-05,
      "model_forward_time": 1.7223248481750488,
      "step": 66081
    },
    {
      "epoch": 6.301975250244141e-05,
      "step": 66081,
      "training_step_time": 8.992875576019287
    },
    {
      "epoch": 6.30207061767578e-05,
      "model_forward_time": 1.7153301239013672,
      "step": 66082
    },
    {
      "epoch": 6.30207061767578e-05,
      "step": 66082,
      "training_step_time": 9.265116214752197
    },
    {
      "epoch": 6.302165985107422e-05,
      "model_forward_time": 2.9369583129882812,
      "step": 66083
    },
    {
      "epoch": 6.302165985107422e-05,
      "step": 66083,
      "training_step_time": 7.798891305923462
    },
    {
      "epoch": 6.302261352539063e-05,
      "model_forward_time": 1.713646411895752,
      "step": 66084
    },
    {
      "epoch": 6.302261352539063e-05,
      "step": 66084,
      "training_step_time": 9.332450151443481
    },
    {
      "epoch": 6.302356719970703e-05,
      "model_forward_time": 1.7104690074920654,
      "step": 66085
    },
    {
      "epoch": 6.302356719970703e-05,
      "step": 66085,
      "training_step_time": 7.737447500228882
    },
    {
      "epoch": 6.302452087402344e-05,
      "model_forward_time": 1.7199244499206543,
      "step": 66086
    },
    {
      "epoch": 6.302452087402344e-05,
      "step": 66086,
      "training_step_time": 6.77592134475708
    },
    {
      "epoch": 6.302547454833985e-05,
      "model_forward_time": 1.329820156097412,
      "step": 66087
    },
    {
      "epoch": 6.302547454833985e-05,
      "step": 66087,
      "training_step_time": 7.458234071731567
    },
    {
      "epoch": 6.302642822265624e-05,
      "model_forward_time": 1.7495801448822021,
      "step": 66088
    },
    {
      "epoch": 6.302642822265624e-05,
      "step": 66088,
      "training_step_time": 7.5201194286346436
    },
    {
      "epoch": 6.302738189697265e-05,
      "model_forward_time": 1.731015920639038,
      "step": 66089
    },
    {
      "epoch": 6.302738189697265e-05,
      "step": 66089,
      "training_step_time": 7.740281105041504
    },
    {
      "epoch": 6.302833557128907e-05,
      "grad_norm": 0.10643837600946426,
      "learning_rate": 2.827964011467862e-06,
      "loss": 0.0506,
      "step": 66090
    },
    {
      "epoch": 6.302833557128907e-05,
      "model_forward_time": 1.7188758850097656,
      "step": 66090
    },
    {
      "dynamics_loss_avg": 0.0413757249712944,
      "epoch": 6.302833557128907e-05,
      "step": 66090
    },
    {
      "action_loss_avg": 0.001979917779681273,
      "epoch": 6.302833557128907e-05,
      "step": 66090
    },
    {
      "epoch": 6.302833557128907e-05,
      "step": 66090,
      "training_step_time": 7.740496397018433
    },
    {
      "epoch": 6.302928924560546e-05,
      "model_forward_time": 1.330826997756958,
      "step": 66091
    },
    {
      "epoch": 6.302928924560546e-05,
      "step": 66091,
      "training_step_time": 7.7115514278411865
    },
    {
      "epoch": 6.303024291992187e-05,
      "model_forward_time": 1.7002708911895752,
      "step": 66092
    },
    {
      "epoch": 6.303024291992187e-05,
      "step": 66092,
      "training_step_time": 9.882706880569458
    },
    {
      "epoch": 6.303119659423828e-05,
      "model_forward_time": 1.7077398300170898,
      "step": 66093
    },
    {
      "epoch": 6.303119659423828e-05,
      "step": 66093,
      "training_step_time": 7.696213245391846
    },
    {
      "epoch": 6.303215026855468e-05,
      "model_forward_time": 1.7172751426696777,
      "step": 66094
    },
    {
      "epoch": 6.303215026855468e-05,
      "step": 66094,
      "training_step_time": 7.724623441696167
    },
    {
      "epoch": 6.303310394287109e-05,
      "model_forward_time": 1.7172877788543701,
      "step": 66095
    },
    {
      "epoch": 6.303310394287109e-05,
      "step": 66095,
      "training_step_time": 6.730932235717773
    },
    {
      "epoch": 6.30340576171875e-05,
      "model_forward_time": 1.713313341140747,
      "step": 66096
    },
    {
      "epoch": 6.30340576171875e-05,
      "step": 66096,
      "training_step_time": 7.431306838989258
    },
    {
      "epoch": 6.30350112915039e-05,
      "model_forward_time": 1.7346408367156982,
      "step": 66097
    },
    {
      "epoch": 6.30350112915039e-05,
      "step": 66097,
      "training_step_time": 7.517401933670044
    },
    {
      "epoch": 6.303596496582031e-05,
      "model_forward_time": 1.7255334854125977,
      "step": 66098
    },
    {
      "epoch": 6.303596496582031e-05,
      "step": 66098,
      "training_step_time": 8.540871381759644
    },
    {
      "epoch": 6.303691864013672e-05,
      "model_forward_time": 1.7106330394744873,
      "step": 66099
    },
    {
      "epoch": 6.303691864013672e-05,
      "step": 66099,
      "training_step_time": 9.643922328948975
    },
    {
      "epoch": 6.303787231445312e-05,
      "grad_norm": 0.07304802536964417,
      "learning_rate": 2.826474822554284e-06,
      "loss": 0.0529,
      "step": 66100
    },
    {
      "epoch": 6.303787231445312e-05,
      "model_forward_time": 1.7276880741119385,
      "step": 66100
    },
    {
      "dynamics_loss_avg": 0.05827359016984701,
      "epoch": 6.303787231445312e-05,
      "step": 66100
    },
    {
      "action_loss_avg": 0.0021703810314647855,
      "epoch": 6.303787231445312e-05,
      "step": 66100
    },
    {
      "epoch": 6.303787231445312e-05,
      "step": 66100,
      "training_step_time": 9.161903619766235
    },
    {
      "epoch": 6.303882598876953e-05,
      "model_forward_time": 1.714972734451294,
      "step": 66101
    },
    {
      "epoch": 6.303882598876953e-05,
      "step": 66101,
      "training_step_time": 8.84619927406311
    },
    {
      "epoch": 6.303977966308594e-05,
      "model_forward_time": 2.5779383182525635,
      "step": 66102
    },
    {
      "epoch": 6.303977966308594e-05,
      "step": 66102,
      "training_step_time": 7.804599761962891
    },
    {
      "epoch": 6.304073333740234e-05,
      "model_forward_time": 1.7062411308288574,
      "step": 66103
    },
    {
      "epoch": 6.304073333740234e-05,
      "step": 66103,
      "training_step_time": 7.791434049606323
    },
    {
      "epoch": 6.304168701171875e-05,
      "model_forward_time": 1.7216477394104004,
      "step": 66104
    },
    {
      "epoch": 6.304168701171875e-05,
      "step": 66104,
      "training_step_time": 6.839439868927002
    },
    {
      "epoch": 6.304264068603516e-05,
      "model_forward_time": 1.7220048904418945,
      "step": 66105
    },
    {
      "epoch": 6.304264068603516e-05,
      "step": 66105,
      "training_step_time": 7.262891530990601
    },
    {
      "epoch": 6.304359436035156e-05,
      "model_forward_time": 1.72666335105896,
      "step": 66106
    },
    {
      "epoch": 6.304359436035156e-05,
      "step": 66106,
      "training_step_time": 7.499713659286499
    },
    {
      "epoch": 6.304454803466797e-05,
      "model_forward_time": 1.7158896923065186,
      "step": 66107
    },
    {
      "epoch": 6.304454803466797e-05,
      "step": 66107,
      "training_step_time": 7.762048006057739
    },
    {
      "epoch": 6.304550170898438e-05,
      "model_forward_time": 1.712885856628418,
      "step": 66108
    },
    {
      "epoch": 6.304550170898438e-05,
      "step": 66108,
      "training_step_time": 9.3348708152771
    },
    {
      "epoch": 6.304645538330078e-05,
      "model_forward_time": 1.7068216800689697,
      "step": 66109
    },
    {
      "epoch": 6.304645538330078e-05,
      "step": 66109,
      "training_step_time": 8.788015365600586
    },
    {
      "epoch": 6.304740905761719e-05,
      "grad_norm": 0.11560987681150436,
      "learning_rate": 2.8249858713341517e-06,
      "loss": 0.0557,
      "step": 66110
    },
    {
      "epoch": 6.304740905761719e-05,
      "model_forward_time": 1.7174341678619385,
      "step": 66110
    },
    {
      "dynamics_loss_avg": 0.059571648761630056,
      "epoch": 6.304740905761719e-05,
      "step": 66110
    },
    {
      "action_loss_avg": 0.0023006881499895828,
      "epoch": 6.304740905761719e-05,
      "step": 66110
    },
    {
      "epoch": 6.304740905761719e-05,
      "step": 66110,
      "training_step_time": 8.87126111984253
    },
    {
      "epoch": 6.30483627319336e-05,
      "model_forward_time": 2.9131948947906494,
      "step": 66111
    },
    {
      "epoch": 6.30483627319336e-05,
      "step": 66111,
      "training_step_time": 8.741358041763306
    },
    {
      "epoch": 6.304931640625e-05,
      "model_forward_time": 1.346644401550293,
      "step": 66112
    },
    {
      "epoch": 6.304931640625e-05,
      "step": 66112,
      "training_step_time": 7.848528146743774
    },
    {
      "epoch": 6.30502700805664e-05,
      "model_forward_time": 1.7171299457550049,
      "step": 66113
    },
    {
      "epoch": 6.30502700805664e-05,
      "step": 66113,
      "training_step_time": 7.706878662109375
    },
    {
      "epoch": 6.305122375488282e-05,
      "model_forward_time": 1.7086997032165527,
      "step": 66114
    },
    {
      "epoch": 6.305122375488282e-05,
      "step": 66114,
      "training_step_time": 6.766977787017822
    },
    {
      "epoch": 6.305217742919921e-05,
      "model_forward_time": 1.7136344909667969,
      "step": 66115
    },
    {
      "epoch": 6.305217742919921e-05,
      "step": 66115,
      "training_step_time": 7.517307758331299
    },
    {
      "epoch": 6.305313110351562e-05,
      "model_forward_time": 1.7224934101104736,
      "step": 66116
    },
    {
      "epoch": 6.305313110351562e-05,
      "step": 66116,
      "training_step_time": 7.678331136703491
    },
    {
      "epoch": 6.305408477783204e-05,
      "model_forward_time": 1.7261829376220703,
      "step": 66117
    },
    {
      "epoch": 6.305408477783204e-05,
      "step": 66117,
      "training_step_time": 8.288360595703125
    },
    {
      "epoch": 6.305503845214843e-05,
      "model_forward_time": 1.7087023258209229,
      "step": 66118
    },
    {
      "epoch": 6.305503845214843e-05,
      "step": 66118,
      "training_step_time": 8.322445154190063
    },
    {
      "epoch": 6.305599212646484e-05,
      "model_forward_time": 1.7169914245605469,
      "step": 66119
    },
    {
      "epoch": 6.305599212646484e-05,
      "step": 66119,
      "training_step_time": 8.593857765197754
    },
    {
      "epoch": 6.305694580078125e-05,
      "grad_norm": 0.07265196740627289,
      "learning_rate": 2.823497157970294e-06,
      "loss": 0.0521,
      "step": 66120
    },
    {
      "epoch": 6.305694580078125e-05,
      "model_forward_time": 2.9266276359558105,
      "step": 66120
    },
    {
      "dynamics_loss_avg": 0.06203436348587275,
      "epoch": 6.305694580078125e-05,
      "step": 66120
    },
    {
      "action_loss_avg": 0.001567669556243345,
      "epoch": 6.305694580078125e-05,
      "step": 66120
    },
    {
      "epoch": 6.305694580078125e-05,
      "step": 66120,
      "training_step_time": 7.785165786743164
    },
    {
      "epoch": 6.305789947509765e-05,
      "model_forward_time": 1.7236371040344238,
      "step": 66121
    },
    {
      "epoch": 6.305789947509765e-05,
      "step": 66121,
      "training_step_time": 7.855240106582642
    },
    {
      "epoch": 6.305885314941406e-05,
      "model_forward_time": 1.7087018489837646,
      "step": 66122
    },
    {
      "epoch": 6.305885314941406e-05,
      "step": 66122,
      "training_step_time": 7.714522838592529
    },
    {
      "epoch": 6.305980682373047e-05,
      "model_forward_time": 1.7082319259643555,
      "step": 66123
    },
    {
      "epoch": 6.305980682373047e-05,
      "step": 66123,
      "training_step_time": 6.696469783782959
    },
    {
      "epoch": 6.306076049804687e-05,
      "model_forward_time": 1.7136292457580566,
      "step": 66124
    },
    {
      "epoch": 6.306076049804687e-05,
      "step": 66124,
      "training_step_time": 7.480463743209839
    },
    {
      "epoch": 6.306171417236328e-05,
      "model_forward_time": 1.7368719577789307,
      "step": 66125
    },
    {
      "epoch": 6.306171417236328e-05,
      "step": 66125,
      "training_step_time": 8.678628921508789
    },
    {
      "epoch": 6.306266784667969e-05,
      "model_forward_time": 1.3425040245056152,
      "step": 66126
    },
    {
      "epoch": 6.306266784667969e-05,
      "step": 66126,
      "training_step_time": 7.680741786956787
    },
    {
      "epoch": 6.306362152099609e-05,
      "model_forward_time": 1.7164573669433594,
      "step": 66127
    },
    {
      "epoch": 6.306362152099609e-05,
      "step": 66127,
      "training_step_time": 8.605607986450195
    },
    {
      "epoch": 6.30645751953125e-05,
      "model_forward_time": 1.709146499633789,
      "step": 66128
    },
    {
      "epoch": 6.30645751953125e-05,
      "step": 66128,
      "training_step_time": 7.799532413482666
    },
    {
      "epoch": 6.306552886962891e-05,
      "model_forward_time": 2.8418145179748535,
      "step": 66129
    },
    {
      "epoch": 6.306552886962891e-05,
      "step": 66129,
      "training_step_time": 9.14034366607666
    },
    {
      "epoch": 6.306648254394531e-05,
      "grad_norm": 0.08806934952735901,
      "learning_rate": 2.8220086826255167e-06,
      "loss": 0.0528,
      "step": 66130
    },
    {
      "epoch": 6.306648254394531e-05,
      "model_forward_time": 1.7089014053344727,
      "step": 66130
    },
    {
      "dynamics_loss_avg": 0.05776597261428833,
      "epoch": 6.306648254394531e-05,
      "step": 66130
    },
    {
      "action_loss_avg": 0.0017676154326181858,
      "epoch": 6.306648254394531e-05,
      "step": 66130
    },
    {
      "epoch": 6.306648254394531e-05,
      "step": 66130,
      "training_step_time": 8.845060348510742
    },
    {
      "epoch": 6.306743621826172e-05,
      "model_forward_time": 1.70430588722229,
      "step": 66131
    },
    {
      "epoch": 6.306743621826172e-05,
      "step": 66131,
      "training_step_time": 7.8403449058532715
    },
    {
      "epoch": 6.306838989257813e-05,
      "model_forward_time": 1.7197937965393066,
      "step": 66132
    },
    {
      "epoch": 6.306838989257813e-05,
      "step": 66132,
      "training_step_time": 6.855426549911499
    },
    {
      "epoch": 6.306934356689453e-05,
      "model_forward_time": 1.7072045803070068,
      "step": 66133
    },
    {
      "epoch": 6.306934356689453e-05,
      "step": 66133,
      "training_step_time": 8.44016408920288
    },
    {
      "epoch": 6.307029724121094e-05,
      "model_forward_time": 1.753725528717041,
      "step": 66134
    },
    {
      "epoch": 6.307029724121094e-05,
      "step": 66134,
      "training_step_time": 7.865437269210815
    },
    {
      "epoch": 6.307125091552735e-05,
      "model_forward_time": 1.7255644798278809,
      "step": 66135
    },
    {
      "epoch": 6.307125091552735e-05,
      "step": 66135,
      "training_step_time": 8.735244989395142
    },
    {
      "epoch": 6.307220458984375e-05,
      "model_forward_time": 1.7055726051330566,
      "step": 66136
    },
    {
      "epoch": 6.307220458984375e-05,
      "step": 66136,
      "training_step_time": 9.97803258895874
    },
    {
      "epoch": 6.307315826416016e-05,
      "model_forward_time": 1.7113492488861084,
      "step": 66137
    },
    {
      "epoch": 6.307315826416016e-05,
      "step": 66137,
      "training_step_time": 8.917646169662476
    },
    {
      "epoch": 6.307411193847657e-05,
      "model_forward_time": 1.7211151123046875,
      "step": 66138
    },
    {
      "epoch": 6.307411193847657e-05,
      "step": 66138,
      "training_step_time": 8.929322719573975
    },
    {
      "epoch": 6.307506561279296e-05,
      "model_forward_time": 1.7141423225402832,
      "step": 66139
    },
    {
      "epoch": 6.307506561279296e-05,
      "step": 66139,
      "training_step_time": 7.899917840957642
    },
    {
      "epoch": 6.307601928710938e-05,
      "grad_norm": 0.10552173852920532,
      "learning_rate": 2.8205204454625934e-06,
      "loss": 0.0523,
      "step": 66140
    },
    {
      "epoch": 6.307601928710938e-05,
      "model_forward_time": 1.7207705974578857,
      "step": 66140
    },
    {
      "dynamics_loss_avg": 0.05950471628457308,
      "epoch": 6.307601928710938e-05,
      "step": 66140
    },
    {
      "action_loss_avg": 0.001372388208983466,
      "epoch": 6.307601928710938e-05,
      "step": 66140
    },
    {
      "epoch": 6.307601928710938e-05,
      "step": 66140,
      "training_step_time": 7.774876594543457
    },
    {
      "epoch": 6.307697296142579e-05,
      "model_forward_time": 1.7160849571228027,
      "step": 66141
    },
    {
      "epoch": 6.307697296142579e-05,
      "step": 66141,
      "training_step_time": 6.788784503936768
    },
    {
      "epoch": 6.307792663574218e-05,
      "model_forward_time": 1.7120893001556396,
      "step": 66142
    },
    {
      "epoch": 6.307792663574218e-05,
      "step": 66142,
      "training_step_time": 7.50808310508728
    },
    {
      "epoch": 6.30788803100586e-05,
      "model_forward_time": 1.73567795753479,
      "step": 66143
    },
    {
      "epoch": 6.30788803100586e-05,
      "step": 66143,
      "training_step_time": 7.8002753257751465
    },
    {
      "epoch": 6.3079833984375e-05,
      "model_forward_time": 1.7208566665649414,
      "step": 66144
    },
    {
      "epoch": 6.3079833984375e-05,
      "step": 66144,
      "training_step_time": 8.447115421295166
    },
    {
      "epoch": 6.30807876586914e-05,
      "model_forward_time": 1.7120847702026367,
      "step": 66145
    },
    {
      "epoch": 6.30807876586914e-05,
      "step": 66145,
      "training_step_time": 9.640395402908325
    },
    {
      "epoch": 6.308174133300781e-05,
      "model_forward_time": 1.719397783279419,
      "step": 66146
    },
    {
      "epoch": 6.308174133300781e-05,
      "step": 66146,
      "training_step_time": 8.480751276016235
    },
    {
      "epoch": 6.308269500732422e-05,
      "model_forward_time": 1.710143804550171,
      "step": 66147
    },
    {
      "epoch": 6.308269500732422e-05,
      "step": 66147,
      "training_step_time": 7.8032591342926025
    },
    {
      "epoch": 6.308364868164062e-05,
      "model_forward_time": 2.899250030517578,
      "step": 66148
    },
    {
      "epoch": 6.308364868164062e-05,
      "step": 66148,
      "training_step_time": 7.7422075271606445
    },
    {
      "epoch": 6.308460235595703e-05,
      "model_forward_time": 1.7099580764770508,
      "step": 66149
    },
    {
      "epoch": 6.308460235595703e-05,
      "step": 66149,
      "training_step_time": 7.704221487045288
    },
    {
      "epoch": 6.308555603027344e-05,
      "grad_norm": 0.09581028670072556,
      "learning_rate": 2.8190324466442797e-06,
      "loss": 0.0545,
      "step": 66150
    },
    {
      "epoch": 6.308555603027344e-05,
      "model_forward_time": 1.706061840057373,
      "step": 66150
    },
    {
      "dynamics_loss_avg": 0.05875408798456192,
      "epoch": 6.308555603027344e-05,
      "step": 66150
    },
    {
      "action_loss_avg": 0.0013317093544173986,
      "epoch": 6.308555603027344e-05,
      "step": 66150
    },
    {
      "epoch": 6.308555603027344e-05,
      "step": 66150,
      "training_step_time": 6.831130743026733
    },
    {
      "epoch": 6.308650970458984e-05,
      "model_forward_time": 1.7068474292755127,
      "step": 66151
    },
    {
      "epoch": 6.308650970458984e-05,
      "step": 66151,
      "training_step_time": 8.410158157348633
    },
    {
      "epoch": 6.308746337890625e-05,
      "model_forward_time": 1.7297847270965576,
      "step": 66152
    },
    {
      "epoch": 6.308746337890625e-05,
      "step": 66152,
      "training_step_time": 7.751583814620972
    },
    {
      "epoch": 6.308841705322266e-05,
      "model_forward_time": 1.7109358310699463,
      "step": 66153
    },
    {
      "epoch": 6.308841705322266e-05,
      "step": 66153,
      "training_step_time": 9.386614084243774
    },
    {
      "epoch": 6.308937072753906e-05,
      "model_forward_time": 0.9878032207489014,
      "step": 66154
    },
    {
      "epoch": 6.308937072753906e-05,
      "step": 66154,
      "training_step_time": 7.675665616989136
    },
    {
      "epoch": 6.309032440185547e-05,
      "model_forward_time": 1.7051615715026855,
      "step": 66155
    },
    {
      "epoch": 6.309032440185547e-05,
      "step": 66155,
      "training_step_time": 9.41537880897522
    },
    {
      "epoch": 6.309127807617188e-05,
      "model_forward_time": 1.7217905521392822,
      "step": 66156
    },
    {
      "epoch": 6.309127807617188e-05,
      "step": 66156,
      "training_step_time": 9.774364948272705
    },
    {
      "epoch": 6.309223175048828e-05,
      "model_forward_time": 2.91913104057312,
      "step": 66157
    },
    {
      "epoch": 6.309223175048828e-05,
      "step": 66157,
      "training_step_time": 8.48752498626709
    },
    {
      "epoch": 6.309318542480469e-05,
      "model_forward_time": 1.7215006351470947,
      "step": 66158
    },
    {
      "epoch": 6.309318542480469e-05,
      "step": 66158,
      "training_step_time": 7.742231845855713
    },
    {
      "epoch": 6.30941390991211e-05,
      "model_forward_time": 1.7173595428466797,
      "step": 66159
    },
    {
      "epoch": 6.30941390991211e-05,
      "step": 66159,
      "training_step_time": 7.746927261352539
    },
    {
      "epoch": 6.30950927734375e-05,
      "grad_norm": 0.10864557325839996,
      "learning_rate": 2.8175446863332967e-06,
      "loss": 0.0522,
      "step": 66160
    },
    {
      "epoch": 6.30950927734375e-05,
      "model_forward_time": 1.7243471145629883,
      "step": 66160
    },
    {
      "dynamics_loss_avg": 0.04462311454117298,
      "epoch": 6.30950927734375e-05,
      "step": 66160
    },
    {
      "action_loss_avg": 0.002287081626127474,
      "epoch": 6.30950927734375e-05,
      "step": 66160
    },
    {
      "epoch": 6.30950927734375e-05,
      "step": 66160,
      "training_step_time": 7.434609651565552
    },
    {
      "epoch": 6.309604644775391e-05,
      "model_forward_time": 1.7359833717346191,
      "step": 66161
    },
    {
      "epoch": 6.309604644775391e-05,
      "step": 66161,
      "training_step_time": 7.694438695907593
    },
    {
      "epoch": 6.309700012207032e-05,
      "model_forward_time": 1.7128143310546875,
      "step": 66162
    },
    {
      "epoch": 6.309700012207032e-05,
      "step": 66162,
      "training_step_time": 7.768388032913208
    },
    {
      "epoch": 6.309795379638672e-05,
      "model_forward_time": 1.334780216217041,
      "step": 66163
    },
    {
      "epoch": 6.309795379638672e-05,
      "step": 66163,
      "training_step_time": 9.502850532531738
    },
    {
      "epoch": 6.309890747070313e-05,
      "model_forward_time": 1.7118806838989258,
      "step": 66164
    },
    {
      "epoch": 6.309890747070313e-05,
      "step": 66164,
      "training_step_time": 8.89611268043518
    },
    {
      "epoch": 6.309986114501954e-05,
      "model_forward_time": 1.3487446308135986,
      "step": 66165
    },
    {
      "epoch": 6.309986114501954e-05,
      "step": 66165,
      "training_step_time": 8.682876110076904
    },
    {
      "epoch": 6.310081481933593e-05,
      "model_forward_time": 1.7141687870025635,
      "step": 66166
    },
    {
      "epoch": 6.310081481933593e-05,
      "step": 66166,
      "training_step_time": 7.857152700424194
    },
    {
      "epoch": 6.310176849365235e-05,
      "model_forward_time": 2.5597808361053467,
      "step": 66167
    },
    {
      "epoch": 6.310176849365235e-05,
      "step": 66167,
      "training_step_time": 7.858847618103027
    },
    {
      "epoch": 6.310272216796876e-05,
      "model_forward_time": 1.718003273010254,
      "step": 66168
    },
    {
      "epoch": 6.310272216796876e-05,
      "step": 66168,
      "training_step_time": 7.664901971817017
    },
    {
      "epoch": 6.310367584228515e-05,
      "model_forward_time": 1.7203772068023682,
      "step": 66169
    },
    {
      "epoch": 6.310367584228515e-05,
      "step": 66169,
      "training_step_time": 7.474524259567261
    },
    {
      "epoch": 6.310462951660156e-05,
      "grad_norm": 0.0963062047958374,
      "learning_rate": 2.8160571646923453e-06,
      "loss": 0.0494,
      "step": 66170
    },
    {
      "epoch": 6.310462951660156e-05,
      "model_forward_time": 1.6985273361206055,
      "step": 66170
    },
    {
      "dynamics_loss_avg": 0.04422249374911189,
      "epoch": 6.310462951660156e-05,
      "step": 66170
    },
    {
      "action_loss_avg": 0.0011491699406178668,
      "epoch": 6.310462951660156e-05,
      "step": 66170
    },
    {
      "epoch": 6.310462951660156e-05,
      "step": 66170,
      "training_step_time": 7.590208530426025
    },
    {
      "epoch": 6.310558319091797e-05,
      "model_forward_time": 1.7513017654418945,
      "step": 66171
    },
    {
      "epoch": 6.310558319091797e-05,
      "step": 66171,
      "training_step_time": 7.8083319664001465
    },
    {
      "epoch": 6.310653686523437e-05,
      "model_forward_time": 1.7149896621704102,
      "step": 66172
    },
    {
      "epoch": 6.310653686523437e-05,
      "step": 66172,
      "training_step_time": 8.575229167938232
    },
    {
      "epoch": 6.310749053955078e-05,
      "model_forward_time": 1.7185235023498535,
      "step": 66173
    },
    {
      "epoch": 6.310749053955078e-05,
      "step": 66173,
      "training_step_time": 8.34039306640625
    },
    {
      "epoch": 6.31084442138672e-05,
      "model_forward_time": 1.717069387435913,
      "step": 66174
    },
    {
      "epoch": 6.31084442138672e-05,
      "step": 66174,
      "training_step_time": 7.838741302490234
    },
    {
      "epoch": 6.310939788818359e-05,
      "model_forward_time": 1.712742567062378,
      "step": 66175
    },
    {
      "epoch": 6.310939788818359e-05,
      "step": 66175,
      "training_step_time": 7.78433084487915
    },
    {
      "epoch": 6.31103515625e-05,
      "model_forward_time": 2.9282338619232178,
      "step": 66176
    },
    {
      "epoch": 6.31103515625e-05,
      "step": 66176,
      "training_step_time": 7.763745307922363
    },
    {
      "epoch": 6.311130523681641e-05,
      "model_forward_time": 1.7116279602050781,
      "step": 66177
    },
    {
      "epoch": 6.311130523681641e-05,
      "step": 66177,
      "training_step_time": 7.725286960601807
    },
    {
      "epoch": 6.311225891113281e-05,
      "model_forward_time": 1.7159628868103027,
      "step": 66178
    },
    {
      "epoch": 6.311225891113281e-05,
      "step": 66178,
      "training_step_time": 7.73118257522583
    },
    {
      "epoch": 6.311321258544922e-05,
      "model_forward_time": 1.7235116958618164,
      "step": 66179
    },
    {
      "epoch": 6.311321258544922e-05,
      "step": 66179,
      "training_step_time": 8.040309190750122
    },
    {
      "epoch": 6.311416625976563e-05,
      "grad_norm": 0.08408176898956299,
      "learning_rate": 2.8145698818840996e-06,
      "loss": 0.0529,
      "step": 66180
    },
    {
      "epoch": 6.311416625976563e-05,
      "model_forward_time": 1.7421765327453613,
      "step": 66180
    },
    {
      "dynamics_loss_avg": 0.0508293729275465,
      "epoch": 6.311416625976563e-05,
      "step": 66180
    },
    {
      "action_loss_avg": 0.0010717615426983685,
      "epoch": 6.311416625976563e-05,
      "step": 66180
    },
    {
      "epoch": 6.311416625976563e-05,
      "step": 66180,
      "training_step_time": 8.461748361587524
    },
    {
      "epoch": 6.311511993408203e-05,
      "model_forward_time": 1.7134003639221191,
      "step": 66181
    },
    {
      "epoch": 6.311511993408203e-05,
      "step": 66181,
      "training_step_time": 8.477864980697632
    },
    {
      "epoch": 6.311607360839844e-05,
      "model_forward_time": 1.3556981086730957,
      "step": 66182
    },
    {
      "epoch": 6.311607360839844e-05,
      "step": 66182,
      "training_step_time": 9.704440116882324
    },
    {
      "epoch": 6.311702728271485e-05,
      "model_forward_time": 1.7146563529968262,
      "step": 66183
    },
    {
      "epoch": 6.311702728271485e-05,
      "step": 66183,
      "training_step_time": 7.84608793258667
    },
    {
      "epoch": 6.311798095703125e-05,
      "model_forward_time": 1.702591896057129,
      "step": 66184
    },
    {
      "epoch": 6.311798095703125e-05,
      "step": 66184,
      "training_step_time": 9.466074228286743
    },
    {
      "epoch": 6.311893463134766e-05,
      "model_forward_time": 2.848503828048706,
      "step": 66185
    },
    {
      "epoch": 6.311893463134766e-05,
      "step": 66185,
      "training_step_time": 8.722842931747437
    },
    {
      "epoch": 6.311988830566407e-05,
      "model_forward_time": 1.7105844020843506,
      "step": 66186
    },
    {
      "epoch": 6.311988830566407e-05,
      "step": 66186,
      "training_step_time": 7.7349324226379395
    },
    {
      "epoch": 6.312084197998047e-05,
      "model_forward_time": 1.7088847160339355,
      "step": 66187
    },
    {
      "epoch": 6.312084197998047e-05,
      "step": 66187,
      "training_step_time": 7.7394492626190186
    },
    {
      "epoch": 6.312179565429688e-05,
      "model_forward_time": 1.7010579109191895,
      "step": 66188
    },
    {
      "epoch": 6.312179565429688e-05,
      "step": 66188,
      "training_step_time": 8.367233276367188
    },
    {
      "epoch": 6.312274932861327e-05,
      "model_forward_time": 1.7273483276367188,
      "step": 66189
    },
    {
      "epoch": 6.312274932861327e-05,
      "step": 66189,
      "training_step_time": 7.686356782913208
    },
    {
      "epoch": 6.312370300292969e-05,
      "grad_norm": 0.10957516729831696,
      "learning_rate": 2.8130828380712067e-06,
      "loss": 0.0517,
      "step": 66190
    },
    {
      "epoch": 6.312370300292969e-05,
      "model_forward_time": 1.7165157794952393,
      "step": 66190
    },
    {
      "dynamics_loss_avg": 0.051423351652920245,
      "epoch": 6.312370300292969e-05,
      "step": 66190
    },
    {
      "action_loss_avg": 0.0011967982281930745,
      "epoch": 6.312370300292969e-05,
      "step": 66190
    },
    {
      "epoch": 6.312370300292969e-05,
      "step": 66190,
      "training_step_time": 8.318593978881836
    },
    {
      "epoch": 6.31246566772461e-05,
      "model_forward_time": 1.7355704307556152,
      "step": 66191
    },
    {
      "epoch": 6.31246566772461e-05,
      "step": 66191,
      "training_step_time": 9.403056383132935
    },
    {
      "epoch": 6.31256103515625e-05,
      "model_forward_time": 1.7197620868682861,
      "step": 66192
    },
    {
      "epoch": 6.31256103515625e-05,
      "step": 66192,
      "training_step_time": 9.452438831329346
    },
    {
      "epoch": 6.31265640258789e-05,
      "model_forward_time": 1.714188575744629,
      "step": 66193
    },
    {
      "epoch": 6.31265640258789e-05,
      "step": 66193,
      "training_step_time": 7.7322611808776855
    },
    {
      "epoch": 6.312751770019531e-05,
      "model_forward_time": 1.325761079788208,
      "step": 66194
    },
    {
      "epoch": 6.312751770019531e-05,
      "step": 66194,
      "training_step_time": 7.757646322250366
    },
    {
      "epoch": 6.312847137451171e-05,
      "model_forward_time": 1.724663257598877,
      "step": 66195
    },
    {
      "epoch": 6.312847137451171e-05,
      "step": 66195,
      "training_step_time": 7.719276428222656
    },
    {
      "epoch": 6.312942504882812e-05,
      "model_forward_time": 1.7005462646484375,
      "step": 66196
    },
    {
      "epoch": 6.312942504882812e-05,
      "step": 66196,
      "training_step_time": 7.7143847942352295
    },
    {
      "epoch": 6.313037872314453e-05,
      "model_forward_time": 1.714355230331421,
      "step": 66197
    },
    {
      "epoch": 6.313037872314453e-05,
      "step": 66197,
      "training_step_time": 7.4637322425842285
    },
    {
      "epoch": 6.313133239746093e-05,
      "model_forward_time": 1.7263853549957275,
      "step": 66198
    },
    {
      "epoch": 6.313133239746093e-05,
      "step": 66198,
      "training_step_time": 7.505835056304932
    },
    {
      "epoch": 6.313228607177734e-05,
      "model_forward_time": 1.715733289718628,
      "step": 66199
    },
    {
      "epoch": 6.313228607177734e-05,
      "step": 66199,
      "training_step_time": 7.70569920539856
    },
    {
      "epoch": 6.313323974609375e-05,
      "grad_norm": 0.17173567414283752,
      "learning_rate": 2.811596033416285e-06,
      "loss": 0.0557,
      "step": 66200
    },
    {
      "epoch": 6.313323974609375e-05,
      "model_forward_time": 1.7320678234100342,
      "step": 66200
    },
    {
      "dynamics_loss_avg": 0.057726665027439596,
      "epoch": 6.313323974609375e-05,
      "step": 66200
    },
    {
      "action_loss_avg": 0.0015935584611725061,
      "epoch": 6.313323974609375e-05,
      "step": 66200
    },
    {
      "epoch": 6.313323974609375e-05,
      "step": 66200,
      "training_step_time": 10.305788278579712
    },
    {
      "epoch": 6.313419342041015e-05,
      "model_forward_time": 1.7059803009033203,
      "step": 66201
    },
    {
      "epoch": 6.313419342041015e-05,
      "step": 66201,
      "training_step_time": 7.870640277862549
    },
    {
      "epoch": 6.313514709472656e-05,
      "model_forward_time": 1.7075178623199463,
      "step": 66202
    },
    {
      "epoch": 6.313514709472656e-05,
      "step": 66202,
      "training_step_time": 7.702490568161011
    },
    {
      "epoch": 6.313610076904297e-05,
      "model_forward_time": 1.7169785499572754,
      "step": 66203
    },
    {
      "epoch": 6.313610076904297e-05,
      "step": 66203,
      "training_step_time": 8.731475353240967
    },
    {
      "epoch": 6.313705444335937e-05,
      "model_forward_time": 2.912924289703369,
      "step": 66204
    },
    {
      "epoch": 6.313705444335937e-05,
      "step": 66204,
      "training_step_time": 8.694996356964111
    },
    {
      "epoch": 6.313800811767578e-05,
      "model_forward_time": 1.72104811668396,
      "step": 66205
    },
    {
      "epoch": 6.313800811767578e-05,
      "step": 66205,
      "training_step_time": 7.711015939712524
    },
    {
      "epoch": 6.313896179199219e-05,
      "model_forward_time": 1.6944072246551514,
      "step": 66206
    },
    {
      "epoch": 6.313896179199219e-05,
      "step": 66206,
      "training_step_time": 7.5477471351623535
    },
    {
      "epoch": 6.313991546630859e-05,
      "model_forward_time": 1.7147471904754639,
      "step": 66207
    },
    {
      "epoch": 6.313991546630859e-05,
      "step": 66207,
      "training_step_time": 7.435399055480957
    },
    {
      "epoch": 6.3140869140625e-05,
      "model_forward_time": 1.7242283821105957,
      "step": 66208
    },
    {
      "epoch": 6.3140869140625e-05,
      "step": 66208,
      "training_step_time": 7.570206880569458
    },
    {
      "epoch": 6.314182281494141e-05,
      "model_forward_time": 1.7432377338409424,
      "step": 66209
    },
    {
      "epoch": 6.314182281494141e-05,
      "step": 66209,
      "training_step_time": 9.027130365371704
    },
    {
      "epoch": 6.31427764892578e-05,
      "grad_norm": 0.11251813173294067,
      "learning_rate": 2.8101094680819306e-06,
      "loss": 0.0541,
      "step": 66210
    },
    {
      "epoch": 6.31427764892578e-05,
      "model_forward_time": 1.721001148223877,
      "step": 66210
    },
    {
      "dynamics_loss_avg": 0.06815588427707553,
      "epoch": 6.31427764892578e-05,
      "step": 66210
    },
    {
      "action_loss_avg": 0.002200223479303531,
      "epoch": 6.31427764892578e-05,
      "step": 66210
    },
    {
      "epoch": 6.31427764892578e-05,
      "step": 66210,
      "training_step_time": 7.943045139312744
    },
    {
      "epoch": 6.314373016357422e-05,
      "model_forward_time": 1.713808298110962,
      "step": 66211
    },
    {
      "epoch": 6.314373016357422e-05,
      "step": 66211,
      "training_step_time": 8.513824939727783
    },
    {
      "epoch": 6.314468383789063e-05,
      "model_forward_time": 1.7071938514709473,
      "step": 66212
    },
    {
      "epoch": 6.314468383789063e-05,
      "step": 66212,
      "training_step_time": 8.918777704238892
    },
    {
      "epoch": 6.314563751220703e-05,
      "model_forward_time": 2.897747278213501,
      "step": 66213
    },
    {
      "epoch": 6.314563751220703e-05,
      "step": 66213,
      "training_step_time": 7.876080274581909
    },
    {
      "epoch": 6.314659118652344e-05,
      "model_forward_time": 1.713794231414795,
      "step": 66214
    },
    {
      "epoch": 6.314659118652344e-05,
      "step": 66214,
      "training_step_time": 7.781513214111328
    },
    {
      "epoch": 6.314754486083985e-05,
      "model_forward_time": 1.326500654220581,
      "step": 66215
    },
    {
      "epoch": 6.314754486083985e-05,
      "step": 66215,
      "training_step_time": 7.485385179519653
    },
    {
      "epoch": 6.314849853515624e-05,
      "model_forward_time": 1.711308479309082,
      "step": 66216
    },
    {
      "epoch": 6.314849853515624e-05,
      "step": 66216,
      "training_step_time": 8.379082679748535
    },
    {
      "epoch": 6.314945220947266e-05,
      "model_forward_time": 1.7162749767303467,
      "step": 66217
    },
    {
      "epoch": 6.314945220947266e-05,
      "step": 66217,
      "training_step_time": 7.841156244277954
    },
    {
      "epoch": 6.315040588378907e-05,
      "model_forward_time": 1.727076768875122,
      "step": 66218
    },
    {
      "epoch": 6.315040588378907e-05,
      "step": 66218,
      "training_step_time": 9.821756839752197
    },
    {
      "epoch": 6.315135955810546e-05,
      "model_forward_time": 1.3257253170013428,
      "step": 66219
    },
    {
      "epoch": 6.315135955810546e-05,
      "step": 66219,
      "training_step_time": 8.824089527130127
    },
    {
      "epoch": 6.315231323242187e-05,
      "grad_norm": 0.1579565852880478,
      "learning_rate": 2.8086231422307152e-06,
      "loss": 0.054,
      "step": 66220
    },
    {
      "epoch": 6.315231323242187e-05,
      "model_forward_time": 1.7162141799926758,
      "step": 66220
    },
    {
      "dynamics_loss_avg": 0.04918276239186525,
      "epoch": 6.315231323242187e-05,
      "step": 66220
    },
    {
      "action_loss_avg": 0.0013296893215738238,
      "epoch": 6.315231323242187e-05,
      "step": 66220
    },
    {
      "epoch": 6.315231323242187e-05,
      "step": 66220,
      "training_step_time": 8.735080242156982
    },
    {
      "epoch": 6.315326690673828e-05,
      "model_forward_time": 1.7142064571380615,
      "step": 66221
    },
    {
      "epoch": 6.315326690673828e-05,
      "step": 66221,
      "training_step_time": 8.794810056686401
    },
    {
      "epoch": 6.315422058105468e-05,
      "model_forward_time": 2.8807222843170166,
      "step": 66222
    },
    {
      "epoch": 6.315422058105468e-05,
      "step": 66222,
      "training_step_time": 7.855750799179077
    },
    {
      "epoch": 6.315517425537109e-05,
      "model_forward_time": 1.718226671218872,
      "step": 66223
    },
    {
      "epoch": 6.315517425537109e-05,
      "step": 66223,
      "training_step_time": 7.610730171203613
    },
    {
      "epoch": 6.31561279296875e-05,
      "model_forward_time": 1.7108378410339355,
      "step": 66224
    },
    {
      "epoch": 6.31561279296875e-05,
      "step": 66224,
      "training_step_time": 7.77796196937561
    },
    {
      "epoch": 6.31570816040039e-05,
      "model_forward_time": 1.7156639099121094,
      "step": 66225
    },
    {
      "epoch": 6.31570816040039e-05,
      "step": 66225,
      "training_step_time": 7.459129095077515
    },
    {
      "epoch": 6.315803527832031e-05,
      "model_forward_time": 1.7364823818206787,
      "step": 66226
    },
    {
      "epoch": 6.315803527832031e-05,
      "step": 66226,
      "training_step_time": 7.802727937698364
    },
    {
      "epoch": 6.315898895263672e-05,
      "model_forward_time": 1.7132022380828857,
      "step": 66227
    },
    {
      "epoch": 6.315898895263672e-05,
      "step": 66227,
      "training_step_time": 8.368642568588257
    },
    {
      "epoch": 6.315994262695312e-05,
      "model_forward_time": 1.7245755195617676,
      "step": 66228
    },
    {
      "epoch": 6.315994262695312e-05,
      "step": 66228,
      "training_step_time": 8.631689071655273
    },
    {
      "epoch": 6.316089630126953e-05,
      "model_forward_time": 1.7105393409729004,
      "step": 66229
    },
    {
      "epoch": 6.316089630126953e-05,
      "step": 66229,
      "training_step_time": 9.699249744415283
    },
    {
      "epoch": 6.316184997558594e-05,
      "grad_norm": 0.11172664910554886,
      "learning_rate": 2.807137056025176e-06,
      "loss": 0.0555,
      "step": 66230
    },
    {
      "epoch": 6.316184997558594e-05,
      "model_forward_time": 1.7122271060943604,
      "step": 66230
    },
    {
      "dynamics_loss_avg": 0.035048763081431386,
      "epoch": 6.316184997558594e-05,
      "step": 66230
    },
    {
      "action_loss_avg": 0.00101034003746463,
      "epoch": 6.316184997558594e-05,
      "step": 66230
    },
    {
      "epoch": 6.316184997558594e-05,
      "step": 66230,
      "training_step_time": 7.771501064300537
    },
    {
      "epoch": 6.316280364990234e-05,
      "model_forward_time": 2.8452255725860596,
      "step": 66231
    },
    {
      "epoch": 6.316280364990234e-05,
      "step": 66231,
      "training_step_time": 7.83221173286438
    },
    {
      "epoch": 6.316375732421875e-05,
      "model_forward_time": 1.729593276977539,
      "step": 66232
    },
    {
      "epoch": 6.316375732421875e-05,
      "step": 66232,
      "training_step_time": 7.893348693847656
    },
    {
      "epoch": 6.316471099853516e-05,
      "model_forward_time": 1.7171351909637451,
      "step": 66233
    },
    {
      "epoch": 6.316471099853516e-05,
      "step": 66233,
      "training_step_time": 7.798382997512817
    },
    {
      "epoch": 6.316566467285156e-05,
      "model_forward_time": 1.7236053943634033,
      "step": 66234
    },
    {
      "epoch": 6.316566467285156e-05,
      "step": 66234,
      "training_step_time": 7.522958517074585
    },
    {
      "epoch": 6.316661834716797e-05,
      "model_forward_time": 1.7351374626159668,
      "step": 66235
    },
    {
      "epoch": 6.316661834716797e-05,
      "step": 66235,
      "training_step_time": 7.766648292541504
    },
    {
      "epoch": 6.316757202148438e-05,
      "model_forward_time": 1.7168004512786865,
      "step": 66236
    },
    {
      "epoch": 6.316757202148438e-05,
      "step": 66236,
      "training_step_time": 7.6952431201934814
    },
    {
      "epoch": 6.316852569580078e-05,
      "model_forward_time": 1.7424323558807373,
      "step": 66237
    },
    {
      "epoch": 6.316852569580078e-05,
      "step": 66237,
      "training_step_time": 8.563700199127197
    },
    {
      "epoch": 6.316947937011719e-05,
      "model_forward_time": 1.7184927463531494,
      "step": 66238
    },
    {
      "epoch": 6.316947937011719e-05,
      "step": 66238,
      "training_step_time": 9.914612531661987
    },
    {
      "epoch": 6.31704330444336e-05,
      "model_forward_time": 1.7088701725006104,
      "step": 66239
    },
    {
      "epoch": 6.31704330444336e-05,
      "step": 66239,
      "training_step_time": 7.675493240356445
    },
    {
      "epoch": 6.317138671875e-05,
      "grad_norm": 0.12809143960475922,
      "learning_rate": 2.805651209627831e-06,
      "loss": 0.0539,
      "step": 66240
    },
    {
      "epoch": 6.317138671875e-05,
      "model_forward_time": 2.809504270553589,
      "step": 66240
    },
    {
      "dynamics_loss_avg": 0.04245410533621907,
      "epoch": 6.317138671875e-05,
      "step": 66240
    },
    {
      "action_loss_avg": 0.00120902732887771,
      "epoch": 6.317138671875e-05,
      "step": 66240
    },
    {
      "epoch": 6.317138671875e-05,
      "step": 66240,
      "training_step_time": 7.801859617233276
    },
    {
      "epoch": 6.31723403930664e-05,
      "model_forward_time": 1.7125513553619385,
      "step": 66241
    },
    {
      "epoch": 6.31723403930664e-05,
      "step": 66241,
      "training_step_time": 7.785343885421753
    },
    {
      "epoch": 6.317329406738282e-05,
      "model_forward_time": 1.7116777896881104,
      "step": 66242
    },
    {
      "epoch": 6.317329406738282e-05,
      "step": 66242,
      "training_step_time": 8.367400646209717
    },
    {
      "epoch": 6.317424774169921e-05,
      "model_forward_time": 1.7132086753845215,
      "step": 66243
    },
    {
      "epoch": 6.317424774169921e-05,
      "step": 66243,
      "training_step_time": 7.435270547866821
    },
    {
      "epoch": 6.317520141601562e-05,
      "model_forward_time": 1.7111449241638184,
      "step": 66244
    },
    {
      "epoch": 6.317520141601562e-05,
      "step": 66244,
      "training_step_time": 7.738945960998535
    },
    {
      "epoch": 6.317615509033204e-05,
      "model_forward_time": 1.7164292335510254,
      "step": 66245
    },
    {
      "epoch": 6.317615509033204e-05,
      "step": 66245,
      "training_step_time": 7.723628759384155
    },
    {
      "epoch": 6.317710876464843e-05,
      "model_forward_time": 1.7699627876281738,
      "step": 66246
    },
    {
      "epoch": 6.317710876464843e-05,
      "step": 66246,
      "training_step_time": 7.803393602371216
    },
    {
      "epoch": 6.317806243896484e-05,
      "model_forward_time": 1.7115285396575928,
      "step": 66247
    },
    {
      "epoch": 6.317806243896484e-05,
      "step": 66247,
      "training_step_time": 9.409311056137085
    },
    {
      "epoch": 6.317901611328125e-05,
      "model_forward_time": 1.7174599170684814,
      "step": 66248
    },
    {
      "epoch": 6.317901611328125e-05,
      "step": 66248,
      "training_step_time": 7.826646566390991
    },
    {
      "epoch": 6.317996978759765e-05,
      "model_forward_time": 1.7100646495819092,
      "step": 66249
    },
    {
      "epoch": 6.317996978759765e-05,
      "step": 66249,
      "training_step_time": 7.7880754470825195
    },
    {
      "epoch": 6.318092346191406e-05,
      "grad_norm": 0.09139583259820938,
      "learning_rate": 2.804165603201171e-06,
      "loss": 0.0558,
      "step": 66250
    },
    {
      "epoch": 6.318092346191406e-05,
      "model_forward_time": 2.9664363861083984,
      "step": 66250
    },
    {
      "dynamics_loss_avg": 0.06545344293117523,
      "epoch": 6.318092346191406e-05,
      "step": 66250
    },
    {
      "action_loss_avg": 0.0021022737520979716,
      "epoch": 6.318092346191406e-05,
      "step": 66250
    },
    {
      "epoch": 6.318092346191406e-05,
      "step": 66250,
      "training_step_time": 7.842486381530762
    },
    {
      "epoch": 6.318187713623047e-05,
      "model_forward_time": 1.7213966846466064,
      "step": 66251
    },
    {
      "epoch": 6.318187713623047e-05,
      "step": 66251,
      "training_step_time": 7.617191791534424
    },
    {
      "epoch": 6.318283081054687e-05,
      "model_forward_time": 1.7068650722503662,
      "step": 66252
    },
    {
      "epoch": 6.318283081054687e-05,
      "step": 66252,
      "training_step_time": 8.501704692840576
    },
    {
      "epoch": 6.318378448486328e-05,
      "model_forward_time": 1.706446886062622,
      "step": 66253
    },
    {
      "epoch": 6.318378448486328e-05,
      "step": 66253,
      "training_step_time": 8.207919359207153
    },
    {
      "epoch": 6.318473815917969e-05,
      "model_forward_time": 1.732733964920044,
      "step": 66254
    },
    {
      "epoch": 6.318473815917969e-05,
      "step": 66254,
      "training_step_time": 8.363184928894043
    },
    {
      "epoch": 6.318569183349609e-05,
      "model_forward_time": 1.7213830947875977,
      "step": 66255
    },
    {
      "epoch": 6.318569183349609e-05,
      "step": 66255,
      "training_step_time": 9.412060976028442
    },
    {
      "epoch": 6.31866455078125e-05,
      "model_forward_time": 1.329153299331665,
      "step": 66256
    },
    {
      "epoch": 6.31866455078125e-05,
      "step": 66256,
      "training_step_time": 9.719369411468506
    },
    {
      "epoch": 6.318759918212891e-05,
      "model_forward_time": 1.7171673774719238,
      "step": 66257
    },
    {
      "epoch": 6.318759918212891e-05,
      "step": 66257,
      "training_step_time": 8.808714151382446
    },
    {
      "epoch": 6.318855285644531e-05,
      "model_forward_time": 1.7137608528137207,
      "step": 66258
    },
    {
      "epoch": 6.318855285644531e-05,
      "step": 66258,
      "training_step_time": 7.73876690864563
    },
    {
      "epoch": 6.318950653076172e-05,
      "model_forward_time": 2.902592897415161,
      "step": 66259
    },
    {
      "epoch": 6.318950653076172e-05,
      "step": 66259,
      "training_step_time": 7.747092247009277
    },
    {
      "epoch": 6.319046020507813e-05,
      "grad_norm": 0.09004615992307663,
      "learning_rate": 2.802680236907661e-06,
      "loss": 0.0588,
      "step": 66260
    },
    {
      "epoch": 6.319046020507813e-05,
      "model_forward_time": 1.7163970470428467,
      "step": 66260
    },
    {
      "dynamics_loss_avg": 0.06965540684759616,
      "epoch": 6.319046020507813e-05,
      "step": 66260
    },
    {
      "action_loss_avg": 0.001565070595825091,
      "epoch": 6.319046020507813e-05,
      "step": 66260
    },
    {
      "epoch": 6.319046020507813e-05,
      "step": 66260,
      "training_step_time": 7.761401414871216
    },
    {
      "epoch": 6.319141387939453e-05,
      "model_forward_time": 1.7085397243499756,
      "step": 66261
    },
    {
      "epoch": 6.319141387939453e-05,
      "step": 66261,
      "training_step_time": 7.675797700881958
    },
    {
      "epoch": 6.319236755371094e-05,
      "model_forward_time": 1.7180125713348389,
      "step": 66262
    },
    {
      "epoch": 6.319236755371094e-05,
      "step": 66262,
      "training_step_time": 7.6405189037323
    },
    {
      "epoch": 6.319332122802735e-05,
      "model_forward_time": 1.7563371658325195,
      "step": 66263
    },
    {
      "epoch": 6.319332122802735e-05,
      "step": 66263,
      "training_step_time": 7.61598801612854
    },
    {
      "epoch": 6.319427490234375e-05,
      "model_forward_time": 1.710721731185913,
      "step": 66264
    },
    {
      "epoch": 6.319427490234375e-05,
      "step": 66264,
      "training_step_time": 8.30201005935669
    },
    {
      "epoch": 6.319522857666016e-05,
      "model_forward_time": 0.9993526935577393,
      "step": 66265
    },
    {
      "epoch": 6.319522857666016e-05,
      "step": 66265,
      "training_step_time": 10.340795516967773
    },
    {
      "epoch": 6.319618225097657e-05,
      "model_forward_time": 1.7062397003173828,
      "step": 66266
    },
    {
      "epoch": 6.319618225097657e-05,
      "step": 66266,
      "training_step_time": 7.801594972610474
    },
    {
      "epoch": 6.319713592529296e-05,
      "model_forward_time": 1.7185134887695312,
      "step": 66267
    },
    {
      "epoch": 6.319713592529296e-05,
      "step": 66267,
      "training_step_time": 9.231138229370117
    },
    {
      "epoch": 6.319808959960938e-05,
      "model_forward_time": 2.8193037509918213,
      "step": 66268
    },
    {
      "epoch": 6.319808959960938e-05,
      "step": 66268,
      "training_step_time": 8.762878179550171
    },
    {
      "epoch": 6.319904327392579e-05,
      "model_forward_time": 1.3452727794647217,
      "step": 66269
    },
    {
      "epoch": 6.319904327392579e-05,
      "step": 66269,
      "training_step_time": 8.869178533554077
    },
    {
      "epoch": 6.319999694824218e-05,
      "grad_norm": 0.09397779405117035,
      "learning_rate": 2.8011951109097348e-06,
      "loss": 0.0546,
      "step": 66270
    },
    {
      "epoch": 6.319999694824218e-05,
      "model_forward_time": 1.7062954902648926,
      "step": 66270
    },
    {
      "dynamics_loss_avg": 0.06498468481004238,
      "epoch": 6.319999694824218e-05,
      "step": 66270
    },
    {
      "action_loss_avg": 0.0033866875994135624,
      "epoch": 6.319999694824218e-05,
      "step": 66270
    },
    {
      "epoch": 6.319999694824218e-05,
      "step": 66270,
      "training_step_time": 7.676218032836914
    },
    {
      "epoch": 6.32009506225586e-05,
      "model_forward_time": 1.7218644618988037,
      "step": 66271
    },
    {
      "epoch": 6.32009506225586e-05,
      "step": 66271,
      "training_step_time": 7.506703853607178
    },
    {
      "epoch": 6.3201904296875e-05,
      "model_forward_time": 1.7187154293060303,
      "step": 66272
    },
    {
      "epoch": 6.3201904296875e-05,
      "step": 66272,
      "training_step_time": 7.705050230026245
    },
    {
      "epoch": 6.32028579711914e-05,
      "model_forward_time": 1.7180323600769043,
      "step": 66273
    },
    {
      "epoch": 6.32028579711914e-05,
      "step": 66273,
      "training_step_time": 8.693009376525879
    },
    {
      "epoch": 6.320381164550781e-05,
      "model_forward_time": 1.7226476669311523,
      "step": 66274
    },
    {
      "epoch": 6.320381164550781e-05,
      "step": 66274,
      "training_step_time": 7.803248643875122
    },
    {
      "epoch": 6.320476531982422e-05,
      "model_forward_time": 1.7082264423370361,
      "step": 66275
    },
    {
      "epoch": 6.320476531982422e-05,
      "step": 66275,
      "training_step_time": 8.55254054069519
    },
    {
      "epoch": 6.320571899414062e-05,
      "model_forward_time": 1.7180426120758057,
      "step": 66276
    },
    {
      "epoch": 6.320571899414062e-05,
      "step": 66276,
      "training_step_time": 8.429004192352295
    },
    {
      "epoch": 6.320667266845703e-05,
      "model_forward_time": 1.7114310264587402,
      "step": 66277
    },
    {
      "epoch": 6.320667266845703e-05,
      "step": 66277,
      "training_step_time": 8.788234949111938
    },
    {
      "epoch": 6.320762634277344e-05,
      "model_forward_time": 2.978314161300659,
      "step": 66278
    },
    {
      "epoch": 6.320762634277344e-05,
      "step": 66278,
      "training_step_time": 7.881064414978027
    },
    {
      "epoch": 6.320858001708984e-05,
      "model_forward_time": 1.7213120460510254,
      "step": 66279
    },
    {
      "epoch": 6.320858001708984e-05,
      "step": 66279,
      "training_step_time": 8.335340023040771
    },
    {
      "epoch": 6.320953369140625e-05,
      "grad_norm": 0.08892010897397995,
      "learning_rate": 2.799710225369807e-06,
      "loss": 0.0554,
      "step": 66280
    },
    {
      "epoch": 6.320953369140625e-05,
      "model_forward_time": 1.7086093425750732,
      "step": 66280
    },
    {
      "dynamics_loss_avg": 0.05512060932815075,
      "epoch": 6.320953369140625e-05,
      "step": 66280
    },
    {
      "action_loss_avg": 0.0017189155565574764,
      "epoch": 6.320953369140625e-05,
      "step": 66280
    },
    {
      "epoch": 6.320953369140625e-05,
      "step": 66280,
      "training_step_time": 7.608335018157959
    },
    {
      "epoch": 6.321048736572266e-05,
      "model_forward_time": 1.721696376800537,
      "step": 66281
    },
    {
      "epoch": 6.321048736572266e-05,
      "step": 66281,
      "training_step_time": 7.735986232757568
    },
    {
      "epoch": 6.321144104003906e-05,
      "model_forward_time": 1.8317878246307373,
      "step": 66282
    },
    {
      "epoch": 6.321144104003906e-05,
      "step": 66282,
      "training_step_time": 8.721506834030151
    },
    {
      "epoch": 6.321239471435547e-05,
      "model_forward_time": 1.7073829174041748,
      "step": 66283
    },
    {
      "epoch": 6.321239471435547e-05,
      "step": 66283,
      "training_step_time": 7.77793550491333
    },
    {
      "epoch": 6.321334838867188e-05,
      "model_forward_time": 1.7165579795837402,
      "step": 66284
    },
    {
      "epoch": 6.321334838867188e-05,
      "step": 66284,
      "training_step_time": 8.739288568496704
    },
    {
      "epoch": 6.321430206298828e-05,
      "model_forward_time": 1.704477071762085,
      "step": 66285
    },
    {
      "epoch": 6.321430206298828e-05,
      "step": 66285,
      "training_step_time": 8.384973526000977
    },
    {
      "epoch": 6.321525573730469e-05,
      "model_forward_time": 1.714076280593872,
      "step": 66286
    },
    {
      "epoch": 6.321525573730469e-05,
      "step": 66286,
      "training_step_time": 7.6832170486450195
    },
    {
      "epoch": 6.32162094116211e-05,
      "model_forward_time": 2.86262583732605,
      "step": 66287
    },
    {
      "epoch": 6.32162094116211e-05,
      "step": 66287,
      "training_step_time": 7.7659752368927
    },
    {
      "epoch": 6.32171630859375e-05,
      "model_forward_time": 1.699859380722046,
      "step": 66288
    },
    {
      "epoch": 6.32171630859375e-05,
      "step": 66288,
      "training_step_time": 7.684232234954834
    },
    {
      "epoch": 6.321811676025391e-05,
      "model_forward_time": 1.7113127708435059,
      "step": 66289
    },
    {
      "epoch": 6.321811676025391e-05,
      "step": 66289,
      "training_step_time": 7.669213533401489
    },
    {
      "epoch": 6.321907043457032e-05,
      "grad_norm": 0.09836161136627197,
      "learning_rate": 2.7982255804502588e-06,
      "loss": 0.058,
      "step": 66290
    },
    {
      "epoch": 6.321907043457032e-05,
      "model_forward_time": 1.698899269104004,
      "step": 66290
    },
    {
      "dynamics_loss_avg": 0.06348836552351714,
      "epoch": 6.321907043457032e-05,
      "step": 66290
    },
    {
      "action_loss_avg": 0.0020651472237659617,
      "epoch": 6.321907043457032e-05,
      "step": 66290
    },
    {
      "epoch": 6.321907043457032e-05,
      "step": 66290,
      "training_step_time": 8.774680137634277
    },
    {
      "epoch": 6.322002410888672e-05,
      "model_forward_time": 1.8026740550994873,
      "step": 66291
    },
    {
      "epoch": 6.322002410888672e-05,
      "step": 66291,
      "training_step_time": 7.681763410568237
    },
    {
      "epoch": 6.322097778320313e-05,
      "model_forward_time": 1.7099249362945557,
      "step": 66292
    },
    {
      "epoch": 6.322097778320313e-05,
      "step": 66292,
      "training_step_time": 8.867515802383423
    },
    {
      "epoch": 6.322193145751954e-05,
      "model_forward_time": 1.707245111465454,
      "step": 66293
    },
    {
      "epoch": 6.322193145751954e-05,
      "step": 66293,
      "training_step_time": 7.857973098754883
    },
    {
      "epoch": 6.322288513183593e-05,
      "model_forward_time": 1.715862512588501,
      "step": 66294
    },
    {
      "epoch": 6.322288513183593e-05,
      "step": 66294,
      "training_step_time": 7.836386680603027
    },
    {
      "epoch": 6.322383880615235e-05,
      "model_forward_time": 1.6990365982055664,
      "step": 66295
    },
    {
      "epoch": 6.322383880615235e-05,
      "step": 66295,
      "training_step_time": 7.636563777923584
    },
    {
      "epoch": 6.322479248046876e-05,
      "model_forward_time": 2.8754868507385254,
      "step": 66296
    },
    {
      "epoch": 6.322479248046876e-05,
      "step": 66296,
      "training_step_time": 7.8524556159973145
    },
    {
      "epoch": 6.322574615478515e-05,
      "model_forward_time": 1.7050447463989258,
      "step": 66297
    },
    {
      "epoch": 6.322574615478515e-05,
      "step": 66297,
      "training_step_time": 7.702571153640747
    },
    {
      "epoch": 6.322669982910156e-05,
      "model_forward_time": 1.6956322193145752,
      "step": 66298
    },
    {
      "epoch": 6.322669982910156e-05,
      "step": 66298,
      "training_step_time": 7.669121980667114
    },
    {
      "epoch": 6.322765350341798e-05,
      "model_forward_time": 1.3188989162445068,
      "step": 66299
    },
    {
      "epoch": 6.322765350341798e-05,
      "step": 66299,
      "training_step_time": 7.750486373901367
    },
    {
      "epoch": 6.322860717773437e-05,
      "grad_norm": 0.10465198010206223,
      "learning_rate": 2.796741176313452e-06,
      "loss": 0.0554,
      "step": 66300
    },
    {
      "epoch": 6.322860717773437e-05,
      "model_forward_time": 1.3739702701568604,
      "step": 66300
    },
    {
      "dynamics_loss_avg": 0.059793674014508726,
      "epoch": 6.322860717773437e-05,
      "step": 66300
    },
    {
      "action_loss_avg": 0.0014972465025493876,
      "epoch": 6.322860717773437e-05,
      "step": 66300
    },
    {
      "epoch": 6.322860717773437e-05,
      "step": 66300,
      "training_step_time": 7.554887056350708
    },
    {
      "epoch": 6.322956085205078e-05,
      "model_forward_time": 1.7021636962890625,
      "step": 66301
    },
    {
      "epoch": 6.322956085205078e-05,
      "step": 66301,
      "training_step_time": 9.824188232421875
    },
    {
      "epoch": 6.32305145263672e-05,
      "model_forward_time": 1.7117254734039307,
      "step": 66302
    },
    {
      "epoch": 6.32305145263672e-05,
      "step": 66302,
      "training_step_time": 8.634502649307251
    },
    {
      "epoch": 6.323146820068359e-05,
      "model_forward_time": 1.713151454925537,
      "step": 66303
    },
    {
      "epoch": 6.323146820068359e-05,
      "step": 66303,
      "training_step_time": 10.321083068847656
    },
    {
      "epoch": 6.3232421875e-05,
      "model_forward_time": 1.721189260482788,
      "step": 66304
    },
    {
      "epoch": 6.3232421875e-05,
      "step": 66304,
      "training_step_time": 9.27462887763977
    },
    {
      "epoch": 6.323337554931641e-05,
      "model_forward_time": 1.7129592895507812,
      "step": 66305
    },
    {
      "epoch": 6.323337554931641e-05,
      "step": 66305,
      "training_step_time": 7.754272937774658
    },
    {
      "epoch": 6.323432922363281e-05,
      "model_forward_time": 2.9091291427612305,
      "step": 66306
    },
    {
      "epoch": 6.323432922363281e-05,
      "step": 66306,
      "training_step_time": 8.759728908538818
    },
    {
      "epoch": 6.323528289794922e-05,
      "model_forward_time": 1.7659854888916016,
      "step": 66307
    },
    {
      "epoch": 6.323528289794922e-05,
      "step": 66307,
      "training_step_time": 7.686627388000488
    },
    {
      "epoch": 6.323623657226563e-05,
      "model_forward_time": 1.7249886989593506,
      "step": 66308
    },
    {
      "epoch": 6.323623657226563e-05,
      "step": 66308,
      "training_step_time": 7.615516662597656
    },
    {
      "epoch": 6.323719024658203e-05,
      "model_forward_time": 1.740248203277588,
      "step": 66309
    },
    {
      "epoch": 6.323719024658203e-05,
      "step": 66309,
      "training_step_time": 7.472890853881836
    },
    {
      "epoch": 6.323814392089844e-05,
      "grad_norm": 0.10646022111177444,
      "learning_rate": 2.7952570131217165e-06,
      "loss": 0.058,
      "step": 66310
    },
    {
      "epoch": 6.323814392089844e-05,
      "model_forward_time": 1.7203254699707031,
      "step": 66310
    },
    {
      "dynamics_loss_avg": 0.057748675160109994,
      "epoch": 6.323814392089844e-05,
      "step": 66310
    },
    {
      "action_loss_avg": 0.0009975005144951866,
      "epoch": 6.323814392089844e-05,
      "step": 66310
    },
    {
      "epoch": 6.323814392089844e-05,
      "step": 66310,
      "training_step_time": 8.718666791915894
    },
    {
      "epoch": 6.323909759521485e-05,
      "model_forward_time": 1.7203950881958008,
      "step": 66311
    },
    {
      "epoch": 6.323909759521485e-05,
      "step": 66311,
      "training_step_time": 8.347364902496338
    },
    {
      "epoch": 6.324005126953125e-05,
      "model_forward_time": 1.718153715133667,
      "step": 66312
    },
    {
      "epoch": 6.324005126953125e-05,
      "step": 66312,
      "training_step_time": 7.817244529724121
    },
    {
      "epoch": 6.324100494384766e-05,
      "model_forward_time": 1.34649658203125,
      "step": 66313
    },
    {
      "epoch": 6.324100494384766e-05,
      "step": 66313,
      "training_step_time": 7.688835382461548
    },
    {
      "epoch": 6.324195861816406e-05,
      "model_forward_time": 1.3486778736114502,
      "step": 66314
    },
    {
      "epoch": 6.324195861816406e-05,
      "step": 66314,
      "training_step_time": 8.648959636688232
    },
    {
      "epoch": 6.324291229248047e-05,
      "model_forward_time": 1.7583386898040771,
      "step": 66315
    },
    {
      "epoch": 6.324291229248047e-05,
      "step": 66315,
      "training_step_time": 7.747547149658203
    },
    {
      "epoch": 6.324386596679688e-05,
      "model_forward_time": 1.7351324558258057,
      "step": 66316
    },
    {
      "epoch": 6.324386596679688e-05,
      "step": 66316,
      "training_step_time": 7.698667049407959
    },
    {
      "epoch": 6.324481964111327e-05,
      "model_forward_time": 1.7027592658996582,
      "step": 66317
    },
    {
      "epoch": 6.324481964111327e-05,
      "step": 66317,
      "training_step_time": 7.6828672885894775
    },
    {
      "epoch": 6.324577331542969e-05,
      "model_forward_time": 1.707298755645752,
      "step": 66318
    },
    {
      "epoch": 6.324577331542969e-05,
      "step": 66318,
      "training_step_time": 7.749103784561157
    },
    {
      "epoch": 6.32467269897461e-05,
      "model_forward_time": 1.8148436546325684,
      "step": 66319
    },
    {
      "epoch": 6.32467269897461e-05,
      "step": 66319,
      "training_step_time": 8.409536838531494
    },
    {
      "epoch": 6.32476806640625e-05,
      "grad_norm": 0.20001530647277832,
      "learning_rate": 2.7937730910373615e-06,
      "loss": 0.0618,
      "step": 66320
    },
    {
      "epoch": 6.32476806640625e-05,
      "model_forward_time": 1.3196747303009033,
      "step": 66320
    },
    {
      "dynamics_loss_avg": 0.06199129670858383,
      "epoch": 6.32476806640625e-05,
      "step": 66320
    },
    {
      "action_loss_avg": 0.0018381345144007355,
      "epoch": 6.32476806640625e-05,
      "step": 66320
    },
    {
      "epoch": 6.32476806640625e-05,
      "step": 66320,
      "training_step_time": 9.360387325286865
    },
    {
      "epoch": 6.32486343383789e-05,
      "model_forward_time": 1.7299902439117432,
      "step": 66321
    },
    {
      "epoch": 6.32486343383789e-05,
      "step": 66321,
      "training_step_time": 9.998110055923462
    },
    {
      "epoch": 6.324958801269532e-05,
      "model_forward_time": 1.717574119567871,
      "step": 66322
    },
    {
      "epoch": 6.324958801269532e-05,
      "step": 66322,
      "training_step_time": 7.778352499008179
    },
    {
      "epoch": 6.325054168701171e-05,
      "model_forward_time": 1.7221505641937256,
      "step": 66323
    },
    {
      "epoch": 6.325054168701171e-05,
      "step": 66323,
      "training_step_time": 8.826520919799805
    },
    {
      "epoch": 6.325149536132812e-05,
      "model_forward_time": 1.7106387615203857,
      "step": 66324
    },
    {
      "epoch": 6.325149536132812e-05,
      "step": 66324,
      "training_step_time": 7.821096658706665
    },
    {
      "epoch": 6.325244903564453e-05,
      "model_forward_time": 2.951500177383423,
      "step": 66325
    },
    {
      "epoch": 6.325244903564453e-05,
      "step": 66325,
      "training_step_time": 8.598131656646729
    },
    {
      "epoch": 6.325340270996093e-05,
      "model_forward_time": 1.7106859683990479,
      "step": 66326
    },
    {
      "epoch": 6.325340270996093e-05,
      "step": 66326,
      "training_step_time": 7.7434587478637695
    },
    {
      "epoch": 6.325435638427734e-05,
      "model_forward_time": 1.753678321838379,
      "step": 66327
    },
    {
      "epoch": 6.325435638427734e-05,
      "step": 66327,
      "training_step_time": 7.536463737487793
    },
    {
      "epoch": 6.325531005859375e-05,
      "model_forward_time": 1.801156997680664,
      "step": 66328
    },
    {
      "epoch": 6.325531005859375e-05,
      "step": 66328,
      "training_step_time": 8.494325876235962
    },
    {
      "epoch": 6.325626373291015e-05,
      "model_forward_time": 1.7101433277130127,
      "step": 66329
    },
    {
      "epoch": 6.325626373291015e-05,
      "step": 66329,
      "training_step_time": 7.859696865081787
    },
    {
      "epoch": 6.325721740722656e-05,
      "grad_norm": 0.12386945635080338,
      "learning_rate": 2.7922894102226628e-06,
      "loss": 0.0627,
      "step": 66330
    },
    {
      "epoch": 6.325721740722656e-05,
      "model_forward_time": 1.7127764225006104,
      "step": 66330
    },
    {
      "dynamics_loss_avg": 0.06073709120973945,
      "epoch": 6.325721740722656e-05,
      "step": 66330
    },
    {
      "action_loss_avg": 0.002791068563237786,
      "epoch": 6.325721740722656e-05,
      "step": 66330
    },
    {
      "epoch": 6.325721740722656e-05,
      "step": 66330,
      "training_step_time": 9.021947145462036
    },
    {
      "epoch": 6.325817108154297e-05,
      "model_forward_time": 1.717231273651123,
      "step": 66331
    },
    {
      "epoch": 6.325817108154297e-05,
      "step": 66331,
      "training_step_time": 7.804229021072388
    },
    {
      "epoch": 6.325912475585937e-05,
      "model_forward_time": 1.7150447368621826,
      "step": 66332
    },
    {
      "epoch": 6.325912475585937e-05,
      "step": 66332,
      "training_step_time": 8.717024326324463
    },
    {
      "epoch": 6.326007843017578e-05,
      "model_forward_time": 1.7258126735687256,
      "step": 66333
    },
    {
      "epoch": 6.326007843017578e-05,
      "step": 66333,
      "training_step_time": 7.731135845184326
    },
    {
      "epoch": 6.326103210449219e-05,
      "model_forward_time": 2.8797664642333984,
      "step": 66334
    },
    {
      "epoch": 6.326103210449219e-05,
      "step": 66334,
      "training_step_time": 7.793006420135498
    },
    {
      "epoch": 6.326198577880859e-05,
      "model_forward_time": 1.7136776447296143,
      "step": 66335
    },
    {
      "epoch": 6.326198577880859e-05,
      "step": 66335,
      "training_step_time": 7.741089105606079
    },
    {
      "epoch": 6.3262939453125e-05,
      "model_forward_time": 1.7689058780670166,
      "step": 66336
    },
    {
      "epoch": 6.3262939453125e-05,
      "step": 66336,
      "training_step_time": 7.5293731689453125
    },
    {
      "epoch": 6.326389312744141e-05,
      "model_forward_time": 1.7403545379638672,
      "step": 66337
    },
    {
      "epoch": 6.326389312744141e-05,
      "step": 66337,
      "training_step_time": 7.70856499671936
    },
    {
      "epoch": 6.32648468017578e-05,
      "model_forward_time": 1.7163617610931396,
      "step": 66338
    },
    {
      "epoch": 6.32648468017578e-05,
      "step": 66338,
      "training_step_time": 7.794464111328125
    },
    {
      "epoch": 6.326580047607422e-05,
      "model_forward_time": 1.7220561504364014,
      "step": 66339
    },
    {
      "epoch": 6.326580047607422e-05,
      "step": 66339,
      "training_step_time": 9.454282522201538
    },
    {
      "epoch": 6.326675415039063e-05,
      "grad_norm": 0.13092370331287384,
      "learning_rate": 2.7908059708398762e-06,
      "loss": 0.056,
      "step": 66340
    },
    {
      "epoch": 6.326675415039063e-05,
      "model_forward_time": 1.71872878074646,
      "step": 66340
    },
    {
      "dynamics_loss_avg": 0.04797105435281992,
      "epoch": 6.326675415039063e-05,
      "step": 66340
    },
    {
      "action_loss_avg": 0.0020683826878666878,
      "epoch": 6.326675415039063e-05,
      "step": 66340
    },
    {
      "epoch": 6.326675415039063e-05,
      "step": 66340,
      "training_step_time": 9.537265300750732
    },
    {
      "epoch": 6.326770782470703e-05,
      "model_forward_time": 1.7125046253204346,
      "step": 66341
    },
    {
      "epoch": 6.326770782470703e-05,
      "step": 66341,
      "training_step_time": 7.854207992553711
    },
    {
      "epoch": 6.326866149902344e-05,
      "model_forward_time": 1.7146570682525635,
      "step": 66342
    },
    {
      "epoch": 6.326866149902344e-05,
      "step": 66342,
      "training_step_time": 7.7289605140686035
    },
    {
      "epoch": 6.326961517333985e-05,
      "model_forward_time": 2.929471731185913,
      "step": 66343
    },
    {
      "epoch": 6.326961517333985e-05,
      "step": 66343,
      "training_step_time": 7.824781894683838
    },
    {
      "epoch": 6.327056884765624e-05,
      "model_forward_time": 1.3451552391052246,
      "step": 66344
    },
    {
      "epoch": 6.327056884765624e-05,
      "step": 66344,
      "training_step_time": 7.672020435333252
    },
    {
      "epoch": 6.327152252197266e-05,
      "model_forward_time": 1.7091648578643799,
      "step": 66345
    },
    {
      "epoch": 6.327152252197266e-05,
      "step": 66345,
      "training_step_time": 8.430314064025879
    },
    {
      "epoch": 6.327247619628907e-05,
      "model_forward_time": 1.7195212841033936,
      "step": 66346
    },
    {
      "epoch": 6.327247619628907e-05,
      "step": 66346,
      "training_step_time": 8.392833948135376
    },
    {
      "epoch": 6.327342987060546e-05,
      "model_forward_time": 1.338135004043579,
      "step": 66347
    },
    {
      "epoch": 6.327342987060546e-05,
      "step": 66347,
      "training_step_time": 8.826201915740967
    },
    {
      "epoch": 6.327438354492187e-05,
      "model_forward_time": 1.7329213619232178,
      "step": 66348
    },
    {
      "epoch": 6.327438354492187e-05,
      "step": 66348,
      "training_step_time": 8.743905544281006
    },
    {
      "epoch": 6.327533721923828e-05,
      "model_forward_time": 1.7113006114959717,
      "step": 66349
    },
    {
      "epoch": 6.327533721923828e-05,
      "step": 66349,
      "training_step_time": 7.817438125610352
    },
    {
      "epoch": 6.327629089355468e-05,
      "grad_norm": 0.09699097275733948,
      "learning_rate": 2.789322773051225e-06,
      "loss": 0.0593,
      "step": 66350
    },
    {
      "epoch": 6.327629089355468e-05,
      "model_forward_time": 1.7187778949737549,
      "step": 66350
    },
    {
      "dynamics_loss_avg": 0.061780227534472945,
      "epoch": 6.327629089355468e-05,
      "step": 66350
    },
    {
      "action_loss_avg": 0.001956391689600423,
      "epoch": 6.327629089355468e-05,
      "step": 66350
    },
    {
      "epoch": 6.327629089355468e-05,
      "step": 66350,
      "training_step_time": 8.538456916809082
    },
    {
      "epoch": 6.327724456787109e-05,
      "model_forward_time": 1.6975548267364502,
      "step": 66351
    },
    {
      "epoch": 6.327724456787109e-05,
      "step": 66351,
      "training_step_time": 7.792722940444946
    },
    {
      "epoch": 6.32781982421875e-05,
      "model_forward_time": 2.891296148300171,
      "step": 66352
    },
    {
      "epoch": 6.32781982421875e-05,
      "step": 66352,
      "training_step_time": 7.806617021560669
    },
    {
      "epoch": 6.32791519165039e-05,
      "model_forward_time": 1.7116525173187256,
      "step": 66353
    },
    {
      "epoch": 6.32791519165039e-05,
      "step": 66353,
      "training_step_time": 7.850647687911987
    },
    {
      "epoch": 6.328010559082031e-05,
      "model_forward_time": 1.708766222000122,
      "step": 66354
    },
    {
      "epoch": 6.328010559082031e-05,
      "step": 66354,
      "training_step_time": 7.553619146347046
    },
    {
      "epoch": 6.328105926513672e-05,
      "model_forward_time": 1.7051103115081787,
      "step": 66355
    },
    {
      "epoch": 6.328105926513672e-05,
      "step": 66355,
      "training_step_time": 8.431466817855835
    },
    {
      "epoch": 6.328201293945312e-05,
      "model_forward_time": 1.8127212524414062,
      "step": 66356
    },
    {
      "epoch": 6.328201293945312e-05,
      "step": 66356,
      "training_step_time": 7.823270559310913
    },
    {
      "epoch": 6.328296661376953e-05,
      "model_forward_time": 1.7039682865142822,
      "step": 66357
    },
    {
      "epoch": 6.328296661376953e-05,
      "step": 66357,
      "training_step_time": 9.926001787185669
    },
    {
      "epoch": 6.328392028808594e-05,
      "model_forward_time": 1.710150957107544,
      "step": 66358
    },
    {
      "epoch": 6.328392028808594e-05,
      "step": 66358,
      "training_step_time": 9.597033262252808
    },
    {
      "epoch": 6.328487396240234e-05,
      "model_forward_time": 1.699713945388794,
      "step": 66359
    },
    {
      "epoch": 6.328487396240234e-05,
      "step": 66359,
      "training_step_time": 7.7480857372283936
    },
    {
      "epoch": 6.328582763671875e-05,
      "grad_norm": 0.09330123662948608,
      "learning_rate": 2.7878398170189127e-06,
      "loss": 0.0561,
      "step": 66360
    },
    {
      "epoch": 6.328582763671875e-05,
      "model_forward_time": 1.7115607261657715,
      "step": 66360
    },
    {
      "dynamics_loss_avg": 0.07653746344149112,
      "epoch": 6.328582763671875e-05,
      "step": 66360
    },
    {
      "action_loss_avg": 0.004585823390516453,
      "epoch": 6.328582763671875e-05,
      "step": 66360
    },
    {
      "epoch": 6.328582763671875e-05,
      "step": 66360,
      "training_step_time": 7.67621111869812
    },
    {
      "epoch": 6.328678131103516e-05,
      "model_forward_time": 2.8441734313964844,
      "step": 66361
    },
    {
      "epoch": 6.328678131103516e-05,
      "step": 66361,
      "training_step_time": 9.593392610549927
    },
    {
      "epoch": 6.328773498535156e-05,
      "model_forward_time": 1.7156691551208496,
      "step": 66362
    },
    {
      "epoch": 6.328773498535156e-05,
      "step": 66362,
      "training_step_time": 7.7241082191467285
    },
    {
      "epoch": 6.328868865966797e-05,
      "model_forward_time": 1.316701889038086,
      "step": 66363
    },
    {
      "epoch": 6.328868865966797e-05,
      "step": 66363,
      "training_step_time": 7.9563374519348145
    },
    {
      "epoch": 6.328964233398438e-05,
      "model_forward_time": 1.7069289684295654,
      "step": 66364
    },
    {
      "epoch": 6.328964233398438e-05,
      "step": 66364,
      "training_step_time": 7.639217853546143
    },
    {
      "epoch": 6.329059600830078e-05,
      "model_forward_time": 1.811316967010498,
      "step": 66365
    },
    {
      "epoch": 6.329059600830078e-05,
      "step": 66365,
      "training_step_time": 7.7508111000061035
    },
    {
      "epoch": 6.329154968261719e-05,
      "model_forward_time": 1.7083194255828857,
      "step": 66366
    },
    {
      "epoch": 6.329154968261719e-05,
      "step": 66366,
      "training_step_time": 7.706331253051758
    },
    {
      "epoch": 6.32925033569336e-05,
      "model_forward_time": 1.7017557621002197,
      "step": 66367
    },
    {
      "epoch": 6.32925033569336e-05,
      "step": 66367,
      "training_step_time": 10.518980264663696
    },
    {
      "epoch": 6.329345703125e-05,
      "model_forward_time": 1.7094075679779053,
      "step": 66368
    },
    {
      "epoch": 6.329345703125e-05,
      "step": 66368,
      "training_step_time": 7.650658130645752
    },
    {
      "epoch": 6.32944107055664e-05,
      "model_forward_time": 1.7142930030822754,
      "step": 66369
    },
    {
      "epoch": 6.32944107055664e-05,
      "step": 66369,
      "training_step_time": 7.6891396045684814
    },
    {
      "epoch": 6.329536437988282e-05,
      "grad_norm": 0.07323101907968521,
      "learning_rate": 2.786357102905111e-06,
      "loss": 0.0583,
      "step": 66370
    },
    {
      "epoch": 6.329536437988282e-05,
      "model_forward_time": 1.7051002979278564,
      "step": 66370
    },
    {
      "dynamics_loss_avg": 0.056796715781092647,
      "epoch": 6.329536437988282e-05,
      "step": 66370
    },
    {
      "action_loss_avg": 0.0016757322184275836,
      "epoch": 6.329536437988282e-05,
      "step": 66370
    },
    {
      "epoch": 6.329536437988282e-05,
      "step": 66370,
      "training_step_time": 8.646953105926514
    },
    {
      "epoch": 6.329631805419921e-05,
      "model_forward_time": 1.7094190120697021,
      "step": 66371
    },
    {
      "epoch": 6.329631805419921e-05,
      "step": 66371,
      "training_step_time": 7.683465957641602
    },
    {
      "epoch": 6.329727172851562e-05,
      "model_forward_time": 1.7159314155578613,
      "step": 66372
    },
    {
      "epoch": 6.329727172851562e-05,
      "step": 66372,
      "training_step_time": 7.203453063964844
    },
    {
      "epoch": 6.329822540283204e-05,
      "model_forward_time": 1.7097711563110352,
      "step": 66373
    },
    {
      "epoch": 6.329822540283204e-05,
      "step": 66373,
      "training_step_time": 7.45277738571167
    },
    {
      "epoch": 6.329917907714843e-05,
      "model_forward_time": 1.7707738876342773,
      "step": 66374
    },
    {
      "epoch": 6.329917907714843e-05,
      "step": 66374,
      "training_step_time": 9.653284788131714
    },
    {
      "epoch": 6.330013275146484e-05,
      "model_forward_time": 1.0002191066741943,
      "step": 66375
    },
    {
      "epoch": 6.330013275146484e-05,
      "step": 66375,
      "training_step_time": 8.11918020248413
    },
    {
      "epoch": 6.330108642578125e-05,
      "model_forward_time": 1.7112081050872803,
      "step": 66376
    },
    {
      "epoch": 6.330108642578125e-05,
      "step": 66376,
      "training_step_time": 7.719081163406372
    },
    {
      "epoch": 6.330204010009765e-05,
      "model_forward_time": 1.7090866565704346,
      "step": 66377
    },
    {
      "epoch": 6.330204010009765e-05,
      "step": 66377,
      "training_step_time": 7.700797080993652
    },
    {
      "epoch": 6.330299377441406e-05,
      "model_forward_time": 0.9935176372528076,
      "step": 66378
    },
    {
      "epoch": 6.330299377441406e-05,
      "step": 66378,
      "training_step_time": 8.190776348114014
    },
    {
      "epoch": 6.330394744873047e-05,
      "model_forward_time": 1.6976466178894043,
      "step": 66379
    },
    {
      "epoch": 6.330394744873047e-05,
      "step": 66379,
      "training_step_time": 7.599257469177246
    },
    {
      "epoch": 6.330490112304687e-05,
      "grad_norm": 0.08316592127084732,
      "learning_rate": 2.784874630871971e-06,
      "loss": 0.0579,
      "step": 66380
    },
    {
      "epoch": 6.330490112304687e-05,
      "model_forward_time": 1.7091212272644043,
      "step": 66380
    },
    {
      "dynamics_loss_avg": 0.05665044374763965,
      "epoch": 6.330490112304687e-05,
      "step": 66380
    },
    {
      "action_loss_avg": 0.0011405462311813608,
      "epoch": 6.330490112304687e-05,
      "step": 66380
    },
    {
      "epoch": 6.330490112304687e-05,
      "step": 66380,
      "training_step_time": 8.678699731826782
    },
    {
      "epoch": 6.330585479736328e-05,
      "model_forward_time": 1.7195429801940918,
      "step": 66381
    },
    {
      "epoch": 6.330585479736328e-05,
      "step": 66381,
      "training_step_time": 7.586292028427124
    },
    {
      "epoch": 6.330680847167969e-05,
      "model_forward_time": 1.7139084339141846,
      "step": 66382
    },
    {
      "epoch": 6.330680847167969e-05,
      "step": 66382,
      "training_step_time": 7.490497827529907
    },
    {
      "epoch": 6.330776214599609e-05,
      "model_forward_time": 1.7650306224822998,
      "step": 66383
    },
    {
      "epoch": 6.330776214599609e-05,
      "step": 66383,
      "training_step_time": 9.099738359451294
    },
    {
      "epoch": 6.33087158203125e-05,
      "model_forward_time": 1.7099196910858154,
      "step": 66384
    },
    {
      "epoch": 6.33087158203125e-05,
      "step": 66384,
      "training_step_time": 8.184057474136353
    },
    {
      "epoch": 6.330966949462891e-05,
      "model_forward_time": 1.7189810276031494,
      "step": 66385
    },
    {
      "epoch": 6.330966949462891e-05,
      "step": 66385,
      "training_step_time": 7.735350847244263
    },
    {
      "epoch": 6.331062316894531e-05,
      "model_forward_time": 1.7216501235961914,
      "step": 66386
    },
    {
      "epoch": 6.331062316894531e-05,
      "step": 66386,
      "training_step_time": 8.30258560180664
    },
    {
      "epoch": 6.331157684326172e-05,
      "model_forward_time": 1.7106642723083496,
      "step": 66387
    },
    {
      "epoch": 6.331157684326172e-05,
      "step": 66387,
      "training_step_time": 8.88122844696045
    },
    {
      "epoch": 6.331253051757813e-05,
      "model_forward_time": 1.723250150680542,
      "step": 66388
    },
    {
      "epoch": 6.331253051757813e-05,
      "step": 66388,
      "training_step_time": 8.462302207946777
    },
    {
      "epoch": 6.331348419189453e-05,
      "model_forward_time": 1.7152135372161865,
      "step": 66389
    },
    {
      "epoch": 6.331348419189453e-05,
      "step": 66389,
      "training_step_time": 7.579426527023315
    },
    {
      "epoch": 6.331443786621094e-05,
      "grad_norm": 0.12858101725578308,
      "learning_rate": 2.783392401081609e-06,
      "loss": 0.0626,
      "step": 66390
    },
    {
      "epoch": 6.331443786621094e-05,
      "model_forward_time": 2.9627387523651123,
      "step": 66390
    },
    {
      "dynamics_loss_avg": 0.06475571766495705,
      "epoch": 6.331443786621094e-05,
      "step": 66390
    },
    {
      "action_loss_avg": 0.0021707698062527923,
      "epoch": 6.331443786621094e-05,
      "step": 66390
    },
    {
      "epoch": 6.331443786621094e-05,
      "step": 66390,
      "training_step_time": 7.709534406661987
    },
    {
      "epoch": 6.331539154052735e-05,
      "model_forward_time": 1.7893702983856201,
      "step": 66391
    },
    {
      "epoch": 6.331539154052735e-05,
      "step": 66391,
      "training_step_time": 7.469954967498779
    },
    {
      "epoch": 6.331634521484375e-05,
      "model_forward_time": 1.7198817729949951,
      "step": 66392
    },
    {
      "epoch": 6.331634521484375e-05,
      "step": 66392,
      "training_step_time": 8.709134340286255
    },
    {
      "epoch": 6.331729888916016e-05,
      "model_forward_time": 1.8222947120666504,
      "step": 66393
    },
    {
      "epoch": 6.331729888916016e-05,
      "step": 66393,
      "training_step_time": 7.57861328125
    },
    {
      "epoch": 6.331825256347657e-05,
      "model_forward_time": 2.253661632537842,
      "step": 66394
    },
    {
      "epoch": 6.331825256347657e-05,
      "step": 66394,
      "training_step_time": 11.631377696990967
    },
    {
      "epoch": 6.331920623779296e-05,
      "model_forward_time": 1.6985692977905273,
      "step": 66395
    },
    {
      "epoch": 6.331920623779296e-05,
      "step": 66395,
      "training_step_time": 6.725780725479126
    },
    {
      "epoch": 6.332015991210938e-05,
      "model_forward_time": 1.6986899375915527,
      "step": 66396
    },
    {
      "epoch": 6.332015991210938e-05,
      "step": 66396,
      "training_step_time": 6.710608720779419
    },
    {
      "epoch": 6.332111358642579e-05,
      "model_forward_time": 1.6952753067016602,
      "step": 66397
    },
    {
      "epoch": 6.332111358642579e-05,
      "step": 66397,
      "training_step_time": 6.647363662719727
    },
    {
      "epoch": 6.332206726074218e-05,
      "model_forward_time": 1.3256418704986572,
      "step": 66398
    },
    {
      "epoch": 6.332206726074218e-05,
      "step": 66398,
      "training_step_time": 6.69410514831543
    },
    {
      "epoch": 6.33230209350586e-05,
      "model_forward_time": 1.711515188217163,
      "step": 66399
    },
    {
      "epoch": 6.33230209350586e-05,
      "step": 66399,
      "training_step_time": 6.678813695907593
    },
    {
      "epoch": 6.3323974609375e-05,
      "grad_norm": 0.08214686065912247,
      "learning_rate": 2.781910413696123e-06,
      "loss": 0.0576,
      "step": 66400
    },
    {
      "epoch": 6.3323974609375e-05,
      "model_forward_time": 1.7177848815917969,
      "step": 66400
    },
    {
      "dynamics_loss_avg": 0.06177808586508036,
      "epoch": 6.3323974609375e-05,
      "step": 66400
    },
    {
      "action_loss_avg": 0.002141824143473059,
      "epoch": 6.3323974609375e-05,
      "step": 66400
    },
    {
      "epoch": 6.3323974609375e-05,
      "step": 66400,
      "training_step_time": 6.657132863998413
    },
    {
      "epoch": 6.33249282836914e-05,
      "model_forward_time": 1.713571548461914,
      "step": 66401
    },
    {
      "epoch": 6.33249282836914e-05,
      "step": 66401,
      "training_step_time": 6.665499210357666
    },
    {
      "epoch": 6.332588195800781e-05,
      "model_forward_time": 1.7126705646514893,
      "step": 66402
    },
    {
      "epoch": 6.332588195800781e-05,
      "step": 66402,
      "training_step_time": 6.71361517906189
    },
    {
      "epoch": 6.332683563232422e-05,
      "model_forward_time": 2.8692145347595215,
      "step": 66403
    },
    {
      "epoch": 6.332683563232422e-05,
      "step": 66403,
      "training_step_time": 7.615585803985596
    },
    {
      "epoch": 6.332778930664062e-05,
      "model_forward_time": 1.711226224899292,
      "step": 66404
    },
    {
      "epoch": 6.332778930664062e-05,
      "step": 66404,
      "training_step_time": 7.723630905151367
    },
    {
      "epoch": 6.332874298095703e-05,
      "model_forward_time": 1.7239556312561035,
      "step": 66405
    },
    {
      "epoch": 6.332874298095703e-05,
      "step": 66405,
      "training_step_time": 6.7330546379089355
    },
    {
      "epoch": 6.332969665527344e-05,
      "model_forward_time": 1.7107939720153809,
      "step": 66406
    },
    {
      "epoch": 6.332969665527344e-05,
      "step": 66406,
      "training_step_time": 6.6758034229278564
    },
    {
      "epoch": 6.333065032958984e-05,
      "model_forward_time": 1.7133607864379883,
      "step": 66407
    },
    {
      "epoch": 6.333065032958984e-05,
      "step": 66407,
      "training_step_time": 6.675909996032715
    },
    {
      "epoch": 6.333160400390625e-05,
      "model_forward_time": 1.7116591930389404,
      "step": 66408
    },
    {
      "epoch": 6.333160400390625e-05,
      "step": 66408,
      "training_step_time": 6.727186679840088
    },
    {
      "epoch": 6.333255767822266e-05,
      "model_forward_time": 1.7146961688995361,
      "step": 66409
    },
    {
      "epoch": 6.333255767822266e-05,
      "step": 66409,
      "training_step_time": 6.688082218170166
    },
    {
      "epoch": 6.333351135253906e-05,
      "grad_norm": 0.09453681856393814,
      "learning_rate": 2.7804286688775773e-06,
      "loss": 0.0577,
      "step": 66410
    },
    {
      "epoch": 6.333351135253906e-05,
      "model_forward_time": 1.7245798110961914,
      "step": 66410
    },
    {
      "dynamics_loss_avg": 0.06983677446842193,
      "epoch": 6.333351135253906e-05,
      "step": 66410
    },
    {
      "action_loss_avg": 0.0023843474802561103,
      "epoch": 6.333351135253906e-05,
      "step": 66410
    },
    {
      "epoch": 6.333351135253906e-05,
      "step": 66410,
      "training_step_time": 6.727970600128174
    },
    {
      "epoch": 6.333446502685547e-05,
      "model_forward_time": 1.721614122390747,
      "step": 66411
    },
    {
      "epoch": 6.333446502685547e-05,
      "step": 66411,
      "training_step_time": 6.752148866653442
    },
    {
      "epoch": 6.333541870117188e-05,
      "model_forward_time": 1.742682695388794,
      "step": 66412
    },
    {
      "epoch": 6.333541870117188e-05,
      "step": 66412,
      "training_step_time": 9.536880254745483
    },
    {
      "epoch": 6.333637237548828e-05,
      "model_forward_time": 1.7157371044158936,
      "step": 66413
    },
    {
      "epoch": 6.333637237548828e-05,
      "step": 66413,
      "training_step_time": 7.805123805999756
    },
    {
      "epoch": 6.333732604980469e-05,
      "model_forward_time": 1.7222700119018555,
      "step": 66414
    },
    {
      "epoch": 6.333732604980469e-05,
      "step": 66414,
      "training_step_time": 6.64053201675415
    },
    {
      "epoch": 6.33382797241211e-05,
      "model_forward_time": 1.7083024978637695,
      "step": 66415
    },
    {
      "epoch": 6.33382797241211e-05,
      "step": 66415,
      "training_step_time": 6.70648455619812
    },
    {
      "epoch": 6.33392333984375e-05,
      "model_forward_time": 1.7169983386993408,
      "step": 66416
    },
    {
      "epoch": 6.33392333984375e-05,
      "step": 66416,
      "training_step_time": 6.715308666229248
    },
    {
      "epoch": 6.334018707275391e-05,
      "model_forward_time": 1.6966559886932373,
      "step": 66417
    },
    {
      "epoch": 6.334018707275391e-05,
      "step": 66417,
      "training_step_time": 6.740359306335449
    },
    {
      "epoch": 6.334114074707032e-05,
      "model_forward_time": 1.702747106552124,
      "step": 66418
    },
    {
      "epoch": 6.334114074707032e-05,
      "step": 66418,
      "training_step_time": 6.722793340682983
    },
    {
      "epoch": 6.334209442138672e-05,
      "model_forward_time": 1.719362735748291,
      "step": 66419
    },
    {
      "epoch": 6.334209442138672e-05,
      "step": 66419,
      "training_step_time": 6.68169641494751
    },
    {
      "epoch": 6.334304809570313e-05,
      "grad_norm": 0.09920506924390793,
      "learning_rate": 2.778947166788015e-06,
      "loss": 0.059,
      "step": 66420
    },
    {
      "epoch": 6.334304809570313e-05,
      "model_forward_time": 1.7110466957092285,
      "step": 66420
    },
    {
      "dynamics_loss_avg": 0.07140905223786831,
      "epoch": 6.334304809570313e-05,
      "step": 66420
    },
    {
      "action_loss_avg": 0.0023063484695740044,
      "epoch": 6.334304809570313e-05,
      "step": 66420
    },
    {
      "epoch": 6.334304809570313e-05,
      "step": 66420,
      "training_step_time": 6.724280834197998
    },
    {
      "epoch": 6.334400177001954e-05,
      "model_forward_time": 1.7027766704559326,
      "step": 66421
    },
    {
      "epoch": 6.334400177001954e-05,
      "step": 66421,
      "training_step_time": 7.1712915897369385
    },
    {
      "epoch": 6.334495544433593e-05,
      "model_forward_time": 2.9306373596191406,
      "step": 66422
    },
    {
      "epoch": 6.334495544433593e-05,
      "step": 66422,
      "training_step_time": 7.781418561935425
    },
    {
      "epoch": 6.334590911865235e-05,
      "model_forward_time": 1.7088592052459717,
      "step": 66423
    },
    {
      "epoch": 6.334590911865235e-05,
      "step": 66423,
      "training_step_time": 7.593704462051392
    },
    {
      "epoch": 6.334686279296876e-05,
      "model_forward_time": 1.7153761386871338,
      "step": 66424
    },
    {
      "epoch": 6.334686279296876e-05,
      "step": 66424,
      "training_step_time": 6.762763738632202
    },
    {
      "epoch": 6.334781646728515e-05,
      "model_forward_time": 1.7026779651641846,
      "step": 66425
    },
    {
      "epoch": 6.334781646728515e-05,
      "step": 66425,
      "training_step_time": 6.717312574386597
    },
    {
      "epoch": 6.334877014160156e-05,
      "model_forward_time": 1.7027218341827393,
      "step": 66426
    },
    {
      "epoch": 6.334877014160156e-05,
      "step": 66426,
      "training_step_time": 6.729978799819946
    },
    {
      "epoch": 6.334972381591798e-05,
      "model_forward_time": 1.7049193382263184,
      "step": 66427
    },
    {
      "epoch": 6.334972381591798e-05,
      "step": 66427,
      "training_step_time": 6.722606658935547
    },
    {
      "epoch": 6.335067749023437e-05,
      "model_forward_time": 1.3220019340515137,
      "step": 66428
    },
    {
      "epoch": 6.335067749023437e-05,
      "step": 66428,
      "training_step_time": 6.733581066131592
    },
    {
      "epoch": 6.335163116455078e-05,
      "model_forward_time": 1.7167644500732422,
      "step": 66429
    },
    {
      "epoch": 6.335163116455078e-05,
      "step": 66429,
      "training_step_time": 6.689173460006714
    },
    {
      "epoch": 6.33525848388672e-05,
      "grad_norm": 0.11553309857845306,
      "learning_rate": 2.777465907589452e-06,
      "loss": 0.0575,
      "step": 66430
    },
    {
      "epoch": 6.33525848388672e-05,
      "model_forward_time": 1.7088792324066162,
      "step": 66430
    },
    {
      "dynamics_loss_avg": 0.06370872519910335,
      "epoch": 6.33525848388672e-05,
      "step": 66430
    },
    {
      "action_loss_avg": 0.002010935335420072,
      "epoch": 6.33525848388672e-05,
      "step": 66430
    },
    {
      "epoch": 6.33525848388672e-05,
      "step": 66430,
      "training_step_time": 6.717398166656494
    },
    {
      "epoch": 6.335353851318359e-05,
      "model_forward_time": 2.904977798461914,
      "step": 66431
    },
    {
      "epoch": 6.335353851318359e-05,
      "step": 66431,
      "training_step_time": 11.408840656280518
    },
    {
      "epoch": 6.33544921875e-05,
      "model_forward_time": 1.7332098484039307,
      "step": 66432
    },
    {
      "epoch": 6.33544921875e-05,
      "step": 66432,
      "training_step_time": 7.602703809738159
    },
    {
      "epoch": 6.335544586181641e-05,
      "model_forward_time": 1.7069478034973145,
      "step": 66433
    },
    {
      "epoch": 6.335544586181641e-05,
      "step": 66433,
      "training_step_time": 6.696678638458252
    },
    {
      "epoch": 6.335639953613281e-05,
      "model_forward_time": 1.7041656970977783,
      "step": 66434
    },
    {
      "epoch": 6.335639953613281e-05,
      "step": 66434,
      "training_step_time": 6.701551675796509
    },
    {
      "epoch": 6.335735321044922e-05,
      "model_forward_time": 1.6978411674499512,
      "step": 66435
    },
    {
      "epoch": 6.335735321044922e-05,
      "step": 66435,
      "training_step_time": 6.724853754043579
    },
    {
      "epoch": 6.335830688476563e-05,
      "model_forward_time": 1.7086150646209717,
      "step": 66436
    },
    {
      "epoch": 6.335830688476563e-05,
      "step": 66436,
      "training_step_time": 6.674203395843506
    },
    {
      "epoch": 6.335926055908203e-05,
      "model_forward_time": 1.709531545639038,
      "step": 66437
    },
    {
      "epoch": 6.335926055908203e-05,
      "step": 66437,
      "training_step_time": 6.6934263706207275
    },
    {
      "epoch": 6.336021423339844e-05,
      "model_forward_time": 1.7011682987213135,
      "step": 66438
    },
    {
      "epoch": 6.336021423339844e-05,
      "step": 66438,
      "training_step_time": 6.7115466594696045
    },
    {
      "epoch": 6.336116790771484e-05,
      "model_forward_time": 1.704437255859375,
      "step": 66439
    },
    {
      "epoch": 6.336116790771484e-05,
      "step": 66439,
      "training_step_time": 6.7235047817230225
    },
    {
      "epoch": 6.336212158203125e-05,
      "grad_norm": 0.09599985182285309,
      "learning_rate": 2.775984891443878e-06,
      "loss": 0.0568,
      "step": 66440
    },
    {
      "epoch": 6.336212158203125e-05,
      "model_forward_time": 2.829002618789673,
      "step": 66440
    },
    {
      "dynamics_loss_avg": 0.047296914551407096,
      "epoch": 6.336212158203125e-05,
      "step": 66440
    },
    {
      "action_loss_avg": 0.0009121319541009143,
      "epoch": 6.336212158203125e-05,
      "step": 66440
    },
    {
      "epoch": 6.336212158203125e-05,
      "step": 66440,
      "training_step_time": 10.690306425094604
    },
    {
      "epoch": 6.336307525634766e-05,
      "model_forward_time": 1.6974091529846191,
      "step": 66441
    },
    {
      "epoch": 6.336307525634766e-05,
      "step": 66441,
      "training_step_time": 7.698859214782715
    },
    {
      "epoch": 6.336402893066406e-05,
      "model_forward_time": 1.777341604232788,
      "step": 66442
    },
    {
      "epoch": 6.336402893066406e-05,
      "step": 66442,
      "training_step_time": 7.673917055130005
    },
    {
      "epoch": 6.336498260498047e-05,
      "model_forward_time": 1.7103557586669922,
      "step": 66443
    },
    {
      "epoch": 6.336498260498047e-05,
      "step": 66443,
      "training_step_time": 6.739578723907471
    },
    {
      "epoch": 6.336593627929688e-05,
      "model_forward_time": 1.7006916999816895,
      "step": 66444
    },
    {
      "epoch": 6.336593627929688e-05,
      "step": 66444,
      "training_step_time": 6.697686672210693
    },
    {
      "epoch": 6.336688995361327e-05,
      "model_forward_time": 1.6977660655975342,
      "step": 66445
    },
    {
      "epoch": 6.336688995361327e-05,
      "step": 66445,
      "training_step_time": 6.702155113220215
    },
    {
      "epoch": 6.336784362792969e-05,
      "model_forward_time": 1.6943027973175049,
      "step": 66446
    },
    {
      "epoch": 6.336784362792969e-05,
      "step": 66446,
      "training_step_time": 6.7617316246032715
    },
    {
      "epoch": 6.33687973022461e-05,
      "model_forward_time": 1.6929140090942383,
      "step": 66447
    },
    {
      "epoch": 6.33687973022461e-05,
      "step": 66447,
      "training_step_time": 6.703904151916504
    },
    {
      "epoch": 6.33697509765625e-05,
      "model_forward_time": 1.6996710300445557,
      "step": 66448
    },
    {
      "epoch": 6.33697509765625e-05,
      "step": 66448,
      "training_step_time": 6.715738773345947
    },
    {
      "epoch": 6.33707046508789e-05,
      "model_forward_time": 2.8258683681488037,
      "step": 66449
    },
    {
      "epoch": 6.33707046508789e-05,
      "step": 66449,
      "training_step_time": 7.740040302276611
    },
    {
      "epoch": 6.337165832519532e-05,
      "grad_norm": 0.07709290087223053,
      "learning_rate": 2.77450411851325e-06,
      "loss": 0.0574,
      "step": 66450
    },
    {
      "epoch": 6.337165832519532e-05,
      "model_forward_time": 1.8136286735534668,
      "step": 66450
    },
    {
      "dynamics_loss_avg": 0.04629450235515833,
      "epoch": 6.337165832519532e-05,
      "step": 66450
    },
    {
      "action_loss_avg": 0.0012769358116202056,
      "epoch": 6.337165832519532e-05,
      "step": 66450
    },
    {
      "epoch": 6.337165832519532e-05,
      "step": 66450,
      "training_step_time": 8.640976667404175
    },
    {
      "epoch": 6.337261199951171e-05,
      "model_forward_time": 1.689284324645996,
      "step": 66451
    },
    {
      "epoch": 6.337261199951171e-05,
      "step": 66451,
      "training_step_time": 7.596866130828857
    },
    {
      "epoch": 6.337356567382812e-05,
      "model_forward_time": 1.7132046222686768,
      "step": 66452
    },
    {
      "epoch": 6.337356567382812e-05,
      "step": 66452,
      "training_step_time": 6.696581840515137
    },
    {
      "epoch": 6.337451934814453e-05,
      "model_forward_time": 1.7047169208526611,
      "step": 66453
    },
    {
      "epoch": 6.337451934814453e-05,
      "step": 66453,
      "training_step_time": 6.714447736740112
    },
    {
      "epoch": 6.337547302246093e-05,
      "model_forward_time": 1.699575424194336,
      "step": 66454
    },
    {
      "epoch": 6.337547302246093e-05,
      "step": 66454,
      "training_step_time": 6.7084338665008545
    },
    {
      "epoch": 6.337642669677734e-05,
      "model_forward_time": 1.7047019004821777,
      "step": 66455
    },
    {
      "epoch": 6.337642669677734e-05,
      "step": 66455,
      "training_step_time": 6.704426050186157
    },
    {
      "epoch": 6.337738037109375e-05,
      "model_forward_time": 1.6983146667480469,
      "step": 66456
    },
    {
      "epoch": 6.337738037109375e-05,
      "step": 66456,
      "training_step_time": 6.725001335144043
    },
    {
      "epoch": 6.337833404541015e-05,
      "model_forward_time": 1.7010748386383057,
      "step": 66457
    },
    {
      "epoch": 6.337833404541015e-05,
      "step": 66457,
      "training_step_time": 6.716290473937988
    },
    {
      "epoch": 6.337928771972656e-05,
      "model_forward_time": 1.7145957946777344,
      "step": 66458
    },
    {
      "epoch": 6.337928771972656e-05,
      "step": 66458,
      "training_step_time": 7.640126466751099
    },
    {
      "epoch": 6.338024139404297e-05,
      "model_forward_time": 2.8497064113616943,
      "step": 66459
    },
    {
      "epoch": 6.338024139404297e-05,
      "step": 66459,
      "training_step_time": 10.277266502380371
    },
    {
      "epoch": 6.338119506835937e-05,
      "grad_norm": 0.0918455719947815,
      "learning_rate": 2.7730235889595058e-06,
      "loss": 0.0538,
      "step": 66460
    },
    {
      "epoch": 6.338119506835937e-05,
      "model_forward_time": 1.7089600563049316,
      "step": 66460
    },
    {
      "dynamics_loss_avg": 0.05304436385631561,
      "epoch": 6.338119506835937e-05,
      "step": 66460
    },
    {
      "action_loss_avg": 0.0014686395501485095,
      "epoch": 6.338119506835937e-05,
      "step": 66460
    },
    {
      "epoch": 6.338119506835937e-05,
      "step": 66460,
      "training_step_time": 7.5726354122161865
    },
    {
      "epoch": 6.338214874267578e-05,
      "model_forward_time": 1.7002909183502197,
      "step": 66461
    },
    {
      "epoch": 6.338214874267578e-05,
      "step": 66461,
      "training_step_time": 7.464280605316162
    },
    {
      "epoch": 6.338310241699219e-05,
      "model_forward_time": 1.7034270763397217,
      "step": 66462
    },
    {
      "epoch": 6.338310241699219e-05,
      "step": 66462,
      "training_step_time": 6.694969415664673
    },
    {
      "epoch": 6.338405609130859e-05,
      "model_forward_time": 1.7025575637817383,
      "step": 66463
    },
    {
      "epoch": 6.338405609130859e-05,
      "step": 66463,
      "training_step_time": 6.755698919296265
    },
    {
      "epoch": 6.3385009765625e-05,
      "model_forward_time": 1.7108485698699951,
      "step": 66464
    },
    {
      "epoch": 6.3385009765625e-05,
      "step": 66464,
      "training_step_time": 6.730366468429565
    },
    {
      "epoch": 6.338596343994141e-05,
      "model_forward_time": 1.7065248489379883,
      "step": 66465
    },
    {
      "epoch": 6.338596343994141e-05,
      "step": 66465,
      "training_step_time": 6.735415935516357
    },
    {
      "epoch": 6.33869171142578e-05,
      "model_forward_time": 1.7078170776367188,
      "step": 66466
    },
    {
      "epoch": 6.33869171142578e-05,
      "step": 66466,
      "training_step_time": 6.7287092208862305
    },
    {
      "epoch": 6.338787078857422e-05,
      "model_forward_time": 1.7012720108032227,
      "step": 66467
    },
    {
      "epoch": 6.338787078857422e-05,
      "step": 66467,
      "training_step_time": 7.635668516159058
    },
    {
      "epoch": 6.338882446289063e-05,
      "model_forward_time": 2.3143157958984375,
      "step": 66468
    },
    {
      "epoch": 6.338882446289063e-05,
      "step": 66468,
      "training_step_time": 10.182286262512207
    },
    {
      "epoch": 6.338977813720703e-05,
      "model_forward_time": 1.776393175125122,
      "step": 66469
    },
    {
      "epoch": 6.338977813720703e-05,
      "step": 66469,
      "training_step_time": 8.52864122390747
    },
    {
      "epoch": 6.339073181152344e-05,
      "grad_norm": 0.10601523518562317,
      "learning_rate": 2.7715433029445555e-06,
      "loss": 0.0542,
      "step": 66470
    },
    {
      "epoch": 6.339073181152344e-05,
      "model_forward_time": 1.7896151542663574,
      "step": 66470
    },
    {
      "dynamics_loss_avg": 0.07429456133395433,
      "epoch": 6.339073181152344e-05,
      "step": 66470
    },
    {
      "action_loss_avg": 0.002394057923811488,
      "epoch": 6.339073181152344e-05,
      "step": 66470
    },
    {
      "epoch": 6.339073181152344e-05,
      "step": 66470,
      "training_step_time": 7.614224433898926
    },
    {
      "epoch": 6.339168548583985e-05,
      "model_forward_time": 1.7456302642822266,
      "step": 66471
    },
    {
      "epoch": 6.339168548583985e-05,
      "step": 66471,
      "training_step_time": 6.723904371261597
    },
    {
      "epoch": 6.339263916015624e-05,
      "model_forward_time": 1.7836096286773682,
      "step": 66472
    },
    {
      "epoch": 6.339263916015624e-05,
      "step": 66472,
      "training_step_time": 6.707867383956909
    },
    {
      "epoch": 6.339359283447266e-05,
      "model_forward_time": 1.7646310329437256,
      "step": 66473
    },
    {
      "epoch": 6.339359283447266e-05,
      "step": 66473,
      "training_step_time": 6.729673147201538
    },
    {
      "epoch": 6.339454650878907e-05,
      "model_forward_time": 1.7787721157073975,
      "step": 66474
    },
    {
      "epoch": 6.339454650878907e-05,
      "step": 66474,
      "training_step_time": 6.682697534561157
    },
    {
      "epoch": 6.339550018310546e-05,
      "model_forward_time": 1.7747077941894531,
      "step": 66475
    },
    {
      "epoch": 6.339550018310546e-05,
      "step": 66475,
      "training_step_time": 6.685615539550781
    },
    {
      "epoch": 6.339645385742187e-05,
      "model_forward_time": 1.7769672870635986,
      "step": 66476
    },
    {
      "epoch": 6.339645385742187e-05,
      "step": 66476,
      "training_step_time": 7.488394737243652
    },
    {
      "epoch": 6.339740753173828e-05,
      "model_forward_time": 2.3350062370300293,
      "step": 66477
    },
    {
      "epoch": 6.339740753173828e-05,
      "step": 66477,
      "training_step_time": 10.11966609954834
    },
    {
      "epoch": 6.339836120605468e-05,
      "model_forward_time": 1.8619768619537354,
      "step": 66478
    },
    {
      "epoch": 6.339836120605468e-05,
      "step": 66478,
      "training_step_time": 7.764995098114014
    },
    {
      "epoch": 6.339931488037109e-05,
      "model_forward_time": 1.8739674091339111,
      "step": 66479
    },
    {
      "epoch": 6.339931488037109e-05,
      "step": 66479,
      "training_step_time": 7.5261759757995605
    },
    {
      "epoch": 6.34002685546875e-05,
      "grad_norm": 0.12342948466539383,
      "learning_rate": 2.770063260630277e-06,
      "loss": 0.0531,
      "step": 66480
    },
    {
      "epoch": 6.34002685546875e-05,
      "model_forward_time": 1.8497631549835205,
      "step": 66480
    },
    {
      "dynamics_loss_avg": 0.08200544621795416,
      "epoch": 6.34002685546875e-05,
      "step": 66480
    },
    {
      "action_loss_avg": 0.0032182104856474324,
      "epoch": 6.34002685546875e-05,
      "step": 66480
    },
    {
      "epoch": 6.34002685546875e-05,
      "step": 66480,
      "training_step_time": 6.6143951416015625
    },
    {
      "epoch": 6.34012222290039e-05,
      "model_forward_time": 1.8693931102752686,
      "step": 66481
    },
    {
      "epoch": 6.34012222290039e-05,
      "step": 66481,
      "training_step_time": 6.6167075634002686
    },
    {
      "epoch": 6.340217590332031e-05,
      "model_forward_time": 1.8887405395507812,
      "step": 66482
    },
    {
      "epoch": 6.340217590332031e-05,
      "step": 66482,
      "training_step_time": 6.612384796142578
    },
    {
      "epoch": 6.340312957763672e-05,
      "model_forward_time": 1.8473279476165771,
      "step": 66483
    },
    {
      "epoch": 6.340312957763672e-05,
      "step": 66483,
      "training_step_time": 6.597205400466919
    },
    {
      "epoch": 6.340408325195312e-05,
      "model_forward_time": 1.8290064334869385,
      "step": 66484
    },
    {
      "epoch": 6.340408325195312e-05,
      "step": 66484,
      "training_step_time": 6.643722295761108
    },
    {
      "epoch": 6.340503692626953e-05,
      "model_forward_time": 1.843825340270996,
      "step": 66485
    },
    {
      "epoch": 6.340503692626953e-05,
      "step": 66485,
      "training_step_time": 7.5579869747161865
    },
    {
      "epoch": 6.340599060058594e-05,
      "model_forward_time": 1.722707986831665,
      "step": 66486
    },
    {
      "epoch": 6.340599060058594e-05,
      "step": 66486,
      "training_step_time": 10.62233567237854
    },
    {
      "epoch": 6.340694427490234e-05,
      "model_forward_time": 1.8837995529174805,
      "step": 66487
    },
    {
      "epoch": 6.340694427490234e-05,
      "step": 66487,
      "training_step_time": 7.819787263870239
    },
    {
      "epoch": 6.340789794921875e-05,
      "model_forward_time": 1.868701696395874,
      "step": 66488
    },
    {
      "epoch": 6.340789794921875e-05,
      "step": 66488,
      "training_step_time": 7.5472869873046875
    },
    {
      "epoch": 6.340885162353516e-05,
      "model_forward_time": 1.8487443923950195,
      "step": 66489
    },
    {
      "epoch": 6.340885162353516e-05,
      "step": 66489,
      "training_step_time": 6.67127799987793
    },
    {
      "epoch": 6.340980529785156e-05,
      "grad_norm": 0.08977067470550537,
      "learning_rate": 2.768583462178528e-06,
      "loss": 0.0545,
      "step": 66490
    },
    {
      "epoch": 6.340980529785156e-05,
      "model_forward_time": 1.8435702323913574,
      "step": 66490
    },
    {
      "dynamics_loss_avg": 0.04585767760872841,
      "epoch": 6.340980529785156e-05,
      "step": 66490
    },
    {
      "action_loss_avg": 0.001545428071403876,
      "epoch": 6.340980529785156e-05,
      "step": 66490
    },
    {
      "epoch": 6.340980529785156e-05,
      "step": 66490,
      "training_step_time": 6.631101846694946
    },
    {
      "epoch": 6.341075897216797e-05,
      "model_forward_time": 1.8751766681671143,
      "step": 66491
    },
    {
      "epoch": 6.341075897216797e-05,
      "step": 66491,
      "training_step_time": 6.6236960887908936
    },
    {
      "epoch": 6.341171264648438e-05,
      "model_forward_time": 1.877467393875122,
      "step": 66492
    },
    {
      "epoch": 6.341171264648438e-05,
      "step": 66492,
      "training_step_time": 6.656260967254639
    },
    {
      "epoch": 6.341266632080078e-05,
      "model_forward_time": 1.8806049823760986,
      "step": 66493
    },
    {
      "epoch": 6.341266632080078e-05,
      "step": 66493,
      "training_step_time": 6.624309539794922
    },
    {
      "epoch": 6.341361999511719e-05,
      "model_forward_time": 1.8921236991882324,
      "step": 66494
    },
    {
      "epoch": 6.341361999511719e-05,
      "step": 66494,
      "training_step_time": 7.535285472869873
    },
    {
      "epoch": 6.34145736694336e-05,
      "model_forward_time": 1.8713710308074951,
      "step": 66495
    },
    {
      "epoch": 6.34145736694336e-05,
      "step": 66495,
      "training_step_time": 9.639091491699219
    },
    {
      "epoch": 6.341552734375e-05,
      "model_forward_time": 2.381680727005005,
      "step": 66496
    },
    {
      "epoch": 6.341552734375e-05,
      "step": 66496,
      "training_step_time": 7.697580099105835
    },
    {
      "epoch": 6.34164810180664e-05,
      "model_forward_time": 1.1580030918121338,
      "step": 66497
    },
    {
      "epoch": 6.34164810180664e-05,
      "step": 66497,
      "training_step_time": 7.535364389419556
    },
    {
      "epoch": 6.341743469238282e-05,
      "model_forward_time": 1.8852334022521973,
      "step": 66498
    },
    {
      "epoch": 6.341743469238282e-05,
      "step": 66498,
      "training_step_time": 7.412538051605225
    },
    {
      "epoch": 6.341838836669921e-05,
      "model_forward_time": 1.5399212837219238,
      "step": 66499
    },
    {
      "epoch": 6.341838836669921e-05,
      "step": 66499,
      "training_step_time": 6.5945210456848145
    },
    {
      "epoch": 6.341934204101562e-05,
      "grad_norm": 0.08433590829372406,
      "learning_rate": 2.767103907751136e-06,
      "loss": 0.0555,
      "step": 66500
    },
    {
      "epoch": 6.341934204101562e-05,
      "model_forward_time": 1.857010841369629,
      "step": 66500
    },
    {
      "dynamics_loss_avg": 0.05243666227906942,
      "epoch": 6.341934204101562e-05,
      "step": 66500
    },
    {
      "action_loss_avg": 0.0011642941157333554,
      "epoch": 6.341934204101562e-05,
      "step": 66500
    },
    {
      "epoch": 6.341934204101562e-05,
      "step": 66500,
      "training_step_time": 6.632972478866577
    },
    {
      "epoch": 6.342029571533204e-05,
      "model_forward_time": 1.8641197681427002,
      "step": 66501
    },
    {
      "epoch": 6.342029571533204e-05,
      "step": 66501,
      "training_step_time": 6.595934867858887
    },
    {
      "epoch": 6.342124938964843e-05,
      "model_forward_time": 1.848231554031372,
      "step": 66502
    },
    {
      "epoch": 6.342124938964843e-05,
      "step": 66502,
      "training_step_time": 6.601464033126831
    },
    {
      "epoch": 6.342220306396484e-05,
      "model_forward_time": 1.8351552486419678,
      "step": 66503
    },
    {
      "epoch": 6.342220306396484e-05,
      "step": 66503,
      "training_step_time": 7.476593255996704
    },
    {
      "epoch": 6.342315673828125e-05,
      "model_forward_time": 1.8653144836425781,
      "step": 66504
    },
    {
      "epoch": 6.342315673828125e-05,
      "step": 66504,
      "training_step_time": 8.60530972480774
    },
    {
      "epoch": 6.342411041259765e-05,
      "model_forward_time": 2.4345803260803223,
      "step": 66505
    },
    {
      "epoch": 6.342411041259765e-05,
      "step": 66505,
      "training_step_time": 8.238590240478516
    },
    {
      "epoch": 6.342506408691406e-05,
      "model_forward_time": 1.8673839569091797,
      "step": 66506
    },
    {
      "epoch": 6.342506408691406e-05,
      "step": 66506,
      "training_step_time": 7.670308589935303
    },
    {
      "epoch": 6.342601776123047e-05,
      "model_forward_time": 1.840700387954712,
      "step": 66507
    },
    {
      "epoch": 6.342601776123047e-05,
      "step": 66507,
      "training_step_time": 7.452632665634155
    },
    {
      "epoch": 6.342697143554687e-05,
      "model_forward_time": 1.4796137809753418,
      "step": 66508
    },
    {
      "epoch": 6.342697143554687e-05,
      "step": 66508,
      "training_step_time": 6.6355977058410645
    },
    {
      "epoch": 6.342792510986328e-05,
      "model_forward_time": 1.854490041732788,
      "step": 66509
    },
    {
      "epoch": 6.342792510986328e-05,
      "step": 66509,
      "training_step_time": 6.615070819854736
    },
    {
      "epoch": 6.342887878417969e-05,
      "grad_norm": 0.10942693054676056,
      "learning_rate": 2.7656245975099066e-06,
      "loss": 0.0539,
      "step": 66510
    },
    {
      "epoch": 6.342887878417969e-05,
      "model_forward_time": 1.8011107444763184,
      "step": 66510
    },
    {
      "dynamics_loss_avg": 0.056571141630411145,
      "epoch": 6.342887878417969e-05,
      "step": 66510
    },
    {
      "action_loss_avg": 0.0011619601107668131,
      "epoch": 6.342887878417969e-05,
      "step": 66510
    },
    {
      "epoch": 6.342887878417969e-05,
      "step": 66510,
      "training_step_time": 6.605987548828125
    },
    {
      "epoch": 6.342983245849609e-05,
      "model_forward_time": 1.8373675346374512,
      "step": 66511
    },
    {
      "epoch": 6.342983245849609e-05,
      "step": 66511,
      "training_step_time": 6.647072792053223
    },
    {
      "epoch": 6.34307861328125e-05,
      "model_forward_time": 1.8691868782043457,
      "step": 66512
    },
    {
      "epoch": 6.34307861328125e-05,
      "step": 66512,
      "training_step_time": 7.525698900222778
    },
    {
      "epoch": 6.343173980712891e-05,
      "model_forward_time": 1.842433214187622,
      "step": 66513
    },
    {
      "epoch": 6.343173980712891e-05,
      "step": 66513,
      "training_step_time": 8.432759523391724
    },
    {
      "epoch": 6.343269348144531e-05,
      "model_forward_time": 1.8205640316009521,
      "step": 66514
    },
    {
      "epoch": 6.343269348144531e-05,
      "step": 66514,
      "training_step_time": 8.84498119354248
    },
    {
      "epoch": 6.343364715576172e-05,
      "model_forward_time": 2.3997960090637207,
      "step": 66515
    },
    {
      "epoch": 6.343364715576172e-05,
      "step": 66515,
      "training_step_time": 7.814189672470093
    },
    {
      "epoch": 6.343460083007813e-05,
      "model_forward_time": 1.8361804485321045,
      "step": 66516
    },
    {
      "epoch": 6.343460083007813e-05,
      "step": 66516,
      "training_step_time": 7.550528049468994
    },
    {
      "epoch": 6.343555450439453e-05,
      "model_forward_time": 1.863739013671875,
      "step": 66517
    },
    {
      "epoch": 6.343555450439453e-05,
      "step": 66517,
      "training_step_time": 6.667089223861694
    },
    {
      "epoch": 6.343650817871094e-05,
      "model_forward_time": 1.8617379665374756,
      "step": 66518
    },
    {
      "epoch": 6.343650817871094e-05,
      "step": 66518,
      "training_step_time": 6.662110090255737
    },
    {
      "epoch": 6.343746185302735e-05,
      "model_forward_time": 1.8607151508331299,
      "step": 66519
    },
    {
      "epoch": 6.343746185302735e-05,
      "step": 66519,
      "training_step_time": 6.656812906265259
    },
    {
      "epoch": 6.343841552734375e-05,
      "grad_norm": 0.09472060203552246,
      "learning_rate": 2.7641455316166095e-06,
      "loss": 0.0542,
      "step": 66520
    },
    {
      "epoch": 6.343841552734375e-05,
      "model_forward_time": 1.8585875034332275,
      "step": 66520
    },
    {
      "dynamics_loss_avg": 0.0529376070946455,
      "epoch": 6.343841552734375e-05,
      "step": 66520
    },
    {
      "action_loss_avg": 0.0012874129926785827,
      "epoch": 6.343841552734375e-05,
      "step": 66520
    },
    {
      "epoch": 6.343841552734375e-05,
      "step": 66520,
      "training_step_time": 6.622803688049316
    },
    {
      "epoch": 6.343936920166016e-05,
      "model_forward_time": 1.8511912822723389,
      "step": 66521
    },
    {
      "epoch": 6.343936920166016e-05,
      "step": 66521,
      "training_step_time": 6.624540328979492
    },
    {
      "epoch": 6.344032287597657e-05,
      "model_forward_time": 1.8432424068450928,
      "step": 66522
    },
    {
      "epoch": 6.344032287597657e-05,
      "step": 66522,
      "training_step_time": 9.247725248336792
    },
    {
      "epoch": 6.344127655029297e-05,
      "model_forward_time": 1.8427200317382812,
      "step": 66523
    },
    {
      "epoch": 6.344127655029297e-05,
      "step": 66523,
      "training_step_time": 12.21778678894043
    },
    {
      "epoch": 6.344223022460938e-05,
      "model_forward_time": 2.396101713180542,
      "step": 66524
    },
    {
      "epoch": 6.344223022460938e-05,
      "step": 66524,
      "training_step_time": 7.720027685165405
    },
    {
      "epoch": 6.344318389892579e-05,
      "model_forward_time": 1.8519868850708008,
      "step": 66525
    },
    {
      "epoch": 6.344318389892579e-05,
      "step": 66525,
      "training_step_time": 7.57427453994751
    },
    {
      "epoch": 6.344413757324218e-05,
      "model_forward_time": 1.8762047290802002,
      "step": 66526
    },
    {
      "epoch": 6.344413757324218e-05,
      "step": 66526,
      "training_step_time": 7.4767234325408936
    },
    {
      "epoch": 6.34450912475586e-05,
      "model_forward_time": 1.882798194885254,
      "step": 66527
    },
    {
      "epoch": 6.34450912475586e-05,
      "step": 66527,
      "training_step_time": 6.625324010848999
    },
    {
      "epoch": 6.3446044921875e-05,
      "model_forward_time": 1.8824849128723145,
      "step": 66528
    },
    {
      "epoch": 6.3446044921875e-05,
      "step": 66528,
      "training_step_time": 6.614654302597046
    },
    {
      "epoch": 6.34469985961914e-05,
      "model_forward_time": 1.846621036529541,
      "step": 66529
    },
    {
      "epoch": 6.34469985961914e-05,
      "step": 66529,
      "training_step_time": 6.6515278816223145
    },
    {
      "epoch": 6.344795227050781e-05,
      "grad_norm": 0.09783270955085754,
      "learning_rate": 2.7626667102329976e-06,
      "loss": 0.0543,
      "step": 66530
    },
    {
      "epoch": 6.344795227050781e-05,
      "model_forward_time": 1.8521792888641357,
      "step": 66530
    },
    {
      "dynamics_loss_avg": 0.06638428904116153,
      "epoch": 6.344795227050781e-05,
      "step": 66530
    },
    {
      "action_loss_avg": 0.0018565427628345788,
      "epoch": 6.344795227050781e-05,
      "step": 66530
    },
    {
      "epoch": 6.344795227050781e-05,
      "step": 66530,
      "training_step_time": 6.653278350830078
    },
    {
      "epoch": 6.344890594482422e-05,
      "model_forward_time": 1.852874755859375,
      "step": 66531
    },
    {
      "epoch": 6.344890594482422e-05,
      "step": 66531,
      "training_step_time": 7.57233190536499
    },
    {
      "epoch": 6.344985961914062e-05,
      "model_forward_time": 1.8612473011016846,
      "step": 66532
    },
    {
      "epoch": 6.344985961914062e-05,
      "step": 66532,
      "training_step_time": 10.358933687210083
    },
    {
      "epoch": 6.345081329345703e-05,
      "model_forward_time": 2.3729910850524902,
      "step": 66533
    },
    {
      "epoch": 6.345081329345703e-05,
      "step": 66533,
      "training_step_time": 7.755019664764404
    },
    {
      "epoch": 6.345176696777344e-05,
      "model_forward_time": 1.8901934623718262,
      "step": 66534
    },
    {
      "epoch": 6.345176696777344e-05,
      "step": 66534,
      "training_step_time": 7.59503698348999
    },
    {
      "epoch": 6.345272064208984e-05,
      "model_forward_time": 1.869128942489624,
      "step": 66535
    },
    {
      "epoch": 6.345272064208984e-05,
      "step": 66535,
      "training_step_time": 7.4892051219940186
    },
    {
      "epoch": 6.345367431640625e-05,
      "model_forward_time": 1.8335492610931396,
      "step": 66536
    },
    {
      "epoch": 6.345367431640625e-05,
      "step": 66536,
      "training_step_time": 6.622730731964111
    },
    {
      "epoch": 6.345462799072266e-05,
      "model_forward_time": 1.8591043949127197,
      "step": 66537
    },
    {
      "epoch": 6.345462799072266e-05,
      "step": 66537,
      "training_step_time": 6.6626527309417725
    },
    {
      "epoch": 6.345558166503906e-05,
      "model_forward_time": 1.8687922954559326,
      "step": 66538
    },
    {
      "epoch": 6.345558166503906e-05,
      "step": 66538,
      "training_step_time": 6.638201475143433
    },
    {
      "epoch": 6.345653533935547e-05,
      "model_forward_time": 1.8539137840270996,
      "step": 66539
    },
    {
      "epoch": 6.345653533935547e-05,
      "step": 66539,
      "training_step_time": 6.676973819732666
    },
    {
      "epoch": 6.345748901367188e-05,
      "grad_norm": 0.09265057742595673,
      "learning_rate": 2.761188133520788e-06,
      "loss": 0.057,
      "step": 66540
    },
    {
      "epoch": 6.345748901367188e-05,
      "model_forward_time": 1.851485013961792,
      "step": 66540
    },
    {
      "dynamics_loss_avg": 0.060599908232688904,
      "epoch": 6.345748901367188e-05,
      "step": 66540
    },
    {
      "action_loss_avg": 0.0009812676609726623,
      "epoch": 6.345748901367188e-05,
      "step": 66540
    },
    {
      "epoch": 6.345748901367188e-05,
      "step": 66540,
      "training_step_time": 7.539753198623657
    },
    {
      "epoch": 6.345844268798828e-05,
      "model_forward_time": 1.844017744064331,
      "step": 66541
    },
    {
      "epoch": 6.345844268798828e-05,
      "step": 66541,
      "training_step_time": 10.188135147094727
    },
    {
      "epoch": 6.345939636230469e-05,
      "model_forward_time": 2.4182450771331787,
      "step": 66542
    },
    {
      "epoch": 6.345939636230469e-05,
      "step": 66542,
      "training_step_time": 8.721569061279297
    },
    {
      "epoch": 6.34603500366211e-05,
      "model_forward_time": 1.8559458255767822,
      "step": 66543
    },
    {
      "epoch": 6.34603500366211e-05,
      "step": 66543,
      "training_step_time": 7.674393177032471
    },
    {
      "epoch": 6.34613037109375e-05,
      "model_forward_time": 1.8275091648101807,
      "step": 66544
    },
    {
      "epoch": 6.34613037109375e-05,
      "step": 66544,
      "training_step_time": 7.4929821491241455
    },
    {
      "epoch": 6.346225738525391e-05,
      "model_forward_time": 1.5011391639709473,
      "step": 66545
    },
    {
      "epoch": 6.346225738525391e-05,
      "step": 66545,
      "training_step_time": 6.7157816886901855
    },
    {
      "epoch": 6.346321105957032e-05,
      "model_forward_time": 1.8518731594085693,
      "step": 66546
    },
    {
      "epoch": 6.346321105957032e-05,
      "step": 66546,
      "training_step_time": 6.704533100128174
    },
    {
      "epoch": 6.346416473388672e-05,
      "model_forward_time": 1.8371260166168213,
      "step": 66547
    },
    {
      "epoch": 6.346416473388672e-05,
      "step": 66547,
      "training_step_time": 6.716944217681885
    },
    {
      "epoch": 6.346511840820313e-05,
      "model_forward_time": 1.8890490531921387,
      "step": 66548
    },
    {
      "epoch": 6.346511840820313e-05,
      "step": 66548,
      "training_step_time": 6.781311988830566
    },
    {
      "epoch": 6.346607208251954e-05,
      "model_forward_time": 1.8751707077026367,
      "step": 66549
    },
    {
      "epoch": 6.346607208251954e-05,
      "step": 66549,
      "training_step_time": 7.590729236602783
    },
    {
      "epoch": 6.346702575683593e-05,
      "grad_norm": 0.10666768252849579,
      "learning_rate": 2.7597098016416786e-06,
      "loss": 0.0561,
      "step": 66550
    },
    {
      "epoch": 6.346702575683593e-05,
      "model_forward_time": 1.8766956329345703,
      "step": 66550
    },
    {
      "dynamics_loss_avg": 0.04072264842689037,
      "epoch": 6.346702575683593e-05,
      "step": 66550
    },
    {
      "action_loss_avg": 0.0012602191884070635,
      "epoch": 6.346702575683593e-05,
      "step": 66550
    },
    {
      "epoch": 6.346702575683593e-05,
      "step": 66550,
      "training_step_time": 8.733038425445557
    },
    {
      "epoch": 6.346797943115235e-05,
      "model_forward_time": 2.381812572479248,
      "step": 66551
    },
    {
      "epoch": 6.346797943115235e-05,
      "step": 66551,
      "training_step_time": 9.845987319946289
    },
    {
      "epoch": 6.346893310546876e-05,
      "model_forward_time": 1.8548355102539062,
      "step": 66552
    },
    {
      "epoch": 6.346893310546876e-05,
      "step": 66552,
      "training_step_time": 7.783132076263428
    },
    {
      "epoch": 6.346988677978515e-05,
      "model_forward_time": 1.1689603328704834,
      "step": 66553
    },
    {
      "epoch": 6.346988677978515e-05,
      "step": 66553,
      "training_step_time": 7.202873468399048
    },
    {
      "epoch": 6.347084045410156e-05,
      "model_forward_time": 1.8248019218444824,
      "step": 66554
    },
    {
      "epoch": 6.347084045410156e-05,
      "step": 66554,
      "training_step_time": 7.4891276359558105
    },
    {
      "epoch": 6.347179412841798e-05,
      "model_forward_time": 1.8485784530639648,
      "step": 66555
    },
    {
      "epoch": 6.347179412841798e-05,
      "step": 66555,
      "training_step_time": 6.69229793548584
    },
    {
      "epoch": 6.347274780273437e-05,
      "model_forward_time": 1.8705792427062988,
      "step": 66556
    },
    {
      "epoch": 6.347274780273437e-05,
      "step": 66556,
      "training_step_time": 6.66405725479126
    },
    {
      "epoch": 6.347370147705078e-05,
      "model_forward_time": 1.8315021991729736,
      "step": 66557
    },
    {
      "epoch": 6.347370147705078e-05,
      "step": 66557,
      "training_step_time": 6.65499472618103
    },
    {
      "epoch": 6.34746551513672e-05,
      "model_forward_time": 1.8390705585479736,
      "step": 66558
    },
    {
      "epoch": 6.34746551513672e-05,
      "step": 66558,
      "training_step_time": 7.5304481983184814
    },
    {
      "epoch": 6.347560882568359e-05,
      "model_forward_time": 1.8686707019805908,
      "step": 66559
    },
    {
      "epoch": 6.347560882568359e-05,
      "step": 66559,
      "training_step_time": 8.886314868927002
    },
    {
      "epoch": 6.34765625e-05,
      "grad_norm": 0.09615937620401382,
      "learning_rate": 2.7582317147573366e-06,
      "loss": 0.0577,
      "step": 66560
    },
    {
      "epoch": 6.34765625e-05,
      "model_forward_time": 1.8456413745880127,
      "step": 66560
    },
    {
      "dynamics_loss_avg": 0.057813248969614504,
      "epoch": 6.34765625e-05,
      "step": 66560
    },
    {
      "action_loss_avg": 0.001388544379733503,
      "epoch": 6.34765625e-05,
      "step": 66560
    },
    {
      "epoch": 6.34765625e-05,
      "step": 66560,
      "training_step_time": 9.816756248474121
    },
    {
      "epoch": 6.347751617431641e-05,
      "model_forward_time": 2.380397081375122,
      "step": 66561
    },
    {
      "epoch": 6.347751617431641e-05,
      "step": 66561,
      "training_step_time": 8.455953121185303
    },
    {
      "epoch": 6.347846984863281e-05,
      "model_forward_time": 1.8400757312774658,
      "step": 66562
    },
    {
      "epoch": 6.347846984863281e-05,
      "step": 66562,
      "training_step_time": 7.527072429656982
    },
    {
      "epoch": 6.347942352294922e-05,
      "model_forward_time": 1.8527674674987793,
      "step": 66563
    },
    {
      "epoch": 6.347942352294922e-05,
      "step": 66563,
      "training_step_time": 7.465627670288086
    },
    {
      "epoch": 6.348037719726562e-05,
      "model_forward_time": 1.8594937324523926,
      "step": 66564
    },
    {
      "epoch": 6.348037719726562e-05,
      "step": 66564,
      "training_step_time": 6.727830171585083
    },
    {
      "epoch": 6.348133087158203e-05,
      "model_forward_time": 1.8773548603057861,
      "step": 66565
    },
    {
      "epoch": 6.348133087158203e-05,
      "step": 66565,
      "training_step_time": 6.683627605438232
    },
    {
      "epoch": 6.348228454589844e-05,
      "model_forward_time": 1.8456594944000244,
      "step": 66566
    },
    {
      "epoch": 6.348228454589844e-05,
      "step": 66566,
      "training_step_time": 6.68597149848938
    },
    {
      "epoch": 6.348323822021484e-05,
      "model_forward_time": 1.8375985622406006,
      "step": 66567
    },
    {
      "epoch": 6.348323822021484e-05,
      "step": 66567,
      "training_step_time": 7.274105072021484
    },
    {
      "epoch": 6.348419189453125e-05,
      "model_forward_time": 1.8734676837921143,
      "step": 66568
    },
    {
      "epoch": 6.348419189453125e-05,
      "step": 66568,
      "training_step_time": 7.909759998321533
    },
    {
      "epoch": 6.348514556884766e-05,
      "model_forward_time": 1.850555419921875,
      "step": 66569
    },
    {
      "epoch": 6.348514556884766e-05,
      "step": 66569,
      "training_step_time": 8.871238946914673
    },
    {
      "epoch": 6.348609924316406e-05,
      "grad_norm": 0.08857206255197525,
      "learning_rate": 2.7567538730294063e-06,
      "loss": 0.0582,
      "step": 66570
    },
    {
      "epoch": 6.348609924316406e-05,
      "model_forward_time": 2.441239833831787,
      "step": 66570
    },
    {
      "dynamics_loss_avg": 0.06789345908910036,
      "epoch": 6.348609924316406e-05,
      "step": 66570
    },
    {
      "action_loss_avg": 0.002370515698567033,
      "epoch": 6.348609924316406e-05,
      "step": 66570
    },
    {
      "epoch": 6.348609924316406e-05,
      "step": 66570,
      "training_step_time": 10.471707582473755
    },
    {
      "epoch": 6.348705291748047e-05,
      "model_forward_time": 1.8459532260894775,
      "step": 66571
    },
    {
      "epoch": 6.348705291748047e-05,
      "step": 66571,
      "training_step_time": 7.938246488571167
    },
    {
      "epoch": 6.348800659179688e-05,
      "model_forward_time": 1.852081060409546,
      "step": 66572
    },
    {
      "epoch": 6.348800659179688e-05,
      "step": 66572,
      "training_step_time": 7.69684910774231
    },
    {
      "epoch": 6.348896026611327e-05,
      "model_forward_time": 1.8607556819915771,
      "step": 66573
    },
    {
      "epoch": 6.348896026611327e-05,
      "step": 66573,
      "training_step_time": 6.792266845703125
    },
    {
      "epoch": 6.348991394042969e-05,
      "model_forward_time": 1.863379955291748,
      "step": 66574
    },
    {
      "epoch": 6.348991394042969e-05,
      "step": 66574,
      "training_step_time": 6.843792200088501
    },
    {
      "epoch": 6.34908676147461e-05,
      "model_forward_time": 1.8365957736968994,
      "step": 66575
    },
    {
      "epoch": 6.34908676147461e-05,
      "step": 66575,
      "training_step_time": 6.839720010757446
    },
    {
      "epoch": 6.34918212890625e-05,
      "model_forward_time": 1.4989047050476074,
      "step": 66576
    },
    {
      "epoch": 6.34918212890625e-05,
      "step": 66576,
      "training_step_time": 6.739389896392822
    },
    {
      "epoch": 6.34927749633789e-05,
      "model_forward_time": 1.863860845565796,
      "step": 66577
    },
    {
      "epoch": 6.34927749633789e-05,
      "step": 66577,
      "training_step_time": 7.7975523471832275
    },
    {
      "epoch": 6.349372863769532e-05,
      "model_forward_time": 1.8548004627227783,
      "step": 66578
    },
    {
      "epoch": 6.349372863769532e-05,
      "step": 66578,
      "training_step_time": 9.586917161941528
    },
    {
      "epoch": 6.349468231201171e-05,
      "model_forward_time": 1.1850082874298096,
      "step": 66579
    },
    {
      "epoch": 6.349468231201171e-05,
      "step": 66579,
      "training_step_time": 10.2588791847229
    },
    {
      "epoch": 6.349563598632812e-05,
      "grad_norm": 0.13499726355075836,
      "learning_rate": 2.7552762766194975e-06,
      "loss": 0.0553,
      "step": 66580
    },
    {
      "epoch": 6.349563598632812e-05,
      "model_forward_time": 2.3604323863983154,
      "step": 66580
    },
    {
      "dynamics_loss_avg": 0.03758228542283178,
      "epoch": 6.349563598632812e-05,
      "step": 66580
    },
    {
      "action_loss_avg": 0.0012275014218175785,
      "epoch": 6.349563598632812e-05,
      "step": 66580
    },
    {
      "epoch": 6.349563598632812e-05,
      "step": 66580,
      "training_step_time": 7.834665060043335
    },
    {
      "epoch": 6.349658966064453e-05,
      "model_forward_time": 1.5332427024841309,
      "step": 66581
    },
    {
      "epoch": 6.349658966064453e-05,
      "step": 66581,
      "training_step_time": 8.370946407318115
    },
    {
      "epoch": 6.349754333496093e-05,
      "model_forward_time": 1.8555176258087158,
      "step": 66582
    },
    {
      "epoch": 6.349754333496093e-05,
      "step": 66582,
      "training_step_time": 6.91127347946167
    },
    {
      "epoch": 6.349849700927734e-05,
      "model_forward_time": 1.8396775722503662,
      "step": 66583
    },
    {
      "epoch": 6.349849700927734e-05,
      "step": 66583,
      "training_step_time": 6.896975755691528
    },
    {
      "epoch": 6.349945068359375e-05,
      "model_forward_time": 1.8620707988739014,
      "step": 66584
    },
    {
      "epoch": 6.349945068359375e-05,
      "step": 66584,
      "training_step_time": 6.8387556076049805
    },
    {
      "epoch": 6.350040435791015e-05,
      "model_forward_time": 1.8426518440246582,
      "step": 66585
    },
    {
      "epoch": 6.350040435791015e-05,
      "step": 66585,
      "training_step_time": 6.782900810241699
    },
    {
      "epoch": 6.350135803222656e-05,
      "model_forward_time": 1.877511978149414,
      "step": 66586
    },
    {
      "epoch": 6.350135803222656e-05,
      "step": 66586,
      "training_step_time": 7.8260416984558105
    },
    {
      "epoch": 6.350231170654297e-05,
      "model_forward_time": 1.8379054069519043,
      "step": 66587
    },
    {
      "epoch": 6.350231170654297e-05,
      "step": 66587,
      "training_step_time": 8.289723634719849
    },
    {
      "epoch": 6.350326538085937e-05,
      "model_forward_time": 1.8699216842651367,
      "step": 66588
    },
    {
      "epoch": 6.350326538085937e-05,
      "step": 66588,
      "training_step_time": 10.219832181930542
    },
    {
      "epoch": 6.350421905517578e-05,
      "model_forward_time": 2.4005863666534424,
      "step": 66589
    },
    {
      "epoch": 6.350421905517578e-05,
      "step": 66589,
      "training_step_time": 7.881359577178955
    },
    {
      "epoch": 6.350517272949219e-05,
      "grad_norm": 0.09627602249383926,
      "learning_rate": 2.7537989256892024e-06,
      "loss": 0.0597,
      "step": 66590
    },
    {
      "epoch": 6.350517272949219e-05,
      "model_forward_time": 1.8736698627471924,
      "step": 66590
    },
    {
      "dynamics_loss_avg": 0.05581378396600485,
      "epoch": 6.350517272949219e-05,
      "step": 66590
    },
    {
      "action_loss_avg": 0.0020830130903050303,
      "epoch": 6.350517272949219e-05,
      "step": 66590
    },
    {
      "epoch": 6.350517272949219e-05,
      "step": 66590,
      "training_step_time": 7.336771249771118
    },
    {
      "epoch": 6.350612640380859e-05,
      "model_forward_time": 1.8546533584594727,
      "step": 66591
    },
    {
      "epoch": 6.350612640380859e-05,
      "step": 66591,
      "training_step_time": 7.503602743148804
    },
    {
      "epoch": 6.3507080078125e-05,
      "model_forward_time": 1.449232816696167,
      "step": 66592
    },
    {
      "epoch": 6.3507080078125e-05,
      "step": 66592,
      "training_step_time": 6.620653390884399
    },
    {
      "epoch": 6.350803375244141e-05,
      "model_forward_time": 1.8566439151763916,
      "step": 66593
    },
    {
      "epoch": 6.350803375244141e-05,
      "step": 66593,
      "training_step_time": 6.709147691726685
    },
    {
      "epoch": 6.350898742675781e-05,
      "model_forward_time": 1.8612360954284668,
      "step": 66594
    },
    {
      "epoch": 6.350898742675781e-05,
      "step": 66594,
      "training_step_time": 6.699327707290649
    },
    {
      "epoch": 6.350994110107422e-05,
      "model_forward_time": 1.881659746170044,
      "step": 66595
    },
    {
      "epoch": 6.350994110107422e-05,
      "step": 66595,
      "training_step_time": 9.518547296524048
    },
    {
      "epoch": 6.351089477539063e-05,
      "model_forward_time": 1.8451974391937256,
      "step": 66596
    },
    {
      "epoch": 6.351089477539063e-05,
      "step": 66596,
      "training_step_time": 10.076054096221924
    },
    {
      "epoch": 6.351184844970703e-05,
      "model_forward_time": 1.885732889175415,
      "step": 66597
    },
    {
      "epoch": 6.351184844970703e-05,
      "step": 66597,
      "training_step_time": 9.221145629882812
    },
    {
      "epoch": 6.351280212402344e-05,
      "model_forward_time": 2.398899555206299,
      "step": 66598
    },
    {
      "epoch": 6.351280212402344e-05,
      "step": 66598,
      "training_step_time": 7.677227735519409
    },
    {
      "epoch": 6.351375579833985e-05,
      "model_forward_time": 1.8768255710601807,
      "step": 66599
    },
    {
      "epoch": 6.351375579833985e-05,
      "step": 66599,
      "training_step_time": 8.136221647262573
    },
    {
      "epoch": 6.351470947265624e-05,
      "grad_norm": 0.09313911199569702,
      "learning_rate": 2.7523218204000772e-06,
      "loss": 0.0594,
      "step": 66600
    },
    {
      "epoch": 6.351470947265624e-05,
      "model_forward_time": 1.8465747833251953,
      "step": 66600
    },
    {
      "dynamics_loss_avg": 0.06411289311945438,
      "epoch": 6.351470947265624e-05,
      "step": 66600
    },
    {
      "action_loss_avg": 0.0018736039171926677,
      "epoch": 6.351470947265624e-05,
      "step": 66600
    },
    {
      "epoch": 6.351470947265624e-05,
      "step": 66600,
      "training_step_time": 7.495006561279297
    },
    {
      "epoch": 6.351566314697266e-05,
      "model_forward_time": 1.8646628856658936,
      "step": 66601
    },
    {
      "epoch": 6.351566314697266e-05,
      "step": 66601,
      "training_step_time": 6.662400245666504
    },
    {
      "epoch": 6.351661682128907e-05,
      "model_forward_time": 1.885030746459961,
      "step": 66602
    },
    {
      "epoch": 6.351661682128907e-05,
      "step": 66602,
      "training_step_time": 6.662230014801025
    },
    {
      "epoch": 6.351757049560546e-05,
      "model_forward_time": 1.9010906219482422,
      "step": 66603
    },
    {
      "epoch": 6.351757049560546e-05,
      "step": 66603,
      "training_step_time": 6.655004262924194
    },
    {
      "epoch": 6.351852416992187e-05,
      "model_forward_time": 1.8679101467132568,
      "step": 66604
    },
    {
      "epoch": 6.351852416992187e-05,
      "step": 66604,
      "training_step_time": 7.7233686447143555
    },
    {
      "epoch": 6.351947784423829e-05,
      "model_forward_time": 1.8734080791473389,
      "step": 66605
    },
    {
      "epoch": 6.351947784423829e-05,
      "step": 66605,
      "training_step_time": 8.176552295684814
    },
    {
      "epoch": 6.352043151855468e-05,
      "model_forward_time": 1.8626103401184082,
      "step": 66606
    },
    {
      "epoch": 6.352043151855468e-05,
      "step": 66606,
      "training_step_time": 8.54532265663147
    },
    {
      "epoch": 6.352138519287109e-05,
      "model_forward_time": 1.8179841041564941,
      "step": 66607
    },
    {
      "epoch": 6.352138519287109e-05,
      "step": 66607,
      "training_step_time": 7.575856924057007
    },
    {
      "epoch": 6.35223388671875e-05,
      "model_forward_time": 2.3938498497009277,
      "step": 66608
    },
    {
      "epoch": 6.35223388671875e-05,
      "step": 66608,
      "training_step_time": 7.552173376083374
    },
    {
      "epoch": 6.35232925415039e-05,
      "model_forward_time": 1.867621660232544,
      "step": 66609
    },
    {
      "epoch": 6.35232925415039e-05,
      "step": 66609,
      "training_step_time": 7.468932390213013
    },
    {
      "epoch": 6.352424621582031e-05,
      "grad_norm": 0.1275230050086975,
      "learning_rate": 2.750844960913661e-06,
      "loss": 0.0562,
      "step": 66610
    },
    {
      "epoch": 6.352424621582031e-05,
      "model_forward_time": 1.859668254852295,
      "step": 66610
    },
    {
      "dynamics_loss_avg": 0.0686970293521881,
      "epoch": 6.352424621582031e-05,
      "step": 66610
    },
    {
      "action_loss_avg": 0.0024487622606102375,
      "epoch": 6.352424621582031e-05,
      "step": 66610
    },
    {
      "epoch": 6.352424621582031e-05,
      "step": 66610,
      "training_step_time": 6.637400388717651
    },
    {
      "epoch": 6.352519989013672e-05,
      "model_forward_time": 1.8389039039611816,
      "step": 66611
    },
    {
      "epoch": 6.352519989013672e-05,
      "step": 66611,
      "training_step_time": 6.641531944274902
    },
    {
      "epoch": 6.352615356445312e-05,
      "model_forward_time": 1.5266203880310059,
      "step": 66612
    },
    {
      "epoch": 6.352615356445312e-05,
      "step": 66612,
      "training_step_time": 6.619867563247681
    },
    {
      "epoch": 6.352710723876953e-05,
      "model_forward_time": 1.8913655281066895,
      "step": 66613
    },
    {
      "epoch": 6.352710723876953e-05,
      "step": 66613,
      "training_step_time": 7.6461265087127686
    },
    {
      "epoch": 6.352806091308594e-05,
      "model_forward_time": 1.8906023502349854,
      "step": 66614
    },
    {
      "epoch": 6.352806091308594e-05,
      "step": 66614,
      "training_step_time": 9.967597484588623
    },
    {
      "epoch": 6.352901458740234e-05,
      "model_forward_time": 1.1845803260803223,
      "step": 66615
    },
    {
      "epoch": 6.352901458740234e-05,
      "step": 66615,
      "training_step_time": 10.822886943817139
    },
    {
      "epoch": 6.352996826171875e-05,
      "model_forward_time": 1.88264799118042,
      "step": 66616
    },
    {
      "epoch": 6.352996826171875e-05,
      "step": 66616,
      "training_step_time": 8.350891590118408
    },
    {
      "epoch": 6.353092193603516e-05,
      "model_forward_time": 2.394062042236328,
      "step": 66617
    },
    {
      "epoch": 6.353092193603516e-05,
      "step": 66617,
      "training_step_time": 7.679697751998901
    },
    {
      "epoch": 6.353187561035156e-05,
      "model_forward_time": 1.840428352355957,
      "step": 66618
    },
    {
      "epoch": 6.353187561035156e-05,
      "step": 66618,
      "training_step_time": 7.523321866989136
    },
    {
      "epoch": 6.353282928466797e-05,
      "model_forward_time": 1.897798776626587,
      "step": 66619
    },
    {
      "epoch": 6.353282928466797e-05,
      "step": 66619,
      "training_step_time": 6.692719221115112
    },
    {
      "epoch": 6.353378295898438e-05,
      "grad_norm": 0.09497455507516861,
      "learning_rate": 2.749368347391454e-06,
      "loss": 0.0557,
      "step": 66620
    },
    {
      "epoch": 6.353378295898438e-05,
      "model_forward_time": 1.8832087516784668,
      "step": 66620
    },
    {
      "dynamics_loss_avg": 0.05900968788191676,
      "epoch": 6.353378295898438e-05,
      "step": 66620
    },
    {
      "action_loss_avg": 0.002463117957813665,
      "epoch": 6.353378295898438e-05,
      "step": 66620
    },
    {
      "epoch": 6.353378295898438e-05,
      "step": 66620,
      "training_step_time": 6.713272333145142
    },
    {
      "epoch": 6.353473663330078e-05,
      "model_forward_time": 1.5558021068572998,
      "step": 66621
    },
    {
      "epoch": 6.353473663330078e-05,
      "step": 66621,
      "training_step_time": 6.71133279800415
    },
    {
      "epoch": 6.353569030761719e-05,
      "model_forward_time": 1.8796603679656982,
      "step": 66622
    },
    {
      "epoch": 6.353569030761719e-05,
      "step": 66622,
      "training_step_time": 7.712639808654785
    },
    {
      "epoch": 6.35366439819336e-05,
      "model_forward_time": 1.8276119232177734,
      "step": 66623
    },
    {
      "epoch": 6.35366439819336e-05,
      "step": 66623,
      "training_step_time": 7.615864515304565
    },
    {
      "epoch": 6.353759765625e-05,
      "model_forward_time": 1.8791794776916504,
      "step": 66624
    },
    {
      "epoch": 6.353759765625e-05,
      "step": 66624,
      "training_step_time": 8.621712923049927
    },
    {
      "epoch": 6.35385513305664e-05,
      "model_forward_time": 1.8737201690673828,
      "step": 66625
    },
    {
      "epoch": 6.35385513305664e-05,
      "step": 66625,
      "training_step_time": 10.56806206703186
    },
    {
      "epoch": 6.353950500488282e-05,
      "model_forward_time": 1.8673744201660156,
      "step": 66626
    },
    {
      "epoch": 6.353950500488282e-05,
      "step": 66626,
      "training_step_time": 7.799276828765869
    },
    {
      "epoch": 6.354045867919921e-05,
      "model_forward_time": 2.370330333709717,
      "step": 66627
    },
    {
      "epoch": 6.354045867919921e-05,
      "step": 66627,
      "training_step_time": 7.256384611129761
    },
    {
      "epoch": 6.354141235351563e-05,
      "model_forward_time": 1.8583950996398926,
      "step": 66628
    },
    {
      "epoch": 6.354141235351563e-05,
      "step": 66628,
      "training_step_time": 7.500669479370117
    },
    {
      "epoch": 6.354236602783204e-05,
      "model_forward_time": 1.8788728713989258,
      "step": 66629
    },
    {
      "epoch": 6.354236602783204e-05,
      "step": 66629,
      "training_step_time": 6.664097547531128
    },
    {
      "epoch": 6.354331970214843e-05,
      "grad_norm": 0.09668564051389694,
      "learning_rate": 2.7478919799949446e-06,
      "loss": 0.0605,
      "step": 66630
    },
    {
      "epoch": 6.354331970214843e-05,
      "model_forward_time": 1.8874175548553467,
      "step": 66630
    },
    {
      "dynamics_loss_avg": 0.056415400467813015,
      "epoch": 6.354331970214843e-05,
      "step": 66630
    },
    {
      "action_loss_avg": 0.0017897845100378618,
      "epoch": 6.354331970214843e-05,
      "step": 66630
    },
    {
      "epoch": 6.354331970214843e-05,
      "step": 66630,
      "training_step_time": 6.676966190338135
    },
    {
      "epoch": 6.354427337646484e-05,
      "model_forward_time": 1.8780126571655273,
      "step": 66631
    },
    {
      "epoch": 6.354427337646484e-05,
      "step": 66631,
      "training_step_time": 7.589351415634155
    },
    {
      "epoch": 6.354522705078125e-05,
      "model_forward_time": 1.8344717025756836,
      "step": 66632
    },
    {
      "epoch": 6.354522705078125e-05,
      "step": 66632,
      "training_step_time": 7.809973478317261
    },
    {
      "epoch": 6.354618072509765e-05,
      "model_forward_time": 1.8720440864562988,
      "step": 66633
    },
    {
      "epoch": 6.354618072509765e-05,
      "step": 66633,
      "training_step_time": 8.22067904472351
    },
    {
      "epoch": 6.354713439941406e-05,
      "model_forward_time": 1.8487119674682617,
      "step": 66634
    },
    {
      "epoch": 6.354713439941406e-05,
      "step": 66634,
      "training_step_time": 9.344236612319946
    },
    {
      "epoch": 6.354808807373047e-05,
      "model_forward_time": 1.8641951084136963,
      "step": 66635
    },
    {
      "epoch": 6.354808807373047e-05,
      "step": 66635,
      "training_step_time": 7.80477499961853
    },
    {
      "epoch": 6.354904174804687e-05,
      "model_forward_time": 1.9952056407928467,
      "step": 66636
    },
    {
      "epoch": 6.354904174804687e-05,
      "step": 66636,
      "training_step_time": 7.584164619445801
    },
    {
      "epoch": 6.354999542236328e-05,
      "model_forward_time": 1.4863097667694092,
      "step": 66637
    },
    {
      "epoch": 6.354999542236328e-05,
      "step": 66637,
      "training_step_time": 7.543363332748413
    },
    {
      "epoch": 6.355094909667969e-05,
      "model_forward_time": 1.8706417083740234,
      "step": 66638
    },
    {
      "epoch": 6.355094909667969e-05,
      "step": 66638,
      "training_step_time": 6.700423002243042
    },
    {
      "epoch": 6.355190277099609e-05,
      "model_forward_time": 1.8604261875152588,
      "step": 66639
    },
    {
      "epoch": 6.355190277099609e-05,
      "step": 66639,
      "training_step_time": 6.702474594116211
    },
    {
      "epoch": 6.35528564453125e-05,
      "grad_norm": 0.1168118491768837,
      "learning_rate": 2.7464158588855804e-06,
      "loss": 0.0552,
      "step": 66640
    },
    {
      "epoch": 6.35528564453125e-05,
      "model_forward_time": 1.879403829574585,
      "step": 66640
    },
    {
      "dynamics_loss_avg": 0.06951492354273796,
      "epoch": 6.35528564453125e-05,
      "step": 66640
    },
    {
      "action_loss_avg": 0.0020178101724013684,
      "epoch": 6.35528564453125e-05,
      "step": 66640
    },
    {
      "epoch": 6.35528564453125e-05,
      "step": 66640,
      "training_step_time": 7.595533847808838
    },
    {
      "epoch": 6.355381011962891e-05,
      "model_forward_time": 1.8439216613769531,
      "step": 66641
    },
    {
      "epoch": 6.355381011962891e-05,
      "step": 66641,
      "training_step_time": 7.727816104888916
    },
    {
      "epoch": 6.355476379394531e-05,
      "model_forward_time": 1.8759329319000244,
      "step": 66642
    },
    {
      "epoch": 6.355476379394531e-05,
      "step": 66642,
      "training_step_time": 9.431337833404541
    },
    {
      "epoch": 6.355571746826172e-05,
      "model_forward_time": 1.8364017009735107,
      "step": 66643
    },
    {
      "epoch": 6.355571746826172e-05,
      "step": 66643,
      "training_step_time": 9.946462869644165
    },
    {
      "epoch": 6.355667114257813e-05,
      "model_forward_time": 1.8738563060760498,
      "step": 66644
    },
    {
      "epoch": 6.355667114257813e-05,
      "step": 66644,
      "training_step_time": 7.804915428161621
    },
    {
      "epoch": 6.355762481689453e-05,
      "model_forward_time": 1.9884166717529297,
      "step": 66645
    },
    {
      "epoch": 6.355762481689453e-05,
      "step": 66645,
      "training_step_time": 7.702928304672241
    },
    {
      "epoch": 6.355857849121094e-05,
      "model_forward_time": 1.875910997390747,
      "step": 66646
    },
    {
      "epoch": 6.355857849121094e-05,
      "step": 66646,
      "training_step_time": 7.528992652893066
    },
    {
      "epoch": 6.355953216552735e-05,
      "model_forward_time": 1.885880947113037,
      "step": 66647
    },
    {
      "epoch": 6.355953216552735e-05,
      "step": 66647,
      "training_step_time": 6.687947034835815
    },
    {
      "epoch": 6.356048583984375e-05,
      "model_forward_time": 1.8791897296905518,
      "step": 66648
    },
    {
      "epoch": 6.356048583984375e-05,
      "step": 66648,
      "training_step_time": 6.720431327819824
    },
    {
      "epoch": 6.356143951416016e-05,
      "model_forward_time": 1.8474056720733643,
      "step": 66649
    },
    {
      "epoch": 6.356143951416016e-05,
      "step": 66649,
      "training_step_time": 7.615740537643433
    },
    {
      "epoch": 6.356239318847657e-05,
      "grad_norm": 0.0869007557630539,
      "learning_rate": 2.744939984224792e-06,
      "loss": 0.0569,
      "step": 66650
    },
    {
      "epoch": 6.356239318847657e-05,
      "model_forward_time": 1.8818202018737793,
      "step": 66650
    },
    {
      "dynamics_loss_avg": 0.053592351637780665,
      "epoch": 6.356239318847657e-05,
      "step": 66650
    },
    {
      "action_loss_avg": 0.0013774186169030144,
      "epoch": 6.356239318847657e-05,
      "step": 66650
    },
    {
      "epoch": 6.356239318847657e-05,
      "step": 66650,
      "training_step_time": 7.779849052429199
    },
    {
      "epoch": 6.356334686279297e-05,
      "model_forward_time": 1.905224323272705,
      "step": 66651
    },
    {
      "epoch": 6.356334686279297e-05,
      "step": 66651,
      "training_step_time": 8.817636251449585
    },
    {
      "epoch": 6.356430053710938e-05,
      "model_forward_time": 1.8351573944091797,
      "step": 66652
    },
    {
      "epoch": 6.356430053710938e-05,
      "step": 66652,
      "training_step_time": 8.81569528579712
    },
    {
      "epoch": 6.356525421142579e-05,
      "model_forward_time": 1.8947036266326904,
      "step": 66653
    },
    {
      "epoch": 6.356525421142579e-05,
      "step": 66653,
      "training_step_time": 7.830530881881714
    },
    {
      "epoch": 6.356620788574218e-05,
      "model_forward_time": 2.3856709003448486,
      "step": 66654
    },
    {
      "epoch": 6.356620788574218e-05,
      "step": 66654,
      "training_step_time": 7.755111217498779
    },
    {
      "epoch": 6.35671615600586e-05,
      "model_forward_time": 1.8883295059204102,
      "step": 66655
    },
    {
      "epoch": 6.35671615600586e-05,
      "step": 66655,
      "training_step_time": 7.2529754638671875
    },
    {
      "epoch": 6.3568115234375e-05,
      "model_forward_time": 1.8527977466583252,
      "step": 66656
    },
    {
      "epoch": 6.3568115234375e-05,
      "step": 66656,
      "training_step_time": 7.544812440872192
    },
    {
      "epoch": 6.35690689086914e-05,
      "model_forward_time": 1.8920419216156006,
      "step": 66657
    },
    {
      "epoch": 6.35690689086914e-05,
      "step": 66657,
      "training_step_time": 6.771865606307983
    },
    {
      "epoch": 6.357002258300781e-05,
      "model_forward_time": 1.8548917770385742,
      "step": 66658
    },
    {
      "epoch": 6.357002258300781e-05,
      "step": 66658,
      "training_step_time": 7.649980545043945
    },
    {
      "epoch": 6.357097625732422e-05,
      "model_forward_time": 1.8803577423095703,
      "step": 66659
    },
    {
      "epoch": 6.357097625732422e-05,
      "step": 66659,
      "training_step_time": 7.7465198040008545
    },
    {
      "epoch": 6.357192993164062e-05,
      "grad_norm": 0.08543383330106735,
      "learning_rate": 2.7434643561739737e-06,
      "loss": 0.0554,
      "step": 66660
    },
    {
      "epoch": 6.357192993164062e-05,
      "model_forward_time": 1.8993680477142334,
      "step": 66660
    },
    {
      "dynamics_loss_avg": 0.057868413627147675,
      "epoch": 6.357192993164062e-05,
      "step": 66660
    },
    {
      "action_loss_avg": 0.0012914366874611006,
      "epoch": 6.357192993164062e-05,
      "step": 66660
    },
    {
      "epoch": 6.357192993164062e-05,
      "step": 66660,
      "training_step_time": 7.809362411499023
    },
    {
      "epoch": 6.357288360595703e-05,
      "model_forward_time": 1.8481965065002441,
      "step": 66661
    },
    {
      "epoch": 6.357288360595703e-05,
      "step": 66661,
      "training_step_time": 9.850306272506714
    },
    {
      "epoch": 6.357383728027344e-05,
      "model_forward_time": 1.870962381362915,
      "step": 66662
    },
    {
      "epoch": 6.357383728027344e-05,
      "step": 66662,
      "training_step_time": 10.53876519203186
    },
    {
      "epoch": 6.357479095458984e-05,
      "model_forward_time": 1.904557466506958,
      "step": 66663
    },
    {
      "epoch": 6.357479095458984e-05,
      "step": 66663,
      "training_step_time": 7.65633225440979
    },
    {
      "epoch": 6.357574462890625e-05,
      "model_forward_time": 2.419161319732666,
      "step": 66664
    },
    {
      "epoch": 6.357574462890625e-05,
      "step": 66664,
      "training_step_time": 7.6199376583099365
    },
    {
      "epoch": 6.357669830322266e-05,
      "model_forward_time": 1.8366327285766602,
      "step": 66665
    },
    {
      "epoch": 6.357669830322266e-05,
      "step": 66665,
      "training_step_time": 7.565763711929321
    },
    {
      "epoch": 6.357765197753906e-05,
      "model_forward_time": 1.8742408752441406,
      "step": 66666
    },
    {
      "epoch": 6.357765197753906e-05,
      "step": 66666,
      "training_step_time": 6.7909040451049805
    },
    {
      "epoch": 6.357860565185547e-05,
      "model_forward_time": 1.8836538791656494,
      "step": 66667
    },
    {
      "epoch": 6.357860565185547e-05,
      "step": 66667,
      "training_step_time": 7.645727634429932
    },
    {
      "epoch": 6.357955932617188e-05,
      "model_forward_time": 1.8862438201904297,
      "step": 66668
    },
    {
      "epoch": 6.357955932617188e-05,
      "step": 66668,
      "training_step_time": 8.161134004592896
    },
    {
      "epoch": 6.358051300048828e-05,
      "model_forward_time": 1.850855827331543,
      "step": 66669
    },
    {
      "epoch": 6.358051300048828e-05,
      "step": 66669,
      "training_step_time": 7.7435126304626465
    },
    {
      "epoch": 6.358146667480469e-05,
      "grad_norm": 0.08856362849473953,
      "learning_rate": 2.7419889748945017e-06,
      "loss": 0.0554,
      "step": 66670
    },
    {
      "epoch": 6.358146667480469e-05,
      "model_forward_time": 1.8524677753448486,
      "step": 66670
    },
    {
      "dynamics_loss_avg": 0.054422990418970585,
      "epoch": 6.358146667480469e-05,
      "step": 66670
    },
    {
      "action_loss_avg": 0.0016733522003050893,
      "epoch": 6.358146667480469e-05,
      "step": 66670
    },
    {
      "epoch": 6.358146667480469e-05,
      "step": 66670,
      "training_step_time": 9.60123062133789
    },
    {
      "epoch": 6.35824203491211e-05,
      "model_forward_time": 1.8422718048095703,
      "step": 66671
    },
    {
      "epoch": 6.35824203491211e-05,
      "step": 66671,
      "training_step_time": 10.419417142868042
    },
    {
      "epoch": 6.35833740234375e-05,
      "model_forward_time": 1.8435349464416504,
      "step": 66672
    },
    {
      "epoch": 6.35833740234375e-05,
      "step": 66672,
      "training_step_time": 7.6568238735198975
    },
    {
      "epoch": 6.358432769775391e-05,
      "model_forward_time": 2.0817978382110596,
      "step": 66673
    },
    {
      "epoch": 6.358432769775391e-05,
      "step": 66673,
      "training_step_time": 7.592166900634766
    },
    {
      "epoch": 6.358528137207032e-05,
      "model_forward_time": 1.873192310333252,
      "step": 66674
    },
    {
      "epoch": 6.358528137207032e-05,
      "step": 66674,
      "training_step_time": 7.5219316482543945
    },
    {
      "epoch": 6.358623504638672e-05,
      "model_forward_time": 1.865086555480957,
      "step": 66675
    },
    {
      "epoch": 6.358623504638672e-05,
      "step": 66675,
      "training_step_time": 6.732080459594727
    },
    {
      "epoch": 6.358718872070313e-05,
      "model_forward_time": 1.8406946659088135,
      "step": 66676
    },
    {
      "epoch": 6.358718872070313e-05,
      "step": 66676,
      "training_step_time": 7.56820273399353
    },
    {
      "epoch": 6.358814239501954e-05,
      "model_forward_time": 1.898129940032959,
      "step": 66677
    },
    {
      "epoch": 6.358814239501954e-05,
      "step": 66677,
      "training_step_time": 7.616523504257202
    },
    {
      "epoch": 6.358909606933593e-05,
      "model_forward_time": 1.856060266494751,
      "step": 66678
    },
    {
      "epoch": 6.358909606933593e-05,
      "step": 66678,
      "training_step_time": 7.754820108413696
    },
    {
      "epoch": 6.359004974365235e-05,
      "model_forward_time": 1.8389723300933838,
      "step": 66679
    },
    {
      "epoch": 6.359004974365235e-05,
      "step": 66679,
      "training_step_time": 8.539725065231323
    },
    {
      "epoch": 6.359100341796876e-05,
      "grad_norm": 0.09277164936065674,
      "learning_rate": 2.7405138405477204e-06,
      "loss": 0.0562,
      "step": 66680
    },
    {
      "epoch": 6.359100341796876e-05,
      "model_forward_time": 1.8741021156311035,
      "step": 66680
    },
    {
      "dynamics_loss_avg": 0.07051398791372776,
      "epoch": 6.359100341796876e-05,
      "step": 66680
    },
    {
      "action_loss_avg": 0.002244091755710542,
      "epoch": 6.359100341796876e-05,
      "step": 66680
    },
    {
      "epoch": 6.359100341796876e-05,
      "step": 66680,
      "training_step_time": 9.43406343460083
    },
    {
      "epoch": 6.359195709228515e-05,
      "model_forward_time": 1.839946985244751,
      "step": 66681
    },
    {
      "epoch": 6.359195709228515e-05,
      "step": 66681,
      "training_step_time": 7.76047682762146
    },
    {
      "epoch": 6.359291076660156e-05,
      "model_forward_time": 2.4604785442352295,
      "step": 66682
    },
    {
      "epoch": 6.359291076660156e-05,
      "step": 66682,
      "training_step_time": 7.672120809555054
    },
    {
      "epoch": 6.359386444091798e-05,
      "model_forward_time": 1.8825788497924805,
      "step": 66683
    },
    {
      "epoch": 6.359386444091798e-05,
      "step": 66683,
      "training_step_time": 7.705024242401123
    },
    {
      "epoch": 6.359481811523437e-05,
      "model_forward_time": 1.8541553020477295,
      "step": 66684
    },
    {
      "epoch": 6.359481811523437e-05,
      "step": 66684,
      "training_step_time": 6.762027978897095
    },
    {
      "epoch": 6.359577178955078e-05,
      "model_forward_time": 1.8824326992034912,
      "step": 66685
    },
    {
      "epoch": 6.359577178955078e-05,
      "step": 66685,
      "training_step_time": 7.609455347061157
    },
    {
      "epoch": 6.35967254638672e-05,
      "model_forward_time": 1.876840591430664,
      "step": 66686
    },
    {
      "epoch": 6.35967254638672e-05,
      "step": 66686,
      "training_step_time": 7.4604408740997314
    },
    {
      "epoch": 6.359767913818359e-05,
      "model_forward_time": 1.8744208812713623,
      "step": 66687
    },
    {
      "epoch": 6.359767913818359e-05,
      "step": 66687,
      "training_step_time": 8.269929647445679
    },
    {
      "epoch": 6.35986328125e-05,
      "model_forward_time": 1.8610575199127197,
      "step": 66688
    },
    {
      "epoch": 6.35986328125e-05,
      "step": 66688,
      "training_step_time": 8.650651931762695
    },
    {
      "epoch": 6.35995864868164e-05,
      "model_forward_time": 1.8907787799835205,
      "step": 66689
    },
    {
      "epoch": 6.35995864868164e-05,
      "step": 66689,
      "training_step_time": 8.793316841125488
    },
    {
      "epoch": 6.360054016113281e-05,
      "grad_norm": 0.09655065834522247,
      "learning_rate": 2.73903895329495e-06,
      "loss": 0.0557,
      "step": 66690
    },
    {
      "epoch": 6.360054016113281e-05,
      "model_forward_time": 1.8440313339233398,
      "step": 66690
    },
    {
      "dynamics_loss_avg": 0.07298763506114483,
      "epoch": 6.360054016113281e-05,
      "step": 66690
    },
    {
      "action_loss_avg": 0.0017241649591596797,
      "epoch": 6.360054016113281e-05,
      "step": 66690
    },
    {
      "epoch": 6.360054016113281e-05,
      "step": 66690,
      "training_step_time": 7.80161714553833
    },
    {
      "epoch": 6.360149383544922e-05,
      "model_forward_time": 2.4231324195861816,
      "step": 66691
    },
    {
      "epoch": 6.360149383544922e-05,
      "step": 66691,
      "training_step_time": 7.577527046203613
    },
    {
      "epoch": 6.360244750976562e-05,
      "model_forward_time": 1.8658339977264404,
      "step": 66692
    },
    {
      "epoch": 6.360244750976562e-05,
      "step": 66692,
      "training_step_time": 7.671565294265747
    },
    {
      "epoch": 6.360340118408203e-05,
      "model_forward_time": 1.858921766281128,
      "step": 66693
    },
    {
      "epoch": 6.360340118408203e-05,
      "step": 66693,
      "training_step_time": 6.732171297073364
    },
    {
      "epoch": 6.360435485839844e-05,
      "model_forward_time": 1.8396008014678955,
      "step": 66694
    },
    {
      "epoch": 6.360435485839844e-05,
      "step": 66694,
      "training_step_time": 7.567654609680176
    },
    {
      "epoch": 6.360530853271484e-05,
      "model_forward_time": 1.5238890647888184,
      "step": 66695
    },
    {
      "epoch": 6.360530853271484e-05,
      "step": 66695,
      "training_step_time": 7.463183403015137
    },
    {
      "epoch": 6.360626220703125e-05,
      "model_forward_time": 1.858429193496704,
      "step": 66696
    },
    {
      "epoch": 6.360626220703125e-05,
      "step": 66696,
      "training_step_time": 7.662633657455444
    },
    {
      "epoch": 6.360721588134766e-05,
      "model_forward_time": 1.4970636367797852,
      "step": 66697
    },
    {
      "epoch": 6.360721588134766e-05,
      "step": 66697,
      "training_step_time": 8.250586986541748
    },
    {
      "epoch": 6.360816955566406e-05,
      "model_forward_time": 1.8792178630828857,
      "step": 66698
    },
    {
      "epoch": 6.360816955566406e-05,
      "step": 66698,
      "training_step_time": 10.821224212646484
    },
    {
      "epoch": 6.360912322998047e-05,
      "model_forward_time": 1.858778715133667,
      "step": 66699
    },
    {
      "epoch": 6.360912322998047e-05,
      "step": 66699,
      "training_step_time": 8.340736150741577
    },
    {
      "epoch": 6.361007690429688e-05,
      "grad_norm": 0.1020917147397995,
      "learning_rate": 2.7375643132974783e-06,
      "loss": 0.0551,
      "step": 66700
    },
    {
      "epoch": 6.361007690429688e-05,
      "model_forward_time": 1.8537631034851074,
      "step": 66700
    },
    {
      "dynamics_loss_avg": 0.061440818943083286,
      "epoch": 6.361007690429688e-05,
      "step": 66700
    },
    {
      "action_loss_avg": 0.0020597546885255725,
      "epoch": 6.361007690429688e-05,
      "step": 66700
    },
    {
      "epoch": 6.361007690429688e-05,
      "step": 66700,
      "training_step_time": 7.766368389129639
    },
    {
      "epoch": 6.361103057861327e-05,
      "model_forward_time": 2.4074645042419434,
      "step": 66701
    },
    {
      "epoch": 6.361103057861327e-05,
      "step": 66701,
      "training_step_time": 7.5322723388671875
    },
    {
      "epoch": 6.361198425292969e-05,
      "model_forward_time": 1.8682539463043213,
      "step": 66702
    },
    {
      "epoch": 6.361198425292969e-05,
      "step": 66702,
      "training_step_time": 6.794259309768677
    },
    {
      "epoch": 6.36129379272461e-05,
      "model_forward_time": 1.8695588111877441,
      "step": 66703
    },
    {
      "epoch": 6.36129379272461e-05,
      "step": 66703,
      "training_step_time": 7.65297532081604
    },
    {
      "epoch": 6.36138916015625e-05,
      "model_forward_time": 1.8301451206207275,
      "step": 66704
    },
    {
      "epoch": 6.36138916015625e-05,
      "step": 66704,
      "training_step_time": 7.503983974456787
    },
    {
      "epoch": 6.36148452758789e-05,
      "model_forward_time": 1.8586843013763428,
      "step": 66705
    },
    {
      "epoch": 6.36148452758789e-05,
      "step": 66705,
      "training_step_time": 8.449333429336548
    },
    {
      "epoch": 6.361579895019532e-05,
      "model_forward_time": 1.8702709674835205,
      "step": 66706
    },
    {
      "epoch": 6.361579895019532e-05,
      "step": 66706,
      "training_step_time": 8.336692333221436
    },
    {
      "epoch": 6.361675262451171e-05,
      "model_forward_time": 1.8560762405395508,
      "step": 66707
    },
    {
      "epoch": 6.361675262451171e-05,
      "step": 66707,
      "training_step_time": 8.614909648895264
    },
    {
      "epoch": 6.361770629882812e-05,
      "model_forward_time": 1.855050802230835,
      "step": 66708
    },
    {
      "epoch": 6.361770629882812e-05,
      "step": 66708,
      "training_step_time": 7.757284879684448
    },
    {
      "epoch": 6.361865997314453e-05,
      "model_forward_time": 1.8411126136779785,
      "step": 66709
    },
    {
      "epoch": 6.361865997314453e-05,
      "step": 66709,
      "training_step_time": 8.288130521774292
    },
    {
      "epoch": 6.361961364746093e-05,
      "grad_norm": 0.0740431398153305,
      "learning_rate": 2.736089920716572e-06,
      "loss": 0.0554,
      "step": 66710
    },
    {
      "epoch": 6.361961364746093e-05,
      "model_forward_time": 2.3977720737457275,
      "step": 66710
    },
    {
      "dynamics_loss_avg": 0.06516084857285023,
      "epoch": 6.361961364746093e-05,
      "step": 66710
    },
    {
      "action_loss_avg": 0.00195905581349507,
      "epoch": 6.361961364746093e-05,
      "step": 66710
    },
    {
      "epoch": 6.361961364746093e-05,
      "step": 66710,
      "training_step_time": 8.653268575668335
    },
    {
      "epoch": 6.362056732177734e-05,
      "model_forward_time": 1.8533639907836914,
      "step": 66711
    },
    {
      "epoch": 6.362056732177734e-05,
      "step": 66711,
      "training_step_time": 6.759125471115112
    },
    {
      "epoch": 6.362152099609375e-05,
      "model_forward_time": 1.8280534744262695,
      "step": 66712
    },
    {
      "epoch": 6.362152099609375e-05,
      "step": 66712,
      "training_step_time": 7.602648019790649
    },
    {
      "epoch": 6.362247467041015e-05,
      "model_forward_time": 1.8560938835144043,
      "step": 66713
    },
    {
      "epoch": 6.362247467041015e-05,
      "step": 66713,
      "training_step_time": 7.4871909618377686
    },
    {
      "epoch": 6.362342834472656e-05,
      "model_forward_time": 1.8376274108886719,
      "step": 66714
    },
    {
      "epoch": 6.362342834472656e-05,
      "step": 66714,
      "training_step_time": 7.7244768142700195
    },
    {
      "epoch": 6.362438201904297e-05,
      "model_forward_time": 1.827960729598999,
      "step": 66715
    },
    {
      "epoch": 6.362438201904297e-05,
      "step": 66715,
      "training_step_time": 7.708311080932617
    },
    {
      "epoch": 6.362533569335937e-05,
      "model_forward_time": 1.8800508975982666,
      "step": 66716
    },
    {
      "epoch": 6.362533569335937e-05,
      "step": 66716,
      "training_step_time": 7.852585554122925
    },
    {
      "epoch": 6.362628936767578e-05,
      "model_forward_time": 1.8477919101715088,
      "step": 66717
    },
    {
      "epoch": 6.362628936767578e-05,
      "step": 66717,
      "training_step_time": 9.357678413391113
    },
    {
      "epoch": 6.362724304199219e-05,
      "model_forward_time": 1.8630917072296143,
      "step": 66718
    },
    {
      "epoch": 6.362724304199219e-05,
      "step": 66718,
      "training_step_time": 7.775092601776123
    },
    {
      "epoch": 6.362819671630859e-05,
      "model_forward_time": 2.385263204574585,
      "step": 66719
    },
    {
      "epoch": 6.362819671630859e-05,
      "step": 66719,
      "training_step_time": 7.663813829421997
    },
    {
      "epoch": 6.3629150390625e-05,
      "grad_norm": 0.09644068777561188,
      "learning_rate": 2.7346157757134704e-06,
      "loss": 0.0556,
      "step": 66720
    },
    {
      "epoch": 6.3629150390625e-05,
      "model_forward_time": 1.8484163284301758,
      "step": 66720
    },
    {
      "dynamics_loss_avg": 0.04474514201283455,
      "epoch": 6.3629150390625e-05,
      "step": 66720
    },
    {
      "action_loss_avg": 0.0012873419444076716,
      "epoch": 6.3629150390625e-05,
      "step": 66720
    },
    {
      "epoch": 6.3629150390625e-05,
      "step": 66720,
      "training_step_time": 7.563488483428955
    },
    {
      "epoch": 6.363010406494141e-05,
      "model_forward_time": 1.8523662090301514,
      "step": 66721
    },
    {
      "epoch": 6.363010406494141e-05,
      "step": 66721,
      "training_step_time": 6.730571031570435
    },
    {
      "epoch": 6.363105773925781e-05,
      "model_forward_time": 1.8504438400268555,
      "step": 66722
    },
    {
      "epoch": 6.363105773925781e-05,
      "step": 66722,
      "training_step_time": 7.602883815765381
    },
    {
      "epoch": 6.363201141357422e-05,
      "model_forward_time": 1.8562941551208496,
      "step": 66723
    },
    {
      "epoch": 6.363201141357422e-05,
      "step": 66723,
      "training_step_time": 7.662592887878418
    },
    {
      "epoch": 6.363296508789063e-05,
      "model_forward_time": 1.8441896438598633,
      "step": 66724
    },
    {
      "epoch": 6.363296508789063e-05,
      "step": 66724,
      "training_step_time": 7.662661552429199
    },
    {
      "epoch": 6.363391876220703e-05,
      "model_forward_time": 1.5516841411590576,
      "step": 66725
    },
    {
      "epoch": 6.363391876220703e-05,
      "step": 66725,
      "training_step_time": 9.587235689163208
    },
    {
      "epoch": 6.363487243652344e-05,
      "model_forward_time": 1.842257022857666,
      "step": 66726
    },
    {
      "epoch": 6.363487243652344e-05,
      "step": 66726,
      "training_step_time": 10.830873727798462
    },
    {
      "epoch": 6.363582611083985e-05,
      "model_forward_time": 1.8659827709197998,
      "step": 66727
    },
    {
      "epoch": 6.363582611083985e-05,
      "step": 66727,
      "training_step_time": 7.764830827713013
    },
    {
      "epoch": 6.363677978515624e-05,
      "model_forward_time": 1.8530304431915283,
      "step": 66728
    },
    {
      "epoch": 6.363677978515624e-05,
      "step": 66728,
      "training_step_time": 8.707722425460815
    },
    {
      "epoch": 6.363773345947266e-05,
      "model_forward_time": 1.8791275024414062,
      "step": 66729
    },
    {
      "epoch": 6.363773345947266e-05,
      "step": 66729,
      "training_step_time": 7.595464706420898
    },
    {
      "epoch": 6.363868713378907e-05,
      "grad_norm": 0.08226367086172104,
      "learning_rate": 2.733141878449379e-06,
      "loss": 0.0545,
      "step": 66730
    },
    {
      "epoch": 6.363868713378907e-05,
      "model_forward_time": 1.831958293914795,
      "step": 66730
    },
    {
      "dynamics_loss_avg": 0.04200892541557551,
      "epoch": 6.363868713378907e-05,
      "step": 66730
    },
    {
      "action_loss_avg": 0.001332843626732938,
      "epoch": 6.363868713378907e-05,
      "step": 66730
    },
    {
      "epoch": 6.363868713378907e-05,
      "step": 66730,
      "training_step_time": 6.7592551708221436
    },
    {
      "epoch": 6.363964080810546e-05,
      "model_forward_time": 1.8355190753936768,
      "step": 66731
    },
    {
      "epoch": 6.363964080810546e-05,
      "step": 66731,
      "training_step_time": 7.62302565574646
    },
    {
      "epoch": 6.364059448242187e-05,
      "model_forward_time": 1.886850118637085,
      "step": 66732
    },
    {
      "epoch": 6.364059448242187e-05,
      "step": 66732,
      "training_step_time": 8.730911016464233
    },
    {
      "epoch": 6.364154815673829e-05,
      "model_forward_time": 1.85111403465271,
      "step": 66733
    },
    {
      "epoch": 6.364154815673829e-05,
      "step": 66733,
      "training_step_time": 7.609942436218262
    },
    {
      "epoch": 6.364250183105468e-05,
      "model_forward_time": 1.8594520092010498,
      "step": 66734
    },
    {
      "epoch": 6.364250183105468e-05,
      "step": 66734,
      "training_step_time": 8.73235034942627
    },
    {
      "epoch": 6.364345550537109e-05,
      "model_forward_time": 1.8431994915008545,
      "step": 66735
    },
    {
      "epoch": 6.364345550537109e-05,
      "step": 66735,
      "training_step_time": 8.311809301376343
    },
    {
      "epoch": 6.36444091796875e-05,
      "model_forward_time": 1.83034086227417,
      "step": 66736
    },
    {
      "epoch": 6.36444091796875e-05,
      "step": 66736,
      "training_step_time": 7.744242191314697
    },
    {
      "epoch": 6.36453628540039e-05,
      "model_forward_time": 1.8868486881256104,
      "step": 66737
    },
    {
      "epoch": 6.36453628540039e-05,
      "step": 66737,
      "training_step_time": 8.607184648513794
    },
    {
      "epoch": 6.364631652832031e-05,
      "model_forward_time": 2.3955628871917725,
      "step": 66738
    },
    {
      "epoch": 6.364631652832031e-05,
      "step": 66738,
      "training_step_time": 7.518602609634399
    },
    {
      "epoch": 6.364727020263672e-05,
      "model_forward_time": 1.877615213394165,
      "step": 66739
    },
    {
      "epoch": 6.364727020263672e-05,
      "step": 66739,
      "training_step_time": 6.732947826385498
    },
    {
      "epoch": 6.364822387695312e-05,
      "grad_norm": 0.11151503771543503,
      "learning_rate": 2.7316682290854846e-06,
      "loss": 0.0573,
      "step": 66740
    },
    {
      "epoch": 6.364822387695312e-05,
      "model_forward_time": 1.1884396076202393,
      "step": 66740
    },
    {
      "dynamics_loss_avg": 0.047018838487565516,
      "epoch": 6.364822387695312e-05,
      "step": 66740
    },
    {
      "action_loss_avg": 0.0016245630336925387,
      "epoch": 6.364822387695312e-05,
      "step": 66740
    },
    {
      "epoch": 6.364822387695312e-05,
      "step": 66740,
      "training_step_time": 7.64061975479126
    },
    {
      "epoch": 6.364917755126953e-05,
      "model_forward_time": 1.8608052730560303,
      "step": 66741
    },
    {
      "epoch": 6.364917755126953e-05,
      "step": 66741,
      "training_step_time": 7.667170763015747
    },
    {
      "epoch": 6.365013122558594e-05,
      "model_forward_time": 1.8532540798187256,
      "step": 66742
    },
    {
      "epoch": 6.365013122558594e-05,
      "step": 66742,
      "training_step_time": 7.656842470169067
    },
    {
      "epoch": 6.365108489990234e-05,
      "model_forward_time": 1.8849098682403564,
      "step": 66743
    },
    {
      "epoch": 6.365108489990234e-05,
      "step": 66743,
      "training_step_time": 8.422399282455444
    },
    {
      "epoch": 6.365203857421875e-05,
      "model_forward_time": 1.838301181793213,
      "step": 66744
    },
    {
      "epoch": 6.365203857421875e-05,
      "step": 66744,
      "training_step_time": 9.230245113372803
    },
    {
      "epoch": 6.365299224853516e-05,
      "model_forward_time": 1.8224194049835205,
      "step": 66745
    },
    {
      "epoch": 6.365299224853516e-05,
      "step": 66745,
      "training_step_time": 9.42024278640747
    },
    {
      "epoch": 6.365394592285156e-05,
      "model_forward_time": 1.8561928272247314,
      "step": 66746
    },
    {
      "epoch": 6.365394592285156e-05,
      "step": 66746,
      "training_step_time": 8.480644941329956
    },
    {
      "epoch": 6.365489959716797e-05,
      "model_forward_time": 2.377641439437866,
      "step": 66747
    },
    {
      "epoch": 6.365489959716797e-05,
      "step": 66747,
      "training_step_time": 7.5976784229278564
    },
    {
      "epoch": 6.365585327148438e-05,
      "model_forward_time": 1.8779714107513428,
      "step": 66748
    },
    {
      "epoch": 6.365585327148438e-05,
      "step": 66748,
      "training_step_time": 7.132157564163208
    },
    {
      "epoch": 6.365680694580078e-05,
      "model_forward_time": 1.8611712455749512,
      "step": 66749
    },
    {
      "epoch": 6.365680694580078e-05,
      "step": 66749,
      "training_step_time": 7.606972694396973
    },
    {
      "epoch": 6.365776062011719e-05,
      "grad_norm": 0.1127181202173233,
      "learning_rate": 2.7301948277829415e-06,
      "loss": 0.0551,
      "step": 66750
    },
    {
      "epoch": 6.365776062011719e-05,
      "model_forward_time": 1.8359971046447754,
      "step": 66750
    },
    {
      "dynamics_loss_avg": 0.05254400204867125,
      "epoch": 6.365776062011719e-05,
      "step": 66750
    },
    {
      "action_loss_avg": 0.0013840126543072984,
      "epoch": 6.365776062011719e-05,
      "step": 66750
    },
    {
      "epoch": 6.365776062011719e-05,
      "step": 66750,
      "training_step_time": 7.738648414611816
    },
    {
      "epoch": 6.36587142944336e-05,
      "model_forward_time": 1.872166633605957,
      "step": 66751
    },
    {
      "epoch": 6.36587142944336e-05,
      "step": 66751,
      "training_step_time": 7.692318916320801
    },
    {
      "epoch": 6.365966796875e-05,
      "model_forward_time": 1.8700613975524902,
      "step": 66752
    },
    {
      "epoch": 6.365966796875e-05,
      "step": 66752,
      "training_step_time": 8.27850866317749
    },
    {
      "epoch": 6.36606216430664e-05,
      "model_forward_time": 1.8506524562835693,
      "step": 66753
    },
    {
      "epoch": 6.36606216430664e-05,
      "step": 66753,
      "training_step_time": 10.49654507637024
    },
    {
      "epoch": 6.366157531738282e-05,
      "model_forward_time": 1.825254201889038,
      "step": 66754
    },
    {
      "epoch": 6.366157531738282e-05,
      "step": 66754,
      "training_step_time": 9.33041524887085
    },
    {
      "epoch": 6.366252899169921e-05,
      "model_forward_time": 1.8909687995910645,
      "step": 66755
    },
    {
      "epoch": 6.366252899169921e-05,
      "step": 66755,
      "training_step_time": 7.742121934890747
    },
    {
      "epoch": 6.366348266601563e-05,
      "model_forward_time": 2.387084484100342,
      "step": 66756
    },
    {
      "epoch": 6.366348266601563e-05,
      "step": 66756,
      "training_step_time": 8.788547039031982
    },
    {
      "epoch": 6.366443634033204e-05,
      "model_forward_time": 1.8766605854034424,
      "step": 66757
    },
    {
      "epoch": 6.366443634033204e-05,
      "step": 66757,
      "training_step_time": 7.17227840423584
    },
    {
      "epoch": 6.366539001464843e-05,
      "model_forward_time": 1.8666431903839111,
      "step": 66758
    },
    {
      "epoch": 6.366539001464843e-05,
      "step": 66758,
      "training_step_time": 7.648264408111572
    },
    {
      "epoch": 6.366634368896484e-05,
      "model_forward_time": 1.8394103050231934,
      "step": 66759
    },
    {
      "epoch": 6.366634368896484e-05,
      "step": 66759,
      "training_step_time": 7.756259202957153
    },
    {
      "epoch": 6.366729736328125e-05,
      "grad_norm": 0.1447267383337021,
      "learning_rate": 2.7287216747028818e-06,
      "loss": 0.0546,
      "step": 66760
    },
    {
      "epoch": 6.366729736328125e-05,
      "model_forward_time": 1.8575165271759033,
      "step": 66760
    },
    {
      "dynamics_loss_avg": 0.055621227622032164,
      "epoch": 6.366729736328125e-05,
      "step": 66760
    },
    {
      "action_loss_avg": 0.001675488901673816,
      "epoch": 6.366729736328125e-05,
      "step": 66760
    },
    {
      "epoch": 6.366729736328125e-05,
      "step": 66760,
      "training_step_time": 7.781421899795532
    },
    {
      "epoch": 6.366825103759765e-05,
      "model_forward_time": 1.8617427349090576,
      "step": 66761
    },
    {
      "epoch": 6.366825103759765e-05,
      "step": 66761,
      "training_step_time": 7.7228922843933105
    },
    {
      "epoch": 6.366920471191406e-05,
      "model_forward_time": 1.8455348014831543,
      "step": 66762
    },
    {
      "epoch": 6.366920471191406e-05,
      "step": 66762,
      "training_step_time": 9.512617349624634
    },
    {
      "epoch": 6.367015838623047e-05,
      "model_forward_time": 1.8742859363555908,
      "step": 66763
    },
    {
      "epoch": 6.367015838623047e-05,
      "step": 66763,
      "training_step_time": 8.884024143218994
    },
    {
      "epoch": 6.367111206054687e-05,
      "model_forward_time": 1.814096450805664,
      "step": 66764
    },
    {
      "epoch": 6.367111206054687e-05,
      "step": 66764,
      "training_step_time": 7.693425416946411
    },
    {
      "epoch": 6.367206573486328e-05,
      "model_forward_time": 2.43477725982666,
      "step": 66765
    },
    {
      "epoch": 6.367206573486328e-05,
      "step": 66765,
      "training_step_time": 7.7852654457092285
    },
    {
      "epoch": 6.367301940917969e-05,
      "model_forward_time": 1.8437504768371582,
      "step": 66766
    },
    {
      "epoch": 6.367301940917969e-05,
      "step": 66766,
      "training_step_time": 7.231013059616089
    },
    {
      "epoch": 6.367397308349609e-05,
      "model_forward_time": 1.8295722007751465,
      "step": 66767
    },
    {
      "epoch": 6.367397308349609e-05,
      "step": 66767,
      "training_step_time": 7.626768112182617
    },
    {
      "epoch": 6.36749267578125e-05,
      "model_forward_time": 1.8550314903259277,
      "step": 66768
    },
    {
      "epoch": 6.36749267578125e-05,
      "step": 66768,
      "training_step_time": 8.440199375152588
    },
    {
      "epoch": 6.367588043212891e-05,
      "model_forward_time": 1.8727400302886963,
      "step": 66769
    },
    {
      "epoch": 6.367588043212891e-05,
      "step": 66769,
      "training_step_time": 7.802824020385742
    },
    {
      "epoch": 6.367683410644531e-05,
      "grad_norm": 0.08996313065290451,
      "learning_rate": 2.7272487700064027e-06,
      "loss": 0.061,
      "step": 66770
    },
    {
      "epoch": 6.367683410644531e-05,
      "model_forward_time": 1.8486206531524658,
      "step": 66770
    },
    {
      "dynamics_loss_avg": 0.05593111710622907,
      "epoch": 6.367683410644531e-05,
      "step": 66770
    },
    {
      "action_loss_avg": 0.0012151078582974151,
      "epoch": 6.367683410644531e-05,
      "step": 66770
    },
    {
      "epoch": 6.367683410644531e-05,
      "step": 66770,
      "training_step_time": 7.6142120361328125
    },
    {
      "epoch": 6.367778778076172e-05,
      "model_forward_time": 1.8586697578430176,
      "step": 66771
    },
    {
      "epoch": 6.367778778076172e-05,
      "step": 66771,
      "training_step_time": 7.744929075241089
    },
    {
      "epoch": 6.367874145507813e-05,
      "model_forward_time": 1.8427939414978027,
      "step": 66772
    },
    {
      "epoch": 6.367874145507813e-05,
      "step": 66772,
      "training_step_time": 8.557217121124268
    },
    {
      "epoch": 6.367969512939453e-05,
      "model_forward_time": 1.850740671157837,
      "step": 66773
    },
    {
      "epoch": 6.367969512939453e-05,
      "step": 66773,
      "training_step_time": 7.884836673736572
    },
    {
      "epoch": 6.368064880371094e-05,
      "model_forward_time": 1.861840009689331,
      "step": 66774
    },
    {
      "epoch": 6.368064880371094e-05,
      "step": 66774,
      "training_step_time": 9.620649337768555
    },
    {
      "epoch": 6.368160247802735e-05,
      "model_forward_time": 1.8730263710021973,
      "step": 66775
    },
    {
      "epoch": 6.368160247802735e-05,
      "step": 66775,
      "training_step_time": 7.625499248504639
    },
    {
      "epoch": 6.368255615234375e-05,
      "model_forward_time": 1.885490894317627,
      "step": 66776
    },
    {
      "epoch": 6.368255615234375e-05,
      "step": 66776,
      "training_step_time": 7.674793720245361
    },
    {
      "epoch": 6.368350982666016e-05,
      "model_forward_time": 1.8349332809448242,
      "step": 66777
    },
    {
      "epoch": 6.368350982666016e-05,
      "step": 66777,
      "training_step_time": 7.256162166595459
    },
    {
      "epoch": 6.368446350097657e-05,
      "model_forward_time": 1.8283047676086426,
      "step": 66778
    },
    {
      "epoch": 6.368446350097657e-05,
      "step": 66778,
      "training_step_time": 8.76504135131836
    },
    {
      "epoch": 6.368541717529297e-05,
      "model_forward_time": 1.8523836135864258,
      "step": 66779
    },
    {
      "epoch": 6.368541717529297e-05,
      "step": 66779,
      "training_step_time": 7.584190845489502
    },
    {
      "epoch": 6.368637084960938e-05,
      "grad_norm": 0.09125060588121414,
      "learning_rate": 2.725776113854583e-06,
      "loss": 0.057,
      "step": 66780
    },
    {
      "epoch": 6.368637084960938e-05,
      "model_forward_time": 1.852252721786499,
      "step": 66780
    },
    {
      "dynamics_loss_avg": 0.0424739183858037,
      "epoch": 6.368637084960938e-05,
      "step": 66780
    },
    {
      "action_loss_avg": 0.001124699885258451,
      "epoch": 6.368637084960938e-05,
      "step": 66780
    },
    {
      "epoch": 6.368637084960938e-05,
      "step": 66780,
      "training_step_time": 8.618751525878906
    },
    {
      "epoch": 6.368732452392579e-05,
      "model_forward_time": 1.8582956790924072,
      "step": 66781
    },
    {
      "epoch": 6.368732452392579e-05,
      "step": 66781,
      "training_step_time": 8.694040060043335
    },
    {
      "epoch": 6.368827819824218e-05,
      "model_forward_time": 1.881096601486206,
      "step": 66782
    },
    {
      "epoch": 6.368827819824218e-05,
      "step": 66782,
      "training_step_time": 8.469706058502197
    },
    {
      "epoch": 6.36892318725586e-05,
      "model_forward_time": 1.8878967761993408,
      "step": 66783
    },
    {
      "epoch": 6.36892318725586e-05,
      "step": 66783,
      "training_step_time": 8.803759336471558
    },
    {
      "epoch": 6.3690185546875e-05,
      "model_forward_time": 2.395598888397217,
      "step": 66784
    },
    {
      "epoch": 6.3690185546875e-05,
      "step": 66784,
      "training_step_time": 7.758388519287109
    },
    {
      "epoch": 6.36911392211914e-05,
      "model_forward_time": 1.7648088932037354,
      "step": 66785
    },
    {
      "epoch": 6.36911392211914e-05,
      "step": 66785,
      "training_step_time": 7.68067193031311
    },
    {
      "epoch": 6.369209289550781e-05,
      "model_forward_time": 1.711686611175537,
      "step": 66786
    },
    {
      "epoch": 6.369209289550781e-05,
      "step": 66786,
      "training_step_time": 7.672279119491577
    },
    {
      "epoch": 6.369304656982422e-05,
      "model_forward_time": 1.7165882587432861,
      "step": 66787
    },
    {
      "epoch": 6.369304656982422e-05,
      "step": 66787,
      "training_step_time": 8.819591760635376
    },
    {
      "epoch": 6.369400024414062e-05,
      "model_forward_time": 1.3309826850891113,
      "step": 66788
    },
    {
      "epoch": 6.369400024414062e-05,
      "step": 66788,
      "training_step_time": 7.7728047370910645
    },
    {
      "epoch": 6.369495391845703e-05,
      "model_forward_time": 1.7247734069824219,
      "step": 66789
    },
    {
      "epoch": 6.369495391845703e-05,
      "step": 66789,
      "training_step_time": 7.833407402038574
    },
    {
      "epoch": 6.369590759277344e-05,
      "grad_norm": 0.082624152302742,
      "learning_rate": 2.7243037064084664e-06,
      "loss": 0.0585,
      "step": 66790
    },
    {
      "epoch": 6.369590759277344e-05,
      "model_forward_time": 1.745065450668335,
      "step": 66790
    },
    {
      "dynamics_loss_avg": 0.04807045375928283,
      "epoch": 6.369590759277344e-05,
      "step": 66790
    },
    {
      "action_loss_avg": 0.0016395277110859753,
      "epoch": 6.369590759277344e-05,
      "step": 66790
    },
    {
      "epoch": 6.369590759277344e-05,
      "step": 66790,
      "training_step_time": 7.801664352416992
    },
    {
      "epoch": 6.369686126708984e-05,
      "model_forward_time": 1.830282211303711,
      "step": 66791
    },
    {
      "epoch": 6.369686126708984e-05,
      "step": 66791,
      "training_step_time": 10.408097505569458
    },
    {
      "epoch": 6.369781494140625e-05,
      "model_forward_time": 1.8429570198059082,
      "step": 66792
    },
    {
      "epoch": 6.369781494140625e-05,
      "step": 66792,
      "training_step_time": 7.773080348968506
    },
    {
      "epoch": 6.369876861572266e-05,
      "model_forward_time": 2.3651163578033447,
      "step": 66793
    },
    {
      "epoch": 6.369876861572266e-05,
      "step": 66793,
      "training_step_time": 7.727717161178589
    },
    {
      "epoch": 6.369972229003906e-05,
      "model_forward_time": 1.8256216049194336,
      "step": 66794
    },
    {
      "epoch": 6.369972229003906e-05,
      "step": 66794,
      "training_step_time": 7.6027443408966064
    },
    {
      "epoch": 6.370067596435547e-05,
      "model_forward_time": 1.8401985168457031,
      "step": 66795
    },
    {
      "epoch": 6.370067596435547e-05,
      "step": 66795,
      "training_step_time": 7.684993743896484
    },
    {
      "epoch": 6.370162963867188e-05,
      "model_forward_time": 1.8440415859222412,
      "step": 66796
    },
    {
      "epoch": 6.370162963867188e-05,
      "step": 66796,
      "training_step_time": 7.792436361312866
    },
    {
      "epoch": 6.370258331298828e-05,
      "model_forward_time": 1.8351781368255615,
      "step": 66797
    },
    {
      "epoch": 6.370258331298828e-05,
      "step": 66797,
      "training_step_time": 7.707171678543091
    },
    {
      "epoch": 6.370353698730469e-05,
      "model_forward_time": 1.8437886238098145,
      "step": 66798
    },
    {
      "epoch": 6.370353698730469e-05,
      "step": 66798,
      "training_step_time": 9.620927095413208
    },
    {
      "epoch": 6.37044906616211e-05,
      "model_forward_time": 1.8322958946228027,
      "step": 66799
    },
    {
      "epoch": 6.37044906616211e-05,
      "step": 66799,
      "training_step_time": 9.459511041641235
    },
    {
      "epoch": 6.37054443359375e-05,
      "grad_norm": 0.08918905258178711,
      "learning_rate": 2.722831547829077e-06,
      "loss": 0.0606,
      "step": 66800
    },
    {
      "epoch": 6.37054443359375e-05,
      "model_forward_time": 1.8403115272521973,
      "step": 66800
    },
    {
      "dynamics_loss_avg": 0.049746555276215076,
      "epoch": 6.37054443359375e-05,
      "step": 66800
    },
    {
      "action_loss_avg": 0.0010346752329496666,
      "epoch": 6.37054443359375e-05,
      "step": 66800
    },
    {
      "epoch": 6.37054443359375e-05,
      "step": 66800,
      "training_step_time": 7.741396188735962
    },
    {
      "epoch": 6.370639801025391e-05,
      "model_forward_time": 1.8536715507507324,
      "step": 66801
    },
    {
      "epoch": 6.370639801025391e-05,
      "step": 66801,
      "training_step_time": 7.795397043228149
    },
    {
      "epoch": 6.370735168457032e-05,
      "model_forward_time": 2.4185292720794678,
      "step": 66802
    },
    {
      "epoch": 6.370735168457032e-05,
      "step": 66802,
      "training_step_time": 7.729848384857178
    },
    {
      "epoch": 6.370830535888672e-05,
      "model_forward_time": 1.8363902568817139,
      "step": 66803
    },
    {
      "epoch": 6.370830535888672e-05,
      "step": 66803,
      "training_step_time": 7.163247108459473
    },
    {
      "epoch": 6.370925903320313e-05,
      "model_forward_time": 1.800931453704834,
      "step": 66804
    },
    {
      "epoch": 6.370925903320313e-05,
      "step": 66804,
      "training_step_time": 8.572571516036987
    },
    {
      "epoch": 6.371021270751954e-05,
      "model_forward_time": 1.8122544288635254,
      "step": 66805
    },
    {
      "epoch": 6.371021270751954e-05,
      "step": 66805,
      "training_step_time": 7.684630870819092
    },
    {
      "epoch": 6.371116638183594e-05,
      "model_forward_time": 1.8151493072509766,
      "step": 66806
    },
    {
      "epoch": 6.371116638183594e-05,
      "step": 66806,
      "training_step_time": 7.732774019241333
    },
    {
      "epoch": 6.371212005615235e-05,
      "model_forward_time": 1.8215582370758057,
      "step": 66807
    },
    {
      "epoch": 6.371212005615235e-05,
      "step": 66807,
      "training_step_time": 8.401943683624268
    },
    {
      "epoch": 6.371307373046876e-05,
      "model_forward_time": 1.811633825302124,
      "step": 66808
    },
    {
      "epoch": 6.371307373046876e-05,
      "step": 66808,
      "training_step_time": 7.838966608047485
    },
    {
      "epoch": 6.371402740478515e-05,
      "model_forward_time": 1.8209795951843262,
      "step": 66809
    },
    {
      "epoch": 6.371402740478515e-05,
      "step": 66809,
      "training_step_time": 9.536806344985962
    },
    {
      "epoch": 6.371498107910156e-05,
      "grad_norm": 0.09338313341140747,
      "learning_rate": 2.7213596382774023e-06,
      "loss": 0.0577,
      "step": 66810
    },
    {
      "epoch": 6.371498107910156e-05,
      "model_forward_time": 1.8096823692321777,
      "step": 66810
    },
    {
      "dynamics_loss_avg": 0.05996456053107977,
      "epoch": 6.371498107910156e-05,
      "step": 66810
    },
    {
      "action_loss_avg": 0.0015160616865614429,
      "epoch": 6.371498107910156e-05,
      "step": 66810
    },
    {
      "epoch": 6.371498107910156e-05,
      "step": 66810,
      "training_step_time": 8.648699760437012
    },
    {
      "epoch": 6.371593475341798e-05,
      "model_forward_time": 2.3572890758514404,
      "step": 66811
    },
    {
      "epoch": 6.371593475341798e-05,
      "step": 66811,
      "training_step_time": 7.806867837905884
    },
    {
      "epoch": 6.371688842773437e-05,
      "model_forward_time": 1.829557180404663,
      "step": 66812
    },
    {
      "epoch": 6.371688842773437e-05,
      "step": 66812,
      "training_step_time": 7.757966041564941
    },
    {
      "epoch": 6.371784210205078e-05,
      "model_forward_time": 1.8128459453582764,
      "step": 66813
    },
    {
      "epoch": 6.371784210205078e-05,
      "step": 66813,
      "training_step_time": 7.696094274520874
    },
    {
      "epoch": 6.371879577636718e-05,
      "model_forward_time": 1.8041064739227295,
      "step": 66814
    },
    {
      "epoch": 6.371879577636718e-05,
      "step": 66814,
      "training_step_time": 7.7397520542144775
    },
    {
      "epoch": 6.371974945068359e-05,
      "model_forward_time": 1.8125555515289307,
      "step": 66815
    },
    {
      "epoch": 6.371974945068359e-05,
      "step": 66815,
      "training_step_time": 7.74045991897583
    },
    {
      "epoch": 6.3720703125e-05,
      "model_forward_time": 1.4470646381378174,
      "step": 66816
    },
    {
      "epoch": 6.3720703125e-05,
      "step": 66816,
      "training_step_time": 8.735471725463867
    },
    {
      "epoch": 6.37216567993164e-05,
      "model_forward_time": 1.8182311058044434,
      "step": 66817
    },
    {
      "epoch": 6.37216567993164e-05,
      "step": 66817,
      "training_step_time": 7.769812822341919
    },
    {
      "epoch": 6.372261047363281e-05,
      "model_forward_time": 1.4575450420379639,
      "step": 66818
    },
    {
      "epoch": 6.372261047363281e-05,
      "step": 66818,
      "training_step_time": 7.97352933883667
    },
    {
      "epoch": 6.372356414794922e-05,
      "model_forward_time": 1.8252406120300293,
      "step": 66819
    },
    {
      "epoch": 6.372356414794922e-05,
      "step": 66819,
      "training_step_time": 7.8461127281188965
    },
    {
      "epoch": 6.372451782226562e-05,
      "grad_norm": 0.1048017293214798,
      "learning_rate": 2.7198879779144147e-06,
      "loss": 0.0594,
      "step": 66820
    },
    {
      "epoch": 6.372451782226562e-05,
      "model_forward_time": 1.83107328414917,
      "step": 66820
    },
    {
      "dynamics_loss_avg": 0.059834665060043334,
      "epoch": 6.372451782226562e-05,
      "step": 66820
    },
    {
      "action_loss_avg": 0.001597813208354637,
      "epoch": 6.372451782226562e-05,
      "step": 66820
    },
    {
      "epoch": 6.372451782226562e-05,
      "step": 66820,
      "training_step_time": 8.33404541015625
    },
    {
      "epoch": 6.372547149658203e-05,
      "model_forward_time": 2.341973066329956,
      "step": 66821
    },
    {
      "epoch": 6.372547149658203e-05,
      "step": 66821,
      "training_step_time": 7.788142681121826
    },
    {
      "epoch": 6.372642517089844e-05,
      "model_forward_time": 1.8318994045257568,
      "step": 66822
    },
    {
      "epoch": 6.372642517089844e-05,
      "step": 66822,
      "training_step_time": 7.702252626419067
    },
    {
      "epoch": 6.372737884521484e-05,
      "model_forward_time": 1.8275105953216553,
      "step": 66823
    },
    {
      "epoch": 6.372737884521484e-05,
      "step": 66823,
      "training_step_time": 7.674444913864136
    },
    {
      "epoch": 6.372833251953125e-05,
      "model_forward_time": 1.832763671875,
      "step": 66824
    },
    {
      "epoch": 6.372833251953125e-05,
      "step": 66824,
      "training_step_time": 7.674579381942749
    },
    {
      "epoch": 6.372928619384766e-05,
      "model_forward_time": 1.835252046585083,
      "step": 66825
    },
    {
      "epoch": 6.372928619384766e-05,
      "step": 66825,
      "training_step_time": 8.742769241333008
    },
    {
      "epoch": 6.373023986816406e-05,
      "model_forward_time": 1.8369672298431396,
      "step": 66826
    },
    {
      "epoch": 6.373023986816406e-05,
      "step": 66826,
      "training_step_time": 7.7019970417022705
    },
    {
      "epoch": 6.373119354248047e-05,
      "model_forward_time": 1.8300237655639648,
      "step": 66827
    },
    {
      "epoch": 6.373119354248047e-05,
      "step": 66827,
      "training_step_time": 8.023046255111694
    },
    {
      "epoch": 6.373214721679688e-05,
      "model_forward_time": 1.8514349460601807,
      "step": 66828
    },
    {
      "epoch": 6.373214721679688e-05,
      "step": 66828,
      "training_step_time": 8.367714405059814
    },
    {
      "epoch": 6.373310089111328e-05,
      "model_forward_time": 1.8444218635559082,
      "step": 66829
    },
    {
      "epoch": 6.373310089111328e-05,
      "step": 66829,
      "training_step_time": 7.791104078292847
    },
    {
      "epoch": 6.373405456542969e-05,
      "grad_norm": 0.08417569845914841,
      "learning_rate": 2.718416566901049e-06,
      "loss": 0.0542,
      "step": 66830
    },
    {
      "epoch": 6.373405456542969e-05,
      "model_forward_time": 2.3792264461517334,
      "step": 66830
    },
    {
      "dynamics_loss_avg": 0.05954143637791276,
      "epoch": 6.373405456542969e-05,
      "step": 66830
    },
    {
      "action_loss_avg": 0.002466706099221483,
      "epoch": 6.373405456542969e-05,
      "step": 66830
    },
    {
      "epoch": 6.373405456542969e-05,
      "step": 66830,
      "training_step_time": 7.762409210205078
    },
    {
      "epoch": 6.37350082397461e-05,
      "model_forward_time": 1.4749155044555664,
      "step": 66831
    },
    {
      "epoch": 6.37350082397461e-05,
      "step": 66831,
      "training_step_time": 7.637732982635498
    },
    {
      "epoch": 6.37359619140625e-05,
      "model_forward_time": 1.8382680416107178,
      "step": 66832
    },
    {
      "epoch": 6.37359619140625e-05,
      "step": 66832,
      "training_step_time": 7.563746213912964
    },
    {
      "epoch": 6.37369155883789e-05,
      "model_forward_time": 1.8313922882080078,
      "step": 66833
    },
    {
      "epoch": 6.37369155883789e-05,
      "step": 66833,
      "training_step_time": 7.800583124160767
    },
    {
      "epoch": 6.373786926269532e-05,
      "model_forward_time": 1.8361682891845703,
      "step": 66834
    },
    {
      "epoch": 6.373786926269532e-05,
      "step": 66834,
      "training_step_time": 7.784132957458496
    },
    {
      "epoch": 6.373882293701171e-05,
      "model_forward_time": 1.8401563167572021,
      "step": 66835
    },
    {
      "epoch": 6.373882293701171e-05,
      "step": 66835,
      "training_step_time": 8.321521759033203
    },
    {
      "epoch": 6.373977661132812e-05,
      "model_forward_time": 1.8431932926177979,
      "step": 66836
    },
    {
      "epoch": 6.373977661132812e-05,
      "step": 66836,
      "training_step_time": 8.568275928497314
    },
    {
      "epoch": 6.374073028564453e-05,
      "model_forward_time": 1.8427667617797852,
      "step": 66837
    },
    {
      "epoch": 6.374073028564453e-05,
      "step": 66837,
      "training_step_time": 8.615046501159668
    },
    {
      "epoch": 6.374168395996093e-05,
      "model_forward_time": 1.446674108505249,
      "step": 66838
    },
    {
      "epoch": 6.374168395996093e-05,
      "step": 66838,
      "training_step_time": 7.704989671707153
    },
    {
      "epoch": 6.374263763427734e-05,
      "model_forward_time": 1.8358702659606934,
      "step": 66839
    },
    {
      "epoch": 6.374263763427734e-05,
      "step": 66839,
      "training_step_time": 8.899153232574463
    },
    {
      "epoch": 6.374359130859375e-05,
      "grad_norm": 0.1042943075299263,
      "learning_rate": 2.716945405398218e-06,
      "loss": 0.0554,
      "step": 66840
    },
    {
      "epoch": 6.374359130859375e-05,
      "model_forward_time": 1.8203144073486328,
      "step": 66840
    },
    {
      "dynamics_loss_avg": 0.04241360649466515,
      "epoch": 6.374359130859375e-05,
      "step": 66840
    },
    {
      "action_loss_avg": 0.0015927649568766356,
      "epoch": 6.374359130859375e-05,
      "step": 66840
    },
    {
      "epoch": 6.374359130859375e-05,
      "step": 66840,
      "training_step_time": 7.721620082855225
    },
    {
      "epoch": 6.374454498291015e-05,
      "model_forward_time": 1.84201979637146,
      "step": 66841
    },
    {
      "epoch": 6.374454498291015e-05,
      "step": 66841,
      "training_step_time": 7.598023414611816
    },
    {
      "epoch": 6.374549865722656e-05,
      "model_forward_time": 1.8231494426727295,
      "step": 66842
    },
    {
      "epoch": 6.374549865722656e-05,
      "step": 66842,
      "training_step_time": 7.939241886138916
    },
    {
      "epoch": 6.374645233154297e-05,
      "model_forward_time": 1.83817720413208,
      "step": 66843
    },
    {
      "epoch": 6.374645233154297e-05,
      "step": 66843,
      "training_step_time": 7.711137771606445
    },
    {
      "epoch": 6.374740600585937e-05,
      "model_forward_time": 1.8298060894012451,
      "step": 66844
    },
    {
      "epoch": 6.374740600585937e-05,
      "step": 66844,
      "training_step_time": 8.439368486404419
    },
    {
      "epoch": 6.374835968017578e-05,
      "model_forward_time": 1.4694068431854248,
      "step": 66845
    },
    {
      "epoch": 6.374835968017578e-05,
      "step": 66845,
      "training_step_time": 8.835155725479126
    },
    {
      "epoch": 6.374931335449219e-05,
      "model_forward_time": 1.1189486980438232,
      "step": 66846
    },
    {
      "epoch": 6.374931335449219e-05,
      "step": 66846,
      "training_step_time": 8.732482671737671
    },
    {
      "epoch": 6.375026702880859e-05,
      "model_forward_time": 1.824373722076416,
      "step": 66847
    },
    {
      "epoch": 6.375026702880859e-05,
      "step": 66847,
      "training_step_time": 8.71545672416687
    },
    {
      "epoch": 6.3751220703125e-05,
      "model_forward_time": 1.83431077003479,
      "step": 66848
    },
    {
      "epoch": 6.3751220703125e-05,
      "step": 66848,
      "training_step_time": 8.793635368347168
    },
    {
      "epoch": 6.375217437744141e-05,
      "model_forward_time": 2.3774521350860596,
      "step": 66849
    },
    {
      "epoch": 6.375217437744141e-05,
      "step": 66849,
      "training_step_time": 7.6030495166778564
    },
    {
      "epoch": 6.375312805175781e-05,
      "grad_norm": 0.08407783508300781,
      "learning_rate": 2.7154744935668033e-06,
      "loss": 0.0573,
      "step": 66850
    },
    {
      "epoch": 6.375312805175781e-05,
      "model_forward_time": 1.1302433013916016,
      "step": 66850
    },
    {
      "dynamics_loss_avg": 0.04944325722754002,
      "epoch": 6.375312805175781e-05,
      "step": 66850
    },
    {
      "action_loss_avg": 0.0008712525130249559,
      "epoch": 6.375312805175781e-05,
      "step": 66850
    },
    {
      "epoch": 6.375312805175781e-05,
      "step": 66850,
      "training_step_time": 7.470239877700806
    },
    {
      "epoch": 6.375408172607422e-05,
      "model_forward_time": 1.8439304828643799,
      "step": 66851
    },
    {
      "epoch": 6.375408172607422e-05,
      "step": 66851,
      "training_step_time": 7.633549451828003
    },
    {
      "epoch": 6.375503540039063e-05,
      "model_forward_time": 1.8240807056427002,
      "step": 66852
    },
    {
      "epoch": 6.375503540039063e-05,
      "step": 66852,
      "training_step_time": 7.547442197799683
    },
    {
      "epoch": 6.375598907470703e-05,
      "model_forward_time": 1.8437902927398682,
      "step": 66853
    },
    {
      "epoch": 6.375598907470703e-05,
      "step": 66853,
      "training_step_time": 9.01297116279602
    },
    {
      "epoch": 6.375694274902344e-05,
      "model_forward_time": 1.8336946964263916,
      "step": 66854
    },
    {
      "epoch": 6.375694274902344e-05,
      "step": 66854,
      "training_step_time": 9.177125453948975
    },
    {
      "epoch": 6.375789642333985e-05,
      "model_forward_time": 1.828096628189087,
      "step": 66855
    },
    {
      "epoch": 6.375789642333985e-05,
      "step": 66855,
      "training_step_time": 8.884721279144287
    },
    {
      "epoch": 6.375885009765624e-05,
      "model_forward_time": 1.8352196216583252,
      "step": 66856
    },
    {
      "epoch": 6.375885009765624e-05,
      "step": 66856,
      "training_step_time": 7.680165767669678
    },
    {
      "epoch": 6.375980377197266e-05,
      "model_forward_time": 1.833745002746582,
      "step": 66857
    },
    {
      "epoch": 6.375980377197266e-05,
      "step": 66857,
      "training_step_time": 7.578290224075317
    },
    {
      "epoch": 6.376075744628907e-05,
      "model_forward_time": 1.8274593353271484,
      "step": 66858
    },
    {
      "epoch": 6.376075744628907e-05,
      "step": 66858,
      "training_step_time": 8.822925329208374
    },
    {
      "epoch": 6.376171112060546e-05,
      "model_forward_time": 2.378509998321533,
      "step": 66859
    },
    {
      "epoch": 6.376171112060546e-05,
      "step": 66859,
      "training_step_time": 7.623351335525513
    },
    {
      "epoch": 6.376266479492187e-05,
      "grad_norm": 0.10967817902565002,
      "learning_rate": 2.714003831567665e-06,
      "loss": 0.0572,
      "step": 66860
    },
    {
      "epoch": 6.376266479492187e-05,
      "model_forward_time": 1.8286094665527344,
      "step": 66860
    },
    {
      "dynamics_loss_avg": 0.07643986009061336,
      "epoch": 6.376266479492187e-05,
      "step": 66860
    },
    {
      "action_loss_avg": 0.0021771699539385736,
      "epoch": 6.376266479492187e-05,
      "step": 66860
    },
    {
      "epoch": 6.376266479492187e-05,
      "step": 66860,
      "training_step_time": 7.647686004638672
    },
    {
      "epoch": 6.376361846923829e-05,
      "model_forward_time": 1.4804282188415527,
      "step": 66861
    },
    {
      "epoch": 6.376361846923829e-05,
      "step": 66861,
      "training_step_time": 7.4365856647491455
    },
    {
      "epoch": 6.376457214355468e-05,
      "model_forward_time": 1.8444066047668457,
      "step": 66862
    },
    {
      "epoch": 6.376457214355468e-05,
      "step": 66862,
      "training_step_time": 7.650196313858032
    },
    {
      "epoch": 6.37655258178711e-05,
      "model_forward_time": 1.8154804706573486,
      "step": 66863
    },
    {
      "epoch": 6.37655258178711e-05,
      "step": 66863,
      "training_step_time": 8.577077388763428
    },
    {
      "epoch": 6.37664794921875e-05,
      "model_forward_time": 1.8282558917999268,
      "step": 66864
    },
    {
      "epoch": 6.37664794921875e-05,
      "step": 66864,
      "training_step_time": 8.733900785446167
    },
    {
      "epoch": 6.37674331665039e-05,
      "model_forward_time": 1.8311781883239746,
      "step": 66865
    },
    {
      "epoch": 6.37674331665039e-05,
      "step": 66865,
      "training_step_time": 9.070772171020508
    },
    {
      "epoch": 6.376838684082031e-05,
      "model_forward_time": 1.842580795288086,
      "step": 66866
    },
    {
      "epoch": 6.376838684082031e-05,
      "step": 66866,
      "training_step_time": 7.594834566116333
    },
    {
      "epoch": 6.376934051513672e-05,
      "model_forward_time": 1.8234014511108398,
      "step": 66867
    },
    {
      "epoch": 6.376934051513672e-05,
      "step": 66867,
      "training_step_time": 7.585423707962036
    },
    {
      "epoch": 6.377029418945312e-05,
      "model_forward_time": 2.3765339851379395,
      "step": 66868
    },
    {
      "epoch": 6.377029418945312e-05,
      "step": 66868,
      "training_step_time": 7.669856071472168
    },
    {
      "epoch": 6.377124786376953e-05,
      "model_forward_time": 1.8330190181732178,
      "step": 66869
    },
    {
      "epoch": 6.377124786376953e-05,
      "step": 66869,
      "training_step_time": 7.528295040130615
    },
    {
      "epoch": 6.377220153808594e-05,
      "grad_norm": 0.10015812516212463,
      "learning_rate": 2.712533419561628e-06,
      "loss": 0.0561,
      "step": 66870
    },
    {
      "epoch": 6.377220153808594e-05,
      "model_forward_time": 1.8282670974731445,
      "step": 66870
    },
    {
      "dynamics_loss_avg": 0.04750144761055708,
      "epoch": 6.377220153808594e-05,
      "step": 66870
    },
    {
      "action_loss_avg": 0.0015952338289935142,
      "epoch": 6.377220153808594e-05,
      "step": 66870
    },
    {
      "epoch": 6.377220153808594e-05,
      "step": 66870,
      "training_step_time": 7.583472013473511
    },
    {
      "epoch": 6.377315521240234e-05,
      "model_forward_time": 1.8337926864624023,
      "step": 66871
    },
    {
      "epoch": 6.377315521240234e-05,
      "step": 66871,
      "training_step_time": 7.5970299243927
    },
    {
      "epoch": 6.377410888671875e-05,
      "model_forward_time": 1.8171768188476562,
      "step": 66872
    },
    {
      "epoch": 6.377410888671875e-05,
      "step": 66872,
      "training_step_time": 10.705747842788696
    },
    {
      "epoch": 6.377506256103516e-05,
      "model_forward_time": 1.4644277095794678,
      "step": 66873
    },
    {
      "epoch": 6.377506256103516e-05,
      "step": 66873,
      "training_step_time": 8.766312599182129
    },
    {
      "epoch": 6.377601623535156e-05,
      "model_forward_time": 1.8366210460662842,
      "step": 66874
    },
    {
      "epoch": 6.377601623535156e-05,
      "step": 66874,
      "training_step_time": 8.197072744369507
    },
    {
      "epoch": 6.377696990966797e-05,
      "model_forward_time": 1.8212034702301025,
      "step": 66875
    },
    {
      "epoch": 6.377696990966797e-05,
      "step": 66875,
      "training_step_time": 7.678798675537109
    },
    {
      "epoch": 6.377792358398438e-05,
      "model_forward_time": 1.835167646408081,
      "step": 66876
    },
    {
      "epoch": 6.377792358398438e-05,
      "step": 66876,
      "training_step_time": 7.50808048248291
    },
    {
      "epoch": 6.377887725830078e-05,
      "model_forward_time": 2.3795182704925537,
      "step": 66877
    },
    {
      "epoch": 6.377887725830078e-05,
      "step": 66877,
      "training_step_time": 7.762833833694458
    },
    {
      "epoch": 6.377983093261719e-05,
      "model_forward_time": 1.8276293277740479,
      "step": 66878
    },
    {
      "epoch": 6.377983093261719e-05,
      "step": 66878,
      "training_step_time": 8.382403135299683
    },
    {
      "epoch": 6.37807846069336e-05,
      "model_forward_time": 1.8346149921417236,
      "step": 66879
    },
    {
      "epoch": 6.37807846069336e-05,
      "step": 66879,
      "training_step_time": 7.57727575302124
    },
    {
      "epoch": 6.378173828125e-05,
      "grad_norm": 0.11173485219478607,
      "learning_rate": 2.7110632577094996e-06,
      "loss": 0.0535,
      "step": 66880
    },
    {
      "epoch": 6.378173828125e-05,
      "model_forward_time": 1.8231110572814941,
      "step": 66880
    },
    {
      "dynamics_loss_avg": 0.05467689000070095,
      "epoch": 6.378173828125e-05,
      "step": 66880
    },
    {
      "action_loss_avg": 0.0010565506061539053,
      "epoch": 6.378173828125e-05,
      "step": 66880
    },
    {
      "epoch": 6.378173828125e-05,
      "step": 66880,
      "training_step_time": 7.5450170040130615
    },
    {
      "epoch": 6.37826919555664e-05,
      "model_forward_time": 1.8155388832092285,
      "step": 66881
    },
    {
      "epoch": 6.37826919555664e-05,
      "step": 66881,
      "training_step_time": 8.616719961166382
    },
    {
      "epoch": 6.378364562988282e-05,
      "model_forward_time": 1.8285396099090576,
      "step": 66882
    },
    {
      "epoch": 6.378364562988282e-05,
      "step": 66882,
      "training_step_time": 9.032296180725098
    },
    {
      "epoch": 6.378459930419921e-05,
      "model_forward_time": 1.8204920291900635,
      "step": 66883
    },
    {
      "epoch": 6.378459930419921e-05,
      "step": 66883,
      "training_step_time": 9.384360551834106
    },
    {
      "epoch": 6.378555297851563e-05,
      "model_forward_time": 1.8221611976623535,
      "step": 66884
    },
    {
      "epoch": 6.378555297851563e-05,
      "step": 66884,
      "training_step_time": 7.807164430618286
    },
    {
      "epoch": 6.378650665283204e-05,
      "model_forward_time": 1.8380424976348877,
      "step": 66885
    },
    {
      "epoch": 6.378650665283204e-05,
      "step": 66885,
      "training_step_time": 7.568529367446899
    },
    {
      "epoch": 6.378746032714843e-05,
      "model_forward_time": 2.400186538696289,
      "step": 66886
    },
    {
      "epoch": 6.378746032714843e-05,
      "step": 66886,
      "training_step_time": 7.746136426925659
    },
    {
      "epoch": 6.378841400146484e-05,
      "model_forward_time": 1.8319292068481445,
      "step": 66887
    },
    {
      "epoch": 6.378841400146484e-05,
      "step": 66887,
      "training_step_time": 7.404566526412964
    },
    {
      "epoch": 6.378936767578125e-05,
      "model_forward_time": 1.8429336547851562,
      "step": 66888
    },
    {
      "epoch": 6.378936767578125e-05,
      "step": 66888,
      "training_step_time": 8.484992980957031
    },
    {
      "epoch": 6.379032135009765e-05,
      "model_forward_time": 1.837859869003296,
      "step": 66889
    },
    {
      "epoch": 6.379032135009765e-05,
      "step": 66889,
      "training_step_time": 8.621610641479492
    },
    {
      "epoch": 6.379127502441406e-05,
      "grad_norm": 0.09739638864994049,
      "learning_rate": 2.7095933461720507e-06,
      "loss": 0.0567,
      "step": 66890
    },
    {
      "epoch": 6.379127502441406e-05,
      "model_forward_time": 1.8389084339141846,
      "step": 66890
    },
    {
      "dynamics_loss_avg": 0.06872187051922082,
      "epoch": 6.379127502441406e-05,
      "step": 66890
    },
    {
      "action_loss_avg": 0.0014074557286221534,
      "epoch": 6.379127502441406e-05,
      "step": 66890
    },
    {
      "epoch": 6.379127502441406e-05,
      "step": 66890,
      "training_step_time": 8.361660242080688
    },
    {
      "epoch": 6.379222869873047e-05,
      "model_forward_time": 1.8275423049926758,
      "step": 66891
    },
    {
      "epoch": 6.379222869873047e-05,
      "step": 66891,
      "training_step_time": 8.74968433380127
    },
    {
      "epoch": 6.379318237304687e-05,
      "model_forward_time": 1.8316385746002197,
      "step": 66892
    },
    {
      "epoch": 6.379318237304687e-05,
      "step": 66892,
      "training_step_time": 7.624785423278809
    },
    {
      "epoch": 6.379413604736328e-05,
      "model_forward_time": 1.8326029777526855,
      "step": 66893
    },
    {
      "epoch": 6.379413604736328e-05,
      "step": 66893,
      "training_step_time": 7.670889854431152
    },
    {
      "epoch": 6.379508972167969e-05,
      "model_forward_time": 1.8436834812164307,
      "step": 66894
    },
    {
      "epoch": 6.379508972167969e-05,
      "step": 66894,
      "training_step_time": 7.4650702476501465
    },
    {
      "epoch": 6.379604339599609e-05,
      "model_forward_time": 1.8448207378387451,
      "step": 66895
    },
    {
      "epoch": 6.379604339599609e-05,
      "step": 66895,
      "training_step_time": 8.663795471191406
    },
    {
      "epoch": 6.37969970703125e-05,
      "model_forward_time": 1.8226878643035889,
      "step": 66896
    },
    {
      "epoch": 6.37969970703125e-05,
      "step": 66896,
      "training_step_time": 7.709262132644653
    },
    {
      "epoch": 6.379795074462891e-05,
      "model_forward_time": 1.8483250141143799,
      "step": 66897
    },
    {
      "epoch": 6.379795074462891e-05,
      "step": 66897,
      "training_step_time": 7.602006196975708
    },
    {
      "epoch": 6.379890441894531e-05,
      "model_forward_time": 1.8429069519042969,
      "step": 66898
    },
    {
      "epoch": 6.379890441894531e-05,
      "step": 66898,
      "training_step_time": 7.65768027305603
    },
    {
      "epoch": 6.379985809326172e-05,
      "model_forward_time": 1.8411200046539307,
      "step": 66899
    },
    {
      "epoch": 6.379985809326172e-05,
      "step": 66899,
      "training_step_time": 8.356546640396118
    },
    {
      "epoch": 6.380081176757813e-05,
      "grad_norm": 0.0929456427693367,
      "learning_rate": 2.7081236851100323e-06,
      "loss": 0.0531,
      "step": 66900
    },
    {
      "epoch": 6.380081176757813e-05,
      "model_forward_time": 1.8401398658752441,
      "step": 66900
    },
    {
      "dynamics_loss_avg": 0.056764856725931165,
      "epoch": 6.380081176757813e-05,
      "step": 66900
    },
    {
      "action_loss_avg": 0.0016206118278205394,
      "epoch": 6.380081176757813e-05,
      "step": 66900
    },
    {
      "epoch": 6.380081176757813e-05,
      "step": 66900,
      "training_step_time": 10.201710224151611
    },
    {
      "epoch": 6.380176544189453e-05,
      "model_forward_time": 1.8202552795410156,
      "step": 66901
    },
    {
      "epoch": 6.380176544189453e-05,
      "step": 66901,
      "training_step_time": 10.327020406723022
    },
    {
      "epoch": 6.380271911621094e-05,
      "model_forward_time": 1.828070878982544,
      "step": 66902
    },
    {
      "epoch": 6.380271911621094e-05,
      "step": 66902,
      "training_step_time": 8.151004791259766
    },
    {
      "epoch": 6.380367279052735e-05,
      "model_forward_time": 1.8149232864379883,
      "step": 66903
    },
    {
      "epoch": 6.380367279052735e-05,
      "step": 66903,
      "training_step_time": 8.105080366134644
    },
    {
      "epoch": 6.380462646484375e-05,
      "model_forward_time": 1.8289515972137451,
      "step": 66904
    },
    {
      "epoch": 6.380462646484375e-05,
      "step": 66904,
      "training_step_time": 7.519967317581177
    },
    {
      "epoch": 6.380558013916016e-05,
      "model_forward_time": 2.3702211380004883,
      "step": 66905
    },
    {
      "epoch": 6.380558013916016e-05,
      "step": 66905,
      "training_step_time": 7.759744882583618
    },
    {
      "epoch": 6.380653381347657e-05,
      "model_forward_time": 1.826019048690796,
      "step": 66906
    },
    {
      "epoch": 6.380653381347657e-05,
      "step": 66906,
      "training_step_time": 7.6093058586120605
    },
    {
      "epoch": 6.380748748779297e-05,
      "model_forward_time": 1.8224713802337646,
      "step": 66907
    },
    {
      "epoch": 6.380748748779297e-05,
      "step": 66907,
      "training_step_time": 8.148940086364746
    },
    {
      "epoch": 6.380844116210938e-05,
      "model_forward_time": 1.8282396793365479,
      "step": 66908
    },
    {
      "epoch": 6.380844116210938e-05,
      "step": 66908,
      "training_step_time": 7.426234006881714
    },
    {
      "epoch": 6.380939483642579e-05,
      "model_forward_time": 1.8259913921356201,
      "step": 66909
    },
    {
      "epoch": 6.380939483642579e-05,
      "step": 66909,
      "training_step_time": 8.602423906326294
    },
    {
      "epoch": 6.381034851074218e-05,
      "grad_norm": 0.12184806913137436,
      "learning_rate": 2.706654274684159e-06,
      "loss": 0.0545,
      "step": 66910
    },
    {
      "epoch": 6.381034851074218e-05,
      "model_forward_time": 1.836667776107788,
      "step": 66910
    },
    {
      "dynamics_loss_avg": 0.056847217865288256,
      "epoch": 6.381034851074218e-05,
      "step": 66910
    },
    {
      "action_loss_avg": 0.0018525870633311569,
      "epoch": 6.381034851074218e-05,
      "step": 66910
    },
    {
      "epoch": 6.381034851074218e-05,
      "step": 66910,
      "training_step_time": 7.865277051925659
    },
    {
      "epoch": 6.38113021850586e-05,
      "model_forward_time": 1.830876350402832,
      "step": 66911
    },
    {
      "epoch": 6.38113021850586e-05,
      "step": 66911,
      "training_step_time": 7.67415976524353
    },
    {
      "epoch": 6.3812255859375e-05,
      "model_forward_time": 1.827174186706543,
      "step": 66912
    },
    {
      "epoch": 6.3812255859375e-05,
      "step": 66912,
      "training_step_time": 8.573534488677979
    },
    {
      "epoch": 6.38132095336914e-05,
      "model_forward_time": 1.8300328254699707,
      "step": 66913
    },
    {
      "epoch": 6.38132095336914e-05,
      "step": 66913,
      "training_step_time": 7.476374626159668
    },
    {
      "epoch": 6.381416320800781e-05,
      "model_forward_time": 2.3431849479675293,
      "step": 66914
    },
    {
      "epoch": 6.381416320800781e-05,
      "step": 66914,
      "training_step_time": 7.820376634597778
    },
    {
      "epoch": 6.381511688232422e-05,
      "model_forward_time": 1.8290443420410156,
      "step": 66915
    },
    {
      "epoch": 6.381511688232422e-05,
      "step": 66915,
      "training_step_time": 8.638584852218628
    },
    {
      "epoch": 6.381607055664062e-05,
      "model_forward_time": 1.469902515411377,
      "step": 66916
    },
    {
      "epoch": 6.381607055664062e-05,
      "step": 66916,
      "training_step_time": 7.474283456802368
    },
    {
      "epoch": 6.381702423095703e-05,
      "model_forward_time": 1.8208513259887695,
      "step": 66917
    },
    {
      "epoch": 6.381702423095703e-05,
      "step": 66917,
      "training_step_time": 7.621002674102783
    },
    {
      "epoch": 6.381797790527344e-05,
      "model_forward_time": 1.8381192684173584,
      "step": 66918
    },
    {
      "epoch": 6.381797790527344e-05,
      "step": 66918,
      "training_step_time": 9.31638240814209
    },
    {
      "epoch": 6.381893157958984e-05,
      "model_forward_time": 1.8336257934570312,
      "step": 66919
    },
    {
      "epoch": 6.381893157958984e-05,
      "step": 66919,
      "training_step_time": 9.849271059036255
    },
    {
      "epoch": 6.381988525390625e-05,
      "grad_norm": 0.09411561489105225,
      "learning_rate": 2.705185115055127e-06,
      "loss": 0.0525,
      "step": 66920
    },
    {
      "epoch": 6.381988525390625e-05,
      "model_forward_time": 1.831533670425415,
      "step": 66920
    },
    {
      "dynamics_loss_avg": 0.05874704495072365,
      "epoch": 6.381988525390625e-05,
      "step": 66920
    },
    {
      "action_loss_avg": 0.0016379528446123005,
      "epoch": 6.381988525390625e-05,
      "step": 66920
    },
    {
      "epoch": 6.381988525390625e-05,
      "step": 66920,
      "training_step_time": 7.7371838092803955
    },
    {
      "epoch": 6.382083892822266e-05,
      "model_forward_time": 1.8344700336456299,
      "step": 66921
    },
    {
      "epoch": 6.382083892822266e-05,
      "step": 66921,
      "training_step_time": 8.471742391586304
    },
    {
      "epoch": 6.382179260253906e-05,
      "model_forward_time": 1.8404500484466553,
      "step": 66922
    },
    {
      "epoch": 6.382179260253906e-05,
      "step": 66922,
      "training_step_time": 7.599745988845825
    },
    {
      "epoch": 6.382274627685547e-05,
      "model_forward_time": 2.3659253120422363,
      "step": 66923
    },
    {
      "epoch": 6.382274627685547e-05,
      "step": 66923,
      "training_step_time": 7.869597434997559
    },
    {
      "epoch": 6.382369995117188e-05,
      "model_forward_time": 1.8246097564697266,
      "step": 66924
    },
    {
      "epoch": 6.382369995117188e-05,
      "step": 66924,
      "training_step_time": 7.987417936325073
    },
    {
      "epoch": 6.382465362548828e-05,
      "model_forward_time": 1.8341257572174072,
      "step": 66925
    },
    {
      "epoch": 6.382465362548828e-05,
      "step": 66925,
      "training_step_time": 7.670248746871948
    },
    {
      "epoch": 6.382560729980469e-05,
      "model_forward_time": 1.831373691558838,
      "step": 66926
    },
    {
      "epoch": 6.382560729980469e-05,
      "step": 66926,
      "training_step_time": 7.732286214828491
    },
    {
      "epoch": 6.38265609741211e-05,
      "model_forward_time": 1.837967872619629,
      "step": 66927
    },
    {
      "epoch": 6.38265609741211e-05,
      "step": 66927,
      "training_step_time": 8.615860939025879
    },
    {
      "epoch": 6.38275146484375e-05,
      "model_forward_time": 1.8276214599609375,
      "step": 66928
    },
    {
      "epoch": 6.38275146484375e-05,
      "step": 66928,
      "training_step_time": 8.390106678009033
    },
    {
      "epoch": 6.382846832275391e-05,
      "model_forward_time": 1.827014684677124,
      "step": 66929
    },
    {
      "epoch": 6.382846832275391e-05,
      "step": 66929,
      "training_step_time": 8.730536937713623
    },
    {
      "epoch": 6.382942199707032e-05,
      "grad_norm": 0.07472213357686996,
      "learning_rate": 2.7037162063836e-06,
      "loss": 0.0552,
      "step": 66930
    },
    {
      "epoch": 6.382942199707032e-05,
      "model_forward_time": 1.8209400177001953,
      "step": 66930
    },
    {
      "dynamics_loss_avg": 0.04760145265609026,
      "epoch": 6.382942199707032e-05,
      "step": 66930
    },
    {
      "action_loss_avg": 0.0016755894204834477,
      "epoch": 6.382942199707032e-05,
      "step": 66930
    },
    {
      "epoch": 6.382942199707032e-05,
      "step": 66930,
      "training_step_time": 8.434210538864136
    },
    {
      "epoch": 6.383037567138672e-05,
      "model_forward_time": 1.8483695983886719,
      "step": 66931
    },
    {
      "epoch": 6.383037567138672e-05,
      "step": 66931,
      "training_step_time": 7.541675090789795
    },
    {
      "epoch": 6.383132934570313e-05,
      "model_forward_time": 2.3751957416534424,
      "step": 66932
    },
    {
      "epoch": 6.383132934570313e-05,
      "step": 66932,
      "training_step_time": 7.7324559688568115
    },
    {
      "epoch": 6.383228302001954e-05,
      "model_forward_time": 1.8289358615875244,
      "step": 66933
    },
    {
      "epoch": 6.383228302001954e-05,
      "step": 66933,
      "training_step_time": 7.714179992675781
    },
    {
      "epoch": 6.383323669433594e-05,
      "model_forward_time": 1.839385747909546,
      "step": 66934
    },
    {
      "epoch": 6.383323669433594e-05,
      "step": 66934,
      "training_step_time": 7.632727146148682
    },
    {
      "epoch": 6.383419036865235e-05,
      "model_forward_time": 1.8436329364776611,
      "step": 66935
    },
    {
      "epoch": 6.383419036865235e-05,
      "step": 66935,
      "training_step_time": 8.701263904571533
    },
    {
      "epoch": 6.383514404296876e-05,
      "model_forward_time": 1.8300206661224365,
      "step": 66936
    },
    {
      "epoch": 6.383514404296876e-05,
      "step": 66936,
      "training_step_time": 7.6733558177948
    },
    {
      "epoch": 6.383609771728515e-05,
      "model_forward_time": 1.8318119049072266,
      "step": 66937
    },
    {
      "epoch": 6.383609771728515e-05,
      "step": 66937,
      "training_step_time": 7.871580362319946
    },
    {
      "epoch": 6.383705139160156e-05,
      "model_forward_time": 1.8383324146270752,
      "step": 66938
    },
    {
      "epoch": 6.383705139160156e-05,
      "step": 66938,
      "training_step_time": 7.754469871520996
    },
    {
      "epoch": 6.383800506591796e-05,
      "model_forward_time": 1.8370676040649414,
      "step": 66939
    },
    {
      "epoch": 6.383800506591796e-05,
      "step": 66939,
      "training_step_time": 7.720545291900635
    },
    {
      "epoch": 6.383895874023437e-05,
      "grad_norm": 0.07906457781791687,
      "learning_rate": 2.702247548830218e-06,
      "loss": 0.0555,
      "step": 66940
    },
    {
      "epoch": 6.383895874023437e-05,
      "model_forward_time": 1.834028720855713,
      "step": 66940
    },
    {
      "dynamics_loss_avg": 0.04810748752206564,
      "epoch": 6.383895874023437e-05,
      "step": 66940
    },
    {
      "action_loss_avg": 0.0014606115611968562,
      "epoch": 6.383895874023437e-05,
      "step": 66940
    },
    {
      "epoch": 6.383895874023437e-05,
      "step": 66940,
      "training_step_time": 7.5921311378479
    },
    {
      "epoch": 6.383991241455078e-05,
      "model_forward_time": 2.3961598873138428,
      "step": 66941
    },
    {
      "epoch": 6.383991241455078e-05,
      "step": 66941,
      "training_step_time": 8.351754426956177
    },
    {
      "epoch": 6.384086608886718e-05,
      "model_forward_time": 1.462946891784668,
      "step": 66942
    },
    {
      "epoch": 6.384086608886718e-05,
      "step": 66942,
      "training_step_time": 7.768753528594971
    },
    {
      "epoch": 6.384181976318359e-05,
      "model_forward_time": 1.8347418308258057,
      "step": 66943
    },
    {
      "epoch": 6.384181976318359e-05,
      "step": 66943,
      "training_step_time": 7.714830160140991
    },
    {
      "epoch": 6.38427734375e-05,
      "model_forward_time": 1.4599292278289795,
      "step": 66944
    },
    {
      "epoch": 6.38427734375e-05,
      "step": 66944,
      "training_step_time": 9.020676374435425
    },
    {
      "epoch": 6.38437271118164e-05,
      "model_forward_time": 1.825340986251831,
      "step": 66945
    },
    {
      "epoch": 6.38437271118164e-05,
      "step": 66945,
      "training_step_time": 9.622084856033325
    },
    {
      "epoch": 6.384468078613281e-05,
      "model_forward_time": 1.8398468494415283,
      "step": 66946
    },
    {
      "epoch": 6.384468078613281e-05,
      "step": 66946,
      "training_step_time": 8.380695104598999
    },
    {
      "epoch": 6.384563446044922e-05,
      "model_forward_time": 1.8305189609527588,
      "step": 66947
    },
    {
      "epoch": 6.384563446044922e-05,
      "step": 66947,
      "training_step_time": 9.664652109146118
    },
    {
      "epoch": 6.384658813476562e-05,
      "model_forward_time": 1.834733247756958,
      "step": 66948
    },
    {
      "epoch": 6.384658813476562e-05,
      "step": 66948,
      "training_step_time": 7.705602169036865
    },
    {
      "epoch": 6.384754180908203e-05,
      "model_forward_time": 1.8326807022094727,
      "step": 66949
    },
    {
      "epoch": 6.384754180908203e-05,
      "step": 66949,
      "training_step_time": 8.399750232696533
    },
    {
      "epoch": 6.384849548339844e-05,
      "grad_norm": 0.0856301411986351,
      "learning_rate": 2.7007791425555884e-06,
      "loss": 0.0523,
      "step": 66950
    },
    {
      "epoch": 6.384849548339844e-05,
      "model_forward_time": 1.8452293872833252,
      "step": 66950
    },
    {
      "dynamics_loss_avg": 0.04799921680241823,
      "epoch": 6.384849548339844e-05,
      "step": 66950
    },
    {
      "action_loss_avg": 0.0016359134315280244,
      "epoch": 6.384849548339844e-05,
      "step": 66950
    },
    {
      "epoch": 6.384849548339844e-05,
      "step": 66950,
      "training_step_time": 7.587060213088989
    },
    {
      "epoch": 6.384944915771484e-05,
      "model_forward_time": 2.4061031341552734,
      "step": 66951
    },
    {
      "epoch": 6.384944915771484e-05,
      "step": 66951,
      "training_step_time": 7.682152271270752
    },
    {
      "epoch": 6.385040283203125e-05,
      "model_forward_time": 1.8458008766174316,
      "step": 66952
    },
    {
      "epoch": 6.385040283203125e-05,
      "step": 66952,
      "training_step_time": 7.635457515716553
    },
    {
      "epoch": 6.385135650634766e-05,
      "model_forward_time": 1.829707145690918,
      "step": 66953
    },
    {
      "epoch": 6.385135650634766e-05,
      "step": 66953,
      "training_step_time": 7.825549840927124
    },
    {
      "epoch": 6.385231018066406e-05,
      "model_forward_time": 1.842738151550293,
      "step": 66954
    },
    {
      "epoch": 6.385231018066406e-05,
      "step": 66954,
      "training_step_time": 7.646484851837158
    },
    {
      "epoch": 6.385326385498047e-05,
      "model_forward_time": 1.8268051147460938,
      "step": 66955
    },
    {
      "epoch": 6.385326385498047e-05,
      "step": 66955,
      "training_step_time": 8.642194509506226
    },
    {
      "epoch": 6.385421752929688e-05,
      "model_forward_time": 1.8324267864227295,
      "step": 66956
    },
    {
      "epoch": 6.385421752929688e-05,
      "step": 66956,
      "training_step_time": 10.221313714981079
    },
    {
      "epoch": 6.385517120361328e-05,
      "model_forward_time": 1.8337137699127197,
      "step": 66957
    },
    {
      "epoch": 6.385517120361328e-05,
      "step": 66957,
      "training_step_time": 8.79036569595337
    },
    {
      "epoch": 6.385612487792969e-05,
      "model_forward_time": 1.832460641860962,
      "step": 66958
    },
    {
      "epoch": 6.385612487792969e-05,
      "step": 66958,
      "training_step_time": 9.500466346740723
    },
    {
      "epoch": 6.38570785522461e-05,
      "model_forward_time": 1.8347432613372803,
      "step": 66959
    },
    {
      "epoch": 6.38570785522461e-05,
      "step": 66959,
      "training_step_time": 7.604629755020142
    },
    {
      "epoch": 6.38580322265625e-05,
      "grad_norm": 0.20937804877758026,
      "learning_rate": 2.6993109877202942e-06,
      "loss": 0.0562,
      "step": 66960
    },
    {
      "epoch": 6.38580322265625e-05,
      "model_forward_time": 2.3699848651885986,
      "step": 66960
    },
    {
      "dynamics_loss_avg": 0.06657429989427328,
      "epoch": 6.38580322265625e-05,
      "step": 66960
    },
    {
      "action_loss_avg": 0.0029095440811943265,
      "epoch": 6.38580322265625e-05,
      "step": 66960
    },
    {
      "epoch": 6.38580322265625e-05,
      "step": 66960,
      "training_step_time": 8.402157068252563
    },
    {
      "epoch": 6.38589859008789e-05,
      "model_forward_time": 1.8456671237945557,
      "step": 66961
    },
    {
      "epoch": 6.38589859008789e-05,
      "step": 66961,
      "training_step_time": 7.683854579925537
    },
    {
      "epoch": 6.385993957519532e-05,
      "model_forward_time": 1.8307418823242188,
      "step": 66962
    },
    {
      "epoch": 6.385993957519532e-05,
      "step": 66962,
      "training_step_time": 7.627712726593018
    },
    {
      "epoch": 6.386089324951171e-05,
      "model_forward_time": 1.833268642425537,
      "step": 66963
    },
    {
      "epoch": 6.386089324951171e-05,
      "step": 66963,
      "training_step_time": 8.965133428573608
    },
    {
      "epoch": 6.386184692382812e-05,
      "model_forward_time": 1.8313829898834229,
      "step": 66964
    },
    {
      "epoch": 6.386184692382812e-05,
      "step": 66964,
      "training_step_time": 7.531364679336548
    },
    {
      "epoch": 6.386280059814453e-05,
      "model_forward_time": 1.823636770248413,
      "step": 66965
    },
    {
      "epoch": 6.386280059814453e-05,
      "step": 66965,
      "training_step_time": 9.83986496925354
    },
    {
      "epoch": 6.386375427246093e-05,
      "model_forward_time": 1.828345775604248,
      "step": 66966
    },
    {
      "epoch": 6.386375427246093e-05,
      "step": 66966,
      "training_step_time": 7.715052127838135
    },
    {
      "epoch": 6.386470794677734e-05,
      "model_forward_time": 1.8345887660980225,
      "step": 66967
    },
    {
      "epoch": 6.386470794677734e-05,
      "step": 66967,
      "training_step_time": 8.21603775024414
    },
    {
      "epoch": 6.386566162109375e-05,
      "model_forward_time": 1.8380522727966309,
      "step": 66968
    },
    {
      "epoch": 6.386566162109375e-05,
      "step": 66968,
      "training_step_time": 8.527870178222656
    },
    {
      "epoch": 6.386661529541015e-05,
      "model_forward_time": 2.027644395828247,
      "step": 66969
    },
    {
      "epoch": 6.386661529541015e-05,
      "step": 66969,
      "training_step_time": 7.741440296173096
    },
    {
      "epoch": 6.386756896972656e-05,
      "grad_norm": 0.08846588432788849,
      "learning_rate": 2.6978430844848936e-06,
      "loss": 0.0526,
      "step": 66970
    },
    {
      "epoch": 6.386756896972656e-05,
      "model_forward_time": 1.8260509967803955,
      "step": 66970
    },
    {
      "dynamics_loss_avg": 0.04461613055318594,
      "epoch": 6.386756896972656e-05,
      "step": 66970
    },
    {
      "action_loss_avg": 0.0018823668855475263,
      "epoch": 6.386756896972656e-05,
      "step": 66970
    },
    {
      "epoch": 6.386756896972656e-05,
      "step": 66970,
      "training_step_time": 7.602635145187378
    },
    {
      "epoch": 6.386852264404297e-05,
      "model_forward_time": 1.8259367942810059,
      "step": 66971
    },
    {
      "epoch": 6.386852264404297e-05,
      "step": 66971,
      "training_step_time": 7.642656326293945
    },
    {
      "epoch": 6.386947631835937e-05,
      "model_forward_time": 1.8472278118133545,
      "step": 66972
    },
    {
      "epoch": 6.386947631835937e-05,
      "step": 66972,
      "training_step_time": 7.900136709213257
    },
    {
      "epoch": 6.387042999267578e-05,
      "model_forward_time": 1.8308770656585693,
      "step": 66973
    },
    {
      "epoch": 6.387042999267578e-05,
      "step": 66973,
      "training_step_time": 8.712551355361938
    },
    {
      "epoch": 6.387138366699219e-05,
      "model_forward_time": 1.8383996486663818,
      "step": 66974
    },
    {
      "epoch": 6.387138366699219e-05,
      "step": 66974,
      "training_step_time": 8.539741039276123
    },
    {
      "epoch": 6.387233734130859e-05,
      "model_forward_time": 1.832292079925537,
      "step": 66975
    },
    {
      "epoch": 6.387233734130859e-05,
      "step": 66975,
      "training_step_time": 7.680582523345947
    },
    {
      "epoch": 6.3873291015625e-05,
      "model_forward_time": 1.837313175201416,
      "step": 66976
    },
    {
      "epoch": 6.3873291015625e-05,
      "step": 66976,
      "training_step_time": 9.212304830551147
    },
    {
      "epoch": 6.387424468994141e-05,
      "model_forward_time": 1.8346288204193115,
      "step": 66977
    },
    {
      "epoch": 6.387424468994141e-05,
      "step": 66977,
      "training_step_time": 8.5334153175354
    },
    {
      "epoch": 6.387519836425781e-05,
      "model_forward_time": 2.3934829235076904,
      "step": 66978
    },
    {
      "epoch": 6.387519836425781e-05,
      "step": 66978,
      "training_step_time": 7.6800220012664795
    },
    {
      "epoch": 6.387615203857422e-05,
      "model_forward_time": 1.8305811882019043,
      "step": 66979
    },
    {
      "epoch": 6.387615203857422e-05,
      "step": 66979,
      "training_step_time": 7.8393495082855225
    },
    {
      "epoch": 6.387710571289063e-05,
      "grad_norm": 0.09424804151058197,
      "learning_rate": 2.69637543300991e-06,
      "loss": 0.0562,
      "step": 66980
    },
    {
      "epoch": 6.387710571289063e-05,
      "model_forward_time": 1.830368995666504,
      "step": 66980
    },
    {
      "dynamics_loss_avg": 0.05711419247090817,
      "epoch": 6.387710571289063e-05,
      "step": 66980
    },
    {
      "action_loss_avg": 0.0018262471814523451,
      "epoch": 6.387710571289063e-05,
      "step": 66980
    },
    {
      "epoch": 6.387710571289063e-05,
      "step": 66980,
      "training_step_time": 7.706009149551392
    },
    {
      "epoch": 6.387805938720703e-05,
      "model_forward_time": 1.8401000499725342,
      "step": 66981
    },
    {
      "epoch": 6.387805938720703e-05,
      "step": 66981,
      "training_step_time": 8.047248840332031
    },
    {
      "epoch": 6.387901306152344e-05,
      "model_forward_time": 1.8422706127166748,
      "step": 66982
    },
    {
      "epoch": 6.387901306152344e-05,
      "step": 66982,
      "training_step_time": 7.635295391082764
    },
    {
      "epoch": 6.387996673583985e-05,
      "model_forward_time": 1.847388744354248,
      "step": 66983
    },
    {
      "epoch": 6.387996673583985e-05,
      "step": 66983,
      "training_step_time": 10.302278757095337
    },
    {
      "epoch": 6.388092041015624e-05,
      "model_forward_time": 1.8446815013885498,
      "step": 66984
    },
    {
      "epoch": 6.388092041015624e-05,
      "step": 66984,
      "training_step_time": 8.274173259735107
    },
    {
      "epoch": 6.388187408447266e-05,
      "model_forward_time": 1.8332726955413818,
      "step": 66985
    },
    {
      "epoch": 6.388187408447266e-05,
      "step": 66985,
      "training_step_time": 9.46754765510559
    },
    {
      "epoch": 6.388282775878907e-05,
      "model_forward_time": 1.8262109756469727,
      "step": 66986
    },
    {
      "epoch": 6.388282775878907e-05,
      "step": 66986,
      "training_step_time": 8.61730432510376
    },
    {
      "epoch": 6.388378143310546e-05,
      "model_forward_time": 2.3792903423309326,
      "step": 66987
    },
    {
      "epoch": 6.388378143310546e-05,
      "step": 66987,
      "training_step_time": 7.463705539703369
    },
    {
      "epoch": 6.388473510742187e-05,
      "model_forward_time": 1.8385250568389893,
      "step": 66988
    },
    {
      "epoch": 6.388473510742187e-05,
      "step": 66988,
      "training_step_time": 7.675767183303833
    },
    {
      "epoch": 6.388568878173829e-05,
      "model_forward_time": 1.8269343376159668,
      "step": 66989
    },
    {
      "epoch": 6.388568878173829e-05,
      "step": 66989,
      "training_step_time": 7.68059778213501
    },
    {
      "epoch": 6.388664245605468e-05,
      "grad_norm": 0.09737236797809601,
      "learning_rate": 2.694908033455848e-06,
      "loss": 0.058,
      "step": 66990
    },
    {
      "epoch": 6.388664245605468e-05,
      "model_forward_time": 1.825505018234253,
      "step": 66990
    },
    {
      "dynamics_loss_avg": 0.07006811276078224,
      "epoch": 6.388664245605468e-05,
      "step": 66990
    },
    {
      "action_loss_avg": 0.0022713540063705297,
      "epoch": 6.388664245605468e-05,
      "step": 66990
    },
    {
      "epoch": 6.388664245605468e-05,
      "step": 66990,
      "training_step_time": 7.797386646270752
    },
    {
      "epoch": 6.38875961303711e-05,
      "model_forward_time": 1.8285324573516846,
      "step": 66991
    },
    {
      "epoch": 6.38875961303711e-05,
      "step": 66991,
      "training_step_time": 8.399180173873901
    },
    {
      "epoch": 6.38885498046875e-05,
      "model_forward_time": 1.8335943222045898,
      "step": 66992
    },
    {
      "epoch": 6.38885498046875e-05,
      "step": 66992,
      "training_step_time": 9.50541067123413
    },
    {
      "epoch": 6.38895034790039e-05,
      "model_forward_time": 1.8335363864898682,
      "step": 66993
    },
    {
      "epoch": 6.38895034790039e-05,
      "step": 66993,
      "training_step_time": 9.234286069869995
    },
    {
      "epoch": 6.389045715332031e-05,
      "model_forward_time": 1.8305745124816895,
      "step": 66994
    },
    {
      "epoch": 6.389045715332031e-05,
      "step": 66994,
      "training_step_time": 10.402118921279907
    },
    {
      "epoch": 6.389141082763672e-05,
      "model_forward_time": 1.8116040229797363,
      "step": 66995
    },
    {
      "epoch": 6.389141082763672e-05,
      "step": 66995,
      "training_step_time": 7.6278076171875
    },
    {
      "epoch": 6.389236450195312e-05,
      "model_forward_time": 1.8356032371520996,
      "step": 66996
    },
    {
      "epoch": 6.389236450195312e-05,
      "step": 66996,
      "training_step_time": 7.6106603145599365
    },
    {
      "epoch": 6.389331817626953e-05,
      "model_forward_time": 2.3949756622314453,
      "step": 66997
    },
    {
      "epoch": 6.389331817626953e-05,
      "step": 66997,
      "training_step_time": 7.753663778305054
    },
    {
      "epoch": 6.389427185058594e-05,
      "model_forward_time": 1.8368995189666748,
      "step": 66998
    },
    {
      "epoch": 6.389427185058594e-05,
      "step": 66998,
      "training_step_time": 7.948148965835571
    },
    {
      "epoch": 6.389522552490234e-05,
      "model_forward_time": 1.8339359760284424,
      "step": 66999
    },
    {
      "epoch": 6.389522552490234e-05,
      "step": 66999,
      "training_step_time": 7.828271389007568
    },
    {
      "epoch": 6.389617919921875e-05,
      "grad_norm": 0.10022782534360886,
      "learning_rate": 2.6934408859831735e-06,
      "loss": 0.0553,
      "step": 67000
    },
    {
      "epoch": 6.389617919921875e-05,
      "model_forward_time": 1.748814344406128,
      "step": 67000
    },
    {
      "dynamics_loss_avg": 0.055437554605305195,
      "epoch": 6.389617919921875e-05,
      "step": 67000
    },
    {
      "action_loss_avg": 0.002451603749068454,
      "epoch": 6.389617919921875e-05,
      "step": 67000
    },
    {
      "epoch": 6.389617919921875e-05,
      "step": 67000,
      "training_step_time": 10.061714887619019
    },
    {
      "epoch": 6.389713287353516e-05,
      "model_forward_time": 1.821533441543579,
      "step": 67001
    },
    {
      "epoch": 6.389713287353516e-05,
      "step": 67001,
      "training_step_time": 9.447328567504883
    },
    {
      "epoch": 6.389808654785156e-05,
      "model_forward_time": 1.847675085067749,
      "step": 67002
    },
    {
      "epoch": 6.389808654785156e-05,
      "step": 67002,
      "training_step_time": 12.677127361297607
    },
    {
      "epoch": 6.389904022216797e-05,
      "model_forward_time": 1.833282470703125,
      "step": 67003
    },
    {
      "epoch": 6.389904022216797e-05,
      "step": 67003,
      "training_step_time": 7.759264945983887
    },
    {
      "epoch": 6.389999389648438e-05,
      "model_forward_time": 1.8240852355957031,
      "step": 67004
    },
    {
      "epoch": 6.389999389648438e-05,
      "step": 67004,
      "training_step_time": 7.787264347076416
    },
    {
      "epoch": 6.390094757080078e-05,
      "model_forward_time": 2.9557383060455322,
      "step": 67005
    },
    {
      "epoch": 6.390094757080078e-05,
      "step": 67005,
      "training_step_time": 7.799891710281372
    },
    {
      "epoch": 6.390190124511719e-05,
      "model_forward_time": 1.8326950073242188,
      "step": 67006
    },
    {
      "epoch": 6.390190124511719e-05,
      "step": 67006,
      "training_step_time": 8.2545325756073
    },
    {
      "epoch": 6.39028549194336e-05,
      "model_forward_time": 1.376283884048462,
      "step": 67007
    },
    {
      "epoch": 6.39028549194336e-05,
      "step": 67007,
      "training_step_time": 8.717548847198486
    },
    {
      "epoch": 6.390380859375e-05,
      "model_forward_time": 1.7472398281097412,
      "step": 67008
    },
    {
      "epoch": 6.390380859375e-05,
      "step": 67008,
      "training_step_time": 7.770378351211548
    },
    {
      "epoch": 6.39047622680664e-05,
      "model_forward_time": 1.704801082611084,
      "step": 67009
    },
    {
      "epoch": 6.39047622680664e-05,
      "step": 67009,
      "training_step_time": 8.988803148269653
    },
    {
      "epoch": 6.390571594238282e-05,
      "grad_norm": 0.09294185042381287,
      "learning_rate": 2.6919739907523403e-06,
      "loss": 0.0594,
      "step": 67010
    },
    {
      "epoch": 6.390571594238282e-05,
      "model_forward_time": 1.6973533630371094,
      "step": 67010
    },
    {
      "dynamics_loss_avg": 0.05687672086060047,
      "epoch": 6.390571594238282e-05,
      "step": 67010
    },
    {
      "action_loss_avg": 0.003055886345100589,
      "epoch": 6.390571594238282e-05,
      "step": 67010
    },
    {
      "epoch": 6.390571594238282e-05,
      "step": 67010,
      "training_step_time": 8.44059944152832
    },
    {
      "epoch": 6.390666961669921e-05,
      "model_forward_time": 1.7120451927185059,
      "step": 67011
    },
    {
      "epoch": 6.390666961669921e-05,
      "step": 67011,
      "training_step_time": 9.717244863510132
    },
    {
      "epoch": 6.390762329101563e-05,
      "model_forward_time": 1.7070484161376953,
      "step": 67012
    },
    {
      "epoch": 6.390762329101563e-05,
      "step": 67012,
      "training_step_time": 7.729647636413574
    },
    {
      "epoch": 6.390857696533204e-05,
      "model_forward_time": 1.7205498218536377,
      "step": 67013
    },
    {
      "epoch": 6.390857696533204e-05,
      "step": 67013,
      "training_step_time": 7.680605173110962
    },
    {
      "epoch": 6.390953063964843e-05,
      "model_forward_time": 2.7827861309051514,
      "step": 67014
    },
    {
      "epoch": 6.390953063964843e-05,
      "step": 67014,
      "training_step_time": 8.504362106323242
    },
    {
      "epoch": 6.391048431396484e-05,
      "model_forward_time": 1.7061994075775146,
      "step": 67015
    },
    {
      "epoch": 6.391048431396484e-05,
      "step": 67015,
      "training_step_time": 7.730777025222778
    },
    {
      "epoch": 6.391143798828126e-05,
      "model_forward_time": 1.7074642181396484,
      "step": 67016
    },
    {
      "epoch": 6.391143798828126e-05,
      "step": 67016,
      "training_step_time": 7.730355978012085
    },
    {
      "epoch": 6.391239166259765e-05,
      "model_forward_time": 1.7041137218475342,
      "step": 67017
    },
    {
      "epoch": 6.391239166259765e-05,
      "step": 67017,
      "training_step_time": 7.7208476066589355
    },
    {
      "epoch": 6.391334533691406e-05,
      "model_forward_time": 1.7151715755462646,
      "step": 67018
    },
    {
      "epoch": 6.391334533691406e-05,
      "step": 67018,
      "training_step_time": 8.473339796066284
    },
    {
      "epoch": 6.391429901123047e-05,
      "model_forward_time": 1.7043213844299316,
      "step": 67019
    },
    {
      "epoch": 6.391429901123047e-05,
      "step": 67019,
      "training_step_time": 7.813939571380615
    },
    {
      "epoch": 6.391525268554687e-05,
      "grad_norm": 0.11133648455142975,
      "learning_rate": 2.690507347923759e-06,
      "loss": 0.0528,
      "step": 67020
    },
    {
      "epoch": 6.391525268554687e-05,
      "model_forward_time": 1.7076966762542725,
      "step": 67020
    },
    {
      "dynamics_loss_avg": 0.05408484153449535,
      "epoch": 6.391525268554687e-05,
      "step": 67020
    },
    {
      "action_loss_avg": 0.0017173549480503426,
      "epoch": 6.391525268554687e-05,
      "step": 67020
    },
    {
      "epoch": 6.391525268554687e-05,
      "step": 67020,
      "training_step_time": 9.58149766921997
    },
    {
      "epoch": 6.391620635986328e-05,
      "model_forward_time": 1.716616153717041,
      "step": 67021
    },
    {
      "epoch": 6.391620635986328e-05,
      "step": 67021,
      "training_step_time": 7.8510332107543945
    },
    {
      "epoch": 6.391716003417969e-05,
      "model_forward_time": 1.723588466644287,
      "step": 67022
    },
    {
      "epoch": 6.391716003417969e-05,
      "step": 67022,
      "training_step_time": 8.299975633621216
    },
    {
      "epoch": 6.391811370849609e-05,
      "model_forward_time": 1.718806266784668,
      "step": 67023
    },
    {
      "epoch": 6.391811370849609e-05,
      "step": 67023,
      "training_step_time": 8.424628019332886
    },
    {
      "epoch": 6.39190673828125e-05,
      "model_forward_time": 2.9653050899505615,
      "step": 67024
    },
    {
      "epoch": 6.39190673828125e-05,
      "step": 67024,
      "training_step_time": 7.838557958602905
    },
    {
      "epoch": 6.392002105712891e-05,
      "model_forward_time": 1.7144851684570312,
      "step": 67025
    },
    {
      "epoch": 6.392002105712891e-05,
      "step": 67025,
      "training_step_time": 7.72441291809082
    },
    {
      "epoch": 6.392097473144531e-05,
      "model_forward_time": 1.7044153213500977,
      "step": 67026
    },
    {
      "epoch": 6.392097473144531e-05,
      "step": 67026,
      "training_step_time": 7.742419004440308
    },
    {
      "epoch": 6.392192840576172e-05,
      "model_forward_time": 1.7075235843658447,
      "step": 67027
    },
    {
      "epoch": 6.392192840576172e-05,
      "step": 67027,
      "training_step_time": 7.691410541534424
    },
    {
      "epoch": 6.392288208007813e-05,
      "model_forward_time": 1.732924461364746,
      "step": 67028
    },
    {
      "epoch": 6.392288208007813e-05,
      "step": 67028,
      "training_step_time": 7.864664316177368
    },
    {
      "epoch": 6.392383575439453e-05,
      "model_forward_time": 1.7123620510101318,
      "step": 67029
    },
    {
      "epoch": 6.392383575439453e-05,
      "step": 67029,
      "training_step_time": 10.909006357192993
    },
    {
      "epoch": 6.392478942871094e-05,
      "grad_norm": 0.10154274106025696,
      "learning_rate": 2.6890409576578234e-06,
      "loss": 0.0565,
      "step": 67030
    },
    {
      "epoch": 6.392478942871094e-05,
      "model_forward_time": 1.3415868282318115,
      "step": 67030
    },
    {
      "dynamics_loss_avg": 0.04287289585918188,
      "epoch": 6.392478942871094e-05,
      "step": 67030
    },
    {
      "action_loss_avg": 0.0017180948227178306,
      "epoch": 6.392478942871094e-05,
      "step": 67030
    },
    {
      "epoch": 6.392478942871094e-05,
      "step": 67030,
      "training_step_time": 9.351624250411987
    },
    {
      "epoch": 6.392574310302735e-05,
      "model_forward_time": 1.7128911018371582,
      "step": 67031
    },
    {
      "epoch": 6.392574310302735e-05,
      "step": 67031,
      "training_step_time": 8.761620283126831
    },
    {
      "epoch": 6.392669677734375e-05,
      "model_forward_time": 1.7192256450653076,
      "step": 67032
    },
    {
      "epoch": 6.392669677734375e-05,
      "step": 67032,
      "training_step_time": 7.652275800704956
    },
    {
      "epoch": 6.392765045166016e-05,
      "model_forward_time": 2.920950412750244,
      "step": 67033
    },
    {
      "epoch": 6.392765045166016e-05,
      "step": 67033,
      "training_step_time": 7.776623964309692
    },
    {
      "epoch": 6.392860412597657e-05,
      "model_forward_time": 1.7152009010314941,
      "step": 67034
    },
    {
      "epoch": 6.392860412597657e-05,
      "step": 67034,
      "training_step_time": 7.66643500328064
    },
    {
      "epoch": 6.392955780029297e-05,
      "model_forward_time": 1.7224507331848145,
      "step": 67035
    },
    {
      "epoch": 6.392955780029297e-05,
      "step": 67035,
      "training_step_time": 8.617060661315918
    },
    {
      "epoch": 6.393051147460938e-05,
      "model_forward_time": 1.7223825454711914,
      "step": 67036
    },
    {
      "epoch": 6.393051147460938e-05,
      "step": 67036,
      "training_step_time": 7.64701509475708
    },
    {
      "epoch": 6.393146514892579e-05,
      "model_forward_time": 1.7346878051757812,
      "step": 67037
    },
    {
      "epoch": 6.393146514892579e-05,
      "step": 67037,
      "training_step_time": 7.725204944610596
    },
    {
      "epoch": 6.393241882324218e-05,
      "model_forward_time": 1.711796522140503,
      "step": 67038
    },
    {
      "epoch": 6.393241882324218e-05,
      "step": 67038,
      "training_step_time": 9.891112089157104
    },
    {
      "epoch": 6.39333724975586e-05,
      "model_forward_time": 1.719855546951294,
      "step": 67039
    },
    {
      "epoch": 6.39333724975586e-05,
      "step": 67039,
      "training_step_time": 7.703927278518677
    },
    {
      "epoch": 6.3934326171875e-05,
      "grad_norm": 0.1429678201675415,
      "learning_rate": 2.687574820114892e-06,
      "loss": 0.0564,
      "step": 67040
    },
    {
      "epoch": 6.3934326171875e-05,
      "model_forward_time": 1.7144362926483154,
      "step": 67040
    },
    {
      "dynamics_loss_avg": 0.04503973722457886,
      "epoch": 6.3934326171875e-05,
      "step": 67040
    },
    {
      "action_loss_avg": 0.001985373836942017,
      "epoch": 6.3934326171875e-05,
      "step": 67040
    },
    {
      "epoch": 6.3934326171875e-05,
      "step": 67040,
      "training_step_time": 7.708825588226318
    },
    {
      "epoch": 6.39352798461914e-05,
      "model_forward_time": 1.7488090991973877,
      "step": 67041
    },
    {
      "epoch": 6.39352798461914e-05,
      "step": 67041,
      "training_step_time": 8.565322875976562
    },
    {
      "epoch": 6.393623352050781e-05,
      "model_forward_time": 2.859809398651123,
      "step": 67042
    },
    {
      "epoch": 6.393623352050781e-05,
      "step": 67042,
      "training_step_time": 7.671081066131592
    },
    {
      "epoch": 6.393718719482422e-05,
      "model_forward_time": 1.7240374088287354,
      "step": 67043
    },
    {
      "epoch": 6.393718719482422e-05,
      "step": 67043,
      "training_step_time": 7.641189813613892
    },
    {
      "epoch": 6.393814086914062e-05,
      "model_forward_time": 1.7280540466308594,
      "step": 67044
    },
    {
      "epoch": 6.393814086914062e-05,
      "step": 67044,
      "training_step_time": 7.605221509933472
    },
    {
      "epoch": 6.393909454345703e-05,
      "model_forward_time": 1.7199199199676514,
      "step": 67045
    },
    {
      "epoch": 6.393909454345703e-05,
      "step": 67045,
      "training_step_time": 7.697484970092773
    },
    {
      "epoch": 6.394004821777344e-05,
      "model_forward_time": 1.7354485988616943,
      "step": 67046
    },
    {
      "epoch": 6.394004821777344e-05,
      "step": 67046,
      "training_step_time": 7.724252223968506
    },
    {
      "epoch": 6.394100189208984e-05,
      "model_forward_time": 1.7256536483764648,
      "step": 67047
    },
    {
      "epoch": 6.394100189208984e-05,
      "step": 67047,
      "training_step_time": 9.343913078308105
    },
    {
      "epoch": 6.394195556640625e-05,
      "model_forward_time": 1.7177035808563232,
      "step": 67048
    },
    {
      "epoch": 6.394195556640625e-05,
      "step": 67048,
      "training_step_time": 10.335799217224121
    },
    {
      "epoch": 6.394290924072266e-05,
      "model_forward_time": 1.7191145420074463,
      "step": 67049
    },
    {
      "epoch": 6.394290924072266e-05,
      "step": 67049,
      "training_step_time": 7.717813968658447
    },
    {
      "epoch": 6.394386291503906e-05,
      "grad_norm": 0.13128706812858582,
      "learning_rate": 2.686108935455305e-06,
      "loss": 0.0541,
      "step": 67050
    },
    {
      "epoch": 6.394386291503906e-05,
      "model_forward_time": 1.7443914413452148,
      "step": 67050
    },
    {
      "dynamics_loss_avg": 0.041155735962092876,
      "epoch": 6.394386291503906e-05,
      "step": 67050
    },
    {
      "action_loss_avg": 0.0010151393886189907,
      "epoch": 6.394386291503906e-05,
      "step": 67050
    },
    {
      "epoch": 6.394386291503906e-05,
      "step": 67050,
      "training_step_time": 8.710340976715088
    },
    {
      "epoch": 6.394481658935547e-05,
      "model_forward_time": 2.825791835784912,
      "step": 67051
    },
    {
      "epoch": 6.394481658935547e-05,
      "step": 67051,
      "training_step_time": 7.67163348197937
    },
    {
      "epoch": 6.394577026367188e-05,
      "model_forward_time": 1.7264673709869385,
      "step": 67052
    },
    {
      "epoch": 6.394577026367188e-05,
      "step": 67052,
      "training_step_time": 7.68462610244751
    },
    {
      "epoch": 6.394672393798828e-05,
      "model_forward_time": 1.7359943389892578,
      "step": 67053
    },
    {
      "epoch": 6.394672393798828e-05,
      "step": 67053,
      "training_step_time": 7.770940542221069
    },
    {
      "epoch": 6.394767761230469e-05,
      "model_forward_time": 1.7207164764404297,
      "step": 67054
    },
    {
      "epoch": 6.394767761230469e-05,
      "step": 67054,
      "training_step_time": 7.677520751953125
    },
    {
      "epoch": 6.39486312866211e-05,
      "model_forward_time": 1.7398691177368164,
      "step": 67055
    },
    {
      "epoch": 6.39486312866211e-05,
      "step": 67055,
      "training_step_time": 10.46898341178894
    },
    {
      "epoch": 6.39495849609375e-05,
      "model_forward_time": 1.7152817249298096,
      "step": 67056
    },
    {
      "epoch": 6.39495849609375e-05,
      "step": 67056,
      "training_step_time": 9.796189546585083
    },
    {
      "epoch": 6.395053863525391e-05,
      "model_forward_time": 1.734583854675293,
      "step": 67057
    },
    {
      "epoch": 6.395053863525391e-05,
      "step": 67057,
      "training_step_time": 6.788424015045166
    },
    {
      "epoch": 6.395149230957032e-05,
      "model_forward_time": 1.7372047901153564,
      "step": 67058
    },
    {
      "epoch": 6.395149230957032e-05,
      "step": 67058,
      "training_step_time": 6.792148590087891
    },
    {
      "epoch": 6.395244598388672e-05,
      "model_forward_time": 1.723708152770996,
      "step": 67059
    },
    {
      "epoch": 6.395244598388672e-05,
      "step": 67059,
      "training_step_time": 6.835734128952026
    },
    {
      "epoch": 6.395339965820313e-05,
      "grad_norm": 0.10493569076061249,
      "learning_rate": 2.6846433038393603e-06,
      "loss": 0.0569,
      "step": 67060
    },
    {
      "epoch": 6.395339965820313e-05,
      "model_forward_time": 1.7282938957214355,
      "step": 67060
    },
    {
      "dynamics_loss_avg": 0.05076706074178219,
      "epoch": 6.395339965820313e-05,
      "step": 67060
    },
    {
      "action_loss_avg": 0.0012398508726619184,
      "epoch": 6.395339965820313e-05,
      "step": 67060
    },
    {
      "epoch": 6.395339965820313e-05,
      "step": 67060,
      "training_step_time": 6.778712034225464
    },
    {
      "epoch": 6.395435333251954e-05,
      "model_forward_time": 1.7305018901824951,
      "step": 67061
    },
    {
      "epoch": 6.395435333251954e-05,
      "step": 67061,
      "training_step_time": 6.87966251373291
    },
    {
      "epoch": 6.395530700683594e-05,
      "model_forward_time": 1.7408983707427979,
      "step": 67062
    },
    {
      "epoch": 6.395530700683594e-05,
      "step": 67062,
      "training_step_time": 6.856299161911011
    },
    {
      "epoch": 6.395626068115235e-05,
      "model_forward_time": 1.7421581745147705,
      "step": 67063
    },
    {
      "epoch": 6.395626068115235e-05,
      "step": 67063,
      "training_step_time": 6.912152051925659
    },
    {
      "epoch": 6.395721435546874e-05,
      "model_forward_time": 1.7212629318237305,
      "step": 67064
    },
    {
      "epoch": 6.395721435546874e-05,
      "step": 67064,
      "training_step_time": 7.906212568283081
    },
    {
      "epoch": 6.395816802978515e-05,
      "model_forward_time": 2.9111316204071045,
      "step": 67065
    },
    {
      "epoch": 6.395816802978515e-05,
      "step": 67065,
      "training_step_time": 7.876328945159912
    },
    {
      "epoch": 6.395912170410156e-05,
      "model_forward_time": 1.7439749240875244,
      "step": 67066
    },
    {
      "epoch": 6.395912170410156e-05,
      "step": 67066,
      "training_step_time": 7.9870383739471436
    },
    {
      "epoch": 6.396007537841796e-05,
      "model_forward_time": 1.7379932403564453,
      "step": 67067
    },
    {
      "epoch": 6.396007537841796e-05,
      "step": 67067,
      "training_step_time": 6.927615165710449
    },
    {
      "epoch": 6.396102905273437e-05,
      "model_forward_time": 1.7252094745635986,
      "step": 67068
    },
    {
      "epoch": 6.396102905273437e-05,
      "step": 67068,
      "training_step_time": 6.933892011642456
    },
    {
      "epoch": 6.396198272705078e-05,
      "model_forward_time": 1.731482744216919,
      "step": 67069
    },
    {
      "epoch": 6.396198272705078e-05,
      "step": 67069,
      "training_step_time": 6.940009593963623
    },
    {
      "epoch": 6.396293640136718e-05,
      "grad_norm": 0.10524323582649231,
      "learning_rate": 2.6831779254273483e-06,
      "loss": 0.0567,
      "step": 67070
    },
    {
      "epoch": 6.396293640136718e-05,
      "model_forward_time": 1.733506679534912,
      "step": 67070
    },
    {
      "dynamics_loss_avg": 0.05604873336851597,
      "epoch": 6.396293640136718e-05,
      "step": 67070
    },
    {
      "action_loss_avg": 0.001269949451670982,
      "epoch": 6.396293640136718e-05,
      "step": 67070
    },
    {
      "epoch": 6.396293640136718e-05,
      "step": 67070,
      "training_step_time": 6.943997383117676
    },
    {
      "epoch": 6.396389007568359e-05,
      "model_forward_time": 1.7230706214904785,
      "step": 67071
    },
    {
      "epoch": 6.396389007568359e-05,
      "step": 67071,
      "training_step_time": 6.9284446239471436
    },
    {
      "epoch": 6.396484375e-05,
      "model_forward_time": 1.7282452583312988,
      "step": 67072
    },
    {
      "epoch": 6.396484375e-05,
      "step": 67072,
      "training_step_time": 6.849413633346558
    },
    {
      "epoch": 6.39657974243164e-05,
      "model_forward_time": 1.7184438705444336,
      "step": 67073
    },
    {
      "epoch": 6.39657974243164e-05,
      "step": 67073,
      "training_step_time": 6.86351752281189
    },
    {
      "epoch": 6.396675109863281e-05,
      "model_forward_time": 2.912562370300293,
      "step": 67074
    },
    {
      "epoch": 6.396675109863281e-05,
      "step": 67074,
      "training_step_time": 7.790545463562012
    },
    {
      "epoch": 6.396770477294922e-05,
      "model_forward_time": 1.7386369705200195,
      "step": 67075
    },
    {
      "epoch": 6.396770477294922e-05,
      "step": 67075,
      "training_step_time": 7.832220792770386
    },
    {
      "epoch": 6.396865844726562e-05,
      "model_forward_time": 1.7266132831573486,
      "step": 67076
    },
    {
      "epoch": 6.396865844726562e-05,
      "step": 67076,
      "training_step_time": 7.713230848312378
    },
    {
      "epoch": 6.396961212158203e-05,
      "model_forward_time": 1.7214014530181885,
      "step": 67077
    },
    {
      "epoch": 6.396961212158203e-05,
      "step": 67077,
      "training_step_time": 6.868687868118286
    },
    {
      "epoch": 6.397056579589844e-05,
      "model_forward_time": 1.7312853336334229,
      "step": 67078
    },
    {
      "epoch": 6.397056579589844e-05,
      "step": 67078,
      "training_step_time": 6.851720094680786
    },
    {
      "epoch": 6.397151947021484e-05,
      "model_forward_time": 1.7286264896392822,
      "step": 67079
    },
    {
      "epoch": 6.397151947021484e-05,
      "step": 67079,
      "training_step_time": 6.848710536956787
    },
    {
      "epoch": 6.397247314453125e-05,
      "grad_norm": 0.07574649155139923,
      "learning_rate": 2.681712800379512e-06,
      "loss": 0.0541,
      "step": 67080
    },
    {
      "epoch": 6.397247314453125e-05,
      "model_forward_time": 1.758436918258667,
      "step": 67080
    },
    {
      "dynamics_loss_avg": 0.050231313705444335,
      "epoch": 6.397247314453125e-05,
      "step": 67080
    },
    {
      "action_loss_avg": 0.0025049161515198646,
      "epoch": 6.397247314453125e-05,
      "step": 67080
    },
    {
      "epoch": 6.397247314453125e-05,
      "step": 67080,
      "training_step_time": 6.877628803253174
    },
    {
      "epoch": 6.397342681884766e-05,
      "model_forward_time": 1.7367777824401855,
      "step": 67081
    },
    {
      "epoch": 6.397342681884766e-05,
      "step": 67081,
      "training_step_time": 6.838506460189819
    },
    {
      "epoch": 6.397438049316406e-05,
      "model_forward_time": 1.7237720489501953,
      "step": 67082
    },
    {
      "epoch": 6.397438049316406e-05,
      "step": 67082,
      "training_step_time": 6.802293300628662
    },
    {
      "epoch": 6.397533416748047e-05,
      "model_forward_time": 2.9646670818328857,
      "step": 67083
    },
    {
      "epoch": 6.397533416748047e-05,
      "step": 67083,
      "training_step_time": 7.845236539840698
    },
    {
      "epoch": 6.397628784179688e-05,
      "model_forward_time": 1.7280588150024414,
      "step": 67084
    },
    {
      "epoch": 6.397628784179688e-05,
      "step": 67084,
      "training_step_time": 7.805952072143555
    },
    {
      "epoch": 6.397724151611328e-05,
      "model_forward_time": 1.7259843349456787,
      "step": 67085
    },
    {
      "epoch": 6.397724151611328e-05,
      "step": 67085,
      "training_step_time": 7.6700615882873535
    },
    {
      "epoch": 6.397819519042969e-05,
      "model_forward_time": 1.7210218906402588,
      "step": 67086
    },
    {
      "epoch": 6.397819519042969e-05,
      "step": 67086,
      "training_step_time": 6.824272632598877
    },
    {
      "epoch": 6.39791488647461e-05,
      "model_forward_time": 1.726423978805542,
      "step": 67087
    },
    {
      "epoch": 6.39791488647461e-05,
      "step": 67087,
      "training_step_time": 6.8490118980407715
    },
    {
      "epoch": 6.39801025390625e-05,
      "model_forward_time": 1.7252674102783203,
      "step": 67088
    },
    {
      "epoch": 6.39801025390625e-05,
      "step": 67088,
      "training_step_time": 6.7698423862457275
    },
    {
      "epoch": 6.39810562133789e-05,
      "model_forward_time": 1.7193934917449951,
      "step": 67089
    },
    {
      "epoch": 6.39810562133789e-05,
      "step": 67089,
      "training_step_time": 6.78259015083313
    },
    {
      "epoch": 6.398200988769532e-05,
      "grad_norm": 0.08253037929534912,
      "learning_rate": 2.6802479288560814e-06,
      "loss": 0.0541,
      "step": 67090
    },
    {
      "epoch": 6.398200988769532e-05,
      "model_forward_time": 1.7220678329467773,
      "step": 67090
    },
    {
      "dynamics_loss_avg": 0.03974965941160917,
      "epoch": 6.398200988769532e-05,
      "step": 67090
    },
    {
      "action_loss_avg": 0.0018103857495589183,
      "epoch": 6.398200988769532e-05,
      "step": 67090
    },
    {
      "epoch": 6.398200988769532e-05,
      "step": 67090,
      "training_step_time": 6.839293479919434
    },
    {
      "epoch": 6.398296356201171e-05,
      "model_forward_time": 1.72648286819458,
      "step": 67091
    },
    {
      "epoch": 6.398296356201171e-05,
      "step": 67091,
      "training_step_time": 6.806665420532227
    },
    {
      "epoch": 6.398391723632812e-05,
      "model_forward_time": 2.986593723297119,
      "step": 67092
    },
    {
      "epoch": 6.398391723632812e-05,
      "step": 67092,
      "training_step_time": 7.821111679077148
    },
    {
      "epoch": 6.398487091064453e-05,
      "model_forward_time": 1.7283921241760254,
      "step": 67093
    },
    {
      "epoch": 6.398487091064453e-05,
      "step": 67093,
      "training_step_time": 9.330818176269531
    },
    {
      "epoch": 6.398582458496093e-05,
      "model_forward_time": 1.7191402912139893,
      "step": 67094
    },
    {
      "epoch": 6.398582458496093e-05,
      "step": 67094,
      "training_step_time": 7.777117967605591
    },
    {
      "epoch": 6.398677825927734e-05,
      "model_forward_time": 1.7332837581634521,
      "step": 67095
    },
    {
      "epoch": 6.398677825927734e-05,
      "step": 67095,
      "training_step_time": 6.812598705291748
    },
    {
      "epoch": 6.398773193359375e-05,
      "model_forward_time": 1.7319767475128174,
      "step": 67096
    },
    {
      "epoch": 6.398773193359375e-05,
      "step": 67096,
      "training_step_time": 6.673810720443726
    },
    {
      "epoch": 6.398868560791015e-05,
      "model_forward_time": 1.715846300125122,
      "step": 67097
    },
    {
      "epoch": 6.398868560791015e-05,
      "step": 67097,
      "training_step_time": 6.759044647216797
    },
    {
      "epoch": 6.398963928222656e-05,
      "model_forward_time": 1.7217564582824707,
      "step": 67098
    },
    {
      "epoch": 6.398963928222656e-05,
      "step": 67098,
      "training_step_time": 6.79689884185791
    },
    {
      "epoch": 6.399059295654297e-05,
      "model_forward_time": 1.7503204345703125,
      "step": 67099
    },
    {
      "epoch": 6.399059295654297e-05,
      "step": 67099,
      "training_step_time": 6.778844594955444
    },
    {
      "epoch": 6.399154663085937e-05,
      "grad_norm": 0.1021529883146286,
      "learning_rate": 2.6787833110172467e-06,
      "loss": 0.0559,
      "step": 67100
    },
    {
      "epoch": 6.399154663085937e-05,
      "model_forward_time": 1.721472978591919,
      "step": 67100
    },
    {
      "dynamics_loss_avg": 0.0553786862641573,
      "epoch": 6.399154663085937e-05,
      "step": 67100
    },
    {
      "action_loss_avg": 0.001418911590008065,
      "epoch": 6.399154663085937e-05,
      "step": 67100
    },
    {
      "epoch": 6.399154663085937e-05,
      "step": 67100,
      "training_step_time": 6.802858591079712
    },
    {
      "epoch": 6.399250030517578e-05,
      "model_forward_time": 2.8836357593536377,
      "step": 67101
    },
    {
      "epoch": 6.399250030517578e-05,
      "step": 67101,
      "training_step_time": 7.818798065185547
    },
    {
      "epoch": 6.399345397949219e-05,
      "model_forward_time": 1.849412202835083,
      "step": 67102
    },
    {
      "epoch": 6.399345397949219e-05,
      "step": 67102,
      "training_step_time": 7.858529806137085
    },
    {
      "epoch": 6.399440765380859e-05,
      "model_forward_time": 1.3591382503509521,
      "step": 67103
    },
    {
      "epoch": 6.399440765380859e-05,
      "step": 67103,
      "training_step_time": 7.749125719070435
    },
    {
      "epoch": 6.3995361328125e-05,
      "model_forward_time": 1.731339454650879,
      "step": 67104
    },
    {
      "epoch": 6.3995361328125e-05,
      "step": 67104,
      "training_step_time": 6.813818693161011
    },
    {
      "epoch": 6.399631500244141e-05,
      "model_forward_time": 1.7272920608520508,
      "step": 67105
    },
    {
      "epoch": 6.399631500244141e-05,
      "step": 67105,
      "training_step_time": 6.743423938751221
    },
    {
      "epoch": 6.399726867675781e-05,
      "model_forward_time": 1.7210495471954346,
      "step": 67106
    },
    {
      "epoch": 6.399726867675781e-05,
      "step": 67106,
      "training_step_time": 6.773160696029663
    },
    {
      "epoch": 6.399822235107422e-05,
      "model_forward_time": 1.7209093570709229,
      "step": 67107
    },
    {
      "epoch": 6.399822235107422e-05,
      "step": 67107,
      "training_step_time": 6.793615341186523
    },
    {
      "epoch": 6.399917602539063e-05,
      "model_forward_time": 1.7174146175384521,
      "step": 67108
    },
    {
      "epoch": 6.399917602539063e-05,
      "step": 67108,
      "training_step_time": 6.772198438644409
    },
    {
      "epoch": 6.400012969970703e-05,
      "model_forward_time": 1.7251884937286377,
      "step": 67109
    },
    {
      "epoch": 6.400012969970703e-05,
      "step": 67109,
      "training_step_time": 6.784197568893433
    },
    {
      "epoch": 6.400108337402344e-05,
      "grad_norm": 0.09352309256792068,
      "learning_rate": 2.6773189470231807e-06,
      "loss": 0.0568,
      "step": 67110
    },
    {
      "epoch": 6.400108337402344e-05,
      "model_forward_time": 1.7384843826293945,
      "step": 67110
    },
    {
      "dynamics_loss_avg": 0.07422195971012116,
      "epoch": 6.400108337402344e-05,
      "step": 67110
    },
    {
      "action_loss_avg": 0.002990026696352288,
      "epoch": 6.400108337402344e-05,
      "step": 67110
    },
    {
      "epoch": 6.400108337402344e-05,
      "step": 67110,
      "training_step_time": 8.794297218322754
    },
    {
      "epoch": 6.400203704833985e-05,
      "model_forward_time": 1.853198766708374,
      "step": 67111
    },
    {
      "epoch": 6.400203704833985e-05,
      "step": 67111,
      "training_step_time": 8.72779130935669
    },
    {
      "epoch": 6.400299072265625e-05,
      "model_forward_time": 1.7329461574554443,
      "step": 67112
    },
    {
      "epoch": 6.400299072265625e-05,
      "step": 67112,
      "training_step_time": 8.765477895736694
    },
    {
      "epoch": 6.400394439697266e-05,
      "model_forward_time": 1.7264928817749023,
      "step": 67113
    },
    {
      "epoch": 6.400394439697266e-05,
      "step": 67113,
      "training_step_time": 7.562119722366333
    },
    {
      "epoch": 6.400489807128907e-05,
      "model_forward_time": 1.7321398258209229,
      "step": 67114
    },
    {
      "epoch": 6.400489807128907e-05,
      "step": 67114,
      "training_step_time": 6.743835687637329
    },
    {
      "epoch": 6.400585174560546e-05,
      "model_forward_time": 1.7432284355163574,
      "step": 67115
    },
    {
      "epoch": 6.400585174560546e-05,
      "step": 67115,
      "training_step_time": 6.769838333129883
    },
    {
      "epoch": 6.400680541992187e-05,
      "model_forward_time": 1.722757339477539,
      "step": 67116
    },
    {
      "epoch": 6.400680541992187e-05,
      "step": 67116,
      "training_step_time": 6.844505310058594
    },
    {
      "epoch": 6.400775909423829e-05,
      "model_forward_time": 1.7214381694793701,
      "step": 67117
    },
    {
      "epoch": 6.400775909423829e-05,
      "step": 67117,
      "training_step_time": 6.807322025299072
    },
    {
      "epoch": 6.400871276855468e-05,
      "model_forward_time": 1.7252237796783447,
      "step": 67118
    },
    {
      "epoch": 6.400871276855468e-05,
      "step": 67118,
      "training_step_time": 6.683483123779297
    },
    {
      "epoch": 6.40096664428711e-05,
      "model_forward_time": 1.721780776977539,
      "step": 67119
    },
    {
      "epoch": 6.40096664428711e-05,
      "step": 67119,
      "training_step_time": 7.783825159072876
    },
    {
      "epoch": 6.40106201171875e-05,
      "grad_norm": 0.12445579469203949,
      "learning_rate": 2.675854837034022e-06,
      "loss": 0.0572,
      "step": 67120
    },
    {
      "epoch": 6.40106201171875e-05,
      "model_forward_time": 2.871450424194336,
      "step": 67120
    },
    {
      "dynamics_loss_avg": 0.04972325302660465,
      "epoch": 6.40106201171875e-05,
      "step": 67120
    },
    {
      "action_loss_avg": 0.0012915597922983578,
      "epoch": 6.40106201171875e-05,
      "step": 67120
    },
    {
      "epoch": 6.40106201171875e-05,
      "step": 67120,
      "training_step_time": 7.669880151748657
    },
    {
      "epoch": 6.40115737915039e-05,
      "model_forward_time": 1.7231025695800781,
      "step": 67121
    },
    {
      "epoch": 6.40115737915039e-05,
      "step": 67121,
      "training_step_time": 7.757167816162109
    },
    {
      "epoch": 6.401252746582031e-05,
      "model_forward_time": 1.7238624095916748,
      "step": 67122
    },
    {
      "epoch": 6.401252746582031e-05,
      "step": 67122,
      "training_step_time": 7.5961549282073975
    },
    {
      "epoch": 6.401348114013672e-05,
      "model_forward_time": 1.733642339706421,
      "step": 67123
    },
    {
      "epoch": 6.401348114013672e-05,
      "step": 67123,
      "training_step_time": 6.7517664432525635
    },
    {
      "epoch": 6.401443481445312e-05,
      "model_forward_time": 1.726802110671997,
      "step": 67124
    },
    {
      "epoch": 6.401443481445312e-05,
      "step": 67124,
      "training_step_time": 6.777608156204224
    },
    {
      "epoch": 6.401538848876953e-05,
      "model_forward_time": 1.7227222919464111,
      "step": 67125
    },
    {
      "epoch": 6.401538848876953e-05,
      "step": 67125,
      "training_step_time": 6.826921701431274
    },
    {
      "epoch": 6.401634216308594e-05,
      "model_forward_time": 1.7242765426635742,
      "step": 67126
    },
    {
      "epoch": 6.401634216308594e-05,
      "step": 67126,
      "training_step_time": 6.789108753204346
    },
    {
      "epoch": 6.401729583740234e-05,
      "model_forward_time": 1.732755422592163,
      "step": 67127
    },
    {
      "epoch": 6.401729583740234e-05,
      "step": 67127,
      "training_step_time": 6.790859222412109
    },
    {
      "epoch": 6.401824951171875e-05,
      "model_forward_time": 1.7098901271820068,
      "step": 67128
    },
    {
      "epoch": 6.401824951171875e-05,
      "step": 67128,
      "training_step_time": 6.804110527038574
    },
    {
      "epoch": 6.401920318603516e-05,
      "model_forward_time": 2.8899343013763428,
      "step": 67129
    },
    {
      "epoch": 6.401920318603516e-05,
      "step": 67129,
      "training_step_time": 7.7946836948394775
    },
    {
      "epoch": 6.402015686035156e-05,
      "grad_norm": 0.07774818688631058,
      "learning_rate": 2.6743909812098833e-06,
      "loss": 0.0584,
      "step": 67130
    },
    {
      "epoch": 6.402015686035156e-05,
      "model_forward_time": 1.7243378162384033,
      "step": 67130
    },
    {
      "dynamics_loss_avg": 0.05319015691056848,
      "epoch": 6.402015686035156e-05,
      "step": 67130
    },
    {
      "action_loss_avg": 0.0017615498014492913,
      "epoch": 6.402015686035156e-05,
      "step": 67130
    },
    {
      "epoch": 6.402015686035156e-05,
      "step": 67130,
      "training_step_time": 8.351561307907104
    },
    {
      "epoch": 6.402111053466797e-05,
      "model_forward_time": 1.7257282733917236,
      "step": 67131
    },
    {
      "epoch": 6.402111053466797e-05,
      "step": 67131,
      "training_step_time": 7.6585142612457275
    },
    {
      "epoch": 6.402206420898438e-05,
      "model_forward_time": 1.734544277191162,
      "step": 67132
    },
    {
      "epoch": 6.402206420898438e-05,
      "step": 67132,
      "training_step_time": 6.786537170410156
    },
    {
      "epoch": 6.402301788330078e-05,
      "model_forward_time": 1.7328402996063232,
      "step": 67133
    },
    {
      "epoch": 6.402301788330078e-05,
      "step": 67133,
      "training_step_time": 6.723260164260864
    },
    {
      "epoch": 6.402397155761719e-05,
      "model_forward_time": 1.724557876586914,
      "step": 67134
    },
    {
      "epoch": 6.402397155761719e-05,
      "step": 67134,
      "training_step_time": 6.766256332397461
    },
    {
      "epoch": 6.40249252319336e-05,
      "model_forward_time": 1.7192020416259766,
      "step": 67135
    },
    {
      "epoch": 6.40249252319336e-05,
      "step": 67135,
      "training_step_time": 6.753921747207642
    },
    {
      "epoch": 6.402587890625e-05,
      "model_forward_time": 1.7277076244354248,
      "step": 67136
    },
    {
      "epoch": 6.402587890625e-05,
      "step": 67136,
      "training_step_time": 6.685034990310669
    },
    {
      "epoch": 6.40268325805664e-05,
      "model_forward_time": 1.7274298667907715,
      "step": 67137
    },
    {
      "epoch": 6.40268325805664e-05,
      "step": 67137,
      "training_step_time": 6.731634616851807
    },
    {
      "epoch": 6.402778625488282e-05,
      "model_forward_time": 2.921703338623047,
      "step": 67138
    },
    {
      "epoch": 6.402778625488282e-05,
      "step": 67138,
      "training_step_time": 7.7928783893585205
    },
    {
      "epoch": 6.402873992919921e-05,
      "model_forward_time": 1.722738265991211,
      "step": 67139
    },
    {
      "epoch": 6.402873992919921e-05,
      "step": 67139,
      "training_step_time": 12.000385761260986
    },
    {
      "epoch": 6.402969360351563e-05,
      "grad_norm": 0.09292872995138168,
      "learning_rate": 2.67292737971085e-06,
      "loss": 0.0596,
      "step": 67140
    },
    {
      "epoch": 6.402969360351563e-05,
      "model_forward_time": 1.7248592376708984,
      "step": 67140
    },
    {
      "dynamics_loss_avg": 0.05155236702412367,
      "epoch": 6.402969360351563e-05,
      "step": 67140
    },
    {
      "action_loss_avg": 0.001957006604061462,
      "epoch": 6.402969360351563e-05,
      "step": 67140
    },
    {
      "epoch": 6.402969360351563e-05,
      "step": 67140,
      "training_step_time": 7.667329549789429
    },
    {
      "epoch": 6.403064727783204e-05,
      "model_forward_time": 1.7294843196868896,
      "step": 67141
    },
    {
      "epoch": 6.403064727783204e-05,
      "step": 67141,
      "training_step_time": 6.754150867462158
    },
    {
      "epoch": 6.403160095214843e-05,
      "model_forward_time": 1.7301509380340576,
      "step": 67142
    },
    {
      "epoch": 6.403160095214843e-05,
      "step": 67142,
      "training_step_time": 6.748386383056641
    },
    {
      "epoch": 6.403255462646484e-05,
      "model_forward_time": 1.7355589866638184,
      "step": 67143
    },
    {
      "epoch": 6.403255462646484e-05,
      "step": 67143,
      "training_step_time": 6.74040675163269
    },
    {
      "epoch": 6.403350830078126e-05,
      "model_forward_time": 1.7301521301269531,
      "step": 67144
    },
    {
      "epoch": 6.403350830078126e-05,
      "step": 67144,
      "training_step_time": 6.758638858795166
    },
    {
      "epoch": 6.403446197509765e-05,
      "model_forward_time": 1.7179393768310547,
      "step": 67145
    },
    {
      "epoch": 6.403446197509765e-05,
      "step": 67145,
      "training_step_time": 6.761066436767578
    },
    {
      "epoch": 6.403541564941406e-05,
      "model_forward_time": 1.7227826118469238,
      "step": 67146
    },
    {
      "epoch": 6.403541564941406e-05,
      "step": 67146,
      "training_step_time": 6.7673420906066895
    },
    {
      "epoch": 6.403636932373047e-05,
      "model_forward_time": 2.947014331817627,
      "step": 67147
    },
    {
      "epoch": 6.403636932373047e-05,
      "step": 67147,
      "training_step_time": 7.822150468826294
    },
    {
      "epoch": 6.403732299804687e-05,
      "model_forward_time": 1.7369327545166016,
      "step": 67148
    },
    {
      "epoch": 6.403732299804687e-05,
      "step": 67148,
      "training_step_time": 10.160662651062012
    },
    {
      "epoch": 6.403827667236328e-05,
      "model_forward_time": 1.7301921844482422,
      "step": 67149
    },
    {
      "epoch": 6.403827667236328e-05,
      "step": 67149,
      "training_step_time": 8.91172194480896
    },
    {
      "epoch": 6.403923034667969e-05,
      "grad_norm": 0.10382182896137238,
      "learning_rate": 2.671464032696982e-06,
      "loss": 0.0567,
      "step": 67150
    },
    {
      "epoch": 6.403923034667969e-05,
      "model_forward_time": 1.7277495861053467,
      "step": 67150
    },
    {
      "dynamics_loss_avg": 0.0624107925221324,
      "epoch": 6.403923034667969e-05,
      "step": 67150
    },
    {
      "action_loss_avg": 0.0014558572554960846,
      "epoch": 6.403923034667969e-05,
      "step": 67150
    },
    {
      "epoch": 6.403923034667969e-05,
      "step": 67150,
      "training_step_time": 6.862247943878174
    },
    {
      "epoch": 6.404018402099609e-05,
      "model_forward_time": 1.7246944904327393,
      "step": 67151
    },
    {
      "epoch": 6.404018402099609e-05,
      "step": 67151,
      "training_step_time": 6.880651950836182
    },
    {
      "epoch": 6.40411376953125e-05,
      "model_forward_time": 1.7152841091156006,
      "step": 67152
    },
    {
      "epoch": 6.40411376953125e-05,
      "step": 67152,
      "training_step_time": 6.924309015274048
    },
    {
      "epoch": 6.404209136962891e-05,
      "model_forward_time": 1.7200732231140137,
      "step": 67153
    },
    {
      "epoch": 6.404209136962891e-05,
      "step": 67153,
      "training_step_time": 6.908763885498047
    },
    {
      "epoch": 6.404304504394531e-05,
      "model_forward_time": 1.72214674949646,
      "step": 67154
    },
    {
      "epoch": 6.404304504394531e-05,
      "step": 67154,
      "training_step_time": 6.876206398010254
    },
    {
      "epoch": 6.404399871826172e-05,
      "model_forward_time": 1.7111520767211914,
      "step": 67155
    },
    {
      "epoch": 6.404399871826172e-05,
      "step": 67155,
      "training_step_time": 6.9068922996521
    },
    {
      "epoch": 6.404495239257813e-05,
      "model_forward_time": 2.9002561569213867,
      "step": 67156
    },
    {
      "epoch": 6.404495239257813e-05,
      "step": 67156,
      "training_step_time": 7.831122636795044
    },
    {
      "epoch": 6.404590606689453e-05,
      "model_forward_time": 1.7125494480133057,
      "step": 67157
    },
    {
      "epoch": 6.404590606689453e-05,
      "step": 67157,
      "training_step_time": 8.755463600158691
    },
    {
      "epoch": 6.404685974121094e-05,
      "model_forward_time": 1.7156355381011963,
      "step": 67158
    },
    {
      "epoch": 6.404685974121094e-05,
      "step": 67158,
      "training_step_time": 9.445974349975586
    },
    {
      "epoch": 6.404781341552735e-05,
      "model_forward_time": 1.7109558582305908,
      "step": 67159
    },
    {
      "epoch": 6.404781341552735e-05,
      "step": 67159,
      "training_step_time": 7.845977544784546
    },
    {
      "epoch": 6.404876708984375e-05,
      "grad_norm": 0.10356659442186356,
      "learning_rate": 2.6700009403283056e-06,
      "loss": 0.0591,
      "step": 67160
    },
    {
      "epoch": 6.404876708984375e-05,
      "model_forward_time": 1.7118825912475586,
      "step": 67160
    },
    {
      "dynamics_loss_avg": 0.04942169152200222,
      "epoch": 6.404876708984375e-05,
      "step": 67160
    },
    {
      "action_loss_avg": 0.0016387416602810846,
      "epoch": 6.404876708984375e-05,
      "step": 67160
    },
    {
      "epoch": 6.404876708984375e-05,
      "step": 67160,
      "training_step_time": 6.942252159118652
    },
    {
      "epoch": 6.404972076416016e-05,
      "model_forward_time": 1.7140955924987793,
      "step": 67161
    },
    {
      "epoch": 6.404972076416016e-05,
      "step": 67161,
      "training_step_time": 6.927482843399048
    },
    {
      "epoch": 6.405067443847657e-05,
      "model_forward_time": 1.7129220962524414,
      "step": 67162
    },
    {
      "epoch": 6.405067443847657e-05,
      "step": 67162,
      "training_step_time": 6.940323352813721
    },
    {
      "epoch": 6.405162811279297e-05,
      "model_forward_time": 1.7143919467926025,
      "step": 67163
    },
    {
      "epoch": 6.405162811279297e-05,
      "step": 67163,
      "training_step_time": 6.980945110321045
    },
    {
      "epoch": 6.405258178710938e-05,
      "model_forward_time": 1.715306282043457,
      "step": 67164
    },
    {
      "epoch": 6.405258178710938e-05,
      "step": 67164,
      "training_step_time": 6.935412645339966
    },
    {
      "epoch": 6.405353546142579e-05,
      "model_forward_time": 1.7816433906555176,
      "step": 67165
    },
    {
      "epoch": 6.405353546142579e-05,
      "step": 67165,
      "training_step_time": 7.776110410690308
    },
    {
      "epoch": 6.405448913574218e-05,
      "model_forward_time": 2.528160333633423,
      "step": 67166
    },
    {
      "epoch": 6.405448913574218e-05,
      "step": 67166,
      "training_step_time": 8.62428092956543
    },
    {
      "epoch": 6.40554428100586e-05,
      "model_forward_time": 1.7189569473266602,
      "step": 67167
    },
    {
      "epoch": 6.40554428100586e-05,
      "step": 67167,
      "training_step_time": 9.773144483566284
    },
    {
      "epoch": 6.4056396484375e-05,
      "model_forward_time": 1.7160170078277588,
      "step": 67168
    },
    {
      "epoch": 6.4056396484375e-05,
      "step": 67168,
      "training_step_time": 7.797684669494629
    },
    {
      "epoch": 6.40573501586914e-05,
      "model_forward_time": 1.717536449432373,
      "step": 67169
    },
    {
      "epoch": 6.40573501586914e-05,
      "step": 67169,
      "training_step_time": 6.924039602279663
    },
    {
      "epoch": 6.405830383300781e-05,
      "grad_norm": 0.1348746418952942,
      "learning_rate": 2.6685381027648215e-06,
      "loss": 0.0588,
      "step": 67170
    },
    {
      "epoch": 6.405830383300781e-05,
      "model_forward_time": 1.7210242748260498,
      "step": 67170
    },
    {
      "dynamics_loss_avg": 0.052277164347469804,
      "epoch": 6.405830383300781e-05,
      "step": 67170
    },
    {
      "action_loss_avg": 0.002293397524044849,
      "epoch": 6.405830383300781e-05,
      "step": 67170
    },
    {
      "epoch": 6.405830383300781e-05,
      "step": 67170,
      "training_step_time": 6.873907804489136
    },
    {
      "epoch": 6.405925750732422e-05,
      "model_forward_time": 1.7173705101013184,
      "step": 67171
    },
    {
      "epoch": 6.405925750732422e-05,
      "step": 67171,
      "training_step_time": 7.065188884735107
    },
    {
      "epoch": 6.406021118164062e-05,
      "model_forward_time": 1.7166762351989746,
      "step": 67172
    },
    {
      "epoch": 6.406021118164062e-05,
      "step": 67172,
      "training_step_time": 6.890010356903076
    },
    {
      "epoch": 6.406116485595703e-05,
      "model_forward_time": 1.7194552421569824,
      "step": 67173
    },
    {
      "epoch": 6.406116485595703e-05,
      "step": 67173,
      "training_step_time": 6.658893585205078
    },
    {
      "epoch": 6.406211853027344e-05,
      "model_forward_time": 1.7522790431976318,
      "step": 67174
    },
    {
      "epoch": 6.406211853027344e-05,
      "step": 67174,
      "training_step_time": 7.484415531158447
    },
    {
      "epoch": 6.406307220458984e-05,
      "model_forward_time": 2.911332368850708,
      "step": 67175
    },
    {
      "epoch": 6.406307220458984e-05,
      "step": 67175,
      "training_step_time": 9.375268936157227
    },
    {
      "epoch": 6.406402587890625e-05,
      "model_forward_time": 1.7170650959014893,
      "step": 67176
    },
    {
      "epoch": 6.406402587890625e-05,
      "step": 67176,
      "training_step_time": 11.419701099395752
    },
    {
      "epoch": 6.406497955322266e-05,
      "model_forward_time": 1.720020055770874,
      "step": 67177
    },
    {
      "epoch": 6.406497955322266e-05,
      "step": 67177,
      "training_step_time": 7.59758448600769
    },
    {
      "epoch": 6.406593322753906e-05,
      "model_forward_time": 1.7213163375854492,
      "step": 67178
    },
    {
      "epoch": 6.406593322753906e-05,
      "step": 67178,
      "training_step_time": 6.68875789642334
    },
    {
      "epoch": 6.406688690185547e-05,
      "model_forward_time": 1.7220497131347656,
      "step": 67179
    },
    {
      "epoch": 6.406688690185547e-05,
      "step": 67179,
      "training_step_time": 6.6819236278533936
    },
    {
      "epoch": 6.406784057617188e-05,
      "grad_norm": 0.08942124247550964,
      "learning_rate": 2.667075520166509e-06,
      "loss": 0.0607,
      "step": 67180
    },
    {
      "epoch": 6.406784057617188e-05,
      "model_forward_time": 1.7136447429656982,
      "step": 67180
    },
    {
      "dynamics_loss_avg": 0.053054826147854325,
      "epoch": 6.406784057617188e-05,
      "step": 67180
    },
    {
      "action_loss_avg": 0.0017582568689249456,
      "epoch": 6.406784057617188e-05,
      "step": 67180
    },
    {
      "epoch": 6.406784057617188e-05,
      "step": 67180,
      "training_step_time": 6.697914123535156
    },
    {
      "epoch": 6.406879425048828e-05,
      "model_forward_time": 1.7202634811401367,
      "step": 67181
    },
    {
      "epoch": 6.406879425048828e-05,
      "step": 67181,
      "training_step_time": 6.707886219024658
    },
    {
      "epoch": 6.406974792480469e-05,
      "model_forward_time": 1.7099010944366455,
      "step": 67182
    },
    {
      "epoch": 6.406974792480469e-05,
      "step": 67182,
      "training_step_time": 6.717458486557007
    },
    {
      "epoch": 6.40707015991211e-05,
      "model_forward_time": 1.7701401710510254,
      "step": 67183
    },
    {
      "epoch": 6.40707015991211e-05,
      "step": 67183,
      "training_step_time": 7.504137277603149
    },
    {
      "epoch": 6.40716552734375e-05,
      "model_forward_time": 2.9121594429016113,
      "step": 67184
    },
    {
      "epoch": 6.40716552734375e-05,
      "step": 67184,
      "training_step_time": 9.490330457687378
    },
    {
      "epoch": 6.407260894775391e-05,
      "model_forward_time": 1.711911678314209,
      "step": 67185
    },
    {
      "epoch": 6.407260894775391e-05,
      "step": 67185,
      "training_step_time": 11.038827896118164
    },
    {
      "epoch": 6.407356262207032e-05,
      "model_forward_time": 1.7105164527893066,
      "step": 67186
    },
    {
      "epoch": 6.407356262207032e-05,
      "step": 67186,
      "training_step_time": 7.648858547210693
    },
    {
      "epoch": 6.407451629638672e-05,
      "model_forward_time": 1.713684320449829,
      "step": 67187
    },
    {
      "epoch": 6.407451629638672e-05,
      "step": 67187,
      "training_step_time": 7.507634162902832
    },
    {
      "epoch": 6.407546997070313e-05,
      "model_forward_time": 1.7178819179534912,
      "step": 67188
    },
    {
      "epoch": 6.407546997070313e-05,
      "step": 67188,
      "training_step_time": 6.664913177490234
    },
    {
      "epoch": 6.407642364501952e-05,
      "model_forward_time": 1.7176506519317627,
      "step": 67189
    },
    {
      "epoch": 6.407642364501952e-05,
      "step": 67189,
      "training_step_time": 6.711803913116455
    },
    {
      "epoch": 6.407737731933594e-05,
      "grad_norm": 0.09873031079769135,
      "learning_rate": 2.6656131926933066e-06,
      "loss": 0.059,
      "step": 67190
    },
    {
      "epoch": 6.407737731933594e-05,
      "model_forward_time": 1.7063462734222412,
      "step": 67190
    },
    {
      "dynamics_loss_avg": 0.05999622847884893,
      "epoch": 6.407737731933594e-05,
      "step": 67190
    },
    {
      "action_loss_avg": 0.001619999756803736,
      "epoch": 6.407737731933594e-05,
      "step": 67190
    },
    {
      "epoch": 6.407737731933594e-05,
      "step": 67190,
      "training_step_time": 6.699587821960449
    },
    {
      "epoch": 6.407833099365235e-05,
      "model_forward_time": 1.7162015438079834,
      "step": 67191
    },
    {
      "epoch": 6.407833099365235e-05,
      "step": 67191,
      "training_step_time": 6.722302675247192
    },
    {
      "epoch": 6.407928466796874e-05,
      "model_forward_time": 1.7371463775634766,
      "step": 67192
    },
    {
      "epoch": 6.407928466796874e-05,
      "step": 67192,
      "training_step_time": 7.4927191734313965
    },
    {
      "epoch": 6.408023834228515e-05,
      "model_forward_time": 2.8999252319335938,
      "step": 67193
    },
    {
      "epoch": 6.408023834228515e-05,
      "step": 67193,
      "training_step_time": 8.424464225769043
    },
    {
      "epoch": 6.408119201660156e-05,
      "model_forward_time": 1.7090787887573242,
      "step": 67194
    },
    {
      "epoch": 6.408119201660156e-05,
      "step": 67194,
      "training_step_time": 7.952631950378418
    },
    {
      "epoch": 6.408214569091796e-05,
      "model_forward_time": 1.711421251296997,
      "step": 67195
    },
    {
      "epoch": 6.408214569091796e-05,
      "step": 67195,
      "training_step_time": 7.9619951248168945
    },
    {
      "epoch": 6.408309936523437e-05,
      "model_forward_time": 1.7201635837554932,
      "step": 67196
    },
    {
      "epoch": 6.408309936523437e-05,
      "step": 67196,
      "training_step_time": 7.691036701202393
    },
    {
      "epoch": 6.408405303955078e-05,
      "model_forward_time": 1.716566801071167,
      "step": 67197
    },
    {
      "epoch": 6.408405303955078e-05,
      "step": 67197,
      "training_step_time": 6.887023687362671
    },
    {
      "epoch": 6.408500671386718e-05,
      "model_forward_time": 1.7156407833099365,
      "step": 67198
    },
    {
      "epoch": 6.408500671386718e-05,
      "step": 67198,
      "training_step_time": 6.879565715789795
    },
    {
      "epoch": 6.408596038818359e-05,
      "model_forward_time": 1.7170524597167969,
      "step": 67199
    },
    {
      "epoch": 6.408596038818359e-05,
      "step": 67199,
      "training_step_time": 6.877910614013672
    },
    {
      "epoch": 6.40869140625e-05,
      "grad_norm": 0.11285188794136047,
      "learning_rate": 2.664151120505136e-06,
      "loss": 0.0594,
      "step": 67200
    },
    {
      "epoch": 6.40869140625e-05,
      "model_forward_time": 1.3554365634918213,
      "step": 67200
    },
    {
      "dynamics_loss_avg": 0.06635654978454113,
      "epoch": 6.40869140625e-05,
      "step": 67200
    },
    {
      "action_loss_avg": 0.00150618361949455,
      "epoch": 6.40869140625e-05,
      "step": 67200
    },
    {
      "epoch": 6.40869140625e-05,
      "step": 67200,
      "training_step_time": 6.822753190994263
    },
    {
      "epoch": 6.40878677368164e-05,
      "model_forward_time": 1.861088514328003,
      "step": 67201
    },
    {
      "epoch": 6.40878677368164e-05,
      "step": 67201,
      "training_step_time": 7.458888053894043
    },
    {
      "epoch": 6.408882141113281e-05,
      "model_forward_time": 1.7240548133850098,
      "step": 67202
    },
    {
      "epoch": 6.408882141113281e-05,
      "step": 67202,
      "training_step_time": 9.089720249176025
    },
    {
      "epoch": 6.408977508544922e-05,
      "model_forward_time": 2.949185609817505,
      "step": 67203
    },
    {
      "epoch": 6.408977508544922e-05,
      "step": 67203,
      "training_step_time": 8.90768027305603
    },
    {
      "epoch": 6.409072875976562e-05,
      "model_forward_time": 1.721320390701294,
      "step": 67204
    },
    {
      "epoch": 6.409072875976562e-05,
      "step": 67204,
      "training_step_time": 9.812780618667603
    },
    {
      "epoch": 6.409168243408203e-05,
      "model_forward_time": 1.7189631462097168,
      "step": 67205
    },
    {
      "epoch": 6.409168243408203e-05,
      "step": 67205,
      "training_step_time": 7.771963119506836
    },
    {
      "epoch": 6.409263610839844e-05,
      "model_forward_time": 1.7230143547058105,
      "step": 67206
    },
    {
      "epoch": 6.409263610839844e-05,
      "step": 67206,
      "training_step_time": 6.871809959411621
    },
    {
      "epoch": 6.409358978271484e-05,
      "model_forward_time": 1.731909990310669,
      "step": 67207
    },
    {
      "epoch": 6.409358978271484e-05,
      "step": 67207,
      "training_step_time": 6.726097345352173
    },
    {
      "epoch": 6.409454345703125e-05,
      "model_forward_time": 1.7270910739898682,
      "step": 67208
    },
    {
      "epoch": 6.409454345703125e-05,
      "step": 67208,
      "training_step_time": 6.720078706741333
    },
    {
      "epoch": 6.409549713134766e-05,
      "model_forward_time": 1.7374465465545654,
      "step": 67209
    },
    {
      "epoch": 6.409549713134766e-05,
      "step": 67209,
      "training_step_time": 6.75795841217041
    },
    {
      "epoch": 6.409645080566406e-05,
      "grad_norm": 0.08479435741901398,
      "learning_rate": 2.6626893037618873e-06,
      "loss": 0.0572,
      "step": 67210
    },
    {
      "epoch": 6.409645080566406e-05,
      "model_forward_time": 1.50343918800354,
      "step": 67210
    },
    {
      "dynamics_loss_avg": 0.060507090762257576,
      "epoch": 6.409645080566406e-05,
      "step": 67210
    },
    {
      "action_loss_avg": 0.003065330570098013,
      "epoch": 6.409645080566406e-05,
      "step": 67210
    },
    {
      "epoch": 6.409645080566406e-05,
      "step": 67210,
      "training_step_time": 7.502084970474243
    },
    {
      "epoch": 6.409740447998047e-05,
      "model_forward_time": 1.7230005264282227,
      "step": 67211
    },
    {
      "epoch": 6.409740447998047e-05,
      "step": 67211,
      "training_step_time": 8.511759042739868
    },
    {
      "epoch": 6.409835815429688e-05,
      "model_forward_time": 2.917402505874634,
      "step": 67212
    },
    {
      "epoch": 6.409835815429688e-05,
      "step": 67212,
      "training_step_time": 10.248051166534424
    },
    {
      "epoch": 6.409931182861328e-05,
      "model_forward_time": 1.7269904613494873,
      "step": 67213
    },
    {
      "epoch": 6.409931182861328e-05,
      "step": 67213,
      "training_step_time": 10.44710898399353
    },
    {
      "epoch": 6.410026550292969e-05,
      "model_forward_time": 1.7253515720367432,
      "step": 67214
    },
    {
      "epoch": 6.410026550292969e-05,
      "step": 67214,
      "training_step_time": 7.613478899002075
    },
    {
      "epoch": 6.41012191772461e-05,
      "model_forward_time": 1.7261738777160645,
      "step": 67215
    },
    {
      "epoch": 6.41012191772461e-05,
      "step": 67215,
      "training_step_time": 6.684811115264893
    },
    {
      "epoch": 6.41021728515625e-05,
      "model_forward_time": 1.7402637004852295,
      "step": 67216
    },
    {
      "epoch": 6.41021728515625e-05,
      "step": 67216,
      "training_step_time": 6.672621488571167
    },
    {
      "epoch": 6.41031265258789e-05,
      "model_forward_time": 1.7316563129425049,
      "step": 67217
    },
    {
      "epoch": 6.41031265258789e-05,
      "step": 67217,
      "training_step_time": 6.701643943786621
    },
    {
      "epoch": 6.410408020019532e-05,
      "model_forward_time": 1.7227404117584229,
      "step": 67218
    },
    {
      "epoch": 6.410408020019532e-05,
      "step": 67218,
      "training_step_time": 6.675571441650391
    },
    {
      "epoch": 6.410503387451171e-05,
      "model_forward_time": 1.853212594985962,
      "step": 67219
    },
    {
      "epoch": 6.410503387451171e-05,
      "step": 67219,
      "training_step_time": 7.467379093170166
    },
    {
      "epoch": 6.410598754882812e-05,
      "grad_norm": 0.08651099354028702,
      "learning_rate": 2.6612277426234244e-06,
      "loss": 0.0588,
      "step": 67220
    },
    {
      "epoch": 6.410598754882812e-05,
      "model_forward_time": 1.7407503128051758,
      "step": 67220
    },
    {
      "dynamics_loss_avg": 0.052535415068268775,
      "epoch": 6.410598754882812e-05,
      "step": 67220
    },
    {
      "action_loss_avg": 0.0013215376791777089,
      "epoch": 6.410598754882812e-05,
      "step": 67220
    },
    {
      "epoch": 6.410598754882812e-05,
      "step": 67220,
      "training_step_time": 7.512479305267334
    },
    {
      "epoch": 6.410694122314453e-05,
      "model_forward_time": 2.902757406234741,
      "step": 67221
    },
    {
      "epoch": 6.410694122314453e-05,
      "step": 67221,
      "training_step_time": 7.652519464492798
    },
    {
      "epoch": 6.410789489746093e-05,
      "model_forward_time": 1.7305560111999512,
      "step": 67222
    },
    {
      "epoch": 6.410789489746093e-05,
      "step": 67222,
      "training_step_time": 8.531704664230347
    },
    {
      "epoch": 6.410884857177734e-05,
      "model_forward_time": 1.7375781536102295,
      "step": 67223
    },
    {
      "epoch": 6.410884857177734e-05,
      "step": 67223,
      "training_step_time": 7.594761371612549
    },
    {
      "epoch": 6.410980224609375e-05,
      "model_forward_time": 1.7363910675048828,
      "step": 67224
    },
    {
      "epoch": 6.410980224609375e-05,
      "step": 67224,
      "training_step_time": 7.358643531799316
    },
    {
      "epoch": 6.411075592041015e-05,
      "model_forward_time": 1.7396929264068604,
      "step": 67225
    },
    {
      "epoch": 6.411075592041015e-05,
      "step": 67225,
      "training_step_time": 6.650445222854614
    },
    {
      "epoch": 6.411170959472656e-05,
      "model_forward_time": 1.7345263957977295,
      "step": 67226
    },
    {
      "epoch": 6.411170959472656e-05,
      "step": 67226,
      "training_step_time": 6.6466758251190186
    },
    {
      "epoch": 6.411266326904297e-05,
      "model_forward_time": 1.7358644008636475,
      "step": 67227
    },
    {
      "epoch": 6.411266326904297e-05,
      "step": 67227,
      "training_step_time": 6.67107892036438
    },
    {
      "epoch": 6.411361694335937e-05,
      "model_forward_time": 1.8588824272155762,
      "step": 67228
    },
    {
      "epoch": 6.411361694335937e-05,
      "step": 67228,
      "training_step_time": 7.4545698165893555
    },
    {
      "epoch": 6.411457061767578e-05,
      "model_forward_time": 1.737145185470581,
      "step": 67229
    },
    {
      "epoch": 6.411457061767578e-05,
      "step": 67229,
      "training_step_time": 7.445004940032959
    },
    {
      "epoch": 6.411552429199219e-05,
      "grad_norm": 0.10547605156898499,
      "learning_rate": 2.659766437249577e-06,
      "loss": 0.0584,
      "step": 67230
    },
    {
      "epoch": 6.411552429199219e-05,
      "model_forward_time": 2.857966661453247,
      "step": 67230
    },
    {
      "dynamics_loss_avg": 0.05104455128312111,
      "epoch": 6.411552429199219e-05,
      "step": 67230
    },
    {
      "action_loss_avg": 0.0022511055634822697,
      "epoch": 6.411552429199219e-05,
      "step": 67230
    },
    {
      "epoch": 6.411552429199219e-05,
      "step": 67230,
      "training_step_time": 8.984880685806274
    },
    {
      "epoch": 6.411647796630859e-05,
      "model_forward_time": 1.728905439376831,
      "step": 67231
    },
    {
      "epoch": 6.411647796630859e-05,
      "step": 67231,
      "training_step_time": 9.473883390426636
    },
    {
      "epoch": 6.4117431640625e-05,
      "model_forward_time": 1.7342040538787842,
      "step": 67232
    },
    {
      "epoch": 6.4117431640625e-05,
      "step": 67232,
      "training_step_time": 7.581302642822266
    },
    {
      "epoch": 6.411838531494141e-05,
      "model_forward_time": 1.7393848896026611,
      "step": 67233
    },
    {
      "epoch": 6.411838531494141e-05,
      "step": 67233,
      "training_step_time": 7.382002353668213
    },
    {
      "epoch": 6.411933898925781e-05,
      "model_forward_time": 1.7275242805480957,
      "step": 67234
    },
    {
      "epoch": 6.411933898925781e-05,
      "step": 67234,
      "training_step_time": 6.609342336654663
    },
    {
      "epoch": 6.412029266357422e-05,
      "model_forward_time": 1.7325925827026367,
      "step": 67235
    },
    {
      "epoch": 6.412029266357422e-05,
      "step": 67235,
      "training_step_time": 6.623405456542969
    },
    {
      "epoch": 6.412124633789063e-05,
      "model_forward_time": 1.748645305633545,
      "step": 67236
    },
    {
      "epoch": 6.412124633789063e-05,
      "step": 67236,
      "training_step_time": 6.605376720428467
    },
    {
      "epoch": 6.412220001220703e-05,
      "model_forward_time": 1.7346243858337402,
      "step": 67237
    },
    {
      "epoch": 6.412220001220703e-05,
      "step": 67237,
      "training_step_time": 7.628393650054932
    },
    {
      "epoch": 6.412315368652344e-05,
      "model_forward_time": 1.739271879196167,
      "step": 67238
    },
    {
      "epoch": 6.412315368652344e-05,
      "step": 67238,
      "training_step_time": 8.492849826812744
    },
    {
      "epoch": 6.412410736083985e-05,
      "model_forward_time": 1.7345881462097168,
      "step": 67239
    },
    {
      "epoch": 6.412410736083985e-05,
      "step": 67239,
      "training_step_time": 8.031212329864502
    },
    {
      "epoch": 6.412506103515625e-05,
      "grad_norm": 0.09070658683776855,
      "learning_rate": 2.658305387800156e-06,
      "loss": 0.055,
      "step": 67240
    },
    {
      "epoch": 6.412506103515625e-05,
      "model_forward_time": 2.2703957557678223,
      "step": 67240
    },
    {
      "dynamics_loss_avg": 0.05417335759848356,
      "epoch": 6.412506103515625e-05,
      "step": 67240
    },
    {
      "action_loss_avg": 0.0014216958137694747,
      "epoch": 6.412506103515625e-05,
      "step": 67240
    },
    {
      "epoch": 6.412506103515625e-05,
      "step": 67240,
      "training_step_time": 8.598212242126465
    },
    {
      "epoch": 6.412601470947266e-05,
      "model_forward_time": 1.7280137538909912,
      "step": 67241
    },
    {
      "epoch": 6.412601470947266e-05,
      "step": 67241,
      "training_step_time": 7.733938217163086
    },
    {
      "epoch": 6.412696838378907e-05,
      "model_forward_time": 1.7217540740966797,
      "step": 67242
    },
    {
      "epoch": 6.412696838378907e-05,
      "step": 67242,
      "training_step_time": 7.618258953094482
    },
    {
      "epoch": 6.412792205810546e-05,
      "model_forward_time": 1.7247869968414307,
      "step": 67243
    },
    {
      "epoch": 6.412792205810546e-05,
      "step": 67243,
      "training_step_time": 6.664465665817261
    },
    {
      "epoch": 6.412887573242187e-05,
      "model_forward_time": 1.7182550430297852,
      "step": 67244
    },
    {
      "epoch": 6.412887573242187e-05,
      "step": 67244,
      "training_step_time": 6.6622703075408936
    },
    {
      "epoch": 6.412982940673829e-05,
      "model_forward_time": 1.7175207138061523,
      "step": 67245
    },
    {
      "epoch": 6.412982940673829e-05,
      "step": 67245,
      "training_step_time": 6.658300161361694
    },
    {
      "epoch": 6.413078308105468e-05,
      "model_forward_time": 1.717862606048584,
      "step": 67246
    },
    {
      "epoch": 6.413078308105468e-05,
      "step": 67246,
      "training_step_time": 7.688554286956787
    },
    {
      "epoch": 6.41317367553711e-05,
      "model_forward_time": 1.7211418151855469,
      "step": 67247
    },
    {
      "epoch": 6.41317367553711e-05,
      "step": 67247,
      "training_step_time": 7.5347580909729
    },
    {
      "epoch": 6.41326904296875e-05,
      "model_forward_time": 1.7340953350067139,
      "step": 67248
    },
    {
      "epoch": 6.41326904296875e-05,
      "step": 67248,
      "training_step_time": 7.585543632507324
    },
    {
      "epoch": 6.41336441040039e-05,
      "model_forward_time": 2.290271520614624,
      "step": 67249
    },
    {
      "epoch": 6.41336441040039e-05,
      "step": 67249,
      "training_step_time": 8.839919090270996
    },
    {
      "epoch": 6.413459777832031e-05,
      "grad_norm": 0.13317885994911194,
      "learning_rate": 2.6568445944349327e-06,
      "loss": 0.0594,
      "step": 67250
    },
    {
      "epoch": 6.413459777832031e-05,
      "model_forward_time": 1.7208430767059326,
      "step": 67250
    },
    {
      "dynamics_loss_avg": 0.06727031534537672,
      "epoch": 6.413459777832031e-05,
      "step": 67250
    },
    {
      "action_loss_avg": 0.0010853705520275981,
      "epoch": 6.413459777832031e-05,
      "step": 67250
    },
    {
      "epoch": 6.413459777832031e-05,
      "step": 67250,
      "training_step_time": 10.719231367111206
    },
    {
      "epoch": 6.413555145263672e-05,
      "model_forward_time": 1.7189218997955322,
      "step": 67251
    },
    {
      "epoch": 6.413555145263672e-05,
      "step": 67251,
      "training_step_time": 7.651053428649902
    },
    {
      "epoch": 6.413650512695312e-05,
      "model_forward_time": 1.7272651195526123,
      "step": 67252
    },
    {
      "epoch": 6.413650512695312e-05,
      "step": 67252,
      "training_step_time": 6.623451232910156
    },
    {
      "epoch": 6.413745880126953e-05,
      "model_forward_time": 1.720519781112671,
      "step": 67253
    },
    {
      "epoch": 6.413745880126953e-05,
      "step": 67253,
      "training_step_time": 6.634844779968262
    },
    {
      "epoch": 6.413841247558594e-05,
      "model_forward_time": 1.7216486930847168,
      "step": 67254
    },
    {
      "epoch": 6.413841247558594e-05,
      "step": 67254,
      "training_step_time": 6.650274038314819
    },
    {
      "epoch": 6.413936614990234e-05,
      "model_forward_time": 1.719822883605957,
      "step": 67255
    },
    {
      "epoch": 6.413936614990234e-05,
      "step": 67255,
      "training_step_time": 7.603384494781494
    },
    {
      "epoch": 6.414031982421875e-05,
      "model_forward_time": 1.7182481288909912,
      "step": 67256
    },
    {
      "epoch": 6.414031982421875e-05,
      "step": 67256,
      "training_step_time": 8.140359878540039
    },
    {
      "epoch": 6.414127349853516e-05,
      "model_forward_time": 1.7297930717468262,
      "step": 67257
    },
    {
      "epoch": 6.414127349853516e-05,
      "step": 67257,
      "training_step_time": 7.502503871917725
    },
    {
      "epoch": 6.414222717285156e-05,
      "model_forward_time": 2.3078839778900146,
      "step": 67258
    },
    {
      "epoch": 6.414222717285156e-05,
      "step": 67258,
      "training_step_time": 8.645640850067139
    },
    {
      "epoch": 6.414318084716797e-05,
      "model_forward_time": 1.7204868793487549,
      "step": 67259
    },
    {
      "epoch": 6.414318084716797e-05,
      "step": 67259,
      "training_step_time": 10.247116088867188
    },
    {
      "epoch": 6.414413452148438e-05,
      "grad_norm": 0.10556676238775253,
      "learning_rate": 2.6553840573136657e-06,
      "loss": 0.0569,
      "step": 67260
    },
    {
      "epoch": 6.414413452148438e-05,
      "model_forward_time": 1.7213270664215088,
      "step": 67260
    },
    {
      "dynamics_loss_avg": 0.044420497678220275,
      "epoch": 6.414413452148438e-05,
      "step": 67260
    },
    {
      "action_loss_avg": 0.002394544376875274,
      "epoch": 6.414413452148438e-05,
      "step": 67260
    },
    {
      "epoch": 6.414413452148438e-05,
      "step": 67260,
      "training_step_time": 7.582502603530884
    },
    {
      "epoch": 6.414508819580078e-05,
      "model_forward_time": 1.7260658740997314,
      "step": 67261
    },
    {
      "epoch": 6.414508819580078e-05,
      "step": 67261,
      "training_step_time": 6.648711919784546
    },
    {
      "epoch": 6.414604187011719e-05,
      "model_forward_time": 1.7233726978302002,
      "step": 67262
    },
    {
      "epoch": 6.414604187011719e-05,
      "step": 67262,
      "training_step_time": 6.6369242668151855
    },
    {
      "epoch": 6.41469955444336e-05,
      "model_forward_time": 1.71681547164917,
      "step": 67263
    },
    {
      "epoch": 6.41469955444336e-05,
      "step": 67263,
      "training_step_time": 6.625420093536377
    },
    {
      "epoch": 6.414794921875e-05,
      "model_forward_time": 1.7264676094055176,
      "step": 67264
    },
    {
      "epoch": 6.414794921875e-05,
      "step": 67264,
      "training_step_time": 7.580387353897095
    },
    {
      "epoch": 6.414890289306641e-05,
      "model_forward_time": 1.7233703136444092,
      "step": 67265
    },
    {
      "epoch": 6.414890289306641e-05,
      "step": 67265,
      "training_step_time": 7.471727132797241
    },
    {
      "epoch": 6.414985656738282e-05,
      "model_forward_time": 1.7304332256317139,
      "step": 67266
    },
    {
      "epoch": 6.414985656738282e-05,
      "step": 67266,
      "training_step_time": 7.548442363739014
    },
    {
      "epoch": 6.415081024169921e-05,
      "model_forward_time": 2.347301721572876,
      "step": 67267
    },
    {
      "epoch": 6.415081024169921e-05,
      "step": 67267,
      "training_step_time": 9.033510446548462
    },
    {
      "epoch": 6.415176391601563e-05,
      "model_forward_time": 1.7208189964294434,
      "step": 67268
    },
    {
      "epoch": 6.415176391601563e-05,
      "step": 67268,
      "training_step_time": 11.083479404449463
    },
    {
      "epoch": 6.415271759033204e-05,
      "model_forward_time": 1.725102424621582,
      "step": 67269
    },
    {
      "epoch": 6.415271759033204e-05,
      "step": 67269,
      "training_step_time": 9.67067265510559
    },
    {
      "epoch": 6.415367126464843e-05,
      "grad_norm": 0.11614267528057098,
      "learning_rate": 2.6539237765960706e-06,
      "loss": 0.0528,
      "step": 67270
    },
    {
      "epoch": 6.415367126464843e-05,
      "model_forward_time": 1.7204337120056152,
      "step": 67270
    },
    {
      "dynamics_loss_avg": 0.04815841261297464,
      "epoch": 6.415367126464843e-05,
      "step": 67270
    },
    {
      "action_loss_avg": 0.0015283351065590977,
      "epoch": 6.415367126464843e-05,
      "step": 67270
    },
    {
      "epoch": 6.415367126464843e-05,
      "step": 67270,
      "training_step_time": 7.164427280426025
    },
    {
      "epoch": 6.415462493896484e-05,
      "model_forward_time": 1.7194786071777344,
      "step": 67271
    },
    {
      "epoch": 6.415462493896484e-05,
      "step": 67271,
      "training_step_time": 6.688960552215576
    },
    {
      "epoch": 6.415557861328126e-05,
      "model_forward_time": 1.7223711013793945,
      "step": 67272
    },
    {
      "epoch": 6.415557861328126e-05,
      "step": 67272,
      "training_step_time": 6.684702396392822
    },
    {
      "epoch": 6.415653228759765e-05,
      "model_forward_time": 1.719437837600708,
      "step": 67273
    },
    {
      "epoch": 6.415653228759765e-05,
      "step": 67273,
      "training_step_time": 6.677629470825195
    },
    {
      "epoch": 6.415748596191406e-05,
      "model_forward_time": 1.7121577262878418,
      "step": 67274
    },
    {
      "epoch": 6.415748596191406e-05,
      "step": 67274,
      "training_step_time": 7.5975117683410645
    },
    {
      "epoch": 6.415843963623047e-05,
      "model_forward_time": 1.3365757465362549,
      "step": 67275
    },
    {
      "epoch": 6.415843963623047e-05,
      "step": 67275,
      "training_step_time": 7.521311283111572
    },
    {
      "epoch": 6.415939331054687e-05,
      "model_forward_time": 2.2854421138763428,
      "step": 67276
    },
    {
      "epoch": 6.415939331054687e-05,
      "step": 67276,
      "training_step_time": 8.759196043014526
    },
    {
      "epoch": 6.416034698486328e-05,
      "model_forward_time": 1.7250802516937256,
      "step": 67277
    },
    {
      "epoch": 6.416034698486328e-05,
      "step": 67277,
      "training_step_time": 8.549312353134155
    },
    {
      "epoch": 6.416130065917969e-05,
      "model_forward_time": 1.7247908115386963,
      "step": 67278
    },
    {
      "epoch": 6.416130065917969e-05,
      "step": 67278,
      "training_step_time": 8.691507816314697
    },
    {
      "epoch": 6.416225433349609e-05,
      "model_forward_time": 1.7232942581176758,
      "step": 67279
    },
    {
      "epoch": 6.416225433349609e-05,
      "step": 67279,
      "training_step_time": 7.5242979526519775
    },
    {
      "epoch": 6.41632080078125e-05,
      "grad_norm": 0.10184986889362335,
      "learning_rate": 2.652463752441847e-06,
      "loss": 0.0542,
      "step": 67280
    },
    {
      "epoch": 6.41632080078125e-05,
      "model_forward_time": 1.730712890625,
      "step": 67280
    },
    {
      "dynamics_loss_avg": 0.050018937513232234,
      "epoch": 6.41632080078125e-05,
      "step": 67280
    },
    {
      "action_loss_avg": 0.0015812244324479251,
      "epoch": 6.41632080078125e-05,
      "step": 67280
    },
    {
      "epoch": 6.41632080078125e-05,
      "step": 67280,
      "training_step_time": 6.6856348514556885
    },
    {
      "epoch": 6.416416168212891e-05,
      "model_forward_time": 1.7231836318969727,
      "step": 67281
    },
    {
      "epoch": 6.416416168212891e-05,
      "step": 67281,
      "training_step_time": 6.665715217590332
    },
    {
      "epoch": 6.416511535644531e-05,
      "model_forward_time": 1.7241218090057373,
      "step": 67282
    },
    {
      "epoch": 6.416511535644531e-05,
      "step": 67282,
      "training_step_time": 6.70215368270874
    },
    {
      "epoch": 6.416606903076172e-05,
      "model_forward_time": 1.7178843021392822,
      "step": 67283
    },
    {
      "epoch": 6.416606903076172e-05,
      "step": 67283,
      "training_step_time": 7.62479043006897
    },
    {
      "epoch": 6.416702270507813e-05,
      "model_forward_time": 1.752108097076416,
      "step": 67284
    },
    {
      "epoch": 6.416702270507813e-05,
      "step": 67284,
      "training_step_time": 7.530330181121826
    },
    {
      "epoch": 6.416797637939453e-05,
      "model_forward_time": 1.738591194152832,
      "step": 67285
    },
    {
      "epoch": 6.416797637939453e-05,
      "step": 67285,
      "training_step_time": 8.616729497909546
    },
    {
      "epoch": 6.416893005371094e-05,
      "model_forward_time": 2.2982914447784424,
      "step": 67286
    },
    {
      "epoch": 6.416893005371094e-05,
      "step": 67286,
      "training_step_time": 9.66097092628479
    },
    {
      "epoch": 6.416988372802735e-05,
      "model_forward_time": 1.7518610954284668,
      "step": 67287
    },
    {
      "epoch": 6.416988372802735e-05,
      "step": 67287,
      "training_step_time": 9.955450534820557
    },
    {
      "epoch": 6.417083740234375e-05,
      "model_forward_time": 1.7337031364440918,
      "step": 67288
    },
    {
      "epoch": 6.417083740234375e-05,
      "step": 67288,
      "training_step_time": 7.532180547714233
    },
    {
      "epoch": 6.417179107666016e-05,
      "model_forward_time": 1.7425215244293213,
      "step": 67289
    },
    {
      "epoch": 6.417179107666016e-05,
      "step": 67289,
      "training_step_time": 6.689083099365234
    },
    {
      "epoch": 6.417274475097657e-05,
      "grad_norm": 0.10825745016336441,
      "learning_rate": 2.651003985010655e-06,
      "loss": 0.058,
      "step": 67290
    },
    {
      "epoch": 6.417274475097657e-05,
      "model_forward_time": 1.7368967533111572,
      "step": 67290
    },
    {
      "dynamics_loss_avg": 0.04909944534301758,
      "epoch": 6.417274475097657e-05,
      "step": 67290
    },
    {
      "action_loss_avg": 0.001628454255114775,
      "epoch": 6.417274475097657e-05,
      "step": 67290
    },
    {
      "epoch": 6.417274475097657e-05,
      "step": 67290,
      "training_step_time": 6.670223712921143
    },
    {
      "epoch": 6.417369842529297e-05,
      "model_forward_time": 1.7296373844146729,
      "step": 67291
    },
    {
      "epoch": 6.417369842529297e-05,
      "step": 67291,
      "training_step_time": 6.683881998062134
    },
    {
      "epoch": 6.417465209960938e-05,
      "model_forward_time": 1.741365909576416,
      "step": 67292
    },
    {
      "epoch": 6.417465209960938e-05,
      "step": 67292,
      "training_step_time": 7.57296895980835
    },
    {
      "epoch": 6.417560577392579e-05,
      "model_forward_time": 1.729222059249878,
      "step": 67293
    },
    {
      "epoch": 6.417560577392579e-05,
      "step": 67293,
      "training_step_time": 7.534612655639648
    },
    {
      "epoch": 6.417655944824218e-05,
      "model_forward_time": 1.7503044605255127,
      "step": 67294
    },
    {
      "epoch": 6.417655944824218e-05,
      "step": 67294,
      "training_step_time": 7.471813201904297
    },
    {
      "epoch": 6.41775131225586e-05,
      "model_forward_time": 2.351358413696289,
      "step": 67295
    },
    {
      "epoch": 6.41775131225586e-05,
      "step": 67295,
      "training_step_time": 9.328632593154907
    },
    {
      "epoch": 6.4178466796875e-05,
      "model_forward_time": 1.737518310546875,
      "step": 67296
    },
    {
      "epoch": 6.4178466796875e-05,
      "step": 67296,
      "training_step_time": 8.688083410263062
    },
    {
      "epoch": 6.41794204711914e-05,
      "model_forward_time": 1.7317116260528564,
      "step": 67297
    },
    {
      "epoch": 6.41794204711914e-05,
      "step": 67297,
      "training_step_time": 7.819207668304443
    },
    {
      "epoch": 6.418037414550781e-05,
      "model_forward_time": 1.7361152172088623,
      "step": 67298
    },
    {
      "epoch": 6.418037414550781e-05,
      "step": 67298,
      "training_step_time": 6.871624231338501
    },
    {
      "epoch": 6.418132781982423e-05,
      "model_forward_time": 1.7365105152130127,
      "step": 67299
    },
    {
      "epoch": 6.418132781982423e-05,
      "step": 67299,
      "training_step_time": 6.864572525024414
    },
    {
      "epoch": 6.418228149414062e-05,
      "grad_norm": 0.09765782952308655,
      "learning_rate": 2.6495444744621378e-06,
      "loss": 0.0587,
      "step": 67300
    },
    {
      "epoch": 6.418228149414062e-05,
      "model_forward_time": 1.7361111640930176,
      "step": 67300
    },
    {
      "dynamics_loss_avg": 0.056826427765190604,
      "epoch": 6.418228149414062e-05,
      "step": 67300
    },
    {
      "action_loss_avg": 0.0014480072248261421,
      "epoch": 6.418228149414062e-05,
      "step": 67300
    },
    {
      "epoch": 6.418228149414062e-05,
      "step": 67300,
      "training_step_time": 6.856024503707886
    },
    {
      "epoch": 6.418323516845703e-05,
      "model_forward_time": 1.7315869331359863,
      "step": 67301
    },
    {
      "epoch": 6.418323516845703e-05,
      "step": 67301,
      "training_step_time": 7.865854501724243
    },
    {
      "epoch": 6.418418884277344e-05,
      "model_forward_time": 1.741926908493042,
      "step": 67302
    },
    {
      "epoch": 6.418418884277344e-05,
      "step": 67302,
      "training_step_time": 7.532937526702881
    },
    {
      "epoch": 6.418514251708984e-05,
      "model_forward_time": 1.7420575618743896,
      "step": 67303
    },
    {
      "epoch": 6.418514251708984e-05,
      "step": 67303,
      "training_step_time": 7.548992395401001
    },
    {
      "epoch": 6.418609619140625e-05,
      "model_forward_time": 2.3400001525878906,
      "step": 67304
    },
    {
      "epoch": 6.418609619140625e-05,
      "step": 67304,
      "training_step_time": 8.792356729507446
    },
    {
      "epoch": 6.418704986572266e-05,
      "model_forward_time": 1.7621631622314453,
      "step": 67305
    },
    {
      "epoch": 6.418704986572266e-05,
      "step": 67305,
      "training_step_time": 10.374957084655762
    },
    {
      "epoch": 6.418800354003906e-05,
      "model_forward_time": 1.741137981414795,
      "step": 67306
    },
    {
      "epoch": 6.418800354003906e-05,
      "step": 67306,
      "training_step_time": 7.940903425216675
    },
    {
      "epoch": 6.418895721435547e-05,
      "model_forward_time": 1.7481496334075928,
      "step": 67307
    },
    {
      "epoch": 6.418895721435547e-05,
      "step": 67307,
      "training_step_time": 7.380841493606567
    },
    {
      "epoch": 6.418991088867188e-05,
      "model_forward_time": 1.7399137020111084,
      "step": 67308
    },
    {
      "epoch": 6.418991088867188e-05,
      "step": 67308,
      "training_step_time": 6.901857376098633
    },
    {
      "epoch": 6.419086456298828e-05,
      "model_forward_time": 1.760704517364502,
      "step": 67309
    },
    {
      "epoch": 6.419086456298828e-05,
      "step": 67309,
      "training_step_time": 6.9733500480651855
    },
    {
      "epoch": 6.419181823730469e-05,
      "grad_norm": 0.08444872498512268,
      "learning_rate": 2.6480852209559015e-06,
      "loss": 0.0528,
      "step": 67310
    },
    {
      "epoch": 6.419181823730469e-05,
      "model_forward_time": 1.7594354152679443,
      "step": 67310
    },
    {
      "dynamics_loss_avg": 0.049145590234547856,
      "epoch": 6.419181823730469e-05,
      "step": 67310
    },
    {
      "action_loss_avg": 0.0015090187691384927,
      "epoch": 6.419181823730469e-05,
      "step": 67310
    },
    {
      "epoch": 6.419181823730469e-05,
      "step": 67310,
      "training_step_time": 7.704050540924072
    },
    {
      "epoch": 6.41927719116211e-05,
      "model_forward_time": 1.7391505241394043,
      "step": 67311
    },
    {
      "epoch": 6.41927719116211e-05,
      "step": 67311,
      "training_step_time": 7.752395153045654
    },
    {
      "epoch": 6.41937255859375e-05,
      "model_forward_time": 1.7604117393493652,
      "step": 67312
    },
    {
      "epoch": 6.41937255859375e-05,
      "step": 67312,
      "training_step_time": 7.73646354675293
    },
    {
      "epoch": 6.419467926025391e-05,
      "model_forward_time": 1.9505400657653809,
      "step": 67313
    },
    {
      "epoch": 6.419467926025391e-05,
      "step": 67313,
      "training_step_time": 8.026439905166626
    },
    {
      "epoch": 6.41956329345703e-05,
      "model_forward_time": 1.373920202255249,
      "step": 67314
    },
    {
      "epoch": 6.41956329345703e-05,
      "step": 67314,
      "training_step_time": 8.67823052406311
    },
    {
      "epoch": 6.419658660888672e-05,
      "model_forward_time": 1.7488882541656494,
      "step": 67315
    },
    {
      "epoch": 6.419658660888672e-05,
      "step": 67315,
      "training_step_time": 7.896210432052612
    },
    {
      "epoch": 6.419754028320313e-05,
      "model_forward_time": 1.748729944229126,
      "step": 67316
    },
    {
      "epoch": 6.419754028320313e-05,
      "step": 67316,
      "training_step_time": 7.7759997844696045
    },
    {
      "epoch": 6.419849395751952e-05,
      "model_forward_time": 1.7561445236206055,
      "step": 67317
    },
    {
      "epoch": 6.419849395751952e-05,
      "step": 67317,
      "training_step_time": 6.998871564865112
    },
    {
      "epoch": 6.419944763183594e-05,
      "model_forward_time": 1.7473397254943848,
      "step": 67318
    },
    {
      "epoch": 6.419944763183594e-05,
      "step": 67318,
      "training_step_time": 6.990449666976929
    },
    {
      "epoch": 6.420040130615235e-05,
      "model_forward_time": 1.7410926818847656,
      "step": 67319
    },
    {
      "epoch": 6.420040130615235e-05,
      "step": 67319,
      "training_step_time": 8.68034315109253
    },
    {
      "epoch": 6.420135498046874e-05,
      "grad_norm": 0.08668316900730133,
      "learning_rate": 2.6466262246515293e-06,
      "loss": 0.0544,
      "step": 67320
    },
    {
      "epoch": 6.420135498046874e-05,
      "model_forward_time": 1.7457571029663086,
      "step": 67320
    },
    {
      "dynamics_loss_avg": 0.04914404284209013,
      "epoch": 6.420135498046874e-05,
      "step": 67320
    },
    {
      "action_loss_avg": 0.001641551626380533,
      "epoch": 6.420135498046874e-05,
      "step": 67320
    },
    {
      "epoch": 6.420135498046874e-05,
      "step": 67320,
      "training_step_time": 7.736534118652344
    },
    {
      "epoch": 6.420230865478515e-05,
      "model_forward_time": 1.7451298236846924,
      "step": 67321
    },
    {
      "epoch": 6.420230865478515e-05,
      "step": 67321,
      "training_step_time": 7.742970943450928
    },
    {
      "epoch": 6.420326232910157e-05,
      "model_forward_time": 1.7353644371032715,
      "step": 67322
    },
    {
      "epoch": 6.420326232910157e-05,
      "step": 67322,
      "training_step_time": 7.940924882888794
    },
    {
      "epoch": 6.420421600341796e-05,
      "model_forward_time": 2.3814077377319336,
      "step": 67323
    },
    {
      "epoch": 6.420421600341796e-05,
      "step": 67323,
      "training_step_time": 7.9690260887146
    },
    {
      "epoch": 6.420516967773437e-05,
      "model_forward_time": 1.739631175994873,
      "step": 67324
    },
    {
      "epoch": 6.420516967773437e-05,
      "step": 67324,
      "training_step_time": 8.88867974281311
    },
    {
      "epoch": 6.420612335205078e-05,
      "model_forward_time": 1.7440259456634521,
      "step": 67325
    },
    {
      "epoch": 6.420612335205078e-05,
      "step": 67325,
      "training_step_time": 7.879576206207275
    },
    {
      "epoch": 6.420707702636718e-05,
      "model_forward_time": 1.7373685836791992,
      "step": 67326
    },
    {
      "epoch": 6.420707702636718e-05,
      "step": 67326,
      "training_step_time": 7.415248394012451
    },
    {
      "epoch": 6.420803070068359e-05,
      "model_forward_time": 1.7489142417907715,
      "step": 67327
    },
    {
      "epoch": 6.420803070068359e-05,
      "step": 67327,
      "training_step_time": 6.959731101989746
    },
    {
      "epoch": 6.4208984375e-05,
      "model_forward_time": 1.7420570850372314,
      "step": 67328
    },
    {
      "epoch": 6.4208984375e-05,
      "step": 67328,
      "training_step_time": 7.809979677200317
    },
    {
      "epoch": 6.42099380493164e-05,
      "model_forward_time": 1.7352688312530518,
      "step": 67329
    },
    {
      "epoch": 6.42099380493164e-05,
      "step": 67329,
      "training_step_time": 7.872246742248535
    },
    {
      "epoch": 6.421089172363281e-05,
      "grad_norm": 0.07243054360151291,
      "learning_rate": 2.6451674857085746e-06,
      "loss": 0.0575,
      "step": 67330
    },
    {
      "epoch": 6.421089172363281e-05,
      "model_forward_time": 1.7380425930023193,
      "step": 67330
    },
    {
      "dynamics_loss_avg": 0.06893915925174951,
      "epoch": 6.421089172363281e-05,
      "step": 67330
    },
    {
      "action_loss_avg": 0.0030687126447446643,
      "epoch": 6.421089172363281e-05,
      "step": 67330
    },
    {
      "epoch": 6.421089172363281e-05,
      "step": 67330,
      "training_step_time": 7.813680410385132
    },
    {
      "epoch": 6.421184539794922e-05,
      "model_forward_time": 1.7366552352905273,
      "step": 67331
    },
    {
      "epoch": 6.421184539794922e-05,
      "step": 67331,
      "training_step_time": 8.750607967376709
    },
    {
      "epoch": 6.421279907226562e-05,
      "model_forward_time": 2.3699426651000977,
      "step": 67332
    },
    {
      "epoch": 6.421279907226562e-05,
      "step": 67332,
      "training_step_time": 9.707037687301636
    },
    {
      "epoch": 6.421375274658203e-05,
      "model_forward_time": 1.7474031448364258,
      "step": 67333
    },
    {
      "epoch": 6.421375274658203e-05,
      "step": 67333,
      "training_step_time": 9.527084589004517
    },
    {
      "epoch": 6.421470642089844e-05,
      "model_forward_time": 1.7349045276641846,
      "step": 67334
    },
    {
      "epoch": 6.421470642089844e-05,
      "step": 67334,
      "training_step_time": 9.033931970596313
    },
    {
      "epoch": 6.421566009521484e-05,
      "model_forward_time": 1.7570176124572754,
      "step": 67335
    },
    {
      "epoch": 6.421566009521484e-05,
      "step": 67335,
      "training_step_time": 7.375413417816162
    },
    {
      "epoch": 6.421661376953125e-05,
      "model_forward_time": 1.752638339996338,
      "step": 67336
    },
    {
      "epoch": 6.421661376953125e-05,
      "step": 67336,
      "training_step_time": 6.917356252670288
    },
    {
      "epoch": 6.421756744384766e-05,
      "model_forward_time": 1.7411510944366455,
      "step": 67337
    },
    {
      "epoch": 6.421756744384766e-05,
      "step": 67337,
      "training_step_time": 7.734766721725464
    },
    {
      "epoch": 6.421852111816406e-05,
      "model_forward_time": 1.7332487106323242,
      "step": 67338
    },
    {
      "epoch": 6.421852111816406e-05,
      "step": 67338,
      "training_step_time": 7.793774127960205
    },
    {
      "epoch": 6.421947479248047e-05,
      "model_forward_time": 1.7454793453216553,
      "step": 67339
    },
    {
      "epoch": 6.421947479248047e-05,
      "step": 67339,
      "training_step_time": 8.62877607345581
    },
    {
      "epoch": 6.422042846679688e-05,
      "grad_norm": 0.06696047633886337,
      "learning_rate": 2.6437090042865655e-06,
      "loss": 0.0548,
      "step": 67340
    },
    {
      "epoch": 6.422042846679688e-05,
      "model_forward_time": 1.7358131408691406,
      "step": 67340
    },
    {
      "dynamics_loss_avg": 0.05005311043933034,
      "epoch": 6.422042846679688e-05,
      "step": 67340
    },
    {
      "action_loss_avg": 0.0017385599872795864,
      "epoch": 6.422042846679688e-05,
      "step": 67340
    },
    {
      "epoch": 6.422042846679688e-05,
      "step": 67340,
      "training_step_time": 9.153832912445068
    },
    {
      "epoch": 6.422138214111328e-05,
      "model_forward_time": 2.360596179962158,
      "step": 67341
    },
    {
      "epoch": 6.422138214111328e-05,
      "step": 67341,
      "training_step_time": 8.99999475479126
    },
    {
      "epoch": 6.422233581542969e-05,
      "model_forward_time": 1.7552447319030762,
      "step": 67342
    },
    {
      "epoch": 6.422233581542969e-05,
      "step": 67342,
      "training_step_time": 7.957969903945923
    },
    {
      "epoch": 6.42232894897461e-05,
      "model_forward_time": 1.7442326545715332,
      "step": 67343
    },
    {
      "epoch": 6.42232894897461e-05,
      "step": 67343,
      "training_step_time": 7.922850847244263
    },
    {
      "epoch": 6.42242431640625e-05,
      "model_forward_time": 1.736536979675293,
      "step": 67344
    },
    {
      "epoch": 6.42242431640625e-05,
      "step": 67344,
      "training_step_time": 7.7818403244018555
    },
    {
      "epoch": 6.42251968383789e-05,
      "model_forward_time": 1.7505228519439697,
      "step": 67345
    },
    {
      "epoch": 6.42251968383789e-05,
      "step": 67345,
      "training_step_time": 6.971587896347046
    },
    {
      "epoch": 6.422615051269532e-05,
      "model_forward_time": 1.3858377933502197,
      "step": 67346
    },
    {
      "epoch": 6.422615051269532e-05,
      "step": 67346,
      "training_step_time": 7.7754716873168945
    },
    {
      "epoch": 6.422710418701171e-05,
      "model_forward_time": 1.7428500652313232,
      "step": 67347
    },
    {
      "epoch": 6.422710418701171e-05,
      "step": 67347,
      "training_step_time": 7.79973840713501
    },
    {
      "epoch": 6.422805786132812e-05,
      "model_forward_time": 1.7434215545654297,
      "step": 67348
    },
    {
      "epoch": 6.422805786132812e-05,
      "step": 67348,
      "training_step_time": 7.864975214004517
    },
    {
      "epoch": 6.422901153564453e-05,
      "model_forward_time": 1.7551631927490234,
      "step": 67349
    },
    {
      "epoch": 6.422901153564453e-05,
      "step": 67349,
      "training_step_time": 9.647835969924927
    },
    {
      "epoch": 6.422996520996093e-05,
      "grad_norm": 0.08148102462291718,
      "learning_rate": 2.642250780544995e-06,
      "loss": 0.0536,
      "step": 67350
    },
    {
      "epoch": 6.422996520996093e-05,
      "model_forward_time": 2.3365068435668945,
      "step": 67350
    },
    {
      "dynamics_loss_avg": 0.05051360847428441,
      "epoch": 6.422996520996093e-05,
      "step": 67350
    },
    {
      "action_loss_avg": 0.001207888376666233,
      "epoch": 6.422996520996093e-05,
      "step": 67350
    },
    {
      "epoch": 6.422996520996093e-05,
      "step": 67350,
      "training_step_time": 7.991574287414551
    },
    {
      "epoch": 6.423091888427734e-05,
      "model_forward_time": 1.7430837154388428,
      "step": 67351
    },
    {
      "epoch": 6.423091888427734e-05,
      "step": 67351,
      "training_step_time": 9.623910427093506
    },
    {
      "epoch": 6.423187255859375e-05,
      "model_forward_time": 1.74519944190979,
      "step": 67352
    },
    {
      "epoch": 6.423187255859375e-05,
      "step": 67352,
      "training_step_time": 8.922170162200928
    },
    {
      "epoch": 6.423282623291015e-05,
      "model_forward_time": 1.745288372039795,
      "step": 67353
    },
    {
      "epoch": 6.423282623291015e-05,
      "step": 67353,
      "training_step_time": 7.817697286605835
    },
    {
      "epoch": 6.423377990722656e-05,
      "model_forward_time": 1.7349936962127686,
      "step": 67354
    },
    {
      "epoch": 6.423377990722656e-05,
      "step": 67354,
      "training_step_time": 6.95624041557312
    },
    {
      "epoch": 6.423473358154297e-05,
      "model_forward_time": 1.7339212894439697,
      "step": 67355
    },
    {
      "epoch": 6.423473358154297e-05,
      "step": 67355,
      "training_step_time": 7.790345668792725
    },
    {
      "epoch": 6.423568725585937e-05,
      "model_forward_time": 1.7478718757629395,
      "step": 67356
    },
    {
      "epoch": 6.423568725585937e-05,
      "step": 67356,
      "training_step_time": 7.799417018890381
    },
    {
      "epoch": 6.423664093017578e-05,
      "model_forward_time": 1.7337872982025146,
      "step": 67357
    },
    {
      "epoch": 6.423664093017578e-05,
      "step": 67357,
      "training_step_time": 7.761262655258179
    },
    {
      "epoch": 6.423759460449219e-05,
      "model_forward_time": 1.7373790740966797,
      "step": 67358
    },
    {
      "epoch": 6.423759460449219e-05,
      "step": 67358,
      "training_step_time": 8.376829624176025
    },
    {
      "epoch": 6.423854827880859e-05,
      "model_forward_time": 2.3369288444519043,
      "step": 67359
    },
    {
      "epoch": 6.423854827880859e-05,
      "step": 67359,
      "training_step_time": 9.001928329467773
    },
    {
      "epoch": 6.4239501953125e-05,
      "grad_norm": 0.1008775383234024,
      "learning_rate": 2.640792814643336e-06,
      "loss": 0.0575,
      "step": 67360
    },
    {
      "epoch": 6.4239501953125e-05,
      "model_forward_time": 1.7377784252166748,
      "step": 67360
    },
    {
      "dynamics_loss_avg": 0.045798959024250505,
      "epoch": 6.4239501953125e-05,
      "step": 67360
    },
    {
      "action_loss_avg": 0.002181580726755783,
      "epoch": 6.4239501953125e-05,
      "step": 67360
    },
    {
      "epoch": 6.4239501953125e-05,
      "step": 67360,
      "training_step_time": 9.383784294128418
    },
    {
      "epoch": 6.424045562744141e-05,
      "model_forward_time": 1.7360782623291016,
      "step": 67361
    },
    {
      "epoch": 6.424045562744141e-05,
      "step": 67361,
      "training_step_time": 7.877653121948242
    },
    {
      "epoch": 6.424140930175781e-05,
      "model_forward_time": 1.7386341094970703,
      "step": 67362
    },
    {
      "epoch": 6.424140930175781e-05,
      "step": 67362,
      "training_step_time": 7.771122694015503
    },
    {
      "epoch": 6.424236297607422e-05,
      "model_forward_time": 1.764232873916626,
      "step": 67363
    },
    {
      "epoch": 6.424236297607422e-05,
      "step": 67363,
      "training_step_time": 7.547899961471558
    },
    {
      "epoch": 6.424331665039063e-05,
      "model_forward_time": 1.7410478591918945,
      "step": 67364
    },
    {
      "epoch": 6.424331665039063e-05,
      "step": 67364,
      "training_step_time": 7.752913951873779
    },
    {
      "epoch": 6.424427032470703e-05,
      "model_forward_time": 1.7470743656158447,
      "step": 67365
    },
    {
      "epoch": 6.424427032470703e-05,
      "step": 67365,
      "training_step_time": 7.768351793289185
    },
    {
      "epoch": 6.424522399902344e-05,
      "model_forward_time": 1.747446060180664,
      "step": 67366
    },
    {
      "epoch": 6.424522399902344e-05,
      "step": 67366,
      "training_step_time": 7.76315712928772
    },
    {
      "epoch": 6.424617767333985e-05,
      "model_forward_time": 1.7443690299987793,
      "step": 67367
    },
    {
      "epoch": 6.424617767333985e-05,
      "step": 67367,
      "training_step_time": 7.8315980434417725
    },
    {
      "epoch": 6.424713134765625e-05,
      "model_forward_time": 1.7355647087097168,
      "step": 67368
    },
    {
      "epoch": 6.424713134765625e-05,
      "step": 67368,
      "training_step_time": 10.79387617111206
    },
    {
      "epoch": 6.424808502197266e-05,
      "model_forward_time": 2.3827502727508545,
      "step": 67369
    },
    {
      "epoch": 6.424808502197266e-05,
      "step": 67369,
      "training_step_time": 9.233654260635376
    },
    {
      "epoch": 6.424903869628907e-05,
      "grad_norm": 0.10497024655342102,
      "learning_rate": 2.6393351067410263e-06,
      "loss": 0.055,
      "step": 67370
    },
    {
      "epoch": 6.424903869628907e-05,
      "model_forward_time": 1.7400519847869873,
      "step": 67370
    },
    {
      "dynamics_loss_avg": 0.05721698552370071,
      "epoch": 6.424903869628907e-05,
      "step": 67370
    },
    {
      "action_loss_avg": 0.0012825368903577328,
      "epoch": 6.424903869628907e-05,
      "step": 67370
    },
    {
      "epoch": 6.424903869628907e-05,
      "step": 67370,
      "training_step_time": 7.720067501068115
    },
    {
      "epoch": 6.424999237060546e-05,
      "model_forward_time": 1.7381348609924316,
      "step": 67371
    },
    {
      "epoch": 6.424999237060546e-05,
      "step": 67371,
      "training_step_time": 7.713045120239258
    },
    {
      "epoch": 6.425094604492187e-05,
      "model_forward_time": 1.7409753799438477,
      "step": 67372
    },
    {
      "epoch": 6.425094604492187e-05,
      "step": 67372,
      "training_step_time": 7.288757085800171
    },
    {
      "epoch": 6.425189971923829e-05,
      "model_forward_time": 1.7414519786834717,
      "step": 67373
    },
    {
      "epoch": 6.425189971923829e-05,
      "step": 67373,
      "training_step_time": 7.453315019607544
    },
    {
      "epoch": 6.425285339355468e-05,
      "model_forward_time": 1.7454068660736084,
      "step": 67374
    },
    {
      "epoch": 6.425285339355468e-05,
      "step": 67374,
      "training_step_time": 7.452908039093018
    },
    {
      "epoch": 6.42538070678711e-05,
      "model_forward_time": 1.7425117492675781,
      "step": 67375
    },
    {
      "epoch": 6.42538070678711e-05,
      "step": 67375,
      "training_step_time": 7.4499499797821045
    },
    {
      "epoch": 6.42547607421875e-05,
      "model_forward_time": 1.7392997741699219,
      "step": 67376
    },
    {
      "epoch": 6.42547607421875e-05,
      "step": 67376,
      "training_step_time": 7.5939741134643555
    },
    {
      "epoch": 6.42557144165039e-05,
      "model_forward_time": 1.7435240745544434,
      "step": 67377
    },
    {
      "epoch": 6.42557144165039e-05,
      "step": 67377,
      "training_step_time": 7.7174084186553955
    },
    {
      "epoch": 6.425666809082031e-05,
      "model_forward_time": 2.3502748012542725,
      "step": 67378
    },
    {
      "epoch": 6.425666809082031e-05,
      "step": 67378,
      "training_step_time": 8.810566902160645
    },
    {
      "epoch": 6.425762176513672e-05,
      "model_forward_time": 1.7346646785736084,
      "step": 67379
    },
    {
      "epoch": 6.425762176513672e-05,
      "step": 67379,
      "training_step_time": 8.433703899383545
    },
    {
      "epoch": 6.425857543945312e-05,
      "grad_norm": 0.12736402451992035,
      "learning_rate": 2.6378776569974796e-06,
      "loss": 0.0574,
      "step": 67380
    },
    {
      "epoch": 6.425857543945312e-05,
      "model_forward_time": 1.7332427501678467,
      "step": 67380
    },
    {
      "dynamics_loss_avg": 0.044750062935054305,
      "epoch": 6.425857543945312e-05,
      "step": 67380
    },
    {
      "action_loss_avg": 0.0015827465860638768,
      "epoch": 6.425857543945312e-05,
      "step": 67380
    },
    {
      "epoch": 6.425857543945312e-05,
      "step": 67380,
      "training_step_time": 7.662799596786499
    },
    {
      "epoch": 6.425952911376953e-05,
      "model_forward_time": 1.74727201461792,
      "step": 67381
    },
    {
      "epoch": 6.425952911376953e-05,
      "step": 67381,
      "training_step_time": 7.9455344676971436
    },
    {
      "epoch": 6.426048278808594e-05,
      "model_forward_time": 1.735213279724121,
      "step": 67382
    },
    {
      "epoch": 6.426048278808594e-05,
      "step": 67382,
      "training_step_time": 7.528519630432129
    },
    {
      "epoch": 6.426143646240234e-05,
      "model_forward_time": 1.7372064590454102,
      "step": 67383
    },
    {
      "epoch": 6.426143646240234e-05,
      "step": 67383,
      "training_step_time": 6.716165781021118
    },
    {
      "epoch": 6.426239013671875e-05,
      "model_forward_time": 1.728212833404541,
      "step": 67384
    },
    {
      "epoch": 6.426239013671875e-05,
      "step": 67384,
      "training_step_time": 7.537938356399536
    },
    {
      "epoch": 6.426334381103516e-05,
      "model_forward_time": 1.7325377464294434,
      "step": 67385
    },
    {
      "epoch": 6.426334381103516e-05,
      "step": 67385,
      "training_step_time": 7.6474692821502686
    },
    {
      "epoch": 6.426429748535156e-05,
      "model_forward_time": 1.7400052547454834,
      "step": 67386
    },
    {
      "epoch": 6.426429748535156e-05,
      "step": 67386,
      "training_step_time": 8.445509433746338
    },
    {
      "epoch": 6.426525115966797e-05,
      "model_forward_time": 2.3142495155334473,
      "step": 67387
    },
    {
      "epoch": 6.426525115966797e-05,
      "step": 67387,
      "training_step_time": 9.228916645050049
    },
    {
      "epoch": 6.426620483398438e-05,
      "model_forward_time": 1.739717721939087,
      "step": 67388
    },
    {
      "epoch": 6.426620483398438e-05,
      "step": 67388,
      "training_step_time": 10.455716609954834
    },
    {
      "epoch": 6.426715850830078e-05,
      "model_forward_time": 1.7389271259307861,
      "step": 67389
    },
    {
      "epoch": 6.426715850830078e-05,
      "step": 67389,
      "training_step_time": 7.651776552200317
    },
    {
      "epoch": 6.426811218261719e-05,
      "grad_norm": 0.09201398491859436,
      "learning_rate": 2.6364204655720817e-06,
      "loss": 0.0567,
      "step": 67390
    },
    {
      "epoch": 6.426811218261719e-05,
      "model_forward_time": 1.7891340255737305,
      "step": 67390
    },
    {
      "dynamics_loss_avg": 0.03757126713171601,
      "epoch": 6.426811218261719e-05,
      "step": 67390
    },
    {
      "action_loss_avg": 0.0009591303503839299,
      "epoch": 6.426811218261719e-05,
      "step": 67390
    },
    {
      "epoch": 6.426811218261719e-05,
      "step": 67390,
      "training_step_time": 7.587903022766113
    },
    {
      "epoch": 6.42690658569336e-05,
      "model_forward_time": 1.7613227367401123,
      "step": 67391
    },
    {
      "epoch": 6.42690658569336e-05,
      "step": 67391,
      "training_step_time": 7.55232310295105
    },
    {
      "epoch": 6.427001953125e-05,
      "model_forward_time": 1.7545874118804932,
      "step": 67392
    },
    {
      "epoch": 6.427001953125e-05,
      "step": 67392,
      "training_step_time": 6.739417552947998
    },
    {
      "epoch": 6.427097320556641e-05,
      "model_forward_time": 1.7300944328308105,
      "step": 67393
    },
    {
      "epoch": 6.427097320556641e-05,
      "step": 67393,
      "training_step_time": 7.556820392608643
    },
    {
      "epoch": 6.427192687988282e-05,
      "model_forward_time": 1.7480521202087402,
      "step": 67394
    },
    {
      "epoch": 6.427192687988282e-05,
      "step": 67394,
      "training_step_time": 7.593303680419922
    },
    {
      "epoch": 6.427288055419921e-05,
      "model_forward_time": 1.742053747177124,
      "step": 67395
    },
    {
      "epoch": 6.427288055419921e-05,
      "step": 67395,
      "training_step_time": 8.315584897994995
    },
    {
      "epoch": 6.427383422851563e-05,
      "model_forward_time": 2.3078324794769287,
      "step": 67396
    },
    {
      "epoch": 6.427383422851563e-05,
      "step": 67396,
      "training_step_time": 8.751356601715088
    },
    {
      "epoch": 6.427478790283204e-05,
      "model_forward_time": 1.7805233001708984,
      "step": 67397
    },
    {
      "epoch": 6.427478790283204e-05,
      "step": 67397,
      "training_step_time": 10.458546161651611
    },
    {
      "epoch": 6.427574157714843e-05,
      "model_forward_time": 1.7483758926391602,
      "step": 67398
    },
    {
      "epoch": 6.427574157714843e-05,
      "step": 67398,
      "training_step_time": 9.664292573928833
    },
    {
      "epoch": 6.427669525146484e-05,
      "model_forward_time": 1.7433836460113525,
      "step": 67399
    },
    {
      "epoch": 6.427669525146484e-05,
      "step": 67399,
      "training_step_time": 7.580293893814087
    },
    {
      "epoch": 6.427764892578126e-05,
      "grad_norm": 0.09943339228630066,
      "learning_rate": 2.6349635326241896e-06,
      "loss": 0.0564,
      "step": 67400
    },
    {
      "epoch": 6.427764892578126e-05,
      "model_forward_time": 1.7183036804199219,
      "step": 67400
    },
    {
      "dynamics_loss_avg": 0.06156219523400068,
      "epoch": 6.427764892578126e-05,
      "step": 67400
    },
    {
      "action_loss_avg": 0.002235480770468712,
      "epoch": 6.427764892578126e-05,
      "step": 67400
    },
    {
      "epoch": 6.427764892578126e-05,
      "step": 67400,
      "training_step_time": 7.497642278671265
    },
    {
      "epoch": 6.427860260009765e-05,
      "model_forward_time": 1.7275986671447754,
      "step": 67401
    },
    {
      "epoch": 6.427860260009765e-05,
      "step": 67401,
      "training_step_time": 6.757145404815674
    },
    {
      "epoch": 6.427955627441406e-05,
      "model_forward_time": 1.7247953414916992,
      "step": 67402
    },
    {
      "epoch": 6.427955627441406e-05,
      "step": 67402,
      "training_step_time": 7.54480504989624
    },
    {
      "epoch": 6.428050994873047e-05,
      "model_forward_time": 1.7265198230743408,
      "step": 67403
    },
    {
      "epoch": 6.428050994873047e-05,
      "step": 67403,
      "training_step_time": 7.551473379135132
    },
    {
      "epoch": 6.428146362304687e-05,
      "model_forward_time": 1.735405445098877,
      "step": 67404
    },
    {
      "epoch": 6.428146362304687e-05,
      "step": 67404,
      "training_step_time": 7.531623363494873
    },
    {
      "epoch": 6.428241729736328e-05,
      "model_forward_time": 2.2767772674560547,
      "step": 67405
    },
    {
      "epoch": 6.428241729736328e-05,
      "step": 67405,
      "training_step_time": 7.67341947555542
    },
    {
      "epoch": 6.428337097167969e-05,
      "model_forward_time": 1.757551670074463,
      "step": 67406
    },
    {
      "epoch": 6.428337097167969e-05,
      "step": 67406,
      "training_step_time": 9.241561889648438
    },
    {
      "epoch": 6.428432464599609e-05,
      "model_forward_time": 1.7276360988616943,
      "step": 67407
    },
    {
      "epoch": 6.428432464599609e-05,
      "step": 67407,
      "training_step_time": 7.713725566864014
    },
    {
      "epoch": 6.42852783203125e-05,
      "model_forward_time": 1.734750747680664,
      "step": 67408
    },
    {
      "epoch": 6.42852783203125e-05,
      "step": 67408,
      "training_step_time": 7.61183762550354
    },
    {
      "epoch": 6.428623199462891e-05,
      "model_forward_time": 1.7430613040924072,
      "step": 67409
    },
    {
      "epoch": 6.428623199462891e-05,
      "step": 67409,
      "training_step_time": 7.875211954116821
    },
    {
      "epoch": 6.428718566894531e-05,
      "grad_norm": 0.08041230589151382,
      "learning_rate": 2.6335068583131273e-06,
      "loss": 0.0572,
      "step": 67410
    },
    {
      "epoch": 6.428718566894531e-05,
      "model_forward_time": 1.7289278507232666,
      "step": 67410
    },
    {
      "dynamics_loss_avg": 0.060499631240963934,
      "epoch": 6.428718566894531e-05,
      "step": 67410
    },
    {
      "action_loss_avg": 0.0032490017300006004,
      "epoch": 6.428718566894531e-05,
      "step": 67410
    },
    {
      "epoch": 6.428718566894531e-05,
      "step": 67410,
      "training_step_time": 6.701273202896118
    },
    {
      "epoch": 6.428813934326172e-05,
      "model_forward_time": 1.725534200668335,
      "step": 67411
    },
    {
      "epoch": 6.428813934326172e-05,
      "step": 67411,
      "training_step_time": 7.493427276611328
    },
    {
      "epoch": 6.428909301757813e-05,
      "model_forward_time": 1.734471321105957,
      "step": 67412
    },
    {
      "epoch": 6.428909301757813e-05,
      "step": 67412,
      "training_step_time": 8.244436979293823
    },
    {
      "epoch": 6.429004669189453e-05,
      "model_forward_time": 1.7318780422210693,
      "step": 67413
    },
    {
      "epoch": 6.429004669189453e-05,
      "step": 67413,
      "training_step_time": 7.78608512878418
    },
    {
      "epoch": 6.429100036621094e-05,
      "model_forward_time": 1.7294926643371582,
      "step": 67414
    },
    {
      "epoch": 6.429100036621094e-05,
      "step": 67414,
      "training_step_time": 8.997343301773071
    },
    {
      "epoch": 6.429195404052735e-05,
      "model_forward_time": 1.7354402542114258,
      "step": 67415
    },
    {
      "epoch": 6.429195404052735e-05,
      "step": 67415,
      "training_step_time": 9.477418422698975
    },
    {
      "epoch": 6.429290771484375e-05,
      "model_forward_time": 1.4444489479064941,
      "step": 67416
    },
    {
      "epoch": 6.429290771484375e-05,
      "step": 67416,
      "training_step_time": 7.936649322509766
    },
    {
      "epoch": 6.429386138916016e-05,
      "model_forward_time": 1.739264965057373,
      "step": 67417
    },
    {
      "epoch": 6.429386138916016e-05,
      "step": 67417,
      "training_step_time": 7.920954942703247
    },
    {
      "epoch": 6.429481506347657e-05,
      "model_forward_time": 1.741302490234375,
      "step": 67418
    },
    {
      "epoch": 6.429481506347657e-05,
      "step": 67418,
      "training_step_time": 7.4897706508636475
    },
    {
      "epoch": 6.429576873779297e-05,
      "model_forward_time": 1.7277898788452148,
      "step": 67419
    },
    {
      "epoch": 6.429576873779297e-05,
      "step": 67419,
      "training_step_time": 7.71670937538147
    },
    {
      "epoch": 6.429672241210938e-05,
      "grad_norm": 0.0928114727139473,
      "learning_rate": 2.6320504427981975e-06,
      "loss": 0.0548,
      "step": 67420
    },
    {
      "epoch": 6.429672241210938e-05,
      "model_forward_time": 1.7225067615509033,
      "step": 67420
    },
    {
      "dynamics_loss_avg": 0.05747878290712834,
      "epoch": 6.429672241210938e-05,
      "step": 67420
    },
    {
      "action_loss_avg": 0.0012965691188583151,
      "epoch": 6.429672241210938e-05,
      "step": 67420
    },
    {
      "epoch": 6.429672241210938e-05,
      "step": 67420,
      "training_step_time": 7.602293014526367
    },
    {
      "epoch": 6.429767608642579e-05,
      "model_forward_time": 1.7395687103271484,
      "step": 67421
    },
    {
      "epoch": 6.429767608642579e-05,
      "step": 67421,
      "training_step_time": 8.60489296913147
    },
    {
      "epoch": 6.429862976074218e-05,
      "model_forward_time": 1.7342808246612549,
      "step": 67422
    },
    {
      "epoch": 6.429862976074218e-05,
      "step": 67422,
      "training_step_time": 7.830999374389648
    },
    {
      "epoch": 6.42995834350586e-05,
      "model_forward_time": 1.7494337558746338,
      "step": 67423
    },
    {
      "epoch": 6.42995834350586e-05,
      "step": 67423,
      "training_step_time": 8.495611906051636
    },
    {
      "epoch": 6.4300537109375e-05,
      "model_forward_time": 2.283855438232422,
      "step": 67424
    },
    {
      "epoch": 6.4300537109375e-05,
      "step": 67424,
      "training_step_time": 9.016700267791748
    },
    {
      "epoch": 6.43014907836914e-05,
      "model_forward_time": 1.7125108242034912,
      "step": 67425
    },
    {
      "epoch": 6.43014907836914e-05,
      "step": 67425,
      "training_step_time": 10.65192723274231
    },
    {
      "epoch": 6.430244445800781e-05,
      "model_forward_time": 1.731501579284668,
      "step": 67426
    },
    {
      "epoch": 6.430244445800781e-05,
      "step": 67426,
      "training_step_time": 7.587429761886597
    },
    {
      "epoch": 6.430339813232423e-05,
      "model_forward_time": 1.7310082912445068,
      "step": 67427
    },
    {
      "epoch": 6.430339813232423e-05,
      "step": 67427,
      "training_step_time": 7.741203546524048
    },
    {
      "epoch": 6.430435180664062e-05,
      "model_forward_time": 1.7930693626403809,
      "step": 67428
    },
    {
      "epoch": 6.430435180664062e-05,
      "step": 67428,
      "training_step_time": 7.700597047805786
    },
    {
      "epoch": 6.430530548095703e-05,
      "model_forward_time": 1.7396631240844727,
      "step": 67429
    },
    {
      "epoch": 6.430530548095703e-05,
      "step": 67429,
      "training_step_time": 6.8526997566223145
    },
    {
      "epoch": 6.430625915527344e-05,
      "grad_norm": 0.12837912142276764,
      "learning_rate": 2.6305942862386736e-06,
      "loss": 0.0589,
      "step": 67430
    },
    {
      "epoch": 6.430625915527344e-05,
      "model_forward_time": 1.7162697315216064,
      "step": 67430
    },
    {
      "dynamics_loss_avg": 0.06112838294357061,
      "epoch": 6.430625915527344e-05,
      "step": 67430
    },
    {
      "action_loss_avg": 0.003022541651444044,
      "epoch": 6.430625915527344e-05,
      "step": 67430
    },
    {
      "epoch": 6.430625915527344e-05,
      "step": 67430,
      "training_step_time": 7.666951894760132
    },
    {
      "epoch": 6.430721282958984e-05,
      "model_forward_time": 1.7297868728637695,
      "step": 67431
    },
    {
      "epoch": 6.430721282958984e-05,
      "step": 67431,
      "training_step_time": 7.896998167037964
    },
    {
      "epoch": 6.430816650390625e-05,
      "model_forward_time": 1.736417531967163,
      "step": 67432
    },
    {
      "epoch": 6.430816650390625e-05,
      "step": 67432,
      "training_step_time": 9.232723474502563
    },
    {
      "epoch": 6.430912017822266e-05,
      "model_forward_time": 2.2914135456085205,
      "step": 67433
    },
    {
      "epoch": 6.430912017822266e-05,
      "step": 67433,
      "training_step_time": 9.396923065185547
    },
    {
      "epoch": 6.431007385253906e-05,
      "model_forward_time": 1.7771129608154297,
      "step": 67434
    },
    {
      "epoch": 6.431007385253906e-05,
      "step": 67434,
      "training_step_time": 9.789517641067505
    },
    {
      "epoch": 6.431102752685547e-05,
      "model_forward_time": 1.7394490242004395,
      "step": 67435
    },
    {
      "epoch": 6.431102752685547e-05,
      "step": 67435,
      "training_step_time": 8.495996713638306
    },
    {
      "epoch": 6.431198120117188e-05,
      "model_forward_time": 1.3685979843139648,
      "step": 67436
    },
    {
      "epoch": 6.431198120117188e-05,
      "step": 67436,
      "training_step_time": 8.584628343582153
    },
    {
      "epoch": 6.431293487548828e-05,
      "model_forward_time": 1.740480899810791,
      "step": 67437
    },
    {
      "epoch": 6.431293487548828e-05,
      "step": 67437,
      "training_step_time": 7.761717796325684
    },
    {
      "epoch": 6.431388854980469e-05,
      "model_forward_time": 1.732800006866455,
      "step": 67438
    },
    {
      "epoch": 6.431388854980469e-05,
      "step": 67438,
      "training_step_time": 6.8693907260894775
    },
    {
      "epoch": 6.431484222412109e-05,
      "model_forward_time": 1.7394602298736572,
      "step": 67439
    },
    {
      "epoch": 6.431484222412109e-05,
      "step": 67439,
      "training_step_time": 7.692524433135986
    },
    {
      "epoch": 6.43157958984375e-05,
      "grad_norm": 0.095387302339077,
      "learning_rate": 2.6291383887937934e-06,
      "loss": 0.0558,
      "step": 67440
    },
    {
      "epoch": 6.43157958984375e-05,
      "model_forward_time": 1.7291886806488037,
      "step": 67440
    },
    {
      "dynamics_loss_avg": 0.03350331746041775,
      "epoch": 6.43157958984375e-05,
      "step": 67440
    },
    {
      "action_loss_avg": 0.0013373268157010898,
      "epoch": 6.43157958984375e-05,
      "step": 67440
    },
    {
      "epoch": 6.43157958984375e-05,
      "step": 67440,
      "training_step_time": 7.860260725021362
    },
    {
      "epoch": 6.431674957275391e-05,
      "model_forward_time": 1.7360308170318604,
      "step": 67441
    },
    {
      "epoch": 6.431674957275391e-05,
      "step": 67441,
      "training_step_time": 7.811870336532593
    },
    {
      "epoch": 6.43177032470703e-05,
      "model_forward_time": 2.285506010055542,
      "step": 67442
    },
    {
      "epoch": 6.43177032470703e-05,
      "step": 67442,
      "training_step_time": 10.294227361679077
    },
    {
      "epoch": 6.431865692138672e-05,
      "model_forward_time": 1.7523300647735596,
      "step": 67443
    },
    {
      "epoch": 6.431865692138672e-05,
      "step": 67443,
      "training_step_time": 9.038338899612427
    },
    {
      "epoch": 6.431961059570313e-05,
      "model_forward_time": 1.3781235218048096,
      "step": 67444
    },
    {
      "epoch": 6.431961059570313e-05,
      "step": 67444,
      "training_step_time": 8.879753828048706
    },
    {
      "epoch": 6.432056427001952e-05,
      "model_forward_time": 1.7467539310455322,
      "step": 67445
    },
    {
      "epoch": 6.432056427001952e-05,
      "step": 67445,
      "training_step_time": 8.701112747192383
    },
    {
      "epoch": 6.432151794433594e-05,
      "model_forward_time": 1.7296984195709229,
      "step": 67446
    },
    {
      "epoch": 6.432151794433594e-05,
      "step": 67446,
      "training_step_time": 7.8869757652282715
    },
    {
      "epoch": 6.432247161865235e-05,
      "model_forward_time": 1.755561113357544,
      "step": 67447
    },
    {
      "epoch": 6.432247161865235e-05,
      "step": 67447,
      "training_step_time": 6.929509162902832
    },
    {
      "epoch": 6.432342529296874e-05,
      "model_forward_time": 1.7216248512268066,
      "step": 67448
    },
    {
      "epoch": 6.432342529296874e-05,
      "step": 67448,
      "training_step_time": 7.735728979110718
    },
    {
      "epoch": 6.432437896728515e-05,
      "model_forward_time": 1.7384059429168701,
      "step": 67449
    },
    {
      "epoch": 6.432437896728515e-05,
      "step": 67449,
      "training_step_time": 7.646424055099487
    },
    {
      "epoch": 6.432533264160157e-05,
      "grad_norm": 0.1112385094165802,
      "learning_rate": 2.6276827506227744e-06,
      "loss": 0.057,
      "step": 67450
    },
    {
      "epoch": 6.432533264160157e-05,
      "model_forward_time": 1.7171614170074463,
      "step": 67450
    },
    {
      "dynamics_loss_avg": 0.04435953916981816,
      "epoch": 6.432533264160157e-05,
      "step": 67450
    },
    {
      "action_loss_avg": 0.001752494860556908,
      "epoch": 6.432533264160157e-05,
      "step": 67450
    },
    {
      "epoch": 6.432533264160157e-05,
      "step": 67450,
      "training_step_time": 7.580386400222778
    },
    {
      "epoch": 6.432628631591796e-05,
      "model_forward_time": 1.72849440574646,
      "step": 67451
    },
    {
      "epoch": 6.432628631591796e-05,
      "step": 67451,
      "training_step_time": 8.286389589309692
    },
    {
      "epoch": 6.432723999023437e-05,
      "model_forward_time": 2.2767114639282227,
      "step": 67452
    },
    {
      "epoch": 6.432723999023437e-05,
      "step": 67452,
      "training_step_time": 8.444713592529297
    },
    {
      "epoch": 6.432819366455078e-05,
      "model_forward_time": 1.8072590827941895,
      "step": 67453
    },
    {
      "epoch": 6.432819366455078e-05,
      "step": 67453,
      "training_step_time": 8.572669506072998
    },
    {
      "epoch": 6.432914733886718e-05,
      "model_forward_time": 1.731783151626587,
      "step": 67454
    },
    {
      "epoch": 6.432914733886718e-05,
      "step": 67454,
      "training_step_time": 7.719892978668213
    },
    {
      "epoch": 6.433010101318359e-05,
      "model_forward_time": 1.7217869758605957,
      "step": 67455
    },
    {
      "epoch": 6.433010101318359e-05,
      "step": 67455,
      "training_step_time": 7.611446380615234
    },
    {
      "epoch": 6.43310546875e-05,
      "model_forward_time": 1.7224812507629395,
      "step": 67456
    },
    {
      "epoch": 6.43310546875e-05,
      "step": 67456,
      "training_step_time": 7.347938537597656
    },
    {
      "epoch": 6.43320083618164e-05,
      "model_forward_time": 1.7447621822357178,
      "step": 67457
    },
    {
      "epoch": 6.43320083618164e-05,
      "step": 67457,
      "training_step_time": 7.694581031799316
    },
    {
      "epoch": 6.433296203613281e-05,
      "model_forward_time": 1.7479465007781982,
      "step": 67458
    },
    {
      "epoch": 6.433296203613281e-05,
      "step": 67458,
      "training_step_time": 7.584012508392334
    },
    {
      "epoch": 6.433391571044922e-05,
      "model_forward_time": 1.735137701034546,
      "step": 67459
    },
    {
      "epoch": 6.433391571044922e-05,
      "step": 67459,
      "training_step_time": 7.578194618225098
    },
    {
      "epoch": 6.433486938476562e-05,
      "grad_norm": 0.0970076248049736,
      "learning_rate": 2.6262273718848037e-06,
      "loss": 0.0547,
      "step": 67460
    },
    {
      "epoch": 6.433486938476562e-05,
      "model_forward_time": 1.7254629135131836,
      "step": 67460
    },
    {
      "dynamics_loss_avg": 0.06364820748567582,
      "epoch": 6.433486938476562e-05,
      "step": 67460
    },
    {
      "action_loss_avg": 0.0026963191194226965,
      "epoch": 6.433486938476562e-05,
      "step": 67460
    },
    {
      "epoch": 6.433486938476562e-05,
      "step": 67460,
      "training_step_time": 9.409342527389526
    },
    {
      "epoch": 6.433582305908203e-05,
      "model_forward_time": 2.270761013031006,
      "step": 67461
    },
    {
      "epoch": 6.433582305908203e-05,
      "step": 67461,
      "training_step_time": 7.80887508392334
    },
    {
      "epoch": 6.433677673339844e-05,
      "model_forward_time": 1.8082890510559082,
      "step": 67462
    },
    {
      "epoch": 6.433677673339844e-05,
      "step": 67462,
      "training_step_time": 8.698858499526978
    },
    {
      "epoch": 6.433773040771484e-05,
      "model_forward_time": 1.7394604682922363,
      "step": 67463
    },
    {
      "epoch": 6.433773040771484e-05,
      "step": 67463,
      "training_step_time": 7.74519944190979
    },
    {
      "epoch": 6.433868408203125e-05,
      "model_forward_time": 1.732848882675171,
      "step": 67464
    },
    {
      "epoch": 6.433868408203125e-05,
      "step": 67464,
      "training_step_time": 8.188120126724243
    },
    {
      "epoch": 6.433963775634766e-05,
      "model_forward_time": 1.7191166877746582,
      "step": 67465
    },
    {
      "epoch": 6.433963775634766e-05,
      "step": 67465,
      "training_step_time": 6.956777334213257
    },
    {
      "epoch": 6.434059143066406e-05,
      "model_forward_time": 1.7381515502929688,
      "step": 67466
    },
    {
      "epoch": 6.434059143066406e-05,
      "step": 67466,
      "training_step_time": 7.651634454727173
    },
    {
      "epoch": 6.434154510498047e-05,
      "model_forward_time": 1.7333965301513672,
      "step": 67467
    },
    {
      "epoch": 6.434154510498047e-05,
      "step": 67467,
      "training_step_time": 7.609029769897461
    },
    {
      "epoch": 6.434249877929688e-05,
      "model_forward_time": 1.7437868118286133,
      "step": 67468
    },
    {
      "epoch": 6.434249877929688e-05,
      "step": 67468,
      "training_step_time": 8.701628684997559
    },
    {
      "epoch": 6.434345245361328e-05,
      "model_forward_time": 1.7346484661102295,
      "step": 67469
    },
    {
      "epoch": 6.434345245361328e-05,
      "step": 67469,
      "training_step_time": 7.911437034606934
    },
    {
      "epoch": 6.434440612792969e-05,
      "grad_norm": 0.0976695641875267,
      "learning_rate": 2.624772252739041e-06,
      "loss": 0.0541,
      "step": 67470
    },
    {
      "epoch": 6.434440612792969e-05,
      "model_forward_time": 2.289196729660034,
      "step": 67470
    },
    {
      "dynamics_loss_avg": 0.05981671158224344,
      "epoch": 6.434440612792969e-05,
      "step": 67470
    },
    {
      "action_loss_avg": 0.002248366753337905,
      "epoch": 6.434440612792969e-05,
      "step": 67470
    },
    {
      "epoch": 6.434440612792969e-05,
      "step": 67470,
      "training_step_time": 9.01969313621521
    },
    {
      "epoch": 6.43453598022461e-05,
      "model_forward_time": 1.7835068702697754,
      "step": 67471
    },
    {
      "epoch": 6.43453598022461e-05,
      "step": 67471,
      "training_step_time": 9.979767084121704
    },
    {
      "epoch": 6.43463134765625e-05,
      "model_forward_time": 1.7224342823028564,
      "step": 67472
    },
    {
      "epoch": 6.43463134765625e-05,
      "step": 67472,
      "training_step_time": 7.889150381088257
    },
    {
      "epoch": 6.43472671508789e-05,
      "model_forward_time": 1.7366440296173096,
      "step": 67473
    },
    {
      "epoch": 6.43472671508789e-05,
      "step": 67473,
      "training_step_time": 9.278184652328491
    },
    {
      "epoch": 6.434822082519532e-05,
      "model_forward_time": 1.748628854751587,
      "step": 67474
    },
    {
      "epoch": 6.434822082519532e-05,
      "step": 67474,
      "training_step_time": 7.606651782989502
    },
    {
      "epoch": 6.434917449951171e-05,
      "model_forward_time": 1.7524008750915527,
      "step": 67475
    },
    {
      "epoch": 6.434917449951171e-05,
      "step": 67475,
      "training_step_time": 8.648313522338867
    },
    {
      "epoch": 6.435012817382812e-05,
      "model_forward_time": 1.7393174171447754,
      "step": 67476
    },
    {
      "epoch": 6.435012817382812e-05,
      "step": 67476,
      "training_step_time": 7.586188554763794
    },
    {
      "epoch": 6.435108184814453e-05,
      "model_forward_time": 1.7396113872528076,
      "step": 67477
    },
    {
      "epoch": 6.435108184814453e-05,
      "step": 67477,
      "training_step_time": 7.795320987701416
    },
    {
      "epoch": 6.435203552246093e-05,
      "model_forward_time": 1.7398159503936768,
      "step": 67478
    },
    {
      "epoch": 6.435203552246093e-05,
      "step": 67478,
      "training_step_time": 7.819711685180664
    },
    {
      "epoch": 6.435298919677734e-05,
      "model_forward_time": 2.3329427242279053,
      "step": 67479
    },
    {
      "epoch": 6.435298919677734e-05,
      "step": 67479,
      "training_step_time": 9.882826089859009
    },
    {
      "epoch": 6.435394287109375e-05,
      "grad_norm": 0.13411833345890045,
      "learning_rate": 2.6233173933446105e-06,
      "loss": 0.0583,
      "step": 67480
    },
    {
      "epoch": 6.435394287109375e-05,
      "model_forward_time": 1.7781133651733398,
      "step": 67480
    },
    {
      "dynamics_loss_avg": 0.04272084962576628,
      "epoch": 6.435394287109375e-05,
      "step": 67480
    },
    {
      "action_loss_avg": 0.0012180734833236785,
      "epoch": 6.435394287109375e-05,
      "step": 67480
    },
    {
      "epoch": 6.435394287109375e-05,
      "step": 67480,
      "training_step_time": 8.380595684051514
    },
    {
      "epoch": 6.435489654541015e-05,
      "model_forward_time": 1.7478516101837158,
      "step": 67481
    },
    {
      "epoch": 6.435489654541015e-05,
      "step": 67481,
      "training_step_time": 9.158589601516724
    },
    {
      "epoch": 6.435585021972656e-05,
      "model_forward_time": 1.7448630332946777,
      "step": 67482
    },
    {
      "epoch": 6.435585021972656e-05,
      "step": 67482,
      "training_step_time": 7.824233055114746
    },
    {
      "epoch": 6.435680389404297e-05,
      "model_forward_time": 1.7425146102905273,
      "step": 67483
    },
    {
      "epoch": 6.435680389404297e-05,
      "step": 67483,
      "training_step_time": 7.673847198486328
    },
    {
      "epoch": 6.435775756835937e-05,
      "model_forward_time": 1.7378606796264648,
      "step": 67484
    },
    {
      "epoch": 6.435775756835937e-05,
      "step": 67484,
      "training_step_time": 7.848509311676025
    },
    {
      "epoch": 6.435871124267578e-05,
      "model_forward_time": 1.7335965633392334,
      "step": 67485
    },
    {
      "epoch": 6.435871124267578e-05,
      "step": 67485,
      "training_step_time": 7.529636859893799
    },
    {
      "epoch": 6.435966491699219e-05,
      "model_forward_time": 1.7426135540008545,
      "step": 67486
    },
    {
      "epoch": 6.435966491699219e-05,
      "step": 67486,
      "training_step_time": 9.285329103469849
    },
    {
      "epoch": 6.436061859130859e-05,
      "model_forward_time": 1.7500827312469482,
      "step": 67487
    },
    {
      "epoch": 6.436061859130859e-05,
      "step": 67487,
      "training_step_time": 7.884657859802246
    },
    {
      "epoch": 6.4361572265625e-05,
      "model_forward_time": 2.295142889022827,
      "step": 67488
    },
    {
      "epoch": 6.4361572265625e-05,
      "step": 67488,
      "training_step_time": 9.459474563598633
    },
    {
      "epoch": 6.436252593994141e-05,
      "model_forward_time": 1.7634367942810059,
      "step": 67489
    },
    {
      "epoch": 6.436252593994141e-05,
      "step": 67489,
      "training_step_time": 7.966881513595581
    },
    {
      "epoch": 6.436347961425781e-05,
      "grad_norm": 0.09553420543670654,
      "learning_rate": 2.621862793860619e-06,
      "loss": 0.0585,
      "step": 67490
    },
    {
      "epoch": 6.436347961425781e-05,
      "model_forward_time": 1.7427723407745361,
      "step": 67490
    },
    {
      "dynamics_loss_avg": 0.060424196906387805,
      "epoch": 6.436347961425781e-05,
      "step": 67490
    },
    {
      "action_loss_avg": 0.0020778237900231033,
      "epoch": 6.436347961425781e-05,
      "step": 67490
    },
    {
      "epoch": 6.436347961425781e-05,
      "step": 67490,
      "training_step_time": 8.928223371505737
    },
    {
      "epoch": 6.436443328857422e-05,
      "model_forward_time": 1.7393584251403809,
      "step": 67491
    },
    {
      "epoch": 6.436443328857422e-05,
      "step": 67491,
      "training_step_time": 7.743213653564453
    },
    {
      "epoch": 6.436538696289063e-05,
      "model_forward_time": 1.747666835784912,
      "step": 67492
    },
    {
      "epoch": 6.436538696289063e-05,
      "step": 67492,
      "training_step_time": 8.390111684799194
    },
    {
      "epoch": 6.436634063720703e-05,
      "model_forward_time": 1.74521803855896,
      "step": 67493
    },
    {
      "epoch": 6.436634063720703e-05,
      "step": 67493,
      "training_step_time": 7.475650787353516
    },
    {
      "epoch": 6.436729431152344e-05,
      "model_forward_time": 1.7340712547302246,
      "step": 67494
    },
    {
      "epoch": 6.436729431152344e-05,
      "step": 67494,
      "training_step_time": 7.533675909042358
    },
    {
      "epoch": 6.436824798583985e-05,
      "model_forward_time": 1.7421045303344727,
      "step": 67495
    },
    {
      "epoch": 6.436824798583985e-05,
      "step": 67495,
      "training_step_time": 7.5441460609436035
    },
    {
      "epoch": 6.436920166015625e-05,
      "model_forward_time": 1.740851640701294,
      "step": 67496
    },
    {
      "epoch": 6.436920166015625e-05,
      "step": 67496,
      "training_step_time": 8.372151613235474
    },
    {
      "epoch": 6.437015533447266e-05,
      "model_forward_time": 1.7416558265686035,
      "step": 67497
    },
    {
      "epoch": 6.437015533447266e-05,
      "step": 67497,
      "training_step_time": 7.634936571121216
    },
    {
      "epoch": 6.437110900878907e-05,
      "model_forward_time": 2.307316780090332,
      "step": 67498
    },
    {
      "epoch": 6.437110900878907e-05,
      "step": 67498,
      "training_step_time": 10.029614448547363
    },
    {
      "epoch": 6.437206268310546e-05,
      "model_forward_time": 1.8312301635742188,
      "step": 67499
    },
    {
      "epoch": 6.437206268310546e-05,
      "step": 67499,
      "training_step_time": 7.6073157787323
    },
    {
      "epoch": 6.437301635742188e-05,
      "grad_norm": 0.18457961082458496,
      "learning_rate": 2.620408454446135e-06,
      "loss": 0.0564,
      "step": 67500
    },
    {
      "epoch": 6.437301635742188e-05,
      "model_forward_time": 1.748511791229248,
      "step": 67500
    },
    {
      "dynamics_loss_avg": 0.04557913187891245,
      "epoch": 6.437301635742188e-05,
      "step": 67500
    },
    {
      "action_loss_avg": 0.0016176968929357827,
      "epoch": 6.437301635742188e-05,
      "step": 67500
    },
    {
      "epoch": 6.437301635742188e-05,
      "step": 67500,
      "training_step_time": 9.088402509689331
    },
    {
      "epoch": 6.437397003173829e-05,
      "model_forward_time": 1.7503395080566406,
      "step": 67501
    },
    {
      "epoch": 6.437397003173829e-05,
      "step": 67501,
      "training_step_time": 7.510602712631226
    },
    {
      "epoch": 6.437492370605468e-05,
      "model_forward_time": 1.753631830215454,
      "step": 67502
    },
    {
      "epoch": 6.437492370605468e-05,
      "step": 67502,
      "training_step_time": 7.614960193634033
    },
    {
      "epoch": 6.43758773803711e-05,
      "model_forward_time": 1.7416224479675293,
      "step": 67503
    },
    {
      "epoch": 6.43758773803711e-05,
      "step": 67503,
      "training_step_time": 8.535562753677368
    },
    {
      "epoch": 6.43768310546875e-05,
      "model_forward_time": 1.7535195350646973,
      "step": 67504
    },
    {
      "epoch": 6.43768310546875e-05,
      "step": 67504,
      "training_step_time": 7.840919017791748
    },
    {
      "epoch": 6.43777847290039e-05,
      "model_forward_time": 1.7595279216766357,
      "step": 67505
    },
    {
      "epoch": 6.43777847290039e-05,
      "step": 67505,
      "training_step_time": 8.778234958648682
    },
    {
      "epoch": 6.437873840332031e-05,
      "model_forward_time": 1.7527251243591309,
      "step": 67506
    },
    {
      "epoch": 6.437873840332031e-05,
      "step": 67506,
      "training_step_time": 8.496036767959595
    },
    {
      "epoch": 6.437969207763672e-05,
      "model_forward_time": 2.30500864982605,
      "step": 67507
    },
    {
      "epoch": 6.437969207763672e-05,
      "step": 67507,
      "training_step_time": 8.542977809906006
    },
    {
      "epoch": 6.438064575195312e-05,
      "model_forward_time": 1.8123729228973389,
      "step": 67508
    },
    {
      "epoch": 6.438064575195312e-05,
      "step": 67508,
      "training_step_time": 9.464120864868164
    },
    {
      "epoch": 6.438159942626953e-05,
      "model_forward_time": 1.7513635158538818,
      "step": 67509
    },
    {
      "epoch": 6.438159942626953e-05,
      "step": 67509,
      "training_step_time": 7.88141942024231
    },
    {
      "epoch": 6.438255310058594e-05,
      "grad_norm": 0.10215673595666885,
      "learning_rate": 2.6189543752602047e-06,
      "loss": 0.0582,
      "step": 67510
    },
    {
      "epoch": 6.438255310058594e-05,
      "model_forward_time": 1.7425241470336914,
      "step": 67510
    },
    {
      "dynamics_loss_avg": 0.04778550434857607,
      "epoch": 6.438255310058594e-05,
      "step": 67510
    },
    {
      "action_loss_avg": 0.001174490651465021,
      "epoch": 6.438255310058594e-05,
      "step": 67510
    },
    {
      "epoch": 6.438255310058594e-05,
      "step": 67510,
      "training_step_time": 8.616042613983154
    },
    {
      "epoch": 6.438350677490234e-05,
      "model_forward_time": 1.7422590255737305,
      "step": 67511
    },
    {
      "epoch": 6.438350677490234e-05,
      "step": 67511,
      "training_step_time": 7.8529582023620605
    },
    {
      "epoch": 6.438446044921875e-05,
      "model_forward_time": 1.7345149517059326,
      "step": 67512
    },
    {
      "epoch": 6.438446044921875e-05,
      "step": 67512,
      "training_step_time": 8.762609958648682
    },
    {
      "epoch": 6.438541412353516e-05,
      "model_forward_time": 1.7373511791229248,
      "step": 67513
    },
    {
      "epoch": 6.438541412353516e-05,
      "step": 67513,
      "training_step_time": 8.491912364959717
    },
    {
      "epoch": 6.438636779785156e-05,
      "model_forward_time": 1.7327287197113037,
      "step": 67514
    },
    {
      "epoch": 6.438636779785156e-05,
      "step": 67514,
      "training_step_time": 7.736559867858887
    },
    {
      "epoch": 6.438732147216797e-05,
      "model_forward_time": 1.736238956451416,
      "step": 67515
    },
    {
      "epoch": 6.438732147216797e-05,
      "step": 67515,
      "training_step_time": 8.160578727722168
    },
    {
      "epoch": 6.438827514648438e-05,
      "model_forward_time": 2.304276466369629,
      "step": 67516
    },
    {
      "epoch": 6.438827514648438e-05,
      "step": 67516,
      "training_step_time": 7.6127941608428955
    },
    {
      "epoch": 6.438922882080078e-05,
      "model_forward_time": 1.7736246585845947,
      "step": 67517
    },
    {
      "epoch": 6.438922882080078e-05,
      "step": 67517,
      "training_step_time": 7.83953332901001
    },
    {
      "epoch": 6.439018249511719e-05,
      "model_forward_time": 1.7352111339569092,
      "step": 67518
    },
    {
      "epoch": 6.439018249511719e-05,
      "step": 67518,
      "training_step_time": 7.654659986495972
    },
    {
      "epoch": 6.43911361694336e-05,
      "model_forward_time": 1.74314284324646,
      "step": 67519
    },
    {
      "epoch": 6.43911361694336e-05,
      "step": 67519,
      "training_step_time": 8.47382926940918
    },
    {
      "epoch": 6.439208984375e-05,
      "grad_norm": 0.1165510043501854,
      "learning_rate": 2.6175005564618445e-06,
      "loss": 0.0558,
      "step": 67520
    },
    {
      "epoch": 6.439208984375e-05,
      "model_forward_time": 1.7391884326934814,
      "step": 67520
    },
    {
      "dynamics_loss_avg": 0.04543913034722209,
      "epoch": 6.439208984375e-05,
      "step": 67520
    },
    {
      "action_loss_avg": 0.0012971970776561649,
      "epoch": 6.439208984375e-05,
      "step": 67520
    },
    {
      "epoch": 6.439208984375e-05,
      "step": 67520,
      "training_step_time": 7.6177709102630615
    },
    {
      "epoch": 6.439304351806641e-05,
      "model_forward_time": 1.7365331649780273,
      "step": 67521
    },
    {
      "epoch": 6.439304351806641e-05,
      "step": 67521,
      "training_step_time": 7.540505409240723
    },
    {
      "epoch": 6.439399719238282e-05,
      "model_forward_time": 1.7454917430877686,
      "step": 67522
    },
    {
      "epoch": 6.439399719238282e-05,
      "step": 67522,
      "training_step_time": 8.207419157028198
    },
    {
      "epoch": 6.439495086669922e-05,
      "model_forward_time": 1.7314321994781494,
      "step": 67523
    },
    {
      "epoch": 6.439495086669922e-05,
      "step": 67523,
      "training_step_time": 8.52511739730835
    },
    {
      "epoch": 6.439590454101563e-05,
      "model_forward_time": 1.734694242477417,
      "step": 67524
    },
    {
      "epoch": 6.439590454101563e-05,
      "step": 67524,
      "training_step_time": 10.42108678817749
    },
    {
      "epoch": 6.439685821533204e-05,
      "model_forward_time": 2.2985687255859375,
      "step": 67525
    },
    {
      "epoch": 6.439685821533204e-05,
      "step": 67525,
      "training_step_time": 7.598398447036743
    },
    {
      "epoch": 6.439781188964843e-05,
      "model_forward_time": 1.7624568939208984,
      "step": 67526
    },
    {
      "epoch": 6.439781188964843e-05,
      "step": 67526,
      "training_step_time": 8.624064922332764
    },
    {
      "epoch": 6.439876556396484e-05,
      "model_forward_time": 1.7656259536743164,
      "step": 67527
    },
    {
      "epoch": 6.439876556396484e-05,
      "step": 67527,
      "training_step_time": 7.605838060379028
    },
    {
      "epoch": 6.439971923828126e-05,
      "model_forward_time": 1.7220475673675537,
      "step": 67528
    },
    {
      "epoch": 6.439971923828126e-05,
      "step": 67528,
      "training_step_time": 7.599773645401001
    },
    {
      "epoch": 6.440067291259765e-05,
      "model_forward_time": 1.7262427806854248,
      "step": 67529
    },
    {
      "epoch": 6.440067291259765e-05,
      "step": 67529,
      "training_step_time": 7.5958051681518555
    },
    {
      "epoch": 6.440162658691406e-05,
      "grad_norm": 0.12813884019851685,
      "learning_rate": 2.6160469982100426e-06,
      "loss": 0.0599,
      "step": 67530
    },
    {
      "epoch": 6.440162658691406e-05,
      "model_forward_time": 1.7261192798614502,
      "step": 67530
    },
    {
      "dynamics_loss_avg": 0.0516227088868618,
      "epoch": 6.440162658691406e-05,
      "step": 67530
    },
    {
      "action_loss_avg": 0.0009164400195004419,
      "epoch": 6.440162658691406e-05,
      "step": 67530
    },
    {
      "epoch": 6.440162658691406e-05,
      "step": 67530,
      "training_step_time": 7.672489404678345
    },
    {
      "epoch": 6.440258026123047e-05,
      "model_forward_time": 1.7195143699645996,
      "step": 67531
    },
    {
      "epoch": 6.440258026123047e-05,
      "step": 67531,
      "training_step_time": 7.529251575469971
    },
    {
      "epoch": 6.440353393554687e-05,
      "model_forward_time": 1.7299673557281494,
      "step": 67532
    },
    {
      "epoch": 6.440353393554687e-05,
      "step": 67532,
      "training_step_time": 9.048285007476807
    },
    {
      "epoch": 6.440448760986328e-05,
      "model_forward_time": 1.730595588684082,
      "step": 67533
    },
    {
      "epoch": 6.440448760986328e-05,
      "step": 67533,
      "training_step_time": 8.669631958007812
    },
    {
      "epoch": 6.44054412841797e-05,
      "model_forward_time": 2.2531657218933105,
      "step": 67534
    },
    {
      "epoch": 6.44054412841797e-05,
      "step": 67534,
      "training_step_time": 7.988930702209473
    },
    {
      "epoch": 6.440639495849609e-05,
      "model_forward_time": 1.7472565174102783,
      "step": 67535
    },
    {
      "epoch": 6.440639495849609e-05,
      "step": 67535,
      "training_step_time": 9.83376955986023
    },
    {
      "epoch": 6.44073486328125e-05,
      "model_forward_time": 1.7968649864196777,
      "step": 67536
    },
    {
      "epoch": 6.44073486328125e-05,
      "step": 67536,
      "training_step_time": 7.8880908489227295
    },
    {
      "epoch": 6.440830230712891e-05,
      "model_forward_time": 1.7235174179077148,
      "step": 67537
    },
    {
      "epoch": 6.440830230712891e-05,
      "step": 67537,
      "training_step_time": 8.594354629516602
    },
    {
      "epoch": 6.440925598144531e-05,
      "model_forward_time": 1.7322142124176025,
      "step": 67538
    },
    {
      "epoch": 6.440925598144531e-05,
      "step": 67538,
      "training_step_time": 8.456894397735596
    },
    {
      "epoch": 6.441020965576172e-05,
      "model_forward_time": 1.7365891933441162,
      "step": 67539
    },
    {
      "epoch": 6.441020965576172e-05,
      "step": 67539,
      "training_step_time": 7.502832412719727
    },
    {
      "epoch": 6.441116333007813e-05,
      "grad_norm": 0.10771695524454117,
      "learning_rate": 2.6145937006637557e-06,
      "loss": 0.0545,
      "step": 67540
    },
    {
      "epoch": 6.441116333007813e-05,
      "model_forward_time": 1.7193405628204346,
      "step": 67540
    },
    {
      "dynamics_loss_avg": 0.05261161997914314,
      "epoch": 6.441116333007813e-05,
      "step": 67540
    },
    {
      "action_loss_avg": 0.0015462912153452635,
      "epoch": 6.441116333007813e-05,
      "step": 67540
    },
    {
      "epoch": 6.441116333007813e-05,
      "step": 67540,
      "training_step_time": 8.2507164478302
    },
    {
      "epoch": 6.441211700439453e-05,
      "model_forward_time": 1.7281103134155273,
      "step": 67541
    },
    {
      "epoch": 6.441211700439453e-05,
      "step": 67541,
      "training_step_time": 8.899409770965576
    },
    {
      "epoch": 6.441307067871094e-05,
      "model_forward_time": 1.7283225059509277,
      "step": 67542
    },
    {
      "epoch": 6.441307067871094e-05,
      "step": 67542,
      "training_step_time": 7.8243794441223145
    },
    {
      "epoch": 6.441402435302735e-05,
      "model_forward_time": 1.718707799911499,
      "step": 67543
    },
    {
      "epoch": 6.441402435302735e-05,
      "step": 67543,
      "training_step_time": 9.028408765792847
    },
    {
      "epoch": 6.441497802734375e-05,
      "model_forward_time": 1.7217721939086914,
      "step": 67544
    },
    {
      "epoch": 6.441497802734375e-05,
      "step": 67544,
      "training_step_time": 7.898584365844727
    },
    {
      "epoch": 6.441593170166016e-05,
      "model_forward_time": 1.719306468963623,
      "step": 67545
    },
    {
      "epoch": 6.441593170166016e-05,
      "step": 67545,
      "training_step_time": 8.85361647605896
    },
    {
      "epoch": 6.441688537597657e-05,
      "model_forward_time": 1.727797269821167,
      "step": 67546
    },
    {
      "epoch": 6.441688537597657e-05,
      "step": 67546,
      "training_step_time": 7.880917549133301
    },
    {
      "epoch": 6.441783905029297e-05,
      "model_forward_time": 1.7246899604797363,
      "step": 67547
    },
    {
      "epoch": 6.441783905029297e-05,
      "step": 67547,
      "training_step_time": 7.856138229370117
    },
    {
      "epoch": 6.441879272460938e-05,
      "model_forward_time": 1.7198631763458252,
      "step": 67548
    },
    {
      "epoch": 6.441879272460938e-05,
      "step": 67548,
      "training_step_time": 7.672271490097046
    },
    {
      "epoch": 6.441974639892579e-05,
      "model_forward_time": 1.7369365692138672,
      "step": 67549
    },
    {
      "epoch": 6.441974639892579e-05,
      "step": 67549,
      "training_step_time": 7.8657495975494385
    },
    {
      "epoch": 6.442070007324218e-05,
      "grad_norm": 0.07891515642404556,
      "learning_rate": 2.613140663981917e-06,
      "loss": 0.0581,
      "step": 67550
    },
    {
      "epoch": 6.442070007324218e-05,
      "model_forward_time": 1.7162318229675293,
      "step": 67550
    },
    {
      "dynamics_loss_avg": 0.043633130565285685,
      "epoch": 6.442070007324218e-05,
      "step": 67550
    },
    {
      "action_loss_avg": 0.0009039432072313503,
      "epoch": 6.442070007324218e-05,
      "step": 67550
    },
    {
      "epoch": 6.442070007324218e-05,
      "step": 67550,
      "training_step_time": 8.917147159576416
    },
    {
      "epoch": 6.44216537475586e-05,
      "model_forward_time": 1.3570101261138916,
      "step": 67551
    },
    {
      "epoch": 6.44216537475586e-05,
      "step": 67551,
      "training_step_time": 8.71789813041687
    },
    {
      "epoch": 6.4422607421875e-05,
      "model_forward_time": 1.7181596755981445,
      "step": 67552
    },
    {
      "epoch": 6.4422607421875e-05,
      "step": 67552,
      "training_step_time": 7.8147571086883545
    },
    {
      "epoch": 6.44235610961914e-05,
      "model_forward_time": 2.8608131408691406,
      "step": 67553
    },
    {
      "epoch": 6.44235610961914e-05,
      "step": 67553,
      "training_step_time": 7.861688852310181
    },
    {
      "epoch": 6.442451477050781e-05,
      "model_forward_time": 1.720005989074707,
      "step": 67554
    },
    {
      "epoch": 6.442451477050781e-05,
      "step": 67554,
      "training_step_time": 8.401077508926392
    },
    {
      "epoch": 6.442546844482423e-05,
      "model_forward_time": 1.7139062881469727,
      "step": 67555
    },
    {
      "epoch": 6.442546844482423e-05,
      "step": 67555,
      "training_step_time": 7.854020118713379
    },
    {
      "epoch": 6.442642211914062e-05,
      "model_forward_time": 1.7137424945831299,
      "step": 67556
    },
    {
      "epoch": 6.442642211914062e-05,
      "step": 67556,
      "training_step_time": 7.872447729110718
    },
    {
      "epoch": 6.442737579345703e-05,
      "model_forward_time": 1.3491652011871338,
      "step": 67557
    },
    {
      "epoch": 6.442737579345703e-05,
      "step": 67557,
      "training_step_time": 7.583613395690918
    },
    {
      "epoch": 6.442832946777344e-05,
      "model_forward_time": 1.3542954921722412,
      "step": 67558
    },
    {
      "epoch": 6.442832946777344e-05,
      "step": 67558,
      "training_step_time": 7.733960390090942
    },
    {
      "epoch": 6.442928314208984e-05,
      "model_forward_time": 1.7306945323944092,
      "step": 67559
    },
    {
      "epoch": 6.442928314208984e-05,
      "step": 67559,
      "training_step_time": 8.0196533203125
    },
    {
      "epoch": 6.443023681640625e-05,
      "grad_norm": 0.08204958587884903,
      "learning_rate": 2.6116878883234243e-06,
      "loss": 0.0526,
      "step": 67560
    },
    {
      "epoch": 6.443023681640625e-05,
      "model_forward_time": 1.7378578186035156,
      "step": 67560
    },
    {
      "dynamics_loss_avg": 0.06057594455778599,
      "epoch": 6.443023681640625e-05,
      "step": 67560
    },
    {
      "action_loss_avg": 0.0023793517437297853,
      "epoch": 6.443023681640625e-05,
      "step": 67560
    },
    {
      "epoch": 6.443023681640625e-05,
      "step": 67560,
      "training_step_time": 7.656398773193359
    },
    {
      "epoch": 6.443119049072266e-05,
      "model_forward_time": 1.7283453941345215,
      "step": 67561
    },
    {
      "epoch": 6.443119049072266e-05,
      "step": 67561,
      "training_step_time": 7.557234287261963
    },
    {
      "epoch": 6.443214416503906e-05,
      "model_forward_time": 2.8619143962860107,
      "step": 67562
    },
    {
      "epoch": 6.443214416503906e-05,
      "step": 67562,
      "training_step_time": 7.7645041942596436
    },
    {
      "epoch": 6.443309783935547e-05,
      "model_forward_time": 1.3675317764282227,
      "step": 67563
    },
    {
      "epoch": 6.443309783935547e-05,
      "step": 67563,
      "training_step_time": 9.428414821624756
    },
    {
      "epoch": 6.443405151367187e-05,
      "model_forward_time": 1.7301089763641357,
      "step": 67564
    },
    {
      "epoch": 6.443405151367187e-05,
      "step": 67564,
      "training_step_time": 8.589917182922363
    },
    {
      "epoch": 6.443500518798828e-05,
      "model_forward_time": 1.734957218170166,
      "step": 67565
    },
    {
      "epoch": 6.443500518798828e-05,
      "step": 67565,
      "training_step_time": 8.160350322723389
    },
    {
      "epoch": 6.443595886230469e-05,
      "model_forward_time": 1.833350658416748,
      "step": 67566
    },
    {
      "epoch": 6.443595886230469e-05,
      "step": 67566,
      "training_step_time": 7.355330228805542
    },
    {
      "epoch": 6.443691253662109e-05,
      "model_forward_time": 1.7376587390899658,
      "step": 67567
    },
    {
      "epoch": 6.443691253662109e-05,
      "step": 67567,
      "training_step_time": 7.4993767738342285
    },
    {
      "epoch": 6.44378662109375e-05,
      "model_forward_time": 1.8121981620788574,
      "step": 67568
    },
    {
      "epoch": 6.44378662109375e-05,
      "step": 67568,
      "training_step_time": 7.497998237609863
    },
    {
      "epoch": 6.443881988525391e-05,
      "model_forward_time": 1.7313568592071533,
      "step": 67569
    },
    {
      "epoch": 6.443881988525391e-05,
      "step": 67569,
      "training_step_time": 8.268446207046509
    },
    {
      "epoch": 6.44397735595703e-05,
      "grad_norm": 0.1333833783864975,
      "learning_rate": 2.610235373847153e-06,
      "loss": 0.0543,
      "step": 67570
    },
    {
      "epoch": 6.44397735595703e-05,
      "model_forward_time": 1.7385475635528564,
      "step": 67570
    },
    {
      "dynamics_loss_avg": 0.06347424015402794,
      "epoch": 6.44397735595703e-05,
      "step": 67570
    },
    {
      "action_loss_avg": 0.0017920884405612015,
      "epoch": 6.44397735595703e-05,
      "step": 67570
    },
    {
      "epoch": 6.44397735595703e-05,
      "step": 67570,
      "training_step_time": 7.545654535293579
    },
    {
      "epoch": 6.444072723388672e-05,
      "model_forward_time": 1.7315106391906738,
      "step": 67571
    },
    {
      "epoch": 6.444072723388672e-05,
      "step": 67571,
      "training_step_time": 8.186790227890015
    },
    {
      "epoch": 6.444168090820313e-05,
      "model_forward_time": 2.930516004562378,
      "step": 67572
    },
    {
      "epoch": 6.444168090820313e-05,
      "step": 67572,
      "training_step_time": 8.507849931716919
    },
    {
      "epoch": 6.444263458251952e-05,
      "model_forward_time": 1.7432072162628174,
      "step": 67573
    },
    {
      "epoch": 6.444263458251952e-05,
      "step": 67573,
      "training_step_time": 8.27545166015625
    },
    {
      "epoch": 6.444358825683594e-05,
      "model_forward_time": 1.7390203475952148,
      "step": 67574
    },
    {
      "epoch": 6.444358825683594e-05,
      "step": 67574,
      "training_step_time": 7.561493158340454
    },
    {
      "epoch": 6.444454193115235e-05,
      "model_forward_time": 1.7340571880340576,
      "step": 67575
    },
    {
      "epoch": 6.444454193115235e-05,
      "step": 67575,
      "training_step_time": 8.213997840881348
    },
    {
      "epoch": 6.444549560546874e-05,
      "model_forward_time": 1.7240087985992432,
      "step": 67576
    },
    {
      "epoch": 6.444549560546874e-05,
      "step": 67576,
      "training_step_time": 8.235036373138428
    },
    {
      "epoch": 6.444644927978515e-05,
      "model_forward_time": 1.7864108085632324,
      "step": 67577
    },
    {
      "epoch": 6.444644927978515e-05,
      "step": 67577,
      "training_step_time": 7.523171901702881
    },
    {
      "epoch": 6.444740295410157e-05,
      "model_forward_time": 1.7689135074615479,
      "step": 67578
    },
    {
      "epoch": 6.444740295410157e-05,
      "step": 67578,
      "training_step_time": 8.328652620315552
    },
    {
      "epoch": 6.444835662841796e-05,
      "model_forward_time": 1.752427339553833,
      "step": 67579
    },
    {
      "epoch": 6.444835662841796e-05,
      "step": 67579,
      "training_step_time": 7.588476181030273
    },
    {
      "epoch": 6.444931030273437e-05,
      "grad_norm": 0.10864558070898056,
      "learning_rate": 2.6087831207119495e-06,
      "loss": 0.0579,
      "step": 67580
    },
    {
      "epoch": 6.444931030273437e-05,
      "model_forward_time": 1.7574717998504639,
      "step": 67580
    },
    {
      "dynamics_loss_avg": 0.0626759272068739,
      "epoch": 6.444931030273437e-05,
      "step": 67580
    },
    {
      "action_loss_avg": 0.0023381394159514455,
      "epoch": 6.444931030273437e-05,
      "step": 67580
    },
    {
      "epoch": 6.444931030273437e-05,
      "step": 67580,
      "training_step_time": 9.229080438613892
    },
    {
      "epoch": 6.445026397705078e-05,
      "model_forward_time": 2.6170358657836914,
      "step": 67581
    },
    {
      "epoch": 6.445026397705078e-05,
      "step": 67581,
      "training_step_time": 8.867433071136475
    },
    {
      "epoch": 6.445121765136718e-05,
      "model_forward_time": 1.7571215629577637,
      "step": 67582
    },
    {
      "epoch": 6.445121765136718e-05,
      "step": 67582,
      "training_step_time": 8.616273403167725
    },
    {
      "epoch": 6.445217132568359e-05,
      "model_forward_time": 1.75730562210083,
      "step": 67583
    },
    {
      "epoch": 6.445217132568359e-05,
      "step": 67583,
      "training_step_time": 7.656737565994263
    },
    {
      "epoch": 6.4453125e-05,
      "model_forward_time": 1.758190631866455,
      "step": 67584
    },
    {
      "epoch": 6.4453125e-05,
      "step": 67584,
      "training_step_time": 7.500040531158447
    },
    {
      "epoch": 6.44540786743164e-05,
      "model_forward_time": 1.7601194381713867,
      "step": 67585
    },
    {
      "epoch": 6.44540786743164e-05,
      "step": 67585,
      "training_step_time": 7.267978668212891
    },
    {
      "epoch": 6.445503234863281e-05,
      "model_forward_time": 1.7919895648956299,
      "step": 67586
    },
    {
      "epoch": 6.445503234863281e-05,
      "step": 67586,
      "training_step_time": 7.476415634155273
    },
    {
      "epoch": 6.445598602294922e-05,
      "model_forward_time": 1.7570569515228271,
      "step": 67587
    },
    {
      "epoch": 6.445598602294922e-05,
      "step": 67587,
      "training_step_time": 7.517125129699707
    },
    {
      "epoch": 6.445693969726562e-05,
      "model_forward_time": 1.7624444961547852,
      "step": 67588
    },
    {
      "epoch": 6.445693969726562e-05,
      "step": 67588,
      "training_step_time": 7.704968214035034
    },
    {
      "epoch": 6.445789337158203e-05,
      "model_forward_time": 1.7790658473968506,
      "step": 67589
    },
    {
      "epoch": 6.445789337158203e-05,
      "step": 67589,
      "training_step_time": 7.671555042266846
    },
    {
      "epoch": 6.445884704589844e-05,
      "grad_norm": 0.08413589745759964,
      "learning_rate": 2.60733112907663e-06,
      "loss": 0.0542,
      "step": 67590
    },
    {
      "epoch": 6.445884704589844e-05,
      "model_forward_time": 2.919635534286499,
      "step": 67590
    },
    {
      "dynamics_loss_avg": 0.03814328042790294,
      "epoch": 6.445884704589844e-05,
      "step": 67590
    },
    {
      "action_loss_avg": 0.0014768600463867187,
      "epoch": 6.445884704589844e-05,
      "step": 67590
    },
    {
      "epoch": 6.445884704589844e-05,
      "step": 67590,
      "training_step_time": 10.469245910644531
    },
    {
      "epoch": 6.445980072021484e-05,
      "model_forward_time": 1.768444299697876,
      "step": 67591
    },
    {
      "epoch": 6.445980072021484e-05,
      "step": 67591,
      "training_step_time": 9.610996961593628
    },
    {
      "epoch": 6.446075439453125e-05,
      "model_forward_time": 1.7603867053985596,
      "step": 67592
    },
    {
      "epoch": 6.446075439453125e-05,
      "step": 67592,
      "training_step_time": 7.726929426193237
    },
    {
      "epoch": 6.446170806884766e-05,
      "model_forward_time": 1.7691924571990967,
      "step": 67593
    },
    {
      "epoch": 6.446170806884766e-05,
      "step": 67593,
      "training_step_time": 7.614380836486816
    },
    {
      "epoch": 6.446266174316406e-05,
      "model_forward_time": 1.744518518447876,
      "step": 67594
    },
    {
      "epoch": 6.446266174316406e-05,
      "step": 67594,
      "training_step_time": 7.345090866088867
    },
    {
      "epoch": 6.446361541748047e-05,
      "model_forward_time": 1.7360498905181885,
      "step": 67595
    },
    {
      "epoch": 6.446361541748047e-05,
      "step": 67595,
      "training_step_time": 7.515243768692017
    },
    {
      "epoch": 6.446456909179688e-05,
      "model_forward_time": 1.8831501007080078,
      "step": 67596
    },
    {
      "epoch": 6.446456909179688e-05,
      "step": 67596,
      "training_step_time": 7.576326847076416
    },
    {
      "epoch": 6.446552276611328e-05,
      "model_forward_time": 1.7375223636627197,
      "step": 67597
    },
    {
      "epoch": 6.446552276611328e-05,
      "step": 67597,
      "training_step_time": 7.515534162521362
    },
    {
      "epoch": 6.446647644042969e-05,
      "model_forward_time": 1.7415759563446045,
      "step": 67598
    },
    {
      "epoch": 6.446647644042969e-05,
      "step": 67598,
      "training_step_time": 8.650942325592041
    },
    {
      "epoch": 6.44674301147461e-05,
      "model_forward_time": 2.862912893295288,
      "step": 67599
    },
    {
      "epoch": 6.44674301147461e-05,
      "step": 67599,
      "training_step_time": 7.595724582672119
    },
    {
      "epoch": 6.44683837890625e-05,
      "grad_norm": 0.12107491493225098,
      "learning_rate": 2.6058793990999785e-06,
      "loss": 0.0556,
      "step": 67600
    },
    {
      "epoch": 6.44683837890625e-05,
      "model_forward_time": 1.7423622608184814,
      "step": 67600
    },
    {
      "dynamics_loss_avg": 0.056960318423807624,
      "epoch": 6.44683837890625e-05,
      "step": 67600
    },
    {
      "action_loss_avg": 0.0018794183095451444,
      "epoch": 6.44683837890625e-05,
      "step": 67600
    },
    {
      "epoch": 6.44683837890625e-05,
      "step": 67600,
      "training_step_time": 9.27231216430664
    },
    {
      "epoch": 6.44693374633789e-05,
      "model_forward_time": 1.735870599746704,
      "step": 67601
    },
    {
      "epoch": 6.44693374633789e-05,
      "step": 67601,
      "training_step_time": 7.64360785484314
    },
    {
      "epoch": 6.447029113769532e-05,
      "model_forward_time": 1.7334558963775635,
      "step": 67602
    },
    {
      "epoch": 6.447029113769532e-05,
      "step": 67602,
      "training_step_time": 7.56084942817688
    },
    {
      "epoch": 6.447124481201171e-05,
      "model_forward_time": 1.7379395961761475,
      "step": 67603
    },
    {
      "epoch": 6.447124481201171e-05,
      "step": 67603,
      "training_step_time": 7.3713603019714355
    },
    {
      "epoch": 6.447219848632812e-05,
      "model_forward_time": 1.7320778369903564,
      "step": 67604
    },
    {
      "epoch": 6.447219848632812e-05,
      "step": 67604,
      "training_step_time": 7.499093770980835
    },
    {
      "epoch": 6.447315216064454e-05,
      "model_forward_time": 1.8640727996826172,
      "step": 67605
    },
    {
      "epoch": 6.447315216064454e-05,
      "step": 67605,
      "training_step_time": 8.565679788589478
    },
    {
      "epoch": 6.447410583496093e-05,
      "model_forward_time": 1.733008623123169,
      "step": 67606
    },
    {
      "epoch": 6.447410583496093e-05,
      "step": 67606,
      "training_step_time": 7.554516792297363
    },
    {
      "epoch": 6.447505950927734e-05,
      "model_forward_time": 1.7308807373046875,
      "step": 67607
    },
    {
      "epoch": 6.447505950927734e-05,
      "step": 67607,
      "training_step_time": 7.677521228790283
    },
    {
      "epoch": 6.447601318359375e-05,
      "model_forward_time": 1.7426378726959229,
      "step": 67608
    },
    {
      "epoch": 6.447601318359375e-05,
      "step": 67608,
      "training_step_time": 7.69326376914978
    },
    {
      "epoch": 6.447696685791015e-05,
      "model_forward_time": 2.907771587371826,
      "step": 67609
    },
    {
      "epoch": 6.447696685791015e-05,
      "step": 67609,
      "training_step_time": 7.65535306930542
    },
    {
      "epoch": 6.447792053222656e-05,
      "grad_norm": 0.1254039704799652,
      "learning_rate": 2.6044279309407596e-06,
      "loss": 0.0537,
      "step": 67610
    },
    {
      "epoch": 6.447792053222656e-05,
      "model_forward_time": 1.7313251495361328,
      "step": 67610
    },
    {
      "dynamics_loss_avg": 0.06253024376928806,
      "epoch": 6.447792053222656e-05,
      "step": 67610
    },
    {
      "action_loss_avg": 0.003197612060466781,
      "epoch": 6.447792053222656e-05,
      "step": 67610
    },
    {
      "epoch": 6.447792053222656e-05,
      "step": 67610,
      "training_step_time": 7.634109735488892
    },
    {
      "epoch": 6.447887420654297e-05,
      "model_forward_time": 1.7700560092926025,
      "step": 67611
    },
    {
      "epoch": 6.447887420654297e-05,
      "step": 67611,
      "training_step_time": 7.569085597991943
    },
    {
      "epoch": 6.447982788085937e-05,
      "model_forward_time": 1.7609283924102783,
      "step": 67612
    },
    {
      "epoch": 6.447982788085937e-05,
      "step": 67612,
      "training_step_time": 7.343632459640503
    },
    {
      "epoch": 6.448078155517578e-05,
      "model_forward_time": 1.7647302150726318,
      "step": 67613
    },
    {
      "epoch": 6.448078155517578e-05,
      "step": 67613,
      "training_step_time": 7.506109237670898
    },
    {
      "epoch": 6.448173522949219e-05,
      "model_forward_time": 1.8536174297332764,
      "step": 67614
    },
    {
      "epoch": 6.448173522949219e-05,
      "step": 67614,
      "training_step_time": 7.750293731689453
    },
    {
      "epoch": 6.448268890380859e-05,
      "model_forward_time": 1.7600865364074707,
      "step": 67615
    },
    {
      "epoch": 6.448268890380859e-05,
      "step": 67615,
      "training_step_time": 7.814660310745239
    },
    {
      "epoch": 6.4483642578125e-05,
      "model_forward_time": 1.757138729095459,
      "step": 67616
    },
    {
      "epoch": 6.4483642578125e-05,
      "step": 67616,
      "training_step_time": 7.880541086196899
    },
    {
      "epoch": 6.448459625244141e-05,
      "model_forward_time": 1.7612650394439697,
      "step": 67617
    },
    {
      "epoch": 6.448459625244141e-05,
      "step": 67617,
      "training_step_time": 9.60150694847107
    },
    {
      "epoch": 6.448554992675781e-05,
      "model_forward_time": 2.9465668201446533,
      "step": 67618
    },
    {
      "epoch": 6.448554992675781e-05,
      "step": 67618,
      "training_step_time": 8.570841073989868
    },
    {
      "epoch": 6.448650360107422e-05,
      "model_forward_time": 1.7504854202270508,
      "step": 67619
    },
    {
      "epoch": 6.448650360107422e-05,
      "step": 67619,
      "training_step_time": 7.857850074768066
    },
    {
      "epoch": 6.448745727539063e-05,
      "grad_norm": 0.08633031696081161,
      "learning_rate": 2.6029767247576974e-06,
      "loss": 0.0543,
      "step": 67620
    },
    {
      "epoch": 6.448745727539063e-05,
      "model_forward_time": 1.7653369903564453,
      "step": 67620
    },
    {
      "dynamics_loss_avg": 0.04060137718915939,
      "epoch": 6.448745727539063e-05,
      "step": 67620
    },
    {
      "action_loss_avg": 0.0012787687446689233,
      "epoch": 6.448745727539063e-05,
      "step": 67620
    },
    {
      "epoch": 6.448745727539063e-05,
      "step": 67620,
      "training_step_time": 7.633127450942993
    },
    {
      "epoch": 6.448841094970703e-05,
      "model_forward_time": 1.746793508529663,
      "step": 67621
    },
    {
      "epoch": 6.448841094970703e-05,
      "step": 67621,
      "training_step_time": 7.598910331726074
    },
    {
      "epoch": 6.448936462402344e-05,
      "model_forward_time": 1.7522501945495605,
      "step": 67622
    },
    {
      "epoch": 6.448936462402344e-05,
      "step": 67622,
      "training_step_time": 7.7967610359191895
    },
    {
      "epoch": 6.449031829833985e-05,
      "model_forward_time": 1.7743127346038818,
      "step": 67623
    },
    {
      "epoch": 6.449031829833985e-05,
      "step": 67623,
      "training_step_time": 7.7675957679748535
    },
    {
      "epoch": 6.449127197265625e-05,
      "model_forward_time": 1.7409436702728271,
      "step": 67624
    },
    {
      "epoch": 6.449127197265625e-05,
      "step": 67624,
      "training_step_time": 7.894385814666748
    },
    {
      "epoch": 6.449222564697266e-05,
      "model_forward_time": 1.7501018047332764,
      "step": 67625
    },
    {
      "epoch": 6.449222564697266e-05,
      "step": 67625,
      "training_step_time": 7.918732643127441
    },
    {
      "epoch": 6.449317932128907e-05,
      "model_forward_time": 1.7328128814697266,
      "step": 67626
    },
    {
      "epoch": 6.449317932128907e-05,
      "step": 67626,
      "training_step_time": 8.75675082206726
    },
    {
      "epoch": 6.449413299560546e-05,
      "model_forward_time": 2.9791834354400635,
      "step": 67627
    },
    {
      "epoch": 6.449413299560546e-05,
      "step": 67627,
      "training_step_time": 8.753232955932617
    },
    {
      "epoch": 6.449508666992188e-05,
      "model_forward_time": 1.3735311031341553,
      "step": 67628
    },
    {
      "epoch": 6.449508666992188e-05,
      "step": 67628,
      "training_step_time": 7.845321178436279
    },
    {
      "epoch": 6.449604034423829e-05,
      "model_forward_time": 1.7459394931793213,
      "step": 67629
    },
    {
      "epoch": 6.449604034423829e-05,
      "step": 67629,
      "training_step_time": 9.636814832687378
    },
    {
      "epoch": 6.449699401855468e-05,
      "grad_norm": 0.11287864297628403,
      "learning_rate": 2.601525780709497e-06,
      "loss": 0.0553,
      "step": 67630
    },
    {
      "epoch": 6.449699401855468e-05,
      "model_forward_time": 1.7484886646270752,
      "step": 67630
    },
    {
      "dynamics_loss_avg": 0.04846640545874834,
      "epoch": 6.449699401855468e-05,
      "step": 67630
    },
    {
      "action_loss_avg": 0.0013329844980034978,
      "epoch": 6.449699401855468e-05,
      "step": 67630
    },
    {
      "epoch": 6.449699401855468e-05,
      "step": 67630,
      "training_step_time": 8.563239812850952
    },
    {
      "epoch": 6.44979476928711e-05,
      "model_forward_time": 1.7469284534454346,
      "step": 67631
    },
    {
      "epoch": 6.44979476928711e-05,
      "step": 67631,
      "training_step_time": 7.796879768371582
    },
    {
      "epoch": 6.44989013671875e-05,
      "model_forward_time": 1.7470941543579102,
      "step": 67632
    },
    {
      "epoch": 6.44989013671875e-05,
      "step": 67632,
      "training_step_time": 8.79880142211914
    },
    {
      "epoch": 6.44998550415039e-05,
      "model_forward_time": 1.7409746646881104,
      "step": 67633
    },
    {
      "epoch": 6.44998550415039e-05,
      "step": 67633,
      "training_step_time": 8.824517011642456
    },
    {
      "epoch": 6.450080871582031e-05,
      "model_forward_time": 1.7416818141937256,
      "step": 67634
    },
    {
      "epoch": 6.450080871582031e-05,
      "step": 67634,
      "training_step_time": 7.734560012817383
    },
    {
      "epoch": 6.450176239013672e-05,
      "model_forward_time": 1.7391998767852783,
      "step": 67635
    },
    {
      "epoch": 6.450176239013672e-05,
      "step": 67635,
      "training_step_time": 7.886088609695435
    },
    {
      "epoch": 6.450271606445312e-05,
      "model_forward_time": 2.927098274230957,
      "step": 67636
    },
    {
      "epoch": 6.450271606445312e-05,
      "step": 67636,
      "training_step_time": 7.930099010467529
    },
    {
      "epoch": 6.450366973876953e-05,
      "model_forward_time": 1.745311975479126,
      "step": 67637
    },
    {
      "epoch": 6.450366973876953e-05,
      "step": 67637,
      "training_step_time": 7.8526952266693115
    },
    {
      "epoch": 6.450462341308594e-05,
      "model_forward_time": 1.7381927967071533,
      "step": 67638
    },
    {
      "epoch": 6.450462341308594e-05,
      "step": 67638,
      "training_step_time": 8.824321985244751
    },
    {
      "epoch": 6.450557708740234e-05,
      "model_forward_time": 1.7513909339904785,
      "step": 67639
    },
    {
      "epoch": 6.450557708740234e-05,
      "step": 67639,
      "training_step_time": 7.873507499694824
    },
    {
      "epoch": 6.450653076171875e-05,
      "grad_norm": 0.07314585894346237,
      "learning_rate": 2.600075098954832e-06,
      "loss": 0.0559,
      "step": 67640
    },
    {
      "epoch": 6.450653076171875e-05,
      "model_forward_time": 1.738846778869629,
      "step": 67640
    },
    {
      "dynamics_loss_avg": 0.052109626680612565,
      "epoch": 6.450653076171875e-05,
      "step": 67640
    },
    {
      "action_loss_avg": 0.0019592128985095767,
      "epoch": 6.450653076171875e-05,
      "step": 67640
    },
    {
      "epoch": 6.450653076171875e-05,
      "step": 67640,
      "training_step_time": 7.843540668487549
    },
    {
      "epoch": 6.450748443603516e-05,
      "model_forward_time": 1.7421314716339111,
      "step": 67641
    },
    {
      "epoch": 6.450748443603516e-05,
      "step": 67641,
      "training_step_time": 7.754406690597534
    },
    {
      "epoch": 6.450843811035156e-05,
      "model_forward_time": 1.8830926418304443,
      "step": 67642
    },
    {
      "epoch": 6.450843811035156e-05,
      "step": 67642,
      "training_step_time": 7.824147939682007
    },
    {
      "epoch": 6.450939178466797e-05,
      "model_forward_time": 1.740412950515747,
      "step": 67643
    },
    {
      "epoch": 6.450939178466797e-05,
      "step": 67643,
      "training_step_time": 7.966047525405884
    },
    {
      "epoch": 6.451034545898438e-05,
      "model_forward_time": 1.7390029430389404,
      "step": 67644
    },
    {
      "epoch": 6.451034545898438e-05,
      "step": 67644,
      "training_step_time": 7.834222078323364
    },
    {
      "epoch": 6.451129913330078e-05,
      "model_forward_time": 1.7432475090026855,
      "step": 67645
    },
    {
      "epoch": 6.451129913330078e-05,
      "step": 67645,
      "training_step_time": 8.848506689071655
    },
    {
      "epoch": 6.451225280761719e-05,
      "model_forward_time": 1.7409696578979492,
      "step": 67646
    },
    {
      "epoch": 6.451225280761719e-05,
      "step": 67646,
      "training_step_time": 8.302454710006714
    },
    {
      "epoch": 6.45132064819336e-05,
      "model_forward_time": 1.73844313621521,
      "step": 67647
    },
    {
      "epoch": 6.45132064819336e-05,
      "step": 67647,
      "training_step_time": 8.795977115631104
    },
    {
      "epoch": 6.451416015625e-05,
      "model_forward_time": 1.7402691841125488,
      "step": 67648
    },
    {
      "epoch": 6.451416015625e-05,
      "step": 67648,
      "training_step_time": 7.845369815826416
    },
    {
      "epoch": 6.451511383056641e-05,
      "model_forward_time": 1.7717211246490479,
      "step": 67649
    },
    {
      "epoch": 6.451511383056641e-05,
      "step": 67649,
      "training_step_time": 7.745203495025635
    },
    {
      "epoch": 6.451606750488282e-05,
      "grad_norm": 0.10636377334594727,
      "learning_rate": 2.598624679652346e-06,
      "loss": 0.0518,
      "step": 67650
    },
    {
      "epoch": 6.451606750488282e-05,
      "model_forward_time": 1.7721161842346191,
      "step": 67650
    },
    {
      "dynamics_loss_avg": 0.06122151296585798,
      "epoch": 6.451606750488282e-05,
      "step": 67650
    },
    {
      "action_loss_avg": 0.002197606541449204,
      "epoch": 6.451606750488282e-05,
      "step": 67650
    },
    {
      "epoch": 6.451606750488282e-05,
      "step": 67650,
      "training_step_time": 7.544593334197998
    },
    {
      "epoch": 6.451702117919922e-05,
      "model_forward_time": 1.804558515548706,
      "step": 67651
    },
    {
      "epoch": 6.451702117919922e-05,
      "step": 67651,
      "training_step_time": 8.771170854568481
    },
    {
      "epoch": 6.451797485351563e-05,
      "model_forward_time": 1.7473573684692383,
      "step": 67652
    },
    {
      "epoch": 6.451797485351563e-05,
      "step": 67652,
      "training_step_time": 8.309216499328613
    },
    {
      "epoch": 6.451892852783204e-05,
      "model_forward_time": 1.7496724128723145,
      "step": 67653
    },
    {
      "epoch": 6.451892852783204e-05,
      "step": 67653,
      "training_step_time": 7.877999544143677
    },
    {
      "epoch": 6.451988220214843e-05,
      "model_forward_time": 1.7375953197479248,
      "step": 67654
    },
    {
      "epoch": 6.451988220214843e-05,
      "step": 67654,
      "training_step_time": 7.898361921310425
    },
    {
      "epoch": 6.452083587646484e-05,
      "model_forward_time": 2.9236507415771484,
      "step": 67655
    },
    {
      "epoch": 6.452083587646484e-05,
      "step": 67655,
      "training_step_time": 7.918312311172485
    },
    {
      "epoch": 6.452178955078126e-05,
      "model_forward_time": 1.7354340553283691,
      "step": 67656
    },
    {
      "epoch": 6.452178955078126e-05,
      "step": 67656,
      "training_step_time": 7.841535806655884
    },
    {
      "epoch": 6.452274322509765e-05,
      "model_forward_time": 1.7410337924957275,
      "step": 67657
    },
    {
      "epoch": 6.452274322509765e-05,
      "step": 67657,
      "training_step_time": 8.435640811920166
    },
    {
      "epoch": 6.452369689941406e-05,
      "model_forward_time": 1.758354663848877,
      "step": 67658
    },
    {
      "epoch": 6.452369689941406e-05,
      "step": 67658,
      "training_step_time": 7.821001052856445
    },
    {
      "epoch": 6.452465057373047e-05,
      "model_forward_time": 1.7361891269683838,
      "step": 67659
    },
    {
      "epoch": 6.452465057373047e-05,
      "step": 67659,
      "training_step_time": 7.535881519317627
    },
    {
      "epoch": 6.452560424804687e-05,
      "grad_norm": 0.10280590504407883,
      "learning_rate": 2.597174522960655e-06,
      "loss": 0.0558,
      "step": 67660
    },
    {
      "epoch": 6.452560424804687e-05,
      "model_forward_time": 1.7910616397857666,
      "step": 67660
    },
    {
      "dynamics_loss_avg": 0.05535285044461489,
      "epoch": 6.452560424804687e-05,
      "step": 67660
    },
    {
      "action_loss_avg": 0.0014024632808286696,
      "epoch": 6.452560424804687e-05,
      "step": 67660
    },
    {
      "epoch": 6.452560424804687e-05,
      "step": 67660,
      "training_step_time": 7.780112266540527
    },
    {
      "epoch": 6.452655792236328e-05,
      "model_forward_time": 1.7328262329101562,
      "step": 67661
    },
    {
      "epoch": 6.452655792236328e-05,
      "step": 67661,
      "training_step_time": 8.452489376068115
    },
    {
      "epoch": 6.45275115966797e-05,
      "model_forward_time": 1.74627685546875,
      "step": 67662
    },
    {
      "epoch": 6.45275115966797e-05,
      "step": 67662,
      "training_step_time": 7.910310506820679
    },
    {
      "epoch": 6.452846527099609e-05,
      "model_forward_time": 1.7505366802215576,
      "step": 67663
    },
    {
      "epoch": 6.452846527099609e-05,
      "step": 67663,
      "training_step_time": 8.861519575119019
    },
    {
      "epoch": 6.45294189453125e-05,
      "model_forward_time": 2.9777097702026367,
      "step": 67664
    },
    {
      "epoch": 6.45294189453125e-05,
      "step": 67664,
      "training_step_time": 8.781224966049194
    },
    {
      "epoch": 6.453037261962891e-05,
      "model_forward_time": 1.7469735145568848,
      "step": 67665
    },
    {
      "epoch": 6.453037261962891e-05,
      "step": 67665,
      "training_step_time": 9.714581966400146
    },
    {
      "epoch": 6.453132629394531e-05,
      "model_forward_time": 1.7391831874847412,
      "step": 67666
    },
    {
      "epoch": 6.453132629394531e-05,
      "step": 67666,
      "training_step_time": 7.756930351257324
    },
    {
      "epoch": 6.453227996826172e-05,
      "model_forward_time": 1.7515456676483154,
      "step": 67667
    },
    {
      "epoch": 6.453227996826172e-05,
      "step": 67667,
      "training_step_time": 7.795180559158325
    },
    {
      "epoch": 6.453323364257813e-05,
      "model_forward_time": 1.7424077987670898,
      "step": 67668
    },
    {
      "epoch": 6.453323364257813e-05,
      "step": 67668,
      "training_step_time": 7.718562126159668
    },
    {
      "epoch": 6.453418731689453e-05,
      "model_forward_time": 1.7555434703826904,
      "step": 67669
    },
    {
      "epoch": 6.453418731689453e-05,
      "step": 67669,
      "training_step_time": 8.412409543991089
    },
    {
      "epoch": 6.453514099121094e-05,
      "grad_norm": 0.11280739307403564,
      "learning_rate": 2.595724629038344e-06,
      "loss": 0.059,
      "step": 67670
    },
    {
      "epoch": 6.453514099121094e-05,
      "model_forward_time": 1.7446115016937256,
      "step": 67670
    },
    {
      "dynamics_loss_avg": 0.05820155031979084,
      "epoch": 6.453514099121094e-05,
      "step": 67670
    },
    {
      "action_loss_avg": 0.0018152488948544488,
      "epoch": 6.453514099121094e-05,
      "step": 67670
    },
    {
      "epoch": 6.453514099121094e-05,
      "step": 67670,
      "training_step_time": 8.705161333084106
    },
    {
      "epoch": 6.453609466552735e-05,
      "model_forward_time": 1.7404136657714844,
      "step": 67671
    },
    {
      "epoch": 6.453609466552735e-05,
      "step": 67671,
      "training_step_time": 9.408930540084839
    },
    {
      "epoch": 6.453704833984375e-05,
      "model_forward_time": 1.7406175136566162,
      "step": 67672
    },
    {
      "epoch": 6.453704833984375e-05,
      "step": 67672,
      "training_step_time": 7.744010925292969
    },
    {
      "epoch": 6.453800201416016e-05,
      "model_forward_time": 2.9059298038482666,
      "step": 67673
    },
    {
      "epoch": 6.453800201416016e-05,
      "step": 67673,
      "training_step_time": 9.415362119674683
    },
    {
      "epoch": 6.453895568847657e-05,
      "model_forward_time": 1.7438130378723145,
      "step": 67674
    },
    {
      "epoch": 6.453895568847657e-05,
      "step": 67674,
      "training_step_time": 7.791804790496826
    },
    {
      "epoch": 6.453990936279297e-05,
      "model_forward_time": 1.7327940464019775,
      "step": 67675
    },
    {
      "epoch": 6.453990936279297e-05,
      "step": 67675,
      "training_step_time": 9.62592363357544
    },
    {
      "epoch": 6.454086303710938e-05,
      "model_forward_time": 1.7524855136871338,
      "step": 67676
    },
    {
      "epoch": 6.454086303710938e-05,
      "step": 67676,
      "training_step_time": 8.522966384887695
    },
    {
      "epoch": 6.454181671142579e-05,
      "model_forward_time": 1.7366118431091309,
      "step": 67677
    },
    {
      "epoch": 6.454181671142579e-05,
      "step": 67677,
      "training_step_time": 7.7046058177948
    },
    {
      "epoch": 6.454277038574218e-05,
      "model_forward_time": 1.744286060333252,
      "step": 67678
    },
    {
      "epoch": 6.454277038574218e-05,
      "step": 67678,
      "training_step_time": 7.758055686950684
    },
    {
      "epoch": 6.45437240600586e-05,
      "model_forward_time": 1.8598158359527588,
      "step": 67679
    },
    {
      "epoch": 6.45437240600586e-05,
      "step": 67679,
      "training_step_time": 7.770287990570068
    },
    {
      "epoch": 6.4544677734375e-05,
      "grad_norm": 0.08812151104211807,
      "learning_rate": 2.594274998043975e-06,
      "loss": 0.0525,
      "step": 67680
    },
    {
      "epoch": 6.4544677734375e-05,
      "model_forward_time": 1.7435905933380127,
      "step": 67680
    },
    {
      "dynamics_loss_avg": 0.05281379446387291,
      "epoch": 6.4544677734375e-05,
      "step": 67680
    },
    {
      "action_loss_avg": 0.0023123500985093417,
      "epoch": 6.4544677734375e-05,
      "step": 67680
    },
    {
      "epoch": 6.4544677734375e-05,
      "step": 67680,
      "training_step_time": 7.8901214599609375
    },
    {
      "epoch": 6.45456314086914e-05,
      "model_forward_time": 1.7522578239440918,
      "step": 67681
    },
    {
      "epoch": 6.45456314086914e-05,
      "step": 67681,
      "training_step_time": 10.405022382736206
    },
    {
      "epoch": 6.454658508300781e-05,
      "model_forward_time": 2.9120349884033203,
      "step": 67682
    },
    {
      "epoch": 6.454658508300781e-05,
      "step": 67682,
      "training_step_time": 9.663862705230713
    },
    {
      "epoch": 6.454753875732423e-05,
      "model_forward_time": 1.7480895519256592,
      "step": 67683
    },
    {
      "epoch": 6.454753875732423e-05,
      "step": 67683,
      "training_step_time": 8.676571369171143
    },
    {
      "epoch": 6.454849243164062e-05,
      "model_forward_time": 1.746608018875122,
      "step": 67684
    },
    {
      "epoch": 6.454849243164062e-05,
      "step": 67684,
      "training_step_time": 7.883716583251953
    },
    {
      "epoch": 6.454944610595703e-05,
      "model_forward_time": 1.7431600093841553,
      "step": 67685
    },
    {
      "epoch": 6.454944610595703e-05,
      "step": 67685,
      "training_step_time": 7.896233320236206
    },
    {
      "epoch": 6.455039978027344e-05,
      "model_forward_time": 1.7407617568969727,
      "step": 67686
    },
    {
      "epoch": 6.455039978027344e-05,
      "step": 67686,
      "training_step_time": 7.822834491729736
    },
    {
      "epoch": 6.455135345458984e-05,
      "model_forward_time": 1.7542524337768555,
      "step": 67687
    },
    {
      "epoch": 6.455135345458984e-05,
      "step": 67687,
      "training_step_time": 7.772297382354736
    },
    {
      "epoch": 6.455230712890625e-05,
      "model_forward_time": 1.8611299991607666,
      "step": 67688
    },
    {
      "epoch": 6.455230712890625e-05,
      "step": 67688,
      "training_step_time": 8.823294878005981
    },
    {
      "epoch": 6.455326080322265e-05,
      "model_forward_time": 1.7407429218292236,
      "step": 67689
    },
    {
      "epoch": 6.455326080322265e-05,
      "step": 67689,
      "training_step_time": 8.930068731307983
    },
    {
      "epoch": 6.455421447753906e-05,
      "grad_norm": 0.109101802110672,
      "learning_rate": 2.5928256301360733e-06,
      "loss": 0.0546,
      "step": 67690
    },
    {
      "epoch": 6.455421447753906e-05,
      "model_forward_time": 1.7460455894470215,
      "step": 67690
    },
    {
      "dynamics_loss_avg": 0.04346459656953812,
      "epoch": 6.455421447753906e-05,
      "step": 67690
    },
    {
      "action_loss_avg": 0.0018474875949323177,
      "epoch": 6.455421447753906e-05,
      "step": 67690
    },
    {
      "epoch": 6.455421447753906e-05,
      "step": 67690,
      "training_step_time": 8.553619146347046
    },
    {
      "epoch": 6.455516815185547e-05,
      "model_forward_time": 2.9219112396240234,
      "step": 67691
    },
    {
      "epoch": 6.455516815185547e-05,
      "step": 67691,
      "training_step_time": 9.092707872390747
    },
    {
      "epoch": 6.455612182617187e-05,
      "model_forward_time": 1.736086368560791,
      "step": 67692
    },
    {
      "epoch": 6.455612182617187e-05,
      "step": 67692,
      "training_step_time": 7.812816381454468
    },
    {
      "epoch": 6.455707550048828e-05,
      "model_forward_time": 1.7435059547424316,
      "step": 67693
    },
    {
      "epoch": 6.455707550048828e-05,
      "step": 67693,
      "training_step_time": 7.839405298233032
    },
    {
      "epoch": 6.455802917480469e-05,
      "model_forward_time": 1.7477679252624512,
      "step": 67694
    },
    {
      "epoch": 6.455802917480469e-05,
      "step": 67694,
      "training_step_time": 7.9128258228302
    },
    {
      "epoch": 6.455898284912109e-05,
      "model_forward_time": 1.7444796562194824,
      "step": 67695
    },
    {
      "epoch": 6.455898284912109e-05,
      "step": 67695,
      "training_step_time": 7.6302080154418945
    },
    {
      "epoch": 6.45599365234375e-05,
      "model_forward_time": 1.734588623046875,
      "step": 67696
    },
    {
      "epoch": 6.45599365234375e-05,
      "step": 67696,
      "training_step_time": 7.679211616516113
    },
    {
      "epoch": 6.456089019775391e-05,
      "model_forward_time": 1.8229775428771973,
      "step": 67697
    },
    {
      "epoch": 6.456089019775391e-05,
      "step": 67697,
      "training_step_time": 8.391480207443237
    },
    {
      "epoch": 6.45618438720703e-05,
      "model_forward_time": 1.3814172744750977,
      "step": 67698
    },
    {
      "epoch": 6.45618438720703e-05,
      "step": 67698,
      "training_step_time": 8.921179294586182
    },
    {
      "epoch": 6.456279754638672e-05,
      "model_forward_time": 1.753237247467041,
      "step": 67699
    },
    {
      "epoch": 6.456279754638672e-05,
      "step": 67699,
      "training_step_time": 8.48865032196045
    },
    {
      "epoch": 6.456375122070313e-05,
      "grad_norm": 0.10093820095062256,
      "learning_rate": 2.591376525473142e-06,
      "loss": 0.0563,
      "step": 67700
    },
    {
      "epoch": 6.456375122070313e-05,
      "model_forward_time": 1.7419147491455078,
      "step": 67700
    },
    {
      "dynamics_loss_avg": 0.0470227618701756,
      "epoch": 6.456375122070313e-05,
      "step": 67700
    },
    {
      "action_loss_avg": 0.001080526551231742,
      "epoch": 6.456375122070313e-05,
      "step": 67700
    },
    {
      "epoch": 6.456375122070313e-05,
      "step": 67700,
      "training_step_time": 8.407312154769897
    },
    {
      "epoch": 6.456470489501952e-05,
      "model_forward_time": 2.9800145626068115,
      "step": 67701
    },
    {
      "epoch": 6.456470489501952e-05,
      "step": 67701,
      "training_step_time": 9.934106588363647
    },
    {
      "epoch": 6.456565856933594e-05,
      "model_forward_time": 1.7444484233856201,
      "step": 67702
    },
    {
      "epoch": 6.456565856933594e-05,
      "step": 67702,
      "training_step_time": 7.804328203201294
    },
    {
      "epoch": 6.456661224365235e-05,
      "model_forward_time": 1.7409281730651855,
      "step": 67703
    },
    {
      "epoch": 6.456661224365235e-05,
      "step": 67703,
      "training_step_time": 7.807643413543701
    },
    {
      "epoch": 6.456756591796874e-05,
      "model_forward_time": 1.7409663200378418,
      "step": 67704
    },
    {
      "epoch": 6.456756591796874e-05,
      "step": 67704,
      "training_step_time": 8.355188131332397
    },
    {
      "epoch": 6.456851959228515e-05,
      "model_forward_time": 1.738929271697998,
      "step": 67705
    },
    {
      "epoch": 6.456851959228515e-05,
      "step": 67705,
      "training_step_time": 7.719746112823486
    },
    {
      "epoch": 6.456947326660157e-05,
      "model_forward_time": 1.8091259002685547,
      "step": 67706
    },
    {
      "epoch": 6.456947326660157e-05,
      "step": 67706,
      "training_step_time": 7.75276255607605
    },
    {
      "epoch": 6.457042694091796e-05,
      "model_forward_time": 1.735872745513916,
      "step": 67707
    },
    {
      "epoch": 6.457042694091796e-05,
      "step": 67707,
      "training_step_time": 8.407744884490967
    },
    {
      "epoch": 6.457138061523437e-05,
      "model_forward_time": 1.7472026348114014,
      "step": 67708
    },
    {
      "epoch": 6.457138061523437e-05,
      "step": 67708,
      "training_step_time": 7.83588981628418
    },
    {
      "epoch": 6.457233428955078e-05,
      "model_forward_time": 1.7436208724975586,
      "step": 67709
    },
    {
      "epoch": 6.457233428955078e-05,
      "step": 67709,
      "training_step_time": 8.60199499130249
    },
    {
      "epoch": 6.457328796386718e-05,
      "grad_norm": 0.0945076197385788,
      "learning_rate": 2.5899276842136522e-06,
      "loss": 0.0576,
      "step": 67710
    },
    {
      "epoch": 6.457328796386718e-05,
      "model_forward_time": 2.9247829914093018,
      "step": 67710
    },
    {
      "dynamics_loss_avg": 0.054073890950530765,
      "epoch": 6.457328796386718e-05,
      "step": 67710
    },
    {
      "action_loss_avg": 0.0018971070618135854,
      "epoch": 6.457328796386718e-05,
      "step": 67710
    },
    {
      "epoch": 6.457328796386718e-05,
      "step": 67710,
      "training_step_time": 8.45026707649231
    },
    {
      "epoch": 6.457424163818359e-05,
      "model_forward_time": 1.7542457580566406,
      "step": 67711
    },
    {
      "epoch": 6.457424163818359e-05,
      "step": 67711,
      "training_step_time": 8.929049968719482
    },
    {
      "epoch": 6.45751953125e-05,
      "model_forward_time": 1.7364985942840576,
      "step": 67712
    },
    {
      "epoch": 6.45751953125e-05,
      "step": 67712,
      "training_step_time": 7.81949257850647
    },
    {
      "epoch": 6.45761489868164e-05,
      "model_forward_time": 1.7445766925811768,
      "step": 67713
    },
    {
      "epoch": 6.45761489868164e-05,
      "step": 67713,
      "training_step_time": 7.786149740219116
    },
    {
      "epoch": 6.457710266113281e-05,
      "model_forward_time": 1.7367918491363525,
      "step": 67714
    },
    {
      "epoch": 6.457710266113281e-05,
      "step": 67714,
      "training_step_time": 8.590993642807007
    },
    {
      "epoch": 6.457805633544922e-05,
      "model_forward_time": 1.791102409362793,
      "step": 67715
    },
    {
      "epoch": 6.457805633544922e-05,
      "step": 67715,
      "training_step_time": 7.738141298294067
    },
    {
      "epoch": 6.457901000976562e-05,
      "model_forward_time": 1.7550451755523682,
      "step": 67716
    },
    {
      "epoch": 6.457901000976562e-05,
      "step": 67716,
      "training_step_time": 7.910714149475098
    },
    {
      "epoch": 6.457996368408203e-05,
      "model_forward_time": 1.7413816452026367,
      "step": 67717
    },
    {
      "epoch": 6.457996368408203e-05,
      "step": 67717,
      "training_step_time": 7.847450494766235
    },
    {
      "epoch": 6.458091735839844e-05,
      "model_forward_time": 1.7413911819458008,
      "step": 67718
    },
    {
      "epoch": 6.458091735839844e-05,
      "step": 67718,
      "training_step_time": 7.86753511428833
    },
    {
      "epoch": 6.458187103271484e-05,
      "model_forward_time": 2.9155020713806152,
      "step": 67719
    },
    {
      "epoch": 6.458187103271484e-05,
      "step": 67719,
      "training_step_time": 9.485234498977661
    },
    {
      "epoch": 6.458282470703125e-05,
      "grad_norm": 0.0904647707939148,
      "learning_rate": 2.58847910651605e-06,
      "loss": 0.055,
      "step": 67720
    },
    {
      "epoch": 6.458282470703125e-05,
      "model_forward_time": 1.7434449195861816,
      "step": 67720
    },
    {
      "dynamics_loss_avg": 0.05184816550463438,
      "epoch": 6.458282470703125e-05,
      "step": 67720
    },
    {
      "action_loss_avg": 0.0017134587658802047,
      "epoch": 6.458282470703125e-05,
      "step": 67720
    },
    {
      "epoch": 6.458282470703125e-05,
      "step": 67720,
      "training_step_time": 8.878727436065674
    },
    {
      "epoch": 6.458377838134766e-05,
      "model_forward_time": 1.7380683422088623,
      "step": 67721
    },
    {
      "epoch": 6.458377838134766e-05,
      "step": 67721,
      "training_step_time": 8.96312141418457
    },
    {
      "epoch": 6.458473205566406e-05,
      "model_forward_time": 1.7375669479370117,
      "step": 67722
    },
    {
      "epoch": 6.458473205566406e-05,
      "step": 67722,
      "training_step_time": 7.8140153884887695
    },
    {
      "epoch": 6.458568572998047e-05,
      "model_forward_time": 1.7521955966949463,
      "step": 67723
    },
    {
      "epoch": 6.458568572998047e-05,
      "step": 67723,
      "training_step_time": 7.715668439865112
    },
    {
      "epoch": 6.458663940429688e-05,
      "model_forward_time": 1.7386436462402344,
      "step": 67724
    },
    {
      "epoch": 6.458663940429688e-05,
      "step": 67724,
      "training_step_time": 7.710873126983643
    },
    {
      "epoch": 6.458759307861328e-05,
      "model_forward_time": 1.8783938884735107,
      "step": 67725
    },
    {
      "epoch": 6.458759307861328e-05,
      "step": 67725,
      "training_step_time": 9.457829475402832
    },
    {
      "epoch": 6.458854675292969e-05,
      "model_forward_time": 1.740880012512207,
      "step": 67726
    },
    {
      "epoch": 6.458854675292969e-05,
      "step": 67726,
      "training_step_time": 7.91745138168335
    },
    {
      "epoch": 6.45895004272461e-05,
      "model_forward_time": 1.7432911396026611,
      "step": 67727
    },
    {
      "epoch": 6.45895004272461e-05,
      "step": 67727,
      "training_step_time": 7.776778936386108
    },
    {
      "epoch": 6.45904541015625e-05,
      "model_forward_time": 2.896963357925415,
      "step": 67728
    },
    {
      "epoch": 6.45904541015625e-05,
      "step": 67728,
      "training_step_time": 7.841153621673584
    },
    {
      "epoch": 6.45914077758789e-05,
      "model_forward_time": 1.745194435119629,
      "step": 67729
    },
    {
      "epoch": 6.45914077758789e-05,
      "step": 67729,
      "training_step_time": 8.394766807556152
    },
    {
      "epoch": 6.459236145019532e-05,
      "grad_norm": 0.10223332792520523,
      "learning_rate": 2.5870307925387437e-06,
      "loss": 0.0581,
      "step": 67730
    },
    {
      "epoch": 6.459236145019532e-05,
      "model_forward_time": 1.7430579662322998,
      "step": 67730
    },
    {
      "dynamics_loss_avg": 0.0593650815077126,
      "epoch": 6.459236145019532e-05,
      "step": 67730
    },
    {
      "action_loss_avg": 0.001689934334717691,
      "epoch": 6.459236145019532e-05,
      "step": 67730
    },
    {
      "epoch": 6.459236145019532e-05,
      "step": 67730,
      "training_step_time": 8.4725980758667
    },
    {
      "epoch": 6.459331512451171e-05,
      "model_forward_time": 1.7350633144378662,
      "step": 67731
    },
    {
      "epoch": 6.459331512451171e-05,
      "step": 67731,
      "training_step_time": 8.356784105300903
    },
    {
      "epoch": 6.459426879882812e-05,
      "model_forward_time": 1.7368128299713135,
      "step": 67732
    },
    {
      "epoch": 6.459426879882812e-05,
      "step": 67732,
      "training_step_time": 7.6484215259552
    },
    {
      "epoch": 6.459522247314454e-05,
      "model_forward_time": 1.7430436611175537,
      "step": 67733
    },
    {
      "epoch": 6.459522247314454e-05,
      "step": 67733,
      "training_step_time": 7.711331844329834
    },
    {
      "epoch": 6.459617614746093e-05,
      "model_forward_time": 1.8229012489318848,
      "step": 67734
    },
    {
      "epoch": 6.459617614746093e-05,
      "step": 67734,
      "training_step_time": 8.896492958068848
    },
    {
      "epoch": 6.459712982177734e-05,
      "model_forward_time": 1.7312214374542236,
      "step": 67735
    },
    {
      "epoch": 6.459712982177734e-05,
      "step": 67735,
      "training_step_time": 9.830609321594238
    },
    {
      "epoch": 6.459808349609375e-05,
      "model_forward_time": 1.38205885887146,
      "step": 67736
    },
    {
      "epoch": 6.459808349609375e-05,
      "step": 67736,
      "training_step_time": 6.880257606506348
    },
    {
      "epoch": 6.459903717041015e-05,
      "model_forward_time": 1.7420103549957275,
      "step": 67737
    },
    {
      "epoch": 6.459903717041015e-05,
      "step": 67737,
      "training_step_time": 6.875525236129761
    },
    {
      "epoch": 6.459999084472656e-05,
      "model_forward_time": 1.7552127838134766,
      "step": 67738
    },
    {
      "epoch": 6.459999084472656e-05,
      "step": 67738,
      "training_step_time": 6.890506029129028
    },
    {
      "epoch": 6.460094451904297e-05,
      "model_forward_time": 1.745326280593872,
      "step": 67739
    },
    {
      "epoch": 6.460094451904297e-05,
      "step": 67739,
      "training_step_time": 6.863704204559326
    },
    {
      "epoch": 6.460189819335937e-05,
      "grad_norm": 0.10734806209802628,
      "learning_rate": 2.5855827424401252e-06,
      "loss": 0.0576,
      "step": 67740
    },
    {
      "epoch": 6.460189819335937e-05,
      "model_forward_time": 1.73514986038208,
      "step": 67740
    },
    {
      "dynamics_loss_avg": 0.059097870439291,
      "epoch": 6.460189819335937e-05,
      "step": 67740
    },
    {
      "action_loss_avg": 0.0026722019072622063,
      "epoch": 6.460189819335937e-05,
      "step": 67740
    },
    {
      "epoch": 6.460189819335937e-05,
      "step": 67740,
      "training_step_time": 6.901917934417725
    },
    {
      "epoch": 6.460285186767578e-05,
      "model_forward_time": 1.738111972808838,
      "step": 67741
    },
    {
      "epoch": 6.460285186767578e-05,
      "step": 67741,
      "training_step_time": 6.827454566955566
    },
    {
      "epoch": 6.460380554199219e-05,
      "model_forward_time": 1.8040685653686523,
      "step": 67742
    },
    {
      "epoch": 6.460380554199219e-05,
      "step": 67742,
      "training_step_time": 6.855120658874512
    },
    {
      "epoch": 6.460475921630859e-05,
      "model_forward_time": 1.7454707622528076,
      "step": 67743
    },
    {
      "epoch": 6.460475921630859e-05,
      "step": 67743,
      "training_step_time": 6.8914830684661865
    },
    {
      "epoch": 6.4605712890625e-05,
      "model_forward_time": 2.981928586959839,
      "step": 67744
    },
    {
      "epoch": 6.4605712890625e-05,
      "step": 67744,
      "training_step_time": 7.943291902542114
    },
    {
      "epoch": 6.460666656494141e-05,
      "model_forward_time": 1.736022710800171,
      "step": 67745
    },
    {
      "epoch": 6.460666656494141e-05,
      "step": 67745,
      "training_step_time": 7.902482748031616
    },
    {
      "epoch": 6.460762023925781e-05,
      "model_forward_time": 1.7378792762756348,
      "step": 67746
    },
    {
      "epoch": 6.460762023925781e-05,
      "step": 67746,
      "training_step_time": 6.88288426399231
    },
    {
      "epoch": 6.460857391357422e-05,
      "model_forward_time": 1.7445533275604248,
      "step": 67747
    },
    {
      "epoch": 6.460857391357422e-05,
      "step": 67747,
      "training_step_time": 6.8697826862335205
    },
    {
      "epoch": 6.460952758789063e-05,
      "model_forward_time": 1.7463493347167969,
      "step": 67748
    },
    {
      "epoch": 6.460952758789063e-05,
      "step": 67748,
      "training_step_time": 6.86221170425415
    },
    {
      "epoch": 6.461048126220703e-05,
      "model_forward_time": 1.7672615051269531,
      "step": 67749
    },
    {
      "epoch": 6.461048126220703e-05,
      "step": 67749,
      "training_step_time": 6.845505475997925
    },
    {
      "epoch": 6.461143493652344e-05,
      "grad_norm": 0.10780651867389679,
      "learning_rate": 2.5841349563785455e-06,
      "loss": 0.0596,
      "step": 67750
    },
    {
      "epoch": 6.461143493652344e-05,
      "model_forward_time": 1.7497782707214355,
      "step": 67750
    },
    {
      "dynamics_loss_avg": 0.04425700157880783,
      "epoch": 6.461143493652344e-05,
      "step": 67750
    },
    {
      "action_loss_avg": 0.001731400369317271,
      "epoch": 6.461143493652344e-05,
      "step": 67750
    },
    {
      "epoch": 6.461143493652344e-05,
      "step": 67750,
      "training_step_time": 6.8709118366241455
    },
    {
      "epoch": 6.461238861083985e-05,
      "model_forward_time": 1.7431349754333496,
      "step": 67751
    },
    {
      "epoch": 6.461238861083985e-05,
      "step": 67751,
      "training_step_time": 6.837196350097656
    },
    {
      "epoch": 6.461334228515625e-05,
      "model_forward_time": 1.746368408203125,
      "step": 67752
    },
    {
      "epoch": 6.461334228515625e-05,
      "step": 67752,
      "training_step_time": 6.845429420471191
    },
    {
      "epoch": 6.461429595947266e-05,
      "model_forward_time": 3.0359270572662354,
      "step": 67753
    },
    {
      "epoch": 6.461429595947266e-05,
      "step": 67753,
      "training_step_time": 7.9710612297058105
    },
    {
      "epoch": 6.461524963378907e-05,
      "model_forward_time": 1.734943151473999,
      "step": 67754
    },
    {
      "epoch": 6.461524963378907e-05,
      "step": 67754,
      "training_step_time": 7.941669464111328
    },
    {
      "epoch": 6.461620330810546e-05,
      "model_forward_time": 1.7439851760864258,
      "step": 67755
    },
    {
      "epoch": 6.461620330810546e-05,
      "step": 67755,
      "training_step_time": 7.789735317230225
    },
    {
      "epoch": 6.461715698242188e-05,
      "model_forward_time": 1.387233018875122,
      "step": 67756
    },
    {
      "epoch": 6.461715698242188e-05,
      "step": 67756,
      "training_step_time": 6.873351097106934
    },
    {
      "epoch": 6.461811065673829e-05,
      "model_forward_time": 1.7096340656280518,
      "step": 67757
    },
    {
      "epoch": 6.461811065673829e-05,
      "step": 67757,
      "training_step_time": 6.818588972091675
    },
    {
      "epoch": 6.461906433105468e-05,
      "model_forward_time": 1.7190592288970947,
      "step": 67758
    },
    {
      "epoch": 6.461906433105468e-05,
      "step": 67758,
      "training_step_time": 6.831369161605835
    },
    {
      "epoch": 6.46200180053711e-05,
      "model_forward_time": 1.7209877967834473,
      "step": 67759
    },
    {
      "epoch": 6.46200180053711e-05,
      "step": 67759,
      "training_step_time": 6.840889930725098
    },
    {
      "epoch": 6.46209716796875e-05,
      "grad_norm": 0.09400800615549088,
      "learning_rate": 2.5826874345123353e-06,
      "loss": 0.0578,
      "step": 67760
    },
    {
      "epoch": 6.46209716796875e-05,
      "model_forward_time": 1.710981845855713,
      "step": 67760
    },
    {
      "dynamics_loss_avg": 0.042228436656296255,
      "epoch": 6.46209716796875e-05,
      "step": 67760
    },
    {
      "action_loss_avg": 0.0018431556876748801,
      "epoch": 6.46209716796875e-05,
      "step": 67760
    },
    {
      "epoch": 6.46209716796875e-05,
      "step": 67760,
      "training_step_time": 6.866437196731567
    },
    {
      "epoch": 6.46219253540039e-05,
      "model_forward_time": 1.7215824127197266,
      "step": 67761
    },
    {
      "epoch": 6.46219253540039e-05,
      "step": 67761,
      "training_step_time": 6.889100790023804
    },
    {
      "epoch": 6.462287902832031e-05,
      "model_forward_time": 2.934044361114502,
      "step": 67762
    },
    {
      "epoch": 6.462287902832031e-05,
      "step": 67762,
      "training_step_time": 7.88129734992981
    },
    {
      "epoch": 6.462383270263672e-05,
      "model_forward_time": 1.7267675399780273,
      "step": 67763
    },
    {
      "epoch": 6.462383270263672e-05,
      "step": 67763,
      "training_step_time": 9.747868061065674
    },
    {
      "epoch": 6.462478637695312e-05,
      "model_forward_time": 1.7265470027923584,
      "step": 67764
    },
    {
      "epoch": 6.462478637695312e-05,
      "step": 67764,
      "training_step_time": 7.838721752166748
    },
    {
      "epoch": 6.462574005126953e-05,
      "model_forward_time": 1.7074556350708008,
      "step": 67765
    },
    {
      "epoch": 6.462574005126953e-05,
      "step": 67765,
      "training_step_time": 6.855207681655884
    },
    {
      "epoch": 6.462669372558594e-05,
      "model_forward_time": 1.7236385345458984,
      "step": 67766
    },
    {
      "epoch": 6.462669372558594e-05,
      "step": 67766,
      "training_step_time": 6.891084671020508
    },
    {
      "epoch": 6.462764739990234e-05,
      "model_forward_time": 1.724858283996582,
      "step": 67767
    },
    {
      "epoch": 6.462764739990234e-05,
      "step": 67767,
      "training_step_time": 6.856609106063843
    },
    {
      "epoch": 6.462860107421875e-05,
      "model_forward_time": 1.7356252670288086,
      "step": 67768
    },
    {
      "epoch": 6.462860107421875e-05,
      "step": 67768,
      "training_step_time": 6.843710422515869
    },
    {
      "epoch": 6.462955474853516e-05,
      "model_forward_time": 1.7301459312438965,
      "step": 67769
    },
    {
      "epoch": 6.462955474853516e-05,
      "step": 67769,
      "training_step_time": 6.838312864303589
    },
    {
      "epoch": 6.463050842285156e-05,
      "grad_norm": 0.09424153715372086,
      "learning_rate": 2.581240176999792e-06,
      "loss": 0.0582,
      "step": 67770
    },
    {
      "epoch": 6.463050842285156e-05,
      "model_forward_time": 1.7197558879852295,
      "step": 67770
    },
    {
      "dynamics_loss_avg": 0.06092100301757455,
      "epoch": 6.463050842285156e-05,
      "step": 67770
    },
    {
      "action_loss_avg": 0.002318203798495233,
      "epoch": 6.463050842285156e-05,
      "step": 67770
    },
    {
      "epoch": 6.463050842285156e-05,
      "step": 67770,
      "training_step_time": 6.829551935195923
    },
    {
      "epoch": 6.463146209716797e-05,
      "model_forward_time": 2.8993453979492188,
      "step": 67771
    },
    {
      "epoch": 6.463146209716797e-05,
      "step": 67771,
      "training_step_time": 8.858663558959961
    },
    {
      "epoch": 6.463241577148438e-05,
      "model_forward_time": 1.7293593883514404,
      "step": 67772
    },
    {
      "epoch": 6.463241577148438e-05,
      "step": 67772,
      "training_step_time": 8.434939861297607
    },
    {
      "epoch": 6.463336944580078e-05,
      "model_forward_time": 1.726289987564087,
      "step": 67773
    },
    {
      "epoch": 6.463336944580078e-05,
      "step": 67773,
      "training_step_time": 7.802250623703003
    },
    {
      "epoch": 6.463432312011719e-05,
      "model_forward_time": 1.7163197994232178,
      "step": 67774
    },
    {
      "epoch": 6.463432312011719e-05,
      "step": 67774,
      "training_step_time": 6.882735252380371
    },
    {
      "epoch": 6.46352767944336e-05,
      "model_forward_time": 1.7190353870391846,
      "step": 67775
    },
    {
      "epoch": 6.46352767944336e-05,
      "step": 67775,
      "training_step_time": 6.831909418106079
    },
    {
      "epoch": 6.463623046875e-05,
      "model_forward_time": 1.7258288860321045,
      "step": 67776
    },
    {
      "epoch": 6.463623046875e-05,
      "step": 67776,
      "training_step_time": 6.813476800918579
    },
    {
      "epoch": 6.463718414306641e-05,
      "model_forward_time": 1.7254760265350342,
      "step": 67777
    },
    {
      "epoch": 6.463718414306641e-05,
      "step": 67777,
      "training_step_time": 6.831765651702881
    },
    {
      "epoch": 6.463813781738282e-05,
      "model_forward_time": 1.7191693782806396,
      "step": 67778
    },
    {
      "epoch": 6.463813781738282e-05,
      "step": 67778,
      "training_step_time": 6.958723306655884
    },
    {
      "epoch": 6.463909149169922e-05,
      "model_forward_time": 1.3413324356079102,
      "step": 67779
    },
    {
      "epoch": 6.463909149169922e-05,
      "step": 67779,
      "training_step_time": 6.820738315582275
    },
    {
      "epoch": 6.464004516601563e-05,
      "grad_norm": 0.10853651165962219,
      "learning_rate": 2.579793183999189e-06,
      "loss": 0.0557,
      "step": 67780
    },
    {
      "epoch": 6.464004516601563e-05,
      "model_forward_time": 2.8705859184265137,
      "step": 67780
    },
    {
      "dynamics_loss_avg": 0.04226490203291178,
      "epoch": 6.464004516601563e-05,
      "step": 67780
    },
    {
      "action_loss_avg": 0.0018880033254390582,
      "epoch": 6.464004516601563e-05,
      "step": 67780
    },
    {
      "epoch": 6.464004516601563e-05,
      "step": 67780,
      "training_step_time": 7.815098524093628
    },
    {
      "epoch": 6.464099884033204e-05,
      "model_forward_time": 1.7554666996002197,
      "step": 67781
    },
    {
      "epoch": 6.464099884033204e-05,
      "step": 67781,
      "training_step_time": 7.898155450820923
    },
    {
      "epoch": 6.464195251464843e-05,
      "model_forward_time": 1.7310302257537842,
      "step": 67782
    },
    {
      "epoch": 6.464195251464843e-05,
      "step": 67782,
      "training_step_time": 7.914167165756226
    },
    {
      "epoch": 6.464290618896484e-05,
      "model_forward_time": 1.7195119857788086,
      "step": 67783
    },
    {
      "epoch": 6.464290618896484e-05,
      "step": 67783,
      "training_step_time": 6.880960464477539
    },
    {
      "epoch": 6.464385986328126e-05,
      "model_forward_time": 1.741062879562378,
      "step": 67784
    },
    {
      "epoch": 6.464385986328126e-05,
      "step": 67784,
      "training_step_time": 6.897924423217773
    },
    {
      "epoch": 6.464481353759765e-05,
      "model_forward_time": 1.7332987785339355,
      "step": 67785
    },
    {
      "epoch": 6.464481353759765e-05,
      "step": 67785,
      "training_step_time": 6.902886152267456
    },
    {
      "epoch": 6.464576721191406e-05,
      "model_forward_time": 1.729586124420166,
      "step": 67786
    },
    {
      "epoch": 6.464576721191406e-05,
      "step": 67786,
      "training_step_time": 6.8907599449157715
    },
    {
      "epoch": 6.464672088623047e-05,
      "model_forward_time": 1.7165765762329102,
      "step": 67787
    },
    {
      "epoch": 6.464672088623047e-05,
      "step": 67787,
      "training_step_time": 6.939422845840454
    },
    {
      "epoch": 6.464767456054687e-05,
      "model_forward_time": 1.7288565635681152,
      "step": 67788
    },
    {
      "epoch": 6.464767456054687e-05,
      "step": 67788,
      "training_step_time": 6.906429052352905
    },
    {
      "epoch": 6.464862823486328e-05,
      "model_forward_time": 1.733577013015747,
      "step": 67789
    },
    {
      "epoch": 6.464862823486328e-05,
      "step": 67789,
      "training_step_time": 6.92496132850647
    },
    {
      "epoch": 6.46495819091797e-05,
      "grad_norm": 0.12757298350334167,
      "learning_rate": 2.5783464556687626e-06,
      "loss": 0.0583,
      "step": 67790
    },
    {
      "epoch": 6.46495819091797e-05,
      "model_forward_time": 2.9259541034698486,
      "step": 67790
    },
    {
      "dynamics_loss_avg": 0.05976740047335625,
      "epoch": 6.46495819091797e-05,
      "step": 67790
    },
    {
      "action_loss_avg": 0.0016660586697980762,
      "epoch": 6.46495819091797e-05,
      "step": 67790
    },
    {
      "epoch": 6.46495819091797e-05,
      "step": 67790,
      "training_step_time": 7.909257411956787
    },
    {
      "epoch": 6.465053558349609e-05,
      "model_forward_time": 1.7279274463653564,
      "step": 67791
    },
    {
      "epoch": 6.465053558349609e-05,
      "step": 67791,
      "training_step_time": 7.860126972198486
    },
    {
      "epoch": 6.46514892578125e-05,
      "model_forward_time": 1.7227146625518799,
      "step": 67792
    },
    {
      "epoch": 6.46514892578125e-05,
      "step": 67792,
      "training_step_time": 7.702715635299683
    },
    {
      "epoch": 6.465244293212891e-05,
      "model_forward_time": 1.7259538173675537,
      "step": 67793
    },
    {
      "epoch": 6.465244293212891e-05,
      "step": 67793,
      "training_step_time": 6.948224306106567
    },
    {
      "epoch": 6.465339660644531e-05,
      "model_forward_time": 1.7262132167816162,
      "step": 67794
    },
    {
      "epoch": 6.465339660644531e-05,
      "step": 67794,
      "training_step_time": 6.867470979690552
    },
    {
      "epoch": 6.465435028076172e-05,
      "model_forward_time": 1.7325589656829834,
      "step": 67795
    },
    {
      "epoch": 6.465435028076172e-05,
      "step": 67795,
      "training_step_time": 6.837859153747559
    },
    {
      "epoch": 6.465530395507813e-05,
      "model_forward_time": 1.7249205112457275,
      "step": 67796
    },
    {
      "epoch": 6.465530395507813e-05,
      "step": 67796,
      "training_step_time": 6.920223951339722
    },
    {
      "epoch": 6.465625762939453e-05,
      "model_forward_time": 1.726144552230835,
      "step": 67797
    },
    {
      "epoch": 6.465625762939453e-05,
      "step": 67797,
      "training_step_time": 6.686352491378784
    },
    {
      "epoch": 6.465721130371094e-05,
      "model_forward_time": 1.7217257022857666,
      "step": 67798
    },
    {
      "epoch": 6.465721130371094e-05,
      "step": 67798,
      "training_step_time": 6.8779213428497314
    },
    {
      "epoch": 6.465816497802735e-05,
      "model_forward_time": 2.9473876953125,
      "step": 67799
    },
    {
      "epoch": 6.465816497802735e-05,
      "step": 67799,
      "training_step_time": 7.893025159835815
    },
    {
      "epoch": 6.465911865234375e-05,
      "grad_norm": 0.10503090918064117,
      "learning_rate": 2.5768999921667294e-06,
      "loss": 0.0545,
      "step": 67800
    },
    {
      "epoch": 6.465911865234375e-05,
      "model_forward_time": 1.7247545719146729,
      "step": 67800
    },
    {
      "dynamics_loss_avg": 0.0433656569570303,
      "epoch": 6.465911865234375e-05,
      "step": 67800
    },
    {
      "action_loss_avg": 0.0014742088998900727,
      "epoch": 6.465911865234375e-05,
      "step": 67800
    },
    {
      "epoch": 6.465911865234375e-05,
      "step": 67800,
      "training_step_time": 7.969107151031494
    },
    {
      "epoch": 6.466007232666016e-05,
      "model_forward_time": 1.7276194095611572,
      "step": 67801
    },
    {
      "epoch": 6.466007232666016e-05,
      "step": 67801,
      "training_step_time": 7.835941314697266
    },
    {
      "epoch": 6.466102600097657e-05,
      "model_forward_time": 1.7247421741485596,
      "step": 67802
    },
    {
      "epoch": 6.466102600097657e-05,
      "step": 67802,
      "training_step_time": 6.872675657272339
    },
    {
      "epoch": 6.466197967529297e-05,
      "model_forward_time": 1.7211267948150635,
      "step": 67803
    },
    {
      "epoch": 6.466197967529297e-05,
      "step": 67803,
      "training_step_time": 6.913361310958862
    },
    {
      "epoch": 6.466293334960938e-05,
      "model_forward_time": 1.7287425994873047,
      "step": 67804
    },
    {
      "epoch": 6.466293334960938e-05,
      "step": 67804,
      "training_step_time": 6.877853870391846
    },
    {
      "epoch": 6.466388702392579e-05,
      "model_forward_time": 1.72585129737854,
      "step": 67805
    },
    {
      "epoch": 6.466388702392579e-05,
      "step": 67805,
      "training_step_time": 6.848811149597168
    },
    {
      "epoch": 6.466484069824219e-05,
      "model_forward_time": 1.7241568565368652,
      "step": 67806
    },
    {
      "epoch": 6.466484069824219e-05,
      "step": 67806,
      "training_step_time": 6.927306890487671
    },
    {
      "epoch": 6.46657943725586e-05,
      "model_forward_time": 1.7231266498565674,
      "step": 67807
    },
    {
      "epoch": 6.46657943725586e-05,
      "step": 67807,
      "training_step_time": 6.918718338012695
    },
    {
      "epoch": 6.4666748046875e-05,
      "model_forward_time": 2.8907947540283203,
      "step": 67808
    },
    {
      "epoch": 6.4666748046875e-05,
      "step": 67808,
      "training_step_time": 7.808423280715942
    },
    {
      "epoch": 6.46677017211914e-05,
      "model_forward_time": 1.761460781097412,
      "step": 67809
    },
    {
      "epoch": 6.46677017211914e-05,
      "step": 67809,
      "training_step_time": 10.452829599380493
    },
    {
      "epoch": 6.466865539550781e-05,
      "grad_norm": 0.1244988813996315,
      "learning_rate": 2.575453793651267e-06,
      "loss": 0.0523,
      "step": 67810
    },
    {
      "epoch": 6.466865539550781e-05,
      "model_forward_time": 1.723761796951294,
      "step": 67810
    },
    {
      "dynamics_loss_avg": 0.044858713448047635,
      "epoch": 6.466865539550781e-05,
      "step": 67810
    },
    {
      "action_loss_avg": 0.0014468737557763235,
      "epoch": 6.466865539550781e-05,
      "step": 67810
    },
    {
      "epoch": 6.466865539550781e-05,
      "step": 67810,
      "training_step_time": 8.565810680389404
    },
    {
      "epoch": 6.466960906982423e-05,
      "model_forward_time": 1.7249095439910889,
      "step": 67811
    },
    {
      "epoch": 6.466960906982423e-05,
      "step": 67811,
      "training_step_time": 6.896812438964844
    },
    {
      "epoch": 6.467056274414062e-05,
      "model_forward_time": 1.7286646366119385,
      "step": 67812
    },
    {
      "epoch": 6.467056274414062e-05,
      "step": 67812,
      "training_step_time": 6.9605393409729
    },
    {
      "epoch": 6.467151641845703e-05,
      "model_forward_time": 1.3628361225128174,
      "step": 67813
    },
    {
      "epoch": 6.467151641845703e-05,
      "step": 67813,
      "training_step_time": 6.9433207511901855
    },
    {
      "epoch": 6.467247009277343e-05,
      "model_forward_time": 1.7260329723358154,
      "step": 67814
    },
    {
      "epoch": 6.467247009277343e-05,
      "step": 67814,
      "training_step_time": 6.974087476730347
    },
    {
      "epoch": 6.467342376708984e-05,
      "model_forward_time": 1.7213592529296875,
      "step": 67815
    },
    {
      "epoch": 6.467342376708984e-05,
      "step": 67815,
      "training_step_time": 6.930025577545166
    },
    {
      "epoch": 6.467437744140625e-05,
      "model_forward_time": 1.7238373756408691,
      "step": 67816
    },
    {
      "epoch": 6.467437744140625e-05,
      "step": 67816,
      "training_step_time": 6.9438464641571045
    },
    {
      "epoch": 6.467533111572265e-05,
      "model_forward_time": 2.885741949081421,
      "step": 67817
    },
    {
      "epoch": 6.467533111572265e-05,
      "step": 67817,
      "training_step_time": 7.978997230529785
    },
    {
      "epoch": 6.467628479003906e-05,
      "model_forward_time": 1.3717679977416992,
      "step": 67818
    },
    {
      "epoch": 6.467628479003906e-05,
      "step": 67818,
      "training_step_time": 11.88603687286377
    },
    {
      "epoch": 6.467723846435547e-05,
      "model_forward_time": 1.7212653160095215,
      "step": 67819
    },
    {
      "epoch": 6.467723846435547e-05,
      "step": 67819,
      "training_step_time": 7.902963399887085
    },
    {
      "epoch": 6.467819213867187e-05,
      "grad_norm": 0.0835881158709526,
      "learning_rate": 2.574007860280535e-06,
      "loss": 0.053,
      "step": 67820
    },
    {
      "epoch": 6.467819213867187e-05,
      "model_forward_time": 1.7287101745605469,
      "step": 67820
    },
    {
      "dynamics_loss_avg": 0.059432696737349036,
      "epoch": 6.467819213867187e-05,
      "step": 67820
    },
    {
      "action_loss_avg": 0.002075381274335086,
      "epoch": 6.467819213867187e-05,
      "step": 67820
    },
    {
      "epoch": 6.467819213867187e-05,
      "step": 67820,
      "training_step_time": 7.453194618225098
    },
    {
      "epoch": 6.467914581298828e-05,
      "model_forward_time": 1.7260336875915527,
      "step": 67821
    },
    {
      "epoch": 6.467914581298828e-05,
      "step": 67821,
      "training_step_time": 6.93475341796875
    },
    {
      "epoch": 6.468009948730469e-05,
      "model_forward_time": 1.7150306701660156,
      "step": 67822
    },
    {
      "epoch": 6.468009948730469e-05,
      "step": 67822,
      "training_step_time": 6.938629150390625
    },
    {
      "epoch": 6.468105316162109e-05,
      "model_forward_time": 1.7243068218231201,
      "step": 67823
    },
    {
      "epoch": 6.468105316162109e-05,
      "step": 67823,
      "training_step_time": 6.955005407333374
    },
    {
      "epoch": 6.46820068359375e-05,
      "model_forward_time": 1.350679874420166,
      "step": 67824
    },
    {
      "epoch": 6.46820068359375e-05,
      "step": 67824,
      "training_step_time": 6.95088005065918
    },
    {
      "epoch": 6.468296051025391e-05,
      "model_forward_time": 1.725313425064087,
      "step": 67825
    },
    {
      "epoch": 6.468296051025391e-05,
      "step": 67825,
      "training_step_time": 6.9683310985565186
    },
    {
      "epoch": 6.46839141845703e-05,
      "model_forward_time": 1.72422456741333,
      "step": 67826
    },
    {
      "epoch": 6.46839141845703e-05,
      "step": 67826,
      "training_step_time": 7.941709041595459
    },
    {
      "epoch": 6.468486785888672e-05,
      "model_forward_time": 2.873237133026123,
      "step": 67827
    },
    {
      "epoch": 6.468486785888672e-05,
      "step": 67827,
      "training_step_time": 11.52480697631836
    },
    {
      "epoch": 6.468582153320313e-05,
      "model_forward_time": 1.7247252464294434,
      "step": 67828
    },
    {
      "epoch": 6.468582153320313e-05,
      "step": 67828,
      "training_step_time": 10.868741273880005
    },
    {
      "epoch": 6.468677520751953e-05,
      "model_forward_time": 1.7261857986450195,
      "step": 67829
    },
    {
      "epoch": 6.468677520751953e-05,
      "step": 67829,
      "training_step_time": 7.780521869659424
    },
    {
      "epoch": 6.468772888183594e-05,
      "grad_norm": 0.1410679966211319,
      "learning_rate": 2.572562192212652e-06,
      "loss": 0.0585,
      "step": 67830
    },
    {
      "epoch": 6.468772888183594e-05,
      "model_forward_time": 1.7216222286224365,
      "step": 67830
    },
    {
      "dynamics_loss_avg": 0.07286706613376737,
      "epoch": 6.468772888183594e-05,
      "step": 67830
    },
    {
      "action_loss_avg": 0.0022736103710485622,
      "epoch": 6.468772888183594e-05,
      "step": 67830
    },
    {
      "epoch": 6.468772888183594e-05,
      "step": 67830,
      "training_step_time": 6.899045467376709
    },
    {
      "epoch": 6.468868255615235e-05,
      "model_forward_time": 1.7243902683258057,
      "step": 67831
    },
    {
      "epoch": 6.468868255615235e-05,
      "step": 67831,
      "training_step_time": 6.9148805141448975
    },
    {
      "epoch": 6.468963623046874e-05,
      "model_forward_time": 1.719684362411499,
      "step": 67832
    },
    {
      "epoch": 6.468963623046874e-05,
      "step": 67832,
      "training_step_time": 6.882444620132446
    },
    {
      "epoch": 6.469058990478515e-05,
      "model_forward_time": 1.7111220359802246,
      "step": 67833
    },
    {
      "epoch": 6.469058990478515e-05,
      "step": 67833,
      "training_step_time": 6.8945934772491455
    },
    {
      "epoch": 6.469154357910157e-05,
      "model_forward_time": 1.7199954986572266,
      "step": 67834
    },
    {
      "epoch": 6.469154357910157e-05,
      "step": 67834,
      "training_step_time": 6.95330023765564
    },
    {
      "epoch": 6.469249725341796e-05,
      "model_forward_time": 1.725456953048706,
      "step": 67835
    },
    {
      "epoch": 6.469249725341796e-05,
      "step": 67835,
      "training_step_time": 7.888343095779419
    },
    {
      "epoch": 6.469345092773437e-05,
      "model_forward_time": 2.8894846439361572,
      "step": 67836
    },
    {
      "epoch": 6.469345092773437e-05,
      "step": 67836,
      "training_step_time": 9.907345294952393
    },
    {
      "epoch": 6.469440460205078e-05,
      "model_forward_time": 1.7215373516082764,
      "step": 67837
    },
    {
      "epoch": 6.469440460205078e-05,
      "step": 67837,
      "training_step_time": 9.556936025619507
    },
    {
      "epoch": 6.469535827636718e-05,
      "model_forward_time": 1.724696397781372,
      "step": 67838
    },
    {
      "epoch": 6.469535827636718e-05,
      "step": 67838,
      "training_step_time": 7.90078330039978
    },
    {
      "epoch": 6.469631195068359e-05,
      "model_forward_time": 1.7300846576690674,
      "step": 67839
    },
    {
      "epoch": 6.469631195068359e-05,
      "step": 67839,
      "training_step_time": 6.938040733337402
    },
    {
      "epoch": 6.4697265625e-05,
      "grad_norm": 0.12773862481117249,
      "learning_rate": 2.5711167896057222e-06,
      "loss": 0.0569,
      "step": 67840
    },
    {
      "epoch": 6.4697265625e-05,
      "model_forward_time": 1.7221932411193848,
      "step": 67840
    },
    {
      "dynamics_loss_avg": 0.06479337997734547,
      "epoch": 6.4697265625e-05,
      "step": 67840
    },
    {
      "action_loss_avg": 0.0021046126435976475,
      "epoch": 6.4697265625e-05,
      "step": 67840
    },
    {
      "epoch": 6.4697265625e-05,
      "step": 67840,
      "training_step_time": 6.927856206893921
    },
    {
      "epoch": 6.46982192993164e-05,
      "model_forward_time": 1.730520486831665,
      "step": 67841
    },
    {
      "epoch": 6.46982192993164e-05,
      "step": 67841,
      "training_step_time": 6.981076955795288
    },
    {
      "epoch": 6.469917297363281e-05,
      "model_forward_time": 1.7274577617645264,
      "step": 67842
    },
    {
      "epoch": 6.469917297363281e-05,
      "step": 67842,
      "training_step_time": 6.967554330825806
    },
    {
      "epoch": 6.470012664794922e-05,
      "model_forward_time": 1.7221441268920898,
      "step": 67843
    },
    {
      "epoch": 6.470012664794922e-05,
      "step": 67843,
      "training_step_time": 7.006804466247559
    },
    {
      "epoch": 6.470108032226562e-05,
      "model_forward_time": 1.7238094806671143,
      "step": 67844
    },
    {
      "epoch": 6.470108032226562e-05,
      "step": 67844,
      "training_step_time": 7.878728151321411
    },
    {
      "epoch": 6.470203399658203e-05,
      "model_forward_time": 2.894911527633667,
      "step": 67845
    },
    {
      "epoch": 6.470203399658203e-05,
      "step": 67845,
      "training_step_time": 7.924137353897095
    },
    {
      "epoch": 6.470298767089844e-05,
      "model_forward_time": 1.7320387363433838,
      "step": 67846
    },
    {
      "epoch": 6.470298767089844e-05,
      "step": 67846,
      "training_step_time": 9.287019968032837
    },
    {
      "epoch": 6.470394134521484e-05,
      "model_forward_time": 1.7320499420166016,
      "step": 67847
    },
    {
      "epoch": 6.470394134521484e-05,
      "step": 67847,
      "training_step_time": 7.932379484176636
    },
    {
      "epoch": 6.470489501953125e-05,
      "model_forward_time": 1.727726697921753,
      "step": 67848
    },
    {
      "epoch": 6.470489501953125e-05,
      "step": 67848,
      "training_step_time": 6.950723171234131
    },
    {
      "epoch": 6.470584869384766e-05,
      "model_forward_time": 1.723478078842163,
      "step": 67849
    },
    {
      "epoch": 6.470584869384766e-05,
      "step": 67849,
      "training_step_time": 6.969825744628906
    },
    {
      "epoch": 6.470680236816406e-05,
      "grad_norm": 0.08067543059587479,
      "learning_rate": 2.5696716526178057e-06,
      "loss": 0.057,
      "step": 67850
    },
    {
      "epoch": 6.470680236816406e-05,
      "model_forward_time": 1.7331523895263672,
      "step": 67850
    },
    {
      "dynamics_loss_avg": 0.06130213811993599,
      "epoch": 6.470680236816406e-05,
      "step": 67850
    },
    {
      "action_loss_avg": 0.0018933115032268688,
      "epoch": 6.470680236816406e-05,
      "step": 67850
    },
    {
      "epoch": 6.470680236816406e-05,
      "step": 67850,
      "training_step_time": 6.827205181121826
    },
    {
      "epoch": 6.470775604248047e-05,
      "model_forward_time": 1.7279176712036133,
      "step": 67851
    },
    {
      "epoch": 6.470775604248047e-05,
      "step": 67851,
      "training_step_time": 7.0032148361206055
    },
    {
      "epoch": 6.470870971679688e-05,
      "model_forward_time": 1.7188808917999268,
      "step": 67852
    },
    {
      "epoch": 6.470870971679688e-05,
      "step": 67852,
      "training_step_time": 7.0838611125946045
    },
    {
      "epoch": 6.470966339111328e-05,
      "model_forward_time": 1.7245113849639893,
      "step": 67853
    },
    {
      "epoch": 6.470966339111328e-05,
      "step": 67853,
      "training_step_time": 7.01991081237793
    },
    {
      "epoch": 6.471061706542969e-05,
      "model_forward_time": 2.4819304943084717,
      "step": 67854
    },
    {
      "epoch": 6.471061706542969e-05,
      "step": 67854,
      "training_step_time": 9.869895219802856
    },
    {
      "epoch": 6.47115707397461e-05,
      "model_forward_time": 1.716508150100708,
      "step": 67855
    },
    {
      "epoch": 6.47115707397461e-05,
      "step": 67855,
      "training_step_time": 10.110110521316528
    },
    {
      "epoch": 6.47125244140625e-05,
      "model_forward_time": 1.7314088344573975,
      "step": 67856
    },
    {
      "epoch": 6.47125244140625e-05,
      "step": 67856,
      "training_step_time": 8.51424765586853
    },
    {
      "epoch": 6.47134780883789e-05,
      "model_forward_time": 1.7297544479370117,
      "step": 67857
    },
    {
      "epoch": 6.47134780883789e-05,
      "step": 67857,
      "training_step_time": 7.249974727630615
    },
    {
      "epoch": 6.471443176269532e-05,
      "model_forward_time": 1.7287819385528564,
      "step": 67858
    },
    {
      "epoch": 6.471443176269532e-05,
      "step": 67858,
      "training_step_time": 6.824135780334473
    },
    {
      "epoch": 6.471538543701171e-05,
      "model_forward_time": 1.731050729751587,
      "step": 67859
    },
    {
      "epoch": 6.471538543701171e-05,
      "step": 67859,
      "training_step_time": 6.770776748657227
    },
    {
      "epoch": 6.471633911132812e-05,
      "grad_norm": 0.09614132344722748,
      "learning_rate": 2.5682267814069463e-06,
      "loss": 0.0552,
      "step": 67860
    },
    {
      "epoch": 6.471633911132812e-05,
      "model_forward_time": 1.731978178024292,
      "step": 67860
    },
    {
      "dynamics_loss_avg": 0.04991662390530109,
      "epoch": 6.471633911132812e-05,
      "step": 67860
    },
    {
      "action_loss_avg": 0.001280693692388013,
      "epoch": 6.471633911132812e-05,
      "step": 67860
    },
    {
      "epoch": 6.471633911132812e-05,
      "step": 67860,
      "training_step_time": 6.798043489456177
    },
    {
      "epoch": 6.471729278564454e-05,
      "model_forward_time": 1.7270290851593018,
      "step": 67861
    },
    {
      "epoch": 6.471729278564454e-05,
      "step": 67861,
      "training_step_time": 6.865138530731201
    },
    {
      "epoch": 6.471824645996093e-05,
      "model_forward_time": 1.719879388809204,
      "step": 67862
    },
    {
      "epoch": 6.471824645996093e-05,
      "step": 67862,
      "training_step_time": 6.747114181518555
    },
    {
      "epoch": 6.471920013427734e-05,
      "model_forward_time": 1.7615869045257568,
      "step": 67863
    },
    {
      "epoch": 6.471920013427734e-05,
      "step": 67863,
      "training_step_time": 9.270490884780884
    },
    {
      "epoch": 6.472015380859375e-05,
      "model_forward_time": 2.9616312980651855,
      "step": 67864
    },
    {
      "epoch": 6.472015380859375e-05,
      "step": 67864,
      "training_step_time": 10.36634373664856
    },
    {
      "epoch": 6.472110748291015e-05,
      "model_forward_time": 1.3625383377075195,
      "step": 67865
    },
    {
      "epoch": 6.472110748291015e-05,
      "step": 67865,
      "training_step_time": 8.342912435531616
    },
    {
      "epoch": 6.472206115722656e-05,
      "model_forward_time": 1.728865623474121,
      "step": 67866
    },
    {
      "epoch": 6.472206115722656e-05,
      "step": 67866,
      "training_step_time": 7.49459433555603
    },
    {
      "epoch": 6.472301483154297e-05,
      "model_forward_time": 1.7290587425231934,
      "step": 67867
    },
    {
      "epoch": 6.472301483154297e-05,
      "step": 67867,
      "training_step_time": 6.815974235534668
    },
    {
      "epoch": 6.472396850585937e-05,
      "model_forward_time": 1.72868013381958,
      "step": 67868
    },
    {
      "epoch": 6.472396850585937e-05,
      "step": 67868,
      "training_step_time": 6.814575433731079
    },
    {
      "epoch": 6.472492218017578e-05,
      "model_forward_time": 1.7268600463867188,
      "step": 67869
    },
    {
      "epoch": 6.472492218017578e-05,
      "step": 67869,
      "training_step_time": 6.850205421447754
    },
    {
      "epoch": 6.472587585449219e-05,
      "grad_norm": 0.08550038188695908,
      "learning_rate": 2.5667821761311472e-06,
      "loss": 0.0532,
      "step": 67870
    },
    {
      "epoch": 6.472587585449219e-05,
      "model_forward_time": 1.7319889068603516,
      "step": 67870
    },
    {
      "dynamics_loss_avg": 0.058761556446552274,
      "epoch": 6.472587585449219e-05,
      "step": 67870
    },
    {
      "action_loss_avg": 0.0022427541902288795,
      "epoch": 6.472587585449219e-05,
      "step": 67870
    },
    {
      "epoch": 6.472587585449219e-05,
      "step": 67870,
      "training_step_time": 6.82153844833374
    },
    {
      "epoch": 6.472682952880859e-05,
      "model_forward_time": 1.722682237625122,
      "step": 67871
    },
    {
      "epoch": 6.472682952880859e-05,
      "step": 67871,
      "training_step_time": 6.790404796600342
    },
    {
      "epoch": 6.4727783203125e-05,
      "model_forward_time": 1.7440788745880127,
      "step": 67872
    },
    {
      "epoch": 6.4727783203125e-05,
      "step": 67872,
      "training_step_time": 8.466917991638184
    },
    {
      "epoch": 6.472873687744141e-05,
      "model_forward_time": 2.941615343093872,
      "step": 67873
    },
    {
      "epoch": 6.472873687744141e-05,
      "step": 67873,
      "training_step_time": 12.848631381988525
    },
    {
      "epoch": 6.472969055175781e-05,
      "model_forward_time": 1.7301225662231445,
      "step": 67874
    },
    {
      "epoch": 6.472969055175781e-05,
      "step": 67874,
      "training_step_time": 7.765677213668823
    },
    {
      "epoch": 6.473064422607422e-05,
      "model_forward_time": 1.7298283576965332,
      "step": 67875
    },
    {
      "epoch": 6.473064422607422e-05,
      "step": 67875,
      "training_step_time": 7.5908119678497314
    },
    {
      "epoch": 6.473159790039063e-05,
      "model_forward_time": 1.7264769077301025,
      "step": 67876
    },
    {
      "epoch": 6.473159790039063e-05,
      "step": 67876,
      "training_step_time": 6.8722004890441895
    },
    {
      "epoch": 6.473255157470703e-05,
      "model_forward_time": 1.7218573093414307,
      "step": 67877
    },
    {
      "epoch": 6.473255157470703e-05,
      "step": 67877,
      "training_step_time": 6.790876150131226
    },
    {
      "epoch": 6.473350524902344e-05,
      "model_forward_time": 1.7301487922668457,
      "step": 67878
    },
    {
      "epoch": 6.473350524902344e-05,
      "step": 67878,
      "training_step_time": 6.775403022766113
    },
    {
      "epoch": 6.473445892333985e-05,
      "model_forward_time": 1.7254443168640137,
      "step": 67879
    },
    {
      "epoch": 6.473445892333985e-05,
      "step": 67879,
      "training_step_time": 6.823113203048706
    },
    {
      "epoch": 6.473541259765625e-05,
      "grad_norm": 0.09844013303518295,
      "learning_rate": 2.5653378369483917e-06,
      "loss": 0.0553,
      "step": 67880
    },
    {
      "epoch": 6.473541259765625e-05,
      "model_forward_time": 1.715475082397461,
      "step": 67880
    },
    {
      "dynamics_loss_avg": 0.04535104129463434,
      "epoch": 6.473541259765625e-05,
      "step": 67880
    },
    {
      "action_loss_avg": 0.0017056894168490544,
      "epoch": 6.473541259765625e-05,
      "step": 67880
    },
    {
      "epoch": 6.473541259765625e-05,
      "step": 67880,
      "training_step_time": 6.748168468475342
    },
    {
      "epoch": 6.473636627197266e-05,
      "model_forward_time": 1.7346034049987793,
      "step": 67881
    },
    {
      "epoch": 6.473636627197266e-05,
      "step": 67881,
      "training_step_time": 7.576512575149536
    },
    {
      "epoch": 6.473731994628907e-05,
      "model_forward_time": 2.9030449390411377,
      "step": 67882
    },
    {
      "epoch": 6.473731994628907e-05,
      "step": 67882,
      "training_step_time": 8.388012409210205
    },
    {
      "epoch": 6.473827362060546e-05,
      "model_forward_time": 1.7233407497406006,
      "step": 67883
    },
    {
      "epoch": 6.473827362060546e-05,
      "step": 67883,
      "training_step_time": 11.175591230392456
    },
    {
      "epoch": 6.473922729492188e-05,
      "model_forward_time": 1.7259297370910645,
      "step": 67884
    },
    {
      "epoch": 6.473922729492188e-05,
      "step": 67884,
      "training_step_time": 7.583849191665649
    },
    {
      "epoch": 6.474018096923829e-05,
      "model_forward_time": 1.729356288909912,
      "step": 67885
    },
    {
      "epoch": 6.474018096923829e-05,
      "step": 67885,
      "training_step_time": 6.798020839691162
    },
    {
      "epoch": 6.474113464355468e-05,
      "model_forward_time": 1.7300777435302734,
      "step": 67886
    },
    {
      "epoch": 6.474113464355468e-05,
      "step": 67886,
      "training_step_time": 6.710239410400391
    },
    {
      "epoch": 6.47420883178711e-05,
      "model_forward_time": 1.724827527999878,
      "step": 67887
    },
    {
      "epoch": 6.47420883178711e-05,
      "step": 67887,
      "training_step_time": 6.73735237121582
    },
    {
      "epoch": 6.47430419921875e-05,
      "model_forward_time": 1.728874921798706,
      "step": 67888
    },
    {
      "epoch": 6.47430419921875e-05,
      "step": 67888,
      "training_step_time": 6.751866579055786
    },
    {
      "epoch": 6.47439956665039e-05,
      "model_forward_time": 1.7306733131408691,
      "step": 67889
    },
    {
      "epoch": 6.47439956665039e-05,
      "step": 67889,
      "training_step_time": 6.727412462234497
    },
    {
      "epoch": 6.474494934082031e-05,
      "grad_norm": 0.08481921255588531,
      "learning_rate": 2.5638937640166293e-06,
      "loss": 0.0568,
      "step": 67890
    },
    {
      "epoch": 6.474494934082031e-05,
      "model_forward_time": 1.7484447956085205,
      "step": 67890
    },
    {
      "dynamics_loss_avg": 0.06434678081423044,
      "epoch": 6.474494934082031e-05,
      "step": 67890
    },
    {
      "action_loss_avg": 0.002337443706346676,
      "epoch": 6.474494934082031e-05,
      "step": 67890
    },
    {
      "epoch": 6.474494934082031e-05,
      "step": 67890,
      "training_step_time": 7.562872886657715
    },
    {
      "epoch": 6.474590301513672e-05,
      "model_forward_time": 2.874577522277832,
      "step": 67891
    },
    {
      "epoch": 6.474590301513672e-05,
      "step": 67891,
      "training_step_time": 8.173901319503784
    },
    {
      "epoch": 6.474685668945312e-05,
      "model_forward_time": 1.7189898490905762,
      "step": 67892
    },
    {
      "epoch": 6.474685668945312e-05,
      "step": 67892,
      "training_step_time": 9.363863468170166
    },
    {
      "epoch": 6.474781036376953e-05,
      "model_forward_time": 1.7309322357177734,
      "step": 67893
    },
    {
      "epoch": 6.474781036376953e-05,
      "step": 67893,
      "training_step_time": 7.544846296310425
    },
    {
      "epoch": 6.474876403808594e-05,
      "model_forward_time": 1.7242636680603027,
      "step": 67894
    },
    {
      "epoch": 6.474876403808594e-05,
      "step": 67894,
      "training_step_time": 6.717291355133057
    },
    {
      "epoch": 6.474971771240234e-05,
      "model_forward_time": 1.7140476703643799,
      "step": 67895
    },
    {
      "epoch": 6.474971771240234e-05,
      "step": 67895,
      "training_step_time": 6.706241607666016
    },
    {
      "epoch": 6.475067138671875e-05,
      "model_forward_time": 1.714064598083496,
      "step": 67896
    },
    {
      "epoch": 6.475067138671875e-05,
      "step": 67896,
      "training_step_time": 6.699382543563843
    },
    {
      "epoch": 6.475162506103516e-05,
      "model_forward_time": 1.7235069274902344,
      "step": 67897
    },
    {
      "epoch": 6.475162506103516e-05,
      "step": 67897,
      "training_step_time": 6.710473537445068
    },
    {
      "epoch": 6.475257873535156e-05,
      "model_forward_time": 1.7257637977600098,
      "step": 67898
    },
    {
      "epoch": 6.475257873535156e-05,
      "step": 67898,
      "training_step_time": 6.737306118011475
    },
    {
      "epoch": 6.475353240966797e-05,
      "model_forward_time": 1.7429571151733398,
      "step": 67899
    },
    {
      "epoch": 6.475353240966797e-05,
      "step": 67899,
      "training_step_time": 8.143288373947144
    },
    {
      "epoch": 6.475448608398438e-05,
      "grad_norm": 0.1049480065703392,
      "learning_rate": 2.5624499574937843e-06,
      "loss": 0.0562,
      "step": 67900
    },
    {
      "epoch": 6.475448608398438e-05,
      "model_forward_time": 1.7214241027832031,
      "step": 67900
    },
    {
      "dynamics_loss_avg": 0.06253710826858878,
      "epoch": 6.475448608398438e-05,
      "step": 67900
    },
    {
      "action_loss_avg": 0.001171560847433284,
      "epoch": 6.475448608398438e-05,
      "step": 67900
    },
    {
      "epoch": 6.475448608398438e-05,
      "step": 67900,
      "training_step_time": 9.54930329322815
    },
    {
      "epoch": 6.475543975830078e-05,
      "model_forward_time": 2.9465012550354004,
      "step": 67901
    },
    {
      "epoch": 6.475543975830078e-05,
      "step": 67901,
      "training_step_time": 9.16835355758667
    },
    {
      "epoch": 6.475639343261719e-05,
      "model_forward_time": 1.7244832515716553,
      "step": 67902
    },
    {
      "epoch": 6.475639343261719e-05,
      "step": 67902,
      "training_step_time": 7.61875319480896
    },
    {
      "epoch": 6.47573471069336e-05,
      "model_forward_time": 1.7201135158538818,
      "step": 67903
    },
    {
      "epoch": 6.47573471069336e-05,
      "step": 67903,
      "training_step_time": 7.5424158573150635
    },
    {
      "epoch": 6.475830078125e-05,
      "model_forward_time": 1.7227180004119873,
      "step": 67904
    },
    {
      "epoch": 6.475830078125e-05,
      "step": 67904,
      "training_step_time": 6.765002727508545
    },
    {
      "epoch": 6.475925445556641e-05,
      "model_forward_time": 1.720390796661377,
      "step": 67905
    },
    {
      "epoch": 6.475925445556641e-05,
      "step": 67905,
      "training_step_time": 6.747270584106445
    },
    {
      "epoch": 6.476020812988282e-05,
      "model_forward_time": 1.716801643371582,
      "step": 67906
    },
    {
      "epoch": 6.476020812988282e-05,
      "step": 67906,
      "training_step_time": 6.764025688171387
    },
    {
      "epoch": 6.476116180419922e-05,
      "model_forward_time": 1.7282938957214355,
      "step": 67907
    },
    {
      "epoch": 6.476116180419922e-05,
      "step": 67907,
      "training_step_time": 6.749656677246094
    },
    {
      "epoch": 6.476211547851563e-05,
      "model_forward_time": 1.7637848854064941,
      "step": 67908
    },
    {
      "epoch": 6.476211547851563e-05,
      "step": 67908,
      "training_step_time": 8.638205289840698
    },
    {
      "epoch": 6.476306915283204e-05,
      "model_forward_time": 1.728882074356079,
      "step": 67909
    },
    {
      "epoch": 6.476306915283204e-05,
      "step": 67909,
      "training_step_time": 8.757463455200195
    },
    {
      "epoch": 6.476402282714843e-05,
      "grad_norm": 0.07316932082176208,
      "learning_rate": 2.5610064175377456e-06,
      "loss": 0.0551,
      "step": 67910
    },
    {
      "epoch": 6.476402282714843e-05,
      "model_forward_time": 2.9526772499084473,
      "step": 67910
    },
    {
      "dynamics_loss_avg": 0.05478704161942005,
      "epoch": 6.476402282714843e-05,
      "step": 67910
    },
    {
      "action_loss_avg": 0.0017328066693153231,
      "epoch": 6.476402282714843e-05,
      "step": 67910
    },
    {
      "epoch": 6.476402282714843e-05,
      "step": 67910,
      "training_step_time": 9.874962329864502
    },
    {
      "epoch": 6.476497650146485e-05,
      "model_forward_time": 1.7336435317993164,
      "step": 67911
    },
    {
      "epoch": 6.476497650146485e-05,
      "step": 67911,
      "training_step_time": 9.055131196975708
    },
    {
      "epoch": 6.476593017578126e-05,
      "model_forward_time": 1.72983980178833,
      "step": 67912
    },
    {
      "epoch": 6.476593017578126e-05,
      "step": 67912,
      "training_step_time": 7.637265920639038
    },
    {
      "epoch": 6.476688385009765e-05,
      "model_forward_time": 1.7281219959259033,
      "step": 67913
    },
    {
      "epoch": 6.476688385009765e-05,
      "step": 67913,
      "training_step_time": 6.726305723190308
    },
    {
      "epoch": 6.476783752441406e-05,
      "model_forward_time": 1.7305293083190918,
      "step": 67914
    },
    {
      "epoch": 6.476783752441406e-05,
      "step": 67914,
      "training_step_time": 6.764491558074951
    },
    {
      "epoch": 6.476879119873047e-05,
      "model_forward_time": 1.7384912967681885,
      "step": 67915
    },
    {
      "epoch": 6.476879119873047e-05,
      "step": 67915,
      "training_step_time": 6.741297721862793
    },
    {
      "epoch": 6.476974487304687e-05,
      "model_forward_time": 1.7353169918060303,
      "step": 67916
    },
    {
      "epoch": 6.476974487304687e-05,
      "step": 67916,
      "training_step_time": 6.736750364303589
    },
    {
      "epoch": 6.477069854736328e-05,
      "model_forward_time": 1.7696683406829834,
      "step": 67917
    },
    {
      "epoch": 6.477069854736328e-05,
      "step": 67917,
      "training_step_time": 7.533858776092529
    },
    {
      "epoch": 6.47716522216797e-05,
      "model_forward_time": 1.728337287902832,
      "step": 67918
    },
    {
      "epoch": 6.47716522216797e-05,
      "step": 67918,
      "training_step_time": 8.32375693321228
    },
    {
      "epoch": 6.477260589599609e-05,
      "model_forward_time": 2.9662272930145264,
      "step": 67919
    },
    {
      "epoch": 6.477260589599609e-05,
      "step": 67919,
      "training_step_time": 10.309888362884521
    },
    {
      "epoch": 6.47735595703125e-05,
      "grad_norm": 0.08246531337499619,
      "learning_rate": 2.5595631443063783e-06,
      "loss": 0.0562,
      "step": 67920
    },
    {
      "epoch": 6.47735595703125e-05,
      "model_forward_time": 1.7177379131317139,
      "step": 67920
    },
    {
      "dynamics_loss_avg": 0.04399473499506712,
      "epoch": 6.47735595703125e-05,
      "step": 67920
    },
    {
      "action_loss_avg": 0.0009479936328716576,
      "epoch": 6.47735595703125e-05,
      "step": 67920
    },
    {
      "epoch": 6.47735595703125e-05,
      "step": 67920,
      "training_step_time": 9.562967300415039
    },
    {
      "epoch": 6.477451324462891e-05,
      "model_forward_time": 1.726931095123291,
      "step": 67921
    },
    {
      "epoch": 6.477451324462891e-05,
      "step": 67921,
      "training_step_time": 7.709693670272827
    },
    {
      "epoch": 6.477546691894531e-05,
      "model_forward_time": 1.7184195518493652,
      "step": 67922
    },
    {
      "epoch": 6.477546691894531e-05,
      "step": 67922,
      "training_step_time": 6.669684171676636
    },
    {
      "epoch": 6.477642059326172e-05,
      "model_forward_time": 1.7286415100097656,
      "step": 67923
    },
    {
      "epoch": 6.477642059326172e-05,
      "step": 67923,
      "training_step_time": 6.812070608139038
    },
    {
      "epoch": 6.477737426757813e-05,
      "model_forward_time": 1.729600429534912,
      "step": 67924
    },
    {
      "epoch": 6.477737426757813e-05,
      "step": 67924,
      "training_step_time": 6.77053427696228
    },
    {
      "epoch": 6.477832794189453e-05,
      "model_forward_time": 1.7265820503234863,
      "step": 67925
    },
    {
      "epoch": 6.477832794189453e-05,
      "step": 67925,
      "training_step_time": 6.73993706703186
    },
    {
      "epoch": 6.477928161621094e-05,
      "model_forward_time": 1.7572064399719238,
      "step": 67926
    },
    {
      "epoch": 6.477928161621094e-05,
      "step": 67926,
      "training_step_time": 7.505434989929199
    },
    {
      "epoch": 6.478023529052735e-05,
      "model_forward_time": 1.732300043106079,
      "step": 67927
    },
    {
      "epoch": 6.478023529052735e-05,
      "step": 67927,
      "training_step_time": 7.5866758823394775
    },
    {
      "epoch": 6.478118896484375e-05,
      "model_forward_time": 2.896383285522461,
      "step": 67928
    },
    {
      "epoch": 6.478118896484375e-05,
      "step": 67928,
      "training_step_time": 8.229673147201538
    },
    {
      "epoch": 6.478214263916016e-05,
      "model_forward_time": 1.7414815425872803,
      "step": 67929
    },
    {
      "epoch": 6.478214263916016e-05,
      "step": 67929,
      "training_step_time": 8.922014713287354
    },
    {
      "epoch": 6.478309631347657e-05,
      "grad_norm": 0.11048728227615356,
      "learning_rate": 2.5581201379575176e-06,
      "loss": 0.0541,
      "step": 67930
    },
    {
      "epoch": 6.478309631347657e-05,
      "model_forward_time": 1.7356913089752197,
      "step": 67930
    },
    {
      "dynamics_loss_avg": 0.04295133762061596,
      "epoch": 6.478309631347657e-05,
      "step": 67930
    },
    {
      "action_loss_avg": 0.0017251395562198012,
      "epoch": 6.478309631347657e-05,
      "step": 67930
    },
    {
      "epoch": 6.478309631347657e-05,
      "step": 67930,
      "training_step_time": 7.673117160797119
    },
    {
      "epoch": 6.478404998779297e-05,
      "model_forward_time": 1.7260453701019287,
      "step": 67931
    },
    {
      "epoch": 6.478404998779297e-05,
      "step": 67931,
      "training_step_time": 7.3912272453308105
    },
    {
      "epoch": 6.478500366210938e-05,
      "model_forward_time": 1.7278225421905518,
      "step": 67932
    },
    {
      "epoch": 6.478500366210938e-05,
      "step": 67932,
      "training_step_time": 6.739165782928467
    },
    {
      "epoch": 6.478595733642579e-05,
      "model_forward_time": 1.7310423851013184,
      "step": 67933
    },
    {
      "epoch": 6.478595733642579e-05,
      "step": 67933,
      "training_step_time": 6.770668983459473
    },
    {
      "epoch": 6.478691101074219e-05,
      "model_forward_time": 1.7367281913757324,
      "step": 67934
    },
    {
      "epoch": 6.478691101074219e-05,
      "step": 67934,
      "training_step_time": 6.749268531799316
    },
    {
      "epoch": 6.47878646850586e-05,
      "model_forward_time": 1.7369003295898438,
      "step": 67935
    },
    {
      "epoch": 6.47878646850586e-05,
      "step": 67935,
      "training_step_time": 7.537728309631348
    },
    {
      "epoch": 6.4788818359375e-05,
      "model_forward_time": 1.7141056060791016,
      "step": 67936
    },
    {
      "epoch": 6.4788818359375e-05,
      "step": 67936,
      "training_step_time": 7.60640025138855
    },
    {
      "epoch": 6.47897720336914e-05,
      "model_forward_time": 2.900766134262085,
      "step": 67937
    },
    {
      "epoch": 6.47897720336914e-05,
      "step": 67937,
      "training_step_time": 8.529984474182129
    },
    {
      "epoch": 6.479072570800781e-05,
      "model_forward_time": 1.7357797622680664,
      "step": 67938
    },
    {
      "epoch": 6.479072570800781e-05,
      "step": 67938,
      "training_step_time": 8.783273458480835
    },
    {
      "epoch": 6.479167938232421e-05,
      "model_forward_time": 1.7137296199798584,
      "step": 67939
    },
    {
      "epoch": 6.479167938232421e-05,
      "step": 67939,
      "training_step_time": 7.633797883987427
    },
    {
      "epoch": 6.479263305664062e-05,
      "grad_norm": 0.10540282726287842,
      "learning_rate": 2.5566773986489657e-06,
      "loss": 0.0538,
      "step": 67940
    },
    {
      "epoch": 6.479263305664062e-05,
      "model_forward_time": 1.7206218242645264,
      "step": 67940
    },
    {
      "dynamics_loss_avg": 0.055710804834961894,
      "epoch": 6.479263305664062e-05,
      "step": 67940
    },
    {
      "action_loss_avg": 0.001768816285766661,
      "epoch": 6.479263305664062e-05,
      "step": 67940
    },
    {
      "epoch": 6.479263305664062e-05,
      "step": 67940,
      "training_step_time": 7.678062677383423
    },
    {
      "epoch": 6.479358673095703e-05,
      "model_forward_time": 1.7152783870697021,
      "step": 67941
    },
    {
      "epoch": 6.479358673095703e-05,
      "step": 67941,
      "training_step_time": 6.766857147216797
    },
    {
      "epoch": 6.479454040527343e-05,
      "model_forward_time": 1.71402907371521,
      "step": 67942
    },
    {
      "epoch": 6.479454040527343e-05,
      "step": 67942,
      "training_step_time": 6.721336364746094
    },
    {
      "epoch": 6.479549407958984e-05,
      "model_forward_time": 1.7202868461608887,
      "step": 67943
    },
    {
      "epoch": 6.479549407958984e-05,
      "step": 67943,
      "training_step_time": 6.770843267440796
    },
    {
      "epoch": 6.479644775390625e-05,
      "model_forward_time": 1.7586829662322998,
      "step": 67944
    },
    {
      "epoch": 6.479644775390625e-05,
      "step": 67944,
      "training_step_time": 7.5493669509887695
    },
    {
      "epoch": 6.479740142822265e-05,
      "model_forward_time": 1.7177119255065918,
      "step": 67945
    },
    {
      "epoch": 6.479740142822265e-05,
      "step": 67945,
      "training_step_time": 7.575717926025391
    },
    {
      "epoch": 6.479835510253906e-05,
      "model_forward_time": 2.931039333343506,
      "step": 67946
    },
    {
      "epoch": 6.479835510253906e-05,
      "step": 67946,
      "training_step_time": 8.97252368927002
    },
    {
      "epoch": 6.479930877685547e-05,
      "model_forward_time": 1.7227935791015625,
      "step": 67947
    },
    {
      "epoch": 6.479930877685547e-05,
      "step": 67947,
      "training_step_time": 9.42900037765503
    },
    {
      "epoch": 6.480026245117187e-05,
      "model_forward_time": 1.7150704860687256,
      "step": 67948
    },
    {
      "epoch": 6.480026245117187e-05,
      "step": 67948,
      "training_step_time": 7.603939533233643
    },
    {
      "epoch": 6.480121612548828e-05,
      "model_forward_time": 1.7209453582763672,
      "step": 67949
    },
    {
      "epoch": 6.480121612548828e-05,
      "step": 67949,
      "training_step_time": 7.635150194168091
    },
    {
      "epoch": 6.480216979980469e-05,
      "grad_norm": 0.11271769553422928,
      "learning_rate": 2.5552349265385003e-06,
      "loss": 0.0547,
      "step": 67950
    },
    {
      "epoch": 6.480216979980469e-05,
      "model_forward_time": 1.712369441986084,
      "step": 67950
    },
    {
      "dynamics_loss_avg": 0.044236138463020325,
      "epoch": 6.480216979980469e-05,
      "step": 67950
    },
    {
      "action_loss_avg": 0.0016737890837248415,
      "epoch": 6.480216979980469e-05,
      "step": 67950
    },
    {
      "epoch": 6.480216979980469e-05,
      "step": 67950,
      "training_step_time": 6.774349927902222
    },
    {
      "epoch": 6.480312347412109e-05,
      "model_forward_time": 1.7251310348510742,
      "step": 67951
    },
    {
      "epoch": 6.480312347412109e-05,
      "step": 67951,
      "training_step_time": 6.7604758739471436
    },
    {
      "epoch": 6.48040771484375e-05,
      "model_forward_time": 1.7174694538116455,
      "step": 67952
    },
    {
      "epoch": 6.48040771484375e-05,
      "step": 67952,
      "training_step_time": 6.807899236679077
    },
    {
      "epoch": 6.480503082275391e-05,
      "model_forward_time": 1.7277753353118896,
      "step": 67953
    },
    {
      "epoch": 6.480503082275391e-05,
      "step": 67953,
      "training_step_time": 8.048359155654907
    },
    {
      "epoch": 6.48059844970703e-05,
      "model_forward_time": 1.719691276550293,
      "step": 67954
    },
    {
      "epoch": 6.48059844970703e-05,
      "step": 67954,
      "training_step_time": 7.657364845275879
    },
    {
      "epoch": 6.480693817138672e-05,
      "model_forward_time": 2.914580821990967,
      "step": 67955
    },
    {
      "epoch": 6.480693817138672e-05,
      "step": 67955,
      "training_step_time": 7.691031217575073
    },
    {
      "epoch": 6.480789184570313e-05,
      "model_forward_time": 1.7183127403259277,
      "step": 67956
    },
    {
      "epoch": 6.480789184570313e-05,
      "step": 67956,
      "training_step_time": 10.246407508850098
    },
    {
      "epoch": 6.480884552001953e-05,
      "model_forward_time": 1.7125778198242188,
      "step": 67957
    },
    {
      "epoch": 6.480884552001953e-05,
      "step": 67957,
      "training_step_time": 9.19652533531189
    },
    {
      "epoch": 6.480979919433594e-05,
      "model_forward_time": 1.7200822830200195,
      "step": 67958
    },
    {
      "epoch": 6.480979919433594e-05,
      "step": 67958,
      "training_step_time": 8.261335611343384
    },
    {
      "epoch": 6.481075286865235e-05,
      "model_forward_time": 1.720057725906372,
      "step": 67959
    },
    {
      "epoch": 6.481075286865235e-05,
      "step": 67959,
      "training_step_time": 6.739920139312744
    },
    {
      "epoch": 6.481170654296874e-05,
      "grad_norm": 0.12006852775812149,
      "learning_rate": 2.5537927217838678e-06,
      "loss": 0.0545,
      "step": 67960
    },
    {
      "epoch": 6.481170654296874e-05,
      "model_forward_time": 1.7217185497283936,
      "step": 67960
    },
    {
      "dynamics_loss_avg": 0.049632414057850835,
      "epoch": 6.481170654296874e-05,
      "step": 67960
    },
    {
      "action_loss_avg": 0.0013214629841968417,
      "epoch": 6.481170654296874e-05,
      "step": 67960
    },
    {
      "epoch": 6.481170654296874e-05,
      "step": 67960,
      "training_step_time": 6.778168201446533
    },
    {
      "epoch": 6.481266021728515e-05,
      "model_forward_time": 1.718857765197754,
      "step": 67961
    },
    {
      "epoch": 6.481266021728515e-05,
      "step": 67961,
      "training_step_time": 6.759977102279663
    },
    {
      "epoch": 6.481361389160157e-05,
      "model_forward_time": 1.7499699592590332,
      "step": 67962
    },
    {
      "epoch": 6.481361389160157e-05,
      "step": 67962,
      "training_step_time": 7.581944704055786
    },
    {
      "epoch": 6.481456756591796e-05,
      "model_forward_time": 1.7157893180847168,
      "step": 67963
    },
    {
      "epoch": 6.481456756591796e-05,
      "step": 67963,
      "training_step_time": 7.650786399841309
    },
    {
      "epoch": 6.481552124023437e-05,
      "model_forward_time": 2.924161911010742,
      "step": 67964
    },
    {
      "epoch": 6.481552124023437e-05,
      "step": 67964,
      "training_step_time": 8.23909044265747
    },
    {
      "epoch": 6.481647491455078e-05,
      "model_forward_time": 1.7237625122070312,
      "step": 67965
    },
    {
      "epoch": 6.481647491455078e-05,
      "step": 67965,
      "training_step_time": 9.600036144256592
    },
    {
      "epoch": 6.481742858886718e-05,
      "model_forward_time": 1.72733473777771,
      "step": 67966
    },
    {
      "epoch": 6.481742858886718e-05,
      "step": 67966,
      "training_step_time": 7.7870261669158936
    },
    {
      "epoch": 6.481838226318359e-05,
      "model_forward_time": 1.7175650596618652,
      "step": 67967
    },
    {
      "epoch": 6.481838226318359e-05,
      "step": 67967,
      "training_step_time": 9.48538589477539
    },
    {
      "epoch": 6.48193359375e-05,
      "model_forward_time": 1.7247929573059082,
      "step": 67968
    },
    {
      "epoch": 6.48193359375e-05,
      "step": 67968,
      "training_step_time": 7.395498275756836
    },
    {
      "epoch": 6.48202896118164e-05,
      "model_forward_time": 1.718996286392212,
      "step": 67969
    },
    {
      "epoch": 6.48202896118164e-05,
      "step": 67969,
      "training_step_time": 6.687756299972534
    },
    {
      "epoch": 6.482124328613281e-05,
      "grad_norm": 0.10293012857437134,
      "learning_rate": 2.5523507845427864e-06,
      "loss": 0.0562,
      "step": 67970
    },
    {
      "epoch": 6.482124328613281e-05,
      "model_forward_time": 1.7186150550842285,
      "step": 67970
    },
    {
      "dynamics_loss_avg": 0.05134700872004032,
      "epoch": 6.482124328613281e-05,
      "step": 67970
    },
    {
      "action_loss_avg": 0.0011494809499708936,
      "epoch": 6.482124328613281e-05,
      "step": 67970
    },
    {
      "epoch": 6.482124328613281e-05,
      "step": 67970,
      "training_step_time": 6.752191543579102
    },
    {
      "epoch": 6.482219696044922e-05,
      "model_forward_time": 1.7507121562957764,
      "step": 67971
    },
    {
      "epoch": 6.482219696044922e-05,
      "step": 67971,
      "training_step_time": 7.53557276725769
    },
    {
      "epoch": 6.482315063476562e-05,
      "model_forward_time": 1.7163267135620117,
      "step": 67972
    },
    {
      "epoch": 6.482315063476562e-05,
      "step": 67972,
      "training_step_time": 7.565488815307617
    },
    {
      "epoch": 6.482410430908203e-05,
      "model_forward_time": 2.9172163009643555,
      "step": 67973
    },
    {
      "epoch": 6.482410430908203e-05,
      "step": 67973,
      "training_step_time": 7.689897060394287
    },
    {
      "epoch": 6.482505798339844e-05,
      "model_forward_time": 1.7293143272399902,
      "step": 67974
    },
    {
      "epoch": 6.482505798339844e-05,
      "step": 67974,
      "training_step_time": 7.6681132316589355
    },
    {
      "epoch": 6.482601165771484e-05,
      "model_forward_time": 1.7314064502716064,
      "step": 67975
    },
    {
      "epoch": 6.482601165771484e-05,
      "step": 67975,
      "training_step_time": 8.882480382919312
    },
    {
      "epoch": 6.482696533203125e-05,
      "model_forward_time": 1.7178173065185547,
      "step": 67976
    },
    {
      "epoch": 6.482696533203125e-05,
      "step": 67976,
      "training_step_time": 7.564725637435913
    },
    {
      "epoch": 6.482791900634766e-05,
      "model_forward_time": 1.7210419178009033,
      "step": 67977
    },
    {
      "epoch": 6.482791900634766e-05,
      "step": 67977,
      "training_step_time": 7.511507749557495
    },
    {
      "epoch": 6.482887268066406e-05,
      "model_forward_time": 1.7192909717559814,
      "step": 67978
    },
    {
      "epoch": 6.482887268066406e-05,
      "step": 67978,
      "training_step_time": 6.707951784133911
    },
    {
      "epoch": 6.482982635498047e-05,
      "model_forward_time": 1.7198536396026611,
      "step": 67979
    },
    {
      "epoch": 6.482982635498047e-05,
      "step": 67979,
      "training_step_time": 6.728763103485107
    },
    {
      "epoch": 6.483078002929688e-05,
      "grad_norm": 0.10226930677890778,
      "learning_rate": 2.5509091149729417e-06,
      "loss": 0.054,
      "step": 67980
    },
    {
      "epoch": 6.483078002929688e-05,
      "model_forward_time": 1.739030361175537,
      "step": 67980
    },
    {
      "dynamics_loss_avg": 0.05690758489072323,
      "epoch": 6.483078002929688e-05,
      "step": 67980
    },
    {
      "action_loss_avg": 0.003368586563738063,
      "epoch": 6.483078002929688e-05,
      "step": 67980
    },
    {
      "epoch": 6.483078002929688e-05,
      "step": 67980,
      "training_step_time": 7.532880067825317
    },
    {
      "epoch": 6.483173370361328e-05,
      "model_forward_time": 1.719310998916626,
      "step": 67981
    },
    {
      "epoch": 6.483173370361328e-05,
      "step": 67981,
      "training_step_time": 7.5962865352630615
    },
    {
      "epoch": 6.483268737792969e-05,
      "model_forward_time": 2.928245782852173,
      "step": 67982
    },
    {
      "epoch": 6.483268737792969e-05,
      "step": 67982,
      "training_step_time": 7.630731582641602
    },
    {
      "epoch": 6.48336410522461e-05,
      "model_forward_time": 1.7176251411437988,
      "step": 67983
    },
    {
      "epoch": 6.48336410522461e-05,
      "step": 67983,
      "training_step_time": 9.20887017250061
    },
    {
      "epoch": 6.48345947265625e-05,
      "model_forward_time": 1.7280726432800293,
      "step": 67984
    },
    {
      "epoch": 6.48345947265625e-05,
      "step": 67984,
      "training_step_time": 9.686409950256348
    },
    {
      "epoch": 6.48355484008789e-05,
      "model_forward_time": 1.7170026302337646,
      "step": 67985
    },
    {
      "epoch": 6.48355484008789e-05,
      "step": 67985,
      "training_step_time": 7.56580376625061
    },
    {
      "epoch": 6.483650207519532e-05,
      "model_forward_time": 1.709501028060913,
      "step": 67986
    },
    {
      "epoch": 6.483650207519532e-05,
      "step": 67986,
      "training_step_time": 7.539935827255249
    },
    {
      "epoch": 6.483745574951171e-05,
      "model_forward_time": 1.7221238613128662,
      "step": 67987
    },
    {
      "epoch": 6.483745574951171e-05,
      "step": 67987,
      "training_step_time": 7.281672239303589
    },
    {
      "epoch": 6.483840942382812e-05,
      "model_forward_time": 1.3574168682098389,
      "step": 67988
    },
    {
      "epoch": 6.483840942382812e-05,
      "step": 67988,
      "training_step_time": 6.68600606918335
    },
    {
      "epoch": 6.483936309814454e-05,
      "model_forward_time": 1.7427489757537842,
      "step": 67989
    },
    {
      "epoch": 6.483936309814454e-05,
      "step": 67989,
      "training_step_time": 7.516361951828003
    },
    {
      "epoch": 6.484031677246093e-05,
      "grad_norm": 0.06318549811840057,
      "learning_rate": 2.549467713231996e-06,
      "loss": 0.0558,
      "step": 67990
    },
    {
      "epoch": 6.484031677246093e-05,
      "model_forward_time": 1.718860149383545,
      "step": 67990
    },
    {
      "dynamics_loss_avg": 0.04561313930898905,
      "epoch": 6.484031677246093e-05,
      "step": 67990
    },
    {
      "action_loss_avg": 0.001650883018737659,
      "epoch": 6.484031677246093e-05,
      "step": 67990
    },
    {
      "epoch": 6.484031677246093e-05,
      "step": 67990,
      "training_step_time": 7.506089448928833
    },
    {
      "epoch": 6.484127044677734e-05,
      "model_forward_time": 2.9022786617279053,
      "step": 67991
    },
    {
      "epoch": 6.484127044677734e-05,
      "step": 67991,
      "training_step_time": 8.695868492126465
    },
    {
      "epoch": 6.484222412109375e-05,
      "model_forward_time": 1.7309541702270508,
      "step": 67992
    },
    {
      "epoch": 6.484222412109375e-05,
      "step": 67992,
      "training_step_time": 8.523983716964722
    },
    {
      "epoch": 6.484317779541015e-05,
      "model_forward_time": 1.3508107662200928,
      "step": 67993
    },
    {
      "epoch": 6.484317779541015e-05,
      "step": 67993,
      "training_step_time": 7.833837985992432
    },
    {
      "epoch": 6.484413146972656e-05,
      "model_forward_time": 1.7271621227264404,
      "step": 67994
    },
    {
      "epoch": 6.484413146972656e-05,
      "step": 67994,
      "training_step_time": 7.589296340942383
    },
    {
      "epoch": 6.484508514404297e-05,
      "model_forward_time": 1.7294254302978516,
      "step": 67995
    },
    {
      "epoch": 6.484508514404297e-05,
      "step": 67995,
      "training_step_time": 7.513259649276733
    },
    {
      "epoch": 6.484603881835937e-05,
      "model_forward_time": 1.7305653095245361,
      "step": 67996
    },
    {
      "epoch": 6.484603881835937e-05,
      "step": 67996,
      "training_step_time": 7.5039894580841064
    },
    {
      "epoch": 6.484699249267578e-05,
      "model_forward_time": 1.7406964302062988,
      "step": 67997
    },
    {
      "epoch": 6.484699249267578e-05,
      "step": 67997,
      "training_step_time": 6.732390403747559
    },
    {
      "epoch": 6.484794616699219e-05,
      "model_forward_time": 1.7876558303833008,
      "step": 67998
    },
    {
      "epoch": 6.484794616699219e-05,
      "step": 67998,
      "training_step_time": 7.700929641723633
    },
    {
      "epoch": 6.484889984130859e-05,
      "model_forward_time": 1.8267672061920166,
      "step": 67999
    },
    {
      "epoch": 6.484889984130859e-05,
      "step": 67999,
      "training_step_time": 7.475815296173096
    },
    {
      "epoch": 6.4849853515625e-05,
      "grad_norm": 0.09612499177455902,
      "learning_rate": 2.548026579477575e-06,
      "loss": 0.0557,
      "step": 68000
    },
    {
      "epoch": 6.4849853515625e-05,
      "model_forward_time": 2.9609289169311523,
      "step": 68000
    },
    {
      "dynamics_loss_avg": 0.05007807649672032,
      "epoch": 6.4849853515625e-05,
      "step": 68000
    },
    {
      "action_loss_avg": 0.0014830069558229297,
      "epoch": 6.4849853515625e-05,
      "step": 68000
    },
    {
      "epoch": 6.4849853515625e-05,
      "step": 68000,
      "training_step_time": 9.795827388763428
    },
    {
      "epoch": 6.485080718994141e-05,
      "model_forward_time": 1.7179818153381348,
      "step": 68001
    },
    {
      "epoch": 6.485080718994141e-05,
      "step": 68001,
      "training_step_time": 8.760145902633667
    },
    {
      "epoch": 6.485176086425781e-05,
      "model_forward_time": 1.71478271484375,
      "step": 68002
    },
    {
      "epoch": 6.485176086425781e-05,
      "step": 68002,
      "training_step_time": 10.000298261642456
    },
    {
      "epoch": 6.485271453857422e-05,
      "model_forward_time": 1.7994155883789062,
      "step": 68003
    },
    {
      "epoch": 6.485271453857422e-05,
      "step": 68003,
      "training_step_time": 10.860418319702148
    },
    {
      "epoch": 6.485366821289063e-05,
      "model_forward_time": 1.0205483436584473,
      "step": 68004
    },
    {
      "epoch": 6.485366821289063e-05,
      "step": 68004,
      "training_step_time": 7.545880556106567
    },
    {
      "epoch": 6.485462188720703e-05,
      "model_forward_time": 1.7214539051055908,
      "step": 68005
    },
    {
      "epoch": 6.485462188720703e-05,
      "step": 68005,
      "training_step_time": 7.653473377227783
    },
    {
      "epoch": 6.485557556152344e-05,
      "model_forward_time": 1.7695362567901611,
      "step": 68006
    },
    {
      "epoch": 6.485557556152344e-05,
      "step": 68006,
      "training_step_time": 7.1101295948028564
    },
    {
      "epoch": 6.485652923583985e-05,
      "model_forward_time": 1.721982717514038,
      "step": 68007
    },
    {
      "epoch": 6.485652923583985e-05,
      "step": 68007,
      "training_step_time": 6.618628978729248
    },
    {
      "epoch": 6.485748291015625e-05,
      "model_forward_time": 1.724142074584961,
      "step": 68008
    },
    {
      "epoch": 6.485748291015625e-05,
      "step": 68008,
      "training_step_time": 8.536674737930298
    },
    {
      "epoch": 6.485843658447266e-05,
      "model_forward_time": 2.864847421646118,
      "step": 68009
    },
    {
      "epoch": 6.485843658447266e-05,
      "step": 68009,
      "training_step_time": 9.19493556022644
    },
    {
      "epoch": 6.485939025878907e-05,
      "grad_norm": 0.07783625274896622,
      "learning_rate": 2.5465857138672834e-06,
      "loss": 0.061,
      "step": 68010
    },
    {
      "epoch": 6.485939025878907e-05,
      "model_forward_time": 1.724485158920288,
      "step": 68010
    },
    {
      "dynamics_loss_avg": 0.06734323874115944,
      "epoch": 6.485939025878907e-05,
      "step": 68010
    },
    {
      "action_loss_avg": 0.0014425299974391236,
      "epoch": 6.485939025878907e-05,
      "step": 68010
    },
    {
      "epoch": 6.485939025878907e-05,
      "step": 68010,
      "training_step_time": 8.418074131011963
    },
    {
      "epoch": 6.486034393310546e-05,
      "model_forward_time": 1.7188231945037842,
      "step": 68011
    },
    {
      "epoch": 6.486034393310546e-05,
      "step": 68011,
      "training_step_time": 8.414927244186401
    },
    {
      "epoch": 6.486129760742188e-05,
      "model_forward_time": 1.772519826889038,
      "step": 68012
    },
    {
      "epoch": 6.486129760742188e-05,
      "step": 68012,
      "training_step_time": 7.78155779838562
    },
    {
      "epoch": 6.486225128173829e-05,
      "model_forward_time": 1.71958327293396,
      "step": 68013
    },
    {
      "epoch": 6.486225128173829e-05,
      "step": 68013,
      "training_step_time": 7.578133583068848
    },
    {
      "epoch": 6.486320495605468e-05,
      "model_forward_time": 1.7334682941436768,
      "step": 68014
    },
    {
      "epoch": 6.486320495605468e-05,
      "step": 68014,
      "training_step_time": 7.62064790725708
    },
    {
      "epoch": 6.48641586303711e-05,
      "model_forward_time": 1.7181975841522217,
      "step": 68015
    },
    {
      "epoch": 6.48641586303711e-05,
      "step": 68015,
      "training_step_time": 7.319535732269287
    },
    {
      "epoch": 6.48651123046875e-05,
      "model_forward_time": 1.716540813446045,
      "step": 68016
    },
    {
      "epoch": 6.48651123046875e-05,
      "step": 68016,
      "training_step_time": 7.124606609344482
    },
    {
      "epoch": 6.48660659790039e-05,
      "model_forward_time": 1.7218446731567383,
      "step": 68017
    },
    {
      "epoch": 6.48660659790039e-05,
      "step": 68017,
      "training_step_time": 7.916300058364868
    },
    {
      "epoch": 6.486701965332031e-05,
      "model_forward_time": 2.8235020637512207,
      "step": 68018
    },
    {
      "epoch": 6.486701965332031e-05,
      "step": 68018,
      "training_step_time": 7.618504524230957
    },
    {
      "epoch": 6.486797332763672e-05,
      "model_forward_time": 1.7224314212799072,
      "step": 68019
    },
    {
      "epoch": 6.486797332763672e-05,
      "step": 68019,
      "training_step_time": 7.569802761077881
    },
    {
      "epoch": 6.486892700195312e-05,
      "grad_norm": 0.13656644523143768,
      "learning_rate": 2.545145116558685e-06,
      "loss": 0.0601,
      "step": 68020
    },
    {
      "epoch": 6.486892700195312e-05,
      "model_forward_time": 1.7187774181365967,
      "step": 68020
    },
    {
      "dynamics_loss_avg": 0.05986944269388914,
      "epoch": 6.486892700195312e-05,
      "step": 68020
    },
    {
      "action_loss_avg": 0.0018708481278736145,
      "epoch": 6.486892700195312e-05,
      "step": 68020
    },
    {
      "epoch": 6.486892700195312e-05,
      "step": 68020,
      "training_step_time": 8.831257343292236
    },
    {
      "epoch": 6.486988067626953e-05,
      "model_forward_time": 1.7420783042907715,
      "step": 68021
    },
    {
      "epoch": 6.486988067626953e-05,
      "step": 68021,
      "training_step_time": 7.76032567024231
    },
    {
      "epoch": 6.487083435058594e-05,
      "model_forward_time": 1.7185618877410889,
      "step": 68022
    },
    {
      "epoch": 6.487083435058594e-05,
      "step": 68022,
      "training_step_time": 7.60796594619751
    },
    {
      "epoch": 6.487178802490234e-05,
      "model_forward_time": 1.7257201671600342,
      "step": 68023
    },
    {
      "epoch": 6.487178802490234e-05,
      "step": 68023,
      "training_step_time": 7.587040901184082
    },
    {
      "epoch": 6.487274169921875e-05,
      "model_forward_time": 1.720341682434082,
      "step": 68024
    },
    {
      "epoch": 6.487274169921875e-05,
      "step": 68024,
      "training_step_time": 7.351351022720337
    },
    {
      "epoch": 6.487369537353516e-05,
      "model_forward_time": 1.7172539234161377,
      "step": 68025
    },
    {
      "epoch": 6.487369537353516e-05,
      "step": 68025,
      "training_step_time": 7.153501033782959
    },
    {
      "epoch": 6.487464904785156e-05,
      "model_forward_time": 1.71693754196167,
      "step": 68026
    },
    {
      "epoch": 6.487464904785156e-05,
      "step": 68026,
      "training_step_time": 6.621322393417358
    },
    {
      "epoch": 6.487560272216797e-05,
      "model_forward_time": 1.717512845993042,
      "step": 68027
    },
    {
      "epoch": 6.487560272216797e-05,
      "step": 68027,
      "training_step_time": 7.653820991516113
    },
    {
      "epoch": 6.487655639648438e-05,
      "model_forward_time": 2.895528554916382,
      "step": 68028
    },
    {
      "epoch": 6.487655639648438e-05,
      "step": 68028,
      "training_step_time": 7.719609975814819
    },
    {
      "epoch": 6.487751007080078e-05,
      "model_forward_time": 1.7212181091308594,
      "step": 68029
    },
    {
      "epoch": 6.487751007080078e-05,
      "step": 68029,
      "training_step_time": 8.680249214172363
    },
    {
      "epoch": 6.487846374511719e-05,
      "grad_norm": 0.156297966837883,
      "learning_rate": 2.543704787709329e-06,
      "loss": 0.0605,
      "step": 68030
    },
    {
      "epoch": 6.487846374511719e-05,
      "model_forward_time": 1.7163472175598145,
      "step": 68030
    },
    {
      "dynamics_loss_avg": 0.06410706546157599,
      "epoch": 6.487846374511719e-05,
      "step": 68030
    },
    {
      "action_loss_avg": 0.0025910009542712944,
      "epoch": 6.487846374511719e-05,
      "step": 68030
    },
    {
      "epoch": 6.487846374511719e-05,
      "step": 68030,
      "training_step_time": 8.68785834312439
    },
    {
      "epoch": 6.48794174194336e-05,
      "model_forward_time": 1.7189874649047852,
      "step": 68031
    },
    {
      "epoch": 6.48794174194336e-05,
      "step": 68031,
      "training_step_time": 7.699791669845581
    },
    {
      "epoch": 6.488037109375e-05,
      "model_forward_time": 1.7171814441680908,
      "step": 68032
    },
    {
      "epoch": 6.488037109375e-05,
      "step": 68032,
      "training_step_time": 7.644798755645752
    },
    {
      "epoch": 6.488132476806641e-05,
      "model_forward_time": 1.7166423797607422,
      "step": 68033
    },
    {
      "epoch": 6.488132476806641e-05,
      "step": 68033,
      "training_step_time": 7.314304351806641
    },
    {
      "epoch": 6.488227844238282e-05,
      "model_forward_time": 1.7163867950439453,
      "step": 68034
    },
    {
      "epoch": 6.488227844238282e-05,
      "step": 68034,
      "training_step_time": 7.103679180145264
    },
    {
      "epoch": 6.488323211669922e-05,
      "model_forward_time": 1.7618632316589355,
      "step": 68035
    },
    {
      "epoch": 6.488323211669922e-05,
      "step": 68035,
      "training_step_time": 6.578973770141602
    },
    {
      "epoch": 6.488418579101563e-05,
      "model_forward_time": 1.720595121383667,
      "step": 68036
    },
    {
      "epoch": 6.488418579101563e-05,
      "step": 68036,
      "training_step_time": 7.513359069824219
    },
    {
      "epoch": 6.488513946533204e-05,
      "model_forward_time": 2.9163122177124023,
      "step": 68037
    },
    {
      "epoch": 6.488513946533204e-05,
      "step": 68037,
      "training_step_time": 7.653989315032959
    },
    {
      "epoch": 6.488609313964843e-05,
      "model_forward_time": 1.7178866863250732,
      "step": 68038
    },
    {
      "epoch": 6.488609313964843e-05,
      "step": 68038,
      "training_step_time": 8.572080135345459
    },
    {
      "epoch": 6.488704681396485e-05,
      "model_forward_time": 1.725876808166504,
      "step": 68039
    },
    {
      "epoch": 6.488704681396485e-05,
      "step": 68039,
      "training_step_time": 7.6794517040252686
    },
    {
      "epoch": 6.488800048828126e-05,
      "grad_norm": 0.09932336956262589,
      "learning_rate": 2.542264727476723e-06,
      "loss": 0.0548,
      "step": 68040
    },
    {
      "epoch": 6.488800048828126e-05,
      "model_forward_time": 1.8010258674621582,
      "step": 68040
    },
    {
      "dynamics_loss_avg": 0.05035028401762247,
      "epoch": 6.488800048828126e-05,
      "step": 68040
    },
    {
      "action_loss_avg": 0.0017156052505015395,
      "epoch": 6.488800048828126e-05,
      "step": 68040
    },
    {
      "epoch": 6.488800048828126e-05,
      "step": 68040,
      "training_step_time": 7.806646823883057
    },
    {
      "epoch": 6.488895416259765e-05,
      "model_forward_time": 1.719240427017212,
      "step": 68041
    },
    {
      "epoch": 6.488895416259765e-05,
      "step": 68041,
      "training_step_time": 8.491430282592773
    },
    {
      "epoch": 6.488990783691406e-05,
      "model_forward_time": 1.7508811950683594,
      "step": 68042
    },
    {
      "epoch": 6.488990783691406e-05,
      "step": 68042,
      "training_step_time": 7.315726280212402
    },
    {
      "epoch": 6.489086151123047e-05,
      "model_forward_time": 1.7193033695220947,
      "step": 68043
    },
    {
      "epoch": 6.489086151123047e-05,
      "step": 68043,
      "training_step_time": 7.114012002944946
    },
    {
      "epoch": 6.489181518554687e-05,
      "model_forward_time": 1.7178540229797363,
      "step": 68044
    },
    {
      "epoch": 6.489181518554687e-05,
      "step": 68044,
      "training_step_time": 6.577751636505127
    },
    {
      "epoch": 6.489276885986328e-05,
      "model_forward_time": 1.722348928451538,
      "step": 68045
    },
    {
      "epoch": 6.489276885986328e-05,
      "step": 68045,
      "training_step_time": 7.508604049682617
    },
    {
      "epoch": 6.48937225341797e-05,
      "model_forward_time": 2.8761816024780273,
      "step": 68046
    },
    {
      "epoch": 6.48937225341797e-05,
      "step": 68046,
      "training_step_time": 8.709326982498169
    },
    {
      "epoch": 6.489467620849609e-05,
      "model_forward_time": 1.7222943305969238,
      "step": 68047
    },
    {
      "epoch": 6.489467620849609e-05,
      "step": 68047,
      "training_step_time": 8.45722246170044
    },
    {
      "epoch": 6.48956298828125e-05,
      "model_forward_time": 1.7358081340789795,
      "step": 68048
    },
    {
      "epoch": 6.48956298828125e-05,
      "step": 68048,
      "training_step_time": 8.62932300567627
    },
    {
      "epoch": 6.489658355712891e-05,
      "model_forward_time": 1.8132081031799316,
      "step": 68049
    },
    {
      "epoch": 6.489658355712891e-05,
      "step": 68049,
      "training_step_time": 7.780113220214844
    },
    {
      "epoch": 6.489753723144531e-05,
      "grad_norm": 0.09825210273265839,
      "learning_rate": 2.5408249360183536e-06,
      "loss": 0.0592,
      "step": 68050
    },
    {
      "epoch": 6.489753723144531e-05,
      "model_forward_time": 1.7298436164855957,
      "step": 68050
    },
    {
      "dynamics_loss_avg": 0.059853716567158696,
      "epoch": 6.489753723144531e-05,
      "step": 68050
    },
    {
      "action_loss_avg": 0.0025540331786032765,
      "epoch": 6.489753723144531e-05,
      "step": 68050
    },
    {
      "epoch": 6.489753723144531e-05,
      "step": 68050,
      "training_step_time": 7.6189165115356445
    },
    {
      "epoch": 6.489849090576172e-05,
      "model_forward_time": 1.7417001724243164,
      "step": 68051
    },
    {
      "epoch": 6.489849090576172e-05,
      "step": 68051,
      "training_step_time": 7.52056360244751
    },
    {
      "epoch": 6.489944458007813e-05,
      "model_forward_time": 1.7314674854278564,
      "step": 68052
    },
    {
      "epoch": 6.489944458007813e-05,
      "step": 68052,
      "training_step_time": 6.5486297607421875
    },
    {
      "epoch": 6.490039825439453e-05,
      "model_forward_time": 1.7355716228485107,
      "step": 68053
    },
    {
      "epoch": 6.490039825439453e-05,
      "step": 68053,
      "training_step_time": 7.118384838104248
    },
    {
      "epoch": 6.490135192871094e-05,
      "model_forward_time": 1.7353837490081787,
      "step": 68054
    },
    {
      "epoch": 6.490135192871094e-05,
      "step": 68054,
      "training_step_time": 7.515599012374878
    },
    {
      "epoch": 6.490230560302735e-05,
      "model_forward_time": 2.8225553035736084,
      "step": 68055
    },
    {
      "epoch": 6.490230560302735e-05,
      "step": 68055,
      "training_step_time": 8.360009908676147
    },
    {
      "epoch": 6.490325927734375e-05,
      "model_forward_time": 1.7247822284698486,
      "step": 68056
    },
    {
      "epoch": 6.490325927734375e-05,
      "step": 68056,
      "training_step_time": 8.429606437683105
    },
    {
      "epoch": 6.490421295166016e-05,
      "model_forward_time": 1.72011137008667,
      "step": 68057
    },
    {
      "epoch": 6.490421295166016e-05,
      "step": 68057,
      "training_step_time": 7.630930662155151
    },
    {
      "epoch": 6.490516662597657e-05,
      "model_forward_time": 1.7574388980865479,
      "step": 68058
    },
    {
      "epoch": 6.490516662597657e-05,
      "step": 68058,
      "training_step_time": 8.399713277816772
    },
    {
      "epoch": 6.490612030029297e-05,
      "model_forward_time": 1.7198832035064697,
      "step": 68059
    },
    {
      "epoch": 6.490612030029297e-05,
      "step": 68059,
      "training_step_time": 7.572861671447754
    },
    {
      "epoch": 6.490707397460938e-05,
      "grad_norm": 0.12212272733449936,
      "learning_rate": 2.5393854134916697e-06,
      "loss": 0.0602,
      "step": 68060
    },
    {
      "epoch": 6.490707397460938e-05,
      "model_forward_time": 1.720602035522461,
      "step": 68060
    },
    {
      "dynamics_loss_avg": 0.061325259134173395,
      "epoch": 6.490707397460938e-05,
      "step": 68060
    },
    {
      "action_loss_avg": 0.0013700015842914581,
      "epoch": 6.490707397460938e-05,
      "step": 68060
    },
    {
      "epoch": 6.490707397460938e-05,
      "step": 68060,
      "training_step_time": 7.594099998474121
    },
    {
      "epoch": 6.490802764892579e-05,
      "model_forward_time": 1.7234158515930176,
      "step": 68061
    },
    {
      "epoch": 6.490802764892579e-05,
      "step": 68061,
      "training_step_time": 7.3006370067596436
    },
    {
      "epoch": 6.490898132324219e-05,
      "model_forward_time": 1.7088825702667236,
      "step": 68062
    },
    {
      "epoch": 6.490898132324219e-05,
      "step": 68062,
      "training_step_time": 7.11121392250061
    },
    {
      "epoch": 6.49099349975586e-05,
      "model_forward_time": 1.720292568206787,
      "step": 68063
    },
    {
      "epoch": 6.49099349975586e-05,
      "step": 68063,
      "training_step_time": 7.496545076370239
    },
    {
      "epoch": 6.4910888671875e-05,
      "model_forward_time": 2.8752877712249756,
      "step": 68064
    },
    {
      "epoch": 6.4910888671875e-05,
      "step": 68064,
      "training_step_time": 8.281473398208618
    },
    {
      "epoch": 6.49118423461914e-05,
      "model_forward_time": 1.7256364822387695,
      "step": 68065
    },
    {
      "epoch": 6.49118423461914e-05,
      "step": 68065,
      "training_step_time": 7.60506796836853
    },
    {
      "epoch": 6.491279602050781e-05,
      "model_forward_time": 1.7211861610412598,
      "step": 68066
    },
    {
      "epoch": 6.491279602050781e-05,
      "step": 68066,
      "training_step_time": 7.632503032684326
    },
    {
      "epoch": 6.491374969482421e-05,
      "model_forward_time": 1.7389273643493652,
      "step": 68067
    },
    {
      "epoch": 6.491374969482421e-05,
      "step": 68067,
      "training_step_time": 8.415755271911621
    },
    {
      "epoch": 6.491470336914062e-05,
      "model_forward_time": 1.7187001705169678,
      "step": 68068
    },
    {
      "epoch": 6.491470336914062e-05,
      "step": 68068,
      "training_step_time": 7.5240185260772705
    },
    {
      "epoch": 6.491565704345703e-05,
      "model_forward_time": 1.7214853763580322,
      "step": 68069
    },
    {
      "epoch": 6.491565704345703e-05,
      "step": 68069,
      "training_step_time": 7.508612632751465
    },
    {
      "epoch": 6.491661071777343e-05,
      "grad_norm": 0.09688528627157211,
      "learning_rate": 2.5379461600540996e-06,
      "loss": 0.0555,
      "step": 68070
    },
    {
      "epoch": 6.491661071777343e-05,
      "model_forward_time": 1.7526724338531494,
      "step": 68070
    },
    {
      "dynamics_loss_avg": 0.05384486131370068,
      "epoch": 6.491661071777343e-05,
      "step": 68070
    },
    {
      "action_loss_avg": 0.0018702236644458025,
      "epoch": 6.491661071777343e-05,
      "step": 68070
    },
    {
      "epoch": 6.491661071777343e-05,
      "step": 68070,
      "training_step_time": 7.221593856811523
    },
    {
      "epoch": 6.491756439208984e-05,
      "model_forward_time": 1.7345519065856934,
      "step": 68071
    },
    {
      "epoch": 6.491756439208984e-05,
      "step": 68071,
      "training_step_time": 7.058078765869141
    },
    {
      "epoch": 6.491851806640625e-05,
      "model_forward_time": 1.7221388816833496,
      "step": 68072
    },
    {
      "epoch": 6.491851806640625e-05,
      "step": 68072,
      "training_step_time": 6.728004693984985
    },
    {
      "epoch": 6.491947174072265e-05,
      "model_forward_time": 1.7736256122589111,
      "step": 68073
    },
    {
      "epoch": 6.491947174072265e-05,
      "step": 68073,
      "training_step_time": 10.13625168800354
    },
    {
      "epoch": 6.492042541503906e-05,
      "model_forward_time": 1.7188048362731934,
      "step": 68074
    },
    {
      "epoch": 6.492042541503906e-05,
      "step": 68074,
      "training_step_time": 9.55001187324524
    },
    {
      "epoch": 6.492137908935547e-05,
      "model_forward_time": 1.716928482055664,
      "step": 68075
    },
    {
      "epoch": 6.492137908935547e-05,
      "step": 68075,
      "training_step_time": 8.449913024902344
    },
    {
      "epoch": 6.492233276367187e-05,
      "model_forward_time": 1.724015474319458,
      "step": 68076
    },
    {
      "epoch": 6.492233276367187e-05,
      "step": 68076,
      "training_step_time": 9.254487991333008
    },
    {
      "epoch": 6.492328643798828e-05,
      "model_forward_time": 1.80863356590271,
      "step": 68077
    },
    {
      "epoch": 6.492328643798828e-05,
      "step": 68077,
      "training_step_time": 7.639560699462891
    },
    {
      "epoch": 6.492424011230469e-05,
      "model_forward_time": 1.7254478931427002,
      "step": 68078
    },
    {
      "epoch": 6.492424011230469e-05,
      "step": 68078,
      "training_step_time": 7.653716087341309
    },
    {
      "epoch": 6.492519378662109e-05,
      "model_forward_time": 1.7277843952178955,
      "step": 68079
    },
    {
      "epoch": 6.492519378662109e-05,
      "step": 68079,
      "training_step_time": 7.529875040054321
    },
    {
      "epoch": 6.49261474609375e-05,
      "grad_norm": 0.08809205144643784,
      "learning_rate": 2.5365071758630318e-06,
      "loss": 0.0577,
      "step": 68080
    },
    {
      "epoch": 6.49261474609375e-05,
      "model_forward_time": 1.7210609912872314,
      "step": 68080
    },
    {
      "dynamics_loss_avg": 0.06573056504130363,
      "epoch": 6.49261474609375e-05,
      "step": 68080
    },
    {
      "action_loss_avg": 0.002372664853464812,
      "epoch": 6.49261474609375e-05,
      "step": 68080
    },
    {
      "epoch": 6.49261474609375e-05,
      "step": 68080,
      "training_step_time": 7.137840747833252
    },
    {
      "epoch": 6.492710113525391e-05,
      "model_forward_time": 1.7237787246704102,
      "step": 68081
    },
    {
      "epoch": 6.492710113525391e-05,
      "step": 68081,
      "training_step_time": 6.799348592758179
    },
    {
      "epoch": 6.49280548095703e-05,
      "model_forward_time": 1.7311334609985352,
      "step": 68082
    },
    {
      "epoch": 6.49280548095703e-05,
      "step": 68082,
      "training_step_time": 7.6991050243377686
    },
    {
      "epoch": 6.492900848388672e-05,
      "model_forward_time": 2.503178596496582,
      "step": 68083
    },
    {
      "epoch": 6.492900848388672e-05,
      "step": 68083,
      "training_step_time": 9.772708892822266
    },
    {
      "epoch": 6.492996215820313e-05,
      "model_forward_time": 1.728318214416504,
      "step": 68084
    },
    {
      "epoch": 6.492996215820313e-05,
      "step": 68084,
      "training_step_time": 7.859078645706177
    },
    {
      "epoch": 6.493091583251953e-05,
      "model_forward_time": 1.812986135482788,
      "step": 68085
    },
    {
      "epoch": 6.493091583251953e-05,
      "step": 68085,
      "training_step_time": 8.711153030395508
    },
    {
      "epoch": 6.493186950683594e-05,
      "model_forward_time": 1.7875943183898926,
      "step": 68086
    },
    {
      "epoch": 6.493186950683594e-05,
      "step": 68086,
      "training_step_time": 9.277896642684937
    },
    {
      "epoch": 6.493282318115235e-05,
      "model_forward_time": 1.7397465705871582,
      "step": 68087
    },
    {
      "epoch": 6.493282318115235e-05,
      "step": 68087,
      "training_step_time": 8.617714881896973
    },
    {
      "epoch": 6.493377685546874e-05,
      "model_forward_time": 1.7334246635437012,
      "step": 68088
    },
    {
      "epoch": 6.493377685546874e-05,
      "step": 68088,
      "training_step_time": 7.514084815979004
    },
    {
      "epoch": 6.493473052978515e-05,
      "model_forward_time": 1.7334625720977783,
      "step": 68089
    },
    {
      "epoch": 6.493473052978515e-05,
      "step": 68089,
      "training_step_time": 7.227615833282471
    },
    {
      "epoch": 6.493568420410157e-05,
      "grad_norm": 0.08879362046718597,
      "learning_rate": 2.535068461075839e-06,
      "loss": 0.0582,
      "step": 68090
    },
    {
      "epoch": 6.493568420410157e-05,
      "model_forward_time": 1.7256860733032227,
      "step": 68090
    },
    {
      "dynamics_loss_avg": 0.060666900128126144,
      "epoch": 6.493568420410157e-05,
      "step": 68090
    },
    {
      "action_loss_avg": 0.001405642757890746,
      "epoch": 6.493568420410157e-05,
      "step": 68090
    },
    {
      "epoch": 6.493568420410157e-05,
      "step": 68090,
      "training_step_time": 6.834402799606323
    },
    {
      "epoch": 6.493663787841796e-05,
      "model_forward_time": 1.7403628826141357,
      "step": 68091
    },
    {
      "epoch": 6.493663787841796e-05,
      "step": 68091,
      "training_step_time": 7.5692150592803955
    },
    {
      "epoch": 6.493759155273437e-05,
      "model_forward_time": 2.9147112369537354,
      "step": 68092
    },
    {
      "epoch": 6.493759155273437e-05,
      "step": 68092,
      "training_step_time": 8.281006574630737
    },
    {
      "epoch": 6.493854522705078e-05,
      "model_forward_time": 1.7320806980133057,
      "step": 68093
    },
    {
      "epoch": 6.493854522705078e-05,
      "step": 68093,
      "training_step_time": 7.992276906967163
    },
    {
      "epoch": 6.493949890136718e-05,
      "model_forward_time": 1.7306745052337646,
      "step": 68094
    },
    {
      "epoch": 6.493949890136718e-05,
      "step": 68094,
      "training_step_time": 8.435360193252563
    },
    {
      "epoch": 6.494045257568359e-05,
      "model_forward_time": 1.7443106174468994,
      "step": 68095
    },
    {
      "epoch": 6.494045257568359e-05,
      "step": 68095,
      "training_step_time": 8.77436900138855
    },
    {
      "epoch": 6.494140625e-05,
      "model_forward_time": 1.7311794757843018,
      "step": 68096
    },
    {
      "epoch": 6.494140625e-05,
      "step": 68096,
      "training_step_time": 7.705436706542969
    },
    {
      "epoch": 6.49423599243164e-05,
      "model_forward_time": 1.7340426445007324,
      "step": 68097
    },
    {
      "epoch": 6.49423599243164e-05,
      "step": 68097,
      "training_step_time": 7.593396425247192
    },
    {
      "epoch": 6.494331359863281e-05,
      "model_forward_time": 1.7210886478424072,
      "step": 68098
    },
    {
      "epoch": 6.494331359863281e-05,
      "step": 68098,
      "training_step_time": 7.410999298095703
    },
    {
      "epoch": 6.494426727294922e-05,
      "model_forward_time": 1.734015703201294,
      "step": 68099
    },
    {
      "epoch": 6.494426727294922e-05,
      "step": 68099,
      "training_step_time": 6.709424018859863
    },
    {
      "epoch": 6.494522094726562e-05,
      "grad_norm": 0.3834116756916046,
      "learning_rate": 2.5336300158498518e-06,
      "loss": 0.0572,
      "step": 68100
    },
    {
      "epoch": 6.494522094726562e-05,
      "model_forward_time": 1.7289471626281738,
      "step": 68100
    },
    {
      "dynamics_loss_avg": 0.07201736755669116,
      "epoch": 6.494522094726562e-05,
      "step": 68100
    },
    {
      "action_loss_avg": 0.0014901450253091753,
      "epoch": 6.494522094726562e-05,
      "step": 68100
    },
    {
      "epoch": 6.494522094726562e-05,
      "step": 68100,
      "training_step_time": 8.582518339157104
    },
    {
      "epoch": 6.494617462158203e-05,
      "model_forward_time": 2.867169141769409,
      "step": 68101
    },
    {
      "epoch": 6.494617462158203e-05,
      "step": 68101,
      "training_step_time": 7.706371784210205
    },
    {
      "epoch": 6.494712829589844e-05,
      "model_forward_time": 1.7317416667938232,
      "step": 68102
    },
    {
      "epoch": 6.494712829589844e-05,
      "step": 68102,
      "training_step_time": 8.878234386444092
    },
    {
      "epoch": 6.494808197021484e-05,
      "model_forward_time": 1.7242841720581055,
      "step": 68103
    },
    {
      "epoch": 6.494808197021484e-05,
      "step": 68103,
      "training_step_time": 9.600302457809448
    },
    {
      "epoch": 6.494903564453125e-05,
      "model_forward_time": 1.7348217964172363,
      "step": 68104
    },
    {
      "epoch": 6.494903564453125e-05,
      "step": 68104,
      "training_step_time": 8.291682243347168
    },
    {
      "epoch": 6.494998931884766e-05,
      "model_forward_time": 1.4574174880981445,
      "step": 68105
    },
    {
      "epoch": 6.494998931884766e-05,
      "step": 68105,
      "training_step_time": 8.315653085708618
    },
    {
      "epoch": 6.495094299316406e-05,
      "model_forward_time": 1.7396726608276367,
      "step": 68106
    },
    {
      "epoch": 6.495094299316406e-05,
      "step": 68106,
      "training_step_time": 7.6381378173828125
    },
    {
      "epoch": 6.495189666748047e-05,
      "model_forward_time": 1.7330176830291748,
      "step": 68107
    },
    {
      "epoch": 6.495189666748047e-05,
      "step": 68107,
      "training_step_time": 7.303758144378662
    },
    {
      "epoch": 6.495285034179688e-05,
      "model_forward_time": 1.7348294258117676,
      "step": 68108
    },
    {
      "epoch": 6.495285034179688e-05,
      "step": 68108,
      "training_step_time": 6.769374847412109
    },
    {
      "epoch": 6.495380401611328e-05,
      "model_forward_time": 1.7329261302947998,
      "step": 68109
    },
    {
      "epoch": 6.495380401611328e-05,
      "step": 68109,
      "training_step_time": 7.448768854141235
    },
    {
      "epoch": 6.495475769042969e-05,
      "grad_norm": 0.1739233434200287,
      "learning_rate": 2.5321918403423793e-06,
      "loss": 0.056,
      "step": 68110
    },
    {
      "epoch": 6.495475769042969e-05,
      "model_forward_time": 2.8047354221343994,
      "step": 68110
    },
    {
      "dynamics_loss_avg": 0.06797654088586569,
      "epoch": 6.495475769042969e-05,
      "step": 68110
    },
    {
      "action_loss_avg": 0.0020739637839142233,
      "epoch": 6.495475769042969e-05,
      "step": 68110
    },
    {
      "epoch": 6.495475769042969e-05,
      "step": 68110,
      "training_step_time": 7.568024396896362
    },
    {
      "epoch": 6.49557113647461e-05,
      "model_forward_time": 1.7416725158691406,
      "step": 68111
    },
    {
      "epoch": 6.49557113647461e-05,
      "step": 68111,
      "training_step_time": 8.567150115966797
    },
    {
      "epoch": 6.49566650390625e-05,
      "model_forward_time": 1.7169756889343262,
      "step": 68112
    },
    {
      "epoch": 6.49566650390625e-05,
      "step": 68112,
      "training_step_time": 9.446241617202759
    },
    {
      "epoch": 6.49576187133789e-05,
      "model_forward_time": 1.7373316287994385,
      "step": 68113
    },
    {
      "epoch": 6.49576187133789e-05,
      "step": 68113,
      "training_step_time": 9.505273342132568
    },
    {
      "epoch": 6.495857238769532e-05,
      "model_forward_time": 1.80106520652771,
      "step": 68114
    },
    {
      "epoch": 6.495857238769532e-05,
      "step": 68114,
      "training_step_time": 7.548161745071411
    },
    {
      "epoch": 6.495952606201171e-05,
      "model_forward_time": 1.7325189113616943,
      "step": 68115
    },
    {
      "epoch": 6.495952606201171e-05,
      "step": 68115,
      "training_step_time": 7.689244985580444
    },
    {
      "epoch": 6.496047973632812e-05,
      "model_forward_time": 1.7247507572174072,
      "step": 68116
    },
    {
      "epoch": 6.496047973632812e-05,
      "step": 68116,
      "training_step_time": 7.456248760223389
    },
    {
      "epoch": 6.496143341064454e-05,
      "model_forward_time": 1.7424983978271484,
      "step": 68117
    },
    {
      "epoch": 6.496143341064454e-05,
      "step": 68117,
      "training_step_time": 6.720654726028442
    },
    {
      "epoch": 6.496238708496093e-05,
      "model_forward_time": 1.720776081085205,
      "step": 68118
    },
    {
      "epoch": 6.496238708496093e-05,
      "step": 68118,
      "training_step_time": 7.0231711864471436
    },
    {
      "epoch": 6.496334075927734e-05,
      "model_forward_time": 1.7133750915527344,
      "step": 68119
    },
    {
      "epoch": 6.496334075927734e-05,
      "step": 68119,
      "training_step_time": 8.401657819747925
    },
    {
      "epoch": 6.496429443359375e-05,
      "grad_norm": 0.08422674983739853,
      "learning_rate": 2.5307539347106957e-06,
      "loss": 0.0568,
      "step": 68120
    },
    {
      "epoch": 6.496429443359375e-05,
      "model_forward_time": 2.866461992263794,
      "step": 68120
    },
    {
      "dynamics_loss_avg": 0.060396283864974976,
      "epoch": 6.496429443359375e-05,
      "step": 68120
    },
    {
      "action_loss_avg": 0.002664701966568828,
      "epoch": 6.496429443359375e-05,
      "step": 68120
    },
    {
      "epoch": 6.496429443359375e-05,
      "step": 68120,
      "training_step_time": 8.607394456863403
    },
    {
      "epoch": 6.496524810791015e-05,
      "model_forward_time": 1.7364788055419922,
      "step": 68121
    },
    {
      "epoch": 6.496524810791015e-05,
      "step": 68121,
      "training_step_time": 8.791759967803955
    },
    {
      "epoch": 6.496620178222656e-05,
      "model_forward_time": 1.738692045211792,
      "step": 68122
    },
    {
      "epoch": 6.496620178222656e-05,
      "step": 68122,
      "training_step_time": 8.476596593856812
    },
    {
      "epoch": 6.496715545654297e-05,
      "model_forward_time": 1.81156325340271,
      "step": 68123
    },
    {
      "epoch": 6.496715545654297e-05,
      "step": 68123,
      "training_step_time": 7.638710021972656
    },
    {
      "epoch": 6.496810913085937e-05,
      "model_forward_time": 1.7298576831817627,
      "step": 68124
    },
    {
      "epoch": 6.496810913085937e-05,
      "step": 68124,
      "training_step_time": 7.694498300552368
    },
    {
      "epoch": 6.496906280517578e-05,
      "model_forward_time": 1.7443797588348389,
      "step": 68125
    },
    {
      "epoch": 6.496906280517578e-05,
      "step": 68125,
      "training_step_time": 7.3919594287872314
    },
    {
      "epoch": 6.497001647949219e-05,
      "model_forward_time": 1.3773822784423828,
      "step": 68126
    },
    {
      "epoch": 6.497001647949219e-05,
      "step": 68126,
      "training_step_time": 6.764617443084717
    },
    {
      "epoch": 6.497097015380859e-05,
      "model_forward_time": 1.7282681465148926,
      "step": 68127
    },
    {
      "epoch": 6.497097015380859e-05,
      "step": 68127,
      "training_step_time": 7.362031936645508
    },
    {
      "epoch": 6.4971923828125e-05,
      "model_forward_time": 1.743626356124878,
      "step": 68128
    },
    {
      "epoch": 6.4971923828125e-05,
      "step": 68128,
      "training_step_time": 8.471813678741455
    },
    {
      "epoch": 6.497287750244141e-05,
      "model_forward_time": 2.857393741607666,
      "step": 68129
    },
    {
      "epoch": 6.497287750244141e-05,
      "step": 68129,
      "training_step_time": 10.169564008712769
    },
    {
      "epoch": 6.497383117675781e-05,
      "grad_norm": 0.08676434308290482,
      "learning_rate": 2.529316299112049e-06,
      "loss": 0.0574,
      "step": 68130
    },
    {
      "epoch": 6.497383117675781e-05,
      "model_forward_time": 1.7290520668029785,
      "step": 68130
    },
    {
      "dynamics_loss_avg": 0.06586997658014297,
      "epoch": 6.497383117675781e-05,
      "step": 68130
    },
    {
      "action_loss_avg": 0.0019161678617820143,
      "epoch": 6.497383117675781e-05,
      "step": 68130
    },
    {
      "epoch": 6.497383117675781e-05,
      "step": 68130,
      "training_step_time": 9.170836925506592
    },
    {
      "epoch": 6.497478485107422e-05,
      "model_forward_time": 1.7523176670074463,
      "step": 68131
    },
    {
      "epoch": 6.497478485107422e-05,
      "step": 68131,
      "training_step_time": 7.81768274307251
    },
    {
      "epoch": 6.497573852539063e-05,
      "model_forward_time": 1.3811633586883545,
      "step": 68132
    },
    {
      "epoch": 6.497573852539063e-05,
      "step": 68132,
      "training_step_time": 7.552142381668091
    },
    {
      "epoch": 6.497669219970703e-05,
      "model_forward_time": 1.7532386779785156,
      "step": 68133
    },
    {
      "epoch": 6.497669219970703e-05,
      "step": 68133,
      "training_step_time": 7.544997930526733
    },
    {
      "epoch": 6.497764587402344e-05,
      "model_forward_time": 1.7534151077270508,
      "step": 68134
    },
    {
      "epoch": 6.497764587402344e-05,
      "step": 68134,
      "training_step_time": 9.391429662704468
    },
    {
      "epoch": 6.497859954833985e-05,
      "model_forward_time": 1.7530791759490967,
      "step": 68135
    },
    {
      "epoch": 6.497859954833985e-05,
      "step": 68135,
      "training_step_time": 7.314847469329834
    },
    {
      "epoch": 6.497955322265625e-05,
      "model_forward_time": 1.7538244724273682,
      "step": 68136
    },
    {
      "epoch": 6.497955322265625e-05,
      "step": 68136,
      "training_step_time": 6.768251180648804
    },
    {
      "epoch": 6.498050689697266e-05,
      "model_forward_time": 1.7532362937927246,
      "step": 68137
    },
    {
      "epoch": 6.498050689697266e-05,
      "step": 68137,
      "training_step_time": 8.15925908088684
    },
    {
      "epoch": 6.498146057128907e-05,
      "model_forward_time": 2.891636610031128,
      "step": 68138
    },
    {
      "epoch": 6.498146057128907e-05,
      "step": 68138,
      "training_step_time": 7.6088666915893555
    },
    {
      "epoch": 6.498241424560546e-05,
      "model_forward_time": 1.743344783782959,
      "step": 68139
    },
    {
      "epoch": 6.498241424560546e-05,
      "step": 68139,
      "training_step_time": 8.987114906311035
    },
    {
      "epoch": 6.498336791992188e-05,
      "grad_norm": 0.08497625589370728,
      "learning_rate": 2.527878933703657e-06,
      "loss": 0.0541,
      "step": 68140
    },
    {
      "epoch": 6.498336791992188e-05,
      "model_forward_time": 1.734579086303711,
      "step": 68140
    },
    {
      "dynamics_loss_avg": 0.0579071793705225,
      "epoch": 6.498336791992188e-05,
      "step": 68140
    },
    {
      "action_loss_avg": 0.001801849715411663,
      "epoch": 6.498336791992188e-05,
      "step": 68140
    },
    {
      "epoch": 6.498336791992188e-05,
      "step": 68140,
      "training_step_time": 9.448471546173096
    },
    {
      "epoch": 6.498432159423829e-05,
      "model_forward_time": 1.0154321193695068,
      "step": 68141
    },
    {
      "epoch": 6.498432159423829e-05,
      "step": 68141,
      "training_step_time": 7.729937791824341
    },
    {
      "epoch": 6.498527526855468e-05,
      "model_forward_time": 1.7175509929656982,
      "step": 68142
    },
    {
      "epoch": 6.498527526855468e-05,
      "step": 68142,
      "training_step_time": 8.653733253479004
    },
    {
      "epoch": 6.49862289428711e-05,
      "model_forward_time": 1.7184255123138428,
      "step": 68143
    },
    {
      "epoch": 6.49862289428711e-05,
      "step": 68143,
      "training_step_time": 7.544980525970459
    },
    {
      "epoch": 6.49871826171875e-05,
      "model_forward_time": 1.7419545650482178,
      "step": 68144
    },
    {
      "epoch": 6.49871826171875e-05,
      "step": 68144,
      "training_step_time": 7.376244068145752
    },
    {
      "epoch": 6.49881362915039e-05,
      "model_forward_time": 1.7330536842346191,
      "step": 68145
    },
    {
      "epoch": 6.49881362915039e-05,
      "step": 68145,
      "training_step_time": 6.69679069519043
    },
    {
      "epoch": 6.498908996582031e-05,
      "model_forward_time": 1.729694128036499,
      "step": 68146
    },
    {
      "epoch": 6.498908996582031e-05,
      "step": 68146,
      "training_step_time": 7.548134088516235
    },
    {
      "epoch": 6.499004364013672e-05,
      "model_forward_time": 1.7473978996276855,
      "step": 68147
    },
    {
      "epoch": 6.499004364013672e-05,
      "step": 68147,
      "training_step_time": 9.444922924041748
    },
    {
      "epoch": 6.499099731445312e-05,
      "model_forward_time": 1.7112643718719482,
      "step": 68148
    },
    {
      "epoch": 6.499099731445312e-05,
      "step": 68148,
      "training_step_time": 9.856566667556763
    },
    {
      "epoch": 6.499195098876953e-05,
      "model_forward_time": 1.7305066585540771,
      "step": 68149
    },
    {
      "epoch": 6.499195098876953e-05,
      "step": 68149,
      "training_step_time": 8.530614137649536
    },
    {
      "epoch": 6.499290466308594e-05,
      "grad_norm": 0.11892613768577576,
      "learning_rate": 2.5264418386427104e-06,
      "loss": 0.0553,
      "step": 68150
    },
    {
      "epoch": 6.499290466308594e-05,
      "model_forward_time": 1.7186863422393799,
      "step": 68150
    },
    {
      "dynamics_loss_avg": 0.0773774846456945,
      "epoch": 6.499290466308594e-05,
      "step": 68150
    },
    {
      "action_loss_avg": 0.002191777201369405,
      "epoch": 6.499290466308594e-05,
      "step": 68150
    },
    {
      "epoch": 6.499290466308594e-05,
      "step": 68150,
      "training_step_time": 8.200613975524902
    },
    {
      "epoch": 6.499385833740234e-05,
      "model_forward_time": 1.7929482460021973,
      "step": 68151
    },
    {
      "epoch": 6.499385833740234e-05,
      "step": 68151,
      "training_step_time": 7.530694484710693
    },
    {
      "epoch": 6.499481201171875e-05,
      "model_forward_time": 1.7395334243774414,
      "step": 68152
    },
    {
      "epoch": 6.499481201171875e-05,
      "step": 68152,
      "training_step_time": 7.695506811141968
    },
    {
      "epoch": 6.499576568603516e-05,
      "model_forward_time": 1.7393226623535156,
      "step": 68153
    },
    {
      "epoch": 6.499576568603516e-05,
      "step": 68153,
      "training_step_time": 7.3763368129730225
    },
    {
      "epoch": 6.499671936035156e-05,
      "model_forward_time": 1.7357816696166992,
      "step": 68154
    },
    {
      "epoch": 6.499671936035156e-05,
      "step": 68154,
      "training_step_time": 6.688699245452881
    },
    {
      "epoch": 6.499767303466797e-05,
      "model_forward_time": 1.718982219696045,
      "step": 68155
    },
    {
      "epoch": 6.499767303466797e-05,
      "step": 68155,
      "training_step_time": 8.5171058177948
    },
    {
      "epoch": 6.499862670898438e-05,
      "model_forward_time": 1.735971212387085,
      "step": 68156
    },
    {
      "epoch": 6.499862670898438e-05,
      "step": 68156,
      "training_step_time": 9.281192064285278
    },
    {
      "epoch": 6.499958038330078e-05,
      "model_forward_time": 2.883084535598755,
      "step": 68157
    },
    {
      "epoch": 6.499958038330078e-05,
      "step": 68157,
      "training_step_time": 7.774301290512085
    },
    {
      "epoch": 6.500053405761719e-05,
      "model_forward_time": 1.7255568504333496,
      "step": 68158
    },
    {
      "epoch": 6.500053405761719e-05,
      "step": 68158,
      "training_step_time": 7.589505434036255
    },
    {
      "epoch": 6.50014877319336e-05,
      "model_forward_time": 1.74373197555542,
      "step": 68159
    },
    {
      "epoch": 6.50014877319336e-05,
      "step": 68159,
      "training_step_time": 7.634819746017456
    },
    {
      "epoch": 6.500244140625e-05,
      "grad_norm": 0.10511309653520584,
      "learning_rate": 2.525005014086363e-06,
      "loss": 0.0528,
      "step": 68160
    },
    {
      "epoch": 6.500244140625e-05,
      "model_forward_time": 1.783611536026001,
      "step": 68160
    },
    {
      "dynamics_loss_avg": 0.046991529129445556,
      "epoch": 6.500244140625e-05,
      "step": 68160
    },
    {
      "action_loss_avg": 0.002210068082786165,
      "epoch": 6.500244140625e-05,
      "step": 68160
    },
    {
      "epoch": 6.500244140625e-05,
      "step": 68160,
      "training_step_time": 7.585230112075806
    },
    {
      "epoch": 6.500339508056641e-05,
      "model_forward_time": 1.7428100109100342,
      "step": 68161
    },
    {
      "epoch": 6.500339508056641e-05,
      "step": 68161,
      "training_step_time": 7.713734865188599
    },
    {
      "epoch": 6.500434875488282e-05,
      "model_forward_time": 1.735459327697754,
      "step": 68162
    },
    {
      "epoch": 6.500434875488282e-05,
      "step": 68162,
      "training_step_time": 7.3456878662109375
    },
    {
      "epoch": 6.500530242919922e-05,
      "model_forward_time": 1.744265079498291,
      "step": 68163
    },
    {
      "epoch": 6.500530242919922e-05,
      "step": 68163,
      "training_step_time": 6.723877906799316
    },
    {
      "epoch": 6.500625610351563e-05,
      "model_forward_time": 1.7492969036102295,
      "step": 68164
    },
    {
      "epoch": 6.500625610351563e-05,
      "step": 68164,
      "training_step_time": 7.451476573944092
    },
    {
      "epoch": 6.500720977783204e-05,
      "model_forward_time": 1.7469873428344727,
      "step": 68165
    },
    {
      "epoch": 6.500720977783204e-05,
      "step": 68165,
      "training_step_time": 8.568730592727661
    },
    {
      "epoch": 6.500816345214843e-05,
      "model_forward_time": 2.835999011993408,
      "step": 68166
    },
    {
      "epoch": 6.500816345214843e-05,
      "step": 68166,
      "training_step_time": 8.613077402114868
    },
    {
      "epoch": 6.500911712646485e-05,
      "model_forward_time": 1.719864845275879,
      "step": 68167
    },
    {
      "epoch": 6.500911712646485e-05,
      "step": 68167,
      "training_step_time": 8.258805751800537
    },
    {
      "epoch": 6.501007080078126e-05,
      "model_forward_time": 1.751279592514038,
      "step": 68168
    },
    {
      "epoch": 6.501007080078126e-05,
      "step": 68168,
      "training_step_time": 9.365398645401001
    },
    {
      "epoch": 6.501102447509765e-05,
      "model_forward_time": 1.8050580024719238,
      "step": 68169
    },
    {
      "epoch": 6.501102447509765e-05,
      "step": 68169,
      "training_step_time": 8.315436601638794
    },
    {
      "epoch": 6.501197814941406e-05,
      "grad_norm": 0.0849713683128357,
      "learning_rate": 2.5235684601917465e-06,
      "loss": 0.0537,
      "step": 68170
    },
    {
      "epoch": 6.501197814941406e-05,
      "model_forward_time": 1.7364585399627686,
      "step": 68170
    },
    {
      "dynamics_loss_avg": 0.05053453547880053,
      "epoch": 6.501197814941406e-05,
      "step": 68170
    },
    {
      "action_loss_avg": 0.0013102509372401982,
      "epoch": 6.501197814941406e-05,
      "step": 68170
    },
    {
      "epoch": 6.501197814941406e-05,
      "step": 68170,
      "training_step_time": 7.686639308929443
    },
    {
      "epoch": 6.501293182373047e-05,
      "model_forward_time": 1.742835521697998,
      "step": 68171
    },
    {
      "epoch": 6.501293182373047e-05,
      "step": 68171,
      "training_step_time": 8.187445878982544
    },
    {
      "epoch": 6.501388549804687e-05,
      "model_forward_time": 1.7517664432525635,
      "step": 68172
    },
    {
      "epoch": 6.501388549804687e-05,
      "step": 68172,
      "training_step_time": 7.404759168624878
    },
    {
      "epoch": 6.501483917236328e-05,
      "model_forward_time": 1.7511632442474365,
      "step": 68173
    },
    {
      "epoch": 6.501483917236328e-05,
      "step": 68173,
      "training_step_time": 7.1638524532318115
    },
    {
      "epoch": 6.50157928466797e-05,
      "model_forward_time": 1.7485721111297607,
      "step": 68174
    },
    {
      "epoch": 6.50157928466797e-05,
      "step": 68174,
      "training_step_time": 7.634003639221191
    },
    {
      "epoch": 6.501674652099609e-05,
      "model_forward_time": 2.9104385375976562,
      "step": 68175
    },
    {
      "epoch": 6.501674652099609e-05,
      "step": 68175,
      "training_step_time": 7.643110036849976
    },
    {
      "epoch": 6.50177001953125e-05,
      "model_forward_time": 1.7510099411010742,
      "step": 68176
    },
    {
      "epoch": 6.50177001953125e-05,
      "step": 68176,
      "training_step_time": 7.777980089187622
    },
    {
      "epoch": 6.501865386962891e-05,
      "model_forward_time": 1.7500317096710205,
      "step": 68177
    },
    {
      "epoch": 6.501865386962891e-05,
      "step": 68177,
      "training_step_time": 10.00871992111206
    },
    {
      "epoch": 6.501960754394531e-05,
      "model_forward_time": 1.7813918590545654,
      "step": 68178
    },
    {
      "epoch": 6.501960754394531e-05,
      "step": 68178,
      "training_step_time": 8.4025137424469
    },
    {
      "epoch": 6.502056121826172e-05,
      "model_forward_time": 1.7474517822265625,
      "step": 68179
    },
    {
      "epoch": 6.502056121826172e-05,
      "step": 68179,
      "training_step_time": 7.702120780944824
    },
    {
      "epoch": 6.502151489257813e-05,
      "grad_norm": 0.08658261597156525,
      "learning_rate": 2.522132177115962e-06,
      "loss": 0.0529,
      "step": 68180
    },
    {
      "epoch": 6.502151489257813e-05,
      "model_forward_time": 1.7460904121398926,
      "step": 68180
    },
    {
      "dynamics_loss_avg": 0.05706932246685028,
      "epoch": 6.502151489257813e-05,
      "step": 68180
    },
    {
      "action_loss_avg": 0.00151456919265911,
      "epoch": 6.502151489257813e-05,
      "step": 68180
    },
    {
      "epoch": 6.502151489257813e-05,
      "step": 68180,
      "training_step_time": 8.646450757980347
    },
    {
      "epoch": 6.502246856689453e-05,
      "model_forward_time": 1.7383754253387451,
      "step": 68181
    },
    {
      "epoch": 6.502246856689453e-05,
      "step": 68181,
      "training_step_time": 7.447662830352783
    },
    {
      "epoch": 6.502342224121094e-05,
      "model_forward_time": 1.7359492778778076,
      "step": 68182
    },
    {
      "epoch": 6.502342224121094e-05,
      "step": 68182,
      "training_step_time": 7.595418453216553
    },
    {
      "epoch": 6.502437591552735e-05,
      "model_forward_time": 1.733942985534668,
      "step": 68183
    },
    {
      "epoch": 6.502437591552735e-05,
      "step": 68183,
      "training_step_time": 7.697784662246704
    },
    {
      "epoch": 6.502532958984375e-05,
      "model_forward_time": 2.9016199111938477,
      "step": 68184
    },
    {
      "epoch": 6.502532958984375e-05,
      "step": 68184,
      "training_step_time": 7.69977068901062
    },
    {
      "epoch": 6.502628326416016e-05,
      "model_forward_time": 1.7416188716888428,
      "step": 68185
    },
    {
      "epoch": 6.502628326416016e-05,
      "step": 68185,
      "training_step_time": 7.908618927001953
    },
    {
      "epoch": 6.502723693847657e-05,
      "model_forward_time": 1.7303717136383057,
      "step": 68186
    },
    {
      "epoch": 6.502723693847657e-05,
      "step": 68186,
      "training_step_time": 8.787505626678467
    },
    {
      "epoch": 6.502819061279297e-05,
      "model_forward_time": 1.7475483417510986,
      "step": 68187
    },
    {
      "epoch": 6.502819061279297e-05,
      "step": 68187,
      "training_step_time": 8.989966630935669
    },
    {
      "epoch": 6.502914428710938e-05,
      "model_forward_time": 1.7361302375793457,
      "step": 68188
    },
    {
      "epoch": 6.502914428710938e-05,
      "step": 68188,
      "training_step_time": 7.708246231079102
    },
    {
      "epoch": 6.503009796142577e-05,
      "model_forward_time": 1.741408109664917,
      "step": 68189
    },
    {
      "epoch": 6.503009796142577e-05,
      "step": 68189,
      "training_step_time": 7.661846876144409
    },
    {
      "epoch": 6.503105163574219e-05,
      "grad_norm": 0.09330250322818756,
      "learning_rate": 2.5206961650160744e-06,
      "loss": 0.0582,
      "step": 68190
    },
    {
      "epoch": 6.503105163574219e-05,
      "model_forward_time": 1.7425265312194824,
      "step": 68190
    },
    {
      "dynamics_loss_avg": 0.05092110224068165,
      "epoch": 6.503105163574219e-05,
      "step": 68190
    },
    {
      "action_loss_avg": 0.0012483913800679148,
      "epoch": 6.503105163574219e-05,
      "step": 68190
    },
    {
      "epoch": 6.503105163574219e-05,
      "step": 68190,
      "training_step_time": 7.593691825866699
    },
    {
      "epoch": 6.50320053100586e-05,
      "model_forward_time": 1.7358131408691406,
      "step": 68191
    },
    {
      "epoch": 6.50320053100586e-05,
      "step": 68191,
      "training_step_time": 7.643571853637695
    },
    {
      "epoch": 6.5032958984375e-05,
      "model_forward_time": 1.733510971069336,
      "step": 68192
    },
    {
      "epoch": 6.5032958984375e-05,
      "step": 68192,
      "training_step_time": 8.645091533660889
    },
    {
      "epoch": 6.50339126586914e-05,
      "model_forward_time": 2.8568060398101807,
      "step": 68193
    },
    {
      "epoch": 6.50339126586914e-05,
      "step": 68193,
      "training_step_time": 8.12390422821045
    },
    {
      "epoch": 6.503486633300781e-05,
      "model_forward_time": 1.7400035858154297,
      "step": 68194
    },
    {
      "epoch": 6.503486633300781e-05,
      "step": 68194,
      "training_step_time": 9.63380765914917
    },
    {
      "epoch": 6.503582000732421e-05,
      "model_forward_time": 1.738325595855713,
      "step": 68195
    },
    {
      "epoch": 6.503582000732421e-05,
      "step": 68195,
      "training_step_time": 8.929582595825195
    },
    {
      "epoch": 6.503677368164062e-05,
      "model_forward_time": 1.7374703884124756,
      "step": 68196
    },
    {
      "epoch": 6.503677368164062e-05,
      "step": 68196,
      "training_step_time": 8.376008033752441
    },
    {
      "epoch": 6.503772735595703e-05,
      "model_forward_time": 1.8052928447723389,
      "step": 68197
    },
    {
      "epoch": 6.503772735595703e-05,
      "step": 68197,
      "training_step_time": 9.161049127578735
    },
    {
      "epoch": 6.503868103027343e-05,
      "model_forward_time": 1.730027198791504,
      "step": 68198
    },
    {
      "epoch": 6.503868103027343e-05,
      "step": 68198,
      "training_step_time": 7.680256128311157
    },
    {
      "epoch": 6.503963470458984e-05,
      "model_forward_time": 1.7346034049987793,
      "step": 68199
    },
    {
      "epoch": 6.503963470458984e-05,
      "step": 68199,
      "training_step_time": 7.6598145961761475
    },
    {
      "epoch": 6.504058837890625e-05,
      "grad_norm": 0.10279447585344315,
      "learning_rate": 2.5192604240491294e-06,
      "loss": 0.0558,
      "step": 68200
    },
    {
      "epoch": 6.504058837890625e-05,
      "model_forward_time": 1.7392990589141846,
      "step": 68200
    },
    {
      "dynamics_loss_avg": 0.06160362623631954,
      "epoch": 6.504058837890625e-05,
      "step": 68200
    },
    {
      "action_loss_avg": 0.0015272425021976232,
      "epoch": 6.504058837890625e-05,
      "step": 68200
    },
    {
      "epoch": 6.504058837890625e-05,
      "step": 68200,
      "training_step_time": 7.548314332962036
    },
    {
      "epoch": 6.504154205322265e-05,
      "model_forward_time": 1.7307848930358887,
      "step": 68201
    },
    {
      "epoch": 6.504154205322265e-05,
      "step": 68201,
      "training_step_time": 7.6207921504974365
    },
    {
      "epoch": 6.504249572753906e-05,
      "model_forward_time": 1.7327826023101807,
      "step": 68202
    },
    {
      "epoch": 6.504249572753906e-05,
      "step": 68202,
      "training_step_time": 8.279785633087158
    },
    {
      "epoch": 6.504344940185547e-05,
      "model_forward_time": 1.727750301361084,
      "step": 68203
    },
    {
      "epoch": 6.504344940185547e-05,
      "step": 68203,
      "training_step_time": 9.83904242515564
    },
    {
      "epoch": 6.504440307617187e-05,
      "model_forward_time": 1.7437551021575928,
      "step": 68204
    },
    {
      "epoch": 6.504440307617187e-05,
      "step": 68204,
      "training_step_time": 7.877867937088013
    },
    {
      "epoch": 6.504535675048828e-05,
      "model_forward_time": 1.369062900543213,
      "step": 68205
    },
    {
      "epoch": 6.504535675048828e-05,
      "step": 68205,
      "training_step_time": 7.754433631896973
    },
    {
      "epoch": 6.504631042480469e-05,
      "model_forward_time": 1.8136820793151855,
      "step": 68206
    },
    {
      "epoch": 6.504631042480469e-05,
      "step": 68206,
      "training_step_time": 7.662580490112305
    },
    {
      "epoch": 6.504726409912109e-05,
      "model_forward_time": 1.734503984451294,
      "step": 68207
    },
    {
      "epoch": 6.504726409912109e-05,
      "step": 68207,
      "training_step_time": 8.709940671920776
    },
    {
      "epoch": 6.50482177734375e-05,
      "model_forward_time": 1.7326619625091553,
      "step": 68208
    },
    {
      "epoch": 6.50482177734375e-05,
      "step": 68208,
      "training_step_time": 7.632054090499878
    },
    {
      "epoch": 6.504917144775391e-05,
      "model_forward_time": 1.7486820220947266,
      "step": 68209
    },
    {
      "epoch": 6.504917144775391e-05,
      "step": 68209,
      "training_step_time": 7.592102766036987
    },
    {
      "epoch": 6.50501251220703e-05,
      "grad_norm": 0.1204085573554039,
      "learning_rate": 2.5178249543721307e-06,
      "loss": 0.0548,
      "step": 68210
    },
    {
      "epoch": 6.50501251220703e-05,
      "model_forward_time": 1.7286155223846436,
      "step": 68210
    },
    {
      "dynamics_loss_avg": 0.057072573155164716,
      "epoch": 6.50501251220703e-05,
      "step": 68210
    },
    {
      "action_loss_avg": 0.0027714081923477353,
      "epoch": 6.50501251220703e-05,
      "step": 68210
    },
    {
      "epoch": 6.50501251220703e-05,
      "step": 68210,
      "training_step_time": 7.121756553649902
    },
    {
      "epoch": 6.505107879638672e-05,
      "model_forward_time": 1.7371320724487305,
      "step": 68211
    },
    {
      "epoch": 6.505107879638672e-05,
      "step": 68211,
      "training_step_time": 8.576320886611938
    },
    {
      "epoch": 6.505203247070313e-05,
      "model_forward_time": 2.888345241546631,
      "step": 68212
    },
    {
      "epoch": 6.505203247070313e-05,
      "step": 68212,
      "training_step_time": 8.328668594360352
    },
    {
      "epoch": 6.505298614501953e-05,
      "model_forward_time": 1.7379789352416992,
      "step": 68213
    },
    {
      "epoch": 6.505298614501953e-05,
      "step": 68213,
      "training_step_time": 8.43723440170288
    },
    {
      "epoch": 6.505393981933594e-05,
      "model_forward_time": 1.7227895259857178,
      "step": 68214
    },
    {
      "epoch": 6.505393981933594e-05,
      "step": 68214,
      "training_step_time": 7.679873943328857
    },
    {
      "epoch": 6.505489349365235e-05,
      "model_forward_time": 1.748595952987671,
      "step": 68215
    },
    {
      "epoch": 6.505489349365235e-05,
      "step": 68215,
      "training_step_time": 7.586975574493408
    },
    {
      "epoch": 6.505584716796874e-05,
      "model_forward_time": 1.7349302768707275,
      "step": 68216
    },
    {
      "epoch": 6.505584716796874e-05,
      "step": 68216,
      "training_step_time": 8.233706712722778
    },
    {
      "epoch": 6.505680084228516e-05,
      "model_forward_time": 1.7409915924072266,
      "step": 68217
    },
    {
      "epoch": 6.505680084228516e-05,
      "step": 68217,
      "training_step_time": 7.534384489059448
    },
    {
      "epoch": 6.505775451660157e-05,
      "model_forward_time": 1.7308862209320068,
      "step": 68218
    },
    {
      "epoch": 6.505775451660157e-05,
      "step": 68218,
      "training_step_time": 7.450251817703247
    },
    {
      "epoch": 6.505870819091796e-05,
      "model_forward_time": 1.7262303829193115,
      "step": 68219
    },
    {
      "epoch": 6.505870819091796e-05,
      "step": 68219,
      "training_step_time": 6.973348140716553
    },
    {
      "epoch": 6.505966186523437e-05,
      "grad_norm": 0.0887565165758133,
      "learning_rate": 2.5163897561420668e-06,
      "loss": 0.0564,
      "step": 68220
    },
    {
      "epoch": 6.505966186523437e-05,
      "model_forward_time": 1.7321629524230957,
      "step": 68220
    },
    {
      "dynamics_loss_avg": 0.05660277716815472,
      "epoch": 6.505966186523437e-05,
      "step": 68220
    },
    {
      "action_loss_avg": 0.0014417330268770457,
      "epoch": 6.505966186523437e-05,
      "step": 68220
    },
    {
      "epoch": 6.505966186523437e-05,
      "step": 68220,
      "training_step_time": 7.604719638824463
    },
    {
      "epoch": 6.506061553955078e-05,
      "model_forward_time": 2.882596015930176,
      "step": 68221
    },
    {
      "epoch": 6.506061553955078e-05,
      "step": 68221,
      "training_step_time": 7.663440465927124
    },
    {
      "epoch": 6.506156921386718e-05,
      "model_forward_time": 1.7305643558502197,
      "step": 68222
    },
    {
      "epoch": 6.506156921386718e-05,
      "step": 68222,
      "training_step_time": 8.232667446136475
    },
    {
      "epoch": 6.506252288818359e-05,
      "model_forward_time": 1.727107048034668,
      "step": 68223
    },
    {
      "epoch": 6.506252288818359e-05,
      "step": 68223,
      "training_step_time": 8.816958665847778
    },
    {
      "epoch": 6.50634765625e-05,
      "model_forward_time": 1.7395071983337402,
      "step": 68224
    },
    {
      "epoch": 6.50634765625e-05,
      "step": 68224,
      "training_step_time": 7.6490583419799805
    },
    {
      "epoch": 6.50644302368164e-05,
      "model_forward_time": 1.7307109832763672,
      "step": 68225
    },
    {
      "epoch": 6.50644302368164e-05,
      "step": 68225,
      "training_step_time": 8.559232950210571
    },
    {
      "epoch": 6.506538391113281e-05,
      "model_forward_time": 1.7391455173492432,
      "step": 68226
    },
    {
      "epoch": 6.506538391113281e-05,
      "step": 68226,
      "training_step_time": 7.674016952514648
    },
    {
      "epoch": 6.506633758544922e-05,
      "model_forward_time": 1.7299463748931885,
      "step": 68227
    },
    {
      "epoch": 6.506633758544922e-05,
      "step": 68227,
      "training_step_time": 7.452680587768555
    },
    {
      "epoch": 6.506729125976562e-05,
      "model_forward_time": 1.7335898876190186,
      "step": 68228
    },
    {
      "epoch": 6.506729125976562e-05,
      "step": 68228,
      "training_step_time": 7.619263410568237
    },
    {
      "epoch": 6.506824493408203e-05,
      "model_forward_time": 1.735501766204834,
      "step": 68229
    },
    {
      "epoch": 6.506824493408203e-05,
      "step": 68229,
      "training_step_time": 7.647933483123779
    },
    {
      "epoch": 6.506919860839844e-05,
      "grad_norm": 0.1373671144247055,
      "learning_rate": 2.5149548295158833e-06,
      "loss": 0.0547,
      "step": 68230
    },
    {
      "epoch": 6.506919860839844e-05,
      "model_forward_time": 2.8463242053985596,
      "step": 68230
    },
    {
      "dynamics_loss_avg": 0.058342592976987365,
      "epoch": 6.506919860839844e-05,
      "step": 68230
    },
    {
      "action_loss_avg": 0.0024201278982218357,
      "epoch": 6.506919860839844e-05,
      "step": 68230
    },
    {
      "epoch": 6.506919860839844e-05,
      "step": 68230,
      "training_step_time": 9.375498533248901
    },
    {
      "epoch": 6.507015228271484e-05,
      "model_forward_time": 1.7290749549865723,
      "step": 68231
    },
    {
      "epoch": 6.507015228271484e-05,
      "step": 68231,
      "training_step_time": 7.92237663269043
    },
    {
      "epoch": 6.507110595703125e-05,
      "model_forward_time": 1.726698637008667,
      "step": 68232
    },
    {
      "epoch": 6.507110595703125e-05,
      "step": 68232,
      "training_step_time": 7.733932971954346
    },
    {
      "epoch": 6.507205963134766e-05,
      "model_forward_time": 1.7297139167785645,
      "step": 68233
    },
    {
      "epoch": 6.507205963134766e-05,
      "step": 68233,
      "training_step_time": 7.619399070739746
    },
    {
      "epoch": 6.507301330566406e-05,
      "model_forward_time": 1.753605604171753,
      "step": 68234
    },
    {
      "epoch": 6.507301330566406e-05,
      "step": 68234,
      "training_step_time": 9.375950813293457
    },
    {
      "epoch": 6.507396697998047e-05,
      "model_forward_time": 1.7285068035125732,
      "step": 68235
    },
    {
      "epoch": 6.507396697998047e-05,
      "step": 68235,
      "training_step_time": 7.59943699836731
    },
    {
      "epoch": 6.507492065429688e-05,
      "model_forward_time": 1.7617056369781494,
      "step": 68236
    },
    {
      "epoch": 6.507492065429688e-05,
      "step": 68236,
      "training_step_time": 7.7062668800354
    },
    {
      "epoch": 6.507587432861328e-05,
      "model_forward_time": 1.7305490970611572,
      "step": 68237
    },
    {
      "epoch": 6.507587432861328e-05,
      "step": 68237,
      "training_step_time": 7.643280982971191
    },
    {
      "epoch": 6.507682800292969e-05,
      "model_forward_time": 1.7292087078094482,
      "step": 68238
    },
    {
      "epoch": 6.507682800292969e-05,
      "step": 68238,
      "training_step_time": 7.590979814529419
    },
    {
      "epoch": 6.50777816772461e-05,
      "model_forward_time": 2.840725898742676,
      "step": 68239
    },
    {
      "epoch": 6.50777816772461e-05,
      "step": 68239,
      "training_step_time": 8.258749961853027
    },
    {
      "epoch": 6.50787353515625e-05,
      "grad_norm": 0.2213962823152542,
      "learning_rate": 2.513520174650505e-06,
      "loss": 0.0545,
      "step": 68240
    },
    {
      "epoch": 6.50787353515625e-05,
      "model_forward_time": 1.7270803451538086,
      "step": 68240
    },
    {
      "dynamics_loss_avg": 0.060282588191330436,
      "epoch": 6.50787353515625e-05,
      "step": 68240
    },
    {
      "action_loss_avg": 0.0030318599863676356,
      "epoch": 6.50787353515625e-05,
      "step": 68240
    },
    {
      "epoch": 6.50787353515625e-05,
      "step": 68240,
      "training_step_time": 8.954120397567749
    },
    {
      "epoch": 6.50796890258789e-05,
      "model_forward_time": 1.7358524799346924,
      "step": 68241
    },
    {
      "epoch": 6.50796890258789e-05,
      "step": 68241,
      "training_step_time": 8.3909170627594
    },
    {
      "epoch": 6.508064270019532e-05,
      "model_forward_time": 1.7216243743896484,
      "step": 68242
    },
    {
      "epoch": 6.508064270019532e-05,
      "step": 68242,
      "training_step_time": 8.264554500579834
    },
    {
      "epoch": 6.508159637451171e-05,
      "model_forward_time": 1.749277114868164,
      "step": 68243
    },
    {
      "epoch": 6.508159637451171e-05,
      "step": 68243,
      "training_step_time": 8.685066223144531
    },
    {
      "epoch": 6.508255004882812e-05,
      "model_forward_time": 1.736177921295166,
      "step": 68244
    },
    {
      "epoch": 6.508255004882812e-05,
      "step": 68244,
      "training_step_time": 7.751933813095093
    },
    {
      "epoch": 6.508350372314454e-05,
      "model_forward_time": 1.7407829761505127,
      "step": 68245
    },
    {
      "epoch": 6.508350372314454e-05,
      "step": 68245,
      "training_step_time": 7.711380958557129
    },
    {
      "epoch": 6.508445739746093e-05,
      "model_forward_time": 1.7277588844299316,
      "step": 68246
    },
    {
      "epoch": 6.508445739746093e-05,
      "step": 68246,
      "training_step_time": 7.554694414138794
    },
    {
      "epoch": 6.508541107177734e-05,
      "model_forward_time": 1.7322916984558105,
      "step": 68247
    },
    {
      "epoch": 6.508541107177734e-05,
      "step": 68247,
      "training_step_time": 7.564147233963013
    },
    {
      "epoch": 6.508636474609375e-05,
      "model_forward_time": 1.7338616847991943,
      "step": 68248
    },
    {
      "epoch": 6.508636474609375e-05,
      "step": 68248,
      "training_step_time": 7.579320907592773
    },
    {
      "epoch": 6.508731842041015e-05,
      "model_forward_time": 2.9299652576446533,
      "step": 68249
    },
    {
      "epoch": 6.508731842041015e-05,
      "step": 68249,
      "training_step_time": 9.181939125061035
    },
    {
      "epoch": 6.508827209472656e-05,
      "grad_norm": 0.09627903252840042,
      "learning_rate": 2.512085791702821e-06,
      "loss": 0.0557,
      "step": 68250
    },
    {
      "epoch": 6.508827209472656e-05,
      "model_forward_time": 1.726189136505127,
      "step": 68250
    },
    {
      "dynamics_loss_avg": 0.0520691042765975,
      "epoch": 6.508827209472656e-05,
      "step": 68250
    },
    {
      "action_loss_avg": 0.001679331116611138,
      "epoch": 6.508827209472656e-05,
      "step": 68250
    },
    {
      "epoch": 6.508827209472656e-05,
      "step": 68250,
      "training_step_time": 7.6864073276519775
    },
    {
      "epoch": 6.508922576904297e-05,
      "model_forward_time": 1.7226691246032715,
      "step": 68251
    },
    {
      "epoch": 6.508922576904297e-05,
      "step": 68251,
      "training_step_time": 8.524718046188354
    },
    {
      "epoch": 6.509017944335937e-05,
      "model_forward_time": 1.7493224143981934,
      "step": 68252
    },
    {
      "epoch": 6.509017944335937e-05,
      "step": 68252,
      "training_step_time": 7.653951406478882
    },
    {
      "epoch": 6.509113311767578e-05,
      "model_forward_time": 1.7276604175567627,
      "step": 68253
    },
    {
      "epoch": 6.509113311767578e-05,
      "step": 68253,
      "training_step_time": 7.557310581207275
    },
    {
      "epoch": 6.509208679199219e-05,
      "model_forward_time": 1.7454323768615723,
      "step": 68254
    },
    {
      "epoch": 6.509208679199219e-05,
      "step": 68254,
      "training_step_time": 7.495792627334595
    },
    {
      "epoch": 6.509304046630859e-05,
      "model_forward_time": 1.7302343845367432,
      "step": 68255
    },
    {
      "epoch": 6.509304046630859e-05,
      "step": 68255,
      "training_step_time": 7.500866651535034
    },
    {
      "epoch": 6.5093994140625e-05,
      "model_forward_time": 1.7382447719573975,
      "step": 68256
    },
    {
      "epoch": 6.5093994140625e-05,
      "step": 68256,
      "training_step_time": 8.40803599357605
    },
    {
      "epoch": 6.509494781494141e-05,
      "model_forward_time": 1.7132411003112793,
      "step": 68257
    },
    {
      "epoch": 6.509494781494141e-05,
      "step": 68257,
      "training_step_time": 8.225260972976685
    },
    {
      "epoch": 6.509590148925781e-05,
      "model_forward_time": 2.920171022415161,
      "step": 68258
    },
    {
      "epoch": 6.509590148925781e-05,
      "step": 68258,
      "training_step_time": 8.713283777236938
    },
    {
      "epoch": 6.509685516357422e-05,
      "model_forward_time": 1.7206923961639404,
      "step": 68259
    },
    {
      "epoch": 6.509685516357422e-05,
      "step": 68259,
      "training_step_time": 8.58667254447937
    },
    {
      "epoch": 6.509780883789063e-05,
      "grad_norm": 0.09817571938037872,
      "learning_rate": 2.5106516808296955e-06,
      "loss": 0.056,
      "step": 68260
    },
    {
      "epoch": 6.509780883789063e-05,
      "model_forward_time": 1.7291147708892822,
      "step": 68260
    },
    {
      "dynamics_loss_avg": 0.06274103950709105,
      "epoch": 6.509780883789063e-05,
      "step": 68260
    },
    {
      "action_loss_avg": 0.001091246458236128,
      "epoch": 6.509780883789063e-05,
      "step": 68260
    },
    {
      "epoch": 6.509780883789063e-05,
      "step": 68260,
      "training_step_time": 7.648092269897461
    },
    {
      "epoch": 6.509876251220703e-05,
      "model_forward_time": 1.7336664199829102,
      "step": 68261
    },
    {
      "epoch": 6.509876251220703e-05,
      "step": 68261,
      "training_step_time": 8.30623745918274
    },
    {
      "epoch": 6.509971618652344e-05,
      "model_forward_time": 1.7258245944976807,
      "step": 68262
    },
    {
      "epoch": 6.509971618652344e-05,
      "step": 68262,
      "training_step_time": 8.610966920852661
    },
    {
      "epoch": 6.510066986083985e-05,
      "model_forward_time": 1.7287545204162598,
      "step": 68263
    },
    {
      "epoch": 6.510066986083985e-05,
      "step": 68263,
      "training_step_time": 7.616403579711914
    },
    {
      "epoch": 6.510162353515625e-05,
      "model_forward_time": 1.7329990863800049,
      "step": 68264
    },
    {
      "epoch": 6.510162353515625e-05,
      "step": 68264,
      "training_step_time": 8.465626001358032
    },
    {
      "epoch": 6.510257720947266e-05,
      "model_forward_time": 1.7340056896209717,
      "step": 68265
    },
    {
      "epoch": 6.510257720947266e-05,
      "step": 68265,
      "training_step_time": 7.526698112487793
    },
    {
      "epoch": 6.510353088378907e-05,
      "model_forward_time": 1.7263622283935547,
      "step": 68266
    },
    {
      "epoch": 6.510353088378907e-05,
      "step": 68266,
      "training_step_time": 7.660503625869751
    },
    {
      "epoch": 6.510448455810546e-05,
      "model_forward_time": 2.9014031887054443,
      "step": 68267
    },
    {
      "epoch": 6.510448455810546e-05,
      "step": 68267,
      "training_step_time": 7.678112268447876
    },
    {
      "epoch": 6.510543823242188e-05,
      "model_forward_time": 1.7213153839111328,
      "step": 68268
    },
    {
      "epoch": 6.510543823242188e-05,
      "step": 68268,
      "training_step_time": 7.884785413742065
    },
    {
      "epoch": 6.510639190673829e-05,
      "model_forward_time": 1.7284648418426514,
      "step": 68269
    },
    {
      "epoch": 6.510639190673829e-05,
      "step": 68269,
      "training_step_time": 7.652282476425171
    },
    {
      "epoch": 6.510734558105468e-05,
      "grad_norm": 0.0854099839925766,
      "learning_rate": 2.5092178421879555e-06,
      "loss": 0.0559,
      "step": 68270
    },
    {
      "epoch": 6.510734558105468e-05,
      "model_forward_time": 1.7200403213500977,
      "step": 68270
    },
    {
      "dynamics_loss_avg": 0.057384158857166766,
      "epoch": 6.510734558105468e-05,
      "step": 68270
    },
    {
      "action_loss_avg": 0.0019022102700546385,
      "epoch": 6.510734558105468e-05,
      "step": 68270
    },
    {
      "epoch": 6.510734558105468e-05,
      "step": 68270,
      "training_step_time": 7.504289627075195
    },
    {
      "epoch": 6.51082992553711e-05,
      "model_forward_time": 1.7283504009246826,
      "step": 68271
    },
    {
      "epoch": 6.51082992553711e-05,
      "step": 68271,
      "training_step_time": 8.19866418838501
    },
    {
      "epoch": 6.51092529296875e-05,
      "model_forward_time": 1.7264442443847656,
      "step": 68272
    },
    {
      "epoch": 6.51092529296875e-05,
      "step": 68272,
      "training_step_time": 7.605258226394653
    },
    {
      "epoch": 6.51102066040039e-05,
      "model_forward_time": 1.7168490886688232,
      "step": 68273
    },
    {
      "epoch": 6.51102066040039e-05,
      "step": 68273,
      "training_step_time": 7.400845289230347
    },
    {
      "epoch": 6.511116027832031e-05,
      "model_forward_time": 1.7252271175384521,
      "step": 68274
    },
    {
      "epoch": 6.511116027832031e-05,
      "step": 68274,
      "training_step_time": 7.521247386932373
    },
    {
      "epoch": 6.511211395263672e-05,
      "model_forward_time": 1.7174057960510254,
      "step": 68275
    },
    {
      "epoch": 6.511211395263672e-05,
      "step": 68275,
      "training_step_time": 7.600855827331543
    },
    {
      "epoch": 6.511306762695312e-05,
      "model_forward_time": 2.8533616065979004,
      "step": 68276
    },
    {
      "epoch": 6.511306762695312e-05,
      "step": 68276,
      "training_step_time": 8.675923824310303
    },
    {
      "epoch": 6.511402130126953e-05,
      "model_forward_time": 1.7166850566864014,
      "step": 68277
    },
    {
      "epoch": 6.511402130126953e-05,
      "step": 68277,
      "training_step_time": 8.957395553588867
    },
    {
      "epoch": 6.511497497558594e-05,
      "model_forward_time": 1.7199962139129639,
      "step": 68278
    },
    {
      "epoch": 6.511497497558594e-05,
      "step": 68278,
      "training_step_time": 8.649619102478027
    },
    {
      "epoch": 6.511592864990234e-05,
      "model_forward_time": 1.7074415683746338,
      "step": 68279
    },
    {
      "epoch": 6.511592864990234e-05,
      "step": 68279,
      "training_step_time": 8.088146209716797
    },
    {
      "epoch": 6.511688232421875e-05,
      "grad_norm": 0.1039806455373764,
      "learning_rate": 2.5077842759344118e-06,
      "loss": 0.0567,
      "step": 68280
    },
    {
      "epoch": 6.511688232421875e-05,
      "model_forward_time": 1.7448227405548096,
      "step": 68280
    },
    {
      "dynamics_loss_avg": 0.0642674658447504,
      "epoch": 6.511688232421875e-05,
      "step": 68280
    },
    {
      "action_loss_avg": 0.0020130132674239577,
      "epoch": 6.511688232421875e-05,
      "step": 68280
    },
    {
      "epoch": 6.511688232421875e-05,
      "step": 68280,
      "training_step_time": 7.668623208999634
    },
    {
      "epoch": 6.511783599853516e-05,
      "model_forward_time": 1.7168216705322266,
      "step": 68281
    },
    {
      "epoch": 6.511783599853516e-05,
      "step": 68281,
      "training_step_time": 7.623057842254639
    },
    {
      "epoch": 6.511878967285156e-05,
      "model_forward_time": 1.756673812866211,
      "step": 68282
    },
    {
      "epoch": 6.511878967285156e-05,
      "step": 68282,
      "training_step_time": 7.544045448303223
    },
    {
      "epoch": 6.511974334716797e-05,
      "model_forward_time": 1.725304126739502,
      "step": 68283
    },
    {
      "epoch": 6.511974334716797e-05,
      "step": 68283,
      "training_step_time": 7.482374668121338
    },
    {
      "epoch": 6.512069702148438e-05,
      "model_forward_time": 1.7225275039672852,
      "step": 68284
    },
    {
      "epoch": 6.512069702148438e-05,
      "step": 68284,
      "training_step_time": 7.592284917831421
    },
    {
      "epoch": 6.512165069580078e-05,
      "model_forward_time": 2.870033025741577,
      "step": 68285
    },
    {
      "epoch": 6.512165069580078e-05,
      "step": 68285,
      "training_step_time": 8.717490911483765
    },
    {
      "epoch": 6.512260437011719e-05,
      "model_forward_time": 1.368239164352417,
      "step": 68286
    },
    {
      "epoch": 6.512260437011719e-05,
      "step": 68286,
      "training_step_time": 7.737590789794922
    },
    {
      "epoch": 6.51235580444336e-05,
      "model_forward_time": 1.722745656967163,
      "step": 68287
    },
    {
      "epoch": 6.51235580444336e-05,
      "step": 68287,
      "training_step_time": 8.639177083969116
    },
    {
      "epoch": 6.512451171875e-05,
      "model_forward_time": 1.7241628170013428,
      "step": 68288
    },
    {
      "epoch": 6.512451171875e-05,
      "step": 68288,
      "training_step_time": 8.29486346244812
    },
    {
      "epoch": 6.512546539306641e-05,
      "model_forward_time": 1.7502398490905762,
      "step": 68289
    },
    {
      "epoch": 6.512546539306641e-05,
      "step": 68289,
      "training_step_time": 7.584716796875
    },
    {
      "epoch": 6.512641906738282e-05,
      "grad_norm": 0.10978974401950836,
      "learning_rate": 2.50635098222583e-06,
      "loss": 0.0567,
      "step": 68290
    },
    {
      "epoch": 6.512641906738282e-05,
      "model_forward_time": 1.7318644523620605,
      "step": 68290
    },
    {
      "dynamics_loss_avg": 0.06702671572566032,
      "epoch": 6.512641906738282e-05,
      "step": 68290
    },
    {
      "action_loss_avg": 0.002145185446715914,
      "epoch": 6.512641906738282e-05,
      "step": 68290
    },
    {
      "epoch": 6.512641906738282e-05,
      "step": 68290,
      "training_step_time": 7.652643442153931
    },
    {
      "epoch": 6.512737274169922e-05,
      "model_forward_time": 1.7476823329925537,
      "step": 68291
    },
    {
      "epoch": 6.512737274169922e-05,
      "step": 68291,
      "training_step_time": 8.477818965911865
    },
    {
      "epoch": 6.512832641601563e-05,
      "model_forward_time": 1.731339931488037,
      "step": 68292
    },
    {
      "epoch": 6.512832641601563e-05,
      "step": 68292,
      "training_step_time": 7.3883373737335205
    },
    {
      "epoch": 6.512928009033204e-05,
      "model_forward_time": 1.7263140678405762,
      "step": 68293
    },
    {
      "epoch": 6.512928009033204e-05,
      "step": 68293,
      "training_step_time": 7.564188480377197
    },
    {
      "epoch": 6.513023376464843e-05,
      "model_forward_time": 1.7221667766571045,
      "step": 68294
    },
    {
      "epoch": 6.513023376464843e-05,
      "step": 68294,
      "training_step_time": 8.168958187103271
    },
    {
      "epoch": 6.513118743896485e-05,
      "model_forward_time": 2.9324941635131836,
      "step": 68295
    },
    {
      "epoch": 6.513118743896485e-05,
      "step": 68295,
      "training_step_time": 8.488710880279541
    },
    {
      "epoch": 6.513214111328126e-05,
      "model_forward_time": 1.7227935791015625,
      "step": 68296
    },
    {
      "epoch": 6.513214111328126e-05,
      "step": 68296,
      "training_step_time": 8.873699426651001
    },
    {
      "epoch": 6.513309478759765e-05,
      "model_forward_time": 1.7321007251739502,
      "step": 68297
    },
    {
      "epoch": 6.513309478759765e-05,
      "step": 68297,
      "training_step_time": 7.81582236289978
    },
    {
      "epoch": 6.513404846191406e-05,
      "model_forward_time": 1.7473218441009521,
      "step": 68298
    },
    {
      "epoch": 6.513404846191406e-05,
      "step": 68298,
      "training_step_time": 9.771047592163086
    },
    {
      "epoch": 6.513500213623048e-05,
      "model_forward_time": 1.7218053340911865,
      "step": 68299
    },
    {
      "epoch": 6.513500213623048e-05,
      "step": 68299,
      "training_step_time": 7.761661529541016
    },
    {
      "epoch": 6.513595581054687e-05,
      "grad_norm": 0.0912138819694519,
      "learning_rate": 2.5049179612189578e-06,
      "loss": 0.0538,
      "step": 68300
    },
    {
      "epoch": 6.513595581054687e-05,
      "model_forward_time": 1.7407381534576416,
      "step": 68300
    },
    {
      "dynamics_loss_avg": 0.04502191320061684,
      "epoch": 6.513595581054687e-05,
      "step": 68300
    },
    {
      "action_loss_avg": 0.0012004956923192367,
      "epoch": 6.513595581054687e-05,
      "step": 68300
    },
    {
      "epoch": 6.513595581054687e-05,
      "step": 68300,
      "training_step_time": 7.675760269165039
    },
    {
      "epoch": 6.513690948486328e-05,
      "model_forward_time": 1.359243392944336,
      "step": 68301
    },
    {
      "epoch": 6.513690948486328e-05,
      "step": 68301,
      "training_step_time": 7.521082401275635
    },
    {
      "epoch": 6.51378631591797e-05,
      "model_forward_time": 1.7344648838043213,
      "step": 68302
    },
    {
      "epoch": 6.51378631591797e-05,
      "step": 68302,
      "training_step_time": 7.6975321769714355
    },
    {
      "epoch": 6.513881683349609e-05,
      "model_forward_time": 1.7337884902954102,
      "step": 68303
    },
    {
      "epoch": 6.513881683349609e-05,
      "step": 68303,
      "training_step_time": 7.657052040100098
    },
    {
      "epoch": 6.51397705078125e-05,
      "model_forward_time": 2.888638973236084,
      "step": 68304
    },
    {
      "epoch": 6.51397705078125e-05,
      "step": 68304,
      "training_step_time": 7.750239610671997
    },
    {
      "epoch": 6.514072418212891e-05,
      "model_forward_time": 1.7276928424835205,
      "step": 68305
    },
    {
      "epoch": 6.514072418212891e-05,
      "step": 68305,
      "training_step_time": 7.770277500152588
    },
    {
      "epoch": 6.514167785644531e-05,
      "model_forward_time": 1.7264864444732666,
      "step": 68306
    },
    {
      "epoch": 6.514167785644531e-05,
      "step": 68306,
      "training_step_time": 8.792822122573853
    },
    {
      "epoch": 6.514263153076172e-05,
      "model_forward_time": 1.7293727397918701,
      "step": 68307
    },
    {
      "epoch": 6.514263153076172e-05,
      "step": 68307,
      "training_step_time": 7.7415454387664795
    },
    {
      "epoch": 6.514358520507813e-05,
      "model_forward_time": 1.7359778881072998,
      "step": 68308
    },
    {
      "epoch": 6.514358520507813e-05,
      "step": 68308,
      "training_step_time": 7.569627523422241
    },
    {
      "epoch": 6.514453887939453e-05,
      "model_forward_time": 1.7321021556854248,
      "step": 68309
    },
    {
      "epoch": 6.514453887939453e-05,
      "step": 68309,
      "training_step_time": 7.510529279708862
    },
    {
      "epoch": 6.514549255371094e-05,
      "grad_norm": 0.09960133582353592,
      "learning_rate": 2.5034852130705044e-06,
      "loss": 0.0514,
      "step": 68310
    },
    {
      "epoch": 6.514549255371094e-05,
      "model_forward_time": 1.7280020713806152,
      "step": 68310
    },
    {
      "dynamics_loss_avg": 0.05084518883377313,
      "epoch": 6.514549255371094e-05,
      "step": 68310
    },
    {
      "action_loss_avg": 0.0013416500703897327,
      "epoch": 6.514549255371094e-05,
      "step": 68310
    },
    {
      "epoch": 6.514549255371094e-05,
      "step": 68310,
      "training_step_time": 7.50400972366333
    },
    {
      "epoch": 6.514644622802735e-05,
      "model_forward_time": 1.7320466041564941,
      "step": 68311
    },
    {
      "epoch": 6.514644622802735e-05,
      "step": 68311,
      "training_step_time": 7.558566093444824
    },
    {
      "epoch": 6.514739990234375e-05,
      "model_forward_time": 1.7155823707580566,
      "step": 68312
    },
    {
      "epoch": 6.514739990234375e-05,
      "step": 68312,
      "training_step_time": 9.46160340309143
    },
    {
      "epoch": 6.514835357666016e-05,
      "model_forward_time": 2.893191337585449,
      "step": 68313
    },
    {
      "epoch": 6.514835357666016e-05,
      "step": 68313,
      "training_step_time": 7.661750316619873
    },
    {
      "epoch": 6.514930725097656e-05,
      "model_forward_time": 1.7355408668518066,
      "step": 68314
    },
    {
      "epoch": 6.514930725097656e-05,
      "step": 68314,
      "training_step_time": 9.265633821487427
    },
    {
      "epoch": 6.515026092529297e-05,
      "model_forward_time": 1.7333042621612549,
      "step": 68315
    },
    {
      "epoch": 6.515026092529297e-05,
      "step": 68315,
      "training_step_time": 8.703189611434937
    },
    {
      "epoch": 6.515121459960938e-05,
      "model_forward_time": 1.7295820713043213,
      "step": 68316
    },
    {
      "epoch": 6.515121459960938e-05,
      "step": 68316,
      "training_step_time": 9.329366683959961
    },
    {
      "epoch": 6.515216827392577e-05,
      "model_forward_time": 1.7516906261444092,
      "step": 68317
    },
    {
      "epoch": 6.515216827392577e-05,
      "step": 68317,
      "training_step_time": 7.605020999908447
    },
    {
      "epoch": 6.515312194824219e-05,
      "model_forward_time": 1.73246169090271,
      "step": 68318
    },
    {
      "epoch": 6.515312194824219e-05,
      "step": 68318,
      "training_step_time": 7.713611602783203
    },
    {
      "epoch": 6.51540756225586e-05,
      "model_forward_time": 1.7348213195800781,
      "step": 68319
    },
    {
      "epoch": 6.51540756225586e-05,
      "step": 68319,
      "training_step_time": 7.564848899841309
    },
    {
      "epoch": 6.5155029296875e-05,
      "grad_norm": 0.07153748720884323,
      "learning_rate": 2.5020527379371574e-06,
      "loss": 0.0536,
      "step": 68320
    },
    {
      "epoch": 6.5155029296875e-05,
      "model_forward_time": 1.7261991500854492,
      "step": 68320
    },
    {
      "dynamics_loss_avg": 0.06163378581404686,
      "epoch": 6.5155029296875e-05,
      "step": 68320
    },
    {
      "action_loss_avg": 0.0023928132723085582,
      "epoch": 6.5155029296875e-05,
      "step": 68320
    },
    {
      "epoch": 6.5155029296875e-05,
      "step": 68320,
      "training_step_time": 7.529277324676514
    },
    {
      "epoch": 6.51559829711914e-05,
      "model_forward_time": 1.7298698425292969,
      "step": 68321
    },
    {
      "epoch": 6.51559829711914e-05,
      "step": 68321,
      "training_step_time": 8.21457052230835
    },
    {
      "epoch": 6.515693664550782e-05,
      "model_forward_time": 2.8415415287017822,
      "step": 68322
    },
    {
      "epoch": 6.515693664550782e-05,
      "step": 68322,
      "training_step_time": 9.287502765655518
    },
    {
      "epoch": 6.515789031982421e-05,
      "model_forward_time": 1.7295196056365967,
      "step": 68323
    },
    {
      "epoch": 6.515789031982421e-05,
      "step": 68323,
      "training_step_time": 7.796367883682251
    },
    {
      "epoch": 6.515884399414062e-05,
      "model_forward_time": 1.72523832321167,
      "step": 68324
    },
    {
      "epoch": 6.515884399414062e-05,
      "step": 68324,
      "training_step_time": 8.595624923706055
    },
    {
      "epoch": 6.515979766845703e-05,
      "model_forward_time": 1.706620216369629,
      "step": 68325
    },
    {
      "epoch": 6.515979766845703e-05,
      "step": 68325,
      "training_step_time": 7.703569650650024
    },
    {
      "epoch": 6.516075134277343e-05,
      "model_forward_time": 1.745335340499878,
      "step": 68326
    },
    {
      "epoch": 6.516075134277343e-05,
      "step": 68326,
      "training_step_time": 7.667366981506348
    },
    {
      "epoch": 6.516170501708984e-05,
      "model_forward_time": 1.7190914154052734,
      "step": 68327
    },
    {
      "epoch": 6.516170501708984e-05,
      "step": 68327,
      "training_step_time": 7.795843839645386
    },
    {
      "epoch": 6.516265869140625e-05,
      "model_forward_time": 1.7532885074615479,
      "step": 68328
    },
    {
      "epoch": 6.516265869140625e-05,
      "step": 68328,
      "training_step_time": 7.773977041244507
    },
    {
      "epoch": 6.516361236572265e-05,
      "model_forward_time": 1.7324414253234863,
      "step": 68329
    },
    {
      "epoch": 6.516361236572265e-05,
      "step": 68329,
      "training_step_time": 8.217114210128784
    },
    {
      "epoch": 6.516456604003906e-05,
      "grad_norm": 0.06959950178861618,
      "learning_rate": 2.500620535975564e-06,
      "loss": 0.0546,
      "step": 68330
    },
    {
      "epoch": 6.516456604003906e-05,
      "model_forward_time": 1.727102518081665,
      "step": 68330
    },
    {
      "dynamics_loss_avg": 0.07116932943463325,
      "epoch": 6.516456604003906e-05,
      "step": 68330
    },
    {
      "action_loss_avg": 0.0027364222856704145,
      "epoch": 6.516456604003906e-05,
      "step": 68330
    },
    {
      "epoch": 6.516456604003906e-05,
      "step": 68330,
      "training_step_time": 7.664607524871826
    },
    {
      "epoch": 6.516551971435547e-05,
      "model_forward_time": 1.7190146446228027,
      "step": 68331
    },
    {
      "epoch": 6.516551971435547e-05,
      "step": 68331,
      "training_step_time": 8.17325758934021
    },
    {
      "epoch": 6.516647338867187e-05,
      "model_forward_time": 1.720139503479004,
      "step": 68332
    },
    {
      "epoch": 6.516647338867187e-05,
      "step": 68332,
      "training_step_time": 7.669345378875732
    },
    {
      "epoch": 6.516742706298828e-05,
      "model_forward_time": 1.7248530387878418,
      "step": 68333
    },
    {
      "epoch": 6.516742706298828e-05,
      "step": 68333,
      "training_step_time": 7.890238523483276
    },
    {
      "epoch": 6.516838073730469e-05,
      "model_forward_time": 1.7225236892700195,
      "step": 68334
    },
    {
      "epoch": 6.516838073730469e-05,
      "step": 68334,
      "training_step_time": 8.46316385269165
    },
    {
      "epoch": 6.516933441162109e-05,
      "model_forward_time": 1.7363545894622803,
      "step": 68335
    },
    {
      "epoch": 6.516933441162109e-05,
      "step": 68335,
      "training_step_time": 7.842027425765991
    },
    {
      "epoch": 6.51702880859375e-05,
      "model_forward_time": 1.7226808071136475,
      "step": 68336
    },
    {
      "epoch": 6.51702880859375e-05,
      "step": 68336,
      "training_step_time": 7.822699069976807
    },
    {
      "epoch": 6.517124176025391e-05,
      "model_forward_time": 1.7923595905303955,
      "step": 68337
    },
    {
      "epoch": 6.517124176025391e-05,
      "step": 68337,
      "training_step_time": 7.846256732940674
    },
    {
      "epoch": 6.51721954345703e-05,
      "model_forward_time": 1.7251651287078857,
      "step": 68338
    },
    {
      "epoch": 6.51721954345703e-05,
      "step": 68338,
      "training_step_time": 6.898664474487305
    },
    {
      "epoch": 6.517314910888672e-05,
      "model_forward_time": 1.7325587272644043,
      "step": 68339
    },
    {
      "epoch": 6.517314910888672e-05,
      "step": 68339,
      "training_step_time": 7.782313346862793
    },
    {
      "epoch": 6.517410278320313e-05,
      "grad_norm": 0.09201744943857193,
      "learning_rate": 2.4991886073423517e-06,
      "loss": 0.0549,
      "step": 68340
    },
    {
      "epoch": 6.517410278320313e-05,
      "model_forward_time": 1.7224841117858887,
      "step": 68340
    },
    {
      "dynamics_loss_avg": 0.0729591578245163,
      "epoch": 6.517410278320313e-05,
      "step": 68340
    },
    {
      "action_loss_avg": 0.0029734513896983116,
      "epoch": 6.517410278320313e-05,
      "step": 68340
    },
    {
      "epoch": 6.517410278320313e-05,
      "step": 68340,
      "training_step_time": 7.793713331222534
    },
    {
      "epoch": 6.517505645751953e-05,
      "model_forward_time": 2.9131038188934326,
      "step": 68341
    },
    {
      "epoch": 6.517505645751953e-05,
      "step": 68341,
      "training_step_time": 8.836532831192017
    },
    {
      "epoch": 6.517601013183594e-05,
      "model_forward_time": 1.7220463752746582,
      "step": 68342
    },
    {
      "epoch": 6.517601013183594e-05,
      "step": 68342,
      "training_step_time": 7.90936541557312
    },
    {
      "epoch": 6.517696380615235e-05,
      "model_forward_time": 1.720001220703125,
      "step": 68343
    },
    {
      "epoch": 6.517696380615235e-05,
      "step": 68343,
      "training_step_time": 9.384034395217896
    },
    {
      "epoch": 6.517791748046874e-05,
      "model_forward_time": 1.7057740688323975,
      "step": 68344
    },
    {
      "epoch": 6.517791748046874e-05,
      "step": 68344,
      "training_step_time": 7.731663942337036
    },
    {
      "epoch": 6.517887115478516e-05,
      "model_forward_time": 1.7330043315887451,
      "step": 68345
    },
    {
      "epoch": 6.517887115478516e-05,
      "step": 68345,
      "training_step_time": 7.765708923339844
    },
    {
      "epoch": 6.517982482910157e-05,
      "model_forward_time": 1.7243492603302002,
      "step": 68346
    },
    {
      "epoch": 6.517982482910157e-05,
      "step": 68346,
      "training_step_time": 7.786897659301758
    },
    {
      "epoch": 6.518077850341796e-05,
      "model_forward_time": 1.7118418216705322,
      "step": 68347
    },
    {
      "epoch": 6.518077850341796e-05,
      "step": 68347,
      "training_step_time": 7.645616054534912
    },
    {
      "epoch": 6.518173217773437e-05,
      "model_forward_time": 1.7213773727416992,
      "step": 68348
    },
    {
      "epoch": 6.518173217773437e-05,
      "step": 68348,
      "training_step_time": 8.628745794296265
    },
    {
      "epoch": 6.518268585205078e-05,
      "model_forward_time": 1.3515913486480713,
      "step": 68349
    },
    {
      "epoch": 6.518268585205078e-05,
      "step": 68349,
      "training_step_time": 8.464721918106079
    },
    {
      "epoch": 6.518363952636718e-05,
      "grad_norm": 0.11212238669395447,
      "learning_rate": 2.4977569521941134e-06,
      "loss": 0.0516,
      "step": 68350
    },
    {
      "epoch": 6.518363952636718e-05,
      "model_forward_time": 2.8699584007263184,
      "step": 68350
    },
    {
      "dynamics_loss_avg": 0.05445360094308853,
      "epoch": 6.518363952636718e-05,
      "step": 68350
    },
    {
      "action_loss_avg": 0.0019745150639209896,
      "epoch": 6.518363952636718e-05,
      "step": 68350
    },
    {
      "epoch": 6.518363952636718e-05,
      "step": 68350,
      "training_step_time": 7.796400547027588
    },
    {
      "epoch": 6.518459320068359e-05,
      "model_forward_time": 1.354114294052124,
      "step": 68351
    },
    {
      "epoch": 6.518459320068359e-05,
      "step": 68351,
      "training_step_time": 9.387398481369019
    },
    {
      "epoch": 6.5185546875e-05,
      "model_forward_time": 1.6994507312774658,
      "step": 68352
    },
    {
      "epoch": 6.5185546875e-05,
      "step": 68352,
      "training_step_time": 8.03731393814087
    },
    {
      "epoch": 6.51865005493164e-05,
      "model_forward_time": 1.7132465839385986,
      "step": 68353
    },
    {
      "epoch": 6.51865005493164e-05,
      "step": 68353,
      "training_step_time": 8.490320920944214
    },
    {
      "epoch": 6.518745422363281e-05,
      "model_forward_time": 1.7360260486602783,
      "step": 68354
    },
    {
      "epoch": 6.518745422363281e-05,
      "step": 68354,
      "training_step_time": 8.825324296951294
    },
    {
      "epoch": 6.518840789794922e-05,
      "model_forward_time": 1.727034091949463,
      "step": 68355
    },
    {
      "epoch": 6.518840789794922e-05,
      "step": 68355,
      "training_step_time": 7.737873554229736
    },
    {
      "epoch": 6.518936157226562e-05,
      "model_forward_time": 1.715303659439087,
      "step": 68356
    },
    {
      "epoch": 6.518936157226562e-05,
      "step": 68356,
      "training_step_time": 7.5676586627960205
    },
    {
      "epoch": 6.519031524658203e-05,
      "model_forward_time": 1.7216198444366455,
      "step": 68357
    },
    {
      "epoch": 6.519031524658203e-05,
      "step": 68357,
      "training_step_time": 7.669524908065796
    },
    {
      "epoch": 6.519126892089844e-05,
      "model_forward_time": 1.7209079265594482,
      "step": 68358
    },
    {
      "epoch": 6.519126892089844e-05,
      "step": 68358,
      "training_step_time": 9.607348442077637
    },
    {
      "epoch": 6.519222259521484e-05,
      "model_forward_time": 2.81484055519104,
      "step": 68359
    },
    {
      "epoch": 6.519222259521484e-05,
      "step": 68359,
      "training_step_time": 8.748015403747559
    },
    {
      "epoch": 6.519317626953125e-05,
      "grad_norm": 0.08859940618276596,
      "learning_rate": 2.4963255706874146e-06,
      "loss": 0.0536,
      "step": 68360
    },
    {
      "epoch": 6.519317626953125e-05,
      "model_forward_time": 1.7138187885284424,
      "step": 68360
    },
    {
      "dynamics_loss_avg": 0.05513972733169794,
      "epoch": 6.519317626953125e-05,
      "step": 68360
    },
    {
      "action_loss_avg": 0.0015909637295408174,
      "epoch": 6.519317626953125e-05,
      "step": 68360
    },
    {
      "epoch": 6.519317626953125e-05,
      "step": 68360,
      "training_step_time": 7.791834831237793
    },
    {
      "epoch": 6.519412994384766e-05,
      "model_forward_time": 1.7194647789001465,
      "step": 68361
    },
    {
      "epoch": 6.519412994384766e-05,
      "step": 68361,
      "training_step_time": 8.963316202163696
    },
    {
      "epoch": 6.519508361816406e-05,
      "model_forward_time": 1.7150330543518066,
      "step": 68362
    },
    {
      "epoch": 6.519508361816406e-05,
      "step": 68362,
      "training_step_time": 8.31920599937439
    },
    {
      "epoch": 6.519603729248047e-05,
      "model_forward_time": 1.7277917861938477,
      "step": 68363
    },
    {
      "epoch": 6.519603729248047e-05,
      "step": 68363,
      "training_step_time": 8.601945638656616
    },
    {
      "epoch": 6.519699096679688e-05,
      "model_forward_time": 1.719080924987793,
      "step": 68364
    },
    {
      "epoch": 6.519699096679688e-05,
      "step": 68364,
      "training_step_time": 7.716135740280151
    },
    {
      "epoch": 6.519794464111328e-05,
      "model_forward_time": 1.7443115711212158,
      "step": 68365
    },
    {
      "epoch": 6.519794464111328e-05,
      "step": 68365,
      "training_step_time": 7.7437450885772705
    },
    {
      "epoch": 6.519889831542969e-05,
      "model_forward_time": 1.7166962623596191,
      "step": 68366
    },
    {
      "epoch": 6.519889831542969e-05,
      "step": 68366,
      "training_step_time": 7.65606951713562
    },
    {
      "epoch": 6.51998519897461e-05,
      "model_forward_time": 1.7244884967803955,
      "step": 68367
    },
    {
      "epoch": 6.51998519897461e-05,
      "step": 68367,
      "training_step_time": 7.760189771652222
    },
    {
      "epoch": 6.52008056640625e-05,
      "model_forward_time": 2.8114068508148193,
      "step": 68368
    },
    {
      "epoch": 6.52008056640625e-05,
      "step": 68368,
      "training_step_time": 8.554732084274292
    },
    {
      "epoch": 6.52017593383789e-05,
      "model_forward_time": 1.717878818511963,
      "step": 68369
    },
    {
      "epoch": 6.52017593383789e-05,
      "step": 68369,
      "training_step_time": 8.826695442199707
    },
    {
      "epoch": 6.520271301269532e-05,
      "grad_norm": 0.09093654900789261,
      "learning_rate": 2.494894462978785e-06,
      "loss": 0.0533,
      "step": 68370
    },
    {
      "epoch": 6.520271301269532e-05,
      "model_forward_time": 1.7182092666625977,
      "step": 68370
    },
    {
      "dynamics_loss_avg": 0.05894365347921848,
      "epoch": 6.520271301269532e-05,
      "step": 68370
    },
    {
      "action_loss_avg": 0.0014496173913357779,
      "epoch": 6.520271301269532e-05,
      "step": 68370
    },
    {
      "epoch": 6.520271301269532e-05,
      "step": 68370,
      "training_step_time": 7.8620805740356445
    },
    {
      "epoch": 6.520366668701171e-05,
      "model_forward_time": 1.7203962802886963,
      "step": 68371
    },
    {
      "epoch": 6.520366668701171e-05,
      "step": 68371,
      "training_step_time": 8.677165985107422
    },
    {
      "epoch": 6.520462036132812e-05,
      "model_forward_time": 1.720595359802246,
      "step": 68372
    },
    {
      "epoch": 6.520462036132812e-05,
      "step": 68372,
      "training_step_time": 7.67669939994812
    },
    {
      "epoch": 6.520557403564454e-05,
      "model_forward_time": 1.7324655055999756,
      "step": 68373
    },
    {
      "epoch": 6.520557403564454e-05,
      "step": 68373,
      "training_step_time": 8.646529197692871
    },
    {
      "epoch": 6.520652770996093e-05,
      "model_forward_time": 1.7389116287231445,
      "step": 68374
    },
    {
      "epoch": 6.520652770996093e-05,
      "step": 68374,
      "training_step_time": 7.686500549316406
    },
    {
      "epoch": 6.520748138427734e-05,
      "model_forward_time": 1.7147610187530518,
      "step": 68375
    },
    {
      "epoch": 6.520748138427734e-05,
      "step": 68375,
      "training_step_time": 7.918524980545044
    },
    {
      "epoch": 6.520843505859375e-05,
      "model_forward_time": 1.7140755653381348,
      "step": 68376
    },
    {
      "epoch": 6.520843505859375e-05,
      "step": 68376,
      "training_step_time": 7.75834846496582
    },
    {
      "epoch": 6.520938873291015e-05,
      "model_forward_time": 1.7242515087127686,
      "step": 68377
    },
    {
      "epoch": 6.520938873291015e-05,
      "step": 68377,
      "training_step_time": 8.621807336807251
    },
    {
      "epoch": 6.521034240722656e-05,
      "model_forward_time": 2.8600237369537354,
      "step": 68378
    },
    {
      "epoch": 6.521034240722656e-05,
      "step": 68378,
      "training_step_time": 7.6786582469940186
    },
    {
      "epoch": 6.521129608154297e-05,
      "model_forward_time": 1.721975326538086,
      "step": 68379
    },
    {
      "epoch": 6.521129608154297e-05,
      "step": 68379,
      "training_step_time": 10.148067951202393
    },
    {
      "epoch": 6.521224975585937e-05,
      "grad_norm": 0.08303002268075943,
      "learning_rate": 2.493463629224731e-06,
      "loss": 0.0546,
      "step": 68380
    },
    {
      "epoch": 6.521224975585937e-05,
      "model_forward_time": 1.7162737846374512,
      "step": 68380
    },
    {
      "dynamics_loss_avg": 0.05957706328481436,
      "epoch": 6.521224975585937e-05,
      "step": 68380
    },
    {
      "action_loss_avg": 0.002057094141491689,
      "epoch": 6.521224975585937e-05,
      "step": 68380
    },
    {
      "epoch": 6.521224975585937e-05,
      "step": 68380,
      "training_step_time": 10.902922868728638
    },
    {
      "epoch": 6.521320343017578e-05,
      "model_forward_time": 1.7185673713684082,
      "step": 68381
    },
    {
      "epoch": 6.521320343017578e-05,
      "step": 68381,
      "training_step_time": 7.779556035995483
    },
    {
      "epoch": 6.521415710449219e-05,
      "model_forward_time": 1.7568950653076172,
      "step": 68382
    },
    {
      "epoch": 6.521415710449219e-05,
      "step": 68382,
      "training_step_time": 8.581427812576294
    },
    {
      "epoch": 6.521511077880859e-05,
      "model_forward_time": 1.7477262020111084,
      "step": 68383
    },
    {
      "epoch": 6.521511077880859e-05,
      "step": 68383,
      "training_step_time": 7.695963621139526
    },
    {
      "epoch": 6.5216064453125e-05,
      "model_forward_time": 1.7103919982910156,
      "step": 68384
    },
    {
      "epoch": 6.5216064453125e-05,
      "step": 68384,
      "training_step_time": 8.126926898956299
    },
    {
      "epoch": 6.521701812744141e-05,
      "model_forward_time": 1.7279150485992432,
      "step": 68385
    },
    {
      "epoch": 6.521701812744141e-05,
      "step": 68385,
      "training_step_time": 7.7649853229522705
    },
    {
      "epoch": 6.521797180175781e-05,
      "model_forward_time": 1.0230019092559814,
      "step": 68386
    },
    {
      "epoch": 6.521797180175781e-05,
      "step": 68386,
      "training_step_time": 7.8085618019104
    },
    {
      "epoch": 6.521892547607422e-05,
      "model_forward_time": 2.872378349304199,
      "step": 68387
    },
    {
      "epoch": 6.521892547607422e-05,
      "step": 68387,
      "training_step_time": 7.704858303070068
    },
    {
      "epoch": 6.521987915039063e-05,
      "model_forward_time": 1.7190327644348145,
      "step": 68388
    },
    {
      "epoch": 6.521987915039063e-05,
      "step": 68388,
      "training_step_time": 8.80626916885376
    },
    {
      "epoch": 6.522083282470703e-05,
      "model_forward_time": 1.7127909660339355,
      "step": 68389
    },
    {
      "epoch": 6.522083282470703e-05,
      "step": 68389,
      "training_step_time": 8.821687459945679
    },
    {
      "epoch": 6.522178649902344e-05,
      "grad_norm": 0.08576541393995285,
      "learning_rate": 2.492033069581728e-06,
      "loss": 0.056,
      "step": 68390
    },
    {
      "epoch": 6.522178649902344e-05,
      "model_forward_time": 1.7141048908233643,
      "step": 68390
    },
    {
      "dynamics_loss_avg": 0.04998696818947792,
      "epoch": 6.522178649902344e-05,
      "step": 68390
    },
    {
      "action_loss_avg": 0.0020008766936371104,
      "epoch": 6.522178649902344e-05,
      "step": 68390
    },
    {
      "epoch": 6.522178649902344e-05,
      "step": 68390,
      "training_step_time": 8.198741674423218
    },
    {
      "epoch": 6.522274017333985e-05,
      "model_forward_time": 1.7395756244659424,
      "step": 68391
    },
    {
      "epoch": 6.522274017333985e-05,
      "step": 68391,
      "training_step_time": 7.659639596939087
    },
    {
      "epoch": 6.522369384765625e-05,
      "model_forward_time": 1.7305991649627686,
      "step": 68392
    },
    {
      "epoch": 6.522369384765625e-05,
      "step": 68392,
      "training_step_time": 8.125518798828125
    },
    {
      "epoch": 6.522464752197266e-05,
      "model_forward_time": 1.7238664627075195,
      "step": 68393
    },
    {
      "epoch": 6.522464752197266e-05,
      "step": 68393,
      "training_step_time": 7.5637712478637695
    },
    {
      "epoch": 6.522560119628907e-05,
      "model_forward_time": 1.7194862365722656,
      "step": 68394
    },
    {
      "epoch": 6.522560119628907e-05,
      "step": 68394,
      "training_step_time": 7.617692708969116
    },
    {
      "epoch": 6.522655487060546e-05,
      "model_forward_time": 1.7265002727508545,
      "step": 68395
    },
    {
      "epoch": 6.522655487060546e-05,
      "step": 68395,
      "training_step_time": 8.37439250946045
    },
    {
      "epoch": 6.522750854492188e-05,
      "model_forward_time": 2.8553476333618164,
      "step": 68396
    },
    {
      "epoch": 6.522750854492188e-05,
      "step": 68396,
      "training_step_time": 8.46341061592102
    },
    {
      "epoch": 6.522846221923829e-05,
      "model_forward_time": 1.720954418182373,
      "step": 68397
    },
    {
      "epoch": 6.522846221923829e-05,
      "step": 68397,
      "training_step_time": 7.668816328048706
    },
    {
      "epoch": 6.522941589355468e-05,
      "model_forward_time": 1.7187623977661133,
      "step": 68398
    },
    {
      "epoch": 6.522941589355468e-05,
      "step": 68398,
      "training_step_time": 8.543415546417236
    },
    {
      "epoch": 6.52303695678711e-05,
      "model_forward_time": 1.7159171104431152,
      "step": 68399
    },
    {
      "epoch": 6.52303695678711e-05,
      "step": 68399,
      "training_step_time": 8.61773419380188
    },
    {
      "epoch": 6.52313232421875e-05,
      "grad_norm": 0.07335422933101654,
      "learning_rate": 2.4906027842062134e-06,
      "loss": 0.0522,
      "step": 68400
    },
    {
      "epoch": 6.52313232421875e-05,
      "model_forward_time": 1.7435095310211182,
      "step": 68400
    },
    {
      "dynamics_loss_avg": 0.050935913994908336,
      "epoch": 6.52313232421875e-05,
      "step": 68400
    },
    {
      "action_loss_avg": 0.002676094591151923,
      "epoch": 6.52313232421875e-05,
      "step": 68400
    },
    {
      "epoch": 6.52313232421875e-05,
      "step": 68400,
      "training_step_time": 9.019243001937866
    },
    {
      "epoch": 6.52322769165039e-05,
      "model_forward_time": 1.7229201793670654,
      "step": 68401
    },
    {
      "epoch": 6.52322769165039e-05,
      "step": 68401,
      "training_step_time": 8.550299167633057
    },
    {
      "epoch": 6.523323059082031e-05,
      "model_forward_time": 1.731734275817871,
      "step": 68402
    },
    {
      "epoch": 6.523323059082031e-05,
      "step": 68402,
      "training_step_time": 6.993720054626465
    },
    {
      "epoch": 6.523418426513672e-05,
      "model_forward_time": 1.728367567062378,
      "step": 68403
    },
    {
      "epoch": 6.523418426513672e-05,
      "step": 68403,
      "training_step_time": 7.030113697052002
    },
    {
      "epoch": 6.523513793945312e-05,
      "model_forward_time": 1.7096025943756104,
      "step": 68404
    },
    {
      "epoch": 6.523513793945312e-05,
      "step": 68404,
      "training_step_time": 7.069637060165405
    },
    {
      "epoch": 6.523609161376953e-05,
      "model_forward_time": 1.7242391109466553,
      "step": 68405
    },
    {
      "epoch": 6.523609161376953e-05,
      "step": 68405,
      "training_step_time": 7.0226874351501465
    },
    {
      "epoch": 6.523704528808594e-05,
      "model_forward_time": 1.357649803161621,
      "step": 68406
    },
    {
      "epoch": 6.523704528808594e-05,
      "step": 68406,
      "training_step_time": 6.961201429367065
    },
    {
      "epoch": 6.523799896240234e-05,
      "model_forward_time": 1.7230618000030518,
      "step": 68407
    },
    {
      "epoch": 6.523799896240234e-05,
      "step": 68407,
      "training_step_time": 7.03589391708374
    },
    {
      "epoch": 6.523895263671875e-05,
      "model_forward_time": 1.7188301086425781,
      "step": 68408
    },
    {
      "epoch": 6.523895263671875e-05,
      "step": 68408,
      "training_step_time": 6.963350057601929
    },
    {
      "epoch": 6.523990631103516e-05,
      "model_forward_time": 1.7267649173736572,
      "step": 68409
    },
    {
      "epoch": 6.523990631103516e-05,
      "step": 68409,
      "training_step_time": 6.888775110244751
    },
    {
      "epoch": 6.524085998535156e-05,
      "grad_norm": 0.06996476650238037,
      "learning_rate": 2.48917277325461e-06,
      "loss": 0.0539,
      "step": 68410
    },
    {
      "epoch": 6.524085998535156e-05,
      "model_forward_time": 2.8333628177642822,
      "step": 68410
    },
    {
      "dynamics_loss_avg": 0.0641864025965333,
      "epoch": 6.524085998535156e-05,
      "step": 68410
    },
    {
      "action_loss_avg": 0.001311551165417768,
      "epoch": 6.524085998535156e-05,
      "step": 68410
    },
    {
      "epoch": 6.524085998535156e-05,
      "step": 68410,
      "training_step_time": 11.035617113113403
    },
    {
      "epoch": 6.524181365966797e-05,
      "model_forward_time": 1.71897554397583,
      "step": 68411
    },
    {
      "epoch": 6.524181365966797e-05,
      "step": 68411,
      "training_step_time": 7.954774618148804
    },
    {
      "epoch": 6.524276733398438e-05,
      "model_forward_time": 1.7307462692260742,
      "step": 68412
    },
    {
      "epoch": 6.524276733398438e-05,
      "step": 68412,
      "training_step_time": 6.946062326431274
    },
    {
      "epoch": 6.524372100830078e-05,
      "model_forward_time": 1.7196550369262695,
      "step": 68413
    },
    {
      "epoch": 6.524372100830078e-05,
      "step": 68413,
      "training_step_time": 6.856436252593994
    },
    {
      "epoch": 6.524467468261719e-05,
      "model_forward_time": 1.7346839904785156,
      "step": 68414
    },
    {
      "epoch": 6.524467468261719e-05,
      "step": 68414,
      "training_step_time": 6.897496461868286
    },
    {
      "epoch": 6.52456283569336e-05,
      "model_forward_time": 1.7241714000701904,
      "step": 68415
    },
    {
      "epoch": 6.52456283569336e-05,
      "step": 68415,
      "training_step_time": 6.974249362945557
    },
    {
      "epoch": 6.524658203125e-05,
      "model_forward_time": 1.721306324005127,
      "step": 68416
    },
    {
      "epoch": 6.524658203125e-05,
      "step": 68416,
      "training_step_time": 6.895884037017822
    },
    {
      "epoch": 6.524753570556641e-05,
      "model_forward_time": 1.7245514392852783,
      "step": 68417
    },
    {
      "epoch": 6.524753570556641e-05,
      "step": 68417,
      "training_step_time": 6.820062160491943
    },
    {
      "epoch": 6.524848937988282e-05,
      "model_forward_time": 1.7212989330291748,
      "step": 68418
    },
    {
      "epoch": 6.524848937988282e-05,
      "step": 68418,
      "training_step_time": 6.987445116043091
    },
    {
      "epoch": 6.524944305419922e-05,
      "model_forward_time": 2.85801100730896,
      "step": 68419
    },
    {
      "epoch": 6.524944305419922e-05,
      "step": 68419,
      "training_step_time": 10.70098876953125
    },
    {
      "epoch": 6.525039672851563e-05,
      "grad_norm": 0.1024940237402916,
      "learning_rate": 2.487743036883295e-06,
      "loss": 0.0553,
      "step": 68420
    },
    {
      "epoch": 6.525039672851563e-05,
      "model_forward_time": 1.7276558876037598,
      "step": 68420
    },
    {
      "dynamics_loss_avg": 0.06303106341511011,
      "epoch": 6.525039672851563e-05,
      "step": 68420
    },
    {
      "action_loss_avg": 0.0010536229994613678,
      "epoch": 6.525039672851563e-05,
      "step": 68420
    },
    {
      "epoch": 6.525039672851563e-05,
      "step": 68420,
      "training_step_time": 10.803001880645752
    },
    {
      "epoch": 6.525135040283204e-05,
      "model_forward_time": 1.722365379333496,
      "step": 68421
    },
    {
      "epoch": 6.525135040283204e-05,
      "step": 68421,
      "training_step_time": 6.9224348068237305
    },
    {
      "epoch": 6.525230407714843e-05,
      "model_forward_time": 1.7253611087799072,
      "step": 68422
    },
    {
      "epoch": 6.525230407714843e-05,
      "step": 68422,
      "training_step_time": 6.831745624542236
    },
    {
      "epoch": 6.525325775146485e-05,
      "model_forward_time": 1.7261130809783936,
      "step": 68423
    },
    {
      "epoch": 6.525325775146485e-05,
      "step": 68423,
      "training_step_time": 6.834383726119995
    },
    {
      "epoch": 6.525421142578126e-05,
      "model_forward_time": 1.7241127490997314,
      "step": 68424
    },
    {
      "epoch": 6.525421142578126e-05,
      "step": 68424,
      "training_step_time": 6.905472040176392
    },
    {
      "epoch": 6.525516510009765e-05,
      "model_forward_time": 1.7282652854919434,
      "step": 68425
    },
    {
      "epoch": 6.525516510009765e-05,
      "step": 68425,
      "training_step_time": 6.86116623878479
    },
    {
      "epoch": 6.525611877441406e-05,
      "model_forward_time": 1.7181830406188965,
      "step": 68426
    },
    {
      "epoch": 6.525611877441406e-05,
      "step": 68426,
      "training_step_time": 6.923375368118286
    },
    {
      "epoch": 6.525707244873048e-05,
      "model_forward_time": 1.7244036197662354,
      "step": 68427
    },
    {
      "epoch": 6.525707244873048e-05,
      "step": 68427,
      "training_step_time": 7.011475324630737
    },
    {
      "epoch": 6.525802612304687e-05,
      "model_forward_time": 2.8592307567596436,
      "step": 68428
    },
    {
      "epoch": 6.525802612304687e-05,
      "step": 68428,
      "training_step_time": 9.491393327713013
    },
    {
      "epoch": 6.525897979736328e-05,
      "model_forward_time": 1.7291688919067383,
      "step": 68429
    },
    {
      "epoch": 6.525897979736328e-05,
      "step": 68429,
      "training_step_time": 7.994756698608398
    },
    {
      "epoch": 6.52599334716797e-05,
      "grad_norm": 0.14134344458580017,
      "learning_rate": 2.4863135752486262e-06,
      "loss": 0.0563,
      "step": 68430
    },
    {
      "epoch": 6.52599334716797e-05,
      "model_forward_time": 1.7244529724121094,
      "step": 68430
    },
    {
      "dynamics_loss_avg": 0.04766157800331712,
      "epoch": 6.52599334716797e-05,
      "step": 68430
    },
    {
      "action_loss_avg": 0.00223568283545319,
      "epoch": 6.52599334716797e-05,
      "step": 68430
    },
    {
      "epoch": 6.52599334716797e-05,
      "step": 68430,
      "training_step_time": 7.816043138504028
    },
    {
      "epoch": 6.526088714599609e-05,
      "model_forward_time": 1.717411994934082,
      "step": 68431
    },
    {
      "epoch": 6.526088714599609e-05,
      "step": 68431,
      "training_step_time": 6.8721702098846436
    },
    {
      "epoch": 6.52618408203125e-05,
      "model_forward_time": 1.722276210784912,
      "step": 68432
    },
    {
      "epoch": 6.52618408203125e-05,
      "step": 68432,
      "training_step_time": 6.855916261672974
    },
    {
      "epoch": 6.526279449462891e-05,
      "model_forward_time": 1.7289376258850098,
      "step": 68433
    },
    {
      "epoch": 6.526279449462891e-05,
      "step": 68433,
      "training_step_time": 6.912866592407227
    },
    {
      "epoch": 6.526374816894531e-05,
      "model_forward_time": 1.7266833782196045,
      "step": 68434
    },
    {
      "epoch": 6.526374816894531e-05,
      "step": 68434,
      "training_step_time": 6.803537368774414
    },
    {
      "epoch": 6.526470184326172e-05,
      "model_forward_time": 1.7226157188415527,
      "step": 68435
    },
    {
      "epoch": 6.526470184326172e-05,
      "step": 68435,
      "training_step_time": 7.015153169631958
    },
    {
      "epoch": 6.526565551757813e-05,
      "model_forward_time": 1.7208797931671143,
      "step": 68436
    },
    {
      "epoch": 6.526565551757813e-05,
      "step": 68436,
      "training_step_time": 6.881144046783447
    },
    {
      "epoch": 6.526660919189453e-05,
      "model_forward_time": 2.8946549892425537,
      "step": 68437
    },
    {
      "epoch": 6.526660919189453e-05,
      "step": 68437,
      "training_step_time": 7.901016712188721
    },
    {
      "epoch": 6.526756286621094e-05,
      "model_forward_time": 1.7210662364959717,
      "step": 68438
    },
    {
      "epoch": 6.526756286621094e-05,
      "step": 68438,
      "training_step_time": 11.304184913635254
    },
    {
      "epoch": 6.526851654052734e-05,
      "model_forward_time": 1.7299137115478516,
      "step": 68439
    },
    {
      "epoch": 6.526851654052734e-05,
      "step": 68439,
      "training_step_time": 7.813952207565308
    },
    {
      "epoch": 6.526947021484375e-05,
      "grad_norm": 0.09351389855146408,
      "learning_rate": 2.4848843885069236e-06,
      "loss": 0.0551,
      "step": 68440
    },
    {
      "epoch": 6.526947021484375e-05,
      "model_forward_time": 1.7282989025115967,
      "step": 68440
    },
    {
      "dynamics_loss_avg": 0.06485534813255071,
      "epoch": 6.526947021484375e-05,
      "step": 68440
    },
    {
      "action_loss_avg": 0.0016397904546465725,
      "epoch": 6.526947021484375e-05,
      "step": 68440
    },
    {
      "epoch": 6.526947021484375e-05,
      "step": 68440,
      "training_step_time": 6.906438589096069
    },
    {
      "epoch": 6.527042388916016e-05,
      "model_forward_time": 1.7217094898223877,
      "step": 68441
    },
    {
      "epoch": 6.527042388916016e-05,
      "step": 68441,
      "training_step_time": 6.869751214981079
    },
    {
      "epoch": 6.527137756347656e-05,
      "model_forward_time": 1.7151904106140137,
      "step": 68442
    },
    {
      "epoch": 6.527137756347656e-05,
      "step": 68442,
      "training_step_time": 6.8963303565979
    },
    {
      "epoch": 6.527233123779297e-05,
      "model_forward_time": 1.7221522331237793,
      "step": 68443
    },
    {
      "epoch": 6.527233123779297e-05,
      "step": 68443,
      "training_step_time": 6.809894561767578
    },
    {
      "epoch": 6.527328491210938e-05,
      "model_forward_time": 1.7289538383483887,
      "step": 68444
    },
    {
      "epoch": 6.527328491210938e-05,
      "step": 68444,
      "training_step_time": 6.792893648147583
    },
    {
      "epoch": 6.527423858642577e-05,
      "model_forward_time": 1.7213516235351562,
      "step": 68445
    },
    {
      "epoch": 6.527423858642577e-05,
      "step": 68445,
      "training_step_time": 6.851813077926636
    },
    {
      "epoch": 6.527519226074219e-05,
      "model_forward_time": 1.7186319828033447,
      "step": 68446
    },
    {
      "epoch": 6.527519226074219e-05,
      "step": 68446,
      "training_step_time": 8.00035572052002
    },
    {
      "epoch": 6.52761459350586e-05,
      "model_forward_time": 1.7439115047454834,
      "step": 68447
    },
    {
      "epoch": 6.52761459350586e-05,
      "step": 68447,
      "training_step_time": 10.767574787139893
    },
    {
      "epoch": 6.5277099609375e-05,
      "model_forward_time": 1.7323157787322998,
      "step": 68448
    },
    {
      "epoch": 6.5277099609375e-05,
      "step": 68448,
      "training_step_time": 7.780827522277832
    },
    {
      "epoch": 6.52780532836914e-05,
      "model_forward_time": 1.743720293045044,
      "step": 68449
    },
    {
      "epoch": 6.52780532836914e-05,
      "step": 68449,
      "training_step_time": 7.266522407531738
    },
    {
      "epoch": 6.527900695800782e-05,
      "grad_norm": 0.0987987220287323,
      "learning_rate": 2.4834554768144843e-06,
      "loss": 0.0557,
      "step": 68450
    },
    {
      "epoch": 6.527900695800782e-05,
      "model_forward_time": 1.372161865234375,
      "step": 68450
    },
    {
      "dynamics_loss_avg": 0.06118877120316028,
      "epoch": 6.527900695800782e-05,
      "step": 68450
    },
    {
      "action_loss_avg": 0.0010253110391204245,
      "epoch": 6.527900695800782e-05,
      "step": 68450
    },
    {
      "epoch": 6.527900695800782e-05,
      "step": 68450,
      "training_step_time": 6.924678802490234
    },
    {
      "epoch": 6.527996063232421e-05,
      "model_forward_time": 1.7398452758789062,
      "step": 68451
    },
    {
      "epoch": 6.527996063232421e-05,
      "step": 68451,
      "training_step_time": 6.927858114242554
    },
    {
      "epoch": 6.528091430664062e-05,
      "model_forward_time": 1.7311758995056152,
      "step": 68452
    },
    {
      "epoch": 6.528091430664062e-05,
      "step": 68452,
      "training_step_time": 6.894744634628296
    },
    {
      "epoch": 6.528186798095703e-05,
      "model_forward_time": 1.728057861328125,
      "step": 68453
    },
    {
      "epoch": 6.528186798095703e-05,
      "step": 68453,
      "training_step_time": 6.879943132400513
    },
    {
      "epoch": 6.528282165527343e-05,
      "model_forward_time": 1.7396454811096191,
      "step": 68454
    },
    {
      "epoch": 6.528282165527343e-05,
      "step": 68454,
      "training_step_time": 6.829270839691162
    },
    {
      "epoch": 6.528377532958984e-05,
      "model_forward_time": 1.7372050285339355,
      "step": 68455
    },
    {
      "epoch": 6.528377532958984e-05,
      "step": 68455,
      "training_step_time": 7.655801296234131
    },
    {
      "epoch": 6.528472900390625e-05,
      "model_forward_time": 2.971571207046509,
      "step": 68456
    },
    {
      "epoch": 6.528472900390625e-05,
      "step": 68456,
      "training_step_time": 10.420119524002075
    },
    {
      "epoch": 6.528568267822265e-05,
      "model_forward_time": 1.7349627017974854,
      "step": 68457
    },
    {
      "epoch": 6.528568267822265e-05,
      "step": 68457,
      "training_step_time": 7.883512258529663
    },
    {
      "epoch": 6.528663635253906e-05,
      "model_forward_time": 1.738264560699463,
      "step": 68458
    },
    {
      "epoch": 6.528663635253906e-05,
      "step": 68458,
      "training_step_time": 7.679414749145508
    },
    {
      "epoch": 6.528759002685547e-05,
      "model_forward_time": 1.74237060546875,
      "step": 68459
    },
    {
      "epoch": 6.528759002685547e-05,
      "step": 68459,
      "training_step_time": 6.6881263256073
    },
    {
      "epoch": 6.528854370117187e-05,
      "grad_norm": 0.0844726413488388,
      "learning_rate": 2.482026840327567e-06,
      "loss": 0.0549,
      "step": 68460
    },
    {
      "epoch": 6.528854370117187e-05,
      "model_forward_time": 1.7526137828826904,
      "step": 68460
    },
    {
      "dynamics_loss_avg": 0.04326251931488514,
      "epoch": 6.528854370117187e-05,
      "step": 68460
    },
    {
      "action_loss_avg": 0.0010165734332986177,
      "epoch": 6.528854370117187e-05,
      "step": 68460
    },
    {
      "epoch": 6.528854370117187e-05,
      "step": 68460,
      "training_step_time": 6.745009899139404
    },
    {
      "epoch": 6.528949737548828e-05,
      "model_forward_time": 1.737762689590454,
      "step": 68461
    },
    {
      "epoch": 6.528949737548828e-05,
      "step": 68461,
      "training_step_time": 6.7510316371917725
    },
    {
      "epoch": 6.529045104980469e-05,
      "model_forward_time": 1.7410974502563477,
      "step": 68462
    },
    {
      "epoch": 6.529045104980469e-05,
      "step": 68462,
      "training_step_time": 6.725471496582031
    },
    {
      "epoch": 6.529140472412109e-05,
      "model_forward_time": 1.7400014400482178,
      "step": 68463
    },
    {
      "epoch": 6.529140472412109e-05,
      "step": 68463,
      "training_step_time": 6.727081775665283
    },
    {
      "epoch": 6.52923583984375e-05,
      "model_forward_time": 1.7350263595581055,
      "step": 68464
    },
    {
      "epoch": 6.52923583984375e-05,
      "step": 68464,
      "training_step_time": 6.8261518478393555
    },
    {
      "epoch": 6.529331207275391e-05,
      "model_forward_time": 2.8552019596099854,
      "step": 68465
    },
    {
      "epoch": 6.529331207275391e-05,
      "step": 68465,
      "training_step_time": 7.747162103652954
    },
    {
      "epoch": 6.52942657470703e-05,
      "model_forward_time": 1.7473456859588623,
      "step": 68466
    },
    {
      "epoch": 6.52942657470703e-05,
      "step": 68466,
      "training_step_time": 8.56856369972229
    },
    {
      "epoch": 6.529521942138672e-05,
      "model_forward_time": 1.7259423732757568,
      "step": 68467
    },
    {
      "epoch": 6.529521942138672e-05,
      "step": 68467,
      "training_step_time": 7.622280597686768
    },
    {
      "epoch": 6.529617309570313e-05,
      "model_forward_time": 1.7413280010223389,
      "step": 68468
    },
    {
      "epoch": 6.529617309570313e-05,
      "step": 68468,
      "training_step_time": 6.823206663131714
    },
    {
      "epoch": 6.529712677001953e-05,
      "model_forward_time": 1.735689401626587,
      "step": 68469
    },
    {
      "epoch": 6.529712677001953e-05,
      "step": 68469,
      "training_step_time": 6.74944281578064
    },
    {
      "epoch": 6.529808044433594e-05,
      "grad_norm": 0.08069486171007156,
      "learning_rate": 2.480598479202413e-06,
      "loss": 0.0551,
      "step": 68470
    },
    {
      "epoch": 6.529808044433594e-05,
      "model_forward_time": 1.7480401992797852,
      "step": 68470
    },
    {
      "dynamics_loss_avg": 0.07001680890098214,
      "epoch": 6.529808044433594e-05,
      "step": 68470
    },
    {
      "action_loss_avg": 0.0018013771157711744,
      "epoch": 6.529808044433594e-05,
      "step": 68470
    },
    {
      "epoch": 6.529808044433594e-05,
      "step": 68470,
      "training_step_time": 6.740811109542847
    },
    {
      "epoch": 6.529903411865235e-05,
      "model_forward_time": 1.738269329071045,
      "step": 68471
    },
    {
      "epoch": 6.529903411865235e-05,
      "step": 68471,
      "training_step_time": 6.802116870880127
    },
    {
      "epoch": 6.529998779296874e-05,
      "model_forward_time": 1.7366607189178467,
      "step": 68472
    },
    {
      "epoch": 6.529998779296874e-05,
      "step": 68472,
      "training_step_time": 6.755731105804443
    },
    {
      "epoch": 6.530094146728516e-05,
      "model_forward_time": 1.7432045936584473,
      "step": 68473
    },
    {
      "epoch": 6.530094146728516e-05,
      "step": 68473,
      "training_step_time": 6.712970018386841
    },
    {
      "epoch": 6.530189514160157e-05,
      "model_forward_time": 2.9513325691223145,
      "step": 68474
    },
    {
      "epoch": 6.530189514160157e-05,
      "step": 68474,
      "training_step_time": 7.733919858932495
    },
    {
      "epoch": 6.530284881591796e-05,
      "model_forward_time": 1.7361910343170166,
      "step": 68475
    },
    {
      "epoch": 6.530284881591796e-05,
      "step": 68475,
      "training_step_time": 8.875524759292603
    },
    {
      "epoch": 6.530380249023437e-05,
      "model_forward_time": 1.7431271076202393,
      "step": 68476
    },
    {
      "epoch": 6.530380249023437e-05,
      "step": 68476,
      "training_step_time": 7.650015354156494
    },
    {
      "epoch": 6.530475616455078e-05,
      "model_forward_time": 1.736063003540039,
      "step": 68477
    },
    {
      "epoch": 6.530475616455078e-05,
      "step": 68477,
      "training_step_time": 6.718674659729004
    },
    {
      "epoch": 6.530570983886718e-05,
      "model_forward_time": 1.7390475273132324,
      "step": 68478
    },
    {
      "epoch": 6.530570983886718e-05,
      "step": 68478,
      "training_step_time": 6.8583221435546875
    },
    {
      "epoch": 6.530666351318359e-05,
      "model_forward_time": 1.749957799911499,
      "step": 68479
    },
    {
      "epoch": 6.530666351318359e-05,
      "step": 68479,
      "training_step_time": 6.838798999786377
    },
    {
      "epoch": 6.53076171875e-05,
      "grad_norm": 0.126083105802536,
      "learning_rate": 2.4791703935952193e-06,
      "loss": 0.0555,
      "step": 68480
    },
    {
      "epoch": 6.53076171875e-05,
      "model_forward_time": 1.7460052967071533,
      "step": 68480
    },
    {
      "dynamics_loss_avg": 0.05154620800167322,
      "epoch": 6.53076171875e-05,
      "step": 68480
    },
    {
      "action_loss_avg": 0.0012604810181073844,
      "epoch": 6.53076171875e-05,
      "step": 68480
    },
    {
      "epoch": 6.53076171875e-05,
      "step": 68480,
      "training_step_time": 6.734345436096191
    },
    {
      "epoch": 6.53085708618164e-05,
      "model_forward_time": 1.741912603378296,
      "step": 68481
    },
    {
      "epoch": 6.53085708618164e-05,
      "step": 68481,
      "training_step_time": 6.739107370376587
    },
    {
      "epoch": 6.530952453613281e-05,
      "model_forward_time": 1.7381417751312256,
      "step": 68482
    },
    {
      "epoch": 6.530952453613281e-05,
      "step": 68482,
      "training_step_time": 6.83136510848999
    },
    {
      "epoch": 6.531047821044922e-05,
      "model_forward_time": 2.907982349395752,
      "step": 68483
    },
    {
      "epoch": 6.531047821044922e-05,
      "step": 68483,
      "training_step_time": 7.751323223114014
    },
    {
      "epoch": 6.531143188476562e-05,
      "model_forward_time": 1.7385458946228027,
      "step": 68484
    },
    {
      "epoch": 6.531143188476562e-05,
      "step": 68484,
      "training_step_time": 10.476296663284302
    },
    {
      "epoch": 6.531238555908203e-05,
      "model_forward_time": 1.7387363910675049,
      "step": 68485
    },
    {
      "epoch": 6.531238555908203e-05,
      "step": 68485,
      "training_step_time": 7.701124906539917
    },
    {
      "epoch": 6.531333923339844e-05,
      "model_forward_time": 1.7429983615875244,
      "step": 68486
    },
    {
      "epoch": 6.531333923339844e-05,
      "step": 68486,
      "training_step_time": 7.679238319396973
    },
    {
      "epoch": 6.531429290771484e-05,
      "model_forward_time": 1.740906000137329,
      "step": 68487
    },
    {
      "epoch": 6.531429290771484e-05,
      "step": 68487,
      "training_step_time": 6.8467934131622314
    },
    {
      "epoch": 6.531524658203125e-05,
      "model_forward_time": 1.737614631652832,
      "step": 68488
    },
    {
      "epoch": 6.531524658203125e-05,
      "step": 68488,
      "training_step_time": 6.899939775466919
    },
    {
      "epoch": 6.531620025634766e-05,
      "model_forward_time": 1.7399768829345703,
      "step": 68489
    },
    {
      "epoch": 6.531620025634766e-05,
      "step": 68489,
      "training_step_time": 6.88305139541626
    },
    {
      "epoch": 6.531715393066406e-05,
      "grad_norm": 0.1115349531173706,
      "learning_rate": 2.4777425836621634e-06,
      "loss": 0.0573,
      "step": 68490
    },
    {
      "epoch": 6.531715393066406e-05,
      "model_forward_time": 1.7355971336364746,
      "step": 68490
    },
    {
      "dynamics_loss_avg": 0.06845568288117647,
      "epoch": 6.531715393066406e-05,
      "step": 68490
    },
    {
      "action_loss_avg": 0.00224092694115825,
      "epoch": 6.531715393066406e-05,
      "step": 68490
    },
    {
      "epoch": 6.531715393066406e-05,
      "step": 68490,
      "training_step_time": 6.960465908050537
    },
    {
      "epoch": 6.531810760498047e-05,
      "model_forward_time": 1.7356009483337402,
      "step": 68491
    },
    {
      "epoch": 6.531810760498047e-05,
      "step": 68491,
      "training_step_time": 6.864349365234375
    },
    {
      "epoch": 6.531906127929688e-05,
      "model_forward_time": 1.8070805072784424,
      "step": 68492
    },
    {
      "epoch": 6.531906127929688e-05,
      "step": 68492,
      "training_step_time": 8.568233728408813
    },
    {
      "epoch": 6.532001495361328e-05,
      "model_forward_time": 2.8713250160217285,
      "step": 68493
    },
    {
      "epoch": 6.532001495361328e-05,
      "step": 68493,
      "training_step_time": 10.738141298294067
    },
    {
      "epoch": 6.532096862792969e-05,
      "model_forward_time": 1.749765396118164,
      "step": 68494
    },
    {
      "epoch": 6.532096862792969e-05,
      "step": 68494,
      "training_step_time": 7.7176501750946045
    },
    {
      "epoch": 6.53219223022461e-05,
      "model_forward_time": 1.742501974105835,
      "step": 68495
    },
    {
      "epoch": 6.53219223022461e-05,
      "step": 68495,
      "training_step_time": 7.65504789352417
    },
    {
      "epoch": 6.53228759765625e-05,
      "model_forward_time": 1.740464687347412,
      "step": 68496
    },
    {
      "epoch": 6.53228759765625e-05,
      "step": 68496,
      "training_step_time": 6.832165718078613
    },
    {
      "epoch": 6.53238296508789e-05,
      "model_forward_time": 1.7378904819488525,
      "step": 68497
    },
    {
      "epoch": 6.53238296508789e-05,
      "step": 68497,
      "training_step_time": 6.776187896728516
    },
    {
      "epoch": 6.532478332519532e-05,
      "model_forward_time": 1.7374353408813477,
      "step": 68498
    },
    {
      "epoch": 6.532478332519532e-05,
      "step": 68498,
      "training_step_time": 6.756611347198486
    },
    {
      "epoch": 6.532573699951171e-05,
      "model_forward_time": 1.7382011413574219,
      "step": 68499
    },
    {
      "epoch": 6.532573699951171e-05,
      "step": 68499,
      "training_step_time": 6.822854042053223
    },
    {
      "epoch": 6.532669067382812e-05,
      "grad_norm": 0.10701357573270798,
      "learning_rate": 2.476315049559384e-06,
      "loss": 0.0577,
      "step": 68500
    },
    {
      "epoch": 6.532669067382812e-05,
      "model_forward_time": 1.7328975200653076,
      "step": 68500
    },
    {
      "dynamics_loss_avg": 0.062098391354084015,
      "epoch": 6.532669067382812e-05,
      "step": 68500
    },
    {
      "action_loss_avg": 0.0013575797813246027,
      "epoch": 6.532669067382812e-05,
      "step": 68500
    },
    {
      "epoch": 6.532669067382812e-05,
      "step": 68500,
      "training_step_time": 6.786397695541382
    },
    {
      "epoch": 6.532764434814454e-05,
      "model_forward_time": 1.8050270080566406,
      "step": 68501
    },
    {
      "epoch": 6.532764434814454e-05,
      "step": 68501,
      "training_step_time": 7.764343738555908
    },
    {
      "epoch": 6.532859802246093e-05,
      "model_forward_time": 2.9523701667785645,
      "step": 68502
    },
    {
      "epoch": 6.532859802246093e-05,
      "step": 68502,
      "training_step_time": 10.840315103530884
    },
    {
      "epoch": 6.532955169677734e-05,
      "model_forward_time": 1.7462148666381836,
      "step": 68503
    },
    {
      "epoch": 6.532955169677734e-05,
      "step": 68503,
      "training_step_time": 8.462873220443726
    },
    {
      "epoch": 6.533050537109375e-05,
      "model_forward_time": 1.7365467548370361,
      "step": 68504
    },
    {
      "epoch": 6.533050537109375e-05,
      "step": 68504,
      "training_step_time": 7.710417985916138
    },
    {
      "epoch": 6.533145904541015e-05,
      "model_forward_time": 1.7343730926513672,
      "step": 68505
    },
    {
      "epoch": 6.533145904541015e-05,
      "step": 68505,
      "training_step_time": 6.769315719604492
    },
    {
      "epoch": 6.533241271972656e-05,
      "model_forward_time": 1.7466833591461182,
      "step": 68506
    },
    {
      "epoch": 6.533241271972656e-05,
      "step": 68506,
      "training_step_time": 6.710616588592529
    },
    {
      "epoch": 6.533336639404297e-05,
      "model_forward_time": 1.7673592567443848,
      "step": 68507
    },
    {
      "epoch": 6.533336639404297e-05,
      "step": 68507,
      "training_step_time": 6.772185325622559
    },
    {
      "epoch": 6.533432006835937e-05,
      "model_forward_time": 1.737259864807129,
      "step": 68508
    },
    {
      "epoch": 6.533432006835937e-05,
      "step": 68508,
      "training_step_time": 6.710198879241943
    },
    {
      "epoch": 6.533527374267578e-05,
      "model_forward_time": 1.7438759803771973,
      "step": 68509
    },
    {
      "epoch": 6.533527374267578e-05,
      "step": 68509,
      "training_step_time": 6.777071475982666
    },
    {
      "epoch": 6.533622741699219e-05,
      "grad_norm": 0.1492186039686203,
      "learning_rate": 2.4748877914429987e-06,
      "loss": 0.0568,
      "step": 68510
    },
    {
      "epoch": 6.533622741699219e-05,
      "model_forward_time": 1.8055601119995117,
      "step": 68510
    },
    {
      "dynamics_loss_avg": 0.06303441300988197,
      "epoch": 6.533622741699219e-05,
      "step": 68510
    },
    {
      "action_loss_avg": 0.0029042881680652497,
      "epoch": 6.533622741699219e-05,
      "step": 68510
    },
    {
      "epoch": 6.533622741699219e-05,
      "step": 68510,
      "training_step_time": 7.7646403312683105
    },
    {
      "epoch": 6.533718109130859e-05,
      "model_forward_time": 2.8999085426330566,
      "step": 68511
    },
    {
      "epoch": 6.533718109130859e-05,
      "step": 68511,
      "training_step_time": 9.22527003288269
    },
    {
      "epoch": 6.5338134765625e-05,
      "model_forward_time": 1.7500200271606445,
      "step": 68512
    },
    {
      "epoch": 6.5338134765625e-05,
      "step": 68512,
      "training_step_time": 10.070072650909424
    },
    {
      "epoch": 6.533908843994141e-05,
      "model_forward_time": 1.7369203567504883,
      "step": 68513
    },
    {
      "epoch": 6.533908843994141e-05,
      "step": 68513,
      "training_step_time": 7.736929416656494
    },
    {
      "epoch": 6.534004211425781e-05,
      "model_forward_time": 1.7456591129302979,
      "step": 68514
    },
    {
      "epoch": 6.534004211425781e-05,
      "step": 68514,
      "training_step_time": 6.853057146072388
    },
    {
      "epoch": 6.534099578857422e-05,
      "model_forward_time": 1.7542157173156738,
      "step": 68515
    },
    {
      "epoch": 6.534099578857422e-05,
      "step": 68515,
      "training_step_time": 6.818514347076416
    },
    {
      "epoch": 6.534194946289063e-05,
      "model_forward_time": 1.745002269744873,
      "step": 68516
    },
    {
      "epoch": 6.534194946289063e-05,
      "step": 68516,
      "training_step_time": 6.874407052993774
    },
    {
      "epoch": 6.534290313720703e-05,
      "model_forward_time": 1.736494541168213,
      "step": 68517
    },
    {
      "epoch": 6.534290313720703e-05,
      "step": 68517,
      "training_step_time": 6.803291082382202
    },
    {
      "epoch": 6.534385681152344e-05,
      "model_forward_time": 1.742037296295166,
      "step": 68518
    },
    {
      "epoch": 6.534385681152344e-05,
      "step": 68518,
      "training_step_time": 6.880376100540161
    },
    {
      "epoch": 6.534481048583985e-05,
      "model_forward_time": 1.823627233505249,
      "step": 68519
    },
    {
      "epoch": 6.534481048583985e-05,
      "step": 68519,
      "training_step_time": 7.817561388015747
    },
    {
      "epoch": 6.534576416015625e-05,
      "grad_norm": 0.07998969405889511,
      "learning_rate": 2.473460809469086e-06,
      "loss": 0.0601,
      "step": 68520
    },
    {
      "epoch": 6.534576416015625e-05,
      "model_forward_time": 2.8717668056488037,
      "step": 68520
    },
    {
      "dynamics_loss_avg": 0.06995028890669346,
      "epoch": 6.534576416015625e-05,
      "step": 68520
    },
    {
      "action_loss_avg": 0.0012773035938153043,
      "epoch": 6.534576416015625e-05,
      "step": 68520
    },
    {
      "epoch": 6.534576416015625e-05,
      "step": 68520,
      "training_step_time": 9.870965003967285
    },
    {
      "epoch": 6.534671783447266e-05,
      "model_forward_time": 1.743523359298706,
      "step": 68521
    },
    {
      "epoch": 6.534671783447266e-05,
      "step": 68521,
      "training_step_time": 8.521538734436035
    },
    {
      "epoch": 6.534767150878907e-05,
      "model_forward_time": 1.7438654899597168,
      "step": 68522
    },
    {
      "epoch": 6.534767150878907e-05,
      "step": 68522,
      "training_step_time": 8.429601192474365
    },
    {
      "epoch": 6.534862518310547e-05,
      "model_forward_time": 1.7396879196166992,
      "step": 68523
    },
    {
      "epoch": 6.534862518310547e-05,
      "step": 68523,
      "training_step_time": 7.78007698059082
    },
    {
      "epoch": 6.534957885742188e-05,
      "model_forward_time": 1.7503633499145508,
      "step": 68524
    },
    {
      "epoch": 6.534957885742188e-05,
      "step": 68524,
      "training_step_time": 6.780203104019165
    },
    {
      "epoch": 6.535053253173829e-05,
      "model_forward_time": 1.7454917430877686,
      "step": 68525
    },
    {
      "epoch": 6.535053253173829e-05,
      "step": 68525,
      "training_step_time": 6.848251819610596
    },
    {
      "epoch": 6.535148620605468e-05,
      "model_forward_time": 1.739889144897461,
      "step": 68526
    },
    {
      "epoch": 6.535148620605468e-05,
      "step": 68526,
      "training_step_time": 6.891253471374512
    },
    {
      "epoch": 6.53524398803711e-05,
      "model_forward_time": 1.7408592700958252,
      "step": 68527
    },
    {
      "epoch": 6.53524398803711e-05,
      "step": 68527,
      "training_step_time": 6.755132675170898
    },
    {
      "epoch": 6.53533935546875e-05,
      "model_forward_time": 1.806563377380371,
      "step": 68528
    },
    {
      "epoch": 6.53533935546875e-05,
      "step": 68528,
      "training_step_time": 7.835833311080933
    },
    {
      "epoch": 6.53543472290039e-05,
      "model_forward_time": 2.877687692642212,
      "step": 68529
    },
    {
      "epoch": 6.53543472290039e-05,
      "step": 68529,
      "training_step_time": 8.93124508857727
    },
    {
      "epoch": 6.535530090332031e-05,
      "grad_norm": 0.13569030165672302,
      "learning_rate": 2.4720341037937007e-06,
      "loss": 0.0573,
      "step": 68530
    },
    {
      "epoch": 6.535530090332031e-05,
      "model_forward_time": 1.7511498928070068,
      "step": 68530
    },
    {
      "dynamics_loss_avg": 0.06639947053045034,
      "epoch": 6.535530090332031e-05,
      "step": 68530
    },
    {
      "action_loss_avg": 0.0013710376442759298,
      "epoch": 6.535530090332031e-05,
      "step": 68530
    },
    {
      "epoch": 6.535530090332031e-05,
      "step": 68530,
      "training_step_time": 9.878774166107178
    },
    {
      "epoch": 6.535625457763672e-05,
      "model_forward_time": 1.744152307510376,
      "step": 68531
    },
    {
      "epoch": 6.535625457763672e-05,
      "step": 68531,
      "training_step_time": 8.497253179550171
    },
    {
      "epoch": 6.535720825195312e-05,
      "model_forward_time": 1.753438949584961,
      "step": 68532
    },
    {
      "epoch": 6.535720825195312e-05,
      "step": 68532,
      "training_step_time": 7.757181167602539
    },
    {
      "epoch": 6.535816192626953e-05,
      "model_forward_time": 1.7409019470214844,
      "step": 68533
    },
    {
      "epoch": 6.535816192626953e-05,
      "step": 68533,
      "training_step_time": 6.791371583938599
    },
    {
      "epoch": 6.535911560058594e-05,
      "model_forward_time": 1.7451224327087402,
      "step": 68534
    },
    {
      "epoch": 6.535911560058594e-05,
      "step": 68534,
      "training_step_time": 6.8606414794921875
    },
    {
      "epoch": 6.536006927490234e-05,
      "model_forward_time": 1.7421212196350098,
      "step": 68535
    },
    {
      "epoch": 6.536006927490234e-05,
      "step": 68535,
      "training_step_time": 6.877035856246948
    },
    {
      "epoch": 6.536102294921875e-05,
      "model_forward_time": 1.742143154144287,
      "step": 68536
    },
    {
      "epoch": 6.536102294921875e-05,
      "step": 68536,
      "training_step_time": 6.8822455406188965
    },
    {
      "epoch": 6.536197662353516e-05,
      "model_forward_time": 1.8025953769683838,
      "step": 68537
    },
    {
      "epoch": 6.536197662353516e-05,
      "step": 68537,
      "training_step_time": 7.775795221328735
    },
    {
      "epoch": 6.536293029785156e-05,
      "model_forward_time": 1.7386915683746338,
      "step": 68538
    },
    {
      "epoch": 6.536293029785156e-05,
      "step": 68538,
      "training_step_time": 8.071686506271362
    },
    {
      "epoch": 6.536388397216797e-05,
      "model_forward_time": 1.741091012954712,
      "step": 68539
    },
    {
      "epoch": 6.536388397216797e-05,
      "step": 68539,
      "training_step_time": 9.577142000198364
    },
    {
      "epoch": 6.536483764648438e-05,
      "grad_norm": 0.11144448071718216,
      "learning_rate": 2.4706076745728647e-06,
      "loss": 0.0561,
      "step": 68540
    },
    {
      "epoch": 6.536483764648438e-05,
      "model_forward_time": 1.7483930587768555,
      "step": 68540
    },
    {
      "dynamics_loss_avg": 0.05567585323005915,
      "epoch": 6.536483764648438e-05,
      "step": 68540
    },
    {
      "action_loss_avg": 0.0015952660294715315,
      "epoch": 6.536483764648438e-05,
      "step": 68540
    },
    {
      "epoch": 6.536483764648438e-05,
      "step": 68540,
      "training_step_time": 7.886991024017334
    },
    {
      "epoch": 6.536579132080078e-05,
      "model_forward_time": 1.7390282154083252,
      "step": 68541
    },
    {
      "epoch": 6.536579132080078e-05,
      "step": 68541,
      "training_step_time": 7.666167974472046
    },
    {
      "epoch": 6.536674499511719e-05,
      "model_forward_time": 1.745880365371704,
      "step": 68542
    },
    {
      "epoch": 6.536674499511719e-05,
      "step": 68542,
      "training_step_time": 6.776897430419922
    },
    {
      "epoch": 6.53676986694336e-05,
      "model_forward_time": 1.7433550357818604,
      "step": 68543
    },
    {
      "epoch": 6.53676986694336e-05,
      "step": 68543,
      "training_step_time": 6.88356614112854
    },
    {
      "epoch": 6.536865234375e-05,
      "model_forward_time": 1.7572731971740723,
      "step": 68544
    },
    {
      "epoch": 6.536865234375e-05,
      "step": 68544,
      "training_step_time": 6.779881477355957
    },
    {
      "epoch": 6.536960601806641e-05,
      "model_forward_time": 1.7556378841400146,
      "step": 68545
    },
    {
      "epoch": 6.536960601806641e-05,
      "step": 68545,
      "training_step_time": 6.795793533325195
    },
    {
      "epoch": 6.537055969238282e-05,
      "model_forward_time": 1.7626934051513672,
      "step": 68546
    },
    {
      "epoch": 6.537055969238282e-05,
      "step": 68546,
      "training_step_time": 7.731022834777832
    },
    {
      "epoch": 6.537151336669922e-05,
      "model_forward_time": 1.7447149753570557,
      "step": 68547
    },
    {
      "epoch": 6.537151336669922e-05,
      "step": 68547,
      "training_step_time": 7.412003517150879
    },
    {
      "epoch": 6.537246704101563e-05,
      "model_forward_time": 2.9301297664642334,
      "step": 68548
    },
    {
      "epoch": 6.537246704101563e-05,
      "step": 68548,
      "training_step_time": 9.8001229763031
    },
    {
      "epoch": 6.537342071533204e-05,
      "model_forward_time": 1.7415573596954346,
      "step": 68549
    },
    {
      "epoch": 6.537342071533204e-05,
      "step": 68549,
      "training_step_time": 8.621537208557129
    },
    {
      "epoch": 6.537437438964843e-05,
      "grad_norm": 0.19369500875473022,
      "learning_rate": 2.469181521962573e-06,
      "loss": 0.0588,
      "step": 68550
    },
    {
      "epoch": 6.537437438964843e-05,
      "model_forward_time": 1.7525660991668701,
      "step": 68550
    },
    {
      "dynamics_loss_avg": 0.05370548870414495,
      "epoch": 6.537437438964843e-05,
      "step": 68550
    },
    {
      "action_loss_avg": 0.0016695255355443806,
      "epoch": 6.537437438964843e-05,
      "step": 68550
    },
    {
      "epoch": 6.537437438964843e-05,
      "step": 68550,
      "training_step_time": 7.664236307144165
    },
    {
      "epoch": 6.537532806396485e-05,
      "model_forward_time": 1.7415359020233154,
      "step": 68551
    },
    {
      "epoch": 6.537532806396485e-05,
      "step": 68551,
      "training_step_time": 7.513501882553101
    },
    {
      "epoch": 6.537628173828126e-05,
      "model_forward_time": 1.73994779586792,
      "step": 68552
    },
    {
      "epoch": 6.537628173828126e-05,
      "step": 68552,
      "training_step_time": 6.867919921875
    },
    {
      "epoch": 6.537723541259765e-05,
      "model_forward_time": 1.7467639446258545,
      "step": 68553
    },
    {
      "epoch": 6.537723541259765e-05,
      "step": 68553,
      "training_step_time": 6.837110996246338
    },
    {
      "epoch": 6.537818908691406e-05,
      "model_forward_time": 1.7411315441131592,
      "step": 68554
    },
    {
      "epoch": 6.537818908691406e-05,
      "step": 68554,
      "training_step_time": 6.846355676651001
    },
    {
      "epoch": 6.537914276123048e-05,
      "model_forward_time": 1.7389111518859863,
      "step": 68555
    },
    {
      "epoch": 6.537914276123048e-05,
      "step": 68555,
      "training_step_time": 7.741629600524902
    },
    {
      "epoch": 6.538009643554687e-05,
      "model_forward_time": 1.8058464527130127,
      "step": 68556
    },
    {
      "epoch": 6.538009643554687e-05,
      "step": 68556,
      "training_step_time": 8.638073682785034
    },
    {
      "epoch": 6.538105010986328e-05,
      "model_forward_time": 2.8927676677703857,
      "step": 68557
    },
    {
      "epoch": 6.538105010986328e-05,
      "step": 68557,
      "training_step_time": 10.120187044143677
    },
    {
      "epoch": 6.53820037841797e-05,
      "model_forward_time": 1.748880386352539,
      "step": 68558
    },
    {
      "epoch": 6.53820037841797e-05,
      "step": 68558,
      "training_step_time": 8.819184064865112
    },
    {
      "epoch": 6.538295745849609e-05,
      "model_forward_time": 1.7432801723480225,
      "step": 68559
    },
    {
      "epoch": 6.538295745849609e-05,
      "step": 68559,
      "training_step_time": 7.723167181015015
    },
    {
      "epoch": 6.53839111328125e-05,
      "grad_norm": 0.09619391709566116,
      "learning_rate": 2.467755646118783e-06,
      "loss": 0.0549,
      "step": 68560
    },
    {
      "epoch": 6.53839111328125e-05,
      "model_forward_time": 1.756289005279541,
      "step": 68560
    },
    {
      "dynamics_loss_avg": 0.060933491215109825,
      "epoch": 6.53839111328125e-05,
      "step": 68560
    },
    {
      "action_loss_avg": 0.0022433788515627386,
      "epoch": 6.53839111328125e-05,
      "step": 68560
    },
    {
      "epoch": 6.53839111328125e-05,
      "step": 68560,
      "training_step_time": 7.681469917297363
    },
    {
      "epoch": 6.538486480712891e-05,
      "model_forward_time": 1.7608582973480225,
      "step": 68561
    },
    {
      "epoch": 6.538486480712891e-05,
      "step": 68561,
      "training_step_time": 6.807308197021484
    },
    {
      "epoch": 6.538581848144531e-05,
      "model_forward_time": 1.7477972507476807,
      "step": 68562
    },
    {
      "epoch": 6.538581848144531e-05,
      "step": 68562,
      "training_step_time": 6.790054798126221
    },
    {
      "epoch": 6.538677215576172e-05,
      "model_forward_time": 1.7394294738769531,
      "step": 68563
    },
    {
      "epoch": 6.538677215576172e-05,
      "step": 68563,
      "training_step_time": 6.7443201541900635
    },
    {
      "epoch": 6.538772583007812e-05,
      "model_forward_time": 1.746403455734253,
      "step": 68564
    },
    {
      "epoch": 6.538772583007812e-05,
      "step": 68564,
      "training_step_time": 7.756686687469482
    },
    {
      "epoch": 6.538867950439453e-05,
      "model_forward_time": 1.8079359531402588,
      "step": 68565
    },
    {
      "epoch": 6.538867950439453e-05,
      "step": 68565,
      "training_step_time": 7.434202671051025
    },
    {
      "epoch": 6.538963317871094e-05,
      "model_forward_time": 2.8930697441101074,
      "step": 68566
    },
    {
      "epoch": 6.538963317871094e-05,
      "step": 68566,
      "training_step_time": 8.197220087051392
    },
    {
      "epoch": 6.539058685302734e-05,
      "model_forward_time": 1.7448859214782715,
      "step": 68567
    },
    {
      "epoch": 6.539058685302734e-05,
      "step": 68567,
      "training_step_time": 8.848164081573486
    },
    {
      "epoch": 6.539154052734375e-05,
      "model_forward_time": 1.7455708980560303,
      "step": 68568
    },
    {
      "epoch": 6.539154052734375e-05,
      "step": 68568,
      "training_step_time": 9.577620267868042
    },
    {
      "epoch": 6.539249420166016e-05,
      "model_forward_time": 1.7469656467437744,
      "step": 68569
    },
    {
      "epoch": 6.539249420166016e-05,
      "step": 68569,
      "training_step_time": 7.646569728851318
    },
    {
      "epoch": 6.539344787597656e-05,
      "grad_norm": 0.1196051687002182,
      "learning_rate": 2.46633004719743e-06,
      "loss": 0.0567,
      "step": 68570
    },
    {
      "epoch": 6.539344787597656e-05,
      "model_forward_time": 1.7603604793548584,
      "step": 68570
    },
    {
      "dynamics_loss_avg": 0.059238641150295736,
      "epoch": 6.539344787597656e-05,
      "step": 68570
    },
    {
      "action_loss_avg": 0.0014587434416171163,
      "epoch": 6.539344787597656e-05,
      "step": 68570
    },
    {
      "epoch": 6.539344787597656e-05,
      "step": 68570,
      "training_step_time": 6.800965785980225
    },
    {
      "epoch": 6.539440155029297e-05,
      "model_forward_time": 1.746758222579956,
      "step": 68571
    },
    {
      "epoch": 6.539440155029297e-05,
      "step": 68571,
      "training_step_time": 6.812091827392578
    },
    {
      "epoch": 6.539535522460938e-05,
      "model_forward_time": 1.7448561191558838,
      "step": 68572
    },
    {
      "epoch": 6.539535522460938e-05,
      "step": 68572,
      "training_step_time": 6.806816339492798
    },
    {
      "epoch": 6.539630889892577e-05,
      "model_forward_time": 1.7547392845153809,
      "step": 68573
    },
    {
      "epoch": 6.539630889892577e-05,
      "step": 68573,
      "training_step_time": 7.691203594207764
    },
    {
      "epoch": 6.539726257324219e-05,
      "model_forward_time": 1.8076269626617432,
      "step": 68574
    },
    {
      "epoch": 6.539726257324219e-05,
      "step": 68574,
      "training_step_time": 7.430349826812744
    },
    {
      "epoch": 6.53982162475586e-05,
      "model_forward_time": 2.8902406692504883,
      "step": 68575
    },
    {
      "epoch": 6.53982162475586e-05,
      "step": 68575,
      "training_step_time": 8.43354868888855
    },
    {
      "epoch": 6.5399169921875e-05,
      "model_forward_time": 1.744636058807373,
      "step": 68576
    },
    {
      "epoch": 6.5399169921875e-05,
      "step": 68576,
      "training_step_time": 8.886773586273193
    },
    {
      "epoch": 6.54001235961914e-05,
      "model_forward_time": 1.744758129119873,
      "step": 68577
    },
    {
      "epoch": 6.54001235961914e-05,
      "step": 68577,
      "training_step_time": 7.763123273849487
    },
    {
      "epoch": 6.540107727050782e-05,
      "model_forward_time": 1.7364087104797363,
      "step": 68578
    },
    {
      "epoch": 6.540107727050782e-05,
      "step": 68578,
      "training_step_time": 7.653423309326172
    },
    {
      "epoch": 6.540203094482421e-05,
      "model_forward_time": 1.739457607269287,
      "step": 68579
    },
    {
      "epoch": 6.540203094482421e-05,
      "step": 68579,
      "training_step_time": 6.811303377151489
    },
    {
      "epoch": 6.540298461914062e-05,
      "grad_norm": 0.07919692993164062,
      "learning_rate": 2.464904725354412e-06,
      "loss": 0.0581,
      "step": 68580
    },
    {
      "epoch": 6.540298461914062e-05,
      "model_forward_time": 1.750068187713623,
      "step": 68580
    },
    {
      "dynamics_loss_avg": 0.07791999094188214,
      "epoch": 6.540298461914062e-05,
      "step": 68580
    },
    {
      "action_loss_avg": 0.0012650657183257863,
      "epoch": 6.540298461914062e-05,
      "step": 68580
    },
    {
      "epoch": 6.540298461914062e-05,
      "step": 68580,
      "training_step_time": 6.797924280166626
    },
    {
      "epoch": 6.540393829345703e-05,
      "model_forward_time": 1.7295536994934082,
      "step": 68581
    },
    {
      "epoch": 6.540393829345703e-05,
      "step": 68581,
      "training_step_time": 6.772868394851685
    },
    {
      "epoch": 6.540489196777343e-05,
      "model_forward_time": 1.7402684688568115,
      "step": 68582
    },
    {
      "epoch": 6.540489196777343e-05,
      "step": 68582,
      "training_step_time": 7.7679665088653564
    },
    {
      "epoch": 6.540584564208984e-05,
      "model_forward_time": 1.8120677471160889,
      "step": 68583
    },
    {
      "epoch": 6.540584564208984e-05,
      "step": 68583,
      "training_step_time": 7.390103578567505
    },
    {
      "epoch": 6.540679931640625e-05,
      "model_forward_time": 2.8760087490081787,
      "step": 68584
    },
    {
      "epoch": 6.540679931640625e-05,
      "step": 68584,
      "training_step_time": 8.219623327255249
    },
    {
      "epoch": 6.540775299072265e-05,
      "model_forward_time": 1.740842342376709,
      "step": 68585
    },
    {
      "epoch": 6.540775299072265e-05,
      "step": 68585,
      "training_step_time": 9.635416746139526
    },
    {
      "epoch": 6.540870666503906e-05,
      "model_forward_time": 1.7419133186340332,
      "step": 68586
    },
    {
      "epoch": 6.540870666503906e-05,
      "step": 68586,
      "training_step_time": 9.330072402954102
    },
    {
      "epoch": 6.540966033935547e-05,
      "model_forward_time": 1.7405853271484375,
      "step": 68587
    },
    {
      "epoch": 6.540966033935547e-05,
      "step": 68587,
      "training_step_time": 8.360706806182861
    },
    {
      "epoch": 6.541061401367187e-05,
      "model_forward_time": 1.7372558116912842,
      "step": 68588
    },
    {
      "epoch": 6.541061401367187e-05,
      "step": 68588,
      "training_step_time": 7.5261070728302
    },
    {
      "epoch": 6.541156768798828e-05,
      "model_forward_time": 1.7456104755401611,
      "step": 68589
    },
    {
      "epoch": 6.541156768798828e-05,
      "step": 68589,
      "training_step_time": 6.796288013458252
    },
    {
      "epoch": 6.541252136230469e-05,
      "grad_norm": 0.09523237496614456,
      "learning_rate": 2.463479680745602e-06,
      "loss": 0.0587,
      "step": 68590
    },
    {
      "epoch": 6.541252136230469e-05,
      "model_forward_time": 1.7483744621276855,
      "step": 68590
    },
    {
      "dynamics_loss_avg": 0.05882561821490526,
      "epoch": 6.541252136230469e-05,
      "step": 68590
    },
    {
      "action_loss_avg": 0.0016677056148182601,
      "epoch": 6.541252136230469e-05,
      "step": 68590
    },
    {
      "epoch": 6.541252136230469e-05,
      "step": 68590,
      "training_step_time": 6.8302836418151855
    },
    {
      "epoch": 6.541347503662109e-05,
      "model_forward_time": 1.7359209060668945,
      "step": 68591
    },
    {
      "epoch": 6.541347503662109e-05,
      "step": 68591,
      "training_step_time": 7.818379163742065
    },
    {
      "epoch": 6.54144287109375e-05,
      "model_forward_time": 1.804823637008667,
      "step": 68592
    },
    {
      "epoch": 6.54144287109375e-05,
      "step": 68592,
      "training_step_time": 7.482420206069946
    },
    {
      "epoch": 6.541538238525391e-05,
      "model_forward_time": 1.742518663406372,
      "step": 68593
    },
    {
      "epoch": 6.541538238525391e-05,
      "step": 68593,
      "training_step_time": 8.323879480361938
    },
    {
      "epoch": 6.541633605957031e-05,
      "model_forward_time": 2.909987449645996,
      "step": 68594
    },
    {
      "epoch": 6.541633605957031e-05,
      "step": 68594,
      "training_step_time": 9.408071279525757
    },
    {
      "epoch": 6.541728973388672e-05,
      "model_forward_time": 1.7375917434692383,
      "step": 68595
    },
    {
      "epoch": 6.541728973388672e-05,
      "step": 68595,
      "training_step_time": 8.730653524398804
    },
    {
      "epoch": 6.541824340820313e-05,
      "model_forward_time": 1.7367942333221436,
      "step": 68596
    },
    {
      "epoch": 6.541824340820313e-05,
      "step": 68596,
      "training_step_time": 8.2049880027771
    },
    {
      "epoch": 6.541919708251953e-05,
      "model_forward_time": 1.7538268566131592,
      "step": 68597
    },
    {
      "epoch": 6.541919708251953e-05,
      "step": 68597,
      "training_step_time": 7.630590200424194
    },
    {
      "epoch": 6.542015075683594e-05,
      "model_forward_time": 1.7562243938446045,
      "step": 68598
    },
    {
      "epoch": 6.542015075683594e-05,
      "step": 68598,
      "training_step_time": 6.766124725341797
    },
    {
      "epoch": 6.542110443115235e-05,
      "model_forward_time": 1.751661777496338,
      "step": 68599
    },
    {
      "epoch": 6.542110443115235e-05,
      "step": 68599,
      "training_step_time": 6.743314743041992
    },
    {
      "epoch": 6.542205810546874e-05,
      "grad_norm": 0.12503772974014282,
      "learning_rate": 2.46205491352684e-06,
      "loss": 0.0576,
      "step": 68600
    },
    {
      "epoch": 6.542205810546874e-05,
      "model_forward_time": 1.7594895362854004,
      "step": 68600
    },
    {
      "dynamics_loss_avg": 0.05633536074310541,
      "epoch": 6.542205810546874e-05,
      "step": 68600
    },
    {
      "action_loss_avg": 0.0011804801994003355,
      "epoch": 6.542205810546874e-05,
      "step": 68600
    },
    {
      "epoch": 6.542205810546874e-05,
      "step": 68600,
      "training_step_time": 7.704534530639648
    },
    {
      "epoch": 6.542301177978516e-05,
      "model_forward_time": 1.7879390716552734,
      "step": 68601
    },
    {
      "epoch": 6.542301177978516e-05,
      "step": 68601,
      "training_step_time": 7.393224477767944
    },
    {
      "epoch": 6.542396545410157e-05,
      "model_forward_time": 1.752955436706543,
      "step": 68602
    },
    {
      "epoch": 6.542396545410157e-05,
      "step": 68602,
      "training_step_time": 7.949466228485107
    },
    {
      "epoch": 6.542491912841796e-05,
      "model_forward_time": 2.960925817489624,
      "step": 68603
    },
    {
      "epoch": 6.542491912841796e-05,
      "step": 68603,
      "training_step_time": 8.782734870910645
    },
    {
      "epoch": 6.542587280273437e-05,
      "model_forward_time": 1.7538378238677979,
      "step": 68604
    },
    {
      "epoch": 6.542587280273437e-05,
      "step": 68604,
      "training_step_time": 8.73718547821045
    },
    {
      "epoch": 6.542682647705079e-05,
      "model_forward_time": 1.754289150238037,
      "step": 68605
    },
    {
      "epoch": 6.542682647705079e-05,
      "step": 68605,
      "training_step_time": 8.24453353881836
    },
    {
      "epoch": 6.542778015136718e-05,
      "model_forward_time": 1.7616634368896484,
      "step": 68606
    },
    {
      "epoch": 6.542778015136718e-05,
      "step": 68606,
      "training_step_time": 8.653873682022095
    },
    {
      "epoch": 6.542873382568359e-05,
      "model_forward_time": 1.7609236240386963,
      "step": 68607
    },
    {
      "epoch": 6.542873382568359e-05,
      "step": 68607,
      "training_step_time": 6.784613370895386
    },
    {
      "epoch": 6.54296875e-05,
      "model_forward_time": 1.7554495334625244,
      "step": 68608
    },
    {
      "epoch": 6.54296875e-05,
      "step": 68608,
      "training_step_time": 6.793644428253174
    },
    {
      "epoch": 6.54306411743164e-05,
      "model_forward_time": 1.7501308917999268,
      "step": 68609
    },
    {
      "epoch": 6.54306411743164e-05,
      "step": 68609,
      "training_step_time": 7.728506565093994
    },
    {
      "epoch": 6.543159484863281e-05,
      "grad_norm": 0.10362230241298676,
      "learning_rate": 2.4606304238539377e-06,
      "loss": 0.0549,
      "step": 68610
    },
    {
      "epoch": 6.543159484863281e-05,
      "model_forward_time": 1.7715353965759277,
      "step": 68610
    },
    {
      "dynamics_loss_avg": 0.04911968894302845,
      "epoch": 6.543159484863281e-05,
      "step": 68610
    },
    {
      "action_loss_avg": 0.0017593696422409267,
      "epoch": 6.543159484863281e-05,
      "step": 68610
    },
    {
      "epoch": 6.543159484863281e-05,
      "step": 68610,
      "training_step_time": 7.480137348175049
    },
    {
      "epoch": 6.543254852294922e-05,
      "model_forward_time": 1.758573293685913,
      "step": 68611
    },
    {
      "epoch": 6.543254852294922e-05,
      "step": 68611,
      "training_step_time": 8.114013671875
    },
    {
      "epoch": 6.543350219726562e-05,
      "model_forward_time": 2.9260811805725098,
      "step": 68612
    },
    {
      "epoch": 6.543350219726562e-05,
      "step": 68612,
      "training_step_time": 8.779600381851196
    },
    {
      "epoch": 6.543445587158203e-05,
      "model_forward_time": 1.7596299648284912,
      "step": 68613
    },
    {
      "epoch": 6.543445587158203e-05,
      "step": 68613,
      "training_step_time": 8.49068284034729
    },
    {
      "epoch": 6.543540954589844e-05,
      "model_forward_time": 1.75266432762146,
      "step": 68614
    },
    {
      "epoch": 6.543540954589844e-05,
      "step": 68614,
      "training_step_time": 7.710489749908447
    },
    {
      "epoch": 6.543636322021484e-05,
      "model_forward_time": 1.758559226989746,
      "step": 68615
    },
    {
      "epoch": 6.543636322021484e-05,
      "step": 68615,
      "training_step_time": 7.740239143371582
    },
    {
      "epoch": 6.543731689453125e-05,
      "model_forward_time": 1.7584257125854492,
      "step": 68616
    },
    {
      "epoch": 6.543731689453125e-05,
      "step": 68616,
      "training_step_time": 7.562520503997803
    },
    {
      "epoch": 6.543827056884766e-05,
      "model_forward_time": 1.7456047534942627,
      "step": 68617
    },
    {
      "epoch": 6.543827056884766e-05,
      "step": 68617,
      "training_step_time": 6.885998249053955
    },
    {
      "epoch": 6.543922424316406e-05,
      "model_forward_time": 1.7506120204925537,
      "step": 68618
    },
    {
      "epoch": 6.543922424316406e-05,
      "step": 68618,
      "training_step_time": 7.824949264526367
    },
    {
      "epoch": 6.544017791748047e-05,
      "model_forward_time": 1.7803168296813965,
      "step": 68619
    },
    {
      "epoch": 6.544017791748047e-05,
      "step": 68619,
      "training_step_time": 7.533960819244385
    },
    {
      "epoch": 6.544113159179688e-05,
      "grad_norm": 0.09735153615474701,
      "learning_rate": 2.4592062118826755e-06,
      "loss": 0.0568,
      "step": 68620
    },
    {
      "epoch": 6.544113159179688e-05,
      "model_forward_time": 1.8409054279327393,
      "step": 68620
    },
    {
      "dynamics_loss_avg": 0.052696249447762966,
      "epoch": 6.544113159179688e-05,
      "step": 68620
    },
    {
      "action_loss_avg": 0.0013174027728382498,
      "epoch": 6.544113159179688e-05,
      "step": 68620
    },
    {
      "epoch": 6.544113159179688e-05,
      "step": 68620,
      "training_step_time": 7.5603179931640625
    },
    {
      "epoch": 6.544208526611328e-05,
      "model_forward_time": 2.8943445682525635,
      "step": 68621
    },
    {
      "epoch": 6.544208526611328e-05,
      "step": 68621,
      "training_step_time": 8.39720368385315
    },
    {
      "epoch": 6.544303894042969e-05,
      "model_forward_time": 1.7530508041381836,
      "step": 68622
    },
    {
      "epoch": 6.544303894042969e-05,
      "step": 68622,
      "training_step_time": 10.891812562942505
    },
    {
      "epoch": 6.54439926147461e-05,
      "model_forward_time": 1.75565767288208,
      "step": 68623
    },
    {
      "epoch": 6.54439926147461e-05,
      "step": 68623,
      "training_step_time": 8.705299377441406
    },
    {
      "epoch": 6.54449462890625e-05,
      "model_forward_time": 1.764592170715332,
      "step": 68624
    },
    {
      "epoch": 6.54449462890625e-05,
      "step": 68624,
      "training_step_time": 7.791256666183472
    },
    {
      "epoch": 6.54458999633789e-05,
      "model_forward_time": 1.748809814453125,
      "step": 68625
    },
    {
      "epoch": 6.54458999633789e-05,
      "step": 68625,
      "training_step_time": 7.551689863204956
    },
    {
      "epoch": 6.544685363769532e-05,
      "model_forward_time": 1.755558729171753,
      "step": 68626
    },
    {
      "epoch": 6.544685363769532e-05,
      "step": 68626,
      "training_step_time": 6.796926736831665
    },
    {
      "epoch": 6.544780731201171e-05,
      "model_forward_time": 1.7536509037017822,
      "step": 68627
    },
    {
      "epoch": 6.544780731201171e-05,
      "step": 68627,
      "training_step_time": 7.808468818664551
    },
    {
      "epoch": 6.544876098632813e-05,
      "model_forward_time": 1.7566184997558594,
      "step": 68628
    },
    {
      "epoch": 6.544876098632813e-05,
      "step": 68628,
      "training_step_time": 7.4839088916778564
    },
    {
      "epoch": 6.544971466064454e-05,
      "model_forward_time": 1.8224892616271973,
      "step": 68629
    },
    {
      "epoch": 6.544971466064454e-05,
      "step": 68629,
      "training_step_time": 7.555914878845215
    },
    {
      "epoch": 6.545066833496093e-05,
      "grad_norm": 0.07286198437213898,
      "learning_rate": 2.4577822777688005e-06,
      "loss": 0.0563,
      "step": 68630
    },
    {
      "epoch": 6.545066833496093e-05,
      "model_forward_time": 2.900705337524414,
      "step": 68630
    },
    {
      "dynamics_loss_avg": 0.06634475886821747,
      "epoch": 6.545066833496093e-05,
      "step": 68630
    },
    {
      "action_loss_avg": 0.0043103904987219725,
      "epoch": 6.545066833496093e-05,
      "step": 68630
    },
    {
      "epoch": 6.545066833496093e-05,
      "step": 68630,
      "training_step_time": 7.7468297481536865
    },
    {
      "epoch": 6.545162200927734e-05,
      "model_forward_time": 1.3915002346038818,
      "step": 68631
    },
    {
      "epoch": 6.545162200927734e-05,
      "step": 68631,
      "training_step_time": 9.446235179901123
    },
    {
      "epoch": 6.545257568359375e-05,
      "model_forward_time": 1.7554426193237305,
      "step": 68632
    },
    {
      "epoch": 6.545257568359375e-05,
      "step": 68632,
      "training_step_time": 7.938461065292358
    },
    {
      "epoch": 6.545352935791015e-05,
      "model_forward_time": 1.7571592330932617,
      "step": 68633
    },
    {
      "epoch": 6.545352935791015e-05,
      "step": 68633,
      "training_step_time": 8.759652137756348
    },
    {
      "epoch": 6.545448303222656e-05,
      "model_forward_time": 1.7546305656433105,
      "step": 68634
    },
    {
      "epoch": 6.545448303222656e-05,
      "step": 68634,
      "training_step_time": 7.676794052124023
    },
    {
      "epoch": 6.545543670654297e-05,
      "model_forward_time": 1.7286992073059082,
      "step": 68635
    },
    {
      "epoch": 6.545543670654297e-05,
      "step": 68635,
      "training_step_time": 7.08223032951355
    },
    {
      "epoch": 6.545639038085937e-05,
      "model_forward_time": 1.7441327571868896,
      "step": 68636
    },
    {
      "epoch": 6.545639038085937e-05,
      "step": 68636,
      "training_step_time": 7.7250282764434814
    },
    {
      "epoch": 6.545734405517578e-05,
      "model_forward_time": 1.7572736740112305,
      "step": 68637
    },
    {
      "epoch": 6.545734405517578e-05,
      "step": 68637,
      "training_step_time": 8.194379329681396
    },
    {
      "epoch": 6.545829772949219e-05,
      "model_forward_time": 1.7917110919952393,
      "step": 68638
    },
    {
      "epoch": 6.545829772949219e-05,
      "step": 68638,
      "training_step_time": 7.508247375488281
    },
    {
      "epoch": 6.545925140380859e-05,
      "model_forward_time": 1.7429871559143066,
      "step": 68639
    },
    {
      "epoch": 6.545925140380859e-05,
      "step": 68639,
      "training_step_time": 8.040168285369873
    },
    {
      "epoch": 6.5460205078125e-05,
      "grad_norm": 0.11369588971138,
      "learning_rate": 2.4563586216680346e-06,
      "loss": 0.0539,
      "step": 68640
    },
    {
      "epoch": 6.5460205078125e-05,
      "model_forward_time": 2.924220085144043,
      "step": 68640
    },
    {
      "dynamics_loss_avg": 0.04637712575495243,
      "epoch": 6.5460205078125e-05,
      "step": 68640
    },
    {
      "action_loss_avg": 0.001866105874069035,
      "epoch": 6.5460205078125e-05,
      "step": 68640
    },
    {
      "epoch": 6.5460205078125e-05,
      "step": 68640,
      "training_step_time": 9.363800048828125
    },
    {
      "epoch": 6.546115875244141e-05,
      "model_forward_time": 1.742276906967163,
      "step": 68641
    },
    {
      "epoch": 6.546115875244141e-05,
      "step": 68641,
      "training_step_time": 8.626604557037354
    },
    {
      "epoch": 6.546211242675781e-05,
      "model_forward_time": 1.7453088760375977,
      "step": 68642
    },
    {
      "epoch": 6.546211242675781e-05,
      "step": 68642,
      "training_step_time": 7.656056642532349
    },
    {
      "epoch": 6.546306610107422e-05,
      "model_forward_time": 1.7636322975158691,
      "step": 68643
    },
    {
      "epoch": 6.546306610107422e-05,
      "step": 68643,
      "training_step_time": 7.749198913574219
    },
    {
      "epoch": 6.546401977539063e-05,
      "model_forward_time": 1.74379563331604,
      "step": 68644
    },
    {
      "epoch": 6.546401977539063e-05,
      "step": 68644,
      "training_step_time": 7.6257102489471436
    },
    {
      "epoch": 6.546497344970703e-05,
      "model_forward_time": 1.7482075691223145,
      "step": 68645
    },
    {
      "epoch": 6.546497344970703e-05,
      "step": 68645,
      "training_step_time": 7.620753049850464
    },
    {
      "epoch": 6.546592712402344e-05,
      "model_forward_time": 1.749873399734497,
      "step": 68646
    },
    {
      "epoch": 6.546592712402344e-05,
      "step": 68646,
      "training_step_time": 6.764096736907959
    },
    {
      "epoch": 6.546688079833985e-05,
      "model_forward_time": 1.7469959259033203,
      "step": 68647
    },
    {
      "epoch": 6.546688079833985e-05,
      "step": 68647,
      "training_step_time": 7.9499335289001465
    },
    {
      "epoch": 6.546783447265625e-05,
      "model_forward_time": 1.751133918762207,
      "step": 68648
    },
    {
      "epoch": 6.546783447265625e-05,
      "step": 68648,
      "training_step_time": 8.049323081970215
    },
    {
      "epoch": 6.546878814697266e-05,
      "model_forward_time": 2.893554925918579,
      "step": 68649
    },
    {
      "epoch": 6.546878814697266e-05,
      "step": 68649,
      "training_step_time": 8.726969957351685
    },
    {
      "epoch": 6.546974182128907e-05,
      "grad_norm": 0.08851579576730728,
      "learning_rate": 2.4549352437360624e-06,
      "loss": 0.0561,
      "step": 68650
    },
    {
      "epoch": 6.546974182128907e-05,
      "model_forward_time": 1.7258355617523193,
      "step": 68650
    },
    {
      "dynamics_loss_avg": 0.05344330444931984,
      "epoch": 6.546974182128907e-05,
      "step": 68650
    },
    {
      "action_loss_avg": 0.0009373171022161841,
      "epoch": 6.546974182128907e-05,
      "step": 68650
    },
    {
      "epoch": 6.546974182128907e-05,
      "step": 68650,
      "training_step_time": 7.836899042129517
    },
    {
      "epoch": 6.547069549560547e-05,
      "model_forward_time": 1.7399218082427979,
      "step": 68651
    },
    {
      "epoch": 6.547069549560547e-05,
      "step": 68651,
      "training_step_time": 7.6219305992126465
    },
    {
      "epoch": 6.547164916992188e-05,
      "model_forward_time": 1.7387089729309082,
      "step": 68652
    },
    {
      "epoch": 6.547164916992188e-05,
      "step": 68652,
      "training_step_time": 8.205121278762817
    },
    {
      "epoch": 6.547260284423829e-05,
      "model_forward_time": 1.7464821338653564,
      "step": 68653
    },
    {
      "epoch": 6.547260284423829e-05,
      "step": 68653,
      "training_step_time": 7.550414562225342
    },
    {
      "epoch": 6.547355651855468e-05,
      "model_forward_time": 1.7389123439788818,
      "step": 68654
    },
    {
      "epoch": 6.547355651855468e-05,
      "step": 68654,
      "training_step_time": 6.749033212661743
    },
    {
      "epoch": 6.54745101928711e-05,
      "model_forward_time": 1.7371528148651123,
      "step": 68655
    },
    {
      "epoch": 6.54745101928711e-05,
      "step": 68655,
      "training_step_time": 7.803850889205933
    },
    {
      "epoch": 6.54754638671875e-05,
      "model_forward_time": 1.7356500625610352,
      "step": 68656
    },
    {
      "epoch": 6.54754638671875e-05,
      "step": 68656,
      "training_step_time": 7.502193927764893
    },
    {
      "epoch": 6.54764175415039e-05,
      "model_forward_time": 1.8043713569641113,
      "step": 68657
    },
    {
      "epoch": 6.54764175415039e-05,
      "step": 68657,
      "training_step_time": 8.457045793533325
    },
    {
      "epoch": 6.547737121582031e-05,
      "model_forward_time": 2.8885915279388428,
      "step": 68658
    },
    {
      "epoch": 6.547737121582031e-05,
      "step": 68658,
      "training_step_time": 8.494449138641357
    },
    {
      "epoch": 6.547832489013672e-05,
      "model_forward_time": 1.3941068649291992,
      "step": 68659
    },
    {
      "epoch": 6.547832489013672e-05,
      "step": 68659,
      "training_step_time": 8.892308712005615
    },
    {
      "epoch": 6.547927856445312e-05,
      "grad_norm": 0.07320345193147659,
      "learning_rate": 2.453512144128549e-06,
      "loss": 0.0593,
      "step": 68660
    },
    {
      "epoch": 6.547927856445312e-05,
      "model_forward_time": 1.7477946281433105,
      "step": 68660
    },
    {
      "dynamics_loss_avg": 0.05837544370442629,
      "epoch": 6.547927856445312e-05,
      "step": 68660
    },
    {
      "action_loss_avg": 0.0008308255288284272,
      "epoch": 6.547927856445312e-05,
      "step": 68660
    },
    {
      "epoch": 6.547927856445312e-05,
      "step": 68660,
      "training_step_time": 9.550942420959473
    },
    {
      "epoch": 6.548023223876953e-05,
      "model_forward_time": 1.7351820468902588,
      "step": 68661
    },
    {
      "epoch": 6.548023223876953e-05,
      "step": 68661,
      "training_step_time": 7.68737268447876
    },
    {
      "epoch": 6.548118591308594e-05,
      "model_forward_time": 1.7367608547210693,
      "step": 68662
    },
    {
      "epoch": 6.548118591308594e-05,
      "step": 68662,
      "training_step_time": 7.557805299758911
    },
    {
      "epoch": 6.548213958740234e-05,
      "model_forward_time": 1.384049892425537,
      "step": 68663
    },
    {
      "epoch": 6.548213958740234e-05,
      "step": 68663,
      "training_step_time": 6.985940217971802
    },
    {
      "epoch": 6.548309326171875e-05,
      "model_forward_time": 1.7415783405303955,
      "step": 68664
    },
    {
      "epoch": 6.548309326171875e-05,
      "step": 68664,
      "training_step_time": 7.726084232330322
    },
    {
      "epoch": 6.548404693603516e-05,
      "model_forward_time": 1.739511489868164,
      "step": 68665
    },
    {
      "epoch": 6.548404693603516e-05,
      "step": 68665,
      "training_step_time": 7.457710266113281
    },
    {
      "epoch": 6.548500061035156e-05,
      "model_forward_time": 1.8031818866729736,
      "step": 68666
    },
    {
      "epoch": 6.548500061035156e-05,
      "step": 68666,
      "training_step_time": 7.508795499801636
    },
    {
      "epoch": 6.548595428466797e-05,
      "model_forward_time": 2.816906690597534,
      "step": 68667
    },
    {
      "epoch": 6.548595428466797e-05,
      "step": 68667,
      "training_step_time": 8.178556680679321
    },
    {
      "epoch": 6.548690795898438e-05,
      "model_forward_time": 1.7376649379730225,
      "step": 68668
    },
    {
      "epoch": 6.548690795898438e-05,
      "step": 68668,
      "training_step_time": 8.799962759017944
    },
    {
      "epoch": 6.548786163330078e-05,
      "model_forward_time": 1.7525310516357422,
      "step": 68669
    },
    {
      "epoch": 6.548786163330078e-05,
      "step": 68669,
      "training_step_time": 7.894866943359375
    },
    {
      "epoch": 6.548881530761719e-05,
      "grad_norm": 0.07094945013523102,
      "learning_rate": 2.4520893230011174e-06,
      "loss": 0.0532,
      "step": 68670
    },
    {
      "epoch": 6.548881530761719e-05,
      "model_forward_time": 1.7628881931304932,
      "step": 68670
    },
    {
      "dynamics_loss_avg": 0.05429094024002552,
      "epoch": 6.548881530761719e-05,
      "step": 68670
    },
    {
      "action_loss_avg": 0.0018196948250988499,
      "epoch": 6.548881530761719e-05,
      "step": 68670
    },
    {
      "epoch": 6.548881530761719e-05,
      "step": 68670,
      "training_step_time": 7.66637659072876
    },
    {
      "epoch": 6.54897689819336e-05,
      "model_forward_time": 1.7447116374969482,
      "step": 68671
    },
    {
      "epoch": 6.54897689819336e-05,
      "step": 68671,
      "training_step_time": 7.327266216278076
    },
    {
      "epoch": 6.549072265625e-05,
      "model_forward_time": 1.7517852783203125,
      "step": 68672
    },
    {
      "epoch": 6.549072265625e-05,
      "step": 68672,
      "training_step_time": 7.557382822036743
    },
    {
      "epoch": 6.549167633056641e-05,
      "model_forward_time": 1.751664638519287,
      "step": 68673
    },
    {
      "epoch": 6.549167633056641e-05,
      "step": 68673,
      "training_step_time": 7.71802830696106
    },
    {
      "epoch": 6.549263000488282e-05,
      "model_forward_time": 1.7551240921020508,
      "step": 68674
    },
    {
      "epoch": 6.549263000488282e-05,
      "step": 68674,
      "training_step_time": 7.4586334228515625
    },
    {
      "epoch": 6.549358367919922e-05,
      "model_forward_time": 1.7969708442687988,
      "step": 68675
    },
    {
      "epoch": 6.549358367919922e-05,
      "step": 68675,
      "training_step_time": 8.39937138557434
    },
    {
      "epoch": 6.549453735351563e-05,
      "model_forward_time": 1.7523984909057617,
      "step": 68676
    },
    {
      "epoch": 6.549453735351563e-05,
      "step": 68676,
      "training_step_time": 10.318402290344238
    },
    {
      "epoch": 6.549549102783204e-05,
      "model_forward_time": 2.918579578399658,
      "step": 68677
    },
    {
      "epoch": 6.549549102783204e-05,
      "step": 68677,
      "training_step_time": 8.50147032737732
    },
    {
      "epoch": 6.549644470214843e-05,
      "model_forward_time": 1.7481303215026855,
      "step": 68678
    },
    {
      "epoch": 6.549644470214843e-05,
      "step": 68678,
      "training_step_time": 7.839705944061279
    },
    {
      "epoch": 6.549739837646485e-05,
      "model_forward_time": 1.7670209407806396,
      "step": 68679
    },
    {
      "epoch": 6.549739837646485e-05,
      "step": 68679,
      "training_step_time": 7.677253246307373
    },
    {
      "epoch": 6.549835205078126e-05,
      "grad_norm": 0.10278613120317459,
      "learning_rate": 2.4506667805093682e-06,
      "loss": 0.0517,
      "step": 68680
    },
    {
      "epoch": 6.549835205078126e-05,
      "model_forward_time": 1.756460428237915,
      "step": 68680
    },
    {
      "dynamics_loss_avg": 0.054533619433641434,
      "epoch": 6.549835205078126e-05,
      "step": 68680
    },
    {
      "action_loss_avg": 0.0022637195012066512,
      "epoch": 6.549835205078126e-05,
      "step": 68680
    },
    {
      "epoch": 6.549835205078126e-05,
      "step": 68680,
      "training_step_time": 7.715528249740601
    },
    {
      "epoch": 6.549930572509765e-05,
      "model_forward_time": 1.7513489723205566,
      "step": 68681
    },
    {
      "epoch": 6.549930572509765e-05,
      "step": 68681,
      "training_step_time": 7.633753061294556
    },
    {
      "epoch": 6.550025939941406e-05,
      "model_forward_time": 1.7463634014129639,
      "step": 68682
    },
    {
      "epoch": 6.550025939941406e-05,
      "step": 68682,
      "training_step_time": 7.749021053314209
    },
    {
      "epoch": 6.550121307373048e-05,
      "model_forward_time": 1.752737283706665,
      "step": 68683
    },
    {
      "epoch": 6.550121307373048e-05,
      "step": 68683,
      "training_step_time": 7.3985435962677
    },
    {
      "epoch": 6.550216674804687e-05,
      "model_forward_time": 1.7554850578308105,
      "step": 68684
    },
    {
      "epoch": 6.550216674804687e-05,
      "step": 68684,
      "training_step_time": 7.578565835952759
    },
    {
      "epoch": 6.550312042236328e-05,
      "model_forward_time": 1.8215909004211426,
      "step": 68685
    },
    {
      "epoch": 6.550312042236328e-05,
      "step": 68685,
      "training_step_time": 7.69210958480835
    },
    {
      "epoch": 6.55040740966797e-05,
      "model_forward_time": 2.9290220737457275,
      "step": 68686
    },
    {
      "epoch": 6.55040740966797e-05,
      "step": 68686,
      "training_step_time": 9.537420272827148
    },
    {
      "epoch": 6.550502777099609e-05,
      "model_forward_time": 1.7516546249389648,
      "step": 68687
    },
    {
      "epoch": 6.550502777099609e-05,
      "step": 68687,
      "training_step_time": 9.584566831588745
    },
    {
      "epoch": 6.55059814453125e-05,
      "model_forward_time": 1.749985933303833,
      "step": 68688
    },
    {
      "epoch": 6.55059814453125e-05,
      "step": 68688,
      "training_step_time": 8.526522397994995
    },
    {
      "epoch": 6.55069351196289e-05,
      "model_forward_time": 1.7485933303833008,
      "step": 68689
    },
    {
      "epoch": 6.55069351196289e-05,
      "step": 68689,
      "training_step_time": 7.637512683868408
    },
    {
      "epoch": 6.550788879394531e-05,
      "grad_norm": 0.08750887960195541,
      "learning_rate": 2.449244516808866e-06,
      "loss": 0.0536,
      "step": 68690
    },
    {
      "epoch": 6.550788879394531e-05,
      "model_forward_time": 1.7521142959594727,
      "step": 68690
    },
    {
      "dynamics_loss_avg": 0.06136759389191866,
      "epoch": 6.550788879394531e-05,
      "step": 68690
    },
    {
      "action_loss_avg": 0.002115498497732915,
      "epoch": 6.550788879394531e-05,
      "step": 68690
    },
    {
      "epoch": 6.550788879394531e-05,
      "step": 68690,
      "training_step_time": 7.337141752243042
    },
    {
      "epoch": 6.550884246826172e-05,
      "model_forward_time": 1.753328561782837,
      "step": 68691
    },
    {
      "epoch": 6.550884246826172e-05,
      "step": 68691,
      "training_step_time": 7.6884729862213135
    },
    {
      "epoch": 6.550979614257812e-05,
      "model_forward_time": 1.7533578872680664,
      "step": 68692
    },
    {
      "epoch": 6.550979614257812e-05,
      "step": 68692,
      "training_step_time": 7.384586572647095
    },
    {
      "epoch": 6.551074981689453e-05,
      "model_forward_time": 1.770195722579956,
      "step": 68693
    },
    {
      "epoch": 6.551074981689453e-05,
      "step": 68693,
      "training_step_time": 7.596276044845581
    },
    {
      "epoch": 6.551170349121094e-05,
      "model_forward_time": 1.8263275623321533,
      "step": 68694
    },
    {
      "epoch": 6.551170349121094e-05,
      "step": 68694,
      "training_step_time": 7.6004228591918945
    },
    {
      "epoch": 6.551265716552734e-05,
      "model_forward_time": 2.8848352432250977,
      "step": 68695
    },
    {
      "epoch": 6.551265716552734e-05,
      "step": 68695,
      "training_step_time": 9.60979676246643
    },
    {
      "epoch": 6.551361083984375e-05,
      "model_forward_time": 1.7554378509521484,
      "step": 68696
    },
    {
      "epoch": 6.551361083984375e-05,
      "step": 68696,
      "training_step_time": 7.779950857162476
    },
    {
      "epoch": 6.551456451416016e-05,
      "model_forward_time": 1.751786231994629,
      "step": 68697
    },
    {
      "epoch": 6.551456451416016e-05,
      "step": 68697,
      "training_step_time": 10.162152767181396
    },
    {
      "epoch": 6.551551818847656e-05,
      "model_forward_time": 1.7548909187316895,
      "step": 68698
    },
    {
      "epoch": 6.551551818847656e-05,
      "step": 68698,
      "training_step_time": 7.663445472717285
    },
    {
      "epoch": 6.551647186279297e-05,
      "model_forward_time": 1.7452771663665771,
      "step": 68699
    },
    {
      "epoch": 6.551647186279297e-05,
      "step": 68699,
      "training_step_time": 7.337526082992554
    },
    {
      "epoch": 6.551742553710938e-05,
      "grad_norm": 0.09843365103006363,
      "learning_rate": 2.447822532055149e-06,
      "loss": 0.0532,
      "step": 68700
    },
    {
      "epoch": 6.551742553710938e-05,
      "model_forward_time": 1.763038158416748,
      "step": 68700
    },
    {
      "dynamics_loss_avg": 0.06938157230615616,
      "epoch": 6.551742553710938e-05,
      "step": 68700
    },
    {
      "action_loss_avg": 0.0017109395936131476,
      "epoch": 6.551742553710938e-05,
      "step": 68700
    },
    {
      "epoch": 6.551742553710938e-05,
      "step": 68700,
      "training_step_time": 8.650909423828125
    },
    {
      "epoch": 6.551837921142577e-05,
      "model_forward_time": 1.7601213455200195,
      "step": 68701
    },
    {
      "epoch": 6.551837921142577e-05,
      "step": 68701,
      "training_step_time": 7.410293340682983
    },
    {
      "epoch": 6.551933288574219e-05,
      "model_forward_time": 1.7514660358428955,
      "step": 68702
    },
    {
      "epoch": 6.551933288574219e-05,
      "step": 68702,
      "training_step_time": 7.578833341598511
    },
    {
      "epoch": 6.55202865600586e-05,
      "model_forward_time": 1.7551724910736084,
      "step": 68703
    },
    {
      "epoch": 6.55202865600586e-05,
      "step": 68703,
      "training_step_time": 8.179516315460205
    },
    {
      "epoch": 6.5521240234375e-05,
      "model_forward_time": 2.8894081115722656,
      "step": 68704
    },
    {
      "epoch": 6.5521240234375e-05,
      "step": 68704,
      "training_step_time": 8.239764213562012
    },
    {
      "epoch": 6.55221939086914e-05,
      "model_forward_time": 1.7401092052459717,
      "step": 68705
    },
    {
      "epoch": 6.55221939086914e-05,
      "step": 68705,
      "training_step_time": 7.778557538986206
    },
    {
      "epoch": 6.552314758300782e-05,
      "model_forward_time": 1.7438125610351562,
      "step": 68706
    },
    {
      "epoch": 6.552314758300782e-05,
      "step": 68706,
      "training_step_time": 9.66670560836792
    },
    {
      "epoch": 6.552410125732421e-05,
      "model_forward_time": 1.7439146041870117,
      "step": 68707
    },
    {
      "epoch": 6.552410125732421e-05,
      "step": 68707,
      "training_step_time": 7.66225528717041
    },
    {
      "epoch": 6.552505493164062e-05,
      "model_forward_time": 1.7226450443267822,
      "step": 68708
    },
    {
      "epoch": 6.552505493164062e-05,
      "step": 68708,
      "training_step_time": 7.298207521438599
    },
    {
      "epoch": 6.552600860595703e-05,
      "model_forward_time": 1.732093095779419,
      "step": 68709
    },
    {
      "epoch": 6.552600860595703e-05,
      "step": 68709,
      "training_step_time": 7.5273497104644775
    },
    {
      "epoch": 6.552696228027343e-05,
      "grad_norm": 0.0919024720788002,
      "learning_rate": 2.4464008264037215e-06,
      "loss": 0.0512,
      "step": 68710
    },
    {
      "epoch": 6.552696228027343e-05,
      "model_forward_time": 1.7297396659851074,
      "step": 68710
    },
    {
      "dynamics_loss_avg": 0.048230011388659476,
      "epoch": 6.552696228027343e-05,
      "step": 68710
    },
    {
      "action_loss_avg": 0.00248355814255774,
      "epoch": 6.552696228027343e-05,
      "step": 68710
    },
    {
      "epoch": 6.552696228027343e-05,
      "step": 68710,
      "training_step_time": 7.758692502975464
    },
    {
      "epoch": 6.552791595458984e-05,
      "model_forward_time": 1.7331998348236084,
      "step": 68711
    },
    {
      "epoch": 6.552791595458984e-05,
      "step": 68711,
      "training_step_time": 7.528043985366821
    },
    {
      "epoch": 6.552886962890625e-05,
      "model_forward_time": 1.7611820697784424,
      "step": 68712
    },
    {
      "epoch": 6.552886962890625e-05,
      "step": 68712,
      "training_step_time": 8.269273519515991
    },
    {
      "epoch": 6.552982330322265e-05,
      "model_forward_time": 2.8700976371765137,
      "step": 68713
    },
    {
      "epoch": 6.552982330322265e-05,
      "step": 68713,
      "training_step_time": 8.44632625579834
    },
    {
      "epoch": 6.553077697753906e-05,
      "model_forward_time": 1.7620680332183838,
      "step": 68714
    },
    {
      "epoch": 6.553077697753906e-05,
      "step": 68714,
      "training_step_time": 8.391707181930542
    },
    {
      "epoch": 6.553173065185547e-05,
      "model_forward_time": 1.7456724643707275,
      "step": 68715
    },
    {
      "epoch": 6.553173065185547e-05,
      "step": 68715,
      "training_step_time": 8.311236143112183
    },
    {
      "epoch": 6.553268432617187e-05,
      "model_forward_time": 1.7445204257965088,
      "step": 68716
    },
    {
      "epoch": 6.553268432617187e-05,
      "step": 68716,
      "training_step_time": 7.601263523101807
    },
    {
      "epoch": 6.553363800048828e-05,
      "model_forward_time": 1.73091459274292,
      "step": 68717
    },
    {
      "epoch": 6.553363800048828e-05,
      "step": 68717,
      "training_step_time": 7.563409090042114
    },
    {
      "epoch": 6.553459167480469e-05,
      "model_forward_time": 1.727184534072876,
      "step": 68718
    },
    {
      "epoch": 6.553459167480469e-05,
      "step": 68718,
      "training_step_time": 7.433182001113892
    },
    {
      "epoch": 6.553554534912109e-05,
      "model_forward_time": 1.7351481914520264,
      "step": 68719
    },
    {
      "epoch": 6.553554534912109e-05,
      "step": 68719,
      "training_step_time": 7.623342275619507
    },
    {
      "epoch": 6.55364990234375e-05,
      "grad_norm": 0.10381262749433517,
      "learning_rate": 2.4449794000100608e-06,
      "loss": 0.0512,
      "step": 68720
    },
    {
      "epoch": 6.55364990234375e-05,
      "model_forward_time": 1.740020990371704,
      "step": 68720
    },
    {
      "dynamics_loss_avg": 0.05240446217358112,
      "epoch": 6.55364990234375e-05,
      "step": 68720
    },
    {
      "action_loss_avg": 0.0018270407803356648,
      "epoch": 6.55364990234375e-05,
      "step": 68720
    },
    {
      "epoch": 6.55364990234375e-05,
      "step": 68720,
      "training_step_time": 7.472829580307007
    },
    {
      "epoch": 6.553745269775391e-05,
      "model_forward_time": 1.7327313423156738,
      "step": 68721
    },
    {
      "epoch": 6.553745269775391e-05,
      "step": 68721,
      "training_step_time": 8.247607946395874
    },
    {
      "epoch": 6.553840637207031e-05,
      "model_forward_time": 1.738731861114502,
      "step": 68722
    },
    {
      "epoch": 6.553840637207031e-05,
      "step": 68722,
      "training_step_time": 7.596707820892334
    },
    {
      "epoch": 6.553936004638672e-05,
      "model_forward_time": 2.9265403747558594,
      "step": 68723
    },
    {
      "epoch": 6.553936004638672e-05,
      "step": 68723,
      "training_step_time": 9.478120803833008
    },
    {
      "epoch": 6.554031372070313e-05,
      "model_forward_time": 1.7386655807495117,
      "step": 68724
    },
    {
      "epoch": 6.554031372070313e-05,
      "step": 68724,
      "training_step_time": 7.601605176925659
    },
    {
      "epoch": 6.554126739501953e-05,
      "model_forward_time": 1.7377183437347412,
      "step": 68725
    },
    {
      "epoch": 6.554126739501953e-05,
      "step": 68725,
      "training_step_time": 8.566864252090454
    },
    {
      "epoch": 6.554222106933594e-05,
      "model_forward_time": 1.7472419738769531,
      "step": 68726
    },
    {
      "epoch": 6.554222106933594e-05,
      "step": 68726,
      "training_step_time": 7.594790458679199
    },
    {
      "epoch": 6.554317474365235e-05,
      "model_forward_time": 1.694244623184204,
      "step": 68727
    },
    {
      "epoch": 6.554317474365235e-05,
      "step": 68727,
      "training_step_time": 7.348701477050781
    },
    {
      "epoch": 6.554412841796874e-05,
      "model_forward_time": 1.7005836963653564,
      "step": 68728
    },
    {
      "epoch": 6.554412841796874e-05,
      "step": 68728,
      "training_step_time": 7.646623134613037
    },
    {
      "epoch": 6.554508209228516e-05,
      "model_forward_time": 1.7062077522277832,
      "step": 68729
    },
    {
      "epoch": 6.554508209228516e-05,
      "step": 68729,
      "training_step_time": 7.463845491409302
    },
    {
      "epoch": 6.554603576660157e-05,
      "grad_norm": 0.09314588457345963,
      "learning_rate": 2.443558253029611e-06,
      "loss": 0.0511,
      "step": 68730
    },
    {
      "epoch": 6.554603576660157e-05,
      "model_forward_time": 1.7037262916564941,
      "step": 68730
    },
    {
      "dynamics_loss_avg": 0.049884643778204917,
      "epoch": 6.554603576660157e-05,
      "step": 68730
    },
    {
      "action_loss_avg": 0.001219338973169215,
      "epoch": 6.554603576660157e-05,
      "step": 68730
    },
    {
      "epoch": 6.554603576660157e-05,
      "step": 68730,
      "training_step_time": 7.879591703414917
    },
    {
      "epoch": 6.554698944091796e-05,
      "model_forward_time": 1.7116055488586426,
      "step": 68731
    },
    {
      "epoch": 6.554698944091796e-05,
      "step": 68731,
      "training_step_time": 8.381240844726562
    },
    {
      "epoch": 6.554794311523437e-05,
      "model_forward_time": 2.299886703491211,
      "step": 68732
    },
    {
      "epoch": 6.554794311523437e-05,
      "step": 68732,
      "training_step_time": 7.684949159622192
    },
    {
      "epoch": 6.554889678955079e-05,
      "model_forward_time": 1.7064754962921143,
      "step": 68733
    },
    {
      "epoch": 6.554889678955079e-05,
      "step": 68733,
      "training_step_time": 9.438107967376709
    },
    {
      "epoch": 6.554985046386718e-05,
      "model_forward_time": 1.7151679992675781,
      "step": 68734
    },
    {
      "epoch": 6.554985046386718e-05,
      "step": 68734,
      "training_step_time": 7.614304542541504
    },
    {
      "epoch": 6.555080413818359e-05,
      "model_forward_time": 1.6967852115631104,
      "step": 68735
    },
    {
      "epoch": 6.555080413818359e-05,
      "step": 68735,
      "training_step_time": 7.607205629348755
    },
    {
      "epoch": 6.55517578125e-05,
      "model_forward_time": 1.7081928253173828,
      "step": 68736
    },
    {
      "epoch": 6.55517578125e-05,
      "step": 68736,
      "training_step_time": 7.228256464004517
    },
    {
      "epoch": 6.55527114868164e-05,
      "model_forward_time": 1.7090742588043213,
      "step": 68737
    },
    {
      "epoch": 6.55527114868164e-05,
      "step": 68737,
      "training_step_time": 7.709896087646484
    },
    {
      "epoch": 6.555366516113281e-05,
      "model_forward_time": 1.710172176361084,
      "step": 68738
    },
    {
      "epoch": 6.555366516113281e-05,
      "step": 68738,
      "training_step_time": 7.4900963306427
    },
    {
      "epoch": 6.555461883544922e-05,
      "model_forward_time": 1.7014288902282715,
      "step": 68739
    },
    {
      "epoch": 6.555461883544922e-05,
      "step": 68739,
      "training_step_time": 7.456845045089722
    },
    {
      "epoch": 6.555557250976562e-05,
      "grad_norm": 0.1307845115661621,
      "learning_rate": 2.442137385617789e-06,
      "loss": 0.0559,
      "step": 68740
    },
    {
      "epoch": 6.555557250976562e-05,
      "model_forward_time": 1.7040536403656006,
      "step": 68740
    },
    {
      "dynamics_loss_avg": 0.050402211397886275,
      "epoch": 6.555557250976562e-05,
      "step": 68740
    },
    {
      "action_loss_avg": 0.00147570307017304,
      "epoch": 6.555557250976562e-05,
      "step": 68740
    },
    {
      "epoch": 6.555557250976562e-05,
      "step": 68740,
      "training_step_time": 7.614398956298828
    },
    {
      "epoch": 6.555652618408203e-05,
      "model_forward_time": 2.3148884773254395,
      "step": 68741
    },
    {
      "epoch": 6.555652618408203e-05,
      "step": 68741,
      "training_step_time": 7.799931764602661
    },
    {
      "epoch": 6.555747985839844e-05,
      "model_forward_time": 1.7081987857818604,
      "step": 68742
    },
    {
      "epoch": 6.555747985839844e-05,
      "step": 68742,
      "training_step_time": 8.949061155319214
    },
    {
      "epoch": 6.555843353271484e-05,
      "model_forward_time": 1.7147715091705322,
      "step": 68743
    },
    {
      "epoch": 6.555843353271484e-05,
      "step": 68743,
      "training_step_time": 7.586049795150757
    },
    {
      "epoch": 6.555938720703125e-05,
      "model_forward_time": 1.7077195644378662,
      "step": 68744
    },
    {
      "epoch": 6.555938720703125e-05,
      "step": 68744,
      "training_step_time": 8.219597578048706
    },
    {
      "epoch": 6.556034088134766e-05,
      "model_forward_time": 1.7006142139434814,
      "step": 68745
    },
    {
      "epoch": 6.556034088134766e-05,
      "step": 68745,
      "training_step_time": 7.295730352401733
    },
    {
      "epoch": 6.556129455566406e-05,
      "model_forward_time": 1.7124903202056885,
      "step": 68746
    },
    {
      "epoch": 6.556129455566406e-05,
      "step": 68746,
      "training_step_time": 7.65404486656189
    },
    {
      "epoch": 6.556224822998047e-05,
      "model_forward_time": 1.712679147720337,
      "step": 68747
    },
    {
      "epoch": 6.556224822998047e-05,
      "step": 68747,
      "training_step_time": 7.569636583328247
    },
    {
      "epoch": 6.556320190429688e-05,
      "model_forward_time": 1.7062950134277344,
      "step": 68748
    },
    {
      "epoch": 6.556320190429688e-05,
      "step": 68748,
      "training_step_time": 8.290616273880005
    },
    {
      "epoch": 6.556415557861328e-05,
      "model_forward_time": 1.7051961421966553,
      "step": 68749
    },
    {
      "epoch": 6.556415557861328e-05,
      "step": 68749,
      "training_step_time": 7.5725672245025635
    },
    {
      "epoch": 6.556510925292969e-05,
      "grad_norm": 0.0850796103477478,
      "learning_rate": 2.440716797929975e-06,
      "loss": 0.0532,
      "step": 68750
    },
    {
      "epoch": 6.556510925292969e-05,
      "model_forward_time": 2.8886373043060303,
      "step": 68750
    },
    {
      "dynamics_loss_avg": 0.052101961150765416,
      "epoch": 6.556510925292969e-05,
      "step": 68750
    },
    {
      "action_loss_avg": 0.0019085856445599347,
      "epoch": 6.556510925292969e-05,
      "step": 68750
    },
    {
      "epoch": 6.556510925292969e-05,
      "step": 68750,
      "training_step_time": 7.682199239730835
    },
    {
      "epoch": 6.55660629272461e-05,
      "model_forward_time": 1.7247838973999023,
      "step": 68751
    },
    {
      "epoch": 6.55660629272461e-05,
      "step": 68751,
      "training_step_time": 9.736968994140625
    },
    {
      "epoch": 6.55670166015625e-05,
      "model_forward_time": 1.7234022617340088,
      "step": 68752
    },
    {
      "epoch": 6.55670166015625e-05,
      "step": 68752,
      "training_step_time": 8.327897548675537
    },
    {
      "epoch": 6.55679702758789e-05,
      "model_forward_time": 1.738849401473999,
      "step": 68753
    },
    {
      "epoch": 6.55679702758789e-05,
      "step": 68753,
      "training_step_time": 8.19857668876648
    },
    {
      "epoch": 6.556892395019532e-05,
      "model_forward_time": 1.7400379180908203,
      "step": 68754
    },
    {
      "epoch": 6.556892395019532e-05,
      "step": 68754,
      "training_step_time": 7.685991048812866
    },
    {
      "epoch": 6.556987762451171e-05,
      "model_forward_time": 1.7405354976654053,
      "step": 68755
    },
    {
      "epoch": 6.556987762451171e-05,
      "step": 68755,
      "training_step_time": 7.510238170623779
    },
    {
      "epoch": 6.557083129882813e-05,
      "model_forward_time": 1.7380874156951904,
      "step": 68756
    },
    {
      "epoch": 6.557083129882813e-05,
      "step": 68756,
      "training_step_time": 7.735372066497803
    },
    {
      "epoch": 6.557178497314454e-05,
      "model_forward_time": 1.734785556793213,
      "step": 68757
    },
    {
      "epoch": 6.557178497314454e-05,
      "step": 68757,
      "training_step_time": 7.418678522109985
    },
    {
      "epoch": 6.557273864746093e-05,
      "model_forward_time": 1.7961201667785645,
      "step": 68758
    },
    {
      "epoch": 6.557273864746093e-05,
      "step": 68758,
      "training_step_time": 7.756504535675049
    },
    {
      "epoch": 6.557369232177734e-05,
      "model_forward_time": 2.8751227855682373,
      "step": 68759
    },
    {
      "epoch": 6.557369232177734e-05,
      "step": 68759,
      "training_step_time": 9.099801778793335
    },
    {
      "epoch": 6.557464599609375e-05,
      "grad_norm": 0.0979737862944603,
      "learning_rate": 2.439296490121526e-06,
      "loss": 0.0569,
      "step": 68760
    },
    {
      "epoch": 6.557464599609375e-05,
      "model_forward_time": 1.7379875183105469,
      "step": 68760
    },
    {
      "dynamics_loss_avg": 0.04720798544585705,
      "epoch": 6.557464599609375e-05,
      "step": 68760
    },
    {
      "action_loss_avg": 0.0017809275828767567,
      "epoch": 6.557464599609375e-05,
      "step": 68760
    },
    {
      "epoch": 6.557464599609375e-05,
      "step": 68760,
      "training_step_time": 8.883858680725098
    },
    {
      "epoch": 6.557559967041015e-05,
      "model_forward_time": 1.7433929443359375,
      "step": 68761
    },
    {
      "epoch": 6.557559967041015e-05,
      "step": 68761,
      "training_step_time": 9.110616445541382
    },
    {
      "epoch": 6.557655334472656e-05,
      "model_forward_time": 1.7324244976043701,
      "step": 68762
    },
    {
      "epoch": 6.557655334472656e-05,
      "step": 68762,
      "training_step_time": 8.692502498626709
    },
    {
      "epoch": 6.557750701904297e-05,
      "model_forward_time": 1.7253484725952148,
      "step": 68763
    },
    {
      "epoch": 6.557750701904297e-05,
      "step": 68763,
      "training_step_time": 7.694365739822388
    },
    {
      "epoch": 6.557846069335937e-05,
      "model_forward_time": 1.0522289276123047,
      "step": 68764
    },
    {
      "epoch": 6.557846069335937e-05,
      "step": 68764,
      "training_step_time": 7.579230785369873
    },
    {
      "epoch": 6.557941436767578e-05,
      "model_forward_time": 1.7326610088348389,
      "step": 68765
    },
    {
      "epoch": 6.557941436767578e-05,
      "step": 68765,
      "training_step_time": 7.683542490005493
    },
    {
      "epoch": 6.558036804199219e-05,
      "model_forward_time": 1.7418556213378906,
      "step": 68766
    },
    {
      "epoch": 6.558036804199219e-05,
      "step": 68766,
      "training_step_time": 7.321304798126221
    },
    {
      "epoch": 6.558132171630859e-05,
      "model_forward_time": 1.7950670719146729,
      "step": 68767
    },
    {
      "epoch": 6.558132171630859e-05,
      "step": 68767,
      "training_step_time": 8.591087818145752
    },
    {
      "epoch": 6.5582275390625e-05,
      "model_forward_time": 1.738128662109375,
      "step": 68768
    },
    {
      "epoch": 6.5582275390625e-05,
      "step": 68768,
      "training_step_time": 8.694938659667969
    },
    {
      "epoch": 6.558322906494141e-05,
      "model_forward_time": 2.934307098388672,
      "step": 68769
    },
    {
      "epoch": 6.558322906494141e-05,
      "step": 68769,
      "training_step_time": 8.39233922958374
    },
    {
      "epoch": 6.558418273925781e-05,
      "grad_norm": 0.11273560672998428,
      "learning_rate": 2.4378764623477608e-06,
      "loss": 0.056,
      "step": 68770
    },
    {
      "epoch": 6.558418273925781e-05,
      "model_forward_time": 1.7320644855499268,
      "step": 68770
    },
    {
      "dynamics_loss_avg": 0.054982255399227145,
      "epoch": 6.558418273925781e-05,
      "step": 68770
    },
    {
      "action_loss_avg": 0.0009663223216193728,
      "epoch": 6.558418273925781e-05,
      "step": 68770
    },
    {
      "epoch": 6.558418273925781e-05,
      "step": 68770,
      "training_step_time": 8.475335359573364
    },
    {
      "epoch": 6.558513641357422e-05,
      "model_forward_time": 1.7445752620697021,
      "step": 68771
    },
    {
      "epoch": 6.558513641357422e-05,
      "step": 68771,
      "training_step_time": 9.050993204116821
    },
    {
      "epoch": 6.558609008789063e-05,
      "model_forward_time": 1.7365286350250244,
      "step": 68772
    },
    {
      "epoch": 6.558609008789063e-05,
      "step": 68772,
      "training_step_time": 7.711250066757202
    },
    {
      "epoch": 6.558704376220703e-05,
      "model_forward_time": 1.742077112197876,
      "step": 68773
    },
    {
      "epoch": 6.558704376220703e-05,
      "step": 68773,
      "training_step_time": 7.6032493114471436
    },
    {
      "epoch": 6.558799743652344e-05,
      "model_forward_time": 1.7403461933135986,
      "step": 68774
    },
    {
      "epoch": 6.558799743652344e-05,
      "step": 68774,
      "training_step_time": 7.377969741821289
    },
    {
      "epoch": 6.558895111083985e-05,
      "model_forward_time": 1.7394614219665527,
      "step": 68775
    },
    {
      "epoch": 6.558895111083985e-05,
      "step": 68775,
      "training_step_time": 7.725276470184326
    },
    {
      "epoch": 6.558990478515625e-05,
      "model_forward_time": 1.7568013668060303,
      "step": 68776
    },
    {
      "epoch": 6.558990478515625e-05,
      "step": 68776,
      "training_step_time": 7.602109670639038
    },
    {
      "epoch": 6.559085845947266e-05,
      "model_forward_time": 1.752342939376831,
      "step": 68777
    },
    {
      "epoch": 6.559085845947266e-05,
      "step": 68777,
      "training_step_time": 10.035489559173584
    },
    {
      "epoch": 6.559181213378907e-05,
      "model_forward_time": 2.894758939743042,
      "step": 68778
    },
    {
      "epoch": 6.559181213378907e-05,
      "step": 68778,
      "training_step_time": 8.206406593322754
    },
    {
      "epoch": 6.559276580810547e-05,
      "model_forward_time": 1.7418975830078125,
      "step": 68779
    },
    {
      "epoch": 6.559276580810547e-05,
      "step": 68779,
      "training_step_time": 8.750424146652222
    },
    {
      "epoch": 6.559371948242188e-05,
      "grad_norm": 0.09304753690958023,
      "learning_rate": 2.436456714763974e-06,
      "loss": 0.0546,
      "step": 68780
    },
    {
      "epoch": 6.559371948242188e-05,
      "model_forward_time": 1.7269330024719238,
      "step": 68780
    },
    {
      "dynamics_loss_avg": 0.05781589699909091,
      "epoch": 6.559371948242188e-05,
      "step": 68780
    },
    {
      "action_loss_avg": 0.004968732764245943,
      "epoch": 6.559371948242188e-05,
      "step": 68780
    },
    {
      "epoch": 6.559371948242188e-05,
      "step": 68780,
      "training_step_time": 7.6095130443573
    },
    {
      "epoch": 6.559467315673829e-05,
      "model_forward_time": 1.7387020587921143,
      "step": 68781
    },
    {
      "epoch": 6.559467315673829e-05,
      "step": 68781,
      "training_step_time": 8.227816104888916
    },
    {
      "epoch": 6.559562683105468e-05,
      "model_forward_time": 1.7399523258209229,
      "step": 68782
    },
    {
      "epoch": 6.559562683105468e-05,
      "step": 68782,
      "training_step_time": 7.429640293121338
    },
    {
      "epoch": 6.55965805053711e-05,
      "model_forward_time": 1.7410011291503906,
      "step": 68783
    },
    {
      "epoch": 6.55965805053711e-05,
      "step": 68783,
      "training_step_time": 7.379304647445679
    },
    {
      "epoch": 6.55975341796875e-05,
      "model_forward_time": 1.7308216094970703,
      "step": 68784
    },
    {
      "epoch": 6.55975341796875e-05,
      "step": 68784,
      "training_step_time": 7.739461660385132
    },
    {
      "epoch": 6.55984878540039e-05,
      "model_forward_time": 1.737403392791748,
      "step": 68785
    },
    {
      "epoch": 6.55984878540039e-05,
      "step": 68785,
      "training_step_time": 8.188118696212769
    },
    {
      "epoch": 6.559944152832031e-05,
      "model_forward_time": 1.3809947967529297,
      "step": 68786
    },
    {
      "epoch": 6.559944152832031e-05,
      "step": 68786,
      "training_step_time": 8.596828699111938
    },
    {
      "epoch": 6.560039520263672e-05,
      "model_forward_time": 2.8939294815063477,
      "step": 68787
    },
    {
      "epoch": 6.560039520263672e-05,
      "step": 68787,
      "training_step_time": 7.824826955795288
    },
    {
      "epoch": 6.560134887695312e-05,
      "model_forward_time": 1.738276481628418,
      "step": 68788
    },
    {
      "epoch": 6.560134887695312e-05,
      "step": 68788,
      "training_step_time": 9.825835704803467
    },
    {
      "epoch": 6.560230255126953e-05,
      "model_forward_time": 1.7195870876312256,
      "step": 68789
    },
    {
      "epoch": 6.560230255126953e-05,
      "step": 68789,
      "training_step_time": 7.667978763580322
    },
    {
      "epoch": 6.560325622558594e-05,
      "grad_norm": 0.08560772985219955,
      "learning_rate": 2.4350372475254265e-06,
      "loss": 0.0582,
      "step": 68790
    },
    {
      "epoch": 6.560325622558594e-05,
      "model_forward_time": 1.7390162944793701,
      "step": 68790
    },
    {
      "dynamics_loss_avg": 0.04958796463906765,
      "epoch": 6.560325622558594e-05,
      "step": 68790
    },
    {
      "action_loss_avg": 0.0017285828886087983,
      "epoch": 6.560325622558594e-05,
      "step": 68790
    },
    {
      "epoch": 6.560325622558594e-05,
      "step": 68790,
      "training_step_time": 8.681291818618774
    },
    {
      "epoch": 6.560420989990234e-05,
      "model_forward_time": 1.7257308959960938,
      "step": 68791
    },
    {
      "epoch": 6.560420989990234e-05,
      "step": 68791,
      "training_step_time": 7.702112436294556
    },
    {
      "epoch": 6.560516357421875e-05,
      "model_forward_time": 1.742757797241211,
      "step": 68792
    },
    {
      "epoch": 6.560516357421875e-05,
      "step": 68792,
      "training_step_time": 7.593418598175049
    },
    {
      "epoch": 6.560611724853516e-05,
      "model_forward_time": 1.7319371700286865,
      "step": 68793
    },
    {
      "epoch": 6.560611724853516e-05,
      "step": 68793,
      "training_step_time": 7.793091297149658
    },
    {
      "epoch": 6.560707092285156e-05,
      "model_forward_time": 1.7138497829437256,
      "step": 68794
    },
    {
      "epoch": 6.560707092285156e-05,
      "step": 68794,
      "training_step_time": 7.689916133880615
    },
    {
      "epoch": 6.560802459716797e-05,
      "model_forward_time": 1.7095096111297607,
      "step": 68795
    },
    {
      "epoch": 6.560802459716797e-05,
      "step": 68795,
      "training_step_time": 8.684322357177734
    },
    {
      "epoch": 6.560897827148438e-05,
      "model_forward_time": 1.7302982807159424,
      "step": 68796
    },
    {
      "epoch": 6.560897827148438e-05,
      "step": 68796,
      "training_step_time": 8.263887405395508
    },
    {
      "epoch": 6.560993194580078e-05,
      "model_forward_time": 1.7147486209869385,
      "step": 68797
    },
    {
      "epoch": 6.560993194580078e-05,
      "step": 68797,
      "training_step_time": 7.739941596984863
    },
    {
      "epoch": 6.561088562011719e-05,
      "model_forward_time": 1.7153346538543701,
      "step": 68798
    },
    {
      "epoch": 6.561088562011719e-05,
      "step": 68798,
      "training_step_time": 8.294115781784058
    },
    {
      "epoch": 6.56118392944336e-05,
      "model_forward_time": 1.7210874557495117,
      "step": 68799
    },
    {
      "epoch": 6.56118392944336e-05,
      "step": 68799,
      "training_step_time": 7.608843803405762
    },
    {
      "epoch": 6.561279296875e-05,
      "grad_norm": 0.10836225003004074,
      "learning_rate": 2.433618060787351e-06,
      "loss": 0.054,
      "step": 68800
    },
    {
      "epoch": 6.561279296875e-05,
      "model_forward_time": 1.7244372367858887,
      "step": 68800
    },
    {
      "dynamics_loss_avg": 0.05092533379793167,
      "epoch": 6.561279296875e-05,
      "step": 68800
    },
    {
      "action_loss_avg": 0.0011490054603200406,
      "epoch": 6.561279296875e-05,
      "step": 68800
    },
    {
      "epoch": 6.561279296875e-05,
      "step": 68800,
      "training_step_time": 7.7566046714782715
    },
    {
      "epoch": 6.561374664306641e-05,
      "model_forward_time": 1.729212760925293,
      "step": 68801
    },
    {
      "epoch": 6.561374664306641e-05,
      "step": 68801,
      "training_step_time": 7.62608790397644
    },
    {
      "epoch": 6.561470031738282e-05,
      "model_forward_time": 1.7374820709228516,
      "step": 68802
    },
    {
      "epoch": 6.561470031738282e-05,
      "step": 68802,
      "training_step_time": 7.788069725036621
    },
    {
      "epoch": 6.561565399169922e-05,
      "model_forward_time": 1.7296466827392578,
      "step": 68803
    },
    {
      "epoch": 6.561565399169922e-05,
      "step": 68803,
      "training_step_time": 7.606744050979614
    },
    {
      "epoch": 6.561660766601563e-05,
      "model_forward_time": 1.799332857131958,
      "step": 68804
    },
    {
      "epoch": 6.561660766601563e-05,
      "step": 68804,
      "training_step_time": 7.7429327964782715
    },
    {
      "epoch": 6.561756134033204e-05,
      "model_forward_time": 1.7265727519989014,
      "step": 68805
    },
    {
      "epoch": 6.561756134033204e-05,
      "step": 68805,
      "training_step_time": 7.860104322433472
    },
    {
      "epoch": 6.561851501464844e-05,
      "model_forward_time": 2.11881160736084,
      "step": 68806
    },
    {
      "epoch": 6.561851501464844e-05,
      "step": 68806,
      "training_step_time": 8.793959379196167
    },
    {
      "epoch": 6.561946868896485e-05,
      "model_forward_time": 1.7394990921020508,
      "step": 68807
    },
    {
      "epoch": 6.561946868896485e-05,
      "step": 68807,
      "training_step_time": 9.813210725784302
    },
    {
      "epoch": 6.562042236328126e-05,
      "model_forward_time": 1.3725085258483887,
      "step": 68808
    },
    {
      "epoch": 6.562042236328126e-05,
      "step": 68808,
      "training_step_time": 7.61491584777832
    },
    {
      "epoch": 6.562137603759765e-05,
      "model_forward_time": 1.7147881984710693,
      "step": 68809
    },
    {
      "epoch": 6.562137603759765e-05,
      "step": 68809,
      "training_step_time": 7.809895277023315
    },
    {
      "epoch": 6.562232971191406e-05,
      "grad_norm": 0.1050807312130928,
      "learning_rate": 2.432199154704944e-06,
      "loss": 0.0557,
      "step": 68810
    },
    {
      "epoch": 6.562232971191406e-05,
      "model_forward_time": 1.7262523174285889,
      "step": 68810
    },
    {
      "dynamics_loss_avg": 0.04549590647220612,
      "epoch": 6.562232971191406e-05,
      "step": 68810
    },
    {
      "action_loss_avg": 0.0017059523641364649,
      "epoch": 6.562232971191406e-05,
      "step": 68810
    },
    {
      "epoch": 6.562232971191406e-05,
      "step": 68810,
      "training_step_time": 7.63587212562561
    },
    {
      "epoch": 6.562328338623048e-05,
      "model_forward_time": 1.7468714714050293,
      "step": 68811
    },
    {
      "epoch": 6.562328338623048e-05,
      "step": 68811,
      "training_step_time": 7.6948113441467285
    },
    {
      "epoch": 6.562423706054687e-05,
      "model_forward_time": 1.7536110877990723,
      "step": 68812
    },
    {
      "epoch": 6.562423706054687e-05,
      "step": 68812,
      "training_step_time": 7.819240570068359
    },
    {
      "epoch": 6.562519073486328e-05,
      "model_forward_time": 1.8178071975708008,
      "step": 68813
    },
    {
      "epoch": 6.562519073486328e-05,
      "step": 68813,
      "training_step_time": 7.76306939125061
    },
    {
      "epoch": 6.562614440917968e-05,
      "model_forward_time": 1.7463161945343018,
      "step": 68814
    },
    {
      "epoch": 6.562614440917968e-05,
      "step": 68814,
      "training_step_time": 9.228806495666504
    },
    {
      "epoch": 6.562709808349609e-05,
      "model_forward_time": 1.7527344226837158,
      "step": 68815
    },
    {
      "epoch": 6.562709808349609e-05,
      "step": 68815,
      "training_step_time": 9.307447910308838
    },
    {
      "epoch": 6.56280517578125e-05,
      "model_forward_time": 1.7502422332763672,
      "step": 68816
    },
    {
      "epoch": 6.56280517578125e-05,
      "step": 68816,
      "training_step_time": 8.332772970199585
    },
    {
      "epoch": 6.56290054321289e-05,
      "model_forward_time": 1.7578215599060059,
      "step": 68817
    },
    {
      "epoch": 6.56290054321289e-05,
      "step": 68817,
      "training_step_time": 7.701796770095825
    },
    {
      "epoch": 6.562995910644531e-05,
      "model_forward_time": 1.7494635581970215,
      "step": 68818
    },
    {
      "epoch": 6.562995910644531e-05,
      "step": 68818,
      "training_step_time": 7.710707426071167
    },
    {
      "epoch": 6.563091278076172e-05,
      "model_forward_time": 1.7495713233947754,
      "step": 68819
    },
    {
      "epoch": 6.563091278076172e-05,
      "step": 68819,
      "training_step_time": 7.31865382194519
    },
    {
      "epoch": 6.563186645507812e-05,
      "grad_norm": 0.14818888902664185,
      "learning_rate": 2.430780529433377e-06,
      "loss": 0.0534,
      "step": 68820
    },
    {
      "epoch": 6.563186645507812e-05,
      "model_forward_time": 1.7541029453277588,
      "step": 68820
    },
    {
      "dynamics_loss_avg": 0.07483105696737766,
      "epoch": 6.563186645507812e-05,
      "step": 68820
    },
    {
      "action_loss_avg": 0.0021600170875899494,
      "epoch": 6.563186645507812e-05,
      "step": 68820
    },
    {
      "epoch": 6.563186645507812e-05,
      "step": 68820,
      "training_step_time": 7.665452480316162
    },
    {
      "epoch": 6.563282012939453e-05,
      "model_forward_time": 1.749603033065796,
      "step": 68821
    },
    {
      "epoch": 6.563282012939453e-05,
      "step": 68821,
      "training_step_time": 7.803231716156006
    },
    {
      "epoch": 6.563377380371094e-05,
      "model_forward_time": 1.8029193878173828,
      "step": 68822
    },
    {
      "epoch": 6.563377380371094e-05,
      "step": 68822,
      "training_step_time": 7.731673240661621
    },
    {
      "epoch": 6.563472747802734e-05,
      "model_forward_time": 1.7533414363861084,
      "step": 68823
    },
    {
      "epoch": 6.563472747802734e-05,
      "step": 68823,
      "training_step_time": 8.54535174369812
    },
    {
      "epoch": 6.563568115234375e-05,
      "model_forward_time": 1.7325119972229004,
      "step": 68824
    },
    {
      "epoch": 6.563568115234375e-05,
      "step": 68824,
      "training_step_time": 9.849914789199829
    },
    {
      "epoch": 6.563663482666016e-05,
      "model_forward_time": 2.955717086791992,
      "step": 68825
    },
    {
      "epoch": 6.563663482666016e-05,
      "step": 68825,
      "training_step_time": 7.813488483428955
    },
    {
      "epoch": 6.563758850097656e-05,
      "model_forward_time": 1.7325973510742188,
      "step": 68826
    },
    {
      "epoch": 6.563758850097656e-05,
      "step": 68826,
      "training_step_time": 7.70691704750061
    },
    {
      "epoch": 6.563854217529297e-05,
      "model_forward_time": 1.7253632545471191,
      "step": 68827
    },
    {
      "epoch": 6.563854217529297e-05,
      "step": 68827,
      "training_step_time": 8.284895896911621
    },
    {
      "epoch": 6.563949584960938e-05,
      "model_forward_time": 1.7439401149749756,
      "step": 68828
    },
    {
      "epoch": 6.563949584960938e-05,
      "step": 68828,
      "training_step_time": 7.348162412643433
    },
    {
      "epoch": 6.564044952392578e-05,
      "model_forward_time": 1.7287530899047852,
      "step": 68829
    },
    {
      "epoch": 6.564044952392578e-05,
      "step": 68829,
      "training_step_time": 7.590805292129517
    },
    {
      "epoch": 6.564140319824219e-05,
      "grad_norm": 0.10345165431499481,
      "learning_rate": 2.4293621851277897e-06,
      "loss": 0.0567,
      "step": 68830
    },
    {
      "epoch": 6.564140319824219e-05,
      "model_forward_time": 1.7309045791625977,
      "step": 68830
    },
    {
      "dynamics_loss_avg": 0.06681888774037362,
      "epoch": 6.564140319824219e-05,
      "step": 68830
    },
    {
      "action_loss_avg": 0.0017468996229581535,
      "epoch": 6.564140319824219e-05,
      "step": 68830
    },
    {
      "epoch": 6.564140319824219e-05,
      "step": 68830,
      "training_step_time": 7.871040344238281
    },
    {
      "epoch": 6.56423568725586e-05,
      "model_forward_time": 1.779287576675415,
      "step": 68831
    },
    {
      "epoch": 6.56423568725586e-05,
      "step": 68831,
      "training_step_time": 8.29338550567627
    },
    {
      "epoch": 6.5643310546875e-05,
      "model_forward_time": 1.7271325588226318,
      "step": 68832
    },
    {
      "epoch": 6.5643310546875e-05,
      "step": 68832,
      "training_step_time": 7.699814081192017
    },
    {
      "epoch": 6.56442642211914e-05,
      "model_forward_time": 1.726205825805664,
      "step": 68833
    },
    {
      "epoch": 6.56442642211914e-05,
      "step": 68833,
      "training_step_time": 8.952334642410278
    },
    {
      "epoch": 6.564521789550782e-05,
      "model_forward_time": 2.8940460681915283,
      "step": 68834
    },
    {
      "epoch": 6.564521789550782e-05,
      "step": 68834,
      "training_step_time": 8.321818113327026
    },
    {
      "epoch": 6.564617156982421e-05,
      "model_forward_time": 1.741457462310791,
      "step": 68835
    },
    {
      "epoch": 6.564617156982421e-05,
      "step": 68835,
      "training_step_time": 8.440662384033203
    },
    {
      "epoch": 6.564712524414062e-05,
      "model_forward_time": 1.7312138080596924,
      "step": 68836
    },
    {
      "epoch": 6.564712524414062e-05,
      "step": 68836,
      "training_step_time": 8.330174446105957
    },
    {
      "epoch": 6.564807891845703e-05,
      "model_forward_time": 1.7278578281402588,
      "step": 68837
    },
    {
      "epoch": 6.564807891845703e-05,
      "step": 68837,
      "training_step_time": 7.7150328159332275
    },
    {
      "epoch": 6.564903259277343e-05,
      "model_forward_time": 1.351715087890625,
      "step": 68838
    },
    {
      "epoch": 6.564903259277343e-05,
      "step": 68838,
      "training_step_time": 7.558562994003296
    },
    {
      "epoch": 6.564998626708984e-05,
      "model_forward_time": 1.7359473705291748,
      "step": 68839
    },
    {
      "epoch": 6.564998626708984e-05,
      "step": 68839,
      "training_step_time": 7.810633897781372
    },
    {
      "epoch": 6.565093994140625e-05,
      "grad_norm": 0.09111825376749039,
      "learning_rate": 2.427944121943288e-06,
      "loss": 0.0562,
      "step": 68840
    },
    {
      "epoch": 6.565093994140625e-05,
      "model_forward_time": 1.0345969200134277,
      "step": 68840
    },
    {
      "dynamics_loss_avg": 0.06466729827225208,
      "epoch": 6.565093994140625e-05,
      "step": 68840
    },
    {
      "action_loss_avg": 0.0033024548829416744,
      "epoch": 6.565093994140625e-05,
      "step": 68840
    },
    {
      "epoch": 6.565093994140625e-05,
      "step": 68840,
      "training_step_time": 7.560994386672974
    },
    {
      "epoch": 6.565189361572265e-05,
      "model_forward_time": 1.7417783737182617,
      "step": 68841
    },
    {
      "epoch": 6.565189361572265e-05,
      "step": 68841,
      "training_step_time": 7.770256280899048
    },
    {
      "epoch": 6.565284729003906e-05,
      "model_forward_time": 1.733652114868164,
      "step": 68842
    },
    {
      "epoch": 6.565284729003906e-05,
      "step": 68842,
      "training_step_time": 7.749570846557617
    },
    {
      "epoch": 6.565380096435547e-05,
      "model_forward_time": 1.7293674945831299,
      "step": 68843
    },
    {
      "epoch": 6.565380096435547e-05,
      "step": 68843,
      "training_step_time": 9.386845111846924
    },
    {
      "epoch": 6.565475463867187e-05,
      "model_forward_time": 2.9024009704589844,
      "step": 68844
    },
    {
      "epoch": 6.565475463867187e-05,
      "step": 68844,
      "training_step_time": 8.461896419525146
    },
    {
      "epoch": 6.565570831298828e-05,
      "model_forward_time": 1.7310614585876465,
      "step": 68845
    },
    {
      "epoch": 6.565570831298828e-05,
      "step": 68845,
      "training_step_time": 7.702802658081055
    },
    {
      "epoch": 6.565666198730469e-05,
      "model_forward_time": 1.730410099029541,
      "step": 68846
    },
    {
      "epoch": 6.565666198730469e-05,
      "step": 68846,
      "training_step_time": 7.769458770751953
    },
    {
      "epoch": 6.565761566162109e-05,
      "model_forward_time": 1.7385280132293701,
      "step": 68847
    },
    {
      "epoch": 6.565761566162109e-05,
      "step": 68847,
      "training_step_time": 7.640846252441406
    },
    {
      "epoch": 6.56585693359375e-05,
      "model_forward_time": 1.7343254089355469,
      "step": 68848
    },
    {
      "epoch": 6.56585693359375e-05,
      "step": 68848,
      "training_step_time": 7.815075874328613
    },
    {
      "epoch": 6.565952301025391e-05,
      "model_forward_time": 1.7313661575317383,
      "step": 68849
    },
    {
      "epoch": 6.565952301025391e-05,
      "step": 68849,
      "training_step_time": 7.544026613235474
    },
    {
      "epoch": 6.566047668457031e-05,
      "grad_norm": 0.09118731319904327,
      "learning_rate": 2.4265263400349497e-06,
      "loss": 0.0583,
      "step": 68850
    },
    {
      "epoch": 6.566047668457031e-05,
      "model_forward_time": 1.812983751296997,
      "step": 68850
    },
    {
      "dynamics_loss_avg": 0.047936751320958135,
      "epoch": 6.566047668457031e-05,
      "step": 68850
    },
    {
      "action_loss_avg": 0.0019908759219106286,
      "epoch": 6.566047668457031e-05,
      "step": 68850
    },
    {
      "epoch": 6.566047668457031e-05,
      "step": 68850,
      "training_step_time": 7.72533655166626
    },
    {
      "epoch": 6.566143035888672e-05,
      "model_forward_time": 1.7223970890045166,
      "step": 68851
    },
    {
      "epoch": 6.566143035888672e-05,
      "step": 68851,
      "training_step_time": 7.73593282699585
    },
    {
      "epoch": 6.566238403320313e-05,
      "model_forward_time": 1.7265088558197021,
      "step": 68852
    },
    {
      "epoch": 6.566238403320313e-05,
      "step": 68852,
      "training_step_time": 8.643825769424438
    },
    {
      "epoch": 6.566333770751953e-05,
      "model_forward_time": 2.906582832336426,
      "step": 68853
    },
    {
      "epoch": 6.566333770751953e-05,
      "step": 68853,
      "training_step_time": 9.689250469207764
    },
    {
      "epoch": 6.566429138183594e-05,
      "model_forward_time": 1.735849142074585,
      "step": 68854
    },
    {
      "epoch": 6.566429138183594e-05,
      "step": 68854,
      "training_step_time": 8.410412311553955
    },
    {
      "epoch": 6.566524505615235e-05,
      "model_forward_time": 1.3763327598571777,
      "step": 68855
    },
    {
      "epoch": 6.566524505615235e-05,
      "step": 68855,
      "training_step_time": 8.3632972240448
    },
    {
      "epoch": 6.566619873046874e-05,
      "model_forward_time": 1.7281396389007568,
      "step": 68856
    },
    {
      "epoch": 6.566619873046874e-05,
      "step": 68856,
      "training_step_time": 7.647359371185303
    },
    {
      "epoch": 6.566715240478516e-05,
      "model_forward_time": 1.7443335056304932,
      "step": 68857
    },
    {
      "epoch": 6.566715240478516e-05,
      "step": 68857,
      "training_step_time": 7.621168851852417
    },
    {
      "epoch": 6.566810607910157e-05,
      "model_forward_time": 1.7360460758209229,
      "step": 68858
    },
    {
      "epoch": 6.566810607910157e-05,
      "step": 68858,
      "training_step_time": 7.8205485343933105
    },
    {
      "epoch": 6.566905975341796e-05,
      "model_forward_time": 1.7980632781982422,
      "step": 68859
    },
    {
      "epoch": 6.566905975341796e-05,
      "step": 68859,
      "training_step_time": 8.551290035247803
    },
    {
      "epoch": 6.567001342773437e-05,
      "grad_norm": 0.09804509580135345,
      "learning_rate": 2.4251088395578214e-06,
      "loss": 0.0598,
      "step": 68860
    },
    {
      "epoch": 6.567001342773437e-05,
      "model_forward_time": 1.7379591464996338,
      "step": 68860
    },
    {
      "dynamics_loss_avg": 0.05419224482029676,
      "epoch": 6.567001342773437e-05,
      "step": 68860
    },
    {
      "action_loss_avg": 0.0016294590779580176,
      "epoch": 6.567001342773437e-05,
      "step": 68860
    },
    {
      "epoch": 6.567001342773437e-05,
      "step": 68860,
      "training_step_time": 9.271741151809692
    },
    {
      "epoch": 6.567096710205079e-05,
      "model_forward_time": 1.7372703552246094,
      "step": 68861
    },
    {
      "epoch": 6.567096710205079e-05,
      "step": 68861,
      "training_step_time": 10.364433765411377
    },
    {
      "epoch": 6.567192077636718e-05,
      "model_forward_time": 2.8993728160858154,
      "step": 68862
    },
    {
      "epoch": 6.567192077636718e-05,
      "step": 68862,
      "training_step_time": 7.735980987548828
    },
    {
      "epoch": 6.56728744506836e-05,
      "model_forward_time": 1.7290019989013672,
      "step": 68863
    },
    {
      "epoch": 6.56728744506836e-05,
      "step": 68863,
      "training_step_time": 7.692173480987549
    },
    {
      "epoch": 6.5673828125e-05,
      "model_forward_time": 1.7334957122802734,
      "step": 68864
    },
    {
      "epoch": 6.5673828125e-05,
      "step": 68864,
      "training_step_time": 7.713752269744873
    },
    {
      "epoch": 6.56747817993164e-05,
      "model_forward_time": 1.7338883876800537,
      "step": 68865
    },
    {
      "epoch": 6.56747817993164e-05,
      "step": 68865,
      "training_step_time": 8.314162731170654
    },
    {
      "epoch": 6.567573547363281e-05,
      "model_forward_time": 1.732283592224121,
      "step": 68866
    },
    {
      "epoch": 6.567573547363281e-05,
      "step": 68866,
      "training_step_time": 7.313978433609009
    },
    {
      "epoch": 6.567668914794922e-05,
      "model_forward_time": 1.7369823455810547,
      "step": 68867
    },
    {
      "epoch": 6.567668914794922e-05,
      "step": 68867,
      "training_step_time": 7.814535140991211
    },
    {
      "epoch": 6.567764282226562e-05,
      "model_forward_time": 1.7802608013153076,
      "step": 68868
    },
    {
      "epoch": 6.567764282226562e-05,
      "step": 68868,
      "training_step_time": 7.722655534744263
    },
    {
      "epoch": 6.567859649658203e-05,
      "model_forward_time": 1.7339165210723877,
      "step": 68869
    },
    {
      "epoch": 6.567859649658203e-05,
      "step": 68869,
      "training_step_time": 9.295101881027222
    },
    {
      "epoch": 6.567955017089844e-05,
      "grad_norm": 0.12293911725282669,
      "learning_rate": 2.4236916206669214e-06,
      "loss": 0.0558,
      "step": 68870
    },
    {
      "epoch": 6.567955017089844e-05,
      "model_forward_time": 1.7337393760681152,
      "step": 68870
    },
    {
      "dynamics_loss_avg": 0.054693544562906024,
      "epoch": 6.567955017089844e-05,
      "step": 68870
    },
    {
      "action_loss_avg": 0.0030380493961274626,
      "epoch": 6.567955017089844e-05,
      "step": 68870
    },
    {
      "epoch": 6.567955017089844e-05,
      "step": 68870,
      "training_step_time": 9.236913681030273
    },
    {
      "epoch": 6.568050384521484e-05,
      "model_forward_time": 2.8563268184661865,
      "step": 68871
    },
    {
      "epoch": 6.568050384521484e-05,
      "step": 68871,
      "training_step_time": 7.690536022186279
    },
    {
      "epoch": 6.568145751953125e-05,
      "model_forward_time": 1.7728471755981445,
      "step": 68872
    },
    {
      "epoch": 6.568145751953125e-05,
      "step": 68872,
      "training_step_time": 7.699276685714722
    },
    {
      "epoch": 6.568241119384766e-05,
      "model_forward_time": 1.7263741493225098,
      "step": 68873
    },
    {
      "epoch": 6.568241119384766e-05,
      "step": 68873,
      "training_step_time": 7.592311382293701
    },
    {
      "epoch": 6.568336486816406e-05,
      "model_forward_time": 1.7371792793273926,
      "step": 68874
    },
    {
      "epoch": 6.568336486816406e-05,
      "step": 68874,
      "training_step_time": 7.696624040603638
    },
    {
      "epoch": 6.568431854248047e-05,
      "model_forward_time": 1.7389178276062012,
      "step": 68875
    },
    {
      "epoch": 6.568431854248047e-05,
      "step": 68875,
      "training_step_time": 7.66656756401062
    },
    {
      "epoch": 6.568527221679688e-05,
      "model_forward_time": 1.7313950061798096,
      "step": 68876
    },
    {
      "epoch": 6.568527221679688e-05,
      "step": 68876,
      "training_step_time": 7.73083233833313
    },
    {
      "epoch": 6.568622589111328e-05,
      "model_forward_time": 1.761425495147705,
      "step": 68877
    },
    {
      "epoch": 6.568622589111328e-05,
      "step": 68877,
      "training_step_time": 9.270567655563354
    },
    {
      "epoch": 6.568717956542969e-05,
      "model_forward_time": 1.7596898078918457,
      "step": 68878
    },
    {
      "epoch": 6.568717956542969e-05,
      "step": 68878,
      "training_step_time": 7.620937824249268
    },
    {
      "epoch": 6.56881332397461e-05,
      "model_forward_time": 1.732234239578247,
      "step": 68879
    },
    {
      "epoch": 6.56881332397461e-05,
      "step": 68879,
      "training_step_time": 8.930756092071533
    },
    {
      "epoch": 6.56890869140625e-05,
      "grad_norm": 0.10644765943288803,
      "learning_rate": 2.4222746835172308e-06,
      "loss": 0.0568,
      "step": 68880
    },
    {
      "epoch": 6.56890869140625e-05,
      "model_forward_time": 2.8657584190368652,
      "step": 68880
    },
    {
      "dynamics_loss_avg": 0.06790910363197326,
      "epoch": 6.56890869140625e-05,
      "step": 68880
    },
    {
      "action_loss_avg": 0.001412474559037946,
      "epoch": 6.56890869140625e-05,
      "step": 68880
    },
    {
      "epoch": 6.56890869140625e-05,
      "step": 68880,
      "training_step_time": 8.3726224899292
    },
    {
      "epoch": 6.56900405883789e-05,
      "model_forward_time": 1.7532963752746582,
      "step": 68881
    },
    {
      "epoch": 6.56900405883789e-05,
      "step": 68881,
      "training_step_time": 8.99947738647461
    },
    {
      "epoch": 6.569099426269532e-05,
      "model_forward_time": 1.7414085865020752,
      "step": 68882
    },
    {
      "epoch": 6.569099426269532e-05,
      "step": 68882,
      "training_step_time": 7.449411153793335
    },
    {
      "epoch": 6.569194793701171e-05,
      "model_forward_time": 1.741065502166748,
      "step": 68883
    },
    {
      "epoch": 6.569194793701171e-05,
      "step": 68883,
      "training_step_time": 7.727860689163208
    },
    {
      "epoch": 6.569290161132813e-05,
      "model_forward_time": 1.3829638957977295,
      "step": 68884
    },
    {
      "epoch": 6.569290161132813e-05,
      "step": 68884,
      "training_step_time": 7.696490526199341
    },
    {
      "epoch": 6.569385528564454e-05,
      "model_forward_time": 1.7448434829711914,
      "step": 68885
    },
    {
      "epoch": 6.569385528564454e-05,
      "step": 68885,
      "training_step_time": 7.6015472412109375
    },
    {
      "epoch": 6.569480895996093e-05,
      "model_forward_time": 1.7586297988891602,
      "step": 68886
    },
    {
      "epoch": 6.569480895996093e-05,
      "step": 68886,
      "training_step_time": 7.6973443031311035
    },
    {
      "epoch": 6.569576263427734e-05,
      "model_forward_time": 1.8218634128570557,
      "step": 68887
    },
    {
      "epoch": 6.569576263427734e-05,
      "step": 68887,
      "training_step_time": 7.629534006118774
    },
    {
      "epoch": 6.569671630859375e-05,
      "model_forward_time": 1.7522506713867188,
      "step": 68888
    },
    {
      "epoch": 6.569671630859375e-05,
      "step": 68888,
      "training_step_time": 8.726130485534668
    },
    {
      "epoch": 6.569766998291015e-05,
      "model_forward_time": 1.751896858215332,
      "step": 68889
    },
    {
      "epoch": 6.569766998291015e-05,
      "step": 68889,
      "training_step_time": 7.692175388336182
    },
    {
      "epoch": 6.569862365722656e-05,
      "grad_norm": 0.0959269255399704,
      "learning_rate": 2.420858028263707e-06,
      "loss": 0.0564,
      "step": 68890
    },
    {
      "epoch": 6.569862365722656e-05,
      "model_forward_time": 2.9568288326263428,
      "step": 68890
    },
    {
      "dynamics_loss_avg": 0.0553082013502717,
      "epoch": 6.569862365722656e-05,
      "step": 68890
    },
    {
      "action_loss_avg": 0.0018231448775622994,
      "epoch": 6.569862365722656e-05,
      "step": 68890
    },
    {
      "epoch": 6.569862365722656e-05,
      "step": 68890,
      "training_step_time": 9.609944105148315
    },
    {
      "epoch": 6.569957733154297e-05,
      "model_forward_time": 1.7431035041809082,
      "step": 68891
    },
    {
      "epoch": 6.569957733154297e-05,
      "step": 68891,
      "training_step_time": 7.615415096282959
    },
    {
      "epoch": 6.570053100585937e-05,
      "model_forward_time": 1.7587852478027344,
      "step": 68892
    },
    {
      "epoch": 6.570053100585937e-05,
      "step": 68892,
      "training_step_time": 7.707547664642334
    },
    {
      "epoch": 6.570148468017578e-05,
      "model_forward_time": 1.7479617595672607,
      "step": 68893
    },
    {
      "epoch": 6.570148468017578e-05,
      "step": 68893,
      "training_step_time": 8.606810569763184
    },
    {
      "epoch": 6.570243835449219e-05,
      "model_forward_time": 1.7380993366241455,
      "step": 68894
    },
    {
      "epoch": 6.570243835449219e-05,
      "step": 68894,
      "training_step_time": 8.637304544448853
    },
    {
      "epoch": 6.570339202880859e-05,
      "model_forward_time": 1.772902250289917,
      "step": 68895
    },
    {
      "epoch": 6.570339202880859e-05,
      "step": 68895,
      "training_step_time": 7.689231634140015
    },
    {
      "epoch": 6.5704345703125e-05,
      "model_forward_time": 1.8203296661376953,
      "step": 68896
    },
    {
      "epoch": 6.5704345703125e-05,
      "step": 68896,
      "training_step_time": 8.098775625228882
    },
    {
      "epoch": 6.570529937744141e-05,
      "model_forward_time": 1.7612864971160889,
      "step": 68897
    },
    {
      "epoch": 6.570529937744141e-05,
      "step": 68897,
      "training_step_time": 7.677761554718018
    },
    {
      "epoch": 6.570625305175781e-05,
      "model_forward_time": 1.7409539222717285,
      "step": 68898
    },
    {
      "epoch": 6.570625305175781e-05,
      "step": 68898,
      "training_step_time": 7.697231292724609
    },
    {
      "epoch": 6.570720672607422e-05,
      "model_forward_time": 2.9437334537506104,
      "step": 68899
    },
    {
      "epoch": 6.570720672607422e-05,
      "step": 68899,
      "training_step_time": 7.785130262374878
    },
    {
      "epoch": 6.570816040039063e-05,
      "grad_norm": 0.09257397800683975,
      "learning_rate": 2.4194416550612685e-06,
      "loss": 0.0601,
      "step": 68900
    },
    {
      "epoch": 6.570816040039063e-05,
      "model_forward_time": 1.753678560256958,
      "step": 68900
    },
    {
      "dynamics_loss_avg": 0.05791090819984675,
      "epoch": 6.570816040039063e-05,
      "step": 68900
    },
    {
      "action_loss_avg": 0.0025826476980000735,
      "epoch": 6.570816040039063e-05,
      "step": 68900
    },
    {
      "epoch": 6.570816040039063e-05,
      "step": 68900,
      "training_step_time": 7.760568618774414
    },
    {
      "epoch": 6.570911407470703e-05,
      "model_forward_time": 1.759155035018921,
      "step": 68901
    },
    {
      "epoch": 6.570911407470703e-05,
      "step": 68901,
      "training_step_time": 7.676484823226929
    },
    {
      "epoch": 6.571006774902344e-05,
      "model_forward_time": 1.7478797435760498,
      "step": 68902
    },
    {
      "epoch": 6.571006774902344e-05,
      "step": 68902,
      "training_step_time": 7.684748888015747
    },
    {
      "epoch": 6.571102142333985e-05,
      "model_forward_time": 1.7588319778442383,
      "step": 68903
    },
    {
      "epoch": 6.571102142333985e-05,
      "step": 68903,
      "training_step_time": 7.775864601135254
    },
    {
      "epoch": 6.571197509765625e-05,
      "model_forward_time": 1.7839560508728027,
      "step": 68904
    },
    {
      "epoch": 6.571197509765625e-05,
      "step": 68904,
      "training_step_time": 7.836405038833618
    },
    {
      "epoch": 6.571292877197266e-05,
      "model_forward_time": 1.8181962966918945,
      "step": 68905
    },
    {
      "epoch": 6.571292877197266e-05,
      "step": 68905,
      "training_step_time": 7.6648828983306885
    },
    {
      "epoch": 6.571388244628907e-05,
      "model_forward_time": 1.762629747390747,
      "step": 68906
    },
    {
      "epoch": 6.571388244628907e-05,
      "step": 68906,
      "training_step_time": 8.16903305053711
    },
    {
      "epoch": 6.571483612060547e-05,
      "model_forward_time": 1.7515077590942383,
      "step": 68907
    },
    {
      "epoch": 6.571483612060547e-05,
      "step": 68907,
      "training_step_time": 7.735217809677124
    },
    {
      "epoch": 6.571578979492188e-05,
      "model_forward_time": 2.9238998889923096,
      "step": 68908
    },
    {
      "epoch": 6.571578979492188e-05,
      "step": 68908,
      "training_step_time": 7.7766687870025635
    },
    {
      "epoch": 6.571674346923829e-05,
      "model_forward_time": 1.7632241249084473,
      "step": 68909
    },
    {
      "epoch": 6.571674346923829e-05,
      "step": 68909,
      "training_step_time": 7.68455171585083
    },
    {
      "epoch": 6.571769714355468e-05,
      "grad_norm": 0.09983430802822113,
      "learning_rate": 2.418025564064812e-06,
      "loss": 0.0572,
      "step": 68910
    },
    {
      "epoch": 6.571769714355468e-05,
      "model_forward_time": 1.747164011001587,
      "step": 68910
    },
    {
      "dynamics_loss_avg": 0.05373339299112558,
      "epoch": 6.571769714355468e-05,
      "step": 68910
    },
    {
      "action_loss_avg": 0.00149173850659281,
      "epoch": 6.571769714355468e-05,
      "step": 68910
    },
    {
      "epoch": 6.571769714355468e-05,
      "step": 68910,
      "training_step_time": 7.725362777709961
    },
    {
      "epoch": 6.57186508178711e-05,
      "model_forward_time": 1.7504973411560059,
      "step": 68911
    },
    {
      "epoch": 6.57186508178711e-05,
      "step": 68911,
      "training_step_time": 7.720057010650635
    },
    {
      "epoch": 6.57196044921875e-05,
      "model_forward_time": 1.7579376697540283,
      "step": 68912
    },
    {
      "epoch": 6.57196044921875e-05,
      "step": 68912,
      "training_step_time": 8.664282321929932
    },
    {
      "epoch": 6.57205581665039e-05,
      "model_forward_time": 1.752406358718872,
      "step": 68913
    },
    {
      "epoch": 6.57205581665039e-05,
      "step": 68913,
      "training_step_time": 7.791250944137573
    },
    {
      "epoch": 6.572151184082031e-05,
      "model_forward_time": 1.7807762622833252,
      "step": 68914
    },
    {
      "epoch": 6.572151184082031e-05,
      "step": 68914,
      "training_step_time": 7.529183387756348
    },
    {
      "epoch": 6.572246551513672e-05,
      "model_forward_time": 1.7495265007019043,
      "step": 68915
    },
    {
      "epoch": 6.572246551513672e-05,
      "step": 68915,
      "training_step_time": 7.668940544128418
    },
    {
      "epoch": 6.572341918945312e-05,
      "model_forward_time": 1.3850157260894775,
      "step": 68916
    },
    {
      "epoch": 6.572341918945312e-05,
      "step": 68916,
      "training_step_time": 7.810094118118286
    },
    {
      "epoch": 6.572437286376953e-05,
      "model_forward_time": 2.9497880935668945,
      "step": 68917
    },
    {
      "epoch": 6.572437286376953e-05,
      "step": 68917,
      "training_step_time": 8.374576091766357
    },
    {
      "epoch": 6.572532653808594e-05,
      "model_forward_time": 1.7534220218658447,
      "step": 68918
    },
    {
      "epoch": 6.572532653808594e-05,
      "step": 68918,
      "training_step_time": 7.668094158172607
    },
    {
      "epoch": 6.572628021240234e-05,
      "model_forward_time": 1.7494137287139893,
      "step": 68919
    },
    {
      "epoch": 6.572628021240234e-05,
      "step": 68919,
      "training_step_time": 7.70296573638916
    },
    {
      "epoch": 6.572723388671875e-05,
      "grad_norm": 0.12482527643442154,
      "learning_rate": 2.4166097554291977e-06,
      "loss": 0.0581,
      "step": 68920
    },
    {
      "epoch": 6.572723388671875e-05,
      "model_forward_time": 1.7499291896820068,
      "step": 68920
    },
    {
      "dynamics_loss_avg": 0.06176981097087264,
      "epoch": 6.572723388671875e-05,
      "step": 68920
    },
    {
      "action_loss_avg": 0.001908189948881045,
      "epoch": 6.572723388671875e-05,
      "step": 68920
    },
    {
      "epoch": 6.572723388671875e-05,
      "step": 68920,
      "training_step_time": 8.29995846748352
    },
    {
      "epoch": 6.572818756103516e-05,
      "model_forward_time": 1.759584665298462,
      "step": 68921
    },
    {
      "epoch": 6.572818756103516e-05,
      "step": 68921,
      "training_step_time": 8.505757808685303
    },
    {
      "epoch": 6.572914123535156e-05,
      "model_forward_time": 1.7476835250854492,
      "step": 68922
    },
    {
      "epoch": 6.572914123535156e-05,
      "step": 68922,
      "training_step_time": 7.7407386302948
    },
    {
      "epoch": 6.573009490966797e-05,
      "model_forward_time": 1.7587227821350098,
      "step": 68923
    },
    {
      "epoch": 6.573009490966797e-05,
      "step": 68923,
      "training_step_time": 9.623266696929932
    },
    {
      "epoch": 6.573104858398438e-05,
      "model_forward_time": 1.7413995265960693,
      "step": 68924
    },
    {
      "epoch": 6.573104858398438e-05,
      "step": 68924,
      "training_step_time": 7.709278583526611
    },
    {
      "epoch": 6.573200225830078e-05,
      "model_forward_time": 1.7502391338348389,
      "step": 68925
    },
    {
      "epoch": 6.573200225830078e-05,
      "step": 68925,
      "training_step_time": 8.541494369506836
    },
    {
      "epoch": 6.573295593261719e-05,
      "model_forward_time": 2.883164167404175,
      "step": 68926
    },
    {
      "epoch": 6.573295593261719e-05,
      "step": 68926,
      "training_step_time": 7.802582263946533
    },
    {
      "epoch": 6.57339096069336e-05,
      "model_forward_time": 1.7609889507293701,
      "step": 68927
    },
    {
      "epoch": 6.57339096069336e-05,
      "step": 68927,
      "training_step_time": 9.47117304801941
    },
    {
      "epoch": 6.573486328125e-05,
      "model_forward_time": 1.7705678939819336,
      "step": 68928
    },
    {
      "epoch": 6.573486328125e-05,
      "step": 68928,
      "training_step_time": 7.730409383773804
    },
    {
      "epoch": 6.573581695556641e-05,
      "model_forward_time": 1.756331205368042,
      "step": 68929
    },
    {
      "epoch": 6.573581695556641e-05,
      "step": 68929,
      "training_step_time": 7.77367091178894
    },
    {
      "epoch": 6.573677062988282e-05,
      "grad_norm": 0.10392522066831589,
      "learning_rate": 2.4151942293092584e-06,
      "loss": 0.0577,
      "step": 68930
    },
    {
      "epoch": 6.573677062988282e-05,
      "model_forward_time": 1.7504775524139404,
      "step": 68930
    },
    {
      "dynamics_loss_avg": 0.07586899884045124,
      "epoch": 6.573677062988282e-05,
      "step": 68930
    },
    {
      "action_loss_avg": 0.0022924342280020936,
      "epoch": 6.573677062988282e-05,
      "step": 68930
    },
    {
      "epoch": 6.573677062988282e-05,
      "step": 68930,
      "training_step_time": 7.675612688064575
    },
    {
      "epoch": 6.573772430419922e-05,
      "model_forward_time": 1.7586066722869873,
      "step": 68931
    },
    {
      "epoch": 6.573772430419922e-05,
      "step": 68931,
      "training_step_time": 7.8019890785217285
    },
    {
      "epoch": 6.573867797851563e-05,
      "model_forward_time": 1.743628978729248,
      "step": 68932
    },
    {
      "epoch": 6.573867797851563e-05,
      "step": 68932,
      "training_step_time": 7.752458810806274
    },
    {
      "epoch": 6.573963165283204e-05,
      "model_forward_time": 1.8252036571502686,
      "step": 68933
    },
    {
      "epoch": 6.573963165283204e-05,
      "step": 68933,
      "training_step_time": 7.6827991008758545
    },
    {
      "epoch": 6.574058532714844e-05,
      "model_forward_time": 1.7515406608581543,
      "step": 68934
    },
    {
      "epoch": 6.574058532714844e-05,
      "step": 68934,
      "training_step_time": 9.408294439315796
    },
    {
      "epoch": 6.574153900146485e-05,
      "model_forward_time": 2.96583890914917,
      "step": 68935
    },
    {
      "epoch": 6.574153900146485e-05,
      "step": 68935,
      "training_step_time": 10.404663324356079
    },
    {
      "epoch": 6.574249267578126e-05,
      "model_forward_time": 1.7539012432098389,
      "step": 68936
    },
    {
      "epoch": 6.574249267578126e-05,
      "step": 68936,
      "training_step_time": 9.656137704849243
    },
    {
      "epoch": 6.574344635009765e-05,
      "model_forward_time": 1.7506227493286133,
      "step": 68937
    },
    {
      "epoch": 6.574344635009765e-05,
      "step": 68937,
      "training_step_time": 8.58479356765747
    },
    {
      "epoch": 6.574440002441406e-05,
      "model_forward_time": 1.7573046684265137,
      "step": 68938
    },
    {
      "epoch": 6.574440002441406e-05,
      "step": 68938,
      "training_step_time": 7.637852430343628
    },
    {
      "epoch": 6.574535369873046e-05,
      "model_forward_time": 1.7484288215637207,
      "step": 68939
    },
    {
      "epoch": 6.574535369873046e-05,
      "step": 68939,
      "training_step_time": 8.315509557723999
    },
    {
      "epoch": 6.574630737304687e-05,
      "grad_norm": 0.10102315992116928,
      "learning_rate": 2.4137789858597897e-06,
      "loss": 0.0575,
      "step": 68940
    },
    {
      "epoch": 6.574630737304687e-05,
      "model_forward_time": 1.746311902999878,
      "step": 68940
    },
    {
      "dynamics_loss_avg": 0.050730387587100266,
      "epoch": 6.574630737304687e-05,
      "step": 68940
    },
    {
      "action_loss_avg": 0.0014115544327069074,
      "epoch": 6.574630737304687e-05,
      "step": 68940
    },
    {
      "epoch": 6.574630737304687e-05,
      "step": 68940,
      "training_step_time": 7.804758310317993
    },
    {
      "epoch": 6.574726104736328e-05,
      "model_forward_time": 1.776759386062622,
      "step": 68941
    },
    {
      "epoch": 6.574726104736328e-05,
      "step": 68941,
      "training_step_time": 7.781540155410767
    },
    {
      "epoch": 6.574821472167968e-05,
      "model_forward_time": 1.8083117008209229,
      "step": 68942
    },
    {
      "epoch": 6.574821472167968e-05,
      "step": 68942,
      "training_step_time": 7.612843036651611
    },
    {
      "epoch": 6.574916839599609e-05,
      "model_forward_time": 1.7572152614593506,
      "step": 68943
    },
    {
      "epoch": 6.574916839599609e-05,
      "step": 68943,
      "training_step_time": 8.75367259979248
    },
    {
      "epoch": 6.57501220703125e-05,
      "model_forward_time": 2.258896827697754,
      "step": 68944
    },
    {
      "epoch": 6.57501220703125e-05,
      "step": 68944,
      "training_step_time": 9.057929515838623
    },
    {
      "epoch": 6.57510757446289e-05,
      "model_forward_time": 1.7091069221496582,
      "step": 68945
    },
    {
      "epoch": 6.57510757446289e-05,
      "step": 68945,
      "training_step_time": 8.317692756652832
    },
    {
      "epoch": 6.575202941894531e-05,
      "model_forward_time": 1.7092559337615967,
      "step": 68946
    },
    {
      "epoch": 6.575202941894531e-05,
      "step": 68946,
      "training_step_time": 7.719065427780151
    },
    {
      "epoch": 6.575298309326172e-05,
      "model_forward_time": 1.7055225372314453,
      "step": 68947
    },
    {
      "epoch": 6.575298309326172e-05,
      "step": 68947,
      "training_step_time": 7.765692472457886
    },
    {
      "epoch": 6.575393676757812e-05,
      "model_forward_time": 1.7096648216247559,
      "step": 68948
    },
    {
      "epoch": 6.575393676757812e-05,
      "step": 68948,
      "training_step_time": 7.67823600769043
    },
    {
      "epoch": 6.575489044189453e-05,
      "model_forward_time": 1.7049353122711182,
      "step": 68949
    },
    {
      "epoch": 6.575489044189453e-05,
      "step": 68949,
      "training_step_time": 7.706800222396851
    },
    {
      "epoch": 6.575584411621094e-05,
      "grad_norm": 0.0860515758395195,
      "learning_rate": 2.412364025235565e-06,
      "loss": 0.0605,
      "step": 68950
    },
    {
      "epoch": 6.575584411621094e-05,
      "model_forward_time": 1.705552577972412,
      "step": 68950
    },
    {
      "dynamics_loss_avg": 0.05360077768564224,
      "epoch": 6.575584411621094e-05,
      "step": 68950
    },
    {
      "action_loss_avg": 0.0009847404202446342,
      "epoch": 6.575584411621094e-05,
      "step": 68950
    },
    {
      "epoch": 6.575584411621094e-05,
      "step": 68950,
      "training_step_time": 7.72291374206543
    },
    {
      "epoch": 6.575679779052734e-05,
      "model_forward_time": 1.7052981853485107,
      "step": 68951
    },
    {
      "epoch": 6.575679779052734e-05,
      "step": 68951,
      "training_step_time": 7.672194957733154
    },
    {
      "epoch": 6.575775146484375e-05,
      "model_forward_time": 1.7081105709075928,
      "step": 68952
    },
    {
      "epoch": 6.575775146484375e-05,
      "step": 68952,
      "training_step_time": 8.82021188735962
    },
    {
      "epoch": 6.575870513916016e-05,
      "model_forward_time": 1.7182199954986572,
      "step": 68953
    },
    {
      "epoch": 6.575870513916016e-05,
      "step": 68953,
      "training_step_time": 9.798739671707153
    },
    {
      "epoch": 6.575965881347656e-05,
      "model_forward_time": 1.338374376296997,
      "step": 68954
    },
    {
      "epoch": 6.575965881347656e-05,
      "step": 68954,
      "training_step_time": 7.541169881820679
    },
    {
      "epoch": 6.576061248779297e-05,
      "model_forward_time": 1.706758737564087,
      "step": 68955
    },
    {
      "epoch": 6.576061248779297e-05,
      "step": 68955,
      "training_step_time": 8.729295253753662
    },
    {
      "epoch": 6.576156616210938e-05,
      "model_forward_time": 1.7079505920410156,
      "step": 68956
    },
    {
      "epoch": 6.576156616210938e-05,
      "step": 68956,
      "training_step_time": 7.71394944190979
    },
    {
      "epoch": 6.576251983642578e-05,
      "model_forward_time": 1.7094347476959229,
      "step": 68957
    },
    {
      "epoch": 6.576251983642578e-05,
      "step": 68957,
      "training_step_time": 8.396708250045776
    },
    {
      "epoch": 6.576347351074219e-05,
      "model_forward_time": 1.7074565887451172,
      "step": 68958
    },
    {
      "epoch": 6.576347351074219e-05,
      "step": 68958,
      "training_step_time": 7.611392974853516
    },
    {
      "epoch": 6.57644271850586e-05,
      "model_forward_time": 1.7055552005767822,
      "step": 68959
    },
    {
      "epoch": 6.57644271850586e-05,
      "step": 68959,
      "training_step_time": 8.913378477096558
    },
    {
      "epoch": 6.5765380859375e-05,
      "grad_norm": 0.09033890068531036,
      "learning_rate": 2.410949347591318e-06,
      "loss": 0.0603,
      "step": 68960
    },
    {
      "epoch": 6.5765380859375e-05,
      "model_forward_time": 1.7062442302703857,
      "step": 68960
    },
    {
      "dynamics_loss_avg": 0.06463035698980094,
      "epoch": 6.5765380859375e-05,
      "step": 68960
    },
    {
      "action_loss_avg": 0.0018122304609278217,
      "epoch": 6.5765380859375e-05,
      "step": 68960
    },
    {
      "epoch": 6.5765380859375e-05,
      "step": 68960,
      "training_step_time": 7.654997110366821
    },
    {
      "epoch": 6.57663345336914e-05,
      "model_forward_time": 1.7087414264678955,
      "step": 68961
    },
    {
      "epoch": 6.57663345336914e-05,
      "step": 68961,
      "training_step_time": 8.465406894683838
    },
    {
      "epoch": 6.576728820800782e-05,
      "model_forward_time": 1.756453037261963,
      "step": 68962
    },
    {
      "epoch": 6.576728820800782e-05,
      "step": 68962,
      "training_step_time": 8.610594511032104
    },
    {
      "epoch": 6.576824188232421e-05,
      "model_forward_time": 2.2709615230560303,
      "step": 68963
    },
    {
      "epoch": 6.576824188232421e-05,
      "step": 68963,
      "training_step_time": 8.295955896377563
    },
    {
      "epoch": 6.576919555664062e-05,
      "model_forward_time": 1.7189955711364746,
      "step": 68964
    },
    {
      "epoch": 6.576919555664062e-05,
      "step": 68964,
      "training_step_time": 8.583034038543701
    },
    {
      "epoch": 6.577014923095703e-05,
      "model_forward_time": 1.7098822593688965,
      "step": 68965
    },
    {
      "epoch": 6.577014923095703e-05,
      "step": 68965,
      "training_step_time": 8.735537767410278
    },
    {
      "epoch": 6.577110290527343e-05,
      "model_forward_time": 1.7157423496246338,
      "step": 68966
    },
    {
      "epoch": 6.577110290527343e-05,
      "step": 68966,
      "training_step_time": 7.653636455535889
    },
    {
      "epoch": 6.577205657958984e-05,
      "model_forward_time": 1.7104923725128174,
      "step": 68967
    },
    {
      "epoch": 6.577205657958984e-05,
      "step": 68967,
      "training_step_time": 8.57120394706726
    },
    {
      "epoch": 6.577301025390625e-05,
      "model_forward_time": 1.7103595733642578,
      "step": 68968
    },
    {
      "epoch": 6.577301025390625e-05,
      "step": 68968,
      "training_step_time": 7.784111738204956
    },
    {
      "epoch": 6.577396392822265e-05,
      "model_forward_time": 1.7126381397247314,
      "step": 68969
    },
    {
      "epoch": 6.577396392822265e-05,
      "step": 68969,
      "training_step_time": 7.764814376831055
    },
    {
      "epoch": 6.577491760253906e-05,
      "grad_norm": 0.0953606516122818,
      "learning_rate": 2.409534953081758e-06,
      "loss": 0.0552,
      "step": 68970
    },
    {
      "epoch": 6.577491760253906e-05,
      "model_forward_time": 1.7044885158538818,
      "step": 68970
    },
    {
      "dynamics_loss_avg": 0.054566537216305736,
      "epoch": 6.577491760253906e-05,
      "step": 68970
    },
    {
      "action_loss_avg": 0.0016243429330643267,
      "epoch": 6.577491760253906e-05,
      "step": 68970
    },
    {
      "epoch": 6.577491760253906e-05,
      "step": 68970,
      "training_step_time": 7.685533285140991
    },
    {
      "epoch": 6.577587127685547e-05,
      "model_forward_time": 1.7558586597442627,
      "step": 68971
    },
    {
      "epoch": 6.577587127685547e-05,
      "step": 68971,
      "training_step_time": 7.782063007354736
    },
    {
      "epoch": 6.577682495117187e-05,
      "model_forward_time": 2.2723395824432373,
      "step": 68972
    },
    {
      "epoch": 6.577682495117187e-05,
      "step": 68972,
      "training_step_time": 8.447594165802002
    },
    {
      "epoch": 6.577777862548828e-05,
      "model_forward_time": 1.7106072902679443,
      "step": 68973
    },
    {
      "epoch": 6.577777862548828e-05,
      "step": 68973,
      "training_step_time": 8.837300539016724
    },
    {
      "epoch": 6.577873229980469e-05,
      "model_forward_time": 1.7050468921661377,
      "step": 68974
    },
    {
      "epoch": 6.577873229980469e-05,
      "step": 68974,
      "training_step_time": 7.824024438858032
    },
    {
      "epoch": 6.577968597412109e-05,
      "model_forward_time": 1.7195658683776855,
      "step": 68975
    },
    {
      "epoch": 6.577968597412109e-05,
      "step": 68975,
      "training_step_time": 7.706888198852539
    },
    {
      "epoch": 6.57806396484375e-05,
      "model_forward_time": 1.7173988819122314,
      "step": 68976
    },
    {
      "epoch": 6.57806396484375e-05,
      "step": 68976,
      "training_step_time": 8.271971702575684
    },
    {
      "epoch": 6.578159332275391e-05,
      "model_forward_time": 1.7214317321777344,
      "step": 68977
    },
    {
      "epoch": 6.578159332275391e-05,
      "step": 68977,
      "training_step_time": 7.799885034561157
    },
    {
      "epoch": 6.578254699707031e-05,
      "model_forward_time": 1.7217919826507568,
      "step": 68978
    },
    {
      "epoch": 6.578254699707031e-05,
      "step": 68978,
      "training_step_time": 7.790311574935913
    },
    {
      "epoch": 6.578350067138672e-05,
      "model_forward_time": 1.7177801132202148,
      "step": 68979
    },
    {
      "epoch": 6.578350067138672e-05,
      "step": 68979,
      "training_step_time": 8.690251588821411
    },
    {
      "epoch": 6.578445434570313e-05,
      "grad_norm": 0.11884184181690216,
      "learning_rate": 2.4081208418615604e-06,
      "loss": 0.0614,
      "step": 68980
    },
    {
      "epoch": 6.578445434570313e-05,
      "model_forward_time": 1.7514173984527588,
      "step": 68980
    },
    {
      "dynamics_loss_avg": 0.0526893075555563,
      "epoch": 6.578445434570313e-05,
      "step": 68980
    },
    {
      "action_loss_avg": 0.002668936541886069,
      "epoch": 6.578445434570313e-05,
      "step": 68980
    },
    {
      "epoch": 6.578445434570313e-05,
      "step": 68980,
      "training_step_time": 8.702566385269165
    },
    {
      "epoch": 6.578540802001953e-05,
      "model_forward_time": 2.301607608795166,
      "step": 68981
    },
    {
      "epoch": 6.578540802001953e-05,
      "step": 68981,
      "training_step_time": 8.749694585800171
    },
    {
      "epoch": 6.578636169433594e-05,
      "model_forward_time": 1.7148211002349854,
      "step": 68982
    },
    {
      "epoch": 6.578636169433594e-05,
      "step": 68982,
      "training_step_time": 8.41642451286316
    },
    {
      "epoch": 6.578731536865235e-05,
      "model_forward_time": 1.7187092304229736,
      "step": 68983
    },
    {
      "epoch": 6.578731536865235e-05,
      "step": 68983,
      "training_step_time": 8.345680952072144
    },
    {
      "epoch": 6.578826904296874e-05,
      "model_forward_time": 1.7174570560455322,
      "step": 68984
    },
    {
      "epoch": 6.578826904296874e-05,
      "step": 68984,
      "training_step_time": 7.760146141052246
    },
    {
      "epoch": 6.578922271728516e-05,
      "model_forward_time": 1.3481249809265137,
      "step": 68985
    },
    {
      "epoch": 6.578922271728516e-05,
      "step": 68985,
      "training_step_time": 7.727707862854004
    },
    {
      "epoch": 6.579017639160157e-05,
      "model_forward_time": 1.7192256450653076,
      "step": 68986
    },
    {
      "epoch": 6.579017639160157e-05,
      "step": 68986,
      "training_step_time": 7.7979795932769775
    },
    {
      "epoch": 6.579113006591796e-05,
      "model_forward_time": 1.7078452110290527,
      "step": 68987
    },
    {
      "epoch": 6.579113006591796e-05,
      "step": 68987,
      "training_step_time": 7.716438293457031
    },
    {
      "epoch": 6.579208374023437e-05,
      "model_forward_time": 1.7009379863739014,
      "step": 68988
    },
    {
      "epoch": 6.579208374023437e-05,
      "step": 68988,
      "training_step_time": 8.312883377075195
    },
    {
      "epoch": 6.579303741455079e-05,
      "model_forward_time": 1.721492052078247,
      "step": 68989
    },
    {
      "epoch": 6.579303741455079e-05,
      "step": 68989,
      "training_step_time": 9.35940432548523
    },
    {
      "epoch": 6.579399108886718e-05,
      "grad_norm": 0.1854199767112732,
      "learning_rate": 2.4067070140853725e-06,
      "loss": 0.061,
      "step": 68990
    },
    {
      "epoch": 6.579399108886718e-05,
      "model_forward_time": 1.704618215560913,
      "step": 68990
    },
    {
      "dynamics_loss_avg": 0.051499956659972666,
      "epoch": 6.579399108886718e-05,
      "step": 68990
    },
    {
      "action_loss_avg": 0.001758806104771793,
      "epoch": 6.579399108886718e-05,
      "step": 68990
    },
    {
      "epoch": 6.579399108886718e-05,
      "step": 68990,
      "training_step_time": 9.158221960067749
    },
    {
      "epoch": 6.57949447631836e-05,
      "model_forward_time": 1.7120041847229004,
      "step": 68991
    },
    {
      "epoch": 6.57949447631836e-05,
      "step": 68991,
      "training_step_time": 8.82740592956543
    },
    {
      "epoch": 6.57958984375e-05,
      "model_forward_time": 1.7079002857208252,
      "step": 68992
    },
    {
      "epoch": 6.57958984375e-05,
      "step": 68992,
      "training_step_time": 7.902997016906738
    },
    {
      "epoch": 6.57968521118164e-05,
      "model_forward_time": 1.7081305980682373,
      "step": 68993
    },
    {
      "epoch": 6.57968521118164e-05,
      "step": 68993,
      "training_step_time": 8.463773250579834
    },
    {
      "epoch": 6.579780578613281e-05,
      "model_forward_time": 1.7085249423980713,
      "step": 68994
    },
    {
      "epoch": 6.579780578613281e-05,
      "step": 68994,
      "training_step_time": 7.699158191680908
    },
    {
      "epoch": 6.579875946044922e-05,
      "model_forward_time": 1.706446886062622,
      "step": 68995
    },
    {
      "epoch": 6.579875946044922e-05,
      "step": 68995,
      "training_step_time": 7.694560527801514
    },
    {
      "epoch": 6.579971313476562e-05,
      "model_forward_time": 1.7097725868225098,
      "step": 68996
    },
    {
      "epoch": 6.579971313476562e-05,
      "step": 68996,
      "training_step_time": 8.529731035232544
    },
    {
      "epoch": 6.580066680908203e-05,
      "model_forward_time": 1.7147610187530518,
      "step": 68997
    },
    {
      "epoch": 6.580066680908203e-05,
      "step": 68997,
      "training_step_time": 7.753672122955322
    },
    {
      "epoch": 6.580162048339844e-05,
      "model_forward_time": 1.7388052940368652,
      "step": 68998
    },
    {
      "epoch": 6.580162048339844e-05,
      "step": 68998,
      "training_step_time": 10.213082075119019
    },
    {
      "epoch": 6.580257415771484e-05,
      "model_forward_time": 1.715794563293457,
      "step": 68999
    },
    {
      "epoch": 6.580257415771484e-05,
      "step": 68999,
      "training_step_time": 9.597864389419556
    },
    {
      "epoch": 6.580352783203125e-05,
      "grad_norm": 0.11682091653347015,
      "learning_rate": 2.405293469907805e-06,
      "loss": 0.0565,
      "step": 69000
    },
    {
      "epoch": 6.580352783203125e-05,
      "model_forward_time": 1.7514328956604004,
      "step": 69000
    },
    {
      "dynamics_loss_avg": 0.05539444461464882,
      "epoch": 6.580352783203125e-05,
      "step": 69000
    },
    {
      "action_loss_avg": 0.0012912557052914054,
      "epoch": 6.580352783203125e-05,
      "step": 69000
    },
    {
      "epoch": 6.580352783203125e-05,
      "step": 69000,
      "training_step_time": 7.235509157180786
    },
    {
      "epoch": 6.580448150634766e-05,
      "model_forward_time": 1.731024980545044,
      "step": 69001
    },
    {
      "epoch": 6.580448150634766e-05,
      "step": 69001,
      "training_step_time": 7.839504241943359
    },
    {
      "epoch": 6.580543518066406e-05,
      "model_forward_time": 1.7322068214416504,
      "step": 69002
    },
    {
      "epoch": 6.580543518066406e-05,
      "step": 69002,
      "training_step_time": 7.712489604949951
    },
    {
      "epoch": 6.580638885498047e-05,
      "model_forward_time": 1.727679967880249,
      "step": 69003
    },
    {
      "epoch": 6.580638885498047e-05,
      "step": 69003,
      "training_step_time": 9.355520963668823
    },
    {
      "epoch": 6.580734252929688e-05,
      "model_forward_time": 1.7253868579864502,
      "step": 69004
    },
    {
      "epoch": 6.580734252929688e-05,
      "step": 69004,
      "training_step_time": 8.536600112915039
    },
    {
      "epoch": 6.580829620361328e-05,
      "model_forward_time": 1.7329785823822021,
      "step": 69005
    },
    {
      "epoch": 6.580829620361328e-05,
      "step": 69005,
      "training_step_time": 7.707582950592041
    },
    {
      "epoch": 6.580924987792969e-05,
      "model_forward_time": 1.7324504852294922,
      "step": 69006
    },
    {
      "epoch": 6.580924987792969e-05,
      "step": 69006,
      "training_step_time": 10.331923961639404
    },
    {
      "epoch": 6.58102035522461e-05,
      "model_forward_time": 1.7266311645507812,
      "step": 69007
    },
    {
      "epoch": 6.58102035522461e-05,
      "step": 69007,
      "training_step_time": 9.256472110748291
    },
    {
      "epoch": 6.58111572265625e-05,
      "model_forward_time": 2.8163275718688965,
      "step": 69008
    },
    {
      "epoch": 6.58111572265625e-05,
      "step": 69008,
      "training_step_time": 9.991772890090942
    },
    {
      "epoch": 6.58121109008789e-05,
      "model_forward_time": 1.7199149131774902,
      "step": 69009
    },
    {
      "epoch": 6.58121109008789e-05,
      "step": 69009,
      "training_step_time": 7.64865255355835
    },
    {
      "epoch": 6.581306457519532e-05,
      "grad_norm": 0.1014939472079277,
      "learning_rate": 2.4038802094834446e-06,
      "loss": 0.0593,
      "step": 69010
    },
    {
      "epoch": 6.581306457519532e-05,
      "model_forward_time": 1.719567060470581,
      "step": 69010
    },
    {
      "dynamics_loss_avg": 0.05037106145173311,
      "epoch": 6.581306457519532e-05,
      "step": 69010
    },
    {
      "action_loss_avg": 0.0013046318432316183,
      "epoch": 6.581306457519532e-05,
      "step": 69010
    },
    {
      "epoch": 6.581306457519532e-05,
      "step": 69010,
      "training_step_time": 7.839110612869263
    },
    {
      "epoch": 6.581401824951171e-05,
      "model_forward_time": 1.7301487922668457,
      "step": 69011
    },
    {
      "epoch": 6.581401824951171e-05,
      "step": 69011,
      "training_step_time": 8.628996133804321
    },
    {
      "epoch": 6.581497192382813e-05,
      "model_forward_time": 1.7229864597320557,
      "step": 69012
    },
    {
      "epoch": 6.581497192382813e-05,
      "step": 69012,
      "training_step_time": 8.788257360458374
    },
    {
      "epoch": 6.581592559814454e-05,
      "model_forward_time": 1.7267417907714844,
      "step": 69013
    },
    {
      "epoch": 6.581592559814454e-05,
      "step": 69013,
      "training_step_time": 7.523615598678589
    },
    {
      "epoch": 6.581687927246093e-05,
      "model_forward_time": 1.714324712753296,
      "step": 69014
    },
    {
      "epoch": 6.581687927246093e-05,
      "step": 69014,
      "training_step_time": 7.706078290939331
    },
    {
      "epoch": 6.581783294677734e-05,
      "model_forward_time": 1.7201697826385498,
      "step": 69015
    },
    {
      "epoch": 6.581783294677734e-05,
      "step": 69015,
      "training_step_time": 8.830881595611572
    },
    {
      "epoch": 6.581878662109376e-05,
      "model_forward_time": 1.3372371196746826,
      "step": 69016
    },
    {
      "epoch": 6.581878662109376e-05,
      "step": 69016,
      "training_step_time": 9.312740564346313
    },
    {
      "epoch": 6.581974029541015e-05,
      "model_forward_time": 1.7705163955688477,
      "step": 69017
    },
    {
      "epoch": 6.581974029541015e-05,
      "step": 69017,
      "training_step_time": 7.688176155090332
    },
    {
      "epoch": 6.582069396972656e-05,
      "model_forward_time": 2.278092622756958,
      "step": 69018
    },
    {
      "epoch": 6.582069396972656e-05,
      "step": 69018,
      "training_step_time": 8.153778791427612
    },
    {
      "epoch": 6.582164764404297e-05,
      "model_forward_time": 1.711923599243164,
      "step": 69019
    },
    {
      "epoch": 6.582164764404297e-05,
      "step": 69019,
      "training_step_time": 7.763198137283325
    },
    {
      "epoch": 6.582260131835937e-05,
      "grad_norm": 0.10174303501844406,
      "learning_rate": 2.40246723296684e-06,
      "loss": 0.0586,
      "step": 69020
    },
    {
      "epoch": 6.582260131835937e-05,
      "model_forward_time": 1.7127799987792969,
      "step": 69020
    },
    {
      "dynamics_loss_avg": 0.05330334380269051,
      "epoch": 6.582260131835937e-05,
      "step": 69020
    },
    {
      "action_loss_avg": 0.0013914796290919185,
      "epoch": 6.582260131835937e-05,
      "step": 69020
    },
    {
      "epoch": 6.582260131835937e-05,
      "step": 69020,
      "training_step_time": 7.645464897155762
    },
    {
      "epoch": 6.582355499267578e-05,
      "model_forward_time": 1.7076315879821777,
      "step": 69021
    },
    {
      "epoch": 6.582355499267578e-05,
      "step": 69021,
      "training_step_time": 7.482249021530151
    },
    {
      "epoch": 6.582450866699219e-05,
      "model_forward_time": 1.7133593559265137,
      "step": 69022
    },
    {
      "epoch": 6.582450866699219e-05,
      "step": 69022,
      "training_step_time": 7.525366544723511
    },
    {
      "epoch": 6.582546234130859e-05,
      "model_forward_time": 1.7082858085632324,
      "step": 69023
    },
    {
      "epoch": 6.582546234130859e-05,
      "step": 69023,
      "training_step_time": 8.786344051361084
    },
    {
      "epoch": 6.5826416015625e-05,
      "model_forward_time": 1.7184224128723145,
      "step": 69024
    },
    {
      "epoch": 6.5826416015625e-05,
      "step": 69024,
      "training_step_time": 9.216641426086426
    },
    {
      "epoch": 6.582736968994141e-05,
      "model_forward_time": 1.7153964042663574,
      "step": 69025
    },
    {
      "epoch": 6.582736968994141e-05,
      "step": 69025,
      "training_step_time": 7.6747145652771
    },
    {
      "epoch": 6.582832336425781e-05,
      "model_forward_time": 1.7782940864562988,
      "step": 69026
    },
    {
      "epoch": 6.582832336425781e-05,
      "step": 69026,
      "training_step_time": 8.587270975112915
    },
    {
      "epoch": 6.582927703857422e-05,
      "model_forward_time": 2.2658660411834717,
      "step": 69027
    },
    {
      "epoch": 6.582927703857422e-05,
      "step": 69027,
      "training_step_time": 9.900853157043457
    },
    {
      "epoch": 6.583023071289063e-05,
      "model_forward_time": 1.7389585971832275,
      "step": 69028
    },
    {
      "epoch": 6.583023071289063e-05,
      "step": 69028,
      "training_step_time": 8.332416534423828
    },
    {
      "epoch": 6.583118438720703e-05,
      "model_forward_time": 1.7101175785064697,
      "step": 69029
    },
    {
      "epoch": 6.583118438720703e-05,
      "step": 69029,
      "training_step_time": 8.858637809753418
    },
    {
      "epoch": 6.583213806152344e-05,
      "grad_norm": 0.10029368847608566,
      "learning_rate": 2.4010545405125126e-06,
      "loss": 0.0579,
      "step": 69030
    },
    {
      "epoch": 6.583213806152344e-05,
      "model_forward_time": 1.7212591171264648,
      "step": 69030
    },
    {
      "dynamics_loss_avg": 0.05386996325105429,
      "epoch": 6.583213806152344e-05,
      "step": 69030
    },
    {
      "action_loss_avg": 0.002874995311140083,
      "epoch": 6.583213806152344e-05,
      "step": 69030
    },
    {
      "epoch": 6.583213806152344e-05,
      "step": 69030,
      "training_step_time": 7.497818231582642
    },
    {
      "epoch": 6.583309173583985e-05,
      "model_forward_time": 1.718855619430542,
      "step": 69031
    },
    {
      "epoch": 6.583309173583985e-05,
      "step": 69031,
      "training_step_time": 7.534041166305542
    },
    {
      "epoch": 6.583404541015625e-05,
      "model_forward_time": 1.7152881622314453,
      "step": 69032
    },
    {
      "epoch": 6.583404541015625e-05,
      "step": 69032,
      "training_step_time": 8.72129201889038
    },
    {
      "epoch": 6.583499908447266e-05,
      "model_forward_time": 1.719071388244629,
      "step": 69033
    },
    {
      "epoch": 6.583499908447266e-05,
      "step": 69033,
      "training_step_time": 7.873606204986572
    },
    {
      "epoch": 6.583595275878907e-05,
      "model_forward_time": 1.7236089706420898,
      "step": 69034
    },
    {
      "epoch": 6.583595275878907e-05,
      "step": 69034,
      "training_step_time": 7.597385406494141
    },
    {
      "epoch": 6.583690643310547e-05,
      "model_forward_time": 1.7763590812683105,
      "step": 69035
    },
    {
      "epoch": 6.583690643310547e-05,
      "step": 69035,
      "training_step_time": 7.763226270675659
    },
    {
      "epoch": 6.583786010742188e-05,
      "model_forward_time": 2.2992820739746094,
      "step": 69036
    },
    {
      "epoch": 6.583786010742188e-05,
      "step": 69036,
      "training_step_time": 9.575023889541626
    },
    {
      "epoch": 6.583881378173829e-05,
      "model_forward_time": 1.717233419418335,
      "step": 69037
    },
    {
      "epoch": 6.583881378173829e-05,
      "step": 69037,
      "training_step_time": 8.520759344100952
    },
    {
      "epoch": 6.583976745605468e-05,
      "model_forward_time": 1.7237517833709717,
      "step": 69038
    },
    {
      "epoch": 6.583976745605468e-05,
      "step": 69038,
      "training_step_time": 8.30896258354187
    },
    {
      "epoch": 6.58407211303711e-05,
      "model_forward_time": 1.3457133769989014,
      "step": 69039
    },
    {
      "epoch": 6.58407211303711e-05,
      "step": 69039,
      "training_step_time": 7.508490562438965
    },
    {
      "epoch": 6.58416748046875e-05,
      "grad_norm": 0.12815631926059723,
      "learning_rate": 2.399642132274953e-06,
      "loss": 0.0582,
      "step": 69040
    },
    {
      "epoch": 6.58416748046875e-05,
      "model_forward_time": 1.7243702411651611,
      "step": 69040
    },
    {
      "dynamics_loss_avg": 0.05147231761366129,
      "epoch": 6.58416748046875e-05,
      "step": 69040
    },
    {
      "action_loss_avg": 0.0013359435484744608,
      "epoch": 6.58416748046875e-05,
      "step": 69040
    },
    {
      "epoch": 6.58416748046875e-05,
      "step": 69040,
      "training_step_time": 7.587955474853516
    },
    {
      "epoch": 6.58426284790039e-05,
      "model_forward_time": 1.7293000221252441,
      "step": 69041
    },
    {
      "epoch": 6.58426284790039e-05,
      "step": 69041,
      "training_step_time": 7.649080276489258
    },
    {
      "epoch": 6.584358215332031e-05,
      "model_forward_time": 1.7220492362976074,
      "step": 69042
    },
    {
      "epoch": 6.584358215332031e-05,
      "step": 69042,
      "training_step_time": 7.618354320526123
    },
    {
      "epoch": 6.584453582763672e-05,
      "model_forward_time": 1.732503890991211,
      "step": 69043
    },
    {
      "epoch": 6.584453582763672e-05,
      "step": 69043,
      "training_step_time": 8.596241474151611
    },
    {
      "epoch": 6.584548950195312e-05,
      "model_forward_time": 1.7716636657714844,
      "step": 69044
    },
    {
      "epoch": 6.584548950195312e-05,
      "step": 69044,
      "training_step_time": 8.39830994606018
    },
    {
      "epoch": 6.584644317626953e-05,
      "model_forward_time": 2.2944495677948,
      "step": 69045
    },
    {
      "epoch": 6.584644317626953e-05,
      "step": 69045,
      "training_step_time": 7.747220516204834
    },
    {
      "epoch": 6.584739685058594e-05,
      "model_forward_time": 1.7199475765228271,
      "step": 69046
    },
    {
      "epoch": 6.584739685058594e-05,
      "step": 69046,
      "training_step_time": 8.64437747001648
    },
    {
      "epoch": 6.584835052490234e-05,
      "model_forward_time": 1.72212553024292,
      "step": 69047
    },
    {
      "epoch": 6.584835052490234e-05,
      "step": 69047,
      "training_step_time": 9.564682483673096
    },
    {
      "epoch": 6.584930419921875e-05,
      "model_forward_time": 1.71101713180542,
      "step": 69048
    },
    {
      "epoch": 6.584930419921875e-05,
      "step": 69048,
      "training_step_time": 8.33909296989441
    },
    {
      "epoch": 6.585025787353516e-05,
      "model_forward_time": 1.7177822589874268,
      "step": 69049
    },
    {
      "epoch": 6.585025787353516e-05,
      "step": 69049,
      "training_step_time": 7.6900060176849365
    },
    {
      "epoch": 6.585121154785156e-05,
      "grad_norm": 0.11338946223258972,
      "learning_rate": 2.3982300084086224e-06,
      "loss": 0.0614,
      "step": 69050
    },
    {
      "epoch": 6.585121154785156e-05,
      "model_forward_time": 1.7248358726501465,
      "step": 69050
    },
    {
      "dynamics_loss_avg": 0.052889058366417885,
      "epoch": 6.585121154785156e-05,
      "step": 69050
    },
    {
      "action_loss_avg": 0.001332407386507839,
      "epoch": 6.585121154785156e-05,
      "step": 69050
    },
    {
      "epoch": 6.585121154785156e-05,
      "step": 69050,
      "training_step_time": 7.629091024398804
    },
    {
      "epoch": 6.585216522216797e-05,
      "model_forward_time": 1.7222273349761963,
      "step": 69051
    },
    {
      "epoch": 6.585216522216797e-05,
      "step": 69051,
      "training_step_time": 8.5086829662323
    },
    {
      "epoch": 6.585311889648438e-05,
      "model_forward_time": 1.7159357070922852,
      "step": 69052
    },
    {
      "epoch": 6.585311889648438e-05,
      "step": 69052,
      "training_step_time": 7.917239427566528
    },
    {
      "epoch": 6.585407257080078e-05,
      "model_forward_time": 1.7466015815734863,
      "step": 69053
    },
    {
      "epoch": 6.585407257080078e-05,
      "step": 69053,
      "training_step_time": 9.521068811416626
    },
    {
      "epoch": 6.585502624511719e-05,
      "model_forward_time": 1.7112312316894531,
      "step": 69054
    },
    {
      "epoch": 6.585502624511719e-05,
      "step": 69054,
      "training_step_time": 8.211336135864258
    },
    {
      "epoch": 6.58559799194336e-05,
      "model_forward_time": 1.7203454971313477,
      "step": 69055
    },
    {
      "epoch": 6.58559799194336e-05,
      "step": 69055,
      "training_step_time": 7.626796245574951
    },
    {
      "epoch": 6.585693359375e-05,
      "model_forward_time": 1.7233562469482422,
      "step": 69056
    },
    {
      "epoch": 6.585693359375e-05,
      "step": 69056,
      "training_step_time": 7.766709566116333
    },
    {
      "epoch": 6.585788726806641e-05,
      "model_forward_time": 1.7248561382293701,
      "step": 69057
    },
    {
      "epoch": 6.585788726806641e-05,
      "step": 69057,
      "training_step_time": 8.648138284683228
    },
    {
      "epoch": 6.585884094238282e-05,
      "model_forward_time": 1.7169063091278076,
      "step": 69058
    },
    {
      "epoch": 6.585884094238282e-05,
      "step": 69058,
      "training_step_time": 9.70857286453247
    },
    {
      "epoch": 6.585979461669922e-05,
      "model_forward_time": 1.7277228832244873,
      "step": 69059
    },
    {
      "epoch": 6.585979461669922e-05,
      "step": 69059,
      "training_step_time": 6.740026235580444
    },
    {
      "epoch": 6.586074829101563e-05,
      "grad_norm": 0.10284624248743057,
      "learning_rate": 2.396818169067944e-06,
      "loss": 0.0603,
      "step": 69060
    },
    {
      "epoch": 6.586074829101563e-05,
      "model_forward_time": 1.7176799774169922,
      "step": 69060
    },
    {
      "dynamics_loss_avg": 0.061321716755628586,
      "epoch": 6.586074829101563e-05,
      "step": 69060
    },
    {
      "action_loss_avg": 0.002913186769001186,
      "epoch": 6.586074829101563e-05,
      "step": 69060
    },
    {
      "epoch": 6.586074829101563e-05,
      "step": 69060,
      "training_step_time": 6.679224967956543
    },
    {
      "epoch": 6.586170196533204e-05,
      "model_forward_time": 1.7163293361663818,
      "step": 69061
    },
    {
      "epoch": 6.586170196533204e-05,
      "step": 69061,
      "training_step_time": 6.814236164093018
    },
    {
      "epoch": 6.586265563964844e-05,
      "model_forward_time": 1.726464033126831,
      "step": 69062
    },
    {
      "epoch": 6.586265563964844e-05,
      "step": 69062,
      "training_step_time": 6.770880699157715
    },
    {
      "epoch": 6.586360931396485e-05,
      "model_forward_time": 1.7209041118621826,
      "step": 69063
    },
    {
      "epoch": 6.586360931396485e-05,
      "step": 69063,
      "training_step_time": 6.790042877197266
    },
    {
      "epoch": 6.586456298828124e-05,
      "model_forward_time": 1.7196552753448486,
      "step": 69064
    },
    {
      "epoch": 6.586456298828124e-05,
      "step": 69064,
      "training_step_time": 6.841793060302734
    },
    {
      "epoch": 6.586551666259765e-05,
      "model_forward_time": 1.7135121822357178,
      "step": 69065
    },
    {
      "epoch": 6.586551666259765e-05,
      "step": 69065,
      "training_step_time": 6.785726547241211
    },
    {
      "epoch": 6.586647033691406e-05,
      "model_forward_time": 1.720355749130249,
      "step": 69066
    },
    {
      "epoch": 6.586647033691406e-05,
      "step": 69066,
      "training_step_time": 7.7051215171813965
    },
    {
      "epoch": 6.586742401123046e-05,
      "model_forward_time": 1.7242791652679443,
      "step": 69067
    },
    {
      "epoch": 6.586742401123046e-05,
      "step": 69067,
      "training_step_time": 6.749935865402222
    },
    {
      "epoch": 6.586837768554687e-05,
      "model_forward_time": 1.7229058742523193,
      "step": 69068
    },
    {
      "epoch": 6.586837768554687e-05,
      "step": 69068,
      "training_step_time": 7.820951461791992
    },
    {
      "epoch": 6.586933135986328e-05,
      "model_forward_time": 1.7503430843353271,
      "step": 69069
    },
    {
      "epoch": 6.586933135986328e-05,
      "step": 69069,
      "training_step_time": 6.751710891723633
    },
    {
      "epoch": 6.587028503417968e-05,
      "grad_norm": 0.3464787006378174,
      "learning_rate": 2.395406614407318e-06,
      "loss": 0.0566,
      "step": 69070
    },
    {
      "epoch": 6.587028503417968e-05,
      "model_forward_time": 1.7285797595977783,
      "step": 69070
    },
    {
      "dynamics_loss_avg": 0.04860330736264586,
      "epoch": 6.587028503417968e-05,
      "step": 69070
    },
    {
      "action_loss_avg": 0.0016615193337202073,
      "epoch": 6.587028503417968e-05,
      "step": 69070
    },
    {
      "epoch": 6.587028503417968e-05,
      "step": 69070,
      "training_step_time": 6.762084722518921
    },
    {
      "epoch": 6.587123870849609e-05,
      "model_forward_time": 1.722379446029663,
      "step": 69071
    },
    {
      "epoch": 6.587123870849609e-05,
      "step": 69071,
      "training_step_time": 6.8193747997283936
    },
    {
      "epoch": 6.58721923828125e-05,
      "model_forward_time": 1.7248506546020508,
      "step": 69072
    },
    {
      "epoch": 6.58721923828125e-05,
      "step": 69072,
      "training_step_time": 6.805812358856201
    },
    {
      "epoch": 6.58731460571289e-05,
      "model_forward_time": 1.7268590927124023,
      "step": 69073
    },
    {
      "epoch": 6.58731460571289e-05,
      "step": 69073,
      "training_step_time": 6.829927682876587
    },
    {
      "epoch": 6.587409973144531e-05,
      "model_forward_time": 1.712876319885254,
      "step": 69074
    },
    {
      "epoch": 6.587409973144531e-05,
      "step": 69074,
      "training_step_time": 6.777370452880859
    },
    {
      "epoch": 6.587505340576172e-05,
      "model_forward_time": 1.732635259628296,
      "step": 69075
    },
    {
      "epoch": 6.587505340576172e-05,
      "step": 69075,
      "training_step_time": 7.357539415359497
    },
    {
      "epoch": 6.587600708007812e-05,
      "model_forward_time": 1.7429814338684082,
      "step": 69076
    },
    {
      "epoch": 6.587600708007812e-05,
      "step": 69076,
      "training_step_time": 7.223138809204102
    },
    {
      "epoch": 6.587696075439453e-05,
      "model_forward_time": 1.7123064994812012,
      "step": 69077
    },
    {
      "epoch": 6.587696075439453e-05,
      "step": 69077,
      "training_step_time": 9.762665271759033
    },
    {
      "epoch": 6.587791442871094e-05,
      "model_forward_time": 1.7180628776550293,
      "step": 69078
    },
    {
      "epoch": 6.587791442871094e-05,
      "step": 69078,
      "training_step_time": 6.84676718711853
    },
    {
      "epoch": 6.587886810302734e-05,
      "model_forward_time": 1.7185606956481934,
      "step": 69079
    },
    {
      "epoch": 6.587886810302734e-05,
      "step": 69079,
      "training_step_time": 6.795511722564697
    },
    {
      "epoch": 6.587982177734375e-05,
      "grad_norm": 0.0891793891787529,
      "learning_rate": 2.39399534458111e-06,
      "loss": 0.0581,
      "step": 69080
    },
    {
      "epoch": 6.587982177734375e-05,
      "model_forward_time": 1.7105867862701416,
      "step": 69080
    },
    {
      "dynamics_loss_avg": 0.0529098185710609,
      "epoch": 6.587982177734375e-05,
      "step": 69080
    },
    {
      "action_loss_avg": 0.0012907053751405329,
      "epoch": 6.587982177734375e-05,
      "step": 69080
    },
    {
      "epoch": 6.587982177734375e-05,
      "step": 69080,
      "training_step_time": 6.821549654006958
    },
    {
      "epoch": 6.588077545166016e-05,
      "model_forward_time": 1.7111830711364746,
      "step": 69081
    },
    {
      "epoch": 6.588077545166016e-05,
      "step": 69081,
      "training_step_time": 6.825520038604736
    },
    {
      "epoch": 6.588172912597656e-05,
      "model_forward_time": 1.7132694721221924,
      "step": 69082
    },
    {
      "epoch": 6.588172912597656e-05,
      "step": 69082,
      "training_step_time": 6.764251470565796
    },
    {
      "epoch": 6.588268280029297e-05,
      "model_forward_time": 1.7197904586791992,
      "step": 69083
    },
    {
      "epoch": 6.588268280029297e-05,
      "step": 69083,
      "training_step_time": 6.761202335357666
    },
    {
      "epoch": 6.588363647460938e-05,
      "model_forward_time": 1.7067406177520752,
      "step": 69084
    },
    {
      "epoch": 6.588363647460938e-05,
      "step": 69084,
      "training_step_time": 6.755490303039551
    },
    {
      "epoch": 6.588459014892578e-05,
      "model_forward_time": 2.2847814559936523,
      "step": 69085
    },
    {
      "epoch": 6.588459014892578e-05,
      "step": 69085,
      "training_step_time": 7.511322736740112
    },
    {
      "epoch": 6.588554382324219e-05,
      "model_forward_time": 1.711963176727295,
      "step": 69086
    },
    {
      "epoch": 6.588554382324219e-05,
      "step": 69086,
      "training_step_time": 10.697725772857666
    },
    {
      "epoch": 6.58864974975586e-05,
      "model_forward_time": 1.7099313735961914,
      "step": 69087
    },
    {
      "epoch": 6.58864974975586e-05,
      "step": 69087,
      "training_step_time": 7.489627838134766
    },
    {
      "epoch": 6.5887451171875e-05,
      "model_forward_time": 1.7127807140350342,
      "step": 69088
    },
    {
      "epoch": 6.5887451171875e-05,
      "step": 69088,
      "training_step_time": 6.777963399887085
    },
    {
      "epoch": 6.58884048461914e-05,
      "model_forward_time": 1.7083055973052979,
      "step": 69089
    },
    {
      "epoch": 6.58884048461914e-05,
      "step": 69089,
      "training_step_time": 6.800530195236206
    },
    {
      "epoch": 6.588935852050782e-05,
      "grad_norm": 0.11083252727985382,
      "learning_rate": 2.392584359743651e-06,
      "loss": 0.0591,
      "step": 69090
    },
    {
      "epoch": 6.588935852050782e-05,
      "model_forward_time": 1.7150311470031738,
      "step": 69090
    },
    {
      "dynamics_loss_avg": 0.04698201920837164,
      "epoch": 6.588935852050782e-05,
      "step": 69090
    },
    {
      "action_loss_avg": 0.0009670338840805925,
      "epoch": 6.588935852050782e-05,
      "step": 69090
    },
    {
      "epoch": 6.588935852050782e-05,
      "step": 69090,
      "training_step_time": 6.693382263183594
    },
    {
      "epoch": 6.589031219482421e-05,
      "model_forward_time": 1.712031602859497,
      "step": 69091
    },
    {
      "epoch": 6.589031219482421e-05,
      "step": 69091,
      "training_step_time": 6.74591588973999
    },
    {
      "epoch": 6.589126586914062e-05,
      "model_forward_time": 1.705657958984375,
      "step": 69092
    },
    {
      "epoch": 6.589126586914062e-05,
      "step": 69092,
      "training_step_time": 6.739483118057251
    },
    {
      "epoch": 6.589221954345703e-05,
      "model_forward_time": 1.7121763229370117,
      "step": 69093
    },
    {
      "epoch": 6.589221954345703e-05,
      "step": 69093,
      "training_step_time": 6.712188959121704
    },
    {
      "epoch": 6.589317321777343e-05,
      "model_forward_time": 2.281541585922241,
      "step": 69094
    },
    {
      "epoch": 6.589317321777343e-05,
      "step": 69094,
      "training_step_time": 7.4970622062683105
    },
    {
      "epoch": 6.589412689208984e-05,
      "model_forward_time": 1.7114858627319336,
      "step": 69095
    },
    {
      "epoch": 6.589412689208984e-05,
      "step": 69095,
      "training_step_time": 9.76753830909729
    },
    {
      "epoch": 6.589508056640625e-05,
      "model_forward_time": 1.3362154960632324,
      "step": 69096
    },
    {
      "epoch": 6.589508056640625e-05,
      "step": 69096,
      "training_step_time": 7.565821409225464
    },
    {
      "epoch": 6.589603424072265e-05,
      "model_forward_time": 1.7049891948699951,
      "step": 69097
    },
    {
      "epoch": 6.589603424072265e-05,
      "step": 69097,
      "training_step_time": 6.746291399002075
    },
    {
      "epoch": 6.589698791503906e-05,
      "model_forward_time": 1.7206289768218994,
      "step": 69098
    },
    {
      "epoch": 6.589698791503906e-05,
      "step": 69098,
      "training_step_time": 6.72127103805542
    },
    {
      "epoch": 6.589794158935547e-05,
      "model_forward_time": 1.7118191719055176,
      "step": 69099
    },
    {
      "epoch": 6.589794158935547e-05,
      "step": 69099,
      "training_step_time": 6.73078727722168
    },
    {
      "epoch": 6.589889526367187e-05,
      "grad_norm": 0.0827772319316864,
      "learning_rate": 2.3911736600492463e-06,
      "loss": 0.0562,
      "step": 69100
    },
    {
      "epoch": 6.589889526367187e-05,
      "model_forward_time": 1.7098636627197266,
      "step": 69100
    },
    {
      "dynamics_loss_avg": 0.04094485566020012,
      "epoch": 6.589889526367187e-05,
      "step": 69100
    },
    {
      "action_loss_avg": 0.0013146487530320884,
      "epoch": 6.589889526367187e-05,
      "step": 69100
    },
    {
      "epoch": 6.589889526367187e-05,
      "step": 69100,
      "training_step_time": 6.740417003631592
    },
    {
      "epoch": 6.589984893798828e-05,
      "model_forward_time": 1.7160661220550537,
      "step": 69101
    },
    {
      "epoch": 6.589984893798828e-05,
      "step": 69101,
      "training_step_time": 6.8022987842559814
    },
    {
      "epoch": 6.590080261230469e-05,
      "model_forward_time": 1.7371985912322998,
      "step": 69102
    },
    {
      "epoch": 6.590080261230469e-05,
      "step": 69102,
      "training_step_time": 6.808972358703613
    },
    {
      "epoch": 6.590175628662109e-05,
      "model_forward_time": 2.284925937652588,
      "step": 69103
    },
    {
      "epoch": 6.590175628662109e-05,
      "step": 69103,
      "training_step_time": 7.575966119766235
    },
    {
      "epoch": 6.59027099609375e-05,
      "model_forward_time": 1.7321414947509766,
      "step": 69104
    },
    {
      "epoch": 6.59027099609375e-05,
      "step": 69104,
      "training_step_time": 8.773168325424194
    },
    {
      "epoch": 6.590366363525391e-05,
      "model_forward_time": 1.711674451828003,
      "step": 69105
    },
    {
      "epoch": 6.590366363525391e-05,
      "step": 69105,
      "training_step_time": 8.52790880203247
    },
    {
      "epoch": 6.590461730957031e-05,
      "model_forward_time": 1.7075958251953125,
      "step": 69106
    },
    {
      "epoch": 6.590461730957031e-05,
      "step": 69106,
      "training_step_time": 7.680354118347168
    },
    {
      "epoch": 6.590557098388672e-05,
      "model_forward_time": 1.7125825881958008,
      "step": 69107
    },
    {
      "epoch": 6.590557098388672e-05,
      "step": 69107,
      "training_step_time": 6.752512693405151
    },
    {
      "epoch": 6.590652465820313e-05,
      "model_forward_time": 1.7092258930206299,
      "step": 69108
    },
    {
      "epoch": 6.590652465820313e-05,
      "step": 69108,
      "training_step_time": 6.791916608810425
    },
    {
      "epoch": 6.590747833251953e-05,
      "model_forward_time": 1.7146451473236084,
      "step": 69109
    },
    {
      "epoch": 6.590747833251953e-05,
      "step": 69109,
      "training_step_time": 6.710750102996826
    },
    {
      "epoch": 6.590843200683594e-05,
      "grad_norm": 0.08836400508880615,
      "learning_rate": 2.3897632456521675e-06,
      "loss": 0.0579,
      "step": 69110
    },
    {
      "epoch": 6.590843200683594e-05,
      "model_forward_time": 1.7089881896972656,
      "step": 69110
    },
    {
      "dynamics_loss_avg": 0.05830759331583977,
      "epoch": 6.590843200683594e-05,
      "step": 69110
    },
    {
      "action_loss_avg": 0.0014827614155365153,
      "epoch": 6.590843200683594e-05,
      "step": 69110
    },
    {
      "epoch": 6.590843200683594e-05,
      "step": 69110,
      "training_step_time": 6.760935068130493
    },
    {
      "epoch": 6.590938568115235e-05,
      "model_forward_time": 1.7147274017333984,
      "step": 69111
    },
    {
      "epoch": 6.590938568115235e-05,
      "step": 69111,
      "training_step_time": 6.724829435348511
    },
    {
      "epoch": 6.591033935546875e-05,
      "model_forward_time": 2.2857437133789062,
      "step": 69112
    },
    {
      "epoch": 6.591033935546875e-05,
      "step": 69112,
      "training_step_time": 7.079735279083252
    },
    {
      "epoch": 6.591129302978516e-05,
      "model_forward_time": 1.7026724815368652,
      "step": 69113
    },
    {
      "epoch": 6.591129302978516e-05,
      "step": 69113,
      "training_step_time": 9.710686206817627
    },
    {
      "epoch": 6.591224670410157e-05,
      "model_forward_time": 1.7185139656066895,
      "step": 69114
    },
    {
      "epoch": 6.591224670410157e-05,
      "step": 69114,
      "training_step_time": 8.436328411102295
    },
    {
      "epoch": 6.591320037841796e-05,
      "model_forward_time": 1.7107717990875244,
      "step": 69115
    },
    {
      "epoch": 6.591320037841796e-05,
      "step": 69115,
      "training_step_time": 7.6838085651397705
    },
    {
      "epoch": 6.591415405273437e-05,
      "model_forward_time": 1.7092173099517822,
      "step": 69116
    },
    {
      "epoch": 6.591415405273437e-05,
      "step": 69116,
      "training_step_time": 6.73747992515564
    },
    {
      "epoch": 6.591510772705079e-05,
      "model_forward_time": 1.721658706665039,
      "step": 69117
    },
    {
      "epoch": 6.591510772705079e-05,
      "step": 69117,
      "training_step_time": 6.763920783996582
    },
    {
      "epoch": 6.591606140136718e-05,
      "model_forward_time": 1.7319860458374023,
      "step": 69118
    },
    {
      "epoch": 6.591606140136718e-05,
      "step": 69118,
      "training_step_time": 6.699315309524536
    },
    {
      "epoch": 6.59170150756836e-05,
      "model_forward_time": 1.7075858116149902,
      "step": 69119
    },
    {
      "epoch": 6.59170150756836e-05,
      "step": 69119,
      "training_step_time": 6.782560586929321
    },
    {
      "epoch": 6.591796875e-05,
      "grad_norm": 0.10769647359848022,
      "learning_rate": 2.388353116706658e-06,
      "loss": 0.0561,
      "step": 69120
    },
    {
      "epoch": 6.591796875e-05,
      "model_forward_time": 1.7168033123016357,
      "step": 69120
    },
    {
      "dynamics_loss_avg": 0.054014799930155274,
      "epoch": 6.591796875e-05,
      "step": 69120
    },
    {
      "action_loss_avg": 0.0015929422661429271,
      "epoch": 6.591796875e-05,
      "step": 69120
    },
    {
      "epoch": 6.591796875e-05,
      "step": 69120,
      "training_step_time": 6.707479476928711
    },
    {
      "epoch": 6.59189224243164e-05,
      "model_forward_time": 2.282172918319702,
      "step": 69121
    },
    {
      "epoch": 6.59189224243164e-05,
      "step": 69121,
      "training_step_time": 7.130199670791626
    },
    {
      "epoch": 6.591987609863281e-05,
      "model_forward_time": 1.7043561935424805,
      "step": 69122
    },
    {
      "epoch": 6.591987609863281e-05,
      "step": 69122,
      "training_step_time": 10.355256795883179
    },
    {
      "epoch": 6.592082977294922e-05,
      "model_forward_time": 1.7077414989471436,
      "step": 69123
    },
    {
      "epoch": 6.592082977294922e-05,
      "step": 69123,
      "training_step_time": 8.889792203903198
    },
    {
      "epoch": 6.592178344726562e-05,
      "model_forward_time": 1.7074711322784424,
      "step": 69124
    },
    {
      "epoch": 6.592178344726562e-05,
      "step": 69124,
      "training_step_time": 7.546453952789307
    },
    {
      "epoch": 6.592273712158203e-05,
      "model_forward_time": 1.7091929912567139,
      "step": 69125
    },
    {
      "epoch": 6.592273712158203e-05,
      "step": 69125,
      "training_step_time": 7.0670859813690186
    },
    {
      "epoch": 6.592369079589844e-05,
      "model_forward_time": 1.7075257301330566,
      "step": 69126
    },
    {
      "epoch": 6.592369079589844e-05,
      "step": 69126,
      "training_step_time": 6.734468460083008
    },
    {
      "epoch": 6.592464447021484e-05,
      "model_forward_time": 1.717245101928711,
      "step": 69127
    },
    {
      "epoch": 6.592464447021484e-05,
      "step": 69127,
      "training_step_time": 6.742238283157349
    },
    {
      "epoch": 6.592559814453125e-05,
      "model_forward_time": 1.7162830829620361,
      "step": 69128
    },
    {
      "epoch": 6.592559814453125e-05,
      "step": 69128,
      "training_step_time": 6.754811525344849
    },
    {
      "epoch": 6.592655181884766e-05,
      "model_forward_time": 1.7099194526672363,
      "step": 69129
    },
    {
      "epoch": 6.592655181884766e-05,
      "step": 69129,
      "training_step_time": 6.727329730987549
    },
    {
      "epoch": 6.592750549316406e-05,
      "grad_norm": 0.09639107435941696,
      "learning_rate": 2.3869432733669234e-06,
      "loss": 0.0599,
      "step": 69130
    },
    {
      "epoch": 6.592750549316406e-05,
      "model_forward_time": 2.2987959384918213,
      "step": 69130
    },
    {
      "dynamics_loss_avg": 0.06484890542924404,
      "epoch": 6.592750549316406e-05,
      "step": 69130
    },
    {
      "action_loss_avg": 0.0022460342181148006,
      "epoch": 6.592750549316406e-05,
      "step": 69130
    },
    {
      "epoch": 6.592750549316406e-05,
      "step": 69130,
      "training_step_time": 7.143326044082642
    },
    {
      "epoch": 6.592845916748047e-05,
      "model_forward_time": 1.8123486042022705,
      "step": 69131
    },
    {
      "epoch": 6.592845916748047e-05,
      "step": 69131,
      "training_step_time": 9.827112436294556
    },
    {
      "epoch": 6.592941284179688e-05,
      "model_forward_time": 1.8133070468902588,
      "step": 69132
    },
    {
      "epoch": 6.592941284179688e-05,
      "step": 69132,
      "training_step_time": 9.670784711837769
    },
    {
      "epoch": 6.593036651611328e-05,
      "model_forward_time": 1.7749199867248535,
      "step": 69133
    },
    {
      "epoch": 6.593036651611328e-05,
      "step": 69133,
      "training_step_time": 7.602134943008423
    },
    {
      "epoch": 6.593132019042969e-05,
      "model_forward_time": 1.7355616092681885,
      "step": 69134
    },
    {
      "epoch": 6.593132019042969e-05,
      "step": 69134,
      "training_step_time": 7.13509202003479
    },
    {
      "epoch": 6.59322738647461e-05,
      "model_forward_time": 1.7114906311035156,
      "step": 69135
    },
    {
      "epoch": 6.59322738647461e-05,
      "step": 69135,
      "training_step_time": 6.6743714809417725
    },
    {
      "epoch": 6.59332275390625e-05,
      "model_forward_time": 1.7089829444885254,
      "step": 69136
    },
    {
      "epoch": 6.59332275390625e-05,
      "step": 69136,
      "training_step_time": 6.723847389221191
    },
    {
      "epoch": 6.59341812133789e-05,
      "model_forward_time": 1.7123680114746094,
      "step": 69137
    },
    {
      "epoch": 6.59341812133789e-05,
      "step": 69137,
      "training_step_time": 6.668402194976807
    },
    {
      "epoch": 6.593513488769532e-05,
      "model_forward_time": 1.7120447158813477,
      "step": 69138
    },
    {
      "epoch": 6.593513488769532e-05,
      "step": 69138,
      "training_step_time": 6.678368806838989
    },
    {
      "epoch": 6.593608856201171e-05,
      "model_forward_time": 1.7158021926879883,
      "step": 69139
    },
    {
      "epoch": 6.593608856201171e-05,
      "step": 69139,
      "training_step_time": 7.688941478729248
    },
    {
      "epoch": 6.593704223632813e-05,
      "grad_norm": 0.10654138773679733,
      "learning_rate": 2.385533715787145e-06,
      "loss": 0.0578,
      "step": 69140
    },
    {
      "epoch": 6.593704223632813e-05,
      "model_forward_time": 1.7083005905151367,
      "step": 69140
    },
    {
      "dynamics_loss_avg": 0.04486912814900279,
      "epoch": 6.593704223632813e-05,
      "step": 69140
    },
    {
      "action_loss_avg": 0.0009905796963721515,
      "epoch": 6.593704223632813e-05,
      "step": 69140
    },
    {
      "epoch": 6.593704223632813e-05,
      "step": 69140,
      "training_step_time": 9.241705894470215
    },
    {
      "epoch": 6.593799591064454e-05,
      "model_forward_time": 1.711613416671753,
      "step": 69141
    },
    {
      "epoch": 6.593799591064454e-05,
      "step": 69141,
      "training_step_time": 9.88785982131958
    },
    {
      "epoch": 6.593894958496093e-05,
      "model_forward_time": 1.7093334197998047,
      "step": 69142
    },
    {
      "epoch": 6.593894958496093e-05,
      "step": 69142,
      "training_step_time": 7.524200916290283
    },
    {
      "epoch": 6.593990325927734e-05,
      "model_forward_time": 1.711268424987793,
      "step": 69143
    },
    {
      "epoch": 6.593990325927734e-05,
      "step": 69143,
      "training_step_time": 7.67583703994751
    },
    {
      "epoch": 6.594085693359376e-05,
      "model_forward_time": 1.6925687789916992,
      "step": 69144
    },
    {
      "epoch": 6.594085693359376e-05,
      "step": 69144,
      "training_step_time": 6.97433614730835
    },
    {
      "epoch": 6.594181060791015e-05,
      "model_forward_time": 1.7118024826049805,
      "step": 69145
    },
    {
      "epoch": 6.594181060791015e-05,
      "step": 69145,
      "training_step_time": 6.672361373901367
    },
    {
      "epoch": 6.594276428222656e-05,
      "model_forward_time": 1.7170493602752686,
      "step": 69146
    },
    {
      "epoch": 6.594276428222656e-05,
      "step": 69146,
      "training_step_time": 6.656971454620361
    },
    {
      "epoch": 6.594371795654297e-05,
      "model_forward_time": 1.715214490890503,
      "step": 69147
    },
    {
      "epoch": 6.594371795654297e-05,
      "step": 69147,
      "training_step_time": 6.67370343208313
    },
    {
      "epoch": 6.594467163085937e-05,
      "model_forward_time": 1.7097976207733154,
      "step": 69148
    },
    {
      "epoch": 6.594467163085937e-05,
      "step": 69148,
      "training_step_time": 7.012087106704712
    },
    {
      "epoch": 6.594562530517578e-05,
      "model_forward_time": 2.297147750854492,
      "step": 69149
    },
    {
      "epoch": 6.594562530517578e-05,
      "step": 69149,
      "training_step_time": 8.122722148895264
    },
    {
      "epoch": 6.594657897949219e-05,
      "grad_norm": 0.09906253218650818,
      "learning_rate": 2.3841244441214674e-06,
      "loss": 0.0541,
      "step": 69150
    },
    {
      "epoch": 6.594657897949219e-05,
      "model_forward_time": 1.7053122520446777,
      "step": 69150
    },
    {
      "dynamics_loss_avg": 0.050609343219548464,
      "epoch": 6.594657897949219e-05,
      "step": 69150
    },
    {
      "action_loss_avg": 0.0018147579336073249,
      "epoch": 6.594657897949219e-05,
      "step": 69150
    },
    {
      "epoch": 6.594657897949219e-05,
      "step": 69150,
      "training_step_time": 9.06656289100647
    },
    {
      "epoch": 6.594753265380859e-05,
      "model_forward_time": 1.719024896621704,
      "step": 69151
    },
    {
      "epoch": 6.594753265380859e-05,
      "step": 69151,
      "training_step_time": 7.604119300842285
    },
    {
      "epoch": 6.5948486328125e-05,
      "model_forward_time": 1.7131381034851074,
      "step": 69152
    },
    {
      "epoch": 6.5948486328125e-05,
      "step": 69152,
      "training_step_time": 7.62861180305481
    },
    {
      "epoch": 6.594944000244141e-05,
      "model_forward_time": 1.711099624633789,
      "step": 69153
    },
    {
      "epoch": 6.594944000244141e-05,
      "step": 69153,
      "training_step_time": 7.064553737640381
    },
    {
      "epoch": 6.595039367675781e-05,
      "model_forward_time": 1.7157986164093018,
      "step": 69154
    },
    {
      "epoch": 6.595039367675781e-05,
      "step": 69154,
      "training_step_time": 6.665069818496704
    },
    {
      "epoch": 6.595134735107422e-05,
      "model_forward_time": 1.716010570526123,
      "step": 69155
    },
    {
      "epoch": 6.595134735107422e-05,
      "step": 69155,
      "training_step_time": 6.657788515090942
    },
    {
      "epoch": 6.595230102539063e-05,
      "model_forward_time": 1.7161386013031006,
      "step": 69156
    },
    {
      "epoch": 6.595230102539063e-05,
      "step": 69156,
      "training_step_time": 6.7106640338897705
    },
    {
      "epoch": 6.595325469970703e-05,
      "model_forward_time": 1.7118678092956543,
      "step": 69157
    },
    {
      "epoch": 6.595325469970703e-05,
      "step": 69157,
      "training_step_time": 6.981899738311768
    },
    {
      "epoch": 6.595420837402344e-05,
      "model_forward_time": 2.2887051105499268,
      "step": 69158
    },
    {
      "epoch": 6.595420837402344e-05,
      "step": 69158,
      "training_step_time": 8.603787422180176
    },
    {
      "epoch": 6.595516204833985e-05,
      "model_forward_time": 1.7076101303100586,
      "step": 69159
    },
    {
      "epoch": 6.595516204833985e-05,
      "step": 69159,
      "training_step_time": 9.722265005111694
    },
    {
      "epoch": 6.595611572265625e-05,
      "grad_norm": 0.10656917095184326,
      "learning_rate": 2.3827154585240076e-06,
      "loss": 0.0568,
      "step": 69160
    },
    {
      "epoch": 6.595611572265625e-05,
      "model_forward_time": 1.7097365856170654,
      "step": 69160
    },
    {
      "dynamics_loss_avg": 0.04639059565961361,
      "epoch": 6.595611572265625e-05,
      "step": 69160
    },
    {
      "action_loss_avg": 0.0013758368382696061,
      "epoch": 6.595611572265625e-05,
      "step": 69160
    },
    {
      "epoch": 6.595611572265625e-05,
      "step": 69160,
      "training_step_time": 7.627350807189941
    },
    {
      "epoch": 6.595706939697266e-05,
      "model_forward_time": 1.7110209465026855,
      "step": 69161
    },
    {
      "epoch": 6.595706939697266e-05,
      "step": 69161,
      "training_step_time": 7.588549613952637
    },
    {
      "epoch": 6.595802307128907e-05,
      "model_forward_time": 1.706491470336914,
      "step": 69162
    },
    {
      "epoch": 6.595802307128907e-05,
      "step": 69162,
      "training_step_time": 7.087335109710693
    },
    {
      "epoch": 6.595897674560547e-05,
      "model_forward_time": 1.7117929458618164,
      "step": 69163
    },
    {
      "epoch": 6.595897674560547e-05,
      "step": 69163,
      "training_step_time": 6.665924787521362
    },
    {
      "epoch": 6.595993041992188e-05,
      "model_forward_time": 1.7100276947021484,
      "step": 69164
    },
    {
      "epoch": 6.595993041992188e-05,
      "step": 69164,
      "training_step_time": 6.760178089141846
    },
    {
      "epoch": 6.596088409423829e-05,
      "model_forward_time": 1.706458330154419,
      "step": 69165
    },
    {
      "epoch": 6.596088409423829e-05,
      "step": 69165,
      "training_step_time": 6.735124588012695
    },
    {
      "epoch": 6.596183776855468e-05,
      "model_forward_time": 1.7148146629333496,
      "step": 69166
    },
    {
      "epoch": 6.596183776855468e-05,
      "step": 69166,
      "training_step_time": 7.022827386856079
    },
    {
      "epoch": 6.59627914428711e-05,
      "model_forward_time": 2.303180694580078,
      "step": 69167
    },
    {
      "epoch": 6.59627914428711e-05,
      "step": 69167,
      "training_step_time": 8.22587513923645
    },
    {
      "epoch": 6.59637451171875e-05,
      "model_forward_time": 1.7061281204223633,
      "step": 69168
    },
    {
      "epoch": 6.59637451171875e-05,
      "step": 69168,
      "training_step_time": 10.745381593704224
    },
    {
      "epoch": 6.59646987915039e-05,
      "model_forward_time": 1.7177143096923828,
      "step": 69169
    },
    {
      "epoch": 6.59646987915039e-05,
      "step": 69169,
      "training_step_time": 7.777261257171631
    },
    {
      "epoch": 6.596565246582031e-05,
      "grad_norm": 0.11217206716537476,
      "learning_rate": 2.38130675914885e-06,
      "loss": 0.0549,
      "step": 69170
    },
    {
      "epoch": 6.596565246582031e-05,
      "model_forward_time": 1.7116174697875977,
      "step": 69170
    },
    {
      "dynamics_loss_avg": 0.055058271251618864,
      "epoch": 6.596565246582031e-05,
      "step": 69170
    },
    {
      "action_loss_avg": 0.0012372516037430614,
      "epoch": 6.596565246582031e-05,
      "step": 69170
    },
    {
      "epoch": 6.596565246582031e-05,
      "step": 69170,
      "training_step_time": 7.614484786987305
    },
    {
      "epoch": 6.596660614013672e-05,
      "model_forward_time": 1.7154266834259033,
      "step": 69171
    },
    {
      "epoch": 6.596660614013672e-05,
      "step": 69171,
      "training_step_time": 7.145914316177368
    },
    {
      "epoch": 6.596755981445312e-05,
      "model_forward_time": 1.7205595970153809,
      "step": 69172
    },
    {
      "epoch": 6.596755981445312e-05,
      "step": 69172,
      "training_step_time": 6.714518070220947
    },
    {
      "epoch": 6.596851348876953e-05,
      "model_forward_time": 1.7181448936462402,
      "step": 69173
    },
    {
      "epoch": 6.596851348876953e-05,
      "step": 69173,
      "training_step_time": 6.784281492233276
    },
    {
      "epoch": 6.596946716308594e-05,
      "model_forward_time": 1.714433193206787,
      "step": 69174
    },
    {
      "epoch": 6.596946716308594e-05,
      "step": 69174,
      "training_step_time": 6.73842191696167
    },
    {
      "epoch": 6.597042083740234e-05,
      "model_forward_time": 1.7113401889801025,
      "step": 69175
    },
    {
      "epoch": 6.597042083740234e-05,
      "step": 69175,
      "training_step_time": 7.086186408996582
    },
    {
      "epoch": 6.597137451171875e-05,
      "model_forward_time": 2.313220977783203,
      "step": 69176
    },
    {
      "epoch": 6.597137451171875e-05,
      "step": 69176,
      "training_step_time": 8.658896923065186
    },
    {
      "epoch": 6.597232818603516e-05,
      "model_forward_time": 1.708500623703003,
      "step": 69177
    },
    {
      "epoch": 6.597232818603516e-05,
      "step": 69177,
      "training_step_time": 8.399675607681274
    },
    {
      "epoch": 6.597328186035156e-05,
      "model_forward_time": 1.7121775150299072,
      "step": 69178
    },
    {
      "epoch": 6.597328186035156e-05,
      "step": 69178,
      "training_step_time": 8.281595230102539
    },
    {
      "epoch": 6.597423553466797e-05,
      "model_forward_time": 1.736067295074463,
      "step": 69179
    },
    {
      "epoch": 6.597423553466797e-05,
      "step": 69179,
      "training_step_time": 7.6530187129974365
    },
    {
      "epoch": 6.597518920898438e-05,
      "grad_norm": 0.0965052992105484,
      "learning_rate": 2.3798983461500498e-06,
      "loss": 0.0554,
      "step": 69180
    },
    {
      "epoch": 6.597518920898438e-05,
      "model_forward_time": 1.7205188274383545,
      "step": 69180
    },
    {
      "dynamics_loss_avg": 0.046454920619726184,
      "epoch": 6.597518920898438e-05,
      "step": 69180
    },
    {
      "action_loss_avg": 0.0014331963378936053,
      "epoch": 6.597518920898438e-05,
      "step": 69180
    },
    {
      "epoch": 6.597518920898438e-05,
      "step": 69180,
      "training_step_time": 7.731731414794922
    },
    {
      "epoch": 6.597614288330078e-05,
      "model_forward_time": 1.7088403701782227,
      "step": 69181
    },
    {
      "epoch": 6.597614288330078e-05,
      "step": 69181,
      "training_step_time": 6.766478776931763
    },
    {
      "epoch": 6.597709655761719e-05,
      "model_forward_time": 1.717055082321167,
      "step": 69182
    },
    {
      "epoch": 6.597709655761719e-05,
      "step": 69182,
      "training_step_time": 6.773585319519043
    },
    {
      "epoch": 6.59780502319336e-05,
      "model_forward_time": 1.7193577289581299,
      "step": 69183
    },
    {
      "epoch": 6.59780502319336e-05,
      "step": 69183,
      "training_step_time": 6.821303129196167
    },
    {
      "epoch": 6.597900390625e-05,
      "model_forward_time": 1.7117490768432617,
      "step": 69184
    },
    {
      "epoch": 6.597900390625e-05,
      "step": 69184,
      "training_step_time": 7.081803321838379
    },
    {
      "epoch": 6.597995758056641e-05,
      "model_forward_time": 2.294419527053833,
      "step": 69185
    },
    {
      "epoch": 6.597995758056641e-05,
      "step": 69185,
      "training_step_time": 8.95679759979248
    },
    {
      "epoch": 6.598091125488282e-05,
      "model_forward_time": 1.7141640186309814,
      "step": 69186
    },
    {
      "epoch": 6.598091125488282e-05,
      "step": 69186,
      "training_step_time": 8.285588264465332
    },
    {
      "epoch": 6.598186492919922e-05,
      "model_forward_time": 1.7234771251678467,
      "step": 69187
    },
    {
      "epoch": 6.598186492919922e-05,
      "step": 69187,
      "training_step_time": 8.894137382507324
    },
    {
      "epoch": 6.598281860351563e-05,
      "model_forward_time": 1.7144958972930908,
      "step": 69188
    },
    {
      "epoch": 6.598281860351563e-05,
      "step": 69188,
      "training_step_time": 7.717843055725098
    },
    {
      "epoch": 6.598377227783202e-05,
      "model_forward_time": 1.7210180759429932,
      "step": 69189
    },
    {
      "epoch": 6.598377227783202e-05,
      "step": 69189,
      "training_step_time": 7.665887355804443
    },
    {
      "epoch": 6.598472595214844e-05,
      "grad_norm": 0.10377799719572067,
      "learning_rate": 2.3784902196816256e-06,
      "loss": 0.0546,
      "step": 69190
    },
    {
      "epoch": 6.598472595214844e-05,
      "model_forward_time": 1.7046799659729004,
      "step": 69190
    },
    {
      "dynamics_loss_avg": 0.03924309611320496,
      "epoch": 6.598472595214844e-05,
      "step": 69190
    },
    {
      "action_loss_avg": 0.0012660123786190525,
      "epoch": 6.598472595214844e-05,
      "step": 69190
    },
    {
      "epoch": 6.598472595214844e-05,
      "step": 69190,
      "training_step_time": 7.092257738113403
    },
    {
      "epoch": 6.598567962646485e-05,
      "model_forward_time": 1.7108192443847656,
      "step": 69191
    },
    {
      "epoch": 6.598567962646485e-05,
      "step": 69191,
      "training_step_time": 6.756988048553467
    },
    {
      "epoch": 6.598663330078124e-05,
      "model_forward_time": 1.7109882831573486,
      "step": 69192
    },
    {
      "epoch": 6.598663330078124e-05,
      "step": 69192,
      "training_step_time": 6.746037483215332
    },
    {
      "epoch": 6.598758697509765e-05,
      "model_forward_time": 1.7050602436065674,
      "step": 69193
    },
    {
      "epoch": 6.598758697509765e-05,
      "step": 69193,
      "training_step_time": 7.032118558883667
    },
    {
      "epoch": 6.598854064941406e-05,
      "model_forward_time": 1.7455589771270752,
      "step": 69194
    },
    {
      "epoch": 6.598854064941406e-05,
      "step": 69194,
      "training_step_time": 7.578561544418335
    },
    {
      "epoch": 6.598949432373046e-05,
      "model_forward_time": 2.3816030025482178,
      "step": 69195
    },
    {
      "epoch": 6.598949432373046e-05,
      "step": 69195,
      "training_step_time": 8.242294788360596
    },
    {
      "epoch": 6.599044799804687e-05,
      "model_forward_time": 1.8201539516448975,
      "step": 69196
    },
    {
      "epoch": 6.599044799804687e-05,
      "step": 69196,
      "training_step_time": 8.863928079605103
    },
    {
      "epoch": 6.599140167236328e-05,
      "model_forward_time": 1.4573414325714111,
      "step": 69197
    },
    {
      "epoch": 6.599140167236328e-05,
      "step": 69197,
      "training_step_time": 7.706680059432983
    },
    {
      "epoch": 6.599235534667968e-05,
      "model_forward_time": 1.8244340419769287,
      "step": 69198
    },
    {
      "epoch": 6.599235534667968e-05,
      "step": 69198,
      "training_step_time": 7.683108329772949
    },
    {
      "epoch": 6.599330902099609e-05,
      "model_forward_time": 1.8245561122894287,
      "step": 69199
    },
    {
      "epoch": 6.599330902099609e-05,
      "step": 69199,
      "training_step_time": 7.133316278457642
    },
    {
      "epoch": 6.59942626953125e-05,
      "grad_norm": 0.10559555143117905,
      "learning_rate": 2.377082379897572e-06,
      "loss": 0.0575,
      "step": 69200
    },
    {
      "epoch": 6.59942626953125e-05,
      "model_forward_time": 1.824666976928711,
      "step": 69200
    },
    {
      "dynamics_loss_avg": 0.06708260104060174,
      "epoch": 6.59942626953125e-05,
      "step": 69200
    },
    {
      "action_loss_avg": 0.0019838663283735514,
      "epoch": 6.59942626953125e-05,
      "step": 69200
    },
    {
      "epoch": 6.59942626953125e-05,
      "step": 69200,
      "training_step_time": 6.739711761474609
    },
    {
      "epoch": 6.59952163696289e-05,
      "model_forward_time": 1.8194255828857422,
      "step": 69201
    },
    {
      "epoch": 6.59952163696289e-05,
      "step": 69201,
      "training_step_time": 6.764634132385254
    },
    {
      "epoch": 6.599617004394531e-05,
      "model_forward_time": 1.8242249488830566,
      "step": 69202
    },
    {
      "epoch": 6.599617004394531e-05,
      "step": 69202,
      "training_step_time": 7.03315544128418
    },
    {
      "epoch": 6.599712371826172e-05,
      "model_forward_time": 1.8287568092346191,
      "step": 69203
    },
    {
      "epoch": 6.599712371826172e-05,
      "step": 69203,
      "training_step_time": 8.552765130996704
    },
    {
      "epoch": 6.599807739257812e-05,
      "model_forward_time": 2.369239568710327,
      "step": 69204
    },
    {
      "epoch": 6.599807739257812e-05,
      "step": 69204,
      "training_step_time": 7.608495235443115
    },
    {
      "epoch": 6.599903106689453e-05,
      "model_forward_time": 1.832564115524292,
      "step": 69205
    },
    {
      "epoch": 6.599903106689453e-05,
      "step": 69205,
      "training_step_time": 9.769165754318237
    },
    {
      "epoch": 6.599998474121094e-05,
      "model_forward_time": 1.8139286041259766,
      "step": 69206
    },
    {
      "epoch": 6.599998474121094e-05,
      "step": 69206,
      "training_step_time": 7.734137535095215
    },
    {
      "epoch": 6.600093841552734e-05,
      "model_forward_time": 1.8222813606262207,
      "step": 69207
    },
    {
      "epoch": 6.600093841552734e-05,
      "step": 69207,
      "training_step_time": 7.581171274185181
    },
    {
      "epoch": 6.600189208984375e-05,
      "model_forward_time": 1.819856882095337,
      "step": 69208
    },
    {
      "epoch": 6.600189208984375e-05,
      "step": 69208,
      "training_step_time": 7.110588312149048
    },
    {
      "epoch": 6.600284576416016e-05,
      "model_forward_time": 1.83335542678833,
      "step": 69209
    },
    {
      "epoch": 6.600284576416016e-05,
      "step": 69209,
      "training_step_time": 7.099323034286499
    },
    {
      "epoch": 6.600379943847656e-05,
      "grad_norm": 0.12348770350217819,
      "learning_rate": 2.3756748269518426e-06,
      "loss": 0.0556,
      "step": 69210
    },
    {
      "epoch": 6.600379943847656e-05,
      "model_forward_time": 1.8248002529144287,
      "step": 69210
    },
    {
      "dynamics_loss_avg": 0.06327722240239382,
      "epoch": 6.600379943847656e-05,
      "step": 69210
    },
    {
      "action_loss_avg": 0.0017043847881723195,
      "epoch": 6.600379943847656e-05,
      "step": 69210
    },
    {
      "epoch": 6.600379943847656e-05,
      "step": 69210,
      "training_step_time": 6.744726657867432
    },
    {
      "epoch": 6.600475311279297e-05,
      "model_forward_time": 1.8150572776794434,
      "step": 69211
    },
    {
      "epoch": 6.600475311279297e-05,
      "step": 69211,
      "training_step_time": 7.0259177684783936
    },
    {
      "epoch": 6.600570678710938e-05,
      "model_forward_time": 1.8399016857147217,
      "step": 69212
    },
    {
      "epoch": 6.600570678710938e-05,
      "step": 69212,
      "training_step_time": 8.474544048309326
    },
    {
      "epoch": 6.600666046142578e-05,
      "model_forward_time": 2.3758273124694824,
      "step": 69213
    },
    {
      "epoch": 6.600666046142578e-05,
      "step": 69213,
      "training_step_time": 8.189539432525635
    },
    {
      "epoch": 6.600761413574219e-05,
      "model_forward_time": 1.8359463214874268,
      "step": 69214
    },
    {
      "epoch": 6.600761413574219e-05,
      "step": 69214,
      "training_step_time": 9.893580436706543
    },
    {
      "epoch": 6.60085678100586e-05,
      "model_forward_time": 1.8313746452331543,
      "step": 69215
    },
    {
      "epoch": 6.60085678100586e-05,
      "step": 69215,
      "training_step_time": 8.645162105560303
    },
    {
      "epoch": 6.6009521484375e-05,
      "model_forward_time": 1.8298113346099854,
      "step": 69216
    },
    {
      "epoch": 6.6009521484375e-05,
      "step": 69216,
      "training_step_time": 7.584105491638184
    },
    {
      "epoch": 6.60104751586914e-05,
      "model_forward_time": 1.8319618701934814,
      "step": 69217
    },
    {
      "epoch": 6.60104751586914e-05,
      "step": 69217,
      "training_step_time": 7.662636041641235
    },
    {
      "epoch": 6.601142883300782e-05,
      "model_forward_time": 1.8325285911560059,
      "step": 69218
    },
    {
      "epoch": 6.601142883300782e-05,
      "step": 69218,
      "training_step_time": 7.045990228652954
    },
    {
      "epoch": 6.601238250732421e-05,
      "model_forward_time": 1.8356504440307617,
      "step": 69219
    },
    {
      "epoch": 6.601238250732421e-05,
      "step": 69219,
      "training_step_time": 6.656886339187622
    },
    {
      "epoch": 6.601333618164062e-05,
      "grad_norm": 0.10597091913223267,
      "learning_rate": 2.374267560998371e-06,
      "loss": 0.0568,
      "step": 69220
    },
    {
      "epoch": 6.601333618164062e-05,
      "model_forward_time": 1.831432580947876,
      "step": 69220
    },
    {
      "dynamics_loss_avg": 0.04378350619226694,
      "epoch": 6.601333618164062e-05,
      "step": 69220
    },
    {
      "action_loss_avg": 0.0009508296236163006,
      "epoch": 6.601333618164062e-05,
      "step": 69220
    },
    {
      "epoch": 6.601333618164062e-05,
      "step": 69220,
      "training_step_time": 7.049499034881592
    },
    {
      "epoch": 6.601428985595703e-05,
      "model_forward_time": 1.8262228965759277,
      "step": 69221
    },
    {
      "epoch": 6.601428985595703e-05,
      "step": 69221,
      "training_step_time": 7.538607597351074
    },
    {
      "epoch": 6.601524353027343e-05,
      "model_forward_time": 2.3822789192199707,
      "step": 69222
    },
    {
      "epoch": 6.601524353027343e-05,
      "step": 69222,
      "training_step_time": 7.625509262084961
    },
    {
      "epoch": 6.601619720458984e-05,
      "model_forward_time": 1.454498529434204,
      "step": 69223
    },
    {
      "epoch": 6.601619720458984e-05,
      "step": 69223,
      "training_step_time": 10.008321046829224
    },
    {
      "epoch": 6.601715087890625e-05,
      "model_forward_time": 1.454519510269165,
      "step": 69224
    },
    {
      "epoch": 6.601715087890625e-05,
      "step": 69224,
      "training_step_time": 8.587530612945557
    },
    {
      "epoch": 6.601810455322265e-05,
      "model_forward_time": 1.46763014793396,
      "step": 69225
    },
    {
      "epoch": 6.601810455322265e-05,
      "step": 69225,
      "training_step_time": 8.379901647567749
    },
    {
      "epoch": 6.601905822753906e-05,
      "model_forward_time": 1.820502519607544,
      "step": 69226
    },
    {
      "epoch": 6.601905822753906e-05,
      "step": 69226,
      "training_step_time": 7.615249156951904
    },
    {
      "epoch": 6.602001190185547e-05,
      "model_forward_time": 1.813645362854004,
      "step": 69227
    },
    {
      "epoch": 6.602001190185547e-05,
      "step": 69227,
      "training_step_time": 7.126520872116089
    },
    {
      "epoch": 6.602096557617187e-05,
      "model_forward_time": 1.833465337753296,
      "step": 69228
    },
    {
      "epoch": 6.602096557617187e-05,
      "step": 69228,
      "training_step_time": 6.699944496154785
    },
    {
      "epoch": 6.602191925048828e-05,
      "model_forward_time": 1.8346772193908691,
      "step": 69229
    },
    {
      "epoch": 6.602191925048828e-05,
      "step": 69229,
      "training_step_time": 7.007708549499512
    },
    {
      "epoch": 6.602287292480469e-05,
      "grad_norm": 0.1358443647623062,
      "learning_rate": 2.3728605821910476e-06,
      "loss": 0.0573,
      "step": 69230
    },
    {
      "epoch": 6.602287292480469e-05,
      "model_forward_time": 1.8260221481323242,
      "step": 69230
    },
    {
      "dynamics_loss_avg": 0.0605875076726079,
      "epoch": 6.602287292480469e-05,
      "step": 69230
    },
    {
      "action_loss_avg": 0.0015562874352326617,
      "epoch": 6.602287292480469e-05,
      "step": 69230
    },
    {
      "epoch": 6.602287292480469e-05,
      "step": 69230,
      "training_step_time": 7.841447591781616
    },
    {
      "epoch": 6.602382659912109e-05,
      "model_forward_time": 1.7741470336914062,
      "step": 69231
    },
    {
      "epoch": 6.602382659912109e-05,
      "step": 69231,
      "training_step_time": 8.37377142906189
    },
    {
      "epoch": 6.60247802734375e-05,
      "model_forward_time": 2.34725284576416,
      "step": 69232
    },
    {
      "epoch": 6.60247802734375e-05,
      "step": 69232,
      "training_step_time": 8.678282260894775
    },
    {
      "epoch": 6.602573394775391e-05,
      "model_forward_time": 1.8257129192352295,
      "step": 69233
    },
    {
      "epoch": 6.602573394775391e-05,
      "step": 69233,
      "training_step_time": 8.374557971954346
    },
    {
      "epoch": 6.602668762207031e-05,
      "model_forward_time": 1.8055295944213867,
      "step": 69234
    },
    {
      "epoch": 6.602668762207031e-05,
      "step": 69234,
      "training_step_time": 7.660924196243286
    },
    {
      "epoch": 6.602764129638672e-05,
      "model_forward_time": 1.7939658164978027,
      "step": 69235
    },
    {
      "epoch": 6.602764129638672e-05,
      "step": 69235,
      "training_step_time": 7.702145338058472
    },
    {
      "epoch": 6.602859497070313e-05,
      "model_forward_time": 1.809173822402954,
      "step": 69236
    },
    {
      "epoch": 6.602859497070313e-05,
      "step": 69236,
      "training_step_time": 7.160678863525391
    },
    {
      "epoch": 6.602954864501953e-05,
      "model_forward_time": 1.8118257522583008,
      "step": 69237
    },
    {
      "epoch": 6.602954864501953e-05,
      "step": 69237,
      "training_step_time": 6.756535530090332
    },
    {
      "epoch": 6.603050231933594e-05,
      "model_forward_time": 1.8092620372772217,
      "step": 69238
    },
    {
      "epoch": 6.603050231933594e-05,
      "step": 69238,
      "training_step_time": 7.035282373428345
    },
    {
      "epoch": 6.603145599365235e-05,
      "model_forward_time": 1.8106420040130615,
      "step": 69239
    },
    {
      "epoch": 6.603145599365235e-05,
      "step": 69239,
      "training_step_time": 7.403605222702026
    },
    {
      "epoch": 6.603240966796875e-05,
      "grad_norm": 0.07707411795854568,
      "learning_rate": 2.3714538906837452e-06,
      "loss": 0.0571,
      "step": 69240
    },
    {
      "epoch": 6.603240966796875e-05,
      "model_forward_time": 1.8177285194396973,
      "step": 69240
    },
    {
      "dynamics_loss_avg": 0.05315348207950592,
      "epoch": 6.603240966796875e-05,
      "step": 69240
    },
    {
      "action_loss_avg": 0.0008712512208148837,
      "epoch": 6.603240966796875e-05,
      "step": 69240
    },
    {
      "epoch": 6.603240966796875e-05,
      "step": 69240,
      "training_step_time": 7.646897315979004
    },
    {
      "epoch": 6.603336334228516e-05,
      "model_forward_time": 2.2495052814483643,
      "step": 69241
    },
    {
      "epoch": 6.603336334228516e-05,
      "step": 69241,
      "training_step_time": 7.783953428268433
    },
    {
      "epoch": 6.603431701660157e-05,
      "model_forward_time": 1.7111856937408447,
      "step": 69242
    },
    {
      "epoch": 6.603431701660157e-05,
      "step": 69242,
      "training_step_time": 8.779703378677368
    },
    {
      "epoch": 6.603527069091796e-05,
      "model_forward_time": 1.7229983806610107,
      "step": 69243
    },
    {
      "epoch": 6.603527069091796e-05,
      "step": 69243,
      "training_step_time": 8.475396633148193
    },
    {
      "epoch": 6.603622436523437e-05,
      "model_forward_time": 1.727430820465088,
      "step": 69244
    },
    {
      "epoch": 6.603622436523437e-05,
      "step": 69244,
      "training_step_time": 7.620590686798096
    },
    {
      "epoch": 6.603717803955079e-05,
      "model_forward_time": 1.72873854637146,
      "step": 69245
    },
    {
      "epoch": 6.603717803955079e-05,
      "step": 69245,
      "training_step_time": 7.150570392608643
    },
    {
      "epoch": 6.603813171386718e-05,
      "model_forward_time": 1.7321767807006836,
      "step": 69246
    },
    {
      "epoch": 6.603813171386718e-05,
      "step": 69246,
      "training_step_time": 7.1019346714019775
    },
    {
      "epoch": 6.60390853881836e-05,
      "model_forward_time": 1.7418313026428223,
      "step": 69247
    },
    {
      "epoch": 6.60390853881836e-05,
      "step": 69247,
      "training_step_time": 7.623914957046509
    },
    {
      "epoch": 6.60400390625e-05,
      "model_forward_time": 1.7285676002502441,
      "step": 69248
    },
    {
      "epoch": 6.60400390625e-05,
      "step": 69248,
      "training_step_time": 7.43632698059082
    },
    {
      "epoch": 6.60409927368164e-05,
      "model_forward_time": 1.7332239151000977,
      "step": 69249
    },
    {
      "epoch": 6.60409927368164e-05,
      "step": 69249,
      "training_step_time": 7.699974298477173
    },
    {
      "epoch": 6.604194641113281e-05,
      "grad_norm": 0.10861939936876297,
      "learning_rate": 2.3700474866302915e-06,
      "loss": 0.0558,
      "step": 69250
    },
    {
      "epoch": 6.604194641113281e-05,
      "model_forward_time": 2.2905125617980957,
      "step": 69250
    },
    {
      "dynamics_loss_avg": 0.04965563379228115,
      "epoch": 6.604194641113281e-05,
      "step": 69250
    },
    {
      "action_loss_avg": 0.0009846389963058754,
      "epoch": 6.604194641113281e-05,
      "step": 69250
    },
    {
      "epoch": 6.604194641113281e-05,
      "step": 69250,
      "training_step_time": 7.707411766052246
    },
    {
      "epoch": 6.604290008544922e-05,
      "model_forward_time": 1.7262017726898193,
      "step": 69251
    },
    {
      "epoch": 6.604290008544922e-05,
      "step": 69251,
      "training_step_time": 10.369136095046997
    },
    {
      "epoch": 6.604385375976562e-05,
      "model_forward_time": 1.729017972946167,
      "step": 69252
    },
    {
      "epoch": 6.604385375976562e-05,
      "step": 69252,
      "training_step_time": 8.309879541397095
    },
    {
      "epoch": 6.604480743408203e-05,
      "model_forward_time": 1.7208735942840576,
      "step": 69253
    },
    {
      "epoch": 6.604480743408203e-05,
      "step": 69253,
      "training_step_time": 7.402990341186523
    },
    {
      "epoch": 6.604576110839844e-05,
      "model_forward_time": 1.720430850982666,
      "step": 69254
    },
    {
      "epoch": 6.604576110839844e-05,
      "step": 69254,
      "training_step_time": 7.7384374141693115
    },
    {
      "epoch": 6.604671478271484e-05,
      "model_forward_time": 1.726395845413208,
      "step": 69255
    },
    {
      "epoch": 6.604671478271484e-05,
      "step": 69255,
      "training_step_time": 7.044973611831665
    },
    {
      "epoch": 6.604766845703125e-05,
      "model_forward_time": 1.728562355041504,
      "step": 69256
    },
    {
      "epoch": 6.604766845703125e-05,
      "step": 69256,
      "training_step_time": 7.227658033370972
    },
    {
      "epoch": 6.604862213134766e-05,
      "model_forward_time": 1.698150634765625,
      "step": 69257
    },
    {
      "epoch": 6.604862213134766e-05,
      "step": 69257,
      "training_step_time": 8.599019050598145
    },
    {
      "epoch": 6.604957580566406e-05,
      "model_forward_time": 1.713754415512085,
      "step": 69258
    },
    {
      "epoch": 6.604957580566406e-05,
      "step": 69258,
      "training_step_time": 7.625076055526733
    },
    {
      "epoch": 6.605052947998047e-05,
      "model_forward_time": 2.2645723819732666,
      "step": 69259
    },
    {
      "epoch": 6.605052947998047e-05,
      "step": 69259,
      "training_step_time": 8.44308352470398
    },
    {
      "epoch": 6.605148315429688e-05,
      "grad_norm": 0.10321681946516037,
      "learning_rate": 2.368641370184492e-06,
      "loss": 0.0568,
      "step": 69260
    },
    {
      "epoch": 6.605148315429688e-05,
      "model_forward_time": 1.710545539855957,
      "step": 69260
    },
    {
      "dynamics_loss_avg": 0.06302165798842907,
      "epoch": 6.605148315429688e-05,
      "step": 69260
    },
    {
      "action_loss_avg": 0.0015086808649357407,
      "epoch": 6.605148315429688e-05,
      "step": 69260
    },
    {
      "epoch": 6.605148315429688e-05,
      "step": 69260,
      "training_step_time": 7.760472059249878
    },
    {
      "epoch": 6.605243682861328e-05,
      "model_forward_time": 1.3362252712249756,
      "step": 69261
    },
    {
      "epoch": 6.605243682861328e-05,
      "step": 69261,
      "training_step_time": 9.544173240661621
    },
    {
      "epoch": 6.605339050292969e-05,
      "model_forward_time": 1.7194881439208984,
      "step": 69262
    },
    {
      "epoch": 6.605339050292969e-05,
      "step": 69262,
      "training_step_time": 7.578593492507935
    },
    {
      "epoch": 6.60543441772461e-05,
      "model_forward_time": 1.7144761085510254,
      "step": 69263
    },
    {
      "epoch": 6.60543441772461e-05,
      "step": 69263,
      "training_step_time": 7.734055042266846
    },
    {
      "epoch": 6.60552978515625e-05,
      "model_forward_time": 1.717984676361084,
      "step": 69264
    },
    {
      "epoch": 6.60552978515625e-05,
      "step": 69264,
      "training_step_time": 7.654318332672119
    },
    {
      "epoch": 6.605625152587891e-05,
      "model_forward_time": 1.7158079147338867,
      "step": 69265
    },
    {
      "epoch": 6.605625152587891e-05,
      "step": 69265,
      "training_step_time": 6.661522626876831
    },
    {
      "epoch": 6.605720520019532e-05,
      "model_forward_time": 1.7869293689727783,
      "step": 69266
    },
    {
      "epoch": 6.605720520019532e-05,
      "step": 69266,
      "training_step_time": 8.646735906600952
    },
    {
      "epoch": 6.605815887451171e-05,
      "model_forward_time": 1.7150964736938477,
      "step": 69267
    },
    {
      "epoch": 6.605815887451171e-05,
      "step": 69267,
      "training_step_time": 7.658154010772705
    },
    {
      "epoch": 6.605911254882813e-05,
      "model_forward_time": 2.257699728012085,
      "step": 69268
    },
    {
      "epoch": 6.605911254882813e-05,
      "step": 69268,
      "training_step_time": 7.602408409118652
    },
    {
      "epoch": 6.606006622314454e-05,
      "model_forward_time": 1.7154614925384521,
      "step": 69269
    },
    {
      "epoch": 6.606006622314454e-05,
      "step": 69269,
      "training_step_time": 8.537596464157104
    },
    {
      "epoch": 6.606101989746093e-05,
      "grad_norm": 0.1866084188222885,
      "learning_rate": 2.367235541500114e-06,
      "loss": 0.055,
      "step": 69270
    },
    {
      "epoch": 6.606101989746093e-05,
      "model_forward_time": 1.7045845985412598,
      "step": 69270
    },
    {
      "dynamics_loss_avg": 0.05192809198051691,
      "epoch": 6.606101989746093e-05,
      "step": 69270
    },
    {
      "action_loss_avg": 0.001755264916573651,
      "epoch": 6.606101989746093e-05,
      "step": 69270
    },
    {
      "epoch": 6.606101989746093e-05,
      "step": 69270,
      "training_step_time": 8.697546243667603
    },
    {
      "epoch": 6.606197357177734e-05,
      "model_forward_time": 1.345137596130371,
      "step": 69271
    },
    {
      "epoch": 6.606197357177734e-05,
      "step": 69271,
      "training_step_time": 7.646273374557495
    },
    {
      "epoch": 6.606292724609376e-05,
      "model_forward_time": 1.7151317596435547,
      "step": 69272
    },
    {
      "epoch": 6.606292724609376e-05,
      "step": 69272,
      "training_step_time": 8.749202013015747
    },
    {
      "epoch": 6.606388092041015e-05,
      "model_forward_time": 1.7178096771240234,
      "step": 69273
    },
    {
      "epoch": 6.606388092041015e-05,
      "step": 69273,
      "training_step_time": 7.08591365814209
    },
    {
      "epoch": 6.606483459472656e-05,
      "model_forward_time": 1.72298264503479,
      "step": 69274
    },
    {
      "epoch": 6.606483459472656e-05,
      "step": 69274,
      "training_step_time": 7.49691367149353
    },
    {
      "epoch": 6.606578826904297e-05,
      "model_forward_time": 1.7714118957519531,
      "step": 69275
    },
    {
      "epoch": 6.606578826904297e-05,
      "step": 69275,
      "training_step_time": 7.705065488815308
    },
    {
      "epoch": 6.606674194335937e-05,
      "model_forward_time": 1.7152702808380127,
      "step": 69276
    },
    {
      "epoch": 6.606674194335937e-05,
      "step": 69276,
      "training_step_time": 7.610631942749023
    },
    {
      "epoch": 6.606769561767578e-05,
      "model_forward_time": 1.7231788635253906,
      "step": 69277
    },
    {
      "epoch": 6.606769561767578e-05,
      "step": 69277,
      "training_step_time": 7.57562255859375
    },
    {
      "epoch": 6.606864929199219e-05,
      "model_forward_time": 2.267498016357422,
      "step": 69278
    },
    {
      "epoch": 6.606864929199219e-05,
      "step": 69278,
      "training_step_time": 8.801028966903687
    },
    {
      "epoch": 6.606960296630859e-05,
      "model_forward_time": 1.7094292640686035,
      "step": 69279
    },
    {
      "epoch": 6.606960296630859e-05,
      "step": 69279,
      "training_step_time": 9.734578847885132
    },
    {
      "epoch": 6.6070556640625e-05,
      "grad_norm": 0.12377478182315826,
      "learning_rate": 2.3658300007308995e-06,
      "loss": 0.0531,
      "step": 69280
    },
    {
      "epoch": 6.6070556640625e-05,
      "model_forward_time": 1.714332103729248,
      "step": 69280
    },
    {
      "dynamics_loss_avg": 0.05822510588914156,
      "epoch": 6.6070556640625e-05,
      "step": 69280
    },
    {
      "action_loss_avg": 0.0014390255062608048,
      "epoch": 6.6070556640625e-05,
      "step": 69280
    },
    {
      "epoch": 6.6070556640625e-05,
      "step": 69280,
      "training_step_time": 7.614369869232178
    },
    {
      "epoch": 6.607151031494141e-05,
      "model_forward_time": 1.719480037689209,
      "step": 69281
    },
    {
      "epoch": 6.607151031494141e-05,
      "step": 69281,
      "training_step_time": 7.655048608779907
    },
    {
      "epoch": 6.607246398925781e-05,
      "model_forward_time": 1.7065346240997314,
      "step": 69282
    },
    {
      "epoch": 6.607246398925781e-05,
      "step": 69282,
      "training_step_time": 7.060757637023926
    },
    {
      "epoch": 6.607341766357422e-05,
      "model_forward_time": 1.7273495197296143,
      "step": 69283
    },
    {
      "epoch": 6.607341766357422e-05,
      "step": 69283,
      "training_step_time": 6.678964376449585
    },
    {
      "epoch": 6.607437133789063e-05,
      "model_forward_time": 1.7731988430023193,
      "step": 69284
    },
    {
      "epoch": 6.607437133789063e-05,
      "step": 69284,
      "training_step_time": 7.693493127822876
    },
    {
      "epoch": 6.607532501220703e-05,
      "model_forward_time": 1.727494239807129,
      "step": 69285
    },
    {
      "epoch": 6.607532501220703e-05,
      "step": 69285,
      "training_step_time": 7.602402210235596
    },
    {
      "epoch": 6.607627868652344e-05,
      "model_forward_time": 1.7275958061218262,
      "step": 69286
    },
    {
      "epoch": 6.607627868652344e-05,
      "step": 69286,
      "training_step_time": 7.653533697128296
    },
    {
      "epoch": 6.607723236083985e-05,
      "model_forward_time": 2.2811944484710693,
      "step": 69287
    },
    {
      "epoch": 6.607723236083985e-05,
      "step": 69287,
      "training_step_time": 7.78833794593811
    },
    {
      "epoch": 6.607818603515625e-05,
      "model_forward_time": 1.7299165725708008,
      "step": 69288
    },
    {
      "epoch": 6.607818603515625e-05,
      "step": 69288,
      "training_step_time": 8.86897897720337
    },
    {
      "epoch": 6.607913970947266e-05,
      "model_forward_time": 1.721909999847412,
      "step": 69289
    },
    {
      "epoch": 6.607913970947266e-05,
      "step": 69289,
      "training_step_time": 9.322017431259155
    },
    {
      "epoch": 6.608009338378907e-05,
      "grad_norm": 0.3680681586265564,
      "learning_rate": 2.3644247480305548e-06,
      "loss": 0.0575,
      "step": 69290
    },
    {
      "epoch": 6.608009338378907e-05,
      "model_forward_time": 1.7267065048217773,
      "step": 69290
    },
    {
      "dynamics_loss_avg": 0.04903913252055645,
      "epoch": 6.608009338378907e-05,
      "step": 69290
    },
    {
      "action_loss_avg": 0.0018840779375750571,
      "epoch": 6.608009338378907e-05,
      "step": 69290
    },
    {
      "epoch": 6.608009338378907e-05,
      "step": 69290,
      "training_step_time": 7.380232810974121
    },
    {
      "epoch": 6.608104705810547e-05,
      "model_forward_time": 1.727351188659668,
      "step": 69291
    },
    {
      "epoch": 6.608104705810547e-05,
      "step": 69291,
      "training_step_time": 7.684686899185181
    },
    {
      "epoch": 6.608200073242188e-05,
      "model_forward_time": 1.7275254726409912,
      "step": 69292
    },
    {
      "epoch": 6.608200073242188e-05,
      "step": 69292,
      "training_step_time": 6.704592704772949
    },
    {
      "epoch": 6.608295440673829e-05,
      "model_forward_time": 1.747896432876587,
      "step": 69293
    },
    {
      "epoch": 6.608295440673829e-05,
      "step": 69293,
      "training_step_time": 8.68889307975769
    },
    {
      "epoch": 6.608390808105468e-05,
      "model_forward_time": 1.728123664855957,
      "step": 69294
    },
    {
      "epoch": 6.608390808105468e-05,
      "step": 69294,
      "training_step_time": 7.59389328956604
    },
    {
      "epoch": 6.60848617553711e-05,
      "model_forward_time": 1.7270383834838867,
      "step": 69295
    },
    {
      "epoch": 6.60848617553711e-05,
      "step": 69295,
      "training_step_time": 7.772008180618286
    },
    {
      "epoch": 6.60858154296875e-05,
      "model_forward_time": 2.3118083477020264,
      "step": 69296
    },
    {
      "epoch": 6.60858154296875e-05,
      "step": 69296,
      "training_step_time": 8.240939378738403
    },
    {
      "epoch": 6.60867691040039e-05,
      "model_forward_time": 1.6990580558776855,
      "step": 69297
    },
    {
      "epoch": 6.60867691040039e-05,
      "step": 69297,
      "training_step_time": 8.793407917022705
    },
    {
      "epoch": 6.608772277832031e-05,
      "model_forward_time": 1.7117598056793213,
      "step": 69298
    },
    {
      "epoch": 6.608772277832031e-05,
      "step": 69298,
      "training_step_time": 8.43095850944519
    },
    {
      "epoch": 6.608867645263673e-05,
      "model_forward_time": 1.6907885074615479,
      "step": 69299
    },
    {
      "epoch": 6.608867645263673e-05,
      "step": 69299,
      "training_step_time": 7.452706813812256
    },
    {
      "epoch": 6.608963012695312e-05,
      "grad_norm": 0.07212075591087341,
      "learning_rate": 2.3630197835527597e-06,
      "loss": 0.0523,
      "step": 69300
    },
    {
      "epoch": 6.608963012695312e-05,
      "model_forward_time": 1.7058494091033936,
      "step": 69300
    },
    {
      "dynamics_loss_avg": 0.05480842534452677,
      "epoch": 6.608963012695312e-05,
      "step": 69300
    },
    {
      "action_loss_avg": 0.0013844100409187376,
      "epoch": 6.608963012695312e-05,
      "step": 69300
    },
    {
      "epoch": 6.608963012695312e-05,
      "step": 69300,
      "training_step_time": 7.672310829162598
    },
    {
      "epoch": 6.609058380126953e-05,
      "model_forward_time": 1.718552589416504,
      "step": 69301
    },
    {
      "epoch": 6.609058380126953e-05,
      "step": 69301,
      "training_step_time": 6.715935468673706
    },
    {
      "epoch": 6.609153747558594e-05,
      "model_forward_time": 1.7262790203094482,
      "step": 69302
    },
    {
      "epoch": 6.609153747558594e-05,
      "step": 69302,
      "training_step_time": 7.650210618972778
    },
    {
      "epoch": 6.609249114990234e-05,
      "model_forward_time": 1.712026834487915,
      "step": 69303
    },
    {
      "epoch": 6.609249114990234e-05,
      "step": 69303,
      "training_step_time": 7.588660478591919
    },
    {
      "epoch": 6.609344482421875e-05,
      "model_forward_time": 1.7129650115966797,
      "step": 69304
    },
    {
      "epoch": 6.609344482421875e-05,
      "step": 69304,
      "training_step_time": 8.47946047782898
    },
    {
      "epoch": 6.609439849853516e-05,
      "model_forward_time": 2.2830429077148438,
      "step": 69305
    },
    {
      "epoch": 6.609439849853516e-05,
      "step": 69305,
      "training_step_time": 7.570762872695923
    },
    {
      "epoch": 6.609535217285156e-05,
      "model_forward_time": 1.7152068614959717,
      "step": 69306
    },
    {
      "epoch": 6.609535217285156e-05,
      "step": 69306,
      "training_step_time": 9.656529188156128
    },
    {
      "epoch": 6.609630584716797e-05,
      "model_forward_time": 1.7181220054626465,
      "step": 69307
    },
    {
      "epoch": 6.609630584716797e-05,
      "step": 69307,
      "training_step_time": 8.770218133926392
    },
    {
      "epoch": 6.609725952148438e-05,
      "model_forward_time": 1.7094309329986572,
      "step": 69308
    },
    {
      "epoch": 6.609725952148438e-05,
      "step": 69308,
      "training_step_time": 7.55872917175293
    },
    {
      "epoch": 6.609821319580078e-05,
      "model_forward_time": 1.7247581481933594,
      "step": 69309
    },
    {
      "epoch": 6.609821319580078e-05,
      "step": 69309,
      "training_step_time": 7.621438264846802
    },
    {
      "epoch": 6.609916687011719e-05,
      "grad_norm": 0.07644302397966385,
      "learning_rate": 2.3616151074511546e-06,
      "loss": 0.0523,
      "step": 69310
    },
    {
      "epoch": 6.609916687011719e-05,
      "model_forward_time": 1.7081892490386963,
      "step": 69310
    },
    {
      "dynamics_loss_avg": 0.054556934721767904,
      "epoch": 6.609916687011719e-05,
      "step": 69310
    },
    {
      "action_loss_avg": 0.0013888275658246129,
      "epoch": 6.609916687011719e-05,
      "step": 69310
    },
    {
      "epoch": 6.609916687011719e-05,
      "step": 69310,
      "training_step_time": 6.67756986618042
    },
    {
      "epoch": 6.61001205444336e-05,
      "model_forward_time": 1.7222886085510254,
      "step": 69311
    },
    {
      "epoch": 6.61001205444336e-05,
      "step": 69311,
      "training_step_time": 7.5157880783081055
    },
    {
      "epoch": 6.610107421875e-05,
      "model_forward_time": 1.8017585277557373,
      "step": 69312
    },
    {
      "epoch": 6.610107421875e-05,
      "step": 69312,
      "training_step_time": 7.611084222793579
    },
    {
      "epoch": 6.610202789306641e-05,
      "model_forward_time": 1.7248108386993408,
      "step": 69313
    },
    {
      "epoch": 6.610202789306641e-05,
      "step": 69313,
      "training_step_time": 8.494363069534302
    },
    {
      "epoch": 6.61029815673828e-05,
      "model_forward_time": 1.7198796272277832,
      "step": 69314
    },
    {
      "epoch": 6.61029815673828e-05,
      "step": 69314,
      "training_step_time": 8.180964469909668
    },
    {
      "epoch": 6.610393524169922e-05,
      "model_forward_time": 1.7174427509307861,
      "step": 69315
    },
    {
      "epoch": 6.610393524169922e-05,
      "step": 69315,
      "training_step_time": 8.544091939926147
    },
    {
      "epoch": 6.610488891601563e-05,
      "model_forward_time": 1.7010204792022705,
      "step": 69316
    },
    {
      "epoch": 6.610488891601563e-05,
      "step": 69316,
      "training_step_time": 7.628502130508423
    },
    {
      "epoch": 6.610584259033202e-05,
      "model_forward_time": 1.7231833934783936,
      "step": 69317
    },
    {
      "epoch": 6.610584259033202e-05,
      "step": 69317,
      "training_step_time": 8.628148794174194
    },
    {
      "epoch": 6.610679626464844e-05,
      "model_forward_time": 1.743065595626831,
      "step": 69318
    },
    {
      "epoch": 6.610679626464844e-05,
      "step": 69318,
      "training_step_time": 7.589824438095093
    },
    {
      "epoch": 6.610774993896485e-05,
      "model_forward_time": 1.3592004776000977,
      "step": 69319
    },
    {
      "epoch": 6.610774993896485e-05,
      "step": 69319,
      "training_step_time": 6.7083728313446045
    },
    {
      "epoch": 6.610870361328124e-05,
      "grad_norm": 0.10735151171684265,
      "learning_rate": 2.3602107198793543e-06,
      "loss": 0.0559,
      "step": 69320
    },
    {
      "epoch": 6.610870361328124e-05,
      "model_forward_time": 1.730109691619873,
      "step": 69320
    },
    {
      "dynamics_loss_avg": 0.05131450295448303,
      "epoch": 6.610870361328124e-05,
      "step": 69320
    },
    {
      "action_loss_avg": 0.0016643497510813177,
      "epoch": 6.610870361328124e-05,
      "step": 69320
    },
    {
      "epoch": 6.610870361328124e-05,
      "step": 69320,
      "training_step_time": 7.469689846038818
    },
    {
      "epoch": 6.610965728759765e-05,
      "model_forward_time": 1.8073441982269287,
      "step": 69321
    },
    {
      "epoch": 6.610965728759765e-05,
      "step": 69321,
      "training_step_time": 7.684640407562256
    },
    {
      "epoch": 6.611061096191407e-05,
      "model_forward_time": 1.7398946285247803,
      "step": 69322
    },
    {
      "epoch": 6.611061096191407e-05,
      "step": 69322,
      "training_step_time": 7.591918706893921
    },
    {
      "epoch": 6.611156463623046e-05,
      "model_forward_time": 1.7249772548675537,
      "step": 69323
    },
    {
      "epoch": 6.611156463623046e-05,
      "step": 69323,
      "training_step_time": 8.690837860107422
    },
    {
      "epoch": 6.611251831054687e-05,
      "model_forward_time": 2.3027586936950684,
      "step": 69324
    },
    {
      "epoch": 6.611251831054687e-05,
      "step": 69324,
      "training_step_time": 8.209991693496704
    },
    {
      "epoch": 6.611347198486328e-05,
      "model_forward_time": 1.7353260517120361,
      "step": 69325
    },
    {
      "epoch": 6.611347198486328e-05,
      "step": 69325,
      "training_step_time": 7.65917444229126
    },
    {
      "epoch": 6.611442565917968e-05,
      "model_forward_time": 1.7347784042358398,
      "step": 69326
    },
    {
      "epoch": 6.611442565917968e-05,
      "step": 69326,
      "training_step_time": 7.637816429138184
    },
    {
      "epoch": 6.611537933349609e-05,
      "model_forward_time": 1.7423388957977295,
      "step": 69327
    },
    {
      "epoch": 6.611537933349609e-05,
      "step": 69327,
      "training_step_time": 7.5189714431762695
    },
    {
      "epoch": 6.61163330078125e-05,
      "model_forward_time": 1.7388875484466553,
      "step": 69328
    },
    {
      "epoch": 6.61163330078125e-05,
      "step": 69328,
      "training_step_time": 7.667804718017578
    },
    {
      "epoch": 6.61172866821289e-05,
      "model_forward_time": 1.7269692420959473,
      "step": 69329
    },
    {
      "epoch": 6.61172866821289e-05,
      "step": 69329,
      "training_step_time": 7.224529027938843
    },
    {
      "epoch": 6.611824035644531e-05,
      "grad_norm": 0.1014789566397667,
      "learning_rate": 2.3588066209909426e-06,
      "loss": 0.0533,
      "step": 69330
    },
    {
      "epoch": 6.611824035644531e-05,
      "model_forward_time": 1.798938512802124,
      "step": 69330
    },
    {
      "dynamics_loss_avg": 0.05025413278490305,
      "epoch": 6.611824035644531e-05,
      "step": 69330
    },
    {
      "action_loss_avg": 0.0013681382813956588,
      "epoch": 6.611824035644531e-05,
      "step": 69330
    },
    {
      "epoch": 6.611824035644531e-05,
      "step": 69330,
      "training_step_time": 7.643105745315552
    },
    {
      "epoch": 6.611919403076172e-05,
      "model_forward_time": 1.73433256149292,
      "step": 69331
    },
    {
      "epoch": 6.611919403076172e-05,
      "step": 69331,
      "training_step_time": 7.5365564823150635
    },
    {
      "epoch": 6.612014770507812e-05,
      "model_forward_time": 1.7246067523956299,
      "step": 69332
    },
    {
      "epoch": 6.612014770507812e-05,
      "step": 69332,
      "training_step_time": 9.455505609512329
    },
    {
      "epoch": 6.612110137939453e-05,
      "model_forward_time": 2.2836201190948486,
      "step": 69333
    },
    {
      "epoch": 6.612110137939453e-05,
      "step": 69333,
      "training_step_time": 9.714869260787964
    },
    {
      "epoch": 6.612205505371094e-05,
      "model_forward_time": 1.7265427112579346,
      "step": 69334
    },
    {
      "epoch": 6.612205505371094e-05,
      "step": 69334,
      "training_step_time": 9.615456104278564
    },
    {
      "epoch": 6.612300872802734e-05,
      "model_forward_time": 1.7269747257232666,
      "step": 69335
    },
    {
      "epoch": 6.612300872802734e-05,
      "step": 69335,
      "training_step_time": 7.714955568313599
    },
    {
      "epoch": 6.612396240234375e-05,
      "model_forward_time": 1.7272052764892578,
      "step": 69336
    },
    {
      "epoch": 6.612396240234375e-05,
      "step": 69336,
      "training_step_time": 7.623609781265259
    },
    {
      "epoch": 6.612491607666016e-05,
      "model_forward_time": 1.7336132526397705,
      "step": 69337
    },
    {
      "epoch": 6.612491607666016e-05,
      "step": 69337,
      "training_step_time": 7.340426683425903
    },
    {
      "epoch": 6.612586975097656e-05,
      "model_forward_time": 1.7274434566497803,
      "step": 69338
    },
    {
      "epoch": 6.612586975097656e-05,
      "step": 69338,
      "training_step_time": 7.293423414230347
    },
    {
      "epoch": 6.612682342529297e-05,
      "model_forward_time": 1.8015005588531494,
      "step": 69339
    },
    {
      "epoch": 6.612682342529297e-05,
      "step": 69339,
      "training_step_time": 7.755460023880005
    },
    {
      "epoch": 6.612777709960938e-05,
      "grad_norm": 0.09202267229557037,
      "learning_rate": 2.3574028109394666e-06,
      "loss": 0.053,
      "step": 69340
    },
    {
      "epoch": 6.612777709960938e-05,
      "model_forward_time": 1.7210707664489746,
      "step": 69340
    },
    {
      "dynamics_loss_avg": 0.057256128266453744,
      "epoch": 6.612777709960938e-05,
      "step": 69340
    },
    {
      "action_loss_avg": 0.001827286431216635,
      "epoch": 6.612777709960938e-05,
      "step": 69340
    },
    {
      "epoch": 6.612777709960938e-05,
      "step": 69340,
      "training_step_time": 7.619284629821777
    },
    {
      "epoch": 6.612873077392578e-05,
      "model_forward_time": 1.7258892059326172,
      "step": 69341
    },
    {
      "epoch": 6.612873077392578e-05,
      "step": 69341,
      "training_step_time": 8.266539812088013
    },
    {
      "epoch": 6.612968444824219e-05,
      "model_forward_time": 2.302215099334717,
      "step": 69342
    },
    {
      "epoch": 6.612968444824219e-05,
      "step": 69342,
      "training_step_time": 8.229309797286987
    },
    {
      "epoch": 6.61306381225586e-05,
      "model_forward_time": 1.735400676727295,
      "step": 69343
    },
    {
      "epoch": 6.61306381225586e-05,
      "step": 69343,
      "training_step_time": 8.556673765182495
    },
    {
      "epoch": 6.6131591796875e-05,
      "model_forward_time": 1.7308118343353271,
      "step": 69344
    },
    {
      "epoch": 6.6131591796875e-05,
      "step": 69344,
      "training_step_time": 7.693631649017334
    },
    {
      "epoch": 6.61325454711914e-05,
      "model_forward_time": 1.7269277572631836,
      "step": 69345
    },
    {
      "epoch": 6.61325454711914e-05,
      "step": 69345,
      "training_step_time": 7.627616882324219
    },
    {
      "epoch": 6.613349914550782e-05,
      "model_forward_time": 1.7375257015228271,
      "step": 69346
    },
    {
      "epoch": 6.613349914550782e-05,
      "step": 69346,
      "training_step_time": 7.214737415313721
    },
    {
      "epoch": 6.613445281982421e-05,
      "model_forward_time": 1.728367567062378,
      "step": 69347
    },
    {
      "epoch": 6.613445281982421e-05,
      "step": 69347,
      "training_step_time": 7.314504384994507
    },
    {
      "epoch": 6.613540649414062e-05,
      "model_forward_time": 1.7920479774475098,
      "step": 69348
    },
    {
      "epoch": 6.613540649414062e-05,
      "step": 69348,
      "training_step_time": 7.684926986694336
    },
    {
      "epoch": 6.613636016845703e-05,
      "model_forward_time": 1.7234880924224854,
      "step": 69349
    },
    {
      "epoch": 6.613636016845703e-05,
      "step": 69349,
      "training_step_time": 8.45446252822876
    },
    {
      "epoch": 6.613731384277343e-05,
      "grad_norm": 0.08762223273515701,
      "learning_rate": 2.3559992898784454e-06,
      "loss": 0.0532,
      "step": 69350
    },
    {
      "epoch": 6.613731384277343e-05,
      "model_forward_time": 1.0076704025268555,
      "step": 69350
    },
    {
      "dynamics_loss_avg": 0.05793045964092016,
      "epoch": 6.613731384277343e-05,
      "step": 69350
    },
    {
      "action_loss_avg": 0.0013617668446386233,
      "epoch": 6.613731384277343e-05,
      "step": 69350
    },
    {
      "epoch": 6.613731384277343e-05,
      "step": 69350,
      "training_step_time": 8.686973333358765
    },
    {
      "epoch": 6.613826751708984e-05,
      "model_forward_time": 1.7334632873535156,
      "step": 69351
    },
    {
      "epoch": 6.613826751708984e-05,
      "step": 69351,
      "training_step_time": 8.195526123046875
    },
    {
      "epoch": 6.613922119140625e-05,
      "model_forward_time": 1.9007534980773926,
      "step": 69352
    },
    {
      "epoch": 6.613922119140625e-05,
      "step": 69352,
      "training_step_time": 7.811584711074829
    },
    {
      "epoch": 6.614017486572265e-05,
      "model_forward_time": 1.7225816249847412,
      "step": 69353
    },
    {
      "epoch": 6.614017486572265e-05,
      "step": 69353,
      "training_step_time": 9.626315832138062
    },
    {
      "epoch": 6.614112854003906e-05,
      "model_forward_time": 1.7206566333770752,
      "step": 69354
    },
    {
      "epoch": 6.614112854003906e-05,
      "step": 69354,
      "training_step_time": 7.542070388793945
    },
    {
      "epoch": 6.614208221435547e-05,
      "model_forward_time": 1.748436450958252,
      "step": 69355
    },
    {
      "epoch": 6.614208221435547e-05,
      "step": 69355,
      "training_step_time": 7.359691858291626
    },
    {
      "epoch": 6.614303588867187e-05,
      "model_forward_time": 1.7495644092559814,
      "step": 69356
    },
    {
      "epoch": 6.614303588867187e-05,
      "step": 69356,
      "training_step_time": 7.669715166091919
    },
    {
      "epoch": 6.614398956298828e-05,
      "model_forward_time": 1.8278632164001465,
      "step": 69357
    },
    {
      "epoch": 6.614398956298828e-05,
      "step": 69357,
      "training_step_time": 7.671335220336914
    },
    {
      "epoch": 6.614494323730469e-05,
      "model_forward_time": 1.754056453704834,
      "step": 69358
    },
    {
      "epoch": 6.614494323730469e-05,
      "step": 69358,
      "training_step_time": 7.696943759918213
    },
    {
      "epoch": 6.614589691162109e-05,
      "model_forward_time": 1.748779058456421,
      "step": 69359
    },
    {
      "epoch": 6.614589691162109e-05,
      "step": 69359,
      "training_step_time": 7.60052752494812
    },
    {
      "epoch": 6.61468505859375e-05,
      "grad_norm": 0.07579466700553894,
      "learning_rate": 2.354596057961367e-06,
      "loss": 0.0522,
      "step": 69360
    },
    {
      "epoch": 6.61468505859375e-05,
      "model_forward_time": 1.738309383392334,
      "step": 69360
    },
    {
      "dynamics_loss_avg": 0.032050991989672185,
      "epoch": 6.61468505859375e-05,
      "step": 69360
    },
    {
      "action_loss_avg": 0.0013034031784627587,
      "epoch": 6.61468505859375e-05,
      "step": 69360
    },
    {
      "epoch": 6.61468505859375e-05,
      "step": 69360,
      "training_step_time": 8.183830976486206
    },
    {
      "epoch": 6.614780426025391e-05,
      "model_forward_time": 2.2966902256011963,
      "step": 69361
    },
    {
      "epoch": 6.614780426025391e-05,
      "step": 69361,
      "training_step_time": 9.42407774925232
    },
    {
      "epoch": 6.614875793457031e-05,
      "model_forward_time": 1.7378325462341309,
      "step": 69362
    },
    {
      "epoch": 6.614875793457031e-05,
      "step": 69362,
      "training_step_time": 7.690675973892212
    },
    {
      "epoch": 6.614971160888672e-05,
      "model_forward_time": 1.7409887313842773,
      "step": 69363
    },
    {
      "epoch": 6.614971160888672e-05,
      "step": 69363,
      "training_step_time": 8.603793859481812
    },
    {
      "epoch": 6.615066528320313e-05,
      "model_forward_time": 1.741666316986084,
      "step": 69364
    },
    {
      "epoch": 6.615066528320313e-05,
      "step": 69364,
      "training_step_time": 7.372185468673706
    },
    {
      "epoch": 6.615161895751953e-05,
      "model_forward_time": 1.737694501876831,
      "step": 69365
    },
    {
      "epoch": 6.615161895751953e-05,
      "step": 69365,
      "training_step_time": 7.677216053009033
    },
    {
      "epoch": 6.615257263183594e-05,
      "model_forward_time": 1.8113152980804443,
      "step": 69366
    },
    {
      "epoch": 6.615257263183594e-05,
      "step": 69366,
      "training_step_time": 7.702224969863892
    },
    {
      "epoch": 6.615352630615235e-05,
      "model_forward_time": 1.718714952468872,
      "step": 69367
    },
    {
      "epoch": 6.615352630615235e-05,
      "step": 69367,
      "training_step_time": 7.574344158172607
    },
    {
      "epoch": 6.615447998046875e-05,
      "model_forward_time": 1.720062494277954,
      "step": 69368
    },
    {
      "epoch": 6.615447998046875e-05,
      "step": 69368,
      "training_step_time": 7.64825701713562
    },
    {
      "epoch": 6.615543365478516e-05,
      "model_forward_time": 1.7899072170257568,
      "step": 69369
    },
    {
      "epoch": 6.615543365478516e-05,
      "step": 69369,
      "training_step_time": 8.354954242706299
    },
    {
      "epoch": 6.615638732910157e-05,
      "grad_norm": 0.11253838986158371,
      "learning_rate": 2.3531931153416883e-06,
      "loss": 0.0573,
      "step": 69370
    },
    {
      "epoch": 6.615638732910157e-05,
      "model_forward_time": 2.3039283752441406,
      "step": 69370
    },
    {
      "dynamics_loss_avg": 0.0492713637650013,
      "epoch": 6.615638732910157e-05,
      "step": 69370
    },
    {
      "action_loss_avg": 0.0015991938766092063,
      "epoch": 6.615638732910157e-05,
      "step": 69370
    },
    {
      "epoch": 6.615638732910157e-05,
      "step": 69370,
      "training_step_time": 7.883515357971191
    },
    {
      "epoch": 6.615734100341796e-05,
      "model_forward_time": 1.723620891571045,
      "step": 69371
    },
    {
      "epoch": 6.615734100341796e-05,
      "step": 69371,
      "training_step_time": 7.799534797668457
    },
    {
      "epoch": 6.615829467773437e-05,
      "model_forward_time": 1.7322216033935547,
      "step": 69372
    },
    {
      "epoch": 6.615829467773437e-05,
      "step": 69372,
      "training_step_time": 8.644759178161621
    },
    {
      "epoch": 6.615924835205079e-05,
      "model_forward_time": 1.74861478805542,
      "step": 69373
    },
    {
      "epoch": 6.615924835205079e-05,
      "step": 69373,
      "training_step_time": 8.63909101486206
    },
    {
      "epoch": 6.616020202636718e-05,
      "model_forward_time": 1.739717721939087,
      "step": 69374
    },
    {
      "epoch": 6.616020202636718e-05,
      "step": 69374,
      "training_step_time": 7.796728610992432
    },
    {
      "epoch": 6.61611557006836e-05,
      "model_forward_time": 1.7366862297058105,
      "step": 69375
    },
    {
      "epoch": 6.61611557006836e-05,
      "step": 69375,
      "training_step_time": 7.436376333236694
    },
    {
      "epoch": 6.6162109375e-05,
      "model_forward_time": 1.7381200790405273,
      "step": 69376
    },
    {
      "epoch": 6.6162109375e-05,
      "step": 69376,
      "training_step_time": 7.632009029388428
    },
    {
      "epoch": 6.61630630493164e-05,
      "model_forward_time": 1.7375235557556152,
      "step": 69377
    },
    {
      "epoch": 6.61630630493164e-05,
      "step": 69377,
      "training_step_time": 8.682482719421387
    },
    {
      "epoch": 6.616401672363281e-05,
      "model_forward_time": 1.3780157566070557,
      "step": 69378
    },
    {
      "epoch": 6.616401672363281e-05,
      "step": 69378,
      "training_step_time": 7.696934223175049
    },
    {
      "epoch": 6.616497039794922e-05,
      "model_forward_time": 1.3749241828918457,
      "step": 69379
    },
    {
      "epoch": 6.616497039794922e-05,
      "step": 69379,
      "training_step_time": 8.771498203277588
    },
    {
      "epoch": 6.616592407226562e-05,
      "grad_norm": 0.08442680537700653,
      "learning_rate": 2.3517904621728294e-06,
      "loss": 0.0537,
      "step": 69380
    },
    {
      "epoch": 6.616592407226562e-05,
      "model_forward_time": 2.3015389442443848,
      "step": 69380
    },
    {
      "dynamics_loss_avg": 0.0386817323975265,
      "epoch": 6.616592407226562e-05,
      "step": 69380
    },
    {
      "action_loss_avg": 0.0015675673319492489,
      "epoch": 6.616592407226562e-05,
      "step": 69380
    },
    {
      "epoch": 6.616592407226562e-05,
      "step": 69380,
      "training_step_time": 9.816660165786743
    },
    {
      "epoch": 6.616687774658203e-05,
      "model_forward_time": 1.7390038967132568,
      "step": 69381
    },
    {
      "epoch": 6.616687774658203e-05,
      "step": 69381,
      "training_step_time": 7.598663330078125
    },
    {
      "epoch": 6.616783142089844e-05,
      "model_forward_time": 1.7164788246154785,
      "step": 69382
    },
    {
      "epoch": 6.616783142089844e-05,
      "step": 69382,
      "training_step_time": 7.725600242614746
    },
    {
      "epoch": 6.616878509521484e-05,
      "model_forward_time": 1.725886344909668,
      "step": 69383
    },
    {
      "epoch": 6.616878509521484e-05,
      "step": 69383,
      "training_step_time": 8.758111476898193
    },
    {
      "epoch": 6.616973876953125e-05,
      "model_forward_time": 1.7150559425354004,
      "step": 69384
    },
    {
      "epoch": 6.616973876953125e-05,
      "step": 69384,
      "training_step_time": 7.421243906021118
    },
    {
      "epoch": 6.617069244384766e-05,
      "model_forward_time": 1.7325758934020996,
      "step": 69385
    },
    {
      "epoch": 6.617069244384766e-05,
      "step": 69385,
      "training_step_time": 7.643566846847534
    },
    {
      "epoch": 6.617164611816406e-05,
      "model_forward_time": 1.7124240398406982,
      "step": 69386
    },
    {
      "epoch": 6.617164611816406e-05,
      "step": 69386,
      "training_step_time": 8.483070373535156
    },
    {
      "epoch": 6.617259979248047e-05,
      "model_forward_time": 1.7425167560577393,
      "step": 69387
    },
    {
      "epoch": 6.617259979248047e-05,
      "step": 69387,
      "training_step_time": 7.7179341316223145
    },
    {
      "epoch": 6.617355346679688e-05,
      "model_forward_time": 1.7291536331176758,
      "step": 69388
    },
    {
      "epoch": 6.617355346679688e-05,
      "step": 69388,
      "training_step_time": 8.450803995132446
    },
    {
      "epoch": 6.617450714111328e-05,
      "model_forward_time": 2.2656350135803223,
      "step": 69389
    },
    {
      "epoch": 6.617450714111328e-05,
      "step": 69389,
      "training_step_time": 8.402954816818237
    },
    {
      "epoch": 6.617546081542969e-05,
      "grad_norm": 0.10591951757669449,
      "learning_rate": 2.3503880986081855e-06,
      "loss": 0.0566,
      "step": 69390
    },
    {
      "epoch": 6.617546081542969e-05,
      "model_forward_time": 1.7146337032318115,
      "step": 69390
    },
    {
      "dynamics_loss_avg": 0.061172022670507434,
      "epoch": 6.617546081542969e-05,
      "step": 69390
    },
    {
      "action_loss_avg": 0.00210679195006378,
      "epoch": 6.617546081542969e-05,
      "step": 69390
    },
    {
      "epoch": 6.617546081542969e-05,
      "step": 69390,
      "training_step_time": 7.638291358947754
    },
    {
      "epoch": 6.61764144897461e-05,
      "model_forward_time": 1.714104413986206,
      "step": 69391
    },
    {
      "epoch": 6.61764144897461e-05,
      "step": 69391,
      "training_step_time": 7.706015110015869
    },
    {
      "epoch": 6.61773681640625e-05,
      "model_forward_time": 1.7207210063934326,
      "step": 69392
    },
    {
      "epoch": 6.61773681640625e-05,
      "step": 69392,
      "training_step_time": 7.455183029174805
    },
    {
      "epoch": 6.617832183837891e-05,
      "model_forward_time": 1.7150323390960693,
      "step": 69393
    },
    {
      "epoch": 6.617832183837891e-05,
      "step": 69393,
      "training_step_time": 7.41386342048645
    },
    {
      "epoch": 6.617927551269532e-05,
      "model_forward_time": 1.7321157455444336,
      "step": 69394
    },
    {
      "epoch": 6.617927551269532e-05,
      "step": 69394,
      "training_step_time": 7.618057489395142
    },
    {
      "epoch": 6.618022918701171e-05,
      "model_forward_time": 1.725184679031372,
      "step": 69395
    },
    {
      "epoch": 6.618022918701171e-05,
      "step": 69395,
      "training_step_time": 7.565523386001587
    },
    {
      "epoch": 6.618118286132813e-05,
      "model_forward_time": 1.7241251468658447,
      "step": 69396
    },
    {
      "epoch": 6.618118286132813e-05,
      "step": 69396,
      "training_step_time": 9.716962575912476
    },
    {
      "epoch": 6.618213653564454e-05,
      "model_forward_time": 1.7180147171020508,
      "step": 69397
    },
    {
      "epoch": 6.618213653564454e-05,
      "step": 69397,
      "training_step_time": 8.216074705123901
    },
    {
      "epoch": 6.618309020996093e-05,
      "model_forward_time": 2.2733495235443115,
      "step": 69398
    },
    {
      "epoch": 6.618309020996093e-05,
      "step": 69398,
      "training_step_time": 8.30977749824524
    },
    {
      "epoch": 6.618404388427734e-05,
      "model_forward_time": 1.722611427307129,
      "step": 69399
    },
    {
      "epoch": 6.618404388427734e-05,
      "step": 69399,
      "training_step_time": 8.561784505844116
    },
    {
      "epoch": 6.618499755859376e-05,
      "grad_norm": 0.10416267812252045,
      "learning_rate": 2.3489860248011136e-06,
      "loss": 0.0556,
      "step": 69400
    },
    {
      "epoch": 6.618499755859376e-05,
      "model_forward_time": 1.7179312705993652,
      "step": 69400
    },
    {
      "dynamics_loss_avg": 0.04036323968321085,
      "epoch": 6.618499755859376e-05,
      "step": 69400
    },
    {
      "action_loss_avg": 0.0009351577085908502,
      "epoch": 6.618499755859376e-05,
      "step": 69400
    },
    {
      "epoch": 6.618499755859376e-05,
      "step": 69400,
      "training_step_time": 8.743927001953125
    },
    {
      "epoch": 6.618595123291015e-05,
      "model_forward_time": 1.7262141704559326,
      "step": 69401
    },
    {
      "epoch": 6.618595123291015e-05,
      "step": 69401,
      "training_step_time": 7.468776226043701
    },
    {
      "epoch": 6.618690490722656e-05,
      "model_forward_time": 1.7276356220245361,
      "step": 69402
    },
    {
      "epoch": 6.618690490722656e-05,
      "step": 69402,
      "training_step_time": 7.746441602706909
    },
    {
      "epoch": 6.618785858154297e-05,
      "model_forward_time": 1.7236361503601074,
      "step": 69403
    },
    {
      "epoch": 6.618785858154297e-05,
      "step": 69403,
      "training_step_time": 7.661616563796997
    },
    {
      "epoch": 6.618881225585937e-05,
      "model_forward_time": 1.7018101215362549,
      "step": 69404
    },
    {
      "epoch": 6.618881225585937e-05,
      "step": 69404,
      "training_step_time": 7.16256856918335
    },
    {
      "epoch": 6.618976593017578e-05,
      "model_forward_time": 1.7087972164154053,
      "step": 69405
    },
    {
      "epoch": 6.618976593017578e-05,
      "step": 69405,
      "training_step_time": 7.700948715209961
    },
    {
      "epoch": 6.619071960449219e-05,
      "model_forward_time": 1.338728666305542,
      "step": 69406
    },
    {
      "epoch": 6.619071960449219e-05,
      "step": 69406,
      "training_step_time": 7.6401379108428955
    },
    {
      "epoch": 6.619167327880859e-05,
      "model_forward_time": 2.841848134994507,
      "step": 69407
    },
    {
      "epoch": 6.619167327880859e-05,
      "step": 69407,
      "training_step_time": 7.868528366088867
    },
    {
      "epoch": 6.6192626953125e-05,
      "model_forward_time": 1.7449116706848145,
      "step": 69408
    },
    {
      "epoch": 6.6192626953125e-05,
      "step": 69408,
      "training_step_time": 7.722178936004639
    },
    {
      "epoch": 6.619358062744141e-05,
      "model_forward_time": 1.708097219467163,
      "step": 69409
    },
    {
      "epoch": 6.619358062744141e-05,
      "step": 69409,
      "training_step_time": 8.457528114318848
    },
    {
      "epoch": 6.619453430175781e-05,
      "grad_norm": 0.13138966262340546,
      "learning_rate": 2.3475842409049465e-06,
      "loss": 0.058,
      "step": 69410
    },
    {
      "epoch": 6.619453430175781e-05,
      "model_forward_time": 1.7139782905578613,
      "step": 69410
    },
    {
      "dynamics_loss_avg": 0.07225439753383398,
      "epoch": 6.619453430175781e-05,
      "step": 69410
    },
    {
      "action_loss_avg": 0.0019238374952692538,
      "epoch": 6.619453430175781e-05,
      "step": 69410
    },
    {
      "epoch": 6.619453430175781e-05,
      "step": 69410,
      "training_step_time": 7.465577125549316
    },
    {
      "epoch": 6.619548797607422e-05,
      "model_forward_time": 1.7242183685302734,
      "step": 69411
    },
    {
      "epoch": 6.619548797607422e-05,
      "step": 69411,
      "training_step_time": 7.681456565856934
    },
    {
      "epoch": 6.619644165039063e-05,
      "model_forward_time": 1.7152385711669922,
      "step": 69412
    },
    {
      "epoch": 6.619644165039063e-05,
      "step": 69412,
      "training_step_time": 8.079686164855957
    },
    {
      "epoch": 6.619739532470703e-05,
      "model_forward_time": 1.7087574005126953,
      "step": 69413
    },
    {
      "epoch": 6.619739532470703e-05,
      "step": 69413,
      "training_step_time": 7.4021453857421875
    },
    {
      "epoch": 6.619834899902344e-05,
      "model_forward_time": 1.7145462036132812,
      "step": 69414
    },
    {
      "epoch": 6.619834899902344e-05,
      "step": 69414,
      "training_step_time": 7.713320970535278
    },
    {
      "epoch": 6.619930267333985e-05,
      "model_forward_time": 1.7069342136383057,
      "step": 69415
    },
    {
      "epoch": 6.619930267333985e-05,
      "step": 69415,
      "training_step_time": 10.507092952728271
    },
    {
      "epoch": 6.620025634765625e-05,
      "model_forward_time": 1.7124578952789307,
      "step": 69416
    },
    {
      "epoch": 6.620025634765625e-05,
      "step": 69416,
      "training_step_time": 8.285495042800903
    },
    {
      "epoch": 6.620121002197266e-05,
      "model_forward_time": 1.7187659740447998,
      "step": 69417
    },
    {
      "epoch": 6.620121002197266e-05,
      "step": 69417,
      "training_step_time": 8.50946831703186
    },
    {
      "epoch": 6.620216369628907e-05,
      "model_forward_time": 1.707585096359253,
      "step": 69418
    },
    {
      "epoch": 6.620216369628907e-05,
      "step": 69418,
      "training_step_time": 8.453704118728638
    },
    {
      "epoch": 6.620311737060547e-05,
      "model_forward_time": 1.7154710292816162,
      "step": 69419
    },
    {
      "epoch": 6.620311737060547e-05,
      "step": 69419,
      "training_step_time": 7.62494683265686
    },
    {
      "epoch": 6.620407104492188e-05,
      "grad_norm": 0.10403267294168472,
      "learning_rate": 2.346182747072975e-06,
      "loss": 0.0528,
      "step": 69420
    },
    {
      "epoch": 6.620407104492188e-05,
      "model_forward_time": 1.750868558883667,
      "step": 69420
    },
    {
      "dynamics_loss_avg": 0.05014126356691122,
      "epoch": 6.620407104492188e-05,
      "step": 69420
    },
    {
      "action_loss_avg": 0.0016166436776984482,
      "epoch": 6.620407104492188e-05,
      "step": 69420
    },
    {
      "epoch": 6.620407104492188e-05,
      "step": 69420,
      "training_step_time": 7.650460243225098
    },
    {
      "epoch": 6.620502471923829e-05,
      "model_forward_time": 1.706892490386963,
      "step": 69421
    },
    {
      "epoch": 6.620502471923829e-05,
      "step": 69421,
      "training_step_time": 7.482477188110352
    },
    {
      "epoch": 6.620597839355468e-05,
      "model_forward_time": 1.707123041152954,
      "step": 69422
    },
    {
      "epoch": 6.620597839355468e-05,
      "step": 69422,
      "training_step_time": 8.589409112930298
    },
    {
      "epoch": 6.62069320678711e-05,
      "model_forward_time": 1.7078094482421875,
      "step": 69423
    },
    {
      "epoch": 6.62069320678711e-05,
      "step": 69423,
      "training_step_time": 8.468042612075806
    },
    {
      "epoch": 6.62078857421875e-05,
      "model_forward_time": 1.7173125743865967,
      "step": 69424
    },
    {
      "epoch": 6.62078857421875e-05,
      "step": 69424,
      "training_step_time": 7.530799150466919
    },
    {
      "epoch": 6.62088394165039e-05,
      "model_forward_time": 1.7256741523742676,
      "step": 69425
    },
    {
      "epoch": 6.62088394165039e-05,
      "step": 69425,
      "training_step_time": 7.67329478263855
    },
    {
      "epoch": 6.620979309082031e-05,
      "model_forward_time": 2.904958486557007,
      "step": 69426
    },
    {
      "epoch": 6.620979309082031e-05,
      "step": 69426,
      "training_step_time": 9.726629257202148
    },
    {
      "epoch": 6.621074676513673e-05,
      "model_forward_time": 1.7231104373931885,
      "step": 69427
    },
    {
      "epoch": 6.621074676513673e-05,
      "step": 69427,
      "training_step_time": 8.274036884307861
    },
    {
      "epoch": 6.621170043945312e-05,
      "model_forward_time": 1.7205078601837158,
      "step": 69428
    },
    {
      "epoch": 6.621170043945312e-05,
      "step": 69428,
      "training_step_time": 7.634385585784912
    },
    {
      "epoch": 6.621265411376953e-05,
      "model_forward_time": 1.72385835647583,
      "step": 69429
    },
    {
      "epoch": 6.621265411376953e-05,
      "step": 69429,
      "training_step_time": 8.21876573562622
    },
    {
      "epoch": 6.621360778808594e-05,
      "grad_norm": 0.08390244841575623,
      "learning_rate": 2.344781543458472e-06,
      "loss": 0.0547,
      "step": 69430
    },
    {
      "epoch": 6.621360778808594e-05,
      "model_forward_time": 1.7224314212799072,
      "step": 69430
    },
    {
      "dynamics_loss_avg": 0.039915981329977514,
      "epoch": 6.621360778808594e-05,
      "step": 69430
    },
    {
      "action_loss_avg": 0.0012985602050321176,
      "epoch": 6.621360778808594e-05,
      "step": 69430
    },
    {
      "epoch": 6.621360778808594e-05,
      "step": 69430,
      "training_step_time": 7.534527063369751
    },
    {
      "epoch": 6.621456146240234e-05,
      "model_forward_time": 1.7205920219421387,
      "step": 69431
    },
    {
      "epoch": 6.621456146240234e-05,
      "step": 69431,
      "training_step_time": 7.60397744178772
    },
    {
      "epoch": 6.621551513671875e-05,
      "model_forward_time": 1.7249813079833984,
      "step": 69432
    },
    {
      "epoch": 6.621551513671875e-05,
      "step": 69432,
      "training_step_time": 7.646100044250488
    },
    {
      "epoch": 6.621646881103516e-05,
      "model_forward_time": 1.7266998291015625,
      "step": 69433
    },
    {
      "epoch": 6.621646881103516e-05,
      "step": 69433,
      "training_step_time": 8.564819574356079
    },
    {
      "epoch": 6.621742248535156e-05,
      "model_forward_time": 1.7102570533752441,
      "step": 69434
    },
    {
      "epoch": 6.621742248535156e-05,
      "step": 69434,
      "training_step_time": 7.614970922470093
    },
    {
      "epoch": 6.621837615966797e-05,
      "model_forward_time": 2.8695433139801025,
      "step": 69435
    },
    {
      "epoch": 6.621837615966797e-05,
      "step": 69435,
      "training_step_time": 9.33763313293457
    },
    {
      "epoch": 6.621932983398438e-05,
      "model_forward_time": 1.7246654033660889,
      "step": 69436
    },
    {
      "epoch": 6.621932983398438e-05,
      "step": 69436,
      "training_step_time": 9.600794792175293
    },
    {
      "epoch": 6.622028350830078e-05,
      "model_forward_time": 1.708266258239746,
      "step": 69437
    },
    {
      "epoch": 6.622028350830078e-05,
      "step": 69437,
      "training_step_time": 7.672549724578857
    },
    {
      "epoch": 6.622123718261719e-05,
      "model_forward_time": 1.7108368873596191,
      "step": 69438
    },
    {
      "epoch": 6.622123718261719e-05,
      "step": 69438,
      "training_step_time": 7.645178556442261
    },
    {
      "epoch": 6.622219085693359e-05,
      "model_forward_time": 1.7343122959136963,
      "step": 69439
    },
    {
      "epoch": 6.622219085693359e-05,
      "step": 69439,
      "training_step_time": 7.447599411010742
    },
    {
      "epoch": 6.622314453125e-05,
      "grad_norm": 0.10681919008493423,
      "learning_rate": 2.3433806302146654e-06,
      "loss": 0.0585,
      "step": 69440
    },
    {
      "epoch": 6.622314453125e-05,
      "model_forward_time": 1.7153408527374268,
      "step": 69440
    },
    {
      "dynamics_loss_avg": 0.0695985198020935,
      "epoch": 6.622314453125e-05,
      "step": 69440
    },
    {
      "action_loss_avg": 0.0020309432758949695,
      "epoch": 6.622314453125e-05,
      "step": 69440
    },
    {
      "epoch": 6.622314453125e-05,
      "step": 69440,
      "training_step_time": 7.533914804458618
    },
    {
      "epoch": 6.622409820556641e-05,
      "model_forward_time": 1.7091877460479736,
      "step": 69441
    },
    {
      "epoch": 6.622409820556641e-05,
      "step": 69441,
      "training_step_time": 8.435379981994629
    },
    {
      "epoch": 6.62250518798828e-05,
      "model_forward_time": 1.700577974319458,
      "step": 69442
    },
    {
      "epoch": 6.62250518798828e-05,
      "step": 69442,
      "training_step_time": 7.555043935775757
    },
    {
      "epoch": 6.622600555419922e-05,
      "model_forward_time": 1.7031149864196777,
      "step": 69443
    },
    {
      "epoch": 6.622600555419922e-05,
      "step": 69443,
      "training_step_time": 8.316802978515625
    },
    {
      "epoch": 6.622695922851563e-05,
      "model_forward_time": 2.865651845932007,
      "step": 69444
    },
    {
      "epoch": 6.622695922851563e-05,
      "step": 69444,
      "training_step_time": 9.218272686004639
    },
    {
      "epoch": 6.622791290283202e-05,
      "model_forward_time": 1.7173736095428467,
      "step": 69445
    },
    {
      "epoch": 6.622791290283202e-05,
      "step": 69445,
      "training_step_time": 8.692772626876831
    },
    {
      "epoch": 6.622886657714844e-05,
      "model_forward_time": 1.7319631576538086,
      "step": 69446
    },
    {
      "epoch": 6.622886657714844e-05,
      "step": 69446,
      "training_step_time": 7.674324035644531
    },
    {
      "epoch": 6.622982025146485e-05,
      "model_forward_time": 1.7074368000030518,
      "step": 69447
    },
    {
      "epoch": 6.622982025146485e-05,
      "step": 69447,
      "training_step_time": 7.636173725128174
    },
    {
      "epoch": 6.623077392578124e-05,
      "model_forward_time": 1.7292284965515137,
      "step": 69448
    },
    {
      "epoch": 6.623077392578124e-05,
      "step": 69448,
      "training_step_time": 7.443145513534546
    },
    {
      "epoch": 6.623172760009765e-05,
      "model_forward_time": 1.7089297771453857,
      "step": 69449
    },
    {
      "epoch": 6.623172760009765e-05,
      "step": 69449,
      "training_step_time": 7.5500171184539795
    },
    {
      "epoch": 6.623268127441407e-05,
      "grad_norm": 0.09890986979007721,
      "learning_rate": 2.3419800074947615e-06,
      "loss": 0.0542,
      "step": 69450
    },
    {
      "epoch": 6.623268127441407e-05,
      "model_forward_time": 1.7063498497009277,
      "step": 69450
    },
    {
      "dynamics_loss_avg": 0.04909509308636188,
      "epoch": 6.623268127441407e-05,
      "step": 69450
    },
    {
      "action_loss_avg": 0.0009671965846791864,
      "epoch": 6.623268127441407e-05,
      "step": 69450
    },
    {
      "epoch": 6.623268127441407e-05,
      "step": 69450,
      "training_step_time": 8.638794183731079
    },
    {
      "epoch": 6.623363494873046e-05,
      "model_forward_time": 1.707313060760498,
      "step": 69451
    },
    {
      "epoch": 6.623363494873046e-05,
      "step": 69451,
      "training_step_time": 7.46648383140564
    },
    {
      "epoch": 6.623458862304687e-05,
      "model_forward_time": 1.7067632675170898,
      "step": 69452
    },
    {
      "epoch": 6.623458862304687e-05,
      "step": 69452,
      "training_step_time": 7.59854793548584
    },
    {
      "epoch": 6.623554229736328e-05,
      "model_forward_time": 2.8216476440429688,
      "step": 69453
    },
    {
      "epoch": 6.623554229736328e-05,
      "step": 69453,
      "training_step_time": 9.841039657592773
    },
    {
      "epoch": 6.623649597167968e-05,
      "model_forward_time": 1.7129449844360352,
      "step": 69454
    },
    {
      "epoch": 6.623649597167968e-05,
      "step": 69454,
      "training_step_time": 7.641719102859497
    },
    {
      "epoch": 6.623744964599609e-05,
      "model_forward_time": 1.7467422485351562,
      "step": 69455
    },
    {
      "epoch": 6.623744964599609e-05,
      "step": 69455,
      "training_step_time": 7.651913642883301
    },
    {
      "epoch": 6.62384033203125e-05,
      "model_forward_time": 1.7216830253601074,
      "step": 69456
    },
    {
      "epoch": 6.62384033203125e-05,
      "step": 69456,
      "training_step_time": 7.639852285385132
    },
    {
      "epoch": 6.62393569946289e-05,
      "model_forward_time": 1.7277991771697998,
      "step": 69457
    },
    {
      "epoch": 6.62393569946289e-05,
      "step": 69457,
      "training_step_time": 7.413705348968506
    },
    {
      "epoch": 6.624031066894531e-05,
      "model_forward_time": 1.713458776473999,
      "step": 69458
    },
    {
      "epoch": 6.624031066894531e-05,
      "step": 69458,
      "training_step_time": 9.00912094116211
    },
    {
      "epoch": 6.624126434326172e-05,
      "model_forward_time": 1.7163689136505127,
      "step": 69459
    },
    {
      "epoch": 6.624126434326172e-05,
      "step": 69459,
      "training_step_time": 8.187642097473145
    },
    {
      "epoch": 6.624221801757812e-05,
      "grad_norm": 0.07399123907089233,
      "learning_rate": 2.340579675451925e-06,
      "loss": 0.0567,
      "step": 69460
    },
    {
      "epoch": 6.624221801757812e-05,
      "model_forward_time": 1.7059450149536133,
      "step": 69460
    },
    {
      "dynamics_loss_avg": 0.048634615167975424,
      "epoch": 6.624221801757812e-05,
      "step": 69460
    },
    {
      "action_loss_avg": 0.001085649020387791,
      "epoch": 6.624221801757812e-05,
      "step": 69460
    },
    {
      "epoch": 6.624221801757812e-05,
      "step": 69460,
      "training_step_time": 7.723932266235352
    },
    {
      "epoch": 6.624317169189453e-05,
      "model_forward_time": 1.7064967155456543,
      "step": 69461
    },
    {
      "epoch": 6.624317169189453e-05,
      "step": 69461,
      "training_step_time": 8.342602729797363
    },
    {
      "epoch": 6.624412536621094e-05,
      "model_forward_time": 2.419915199279785,
      "step": 69462
    },
    {
      "epoch": 6.624412536621094e-05,
      "step": 69462,
      "training_step_time": 8.524561882019043
    },
    {
      "epoch": 6.624507904052734e-05,
      "model_forward_time": 1.7235045433044434,
      "step": 69463
    },
    {
      "epoch": 6.624507904052734e-05,
      "step": 69463,
      "training_step_time": 8.712121486663818
    },
    {
      "epoch": 6.624603271484375e-05,
      "model_forward_time": 1.7551565170288086,
      "step": 69464
    },
    {
      "epoch": 6.624603271484375e-05,
      "step": 69464,
      "training_step_time": 10.374482870101929
    },
    {
      "epoch": 6.624698638916016e-05,
      "model_forward_time": 1.727637767791748,
      "step": 69465
    },
    {
      "epoch": 6.624698638916016e-05,
      "step": 69465,
      "training_step_time": 8.570416450500488
    },
    {
      "epoch": 6.624794006347656e-05,
      "model_forward_time": 1.7398250102996826,
      "step": 69466
    },
    {
      "epoch": 6.624794006347656e-05,
      "step": 69466,
      "training_step_time": 7.366097927093506
    },
    {
      "epoch": 6.624889373779297e-05,
      "model_forward_time": 1.725240707397461,
      "step": 69467
    },
    {
      "epoch": 6.624889373779297e-05,
      "step": 69467,
      "training_step_time": 8.77030086517334
    },
    {
      "epoch": 6.624984741210938e-05,
      "model_forward_time": 1.7292289733886719,
      "step": 69468
    },
    {
      "epoch": 6.624984741210938e-05,
      "step": 69468,
      "training_step_time": 7.583879232406616
    },
    {
      "epoch": 6.625080108642578e-05,
      "model_forward_time": 1.7312209606170654,
      "step": 69469
    },
    {
      "epoch": 6.625080108642578e-05,
      "step": 69469,
      "training_step_time": 7.535912036895752
    },
    {
      "epoch": 6.625175476074219e-05,
      "grad_norm": 0.0858091413974762,
      "learning_rate": 2.339179634239299e-06,
      "loss": 0.0542,
      "step": 69470
    },
    {
      "epoch": 6.625175476074219e-05,
      "model_forward_time": 1.7161147594451904,
      "step": 69470
    },
    {
      "dynamics_loss_avg": 0.05425448706373572,
      "epoch": 6.625175476074219e-05,
      "step": 69470
    },
    {
      "action_loss_avg": 0.0014336675172671675,
      "epoch": 6.625175476074219e-05,
      "step": 69470
    },
    {
      "epoch": 6.625175476074219e-05,
      "step": 69470,
      "training_step_time": 8.838000774383545
    },
    {
      "epoch": 6.62527084350586e-05,
      "model_forward_time": 1.7257375717163086,
      "step": 69471
    },
    {
      "epoch": 6.62527084350586e-05,
      "step": 69471,
      "training_step_time": 7.782018184661865
    },
    {
      "epoch": 6.6253662109375e-05,
      "model_forward_time": 2.863175868988037,
      "step": 69472
    },
    {
      "epoch": 6.6253662109375e-05,
      "step": 69472,
      "training_step_time": 8.644075393676758
    },
    {
      "epoch": 6.62546157836914e-05,
      "model_forward_time": 1.7407777309417725,
      "step": 69473
    },
    {
      "epoch": 6.62546157836914e-05,
      "step": 69473,
      "training_step_time": 8.688538312911987
    },
    {
      "epoch": 6.625556945800782e-05,
      "model_forward_time": 1.721153736114502,
      "step": 69474
    },
    {
      "epoch": 6.625556945800782e-05,
      "step": 69474,
      "training_step_time": 8.471700668334961
    },
    {
      "epoch": 6.625652313232421e-05,
      "model_forward_time": 1.7287507057189941,
      "step": 69475
    },
    {
      "epoch": 6.625652313232421e-05,
      "step": 69475,
      "training_step_time": 7.667774438858032
    },
    {
      "epoch": 6.625747680664062e-05,
      "model_forward_time": 1.7492055892944336,
      "step": 69476
    },
    {
      "epoch": 6.625747680664062e-05,
      "step": 69476,
      "training_step_time": 7.42525839805603
    },
    {
      "epoch": 6.625843048095703e-05,
      "model_forward_time": 1.7243826389312744,
      "step": 69477
    },
    {
      "epoch": 6.625843048095703e-05,
      "step": 69477,
      "training_step_time": 8.385364532470703
    },
    {
      "epoch": 6.625938415527343e-05,
      "model_forward_time": 1.7201995849609375,
      "step": 69478
    },
    {
      "epoch": 6.625938415527343e-05,
      "step": 69478,
      "training_step_time": 8.44770622253418
    },
    {
      "epoch": 6.626033782958984e-05,
      "model_forward_time": 1.7250146865844727,
      "step": 69479
    },
    {
      "epoch": 6.626033782958984e-05,
      "step": 69479,
      "training_step_time": 8.528692960739136
    },
    {
      "epoch": 6.626129150390625e-05,
      "grad_norm": 0.08820876479148865,
      "learning_rate": 2.337779884009984e-06,
      "loss": 0.0537,
      "step": 69480
    },
    {
      "epoch": 6.626129150390625e-05,
      "model_forward_time": 1.7203443050384521,
      "step": 69480
    },
    {
      "dynamics_loss_avg": 0.04542474392801523,
      "epoch": 6.626129150390625e-05,
      "step": 69480
    },
    {
      "action_loss_avg": 0.0014523245714372023,
      "epoch": 6.626129150390625e-05,
      "step": 69480
    },
    {
      "epoch": 6.626129150390625e-05,
      "step": 69480,
      "training_step_time": 8.671380758285522
    },
    {
      "epoch": 6.626224517822265e-05,
      "model_forward_time": 2.8449859619140625,
      "step": 69481
    },
    {
      "epoch": 6.626224517822265e-05,
      "step": 69481,
      "training_step_time": 8.303615093231201
    },
    {
      "epoch": 6.626319885253906e-05,
      "model_forward_time": 1.7311642169952393,
      "step": 69482
    },
    {
      "epoch": 6.626319885253906e-05,
      "step": 69482,
      "training_step_time": 8.51271677017212
    },
    {
      "epoch": 6.626415252685547e-05,
      "model_forward_time": 1.7245395183563232,
      "step": 69483
    },
    {
      "epoch": 6.626415252685547e-05,
      "step": 69483,
      "training_step_time": 7.735412836074829
    },
    {
      "epoch": 6.626510620117187e-05,
      "model_forward_time": 1.7192981243133545,
      "step": 69484
    },
    {
      "epoch": 6.626510620117187e-05,
      "step": 69484,
      "training_step_time": 7.597558498382568
    },
    {
      "epoch": 6.626605987548828e-05,
      "model_forward_time": 1.738224983215332,
      "step": 69485
    },
    {
      "epoch": 6.626605987548828e-05,
      "step": 69485,
      "training_step_time": 7.432304382324219
    },
    {
      "epoch": 6.626701354980469e-05,
      "model_forward_time": 1.7189583778381348,
      "step": 69486
    },
    {
      "epoch": 6.626701354980469e-05,
      "step": 69486,
      "training_step_time": 7.605651617050171
    },
    {
      "epoch": 6.626796722412109e-05,
      "model_forward_time": 1.7194252014160156,
      "step": 69487
    },
    {
      "epoch": 6.626796722412109e-05,
      "step": 69487,
      "training_step_time": 7.5747389793396
    },
    {
      "epoch": 6.62689208984375e-05,
      "model_forward_time": 1.7275140285491943,
      "step": 69488
    },
    {
      "epoch": 6.62689208984375e-05,
      "step": 69488,
      "training_step_time": 8.633864641189575
    },
    {
      "epoch": 6.626987457275391e-05,
      "model_forward_time": 1.718254566192627,
      "step": 69489
    },
    {
      "epoch": 6.626987457275391e-05,
      "step": 69489,
      "training_step_time": 8.633879661560059
    },
    {
      "epoch": 6.627082824707031e-05,
      "grad_norm": 0.11224652081727982,
      "learning_rate": 2.3363804249170626e-06,
      "loss": 0.0558,
      "step": 69490
    },
    {
      "epoch": 6.627082824707031e-05,
      "model_forward_time": 2.888310432434082,
      "step": 69490
    },
    {
      "dynamics_loss_avg": 0.0472631661221385,
      "epoch": 6.627082824707031e-05,
      "step": 69490
    },
    {
      "action_loss_avg": 0.0013385414436925202,
      "epoch": 6.627082824707031e-05,
      "step": 69490
    },
    {
      "epoch": 6.627082824707031e-05,
      "step": 69490,
      "training_step_time": 7.841659784317017
    },
    {
      "epoch": 6.627178192138672e-05,
      "model_forward_time": 1.7259957790374756,
      "step": 69491
    },
    {
      "epoch": 6.627178192138672e-05,
      "step": 69491,
      "training_step_time": 9.07044243812561
    },
    {
      "epoch": 6.627273559570313e-05,
      "model_forward_time": 1.727926254272461,
      "step": 69492
    },
    {
      "epoch": 6.627273559570313e-05,
      "step": 69492,
      "training_step_time": 7.7125632762908936
    },
    {
      "epoch": 6.627368927001953e-05,
      "model_forward_time": 1.723618984222412,
      "step": 69493
    },
    {
      "epoch": 6.627368927001953e-05,
      "step": 69493,
      "training_step_time": 8.665083408355713
    },
    {
      "epoch": 6.627464294433594e-05,
      "model_forward_time": 1.7622501850128174,
      "step": 69494
    },
    {
      "epoch": 6.627464294433594e-05,
      "step": 69494,
      "training_step_time": 7.435168027877808
    },
    {
      "epoch": 6.627559661865235e-05,
      "model_forward_time": 1.7183127403259277,
      "step": 69495
    },
    {
      "epoch": 6.627559661865235e-05,
      "step": 69495,
      "training_step_time": 7.517630577087402
    },
    {
      "epoch": 6.627655029296875e-05,
      "model_forward_time": 1.7213144302368164,
      "step": 69496
    },
    {
      "epoch": 6.627655029296875e-05,
      "step": 69496,
      "training_step_time": 7.670198678970337
    },
    {
      "epoch": 6.627750396728516e-05,
      "model_forward_time": 1.7212965488433838,
      "step": 69497
    },
    {
      "epoch": 6.627750396728516e-05,
      "step": 69497,
      "training_step_time": 7.526845216751099
    },
    {
      "epoch": 6.627845764160157e-05,
      "model_forward_time": 1.7203786373138428,
      "step": 69498
    },
    {
      "epoch": 6.627845764160157e-05,
      "step": 69498,
      "training_step_time": 8.416992902755737
    },
    {
      "epoch": 6.627941131591796e-05,
      "model_forward_time": 2.8571887016296387,
      "step": 69499
    },
    {
      "epoch": 6.627941131591796e-05,
      "step": 69499,
      "training_step_time": 8.383714199066162
    },
    {
      "epoch": 6.628036499023438e-05,
      "grad_norm": 0.10515649616718292,
      "learning_rate": 2.3349812571135704e-06,
      "loss": 0.0558,
      "step": 69500
    },
    {
      "epoch": 6.628036499023438e-05,
      "model_forward_time": 1.7268240451812744,
      "step": 69500
    },
    {
      "dynamics_loss_avg": 0.05751714501529932,
      "epoch": 6.628036499023438e-05,
      "step": 69500
    },
    {
      "action_loss_avg": 0.0011127311910968274,
      "epoch": 6.628036499023438e-05,
      "step": 69500
    },
    {
      "epoch": 6.628036499023438e-05,
      "step": 69500,
      "training_step_time": 10.074087858200073
    },
    {
      "epoch": 6.628131866455079e-05,
      "model_forward_time": 1.749361276626587,
      "step": 69501
    },
    {
      "epoch": 6.628131866455079e-05,
      "step": 69501,
      "training_step_time": 7.761494398117065
    },
    {
      "epoch": 6.628227233886718e-05,
      "model_forward_time": 1.722684621810913,
      "step": 69502
    },
    {
      "epoch": 6.628227233886718e-05,
      "step": 69502,
      "training_step_time": 7.643179178237915
    },
    {
      "epoch": 6.62832260131836e-05,
      "model_forward_time": 1.7468221187591553,
      "step": 69503
    },
    {
      "epoch": 6.62832260131836e-05,
      "step": 69503,
      "training_step_time": 7.367069482803345
    },
    {
      "epoch": 6.62841796875e-05,
      "model_forward_time": 1.7247545719146729,
      "step": 69504
    },
    {
      "epoch": 6.62841796875e-05,
      "step": 69504,
      "training_step_time": 7.594438076019287
    },
    {
      "epoch": 6.62851333618164e-05,
      "model_forward_time": 1.7159404754638672,
      "step": 69505
    },
    {
      "epoch": 6.62851333618164e-05,
      "step": 69505,
      "training_step_time": 7.597039222717285
    },
    {
      "epoch": 6.628608703613281e-05,
      "model_forward_time": 1.7188060283660889,
      "step": 69506
    },
    {
      "epoch": 6.628608703613281e-05,
      "step": 69506,
      "training_step_time": 7.579272270202637
    },
    {
      "epoch": 6.628704071044922e-05,
      "model_forward_time": 1.3480093479156494,
      "step": 69507
    },
    {
      "epoch": 6.628704071044922e-05,
      "step": 69507,
      "training_step_time": 8.88669204711914
    },
    {
      "epoch": 6.628799438476562e-05,
      "model_forward_time": 1.7214031219482422,
      "step": 69508
    },
    {
      "epoch": 6.628799438476562e-05,
      "step": 69508,
      "training_step_time": 7.828901529312134
    },
    {
      "epoch": 6.628894805908203e-05,
      "model_forward_time": 2.921251058578491,
      "step": 69509
    },
    {
      "epoch": 6.628894805908203e-05,
      "step": 69509,
      "training_step_time": 9.292684078216553
    },
    {
      "epoch": 6.628990173339844e-05,
      "grad_norm": 0.08828197419643402,
      "learning_rate": 2.333582380752523e-06,
      "loss": 0.0577,
      "step": 69510
    },
    {
      "epoch": 6.628990173339844e-05,
      "model_forward_time": 1.732191562652588,
      "step": 69510
    },
    {
      "dynamics_loss_avg": 0.050438433140516284,
      "epoch": 6.628990173339844e-05,
      "step": 69510
    },
    {
      "action_loss_avg": 0.0016966663679340854,
      "epoch": 6.628990173339844e-05,
      "step": 69510
    },
    {
      "epoch": 6.628990173339844e-05,
      "step": 69510,
      "training_step_time": 8.161387205123901
    },
    {
      "epoch": 6.629085540771484e-05,
      "model_forward_time": 1.7213091850280762,
      "step": 69511
    },
    {
      "epoch": 6.629085540771484e-05,
      "step": 69511,
      "training_step_time": 7.6923205852508545
    },
    {
      "epoch": 6.629180908203125e-05,
      "model_forward_time": 1.728776216506958,
      "step": 69512
    },
    {
      "epoch": 6.629180908203125e-05,
      "step": 69512,
      "training_step_time": 8.145559072494507
    },
    {
      "epoch": 6.629276275634766e-05,
      "model_forward_time": 1.7305552959442139,
      "step": 69513
    },
    {
      "epoch": 6.629276275634766e-05,
      "step": 69513,
      "training_step_time": 7.527297735214233
    },
    {
      "epoch": 6.629371643066406e-05,
      "model_forward_time": 1.3487567901611328,
      "step": 69514
    },
    {
      "epoch": 6.629371643066406e-05,
      "step": 69514,
      "training_step_time": 7.6203742027282715
    },
    {
      "epoch": 6.629467010498047e-05,
      "model_forward_time": 1.7199037075042725,
      "step": 69515
    },
    {
      "epoch": 6.629467010498047e-05,
      "step": 69515,
      "training_step_time": 8.444465398788452
    },
    {
      "epoch": 6.629562377929688e-05,
      "model_forward_time": 1.7486751079559326,
      "step": 69516
    },
    {
      "epoch": 6.629562377929688e-05,
      "step": 69516,
      "training_step_time": 7.595704793930054
    },
    {
      "epoch": 6.629657745361328e-05,
      "model_forward_time": 1.7232661247253418,
      "step": 69517
    },
    {
      "epoch": 6.629657745361328e-05,
      "step": 69517,
      "training_step_time": 8.515966892242432
    },
    {
      "epoch": 6.629753112792969e-05,
      "model_forward_time": 2.9571611881256104,
      "step": 69518
    },
    {
      "epoch": 6.629753112792969e-05,
      "step": 69518,
      "training_step_time": 8.9219970703125
    },
    {
      "epoch": 6.62984848022461e-05,
      "model_forward_time": 1.7179882526397705,
      "step": 69519
    },
    {
      "epoch": 6.62984848022461e-05,
      "step": 69519,
      "training_step_time": 7.614022254943848
    },
    {
      "epoch": 6.62994384765625e-05,
      "grad_norm": 0.10194453597068787,
      "learning_rate": 2.3321837959868964e-06,
      "loss": 0.0558,
      "step": 69520
    },
    {
      "epoch": 6.62994384765625e-05,
      "model_forward_time": 1.7175862789154053,
      "step": 69520
    },
    {
      "dynamics_loss_avg": 0.05610930696129799,
      "epoch": 6.62994384765625e-05,
      "step": 69520
    },
    {
      "action_loss_avg": 0.0015548231196589768,
      "epoch": 6.62994384765625e-05,
      "step": 69520
    },
    {
      "epoch": 6.62994384765625e-05,
      "step": 69520,
      "training_step_time": 8.284288167953491
    },
    {
      "epoch": 6.630039215087891e-05,
      "model_forward_time": 1.737117052078247,
      "step": 69521
    },
    {
      "epoch": 6.630039215087891e-05,
      "step": 69521,
      "training_step_time": 7.39555811882019
    },
    {
      "epoch": 6.630134582519532e-05,
      "model_forward_time": 1.7239751815795898,
      "step": 69522
    },
    {
      "epoch": 6.630134582519532e-05,
      "step": 69522,
      "training_step_time": 8.097488641738892
    },
    {
      "epoch": 6.630229949951172e-05,
      "model_forward_time": 1.7257285118103027,
      "step": 69523
    },
    {
      "epoch": 6.630229949951172e-05,
      "step": 69523,
      "training_step_time": 8.172940015792847
    },
    {
      "epoch": 6.630325317382813e-05,
      "model_forward_time": 1.7187590599060059,
      "step": 69524
    },
    {
      "epoch": 6.630325317382813e-05,
      "step": 69524,
      "training_step_time": 8.686184167861938
    },
    {
      "epoch": 6.630420684814454e-05,
      "model_forward_time": 1.7260468006134033,
      "step": 69525
    },
    {
      "epoch": 6.630420684814454e-05,
      "step": 69525,
      "training_step_time": 7.54873251914978
    },
    {
      "epoch": 6.630516052246093e-05,
      "model_forward_time": 1.7187645435333252,
      "step": 69526
    },
    {
      "epoch": 6.630516052246093e-05,
      "step": 69526,
      "training_step_time": 9.70034408569336
    },
    {
      "epoch": 6.630611419677734e-05,
      "model_forward_time": 2.899822473526001,
      "step": 69527
    },
    {
      "epoch": 6.630611419677734e-05,
      "step": 69527,
      "training_step_time": 7.810354232788086
    },
    {
      "epoch": 6.630706787109376e-05,
      "model_forward_time": 1.7245924472808838,
      "step": 69528
    },
    {
      "epoch": 6.630706787109376e-05,
      "step": 69528,
      "training_step_time": 7.646228075027466
    },
    {
      "epoch": 6.630802154541015e-05,
      "model_forward_time": 1.7155864238739014,
      "step": 69529
    },
    {
      "epoch": 6.630802154541015e-05,
      "step": 69529,
      "training_step_time": 7.732272624969482
    },
    {
      "epoch": 6.630897521972656e-05,
      "grad_norm": 0.10638488084077835,
      "learning_rate": 2.3307855029696388e-06,
      "loss": 0.0558,
      "step": 69530
    },
    {
      "epoch": 6.630897521972656e-05,
      "model_forward_time": 1.774784803390503,
      "step": 69530
    },
    {
      "dynamics_loss_avg": 0.03524897554889321,
      "epoch": 6.630897521972656e-05,
      "step": 69530
    },
    {
      "action_loss_avg": 0.0011414489010348916,
      "epoch": 6.630897521972656e-05,
      "step": 69530
    },
    {
      "epoch": 6.630897521972656e-05,
      "step": 69530,
      "training_step_time": 7.623909950256348
    },
    {
      "epoch": 6.630992889404297e-05,
      "model_forward_time": 1.7243309020996094,
      "step": 69531
    },
    {
      "epoch": 6.630992889404297e-05,
      "step": 69531,
      "training_step_time": 7.507193565368652
    },
    {
      "epoch": 6.631088256835937e-05,
      "model_forward_time": 1.714223861694336,
      "step": 69532
    },
    {
      "epoch": 6.631088256835937e-05,
      "step": 69532,
      "training_step_time": 8.469605684280396
    },
    {
      "epoch": 6.631183624267578e-05,
      "model_forward_time": 1.719167947769165,
      "step": 69533
    },
    {
      "epoch": 6.631183624267578e-05,
      "step": 69533,
      "training_step_time": 8.516701698303223
    },
    {
      "epoch": 6.63127899169922e-05,
      "model_forward_time": 1.3461148738861084,
      "step": 69534
    },
    {
      "epoch": 6.63127899169922e-05,
      "step": 69534,
      "training_step_time": 7.532415151596069
    },
    {
      "epoch": 6.631374359130859e-05,
      "model_forward_time": 1.7371275424957275,
      "step": 69535
    },
    {
      "epoch": 6.631374359130859e-05,
      "step": 69535,
      "training_step_time": 7.638887882232666
    },
    {
      "epoch": 6.6314697265625e-05,
      "model_forward_time": 2.867358446121216,
      "step": 69536
    },
    {
      "epoch": 6.6314697265625e-05,
      "step": 69536,
      "training_step_time": 7.803104877471924
    },
    {
      "epoch": 6.631565093994141e-05,
      "model_forward_time": 1.7250187397003174,
      "step": 69537
    },
    {
      "epoch": 6.631565093994141e-05,
      "step": 69537,
      "training_step_time": 9.514015436172485
    },
    {
      "epoch": 6.631660461425781e-05,
      "model_forward_time": 1.7162532806396484,
      "step": 69538
    },
    {
      "epoch": 6.631660461425781e-05,
      "step": 69538,
      "training_step_time": 8.6781485080719
    },
    {
      "epoch": 6.631755828857422e-05,
      "model_forward_time": 1.7373111248016357,
      "step": 69539
    },
    {
      "epoch": 6.631755828857422e-05,
      "step": 69539,
      "training_step_time": 7.477191209793091
    },
    {
      "epoch": 6.631851196289063e-05,
      "grad_norm": 0.08969065546989441,
      "learning_rate": 2.3293875018536648e-06,
      "loss": 0.0559,
      "step": 69540
    },
    {
      "epoch": 6.631851196289063e-05,
      "model_forward_time": 1.719672679901123,
      "step": 69540
    },
    {
      "dynamics_loss_avg": 0.05139171034097671,
      "epoch": 6.631851196289063e-05,
      "step": 69540
    },
    {
      "action_loss_avg": 0.0015109630767256021,
      "epoch": 6.631851196289063e-05,
      "step": 69540
    },
    {
      "epoch": 6.631851196289063e-05,
      "step": 69540,
      "training_step_time": 7.602452993392944
    },
    {
      "epoch": 6.631946563720703e-05,
      "model_forward_time": 1.7211127281188965,
      "step": 69541
    },
    {
      "epoch": 6.631946563720703e-05,
      "step": 69541,
      "training_step_time": 7.611677169799805
    },
    {
      "epoch": 6.632041931152344e-05,
      "model_forward_time": 1.72403883934021,
      "step": 69542
    },
    {
      "epoch": 6.632041931152344e-05,
      "step": 69542,
      "training_step_time": 7.545805931091309
    },
    {
      "epoch": 6.632137298583985e-05,
      "model_forward_time": 1.720201015472412,
      "step": 69543
    },
    {
      "epoch": 6.632137298583985e-05,
      "step": 69543,
      "training_step_time": 7.572435140609741
    },
    {
      "epoch": 6.632232666015625e-05,
      "model_forward_time": 1.730013370513916,
      "step": 69544
    },
    {
      "epoch": 6.632232666015625e-05,
      "step": 69544,
      "training_step_time": 7.702141046524048
    },
    {
      "epoch": 6.632328033447266e-05,
      "model_forward_time": 1.7249104976654053,
      "step": 69545
    },
    {
      "epoch": 6.632328033447266e-05,
      "step": 69545,
      "training_step_time": 8.83777117729187
    },
    {
      "epoch": 6.632423400878907e-05,
      "model_forward_time": 1.7218687534332275,
      "step": 69546
    },
    {
      "epoch": 6.632423400878907e-05,
      "step": 69546,
      "training_step_time": 7.7303786277771
    },
    {
      "epoch": 6.632518768310547e-05,
      "model_forward_time": 1.720026969909668,
      "step": 69547
    },
    {
      "epoch": 6.632518768310547e-05,
      "step": 69547,
      "training_step_time": 7.7089526653289795
    },
    {
      "epoch": 6.632614135742188e-05,
      "model_forward_time": 1.7245800495147705,
      "step": 69548
    },
    {
      "epoch": 6.632614135742188e-05,
      "step": 69548,
      "training_step_time": 7.632622003555298
    },
    {
      "epoch": 6.632709503173829e-05,
      "model_forward_time": 1.7261216640472412,
      "step": 69549
    },
    {
      "epoch": 6.632709503173829e-05,
      "step": 69549,
      "training_step_time": 8.103667736053467
    },
    {
      "epoch": 6.632804870605468e-05,
      "grad_norm": 0.09789519757032394,
      "learning_rate": 2.327989792791861e-06,
      "loss": 0.0544,
      "step": 69550
    },
    {
      "epoch": 6.632804870605468e-05,
      "model_forward_time": 1.7185990810394287,
      "step": 69550
    },
    {
      "dynamics_loss_avg": 0.04890948813408613,
      "epoch": 6.632804870605468e-05,
      "step": 69550
    },
    {
      "action_loss_avg": 0.001659964193822816,
      "epoch": 6.632804870605468e-05,
      "step": 69550
    },
    {
      "epoch": 6.632804870605468e-05,
      "step": 69550,
      "training_step_time": 7.646140813827515
    },
    {
      "epoch": 6.63290023803711e-05,
      "model_forward_time": 1.722259759902954,
      "step": 69551
    },
    {
      "epoch": 6.63290023803711e-05,
      "step": 69551,
      "training_step_time": 7.574565172195435
    },
    {
      "epoch": 6.63299560546875e-05,
      "model_forward_time": 1.724299669265747,
      "step": 69552
    },
    {
      "epoch": 6.63299560546875e-05,
      "step": 69552,
      "training_step_time": 7.529033422470093
    },
    {
      "epoch": 6.63309097290039e-05,
      "model_forward_time": 1.331115484237671,
      "step": 69553
    },
    {
      "epoch": 6.63309097290039e-05,
      "step": 69553,
      "training_step_time": 7.678978681564331
    },
    {
      "epoch": 6.633186340332031e-05,
      "model_forward_time": 1.7411341667175293,
      "step": 69554
    },
    {
      "epoch": 6.633186340332031e-05,
      "step": 69554,
      "training_step_time": 8.312792539596558
    },
    {
      "epoch": 6.633281707763673e-05,
      "model_forward_time": 2.9330575466156006,
      "step": 69555
    },
    {
      "epoch": 6.633281707763673e-05,
      "step": 69555,
      "training_step_time": 7.645476341247559
    },
    {
      "epoch": 6.633377075195312e-05,
      "model_forward_time": 1.7253026962280273,
      "step": 69556
    },
    {
      "epoch": 6.633377075195312e-05,
      "step": 69556,
      "training_step_time": 9.435260772705078
    },
    {
      "epoch": 6.633472442626953e-05,
      "model_forward_time": 1.7198245525360107,
      "step": 69557
    },
    {
      "epoch": 6.633472442626953e-05,
      "step": 69557,
      "training_step_time": 7.625439405441284
    },
    {
      "epoch": 6.633567810058594e-05,
      "model_forward_time": 1.7397196292877197,
      "step": 69558
    },
    {
      "epoch": 6.633567810058594e-05,
      "step": 69558,
      "training_step_time": 7.537424325942993
    },
    {
      "epoch": 6.633663177490234e-05,
      "model_forward_time": 1.7224457263946533,
      "step": 69559
    },
    {
      "epoch": 6.633663177490234e-05,
      "step": 69559,
      "training_step_time": 7.570937633514404
    },
    {
      "epoch": 6.633758544921875e-05,
      "grad_norm": 0.10263006389141083,
      "learning_rate": 2.326592375937074e-06,
      "loss": 0.0576,
      "step": 69560
    },
    {
      "epoch": 6.633758544921875e-05,
      "model_forward_time": 1.7189152240753174,
      "step": 69560
    },
    {
      "dynamics_loss_avg": 0.0522063110023737,
      "epoch": 6.633758544921875e-05,
      "step": 69560
    },
    {
      "action_loss_avg": 0.0013513971032807603,
      "epoch": 6.633758544921875e-05,
      "step": 69560
    },
    {
      "epoch": 6.633758544921875e-05,
      "step": 69560,
      "training_step_time": 7.586343050003052
    },
    {
      "epoch": 6.633853912353516e-05,
      "model_forward_time": 1.7218396663665771,
      "step": 69561
    },
    {
      "epoch": 6.633853912353516e-05,
      "step": 69561,
      "training_step_time": 8.64403223991394
    },
    {
      "epoch": 6.633949279785156e-05,
      "model_forward_time": 1.7265818119049072,
      "step": 69562
    },
    {
      "epoch": 6.633949279785156e-05,
      "step": 69562,
      "training_step_time": 8.276089191436768
    },
    {
      "epoch": 6.634044647216797e-05,
      "model_forward_time": 1.7449347972869873,
      "step": 69563
    },
    {
      "epoch": 6.634044647216797e-05,
      "step": 69563,
      "training_step_time": 7.689796686172485
    },
    {
      "epoch": 6.634140014648437e-05,
      "model_forward_time": 2.8889882564544678,
      "step": 69564
    },
    {
      "epoch": 6.634140014648437e-05,
      "step": 69564,
      "training_step_time": 7.634923934936523
    },
    {
      "epoch": 6.634235382080078e-05,
      "model_forward_time": 1.7155566215515137,
      "step": 69565
    },
    {
      "epoch": 6.634235382080078e-05,
      "step": 69565,
      "training_step_time": 8.291834115982056
    },
    {
      "epoch": 6.634330749511719e-05,
      "model_forward_time": 1.7210490703582764,
      "step": 69566
    },
    {
      "epoch": 6.634330749511719e-05,
      "step": 69566,
      "training_step_time": 7.705514430999756
    },
    {
      "epoch": 6.634426116943359e-05,
      "model_forward_time": 1.741257905960083,
      "step": 69567
    },
    {
      "epoch": 6.634426116943359e-05,
      "step": 69567,
      "training_step_time": 7.530077934265137
    },
    {
      "epoch": 6.634521484375e-05,
      "model_forward_time": 1.7270333766937256,
      "step": 69568
    },
    {
      "epoch": 6.634521484375e-05,
      "step": 69568,
      "training_step_time": 7.613816261291504
    },
    {
      "epoch": 6.634616851806641e-05,
      "model_forward_time": 1.7267377376556396,
      "step": 69569
    },
    {
      "epoch": 6.634616851806641e-05,
      "step": 69569,
      "training_step_time": 8.058327198028564
    },
    {
      "epoch": 6.63471221923828e-05,
      "grad_norm": 0.14062003791332245,
      "learning_rate": 2.325195251442125e-06,
      "loss": 0.0589,
      "step": 69570
    },
    {
      "epoch": 6.63471221923828e-05,
      "model_forward_time": 1.7231569290161133,
      "step": 69570
    },
    {
      "dynamics_loss_avg": 0.05862155891954899,
      "epoch": 6.63471221923828e-05,
      "step": 69570
    },
    {
      "action_loss_avg": 0.0017358849465381355,
      "epoch": 6.63471221923828e-05,
      "step": 69570
    },
    {
      "epoch": 6.63471221923828e-05,
      "step": 69570,
      "training_step_time": 8.409963846206665
    },
    {
      "epoch": 6.634807586669922e-05,
      "model_forward_time": 1.7218687534332275,
      "step": 69571
    },
    {
      "epoch": 6.634807586669922e-05,
      "step": 69571,
      "training_step_time": 8.122892618179321
    },
    {
      "epoch": 6.634902954101563e-05,
      "model_forward_time": 1.7364788055419922,
      "step": 69572
    },
    {
      "epoch": 6.634902954101563e-05,
      "step": 69572,
      "training_step_time": 8.684505701065063
    },
    {
      "epoch": 6.634998321533202e-05,
      "model_forward_time": 2.8426942825317383,
      "step": 69573
    },
    {
      "epoch": 6.634998321533202e-05,
      "step": 69573,
      "training_step_time": 9.348455429077148
    },
    {
      "epoch": 6.635093688964844e-05,
      "model_forward_time": 1.7219154834747314,
      "step": 69574
    },
    {
      "epoch": 6.635093688964844e-05,
      "step": 69574,
      "training_step_time": 8.320688486099243
    },
    {
      "epoch": 6.635189056396485e-05,
      "model_forward_time": 1.7160775661468506,
      "step": 69575
    },
    {
      "epoch": 6.635189056396485e-05,
      "step": 69575,
      "training_step_time": 8.791006326675415
    },
    {
      "epoch": 6.635284423828124e-05,
      "model_forward_time": 1.7431681156158447,
      "step": 69576
    },
    {
      "epoch": 6.635284423828124e-05,
      "step": 69576,
      "training_step_time": 7.49084210395813
    },
    {
      "epoch": 6.635379791259765e-05,
      "model_forward_time": 1.7263514995574951,
      "step": 69577
    },
    {
      "epoch": 6.635379791259765e-05,
      "step": 69577,
      "training_step_time": 7.597590923309326
    },
    {
      "epoch": 6.635475158691407e-05,
      "model_forward_time": 1.7250843048095703,
      "step": 69578
    },
    {
      "epoch": 6.635475158691407e-05,
      "step": 69578,
      "training_step_time": 7.54278826713562
    },
    {
      "epoch": 6.635570526123046e-05,
      "model_forward_time": 1.7216103076934814,
      "step": 69579
    },
    {
      "epoch": 6.635570526123046e-05,
      "step": 69579,
      "training_step_time": 7.474508285522461
    },
    {
      "epoch": 6.635665893554687e-05,
      "grad_norm": 0.08683576434850693,
      "learning_rate": 2.323798419459804e-06,
      "loss": 0.0564,
      "step": 69580
    },
    {
      "epoch": 6.635665893554687e-05,
      "model_forward_time": 1.7190032005310059,
      "step": 69580
    },
    {
      "dynamics_loss_avg": 0.05011162776499987,
      "epoch": 6.635665893554687e-05,
      "step": 69580
    },
    {
      "action_loss_avg": 0.0015847689006477594,
      "epoch": 6.635665893554687e-05,
      "step": 69580
    },
    {
      "epoch": 6.635665893554687e-05,
      "step": 69580,
      "training_step_time": 7.659400224685669
    },
    {
      "epoch": 6.635761260986328e-05,
      "model_forward_time": 1.7278754711151123,
      "step": 69581
    },
    {
      "epoch": 6.635761260986328e-05,
      "step": 69581,
      "training_step_time": 7.722502708435059
    },
    {
      "epoch": 6.635856628417968e-05,
      "model_forward_time": 2.844240188598633,
      "step": 69582
    },
    {
      "epoch": 6.635856628417968e-05,
      "step": 69582,
      "training_step_time": 8.555272817611694
    },
    {
      "epoch": 6.635951995849609e-05,
      "model_forward_time": 1.7222611904144287,
      "step": 69583
    },
    {
      "epoch": 6.635951995849609e-05,
      "step": 69583,
      "training_step_time": 8.522053003311157
    },
    {
      "epoch": 6.63604736328125e-05,
      "model_forward_time": 1.723210334777832,
      "step": 69584
    },
    {
      "epoch": 6.63604736328125e-05,
      "step": 69584,
      "training_step_time": 8.483208417892456
    },
    {
      "epoch": 6.63614273071289e-05,
      "model_forward_time": 1.726316213607788,
      "step": 69585
    },
    {
      "epoch": 6.63614273071289e-05,
      "step": 69585,
      "training_step_time": 8.087829828262329
    },
    {
      "epoch": 6.636238098144531e-05,
      "model_forward_time": 1.7195897102355957,
      "step": 69586
    },
    {
      "epoch": 6.636238098144531e-05,
      "step": 69586,
      "training_step_time": 7.649260759353638
    },
    {
      "epoch": 6.636333465576172e-05,
      "model_forward_time": 1.7259712219238281,
      "step": 69587
    },
    {
      "epoch": 6.636333465576172e-05,
      "step": 69587,
      "training_step_time": 8.562910795211792
    },
    {
      "epoch": 6.636428833007812e-05,
      "model_forward_time": 1.724487066268921,
      "step": 69588
    },
    {
      "epoch": 6.636428833007812e-05,
      "step": 69588,
      "training_step_time": 7.517428874969482
    },
    {
      "epoch": 6.636524200439453e-05,
      "model_forward_time": 1.7242183685302734,
      "step": 69589
    },
    {
      "epoch": 6.636524200439453e-05,
      "step": 69589,
      "training_step_time": 8.118854761123657
    },
    {
      "epoch": 6.636619567871094e-05,
      "grad_norm": 0.09287623316049576,
      "learning_rate": 2.3224018801428627e-06,
      "loss": 0.0605,
      "step": 69590
    },
    {
      "epoch": 6.636619567871094e-05,
      "model_forward_time": 1.7127082347869873,
      "step": 69590
    },
    {
      "dynamics_loss_avg": 0.04437712468206882,
      "epoch": 6.636619567871094e-05,
      "step": 69590
    },
    {
      "action_loss_avg": 0.001002523404895328,
      "epoch": 6.636619567871094e-05,
      "step": 69590
    },
    {
      "epoch": 6.636619567871094e-05,
      "step": 69590,
      "training_step_time": 8.77597713470459
    },
    {
      "epoch": 6.636714935302734e-05,
      "model_forward_time": 1.741377830505371,
      "step": 69591
    },
    {
      "epoch": 6.636714935302734e-05,
      "step": 69591,
      "training_step_time": 8.253401756286621
    },
    {
      "epoch": 6.636810302734375e-05,
      "model_forward_time": 1.7183358669281006,
      "step": 69592
    },
    {
      "epoch": 6.636810302734375e-05,
      "step": 69592,
      "training_step_time": 7.545431852340698
    },
    {
      "epoch": 6.636905670166016e-05,
      "model_forward_time": 1.7211992740631104,
      "step": 69593
    },
    {
      "epoch": 6.636905670166016e-05,
      "step": 69593,
      "training_step_time": 7.660253047943115
    },
    {
      "epoch": 6.637001037597656e-05,
      "model_forward_time": 1.700711965560913,
      "step": 69594
    },
    {
      "epoch": 6.637001037597656e-05,
      "step": 69594,
      "training_step_time": 7.596340179443359
    },
    {
      "epoch": 6.637096405029297e-05,
      "model_forward_time": 1.7434773445129395,
      "step": 69595
    },
    {
      "epoch": 6.637096405029297e-05,
      "step": 69595,
      "training_step_time": 8.454302310943604
    },
    {
      "epoch": 6.637191772460938e-05,
      "model_forward_time": 1.7239177227020264,
      "step": 69596
    },
    {
      "epoch": 6.637191772460938e-05,
      "step": 69596,
      "training_step_time": 7.702450752258301
    },
    {
      "epoch": 6.637287139892578e-05,
      "model_forward_time": 1.723219394683838,
      "step": 69597
    },
    {
      "epoch": 6.637287139892578e-05,
      "step": 69597,
      "training_step_time": 7.6244893074035645
    },
    {
      "epoch": 6.637382507324219e-05,
      "model_forward_time": 1.7067639827728271,
      "step": 69598
    },
    {
      "epoch": 6.637382507324219e-05,
      "step": 69598,
      "training_step_time": 7.643555641174316
    },
    {
      "epoch": 6.63747787475586e-05,
      "model_forward_time": 1.7097370624542236,
      "step": 69599
    },
    {
      "epoch": 6.63747787475586e-05,
      "step": 69599,
      "training_step_time": 9.33298635482788
    },
    {
      "epoch": 6.6375732421875e-05,
      "grad_norm": 0.2240375429391861,
      "learning_rate": 2.3210056336440274e-06,
      "loss": 0.058,
      "step": 69600
    },
    {
      "epoch": 6.6375732421875e-05,
      "model_forward_time": 1.7386023998260498,
      "step": 69600
    },
    {
      "dynamics_loss_avg": 0.05046792048960924,
      "epoch": 6.6375732421875e-05,
      "step": 69600
    },
    {
      "action_loss_avg": 0.0014744024141691626,
      "epoch": 6.6375732421875e-05,
      "step": 69600
    },
    {
      "epoch": 6.6375732421875e-05,
      "step": 69600,
      "training_step_time": 7.780888557434082
    },
    {
      "epoch": 6.63766860961914e-05,
      "model_forward_time": 2.8780949115753174,
      "step": 69601
    },
    {
      "epoch": 6.63766860961914e-05,
      "step": 69601,
      "training_step_time": 7.61436653137207
    },
    {
      "epoch": 6.637763977050782e-05,
      "model_forward_time": 1.708137035369873,
      "step": 69602
    },
    {
      "epoch": 6.637763977050782e-05,
      "step": 69602,
      "training_step_time": 8.9433114528656
    },
    {
      "epoch": 6.637859344482421e-05,
      "model_forward_time": 1.7068040370941162,
      "step": 69603
    },
    {
      "epoch": 6.637859344482421e-05,
      "step": 69603,
      "training_step_time": 7.685954809188843
    },
    {
      "epoch": 6.637954711914062e-05,
      "model_forward_time": 1.736098289489746,
      "step": 69604
    },
    {
      "epoch": 6.637954711914062e-05,
      "step": 69604,
      "training_step_time": 7.527218580245972
    },
    {
      "epoch": 6.638050079345704e-05,
      "model_forward_time": 1.7174274921417236,
      "step": 69605
    },
    {
      "epoch": 6.638050079345704e-05,
      "step": 69605,
      "training_step_time": 7.664458513259888
    },
    {
      "epoch": 6.638145446777343e-05,
      "model_forward_time": 1.703643560409546,
      "step": 69606
    },
    {
      "epoch": 6.638145446777343e-05,
      "step": 69606,
      "training_step_time": 9.151774883270264
    },
    {
      "epoch": 6.638240814208984e-05,
      "model_forward_time": 1.7023441791534424,
      "step": 69607
    },
    {
      "epoch": 6.638240814208984e-05,
      "step": 69607,
      "training_step_time": 7.5093183517456055
    },
    {
      "epoch": 6.638336181640625e-05,
      "model_forward_time": 1.7080397605895996,
      "step": 69608
    },
    {
      "epoch": 6.638336181640625e-05,
      "step": 69608,
      "training_step_time": 8.21995997428894
    },
    {
      "epoch": 6.638431549072265e-05,
      "model_forward_time": 1.731581449508667,
      "step": 69609
    },
    {
      "epoch": 6.638431549072265e-05,
      "step": 69609,
      "training_step_time": 7.7672483921051025
    },
    {
      "epoch": 6.638526916503906e-05,
      "grad_norm": 0.10941494256258011,
      "learning_rate": 2.3196096801159844e-06,
      "loss": 0.0581,
      "step": 69610
    },
    {
      "epoch": 6.638526916503906e-05,
      "model_forward_time": 2.8958237171173096,
      "step": 69610
    },
    {
      "dynamics_loss_avg": 0.052159950323402884,
      "epoch": 6.638526916503906e-05,
      "step": 69610
    },
    {
      "action_loss_avg": 0.0016528131469385698,
      "epoch": 6.638526916503906e-05,
      "step": 69610
    },
    {
      "epoch": 6.638526916503906e-05,
      "step": 69610,
      "training_step_time": 8.566581726074219
    },
    {
      "epoch": 6.638622283935547e-05,
      "model_forward_time": 1.7080225944519043,
      "step": 69611
    },
    {
      "epoch": 6.638622283935547e-05,
      "step": 69611,
      "training_step_time": 7.613093852996826
    },
    {
      "epoch": 6.638717651367187e-05,
      "model_forward_time": 1.3371949195861816,
      "step": 69612
    },
    {
      "epoch": 6.638717651367187e-05,
      "step": 69612,
      "training_step_time": 9.660646200180054
    },
    {
      "epoch": 6.638813018798828e-05,
      "model_forward_time": 1.7277603149414062,
      "step": 69613
    },
    {
      "epoch": 6.638813018798828e-05,
      "step": 69613,
      "training_step_time": 7.539782524108887
    },
    {
      "epoch": 6.638908386230469e-05,
      "model_forward_time": 1.7107203006744385,
      "step": 69614
    },
    {
      "epoch": 6.638908386230469e-05,
      "step": 69614,
      "training_step_time": 7.722048759460449
    },
    {
      "epoch": 6.639003753662109e-05,
      "model_forward_time": 1.7178905010223389,
      "step": 69615
    },
    {
      "epoch": 6.639003753662109e-05,
      "step": 69615,
      "training_step_time": 8.208128690719604
    },
    {
      "epoch": 6.63909912109375e-05,
      "model_forward_time": 1.7025372982025146,
      "step": 69616
    },
    {
      "epoch": 6.63909912109375e-05,
      "step": 69616,
      "training_step_time": 7.53409218788147
    },
    {
      "epoch": 6.639194488525391e-05,
      "model_forward_time": 1.7193725109100342,
      "step": 69617
    },
    {
      "epoch": 6.639194488525391e-05,
      "step": 69617,
      "training_step_time": 7.614625453948975
    },
    {
      "epoch": 6.639289855957031e-05,
      "model_forward_time": 1.743107557296753,
      "step": 69618
    },
    {
      "epoch": 6.639289855957031e-05,
      "step": 69618,
      "training_step_time": 7.652601957321167
    },
    {
      "epoch": 6.639385223388672e-05,
      "model_forward_time": 2.927089214324951,
      "step": 69619
    },
    {
      "epoch": 6.639385223388672e-05,
      "step": 69619,
      "training_step_time": 8.488228797912598
    },
    {
      "epoch": 6.639480590820313e-05,
      "grad_norm": 0.10561475902795792,
      "learning_rate": 2.318214019711401e-06,
      "loss": 0.0554,
      "step": 69620
    },
    {
      "epoch": 6.639480590820313e-05,
      "model_forward_time": 1.7235400676727295,
      "step": 69620
    },
    {
      "dynamics_loss_avg": 0.04906695391982794,
      "epoch": 6.639480590820313e-05,
      "step": 69620
    },
    {
      "action_loss_avg": 0.001826864088070579,
      "epoch": 6.639480590820313e-05,
      "step": 69620
    },
    {
      "epoch": 6.639480590820313e-05,
      "step": 69620,
      "training_step_time": 7.604925632476807
    },
    {
      "epoch": 6.639575958251953e-05,
      "model_forward_time": 1.7263150215148926,
      "step": 69621
    },
    {
      "epoch": 6.639575958251953e-05,
      "step": 69621,
      "training_step_time": 9.202400922775269
    },
    {
      "epoch": 6.639671325683594e-05,
      "model_forward_time": 1.7460556030273438,
      "step": 69622
    },
    {
      "epoch": 6.639671325683594e-05,
      "step": 69622,
      "training_step_time": 8.360291957855225
    },
    {
      "epoch": 6.639766693115235e-05,
      "model_forward_time": 1.7259280681610107,
      "step": 69623
    },
    {
      "epoch": 6.639766693115235e-05,
      "step": 69623,
      "training_step_time": 8.6121666431427
    },
    {
      "epoch": 6.639862060546875e-05,
      "model_forward_time": 1.7302160263061523,
      "step": 69624
    },
    {
      "epoch": 6.639862060546875e-05,
      "step": 69624,
      "training_step_time": 7.545745372772217
    },
    {
      "epoch": 6.639957427978516e-05,
      "model_forward_time": 1.7271645069122314,
      "step": 69625
    },
    {
      "epoch": 6.639957427978516e-05,
      "step": 69625,
      "training_step_time": 7.486353158950806
    },
    {
      "epoch": 6.640052795410157e-05,
      "model_forward_time": 1.7287933826446533,
      "step": 69626
    },
    {
      "epoch": 6.640052795410157e-05,
      "step": 69626,
      "training_step_time": 8.623127698898315
    },
    {
      "epoch": 6.640148162841796e-05,
      "model_forward_time": 1.7289924621582031,
      "step": 69627
    },
    {
      "epoch": 6.640148162841796e-05,
      "step": 69627,
      "training_step_time": 8.536975383758545
    },
    {
      "epoch": 6.640243530273438e-05,
      "model_forward_time": 2.895580768585205,
      "step": 69628
    },
    {
      "epoch": 6.640243530273438e-05,
      "step": 69628,
      "training_step_time": 8.566904544830322
    },
    {
      "epoch": 6.640338897705079e-05,
      "model_forward_time": 1.7238976955413818,
      "step": 69629
    },
    {
      "epoch": 6.640338897705079e-05,
      "step": 69629,
      "training_step_time": 7.595422983169556
    },
    {
      "epoch": 6.640434265136718e-05,
      "grad_norm": 0.09739143401384354,
      "learning_rate": 2.3168186525828983e-06,
      "loss": 0.0559,
      "step": 69630
    },
    {
      "epoch": 6.640434265136718e-05,
      "model_forward_time": 1.7271149158477783,
      "step": 69630
    },
    {
      "dynamics_loss_avg": 0.054065341129899025,
      "epoch": 6.640434265136718e-05,
      "step": 69630
    },
    {
      "action_loss_avg": 0.0013302598585141824,
      "epoch": 6.640434265136718e-05,
      "step": 69630
    },
    {
      "epoch": 6.640434265136718e-05,
      "step": 69630,
      "training_step_time": 8.720840454101562
    },
    {
      "epoch": 6.64052963256836e-05,
      "model_forward_time": 1.7407214641571045,
      "step": 69631
    },
    {
      "epoch": 6.64052963256836e-05,
      "step": 69631,
      "training_step_time": 7.60936713218689
    },
    {
      "epoch": 6.640625e-05,
      "model_forward_time": 1.6989715099334717,
      "step": 69632
    },
    {
      "epoch": 6.640625e-05,
      "step": 69632,
      "training_step_time": 7.583014726638794
    },
    {
      "epoch": 6.64072036743164e-05,
      "model_forward_time": 1.7065410614013672,
      "step": 69633
    },
    {
      "epoch": 6.64072036743164e-05,
      "step": 69633,
      "training_step_time": 8.04125428199768
    },
    {
      "epoch": 6.640815734863281e-05,
      "model_forward_time": 1.7069077491760254,
      "step": 69634
    },
    {
      "epoch": 6.640815734863281e-05,
      "step": 69634,
      "training_step_time": 7.480439186096191
    },
    {
      "epoch": 6.640911102294922e-05,
      "model_forward_time": 1.7082710266113281,
      "step": 69635
    },
    {
      "epoch": 6.640911102294922e-05,
      "step": 69635,
      "training_step_time": 7.635576009750366
    },
    {
      "epoch": 6.641006469726562e-05,
      "model_forward_time": 1.7092640399932861,
      "step": 69636
    },
    {
      "epoch": 6.641006469726562e-05,
      "step": 69636,
      "training_step_time": 8.742347955703735
    },
    {
      "epoch": 6.641101837158203e-05,
      "model_forward_time": 2.8089778423309326,
      "step": 69637
    },
    {
      "epoch": 6.641101837158203e-05,
      "step": 69637,
      "training_step_time": 8.848949432373047
    },
    {
      "epoch": 6.641197204589844e-05,
      "model_forward_time": 1.703230619430542,
      "step": 69638
    },
    {
      "epoch": 6.641197204589844e-05,
      "step": 69638,
      "training_step_time": 7.59502911567688
    },
    {
      "epoch": 6.641292572021484e-05,
      "model_forward_time": 1.7085392475128174,
      "step": 69639
    },
    {
      "epoch": 6.641292572021484e-05,
      "step": 69639,
      "training_step_time": 8.555966138839722
    },
    {
      "epoch": 6.641387939453125e-05,
      "grad_norm": 0.09920988976955414,
      "learning_rate": 2.315423578883076e-06,
      "loss": 0.0544,
      "step": 69640
    },
    {
      "epoch": 6.641387939453125e-05,
      "model_forward_time": 1.7050056457519531,
      "step": 69640
    },
    {
      "dynamics_loss_avg": 0.06466747932136059,
      "epoch": 6.641387939453125e-05,
      "step": 69640
    },
    {
      "action_loss_avg": 0.0012124138767831028,
      "epoch": 6.641387939453125e-05,
      "step": 69640
    },
    {
      "epoch": 6.641387939453125e-05,
      "step": 69640,
      "training_step_time": 7.720005035400391
    },
    {
      "epoch": 6.641483306884766e-05,
      "model_forward_time": 1.7323241233825684,
      "step": 69641
    },
    {
      "epoch": 6.641483306884766e-05,
      "step": 69641,
      "training_step_time": 8.055638074874878
    },
    {
      "epoch": 6.641578674316406e-05,
      "model_forward_time": 1.7095680236816406,
      "step": 69642
    },
    {
      "epoch": 6.641578674316406e-05,
      "step": 69642,
      "training_step_time": 8.40027379989624
    },
    {
      "epoch": 6.641674041748047e-05,
      "model_forward_time": 1.7123780250549316,
      "step": 69643
    },
    {
      "epoch": 6.641674041748047e-05,
      "step": 69643,
      "training_step_time": 7.521949291229248
    },
    {
      "epoch": 6.641769409179688e-05,
      "model_forward_time": 1.3343961238861084,
      "step": 69644
    },
    {
      "epoch": 6.641769409179688e-05,
      "step": 69644,
      "training_step_time": 7.546401023864746
    },
    {
      "epoch": 6.641864776611328e-05,
      "model_forward_time": 1.7103040218353271,
      "step": 69645
    },
    {
      "epoch": 6.641864776611328e-05,
      "step": 69645,
      "training_step_time": 7.567617893218994
    },
    {
      "epoch": 6.641960144042969e-05,
      "model_forward_time": 1.7376554012298584,
      "step": 69646
    },
    {
      "epoch": 6.641960144042969e-05,
      "step": 69646,
      "training_step_time": 7.742063760757446
    },
    {
      "epoch": 6.64205551147461e-05,
      "model_forward_time": 2.894505262374878,
      "step": 69647
    },
    {
      "epoch": 6.64205551147461e-05,
      "step": 69647,
      "training_step_time": 7.660656213760376
    },
    {
      "epoch": 6.64215087890625e-05,
      "model_forward_time": 1.7140710353851318,
      "step": 69648
    },
    {
      "epoch": 6.64215087890625e-05,
      "step": 69648,
      "training_step_time": 8.654821872711182
    },
    {
      "epoch": 6.642246246337891e-05,
      "model_forward_time": 1.7043662071228027,
      "step": 69649
    },
    {
      "epoch": 6.642246246337891e-05,
      "step": 69649,
      "training_step_time": 8.457178115844727
    },
    {
      "epoch": 6.642341613769532e-05,
      "grad_norm": 0.08538362383842468,
      "learning_rate": 2.314028798764493e-06,
      "loss": 0.0554,
      "step": 69650
    },
    {
      "epoch": 6.642341613769532e-05,
      "model_forward_time": 1.7244229316711426,
      "step": 69650
    },
    {
      "dynamics_loss_avg": 0.05380776170641184,
      "epoch": 6.642341613769532e-05,
      "step": 69650
    },
    {
      "action_loss_avg": 0.0016052633698564022,
      "epoch": 6.642341613769532e-05,
      "step": 69650
    },
    {
      "epoch": 6.642341613769532e-05,
      "step": 69650,
      "training_step_time": 7.485274076461792
    },
    {
      "epoch": 6.642436981201172e-05,
      "model_forward_time": 1.7184531688690186,
      "step": 69651
    },
    {
      "epoch": 6.642436981201172e-05,
      "step": 69651,
      "training_step_time": 8.070730209350586
    },
    {
      "epoch": 6.642532348632813e-05,
      "model_forward_time": 1.7160916328430176,
      "step": 69652
    },
    {
      "epoch": 6.642532348632813e-05,
      "step": 69652,
      "training_step_time": 7.576709985733032
    },
    {
      "epoch": 6.642627716064454e-05,
      "model_forward_time": 1.7083947658538818,
      "step": 69653
    },
    {
      "epoch": 6.642627716064454e-05,
      "step": 69653,
      "training_step_time": 8.345170736312866
    },
    {
      "epoch": 6.642723083496093e-05,
      "model_forward_time": 1.718613624572754,
      "step": 69654
    },
    {
      "epoch": 6.642723083496093e-05,
      "step": 69654,
      "training_step_time": 7.618007183074951
    },
    {
      "epoch": 6.642818450927734e-05,
      "model_forward_time": 1.734856367111206,
      "step": 69655
    },
    {
      "epoch": 6.642818450927734e-05,
      "step": 69655,
      "training_step_time": 8.808912515640259
    },
    {
      "epoch": 6.642913818359376e-05,
      "model_forward_time": 2.9486522674560547,
      "step": 69656
    },
    {
      "epoch": 6.642913818359376e-05,
      "step": 69656,
      "training_step_time": 8.243366718292236
    },
    {
      "epoch": 6.643009185791015e-05,
      "model_forward_time": 1.7099220752716064,
      "step": 69657
    },
    {
      "epoch": 6.643009185791015e-05,
      "step": 69657,
      "training_step_time": 8.395576238632202
    },
    {
      "epoch": 6.643104553222656e-05,
      "model_forward_time": 1.7146456241607666,
      "step": 69658
    },
    {
      "epoch": 6.643104553222656e-05,
      "step": 69658,
      "training_step_time": 7.669687747955322
    },
    {
      "epoch": 6.643199920654297e-05,
      "model_forward_time": 1.7287695407867432,
      "step": 69659
    },
    {
      "epoch": 6.643199920654297e-05,
      "step": 69659,
      "training_step_time": 7.510466575622559
    },
    {
      "epoch": 6.643295288085937e-05,
      "grad_norm": 0.07763536274433136,
      "learning_rate": 2.3126343123796834e-06,
      "loss": 0.0583,
      "step": 69660
    },
    {
      "epoch": 6.643295288085937e-05,
      "model_forward_time": 1.7058017253875732,
      "step": 69660
    },
    {
      "dynamics_loss_avg": 0.05447374936193228,
      "epoch": 6.643295288085937e-05,
      "step": 69660
    },
    {
      "action_loss_avg": 0.0011427177320001646,
      "epoch": 6.643295288085937e-05,
      "step": 69660
    },
    {
      "epoch": 6.643295288085937e-05,
      "step": 69660,
      "training_step_time": 7.55861759185791
    },
    {
      "epoch": 6.643390655517578e-05,
      "model_forward_time": 1.7127304077148438,
      "step": 69661
    },
    {
      "epoch": 6.643390655517578e-05,
      "step": 69661,
      "training_step_time": 8.548771619796753
    },
    {
      "epoch": 6.64348602294922e-05,
      "model_forward_time": 1.7125141620635986,
      "step": 69662
    },
    {
      "epoch": 6.64348602294922e-05,
      "step": 69662,
      "training_step_time": 8.41282320022583
    },
    {
      "epoch": 6.643581390380859e-05,
      "model_forward_time": 1.7089059352874756,
      "step": 69663
    },
    {
      "epoch": 6.643581390380859e-05,
      "step": 69663,
      "training_step_time": 8.57031798362732
    },
    {
      "epoch": 6.6436767578125e-05,
      "model_forward_time": 1.7389838695526123,
      "step": 69664
    },
    {
      "epoch": 6.6436767578125e-05,
      "step": 69664,
      "training_step_time": 9.238795280456543
    },
    {
      "epoch": 6.643772125244141e-05,
      "model_forward_time": 2.880185604095459,
      "step": 69665
    },
    {
      "epoch": 6.643772125244141e-05,
      "step": 69665,
      "training_step_time": 7.769856214523315
    },
    {
      "epoch": 6.643867492675781e-05,
      "model_forward_time": 1.7185673713684082,
      "step": 69666
    },
    {
      "epoch": 6.643867492675781e-05,
      "step": 69666,
      "training_step_time": 7.546017646789551
    },
    {
      "epoch": 6.643962860107422e-05,
      "model_forward_time": 1.707475185394287,
      "step": 69667
    },
    {
      "epoch": 6.643962860107422e-05,
      "step": 69667,
      "training_step_time": 7.744427442550659
    },
    {
      "epoch": 6.644058227539063e-05,
      "model_forward_time": 1.7284736633300781,
      "step": 69668
    },
    {
      "epoch": 6.644058227539063e-05,
      "step": 69668,
      "training_step_time": 8.337013959884644
    },
    {
      "epoch": 6.644153594970703e-05,
      "model_forward_time": 1.7247645854949951,
      "step": 69669
    },
    {
      "epoch": 6.644153594970703e-05,
      "step": 69669,
      "training_step_time": 7.5717926025390625
    },
    {
      "epoch": 6.644248962402344e-05,
      "grad_norm": 0.09172878414392471,
      "learning_rate": 2.3112401198811416e-06,
      "loss": 0.0563,
      "step": 69670
    },
    {
      "epoch": 6.644248962402344e-05,
      "model_forward_time": 1.7070398330688477,
      "step": 69670
    },
    {
      "dynamics_loss_avg": 0.05497704837471247,
      "epoch": 6.644248962402344e-05,
      "step": 69670
    },
    {
      "action_loss_avg": 0.001228278299095109,
      "epoch": 6.644248962402344e-05,
      "step": 69670
    },
    {
      "epoch": 6.644248962402344e-05,
      "step": 69670,
      "training_step_time": 7.543474197387695
    },
    {
      "epoch": 6.644344329833985e-05,
      "model_forward_time": 1.7144534587860107,
      "step": 69671
    },
    {
      "epoch": 6.644344329833985e-05,
      "step": 69671,
      "training_step_time": 7.603545665740967
    },
    {
      "epoch": 6.644439697265625e-05,
      "model_forward_time": 1.7301216125488281,
      "step": 69672
    },
    {
      "epoch": 6.644439697265625e-05,
      "step": 69672,
      "training_step_time": 8.412920475006104
    },
    {
      "epoch": 6.644535064697266e-05,
      "model_forward_time": 1.7382049560546875,
      "step": 69673
    },
    {
      "epoch": 6.644535064697266e-05,
      "step": 69673,
      "training_step_time": 8.650739431381226
    },
    {
      "epoch": 6.644630432128907e-05,
      "model_forward_time": 2.901592254638672,
      "step": 69674
    },
    {
      "epoch": 6.644630432128907e-05,
      "step": 69674,
      "training_step_time": 7.7705018520355225
    },
    {
      "epoch": 6.644725799560547e-05,
      "model_forward_time": 1.7276930809020996,
      "step": 69675
    },
    {
      "epoch": 6.644725799560547e-05,
      "step": 69675,
      "training_step_time": 8.312053918838501
    },
    {
      "epoch": 6.644821166992188e-05,
      "model_forward_time": 1.7153384685516357,
      "step": 69676
    },
    {
      "epoch": 6.644821166992188e-05,
      "step": 69676,
      "training_step_time": 8.47284483909607
    },
    {
      "epoch": 6.644916534423829e-05,
      "model_forward_time": 1.7359693050384521,
      "step": 69677
    },
    {
      "epoch": 6.644916534423829e-05,
      "step": 69677,
      "training_step_time": 7.601906776428223
    },
    {
      "epoch": 6.645011901855468e-05,
      "model_forward_time": 1.7054178714752197,
      "step": 69678
    },
    {
      "epoch": 6.645011901855468e-05,
      "step": 69678,
      "training_step_time": 7.528067588806152
    },
    {
      "epoch": 6.64510726928711e-05,
      "model_forward_time": 1.7186975479125977,
      "step": 69679
    },
    {
      "epoch": 6.64510726928711e-05,
      "step": 69679,
      "training_step_time": 8.56242847442627
    },
    {
      "epoch": 6.64520263671875e-05,
      "grad_norm": 0.09001964330673218,
      "learning_rate": 2.309846221421342e-06,
      "loss": 0.0564,
      "step": 69680
    },
    {
      "epoch": 6.64520263671875e-05,
      "model_forward_time": 1.7178497314453125,
      "step": 69680
    },
    {
      "dynamics_loss_avg": 0.05467351172119379,
      "epoch": 6.64520263671875e-05,
      "step": 69680
    },
    {
      "action_loss_avg": 0.0011851911258418113,
      "epoch": 6.64520263671875e-05,
      "step": 69680
    },
    {
      "epoch": 6.64520263671875e-05,
      "step": 69680,
      "training_step_time": 7.74138617515564
    },
    {
      "epoch": 6.64529800415039e-05,
      "model_forward_time": 1.726912021636963,
      "step": 69681
    },
    {
      "epoch": 6.64529800415039e-05,
      "step": 69681,
      "training_step_time": 8.585530519485474
    },
    {
      "epoch": 6.645393371582031e-05,
      "model_forward_time": 1.0341451168060303,
      "step": 69682
    },
    {
      "epoch": 6.645393371582031e-05,
      "step": 69682,
      "training_step_time": 10.258623600006104
    },
    {
      "epoch": 6.645488739013673e-05,
      "model_forward_time": 1.7110397815704346,
      "step": 69683
    },
    {
      "epoch": 6.645488739013673e-05,
      "step": 69683,
      "training_step_time": 8.2839515209198
    },
    {
      "epoch": 6.645584106445312e-05,
      "model_forward_time": 1.7376980781555176,
      "step": 69684
    },
    {
      "epoch": 6.645584106445312e-05,
      "step": 69684,
      "training_step_time": 8.341256141662598
    },
    {
      "epoch": 6.645679473876953e-05,
      "model_forward_time": 1.3404450416564941,
      "step": 69685
    },
    {
      "epoch": 6.645679473876953e-05,
      "step": 69685,
      "training_step_time": 8.292290449142456
    },
    {
      "epoch": 6.645774841308594e-05,
      "model_forward_time": 1.7226130962371826,
      "step": 69686
    },
    {
      "epoch": 6.645774841308594e-05,
      "step": 69686,
      "training_step_time": 9.527905702590942
    },
    {
      "epoch": 6.645870208740234e-05,
      "model_forward_time": 1.7239336967468262,
      "step": 69687
    },
    {
      "epoch": 6.645870208740234e-05,
      "step": 69687,
      "training_step_time": 7.543016672134399
    },
    {
      "epoch": 6.645965576171875e-05,
      "model_forward_time": 1.7248127460479736,
      "step": 69688
    },
    {
      "epoch": 6.645965576171875e-05,
      "step": 69688,
      "training_step_time": 7.594284534454346
    },
    {
      "epoch": 6.646060943603515e-05,
      "model_forward_time": 1.725097894668579,
      "step": 69689
    },
    {
      "epoch": 6.646060943603515e-05,
      "step": 69689,
      "training_step_time": 7.6053078174591064
    },
    {
      "epoch": 6.646156311035156e-05,
      "grad_norm": 0.0812809094786644,
      "learning_rate": 2.3084526171527124e-06,
      "loss": 0.0538,
      "step": 69690
    },
    {
      "epoch": 6.646156311035156e-05,
      "model_forward_time": 1.7188925743103027,
      "step": 69690
    },
    {
      "dynamics_loss_avg": 0.06690829750150443,
      "epoch": 6.646156311035156e-05,
      "step": 69690
    },
    {
      "action_loss_avg": 0.0017901502083986997,
      "epoch": 6.646156311035156e-05,
      "step": 69690
    },
    {
      "epoch": 6.646156311035156e-05,
      "step": 69690,
      "training_step_time": 7.5939788818359375
    },
    {
      "epoch": 6.646251678466797e-05,
      "model_forward_time": 1.7437257766723633,
      "step": 69691
    },
    {
      "epoch": 6.646251678466797e-05,
      "step": 69691,
      "training_step_time": 7.616387367248535
    },
    {
      "epoch": 6.646347045898437e-05,
      "model_forward_time": 1.727142333984375,
      "step": 69692
    },
    {
      "epoch": 6.646347045898437e-05,
      "step": 69692,
      "training_step_time": 7.870559215545654
    },
    {
      "epoch": 6.646442413330078e-05,
      "model_forward_time": 2.917973518371582,
      "step": 69693
    },
    {
      "epoch": 6.646442413330078e-05,
      "step": 69693,
      "training_step_time": 8.564104557037354
    },
    {
      "epoch": 6.646537780761719e-05,
      "model_forward_time": 1.7318716049194336,
      "step": 69694
    },
    {
      "epoch": 6.646537780761719e-05,
      "step": 69694,
      "training_step_time": 7.609423875808716
    },
    {
      "epoch": 6.646633148193359e-05,
      "model_forward_time": 1.719531536102295,
      "step": 69695
    },
    {
      "epoch": 6.646633148193359e-05,
      "step": 69695,
      "training_step_time": 7.638734340667725
    },
    {
      "epoch": 6.646728515625e-05,
      "model_forward_time": 1.0555346012115479,
      "step": 69696
    },
    {
      "epoch": 6.646728515625e-05,
      "step": 69696,
      "training_step_time": 7.544775485992432
    },
    {
      "epoch": 6.646823883056641e-05,
      "model_forward_time": 1.7411034107208252,
      "step": 69697
    },
    {
      "epoch": 6.646823883056641e-05,
      "step": 69697,
      "training_step_time": 7.581016302108765
    },
    {
      "epoch": 6.64691925048828e-05,
      "model_forward_time": 1.7390656471252441,
      "step": 69698
    },
    {
      "epoch": 6.64691925048828e-05,
      "step": 69698,
      "training_step_time": 8.62635588645935
    },
    {
      "epoch": 6.647014617919922e-05,
      "model_forward_time": 1.7407941818237305,
      "step": 69699
    },
    {
      "epoch": 6.647014617919922e-05,
      "step": 69699,
      "training_step_time": 7.541745662689209
    },
    {
      "epoch": 6.647109985351563e-05,
      "grad_norm": 0.08048008382320404,
      "learning_rate": 2.30705930722766e-06,
      "loss": 0.0544,
      "step": 69700
    },
    {
      "epoch": 6.647109985351563e-05,
      "model_forward_time": 1.7552599906921387,
      "step": 69700
    },
    {
      "dynamics_loss_avg": 0.04544173050671816,
      "epoch": 6.647109985351563e-05,
      "step": 69700
    },
    {
      "action_loss_avg": 0.0010209260013652965,
      "epoch": 6.647109985351563e-05,
      "step": 69700
    },
    {
      "epoch": 6.647109985351563e-05,
      "step": 69700,
      "training_step_time": 7.59185004234314
    },
    {
      "epoch": 6.647205352783202e-05,
      "model_forward_time": 1.7372093200683594,
      "step": 69701
    },
    {
      "epoch": 6.647205352783202e-05,
      "step": 69701,
      "training_step_time": 9.40267038345337
    },
    {
      "epoch": 6.647300720214844e-05,
      "model_forward_time": 2.937204122543335,
      "step": 69702
    },
    {
      "epoch": 6.647300720214844e-05,
      "step": 69702,
      "training_step_time": 8.964551210403442
    },
    {
      "epoch": 6.647396087646485e-05,
      "model_forward_time": 1.733926773071289,
      "step": 69703
    },
    {
      "epoch": 6.647396087646485e-05,
      "step": 69703,
      "training_step_time": 7.619380235671997
    },
    {
      "epoch": 6.647491455078124e-05,
      "model_forward_time": 1.7338640689849854,
      "step": 69704
    },
    {
      "epoch": 6.647491455078124e-05,
      "step": 69704,
      "training_step_time": 8.374678134918213
    },
    {
      "epoch": 6.647586822509765e-05,
      "model_forward_time": 1.7611815929412842,
      "step": 69705
    },
    {
      "epoch": 6.647586822509765e-05,
      "step": 69705,
      "training_step_time": 8.295740365982056
    },
    {
      "epoch": 6.647682189941407e-05,
      "model_forward_time": 1.7364227771759033,
      "step": 69706
    },
    {
      "epoch": 6.647682189941407e-05,
      "step": 69706,
      "training_step_time": 7.686442136764526
    },
    {
      "epoch": 6.647777557373046e-05,
      "model_forward_time": 1.734142780303955,
      "step": 69707
    },
    {
      "epoch": 6.647777557373046e-05,
      "step": 69707,
      "training_step_time": 8.6142578125
    },
    {
      "epoch": 6.647872924804687e-05,
      "model_forward_time": 1.738779067993164,
      "step": 69708
    },
    {
      "epoch": 6.647872924804687e-05,
      "step": 69708,
      "training_step_time": 8.171865940093994
    },
    {
      "epoch": 6.647968292236328e-05,
      "model_forward_time": 1.736419677734375,
      "step": 69709
    },
    {
      "epoch": 6.647968292236328e-05,
      "step": 69709,
      "training_step_time": 7.620633602142334
    },
    {
      "epoch": 6.648063659667968e-05,
      "grad_norm": 0.09871567785739899,
      "learning_rate": 2.3056662917985518e-06,
      "loss": 0.0515,
      "step": 69710
    },
    {
      "epoch": 6.648063659667968e-05,
      "model_forward_time": 1.7368841171264648,
      "step": 69710
    },
    {
      "dynamics_loss_avg": 0.04866101797670126,
      "epoch": 6.648063659667968e-05,
      "step": 69710
    },
    {
      "action_loss_avg": 0.0013798905099974945,
      "epoch": 6.648063659667968e-05,
      "step": 69710
    },
    {
      "epoch": 6.648063659667968e-05,
      "step": 69710,
      "training_step_time": 8.556179285049438
    },
    {
      "epoch": 6.648159027099609e-05,
      "model_forward_time": 2.883155345916748,
      "step": 69711
    },
    {
      "epoch": 6.648159027099609e-05,
      "step": 69711,
      "training_step_time": 8.56351613998413
    },
    {
      "epoch": 6.64825439453125e-05,
      "model_forward_time": 1.7579622268676758,
      "step": 69712
    },
    {
      "epoch": 6.64825439453125e-05,
      "step": 69712,
      "training_step_time": 7.65914511680603
    },
    {
      "epoch": 6.64834976196289e-05,
      "model_forward_time": 1.7476420402526855,
      "step": 69713
    },
    {
      "epoch": 6.64834976196289e-05,
      "step": 69713,
      "training_step_time": 7.777223825454712
    },
    {
      "epoch": 6.648445129394531e-05,
      "model_forward_time": 1.8048312664031982,
      "step": 69714
    },
    {
      "epoch": 6.648445129394531e-05,
      "step": 69714,
      "training_step_time": 8.651384592056274
    },
    {
      "epoch": 6.648540496826172e-05,
      "model_forward_time": 1.750293493270874,
      "step": 69715
    },
    {
      "epoch": 6.648540496826172e-05,
      "step": 69715,
      "training_step_time": 7.644105672836304
    },
    {
      "epoch": 6.648635864257812e-05,
      "model_forward_time": 1.745471477508545,
      "step": 69716
    },
    {
      "epoch": 6.648635864257812e-05,
      "step": 69716,
      "training_step_time": 7.6962456703186035
    },
    {
      "epoch": 6.648731231689453e-05,
      "model_forward_time": 1.7595224380493164,
      "step": 69717
    },
    {
      "epoch": 6.648731231689453e-05,
      "step": 69717,
      "training_step_time": 7.613097906112671
    },
    {
      "epoch": 6.648826599121094e-05,
      "model_forward_time": 1.7665982246398926,
      "step": 69718
    },
    {
      "epoch": 6.648826599121094e-05,
      "step": 69718,
      "training_step_time": 7.569849967956543
    },
    {
      "epoch": 6.648921966552734e-05,
      "model_forward_time": 1.7426888942718506,
      "step": 69719
    },
    {
      "epoch": 6.648921966552734e-05,
      "step": 69719,
      "training_step_time": 8.388584852218628
    },
    {
      "epoch": 6.649017333984375e-05,
      "grad_norm": 0.09835349023342133,
      "learning_rate": 2.3042735710177285e-06,
      "loss": 0.0564,
      "step": 69720
    },
    {
      "epoch": 6.649017333984375e-05,
      "model_forward_time": 1.7476692199707031,
      "step": 69720
    },
    {
      "dynamics_loss_avg": 0.0544468604028225,
      "epoch": 6.649017333984375e-05,
      "step": 69720
    },
    {
      "action_loss_avg": 0.0017944537656148895,
      "epoch": 6.649017333984375e-05,
      "step": 69720
    },
    {
      "epoch": 6.649017333984375e-05,
      "step": 69720,
      "training_step_time": 8.303001403808594
    },
    {
      "epoch": 6.649112701416016e-05,
      "model_forward_time": 1.765312671661377,
      "step": 69721
    },
    {
      "epoch": 6.649112701416016e-05,
      "step": 69721,
      "training_step_time": 7.576018571853638
    },
    {
      "epoch": 6.649208068847656e-05,
      "model_forward_time": 1.7453384399414062,
      "step": 69722
    },
    {
      "epoch": 6.649208068847656e-05,
      "step": 69722,
      "training_step_time": 7.806307077407837
    },
    {
      "epoch": 6.649303436279297e-05,
      "model_forward_time": 1.7963523864746094,
      "step": 69723
    },
    {
      "epoch": 6.649303436279297e-05,
      "step": 69723,
      "training_step_time": 7.723891258239746
    },
    {
      "epoch": 6.649398803710938e-05,
      "model_forward_time": 1.7506871223449707,
      "step": 69724
    },
    {
      "epoch": 6.649398803710938e-05,
      "step": 69724,
      "training_step_time": 7.663416862487793
    },
    {
      "epoch": 6.649494171142578e-05,
      "model_forward_time": 1.7495479583740234,
      "step": 69725
    },
    {
      "epoch": 6.649494171142578e-05,
      "step": 69725,
      "training_step_time": 8.171175003051758
    },
    {
      "epoch": 6.649589538574219e-05,
      "model_forward_time": 1.7557320594787598,
      "step": 69726
    },
    {
      "epoch": 6.649589538574219e-05,
      "step": 69726,
      "training_step_time": 7.65171217918396
    },
    {
      "epoch": 6.64968490600586e-05,
      "model_forward_time": 1.7417786121368408,
      "step": 69727
    },
    {
      "epoch": 6.64968490600586e-05,
      "step": 69727,
      "training_step_time": 7.600367784500122
    },
    {
      "epoch": 6.6497802734375e-05,
      "model_forward_time": 1.7445721626281738,
      "step": 69728
    },
    {
      "epoch": 6.6497802734375e-05,
      "step": 69728,
      "training_step_time": 8.693358659744263
    },
    {
      "epoch": 6.64987564086914e-05,
      "model_forward_time": 1.753159761428833,
      "step": 69729
    },
    {
      "epoch": 6.64987564086914e-05,
      "step": 69729,
      "training_step_time": 7.752949953079224
    },
    {
      "epoch": 6.649971008300782e-05,
      "grad_norm": 0.08900722861289978,
      "learning_rate": 2.3028811450374923e-06,
      "loss": 0.0546,
      "step": 69730
    },
    {
      "epoch": 6.649971008300782e-05,
      "model_forward_time": 2.939851999282837,
      "step": 69730
    },
    {
      "dynamics_loss_avg": 0.05143680218607187,
      "epoch": 6.649971008300782e-05,
      "step": 69730
    },
    {
      "action_loss_avg": 0.0010439409408718348,
      "epoch": 6.649971008300782e-05,
      "step": 69730
    },
    {
      "epoch": 6.649971008300782e-05,
      "step": 69730,
      "training_step_time": 8.305139064788818
    },
    {
      "epoch": 6.650066375732421e-05,
      "model_forward_time": 1.7575347423553467,
      "step": 69731
    },
    {
      "epoch": 6.650066375732421e-05,
      "step": 69731,
      "training_step_time": 8.629340887069702
    },
    {
      "epoch": 6.650161743164062e-05,
      "model_forward_time": 1.7523491382598877,
      "step": 69732
    },
    {
      "epoch": 6.650161743164062e-05,
      "step": 69732,
      "training_step_time": 8.623222827911377
    },
    {
      "epoch": 6.650257110595704e-05,
      "model_forward_time": 1.7994880676269531,
      "step": 69733
    },
    {
      "epoch": 6.650257110595704e-05,
      "step": 69733,
      "training_step_time": 7.580230474472046
    },
    {
      "epoch": 6.650352478027343e-05,
      "model_forward_time": 1.748852252960205,
      "step": 69734
    },
    {
      "epoch": 6.650352478027343e-05,
      "step": 69734,
      "training_step_time": 8.492886781692505
    },
    {
      "epoch": 6.650447845458984e-05,
      "model_forward_time": 1.7482433319091797,
      "step": 69735
    },
    {
      "epoch": 6.650447845458984e-05,
      "step": 69735,
      "training_step_time": 7.5904505252838135
    },
    {
      "epoch": 6.650543212890625e-05,
      "model_forward_time": 1.7550501823425293,
      "step": 69736
    },
    {
      "epoch": 6.650543212890625e-05,
      "step": 69736,
      "training_step_time": 7.65129828453064
    },
    {
      "epoch": 6.650638580322265e-05,
      "model_forward_time": 1.7534432411193848,
      "step": 69737
    },
    {
      "epoch": 6.650638580322265e-05,
      "step": 69737,
      "training_step_time": 8.123881101608276
    },
    {
      "epoch": 6.650733947753906e-05,
      "model_forward_time": 1.7453043460845947,
      "step": 69738
    },
    {
      "epoch": 6.650733947753906e-05,
      "step": 69738,
      "training_step_time": 7.8547117710113525
    },
    {
      "epoch": 6.650829315185547e-05,
      "model_forward_time": 3.0911641120910645,
      "step": 69739
    },
    {
      "epoch": 6.650829315185547e-05,
      "step": 69739,
      "training_step_time": 10.681718349456787
    },
    {
      "epoch": 6.650924682617187e-05,
      "grad_norm": 0.09104376286268234,
      "learning_rate": 2.301489014010119e-06,
      "loss": 0.0551,
      "step": 69740
    },
    {
      "epoch": 6.650924682617187e-05,
      "model_forward_time": 1.7538886070251465,
      "step": 69740
    },
    {
      "dynamics_loss_avg": 0.05372490063309669,
      "epoch": 6.650924682617187e-05,
      "step": 69740
    },
    {
      "action_loss_avg": 0.0012216126604471355,
      "epoch": 6.650924682617187e-05,
      "step": 69740
    },
    {
      "epoch": 6.650924682617187e-05,
      "step": 69740,
      "training_step_time": 6.778253078460693
    },
    {
      "epoch": 6.651020050048828e-05,
      "model_forward_time": 1.7507588863372803,
      "step": 69741
    },
    {
      "epoch": 6.651020050048828e-05,
      "step": 69741,
      "training_step_time": 6.786925315856934
    },
    {
      "epoch": 6.651115417480469e-05,
      "model_forward_time": 1.7528345584869385,
      "step": 69742
    },
    {
      "epoch": 6.651115417480469e-05,
      "step": 69742,
      "training_step_time": 6.804381608963013
    },
    {
      "epoch": 6.651210784912109e-05,
      "model_forward_time": 1.7453458309173584,
      "step": 69743
    },
    {
      "epoch": 6.651210784912109e-05,
      "step": 69743,
      "training_step_time": 6.754507064819336
    },
    {
      "epoch": 6.65130615234375e-05,
      "model_forward_time": 1.7404959201812744,
      "step": 69744
    },
    {
      "epoch": 6.65130615234375e-05,
      "step": 69744,
      "training_step_time": 6.7172276973724365
    },
    {
      "epoch": 6.651401519775391e-05,
      "model_forward_time": 1.7447667121887207,
      "step": 69745
    },
    {
      "epoch": 6.651401519775391e-05,
      "step": 69745,
      "training_step_time": 6.764800071716309
    },
    {
      "epoch": 6.651496887207031e-05,
      "model_forward_time": 1.7502667903900146,
      "step": 69746
    },
    {
      "epoch": 6.651496887207031e-05,
      "step": 69746,
      "training_step_time": 6.705821752548218
    },
    {
      "epoch": 6.651592254638672e-05,
      "model_forward_time": 1.739605188369751,
      "step": 69747
    },
    {
      "epoch": 6.651592254638672e-05,
      "step": 69747,
      "training_step_time": 6.717602014541626
    },
    {
      "epoch": 6.651687622070313e-05,
      "model_forward_time": 2.88445782661438,
      "step": 69748
    },
    {
      "epoch": 6.651687622070313e-05,
      "step": 69748,
      "training_step_time": 10.794479846954346
    },
    {
      "epoch": 6.651782989501953e-05,
      "model_forward_time": 1.7554402351379395,
      "step": 69749
    },
    {
      "epoch": 6.651782989501953e-05,
      "step": 69749,
      "training_step_time": 7.690448045730591
    },
    {
      "epoch": 6.651878356933594e-05,
      "grad_norm": 0.11282142996788025,
      "learning_rate": 2.30009717808785e-06,
      "loss": 0.0543,
      "step": 69750
    },
    {
      "epoch": 6.651878356933594e-05,
      "model_forward_time": 1.7480390071868896,
      "step": 69750
    },
    {
      "dynamics_loss_avg": 0.047528144996613265,
      "epoch": 6.651878356933594e-05,
      "step": 69750
    },
    {
      "action_loss_avg": 0.0011475493607576936,
      "epoch": 6.651878356933594e-05,
      "step": 69750
    },
    {
      "epoch": 6.651878356933594e-05,
      "step": 69750,
      "training_step_time": 6.7871198654174805
    },
    {
      "epoch": 6.651973724365235e-05,
      "model_forward_time": 1.744232416152954,
      "step": 69751
    },
    {
      "epoch": 6.651973724365235e-05,
      "step": 69751,
      "training_step_time": 6.7592339515686035
    },
    {
      "epoch": 6.652069091796875e-05,
      "model_forward_time": 1.7454509735107422,
      "step": 69752
    },
    {
      "epoch": 6.652069091796875e-05,
      "step": 69752,
      "training_step_time": 6.7557337284088135
    },
    {
      "epoch": 6.652164459228516e-05,
      "model_forward_time": 1.7437939643859863,
      "step": 69753
    },
    {
      "epoch": 6.652164459228516e-05,
      "step": 69753,
      "training_step_time": 6.827387571334839
    },
    {
      "epoch": 6.652259826660157e-05,
      "model_forward_time": 1.746811866760254,
      "step": 69754
    },
    {
      "epoch": 6.652259826660157e-05,
      "step": 69754,
      "training_step_time": 6.758248567581177
    },
    {
      "epoch": 6.652355194091796e-05,
      "model_forward_time": 1.0776193141937256,
      "step": 69755
    },
    {
      "epoch": 6.652355194091796e-05,
      "step": 69755,
      "training_step_time": 6.762175798416138
    },
    {
      "epoch": 6.652450561523438e-05,
      "model_forward_time": 1.7635834217071533,
      "step": 69756
    },
    {
      "epoch": 6.652450561523438e-05,
      "step": 69756,
      "training_step_time": 6.822230815887451
    },
    {
      "epoch": 6.652545928955079e-05,
      "model_forward_time": 1.7430098056793213,
      "step": 69757
    },
    {
      "epoch": 6.652545928955079e-05,
      "step": 69757,
      "training_step_time": 6.783657550811768
    },
    {
      "epoch": 6.652641296386718e-05,
      "model_forward_time": 2.8553385734558105,
      "step": 69758
    },
    {
      "epoch": 6.652641296386718e-05,
      "step": 69758,
      "training_step_time": 7.886895179748535
    },
    {
      "epoch": 6.65273666381836e-05,
      "model_forward_time": 1.749453067779541,
      "step": 69759
    },
    {
      "epoch": 6.65273666381836e-05,
      "step": 69759,
      "training_step_time": 6.805675506591797
    },
    {
      "epoch": 6.65283203125e-05,
      "grad_norm": 0.09586016833782196,
      "learning_rate": 2.2987056374228966e-06,
      "loss": 0.0536,
      "step": 69760
    },
    {
      "epoch": 6.65283203125e-05,
      "model_forward_time": 1.7695834636688232,
      "step": 69760
    },
    {
      "dynamics_loss_avg": 0.05059604980051517,
      "epoch": 6.65283203125e-05,
      "step": 69760
    },
    {
      "action_loss_avg": 0.0014598833571653814,
      "epoch": 6.65283203125e-05,
      "step": 69760
    },
    {
      "epoch": 6.65283203125e-05,
      "step": 69760,
      "training_step_time": 6.806644678115845
    },
    {
      "epoch": 6.65292739868164e-05,
      "model_forward_time": 1.7425806522369385,
      "step": 69761
    },
    {
      "epoch": 6.65292739868164e-05,
      "step": 69761,
      "training_step_time": 6.752141952514648
    },
    {
      "epoch": 6.653022766113281e-05,
      "model_forward_time": 1.7554411888122559,
      "step": 69762
    },
    {
      "epoch": 6.653022766113281e-05,
      "step": 69762,
      "training_step_time": 6.794972658157349
    },
    {
      "epoch": 6.653118133544922e-05,
      "model_forward_time": 1.7801806926727295,
      "step": 69763
    },
    {
      "epoch": 6.653118133544922e-05,
      "step": 69763,
      "training_step_time": 6.741515874862671
    },
    {
      "epoch": 6.653213500976562e-05,
      "model_forward_time": 1.7613003253936768,
      "step": 69764
    },
    {
      "epoch": 6.653213500976562e-05,
      "step": 69764,
      "training_step_time": 6.7663328647613525
    },
    {
      "epoch": 6.653308868408203e-05,
      "model_forward_time": 1.3942577838897705,
      "step": 69765
    },
    {
      "epoch": 6.653308868408203e-05,
      "step": 69765,
      "training_step_time": 6.809868097305298
    },
    {
      "epoch": 6.653404235839844e-05,
      "model_forward_time": 1.7618939876556396,
      "step": 69766
    },
    {
      "epoch": 6.653404235839844e-05,
      "step": 69766,
      "training_step_time": 7.826571702957153
    },
    {
      "epoch": 6.653499603271484e-05,
      "model_forward_time": 2.8706881999969482,
      "step": 69767
    },
    {
      "epoch": 6.653499603271484e-05,
      "step": 69767,
      "training_step_time": 7.788243532180786
    },
    {
      "epoch": 6.653594970703125e-05,
      "model_forward_time": 1.7527813911437988,
      "step": 69768
    },
    {
      "epoch": 6.653594970703125e-05,
      "step": 69768,
      "training_step_time": 7.538698196411133
    },
    {
      "epoch": 6.653690338134766e-05,
      "model_forward_time": 1.7517173290252686,
      "step": 69769
    },
    {
      "epoch": 6.653690338134766e-05,
      "step": 69769,
      "training_step_time": 6.763539552688599
    },
    {
      "epoch": 6.653785705566406e-05,
      "grad_norm": 0.10462526977062225,
      "learning_rate": 2.2973143921674314e-06,
      "loss": 0.055,
      "step": 69770
    },
    {
      "epoch": 6.653785705566406e-05,
      "model_forward_time": 1.7681100368499756,
      "step": 69770
    },
    {
      "dynamics_loss_avg": 0.043565662764012816,
      "epoch": 6.653785705566406e-05,
      "step": 69770
    },
    {
      "action_loss_avg": 0.0010878644970944151,
      "epoch": 6.653785705566406e-05,
      "step": 69770
    },
    {
      "epoch": 6.653785705566406e-05,
      "step": 69770,
      "training_step_time": 6.822211980819702
    },
    {
      "epoch": 6.653881072998047e-05,
      "model_forward_time": 1.7530088424682617,
      "step": 69771
    },
    {
      "epoch": 6.653881072998047e-05,
      "step": 69771,
      "training_step_time": 6.787219047546387
    },
    {
      "epoch": 6.653976440429688e-05,
      "model_forward_time": 1.7507617473602295,
      "step": 69772
    },
    {
      "epoch": 6.653976440429688e-05,
      "step": 69772,
      "training_step_time": 6.789251327514648
    },
    {
      "epoch": 6.654071807861328e-05,
      "model_forward_time": 1.7523598670959473,
      "step": 69773
    },
    {
      "epoch": 6.654071807861328e-05,
      "step": 69773,
      "training_step_time": 6.752822399139404
    },
    {
      "epoch": 6.654167175292969e-05,
      "model_forward_time": 1.7785253524780273,
      "step": 69774
    },
    {
      "epoch": 6.654167175292969e-05,
      "step": 69774,
      "training_step_time": 6.774331569671631
    },
    {
      "epoch": 6.65426254272461e-05,
      "model_forward_time": 1.7636363506317139,
      "step": 69775
    },
    {
      "epoch": 6.65426254272461e-05,
      "step": 69775,
      "training_step_time": 6.835744380950928
    },
    {
      "epoch": 6.65435791015625e-05,
      "model_forward_time": 2.921874523162842,
      "step": 69776
    },
    {
      "epoch": 6.65435791015625e-05,
      "step": 69776,
      "training_step_time": 9.888509035110474
    },
    {
      "epoch": 6.654453277587891e-05,
      "model_forward_time": 1.756946325302124,
      "step": 69777
    },
    {
      "epoch": 6.654453277587891e-05,
      "step": 69777,
      "training_step_time": 7.81113338470459
    },
    {
      "epoch": 6.654548645019532e-05,
      "model_forward_time": 1.763749361038208,
      "step": 69778
    },
    {
      "epoch": 6.654548645019532e-05,
      "step": 69778,
      "training_step_time": 6.844894886016846
    },
    {
      "epoch": 6.654644012451172e-05,
      "model_forward_time": 1.7653422355651855,
      "step": 69779
    },
    {
      "epoch": 6.654644012451172e-05,
      "step": 69779,
      "training_step_time": 6.795598745346069
    },
    {
      "epoch": 6.654739379882813e-05,
      "grad_norm": 0.08709294348955154,
      "learning_rate": 2.295923442473601e-06,
      "loss": 0.0545,
      "step": 69780
    },
    {
      "epoch": 6.654739379882813e-05,
      "model_forward_time": 1.7601313591003418,
      "step": 69780
    },
    {
      "dynamics_loss_avg": 0.03794264756143093,
      "epoch": 6.654739379882813e-05,
      "step": 69780
    },
    {
      "action_loss_avg": 0.0010235063586151228,
      "epoch": 6.654739379882813e-05,
      "step": 69780
    },
    {
      "epoch": 6.654739379882813e-05,
      "step": 69780,
      "training_step_time": 6.865888595581055
    },
    {
      "epoch": 6.654834747314454e-05,
      "model_forward_time": 1.7576589584350586,
      "step": 69781
    },
    {
      "epoch": 6.654834747314454e-05,
      "step": 69781,
      "training_step_time": 6.894551753997803
    },
    {
      "epoch": 6.654930114746093e-05,
      "model_forward_time": 1.7609612941741943,
      "step": 69782
    },
    {
      "epoch": 6.654930114746093e-05,
      "step": 69782,
      "training_step_time": 6.822198867797852
    },
    {
      "epoch": 6.655025482177734e-05,
      "model_forward_time": 1.764467477798462,
      "step": 69783
    },
    {
      "epoch": 6.655025482177734e-05,
      "step": 69783,
      "training_step_time": 6.81948184967041
    },
    {
      "epoch": 6.655120849609376e-05,
      "model_forward_time": 1.7925834655761719,
      "step": 69784
    },
    {
      "epoch": 6.655120849609376e-05,
      "step": 69784,
      "training_step_time": 6.81476902961731
    },
    {
      "epoch": 6.655216217041015e-05,
      "model_forward_time": 2.911581039428711,
      "step": 69785
    },
    {
      "epoch": 6.655216217041015e-05,
      "step": 69785,
      "training_step_time": 7.820887804031372
    },
    {
      "epoch": 6.655311584472656e-05,
      "model_forward_time": 1.7626276016235352,
      "step": 69786
    },
    {
      "epoch": 6.655311584472656e-05,
      "step": 69786,
      "training_step_time": 7.663909435272217
    },
    {
      "epoch": 6.655406951904297e-05,
      "model_forward_time": 1.7651033401489258,
      "step": 69787
    },
    {
      "epoch": 6.655406951904297e-05,
      "step": 69787,
      "training_step_time": 6.737766981124878
    },
    {
      "epoch": 6.655502319335937e-05,
      "model_forward_time": 1.756605863571167,
      "step": 69788
    },
    {
      "epoch": 6.655502319335937e-05,
      "step": 69788,
      "training_step_time": 6.753462791442871
    },
    {
      "epoch": 6.655597686767578e-05,
      "model_forward_time": 1.7623672485351562,
      "step": 69789
    },
    {
      "epoch": 6.655597686767578e-05,
      "step": 69789,
      "training_step_time": 6.730135440826416
    },
    {
      "epoch": 6.65569305419922e-05,
      "grad_norm": 0.08902361243963242,
      "learning_rate": 2.2945327884935192e-06,
      "loss": 0.057,
      "step": 69790
    },
    {
      "epoch": 6.65569305419922e-05,
      "model_forward_time": 1.765319585800171,
      "step": 69790
    },
    {
      "dynamics_loss_avg": 0.04180220328271389,
      "epoch": 6.65569305419922e-05,
      "step": 69790
    },
    {
      "action_loss_avg": 0.001028157863765955,
      "epoch": 6.65569305419922e-05,
      "step": 69790
    },
    {
      "epoch": 6.65569305419922e-05,
      "step": 69790,
      "training_step_time": 6.776996374130249
    },
    {
      "epoch": 6.655788421630859e-05,
      "model_forward_time": 1.7570273876190186,
      "step": 69791
    },
    {
      "epoch": 6.655788421630859e-05,
      "step": 69791,
      "training_step_time": 6.719079971313477
    },
    {
      "epoch": 6.6558837890625e-05,
      "model_forward_time": 1.7595045566558838,
      "step": 69792
    },
    {
      "epoch": 6.6558837890625e-05,
      "step": 69792,
      "training_step_time": 6.754490375518799
    },
    {
      "epoch": 6.655979156494141e-05,
      "model_forward_time": 1.7551147937774658,
      "step": 69793
    },
    {
      "epoch": 6.655979156494141e-05,
      "step": 69793,
      "training_step_time": 6.710838317871094
    },
    {
      "epoch": 6.656074523925781e-05,
      "model_forward_time": 1.7953178882598877,
      "step": 69794
    },
    {
      "epoch": 6.656074523925781e-05,
      "step": 69794,
      "training_step_time": 8.265030860900879
    },
    {
      "epoch": 6.656169891357422e-05,
      "model_forward_time": 2.905228853225708,
      "step": 69795
    },
    {
      "epoch": 6.656169891357422e-05,
      "step": 69795,
      "training_step_time": 7.8052873611450195
    },
    {
      "epoch": 6.656265258789063e-05,
      "model_forward_time": 1.764244556427002,
      "step": 69796
    },
    {
      "epoch": 6.656265258789063e-05,
      "step": 69796,
      "training_step_time": 6.694616794586182
    },
    {
      "epoch": 6.656360626220703e-05,
      "model_forward_time": 1.7542221546173096,
      "step": 69797
    },
    {
      "epoch": 6.656360626220703e-05,
      "step": 69797,
      "training_step_time": 6.694223165512085
    },
    {
      "epoch": 6.656455993652344e-05,
      "model_forward_time": 1.7554214000701904,
      "step": 69798
    },
    {
      "epoch": 6.656455993652344e-05,
      "step": 69798,
      "training_step_time": 6.678921699523926
    },
    {
      "epoch": 6.656551361083985e-05,
      "model_forward_time": 1.7538256645202637,
      "step": 69799
    },
    {
      "epoch": 6.656551361083985e-05,
      "step": 69799,
      "training_step_time": 6.720844745635986
    },
    {
      "epoch": 6.656646728515625e-05,
      "grad_norm": 0.11882337927818298,
      "learning_rate": 2.293142430379262e-06,
      "loss": 0.0581,
      "step": 69800
    },
    {
      "epoch": 6.656646728515625e-05,
      "model_forward_time": 1.7625572681427002,
      "step": 69800
    },
    {
      "dynamics_loss_avg": 0.05415244046598673,
      "epoch": 6.656646728515625e-05,
      "step": 69800
    },
    {
      "action_loss_avg": 0.0016000595351215452,
      "epoch": 6.656646728515625e-05,
      "step": 69800
    },
    {
      "epoch": 6.656646728515625e-05,
      "step": 69800,
      "training_step_time": 6.667243480682373
    },
    {
      "epoch": 6.656742095947266e-05,
      "model_forward_time": 1.7614641189575195,
      "step": 69801
    },
    {
      "epoch": 6.656742095947266e-05,
      "step": 69801,
      "training_step_time": 6.718800783157349
    },
    {
      "epoch": 6.656837463378907e-05,
      "model_forward_time": 1.7599279880523682,
      "step": 69802
    },
    {
      "epoch": 6.656837463378907e-05,
      "step": 69802,
      "training_step_time": 6.688324213027954
    },
    {
      "epoch": 6.656932830810547e-05,
      "model_forward_time": 1.7593722343444824,
      "step": 69803
    },
    {
      "epoch": 6.656932830810547e-05,
      "step": 69803,
      "training_step_time": 8.98896050453186
    },
    {
      "epoch": 6.657028198242188e-05,
      "model_forward_time": 2.9198436737060547,
      "step": 69804
    },
    {
      "epoch": 6.657028198242188e-05,
      "step": 69804,
      "training_step_time": 8.811728954315186
    },
    {
      "epoch": 6.657123565673829e-05,
      "model_forward_time": 1.7542250156402588,
      "step": 69805
    },
    {
      "epoch": 6.657123565673829e-05,
      "step": 69805,
      "training_step_time": 6.708771705627441
    },
    {
      "epoch": 6.657218933105469e-05,
      "model_forward_time": 1.7585773468017578,
      "step": 69806
    },
    {
      "epoch": 6.657218933105469e-05,
      "step": 69806,
      "training_step_time": 6.681173086166382
    },
    {
      "epoch": 6.65731430053711e-05,
      "model_forward_time": 1.762493371963501,
      "step": 69807
    },
    {
      "epoch": 6.65731430053711e-05,
      "step": 69807,
      "training_step_time": 6.752244710922241
    },
    {
      "epoch": 6.65740966796875e-05,
      "model_forward_time": 1.7442233562469482,
      "step": 69808
    },
    {
      "epoch": 6.65740966796875e-05,
      "step": 69808,
      "training_step_time": 6.695174217224121
    },
    {
      "epoch": 6.65750503540039e-05,
      "model_forward_time": 1.7405009269714355,
      "step": 69809
    },
    {
      "epoch": 6.65750503540039e-05,
      "step": 69809,
      "training_step_time": 6.664896011352539
    },
    {
      "epoch": 6.657600402832031e-05,
      "grad_norm": 0.1130727156996727,
      "learning_rate": 2.291752368282879e-06,
      "loss": 0.0596,
      "step": 69810
    },
    {
      "epoch": 6.657600402832031e-05,
      "model_forward_time": 1.7452948093414307,
      "step": 69810
    },
    {
      "dynamics_loss_avg": 0.05195995224639773,
      "epoch": 6.657600402832031e-05,
      "step": 69810
    },
    {
      "action_loss_avg": 0.0011728940880857408,
      "epoch": 6.657600402832031e-05,
      "step": 69810
    },
    {
      "epoch": 6.657600402832031e-05,
      "step": 69810,
      "training_step_time": 6.727656602859497
    },
    {
      "epoch": 6.657695770263673e-05,
      "model_forward_time": 1.7449870109558105,
      "step": 69811
    },
    {
      "epoch": 6.657695770263673e-05,
      "step": 69811,
      "training_step_time": 6.659790992736816
    },
    {
      "epoch": 6.657791137695312e-05,
      "model_forward_time": 1.7329814434051514,
      "step": 69812
    },
    {
      "epoch": 6.657791137695312e-05,
      "step": 69812,
      "training_step_time": 8.721620321273804
    },
    {
      "epoch": 6.657886505126953e-05,
      "model_forward_time": 2.8606388568878174,
      "step": 69813
    },
    {
      "epoch": 6.657886505126953e-05,
      "step": 69813,
      "training_step_time": 7.749127388000488
    },
    {
      "epoch": 6.657981872558593e-05,
      "model_forward_time": 1.7381763458251953,
      "step": 69814
    },
    {
      "epoch": 6.657981872558593e-05,
      "step": 69814,
      "training_step_time": 7.674587965011597
    },
    {
      "epoch": 6.658077239990234e-05,
      "model_forward_time": 1.7682745456695557,
      "step": 69815
    },
    {
      "epoch": 6.658077239990234e-05,
      "step": 69815,
      "training_step_time": 6.695294141769409
    },
    {
      "epoch": 6.658172607421875e-05,
      "model_forward_time": 1.7437636852264404,
      "step": 69816
    },
    {
      "epoch": 6.658172607421875e-05,
      "step": 69816,
      "training_step_time": 6.689310312271118
    },
    {
      "epoch": 6.658267974853515e-05,
      "model_forward_time": 1.7421436309814453,
      "step": 69817
    },
    {
      "epoch": 6.658267974853515e-05,
      "step": 69817,
      "training_step_time": 6.694182634353638
    },
    {
      "epoch": 6.658363342285156e-05,
      "model_forward_time": 1.744433879852295,
      "step": 69818
    },
    {
      "epoch": 6.658363342285156e-05,
      "step": 69818,
      "training_step_time": 6.6689293384552
    },
    {
      "epoch": 6.658458709716797e-05,
      "model_forward_time": 1.7378158569335938,
      "step": 69819
    },
    {
      "epoch": 6.658458709716797e-05,
      "step": 69819,
      "training_step_time": 6.688779830932617
    },
    {
      "epoch": 6.658554077148437e-05,
      "grad_norm": 0.08331433683633804,
      "learning_rate": 2.290362602356385e-06,
      "loss": 0.0566,
      "step": 69820
    },
    {
      "epoch": 6.658554077148437e-05,
      "model_forward_time": 1.7530345916748047,
      "step": 69820
    },
    {
      "dynamics_loss_avg": 0.058409125963225964,
      "epoch": 6.658554077148437e-05,
      "step": 69820
    },
    {
      "action_loss_avg": 0.0018306418147403746,
      "epoch": 6.658554077148437e-05,
      "step": 69820
    },
    {
      "epoch": 6.658554077148437e-05,
      "step": 69820,
      "training_step_time": 6.712049722671509
    },
    {
      "epoch": 6.658649444580078e-05,
      "model_forward_time": 1.7362568378448486,
      "step": 69821
    },
    {
      "epoch": 6.658649444580078e-05,
      "step": 69821,
      "training_step_time": 8.702451944351196
    },
    {
      "epoch": 6.658744812011719e-05,
      "model_forward_time": 2.8817083835601807,
      "step": 69822
    },
    {
      "epoch": 6.658744812011719e-05,
      "step": 69822,
      "training_step_time": 10.704244613647461
    },
    {
      "epoch": 6.658840179443359e-05,
      "model_forward_time": 1.8786718845367432,
      "step": 69823
    },
    {
      "epoch": 6.658840179443359e-05,
      "step": 69823,
      "training_step_time": 7.2791101932525635
    },
    {
      "epoch": 6.658935546875e-05,
      "model_forward_time": 1.7533838748931885,
      "step": 69824
    },
    {
      "epoch": 6.658935546875e-05,
      "step": 69824,
      "training_step_time": 6.692556619644165
    },
    {
      "epoch": 6.659030914306641e-05,
      "model_forward_time": 1.7561557292938232,
      "step": 69825
    },
    {
      "epoch": 6.659030914306641e-05,
      "step": 69825,
      "training_step_time": 6.739335060119629
    },
    {
      "epoch": 6.65912628173828e-05,
      "model_forward_time": 1.7492034435272217,
      "step": 69826
    },
    {
      "epoch": 6.65912628173828e-05,
      "step": 69826,
      "training_step_time": 6.742001056671143
    },
    {
      "epoch": 6.659221649169922e-05,
      "model_forward_time": 1.743943691253662,
      "step": 69827
    },
    {
      "epoch": 6.659221649169922e-05,
      "step": 69827,
      "training_step_time": 6.713652849197388
    },
    {
      "epoch": 6.659317016601563e-05,
      "model_forward_time": 1.7349064350128174,
      "step": 69828
    },
    {
      "epoch": 6.659317016601563e-05,
      "step": 69828,
      "training_step_time": 6.694340229034424
    },
    {
      "epoch": 6.659412384033203e-05,
      "model_forward_time": 1.7503132820129395,
      "step": 69829
    },
    {
      "epoch": 6.659412384033203e-05,
      "step": 69829,
      "training_step_time": 6.650040626525879
    },
    {
      "epoch": 6.659507751464844e-05,
      "grad_norm": 0.07906324416399002,
      "learning_rate": 2.288973132751765e-06,
      "loss": 0.057,
      "step": 69830
    },
    {
      "epoch": 6.659507751464844e-05,
      "model_forward_time": 1.8765404224395752,
      "step": 69830
    },
    {
      "dynamics_loss_avg": 0.0561161458492279,
      "epoch": 6.659507751464844e-05,
      "step": 69830
    },
    {
      "action_loss_avg": 0.0013919581833761185,
      "epoch": 6.659507751464844e-05,
      "step": 69830
    },
    {
      "epoch": 6.659507751464844e-05,
      "step": 69830,
      "training_step_time": 7.649256706237793
    },
    {
      "epoch": 6.659603118896485e-05,
      "model_forward_time": 2.9114811420440674,
      "step": 69831
    },
    {
      "epoch": 6.659603118896485e-05,
      "step": 69831,
      "training_step_time": 9.535650730133057
    },
    {
      "epoch": 6.659698486328124e-05,
      "model_forward_time": 1.758608341217041,
      "step": 69832
    },
    {
      "epoch": 6.659698486328124e-05,
      "step": 69832,
      "training_step_time": 7.704328536987305
    },
    {
      "epoch": 6.659793853759765e-05,
      "model_forward_time": 1.4019248485565186,
      "step": 69833
    },
    {
      "epoch": 6.659793853759765e-05,
      "step": 69833,
      "training_step_time": 6.7145867347717285
    },
    {
      "epoch": 6.659889221191407e-05,
      "model_forward_time": 1.76080322265625,
      "step": 69834
    },
    {
      "epoch": 6.659889221191407e-05,
      "step": 69834,
      "training_step_time": 6.673160791397095
    },
    {
      "epoch": 6.659984588623046e-05,
      "model_forward_time": 1.7599842548370361,
      "step": 69835
    },
    {
      "epoch": 6.659984588623046e-05,
      "step": 69835,
      "training_step_time": 6.669183254241943
    },
    {
      "epoch": 6.660079956054687e-05,
      "model_forward_time": 1.7545652389526367,
      "step": 69836
    },
    {
      "epoch": 6.660079956054687e-05,
      "step": 69836,
      "training_step_time": 6.704355239868164
    },
    {
      "epoch": 6.660175323486328e-05,
      "model_forward_time": 1.7608184814453125,
      "step": 69837
    },
    {
      "epoch": 6.660175323486328e-05,
      "step": 69837,
      "training_step_time": 6.627837419509888
    },
    {
      "epoch": 6.660270690917968e-05,
      "model_forward_time": 1.7523317337036133,
      "step": 69838
    },
    {
      "epoch": 6.660270690917968e-05,
      "step": 69838,
      "training_step_time": 6.661061763763428
    },
    {
      "epoch": 6.660366058349609e-05,
      "model_forward_time": 1.8627057075500488,
      "step": 69839
    },
    {
      "epoch": 6.660366058349609e-05,
      "step": 69839,
      "training_step_time": 7.666993856430054
    },
    {
      "epoch": 6.66046142578125e-05,
      "grad_norm": 0.09126093983650208,
      "learning_rate": 2.2875839596209652e-06,
      "loss": 0.0575,
      "step": 69840
    },
    {
      "epoch": 6.66046142578125e-05,
      "model_forward_time": 1.076998233795166,
      "step": 69840
    },
    {
      "dynamics_loss_avg": 0.05603656070306897,
      "epoch": 6.66046142578125e-05,
      "step": 69840
    },
    {
      "action_loss_avg": 0.0018225815030746162,
      "epoch": 6.66046142578125e-05,
      "step": 69840
    },
    {
      "epoch": 6.66046142578125e-05,
      "step": 69840,
      "training_step_time": 10.281941890716553
    },
    {
      "epoch": 6.66055679321289e-05,
      "model_forward_time": 2.9389472007751465,
      "step": 69841
    },
    {
      "epoch": 6.66055679321289e-05,
      "step": 69841,
      "training_step_time": 7.787329435348511
    },
    {
      "epoch": 6.660652160644531e-05,
      "model_forward_time": 1.7694618701934814,
      "step": 69842
    },
    {
      "epoch": 6.660652160644531e-05,
      "step": 69842,
      "training_step_time": 6.6704261302948
    },
    {
      "epoch": 6.660747528076172e-05,
      "model_forward_time": 1.7541956901550293,
      "step": 69843
    },
    {
      "epoch": 6.660747528076172e-05,
      "step": 69843,
      "training_step_time": 6.72438645362854
    },
    {
      "epoch": 6.660842895507812e-05,
      "model_forward_time": 1.7650120258331299,
      "step": 69844
    },
    {
      "epoch": 6.660842895507812e-05,
      "step": 69844,
      "training_step_time": 6.754323244094849
    },
    {
      "epoch": 6.660938262939453e-05,
      "model_forward_time": 1.4004216194152832,
      "step": 69845
    },
    {
      "epoch": 6.660938262939453e-05,
      "step": 69845,
      "training_step_time": 6.729703664779663
    },
    {
      "epoch": 6.661033630371094e-05,
      "model_forward_time": 1.760279893875122,
      "step": 69846
    },
    {
      "epoch": 6.661033630371094e-05,
      "step": 69846,
      "training_step_time": 6.74596643447876
    },
    {
      "epoch": 6.661128997802734e-05,
      "model_forward_time": 1.756133794784546,
      "step": 69847
    },
    {
      "epoch": 6.661128997802734e-05,
      "step": 69847,
      "training_step_time": 6.724022150039673
    },
    {
      "epoch": 6.661224365234375e-05,
      "model_forward_time": 1.833658218383789,
      "step": 69848
    },
    {
      "epoch": 6.661224365234375e-05,
      "step": 69848,
      "training_step_time": 7.6155922412872314
    },
    {
      "epoch": 6.661319732666016e-05,
      "model_forward_time": 1.7630774974822998,
      "step": 69849
    },
    {
      "epoch": 6.661319732666016e-05,
      "step": 69849,
      "training_step_time": 10.307795524597168
    },
    {
      "epoch": 6.661415100097656e-05,
      "grad_norm": 0.09946057945489883,
      "learning_rate": 2.2861950831159074e-06,
      "loss": 0.0579,
      "step": 69850
    },
    {
      "epoch": 6.661415100097656e-05,
      "model_forward_time": 2.53210711479187,
      "step": 69850
    },
    {
      "dynamics_loss_avg": 0.0644168940372765,
      "epoch": 6.661415100097656e-05,
      "step": 69850
    },
    {
      "action_loss_avg": 0.0024498555460013447,
      "epoch": 6.661415100097656e-05,
      "step": 69850
    },
    {
      "epoch": 6.661415100097656e-05,
      "step": 69850,
      "training_step_time": 8.624815702438354
    },
    {
      "epoch": 6.661510467529297e-05,
      "model_forward_time": 1.763439416885376,
      "step": 69851
    },
    {
      "epoch": 6.661510467529297e-05,
      "step": 69851,
      "training_step_time": 7.634843111038208
    },
    {
      "epoch": 6.661605834960938e-05,
      "model_forward_time": 1.7630012035369873,
      "step": 69852
    },
    {
      "epoch": 6.661605834960938e-05,
      "step": 69852,
      "training_step_time": 6.752587795257568
    },
    {
      "epoch": 6.661701202392578e-05,
      "model_forward_time": 1.7770259380340576,
      "step": 69853
    },
    {
      "epoch": 6.661701202392578e-05,
      "step": 69853,
      "training_step_time": 6.7859978675842285
    },
    {
      "epoch": 6.661796569824219e-05,
      "model_forward_time": 1.7669103145599365,
      "step": 69854
    },
    {
      "epoch": 6.661796569824219e-05,
      "step": 69854,
      "training_step_time": 6.731504678726196
    },
    {
      "epoch": 6.66189193725586e-05,
      "model_forward_time": 1.7640657424926758,
      "step": 69855
    },
    {
      "epoch": 6.66189193725586e-05,
      "step": 69855,
      "training_step_time": 6.739004135131836
    },
    {
      "epoch": 6.6619873046875e-05,
      "model_forward_time": 1.7649426460266113,
      "step": 69856
    },
    {
      "epoch": 6.6619873046875e-05,
      "step": 69856,
      "training_step_time": 6.75575590133667
    },
    {
      "epoch": 6.66208267211914e-05,
      "model_forward_time": 1.7596538066864014,
      "step": 69857
    },
    {
      "epoch": 6.66208267211914e-05,
      "step": 69857,
      "training_step_time": 8.507730722427368
    },
    {
      "epoch": 6.662178039550782e-05,
      "model_forward_time": 1.7631866931915283,
      "step": 69858
    },
    {
      "epoch": 6.662178039550782e-05,
      "step": 69858,
      "training_step_time": 9.14072322845459
    },
    {
      "epoch": 6.662273406982421e-05,
      "model_forward_time": 2.8343420028686523,
      "step": 69859
    },
    {
      "epoch": 6.662273406982421e-05,
      "step": 69859,
      "training_step_time": 8.724469900131226
    },
    {
      "epoch": 6.662368774414062e-05,
      "grad_norm": 0.13345612585544586,
      "learning_rate": 2.284806503388472e-06,
      "loss": 0.0567,
      "step": 69860
    },
    {
      "epoch": 6.662368774414062e-05,
      "model_forward_time": 1.7705175876617432,
      "step": 69860
    },
    {
      "dynamics_loss_avg": 0.04631334189325571,
      "epoch": 6.662368774414062e-05,
      "step": 69860
    },
    {
      "action_loss_avg": 0.0014373392099514604,
      "epoch": 6.662368774414062e-05,
      "step": 69860
    },
    {
      "epoch": 6.662368774414062e-05,
      "step": 69860,
      "training_step_time": 7.801637649536133
    },
    {
      "epoch": 6.662464141845704e-05,
      "model_forward_time": 1.7184464931488037,
      "step": 69861
    },
    {
      "epoch": 6.662464141845704e-05,
      "step": 69861,
      "training_step_time": 6.7485668659210205
    },
    {
      "epoch": 6.662559509277343e-05,
      "model_forward_time": 1.7459182739257812,
      "step": 69862
    },
    {
      "epoch": 6.662559509277343e-05,
      "step": 69862,
      "training_step_time": 6.707186222076416
    },
    {
      "epoch": 6.662654876708984e-05,
      "model_forward_time": 1.7396275997161865,
      "step": 69863
    },
    {
      "epoch": 6.662654876708984e-05,
      "step": 69863,
      "training_step_time": 6.711501598358154
    },
    {
      "epoch": 6.662750244140625e-05,
      "model_forward_time": 1.7499468326568604,
      "step": 69864
    },
    {
      "epoch": 6.662750244140625e-05,
      "step": 69864,
      "training_step_time": 6.803143501281738
    },
    {
      "epoch": 6.662845611572265e-05,
      "model_forward_time": 1.7451605796813965,
      "step": 69865
    },
    {
      "epoch": 6.662845611572265e-05,
      "step": 69865,
      "training_step_time": 6.762498378753662
    },
    {
      "epoch": 6.662940979003906e-05,
      "model_forward_time": 1.7483413219451904,
      "step": 69866
    },
    {
      "epoch": 6.662940979003906e-05,
      "step": 69866,
      "training_step_time": 7.68143892288208
    },
    {
      "epoch": 6.663036346435547e-05,
      "model_forward_time": 1.8400509357452393,
      "step": 69867
    },
    {
      "epoch": 6.663036346435547e-05,
      "step": 69867,
      "training_step_time": 8.67662525177002
    },
    {
      "epoch": 6.663131713867187e-05,
      "model_forward_time": 1.7424564361572266,
      "step": 69868
    },
    {
      "epoch": 6.663131713867187e-05,
      "step": 69868,
      "training_step_time": 8.670399188995361
    },
    {
      "epoch": 6.663227081298828e-05,
      "model_forward_time": 2.910782814025879,
      "step": 69869
    },
    {
      "epoch": 6.663227081298828e-05,
      "step": 69869,
      "training_step_time": 7.778048515319824
    },
    {
      "epoch": 6.663322448730469e-05,
      "grad_norm": 0.07652571052312851,
      "learning_rate": 2.283418220590518e-06,
      "loss": 0.0587,
      "step": 69870
    },
    {
      "epoch": 6.663322448730469e-05,
      "model_forward_time": 1.7316820621490479,
      "step": 69870
    },
    {
      "dynamics_loss_avg": 0.04813660196959972,
      "epoch": 6.663322448730469e-05,
      "step": 69870
    },
    {
      "action_loss_avg": 0.0009250596107449382,
      "epoch": 6.663322448730469e-05,
      "step": 69870
    },
    {
      "epoch": 6.663322448730469e-05,
      "step": 69870,
      "training_step_time": 6.745163917541504
    },
    {
      "epoch": 6.663417816162109e-05,
      "model_forward_time": 1.7513842582702637,
      "step": 69871
    },
    {
      "epoch": 6.663417816162109e-05,
      "step": 69871,
      "training_step_time": 6.735710144042969
    },
    {
      "epoch": 6.66351318359375e-05,
      "model_forward_time": 1.729595422744751,
      "step": 69872
    },
    {
      "epoch": 6.66351318359375e-05,
      "step": 69872,
      "training_step_time": 6.785459756851196
    },
    {
      "epoch": 6.663608551025391e-05,
      "model_forward_time": 1.7438783645629883,
      "step": 69873
    },
    {
      "epoch": 6.663608551025391e-05,
      "step": 69873,
      "training_step_time": 6.767592668533325
    },
    {
      "epoch": 6.663703918457031e-05,
      "model_forward_time": 1.7560598850250244,
      "step": 69874
    },
    {
      "epoch": 6.663703918457031e-05,
      "step": 69874,
      "training_step_time": 6.688243389129639
    },
    {
      "epoch": 6.663799285888672e-05,
      "model_forward_time": 1.7508463859558105,
      "step": 69875
    },
    {
      "epoch": 6.663799285888672e-05,
      "step": 69875,
      "training_step_time": 7.705557823181152
    },
    {
      "epoch": 6.663894653320313e-05,
      "model_forward_time": 1.8489859104156494,
      "step": 69876
    },
    {
      "epoch": 6.663894653320313e-05,
      "step": 69876,
      "training_step_time": 8.257983922958374
    },
    {
      "epoch": 6.663990020751953e-05,
      "model_forward_time": 1.747490644454956,
      "step": 69877
    },
    {
      "epoch": 6.663990020751953e-05,
      "step": 69877,
      "training_step_time": 7.794646739959717
    },
    {
      "epoch": 6.664085388183594e-05,
      "model_forward_time": 2.9359261989593506,
      "step": 69878
    },
    {
      "epoch": 6.664085388183594e-05,
      "step": 69878,
      "training_step_time": 7.80239462852478
    },
    {
      "epoch": 6.664180755615235e-05,
      "model_forward_time": 1.7460014820098877,
      "step": 69879
    },
    {
      "epoch": 6.664180755615235e-05,
      "step": 69879,
      "training_step_time": 6.772582054138184
    },
    {
      "epoch": 6.664276123046875e-05,
      "grad_norm": 0.08154451102018356,
      "learning_rate": 2.282030234873862e-06,
      "loss": 0.0563,
      "step": 69880
    },
    {
      "epoch": 6.664276123046875e-05,
      "model_forward_time": 1.7215323448181152,
      "step": 69880
    },
    {
      "dynamics_loss_avg": 0.04949507880955935,
      "epoch": 6.664276123046875e-05,
      "step": 69880
    },
    {
      "action_loss_avg": 0.0014140242303255946,
      "epoch": 6.664276123046875e-05,
      "step": 69880
    },
    {
      "epoch": 6.664276123046875e-05,
      "step": 69880,
      "training_step_time": 6.8060462474823
    },
    {
      "epoch": 6.664371490478516e-05,
      "model_forward_time": 1.7525944709777832,
      "step": 69881
    },
    {
      "epoch": 6.664371490478516e-05,
      "step": 69881,
      "training_step_time": 6.793537616729736
    },
    {
      "epoch": 6.664466857910157e-05,
      "model_forward_time": 1.8039968013763428,
      "step": 69882
    },
    {
      "epoch": 6.664466857910157e-05,
      "step": 69882,
      "training_step_time": 6.755019903182983
    },
    {
      "epoch": 6.664562225341796e-05,
      "model_forward_time": 1.7676713466644287,
      "step": 69883
    },
    {
      "epoch": 6.664562225341796e-05,
      "step": 69883,
      "training_step_time": 6.762385845184326
    },
    {
      "epoch": 6.664657592773438e-05,
      "model_forward_time": 1.7413067817687988,
      "step": 69884
    },
    {
      "epoch": 6.664657592773438e-05,
      "step": 69884,
      "training_step_time": 7.680771350860596
    },
    {
      "epoch": 6.664752960205079e-05,
      "model_forward_time": 1.779212474822998,
      "step": 69885
    },
    {
      "epoch": 6.664752960205079e-05,
      "step": 69885,
      "training_step_time": 7.550761938095093
    },
    {
      "epoch": 6.664848327636718e-05,
      "model_forward_time": 1.7461683750152588,
      "step": 69886
    },
    {
      "epoch": 6.664848327636718e-05,
      "step": 69886,
      "training_step_time": 11.418121337890625
    },
    {
      "epoch": 6.66494369506836e-05,
      "model_forward_time": 2.8981711864471436,
      "step": 69887
    },
    {
      "epoch": 6.66494369506836e-05,
      "step": 69887,
      "training_step_time": 7.64326286315918
    },
    {
      "epoch": 6.6650390625e-05,
      "model_forward_time": 1.7416250705718994,
      "step": 69888
    },
    {
      "epoch": 6.6650390625e-05,
      "step": 69888,
      "training_step_time": 7.766258001327515
    },
    {
      "epoch": 6.66513442993164e-05,
      "model_forward_time": 1.754589319229126,
      "step": 69889
    },
    {
      "epoch": 6.66513442993164e-05,
      "step": 69889,
      "training_step_time": 6.803044319152832
    },
    {
      "epoch": 6.665229797363281e-05,
      "grad_norm": 0.12828360497951508,
      "learning_rate": 2.2806425463902948e-06,
      "loss": 0.0537,
      "step": 69890
    },
    {
      "epoch": 6.665229797363281e-05,
      "model_forward_time": 1.731126308441162,
      "step": 69890
    },
    {
      "dynamics_loss_avg": 0.048398510180413724,
      "epoch": 6.665229797363281e-05,
      "step": 69890
    },
    {
      "action_loss_avg": 0.001337379336473532,
      "epoch": 6.665229797363281e-05,
      "step": 69890
    },
    {
      "epoch": 6.665229797363281e-05,
      "step": 69890,
      "training_step_time": 6.749384880065918
    },
    {
      "epoch": 6.665325164794922e-05,
      "model_forward_time": 1.7531154155731201,
      "step": 69891
    },
    {
      "epoch": 6.665325164794922e-05,
      "step": 69891,
      "training_step_time": 6.737754821777344
    },
    {
      "epoch": 6.665420532226562e-05,
      "model_forward_time": 1.757502555847168,
      "step": 69892
    },
    {
      "epoch": 6.665420532226562e-05,
      "step": 69892,
      "training_step_time": 6.807939052581787
    },
    {
      "epoch": 6.665515899658203e-05,
      "model_forward_time": 1.7382075786590576,
      "step": 69893
    },
    {
      "epoch": 6.665515899658203e-05,
      "step": 69893,
      "training_step_time": 7.567392826080322
    },
    {
      "epoch": 6.665611267089844e-05,
      "model_forward_time": 1.750079870223999,
      "step": 69894
    },
    {
      "epoch": 6.665611267089844e-05,
      "step": 69894,
      "training_step_time": 8.421095132827759
    },
    {
      "epoch": 6.665706634521484e-05,
      "model_forward_time": 1.7378499507904053,
      "step": 69895
    },
    {
      "epoch": 6.665706634521484e-05,
      "step": 69895,
      "training_step_time": 8.285934209823608
    },
    {
      "epoch": 6.665802001953125e-05,
      "model_forward_time": 2.8691177368164062,
      "step": 69896
    },
    {
      "epoch": 6.665802001953125e-05,
      "step": 69896,
      "training_step_time": 8.266509294509888
    },
    {
      "epoch": 6.665897369384766e-05,
      "model_forward_time": 1.734954833984375,
      "step": 69897
    },
    {
      "epoch": 6.665897369384766e-05,
      "step": 69897,
      "training_step_time": 7.788494348526001
    },
    {
      "epoch": 6.665992736816406e-05,
      "model_forward_time": 1.751636028289795,
      "step": 69898
    },
    {
      "epoch": 6.665992736816406e-05,
      "step": 69898,
      "training_step_time": 6.78792405128479
    },
    {
      "epoch": 6.666088104248047e-05,
      "model_forward_time": 1.7365326881408691,
      "step": 69899
    },
    {
      "epoch": 6.666088104248047e-05,
      "step": 69899,
      "training_step_time": 6.751733303070068
    },
    {
      "epoch": 6.666183471679688e-05,
      "grad_norm": 0.1365545243024826,
      "learning_rate": 2.2792551552915682e-06,
      "loss": 0.0545,
      "step": 69900
    },
    {
      "epoch": 6.666183471679688e-05,
      "model_forward_time": 1.732194423675537,
      "step": 69900
    },
    {
      "dynamics_loss_avg": 0.05536339655518532,
      "epoch": 6.666183471679688e-05,
      "step": 69900
    },
    {
      "action_loss_avg": 0.0013754858417087234,
      "epoch": 6.666183471679688e-05,
      "step": 69900
    },
    {
      "epoch": 6.666183471679688e-05,
      "step": 69900,
      "training_step_time": 6.8357625007629395
    },
    {
      "epoch": 6.666278839111328e-05,
      "model_forward_time": 1.74595046043396,
      "step": 69901
    },
    {
      "epoch": 6.666278839111328e-05,
      "step": 69901,
      "training_step_time": 6.800315856933594
    },
    {
      "epoch": 6.666374206542969e-05,
      "model_forward_time": 1.7117400169372559,
      "step": 69902
    },
    {
      "epoch": 6.666374206542969e-05,
      "step": 69902,
      "training_step_time": 7.6568756103515625
    },
    {
      "epoch": 6.66646957397461e-05,
      "model_forward_time": 1.740931510925293,
      "step": 69903
    },
    {
      "epoch": 6.66646957397461e-05,
      "step": 69903,
      "training_step_time": 7.707505702972412
    },
    {
      "epoch": 6.66656494140625e-05,
      "model_forward_time": 1.8499510288238525,
      "step": 69904
    },
    {
      "epoch": 6.66656494140625e-05,
      "step": 69904,
      "training_step_time": 9.229085206985474
    },
    {
      "epoch": 6.666660308837891e-05,
      "model_forward_time": 2.8354854583740234,
      "step": 69905
    },
    {
      "epoch": 6.666660308837891e-05,
      "step": 69905,
      "training_step_time": 8.284552097320557
    },
    {
      "epoch": 6.666755676269532e-05,
      "model_forward_time": 1.740436315536499,
      "step": 69906
    },
    {
      "epoch": 6.666755676269532e-05,
      "step": 69906,
      "training_step_time": 7.879642724990845
    },
    {
      "epoch": 6.666851043701172e-05,
      "model_forward_time": 1.7427237033843994,
      "step": 69907
    },
    {
      "epoch": 6.666851043701172e-05,
      "step": 69907,
      "training_step_time": 6.717474699020386
    },
    {
      "epoch": 6.666946411132813e-05,
      "model_forward_time": 1.7409818172454834,
      "step": 69908
    },
    {
      "epoch": 6.666946411132813e-05,
      "step": 69908,
      "training_step_time": 6.753389596939087
    },
    {
      "epoch": 6.667041778564454e-05,
      "model_forward_time": 1.7536160945892334,
      "step": 69909
    },
    {
      "epoch": 6.667041778564454e-05,
      "step": 69909,
      "training_step_time": 6.762355089187622
    },
    {
      "epoch": 6.667137145996093e-05,
      "grad_norm": 0.0813794881105423,
      "learning_rate": 2.277868061729409e-06,
      "loss": 0.0523,
      "step": 69910
    },
    {
      "epoch": 6.667137145996093e-05,
      "model_forward_time": 1.7447428703308105,
      "step": 69910
    },
    {
      "dynamics_loss_avg": 0.05727077852934599,
      "epoch": 6.667137145996093e-05,
      "step": 69910
    },
    {
      "action_loss_avg": 0.001190312550170347,
      "epoch": 6.667137145996093e-05,
      "step": 69910
    },
    {
      "epoch": 6.667137145996093e-05,
      "step": 69910,
      "training_step_time": 6.671842098236084
    },
    {
      "epoch": 6.667232513427735e-05,
      "model_forward_time": 1.7538321018218994,
      "step": 69911
    },
    {
      "epoch": 6.667232513427735e-05,
      "step": 69911,
      "training_step_time": 7.634135723114014
    },
    {
      "epoch": 6.667327880859376e-05,
      "model_forward_time": 1.7538542747497559,
      "step": 69912
    },
    {
      "epoch": 6.667327880859376e-05,
      "step": 69912,
      "training_step_time": 7.684145450592041
    },
    {
      "epoch": 6.667423248291015e-05,
      "model_forward_time": 1.8274257183074951,
      "step": 69913
    },
    {
      "epoch": 6.667423248291015e-05,
      "step": 69913,
      "training_step_time": 10.141496419906616
    },
    {
      "epoch": 6.667518615722656e-05,
      "model_forward_time": 1.7382838726043701,
      "step": 69914
    },
    {
      "epoch": 6.667518615722656e-05,
      "step": 69914,
      "training_step_time": 9.697269201278687
    },
    {
      "epoch": 6.667613983154297e-05,
      "model_forward_time": 2.9126365184783936,
      "step": 69915
    },
    {
      "epoch": 6.667613983154297e-05,
      "step": 69915,
      "training_step_time": 7.878814458847046
    },
    {
      "epoch": 6.667709350585937e-05,
      "model_forward_time": 1.7471516132354736,
      "step": 69916
    },
    {
      "epoch": 6.667709350585937e-05,
      "step": 69916,
      "training_step_time": 6.686935186386108
    },
    {
      "epoch": 6.667804718017578e-05,
      "model_forward_time": 1.7438116073608398,
      "step": 69917
    },
    {
      "epoch": 6.667804718017578e-05,
      "step": 69917,
      "training_step_time": 6.720692873001099
    },
    {
      "epoch": 6.66790008544922e-05,
      "model_forward_time": 1.7610349655151367,
      "step": 69918
    },
    {
      "epoch": 6.66790008544922e-05,
      "step": 69918,
      "training_step_time": 6.693985462188721
    },
    {
      "epoch": 6.667995452880859e-05,
      "model_forward_time": 1.7564220428466797,
      "step": 69919
    },
    {
      "epoch": 6.667995452880859e-05,
      "step": 69919,
      "training_step_time": 6.73879861831665
    },
    {
      "epoch": 6.6680908203125e-05,
      "grad_norm": 0.10126514732837677,
      "learning_rate": 2.276481265855504e-06,
      "loss": 0.0504,
      "step": 69920
    },
    {
      "epoch": 6.6680908203125e-05,
      "model_forward_time": 1.7709681987762451,
      "step": 69920
    },
    {
      "dynamics_loss_avg": 0.05166164375841618,
      "epoch": 6.6680908203125e-05,
      "step": 69920
    },
    {
      "action_loss_avg": 0.0013477372121997178,
      "epoch": 6.6680908203125e-05,
      "step": 69920
    },
    {
      "epoch": 6.6680908203125e-05,
      "step": 69920,
      "training_step_time": 7.621715784072876
    },
    {
      "epoch": 6.668186187744141e-05,
      "model_forward_time": 1.760009527206421,
      "step": 69921
    },
    {
      "epoch": 6.668186187744141e-05,
      "step": 69921,
      "training_step_time": 7.716272592544556
    },
    {
      "epoch": 6.668281555175781e-05,
      "model_forward_time": 1.8312485218048096,
      "step": 69922
    },
    {
      "epoch": 6.668281555175781e-05,
      "step": 69922,
      "training_step_time": 8.72691822052002
    },
    {
      "epoch": 6.668376922607422e-05,
      "model_forward_time": 1.763148307800293,
      "step": 69923
    },
    {
      "epoch": 6.668376922607422e-05,
      "step": 69923,
      "training_step_time": 9.499612808227539
    },
    {
      "epoch": 6.668472290039063e-05,
      "model_forward_time": 2.9453647136688232,
      "step": 69924
    },
    {
      "epoch": 6.668472290039063e-05,
      "step": 69924,
      "training_step_time": 9.584277391433716
    },
    {
      "epoch": 6.668567657470703e-05,
      "model_forward_time": 1.7640318870544434,
      "step": 69925
    },
    {
      "epoch": 6.668567657470703e-05,
      "step": 69925,
      "training_step_time": 7.569866180419922
    },
    {
      "epoch": 6.668663024902344e-05,
      "model_forward_time": 1.7890968322753906,
      "step": 69926
    },
    {
      "epoch": 6.668663024902344e-05,
      "step": 69926,
      "training_step_time": 6.717934846878052
    },
    {
      "epoch": 6.668758392333985e-05,
      "model_forward_time": 1.7646737098693848,
      "step": 69927
    },
    {
      "epoch": 6.668758392333985e-05,
      "step": 69927,
      "training_step_time": 6.739311456680298
    },
    {
      "epoch": 6.668853759765625e-05,
      "model_forward_time": 1.7690999507904053,
      "step": 69928
    },
    {
      "epoch": 6.668853759765625e-05,
      "step": 69928,
      "training_step_time": 6.710571050643921
    },
    {
      "epoch": 6.668949127197266e-05,
      "model_forward_time": 1.7679567337036133,
      "step": 69929
    },
    {
      "epoch": 6.668949127197266e-05,
      "step": 69929,
      "training_step_time": 7.620285987854004
    },
    {
      "epoch": 6.669044494628907e-05,
      "grad_norm": 0.11327220499515533,
      "learning_rate": 2.275094767821513e-06,
      "loss": 0.0515,
      "step": 69930
    },
    {
      "epoch": 6.669044494628907e-05,
      "model_forward_time": 1.781083106994629,
      "step": 69930
    },
    {
      "dynamics_loss_avg": 0.053061875700950625,
      "epoch": 6.669044494628907e-05,
      "step": 69930
    },
    {
      "action_loss_avg": 0.0014908218727214262,
      "epoch": 6.669044494628907e-05,
      "step": 69930
    },
    {
      "epoch": 6.669044494628907e-05,
      "step": 69930,
      "training_step_time": 7.767985582351685
    },
    {
      "epoch": 6.669139862060547e-05,
      "model_forward_time": 1.7983567714691162,
      "step": 69931
    },
    {
      "epoch": 6.669139862060547e-05,
      "step": 69931,
      "training_step_time": 8.260353326797485
    },
    {
      "epoch": 6.669235229492188e-05,
      "model_forward_time": 1.7657580375671387,
      "step": 69932
    },
    {
      "epoch": 6.669235229492188e-05,
      "step": 69932,
      "training_step_time": 9.651875257492065
    },
    {
      "epoch": 6.669330596923829e-05,
      "model_forward_time": 2.923548698425293,
      "step": 69933
    },
    {
      "epoch": 6.669330596923829e-05,
      "step": 69933,
      "training_step_time": 9.059509754180908
    },
    {
      "epoch": 6.669425964355469e-05,
      "model_forward_time": 1.7703220844268799,
      "step": 69934
    },
    {
      "epoch": 6.669425964355469e-05,
      "step": 69934,
      "training_step_time": 7.856055498123169
    },
    {
      "epoch": 6.66952133178711e-05,
      "model_forward_time": 1.7669625282287598,
      "step": 69935
    },
    {
      "epoch": 6.66952133178711e-05,
      "step": 69935,
      "training_step_time": 6.73184609413147
    },
    {
      "epoch": 6.66961669921875e-05,
      "model_forward_time": 1.757843017578125,
      "step": 69936
    },
    {
      "epoch": 6.66961669921875e-05,
      "step": 69936,
      "training_step_time": 6.709820985794067
    },
    {
      "epoch": 6.66971206665039e-05,
      "model_forward_time": 1.7652626037597656,
      "step": 69937
    },
    {
      "epoch": 6.66971206665039e-05,
      "step": 69937,
      "training_step_time": 6.711469411849976
    },
    {
      "epoch": 6.669807434082031e-05,
      "model_forward_time": 1.764923334121704,
      "step": 69938
    },
    {
      "epoch": 6.669807434082031e-05,
      "step": 69938,
      "training_step_time": 7.651026248931885
    },
    {
      "epoch": 6.669902801513671e-05,
      "model_forward_time": 1.7752747535705566,
      "step": 69939
    },
    {
      "epoch": 6.669902801513671e-05,
      "step": 69939,
      "training_step_time": 7.756691932678223
    },
    {
      "epoch": 6.669998168945312e-05,
      "grad_norm": 0.10780124366283417,
      "learning_rate": 2.2737085677790617e-06,
      "loss": 0.0512,
      "step": 69940
    },
    {
      "epoch": 6.669998168945312e-05,
      "model_forward_time": 1.7722587585449219,
      "step": 69940
    },
    {
      "dynamics_loss_avg": 0.05703253783285618,
      "epoch": 6.669998168945312e-05,
      "step": 69940
    },
    {
      "action_loss_avg": 0.001939383745775558,
      "epoch": 6.669998168945312e-05,
      "step": 69940
    },
    {
      "epoch": 6.669998168945312e-05,
      "step": 69940,
      "training_step_time": 7.643062353134155
    },
    {
      "epoch": 6.670093536376953e-05,
      "model_forward_time": 1.7648062705993652,
      "step": 69941
    },
    {
      "epoch": 6.670093536376953e-05,
      "step": 69941,
      "training_step_time": 9.168504238128662
    },
    {
      "epoch": 6.670188903808593e-05,
      "model_forward_time": 2.87272572517395,
      "step": 69942
    },
    {
      "epoch": 6.670188903808593e-05,
      "step": 69942,
      "training_step_time": 9.300939083099365
    },
    {
      "epoch": 6.670284271240234e-05,
      "model_forward_time": 1.7302350997924805,
      "step": 69943
    },
    {
      "epoch": 6.670284271240234e-05,
      "step": 69943,
      "training_step_time": 7.827863693237305
    },
    {
      "epoch": 6.670379638671875e-05,
      "model_forward_time": 1.736861228942871,
      "step": 69944
    },
    {
      "epoch": 6.670379638671875e-05,
      "step": 69944,
      "training_step_time": 6.641082286834717
    },
    {
      "epoch": 6.670475006103515e-05,
      "model_forward_time": 1.7445707321166992,
      "step": 69945
    },
    {
      "epoch": 6.670475006103515e-05,
      "step": 69945,
      "training_step_time": 6.758263826370239
    },
    {
      "epoch": 6.670570373535156e-05,
      "model_forward_time": 1.731048583984375,
      "step": 69946
    },
    {
      "epoch": 6.670570373535156e-05,
      "step": 69946,
      "training_step_time": 6.734245300292969
    },
    {
      "epoch": 6.670665740966797e-05,
      "model_forward_time": 1.7486076354980469,
      "step": 69947
    },
    {
      "epoch": 6.670665740966797e-05,
      "step": 69947,
      "training_step_time": 7.671295881271362
    },
    {
      "epoch": 6.670761108398437e-05,
      "model_forward_time": 1.7544522285461426,
      "step": 69948
    },
    {
      "epoch": 6.670761108398437e-05,
      "step": 69948,
      "training_step_time": 7.797342300415039
    },
    {
      "epoch": 6.670856475830078e-05,
      "model_forward_time": 1.7406971454620361,
      "step": 69949
    },
    {
      "epoch": 6.670856475830078e-05,
      "step": 69949,
      "training_step_time": 7.6616411209106445
    },
    {
      "epoch": 6.670951843261719e-05,
      "grad_norm": 0.10463203489780426,
      "learning_rate": 2.2723226658797443e-06,
      "loss": 0.0532,
      "step": 69950
    },
    {
      "epoch": 6.670951843261719e-05,
      "model_forward_time": 1.8304219245910645,
      "step": 69950
    },
    {
      "dynamics_loss_avg": 0.047524459473788735,
      "epoch": 6.670951843261719e-05,
      "step": 69950
    },
    {
      "action_loss_avg": 0.0014308752201031894,
      "epoch": 6.670951843261719e-05,
      "step": 69950
    },
    {
      "epoch": 6.670951843261719e-05,
      "step": 69950,
      "training_step_time": 9.586439371109009
    },
    {
      "epoch": 6.671047210693359e-05,
      "model_forward_time": 2.8411402702331543,
      "step": 69951
    },
    {
      "epoch": 6.671047210693359e-05,
      "step": 69951,
      "training_step_time": 7.789696455001831
    },
    {
      "epoch": 6.671142578125e-05,
      "model_forward_time": 1.7375874519348145,
      "step": 69952
    },
    {
      "epoch": 6.671142578125e-05,
      "step": 69952,
      "training_step_time": 7.7992682456970215
    },
    {
      "epoch": 6.671237945556641e-05,
      "model_forward_time": 1.742995262145996,
      "step": 69953
    },
    {
      "epoch": 6.671237945556641e-05,
      "step": 69953,
      "training_step_time": 6.706557989120483
    },
    {
      "epoch": 6.67133331298828e-05,
      "model_forward_time": 1.7355732917785645,
      "step": 69954
    },
    {
      "epoch": 6.67133331298828e-05,
      "step": 69954,
      "training_step_time": 6.6952619552612305
    },
    {
      "epoch": 6.671428680419922e-05,
      "model_forward_time": 1.7451889514923096,
      "step": 69955
    },
    {
      "epoch": 6.671428680419922e-05,
      "step": 69955,
      "training_step_time": 6.7086522579193115
    },
    {
      "epoch": 6.671524047851563e-05,
      "model_forward_time": 1.7425239086151123,
      "step": 69956
    },
    {
      "epoch": 6.671524047851563e-05,
      "step": 69956,
      "training_step_time": 7.616740703582764
    },
    {
      "epoch": 6.671619415283203e-05,
      "model_forward_time": 1.7666089534759521,
      "step": 69957
    },
    {
      "epoch": 6.671619415283203e-05,
      "step": 69957,
      "training_step_time": 7.5883636474609375
    },
    {
      "epoch": 6.671714782714844e-05,
      "model_forward_time": 1.7503798007965088,
      "step": 69958
    },
    {
      "epoch": 6.671714782714844e-05,
      "step": 69958,
      "training_step_time": 7.626162528991699
    },
    {
      "epoch": 6.671810150146485e-05,
      "model_forward_time": 1.8367908000946045,
      "step": 69959
    },
    {
      "epoch": 6.671810150146485e-05,
      "step": 69959,
      "training_step_time": 9.204839706420898
    },
    {
      "epoch": 6.671905517578124e-05,
      "grad_norm": 0.1130177304148674,
      "learning_rate": 2.270937062275117e-06,
      "loss": 0.0566,
      "step": 69960
    },
    {
      "epoch": 6.671905517578124e-05,
      "model_forward_time": 1.7382917404174805,
      "step": 69960
    },
    {
      "dynamics_loss_avg": 0.054425022937357424,
      "epoch": 6.671905517578124e-05,
      "step": 69960
    },
    {
      "action_loss_avg": 0.002258783663273789,
      "epoch": 6.671905517578124e-05,
      "step": 69960
    },
    {
      "epoch": 6.671905517578124e-05,
      "step": 69960,
      "training_step_time": 8.237802267074585
    },
    {
      "epoch": 6.672000885009765e-05,
      "model_forward_time": 1.3845582008361816,
      "step": 69961
    },
    {
      "epoch": 6.672000885009765e-05,
      "step": 69961,
      "training_step_time": 8.805480003356934
    },
    {
      "epoch": 6.672096252441407e-05,
      "model_forward_time": 1.7340900897979736,
      "step": 69962
    },
    {
      "epoch": 6.672096252441407e-05,
      "step": 69962,
      "training_step_time": 6.7268171310424805
    },
    {
      "epoch": 6.672191619873046e-05,
      "model_forward_time": 1.755077600479126,
      "step": 69963
    },
    {
      "epoch": 6.672191619873046e-05,
      "step": 69963,
      "training_step_time": 6.750210523605347
    },
    {
      "epoch": 6.672286987304687e-05,
      "model_forward_time": 1.7570271492004395,
      "step": 69964
    },
    {
      "epoch": 6.672286987304687e-05,
      "step": 69964,
      "training_step_time": 6.750498533248901
    },
    {
      "epoch": 6.672382354736328e-05,
      "model_forward_time": 1.734994649887085,
      "step": 69965
    },
    {
      "epoch": 6.672382354736328e-05,
      "step": 69965,
      "training_step_time": 7.579901456832886
    },
    {
      "epoch": 6.672477722167968e-05,
      "model_forward_time": 1.7519779205322266,
      "step": 69966
    },
    {
      "epoch": 6.672477722167968e-05,
      "step": 69966,
      "training_step_time": 8.31207537651062
    },
    {
      "epoch": 6.672573089599609e-05,
      "model_forward_time": 1.7362713813781738,
      "step": 69967
    },
    {
      "epoch": 6.672573089599609e-05,
      "step": 69967,
      "training_step_time": 7.667870998382568
    },
    {
      "epoch": 6.67266845703125e-05,
      "model_forward_time": 1.4086554050445557,
      "step": 69968
    },
    {
      "epoch": 6.67266845703125e-05,
      "step": 69968,
      "training_step_time": 8.384795904159546
    },
    {
      "epoch": 6.67276382446289e-05,
      "model_forward_time": 1.7120542526245117,
      "step": 69969
    },
    {
      "epoch": 6.67276382446289e-05,
      "step": 69969,
      "training_step_time": 9.355907201766968
    },
    {
      "epoch": 6.672859191894531e-05,
      "grad_norm": 0.10361933708190918,
      "learning_rate": 2.2695517571167115e-06,
      "loss": 0.0544,
      "step": 69970
    },
    {
      "epoch": 6.672859191894531e-05,
      "model_forward_time": 2.913133144378662,
      "step": 69970
    },
    {
      "dynamics_loss_avg": 0.03898554537445307,
      "epoch": 6.672859191894531e-05,
      "step": 69970
    },
    {
      "action_loss_avg": 0.0012555342575069517,
      "epoch": 6.672859191894531e-05,
      "step": 69970
    },
    {
      "epoch": 6.672859191894531e-05,
      "step": 69970,
      "training_step_time": 8.71363377571106
    },
    {
      "epoch": 6.672954559326172e-05,
      "model_forward_time": 1.791978120803833,
      "step": 69971
    },
    {
      "epoch": 6.672954559326172e-05,
      "step": 69971,
      "training_step_time": 7.828124284744263
    },
    {
      "epoch": 6.673049926757812e-05,
      "model_forward_time": 1.7480506896972656,
      "step": 69972
    },
    {
      "epoch": 6.673049926757812e-05,
      "step": 69972,
      "training_step_time": 6.714161157608032
    },
    {
      "epoch": 6.673145294189453e-05,
      "model_forward_time": 1.738393783569336,
      "step": 69973
    },
    {
      "epoch": 6.673145294189453e-05,
      "step": 69973,
      "training_step_time": 6.749917507171631
    },
    {
      "epoch": 6.673240661621094e-05,
      "model_forward_time": 1.7450201511383057,
      "step": 69974
    },
    {
      "epoch": 6.673240661621094e-05,
      "step": 69974,
      "training_step_time": 8.055675745010376
    },
    {
      "epoch": 6.673336029052734e-05,
      "model_forward_time": 1.745509386062622,
      "step": 69975
    },
    {
      "epoch": 6.673336029052734e-05,
      "step": 69975,
      "training_step_time": 7.614420413970947
    },
    {
      "epoch": 6.673431396484375e-05,
      "model_forward_time": 1.043605089187622,
      "step": 69976
    },
    {
      "epoch": 6.673431396484375e-05,
      "step": 69976,
      "training_step_time": 7.711678981781006
    },
    {
      "epoch": 6.673526763916016e-05,
      "model_forward_time": 1.7441020011901855,
      "step": 69977
    },
    {
      "epoch": 6.673526763916016e-05,
      "step": 69977,
      "training_step_time": 8.636229515075684
    },
    {
      "epoch": 6.673622131347656e-05,
      "model_forward_time": 1.7547938823699951,
      "step": 69978
    },
    {
      "epoch": 6.673622131347656e-05,
      "step": 69978,
      "training_step_time": 8.567832708358765
    },
    {
      "epoch": 6.673717498779297e-05,
      "model_forward_time": 1.7512385845184326,
      "step": 69979
    },
    {
      "epoch": 6.673717498779297e-05,
      "step": 69979,
      "training_step_time": 9.948867321014404
    },
    {
      "epoch": 6.673812866210938e-05,
      "grad_norm": 0.1844317466020584,
      "learning_rate": 2.268166750556019e-06,
      "loss": 0.0532,
      "step": 69980
    },
    {
      "epoch": 6.673812866210938e-05,
      "model_forward_time": 1.732029914855957,
      "step": 69980
    },
    {
      "dynamics_loss_avg": 0.0443394884467125,
      "epoch": 6.673812866210938e-05,
      "step": 69980
    },
    {
      "action_loss_avg": 0.0012428986781742423,
      "epoch": 6.673812866210938e-05,
      "step": 69980
    },
    {
      "epoch": 6.673812866210938e-05,
      "step": 69980,
      "training_step_time": 7.664628028869629
    },
    {
      "epoch": 6.673908233642578e-05,
      "model_forward_time": 1.7556207180023193,
      "step": 69981
    },
    {
      "epoch": 6.673908233642578e-05,
      "step": 69981,
      "training_step_time": 6.723243713378906
    },
    {
      "epoch": 6.674003601074219e-05,
      "model_forward_time": 1.7563183307647705,
      "step": 69982
    },
    {
      "epoch": 6.674003601074219e-05,
      "step": 69982,
      "training_step_time": 6.737044334411621
    },
    {
      "epoch": 6.67409896850586e-05,
      "model_forward_time": 1.3816752433776855,
      "step": 69983
    },
    {
      "epoch": 6.67409896850586e-05,
      "step": 69983,
      "training_step_time": 7.093263864517212
    },
    {
      "epoch": 6.6741943359375e-05,
      "model_forward_time": 1.7359836101531982,
      "step": 69984
    },
    {
      "epoch": 6.6741943359375e-05,
      "step": 69984,
      "training_step_time": 7.625391960144043
    },
    {
      "epoch": 6.67428970336914e-05,
      "model_forward_time": 1.7564747333526611,
      "step": 69985
    },
    {
      "epoch": 6.67428970336914e-05,
      "step": 69985,
      "training_step_time": 7.715819358825684
    },
    {
      "epoch": 6.674385070800782e-05,
      "model_forward_time": 1.757110595703125,
      "step": 69986
    },
    {
      "epoch": 6.674385070800782e-05,
      "step": 69986,
      "training_step_time": 8.768815279006958
    },
    {
      "epoch": 6.674480438232421e-05,
      "model_forward_time": 1.1061251163482666,
      "step": 69987
    },
    {
      "epoch": 6.674480438232421e-05,
      "step": 69987,
      "training_step_time": 9.289374351501465
    },
    {
      "epoch": 6.674575805664062e-05,
      "model_forward_time": 1.7266826629638672,
      "step": 69988
    },
    {
      "epoch": 6.674575805664062e-05,
      "step": 69988,
      "training_step_time": 8.840132236480713
    },
    {
      "epoch": 6.674671173095704e-05,
      "model_forward_time": 2.8762784004211426,
      "step": 69989
    },
    {
      "epoch": 6.674671173095704e-05,
      "step": 69989,
      "training_step_time": 7.877244710922241
    },
    {
      "epoch": 6.674766540527343e-05,
      "grad_norm": 0.0932563841342926,
      "learning_rate": 2.266782042744503e-06,
      "loss": 0.0547,
      "step": 69990
    },
    {
      "epoch": 6.674766540527343e-05,
      "model_forward_time": 1.7374744415283203,
      "step": 69990
    },
    {
      "dynamics_loss_avg": 0.03778860680758953,
      "epoch": 6.674766540527343e-05,
      "step": 69990
    },
    {
      "action_loss_avg": 0.001999897591304034,
      "epoch": 6.674766540527343e-05,
      "step": 69990
    },
    {
      "epoch": 6.674766540527343e-05,
      "step": 69990,
      "training_step_time": 6.843430042266846
    },
    {
      "epoch": 6.674861907958984e-05,
      "model_forward_time": 1.747344732284546,
      "step": 69991
    },
    {
      "epoch": 6.674861907958984e-05,
      "step": 69991,
      "training_step_time": 6.791754484176636
    },
    {
      "epoch": 6.674957275390625e-05,
      "model_forward_time": 1.7409281730651855,
      "step": 69992
    },
    {
      "epoch": 6.674957275390625e-05,
      "step": 69992,
      "training_step_time": 7.064870595932007
    },
    {
      "epoch": 6.675052642822265e-05,
      "model_forward_time": 1.7588660717010498,
      "step": 69993
    },
    {
      "epoch": 6.675052642822265e-05,
      "step": 69993,
      "training_step_time": 7.582744598388672
    },
    {
      "epoch": 6.675148010253906e-05,
      "model_forward_time": 1.7505979537963867,
      "step": 69994
    },
    {
      "epoch": 6.675148010253906e-05,
      "step": 69994,
      "training_step_time": 7.519451141357422
    },
    {
      "epoch": 6.675243377685547e-05,
      "model_forward_time": 1.747910976409912,
      "step": 69995
    },
    {
      "epoch": 6.675243377685547e-05,
      "step": 69995,
      "training_step_time": 7.794250011444092
    },
    {
      "epoch": 6.675338745117187e-05,
      "model_forward_time": 1.843372106552124,
      "step": 69996
    },
    {
      "epoch": 6.675338745117187e-05,
      "step": 69996,
      "training_step_time": 9.338156938552856
    },
    {
      "epoch": 6.675434112548828e-05,
      "model_forward_time": 1.7560527324676514,
      "step": 69997
    },
    {
      "epoch": 6.675434112548828e-05,
      "step": 69997,
      "training_step_time": 9.815471649169922
    },
    {
      "epoch": 6.675529479980469e-05,
      "model_forward_time": 2.9789671897888184,
      "step": 69998
    },
    {
      "epoch": 6.675529479980469e-05,
      "step": 69998,
      "training_step_time": 8.009467601776123
    },
    {
      "epoch": 6.675624847412109e-05,
      "model_forward_time": 1.7673001289367676,
      "step": 69999
    },
    {
      "epoch": 6.675624847412109e-05,
      "step": 69999,
      "training_step_time": 7.600813627243042
    },
    {
      "epoch": 6.67572021484375e-05,
      "grad_norm": 0.07735328376293182,
      "learning_rate": 2.2653976338335936e-06,
      "loss": 0.0535,
      "step": 70000
    },
    {
      "epoch": 6.67572021484375e-05,
      "model_forward_time": 1.7696428298950195,
      "step": 70000
    },
    {
      "dynamics_loss_avg": 0.042465108633041385,
      "epoch": 6.67572021484375e-05,
      "step": 70000
    },
    {
      "action_loss_avg": 0.001327129773562774,
      "epoch": 6.67572021484375e-05,
      "step": 70000
    },
    {
      "epoch": 6.67572021484375e-05,
      "step": 70000,
      "training_step_time": 6.594804525375366
    },
    {
      "epoch": 6.675815582275391e-05,
      "model_forward_time": 1.7148845195770264,
      "step": 70001
    },
    {
      "epoch": 6.675815582275391e-05,
      "step": 70001,
      "training_step_time": 7.176732540130615
    },
    {
      "epoch": 6.675910949707031e-05,
      "model_forward_time": 1.7117106914520264,
      "step": 70002
    },
    {
      "epoch": 6.675910949707031e-05,
      "step": 70002,
      "training_step_time": 7.7087624073028564
    },
    {
      "epoch": 6.676006317138672e-05,
      "model_forward_time": 1.7056307792663574,
      "step": 70003
    },
    {
      "epoch": 6.676006317138672e-05,
      "step": 70003,
      "training_step_time": 7.788209676742554
    },
    {
      "epoch": 6.676101684570313e-05,
      "model_forward_time": 1.720304250717163,
      "step": 70004
    },
    {
      "epoch": 6.676101684570313e-05,
      "step": 70004,
      "training_step_time": 9.35436463356018
    },
    {
      "epoch": 6.676197052001953e-05,
      "model_forward_time": 1.7134666442871094,
      "step": 70005
    },
    {
      "epoch": 6.676197052001953e-05,
      "step": 70005,
      "training_step_time": 9.931092023849487
    },
    {
      "epoch": 6.676292419433594e-05,
      "model_forward_time": 1.7575979232788086,
      "step": 70006
    },
    {
      "epoch": 6.676292419433594e-05,
      "step": 70006,
      "training_step_time": 10.425070762634277
    },
    {
      "epoch": 6.676387786865235e-05,
      "model_forward_time": 2.7699692249298096,
      "step": 70007
    },
    {
      "epoch": 6.676387786865235e-05,
      "step": 70007,
      "training_step_time": 7.807011842727661
    },
    {
      "epoch": 6.676483154296875e-05,
      "model_forward_time": 1.7035818099975586,
      "step": 70008
    },
    {
      "epoch": 6.676483154296875e-05,
      "step": 70008,
      "training_step_time": 7.65893816947937
    },
    {
      "epoch": 6.676578521728516e-05,
      "model_forward_time": 1.710719347000122,
      "step": 70009
    },
    {
      "epoch": 6.676578521728516e-05,
      "step": 70009,
      "training_step_time": 6.817511558532715
    },
    {
      "epoch": 6.676673889160157e-05,
      "grad_norm": 0.09792298823595047,
      "learning_rate": 2.2640135239746896e-06,
      "loss": 0.0554,
      "step": 70010
    },
    {
      "epoch": 6.676673889160157e-05,
      "model_forward_time": 1.6999428272247314,
      "step": 70010
    },
    {
      "dynamics_loss_avg": 0.05496894530951977,
      "epoch": 6.676673889160157e-05,
      "step": 70010
    },
    {
      "action_loss_avg": 0.0013215235783718526,
      "epoch": 6.676673889160157e-05,
      "step": 70010
    },
    {
      "epoch": 6.676673889160157e-05,
      "step": 70010,
      "training_step_time": 7.170311212539673
    },
    {
      "epoch": 6.676769256591796e-05,
      "model_forward_time": 1.7187471389770508,
      "step": 70011
    },
    {
      "epoch": 6.676769256591796e-05,
      "step": 70011,
      "training_step_time": 7.633200645446777
    },
    {
      "epoch": 6.676864624023438e-05,
      "model_forward_time": 1.7083680629730225,
      "step": 70012
    },
    {
      "epoch": 6.676864624023438e-05,
      "step": 70012,
      "training_step_time": 8.755742311477661
    },
    {
      "epoch": 6.676959991455079e-05,
      "model_forward_time": 1.7149503231048584,
      "step": 70013
    },
    {
      "epoch": 6.676959991455079e-05,
      "step": 70013,
      "training_step_time": 7.743476152420044
    },
    {
      "epoch": 6.677055358886718e-05,
      "model_forward_time": 1.7045009136199951,
      "step": 70014
    },
    {
      "epoch": 6.677055358886718e-05,
      "step": 70014,
      "training_step_time": 9.441740274429321
    },
    {
      "epoch": 6.67715072631836e-05,
      "model_forward_time": 1.7229185104370117,
      "step": 70015
    },
    {
      "epoch": 6.67715072631836e-05,
      "step": 70015,
      "training_step_time": 9.810701847076416
    },
    {
      "epoch": 6.67724609375e-05,
      "model_forward_time": 2.2910664081573486,
      "step": 70016
    },
    {
      "epoch": 6.67724609375e-05,
      "step": 70016,
      "training_step_time": 10.546857595443726
    },
    {
      "epoch": 6.67734146118164e-05,
      "model_forward_time": 1.7279598712921143,
      "step": 70017
    },
    {
      "epoch": 6.67734146118164e-05,
      "step": 70017,
      "training_step_time": 7.738588571548462
    },
    {
      "epoch": 6.677436828613281e-05,
      "model_forward_time": 1.707529067993164,
      "step": 70018
    },
    {
      "epoch": 6.677436828613281e-05,
      "step": 70018,
      "training_step_time": 6.805430889129639
    },
    {
      "epoch": 6.677532196044922e-05,
      "model_forward_time": 1.707231044769287,
      "step": 70019
    },
    {
      "epoch": 6.677532196044922e-05,
      "step": 70019,
      "training_step_time": 7.292365074157715
    },
    {
      "epoch": 6.677627563476562e-05,
      "grad_norm": 0.10314104706048965,
      "learning_rate": 2.2626297133191517e-06,
      "loss": 0.0575,
      "step": 70020
    },
    {
      "epoch": 6.677627563476562e-05,
      "model_forward_time": 1.7044670581817627,
      "step": 70020
    },
    {
      "dynamics_loss_avg": 0.05465585496276617,
      "epoch": 6.677627563476562e-05,
      "step": 70020
    },
    {
      "action_loss_avg": 0.0021883954221266324,
      "epoch": 6.677627563476562e-05,
      "step": 70020
    },
    {
      "epoch": 6.677627563476562e-05,
      "step": 70020,
      "training_step_time": 7.603731393814087
    },
    {
      "epoch": 6.677722930908203e-05,
      "model_forward_time": 1.7026395797729492,
      "step": 70021
    },
    {
      "epoch": 6.677722930908203e-05,
      "step": 70021,
      "training_step_time": 7.6802544593811035
    },
    {
      "epoch": 6.677818298339844e-05,
      "model_forward_time": 1.701106071472168,
      "step": 70022
    },
    {
      "epoch": 6.677818298339844e-05,
      "step": 70022,
      "training_step_time": 8.182175397872925
    },
    {
      "epoch": 6.677913665771484e-05,
      "model_forward_time": 1.7052197456359863,
      "step": 70023
    },
    {
      "epoch": 6.677913665771484e-05,
      "step": 70023,
      "training_step_time": 8.525056838989258
    },
    {
      "epoch": 6.678009033203125e-05,
      "model_forward_time": 1.700871467590332,
      "step": 70024
    },
    {
      "epoch": 6.678009033203125e-05,
      "step": 70024,
      "training_step_time": 7.67165207862854
    },
    {
      "epoch": 6.678104400634766e-05,
      "model_forward_time": 1.71968674659729,
      "step": 70025
    },
    {
      "epoch": 6.678104400634766e-05,
      "step": 70025,
      "training_step_time": 9.95798921585083
    },
    {
      "epoch": 6.678199768066406e-05,
      "model_forward_time": 1.706226110458374,
      "step": 70026
    },
    {
      "epoch": 6.678199768066406e-05,
      "step": 70026,
      "training_step_time": 7.7293455600738525
    },
    {
      "epoch": 6.678295135498047e-05,
      "model_forward_time": 1.7208220958709717,
      "step": 70027
    },
    {
      "epoch": 6.678295135498047e-05,
      "step": 70027,
      "training_step_time": 7.109747648239136
    },
    {
      "epoch": 6.678390502929688e-05,
      "model_forward_time": 1.3347876071929932,
      "step": 70028
    },
    {
      "epoch": 6.678390502929688e-05,
      "step": 70028,
      "training_step_time": 7.142497301101685
    },
    {
      "epoch": 6.678485870361328e-05,
      "model_forward_time": 1.7188341617584229,
      "step": 70029
    },
    {
      "epoch": 6.678485870361328e-05,
      "step": 70029,
      "training_step_time": 7.543622016906738
    },
    {
      "epoch": 6.678581237792969e-05,
      "grad_norm": 0.1265346109867096,
      "learning_rate": 2.261246202018312e-06,
      "loss": 0.0576,
      "step": 70030
    },
    {
      "epoch": 6.678581237792969e-05,
      "model_forward_time": 1.7059030532836914,
      "step": 70030
    },
    {
      "dynamics_loss_avg": 0.05842584278434515,
      "epoch": 6.678581237792969e-05,
      "step": 70030
    },
    {
      "action_loss_avg": 0.002168859663652256,
      "epoch": 6.678581237792969e-05,
      "step": 70030
    },
    {
      "epoch": 6.678581237792969e-05,
      "step": 70030,
      "training_step_time": 7.635905027389526
    },
    {
      "epoch": 6.67867660522461e-05,
      "model_forward_time": 1.7161729335784912,
      "step": 70031
    },
    {
      "epoch": 6.67867660522461e-05,
      "step": 70031,
      "training_step_time": 8.480350732803345
    },
    {
      "epoch": 6.67877197265625e-05,
      "model_forward_time": 1.706979513168335,
      "step": 70032
    },
    {
      "epoch": 6.67877197265625e-05,
      "step": 70032,
      "training_step_time": 7.749377012252808
    },
    {
      "epoch": 6.678867340087891e-05,
      "model_forward_time": 1.7187957763671875,
      "step": 70033
    },
    {
      "epoch": 6.678867340087891e-05,
      "step": 70033,
      "training_step_time": 10.937713861465454
    },
    {
      "epoch": 6.678962707519532e-05,
      "model_forward_time": 1.787374496459961,
      "step": 70034
    },
    {
      "epoch": 6.678962707519532e-05,
      "step": 70034,
      "training_step_time": 8.565624713897705
    },
    {
      "epoch": 6.679058074951172e-05,
      "model_forward_time": 2.2762506008148193,
      "step": 70035
    },
    {
      "epoch": 6.679058074951172e-05,
      "step": 70035,
      "training_step_time": 7.735714673995972
    },
    {
      "epoch": 6.679153442382813e-05,
      "model_forward_time": 1.70814847946167,
      "step": 70036
    },
    {
      "epoch": 6.679153442382813e-05,
      "step": 70036,
      "training_step_time": 7.074108362197876
    },
    {
      "epoch": 6.679248809814454e-05,
      "model_forward_time": 1.7085180282592773,
      "step": 70037
    },
    {
      "epoch": 6.679248809814454e-05,
      "step": 70037,
      "training_step_time": 7.106914281845093
    },
    {
      "epoch": 6.679344177246093e-05,
      "model_forward_time": 1.7160942554473877,
      "step": 70038
    },
    {
      "epoch": 6.679344177246093e-05,
      "step": 70038,
      "training_step_time": 6.744130849838257
    },
    {
      "epoch": 6.679439544677735e-05,
      "model_forward_time": 1.3436195850372314,
      "step": 70039
    },
    {
      "epoch": 6.679439544677735e-05,
      "step": 70039,
      "training_step_time": 8.553333759307861
    },
    {
      "epoch": 6.679534912109376e-05,
      "grad_norm": 0.08247705549001694,
      "learning_rate": 2.259862990223472e-06,
      "loss": 0.0581,
      "step": 70040
    },
    {
      "epoch": 6.679534912109376e-05,
      "model_forward_time": 1.7050139904022217,
      "step": 70040
    },
    {
      "dynamics_loss_avg": 0.06901665888726712,
      "epoch": 6.679534912109376e-05,
      "step": 70040
    },
    {
      "action_loss_avg": 0.0011586111679207534,
      "epoch": 6.679534912109376e-05,
      "step": 70040
    },
    {
      "epoch": 6.679534912109376e-05,
      "step": 70040,
      "training_step_time": 7.63648247718811
    },
    {
      "epoch": 6.679630279541015e-05,
      "model_forward_time": 1.7140758037567139,
      "step": 70041
    },
    {
      "epoch": 6.679630279541015e-05,
      "step": 70041,
      "training_step_time": 8.495170593261719
    },
    {
      "epoch": 6.679725646972656e-05,
      "model_forward_time": 1.7192459106445312,
      "step": 70042
    },
    {
      "epoch": 6.679725646972656e-05,
      "step": 70042,
      "training_step_time": 9.007521629333496
    },
    {
      "epoch": 6.679821014404297e-05,
      "model_forward_time": 1.7660858631134033,
      "step": 70043
    },
    {
      "epoch": 6.679821014404297e-05,
      "step": 70043,
      "training_step_time": 8.701805830001831
    },
    {
      "epoch": 6.679916381835937e-05,
      "model_forward_time": 2.3078839778900146,
      "step": 70044
    },
    {
      "epoch": 6.679916381835937e-05,
      "step": 70044,
      "training_step_time": 7.732184410095215
    },
    {
      "epoch": 6.680011749267578e-05,
      "model_forward_time": 1.3275761604309082,
      "step": 70045
    },
    {
      "epoch": 6.680011749267578e-05,
      "step": 70045,
      "training_step_time": 7.1039252281188965
    },
    {
      "epoch": 6.68010711669922e-05,
      "model_forward_time": 1.7119698524475098,
      "step": 70046
    },
    {
      "epoch": 6.68010711669922e-05,
      "step": 70046,
      "training_step_time": 6.779183626174927
    },
    {
      "epoch": 6.680202484130859e-05,
      "model_forward_time": 1.7204723358154297,
      "step": 70047
    },
    {
      "epoch": 6.680202484130859e-05,
      "step": 70047,
      "training_step_time": 7.14475154876709
    },
    {
      "epoch": 6.6802978515625e-05,
      "model_forward_time": 1.7192273139953613,
      "step": 70048
    },
    {
      "epoch": 6.6802978515625e-05,
      "step": 70048,
      "training_step_time": 7.565103054046631
    },
    {
      "epoch": 6.680393218994141e-05,
      "model_forward_time": 1.7144219875335693,
      "step": 70049
    },
    {
      "epoch": 6.680393218994141e-05,
      "step": 70049,
      "training_step_time": 7.659301042556763
    },
    {
      "epoch": 6.680488586425781e-05,
      "grad_norm": 0.09450996667146683,
      "learning_rate": 2.258480078085894e-06,
      "loss": 0.0583,
      "step": 70050
    },
    {
      "epoch": 6.680488586425781e-05,
      "model_forward_time": 1.7089455127716064,
      "step": 70050
    },
    {
      "dynamics_loss_avg": 0.06707279346883296,
      "epoch": 6.680488586425781e-05,
      "step": 70050
    },
    {
      "action_loss_avg": 0.0020514484320301564,
      "epoch": 6.680488586425781e-05,
      "step": 70050
    },
    {
      "epoch": 6.680488586425781e-05,
      "step": 70050,
      "training_step_time": 8.730172872543335
    },
    {
      "epoch": 6.680583953857422e-05,
      "model_forward_time": 1.7071611881256104,
      "step": 70051
    },
    {
      "epoch": 6.680583953857422e-05,
      "step": 70051,
      "training_step_time": 8.291614294052124
    },
    {
      "epoch": 6.680679321289063e-05,
      "model_forward_time": 1.7476608753204346,
      "step": 70052
    },
    {
      "epoch": 6.680679321289063e-05,
      "step": 70052,
      "training_step_time": 9.551789283752441
    },
    {
      "epoch": 6.680774688720703e-05,
      "model_forward_time": 2.221775531768799,
      "step": 70053
    },
    {
      "epoch": 6.680774688720703e-05,
      "step": 70053,
      "training_step_time": 10.210502624511719
    },
    {
      "epoch": 6.680870056152344e-05,
      "model_forward_time": 1.7147817611694336,
      "step": 70054
    },
    {
      "epoch": 6.680870056152344e-05,
      "step": 70054,
      "training_step_time": 8.36872673034668
    },
    {
      "epoch": 6.680965423583985e-05,
      "model_forward_time": 1.7216265201568604,
      "step": 70055
    },
    {
      "epoch": 6.680965423583985e-05,
      "step": 70055,
      "training_step_time": 6.792009115219116
    },
    {
      "epoch": 6.681060791015625e-05,
      "model_forward_time": 1.7006313800811768,
      "step": 70056
    },
    {
      "epoch": 6.681060791015625e-05,
      "step": 70056,
      "training_step_time": 7.106701374053955
    },
    {
      "epoch": 6.681156158447266e-05,
      "model_forward_time": 1.7088162899017334,
      "step": 70057
    },
    {
      "epoch": 6.681156158447266e-05,
      "step": 70057,
      "training_step_time": 7.597411155700684
    },
    {
      "epoch": 6.681251525878907e-05,
      "model_forward_time": 1.3511137962341309,
      "step": 70058
    },
    {
      "epoch": 6.681251525878907e-05,
      "step": 70058,
      "training_step_time": 7.676080226898193
    },
    {
      "epoch": 6.681346893310547e-05,
      "model_forward_time": 1.709113597869873,
      "step": 70059
    },
    {
      "epoch": 6.681346893310547e-05,
      "step": 70059,
      "training_step_time": 7.691992282867432
    },
    {
      "epoch": 6.681442260742188e-05,
      "grad_norm": 0.13202162086963654,
      "learning_rate": 2.257097465756813e-06,
      "loss": 0.0555,
      "step": 70060
    },
    {
      "epoch": 6.681442260742188e-05,
      "model_forward_time": 1.7033758163452148,
      "step": 70060
    },
    {
      "dynamics_loss_avg": 0.05251271575689316,
      "epoch": 6.681442260742188e-05,
      "step": 70060
    },
    {
      "action_loss_avg": 0.0013966611411888152,
      "epoch": 6.681442260742188e-05,
      "step": 70060
    },
    {
      "epoch": 6.681442260742188e-05,
      "step": 70060,
      "training_step_time": 8.520050764083862
    },
    {
      "epoch": 6.681537628173829e-05,
      "model_forward_time": 1.7277348041534424,
      "step": 70061
    },
    {
      "epoch": 6.681537628173829e-05,
      "step": 70061,
      "training_step_time": 9.924969911575317
    },
    {
      "epoch": 6.681632995605469e-05,
      "model_forward_time": 1.7210302352905273,
      "step": 70062
    },
    {
      "epoch": 6.681632995605469e-05,
      "step": 70062,
      "training_step_time": 9.326306104660034
    },
    {
      "epoch": 6.68172836303711e-05,
      "model_forward_time": 2.265998125076294,
      "step": 70063
    },
    {
      "epoch": 6.68172836303711e-05,
      "step": 70063,
      "training_step_time": 7.740333318710327
    },
    {
      "epoch": 6.68182373046875e-05,
      "model_forward_time": 1.7114684581756592,
      "step": 70064
    },
    {
      "epoch": 6.68182373046875e-05,
      "step": 70064,
      "training_step_time": 7.101257562637329
    },
    {
      "epoch": 6.68191909790039e-05,
      "model_forward_time": 1.3276476860046387,
      "step": 70065
    },
    {
      "epoch": 6.68191909790039e-05,
      "step": 70065,
      "training_step_time": 6.721402883529663
    },
    {
      "epoch": 6.682014465332031e-05,
      "model_forward_time": 1.7084338665008545,
      "step": 70066
    },
    {
      "epoch": 6.682014465332031e-05,
      "step": 70066,
      "training_step_time": 7.504009246826172
    },
    {
      "epoch": 6.682109832763671e-05,
      "model_forward_time": 1.708012580871582,
      "step": 70067
    },
    {
      "epoch": 6.682109832763671e-05,
      "step": 70067,
      "training_step_time": 8.67326545715332
    },
    {
      "epoch": 6.682205200195312e-05,
      "model_forward_time": 1.715902328491211,
      "step": 70068
    },
    {
      "epoch": 6.682205200195312e-05,
      "step": 70068,
      "training_step_time": 7.648100852966309
    },
    {
      "epoch": 6.682300567626953e-05,
      "model_forward_time": 1.6982433795928955,
      "step": 70069
    },
    {
      "epoch": 6.682300567626953e-05,
      "step": 70069,
      "training_step_time": 8.514735460281372
    },
    {
      "epoch": 6.682395935058593e-05,
      "grad_norm": 0.15210291743278503,
      "learning_rate": 2.2557151533874287e-06,
      "loss": 0.0529,
      "step": 70070
    },
    {
      "epoch": 6.682395935058593e-05,
      "model_forward_time": 1.6970467567443848,
      "step": 70070
    },
    {
      "dynamics_loss_avg": 0.046432977076619864,
      "epoch": 6.682395935058593e-05,
      "step": 70070
    },
    {
      "action_loss_avg": 0.0014519728778395803,
      "epoch": 6.682395935058593e-05,
      "step": 70070
    },
    {
      "epoch": 6.682395935058593e-05,
      "step": 70070,
      "training_step_time": 10.051871538162231
    },
    {
      "epoch": 6.682491302490234e-05,
      "model_forward_time": 1.6986098289489746,
      "step": 70071
    },
    {
      "epoch": 6.682491302490234e-05,
      "step": 70071,
      "training_step_time": 8.711968183517456
    },
    {
      "epoch": 6.682586669921875e-05,
      "model_forward_time": 2.2886099815368652,
      "step": 70072
    },
    {
      "epoch": 6.682586669921875e-05,
      "step": 70072,
      "training_step_time": 7.700855255126953
    },
    {
      "epoch": 6.682682037353515e-05,
      "model_forward_time": 1.7031841278076172,
      "step": 70073
    },
    {
      "epoch": 6.682682037353515e-05,
      "step": 70073,
      "training_step_time": 7.073599815368652
    },
    {
      "epoch": 6.682777404785156e-05,
      "model_forward_time": 1.7210655212402344,
      "step": 70074
    },
    {
      "epoch": 6.682777404785156e-05,
      "step": 70074,
      "training_step_time": 6.725973844528198
    },
    {
      "epoch": 6.682872772216797e-05,
      "model_forward_time": 1.7093701362609863,
      "step": 70075
    },
    {
      "epoch": 6.682872772216797e-05,
      "step": 70075,
      "training_step_time": 7.110816478729248
    },
    {
      "epoch": 6.682968139648437e-05,
      "model_forward_time": 1.710468053817749,
      "step": 70076
    },
    {
      "epoch": 6.682968139648437e-05,
      "step": 70076,
      "training_step_time": 7.610079765319824
    },
    {
      "epoch": 6.683063507080078e-05,
      "model_forward_time": 1.7062077522277832,
      "step": 70077
    },
    {
      "epoch": 6.683063507080078e-05,
      "step": 70077,
      "training_step_time": 7.634933710098267
    },
    {
      "epoch": 6.683158874511719e-05,
      "model_forward_time": 1.699141502380371,
      "step": 70078
    },
    {
      "epoch": 6.683158874511719e-05,
      "step": 70078,
      "training_step_time": 9.37918472290039
    },
    {
      "epoch": 6.683254241943359e-05,
      "model_forward_time": 1.3243279457092285,
      "step": 70079
    },
    {
      "epoch": 6.683254241943359e-05,
      "step": 70079,
      "training_step_time": 8.726007461547852
    },
    {
      "epoch": 6.683349609375e-05,
      "grad_norm": 0.10164584219455719,
      "learning_rate": 2.2543331411289116e-06,
      "loss": 0.0536,
      "step": 70080
    },
    {
      "epoch": 6.683349609375e-05,
      "model_forward_time": 1.7839527130126953,
      "step": 70080
    },
    {
      "dynamics_loss_avg": 0.05360569767653942,
      "epoch": 6.683349609375e-05,
      "step": 70080
    },
    {
      "action_loss_avg": 0.0013535900507122278,
      "epoch": 6.683349609375e-05,
      "step": 70080
    },
    {
      "epoch": 6.683349609375e-05,
      "step": 70080,
      "training_step_time": 7.882903814315796
    },
    {
      "epoch": 6.683444976806641e-05,
      "model_forward_time": 1.7106952667236328,
      "step": 70081
    },
    {
      "epoch": 6.683444976806641e-05,
      "step": 70081,
      "training_step_time": 7.752617359161377
    },
    {
      "epoch": 6.68354034423828e-05,
      "model_forward_time": 1.7084481716156006,
      "step": 70082
    },
    {
      "epoch": 6.68354034423828e-05,
      "step": 70082,
      "training_step_time": 7.099500894546509
    },
    {
      "epoch": 6.683635711669922e-05,
      "model_forward_time": 1.7089207172393799,
      "step": 70083
    },
    {
      "epoch": 6.683635711669922e-05,
      "step": 70083,
      "training_step_time": 6.772930383682251
    },
    {
      "epoch": 6.683731079101563e-05,
      "model_forward_time": 1.7022991180419922,
      "step": 70084
    },
    {
      "epoch": 6.683731079101563e-05,
      "step": 70084,
      "training_step_time": 7.12296986579895
    },
    {
      "epoch": 6.683826446533203e-05,
      "model_forward_time": 1.7120513916015625,
      "step": 70085
    },
    {
      "epoch": 6.683826446533203e-05,
      "step": 70085,
      "training_step_time": 7.5408642292022705
    },
    {
      "epoch": 6.683921813964844e-05,
      "model_forward_time": 1.6986358165740967,
      "step": 70086
    },
    {
      "epoch": 6.683921813964844e-05,
      "step": 70086,
      "training_step_time": 7.632092237472534
    },
    {
      "epoch": 6.684017181396485e-05,
      "model_forward_time": 1.7048099040985107,
      "step": 70087
    },
    {
      "epoch": 6.684017181396485e-05,
      "step": 70087,
      "training_step_time": 8.50687289237976
    },
    {
      "epoch": 6.684112548828124e-05,
      "model_forward_time": 1.7119991779327393,
      "step": 70088
    },
    {
      "epoch": 6.684112548828124e-05,
      "step": 70088,
      "training_step_time": 8.288569450378418
    },
    {
      "epoch": 6.684207916259765e-05,
      "model_forward_time": 1.7709603309631348,
      "step": 70089
    },
    {
      "epoch": 6.684207916259765e-05,
      "step": 70089,
      "training_step_time": 8.707989931106567
    },
    {
      "epoch": 6.684303283691407e-05,
      "grad_norm": 0.10381612926721573,
      "learning_rate": 2.2529514291323923e-06,
      "loss": 0.0597,
      "step": 70090
    },
    {
      "epoch": 6.684303283691407e-05,
      "model_forward_time": 1.7063546180725098,
      "step": 70090
    },
    {
      "dynamics_loss_avg": 0.07332682758569717,
      "epoch": 6.684303283691407e-05,
      "step": 70090
    },
    {
      "action_loss_avg": 0.0020906660240143536,
      "epoch": 6.684303283691407e-05,
      "step": 70090
    },
    {
      "epoch": 6.684303283691407e-05,
      "step": 70090,
      "training_step_time": 7.689340829849243
    },
    {
      "epoch": 6.684398651123046e-05,
      "model_forward_time": 2.252642869949341,
      "step": 70091
    },
    {
      "epoch": 6.684398651123046e-05,
      "step": 70091,
      "training_step_time": 7.0814902782440186
    },
    {
      "epoch": 6.684494018554687e-05,
      "model_forward_time": 1.7059576511383057,
      "step": 70092
    },
    {
      "epoch": 6.684494018554687e-05,
      "step": 70092,
      "training_step_time": 6.718962907791138
    },
    {
      "epoch": 6.684589385986328e-05,
      "model_forward_time": 1.6986870765686035,
      "step": 70093
    },
    {
      "epoch": 6.684589385986328e-05,
      "step": 70093,
      "training_step_time": 7.091359376907349
    },
    {
      "epoch": 6.684684753417968e-05,
      "model_forward_time": 1.7042176723480225,
      "step": 70094
    },
    {
      "epoch": 6.684684753417968e-05,
      "step": 70094,
      "training_step_time": 7.467068195343018
    },
    {
      "epoch": 6.684780120849609e-05,
      "model_forward_time": 1.3270289897918701,
      "step": 70095
    },
    {
      "epoch": 6.684780120849609e-05,
      "step": 70095,
      "training_step_time": 8.339726209640503
    },
    {
      "epoch": 6.68487548828125e-05,
      "model_forward_time": 1.6953620910644531,
      "step": 70096
    },
    {
      "epoch": 6.68487548828125e-05,
      "step": 70096,
      "training_step_time": 7.8174872398376465
    },
    {
      "epoch": 6.68497085571289e-05,
      "model_forward_time": 1.7047934532165527,
      "step": 70097
    },
    {
      "epoch": 6.68497085571289e-05,
      "step": 70097,
      "training_step_time": 8.538762092590332
    },
    {
      "epoch": 6.685066223144531e-05,
      "model_forward_time": 1.7501513957977295,
      "step": 70098
    },
    {
      "epoch": 6.685066223144531e-05,
      "step": 70098,
      "training_step_time": 9.808116912841797
    },
    {
      "epoch": 6.685161590576172e-05,
      "model_forward_time": 1.7219555377960205,
      "step": 70099
    },
    {
      "epoch": 6.685161590576172e-05,
      "step": 70099,
      "training_step_time": 9.509535551071167
    },
    {
      "epoch": 6.685256958007812e-05,
      "grad_norm": 0.0951620489358902,
      "learning_rate": 2.251570017548976e-06,
      "loss": 0.0602,
      "step": 70100
    },
    {
      "epoch": 6.685256958007812e-05,
      "model_forward_time": 2.2877297401428223,
      "step": 70100
    },
    {
      "dynamics_loss_avg": 0.06446077767759562,
      "epoch": 6.685256958007812e-05,
      "step": 70100
    },
    {
      "action_loss_avg": 0.0010358326137065888,
      "epoch": 6.685256958007812e-05,
      "step": 70100
    },
    {
      "epoch": 6.685256958007812e-05,
      "step": 70100,
      "training_step_time": 7.667389631271362
    },
    {
      "epoch": 6.685352325439453e-05,
      "model_forward_time": 1.7440745830535889,
      "step": 70101
    },
    {
      "epoch": 6.685352325439453e-05,
      "step": 70101,
      "training_step_time": 7.050702810287476
    },
    {
      "epoch": 6.685447692871094e-05,
      "model_forward_time": 1.7346012592315674,
      "step": 70102
    },
    {
      "epoch": 6.685447692871094e-05,
      "step": 70102,
      "training_step_time": 6.751705646514893
    },
    {
      "epoch": 6.685543060302734e-05,
      "model_forward_time": 1.7303168773651123,
      "step": 70103
    },
    {
      "epoch": 6.685543060302734e-05,
      "step": 70103,
      "training_step_time": 7.562910556793213
    },
    {
      "epoch": 6.685638427734375e-05,
      "model_forward_time": 1.7286021709442139,
      "step": 70104
    },
    {
      "epoch": 6.685638427734375e-05,
      "step": 70104,
      "training_step_time": 8.62930703163147
    },
    {
      "epoch": 6.685733795166016e-05,
      "model_forward_time": 1.738778829574585,
      "step": 70105
    },
    {
      "epoch": 6.685733795166016e-05,
      "step": 70105,
      "training_step_time": 7.67425799369812
    },
    {
      "epoch": 6.685829162597656e-05,
      "model_forward_time": 1.7325012683868408,
      "step": 70106
    },
    {
      "epoch": 6.685829162597656e-05,
      "step": 70106,
      "training_step_time": 8.440837621688843
    },
    {
      "epoch": 6.685924530029297e-05,
      "model_forward_time": 1.7690434455871582,
      "step": 70107
    },
    {
      "epoch": 6.685924530029297e-05,
      "step": 70107,
      "training_step_time": 7.778189420700073
    },
    {
      "epoch": 6.686019897460938e-05,
      "model_forward_time": 1.7296054363250732,
      "step": 70108
    },
    {
      "epoch": 6.686019897460938e-05,
      "step": 70108,
      "training_step_time": 8.865977048873901
    },
    {
      "epoch": 6.686115264892578e-05,
      "model_forward_time": 2.2867379188537598,
      "step": 70109
    },
    {
      "epoch": 6.686115264892578e-05,
      "step": 70109,
      "training_step_time": 7.745488405227661
    },
    {
      "epoch": 6.686210632324219e-05,
      "grad_norm": 0.1186094731092453,
      "learning_rate": 2.2501889065297288e-06,
      "loss": 0.0584,
      "step": 70110
    },
    {
      "epoch": 6.686210632324219e-05,
      "model_forward_time": 1.6981549263000488,
      "step": 70110
    },
    {
      "dynamics_loss_avg": 0.06041893772780895,
      "epoch": 6.686210632324219e-05,
      "step": 70110
    },
    {
      "action_loss_avg": 0.001209735096199438,
      "epoch": 6.686210632324219e-05,
      "step": 70110
    },
    {
      "epoch": 6.686210632324219e-05,
      "step": 70110,
      "training_step_time": 7.058775186538696
    },
    {
      "epoch": 6.68630599975586e-05,
      "model_forward_time": 1.707360029220581,
      "step": 70111
    },
    {
      "epoch": 6.68630599975586e-05,
      "step": 70111,
      "training_step_time": 6.749770164489746
    },
    {
      "epoch": 6.6864013671875e-05,
      "model_forward_time": 1.7030491828918457,
      "step": 70112
    },
    {
      "epoch": 6.6864013671875e-05,
      "step": 70112,
      "training_step_time": 7.587818622589111
    },
    {
      "epoch": 6.68649673461914e-05,
      "model_forward_time": 1.7014005184173584,
      "step": 70113
    },
    {
      "epoch": 6.68649673461914e-05,
      "step": 70113,
      "training_step_time": 7.599327802658081
    },
    {
      "epoch": 6.686592102050782e-05,
      "model_forward_time": 1.7098057270050049,
      "step": 70114
    },
    {
      "epoch": 6.686592102050782e-05,
      "step": 70114,
      "training_step_time": 8.813517808914185
    },
    {
      "epoch": 6.686687469482421e-05,
      "model_forward_time": 1.7046833038330078,
      "step": 70115
    },
    {
      "epoch": 6.686687469482421e-05,
      "step": 70115,
      "training_step_time": 10.347167015075684
    },
    {
      "epoch": 6.686782836914062e-05,
      "model_forward_time": 1.713599443435669,
      "step": 70116
    },
    {
      "epoch": 6.686782836914062e-05,
      "step": 70116,
      "training_step_time": 8.799548387527466
    },
    {
      "epoch": 6.686878204345704e-05,
      "model_forward_time": 1.7062766551971436,
      "step": 70117
    },
    {
      "epoch": 6.686878204345704e-05,
      "step": 70117,
      "training_step_time": 7.784516334533691
    },
    {
      "epoch": 6.686973571777343e-05,
      "model_forward_time": 2.2588348388671875,
      "step": 70118
    },
    {
      "epoch": 6.686973571777343e-05,
      "step": 70118,
      "training_step_time": 8.528009414672852
    },
    {
      "epoch": 6.687068939208984e-05,
      "model_forward_time": 1.7050151824951172,
      "step": 70119
    },
    {
      "epoch": 6.687068939208984e-05,
      "step": 70119,
      "training_step_time": 7.077979326248169
    },
    {
      "epoch": 6.687164306640625e-05,
      "grad_norm": 0.08091558516025543,
      "learning_rate": 2.2488080962256887e-06,
      "loss": 0.0565,
      "step": 70120
    },
    {
      "epoch": 6.687164306640625e-05,
      "model_forward_time": 1.7180778980255127,
      "step": 70120
    },
    {
      "dynamics_loss_avg": 0.05412061037495732,
      "epoch": 6.687164306640625e-05,
      "step": 70120
    },
    {
      "action_loss_avg": 0.0013464658753946424,
      "epoch": 6.687164306640625e-05,
      "step": 70120
    },
    {
      "epoch": 6.687164306640625e-05,
      "step": 70120,
      "training_step_time": 6.733892440795898
    },
    {
      "epoch": 6.687259674072265e-05,
      "model_forward_time": 1.7271649837493896,
      "step": 70121
    },
    {
      "epoch": 6.687259674072265e-05,
      "step": 70121,
      "training_step_time": 7.486402273178101
    },
    {
      "epoch": 6.687355041503906e-05,
      "model_forward_time": 1.7192418575286865,
      "step": 70122
    },
    {
      "epoch": 6.687355041503906e-05,
      "step": 70122,
      "training_step_time": 7.616182804107666
    },
    {
      "epoch": 6.687450408935547e-05,
      "model_forward_time": 1.7218880653381348,
      "step": 70123
    },
    {
      "epoch": 6.687450408935547e-05,
      "step": 70123,
      "training_step_time": 7.719006776809692
    },
    {
      "epoch": 6.687545776367187e-05,
      "model_forward_time": 1.731400966644287,
      "step": 70124
    },
    {
      "epoch": 6.687545776367187e-05,
      "step": 70124,
      "training_step_time": 8.354207038879395
    },
    {
      "epoch": 6.687641143798828e-05,
      "model_forward_time": 1.6953485012054443,
      "step": 70125
    },
    {
      "epoch": 6.687641143798828e-05,
      "step": 70125,
      "training_step_time": 9.071353435516357
    },
    {
      "epoch": 6.687736511230469e-05,
      "model_forward_time": 1.717578649520874,
      "step": 70126
    },
    {
      "epoch": 6.687736511230469e-05,
      "step": 70126,
      "training_step_time": 9.493337154388428
    },
    {
      "epoch": 6.687831878662109e-05,
      "model_forward_time": 1.7032299041748047,
      "step": 70127
    },
    {
      "epoch": 6.687831878662109e-05,
      "step": 70127,
      "training_step_time": 8.701459169387817
    },
    {
      "epoch": 6.68792724609375e-05,
      "model_forward_time": 2.262478828430176,
      "step": 70128
    },
    {
      "epoch": 6.68792724609375e-05,
      "step": 70128,
      "training_step_time": 7.645392656326294
    },
    {
      "epoch": 6.688022613525391e-05,
      "model_forward_time": 1.6945233345031738,
      "step": 70129
    },
    {
      "epoch": 6.688022613525391e-05,
      "step": 70129,
      "training_step_time": 6.764445781707764
    },
    {
      "epoch": 6.688117980957031e-05,
      "grad_norm": 0.08649780601263046,
      "learning_rate": 2.2474275867878586e-06,
      "loss": 0.057,
      "step": 70130
    },
    {
      "epoch": 6.688117980957031e-05,
      "model_forward_time": 1.6970999240875244,
      "step": 70130
    },
    {
      "dynamics_loss_avg": 0.061761272884905336,
      "epoch": 6.688117980957031e-05,
      "step": 70130
    },
    {
      "action_loss_avg": 0.0013391359476372599,
      "epoch": 6.688117980957031e-05,
      "step": 70130
    },
    {
      "epoch": 6.688117980957031e-05,
      "step": 70130,
      "training_step_time": 7.192417860031128
    },
    {
      "epoch": 6.688213348388672e-05,
      "model_forward_time": 1.3257629871368408,
      "step": 70131
    },
    {
      "epoch": 6.688213348388672e-05,
      "step": 70131,
      "training_step_time": 7.596505165100098
    },
    {
      "epoch": 6.688308715820313e-05,
      "model_forward_time": 1.7075955867767334,
      "step": 70132
    },
    {
      "epoch": 6.688308715820313e-05,
      "step": 70132,
      "training_step_time": 7.653157711029053
    },
    {
      "epoch": 6.688404083251953e-05,
      "model_forward_time": 1.7058689594268799,
      "step": 70133
    },
    {
      "epoch": 6.688404083251953e-05,
      "step": 70133,
      "training_step_time": 8.267908096313477
    },
    {
      "epoch": 6.688499450683594e-05,
      "model_forward_time": 1.7022149562835693,
      "step": 70134
    },
    {
      "epoch": 6.688499450683594e-05,
      "step": 70134,
      "training_step_time": 7.685072183609009
    },
    {
      "epoch": 6.688594818115235e-05,
      "model_forward_time": 1.8075470924377441,
      "step": 70135
    },
    {
      "epoch": 6.688594818115235e-05,
      "step": 70135,
      "training_step_time": 8.46656322479248
    },
    {
      "epoch": 6.688690185546875e-05,
      "model_forward_time": 1.704160451889038,
      "step": 70136
    },
    {
      "epoch": 6.688690185546875e-05,
      "step": 70136,
      "training_step_time": 8.47271990776062
    },
    {
      "epoch": 6.688785552978516e-05,
      "model_forward_time": 2.2650082111358643,
      "step": 70137
    },
    {
      "epoch": 6.688785552978516e-05,
      "step": 70137,
      "training_step_time": 8.67913031578064
    },
    {
      "epoch": 6.688880920410157e-05,
      "model_forward_time": 1.701854944229126,
      "step": 70138
    },
    {
      "epoch": 6.688880920410157e-05,
      "step": 70138,
      "training_step_time": 6.801507949829102
    },
    {
      "epoch": 6.688976287841796e-05,
      "model_forward_time": 1.706559658050537,
      "step": 70139
    },
    {
      "epoch": 6.688976287841796e-05,
      "step": 70139,
      "training_step_time": 7.216577768325806
    },
    {
      "epoch": 6.689071655273438e-05,
      "grad_norm": 0.12498513609170914,
      "learning_rate": 2.246047378367211e-06,
      "loss": 0.0536,
      "step": 70140
    },
    {
      "epoch": 6.689071655273438e-05,
      "model_forward_time": 1.7124409675598145,
      "step": 70140
    },
    {
      "dynamics_loss_avg": 0.053662358596920964,
      "epoch": 6.689071655273438e-05,
      "step": 70140
    },
    {
      "action_loss_avg": 0.0017371206544339658,
      "epoch": 6.689071655273438e-05,
      "step": 70140
    },
    {
      "epoch": 6.689071655273438e-05,
      "step": 70140,
      "training_step_time": 7.598780155181885
    },
    {
      "epoch": 6.689167022705079e-05,
      "model_forward_time": 1.7038929462432861,
      "step": 70141
    },
    {
      "epoch": 6.689167022705079e-05,
      "step": 70141,
      "training_step_time": 8.60204005241394
    },
    {
      "epoch": 6.689262390136718e-05,
      "model_forward_time": 1.7055599689483643,
      "step": 70142
    },
    {
      "epoch": 6.689262390136718e-05,
      "step": 70142,
      "training_step_time": 7.725574970245361
    },
    {
      "epoch": 6.68935775756836e-05,
      "model_forward_time": 1.7107813358306885,
      "step": 70143
    },
    {
      "epoch": 6.68935775756836e-05,
      "step": 70143,
      "training_step_time": 8.433201789855957
    },
    {
      "epoch": 6.689453125e-05,
      "model_forward_time": 1.785968542098999,
      "step": 70144
    },
    {
      "epoch": 6.689453125e-05,
      "step": 70144,
      "training_step_time": 8.546802043914795
    },
    {
      "epoch": 6.68954849243164e-05,
      "model_forward_time": 1.719235897064209,
      "step": 70145
    },
    {
      "epoch": 6.68954849243164e-05,
      "step": 70145,
      "training_step_time": 10.191953420639038
    },
    {
      "epoch": 6.689643859863281e-05,
      "model_forward_time": 2.267005681991577,
      "step": 70146
    },
    {
      "epoch": 6.689643859863281e-05,
      "step": 70146,
      "training_step_time": 7.570843696594238
    },
    {
      "epoch": 6.689739227294922e-05,
      "model_forward_time": 1.7053823471069336,
      "step": 70147
    },
    {
      "epoch": 6.689739227294922e-05,
      "step": 70147,
      "training_step_time": 7.5872132778167725
    },
    {
      "epoch": 6.689834594726562e-05,
      "model_forward_time": 1.324925422668457,
      "step": 70148
    },
    {
      "epoch": 6.689834594726562e-05,
      "step": 70148,
      "training_step_time": 8.269728899002075
    },
    {
      "epoch": 6.689929962158203e-05,
      "model_forward_time": 1.6998262405395508,
      "step": 70149
    },
    {
      "epoch": 6.689929962158203e-05,
      "step": 70149,
      "training_step_time": 7.5541558265686035
    },
    {
      "epoch": 6.690025329589844e-05,
      "grad_norm": 0.11222490668296814,
      "learning_rate": 2.24466747111468e-06,
      "loss": 0.0558,
      "step": 70150
    },
    {
      "epoch": 6.690025329589844e-05,
      "model_forward_time": 1.699953317642212,
      "step": 70150
    },
    {
      "dynamics_loss_avg": 0.05192852504551411,
      "epoch": 6.690025329589844e-05,
      "step": 70150
    },
    {
      "action_loss_avg": 0.001216934181866236,
      "epoch": 6.690025329589844e-05,
      "step": 70150
    },
    {
      "epoch": 6.690025329589844e-05,
      "step": 70150,
      "training_step_time": 8.706447124481201
    },
    {
      "epoch": 6.690120697021484e-05,
      "model_forward_time": 1.7058377265930176,
      "step": 70151
    },
    {
      "epoch": 6.690120697021484e-05,
      "step": 70151,
      "training_step_time": 7.696306943893433
    },
    {
      "epoch": 6.690216064453125e-05,
      "model_forward_time": 1.7322425842285156,
      "step": 70152
    },
    {
      "epoch": 6.690216064453125e-05,
      "step": 70152,
      "training_step_time": 8.54966926574707
    },
    {
      "epoch": 6.690311431884766e-05,
      "model_forward_time": 1.7894597053527832,
      "step": 70153
    },
    {
      "epoch": 6.690311431884766e-05,
      "step": 70153,
      "training_step_time": 8.94824481010437
    },
    {
      "epoch": 6.690406799316406e-05,
      "model_forward_time": 1.7322256565093994,
      "step": 70154
    },
    {
      "epoch": 6.690406799316406e-05,
      "step": 70154,
      "training_step_time": 8.647763013839722
    },
    {
      "epoch": 6.690502166748047e-05,
      "model_forward_time": 2.272493600845337,
      "step": 70155
    },
    {
      "epoch": 6.690502166748047e-05,
      "step": 70155,
      "training_step_time": 7.704013824462891
    },
    {
      "epoch": 6.690597534179688e-05,
      "model_forward_time": 1.7692921161651611,
      "step": 70156
    },
    {
      "epoch": 6.690597534179688e-05,
      "step": 70156,
      "training_step_time": 7.621836423873901
    },
    {
      "epoch": 6.690692901611328e-05,
      "model_forward_time": 1.7183794975280762,
      "step": 70157
    },
    {
      "epoch": 6.690692901611328e-05,
      "step": 70157,
      "training_step_time": 7.26175594329834
    },
    {
      "epoch": 6.690788269042969e-05,
      "model_forward_time": 1.717649221420288,
      "step": 70158
    },
    {
      "epoch": 6.690788269042969e-05,
      "step": 70158,
      "training_step_time": 7.5873777866363525
    },
    {
      "epoch": 6.69088363647461e-05,
      "model_forward_time": 1.723768949508667,
      "step": 70159
    },
    {
      "epoch": 6.69088363647461e-05,
      "step": 70159,
      "training_step_time": 7.664707183837891
    },
    {
      "epoch": 6.69097900390625e-05,
      "grad_norm": 0.10002865642309189,
      "learning_rate": 2.2432878651811737e-06,
      "loss": 0.0518,
      "step": 70160
    },
    {
      "epoch": 6.69097900390625e-05,
      "model_forward_time": 1.7222776412963867,
      "step": 70160
    },
    {
      "dynamics_loss_avg": 0.05799141116440296,
      "epoch": 6.69097900390625e-05,
      "step": 70160
    },
    {
      "action_loss_avg": 0.002522444189526141,
      "epoch": 6.69097900390625e-05,
      "step": 70160
    },
    {
      "epoch": 6.69097900390625e-05,
      "step": 70160,
      "training_step_time": 7.737859487533569
    },
    {
      "epoch": 6.691074371337891e-05,
      "model_forward_time": 1.718904972076416,
      "step": 70161
    },
    {
      "epoch": 6.691074371337891e-05,
      "step": 70161,
      "training_step_time": 7.710235357284546
    },
    {
      "epoch": 6.691169738769532e-05,
      "model_forward_time": 1.737290620803833,
      "step": 70162
    },
    {
      "epoch": 6.691169738769532e-05,
      "step": 70162,
      "training_step_time": 10.172789335250854
    },
    {
      "epoch": 6.691265106201172e-05,
      "model_forward_time": 1.3395366668701172,
      "step": 70163
    },
    {
      "epoch": 6.691265106201172e-05,
      "step": 70163,
      "training_step_time": 7.720953941345215
    },
    {
      "epoch": 6.691360473632813e-05,
      "model_forward_time": 1.7220640182495117,
      "step": 70164
    },
    {
      "epoch": 6.691360473632813e-05,
      "step": 70164,
      "training_step_time": 7.705158472061157
    },
    {
      "epoch": 6.691455841064454e-05,
      "model_forward_time": 2.297577142715454,
      "step": 70165
    },
    {
      "epoch": 6.691455841064454e-05,
      "step": 70165,
      "training_step_time": 7.625912427902222
    },
    {
      "epoch": 6.691551208496093e-05,
      "model_forward_time": 1.7216722965240479,
      "step": 70166
    },
    {
      "epoch": 6.691551208496093e-05,
      "step": 70166,
      "training_step_time": 7.209070920944214
    },
    {
      "epoch": 6.691646575927735e-05,
      "model_forward_time": 1.6901650428771973,
      "step": 70167
    },
    {
      "epoch": 6.691646575927735e-05,
      "step": 70167,
      "training_step_time": 7.527062892913818
    },
    {
      "epoch": 6.691741943359376e-05,
      "model_forward_time": 1.718912124633789,
      "step": 70168
    },
    {
      "epoch": 6.691741943359376e-05,
      "step": 70168,
      "training_step_time": 7.632438898086548
    },
    {
      "epoch": 6.691837310791015e-05,
      "model_forward_time": 1.7114672660827637,
      "step": 70169
    },
    {
      "epoch": 6.691837310791015e-05,
      "step": 70169,
      "training_step_time": 9.049381017684937
    },
    {
      "epoch": 6.691932678222656e-05,
      "grad_norm": 0.13823384046554565,
      "learning_rate": 2.2419085607175595e-06,
      "loss": 0.0579,
      "step": 70170
    },
    {
      "epoch": 6.691932678222656e-05,
      "model_forward_time": 1.7036042213439941,
      "step": 70170
    },
    {
      "dynamics_loss_avg": 0.07272167466580867,
      "epoch": 6.691932678222656e-05,
      "step": 70170
    },
    {
      "action_loss_avg": 0.0015827610943233595,
      "epoch": 6.691932678222656e-05,
      "step": 70170
    },
    {
      "epoch": 6.691932678222656e-05,
      "step": 70170,
      "training_step_time": 7.634195566177368
    },
    {
      "epoch": 6.692028045654297e-05,
      "model_forward_time": 1.7124226093292236,
      "step": 70171
    },
    {
      "epoch": 6.692028045654297e-05,
      "step": 70171,
      "training_step_time": 9.460275888442993
    },
    {
      "epoch": 6.692123413085937e-05,
      "model_forward_time": 1.684844732284546,
      "step": 70172
    },
    {
      "epoch": 6.692123413085937e-05,
      "step": 70172,
      "training_step_time": 7.6922523975372314
    },
    {
      "epoch": 6.692218780517578e-05,
      "model_forward_time": 1.694228172302246,
      "step": 70173
    },
    {
      "epoch": 6.692218780517578e-05,
      "step": 70173,
      "training_step_time": 7.688174247741699
    },
    {
      "epoch": 6.69231414794922e-05,
      "model_forward_time": 2.2342729568481445,
      "step": 70174
    },
    {
      "epoch": 6.69231414794922e-05,
      "step": 70174,
      "training_step_time": 7.6159515380859375
    },
    {
      "epoch": 6.692409515380859e-05,
      "model_forward_time": 1.6850454807281494,
      "step": 70175
    },
    {
      "epoch": 6.692409515380859e-05,
      "step": 70175,
      "training_step_time": 7.622615575790405
    },
    {
      "epoch": 6.6925048828125e-05,
      "model_forward_time": 1.6901750564575195,
      "step": 70176
    },
    {
      "epoch": 6.6925048828125e-05,
      "step": 70176,
      "training_step_time": 7.105468034744263
    },
    {
      "epoch": 6.692600250244141e-05,
      "model_forward_time": 1.6875131130218506,
      "step": 70177
    },
    {
      "epoch": 6.692600250244141e-05,
      "step": 70177,
      "training_step_time": 8.60205340385437
    },
    {
      "epoch": 6.692695617675781e-05,
      "model_forward_time": 1.6885662078857422,
      "step": 70178
    },
    {
      "epoch": 6.692695617675781e-05,
      "step": 70178,
      "training_step_time": 7.6291468143463135
    },
    {
      "epoch": 6.692790985107422e-05,
      "model_forward_time": 1.6865530014038086,
      "step": 70179
    },
    {
      "epoch": 6.692790985107422e-05,
      "step": 70179,
      "training_step_time": 8.715999364852905
    },
    {
      "epoch": 6.692886352539063e-05,
      "grad_norm": 0.09882054477930069,
      "learning_rate": 2.2405295578746783e-06,
      "loss": 0.0537,
      "step": 70180
    },
    {
      "epoch": 6.692886352539063e-05,
      "model_forward_time": 1.688662052154541,
      "step": 70180
    },
    {
      "dynamics_loss_avg": 0.039726178348064425,
      "epoch": 6.692886352539063e-05,
      "step": 70180
    },
    {
      "action_loss_avg": 0.0010101379419211298,
      "epoch": 6.692886352539063e-05,
      "step": 70180
    },
    {
      "epoch": 6.692886352539063e-05,
      "step": 70180,
      "training_step_time": 7.84807562828064
    },
    {
      "epoch": 6.692981719970703e-05,
      "model_forward_time": 1.7195968627929688,
      "step": 70181
    },
    {
      "epoch": 6.692981719970703e-05,
      "step": 70181,
      "training_step_time": 8.821327209472656
    },
    {
      "epoch": 6.693077087402344e-05,
      "model_forward_time": 1.686424970626831,
      "step": 70182
    },
    {
      "epoch": 6.693077087402344e-05,
      "step": 70182,
      "training_step_time": 8.84061050415039
    },
    {
      "epoch": 6.693172454833985e-05,
      "model_forward_time": 2.2658979892730713,
      "step": 70183
    },
    {
      "epoch": 6.693172454833985e-05,
      "step": 70183,
      "training_step_time": 7.618748903274536
    },
    {
      "epoch": 6.693267822265625e-05,
      "model_forward_time": 1.681821346282959,
      "step": 70184
    },
    {
      "epoch": 6.693267822265625e-05,
      "step": 70184,
      "training_step_time": 6.791658878326416
    },
    {
      "epoch": 6.693363189697266e-05,
      "model_forward_time": 1.6910209655761719,
      "step": 70185
    },
    {
      "epoch": 6.693363189697266e-05,
      "step": 70185,
      "training_step_time": 7.229598522186279
    },
    {
      "epoch": 6.693458557128907e-05,
      "model_forward_time": 1.6906518936157227,
      "step": 70186
    },
    {
      "epoch": 6.693458557128907e-05,
      "step": 70186,
      "training_step_time": 7.5945539474487305
    },
    {
      "epoch": 6.693553924560547e-05,
      "model_forward_time": 1.6942429542541504,
      "step": 70187
    },
    {
      "epoch": 6.693553924560547e-05,
      "step": 70187,
      "training_step_time": 7.736500978469849
    },
    {
      "epoch": 6.693649291992188e-05,
      "model_forward_time": 1.6897306442260742,
      "step": 70188
    },
    {
      "epoch": 6.693649291992188e-05,
      "step": 70188,
      "training_step_time": 8.471397399902344
    },
    {
      "epoch": 6.693744659423827e-05,
      "model_forward_time": 1.309722900390625,
      "step": 70189
    },
    {
      "epoch": 6.693744659423827e-05,
      "step": 70189,
      "training_step_time": 7.67963719367981
    },
    {
      "epoch": 6.693840026855469e-05,
      "grad_norm": 0.09946108609437943,
      "learning_rate": 2.239150856803336e-06,
      "loss": 0.0581,
      "step": 70190
    },
    {
      "epoch": 6.693840026855469e-05,
      "model_forward_time": 1.7064235210418701,
      "step": 70190
    },
    {
      "dynamics_loss_avg": 0.045785889867693184,
      "epoch": 6.693840026855469e-05,
      "step": 70190
    },
    {
      "action_loss_avg": 0.0013897314900532364,
      "epoch": 6.693840026855469e-05,
      "step": 70190
    },
    {
      "epoch": 6.693840026855469e-05,
      "step": 70190,
      "training_step_time": 8.853529453277588
    },
    {
      "epoch": 6.69393539428711e-05,
      "model_forward_time": 1.708892583847046,
      "step": 70191
    },
    {
      "epoch": 6.69393539428711e-05,
      "step": 70191,
      "training_step_time": 8.36720609664917
    },
    {
      "epoch": 6.69403076171875e-05,
      "model_forward_time": 1.3308615684509277,
      "step": 70192
    },
    {
      "epoch": 6.69403076171875e-05,
      "step": 70192,
      "training_step_time": 8.810020685195923
    },
    {
      "epoch": 6.69412612915039e-05,
      "model_forward_time": 2.263118267059326,
      "step": 70193
    },
    {
      "epoch": 6.69412612915039e-05,
      "step": 70193,
      "training_step_time": 7.378893852233887
    },
    {
      "epoch": 6.694221496582031e-05,
      "model_forward_time": 1.7094717025756836,
      "step": 70194
    },
    {
      "epoch": 6.694221496582031e-05,
      "step": 70194,
      "training_step_time": 7.278948783874512
    },
    {
      "epoch": 6.694316864013671e-05,
      "model_forward_time": 1.71095609664917,
      "step": 70195
    },
    {
      "epoch": 6.694316864013671e-05,
      "step": 70195,
      "training_step_time": 7.450544595718384
    },
    {
      "epoch": 6.694412231445312e-05,
      "model_forward_time": 1.7060279846191406,
      "step": 70196
    },
    {
      "epoch": 6.694412231445312e-05,
      "step": 70196,
      "training_step_time": 8.297497987747192
    },
    {
      "epoch": 6.694507598876953e-05,
      "model_forward_time": 1.7077500820159912,
      "step": 70197
    },
    {
      "epoch": 6.694507598876953e-05,
      "step": 70197,
      "training_step_time": 9.2947678565979
    },
    {
      "epoch": 6.694602966308593e-05,
      "model_forward_time": 1.708322286605835,
      "step": 70198
    },
    {
      "epoch": 6.694602966308593e-05,
      "step": 70198,
      "training_step_time": 7.771601438522339
    },
    {
      "epoch": 6.694698333740234e-05,
      "model_forward_time": 1.716764211654663,
      "step": 70199
    },
    {
      "epoch": 6.694698333740234e-05,
      "step": 70199,
      "training_step_time": 7.810638666152954
    },
    {
      "epoch": 6.694793701171875e-05,
      "grad_norm": 0.10105080157518387,
      "learning_rate": 2.237772457654307e-06,
      "loss": 0.0574,
      "step": 70200
    },
    {
      "epoch": 6.694793701171875e-05,
      "model_forward_time": 1.6776747703552246,
      "step": 70200
    },
    {
      "dynamics_loss_avg": 0.05829950422048569,
      "epoch": 6.694793701171875e-05,
      "step": 70200
    },
    {
      "action_loss_avg": 0.0014452471601543947,
      "epoch": 6.694793701171875e-05,
      "step": 70200
    },
    {
      "epoch": 6.694793701171875e-05,
      "step": 70200,
      "training_step_time": 7.918395042419434
    },
    {
      "epoch": 6.694889068603515e-05,
      "model_forward_time": 1.6936805248260498,
      "step": 70201
    },
    {
      "epoch": 6.694889068603515e-05,
      "step": 70201,
      "training_step_time": 8.621625423431396
    },
    {
      "epoch": 6.694984436035156e-05,
      "model_forward_time": 2.261441707611084,
      "step": 70202
    },
    {
      "epoch": 6.694984436035156e-05,
      "step": 70202,
      "training_step_time": 7.369603633880615
    },
    {
      "epoch": 6.695079803466797e-05,
      "model_forward_time": 1.6966643333435059,
      "step": 70203
    },
    {
      "epoch": 6.695079803466797e-05,
      "step": 70203,
      "training_step_time": 7.313839673995972
    },
    {
      "epoch": 6.695175170898437e-05,
      "model_forward_time": 1.7033641338348389,
      "step": 70204
    },
    {
      "epoch": 6.695175170898437e-05,
      "step": 70204,
      "training_step_time": 7.475084066390991
    },
    {
      "epoch": 6.695270538330078e-05,
      "model_forward_time": 1.702075719833374,
      "step": 70205
    },
    {
      "epoch": 6.695270538330078e-05,
      "step": 70205,
      "training_step_time": 8.787160634994507
    },
    {
      "epoch": 6.695365905761719e-05,
      "model_forward_time": 1.6924426555633545,
      "step": 70206
    },
    {
      "epoch": 6.695365905761719e-05,
      "step": 70206,
      "training_step_time": 7.723490476608276
    },
    {
      "epoch": 6.695461273193359e-05,
      "model_forward_time": 1.6945886611938477,
      "step": 70207
    },
    {
      "epoch": 6.695461273193359e-05,
      "step": 70207,
      "training_step_time": 7.768857955932617
    },
    {
      "epoch": 6.695556640625e-05,
      "model_forward_time": 1.6953985691070557,
      "step": 70208
    },
    {
      "epoch": 6.695556640625e-05,
      "step": 70208,
      "training_step_time": 8.737098693847656
    },
    {
      "epoch": 6.695652008056641e-05,
      "model_forward_time": 1.7359368801116943,
      "step": 70209
    },
    {
      "epoch": 6.695652008056641e-05,
      "step": 70209,
      "training_step_time": 7.848212957382202
    },
    {
      "epoch": 6.69574737548828e-05,
      "grad_norm": 0.08968202769756317,
      "learning_rate": 2.236394360578327e-06,
      "loss": 0.0597,
      "step": 70210
    },
    {
      "epoch": 6.69574737548828e-05,
      "model_forward_time": 1.7237741947174072,
      "step": 70210
    },
    {
      "dynamics_loss_avg": 0.045371942967176435,
      "epoch": 6.69574737548828e-05,
      "step": 70210
    },
    {
      "action_loss_avg": 0.0012040756075293757,
      "epoch": 6.69574737548828e-05,
      "step": 70210
    },
    {
      "epoch": 6.69574737548828e-05,
      "step": 70210,
      "training_step_time": 7.71985387802124
    },
    {
      "epoch": 6.695842742919922e-05,
      "model_forward_time": 2.30712628364563,
      "step": 70211
    },
    {
      "epoch": 6.695842742919922e-05,
      "step": 70211,
      "training_step_time": 7.672758102416992
    },
    {
      "epoch": 6.695938110351563e-05,
      "model_forward_time": 1.7309093475341797,
      "step": 70212
    },
    {
      "epoch": 6.695938110351563e-05,
      "step": 70212,
      "training_step_time": 7.240693092346191
    },
    {
      "epoch": 6.696033477783203e-05,
      "model_forward_time": 1.7287275791168213,
      "step": 70213
    },
    {
      "epoch": 6.696033477783203e-05,
      "step": 70213,
      "training_step_time": 7.424909353256226
    },
    {
      "epoch": 6.696128845214844e-05,
      "model_forward_time": 1.7427778244018555,
      "step": 70214
    },
    {
      "epoch": 6.696128845214844e-05,
      "step": 70214,
      "training_step_time": 7.668670415878296
    },
    {
      "epoch": 6.696224212646485e-05,
      "model_forward_time": 1.7252094745635986,
      "step": 70215
    },
    {
      "epoch": 6.696224212646485e-05,
      "step": 70215,
      "training_step_time": 7.718353986740112
    },
    {
      "epoch": 6.696319580078124e-05,
      "model_forward_time": 1.7316946983337402,
      "step": 70216
    },
    {
      "epoch": 6.696319580078124e-05,
      "step": 70216,
      "training_step_time": 7.6441075801849365
    },
    {
      "epoch": 6.696414947509766e-05,
      "model_forward_time": 1.7363858222961426,
      "step": 70217
    },
    {
      "epoch": 6.696414947509766e-05,
      "step": 70217,
      "training_step_time": 9.321756362915039
    },
    {
      "epoch": 6.696510314941407e-05,
      "model_forward_time": 1.722243309020996,
      "step": 70218
    },
    {
      "epoch": 6.696510314941407e-05,
      "step": 70218,
      "training_step_time": 7.7243170738220215
    },
    {
      "epoch": 6.696605682373046e-05,
      "model_forward_time": 1.7434020042419434,
      "step": 70219
    },
    {
      "epoch": 6.696605682373046e-05,
      "step": 70219,
      "training_step_time": 8.812839031219482
    },
    {
      "epoch": 6.696701049804687e-05,
      "grad_norm": 0.13057737052440643,
      "learning_rate": 2.2350165657261076e-06,
      "loss": 0.0603,
      "step": 70220
    },
    {
      "epoch": 6.696701049804687e-05,
      "model_forward_time": 1.8969628810882568,
      "step": 70220
    },
    {
      "dynamics_loss_avg": 0.05236535286530852,
      "epoch": 6.696701049804687e-05,
      "step": 70220
    },
    {
      "action_loss_avg": 0.0014723083993885666,
      "epoch": 6.696701049804687e-05,
      "step": 70220
    },
    {
      "epoch": 6.696701049804687e-05,
      "step": 70220,
      "training_step_time": 7.6944427490234375
    },
    {
      "epoch": 6.696796417236328e-05,
      "model_forward_time": 1.7287750244140625,
      "step": 70221
    },
    {
      "epoch": 6.696796417236328e-05,
      "step": 70221,
      "training_step_time": 7.2747485637664795
    },
    {
      "epoch": 6.696891784667968e-05,
      "model_forward_time": 1.7367117404937744,
      "step": 70222
    },
    {
      "epoch": 6.696891784667968e-05,
      "step": 70222,
      "training_step_time": 7.427227258682251
    },
    {
      "epoch": 6.696987152099609e-05,
      "model_forward_time": 1.7288546562194824,
      "step": 70223
    },
    {
      "epoch": 6.696987152099609e-05,
      "step": 70223,
      "training_step_time": 7.570414066314697
    },
    {
      "epoch": 6.69708251953125e-05,
      "model_forward_time": 1.7303600311279297,
      "step": 70224
    },
    {
      "epoch": 6.69708251953125e-05,
      "step": 70224,
      "training_step_time": 7.707195043563843
    },
    {
      "epoch": 6.69717788696289e-05,
      "model_forward_time": 1.7335865497589111,
      "step": 70225
    },
    {
      "epoch": 6.69717788696289e-05,
      "step": 70225,
      "training_step_time": 8.836646318435669
    },
    {
      "epoch": 6.697273254394531e-05,
      "model_forward_time": 1.726851463317871,
      "step": 70226
    },
    {
      "epoch": 6.697273254394531e-05,
      "step": 70226,
      "training_step_time": 8.47166657447815
    },
    {
      "epoch": 6.697368621826172e-05,
      "model_forward_time": 1.8066234588623047,
      "step": 70227
    },
    {
      "epoch": 6.697368621826172e-05,
      "step": 70227,
      "training_step_time": 9.493338584899902
    },
    {
      "epoch": 6.697463989257812e-05,
      "model_forward_time": 1.7222764492034912,
      "step": 70228
    },
    {
      "epoch": 6.697463989257812e-05,
      "step": 70228,
      "training_step_time": 7.747591972351074
    },
    {
      "epoch": 6.697559356689453e-05,
      "model_forward_time": 1.7387194633483887,
      "step": 70229
    },
    {
      "epoch": 6.697559356689453e-05,
      "step": 70229,
      "training_step_time": 8.260420083999634
    },
    {
      "epoch": 6.697654724121094e-05,
      "grad_norm": 0.08195972442626953,
      "learning_rate": 2.2336390732483165e-06,
      "loss": 0.0595,
      "step": 70230
    },
    {
      "epoch": 6.697654724121094e-05,
      "model_forward_time": 1.720184564590454,
      "step": 70230
    },
    {
      "dynamics_loss_avg": 0.05850055869668722,
      "epoch": 6.697654724121094e-05,
      "step": 70230
    },
    {
      "action_loss_avg": 0.001155311887850985,
      "epoch": 6.697654724121094e-05,
      "step": 70230
    },
    {
      "epoch": 6.697654724121094e-05,
      "step": 70230,
      "training_step_time": 7.275052547454834
    },
    {
      "epoch": 6.697750091552734e-05,
      "model_forward_time": 1.7290489673614502,
      "step": 70231
    },
    {
      "epoch": 6.697750091552734e-05,
      "step": 70231,
      "training_step_time": 7.496648550033569
    },
    {
      "epoch": 6.697845458984375e-05,
      "model_forward_time": 1.7348463535308838,
      "step": 70232
    },
    {
      "epoch": 6.697845458984375e-05,
      "step": 70232,
      "training_step_time": 8.453685522079468
    },
    {
      "epoch": 6.697940826416016e-05,
      "model_forward_time": 1.6948270797729492,
      "step": 70233
    },
    {
      "epoch": 6.697940826416016e-05,
      "step": 70233,
      "training_step_time": 9.755795240402222
    },
    {
      "epoch": 6.698036193847656e-05,
      "model_forward_time": 1.7191462516784668,
      "step": 70234
    },
    {
      "epoch": 6.698036193847656e-05,
      "step": 70234,
      "training_step_time": 7.772130250930786
    },
    {
      "epoch": 6.698131561279297e-05,
      "model_forward_time": 1.704819917678833,
      "step": 70235
    },
    {
      "epoch": 6.698131561279297e-05,
      "step": 70235,
      "training_step_time": 9.187347173690796
    },
    {
      "epoch": 6.698226928710938e-05,
      "model_forward_time": 1.7599599361419678,
      "step": 70236
    },
    {
      "epoch": 6.698226928710938e-05,
      "step": 70236,
      "training_step_time": 8.92636251449585
    },
    {
      "epoch": 6.698322296142578e-05,
      "model_forward_time": 1.7137999534606934,
      "step": 70237
    },
    {
      "epoch": 6.698322296142578e-05,
      "step": 70237,
      "training_step_time": 7.833930730819702
    },
    {
      "epoch": 6.698417663574219e-05,
      "model_forward_time": 1.7318909168243408,
      "step": 70238
    },
    {
      "epoch": 6.698417663574219e-05,
      "step": 70238,
      "training_step_time": 7.740045070648193
    },
    {
      "epoch": 6.69851303100586e-05,
      "model_forward_time": 2.308424711227417,
      "step": 70239
    },
    {
      "epoch": 6.69851303100586e-05,
      "step": 70239,
      "training_step_time": 7.38283371925354
    },
    {
      "epoch": 6.6986083984375e-05,
      "grad_norm": 0.15236468613147736,
      "learning_rate": 2.2322618832955977e-06,
      "loss": 0.0572,
      "step": 70240
    },
    {
      "epoch": 6.6986083984375e-05,
      "model_forward_time": 1.7180559635162354,
      "step": 70240
    },
    {
      "dynamics_loss_avg": 0.06185142770409584,
      "epoch": 6.6986083984375e-05,
      "step": 70240
    },
    {
      "action_loss_avg": 0.0015893591393250973,
      "epoch": 6.6986083984375e-05,
      "step": 70240
    },
    {
      "epoch": 6.6986083984375e-05,
      "step": 70240,
      "training_step_time": 7.460686445236206
    },
    {
      "epoch": 6.69870376586914e-05,
      "model_forward_time": 1.727189302444458,
      "step": 70241
    },
    {
      "epoch": 6.69870376586914e-05,
      "step": 70241,
      "training_step_time": 7.4188573360443115
    },
    {
      "epoch": 6.698799133300782e-05,
      "model_forward_time": 1.739194393157959,
      "step": 70242
    },
    {
      "epoch": 6.698799133300782e-05,
      "step": 70242,
      "training_step_time": 7.786971092224121
    },
    {
      "epoch": 6.698894500732421e-05,
      "model_forward_time": 1.3429758548736572,
      "step": 70243
    },
    {
      "epoch": 6.698894500732421e-05,
      "step": 70243,
      "training_step_time": 7.737916707992554
    },
    {
      "epoch": 6.698989868164062e-05,
      "model_forward_time": 1.72471284866333,
      "step": 70244
    },
    {
      "epoch": 6.698989868164062e-05,
      "step": 70244,
      "training_step_time": 7.726359844207764
    },
    {
      "epoch": 6.699085235595704e-05,
      "model_forward_time": 1.759403944015503,
      "step": 70245
    },
    {
      "epoch": 6.699085235595704e-05,
      "step": 70245,
      "training_step_time": 10.463069438934326
    },
    {
      "epoch": 6.699180603027343e-05,
      "model_forward_time": 1.7319848537445068,
      "step": 70246
    },
    {
      "epoch": 6.699180603027343e-05,
      "step": 70246,
      "training_step_time": 8.871668100357056
    },
    {
      "epoch": 6.699275970458984e-05,
      "model_forward_time": 1.7357094287872314,
      "step": 70247
    },
    {
      "epoch": 6.699275970458984e-05,
      "step": 70247,
      "training_step_time": 7.758631229400635
    },
    {
      "epoch": 6.699371337890625e-05,
      "model_forward_time": 2.3046014308929443,
      "step": 70248
    },
    {
      "epoch": 6.699371337890625e-05,
      "step": 70248,
      "training_step_time": 8.726420640945435
    },
    {
      "epoch": 6.699466705322265e-05,
      "model_forward_time": 1.706542730331421,
      "step": 70249
    },
    {
      "epoch": 6.699466705322265e-05,
      "step": 70249,
      "training_step_time": 7.3280980587005615
    },
    {
      "epoch": 6.699562072753906e-05,
      "grad_norm": 0.1142408475279808,
      "learning_rate": 2.230884996018558e-06,
      "loss": 0.0566,
      "step": 70250
    },
    {
      "epoch": 6.699562072753906e-05,
      "model_forward_time": 1.3392560482025146,
      "step": 70250
    },
    {
      "dynamics_loss_avg": 0.047134158201515675,
      "epoch": 6.699562072753906e-05,
      "step": 70250
    },
    {
      "action_loss_avg": 0.0019337486286531203,
      "epoch": 6.699562072753906e-05,
      "step": 70250
    },
    {
      "epoch": 6.699562072753906e-05,
      "step": 70250,
      "training_step_time": 7.45904278755188
    },
    {
      "epoch": 6.699657440185547e-05,
      "model_forward_time": 1.7303919792175293,
      "step": 70251
    },
    {
      "epoch": 6.699657440185547e-05,
      "step": 70251,
      "training_step_time": 7.670567274093628
    },
    {
      "epoch": 6.699752807617187e-05,
      "model_forward_time": 1.739377498626709,
      "step": 70252
    },
    {
      "epoch": 6.699752807617187e-05,
      "step": 70252,
      "training_step_time": 7.687732696533203
    },
    {
      "epoch": 6.699848175048828e-05,
      "model_forward_time": 1.7379162311553955,
      "step": 70253
    },
    {
      "epoch": 6.699848175048828e-05,
      "step": 70253,
      "training_step_time": 8.268877983093262
    },
    {
      "epoch": 6.699943542480469e-05,
      "model_forward_time": 1.726874589920044,
      "step": 70254
    },
    {
      "epoch": 6.699943542480469e-05,
      "step": 70254,
      "training_step_time": 8.5552339553833
    },
    {
      "epoch": 6.700038909912109e-05,
      "model_forward_time": 1.8186752796173096,
      "step": 70255
    },
    {
      "epoch": 6.700038909912109e-05,
      "step": 70255,
      "training_step_time": 7.694184064865112
    },
    {
      "epoch": 6.70013427734375e-05,
      "model_forward_time": 1.7335007190704346,
      "step": 70256
    },
    {
      "epoch": 6.70013427734375e-05,
      "step": 70256,
      "training_step_time": 7.724230051040649
    },
    {
      "epoch": 6.700229644775391e-05,
      "model_forward_time": 2.313162088394165,
      "step": 70257
    },
    {
      "epoch": 6.700229644775391e-05,
      "step": 70257,
      "training_step_time": 7.687278985977173
    },
    {
      "epoch": 6.700325012207031e-05,
      "model_forward_time": 1.7294442653656006,
      "step": 70258
    },
    {
      "epoch": 6.700325012207031e-05,
      "step": 70258,
      "training_step_time": 7.600241184234619
    },
    {
      "epoch": 6.700420379638672e-05,
      "model_forward_time": 1.7274529933929443,
      "step": 70259
    },
    {
      "epoch": 6.700420379638672e-05,
      "step": 70259,
      "training_step_time": 7.436669588088989
    },
    {
      "epoch": 6.700515747070313e-05,
      "grad_norm": 0.17664223909378052,
      "learning_rate": 2.229508411567774e-06,
      "loss": 0.0552,
      "step": 70260
    },
    {
      "epoch": 6.700515747070313e-05,
      "model_forward_time": 1.7273039817810059,
      "step": 70260
    },
    {
      "dynamics_loss_avg": 0.0545509722083807,
      "epoch": 6.700515747070313e-05,
      "step": 70260
    },
    {
      "action_loss_avg": 0.0011602869111811743,
      "epoch": 6.700515747070313e-05,
      "step": 70260
    },
    {
      "epoch": 6.700515747070313e-05,
      "step": 70260,
      "training_step_time": 8.757254600524902
    },
    {
      "epoch": 6.700611114501953e-05,
      "model_forward_time": 1.7308027744293213,
      "step": 70261
    },
    {
      "epoch": 6.700611114501953e-05,
      "step": 70261,
      "training_step_time": 8.231173038482666
    },
    {
      "epoch": 6.700706481933594e-05,
      "model_forward_time": 1.7087979316711426,
      "step": 70262
    },
    {
      "epoch": 6.700706481933594e-05,
      "step": 70262,
      "training_step_time": 8.673624753952026
    },
    {
      "epoch": 6.700801849365235e-05,
      "model_forward_time": 1.7345600128173828,
      "step": 70263
    },
    {
      "epoch": 6.700801849365235e-05,
      "step": 70263,
      "training_step_time": 8.954411506652832
    },
    {
      "epoch": 6.700897216796875e-05,
      "model_forward_time": 1.8069803714752197,
      "step": 70264
    },
    {
      "epoch": 6.700897216796875e-05,
      "step": 70264,
      "training_step_time": 10.08296823501587
    },
    {
      "epoch": 6.700992584228516e-05,
      "model_forward_time": 1.7281510829925537,
      "step": 70265
    },
    {
      "epoch": 6.700992584228516e-05,
      "step": 70265,
      "training_step_time": 7.878816366195679
    },
    {
      "epoch": 6.701087951660157e-05,
      "model_forward_time": 1.7327911853790283,
      "step": 70266
    },
    {
      "epoch": 6.701087951660157e-05,
      "step": 70266,
      "training_step_time": 7.800157070159912
    },
    {
      "epoch": 6.701183319091796e-05,
      "model_forward_time": 2.2947375774383545,
      "step": 70267
    },
    {
      "epoch": 6.701183319091796e-05,
      "step": 70267,
      "training_step_time": 7.665151596069336
    },
    {
      "epoch": 6.701278686523438e-05,
      "model_forward_time": 1.732551097869873,
      "step": 70268
    },
    {
      "epoch": 6.701278686523438e-05,
      "step": 70268,
      "training_step_time": 7.419351100921631
    },
    {
      "epoch": 6.701374053955079e-05,
      "model_forward_time": 1.7400376796722412,
      "step": 70269
    },
    {
      "epoch": 6.701374053955079e-05,
      "step": 70269,
      "training_step_time": 7.732458114624023
    },
    {
      "epoch": 6.701469421386718e-05,
      "grad_norm": 0.10723192244768143,
      "learning_rate": 2.2281321300937824e-06,
      "loss": 0.0565,
      "step": 70270
    },
    {
      "epoch": 6.701469421386718e-05,
      "model_forward_time": 1.719092607498169,
      "step": 70270
    },
    {
      "dynamics_loss_avg": 0.055345036834478376,
      "epoch": 6.701469421386718e-05,
      "step": 70270
    },
    {
      "action_loss_avg": 0.0015095756796654313,
      "epoch": 6.701469421386718e-05,
      "step": 70270
    },
    {
      "epoch": 6.701469421386718e-05,
      "step": 70270,
      "training_step_time": 7.760308265686035
    },
    {
      "epoch": 6.70156478881836e-05,
      "model_forward_time": 1.7292978763580322,
      "step": 70271
    },
    {
      "epoch": 6.70156478881836e-05,
      "step": 70271,
      "training_step_time": 7.821398973464966
    },
    {
      "epoch": 6.70166015625e-05,
      "model_forward_time": 1.7353157997131348,
      "step": 70272
    },
    {
      "epoch": 6.70166015625e-05,
      "step": 70272,
      "training_step_time": 8.960823774337769
    },
    {
      "epoch": 6.70175552368164e-05,
      "model_forward_time": 1.808239459991455,
      "step": 70273
    },
    {
      "epoch": 6.70175552368164e-05,
      "step": 70273,
      "training_step_time": 8.346112966537476
    },
    {
      "epoch": 6.701850891113281e-05,
      "model_forward_time": 1.7355568408966064,
      "step": 70274
    },
    {
      "epoch": 6.701850891113281e-05,
      "step": 70274,
      "training_step_time": 8.435875654220581
    },
    {
      "epoch": 6.701946258544922e-05,
      "model_forward_time": 1.7303481101989746,
      "step": 70275
    },
    {
      "epoch": 6.701946258544922e-05,
      "step": 70275,
      "training_step_time": 7.755415201187134
    },
    {
      "epoch": 6.702041625976562e-05,
      "model_forward_time": 2.246978998184204,
      "step": 70276
    },
    {
      "epoch": 6.702041625976562e-05,
      "step": 70276,
      "training_step_time": 7.6275763511657715
    },
    {
      "epoch": 6.702136993408203e-05,
      "model_forward_time": 1.7164595127105713,
      "step": 70277
    },
    {
      "epoch": 6.702136993408203e-05,
      "step": 70277,
      "training_step_time": 6.841169357299805
    },
    {
      "epoch": 6.702232360839844e-05,
      "model_forward_time": 1.7165863513946533,
      "step": 70278
    },
    {
      "epoch": 6.702232360839844e-05,
      "step": 70278,
      "training_step_time": 7.740616083145142
    },
    {
      "epoch": 6.702327728271484e-05,
      "model_forward_time": 1.711143970489502,
      "step": 70279
    },
    {
      "epoch": 6.702327728271484e-05,
      "step": 70279,
      "training_step_time": 7.733684778213501
    },
    {
      "epoch": 6.702423095703125e-05,
      "grad_norm": 0.0830351784825325,
      "learning_rate": 2.2267561517470935e-06,
      "loss": 0.0597,
      "step": 70280
    },
    {
      "epoch": 6.702423095703125e-05,
      "model_forward_time": 1.7121386528015137,
      "step": 70280
    },
    {
      "dynamics_loss_avg": 0.07116125971078872,
      "epoch": 6.702423095703125e-05,
      "step": 70280
    },
    {
      "action_loss_avg": 0.001724532007938251,
      "epoch": 6.702423095703125e-05,
      "step": 70280
    },
    {
      "epoch": 6.702423095703125e-05,
      "step": 70280,
      "training_step_time": 7.7527549266815186
    },
    {
      "epoch": 6.702518463134766e-05,
      "model_forward_time": 1.718327283859253,
      "step": 70281
    },
    {
      "epoch": 6.702518463134766e-05,
      "step": 70281,
      "training_step_time": 9.27862548828125
    },
    {
      "epoch": 6.702613830566406e-05,
      "model_forward_time": 1.408841609954834,
      "step": 70282
    },
    {
      "epoch": 6.702613830566406e-05,
      "step": 70282,
      "training_step_time": 7.795067310333252
    },
    {
      "epoch": 6.702709197998047e-05,
      "model_forward_time": 1.7192420959472656,
      "step": 70283
    },
    {
      "epoch": 6.702709197998047e-05,
      "step": 70283,
      "training_step_time": 8.62296438217163
    },
    {
      "epoch": 6.702804565429688e-05,
      "model_forward_time": 1.7154216766357422,
      "step": 70284
    },
    {
      "epoch": 6.702804565429688e-05,
      "step": 70284,
      "training_step_time": 7.725900173187256
    },
    {
      "epoch": 6.702899932861328e-05,
      "model_forward_time": 2.291567325592041,
      "step": 70285
    },
    {
      "epoch": 6.702899932861328e-05,
      "step": 70285,
      "training_step_time": 7.687313079833984
    },
    {
      "epoch": 6.702995300292969e-05,
      "model_forward_time": 1.71524977684021,
      "step": 70286
    },
    {
      "epoch": 6.702995300292969e-05,
      "step": 70286,
      "training_step_time": 7.687206029891968
    },
    {
      "epoch": 6.70309066772461e-05,
      "model_forward_time": 1.7230873107910156,
      "step": 70287
    },
    {
      "epoch": 6.70309066772461e-05,
      "step": 70287,
      "training_step_time": 7.563662767410278
    },
    {
      "epoch": 6.70318603515625e-05,
      "model_forward_time": 1.7214996814727783,
      "step": 70288
    },
    {
      "epoch": 6.70318603515625e-05,
      "step": 70288,
      "training_step_time": 8.707570552825928
    },
    {
      "epoch": 6.703281402587891e-05,
      "model_forward_time": 1.7047111988067627,
      "step": 70289
    },
    {
      "epoch": 6.703281402587891e-05,
      "step": 70289,
      "training_step_time": 8.802575588226318
    },
    {
      "epoch": 6.703376770019532e-05,
      "grad_norm": 0.0919182077050209,
      "learning_rate": 2.2253804766781844e-06,
      "loss": 0.0569,
      "step": 70290
    },
    {
      "epoch": 6.703376770019532e-05,
      "model_forward_time": 1.7289471626281738,
      "step": 70290
    },
    {
      "dynamics_loss_avg": 0.048141483776271345,
      "epoch": 6.703376770019532e-05,
      "step": 70290
    },
    {
      "action_loss_avg": 0.0008998730627354234,
      "epoch": 6.703376770019532e-05,
      "step": 70290
    },
    {
      "epoch": 6.703376770019532e-05,
      "step": 70290,
      "training_step_time": 8.87858533859253
    },
    {
      "epoch": 6.703472137451172e-05,
      "model_forward_time": 1.7764818668365479,
      "step": 70291
    },
    {
      "epoch": 6.703472137451172e-05,
      "step": 70291,
      "training_step_time": 10.52011489868164
    },
    {
      "epoch": 6.703567504882813e-05,
      "model_forward_time": 1.7117750644683838,
      "step": 70292
    },
    {
      "epoch": 6.703567504882813e-05,
      "step": 70292,
      "training_step_time": 7.742208242416382
    },
    {
      "epoch": 6.703662872314454e-05,
      "model_forward_time": 1.7115261554718018,
      "step": 70293
    },
    {
      "epoch": 6.703662872314454e-05,
      "step": 70293,
      "training_step_time": 8.766039371490479
    },
    {
      "epoch": 6.703758239746093e-05,
      "model_forward_time": 2.2976021766662598,
      "step": 70294
    },
    {
      "epoch": 6.703758239746093e-05,
      "step": 70294,
      "training_step_time": 7.726586580276489
    },
    {
      "epoch": 6.703853607177735e-05,
      "model_forward_time": 1.7165513038635254,
      "step": 70295
    },
    {
      "epoch": 6.703853607177735e-05,
      "step": 70295,
      "training_step_time": 7.649001598358154
    },
    {
      "epoch": 6.703948974609376e-05,
      "model_forward_time": 1.7064123153686523,
      "step": 70296
    },
    {
      "epoch": 6.703948974609376e-05,
      "step": 70296,
      "training_step_time": 7.675307512283325
    },
    {
      "epoch": 6.704044342041015e-05,
      "model_forward_time": 1.7245008945465088,
      "step": 70297
    },
    {
      "epoch": 6.704044342041015e-05,
      "step": 70297,
      "training_step_time": 8.61837100982666
    },
    {
      "epoch": 6.704139709472656e-05,
      "model_forward_time": 1.3269100189208984,
      "step": 70298
    },
    {
      "epoch": 6.704139709472656e-05,
      "step": 70298,
      "training_step_time": 7.7633466720581055
    },
    {
      "epoch": 6.704235076904298e-05,
      "model_forward_time": 1.7127528190612793,
      "step": 70299
    },
    {
      "epoch": 6.704235076904298e-05,
      "step": 70299,
      "training_step_time": 7.895040988922119
    },
    {
      "epoch": 6.704330444335937e-05,
      "grad_norm": 0.07782191038131714,
      "learning_rate": 2.2240051050374935e-06,
      "loss": 0.055,
      "step": 70300
    },
    {
      "epoch": 6.704330444335937e-05,
      "model_forward_time": 1.7440361976623535,
      "step": 70300
    },
    {
      "dynamics_loss_avg": 0.05452477522194386,
      "epoch": 6.704330444335937e-05,
      "step": 70300
    },
    {
      "action_loss_avg": 0.001218030267045833,
      "epoch": 6.704330444335937e-05,
      "step": 70300
    },
    {
      "epoch": 6.704330444335937e-05,
      "step": 70300,
      "training_step_time": 8.43881869316101
    },
    {
      "epoch": 6.704425811767578e-05,
      "model_forward_time": 1.713507890701294,
      "step": 70301
    },
    {
      "epoch": 6.704425811767578e-05,
      "step": 70301,
      "training_step_time": 8.267177820205688
    },
    {
      "epoch": 6.70452117919922e-05,
      "model_forward_time": 1.7168424129486084,
      "step": 70302
    },
    {
      "epoch": 6.70452117919922e-05,
      "step": 70302,
      "training_step_time": 8.637007713317871
    },
    {
      "epoch": 6.704616546630859e-05,
      "model_forward_time": 1.7170648574829102,
      "step": 70303
    },
    {
      "epoch": 6.704616546630859e-05,
      "step": 70303,
      "training_step_time": 7.792800188064575
    },
    {
      "epoch": 6.7047119140625e-05,
      "model_forward_time": 2.2818636894226074,
      "step": 70304
    },
    {
      "epoch": 6.7047119140625e-05,
      "step": 70304,
      "training_step_time": 7.648874759674072
    },
    {
      "epoch": 6.704807281494141e-05,
      "model_forward_time": 1.7177906036376953,
      "step": 70305
    },
    {
      "epoch": 6.704807281494141e-05,
      "step": 70305,
      "training_step_time": 7.485547065734863
    },
    {
      "epoch": 6.704902648925781e-05,
      "model_forward_time": 1.7157020568847656,
      "step": 70306
    },
    {
      "epoch": 6.704902648925781e-05,
      "step": 70306,
      "training_step_time": 8.435035228729248
    },
    {
      "epoch": 6.704998016357422e-05,
      "model_forward_time": 1.723860263824463,
      "step": 70307
    },
    {
      "epoch": 6.704998016357422e-05,
      "step": 70307,
      "training_step_time": 7.781836748123169
    },
    {
      "epoch": 6.705093383789063e-05,
      "model_forward_time": 1.7154319286346436,
      "step": 70308
    },
    {
      "epoch": 6.705093383789063e-05,
      "step": 70308,
      "training_step_time": 7.826894521713257
    },
    {
      "epoch": 6.705188751220703e-05,
      "model_forward_time": 1.7352442741394043,
      "step": 70309
    },
    {
      "epoch": 6.705188751220703e-05,
      "step": 70309,
      "training_step_time": 7.775015115737915
    },
    {
      "epoch": 6.705284118652344e-05,
      "grad_norm": 0.09114803373813629,
      "learning_rate": 2.22263003697543e-06,
      "loss": 0.0567,
      "step": 70310
    },
    {
      "epoch": 6.705284118652344e-05,
      "model_forward_time": 1.7200958728790283,
      "step": 70310
    },
    {
      "dynamics_loss_avg": 0.06563769429922103,
      "epoch": 6.705284118652344e-05,
      "step": 70310
    },
    {
      "action_loss_avg": 0.002115182761917822,
      "epoch": 6.705284118652344e-05,
      "step": 70310
    },
    {
      "epoch": 6.705284118652344e-05,
      "step": 70310,
      "training_step_time": 8.760887861251831
    },
    {
      "epoch": 6.705379486083985e-05,
      "model_forward_time": 1.7149848937988281,
      "step": 70311
    },
    {
      "epoch": 6.705379486083985e-05,
      "step": 70311,
      "training_step_time": 7.673869848251343
    },
    {
      "epoch": 6.705474853515625e-05,
      "model_forward_time": 1.7068495750427246,
      "step": 70312
    },
    {
      "epoch": 6.705474853515625e-05,
      "step": 70312,
      "training_step_time": 7.756275177001953
    },
    {
      "epoch": 6.705570220947266e-05,
      "model_forward_time": 2.281862735748291,
      "step": 70313
    },
    {
      "epoch": 6.705570220947266e-05,
      "step": 70313,
      "training_step_time": 7.715579986572266
    },
    {
      "epoch": 6.705665588378906e-05,
      "model_forward_time": 1.7118818759918213,
      "step": 70314
    },
    {
      "epoch": 6.705665588378906e-05,
      "step": 70314,
      "training_step_time": 8.349183797836304
    },
    {
      "epoch": 6.705760955810547e-05,
      "model_forward_time": 1.7183268070220947,
      "step": 70315
    },
    {
      "epoch": 6.705760955810547e-05,
      "step": 70315,
      "training_step_time": 7.67275333404541
    },
    {
      "epoch": 6.705856323242188e-05,
      "model_forward_time": 1.732210397720337,
      "step": 70316
    },
    {
      "epoch": 6.705856323242188e-05,
      "step": 70316,
      "training_step_time": 8.444596290588379
    },
    {
      "epoch": 6.705951690673827e-05,
      "model_forward_time": 1.7271249294281006,
      "step": 70317
    },
    {
      "epoch": 6.705951690673827e-05,
      "step": 70317,
      "training_step_time": 8.693940877914429
    },
    {
      "epoch": 6.706047058105469e-05,
      "model_forward_time": 1.6838030815124512,
      "step": 70318
    },
    {
      "epoch": 6.706047058105469e-05,
      "step": 70318,
      "training_step_time": 7.8289220333099365
    },
    {
      "epoch": 6.70614242553711e-05,
      "model_forward_time": 1.794785737991333,
      "step": 70319
    },
    {
      "epoch": 6.70614242553711e-05,
      "step": 70319,
      "training_step_time": 7.793440818786621
    },
    {
      "epoch": 6.70623779296875e-05,
      "grad_norm": 0.0871657207608223,
      "learning_rate": 2.2212552726423705e-06,
      "loss": 0.0585,
      "step": 70320
    },
    {
      "epoch": 6.70623779296875e-05,
      "model_forward_time": 1.719081163406372,
      "step": 70320
    },
    {
      "dynamics_loss_avg": 0.06881003547459841,
      "epoch": 6.70623779296875e-05,
      "step": 70320
    },
    {
      "action_loss_avg": 0.0022487423731945453,
      "epoch": 6.70623779296875e-05,
      "step": 70320
    },
    {
      "epoch": 6.70623779296875e-05,
      "step": 70320,
      "training_step_time": 7.657534837722778
    },
    {
      "epoch": 6.70633316040039e-05,
      "model_forward_time": 1.7311139106750488,
      "step": 70321
    },
    {
      "epoch": 6.70633316040039e-05,
      "step": 70321,
      "training_step_time": 8.837780475616455
    },
    {
      "epoch": 6.706428527832032e-05,
      "model_forward_time": 2.3041908740997314,
      "step": 70322
    },
    {
      "epoch": 6.706428527832032e-05,
      "step": 70322,
      "training_step_time": 7.6670918464660645
    },
    {
      "epoch": 6.706523895263671e-05,
      "model_forward_time": 1.352985143661499,
      "step": 70323
    },
    {
      "epoch": 6.706523895263671e-05,
      "step": 70323,
      "training_step_time": 7.585521221160889
    },
    {
      "epoch": 6.706619262695312e-05,
      "model_forward_time": 1.7287461757659912,
      "step": 70324
    },
    {
      "epoch": 6.706619262695312e-05,
      "step": 70324,
      "training_step_time": 7.582617282867432
    },
    {
      "epoch": 6.706714630126953e-05,
      "model_forward_time": 1.730043649673462,
      "step": 70325
    },
    {
      "epoch": 6.706714630126953e-05,
      "step": 70325,
      "training_step_time": 9.30524492263794
    },
    {
      "epoch": 6.706809997558593e-05,
      "model_forward_time": 1.7345960140228271,
      "step": 70326
    },
    {
      "epoch": 6.706809997558593e-05,
      "step": 70326,
      "training_step_time": 7.757258653640747
    },
    {
      "epoch": 6.706905364990234e-05,
      "model_forward_time": 1.7276597023010254,
      "step": 70327
    },
    {
      "epoch": 6.706905364990234e-05,
      "step": 70327,
      "training_step_time": 8.912132024765015
    },
    {
      "epoch": 6.707000732421875e-05,
      "model_forward_time": 1.8177826404571533,
      "step": 70328
    },
    {
      "epoch": 6.707000732421875e-05,
      "step": 70328,
      "training_step_time": 8.336113214492798
    },
    {
      "epoch": 6.707096099853515e-05,
      "model_forward_time": 1.354151964187622,
      "step": 70329
    },
    {
      "epoch": 6.707096099853515e-05,
      "step": 70329,
      "training_step_time": 8.298492670059204
    },
    {
      "epoch": 6.707191467285156e-05,
      "grad_norm": 0.10237377136945724,
      "learning_rate": 2.219880812188658e-06,
      "loss": 0.0575,
      "step": 70330
    },
    {
      "epoch": 6.707191467285156e-05,
      "model_forward_time": 1.7317585945129395,
      "step": 70330
    },
    {
      "dynamics_loss_avg": 0.04764916263520717,
      "epoch": 6.707191467285156e-05,
      "step": 70330
    },
    {
      "action_loss_avg": 0.0010156527190702037,
      "epoch": 6.707191467285156e-05,
      "step": 70330
    },
    {
      "epoch": 6.707191467285156e-05,
      "step": 70330,
      "training_step_time": 8.260991096496582
    },
    {
      "epoch": 6.707286834716797e-05,
      "model_forward_time": 2.301124334335327,
      "step": 70331
    },
    {
      "epoch": 6.707286834716797e-05,
      "step": 70331,
      "training_step_time": 7.617358207702637
    },
    {
      "epoch": 6.707382202148437e-05,
      "model_forward_time": 1.733016014099121,
      "step": 70332
    },
    {
      "epoch": 6.707382202148437e-05,
      "step": 70332,
      "training_step_time": 6.768558025360107
    },
    {
      "epoch": 6.707477569580078e-05,
      "model_forward_time": 1.7185869216918945,
      "step": 70333
    },
    {
      "epoch": 6.707477569580078e-05,
      "step": 70333,
      "training_step_time": 8.08110499382019
    },
    {
      "epoch": 6.707572937011719e-05,
      "model_forward_time": 1.7263126373291016,
      "step": 70334
    },
    {
      "epoch": 6.707572937011719e-05,
      "step": 70334,
      "training_step_time": 7.6172871589660645
    },
    {
      "epoch": 6.707668304443359e-05,
      "model_forward_time": 1.7319881916046143,
      "step": 70335
    },
    {
      "epoch": 6.707668304443359e-05,
      "step": 70335,
      "training_step_time": 8.271706104278564
    },
    {
      "epoch": 6.707763671875e-05,
      "model_forward_time": 1.7359061241149902,
      "step": 70336
    },
    {
      "epoch": 6.707763671875e-05,
      "step": 70336,
      "training_step_time": 8.541213512420654
    },
    {
      "epoch": 6.707859039306641e-05,
      "model_forward_time": 1.8184738159179688,
      "step": 70337
    },
    {
      "epoch": 6.707859039306641e-05,
      "step": 70337,
      "training_step_time": 8.690309762954712
    },
    {
      "epoch": 6.70795440673828e-05,
      "model_forward_time": 1.736093521118164,
      "step": 70338
    },
    {
      "epoch": 6.70795440673828e-05,
      "step": 70338,
      "training_step_time": 8.718708038330078
    },
    {
      "epoch": 6.708049774169922e-05,
      "model_forward_time": 1.7279412746429443,
      "step": 70339
    },
    {
      "epoch": 6.708049774169922e-05,
      "step": 70339,
      "training_step_time": 8.749546527862549
    },
    {
      "epoch": 6.708145141601563e-05,
      "grad_norm": 0.07072746008634567,
      "learning_rate": 2.2185066557645977e-06,
      "loss": 0.0537,
      "step": 70340
    },
    {
      "epoch": 6.708145141601563e-05,
      "model_forward_time": 2.2988715171813965,
      "step": 70340
    },
    {
      "dynamics_loss_avg": 0.07037950120866299,
      "epoch": 6.708145141601563e-05,
      "step": 70340
    },
    {
      "action_loss_avg": 0.0013249670621007681,
      "epoch": 6.708145141601563e-05,
      "step": 70340
    },
    {
      "epoch": 6.708145141601563e-05,
      "step": 70340,
      "training_step_time": 7.657287120819092
    },
    {
      "epoch": 6.708240509033203e-05,
      "model_forward_time": 1.7268176078796387,
      "step": 70341
    },
    {
      "epoch": 6.708240509033203e-05,
      "step": 70341,
      "training_step_time": 7.265669107437134
    },
    {
      "epoch": 6.708335876464844e-05,
      "model_forward_time": 1.730839490890503,
      "step": 70342
    },
    {
      "epoch": 6.708335876464844e-05,
      "step": 70342,
      "training_step_time": 7.534426927566528
    },
    {
      "epoch": 6.708431243896485e-05,
      "model_forward_time": 1.7326099872589111,
      "step": 70343
    },
    {
      "epoch": 6.708431243896485e-05,
      "step": 70343,
      "training_step_time": 7.593898057937622
    },
    {
      "epoch": 6.708526611328124e-05,
      "model_forward_time": 1.733518362045288,
      "step": 70344
    },
    {
      "epoch": 6.708526611328124e-05,
      "step": 70344,
      "training_step_time": 8.284791469573975
    },
    {
      "epoch": 6.708621978759766e-05,
      "model_forward_time": 1.7304189205169678,
      "step": 70345
    },
    {
      "epoch": 6.708621978759766e-05,
      "step": 70345,
      "training_step_time": 7.692323684692383
    },
    {
      "epoch": 6.708717346191407e-05,
      "model_forward_time": 1.7953987121582031,
      "step": 70346
    },
    {
      "epoch": 6.708717346191407e-05,
      "step": 70346,
      "training_step_time": 8.931231021881104
    },
    {
      "epoch": 6.708812713623046e-05,
      "model_forward_time": 1.7262213230133057,
      "step": 70347
    },
    {
      "epoch": 6.708812713623046e-05,
      "step": 70347,
      "training_step_time": 8.531270980834961
    },
    {
      "epoch": 6.708908081054687e-05,
      "model_forward_time": 1.7325360774993896,
      "step": 70348
    },
    {
      "epoch": 6.708908081054687e-05,
      "step": 70348,
      "training_step_time": 9.74226689338684
    },
    {
      "epoch": 6.709003448486328e-05,
      "model_forward_time": 2.315751791000366,
      "step": 70349
    },
    {
      "epoch": 6.709003448486328e-05,
      "step": 70349,
      "training_step_time": 8.671393632888794
    },
    {
      "epoch": 6.709098815917968e-05,
      "grad_norm": 0.07714133709669113,
      "learning_rate": 2.2171328035204704e-06,
      "loss": 0.054,
      "step": 70350
    },
    {
      "epoch": 6.709098815917968e-05,
      "model_forward_time": 1.7213587760925293,
      "step": 70350
    },
    {
      "dynamics_loss_avg": 0.050377510488033295,
      "epoch": 6.709098815917968e-05,
      "step": 70350
    },
    {
      "action_loss_avg": 0.0012692407181020826,
      "epoch": 6.709098815917968e-05,
      "step": 70350
    },
    {
      "epoch": 6.709098815917968e-05,
      "step": 70350,
      "training_step_time": 7.364262580871582
    },
    {
      "epoch": 6.709194183349609e-05,
      "model_forward_time": 1.7260127067565918,
      "step": 70351
    },
    {
      "epoch": 6.709194183349609e-05,
      "step": 70351,
      "training_step_time": 7.504012107849121
    },
    {
      "epoch": 6.70928955078125e-05,
      "model_forward_time": 1.722806453704834,
      "step": 70352
    },
    {
      "epoch": 6.70928955078125e-05,
      "step": 70352,
      "training_step_time": 7.649614095687866
    },
    {
      "epoch": 6.70938491821289e-05,
      "model_forward_time": 1.7377498149871826,
      "step": 70353
    },
    {
      "epoch": 6.70938491821289e-05,
      "step": 70353,
      "training_step_time": 7.737171411514282
    },
    {
      "epoch": 6.709480285644531e-05,
      "model_forward_time": 1.7075674533843994,
      "step": 70354
    },
    {
      "epoch": 6.709480285644531e-05,
      "step": 70354,
      "training_step_time": 7.757108211517334
    },
    {
      "epoch": 6.709575653076172e-05,
      "model_forward_time": 1.7798216342926025,
      "step": 70355
    },
    {
      "epoch": 6.709575653076172e-05,
      "step": 70355,
      "training_step_time": 7.863435983657837
    },
    {
      "epoch": 6.709671020507812e-05,
      "model_forward_time": 1.7367744445800781,
      "step": 70356
    },
    {
      "epoch": 6.709671020507812e-05,
      "step": 70356,
      "training_step_time": 7.793264627456665
    },
    {
      "epoch": 6.709766387939453e-05,
      "model_forward_time": 1.7281644344329834,
      "step": 70357
    },
    {
      "epoch": 6.709766387939453e-05,
      "step": 70357,
      "training_step_time": 7.687225818634033
    },
    {
      "epoch": 6.709861755371094e-05,
      "model_forward_time": 1.7148866653442383,
      "step": 70358
    },
    {
      "epoch": 6.709861755371094e-05,
      "step": 70358,
      "training_step_time": 8.247003555297852
    },
    {
      "epoch": 6.709957122802734e-05,
      "model_forward_time": 1.730827808380127,
      "step": 70359
    },
    {
      "epoch": 6.709957122802734e-05,
      "step": 70359,
      "training_step_time": 7.44539737701416
    },
    {
      "epoch": 6.710052490234375e-05,
      "grad_norm": 0.12187129259109497,
      "learning_rate": 2.2157592556065134e-06,
      "loss": 0.056,
      "step": 70360
    },
    {
      "epoch": 6.710052490234375e-05,
      "model_forward_time": 1.710580587387085,
      "step": 70360
    },
    {
      "dynamics_loss_avg": 0.05424270275980234,
      "epoch": 6.710052490234375e-05,
      "step": 70360
    },
    {
      "action_loss_avg": 0.0011542859399924054,
      "epoch": 6.710052490234375e-05,
      "step": 70360
    },
    {
      "epoch": 6.710052490234375e-05,
      "step": 70360,
      "training_step_time": 7.6115500926971436
    },
    {
      "epoch": 6.710147857666016e-05,
      "model_forward_time": 1.7261404991149902,
      "step": 70361
    },
    {
      "epoch": 6.710147857666016e-05,
      "step": 70361,
      "training_step_time": 7.648900508880615
    },
    {
      "epoch": 6.710243225097656e-05,
      "model_forward_time": 1.742802381515503,
      "step": 70362
    },
    {
      "epoch": 6.710243225097656e-05,
      "step": 70362,
      "training_step_time": 7.698711395263672
    },
    {
      "epoch": 6.710338592529297e-05,
      "model_forward_time": 1.6844596862792969,
      "step": 70363
    },
    {
      "epoch": 6.710338592529297e-05,
      "step": 70363,
      "training_step_time": 7.7933244705200195
    },
    {
      "epoch": 6.710433959960938e-05,
      "model_forward_time": 1.324045181274414,
      "step": 70364
    },
    {
      "epoch": 6.710433959960938e-05,
      "step": 70364,
      "training_step_time": 8.727408409118652
    },
    {
      "epoch": 6.710529327392578e-05,
      "model_forward_time": 1.7120120525360107,
      "step": 70365
    },
    {
      "epoch": 6.710529327392578e-05,
      "step": 70365,
      "training_step_time": 8.617729902267456
    },
    {
      "epoch": 6.710624694824219e-05,
      "model_forward_time": 1.7173032760620117,
      "step": 70366
    },
    {
      "epoch": 6.710624694824219e-05,
      "step": 70366,
      "training_step_time": 8.751540184020996
    },
    {
      "epoch": 6.71072006225586e-05,
      "model_forward_time": 1.7140941619873047,
      "step": 70367
    },
    {
      "epoch": 6.71072006225586e-05,
      "step": 70367,
      "training_step_time": 8.849865674972534
    },
    {
      "epoch": 6.7108154296875e-05,
      "model_forward_time": 2.264484405517578,
      "step": 70368
    },
    {
      "epoch": 6.7108154296875e-05,
      "step": 70368,
      "training_step_time": 7.745432615280151
    },
    {
      "epoch": 6.71091079711914e-05,
      "model_forward_time": 1.7080409526824951,
      "step": 70369
    },
    {
      "epoch": 6.71091079711914e-05,
      "step": 70369,
      "training_step_time": 7.7157697677612305
    },
    {
      "epoch": 6.711006164550782e-05,
      "grad_norm": 0.08898511528968811,
      "learning_rate": 2.2143860121729383e-06,
      "loss": 0.0528,
      "step": 70370
    },
    {
      "epoch": 6.711006164550782e-05,
      "model_forward_time": 1.697713851928711,
      "step": 70370
    },
    {
      "dynamics_loss_avg": 0.046393837220966815,
      "epoch": 6.711006164550782e-05,
      "step": 70370
    },
    {
      "action_loss_avg": 0.0008696443284861744,
      "epoch": 6.711006164550782e-05,
      "step": 70370
    },
    {
      "epoch": 6.711006164550782e-05,
      "step": 70370,
      "training_step_time": 7.69426417350769
    },
    {
      "epoch": 6.711101531982421e-05,
      "model_forward_time": 1.7058355808258057,
      "step": 70371
    },
    {
      "epoch": 6.711101531982421e-05,
      "step": 70371,
      "training_step_time": 7.719252586364746
    },
    {
      "epoch": 6.711196899414062e-05,
      "model_forward_time": 1.7147314548492432,
      "step": 70372
    },
    {
      "epoch": 6.711196899414062e-05,
      "step": 70372,
      "training_step_time": 7.773056745529175
    },
    {
      "epoch": 6.711292266845704e-05,
      "model_forward_time": 1.7094955444335938,
      "step": 70373
    },
    {
      "epoch": 6.711292266845704e-05,
      "step": 70373,
      "training_step_time": 7.642558813095093
    },
    {
      "epoch": 6.711387634277343e-05,
      "model_forward_time": 1.709792137145996,
      "step": 70374
    },
    {
      "epoch": 6.711387634277343e-05,
      "step": 70374,
      "training_step_time": 9.96066665649414
    },
    {
      "epoch": 6.711483001708984e-05,
      "model_forward_time": 1.6989107131958008,
      "step": 70375
    },
    {
      "epoch": 6.711483001708984e-05,
      "step": 70375,
      "training_step_time": 7.734771728515625
    },
    {
      "epoch": 6.711578369140625e-05,
      "model_forward_time": 1.7035934925079346,
      "step": 70376
    },
    {
      "epoch": 6.711578369140625e-05,
      "step": 70376,
      "training_step_time": 7.74128270149231
    },
    {
      "epoch": 6.711673736572265e-05,
      "model_forward_time": 2.2867605686187744,
      "step": 70377
    },
    {
      "epoch": 6.711673736572265e-05,
      "step": 70377,
      "training_step_time": 7.762986421585083
    },
    {
      "epoch": 6.711769104003906e-05,
      "model_forward_time": 1.7050843238830566,
      "step": 70378
    },
    {
      "epoch": 6.711769104003906e-05,
      "step": 70378,
      "training_step_time": 7.687358379364014
    },
    {
      "epoch": 6.711864471435547e-05,
      "model_forward_time": 1.7066268920898438,
      "step": 70379
    },
    {
      "epoch": 6.711864471435547e-05,
      "step": 70379,
      "training_step_time": 7.7232701778411865
    },
    {
      "epoch": 6.711959838867187e-05,
      "grad_norm": 0.07632811367511749,
      "learning_rate": 2.2130130733699206e-06,
      "loss": 0.053,
      "step": 70380
    },
    {
      "epoch": 6.711959838867187e-05,
      "model_forward_time": 1.704930305480957,
      "step": 70380
    },
    {
      "dynamics_loss_avg": 0.057506203278899194,
      "epoch": 6.711959838867187e-05,
      "step": 70380
    },
    {
      "action_loss_avg": 0.0020520001475233586,
      "epoch": 6.711959838867187e-05,
      "step": 70380
    },
    {
      "epoch": 6.711959838867187e-05,
      "step": 70380,
      "training_step_time": 8.554268598556519
    },
    {
      "epoch": 6.712055206298828e-05,
      "model_forward_time": 1.7077417373657227,
      "step": 70381
    },
    {
      "epoch": 6.712055206298828e-05,
      "step": 70381,
      "training_step_time": 7.77997088432312
    },
    {
      "epoch": 6.712150573730469e-05,
      "model_forward_time": 1.7018859386444092,
      "step": 70382
    },
    {
      "epoch": 6.712150573730469e-05,
      "step": 70382,
      "training_step_time": 8.545787811279297
    },
    {
      "epoch": 6.712245941162109e-05,
      "model_forward_time": 1.7668843269348145,
      "step": 70383
    },
    {
      "epoch": 6.712245941162109e-05,
      "step": 70383,
      "training_step_time": 8.448916912078857
    },
    {
      "epoch": 6.71234130859375e-05,
      "model_forward_time": 1.6955983638763428,
      "step": 70384
    },
    {
      "epoch": 6.71234130859375e-05,
      "step": 70384,
      "training_step_time": 8.678917169570923
    },
    {
      "epoch": 6.712436676025391e-05,
      "model_forward_time": 1.708571195602417,
      "step": 70385
    },
    {
      "epoch": 6.712436676025391e-05,
      "step": 70385,
      "training_step_time": 7.703542232513428
    },
    {
      "epoch": 6.712532043457031e-05,
      "model_forward_time": 2.275726795196533,
      "step": 70386
    },
    {
      "epoch": 6.712532043457031e-05,
      "step": 70386,
      "training_step_time": 7.70716667175293
    },
    {
      "epoch": 6.712627410888672e-05,
      "model_forward_time": 1.7043986320495605,
      "step": 70387
    },
    {
      "epoch": 6.712627410888672e-05,
      "step": 70387,
      "training_step_time": 7.565839529037476
    },
    {
      "epoch": 6.712722778320313e-05,
      "model_forward_time": 1.703068733215332,
      "step": 70388
    },
    {
      "epoch": 6.712722778320313e-05,
      "step": 70388,
      "training_step_time": 7.632385969161987
    },
    {
      "epoch": 6.712818145751953e-05,
      "model_forward_time": 1.7228069305419922,
      "step": 70389
    },
    {
      "epoch": 6.712818145751953e-05,
      "step": 70389,
      "training_step_time": 7.662140369415283
    },
    {
      "epoch": 6.712913513183594e-05,
      "grad_norm": 0.06727082282304764,
      "learning_rate": 2.2116404393476054e-06,
      "loss": 0.0525,
      "step": 70390
    },
    {
      "epoch": 6.712913513183594e-05,
      "model_forward_time": 1.714707851409912,
      "step": 70390
    },
    {
      "dynamics_loss_avg": 0.056293419748544696,
      "epoch": 6.712913513183594e-05,
      "step": 70390
    },
    {
      "action_loss_avg": 0.0015606589760864154,
      "epoch": 6.712913513183594e-05,
      "step": 70390
    },
    {
      "epoch": 6.712913513183594e-05,
      "step": 70390,
      "training_step_time": 7.783069372177124
    },
    {
      "epoch": 6.713008880615235e-05,
      "model_forward_time": 1.7157540321350098,
      "step": 70391
    },
    {
      "epoch": 6.713008880615235e-05,
      "step": 70391,
      "training_step_time": 8.398373365402222
    },
    {
      "epoch": 6.713104248046875e-05,
      "model_forward_time": 1.7625586986541748,
      "step": 70392
    },
    {
      "epoch": 6.713104248046875e-05,
      "step": 70392,
      "training_step_time": 7.59390664100647
    },
    {
      "epoch": 6.713199615478516e-05,
      "model_forward_time": 1.717989444732666,
      "step": 70393
    },
    {
      "epoch": 6.713199615478516e-05,
      "step": 70393,
      "training_step_time": 8.554134368896484
    },
    {
      "epoch": 6.713294982910157e-05,
      "model_forward_time": 1.711451530456543,
      "step": 70394
    },
    {
      "epoch": 6.713294982910157e-05,
      "step": 70394,
      "training_step_time": 8.126714944839478
    },
    {
      "epoch": 6.713390350341796e-05,
      "model_forward_time": 2.2974913120269775,
      "step": 70395
    },
    {
      "epoch": 6.713390350341796e-05,
      "step": 70395,
      "training_step_time": 7.5943169593811035
    },
    {
      "epoch": 6.713485717773438e-05,
      "model_forward_time": 1.7319276332855225,
      "step": 70396
    },
    {
      "epoch": 6.713485717773438e-05,
      "step": 70396,
      "training_step_time": 7.586660146713257
    },
    {
      "epoch": 6.713581085205079e-05,
      "model_forward_time": 1.7245628833770752,
      "step": 70397
    },
    {
      "epoch": 6.713581085205079e-05,
      "step": 70397,
      "training_step_time": 8.405234098434448
    },
    {
      "epoch": 6.713676452636718e-05,
      "model_forward_time": 1.7153041362762451,
      "step": 70398
    },
    {
      "epoch": 6.713676452636718e-05,
      "step": 70398,
      "training_step_time": 7.541860342025757
    },
    {
      "epoch": 6.71377182006836e-05,
      "model_forward_time": 1.7226178646087646,
      "step": 70399
    },
    {
      "epoch": 6.71377182006836e-05,
      "step": 70399,
      "training_step_time": 7.649962425231934
    },
    {
      "epoch": 6.7138671875e-05,
      "grad_norm": 0.11435236036777496,
      "learning_rate": 2.210268110256097e-06,
      "loss": 0.0532,
      "step": 70400
    },
    {
      "epoch": 6.7138671875e-05,
      "model_forward_time": 1.7055168151855469,
      "step": 70400
    },
    {
      "dynamics_loss_avg": 0.0584860734641552,
      "epoch": 6.7138671875e-05,
      "step": 70400
    },
    {
      "action_loss_avg": 0.0013594595715403558,
      "epoch": 6.7138671875e-05,
      "step": 70400
    },
    {
      "epoch": 6.7138671875e-05,
      "step": 70400,
      "training_step_time": 8.191478729248047
    },
    {
      "epoch": 6.71396255493164e-05,
      "model_forward_time": 1.7245349884033203,
      "step": 70401
    },
    {
      "epoch": 6.71396255493164e-05,
      "step": 70401,
      "training_step_time": 8.799606323242188
    },
    {
      "epoch": 6.714057922363281e-05,
      "model_forward_time": 1.7113754749298096,
      "step": 70402
    },
    {
      "epoch": 6.714057922363281e-05,
      "step": 70402,
      "training_step_time": 8.469217538833618
    },
    {
      "epoch": 6.714153289794922e-05,
      "model_forward_time": 1.705359935760498,
      "step": 70403
    },
    {
      "epoch": 6.714153289794922e-05,
      "step": 70403,
      "training_step_time": 8.23203182220459
    },
    {
      "epoch": 6.714248657226562e-05,
      "model_forward_time": 1.7114145755767822,
      "step": 70404
    },
    {
      "epoch": 6.714248657226562e-05,
      "step": 70404,
      "training_step_time": 9.171506404876709
    },
    {
      "epoch": 6.714344024658203e-05,
      "model_forward_time": 1.3045594692230225,
      "step": 70405
    },
    {
      "epoch": 6.714344024658203e-05,
      "step": 70405,
      "training_step_time": 7.644937753677368
    },
    {
      "epoch": 6.714439392089844e-05,
      "model_forward_time": 1.6957769393920898,
      "step": 70406
    },
    {
      "epoch": 6.714439392089844e-05,
      "step": 70406,
      "training_step_time": 7.676180362701416
    },
    {
      "epoch": 6.714534759521484e-05,
      "model_forward_time": 1.6999123096466064,
      "step": 70407
    },
    {
      "epoch": 6.714534759521484e-05,
      "step": 70407,
      "training_step_time": 7.62497091293335
    },
    {
      "epoch": 6.714630126953125e-05,
      "model_forward_time": 2.279831647872925,
      "step": 70408
    },
    {
      "epoch": 6.714630126953125e-05,
      "step": 70408,
      "training_step_time": 9.86839771270752
    },
    {
      "epoch": 6.714725494384766e-05,
      "model_forward_time": 1.699613094329834,
      "step": 70409
    },
    {
      "epoch": 6.714725494384766e-05,
      "step": 70409,
      "training_step_time": 6.776634693145752
    },
    {
      "epoch": 6.714820861816406e-05,
      "grad_norm": 0.08804086595773697,
      "learning_rate": 2.2088960862454762e-06,
      "loss": 0.0544,
      "step": 70410
    },
    {
      "epoch": 6.714820861816406e-05,
      "model_forward_time": 1.7001466751098633,
      "step": 70410
    },
    {
      "dynamics_loss_avg": 0.05336897224187851,
      "epoch": 6.714820861816406e-05,
      "step": 70410
    },
    {
      "action_loss_avg": 0.0013911643211031333,
      "epoch": 6.714820861816406e-05,
      "step": 70410
    },
    {
      "epoch": 6.714820861816406e-05,
      "step": 70410,
      "training_step_time": 6.760560512542725
    },
    {
      "epoch": 6.714916229248047e-05,
      "model_forward_time": 1.703366756439209,
      "step": 70411
    },
    {
      "epoch": 6.714916229248047e-05,
      "step": 70411,
      "training_step_time": 6.7341227531433105
    },
    {
      "epoch": 6.715011596679688e-05,
      "model_forward_time": 1.7074625492095947,
      "step": 70412
    },
    {
      "epoch": 6.715011596679688e-05,
      "step": 70412,
      "training_step_time": 6.7839035987854
    },
    {
      "epoch": 6.715106964111328e-05,
      "model_forward_time": 1.6975491046905518,
      "step": 70413
    },
    {
      "epoch": 6.715106964111328e-05,
      "step": 70413,
      "training_step_time": 6.812157392501831
    },
    {
      "epoch": 6.715202331542969e-05,
      "model_forward_time": 1.7044365406036377,
      "step": 70414
    },
    {
      "epoch": 6.715202331542969e-05,
      "step": 70414,
      "training_step_time": 6.745770215988159
    },
    {
      "epoch": 6.71529769897461e-05,
      "model_forward_time": 1.6965956687927246,
      "step": 70415
    },
    {
      "epoch": 6.71529769897461e-05,
      "step": 70415,
      "training_step_time": 6.7415595054626465
    },
    {
      "epoch": 6.71539306640625e-05,
      "model_forward_time": 1.7056431770324707,
      "step": 70416
    },
    {
      "epoch": 6.71539306640625e-05,
      "step": 70416,
      "training_step_time": 7.898101806640625
    },
    {
      "epoch": 6.715488433837891e-05,
      "model_forward_time": 1.7029688358306885,
      "step": 70417
    },
    {
      "epoch": 6.715488433837891e-05,
      "step": 70417,
      "training_step_time": 8.915915250778198
    },
    {
      "epoch": 6.715583801269532e-05,
      "model_forward_time": 1.7002220153808594,
      "step": 70418
    },
    {
      "epoch": 6.715583801269532e-05,
      "step": 70418,
      "training_step_time": 7.758397102355957
    },
    {
      "epoch": 6.715679168701172e-05,
      "model_forward_time": 1.7008872032165527,
      "step": 70419
    },
    {
      "epoch": 6.715679168701172e-05,
      "step": 70419,
      "training_step_time": 6.7246294021606445
    },
    {
      "epoch": 6.715774536132813e-05,
      "grad_norm": 0.11279042810201645,
      "learning_rate": 2.207524367465781e-06,
      "loss": 0.0557,
      "step": 70420
    },
    {
      "epoch": 6.715774536132813e-05,
      "model_forward_time": 1.7015483379364014,
      "step": 70420
    },
    {
      "dynamics_loss_avg": 0.05991086345165968,
      "epoch": 6.715774536132813e-05,
      "step": 70420
    },
    {
      "action_loss_avg": 0.0014713740849401802,
      "epoch": 6.715774536132813e-05,
      "step": 70420
    },
    {
      "epoch": 6.715774536132813e-05,
      "step": 70420,
      "training_step_time": 6.764421463012695
    },
    {
      "epoch": 6.715869903564454e-05,
      "model_forward_time": 1.6993203163146973,
      "step": 70421
    },
    {
      "epoch": 6.715869903564454e-05,
      "step": 70421,
      "training_step_time": 6.742808103561401
    },
    {
      "epoch": 6.715965270996093e-05,
      "model_forward_time": 1.7030949592590332,
      "step": 70422
    },
    {
      "epoch": 6.715965270996093e-05,
      "step": 70422,
      "training_step_time": 6.754677057266235
    },
    {
      "epoch": 6.716060638427735e-05,
      "model_forward_time": 1.317781925201416,
      "step": 70423
    },
    {
      "epoch": 6.716060638427735e-05,
      "step": 70423,
      "training_step_time": 6.831049680709839
    },
    {
      "epoch": 6.716156005859376e-05,
      "model_forward_time": 1.7046515941619873,
      "step": 70424
    },
    {
      "epoch": 6.716156005859376e-05,
      "step": 70424,
      "training_step_time": 6.714529991149902
    },
    {
      "epoch": 6.716251373291015e-05,
      "model_forward_time": 1.7067451477050781,
      "step": 70425
    },
    {
      "epoch": 6.716251373291015e-05,
      "step": 70425,
      "training_step_time": 6.746009826660156
    },
    {
      "epoch": 6.716346740722656e-05,
      "model_forward_time": 2.25649094581604,
      "step": 70426
    },
    {
      "epoch": 6.716346740722656e-05,
      "step": 70426,
      "training_step_time": 8.667008399963379
    },
    {
      "epoch": 6.716442108154298e-05,
      "model_forward_time": 1.6967179775238037,
      "step": 70427
    },
    {
      "epoch": 6.716442108154298e-05,
      "step": 70427,
      "training_step_time": 7.703054666519165
    },
    {
      "epoch": 6.716537475585937e-05,
      "model_forward_time": 1.7016949653625488,
      "step": 70428
    },
    {
      "epoch": 6.716537475585937e-05,
      "step": 70428,
      "training_step_time": 6.725897789001465
    },
    {
      "epoch": 6.716632843017578e-05,
      "model_forward_time": 1.703526496887207,
      "step": 70429
    },
    {
      "epoch": 6.716632843017578e-05,
      "step": 70429,
      "training_step_time": 6.805901050567627
    },
    {
      "epoch": 6.71672821044922e-05,
      "grad_norm": 0.07342994958162308,
      "learning_rate": 2.206152954067024e-06,
      "loss": 0.0576,
      "step": 70430
    },
    {
      "epoch": 6.71672821044922e-05,
      "model_forward_time": 1.7021129131317139,
      "step": 70430
    },
    {
      "dynamics_loss_avg": 0.05572628546506166,
      "epoch": 6.71672821044922e-05,
      "step": 70430
    },
    {
      "action_loss_avg": 0.0012574919790495187,
      "epoch": 6.71672821044922e-05,
      "step": 70430
    },
    {
      "epoch": 6.71672821044922e-05,
      "step": 70430,
      "training_step_time": 6.767066955566406
    },
    {
      "epoch": 6.716823577880859e-05,
      "model_forward_time": 1.7120118141174316,
      "step": 70431
    },
    {
      "epoch": 6.716823577880859e-05,
      "step": 70431,
      "training_step_time": 6.771048545837402
    },
    {
      "epoch": 6.7169189453125e-05,
      "model_forward_time": 1.7188606262207031,
      "step": 70432
    },
    {
      "epoch": 6.7169189453125e-05,
      "step": 70432,
      "training_step_time": 6.813809871673584
    },
    {
      "epoch": 6.717014312744141e-05,
      "model_forward_time": 1.7090840339660645,
      "step": 70433
    },
    {
      "epoch": 6.717014312744141e-05,
      "step": 70433,
      "training_step_time": 6.821310520172119
    },
    {
      "epoch": 6.717109680175781e-05,
      "model_forward_time": 1.7076809406280518,
      "step": 70434
    },
    {
      "epoch": 6.717109680175781e-05,
      "step": 70434,
      "training_step_time": 6.817943811416626
    },
    {
      "epoch": 6.717205047607422e-05,
      "model_forward_time": 2.244905710220337,
      "step": 70435
    },
    {
      "epoch": 6.717205047607422e-05,
      "step": 70435,
      "training_step_time": 9.780693054199219
    },
    {
      "epoch": 6.717300415039063e-05,
      "model_forward_time": 1.7016453742980957,
      "step": 70436
    },
    {
      "epoch": 6.717300415039063e-05,
      "step": 70436,
      "training_step_time": 8.816095352172852
    },
    {
      "epoch": 6.717395782470703e-05,
      "model_forward_time": 1.7064013481140137,
      "step": 70437
    },
    {
      "epoch": 6.717395782470703e-05,
      "step": 70437,
      "training_step_time": 7.5056867599487305
    },
    {
      "epoch": 6.717491149902344e-05,
      "model_forward_time": 1.7087178230285645,
      "step": 70438
    },
    {
      "epoch": 6.717491149902344e-05,
      "step": 70438,
      "training_step_time": 6.789961576461792
    },
    {
      "epoch": 6.717586517333984e-05,
      "model_forward_time": 1.6975338459014893,
      "step": 70439
    },
    {
      "epoch": 6.717586517333984e-05,
      "step": 70439,
      "training_step_time": 6.776285171508789
    },
    {
      "epoch": 6.717681884765625e-05,
      "grad_norm": 0.11468134075403214,
      "learning_rate": 2.204781846199176e-06,
      "loss": 0.0572,
      "step": 70440
    },
    {
      "epoch": 6.717681884765625e-05,
      "model_forward_time": 1.6968975067138672,
      "step": 70440
    },
    {
      "dynamics_loss_avg": 0.06086561102420092,
      "epoch": 6.717681884765625e-05,
      "step": 70440
    },
    {
      "action_loss_avg": 0.0014145189546979965,
      "epoch": 6.717681884765625e-05,
      "step": 70440
    },
    {
      "epoch": 6.717681884765625e-05,
      "step": 70440,
      "training_step_time": 6.794931411743164
    },
    {
      "epoch": 6.717777252197266e-05,
      "model_forward_time": 1.701037883758545,
      "step": 70441
    },
    {
      "epoch": 6.717777252197266e-05,
      "step": 70441,
      "training_step_time": 6.764341354370117
    },
    {
      "epoch": 6.717872619628906e-05,
      "model_forward_time": 1.7038657665252686,
      "step": 70442
    },
    {
      "epoch": 6.717872619628906e-05,
      "step": 70442,
      "training_step_time": 6.771201133728027
    },
    {
      "epoch": 6.717967987060547e-05,
      "model_forward_time": 1.6952171325683594,
      "step": 70443
    },
    {
      "epoch": 6.717967987060547e-05,
      "step": 70443,
      "training_step_time": 6.752793788909912
    },
    {
      "epoch": 6.718063354492188e-05,
      "model_forward_time": 2.2575948238372803,
      "step": 70444
    },
    {
      "epoch": 6.718063354492188e-05,
      "step": 70444,
      "training_step_time": 8.748337984085083
    },
    {
      "epoch": 6.718158721923827e-05,
      "model_forward_time": 1.3267853260040283,
      "step": 70445
    },
    {
      "epoch": 6.718158721923827e-05,
      "step": 70445,
      "training_step_time": 8.678159236907959
    },
    {
      "epoch": 6.718254089355469e-05,
      "model_forward_time": 1.7109458446502686,
      "step": 70446
    },
    {
      "epoch": 6.718254089355469e-05,
      "step": 70446,
      "training_step_time": 7.6981446743011475
    },
    {
      "epoch": 6.71834945678711e-05,
      "model_forward_time": 1.703298568725586,
      "step": 70447
    },
    {
      "epoch": 6.71834945678711e-05,
      "step": 70447,
      "training_step_time": 6.899121046066284
    },
    {
      "epoch": 6.71844482421875e-05,
      "model_forward_time": 1.3269867897033691,
      "step": 70448
    },
    {
      "epoch": 6.71844482421875e-05,
      "step": 70448,
      "training_step_time": 6.867736339569092
    },
    {
      "epoch": 6.71854019165039e-05,
      "model_forward_time": 1.6999177932739258,
      "step": 70449
    },
    {
      "epoch": 6.71854019165039e-05,
      "step": 70449,
      "training_step_time": 6.856638431549072
    },
    {
      "epoch": 6.718635559082032e-05,
      "grad_norm": 0.10299819707870483,
      "learning_rate": 2.2034110440121865e-06,
      "loss": 0.0569,
      "step": 70450
    },
    {
      "epoch": 6.718635559082032e-05,
      "model_forward_time": 1.7108054161071777,
      "step": 70450
    },
    {
      "dynamics_loss_avg": 0.05793197862803936,
      "epoch": 6.718635559082032e-05,
      "step": 70450
    },
    {
      "action_loss_avg": 0.001371931645553559,
      "epoch": 6.718635559082032e-05,
      "step": 70450
    },
    {
      "epoch": 6.718635559082032e-05,
      "step": 70450,
      "training_step_time": 6.869342803955078
    },
    {
      "epoch": 6.718730926513671e-05,
      "model_forward_time": 1.6946732997894287,
      "step": 70451
    },
    {
      "epoch": 6.718730926513671e-05,
      "step": 70451,
      "training_step_time": 6.81458044052124
    },
    {
      "epoch": 6.718826293945312e-05,
      "model_forward_time": 1.72135591506958,
      "step": 70452
    },
    {
      "epoch": 6.718826293945312e-05,
      "step": 70452,
      "training_step_time": 6.741273880004883
    },
    {
      "epoch": 6.718921661376953e-05,
      "model_forward_time": 1.7232232093811035,
      "step": 70453
    },
    {
      "epoch": 6.718921661376953e-05,
      "step": 70453,
      "training_step_time": 8.640360832214355
    },
    {
      "epoch": 6.719017028808593e-05,
      "model_forward_time": 2.2595696449279785,
      "step": 70454
    },
    {
      "epoch": 6.719017028808593e-05,
      "step": 70454,
      "training_step_time": 9.94041919708252
    },
    {
      "epoch": 6.719112396240234e-05,
      "model_forward_time": 1.6991279125213623,
      "step": 70455
    },
    {
      "epoch": 6.719112396240234e-05,
      "step": 70455,
      "training_step_time": 7.661729335784912
    },
    {
      "epoch": 6.719207763671875e-05,
      "model_forward_time": 1.7059593200683594,
      "step": 70456
    },
    {
      "epoch": 6.719207763671875e-05,
      "step": 70456,
      "training_step_time": 7.426248788833618
    },
    {
      "epoch": 6.719303131103515e-05,
      "model_forward_time": 1.6954646110534668,
      "step": 70457
    },
    {
      "epoch": 6.719303131103515e-05,
      "step": 70457,
      "training_step_time": 6.722003936767578
    },
    {
      "epoch": 6.719398498535156e-05,
      "model_forward_time": 1.7018482685089111,
      "step": 70458
    },
    {
      "epoch": 6.719398498535156e-05,
      "step": 70458,
      "training_step_time": 6.754986763000488
    },
    {
      "epoch": 6.719493865966797e-05,
      "model_forward_time": 1.699774980545044,
      "step": 70459
    },
    {
      "epoch": 6.719493865966797e-05,
      "step": 70459,
      "training_step_time": 6.737446546554565
    },
    {
      "epoch": 6.719589233398437e-05,
      "grad_norm": 0.11057339608669281,
      "learning_rate": 2.2020405476559585e-06,
      "loss": 0.0559,
      "step": 70460
    },
    {
      "epoch": 6.719589233398437e-05,
      "model_forward_time": 1.702364444732666,
      "step": 70460
    },
    {
      "dynamics_loss_avg": 0.06880658380687237,
      "epoch": 6.719589233398437e-05,
      "step": 70460
    },
    {
      "action_loss_avg": 0.0016464976710267366,
      "epoch": 6.719589233398437e-05,
      "step": 70460
    },
    {
      "epoch": 6.719589233398437e-05,
      "step": 70460,
      "training_step_time": 6.741269588470459
    },
    {
      "epoch": 6.719684600830078e-05,
      "model_forward_time": 1.703782081604004,
      "step": 70461
    },
    {
      "epoch": 6.719684600830078e-05,
      "step": 70461,
      "training_step_time": 6.745775461196899
    },
    {
      "epoch": 6.719779968261719e-05,
      "model_forward_time": 1.7255430221557617,
      "step": 70462
    },
    {
      "epoch": 6.719779968261719e-05,
      "step": 70462,
      "training_step_time": 9.253175258636475
    },
    {
      "epoch": 6.719875335693359e-05,
      "model_forward_time": 1.7936851978302002,
      "step": 70463
    },
    {
      "epoch": 6.719875335693359e-05,
      "step": 70463,
      "training_step_time": 10.390787601470947
    },
    {
      "epoch": 6.719970703125e-05,
      "model_forward_time": 1.70489501953125,
      "step": 70464
    },
    {
      "epoch": 6.719970703125e-05,
      "step": 70464,
      "training_step_time": 7.720895051956177
    },
    {
      "epoch": 6.720066070556641e-05,
      "model_forward_time": 1.7057685852050781,
      "step": 70465
    },
    {
      "epoch": 6.720066070556641e-05,
      "step": 70465,
      "training_step_time": 7.594012498855591
    },
    {
      "epoch": 6.72016143798828e-05,
      "model_forward_time": 1.7104895114898682,
      "step": 70466
    },
    {
      "epoch": 6.72016143798828e-05,
      "step": 70466,
      "training_step_time": 6.7048420906066895
    },
    {
      "epoch": 6.720256805419922e-05,
      "model_forward_time": 1.7066140174865723,
      "step": 70467
    },
    {
      "epoch": 6.720256805419922e-05,
      "step": 70467,
      "training_step_time": 6.777743339538574
    },
    {
      "epoch": 6.720352172851563e-05,
      "model_forward_time": 1.6997554302215576,
      "step": 70468
    },
    {
      "epoch": 6.720352172851563e-05,
      "step": 70468,
      "training_step_time": 6.744666337966919
    },
    {
      "epoch": 6.720447540283203e-05,
      "model_forward_time": 1.708289623260498,
      "step": 70469
    },
    {
      "epoch": 6.720447540283203e-05,
      "step": 70469,
      "training_step_time": 6.717806816101074
    },
    {
      "epoch": 6.720542907714844e-05,
      "grad_norm": 0.07396838814020157,
      "learning_rate": 2.2006703572803725e-06,
      "loss": 0.0577,
      "step": 70470
    },
    {
      "epoch": 6.720542907714844e-05,
      "model_forward_time": 1.6933953762054443,
      "step": 70470
    },
    {
      "dynamics_loss_avg": 0.0589358052238822,
      "epoch": 6.720542907714844e-05,
      "step": 70470
    },
    {
      "action_loss_avg": 0.0015891121816821395,
      "epoch": 6.720542907714844e-05,
      "step": 70470
    },
    {
      "epoch": 6.720542907714844e-05,
      "step": 70470,
      "training_step_time": 6.7457311153411865
    },
    {
      "epoch": 6.720638275146485e-05,
      "model_forward_time": 1.700568675994873,
      "step": 70471
    },
    {
      "epoch": 6.720638275146485e-05,
      "step": 70471,
      "training_step_time": 6.698820114135742
    },
    {
      "epoch": 6.720733642578124e-05,
      "model_forward_time": 2.249156951904297,
      "step": 70472
    },
    {
      "epoch": 6.720733642578124e-05,
      "step": 70472,
      "training_step_time": 8.469334363937378
    },
    {
      "epoch": 6.720829010009766e-05,
      "model_forward_time": 1.6998014450073242,
      "step": 70473
    },
    {
      "epoch": 6.720829010009766e-05,
      "step": 70473,
      "training_step_time": 7.764130592346191
    },
    {
      "epoch": 6.720924377441407e-05,
      "model_forward_time": 1.7007102966308594,
      "step": 70474
    },
    {
      "epoch": 6.720924377441407e-05,
      "step": 70474,
      "training_step_time": 7.660639524459839
    },
    {
      "epoch": 6.721019744873046e-05,
      "model_forward_time": 1.706218957901001,
      "step": 70475
    },
    {
      "epoch": 6.721019744873046e-05,
      "step": 70475,
      "training_step_time": 6.7755351066589355
    },
    {
      "epoch": 6.721115112304687e-05,
      "model_forward_time": 1.7077620029449463,
      "step": 70476
    },
    {
      "epoch": 6.721115112304687e-05,
      "step": 70476,
      "training_step_time": 6.780903100967407
    },
    {
      "epoch": 6.721210479736328e-05,
      "model_forward_time": 1.7050044536590576,
      "step": 70477
    },
    {
      "epoch": 6.721210479736328e-05,
      "step": 70477,
      "training_step_time": 6.839120864868164
    },
    {
      "epoch": 6.721305847167968e-05,
      "model_forward_time": 1.6871466636657715,
      "step": 70478
    },
    {
      "epoch": 6.721305847167968e-05,
      "step": 70478,
      "training_step_time": 6.808819055557251
    },
    {
      "epoch": 6.721401214599609e-05,
      "model_forward_time": 1.703819751739502,
      "step": 70479
    },
    {
      "epoch": 6.721401214599609e-05,
      "step": 70479,
      "training_step_time": 6.779956579208374
    },
    {
      "epoch": 6.72149658203125e-05,
      "grad_norm": 0.07711105793714523,
      "learning_rate": 2.1993004730352647e-06,
      "loss": 0.0582,
      "step": 70480
    },
    {
      "epoch": 6.72149658203125e-05,
      "model_forward_time": 1.7011933326721191,
      "step": 70480
    },
    {
      "dynamics_loss_avg": 0.04528174679726362,
      "epoch": 6.72149658203125e-05,
      "step": 70480
    },
    {
      "action_loss_avg": 0.0008278317633084953,
      "epoch": 6.72149658203125e-05,
      "step": 70480
    },
    {
      "epoch": 6.72149658203125e-05,
      "step": 70480,
      "training_step_time": 7.654756546020508
    },
    {
      "epoch": 6.72159194946289e-05,
      "model_forward_time": 2.2803385257720947,
      "step": 70481
    },
    {
      "epoch": 6.72159194946289e-05,
      "step": 70481,
      "training_step_time": 9.437479257583618
    },
    {
      "epoch": 6.721687316894531e-05,
      "model_forward_time": 1.7042052745819092,
      "step": 70482
    },
    {
      "epoch": 6.721687316894531e-05,
      "step": 70482,
      "training_step_time": 7.987451076507568
    },
    {
      "epoch": 6.721782684326172e-05,
      "model_forward_time": 1.6957871913909912,
      "step": 70483
    },
    {
      "epoch": 6.721782684326172e-05,
      "step": 70483,
      "training_step_time": 7.691180467605591
    },
    {
      "epoch": 6.721878051757812e-05,
      "model_forward_time": 1.706571102142334,
      "step": 70484
    },
    {
      "epoch": 6.721878051757812e-05,
      "step": 70484,
      "training_step_time": 7.51732873916626
    },
    {
      "epoch": 6.721973419189453e-05,
      "model_forward_time": 1.7085959911346436,
      "step": 70485
    },
    {
      "epoch": 6.721973419189453e-05,
      "step": 70485,
      "training_step_time": 6.823130369186401
    },
    {
      "epoch": 6.722068786621094e-05,
      "model_forward_time": 1.7004470825195312,
      "step": 70486
    },
    {
      "epoch": 6.722068786621094e-05,
      "step": 70486,
      "training_step_time": 6.793666362762451
    },
    {
      "epoch": 6.722164154052734e-05,
      "model_forward_time": 1.7080552577972412,
      "step": 70487
    },
    {
      "epoch": 6.722164154052734e-05,
      "step": 70487,
      "training_step_time": 6.8186681270599365
    },
    {
      "epoch": 6.722259521484375e-05,
      "model_forward_time": 1.700122356414795,
      "step": 70488
    },
    {
      "epoch": 6.722259521484375e-05,
      "step": 70488,
      "training_step_time": 6.816365718841553
    },
    {
      "epoch": 6.722354888916016e-05,
      "model_forward_time": 1.7007639408111572,
      "step": 70489
    },
    {
      "epoch": 6.722354888916016e-05,
      "step": 70489,
      "training_step_time": 6.832349538803101
    },
    {
      "epoch": 6.722450256347656e-05,
      "grad_norm": 0.11915377527475357,
      "learning_rate": 2.1979308950704466e-06,
      "loss": 0.0533,
      "step": 70490
    },
    {
      "epoch": 6.722450256347656e-05,
      "model_forward_time": 2.315617084503174,
      "step": 70490
    },
    {
      "dynamics_loss_avg": 0.055460446327924726,
      "epoch": 6.722450256347656e-05,
      "step": 70490
    },
    {
      "action_loss_avg": 0.0009191543736960739,
      "epoch": 6.722450256347656e-05,
      "step": 70490
    },
    {
      "epoch": 6.722450256347656e-05,
      "step": 70490,
      "training_step_time": 8.641613960266113
    },
    {
      "epoch": 6.722545623779297e-05,
      "model_forward_time": 1.7025361061096191,
      "step": 70491
    },
    {
      "epoch": 6.722545623779297e-05,
      "step": 70491,
      "training_step_time": 9.766511917114258
    },
    {
      "epoch": 6.722640991210938e-05,
      "model_forward_time": 1.7031631469726562,
      "step": 70492
    },
    {
      "epoch": 6.722640991210938e-05,
      "step": 70492,
      "training_step_time": 8.457197189331055
    },
    {
      "epoch": 6.722736358642578e-05,
      "model_forward_time": 1.6897764205932617,
      "step": 70493
    },
    {
      "epoch": 6.722736358642578e-05,
      "step": 70493,
      "training_step_time": 7.6060426235198975
    },
    {
      "epoch": 6.722831726074219e-05,
      "model_forward_time": 1.697887659072876,
      "step": 70494
    },
    {
      "epoch": 6.722831726074219e-05,
      "step": 70494,
      "training_step_time": 6.819709062576294
    },
    {
      "epoch": 6.72292709350586e-05,
      "model_forward_time": 1.3244683742523193,
      "step": 70495
    },
    {
      "epoch": 6.72292709350586e-05,
      "step": 70495,
      "training_step_time": 6.796708106994629
    },
    {
      "epoch": 6.7230224609375e-05,
      "model_forward_time": 1.320512294769287,
      "step": 70496
    },
    {
      "epoch": 6.7230224609375e-05,
      "step": 70496,
      "training_step_time": 6.832229375839233
    },
    {
      "epoch": 6.72311782836914e-05,
      "model_forward_time": 1.7094166278839111,
      "step": 70497
    },
    {
      "epoch": 6.72311782836914e-05,
      "step": 70497,
      "training_step_time": 6.8380866050720215
    },
    {
      "epoch": 6.723213195800782e-05,
      "model_forward_time": 1.6769626140594482,
      "step": 70498
    },
    {
      "epoch": 6.723213195800782e-05,
      "step": 70498,
      "training_step_time": 6.736583948135376
    },
    {
      "epoch": 6.723308563232421e-05,
      "model_forward_time": 2.2711832523345947,
      "step": 70499
    },
    {
      "epoch": 6.723308563232421e-05,
      "step": 70499,
      "training_step_time": 8.533175230026245
    },
    {
      "epoch": 6.723403930664062e-05,
      "grad_norm": 0.12012505531311035,
      "learning_rate": 2.1965616235356925e-06,
      "loss": 0.0553,
      "step": 70500
    },
    {
      "epoch": 6.723403930664062e-05,
      "model_forward_time": 1.6968705654144287,
      "step": 70500
    },
    {
      "dynamics_loss_avg": 0.06124883145093918,
      "epoch": 6.723403930664062e-05,
      "step": 70500
    },
    {
      "action_loss_avg": 0.0025214949913788588,
      "epoch": 6.723403930664062e-05,
      "step": 70500
    },
    {
      "epoch": 6.723403930664062e-05,
      "step": 70500,
      "training_step_time": 8.368708372116089
    },
    {
      "epoch": 6.723499298095704e-05,
      "model_forward_time": 1.697237253189087,
      "step": 70501
    },
    {
      "epoch": 6.723499298095704e-05,
      "step": 70501,
      "training_step_time": 7.696566581726074
    },
    {
      "epoch": 6.723594665527343e-05,
      "model_forward_time": 1.3350086212158203,
      "step": 70502
    },
    {
      "epoch": 6.723594665527343e-05,
      "step": 70502,
      "training_step_time": 7.691246509552002
    },
    {
      "epoch": 6.723690032958984e-05,
      "model_forward_time": 1.7174365520477295,
      "step": 70503
    },
    {
      "epoch": 6.723690032958984e-05,
      "step": 70503,
      "training_step_time": 6.822658538818359
    },
    {
      "epoch": 6.723785400390625e-05,
      "model_forward_time": 1.7038118839263916,
      "step": 70504
    },
    {
      "epoch": 6.723785400390625e-05,
      "step": 70504,
      "training_step_time": 6.840535640716553
    },
    {
      "epoch": 6.723880767822265e-05,
      "model_forward_time": 1.7052786350250244,
      "step": 70505
    },
    {
      "epoch": 6.723880767822265e-05,
      "step": 70505,
      "training_step_time": 6.843527555465698
    },
    {
      "epoch": 6.723976135253906e-05,
      "model_forward_time": 1.7007734775543213,
      "step": 70506
    },
    {
      "epoch": 6.723976135253906e-05,
      "step": 70506,
      "training_step_time": 6.805832386016846
    },
    {
      "epoch": 6.724071502685547e-05,
      "model_forward_time": 1.337214469909668,
      "step": 70507
    },
    {
      "epoch": 6.724071502685547e-05,
      "step": 70507,
      "training_step_time": 6.800500154495239
    },
    {
      "epoch": 6.724166870117187e-05,
      "model_forward_time": 1.7092669010162354,
      "step": 70508
    },
    {
      "epoch": 6.724166870117187e-05,
      "step": 70508,
      "training_step_time": 8.388561725616455
    },
    {
      "epoch": 6.724262237548828e-05,
      "model_forward_time": 2.297618865966797,
      "step": 70509
    },
    {
      "epoch": 6.724262237548828e-05,
      "step": 70509,
      "training_step_time": 8.491141319274902
    },
    {
      "epoch": 6.724357604980469e-05,
      "grad_norm": 0.11751730740070343,
      "learning_rate": 2.1951926585807466e-06,
      "loss": 0.0541,
      "step": 70510
    },
    {
      "epoch": 6.724357604980469e-05,
      "model_forward_time": 1.7055916786193848,
      "step": 70510
    },
    {
      "dynamics_loss_avg": 0.0527438934892416,
      "epoch": 6.724357604980469e-05,
      "step": 70510
    },
    {
      "action_loss_avg": 0.0007261526843649335,
      "epoch": 6.724357604980469e-05,
      "step": 70510
    },
    {
      "epoch": 6.724357604980469e-05,
      "step": 70510,
      "training_step_time": 8.665921926498413
    },
    {
      "epoch": 6.724452972412109e-05,
      "model_forward_time": 1.712402582168579,
      "step": 70511
    },
    {
      "epoch": 6.724452972412109e-05,
      "step": 70511,
      "training_step_time": 7.703664779663086
    },
    {
      "epoch": 6.72454833984375e-05,
      "model_forward_time": 1.7145740985870361,
      "step": 70512
    },
    {
      "epoch": 6.72454833984375e-05,
      "step": 70512,
      "training_step_time": 7.49473237991333
    },
    {
      "epoch": 6.724643707275391e-05,
      "model_forward_time": 1.7111456394195557,
      "step": 70513
    },
    {
      "epoch": 6.724643707275391e-05,
      "step": 70513,
      "training_step_time": 6.774469614028931
    },
    {
      "epoch": 6.724739074707031e-05,
      "model_forward_time": 1.7187485694885254,
      "step": 70514
    },
    {
      "epoch": 6.724739074707031e-05,
      "step": 70514,
      "training_step_time": 6.784278392791748
    },
    {
      "epoch": 6.724834442138672e-05,
      "model_forward_time": 1.6988890171051025,
      "step": 70515
    },
    {
      "epoch": 6.724834442138672e-05,
      "step": 70515,
      "training_step_time": 6.68360161781311
    },
    {
      "epoch": 6.724929809570313e-05,
      "model_forward_time": 1.7155287265777588,
      "step": 70516
    },
    {
      "epoch": 6.724929809570313e-05,
      "step": 70516,
      "training_step_time": 6.793927431106567
    },
    {
      "epoch": 6.725025177001953e-05,
      "model_forward_time": 1.7187738418579102,
      "step": 70517
    },
    {
      "epoch": 6.725025177001953e-05,
      "step": 70517,
      "training_step_time": 7.796074628829956
    },
    {
      "epoch": 6.725120544433594e-05,
      "model_forward_time": 2.3445491790771484,
      "step": 70518
    },
    {
      "epoch": 6.725120544433594e-05,
      "step": 70518,
      "training_step_time": 11.76897644996643
    },
    {
      "epoch": 6.725215911865235e-05,
      "model_forward_time": 1.72385573387146,
      "step": 70519
    },
    {
      "epoch": 6.725215911865235e-05,
      "step": 70519,
      "training_step_time": 9.187793493270874
    },
    {
      "epoch": 6.725311279296875e-05,
      "grad_norm": 0.14430482685565948,
      "learning_rate": 2.193824000355313e-06,
      "loss": 0.0542,
      "step": 70520
    },
    {
      "epoch": 6.725311279296875e-05,
      "model_forward_time": 1.7205452919006348,
      "step": 70520
    },
    {
      "dynamics_loss_avg": 0.037455704808235166,
      "epoch": 6.725311279296875e-05,
      "step": 70520
    },
    {
      "action_loss_avg": 0.0009305345418397337,
      "epoch": 6.725311279296875e-05,
      "step": 70520
    },
    {
      "epoch": 6.725311279296875e-05,
      "step": 70520,
      "training_step_time": 7.68265700340271
    },
    {
      "epoch": 6.725406646728516e-05,
      "model_forward_time": 1.7144629955291748,
      "step": 70521
    },
    {
      "epoch": 6.725406646728516e-05,
      "step": 70521,
      "training_step_time": 7.504646301269531
    },
    {
      "epoch": 6.725502014160157e-05,
      "model_forward_time": 1.7267301082611084,
      "step": 70522
    },
    {
      "epoch": 6.725502014160157e-05,
      "step": 70522,
      "training_step_time": 6.75166654586792
    },
    {
      "epoch": 6.725597381591797e-05,
      "model_forward_time": 1.722949743270874,
      "step": 70523
    },
    {
      "epoch": 6.725597381591797e-05,
      "step": 70523,
      "training_step_time": 6.820578575134277
    },
    {
      "epoch": 6.725692749023438e-05,
      "model_forward_time": 1.7211501598358154,
      "step": 70524
    },
    {
      "epoch": 6.725692749023438e-05,
      "step": 70524,
      "training_step_time": 6.786790609359741
    },
    {
      "epoch": 6.725788116455079e-05,
      "model_forward_time": 1.7114300727844238,
      "step": 70525
    },
    {
      "epoch": 6.725788116455079e-05,
      "step": 70525,
      "training_step_time": 6.745822906494141
    },
    {
      "epoch": 6.725883483886718e-05,
      "model_forward_time": 1.716813087463379,
      "step": 70526
    },
    {
      "epoch": 6.725883483886718e-05,
      "step": 70526,
      "training_step_time": 7.688122034072876
    },
    {
      "epoch": 6.72597885131836e-05,
      "model_forward_time": 2.3252925872802734,
      "step": 70527
    },
    {
      "epoch": 6.72597885131836e-05,
      "step": 70527,
      "training_step_time": 11.241304874420166
    },
    {
      "epoch": 6.72607421875e-05,
      "model_forward_time": 1.7276229858398438,
      "step": 70528
    },
    {
      "epoch": 6.72607421875e-05,
      "step": 70528,
      "training_step_time": 7.789381742477417
    },
    {
      "epoch": 6.72616958618164e-05,
      "model_forward_time": 1.724503517150879,
      "step": 70529
    },
    {
      "epoch": 6.72616958618164e-05,
      "step": 70529,
      "training_step_time": 7.719108819961548
    },
    {
      "epoch": 6.726264953613281e-05,
      "grad_norm": 0.12383459508419037,
      "learning_rate": 2.192455649009067e-06,
      "loss": 0.0544,
      "step": 70530
    },
    {
      "epoch": 6.726264953613281e-05,
      "model_forward_time": 1.7142565250396729,
      "step": 70530
    },
    {
      "dynamics_loss_avg": 0.054314091056585315,
      "epoch": 6.726264953613281e-05,
      "step": 70530
    },
    {
      "action_loss_avg": 0.0011005358886905015,
      "epoch": 6.726264953613281e-05,
      "step": 70530
    },
    {
      "epoch": 6.726264953613281e-05,
      "step": 70530,
      "training_step_time": 7.695595026016235
    },
    {
      "epoch": 6.726360321044922e-05,
      "model_forward_time": 1.7288222312927246,
      "step": 70531
    },
    {
      "epoch": 6.726360321044922e-05,
      "step": 70531,
      "training_step_time": 6.754597902297974
    },
    {
      "epoch": 6.726455688476562e-05,
      "model_forward_time": 1.7398581504821777,
      "step": 70532
    },
    {
      "epoch": 6.726455688476562e-05,
      "step": 70532,
      "training_step_time": 6.757697582244873
    },
    {
      "epoch": 6.726551055908203e-05,
      "model_forward_time": 1.710728645324707,
      "step": 70533
    },
    {
      "epoch": 6.726551055908203e-05,
      "step": 70533,
      "training_step_time": 6.761392831802368
    },
    {
      "epoch": 6.726646423339844e-05,
      "model_forward_time": 1.7232446670532227,
      "step": 70534
    },
    {
      "epoch": 6.726646423339844e-05,
      "step": 70534,
      "training_step_time": 6.764127016067505
    },
    {
      "epoch": 6.726741790771484e-05,
      "model_forward_time": 1.7238647937774658,
      "step": 70535
    },
    {
      "epoch": 6.726741790771484e-05,
      "step": 70535,
      "training_step_time": 7.755072832107544
    },
    {
      "epoch": 6.726837158203125e-05,
      "model_forward_time": 1.7491819858551025,
      "step": 70536
    },
    {
      "epoch": 6.726837158203125e-05,
      "step": 70536,
      "training_step_time": 10.360922574996948
    },
    {
      "epoch": 6.726932525634766e-05,
      "model_forward_time": 1.7021734714508057,
      "step": 70537
    },
    {
      "epoch": 6.726932525634766e-05,
      "step": 70537,
      "training_step_time": 11.068798542022705
    },
    {
      "epoch": 6.727027893066406e-05,
      "model_forward_time": 1.7166755199432373,
      "step": 70538
    },
    {
      "epoch": 6.727027893066406e-05,
      "step": 70538,
      "training_step_time": 7.69660758972168
    },
    {
      "epoch": 6.727123260498047e-05,
      "model_forward_time": 1.7276229858398438,
      "step": 70539
    },
    {
      "epoch": 6.727123260498047e-05,
      "step": 70539,
      "training_step_time": 7.696518182754517
    },
    {
      "epoch": 6.727218627929688e-05,
      "grad_norm": 0.09176139533519745,
      "learning_rate": 2.1910876046916524e-06,
      "loss": 0.0565,
      "step": 70540
    },
    {
      "epoch": 6.727218627929688e-05,
      "model_forward_time": 1.7153944969177246,
      "step": 70540
    },
    {
      "dynamics_loss_avg": 0.06077792625874281,
      "epoch": 6.727218627929688e-05,
      "step": 70540
    },
    {
      "action_loss_avg": 0.0011669446481391787,
      "epoch": 6.727218627929688e-05,
      "step": 70540
    },
    {
      "epoch": 6.727218627929688e-05,
      "step": 70540,
      "training_step_time": 6.809571981430054
    },
    {
      "epoch": 6.727313995361328e-05,
      "model_forward_time": 1.7172164916992188,
      "step": 70541
    },
    {
      "epoch": 6.727313995361328e-05,
      "step": 70541,
      "training_step_time": 6.827303409576416
    },
    {
      "epoch": 6.727409362792969e-05,
      "model_forward_time": 1.72282075881958,
      "step": 70542
    },
    {
      "epoch": 6.727409362792969e-05,
      "step": 70542,
      "training_step_time": 6.745183229446411
    },
    {
      "epoch": 6.72750473022461e-05,
      "model_forward_time": 1.7130773067474365,
      "step": 70543
    },
    {
      "epoch": 6.72750473022461e-05,
      "step": 70543,
      "training_step_time": 6.751139163970947
    },
    {
      "epoch": 6.72760009765625e-05,
      "model_forward_time": 1.3496761322021484,
      "step": 70544
    },
    {
      "epoch": 6.72760009765625e-05,
      "step": 70544,
      "training_step_time": 7.728849172592163
    },
    {
      "epoch": 6.727695465087891e-05,
      "model_forward_time": 1.723501205444336,
      "step": 70545
    },
    {
      "epoch": 6.727695465087891e-05,
      "step": 70545,
      "training_step_time": 9.752845048904419
    },
    {
      "epoch": 6.727790832519532e-05,
      "model_forward_time": 2.286132335662842,
      "step": 70546
    },
    {
      "epoch": 6.727790832519532e-05,
      "step": 70546,
      "training_step_time": 10.356130123138428
    },
    {
      "epoch": 6.727886199951172e-05,
      "model_forward_time": 1.7207627296447754,
      "step": 70547
    },
    {
      "epoch": 6.727886199951172e-05,
      "step": 70547,
      "training_step_time": 8.75832724571228
    },
    {
      "epoch": 6.727981567382813e-05,
      "model_forward_time": 1.7023751735687256,
      "step": 70548
    },
    {
      "epoch": 6.727981567382813e-05,
      "step": 70548,
      "training_step_time": 7.692693471908569
    },
    {
      "epoch": 6.728076934814454e-05,
      "model_forward_time": 1.718790054321289,
      "step": 70549
    },
    {
      "epoch": 6.728076934814454e-05,
      "step": 70549,
      "training_step_time": 6.779803991317749
    },
    {
      "epoch": 6.728172302246093e-05,
      "grad_norm": 0.10175295174121857,
      "learning_rate": 2.1897198675526726e-06,
      "loss": 0.0572,
      "step": 70550
    },
    {
      "epoch": 6.728172302246093e-05,
      "model_forward_time": 1.7101271152496338,
      "step": 70550
    },
    {
      "dynamics_loss_avg": 0.06609097458422183,
      "epoch": 6.728172302246093e-05,
      "step": 70550
    },
    {
      "action_loss_avg": 0.0016110206954181195,
      "epoch": 6.728172302246093e-05,
      "step": 70550
    },
    {
      "epoch": 6.728172302246093e-05,
      "step": 70550,
      "training_step_time": 6.724901914596558
    },
    {
      "epoch": 6.728267669677735e-05,
      "model_forward_time": 1.733017921447754,
      "step": 70551
    },
    {
      "epoch": 6.728267669677735e-05,
      "step": 70551,
      "training_step_time": 6.7366392612457275
    },
    {
      "epoch": 6.728363037109376e-05,
      "model_forward_time": 1.71610426902771,
      "step": 70552
    },
    {
      "epoch": 6.728363037109376e-05,
      "step": 70552,
      "training_step_time": 6.780744314193726
    },
    {
      "epoch": 6.728458404541015e-05,
      "model_forward_time": 1.7152268886566162,
      "step": 70553
    },
    {
      "epoch": 6.728458404541015e-05,
      "step": 70553,
      "training_step_time": 7.72723650932312
    },
    {
      "epoch": 6.728553771972656e-05,
      "model_forward_time": 1.716191291809082,
      "step": 70554
    },
    {
      "epoch": 6.728553771972656e-05,
      "step": 70554,
      "training_step_time": 9.505707025527954
    },
    {
      "epoch": 6.728649139404298e-05,
      "model_forward_time": 2.309314727783203,
      "step": 70555
    },
    {
      "epoch": 6.728649139404298e-05,
      "step": 70555,
      "training_step_time": 10.79899001121521
    },
    {
      "epoch": 6.728744506835937e-05,
      "model_forward_time": 1.7267658710479736,
      "step": 70556
    },
    {
      "epoch": 6.728744506835937e-05,
      "step": 70556,
      "training_step_time": 10.041738033294678
    },
    {
      "epoch": 6.728839874267578e-05,
      "model_forward_time": 1.7153129577636719,
      "step": 70557
    },
    {
      "epoch": 6.728839874267578e-05,
      "step": 70557,
      "training_step_time": 7.749327898025513
    },
    {
      "epoch": 6.72893524169922e-05,
      "model_forward_time": 1.7178945541381836,
      "step": 70558
    },
    {
      "epoch": 6.72893524169922e-05,
      "step": 70558,
      "training_step_time": 7.49318265914917
    },
    {
      "epoch": 6.729030609130859e-05,
      "model_forward_time": 1.6999266147613525,
      "step": 70559
    },
    {
      "epoch": 6.729030609130859e-05,
      "step": 70559,
      "training_step_time": 6.786761522293091
    },
    {
      "epoch": 6.7291259765625e-05,
      "grad_norm": 0.10638733953237534,
      "learning_rate": 2.1883524377417025e-06,
      "loss": 0.0585,
      "step": 70560
    },
    {
      "epoch": 6.7291259765625e-05,
      "model_forward_time": 1.7177674770355225,
      "step": 70560
    },
    {
      "dynamics_loss_avg": 0.0657829198986292,
      "epoch": 6.7291259765625e-05,
      "step": 70560
    },
    {
      "action_loss_avg": 0.0017462569405324758,
      "epoch": 6.7291259765625e-05,
      "step": 70560
    },
    {
      "epoch": 6.7291259765625e-05,
      "step": 70560,
      "training_step_time": 6.8139495849609375
    },
    {
      "epoch": 6.729221343994141e-05,
      "model_forward_time": 1.341116189956665,
      "step": 70561
    },
    {
      "epoch": 6.729221343994141e-05,
      "step": 70561,
      "training_step_time": 6.7747642993927
    },
    {
      "epoch": 6.729316711425781e-05,
      "model_forward_time": 1.725174903869629,
      "step": 70562
    },
    {
      "epoch": 6.729316711425781e-05,
      "step": 70562,
      "training_step_time": 7.665660858154297
    },
    {
      "epoch": 6.729412078857422e-05,
      "model_forward_time": 1.7161097526550293,
      "step": 70563
    },
    {
      "epoch": 6.729412078857422e-05,
      "step": 70563,
      "training_step_time": 10.275260210037231
    },
    {
      "epoch": 6.729507446289062e-05,
      "model_forward_time": 2.311551094055176,
      "step": 70564
    },
    {
      "epoch": 6.729507446289062e-05,
      "step": 70564,
      "training_step_time": 9.147664785385132
    },
    {
      "epoch": 6.729602813720703e-05,
      "model_forward_time": 1.7167487144470215,
      "step": 70565
    },
    {
      "epoch": 6.729602813720703e-05,
      "step": 70565,
      "training_step_time": 7.819129228591919
    },
    {
      "epoch": 6.729698181152344e-05,
      "model_forward_time": 1.726790428161621,
      "step": 70566
    },
    {
      "epoch": 6.729698181152344e-05,
      "step": 70566,
      "training_step_time": 7.698759078979492
    },
    {
      "epoch": 6.729793548583984e-05,
      "model_forward_time": 1.7114453315734863,
      "step": 70567
    },
    {
      "epoch": 6.729793548583984e-05,
      "step": 70567,
      "training_step_time": 7.771308183670044
    },
    {
      "epoch": 6.729888916015625e-05,
      "model_forward_time": 1.7177300453186035,
      "step": 70568
    },
    {
      "epoch": 6.729888916015625e-05,
      "step": 70568,
      "training_step_time": 6.756723642349243
    },
    {
      "epoch": 6.729984283447266e-05,
      "model_forward_time": 1.715804100036621,
      "step": 70569
    },
    {
      "epoch": 6.729984283447266e-05,
      "step": 70569,
      "training_step_time": 6.7632598876953125
    },
    {
      "epoch": 6.730079650878906e-05,
      "grad_norm": 0.10271842777729034,
      "learning_rate": 2.1869853154082828e-06,
      "loss": 0.0598,
      "step": 70570
    },
    {
      "epoch": 6.730079650878906e-05,
      "model_forward_time": 1.7334530353546143,
      "step": 70570
    },
    {
      "dynamics_loss_avg": 0.041596013400703666,
      "epoch": 6.730079650878906e-05,
      "step": 70570
    },
    {
      "action_loss_avg": 0.0007952263404149562,
      "epoch": 6.730079650878906e-05,
      "step": 70570
    },
    {
      "epoch": 6.730079650878906e-05,
      "step": 70570,
      "training_step_time": 6.836151123046875
    },
    {
      "epoch": 6.730175018310547e-05,
      "model_forward_time": 1.7352137565612793,
      "step": 70571
    },
    {
      "epoch": 6.730175018310547e-05,
      "step": 70571,
      "training_step_time": 7.6555016040802
    },
    {
      "epoch": 6.730270385742188e-05,
      "model_forward_time": 1.7160637378692627,
      "step": 70572
    },
    {
      "epoch": 6.730270385742188e-05,
      "step": 70572,
      "training_step_time": 7.701673984527588
    },
    {
      "epoch": 6.730365753173827e-05,
      "model_forward_time": 1.3533756732940674,
      "step": 70573
    },
    {
      "epoch": 6.730365753173827e-05,
      "step": 70573,
      "training_step_time": 9.908108472824097
    },
    {
      "epoch": 6.730461120605469e-05,
      "model_forward_time": 2.2762014865875244,
      "step": 70574
    },
    {
      "epoch": 6.730461120605469e-05,
      "step": 70574,
      "training_step_time": 8.49278473854065
    },
    {
      "epoch": 6.73055648803711e-05,
      "model_forward_time": 1.7251853942871094,
      "step": 70575
    },
    {
      "epoch": 6.73055648803711e-05,
      "step": 70575,
      "training_step_time": 7.746136903762817
    },
    {
      "epoch": 6.73065185546875e-05,
      "model_forward_time": 1.7125253677368164,
      "step": 70576
    },
    {
      "epoch": 6.73065185546875e-05,
      "step": 70576,
      "training_step_time": 8.472277164459229
    },
    {
      "epoch": 6.73074722290039e-05,
      "model_forward_time": 1.7162604331970215,
      "step": 70577
    },
    {
      "epoch": 6.73074722290039e-05,
      "step": 70577,
      "training_step_time": 6.744284391403198
    },
    {
      "epoch": 6.730842590332032e-05,
      "model_forward_time": 1.7260456085205078,
      "step": 70578
    },
    {
      "epoch": 6.730842590332032e-05,
      "step": 70578,
      "training_step_time": 6.811220645904541
    },
    {
      "epoch": 6.730937957763671e-05,
      "model_forward_time": 1.7248561382293701,
      "step": 70579
    },
    {
      "epoch": 6.730937957763671e-05,
      "step": 70579,
      "training_step_time": 6.8199708461761475
    },
    {
      "epoch": 6.731033325195312e-05,
      "grad_norm": 0.0986165776848793,
      "learning_rate": 2.1856185007019215e-06,
      "loss": 0.0585,
      "step": 70580
    },
    {
      "epoch": 6.731033325195312e-05,
      "model_forward_time": 1.715653657913208,
      "step": 70580
    },
    {
      "dynamics_loss_avg": 0.05225133635103703,
      "epoch": 6.731033325195312e-05,
      "step": 70580
    },
    {
      "action_loss_avg": 0.0018328285426832736,
      "epoch": 6.731033325195312e-05,
      "step": 70580
    },
    {
      "epoch": 6.731033325195312e-05,
      "step": 70580,
      "training_step_time": 7.585457801818848
    },
    {
      "epoch": 6.731128692626953e-05,
      "model_forward_time": 1.7186250686645508,
      "step": 70581
    },
    {
      "epoch": 6.731128692626953e-05,
      "step": 70581,
      "training_step_time": 8.396407127380371
    },
    {
      "epoch": 6.731224060058593e-05,
      "model_forward_time": 1.73679780960083,
      "step": 70582
    },
    {
      "epoch": 6.731224060058593e-05,
      "step": 70582,
      "training_step_time": 8.353101015090942
    },
    {
      "epoch": 6.731319427490234e-05,
      "model_forward_time": 2.2756099700927734,
      "step": 70583
    },
    {
      "epoch": 6.731319427490234e-05,
      "step": 70583,
      "training_step_time": 7.936087369918823
    },
    {
      "epoch": 6.731414794921875e-05,
      "model_forward_time": 1.7167754173278809,
      "step": 70584
    },
    {
      "epoch": 6.731414794921875e-05,
      "step": 70584,
      "training_step_time": 7.751983404159546
    },
    {
      "epoch": 6.731510162353515e-05,
      "model_forward_time": 1.720240831375122,
      "step": 70585
    },
    {
      "epoch": 6.731510162353515e-05,
      "step": 70585,
      "training_step_time": 7.696889877319336
    },
    {
      "epoch": 6.731605529785156e-05,
      "model_forward_time": 1.7189459800720215,
      "step": 70586
    },
    {
      "epoch": 6.731605529785156e-05,
      "step": 70586,
      "training_step_time": 7.510751247406006
    },
    {
      "epoch": 6.731700897216797e-05,
      "model_forward_time": 1.340029239654541,
      "step": 70587
    },
    {
      "epoch": 6.731700897216797e-05,
      "step": 70587,
      "training_step_time": 6.7609052658081055
    },
    {
      "epoch": 6.731796264648437e-05,
      "model_forward_time": 1.7268669605255127,
      "step": 70588
    },
    {
      "epoch": 6.731796264648437e-05,
      "step": 70588,
      "training_step_time": 6.813322305679321
    },
    {
      "epoch": 6.731891632080078e-05,
      "model_forward_time": 1.7184782028198242,
      "step": 70589
    },
    {
      "epoch": 6.731891632080078e-05,
      "step": 70589,
      "training_step_time": 7.631645679473877
    },
    {
      "epoch": 6.731986999511719e-05,
      "grad_norm": 0.08363261818885803,
      "learning_rate": 2.1842519937720873e-06,
      "loss": 0.0574,
      "step": 70590
    },
    {
      "epoch": 6.731986999511719e-05,
      "model_forward_time": 1.7141990661621094,
      "step": 70590
    },
    {
      "dynamics_loss_avg": 0.05666674710810184,
      "epoch": 6.731986999511719e-05,
      "step": 70590
    },
    {
      "action_loss_avg": 0.0014832518063485622,
      "epoch": 6.731986999511719e-05,
      "step": 70590
    },
    {
      "epoch": 6.731986999511719e-05,
      "step": 70590,
      "training_step_time": 7.673527956008911
    },
    {
      "epoch": 6.732082366943359e-05,
      "model_forward_time": 1.7266175746917725,
      "step": 70591
    },
    {
      "epoch": 6.732082366943359e-05,
      "step": 70591,
      "training_step_time": 10.645817041397095
    },
    {
      "epoch": 6.732177734375e-05,
      "model_forward_time": 2.2747609615325928,
      "step": 70592
    },
    {
      "epoch": 6.732177734375e-05,
      "step": 70592,
      "training_step_time": 9.860054016113281
    },
    {
      "epoch": 6.732273101806641e-05,
      "model_forward_time": 1.7248880863189697,
      "step": 70593
    },
    {
      "epoch": 6.732273101806641e-05,
      "step": 70593,
      "training_step_time": 8.816444158554077
    },
    {
      "epoch": 6.732368469238281e-05,
      "model_forward_time": 1.7244024276733398,
      "step": 70594
    },
    {
      "epoch": 6.732368469238281e-05,
      "step": 70594,
      "training_step_time": 7.530134677886963
    },
    {
      "epoch": 6.732463836669922e-05,
      "model_forward_time": 1.7262351512908936,
      "step": 70595
    },
    {
      "epoch": 6.732463836669922e-05,
      "step": 70595,
      "training_step_time": 7.7589311599731445
    },
    {
      "epoch": 6.732559204101563e-05,
      "model_forward_time": 1.7122690677642822,
      "step": 70596
    },
    {
      "epoch": 6.732559204101563e-05,
      "step": 70596,
      "training_step_time": 6.743208169937134
    },
    {
      "epoch": 6.732654571533203e-05,
      "model_forward_time": 1.7188973426818848,
      "step": 70597
    },
    {
      "epoch": 6.732654571533203e-05,
      "step": 70597,
      "training_step_time": 6.768331050872803
    },
    {
      "epoch": 6.732749938964844e-05,
      "model_forward_time": 1.726109504699707,
      "step": 70598
    },
    {
      "epoch": 6.732749938964844e-05,
      "step": 70598,
      "training_step_time": 7.697791337966919
    },
    {
      "epoch": 6.732845306396485e-05,
      "model_forward_time": 1.7277400493621826,
      "step": 70599
    },
    {
      "epoch": 6.732845306396485e-05,
      "step": 70599,
      "training_step_time": 7.708970546722412
    },
    {
      "epoch": 6.732940673828124e-05,
      "grad_norm": 0.11855165660381317,
      "learning_rate": 2.182885794768224e-06,
      "loss": 0.0567,
      "step": 70600
    },
    {
      "epoch": 6.732940673828124e-05,
      "model_forward_time": 1.7241802215576172,
      "step": 70600
    },
    {
      "dynamics_loss_avg": 0.06534319519996643,
      "epoch": 6.732940673828124e-05,
      "step": 70600
    },
    {
      "action_loss_avg": 0.0018186645756941289,
      "epoch": 6.732940673828124e-05,
      "step": 70600
    },
    {
      "epoch": 6.732940673828124e-05,
      "step": 70600,
      "training_step_time": 8.959972381591797
    },
    {
      "epoch": 6.733036041259766e-05,
      "model_forward_time": 2.323242664337158,
      "step": 70601
    },
    {
      "epoch": 6.733036041259766e-05,
      "step": 70601,
      "training_step_time": 7.947856187820435
    },
    {
      "epoch": 6.733131408691407e-05,
      "model_forward_time": 1.7285845279693604,
      "step": 70602
    },
    {
      "epoch": 6.733131408691407e-05,
      "step": 70602,
      "training_step_time": 8.511372089385986
    },
    {
      "epoch": 6.733226776123046e-05,
      "model_forward_time": 1.717477560043335,
      "step": 70603
    },
    {
      "epoch": 6.733226776123046e-05,
      "step": 70603,
      "training_step_time": 7.685242414474487
    },
    {
      "epoch": 6.733322143554687e-05,
      "model_forward_time": 1.730034589767456,
      "step": 70604
    },
    {
      "epoch": 6.733322143554687e-05,
      "step": 70604,
      "training_step_time": 7.703259229660034
    },
    {
      "epoch": 6.733417510986329e-05,
      "model_forward_time": 1.7213311195373535,
      "step": 70605
    },
    {
      "epoch": 6.733417510986329e-05,
      "step": 70605,
      "training_step_time": 7.49370813369751
    },
    {
      "epoch": 6.733512878417968e-05,
      "model_forward_time": 1.7170934677124023,
      "step": 70606
    },
    {
      "epoch": 6.733512878417968e-05,
      "step": 70606,
      "training_step_time": 6.7855589389801025
    },
    {
      "epoch": 6.733608245849609e-05,
      "model_forward_time": 1.7330536842346191,
      "step": 70607
    },
    {
      "epoch": 6.733608245849609e-05,
      "step": 70607,
      "training_step_time": 7.687190294265747
    },
    {
      "epoch": 6.73370361328125e-05,
      "model_forward_time": 1.725431203842163,
      "step": 70608
    },
    {
      "epoch": 6.73370361328125e-05,
      "step": 70608,
      "training_step_time": 7.706395626068115
    },
    {
      "epoch": 6.73379898071289e-05,
      "model_forward_time": 1.724303960800171,
      "step": 70609
    },
    {
      "epoch": 6.73379898071289e-05,
      "step": 70609,
      "training_step_time": 10.554066896438599
    },
    {
      "epoch": 6.733894348144531e-05,
      "grad_norm": 0.12040416896343231,
      "learning_rate": 2.181519903839733e-06,
      "loss": 0.0586,
      "step": 70610
    },
    {
      "epoch": 6.733894348144531e-05,
      "model_forward_time": 1.7754173278808594,
      "step": 70610
    },
    {
      "dynamics_loss_avg": 0.05371963195502758,
      "epoch": 6.733894348144531e-05,
      "step": 70610
    },
    {
      "action_loss_avg": 0.001576717710122466,
      "epoch": 6.733894348144531e-05,
      "step": 70610
    },
    {
      "epoch": 6.733894348144531e-05,
      "step": 70610,
      "training_step_time": 8.80441951751709
    },
    {
      "epoch": 6.733989715576172e-05,
      "model_forward_time": 2.2926723957061768,
      "step": 70611
    },
    {
      "epoch": 6.733989715576172e-05,
      "step": 70611,
      "training_step_time": 9.77504825592041
    },
    {
      "epoch": 6.734085083007812e-05,
      "model_forward_time": 1.7232048511505127,
      "step": 70612
    },
    {
      "epoch": 6.734085083007812e-05,
      "step": 70612,
      "training_step_time": 7.780594825744629
    },
    {
      "epoch": 6.734180450439453e-05,
      "model_forward_time": 1.7209892272949219,
      "step": 70613
    },
    {
      "epoch": 6.734180450439453e-05,
      "step": 70613,
      "training_step_time": 7.753137111663818
    },
    {
      "epoch": 6.734275817871094e-05,
      "model_forward_time": 1.7236628532409668,
      "step": 70614
    },
    {
      "epoch": 6.734275817871094e-05,
      "step": 70614,
      "training_step_time": 7.555776834487915
    },
    {
      "epoch": 6.734371185302734e-05,
      "model_forward_time": 1.7313272953033447,
      "step": 70615
    },
    {
      "epoch": 6.734371185302734e-05,
      "step": 70615,
      "training_step_time": 6.8302435874938965
    },
    {
      "epoch": 6.734466552734375e-05,
      "model_forward_time": 1.722909688949585,
      "step": 70616
    },
    {
      "epoch": 6.734466552734375e-05,
      "step": 70616,
      "training_step_time": 7.583726406097412
    },
    {
      "epoch": 6.734561920166016e-05,
      "model_forward_time": 1.7303335666656494,
      "step": 70617
    },
    {
      "epoch": 6.734561920166016e-05,
      "step": 70617,
      "training_step_time": 7.681672811508179
    },
    {
      "epoch": 6.734657287597656e-05,
      "model_forward_time": 1.7175657749176025,
      "step": 70618
    },
    {
      "epoch": 6.734657287597656e-05,
      "step": 70618,
      "training_step_time": 7.74224328994751
    },
    {
      "epoch": 6.734752655029297e-05,
      "model_forward_time": 1.74983811378479,
      "step": 70619
    },
    {
      "epoch": 6.734752655029297e-05,
      "step": 70619,
      "training_step_time": 9.274676322937012
    },
    {
      "epoch": 6.734848022460938e-05,
      "grad_norm": 0.08710242807865143,
      "learning_rate": 2.1801543211359896e-06,
      "loss": 0.0599,
      "step": 70620
    },
    {
      "epoch": 6.734848022460938e-05,
      "model_forward_time": 2.275160312652588,
      "step": 70620
    },
    {
      "dynamics_loss_avg": 0.058336909115314486,
      "epoch": 6.734848022460938e-05,
      "step": 70620
    },
    {
      "action_loss_avg": 0.0010633049998432398,
      "epoch": 6.734848022460938e-05,
      "step": 70620
    },
    {
      "epoch": 6.734848022460938e-05,
      "step": 70620,
      "training_step_time": 8.503302097320557
    },
    {
      "epoch": 6.734943389892578e-05,
      "model_forward_time": 1.7250618934631348,
      "step": 70621
    },
    {
      "epoch": 6.734943389892578e-05,
      "step": 70621,
      "training_step_time": 7.796515703201294
    },
    {
      "epoch": 6.735038757324219e-05,
      "model_forward_time": 1.725135326385498,
      "step": 70622
    },
    {
      "epoch": 6.735038757324219e-05,
      "step": 70622,
      "training_step_time": 7.804553031921387
    },
    {
      "epoch": 6.73513412475586e-05,
      "model_forward_time": 1.728423833847046,
      "step": 70623
    },
    {
      "epoch": 6.73513412475586e-05,
      "step": 70623,
      "training_step_time": 7.5521955490112305
    },
    {
      "epoch": 6.7352294921875e-05,
      "model_forward_time": 1.724215030670166,
      "step": 70624
    },
    {
      "epoch": 6.7352294921875e-05,
      "step": 70624,
      "training_step_time": 6.830115079879761
    },
    {
      "epoch": 6.73532485961914e-05,
      "model_forward_time": 1.738234519958496,
      "step": 70625
    },
    {
      "epoch": 6.73532485961914e-05,
      "step": 70625,
      "training_step_time": 7.616678237915039
    },
    {
      "epoch": 6.735420227050782e-05,
      "model_forward_time": 1.73643159866333,
      "step": 70626
    },
    {
      "epoch": 6.735420227050782e-05,
      "step": 70626,
      "training_step_time": 7.73125696182251
    },
    {
      "epoch": 6.735515594482421e-05,
      "model_forward_time": 1.734318733215332,
      "step": 70627
    },
    {
      "epoch": 6.735515594482421e-05,
      "step": 70627,
      "training_step_time": 7.707559585571289
    },
    {
      "epoch": 6.735610961914063e-05,
      "model_forward_time": 1.7303521633148193,
      "step": 70628
    },
    {
      "epoch": 6.735610961914063e-05,
      "step": 70628,
      "training_step_time": 8.825414180755615
    },
    {
      "epoch": 6.735706329345704e-05,
      "model_forward_time": 2.2691810131073,
      "step": 70629
    },
    {
      "epoch": 6.735706329345704e-05,
      "step": 70629,
      "training_step_time": 10.24117660522461
    },
    {
      "epoch": 6.735801696777343e-05,
      "grad_norm": 0.11410102248191833,
      "learning_rate": 2.178789046806326e-06,
      "loss": 0.0584,
      "step": 70630
    },
    {
      "epoch": 6.735801696777343e-05,
      "model_forward_time": 1.7307720184326172,
      "step": 70630
    },
    {
      "dynamics_loss_avg": 0.06607538610696792,
      "epoch": 6.735801696777343e-05,
      "step": 70630
    },
    {
      "action_loss_avg": 0.0018044619413558395,
      "epoch": 6.735801696777343e-05,
      "step": 70630
    },
    {
      "epoch": 6.735801696777343e-05,
      "step": 70630,
      "training_step_time": 7.833164930343628
    },
    {
      "epoch": 6.735897064208984e-05,
      "model_forward_time": 1.7304134368896484,
      "step": 70631
    },
    {
      "epoch": 6.735897064208984e-05,
      "step": 70631,
      "training_step_time": 7.8284361362457275
    },
    {
      "epoch": 6.735992431640625e-05,
      "model_forward_time": 1.733996868133545,
      "step": 70632
    },
    {
      "epoch": 6.735992431640625e-05,
      "step": 70632,
      "training_step_time": 7.684713840484619
    },
    {
      "epoch": 6.736087799072265e-05,
      "model_forward_time": 1.748948335647583,
      "step": 70633
    },
    {
      "epoch": 6.736087799072265e-05,
      "step": 70633,
      "training_step_time": 6.8210532665252686
    },
    {
      "epoch": 6.736183166503906e-05,
      "model_forward_time": 1.7151994705200195,
      "step": 70634
    },
    {
      "epoch": 6.736183166503906e-05,
      "step": 70634,
      "training_step_time": 7.682683944702148
    },
    {
      "epoch": 6.736278533935547e-05,
      "model_forward_time": 1.7297437191009521,
      "step": 70635
    },
    {
      "epoch": 6.736278533935547e-05,
      "step": 70635,
      "training_step_time": 7.815718650817871
    },
    {
      "epoch": 6.736373901367187e-05,
      "model_forward_time": 1.359201192855835,
      "step": 70636
    },
    {
      "epoch": 6.736373901367187e-05,
      "step": 70636,
      "training_step_time": 7.714232921600342
    },
    {
      "epoch": 6.736469268798828e-05,
      "model_forward_time": 1.7234785556793213,
      "step": 70637
    },
    {
      "epoch": 6.736469268798828e-05,
      "step": 70637,
      "training_step_time": 8.490886211395264
    },
    {
      "epoch": 6.736564636230469e-05,
      "model_forward_time": 2.3497536182403564,
      "step": 70638
    },
    {
      "epoch": 6.736564636230469e-05,
      "step": 70638,
      "training_step_time": 9.637450218200684
    },
    {
      "epoch": 6.736660003662109e-05,
      "model_forward_time": 1.7227156162261963,
      "step": 70639
    },
    {
      "epoch": 6.736660003662109e-05,
      "step": 70639,
      "training_step_time": 8.368527889251709
    },
    {
      "epoch": 6.73675537109375e-05,
      "grad_norm": 0.08696192502975464,
      "learning_rate": 2.1774240810000547e-06,
      "loss": 0.0592,
      "step": 70640
    },
    {
      "epoch": 6.73675537109375e-05,
      "model_forward_time": 1.7314879894256592,
      "step": 70640
    },
    {
      "dynamics_loss_avg": 0.04647404234856367,
      "epoch": 6.73675537109375e-05,
      "step": 70640
    },
    {
      "action_loss_avg": 0.0011105894664069638,
      "epoch": 6.73675537109375e-05,
      "step": 70640
    },
    {
      "epoch": 6.73675537109375e-05,
      "step": 70640,
      "training_step_time": 7.76694917678833
    },
    {
      "epoch": 6.736850738525391e-05,
      "model_forward_time": 1.7266082763671875,
      "step": 70641
    },
    {
      "epoch": 6.736850738525391e-05,
      "step": 70641,
      "training_step_time": 7.749605655670166
    },
    {
      "epoch": 6.736946105957031e-05,
      "model_forward_time": 1.7303552627563477,
      "step": 70642
    },
    {
      "epoch": 6.736946105957031e-05,
      "step": 70642,
      "training_step_time": 7.595979928970337
    },
    {
      "epoch": 6.737041473388672e-05,
      "model_forward_time": 1.7285192012786865,
      "step": 70643
    },
    {
      "epoch": 6.737041473388672e-05,
      "step": 70643,
      "training_step_time": 7.6090192794799805
    },
    {
      "epoch": 6.737136840820313e-05,
      "model_forward_time": 1.7436907291412354,
      "step": 70644
    },
    {
      "epoch": 6.737136840820313e-05,
      "step": 70644,
      "training_step_time": 7.745972633361816
    },
    {
      "epoch": 6.737232208251953e-05,
      "model_forward_time": 1.729496717453003,
      "step": 70645
    },
    {
      "epoch": 6.737232208251953e-05,
      "step": 70645,
      "training_step_time": 7.844542026519775
    },
    {
      "epoch": 6.737327575683594e-05,
      "model_forward_time": 1.7280452251434326,
      "step": 70646
    },
    {
      "epoch": 6.737327575683594e-05,
      "step": 70646,
      "training_step_time": 9.15089464187622
    },
    {
      "epoch": 6.737422943115235e-05,
      "model_forward_time": 2.350257396697998,
      "step": 70647
    },
    {
      "epoch": 6.737422943115235e-05,
      "step": 70647,
      "training_step_time": 8.624271154403687
    },
    {
      "epoch": 6.737518310546875e-05,
      "model_forward_time": 1.723031997680664,
      "step": 70648
    },
    {
      "epoch": 6.737518310546875e-05,
      "step": 70648,
      "training_step_time": 8.620208263397217
    },
    {
      "epoch": 6.737613677978516e-05,
      "model_forward_time": 1.7337145805358887,
      "step": 70649
    },
    {
      "epoch": 6.737613677978516e-05,
      "step": 70649,
      "training_step_time": 8.254454851150513
    },
    {
      "epoch": 6.737709045410157e-05,
      "grad_norm": 0.08342105150222778,
      "learning_rate": 2.1760594238664403e-06,
      "loss": 0.0575,
      "step": 70650
    },
    {
      "epoch": 6.737709045410157e-05,
      "model_forward_time": 1.722379207611084,
      "step": 70650
    },
    {
      "dynamics_loss_avg": 0.05884089097380638,
      "epoch": 6.737709045410157e-05,
      "step": 70650
    },
    {
      "action_loss_avg": 0.001457666838541627,
      "epoch": 6.737709045410157e-05,
      "step": 70650
    },
    {
      "epoch": 6.737709045410157e-05,
      "step": 70650,
      "training_step_time": 7.726893424987793
    },
    {
      "epoch": 6.737804412841797e-05,
      "model_forward_time": 1.7243010997772217,
      "step": 70651
    },
    {
      "epoch": 6.737804412841797e-05,
      "step": 70651,
      "training_step_time": 7.558105707168579
    },
    {
      "epoch": 6.737899780273438e-05,
      "model_forward_time": 1.7246286869049072,
      "step": 70652
    },
    {
      "epoch": 6.737899780273438e-05,
      "step": 70652,
      "training_step_time": 7.598429441452026
    },
    {
      "epoch": 6.737995147705079e-05,
      "model_forward_time": 1.7242867946624756,
      "step": 70653
    },
    {
      "epoch": 6.737995147705079e-05,
      "step": 70653,
      "training_step_time": 7.73848557472229
    },
    {
      "epoch": 6.738090515136718e-05,
      "model_forward_time": 1.728574275970459,
      "step": 70654
    },
    {
      "epoch": 6.738090515136718e-05,
      "step": 70654,
      "training_step_time": 8.226571559906006
    },
    {
      "epoch": 6.73818588256836e-05,
      "model_forward_time": 1.7272863388061523,
      "step": 70655
    },
    {
      "epoch": 6.73818588256836e-05,
      "step": 70655,
      "training_step_time": 7.696711778640747
    },
    {
      "epoch": 6.73828125e-05,
      "model_forward_time": 2.333869695663452,
      "step": 70656
    },
    {
      "epoch": 6.73828125e-05,
      "step": 70656,
      "training_step_time": 9.02319049835205
    },
    {
      "epoch": 6.73837661743164e-05,
      "model_forward_time": 1.7272922992706299,
      "step": 70657
    },
    {
      "epoch": 6.73837661743164e-05,
      "step": 70657,
      "training_step_time": 7.932477235794067
    },
    {
      "epoch": 6.738471984863281e-05,
      "model_forward_time": 1.726618766784668,
      "step": 70658
    },
    {
      "epoch": 6.738471984863281e-05,
      "step": 70658,
      "training_step_time": 8.805467367172241
    },
    {
      "epoch": 6.738567352294922e-05,
      "model_forward_time": 1.7392477989196777,
      "step": 70659
    },
    {
      "epoch": 6.738567352294922e-05,
      "step": 70659,
      "training_step_time": 7.785741806030273
    },
    {
      "epoch": 6.738662719726562e-05,
      "grad_norm": 0.08390868455171585,
      "learning_rate": 2.174695075554723e-06,
      "loss": 0.0575,
      "step": 70660
    },
    {
      "epoch": 6.738662719726562e-05,
      "model_forward_time": 1.7277672290802002,
      "step": 70660
    },
    {
      "dynamics_loss_avg": 0.0708643939346075,
      "epoch": 6.738662719726562e-05,
      "step": 70660
    },
    {
      "action_loss_avg": 0.0019452239212114364,
      "epoch": 6.738662719726562e-05,
      "step": 70660
    },
    {
      "epoch": 6.738662719726562e-05,
      "step": 70660,
      "training_step_time": 7.574435234069824
    },
    {
      "epoch": 6.738758087158203e-05,
      "model_forward_time": 1.7138664722442627,
      "step": 70661
    },
    {
      "epoch": 6.738758087158203e-05,
      "step": 70661,
      "training_step_time": 7.609363794326782
    },
    {
      "epoch": 6.738853454589844e-05,
      "model_forward_time": 1.7334175109863281,
      "step": 70662
    },
    {
      "epoch": 6.738853454589844e-05,
      "step": 70662,
      "training_step_time": 7.7557432651519775
    },
    {
      "epoch": 6.738948822021484e-05,
      "model_forward_time": 1.7314684391021729,
      "step": 70663
    },
    {
      "epoch": 6.738948822021484e-05,
      "step": 70663,
      "training_step_time": 7.662813663482666
    },
    {
      "epoch": 6.739044189453125e-05,
      "model_forward_time": 1.7207999229431152,
      "step": 70664
    },
    {
      "epoch": 6.739044189453125e-05,
      "step": 70664,
      "training_step_time": 9.288471221923828
    },
    {
      "epoch": 6.739139556884766e-05,
      "model_forward_time": 2.316441535949707,
      "step": 70665
    },
    {
      "epoch": 6.739139556884766e-05,
      "step": 70665,
      "training_step_time": 10.483353853225708
    },
    {
      "epoch": 6.739234924316406e-05,
      "model_forward_time": 1.7189886569976807,
      "step": 70666
    },
    {
      "epoch": 6.739234924316406e-05,
      "step": 70666,
      "training_step_time": 9.076047658920288
    },
    {
      "epoch": 6.739330291748047e-05,
      "model_forward_time": 1.707237720489502,
      "step": 70667
    },
    {
      "epoch": 6.739330291748047e-05,
      "step": 70667,
      "training_step_time": 7.80700421333313
    },
    {
      "epoch": 6.739425659179688e-05,
      "model_forward_time": 1.7188737392425537,
      "step": 70668
    },
    {
      "epoch": 6.739425659179688e-05,
      "step": 70668,
      "training_step_time": 7.831655263900757
    },
    {
      "epoch": 6.739521026611328e-05,
      "model_forward_time": 1.7232544422149658,
      "step": 70669
    },
    {
      "epoch": 6.739521026611328e-05,
      "step": 70669,
      "training_step_time": 7.591871976852417
    },
    {
      "epoch": 6.739616394042969e-05,
      "grad_norm": 0.09518875926733017,
      "learning_rate": 2.173331036214103e-06,
      "loss": 0.0536,
      "step": 70670
    },
    {
      "epoch": 6.739616394042969e-05,
      "model_forward_time": 1.704183578491211,
      "step": 70670
    },
    {
      "dynamics_loss_avg": 0.03891957374289632,
      "epoch": 6.739616394042969e-05,
      "step": 70670
    },
    {
      "action_loss_avg": 0.0011539887345861644,
      "epoch": 6.739616394042969e-05,
      "step": 70670
    },
    {
      "epoch": 6.739616394042969e-05,
      "step": 70670,
      "training_step_time": 7.586275100708008
    },
    {
      "epoch": 6.73971176147461e-05,
      "model_forward_time": 1.7200899124145508,
      "step": 70671
    },
    {
      "epoch": 6.73971176147461e-05,
      "step": 70671,
      "training_step_time": 8.242872714996338
    },
    {
      "epoch": 6.73980712890625e-05,
      "model_forward_time": 1.7180097103118896,
      "step": 70672
    },
    {
      "epoch": 6.73980712890625e-05,
      "step": 70672,
      "training_step_time": 7.644654750823975
    },
    {
      "epoch": 6.739902496337891e-05,
      "model_forward_time": 1.711890459060669,
      "step": 70673
    },
    {
      "epoch": 6.739902496337891e-05,
      "step": 70673,
      "training_step_time": 7.873547554016113
    },
    {
      "epoch": 6.739997863769532e-05,
      "model_forward_time": 1.7214186191558838,
      "step": 70674
    },
    {
      "epoch": 6.739997863769532e-05,
      "step": 70674,
      "training_step_time": 9.786023139953613
    },
    {
      "epoch": 6.740093231201172e-05,
      "model_forward_time": 1.8033339977264404,
      "step": 70675
    },
    {
      "epoch": 6.740093231201172e-05,
      "step": 70675,
      "training_step_time": 8.062563180923462
    },
    {
      "epoch": 6.740188598632813e-05,
      "model_forward_time": 1.7218728065490723,
      "step": 70676
    },
    {
      "epoch": 6.740188598632813e-05,
      "step": 70676,
      "training_step_time": 7.906018018722534
    },
    {
      "epoch": 6.740283966064454e-05,
      "model_forward_time": 1.7269287109375,
      "step": 70677
    },
    {
      "epoch": 6.740283966064454e-05,
      "step": 70677,
      "training_step_time": 7.7626447677612305
    },
    {
      "epoch": 6.740379333496093e-05,
      "model_forward_time": 1.7235145568847656,
      "step": 70678
    },
    {
      "epoch": 6.740379333496093e-05,
      "step": 70678,
      "training_step_time": 7.750814914703369
    },
    {
      "epoch": 6.740474700927735e-05,
      "model_forward_time": 1.7087602615356445,
      "step": 70679
    },
    {
      "epoch": 6.740474700927735e-05,
      "step": 70679,
      "training_step_time": 7.645955801010132
    },
    {
      "epoch": 6.740570068359376e-05,
      "grad_norm": 0.11865480989217758,
      "learning_rate": 2.171967305993753e-06,
      "loss": 0.0591,
      "step": 70680
    },
    {
      "epoch": 6.740570068359376e-05,
      "model_forward_time": 1.722200632095337,
      "step": 70680
    },
    {
      "dynamics_loss_avg": 0.048751750215888026,
      "epoch": 6.740570068359376e-05,
      "step": 70680
    },
    {
      "action_loss_avg": 0.0016076880317996256,
      "epoch": 6.740570068359376e-05,
      "step": 70680
    },
    {
      "epoch": 6.740570068359376e-05,
      "step": 70680,
      "training_step_time": 8.737653970718384
    },
    {
      "epoch": 6.740665435791015e-05,
      "model_forward_time": 1.7188410758972168,
      "step": 70681
    },
    {
      "epoch": 6.740665435791015e-05,
      "step": 70681,
      "training_step_time": 7.490262269973755
    },
    {
      "epoch": 6.740760803222656e-05,
      "model_forward_time": 1.7271754741668701,
      "step": 70682
    },
    {
      "epoch": 6.740760803222656e-05,
      "step": 70682,
      "training_step_time": 7.780573606491089
    },
    {
      "epoch": 6.740856170654298e-05,
      "model_forward_time": 1.7284574508666992,
      "step": 70683
    },
    {
      "epoch": 6.740856170654298e-05,
      "step": 70683,
      "training_step_time": 8.63251280784607
    },
    {
      "epoch": 6.740951538085937e-05,
      "model_forward_time": 2.357692241668701,
      "step": 70684
    },
    {
      "epoch": 6.740951538085937e-05,
      "step": 70684,
      "training_step_time": 10.150236129760742
    },
    {
      "epoch": 6.741046905517578e-05,
      "model_forward_time": 1.7192473411560059,
      "step": 70685
    },
    {
      "epoch": 6.741046905517578e-05,
      "step": 70685,
      "training_step_time": 8.57174825668335
    },
    {
      "epoch": 6.74114227294922e-05,
      "model_forward_time": 1.708294153213501,
      "step": 70686
    },
    {
      "epoch": 6.74114227294922e-05,
      "step": 70686,
      "training_step_time": 7.712353706359863
    },
    {
      "epoch": 6.741237640380859e-05,
      "model_forward_time": 1.7290751934051514,
      "step": 70687
    },
    {
      "epoch": 6.741237640380859e-05,
      "step": 70687,
      "training_step_time": 7.798274517059326
    },
    {
      "epoch": 6.7413330078125e-05,
      "model_forward_time": 1.7212364673614502,
      "step": 70688
    },
    {
      "epoch": 6.7413330078125e-05,
      "step": 70688,
      "training_step_time": 7.618133544921875
    },
    {
      "epoch": 6.74142837524414e-05,
      "model_forward_time": 1.7343525886535645,
      "step": 70689
    },
    {
      "epoch": 6.74142837524414e-05,
      "step": 70689,
      "training_step_time": 7.729266405105591
    },
    {
      "epoch": 6.741523742675781e-05,
      "grad_norm": 0.08800879865884781,
      "learning_rate": 2.1706038850428023e-06,
      "loss": 0.0553,
      "step": 70690
    },
    {
      "epoch": 6.741523742675781e-05,
      "model_forward_time": 1.7148823738098145,
      "step": 70690
    },
    {
      "dynamics_loss_avg": 0.0500373749062419,
      "epoch": 6.741523742675781e-05,
      "step": 70690
    },
    {
      "action_loss_avg": 0.0012129997659940273,
      "epoch": 6.741523742675781e-05,
      "step": 70690
    },
    {
      "epoch": 6.741523742675781e-05,
      "step": 70690,
      "training_step_time": 7.930613994598389
    },
    {
      "epoch": 6.741619110107422e-05,
      "model_forward_time": 1.7109017372131348,
      "step": 70691
    },
    {
      "epoch": 6.741619110107422e-05,
      "step": 70691,
      "training_step_time": 7.853961229324341
    },
    {
      "epoch": 6.741714477539062e-05,
      "model_forward_time": 1.706207513809204,
      "step": 70692
    },
    {
      "epoch": 6.741714477539062e-05,
      "step": 70692,
      "training_step_time": 7.879727840423584
    },
    {
      "epoch": 6.741809844970703e-05,
      "model_forward_time": 2.3442840576171875,
      "step": 70693
    },
    {
      "epoch": 6.741809844970703e-05,
      "step": 70693,
      "training_step_time": 10.04487919807434
    },
    {
      "epoch": 6.741905212402344e-05,
      "model_forward_time": 1.7149748802185059,
      "step": 70694
    },
    {
      "epoch": 6.741905212402344e-05,
      "step": 70694,
      "training_step_time": 8.505910634994507
    },
    {
      "epoch": 6.742000579833984e-05,
      "model_forward_time": 1.7156219482421875,
      "step": 70695
    },
    {
      "epoch": 6.742000579833984e-05,
      "step": 70695,
      "training_step_time": 7.745658874511719
    },
    {
      "epoch": 6.742095947265625e-05,
      "model_forward_time": 1.7276225090026855,
      "step": 70696
    },
    {
      "epoch": 6.742095947265625e-05,
      "step": 70696,
      "training_step_time": 7.774433851242065
    },
    {
      "epoch": 6.742191314697266e-05,
      "model_forward_time": 1.712825059890747,
      "step": 70697
    },
    {
      "epoch": 6.742191314697266e-05,
      "step": 70697,
      "training_step_time": 7.76792049407959
    },
    {
      "epoch": 6.742286682128906e-05,
      "model_forward_time": 1.7201740741729736,
      "step": 70698
    },
    {
      "epoch": 6.742286682128906e-05,
      "step": 70698,
      "training_step_time": 7.726015567779541
    },
    {
      "epoch": 6.742382049560547e-05,
      "model_forward_time": 1.7219264507293701,
      "step": 70699
    },
    {
      "epoch": 6.742382049560547e-05,
      "step": 70699,
      "training_step_time": 7.075262784957886
    },
    {
      "epoch": 6.742477416992188e-05,
      "grad_norm": 0.09770109504461288,
      "learning_rate": 2.169240773510361e-06,
      "loss": 0.0555,
      "step": 70700
    },
    {
      "epoch": 6.742477416992188e-05,
      "model_forward_time": 1.7117955684661865,
      "step": 70700
    },
    {
      "dynamics_loss_avg": 0.060667498409748076,
      "epoch": 6.742477416992188e-05,
      "step": 70700
    },
    {
      "action_loss_avg": 0.0013285813911352307,
      "epoch": 6.742477416992188e-05,
      "step": 70700
    },
    {
      "epoch": 6.742477416992188e-05,
      "step": 70700,
      "training_step_time": 8.702350616455078
    },
    {
      "epoch": 6.742572784423827e-05,
      "model_forward_time": 1.7173371315002441,
      "step": 70701
    },
    {
      "epoch": 6.742572784423827e-05,
      "step": 70701,
      "training_step_time": 7.807997941970825
    },
    {
      "epoch": 6.742668151855469e-05,
      "model_forward_time": 2.315948247909546,
      "step": 70702
    },
    {
      "epoch": 6.742668151855469e-05,
      "step": 70702,
      "training_step_time": 7.737266302108765
    },
    {
      "epoch": 6.74276351928711e-05,
      "model_forward_time": 1.7180824279785156,
      "step": 70703
    },
    {
      "epoch": 6.74276351928711e-05,
      "step": 70703,
      "training_step_time": 8.978149652481079
    },
    {
      "epoch": 6.74285888671875e-05,
      "model_forward_time": 1.7234916687011719,
      "step": 70704
    },
    {
      "epoch": 6.74285888671875e-05,
      "step": 70704,
      "training_step_time": 8.747320413589478
    },
    {
      "epoch": 6.74295425415039e-05,
      "model_forward_time": 1.7210252285003662,
      "step": 70705
    },
    {
      "epoch": 6.74295425415039e-05,
      "step": 70705,
      "training_step_time": 7.782138109207153
    },
    {
      "epoch": 6.743049621582032e-05,
      "model_forward_time": 1.749419927597046,
      "step": 70706
    },
    {
      "epoch": 6.743049621582032e-05,
      "step": 70706,
      "training_step_time": 7.680732011795044
    },
    {
      "epoch": 6.743144989013671e-05,
      "model_forward_time": 1.7209033966064453,
      "step": 70707
    },
    {
      "epoch": 6.743144989013671e-05,
      "step": 70707,
      "training_step_time": 7.678898096084595
    },
    {
      "epoch": 6.743240356445312e-05,
      "model_forward_time": 1.720351219177246,
      "step": 70708
    },
    {
      "epoch": 6.743240356445312e-05,
      "step": 70708,
      "training_step_time": 7.782058000564575
    },
    {
      "epoch": 6.743335723876953e-05,
      "model_forward_time": 1.7165112495422363,
      "step": 70709
    },
    {
      "epoch": 6.743335723876953e-05,
      "step": 70709,
      "training_step_time": 7.660207033157349
    },
    {
      "epoch": 6.743431091308593e-05,
      "grad_norm": 0.1029963493347168,
      "learning_rate": 2.1678779715454908e-06,
      "loss": 0.0544,
      "step": 70710
    },
    {
      "epoch": 6.743431091308593e-05,
      "model_forward_time": 1.7280821800231934,
      "step": 70710
    },
    {
      "dynamics_loss_avg": 0.04976553414016962,
      "epoch": 6.743431091308593e-05,
      "step": 70710
    },
    {
      "action_loss_avg": 0.0014721002284204588,
      "epoch": 6.743431091308593e-05,
      "step": 70710
    },
    {
      "epoch": 6.743431091308593e-05,
      "step": 70710,
      "training_step_time": 9.179277420043945
    },
    {
      "epoch": 6.743526458740234e-05,
      "model_forward_time": 2.263159990310669,
      "step": 70711
    },
    {
      "epoch": 6.743526458740234e-05,
      "step": 70711,
      "training_step_time": 7.716551065444946
    },
    {
      "epoch": 6.743621826171875e-05,
      "model_forward_time": 1.7987914085388184,
      "step": 70712
    },
    {
      "epoch": 6.743621826171875e-05,
      "step": 70712,
      "training_step_time": 9.688906192779541
    },
    {
      "epoch": 6.743717193603515e-05,
      "model_forward_time": 1.7194888591766357,
      "step": 70713
    },
    {
      "epoch": 6.743717193603515e-05,
      "step": 70713,
      "training_step_time": 9.390135049819946
    },
    {
      "epoch": 6.743812561035156e-05,
      "model_forward_time": 1.7150421142578125,
      "step": 70714
    },
    {
      "epoch": 6.743812561035156e-05,
      "step": 70714,
      "training_step_time": 7.795889139175415
    },
    {
      "epoch": 6.743907928466797e-05,
      "model_forward_time": 1.7231383323669434,
      "step": 70715
    },
    {
      "epoch": 6.743907928466797e-05,
      "step": 70715,
      "training_step_time": 7.641072511672974
    },
    {
      "epoch": 6.744003295898437e-05,
      "model_forward_time": 1.7110049724578857,
      "step": 70716
    },
    {
      "epoch": 6.744003295898437e-05,
      "step": 70716,
      "training_step_time": 7.693507432937622
    },
    {
      "epoch": 6.744098663330078e-05,
      "model_forward_time": 1.6979167461395264,
      "step": 70717
    },
    {
      "epoch": 6.744098663330078e-05,
      "step": 70717,
      "training_step_time": 7.6488635540008545
    },
    {
      "epoch": 6.744194030761719e-05,
      "model_forward_time": 1.7191667556762695,
      "step": 70718
    },
    {
      "epoch": 6.744194030761719e-05,
      "step": 70718,
      "training_step_time": 7.631826877593994
    },
    {
      "epoch": 6.744289398193359e-05,
      "model_forward_time": 1.7162652015686035,
      "step": 70719
    },
    {
      "epoch": 6.744289398193359e-05,
      "step": 70719,
      "training_step_time": 7.630899906158447
    },
    {
      "epoch": 6.744384765625e-05,
      "grad_norm": 0.08600825071334839,
      "learning_rate": 2.1665154792972292e-06,
      "loss": 0.0558,
      "step": 70720
    },
    {
      "epoch": 6.744384765625e-05,
      "model_forward_time": 2.253920793533325,
      "step": 70720
    },
    {
      "dynamics_loss_avg": 0.04644297044724226,
      "epoch": 6.744384765625e-05,
      "step": 70720
    },
    {
      "action_loss_avg": 0.0010256987065076829,
      "epoch": 6.744384765625e-05,
      "step": 70720
    },
    {
      "epoch": 6.744384765625e-05,
      "step": 70720,
      "training_step_time": 8.593504190444946
    },
    {
      "epoch": 6.744480133056641e-05,
      "model_forward_time": 1.767683744430542,
      "step": 70721
    },
    {
      "epoch": 6.744480133056641e-05,
      "step": 70721,
      "training_step_time": 10.305865049362183
    },
    {
      "epoch": 6.744575500488281e-05,
      "model_forward_time": 1.7172493934631348,
      "step": 70722
    },
    {
      "epoch": 6.744575500488281e-05,
      "step": 70722,
      "training_step_time": 7.90209698677063
    },
    {
      "epoch": 6.744670867919922e-05,
      "model_forward_time": 1.720031976699829,
      "step": 70723
    },
    {
      "epoch": 6.744670867919922e-05,
      "step": 70723,
      "training_step_time": 7.689999580383301
    },
    {
      "epoch": 6.744766235351563e-05,
      "model_forward_time": 1.7267987728118896,
      "step": 70724
    },
    {
      "epoch": 6.744766235351563e-05,
      "step": 70724,
      "training_step_time": 7.681920051574707
    },
    {
      "epoch": 6.744861602783203e-05,
      "model_forward_time": 1.7130405902862549,
      "step": 70725
    },
    {
      "epoch": 6.744861602783203e-05,
      "step": 70725,
      "training_step_time": 7.6759867668151855
    },
    {
      "epoch": 6.744956970214844e-05,
      "model_forward_time": 1.7134852409362793,
      "step": 70726
    },
    {
      "epoch": 6.744956970214844e-05,
      "step": 70726,
      "training_step_time": 7.711923360824585
    },
    {
      "epoch": 6.745052337646485e-05,
      "model_forward_time": 1.7178187370300293,
      "step": 70727
    },
    {
      "epoch": 6.745052337646485e-05,
      "step": 70727,
      "training_step_time": 7.685284614562988
    },
    {
      "epoch": 6.745147705078124e-05,
      "model_forward_time": 1.7169864177703857,
      "step": 70728
    },
    {
      "epoch": 6.745147705078124e-05,
      "step": 70728,
      "training_step_time": 8.441200256347656
    },
    {
      "epoch": 6.745243072509766e-05,
      "model_forward_time": 1.721904993057251,
      "step": 70729
    },
    {
      "epoch": 6.745243072509766e-05,
      "step": 70729,
      "training_step_time": 8.776904344558716
    },
    {
      "epoch": 6.745338439941407e-05,
      "grad_norm": 0.10059129446744919,
      "learning_rate": 2.1651532969145733e-06,
      "loss": 0.0602,
      "step": 70730
    },
    {
      "epoch": 6.745338439941407e-05,
      "model_forward_time": 2.3357956409454346,
      "step": 70730
    },
    {
      "dynamics_loss_avg": 0.05897313859313726,
      "epoch": 6.745338439941407e-05,
      "step": 70730
    },
    {
      "action_loss_avg": 0.001488817149947863,
      "epoch": 6.745338439941407e-05,
      "step": 70730
    },
    {
      "epoch": 6.745338439941407e-05,
      "step": 70730,
      "training_step_time": 9.450709581375122
    },
    {
      "epoch": 6.745433807373046e-05,
      "model_forward_time": 1.7110116481781006,
      "step": 70731
    },
    {
      "epoch": 6.745433807373046e-05,
      "step": 70731,
      "training_step_time": 7.845289945602417
    },
    {
      "epoch": 6.745529174804687e-05,
      "model_forward_time": 1.7187743186950684,
      "step": 70732
    },
    {
      "epoch": 6.745529174804687e-05,
      "step": 70732,
      "training_step_time": 8.755498170852661
    },
    {
      "epoch": 6.745624542236329e-05,
      "model_forward_time": 1.7270545959472656,
      "step": 70733
    },
    {
      "epoch": 6.745624542236329e-05,
      "step": 70733,
      "training_step_time": 7.761044263839722
    },
    {
      "epoch": 6.745719909667968e-05,
      "model_forward_time": 1.7243773937225342,
      "step": 70734
    },
    {
      "epoch": 6.745719909667968e-05,
      "step": 70734,
      "training_step_time": 7.718550682067871
    },
    {
      "epoch": 6.745815277099609e-05,
      "model_forward_time": 1.7202534675598145,
      "step": 70735
    },
    {
      "epoch": 6.745815277099609e-05,
      "step": 70735,
      "training_step_time": 8.777042865753174
    },
    {
      "epoch": 6.74591064453125e-05,
      "model_forward_time": 1.7306528091430664,
      "step": 70736
    },
    {
      "epoch": 6.74591064453125e-05,
      "step": 70736,
      "training_step_time": 7.624652862548828
    },
    {
      "epoch": 6.74600601196289e-05,
      "model_forward_time": 1.7149171829223633,
      "step": 70737
    },
    {
      "epoch": 6.74600601196289e-05,
      "step": 70737,
      "training_step_time": 7.59993577003479
    },
    {
      "epoch": 6.746101379394531e-05,
      "model_forward_time": 1.7191405296325684,
      "step": 70738
    },
    {
      "epoch": 6.746101379394531e-05,
      "step": 70738,
      "training_step_time": 8.29855990409851
    },
    {
      "epoch": 6.746196746826172e-05,
      "model_forward_time": 2.3110146522521973,
      "step": 70739
    },
    {
      "epoch": 6.746196746826172e-05,
      "step": 70739,
      "training_step_time": 10.609581470489502
    },
    {
      "epoch": 6.746292114257812e-05,
      "grad_norm": 0.10276270657777786,
      "learning_rate": 2.16379142454649e-06,
      "loss": 0.0584,
      "step": 70740
    },
    {
      "epoch": 6.746292114257812e-05,
      "model_forward_time": 1.7169458866119385,
      "step": 70740
    },
    {
      "dynamics_loss_avg": 0.05155444443225861,
      "epoch": 6.746292114257812e-05,
      "step": 70740
    },
    {
      "action_loss_avg": 0.0009444984636502341,
      "epoch": 6.746292114257812e-05,
      "step": 70740
    },
    {
      "epoch": 6.746292114257812e-05,
      "step": 70740,
      "training_step_time": 8.772037982940674
    },
    {
      "epoch": 6.746387481689453e-05,
      "model_forward_time": 1.7115974426269531,
      "step": 70741
    },
    {
      "epoch": 6.746387481689453e-05,
      "step": 70741,
      "training_step_time": 7.6988608837127686
    },
    {
      "epoch": 6.746482849121094e-05,
      "model_forward_time": 1.716888189315796,
      "step": 70742
    },
    {
      "epoch": 6.746482849121094e-05,
      "step": 70742,
      "training_step_time": 7.717792987823486
    },
    {
      "epoch": 6.746578216552734e-05,
      "model_forward_time": 1.724484920501709,
      "step": 70743
    },
    {
      "epoch": 6.746578216552734e-05,
      "step": 70743,
      "training_step_time": 7.1291115283966064
    },
    {
      "epoch": 6.746673583984375e-05,
      "model_forward_time": 1.7195773124694824,
      "step": 70744
    },
    {
      "epoch": 6.746673583984375e-05,
      "step": 70744,
      "training_step_time": 7.6953325271606445
    },
    {
      "epoch": 6.746768951416016e-05,
      "model_forward_time": 1.7199132442474365,
      "step": 70745
    },
    {
      "epoch": 6.746768951416016e-05,
      "step": 70745,
      "training_step_time": 7.78230357170105
    },
    {
      "epoch": 6.746864318847656e-05,
      "model_forward_time": 1.7200264930725098,
      "step": 70746
    },
    {
      "epoch": 6.746864318847656e-05,
      "step": 70746,
      "training_step_time": 8.340800046920776
    },
    {
      "epoch": 6.746959686279297e-05,
      "model_forward_time": 1.7168426513671875,
      "step": 70747
    },
    {
      "epoch": 6.746959686279297e-05,
      "step": 70747,
      "training_step_time": 8.884892225265503
    },
    {
      "epoch": 6.747055053710938e-05,
      "model_forward_time": 2.2735836505889893,
      "step": 70748
    },
    {
      "epoch": 6.747055053710938e-05,
      "step": 70748,
      "training_step_time": 8.777955055236816
    },
    {
      "epoch": 6.747150421142578e-05,
      "model_forward_time": 1.8038675785064697,
      "step": 70749
    },
    {
      "epoch": 6.747150421142578e-05,
      "step": 70749,
      "training_step_time": 9.959856986999512
    },
    {
      "epoch": 6.747245788574219e-05,
      "grad_norm": 0.10377722978591919,
      "learning_rate": 2.1624298623419133e-06,
      "loss": 0.0588,
      "step": 70750
    },
    {
      "epoch": 6.747245788574219e-05,
      "model_forward_time": 1.7237601280212402,
      "step": 70750
    },
    {
      "dynamics_loss_avg": 0.03738053292036057,
      "epoch": 6.747245788574219e-05,
      "step": 70750
    },
    {
      "action_loss_avg": 0.0009512228338280692,
      "epoch": 6.747245788574219e-05,
      "step": 70750
    },
    {
      "epoch": 6.747245788574219e-05,
      "step": 70750,
      "training_step_time": 9.646056652069092
    },
    {
      "epoch": 6.74734115600586e-05,
      "model_forward_time": 1.71626615524292,
      "step": 70751
    },
    {
      "epoch": 6.74734115600586e-05,
      "step": 70751,
      "training_step_time": 8.213566064834595
    },
    {
      "epoch": 6.7474365234375e-05,
      "model_forward_time": 1.7174808979034424,
      "step": 70752
    },
    {
      "epoch": 6.7474365234375e-05,
      "step": 70752,
      "training_step_time": 7.465254783630371
    },
    {
      "epoch": 6.74753189086914e-05,
      "model_forward_time": 1.715611219406128,
      "step": 70753
    },
    {
      "epoch": 6.74753189086914e-05,
      "step": 70753,
      "training_step_time": 8.63740611076355
    },
    {
      "epoch": 6.747627258300782e-05,
      "model_forward_time": 1.7319281101226807,
      "step": 70754
    },
    {
      "epoch": 6.747627258300782e-05,
      "step": 70754,
      "training_step_time": 7.670289754867554
    },
    {
      "epoch": 6.747722625732421e-05,
      "model_forward_time": 1.7061500549316406,
      "step": 70755
    },
    {
      "epoch": 6.747722625732421e-05,
      "step": 70755,
      "training_step_time": 7.618628740310669
    },
    {
      "epoch": 6.747817993164063e-05,
      "model_forward_time": 1.725757122039795,
      "step": 70756
    },
    {
      "epoch": 6.747817993164063e-05,
      "step": 70756,
      "training_step_time": 8.538330078125
    },
    {
      "epoch": 6.747913360595704e-05,
      "model_forward_time": 2.2685129642486572,
      "step": 70757
    },
    {
      "epoch": 6.747913360595704e-05,
      "step": 70757,
      "training_step_time": 8.435248851776123
    },
    {
      "epoch": 6.748008728027343e-05,
      "model_forward_time": 1.7808215618133545,
      "step": 70758
    },
    {
      "epoch": 6.748008728027343e-05,
      "step": 70758,
      "training_step_time": 7.978324890136719
    },
    {
      "epoch": 6.748104095458984e-05,
      "model_forward_time": 1.7139787673950195,
      "step": 70759
    },
    {
      "epoch": 6.748104095458984e-05,
      "step": 70759,
      "training_step_time": 7.87551736831665
    },
    {
      "epoch": 6.748199462890625e-05,
      "grad_norm": 0.06838610768318176,
      "learning_rate": 2.1610686104497413e-06,
      "loss": 0.0582,
      "step": 70760
    },
    {
      "epoch": 6.748199462890625e-05,
      "model_forward_time": 1.7175114154815674,
      "step": 70760
    },
    {
      "dynamics_loss_avg": 0.05375617630779743,
      "epoch": 6.748199462890625e-05,
      "step": 70760
    },
    {
      "action_loss_avg": 0.0018025973928160965,
      "epoch": 6.748199462890625e-05,
      "step": 70760
    },
    {
      "epoch": 6.748199462890625e-05,
      "step": 70760,
      "training_step_time": 7.713924884796143
    },
    {
      "epoch": 6.748294830322265e-05,
      "model_forward_time": 1.7185850143432617,
      "step": 70761
    },
    {
      "epoch": 6.748294830322265e-05,
      "step": 70761,
      "training_step_time": 7.664329767227173
    },
    {
      "epoch": 6.748390197753906e-05,
      "model_forward_time": 1.7300360202789307,
      "step": 70762
    },
    {
      "epoch": 6.748390197753906e-05,
      "step": 70762,
      "training_step_time": 7.7094621658325195
    },
    {
      "epoch": 6.748485565185547e-05,
      "model_forward_time": 1.7209186553955078,
      "step": 70763
    },
    {
      "epoch": 6.748485565185547e-05,
      "step": 70763,
      "training_step_time": 7.782527208328247
    },
    {
      "epoch": 6.748580932617187e-05,
      "model_forward_time": 1.7103722095489502,
      "step": 70764
    },
    {
      "epoch": 6.748580932617187e-05,
      "step": 70764,
      "training_step_time": 7.620373964309692
    },
    {
      "epoch": 6.748676300048828e-05,
      "model_forward_time": 1.7174882888793945,
      "step": 70765
    },
    {
      "epoch": 6.748676300048828e-05,
      "step": 70765,
      "training_step_time": 8.334062814712524
    },
    {
      "epoch": 6.748771667480469e-05,
      "model_forward_time": 2.2698912620544434,
      "step": 70766
    },
    {
      "epoch": 6.748771667480469e-05,
      "step": 70766,
      "training_step_time": 8.30699634552002
    },
    {
      "epoch": 6.748867034912109e-05,
      "model_forward_time": 1.7571256160736084,
      "step": 70767
    },
    {
      "epoch": 6.748867034912109e-05,
      "step": 70767,
      "training_step_time": 9.19105577468872
    },
    {
      "epoch": 6.74896240234375e-05,
      "model_forward_time": 1.7250845432281494,
      "step": 70768
    },
    {
      "epoch": 6.74896240234375e-05,
      "step": 70768,
      "training_step_time": 7.919867038726807
    },
    {
      "epoch": 6.749057769775391e-05,
      "model_forward_time": 1.7197291851043701,
      "step": 70769
    },
    {
      "epoch": 6.749057769775391e-05,
      "step": 70769,
      "training_step_time": 7.71903395652771
    },
    {
      "epoch": 6.749153137207031e-05,
      "grad_norm": 0.07369793951511383,
      "learning_rate": 2.1597076690188368e-06,
      "loss": 0.0577,
      "step": 70770
    },
    {
      "epoch": 6.749153137207031e-05,
      "model_forward_time": 1.729081630706787,
      "step": 70770
    },
    {
      "dynamics_loss_avg": 0.06827969066798686,
      "epoch": 6.749153137207031e-05,
      "step": 70770
    },
    {
      "action_loss_avg": 0.0014003092888742684,
      "epoch": 6.749153137207031e-05,
      "step": 70770
    },
    {
      "epoch": 6.749153137207031e-05,
      "step": 70770,
      "training_step_time": 7.66658616065979
    },
    {
      "epoch": 6.749248504638672e-05,
      "model_forward_time": 1.7223713397979736,
      "step": 70771
    },
    {
      "epoch": 6.749248504638672e-05,
      "step": 70771,
      "training_step_time": 7.495185613632202
    },
    {
      "epoch": 6.749343872070313e-05,
      "model_forward_time": 1.7203209400177002,
      "step": 70772
    },
    {
      "epoch": 6.749343872070313e-05,
      "step": 70772,
      "training_step_time": 7.852594375610352
    },
    {
      "epoch": 6.749439239501953e-05,
      "model_forward_time": 1.723229169845581,
      "step": 70773
    },
    {
      "epoch": 6.749439239501953e-05,
      "step": 70773,
      "training_step_time": 7.789367914199829
    },
    {
      "epoch": 6.749534606933594e-05,
      "model_forward_time": 1.7247531414031982,
      "step": 70774
    },
    {
      "epoch": 6.749534606933594e-05,
      "step": 70774,
      "training_step_time": 7.564374685287476
    },
    {
      "epoch": 6.749629974365235e-05,
      "model_forward_time": 1.7243359088897705,
      "step": 70775
    },
    {
      "epoch": 6.749629974365235e-05,
      "step": 70775,
      "training_step_time": 8.822202920913696
    },
    {
      "epoch": 6.749725341796875e-05,
      "model_forward_time": 1.7420811653137207,
      "step": 70776
    },
    {
      "epoch": 6.749725341796875e-05,
      "step": 70776,
      "training_step_time": 9.46733283996582
    },
    {
      "epoch": 6.749820709228516e-05,
      "model_forward_time": 1.7178819179534912,
      "step": 70777
    },
    {
      "epoch": 6.749820709228516e-05,
      "step": 70777,
      "training_step_time": 8.630431652069092
    },
    {
      "epoch": 6.749916076660157e-05,
      "model_forward_time": 1.7352840900421143,
      "step": 70778
    },
    {
      "epoch": 6.749916076660157e-05,
      "step": 70778,
      "training_step_time": 8.724762439727783
    },
    {
      "epoch": 6.750011444091797e-05,
      "model_forward_time": 1.3449947834014893,
      "step": 70779
    },
    {
      "epoch": 6.750011444091797e-05,
      "step": 70779,
      "training_step_time": 7.72732138633728
    },
    {
      "epoch": 6.750106811523438e-05,
      "grad_norm": 0.09288935363292694,
      "learning_rate": 2.1583470381980315e-06,
      "loss": 0.054,
      "step": 70780
    },
    {
      "epoch": 6.750106811523438e-05,
      "model_forward_time": 1.7305431365966797,
      "step": 70780
    },
    {
      "dynamics_loss_avg": 0.07049623876810074,
      "epoch": 6.750106811523438e-05,
      "step": 70780
    },
    {
      "action_loss_avg": 0.0014950269774999469,
      "epoch": 6.750106811523438e-05,
      "step": 70780
    },
    {
      "epoch": 6.750106811523438e-05,
      "step": 70780,
      "training_step_time": 7.730271339416504
    },
    {
      "epoch": 6.750202178955079e-05,
      "model_forward_time": 1.7217488288879395,
      "step": 70781
    },
    {
      "epoch": 6.750202178955079e-05,
      "step": 70781,
      "training_step_time": 7.922636032104492
    },
    {
      "epoch": 6.750297546386718e-05,
      "model_forward_time": 1.724477767944336,
      "step": 70782
    },
    {
      "epoch": 6.750297546386718e-05,
      "step": 70782,
      "training_step_time": 7.77262806892395
    },
    {
      "epoch": 6.75039291381836e-05,
      "model_forward_time": 1.7327992916107178,
      "step": 70783
    },
    {
      "epoch": 6.75039291381836e-05,
      "step": 70783,
      "training_step_time": 7.516008615493774
    },
    {
      "epoch": 6.75048828125e-05,
      "model_forward_time": 1.7217681407928467,
      "step": 70784
    },
    {
      "epoch": 6.75048828125e-05,
      "step": 70784,
      "training_step_time": 9.28786849975586
    },
    {
      "epoch": 6.75058364868164e-05,
      "model_forward_time": 2.265432357788086,
      "step": 70785
    },
    {
      "epoch": 6.75058364868164e-05,
      "step": 70785,
      "training_step_time": 10.4266197681427
    },
    {
      "epoch": 6.750679016113281e-05,
      "model_forward_time": 1.7194371223449707,
      "step": 70786
    },
    {
      "epoch": 6.750679016113281e-05,
      "step": 70786,
      "training_step_time": 9.269538879394531
    },
    {
      "epoch": 6.750774383544922e-05,
      "model_forward_time": 1.725893497467041,
      "step": 70787
    },
    {
      "epoch": 6.750774383544922e-05,
      "step": 70787,
      "training_step_time": 8.511799573898315
    },
    {
      "epoch": 6.750869750976562e-05,
      "model_forward_time": 1.7261223793029785,
      "step": 70788
    },
    {
      "epoch": 6.750869750976562e-05,
      "step": 70788,
      "training_step_time": 7.671278953552246
    },
    {
      "epoch": 6.750965118408203e-05,
      "model_forward_time": 1.725522756576538,
      "step": 70789
    },
    {
      "epoch": 6.750965118408203e-05,
      "step": 70789,
      "training_step_time": 8.366114854812622
    },
    {
      "epoch": 6.751060485839844e-05,
      "grad_norm": 0.07547774165868759,
      "learning_rate": 2.156986718136123e-06,
      "loss": 0.0575,
      "step": 70790
    },
    {
      "epoch": 6.751060485839844e-05,
      "model_forward_time": 1.725780963897705,
      "step": 70790
    },
    {
      "dynamics_loss_avg": 0.0665402403101325,
      "epoch": 6.751060485839844e-05,
      "step": 70790
    },
    {
      "action_loss_avg": 0.001774504641070962,
      "epoch": 6.751060485839844e-05,
      "step": 70790
    },
    {
      "epoch": 6.751060485839844e-05,
      "step": 70790,
      "training_step_time": 7.998545169830322
    },
    {
      "epoch": 6.751155853271484e-05,
      "model_forward_time": 1.7297251224517822,
      "step": 70791
    },
    {
      "epoch": 6.751155853271484e-05,
      "step": 70791,
      "training_step_time": 7.6409478187561035
    },
    {
      "epoch": 6.751251220703125e-05,
      "model_forward_time": 1.7201972007751465,
      "step": 70792
    },
    {
      "epoch": 6.751251220703125e-05,
      "step": 70792,
      "training_step_time": 7.5306010246276855
    },
    {
      "epoch": 6.751346588134766e-05,
      "model_forward_time": 1.7189314365386963,
      "step": 70793
    },
    {
      "epoch": 6.751346588134766e-05,
      "step": 70793,
      "training_step_time": 8.791867733001709
    },
    {
      "epoch": 6.751441955566406e-05,
      "model_forward_time": 2.2775611877441406,
      "step": 70794
    },
    {
      "epoch": 6.751441955566406e-05,
      "step": 70794,
      "training_step_time": 9.525128364562988
    },
    {
      "epoch": 6.751537322998047e-05,
      "model_forward_time": 1.803687572479248,
      "step": 70795
    },
    {
      "epoch": 6.751537322998047e-05,
      "step": 70795,
      "training_step_time": 8.60921335220337
    },
    {
      "epoch": 6.751632690429688e-05,
      "model_forward_time": 1.7230207920074463,
      "step": 70796
    },
    {
      "epoch": 6.751632690429688e-05,
      "step": 70796,
      "training_step_time": 7.862439155578613
    },
    {
      "epoch": 6.751728057861328e-05,
      "model_forward_time": 1.7283849716186523,
      "step": 70797
    },
    {
      "epoch": 6.751728057861328e-05,
      "step": 70797,
      "training_step_time": 7.709568738937378
    },
    {
      "epoch": 6.751823425292969e-05,
      "model_forward_time": 1.7315711975097656,
      "step": 70798
    },
    {
      "epoch": 6.751823425292969e-05,
      "step": 70798,
      "training_step_time": 7.637794733047485
    },
    {
      "epoch": 6.75191879272461e-05,
      "model_forward_time": 1.7109508514404297,
      "step": 70799
    },
    {
      "epoch": 6.75191879272461e-05,
      "step": 70799,
      "training_step_time": 7.438162803649902
    },
    {
      "epoch": 6.75201416015625e-05,
      "grad_norm": 0.09701074659824371,
      "learning_rate": 2.155626708981871e-06,
      "loss": 0.0545,
      "step": 70800
    },
    {
      "epoch": 6.75201416015625e-05,
      "model_forward_time": 1.7253761291503906,
      "step": 70800
    },
    {
      "dynamics_loss_avg": 0.06320880353450775,
      "epoch": 6.75201416015625e-05,
      "step": 70800
    },
    {
      "action_loss_avg": 0.0014274602406658232,
      "epoch": 6.75201416015625e-05,
      "step": 70800
    },
    {
      "epoch": 6.75201416015625e-05,
      "step": 70800,
      "training_step_time": 7.803438425064087
    },
    {
      "epoch": 6.752109527587891e-05,
      "model_forward_time": 1.7342865467071533,
      "step": 70801
    },
    {
      "epoch": 6.752109527587891e-05,
      "step": 70801,
      "training_step_time": 7.49058985710144
    },
    {
      "epoch": 6.752204895019532e-05,
      "model_forward_time": 1.73398756980896,
      "step": 70802
    },
    {
      "epoch": 6.752204895019532e-05,
      "step": 70802,
      "training_step_time": 7.711530685424805
    },
    {
      "epoch": 6.752300262451172e-05,
      "model_forward_time": 2.278291702270508,
      "step": 70803
    },
    {
      "epoch": 6.752300262451172e-05,
      "step": 70803,
      "training_step_time": 8.58668828010559
    },
    {
      "epoch": 6.752395629882813e-05,
      "model_forward_time": 1.778785228729248,
      "step": 70804
    },
    {
      "epoch": 6.752395629882813e-05,
      "step": 70804,
      "training_step_time": 8.914872646331787
    },
    {
      "epoch": 6.752490997314454e-05,
      "model_forward_time": 1.7141015529632568,
      "step": 70805
    },
    {
      "epoch": 6.752490997314454e-05,
      "step": 70805,
      "training_step_time": 7.890530109405518
    },
    {
      "epoch": 6.752586364746094e-05,
      "model_forward_time": 1.7246627807617188,
      "step": 70806
    },
    {
      "epoch": 6.752586364746094e-05,
      "step": 70806,
      "training_step_time": 7.807556390762329
    },
    {
      "epoch": 6.752681732177735e-05,
      "model_forward_time": 1.7196733951568604,
      "step": 70807
    },
    {
      "epoch": 6.752681732177735e-05,
      "step": 70807,
      "training_step_time": 7.752396106719971
    },
    {
      "epoch": 6.752777099609376e-05,
      "model_forward_time": 1.7138152122497559,
      "step": 70808
    },
    {
      "epoch": 6.752777099609376e-05,
      "step": 70808,
      "training_step_time": 8.221171140670776
    },
    {
      "epoch": 6.752872467041015e-05,
      "model_forward_time": 1.7253479957580566,
      "step": 70809
    },
    {
      "epoch": 6.752872467041015e-05,
      "step": 70809,
      "training_step_time": 8.526789426803589
    },
    {
      "epoch": 6.752967834472656e-05,
      "grad_norm": 0.08595910668373108,
      "learning_rate": 2.1542670108840075e-06,
      "loss": 0.055,
      "step": 70810
    },
    {
      "epoch": 6.752967834472656e-05,
      "model_forward_time": 1.7096881866455078,
      "step": 70810
    },
    {
      "dynamics_loss_avg": 0.05059224758297205,
      "epoch": 6.752967834472656e-05,
      "step": 70810
    },
    {
      "action_loss_avg": 0.0014690798736410216,
      "epoch": 6.752967834472656e-05,
      "step": 70810
    },
    {
      "epoch": 6.752967834472656e-05,
      "step": 70810,
      "training_step_time": 7.826348781585693
    },
    {
      "epoch": 6.753063201904298e-05,
      "model_forward_time": 1.7190062999725342,
      "step": 70811
    },
    {
      "epoch": 6.753063201904298e-05,
      "step": 70811,
      "training_step_time": 9.099382162094116
    },
    {
      "epoch": 6.753158569335937e-05,
      "model_forward_time": 2.2714731693267822,
      "step": 70812
    },
    {
      "epoch": 6.753158569335937e-05,
      "step": 70812,
      "training_step_time": 7.713390350341797
    },
    {
      "epoch": 6.753253936767578e-05,
      "model_forward_time": 1.772752285003662,
      "step": 70813
    },
    {
      "epoch": 6.753253936767578e-05,
      "step": 70813,
      "training_step_time": 9.061724424362183
    },
    {
      "epoch": 6.753349304199218e-05,
      "model_forward_time": 1.7215733528137207,
      "step": 70814
    },
    {
      "epoch": 6.753349304199218e-05,
      "step": 70814,
      "training_step_time": 8.717438459396362
    },
    {
      "epoch": 6.753444671630859e-05,
      "model_forward_time": 1.7375361919403076,
      "step": 70815
    },
    {
      "epoch": 6.753444671630859e-05,
      "step": 70815,
      "training_step_time": 7.746078729629517
    },
    {
      "epoch": 6.7535400390625e-05,
      "model_forward_time": 1.7380671501159668,
      "step": 70816
    },
    {
      "epoch": 6.7535400390625e-05,
      "step": 70816,
      "training_step_time": 7.774052858352661
    },
    {
      "epoch": 6.75363540649414e-05,
      "model_forward_time": 1.733490228652954,
      "step": 70817
    },
    {
      "epoch": 6.75363540649414e-05,
      "step": 70817,
      "training_step_time": 7.679972171783447
    },
    {
      "epoch": 6.753730773925781e-05,
      "model_forward_time": 1.700286865234375,
      "step": 70818
    },
    {
      "epoch": 6.753730773925781e-05,
      "step": 70818,
      "training_step_time": 7.988800287246704
    },
    {
      "epoch": 6.753826141357422e-05,
      "model_forward_time": 1.7056126594543457,
      "step": 70819
    },
    {
      "epoch": 6.753826141357422e-05,
      "step": 70819,
      "training_step_time": 7.802822113037109
    },
    {
      "epoch": 6.753921508789062e-05,
      "grad_norm": 0.09798546135425568,
      "learning_rate": 2.1529076239912212e-06,
      "loss": 0.0531,
      "step": 70820
    },
    {
      "epoch": 6.753921508789062e-05,
      "model_forward_time": 1.7085165977478027,
      "step": 70820
    },
    {
      "dynamics_loss_avg": 0.06707186140120029,
      "epoch": 6.753921508789062e-05,
      "step": 70820
    },
    {
      "action_loss_avg": 0.0015091995519469492,
      "epoch": 6.753921508789062e-05,
      "step": 70820
    },
    {
      "epoch": 6.753921508789062e-05,
      "step": 70820,
      "training_step_time": 10.404748678207397
    },
    {
      "epoch": 6.754016876220703e-05,
      "model_forward_time": 2.238818645477295,
      "step": 70821
    },
    {
      "epoch": 6.754016876220703e-05,
      "step": 70821,
      "training_step_time": 7.47524881362915
    },
    {
      "epoch": 6.754112243652344e-05,
      "model_forward_time": 1.7645773887634277,
      "step": 70822
    },
    {
      "epoch": 6.754112243652344e-05,
      "step": 70822,
      "training_step_time": 9.65336012840271
    },
    {
      "epoch": 6.754207611083984e-05,
      "model_forward_time": 1.6952927112579346,
      "step": 70823
    },
    {
      "epoch": 6.754207611083984e-05,
      "step": 70823,
      "training_step_time": 8.010326147079468
    },
    {
      "epoch": 6.754302978515625e-05,
      "model_forward_time": 1.7073407173156738,
      "step": 70824
    },
    {
      "epoch": 6.754302978515625e-05,
      "step": 70824,
      "training_step_time": 9.158662557601929
    },
    {
      "epoch": 6.754398345947266e-05,
      "model_forward_time": 1.7118573188781738,
      "step": 70825
    },
    {
      "epoch": 6.754398345947266e-05,
      "step": 70825,
      "training_step_time": 7.86434268951416
    },
    {
      "epoch": 6.754493713378906e-05,
      "model_forward_time": 1.7106647491455078,
      "step": 70826
    },
    {
      "epoch": 6.754493713378906e-05,
      "step": 70826,
      "training_step_time": 7.710070371627808
    },
    {
      "epoch": 6.754589080810547e-05,
      "model_forward_time": 1.7149524688720703,
      "step": 70827
    },
    {
      "epoch": 6.754589080810547e-05,
      "step": 70827,
      "training_step_time": 8.021442651748657
    },
    {
      "epoch": 6.754684448242188e-05,
      "model_forward_time": 1.7110347747802734,
      "step": 70828
    },
    {
      "epoch": 6.754684448242188e-05,
      "step": 70828,
      "training_step_time": 7.7818543910980225
    },
    {
      "epoch": 6.754779815673828e-05,
      "model_forward_time": 1.7055118083953857,
      "step": 70829
    },
    {
      "epoch": 6.754779815673828e-05,
      "step": 70829,
      "training_step_time": 8.351827144622803
    },
    {
      "epoch": 6.754875183105469e-05,
      "grad_norm": 0.0983138233423233,
      "learning_rate": 2.1515485484521805e-06,
      "loss": 0.0537,
      "step": 70830
    },
    {
      "epoch": 6.754875183105469e-05,
      "model_forward_time": 1.7128498554229736,
      "step": 70830
    },
    {
      "dynamics_loss_avg": 0.052634593285620215,
      "epoch": 6.754875183105469e-05,
      "step": 70830
    },
    {
      "action_loss_avg": 0.0014733950316440313,
      "epoch": 6.754875183105469e-05,
      "step": 70830
    },
    {
      "epoch": 6.754875183105469e-05,
      "step": 70830,
      "training_step_time": 8.382635116577148
    },
    {
      "epoch": 6.75497055053711e-05,
      "model_forward_time": 2.2484476566314697,
      "step": 70831
    },
    {
      "epoch": 6.75497055053711e-05,
      "step": 70831,
      "training_step_time": 7.793262243270874
    },
    {
      "epoch": 6.75506591796875e-05,
      "model_forward_time": 1.7063617706298828,
      "step": 70832
    },
    {
      "epoch": 6.75506591796875e-05,
      "step": 70832,
      "training_step_time": 7.9915993213653564
    },
    {
      "epoch": 6.75516128540039e-05,
      "model_forward_time": 1.725046157836914,
      "step": 70833
    },
    {
      "epoch": 6.75516128540039e-05,
      "step": 70833,
      "training_step_time": 8.556774377822876
    },
    {
      "epoch": 6.755256652832032e-05,
      "model_forward_time": 1.7287852764129639,
      "step": 70834
    },
    {
      "epoch": 6.755256652832032e-05,
      "step": 70834,
      "training_step_time": 7.8347485065460205
    },
    {
      "epoch": 6.755352020263671e-05,
      "model_forward_time": 1.7533504962921143,
      "step": 70835
    },
    {
      "epoch": 6.755352020263671e-05,
      "step": 70835,
      "training_step_time": 7.743894577026367
    },
    {
      "epoch": 6.755447387695312e-05,
      "model_forward_time": 1.7315030097961426,
      "step": 70836
    },
    {
      "epoch": 6.755447387695312e-05,
      "step": 70836,
      "training_step_time": 7.751858472824097
    },
    {
      "epoch": 6.755542755126953e-05,
      "model_forward_time": 1.7359836101531982,
      "step": 70837
    },
    {
      "epoch": 6.755542755126953e-05,
      "step": 70837,
      "training_step_time": 7.75091028213501
    },
    {
      "epoch": 6.755638122558593e-05,
      "model_forward_time": 1.7419769763946533,
      "step": 70838
    },
    {
      "epoch": 6.755638122558593e-05,
      "step": 70838,
      "training_step_time": 7.825721979141235
    },
    {
      "epoch": 6.755733489990234e-05,
      "model_forward_time": 1.7296714782714844,
      "step": 70839
    },
    {
      "epoch": 6.755733489990234e-05,
      "step": 70839,
      "training_step_time": 8.461063385009766
    },
    {
      "epoch": 6.755828857421875e-05,
      "grad_norm": 0.14901185035705566,
      "learning_rate": 2.150189784415506e-06,
      "loss": 0.0549,
      "step": 70840
    },
    {
      "epoch": 6.755828857421875e-05,
      "model_forward_time": 2.288477659225464,
      "step": 70840
    },
    {
      "dynamics_loss_avg": 0.055879838019609454,
      "epoch": 6.755828857421875e-05,
      "step": 70840
    },
    {
      "action_loss_avg": 0.001370554813183844,
      "epoch": 6.755828857421875e-05,
      "step": 70840
    },
    {
      "epoch": 6.755828857421875e-05,
      "step": 70840,
      "training_step_time": 7.708164691925049
    },
    {
      "epoch": 6.755924224853515e-05,
      "model_forward_time": 1.814039707183838,
      "step": 70841
    },
    {
      "epoch": 6.755924224853515e-05,
      "step": 70841,
      "training_step_time": 8.525360584259033
    },
    {
      "epoch": 6.756019592285156e-05,
      "model_forward_time": 1.7435531616210938,
      "step": 70842
    },
    {
      "epoch": 6.756019592285156e-05,
      "step": 70842,
      "training_step_time": 8.71404218673706
    },
    {
      "epoch": 6.756114959716797e-05,
      "model_forward_time": 1.73876953125,
      "step": 70843
    },
    {
      "epoch": 6.756114959716797e-05,
      "step": 70843,
      "training_step_time": 8.722564220428467
    },
    {
      "epoch": 6.756210327148437e-05,
      "model_forward_time": 1.7421584129333496,
      "step": 70844
    },
    {
      "epoch": 6.756210327148437e-05,
      "step": 70844,
      "training_step_time": 7.501884698867798
    },
    {
      "epoch": 6.756305694580078e-05,
      "model_forward_time": 1.740112066268921,
      "step": 70845
    },
    {
      "epoch": 6.756305694580078e-05,
      "step": 70845,
      "training_step_time": 7.587897062301636
    },
    {
      "epoch": 6.756401062011719e-05,
      "model_forward_time": 1.7341547012329102,
      "step": 70846
    },
    {
      "epoch": 6.756401062011719e-05,
      "step": 70846,
      "training_step_time": 7.668471336364746
    },
    {
      "epoch": 6.756496429443359e-05,
      "model_forward_time": 1.7333614826202393,
      "step": 70847
    },
    {
      "epoch": 6.756496429443359e-05,
      "step": 70847,
      "training_step_time": 7.7914040088653564
    },
    {
      "epoch": 6.756591796875e-05,
      "model_forward_time": 1.7308003902435303,
      "step": 70848
    },
    {
      "epoch": 6.756591796875e-05,
      "step": 70848,
      "training_step_time": 8.750640392303467
    },
    {
      "epoch": 6.756687164306641e-05,
      "model_forward_time": 2.2801926136016846,
      "step": 70849
    },
    {
      "epoch": 6.756687164306641e-05,
      "step": 70849,
      "training_step_time": 8.20885682106018
    },
    {
      "epoch": 6.756782531738281e-05,
      "grad_norm": 0.09097820520401001,
      "learning_rate": 2.148831332029794e-06,
      "loss": 0.0525,
      "step": 70850
    },
    {
      "epoch": 6.756782531738281e-05,
      "model_forward_time": 1.8122777938842773,
      "step": 70850
    },
    {
      "dynamics_loss_avg": 0.051481925509870054,
      "epoch": 6.756782531738281e-05,
      "step": 70850
    },
    {
      "action_loss_avg": 0.0014319675654405728,
      "epoch": 6.756782531738281e-05,
      "step": 70850
    },
    {
      "epoch": 6.756782531738281e-05,
      "step": 70850,
      "training_step_time": 9.475184440612793
    },
    {
      "epoch": 6.756877899169922e-05,
      "model_forward_time": 1.7280924320220947,
      "step": 70851
    },
    {
      "epoch": 6.756877899169922e-05,
      "step": 70851,
      "training_step_time": 7.838081121444702
    },
    {
      "epoch": 6.756973266601563e-05,
      "model_forward_time": 1.7400486469268799,
      "step": 70852
    },
    {
      "epoch": 6.756973266601563e-05,
      "step": 70852,
      "training_step_time": 7.7470619678497314
    },
    {
      "epoch": 6.757068634033203e-05,
      "model_forward_time": 1.7493493556976318,
      "step": 70853
    },
    {
      "epoch": 6.757068634033203e-05,
      "step": 70853,
      "training_step_time": 7.671136379241943
    },
    {
      "epoch": 6.757164001464844e-05,
      "model_forward_time": 1.733673095703125,
      "step": 70854
    },
    {
      "epoch": 6.757164001464844e-05,
      "step": 70854,
      "training_step_time": 6.932987928390503
    },
    {
      "epoch": 6.757259368896485e-05,
      "model_forward_time": 1.731935739517212,
      "step": 70855
    },
    {
      "epoch": 6.757259368896485e-05,
      "step": 70855,
      "training_step_time": 7.743107318878174
    },
    {
      "epoch": 6.757354736328124e-05,
      "model_forward_time": 1.7478227615356445,
      "step": 70856
    },
    {
      "epoch": 6.757354736328124e-05,
      "step": 70856,
      "training_step_time": 7.828398942947388
    },
    {
      "epoch": 6.757450103759766e-05,
      "model_forward_time": 1.7424876689910889,
      "step": 70857
    },
    {
      "epoch": 6.757450103759766e-05,
      "step": 70857,
      "training_step_time": 9.108773469924927
    },
    {
      "epoch": 6.757545471191407e-05,
      "model_forward_time": 2.326481342315674,
      "step": 70858
    },
    {
      "epoch": 6.757545471191407e-05,
      "step": 70858,
      "training_step_time": 8.72366452217102
    },
    {
      "epoch": 6.757640838623046e-05,
      "model_forward_time": 1.790517807006836,
      "step": 70859
    },
    {
      "epoch": 6.757640838623046e-05,
      "step": 70859,
      "training_step_time": 10.638501167297363
    },
    {
      "epoch": 6.757736206054687e-05,
      "grad_norm": 0.10969773679971695,
      "learning_rate": 2.1474731914435992e-06,
      "loss": 0.0538,
      "step": 70860
    },
    {
      "epoch": 6.757736206054687e-05,
      "model_forward_time": 1.747687578201294,
      "step": 70860
    },
    {
      "dynamics_loss_avg": 0.06375213563442231,
      "epoch": 6.757736206054687e-05,
      "step": 70860
    },
    {
      "action_loss_avg": 0.0012162069848272949,
      "epoch": 6.757736206054687e-05,
      "step": 70860
    },
    {
      "epoch": 6.757736206054687e-05,
      "step": 70860,
      "training_step_time": 7.860299825668335
    },
    {
      "epoch": 6.757831573486329e-05,
      "model_forward_time": 1.7338132858276367,
      "step": 70861
    },
    {
      "epoch": 6.757831573486329e-05,
      "step": 70861,
      "training_step_time": 8.30313754081726
    },
    {
      "epoch": 6.757926940917968e-05,
      "model_forward_time": 1.7472643852233887,
      "step": 70862
    },
    {
      "epoch": 6.757926940917968e-05,
      "step": 70862,
      "training_step_time": 7.576104640960693
    },
    {
      "epoch": 6.75802230834961e-05,
      "model_forward_time": 1.7442796230316162,
      "step": 70863
    },
    {
      "epoch": 6.75802230834961e-05,
      "step": 70863,
      "training_step_time": 7.664266347885132
    },
    {
      "epoch": 6.75811767578125e-05,
      "model_forward_time": 1.7022411823272705,
      "step": 70864
    },
    {
      "epoch": 6.75811767578125e-05,
      "step": 70864,
      "training_step_time": 7.821329116821289
    },
    {
      "epoch": 6.75821304321289e-05,
      "model_forward_time": 1.7165019512176514,
      "step": 70865
    },
    {
      "epoch": 6.75821304321289e-05,
      "step": 70865,
      "training_step_time": 7.807331562042236
    },
    {
      "epoch": 6.758308410644531e-05,
      "model_forward_time": 1.7083039283752441,
      "step": 70866
    },
    {
      "epoch": 6.758308410644531e-05,
      "step": 70866,
      "training_step_time": 8.088142156600952
    },
    {
      "epoch": 6.758403778076172e-05,
      "model_forward_time": 2.240321397781372,
      "step": 70867
    },
    {
      "epoch": 6.758403778076172e-05,
      "step": 70867,
      "training_step_time": 7.771672010421753
    },
    {
      "epoch": 6.758499145507812e-05,
      "model_forward_time": 1.7363619804382324,
      "step": 70868
    },
    {
      "epoch": 6.758499145507812e-05,
      "step": 70868,
      "training_step_time": 9.324874639511108
    },
    {
      "epoch": 6.758594512939453e-05,
      "model_forward_time": 1.7115275859832764,
      "step": 70869
    },
    {
      "epoch": 6.758594512939453e-05,
      "step": 70869,
      "training_step_time": 8.002134561538696
    },
    {
      "epoch": 6.758689880371094e-05,
      "grad_norm": 0.12137022614479065,
      "learning_rate": 2.1461153628054493e-06,
      "loss": 0.053,
      "step": 70870
    },
    {
      "epoch": 6.758689880371094e-05,
      "model_forward_time": 1.7169160842895508,
      "step": 70870
    },
    {
      "dynamics_loss_avg": 0.07623390685766936,
      "epoch": 6.758689880371094e-05,
      "step": 70870
    },
    {
      "action_loss_avg": 0.001991199099575169,
      "epoch": 6.758689880371094e-05,
      "step": 70870
    },
    {
      "epoch": 6.758689880371094e-05,
      "step": 70870,
      "training_step_time": 7.810428619384766
    },
    {
      "epoch": 6.758785247802734e-05,
      "model_forward_time": 1.7098884582519531,
      "step": 70871
    },
    {
      "epoch": 6.758785247802734e-05,
      "step": 70871,
      "training_step_time": 7.571916818618774
    },
    {
      "epoch": 6.758880615234375e-05,
      "model_forward_time": 1.715195894241333,
      "step": 70872
    },
    {
      "epoch": 6.758880615234375e-05,
      "step": 70872,
      "training_step_time": 7.699854135513306
    },
    {
      "epoch": 6.758975982666016e-05,
      "model_forward_time": 1.7101972103118896,
      "step": 70873
    },
    {
      "epoch": 6.758975982666016e-05,
      "step": 70873,
      "training_step_time": 7.784360647201538
    },
    {
      "epoch": 6.759071350097656e-05,
      "model_forward_time": 1.7079756259918213,
      "step": 70874
    },
    {
      "epoch": 6.759071350097656e-05,
      "step": 70874,
      "training_step_time": 8.86109972000122
    },
    {
      "epoch": 6.759166717529297e-05,
      "model_forward_time": 1.7090072631835938,
      "step": 70875
    },
    {
      "epoch": 6.759166717529297e-05,
      "step": 70875,
      "training_step_time": 7.71120285987854
    },
    {
      "epoch": 6.759262084960938e-05,
      "model_forward_time": 1.7175333499908447,
      "step": 70876
    },
    {
      "epoch": 6.759262084960938e-05,
      "step": 70876,
      "training_step_time": 8.750568628311157
    },
    {
      "epoch": 6.759357452392578e-05,
      "model_forward_time": 1.71626615524292,
      "step": 70877
    },
    {
      "epoch": 6.759357452392578e-05,
      "step": 70877,
      "training_step_time": 9.602099180221558
    },
    {
      "epoch": 6.759452819824219e-05,
      "model_forward_time": 1.8048298358917236,
      "step": 70878
    },
    {
      "epoch": 6.759452819824219e-05,
      "step": 70878,
      "training_step_time": 10.20371961593628
    },
    {
      "epoch": 6.75954818725586e-05,
      "model_forward_time": 1.733034610748291,
      "step": 70879
    },
    {
      "epoch": 6.75954818725586e-05,
      "step": 70879,
      "training_step_time": 7.763137578964233
    },
    {
      "epoch": 6.7596435546875e-05,
      "grad_norm": 0.09583788365125656,
      "learning_rate": 2.1447578462638287e-06,
      "loss": 0.0538,
      "step": 70880
    },
    {
      "epoch": 6.7596435546875e-05,
      "model_forward_time": 1.72861647605896,
      "step": 70880
    },
    {
      "dynamics_loss_avg": 0.06261624023318291,
      "epoch": 6.7596435546875e-05,
      "step": 70880
    },
    {
      "action_loss_avg": 0.0018844476115191355,
      "epoch": 6.7596435546875e-05,
      "step": 70880
    },
    {
      "epoch": 6.7596435546875e-05,
      "step": 70880,
      "training_step_time": 7.80533504486084
    },
    {
      "epoch": 6.75973892211914e-05,
      "model_forward_time": 1.7309858798980713,
      "step": 70881
    },
    {
      "epoch": 6.75973892211914e-05,
      "step": 70881,
      "training_step_time": 7.702415466308594
    },
    {
      "epoch": 6.759834289550782e-05,
      "model_forward_time": 1.7471094131469727,
      "step": 70882
    },
    {
      "epoch": 6.759834289550782e-05,
      "step": 70882,
      "training_step_time": 8.307538986206055
    },
    {
      "epoch": 6.759929656982421e-05,
      "model_forward_time": 1.730968952178955,
      "step": 70883
    },
    {
      "epoch": 6.759929656982421e-05,
      "step": 70883,
      "training_step_time": 8.611208438873291
    },
    {
      "epoch": 6.760025024414063e-05,
      "model_forward_time": 1.3609960079193115,
      "step": 70884
    },
    {
      "epoch": 6.760025024414063e-05,
      "step": 70884,
      "training_step_time": 7.731558322906494
    },
    {
      "epoch": 6.760120391845704e-05,
      "model_forward_time": 1.7343683242797852,
      "step": 70885
    },
    {
      "epoch": 6.760120391845704e-05,
      "step": 70885,
      "training_step_time": 7.632983922958374
    },
    {
      "epoch": 6.760215759277343e-05,
      "model_forward_time": 2.2990379333496094,
      "step": 70886
    },
    {
      "epoch": 6.760215759277343e-05,
      "step": 70886,
      "training_step_time": 10.86794924736023
    },
    {
      "epoch": 6.760311126708984e-05,
      "model_forward_time": 1.8168883323669434,
      "step": 70887
    },
    {
      "epoch": 6.760311126708984e-05,
      "step": 70887,
      "training_step_time": 9.822854042053223
    },
    {
      "epoch": 6.760406494140625e-05,
      "model_forward_time": 1.733349323272705,
      "step": 70888
    },
    {
      "epoch": 6.760406494140625e-05,
      "step": 70888,
      "training_step_time": 8.712029218673706
    },
    {
      "epoch": 6.760501861572265e-05,
      "model_forward_time": 1.740189552307129,
      "step": 70889
    },
    {
      "epoch": 6.760501861572265e-05,
      "step": 70889,
      "training_step_time": 7.778860569000244
    },
    {
      "epoch": 6.760597229003906e-05,
      "grad_norm": 0.08014963567256927,
      "learning_rate": 2.143400641967201e-06,
      "loss": 0.0539,
      "step": 70890
    },
    {
      "epoch": 6.760597229003906e-05,
      "model_forward_time": 1.7473385334014893,
      "step": 70890
    },
    {
      "dynamics_loss_avg": 0.049604186415672304,
      "epoch": 6.760597229003906e-05,
      "step": 70890
    },
    {
      "action_loss_avg": 0.0008327180272317492,
      "epoch": 6.760597229003906e-05,
      "step": 70890
    },
    {
      "epoch": 6.760597229003906e-05,
      "step": 70890,
      "training_step_time": 8.616148710250854
    },
    {
      "epoch": 6.760692596435547e-05,
      "model_forward_time": 1.7336959838867188,
      "step": 70891
    },
    {
      "epoch": 6.760692596435547e-05,
      "step": 70891,
      "training_step_time": 7.631039142608643
    },
    {
      "epoch": 6.760787963867187e-05,
      "model_forward_time": 1.7316315174102783,
      "step": 70892
    },
    {
      "epoch": 6.760787963867187e-05,
      "step": 70892,
      "training_step_time": 8.193832635879517
    },
    {
      "epoch": 6.760883331298828e-05,
      "model_forward_time": 1.7269644737243652,
      "step": 70893
    },
    {
      "epoch": 6.760883331298828e-05,
      "step": 70893,
      "training_step_time": 9.27971339225769
    },
    {
      "epoch": 6.760978698730469e-05,
      "model_forward_time": 1.7392005920410156,
      "step": 70894
    },
    {
      "epoch": 6.760978698730469e-05,
      "step": 70894,
      "training_step_time": 7.601683139801025
    },
    {
      "epoch": 6.761074066162109e-05,
      "model_forward_time": 2.9098434448242188,
      "step": 70895
    },
    {
      "epoch": 6.761074066162109e-05,
      "step": 70895,
      "training_step_time": 7.776623010635376
    },
    {
      "epoch": 6.76116943359375e-05,
      "model_forward_time": 1.7401838302612305,
      "step": 70896
    },
    {
      "epoch": 6.76116943359375e-05,
      "step": 70896,
      "training_step_time": 8.932921409606934
    },
    {
      "epoch": 6.761264801025391e-05,
      "model_forward_time": 1.7379631996154785,
      "step": 70897
    },
    {
      "epoch": 6.761264801025391e-05,
      "step": 70897,
      "training_step_time": 7.931638240814209
    },
    {
      "epoch": 6.761360168457031e-05,
      "model_forward_time": 1.7393953800201416,
      "step": 70898
    },
    {
      "epoch": 6.761360168457031e-05,
      "step": 70898,
      "training_step_time": 7.7459025382995605
    },
    {
      "epoch": 6.761455535888672e-05,
      "model_forward_time": 1.7413737773895264,
      "step": 70899
    },
    {
      "epoch": 6.761455535888672e-05,
      "step": 70899,
      "training_step_time": 7.540860414505005
    },
    {
      "epoch": 6.761550903320313e-05,
      "grad_norm": 0.08358960598707199,
      "learning_rate": 2.1420437500639824e-06,
      "loss": 0.0554,
      "step": 70900
    },
    {
      "epoch": 6.761550903320313e-05,
      "model_forward_time": 1.725407600402832,
      "step": 70900
    },
    {
      "dynamics_loss_avg": 0.06045923456549644,
      "epoch": 6.761550903320313e-05,
      "step": 70900
    },
    {
      "action_loss_avg": 0.0026830315531697125,
      "epoch": 6.761550903320313e-05,
      "step": 70900
    },
    {
      "epoch": 6.761550903320313e-05,
      "step": 70900,
      "training_step_time": 7.714807987213135
    },
    {
      "epoch": 6.761646270751953e-05,
      "model_forward_time": 1.75657057762146,
      "step": 70901
    },
    {
      "epoch": 6.761646270751953e-05,
      "step": 70901,
      "training_step_time": 7.715254306793213
    },
    {
      "epoch": 6.761741638183594e-05,
      "model_forward_time": 1.8191237449645996,
      "step": 70902
    },
    {
      "epoch": 6.761741638183594e-05,
      "step": 70902,
      "training_step_time": 7.860002040863037
    },
    {
      "epoch": 6.761837005615235e-05,
      "model_forward_time": 1.7447788715362549,
      "step": 70903
    },
    {
      "epoch": 6.761837005615235e-05,
      "step": 70903,
      "training_step_time": 9.609642267227173
    },
    {
      "epoch": 6.761932373046875e-05,
      "model_forward_time": 0.9858841896057129,
      "step": 70904
    },
    {
      "epoch": 6.761932373046875e-05,
      "step": 70904,
      "training_step_time": 8.73340368270874
    },
    {
      "epoch": 6.762027740478516e-05,
      "model_forward_time": 1.7648484706878662,
      "step": 70905
    },
    {
      "epoch": 6.762027740478516e-05,
      "step": 70905,
      "training_step_time": 10.671639919281006
    },
    {
      "epoch": 6.762123107910157e-05,
      "model_forward_time": 1.7251429557800293,
      "step": 70906
    },
    {
      "epoch": 6.762123107910157e-05,
      "step": 70906,
      "training_step_time": 7.907343626022339
    },
    {
      "epoch": 6.762218475341797e-05,
      "model_forward_time": 1.7245368957519531,
      "step": 70907
    },
    {
      "epoch": 6.762218475341797e-05,
      "step": 70907,
      "training_step_time": 7.749876022338867
    },
    {
      "epoch": 6.762313842773438e-05,
      "model_forward_time": 1.7196474075317383,
      "step": 70908
    },
    {
      "epoch": 6.762313842773438e-05,
      "step": 70908,
      "training_step_time": 8.275226593017578
    },
    {
      "epoch": 6.762409210205079e-05,
      "model_forward_time": 1.722632646560669,
      "step": 70909
    },
    {
      "epoch": 6.762409210205079e-05,
      "step": 70909,
      "training_step_time": 7.658398151397705
    },
    {
      "epoch": 6.762504577636718e-05,
      "grad_norm": 0.08277828991413116,
      "learning_rate": 2.1406871707025646e-06,
      "loss": 0.0529,
      "step": 70910
    },
    {
      "epoch": 6.762504577636718e-05,
      "model_forward_time": 1.7203152179718018,
      "step": 70910
    },
    {
      "dynamics_loss_avg": 0.04920140653848648,
      "epoch": 6.762504577636718e-05,
      "step": 70910
    },
    {
      "action_loss_avg": 0.0012731922761304304,
      "epoch": 6.762504577636718e-05,
      "step": 70910
    },
    {
      "epoch": 6.762504577636718e-05,
      "step": 70910,
      "training_step_time": 7.7256951332092285
    },
    {
      "epoch": 6.76259994506836e-05,
      "model_forward_time": 1.7385766506195068,
      "step": 70911
    },
    {
      "epoch": 6.76259994506836e-05,
      "step": 70911,
      "training_step_time": 7.8279101848602295
    },
    {
      "epoch": 6.7626953125e-05,
      "model_forward_time": 1.7264175415039062,
      "step": 70912
    },
    {
      "epoch": 6.7626953125e-05,
      "step": 70912,
      "training_step_time": 7.6896750926971436
    },
    {
      "epoch": 6.76279067993164e-05,
      "model_forward_time": 1.7183892726898193,
      "step": 70913
    },
    {
      "epoch": 6.76279067993164e-05,
      "step": 70913,
      "training_step_time": 8.718013286590576
    },
    {
      "epoch": 6.762886047363281e-05,
      "model_forward_time": 2.307324171066284,
      "step": 70914
    },
    {
      "epoch": 6.762886047363281e-05,
      "step": 70914,
      "training_step_time": 8.299344539642334
    },
    {
      "epoch": 6.762981414794922e-05,
      "model_forward_time": 1.7203004360198975,
      "step": 70915
    },
    {
      "epoch": 6.762981414794922e-05,
      "step": 70915,
      "training_step_time": 7.963010549545288
    },
    {
      "epoch": 6.763076782226562e-05,
      "model_forward_time": 1.7283570766448975,
      "step": 70916
    },
    {
      "epoch": 6.763076782226562e-05,
      "step": 70916,
      "training_step_time": 7.763021945953369
    },
    {
      "epoch": 6.763172149658203e-05,
      "model_forward_time": 1.7229340076446533,
      "step": 70917
    },
    {
      "epoch": 6.763172149658203e-05,
      "step": 70917,
      "training_step_time": 7.811040878295898
    },
    {
      "epoch": 6.763267517089844e-05,
      "model_forward_time": 1.7295444011688232,
      "step": 70918
    },
    {
      "epoch": 6.763267517089844e-05,
      "step": 70918,
      "training_step_time": 7.692691802978516
    },
    {
      "epoch": 6.763362884521484e-05,
      "model_forward_time": 1.718505620956421,
      "step": 70919
    },
    {
      "epoch": 6.763362884521484e-05,
      "step": 70919,
      "training_step_time": 7.77554726600647
    },
    {
      "epoch": 6.763458251953125e-05,
      "grad_norm": 0.113304004073143,
      "learning_rate": 2.1393309040312965e-06,
      "loss": 0.0535,
      "step": 70920
    },
    {
      "epoch": 6.763458251953125e-05,
      "model_forward_time": 1.7294890880584717,
      "step": 70920
    },
    {
      "dynamics_loss_avg": 0.05739947408437729,
      "epoch": 6.763458251953125e-05,
      "step": 70920
    },
    {
      "action_loss_avg": 0.0015935581177473068,
      "epoch": 6.763458251953125e-05,
      "step": 70920
    },
    {
      "epoch": 6.763458251953125e-05,
      "step": 70920,
      "training_step_time": 7.8703453540802
    },
    {
      "epoch": 6.763553619384766e-05,
      "model_forward_time": 1.7289891242980957,
      "step": 70921
    },
    {
      "epoch": 6.763553619384766e-05,
      "step": 70921,
      "training_step_time": 7.805493116378784
    },
    {
      "epoch": 6.763648986816406e-05,
      "model_forward_time": 1.7243759632110596,
      "step": 70922
    },
    {
      "epoch": 6.763648986816406e-05,
      "step": 70922,
      "training_step_time": 7.707341194152832
    },
    {
      "epoch": 6.763744354248047e-05,
      "model_forward_time": 2.289677619934082,
      "step": 70923
    },
    {
      "epoch": 6.763744354248047e-05,
      "step": 70923,
      "training_step_time": 8.439632892608643
    },
    {
      "epoch": 6.763839721679688e-05,
      "model_forward_time": 1.8079252243041992,
      "step": 70924
    },
    {
      "epoch": 6.763839721679688e-05,
      "step": 70924,
      "training_step_time": 8.008761405944824
    },
    {
      "epoch": 6.763935089111328e-05,
      "model_forward_time": 1.7368083000183105,
      "step": 70925
    },
    {
      "epoch": 6.763935089111328e-05,
      "step": 70925,
      "training_step_time": 9.761890172958374
    },
    {
      "epoch": 6.764030456542969e-05,
      "model_forward_time": 1.7269935607910156,
      "step": 70926
    },
    {
      "epoch": 6.764030456542969e-05,
      "step": 70926,
      "training_step_time": 7.7458648681640625
    },
    {
      "epoch": 6.76412582397461e-05,
      "model_forward_time": 1.7290012836456299,
      "step": 70927
    },
    {
      "epoch": 6.76412582397461e-05,
      "step": 70927,
      "training_step_time": 7.687966346740723
    },
    {
      "epoch": 6.76422119140625e-05,
      "model_forward_time": 1.7262144088745117,
      "step": 70928
    },
    {
      "epoch": 6.76422119140625e-05,
      "step": 70928,
      "training_step_time": 7.737595796585083
    },
    {
      "epoch": 6.764316558837891e-05,
      "model_forward_time": 1.7215845584869385,
      "step": 70929
    },
    {
      "epoch": 6.764316558837891e-05,
      "step": 70929,
      "training_step_time": 8.050617218017578
    },
    {
      "epoch": 6.764411926269532e-05,
      "grad_norm": 0.08242633193731308,
      "learning_rate": 2.137974950198502e-06,
      "loss": 0.0562,
      "step": 70930
    },
    {
      "epoch": 6.764411926269532e-05,
      "model_forward_time": 1.730968952178955,
      "step": 70930
    },
    {
      "dynamics_loss_avg": 0.055979359149932864,
      "epoch": 6.764411926269532e-05,
      "step": 70930
    },
    {
      "action_loss_avg": 0.0012719024787656962,
      "epoch": 6.764411926269532e-05,
      "step": 70930
    },
    {
      "epoch": 6.764411926269532e-05,
      "step": 70930,
      "training_step_time": 7.912282466888428
    },
    {
      "epoch": 6.764507293701172e-05,
      "model_forward_time": 1.7167880535125732,
      "step": 70931
    },
    {
      "epoch": 6.764507293701172e-05,
      "step": 70931,
      "training_step_time": 7.7717249393463135
    },
    {
      "epoch": 6.764602661132813e-05,
      "model_forward_time": 2.2856452465057373,
      "step": 70932
    },
    {
      "epoch": 6.764602661132813e-05,
      "step": 70932,
      "training_step_time": 7.772619247436523
    },
    {
      "epoch": 6.764698028564454e-05,
      "model_forward_time": 1.7878062725067139,
      "step": 70933
    },
    {
      "epoch": 6.764698028564454e-05,
      "step": 70933,
      "training_step_time": 8.803572416305542
    },
    {
      "epoch": 6.764793395996094e-05,
      "model_forward_time": 1.7318840026855469,
      "step": 70934
    },
    {
      "epoch": 6.764793395996094e-05,
      "step": 70934,
      "training_step_time": 8.488504648208618
    },
    {
      "epoch": 6.764888763427735e-05,
      "model_forward_time": 1.7234525680541992,
      "step": 70935
    },
    {
      "epoch": 6.764888763427735e-05,
      "step": 70935,
      "training_step_time": 7.7440409660339355
    },
    {
      "epoch": 6.764984130859376e-05,
      "model_forward_time": 1.7186131477355957,
      "step": 70936
    },
    {
      "epoch": 6.764984130859376e-05,
      "step": 70936,
      "training_step_time": 7.48794150352478
    },
    {
      "epoch": 6.765079498291015e-05,
      "model_forward_time": 1.7309954166412354,
      "step": 70937
    },
    {
      "epoch": 6.765079498291015e-05,
      "step": 70937,
      "training_step_time": 7.683543682098389
    },
    {
      "epoch": 6.765174865722656e-05,
      "model_forward_time": 1.7287743091583252,
      "step": 70938
    },
    {
      "epoch": 6.765174865722656e-05,
      "step": 70938,
      "training_step_time": 7.713105201721191
    },
    {
      "epoch": 6.765270233154296e-05,
      "model_forward_time": 1.7186977863311768,
      "step": 70939
    },
    {
      "epoch": 6.765270233154296e-05,
      "step": 70939,
      "training_step_time": 7.775772571563721
    },
    {
      "epoch": 6.765365600585937e-05,
      "grad_norm": 0.09125714004039764,
      "learning_rate": 2.136619309352462e-06,
      "loss": 0.0526,
      "step": 70940
    },
    {
      "epoch": 6.765365600585937e-05,
      "model_forward_time": 1.7230033874511719,
      "step": 70940
    },
    {
      "dynamics_loss_avg": 0.05564767904579639,
      "epoch": 6.765365600585937e-05,
      "step": 70940
    },
    {
      "action_loss_avg": 0.001673282019328326,
      "epoch": 6.765365600585937e-05,
      "step": 70940
    },
    {
      "epoch": 6.765365600585937e-05,
      "step": 70940,
      "training_step_time": 7.655480146408081
    },
    {
      "epoch": 6.765460968017578e-05,
      "model_forward_time": 2.296685218811035,
      "step": 70941
    },
    {
      "epoch": 6.765460968017578e-05,
      "step": 70941,
      "training_step_time": 8.531816244125366
    },
    {
      "epoch": 6.765556335449218e-05,
      "model_forward_time": 1.76202392578125,
      "step": 70942
    },
    {
      "epoch": 6.765556335449218e-05,
      "step": 70942,
      "training_step_time": 10.021178960800171
    },
    {
      "epoch": 6.765651702880859e-05,
      "model_forward_time": 1.7093992233276367,
      "step": 70943
    },
    {
      "epoch": 6.765651702880859e-05,
      "step": 70943,
      "training_step_time": 8.513465881347656
    },
    {
      "epoch": 6.7657470703125e-05,
      "model_forward_time": 1.3370606899261475,
      "step": 70944
    },
    {
      "epoch": 6.7657470703125e-05,
      "step": 70944,
      "training_step_time": 7.668943166732788
    },
    {
      "epoch": 6.76584243774414e-05,
      "model_forward_time": 1.3484067916870117,
      "step": 70945
    },
    {
      "epoch": 6.76584243774414e-05,
      "step": 70945,
      "training_step_time": 7.688969612121582
    },
    {
      "epoch": 6.765937805175781e-05,
      "model_forward_time": 1.7163031101226807,
      "step": 70946
    },
    {
      "epoch": 6.765937805175781e-05,
      "step": 70946,
      "training_step_time": 8.6559419631958
    },
    {
      "epoch": 6.766033172607422e-05,
      "model_forward_time": 1.7220215797424316,
      "step": 70947
    },
    {
      "epoch": 6.766033172607422e-05,
      "step": 70947,
      "training_step_time": 7.605068922042847
    },
    {
      "epoch": 6.766128540039062e-05,
      "model_forward_time": 1.7334506511688232,
      "step": 70948
    },
    {
      "epoch": 6.766128540039062e-05,
      "step": 70948,
      "training_step_time": 7.6863203048706055
    },
    {
      "epoch": 6.766223907470703e-05,
      "model_forward_time": 1.7094285488128662,
      "step": 70949
    },
    {
      "epoch": 6.766223907470703e-05,
      "step": 70949,
      "training_step_time": 7.783317565917969
    },
    {
      "epoch": 6.766319274902344e-05,
      "grad_norm": 0.10924110561609268,
      "learning_rate": 2.135263981641429e-06,
      "loss": 0.0563,
      "step": 70950
    },
    {
      "epoch": 6.766319274902344e-05,
      "model_forward_time": 1.7184062004089355,
      "step": 70950
    },
    {
      "dynamics_loss_avg": 0.05124015659093857,
      "epoch": 6.766319274902344e-05,
      "step": 70950
    },
    {
      "action_loss_avg": 0.00230505209765397,
      "epoch": 6.766319274902344e-05,
      "step": 70950
    },
    {
      "epoch": 6.766319274902344e-05,
      "step": 70950,
      "training_step_time": 7.7245094776153564
    },
    {
      "epoch": 6.766414642333984e-05,
      "model_forward_time": 2.362964391708374,
      "step": 70951
    },
    {
      "epoch": 6.766414642333984e-05,
      "step": 70951,
      "training_step_time": 8.566916942596436
    },
    {
      "epoch": 6.766510009765625e-05,
      "model_forward_time": 1.7093005180358887,
      "step": 70952
    },
    {
      "epoch": 6.766510009765625e-05,
      "step": 70952,
      "training_step_time": 8.958881378173828
    },
    {
      "epoch": 6.766605377197266e-05,
      "model_forward_time": 1.7132635116577148,
      "step": 70953
    },
    {
      "epoch": 6.766605377197266e-05,
      "step": 70953,
      "training_step_time": 7.722666025161743
    },
    {
      "epoch": 6.766700744628906e-05,
      "model_forward_time": 1.3395264148712158,
      "step": 70954
    },
    {
      "epoch": 6.766700744628906e-05,
      "step": 70954,
      "training_step_time": 8.074073314666748
    },
    {
      "epoch": 6.766796112060547e-05,
      "model_forward_time": 1.729630947113037,
      "step": 70955
    },
    {
      "epoch": 6.766796112060547e-05,
      "step": 70955,
      "training_step_time": 8.547794103622437
    },
    {
      "epoch": 6.766891479492188e-05,
      "model_forward_time": 1.3446323871612549,
      "step": 70956
    },
    {
      "epoch": 6.766891479492188e-05,
      "step": 70956,
      "training_step_time": 9.186592102050781
    },
    {
      "epoch": 6.766986846923828e-05,
      "model_forward_time": 1.732497215270996,
      "step": 70957
    },
    {
      "epoch": 6.766986846923828e-05,
      "step": 70957,
      "training_step_time": 7.638437986373901
    },
    {
      "epoch": 6.767082214355469e-05,
      "model_forward_time": 1.7217674255371094,
      "step": 70958
    },
    {
      "epoch": 6.767082214355469e-05,
      "step": 70958,
      "training_step_time": 9.228820323944092
    },
    {
      "epoch": 6.76717758178711e-05,
      "model_forward_time": 1.7234299182891846,
      "step": 70959
    },
    {
      "epoch": 6.76717758178711e-05,
      "step": 70959,
      "training_step_time": 7.665006399154663
    },
    {
      "epoch": 6.76727294921875e-05,
      "grad_norm": 0.10561631619930267,
      "learning_rate": 2.1339089672136205e-06,
      "loss": 0.0557,
      "step": 70960
    },
    {
      "epoch": 6.76727294921875e-05,
      "model_forward_time": 2.3233089447021484,
      "step": 70960
    },
    {
      "dynamics_loss_avg": 0.05829765982925892,
      "epoch": 6.76727294921875e-05,
      "step": 70960
    },
    {
      "action_loss_avg": 0.0010017930762842298,
      "epoch": 6.76727294921875e-05,
      "step": 70960
    },
    {
      "epoch": 6.76727294921875e-05,
      "step": 70960,
      "training_step_time": 9.189707279205322
    },
    {
      "epoch": 6.76736831665039e-05,
      "model_forward_time": 1.728898048400879,
      "step": 70961
    },
    {
      "epoch": 6.76736831665039e-05,
      "step": 70961,
      "training_step_time": 9.088429927825928
    },
    {
      "epoch": 6.767463684082032e-05,
      "model_forward_time": 1.7179930210113525,
      "step": 70962
    },
    {
      "epoch": 6.767463684082032e-05,
      "step": 70962,
      "training_step_time": 8.470763206481934
    },
    {
      "epoch": 6.767559051513671e-05,
      "model_forward_time": 1.7236039638519287,
      "step": 70963
    },
    {
      "epoch": 6.767559051513671e-05,
      "step": 70963,
      "training_step_time": 7.502227306365967
    },
    {
      "epoch": 6.767654418945312e-05,
      "model_forward_time": 1.72288179397583,
      "step": 70964
    },
    {
      "epoch": 6.767654418945312e-05,
      "step": 70964,
      "training_step_time": 7.713824510574341
    },
    {
      "epoch": 6.767749786376953e-05,
      "model_forward_time": 1.7203803062438965,
      "step": 70965
    },
    {
      "epoch": 6.767749786376953e-05,
      "step": 70965,
      "training_step_time": 7.790959358215332
    },
    {
      "epoch": 6.767845153808593e-05,
      "model_forward_time": 1.7283709049224854,
      "step": 70966
    },
    {
      "epoch": 6.767845153808593e-05,
      "step": 70966,
      "training_step_time": 8.101431608200073
    },
    {
      "epoch": 6.767940521240234e-05,
      "model_forward_time": 1.7136006355285645,
      "step": 70967
    },
    {
      "epoch": 6.767940521240234e-05,
      "step": 70967,
      "training_step_time": 7.67995285987854
    },
    {
      "epoch": 6.768035888671875e-05,
      "model_forward_time": 1.7371582984924316,
      "step": 70968
    },
    {
      "epoch": 6.768035888671875e-05,
      "step": 70968,
      "training_step_time": 7.693125247955322
    },
    {
      "epoch": 6.768131256103515e-05,
      "model_forward_time": 2.3195230960845947,
      "step": 70969
    },
    {
      "epoch": 6.768131256103515e-05,
      "step": 70969,
      "training_step_time": 8.50180697441101
    },
    {
      "epoch": 6.768226623535156e-05,
      "grad_norm": 0.10045787692070007,
      "learning_rate": 2.1325542662172195e-06,
      "loss": 0.0565,
      "step": 70970
    },
    {
      "epoch": 6.768226623535156e-05,
      "model_forward_time": 1.8079931735992432,
      "step": 70970
    },
    {
      "dynamics_loss_avg": 0.041628651693463324,
      "epoch": 6.768226623535156e-05,
      "step": 70970
    },
    {
      "action_loss_avg": 0.0018505903601180762,
      "epoch": 6.768226623535156e-05,
      "step": 70970
    },
    {
      "epoch": 6.768226623535156e-05,
      "step": 70970,
      "training_step_time": 7.924358129501343
    },
    {
      "epoch": 6.768321990966797e-05,
      "model_forward_time": 1.361680030822754,
      "step": 70971
    },
    {
      "epoch": 6.768321990966797e-05,
      "step": 70971,
      "training_step_time": 8.660261154174805
    },
    {
      "epoch": 6.768417358398437e-05,
      "model_forward_time": 1.7576909065246582,
      "step": 70972
    },
    {
      "epoch": 6.768417358398437e-05,
      "step": 70972,
      "training_step_time": 7.647444009780884
    },
    {
      "epoch": 6.768512725830078e-05,
      "model_forward_time": 1.735199213027954,
      "step": 70973
    },
    {
      "epoch": 6.768512725830078e-05,
      "step": 70973,
      "training_step_time": 7.64609956741333
    },
    {
      "epoch": 6.768608093261719e-05,
      "model_forward_time": 1.7354588508605957,
      "step": 70974
    },
    {
      "epoch": 6.768608093261719e-05,
      "step": 70974,
      "training_step_time": 9.320849657058716
    },
    {
      "epoch": 6.768703460693359e-05,
      "model_forward_time": 1.7295007705688477,
      "step": 70975
    },
    {
      "epoch": 6.768703460693359e-05,
      "step": 70975,
      "training_step_time": 7.505002737045288
    },
    {
      "epoch": 6.768798828125e-05,
      "model_forward_time": 1.729262113571167,
      "step": 70976
    },
    {
      "epoch": 6.768798828125e-05,
      "step": 70976,
      "training_step_time": 8.629875659942627
    },
    {
      "epoch": 6.768894195556641e-05,
      "model_forward_time": 1.74267578125,
      "step": 70977
    },
    {
      "epoch": 6.768894195556641e-05,
      "step": 70977,
      "training_step_time": 10.021535396575928
    },
    {
      "epoch": 6.768989562988281e-05,
      "model_forward_time": 1.7101569175720215,
      "step": 70978
    },
    {
      "epoch": 6.768989562988281e-05,
      "step": 70978,
      "training_step_time": 7.726465940475464
    },
    {
      "epoch": 6.769084930419922e-05,
      "model_forward_time": 2.340069055557251,
      "step": 70979
    },
    {
      "epoch": 6.769084930419922e-05,
      "step": 70979,
      "training_step_time": 8.849243402481079
    },
    {
      "epoch": 6.769180297851563e-05,
      "grad_norm": 0.09381062537431717,
      "learning_rate": 2.1311998788003714e-06,
      "loss": 0.0572,
      "step": 70980
    },
    {
      "epoch": 6.769180297851563e-05,
      "model_forward_time": 1.719491958618164,
      "step": 70980
    },
    {
      "dynamics_loss_avg": 0.043710795044898984,
      "epoch": 6.769180297851563e-05,
      "step": 70980
    },
    {
      "action_loss_avg": 0.0012840277384384535,
      "epoch": 6.769180297851563e-05,
      "step": 70980
    },
    {
      "epoch": 6.769180297851563e-05,
      "step": 70980,
      "training_step_time": 7.7868897914886475
    },
    {
      "epoch": 6.769275665283203e-05,
      "model_forward_time": 1.73649001121521,
      "step": 70981
    },
    {
      "epoch": 6.769275665283203e-05,
      "step": 70981,
      "training_step_time": 7.63039493560791
    },
    {
      "epoch": 6.769371032714844e-05,
      "model_forward_time": 1.7313451766967773,
      "step": 70982
    },
    {
      "epoch": 6.769371032714844e-05,
      "step": 70982,
      "training_step_time": 8.07037615776062
    },
    {
      "epoch": 6.769466400146485e-05,
      "model_forward_time": 1.7294132709503174,
      "step": 70983
    },
    {
      "epoch": 6.769466400146485e-05,
      "step": 70983,
      "training_step_time": 7.681901454925537
    },
    {
      "epoch": 6.769561767578124e-05,
      "model_forward_time": 1.7319552898406982,
      "step": 70984
    },
    {
      "epoch": 6.769561767578124e-05,
      "step": 70984,
      "training_step_time": 7.646931171417236
    },
    {
      "epoch": 6.769657135009766e-05,
      "model_forward_time": 1.7295401096343994,
      "step": 70985
    },
    {
      "epoch": 6.769657135009766e-05,
      "step": 70985,
      "training_step_time": 8.052576780319214
    },
    {
      "epoch": 6.769752502441407e-05,
      "model_forward_time": 1.7349042892456055,
      "step": 70986
    },
    {
      "epoch": 6.769752502441407e-05,
      "step": 70986,
      "training_step_time": 8.490361452102661
    },
    {
      "epoch": 6.769847869873046e-05,
      "model_forward_time": 1.7275290489196777,
      "step": 70987
    },
    {
      "epoch": 6.769847869873046e-05,
      "step": 70987,
      "training_step_time": 7.754669427871704
    },
    {
      "epoch": 6.769943237304687e-05,
      "model_forward_time": 2.3582100868225098,
      "step": 70988
    },
    {
      "epoch": 6.769943237304687e-05,
      "step": 70988,
      "training_step_time": 8.899313926696777
    },
    {
      "epoch": 6.770038604736329e-05,
      "model_forward_time": 1.7110316753387451,
      "step": 70989
    },
    {
      "epoch": 6.770038604736329e-05,
      "step": 70989,
      "training_step_time": 7.856133937835693
    },
    {
      "epoch": 6.770133972167968e-05,
      "grad_norm": 0.08236188441514969,
      "learning_rate": 2.129845805111191e-06,
      "loss": 0.06,
      "step": 70990
    },
    {
      "epoch": 6.770133972167968e-05,
      "model_forward_time": 1.7232723236083984,
      "step": 70990
    },
    {
      "dynamics_loss_avg": 0.056115427426993846,
      "epoch": 6.770133972167968e-05,
      "step": 70990
    },
    {
      "action_loss_avg": 0.0012165982101578266,
      "epoch": 6.770133972167968e-05,
      "step": 70990
    },
    {
      "epoch": 6.770133972167968e-05,
      "step": 70990,
      "training_step_time": 7.6649909019470215
    },
    {
      "epoch": 6.77022933959961e-05,
      "model_forward_time": 1.7351889610290527,
      "step": 70991
    },
    {
      "epoch": 6.77022933959961e-05,
      "step": 70991,
      "training_step_time": 7.423495292663574
    },
    {
      "epoch": 6.77032470703125e-05,
      "model_forward_time": 1.7374563217163086,
      "step": 70992
    },
    {
      "epoch": 6.77032470703125e-05,
      "step": 70992,
      "training_step_time": 7.685858726501465
    },
    {
      "epoch": 6.77042007446289e-05,
      "model_forward_time": 1.7269868850708008,
      "step": 70993
    },
    {
      "epoch": 6.77042007446289e-05,
      "step": 70993,
      "training_step_time": 7.6976704597473145
    },
    {
      "epoch": 6.770515441894531e-05,
      "model_forward_time": 1.7221457958221436,
      "step": 70994
    },
    {
      "epoch": 6.770515441894531e-05,
      "step": 70994,
      "training_step_time": 7.522448539733887
    },
    {
      "epoch": 6.770610809326172e-05,
      "model_forward_time": 1.7347321510314941,
      "step": 70995
    },
    {
      "epoch": 6.770610809326172e-05,
      "step": 70995,
      "training_step_time": 7.723491668701172
    },
    {
      "epoch": 6.770706176757812e-05,
      "model_forward_time": 1.7021710872650146,
      "step": 70996
    },
    {
      "epoch": 6.770706176757812e-05,
      "step": 70996,
      "training_step_time": 8.806582689285278
    },
    {
      "epoch": 6.770801544189453e-05,
      "model_forward_time": 2.3406667709350586,
      "step": 70997
    },
    {
      "epoch": 6.770801544189453e-05,
      "step": 70997,
      "training_step_time": 7.859516620635986
    },
    {
      "epoch": 6.770896911621094e-05,
      "model_forward_time": 1.7608788013458252,
      "step": 70998
    },
    {
      "epoch": 6.770896911621094e-05,
      "step": 70998,
      "training_step_time": 11.90088963508606
    },
    {
      "epoch": 6.770992279052734e-05,
      "model_forward_time": 1.7172963619232178,
      "step": 70999
    },
    {
      "epoch": 6.770992279052734e-05,
      "step": 70999,
      "training_step_time": 8.300378561019897
    },
    {
      "epoch": 6.771087646484375e-05,
      "grad_norm": 0.09526273608207703,
      "learning_rate": 2.1284920452977603e-06,
      "loss": 0.057,
      "step": 71000
    },
    {
      "epoch": 6.771087646484375e-05,
      "model_forward_time": 1.745434045791626,
      "step": 71000
    },
    {
      "dynamics_loss_avg": 0.055948485247790815,
      "epoch": 6.771087646484375e-05,
      "step": 71000
    },
    {
      "action_loss_avg": 0.0015007031935965643,
      "epoch": 6.771087646484375e-05,
      "step": 71000
    },
    {
      "epoch": 6.771087646484375e-05,
      "step": 71000,
      "training_step_time": 7.279724597930908
    },
    {
      "epoch": 6.771183013916016e-05,
      "model_forward_time": 1.713343620300293,
      "step": 71001
    },
    {
      "epoch": 6.771183013916016e-05,
      "step": 71001,
      "training_step_time": 8.338633060455322
    },
    {
      "epoch": 6.771278381347656e-05,
      "model_forward_time": 1.718977928161621,
      "step": 71002
    },
    {
      "epoch": 6.771278381347656e-05,
      "step": 71002,
      "training_step_time": 7.694042444229126
    },
    {
      "epoch": 6.771373748779297e-05,
      "model_forward_time": 1.7211809158325195,
      "step": 71003
    },
    {
      "epoch": 6.771373748779297e-05,
      "step": 71003,
      "training_step_time": 8.271072626113892
    },
    {
      "epoch": 6.771469116210938e-05,
      "model_forward_time": 1.728762149810791,
      "step": 71004
    },
    {
      "epoch": 6.771469116210938e-05,
      "step": 71004,
      "training_step_time": 7.6452271938323975
    },
    {
      "epoch": 6.771564483642578e-05,
      "model_forward_time": 1.7148687839508057,
      "step": 71005
    },
    {
      "epoch": 6.771564483642578e-05,
      "step": 71005,
      "training_step_time": 9.345755100250244
    },
    {
      "epoch": 6.771659851074219e-05,
      "model_forward_time": 1.7174649238586426,
      "step": 71006
    },
    {
      "epoch": 6.771659851074219e-05,
      "step": 71006,
      "training_step_time": 9.858410835266113
    },
    {
      "epoch": 6.77175521850586e-05,
      "model_forward_time": 1.7119708061218262,
      "step": 71007
    },
    {
      "epoch": 6.77175521850586e-05,
      "step": 71007,
      "training_step_time": 8.920728206634521
    },
    {
      "epoch": 6.7718505859375e-05,
      "model_forward_time": 1.7157964706420898,
      "step": 71008
    },
    {
      "epoch": 6.7718505859375e-05,
      "step": 71008,
      "training_step_time": 9.390975713729858
    },
    {
      "epoch": 6.77194595336914e-05,
      "model_forward_time": 1.7173659801483154,
      "step": 71009
    },
    {
      "epoch": 6.77194595336914e-05,
      "step": 71009,
      "training_step_time": 7.6645588874816895
    },
    {
      "epoch": 6.772041320800782e-05,
      "grad_norm": 0.10121835023164749,
      "learning_rate": 2.12713859950812e-06,
      "loss": 0.0567,
      "step": 71010
    },
    {
      "epoch": 6.772041320800782e-05,
      "model_forward_time": 1.72060227394104,
      "step": 71010
    },
    {
      "dynamics_loss_avg": 0.07078300062566996,
      "epoch": 6.772041320800782e-05,
      "step": 71010
    },
    {
      "action_loss_avg": 0.0016666890995111316,
      "epoch": 6.772041320800782e-05,
      "step": 71010
    },
    {
      "epoch": 6.772041320800782e-05,
      "step": 71010,
      "training_step_time": 7.728027105331421
    },
    {
      "epoch": 6.772136688232421e-05,
      "model_forward_time": 1.7124660015106201,
      "step": 71011
    },
    {
      "epoch": 6.772136688232421e-05,
      "step": 71011,
      "training_step_time": 7.733743906021118
    },
    {
      "epoch": 6.772232055664063e-05,
      "model_forward_time": 1.718867540359497,
      "step": 71012
    },
    {
      "epoch": 6.772232055664063e-05,
      "step": 71012,
      "training_step_time": 8.512458324432373
    },
    {
      "epoch": 6.772327423095704e-05,
      "model_forward_time": 1.7129559516906738,
      "step": 71013
    },
    {
      "epoch": 6.772327423095704e-05,
      "step": 71013,
      "training_step_time": 7.716340780258179
    },
    {
      "epoch": 6.772422790527343e-05,
      "model_forward_time": 1.7091138362884521,
      "step": 71014
    },
    {
      "epoch": 6.772422790527343e-05,
      "step": 71014,
      "training_step_time": 7.754318475723267
    },
    {
      "epoch": 6.772518157958984e-05,
      "model_forward_time": 2.2707653045654297,
      "step": 71015
    },
    {
      "epoch": 6.772518157958984e-05,
      "step": 71015,
      "training_step_time": 7.8738789558410645
    },
    {
      "epoch": 6.772613525390626e-05,
      "model_forward_time": 1.0383272171020508,
      "step": 71016
    },
    {
      "epoch": 6.772613525390626e-05,
      "step": 71016,
      "training_step_time": 8.769627094268799
    },
    {
      "epoch": 6.772708892822265e-05,
      "model_forward_time": 1.7160413265228271,
      "step": 71017
    },
    {
      "epoch": 6.772708892822265e-05,
      "step": 71017,
      "training_step_time": 9.437957286834717
    },
    {
      "epoch": 6.772804260253906e-05,
      "model_forward_time": 1.7146220207214355,
      "step": 71018
    },
    {
      "epoch": 6.772804260253906e-05,
      "step": 71018,
      "training_step_time": 7.670659065246582
    },
    {
      "epoch": 6.772899627685547e-05,
      "model_forward_time": 1.7204327583312988,
      "step": 71019
    },
    {
      "epoch": 6.772899627685547e-05,
      "step": 71019,
      "training_step_time": 7.75052809715271
    },
    {
      "epoch": 6.772994995117187e-05,
      "grad_norm": 0.08853223919868469,
      "learning_rate": 2.1257854678902832e-06,
      "loss": 0.059,
      "step": 71020
    },
    {
      "epoch": 6.772994995117187e-05,
      "model_forward_time": 1.7188873291015625,
      "step": 71020
    },
    {
      "dynamics_loss_avg": 0.06421560943126678,
      "epoch": 6.772994995117187e-05,
      "step": 71020
    },
    {
      "action_loss_avg": 0.0025462302030064164,
      "epoch": 6.772994995117187e-05,
      "step": 71020
    },
    {
      "epoch": 6.772994995117187e-05,
      "step": 71020,
      "training_step_time": 7.70880126953125
    },
    {
      "epoch": 6.773090362548828e-05,
      "model_forward_time": 1.7173287868499756,
      "step": 71021
    },
    {
      "epoch": 6.773090362548828e-05,
      "step": 71021,
      "training_step_time": 8.030407428741455
    },
    {
      "epoch": 6.773185729980469e-05,
      "model_forward_time": 1.711261510848999,
      "step": 71022
    },
    {
      "epoch": 6.773185729980469e-05,
      "step": 71022,
      "training_step_time": 7.719072580337524
    },
    {
      "epoch": 6.773281097412109e-05,
      "model_forward_time": 1.7298402786254883,
      "step": 71023
    },
    {
      "epoch": 6.773281097412109e-05,
      "step": 71023,
      "training_step_time": 8.131904363632202
    },
    {
      "epoch": 6.77337646484375e-05,
      "model_forward_time": 2.2889175415039062,
      "step": 71024
    },
    {
      "epoch": 6.77337646484375e-05,
      "step": 71024,
      "training_step_time": 7.798112869262695
    },
    {
      "epoch": 6.773471832275391e-05,
      "model_forward_time": 1.403766393661499,
      "step": 71025
    },
    {
      "epoch": 6.773471832275391e-05,
      "step": 71025,
      "training_step_time": 9.48156476020813
    },
    {
      "epoch": 6.773567199707031e-05,
      "model_forward_time": 1.731844425201416,
      "step": 71026
    },
    {
      "epoch": 6.773567199707031e-05,
      "step": 71026,
      "training_step_time": 8.379235029220581
    },
    {
      "epoch": 6.773662567138672e-05,
      "model_forward_time": 1.7285985946655273,
      "step": 71027
    },
    {
      "epoch": 6.773662567138672e-05,
      "step": 71027,
      "training_step_time": 7.6311163902282715
    },
    {
      "epoch": 6.773757934570313e-05,
      "model_forward_time": 1.727051019668579,
      "step": 71028
    },
    {
      "epoch": 6.773757934570313e-05,
      "step": 71028,
      "training_step_time": 8.198460578918457
    },
    {
      "epoch": 6.773853302001953e-05,
      "model_forward_time": 1.7384943962097168,
      "step": 71029
    },
    {
      "epoch": 6.773853302001953e-05,
      "step": 71029,
      "training_step_time": 7.802125692367554
    },
    {
      "epoch": 6.773948669433594e-05,
      "grad_norm": 0.09548569470643997,
      "learning_rate": 2.124432650592227e-06,
      "loss": 0.0592,
      "step": 71030
    },
    {
      "epoch": 6.773948669433594e-05,
      "model_forward_time": 1.7315504550933838,
      "step": 71030
    },
    {
      "dynamics_loss_avg": 0.052928286232054235,
      "epoch": 6.773948669433594e-05,
      "step": 71030
    },
    {
      "action_loss_avg": 0.0012235897753271273,
      "epoch": 6.773948669433594e-05,
      "step": 71030
    },
    {
      "epoch": 6.773948669433594e-05,
      "step": 71030,
      "training_step_time": 8.33468508720398
    },
    {
      "epoch": 6.774044036865235e-05,
      "model_forward_time": 1.7256698608398438,
      "step": 71031
    },
    {
      "epoch": 6.774044036865235e-05,
      "step": 71031,
      "training_step_time": 7.6580116748809814
    },
    {
      "epoch": 6.774139404296875e-05,
      "model_forward_time": 1.714277744293213,
      "step": 71032
    },
    {
      "epoch": 6.774139404296875e-05,
      "step": 71032,
      "training_step_time": 8.337561130523682
    },
    {
      "epoch": 6.774234771728516e-05,
      "model_forward_time": 2.2853569984436035,
      "step": 71033
    },
    {
      "epoch": 6.774234771728516e-05,
      "step": 71033,
      "training_step_time": 8.236360549926758
    },
    {
      "epoch": 6.774330139160157e-05,
      "model_forward_time": 1.7616617679595947,
      "step": 71034
    },
    {
      "epoch": 6.774330139160157e-05,
      "step": 71034,
      "training_step_time": 7.673986196517944
    },
    {
      "epoch": 6.774425506591797e-05,
      "model_forward_time": 1.719890832901001,
      "step": 71035
    },
    {
      "epoch": 6.774425506591797e-05,
      "step": 71035,
      "training_step_time": 7.675635814666748
    },
    {
      "epoch": 6.774520874023438e-05,
      "model_forward_time": 1.7053985595703125,
      "step": 71036
    },
    {
      "epoch": 6.774520874023438e-05,
      "step": 71036,
      "training_step_time": 7.529141426086426
    },
    {
      "epoch": 6.774616241455079e-05,
      "model_forward_time": 1.708693504333496,
      "step": 71037
    },
    {
      "epoch": 6.774616241455079e-05,
      "step": 71037,
      "training_step_time": 7.49535346031189
    },
    {
      "epoch": 6.774711608886718e-05,
      "model_forward_time": 1.7199594974517822,
      "step": 71038
    },
    {
      "epoch": 6.774711608886718e-05,
      "step": 71038,
      "training_step_time": 7.7213990688323975
    },
    {
      "epoch": 6.77480697631836e-05,
      "model_forward_time": 1.347639799118042,
      "step": 71039
    },
    {
      "epoch": 6.77480697631836e-05,
      "step": 71039,
      "training_step_time": 7.639057159423828
    },
    {
      "epoch": 6.77490234375e-05,
      "grad_norm": 0.07418087869882584,
      "learning_rate": 2.123080147761894e-06,
      "loss": 0.0587,
      "step": 71040
    },
    {
      "epoch": 6.77490234375e-05,
      "model_forward_time": 1.7280848026275635,
      "step": 71040
    },
    {
      "dynamics_loss_avg": 0.0657321804203093,
      "epoch": 6.77490234375e-05,
      "step": 71040
    },
    {
      "action_loss_avg": 0.0014885864191455766,
      "epoch": 6.77490234375e-05,
      "step": 71040
    },
    {
      "epoch": 6.77490234375e-05,
      "step": 71040,
      "training_step_time": 8.244760751724243
    },
    {
      "epoch": 6.77499771118164e-05,
      "model_forward_time": 1.7347025871276855,
      "step": 71041
    },
    {
      "epoch": 6.77499771118164e-05,
      "step": 71041,
      "training_step_time": 8.468624114990234
    },
    {
      "epoch": 6.775093078613281e-05,
      "model_forward_time": 1.7326312065124512,
      "step": 71042
    },
    {
      "epoch": 6.775093078613281e-05,
      "step": 71042,
      "training_step_time": 8.304007530212402
    },
    {
      "epoch": 6.775188446044922e-05,
      "model_forward_time": 2.331550359725952,
      "step": 71043
    },
    {
      "epoch": 6.775188446044922e-05,
      "step": 71043,
      "training_step_time": 8.767168760299683
    },
    {
      "epoch": 6.775283813476562e-05,
      "model_forward_time": 1.3547775745391846,
      "step": 71044
    },
    {
      "epoch": 6.775283813476562e-05,
      "step": 71044,
      "training_step_time": 8.89089322090149
    },
    {
      "epoch": 6.775379180908203e-05,
      "model_forward_time": 1.7202074527740479,
      "step": 71045
    },
    {
      "epoch": 6.775379180908203e-05,
      "step": 71045,
      "training_step_time": 7.7443578243255615
    },
    {
      "epoch": 6.775474548339844e-05,
      "model_forward_time": 1.700443983078003,
      "step": 71046
    },
    {
      "epoch": 6.775474548339844e-05,
      "step": 71046,
      "training_step_time": 7.525991201400757
    },
    {
      "epoch": 6.775569915771484e-05,
      "model_forward_time": 1.7176880836486816,
      "step": 71047
    },
    {
      "epoch": 6.775569915771484e-05,
      "step": 71047,
      "training_step_time": 9.953712940216064
    },
    {
      "epoch": 6.775665283203125e-05,
      "model_forward_time": 1.71586275100708,
      "step": 71048
    },
    {
      "epoch": 6.775665283203125e-05,
      "step": 71048,
      "training_step_time": 7.59221339225769
    },
    {
      "epoch": 6.775760650634766e-05,
      "model_forward_time": 1.7188735008239746,
      "step": 71049
    },
    {
      "epoch": 6.775760650634766e-05,
      "step": 71049,
      "training_step_time": 7.693174600601196
    },
    {
      "epoch": 6.775856018066406e-05,
      "grad_norm": 0.08976723253726959,
      "learning_rate": 2.121727959547189e-06,
      "loss": 0.0573,
      "step": 71050
    },
    {
      "epoch": 6.775856018066406e-05,
      "model_forward_time": 1.7176637649536133,
      "step": 71050
    },
    {
      "dynamics_loss_avg": 0.06227046325802803,
      "epoch": 6.775856018066406e-05,
      "step": 71050
    },
    {
      "action_loss_avg": 0.0015562816435704008,
      "epoch": 6.775856018066406e-05,
      "step": 71050
    },
    {
      "epoch": 6.775856018066406e-05,
      "step": 71050,
      "training_step_time": 7.552945613861084
    },
    {
      "epoch": 6.775951385498047e-05,
      "model_forward_time": 1.7287321090698242,
      "step": 71051
    },
    {
      "epoch": 6.775951385498047e-05,
      "step": 71051,
      "training_step_time": 7.71767520904541
    },
    {
      "epoch": 6.776046752929688e-05,
      "model_forward_time": 2.3171021938323975,
      "step": 71052
    },
    {
      "epoch": 6.776046752929688e-05,
      "step": 71052,
      "training_step_time": 7.638334035873413
    },
    {
      "epoch": 6.776142120361328e-05,
      "model_forward_time": 1.7241337299346924,
      "step": 71053
    },
    {
      "epoch": 6.776142120361328e-05,
      "step": 71053,
      "training_step_time": 8.686607122421265
    },
    {
      "epoch": 6.776237487792969e-05,
      "model_forward_time": 1.7278118133544922,
      "step": 71054
    },
    {
      "epoch": 6.776237487792969e-05,
      "step": 71054,
      "training_step_time": 8.67249083518982
    },
    {
      "epoch": 6.77633285522461e-05,
      "model_forward_time": 1.7439942359924316,
      "step": 71055
    },
    {
      "epoch": 6.77633285522461e-05,
      "step": 71055,
      "training_step_time": 7.542478084564209
    },
    {
      "epoch": 6.77642822265625e-05,
      "model_forward_time": 1.3612456321716309,
      "step": 71056
    },
    {
      "epoch": 6.77642822265625e-05,
      "step": 71056,
      "training_step_time": 7.726447343826294
    },
    {
      "epoch": 6.776523590087891e-05,
      "model_forward_time": 1.7348182201385498,
      "step": 71057
    },
    {
      "epoch": 6.776523590087891e-05,
      "step": 71057,
      "training_step_time": 8.335271835327148
    },
    {
      "epoch": 6.776618957519532e-05,
      "model_forward_time": 1.7362935543060303,
      "step": 71058
    },
    {
      "epoch": 6.776618957519532e-05,
      "step": 71058,
      "training_step_time": 7.519848585128784
    },
    {
      "epoch": 6.776714324951172e-05,
      "model_forward_time": 1.7387359142303467,
      "step": 71059
    },
    {
      "epoch": 6.776714324951172e-05,
      "step": 71059,
      "training_step_time": 7.671107769012451
    },
    {
      "epoch": 6.776809692382813e-05,
      "grad_norm": 0.09679345041513443,
      "learning_rate": 2.1203760860959887e-06,
      "loss": 0.0548,
      "step": 71060
    },
    {
      "epoch": 6.776809692382813e-05,
      "model_forward_time": 1.7265753746032715,
      "step": 71060
    },
    {
      "dynamics_loss_avg": 0.07484082020819187,
      "epoch": 6.776809692382813e-05,
      "step": 71060
    },
    {
      "action_loss_avg": 0.0018352377344854176,
      "epoch": 6.776809692382813e-05,
      "step": 71060
    },
    {
      "epoch": 6.776809692382813e-05,
      "step": 71060,
      "training_step_time": 7.687366247177124
    },
    {
      "epoch": 6.776905059814454e-05,
      "model_forward_time": 2.3494880199432373,
      "step": 71061
    },
    {
      "epoch": 6.776905059814454e-05,
      "step": 71061,
      "training_step_time": 9.80632495880127
    },
    {
      "epoch": 6.777000427246094e-05,
      "model_forward_time": 1.7548134326934814,
      "step": 71062
    },
    {
      "epoch": 6.777000427246094e-05,
      "step": 71062,
      "training_step_time": 6.72191309928894
    },
    {
      "epoch": 6.777095794677735e-05,
      "model_forward_time": 1.7549476623535156,
      "step": 71063
    },
    {
      "epoch": 6.777095794677735e-05,
      "step": 71063,
      "training_step_time": 6.683704137802124
    },
    {
      "epoch": 6.777191162109374e-05,
      "model_forward_time": 1.7454674243927002,
      "step": 71064
    },
    {
      "epoch": 6.777191162109374e-05,
      "step": 71064,
      "training_step_time": 6.71719217300415
    },
    {
      "epoch": 6.777286529541015e-05,
      "model_forward_time": 1.7508482933044434,
      "step": 71065
    },
    {
      "epoch": 6.777286529541015e-05,
      "step": 71065,
      "training_step_time": 6.680463790893555
    },
    {
      "epoch": 6.777381896972656e-05,
      "model_forward_time": 1.7582051753997803,
      "step": 71066
    },
    {
      "epoch": 6.777381896972656e-05,
      "step": 71066,
      "training_step_time": 6.690190315246582
    },
    {
      "epoch": 6.777477264404296e-05,
      "model_forward_time": 1.7612254619598389,
      "step": 71067
    },
    {
      "epoch": 6.777477264404296e-05,
      "step": 71067,
      "training_step_time": 6.74597954750061
    },
    {
      "epoch": 6.777572631835937e-05,
      "model_forward_time": 1.747955560684204,
      "step": 71068
    },
    {
      "epoch": 6.777572631835937e-05,
      "step": 71068,
      "training_step_time": 6.762257099151611
    },
    {
      "epoch": 6.777667999267578e-05,
      "model_forward_time": 1.7481226921081543,
      "step": 71069
    },
    {
      "epoch": 6.777667999267578e-05,
      "step": 71069,
      "training_step_time": 6.753243923187256
    },
    {
      "epoch": 6.777763366699218e-05,
      "grad_norm": 0.06561116129159927,
      "learning_rate": 2.119024527556127e-06,
      "loss": 0.0567,
      "step": 71070
    },
    {
      "epoch": 6.777763366699218e-05,
      "model_forward_time": 2.3452250957489014,
      "step": 71070
    },
    {
      "dynamics_loss_avg": 0.05577258849516511,
      "epoch": 6.777763366699218e-05,
      "step": 71070
    },
    {
      "action_loss_avg": 0.0016508655622601508,
      "epoch": 6.777763366699218e-05,
      "step": 71070
    },
    {
      "epoch": 6.777763366699218e-05,
      "step": 71070,
      "training_step_time": 9.80896782875061
    },
    {
      "epoch": 6.777858734130859e-05,
      "model_forward_time": 1.75199556350708,
      "step": 71071
    },
    {
      "epoch": 6.777858734130859e-05,
      "step": 71071,
      "training_step_time": 7.7918701171875
    },
    {
      "epoch": 6.7779541015625e-05,
      "model_forward_time": 1.7512638568878174,
      "step": 71072
    },
    {
      "epoch": 6.7779541015625e-05,
      "step": 71072,
      "training_step_time": 6.761397361755371
    },
    {
      "epoch": 6.77804946899414e-05,
      "model_forward_time": 1.7673485279083252,
      "step": 71073
    },
    {
      "epoch": 6.77804946899414e-05,
      "step": 71073,
      "training_step_time": 6.763188600540161
    },
    {
      "epoch": 6.778144836425781e-05,
      "model_forward_time": 1.748767375946045,
      "step": 71074
    },
    {
      "epoch": 6.778144836425781e-05,
      "step": 71074,
      "training_step_time": 6.734972715377808
    },
    {
      "epoch": 6.778240203857422e-05,
      "model_forward_time": 1.759187936782837,
      "step": 71075
    },
    {
      "epoch": 6.778240203857422e-05,
      "step": 71075,
      "training_step_time": 6.712048530578613
    },
    {
      "epoch": 6.778335571289062e-05,
      "model_forward_time": 1.7527985572814941,
      "step": 71076
    },
    {
      "epoch": 6.778335571289062e-05,
      "step": 71076,
      "training_step_time": 6.680332183837891
    },
    {
      "epoch": 6.778430938720703e-05,
      "model_forward_time": 1.7480556964874268,
      "step": 71077
    },
    {
      "epoch": 6.778430938720703e-05,
      "step": 71077,
      "training_step_time": 6.690398454666138
    },
    {
      "epoch": 6.778526306152344e-05,
      "model_forward_time": 1.7541542053222656,
      "step": 71078
    },
    {
      "epoch": 6.778526306152344e-05,
      "step": 71078,
      "training_step_time": 6.683866024017334
    },
    {
      "epoch": 6.778621673583984e-05,
      "model_forward_time": 2.347235679626465,
      "step": 71079
    },
    {
      "epoch": 6.778621673583984e-05,
      "step": 71079,
      "training_step_time": 8.643512725830078
    },
    {
      "epoch": 6.778717041015625e-05,
      "grad_norm": 0.07241711020469666,
      "learning_rate": 2.117673284075415e-06,
      "loss": 0.0534,
      "step": 71080
    },
    {
      "epoch": 6.778717041015625e-05,
      "model_forward_time": 1.7500357627868652,
      "step": 71080
    },
    {
      "dynamics_loss_avg": 0.0458227451890707,
      "epoch": 6.778717041015625e-05,
      "step": 71080
    },
    {
      "action_loss_avg": 0.0012258793925866486,
      "epoch": 6.778717041015625e-05,
      "step": 71080
    },
    {
      "epoch": 6.778717041015625e-05,
      "step": 71080,
      "training_step_time": 7.804807186126709
    },
    {
      "epoch": 6.778812408447266e-05,
      "model_forward_time": 1.746401071548462,
      "step": 71081
    },
    {
      "epoch": 6.778812408447266e-05,
      "step": 71081,
      "training_step_time": 6.681638717651367
    },
    {
      "epoch": 6.778907775878906e-05,
      "model_forward_time": 1.7585625648498535,
      "step": 71082
    },
    {
      "epoch": 6.778907775878906e-05,
      "step": 71082,
      "training_step_time": 6.731646537780762
    },
    {
      "epoch": 6.779003143310547e-05,
      "model_forward_time": 1.757737636566162,
      "step": 71083
    },
    {
      "epoch": 6.779003143310547e-05,
      "step": 71083,
      "training_step_time": 6.677755117416382
    },
    {
      "epoch": 6.779098510742188e-05,
      "model_forward_time": 1.758263111114502,
      "step": 71084
    },
    {
      "epoch": 6.779098510742188e-05,
      "step": 71084,
      "training_step_time": 6.7210211753845215
    },
    {
      "epoch": 6.779193878173828e-05,
      "model_forward_time": 1.7486107349395752,
      "step": 71085
    },
    {
      "epoch": 6.779193878173828e-05,
      "step": 71085,
      "training_step_time": 6.6832356452941895
    },
    {
      "epoch": 6.779289245605469e-05,
      "model_forward_time": 1.7507426738739014,
      "step": 71086
    },
    {
      "epoch": 6.779289245605469e-05,
      "step": 71086,
      "training_step_time": 6.660588502883911
    },
    {
      "epoch": 6.77938461303711e-05,
      "model_forward_time": 1.7483856678009033,
      "step": 71087
    },
    {
      "epoch": 6.77938461303711e-05,
      "step": 71087,
      "training_step_time": 6.662298917770386
    },
    {
      "epoch": 6.77947998046875e-05,
      "model_forward_time": 2.0103139877319336,
      "step": 71088
    },
    {
      "epoch": 6.77947998046875e-05,
      "step": 71088,
      "training_step_time": 7.6883463859558105
    },
    {
      "epoch": 6.77957534790039e-05,
      "model_forward_time": 1.7607371807098389,
      "step": 71089
    },
    {
      "epoch": 6.77957534790039e-05,
      "step": 71089,
      "training_step_time": 8.915558099746704
    },
    {
      "epoch": 6.779670715332032e-05,
      "grad_norm": 0.10366248339414597,
      "learning_rate": 2.1163223558016166e-06,
      "loss": 0.0547,
      "step": 71090
    },
    {
      "epoch": 6.779670715332032e-05,
      "model_forward_time": 1.7573657035827637,
      "step": 71090
    },
    {
      "dynamics_loss_avg": 0.06511203488335013,
      "epoch": 6.779670715332032e-05,
      "step": 71090
    },
    {
      "action_loss_avg": 0.0017969827400520443,
      "epoch": 6.779670715332032e-05,
      "step": 71090
    },
    {
      "epoch": 6.779670715332032e-05,
      "step": 71090,
      "training_step_time": 7.356060981750488
    },
    {
      "epoch": 6.779766082763671e-05,
      "model_forward_time": 1.7782578468322754,
      "step": 71091
    },
    {
      "epoch": 6.779766082763671e-05,
      "step": 71091,
      "training_step_time": 6.71127724647522
    },
    {
      "epoch": 6.779861450195312e-05,
      "model_forward_time": 1.768294334411621,
      "step": 71092
    },
    {
      "epoch": 6.779861450195312e-05,
      "step": 71092,
      "training_step_time": 6.725239276885986
    },
    {
      "epoch": 6.779956817626953e-05,
      "model_forward_time": 1.7673113346099854,
      "step": 71093
    },
    {
      "epoch": 6.779956817626953e-05,
      "step": 71093,
      "training_step_time": 6.6630330085754395
    },
    {
      "epoch": 6.780052185058593e-05,
      "model_forward_time": 1.757401704788208,
      "step": 71094
    },
    {
      "epoch": 6.780052185058593e-05,
      "step": 71094,
      "training_step_time": 6.69382905960083
    },
    {
      "epoch": 6.780147552490234e-05,
      "model_forward_time": 1.7678089141845703,
      "step": 71095
    },
    {
      "epoch": 6.780147552490234e-05,
      "step": 71095,
      "training_step_time": 6.716730833053589
    },
    {
      "epoch": 6.780242919921875e-05,
      "model_forward_time": 1.7648446559906006,
      "step": 71096
    },
    {
      "epoch": 6.780242919921875e-05,
      "step": 71096,
      "training_step_time": 6.675353050231934
    },
    {
      "epoch": 6.780338287353515e-05,
      "model_forward_time": 2.4078304767608643,
      "step": 71097
    },
    {
      "epoch": 6.780338287353515e-05,
      "step": 71097,
      "training_step_time": 8.248740673065186
    },
    {
      "epoch": 6.780433654785156e-05,
      "model_forward_time": 1.7535665035247803,
      "step": 71098
    },
    {
      "epoch": 6.780433654785156e-05,
      "step": 71098,
      "training_step_time": 7.840234756469727
    },
    {
      "epoch": 6.780529022216797e-05,
      "model_forward_time": 1.75132417678833,
      "step": 71099
    },
    {
      "epoch": 6.780529022216797e-05,
      "step": 71099,
      "training_step_time": 7.668276071548462
    },
    {
      "epoch": 6.780624389648437e-05,
      "grad_norm": 0.10117927193641663,
      "learning_rate": 2.1149717428824722e-06,
      "loss": 0.0575,
      "step": 71100
    },
    {
      "epoch": 6.780624389648437e-05,
      "model_forward_time": 1.040436029434204,
      "step": 71100
    },
    {
      "dynamics_loss_avg": 0.04225119203329086,
      "epoch": 6.780624389648437e-05,
      "step": 71100
    },
    {
      "action_loss_avg": 0.0015964079968398437,
      "epoch": 6.780624389648437e-05,
      "step": 71100
    },
    {
      "epoch": 6.780624389648437e-05,
      "step": 71100,
      "training_step_time": 6.6866395473480225
    },
    {
      "epoch": 6.780719757080078e-05,
      "model_forward_time": 1.773388385772705,
      "step": 71101
    },
    {
      "epoch": 6.780719757080078e-05,
      "step": 71101,
      "training_step_time": 6.692501783370972
    },
    {
      "epoch": 6.780815124511719e-05,
      "model_forward_time": 1.691950798034668,
      "step": 71102
    },
    {
      "epoch": 6.780815124511719e-05,
      "step": 71102,
      "training_step_time": 6.686512470245361
    },
    {
      "epoch": 6.780910491943359e-05,
      "model_forward_time": 1.7004592418670654,
      "step": 71103
    },
    {
      "epoch": 6.780910491943359e-05,
      "step": 71103,
      "training_step_time": 6.679235458374023
    },
    {
      "epoch": 6.781005859375e-05,
      "model_forward_time": 1.6905040740966797,
      "step": 71104
    },
    {
      "epoch": 6.781005859375e-05,
      "step": 71104,
      "training_step_time": 6.676439046859741
    },
    {
      "epoch": 6.781101226806641e-05,
      "model_forward_time": 1.6921303272247314,
      "step": 71105
    },
    {
      "epoch": 6.781101226806641e-05,
      "step": 71105,
      "training_step_time": 6.719764709472656
    },
    {
      "epoch": 6.781196594238281e-05,
      "model_forward_time": 2.2792742252349854,
      "step": 71106
    },
    {
      "epoch": 6.781196594238281e-05,
      "step": 71106,
      "training_step_time": 7.780551195144653
    },
    {
      "epoch": 6.781291961669922e-05,
      "model_forward_time": 1.3213555812835693,
      "step": 71107
    },
    {
      "epoch": 6.781291961669922e-05,
      "step": 71107,
      "training_step_time": 9.453927516937256
    },
    {
      "epoch": 6.781387329101563e-05,
      "model_forward_time": 1.7028963565826416,
      "step": 71108
    },
    {
      "epoch": 6.781387329101563e-05,
      "step": 71108,
      "training_step_time": 7.78815484046936
    },
    {
      "epoch": 6.781482696533203e-05,
      "model_forward_time": 1.7046234607696533,
      "step": 71109
    },
    {
      "epoch": 6.781482696533203e-05,
      "step": 71109,
      "training_step_time": 6.721090793609619
    },
    {
      "epoch": 6.781578063964844e-05,
      "grad_norm": 0.08319909125566483,
      "learning_rate": 2.1136214454656785e-06,
      "loss": 0.0567,
      "step": 71110
    },
    {
      "epoch": 6.781578063964844e-05,
      "model_forward_time": 1.6997647285461426,
      "step": 71110
    },
    {
      "dynamics_loss_avg": 0.08220841437578201,
      "epoch": 6.781578063964844e-05,
      "step": 71110
    },
    {
      "action_loss_avg": 0.002436611871235073,
      "epoch": 6.781578063964844e-05,
      "step": 71110
    },
    {
      "epoch": 6.781578063964844e-05,
      "step": 71110,
      "training_step_time": 6.785010814666748
    },
    {
      "epoch": 6.781673431396485e-05,
      "model_forward_time": 1.7002198696136475,
      "step": 71111
    },
    {
      "epoch": 6.781673431396485e-05,
      "step": 71111,
      "training_step_time": 6.717502593994141
    },
    {
      "epoch": 6.781768798828125e-05,
      "model_forward_time": 1.702697515487671,
      "step": 71112
    },
    {
      "epoch": 6.781768798828125e-05,
      "step": 71112,
      "training_step_time": 6.725681304931641
    },
    {
      "epoch": 6.781864166259766e-05,
      "model_forward_time": 1.7100884914398193,
      "step": 71113
    },
    {
      "epoch": 6.781864166259766e-05,
      "step": 71113,
      "training_step_time": 6.750643968582153
    },
    {
      "epoch": 6.781959533691407e-05,
      "model_forward_time": 1.3353161811828613,
      "step": 71114
    },
    {
      "epoch": 6.781959533691407e-05,
      "step": 71114,
      "training_step_time": 6.734943151473999
    },
    {
      "epoch": 6.782054901123046e-05,
      "model_forward_time": 1.7151610851287842,
      "step": 71115
    },
    {
      "epoch": 6.782054901123046e-05,
      "step": 71115,
      "training_step_time": 8.215923070907593
    },
    {
      "epoch": 6.782150268554687e-05,
      "model_forward_time": 2.2491817474365234,
      "step": 71116
    },
    {
      "epoch": 6.782150268554687e-05,
      "step": 71116,
      "training_step_time": 8.635321855545044
    },
    {
      "epoch": 6.782245635986329e-05,
      "model_forward_time": 1.702409029006958,
      "step": 71117
    },
    {
      "epoch": 6.782245635986329e-05,
      "step": 71117,
      "training_step_time": 8.355156421661377
    },
    {
      "epoch": 6.782341003417968e-05,
      "model_forward_time": 1.695335865020752,
      "step": 71118
    },
    {
      "epoch": 6.782341003417968e-05,
      "step": 71118,
      "training_step_time": 7.093314170837402
    },
    {
      "epoch": 6.78243637084961e-05,
      "model_forward_time": 1.700446367263794,
      "step": 71119
    },
    {
      "epoch": 6.78243637084961e-05,
      "step": 71119,
      "training_step_time": 6.678989887237549
    },
    {
      "epoch": 6.78253173828125e-05,
      "grad_norm": 0.09887702763080597,
      "learning_rate": 2.1122714636989055e-06,
      "loss": 0.0555,
      "step": 71120
    },
    {
      "epoch": 6.78253173828125e-05,
      "model_forward_time": 1.694279432296753,
      "step": 71120
    },
    {
      "dynamics_loss_avg": 0.05538457334041595,
      "epoch": 6.78253173828125e-05,
      "step": 71120
    },
    {
      "action_loss_avg": 0.0016375973587855696,
      "epoch": 6.78253173828125e-05,
      "step": 71120
    },
    {
      "epoch": 6.78253173828125e-05,
      "step": 71120,
      "training_step_time": 6.722576856613159
    },
    {
      "epoch": 6.78262710571289e-05,
      "model_forward_time": 1.6981253623962402,
      "step": 71121
    },
    {
      "epoch": 6.78262710571289e-05,
      "step": 71121,
      "training_step_time": 6.72696590423584
    },
    {
      "epoch": 6.782722473144531e-05,
      "model_forward_time": 1.6940128803253174,
      "step": 71122
    },
    {
      "epoch": 6.782722473144531e-05,
      "step": 71122,
      "training_step_time": 6.688509464263916
    },
    {
      "epoch": 6.782817840576172e-05,
      "model_forward_time": 1.6946280002593994,
      "step": 71123
    },
    {
      "epoch": 6.782817840576172e-05,
      "step": 71123,
      "training_step_time": 6.697807550430298
    },
    {
      "epoch": 6.782913208007812e-05,
      "model_forward_time": 1.718630075454712,
      "step": 71124
    },
    {
      "epoch": 6.782913208007812e-05,
      "step": 71124,
      "training_step_time": 7.65721321105957
    },
    {
      "epoch": 6.783008575439453e-05,
      "model_forward_time": 2.27388072013855,
      "step": 71125
    },
    {
      "epoch": 6.783008575439453e-05,
      "step": 71125,
      "training_step_time": 9.8165762424469
    },
    {
      "epoch": 6.783103942871094e-05,
      "model_forward_time": 1.3179843425750732,
      "step": 71126
    },
    {
      "epoch": 6.783103942871094e-05,
      "step": 71126,
      "training_step_time": 7.802783012390137
    },
    {
      "epoch": 6.783199310302734e-05,
      "model_forward_time": 1.704526424407959,
      "step": 71127
    },
    {
      "epoch": 6.783199310302734e-05,
      "step": 71127,
      "training_step_time": 7.091579914093018
    },
    {
      "epoch": 6.783294677734375e-05,
      "model_forward_time": 0.9838166236877441,
      "step": 71128
    },
    {
      "epoch": 6.783294677734375e-05,
      "step": 71128,
      "training_step_time": 6.7465736865997314
    },
    {
      "epoch": 6.783390045166016e-05,
      "model_forward_time": 1.703026294708252,
      "step": 71129
    },
    {
      "epoch": 6.783390045166016e-05,
      "step": 71129,
      "training_step_time": 6.7273268699646
    },
    {
      "epoch": 6.783485412597656e-05,
      "grad_norm": 0.12071739882230759,
      "learning_rate": 2.1109217977297826e-06,
      "loss": 0.056,
      "step": 71130
    },
    {
      "epoch": 6.783485412597656e-05,
      "model_forward_time": 1.694725513458252,
      "step": 71130
    },
    {
      "dynamics_loss_avg": 0.04239637795835734,
      "epoch": 6.783485412597656e-05,
      "step": 71130
    },
    {
      "action_loss_avg": 0.0009820089559070765,
      "epoch": 6.783485412597656e-05,
      "step": 71130
    },
    {
      "epoch": 6.783485412597656e-05,
      "step": 71130,
      "training_step_time": 6.733541965484619
    },
    {
      "epoch": 6.783580780029297e-05,
      "model_forward_time": 1.7014505863189697,
      "step": 71131
    },
    {
      "epoch": 6.783580780029297e-05,
      "step": 71131,
      "training_step_time": 6.792353868484497
    },
    {
      "epoch": 6.783676147460938e-05,
      "model_forward_time": 1.6986896991729736,
      "step": 71132
    },
    {
      "epoch": 6.783676147460938e-05,
      "step": 71132,
      "training_step_time": 6.75322961807251
    },
    {
      "epoch": 6.783771514892578e-05,
      "model_forward_time": 1.7485706806182861,
      "step": 71133
    },
    {
      "epoch": 6.783771514892578e-05,
      "step": 71133,
      "training_step_time": 7.690154790878296
    },
    {
      "epoch": 6.783866882324219e-05,
      "model_forward_time": 2.872821569442749,
      "step": 71134
    },
    {
      "epoch": 6.783866882324219e-05,
      "step": 71134,
      "training_step_time": 8.773507118225098
    },
    {
      "epoch": 6.78396224975586e-05,
      "model_forward_time": 1.7791059017181396,
      "step": 71135
    },
    {
      "epoch": 6.78396224975586e-05,
      "step": 71135,
      "training_step_time": 9.65803337097168
    },
    {
      "epoch": 6.7840576171875e-05,
      "model_forward_time": 1.7643170356750488,
      "step": 71136
    },
    {
      "epoch": 6.7840576171875e-05,
      "step": 71136,
      "training_step_time": 7.644287586212158
    },
    {
      "epoch": 6.78415298461914e-05,
      "model_forward_time": 1.7633018493652344,
      "step": 71137
    },
    {
      "epoch": 6.78415298461914e-05,
      "step": 71137,
      "training_step_time": 6.768767356872559
    },
    {
      "epoch": 6.784248352050782e-05,
      "model_forward_time": 1.761129379272461,
      "step": 71138
    },
    {
      "epoch": 6.784248352050782e-05,
      "step": 71138,
      "training_step_time": 6.741266489028931
    },
    {
      "epoch": 6.784343719482421e-05,
      "model_forward_time": 1.767883539199829,
      "step": 71139
    },
    {
      "epoch": 6.784343719482421e-05,
      "step": 71139,
      "training_step_time": 6.769561767578125
    },
    {
      "epoch": 6.784439086914063e-05,
      "grad_norm": 0.12016172707080841,
      "learning_rate": 2.1095724477059077e-06,
      "loss": 0.0541,
      "step": 71140
    },
    {
      "epoch": 6.784439086914063e-05,
      "model_forward_time": 1.7734549045562744,
      "step": 71140
    },
    {
      "dynamics_loss_avg": 0.04939758665859699,
      "epoch": 6.784439086914063e-05,
      "step": 71140
    },
    {
      "action_loss_avg": 0.0011208905198145658,
      "epoch": 6.784439086914063e-05,
      "step": 71140
    },
    {
      "epoch": 6.784439086914063e-05,
      "step": 71140,
      "training_step_time": 6.751476526260376
    },
    {
      "epoch": 6.784534454345704e-05,
      "model_forward_time": 1.7698156833648682,
      "step": 71141
    },
    {
      "epoch": 6.784534454345704e-05,
      "step": 71141,
      "training_step_time": 6.757091760635376
    },
    {
      "epoch": 6.784629821777343e-05,
      "model_forward_time": 1.7825021743774414,
      "step": 71142
    },
    {
      "epoch": 6.784629821777343e-05,
      "step": 71142,
      "training_step_time": 7.600679636001587
    },
    {
      "epoch": 6.784725189208984e-05,
      "model_forward_time": 1.7600853443145752,
      "step": 71143
    },
    {
      "epoch": 6.784725189208984e-05,
      "step": 71143,
      "training_step_time": 7.605535984039307
    },
    {
      "epoch": 6.784820556640626e-05,
      "model_forward_time": 2.9395558834075928,
      "step": 71144
    },
    {
      "epoch": 6.784820556640626e-05,
      "step": 71144,
      "training_step_time": 9.474689483642578
    },
    {
      "epoch": 6.784915924072265e-05,
      "model_forward_time": 1.7597503662109375,
      "step": 71145
    },
    {
      "epoch": 6.784915924072265e-05,
      "step": 71145,
      "training_step_time": 7.706009149551392
    },
    {
      "epoch": 6.785011291503906e-05,
      "model_forward_time": 1.7626185417175293,
      "step": 71146
    },
    {
      "epoch": 6.785011291503906e-05,
      "step": 71146,
      "training_step_time": 6.739307641983032
    },
    {
      "epoch": 6.785106658935547e-05,
      "model_forward_time": 1.758687973022461,
      "step": 71147
    },
    {
      "epoch": 6.785106658935547e-05,
      "step": 71147,
      "training_step_time": 6.7123329639434814
    },
    {
      "epoch": 6.785202026367187e-05,
      "model_forward_time": 1.7686119079589844,
      "step": 71148
    },
    {
      "epoch": 6.785202026367187e-05,
      "step": 71148,
      "training_step_time": 6.724241018295288
    },
    {
      "epoch": 6.785297393798828e-05,
      "model_forward_time": 1.831113338470459,
      "step": 71149
    },
    {
      "epoch": 6.785297393798828e-05,
      "step": 71149,
      "training_step_time": 6.697453022003174
    },
    {
      "epoch": 6.785392761230469e-05,
      "grad_norm": 0.08333923667669296,
      "learning_rate": 2.108223413774844e-06,
      "loss": 0.0516,
      "step": 71150
    },
    {
      "epoch": 6.785392761230469e-05,
      "model_forward_time": 1.7435145378112793,
      "step": 71150
    },
    {
      "dynamics_loss_avg": 0.06116214729845524,
      "epoch": 6.785392761230469e-05,
      "step": 71150
    },
    {
      "action_loss_avg": 0.0013456430518999697,
      "epoch": 6.785392761230469e-05,
      "step": 71150
    },
    {
      "epoch": 6.785392761230469e-05,
      "step": 71150,
      "training_step_time": 6.720441818237305
    },
    {
      "epoch": 6.785488128662109e-05,
      "model_forward_time": 1.769975185394287,
      "step": 71151
    },
    {
      "epoch": 6.785488128662109e-05,
      "step": 71151,
      "training_step_time": 7.640765190124512
    },
    {
      "epoch": 6.78558349609375e-05,
      "model_forward_time": 1.871309757232666,
      "step": 71152
    },
    {
      "epoch": 6.78558349609375e-05,
      "step": 71152,
      "training_step_time": 8.02754545211792
    },
    {
      "epoch": 6.785678863525391e-05,
      "model_forward_time": 2.9105937480926514,
      "step": 71153
    },
    {
      "epoch": 6.785678863525391e-05,
      "step": 71153,
      "training_step_time": 9.831180572509766
    },
    {
      "epoch": 6.785774230957031e-05,
      "model_forward_time": 1.7648229598999023,
      "step": 71154
    },
    {
      "epoch": 6.785774230957031e-05,
      "step": 71154,
      "training_step_time": 7.824578285217285
    },
    {
      "epoch": 6.785869598388672e-05,
      "model_forward_time": 1.77032470703125,
      "step": 71155
    },
    {
      "epoch": 6.785869598388672e-05,
      "step": 71155,
      "training_step_time": 6.703065395355225
    },
    {
      "epoch": 6.785964965820313e-05,
      "model_forward_time": 1.7596347332000732,
      "step": 71156
    },
    {
      "epoch": 6.785964965820313e-05,
      "step": 71156,
      "training_step_time": 6.760035037994385
    },
    {
      "epoch": 6.786060333251953e-05,
      "model_forward_time": 1.7649528980255127,
      "step": 71157
    },
    {
      "epoch": 6.786060333251953e-05,
      "step": 71157,
      "training_step_time": 6.740421295166016
    },
    {
      "epoch": 6.786155700683594e-05,
      "model_forward_time": 1.764307975769043,
      "step": 71158
    },
    {
      "epoch": 6.786155700683594e-05,
      "step": 71158,
      "training_step_time": 6.754194736480713
    },
    {
      "epoch": 6.786251068115235e-05,
      "model_forward_time": 1.7640798091888428,
      "step": 71159
    },
    {
      "epoch": 6.786251068115235e-05,
      "step": 71159,
      "training_step_time": 6.713623046875
    },
    {
      "epoch": 6.786346435546875e-05,
      "grad_norm": 0.10080127418041229,
      "learning_rate": 2.106874696084122e-06,
      "loss": 0.0546,
      "step": 71160
    },
    {
      "epoch": 6.786346435546875e-05,
      "model_forward_time": 1.7640676498413086,
      "step": 71160
    },
    {
      "dynamics_loss_avg": 0.05931534245610237,
      "epoch": 6.786346435546875e-05,
      "step": 71160
    },
    {
      "action_loss_avg": 0.0020970867830328644,
      "epoch": 6.786346435546875e-05,
      "step": 71160
    },
    {
      "epoch": 6.786346435546875e-05,
      "step": 71160,
      "training_step_time": 7.6406121253967285
    },
    {
      "epoch": 6.786441802978516e-05,
      "model_forward_time": 1.7641229629516602,
      "step": 71161
    },
    {
      "epoch": 6.786441802978516e-05,
      "step": 71161,
      "training_step_time": 7.489904165267944
    },
    {
      "epoch": 6.786537170410157e-05,
      "model_forward_time": 2.9252374172210693,
      "step": 71162
    },
    {
      "epoch": 6.786537170410157e-05,
      "step": 71162,
      "training_step_time": 9.076284646987915
    },
    {
      "epoch": 6.786632537841797e-05,
      "model_forward_time": 1.7635271549224854,
      "step": 71163
    },
    {
      "epoch": 6.786632537841797e-05,
      "step": 71163,
      "training_step_time": 9.473153591156006
    },
    {
      "epoch": 6.786727905273438e-05,
      "model_forward_time": 1.7726850509643555,
      "step": 71164
    },
    {
      "epoch": 6.786727905273438e-05,
      "step": 71164,
      "training_step_time": 7.140417814254761
    },
    {
      "epoch": 6.786823272705079e-05,
      "model_forward_time": 1.7622911930084229,
      "step": 71165
    },
    {
      "epoch": 6.786823272705079e-05,
      "step": 71165,
      "training_step_time": 6.770410060882568
    },
    {
      "epoch": 6.786918640136718e-05,
      "model_forward_time": 1.7689013481140137,
      "step": 71166
    },
    {
      "epoch": 6.786918640136718e-05,
      "step": 71166,
      "training_step_time": 6.793509006500244
    },
    {
      "epoch": 6.78701400756836e-05,
      "model_forward_time": 1.7622268199920654,
      "step": 71167
    },
    {
      "epoch": 6.78701400756836e-05,
      "step": 71167,
      "training_step_time": 6.799100399017334
    },
    {
      "epoch": 6.787109375e-05,
      "model_forward_time": 1.771817922592163,
      "step": 71168
    },
    {
      "epoch": 6.787109375e-05,
      "step": 71168,
      "training_step_time": 6.8344643115997314
    },
    {
      "epoch": 6.78720474243164e-05,
      "model_forward_time": 1.7560827732086182,
      "step": 71169
    },
    {
      "epoch": 6.78720474243164e-05,
      "step": 71169,
      "training_step_time": 7.693146705627441
    },
    {
      "epoch": 6.787300109863281e-05,
      "grad_norm": 0.14708437025547028,
      "learning_rate": 2.1055262947812315e-06,
      "loss": 0.0519,
      "step": 71170
    },
    {
      "epoch": 6.787300109863281e-05,
      "model_forward_time": 1.7645115852355957,
      "step": 71170
    },
    {
      "dynamics_loss_avg": 0.06226331125944853,
      "epoch": 6.787300109863281e-05,
      "step": 71170
    },
    {
      "action_loss_avg": 0.0023646824294701217,
      "epoch": 6.787300109863281e-05,
      "step": 71170
    },
    {
      "epoch": 6.787300109863281e-05,
      "step": 71170,
      "training_step_time": 7.589245557785034
    },
    {
      "epoch": 6.787395477294922e-05,
      "model_forward_time": 2.959754705429077,
      "step": 71171
    },
    {
      "epoch": 6.787395477294922e-05,
      "step": 71171,
      "training_step_time": 9.912264585494995
    },
    {
      "epoch": 6.787490844726562e-05,
      "model_forward_time": 1.7676308155059814,
      "step": 71172
    },
    {
      "epoch": 6.787490844726562e-05,
      "step": 71172,
      "training_step_time": 8.551781177520752
    },
    {
      "epoch": 6.787586212158203e-05,
      "model_forward_time": 1.7982826232910156,
      "step": 71173
    },
    {
      "epoch": 6.787586212158203e-05,
      "step": 71173,
      "training_step_time": 7.810171604156494
    },
    {
      "epoch": 6.787681579589844e-05,
      "model_forward_time": 1.767042875289917,
      "step": 71174
    },
    {
      "epoch": 6.787681579589844e-05,
      "step": 71174,
      "training_step_time": 6.930176258087158
    },
    {
      "epoch": 6.787776947021484e-05,
      "model_forward_time": 1.772956371307373,
      "step": 71175
    },
    {
      "epoch": 6.787776947021484e-05,
      "step": 71175,
      "training_step_time": 6.897305727005005
    },
    {
      "epoch": 6.787872314453125e-05,
      "model_forward_time": 1.763733148574829,
      "step": 71176
    },
    {
      "epoch": 6.787872314453125e-05,
      "step": 71176,
      "training_step_time": 6.877042055130005
    },
    {
      "epoch": 6.787967681884766e-05,
      "model_forward_time": 1.75838303565979,
      "step": 71177
    },
    {
      "epoch": 6.787967681884766e-05,
      "step": 71177,
      "training_step_time": 6.893630266189575
    },
    {
      "epoch": 6.788063049316406e-05,
      "model_forward_time": 1.753709316253662,
      "step": 71178
    },
    {
      "epoch": 6.788063049316406e-05,
      "step": 71178,
      "training_step_time": 7.769737482070923
    },
    {
      "epoch": 6.788158416748047e-05,
      "model_forward_time": 1.7618629932403564,
      "step": 71179
    },
    {
      "epoch": 6.788158416748047e-05,
      "step": 71179,
      "training_step_time": 8.685160160064697
    },
    {
      "epoch": 6.788253784179688e-05,
      "grad_norm": 0.07486803829669952,
      "learning_rate": 2.104178210013636e-06,
      "loss": 0.057,
      "step": 71180
    },
    {
      "epoch": 6.788253784179688e-05,
      "model_forward_time": 2.9311373233795166,
      "step": 71180
    },
    {
      "dynamics_loss_avg": 0.06247839666903019,
      "epoch": 6.788253784179688e-05,
      "step": 71180
    },
    {
      "action_loss_avg": 0.0015809561911737547,
      "epoch": 6.788253784179688e-05,
      "step": 71180
    },
    {
      "epoch": 6.788253784179688e-05,
      "step": 71180,
      "training_step_time": 7.949136018753052
    },
    {
      "epoch": 6.788349151611328e-05,
      "model_forward_time": 1.761436939239502,
      "step": 71181
    },
    {
      "epoch": 6.788349151611328e-05,
      "step": 71181,
      "training_step_time": 9.848747491836548
    },
    {
      "epoch": 6.788444519042969e-05,
      "model_forward_time": 1.766676425933838,
      "step": 71182
    },
    {
      "epoch": 6.788444519042969e-05,
      "step": 71182,
      "training_step_time": 7.875343561172485
    },
    {
      "epoch": 6.78853988647461e-05,
      "model_forward_time": 1.763688564300537,
      "step": 71183
    },
    {
      "epoch": 6.78853988647461e-05,
      "step": 71183,
      "training_step_time": 6.972122669219971
    },
    {
      "epoch": 6.78863525390625e-05,
      "model_forward_time": 1.7597253322601318,
      "step": 71184
    },
    {
      "epoch": 6.78863525390625e-05,
      "step": 71184,
      "training_step_time": 6.934279441833496
    },
    {
      "epoch": 6.788730621337891e-05,
      "model_forward_time": 1.7590787410736084,
      "step": 71185
    },
    {
      "epoch": 6.788730621337891e-05,
      "step": 71185,
      "training_step_time": 6.928384304046631
    },
    {
      "epoch": 6.788825988769532e-05,
      "model_forward_time": 1.7381846904754639,
      "step": 71186
    },
    {
      "epoch": 6.788825988769532e-05,
      "step": 71186,
      "training_step_time": 6.968145847320557
    },
    {
      "epoch": 6.788921356201172e-05,
      "model_forward_time": 1.750868797302246,
      "step": 71187
    },
    {
      "epoch": 6.788921356201172e-05,
      "step": 71187,
      "training_step_time": 7.786854028701782
    },
    {
      "epoch": 6.789016723632813e-05,
      "model_forward_time": 1.771066665649414,
      "step": 71188
    },
    {
      "epoch": 6.789016723632813e-05,
      "step": 71188,
      "training_step_time": 7.439765930175781
    },
    {
      "epoch": 6.789112091064452e-05,
      "model_forward_time": 1.7864484786987305,
      "step": 71189
    },
    {
      "epoch": 6.789112091064452e-05,
      "step": 71189,
      "training_step_time": 10.014026403427124
    },
    {
      "epoch": 6.789207458496094e-05,
      "grad_norm": 0.08122087270021439,
      "learning_rate": 2.1028304419287553e-06,
      "loss": 0.056,
      "step": 71190
    },
    {
      "epoch": 6.789207458496094e-05,
      "model_forward_time": 1.7758474349975586,
      "step": 71190
    },
    {
      "dynamics_loss_avg": 0.07454478107392788,
      "epoch": 6.789207458496094e-05,
      "step": 71190
    },
    {
      "action_loss_avg": 0.002027638102299534,
      "epoch": 6.789207458496094e-05,
      "step": 71190
    },
    {
      "epoch": 6.789207458496094e-05,
      "step": 71190,
      "training_step_time": 9.49995470046997
    },
    {
      "epoch": 6.789302825927735e-05,
      "model_forward_time": 1.754131555557251,
      "step": 71191
    },
    {
      "epoch": 6.789302825927735e-05,
      "step": 71191,
      "training_step_time": 8.505061149597168
    },
    {
      "epoch": 6.789398193359374e-05,
      "model_forward_time": 1.7522892951965332,
      "step": 71192
    },
    {
      "epoch": 6.789398193359374e-05,
      "step": 71192,
      "training_step_time": 7.687273025512695
    },
    {
      "epoch": 6.789493560791015e-05,
      "model_forward_time": 1.752953290939331,
      "step": 71193
    },
    {
      "epoch": 6.789493560791015e-05,
      "step": 71193,
      "training_step_time": 6.880008697509766
    },
    {
      "epoch": 6.789588928222656e-05,
      "model_forward_time": 1.7568376064300537,
      "step": 71194
    },
    {
      "epoch": 6.789588928222656e-05,
      "step": 71194,
      "training_step_time": 6.878490209579468
    },
    {
      "epoch": 6.789684295654296e-05,
      "model_forward_time": 1.753845453262329,
      "step": 71195
    },
    {
      "epoch": 6.789684295654296e-05,
      "step": 71195,
      "training_step_time": 6.886890411376953
    },
    {
      "epoch": 6.789779663085937e-05,
      "model_forward_time": 1.7446999549865723,
      "step": 71196
    },
    {
      "epoch": 6.789779663085937e-05,
      "step": 71196,
      "training_step_time": 7.722024202346802
    },
    {
      "epoch": 6.789875030517578e-05,
      "model_forward_time": 1.7556450366973877,
      "step": 71197
    },
    {
      "epoch": 6.789875030517578e-05,
      "step": 71197,
      "training_step_time": 7.37303614616394
    },
    {
      "epoch": 6.789970397949218e-05,
      "model_forward_time": 1.7530088424682617,
      "step": 71198
    },
    {
      "epoch": 6.789970397949218e-05,
      "step": 71198,
      "training_step_time": 8.162359237670898
    },
    {
      "epoch": 6.790065765380859e-05,
      "model_forward_time": 1.6832685470581055,
      "step": 71199
    },
    {
      "epoch": 6.790065765380859e-05,
      "step": 71199,
      "training_step_time": 9.784146308898926
    },
    {
      "epoch": 6.7901611328125e-05,
      "grad_norm": 0.11894506961107254,
      "learning_rate": 2.101482990673982e-06,
      "loss": 0.0545,
      "step": 71200
    },
    {
      "epoch": 6.7901611328125e-05,
      "model_forward_time": 1.7019076347351074,
      "step": 71200
    },
    {
      "dynamics_loss_avg": 0.06106564272195101,
      "epoch": 6.7901611328125e-05,
      "step": 71200
    },
    {
      "action_loss_avg": 0.001184646735782735,
      "epoch": 6.7901611328125e-05,
      "step": 71200
    },
    {
      "epoch": 6.7901611328125e-05,
      "step": 71200,
      "training_step_time": 8.509855270385742
    },
    {
      "epoch": 6.79025650024414e-05,
      "model_forward_time": 0.9828784465789795,
      "step": 71201
    },
    {
      "epoch": 6.79025650024414e-05,
      "step": 71201,
      "training_step_time": 7.82000732421875
    },
    {
      "epoch": 6.790351867675781e-05,
      "model_forward_time": 1.7081773281097412,
      "step": 71202
    },
    {
      "epoch": 6.790351867675781e-05,
      "step": 71202,
      "training_step_time": 6.759560585021973
    },
    {
      "epoch": 6.790447235107422e-05,
      "model_forward_time": 1.706007480621338,
      "step": 71203
    },
    {
      "epoch": 6.790447235107422e-05,
      "step": 71203,
      "training_step_time": 6.910272121429443
    },
    {
      "epoch": 6.790542602539062e-05,
      "model_forward_time": 1.7052943706512451,
      "step": 71204
    },
    {
      "epoch": 6.790542602539062e-05,
      "step": 71204,
      "training_step_time": 6.919336795806885
    },
    {
      "epoch": 6.790637969970703e-05,
      "model_forward_time": 1.7311649322509766,
      "step": 71205
    },
    {
      "epoch": 6.790637969970703e-05,
      "step": 71205,
      "training_step_time": 7.725980281829834
    },
    {
      "epoch": 6.790733337402344e-05,
      "model_forward_time": 1.7097914218902588,
      "step": 71206
    },
    {
      "epoch": 6.790733337402344e-05,
      "step": 71206,
      "training_step_time": 7.380636692047119
    },
    {
      "epoch": 6.790828704833984e-05,
      "model_forward_time": 1.716135025024414,
      "step": 71207
    },
    {
      "epoch": 6.790828704833984e-05,
      "step": 71207,
      "training_step_time": 8.270551681518555
    },
    {
      "epoch": 6.790924072265625e-05,
      "model_forward_time": 2.260141611099243,
      "step": 71208
    },
    {
      "epoch": 6.790924072265625e-05,
      "step": 71208,
      "training_step_time": 10.969962120056152
    },
    {
      "epoch": 6.791019439697266e-05,
      "model_forward_time": 1.7121622562408447,
      "step": 71209
    },
    {
      "epoch": 6.791019439697266e-05,
      "step": 71209,
      "training_step_time": 8.055763959884644
    },
    {
      "epoch": 6.791114807128906e-05,
      "grad_norm": 0.10342767089605331,
      "learning_rate": 2.1001358563966707e-06,
      "loss": 0.056,
      "step": 71210
    },
    {
      "epoch": 6.791114807128906e-05,
      "model_forward_time": 1.7088522911071777,
      "step": 71210
    },
    {
      "dynamics_loss_avg": 0.04113854989409447,
      "epoch": 6.791114807128906e-05,
      "step": 71210
    },
    {
      "action_loss_avg": 0.0013937454728875308,
      "epoch": 6.791114807128906e-05,
      "step": 71210
    },
    {
      "epoch": 6.791114807128906e-05,
      "step": 71210,
      "training_step_time": 7.796974182128906
    },
    {
      "epoch": 6.791210174560547e-05,
      "model_forward_time": 1.7158682346343994,
      "step": 71211
    },
    {
      "epoch": 6.791210174560547e-05,
      "step": 71211,
      "training_step_time": 6.902043581008911
    },
    {
      "epoch": 6.791305541992188e-05,
      "model_forward_time": 1.7129757404327393,
      "step": 71212
    },
    {
      "epoch": 6.791305541992188e-05,
      "step": 71212,
      "training_step_time": 6.921125173568726
    },
    {
      "epoch": 6.791400909423828e-05,
      "model_forward_time": 1.7097880840301514,
      "step": 71213
    },
    {
      "epoch": 6.791400909423828e-05,
      "step": 71213,
      "training_step_time": 6.795292139053345
    },
    {
      "epoch": 6.791496276855469e-05,
      "model_forward_time": 1.7406005859375,
      "step": 71214
    },
    {
      "epoch": 6.791496276855469e-05,
      "step": 71214,
      "training_step_time": 7.876631736755371
    },
    {
      "epoch": 6.79159164428711e-05,
      "model_forward_time": 1.7075629234313965,
      "step": 71215
    },
    {
      "epoch": 6.79159164428711e-05,
      "step": 71215,
      "training_step_time": 6.905445337295532
    },
    {
      "epoch": 6.79168701171875e-05,
      "model_forward_time": 1.718170166015625,
      "step": 71216
    },
    {
      "epoch": 6.79168701171875e-05,
      "step": 71216,
      "training_step_time": 7.759674787521362
    },
    {
      "epoch": 6.79178237915039e-05,
      "model_forward_time": 1.5273540019989014,
      "step": 71217
    },
    {
      "epoch": 6.79178237915039e-05,
      "step": 71217,
      "training_step_time": 8.859474897384644
    },
    {
      "epoch": 6.791877746582032e-05,
      "model_forward_time": 1.711045265197754,
      "step": 71218
    },
    {
      "epoch": 6.791877746582032e-05,
      "step": 71218,
      "training_step_time": 8.018453121185303
    },
    {
      "epoch": 6.791973114013671e-05,
      "model_forward_time": 1.7176735401153564,
      "step": 71219
    },
    {
      "epoch": 6.791973114013671e-05,
      "step": 71219,
      "training_step_time": 7.792543649673462
    },
    {
      "epoch": 6.792068481445312e-05,
      "grad_norm": 0.09045002609491348,
      "learning_rate": 2.098789039244144e-06,
      "loss": 0.05,
      "step": 71220
    },
    {
      "epoch": 6.792068481445312e-05,
      "model_forward_time": 1.714953899383545,
      "step": 71220
    },
    {
      "dynamics_loss_avg": 0.0448008163832128,
      "epoch": 6.792068481445312e-05,
      "step": 71220
    },
    {
      "action_loss_avg": 0.0016919514862820506,
      "epoch": 6.792068481445312e-05,
      "step": 71220
    },
    {
      "epoch": 6.792068481445312e-05,
      "step": 71220,
      "training_step_time": 6.875222444534302
    },
    {
      "epoch": 6.792163848876953e-05,
      "model_forward_time": 1.7081856727600098,
      "step": 71221
    },
    {
      "epoch": 6.792163848876953e-05,
      "step": 71221,
      "training_step_time": 6.886605739593506
    },
    {
      "epoch": 6.792259216308593e-05,
      "model_forward_time": 1.7090721130371094,
      "step": 71222
    },
    {
      "epoch": 6.792259216308593e-05,
      "step": 71222,
      "training_step_time": 6.932164907455444
    },
    {
      "epoch": 6.792354583740234e-05,
      "model_forward_time": 1.7311985492706299,
      "step": 71223
    },
    {
      "epoch": 6.792354583740234e-05,
      "step": 71223,
      "training_step_time": 7.704326868057251
    },
    {
      "epoch": 6.792449951171875e-05,
      "model_forward_time": 1.6970937252044678,
      "step": 71224
    },
    {
      "epoch": 6.792449951171875e-05,
      "step": 71224,
      "training_step_time": 6.8942952156066895
    },
    {
      "epoch": 6.792545318603515e-05,
      "model_forward_time": 1.707674503326416,
      "step": 71225
    },
    {
      "epoch": 6.792545318603515e-05,
      "step": 71225,
      "training_step_time": 8.716367244720459
    },
    {
      "epoch": 6.792640686035156e-05,
      "model_forward_time": 2.2381041049957275,
      "step": 71226
    },
    {
      "epoch": 6.792640686035156e-05,
      "step": 71226,
      "training_step_time": 10.243896961212158
    },
    {
      "epoch": 6.792736053466797e-05,
      "model_forward_time": 1.7019357681274414,
      "step": 71227
    },
    {
      "epoch": 6.792736053466797e-05,
      "step": 71227,
      "training_step_time": 9.686521053314209
    },
    {
      "epoch": 6.792831420898437e-05,
      "model_forward_time": 1.7127351760864258,
      "step": 71228
    },
    {
      "epoch": 6.792831420898437e-05,
      "step": 71228,
      "training_step_time": 8.864517450332642
    },
    {
      "epoch": 6.792926788330078e-05,
      "model_forward_time": 1.7070813179016113,
      "step": 71229
    },
    {
      "epoch": 6.792926788330078e-05,
      "step": 71229,
      "training_step_time": 7.607299327850342
    },
    {
      "epoch": 6.793022155761719e-05,
      "grad_norm": 0.07433360815048218,
      "learning_rate": 2.097442539363685e-06,
      "loss": 0.0513,
      "step": 71230
    },
    {
      "epoch": 6.793022155761719e-05,
      "model_forward_time": 1.703862190246582,
      "step": 71230
    },
    {
      "dynamics_loss_avg": 0.047935849614441396,
      "epoch": 6.793022155761719e-05,
      "step": 71230
    },
    {
      "action_loss_avg": 0.001141126052243635,
      "epoch": 6.793022155761719e-05,
      "step": 71230
    },
    {
      "epoch": 6.793022155761719e-05,
      "step": 71230,
      "training_step_time": 6.884820938110352
    },
    {
      "epoch": 6.793117523193359e-05,
      "model_forward_time": 1.708672285079956,
      "step": 71231
    },
    {
      "epoch": 6.793117523193359e-05,
      "step": 71231,
      "training_step_time": 6.91835355758667
    },
    {
      "epoch": 6.793212890625e-05,
      "model_forward_time": 1.7361674308776855,
      "step": 71232
    },
    {
      "epoch": 6.793212890625e-05,
      "step": 71232,
      "training_step_time": 7.724439859390259
    },
    {
      "epoch": 6.793308258056641e-05,
      "model_forward_time": 1.7065966129302979,
      "step": 71233
    },
    {
      "epoch": 6.793308258056641e-05,
      "step": 71233,
      "training_step_time": 6.903186559677124
    },
    {
      "epoch": 6.793403625488281e-05,
      "model_forward_time": 1.7012608051300049,
      "step": 71234
    },
    {
      "epoch": 6.793403625488281e-05,
      "step": 71234,
      "training_step_time": 8.104859590530396
    },
    {
      "epoch": 6.793498992919922e-05,
      "model_forward_time": 1.7264466285705566,
      "step": 71235
    },
    {
      "epoch": 6.793498992919922e-05,
      "step": 71235,
      "training_step_time": 10.52667784690857
    },
    {
      "epoch": 6.793594360351563e-05,
      "model_forward_time": 1.8838865756988525,
      "step": 71236
    },
    {
      "epoch": 6.793594360351563e-05,
      "step": 71236,
      "training_step_time": 9.602033615112305
    },
    {
      "epoch": 6.793689727783203e-05,
      "model_forward_time": 1.7032864093780518,
      "step": 71237
    },
    {
      "epoch": 6.793689727783203e-05,
      "step": 71237,
      "training_step_time": 9.039902448654175
    },
    {
      "epoch": 6.793785095214844e-05,
      "model_forward_time": 1.7136297225952148,
      "step": 71238
    },
    {
      "epoch": 6.793785095214844e-05,
      "step": 71238,
      "training_step_time": 7.819026947021484
    },
    {
      "epoch": 6.793880462646485e-05,
      "model_forward_time": 1.7017343044281006,
      "step": 71239
    },
    {
      "epoch": 6.793880462646485e-05,
      "step": 71239,
      "training_step_time": 6.872697591781616
    },
    {
      "epoch": 6.793975830078125e-05,
      "grad_norm": 0.07429344207048416,
      "learning_rate": 2.096096356902545e-06,
      "loss": 0.0556,
      "step": 71240
    },
    {
      "epoch": 6.793975830078125e-05,
      "model_forward_time": 1.7007567882537842,
      "step": 71240
    },
    {
      "dynamics_loss_avg": 0.035298421420156954,
      "epoch": 6.793975830078125e-05,
      "step": 71240
    },
    {
      "action_loss_avg": 0.0008471384571748786,
      "epoch": 6.793975830078125e-05,
      "step": 71240
    },
    {
      "epoch": 6.793975830078125e-05,
      "step": 71240,
      "training_step_time": 6.905219554901123
    },
    {
      "epoch": 6.794071197509766e-05,
      "model_forward_time": 1.7309961318969727,
      "step": 71241
    },
    {
      "epoch": 6.794071197509766e-05,
      "step": 71241,
      "training_step_time": 7.773756980895996
    },
    {
      "epoch": 6.794166564941407e-05,
      "model_forward_time": 1.707263708114624,
      "step": 71242
    },
    {
      "epoch": 6.794166564941407e-05,
      "step": 71242,
      "training_step_time": 6.908721923828125
    },
    {
      "epoch": 6.794261932373046e-05,
      "model_forward_time": 1.705244779586792,
      "step": 71243
    },
    {
      "epoch": 6.794261932373046e-05,
      "step": 71243,
      "training_step_time": 7.9385035037994385
    },
    {
      "epoch": 6.794357299804687e-05,
      "model_forward_time": 1.7203667163848877,
      "step": 71244
    },
    {
      "epoch": 6.794357299804687e-05,
      "step": 71244,
      "training_step_time": 8.797460556030273
    },
    {
      "epoch": 6.794452667236329e-05,
      "model_forward_time": 2.259222984313965,
      "step": 71245
    },
    {
      "epoch": 6.794452667236329e-05,
      "step": 71245,
      "training_step_time": 10.780813694000244
    },
    {
      "epoch": 6.794548034667968e-05,
      "model_forward_time": 1.7115771770477295,
      "step": 71246
    },
    {
      "epoch": 6.794548034667968e-05,
      "step": 71246,
      "training_step_time": 8.028315782546997
    },
    {
      "epoch": 6.79464340209961e-05,
      "model_forward_time": 1.7049708366394043,
      "step": 71247
    },
    {
      "epoch": 6.79464340209961e-05,
      "step": 71247,
      "training_step_time": 7.726480007171631
    },
    {
      "epoch": 6.79473876953125e-05,
      "model_forward_time": 1.709141731262207,
      "step": 71248
    },
    {
      "epoch": 6.79473876953125e-05,
      "step": 71248,
      "training_step_time": 6.864088535308838
    },
    {
      "epoch": 6.79483413696289e-05,
      "model_forward_time": 1.7083492279052734,
      "step": 71249
    },
    {
      "epoch": 6.79483413696289e-05,
      "step": 71249,
      "training_step_time": 6.7579429149627686
    },
    {
      "epoch": 6.794929504394531e-05,
      "grad_norm": 0.08545927703380585,
      "learning_rate": 2.0947504920079437e-06,
      "loss": 0.0561,
      "step": 71250
    },
    {
      "epoch": 6.794929504394531e-05,
      "model_forward_time": 1.7246818542480469,
      "step": 71250
    },
    {
      "dynamics_loss_avg": 0.06205096775665879,
      "epoch": 6.794929504394531e-05,
      "step": 71250
    },
    {
      "action_loss_avg": 0.002199327922426164,
      "epoch": 6.794929504394531e-05,
      "step": 71250
    },
    {
      "epoch": 6.794929504394531e-05,
      "step": 71250,
      "training_step_time": 7.623597621917725
    },
    {
      "epoch": 6.795024871826172e-05,
      "model_forward_time": 1.707092046737671,
      "step": 71251
    },
    {
      "epoch": 6.795024871826172e-05,
      "step": 71251,
      "training_step_time": 6.814537763595581
    },
    {
      "epoch": 6.795120239257812e-05,
      "model_forward_time": 1.7023985385894775,
      "step": 71252
    },
    {
      "epoch": 6.795120239257812e-05,
      "step": 71252,
      "training_step_time": 7.101294279098511
    },
    {
      "epoch": 6.795215606689453e-05,
      "model_forward_time": 1.7228925228118896,
      "step": 71253
    },
    {
      "epoch": 6.795215606689453e-05,
      "step": 71253,
      "training_step_time": 7.728519916534424
    },
    {
      "epoch": 6.795310974121094e-05,
      "model_forward_time": 2.2633309364318848,
      "step": 71254
    },
    {
      "epoch": 6.795310974121094e-05,
      "step": 71254,
      "training_step_time": 10.311057806015015
    },
    {
      "epoch": 6.795406341552734e-05,
      "model_forward_time": 1.7108423709869385,
      "step": 71255
    },
    {
      "epoch": 6.795406341552734e-05,
      "step": 71255,
      "training_step_time": 8.491394281387329
    },
    {
      "epoch": 6.795501708984375e-05,
      "model_forward_time": 1.708024501800537,
      "step": 71256
    },
    {
      "epoch": 6.795501708984375e-05,
      "step": 71256,
      "training_step_time": 8.65458631515503
    },
    {
      "epoch": 6.795597076416016e-05,
      "model_forward_time": 1.7114005088806152,
      "step": 71257
    },
    {
      "epoch": 6.795597076416016e-05,
      "step": 71257,
      "training_step_time": 7.653713703155518
    },
    {
      "epoch": 6.795692443847656e-05,
      "model_forward_time": 1.701202392578125,
      "step": 71258
    },
    {
      "epoch": 6.795692443847656e-05,
      "step": 71258,
      "training_step_time": 6.8083741664886475
    },
    {
      "epoch": 6.795787811279297e-05,
      "model_forward_time": 1.729135513305664,
      "step": 71259
    },
    {
      "epoch": 6.795787811279297e-05,
      "step": 71259,
      "training_step_time": 7.712818622589111
    },
    {
      "epoch": 6.795883178710938e-05,
      "grad_norm": 0.09782532602548599,
      "learning_rate": 2.093404944827059e-06,
      "loss": 0.0579,
      "step": 71260
    },
    {
      "epoch": 6.795883178710938e-05,
      "model_forward_time": 1.7055635452270508,
      "step": 71260
    },
    {
      "dynamics_loss_avg": 0.06061681564897299,
      "epoch": 6.795883178710938e-05,
      "step": 71260
    },
    {
      "action_loss_avg": 0.0018629697966389357,
      "epoch": 6.795883178710938e-05,
      "step": 71260
    },
    {
      "epoch": 6.795883178710938e-05,
      "step": 71260,
      "training_step_time": 6.869413375854492
    },
    {
      "epoch": 6.795978546142578e-05,
      "model_forward_time": 1.713423252105713,
      "step": 71261
    },
    {
      "epoch": 6.795978546142578e-05,
      "step": 71261,
      "training_step_time": 7.090237855911255
    },
    {
      "epoch": 6.796073913574219e-05,
      "model_forward_time": 1.7177186012268066,
      "step": 71262
    },
    {
      "epoch": 6.796073913574219e-05,
      "step": 71262,
      "training_step_time": 7.743071794509888
    },
    {
      "epoch": 6.79616928100586e-05,
      "model_forward_time": 2.254093885421753,
      "step": 71263
    },
    {
      "epoch": 6.79616928100586e-05,
      "step": 71263,
      "training_step_time": 11.374574184417725
    },
    {
      "epoch": 6.7962646484375e-05,
      "model_forward_time": 1.7032182216644287,
      "step": 71264
    },
    {
      "epoch": 6.7962646484375e-05,
      "step": 71264,
      "training_step_time": 9.289696216583252
    },
    {
      "epoch": 6.796360015869141e-05,
      "model_forward_time": 1.7026379108428955,
      "step": 71265
    },
    {
      "epoch": 6.796360015869141e-05,
      "step": 71265,
      "training_step_time": 7.910615921020508
    },
    {
      "epoch": 6.796455383300782e-05,
      "model_forward_time": 1.7039580345153809,
      "step": 71266
    },
    {
      "epoch": 6.796455383300782e-05,
      "step": 71266,
      "training_step_time": 7.805864572525024
    },
    {
      "epoch": 6.796550750732421e-05,
      "model_forward_time": 1.6993792057037354,
      "step": 71267
    },
    {
      "epoch": 6.796550750732421e-05,
      "step": 71267,
      "training_step_time": 6.867408990859985
    },
    {
      "epoch": 6.796646118164063e-05,
      "model_forward_time": 1.724961280822754,
      "step": 71268
    },
    {
      "epoch": 6.796646118164063e-05,
      "step": 71268,
      "training_step_time": 7.714327335357666
    },
    {
      "epoch": 6.796741485595704e-05,
      "model_forward_time": 1.324876070022583,
      "step": 71269
    },
    {
      "epoch": 6.796741485595704e-05,
      "step": 71269,
      "training_step_time": 6.856825351715088
    },
    {
      "epoch": 6.796836853027343e-05,
      "grad_norm": 0.09432897716760635,
      "learning_rate": 2.0920597155070394e-06,
      "loss": 0.0558,
      "step": 71270
    },
    {
      "epoch": 6.796836853027343e-05,
      "model_forward_time": 1.6987857818603516,
      "step": 71270
    },
    {
      "dynamics_loss_avg": 0.05353786554187536,
      "epoch": 6.796836853027343e-05,
      "step": 71270
    },
    {
      "action_loss_avg": 0.0009206924412865191,
      "epoch": 6.796836853027343e-05,
      "step": 71270
    },
    {
      "epoch": 6.796836853027343e-05,
      "step": 71270,
      "training_step_time": 7.117844104766846
    },
    {
      "epoch": 6.796932220458984e-05,
      "model_forward_time": 1.7041947841644287,
      "step": 71271
    },
    {
      "epoch": 6.796932220458984e-05,
      "step": 71271,
      "training_step_time": 9.661985397338867
    },
    {
      "epoch": 6.797027587890626e-05,
      "model_forward_time": 1.3246397972106934,
      "step": 71272
    },
    {
      "epoch": 6.797027587890626e-05,
      "step": 71272,
      "training_step_time": 9.746673822402954
    },
    {
      "epoch": 6.797122955322265e-05,
      "model_forward_time": 2.2784225940704346,
      "step": 71273
    },
    {
      "epoch": 6.797122955322265e-05,
      "step": 71273,
      "training_step_time": 7.886014938354492
    },
    {
      "epoch": 6.797218322753906e-05,
      "model_forward_time": 1.7116894721984863,
      "step": 71274
    },
    {
      "epoch": 6.797218322753906e-05,
      "step": 71274,
      "training_step_time": 8.10854721069336
    },
    {
      "epoch": 6.797313690185547e-05,
      "model_forward_time": 1.7071852684020996,
      "step": 71275
    },
    {
      "epoch": 6.797313690185547e-05,
      "step": 71275,
      "training_step_time": 7.711303234100342
    },
    {
      "epoch": 6.797409057617187e-05,
      "model_forward_time": 1.7138166427612305,
      "step": 71276
    },
    {
      "epoch": 6.797409057617187e-05,
      "step": 71276,
      "training_step_time": 6.933351993560791
    },
    {
      "epoch": 6.797504425048828e-05,
      "model_forward_time": 1.7166450023651123,
      "step": 71277
    },
    {
      "epoch": 6.797504425048828e-05,
      "step": 71277,
      "training_step_time": 7.79159688949585
    },
    {
      "epoch": 6.797599792480469e-05,
      "model_forward_time": 1.7075586318969727,
      "step": 71278
    },
    {
      "epoch": 6.797599792480469e-05,
      "step": 71278,
      "training_step_time": 6.893818140029907
    },
    {
      "epoch": 6.797695159912109e-05,
      "model_forward_time": 1.712632179260254,
      "step": 71279
    },
    {
      "epoch": 6.797695159912109e-05,
      "step": 71279,
      "training_step_time": 7.134896755218506
    },
    {
      "epoch": 6.79779052734375e-05,
      "grad_norm": 0.08817493170499802,
      "learning_rate": 2.0907148041949972e-06,
      "loss": 0.0556,
      "step": 71280
    },
    {
      "epoch": 6.79779052734375e-05,
      "model_forward_time": 1.7037203311920166,
      "step": 71280
    },
    {
      "dynamics_loss_avg": 0.0550929544493556,
      "epoch": 6.79779052734375e-05,
      "step": 71280
    },
    {
      "action_loss_avg": 0.0012805811362341045,
      "epoch": 6.79779052734375e-05,
      "step": 71280
    },
    {
      "epoch": 6.79779052734375e-05,
      "step": 71280,
      "training_step_time": 7.684065818786621
    },
    {
      "epoch": 6.797885894775391e-05,
      "model_forward_time": 1.7204246520996094,
      "step": 71281
    },
    {
      "epoch": 6.797885894775391e-05,
      "step": 71281,
      "training_step_time": 8.815235137939453
    },
    {
      "epoch": 6.797981262207031e-05,
      "model_forward_time": 2.2871272563934326,
      "step": 71282
    },
    {
      "epoch": 6.797981262207031e-05,
      "step": 71282,
      "training_step_time": 8.809258222579956
    },
    {
      "epoch": 6.798076629638672e-05,
      "model_forward_time": 1.7058911323547363,
      "step": 71283
    },
    {
      "epoch": 6.798076629638672e-05,
      "step": 71283,
      "training_step_time": 7.904739618301392
    },
    {
      "epoch": 6.798171997070313e-05,
      "model_forward_time": 1.7041411399841309,
      "step": 71284
    },
    {
      "epoch": 6.798171997070313e-05,
      "step": 71284,
      "training_step_time": 7.63521671295166
    },
    {
      "epoch": 6.798267364501953e-05,
      "model_forward_time": 1.7124323844909668,
      "step": 71285
    },
    {
      "epoch": 6.798267364501953e-05,
      "step": 71285,
      "training_step_time": 6.88568377494812
    },
    {
      "epoch": 6.798362731933594e-05,
      "model_forward_time": 1.7061817646026611,
      "step": 71286
    },
    {
      "epoch": 6.798362731933594e-05,
      "step": 71286,
      "training_step_time": 7.6854798793792725
    },
    {
      "epoch": 6.798458099365235e-05,
      "model_forward_time": 1.7147269248962402,
      "step": 71287
    },
    {
      "epoch": 6.798458099365235e-05,
      "step": 71287,
      "training_step_time": 6.860415935516357
    },
    {
      "epoch": 6.798553466796875e-05,
      "model_forward_time": 0.9948089122772217,
      "step": 71288
    },
    {
      "epoch": 6.798553466796875e-05,
      "step": 71288,
      "training_step_time": 7.127808570861816
    },
    {
      "epoch": 6.798648834228516e-05,
      "model_forward_time": 1.7126410007476807,
      "step": 71289
    },
    {
      "epoch": 6.798648834228516e-05,
      "step": 71289,
      "training_step_time": 7.680302858352661
    },
    {
      "epoch": 6.798744201660157e-05,
      "grad_norm": 0.1207968145608902,
      "learning_rate": 2.0893702110380113e-06,
      "loss": 0.0583,
      "step": 71290
    },
    {
      "epoch": 6.798744201660157e-05,
      "model_forward_time": 1.7254939079284668,
      "step": 71290
    },
    {
      "dynamics_loss_avg": 0.06616697013378144,
      "epoch": 6.798744201660157e-05,
      "step": 71290
    },
    {
      "action_loss_avg": 0.001531755510950461,
      "epoch": 6.798744201660157e-05,
      "step": 71290
    },
    {
      "epoch": 6.798744201660157e-05,
      "step": 71290,
      "training_step_time": 8.363504886627197
    },
    {
      "epoch": 6.798839569091797e-05,
      "model_forward_time": 2.2461445331573486,
      "step": 71291
    },
    {
      "epoch": 6.798839569091797e-05,
      "step": 71291,
      "training_step_time": 8.618757009506226
    },
    {
      "epoch": 6.798934936523438e-05,
      "model_forward_time": 1.7102057933807373,
      "step": 71292
    },
    {
      "epoch": 6.798934936523438e-05,
      "step": 71292,
      "training_step_time": 8.595086812973022
    },
    {
      "epoch": 6.799030303955079e-05,
      "model_forward_time": 1.7142035961151123,
      "step": 71293
    },
    {
      "epoch": 6.799030303955079e-05,
      "step": 71293,
      "training_step_time": 7.844698905944824
    },
    {
      "epoch": 6.799125671386718e-05,
      "model_forward_time": 1.7149384021759033,
      "step": 71294
    },
    {
      "epoch": 6.799125671386718e-05,
      "step": 71294,
      "training_step_time": 7.1689293384552
    },
    {
      "epoch": 6.79922103881836e-05,
      "model_forward_time": 1.7162976264953613,
      "step": 71295
    },
    {
      "epoch": 6.79922103881836e-05,
      "step": 71295,
      "training_step_time": 7.603252649307251
    },
    {
      "epoch": 6.79931640625e-05,
      "model_forward_time": 1.7142856121063232,
      "step": 71296
    },
    {
      "epoch": 6.79931640625e-05,
      "step": 71296,
      "training_step_time": 6.775217771530151
    },
    {
      "epoch": 6.79941177368164e-05,
      "model_forward_time": 1.712202548980713,
      "step": 71297
    },
    {
      "epoch": 6.79941177368164e-05,
      "step": 71297,
      "training_step_time": 7.157177925109863
    },
    {
      "epoch": 6.799507141113281e-05,
      "model_forward_time": 1.7151732444763184,
      "step": 71298
    },
    {
      "epoch": 6.799507141113281e-05,
      "step": 71298,
      "training_step_time": 9.073741436004639
    },
    {
      "epoch": 6.799602508544923e-05,
      "model_forward_time": 1.7201032638549805,
      "step": 71299
    },
    {
      "epoch": 6.799602508544923e-05,
      "step": 71299,
      "training_step_time": 7.705507516860962
    },
    {
      "epoch": 6.799697875976562e-05,
      "grad_norm": 0.11806497722864151,
      "learning_rate": 2.088025936183122e-06,
      "loss": 0.0566,
      "step": 71300
    },
    {
      "epoch": 6.799697875976562e-05,
      "model_forward_time": 1.694875717163086,
      "step": 71300
    },
    {
      "dynamics_loss_avg": 0.055622261948883533,
      "epoch": 6.799697875976562e-05,
      "step": 71300
    },
    {
      "action_loss_avg": 0.002169608970871195,
      "epoch": 6.799697875976562e-05,
      "step": 71300
    },
    {
      "epoch": 6.799697875976562e-05,
      "step": 71300,
      "training_step_time": 8.878156661987305
    },
    {
      "epoch": 6.799793243408203e-05,
      "model_forward_time": 2.2539126873016357,
      "step": 71301
    },
    {
      "epoch": 6.799793243408203e-05,
      "step": 71301,
      "training_step_time": 7.806257247924805
    },
    {
      "epoch": 6.799888610839844e-05,
      "model_forward_time": 1.7002882957458496,
      "step": 71302
    },
    {
      "epoch": 6.799888610839844e-05,
      "step": 71302,
      "training_step_time": 9.520021438598633
    },
    {
      "epoch": 6.799983978271484e-05,
      "model_forward_time": 1.700343370437622,
      "step": 71303
    },
    {
      "epoch": 6.799983978271484e-05,
      "step": 71303,
      "training_step_time": 7.900272846221924
    },
    {
      "epoch": 6.800079345703125e-05,
      "model_forward_time": 1.7032444477081299,
      "step": 71304
    },
    {
      "epoch": 6.800079345703125e-05,
      "step": 71304,
      "training_step_time": 7.8773932456970215
    },
    {
      "epoch": 6.800174713134766e-05,
      "model_forward_time": 1.7022030353546143,
      "step": 71305
    },
    {
      "epoch": 6.800174713134766e-05,
      "step": 71305,
      "training_step_time": 6.820428133010864
    },
    {
      "epoch": 6.800270080566406e-05,
      "model_forward_time": 1.7249364852905273,
      "step": 71306
    },
    {
      "epoch": 6.800270080566406e-05,
      "step": 71306,
      "training_step_time": 7.161696672439575
    },
    {
      "epoch": 6.800365447998047e-05,
      "model_forward_time": 1.7207286357879639,
      "step": 71307
    },
    {
      "epoch": 6.800365447998047e-05,
      "step": 71307,
      "training_step_time": 7.442630052566528
    },
    {
      "epoch": 6.800460815429688e-05,
      "model_forward_time": 1.7268829345703125,
      "step": 71308
    },
    {
      "epoch": 6.800460815429688e-05,
      "step": 71308,
      "training_step_time": 8.771892786026001
    },
    {
      "epoch": 6.800556182861328e-05,
      "model_forward_time": 1.7220265865325928,
      "step": 71309
    },
    {
      "epoch": 6.800556182861328e-05,
      "step": 71309,
      "training_step_time": 10.663126945495605
    },
    {
      "epoch": 6.800651550292969e-05,
      "grad_norm": 0.11208135634660721,
      "learning_rate": 2.0866819797773397e-06,
      "loss": 0.0595,
      "step": 71310
    },
    {
      "epoch": 6.800651550292969e-05,
      "model_forward_time": 2.269268274307251,
      "step": 71310
    },
    {
      "dynamics_loss_avg": 0.05708491001278162,
      "epoch": 6.800651550292969e-05,
      "step": 71310
    },
    {
      "action_loss_avg": 0.0012976495432667433,
      "epoch": 6.800651550292969e-05,
      "step": 71310
    },
    {
      "epoch": 6.800651550292969e-05,
      "step": 71310,
      "training_step_time": 9.633822202682495
    },
    {
      "epoch": 6.80074691772461e-05,
      "model_forward_time": 1.7319858074188232,
      "step": 71311
    },
    {
      "epoch": 6.80074691772461e-05,
      "step": 71311,
      "training_step_time": 7.749102354049683
    },
    {
      "epoch": 6.80084228515625e-05,
      "model_forward_time": 1.7174463272094727,
      "step": 71312
    },
    {
      "epoch": 6.80084228515625e-05,
      "step": 71312,
      "training_step_time": 8.880538702011108
    },
    {
      "epoch": 6.800937652587891e-05,
      "model_forward_time": 1.7204582691192627,
      "step": 71313
    },
    {
      "epoch": 6.800937652587891e-05,
      "step": 71313,
      "training_step_time": 6.768933057785034
    },
    {
      "epoch": 6.80103302001953e-05,
      "model_forward_time": 1.8025057315826416,
      "step": 71314
    },
    {
      "epoch": 6.80103302001953e-05,
      "step": 71314,
      "training_step_time": 7.7130043506622314
    },
    {
      "epoch": 6.801128387451172e-05,
      "model_forward_time": 1.7204508781433105,
      "step": 71315
    },
    {
      "epoch": 6.801128387451172e-05,
      "step": 71315,
      "training_step_time": 7.172297239303589
    },
    {
      "epoch": 6.801223754882813e-05,
      "model_forward_time": 1.7216637134552002,
      "step": 71316
    },
    {
      "epoch": 6.801223754882813e-05,
      "step": 71316,
      "training_step_time": 7.368108034133911
    },
    {
      "epoch": 6.801319122314452e-05,
      "model_forward_time": 1.7257094383239746,
      "step": 71317
    },
    {
      "epoch": 6.801319122314452e-05,
      "step": 71317,
      "training_step_time": 7.718776226043701
    },
    {
      "epoch": 6.801414489746094e-05,
      "model_forward_time": 1.7142517566680908,
      "step": 71318
    },
    {
      "epoch": 6.801414489746094e-05,
      "step": 71318,
      "training_step_time": 9.232812404632568
    },
    {
      "epoch": 6.801509857177735e-05,
      "model_forward_time": 2.2774298191070557,
      "step": 71319
    },
    {
      "epoch": 6.801509857177735e-05,
      "step": 71319,
      "training_step_time": 9.217647790908813
    },
    {
      "epoch": 6.801605224609374e-05,
      "grad_norm": 0.09441899508237839,
      "learning_rate": 2.085338341967634e-06,
      "loss": 0.0608,
      "step": 71320
    },
    {
      "epoch": 6.801605224609374e-05,
      "model_forward_time": 1.7255175113677979,
      "step": 71320
    },
    {
      "dynamics_loss_avg": 0.07189134284853935,
      "epoch": 6.801605224609374e-05,
      "step": 71320
    },
    {
      "action_loss_avg": 0.0026521931460592894,
      "epoch": 6.801605224609374e-05,
      "step": 71320
    },
    {
      "epoch": 6.801605224609374e-05,
      "step": 71320,
      "training_step_time": 9.179527282714844
    },
    {
      "epoch": 6.801700592041015e-05,
      "model_forward_time": 1.7333226203918457,
      "step": 71321
    },
    {
      "epoch": 6.801700592041015e-05,
      "step": 71321,
      "training_step_time": 8.64676570892334
    },
    {
      "epoch": 6.801795959472657e-05,
      "model_forward_time": 1.730635404586792,
      "step": 71322
    },
    {
      "epoch": 6.801795959472657e-05,
      "step": 71322,
      "training_step_time": 7.583102226257324
    },
    {
      "epoch": 6.801891326904296e-05,
      "model_forward_time": 1.7940587997436523,
      "step": 71323
    },
    {
      "epoch": 6.801891326904296e-05,
      "step": 71323,
      "training_step_time": 7.80920147895813
    },
    {
      "epoch": 6.801986694335937e-05,
      "model_forward_time": 1.7411987781524658,
      "step": 71324
    },
    {
      "epoch": 6.801986694335937e-05,
      "step": 71324,
      "training_step_time": 7.225052833557129
    },
    {
      "epoch": 6.802082061767578e-05,
      "model_forward_time": 1.7234878540039062,
      "step": 71325
    },
    {
      "epoch": 6.802082061767578e-05,
      "step": 71325,
      "training_step_time": 7.458219528198242
    },
    {
      "epoch": 6.802177429199218e-05,
      "model_forward_time": 1.7264420986175537,
      "step": 71326
    },
    {
      "epoch": 6.802177429199218e-05,
      "step": 71326,
      "training_step_time": 8.446121215820312
    },
    {
      "epoch": 6.802272796630859e-05,
      "model_forward_time": 1.728841781616211,
      "step": 71327
    },
    {
      "epoch": 6.802272796630859e-05,
      "step": 71327,
      "training_step_time": 7.889968156814575
    },
    {
      "epoch": 6.8023681640625e-05,
      "model_forward_time": 2.2922332286834717,
      "step": 71328
    },
    {
      "epoch": 6.8023681640625e-05,
      "step": 71328,
      "training_step_time": 7.8581109046936035
    },
    {
      "epoch": 6.80246353149414e-05,
      "model_forward_time": 1.7253293991088867,
      "step": 71329
    },
    {
      "epoch": 6.80246353149414e-05,
      "step": 71329,
      "training_step_time": 7.894651174545288
    },
    {
      "epoch": 6.802558898925781e-05,
      "grad_norm": 0.0966670885682106,
      "learning_rate": 2.083995022900946e-06,
      "loss": 0.058,
      "step": 71330
    },
    {
      "epoch": 6.802558898925781e-05,
      "model_forward_time": 1.7138466835021973,
      "step": 71330
    },
    {
      "dynamics_loss_avg": 0.04325644988566637,
      "epoch": 6.802558898925781e-05,
      "step": 71330
    },
    {
      "action_loss_avg": 0.0015619106241501867,
      "epoch": 6.802558898925781e-05,
      "step": 71330
    },
    {
      "epoch": 6.802558898925781e-05,
      "step": 71330,
      "training_step_time": 8.966064929962158
    },
    {
      "epoch": 6.802654266357422e-05,
      "model_forward_time": 1.727449655532837,
      "step": 71331
    },
    {
      "epoch": 6.802654266357422e-05,
      "step": 71331,
      "training_step_time": 7.5836567878723145
    },
    {
      "epoch": 6.802749633789062e-05,
      "model_forward_time": 1.385936975479126,
      "step": 71332
    },
    {
      "epoch": 6.802749633789062e-05,
      "step": 71332,
      "training_step_time": 7.654941082000732
    },
    {
      "epoch": 6.802845001220703e-05,
      "model_forward_time": 1.7352752685546875,
      "step": 71333
    },
    {
      "epoch": 6.802845001220703e-05,
      "step": 71333,
      "training_step_time": 6.855085849761963
    },
    {
      "epoch": 6.802940368652344e-05,
      "model_forward_time": 1.7293155193328857,
      "step": 71334
    },
    {
      "epoch": 6.802940368652344e-05,
      "step": 71334,
      "training_step_time": 7.410823106765747
    },
    {
      "epoch": 6.803035736083984e-05,
      "model_forward_time": 1.7361328601837158,
      "step": 71335
    },
    {
      "epoch": 6.803035736083984e-05,
      "step": 71335,
      "training_step_time": 7.691548109054565
    },
    {
      "epoch": 6.803131103515625e-05,
      "model_forward_time": 1.738755702972412,
      "step": 71336
    },
    {
      "epoch": 6.803131103515625e-05,
      "step": 71336,
      "training_step_time": 7.854571580886841
    },
    {
      "epoch": 6.803226470947266e-05,
      "model_forward_time": 1.726532220840454,
      "step": 71337
    },
    {
      "epoch": 6.803226470947266e-05,
      "step": 71337,
      "training_step_time": 8.386890888214111
    },
    {
      "epoch": 6.803321838378906e-05,
      "model_forward_time": 2.298076629638672,
      "step": 71338
    },
    {
      "epoch": 6.803321838378906e-05,
      "step": 71338,
      "training_step_time": 8.781617879867554
    },
    {
      "epoch": 6.803417205810547e-05,
      "model_forward_time": 1.7262511253356934,
      "step": 71339
    },
    {
      "epoch": 6.803417205810547e-05,
      "step": 71339,
      "training_step_time": 7.911096096038818
    },
    {
      "epoch": 6.803512573242188e-05,
      "grad_norm": 0.10195348411798477,
      "learning_rate": 2.082652022724177e-06,
      "loss": 0.0559,
      "step": 71340
    },
    {
      "epoch": 6.803512573242188e-05,
      "model_forward_time": 1.7228071689605713,
      "step": 71340
    },
    {
      "dynamics_loss_avg": 0.06960477456450462,
      "epoch": 6.803512573242188e-05,
      "step": 71340
    },
    {
      "action_loss_avg": 0.0017140828887932003,
      "epoch": 6.803512573242188e-05,
      "step": 71340
    },
    {
      "epoch": 6.803512573242188e-05,
      "step": 71340,
      "training_step_time": 7.670749664306641
    },
    {
      "epoch": 6.803607940673828e-05,
      "model_forward_time": 1.7646710872650146,
      "step": 71341
    },
    {
      "epoch": 6.803607940673828e-05,
      "step": 71341,
      "training_step_time": 7.86388373374939
    },
    {
      "epoch": 6.803703308105469e-05,
      "model_forward_time": 1.73167085647583,
      "step": 71342
    },
    {
      "epoch": 6.803703308105469e-05,
      "step": 71342,
      "training_step_time": 6.954958438873291
    },
    {
      "epoch": 6.80379867553711e-05,
      "model_forward_time": 1.7298557758331299,
      "step": 71343
    },
    {
      "epoch": 6.80379867553711e-05,
      "step": 71343,
      "training_step_time": 7.230039834976196
    },
    {
      "epoch": 6.80389404296875e-05,
      "model_forward_time": 1.7346704006195068,
      "step": 71344
    },
    {
      "epoch": 6.80389404296875e-05,
      "step": 71344,
      "training_step_time": 8.481250762939453
    },
    {
      "epoch": 6.80398941040039e-05,
      "model_forward_time": 1.7270333766937256,
      "step": 71345
    },
    {
      "epoch": 6.80398941040039e-05,
      "step": 71345,
      "training_step_time": 8.705477714538574
    },
    {
      "epoch": 6.804084777832032e-05,
      "model_forward_time": 1.7332608699798584,
      "step": 71346
    },
    {
      "epoch": 6.804084777832032e-05,
      "step": 71346,
      "training_step_time": 9.575343608856201
    },
    {
      "epoch": 6.804180145263671e-05,
      "model_forward_time": 2.2791030406951904,
      "step": 71347
    },
    {
      "epoch": 6.804180145263671e-05,
      "step": 71347,
      "training_step_time": 8.360365629196167
    },
    {
      "epoch": 6.804275512695312e-05,
      "model_forward_time": 1.7364487648010254,
      "step": 71348
    },
    {
      "epoch": 6.804275512695312e-05,
      "step": 71348,
      "training_step_time": 7.690036058425903
    },
    {
      "epoch": 6.804370880126953e-05,
      "model_forward_time": 1.3553645610809326,
      "step": 71349
    },
    {
      "epoch": 6.804370880126953e-05,
      "step": 71349,
      "training_step_time": 7.842725515365601
    },
    {
      "epoch": 6.804466247558593e-05,
      "grad_norm": 0.08860117197036743,
      "learning_rate": 2.081309341584199e-06,
      "loss": 0.0595,
      "step": 71350
    },
    {
      "epoch": 6.804466247558593e-05,
      "model_forward_time": 1.735764741897583,
      "step": 71350
    },
    {
      "dynamics_loss_avg": 0.06350357420742511,
      "epoch": 6.804466247558593e-05,
      "step": 71350
    },
    {
      "action_loss_avg": 0.001879778571310453,
      "epoch": 6.804466247558593e-05,
      "step": 71350
    },
    {
      "epoch": 6.804466247558593e-05,
      "step": 71350,
      "training_step_time": 7.613489151000977
    },
    {
      "epoch": 6.804561614990234e-05,
      "model_forward_time": 1.74009370803833,
      "step": 71351
    },
    {
      "epoch": 6.804561614990234e-05,
      "step": 71351,
      "training_step_time": 6.779368162155151
    },
    {
      "epoch": 6.804656982421875e-05,
      "model_forward_time": 1.732679843902588,
      "step": 71352
    },
    {
      "epoch": 6.804656982421875e-05,
      "step": 71352,
      "training_step_time": 7.046391487121582
    },
    {
      "epoch": 6.804752349853515e-05,
      "model_forward_time": 1.7288625240325928,
      "step": 71353
    },
    {
      "epoch": 6.804752349853515e-05,
      "step": 71353,
      "training_step_time": 7.565954208374023
    },
    {
      "epoch": 6.804847717285156e-05,
      "model_forward_time": 1.7260935306549072,
      "step": 71354
    },
    {
      "epoch": 6.804847717285156e-05,
      "step": 71354,
      "training_step_time": 9.532560348510742
    },
    {
      "epoch": 6.804943084716797e-05,
      "model_forward_time": 1.7258498668670654,
      "step": 71355
    },
    {
      "epoch": 6.804943084716797e-05,
      "step": 71355,
      "training_step_time": 8.448022365570068
    },
    {
      "epoch": 6.805038452148437e-05,
      "model_forward_time": 2.270965814590454,
      "step": 71356
    },
    {
      "epoch": 6.805038452148437e-05,
      "step": 71356,
      "training_step_time": 9.795287370681763
    },
    {
      "epoch": 6.805133819580078e-05,
      "model_forward_time": 1.7188239097595215,
      "step": 71357
    },
    {
      "epoch": 6.805133819580078e-05,
      "step": 71357,
      "training_step_time": 7.614730596542358
    },
    {
      "epoch": 6.805229187011719e-05,
      "model_forward_time": 1.7245666980743408,
      "step": 71358
    },
    {
      "epoch": 6.805229187011719e-05,
      "step": 71358,
      "training_step_time": 8.377968788146973
    },
    {
      "epoch": 6.805324554443359e-05,
      "model_forward_time": 1.7264065742492676,
      "step": 71359
    },
    {
      "epoch": 6.805324554443359e-05,
      "step": 71359,
      "training_step_time": 8.50304388999939
    },
    {
      "epoch": 6.805419921875e-05,
      "grad_norm": 0.15247562527656555,
      "learning_rate": 2.0799669796278417e-06,
      "loss": 0.0605,
      "step": 71360
    },
    {
      "epoch": 6.805419921875e-05,
      "model_forward_time": 1.7138254642486572,
      "step": 71360
    },
    {
      "dynamics_loss_avg": 0.058319355361163616,
      "epoch": 6.805419921875e-05,
      "step": 71360
    },
    {
      "action_loss_avg": 0.0016319106973242015,
      "epoch": 6.805419921875e-05,
      "step": 71360
    },
    {
      "epoch": 6.805419921875e-05,
      "step": 71360,
      "training_step_time": 6.826555252075195
    },
    {
      "epoch": 6.805515289306641e-05,
      "model_forward_time": 1.7189643383026123,
      "step": 71361
    },
    {
      "epoch": 6.805515289306641e-05,
      "step": 71361,
      "training_step_time": 7.272576808929443
    },
    {
      "epoch": 6.805610656738281e-05,
      "model_forward_time": 1.724693775177002,
      "step": 71362
    },
    {
      "epoch": 6.805610656738281e-05,
      "step": 71362,
      "training_step_time": 7.587306499481201
    },
    {
      "epoch": 6.805706024169922e-05,
      "model_forward_time": 1.7267632484436035,
      "step": 71363
    },
    {
      "epoch": 6.805706024169922e-05,
      "step": 71363,
      "training_step_time": 7.837938070297241
    },
    {
      "epoch": 6.805801391601563e-05,
      "model_forward_time": 1.7128446102142334,
      "step": 71364
    },
    {
      "epoch": 6.805801391601563e-05,
      "step": 71364,
      "training_step_time": 8.348822832107544
    },
    {
      "epoch": 6.805896759033203e-05,
      "model_forward_time": 2.266052007675171,
      "step": 71365
    },
    {
      "epoch": 6.805896759033203e-05,
      "step": 71365,
      "training_step_time": 8.920944690704346
    },
    {
      "epoch": 6.805992126464844e-05,
      "model_forward_time": 1.7143158912658691,
      "step": 71366
    },
    {
      "epoch": 6.805992126464844e-05,
      "step": 71366,
      "training_step_time": 7.733076333999634
    },
    {
      "epoch": 6.806087493896485e-05,
      "model_forward_time": 1.7193336486816406,
      "step": 71367
    },
    {
      "epoch": 6.806087493896485e-05,
      "step": 71367,
      "training_step_time": 7.8310418128967285
    },
    {
      "epoch": 6.806182861328125e-05,
      "model_forward_time": 1.7177391052246094,
      "step": 71368
    },
    {
      "epoch": 6.806182861328125e-05,
      "step": 71368,
      "training_step_time": 7.590915679931641
    },
    {
      "epoch": 6.806278228759766e-05,
      "model_forward_time": 1.7961747646331787,
      "step": 71369
    },
    {
      "epoch": 6.806278228759766e-05,
      "step": 71369,
      "training_step_time": 7.627837419509888
    },
    {
      "epoch": 6.806373596191407e-05,
      "grad_norm": 0.14417189359664917,
      "learning_rate": 2.0786249370019063e-06,
      "loss": 0.0594,
      "step": 71370
    },
    {
      "epoch": 6.806373596191407e-05,
      "model_forward_time": 1.7161455154418945,
      "step": 71370
    },
    {
      "dynamics_loss_avg": 0.0571920657530427,
      "epoch": 6.806373596191407e-05,
      "step": 71370
    },
    {
      "action_loss_avg": 0.002322317793732509,
      "epoch": 6.806373596191407e-05,
      "step": 71370
    },
    {
      "epoch": 6.806373596191407e-05,
      "step": 71370,
      "training_step_time": 7.219874382019043
    },
    {
      "epoch": 6.806468963623046e-05,
      "model_forward_time": 1.7221059799194336,
      "step": 71371
    },
    {
      "epoch": 6.806468963623046e-05,
      "step": 71371,
      "training_step_time": 7.529717445373535
    },
    {
      "epoch": 6.806564331054687e-05,
      "model_forward_time": 1.7209229469299316,
      "step": 71372
    },
    {
      "epoch": 6.806564331054687e-05,
      "step": 71372,
      "training_step_time": 7.766511917114258
    },
    {
      "epoch": 6.806659698486329e-05,
      "model_forward_time": 1.7229323387145996,
      "step": 71373
    },
    {
      "epoch": 6.806659698486329e-05,
      "step": 71373,
      "training_step_time": 8.766059160232544
    },
    {
      "epoch": 6.806755065917968e-05,
      "model_forward_time": 2.285261869430542,
      "step": 71374
    },
    {
      "epoch": 6.806755065917968e-05,
      "step": 71374,
      "training_step_time": 8.551960229873657
    },
    {
      "epoch": 6.80685043334961e-05,
      "model_forward_time": 1.3480727672576904,
      "step": 71375
    },
    {
      "epoch": 6.80685043334961e-05,
      "step": 71375,
      "training_step_time": 7.8211071491241455
    },
    {
      "epoch": 6.80694580078125e-05,
      "model_forward_time": 1.7167143821716309,
      "step": 71376
    },
    {
      "epoch": 6.80694580078125e-05,
      "step": 71376,
      "training_step_time": 7.942203521728516
    },
    {
      "epoch": 6.80704116821289e-05,
      "model_forward_time": 1.7220587730407715,
      "step": 71377
    },
    {
      "epoch": 6.80704116821289e-05,
      "step": 71377,
      "training_step_time": 7.777485609054565
    },
    {
      "epoch": 6.807136535644531e-05,
      "model_forward_time": 1.7964050769805908,
      "step": 71378
    },
    {
      "epoch": 6.807136535644531e-05,
      "step": 71378,
      "training_step_time": 7.733019590377808
    },
    {
      "epoch": 6.807231903076172e-05,
      "model_forward_time": 1.7177743911743164,
      "step": 71379
    },
    {
      "epoch": 6.807231903076172e-05,
      "step": 71379,
      "training_step_time": 8.098969459533691
    },
    {
      "epoch": 6.807327270507812e-05,
      "grad_norm": 0.10868963599205017,
      "learning_rate": 2.0772832138531544e-06,
      "loss": 0.0596,
      "step": 71380
    },
    {
      "epoch": 6.807327270507812e-05,
      "model_forward_time": 1.711963176727295,
      "step": 71380
    },
    {
      "dynamics_loss_avg": 0.05898546911776066,
      "epoch": 6.807327270507812e-05,
      "step": 71380
    },
    {
      "action_loss_avg": 0.0022353575215674935,
      "epoch": 6.807327270507812e-05,
      "step": 71380
    },
    {
      "epoch": 6.807327270507812e-05,
      "step": 71380,
      "training_step_time": 7.63713526725769
    },
    {
      "epoch": 6.807422637939453e-05,
      "model_forward_time": 1.7257351875305176,
      "step": 71381
    },
    {
      "epoch": 6.807422637939453e-05,
      "step": 71381,
      "training_step_time": 7.902630805969238
    },
    {
      "epoch": 6.807518005371094e-05,
      "model_forward_time": 1.3444080352783203,
      "step": 71382
    },
    {
      "epoch": 6.807518005371094e-05,
      "step": 71382,
      "training_step_time": 8.623847961425781
    },
    {
      "epoch": 6.807613372802734e-05,
      "model_forward_time": 1.7268707752227783,
      "step": 71383
    },
    {
      "epoch": 6.807613372802734e-05,
      "step": 71383,
      "training_step_time": 9.161258697509766
    },
    {
      "epoch": 6.807708740234375e-05,
      "model_forward_time": 2.2589237689971924,
      "step": 71384
    },
    {
      "epoch": 6.807708740234375e-05,
      "step": 71384,
      "training_step_time": 9.50561809539795
    },
    {
      "epoch": 6.807804107666016e-05,
      "model_forward_time": 1.7259483337402344,
      "step": 71385
    },
    {
      "epoch": 6.807804107666016e-05,
      "step": 71385,
      "training_step_time": 7.807976722717285
    },
    {
      "epoch": 6.807899475097656e-05,
      "model_forward_time": 1.7170360088348389,
      "step": 71386
    },
    {
      "epoch": 6.807899475097656e-05,
      "step": 71386,
      "training_step_time": 7.7671661376953125
    },
    {
      "epoch": 6.807994842529297e-05,
      "model_forward_time": 1.795121431350708,
      "step": 71387
    },
    {
      "epoch": 6.807994842529297e-05,
      "step": 71387,
      "training_step_time": 7.7506279945373535
    },
    {
      "epoch": 6.808090209960938e-05,
      "model_forward_time": 1.7276279926300049,
      "step": 71388
    },
    {
      "epoch": 6.808090209960938e-05,
      "step": 71388,
      "training_step_time": 7.0268638134002686
    },
    {
      "epoch": 6.808185577392578e-05,
      "model_forward_time": 1.72001051902771,
      "step": 71389
    },
    {
      "epoch": 6.808185577392578e-05,
      "step": 71389,
      "training_step_time": 7.613572835922241
    },
    {
      "epoch": 6.808280944824219e-05,
      "grad_norm": 0.08772064745426178,
      "learning_rate": 2.0759418103283156e-06,
      "loss": 0.0549,
      "step": 71390
    },
    {
      "epoch": 6.808280944824219e-05,
      "model_forward_time": 1.7118215560913086,
      "step": 71390
    },
    {
      "dynamics_loss_avg": 0.044535445235669616,
      "epoch": 6.808280944824219e-05,
      "step": 71390
    },
    {
      "action_loss_avg": 0.0016276112524792551,
      "epoch": 6.808280944824219e-05,
      "step": 71390
    },
    {
      "epoch": 6.808280944824219e-05,
      "step": 71390,
      "training_step_time": 7.624687910079956
    },
    {
      "epoch": 6.80837631225586e-05,
      "model_forward_time": 1.7270734310150146,
      "step": 71391
    },
    {
      "epoch": 6.80837631225586e-05,
      "step": 71391,
      "training_step_time": 9.774143934249878
    },
    {
      "epoch": 6.8084716796875e-05,
      "model_forward_time": 1.7162346839904785,
      "step": 71392
    },
    {
      "epoch": 6.8084716796875e-05,
      "step": 71392,
      "training_step_time": 7.719972133636475
    },
    {
      "epoch": 6.808567047119141e-05,
      "model_forward_time": 2.277571201324463,
      "step": 71393
    },
    {
      "epoch": 6.808567047119141e-05,
      "step": 71393,
      "training_step_time": 8.477692365646362
    },
    {
      "epoch": 6.808662414550782e-05,
      "model_forward_time": 1.7212378978729248,
      "step": 71394
    },
    {
      "epoch": 6.808662414550782e-05,
      "step": 71394,
      "training_step_time": 8.450200080871582
    },
    {
      "epoch": 6.808757781982421e-05,
      "model_forward_time": 1.7285866737365723,
      "step": 71395
    },
    {
      "epoch": 6.808757781982421e-05,
      "step": 71395,
      "training_step_time": 7.934034824371338
    },
    {
      "epoch": 6.808853149414063e-05,
      "model_forward_time": 1.7964301109313965,
      "step": 71396
    },
    {
      "epoch": 6.808853149414063e-05,
      "step": 71396,
      "training_step_time": 7.746252059936523
    },
    {
      "epoch": 6.808948516845704e-05,
      "model_forward_time": 1.7201709747314453,
      "step": 71397
    },
    {
      "epoch": 6.808948516845704e-05,
      "step": 71397,
      "training_step_time": 6.851404905319214
    },
    {
      "epoch": 6.809043884277343e-05,
      "model_forward_time": 1.717043399810791,
      "step": 71398
    },
    {
      "epoch": 6.809043884277343e-05,
      "step": 71398,
      "training_step_time": 7.367243766784668
    },
    {
      "epoch": 6.809139251708984e-05,
      "model_forward_time": 1.7365143299102783,
      "step": 71399
    },
    {
      "epoch": 6.809139251708984e-05,
      "step": 71399,
      "training_step_time": 7.977771520614624
    },
    {
      "epoch": 6.809234619140626e-05,
      "grad_norm": 0.10625002533197403,
      "learning_rate": 2.074600726574085e-06,
      "loss": 0.0588,
      "step": 71400
    },
    {
      "epoch": 6.809234619140626e-05,
      "model_forward_time": 1.7143583297729492,
      "step": 71400
    },
    {
      "dynamics_loss_avg": 0.06518620289862156,
      "epoch": 6.809234619140626e-05,
      "step": 71400
    },
    {
      "action_loss_avg": 0.0017717289563734084,
      "epoch": 6.809234619140626e-05,
      "step": 71400
    },
    {
      "epoch": 6.809234619140626e-05,
      "step": 71400,
      "training_step_time": 7.922047138214111
    },
    {
      "epoch": 6.809329986572265e-05,
      "model_forward_time": 1.7246825695037842,
      "step": 71401
    },
    {
      "epoch": 6.809329986572265e-05,
      "step": 71401,
      "training_step_time": 9.130008459091187
    },
    {
      "epoch": 6.809425354003906e-05,
      "model_forward_time": 1.8900880813598633,
      "step": 71402
    },
    {
      "epoch": 6.809425354003906e-05,
      "step": 71402,
      "training_step_time": 8.338984251022339
    },
    {
      "epoch": 6.809520721435547e-05,
      "model_forward_time": 1.7348036766052246,
      "step": 71403
    },
    {
      "epoch": 6.809520721435547e-05,
      "step": 71403,
      "training_step_time": 8.36517882347107
    },
    {
      "epoch": 6.809616088867187e-05,
      "model_forward_time": 1.7270655632019043,
      "step": 71404
    },
    {
      "epoch": 6.809616088867187e-05,
      "step": 71404,
      "training_step_time": 7.838332176208496
    },
    {
      "epoch": 6.809711456298828e-05,
      "model_forward_time": 1.8149125576019287,
      "step": 71405
    },
    {
      "epoch": 6.809711456298828e-05,
      "step": 71405,
      "training_step_time": 7.708014726638794
    },
    {
      "epoch": 6.809806823730469e-05,
      "model_forward_time": 1.7367053031921387,
      "step": 71406
    },
    {
      "epoch": 6.809806823730469e-05,
      "step": 71406,
      "training_step_time": 6.8467137813568115
    },
    {
      "epoch": 6.809902191162109e-05,
      "model_forward_time": 1.7295217514038086,
      "step": 71407
    },
    {
      "epoch": 6.809902191162109e-05,
      "step": 71407,
      "training_step_time": 7.688815116882324
    },
    {
      "epoch": 6.80999755859375e-05,
      "model_forward_time": 1.7249763011932373,
      "step": 71408
    },
    {
      "epoch": 6.80999755859375e-05,
      "step": 71408,
      "training_step_time": 7.395293474197388
    },
    {
      "epoch": 6.810092926025391e-05,
      "model_forward_time": 1.7547783851623535,
      "step": 71409
    },
    {
      "epoch": 6.810092926025391e-05,
      "step": 71409,
      "training_step_time": 8.65023684501648
    },
    {
      "epoch": 6.810188293457031e-05,
      "grad_norm": 0.09626907110214233,
      "learning_rate": 2.0732599627371215e-06,
      "loss": 0.0562,
      "step": 71410
    },
    {
      "epoch": 6.810188293457031e-05,
      "model_forward_time": 1.7219247817993164,
      "step": 71410
    },
    {
      "dynamics_loss_avg": 0.0816883748397231,
      "epoch": 6.810188293457031e-05,
      "step": 71410
    },
    {
      "action_loss_avg": 0.0020590948639437555,
      "epoch": 6.810188293457031e-05,
      "step": 71410
    },
    {
      "epoch": 6.810188293457031e-05,
      "step": 71410,
      "training_step_time": 8.542662143707275
    },
    {
      "epoch": 6.810283660888672e-05,
      "model_forward_time": 2.2652196884155273,
      "step": 71411
    },
    {
      "epoch": 6.810283660888672e-05,
      "step": 71411,
      "training_step_time": 8.807548761367798
    },
    {
      "epoch": 6.810379028320313e-05,
      "model_forward_time": 1.7288992404937744,
      "step": 71412
    },
    {
      "epoch": 6.810379028320313e-05,
      "step": 71412,
      "training_step_time": 8.740558624267578
    },
    {
      "epoch": 6.810474395751953e-05,
      "model_forward_time": 1.7246224880218506,
      "step": 71413
    },
    {
      "epoch": 6.810474395751953e-05,
      "step": 71413,
      "training_step_time": 7.938427448272705
    },
    {
      "epoch": 6.810569763183594e-05,
      "model_forward_time": 1.765479564666748,
      "step": 71414
    },
    {
      "epoch": 6.810569763183594e-05,
      "step": 71414,
      "training_step_time": 7.782459020614624
    },
    {
      "epoch": 6.810665130615235e-05,
      "model_forward_time": 1.7319145202636719,
      "step": 71415
    },
    {
      "epoch": 6.810665130615235e-05,
      "step": 71415,
      "training_step_time": 6.904017210006714
    },
    {
      "epoch": 6.810760498046875e-05,
      "model_forward_time": 1.725281000137329,
      "step": 71416
    },
    {
      "epoch": 6.810760498046875e-05,
      "step": 71416,
      "training_step_time": 7.652975559234619
    },
    {
      "epoch": 6.810855865478516e-05,
      "model_forward_time": 1.7354872226715088,
      "step": 71417
    },
    {
      "epoch": 6.810855865478516e-05,
      "step": 71417,
      "training_step_time": 7.430146217346191
    },
    {
      "epoch": 6.810951232910157e-05,
      "model_forward_time": 1.7214529514312744,
      "step": 71418
    },
    {
      "epoch": 6.810951232910157e-05,
      "step": 71418,
      "training_step_time": 8.39647912979126
    },
    {
      "epoch": 6.811046600341797e-05,
      "model_forward_time": 1.7225873470306396,
      "step": 71419
    },
    {
      "epoch": 6.811046600341797e-05,
      "step": 71419,
      "training_step_time": 7.588958978652954
    },
    {
      "epoch": 6.811141967773438e-05,
      "grad_norm": 0.08172473311424255,
      "learning_rate": 2.0719195189640466e-06,
      "loss": 0.0575,
      "step": 71420
    },
    {
      "epoch": 6.811141967773438e-05,
      "model_forward_time": 1.7291350364685059,
      "step": 71420
    },
    {
      "dynamics_loss_avg": 0.06221479419618845,
      "epoch": 6.811141967773438e-05,
      "step": 71420
    },
    {
      "action_loss_avg": 0.0016371838399209082,
      "epoch": 6.811141967773438e-05,
      "step": 71420
    },
    {
      "epoch": 6.811141967773438e-05,
      "step": 71420,
      "training_step_time": 8.670974969863892
    },
    {
      "epoch": 6.811237335205079e-05,
      "model_forward_time": 2.2844815254211426,
      "step": 71421
    },
    {
      "epoch": 6.811237335205079e-05,
      "step": 71421,
      "training_step_time": 7.687533855438232
    },
    {
      "epoch": 6.811332702636718e-05,
      "model_forward_time": 1.7329041957855225,
      "step": 71422
    },
    {
      "epoch": 6.811332702636718e-05,
      "step": 71422,
      "training_step_time": 8.541147470474243
    },
    {
      "epoch": 6.81142807006836e-05,
      "model_forward_time": 1.7464394569396973,
      "step": 71423
    },
    {
      "epoch": 6.81142807006836e-05,
      "step": 71423,
      "training_step_time": 7.628780126571655
    },
    {
      "epoch": 6.8115234375e-05,
      "model_forward_time": 1.7297253608703613,
      "step": 71424
    },
    {
      "epoch": 6.8115234375e-05,
      "step": 71424,
      "training_step_time": 7.172017812728882
    },
    {
      "epoch": 6.81161880493164e-05,
      "model_forward_time": 1.728316068649292,
      "step": 71425
    },
    {
      "epoch": 6.81161880493164e-05,
      "step": 71425,
      "training_step_time": 7.604672431945801
    },
    {
      "epoch": 6.811714172363281e-05,
      "model_forward_time": 1.730560302734375,
      "step": 71426
    },
    {
      "epoch": 6.811714172363281e-05,
      "step": 71426,
      "training_step_time": 7.3216872215271
    },
    {
      "epoch": 6.811809539794923e-05,
      "model_forward_time": 1.7316439151763916,
      "step": 71427
    },
    {
      "epoch": 6.811809539794923e-05,
      "step": 71427,
      "training_step_time": 7.771708965301514
    },
    {
      "epoch": 6.811904907226562e-05,
      "model_forward_time": 1.7293391227722168,
      "step": 71428
    },
    {
      "epoch": 6.811904907226562e-05,
      "step": 71428,
      "training_step_time": 7.623477935791016
    },
    {
      "epoch": 6.812000274658203e-05,
      "model_forward_time": 1.7368214130401611,
      "step": 71429
    },
    {
      "epoch": 6.812000274658203e-05,
      "step": 71429,
      "training_step_time": 8.591060400009155
    },
    {
      "epoch": 6.812095642089844e-05,
      "grad_norm": 0.07819544523954391,
      "learning_rate": 2.070579395401453e-06,
      "loss": 0.0566,
      "step": 71430
    },
    {
      "epoch": 6.812095642089844e-05,
      "model_forward_time": 2.2791523933410645,
      "step": 71430
    },
    {
      "dynamics_loss_avg": 0.0486290518194437,
      "epoch": 6.812095642089844e-05,
      "step": 71430
    },
    {
      "action_loss_avg": 0.0013914011098677293,
      "epoch": 6.812095642089844e-05,
      "step": 71430
    },
    {
      "epoch": 6.812095642089844e-05,
      "step": 71430,
      "training_step_time": 8.384138345718384
    },
    {
      "epoch": 6.812191009521484e-05,
      "model_forward_time": 1.7356016635894775,
      "step": 71431
    },
    {
      "epoch": 6.812191009521484e-05,
      "step": 71431,
      "training_step_time": 7.641253471374512
    },
    {
      "epoch": 6.812286376953125e-05,
      "model_forward_time": 1.726489543914795,
      "step": 71432
    },
    {
      "epoch": 6.812286376953125e-05,
      "step": 71432,
      "training_step_time": 7.726551294326782
    },
    {
      "epoch": 6.812381744384766e-05,
      "model_forward_time": 1.72932767868042,
      "step": 71433
    },
    {
      "epoch": 6.812381744384766e-05,
      "step": 71433,
      "training_step_time": 7.426395416259766
    },
    {
      "epoch": 6.812477111816406e-05,
      "model_forward_time": 1.7320809364318848,
      "step": 71434
    },
    {
      "epoch": 6.812477111816406e-05,
      "step": 71434,
      "training_step_time": 7.236170053482056
    },
    {
      "epoch": 6.812572479248047e-05,
      "model_forward_time": 1.7325246334075928,
      "step": 71435
    },
    {
      "epoch": 6.812572479248047e-05,
      "step": 71435,
      "training_step_time": 7.558347940444946
    },
    {
      "epoch": 6.812667846679688e-05,
      "model_forward_time": 1.730912208557129,
      "step": 71436
    },
    {
      "epoch": 6.812667846679688e-05,
      "step": 71436,
      "training_step_time": 9.012352705001831
    },
    {
      "epoch": 6.812763214111328e-05,
      "model_forward_time": 1.7330071926116943,
      "step": 71437
    },
    {
      "epoch": 6.812763214111328e-05,
      "step": 71437,
      "training_step_time": 10.426909685134888
    },
    {
      "epoch": 6.812858581542969e-05,
      "model_forward_time": 1.720578670501709,
      "step": 71438
    },
    {
      "epoch": 6.812858581542969e-05,
      "step": 71438,
      "training_step_time": 9.615048885345459
    },
    {
      "epoch": 6.812953948974609e-05,
      "model_forward_time": 2.301283121109009,
      "step": 71439
    },
    {
      "epoch": 6.812953948974609e-05,
      "step": 71439,
      "training_step_time": 7.710021018981934
    },
    {
      "epoch": 6.81304931640625e-05,
      "grad_norm": 0.08105098456144333,
      "learning_rate": 2.0692395921958904e-06,
      "loss": 0.0578,
      "step": 71440
    },
    {
      "epoch": 6.81304931640625e-05,
      "model_forward_time": 1.726619005203247,
      "step": 71440
    },
    {
      "dynamics_loss_avg": 0.0491170059889555,
      "epoch": 6.81304931640625e-05,
      "step": 71440
    },
    {
      "action_loss_avg": 0.0019696188857778905,
      "epoch": 6.81304931640625e-05,
      "step": 71440
    },
    {
      "epoch": 6.81304931640625e-05,
      "step": 71440,
      "training_step_time": 8.37617540359497
    },
    {
      "epoch": 6.813144683837891e-05,
      "model_forward_time": 1.738205909729004,
      "step": 71441
    },
    {
      "epoch": 6.813144683837891e-05,
      "step": 71441,
      "training_step_time": 7.727114915847778
    },
    {
      "epoch": 6.81324005126953e-05,
      "model_forward_time": 1.8090369701385498,
      "step": 71442
    },
    {
      "epoch": 6.81324005126953e-05,
      "step": 71442,
      "training_step_time": 7.638722896575928
    },
    {
      "epoch": 6.813335418701172e-05,
      "model_forward_time": 1.7308883666992188,
      "step": 71443
    },
    {
      "epoch": 6.813335418701172e-05,
      "step": 71443,
      "training_step_time": 7.278486967086792
    },
    {
      "epoch": 6.813430786132813e-05,
      "model_forward_time": 1.7334179878234863,
      "step": 71444
    },
    {
      "epoch": 6.813430786132813e-05,
      "step": 71444,
      "training_step_time": 7.541182994842529
    },
    {
      "epoch": 6.813526153564452e-05,
      "model_forward_time": 1.7255537509918213,
      "step": 71445
    },
    {
      "epoch": 6.813526153564452e-05,
      "step": 71445,
      "training_step_time": 9.348567962646484
    },
    {
      "epoch": 6.813621520996094e-05,
      "model_forward_time": 1.7377161979675293,
      "step": 71446
    },
    {
      "epoch": 6.813621520996094e-05,
      "step": 71446,
      "training_step_time": 9.362860441207886
    },
    {
      "epoch": 6.813716888427735e-05,
      "model_forward_time": 1.725318431854248,
      "step": 71447
    },
    {
      "epoch": 6.813716888427735e-05,
      "step": 71447,
      "training_step_time": 8.410255432128906
    },
    {
      "epoch": 6.813812255859374e-05,
      "model_forward_time": 2.300045967102051,
      "step": 71448
    },
    {
      "epoch": 6.813812255859374e-05,
      "step": 71448,
      "training_step_time": 8.526885986328125
    },
    {
      "epoch": 6.813907623291015e-05,
      "model_forward_time": 1.7333533763885498,
      "step": 71449
    },
    {
      "epoch": 6.813907623291015e-05,
      "step": 71449,
      "training_step_time": 8.406631708145142
    },
    {
      "epoch": 6.814002990722657e-05,
      "grad_norm": 0.14782658219337463,
      "learning_rate": 2.0679001094938795e-06,
      "loss": 0.058,
      "step": 71450
    },
    {
      "epoch": 6.814002990722657e-05,
      "model_forward_time": 1.7405126094818115,
      "step": 71450
    },
    {
      "dynamics_loss_avg": 0.05056676268577576,
      "epoch": 6.814002990722657e-05,
      "step": 71450
    },
    {
      "action_loss_avg": 0.0011210242082597688,
      "epoch": 6.814002990722657e-05,
      "step": 71450
    },
    {
      "epoch": 6.814002990722657e-05,
      "step": 71450,
      "training_step_time": 7.734011888504028
    },
    {
      "epoch": 6.814098358154296e-05,
      "model_forward_time": 1.7956583499908447,
      "step": 71451
    },
    {
      "epoch": 6.814098358154296e-05,
      "step": 71451,
      "training_step_time": 7.665711879730225
    },
    {
      "epoch": 6.814193725585937e-05,
      "model_forward_time": 1.7238538265228271,
      "step": 71452
    },
    {
      "epoch": 6.814193725585937e-05,
      "step": 71452,
      "training_step_time": 7.2877397537231445
    },
    {
      "epoch": 6.814289093017578e-05,
      "model_forward_time": 1.7389135360717773,
      "step": 71453
    },
    {
      "epoch": 6.814289093017578e-05,
      "step": 71453,
      "training_step_time": 7.582903623580933
    },
    {
      "epoch": 6.814384460449218e-05,
      "model_forward_time": 1.7340521812438965,
      "step": 71454
    },
    {
      "epoch": 6.814384460449218e-05,
      "step": 71454,
      "training_step_time": 7.729288578033447
    },
    {
      "epoch": 6.814479827880859e-05,
      "model_forward_time": 1.7316598892211914,
      "step": 71455
    },
    {
      "epoch": 6.814479827880859e-05,
      "step": 71455,
      "training_step_time": 7.839575290679932
    },
    {
      "epoch": 6.8145751953125e-05,
      "model_forward_time": 1.7367234230041504,
      "step": 71456
    },
    {
      "epoch": 6.8145751953125e-05,
      "step": 71456,
      "training_step_time": 8.511133670806885
    },
    {
      "epoch": 6.81467056274414e-05,
      "model_forward_time": 2.293351888656616,
      "step": 71457
    },
    {
      "epoch": 6.81467056274414e-05,
      "step": 71457,
      "training_step_time": 9.479430913925171
    },
    {
      "epoch": 6.814765930175781e-05,
      "model_forward_time": 1.7399990558624268,
      "step": 71458
    },
    {
      "epoch": 6.814765930175781e-05,
      "step": 71458,
      "training_step_time": 7.704012870788574
    },
    {
      "epoch": 6.814861297607422e-05,
      "model_forward_time": 1.7268545627593994,
      "step": 71459
    },
    {
      "epoch": 6.814861297607422e-05,
      "step": 71459,
      "training_step_time": 7.676307678222656
    },
    {
      "epoch": 6.814956665039062e-05,
      "grad_norm": 0.09382949024438858,
      "learning_rate": 2.066560947441904e-06,
      "loss": 0.0586,
      "step": 71460
    },
    {
      "epoch": 6.814956665039062e-05,
      "model_forward_time": 1.7702155113220215,
      "step": 71460
    },
    {
      "dynamics_loss_avg": 0.03667918499559164,
      "epoch": 6.814956665039062e-05,
      "step": 71460
    },
    {
      "action_loss_avg": 0.0010864487558137625,
      "epoch": 6.814956665039062e-05,
      "step": 71460
    },
    {
      "epoch": 6.814956665039062e-05,
      "step": 71460,
      "training_step_time": 7.641180753707886
    },
    {
      "epoch": 6.815052032470703e-05,
      "model_forward_time": 1.7395482063293457,
      "step": 71461
    },
    {
      "epoch": 6.815052032470703e-05,
      "step": 71461,
      "training_step_time": 7.317406415939331
    },
    {
      "epoch": 6.815147399902344e-05,
      "model_forward_time": 1.7312235832214355,
      "step": 71462
    },
    {
      "epoch": 6.815147399902344e-05,
      "step": 71462,
      "training_step_time": 7.643830299377441
    },
    {
      "epoch": 6.815242767333984e-05,
      "model_forward_time": 1.7305326461791992,
      "step": 71463
    },
    {
      "epoch": 6.815242767333984e-05,
      "step": 71463,
      "training_step_time": 7.659375429153442
    },
    {
      "epoch": 6.815338134765625e-05,
      "model_forward_time": 1.7290260791778564,
      "step": 71464
    },
    {
      "epoch": 6.815338134765625e-05,
      "step": 71464,
      "training_step_time": 7.86138391494751
    },
    {
      "epoch": 6.815433502197266e-05,
      "model_forward_time": 1.7250347137451172,
      "step": 71465
    },
    {
      "epoch": 6.815433502197266e-05,
      "step": 71465,
      "training_step_time": 7.665612459182739
    },
    {
      "epoch": 6.815528869628906e-05,
      "model_forward_time": 1.718496322631836,
      "step": 71466
    },
    {
      "epoch": 6.815528869628906e-05,
      "step": 71466,
      "training_step_time": 9.17597222328186
    },
    {
      "epoch": 6.815624237060547e-05,
      "model_forward_time": 1.7223267555236816,
      "step": 71467
    },
    {
      "epoch": 6.815624237060547e-05,
      "step": 71467,
      "training_step_time": 7.682218074798584
    },
    {
      "epoch": 6.815719604492188e-05,
      "model_forward_time": 1.7237157821655273,
      "step": 71468
    },
    {
      "epoch": 6.815719604492188e-05,
      "step": 71468,
      "training_step_time": 7.649177312850952
    },
    {
      "epoch": 6.815814971923828e-05,
      "model_forward_time": 1.7457361221313477,
      "step": 71469
    },
    {
      "epoch": 6.815814971923828e-05,
      "step": 71469,
      "training_step_time": 7.736035108566284
    },
    {
      "epoch": 6.815910339355469e-05,
      "grad_norm": 0.10121531039476395,
      "learning_rate": 2.065222106186415e-06,
      "loss": 0.0595,
      "step": 71470
    },
    {
      "epoch": 6.815910339355469e-05,
      "model_forward_time": 1.7204420566558838,
      "step": 71470
    },
    {
      "dynamics_loss_avg": 0.05263904761523008,
      "epoch": 6.815910339355469e-05,
      "step": 71470
    },
    {
      "action_loss_avg": 0.0010444710744195618,
      "epoch": 6.815910339355469e-05,
      "step": 71470
    },
    {
      "epoch": 6.815910339355469e-05,
      "step": 71470,
      "training_step_time": 7.238957166671753
    },
    {
      "epoch": 6.81600570678711e-05,
      "model_forward_time": 1.3475890159606934,
      "step": 71471
    },
    {
      "epoch": 6.81600570678711e-05,
      "step": 71471,
      "training_step_time": 7.332921028137207
    },
    {
      "epoch": 6.81610107421875e-05,
      "model_forward_time": 1.720250129699707,
      "step": 71472
    },
    {
      "epoch": 6.81610107421875e-05,
      "step": 71472,
      "training_step_time": 7.637333393096924
    },
    {
      "epoch": 6.81619644165039e-05,
      "model_forward_time": 1.7291641235351562,
      "step": 71473
    },
    {
      "epoch": 6.81619644165039e-05,
      "step": 71473,
      "training_step_time": 7.827085256576538
    },
    {
      "epoch": 6.816291809082032e-05,
      "model_forward_time": 1.7227797508239746,
      "step": 71474
    },
    {
      "epoch": 6.816291809082032e-05,
      "step": 71474,
      "training_step_time": 8.597920417785645
    },
    {
      "epoch": 6.816387176513671e-05,
      "model_forward_time": 1.720198392868042,
      "step": 71475
    },
    {
      "epoch": 6.816387176513671e-05,
      "step": 71475,
      "training_step_time": 9.943472146987915
    },
    {
      "epoch": 6.816482543945312e-05,
      "model_forward_time": 2.2800519466400146,
      "step": 71476
    },
    {
      "epoch": 6.816482543945312e-05,
      "step": 71476,
      "training_step_time": 7.712961673736572
    },
    {
      "epoch": 6.816577911376953e-05,
      "model_forward_time": 1.7209811210632324,
      "step": 71477
    },
    {
      "epoch": 6.816577911376953e-05,
      "step": 71477,
      "training_step_time": 8.682159423828125
    },
    {
      "epoch": 6.816673278808593e-05,
      "model_forward_time": 1.7218542098999023,
      "step": 71478
    },
    {
      "epoch": 6.816673278808593e-05,
      "step": 71478,
      "training_step_time": 7.617339372634888
    },
    {
      "epoch": 6.816768646240234e-05,
      "model_forward_time": 1.7202517986297607,
      "step": 71479
    },
    {
      "epoch": 6.816768646240234e-05,
      "step": 71479,
      "training_step_time": 7.329156160354614
    },
    {
      "epoch": 6.816864013671875e-05,
      "grad_norm": 0.1111459732055664,
      "learning_rate": 2.063883585873823e-06,
      "loss": 0.0575,
      "step": 71480
    },
    {
      "epoch": 6.816864013671875e-05,
      "model_forward_time": 1.7184503078460693,
      "step": 71480
    },
    {
      "dynamics_loss_avg": 0.050911753252148625,
      "epoch": 6.816864013671875e-05,
      "step": 71480
    },
    {
      "action_loss_avg": 0.0012828975566662848,
      "epoch": 6.816864013671875e-05,
      "step": 71480
    },
    {
      "epoch": 6.816864013671875e-05,
      "step": 71480,
      "training_step_time": 7.673484802246094
    },
    {
      "epoch": 6.816959381103515e-05,
      "model_forward_time": 1.7214014530181885,
      "step": 71481
    },
    {
      "epoch": 6.816959381103515e-05,
      "step": 71481,
      "training_step_time": 7.6352033615112305
    },
    {
      "epoch": 6.817054748535156e-05,
      "model_forward_time": 1.7252180576324463,
      "step": 71482
    },
    {
      "epoch": 6.817054748535156e-05,
      "step": 71482,
      "training_step_time": 8.73538327217102
    },
    {
      "epoch": 6.817150115966797e-05,
      "model_forward_time": 1.7290356159210205,
      "step": 71483
    },
    {
      "epoch": 6.817150115966797e-05,
      "step": 71483,
      "training_step_time": 8.506297588348389
    },
    {
      "epoch": 6.817245483398437e-05,
      "model_forward_time": 1.6974592208862305,
      "step": 71484
    },
    {
      "epoch": 6.817245483398437e-05,
      "step": 71484,
      "training_step_time": 7.845738410949707
    },
    {
      "epoch": 6.817340850830078e-05,
      "model_forward_time": 2.262826681137085,
      "step": 71485
    },
    {
      "epoch": 6.817340850830078e-05,
      "step": 71485,
      "training_step_time": 8.809002161026001
    },
    {
      "epoch": 6.817436218261719e-05,
      "model_forward_time": 1.7225749492645264,
      "step": 71486
    },
    {
      "epoch": 6.817436218261719e-05,
      "step": 71486,
      "training_step_time": 8.358678817749023
    },
    {
      "epoch": 6.817531585693359e-05,
      "model_forward_time": 1.7172863483428955,
      "step": 71487
    },
    {
      "epoch": 6.817531585693359e-05,
      "step": 71487,
      "training_step_time": 8.768040657043457
    },
    {
      "epoch": 6.817626953125e-05,
      "model_forward_time": 1.718677282333374,
      "step": 71488
    },
    {
      "epoch": 6.817626953125e-05,
      "step": 71488,
      "training_step_time": 7.685839414596558
    },
    {
      "epoch": 6.817722320556641e-05,
      "model_forward_time": 1.7171790599822998,
      "step": 71489
    },
    {
      "epoch": 6.817722320556641e-05,
      "step": 71489,
      "training_step_time": 7.599294185638428
    },
    {
      "epoch": 6.817817687988281e-05,
      "grad_norm": 0.1399434506893158,
      "learning_rate": 2.0625453866505067e-06,
      "loss": 0.0582,
      "step": 71490
    },
    {
      "epoch": 6.817817687988281e-05,
      "model_forward_time": 1.7202296257019043,
      "step": 71490
    },
    {
      "dynamics_loss_avg": 0.06142205903306604,
      "epoch": 6.817817687988281e-05,
      "step": 71490
    },
    {
      "action_loss_avg": 0.0018083514674799516,
      "epoch": 6.817817687988281e-05,
      "step": 71490
    },
    {
      "epoch": 6.817817687988281e-05,
      "step": 71490,
      "training_step_time": 8.067197799682617
    },
    {
      "epoch": 6.817913055419922e-05,
      "model_forward_time": 1.730741262435913,
      "step": 71491
    },
    {
      "epoch": 6.817913055419922e-05,
      "step": 71491,
      "training_step_time": 8.441806316375732
    },
    {
      "epoch": 6.818008422851563e-05,
      "model_forward_time": 1.7383852005004883,
      "step": 71492
    },
    {
      "epoch": 6.818008422851563e-05,
      "step": 71492,
      "training_step_time": 7.750114679336548
    },
    {
      "epoch": 6.818103790283203e-05,
      "model_forward_time": 1.7281367778778076,
      "step": 71493
    },
    {
      "epoch": 6.818103790283203e-05,
      "step": 71493,
      "training_step_time": 7.670631170272827
    },
    {
      "epoch": 6.818199157714844e-05,
      "model_forward_time": 2.2816667556762695,
      "step": 71494
    },
    {
      "epoch": 6.818199157714844e-05,
      "step": 71494,
      "training_step_time": 7.740010023117065
    },
    {
      "epoch": 6.818294525146485e-05,
      "model_forward_time": 1.724318027496338,
      "step": 71495
    },
    {
      "epoch": 6.818294525146485e-05,
      "step": 71495,
      "training_step_time": 8.267682313919067
    },
    {
      "epoch": 6.818389892578125e-05,
      "model_forward_time": 1.7397830486297607,
      "step": 71496
    },
    {
      "epoch": 6.818389892578125e-05,
      "step": 71496,
      "training_step_time": 7.744375467300415
    },
    {
      "epoch": 6.818485260009766e-05,
      "model_forward_time": 1.4358794689178467,
      "step": 71497
    },
    {
      "epoch": 6.818485260009766e-05,
      "step": 71497,
      "training_step_time": 7.67866587638855
    },
    {
      "epoch": 6.818580627441407e-05,
      "model_forward_time": 1.729954481124878,
      "step": 71498
    },
    {
      "epoch": 6.818580627441407e-05,
      "step": 71498,
      "training_step_time": 7.798454523086548
    },
    {
      "epoch": 6.818675994873046e-05,
      "model_forward_time": 1.7359681129455566,
      "step": 71499
    },
    {
      "epoch": 6.818675994873046e-05,
      "step": 71499,
      "training_step_time": 7.501415491104126
    },
    {
      "epoch": 6.818771362304688e-05,
      "grad_norm": 0.09755796939134598,
      "learning_rate": 2.0612075086628126e-06,
      "loss": 0.055,
      "step": 71500
    },
    {
      "epoch": 6.818771362304688e-05,
      "model_forward_time": 1.723834753036499,
      "step": 71500
    },
    {
      "dynamics_loss_avg": 0.06535313930362463,
      "epoch": 6.818771362304688e-05,
      "step": 71500
    },
    {
      "action_loss_avg": 0.0019400036195293068,
      "epoch": 6.818771362304688e-05,
      "step": 71500
    },
    {
      "epoch": 6.818771362304688e-05,
      "step": 71500,
      "training_step_time": 7.7247068881988525
    },
    {
      "epoch": 6.818866729736329e-05,
      "model_forward_time": 1.734243392944336,
      "step": 71501
    },
    {
      "epoch": 6.818866729736329e-05,
      "step": 71501,
      "training_step_time": 7.864950656890869
    },
    {
      "epoch": 6.818962097167968e-05,
      "model_forward_time": 1.723634958267212,
      "step": 71502
    },
    {
      "epoch": 6.818962097167968e-05,
      "step": 71502,
      "training_step_time": 7.6433165073394775
    },
    {
      "epoch": 6.81905746459961e-05,
      "model_forward_time": 2.2768328189849854,
      "step": 71503
    },
    {
      "epoch": 6.81905746459961e-05,
      "step": 71503,
      "training_step_time": 10.536692380905151
    },
    {
      "epoch": 6.81915283203125e-05,
      "model_forward_time": 1.020770788192749,
      "step": 71504
    },
    {
      "epoch": 6.81915283203125e-05,
      "step": 71504,
      "training_step_time": 7.806321620941162
    },
    {
      "epoch": 6.81924819946289e-05,
      "model_forward_time": 1.7319068908691406,
      "step": 71505
    },
    {
      "epoch": 6.81924819946289e-05,
      "step": 71505,
      "training_step_time": 8.348364353179932
    },
    {
      "epoch": 6.819343566894531e-05,
      "model_forward_time": 1.8236827850341797,
      "step": 71506
    },
    {
      "epoch": 6.819343566894531e-05,
      "step": 71506,
      "training_step_time": 7.714633226394653
    },
    {
      "epoch": 6.819438934326172e-05,
      "model_forward_time": 1.7412974834442139,
      "step": 71507
    },
    {
      "epoch": 6.819438934326172e-05,
      "step": 71507,
      "training_step_time": 7.6083245277404785
    },
    {
      "epoch": 6.819534301757812e-05,
      "model_forward_time": 1.7282376289367676,
      "step": 71508
    },
    {
      "epoch": 6.819534301757812e-05,
      "step": 71508,
      "training_step_time": 7.350977659225464
    },
    {
      "epoch": 6.819629669189453e-05,
      "model_forward_time": 1.7356510162353516,
      "step": 71509
    },
    {
      "epoch": 6.819629669189453e-05,
      "step": 71509,
      "training_step_time": 7.763913154602051
    },
    {
      "epoch": 6.819725036621094e-05,
      "grad_norm": 0.11573085188865662,
      "learning_rate": 2.0598699520570454e-06,
      "loss": 0.0571,
      "step": 71510
    },
    {
      "epoch": 6.819725036621094e-05,
      "model_forward_time": 1.7328870296478271,
      "step": 71510
    },
    {
      "dynamics_loss_avg": 0.07359936796128749,
      "epoch": 6.819725036621094e-05,
      "step": 71510
    },
    {
      "action_loss_avg": 0.0017676778719760478,
      "epoch": 6.819725036621094e-05,
      "step": 71510
    },
    {
      "epoch": 6.819725036621094e-05,
      "step": 71510,
      "training_step_time": 8.368297100067139
    },
    {
      "epoch": 6.819820404052734e-05,
      "model_forward_time": 1.7262341976165771,
      "step": 71511
    },
    {
      "epoch": 6.819820404052734e-05,
      "step": 71511,
      "training_step_time": 7.668820858001709
    },
    {
      "epoch": 6.819915771484375e-05,
      "model_forward_time": 1.72990083694458,
      "step": 71512
    },
    {
      "epoch": 6.819915771484375e-05,
      "step": 71512,
      "training_step_time": 7.769366264343262
    },
    {
      "epoch": 6.820011138916016e-05,
      "model_forward_time": 2.287616491317749,
      "step": 71513
    },
    {
      "epoch": 6.820011138916016e-05,
      "step": 71513,
      "training_step_time": 7.749908447265625
    },
    {
      "epoch": 6.820106506347656e-05,
      "model_forward_time": 1.7291483879089355,
      "step": 71514
    },
    {
      "epoch": 6.820106506347656e-05,
      "step": 71514,
      "training_step_time": 8.762682437896729
    },
    {
      "epoch": 6.820201873779297e-05,
      "model_forward_time": 1.8185982704162598,
      "step": 71515
    },
    {
      "epoch": 6.820201873779297e-05,
      "step": 71515,
      "training_step_time": 7.714954137802124
    },
    {
      "epoch": 6.820297241210938e-05,
      "model_forward_time": 1.7292206287384033,
      "step": 71516
    },
    {
      "epoch": 6.820297241210938e-05,
      "step": 71516,
      "training_step_time": 7.68310284614563
    },
    {
      "epoch": 6.820392608642578e-05,
      "model_forward_time": 1.725731611251831,
      "step": 71517
    },
    {
      "epoch": 6.820392608642578e-05,
      "step": 71517,
      "training_step_time": 7.397710800170898
    },
    {
      "epoch": 6.820487976074219e-05,
      "model_forward_time": 1.739811897277832,
      "step": 71518
    },
    {
      "epoch": 6.820487976074219e-05,
      "step": 71518,
      "training_step_time": 7.624495983123779
    },
    {
      "epoch": 6.82058334350586e-05,
      "model_forward_time": 1.7323760986328125,
      "step": 71519
    },
    {
      "epoch": 6.82058334350586e-05,
      "step": 71519,
      "training_step_time": 7.808389663696289
    },
    {
      "epoch": 6.8206787109375e-05,
      "grad_norm": 0.10561913251876831,
      "learning_rate": 2.0585327169794796e-06,
      "loss": 0.056,
      "step": 71520
    },
    {
      "epoch": 6.8206787109375e-05,
      "model_forward_time": 1.7259259223937988,
      "step": 71520
    },
    {
      "dynamics_loss_avg": 0.06506755501031876,
      "epoch": 6.8206787109375e-05,
      "step": 71520
    },
    {
      "action_loss_avg": 0.002161611267365515,
      "epoch": 6.8206787109375e-05,
      "step": 71520
    },
    {
      "epoch": 6.8206787109375e-05,
      "step": 71520,
      "training_step_time": 8.570849180221558
    },
    {
      "epoch": 6.820774078369141e-05,
      "model_forward_time": 1.7310528755187988,
      "step": 71521
    },
    {
      "epoch": 6.820774078369141e-05,
      "step": 71521,
      "training_step_time": 8.965174674987793
    },
    {
      "epoch": 6.820869445800782e-05,
      "model_forward_time": 2.3093960285186768,
      "step": 71522
    },
    {
      "epoch": 6.820869445800782e-05,
      "step": 71522,
      "training_step_time": 7.737216234207153
    },
    {
      "epoch": 6.820964813232422e-05,
      "model_forward_time": 1.7379701137542725,
      "step": 71523
    },
    {
      "epoch": 6.820964813232422e-05,
      "step": 71523,
      "training_step_time": 8.593129873275757
    },
    {
      "epoch": 6.821060180664063e-05,
      "model_forward_time": 1.815239429473877,
      "step": 71524
    },
    {
      "epoch": 6.821060180664063e-05,
      "step": 71524,
      "training_step_time": 7.65692400932312
    },
    {
      "epoch": 6.821155548095704e-05,
      "model_forward_time": 1.7308409214019775,
      "step": 71525
    },
    {
      "epoch": 6.821155548095704e-05,
      "step": 71525,
      "training_step_time": 7.756229639053345
    },
    {
      "epoch": 6.821250915527343e-05,
      "model_forward_time": 1.7303893566131592,
      "step": 71526
    },
    {
      "epoch": 6.821250915527343e-05,
      "step": 71526,
      "training_step_time": 7.322260618209839
    },
    {
      "epoch": 6.821346282958984e-05,
      "model_forward_time": 1.7305946350097656,
      "step": 71527
    },
    {
      "epoch": 6.821346282958984e-05,
      "step": 71527,
      "training_step_time": 8.523821592330933
    },
    {
      "epoch": 6.821441650390626e-05,
      "model_forward_time": 1.7365500926971436,
      "step": 71528
    },
    {
      "epoch": 6.821441650390626e-05,
      "step": 71528,
      "training_step_time": 8.866804599761963
    },
    {
      "epoch": 6.821537017822265e-05,
      "model_forward_time": 1.726505994796753,
      "step": 71529
    },
    {
      "epoch": 6.821537017822265e-05,
      "step": 71529,
      "training_step_time": 8.238953351974487
    },
    {
      "epoch": 6.821632385253906e-05,
      "grad_norm": 0.09916335344314575,
      "learning_rate": 2.0571958035763535e-06,
      "loss": 0.0543,
      "step": 71530
    },
    {
      "epoch": 6.821632385253906e-05,
      "model_forward_time": 1.720224380493164,
      "step": 71530
    },
    {
      "dynamics_loss_avg": 0.059648534283041954,
      "epoch": 6.821632385253906e-05,
      "step": 71530
    },
    {
      "action_loss_avg": 0.0019539355125743896,
      "epoch": 6.821632385253906e-05,
      "step": 71530
    },
    {
      "epoch": 6.821632385253906e-05,
      "step": 71530,
      "training_step_time": 9.860770463943481
    },
    {
      "epoch": 6.821727752685547e-05,
      "model_forward_time": 1.724198341369629,
      "step": 71531
    },
    {
      "epoch": 6.821727752685547e-05,
      "step": 71531,
      "training_step_time": 9.183519124984741
    },
    {
      "epoch": 6.821823120117187e-05,
      "model_forward_time": 1.7273659706115723,
      "step": 71532
    },
    {
      "epoch": 6.821823120117187e-05,
      "step": 71532,
      "training_step_time": 7.586601257324219
    },
    {
      "epoch": 6.821918487548828e-05,
      "model_forward_time": 1.4342129230499268,
      "step": 71533
    },
    {
      "epoch": 6.821918487548828e-05,
      "step": 71533,
      "training_step_time": 7.7000041007995605
    },
    {
      "epoch": 6.82201385498047e-05,
      "model_forward_time": 1.72749662399292,
      "step": 71534
    },
    {
      "epoch": 6.82201385498047e-05,
      "step": 71534,
      "training_step_time": 7.660745620727539
    },
    {
      "epoch": 6.822109222412109e-05,
      "model_forward_time": 1.7296104431152344,
      "step": 71535
    },
    {
      "epoch": 6.822109222412109e-05,
      "step": 71535,
      "training_step_time": 8.180678844451904
    },
    {
      "epoch": 6.82220458984375e-05,
      "model_forward_time": 1.726952314376831,
      "step": 71536
    },
    {
      "epoch": 6.82220458984375e-05,
      "step": 71536,
      "training_step_time": 7.6109373569488525
    },
    {
      "epoch": 6.822299957275391e-05,
      "model_forward_time": 1.7327759265899658,
      "step": 71537
    },
    {
      "epoch": 6.822299957275391e-05,
      "step": 71537,
      "training_step_time": 7.7234930992126465
    },
    {
      "epoch": 6.822395324707031e-05,
      "model_forward_time": 1.735304355621338,
      "step": 71538
    },
    {
      "epoch": 6.822395324707031e-05,
      "step": 71538,
      "training_step_time": 7.785958528518677
    },
    {
      "epoch": 6.822490692138672e-05,
      "model_forward_time": 1.723881483078003,
      "step": 71539
    },
    {
      "epoch": 6.822490692138672e-05,
      "step": 71539,
      "training_step_time": 9.533859729766846
    },
    {
      "epoch": 6.822586059570313e-05,
      "grad_norm": 0.06671737134456635,
      "learning_rate": 2.0558592119938712e-06,
      "loss": 0.0549,
      "step": 71540
    },
    {
      "epoch": 6.822586059570313e-05,
      "model_forward_time": 1.7295761108398438,
      "step": 71540
    },
    {
      "dynamics_loss_avg": 0.057958252355456354,
      "epoch": 6.822586059570313e-05,
      "step": 71540
    },
    {
      "action_loss_avg": 0.0014471414557192474,
      "epoch": 6.822586059570313e-05,
      "step": 71540
    },
    {
      "epoch": 6.822586059570313e-05,
      "step": 71540,
      "training_step_time": 8.732550144195557
    },
    {
      "epoch": 6.822681427001953e-05,
      "model_forward_time": 2.2828991413116455,
      "step": 71541
    },
    {
      "epoch": 6.822681427001953e-05,
      "step": 71541,
      "training_step_time": 7.758638381958008
    },
    {
      "epoch": 6.822776794433594e-05,
      "model_forward_time": 1.8243420124053955,
      "step": 71542
    },
    {
      "epoch": 6.822776794433594e-05,
      "step": 71542,
      "training_step_time": 7.728287220001221
    },
    {
      "epoch": 6.822872161865235e-05,
      "model_forward_time": 1.7344377040863037,
      "step": 71543
    },
    {
      "epoch": 6.822872161865235e-05,
      "step": 71543,
      "training_step_time": 8.662738561630249
    },
    {
      "epoch": 6.822967529296875e-05,
      "model_forward_time": 1.3550381660461426,
      "step": 71544
    },
    {
      "epoch": 6.822967529296875e-05,
      "step": 71544,
      "training_step_time": 7.5228111743927
    },
    {
      "epoch": 6.823062896728516e-05,
      "model_forward_time": 1.7386319637298584,
      "step": 71545
    },
    {
      "epoch": 6.823062896728516e-05,
      "step": 71545,
      "training_step_time": 7.59470534324646
    },
    {
      "epoch": 6.823158264160157e-05,
      "model_forward_time": 1.730947494506836,
      "step": 71546
    },
    {
      "epoch": 6.823158264160157e-05,
      "step": 71546,
      "training_step_time": 8.17379117012024
    },
    {
      "epoch": 6.823253631591797e-05,
      "model_forward_time": 1.734468936920166,
      "step": 71547
    },
    {
      "epoch": 6.823253631591797e-05,
      "step": 71547,
      "training_step_time": 7.726819038391113
    },
    {
      "epoch": 6.823348999023438e-05,
      "model_forward_time": 1.7391026020050049,
      "step": 71548
    },
    {
      "epoch": 6.823348999023438e-05,
      "step": 71548,
      "training_step_time": 8.437234878540039
    },
    {
      "epoch": 6.823444366455079e-05,
      "model_forward_time": 1.7299838066101074,
      "step": 71549
    },
    {
      "epoch": 6.823444366455079e-05,
      "step": 71549,
      "training_step_time": 10.141441106796265
    },
    {
      "epoch": 6.823539733886718e-05,
      "grad_norm": 0.09087736904621124,
      "learning_rate": 2.054522942378198e-06,
      "loss": 0.0522,
      "step": 71550
    },
    {
      "epoch": 6.823539733886718e-05,
      "model_forward_time": 2.2823474407196045,
      "step": 71550
    },
    {
      "dynamics_loss_avg": 0.0642110399901867,
      "epoch": 6.823539733886718e-05,
      "step": 71550
    },
    {
      "action_loss_avg": 0.0017070399248041213,
      "epoch": 6.823539733886718e-05,
      "step": 71550
    },
    {
      "epoch": 6.823539733886718e-05,
      "step": 71550,
      "training_step_time": 7.716227054595947
    },
    {
      "epoch": 6.82363510131836e-05,
      "model_forward_time": 1.8131382465362549,
      "step": 71551
    },
    {
      "epoch": 6.82363510131836e-05,
      "step": 71551,
      "training_step_time": 7.649888038635254
    },
    {
      "epoch": 6.82373046875e-05,
      "model_forward_time": 1.7254483699798584,
      "step": 71552
    },
    {
      "epoch": 6.82373046875e-05,
      "step": 71552,
      "training_step_time": 8.638433456420898
    },
    {
      "epoch": 6.82382583618164e-05,
      "model_forward_time": 1.3403592109680176,
      "step": 71553
    },
    {
      "epoch": 6.82382583618164e-05,
      "step": 71553,
      "training_step_time": 7.832190752029419
    },
    {
      "epoch": 6.823921203613281e-05,
      "model_forward_time": 1.7156152725219727,
      "step": 71554
    },
    {
      "epoch": 6.823921203613281e-05,
      "step": 71554,
      "training_step_time": 7.620504140853882
    },
    {
      "epoch": 6.824016571044923e-05,
      "model_forward_time": 1.7376775741577148,
      "step": 71555
    },
    {
      "epoch": 6.824016571044923e-05,
      "step": 71555,
      "training_step_time": 7.632303953170776
    },
    {
      "epoch": 6.824111938476562e-05,
      "model_forward_time": 1.7152042388916016,
      "step": 71556
    },
    {
      "epoch": 6.824111938476562e-05,
      "step": 71556,
      "training_step_time": 9.546064138412476
    },
    {
      "epoch": 6.824207305908203e-05,
      "model_forward_time": 1.72731614112854,
      "step": 71557
    },
    {
      "epoch": 6.824207305908203e-05,
      "step": 71557,
      "training_step_time": 8.287757158279419
    },
    {
      "epoch": 6.824302673339844e-05,
      "model_forward_time": 1.7169709205627441,
      "step": 71558
    },
    {
      "epoch": 6.824302673339844e-05,
      "step": 71558,
      "training_step_time": 9.432119607925415
    },
    {
      "epoch": 6.824398040771484e-05,
      "model_forward_time": 2.254733085632324,
      "step": 71559
    },
    {
      "epoch": 6.824398040771484e-05,
      "step": 71559,
      "training_step_time": 8.589549541473389
    },
    {
      "epoch": 6.824493408203125e-05,
      "grad_norm": 0.10813994705677032,
      "learning_rate": 2.053186994875469e-06,
      "loss": 0.0546,
      "step": 71560
    },
    {
      "epoch": 6.824493408203125e-05,
      "model_forward_time": 1.7920141220092773,
      "step": 71560
    },
    {
      "dynamics_loss_avg": 0.052147870790213345,
      "epoch": 6.824493408203125e-05,
      "step": 71560
    },
    {
      "action_loss_avg": 0.0014335377345560119,
      "epoch": 6.824493408203125e-05,
      "step": 71560
    },
    {
      "epoch": 6.824493408203125e-05,
      "step": 71560,
      "training_step_time": 7.7048563957214355
    },
    {
      "epoch": 6.824588775634766e-05,
      "model_forward_time": 1.7261919975280762,
      "step": 71561
    },
    {
      "epoch": 6.824588775634766e-05,
      "step": 71561,
      "training_step_time": 7.662464380264282
    },
    {
      "epoch": 6.824684143066406e-05,
      "model_forward_time": 1.732750415802002,
      "step": 71562
    },
    {
      "epoch": 6.824684143066406e-05,
      "step": 71562,
      "training_step_time": 7.774988174438477
    },
    {
      "epoch": 6.824779510498047e-05,
      "model_forward_time": 1.7226343154907227,
      "step": 71563
    },
    {
      "epoch": 6.824779510498047e-05,
      "step": 71563,
      "training_step_time": 7.6029582023620605
    },
    {
      "epoch": 6.824874877929687e-05,
      "model_forward_time": 1.7242329120635986,
      "step": 71564
    },
    {
      "epoch": 6.824874877929687e-05,
      "step": 71564,
      "training_step_time": 7.654999017715454
    },
    {
      "epoch": 6.824970245361328e-05,
      "model_forward_time": 1.7310631275177002,
      "step": 71565
    },
    {
      "epoch": 6.824970245361328e-05,
      "step": 71565,
      "training_step_time": 9.585534811019897
    },
    {
      "epoch": 6.825065612792969e-05,
      "model_forward_time": 1.7307264804840088,
      "step": 71566
    },
    {
      "epoch": 6.825065612792969e-05,
      "step": 71566,
      "training_step_time": 8.38871169090271
    },
    {
      "epoch": 6.825160980224609e-05,
      "model_forward_time": 1.7245185375213623,
      "step": 71567
    },
    {
      "epoch": 6.825160980224609e-05,
      "step": 71567,
      "training_step_time": 8.51061463356018
    },
    {
      "epoch": 6.82525634765625e-05,
      "model_forward_time": 1.7326202392578125,
      "step": 71568
    },
    {
      "epoch": 6.82525634765625e-05,
      "step": 71568,
      "training_step_time": 8.306251049041748
    },
    {
      "epoch": 6.825351715087891e-05,
      "model_forward_time": 1.6846013069152832,
      "step": 71569
    },
    {
      "epoch": 6.825351715087891e-05,
      "step": 71569,
      "training_step_time": 8.853453636169434
    },
    {
      "epoch": 6.82544708251953e-05,
      "grad_norm": 0.09365907311439514,
      "learning_rate": 2.0518513696317776e-06,
      "loss": 0.0577,
      "step": 71570
    },
    {
      "epoch": 6.82544708251953e-05,
      "model_forward_time": 1.710871696472168,
      "step": 71570
    },
    {
      "dynamics_loss_avg": 0.036316854506731035,
      "epoch": 6.82544708251953e-05,
      "step": 71570
    },
    {
      "action_loss_avg": 0.0013223931251559406,
      "epoch": 6.82544708251953e-05,
      "step": 71570
    },
    {
      "epoch": 6.82544708251953e-05,
      "step": 71570,
      "training_step_time": 7.680823802947998
    },
    {
      "epoch": 6.825542449951172e-05,
      "model_forward_time": 1.7231600284576416,
      "step": 71571
    },
    {
      "epoch": 6.825542449951172e-05,
      "step": 71571,
      "training_step_time": 8.471057415008545
    },
    {
      "epoch": 6.825637817382813e-05,
      "model_forward_time": 1.8302195072174072,
      "step": 71572
    },
    {
      "epoch": 6.825637817382813e-05,
      "step": 71572,
      "training_step_time": 7.40692925453186
    },
    {
      "epoch": 6.825733184814452e-05,
      "model_forward_time": 1.8555662631988525,
      "step": 71573
    },
    {
      "epoch": 6.825733184814452e-05,
      "step": 71573,
      "training_step_time": 7.723070859909058
    },
    {
      "epoch": 6.825828552246094e-05,
      "model_forward_time": 1.8550100326538086,
      "step": 71574
    },
    {
      "epoch": 6.825828552246094e-05,
      "step": 71574,
      "training_step_time": 8.54909634590149
    },
    {
      "epoch": 6.825923919677735e-05,
      "model_forward_time": 1.847001075744629,
      "step": 71575
    },
    {
      "epoch": 6.825923919677735e-05,
      "step": 71575,
      "training_step_time": 9.445464134216309
    },
    {
      "epoch": 6.826019287109374e-05,
      "model_forward_time": 1.4908311367034912,
      "step": 71576
    },
    {
      "epoch": 6.826019287109374e-05,
      "step": 71576,
      "training_step_time": 9.232495307922363
    },
    {
      "epoch": 6.826114654541015e-05,
      "model_forward_time": 1.8209149837493896,
      "step": 71577
    },
    {
      "epoch": 6.826114654541015e-05,
      "step": 71577,
      "training_step_time": 7.749889612197876
    },
    {
      "epoch": 6.826210021972657e-05,
      "model_forward_time": 2.415785312652588,
      "step": 71578
    },
    {
      "epoch": 6.826210021972657e-05,
      "step": 71578,
      "training_step_time": 7.760131359100342
    },
    {
      "epoch": 6.826305389404296e-05,
      "model_forward_time": 1.8544201850891113,
      "step": 71579
    },
    {
      "epoch": 6.826305389404296e-05,
      "step": 71579,
      "training_step_time": 7.68091082572937
    },
    {
      "epoch": 6.826400756835937e-05,
      "grad_norm": 0.0830843523144722,
      "learning_rate": 2.0505160667931885e-06,
      "loss": 0.0549,
      "step": 71580
    },
    {
      "epoch": 6.826400756835937e-05,
      "model_forward_time": 1.8387789726257324,
      "step": 71580
    },
    {
      "dynamics_loss_avg": 0.06383142042905092,
      "epoch": 6.826400756835937e-05,
      "step": 71580
    },
    {
      "action_loss_avg": 0.002200816315598786,
      "epoch": 6.826400756835937e-05,
      "step": 71580
    },
    {
      "epoch": 6.826400756835937e-05,
      "step": 71580,
      "training_step_time": 7.680212497711182
    },
    {
      "epoch": 6.826496124267578e-05,
      "model_forward_time": 1.8389308452606201,
      "step": 71581
    },
    {
      "epoch": 6.826496124267578e-05,
      "step": 71581,
      "training_step_time": 7.751739740371704
    },
    {
      "epoch": 6.826591491699218e-05,
      "model_forward_time": 1.4796252250671387,
      "step": 71582
    },
    {
      "epoch": 6.826591491699218e-05,
      "step": 71582,
      "training_step_time": 7.645134210586548
    },
    {
      "epoch": 6.826686859130859e-05,
      "model_forward_time": 1.8337640762329102,
      "step": 71583
    },
    {
      "epoch": 6.826686859130859e-05,
      "step": 71583,
      "training_step_time": 8.713549375534058
    },
    {
      "epoch": 6.8267822265625e-05,
      "model_forward_time": 1.8606326580047607,
      "step": 71584
    },
    {
      "epoch": 6.8267822265625e-05,
      "step": 71584,
      "training_step_time": 7.791695833206177
    },
    {
      "epoch": 6.82687759399414e-05,
      "model_forward_time": 1.8416938781738281,
      "step": 71585
    },
    {
      "epoch": 6.82687759399414e-05,
      "step": 71585,
      "training_step_time": 7.749430894851685
    },
    {
      "epoch": 6.826972961425781e-05,
      "model_forward_time": 1.8472001552581787,
      "step": 71586
    },
    {
      "epoch": 6.826972961425781e-05,
      "step": 71586,
      "training_step_time": 8.58625316619873
    },
    {
      "epoch": 6.827068328857422e-05,
      "model_forward_time": 1.8531477451324463,
      "step": 71587
    },
    {
      "epoch": 6.827068328857422e-05,
      "step": 71587,
      "training_step_time": 7.7088236808776855
    },
    {
      "epoch": 6.827163696289062e-05,
      "model_forward_time": 2.4018120765686035,
      "step": 71588
    },
    {
      "epoch": 6.827163696289062e-05,
      "step": 71588,
      "training_step_time": 8.713280439376831
    },
    {
      "epoch": 6.827259063720703e-05,
      "model_forward_time": 1.4931490421295166,
      "step": 71589
    },
    {
      "epoch": 6.827259063720703e-05,
      "step": 71589,
      "training_step_time": 8.400664329528809
    },
    {
      "epoch": 6.827354431152344e-05,
      "grad_norm": 0.09399024397134781,
      "learning_rate": 2.049181086505728e-06,
      "loss": 0.0535,
      "step": 71590
    },
    {
      "epoch": 6.827354431152344e-05,
      "model_forward_time": 1.8469138145446777,
      "step": 71590
    },
    {
      "dynamics_loss_avg": 0.0439534766599536,
      "epoch": 6.827354431152344e-05,
      "step": 71590
    },
    {
      "action_loss_avg": 0.0014122159860562532,
      "epoch": 6.827354431152344e-05,
      "step": 71590
    },
    {
      "epoch": 6.827354431152344e-05,
      "step": 71590,
      "training_step_time": 7.706287384033203
    },
    {
      "epoch": 6.827449798583984e-05,
      "model_forward_time": 1.846670150756836,
      "step": 71591
    },
    {
      "epoch": 6.827449798583984e-05,
      "step": 71591,
      "training_step_time": 7.604243040084839
    },
    {
      "epoch": 6.827545166015625e-05,
      "model_forward_time": 1.8468246459960938,
      "step": 71592
    },
    {
      "epoch": 6.827545166015625e-05,
      "step": 71592,
      "training_step_time": 7.672962427139282
    },
    {
      "epoch": 6.827640533447266e-05,
      "model_forward_time": 1.8070495128631592,
      "step": 71593
    },
    {
      "epoch": 6.827640533447266e-05,
      "step": 71593,
      "training_step_time": 7.799992322921753
    },
    {
      "epoch": 6.827735900878906e-05,
      "model_forward_time": 1.831383466720581,
      "step": 71594
    },
    {
      "epoch": 6.827735900878906e-05,
      "step": 71594,
      "training_step_time": 7.7206432819366455
    },
    {
      "epoch": 6.827831268310547e-05,
      "model_forward_time": 1.83235502243042,
      "step": 71595
    },
    {
      "epoch": 6.827831268310547e-05,
      "step": 71595,
      "training_step_time": 8.264835357666016
    },
    {
      "epoch": 6.827926635742188e-05,
      "model_forward_time": 1.8306365013122559,
      "step": 71596
    },
    {
      "epoch": 6.827926635742188e-05,
      "step": 71596,
      "training_step_time": 9.861315965652466
    },
    {
      "epoch": 6.828022003173828e-05,
      "model_forward_time": 2.3514788150787354,
      "step": 71597
    },
    {
      "epoch": 6.828022003173828e-05,
      "step": 71597,
      "training_step_time": 8.344414710998535
    },
    {
      "epoch": 6.828117370605469e-05,
      "model_forward_time": 1.8234350681304932,
      "step": 71598
    },
    {
      "epoch": 6.828117370605469e-05,
      "step": 71598,
      "training_step_time": 7.567432880401611
    },
    {
      "epoch": 6.82821273803711e-05,
      "model_forward_time": 1.8318812847137451,
      "step": 71599
    },
    {
      "epoch": 6.82821273803711e-05,
      "step": 71599,
      "training_step_time": 7.691264629364014
    },
    {
      "epoch": 6.82830810546875e-05,
      "grad_norm": 0.10116373002529144,
      "learning_rate": 2.047846428915388e-06,
      "loss": 0.0596,
      "step": 71600
    },
    {
      "epoch": 6.82830810546875e-05,
      "model_forward_time": 1.825547456741333,
      "step": 71600
    },
    {
      "dynamics_loss_avg": 0.07375750560313463,
      "epoch": 6.82830810546875e-05,
      "step": 71600
    },
    {
      "action_loss_avg": 0.0013210374978370964,
      "epoch": 6.82830810546875e-05,
      "step": 71600
    },
    {
      "epoch": 6.82830810546875e-05,
      "step": 71600,
      "training_step_time": 7.555854797363281
    },
    {
      "epoch": 6.82840347290039e-05,
      "model_forward_time": 1.8269896507263184,
      "step": 71601
    },
    {
      "epoch": 6.82840347290039e-05,
      "step": 71601,
      "training_step_time": 7.6466639041900635
    },
    {
      "epoch": 6.828498840332032e-05,
      "model_forward_time": 1.833249807357788,
      "step": 71602
    },
    {
      "epoch": 6.828498840332032e-05,
      "step": 71602,
      "training_step_time": 8.337010860443115
    },
    {
      "epoch": 6.828594207763671e-05,
      "model_forward_time": 1.832632303237915,
      "step": 71603
    },
    {
      "epoch": 6.828594207763671e-05,
      "step": 71603,
      "training_step_time": 9.948137044906616
    },
    {
      "epoch": 6.828689575195312e-05,
      "model_forward_time": 1.8411903381347656,
      "step": 71604
    },
    {
      "epoch": 6.828689575195312e-05,
      "step": 71604,
      "training_step_time": 7.660435914993286
    },
    {
      "epoch": 6.828784942626954e-05,
      "model_forward_time": 1.8353896141052246,
      "step": 71605
    },
    {
      "epoch": 6.828784942626954e-05,
      "step": 71605,
      "training_step_time": 7.722832918167114
    },
    {
      "epoch": 6.828880310058593e-05,
      "model_forward_time": 2.4069221019744873,
      "step": 71606
    },
    {
      "epoch": 6.828880310058593e-05,
      "step": 71606,
      "training_step_time": 7.645970106124878
    },
    {
      "epoch": 6.828975677490234e-05,
      "model_forward_time": 1.835758924484253,
      "step": 71607
    },
    {
      "epoch": 6.828975677490234e-05,
      "step": 71607,
      "training_step_time": 8.59842038154602
    },
    {
      "epoch": 6.829071044921875e-05,
      "model_forward_time": 1.8365726470947266,
      "step": 71608
    },
    {
      "epoch": 6.829071044921875e-05,
      "step": 71608,
      "training_step_time": 7.682015419006348
    },
    {
      "epoch": 6.829166412353515e-05,
      "model_forward_time": 1.827481746673584,
      "step": 71609
    },
    {
      "epoch": 6.829166412353515e-05,
      "step": 71609,
      "training_step_time": 8.262439250946045
    },
    {
      "epoch": 6.829261779785156e-05,
      "grad_norm": 0.10501865297555923,
      "learning_rate": 2.046512094168123e-06,
      "loss": 0.0574,
      "step": 71610
    },
    {
      "epoch": 6.829261779785156e-05,
      "model_forward_time": 1.8337256908416748,
      "step": 71610
    },
    {
      "dynamics_loss_avg": 0.05520248860120773,
      "epoch": 6.829261779785156e-05,
      "step": 71610
    },
    {
      "action_loss_avg": 0.0015076407202286647,
      "epoch": 6.829261779785156e-05,
      "step": 71610
    },
    {
      "epoch": 6.829261779785156e-05,
      "step": 71610,
      "training_step_time": 7.562576055526733
    },
    {
      "epoch": 6.829357147216797e-05,
      "model_forward_time": 1.8325934410095215,
      "step": 71611
    },
    {
      "epoch": 6.829357147216797e-05,
      "step": 71611,
      "training_step_time": 8.404668092727661
    },
    {
      "epoch": 6.829452514648437e-05,
      "model_forward_time": 1.8367843627929688,
      "step": 71612
    },
    {
      "epoch": 6.829452514648437e-05,
      "step": 71612,
      "training_step_time": 7.670712947845459
    },
    {
      "epoch": 6.829547882080078e-05,
      "model_forward_time": 1.8135757446289062,
      "step": 71613
    },
    {
      "epoch": 6.829547882080078e-05,
      "step": 71613,
      "training_step_time": 7.71701192855835
    },
    {
      "epoch": 6.829643249511719e-05,
      "model_forward_time": 1.832172155380249,
      "step": 71614
    },
    {
      "epoch": 6.829643249511719e-05,
      "step": 71614,
      "training_step_time": 8.82620096206665
    },
    {
      "epoch": 6.829738616943359e-05,
      "model_forward_time": 2.3694255352020264,
      "step": 71615
    },
    {
      "epoch": 6.829738616943359e-05,
      "step": 71615,
      "training_step_time": 7.656264781951904
    },
    {
      "epoch": 6.829833984375e-05,
      "model_forward_time": 1.8236455917358398,
      "step": 71616
    },
    {
      "epoch": 6.829833984375e-05,
      "step": 71616,
      "training_step_time": 7.562600135803223
    },
    {
      "epoch": 6.829929351806641e-05,
      "model_forward_time": 1.8060526847839355,
      "step": 71617
    },
    {
      "epoch": 6.829929351806641e-05,
      "step": 71617,
      "training_step_time": 7.72704553604126
    },
    {
      "epoch": 6.830024719238281e-05,
      "model_forward_time": 1.8192319869995117,
      "step": 71618
    },
    {
      "epoch": 6.830024719238281e-05,
      "step": 71618,
      "training_step_time": 7.74977445602417
    },
    {
      "epoch": 6.830120086669922e-05,
      "model_forward_time": 1.8190581798553467,
      "step": 71619
    },
    {
      "epoch": 6.830120086669922e-05,
      "step": 71619,
      "training_step_time": 7.596981525421143
    },
    {
      "epoch": 6.830215454101563e-05,
      "grad_norm": 0.08897823095321655,
      "learning_rate": 2.0451780824098566e-06,
      "loss": 0.0575,
      "step": 71620
    },
    {
      "epoch": 6.830215454101563e-05,
      "model_forward_time": 1.8218967914581299,
      "step": 71620
    },
    {
      "dynamics_loss_avg": 0.05057111214846373,
      "epoch": 6.830215454101563e-05,
      "step": 71620
    },
    {
      "action_loss_avg": 0.001378860606928356,
      "epoch": 6.830215454101563e-05,
      "step": 71620
    },
    {
      "epoch": 6.830215454101563e-05,
      "step": 71620,
      "training_step_time": 7.643457889556885
    },
    {
      "epoch": 6.830310821533203e-05,
      "model_forward_time": 1.8214855194091797,
      "step": 71621
    },
    {
      "epoch": 6.830310821533203e-05,
      "step": 71621,
      "training_step_time": 8.27555537223816
    },
    {
      "epoch": 6.830406188964844e-05,
      "model_forward_time": 1.7931535243988037,
      "step": 71622
    },
    {
      "epoch": 6.830406188964844e-05,
      "step": 71622,
      "training_step_time": 8.311502456665039
    },
    {
      "epoch": 6.830501556396485e-05,
      "model_forward_time": 1.8182318210601807,
      "step": 71623
    },
    {
      "epoch": 6.830501556396485e-05,
      "step": 71623,
      "training_step_time": 7.627645969390869
    },
    {
      "epoch": 6.830596923828125e-05,
      "model_forward_time": 1.793257236480713,
      "step": 71624
    },
    {
      "epoch": 6.830596923828125e-05,
      "step": 71624,
      "training_step_time": 8.914904832839966
    },
    {
      "epoch": 6.830692291259766e-05,
      "model_forward_time": 1.818314790725708,
      "step": 71625
    },
    {
      "epoch": 6.830692291259766e-05,
      "step": 71625,
      "training_step_time": 8.44845986366272
    },
    {
      "epoch": 6.830787658691407e-05,
      "model_forward_time": 1.8252513408660889,
      "step": 71626
    },
    {
      "epoch": 6.830787658691407e-05,
      "step": 71626,
      "training_step_time": 8.623788118362427
    },
    {
      "epoch": 6.830883026123046e-05,
      "model_forward_time": 1.7975544929504395,
      "step": 71627
    },
    {
      "epoch": 6.830883026123046e-05,
      "step": 71627,
      "training_step_time": 7.585253000259399
    },
    {
      "epoch": 6.830978393554688e-05,
      "model_forward_time": 1.8232433795928955,
      "step": 71628
    },
    {
      "epoch": 6.830978393554688e-05,
      "step": 71628,
      "training_step_time": 8.198513984680176
    },
    {
      "epoch": 6.831073760986329e-05,
      "model_forward_time": 1.8191618919372559,
      "step": 71629
    },
    {
      "epoch": 6.831073760986329e-05,
      "step": 71629,
      "training_step_time": 8.285999059677124
    },
    {
      "epoch": 6.831169128417968e-05,
      "grad_norm": 0.12134057283401489,
      "learning_rate": 2.043844393786471e-06,
      "loss": 0.0587,
      "step": 71630
    },
    {
      "epoch": 6.831169128417968e-05,
      "model_forward_time": 1.8316774368286133,
      "step": 71630
    },
    {
      "dynamics_loss_avg": 0.0806507270783186,
      "epoch": 6.831169128417968e-05,
      "step": 71630
    },
    {
      "action_loss_avg": 0.0015019450394902379,
      "epoch": 6.831169128417968e-05,
      "step": 71630
    },
    {
      "epoch": 6.831169128417968e-05,
      "step": 71630,
      "training_step_time": 7.634902238845825
    },
    {
      "epoch": 6.83126449584961e-05,
      "model_forward_time": 1.8224802017211914,
      "step": 71631
    },
    {
      "epoch": 6.83126449584961e-05,
      "step": 71631,
      "training_step_time": 9.289917945861816
    },
    {
      "epoch": 6.83135986328125e-05,
      "model_forward_time": 1.8032240867614746,
      "step": 71632
    },
    {
      "epoch": 6.83135986328125e-05,
      "step": 71632,
      "training_step_time": 8.645669937133789
    },
    {
      "epoch": 6.83145523071289e-05,
      "model_forward_time": 1.8327245712280273,
      "step": 71633
    },
    {
      "epoch": 6.83145523071289e-05,
      "step": 71633,
      "training_step_time": 8.728163957595825
    },
    {
      "epoch": 6.831550598144531e-05,
      "model_forward_time": 1.6717383861541748,
      "step": 71634
    },
    {
      "epoch": 6.831550598144531e-05,
      "step": 71634,
      "training_step_time": 7.571677923202515
    },
    {
      "epoch": 6.831645965576172e-05,
      "model_forward_time": 1.8291335105895996,
      "step": 71635
    },
    {
      "epoch": 6.831645965576172e-05,
      "step": 71635,
      "training_step_time": 8.733329772949219
    },
    {
      "epoch": 6.831741333007812e-05,
      "model_forward_time": 1.8293545246124268,
      "step": 71636
    },
    {
      "epoch": 6.831741333007812e-05,
      "step": 71636,
      "training_step_time": 7.69701623916626
    },
    {
      "epoch": 6.831836700439453e-05,
      "model_forward_time": 1.694765567779541,
      "step": 71637
    },
    {
      "epoch": 6.831836700439453e-05,
      "step": 71637,
      "training_step_time": 7.151435613632202
    },
    {
      "epoch": 6.831932067871094e-05,
      "model_forward_time": 1.712454080581665,
      "step": 71638
    },
    {
      "epoch": 6.831932067871094e-05,
      "step": 71638,
      "training_step_time": 7.65055513381958
    },
    {
      "epoch": 6.832027435302734e-05,
      "model_forward_time": 1.7122142314910889,
      "step": 71639
    },
    {
      "epoch": 6.832027435302734e-05,
      "step": 71639,
      "training_step_time": 7.74387264251709
    },
    {
      "epoch": 6.832122802734375e-05,
      "grad_norm": 0.08452925831079483,
      "learning_rate": 2.042511028443819e-06,
      "loss": 0.0586,
      "step": 71640
    },
    {
      "epoch": 6.832122802734375e-05,
      "model_forward_time": 1.7164273262023926,
      "step": 71640
    },
    {
      "dynamics_loss_avg": 0.05299391653388739,
      "epoch": 6.832122802734375e-05,
      "step": 71640
    },
    {
      "action_loss_avg": 0.0014158214093185962,
      "epoch": 6.832122802734375e-05,
      "step": 71640
    },
    {
      "epoch": 6.832122802734375e-05,
      "step": 71640,
      "training_step_time": 9.471025705337524
    },
    {
      "epoch": 6.832218170166016e-05,
      "model_forward_time": 1.7109403610229492,
      "step": 71641
    },
    {
      "epoch": 6.832218170166016e-05,
      "step": 71641,
      "training_step_time": 7.611232757568359
    },
    {
      "epoch": 6.832313537597656e-05,
      "model_forward_time": 1.7076256275177002,
      "step": 71642
    },
    {
      "epoch": 6.832313537597656e-05,
      "step": 71642,
      "training_step_time": 8.336340427398682
    },
    {
      "epoch": 6.832408905029297e-05,
      "model_forward_time": 2.2823691368103027,
      "step": 71643
    },
    {
      "epoch": 6.832408905029297e-05,
      "step": 71643,
      "training_step_time": 7.698010683059692
    },
    {
      "epoch": 6.832504272460938e-05,
      "model_forward_time": 1.7076079845428467,
      "step": 71644
    },
    {
      "epoch": 6.832504272460938e-05,
      "step": 71644,
      "training_step_time": 7.581631183624268
    },
    {
      "epoch": 6.832599639892578e-05,
      "model_forward_time": 1.7133560180664062,
      "step": 71645
    },
    {
      "epoch": 6.832599639892578e-05,
      "step": 71645,
      "training_step_time": 7.7164716720581055
    },
    {
      "epoch": 6.832695007324219e-05,
      "model_forward_time": 1.711578130722046,
      "step": 71646
    },
    {
      "epoch": 6.832695007324219e-05,
      "step": 71646,
      "training_step_time": 7.774823188781738
    },
    {
      "epoch": 6.83279037475586e-05,
      "model_forward_time": 1.7098588943481445,
      "step": 71647
    },
    {
      "epoch": 6.83279037475586e-05,
      "step": 71647,
      "training_step_time": 7.681498289108276
    },
    {
      "epoch": 6.8328857421875e-05,
      "model_forward_time": 1.7122623920440674,
      "step": 71648
    },
    {
      "epoch": 6.8328857421875e-05,
      "step": 71648,
      "training_step_time": 8.85399055480957
    },
    {
      "epoch": 6.832981109619141e-05,
      "model_forward_time": 1.7061419486999512,
      "step": 71649
    },
    {
      "epoch": 6.832981109619141e-05,
      "step": 71649,
      "training_step_time": 9.470066547393799
    },
    {
      "epoch": 6.833076477050782e-05,
      "grad_norm": 0.07461866736412048,
      "learning_rate": 2.041177986527712e-06,
      "loss": 0.0607,
      "step": 71650
    },
    {
      "epoch": 6.833076477050782e-05,
      "model_forward_time": 1.7219529151916504,
      "step": 71650
    },
    {
      "dynamics_loss_avg": 0.056497018039226535,
      "epoch": 6.833076477050782e-05,
      "step": 71650
    },
    {
      "action_loss_avg": 0.0014631218175054528,
      "epoch": 6.833076477050782e-05,
      "step": 71650
    },
    {
      "epoch": 6.833076477050782e-05,
      "step": 71650,
      "training_step_time": 8.60891056060791
    },
    {
      "epoch": 6.833171844482422e-05,
      "model_forward_time": 1.7293946743011475,
      "step": 71651
    },
    {
      "epoch": 6.833171844482422e-05,
      "step": 71651,
      "training_step_time": 8.587243556976318
    },
    {
      "epoch": 6.833267211914063e-05,
      "model_forward_time": 2.3234150409698486,
      "step": 71652
    },
    {
      "epoch": 6.833267211914063e-05,
      "step": 71652,
      "training_step_time": 7.708332300186157
    },
    {
      "epoch": 6.833362579345704e-05,
      "model_forward_time": 1.7298274040222168,
      "step": 71653
    },
    {
      "epoch": 6.833362579345704e-05,
      "step": 71653,
      "training_step_time": 7.591879367828369
    },
    {
      "epoch": 6.833457946777343e-05,
      "model_forward_time": 1.728400468826294,
      "step": 71654
    },
    {
      "epoch": 6.833457946777343e-05,
      "step": 71654,
      "training_step_time": 8.579635381698608
    },
    {
      "epoch": 6.833553314208984e-05,
      "model_forward_time": 1.7376289367675781,
      "step": 71655
    },
    {
      "epoch": 6.833553314208984e-05,
      "step": 71655,
      "training_step_time": 7.792081594467163
    },
    {
      "epoch": 6.833648681640626e-05,
      "model_forward_time": 1.7246918678283691,
      "step": 71656
    },
    {
      "epoch": 6.833648681640626e-05,
      "step": 71656,
      "training_step_time": 7.620563507080078
    },
    {
      "epoch": 6.833744049072265e-05,
      "model_forward_time": 1.7251334190368652,
      "step": 71657
    },
    {
      "epoch": 6.833744049072265e-05,
      "step": 71657,
      "training_step_time": 8.894720077514648
    },
    {
      "epoch": 6.833839416503906e-05,
      "model_forward_time": 1.7304883003234863,
      "step": 71658
    },
    {
      "epoch": 6.833839416503906e-05,
      "step": 71658,
      "training_step_time": 7.685834169387817
    },
    {
      "epoch": 6.833934783935547e-05,
      "model_forward_time": 1.769514560699463,
      "step": 71659
    },
    {
      "epoch": 6.833934783935547e-05,
      "step": 71659,
      "training_step_time": 7.7680604457855225
    },
    {
      "epoch": 6.834030151367187e-05,
      "grad_norm": 0.08241162449121475,
      "learning_rate": 2.0398452681839347e-06,
      "loss": 0.0548,
      "step": 71660
    },
    {
      "epoch": 6.834030151367187e-05,
      "model_forward_time": 1.716583013534546,
      "step": 71660
    },
    {
      "dynamics_loss_avg": 0.06374353002756834,
      "epoch": 6.834030151367187e-05,
      "step": 71660
    },
    {
      "action_loss_avg": 0.0014961567590944468,
      "epoch": 6.834030151367187e-05,
      "step": 71660
    },
    {
      "epoch": 6.834030151367187e-05,
      "step": 71660,
      "training_step_time": 7.742832899093628
    },
    {
      "epoch": 6.834125518798828e-05,
      "model_forward_time": 2.2725861072540283,
      "step": 71661
    },
    {
      "epoch": 6.834125518798828e-05,
      "step": 71661,
      "training_step_time": 8.505555868148804
    },
    {
      "epoch": 6.83422088623047e-05,
      "model_forward_time": 1.7258424758911133,
      "step": 71662
    },
    {
      "epoch": 6.83422088623047e-05,
      "step": 71662,
      "training_step_time": 8.306208610534668
    },
    {
      "epoch": 6.834316253662109e-05,
      "model_forward_time": 1.7329339981079102,
      "step": 71663
    },
    {
      "epoch": 6.834316253662109e-05,
      "step": 71663,
      "training_step_time": 7.745557069778442
    },
    {
      "epoch": 6.83441162109375e-05,
      "model_forward_time": 1.723921775817871,
      "step": 71664
    },
    {
      "epoch": 6.83441162109375e-05,
      "step": 71664,
      "training_step_time": 7.674837112426758
    },
    {
      "epoch": 6.834506988525391e-05,
      "model_forward_time": 1.723672866821289,
      "step": 71665
    },
    {
      "epoch": 6.834506988525391e-05,
      "step": 71665,
      "training_step_time": 7.560562372207642
    },
    {
      "epoch": 6.834602355957031e-05,
      "model_forward_time": 1.7172377109527588,
      "step": 71666
    },
    {
      "epoch": 6.834602355957031e-05,
      "step": 71666,
      "training_step_time": 7.6852734088897705
    },
    {
      "epoch": 6.834697723388672e-05,
      "model_forward_time": 1.7263224124908447,
      "step": 71667
    },
    {
      "epoch": 6.834697723388672e-05,
      "step": 71667,
      "training_step_time": 8.508257389068604
    },
    {
      "epoch": 6.834793090820313e-05,
      "model_forward_time": 1.741835117340088,
      "step": 71668
    },
    {
      "epoch": 6.834793090820313e-05,
      "step": 71668,
      "training_step_time": 10.019577741622925
    },
    {
      "epoch": 6.834888458251953e-05,
      "model_forward_time": 1.7065324783325195,
      "step": 71669
    },
    {
      "epoch": 6.834888458251953e-05,
      "step": 71669,
      "training_step_time": 7.621088981628418
    },
    {
      "epoch": 6.834983825683594e-05,
      "grad_norm": 0.08815573155879974,
      "learning_rate": 2.038512873558227e-06,
      "loss": 0.0563,
      "step": 71670
    },
    {
      "epoch": 6.834983825683594e-05,
      "model_forward_time": 1.70335054397583,
      "step": 71670
    },
    {
      "dynamics_loss_avg": 0.05113570485264063,
      "epoch": 6.834983825683594e-05,
      "step": 71670
    },
    {
      "action_loss_avg": 0.0016949948854744433,
      "epoch": 6.834983825683594e-05,
      "step": 71670
    },
    {
      "epoch": 6.834983825683594e-05,
      "step": 71670,
      "training_step_time": 8.634620666503906
    },
    {
      "epoch": 6.835079193115235e-05,
      "model_forward_time": 2.2784602642059326,
      "step": 71671
    },
    {
      "epoch": 6.835079193115235e-05,
      "step": 71671,
      "training_step_time": 7.662887811660767
    },
    {
      "epoch": 6.835174560546875e-05,
      "model_forward_time": 1.7252988815307617,
      "step": 71672
    },
    {
      "epoch": 6.835174560546875e-05,
      "step": 71672,
      "training_step_time": 7.643106460571289
    },
    {
      "epoch": 6.835269927978516e-05,
      "model_forward_time": 1.7360341548919678,
      "step": 71673
    },
    {
      "epoch": 6.835269927978516e-05,
      "step": 71673,
      "training_step_time": 7.711270570755005
    },
    {
      "epoch": 6.835365295410157e-05,
      "model_forward_time": 1.725623607635498,
      "step": 71674
    },
    {
      "epoch": 6.835365295410157e-05,
      "step": 71674,
      "training_step_time": 7.105191946029663
    },
    {
      "epoch": 6.835460662841797e-05,
      "model_forward_time": 1.7252273559570312,
      "step": 71675
    },
    {
      "epoch": 6.835460662841797e-05,
      "step": 71675,
      "training_step_time": 7.718044281005859
    },
    {
      "epoch": 6.835556030273438e-05,
      "model_forward_time": 1.721095323562622,
      "step": 71676
    },
    {
      "epoch": 6.835556030273438e-05,
      "step": 71676,
      "training_step_time": 7.600456953048706
    },
    {
      "epoch": 6.835651397705079e-05,
      "model_forward_time": 1.7469980716705322,
      "step": 71677
    },
    {
      "epoch": 6.835651397705079e-05,
      "step": 71677,
      "training_step_time": 8.497487306594849
    },
    {
      "epoch": 6.835746765136718e-05,
      "model_forward_time": 1.3446450233459473,
      "step": 71678
    },
    {
      "epoch": 6.835746765136718e-05,
      "step": 71678,
      "training_step_time": 8.639380931854248
    },
    {
      "epoch": 6.83584213256836e-05,
      "model_forward_time": 1.7390201091766357,
      "step": 71679
    },
    {
      "epoch": 6.83584213256836e-05,
      "step": 71679,
      "training_step_time": 8.245038509368896
    },
    {
      "epoch": 6.8359375e-05,
      "grad_norm": 0.13335493206977844,
      "learning_rate": 2.0371808027963e-06,
      "loss": 0.056,
      "step": 71680
    },
    {
      "epoch": 6.8359375e-05,
      "model_forward_time": 2.2799530029296875,
      "step": 71680
    },
    {
      "dynamics_loss_avg": 0.07129625678062439,
      "epoch": 6.8359375e-05,
      "step": 71680
    },
    {
      "action_loss_avg": 0.0015140792937017977,
      "epoch": 6.8359375e-05,
      "step": 71680
    },
    {
      "epoch": 6.8359375e-05,
      "step": 71680,
      "training_step_time": 8.334803104400635
    },
    {
      "epoch": 6.83603286743164e-05,
      "model_forward_time": 1.7339801788330078,
      "step": 71681
    },
    {
      "epoch": 6.83603286743164e-05,
      "step": 71681,
      "training_step_time": 7.530776023864746
    },
    {
      "epoch": 6.836128234863281e-05,
      "model_forward_time": 0.9774386882781982,
      "step": 71682
    },
    {
      "epoch": 6.836128234863281e-05,
      "step": 71682,
      "training_step_time": 7.598232746124268
    },
    {
      "epoch": 6.836223602294923e-05,
      "model_forward_time": 1.7051219940185547,
      "step": 71683
    },
    {
      "epoch": 6.836223602294923e-05,
      "step": 71683,
      "training_step_time": 7.564677715301514
    },
    {
      "epoch": 6.836318969726562e-05,
      "model_forward_time": 1.7096772193908691,
      "step": 71684
    },
    {
      "epoch": 6.836318969726562e-05,
      "step": 71684,
      "training_step_time": 8.67026400566101
    },
    {
      "epoch": 6.836414337158203e-05,
      "model_forward_time": 1.7122082710266113,
      "step": 71685
    },
    {
      "epoch": 6.836414337158203e-05,
      "step": 71685,
      "training_step_time": 8.208125114440918
    },
    {
      "epoch": 6.836509704589844e-05,
      "model_forward_time": 1.7160358428955078,
      "step": 71686
    },
    {
      "epoch": 6.836509704589844e-05,
      "step": 71686,
      "training_step_time": 7.57841944694519
    },
    {
      "epoch": 6.836605072021484e-05,
      "model_forward_time": 0.9970686435699463,
      "step": 71687
    },
    {
      "epoch": 6.836605072021484e-05,
      "step": 71687,
      "training_step_time": 7.643625974655151
    },
    {
      "epoch": 6.836700439453125e-05,
      "model_forward_time": 1.7141878604888916,
      "step": 71688
    },
    {
      "epoch": 6.836700439453125e-05,
      "step": 71688,
      "training_step_time": 8.612847089767456
    },
    {
      "epoch": 6.836795806884765e-05,
      "model_forward_time": 1.718597412109375,
      "step": 71689
    },
    {
      "epoch": 6.836795806884765e-05,
      "step": 71689,
      "training_step_time": 7.662367343902588
    },
    {
      "epoch": 6.836891174316406e-05,
      "grad_norm": 0.11576265841722488,
      "learning_rate": 2.0358490560438248e-06,
      "loss": 0.0578,
      "step": 71690
    },
    {
      "epoch": 6.836891174316406e-05,
      "model_forward_time": 2.2623848915100098,
      "step": 71690
    },
    {
      "dynamics_loss_avg": 0.07127772532403469,
      "epoch": 6.836891174316406e-05,
      "step": 71690
    },
    {
      "action_loss_avg": 0.0033693334029521794,
      "epoch": 6.836891174316406e-05,
      "step": 71690
    },
    {
      "epoch": 6.836891174316406e-05,
      "step": 71690,
      "training_step_time": 7.550597667694092
    },
    {
      "epoch": 6.836986541748047e-05,
      "model_forward_time": 1.7135491371154785,
      "step": 71691
    },
    {
      "epoch": 6.836986541748047e-05,
      "step": 71691,
      "training_step_time": 7.715135335922241
    },
    {
      "epoch": 6.837081909179687e-05,
      "model_forward_time": 1.7220654487609863,
      "step": 71692
    },
    {
      "epoch": 6.837081909179687e-05,
      "step": 71692,
      "training_step_time": 7.680413484573364
    },
    {
      "epoch": 6.837177276611328e-05,
      "model_forward_time": 1.3506906032562256,
      "step": 71693
    },
    {
      "epoch": 6.837177276611328e-05,
      "step": 71693,
      "training_step_time": 7.578135967254639
    },
    {
      "epoch": 6.837272644042969e-05,
      "model_forward_time": 1.719829797744751,
      "step": 71694
    },
    {
      "epoch": 6.837272644042969e-05,
      "step": 71694,
      "training_step_time": 7.664352893829346
    },
    {
      "epoch": 6.837368011474609e-05,
      "model_forward_time": 1.7225341796875,
      "step": 71695
    },
    {
      "epoch": 6.837368011474609e-05,
      "step": 71695,
      "training_step_time": 8.813841342926025
    },
    {
      "epoch": 6.83746337890625e-05,
      "model_forward_time": 1.7636916637420654,
      "step": 71696
    },
    {
      "epoch": 6.83746337890625e-05,
      "step": 71696,
      "training_step_time": 7.675188064575195
    },
    {
      "epoch": 6.837558746337891e-05,
      "model_forward_time": 1.7213239669799805,
      "step": 71697
    },
    {
      "epoch": 6.837558746337891e-05,
      "step": 71697,
      "training_step_time": 8.544476509094238
    },
    {
      "epoch": 6.83765411376953e-05,
      "model_forward_time": 1.7195916175842285,
      "step": 71698
    },
    {
      "epoch": 6.83765411376953e-05,
      "step": 71698,
      "training_step_time": 8.143463611602783
    },
    {
      "epoch": 6.837749481201172e-05,
      "model_forward_time": 2.2889959812164307,
      "step": 71699
    },
    {
      "epoch": 6.837749481201172e-05,
      "step": 71699,
      "training_step_time": 8.725550413131714
    },
    {
      "epoch": 6.837844848632813e-05,
      "grad_norm": 0.08587981015443802,
      "learning_rate": 2.0345176334464402e-06,
      "loss": 0.0504,
      "step": 71700
    },
    {
      "epoch": 6.837844848632813e-05,
      "model_forward_time": 1.713033676147461,
      "step": 71700
    },
    {
      "dynamics_loss_avg": 0.0523230142891407,
      "epoch": 6.837844848632813e-05,
      "step": 71700
    },
    {
      "action_loss_avg": 0.0014804239675868303,
      "epoch": 6.837844848632813e-05,
      "step": 71700
    },
    {
      "epoch": 6.837844848632813e-05,
      "step": 71700,
      "training_step_time": 8.687001466751099
    },
    {
      "epoch": 6.837940216064452e-05,
      "model_forward_time": 1.726928472518921,
      "step": 71701
    },
    {
      "epoch": 6.837940216064452e-05,
      "step": 71701,
      "training_step_time": 7.738379716873169
    },
    {
      "epoch": 6.838035583496094e-05,
      "model_forward_time": 1.7163288593292236,
      "step": 71702
    },
    {
      "epoch": 6.838035583496094e-05,
      "step": 71702,
      "training_step_time": 8.446906566619873
    },
    {
      "epoch": 6.838130950927735e-05,
      "model_forward_time": 1.7215473651885986,
      "step": 71703
    },
    {
      "epoch": 6.838130950927735e-05,
      "step": 71703,
      "training_step_time": 7.681565999984741
    },
    {
      "epoch": 6.838226318359374e-05,
      "model_forward_time": 1.7208051681518555,
      "step": 71704
    },
    {
      "epoch": 6.838226318359374e-05,
      "step": 71704,
      "training_step_time": 7.7058234214782715
    },
    {
      "epoch": 6.838321685791015e-05,
      "model_forward_time": 1.7677817344665527,
      "step": 71705
    },
    {
      "epoch": 6.838321685791015e-05,
      "step": 71705,
      "training_step_time": 9.375321388244629
    },
    {
      "epoch": 6.838417053222657e-05,
      "model_forward_time": 1.7176690101623535,
      "step": 71706
    },
    {
      "epoch": 6.838417053222657e-05,
      "step": 71706,
      "training_step_time": 8.7697172164917
    },
    {
      "epoch": 6.838512420654296e-05,
      "model_forward_time": 1.7172739505767822,
      "step": 71707
    },
    {
      "epoch": 6.838512420654296e-05,
      "step": 71707,
      "training_step_time": 7.64508581161499
    },
    {
      "epoch": 6.838607788085937e-05,
      "model_forward_time": 2.2985610961914062,
      "step": 71708
    },
    {
      "epoch": 6.838607788085937e-05,
      "step": 71708,
      "training_step_time": 7.744709730148315
    },
    {
      "epoch": 6.838703155517578e-05,
      "model_forward_time": 1.7222578525543213,
      "step": 71709
    },
    {
      "epoch": 6.838703155517578e-05,
      "step": 71709,
      "training_step_time": 7.688301086425781
    },
    {
      "epoch": 6.838798522949218e-05,
      "grad_norm": 0.07984712719917297,
      "learning_rate": 2.033186535149748e-06,
      "loss": 0.0553,
      "step": 71710
    },
    {
      "epoch": 6.838798522949218e-05,
      "model_forward_time": 1.7160768508911133,
      "step": 71710
    },
    {
      "dynamics_loss_avg": 0.0543364280834794,
      "epoch": 6.838798522949218e-05,
      "step": 71710
    },
    {
      "action_loss_avg": 0.0015426515368744732,
      "epoch": 6.838798522949218e-05,
      "step": 71710
    },
    {
      "epoch": 6.838798522949218e-05,
      "step": 71710,
      "training_step_time": 7.714706182479858
    },
    {
      "epoch": 6.838893890380859e-05,
      "model_forward_time": 1.718665361404419,
      "step": 71711
    },
    {
      "epoch": 6.838893890380859e-05,
      "step": 71711,
      "training_step_time": 7.591299057006836
    },
    {
      "epoch": 6.8389892578125e-05,
      "model_forward_time": 1.7202119827270508,
      "step": 71712
    },
    {
      "epoch": 6.8389892578125e-05,
      "step": 71712,
      "training_step_time": 7.804119348526001
    },
    {
      "epoch": 6.83908462524414e-05,
      "model_forward_time": 1.7189104557037354,
      "step": 71713
    },
    {
      "epoch": 6.83908462524414e-05,
      "step": 71713,
      "training_step_time": 9.110845804214478
    },
    {
      "epoch": 6.839179992675781e-05,
      "model_forward_time": 1.7587645053863525,
      "step": 71714
    },
    {
      "epoch": 6.839179992675781e-05,
      "step": 71714,
      "training_step_time": 9.974902629852295
    },
    {
      "epoch": 6.839275360107422e-05,
      "model_forward_time": 1.7281982898712158,
      "step": 71715
    },
    {
      "epoch": 6.839275360107422e-05,
      "step": 71715,
      "training_step_time": 8.317216157913208
    },
    {
      "epoch": 6.839370727539062e-05,
      "model_forward_time": 1.7168958187103271,
      "step": 71716
    },
    {
      "epoch": 6.839370727539062e-05,
      "step": 71716,
      "training_step_time": 8.062451362609863
    },
    {
      "epoch": 6.839466094970703e-05,
      "model_forward_time": 2.299414873123169,
      "step": 71717
    },
    {
      "epoch": 6.839466094970703e-05,
      "step": 71717,
      "training_step_time": 7.6364569664001465
    },
    {
      "epoch": 6.839561462402344e-05,
      "model_forward_time": 1.7190005779266357,
      "step": 71718
    },
    {
      "epoch": 6.839561462402344e-05,
      "step": 71718,
      "training_step_time": 7.58423376083374
    },
    {
      "epoch": 6.839656829833984e-05,
      "model_forward_time": 1.7243306636810303,
      "step": 71719
    },
    {
      "epoch": 6.839656829833984e-05,
      "step": 71719,
      "training_step_time": 9.628028392791748
    },
    {
      "epoch": 6.839752197265625e-05,
      "grad_norm": 0.0830964520573616,
      "learning_rate": 2.0318557612993185e-06,
      "loss": 0.0586,
      "step": 71720
    },
    {
      "epoch": 6.839752197265625e-05,
      "model_forward_time": 1.7181990146636963,
      "step": 71720
    },
    {
      "dynamics_loss_avg": 0.07700811848044395,
      "epoch": 6.839752197265625e-05,
      "step": 71720
    },
    {
      "action_loss_avg": 0.0019167120335623622,
      "epoch": 6.839752197265625e-05,
      "step": 71720
    },
    {
      "epoch": 6.839752197265625e-05,
      "step": 71720,
      "training_step_time": 7.666382551193237
    },
    {
      "epoch": 6.839847564697266e-05,
      "model_forward_time": 1.726449966430664,
      "step": 71721
    },
    {
      "epoch": 6.839847564697266e-05,
      "step": 71721,
      "training_step_time": 8.664950370788574
    },
    {
      "epoch": 6.839942932128906e-05,
      "model_forward_time": 1.7289535999298096,
      "step": 71722
    },
    {
      "epoch": 6.839942932128906e-05,
      "step": 71722,
      "training_step_time": 7.662166118621826
    },
    {
      "epoch": 6.840038299560547e-05,
      "model_forward_time": 1.7618072032928467,
      "step": 71723
    },
    {
      "epoch": 6.840038299560547e-05,
      "step": 71723,
      "training_step_time": 7.633230924606323
    },
    {
      "epoch": 6.840133666992188e-05,
      "model_forward_time": 1.723069190979004,
      "step": 71724
    },
    {
      "epoch": 6.840133666992188e-05,
      "step": 71724,
      "training_step_time": 8.79328465461731
    },
    {
      "epoch": 6.840229034423828e-05,
      "model_forward_time": 1.7261152267456055,
      "step": 71725
    },
    {
      "epoch": 6.840229034423828e-05,
      "step": 71725,
      "training_step_time": 8.168798208236694
    },
    {
      "epoch": 6.840324401855469e-05,
      "model_forward_time": 1.721376657485962,
      "step": 71726
    },
    {
      "epoch": 6.840324401855469e-05,
      "step": 71726,
      "training_step_time": 8.214875936508179
    },
    {
      "epoch": 6.84041976928711e-05,
      "model_forward_time": 1.7200942039489746,
      "step": 71727
    },
    {
      "epoch": 6.84041976928711e-05,
      "step": 71727,
      "training_step_time": 7.712771415710449
    },
    {
      "epoch": 6.84051513671875e-05,
      "model_forward_time": 1.7267637252807617,
      "step": 71728
    },
    {
      "epoch": 6.84051513671875e-05,
      "step": 71728,
      "training_step_time": 7.488391637802124
    },
    {
      "epoch": 6.84061050415039e-05,
      "model_forward_time": 1.7206215858459473,
      "step": 71729
    },
    {
      "epoch": 6.84061050415039e-05,
      "step": 71729,
      "training_step_time": 7.757863998413086
    },
    {
      "epoch": 6.840705871582032e-05,
      "grad_norm": 0.11076768487691879,
      "learning_rate": 2.030525312040678e-06,
      "loss": 0.0554,
      "step": 71730
    },
    {
      "epoch": 6.840705871582032e-05,
      "model_forward_time": 1.7105093002319336,
      "step": 71730
    },
    {
      "dynamics_loss_avg": 0.060837974213063716,
      "epoch": 6.840705871582032e-05,
      "step": 71730
    },
    {
      "action_loss_avg": 0.0009206514921970665,
      "epoch": 6.840705871582032e-05,
      "step": 71730
    },
    {
      "epoch": 6.840705871582032e-05,
      "step": 71730,
      "training_step_time": 8.65843415260315
    },
    {
      "epoch": 6.840801239013671e-05,
      "model_forward_time": 1.722496509552002,
      "step": 71731
    },
    {
      "epoch": 6.840801239013671e-05,
      "step": 71731,
      "training_step_time": 9.80325984954834
    },
    {
      "epoch": 6.840896606445312e-05,
      "model_forward_time": 1.7325472831726074,
      "step": 71732
    },
    {
      "epoch": 6.840896606445312e-05,
      "step": 71732,
      "training_step_time": 8.431777477264404
    },
    {
      "epoch": 6.840991973876954e-05,
      "model_forward_time": 1.3612146377563477,
      "step": 71733
    },
    {
      "epoch": 6.840991973876954e-05,
      "step": 71733,
      "training_step_time": 8.3878812789917
    },
    {
      "epoch": 6.841087341308593e-05,
      "model_forward_time": 1.7472562789916992,
      "step": 71734
    },
    {
      "epoch": 6.841087341308593e-05,
      "step": 71734,
      "training_step_time": 8.327938795089722
    },
    {
      "epoch": 6.841182708740234e-05,
      "model_forward_time": 1.7293329238891602,
      "step": 71735
    },
    {
      "epoch": 6.841182708740234e-05,
      "step": 71735,
      "training_step_time": 7.580096244812012
    },
    {
      "epoch": 6.841278076171875e-05,
      "model_forward_time": 2.281630039215088,
      "step": 71736
    },
    {
      "epoch": 6.841278076171875e-05,
      "step": 71736,
      "training_step_time": 8.724679470062256
    },
    {
      "epoch": 6.841373443603515e-05,
      "model_forward_time": 1.7404029369354248,
      "step": 71737
    },
    {
      "epoch": 6.841373443603515e-05,
      "step": 71737,
      "training_step_time": 7.680180311203003
    },
    {
      "epoch": 6.841468811035156e-05,
      "model_forward_time": 1.3576457500457764,
      "step": 71738
    },
    {
      "epoch": 6.841468811035156e-05,
      "step": 71738,
      "training_step_time": 7.784104347229004
    },
    {
      "epoch": 6.841564178466797e-05,
      "model_forward_time": 1.735123872756958,
      "step": 71739
    },
    {
      "epoch": 6.841564178466797e-05,
      "step": 71739,
      "training_step_time": 8.648246049880981
    },
    {
      "epoch": 6.841659545898437e-05,
      "grad_norm": 0.07920289039611816,
      "learning_rate": 2.029195187519325e-06,
      "loss": 0.0551,
      "step": 71740
    },
    {
      "epoch": 6.841659545898437e-05,
      "model_forward_time": 1.7205264568328857,
      "step": 71740
    },
    {
      "dynamics_loss_avg": 0.04786853585392237,
      "epoch": 6.841659545898437e-05,
      "step": 71740
    },
    {
      "action_loss_avg": 0.0014409381197765469,
      "epoch": 6.841659545898437e-05,
      "step": 71740
    },
    {
      "epoch": 6.841659545898437e-05,
      "step": 71740,
      "training_step_time": 8.461633443832397
    },
    {
      "epoch": 6.841754913330078e-05,
      "model_forward_time": 2.8401458263397217,
      "step": 71741
    },
    {
      "epoch": 6.841754913330078e-05,
      "step": 71741,
      "training_step_time": 9.723958730697632
    },
    {
      "epoch": 6.841850280761719e-05,
      "model_forward_time": 1.7308175563812256,
      "step": 71742
    },
    {
      "epoch": 6.841850280761719e-05,
      "step": 71742,
      "training_step_time": 6.778929948806763
    },
    {
      "epoch": 6.841945648193359e-05,
      "model_forward_time": 1.7352838516235352,
      "step": 71743
    },
    {
      "epoch": 6.841945648193359e-05,
      "step": 71743,
      "training_step_time": 6.731537580490112
    },
    {
      "epoch": 6.842041015625e-05,
      "model_forward_time": 1.7374451160430908,
      "step": 71744
    },
    {
      "epoch": 6.842041015625e-05,
      "step": 71744,
      "training_step_time": 6.769667863845825
    },
    {
      "epoch": 6.842136383056641e-05,
      "model_forward_time": 1.735170602798462,
      "step": 71745
    },
    {
      "epoch": 6.842136383056641e-05,
      "step": 71745,
      "training_step_time": 6.731244802474976
    },
    {
      "epoch": 6.842231750488281e-05,
      "model_forward_time": 1.7351791858673096,
      "step": 71746
    },
    {
      "epoch": 6.842231750488281e-05,
      "step": 71746,
      "training_step_time": 6.704641580581665
    },
    {
      "epoch": 6.842327117919922e-05,
      "model_forward_time": 1.3752474784851074,
      "step": 71747
    },
    {
      "epoch": 6.842327117919922e-05,
      "step": 71747,
      "training_step_time": 6.792010307312012
    },
    {
      "epoch": 6.842422485351563e-05,
      "model_forward_time": 1.7363145351409912,
      "step": 71748
    },
    {
      "epoch": 6.842422485351563e-05,
      "step": 71748,
      "training_step_time": 6.804275035858154
    },
    {
      "epoch": 6.842517852783203e-05,
      "model_forward_time": 1.732591152191162,
      "step": 71749
    },
    {
      "epoch": 6.842517852783203e-05,
      "step": 71749,
      "training_step_time": 7.8021862506866455
    },
    {
      "epoch": 6.842613220214844e-05,
      "grad_norm": 0.0729498565196991,
      "learning_rate": 2.027865387880722e-06,
      "loss": 0.0569,
      "step": 71750
    },
    {
      "epoch": 6.842613220214844e-05,
      "model_forward_time": 2.287102460861206,
      "step": 71750
    },
    {
      "dynamics_loss_avg": 0.044113189168274405,
      "epoch": 6.842613220214844e-05,
      "step": 71750
    },
    {
      "action_loss_avg": 0.0014758148579858243,
      "epoch": 6.842613220214844e-05,
      "step": 71750
    },
    {
      "epoch": 6.842613220214844e-05,
      "step": 71750,
      "training_step_time": 11.662760734558105
    },
    {
      "epoch": 6.842708587646485e-05,
      "model_forward_time": 1.7514238357543945,
      "step": 71751
    },
    {
      "epoch": 6.842708587646485e-05,
      "step": 71751,
      "training_step_time": 7.842345714569092
    },
    {
      "epoch": 6.842803955078125e-05,
      "model_forward_time": 1.7329661846160889,
      "step": 71752
    },
    {
      "epoch": 6.842803955078125e-05,
      "step": 71752,
      "training_step_time": 6.734632253646851
    },
    {
      "epoch": 6.842899322509766e-05,
      "model_forward_time": 1.7386133670806885,
      "step": 71753
    },
    {
      "epoch": 6.842899322509766e-05,
      "step": 71753,
      "training_step_time": 6.8047356605529785
    },
    {
      "epoch": 6.842994689941407e-05,
      "model_forward_time": 1.7365868091583252,
      "step": 71754
    },
    {
      "epoch": 6.842994689941407e-05,
      "step": 71754,
      "training_step_time": 6.754548072814941
    },
    {
      "epoch": 6.843090057373046e-05,
      "model_forward_time": 1.7343595027923584,
      "step": 71755
    },
    {
      "epoch": 6.843090057373046e-05,
      "step": 71755,
      "training_step_time": 6.815833330154419
    },
    {
      "epoch": 6.843185424804688e-05,
      "model_forward_time": 1.741194725036621,
      "step": 71756
    },
    {
      "epoch": 6.843185424804688e-05,
      "step": 71756,
      "training_step_time": 6.79217791557312
    },
    {
      "epoch": 6.843280792236329e-05,
      "model_forward_time": 1.721020221710205,
      "step": 71757
    },
    {
      "epoch": 6.843280792236329e-05,
      "step": 71757,
      "training_step_time": 6.783033609390259
    },
    {
      "epoch": 6.843376159667968e-05,
      "model_forward_time": 1.739104986190796,
      "step": 71758
    },
    {
      "epoch": 6.843376159667968e-05,
      "step": 71758,
      "training_step_time": 6.8096230030059814
    },
    {
      "epoch": 6.84347152709961e-05,
      "model_forward_time": 2.295433282852173,
      "step": 71759
    },
    {
      "epoch": 6.84347152709961e-05,
      "step": 71759,
      "training_step_time": 7.368091344833374
    },
    {
      "epoch": 6.84356689453125e-05,
      "grad_norm": 0.09078587591648102,
      "learning_rate": 2.0265359132702896e-06,
      "loss": 0.0535,
      "step": 71760
    },
    {
      "epoch": 6.84356689453125e-05,
      "model_forward_time": 1.7308495044708252,
      "step": 71760
    },
    {
      "dynamics_loss_avg": 0.0515609584748745,
      "epoch": 6.84356689453125e-05,
      "step": 71760
    },
    {
      "action_loss_avg": 0.0016925720585277304,
      "epoch": 6.84356689453125e-05,
      "step": 71760
    },
    {
      "epoch": 6.84356689453125e-05,
      "step": 71760,
      "training_step_time": 7.786779403686523
    },
    {
      "epoch": 6.84366226196289e-05,
      "model_forward_time": 1.7353718280792236,
      "step": 71761
    },
    {
      "epoch": 6.84366226196289e-05,
      "step": 71761,
      "training_step_time": 6.764217853546143
    },
    {
      "epoch": 6.843757629394531e-05,
      "model_forward_time": 1.7330238819122314,
      "step": 71762
    },
    {
      "epoch": 6.843757629394531e-05,
      "step": 71762,
      "training_step_time": 6.859040975570679
    },
    {
      "epoch": 6.843852996826172e-05,
      "model_forward_time": 1.7405176162719727,
      "step": 71763
    },
    {
      "epoch": 6.843852996826172e-05,
      "step": 71763,
      "training_step_time": 6.801219940185547
    },
    {
      "epoch": 6.843948364257812e-05,
      "model_forward_time": 1.7278268337249756,
      "step": 71764
    },
    {
      "epoch": 6.843948364257812e-05,
      "step": 71764,
      "training_step_time": 6.791929006576538
    },
    {
      "epoch": 6.844043731689453e-05,
      "model_forward_time": 1.7382166385650635,
      "step": 71765
    },
    {
      "epoch": 6.844043731689453e-05,
      "step": 71765,
      "training_step_time": 6.795350074768066
    },
    {
      "epoch": 6.844139099121094e-05,
      "model_forward_time": 1.7477269172668457,
      "step": 71766
    },
    {
      "epoch": 6.844139099121094e-05,
      "step": 71766,
      "training_step_time": 6.832520008087158
    },
    {
      "epoch": 6.844234466552734e-05,
      "model_forward_time": 1.7385926246643066,
      "step": 71767
    },
    {
      "epoch": 6.844234466552734e-05,
      "step": 71767,
      "training_step_time": 6.806453227996826
    },
    {
      "epoch": 6.844329833984375e-05,
      "model_forward_time": 2.29524564743042,
      "step": 71768
    },
    {
      "epoch": 6.844329833984375e-05,
      "step": 71768,
      "training_step_time": 7.405986547470093
    },
    {
      "epoch": 6.844425201416016e-05,
      "model_forward_time": 1.7390897274017334,
      "step": 71769
    },
    {
      "epoch": 6.844425201416016e-05,
      "step": 71769,
      "training_step_time": 7.849342107772827
    },
    {
      "epoch": 6.844520568847656e-05,
      "grad_norm": 0.07294522970914841,
      "learning_rate": 2.0252067638334183e-06,
      "loss": 0.0579,
      "step": 71770
    },
    {
      "epoch": 6.844520568847656e-05,
      "model_forward_time": 1.7334001064300537,
      "step": 71770
    },
    {
      "dynamics_loss_avg": 0.06202613301575184,
      "epoch": 6.844520568847656e-05,
      "step": 71770
    },
    {
      "action_loss_avg": 0.0015748238191008567,
      "epoch": 6.844520568847656e-05,
      "step": 71770
    },
    {
      "epoch": 6.844520568847656e-05,
      "step": 71770,
      "training_step_time": 7.164179563522339
    },
    {
      "epoch": 6.844615936279297e-05,
      "model_forward_time": 1.735182285308838,
      "step": 71771
    },
    {
      "epoch": 6.844615936279297e-05,
      "step": 71771,
      "training_step_time": 6.767941474914551
    },
    {
      "epoch": 6.844711303710938e-05,
      "model_forward_time": 1.7373652458190918,
      "step": 71772
    },
    {
      "epoch": 6.844711303710938e-05,
      "step": 71772,
      "training_step_time": 6.718249320983887
    },
    {
      "epoch": 6.844806671142578e-05,
      "model_forward_time": 1.7389147281646729,
      "step": 71773
    },
    {
      "epoch": 6.844806671142578e-05,
      "step": 71773,
      "training_step_time": 6.7793238162994385
    },
    {
      "epoch": 6.844902038574219e-05,
      "model_forward_time": 1.738959550857544,
      "step": 71774
    },
    {
      "epoch": 6.844902038574219e-05,
      "step": 71774,
      "training_step_time": 6.740223407745361
    },
    {
      "epoch": 6.84499740600586e-05,
      "model_forward_time": 1.7396440505981445,
      "step": 71775
    },
    {
      "epoch": 6.84499740600586e-05,
      "step": 71775,
      "training_step_time": 6.71384334564209
    },
    {
      "epoch": 6.8450927734375e-05,
      "model_forward_time": 1.7462122440338135,
      "step": 71776
    },
    {
      "epoch": 6.8450927734375e-05,
      "step": 71776,
      "training_step_time": 6.786796569824219
    },
    {
      "epoch": 6.845188140869141e-05,
      "model_forward_time": 2.304811716079712,
      "step": 71777
    },
    {
      "epoch": 6.845188140869141e-05,
      "step": 71777,
      "training_step_time": 7.3694963455200195
    },
    {
      "epoch": 6.845283508300782e-05,
      "model_forward_time": 1.7323684692382812,
      "step": 71778
    },
    {
      "epoch": 6.845283508300782e-05,
      "step": 71778,
      "training_step_time": 11.478180408477783
    },
    {
      "epoch": 6.845378875732422e-05,
      "model_forward_time": 1.7352116107940674,
      "step": 71779
    },
    {
      "epoch": 6.845378875732422e-05,
      "step": 71779,
      "training_step_time": 7.608049154281616
    },
    {
      "epoch": 6.845474243164063e-05,
      "grad_norm": 0.09717808663845062,
      "learning_rate": 2.023877939715464e-06,
      "loss": 0.0562,
      "step": 71780
    },
    {
      "epoch": 6.845474243164063e-05,
      "model_forward_time": 1.7336242198944092,
      "step": 71780
    },
    {
      "dynamics_loss_avg": 0.06721788048744201,
      "epoch": 6.845474243164063e-05,
      "step": 71780
    },
    {
      "action_loss_avg": 0.0012476531308493577,
      "epoch": 6.845474243164063e-05,
      "step": 71780
    },
    {
      "epoch": 6.845474243164063e-05,
      "step": 71780,
      "training_step_time": 6.711533784866333
    },
    {
      "epoch": 6.845569610595704e-05,
      "model_forward_time": 1.7359709739685059,
      "step": 71781
    },
    {
      "epoch": 6.845569610595704e-05,
      "step": 71781,
      "training_step_time": 6.76712703704834
    },
    {
      "epoch": 6.845664978027343e-05,
      "model_forward_time": 1.7454233169555664,
      "step": 71782
    },
    {
      "epoch": 6.845664978027343e-05,
      "step": 71782,
      "training_step_time": 6.746753931045532
    },
    {
      "epoch": 6.845760345458984e-05,
      "model_forward_time": 1.7391211986541748,
      "step": 71783
    },
    {
      "epoch": 6.845760345458984e-05,
      "step": 71783,
      "training_step_time": 6.7575788497924805
    },
    {
      "epoch": 6.845855712890626e-05,
      "model_forward_time": 1.740156888961792,
      "step": 71784
    },
    {
      "epoch": 6.845855712890626e-05,
      "step": 71784,
      "training_step_time": 6.755847930908203
    },
    {
      "epoch": 6.845951080322265e-05,
      "model_forward_time": 1.7348194122314453,
      "step": 71785
    },
    {
      "epoch": 6.845951080322265e-05,
      "step": 71785,
      "training_step_time": 6.729209899902344
    },
    {
      "epoch": 6.846046447753906e-05,
      "model_forward_time": 2.305445909500122,
      "step": 71786
    },
    {
      "epoch": 6.846046447753906e-05,
      "step": 71786,
      "training_step_time": 7.375970840454102
    },
    {
      "epoch": 6.846141815185547e-05,
      "model_forward_time": 1.738037109375,
      "step": 71787
    },
    {
      "epoch": 6.846141815185547e-05,
      "step": 71787,
      "training_step_time": 9.554723501205444
    },
    {
      "epoch": 6.846237182617187e-05,
      "model_forward_time": 1.735302209854126,
      "step": 71788
    },
    {
      "epoch": 6.846237182617187e-05,
      "step": 71788,
      "training_step_time": 7.679737567901611
    },
    {
      "epoch": 6.846332550048828e-05,
      "model_forward_time": 1.734224796295166,
      "step": 71789
    },
    {
      "epoch": 6.846332550048828e-05,
      "step": 71789,
      "training_step_time": 7.305019378662109
    },
    {
      "epoch": 6.84642791748047e-05,
      "grad_norm": 0.0692751333117485,
      "learning_rate": 2.0225494410617447e-06,
      "loss": 0.0556,
      "step": 71790
    },
    {
      "epoch": 6.84642791748047e-05,
      "model_forward_time": 1.7441625595092773,
      "step": 71790
    },
    {
      "dynamics_loss_avg": 0.07216179501265288,
      "epoch": 6.84642791748047e-05,
      "step": 71790
    },
    {
      "action_loss_avg": 0.001993323490023613,
      "epoch": 6.84642791748047e-05,
      "step": 71790
    },
    {
      "epoch": 6.84642791748047e-05,
      "step": 71790,
      "training_step_time": 6.7520458698272705
    },
    {
      "epoch": 6.846523284912109e-05,
      "model_forward_time": 1.736772060394287,
      "step": 71791
    },
    {
      "epoch": 6.846523284912109e-05,
      "step": 71791,
      "training_step_time": 6.766993284225464
    },
    {
      "epoch": 6.84661865234375e-05,
      "model_forward_time": 1.7381021976470947,
      "step": 71792
    },
    {
      "epoch": 6.84661865234375e-05,
      "step": 71792,
      "training_step_time": 6.784266948699951
    },
    {
      "epoch": 6.846714019775391e-05,
      "model_forward_time": 1.730506181716919,
      "step": 71793
    },
    {
      "epoch": 6.846714019775391e-05,
      "step": 71793,
      "training_step_time": 6.791255235671997
    },
    {
      "epoch": 6.846809387207031e-05,
      "model_forward_time": 1.7378895282745361,
      "step": 71794
    },
    {
      "epoch": 6.846809387207031e-05,
      "step": 71794,
      "training_step_time": 6.814393520355225
    },
    {
      "epoch": 6.846904754638672e-05,
      "model_forward_time": 1.7459039688110352,
      "step": 71795
    },
    {
      "epoch": 6.846904754638672e-05,
      "step": 71795,
      "training_step_time": 7.924000978469849
    },
    {
      "epoch": 6.847000122070313e-05,
      "model_forward_time": 1.7333223819732666,
      "step": 71796
    },
    {
      "epoch": 6.847000122070313e-05,
      "step": 71796,
      "training_step_time": 8.341158390045166
    },
    {
      "epoch": 6.847095489501953e-05,
      "model_forward_time": 1.7363510131835938,
      "step": 71797
    },
    {
      "epoch": 6.847095489501953e-05,
      "step": 71797,
      "training_step_time": 7.736568927764893
    },
    {
      "epoch": 6.847190856933594e-05,
      "model_forward_time": 1.7488064765930176,
      "step": 71798
    },
    {
      "epoch": 6.847190856933594e-05,
      "step": 71798,
      "training_step_time": 7.368632555007935
    },
    {
      "epoch": 6.847286224365235e-05,
      "model_forward_time": 1.7338004112243652,
      "step": 71799
    },
    {
      "epoch": 6.847286224365235e-05,
      "step": 71799,
      "training_step_time": 6.788039922714233
    },
    {
      "epoch": 6.847381591796875e-05,
      "grad_norm": 0.0830220878124237,
      "learning_rate": 2.0212212680175402e-06,
      "loss": 0.0549,
      "step": 71800
    },
    {
      "epoch": 6.847381591796875e-05,
      "model_forward_time": 1.7253704071044922,
      "step": 71800
    },
    {
      "dynamics_loss_avg": 0.06260268278419971,
      "epoch": 6.847381591796875e-05,
      "step": 71800
    },
    {
      "action_loss_avg": 0.001202711113728583,
      "epoch": 6.847381591796875e-05,
      "step": 71800
    },
    {
      "epoch": 6.847381591796875e-05,
      "step": 71800,
      "training_step_time": 6.797075510025024
    },
    {
      "epoch": 6.847476959228516e-05,
      "model_forward_time": 1.737865686416626,
      "step": 71801
    },
    {
      "epoch": 6.847476959228516e-05,
      "step": 71801,
      "training_step_time": 6.797516107559204
    },
    {
      "epoch": 6.847572326660157e-05,
      "model_forward_time": 1.7322032451629639,
      "step": 71802
    },
    {
      "epoch": 6.847572326660157e-05,
      "step": 71802,
      "training_step_time": 6.832583904266357
    },
    {
      "epoch": 6.847667694091797e-05,
      "model_forward_time": 1.7466135025024414,
      "step": 71803
    },
    {
      "epoch": 6.847667694091797e-05,
      "step": 71803,
      "training_step_time": 6.827913761138916
    },
    {
      "epoch": 6.847763061523438e-05,
      "model_forward_time": 1.7414531707763672,
      "step": 71804
    },
    {
      "epoch": 6.847763061523438e-05,
      "step": 71804,
      "training_step_time": 7.866280555725098
    },
    {
      "epoch": 6.847858428955079e-05,
      "model_forward_time": 2.3038370609283447,
      "step": 71805
    },
    {
      "epoch": 6.847858428955079e-05,
      "step": 71805,
      "training_step_time": 9.27546238899231
    },
    {
      "epoch": 6.847953796386719e-05,
      "model_forward_time": 1.7288525104522705,
      "step": 71806
    },
    {
      "epoch": 6.847953796386719e-05,
      "step": 71806,
      "training_step_time": 7.839094161987305
    },
    {
      "epoch": 6.84804916381836e-05,
      "model_forward_time": 1.730837345123291,
      "step": 71807
    },
    {
      "epoch": 6.84804916381836e-05,
      "step": 71807,
      "training_step_time": 7.545888185501099
    },
    {
      "epoch": 6.84814453125e-05,
      "model_forward_time": 1.7210259437561035,
      "step": 71808
    },
    {
      "epoch": 6.84814453125e-05,
      "step": 71808,
      "training_step_time": 6.758322715759277
    },
    {
      "epoch": 6.84823989868164e-05,
      "model_forward_time": 1.7177677154541016,
      "step": 71809
    },
    {
      "epoch": 6.84823989868164e-05,
      "step": 71809,
      "training_step_time": 6.7490293979644775
    },
    {
      "epoch": 6.848335266113281e-05,
      "grad_norm": 0.1563003659248352,
      "learning_rate": 2.0198934207281017e-06,
      "loss": 0.0528,
      "step": 71810
    },
    {
      "epoch": 6.848335266113281e-05,
      "model_forward_time": 1.3229031562805176,
      "step": 71810
    },
    {
      "dynamics_loss_avg": 0.05620235595852137,
      "epoch": 6.848335266113281e-05,
      "step": 71810
    },
    {
      "action_loss_avg": 0.0013222559180576354,
      "epoch": 6.848335266113281e-05,
      "step": 71810
    },
    {
      "epoch": 6.848335266113281e-05,
      "step": 71810,
      "training_step_time": 6.790634870529175
    },
    {
      "epoch": 6.848430633544923e-05,
      "model_forward_time": 1.7037930488586426,
      "step": 71811
    },
    {
      "epoch": 6.848430633544923e-05,
      "step": 71811,
      "training_step_time": 6.8072896003723145
    },
    {
      "epoch": 6.848526000976562e-05,
      "model_forward_time": 1.7187206745147705,
      "step": 71812
    },
    {
      "epoch": 6.848526000976562e-05,
      "step": 71812,
      "training_step_time": 6.762601852416992
    },
    {
      "epoch": 6.848621368408203e-05,
      "model_forward_time": 1.7177765369415283,
      "step": 71813
    },
    {
      "epoch": 6.848621368408203e-05,
      "step": 71813,
      "training_step_time": 6.816235303878784
    },
    {
      "epoch": 6.848716735839843e-05,
      "model_forward_time": 2.2599265575408936,
      "step": 71814
    },
    {
      "epoch": 6.848716735839843e-05,
      "step": 71814,
      "training_step_time": 9.703053712844849
    },
    {
      "epoch": 6.848812103271484e-05,
      "model_forward_time": 1.7132041454315186,
      "step": 71815
    },
    {
      "epoch": 6.848812103271484e-05,
      "step": 71815,
      "training_step_time": 8.292307376861572
    },
    {
      "epoch": 6.848907470703125e-05,
      "model_forward_time": 1.7238214015960693,
      "step": 71816
    },
    {
      "epoch": 6.848907470703125e-05,
      "step": 71816,
      "training_step_time": 7.664639711380005
    },
    {
      "epoch": 6.849002838134765e-05,
      "model_forward_time": 1.3626708984375,
      "step": 71817
    },
    {
      "epoch": 6.849002838134765e-05,
      "step": 71817,
      "training_step_time": 6.80962872505188
    },
    {
      "epoch": 6.849098205566406e-05,
      "model_forward_time": 1.7182412147521973,
      "step": 71818
    },
    {
      "epoch": 6.849098205566406e-05,
      "step": 71818,
      "training_step_time": 6.776760578155518
    },
    {
      "epoch": 6.849193572998047e-05,
      "model_forward_time": 1.7254011631011963,
      "step": 71819
    },
    {
      "epoch": 6.849193572998047e-05,
      "step": 71819,
      "training_step_time": 6.760485410690308
    },
    {
      "epoch": 6.849288940429687e-05,
      "grad_norm": 0.09781773388385773,
      "learning_rate": 2.018565899338636e-06,
      "loss": 0.0545,
      "step": 71820
    },
    {
      "epoch": 6.849288940429687e-05,
      "model_forward_time": 1.7201685905456543,
      "step": 71820
    },
    {
      "dynamics_loss_avg": 0.06357402540743351,
      "epoch": 6.849288940429687e-05,
      "step": 71820
    },
    {
      "action_loss_avg": 0.002202074124943465,
      "epoch": 6.849288940429687e-05,
      "step": 71820
    },
    {
      "epoch": 6.849288940429687e-05,
      "step": 71820,
      "training_step_time": 6.721494436264038
    },
    {
      "epoch": 6.849384307861328e-05,
      "model_forward_time": 1.723050832748413,
      "step": 71821
    },
    {
      "epoch": 6.849384307861328e-05,
      "step": 71821,
      "training_step_time": 6.776808023452759
    },
    {
      "epoch": 6.849479675292969e-05,
      "model_forward_time": 1.7233285903930664,
      "step": 71822
    },
    {
      "epoch": 6.849479675292969e-05,
      "step": 71822,
      "training_step_time": 6.8211493492126465
    },
    {
      "epoch": 6.849575042724609e-05,
      "model_forward_time": 2.3128933906555176,
      "step": 71823
    },
    {
      "epoch": 6.849575042724609e-05,
      "step": 71823,
      "training_step_time": 8.725785970687866
    },
    {
      "epoch": 6.84967041015625e-05,
      "model_forward_time": 1.7236547470092773,
      "step": 71824
    },
    {
      "epoch": 6.84967041015625e-05,
      "step": 71824,
      "training_step_time": 7.733182907104492
    },
    {
      "epoch": 6.849765777587891e-05,
      "model_forward_time": 1.7259299755096436,
      "step": 71825
    },
    {
      "epoch": 6.849765777587891e-05,
      "step": 71825,
      "training_step_time": 7.7253968715667725
    },
    {
      "epoch": 6.84986114501953e-05,
      "model_forward_time": 1.711188554763794,
      "step": 71826
    },
    {
      "epoch": 6.84986114501953e-05,
      "step": 71826,
      "training_step_time": 6.824058532714844
    },
    {
      "epoch": 6.849956512451172e-05,
      "model_forward_time": 1.7248167991638184,
      "step": 71827
    },
    {
      "epoch": 6.849956512451172e-05,
      "step": 71827,
      "training_step_time": 6.824324369430542
    },
    {
      "epoch": 6.850051879882813e-05,
      "model_forward_time": 1.3554697036743164,
      "step": 71828
    },
    {
      "epoch": 6.850051879882813e-05,
      "step": 71828,
      "training_step_time": 6.836104869842529
    },
    {
      "epoch": 6.850147247314453e-05,
      "model_forward_time": 1.732942819595337,
      "step": 71829
    },
    {
      "epoch": 6.850147247314453e-05,
      "step": 71829,
      "training_step_time": 6.837756633758545
    },
    {
      "epoch": 6.850242614746094e-05,
      "grad_norm": 0.08021208643913269,
      "learning_rate": 2.0172387039943225e-06,
      "loss": 0.0537,
      "step": 71830
    },
    {
      "epoch": 6.850242614746094e-05,
      "model_forward_time": 1.7240378856658936,
      "step": 71830
    },
    {
      "dynamics_loss_avg": 0.05796052198857069,
      "epoch": 6.850242614746094e-05,
      "step": 71830
    },
    {
      "action_loss_avg": 0.00165991602698341,
      "epoch": 6.850242614746094e-05,
      "step": 71830
    },
    {
      "epoch": 6.850242614746094e-05,
      "step": 71830,
      "training_step_time": 6.807699918746948
    },
    {
      "epoch": 6.850337982177735e-05,
      "model_forward_time": 1.7330586910247803,
      "step": 71831
    },
    {
      "epoch": 6.850337982177735e-05,
      "step": 71831,
      "training_step_time": 6.714524507522583
    },
    {
      "epoch": 6.850433349609374e-05,
      "model_forward_time": 1.7674705982208252,
      "step": 71832
    },
    {
      "epoch": 6.850433349609374e-05,
      "step": 71832,
      "training_step_time": 9.22965145111084
    },
    {
      "epoch": 6.850528717041015e-05,
      "model_forward_time": 2.2719054222106934,
      "step": 71833
    },
    {
      "epoch": 6.850528717041015e-05,
      "step": 71833,
      "training_step_time": 9.642964363098145
    },
    {
      "epoch": 6.850624084472657e-05,
      "model_forward_time": 1.715827465057373,
      "step": 71834
    },
    {
      "epoch": 6.850624084472657e-05,
      "step": 71834,
      "training_step_time": 8.689555883407593
    },
    {
      "epoch": 6.850719451904296e-05,
      "model_forward_time": 1.7040283679962158,
      "step": 71835
    },
    {
      "epoch": 6.850719451904296e-05,
      "step": 71835,
      "training_step_time": 7.292644023895264
    },
    {
      "epoch": 6.850814819335937e-05,
      "model_forward_time": 1.713843584060669,
      "step": 71836
    },
    {
      "epoch": 6.850814819335937e-05,
      "step": 71836,
      "training_step_time": 6.7834813594818115
    },
    {
      "epoch": 6.850910186767578e-05,
      "model_forward_time": 1.716618299484253,
      "step": 71837
    },
    {
      "epoch": 6.850910186767578e-05,
      "step": 71837,
      "training_step_time": 6.813791275024414
    },
    {
      "epoch": 6.851005554199218e-05,
      "model_forward_time": 1.7096788883209229,
      "step": 71838
    },
    {
      "epoch": 6.851005554199218e-05,
      "step": 71838,
      "training_step_time": 6.83183741569519
    },
    {
      "epoch": 6.851100921630859e-05,
      "model_forward_time": 1.723719835281372,
      "step": 71839
    },
    {
      "epoch": 6.851100921630859e-05,
      "step": 71839,
      "training_step_time": 6.767362594604492
    },
    {
      "epoch": 6.8511962890625e-05,
      "grad_norm": 0.07881350815296173,
      "learning_rate": 2.015911834840297e-06,
      "loss": 0.0576,
      "step": 71840
    },
    {
      "epoch": 6.8511962890625e-05,
      "model_forward_time": 1.713709831237793,
      "step": 71840
    },
    {
      "dynamics_loss_avg": 0.06510738898068666,
      "epoch": 6.8511962890625e-05,
      "step": 71840
    },
    {
      "action_loss_avg": 0.0012997915036976337,
      "epoch": 6.8511962890625e-05,
      "step": 71840
    },
    {
      "epoch": 6.8511962890625e-05,
      "step": 71840,
      "training_step_time": 6.7286481857299805
    },
    {
      "epoch": 6.85129165649414e-05,
      "model_forward_time": 1.7922437191009521,
      "step": 71841
    },
    {
      "epoch": 6.85129165649414e-05,
      "step": 71841,
      "training_step_time": 8.199996948242188
    },
    {
      "epoch": 6.851387023925781e-05,
      "model_forward_time": 2.296505928039551,
      "step": 71842
    },
    {
      "epoch": 6.851387023925781e-05,
      "step": 71842,
      "training_step_time": 8.732156991958618
    },
    {
      "epoch": 6.851482391357422e-05,
      "model_forward_time": 1.7265822887420654,
      "step": 71843
    },
    {
      "epoch": 6.851482391357422e-05,
      "step": 71843,
      "training_step_time": 7.657047986984253
    },
    {
      "epoch": 6.851577758789062e-05,
      "model_forward_time": 1.6945977210998535,
      "step": 71844
    },
    {
      "epoch": 6.851577758789062e-05,
      "step": 71844,
      "training_step_time": 7.346968173980713
    },
    {
      "epoch": 6.851673126220703e-05,
      "model_forward_time": 1.7242639064788818,
      "step": 71845
    },
    {
      "epoch": 6.851673126220703e-05,
      "step": 71845,
      "training_step_time": 6.77984619140625
    },
    {
      "epoch": 6.851768493652344e-05,
      "model_forward_time": 1.71824049949646,
      "step": 71846
    },
    {
      "epoch": 6.851768493652344e-05,
      "step": 71846,
      "training_step_time": 6.721825361251831
    },
    {
      "epoch": 6.851863861083984e-05,
      "model_forward_time": 1.723667860031128,
      "step": 71847
    },
    {
      "epoch": 6.851863861083984e-05,
      "step": 71847,
      "training_step_time": 6.736991882324219
    },
    {
      "epoch": 6.851959228515625e-05,
      "model_forward_time": 1.7219629287719727,
      "step": 71848
    },
    {
      "epoch": 6.851959228515625e-05,
      "step": 71848,
      "training_step_time": 6.75260329246521
    },
    {
      "epoch": 6.852054595947266e-05,
      "model_forward_time": 1.7204182147979736,
      "step": 71849
    },
    {
      "epoch": 6.852054595947266e-05,
      "step": 71849,
      "training_step_time": 6.77590012550354
    },
    {
      "epoch": 6.852149963378906e-05,
      "grad_norm": 0.09286762028932571,
      "learning_rate": 2.01458529202167e-06,
      "loss": 0.0536,
      "step": 71850
    },
    {
      "epoch": 6.852149963378906e-05,
      "model_forward_time": 1.7771692276000977,
      "step": 71850
    },
    {
      "dynamics_loss_avg": 0.0632939200848341,
      "epoch": 6.852149963378906e-05,
      "step": 71850
    },
    {
      "action_loss_avg": 0.0022092492843512447,
      "epoch": 6.852149963378906e-05,
      "step": 71850
    },
    {
      "epoch": 6.852149963378906e-05,
      "step": 71850,
      "training_step_time": 8.605344295501709
    },
    {
      "epoch": 6.852245330810547e-05,
      "model_forward_time": 2.2996909618377686,
      "step": 71851
    },
    {
      "epoch": 6.852245330810547e-05,
      "step": 71851,
      "training_step_time": 9.728852987289429
    },
    {
      "epoch": 6.852340698242188e-05,
      "model_forward_time": 1.7218983173370361,
      "step": 71852
    },
    {
      "epoch": 6.852340698242188e-05,
      "step": 71852,
      "training_step_time": 7.66865873336792
    },
    {
      "epoch": 6.852436065673828e-05,
      "model_forward_time": 1.704766035079956,
      "step": 71853
    },
    {
      "epoch": 6.852436065673828e-05,
      "step": 71853,
      "training_step_time": 7.545454502105713
    },
    {
      "epoch": 6.852531433105469e-05,
      "model_forward_time": 1.722346544265747,
      "step": 71854
    },
    {
      "epoch": 6.852531433105469e-05,
      "step": 71854,
      "training_step_time": 6.7648725509643555
    },
    {
      "epoch": 6.85262680053711e-05,
      "model_forward_time": 1.7025859355926514,
      "step": 71855
    },
    {
      "epoch": 6.85262680053711e-05,
      "step": 71855,
      "training_step_time": 6.715761661529541
    },
    {
      "epoch": 6.85272216796875e-05,
      "model_forward_time": 1.7213926315307617,
      "step": 71856
    },
    {
      "epoch": 6.85272216796875e-05,
      "step": 71856,
      "training_step_time": 6.762654781341553
    },
    {
      "epoch": 6.85281753540039e-05,
      "model_forward_time": 1.7184383869171143,
      "step": 71857
    },
    {
      "epoch": 6.85281753540039e-05,
      "step": 71857,
      "training_step_time": 6.811435699462891
    },
    {
      "epoch": 6.852912902832032e-05,
      "model_forward_time": 1.7210354804992676,
      "step": 71858
    },
    {
      "epoch": 6.852912902832032e-05,
      "step": 71858,
      "training_step_time": 6.778626441955566
    },
    {
      "epoch": 6.853008270263671e-05,
      "model_forward_time": 1.7760376930236816,
      "step": 71859
    },
    {
      "epoch": 6.853008270263671e-05,
      "step": 71859,
      "training_step_time": 7.708143949508667
    },
    {
      "epoch": 6.853103637695312e-05,
      "grad_norm": 0.09241771697998047,
      "learning_rate": 2.0132590756835045e-06,
      "loss": 0.055,
      "step": 71860
    },
    {
      "epoch": 6.853103637695312e-05,
      "model_forward_time": 2.2841436862945557,
      "step": 71860
    },
    {
      "dynamics_loss_avg": 0.06322504300624132,
      "epoch": 6.853103637695312e-05,
      "step": 71860
    },
    {
      "action_loss_avg": 0.0013744004536420107,
      "epoch": 6.853103637695312e-05,
      "step": 71860
    },
    {
      "epoch": 6.853103637695312e-05,
      "step": 71860,
      "training_step_time": 9.794487476348877
    },
    {
      "epoch": 6.853199005126954e-05,
      "model_forward_time": 1.7163231372833252,
      "step": 71861
    },
    {
      "epoch": 6.853199005126954e-05,
      "step": 71861,
      "training_step_time": 8.702104091644287
    },
    {
      "epoch": 6.853294372558593e-05,
      "model_forward_time": 1.7126250267028809,
      "step": 71862
    },
    {
      "epoch": 6.853294372558593e-05,
      "step": 71862,
      "training_step_time": 8.201268434524536
    },
    {
      "epoch": 6.853389739990234e-05,
      "model_forward_time": 1.7274446487426758,
      "step": 71863
    },
    {
      "epoch": 6.853389739990234e-05,
      "step": 71863,
      "training_step_time": 6.809372186660767
    },
    {
      "epoch": 6.853485107421875e-05,
      "model_forward_time": 1.718019723892212,
      "step": 71864
    },
    {
      "epoch": 6.853485107421875e-05,
      "step": 71864,
      "training_step_time": 6.805268287658691
    },
    {
      "epoch": 6.853580474853515e-05,
      "model_forward_time": 1.7158691883087158,
      "step": 71865
    },
    {
      "epoch": 6.853580474853515e-05,
      "step": 71865,
      "training_step_time": 6.807146310806274
    },
    {
      "epoch": 6.853675842285156e-05,
      "model_forward_time": 1.7046489715576172,
      "step": 71866
    },
    {
      "epoch": 6.853675842285156e-05,
      "step": 71866,
      "training_step_time": 6.781976222991943
    },
    {
      "epoch": 6.853771209716797e-05,
      "model_forward_time": 1.715956449508667,
      "step": 71867
    },
    {
      "epoch": 6.853771209716797e-05,
      "step": 71867,
      "training_step_time": 6.8263468742370605
    },
    {
      "epoch": 6.853866577148437e-05,
      "model_forward_time": 1.7752602100372314,
      "step": 71868
    },
    {
      "epoch": 6.853866577148437e-05,
      "step": 71868,
      "training_step_time": 7.664691686630249
    },
    {
      "epoch": 6.853961944580078e-05,
      "model_forward_time": 2.2727954387664795,
      "step": 71869
    },
    {
      "epoch": 6.853961944580078e-05,
      "step": 71869,
      "training_step_time": 8.40326714515686
    },
    {
      "epoch": 6.854057312011719e-05,
      "grad_norm": 0.07184597849845886,
      "learning_rate": 2.0119331859708384e-06,
      "loss": 0.0532,
      "step": 71870
    },
    {
      "epoch": 6.854057312011719e-05,
      "model_forward_time": 1.711113691329956,
      "step": 71870
    },
    {
      "dynamics_loss_avg": 0.05087924879044294,
      "epoch": 6.854057312011719e-05,
      "step": 71870
    },
    {
      "action_loss_avg": 0.001465476310113445,
      "epoch": 6.854057312011719e-05,
      "step": 71870
    },
    {
      "epoch": 6.854057312011719e-05,
      "step": 71870,
      "training_step_time": 7.74071192741394
    },
    {
      "epoch": 6.854152679443359e-05,
      "model_forward_time": 1.7199876308441162,
      "step": 71871
    },
    {
      "epoch": 6.854152679443359e-05,
      "step": 71871,
      "training_step_time": 7.676860332489014
    },
    {
      "epoch": 6.854248046875e-05,
      "model_forward_time": 1.7276332378387451,
      "step": 71872
    },
    {
      "epoch": 6.854248046875e-05,
      "step": 71872,
      "training_step_time": 7.281078338623047
    },
    {
      "epoch": 6.854343414306641e-05,
      "model_forward_time": 1.710085153579712,
      "step": 71873
    },
    {
      "epoch": 6.854343414306641e-05,
      "step": 71873,
      "training_step_time": 6.864248752593994
    },
    {
      "epoch": 6.854438781738281e-05,
      "model_forward_time": 1.7209491729736328,
      "step": 71874
    },
    {
      "epoch": 6.854438781738281e-05,
      "step": 71874,
      "training_step_time": 6.835330486297607
    },
    {
      "epoch": 6.854534149169922e-05,
      "model_forward_time": 1.7007091045379639,
      "step": 71875
    },
    {
      "epoch": 6.854534149169922e-05,
      "step": 71875,
      "training_step_time": 6.839998245239258
    },
    {
      "epoch": 6.854629516601563e-05,
      "model_forward_time": 1.7203800678253174,
      "step": 71876
    },
    {
      "epoch": 6.854629516601563e-05,
      "step": 71876,
      "training_step_time": 6.830984354019165
    },
    {
      "epoch": 6.854724884033203e-05,
      "model_forward_time": 1.7819850444793701,
      "step": 71877
    },
    {
      "epoch": 6.854724884033203e-05,
      "step": 71877,
      "training_step_time": 7.664882659912109
    },
    {
      "epoch": 6.854820251464844e-05,
      "model_forward_time": 1.704268455505371,
      "step": 71878
    },
    {
      "epoch": 6.854820251464844e-05,
      "step": 71878,
      "training_step_time": 9.724198818206787
    },
    {
      "epoch": 6.854915618896485e-05,
      "model_forward_time": 2.2672441005706787,
      "step": 71879
    },
    {
      "epoch": 6.854915618896485e-05,
      "step": 71879,
      "training_step_time": 9.573961734771729
    },
    {
      "epoch": 6.855010986328125e-05,
      "grad_norm": 0.09255191683769226,
      "learning_rate": 2.0106076230286654e-06,
      "loss": 0.0565,
      "step": 71880
    },
    {
      "epoch": 6.855010986328125e-05,
      "model_forward_time": 1.7163584232330322,
      "step": 71880
    },
    {
      "dynamics_loss_avg": 0.055218455381691456,
      "epoch": 6.855010986328125e-05,
      "step": 71880
    },
    {
      "action_loss_avg": 0.0011359459953382612,
      "epoch": 6.855010986328125e-05,
      "step": 71880
    },
    {
      "epoch": 6.855010986328125e-05,
      "step": 71880,
      "training_step_time": 7.710971355438232
    },
    {
      "epoch": 6.855106353759766e-05,
      "model_forward_time": 1.7168951034545898,
      "step": 71881
    },
    {
      "epoch": 6.855106353759766e-05,
      "step": 71881,
      "training_step_time": 7.620478868484497
    },
    {
      "epoch": 6.855201721191407e-05,
      "model_forward_time": 1.7047615051269531,
      "step": 71882
    },
    {
      "epoch": 6.855201721191407e-05,
      "step": 71882,
      "training_step_time": 6.838523626327515
    },
    {
      "epoch": 6.855297088623046e-05,
      "model_forward_time": 1.7220144271850586,
      "step": 71883
    },
    {
      "epoch": 6.855297088623046e-05,
      "step": 71883,
      "training_step_time": 6.812575578689575
    },
    {
      "epoch": 6.855392456054688e-05,
      "model_forward_time": 1.7191379070281982,
      "step": 71884
    },
    {
      "epoch": 6.855392456054688e-05,
      "step": 71884,
      "training_step_time": 6.800934076309204
    },
    {
      "epoch": 6.855487823486329e-05,
      "model_forward_time": 1.7210099697113037,
      "step": 71885
    },
    {
      "epoch": 6.855487823486329e-05,
      "step": 71885,
      "training_step_time": 6.8151960372924805
    },
    {
      "epoch": 6.855583190917968e-05,
      "model_forward_time": 1.7604789733886719,
      "step": 71886
    },
    {
      "epoch": 6.855583190917968e-05,
      "step": 71886,
      "training_step_time": 7.711390018463135
    },
    {
      "epoch": 6.85567855834961e-05,
      "model_forward_time": 1.7299983501434326,
      "step": 71887
    },
    {
      "epoch": 6.85567855834961e-05,
      "step": 71887,
      "training_step_time": 8.451513051986694
    },
    {
      "epoch": 6.85577392578125e-05,
      "model_forward_time": 2.2809479236602783,
      "step": 71888
    },
    {
      "epoch": 6.85577392578125e-05,
      "step": 71888,
      "training_step_time": 9.476778030395508
    },
    {
      "epoch": 6.85586929321289e-05,
      "model_forward_time": 1.703120231628418,
      "step": 71889
    },
    {
      "epoch": 6.85586929321289e-05,
      "step": 71889,
      "training_step_time": 7.677325487136841
    },
    {
      "epoch": 6.855964660644531e-05,
      "grad_norm": 0.08279335498809814,
      "learning_rate": 2.009282387001949e-06,
      "loss": 0.0541,
      "step": 71890
    },
    {
      "epoch": 6.855964660644531e-05,
      "model_forward_time": 1.7106645107269287,
      "step": 71890
    },
    {
      "dynamics_loss_avg": 0.06019998714327812,
      "epoch": 6.855964660644531e-05,
      "step": 71890
    },
    {
      "action_loss_avg": 0.0012177026335848495,
      "epoch": 6.855964660644531e-05,
      "step": 71890
    },
    {
      "epoch": 6.855964660644531e-05,
      "step": 71890,
      "training_step_time": 8.249879121780396
    },
    {
      "epoch": 6.856060028076172e-05,
      "model_forward_time": 1.7275736331939697,
      "step": 71891
    },
    {
      "epoch": 6.856060028076172e-05,
      "step": 71891,
      "training_step_time": 6.841536521911621
    },
    {
      "epoch": 6.856155395507812e-05,
      "model_forward_time": 1.3551597595214844,
      "step": 71892
    },
    {
      "epoch": 6.856155395507812e-05,
      "step": 71892,
      "training_step_time": 6.814086437225342
    },
    {
      "epoch": 6.856250762939453e-05,
      "model_forward_time": 1.7162220478057861,
      "step": 71893
    },
    {
      "epoch": 6.856250762939453e-05,
      "step": 71893,
      "training_step_time": 6.827649116516113
    },
    {
      "epoch": 6.856346130371094e-05,
      "model_forward_time": 1.7180604934692383,
      "step": 71894
    },
    {
      "epoch": 6.856346130371094e-05,
      "step": 71894,
      "training_step_time": 6.831814527511597
    },
    {
      "epoch": 6.856441497802734e-05,
      "model_forward_time": 1.7737541198730469,
      "step": 71895
    },
    {
      "epoch": 6.856441497802734e-05,
      "step": 71895,
      "training_step_time": 7.670332431793213
    },
    {
      "epoch": 6.856536865234375e-05,
      "model_forward_time": 1.726806879043579,
      "step": 71896
    },
    {
      "epoch": 6.856536865234375e-05,
      "step": 71896,
      "training_step_time": 9.077428340911865
    },
    {
      "epoch": 6.856632232666016e-05,
      "model_forward_time": 2.2805254459381104,
      "step": 71897
    },
    {
      "epoch": 6.856632232666016e-05,
      "step": 71897,
      "training_step_time": 9.298608779907227
    },
    {
      "epoch": 6.856727600097656e-05,
      "model_forward_time": 1.7129347324371338,
      "step": 71898
    },
    {
      "epoch": 6.856727600097656e-05,
      "step": 71898,
      "training_step_time": 9.138769626617432
    },
    {
      "epoch": 6.856822967529297e-05,
      "model_forward_time": 1.7221195697784424,
      "step": 71899
    },
    {
      "epoch": 6.856822967529297e-05,
      "step": 71899,
      "training_step_time": 7.6917712688446045
    },
    {
      "epoch": 6.856918334960938e-05,
      "grad_norm": 0.09986645728349686,
      "learning_rate": 2.0079574780356116e-06,
      "loss": 0.0556,
      "step": 71900
    },
    {
      "epoch": 6.856918334960938e-05,
      "model_forward_time": 1.7170631885528564,
      "step": 71900
    },
    {
      "dynamics_loss_avg": 0.0525723272934556,
      "epoch": 6.856918334960938e-05,
      "step": 71900
    },
    {
      "action_loss_avg": 0.0014143912470899523,
      "epoch": 6.856918334960938e-05,
      "step": 71900
    },
    {
      "epoch": 6.856918334960938e-05,
      "step": 71900,
      "training_step_time": 7.211655139923096
    },
    {
      "epoch": 6.857013702392578e-05,
      "model_forward_time": 1.7074742317199707,
      "step": 71901
    },
    {
      "epoch": 6.857013702392578e-05,
      "step": 71901,
      "training_step_time": 6.824392557144165
    },
    {
      "epoch": 6.857109069824219e-05,
      "model_forward_time": 1.718827724456787,
      "step": 71902
    },
    {
      "epoch": 6.857109069824219e-05,
      "step": 71902,
      "training_step_time": 6.79294753074646
    },
    {
      "epoch": 6.85720443725586e-05,
      "model_forward_time": 1.724330186843872,
      "step": 71903
    },
    {
      "epoch": 6.85720443725586e-05,
      "step": 71903,
      "training_step_time": 6.829346418380737
    },
    {
      "epoch": 6.8572998046875e-05,
      "model_forward_time": 1.7648508548736572,
      "step": 71904
    },
    {
      "epoch": 6.8572998046875e-05,
      "step": 71904,
      "training_step_time": 7.7200610637664795
    },
    {
      "epoch": 6.857395172119141e-05,
      "model_forward_time": 1.3493778705596924,
      "step": 71905
    },
    {
      "epoch": 6.857395172119141e-05,
      "step": 71905,
      "training_step_time": 8.641376972198486
    },
    {
      "epoch": 6.857490539550782e-05,
      "model_forward_time": 2.260165214538574,
      "step": 71906
    },
    {
      "epoch": 6.857490539550782e-05,
      "step": 71906,
      "training_step_time": 9.728096961975098
    },
    {
      "epoch": 6.857585906982422e-05,
      "model_forward_time": 1.7235279083251953,
      "step": 71907
    },
    {
      "epoch": 6.857585906982422e-05,
      "step": 71907,
      "training_step_time": 9.58895754814148
    },
    {
      "epoch": 6.857681274414063e-05,
      "model_forward_time": 1.7302348613739014,
      "step": 71908
    },
    {
      "epoch": 6.857681274414063e-05,
      "step": 71908,
      "training_step_time": 7.738126277923584
    },
    {
      "epoch": 6.857776641845704e-05,
      "model_forward_time": 1.7131438255310059,
      "step": 71909
    },
    {
      "epoch": 6.857776641845704e-05,
      "step": 71909,
      "training_step_time": 7.354180574417114
    },
    {
      "epoch": 6.857872009277343e-05,
      "grad_norm": 0.09621374309062958,
      "learning_rate": 2.0066328962745487e-06,
      "loss": 0.0544,
      "step": 71910
    },
    {
      "epoch": 6.857872009277343e-05,
      "model_forward_time": 1.732881784439087,
      "step": 71910
    },
    {
      "dynamics_loss_avg": 0.05872973836958408,
      "epoch": 6.857872009277343e-05,
      "step": 71910
    },
    {
      "action_loss_avg": 0.0012225467944517732,
      "epoch": 6.857872009277343e-05,
      "step": 71910
    },
    {
      "epoch": 6.857872009277343e-05,
      "step": 71910,
      "training_step_time": 7.162919044494629
    },
    {
      "epoch": 6.857967376708985e-05,
      "model_forward_time": 1.7327587604522705,
      "step": 71911
    },
    {
      "epoch": 6.857967376708985e-05,
      "step": 71911,
      "training_step_time": 6.8214006423950195
    },
    {
      "epoch": 6.858062744140626e-05,
      "model_forward_time": 1.7301716804504395,
      "step": 71912
    },
    {
      "epoch": 6.858062744140626e-05,
      "step": 71912,
      "training_step_time": 6.8169310092926025
    },
    {
      "epoch": 6.858158111572265e-05,
      "model_forward_time": 1.7661805152893066,
      "step": 71913
    },
    {
      "epoch": 6.858158111572265e-05,
      "step": 71913,
      "training_step_time": 7.681694269180298
    },
    {
      "epoch": 6.858253479003906e-05,
      "model_forward_time": 1.7379047870635986,
      "step": 71914
    },
    {
      "epoch": 6.858253479003906e-05,
      "step": 71914,
      "training_step_time": 7.642805099487305
    },
    {
      "epoch": 6.858348846435547e-05,
      "model_forward_time": 1.7165396213531494,
      "step": 71915
    },
    {
      "epoch": 6.858348846435547e-05,
      "step": 71915,
      "training_step_time": 7.760080814361572
    },
    {
      "epoch": 6.858444213867187e-05,
      "model_forward_time": 2.2986273765563965,
      "step": 71916
    },
    {
      "epoch": 6.858444213867187e-05,
      "step": 71916,
      "training_step_time": 9.772693872451782
    },
    {
      "epoch": 6.858539581298828e-05,
      "model_forward_time": 1.7326738834381104,
      "step": 71917
    },
    {
      "epoch": 6.858539581298828e-05,
      "step": 71917,
      "training_step_time": 7.718154430389404
    },
    {
      "epoch": 6.85863494873047e-05,
      "model_forward_time": 1.7491240501403809,
      "step": 71918
    },
    {
      "epoch": 6.85863494873047e-05,
      "step": 71918,
      "training_step_time": 7.6859283447265625
    },
    {
      "epoch": 6.858730316162109e-05,
      "model_forward_time": 1.734344720840454,
      "step": 71919
    },
    {
      "epoch": 6.858730316162109e-05,
      "step": 71919,
      "training_step_time": 7.145856618881226
    },
    {
      "epoch": 6.85882568359375e-05,
      "grad_norm": 0.2041241079568863,
      "learning_rate": 2.0053086418636102e-06,
      "loss": 0.0578,
      "step": 71920
    },
    {
      "epoch": 6.85882568359375e-05,
      "model_forward_time": 1.7319281101226807,
      "step": 71920
    },
    {
      "dynamics_loss_avg": 0.05631834883242846,
      "epoch": 6.85882568359375e-05,
      "step": 71920
    },
    {
      "action_loss_avg": 0.0013521976390620694,
      "epoch": 6.85882568359375e-05,
      "step": 71920
    },
    {
      "epoch": 6.85882568359375e-05,
      "step": 71920,
      "training_step_time": 6.791402578353882
    },
    {
      "epoch": 6.858921051025391e-05,
      "model_forward_time": 1.7331950664520264,
      "step": 71921
    },
    {
      "epoch": 6.858921051025391e-05,
      "step": 71921,
      "training_step_time": 6.842315196990967
    },
    {
      "epoch": 6.859016418457031e-05,
      "model_forward_time": 1.740455150604248,
      "step": 71922
    },
    {
      "epoch": 6.859016418457031e-05,
      "step": 71922,
      "training_step_time": 7.776292562484741
    },
    {
      "epoch": 6.859111785888672e-05,
      "model_forward_time": 1.7324447631835938,
      "step": 71923
    },
    {
      "epoch": 6.859111785888672e-05,
      "step": 71923,
      "training_step_time": 7.702918767929077
    },
    {
      "epoch": 6.859207153320313e-05,
      "model_forward_time": 1.724691390991211,
      "step": 71924
    },
    {
      "epoch": 6.859207153320313e-05,
      "step": 71924,
      "training_step_time": 8.423547506332397
    },
    {
      "epoch": 6.859302520751953e-05,
      "model_forward_time": 2.3224449157714844,
      "step": 71925
    },
    {
      "epoch": 6.859302520751953e-05,
      "step": 71925,
      "training_step_time": 8.620308637619019
    },
    {
      "epoch": 6.859397888183594e-05,
      "model_forward_time": 1.7278807163238525,
      "step": 71926
    },
    {
      "epoch": 6.859397888183594e-05,
      "step": 71926,
      "training_step_time": 8.783818244934082
    },
    {
      "epoch": 6.859493255615235e-05,
      "model_forward_time": 1.7167718410491943,
      "step": 71927
    },
    {
      "epoch": 6.859493255615235e-05,
      "step": 71927,
      "training_step_time": 7.738788366317749
    },
    {
      "epoch": 6.859588623046875e-05,
      "model_forward_time": 1.7129638195037842,
      "step": 71928
    },
    {
      "epoch": 6.859588623046875e-05,
      "step": 71928,
      "training_step_time": 7.300120830535889
    },
    {
      "epoch": 6.859683990478516e-05,
      "model_forward_time": 1.7113964557647705,
      "step": 71929
    },
    {
      "epoch": 6.859683990478516e-05,
      "step": 71929,
      "training_step_time": 6.835163593292236
    },
    {
      "epoch": 6.859779357910157e-05,
      "grad_norm": 0.090617835521698,
      "learning_rate": 2.003984714947618e-06,
      "loss": 0.0577,
      "step": 71930
    },
    {
      "epoch": 6.859779357910157e-05,
      "model_forward_time": 1.7079262733459473,
      "step": 71930
    },
    {
      "dynamics_loss_avg": 0.04744892511516809,
      "epoch": 6.859779357910157e-05,
      "step": 71930
    },
    {
      "action_loss_avg": 0.0008819309252430685,
      "epoch": 6.859779357910157e-05,
      "step": 71930
    },
    {
      "epoch": 6.859779357910157e-05,
      "step": 71930,
      "training_step_time": 6.808839559555054
    },
    {
      "epoch": 6.859874725341797e-05,
      "model_forward_time": 1.71417236328125,
      "step": 71931
    },
    {
      "epoch": 6.859874725341797e-05,
      "step": 71931,
      "training_step_time": 7.734967231750488
    },
    {
      "epoch": 6.859970092773438e-05,
      "model_forward_time": 1.7276852130889893,
      "step": 71932
    },
    {
      "epoch": 6.859970092773438e-05,
      "step": 71932,
      "training_step_time": 8.361512184143066
    },
    {
      "epoch": 6.860065460205079e-05,
      "model_forward_time": 1.716050148010254,
      "step": 71933
    },
    {
      "epoch": 6.860065460205079e-05,
      "step": 71933,
      "training_step_time": 7.7539660930633545
    },
    {
      "epoch": 6.860160827636719e-05,
      "model_forward_time": 2.267390727996826,
      "step": 71934
    },
    {
      "epoch": 6.860160827636719e-05,
      "step": 71934,
      "training_step_time": 9.482771873474121
    },
    {
      "epoch": 6.86025619506836e-05,
      "model_forward_time": 1.708350658416748,
      "step": 71935
    },
    {
      "epoch": 6.86025619506836e-05,
      "step": 71935,
      "training_step_time": 9.202719449996948
    },
    {
      "epoch": 6.8603515625e-05,
      "model_forward_time": 1.705467700958252,
      "step": 71936
    },
    {
      "epoch": 6.8603515625e-05,
      "step": 71936,
      "training_step_time": 7.766620635986328
    },
    {
      "epoch": 6.86044692993164e-05,
      "model_forward_time": 1.6983609199523926,
      "step": 71937
    },
    {
      "epoch": 6.86044692993164e-05,
      "step": 71937,
      "training_step_time": 7.440664768218994
    },
    {
      "epoch": 6.860542297363281e-05,
      "model_forward_time": 1.7197012901306152,
      "step": 71938
    },
    {
      "epoch": 6.860542297363281e-05,
      "step": 71938,
      "training_step_time": 6.840050220489502
    },
    {
      "epoch": 6.860637664794921e-05,
      "model_forward_time": 1.7283096313476562,
      "step": 71939
    },
    {
      "epoch": 6.860637664794921e-05,
      "step": 71939,
      "training_step_time": 6.842606067657471
    },
    {
      "epoch": 6.860733032226562e-05,
      "grad_norm": 0.08867918699979782,
      "learning_rate": 2.0026611156713515e-06,
      "loss": 0.054,
      "step": 71940
    },
    {
      "epoch": 6.860733032226562e-05,
      "model_forward_time": 1.7191967964172363,
      "step": 71940
    },
    {
      "dynamics_loss_avg": 0.043865359760820864,
      "epoch": 6.860733032226562e-05,
      "step": 71940
    },
    {
      "action_loss_avg": 0.001827909582061693,
      "epoch": 6.860733032226562e-05,
      "step": 71940
    },
    {
      "epoch": 6.860733032226562e-05,
      "step": 71940,
      "training_step_time": 7.746347427368164
    },
    {
      "epoch": 6.860828399658203e-05,
      "model_forward_time": 1.7741360664367676,
      "step": 71941
    },
    {
      "epoch": 6.860828399658203e-05,
      "step": 71941,
      "training_step_time": 7.706030607223511
    },
    {
      "epoch": 6.860923767089843e-05,
      "model_forward_time": 1.7163677215576172,
      "step": 71942
    },
    {
      "epoch": 6.860923767089843e-05,
      "step": 71942,
      "training_step_time": 7.707471132278442
    },
    {
      "epoch": 6.861019134521484e-05,
      "model_forward_time": 2.3068490028381348,
      "step": 71943
    },
    {
      "epoch": 6.861019134521484e-05,
      "step": 71943,
      "training_step_time": 10.10542631149292
    },
    {
      "epoch": 6.861114501953125e-05,
      "model_forward_time": 1.7154006958007812,
      "step": 71944
    },
    {
      "epoch": 6.861114501953125e-05,
      "step": 71944,
      "training_step_time": 8.355963230133057
    },
    {
      "epoch": 6.861209869384765e-05,
      "model_forward_time": 1.7239370346069336,
      "step": 71945
    },
    {
      "epoch": 6.861209869384765e-05,
      "step": 71945,
      "training_step_time": 7.639408349990845
    },
    {
      "epoch": 6.861305236816406e-05,
      "model_forward_time": 1.7210373878479004,
      "step": 71946
    },
    {
      "epoch": 6.861305236816406e-05,
      "step": 71946,
      "training_step_time": 7.746757745742798
    },
    {
      "epoch": 6.861400604248047e-05,
      "model_forward_time": 1.7195429801940918,
      "step": 71947
    },
    {
      "epoch": 6.861400604248047e-05,
      "step": 71947,
      "training_step_time": 6.811025142669678
    },
    {
      "epoch": 6.861495971679687e-05,
      "model_forward_time": 1.7234175205230713,
      "step": 71948
    },
    {
      "epoch": 6.861495971679687e-05,
      "step": 71948,
      "training_step_time": 6.784029245376587
    },
    {
      "epoch": 6.861591339111328e-05,
      "model_forward_time": 1.7116167545318604,
      "step": 71949
    },
    {
      "epoch": 6.861591339111328e-05,
      "step": 71949,
      "training_step_time": 7.718686819076538
    },
    {
      "epoch": 6.861686706542969e-05,
      "grad_norm": 0.08338435739278793,
      "learning_rate": 2.001337844179558e-06,
      "loss": 0.0538,
      "step": 71950
    },
    {
      "epoch": 6.861686706542969e-05,
      "model_forward_time": 1.7574849128723145,
      "step": 71950
    },
    {
      "dynamics_loss_avg": 0.06839221809059381,
      "epoch": 6.861686706542969e-05,
      "step": 71950
    },
    {
      "action_loss_avg": 0.0022452773526310922,
      "epoch": 6.861686706542969e-05,
      "step": 71950
    },
    {
      "epoch": 6.861686706542969e-05,
      "step": 71950,
      "training_step_time": 7.688809633255005
    },
    {
      "epoch": 6.861782073974609e-05,
      "model_forward_time": 1.7283174991607666,
      "step": 71951
    },
    {
      "epoch": 6.861782073974609e-05,
      "step": 71951,
      "training_step_time": 7.648581504821777
    },
    {
      "epoch": 6.86187744140625e-05,
      "model_forward_time": 2.2641851902008057,
      "step": 71952
    },
    {
      "epoch": 6.86187744140625e-05,
      "step": 71952,
      "training_step_time": 8.978883743286133
    },
    {
      "epoch": 6.861972808837891e-05,
      "model_forward_time": 1.7235956192016602,
      "step": 71953
    },
    {
      "epoch": 6.861972808837891e-05,
      "step": 71953,
      "training_step_time": 9.324343204498291
    },
    {
      "epoch": 6.86206817626953e-05,
      "model_forward_time": 1.7251920700073242,
      "step": 71954
    },
    {
      "epoch": 6.86206817626953e-05,
      "step": 71954,
      "training_step_time": 7.762763261795044
    },
    {
      "epoch": 6.862163543701172e-05,
      "model_forward_time": 1.7143189907073975,
      "step": 71955
    },
    {
      "epoch": 6.862163543701172e-05,
      "step": 71955,
      "training_step_time": 7.7709736824035645
    },
    {
      "epoch": 6.862258911132813e-05,
      "model_forward_time": 1.72337007522583,
      "step": 71956
    },
    {
      "epoch": 6.862258911132813e-05,
      "step": 71956,
      "training_step_time": 6.8446667194366455
    },
    {
      "epoch": 6.862354278564453e-05,
      "model_forward_time": 1.713014841079712,
      "step": 71957
    },
    {
      "epoch": 6.862354278564453e-05,
      "step": 71957,
      "training_step_time": 6.759843349456787
    },
    {
      "epoch": 6.862449645996094e-05,
      "model_forward_time": 1.7140181064605713,
      "step": 71958
    },
    {
      "epoch": 6.862449645996094e-05,
      "step": 71958,
      "training_step_time": 7.715093374252319
    },
    {
      "epoch": 6.862545013427735e-05,
      "model_forward_time": 1.742586612701416,
      "step": 71959
    },
    {
      "epoch": 6.862545013427735e-05,
      "step": 71959,
      "training_step_time": 7.726501941680908
    },
    {
      "epoch": 6.862640380859374e-05,
      "grad_norm": 0.07585421949625015,
      "learning_rate": 2.0000149006169506e-06,
      "loss": 0.0559,
      "step": 71960
    },
    {
      "epoch": 6.862640380859374e-05,
      "model_forward_time": 1.7167291641235352,
      "step": 71960
    },
    {
      "dynamics_loss_avg": 0.08076129760593176,
      "epoch": 6.862640380859374e-05,
      "step": 71960
    },
    {
      "action_loss_avg": 0.0025710242509376256,
      "epoch": 6.862640380859374e-05,
      "step": 71960
    },
    {
      "epoch": 6.862640380859374e-05,
      "step": 71960,
      "training_step_time": 7.601163625717163
    },
    {
      "epoch": 6.862735748291015e-05,
      "model_forward_time": 1.708482027053833,
      "step": 71961
    },
    {
      "epoch": 6.862735748291015e-05,
      "step": 71961,
      "training_step_time": 9.397468328475952
    },
    {
      "epoch": 6.862831115722657e-05,
      "model_forward_time": 2.2730114459991455,
      "step": 71962
    },
    {
      "epoch": 6.862831115722657e-05,
      "step": 71962,
      "training_step_time": 7.768223762512207
    },
    {
      "epoch": 6.862926483154296e-05,
      "model_forward_time": 1.7107908725738525,
      "step": 71963
    },
    {
      "epoch": 6.862926483154296e-05,
      "step": 71963,
      "training_step_time": 7.7942705154418945
    },
    {
      "epoch": 6.863021850585937e-05,
      "model_forward_time": 1.70914626121521,
      "step": 71964
    },
    {
      "epoch": 6.863021850585937e-05,
      "step": 71964,
      "training_step_time": 7.644329309463501
    },
    {
      "epoch": 6.863117218017578e-05,
      "model_forward_time": 1.7115507125854492,
      "step": 71965
    },
    {
      "epoch": 6.863117218017578e-05,
      "step": 71965,
      "training_step_time": 7.245835304260254
    },
    {
      "epoch": 6.863212585449218e-05,
      "model_forward_time": 1.7201333045959473,
      "step": 71966
    },
    {
      "epoch": 6.863212585449218e-05,
      "step": 71966,
      "training_step_time": 6.758450746536255
    },
    {
      "epoch": 6.863307952880859e-05,
      "model_forward_time": 1.7124569416046143,
      "step": 71967
    },
    {
      "epoch": 6.863307952880859e-05,
      "step": 71967,
      "training_step_time": 8.218057870864868
    },
    {
      "epoch": 6.8634033203125e-05,
      "model_forward_time": 1.7233233451843262,
      "step": 71968
    },
    {
      "epoch": 6.8634033203125e-05,
      "step": 71968,
      "training_step_time": 7.59618616104126
    },
    {
      "epoch": 6.86349868774414e-05,
      "model_forward_time": 1.727442741394043,
      "step": 71969
    },
    {
      "epoch": 6.86349868774414e-05,
      "step": 71969,
      "training_step_time": 7.647199630737305
    },
    {
      "epoch": 6.863594055175781e-05,
      "grad_norm": 0.07359392940998077,
      "learning_rate": 1.9986922851282047e-06,
      "loss": 0.0568,
      "step": 71970
    },
    {
      "epoch": 6.863594055175781e-05,
      "model_forward_time": 1.3553757667541504,
      "step": 71970
    },
    {
      "dynamics_loss_avg": 0.051652928628027436,
      "epoch": 6.863594055175781e-05,
      "step": 71970
    },
    {
      "action_loss_avg": 0.001355120530934073,
      "epoch": 6.863594055175781e-05,
      "step": 71970
    },
    {
      "epoch": 6.863594055175781e-05,
      "step": 71970,
      "training_step_time": 10.804883241653442
    },
    {
      "epoch": 6.863689422607422e-05,
      "model_forward_time": 2.2938811779022217,
      "step": 71971
    },
    {
      "epoch": 6.863689422607422e-05,
      "step": 71971,
      "training_step_time": 10.464872121810913
    },
    {
      "epoch": 6.863784790039062e-05,
      "model_forward_time": 1.7296452522277832,
      "step": 71972
    },
    {
      "epoch": 6.863784790039062e-05,
      "step": 71972,
      "training_step_time": 9.26991891860962
    },
    {
      "epoch": 6.863880157470703e-05,
      "model_forward_time": 1.7295098304748535,
      "step": 71973
    },
    {
      "epoch": 6.863880157470703e-05,
      "step": 71973,
      "training_step_time": 8.237732410430908
    },
    {
      "epoch": 6.863975524902344e-05,
      "model_forward_time": 1.7249929904937744,
      "step": 71974
    },
    {
      "epoch": 6.863975524902344e-05,
      "step": 71974,
      "training_step_time": 7.282581567764282
    },
    {
      "epoch": 6.864070892333984e-05,
      "model_forward_time": 1.730381965637207,
      "step": 71975
    },
    {
      "epoch": 6.864070892333984e-05,
      "step": 71975,
      "training_step_time": 6.78450083732605
    },
    {
      "epoch": 6.864166259765625e-05,
      "model_forward_time": 1.7207765579223633,
      "step": 71976
    },
    {
      "epoch": 6.864166259765625e-05,
      "step": 71976,
      "training_step_time": 7.738513946533203
    },
    {
      "epoch": 6.864261627197266e-05,
      "model_forward_time": 1.7209851741790771,
      "step": 71977
    },
    {
      "epoch": 6.864261627197266e-05,
      "step": 71977,
      "training_step_time": 7.346471309661865
    },
    {
      "epoch": 6.864356994628906e-05,
      "model_forward_time": 1.7859554290771484,
      "step": 71978
    },
    {
      "epoch": 6.864356994628906e-05,
      "step": 71978,
      "training_step_time": 7.672773122787476
    },
    {
      "epoch": 6.864452362060547e-05,
      "model_forward_time": 1.718310832977295,
      "step": 71979
    },
    {
      "epoch": 6.864452362060547e-05,
      "step": 71979,
      "training_step_time": 9.10949158668518
    },
    {
      "epoch": 6.864547729492188e-05,
      "grad_norm": 0.09044267982244492,
      "learning_rate": 1.9973699978579563e-06,
      "loss": 0.0541,
      "step": 71980
    },
    {
      "epoch": 6.864547729492188e-05,
      "model_forward_time": 2.2981293201446533,
      "step": 71980
    },
    {
      "dynamics_loss_avg": 0.05944126658141613,
      "epoch": 6.864547729492188e-05,
      "step": 71980
    },
    {
      "action_loss_avg": 0.0018013868742855267,
      "epoch": 6.864547729492188e-05,
      "step": 71980
    },
    {
      "epoch": 6.864547729492188e-05,
      "step": 71980,
      "training_step_time": 10.116122484207153
    },
    {
      "epoch": 6.864643096923828e-05,
      "model_forward_time": 1.7225217819213867,
      "step": 71981
    },
    {
      "epoch": 6.864643096923828e-05,
      "step": 71981,
      "training_step_time": 9.345570087432861
    },
    {
      "epoch": 6.864738464355469e-05,
      "model_forward_time": 1.719423770904541,
      "step": 71982
    },
    {
      "epoch": 6.864738464355469e-05,
      "step": 71982,
      "training_step_time": 7.768159866333008
    },
    {
      "epoch": 6.86483383178711e-05,
      "model_forward_time": 1.7167303562164307,
      "step": 71983
    },
    {
      "epoch": 6.86483383178711e-05,
      "step": 71983,
      "training_step_time": 7.299042463302612
    },
    {
      "epoch": 6.86492919921875e-05,
      "model_forward_time": 1.7236993312835693,
      "step": 71984
    },
    {
      "epoch": 6.86492919921875e-05,
      "step": 71984,
      "training_step_time": 6.826392889022827
    },
    {
      "epoch": 6.86502456665039e-05,
      "model_forward_time": 1.7527692317962646,
      "step": 71985
    },
    {
      "epoch": 6.86502456665039e-05,
      "step": 71985,
      "training_step_time": 7.706680059432983
    },
    {
      "epoch": 6.865119934082032e-05,
      "model_forward_time": 1.72538161277771,
      "step": 71986
    },
    {
      "epoch": 6.865119934082032e-05,
      "step": 71986,
      "training_step_time": 7.401109933853149
    },
    {
      "epoch": 6.865215301513671e-05,
      "model_forward_time": 1.783372163772583,
      "step": 71987
    },
    {
      "epoch": 6.865215301513671e-05,
      "step": 71987,
      "training_step_time": 7.698862791061401
    },
    {
      "epoch": 6.865310668945312e-05,
      "model_forward_time": 1.7260615825653076,
      "step": 71988
    },
    {
      "epoch": 6.865310668945312e-05,
      "step": 71988,
      "training_step_time": 9.302502393722534
    },
    {
      "epoch": 6.865406036376954e-05,
      "model_forward_time": 2.317232370376587,
      "step": 71989
    },
    {
      "epoch": 6.865406036376954e-05,
      "step": 71989,
      "training_step_time": 9.469696283340454
    },
    {
      "epoch": 6.865501403808593e-05,
      "grad_norm": 0.13462437689304352,
      "learning_rate": 1.99604803895081e-06,
      "loss": 0.056,
      "step": 71990
    },
    {
      "epoch": 6.865501403808593e-05,
      "model_forward_time": 1.722419023513794,
      "step": 71990
    },
    {
      "dynamics_loss_avg": 0.06084382329136133,
      "epoch": 6.865501403808593e-05,
      "step": 71990
    },
    {
      "action_loss_avg": 0.0012619898363482207,
      "epoch": 6.865501403808593e-05,
      "step": 71990
    },
    {
      "epoch": 6.865501403808593e-05,
      "step": 71990,
      "training_step_time": 9.11162805557251
    },
    {
      "epoch": 6.865596771240234e-05,
      "model_forward_time": 1.7265076637268066,
      "step": 71991
    },
    {
      "epoch": 6.865596771240234e-05,
      "step": 71991,
      "training_step_time": 8.62028694152832
    },
    {
      "epoch": 6.865692138671875e-05,
      "model_forward_time": 1.3627405166625977,
      "step": 71992
    },
    {
      "epoch": 6.865692138671875e-05,
      "step": 71992,
      "training_step_time": 8.391604900360107
    },
    {
      "epoch": 6.865787506103515e-05,
      "model_forward_time": 1.7088172435760498,
      "step": 71993
    },
    {
      "epoch": 6.865787506103515e-05,
      "step": 71993,
      "training_step_time": 7.202463626861572
    },
    {
      "epoch": 6.865882873535156e-05,
      "model_forward_time": 1.716703176498413,
      "step": 71994
    },
    {
      "epoch": 6.865882873535156e-05,
      "step": 71994,
      "training_step_time": 7.684711456298828
    },
    {
      "epoch": 6.865978240966797e-05,
      "model_forward_time": 1.7222301959991455,
      "step": 71995
    },
    {
      "epoch": 6.865978240966797e-05,
      "step": 71995,
      "training_step_time": 7.35744833946228
    },
    {
      "epoch": 6.866073608398437e-05,
      "model_forward_time": 1.7689924240112305,
      "step": 71996
    },
    {
      "epoch": 6.866073608398437e-05,
      "step": 71996,
      "training_step_time": 7.696342468261719
    },
    {
      "epoch": 6.866168975830078e-05,
      "model_forward_time": 1.7238974571228027,
      "step": 71997
    },
    {
      "epoch": 6.866168975830078e-05,
      "step": 71997,
      "training_step_time": 7.631901979446411
    },
    {
      "epoch": 6.866264343261719e-05,
      "model_forward_time": 2.3223578929901123,
      "step": 71998
    },
    {
      "epoch": 6.866264343261719e-05,
      "step": 71998,
      "training_step_time": 7.98312783241272
    },
    {
      "epoch": 6.866359710693359e-05,
      "model_forward_time": 1.7366681098937988,
      "step": 71999
    },
    {
      "epoch": 6.866359710693359e-05,
      "step": 71999,
      "training_step_time": 7.733346223831177
    },
    {
      "epoch": 6.866455078125e-05,
      "grad_norm": 0.10812176018953323,
      "learning_rate": 1.994726408551336e-06,
      "loss": 0.0529,
      "step": 72000
    },
    {
      "epoch": 6.866455078125e-05,
      "model_forward_time": 1.7702841758728027,
      "step": 72000
    },
    {
      "dynamics_loss_avg": 0.04426100812852383,
      "epoch": 6.866455078125e-05,
      "step": 72000
    },
    {
      "action_loss_avg": 0.002209446023334749,
      "epoch": 6.866455078125e-05,
      "step": 72000
    },
    {
      "epoch": 6.866455078125e-05,
      "step": 72000,
      "training_step_time": 10.21438455581665
    },
    {
      "epoch": 6.866550445556641e-05,
      "model_forward_time": 1.744610071182251,
      "step": 72001
    },
    {
      "epoch": 6.866550445556641e-05,
      "step": 72001,
      "training_step_time": 7.386176109313965
    },
    {
      "epoch": 6.866645812988281e-05,
      "model_forward_time": 1.7498927116394043,
      "step": 72002
    },
    {
      "epoch": 6.866645812988281e-05,
      "step": 72002,
      "training_step_time": 7.352047443389893
    },
    {
      "epoch": 6.866741180419922e-05,
      "model_forward_time": 1.7502501010894775,
      "step": 72003
    },
    {
      "epoch": 6.866741180419922e-05,
      "step": 72003,
      "training_step_time": 7.265620231628418
    },
    {
      "epoch": 6.866836547851563e-05,
      "model_forward_time": 1.7485275268554688,
      "step": 72004
    },
    {
      "epoch": 6.866836547851563e-05,
      "step": 72004,
      "training_step_time": 7.882969617843628
    },
    {
      "epoch": 6.866931915283203e-05,
      "model_forward_time": 1.8031957149505615,
      "step": 72005
    },
    {
      "epoch": 6.866931915283203e-05,
      "step": 72005,
      "training_step_time": 7.707546949386597
    },
    {
      "epoch": 6.867027282714844e-05,
      "model_forward_time": 1.7514395713806152,
      "step": 72006
    },
    {
      "epoch": 6.867027282714844e-05,
      "step": 72006,
      "training_step_time": 8.571781873703003
    },
    {
      "epoch": 6.867122650146485e-05,
      "model_forward_time": 2.752166271209717,
      "step": 72007
    },
    {
      "epoch": 6.867122650146485e-05,
      "step": 72007,
      "training_step_time": 12.360681533813477
    },
    {
      "epoch": 6.867218017578125e-05,
      "model_forward_time": 1.7360451221466064,
      "step": 72008
    },
    {
      "epoch": 6.867218017578125e-05,
      "step": 72008,
      "training_step_time": 10.61771297454834
    },
    {
      "epoch": 6.867313385009766e-05,
      "model_forward_time": 1.739262342453003,
      "step": 72009
    },
    {
      "epoch": 6.867313385009766e-05,
      "step": 72009,
      "training_step_time": 7.658975839614868
    },
    {
      "epoch": 6.867408752441407e-05,
      "grad_norm": 0.08124130964279175,
      "learning_rate": 1.993405106804062e-06,
      "loss": 0.0517,
      "step": 72010
    },
    {
      "epoch": 6.867408752441407e-05,
      "model_forward_time": 1.3539936542510986,
      "step": 72010
    },
    {
      "dynamics_loss_avg": 0.04461125684902072,
      "epoch": 6.867408752441407e-05,
      "step": 72010
    },
    {
      "action_loss_avg": 0.0018821614619810134,
      "epoch": 6.867408752441407e-05,
      "step": 72010
    },
    {
      "epoch": 6.867408752441407e-05,
      "step": 72010,
      "training_step_time": 8.338720798492432
    },
    {
      "epoch": 6.867504119873046e-05,
      "model_forward_time": 1.7359709739685059,
      "step": 72011
    },
    {
      "epoch": 6.867504119873046e-05,
      "step": 72011,
      "training_step_time": 7.57038426399231
    },
    {
      "epoch": 6.867599487304688e-05,
      "model_forward_time": 1.7335035800933838,
      "step": 72012
    },
    {
      "epoch": 6.867599487304688e-05,
      "step": 72012,
      "training_step_time": 7.291321754455566
    },
    {
      "epoch": 6.867694854736329e-05,
      "model_forward_time": 1.7325046062469482,
      "step": 72013
    },
    {
      "epoch": 6.867694854736329e-05,
      "step": 72013,
      "training_step_time": 7.832167625427246
    },
    {
      "epoch": 6.867790222167968e-05,
      "model_forward_time": 1.758775234222412,
      "step": 72014
    },
    {
      "epoch": 6.867790222167968e-05,
      "step": 72014,
      "training_step_time": 7.674824237823486
    },
    {
      "epoch": 6.86788558959961e-05,
      "model_forward_time": 1.8363614082336426,
      "step": 72015
    },
    {
      "epoch": 6.86788558959961e-05,
      "step": 72015,
      "training_step_time": 8.716874122619629
    },
    {
      "epoch": 6.86798095703125e-05,
      "model_forward_time": 2.386676073074341,
      "step": 72016
    },
    {
      "epoch": 6.86798095703125e-05,
      "step": 72016,
      "training_step_time": 10.222800016403198
    },
    {
      "epoch": 6.86807632446289e-05,
      "model_forward_time": 1.8429234027862549,
      "step": 72017
    },
    {
      "epoch": 6.86807632446289e-05,
      "step": 72017,
      "training_step_time": 8.346160411834717
    },
    {
      "epoch": 6.868171691894531e-05,
      "model_forward_time": 1.8430988788604736,
      "step": 72018
    },
    {
      "epoch": 6.868171691894531e-05,
      "step": 72018,
      "training_step_time": 7.663082599639893
    },
    {
      "epoch": 6.868267059326172e-05,
      "model_forward_time": 1.7705469131469727,
      "step": 72019
    },
    {
      "epoch": 6.868267059326172e-05,
      "step": 72019,
      "training_step_time": 7.764930725097656
    },
    {
      "epoch": 6.868362426757812e-05,
      "grad_norm": 0.10860062390565872,
      "learning_rate": 1.992084133853487e-06,
      "loss": 0.0555,
      "step": 72020
    },
    {
      "epoch": 6.868362426757812e-05,
      "model_forward_time": 1.73944091796875,
      "step": 72020
    },
    {
      "dynamics_loss_avg": 0.05161565281450749,
      "epoch": 6.868362426757812e-05,
      "step": 72020
    },
    {
      "action_loss_avg": 0.0016720707819331438,
      "epoch": 6.868362426757812e-05,
      "step": 72020
    },
    {
      "epoch": 6.868362426757812e-05,
      "step": 72020,
      "training_step_time": 7.4569618701934814
    },
    {
      "epoch": 6.868457794189453e-05,
      "model_forward_time": 1.7341160774230957,
      "step": 72021
    },
    {
      "epoch": 6.868457794189453e-05,
      "step": 72021,
      "training_step_time": 7.666293144226074
    },
    {
      "epoch": 6.868553161621094e-05,
      "model_forward_time": 1.7424664497375488,
      "step": 72022
    },
    {
      "epoch": 6.868553161621094e-05,
      "step": 72022,
      "training_step_time": 6.775458812713623
    },
    {
      "epoch": 6.868648529052734e-05,
      "model_forward_time": 1.7340881824493408,
      "step": 72023
    },
    {
      "epoch": 6.868648529052734e-05,
      "step": 72023,
      "training_step_time": 7.81801700592041
    },
    {
      "epoch": 6.868743896484375e-05,
      "model_forward_time": 1.7374424934387207,
      "step": 72024
    },
    {
      "epoch": 6.868743896484375e-05,
      "step": 72024,
      "training_step_time": 7.709375858306885
    },
    {
      "epoch": 6.868839263916016e-05,
      "model_forward_time": 1.7608582973480225,
      "step": 72025
    },
    {
      "epoch": 6.868839263916016e-05,
      "step": 72025,
      "training_step_time": 7.86430549621582
    },
    {
      "epoch": 6.868934631347656e-05,
      "model_forward_time": 2.3124196529388428,
      "step": 72026
    },
    {
      "epoch": 6.868934631347656e-05,
      "step": 72026,
      "training_step_time": 9.752774238586426
    },
    {
      "epoch": 6.869029998779297e-05,
      "model_forward_time": 1.7409837245941162,
      "step": 72027
    },
    {
      "epoch": 6.869029998779297e-05,
      "step": 72027,
      "training_step_time": 7.717716455459595
    },
    {
      "epoch": 6.869125366210938e-05,
      "model_forward_time": 1.7424304485321045,
      "step": 72028
    },
    {
      "epoch": 6.869125366210938e-05,
      "step": 72028,
      "training_step_time": 8.290651321411133
    },
    {
      "epoch": 6.869220733642578e-05,
      "model_forward_time": 1.7470664978027344,
      "step": 72029
    },
    {
      "epoch": 6.869220733642578e-05,
      "step": 72029,
      "training_step_time": 7.7698259353637695
    },
    {
      "epoch": 6.869316101074219e-05,
      "grad_norm": 0.15161356329917908,
      "learning_rate": 1.9907634898440646e-06,
      "loss": 0.0534,
      "step": 72030
    },
    {
      "epoch": 6.869316101074219e-05,
      "model_forward_time": 1.7515957355499268,
      "step": 72030
    },
    {
      "dynamics_loss_avg": 0.04351335950195789,
      "epoch": 6.869316101074219e-05,
      "step": 72030
    },
    {
      "action_loss_avg": 0.001882256957469508,
      "epoch": 6.869316101074219e-05,
      "step": 72030
    },
    {
      "epoch": 6.869316101074219e-05,
      "step": 72030,
      "training_step_time": 7.733954429626465
    },
    {
      "epoch": 6.86941146850586e-05,
      "model_forward_time": 1.7463250160217285,
      "step": 72031
    },
    {
      "epoch": 6.86941146850586e-05,
      "step": 72031,
      "training_step_time": 6.771644353866577
    },
    {
      "epoch": 6.8695068359375e-05,
      "model_forward_time": 1.7305049896240234,
      "step": 72032
    },
    {
      "epoch": 6.8695068359375e-05,
      "step": 72032,
      "training_step_time": 7.874737977981567
    },
    {
      "epoch": 6.869602203369141e-05,
      "model_forward_time": 1.7726356983184814,
      "step": 72033
    },
    {
      "epoch": 6.869602203369141e-05,
      "step": 72033,
      "training_step_time": 8.385612487792969
    },
    {
      "epoch": 6.869697570800782e-05,
      "model_forward_time": 1.7425651550292969,
      "step": 72034
    },
    {
      "epoch": 6.869697570800782e-05,
      "step": 72034,
      "training_step_time": 7.854075193405151
    },
    {
      "epoch": 6.869792938232422e-05,
      "model_forward_time": 2.3055920600891113,
      "step": 72035
    },
    {
      "epoch": 6.869792938232422e-05,
      "step": 72035,
      "training_step_time": 9.34306526184082
    },
    {
      "epoch": 6.869888305664063e-05,
      "model_forward_time": 1.7434632778167725,
      "step": 72036
    },
    {
      "epoch": 6.869888305664063e-05,
      "step": 72036,
      "training_step_time": 9.010687351226807
    },
    {
      "epoch": 6.869983673095704e-05,
      "model_forward_time": 1.7246346473693848,
      "step": 72037
    },
    {
      "epoch": 6.869983673095704e-05,
      "step": 72037,
      "training_step_time": 7.721209764480591
    },
    {
      "epoch": 6.870079040527343e-05,
      "model_forward_time": 1.7265055179595947,
      "step": 72038
    },
    {
      "epoch": 6.870079040527343e-05,
      "step": 72038,
      "training_step_time": 8.565251111984253
    },
    {
      "epoch": 6.870174407958985e-05,
      "model_forward_time": 1.7253966331481934,
      "step": 72039
    },
    {
      "epoch": 6.870174407958985e-05,
      "step": 72039,
      "training_step_time": 7.647632122039795
    },
    {
      "epoch": 6.870269775390626e-05,
      "grad_norm": 0.08001337200403214,
      "learning_rate": 1.989443174920227e-06,
      "loss": 0.0541,
      "step": 72040
    },
    {
      "epoch": 6.870269775390626e-05,
      "model_forward_time": 1.7266011238098145,
      "step": 72040
    },
    {
      "dynamics_loss_avg": 0.06816965769976377,
      "epoch": 6.870269775390626e-05,
      "step": 72040
    },
    {
      "action_loss_avg": 0.0027259455120656638,
      "epoch": 6.870269775390626e-05,
      "step": 72040
    },
    {
      "epoch": 6.870269775390626e-05,
      "step": 72040,
      "training_step_time": 6.7528276443481445
    },
    {
      "epoch": 6.870365142822265e-05,
      "model_forward_time": 1.7385718822479248,
      "step": 72041
    },
    {
      "epoch": 6.870365142822265e-05,
      "step": 72041,
      "training_step_time": 7.880005598068237
    },
    {
      "epoch": 6.870460510253906e-05,
      "model_forward_time": 1.7195568084716797,
      "step": 72042
    },
    {
      "epoch": 6.870460510253906e-05,
      "step": 72042,
      "training_step_time": 7.774819374084473
    },
    {
      "epoch": 6.870555877685547e-05,
      "model_forward_time": 1.7279112339019775,
      "step": 72043
    },
    {
      "epoch": 6.870555877685547e-05,
      "step": 72043,
      "training_step_time": 8.567851066589355
    },
    {
      "epoch": 6.870651245117187e-05,
      "model_forward_time": 2.328899621963501,
      "step": 72044
    },
    {
      "epoch": 6.870651245117187e-05,
      "step": 72044,
      "training_step_time": 9.250978231430054
    },
    {
      "epoch": 6.870746612548828e-05,
      "model_forward_time": 1.7276532649993896,
      "step": 72045
    },
    {
      "epoch": 6.870746612548828e-05,
      "step": 72045,
      "training_step_time": 7.927161455154419
    },
    {
      "epoch": 6.87084197998047e-05,
      "model_forward_time": 1.7274608612060547,
      "step": 72046
    },
    {
      "epoch": 6.87084197998047e-05,
      "step": 72046,
      "training_step_time": 9.26089096069336
    },
    {
      "epoch": 6.870937347412109e-05,
      "model_forward_time": 1.7283267974853516,
      "step": 72047
    },
    {
      "epoch": 6.870937347412109e-05,
      "step": 72047,
      "training_step_time": 7.689731597900391
    },
    {
      "epoch": 6.87103271484375e-05,
      "model_forward_time": 1.7212047576904297,
      "step": 72048
    },
    {
      "epoch": 6.87103271484375e-05,
      "step": 72048,
      "training_step_time": 8.564903259277344
    },
    {
      "epoch": 6.871128082275391e-05,
      "model_forward_time": 1.7284765243530273,
      "step": 72049
    },
    {
      "epoch": 6.871128082275391e-05,
      "step": 72049,
      "training_step_time": 7.6814799308776855
    },
    {
      "epoch": 6.871223449707031e-05,
      "grad_norm": 0.12794944643974304,
      "learning_rate": 1.9881231892263554e-06,
      "loss": 0.058,
      "step": 72050
    },
    {
      "epoch": 6.871223449707031e-05,
      "model_forward_time": 1.7228384017944336,
      "step": 72050
    },
    {
      "dynamics_loss_avg": 0.07135703898966313,
      "epoch": 6.871223449707031e-05,
      "step": 72050
    },
    {
      "action_loss_avg": 0.001445738348411396,
      "epoch": 6.871223449707031e-05,
      "step": 72050
    },
    {
      "epoch": 6.871223449707031e-05,
      "step": 72050,
      "training_step_time": 8.686598777770996
    },
    {
      "epoch": 6.871318817138672e-05,
      "model_forward_time": 1.719261646270752,
      "step": 72051
    },
    {
      "epoch": 6.871318817138672e-05,
      "step": 72051,
      "training_step_time": 7.656942367553711
    },
    {
      "epoch": 6.871414184570313e-05,
      "model_forward_time": 1.7227239608764648,
      "step": 72052
    },
    {
      "epoch": 6.871414184570313e-05,
      "step": 72052,
      "training_step_time": 8.740585565567017
    },
    {
      "epoch": 6.871509552001953e-05,
      "model_forward_time": 2.332036018371582,
      "step": 72053
    },
    {
      "epoch": 6.871509552001953e-05,
      "step": 72053,
      "training_step_time": 7.960806131362915
    },
    {
      "epoch": 6.871604919433594e-05,
      "model_forward_time": 1.719674825668335,
      "step": 72054
    },
    {
      "epoch": 6.871604919433594e-05,
      "step": 72054,
      "training_step_time": 7.770944595336914
    },
    {
      "epoch": 6.871700286865235e-05,
      "model_forward_time": 1.7284088134765625,
      "step": 72055
    },
    {
      "epoch": 6.871700286865235e-05,
      "step": 72055,
      "training_step_time": 8.262603282928467
    },
    {
      "epoch": 6.871795654296875e-05,
      "model_forward_time": 1.7296428680419922,
      "step": 72056
    },
    {
      "epoch": 6.871795654296875e-05,
      "step": 72056,
      "training_step_time": 7.842684030532837
    },
    {
      "epoch": 6.871891021728516e-05,
      "model_forward_time": 1.7350006103515625,
      "step": 72057
    },
    {
      "epoch": 6.871891021728516e-05,
      "step": 72057,
      "training_step_time": 7.565768718719482
    },
    {
      "epoch": 6.871986389160157e-05,
      "model_forward_time": 1.7505261898040771,
      "step": 72058
    },
    {
      "epoch": 6.871986389160157e-05,
      "step": 72058,
      "training_step_time": 7.768927812576294
    },
    {
      "epoch": 6.872081756591797e-05,
      "model_forward_time": 1.736619472503662,
      "step": 72059
    },
    {
      "epoch": 6.872081756591797e-05,
      "step": 72059,
      "training_step_time": 7.822005987167358
    },
    {
      "epoch": 6.872177124023438e-05,
      "grad_norm": 0.10769903659820557,
      "learning_rate": 1.9868035329068054e-06,
      "loss": 0.0555,
      "step": 72060
    },
    {
      "epoch": 6.872177124023438e-05,
      "model_forward_time": 1.7249550819396973,
      "step": 72060
    },
    {
      "dynamics_loss_avg": 0.056491083838045596,
      "epoch": 6.872177124023438e-05,
      "step": 72060
    },
    {
      "action_loss_avg": 0.0012337846594164148,
      "epoch": 6.872177124023438e-05,
      "step": 72060
    },
    {
      "epoch": 6.872177124023438e-05,
      "step": 72060,
      "training_step_time": 8.644095420837402
    },
    {
      "epoch": 6.872272491455079e-05,
      "model_forward_time": 1.7428905963897705,
      "step": 72061
    },
    {
      "epoch": 6.872272491455079e-05,
      "step": 72061,
      "training_step_time": 8.464132308959961
    },
    {
      "epoch": 6.872367858886719e-05,
      "model_forward_time": 2.3447372913360596,
      "step": 72062
    },
    {
      "epoch": 6.872367858886719e-05,
      "step": 72062,
      "training_step_time": 8.655014514923096
    },
    {
      "epoch": 6.87246322631836e-05,
      "model_forward_time": 1.736544132232666,
      "step": 72063
    },
    {
      "epoch": 6.87246322631836e-05,
      "step": 72063,
      "training_step_time": 8.565983533859253
    },
    {
      "epoch": 6.87255859375e-05,
      "model_forward_time": 1.7546837329864502,
      "step": 72064
    },
    {
      "epoch": 6.87255859375e-05,
      "step": 72064,
      "training_step_time": 7.7206926345825195
    },
    {
      "epoch": 6.87265396118164e-05,
      "model_forward_time": 1.740088701248169,
      "step": 72065
    },
    {
      "epoch": 6.87265396118164e-05,
      "step": 72065,
      "training_step_time": 7.870957374572754
    },
    {
      "epoch": 6.872749328613281e-05,
      "model_forward_time": 1.7408714294433594,
      "step": 72066
    },
    {
      "epoch": 6.872749328613281e-05,
      "step": 72066,
      "training_step_time": 7.544878244400024
    },
    {
      "epoch": 6.872844696044921e-05,
      "model_forward_time": 1.7380902767181396,
      "step": 72067
    },
    {
      "epoch": 6.872844696044921e-05,
      "step": 72067,
      "training_step_time": 7.7098002433776855
    },
    {
      "epoch": 6.872940063476562e-05,
      "model_forward_time": 1.7433576583862305,
      "step": 72068
    },
    {
      "epoch": 6.872940063476562e-05,
      "step": 72068,
      "training_step_time": 7.0632007122039795
    },
    {
      "epoch": 6.873035430908203e-05,
      "model_forward_time": 1.744873285293579,
      "step": 72069
    },
    {
      "epoch": 6.873035430908203e-05,
      "step": 72069,
      "training_step_time": 7.754347324371338
    },
    {
      "epoch": 6.873130798339843e-05,
      "grad_norm": 0.09780877828598022,
      "learning_rate": 1.985484206105889e-06,
      "loss": 0.0568,
      "step": 72070
    },
    {
      "epoch": 6.873130798339843e-05,
      "model_forward_time": 1.7357916831970215,
      "step": 72070
    },
    {
      "dynamics_loss_avg": 0.0633154472336173,
      "epoch": 6.873130798339843e-05,
      "step": 72070
    },
    {
      "action_loss_avg": 0.0017426306672859936,
      "epoch": 6.873130798339843e-05,
      "step": 72070
    },
    {
      "epoch": 6.873130798339843e-05,
      "step": 72070,
      "training_step_time": 7.688290596008301
    },
    {
      "epoch": 6.873226165771484e-05,
      "model_forward_time": 1.7341268062591553,
      "step": 72071
    },
    {
      "epoch": 6.873226165771484e-05,
      "step": 72071,
      "training_step_time": 8.807806730270386
    },
    {
      "epoch": 6.873321533203125e-05,
      "model_forward_time": 1.7354116439819336,
      "step": 72072
    },
    {
      "epoch": 6.873321533203125e-05,
      "step": 72072,
      "training_step_time": 7.708235740661621
    },
    {
      "epoch": 6.873416900634765e-05,
      "model_forward_time": 1.7235581874847412,
      "step": 72073
    },
    {
      "epoch": 6.873416900634765e-05,
      "step": 72073,
      "training_step_time": 8.397710800170898
    },
    {
      "epoch": 6.873512268066406e-05,
      "model_forward_time": 1.7179949283599854,
      "step": 72074
    },
    {
      "epoch": 6.873512268066406e-05,
      "step": 72074,
      "training_step_time": 7.840667963027954
    },
    {
      "epoch": 6.873607635498047e-05,
      "model_forward_time": 1.719752550125122,
      "step": 72075
    },
    {
      "epoch": 6.873607635498047e-05,
      "step": 72075,
      "training_step_time": 7.675822019577026
    },
    {
      "epoch": 6.873703002929687e-05,
      "model_forward_time": 1.7147674560546875,
      "step": 72076
    },
    {
      "epoch": 6.873703002929687e-05,
      "step": 72076,
      "training_step_time": 7.71940803527832
    },
    {
      "epoch": 6.873798370361328e-05,
      "model_forward_time": 1.7048335075378418,
      "step": 72077
    },
    {
      "epoch": 6.873798370361328e-05,
      "step": 72077,
      "training_step_time": 7.117055177688599
    },
    {
      "epoch": 6.873893737792969e-05,
      "model_forward_time": 1.717116355895996,
      "step": 72078
    },
    {
      "epoch": 6.873893737792969e-05,
      "step": 72078,
      "training_step_time": 7.767133951187134
    },
    {
      "epoch": 6.873989105224609e-05,
      "model_forward_time": 1.7265346050262451,
      "step": 72079
    },
    {
      "epoch": 6.873989105224609e-05,
      "step": 72079,
      "training_step_time": 8.704651117324829
    },
    {
      "epoch": 6.87408447265625e-05,
      "grad_norm": 0.12620821595191956,
      "learning_rate": 1.984165208967889e-06,
      "loss": 0.0586,
      "step": 72080
    },
    {
      "epoch": 6.87408447265625e-05,
      "model_forward_time": 1.7367262840270996,
      "step": 72080
    },
    {
      "dynamics_loss_avg": 0.04781614448875189,
      "epoch": 6.87408447265625e-05,
      "step": 72080
    },
    {
      "action_loss_avg": 0.0011183446476934478,
      "epoch": 6.87408447265625e-05,
      "step": 72080
    },
    {
      "epoch": 6.87408447265625e-05,
      "step": 72080,
      "training_step_time": 7.744544744491577
    },
    {
      "epoch": 6.874179840087891e-05,
      "model_forward_time": 3.0895259380340576,
      "step": 72081
    },
    {
      "epoch": 6.874179840087891e-05,
      "step": 72081,
      "training_step_time": 7.9098498821258545
    },
    {
      "epoch": 6.87427520751953e-05,
      "model_forward_time": 1.9245262145996094,
      "step": 72082
    },
    {
      "epoch": 6.87427520751953e-05,
      "step": 72082,
      "training_step_time": 8.360888481140137
    },
    {
      "epoch": 6.874370574951172e-05,
      "model_forward_time": 1.9422454833984375,
      "step": 72083
    },
    {
      "epoch": 6.874370574951172e-05,
      "step": 72083,
      "training_step_time": 7.754710912704468
    },
    {
      "epoch": 6.874465942382813e-05,
      "model_forward_time": 1.9162623882293701,
      "step": 72084
    },
    {
      "epoch": 6.874465942382813e-05,
      "step": 72084,
      "training_step_time": 8.453541994094849
    },
    {
      "epoch": 6.874561309814453e-05,
      "model_forward_time": 1.9292387962341309,
      "step": 72085
    },
    {
      "epoch": 6.874561309814453e-05,
      "step": 72085,
      "training_step_time": 7.5079240798950195
    },
    {
      "epoch": 6.874656677246094e-05,
      "model_forward_time": 1.8889219760894775,
      "step": 72086
    },
    {
      "epoch": 6.874656677246094e-05,
      "step": 72086,
      "training_step_time": 7.627286434173584
    },
    {
      "epoch": 6.874752044677735e-05,
      "model_forward_time": 1.9313130378723145,
      "step": 72087
    },
    {
      "epoch": 6.874752044677735e-05,
      "step": 72087,
      "training_step_time": 7.6647326946258545
    },
    {
      "epoch": 6.874847412109374e-05,
      "model_forward_time": 1.9282195568084717,
      "step": 72088
    },
    {
      "epoch": 6.874847412109374e-05,
      "step": 72088,
      "training_step_time": 8.823729991912842
    },
    {
      "epoch": 6.874942779541015e-05,
      "model_forward_time": 1.9185094833374023,
      "step": 72089
    },
    {
      "epoch": 6.874942779541015e-05,
      "step": 72089,
      "training_step_time": 7.701807975769043
    },
    {
      "epoch": 6.875038146972657e-05,
      "grad_norm": 0.07751249521970749,
      "learning_rate": 1.9828465416370434e-06,
      "loss": 0.0557,
      "step": 72090
    },
    {
      "epoch": 6.875038146972657e-05,
      "model_forward_time": 3.006655216217041,
      "step": 72090
    },
    {
      "dynamics_loss_avg": 0.06294023506343364,
      "epoch": 6.875038146972657e-05,
      "step": 72090
    },
    {
      "action_loss_avg": 0.0013996830704854802,
      "epoch": 6.875038146972657e-05,
      "step": 72090
    },
    {
      "epoch": 6.875038146972657e-05,
      "step": 72090,
      "training_step_time": 8.375174760818481
    },
    {
      "epoch": 6.875133514404296e-05,
      "model_forward_time": 1.9327600002288818,
      "step": 72091
    },
    {
      "epoch": 6.875133514404296e-05,
      "step": 72091,
      "training_step_time": 7.760920286178589
    },
    {
      "epoch": 6.875228881835937e-05,
      "model_forward_time": 1.9382545948028564,
      "step": 72092
    },
    {
      "epoch": 6.875228881835937e-05,
      "step": 72092,
      "training_step_time": 8.6074857711792
    },
    {
      "epoch": 6.875324249267578e-05,
      "model_forward_time": 1.9048850536346436,
      "step": 72093
    },
    {
      "epoch": 6.875324249267578e-05,
      "step": 72093,
      "training_step_time": 7.848769903182983
    },
    {
      "epoch": 6.875419616699218e-05,
      "model_forward_time": 1.9132401943206787,
      "step": 72094
    },
    {
      "epoch": 6.875419616699218e-05,
      "step": 72094,
      "training_step_time": 7.531858205795288
    },
    {
      "epoch": 6.875514984130859e-05,
      "model_forward_time": 1.9398188591003418,
      "step": 72095
    },
    {
      "epoch": 6.875514984130859e-05,
      "step": 72095,
      "training_step_time": 7.74542498588562
    },
    {
      "epoch": 6.8756103515625e-05,
      "model_forward_time": 1.928992509841919,
      "step": 72096
    },
    {
      "epoch": 6.8756103515625e-05,
      "step": 72096,
      "training_step_time": 7.670603036880493
    },
    {
      "epoch": 6.87570571899414e-05,
      "model_forward_time": 1.905595302581787,
      "step": 72097
    },
    {
      "epoch": 6.87570571899414e-05,
      "step": 72097,
      "training_step_time": 7.654847621917725
    },
    {
      "epoch": 6.875801086425781e-05,
      "model_forward_time": 1.9314758777618408,
      "step": 72098
    },
    {
      "epoch": 6.875801086425781e-05,
      "step": 72098,
      "training_step_time": 7.654310464859009
    },
    {
      "epoch": 6.875896453857422e-05,
      "model_forward_time": 2.967456340789795,
      "step": 72099
    },
    {
      "epoch": 6.875896453857422e-05,
      "step": 72099,
      "training_step_time": 8.770788431167603
    },
    {
      "epoch": 6.875991821289062e-05,
      "grad_norm": 0.09552377462387085,
      "learning_rate": 1.9815282042575686e-06,
      "loss": 0.0535,
      "step": 72100
    },
    {
      "epoch": 6.875991821289062e-05,
      "model_forward_time": 1.93113112449646,
      "step": 72100
    },
    {
      "dynamics_loss_avg": 0.05464161448180675,
      "epoch": 6.875991821289062e-05,
      "step": 72100
    },
    {
      "action_loss_avg": 0.0027950463088927792,
      "epoch": 6.875991821289062e-05,
      "step": 72100
    },
    {
      "epoch": 6.875991821289062e-05,
      "step": 72100,
      "training_step_time": 7.762991666793823
    },
    {
      "epoch": 6.876087188720703e-05,
      "model_forward_time": 1.9319713115692139,
      "step": 72101
    },
    {
      "epoch": 6.876087188720703e-05,
      "step": 72101,
      "training_step_time": 7.677948713302612
    },
    {
      "epoch": 6.876182556152344e-05,
      "model_forward_time": 1.94071626663208,
      "step": 72102
    },
    {
      "epoch": 6.876182556152344e-05,
      "step": 72102,
      "training_step_time": 8.563998222351074
    },
    {
      "epoch": 6.876277923583984e-05,
      "model_forward_time": 1.9528858661651611,
      "step": 72103
    },
    {
      "epoch": 6.876277923583984e-05,
      "step": 72103,
      "training_step_time": 7.660247564315796
    },
    {
      "epoch": 6.876373291015625e-05,
      "model_forward_time": 1.891127109527588,
      "step": 72104
    },
    {
      "epoch": 6.876373291015625e-05,
      "step": 72104,
      "training_step_time": 7.5895092487335205
    },
    {
      "epoch": 6.876468658447266e-05,
      "model_forward_time": 1.8859786987304688,
      "step": 72105
    },
    {
      "epoch": 6.876468658447266e-05,
      "step": 72105,
      "training_step_time": 7.6191606521606445
    },
    {
      "epoch": 6.876564025878906e-05,
      "model_forward_time": 1.9263272285461426,
      "step": 72106
    },
    {
      "epoch": 6.876564025878906e-05,
      "step": 72106,
      "training_step_time": 7.603692293167114
    },
    {
      "epoch": 6.876659393310547e-05,
      "model_forward_time": 1.5452852249145508,
      "step": 72107
    },
    {
      "epoch": 6.876659393310547e-05,
      "step": 72107,
      "training_step_time": 7.704825401306152
    },
    {
      "epoch": 6.876754760742188e-05,
      "model_forward_time": 3.021397590637207,
      "step": 72108
    },
    {
      "epoch": 6.876754760742188e-05,
      "step": 72108,
      "training_step_time": 8.45159125328064
    },
    {
      "epoch": 6.876850128173828e-05,
      "model_forward_time": 1.9036905765533447,
      "step": 72109
    },
    {
      "epoch": 6.876850128173828e-05,
      "step": 72109,
      "training_step_time": 8.462773561477661
    },
    {
      "epoch": 6.876945495605469e-05,
      "grad_norm": 0.08020801842212677,
      "learning_rate": 1.9802101969736286e-06,
      "loss": 0.055,
      "step": 72110
    },
    {
      "epoch": 6.876945495605469e-05,
      "model_forward_time": 1.917130947113037,
      "step": 72110
    },
    {
      "dynamics_loss_avg": 0.05301051437854767,
      "epoch": 6.876945495605469e-05,
      "step": 72110
    },
    {
      "action_loss_avg": 0.002171040413668379,
      "epoch": 6.876945495605469e-05,
      "step": 72110
    },
    {
      "epoch": 6.876945495605469e-05,
      "step": 72110,
      "training_step_time": 7.812150955200195
    },
    {
      "epoch": 6.87704086303711e-05,
      "model_forward_time": 1.911078691482544,
      "step": 72111
    },
    {
      "epoch": 6.87704086303711e-05,
      "step": 72111,
      "training_step_time": 8.835825681686401
    },
    {
      "epoch": 6.87713623046875e-05,
      "model_forward_time": 1.9054229259490967,
      "step": 72112
    },
    {
      "epoch": 6.87713623046875e-05,
      "step": 72112,
      "training_step_time": 7.704286813735962
    },
    {
      "epoch": 6.87723159790039e-05,
      "model_forward_time": 1.9444770812988281,
      "step": 72113
    },
    {
      "epoch": 6.87723159790039e-05,
      "step": 72113,
      "training_step_time": 7.676496744155884
    },
    {
      "epoch": 6.877326965332032e-05,
      "model_forward_time": 1.9217140674591064,
      "step": 72114
    },
    {
      "epoch": 6.877326965332032e-05,
      "step": 72114,
      "training_step_time": 7.108734369277954
    },
    {
      "epoch": 6.877422332763671e-05,
      "model_forward_time": 1.906308650970459,
      "step": 72115
    },
    {
      "epoch": 6.877422332763671e-05,
      "step": 72115,
      "training_step_time": 8.736299514770508
    },
    {
      "epoch": 6.877517700195312e-05,
      "model_forward_time": 1.9046978950500488,
      "step": 72116
    },
    {
      "epoch": 6.877517700195312e-05,
      "step": 72116,
      "training_step_time": 7.793031692504883
    },
    {
      "epoch": 6.877613067626954e-05,
      "model_forward_time": 1.9120891094207764,
      "step": 72117
    },
    {
      "epoch": 6.877613067626954e-05,
      "step": 72117,
      "training_step_time": 7.722625255584717
    },
    {
      "epoch": 6.877708435058593e-05,
      "model_forward_time": 3.0515620708465576,
      "step": 72118
    },
    {
      "epoch": 6.877708435058593e-05,
      "step": 72118,
      "training_step_time": 8.75605821609497
    },
    {
      "epoch": 6.877803802490234e-05,
      "model_forward_time": 1.9332537651062012,
      "step": 72119
    },
    {
      "epoch": 6.877803802490234e-05,
      "step": 72119,
      "training_step_time": 7.727787017822266
    },
    {
      "epoch": 6.877899169921875e-05,
      "grad_norm": 0.1036544144153595,
      "learning_rate": 1.9788925199293636e-06,
      "loss": 0.0566,
      "step": 72120
    },
    {
      "epoch": 6.877899169921875e-05,
      "model_forward_time": 1.9380578994750977,
      "step": 72120
    },
    {
      "dynamics_loss_avg": 0.056708197202533485,
      "epoch": 6.877899169921875e-05,
      "step": 72120
    },
    {
      "action_loss_avg": 0.0018394980957964434,
      "epoch": 6.877899169921875e-05,
      "step": 72120
    },
    {
      "epoch": 6.877899169921875e-05,
      "step": 72120,
      "training_step_time": 8.202582597732544
    },
    {
      "epoch": 6.877994537353515e-05,
      "model_forward_time": 1.9449646472930908,
      "step": 72121
    },
    {
      "epoch": 6.877994537353515e-05,
      "step": 72121,
      "training_step_time": 7.751716136932373
    },
    {
      "epoch": 6.878089904785156e-05,
      "model_forward_time": 1.8765814304351807,
      "step": 72122
    },
    {
      "epoch": 6.878089904785156e-05,
      "step": 72122,
      "training_step_time": 7.480149745941162
    },
    {
      "epoch": 6.878185272216797e-05,
      "model_forward_time": 1.9224522113800049,
      "step": 72123
    },
    {
      "epoch": 6.878185272216797e-05,
      "step": 72123,
      "training_step_time": 6.717962026596069
    },
    {
      "epoch": 6.878280639648437e-05,
      "model_forward_time": 1.9399909973144531,
      "step": 72124
    },
    {
      "epoch": 6.878280639648437e-05,
      "step": 72124,
      "training_step_time": 7.693139553070068
    },
    {
      "epoch": 6.878376007080078e-05,
      "model_forward_time": 1.9433643817901611,
      "step": 72125
    },
    {
      "epoch": 6.878376007080078e-05,
      "step": 72125,
      "training_step_time": 7.671254873275757
    },
    {
      "epoch": 6.878471374511719e-05,
      "model_forward_time": 1.916363000869751,
      "step": 72126
    },
    {
      "epoch": 6.878471374511719e-05,
      "step": 72126,
      "training_step_time": 7.669050931930542
    },
    {
      "epoch": 6.878566741943359e-05,
      "model_forward_time": 3.040679693222046,
      "step": 72127
    },
    {
      "epoch": 6.878566741943359e-05,
      "step": 72127,
      "training_step_time": 10.472278356552124
    },
    {
      "epoch": 6.878662109375e-05,
      "model_forward_time": 1.9480228424072266,
      "step": 72128
    },
    {
      "epoch": 6.878662109375e-05,
      "step": 72128,
      "training_step_time": 7.905921220779419
    },
    {
      "epoch": 6.878757476806641e-05,
      "model_forward_time": 1.8974628448486328,
      "step": 72129
    },
    {
      "epoch": 6.878757476806641e-05,
      "step": 72129,
      "training_step_time": 7.691922426223755
    },
    {
      "epoch": 6.878852844238281e-05,
      "grad_norm": 0.10934605449438095,
      "learning_rate": 1.977575173268868e-06,
      "loss": 0.054,
      "step": 72130
    },
    {
      "epoch": 6.878852844238281e-05,
      "model_forward_time": 1.9251279830932617,
      "step": 72130
    },
    {
      "dynamics_loss_avg": 0.05477957557886839,
      "epoch": 6.878852844238281e-05,
      "step": 72130
    },
    {
      "action_loss_avg": 0.0022987580305198207,
      "epoch": 6.878852844238281e-05,
      "step": 72130
    },
    {
      "epoch": 6.878852844238281e-05,
      "step": 72130,
      "training_step_time": 7.869493246078491
    },
    {
      "epoch": 6.878948211669922e-05,
      "model_forward_time": 1.9042890071868896,
      "step": 72131
    },
    {
      "epoch": 6.878948211669922e-05,
      "step": 72131,
      "training_step_time": 7.589611768722534
    },
    {
      "epoch": 6.879043579101563e-05,
      "model_forward_time": 1.9203004837036133,
      "step": 72132
    },
    {
      "epoch": 6.879043579101563e-05,
      "step": 72132,
      "training_step_time": 7.850452423095703
    },
    {
      "epoch": 6.879138946533203e-05,
      "model_forward_time": 1.9364039897918701,
      "step": 72133
    },
    {
      "epoch": 6.879138946533203e-05,
      "step": 72133,
      "training_step_time": 7.77127480506897
    },
    {
      "epoch": 6.879234313964844e-05,
      "model_forward_time": 1.9364948272705078,
      "step": 72134
    },
    {
      "epoch": 6.879234313964844e-05,
      "step": 72134,
      "training_step_time": 7.702165126800537
    },
    {
      "epoch": 6.879329681396485e-05,
      "model_forward_time": 1.8973126411437988,
      "step": 72135
    },
    {
      "epoch": 6.879329681396485e-05,
      "step": 72135,
      "training_step_time": 7.658263921737671
    },
    {
      "epoch": 6.879425048828125e-05,
      "model_forward_time": 2.995965003967285,
      "step": 72136
    },
    {
      "epoch": 6.879425048828125e-05,
      "step": 72136,
      "training_step_time": 8.393375635147095
    },
    {
      "epoch": 6.879520416259766e-05,
      "model_forward_time": 1.949798583984375,
      "step": 72137
    },
    {
      "epoch": 6.879520416259766e-05,
      "step": 72137,
      "training_step_time": 9.939498901367188
    },
    {
      "epoch": 6.879615783691407e-05,
      "model_forward_time": 1.946216106414795,
      "step": 72138
    },
    {
      "epoch": 6.879615783691407e-05,
      "step": 72138,
      "training_step_time": 9.450550556182861
    },
    {
      "epoch": 6.879711151123046e-05,
      "model_forward_time": 1.9446654319763184,
      "step": 72139
    },
    {
      "epoch": 6.879711151123046e-05,
      "step": 72139,
      "training_step_time": 7.877159357070923
    },
    {
      "epoch": 6.879806518554688e-05,
      "grad_norm": 0.18749181926250458,
      "learning_rate": 1.9762581571362094e-06,
      "loss": 0.0499,
      "step": 72140
    },
    {
      "epoch": 6.879806518554688e-05,
      "model_forward_time": 1.9591903686523438,
      "step": 72140
    },
    {
      "dynamics_loss_avg": 0.06393276862800121,
      "epoch": 6.879806518554688e-05,
      "step": 72140
    },
    {
      "action_loss_avg": 0.0021339453232940286,
      "epoch": 6.879806518554688e-05,
      "step": 72140
    },
    {
      "epoch": 6.879806518554688e-05,
      "step": 72140,
      "training_step_time": 7.5938944816589355
    },
    {
      "epoch": 6.879901885986329e-05,
      "model_forward_time": 1.9435818195343018,
      "step": 72141
    },
    {
      "epoch": 6.879901885986329e-05,
      "step": 72141,
      "training_step_time": 7.1318933963775635
    },
    {
      "epoch": 6.879997253417968e-05,
      "model_forward_time": 1.9342079162597656,
      "step": 72142
    },
    {
      "epoch": 6.879997253417968e-05,
      "step": 72142,
      "training_step_time": 7.723900318145752
    },
    {
      "epoch": 6.88009262084961e-05,
      "model_forward_time": 1.9322822093963623,
      "step": 72143
    },
    {
      "epoch": 6.88009262084961e-05,
      "step": 72143,
      "training_step_time": 9.78816032409668
    },
    {
      "epoch": 6.88018798828125e-05,
      "model_forward_time": 1.8871116638183594,
      "step": 72144
    },
    {
      "epoch": 6.88018798828125e-05,
      "step": 72144,
      "training_step_time": 9.244722366333008
    },
    {
      "epoch": 6.88028335571289e-05,
      "model_forward_time": 3.0311505794525146,
      "step": 72145
    },
    {
      "epoch": 6.88028335571289e-05,
      "step": 72145,
      "training_step_time": 8.617679357528687
    },
    {
      "epoch": 6.880378723144531e-05,
      "model_forward_time": 1.90653395652771,
      "step": 72146
    },
    {
      "epoch": 6.880378723144531e-05,
      "step": 72146,
      "training_step_time": 8.690969228744507
    },
    {
      "epoch": 6.880474090576172e-05,
      "model_forward_time": 1.933537483215332,
      "step": 72147
    },
    {
      "epoch": 6.880474090576172e-05,
      "step": 72147,
      "training_step_time": 8.672337293624878
    },
    {
      "epoch": 6.880569458007812e-05,
      "model_forward_time": 1.9340004920959473,
      "step": 72148
    },
    {
      "epoch": 6.880569458007812e-05,
      "step": 72148,
      "training_step_time": 8.23119330406189
    },
    {
      "epoch": 6.880664825439453e-05,
      "model_forward_time": 1.89975905418396,
      "step": 72149
    },
    {
      "epoch": 6.880664825439453e-05,
      "step": 72149,
      "training_step_time": 7.8909993171691895
    },
    {
      "epoch": 6.880760192871094e-05,
      "grad_norm": 0.10804151743650436,
      "learning_rate": 1.974941471675411e-06,
      "loss": 0.0556,
      "step": 72150
    },
    {
      "epoch": 6.880760192871094e-05,
      "model_forward_time": 1.920790433883667,
      "step": 72150
    },
    {
      "dynamics_loss_avg": 0.05862562656402588,
      "epoch": 6.880760192871094e-05,
      "step": 72150
    },
    {
      "action_loss_avg": 0.002891029795864597,
      "epoch": 6.880760192871094e-05,
      "step": 72150
    },
    {
      "epoch": 6.880760192871094e-05,
      "step": 72150,
      "training_step_time": 7.964823961257935
    },
    {
      "epoch": 6.880855560302734e-05,
      "model_forward_time": 1.945042371749878,
      "step": 72151
    },
    {
      "epoch": 6.880855560302734e-05,
      "step": 72151,
      "training_step_time": 8.360660314559937
    },
    {
      "epoch": 6.880950927734375e-05,
      "model_forward_time": 1.9078412055969238,
      "step": 72152
    },
    {
      "epoch": 6.880950927734375e-05,
      "step": 72152,
      "training_step_time": 7.601529121398926
    },
    {
      "epoch": 6.881046295166016e-05,
      "model_forward_time": 1.8885786533355713,
      "step": 72153
    },
    {
      "epoch": 6.881046295166016e-05,
      "step": 72153,
      "training_step_time": 8.509933710098267
    },
    {
      "epoch": 6.881141662597656e-05,
      "model_forward_time": 2.993762493133545,
      "step": 72154
    },
    {
      "epoch": 6.881141662597656e-05,
      "step": 72154,
      "training_step_time": 9.166963338851929
    },
    {
      "epoch": 6.881237030029297e-05,
      "model_forward_time": 1.9058115482330322,
      "step": 72155
    },
    {
      "epoch": 6.881237030029297e-05,
      "step": 72155,
      "training_step_time": 8.246107578277588
    },
    {
      "epoch": 6.881332397460938e-05,
      "model_forward_time": 1.919813632965088,
      "step": 72156
    },
    {
      "epoch": 6.881332397460938e-05,
      "step": 72156,
      "training_step_time": 7.792969703674316
    },
    {
      "epoch": 6.881427764892578e-05,
      "model_forward_time": 1.932814598083496,
      "step": 72157
    },
    {
      "epoch": 6.881427764892578e-05,
      "step": 72157,
      "training_step_time": 7.609312057495117
    },
    {
      "epoch": 6.881523132324219e-05,
      "model_forward_time": 1.9127702713012695,
      "step": 72158
    },
    {
      "epoch": 6.881523132324219e-05,
      "step": 72158,
      "training_step_time": 7.800763368606567
    },
    {
      "epoch": 6.88161849975586e-05,
      "model_forward_time": 1.8905742168426514,
      "step": 72159
    },
    {
      "epoch": 6.88161849975586e-05,
      "step": 72159,
      "training_step_time": 7.462316036224365
    },
    {
      "epoch": 6.8817138671875e-05,
      "grad_norm": 0.1192740797996521,
      "learning_rate": 1.973625117030465e-06,
      "loss": 0.0549,
      "step": 72160
    },
    {
      "epoch": 6.8817138671875e-05,
      "model_forward_time": 1.9122443199157715,
      "step": 72160
    },
    {
      "dynamics_loss_avg": 0.03443775847554207,
      "epoch": 6.8817138671875e-05,
      "step": 72160
    },
    {
      "action_loss_avg": 0.0013473707338562235,
      "epoch": 6.8817138671875e-05,
      "step": 72160
    },
    {
      "epoch": 6.8817138671875e-05,
      "step": 72160,
      "training_step_time": 7.583166599273682
    },
    {
      "epoch": 6.881809234619141e-05,
      "model_forward_time": 1.9262616634368896,
      "step": 72161
    },
    {
      "epoch": 6.881809234619141e-05,
      "step": 72161,
      "training_step_time": 7.523872375488281
    },
    {
      "epoch": 6.881904602050782e-05,
      "model_forward_time": 1.9389123916625977,
      "step": 72162
    },
    {
      "epoch": 6.881904602050782e-05,
      "step": 72162,
      "training_step_time": 8.7345449924469
    },
    {
      "epoch": 6.881999969482422e-05,
      "model_forward_time": 1.923171043395996,
      "step": 72163
    },
    {
      "epoch": 6.881999969482422e-05,
      "step": 72163,
      "training_step_time": 8.754586219787598
    },
    {
      "epoch": 6.882095336914063e-05,
      "model_forward_time": 1.927938461303711,
      "step": 72164
    },
    {
      "epoch": 6.882095336914063e-05,
      "step": 72164,
      "training_step_time": 8.885612964630127
    },
    {
      "epoch": 6.882190704345704e-05,
      "model_forward_time": 1.907945156097412,
      "step": 72165
    },
    {
      "epoch": 6.882190704345704e-05,
      "step": 72165,
      "training_step_time": 8.961236238479614
    },
    {
      "epoch": 6.882286071777343e-05,
      "model_forward_time": 1.8841240406036377,
      "step": 72166
    },
    {
      "epoch": 6.882286071777343e-05,
      "step": 72166,
      "training_step_time": 7.623148441314697
    },
    {
      "epoch": 6.882381439208985e-05,
      "model_forward_time": 1.8890814781188965,
      "step": 72167
    },
    {
      "epoch": 6.882381439208985e-05,
      "step": 72167,
      "training_step_time": 7.815857410430908
    },
    {
      "epoch": 6.882476806640626e-05,
      "model_forward_time": 1.9280478954315186,
      "step": 72168
    },
    {
      "epoch": 6.882476806640626e-05,
      "step": 72168,
      "training_step_time": 7.50489616394043
    },
    {
      "epoch": 6.882572174072265e-05,
      "model_forward_time": 1.9253556728363037,
      "step": 72169
    },
    {
      "epoch": 6.882572174072265e-05,
      "step": 72169,
      "training_step_time": 7.063901662826538
    },
    {
      "epoch": 6.882667541503906e-05,
      "grad_norm": 0.1094125509262085,
      "learning_rate": 1.972309093345326e-06,
      "loss": 0.053,
      "step": 72170
    },
    {
      "epoch": 6.882667541503906e-05,
      "model_forward_time": 1.8887462615966797,
      "step": 72170
    },
    {
      "dynamics_loss_avg": 0.058891558274626735,
      "epoch": 6.882667541503906e-05,
      "step": 72170
    },
    {
      "action_loss_avg": 0.0015206873882561921,
      "epoch": 6.882667541503906e-05,
      "step": 72170
    },
    {
      "epoch": 6.882667541503906e-05,
      "step": 72170,
      "training_step_time": 7.642654657363892
    },
    {
      "epoch": 6.882762908935547e-05,
      "model_forward_time": 1.9096064567565918,
      "step": 72171
    },
    {
      "epoch": 6.882762908935547e-05,
      "step": 72171,
      "training_step_time": 8.158580303192139
    },
    {
      "epoch": 6.882858276367187e-05,
      "model_forward_time": 1.930882215499878,
      "step": 72172
    },
    {
      "epoch": 6.882858276367187e-05,
      "step": 72172,
      "training_step_time": 7.758822202682495
    },
    {
      "epoch": 6.882953643798828e-05,
      "model_forward_time": 2.577816963195801,
      "step": 72173
    },
    {
      "epoch": 6.882953643798828e-05,
      "step": 72173,
      "training_step_time": 7.797297477722168
    },
    {
      "epoch": 6.88304901123047e-05,
      "model_forward_time": 1.939727783203125,
      "step": 72174
    },
    {
      "epoch": 6.88304901123047e-05,
      "step": 72174,
      "training_step_time": 7.892607927322388
    },
    {
      "epoch": 6.883144378662109e-05,
      "model_forward_time": 1.9268465042114258,
      "step": 72175
    },
    {
      "epoch": 6.883144378662109e-05,
      "step": 72175,
      "training_step_time": 7.835258722305298
    },
    {
      "epoch": 6.88323974609375e-05,
      "model_forward_time": 1.888631820678711,
      "step": 72176
    },
    {
      "epoch": 6.88323974609375e-05,
      "step": 72176,
      "training_step_time": 7.889837265014648
    },
    {
      "epoch": 6.883335113525391e-05,
      "model_forward_time": 1.9314780235290527,
      "step": 72177
    },
    {
      "epoch": 6.883335113525391e-05,
      "step": 72177,
      "training_step_time": 7.560737133026123
    },
    {
      "epoch": 6.883430480957031e-05,
      "model_forward_time": 1.9212098121643066,
      "step": 72178
    },
    {
      "epoch": 6.883430480957031e-05,
      "step": 72178,
      "training_step_time": 7.1260316371917725
    },
    {
      "epoch": 6.883525848388672e-05,
      "model_forward_time": 1.9361755847930908,
      "step": 72179
    },
    {
      "epoch": 6.883525848388672e-05,
      "step": 72179,
      "training_step_time": 7.707445383071899
    },
    {
      "epoch": 6.883621215820313e-05,
      "grad_norm": 0.07768258452415466,
      "learning_rate": 1.970993400763915e-06,
      "loss": 0.0541,
      "step": 72180
    },
    {
      "epoch": 6.883621215820313e-05,
      "model_forward_time": 1.8978347778320312,
      "step": 72180
    },
    {
      "dynamics_loss_avg": 0.05306140575557947,
      "epoch": 6.883621215820313e-05,
      "step": 72180
    },
    {
      "action_loss_avg": 0.00158353561128024,
      "epoch": 6.883621215820313e-05,
      "step": 72180
    },
    {
      "epoch": 6.883621215820313e-05,
      "step": 72180,
      "training_step_time": 8.375638723373413
    },
    {
      "epoch": 6.883716583251953e-05,
      "model_forward_time": 1.932809829711914,
      "step": 72181
    },
    {
      "epoch": 6.883716583251953e-05,
      "step": 72181,
      "training_step_time": 7.7960755825042725
    },
    {
      "epoch": 6.883811950683594e-05,
      "model_forward_time": 3.022008180618286,
      "step": 72182
    },
    {
      "epoch": 6.883811950683594e-05,
      "step": 72182,
      "training_step_time": 8.507834672927856
    },
    {
      "epoch": 6.883907318115235e-05,
      "model_forward_time": 1.9132328033447266,
      "step": 72183
    },
    {
      "epoch": 6.883907318115235e-05,
      "step": 72183,
      "training_step_time": 10.953071594238281
    },
    {
      "epoch": 6.884002685546875e-05,
      "model_forward_time": 1.9336109161376953,
      "step": 72184
    },
    {
      "epoch": 6.884002685546875e-05,
      "step": 72184,
      "training_step_time": 8.717097997665405
    },
    {
      "epoch": 6.884098052978516e-05,
      "model_forward_time": 1.8990910053253174,
      "step": 72185
    },
    {
      "epoch": 6.884098052978516e-05,
      "step": 72185,
      "training_step_time": 7.891138315200806
    },
    {
      "epoch": 6.884193420410157e-05,
      "model_forward_time": 1.9079346656799316,
      "step": 72186
    },
    {
      "epoch": 6.884193420410157e-05,
      "step": 72186,
      "training_step_time": 7.6090593338012695
    },
    {
      "epoch": 6.884288787841797e-05,
      "model_forward_time": 1.5305933952331543,
      "step": 72187
    },
    {
      "epoch": 6.884288787841797e-05,
      "step": 72187,
      "training_step_time": 7.448858976364136
    },
    {
      "epoch": 6.884384155273438e-05,
      "model_forward_time": 1.928161382675171,
      "step": 72188
    },
    {
      "epoch": 6.884384155273438e-05,
      "step": 72188,
      "training_step_time": 7.657779693603516
    },
    {
      "epoch": 6.884479522705077e-05,
      "model_forward_time": 1.9140965938568115,
      "step": 72189
    },
    {
      "epoch": 6.884479522705077e-05,
      "step": 72189,
      "training_step_time": 7.715212345123291
    },
    {
      "epoch": 6.884574890136719e-05,
      "grad_norm": 0.07769880443811417,
      "learning_rate": 1.969678039430109e-06,
      "loss": 0.0587,
      "step": 72190
    },
    {
      "epoch": 6.884574890136719e-05,
      "model_forward_time": 1.9400501251220703,
      "step": 72190
    },
    {
      "dynamics_loss_avg": 0.04931184165179729,
      "epoch": 6.884574890136719e-05,
      "step": 72190
    },
    {
      "action_loss_avg": 0.004555469687329605,
      "epoch": 6.884574890136719e-05,
      "step": 72190
    },
    {
      "epoch": 6.884574890136719e-05,
      "step": 72190,
      "training_step_time": 7.877950429916382
    },
    {
      "epoch": 6.88467025756836e-05,
      "model_forward_time": 3.032033920288086,
      "step": 72191
    },
    {
      "epoch": 6.88467025756836e-05,
      "step": 72191,
      "training_step_time": 9.72211217880249
    },
    {
      "epoch": 6.884765625e-05,
      "model_forward_time": 1.8811869621276855,
      "step": 72192
    },
    {
      "epoch": 6.884765625e-05,
      "step": 72192,
      "training_step_time": 8.443609237670898
    },
    {
      "epoch": 6.88486099243164e-05,
      "model_forward_time": 1.876889705657959,
      "step": 72193
    },
    {
      "epoch": 6.88486099243164e-05,
      "step": 72193,
      "training_step_time": 7.864834308624268
    },
    {
      "epoch": 6.884956359863281e-05,
      "model_forward_time": 1.9247851371765137,
      "step": 72194
    },
    {
      "epoch": 6.884956359863281e-05,
      "step": 72194,
      "training_step_time": 8.65889573097229
    },
    {
      "epoch": 6.885051727294921e-05,
      "model_forward_time": 1.9277231693267822,
      "step": 72195
    },
    {
      "epoch": 6.885051727294921e-05,
      "step": 72195,
      "training_step_time": 7.752802610397339
    },
    {
      "epoch": 6.885147094726562e-05,
      "model_forward_time": 1.8763561248779297,
      "step": 72196
    },
    {
      "epoch": 6.885147094726562e-05,
      "step": 72196,
      "training_step_time": 7.536929368972778
    },
    {
      "epoch": 6.885242462158203e-05,
      "model_forward_time": 1.9190573692321777,
      "step": 72197
    },
    {
      "epoch": 6.885242462158203e-05,
      "step": 72197,
      "training_step_time": 6.928689002990723
    },
    {
      "epoch": 6.885337829589843e-05,
      "model_forward_time": 1.9377617835998535,
      "step": 72198
    },
    {
      "epoch": 6.885337829589843e-05,
      "step": 72198,
      "training_step_time": 7.700308561325073
    },
    {
      "epoch": 6.885433197021484e-05,
      "model_forward_time": 1.8835439682006836,
      "step": 72199
    },
    {
      "epoch": 6.885433197021484e-05,
      "step": 72199,
      "training_step_time": 7.801296949386597
    },
    {
      "epoch": 6.885528564453125e-05,
      "grad_norm": 0.11361680179834366,
      "learning_rate": 1.9683630094877563e-06,
      "loss": 0.0559,
      "step": 72200
    },
    {
      "epoch": 6.885528564453125e-05,
      "model_forward_time": 1.9260778427124023,
      "step": 72200
    },
    {
      "dynamics_loss_avg": 0.049186524003744125,
      "epoch": 6.885528564453125e-05,
      "step": 72200
    },
    {
      "action_loss_avg": 0.0020634613349102436,
      "epoch": 6.885528564453125e-05,
      "step": 72200
    },
    {
      "epoch": 6.885528564453125e-05,
      "step": 72200,
      "training_step_time": 9.972949981689453
    },
    {
      "epoch": 6.885623931884765e-05,
      "model_forward_time": 1.901468276977539,
      "step": 72201
    },
    {
      "epoch": 6.885623931884765e-05,
      "step": 72201,
      "training_step_time": 7.8426947593688965
    },
    {
      "epoch": 6.885719299316406e-05,
      "model_forward_time": 1.9176981449127197,
      "step": 72202
    },
    {
      "epoch": 6.885719299316406e-05,
      "step": 72202,
      "training_step_time": 7.981497526168823
    },
    {
      "epoch": 6.885814666748047e-05,
      "model_forward_time": 1.6793169975280762,
      "step": 72203
    },
    {
      "epoch": 6.885814666748047e-05,
      "step": 72203,
      "training_step_time": 8.540342330932617
    },
    {
      "epoch": 6.885910034179687e-05,
      "model_forward_time": 1.6943705081939697,
      "step": 72204
    },
    {
      "epoch": 6.885910034179687e-05,
      "step": 72204,
      "training_step_time": 7.81688117980957
    },
    {
      "epoch": 6.886005401611328e-05,
      "model_forward_time": 1.6958327293395996,
      "step": 72205
    },
    {
      "epoch": 6.886005401611328e-05,
      "step": 72205,
      "training_step_time": 7.62068510055542
    },
    {
      "epoch": 6.886100769042969e-05,
      "model_forward_time": 1.7040646076202393,
      "step": 72206
    },
    {
      "epoch": 6.886100769042969e-05,
      "step": 72206,
      "training_step_time": 7.179759502410889
    },
    {
      "epoch": 6.886196136474609e-05,
      "model_forward_time": 1.702523946762085,
      "step": 72207
    },
    {
      "epoch": 6.886196136474609e-05,
      "step": 72207,
      "training_step_time": 7.829310894012451
    },
    {
      "epoch": 6.88629150390625e-05,
      "model_forward_time": 1.6972620487213135,
      "step": 72208
    },
    {
      "epoch": 6.88629150390625e-05,
      "step": 72208,
      "training_step_time": 9.095978498458862
    },
    {
      "epoch": 6.886386871337891e-05,
      "model_forward_time": 1.7016801834106445,
      "step": 72209
    },
    {
      "epoch": 6.886386871337891e-05,
      "step": 72209,
      "training_step_time": 8.857436895370483
    },
    {
      "epoch": 6.88648223876953e-05,
      "grad_norm": 0.10500039905309677,
      "learning_rate": 1.9670483110806697e-06,
      "loss": 0.0591,
      "step": 72210
    },
    {
      "epoch": 6.88648223876953e-05,
      "model_forward_time": 2.2663815021514893,
      "step": 72210
    },
    {
      "dynamics_loss_avg": 0.0602027552202344,
      "epoch": 6.88648223876953e-05,
      "step": 72210
    },
    {
      "action_loss_avg": 0.002237173169851303,
      "epoch": 6.88648223876953e-05,
      "step": 72210
    },
    {
      "epoch": 6.88648223876953e-05,
      "step": 72210,
      "training_step_time": 8.274973154067993
    },
    {
      "epoch": 6.886577606201172e-05,
      "model_forward_time": 1.3486788272857666,
      "step": 72211
    },
    {
      "epoch": 6.886577606201172e-05,
      "step": 72211,
      "training_step_time": 7.8304102420806885
    },
    {
      "epoch": 6.886672973632813e-05,
      "model_forward_time": 1.6894917488098145,
      "step": 72212
    },
    {
      "epoch": 6.886672973632813e-05,
      "step": 72212,
      "training_step_time": 7.759068727493286
    },
    {
      "epoch": 6.886768341064453e-05,
      "model_forward_time": 1.694108009338379,
      "step": 72213
    },
    {
      "epoch": 6.886768341064453e-05,
      "step": 72213,
      "training_step_time": 7.7385947704315186
    },
    {
      "epoch": 6.886863708496094e-05,
      "model_forward_time": 1.6977565288543701,
      "step": 72214
    },
    {
      "epoch": 6.886863708496094e-05,
      "step": 72214,
      "training_step_time": 7.485368728637695
    },
    {
      "epoch": 6.886959075927735e-05,
      "model_forward_time": 1.6994295120239258,
      "step": 72215
    },
    {
      "epoch": 6.886959075927735e-05,
      "step": 72215,
      "training_step_time": 7.403642177581787
    },
    {
      "epoch": 6.887054443359374e-05,
      "model_forward_time": 1.6955585479736328,
      "step": 72216
    },
    {
      "epoch": 6.887054443359374e-05,
      "step": 72216,
      "training_step_time": 8.719282627105713
    },
    {
      "epoch": 6.887149810791016e-05,
      "model_forward_time": 1.702904224395752,
      "step": 72217
    },
    {
      "epoch": 6.887149810791016e-05,
      "step": 72217,
      "training_step_time": 8.101314067840576
    },
    {
      "epoch": 6.887245178222657e-05,
      "model_forward_time": 1.69252610206604,
      "step": 72218
    },
    {
      "epoch": 6.887245178222657e-05,
      "step": 72218,
      "training_step_time": 8.75500226020813
    },
    {
      "epoch": 6.887340545654296e-05,
      "model_forward_time": 2.2579617500305176,
      "step": 72219
    },
    {
      "epoch": 6.887340545654296e-05,
      "step": 72219,
      "training_step_time": 9.371551275253296
    },
    {
      "epoch": 6.887435913085937e-05,
      "grad_norm": 0.11777001619338989,
      "learning_rate": 1.9657339443526172e-06,
      "loss": 0.0546,
      "step": 72220
    },
    {
      "epoch": 6.887435913085937e-05,
      "model_forward_time": 1.321263074874878,
      "step": 72220
    },
    {
      "dynamics_loss_avg": 0.04447173085063696,
      "epoch": 6.887435913085937e-05,
      "step": 72220
    },
    {
      "action_loss_avg": 0.0013473040889948606,
      "epoch": 6.887435913085937e-05,
      "step": 72220
    },
    {
      "epoch": 6.887435913085937e-05,
      "step": 72220,
      "training_step_time": 8.012916088104248
    },
    {
      "epoch": 6.887531280517578e-05,
      "model_forward_time": 1.7080378532409668,
      "step": 72221
    },
    {
      "epoch": 6.887531280517578e-05,
      "step": 72221,
      "training_step_time": 8.501162767410278
    },
    {
      "epoch": 6.887626647949218e-05,
      "model_forward_time": 1.6982979774475098,
      "step": 72222
    },
    {
      "epoch": 6.887626647949218e-05,
      "step": 72222,
      "training_step_time": 7.725635766983032
    },
    {
      "epoch": 6.887722015380859e-05,
      "model_forward_time": 1.7025995254516602,
      "step": 72223
    },
    {
      "epoch": 6.887722015380859e-05,
      "step": 72223,
      "training_step_time": 7.812922239303589
    },
    {
      "epoch": 6.8878173828125e-05,
      "model_forward_time": 1.6975297927856445,
      "step": 72224
    },
    {
      "epoch": 6.8878173828125e-05,
      "step": 72224,
      "training_step_time": 7.662228107452393
    },
    {
      "epoch": 6.88791275024414e-05,
      "model_forward_time": 1.6979296207427979,
      "step": 72225
    },
    {
      "epoch": 6.88791275024414e-05,
      "step": 72225,
      "training_step_time": 8.417149305343628
    },
    {
      "epoch": 6.888008117675781e-05,
      "model_forward_time": 1.704911231994629,
      "step": 72226
    },
    {
      "epoch": 6.888008117675781e-05,
      "step": 72226,
      "training_step_time": 7.630341053009033
    },
    {
      "epoch": 6.888103485107422e-05,
      "model_forward_time": 1.7030870914459229,
      "step": 72227
    },
    {
      "epoch": 6.888103485107422e-05,
      "step": 72227,
      "training_step_time": 8.805038213729858
    },
    {
      "epoch": 6.888198852539062e-05,
      "model_forward_time": 2.2558722496032715,
      "step": 72228
    },
    {
      "epoch": 6.888198852539062e-05,
      "step": 72228,
      "training_step_time": 8.517279624938965
    },
    {
      "epoch": 6.888294219970703e-05,
      "model_forward_time": 1.7008564472198486,
      "step": 72229
    },
    {
      "epoch": 6.888294219970703e-05,
      "step": 72229,
      "training_step_time": 9.015526056289673
    },
    {
      "epoch": 6.888389587402344e-05,
      "grad_norm": 0.09024576842784882,
      "learning_rate": 1.964419909447339e-06,
      "loss": 0.057,
      "step": 72230
    },
    {
      "epoch": 6.888389587402344e-05,
      "model_forward_time": 1.6977007389068604,
      "step": 72230
    },
    {
      "dynamics_loss_avg": 0.07019171565771103,
      "epoch": 6.888389587402344e-05,
      "step": 72230
    },
    {
      "action_loss_avg": 0.0013660803495440632,
      "epoch": 6.888389587402344e-05,
      "step": 72230
    },
    {
      "epoch": 6.888389587402344e-05,
      "step": 72230,
      "training_step_time": 7.905045986175537
    },
    {
      "epoch": 6.888484954833984e-05,
      "model_forward_time": 1.6928279399871826,
      "step": 72231
    },
    {
      "epoch": 6.888484954833984e-05,
      "step": 72231,
      "training_step_time": 7.636978387832642
    },
    {
      "epoch": 6.888580322265625e-05,
      "model_forward_time": 1.6930952072143555,
      "step": 72232
    },
    {
      "epoch": 6.888580322265625e-05,
      "step": 72232,
      "training_step_time": 7.819866895675659
    },
    {
      "epoch": 6.888675689697266e-05,
      "model_forward_time": 1.6992876529693604,
      "step": 72233
    },
    {
      "epoch": 6.888675689697266e-05,
      "step": 72233,
      "training_step_time": 7.533899307250977
    },
    {
      "epoch": 6.888771057128906e-05,
      "model_forward_time": 1.6921064853668213,
      "step": 72234
    },
    {
      "epoch": 6.888771057128906e-05,
      "step": 72234,
      "training_step_time": 7.580581426620483
    },
    {
      "epoch": 6.888866424560547e-05,
      "model_forward_time": 1.6945247650146484,
      "step": 72235
    },
    {
      "epoch": 6.888866424560547e-05,
      "step": 72235,
      "training_step_time": 8.2215256690979
    },
    {
      "epoch": 6.888961791992188e-05,
      "model_forward_time": 1.6974754333496094,
      "step": 72236
    },
    {
      "epoch": 6.888961791992188e-05,
      "step": 72236,
      "training_step_time": 7.69167947769165
    },
    {
      "epoch": 6.889057159423828e-05,
      "model_forward_time": 2.2398884296417236,
      "step": 72237
    },
    {
      "epoch": 6.889057159423828e-05,
      "step": 72237,
      "training_step_time": 10.175217628479004
    },
    {
      "epoch": 6.889152526855469e-05,
      "model_forward_time": 1.7038767337799072,
      "step": 72238
    },
    {
      "epoch": 6.889152526855469e-05,
      "step": 72238,
      "training_step_time": 7.939556121826172
    },
    {
      "epoch": 6.88924789428711e-05,
      "model_forward_time": 1.7116591930389404,
      "step": 72239
    },
    {
      "epoch": 6.88924789428711e-05,
      "step": 72239,
      "training_step_time": 8.728657484054565
    },
    {
      "epoch": 6.88934326171875e-05,
      "grad_norm": 0.11805818974971771,
      "learning_rate": 1.963106206508536e-06,
      "loss": 0.0559,
      "step": 72240
    },
    {
      "epoch": 6.88934326171875e-05,
      "model_forward_time": 1.6983928680419922,
      "step": 72240
    },
    {
      "dynamics_loss_avg": 0.04684844687581062,
      "epoch": 6.88934326171875e-05,
      "step": 72240
    },
    {
      "action_loss_avg": 0.0021810604899656027,
      "epoch": 6.88934326171875e-05,
      "step": 72240
    },
    {
      "epoch": 6.88934326171875e-05,
      "step": 72240,
      "training_step_time": 7.570548057556152
    },
    {
      "epoch": 6.88943862915039e-05,
      "model_forward_time": 1.326087474822998,
      "step": 72241
    },
    {
      "epoch": 6.88943862915039e-05,
      "step": 72241,
      "training_step_time": 7.643381118774414
    },
    {
      "epoch": 6.889533996582032e-05,
      "model_forward_time": 1.6985085010528564,
      "step": 72242
    },
    {
      "epoch": 6.889533996582032e-05,
      "step": 72242,
      "training_step_time": 7.581938743591309
    },
    {
      "epoch": 6.889629364013671e-05,
      "model_forward_time": 1.7010776996612549,
      "step": 72243
    },
    {
      "epoch": 6.889629364013671e-05,
      "step": 72243,
      "training_step_time": 7.679453372955322
    },
    {
      "epoch": 6.889724731445312e-05,
      "model_forward_time": 1.6952297687530518,
      "step": 72244
    },
    {
      "epoch": 6.889724731445312e-05,
      "step": 72244,
      "training_step_time": 7.626698970794678
    },
    {
      "epoch": 6.889820098876954e-05,
      "model_forward_time": 1.6955673694610596,
      "step": 72245
    },
    {
      "epoch": 6.889820098876954e-05,
      "step": 72245,
      "training_step_time": 7.664987802505493
    },
    {
      "epoch": 6.889915466308593e-05,
      "model_forward_time": 1.7183706760406494,
      "step": 72246
    },
    {
      "epoch": 6.889915466308593e-05,
      "step": 72246,
      "training_step_time": 9.627472400665283
    },
    {
      "epoch": 6.890010833740234e-05,
      "model_forward_time": 2.2385692596435547,
      "step": 72247
    },
    {
      "epoch": 6.890010833740234e-05,
      "step": 72247,
      "training_step_time": 8.341944932937622
    },
    {
      "epoch": 6.890106201171875e-05,
      "model_forward_time": 1.715301275253296,
      "step": 72248
    },
    {
      "epoch": 6.890106201171875e-05,
      "step": 72248,
      "training_step_time": 8.870170831680298
    },
    {
      "epoch": 6.890201568603515e-05,
      "model_forward_time": 1.701991319656372,
      "step": 72249
    },
    {
      "epoch": 6.890201568603515e-05,
      "step": 72249,
      "training_step_time": 9.776307821273804
    },
    {
      "epoch": 6.890296936035156e-05,
      "grad_norm": 0.09146922826766968,
      "learning_rate": 1.961792835679874e-06,
      "loss": 0.0603,
      "step": 72250
    },
    {
      "epoch": 6.890296936035156e-05,
      "model_forward_time": 1.6867468357086182,
      "step": 72250
    },
    {
      "dynamics_loss_avg": 0.04962766524404287,
      "epoch": 6.890296936035156e-05,
      "step": 72250
    },
    {
      "action_loss_avg": 0.0016844596189912408,
      "epoch": 6.890296936035156e-05,
      "step": 72250
    },
    {
      "epoch": 6.890296936035156e-05,
      "step": 72250,
      "training_step_time": 7.680246114730835
    },
    {
      "epoch": 6.890392303466797e-05,
      "model_forward_time": 1.6965868473052979,
      "step": 72251
    },
    {
      "epoch": 6.890392303466797e-05,
      "step": 72251,
      "training_step_time": 7.654090642929077
    },
    {
      "epoch": 6.890487670898437e-05,
      "model_forward_time": 1.6978034973144531,
      "step": 72252
    },
    {
      "epoch": 6.890487670898437e-05,
      "step": 72252,
      "training_step_time": 7.657162666320801
    },
    {
      "epoch": 6.890583038330078e-05,
      "model_forward_time": 1.3200316429138184,
      "step": 72253
    },
    {
      "epoch": 6.890583038330078e-05,
      "step": 72253,
      "training_step_time": 7.630740165710449
    },
    {
      "epoch": 6.890678405761719e-05,
      "model_forward_time": 1.6947252750396729,
      "step": 72254
    },
    {
      "epoch": 6.890678405761719e-05,
      "step": 72254,
      "training_step_time": 8.788236379623413
    },
    {
      "epoch": 6.890773773193359e-05,
      "model_forward_time": 1.7224702835083008,
      "step": 72255
    },
    {
      "epoch": 6.890773773193359e-05,
      "step": 72255,
      "training_step_time": 8.795654296875
    },
    {
      "epoch": 6.890869140625e-05,
      "model_forward_time": 2.2635655403137207,
      "step": 72256
    },
    {
      "epoch": 6.890869140625e-05,
      "step": 72256,
      "training_step_time": 8.841952085494995
    },
    {
      "epoch": 6.890964508056641e-05,
      "model_forward_time": 1.7188999652862549,
      "step": 72257
    },
    {
      "epoch": 6.890964508056641e-05,
      "step": 72257,
      "training_step_time": 10.720066785812378
    },
    {
      "epoch": 6.891059875488281e-05,
      "model_forward_time": 1.6985571384429932,
      "step": 72258
    },
    {
      "epoch": 6.891059875488281e-05,
      "step": 72258,
      "training_step_time": 7.853089094161987
    },
    {
      "epoch": 6.891155242919922e-05,
      "model_forward_time": 1.6966526508331299,
      "step": 72259
    },
    {
      "epoch": 6.891155242919922e-05,
      "step": 72259,
      "training_step_time": 7.777588367462158
    },
    {
      "epoch": 6.891250610351563e-05,
      "grad_norm": 0.10418988019227982,
      "learning_rate": 1.960479797104977e-06,
      "loss": 0.0576,
      "step": 72260
    },
    {
      "epoch": 6.891250610351563e-05,
      "model_forward_time": 1.7044060230255127,
      "step": 72260
    },
    {
      "dynamics_loss_avg": 0.04955677967518568,
      "epoch": 6.891250610351563e-05,
      "step": 72260
    },
    {
      "action_loss_avg": 0.0010469373344676569,
      "epoch": 6.891250610351563e-05,
      "step": 72260
    },
    {
      "epoch": 6.891250610351563e-05,
      "step": 72260,
      "training_step_time": 7.676218271255493
    },
    {
      "epoch": 6.891345977783203e-05,
      "model_forward_time": 1.697721242904663,
      "step": 72261
    },
    {
      "epoch": 6.891345977783203e-05,
      "step": 72261,
      "training_step_time": 7.613436460494995
    },
    {
      "epoch": 6.891441345214844e-05,
      "model_forward_time": 1.6912238597869873,
      "step": 72262
    },
    {
      "epoch": 6.891441345214844e-05,
      "step": 72262,
      "training_step_time": 7.6242828369140625
    },
    {
      "epoch": 6.891536712646485e-05,
      "model_forward_time": 1.7007551193237305,
      "step": 72263
    },
    {
      "epoch": 6.891536712646485e-05,
      "step": 72263,
      "training_step_time": 7.676577806472778
    },
    {
      "epoch": 6.891632080078125e-05,
      "model_forward_time": 1.7201175689697266,
      "step": 72264
    },
    {
      "epoch": 6.891632080078125e-05,
      "step": 72264,
      "training_step_time": 8.342146396636963
    },
    {
      "epoch": 6.891727447509766e-05,
      "model_forward_time": 2.2548987865448,
      "step": 72265
    },
    {
      "epoch": 6.891727447509766e-05,
      "step": 72265,
      "training_step_time": 7.715404033660889
    },
    {
      "epoch": 6.891822814941407e-05,
      "model_forward_time": 1.7131531238555908,
      "step": 72266
    },
    {
      "epoch": 6.891822814941407e-05,
      "step": 72266,
      "training_step_time": 7.916608095169067
    },
    {
      "epoch": 6.891918182373046e-05,
      "model_forward_time": 1.707277536392212,
      "step": 72267
    },
    {
      "epoch": 6.891918182373046e-05,
      "step": 72267,
      "training_step_time": 9.220041513442993
    },
    {
      "epoch": 6.892013549804688e-05,
      "model_forward_time": 1.6968333721160889,
      "step": 72268
    },
    {
      "epoch": 6.892013549804688e-05,
      "step": 72268,
      "training_step_time": 7.7369067668914795
    },
    {
      "epoch": 6.892108917236329e-05,
      "model_forward_time": 1.701570749282837,
      "step": 72269
    },
    {
      "epoch": 6.892108917236329e-05,
      "step": 72269,
      "training_step_time": 7.635594844818115
    },
    {
      "epoch": 6.892204284667968e-05,
      "grad_norm": 0.12074639648199081,
      "learning_rate": 1.959167090927442e-06,
      "loss": 0.0579,
      "step": 72270
    },
    {
      "epoch": 6.892204284667968e-05,
      "model_forward_time": 1.7042994499206543,
      "step": 72270
    },
    {
      "dynamics_loss_avg": 0.06309633962810039,
      "epoch": 6.892204284667968e-05,
      "step": 72270
    },
    {
      "action_loss_avg": 0.0021157543640583755,
      "epoch": 6.892204284667968e-05,
      "step": 72270
    },
    {
      "epoch": 6.892204284667968e-05,
      "step": 72270,
      "training_step_time": 7.590000867843628
    },
    {
      "epoch": 6.89229965209961e-05,
      "model_forward_time": 1.687835454940796,
      "step": 72271
    },
    {
      "epoch": 6.89229965209961e-05,
      "step": 72271,
      "training_step_time": 8.469660758972168
    },
    {
      "epoch": 6.89239501953125e-05,
      "model_forward_time": 1.7042701244354248,
      "step": 72272
    },
    {
      "epoch": 6.89239501953125e-05,
      "step": 72272,
      "training_step_time": 7.549800872802734
    },
    {
      "epoch": 6.89249038696289e-05,
      "model_forward_time": 1.7028093338012695,
      "step": 72273
    },
    {
      "epoch": 6.89249038696289e-05,
      "step": 72273,
      "training_step_time": 9.186722755432129
    },
    {
      "epoch": 6.892585754394531e-05,
      "model_forward_time": 2.252129077911377,
      "step": 72274
    },
    {
      "epoch": 6.892585754394531e-05,
      "step": 72274,
      "training_step_time": 8.298611402511597
    },
    {
      "epoch": 6.892681121826172e-05,
      "model_forward_time": 1.6946380138397217,
      "step": 72275
    },
    {
      "epoch": 6.892681121826172e-05,
      "step": 72275,
      "training_step_time": 9.595528602600098
    },
    {
      "epoch": 6.892776489257812e-05,
      "model_forward_time": 1.7084157466888428,
      "step": 72276
    },
    {
      "epoch": 6.892776489257812e-05,
      "step": 72276,
      "training_step_time": 7.752383708953857
    },
    {
      "epoch": 6.892871856689453e-05,
      "model_forward_time": 1.6923015117645264,
      "step": 72277
    },
    {
      "epoch": 6.892871856689453e-05,
      "step": 72277,
      "training_step_time": 8.897485733032227
    },
    {
      "epoch": 6.892967224121094e-05,
      "model_forward_time": 1.6935162544250488,
      "step": 72278
    },
    {
      "epoch": 6.892967224121094e-05,
      "step": 72278,
      "training_step_time": 7.668195009231567
    },
    {
      "epoch": 6.893062591552734e-05,
      "model_forward_time": 1.7008659839630127,
      "step": 72279
    },
    {
      "epoch": 6.893062591552734e-05,
      "step": 72279,
      "training_step_time": 7.692684173583984
    },
    {
      "epoch": 6.893157958984375e-05,
      "grad_norm": 0.10546290129423141,
      "learning_rate": 1.9578547172908185e-06,
      "loss": 0.0589,
      "step": 72280
    },
    {
      "epoch": 6.893157958984375e-05,
      "model_forward_time": 1.7035186290740967,
      "step": 72280
    },
    {
      "dynamics_loss_avg": 0.05042999517172575,
      "epoch": 6.893157958984375e-05,
      "step": 72280
    },
    {
      "action_loss_avg": 0.0011876093863975256,
      "epoch": 6.893157958984375e-05,
      "step": 72280
    },
    {
      "epoch": 6.893157958984375e-05,
      "step": 72280,
      "training_step_time": 7.657382011413574
    },
    {
      "epoch": 6.893253326416016e-05,
      "model_forward_time": 1.7055110931396484,
      "step": 72281
    },
    {
      "epoch": 6.893253326416016e-05,
      "step": 72281,
      "training_step_time": 8.470242500305176
    },
    {
      "epoch": 6.893348693847656e-05,
      "model_forward_time": 1.6963589191436768,
      "step": 72282
    },
    {
      "epoch": 6.893348693847656e-05,
      "step": 72282,
      "training_step_time": 8.670919179916382
    },
    {
      "epoch": 6.893444061279297e-05,
      "model_forward_time": 1.70371413230896,
      "step": 72283
    },
    {
      "epoch": 6.893444061279297e-05,
      "step": 72283,
      "training_step_time": 8.51086974143982
    },
    {
      "epoch": 6.893539428710938e-05,
      "model_forward_time": 2.2416179180145264,
      "step": 72284
    },
    {
      "epoch": 6.893539428710938e-05,
      "step": 72284,
      "training_step_time": 8.865144491195679
    },
    {
      "epoch": 6.893634796142578e-05,
      "model_forward_time": 1.7000558376312256,
      "step": 72285
    },
    {
      "epoch": 6.893634796142578e-05,
      "step": 72285,
      "training_step_time": 9.684892177581787
    },
    {
      "epoch": 6.893730163574219e-05,
      "model_forward_time": 1.691333293914795,
      "step": 72286
    },
    {
      "epoch": 6.893730163574219e-05,
      "step": 72286,
      "training_step_time": 9.677314758300781
    },
    {
      "epoch": 6.89382553100586e-05,
      "model_forward_time": 1.7086060047149658,
      "step": 72287
    },
    {
      "epoch": 6.89382553100586e-05,
      "step": 72287,
      "training_step_time": 8.300311088562012
    },
    {
      "epoch": 6.8939208984375e-05,
      "model_forward_time": 1.696598768234253,
      "step": 72288
    },
    {
      "epoch": 6.8939208984375e-05,
      "step": 72288,
      "training_step_time": 7.496914863586426
    },
    {
      "epoch": 6.894016265869141e-05,
      "model_forward_time": 1.6970815658569336,
      "step": 72289
    },
    {
      "epoch": 6.894016265869141e-05,
      "step": 72289,
      "training_step_time": 7.622495651245117
    },
    {
      "epoch": 6.894111633300782e-05,
      "grad_norm": 0.0981520339846611,
      "learning_rate": 1.9565426763386324e-06,
      "loss": 0.0596,
      "step": 72290
    },
    {
      "epoch": 6.894111633300782e-05,
      "model_forward_time": 1.698744773864746,
      "step": 72290
    },
    {
      "dynamics_loss_avg": 0.07351467721164226,
      "epoch": 6.894111633300782e-05,
      "step": 72290
    },
    {
      "action_loss_avg": 0.0028580277517903595,
      "epoch": 6.894111633300782e-05,
      "step": 72290
    },
    {
      "epoch": 6.894111633300782e-05,
      "step": 72290,
      "training_step_time": 8.453251361846924
    },
    {
      "epoch": 6.894207000732422e-05,
      "model_forward_time": 1.6937353610992432,
      "step": 72291
    },
    {
      "epoch": 6.894207000732422e-05,
      "step": 72291,
      "training_step_time": 7.6296656131744385
    },
    {
      "epoch": 6.894302368164063e-05,
      "model_forward_time": 1.7061481475830078,
      "step": 72292
    },
    {
      "epoch": 6.894302368164063e-05,
      "step": 72292,
      "training_step_time": 7.704390048980713
    },
    {
      "epoch": 6.894397735595704e-05,
      "model_forward_time": 2.234827995300293,
      "step": 72293
    },
    {
      "epoch": 6.894397735595704e-05,
      "step": 72293,
      "training_step_time": 7.835820198059082
    },
    {
      "epoch": 6.894493103027343e-05,
      "model_forward_time": 1.7102396488189697,
      "step": 72294
    },
    {
      "epoch": 6.894493103027343e-05,
      "step": 72294,
      "training_step_time": 7.8723344802856445
    },
    {
      "epoch": 6.894588470458985e-05,
      "model_forward_time": 1.6962783336639404,
      "step": 72295
    },
    {
      "epoch": 6.894588470458985e-05,
      "step": 72295,
      "training_step_time": 7.821139335632324
    },
    {
      "epoch": 6.894683837890626e-05,
      "model_forward_time": 1.699150800704956,
      "step": 72296
    },
    {
      "epoch": 6.894683837890626e-05,
      "step": 72296,
      "training_step_time": 7.721368312835693
    },
    {
      "epoch": 6.894779205322265e-05,
      "model_forward_time": 1.7024030685424805,
      "step": 72297
    },
    {
      "epoch": 6.894779205322265e-05,
      "step": 72297,
      "training_step_time": 8.059762477874756
    },
    {
      "epoch": 6.894874572753906e-05,
      "model_forward_time": 1.7001779079437256,
      "step": 72298
    },
    {
      "epoch": 6.894874572753906e-05,
      "step": 72298,
      "training_step_time": 8.353057384490967
    },
    {
      "epoch": 6.894969940185548e-05,
      "model_forward_time": 1.683706521987915,
      "step": 72299
    },
    {
      "epoch": 6.894969940185548e-05,
      "step": 72299,
      "training_step_time": 8.677857637405396
    },
    {
      "epoch": 6.895065307617187e-05,
      "grad_norm": 0.08752156794071198,
      "learning_rate": 1.955230968214362e-06,
      "loss": 0.0579,
      "step": 72300
    },
    {
      "epoch": 6.895065307617187e-05,
      "model_forward_time": 1.6883814334869385,
      "step": 72300
    },
    {
      "dynamics_loss_avg": 0.051930577494204044,
      "epoch": 6.895065307617187e-05,
      "step": 72300
    },
    {
      "action_loss_avg": 0.0015806347917532548,
      "epoch": 6.895065307617187e-05,
      "step": 72300
    },
    {
      "epoch": 6.895065307617187e-05,
      "step": 72300,
      "training_step_time": 8.207610130310059
    },
    {
      "epoch": 6.895160675048828e-05,
      "model_forward_time": 0.9804391860961914,
      "step": 72301
    },
    {
      "epoch": 6.895160675048828e-05,
      "step": 72301,
      "training_step_time": 9.750861644744873
    },
    {
      "epoch": 6.89525604248047e-05,
      "model_forward_time": 2.2486064434051514,
      "step": 72302
    },
    {
      "epoch": 6.89525604248047e-05,
      "step": 72302,
      "training_step_time": 7.782228708267212
    },
    {
      "epoch": 6.895351409912109e-05,
      "model_forward_time": 1.719632863998413,
      "step": 72303
    },
    {
      "epoch": 6.895351409912109e-05,
      "step": 72303,
      "training_step_time": 7.878162622451782
    },
    {
      "epoch": 6.89544677734375e-05,
      "model_forward_time": 1.6970148086547852,
      "step": 72304
    },
    {
      "epoch": 6.89544677734375e-05,
      "step": 72304,
      "training_step_time": 9.407569885253906
    },
    {
      "epoch": 6.895542144775391e-05,
      "model_forward_time": 1.6992371082305908,
      "step": 72305
    },
    {
      "epoch": 6.895542144775391e-05,
      "step": 72305,
      "training_step_time": 7.769567966461182
    },
    {
      "epoch": 6.895637512207031e-05,
      "model_forward_time": 1.6908047199249268,
      "step": 72306
    },
    {
      "epoch": 6.895637512207031e-05,
      "step": 72306,
      "training_step_time": 7.625353574752808
    },
    {
      "epoch": 6.895732879638672e-05,
      "model_forward_time": 1.6917314529418945,
      "step": 72307
    },
    {
      "epoch": 6.895732879638672e-05,
      "step": 72307,
      "training_step_time": 8.205284595489502
    },
    {
      "epoch": 6.895828247070313e-05,
      "model_forward_time": 1.7018730640411377,
      "step": 72308
    },
    {
      "epoch": 6.895828247070313e-05,
      "step": 72308,
      "training_step_time": 7.719407320022583
    },
    {
      "epoch": 6.895923614501953e-05,
      "model_forward_time": 1.6965863704681396,
      "step": 72309
    },
    {
      "epoch": 6.895923614501953e-05,
      "step": 72309,
      "training_step_time": 7.614695072174072
    },
    {
      "epoch": 6.896018981933594e-05,
      "grad_norm": 0.1035546362400055,
      "learning_rate": 1.9539195930614576e-06,
      "loss": 0.0571,
      "step": 72310
    },
    {
      "epoch": 6.896018981933594e-05,
      "model_forward_time": 1.7050864696502686,
      "step": 72310
    },
    {
      "dynamics_loss_avg": 0.06699439212679863,
      "epoch": 6.896018981933594e-05,
      "step": 72310
    },
    {
      "action_loss_avg": 0.0029652201104909182,
      "epoch": 6.896018981933594e-05,
      "step": 72310
    },
    {
      "epoch": 6.896018981933594e-05,
      "step": 72310,
      "training_step_time": 7.706389904022217
    },
    {
      "epoch": 6.896114349365235e-05,
      "model_forward_time": 2.2450034618377686,
      "step": 72311
    },
    {
      "epoch": 6.896114349365235e-05,
      "step": 72311,
      "training_step_time": 8.89491605758667
    },
    {
      "epoch": 6.896209716796875e-05,
      "model_forward_time": 1.7026207447052002,
      "step": 72312
    },
    {
      "epoch": 6.896209716796875e-05,
      "step": 72312,
      "training_step_time": 8.667805671691895
    },
    {
      "epoch": 6.896305084228516e-05,
      "model_forward_time": 1.6915345191955566,
      "step": 72313
    },
    {
      "epoch": 6.896305084228516e-05,
      "step": 72313,
      "training_step_time": 8.954068183898926
    },
    {
      "epoch": 6.896400451660156e-05,
      "model_forward_time": 1.6949737071990967,
      "step": 72314
    },
    {
      "epoch": 6.896400451660156e-05,
      "step": 72314,
      "training_step_time": 8.239960432052612
    },
    {
      "epoch": 6.896495819091797e-05,
      "model_forward_time": 1.6966400146484375,
      "step": 72315
    },
    {
      "epoch": 6.896495819091797e-05,
      "step": 72315,
      "training_step_time": 7.7188591957092285
    },
    {
      "epoch": 6.896591186523438e-05,
      "model_forward_time": 1.698882818222046,
      "step": 72316
    },
    {
      "epoch": 6.896591186523438e-05,
      "step": 72316,
      "training_step_time": 8.270211696624756
    },
    {
      "epoch": 6.896686553955077e-05,
      "model_forward_time": 1.7032973766326904,
      "step": 72317
    },
    {
      "epoch": 6.896686553955077e-05,
      "step": 72317,
      "training_step_time": 7.7313830852508545
    },
    {
      "epoch": 6.896781921386719e-05,
      "model_forward_time": 1.6998264789581299,
      "step": 72318
    },
    {
      "epoch": 6.896781921386719e-05,
      "step": 72318,
      "training_step_time": 8.15232229232788
    },
    {
      "epoch": 6.89687728881836e-05,
      "model_forward_time": 1.699972152709961,
      "step": 72319
    },
    {
      "epoch": 6.89687728881836e-05,
      "step": 72319,
      "training_step_time": 7.689138650894165
    },
    {
      "epoch": 6.89697265625e-05,
      "grad_norm": 0.07804504781961441,
      "learning_rate": 1.952608551023324e-06,
      "loss": 0.0538,
      "step": 72320
    },
    {
      "epoch": 6.89697265625e-05,
      "model_forward_time": 1.7186498641967773,
      "step": 72320
    },
    {
      "dynamics_loss_avg": 0.05889304634183645,
      "epoch": 6.89697265625e-05,
      "step": 72320
    },
    {
      "action_loss_avg": 0.0017892752715852112,
      "epoch": 6.89697265625e-05,
      "step": 72320
    },
    {
      "epoch": 6.89697265625e-05,
      "step": 72320,
      "training_step_time": 7.875050783157349
    },
    {
      "epoch": 6.89706802368164e-05,
      "model_forward_time": 2.2473127841949463,
      "step": 72321
    },
    {
      "epoch": 6.89706802368164e-05,
      "step": 72321,
      "training_step_time": 8.413007974624634
    },
    {
      "epoch": 6.897163391113282e-05,
      "model_forward_time": 1.7026679515838623,
      "step": 72322
    },
    {
      "epoch": 6.897163391113282e-05,
      "step": 72322,
      "training_step_time": 7.706401586532593
    },
    {
      "epoch": 6.897258758544921e-05,
      "model_forward_time": 1.7040205001831055,
      "step": 72323
    },
    {
      "epoch": 6.897258758544921e-05,
      "step": 72323,
      "training_step_time": 7.790006160736084
    },
    {
      "epoch": 6.897354125976562e-05,
      "model_forward_time": 1.7039024829864502,
      "step": 72324
    },
    {
      "epoch": 6.897354125976562e-05,
      "step": 72324,
      "training_step_time": 7.8239922523498535
    },
    {
      "epoch": 6.897449493408203e-05,
      "model_forward_time": 1.7080223560333252,
      "step": 72325
    },
    {
      "epoch": 6.897449493408203e-05,
      "step": 72325,
      "training_step_time": 7.286372184753418
    },
    {
      "epoch": 6.897544860839843e-05,
      "model_forward_time": 1.7094318866729736,
      "step": 72326
    },
    {
      "epoch": 6.897544860839843e-05,
      "step": 72326,
      "training_step_time": 7.724380970001221
    },
    {
      "epoch": 6.897640228271484e-05,
      "model_forward_time": 1.6983025074005127,
      "step": 72327
    },
    {
      "epoch": 6.897640228271484e-05,
      "step": 72327,
      "training_step_time": 7.793111085891724
    },
    {
      "epoch": 6.897735595703125e-05,
      "model_forward_time": 1.7069127559661865,
      "step": 72328
    },
    {
      "epoch": 6.897735595703125e-05,
      "step": 72328,
      "training_step_time": 8.259260416030884
    },
    {
      "epoch": 6.897830963134765e-05,
      "model_forward_time": 1.3355028629302979,
      "step": 72329
    },
    {
      "epoch": 6.897830963134765e-05,
      "step": 72329,
      "training_step_time": 7.725495100021362
    },
    {
      "epoch": 6.897926330566406e-05,
      "grad_norm": 0.08799906820058823,
      "learning_rate": 1.9512978422433397e-06,
      "loss": 0.0585,
      "step": 72330
    },
    {
      "epoch": 6.897926330566406e-05,
      "model_forward_time": 2.232335090637207,
      "step": 72330
    },
    {
      "dynamics_loss_avg": 0.05538357328623533,
      "epoch": 6.897926330566406e-05,
      "step": 72330
    },
    {
      "action_loss_avg": 0.0014408836577786133,
      "epoch": 6.897926330566406e-05,
      "step": 72330
    },
    {
      "epoch": 6.897926330566406e-05,
      "step": 72330,
      "training_step_time": 7.814614534378052
    },
    {
      "epoch": 6.898021697998047e-05,
      "model_forward_time": 1.7246131896972656,
      "step": 72331
    },
    {
      "epoch": 6.898021697998047e-05,
      "step": 72331,
      "training_step_time": 8.327388763427734
    },
    {
      "epoch": 6.898117065429687e-05,
      "model_forward_time": 1.697472333908081,
      "step": 72332
    },
    {
      "epoch": 6.898117065429687e-05,
      "step": 72332,
      "training_step_time": 7.713786363601685
    },
    {
      "epoch": 6.898212432861328e-05,
      "model_forward_time": 1.7212107181549072,
      "step": 72333
    },
    {
      "epoch": 6.898212432861328e-05,
      "step": 72333,
      "training_step_time": 8.125201940536499
    },
    {
      "epoch": 6.898307800292969e-05,
      "model_forward_time": 1.7116267681121826,
      "step": 72334
    },
    {
      "epoch": 6.898307800292969e-05,
      "step": 72334,
      "training_step_time": 7.67182731628418
    },
    {
      "epoch": 6.898403167724609e-05,
      "model_forward_time": 1.7022886276245117,
      "step": 72335
    },
    {
      "epoch": 6.898403167724609e-05,
      "step": 72335,
      "training_step_time": 7.646638631820679
    },
    {
      "epoch": 6.89849853515625e-05,
      "model_forward_time": 1.6938300132751465,
      "step": 72336
    },
    {
      "epoch": 6.89849853515625e-05,
      "step": 72336,
      "training_step_time": 8.370958805084229
    },
    {
      "epoch": 6.898593902587891e-05,
      "model_forward_time": 1.6985549926757812,
      "step": 72337
    },
    {
      "epoch": 6.898593902587891e-05,
      "step": 72337,
      "training_step_time": 7.64794135093689
    },
    {
      "epoch": 6.89868927001953e-05,
      "model_forward_time": 1.3326213359832764,
      "step": 72338
    },
    {
      "epoch": 6.89868927001953e-05,
      "step": 72338,
      "training_step_time": 8.920195817947388
    },
    {
      "epoch": 6.898784637451172e-05,
      "model_forward_time": 2.2522027492523193,
      "step": 72339
    },
    {
      "epoch": 6.898784637451172e-05,
      "step": 72339,
      "training_step_time": 10.052185535430908
    },
    {
      "epoch": 6.898880004882813e-05,
      "grad_norm": 0.30160248279571533,
      "learning_rate": 1.9499874668648376e-06,
      "loss": 0.0559,
      "step": 72340
    },
    {
      "epoch": 6.898880004882813e-05,
      "model_forward_time": 1.7164511680603027,
      "step": 72340
    },
    {
      "dynamics_loss_avg": 0.055486094951629636,
      "epoch": 6.898880004882813e-05,
      "step": 72340
    },
    {
      "action_loss_avg": 0.0022367656347341835,
      "epoch": 6.898880004882813e-05,
      "step": 72340
    },
    {
      "epoch": 6.898880004882813e-05,
      "step": 72340,
      "training_step_time": 9.673738718032837
    },
    {
      "epoch": 6.898975372314453e-05,
      "model_forward_time": 1.6932132244110107,
      "step": 72341
    },
    {
      "epoch": 6.898975372314453e-05,
      "step": 72341,
      "training_step_time": 8.763418912887573
    },
    {
      "epoch": 6.899070739746094e-05,
      "model_forward_time": 1.708935260772705,
      "step": 72342
    },
    {
      "epoch": 6.899070739746094e-05,
      "step": 72342,
      "training_step_time": 8.848323822021484
    },
    {
      "epoch": 6.899166107177735e-05,
      "model_forward_time": 1.6975998878479004,
      "step": 72343
    },
    {
      "epoch": 6.899166107177735e-05,
      "step": 72343,
      "training_step_time": 7.758270502090454
    },
    {
      "epoch": 6.899261474609374e-05,
      "model_forward_time": 1.7030882835388184,
      "step": 72344
    },
    {
      "epoch": 6.899261474609374e-05,
      "step": 72344,
      "training_step_time": 8.428248405456543
    },
    {
      "epoch": 6.899356842041016e-05,
      "model_forward_time": 1.7009565830230713,
      "step": 72345
    },
    {
      "epoch": 6.899356842041016e-05,
      "step": 72345,
      "training_step_time": 8.522010803222656
    },
    {
      "epoch": 6.899452209472657e-05,
      "model_forward_time": 1.7000186443328857,
      "step": 72346
    },
    {
      "epoch": 6.899452209472657e-05,
      "step": 72346,
      "training_step_time": 7.723844766616821
    },
    {
      "epoch": 6.899547576904296e-05,
      "model_forward_time": 1.6919877529144287,
      "step": 72347
    },
    {
      "epoch": 6.899547576904296e-05,
      "step": 72347,
      "training_step_time": 9.128979682922363
    },
    {
      "epoch": 6.899642944335937e-05,
      "model_forward_time": 1.7224714756011963,
      "step": 72348
    },
    {
      "epoch": 6.899642944335937e-05,
      "step": 72348,
      "training_step_time": 8.955960035324097
    },
    {
      "epoch": 6.899738311767578e-05,
      "model_forward_time": 1.715484619140625,
      "step": 72349
    },
    {
      "epoch": 6.899738311767578e-05,
      "step": 72349,
      "training_step_time": 7.968013048171997
    },
    {
      "epoch": 6.899833679199218e-05,
      "grad_norm": 0.07421598583459854,
      "learning_rate": 1.9486774250311215e-06,
      "loss": 0.0564,
      "step": 72350
    },
    {
      "epoch": 6.899833679199218e-05,
      "model_forward_time": 1.6980829238891602,
      "step": 72350
    },
    {
      "dynamics_loss_avg": 0.06709657292813062,
      "epoch": 6.899833679199218e-05,
      "step": 72350
    },
    {
      "action_loss_avg": 0.0027204672514926642,
      "epoch": 6.899833679199218e-05,
      "step": 72350
    },
    {
      "epoch": 6.899833679199218e-05,
      "step": 72350,
      "training_step_time": 8.358401536941528
    },
    {
      "epoch": 6.899929046630859e-05,
      "model_forward_time": 1.7032039165496826,
      "step": 72351
    },
    {
      "epoch": 6.899929046630859e-05,
      "step": 72351,
      "training_step_time": 7.750228643417358
    },
    {
      "epoch": 6.9000244140625e-05,
      "model_forward_time": 1.704253911972046,
      "step": 72352
    },
    {
      "epoch": 6.9000244140625e-05,
      "step": 72352,
      "training_step_time": 7.643220901489258
    },
    {
      "epoch": 6.90011978149414e-05,
      "model_forward_time": 1.6973557472229004,
      "step": 72353
    },
    {
      "epoch": 6.90011978149414e-05,
      "step": 72353,
      "training_step_time": 7.616287708282471
    },
    {
      "epoch": 6.900215148925781e-05,
      "model_forward_time": 1.6800422668457031,
      "step": 72354
    },
    {
      "epoch": 6.900215148925781e-05,
      "step": 72354,
      "training_step_time": 7.727631568908691
    },
    {
      "epoch": 6.900310516357422e-05,
      "model_forward_time": 1.700164794921875,
      "step": 72355
    },
    {
      "epoch": 6.900310516357422e-05,
      "step": 72355,
      "training_step_time": 8.657529592514038
    },
    {
      "epoch": 6.900405883789062e-05,
      "model_forward_time": 1.7027690410614014,
      "step": 72356
    },
    {
      "epoch": 6.900405883789062e-05,
      "step": 72356,
      "training_step_time": 8.452696084976196
    },
    {
      "epoch": 6.900501251220703e-05,
      "model_forward_time": 1.7905681133270264,
      "step": 72357
    },
    {
      "epoch": 6.900501251220703e-05,
      "step": 72357,
      "training_step_time": 7.846070766448975
    },
    {
      "epoch": 6.900596618652344e-05,
      "model_forward_time": 2.2335493564605713,
      "step": 72358
    },
    {
      "epoch": 6.900596618652344e-05,
      "step": 72358,
      "training_step_time": 8.658020973205566
    },
    {
      "epoch": 6.900691986083984e-05,
      "model_forward_time": 1.7120652198791504,
      "step": 72359
    },
    {
      "epoch": 6.900691986083984e-05,
      "step": 72359,
      "training_step_time": 8.577186584472656
    },
    {
      "epoch": 6.900787353515625e-05,
      "grad_norm": 0.10942821949720383,
      "learning_rate": 1.947367716885454e-06,
      "loss": 0.0581,
      "step": 72360
    },
    {
      "epoch": 6.900787353515625e-05,
      "model_forward_time": 1.7000067234039307,
      "step": 72360
    },
    {
      "dynamics_loss_avg": 0.061016397923231123,
      "epoch": 6.900787353515625e-05,
      "step": 72360
    },
    {
      "action_loss_avg": 0.0015237661864375696,
      "epoch": 6.900787353515625e-05,
      "step": 72360
    },
    {
      "epoch": 6.900787353515625e-05,
      "step": 72360,
      "training_step_time": 7.812275171279907
    },
    {
      "epoch": 6.900882720947266e-05,
      "model_forward_time": 1.716672420501709,
      "step": 72361
    },
    {
      "epoch": 6.900882720947266e-05,
      "step": 72361,
      "training_step_time": 7.674551725387573
    },
    {
      "epoch": 6.900978088378906e-05,
      "model_forward_time": 1.706388235092163,
      "step": 72362
    },
    {
      "epoch": 6.900978088378906e-05,
      "step": 72362,
      "training_step_time": 7.771217107772827
    },
    {
      "epoch": 6.901073455810547e-05,
      "model_forward_time": 1.7068583965301514,
      "step": 72363
    },
    {
      "epoch": 6.901073455810547e-05,
      "step": 72363,
      "training_step_time": 7.7251856327056885
    },
    {
      "epoch": 6.901168823242188e-05,
      "model_forward_time": 1.711277723312378,
      "step": 72364
    },
    {
      "epoch": 6.901168823242188e-05,
      "step": 72364,
      "training_step_time": 7.811547517776489
    },
    {
      "epoch": 6.901264190673828e-05,
      "model_forward_time": 1.706392526626587,
      "step": 72365
    },
    {
      "epoch": 6.901264190673828e-05,
      "step": 72365,
      "training_step_time": 7.760226726531982
    },
    {
      "epoch": 6.901359558105469e-05,
      "model_forward_time": 1.7770044803619385,
      "step": 72366
    },
    {
      "epoch": 6.901359558105469e-05,
      "step": 72366,
      "training_step_time": 8.002706527709961
    },
    {
      "epoch": 6.90145492553711e-05,
      "model_forward_time": 2.2495226860046387,
      "step": 72367
    },
    {
      "epoch": 6.90145492553711e-05,
      "step": 72367,
      "training_step_time": 8.837093830108643
    },
    {
      "epoch": 6.90155029296875e-05,
      "model_forward_time": 1.7020471096038818,
      "step": 72368
    },
    {
      "epoch": 6.90155029296875e-05,
      "step": 72368,
      "training_step_time": 7.902263164520264
    },
    {
      "epoch": 6.90164566040039e-05,
      "model_forward_time": 1.7215728759765625,
      "step": 72369
    },
    {
      "epoch": 6.90164566040039e-05,
      "step": 72369,
      "training_step_time": 8.305086374282837
    },
    {
      "epoch": 6.901741027832032e-05,
      "grad_norm": 0.07589278370141983,
      "learning_rate": 1.946058342571065e-06,
      "loss": 0.0559,
      "step": 72370
    },
    {
      "epoch": 6.901741027832032e-05,
      "model_forward_time": 1.702155590057373,
      "step": 72370
    },
    {
      "dynamics_loss_avg": 0.04657268468290567,
      "epoch": 6.901741027832032e-05,
      "step": 72370
    },
    {
      "action_loss_avg": 0.002383286348776892,
      "epoch": 6.901741027832032e-05,
      "step": 72370
    },
    {
      "epoch": 6.901741027832032e-05,
      "step": 72370,
      "training_step_time": 7.7952494621276855
    },
    {
      "epoch": 6.901836395263671e-05,
      "model_forward_time": 1.7095177173614502,
      "step": 72371
    },
    {
      "epoch": 6.901836395263671e-05,
      "step": 72371,
      "training_step_time": 7.742969989776611
    },
    {
      "epoch": 6.901931762695312e-05,
      "model_forward_time": 1.714203119277954,
      "step": 72372
    },
    {
      "epoch": 6.901931762695312e-05,
      "step": 72372,
      "training_step_time": 7.6367151737213135
    },
    {
      "epoch": 6.902027130126954e-05,
      "model_forward_time": 1.7112247943878174,
      "step": 72373
    },
    {
      "epoch": 6.902027130126954e-05,
      "step": 72373,
      "training_step_time": 8.228893041610718
    },
    {
      "epoch": 6.902122497558593e-05,
      "model_forward_time": 1.7049565315246582,
      "step": 72374
    },
    {
      "epoch": 6.902122497558593e-05,
      "step": 72374,
      "training_step_time": 8.666615724563599
    },
    {
      "epoch": 6.902217864990234e-05,
      "model_forward_time": 1.7875540256500244,
      "step": 72375
    },
    {
      "epoch": 6.902217864990234e-05,
      "step": 72375,
      "training_step_time": 7.943042516708374
    },
    {
      "epoch": 6.902313232421875e-05,
      "model_forward_time": 2.2549257278442383,
      "step": 72376
    },
    {
      "epoch": 6.902313232421875e-05,
      "step": 72376,
      "training_step_time": 8.69391679763794
    },
    {
      "epoch": 6.902408599853515e-05,
      "model_forward_time": 1.6947176456451416,
      "step": 72377
    },
    {
      "epoch": 6.902408599853515e-05,
      "step": 72377,
      "training_step_time": 7.847782850265503
    },
    {
      "epoch": 6.902503967285156e-05,
      "model_forward_time": 1.704207181930542,
      "step": 72378
    },
    {
      "epoch": 6.902503967285156e-05,
      "step": 72378,
      "training_step_time": 7.727079629898071
    },
    {
      "epoch": 6.902599334716797e-05,
      "model_forward_time": 1.3388895988464355,
      "step": 72379
    },
    {
      "epoch": 6.902599334716797e-05,
      "step": 72379,
      "training_step_time": 7.7244226932525635
    },
    {
      "epoch": 6.902694702148437e-05,
      "grad_norm": 0.30931687355041504,
      "learning_rate": 1.9447493022311427e-06,
      "loss": 0.0585,
      "step": 72380
    },
    {
      "epoch": 6.902694702148437e-05,
      "model_forward_time": 1.6991102695465088,
      "step": 72380
    },
    {
      "dynamics_loss_avg": 0.07498886324465275,
      "epoch": 6.902694702148437e-05,
      "step": 72380
    },
    {
      "action_loss_avg": 0.003911353158764541,
      "epoch": 6.902694702148437e-05,
      "step": 72380
    },
    {
      "epoch": 6.902694702148437e-05,
      "step": 72380,
      "training_step_time": 7.649864673614502
    },
    {
      "epoch": 6.902790069580078e-05,
      "model_forward_time": 1.3352036476135254,
      "step": 72381
    },
    {
      "epoch": 6.902790069580078e-05,
      "step": 72381,
      "training_step_time": 7.501399517059326
    },
    {
      "epoch": 6.902885437011719e-05,
      "model_forward_time": 1.335576057434082,
      "step": 72382
    },
    {
      "epoch": 6.902885437011719e-05,
      "step": 72382,
      "training_step_time": 8.40170669555664
    },
    {
      "epoch": 6.902980804443359e-05,
      "model_forward_time": 1.7053234577178955,
      "step": 72383
    },
    {
      "epoch": 6.902980804443359e-05,
      "step": 72383,
      "training_step_time": 8.697890043258667
    },
    {
      "epoch": 6.903076171875e-05,
      "model_forward_time": 1.7874493598937988,
      "step": 72384
    },
    {
      "epoch": 6.903076171875e-05,
      "step": 72384,
      "training_step_time": 9.627754926681519
    },
    {
      "epoch": 6.903171539306641e-05,
      "model_forward_time": 1.7085628509521484,
      "step": 72385
    },
    {
      "epoch": 6.903171539306641e-05,
      "step": 72385,
      "training_step_time": 7.675710439682007
    },
    {
      "epoch": 6.903266906738281e-05,
      "model_forward_time": 2.218782901763916,
      "step": 72386
    },
    {
      "epoch": 6.903266906738281e-05,
      "step": 72386,
      "training_step_time": 8.849440574645996
    },
    {
      "epoch": 6.903362274169922e-05,
      "model_forward_time": 1.7083542346954346,
      "step": 72387
    },
    {
      "epoch": 6.903362274169922e-05,
      "step": 72387,
      "training_step_time": 7.69287371635437
    },
    {
      "epoch": 6.903457641601563e-05,
      "model_forward_time": 1.7061591148376465,
      "step": 72388
    },
    {
      "epoch": 6.903457641601563e-05,
      "step": 72388,
      "training_step_time": 7.751811981201172
    },
    {
      "epoch": 6.903553009033203e-05,
      "model_forward_time": 1.713369369506836,
      "step": 72389
    },
    {
      "epoch": 6.903553009033203e-05,
      "step": 72389,
      "training_step_time": 8.752868890762329
    },
    {
      "epoch": 6.903648376464844e-05,
      "grad_norm": 0.0975077822804451,
      "learning_rate": 1.943440596008844e-06,
      "loss": 0.0592,
      "step": 72390
    },
    {
      "epoch": 6.903648376464844e-05,
      "model_forward_time": 1.7075467109680176,
      "step": 72390
    },
    {
      "dynamics_loss_avg": 0.060691472887992856,
      "epoch": 6.903648376464844e-05,
      "step": 72390
    },
    {
      "action_loss_avg": 0.0010406263783806935,
      "epoch": 6.903648376464844e-05,
      "step": 72390
    },
    {
      "epoch": 6.903648376464844e-05,
      "step": 72390,
      "training_step_time": 7.591878652572632
    },
    {
      "epoch": 6.903743743896485e-05,
      "model_forward_time": 1.6988027095794678,
      "step": 72391
    },
    {
      "epoch": 6.903743743896485e-05,
      "step": 72391,
      "training_step_time": 8.366596221923828
    },
    {
      "epoch": 6.903839111328125e-05,
      "model_forward_time": 1.7112641334533691,
      "step": 72392
    },
    {
      "epoch": 6.903839111328125e-05,
      "step": 72392,
      "training_step_time": 8.497102975845337
    },
    {
      "epoch": 6.903934478759766e-05,
      "model_forward_time": 1.7924962043762207,
      "step": 72393
    },
    {
      "epoch": 6.903934478759766e-05,
      "step": 72393,
      "training_step_time": 7.6747541427612305
    },
    {
      "epoch": 6.904029846191407e-05,
      "model_forward_time": 1.7232048511505127,
      "step": 72394
    },
    {
      "epoch": 6.904029846191407e-05,
      "step": 72394,
      "training_step_time": 9.214951276779175
    },
    {
      "epoch": 6.904125213623046e-05,
      "model_forward_time": 2.238651990890503,
      "step": 72395
    },
    {
      "epoch": 6.904125213623046e-05,
      "step": 72395,
      "training_step_time": 8.7618567943573
    },
    {
      "epoch": 6.904220581054688e-05,
      "model_forward_time": 1.7031030654907227,
      "step": 72396
    },
    {
      "epoch": 6.904220581054688e-05,
      "step": 72396,
      "training_step_time": 8.365454196929932
    },
    {
      "epoch": 6.904315948486329e-05,
      "model_forward_time": 1.7093496322631836,
      "step": 72397
    },
    {
      "epoch": 6.904315948486329e-05,
      "step": 72397,
      "training_step_time": 7.794685363769531
    },
    {
      "epoch": 6.904411315917968e-05,
      "model_forward_time": 1.7161691188812256,
      "step": 72398
    },
    {
      "epoch": 6.904411315917968e-05,
      "step": 72398,
      "training_step_time": 7.671806573867798
    },
    {
      "epoch": 6.90450668334961e-05,
      "model_forward_time": 1.706857442855835,
      "step": 72399
    },
    {
      "epoch": 6.90450668334961e-05,
      "step": 72399,
      "training_step_time": 8.348469018936157
    },
    {
      "epoch": 6.90460205078125e-05,
      "grad_norm": 0.1073174849152565,
      "learning_rate": 1.9421322240472856e-06,
      "loss": 0.0581,
      "step": 72400
    },
    {
      "epoch": 6.90460205078125e-05,
      "model_forward_time": 1.7049798965454102,
      "step": 72400
    },
    {
      "dynamics_loss_avg": 0.0697956707328558,
      "epoch": 6.90460205078125e-05,
      "step": 72400
    },
    {
      "action_loss_avg": 0.001556978395092301,
      "epoch": 6.90460205078125e-05,
      "step": 72400
    },
    {
      "epoch": 6.90460205078125e-05,
      "step": 72400,
      "training_step_time": 9.358264207839966
    },
    {
      "epoch": 6.90469741821289e-05,
      "model_forward_time": 1.704383373260498,
      "step": 72401
    },
    {
      "epoch": 6.90469741821289e-05,
      "step": 72401,
      "training_step_time": 9.677684783935547
    },
    {
      "epoch": 6.904792785644531e-05,
      "model_forward_time": 1.7108275890350342,
      "step": 72402
    },
    {
      "epoch": 6.904792785644531e-05,
      "step": 72402,
      "training_step_time": 6.774429082870483
    },
    {
      "epoch": 6.904888153076172e-05,
      "model_forward_time": 1.7121973037719727,
      "step": 72403
    },
    {
      "epoch": 6.904888153076172e-05,
      "step": 72403,
      "training_step_time": 6.764001131057739
    },
    {
      "epoch": 6.904983520507812e-05,
      "model_forward_time": 1.7062015533447266,
      "step": 72404
    },
    {
      "epoch": 6.904983520507812e-05,
      "step": 72404,
      "training_step_time": 6.756351947784424
    },
    {
      "epoch": 6.905078887939453e-05,
      "model_forward_time": 1.7044594287872314,
      "step": 72405
    },
    {
      "epoch": 6.905078887939453e-05,
      "step": 72405,
      "training_step_time": 6.7993223667144775
    },
    {
      "epoch": 6.905174255371094e-05,
      "model_forward_time": 1.703683614730835,
      "step": 72406
    },
    {
      "epoch": 6.905174255371094e-05,
      "step": 72406,
      "training_step_time": 6.76830530166626
    },
    {
      "epoch": 6.905269622802734e-05,
      "model_forward_time": 1.7074968814849854,
      "step": 72407
    },
    {
      "epoch": 6.905269622802734e-05,
      "step": 72407,
      "training_step_time": 6.733227968215942
    },
    {
      "epoch": 6.905364990234375e-05,
      "model_forward_time": 1.7084565162658691,
      "step": 72408
    },
    {
      "epoch": 6.905364990234375e-05,
      "step": 72408,
      "training_step_time": 6.753905534744263
    },
    {
      "epoch": 6.905460357666016e-05,
      "model_forward_time": 1.7065110206604004,
      "step": 72409
    },
    {
      "epoch": 6.905460357666016e-05,
      "step": 72409,
      "training_step_time": 6.699167013168335
    },
    {
      "epoch": 6.905555725097656e-05,
      "grad_norm": 0.11958146840333939,
      "learning_rate": 1.9408241864895494e-06,
      "loss": 0.0571,
      "step": 72410
    },
    {
      "epoch": 6.905555725097656e-05,
      "model_forward_time": 2.2444698810577393,
      "step": 72410
    },
    {
      "dynamics_loss_avg": 0.06459213588386774,
      "epoch": 6.905555725097656e-05,
      "step": 72410
    },
    {
      "action_loss_avg": 0.001858328521484509,
      "epoch": 6.905555725097656e-05,
      "step": 72410
    },
    {
      "epoch": 6.905555725097656e-05,
      "step": 72410,
      "training_step_time": 7.597902774810791
    },
    {
      "epoch": 6.905651092529297e-05,
      "model_forward_time": 1.700747013092041,
      "step": 72411
    },
    {
      "epoch": 6.905651092529297e-05,
      "step": 72411,
      "training_step_time": 7.897642374038696
    },
    {
      "epoch": 6.905746459960938e-05,
      "model_forward_time": 1.712526559829712,
      "step": 72412
    },
    {
      "epoch": 6.905746459960938e-05,
      "step": 72412,
      "training_step_time": 6.816119909286499
    },
    {
      "epoch": 6.905841827392578e-05,
      "model_forward_time": 1.7030029296875,
      "step": 72413
    },
    {
      "epoch": 6.905841827392578e-05,
      "step": 72413,
      "training_step_time": 6.7653608322143555
    },
    {
      "epoch": 6.905937194824219e-05,
      "model_forward_time": 1.709808111190796,
      "step": 72414
    },
    {
      "epoch": 6.905937194824219e-05,
      "step": 72414,
      "training_step_time": 6.795482873916626
    },
    {
      "epoch": 6.90603256225586e-05,
      "model_forward_time": 1.703944444656372,
      "step": 72415
    },
    {
      "epoch": 6.90603256225586e-05,
      "step": 72415,
      "training_step_time": 6.782252788543701
    },
    {
      "epoch": 6.9061279296875e-05,
      "model_forward_time": 1.7034087181091309,
      "step": 72416
    },
    {
      "epoch": 6.9061279296875e-05,
      "step": 72416,
      "training_step_time": 6.732973337173462
    },
    {
      "epoch": 6.906223297119141e-05,
      "model_forward_time": 1.704186201095581,
      "step": 72417
    },
    {
      "epoch": 6.906223297119141e-05,
      "step": 72417,
      "training_step_time": 6.742354154586792
    },
    {
      "epoch": 6.906318664550782e-05,
      "model_forward_time": 1.7190423011779785,
      "step": 72418
    },
    {
      "epoch": 6.906318664550782e-05,
      "step": 72418,
      "training_step_time": 6.769448757171631
    },
    {
      "epoch": 6.906414031982422e-05,
      "model_forward_time": 2.2487425804138184,
      "step": 72419
    },
    {
      "epoch": 6.906414031982422e-05,
      "step": 72419,
      "training_step_time": 7.571851491928101
    },
    {
      "epoch": 6.906509399414063e-05,
      "grad_norm": 0.08032605797052383,
      "learning_rate": 1.939516483478681e-06,
      "loss": 0.0557,
      "step": 72420
    },
    {
      "epoch": 6.906509399414063e-05,
      "model_forward_time": 1.703003168106079,
      "step": 72420
    },
    {
      "dynamics_loss_avg": 0.07139277383685112,
      "epoch": 6.906509399414063e-05,
      "step": 72420
    },
    {
      "action_loss_avg": 0.005830354377394542,
      "epoch": 6.906509399414063e-05,
      "step": 72420
    },
    {
      "epoch": 6.906509399414063e-05,
      "step": 72420,
      "training_step_time": 7.81256890296936
    },
    {
      "epoch": 6.906604766845704e-05,
      "model_forward_time": 1.7065293788909912,
      "step": 72421
    },
    {
      "epoch": 6.906604766845704e-05,
      "step": 72421,
      "training_step_time": 6.720549821853638
    },
    {
      "epoch": 6.906700134277343e-05,
      "model_forward_time": 1.7038860321044922,
      "step": 72422
    },
    {
      "epoch": 6.906700134277343e-05,
      "step": 72422,
      "training_step_time": 6.6944029331207275
    },
    {
      "epoch": 6.906795501708985e-05,
      "model_forward_time": 1.7056560516357422,
      "step": 72423
    },
    {
      "epoch": 6.906795501708985e-05,
      "step": 72423,
      "training_step_time": 6.680782079696655
    },
    {
      "epoch": 6.906890869140626e-05,
      "model_forward_time": 1.334486961364746,
      "step": 72424
    },
    {
      "epoch": 6.906890869140626e-05,
      "step": 72424,
      "training_step_time": 6.699019908905029
    },
    {
      "epoch": 6.906986236572265e-05,
      "model_forward_time": 1.6913228034973145,
      "step": 72425
    },
    {
      "epoch": 6.906986236572265e-05,
      "step": 72425,
      "training_step_time": 6.757511138916016
    },
    {
      "epoch": 6.907081604003906e-05,
      "model_forward_time": 1.706167459487915,
      "step": 72426
    },
    {
      "epoch": 6.907081604003906e-05,
      "step": 72426,
      "training_step_time": 6.690653085708618
    },
    {
      "epoch": 6.907176971435548e-05,
      "model_forward_time": 1.7164716720581055,
      "step": 72427
    },
    {
      "epoch": 6.907176971435548e-05,
      "step": 72427,
      "training_step_time": 6.70451021194458
    },
    {
      "epoch": 6.907272338867187e-05,
      "model_forward_time": 2.280271053314209,
      "step": 72428
    },
    {
      "epoch": 6.907272338867187e-05,
      "step": 72428,
      "training_step_time": 7.6153953075408936
    },
    {
      "epoch": 6.907367706298828e-05,
      "model_forward_time": 1.702176809310913,
      "step": 72429
    },
    {
      "epoch": 6.907367706298828e-05,
      "step": 72429,
      "training_step_time": 7.80242657661438
    },
    {
      "epoch": 6.90746307373047e-05,
      "grad_norm": 0.07693719863891602,
      "learning_rate": 1.9382091151576905e-06,
      "loss": 0.0532,
      "step": 72430
    },
    {
      "epoch": 6.90746307373047e-05,
      "model_forward_time": 1.7083775997161865,
      "step": 72430
    },
    {
      "dynamics_loss_avg": 0.05379279861226678,
      "epoch": 6.90746307373047e-05,
      "step": 72430
    },
    {
      "action_loss_avg": 0.0029088687675539403,
      "epoch": 6.90746307373047e-05,
      "step": 72430
    },
    {
      "epoch": 6.90746307373047e-05,
      "step": 72430,
      "training_step_time": 7.427203416824341
    },
    {
      "epoch": 6.907558441162109e-05,
      "model_forward_time": 1.3230407238006592,
      "step": 72431
    },
    {
      "epoch": 6.907558441162109e-05,
      "step": 72431,
      "training_step_time": 6.715940713882446
    },
    {
      "epoch": 6.90765380859375e-05,
      "model_forward_time": 1.7220535278320312,
      "step": 72432
    },
    {
      "epoch": 6.90765380859375e-05,
      "step": 72432,
      "training_step_time": 6.705660581588745
    },
    {
      "epoch": 6.907749176025391e-05,
      "model_forward_time": 1.7273015975952148,
      "step": 72433
    },
    {
      "epoch": 6.907749176025391e-05,
      "step": 72433,
      "training_step_time": 6.663972854614258
    },
    {
      "epoch": 6.907844543457031e-05,
      "model_forward_time": 1.704819917678833,
      "step": 72434
    },
    {
      "epoch": 6.907844543457031e-05,
      "step": 72434,
      "training_step_time": 6.704573154449463
    },
    {
      "epoch": 6.907939910888672e-05,
      "model_forward_time": 1.7060356140136719,
      "step": 72435
    },
    {
      "epoch": 6.907939910888672e-05,
      "step": 72435,
      "training_step_time": 6.756348133087158
    },
    {
      "epoch": 6.908035278320313e-05,
      "model_forward_time": 1.7078399658203125,
      "step": 72436
    },
    {
      "epoch": 6.908035278320313e-05,
      "step": 72436,
      "training_step_time": 6.722130298614502
    },
    {
      "epoch": 6.908130645751953e-05,
      "model_forward_time": 2.281797170639038,
      "step": 72437
    },
    {
      "epoch": 6.908130645751953e-05,
      "step": 72437,
      "training_step_time": 7.63213849067688
    },
    {
      "epoch": 6.908226013183594e-05,
      "model_forward_time": 1.7132985591888428,
      "step": 72438
    },
    {
      "epoch": 6.908226013183594e-05,
      "step": 72438,
      "training_step_time": 9.453688383102417
    },
    {
      "epoch": 6.908321380615234e-05,
      "model_forward_time": 1.71238374710083,
      "step": 72439
    },
    {
      "epoch": 6.908321380615234e-05,
      "step": 72439,
      "training_step_time": 7.7211925983428955
    },
    {
      "epoch": 6.908416748046875e-05,
      "grad_norm": 0.07667618244886398,
      "learning_rate": 1.9369020816695455e-06,
      "loss": 0.0578,
      "step": 72440
    },
    {
      "epoch": 6.908416748046875e-05,
      "model_forward_time": 1.7159764766693115,
      "step": 72440
    },
    {
      "dynamics_loss_avg": 0.05996016524732113,
      "epoch": 6.908416748046875e-05,
      "step": 72440
    },
    {
      "action_loss_avg": 0.004243938671424985,
      "epoch": 6.908416748046875e-05,
      "step": 72440
    },
    {
      "epoch": 6.908416748046875e-05,
      "step": 72440,
      "training_step_time": 6.7712647914886475
    },
    {
      "epoch": 6.908512115478516e-05,
      "model_forward_time": 1.707280158996582,
      "step": 72441
    },
    {
      "epoch": 6.908512115478516e-05,
      "step": 72441,
      "training_step_time": 6.753829479217529
    },
    {
      "epoch": 6.908607482910156e-05,
      "model_forward_time": 1.7043838500976562,
      "step": 72442
    },
    {
      "epoch": 6.908607482910156e-05,
      "step": 72442,
      "training_step_time": 6.805043935775757
    },
    {
      "epoch": 6.908702850341797e-05,
      "model_forward_time": 1.7079839706420898,
      "step": 72443
    },
    {
      "epoch": 6.908702850341797e-05,
      "step": 72443,
      "training_step_time": 6.7232887744903564
    },
    {
      "epoch": 6.908798217773438e-05,
      "model_forward_time": 1.3359830379486084,
      "step": 72444
    },
    {
      "epoch": 6.908798217773438e-05,
      "step": 72444,
      "training_step_time": 6.731260538101196
    },
    {
      "epoch": 6.908893585205077e-05,
      "model_forward_time": 1.720196008682251,
      "step": 72445
    },
    {
      "epoch": 6.908893585205077e-05,
      "step": 72445,
      "training_step_time": 6.773909568786621
    },
    {
      "epoch": 6.908988952636719e-05,
      "model_forward_time": 2.299387216567993,
      "step": 72446
    },
    {
      "epoch": 6.908988952636719e-05,
      "step": 72446,
      "training_step_time": 8.348951816558838
    },
    {
      "epoch": 6.90908432006836e-05,
      "model_forward_time": 1.7064402103424072,
      "step": 72447
    },
    {
      "epoch": 6.90908432006836e-05,
      "step": 72447,
      "training_step_time": 10.531309843063354
    },
    {
      "epoch": 6.9091796875e-05,
      "model_forward_time": 1.3377161026000977,
      "step": 72448
    },
    {
      "epoch": 6.9091796875e-05,
      "step": 72448,
      "training_step_time": 8.841354370117188
    },
    {
      "epoch": 6.90927505493164e-05,
      "model_forward_time": 1.7048258781433105,
      "step": 72449
    },
    {
      "epoch": 6.90927505493164e-05,
      "step": 72449,
      "training_step_time": 7.381434679031372
    },
    {
      "epoch": 6.909370422363282e-05,
      "grad_norm": 0.08628131449222565,
      "learning_rate": 1.935595383157184e-06,
      "loss": 0.0533,
      "step": 72450
    },
    {
      "epoch": 6.909370422363282e-05,
      "model_forward_time": 1.7153561115264893,
      "step": 72450
    },
    {
      "dynamics_loss_avg": 0.057227997295558455,
      "epoch": 6.909370422363282e-05,
      "step": 72450
    },
    {
      "action_loss_avg": 0.0016833428031532093,
      "epoch": 6.909370422363282e-05,
      "step": 72450
    },
    {
      "epoch": 6.909370422363282e-05,
      "step": 72450,
      "training_step_time": 6.7570977210998535
    },
    {
      "epoch": 6.909465789794921e-05,
      "model_forward_time": 1.706678867340088,
      "step": 72451
    },
    {
      "epoch": 6.909465789794921e-05,
      "step": 72451,
      "training_step_time": 6.800053834915161
    },
    {
      "epoch": 6.909561157226562e-05,
      "model_forward_time": 1.7054388523101807,
      "step": 72452
    },
    {
      "epoch": 6.909561157226562e-05,
      "step": 72452,
      "training_step_time": 6.7771875858306885
    },
    {
      "epoch": 6.909656524658203e-05,
      "model_forward_time": 1.7051482200622559,
      "step": 72453
    },
    {
      "epoch": 6.909656524658203e-05,
      "step": 72453,
      "training_step_time": 6.806031227111816
    },
    {
      "epoch": 6.909751892089843e-05,
      "model_forward_time": 1.7043383121490479,
      "step": 72454
    },
    {
      "epoch": 6.909751892089843e-05,
      "step": 72454,
      "training_step_time": 6.788727760314941
    },
    {
      "epoch": 6.909847259521484e-05,
      "model_forward_time": 1.7854864597320557,
      "step": 72455
    },
    {
      "epoch": 6.909847259521484e-05,
      "step": 72455,
      "training_step_time": 7.581094264984131
    },
    {
      "epoch": 6.909942626953125e-05,
      "model_forward_time": 2.243110179901123,
      "step": 72456
    },
    {
      "epoch": 6.909942626953125e-05,
      "step": 72456,
      "training_step_time": 8.628237962722778
    },
    {
      "epoch": 6.910037994384765e-05,
      "model_forward_time": 1.704296588897705,
      "step": 72457
    },
    {
      "epoch": 6.910037994384765e-05,
      "step": 72457,
      "training_step_time": 7.864135980606079
    },
    {
      "epoch": 6.910133361816406e-05,
      "model_forward_time": 1.7120673656463623,
      "step": 72458
    },
    {
      "epoch": 6.910133361816406e-05,
      "step": 72458,
      "training_step_time": 7.705430746078491
    },
    {
      "epoch": 6.910228729248047e-05,
      "model_forward_time": 1.7075302600860596,
      "step": 72459
    },
    {
      "epoch": 6.910228729248047e-05,
      "step": 72459,
      "training_step_time": 6.805245399475098
    },
    {
      "epoch": 6.910324096679687e-05,
      "grad_norm": 0.07739794999361038,
      "learning_rate": 1.9342890197635057e-06,
      "loss": 0.058,
      "step": 72460
    },
    {
      "epoch": 6.910324096679687e-05,
      "model_forward_time": 1.702472448348999,
      "step": 72460
    },
    {
      "dynamics_loss_avg": 0.08037789016962052,
      "epoch": 6.910324096679687e-05,
      "step": 72460
    },
    {
      "action_loss_avg": 0.00295555905031506,
      "epoch": 6.910324096679687e-05,
      "step": 72460
    },
    {
      "epoch": 6.910324096679687e-05,
      "step": 72460,
      "training_step_time": 6.800354480743408
    },
    {
      "epoch": 6.910419464111328e-05,
      "model_forward_time": 1.69624662399292,
      "step": 72461
    },
    {
      "epoch": 6.910419464111328e-05,
      "step": 72461,
      "training_step_time": 6.784799814224243
    },
    {
      "epoch": 6.910514831542969e-05,
      "model_forward_time": 1.7167253494262695,
      "step": 72462
    },
    {
      "epoch": 6.910514831542969e-05,
      "step": 72462,
      "training_step_time": 6.840880632400513
    },
    {
      "epoch": 6.910610198974609e-05,
      "model_forward_time": 1.700035810470581,
      "step": 72463
    },
    {
      "epoch": 6.910610198974609e-05,
      "step": 72463,
      "training_step_time": 6.713094472885132
    },
    {
      "epoch": 6.91070556640625e-05,
      "model_forward_time": 1.7885315418243408,
      "step": 72464
    },
    {
      "epoch": 6.91070556640625e-05,
      "step": 72464,
      "training_step_time": 7.616532325744629
    },
    {
      "epoch": 6.910800933837891e-05,
      "model_forward_time": 2.2358274459838867,
      "step": 72465
    },
    {
      "epoch": 6.910800933837891e-05,
      "step": 72465,
      "training_step_time": 8.63902997970581
    },
    {
      "epoch": 6.91089630126953e-05,
      "model_forward_time": 1.710266351699829,
      "step": 72466
    },
    {
      "epoch": 6.91089630126953e-05,
      "step": 72466,
      "training_step_time": 9.803727388381958
    },
    {
      "epoch": 6.910991668701172e-05,
      "model_forward_time": 1.7083697319030762,
      "step": 72467
    },
    {
      "epoch": 6.910991668701172e-05,
      "step": 72467,
      "training_step_time": 7.718233823776245
    },
    {
      "epoch": 6.911087036132813e-05,
      "model_forward_time": 1.7127935886383057,
      "step": 72468
    },
    {
      "epoch": 6.911087036132813e-05,
      "step": 72468,
      "training_step_time": 6.803651809692383
    },
    {
      "epoch": 6.911182403564453e-05,
      "model_forward_time": 1.7078917026519775,
      "step": 72469
    },
    {
      "epoch": 6.911182403564453e-05,
      "step": 72469,
      "training_step_time": 6.8513548374176025
    },
    {
      "epoch": 6.911277770996094e-05,
      "grad_norm": 0.08262594044208527,
      "learning_rate": 1.9329829916313684e-06,
      "loss": 0.057,
      "step": 72470
    },
    {
      "epoch": 6.911277770996094e-05,
      "model_forward_time": 1.7039406299591064,
      "step": 72470
    },
    {
      "dynamics_loss_avg": 0.06589841488748789,
      "epoch": 6.911277770996094e-05,
      "step": 72470
    },
    {
      "action_loss_avg": 0.002628466789610684,
      "epoch": 6.911277770996094e-05,
      "step": 72470
    },
    {
      "epoch": 6.911277770996094e-05,
      "step": 72470,
      "training_step_time": 6.8699541091918945
    },
    {
      "epoch": 6.911373138427735e-05,
      "model_forward_time": 1.709313154220581,
      "step": 72471
    },
    {
      "epoch": 6.911373138427735e-05,
      "step": 72471,
      "training_step_time": 6.747941732406616
    },
    {
      "epoch": 6.911468505859374e-05,
      "model_forward_time": 1.7037742137908936,
      "step": 72472
    },
    {
      "epoch": 6.911468505859374e-05,
      "step": 72472,
      "training_step_time": 6.727954387664795
    },
    {
      "epoch": 6.911563873291016e-05,
      "model_forward_time": 1.7952606678009033,
      "step": 72473
    },
    {
      "epoch": 6.911563873291016e-05,
      "step": 72473,
      "training_step_time": 7.648458957672119
    },
    {
      "epoch": 6.911659240722657e-05,
      "model_forward_time": 2.2514686584472656,
      "step": 72474
    },
    {
      "epoch": 6.911659240722657e-05,
      "step": 72474,
      "training_step_time": 7.67591118812561
    },
    {
      "epoch": 6.911754608154296e-05,
      "model_forward_time": 1.704777717590332,
      "step": 72475
    },
    {
      "epoch": 6.911754608154296e-05,
      "step": 72475,
      "training_step_time": 11.844616651535034
    },
    {
      "epoch": 6.911849975585937e-05,
      "model_forward_time": 1.7084293365478516,
      "step": 72476
    },
    {
      "epoch": 6.911849975585937e-05,
      "step": 72476,
      "training_step_time": 7.8405866622924805
    },
    {
      "epoch": 6.911945343017578e-05,
      "model_forward_time": 1.7174279689788818,
      "step": 72477
    },
    {
      "epoch": 6.911945343017578e-05,
      "step": 72477,
      "training_step_time": 6.760942220687866
    },
    {
      "epoch": 6.912040710449218e-05,
      "model_forward_time": 1.7089684009552002,
      "step": 72478
    },
    {
      "epoch": 6.912040710449218e-05,
      "step": 72478,
      "training_step_time": 6.749670505523682
    },
    {
      "epoch": 6.912136077880859e-05,
      "model_forward_time": 1.7071857452392578,
      "step": 72479
    },
    {
      "epoch": 6.912136077880859e-05,
      "step": 72479,
      "training_step_time": 6.871671915054321
    },
    {
      "epoch": 6.9122314453125e-05,
      "grad_norm": 0.08492249995470047,
      "learning_rate": 1.9316772989036013e-06,
      "loss": 0.0532,
      "step": 72480
    },
    {
      "epoch": 6.9122314453125e-05,
      "model_forward_time": 1.7061328887939453,
      "step": 72480
    },
    {
      "dynamics_loss_avg": 0.06455171573907137,
      "epoch": 6.9122314453125e-05,
      "step": 72480
    },
    {
      "action_loss_avg": 0.0018170651223044843,
      "epoch": 6.9122314453125e-05,
      "step": 72480
    },
    {
      "epoch": 6.9122314453125e-05,
      "step": 72480,
      "training_step_time": 6.848611831665039
    },
    {
      "epoch": 6.91232681274414e-05,
      "model_forward_time": 1.7067821025848389,
      "step": 72481
    },
    {
      "epoch": 6.91232681274414e-05,
      "step": 72481,
      "training_step_time": 6.789856910705566
    },
    {
      "epoch": 6.912422180175781e-05,
      "model_forward_time": 1.7965469360351562,
      "step": 72482
    },
    {
      "epoch": 6.912422180175781e-05,
      "step": 72482,
      "training_step_time": 7.687212944030762
    },
    {
      "epoch": 6.912517547607422e-05,
      "model_forward_time": 2.2701094150543213,
      "step": 72483
    },
    {
      "epoch": 6.912517547607422e-05,
      "step": 72483,
      "training_step_time": 7.722442388534546
    },
    {
      "epoch": 6.912612915039062e-05,
      "model_forward_time": 1.7077412605285645,
      "step": 72484
    },
    {
      "epoch": 6.912612915039062e-05,
      "step": 72484,
      "training_step_time": 10.896799802780151
    },
    {
      "epoch": 6.912708282470703e-05,
      "model_forward_time": 1.7128334045410156,
      "step": 72485
    },
    {
      "epoch": 6.912708282470703e-05,
      "step": 72485,
      "training_step_time": 7.803455829620361
    },
    {
      "epoch": 6.912803649902344e-05,
      "model_forward_time": 1.709756851196289,
      "step": 72486
    },
    {
      "epoch": 6.912803649902344e-05,
      "step": 72486,
      "training_step_time": 7.726568222045898
    },
    {
      "epoch": 6.912899017333984e-05,
      "model_forward_time": 1.710798978805542,
      "step": 72487
    },
    {
      "epoch": 6.912899017333984e-05,
      "step": 72487,
      "training_step_time": 6.7861738204956055
    },
    {
      "epoch": 6.912994384765625e-05,
      "model_forward_time": 1.706575870513916,
      "step": 72488
    },
    {
      "epoch": 6.912994384765625e-05,
      "step": 72488,
      "training_step_time": 6.786201238632202
    },
    {
      "epoch": 6.913089752197266e-05,
      "model_forward_time": 1.7031433582305908,
      "step": 72489
    },
    {
      "epoch": 6.913089752197266e-05,
      "step": 72489,
      "training_step_time": 6.784108638763428
    },
    {
      "epoch": 6.913185119628906e-05,
      "grad_norm": 0.11578666418790817,
      "learning_rate": 1.930371941722991e-06,
      "loss": 0.0561,
      "step": 72490
    },
    {
      "epoch": 6.913185119628906e-05,
      "model_forward_time": 1.6965727806091309,
      "step": 72490
    },
    {
      "dynamics_loss_avg": 0.04975213333964348,
      "epoch": 6.913185119628906e-05,
      "step": 72490
    },
    {
      "action_loss_avg": 0.0033431101939640938,
      "epoch": 6.913185119628906e-05,
      "step": 72490
    },
    {
      "epoch": 6.913185119628906e-05,
      "step": 72490,
      "training_step_time": 6.838042497634888
    },
    {
      "epoch": 6.913280487060547e-05,
      "model_forward_time": 1.7888448238372803,
      "step": 72491
    },
    {
      "epoch": 6.913280487060547e-05,
      "step": 72491,
      "training_step_time": 7.680354595184326
    },
    {
      "epoch": 6.913375854492188e-05,
      "model_forward_time": 2.2751340866088867,
      "step": 72492
    },
    {
      "epoch": 6.913375854492188e-05,
      "step": 72492,
      "training_step_time": 8.738555431365967
    },
    {
      "epoch": 6.913471221923828e-05,
      "model_forward_time": 1.7050600051879883,
      "step": 72493
    },
    {
      "epoch": 6.913471221923828e-05,
      "step": 72493,
      "training_step_time": 8.655805587768555
    },
    {
      "epoch": 6.913566589355469e-05,
      "model_forward_time": 1.7108919620513916,
      "step": 72494
    },
    {
      "epoch": 6.913566589355469e-05,
      "step": 72494,
      "training_step_time": 8.782373428344727
    },
    {
      "epoch": 6.91366195678711e-05,
      "model_forward_time": 1.7029716968536377,
      "step": 72495
    },
    {
      "epoch": 6.91366195678711e-05,
      "step": 72495,
      "training_step_time": 7.636907339096069
    },
    {
      "epoch": 6.91375732421875e-05,
      "model_forward_time": 1.7046396732330322,
      "step": 72496
    },
    {
      "epoch": 6.91375732421875e-05,
      "step": 72496,
      "training_step_time": 6.763377904891968
    },
    {
      "epoch": 6.91385269165039e-05,
      "model_forward_time": 1.7075295448303223,
      "step": 72497
    },
    {
      "epoch": 6.91385269165039e-05,
      "step": 72497,
      "training_step_time": 6.798155307769775
    },
    {
      "epoch": 6.913948059082032e-05,
      "model_forward_time": 1.6997499465942383,
      "step": 72498
    },
    {
      "epoch": 6.913948059082032e-05,
      "step": 72498,
      "training_step_time": 6.718058347702026
    },
    {
      "epoch": 6.914043426513671e-05,
      "model_forward_time": 1.7105152606964111,
      "step": 72499
    },
    {
      "epoch": 6.914043426513671e-05,
      "step": 72499,
      "training_step_time": 6.708698987960815
    },
    {
      "epoch": 6.914138793945312e-05,
      "grad_norm": 0.09429214894771576,
      "learning_rate": 1.929066920232292e-06,
      "loss": 0.0587,
      "step": 72500
    },
    {
      "epoch": 6.914138793945312e-05,
      "model_forward_time": 1.7827208042144775,
      "step": 72500
    },
    {
      "dynamics_loss_avg": 0.062060953862965104,
      "epoch": 6.914138793945312e-05,
      "step": 72500
    },
    {
      "action_loss_avg": 0.0017966344370506705,
      "epoch": 6.914138793945312e-05,
      "step": 72500
    },
    {
      "epoch": 6.914138793945312e-05,
      "step": 72500,
      "training_step_time": 7.677399396896362
    },
    {
      "epoch": 6.914234161376954e-05,
      "model_forward_time": 1.705143928527832,
      "step": 72501
    },
    {
      "epoch": 6.914234161376954e-05,
      "step": 72501,
      "training_step_time": 8.633580684661865
    },
    {
      "epoch": 6.914329528808593e-05,
      "model_forward_time": 1.7352676391601562,
      "step": 72502
    },
    {
      "epoch": 6.914329528808593e-05,
      "step": 72502,
      "training_step_time": 8.444810390472412
    },
    {
      "epoch": 6.914424896240234e-05,
      "model_forward_time": 1.707646131515503,
      "step": 72503
    },
    {
      "epoch": 6.914424896240234e-05,
      "step": 72503,
      "training_step_time": 8.801698446273804
    },
    {
      "epoch": 6.914520263671875e-05,
      "model_forward_time": 1.7053077220916748,
      "step": 72504
    },
    {
      "epoch": 6.914520263671875e-05,
      "step": 72504,
      "training_step_time": 7.739183664321899
    },
    {
      "epoch": 6.914615631103515e-05,
      "model_forward_time": 1.704289197921753,
      "step": 72505
    },
    {
      "epoch": 6.914615631103515e-05,
      "step": 72505,
      "training_step_time": 6.831742286682129
    },
    {
      "epoch": 6.914710998535156e-05,
      "model_forward_time": 1.707078456878662,
      "step": 72506
    },
    {
      "epoch": 6.914710998535156e-05,
      "step": 72506,
      "training_step_time": 6.7507781982421875
    },
    {
      "epoch": 6.914806365966797e-05,
      "model_forward_time": 1.709392786026001,
      "step": 72507
    },
    {
      "epoch": 6.914806365966797e-05,
      "step": 72507,
      "training_step_time": 6.785255193710327
    },
    {
      "epoch": 6.914901733398437e-05,
      "model_forward_time": 1.7062625885009766,
      "step": 72508
    },
    {
      "epoch": 6.914901733398437e-05,
      "step": 72508,
      "training_step_time": 6.811643600463867
    },
    {
      "epoch": 6.914997100830078e-05,
      "model_forward_time": 1.3532869815826416,
      "step": 72509
    },
    {
      "epoch": 6.914997100830078e-05,
      "step": 72509,
      "training_step_time": 7.6217100620269775
    },
    {
      "epoch": 6.915092468261719e-05,
      "grad_norm": 0.10172375291585922,
      "learning_rate": 1.9277622345742155e-06,
      "loss": 0.0553,
      "step": 72510
    },
    {
      "epoch": 6.915092468261719e-05,
      "model_forward_time": 1.702650547027588,
      "step": 72510
    },
    {
      "dynamics_loss_avg": 0.05973087791353464,
      "epoch": 6.915092468261719e-05,
      "step": 72510
    },
    {
      "action_loss_avg": 0.0016556486836634577,
      "epoch": 6.915092468261719e-05,
      "step": 72510
    },
    {
      "epoch": 6.915092468261719e-05,
      "step": 72510,
      "training_step_time": 7.500972509384155
    },
    {
      "epoch": 6.915187835693359e-05,
      "model_forward_time": 2.245163917541504,
      "step": 72511
    },
    {
      "epoch": 6.915187835693359e-05,
      "step": 72511,
      "training_step_time": 9.694350719451904
    },
    {
      "epoch": 6.915283203125e-05,
      "model_forward_time": 1.7088532447814941,
      "step": 72512
    },
    {
      "epoch": 6.915283203125e-05,
      "step": 72512,
      "training_step_time": 9.192430973052979
    },
    {
      "epoch": 6.915378570556641e-05,
      "model_forward_time": 1.712508201599121,
      "step": 72513
    },
    {
      "epoch": 6.915378570556641e-05,
      "step": 72513,
      "training_step_time": 7.7892677783966064
    },
    {
      "epoch": 6.915473937988281e-05,
      "model_forward_time": 1.714275598526001,
      "step": 72514
    },
    {
      "epoch": 6.915473937988281e-05,
      "step": 72514,
      "training_step_time": 7.564995765686035
    },
    {
      "epoch": 6.915569305419922e-05,
      "model_forward_time": 1.7052829265594482,
      "step": 72515
    },
    {
      "epoch": 6.915569305419922e-05,
      "step": 72515,
      "training_step_time": 6.741175889968872
    },
    {
      "epoch": 6.915664672851563e-05,
      "model_forward_time": 1.7139387130737305,
      "step": 72516
    },
    {
      "epoch": 6.915664672851563e-05,
      "step": 72516,
      "training_step_time": 6.775407791137695
    },
    {
      "epoch": 6.915760040283203e-05,
      "model_forward_time": 1.7057831287384033,
      "step": 72517
    },
    {
      "epoch": 6.915760040283203e-05,
      "step": 72517,
      "training_step_time": 6.790119647979736
    },
    {
      "epoch": 6.915855407714844e-05,
      "model_forward_time": 1.7334070205688477,
      "step": 72518
    },
    {
      "epoch": 6.915855407714844e-05,
      "step": 72518,
      "training_step_time": 7.6582932472229
    },
    {
      "epoch": 6.915950775146485e-05,
      "model_forward_time": 1.7015516757965088,
      "step": 72519
    },
    {
      "epoch": 6.915950775146485e-05,
      "step": 72519,
      "training_step_time": 7.462922096252441
    },
    {
      "epoch": 6.916046142578125e-05,
      "grad_norm": 0.0927940309047699,
      "learning_rate": 1.926457884891444e-06,
      "loss": 0.0547,
      "step": 72520
    },
    {
      "epoch": 6.916046142578125e-05,
      "model_forward_time": 2.261780261993408,
      "step": 72520
    },
    {
      "dynamics_loss_avg": 0.05542722214013338,
      "epoch": 6.916046142578125e-05,
      "step": 72520
    },
    {
      "action_loss_avg": 0.001239613356301561,
      "epoch": 6.916046142578125e-05,
      "step": 72520
    },
    {
      "epoch": 6.916046142578125e-05,
      "step": 72520,
      "training_step_time": 8.404553174972534
    },
    {
      "epoch": 6.916141510009766e-05,
      "model_forward_time": 1.709728717803955,
      "step": 72521
    },
    {
      "epoch": 6.916141510009766e-05,
      "step": 72521,
      "training_step_time": 8.964107990264893
    },
    {
      "epoch": 6.916236877441407e-05,
      "model_forward_time": 1.7169840335845947,
      "step": 72522
    },
    {
      "epoch": 6.916236877441407e-05,
      "step": 72522,
      "training_step_time": 7.801187515258789
    },
    {
      "epoch": 6.916332244873047e-05,
      "model_forward_time": 1.7078700065612793,
      "step": 72523
    },
    {
      "epoch": 6.916332244873047e-05,
      "step": 72523,
      "training_step_time": 7.575130462646484
    },
    {
      "epoch": 6.916427612304688e-05,
      "model_forward_time": 1.7062089443206787,
      "step": 72524
    },
    {
      "epoch": 6.916427612304688e-05,
      "step": 72524,
      "training_step_time": 6.751515865325928
    },
    {
      "epoch": 6.916522979736329e-05,
      "model_forward_time": 1.7104933261871338,
      "step": 72525
    },
    {
      "epoch": 6.916522979736329e-05,
      "step": 72525,
      "training_step_time": 6.776783227920532
    },
    {
      "epoch": 6.916618347167968e-05,
      "model_forward_time": 1.7100286483764648,
      "step": 72526
    },
    {
      "epoch": 6.916618347167968e-05,
      "step": 72526,
      "training_step_time": 6.800534248352051
    },
    {
      "epoch": 6.91671371459961e-05,
      "model_forward_time": 1.7196619510650635,
      "step": 72527
    },
    {
      "epoch": 6.91671371459961e-05,
      "step": 72527,
      "training_step_time": 7.599911451339722
    },
    {
      "epoch": 6.91680908203125e-05,
      "model_forward_time": 1.6970000267028809,
      "step": 72528
    },
    {
      "epoch": 6.91680908203125e-05,
      "step": 72528,
      "training_step_time": 7.463428735733032
    },
    {
      "epoch": 6.91690444946289e-05,
      "model_forward_time": 2.258780002593994,
      "step": 72529
    },
    {
      "epoch": 6.91690444946289e-05,
      "step": 72529,
      "training_step_time": 9.47391390800476
    },
    {
      "epoch": 6.916999816894531e-05,
      "grad_norm": 0.09819598495960236,
      "learning_rate": 1.925153871326615e-06,
      "loss": 0.0561,
      "step": 72530
    },
    {
      "epoch": 6.916999816894531e-05,
      "model_forward_time": 1.6990110874176025,
      "step": 72530
    },
    {
      "dynamics_loss_avg": 0.05846754210069775,
      "epoch": 6.916999816894531e-05,
      "step": 72530
    },
    {
      "action_loss_avg": 0.002315475681098178,
      "epoch": 6.916999816894531e-05,
      "step": 72530
    },
    {
      "epoch": 6.916999816894531e-05,
      "step": 72530,
      "training_step_time": 9.626749277114868
    },
    {
      "epoch": 6.917095184326172e-05,
      "model_forward_time": 1.7157363891601562,
      "step": 72531
    },
    {
      "epoch": 6.917095184326172e-05,
      "step": 72531,
      "training_step_time": 9.710468530654907
    },
    {
      "epoch": 6.917190551757812e-05,
      "model_forward_time": 1.703831672668457,
      "step": 72532
    },
    {
      "epoch": 6.917190551757812e-05,
      "step": 72532,
      "training_step_time": 7.674297571182251
    },
    {
      "epoch": 6.917285919189453e-05,
      "model_forward_time": 1.7211062908172607,
      "step": 72533
    },
    {
      "epoch": 6.917285919189453e-05,
      "step": 72533,
      "training_step_time": 6.7438063621521
    },
    {
      "epoch": 6.917381286621094e-05,
      "model_forward_time": 1.7056002616882324,
      "step": 72534
    },
    {
      "epoch": 6.917381286621094e-05,
      "step": 72534,
      "training_step_time": 6.7525341510772705
    },
    {
      "epoch": 6.917476654052734e-05,
      "model_forward_time": 1.7000584602355957,
      "step": 72535
    },
    {
      "epoch": 6.917476654052734e-05,
      "step": 72535,
      "training_step_time": 6.768304347991943
    },
    {
      "epoch": 6.917572021484375e-05,
      "model_forward_time": 1.6936352252960205,
      "step": 72536
    },
    {
      "epoch": 6.917572021484375e-05,
      "step": 72536,
      "training_step_time": 6.756536245346069
    },
    {
      "epoch": 6.917667388916016e-05,
      "model_forward_time": 1.7730467319488525,
      "step": 72537
    },
    {
      "epoch": 6.917667388916016e-05,
      "step": 72537,
      "training_step_time": 7.632062911987305
    },
    {
      "epoch": 6.917762756347656e-05,
      "model_forward_time": 2.248721122741699,
      "step": 72538
    },
    {
      "epoch": 6.917762756347656e-05,
      "step": 72538,
      "training_step_time": 7.543477773666382
    },
    {
      "epoch": 6.917858123779297e-05,
      "model_forward_time": 1.7153980731964111,
      "step": 72539
    },
    {
      "epoch": 6.917858123779297e-05,
      "step": 72539,
      "training_step_time": 8.538471698760986
    },
    {
      "epoch": 6.917953491210938e-05,
      "grad_norm": 0.07801499962806702,
      "learning_rate": 1.9238501940223357e-06,
      "loss": 0.0543,
      "step": 72540
    },
    {
      "epoch": 6.917953491210938e-05,
      "model_forward_time": 1.7036137580871582,
      "step": 72540
    },
    {
      "dynamics_loss_avg": 0.06783849447965622,
      "epoch": 6.917953491210938e-05,
      "step": 72540
    },
    {
      "action_loss_avg": 0.0019215004344005139,
      "epoch": 6.917953491210938e-05,
      "step": 72540
    },
    {
      "epoch": 6.917953491210938e-05,
      "step": 72540,
      "training_step_time": 8.521093606948853
    },
    {
      "epoch": 6.918048858642578e-05,
      "model_forward_time": 1.7094640731811523,
      "step": 72541
    },
    {
      "epoch": 6.918048858642578e-05,
      "step": 72541,
      "training_step_time": 7.721549987792969
    },
    {
      "epoch": 6.918144226074219e-05,
      "model_forward_time": 1.7081406116485596,
      "step": 72542
    },
    {
      "epoch": 6.918144226074219e-05,
      "step": 72542,
      "training_step_time": 6.741098642349243
    },
    {
      "epoch": 6.91823959350586e-05,
      "model_forward_time": 1.69891357421875,
      "step": 72543
    },
    {
      "epoch": 6.91823959350586e-05,
      "step": 72543,
      "training_step_time": 6.764715909957886
    },
    {
      "epoch": 6.9183349609375e-05,
      "model_forward_time": 1.7153568267822266,
      "step": 72544
    },
    {
      "epoch": 6.9183349609375e-05,
      "step": 72544,
      "training_step_time": 6.767810583114624
    },
    {
      "epoch": 6.918430328369141e-05,
      "model_forward_time": 1.7132863998413086,
      "step": 72545
    },
    {
      "epoch": 6.918430328369141e-05,
      "step": 72545,
      "training_step_time": 6.772242307662964
    },
    {
      "epoch": 6.918525695800782e-05,
      "model_forward_time": 1.7435650825500488,
      "step": 72546
    },
    {
      "epoch": 6.918525695800782e-05,
      "step": 72546,
      "training_step_time": 7.5963218212127686
    },
    {
      "epoch": 6.918621063232422e-05,
      "model_forward_time": 1.701951265335083,
      "step": 72547
    },
    {
      "epoch": 6.918621063232422e-05,
      "step": 72547,
      "training_step_time": 7.558641195297241
    },
    {
      "epoch": 6.918716430664063e-05,
      "model_forward_time": 2.221571922302246,
      "step": 72548
    },
    {
      "epoch": 6.918716430664063e-05,
      "step": 72548,
      "training_step_time": 8.643378734588623
    },
    {
      "epoch": 6.918811798095704e-05,
      "model_forward_time": 1.7072465419769287,
      "step": 72549
    },
    {
      "epoch": 6.918811798095704e-05,
      "step": 72549,
      "training_step_time": 8.853987455368042
    },
    {
      "epoch": 6.918907165527343e-05,
      "grad_norm": 0.09173674136400223,
      "learning_rate": 1.922546853121175e-06,
      "loss": 0.0549,
      "step": 72550
    },
    {
      "epoch": 6.918907165527343e-05,
      "model_forward_time": 1.6967577934265137,
      "step": 72550
    },
    {
      "dynamics_loss_avg": 0.07345773577690125,
      "epoch": 6.918907165527343e-05,
      "step": 72550
    },
    {
      "action_loss_avg": 0.002892957424046472,
      "epoch": 6.918907165527343e-05,
      "step": 72550
    },
    {
      "epoch": 6.918907165527343e-05,
      "step": 72550,
      "training_step_time": 7.72904109954834
    },
    {
      "epoch": 6.919002532958985e-05,
      "model_forward_time": 1.7062537670135498,
      "step": 72551
    },
    {
      "epoch": 6.919002532958985e-05,
      "step": 72551,
      "training_step_time": 7.520448207855225
    },
    {
      "epoch": 6.919097900390626e-05,
      "model_forward_time": 1.3347489833831787,
      "step": 72552
    },
    {
      "epoch": 6.919097900390626e-05,
      "step": 72552,
      "training_step_time": 6.781901836395264
    },
    {
      "epoch": 6.919193267822265e-05,
      "model_forward_time": 1.7120964527130127,
      "step": 72553
    },
    {
      "epoch": 6.919193267822265e-05,
      "step": 72553,
      "training_step_time": 6.7862725257873535
    },
    {
      "epoch": 6.919288635253906e-05,
      "model_forward_time": 1.6931817531585693,
      "step": 72554
    },
    {
      "epoch": 6.919288635253906e-05,
      "step": 72554,
      "training_step_time": 6.847548246383667
    },
    {
      "epoch": 6.919384002685548e-05,
      "model_forward_time": 1.7278718948364258,
      "step": 72555
    },
    {
      "epoch": 6.919384002685548e-05,
      "step": 72555,
      "training_step_time": 7.60978627204895
    },
    {
      "epoch": 6.919479370117187e-05,
      "model_forward_time": 1.7077350616455078,
      "step": 72556
    },
    {
      "epoch": 6.919479370117187e-05,
      "step": 72556,
      "training_step_time": 7.622399091720581
    },
    {
      "epoch": 6.919574737548828e-05,
      "model_forward_time": 2.2292215824127197,
      "step": 72557
    },
    {
      "epoch": 6.919574737548828e-05,
      "step": 72557,
      "training_step_time": 8.667247772216797
    },
    {
      "epoch": 6.91967010498047e-05,
      "model_forward_time": 1.7203681468963623,
      "step": 72558
    },
    {
      "epoch": 6.91967010498047e-05,
      "step": 72558,
      "training_step_time": 9.96355414390564
    },
    {
      "epoch": 6.919765472412109e-05,
      "model_forward_time": 1.703080415725708,
      "step": 72559
    },
    {
      "epoch": 6.919765472412109e-05,
      "step": 72559,
      "training_step_time": 8.840015649795532
    },
    {
      "epoch": 6.91986083984375e-05,
      "grad_norm": 0.09039834886789322,
      "learning_rate": 1.921243848765665e-06,
      "loss": 0.0556,
      "step": 72560
    },
    {
      "epoch": 6.91986083984375e-05,
      "model_forward_time": 1.7111339569091797,
      "step": 72560
    },
    {
      "dynamics_loss_avg": 0.07666219361126422,
      "epoch": 6.91986083984375e-05,
      "step": 72560
    },
    {
      "action_loss_avg": 0.003260127769317478,
      "epoch": 6.91986083984375e-05,
      "step": 72560
    },
    {
      "epoch": 6.91986083984375e-05,
      "step": 72560,
      "training_step_time": 7.583057165145874
    },
    {
      "epoch": 6.919956207275391e-05,
      "model_forward_time": 1.7149767875671387,
      "step": 72561
    },
    {
      "epoch": 6.919956207275391e-05,
      "step": 72561,
      "training_step_time": 7.529681921005249
    },
    {
      "epoch": 6.920051574707031e-05,
      "model_forward_time": 1.698014259338379,
      "step": 72562
    },
    {
      "epoch": 6.920051574707031e-05,
      "step": 72562,
      "training_step_time": 6.762192726135254
    },
    {
      "epoch": 6.920146942138672e-05,
      "model_forward_time": 1.714822769165039,
      "step": 72563
    },
    {
      "epoch": 6.920146942138672e-05,
      "step": 72563,
      "training_step_time": 6.778099298477173
    },
    {
      "epoch": 6.920242309570312e-05,
      "model_forward_time": 1.3399279117584229,
      "step": 72564
    },
    {
      "epoch": 6.920242309570312e-05,
      "step": 72564,
      "training_step_time": 7.565841913223267
    },
    {
      "epoch": 6.920337677001953e-05,
      "model_forward_time": 1.7077651023864746,
      "step": 72565
    },
    {
      "epoch": 6.920337677001953e-05,
      "step": 72565,
      "training_step_time": 7.593285799026489
    },
    {
      "epoch": 6.920433044433594e-05,
      "model_forward_time": 2.2413032054901123,
      "step": 72566
    },
    {
      "epoch": 6.920433044433594e-05,
      "step": 72566,
      "training_step_time": 9.686918020248413
    },
    {
      "epoch": 6.920528411865234e-05,
      "model_forward_time": 1.7035505771636963,
      "step": 72567
    },
    {
      "epoch": 6.920528411865234e-05,
      "step": 72567,
      "training_step_time": 9.674373865127563
    },
    {
      "epoch": 6.920623779296875e-05,
      "model_forward_time": 1.7120935916900635,
      "step": 72568
    },
    {
      "epoch": 6.920623779296875e-05,
      "step": 72568,
      "training_step_time": 7.728739500045776
    },
    {
      "epoch": 6.920719146728516e-05,
      "model_forward_time": 1.7120084762573242,
      "step": 72569
    },
    {
      "epoch": 6.920719146728516e-05,
      "step": 72569,
      "training_step_time": 7.713382720947266
    },
    {
      "epoch": 6.920814514160156e-05,
      "grad_norm": 0.08121185004711151,
      "learning_rate": 1.9199411810982967e-06,
      "loss": 0.0538,
      "step": 72570
    },
    {
      "epoch": 6.920814514160156e-05,
      "model_forward_time": 1.7063088417053223,
      "step": 72570
    },
    {
      "dynamics_loss_avg": 0.06594668440520764,
      "epoch": 6.920814514160156e-05,
      "step": 72570
    },
    {
      "action_loss_avg": 0.0029094639932736753,
      "epoch": 6.920814514160156e-05,
      "step": 72570
    },
    {
      "epoch": 6.920814514160156e-05,
      "step": 72570,
      "training_step_time": 7.523212194442749
    },
    {
      "epoch": 6.920909881591797e-05,
      "model_forward_time": 1.713721752166748,
      "step": 72571
    },
    {
      "epoch": 6.920909881591797e-05,
      "step": 72571,
      "training_step_time": 6.823386192321777
    },
    {
      "epoch": 6.921005249023438e-05,
      "model_forward_time": 1.7113690376281738,
      "step": 72572
    },
    {
      "epoch": 6.921005249023438e-05,
      "step": 72572,
      "training_step_time": 6.842478036880493
    },
    {
      "epoch": 6.921100616455077e-05,
      "model_forward_time": 1.6999659538269043,
      "step": 72573
    },
    {
      "epoch": 6.921100616455077e-05,
      "step": 72573,
      "training_step_time": 7.473106861114502
    },
    {
      "epoch": 6.921195983886719e-05,
      "model_forward_time": 1.784475564956665,
      "step": 72574
    },
    {
      "epoch": 6.921195983886719e-05,
      "step": 72574,
      "training_step_time": 7.626676082611084
    },
    {
      "epoch": 6.92129135131836e-05,
      "model_forward_time": 2.2681612968444824,
      "step": 72575
    },
    {
      "epoch": 6.92129135131836e-05,
      "step": 72575,
      "training_step_time": 8.757218837738037
    },
    {
      "epoch": 6.92138671875e-05,
      "model_forward_time": 1.7022864818572998,
      "step": 72576
    },
    {
      "epoch": 6.92138671875e-05,
      "step": 72576,
      "training_step_time": 10.253316402435303
    },
    {
      "epoch": 6.92148208618164e-05,
      "model_forward_time": 1.7097244262695312,
      "step": 72577
    },
    {
      "epoch": 6.92148208618164e-05,
      "step": 72577,
      "training_step_time": 9.513248682022095
    },
    {
      "epoch": 6.921577453613282e-05,
      "model_forward_time": 1.7062644958496094,
      "step": 72578
    },
    {
      "epoch": 6.921577453613282e-05,
      "step": 72578,
      "training_step_time": 7.717391490936279
    },
    {
      "epoch": 6.921672821044921e-05,
      "model_forward_time": 1.7097954750061035,
      "step": 72579
    },
    {
      "epoch": 6.921672821044921e-05,
      "step": 72579,
      "training_step_time": 7.474480152130127
    },
    {
      "epoch": 6.921768188476562e-05,
      "grad_norm": 0.08333709090948105,
      "learning_rate": 1.9186388502615323e-06,
      "loss": 0.0538,
      "step": 72580
    },
    {
      "epoch": 6.921768188476562e-05,
      "model_forward_time": 1.7009351253509521,
      "step": 72580
    },
    {
      "dynamics_loss_avg": 0.04663258064538241,
      "epoch": 6.921768188476562e-05,
      "step": 72580
    },
    {
      "action_loss_avg": 0.0018654053594218567,
      "epoch": 6.921768188476562e-05,
      "step": 72580
    },
    {
      "epoch": 6.921768188476562e-05,
      "step": 72580,
      "training_step_time": 6.844096899032593
    },
    {
      "epoch": 6.921863555908203e-05,
      "model_forward_time": 1.7389700412750244,
      "step": 72581
    },
    {
      "epoch": 6.921863555908203e-05,
      "step": 72581,
      "training_step_time": 6.831652879714966
    },
    {
      "epoch": 6.921958923339843e-05,
      "model_forward_time": 1.7492547035217285,
      "step": 72582
    },
    {
      "epoch": 6.921958923339843e-05,
      "step": 72582,
      "training_step_time": 7.424473762512207
    },
    {
      "epoch": 6.922054290771484e-05,
      "model_forward_time": 1.813143253326416,
      "step": 72583
    },
    {
      "epoch": 6.922054290771484e-05,
      "step": 72583,
      "training_step_time": 7.685286521911621
    },
    {
      "epoch": 6.922149658203125e-05,
      "model_forward_time": 1.7516205310821533,
      "step": 72584
    },
    {
      "epoch": 6.922149658203125e-05,
      "step": 72584,
      "training_step_time": 8.792436599731445
    },
    {
      "epoch": 6.922245025634765e-05,
      "model_forward_time": 1.7470943927764893,
      "step": 72585
    },
    {
      "epoch": 6.922245025634765e-05,
      "step": 72585,
      "training_step_time": 8.79679799079895
    },
    {
      "epoch": 6.922340393066406e-05,
      "model_forward_time": 1.7017731666564941,
      "step": 72586
    },
    {
      "epoch": 6.922340393066406e-05,
      "step": 72586,
      "training_step_time": 9.277459621429443
    },
    {
      "epoch": 6.922435760498047e-05,
      "model_forward_time": 1.730750322341919,
      "step": 72587
    },
    {
      "epoch": 6.922435760498047e-05,
      "step": 72587,
      "training_step_time": 7.7731688022613525
    },
    {
      "epoch": 6.922531127929687e-05,
      "model_forward_time": 1.7307767868041992,
      "step": 72588
    },
    {
      "epoch": 6.922531127929687e-05,
      "step": 72588,
      "training_step_time": 7.578387022018433
    },
    {
      "epoch": 6.922626495361328e-05,
      "model_forward_time": 1.7268683910369873,
      "step": 72589
    },
    {
      "epoch": 6.922626495361328e-05,
      "step": 72589,
      "training_step_time": 6.838099002838135
    },
    {
      "epoch": 6.922721862792969e-05,
      "grad_norm": 0.11752376705408096,
      "learning_rate": 1.9173368563977884e-06,
      "loss": 0.0528,
      "step": 72590
    },
    {
      "epoch": 6.922721862792969e-05,
      "model_forward_time": 1.7124171257019043,
      "step": 72590
    },
    {
      "dynamics_loss_avg": 0.0399038347415626,
      "epoch": 6.922721862792969e-05,
      "step": 72590
    },
    {
      "action_loss_avg": 0.0010635257873218506,
      "epoch": 6.922721862792969e-05,
      "step": 72590
    },
    {
      "epoch": 6.922721862792969e-05,
      "step": 72590,
      "training_step_time": 6.841604709625244
    },
    {
      "epoch": 6.922817230224609e-05,
      "model_forward_time": 1.711272954940796,
      "step": 72591
    },
    {
      "epoch": 6.922817230224609e-05,
      "step": 72591,
      "training_step_time": 7.465977668762207
    },
    {
      "epoch": 6.92291259765625e-05,
      "model_forward_time": 1.7524535655975342,
      "step": 72592
    },
    {
      "epoch": 6.92291259765625e-05,
      "step": 72592,
      "training_step_time": 7.6401448249816895
    },
    {
      "epoch": 6.923007965087891e-05,
      "model_forward_time": 1.7058050632476807,
      "step": 72593
    },
    {
      "epoch": 6.923007965087891e-05,
      "step": 72593,
      "training_step_time": 7.805570602416992
    },
    {
      "epoch": 6.923103332519531e-05,
      "model_forward_time": 2.2722396850585938,
      "step": 72594
    },
    {
      "epoch": 6.923103332519531e-05,
      "step": 72594,
      "training_step_time": 9.940589427947998
    },
    {
      "epoch": 6.923198699951172e-05,
      "model_forward_time": 1.7055373191833496,
      "step": 72595
    },
    {
      "epoch": 6.923198699951172e-05,
      "step": 72595,
      "training_step_time": 8.811238765716553
    },
    {
      "epoch": 6.923294067382813e-05,
      "model_forward_time": 1.7158708572387695,
      "step": 72596
    },
    {
      "epoch": 6.923294067382813e-05,
      "step": 72596,
      "training_step_time": 7.759103059768677
    },
    {
      "epoch": 6.923389434814453e-05,
      "model_forward_time": 1.7072601318359375,
      "step": 72597
    },
    {
      "epoch": 6.923389434814453e-05,
      "step": 72597,
      "training_step_time": 7.491079092025757
    },
    {
      "epoch": 6.923484802246094e-05,
      "model_forward_time": 1.7253384590148926,
      "step": 72598
    },
    {
      "epoch": 6.923484802246094e-05,
      "step": 72598,
      "training_step_time": 6.792035818099976
    },
    {
      "epoch": 6.923580169677735e-05,
      "model_forward_time": 1.716646671295166,
      "step": 72599
    },
    {
      "epoch": 6.923580169677735e-05,
      "step": 72599,
      "training_step_time": 6.782130479812622
    },
    {
      "epoch": 6.923675537109374e-05,
      "grad_norm": 0.08075125515460968,
      "learning_rate": 1.9160351996494513e-06,
      "loss": 0.0561,
      "step": 72600
    },
    {
      "epoch": 6.923675537109374e-05,
      "model_forward_time": 1.7214863300323486,
      "step": 72600
    },
    {
      "dynamics_loss_avg": 0.06805803366005421,
      "epoch": 6.923675537109374e-05,
      "step": 72600
    },
    {
      "action_loss_avg": 0.002202189021045342,
      "epoch": 6.923675537109374e-05,
      "step": 72600
    },
    {
      "epoch": 6.923675537109374e-05,
      "step": 72600,
      "training_step_time": 6.8486692905426025
    },
    {
      "epoch": 6.923770904541016e-05,
      "model_forward_time": 1.735992670059204,
      "step": 72601
    },
    {
      "epoch": 6.923770904541016e-05,
      "step": 72601,
      "training_step_time": 7.590918779373169
    },
    {
      "epoch": 6.923866271972657e-05,
      "model_forward_time": 1.7101364135742188,
      "step": 72602
    },
    {
      "epoch": 6.923866271972657e-05,
      "step": 72602,
      "training_step_time": 9.458203554153442
    },
    {
      "epoch": 6.923961639404296e-05,
      "model_forward_time": 2.245431900024414,
      "step": 72603
    },
    {
      "epoch": 6.923961639404296e-05,
      "step": 72603,
      "training_step_time": 8.574959516525269
    },
    {
      "epoch": 6.924057006835937e-05,
      "model_forward_time": 1.7123115062713623,
      "step": 72604
    },
    {
      "epoch": 6.924057006835937e-05,
      "step": 72604,
      "training_step_time": 10.036669492721558
    },
    {
      "epoch": 6.924152374267579e-05,
      "model_forward_time": 1.7216846942901611,
      "step": 72605
    },
    {
      "epoch": 6.924152374267579e-05,
      "step": 72605,
      "training_step_time": 8.268222093582153
    },
    {
      "epoch": 6.924247741699218e-05,
      "model_forward_time": 1.7230989933013916,
      "step": 72606
    },
    {
      "epoch": 6.924247741699218e-05,
      "step": 72606,
      "training_step_time": 7.692484378814697
    },
    {
      "epoch": 6.924343109130859e-05,
      "model_forward_time": 1.712904930114746,
      "step": 72607
    },
    {
      "epoch": 6.924343109130859e-05,
      "step": 72607,
      "training_step_time": 7.561176538467407
    },
    {
      "epoch": 6.9244384765625e-05,
      "model_forward_time": 1.7494370937347412,
      "step": 72608
    },
    {
      "epoch": 6.9244384765625e-05,
      "step": 72608,
      "training_step_time": 6.809619188308716
    },
    {
      "epoch": 6.92453384399414e-05,
      "model_forward_time": 1.715622901916504,
      "step": 72609
    },
    {
      "epoch": 6.92453384399414e-05,
      "step": 72609,
      "training_step_time": 6.866116046905518
    },
    {
      "epoch": 6.924629211425781e-05,
      "grad_norm": 0.09975031763315201,
      "learning_rate": 1.914733880158868e-06,
      "loss": 0.0537,
      "step": 72610
    },
    {
      "epoch": 6.924629211425781e-05,
      "model_forward_time": 1.7217280864715576,
      "step": 72610
    },
    {
      "dynamics_loss_avg": 0.04697969062253833,
      "epoch": 6.924629211425781e-05,
      "step": 72610
    },
    {
      "action_loss_avg": 0.0018148591101635248,
      "epoch": 6.924629211425781e-05,
      "step": 72610
    },
    {
      "epoch": 6.924629211425781e-05,
      "step": 72610,
      "training_step_time": 7.511695623397827
    },
    {
      "epoch": 6.924724578857422e-05,
      "model_forward_time": 1.7915775775909424,
      "step": 72611
    },
    {
      "epoch": 6.924724578857422e-05,
      "step": 72611,
      "training_step_time": 7.6598944664001465
    },
    {
      "epoch": 6.924819946289062e-05,
      "model_forward_time": 2.2610347270965576,
      "step": 72612
    },
    {
      "epoch": 6.924819946289062e-05,
      "step": 72612,
      "training_step_time": 9.610395431518555
    },
    {
      "epoch": 6.924915313720703e-05,
      "model_forward_time": 1.7050139904022217,
      "step": 72613
    },
    {
      "epoch": 6.924915313720703e-05,
      "step": 72613,
      "training_step_time": 10.948697566986084
    },
    {
      "epoch": 6.925010681152344e-05,
      "model_forward_time": 1.719202995300293,
      "step": 72614
    },
    {
      "epoch": 6.925010681152344e-05,
      "step": 72614,
      "training_step_time": 7.721874713897705
    },
    {
      "epoch": 6.925106048583984e-05,
      "model_forward_time": 1.7119810581207275,
      "step": 72615
    },
    {
      "epoch": 6.925106048583984e-05,
      "step": 72615,
      "training_step_time": 7.71673846244812
    },
    {
      "epoch": 6.925201416015625e-05,
      "model_forward_time": 1.7101216316223145,
      "step": 72616
    },
    {
      "epoch": 6.925201416015625e-05,
      "step": 72616,
      "training_step_time": 7.587828874588013
    },
    {
      "epoch": 6.925296783447266e-05,
      "model_forward_time": 1.718522071838379,
      "step": 72617
    },
    {
      "epoch": 6.925296783447266e-05,
      "step": 72617,
      "training_step_time": 6.82032322883606
    },
    {
      "epoch": 6.925392150878906e-05,
      "model_forward_time": 0.9952929019927979,
      "step": 72618
    },
    {
      "epoch": 6.925392150878906e-05,
      "step": 72618,
      "training_step_time": 6.808778285980225
    },
    {
      "epoch": 6.925487518310547e-05,
      "model_forward_time": 1.7129580974578857,
      "step": 72619
    },
    {
      "epoch": 6.925487518310547e-05,
      "step": 72619,
      "training_step_time": 7.530834674835205
    },
    {
      "epoch": 6.925582885742188e-05,
      "grad_norm": 0.08631015568971634,
      "learning_rate": 1.9134328980683508e-06,
      "loss": 0.0557,
      "step": 72620
    },
    {
      "epoch": 6.925582885742188e-05,
      "model_forward_time": 1.3846564292907715,
      "step": 72620
    },
    {
      "dynamics_loss_avg": 0.044174181576818225,
      "epoch": 6.925582885742188e-05,
      "step": 72620
    },
    {
      "action_loss_avg": 0.0025586260482668875,
      "epoch": 6.925582885742188e-05,
      "step": 72620
    },
    {
      "epoch": 6.925582885742188e-05,
      "step": 72620,
      "training_step_time": 8.338554859161377
    },
    {
      "epoch": 6.925678253173828e-05,
      "model_forward_time": 1.7095708847045898,
      "step": 72621
    },
    {
      "epoch": 6.925678253173828e-05,
      "step": 72621,
      "training_step_time": 7.705380439758301
    },
    {
      "epoch": 6.925773620605469e-05,
      "model_forward_time": 2.235104560852051,
      "step": 72622
    },
    {
      "epoch": 6.925773620605469e-05,
      "step": 72622,
      "training_step_time": 8.635619640350342
    },
    {
      "epoch": 6.92586898803711e-05,
      "model_forward_time": 1.716374397277832,
      "step": 72623
    },
    {
      "epoch": 6.92586898803711e-05,
      "step": 72623,
      "training_step_time": 7.759009599685669
    },
    {
      "epoch": 6.92596435546875e-05,
      "model_forward_time": 1.7097547054290771,
      "step": 72624
    },
    {
      "epoch": 6.92596435546875e-05,
      "step": 72624,
      "training_step_time": 7.757616281509399
    },
    {
      "epoch": 6.92605972290039e-05,
      "model_forward_time": 1.7116906642913818,
      "step": 72625
    },
    {
      "epoch": 6.92605972290039e-05,
      "step": 72625,
      "training_step_time": 8.054898977279663
    },
    {
      "epoch": 6.926155090332032e-05,
      "model_forward_time": 1.714125633239746,
      "step": 72626
    },
    {
      "epoch": 6.926155090332032e-05,
      "step": 72626,
      "training_step_time": 6.8474440574646
    },
    {
      "epoch": 6.926250457763671e-05,
      "model_forward_time": 1.7148008346557617,
      "step": 72627
    },
    {
      "epoch": 6.926250457763671e-05,
      "step": 72627,
      "training_step_time": 6.793810129165649
    },
    {
      "epoch": 6.926345825195313e-05,
      "model_forward_time": 1.708085060119629,
      "step": 72628
    },
    {
      "epoch": 6.926345825195313e-05,
      "step": 72628,
      "training_step_time": 7.504336833953857
    },
    {
      "epoch": 6.926441192626954e-05,
      "model_forward_time": 1.7570769786834717,
      "step": 72629
    },
    {
      "epoch": 6.926441192626954e-05,
      "step": 72629,
      "training_step_time": 8.269271850585938
    },
    {
      "epoch": 6.926536560058593e-05,
      "grad_norm": 0.08781483769416809,
      "learning_rate": 1.912132253520169e-06,
      "loss": 0.0565,
      "step": 72630
    },
    {
      "epoch": 6.926536560058593e-05,
      "model_forward_time": 1.7084236145019531,
      "step": 72630
    },
    {
      "dynamics_loss_avg": 0.04424800276756287,
      "epoch": 6.926536560058593e-05,
      "step": 72630
    },
    {
      "action_loss_avg": 0.0016443892935058102,
      "epoch": 6.926536560058593e-05,
      "step": 72630
    },
    {
      "epoch": 6.926536560058593e-05,
      "step": 72630,
      "training_step_time": 7.749194860458374
    },
    {
      "epoch": 6.926631927490234e-05,
      "model_forward_time": 2.2549612522125244,
      "step": 72631
    },
    {
      "epoch": 6.926631927490234e-05,
      "step": 72631,
      "training_step_time": 9.20537257194519
    },
    {
      "epoch": 6.926727294921875e-05,
      "model_forward_time": 1.718418836593628,
      "step": 72632
    },
    {
      "epoch": 6.926727294921875e-05,
      "step": 72632,
      "training_step_time": 7.698345899581909
    },
    {
      "epoch": 6.926822662353515e-05,
      "model_forward_time": 1.7114317417144775,
      "step": 72633
    },
    {
      "epoch": 6.926822662353515e-05,
      "step": 72633,
      "training_step_time": 8.326993465423584
    },
    {
      "epoch": 6.926918029785156e-05,
      "model_forward_time": 1.7024798393249512,
      "step": 72634
    },
    {
      "epoch": 6.926918029785156e-05,
      "step": 72634,
      "training_step_time": 7.529494047164917
    },
    {
      "epoch": 6.927013397216797e-05,
      "model_forward_time": 1.7135090827941895,
      "step": 72635
    },
    {
      "epoch": 6.927013397216797e-05,
      "step": 72635,
      "training_step_time": 7.545602798461914
    },
    {
      "epoch": 6.927108764648437e-05,
      "model_forward_time": 1.7112386226654053,
      "step": 72636
    },
    {
      "epoch": 6.927108764648437e-05,
      "step": 72636,
      "training_step_time": 6.796659231185913
    },
    {
      "epoch": 6.927204132080078e-05,
      "model_forward_time": 1.7066709995269775,
      "step": 72637
    },
    {
      "epoch": 6.927204132080078e-05,
      "step": 72637,
      "training_step_time": 7.4370527267456055
    },
    {
      "epoch": 6.927299499511719e-05,
      "model_forward_time": 1.8143086433410645,
      "step": 72638
    },
    {
      "epoch": 6.927299499511719e-05,
      "step": 72638,
      "training_step_time": 7.582656145095825
    },
    {
      "epoch": 6.927394866943359e-05,
      "model_forward_time": 1.8209035396575928,
      "step": 72639
    },
    {
      "epoch": 6.927394866943359e-05,
      "step": 72639,
      "training_step_time": 9.160340547561646
    },
    {
      "epoch": 6.927490234375e-05,
      "grad_norm": 0.09169228374958038,
      "learning_rate": 1.910831946656564e-06,
      "loss": 0.0551,
      "step": 72640
    },
    {
      "epoch": 6.927490234375e-05,
      "model_forward_time": 2.348482608795166,
      "step": 72640
    },
    {
      "dynamics_loss_avg": 0.04630101397633553,
      "epoch": 6.927490234375e-05,
      "step": 72640
    },
    {
      "action_loss_avg": 0.0013247195980511605,
      "epoch": 6.927490234375e-05,
      "step": 72640
    },
    {
      "epoch": 6.927490234375e-05,
      "step": 72640,
      "training_step_time": 8.68859076499939
    },
    {
      "epoch": 6.927585601806641e-05,
      "model_forward_time": 1.8145811557769775,
      "step": 72641
    },
    {
      "epoch": 6.927585601806641e-05,
      "step": 72641,
      "training_step_time": 9.417169570922852
    },
    {
      "epoch": 6.927680969238281e-05,
      "model_forward_time": 1.8462109565734863,
      "step": 72642
    },
    {
      "epoch": 6.927680969238281e-05,
      "step": 72642,
      "training_step_time": 7.704522132873535
    },
    {
      "epoch": 6.927776336669922e-05,
      "model_forward_time": 1.8357570171356201,
      "step": 72643
    },
    {
      "epoch": 6.927776336669922e-05,
      "step": 72643,
      "training_step_time": 8.762125968933105
    },
    {
      "epoch": 6.927871704101563e-05,
      "model_forward_time": 1.7809219360351562,
      "step": 72644
    },
    {
      "epoch": 6.927871704101563e-05,
      "step": 72644,
      "training_step_time": 7.5826945304870605
    },
    {
      "epoch": 6.927967071533203e-05,
      "model_forward_time": 1.7315657138824463,
      "step": 72645
    },
    {
      "epoch": 6.927967071533203e-05,
      "step": 72645,
      "training_step_time": 6.808585166931152
    },
    {
      "epoch": 6.928062438964844e-05,
      "model_forward_time": 1.731222152709961,
      "step": 72646
    },
    {
      "epoch": 6.928062438964844e-05,
      "step": 72646,
      "training_step_time": 7.485950708389282
    },
    {
      "epoch": 6.928157806396485e-05,
      "model_forward_time": 1.7322025299072266,
      "step": 72647
    },
    {
      "epoch": 6.928157806396485e-05,
      "step": 72647,
      "training_step_time": 7.581301689147949
    },
    {
      "epoch": 6.928253173828125e-05,
      "model_forward_time": 1.7323529720306396,
      "step": 72648
    },
    {
      "epoch": 6.928253173828125e-05,
      "step": 72648,
      "training_step_time": 8.616480112075806
    },
    {
      "epoch": 6.928348541259766e-05,
      "model_forward_time": 2.2727136611938477,
      "step": 72649
    },
    {
      "epoch": 6.928348541259766e-05,
      "step": 72649,
      "training_step_time": 8.697653532028198
    },
    {
      "epoch": 6.928443908691407e-05,
      "grad_norm": 0.09185846149921417,
      "learning_rate": 1.9095319776197308e-06,
      "loss": 0.0521,
      "step": 72650
    },
    {
      "epoch": 6.928443908691407e-05,
      "model_forward_time": 1.7216007709503174,
      "step": 72650
    },
    {
      "dynamics_loss_avg": 0.054910858720541,
      "epoch": 6.928443908691407e-05,
      "step": 72650
    },
    {
      "action_loss_avg": 0.001829690596787259,
      "epoch": 6.928443908691407e-05,
      "step": 72650
    },
    {
      "epoch": 6.928443908691407e-05,
      "step": 72650,
      "training_step_time": 8.002833127975464
    },
    {
      "epoch": 6.928539276123047e-05,
      "model_forward_time": 1.7209405899047852,
      "step": 72651
    },
    {
      "epoch": 6.928539276123047e-05,
      "step": 72651,
      "training_step_time": 8.777722358703613
    },
    {
      "epoch": 6.928634643554688e-05,
      "model_forward_time": 1.7121961116790771,
      "step": 72652
    },
    {
      "epoch": 6.928634643554688e-05,
      "step": 72652,
      "training_step_time": 7.766099214553833
    },
    {
      "epoch": 6.928730010986329e-05,
      "model_forward_time": 1.7252991199493408,
      "step": 72653
    },
    {
      "epoch": 6.928730010986329e-05,
      "step": 72653,
      "training_step_time": 7.628246307373047
    },
    {
      "epoch": 6.928825378417968e-05,
      "model_forward_time": 1.7107806205749512,
      "step": 72654
    },
    {
      "epoch": 6.928825378417968e-05,
      "step": 72654,
      "training_step_time": 6.815831899642944
    },
    {
      "epoch": 6.92892074584961e-05,
      "model_forward_time": 1.718940019607544,
      "step": 72655
    },
    {
      "epoch": 6.92892074584961e-05,
      "step": 72655,
      "training_step_time": 7.5039801597595215
    },
    {
      "epoch": 6.92901611328125e-05,
      "model_forward_time": 1.7106680870056152,
      "step": 72656
    },
    {
      "epoch": 6.92901611328125e-05,
      "step": 72656,
      "training_step_time": 7.588545322418213
    },
    {
      "epoch": 6.92911148071289e-05,
      "model_forward_time": 1.8009326457977295,
      "step": 72657
    },
    {
      "epoch": 6.92911148071289e-05,
      "step": 72657,
      "training_step_time": 8.74933123588562
    },
    {
      "epoch": 6.929206848144531e-05,
      "model_forward_time": 1.3530066013336182,
      "step": 72658
    },
    {
      "epoch": 6.929206848144531e-05,
      "step": 72658,
      "training_step_time": 10.508912801742554
    },
    {
      "epoch": 6.929302215576172e-05,
      "model_forward_time": 1.7165827751159668,
      "step": 72659
    },
    {
      "epoch": 6.929302215576172e-05,
      "step": 72659,
      "training_step_time": 9.519309997558594
    },
    {
      "epoch": 6.929397583007812e-05,
      "grad_norm": 0.08877675235271454,
      "learning_rate": 1.908232346551834e-06,
      "loss": 0.0544,
      "step": 72660
    },
    {
      "epoch": 6.929397583007812e-05,
      "model_forward_time": 1.7412307262420654,
      "step": 72660
    },
    {
      "dynamics_loss_avg": 0.05238630510866642,
      "epoch": 6.929397583007812e-05,
      "step": 72660
    },
    {
      "action_loss_avg": 0.0015156327688600868,
      "epoch": 6.929397583007812e-05,
      "step": 72660
    },
    {
      "epoch": 6.929397583007812e-05,
      "step": 72660,
      "training_step_time": 8.730029821395874
    },
    {
      "epoch": 6.929492950439453e-05,
      "model_forward_time": 1.7083239555358887,
      "step": 72661
    },
    {
      "epoch": 6.929492950439453e-05,
      "step": 72661,
      "training_step_time": 7.751827955245972
    },
    {
      "epoch": 6.929588317871094e-05,
      "model_forward_time": 1.7156867980957031,
      "step": 72662
    },
    {
      "epoch": 6.929588317871094e-05,
      "step": 72662,
      "training_step_time": 7.509381055831909
    },
    {
      "epoch": 6.929683685302734e-05,
      "model_forward_time": 1.7173633575439453,
      "step": 72663
    },
    {
      "epoch": 6.929683685302734e-05,
      "step": 72663,
      "training_step_time": 6.78583550453186
    },
    {
      "epoch": 6.929779052734375e-05,
      "model_forward_time": 1.7170164585113525,
      "step": 72664
    },
    {
      "epoch": 6.929779052734375e-05,
      "step": 72664,
      "training_step_time": 7.410604000091553
    },
    {
      "epoch": 6.929874420166016e-05,
      "model_forward_time": 1.7151150703430176,
      "step": 72665
    },
    {
      "epoch": 6.929874420166016e-05,
      "step": 72665,
      "training_step_time": 7.5529820919036865
    },
    {
      "epoch": 6.929969787597656e-05,
      "model_forward_time": 1.7891199588775635,
      "step": 72666
    },
    {
      "epoch": 6.929969787597656e-05,
      "step": 72666,
      "training_step_time": 8.209947109222412
    },
    {
      "epoch": 6.930065155029297e-05,
      "model_forward_time": 1.6895201206207275,
      "step": 72667
    },
    {
      "epoch": 6.930065155029297e-05,
      "step": 72667,
      "training_step_time": 8.131361246109009
    },
    {
      "epoch": 6.930160522460938e-05,
      "model_forward_time": 2.2469823360443115,
      "step": 72668
    },
    {
      "epoch": 6.930160522460938e-05,
      "step": 72668,
      "training_step_time": 9.27878475189209
    },
    {
      "epoch": 6.930255889892578e-05,
      "model_forward_time": 1.7474291324615479,
      "step": 72669
    },
    {
      "epoch": 6.930255889892578e-05,
      "step": 72669,
      "training_step_time": 7.620384216308594
    },
    {
      "epoch": 6.930351257324219e-05,
      "grad_norm": 0.09191562235355377,
      "learning_rate": 1.9069330535949992e-06,
      "loss": 0.0515,
      "step": 72670
    },
    {
      "epoch": 6.930351257324219e-05,
      "model_forward_time": 1.7031188011169434,
      "step": 72670
    },
    {
      "dynamics_loss_avg": 0.06389885377138853,
      "epoch": 6.930351257324219e-05,
      "step": 72670
    },
    {
      "action_loss_avg": 0.002235811436548829,
      "epoch": 6.930351257324219e-05,
      "step": 72670
    },
    {
      "epoch": 6.930351257324219e-05,
      "step": 72670,
      "training_step_time": 9.30359148979187
    },
    {
      "epoch": 6.93044662475586e-05,
      "model_forward_time": 1.7230486869812012,
      "step": 72671
    },
    {
      "epoch": 6.93044662475586e-05,
      "step": 72671,
      "training_step_time": 7.772779226303101
    },
    {
      "epoch": 6.9305419921875e-05,
      "model_forward_time": 1.7239422798156738,
      "step": 72672
    },
    {
      "epoch": 6.9305419921875e-05,
      "step": 72672,
      "training_step_time": 7.618347406387329
    },
    {
      "epoch": 6.930637359619141e-05,
      "model_forward_time": 1.3575360774993896,
      "step": 72673
    },
    {
      "epoch": 6.930637359619141e-05,
      "step": 72673,
      "training_step_time": 6.902962923049927
    },
    {
      "epoch": 6.930732727050782e-05,
      "model_forward_time": 1.7271828651428223,
      "step": 72674
    },
    {
      "epoch": 6.930732727050782e-05,
      "step": 72674,
      "training_step_time": 7.608968496322632
    },
    {
      "epoch": 6.930828094482422e-05,
      "model_forward_time": 1.7571289539337158,
      "step": 72675
    },
    {
      "epoch": 6.930828094482422e-05,
      "step": 72675,
      "training_step_time": 8.700355529785156
    },
    {
      "epoch": 6.930923461914063e-05,
      "model_forward_time": 1.7154405117034912,
      "step": 72676
    },
    {
      "epoch": 6.930923461914063e-05,
      "step": 72676,
      "training_step_time": 8.764050960540771
    },
    {
      "epoch": 6.931018829345704e-05,
      "model_forward_time": 2.2490122318267822,
      "step": 72677
    },
    {
      "epoch": 6.931018829345704e-05,
      "step": 72677,
      "training_step_time": 8.764033555984497
    },
    {
      "epoch": 6.931114196777343e-05,
      "model_forward_time": 1.7069408893585205,
      "step": 72678
    },
    {
      "epoch": 6.931114196777343e-05,
      "step": 72678,
      "training_step_time": 7.949012517929077
    },
    {
      "epoch": 6.931209564208985e-05,
      "model_forward_time": 1.7271511554718018,
      "step": 72679
    },
    {
      "epoch": 6.931209564208985e-05,
      "step": 72679,
      "training_step_time": 7.9259772300720215
    },
    {
      "epoch": 6.931304931640626e-05,
      "grad_norm": 0.07640716433525085,
      "learning_rate": 1.905634098891317e-06,
      "loss": 0.055,
      "step": 72680
    },
    {
      "epoch": 6.931304931640626e-05,
      "model_forward_time": 1.7140147686004639,
      "step": 72680
    },
    {
      "dynamics_loss_avg": 0.05771108586341143,
      "epoch": 6.931304931640626e-05,
      "step": 72680
    },
    {
      "action_loss_avg": 0.001967882108874619,
      "epoch": 6.931304931640626e-05,
      "step": 72680
    },
    {
      "epoch": 6.931304931640626e-05,
      "step": 72680,
      "training_step_time": 7.705356597900391
    },
    {
      "epoch": 6.931400299072265e-05,
      "model_forward_time": 1.7049121856689453,
      "step": 72681
    },
    {
      "epoch": 6.931400299072265e-05,
      "step": 72681,
      "training_step_time": 7.5420873165130615
    },
    {
      "epoch": 6.931495666503906e-05,
      "model_forward_time": 1.7218706607818604,
      "step": 72682
    },
    {
      "epoch": 6.931495666503906e-05,
      "step": 72682,
      "training_step_time": 6.817606210708618
    },
    {
      "epoch": 6.931591033935548e-05,
      "model_forward_time": 1.71885085105896,
      "step": 72683
    },
    {
      "epoch": 6.931591033935548e-05,
      "step": 72683,
      "training_step_time": 7.526182651519775
    },
    {
      "epoch": 6.931686401367187e-05,
      "model_forward_time": 1.7415049076080322,
      "step": 72684
    },
    {
      "epoch": 6.931686401367187e-05,
      "step": 72684,
      "training_step_time": 7.582234621047974
    },
    {
      "epoch": 6.931781768798828e-05,
      "model_forward_time": 1.7434039115905762,
      "step": 72685
    },
    {
      "epoch": 6.931781768798828e-05,
      "step": 72685,
      "training_step_time": 7.676822662353516
    },
    {
      "epoch": 6.93187713623047e-05,
      "model_forward_time": 2.2515761852264404,
      "step": 72686
    },
    {
      "epoch": 6.93187713623047e-05,
      "step": 72686,
      "training_step_time": 9.23104190826416
    },
    {
      "epoch": 6.931972503662109e-05,
      "model_forward_time": 1.712094783782959,
      "step": 72687
    },
    {
      "epoch": 6.931972503662109e-05,
      "step": 72687,
      "training_step_time": 9.865685939788818
    },
    {
      "epoch": 6.93206787109375e-05,
      "model_forward_time": 1.70884370803833,
      "step": 72688
    },
    {
      "epoch": 6.93206787109375e-05,
      "step": 72688,
      "training_step_time": 9.339576721191406
    },
    {
      "epoch": 6.93216323852539e-05,
      "model_forward_time": 1.700634241104126,
      "step": 72689
    },
    {
      "epoch": 6.93216323852539e-05,
      "step": 72689,
      "training_step_time": 7.727569341659546
    },
    {
      "epoch": 6.932258605957031e-05,
      "grad_norm": 0.0728868842124939,
      "learning_rate": 1.9043354825828358e-06,
      "loss": 0.0557,
      "step": 72690
    },
    {
      "epoch": 6.932258605957031e-05,
      "model_forward_time": 1.7052347660064697,
      "step": 72690
    },
    {
      "dynamics_loss_avg": 0.05066359005868435,
      "epoch": 6.932258605957031e-05,
      "step": 72690
    },
    {
      "action_loss_avg": 0.0017923761683050544,
      "epoch": 6.932258605957031e-05,
      "step": 72690
    },
    {
      "epoch": 6.932258605957031e-05,
      "step": 72690,
      "training_step_time": 7.480917692184448
    },
    {
      "epoch": 6.932353973388672e-05,
      "model_forward_time": 1.7105121612548828,
      "step": 72691
    },
    {
      "epoch": 6.932353973388672e-05,
      "step": 72691,
      "training_step_time": 6.843222618103027
    },
    {
      "epoch": 6.932449340820312e-05,
      "model_forward_time": 1.7086181640625,
      "step": 72692
    },
    {
      "epoch": 6.932449340820312e-05,
      "step": 72692,
      "training_step_time": 7.339966297149658
    },
    {
      "epoch": 6.932544708251953e-05,
      "model_forward_time": 1.7223682403564453,
      "step": 72693
    },
    {
      "epoch": 6.932544708251953e-05,
      "step": 72693,
      "training_step_time": 7.522211313247681
    },
    {
      "epoch": 6.932640075683594e-05,
      "model_forward_time": 1.802443027496338,
      "step": 72694
    },
    {
      "epoch": 6.932640075683594e-05,
      "step": 72694,
      "training_step_time": 7.641427516937256
    },
    {
      "epoch": 6.932735443115234e-05,
      "model_forward_time": 2.2853236198425293,
      "step": 72695
    },
    {
      "epoch": 6.932735443115234e-05,
      "step": 72695,
      "training_step_time": 7.773333311080933
    },
    {
      "epoch": 6.932830810546875e-05,
      "model_forward_time": 1.7350647449493408,
      "step": 72696
    },
    {
      "epoch": 6.932830810546875e-05,
      "step": 72696,
      "training_step_time": 10.488934516906738
    },
    {
      "epoch": 6.932926177978516e-05,
      "model_forward_time": 1.7292404174804688,
      "step": 72697
    },
    {
      "epoch": 6.932926177978516e-05,
      "step": 72697,
      "training_step_time": 7.730999708175659
    },
    {
      "epoch": 6.933021545410156e-05,
      "model_forward_time": 1.736379861831665,
      "step": 72698
    },
    {
      "epoch": 6.933021545410156e-05,
      "step": 72698,
      "training_step_time": 8.280810356140137
    },
    {
      "epoch": 6.933116912841797e-05,
      "model_forward_time": 1.73362135887146,
      "step": 72699
    },
    {
      "epoch": 6.933116912841797e-05,
      "step": 72699,
      "training_step_time": 7.5005271434783936
    },
    {
      "epoch": 6.933212280273438e-05,
      "grad_norm": 0.08581037074327469,
      "learning_rate": 1.9030372048115724e-06,
      "loss": 0.0553,
      "step": 72700
    },
    {
      "epoch": 6.933212280273438e-05,
      "model_forward_time": 1.7272613048553467,
      "step": 72700
    },
    {
      "dynamics_loss_avg": 0.05068246955052018,
      "epoch": 6.933212280273438e-05,
      "step": 72700
    },
    {
      "action_loss_avg": 0.002119551334180869,
      "epoch": 6.933212280273438e-05,
      "step": 72700
    },
    {
      "epoch": 6.933212280273438e-05,
      "step": 72700,
      "training_step_time": 7.5966150760650635
    },
    {
      "epoch": 6.933307647705077e-05,
      "model_forward_time": 1.7331089973449707,
      "step": 72701
    },
    {
      "epoch": 6.933307647705077e-05,
      "step": 72701,
      "training_step_time": 7.218253135681152
    },
    {
      "epoch": 6.933403015136719e-05,
      "model_forward_time": 1.7184741497039795,
      "step": 72702
    },
    {
      "epoch": 6.933403015136719e-05,
      "step": 72702,
      "training_step_time": 7.6568663120269775
    },
    {
      "epoch": 6.93349838256836e-05,
      "model_forward_time": 1.7923860549926758,
      "step": 72703
    },
    {
      "epoch": 6.93349838256836e-05,
      "step": 72703,
      "training_step_time": 8.322615385055542
    },
    {
      "epoch": 6.93359375e-05,
      "model_forward_time": 2.301590919494629,
      "step": 72704
    },
    {
      "epoch": 6.93359375e-05,
      "step": 72704,
      "training_step_time": 8.617546558380127
    },
    {
      "epoch": 6.93368911743164e-05,
      "model_forward_time": 1.7225489616394043,
      "step": 72705
    },
    {
      "epoch": 6.93368911743164e-05,
      "step": 72705,
      "training_step_time": 8.419084310531616
    },
    {
      "epoch": 6.933784484863282e-05,
      "model_forward_time": 1.730259656906128,
      "step": 72706
    },
    {
      "epoch": 6.933784484863282e-05,
      "step": 72706,
      "training_step_time": 7.74315881729126
    },
    {
      "epoch": 6.933879852294921e-05,
      "model_forward_time": 1.7275075912475586,
      "step": 72707
    },
    {
      "epoch": 6.933879852294921e-05,
      "step": 72707,
      "training_step_time": 7.806660890579224
    },
    {
      "epoch": 6.933975219726562e-05,
      "model_forward_time": 1.728778600692749,
      "step": 72708
    },
    {
      "epoch": 6.933975219726562e-05,
      "step": 72708,
      "training_step_time": 7.79828143119812
    },
    {
      "epoch": 6.934070587158203e-05,
      "model_forward_time": 1.736933708190918,
      "step": 72709
    },
    {
      "epoch": 6.934070587158203e-05,
      "step": 72709,
      "training_step_time": 7.504849672317505
    },
    {
      "epoch": 6.934165954589843e-05,
      "grad_norm": 0.08123253285884857,
      "learning_rate": 1.9017392657195055e-06,
      "loss": 0.0538,
      "step": 72710
    },
    {
      "epoch": 6.934165954589843e-05,
      "model_forward_time": 1.7319316864013672,
      "step": 72710
    },
    {
      "dynamics_loss_avg": 0.06590233910828829,
      "epoch": 6.934165954589843e-05,
      "step": 72710
    },
    {
      "action_loss_avg": 0.0012651951692532748,
      "epoch": 6.934165954589843e-05,
      "step": 72710
    },
    {
      "epoch": 6.934165954589843e-05,
      "step": 72710,
      "training_step_time": 7.267598390579224
    },
    {
      "epoch": 6.934261322021484e-05,
      "model_forward_time": 1.725292444229126,
      "step": 72711
    },
    {
      "epoch": 6.934261322021484e-05,
      "step": 72711,
      "training_step_time": 7.6526806354522705
    },
    {
      "epoch": 6.934356689453125e-05,
      "model_forward_time": 1.7695329189300537,
      "step": 72712
    },
    {
      "epoch": 6.934356689453125e-05,
      "step": 72712,
      "training_step_time": 8.37253475189209
    },
    {
      "epoch": 6.934452056884765e-05,
      "model_forward_time": 2.2349698543548584,
      "step": 72713
    },
    {
      "epoch": 6.934452056884765e-05,
      "step": 72713,
      "training_step_time": 9.059989929199219
    },
    {
      "epoch": 6.934547424316406e-05,
      "model_forward_time": 1.7161188125610352,
      "step": 72714
    },
    {
      "epoch": 6.934547424316406e-05,
      "step": 72714,
      "training_step_time": 8.885481595993042
    },
    {
      "epoch": 6.934642791748047e-05,
      "model_forward_time": 1.705030918121338,
      "step": 72715
    },
    {
      "epoch": 6.934642791748047e-05,
      "step": 72715,
      "training_step_time": 8.647748708724976
    },
    {
      "epoch": 6.934738159179687e-05,
      "model_forward_time": 1.7156193256378174,
      "step": 72716
    },
    {
      "epoch": 6.934738159179687e-05,
      "step": 72716,
      "training_step_time": 8.343983888626099
    },
    {
      "epoch": 6.934833526611328e-05,
      "model_forward_time": 1.71463942527771,
      "step": 72717
    },
    {
      "epoch": 6.934833526611328e-05,
      "step": 72717,
      "training_step_time": 7.773102283477783
    },
    {
      "epoch": 6.934928894042969e-05,
      "model_forward_time": 1.7087416648864746,
      "step": 72718
    },
    {
      "epoch": 6.934928894042969e-05,
      "step": 72718,
      "training_step_time": 6.866891145706177
    },
    {
      "epoch": 6.935024261474609e-05,
      "model_forward_time": 1.7120487689971924,
      "step": 72719
    },
    {
      "epoch": 6.935024261474609e-05,
      "step": 72719,
      "training_step_time": 8.28794002532959
    },
    {
      "epoch": 6.93511962890625e-05,
      "grad_norm": 0.09459036588668823,
      "learning_rate": 1.9004416654485724e-06,
      "loss": 0.0533,
      "step": 72720
    },
    {
      "epoch": 6.93511962890625e-05,
      "model_forward_time": 1.7109222412109375,
      "step": 72720
    },
    {
      "dynamics_loss_avg": 0.049631389789283276,
      "epoch": 6.93511962890625e-05,
      "step": 72720
    },
    {
      "action_loss_avg": 0.0019318473583552986,
      "epoch": 6.93511962890625e-05,
      "step": 72720
    },
    {
      "epoch": 6.93511962890625e-05,
      "step": 72720,
      "training_step_time": 7.6246936321258545
    },
    {
      "epoch": 6.935214996337891e-05,
      "model_forward_time": 1.7378146648406982,
      "step": 72721
    },
    {
      "epoch": 6.935214996337891e-05,
      "step": 72721,
      "training_step_time": 7.653282403945923
    },
    {
      "epoch": 6.935310363769531e-05,
      "model_forward_time": 1.7170531749725342,
      "step": 72722
    },
    {
      "epoch": 6.935310363769531e-05,
      "step": 72722,
      "training_step_time": 7.782516717910767
    },
    {
      "epoch": 6.935405731201172e-05,
      "model_forward_time": 2.247812509536743,
      "step": 72723
    },
    {
      "epoch": 6.935405731201172e-05,
      "step": 72723,
      "training_step_time": 9.325258493423462
    },
    {
      "epoch": 6.935501098632813e-05,
      "model_forward_time": 1.7045021057128906,
      "step": 72724
    },
    {
      "epoch": 6.935501098632813e-05,
      "step": 72724,
      "training_step_time": 10.618629217147827
    },
    {
      "epoch": 6.935596466064453e-05,
      "model_forward_time": 1.7219369411468506,
      "step": 72725
    },
    {
      "epoch": 6.935596466064453e-05,
      "step": 72725,
      "training_step_time": 9.51656699180603
    },
    {
      "epoch": 6.935691833496094e-05,
      "model_forward_time": 1.715240478515625,
      "step": 72726
    },
    {
      "epoch": 6.935691833496094e-05,
      "step": 72726,
      "training_step_time": 7.758990287780762
    },
    {
      "epoch": 6.935787200927735e-05,
      "model_forward_time": 1.7091600894927979,
      "step": 72727
    },
    {
      "epoch": 6.935787200927735e-05,
      "step": 72727,
      "training_step_time": 7.570406675338745
    },
    {
      "epoch": 6.935882568359374e-05,
      "model_forward_time": 1.7264652252197266,
      "step": 72728
    },
    {
      "epoch": 6.935882568359374e-05,
      "step": 72728,
      "training_step_time": 7.60839056968689
    },
    {
      "epoch": 6.935977935791016e-05,
      "model_forward_time": 1.714796781539917,
      "step": 72729
    },
    {
      "epoch": 6.935977935791016e-05,
      "step": 72729,
      "training_step_time": 7.668020486831665
    },
    {
      "epoch": 6.936073303222657e-05,
      "grad_norm": 0.15474390983581543,
      "learning_rate": 1.8991444041406787e-06,
      "loss": 0.0563,
      "step": 72730
    },
    {
      "epoch": 6.936073303222657e-05,
      "model_forward_time": 1.7153313159942627,
      "step": 72730
    },
    {
      "dynamics_loss_avg": 0.053737998381257056,
      "epoch": 6.936073303222657e-05,
      "step": 72730
    },
    {
      "action_loss_avg": 0.0013776088890153915,
      "epoch": 6.936073303222657e-05,
      "step": 72730
    },
    {
      "epoch": 6.936073303222657e-05,
      "step": 72730,
      "training_step_time": 7.685077905654907
    },
    {
      "epoch": 6.936168670654296e-05,
      "model_forward_time": 1.7150797843933105,
      "step": 72731
    },
    {
      "epoch": 6.936168670654296e-05,
      "step": 72731,
      "training_step_time": 7.862596035003662
    },
    {
      "epoch": 6.936264038085937e-05,
      "model_forward_time": 2.2631359100341797,
      "step": 72732
    },
    {
      "epoch": 6.936264038085937e-05,
      "step": 72732,
      "training_step_time": 9.944608688354492
    },
    {
      "epoch": 6.936359405517579e-05,
      "model_forward_time": 1.713881254196167,
      "step": 72733
    },
    {
      "epoch": 6.936359405517579e-05,
      "step": 72733,
      "training_step_time": 9.431110620498657
    },
    {
      "epoch": 6.936454772949218e-05,
      "model_forward_time": 1.7180163860321045,
      "step": 72734
    },
    {
      "epoch": 6.936454772949218e-05,
      "step": 72734,
      "training_step_time": 8.434824705123901
    },
    {
      "epoch": 6.936550140380859e-05,
      "model_forward_time": 1.721400260925293,
      "step": 72735
    },
    {
      "epoch": 6.936550140380859e-05,
      "step": 72735,
      "training_step_time": 8.271968841552734
    },
    {
      "epoch": 6.9366455078125e-05,
      "model_forward_time": 1.7117938995361328,
      "step": 72736
    },
    {
      "epoch": 6.9366455078125e-05,
      "step": 72736,
      "training_step_time": 7.700276136398315
    },
    {
      "epoch": 6.93674087524414e-05,
      "model_forward_time": 1.7116003036499023,
      "step": 72737
    },
    {
      "epoch": 6.93674087524414e-05,
      "step": 72737,
      "training_step_time": 7.551676511764526
    },
    {
      "epoch": 6.936836242675781e-05,
      "model_forward_time": 1.7171800136566162,
      "step": 72738
    },
    {
      "epoch": 6.936836242675781e-05,
      "step": 72738,
      "training_step_time": 8.358917951583862
    },
    {
      "epoch": 6.936931610107422e-05,
      "model_forward_time": 1.7097923755645752,
      "step": 72739
    },
    {
      "epoch": 6.936931610107422e-05,
      "step": 72739,
      "training_step_time": 8.508036375045776
    },
    {
      "epoch": 6.937026977539062e-05,
      "grad_norm": 0.07378509640693665,
      "learning_rate": 1.8978474819376913e-06,
      "loss": 0.0551,
      "step": 72740
    },
    {
      "epoch": 6.937026977539062e-05,
      "model_forward_time": 1.8016016483306885,
      "step": 72740
    },
    {
      "dynamics_loss_avg": 0.05809307796880603,
      "epoch": 6.937026977539062e-05,
      "step": 72740
    },
    {
      "action_loss_avg": 0.0026287771470379084,
      "epoch": 6.937026977539062e-05,
      "step": 72740
    },
    {
      "epoch": 6.937026977539062e-05,
      "step": 72740,
      "training_step_time": 7.8373236656188965
    },
    {
      "epoch": 6.937122344970703e-05,
      "model_forward_time": 2.2575278282165527,
      "step": 72741
    },
    {
      "epoch": 6.937122344970703e-05,
      "step": 72741,
      "training_step_time": 10.324375629425049
    },
    {
      "epoch": 6.937217712402344e-05,
      "model_forward_time": 1.716468095779419,
      "step": 72742
    },
    {
      "epoch": 6.937217712402344e-05,
      "step": 72742,
      "training_step_time": 8.628758907318115
    },
    {
      "epoch": 6.937313079833984e-05,
      "model_forward_time": 1.7168986797332764,
      "step": 72743
    },
    {
      "epoch": 6.937313079833984e-05,
      "step": 72743,
      "training_step_time": 7.7577149868011475
    },
    {
      "epoch": 6.937408447265625e-05,
      "model_forward_time": 1.7246904373168945,
      "step": 72744
    },
    {
      "epoch": 6.937408447265625e-05,
      "step": 72744,
      "training_step_time": 7.662569999694824
    },
    {
      "epoch": 6.937503814697266e-05,
      "model_forward_time": 1.711395263671875,
      "step": 72745
    },
    {
      "epoch": 6.937503814697266e-05,
      "step": 72745,
      "training_step_time": 8.276356935501099
    },
    {
      "epoch": 6.937599182128906e-05,
      "model_forward_time": 1.7227938175201416,
      "step": 72746
    },
    {
      "epoch": 6.937599182128906e-05,
      "step": 72746,
      "training_step_time": 7.820465564727783
    },
    {
      "epoch": 6.937694549560547e-05,
      "model_forward_time": 1.7410163879394531,
      "step": 72747
    },
    {
      "epoch": 6.937694549560547e-05,
      "step": 72747,
      "training_step_time": 7.56702733039856
    },
    {
      "epoch": 6.937789916992188e-05,
      "model_forward_time": 1.731858491897583,
      "step": 72748
    },
    {
      "epoch": 6.937789916992188e-05,
      "step": 72748,
      "training_step_time": 7.4611976146698
    },
    {
      "epoch": 6.937885284423828e-05,
      "model_forward_time": 1.7964990139007568,
      "step": 72749
    },
    {
      "epoch": 6.937885284423828e-05,
      "step": 72749,
      "training_step_time": 7.8528947830200195
    },
    {
      "epoch": 6.937980651855469e-05,
      "grad_norm": 0.09676829725503922,
      "learning_rate": 1.8965508989814407e-06,
      "loss": 0.0535,
      "step": 72750
    },
    {
      "epoch": 6.937980651855469e-05,
      "model_forward_time": 2.299980878829956,
      "step": 72750
    },
    {
      "dynamics_loss_avg": 0.057333136163651946,
      "epoch": 6.937980651855469e-05,
      "step": 72750
    },
    {
      "action_loss_avg": 0.0015663344471249729,
      "epoch": 6.937980651855469e-05,
      "step": 72750
    },
    {
      "epoch": 6.937980651855469e-05,
      "step": 72750,
      "training_step_time": 7.9835076332092285
    },
    {
      "epoch": 6.93807601928711e-05,
      "model_forward_time": 1.7285304069519043,
      "step": 72751
    },
    {
      "epoch": 6.93807601928711e-05,
      "step": 72751,
      "training_step_time": 9.213534116744995
    },
    {
      "epoch": 6.93817138671875e-05,
      "model_forward_time": 1.7400689125061035,
      "step": 72752
    },
    {
      "epoch": 6.93817138671875e-05,
      "step": 72752,
      "training_step_time": 8.913603782653809
    },
    {
      "epoch": 6.93826675415039e-05,
      "model_forward_time": 1.7250933647155762,
      "step": 72753
    },
    {
      "epoch": 6.93826675415039e-05,
      "step": 72753,
      "training_step_time": 7.632412433624268
    },
    {
      "epoch": 6.938362121582032e-05,
      "model_forward_time": 1.7245888710021973,
      "step": 72754
    },
    {
      "epoch": 6.938362121582032e-05,
      "step": 72754,
      "training_step_time": 7.696542501449585
    },
    {
      "epoch": 6.938457489013671e-05,
      "model_forward_time": 1.712770700454712,
      "step": 72755
    },
    {
      "epoch": 6.938457489013671e-05,
      "step": 72755,
      "training_step_time": 7.079252243041992
    },
    {
      "epoch": 6.938552856445313e-05,
      "model_forward_time": 1.7297005653381348,
      "step": 72756
    },
    {
      "epoch": 6.938552856445313e-05,
      "step": 72756,
      "training_step_time": 8.22078800201416
    },
    {
      "epoch": 6.938648223876954e-05,
      "model_forward_time": 1.7381556034088135,
      "step": 72757
    },
    {
      "epoch": 6.938648223876954e-05,
      "step": 72757,
      "training_step_time": 8.243112325668335
    },
    {
      "epoch": 6.938743591308593e-05,
      "model_forward_time": 1.7842040061950684,
      "step": 72758
    },
    {
      "epoch": 6.938743591308593e-05,
      "step": 72758,
      "training_step_time": 8.720894813537598
    },
    {
      "epoch": 6.938838958740234e-05,
      "model_forward_time": 2.2735698223114014,
      "step": 72759
    },
    {
      "epoch": 6.938838958740234e-05,
      "step": 72759,
      "training_step_time": 8.474886178970337
    },
    {
      "epoch": 6.938934326171875e-05,
      "grad_norm": 0.08614014089107513,
      "learning_rate": 1.8952546554137157e-06,
      "loss": 0.0516,
      "step": 72760
    },
    {
      "epoch": 6.938934326171875e-05,
      "model_forward_time": 1.7282378673553467,
      "step": 72760
    },
    {
      "dynamics_loss_avg": 0.05600301120430231,
      "epoch": 6.938934326171875e-05,
      "step": 72760
    },
    {
      "action_loss_avg": 0.0011951149383094162,
      "epoch": 6.938934326171875e-05,
      "step": 72760
    },
    {
      "epoch": 6.938934326171875e-05,
      "step": 72760,
      "training_step_time": 8.712144374847412
    },
    {
      "epoch": 6.939029693603515e-05,
      "model_forward_time": 1.7413344383239746,
      "step": 72761
    },
    {
      "epoch": 6.939029693603515e-05,
      "step": 72761,
      "training_step_time": 7.866384029388428
    },
    {
      "epoch": 6.939125061035156e-05,
      "model_forward_time": 1.737276315689087,
      "step": 72762
    },
    {
      "epoch": 6.939125061035156e-05,
      "step": 72762,
      "training_step_time": 8.690861940383911
    },
    {
      "epoch": 6.939220428466797e-05,
      "model_forward_time": 1.3616464138031006,
      "step": 72763
    },
    {
      "epoch": 6.939220428466797e-05,
      "step": 72763,
      "training_step_time": 7.732576370239258
    },
    {
      "epoch": 6.939315795898437e-05,
      "model_forward_time": 1.7304699420928955,
      "step": 72764
    },
    {
      "epoch": 6.939315795898437e-05,
      "step": 72764,
      "training_step_time": 7.21918797492981
    },
    {
      "epoch": 6.939411163330078e-05,
      "model_forward_time": 1.7360775470733643,
      "step": 72765
    },
    {
      "epoch": 6.939411163330078e-05,
      "step": 72765,
      "training_step_time": 7.596399784088135
    },
    {
      "epoch": 6.939506530761719e-05,
      "model_forward_time": 1.7243905067443848,
      "step": 72766
    },
    {
      "epoch": 6.939506530761719e-05,
      "step": 72766,
      "training_step_time": 8.290710926055908
    },
    {
      "epoch": 6.939601898193359e-05,
      "model_forward_time": 1.7631163597106934,
      "step": 72767
    },
    {
      "epoch": 6.939601898193359e-05,
      "step": 72767,
      "training_step_time": 8.396454572677612
    },
    {
      "epoch": 6.939697265625e-05,
      "model_forward_time": 1.732527256011963,
      "step": 72768
    },
    {
      "epoch": 6.939697265625e-05,
      "step": 72768,
      "training_step_time": 9.91641116142273
    },
    {
      "epoch": 6.939792633056641e-05,
      "model_forward_time": 2.270324945449829,
      "step": 72769
    },
    {
      "epoch": 6.939792633056641e-05,
      "step": 72769,
      "training_step_time": 7.784822940826416
    },
    {
      "epoch": 6.939888000488281e-05,
      "grad_norm": 0.0676267221570015,
      "learning_rate": 1.893958751376276e-06,
      "loss": 0.0514,
      "step": 72770
    },
    {
      "epoch": 6.939888000488281e-05,
      "model_forward_time": 1.7287163734436035,
      "step": 72770
    },
    {
      "dynamics_loss_avg": 0.060876058787107466,
      "epoch": 6.939888000488281e-05,
      "step": 72770
    },
    {
      "action_loss_avg": 0.0017247892945306375,
      "epoch": 6.939888000488281e-05,
      "step": 72770
    },
    {
      "epoch": 6.939888000488281e-05,
      "step": 72770,
      "training_step_time": 8.524011850357056
    },
    {
      "epoch": 6.939983367919922e-05,
      "model_forward_time": 1.7318036556243896,
      "step": 72771
    },
    {
      "epoch": 6.939983367919922e-05,
      "step": 72771,
      "training_step_time": 7.826200008392334
    },
    {
      "epoch": 6.940078735351563e-05,
      "model_forward_time": 1.735015630722046,
      "step": 72772
    },
    {
      "epoch": 6.940078735351563e-05,
      "step": 72772,
      "training_step_time": 7.673922777175903
    },
    {
      "epoch": 6.940174102783203e-05,
      "model_forward_time": 1.728250503540039,
      "step": 72773
    },
    {
      "epoch": 6.940174102783203e-05,
      "step": 72773,
      "training_step_time": 7.757375478744507
    },
    {
      "epoch": 6.940269470214844e-05,
      "model_forward_time": 1.7315945625305176,
      "step": 72774
    },
    {
      "epoch": 6.940269470214844e-05,
      "step": 72774,
      "training_step_time": 7.571359634399414
    },
    {
      "epoch": 6.940364837646485e-05,
      "model_forward_time": 1.7325787544250488,
      "step": 72775
    },
    {
      "epoch": 6.940364837646485e-05,
      "step": 72775,
      "training_step_time": 7.446343898773193
    },
    {
      "epoch": 6.940460205078125e-05,
      "model_forward_time": 1.732093095779419,
      "step": 72776
    },
    {
      "epoch": 6.940460205078125e-05,
      "step": 72776,
      "training_step_time": 7.710654258728027
    },
    {
      "epoch": 6.940555572509766e-05,
      "model_forward_time": 1.7266008853912354,
      "step": 72777
    },
    {
      "epoch": 6.940555572509766e-05,
      "step": 72777,
      "training_step_time": 8.702715158462524
    },
    {
      "epoch": 6.940650939941407e-05,
      "model_forward_time": 2.2656967639923096,
      "step": 72778
    },
    {
      "epoch": 6.940650939941407e-05,
      "step": 72778,
      "training_step_time": 9.770467281341553
    },
    {
      "epoch": 6.940746307373047e-05,
      "model_forward_time": 1.7300612926483154,
      "step": 72779
    },
    {
      "epoch": 6.940746307373047e-05,
      "step": 72779,
      "training_step_time": 9.162043571472168
    },
    {
      "epoch": 6.940841674804688e-05,
      "grad_norm": 0.25868847966194153,
      "learning_rate": 1.8926631870108353e-06,
      "loss": 0.052,
      "step": 72780
    },
    {
      "epoch": 6.940841674804688e-05,
      "model_forward_time": 1.7331347465515137,
      "step": 72780
    },
    {
      "dynamics_loss_avg": 0.04598369598388672,
      "epoch": 6.940841674804688e-05,
      "step": 72780
    },
    {
      "action_loss_avg": 0.0015739834343548864,
      "epoch": 6.940841674804688e-05,
      "step": 72780
    },
    {
      "epoch": 6.940841674804688e-05,
      "step": 72780,
      "training_step_time": 7.81924843788147
    },
    {
      "epoch": 6.940937042236329e-05,
      "model_forward_time": 1.726581335067749,
      "step": 72781
    },
    {
      "epoch": 6.940937042236329e-05,
      "step": 72781,
      "training_step_time": 7.60228419303894
    },
    {
      "epoch": 6.941032409667968e-05,
      "model_forward_time": 1.7338590621948242,
      "step": 72782
    },
    {
      "epoch": 6.941032409667968e-05,
      "step": 72782,
      "training_step_time": 7.743259906768799
    },
    {
      "epoch": 6.94112777709961e-05,
      "model_forward_time": 1.7319142818450928,
      "step": 72783
    },
    {
      "epoch": 6.94112777709961e-05,
      "step": 72783,
      "training_step_time": 7.5425026416778564
    },
    {
      "epoch": 6.94122314453125e-05,
      "model_forward_time": 1.7278838157653809,
      "step": 72784
    },
    {
      "epoch": 6.94122314453125e-05,
      "step": 72784,
      "training_step_time": 8.276053190231323
    },
    {
      "epoch": 6.94131851196289e-05,
      "model_forward_time": 1.7218592166900635,
      "step": 72785
    },
    {
      "epoch": 6.94131851196289e-05,
      "step": 72785,
      "training_step_time": 8.61457085609436
    },
    {
      "epoch": 6.941413879394531e-05,
      "model_forward_time": 1.7225162982940674,
      "step": 72786
    },
    {
      "epoch": 6.941413879394531e-05,
      "step": 72786,
      "training_step_time": 7.939728021621704
    },
    {
      "epoch": 6.941509246826172e-05,
      "model_forward_time": 2.2833399772644043,
      "step": 72787
    },
    {
      "epoch": 6.941509246826172e-05,
      "step": 72787,
      "training_step_time": 8.22892451286316
    },
    {
      "epoch": 6.941604614257812e-05,
      "model_forward_time": 1.7115452289581299,
      "step": 72788
    },
    {
      "epoch": 6.941604614257812e-05,
      "step": 72788,
      "training_step_time": 7.814199924468994
    },
    {
      "epoch": 6.941699981689453e-05,
      "model_forward_time": 1.7200102806091309,
      "step": 72789
    },
    {
      "epoch": 6.941699981689453e-05,
      "step": 72789,
      "training_step_time": 7.887888193130493
    },
    {
      "epoch": 6.941795349121094e-05,
      "grad_norm": 0.08823473006486893,
      "learning_rate": 1.8913679624590769e-06,
      "loss": 0.0523,
      "step": 72790
    },
    {
      "epoch": 6.941795349121094e-05,
      "model_forward_time": 1.3497164249420166,
      "step": 72790
    },
    {
      "dynamics_loss_avg": 0.0726372942328453,
      "epoch": 6.941795349121094e-05,
      "step": 72790
    },
    {
      "action_loss_avg": 0.001567376166349277,
      "epoch": 6.941795349121094e-05,
      "step": 72790
    },
    {
      "epoch": 6.941795349121094e-05,
      "step": 72790,
      "training_step_time": 7.672916650772095
    },
    {
      "epoch": 6.941890716552734e-05,
      "model_forward_time": 1.7273805141448975,
      "step": 72791
    },
    {
      "epoch": 6.941890716552734e-05,
      "step": 72791,
      "training_step_time": 7.740698575973511
    },
    {
      "epoch": 6.941986083984375e-05,
      "model_forward_time": 1.7194254398345947,
      "step": 72792
    },
    {
      "epoch": 6.941986083984375e-05,
      "step": 72792,
      "training_step_time": 7.560470819473267
    },
    {
      "epoch": 6.942081451416016e-05,
      "model_forward_time": 1.727278232574463,
      "step": 72793
    },
    {
      "epoch": 6.942081451416016e-05,
      "step": 72793,
      "training_step_time": 8.238890171051025
    },
    {
      "epoch": 6.942176818847656e-05,
      "model_forward_time": 1.7143685817718506,
      "step": 72794
    },
    {
      "epoch": 6.942176818847656e-05,
      "step": 72794,
      "training_step_time": 7.613021612167358
    },
    {
      "epoch": 6.942272186279297e-05,
      "model_forward_time": 1.8084437847137451,
      "step": 72795
    },
    {
      "epoch": 6.942272186279297e-05,
      "step": 72795,
      "training_step_time": 7.922480821609497
    },
    {
      "epoch": 6.942367553710938e-05,
      "model_forward_time": 2.2620670795440674,
      "step": 72796
    },
    {
      "epoch": 6.942367553710938e-05,
      "step": 72796,
      "training_step_time": 8.360910892486572
    },
    {
      "epoch": 6.942462921142578e-05,
      "model_forward_time": 1.7097399234771729,
      "step": 72797
    },
    {
      "epoch": 6.942462921142578e-05,
      "step": 72797,
      "training_step_time": 9.103055238723755
    },
    {
      "epoch": 6.942558288574219e-05,
      "model_forward_time": 1.7200021743774414,
      "step": 72798
    },
    {
      "epoch": 6.942558288574219e-05,
      "step": 72798,
      "training_step_time": 8.577699184417725
    },
    {
      "epoch": 6.94265365600586e-05,
      "model_forward_time": 1.7210543155670166,
      "step": 72799
    },
    {
      "epoch": 6.94265365600586e-05,
      "step": 72799,
      "training_step_time": 7.737155437469482
    },
    {
      "epoch": 6.9427490234375e-05,
      "grad_norm": 0.13359498977661133,
      "learning_rate": 1.8900730778626442e-06,
      "loss": 0.0536,
      "step": 72800
    },
    {
      "epoch": 6.9427490234375e-05,
      "model_forward_time": 1.7314300537109375,
      "step": 72800
    },
    {
      "dynamics_loss_avg": 0.06733765024691821,
      "epoch": 6.9427490234375e-05,
      "step": 72800
    },
    {
      "action_loss_avg": 0.002166394202504307,
      "epoch": 6.9427490234375e-05,
      "step": 72800
    },
    {
      "epoch": 6.9427490234375e-05,
      "step": 72800,
      "training_step_time": 7.737879276275635
    },
    {
      "epoch": 6.942844390869141e-05,
      "model_forward_time": 1.715336561203003,
      "step": 72801
    },
    {
      "epoch": 6.942844390869141e-05,
      "step": 72801,
      "training_step_time": 7.605193138122559
    },
    {
      "epoch": 6.942939758300782e-05,
      "model_forward_time": 1.7204208374023438,
      "step": 72802
    },
    {
      "epoch": 6.942939758300782e-05,
      "step": 72802,
      "training_step_time": 8.230771541595459
    },
    {
      "epoch": 6.943035125732422e-05,
      "model_forward_time": 1.3537321090698242,
      "step": 72803
    },
    {
      "epoch": 6.943035125732422e-05,
      "step": 72803,
      "training_step_time": 7.5912699699401855
    },
    {
      "epoch": 6.943130493164063e-05,
      "model_forward_time": 1.7930536270141602,
      "step": 72804
    },
    {
      "epoch": 6.943130493164063e-05,
      "step": 72804,
      "training_step_time": 7.938395023345947
    },
    {
      "epoch": 6.943225860595704e-05,
      "model_forward_time": 1.7028741836547852,
      "step": 72805
    },
    {
      "epoch": 6.943225860595704e-05,
      "step": 72805,
      "training_step_time": 7.9182116985321045
    },
    {
      "epoch": 6.943321228027344e-05,
      "model_forward_time": 2.2619895935058594,
      "step": 72806
    },
    {
      "epoch": 6.943321228027344e-05,
      "step": 72806,
      "training_step_time": 8.441669702529907
    },
    {
      "epoch": 6.943416595458985e-05,
      "model_forward_time": 1.7204856872558594,
      "step": 72807
    },
    {
      "epoch": 6.943416595458985e-05,
      "step": 72807,
      "training_step_time": 9.534909725189209
    },
    {
      "epoch": 6.943511962890626e-05,
      "model_forward_time": 1.697479248046875,
      "step": 72808
    },
    {
      "epoch": 6.943511962890626e-05,
      "step": 72808,
      "training_step_time": 8.340747356414795
    },
    {
      "epoch": 6.943607330322265e-05,
      "model_forward_time": 1.7215509414672852,
      "step": 72809
    },
    {
      "epoch": 6.943607330322265e-05,
      "step": 72809,
      "training_step_time": 7.710815668106079
    },
    {
      "epoch": 6.943702697753906e-05,
      "grad_norm": 0.07830336689949036,
      "learning_rate": 1.8887785333631459e-06,
      "loss": 0.0517,
      "step": 72810
    },
    {
      "epoch": 6.943702697753906e-05,
      "model_forward_time": 1.728963851928711,
      "step": 72810
    },
    {
      "dynamics_loss_avg": 0.04895954504609108,
      "epoch": 6.943702697753906e-05,
      "step": 72810
    },
    {
      "action_loss_avg": 0.0029697219841182234,
      "epoch": 6.943702697753906e-05,
      "step": 72810
    },
    {
      "epoch": 6.943702697753906e-05,
      "step": 72810,
      "training_step_time": 8.763436555862427
    },
    {
      "epoch": 6.943798065185548e-05,
      "model_forward_time": 1.722733736038208,
      "step": 72811
    },
    {
      "epoch": 6.943798065185548e-05,
      "step": 72811,
      "training_step_time": 7.8129706382751465
    },
    {
      "epoch": 6.943893432617187e-05,
      "model_forward_time": 1.7227938175201416,
      "step": 72812
    },
    {
      "epoch": 6.943893432617187e-05,
      "step": 72812,
      "training_step_time": 7.643768787384033
    },
    {
      "epoch": 6.943988800048828e-05,
      "model_forward_time": 1.7736942768096924,
      "step": 72813
    },
    {
      "epoch": 6.943988800048828e-05,
      "step": 72813,
      "training_step_time": 7.791590690612793
    },
    {
      "epoch": 6.944084167480468e-05,
      "model_forward_time": 1.7275586128234863,
      "step": 72814
    },
    {
      "epoch": 6.944084167480468e-05,
      "step": 72814,
      "training_step_time": 7.926077127456665
    },
    {
      "epoch": 6.944179534912109e-05,
      "model_forward_time": 2.271331310272217,
      "step": 72815
    },
    {
      "epoch": 6.944179534912109e-05,
      "step": 72815,
      "training_step_time": 8.737866401672363
    },
    {
      "epoch": 6.94427490234375e-05,
      "model_forward_time": 1.724461317062378,
      "step": 72816
    },
    {
      "epoch": 6.94427490234375e-05,
      "step": 72816,
      "training_step_time": 7.753664255142212
    },
    {
      "epoch": 6.94437026977539e-05,
      "model_forward_time": 1.72761869430542,
      "step": 72817
    },
    {
      "epoch": 6.94437026977539e-05,
      "step": 72817,
      "training_step_time": 7.83656907081604
    },
    {
      "epoch": 6.944465637207031e-05,
      "model_forward_time": 1.7269985675811768,
      "step": 72818
    },
    {
      "epoch": 6.944465637207031e-05,
      "step": 72818,
      "training_step_time": 7.772604703903198
    },
    {
      "epoch": 6.944561004638672e-05,
      "model_forward_time": 1.7261998653411865,
      "step": 72819
    },
    {
      "epoch": 6.944561004638672e-05,
      "step": 72819,
      "training_step_time": 8.3030264377594
    },
    {
      "epoch": 6.944656372070312e-05,
      "grad_norm": 0.11575642973184586,
      "learning_rate": 1.8874843291021477e-06,
      "loss": 0.0526,
      "step": 72820
    },
    {
      "epoch": 6.944656372070312e-05,
      "model_forward_time": 1.7214887142181396,
      "step": 72820
    },
    {
      "dynamics_loss_avg": 0.03999807834625244,
      "epoch": 6.944656372070312e-05,
      "step": 72820
    },
    {
      "action_loss_avg": 0.0016978527884930373,
      "epoch": 6.944656372070312e-05,
      "step": 72820
    },
    {
      "epoch": 6.944656372070312e-05,
      "step": 72820,
      "training_step_time": 7.292950391769409
    },
    {
      "epoch": 6.944751739501953e-05,
      "model_forward_time": 1.7290434837341309,
      "step": 72821
    },
    {
      "epoch": 6.944751739501953e-05,
      "step": 72821,
      "training_step_time": 7.67539119720459
    },
    {
      "epoch": 6.944847106933594e-05,
      "model_forward_time": 1.744995355606079,
      "step": 72822
    },
    {
      "epoch": 6.944847106933594e-05,
      "step": 72822,
      "training_step_time": 9.063035726547241
    },
    {
      "epoch": 6.944942474365234e-05,
      "model_forward_time": 1.7230854034423828,
      "step": 72823
    },
    {
      "epoch": 6.944942474365234e-05,
      "step": 72823,
      "training_step_time": 7.89822244644165
    },
    {
      "epoch": 6.945037841796875e-05,
      "model_forward_time": 2.2552871704101562,
      "step": 72824
    },
    {
      "epoch": 6.945037841796875e-05,
      "step": 72824,
      "training_step_time": 7.953577995300293
    },
    {
      "epoch": 6.945133209228516e-05,
      "model_forward_time": 1.3414576053619385,
      "step": 72825
    },
    {
      "epoch": 6.945133209228516e-05,
      "step": 72825,
      "training_step_time": 9.01189637184143
    },
    {
      "epoch": 6.945228576660156e-05,
      "model_forward_time": 1.7233147621154785,
      "step": 72826
    },
    {
      "epoch": 6.945228576660156e-05,
      "step": 72826,
      "training_step_time": 7.73037314414978
    },
    {
      "epoch": 6.945323944091797e-05,
      "model_forward_time": 1.7239813804626465,
      "step": 72827
    },
    {
      "epoch": 6.945323944091797e-05,
      "step": 72827,
      "training_step_time": 7.630282163619995
    },
    {
      "epoch": 6.945419311523438e-05,
      "model_forward_time": 1.7295491695404053,
      "step": 72828
    },
    {
      "epoch": 6.945419311523438e-05,
      "step": 72828,
      "training_step_time": 7.658489465713501
    },
    {
      "epoch": 6.945514678955078e-05,
      "model_forward_time": 1.7247209548950195,
      "step": 72829
    },
    {
      "epoch": 6.945514678955078e-05,
      "step": 72829,
      "training_step_time": 7.842914342880249
    },
    {
      "epoch": 6.945610046386719e-05,
      "grad_norm": 0.08403687924146652,
      "learning_rate": 1.8861904652211854e-06,
      "loss": 0.0522,
      "step": 72830
    },
    {
      "epoch": 6.945610046386719e-05,
      "model_forward_time": 1.7163257598876953,
      "step": 72830
    },
    {
      "dynamics_loss_avg": 0.05683945454657078,
      "epoch": 6.945610046386719e-05,
      "step": 72830
    },
    {
      "action_loss_avg": 0.0036020477826241406,
      "epoch": 6.945610046386719e-05,
      "step": 72830
    },
    {
      "epoch": 6.945610046386719e-05,
      "step": 72830,
      "training_step_time": 7.711568355560303
    },
    {
      "epoch": 6.94570541381836e-05,
      "model_forward_time": 1.7192175388336182,
      "step": 72831
    },
    {
      "epoch": 6.94570541381836e-05,
      "step": 72831,
      "training_step_time": 7.622752666473389
    },
    {
      "epoch": 6.94580078125e-05,
      "model_forward_time": 1.7219820022583008,
      "step": 72832
    },
    {
      "epoch": 6.94580078125e-05,
      "step": 72832,
      "training_step_time": 8.052096128463745
    },
    {
      "epoch": 6.94589614868164e-05,
      "model_forward_time": 2.2457191944122314,
      "step": 72833
    },
    {
      "epoch": 6.94589614868164e-05,
      "step": 72833,
      "training_step_time": 8.589625120162964
    },
    {
      "epoch": 6.945991516113282e-05,
      "model_forward_time": 1.719928503036499,
      "step": 72834
    },
    {
      "epoch": 6.945991516113282e-05,
      "step": 72834,
      "training_step_time": 8.478431940078735
    },
    {
      "epoch": 6.946086883544921e-05,
      "model_forward_time": 1.7201995849609375,
      "step": 72835
    },
    {
      "epoch": 6.946086883544921e-05,
      "step": 72835,
      "training_step_time": 7.855623722076416
    },
    {
      "epoch": 6.946182250976562e-05,
      "model_forward_time": 1.733729362487793,
      "step": 72836
    },
    {
      "epoch": 6.946182250976562e-05,
      "step": 72836,
      "training_step_time": 7.576311111450195
    },
    {
      "epoch": 6.946277618408203e-05,
      "model_forward_time": 1.7229080200195312,
      "step": 72837
    },
    {
      "epoch": 6.946277618408203e-05,
      "step": 72837,
      "training_step_time": 7.584425926208496
    },
    {
      "epoch": 6.946372985839843e-05,
      "model_forward_time": 1.7203609943389893,
      "step": 72838
    },
    {
      "epoch": 6.946372985839843e-05,
      "step": 72838,
      "training_step_time": 7.765095233917236
    },
    {
      "epoch": 6.946468353271484e-05,
      "model_forward_time": 1.7241780757904053,
      "step": 72839
    },
    {
      "epoch": 6.946468353271484e-05,
      "step": 72839,
      "training_step_time": 7.647706747055054
    },
    {
      "epoch": 6.946563720703125e-05,
      "grad_norm": 0.08724190294742584,
      "learning_rate": 1.8848969418617507e-06,
      "loss": 0.0525,
      "step": 72840
    },
    {
      "epoch": 6.946563720703125e-05,
      "model_forward_time": 1.733062744140625,
      "step": 72840
    },
    {
      "dynamics_loss_avg": 0.047264781687408684,
      "epoch": 6.946563720703125e-05,
      "step": 72840
    },
    {
      "action_loss_avg": 0.002392975054681301,
      "epoch": 6.946563720703125e-05,
      "step": 72840
    },
    {
      "epoch": 6.946563720703125e-05,
      "step": 72840,
      "training_step_time": 8.500374794006348
    },
    {
      "epoch": 6.946659088134765e-05,
      "model_forward_time": 1.8113694190979004,
      "step": 72841
    },
    {
      "epoch": 6.946659088134765e-05,
      "step": 72841,
      "training_step_time": 8.945867538452148
    },
    {
      "epoch": 6.946754455566406e-05,
      "model_forward_time": 1.6909804344177246,
      "step": 72842
    },
    {
      "epoch": 6.946754455566406e-05,
      "step": 72842,
      "training_step_time": 8.447894811630249
    },
    {
      "epoch": 6.946849822998047e-05,
      "model_forward_time": 2.2729814052581787,
      "step": 72843
    },
    {
      "epoch": 6.946849822998047e-05,
      "step": 72843,
      "training_step_time": 10.064559936523438
    },
    {
      "epoch": 6.946945190429687e-05,
      "model_forward_time": 1.7192838191986084,
      "step": 72844
    },
    {
      "epoch": 6.946945190429687e-05,
      "step": 72844,
      "training_step_time": 9.30784559249878
    },
    {
      "epoch": 6.947040557861328e-05,
      "model_forward_time": 1.7287321090698242,
      "step": 72845
    },
    {
      "epoch": 6.947040557861328e-05,
      "step": 72845,
      "training_step_time": 7.926559686660767
    },
    {
      "epoch": 6.947135925292969e-05,
      "model_forward_time": 1.7260427474975586,
      "step": 72846
    },
    {
      "epoch": 6.947135925292969e-05,
      "step": 72846,
      "training_step_time": 7.675095081329346
    },
    {
      "epoch": 6.947231292724609e-05,
      "model_forward_time": 1.715214729309082,
      "step": 72847
    },
    {
      "epoch": 6.947231292724609e-05,
      "step": 72847,
      "training_step_time": 7.735973358154297
    },
    {
      "epoch": 6.94732666015625e-05,
      "model_forward_time": 1.7284667491912842,
      "step": 72848
    },
    {
      "epoch": 6.94732666015625e-05,
      "step": 72848,
      "training_step_time": 7.658571720123291
    },
    {
      "epoch": 6.947422027587891e-05,
      "model_forward_time": 1.731222152709961,
      "step": 72849
    },
    {
      "epoch": 6.947422027587891e-05,
      "step": 72849,
      "training_step_time": 9.056546688079834
    },
    {
      "epoch": 6.947517395019531e-05,
      "grad_norm": 0.12509237229824066,
      "learning_rate": 1.8836037591653044e-06,
      "loss": 0.0542,
      "step": 72850
    },
    {
      "epoch": 6.947517395019531e-05,
      "model_forward_time": 1.7694683074951172,
      "step": 72850
    },
    {
      "dynamics_loss_avg": 0.04944656528532505,
      "epoch": 6.947517395019531e-05,
      "step": 72850
    },
    {
      "action_loss_avg": 0.0027436667936854065,
      "epoch": 6.947517395019531e-05,
      "step": 72850
    },
    {
      "epoch": 6.947517395019531e-05,
      "step": 72850,
      "training_step_time": 8.808027267456055
    },
    {
      "epoch": 6.947612762451172e-05,
      "model_forward_time": 1.7270314693450928,
      "step": 72851
    },
    {
      "epoch": 6.947612762451172e-05,
      "step": 72851,
      "training_step_time": 7.865371227264404
    },
    {
      "epoch": 6.947708129882813e-05,
      "model_forward_time": 2.2605628967285156,
      "step": 72852
    },
    {
      "epoch": 6.947708129882813e-05,
      "step": 72852,
      "training_step_time": 10.032877206802368
    },
    {
      "epoch": 6.947803497314453e-05,
      "model_forward_time": 1.722886562347412,
      "step": 72853
    },
    {
      "epoch": 6.947803497314453e-05,
      "step": 72853,
      "training_step_time": 7.721086025238037
    },
    {
      "epoch": 6.947898864746094e-05,
      "model_forward_time": 1.7306911945343018,
      "step": 72854
    },
    {
      "epoch": 6.947898864746094e-05,
      "step": 72854,
      "training_step_time": 9.816613674163818
    },
    {
      "epoch": 6.947994232177735e-05,
      "model_forward_time": 1.728548526763916,
      "step": 72855
    },
    {
      "epoch": 6.947994232177735e-05,
      "step": 72855,
      "training_step_time": 8.138181209564209
    },
    {
      "epoch": 6.948089599609374e-05,
      "model_forward_time": 1.722700834274292,
      "step": 72856
    },
    {
      "epoch": 6.948089599609374e-05,
      "step": 72856,
      "training_step_time": 8.204172134399414
    },
    {
      "epoch": 6.948184967041016e-05,
      "model_forward_time": 1.7287416458129883,
      "step": 72857
    },
    {
      "epoch": 6.948184967041016e-05,
      "step": 72857,
      "training_step_time": 7.99240779876709
    },
    {
      "epoch": 6.948280334472657e-05,
      "model_forward_time": 1.7297232151031494,
      "step": 72858
    },
    {
      "epoch": 6.948280334472657e-05,
      "step": 72858,
      "training_step_time": 7.594327688217163
    },
    {
      "epoch": 6.948375701904296e-05,
      "model_forward_time": 1.7580626010894775,
      "step": 72859
    },
    {
      "epoch": 6.948375701904296e-05,
      "step": 72859,
      "training_step_time": 7.710085868835449
    },
    {
      "epoch": 6.948471069335937e-05,
      "grad_norm": 0.09025824069976807,
      "learning_rate": 1.8823109172732622e-06,
      "loss": 0.0511,
      "step": 72860
    },
    {
      "epoch": 6.948471069335937e-05,
      "model_forward_time": 1.7318382263183594,
      "step": 72860
    },
    {
      "dynamics_loss_avg": 0.04740372272208333,
      "epoch": 6.948471069335937e-05,
      "step": 72860
    },
    {
      "action_loss_avg": 0.0018465657863998785,
      "epoch": 6.948471069335937e-05,
      "step": 72860
    },
    {
      "epoch": 6.948471069335937e-05,
      "step": 72860,
      "training_step_time": 8.455227613449097
    },
    {
      "epoch": 6.948566436767579e-05,
      "model_forward_time": 2.313112735748291,
      "step": 72861
    },
    {
      "epoch": 6.948566436767579e-05,
      "step": 72861,
      "training_step_time": 9.165502071380615
    },
    {
      "epoch": 6.948661804199218e-05,
      "model_forward_time": 1.7351691722869873,
      "step": 72862
    },
    {
      "epoch": 6.948661804199218e-05,
      "step": 72862,
      "training_step_time": 8.690122127532959
    },
    {
      "epoch": 6.94875717163086e-05,
      "model_forward_time": 1.742476224899292,
      "step": 72863
    },
    {
      "epoch": 6.94875717163086e-05,
      "step": 72863,
      "training_step_time": 7.835927248001099
    },
    {
      "epoch": 6.9488525390625e-05,
      "model_forward_time": 1.736748218536377,
      "step": 72864
    },
    {
      "epoch": 6.9488525390625e-05,
      "step": 72864,
      "training_step_time": 8.894093990325928
    },
    {
      "epoch": 6.94894790649414e-05,
      "model_forward_time": 1.7336106300354004,
      "step": 72865
    },
    {
      "epoch": 6.94894790649414e-05,
      "step": 72865,
      "training_step_time": 8.715283155441284
    },
    {
      "epoch": 6.949043273925781e-05,
      "model_forward_time": 1.7483172416687012,
      "step": 72866
    },
    {
      "epoch": 6.949043273925781e-05,
      "step": 72866,
      "training_step_time": 7.861764907836914
    },
    {
      "epoch": 6.949138641357422e-05,
      "model_forward_time": 1.754323959350586,
      "step": 72867
    },
    {
      "epoch": 6.949138641357422e-05,
      "step": 72867,
      "training_step_time": 7.650761842727661
    },
    {
      "epoch": 6.949234008789062e-05,
      "model_forward_time": 1.7419826984405518,
      "step": 72868
    },
    {
      "epoch": 6.949234008789062e-05,
      "step": 72868,
      "training_step_time": 7.669180631637573
    },
    {
      "epoch": 6.949329376220703e-05,
      "model_forward_time": 1.741743564605713,
      "step": 72869
    },
    {
      "epoch": 6.949329376220703e-05,
      "step": 72869,
      "training_step_time": 8.801754713058472
    },
    {
      "epoch": 6.949424743652344e-05,
      "grad_norm": 0.10590652376413345,
      "learning_rate": 1.881018416327014e-06,
      "loss": 0.0502,
      "step": 72870
    },
    {
      "epoch": 6.949424743652344e-05,
      "model_forward_time": 2.298447370529175,
      "step": 72870
    },
    {
      "dynamics_loss_avg": 0.047097553685307506,
      "epoch": 6.949424743652344e-05,
      "step": 72870
    },
    {
      "action_loss_avg": 0.0014150503353448584,
      "epoch": 6.949424743652344e-05,
      "step": 72870
    },
    {
      "epoch": 6.949424743652344e-05,
      "step": 72870,
      "training_step_time": 7.879494905471802
    },
    {
      "epoch": 6.949520111083984e-05,
      "model_forward_time": 1.73734712600708,
      "step": 72871
    },
    {
      "epoch": 6.949520111083984e-05,
      "step": 72871,
      "training_step_time": 8.409677982330322
    },
    {
      "epoch": 6.949615478515625e-05,
      "model_forward_time": 1.7455322742462158,
      "step": 72872
    },
    {
      "epoch": 6.949615478515625e-05,
      "step": 72872,
      "training_step_time": 8.317012310028076
    },
    {
      "epoch": 6.949710845947266e-05,
      "model_forward_time": 1.7485287189483643,
      "step": 72873
    },
    {
      "epoch": 6.949710845947266e-05,
      "step": 72873,
      "training_step_time": 7.724126815795898
    },
    {
      "epoch": 6.949806213378906e-05,
      "model_forward_time": 1.738401174545288,
      "step": 72874
    },
    {
      "epoch": 6.949806213378906e-05,
      "step": 72874,
      "training_step_time": 7.687424659729004
    },
    {
      "epoch": 6.949901580810547e-05,
      "model_forward_time": 1.7438838481903076,
      "step": 72875
    },
    {
      "epoch": 6.949901580810547e-05,
      "step": 72875,
      "training_step_time": 7.776824474334717
    },
    {
      "epoch": 6.949996948242188e-05,
      "model_forward_time": 1.7423756122589111,
      "step": 72876
    },
    {
      "epoch": 6.949996948242188e-05,
      "step": 72876,
      "training_step_time": 7.689695119857788
    },
    {
      "epoch": 6.950092315673828e-05,
      "model_forward_time": 1.737245798110962,
      "step": 72877
    },
    {
      "epoch": 6.950092315673828e-05,
      "step": 72877,
      "training_step_time": 7.568675994873047
    },
    {
      "epoch": 6.950187683105469e-05,
      "model_forward_time": 1.8119750022888184,
      "step": 72878
    },
    {
      "epoch": 6.950187683105469e-05,
      "step": 72878,
      "training_step_time": 9.546266078948975
    },
    {
      "epoch": 6.95028305053711e-05,
      "model_forward_time": 2.277244806289673,
      "step": 72879
    },
    {
      "epoch": 6.95028305053711e-05,
      "step": 72879,
      "training_step_time": 8.43736982345581
    },
    {
      "epoch": 6.95037841796875e-05,
      "grad_norm": 0.10117733478546143,
      "learning_rate": 1.8797262564679014e-06,
      "loss": 0.0511,
      "step": 72880
    },
    {
      "epoch": 6.95037841796875e-05,
      "model_forward_time": 1.734954595565796,
      "step": 72880
    },
    {
      "dynamics_loss_avg": 0.05158014465123415,
      "epoch": 6.95037841796875e-05,
      "step": 72880
    },
    {
      "action_loss_avg": 0.0011853803356643767,
      "epoch": 6.95037841796875e-05,
      "step": 72880
    },
    {
      "epoch": 6.95037841796875e-05,
      "step": 72880,
      "training_step_time": 9.605118989944458
    },
    {
      "epoch": 6.95047378540039e-05,
      "model_forward_time": 1.7307348251342773,
      "step": 72881
    },
    {
      "epoch": 6.95047378540039e-05,
      "step": 72881,
      "training_step_time": 7.812888145446777
    },
    {
      "epoch": 6.950569152832032e-05,
      "model_forward_time": 1.733346700668335,
      "step": 72882
    },
    {
      "epoch": 6.950569152832032e-05,
      "step": 72882,
      "training_step_time": 8.877794742584229
    },
    {
      "epoch": 6.950664520263671e-05,
      "model_forward_time": 1.3727495670318604,
      "step": 72883
    },
    {
      "epoch": 6.950664520263671e-05,
      "step": 72883,
      "training_step_time": 8.337836027145386
    },
    {
      "epoch": 6.950759887695313e-05,
      "model_forward_time": 1.7400579452514648,
      "step": 72884
    },
    {
      "epoch": 6.950759887695313e-05,
      "step": 72884,
      "training_step_time": 8.321057796478271
    },
    {
      "epoch": 6.950855255126954e-05,
      "model_forward_time": 1.7416462898254395,
      "step": 72885
    },
    {
      "epoch": 6.950855255126954e-05,
      "step": 72885,
      "training_step_time": 7.652612924575806
    },
    {
      "epoch": 6.950950622558593e-05,
      "model_forward_time": 1.731661081314087,
      "step": 72886
    },
    {
      "epoch": 6.950950622558593e-05,
      "step": 72886,
      "training_step_time": 7.614989757537842
    },
    {
      "epoch": 6.951045989990234e-05,
      "model_forward_time": 1.8076958656311035,
      "step": 72887
    },
    {
      "epoch": 6.951045989990234e-05,
      "step": 72887,
      "training_step_time": 7.892627477645874
    },
    {
      "epoch": 6.951141357421875e-05,
      "model_forward_time": 1.0238044261932373,
      "step": 72888
    },
    {
      "epoch": 6.951141357421875e-05,
      "step": 72888,
      "training_step_time": 7.964416265487671
    },
    {
      "epoch": 6.951236724853515e-05,
      "model_forward_time": 2.283412456512451,
      "step": 72889
    },
    {
      "epoch": 6.951236724853515e-05,
      "step": 72889,
      "training_step_time": 8.855128526687622
    },
    {
      "epoch": 6.951332092285156e-05,
      "grad_norm": 0.09267907589673996,
      "learning_rate": 1.8784344378372355e-06,
      "loss": 0.0545,
      "step": 72890
    },
    {
      "epoch": 6.951332092285156e-05,
      "model_forward_time": 1.7592732906341553,
      "step": 72890
    },
    {
      "dynamics_loss_avg": 0.06031344272196293,
      "epoch": 6.951332092285156e-05,
      "step": 72890
    },
    {
      "action_loss_avg": 0.0016325649747159332,
      "epoch": 6.951332092285156e-05,
      "step": 72890
    },
    {
      "epoch": 6.951332092285156e-05,
      "step": 72890,
      "training_step_time": 7.689920425415039
    },
    {
      "epoch": 6.951427459716797e-05,
      "model_forward_time": 1.7431302070617676,
      "step": 72891
    },
    {
      "epoch": 6.951427459716797e-05,
      "step": 72891,
      "training_step_time": 7.7400898933410645
    },
    {
      "epoch": 6.951522827148437e-05,
      "model_forward_time": 1.7231230735778809,
      "step": 72892
    },
    {
      "epoch": 6.951522827148437e-05,
      "step": 72892,
      "training_step_time": 7.721068382263184
    },
    {
      "epoch": 6.951618194580078e-05,
      "model_forward_time": 1.7312872409820557,
      "step": 72893
    },
    {
      "epoch": 6.951618194580078e-05,
      "step": 72893,
      "training_step_time": 7.681437253952026
    },
    {
      "epoch": 6.951713562011719e-05,
      "model_forward_time": 1.726625680923462,
      "step": 72894
    },
    {
      "epoch": 6.951713562011719e-05,
      "step": 72894,
      "training_step_time": 7.688672780990601
    },
    {
      "epoch": 6.951808929443359e-05,
      "model_forward_time": 1.720979928970337,
      "step": 72895
    },
    {
      "epoch": 6.951808929443359e-05,
      "step": 72895,
      "training_step_time": 9.326263189315796
    },
    {
      "epoch": 6.951904296875e-05,
      "model_forward_time": 1.7691049575805664,
      "step": 72896
    },
    {
      "epoch": 6.951904296875e-05,
      "step": 72896,
      "training_step_time": 7.934990882873535
    },
    {
      "epoch": 6.951999664306641e-05,
      "model_forward_time": 1.7193591594696045,
      "step": 72897
    },
    {
      "epoch": 6.951999664306641e-05,
      "step": 72897,
      "training_step_time": 10.078838586807251
    },
    {
      "epoch": 6.952095031738281e-05,
      "model_forward_time": 2.2914321422576904,
      "step": 72898
    },
    {
      "epoch": 6.952095031738281e-05,
      "step": 72898,
      "training_step_time": 8.491247177124023
    },
    {
      "epoch": 6.952190399169922e-05,
      "model_forward_time": 1.7266595363616943,
      "step": 72899
    },
    {
      "epoch": 6.952190399169922e-05,
      "step": 72899,
      "training_step_time": 7.7247841358184814
    },
    {
      "epoch": 6.952285766601563e-05,
      "grad_norm": 0.07328369468450546,
      "learning_rate": 1.8771429605762841e-06,
      "loss": 0.0515,
      "step": 72900
    },
    {
      "epoch": 6.952285766601563e-05,
      "model_forward_time": 1.7266943454742432,
      "step": 72900
    },
    {
      "dynamics_loss_avg": 0.060375474579632285,
      "epoch": 6.952285766601563e-05,
      "step": 72900
    },
    {
      "action_loss_avg": 0.001690599787980318,
      "epoch": 6.952285766601563e-05,
      "step": 72900
    },
    {
      "epoch": 6.952285766601563e-05,
      "step": 72900,
      "training_step_time": 7.779698371887207
    },
    {
      "epoch": 6.952381134033203e-05,
      "model_forward_time": 1.7279410362243652,
      "step": 72901
    },
    {
      "epoch": 6.952381134033203e-05,
      "step": 72901,
      "training_step_time": 7.752724885940552
    },
    {
      "epoch": 6.952476501464844e-05,
      "model_forward_time": 1.7300775051116943,
      "step": 72902
    },
    {
      "epoch": 6.952476501464844e-05,
      "step": 72902,
      "training_step_time": 7.690446138381958
    },
    {
      "epoch": 6.952571868896485e-05,
      "model_forward_time": 1.7179508209228516,
      "step": 72903
    },
    {
      "epoch": 6.952571868896485e-05,
      "step": 72903,
      "training_step_time": 7.607417106628418
    },
    {
      "epoch": 6.952667236328125e-05,
      "model_forward_time": 1.7203454971313477,
      "step": 72904
    },
    {
      "epoch": 6.952667236328125e-05,
      "step": 72904,
      "training_step_time": 7.42028021812439
    },
    {
      "epoch": 6.952762603759766e-05,
      "model_forward_time": 1.7382404804229736,
      "step": 72905
    },
    {
      "epoch": 6.952762603759766e-05,
      "step": 72905,
      "training_step_time": 8.64835000038147
    },
    {
      "epoch": 6.952857971191407e-05,
      "model_forward_time": 1.7328813076019287,
      "step": 72906
    },
    {
      "epoch": 6.952857971191407e-05,
      "step": 72906,
      "training_step_time": 8.632189750671387
    },
    {
      "epoch": 6.952953338623047e-05,
      "model_forward_time": 1.7070624828338623,
      "step": 72907
    },
    {
      "epoch": 6.952953338623047e-05,
      "step": 72907,
      "training_step_time": 9.742065906524658
    },
    {
      "epoch": 6.953048706054688e-05,
      "model_forward_time": 1.7054638862609863,
      "step": 72908
    },
    {
      "epoch": 6.953048706054688e-05,
      "step": 72908,
      "training_step_time": 7.790159225463867
    },
    {
      "epoch": 6.953144073486329e-05,
      "model_forward_time": 1.7206332683563232,
      "step": 72909
    },
    {
      "epoch": 6.953144073486329e-05,
      "step": 72909,
      "training_step_time": 7.723073244094849
    },
    {
      "epoch": 6.953239440917968e-05,
      "grad_norm": 0.0798540860414505,
      "learning_rate": 1.8758518248262837e-06,
      "loss": 0.0505,
      "step": 72910
    },
    {
      "epoch": 6.953239440917968e-05,
      "model_forward_time": 1.7263853549957275,
      "step": 72910
    },
    {
      "dynamics_loss_avg": 0.06827174462378024,
      "epoch": 6.953239440917968e-05,
      "step": 72910
    },
    {
      "action_loss_avg": 0.004550038464367389,
      "epoch": 6.953239440917968e-05,
      "step": 72910
    },
    {
      "epoch": 6.953239440917968e-05,
      "step": 72910,
      "training_step_time": 8.934063196182251
    },
    {
      "epoch": 6.95333480834961e-05,
      "model_forward_time": 1.7279260158538818,
      "step": 72911
    },
    {
      "epoch": 6.95333480834961e-05,
      "step": 72911,
      "training_step_time": 7.375007390975952
    },
    {
      "epoch": 6.95343017578125e-05,
      "model_forward_time": 1.7314298152923584,
      "step": 72912
    },
    {
      "epoch": 6.95343017578125e-05,
      "step": 72912,
      "training_step_time": 7.7017011642456055
    },
    {
      "epoch": 6.95352554321289e-05,
      "model_forward_time": 1.7282187938690186,
      "step": 72913
    },
    {
      "epoch": 6.95352554321289e-05,
      "step": 72913,
      "training_step_time": 7.553105592727661
    },
    {
      "epoch": 6.953620910644531e-05,
      "model_forward_time": 1.7190899848937988,
      "step": 72914
    },
    {
      "epoch": 6.953620910644531e-05,
      "step": 72914,
      "training_step_time": 7.937847137451172
    },
    {
      "epoch": 6.953716278076172e-05,
      "model_forward_time": 1.7347931861877441,
      "step": 72915
    },
    {
      "epoch": 6.953716278076172e-05,
      "step": 72915,
      "training_step_time": 7.715540885925293
    },
    {
      "epoch": 6.953811645507812e-05,
      "model_forward_time": 1.7303438186645508,
      "step": 72916
    },
    {
      "epoch": 6.953811645507812e-05,
      "step": 72916,
      "training_step_time": 8.41356635093689
    },
    {
      "epoch": 6.953907012939453e-05,
      "model_forward_time": 2.278944969177246,
      "step": 72917
    },
    {
      "epoch": 6.953907012939453e-05,
      "step": 72917,
      "training_step_time": 7.728480815887451
    },
    {
      "epoch": 6.954002380371094e-05,
      "model_forward_time": 1.7406723499298096,
      "step": 72918
    },
    {
      "epoch": 6.954002380371094e-05,
      "step": 72918,
      "training_step_time": 7.73115086555481
    },
    {
      "epoch": 6.954097747802734e-05,
      "model_forward_time": 1.736506462097168,
      "step": 72919
    },
    {
      "epoch": 6.954097747802734e-05,
      "step": 72919,
      "training_step_time": 7.6463587284088135
    },
    {
      "epoch": 6.954193115234375e-05,
      "grad_norm": 0.08768310397863388,
      "learning_rate": 1.8745610307284307e-06,
      "loss": 0.051,
      "step": 72920
    },
    {
      "epoch": 6.954193115234375e-05,
      "model_forward_time": 1.7370707988739014,
      "step": 72920
    },
    {
      "dynamics_loss_avg": 0.05409442167729139,
      "epoch": 6.954193115234375e-05,
      "step": 72920
    },
    {
      "action_loss_avg": 0.0013352943758945912,
      "epoch": 6.954193115234375e-05,
      "step": 72920
    },
    {
      "epoch": 6.954193115234375e-05,
      "step": 72920,
      "training_step_time": 7.228124618530273
    },
    {
      "epoch": 6.954288482666016e-05,
      "model_forward_time": 1.7413852214813232,
      "step": 72921
    },
    {
      "epoch": 6.954288482666016e-05,
      "step": 72921,
      "training_step_time": 7.705355882644653
    },
    {
      "epoch": 6.954383850097656e-05,
      "model_forward_time": 1.732243299484253,
      "step": 72922
    },
    {
      "epoch": 6.954383850097656e-05,
      "step": 72922,
      "training_step_time": 7.48375391960144
    },
    {
      "epoch": 6.954479217529297e-05,
      "model_forward_time": 1.7328996658325195,
      "step": 72923
    },
    {
      "epoch": 6.954479217529297e-05,
      "step": 72923,
      "training_step_time": 7.633898019790649
    },
    {
      "epoch": 6.954574584960938e-05,
      "model_forward_time": 1.8123466968536377,
      "step": 72924
    },
    {
      "epoch": 6.954574584960938e-05,
      "step": 72924,
      "training_step_time": 8.70268702507019
    },
    {
      "epoch": 6.954669952392578e-05,
      "model_forward_time": 1.7341442108154297,
      "step": 72925
    },
    {
      "epoch": 6.954669952392578e-05,
      "step": 72925,
      "training_step_time": 7.775629043579102
    },
    {
      "epoch": 6.954765319824219e-05,
      "model_forward_time": 2.298110246658325,
      "step": 72926
    },
    {
      "epoch": 6.954765319824219e-05,
      "step": 72926,
      "training_step_time": 7.615941286087036
    },
    {
      "epoch": 6.95486068725586e-05,
      "model_forward_time": 1.7346324920654297,
      "step": 72927
    },
    {
      "epoch": 6.95486068725586e-05,
      "step": 72927,
      "training_step_time": 8.21657133102417
    },
    {
      "epoch": 6.9549560546875e-05,
      "model_forward_time": 1.723020315170288,
      "step": 72928
    },
    {
      "epoch": 6.9549560546875e-05,
      "step": 72928,
      "training_step_time": 7.822178363800049
    },
    {
      "epoch": 6.955051422119141e-05,
      "model_forward_time": 1.735868215560913,
      "step": 72929
    },
    {
      "epoch": 6.955051422119141e-05,
      "step": 72929,
      "training_step_time": 7.552240610122681
    },
    {
      "epoch": 6.955146789550782e-05,
      "grad_norm": 0.08384022116661072,
      "learning_rate": 1.8732705784238863e-06,
      "loss": 0.0531,
      "step": 72930
    },
    {
      "epoch": 6.955146789550782e-05,
      "model_forward_time": 1.3610806465148926,
      "step": 72930
    },
    {
      "dynamics_loss_avg": 0.03024394456297159,
      "epoch": 6.955146789550782e-05,
      "step": 72930
    },
    {
      "action_loss_avg": 0.0005976753542199731,
      "epoch": 6.955146789550782e-05,
      "step": 72930
    },
    {
      "epoch": 6.955146789550782e-05,
      "step": 72930,
      "training_step_time": 7.666182279586792
    },
    {
      "epoch": 6.955242156982422e-05,
      "model_forward_time": 1.733525037765503,
      "step": 72931
    },
    {
      "epoch": 6.955242156982422e-05,
      "step": 72931,
      "training_step_time": 7.5969078540802
    },
    {
      "epoch": 6.955337524414063e-05,
      "model_forward_time": 1.732593297958374,
      "step": 72932
    },
    {
      "epoch": 6.955337524414063e-05,
      "step": 72932,
      "training_step_time": 8.205994606018066
    },
    {
      "epoch": 6.955432891845704e-05,
      "model_forward_time": 1.8097262382507324,
      "step": 72933
    },
    {
      "epoch": 6.955432891845704e-05,
      "step": 72933,
      "training_step_time": 7.8354551792144775
    },
    {
      "epoch": 6.955528259277344e-05,
      "model_forward_time": 1.7352180480957031,
      "step": 72934
    },
    {
      "epoch": 6.955528259277344e-05,
      "step": 72934,
      "training_step_time": 9.49680233001709
    },
    {
      "epoch": 6.955623626708985e-05,
      "model_forward_time": 2.291030168533325,
      "step": 72935
    },
    {
      "epoch": 6.955623626708985e-05,
      "step": 72935,
      "training_step_time": 8.658271074295044
    },
    {
      "epoch": 6.955718994140626e-05,
      "model_forward_time": 1.7309932708740234,
      "step": 72936
    },
    {
      "epoch": 6.955718994140626e-05,
      "step": 72936,
      "training_step_time": 7.599922180175781
    },
    {
      "epoch": 6.955814361572265e-05,
      "model_forward_time": 1.7347917556762695,
      "step": 72937
    },
    {
      "epoch": 6.955814361572265e-05,
      "step": 72937,
      "training_step_time": 8.6377592086792
    },
    {
      "epoch": 6.955909729003906e-05,
      "model_forward_time": 1.7279231548309326,
      "step": 72938
    },
    {
      "epoch": 6.955909729003906e-05,
      "step": 72938,
      "training_step_time": 7.610329627990723
    },
    {
      "epoch": 6.956005096435546e-05,
      "model_forward_time": 1.708989143371582,
      "step": 72939
    },
    {
      "epoch": 6.956005096435546e-05,
      "step": 72939,
      "training_step_time": 7.655108451843262
    },
    {
      "epoch": 6.956100463867187e-05,
      "grad_norm": 0.07131379097700119,
      "learning_rate": 1.8719804680537685e-06,
      "loss": 0.0507,
      "step": 72940
    },
    {
      "epoch": 6.956100463867187e-05,
      "model_forward_time": 1.7193944454193115,
      "step": 72940
    },
    {
      "dynamics_loss_avg": 0.04963261317461729,
      "epoch": 6.956100463867187e-05,
      "step": 72940
    },
    {
      "action_loss_avg": 0.0014364545320859178,
      "epoch": 6.956100463867187e-05,
      "step": 72940
    },
    {
      "epoch": 6.956100463867187e-05,
      "step": 72940,
      "training_step_time": 7.62082314491272
    },
    {
      "epoch": 6.956195831298828e-05,
      "model_forward_time": 1.7322924137115479,
      "step": 72941
    },
    {
      "epoch": 6.956195831298828e-05,
      "step": 72941,
      "training_step_time": 7.535256385803223
    },
    {
      "epoch": 6.956291198730468e-05,
      "model_forward_time": 1.809612512588501,
      "step": 72942
    },
    {
      "epoch": 6.956291198730468e-05,
      "step": 72942,
      "training_step_time": 8.553838729858398
    },
    {
      "epoch": 6.956386566162109e-05,
      "model_forward_time": 1.7372856140136719,
      "step": 72943
    },
    {
      "epoch": 6.956386566162109e-05,
      "step": 72943,
      "training_step_time": 9.65067195892334
    },
    {
      "epoch": 6.95648193359375e-05,
      "model_forward_time": 2.305820941925049,
      "step": 72944
    },
    {
      "epoch": 6.95648193359375e-05,
      "step": 72944,
      "training_step_time": 9.89945363998413
    },
    {
      "epoch": 6.95657730102539e-05,
      "model_forward_time": 1.74534273147583,
      "step": 72945
    },
    {
      "epoch": 6.95657730102539e-05,
      "step": 72945,
      "training_step_time": 8.745635509490967
    },
    {
      "epoch": 6.956672668457031e-05,
      "model_forward_time": 1.7413690090179443,
      "step": 72946
    },
    {
      "epoch": 6.956672668457031e-05,
      "step": 72946,
      "training_step_time": 7.513334035873413
    },
    {
      "epoch": 6.956768035888672e-05,
      "model_forward_time": 1.7375807762145996,
      "step": 72947
    },
    {
      "epoch": 6.956768035888672e-05,
      "step": 72947,
      "training_step_time": 7.767831087112427
    },
    {
      "epoch": 6.956863403320312e-05,
      "model_forward_time": 1.7437608242034912,
      "step": 72948
    },
    {
      "epoch": 6.956863403320312e-05,
      "step": 72948,
      "training_step_time": 8.63174033164978
    },
    {
      "epoch": 6.956958770751953e-05,
      "model_forward_time": 1.7254056930541992,
      "step": 72949
    },
    {
      "epoch": 6.956958770751953e-05,
      "step": 72949,
      "training_step_time": 7.542724847793579
    },
    {
      "epoch": 6.957054138183594e-05,
      "grad_norm": 0.08821266889572144,
      "learning_rate": 1.870690699759164e-06,
      "loss": 0.0526,
      "step": 72950
    },
    {
      "epoch": 6.957054138183594e-05,
      "model_forward_time": 1.7242717742919922,
      "step": 72950
    },
    {
      "dynamics_loss_avg": 0.04204823337495327,
      "epoch": 6.957054138183594e-05,
      "step": 72950
    },
    {
      "action_loss_avg": 0.001090253223082982,
      "epoch": 6.957054138183594e-05,
      "step": 72950
    },
    {
      "epoch": 6.957054138183594e-05,
      "step": 72950,
      "training_step_time": 7.427200078964233
    },
    {
      "epoch": 6.957149505615234e-05,
      "model_forward_time": 1.8056538105010986,
      "step": 72951
    },
    {
      "epoch": 6.957149505615234e-05,
      "step": 72951,
      "training_step_time": 7.936333417892456
    },
    {
      "epoch": 6.957244873046875e-05,
      "model_forward_time": 1.7260611057281494,
      "step": 72952
    },
    {
      "epoch": 6.957244873046875e-05,
      "step": 72952,
      "training_step_time": 9.487089395523071
    },
    {
      "epoch": 6.957340240478516e-05,
      "model_forward_time": 1.7179155349731445,
      "step": 72953
    },
    {
      "epoch": 6.957340240478516e-05,
      "step": 72953,
      "training_step_time": 9.734764814376831
    },
    {
      "epoch": 6.957435607910156e-05,
      "model_forward_time": 1.7290759086608887,
      "step": 72954
    },
    {
      "epoch": 6.957435607910156e-05,
      "step": 72954,
      "training_step_time": 8.35899806022644
    },
    {
      "epoch": 6.957530975341797e-05,
      "model_forward_time": 1.724942922592163,
      "step": 72955
    },
    {
      "epoch": 6.957530975341797e-05,
      "step": 72955,
      "training_step_time": 7.676600694656372
    },
    {
      "epoch": 6.957626342773438e-05,
      "model_forward_time": 1.731478214263916,
      "step": 72956
    },
    {
      "epoch": 6.957626342773438e-05,
      "step": 72956,
      "training_step_time": 8.815717220306396
    },
    {
      "epoch": 6.957721710205078e-05,
      "model_forward_time": 1.7308030128479004,
      "step": 72957
    },
    {
      "epoch": 6.957721710205078e-05,
      "step": 72957,
      "training_step_time": 7.865362644195557
    },
    {
      "epoch": 6.957817077636719e-05,
      "model_forward_time": 1.7418019771575928,
      "step": 72958
    },
    {
      "epoch": 6.957817077636719e-05,
      "step": 72958,
      "training_step_time": 7.533137083053589
    },
    {
      "epoch": 6.95791244506836e-05,
      "model_forward_time": 1.728708028793335,
      "step": 72959
    },
    {
      "epoch": 6.95791244506836e-05,
      "step": 72959,
      "training_step_time": 8.480340480804443
    },
    {
      "epoch": 6.9580078125e-05,
      "grad_norm": 0.18376703560352325,
      "learning_rate": 1.8694012736811213e-06,
      "loss": 0.0523,
      "step": 72960
    },
    {
      "epoch": 6.9580078125e-05,
      "model_forward_time": 1.8132858276367188,
      "step": 72960
    },
    {
      "dynamics_loss_avg": 0.0629210633225739,
      "epoch": 6.9580078125e-05,
      "step": 72960
    },
    {
      "action_loss_avg": 0.0028422150178812446,
      "epoch": 6.9580078125e-05,
      "step": 72960
    },
    {
      "epoch": 6.9580078125e-05,
      "step": 72960,
      "training_step_time": 7.978485345840454
    },
    {
      "epoch": 6.95810317993164e-05,
      "model_forward_time": 1.7293660640716553,
      "step": 72961
    },
    {
      "epoch": 6.95810317993164e-05,
      "step": 72961,
      "training_step_time": 9.634871482849121
    },
    {
      "epoch": 6.958198547363282e-05,
      "model_forward_time": 1.7249650955200195,
      "step": 72962
    },
    {
      "epoch": 6.958198547363282e-05,
      "step": 72962,
      "training_step_time": 8.712531805038452
    },
    {
      "epoch": 6.958293914794921e-05,
      "model_forward_time": 2.275573253631592,
      "step": 72963
    },
    {
      "epoch": 6.958293914794921e-05,
      "step": 72963,
      "training_step_time": 8.560811758041382
    },
    {
      "epoch": 6.958389282226562e-05,
      "model_forward_time": 1.730109691619873,
      "step": 72964
    },
    {
      "epoch": 6.958389282226562e-05,
      "step": 72964,
      "training_step_time": 7.602076768875122
    },
    {
      "epoch": 6.958484649658203e-05,
      "model_forward_time": 1.7247655391693115,
      "step": 72965
    },
    {
      "epoch": 6.958484649658203e-05,
      "step": 72965,
      "training_step_time": 7.8568949699401855
    },
    {
      "epoch": 6.958580017089843e-05,
      "model_forward_time": 1.7282297611236572,
      "step": 72966
    },
    {
      "epoch": 6.958580017089843e-05,
      "step": 72966,
      "training_step_time": 7.657040119171143
    },
    {
      "epoch": 6.958675384521484e-05,
      "model_forward_time": 1.725168228149414,
      "step": 72967
    },
    {
      "epoch": 6.958675384521484e-05,
      "step": 72967,
      "training_step_time": 7.713228464126587
    },
    {
      "epoch": 6.958770751953125e-05,
      "model_forward_time": 1.7353301048278809,
      "step": 72968
    },
    {
      "epoch": 6.958770751953125e-05,
      "step": 72968,
      "training_step_time": 7.578347206115723
    },
    {
      "epoch": 6.958866119384765e-05,
      "model_forward_time": 1.8025314807891846,
      "step": 72969
    },
    {
      "epoch": 6.958866119384765e-05,
      "step": 72969,
      "training_step_time": 7.59239935874939
    },
    {
      "epoch": 6.958961486816406e-05,
      "grad_norm": 0.07225274294614792,
      "learning_rate": 1.8681121899606463e-06,
      "loss": 0.0524,
      "step": 72970
    },
    {
      "epoch": 6.958961486816406e-05,
      "model_forward_time": 1.7260289192199707,
      "step": 72970
    },
    {
      "dynamics_loss_avg": 0.05389490127563477,
      "epoch": 6.958961486816406e-05,
      "step": 72970
    },
    {
      "action_loss_avg": 0.003639748936984688,
      "epoch": 6.958961486816406e-05,
      "step": 72970
    },
    {
      "epoch": 6.958961486816406e-05,
      "step": 72970,
      "training_step_time": 8.664059400558472
    },
    {
      "epoch": 6.959056854248047e-05,
      "model_forward_time": 1.371260166168213,
      "step": 72971
    },
    {
      "epoch": 6.959056854248047e-05,
      "step": 72971,
      "training_step_time": 8.47828722000122
    },
    {
      "epoch": 6.959152221679687e-05,
      "model_forward_time": 2.275000810623169,
      "step": 72972
    },
    {
      "epoch": 6.959152221679687e-05,
      "step": 72972,
      "training_step_time": 9.348252534866333
    },
    {
      "epoch": 6.959247589111328e-05,
      "model_forward_time": 1.7459311485290527,
      "step": 72973
    },
    {
      "epoch": 6.959247589111328e-05,
      "step": 72973,
      "training_step_time": 7.615828514099121
    },
    {
      "epoch": 6.959342956542969e-05,
      "model_forward_time": 1.7225558757781982,
      "step": 72974
    },
    {
      "epoch": 6.959342956542969e-05,
      "step": 72974,
      "training_step_time": 10.670467615127563
    },
    {
      "epoch": 6.959438323974609e-05,
      "model_forward_time": 1.7400727272033691,
      "step": 72975
    },
    {
      "epoch": 6.959438323974609e-05,
      "step": 72975,
      "training_step_time": 7.6870129108428955
    },
    {
      "epoch": 6.95953369140625e-05,
      "model_forward_time": 1.7348902225494385,
      "step": 72976
    },
    {
      "epoch": 6.95953369140625e-05,
      "step": 72976,
      "training_step_time": 8.215246438980103
    },
    {
      "epoch": 6.959629058837891e-05,
      "model_forward_time": 1.7449994087219238,
      "step": 72977
    },
    {
      "epoch": 6.959629058837891e-05,
      "step": 72977,
      "training_step_time": 7.547475576400757
    },
    {
      "epoch": 6.959724426269531e-05,
      "model_forward_time": 1.7775840759277344,
      "step": 72978
    },
    {
      "epoch": 6.959724426269531e-05,
      "step": 72978,
      "training_step_time": 8.412835597991943
    },
    {
      "epoch": 6.959819793701172e-05,
      "model_forward_time": 1.7231895923614502,
      "step": 72979
    },
    {
      "epoch": 6.959819793701172e-05,
      "step": 72979,
      "training_step_time": 8.828370809555054
    },
    {
      "epoch": 6.959915161132813e-05,
      "grad_norm": 0.1255117803812027,
      "learning_rate": 1.8668234487387144e-06,
      "loss": 0.0539,
      "step": 72980
    },
    {
      "epoch": 6.959915161132813e-05,
      "model_forward_time": 1.7231614589691162,
      "step": 72980
    },
    {
      "dynamics_loss_avg": 0.06109035853296518,
      "epoch": 6.959915161132813e-05,
      "step": 72980
    },
    {
      "action_loss_avg": 0.00235330116993282,
      "epoch": 6.959915161132813e-05,
      "step": 72980
    },
    {
      "epoch": 6.959915161132813e-05,
      "step": 72980,
      "training_step_time": 7.79827880859375
    },
    {
      "epoch": 6.960010528564453e-05,
      "model_forward_time": 2.272280693054199,
      "step": 72981
    },
    {
      "epoch": 6.960010528564453e-05,
      "step": 72981,
      "training_step_time": 9.155468225479126
    },
    {
      "epoch": 6.960105895996094e-05,
      "model_forward_time": 1.731459140777588,
      "step": 72982
    },
    {
      "epoch": 6.960105895996094e-05,
      "step": 72982,
      "training_step_time": 7.678822994232178
    },
    {
      "epoch": 6.960201263427735e-05,
      "model_forward_time": 1.7208542823791504,
      "step": 72983
    },
    {
      "epoch": 6.960201263427735e-05,
      "step": 72983,
      "training_step_time": 7.923042297363281
    },
    {
      "epoch": 6.960296630859374e-05,
      "model_forward_time": 1.727250337600708,
      "step": 72984
    },
    {
      "epoch": 6.960296630859374e-05,
      "step": 72984,
      "training_step_time": 7.623445272445679
    },
    {
      "epoch": 6.960391998291016e-05,
      "model_forward_time": 1.7325410842895508,
      "step": 72985
    },
    {
      "epoch": 6.960391998291016e-05,
      "step": 72985,
      "training_step_time": 7.68356990814209
    },
    {
      "epoch": 6.960487365722657e-05,
      "model_forward_time": 1.721592664718628,
      "step": 72986
    },
    {
      "epoch": 6.960487365722657e-05,
      "step": 72986,
      "training_step_time": 8.30927848815918
    },
    {
      "epoch": 6.960582733154296e-05,
      "model_forward_time": 1.7766144275665283,
      "step": 72987
    },
    {
      "epoch": 6.960582733154296e-05,
      "step": 72987,
      "training_step_time": 7.613264322280884
    },
    {
      "epoch": 6.960678100585937e-05,
      "model_forward_time": 1.724442958831787,
      "step": 72988
    },
    {
      "epoch": 6.960678100585937e-05,
      "step": 72988,
      "training_step_time": 8.063798427581787
    },
    {
      "epoch": 6.960773468017579e-05,
      "model_forward_time": 1.7238569259643555,
      "step": 72989
    },
    {
      "epoch": 6.960773468017579e-05,
      "step": 72989,
      "training_step_time": 9.233890295028687
    },
    {
      "epoch": 6.960868835449218e-05,
      "grad_norm": 0.18989625573158264,
      "learning_rate": 1.8655350501562592e-06,
      "loss": 0.0551,
      "step": 72990
    },
    {
      "epoch": 6.960868835449218e-05,
      "model_forward_time": 2.2712924480438232,
      "step": 72990
    },
    {
      "dynamics_loss_avg": 0.04694206267595291,
      "epoch": 6.960868835449218e-05,
      "step": 72990
    },
    {
      "action_loss_avg": 0.0015776447718963027,
      "epoch": 6.960868835449218e-05,
      "step": 72990
    },
    {
      "epoch": 6.960868835449218e-05,
      "step": 72990,
      "training_step_time": 9.619844675064087
    },
    {
      "epoch": 6.96096420288086e-05,
      "model_forward_time": 1.736128568649292,
      "step": 72991
    },
    {
      "epoch": 6.96096420288086e-05,
      "step": 72991,
      "training_step_time": 7.635809659957886
    },
    {
      "epoch": 6.9610595703125e-05,
      "model_forward_time": 1.7262039184570312,
      "step": 72992
    },
    {
      "epoch": 6.9610595703125e-05,
      "step": 72992,
      "training_step_time": 7.941559314727783
    },
    {
      "epoch": 6.96115493774414e-05,
      "model_forward_time": 1.7120087146759033,
      "step": 72993
    },
    {
      "epoch": 6.96115493774414e-05,
      "step": 72993,
      "training_step_time": 7.819217205047607
    },
    {
      "epoch": 6.961250305175781e-05,
      "model_forward_time": 1.7395613193511963,
      "step": 72994
    },
    {
      "epoch": 6.961250305175781e-05,
      "step": 72994,
      "training_step_time": 8.749876022338867
    },
    {
      "epoch": 6.961345672607422e-05,
      "model_forward_time": 1.7302017211914062,
      "step": 72995
    },
    {
      "epoch": 6.961345672607422e-05,
      "step": 72995,
      "training_step_time": 7.809779405593872
    },
    {
      "epoch": 6.961441040039062e-05,
      "model_forward_time": 1.7481324672698975,
      "step": 72996
    },
    {
      "epoch": 6.961441040039062e-05,
      "step": 72996,
      "training_step_time": 7.729460716247559
    },
    {
      "epoch": 6.961536407470703e-05,
      "model_forward_time": 1.7458405494689941,
      "step": 72997
    },
    {
      "epoch": 6.961536407470703e-05,
      "step": 72997,
      "training_step_time": 8.666381120681763
    },
    {
      "epoch": 6.961631774902344e-05,
      "model_forward_time": 1.764307975769043,
      "step": 72998
    },
    {
      "epoch": 6.961631774902344e-05,
      "step": 72998,
      "training_step_time": 7.809877872467041
    },
    {
      "epoch": 6.961727142333984e-05,
      "model_forward_time": 2.2501349449157715,
      "step": 72999
    },
    {
      "epoch": 6.961727142333984e-05,
      "step": 72999,
      "training_step_time": 8.844366312026978
    },
    {
      "epoch": 6.961822509765625e-05,
      "grad_norm": 0.11811394989490509,
      "learning_rate": 1.8642469943541797e-06,
      "loss": 0.0536,
      "step": 73000
    },
    {
      "epoch": 6.961822509765625e-05,
      "model_forward_time": 1.7680940628051758,
      "step": 73000
    },
    {
      "dynamics_loss_avg": 0.04230620134621858,
      "epoch": 6.961822509765625e-05,
      "step": 73000
    },
    {
      "action_loss_avg": 0.001475052291061729,
      "epoch": 6.961822509765625e-05,
      "step": 73000
    },
    {
      "epoch": 6.961822509765625e-05,
      "step": 73000,
      "training_step_time": 9.506671667098999
    },
    {
      "epoch": 6.961917877197266e-05,
      "model_forward_time": 1.7234821319580078,
      "step": 73001
    },
    {
      "epoch": 6.961917877197266e-05,
      "step": 73001,
      "training_step_time": 8.94172978401184
    },
    {
      "epoch": 6.962013244628906e-05,
      "model_forward_time": 1.716761589050293,
      "step": 73002
    },
    {
      "epoch": 6.962013244628906e-05,
      "step": 73002,
      "training_step_time": 7.736170291900635
    },
    {
      "epoch": 6.962108612060547e-05,
      "model_forward_time": 1.7187349796295166,
      "step": 73003
    },
    {
      "epoch": 6.962108612060547e-05,
      "step": 73003,
      "training_step_time": 7.607199668884277
    },
    {
      "epoch": 6.962203979492188e-05,
      "model_forward_time": 1.7218682765960693,
      "step": 73004
    },
    {
      "epoch": 6.962203979492188e-05,
      "step": 73004,
      "training_step_time": 7.682181358337402
    },
    {
      "epoch": 6.962299346923828e-05,
      "model_forward_time": 1.7237796783447266,
      "step": 73005
    },
    {
      "epoch": 6.962299346923828e-05,
      "step": 73005,
      "training_step_time": 7.552534818649292
    },
    {
      "epoch": 6.962394714355469e-05,
      "model_forward_time": 1.7102224826812744,
      "step": 73006
    },
    {
      "epoch": 6.962394714355469e-05,
      "step": 73006,
      "training_step_time": 8.380585432052612
    },
    {
      "epoch": 6.96249008178711e-05,
      "model_forward_time": 1.7200937271118164,
      "step": 73007
    },
    {
      "epoch": 6.96249008178711e-05,
      "step": 73007,
      "training_step_time": 9.358121156692505
    },
    {
      "epoch": 6.96258544921875e-05,
      "model_forward_time": 2.7901549339294434,
      "step": 73008
    },
    {
      "epoch": 6.96258544921875e-05,
      "step": 73008,
      "training_step_time": 8.71774435043335
    },
    {
      "epoch": 6.96268081665039e-05,
      "model_forward_time": 1.3432950973510742,
      "step": 73009
    },
    {
      "epoch": 6.96268081665039e-05,
      "step": 73009,
      "training_step_time": 9.666892290115356
    },
    {
      "epoch": 6.962776184082032e-05,
      "grad_norm": 0.07433226704597473,
      "learning_rate": 1.8629592814733328e-06,
      "loss": 0.0538,
      "step": 73010
    },
    {
      "epoch": 6.962776184082032e-05,
      "model_forward_time": 1.724233865737915,
      "step": 73010
    },
    {
      "dynamics_loss_avg": 0.05836831536144018,
      "epoch": 6.962776184082032e-05,
      "step": 73010
    },
    {
      "action_loss_avg": 0.00129548164550215,
      "epoch": 6.962776184082032e-05,
      "step": 73010
    },
    {
      "epoch": 6.962776184082032e-05,
      "step": 73010,
      "training_step_time": 9.231175422668457
    },
    {
      "epoch": 6.962871551513671e-05,
      "model_forward_time": 1.7128560543060303,
      "step": 73011
    },
    {
      "epoch": 6.962871551513671e-05,
      "step": 73011,
      "training_step_time": 9.33422064781189
    },
    {
      "epoch": 6.962966918945313e-05,
      "model_forward_time": 1.7162814140319824,
      "step": 73012
    },
    {
      "epoch": 6.962966918945313e-05,
      "step": 73012,
      "training_step_time": 7.630544424057007
    },
    {
      "epoch": 6.963062286376954e-05,
      "model_forward_time": 1.7130358219146729,
      "step": 73013
    },
    {
      "epoch": 6.963062286376954e-05,
      "step": 73013,
      "training_step_time": 7.689504146575928
    },
    {
      "epoch": 6.963157653808593e-05,
      "model_forward_time": 1.7110469341278076,
      "step": 73014
    },
    {
      "epoch": 6.963157653808593e-05,
      "step": 73014,
      "training_step_time": 7.597409009933472
    },
    {
      "epoch": 6.963253021240234e-05,
      "model_forward_time": 1.7966818809509277,
      "step": 73015
    },
    {
      "epoch": 6.963253021240234e-05,
      "step": 73015,
      "training_step_time": 7.792462110519409
    },
    {
      "epoch": 6.963348388671876e-05,
      "model_forward_time": 1.7100727558135986,
      "step": 73016
    },
    {
      "epoch": 6.963348388671876e-05,
      "step": 73016,
      "training_step_time": 8.541768312454224
    },
    {
      "epoch": 6.963443756103515e-05,
      "model_forward_time": 1.7279517650604248,
      "step": 73017
    },
    {
      "epoch": 6.963443756103515e-05,
      "step": 73017,
      "training_step_time": 8.77123212814331
    },
    {
      "epoch": 6.963539123535156e-05,
      "model_forward_time": 1.7019720077514648,
      "step": 73018
    },
    {
      "epoch": 6.963539123535156e-05,
      "step": 73018,
      "training_step_time": 7.699019193649292
    },
    {
      "epoch": 6.963634490966797e-05,
      "model_forward_time": 1.7027106285095215,
      "step": 73019
    },
    {
      "epoch": 6.963634490966797e-05,
      "step": 73019,
      "training_step_time": 8.495583057403564
    },
    {
      "epoch": 6.963729858398437e-05,
      "grad_norm": 0.22616639733314514,
      "learning_rate": 1.861671911654544e-06,
      "loss": 0.0544,
      "step": 73020
    },
    {
      "epoch": 6.963729858398437e-05,
      "model_forward_time": 1.7048792839050293,
      "step": 73020
    },
    {
      "dynamics_loss_avg": 0.03593871053308249,
      "epoch": 6.963729858398437e-05,
      "step": 73020
    },
    {
      "action_loss_avg": 0.0008855721564032138,
      "epoch": 6.963729858398437e-05,
      "step": 73020
    },
    {
      "epoch": 6.963729858398437e-05,
      "step": 73020,
      "training_step_time": 8.911537885665894
    },
    {
      "epoch": 6.963825225830078e-05,
      "model_forward_time": 1.7100131511688232,
      "step": 73021
    },
    {
      "epoch": 6.963825225830078e-05,
      "step": 73021,
      "training_step_time": 8.839366674423218
    },
    {
      "epoch": 6.963920593261719e-05,
      "model_forward_time": 1.7211439609527588,
      "step": 73022
    },
    {
      "epoch": 6.963920593261719e-05,
      "step": 73022,
      "training_step_time": 7.725744247436523
    },
    {
      "epoch": 6.964015960693359e-05,
      "model_forward_time": 1.7052161693572998,
      "step": 73023
    },
    {
      "epoch": 6.964015960693359e-05,
      "step": 73023,
      "training_step_time": 7.681118011474609
    },
    {
      "epoch": 6.964111328125e-05,
      "model_forward_time": 1.7823426723480225,
      "step": 73024
    },
    {
      "epoch": 6.964111328125e-05,
      "step": 73024,
      "training_step_time": 8.344059467315674
    },
    {
      "epoch": 6.964206695556641e-05,
      "model_forward_time": 1.7342212200164795,
      "step": 73025
    },
    {
      "epoch": 6.964206695556641e-05,
      "step": 73025,
      "training_step_time": 8.635081768035889
    },
    {
      "epoch": 6.964302062988281e-05,
      "model_forward_time": 1.7350995540618896,
      "step": 73026
    },
    {
      "epoch": 6.964302062988281e-05,
      "step": 73026,
      "training_step_time": 7.717489242553711
    },
    {
      "epoch": 6.964397430419922e-05,
      "model_forward_time": 2.2490336894989014,
      "step": 73027
    },
    {
      "epoch": 6.964397430419922e-05,
      "step": 73027,
      "training_step_time": 9.54952883720398
    },
    {
      "epoch": 6.964492797851563e-05,
      "model_forward_time": 1.734816074371338,
      "step": 73028
    },
    {
      "epoch": 6.964492797851563e-05,
      "step": 73028,
      "training_step_time": 8.64609169960022
    },
    {
      "epoch": 6.964588165283203e-05,
      "model_forward_time": 1.7174749374389648,
      "step": 73029
    },
    {
      "epoch": 6.964588165283203e-05,
      "step": 73029,
      "training_step_time": 8.943790197372437
    },
    {
      "epoch": 6.964683532714844e-05,
      "grad_norm": 0.07239729911088943,
      "learning_rate": 1.8603848850385952e-06,
      "loss": 0.0566,
      "step": 73030
    },
    {
      "epoch": 6.964683532714844e-05,
      "model_forward_time": 1.7335302829742432,
      "step": 73030
    },
    {
      "dynamics_loss_avg": 0.05854759160429239,
      "epoch": 6.964683532714844e-05,
      "step": 73030
    },
    {
      "action_loss_avg": 0.0012528472405392676,
      "epoch": 6.964683532714844e-05,
      "step": 73030
    },
    {
      "epoch": 6.964683532714844e-05,
      "step": 73030,
      "training_step_time": 7.800090074539185
    },
    {
      "epoch": 6.964778900146485e-05,
      "model_forward_time": 1.3518600463867188,
      "step": 73031
    },
    {
      "epoch": 6.964778900146485e-05,
      "step": 73031,
      "training_step_time": 8.265655517578125
    },
    {
      "epoch": 6.964874267578125e-05,
      "model_forward_time": 1.732832431793213,
      "step": 73032
    },
    {
      "epoch": 6.964874267578125e-05,
      "step": 73032,
      "training_step_time": 7.681000232696533
    },
    {
      "epoch": 6.964969635009766e-05,
      "model_forward_time": 1.7791285514831543,
      "step": 73033
    },
    {
      "epoch": 6.964969635009766e-05,
      "step": 73033,
      "training_step_time": 8.347489833831787
    },
    {
      "epoch": 6.965065002441407e-05,
      "model_forward_time": 1.7283885478973389,
      "step": 73034
    },
    {
      "epoch": 6.965065002441407e-05,
      "step": 73034,
      "training_step_time": 8.7562096118927
    },
    {
      "epoch": 6.965160369873047e-05,
      "model_forward_time": 1.7334349155426025,
      "step": 73035
    },
    {
      "epoch": 6.965160369873047e-05,
      "step": 73035,
      "training_step_time": 8.396335124969482
    },
    {
      "epoch": 6.965255737304688e-05,
      "model_forward_time": 2.25034499168396,
      "step": 73036
    },
    {
      "epoch": 6.965255737304688e-05,
      "step": 73036,
      "training_step_time": 9.917643308639526
    },
    {
      "epoch": 6.965351104736329e-05,
      "model_forward_time": 1.7200443744659424,
      "step": 73037
    },
    {
      "epoch": 6.965351104736329e-05,
      "step": 73037,
      "training_step_time": 7.763475656509399
    },
    {
      "epoch": 6.965446472167968e-05,
      "model_forward_time": 1.7228362560272217,
      "step": 73038
    },
    {
      "epoch": 6.965446472167968e-05,
      "step": 73038,
      "training_step_time": 8.497834205627441
    },
    {
      "epoch": 6.96554183959961e-05,
      "model_forward_time": 1.7256190776824951,
      "step": 73039
    },
    {
      "epoch": 6.96554183959961e-05,
      "step": 73039,
      "training_step_time": 8.45482850074768
    },
    {
      "epoch": 6.96563720703125e-05,
      "grad_norm": 0.07002007961273193,
      "learning_rate": 1.8590982017662362e-06,
      "loss": 0.0533,
      "step": 73040
    },
    {
      "epoch": 6.96563720703125e-05,
      "model_forward_time": 1.7082781791687012,
      "step": 73040
    },
    {
      "dynamics_loss_avg": 0.04081902662292123,
      "epoch": 6.96563720703125e-05,
      "step": 73040
    },
    {
      "action_loss_avg": 0.0006175382528454066,
      "epoch": 6.96563720703125e-05,
      "step": 73040
    },
    {
      "epoch": 6.96563720703125e-05,
      "step": 73040,
      "training_step_time": 8.622515439987183
    },
    {
      "epoch": 6.96573257446289e-05,
      "model_forward_time": 1.7228336334228516,
      "step": 73041
    },
    {
      "epoch": 6.96573257446289e-05,
      "step": 73041,
      "training_step_time": 8.720031023025513
    },
    {
      "epoch": 6.965827941894531e-05,
      "model_forward_time": 1.7383112907409668,
      "step": 73042
    },
    {
      "epoch": 6.965827941894531e-05,
      "step": 73042,
      "training_step_time": 7.67339825630188
    },
    {
      "epoch": 6.965923309326172e-05,
      "model_forward_time": 1.7109498977661133,
      "step": 73043
    },
    {
      "epoch": 6.965923309326172e-05,
      "step": 73043,
      "training_step_time": 7.759568214416504
    },
    {
      "epoch": 6.966018676757812e-05,
      "model_forward_time": 1.7234630584716797,
      "step": 73044
    },
    {
      "epoch": 6.966018676757812e-05,
      "step": 73044,
      "training_step_time": 8.834522247314453
    },
    {
      "epoch": 6.966114044189453e-05,
      "model_forward_time": 2.2873308658599854,
      "step": 73045
    },
    {
      "epoch": 6.966114044189453e-05,
      "step": 73045,
      "training_step_time": 9.821002721786499
    },
    {
      "epoch": 6.966209411621094e-05,
      "model_forward_time": 1.714184045791626,
      "step": 73046
    },
    {
      "epoch": 6.966209411621094e-05,
      "step": 73046,
      "training_step_time": 8.608165264129639
    },
    {
      "epoch": 6.966304779052734e-05,
      "model_forward_time": 1.7110745906829834,
      "step": 73047
    },
    {
      "epoch": 6.966304779052734e-05,
      "step": 73047,
      "training_step_time": 8.76953125
    },
    {
      "epoch": 6.966400146484375e-05,
      "model_forward_time": 1.695997953414917,
      "step": 73048
    },
    {
      "epoch": 6.966400146484375e-05,
      "step": 73048,
      "training_step_time": 8.507341623306274
    },
    {
      "epoch": 6.966495513916016e-05,
      "model_forward_time": 1.7079086303710938,
      "step": 73049
    },
    {
      "epoch": 6.966495513916016e-05,
      "step": 73049,
      "training_step_time": 8.571510553359985
    },
    {
      "epoch": 6.966590881347656e-05,
      "grad_norm": 0.0875878781080246,
      "learning_rate": 1.8578118619781727e-06,
      "loss": 0.0572,
      "step": 73050
    },
    {
      "epoch": 6.966590881347656e-05,
      "model_forward_time": 1.7189998626708984,
      "step": 73050
    },
    {
      "dynamics_loss_avg": 0.044794817268848416,
      "epoch": 6.966590881347656e-05,
      "step": 73050
    },
    {
      "action_loss_avg": 0.0013226161419879646,
      "epoch": 6.966590881347656e-05,
      "step": 73050
    },
    {
      "epoch": 6.966590881347656e-05,
      "step": 73050,
      "training_step_time": 7.640681028366089
    },
    {
      "epoch": 6.966686248779297e-05,
      "model_forward_time": 1.7170820236206055,
      "step": 73051
    },
    {
      "epoch": 6.966686248779297e-05,
      "step": 73051,
      "training_step_time": 7.638944387435913
    },
    {
      "epoch": 6.966781616210938e-05,
      "model_forward_time": 1.717637538909912,
      "step": 73052
    },
    {
      "epoch": 6.966781616210938e-05,
      "step": 73052,
      "training_step_time": 7.6724019050598145
    },
    {
      "epoch": 6.966876983642578e-05,
      "model_forward_time": 1.711775779724121,
      "step": 73053
    },
    {
      "epoch": 6.966876983642578e-05,
      "step": 73053,
      "training_step_time": 9.911341905593872
    },
    {
      "epoch": 6.966972351074219e-05,
      "model_forward_time": 2.2717783451080322,
      "step": 73054
    },
    {
      "epoch": 6.966972351074219e-05,
      "step": 73054,
      "training_step_time": 7.561701059341431
    },
    {
      "epoch": 6.96706771850586e-05,
      "model_forward_time": 1.7171978950500488,
      "step": 73055
    },
    {
      "epoch": 6.96706771850586e-05,
      "step": 73055,
      "training_step_time": 8.394438982009888
    },
    {
      "epoch": 6.9671630859375e-05,
      "model_forward_time": 2.8435816764831543,
      "step": 73056
    },
    {
      "epoch": 6.9671630859375e-05,
      "step": 73056,
      "training_step_time": 7.671293258666992
    },
    {
      "epoch": 6.967258453369141e-05,
      "model_forward_time": 1.731598138809204,
      "step": 73057
    },
    {
      "epoch": 6.967258453369141e-05,
      "step": 73057,
      "training_step_time": 9.00987720489502
    },
    {
      "epoch": 6.967353820800782e-05,
      "model_forward_time": 1.7186124324798584,
      "step": 73058
    },
    {
      "epoch": 6.967353820800782e-05,
      "step": 73058,
      "training_step_time": 6.877062559127808
    },
    {
      "epoch": 6.967449188232422e-05,
      "model_forward_time": 1.718468189239502,
      "step": 73059
    },
    {
      "epoch": 6.967449188232422e-05,
      "step": 73059,
      "training_step_time": 6.874847412109375
    },
    {
      "epoch": 6.967544555664063e-05,
      "grad_norm": 0.09669838100671768,
      "learning_rate": 1.8565258658150831e-06,
      "loss": 0.0577,
      "step": 73060
    },
    {
      "epoch": 6.967544555664063e-05,
      "model_forward_time": 1.7234885692596436,
      "step": 73060
    },
    {
      "dynamics_loss_avg": 0.051089541241526606,
      "epoch": 6.967544555664063e-05,
      "step": 73060
    },
    {
      "action_loss_avg": 0.0016096224717330188,
      "epoch": 6.967544555664063e-05,
      "step": 73060
    },
    {
      "epoch": 6.967544555664063e-05,
      "step": 73060,
      "training_step_time": 7.021872520446777
    },
    {
      "epoch": 6.967639923095704e-05,
      "model_forward_time": 1.727754831314087,
      "step": 73061
    },
    {
      "epoch": 6.967639923095704e-05,
      "step": 73061,
      "training_step_time": 7.013373613357544
    },
    {
      "epoch": 6.967735290527344e-05,
      "model_forward_time": 1.716325283050537,
      "step": 73062
    },
    {
      "epoch": 6.967735290527344e-05,
      "step": 73062,
      "training_step_time": 7.006820201873779
    },
    {
      "epoch": 6.967830657958985e-05,
      "model_forward_time": 1.7131702899932861,
      "step": 73063
    },
    {
      "epoch": 6.967830657958985e-05,
      "step": 73063,
      "training_step_time": 6.990020751953125
    },
    {
      "epoch": 6.967926025390624e-05,
      "model_forward_time": 1.7326014041900635,
      "step": 73064
    },
    {
      "epoch": 6.967926025390624e-05,
      "step": 73064,
      "training_step_time": 7.037429571151733
    },
    {
      "epoch": 6.968021392822265e-05,
      "model_forward_time": 2.282029151916504,
      "step": 73065
    },
    {
      "epoch": 6.968021392822265e-05,
      "step": 73065,
      "training_step_time": 10.664536952972412
    },
    {
      "epoch": 6.968116760253906e-05,
      "model_forward_time": 1.727050542831421,
      "step": 73066
    },
    {
      "epoch": 6.968116760253906e-05,
      "step": 73066,
      "training_step_time": 10.011014223098755
    },
    {
      "epoch": 6.968212127685546e-05,
      "model_forward_time": 1.7302541732788086,
      "step": 73067
    },
    {
      "epoch": 6.968212127685546e-05,
      "step": 73067,
      "training_step_time": 8.338839530944824
    },
    {
      "epoch": 6.968307495117187e-05,
      "model_forward_time": 1.7251524925231934,
      "step": 73068
    },
    {
      "epoch": 6.968307495117187e-05,
      "step": 73068,
      "training_step_time": 6.861358165740967
    },
    {
      "epoch": 6.968402862548828e-05,
      "model_forward_time": 1.7190251350402832,
      "step": 73069
    },
    {
      "epoch": 6.968402862548828e-05,
      "step": 73069,
      "training_step_time": 6.962793588638306
    },
    {
      "epoch": 6.968498229980468e-05,
      "grad_norm": 0.07729645073413849,
      "learning_rate": 1.8552402134175967e-06,
      "loss": 0.0577,
      "step": 73070
    },
    {
      "epoch": 6.968498229980468e-05,
      "model_forward_time": 1.7042560577392578,
      "step": 73070
    },
    {
      "dynamics_loss_avg": 0.05789063982665539,
      "epoch": 6.968498229980468e-05,
      "step": 73070
    },
    {
      "action_loss_avg": 0.0013561143190599979,
      "epoch": 6.968498229980468e-05,
      "step": 73070
    },
    {
      "epoch": 6.968498229980468e-05,
      "step": 73070,
      "training_step_time": 7.038410663604736
    },
    {
      "epoch": 6.968593597412109e-05,
      "model_forward_time": 1.7269046306610107,
      "step": 73071
    },
    {
      "epoch": 6.968593597412109e-05,
      "step": 73071,
      "training_step_time": 6.981552839279175
    },
    {
      "epoch": 6.96868896484375e-05,
      "model_forward_time": 1.7269561290740967,
      "step": 73072
    },
    {
      "epoch": 6.96868896484375e-05,
      "step": 73072,
      "training_step_time": 6.944872856140137
    },
    {
      "epoch": 6.96878433227539e-05,
      "model_forward_time": 1.7180273532867432,
      "step": 73073
    },
    {
      "epoch": 6.96878433227539e-05,
      "step": 73073,
      "training_step_time": 6.978748083114624
    },
    {
      "epoch": 6.968879699707031e-05,
      "model_forward_time": 1.7229104042053223,
      "step": 73074
    },
    {
      "epoch": 6.968879699707031e-05,
      "step": 73074,
      "training_step_time": 7.008851528167725
    },
    {
      "epoch": 6.968975067138672e-05,
      "model_forward_time": 2.2762370109558105,
      "step": 73075
    },
    {
      "epoch": 6.968975067138672e-05,
      "step": 73075,
      "training_step_time": 12.1951003074646
    },
    {
      "epoch": 6.969070434570312e-05,
      "model_forward_time": 1.7134568691253662,
      "step": 73076
    },
    {
      "epoch": 6.969070434570312e-05,
      "step": 73076,
      "training_step_time": 7.863826036453247
    },
    {
      "epoch": 6.969165802001953e-05,
      "model_forward_time": 1.724778652191162,
      "step": 73077
    },
    {
      "epoch": 6.969165802001953e-05,
      "step": 73077,
      "training_step_time": 7.216952085494995
    },
    {
      "epoch": 6.969261169433594e-05,
      "model_forward_time": 1.7407124042510986,
      "step": 73078
    },
    {
      "epoch": 6.969261169433594e-05,
      "step": 73078,
      "training_step_time": 6.981841802597046
    },
    {
      "epoch": 6.969356536865234e-05,
      "model_forward_time": 1.730447769165039,
      "step": 73079
    },
    {
      "epoch": 6.969356536865234e-05,
      "step": 73079,
      "training_step_time": 6.952096939086914
    },
    {
      "epoch": 6.969451904296875e-05,
      "grad_norm": 0.08885722607374191,
      "learning_rate": 1.8539549049263144e-06,
      "loss": 0.0587,
      "step": 73080
    },
    {
      "epoch": 6.969451904296875e-05,
      "model_forward_time": 1.7126612663269043,
      "step": 73080
    },
    {
      "dynamics_loss_avg": 0.05403587948530912,
      "epoch": 6.969451904296875e-05,
      "step": 73080
    },
    {
      "action_loss_avg": 0.0008873651153407991,
      "epoch": 6.969451904296875e-05,
      "step": 73080
    },
    {
      "epoch": 6.969451904296875e-05,
      "step": 73080,
      "training_step_time": 7.02035117149353
    },
    {
      "epoch": 6.969547271728516e-05,
      "model_forward_time": 1.7269933223724365,
      "step": 73081
    },
    {
      "epoch": 6.969547271728516e-05,
      "step": 73081,
      "training_step_time": 7.001487970352173
    },
    {
      "epoch": 6.969642639160156e-05,
      "model_forward_time": 1.7350285053253174,
      "step": 73082
    },
    {
      "epoch": 6.969642639160156e-05,
      "step": 73082,
      "training_step_time": 6.9024035930633545
    },
    {
      "epoch": 6.969738006591797e-05,
      "model_forward_time": 1.7299981117248535,
      "step": 73083
    },
    {
      "epoch": 6.969738006591797e-05,
      "step": 73083,
      "training_step_time": 6.850486516952515
    },
    {
      "epoch": 6.969833374023438e-05,
      "model_forward_time": 2.303668975830078,
      "step": 73084
    },
    {
      "epoch": 6.969833374023438e-05,
      "step": 73084,
      "training_step_time": 9.27380919456482
    },
    {
      "epoch": 6.969928741455078e-05,
      "model_forward_time": 1.726381778717041,
      "step": 73085
    },
    {
      "epoch": 6.969928741455078e-05,
      "step": 73085,
      "training_step_time": 7.827691316604614
    },
    {
      "epoch": 6.970024108886719e-05,
      "model_forward_time": 1.7218096256256104,
      "step": 73086
    },
    {
      "epoch": 6.970024108886719e-05,
      "step": 73086,
      "training_step_time": 7.084117650985718
    },
    {
      "epoch": 6.97011947631836e-05,
      "model_forward_time": 1.720160961151123,
      "step": 73087
    },
    {
      "epoch": 6.97011947631836e-05,
      "step": 73087,
      "training_step_time": 6.8206541538238525
    },
    {
      "epoch": 6.97021484375e-05,
      "model_forward_time": 1.719306230545044,
      "step": 73088
    },
    {
      "epoch": 6.97021484375e-05,
      "step": 73088,
      "training_step_time": 6.8376617431640625
    },
    {
      "epoch": 6.97031021118164e-05,
      "model_forward_time": 1.7202188968658447,
      "step": 73089
    },
    {
      "epoch": 6.97031021118164e-05,
      "step": 73089,
      "training_step_time": 6.849414348602295
    },
    {
      "epoch": 6.970405578613282e-05,
      "grad_norm": 0.09647233784198761,
      "learning_rate": 1.8526699404817916e-06,
      "loss": 0.0587,
      "step": 73090
    },
    {
      "epoch": 6.970405578613282e-05,
      "model_forward_time": 1.7304797172546387,
      "step": 73090
    },
    {
      "dynamics_loss_avg": 0.04430499747395515,
      "epoch": 6.970405578613282e-05,
      "step": 73090
    },
    {
      "action_loss_avg": 0.0009353124652989209,
      "epoch": 6.970405578613282e-05,
      "step": 73090
    },
    {
      "epoch": 6.970405578613282e-05,
      "step": 73090,
      "training_step_time": 6.832037687301636
    },
    {
      "epoch": 6.970500946044921e-05,
      "model_forward_time": 1.7135536670684814,
      "step": 73091
    },
    {
      "epoch": 6.970500946044921e-05,
      "step": 73091,
      "training_step_time": 6.851543664932251
    },
    {
      "epoch": 6.970596313476562e-05,
      "model_forward_time": 1.7359046936035156,
      "step": 73092
    },
    {
      "epoch": 6.970596313476562e-05,
      "step": 73092,
      "training_step_time": 6.866427183151245
    },
    {
      "epoch": 6.970691680908203e-05,
      "model_forward_time": 2.3063998222351074,
      "step": 73093
    },
    {
      "epoch": 6.970691680908203e-05,
      "step": 73093,
      "training_step_time": 11.00569224357605
    },
    {
      "epoch": 6.970787048339843e-05,
      "model_forward_time": 1.7348101139068604,
      "step": 73094
    },
    {
      "epoch": 6.970787048339843e-05,
      "step": 73094,
      "training_step_time": 7.873685359954834
    },
    {
      "epoch": 6.970882415771484e-05,
      "model_forward_time": 1.7368087768554688,
      "step": 73095
    },
    {
      "epoch": 6.970882415771484e-05,
      "step": 73095,
      "training_step_time": 7.376678228378296
    },
    {
      "epoch": 6.970977783203125e-05,
      "model_forward_time": 1.7467525005340576,
      "step": 73096
    },
    {
      "epoch": 6.970977783203125e-05,
      "step": 73096,
      "training_step_time": 7.0868871212005615
    },
    {
      "epoch": 6.971073150634765e-05,
      "model_forward_time": 1.7331209182739258,
      "step": 73097
    },
    {
      "epoch": 6.971073150634765e-05,
      "step": 73097,
      "training_step_time": 6.824991941452026
    },
    {
      "epoch": 6.971168518066406e-05,
      "model_forward_time": 1.7405328750610352,
      "step": 73098
    },
    {
      "epoch": 6.971168518066406e-05,
      "step": 73098,
      "training_step_time": 6.879227161407471
    },
    {
      "epoch": 6.971263885498047e-05,
      "model_forward_time": 1.7499070167541504,
      "step": 73099
    },
    {
      "epoch": 6.971263885498047e-05,
      "step": 73099,
      "training_step_time": 6.76438307762146
    },
    {
      "epoch": 6.971359252929687e-05,
      "grad_norm": 0.06492739915847778,
      "learning_rate": 1.8513853202245547e-06,
      "loss": 0.0574,
      "step": 73100
    },
    {
      "epoch": 6.971359252929687e-05,
      "model_forward_time": 1.7386887073516846,
      "step": 73100
    },
    {
      "dynamics_loss_avg": 0.05658706557005644,
      "epoch": 6.971359252929687e-05,
      "step": 73100
    },
    {
      "action_loss_avg": 0.0017799403256503865,
      "epoch": 6.971359252929687e-05,
      "step": 73100
    },
    {
      "epoch": 6.971359252929687e-05,
      "step": 73100,
      "training_step_time": 6.7843732833862305
    },
    {
      "epoch": 6.971454620361328e-05,
      "model_forward_time": 1.7520029544830322,
      "step": 73101
    },
    {
      "epoch": 6.971454620361328e-05,
      "step": 73101,
      "training_step_time": 7.122486114501953
    },
    {
      "epoch": 6.971549987792969e-05,
      "model_forward_time": 2.2935969829559326,
      "step": 73102
    },
    {
      "epoch": 6.971549987792969e-05,
      "step": 73102,
      "training_step_time": 10.023729801177979
    },
    {
      "epoch": 6.971645355224609e-05,
      "model_forward_time": 1.7314908504486084,
      "step": 73103
    },
    {
      "epoch": 6.971645355224609e-05,
      "step": 73103,
      "training_step_time": 9.031983137130737
    },
    {
      "epoch": 6.97174072265625e-05,
      "model_forward_time": 1.725984811782837,
      "step": 73104
    },
    {
      "epoch": 6.97174072265625e-05,
      "step": 73104,
      "training_step_time": 7.851921558380127
    },
    {
      "epoch": 6.971836090087891e-05,
      "model_forward_time": 1.7251553535461426,
      "step": 73105
    },
    {
      "epoch": 6.971836090087891e-05,
      "step": 73105,
      "training_step_time": 7.388279914855957
    },
    {
      "epoch": 6.971931457519531e-05,
      "model_forward_time": 1.7261290550231934,
      "step": 73106
    },
    {
      "epoch": 6.971931457519531e-05,
      "step": 73106,
      "training_step_time": 7.170535564422607
    },
    {
      "epoch": 6.972026824951172e-05,
      "model_forward_time": 1.3330228328704834,
      "step": 73107
    },
    {
      "epoch": 6.972026824951172e-05,
      "step": 73107,
      "training_step_time": 7.092611789703369
    },
    {
      "epoch": 6.972122192382813e-05,
      "model_forward_time": 1.7283620834350586,
      "step": 73108
    },
    {
      "epoch": 6.972122192382813e-05,
      "step": 73108,
      "training_step_time": 7.135242700576782
    },
    {
      "epoch": 6.972217559814453e-05,
      "model_forward_time": 1.7341301441192627,
      "step": 73109
    },
    {
      "epoch": 6.972217559814453e-05,
      "step": 73109,
      "training_step_time": 7.157123804092407
    },
    {
      "epoch": 6.972312927246094e-05,
      "grad_norm": 0.1013767421245575,
      "learning_rate": 1.8501010442950811e-06,
      "loss": 0.0577,
      "step": 73110
    },
    {
      "epoch": 6.972312927246094e-05,
      "model_forward_time": 1.739349126815796,
      "step": 73110
    },
    {
      "dynamics_loss_avg": 0.07072919458150864,
      "epoch": 6.972312927246094e-05,
      "step": 73110
    },
    {
      "action_loss_avg": 0.002067011606413871,
      "epoch": 6.972312927246094e-05,
      "step": 73110
    },
    {
      "epoch": 6.972312927246094e-05,
      "step": 73110,
      "training_step_time": 7.1962878704071045
    },
    {
      "epoch": 6.972408294677735e-05,
      "model_forward_time": 2.310753583908081,
      "step": 73111
    },
    {
      "epoch": 6.972408294677735e-05,
      "step": 73111,
      "training_step_time": 9.661202430725098
    },
    {
      "epoch": 6.972503662109375e-05,
      "model_forward_time": 1.7277107238769531,
      "step": 73112
    },
    {
      "epoch": 6.972503662109375e-05,
      "step": 73112,
      "training_step_time": 9.919301748275757
    },
    {
      "epoch": 6.972599029541016e-05,
      "model_forward_time": 1.7295796871185303,
      "step": 73113
    },
    {
      "epoch": 6.972599029541016e-05,
      "step": 73113,
      "training_step_time": 7.948789834976196
    },
    {
      "epoch": 6.972694396972657e-05,
      "model_forward_time": 1.8609485626220703,
      "step": 73114
    },
    {
      "epoch": 6.972694396972657e-05,
      "step": 73114,
      "training_step_time": 7.703229188919067
    },
    {
      "epoch": 6.972789764404296e-05,
      "model_forward_time": 1.9036946296691895,
      "step": 73115
    },
    {
      "epoch": 6.972789764404296e-05,
      "step": 73115,
      "training_step_time": 7.093364238739014
    },
    {
      "epoch": 6.972885131835937e-05,
      "model_forward_time": 1.92149019241333,
      "step": 73116
    },
    {
      "epoch": 6.972885131835937e-05,
      "step": 73116,
      "training_step_time": 7.130626678466797
    },
    {
      "epoch": 6.972980499267579e-05,
      "model_forward_time": 1.9169535636901855,
      "step": 73117
    },
    {
      "epoch": 6.972980499267579e-05,
      "step": 73117,
      "training_step_time": 7.156182050704956
    },
    {
      "epoch": 6.973075866699218e-05,
      "model_forward_time": 1.931502103805542,
      "step": 73118
    },
    {
      "epoch": 6.973075866699218e-05,
      "step": 73118,
      "training_step_time": 7.176660776138306
    },
    {
      "epoch": 6.97317123413086e-05,
      "model_forward_time": 1.9299910068511963,
      "step": 73119
    },
    {
      "epoch": 6.97317123413086e-05,
      "step": 73119,
      "training_step_time": 7.095568656921387
    },
    {
      "epoch": 6.9732666015625e-05,
      "grad_norm": 0.09554561227560043,
      "learning_rate": 1.848817112833825e-06,
      "loss": 0.06,
      "step": 73120
    },
    {
      "epoch": 6.9732666015625e-05,
      "model_forward_time": 3.035189628601074,
      "step": 73120
    },
    {
      "dynamics_loss_avg": 0.06356674451380968,
      "epoch": 6.9732666015625e-05,
      "step": 73120
    },
    {
      "action_loss_avg": 0.0013449598074657843,
      "epoch": 6.9732666015625e-05,
      "step": 73120
    },
    {
      "epoch": 6.9732666015625e-05,
      "step": 73120,
      "training_step_time": 9.098847389221191
    },
    {
      "epoch": 6.97336196899414e-05,
      "model_forward_time": 1.9267246723175049,
      "step": 73121
    },
    {
      "epoch": 6.97336196899414e-05,
      "step": 73121,
      "training_step_time": 8.269803524017334
    },
    {
      "epoch": 6.973457336425781e-05,
      "model_forward_time": 1.8940343856811523,
      "step": 73122
    },
    {
      "epoch": 6.973457336425781e-05,
      "step": 73122,
      "training_step_time": 7.72072696685791
    },
    {
      "epoch": 6.973552703857422e-05,
      "model_forward_time": 1.9264092445373535,
      "step": 73123
    },
    {
      "epoch": 6.973552703857422e-05,
      "step": 73123,
      "training_step_time": 6.782684326171875
    },
    {
      "epoch": 6.973648071289062e-05,
      "model_forward_time": 1.898184061050415,
      "step": 73124
    },
    {
      "epoch": 6.973648071289062e-05,
      "step": 73124,
      "training_step_time": 7.016754865646362
    },
    {
      "epoch": 6.973743438720703e-05,
      "model_forward_time": 1.9353704452514648,
      "step": 73125
    },
    {
      "epoch": 6.973743438720703e-05,
      "step": 73125,
      "training_step_time": 6.824449777603149
    },
    {
      "epoch": 6.973838806152344e-05,
      "model_forward_time": 1.940826416015625,
      "step": 73126
    },
    {
      "epoch": 6.973838806152344e-05,
      "step": 73126,
      "training_step_time": 6.804455280303955
    },
    {
      "epoch": 6.973934173583984e-05,
      "model_forward_time": 1.7857370376586914,
      "step": 73127
    },
    {
      "epoch": 6.973934173583984e-05,
      "step": 73127,
      "training_step_time": 6.77898645401001
    },
    {
      "epoch": 6.974029541015625e-05,
      "model_forward_time": 1.726618766784668,
      "step": 73128
    },
    {
      "epoch": 6.974029541015625e-05,
      "step": 73128,
      "training_step_time": 6.828619480133057
    },
    {
      "epoch": 6.974124908447266e-05,
      "model_forward_time": 2.869239568710327,
      "step": 73129
    },
    {
      "epoch": 6.974124908447266e-05,
      "step": 73129,
      "training_step_time": 7.597565650939941
    },
    {
      "epoch": 6.974220275878906e-05,
      "grad_norm": 0.10561899840831757,
      "learning_rate": 1.8475335259811905e-06,
      "loss": 0.0564,
      "step": 73130
    },
    {
      "epoch": 6.974220275878906e-05,
      "model_forward_time": 1.7302961349487305,
      "step": 73130
    },
    {
      "dynamics_loss_avg": 0.050624447595328095,
      "epoch": 6.974220275878906e-05,
      "step": 73130
    },
    {
      "action_loss_avg": 0.0016581639531068504,
      "epoch": 6.974220275878906e-05,
      "step": 73130
    },
    {
      "epoch": 6.974220275878906e-05,
      "step": 73130,
      "training_step_time": 10.469192743301392
    },
    {
      "epoch": 6.974315643310547e-05,
      "model_forward_time": 1.7276699542999268,
      "step": 73131
    },
    {
      "epoch": 6.974315643310547e-05,
      "step": 73131,
      "training_step_time": 7.748154640197754
    },
    {
      "epoch": 6.974411010742188e-05,
      "model_forward_time": 1.7334585189819336,
      "step": 73132
    },
    {
      "epoch": 6.974411010742188e-05,
      "step": 73132,
      "training_step_time": 7.360759973526001
    },
    {
      "epoch": 6.974506378173828e-05,
      "model_forward_time": 1.7232661247253418,
      "step": 73133
    },
    {
      "epoch": 6.974506378173828e-05,
      "step": 73133,
      "training_step_time": 7.0496909618377686
    },
    {
      "epoch": 6.974601745605469e-05,
      "model_forward_time": 1.725679636001587,
      "step": 73134
    },
    {
      "epoch": 6.974601745605469e-05,
      "step": 73134,
      "training_step_time": 6.79195761680603
    },
    {
      "epoch": 6.97469711303711e-05,
      "model_forward_time": 1.7280387878417969,
      "step": 73135
    },
    {
      "epoch": 6.97469711303711e-05,
      "step": 73135,
      "training_step_time": 6.785571575164795
    },
    {
      "epoch": 6.97479248046875e-05,
      "model_forward_time": 1.730224847793579,
      "step": 73136
    },
    {
      "epoch": 6.97479248046875e-05,
      "step": 73136,
      "training_step_time": 6.827257871627808
    },
    {
      "epoch": 6.97488784790039e-05,
      "model_forward_time": 1.7409496307373047,
      "step": 73137
    },
    {
      "epoch": 6.97488784790039e-05,
      "step": 73137,
      "training_step_time": 6.817094326019287
    },
    {
      "epoch": 6.974983215332032e-05,
      "model_forward_time": 2.870929479598999,
      "step": 73138
    },
    {
      "epoch": 6.974983215332032e-05,
      "step": 73138,
      "training_step_time": 7.663148403167725
    },
    {
      "epoch": 6.975078582763671e-05,
      "model_forward_time": 1.727797031402588,
      "step": 73139
    },
    {
      "epoch": 6.975078582763671e-05,
      "step": 73139,
      "training_step_time": 8.38343596458435
    },
    {
      "epoch": 6.975173950195313e-05,
      "grad_norm": 0.07976095378398895,
      "learning_rate": 1.8462502838775509e-06,
      "loss": 0.0558,
      "step": 73140
    },
    {
      "epoch": 6.975173950195313e-05,
      "model_forward_time": 1.7391993999481201,
      "step": 73140
    },
    {
      "dynamics_loss_avg": 0.06257759630680085,
      "epoch": 6.975173950195313e-05,
      "step": 73140
    },
    {
      "action_loss_avg": 0.0016615781292784958,
      "epoch": 6.975173950195313e-05,
      "step": 73140
    },
    {
      "epoch": 6.975173950195313e-05,
      "step": 73140,
      "training_step_time": 7.88727879524231
    },
    {
      "epoch": 6.975269317626954e-05,
      "model_forward_time": 1.3677606582641602,
      "step": 73141
    },
    {
      "epoch": 6.975269317626954e-05,
      "step": 73141,
      "training_step_time": 7.766051769256592
    },
    {
      "epoch": 6.975364685058593e-05,
      "model_forward_time": 1.7401320934295654,
      "step": 73142
    },
    {
      "epoch": 6.975364685058593e-05,
      "step": 73142,
      "training_step_time": 7.10308313369751
    },
    {
      "epoch": 6.975460052490234e-05,
      "model_forward_time": 1.7392966747283936,
      "step": 73143
    },
    {
      "epoch": 6.975460052490234e-05,
      "step": 73143,
      "training_step_time": 6.7961626052856445
    },
    {
      "epoch": 6.975555419921876e-05,
      "model_forward_time": 1.7405674457550049,
      "step": 73144
    },
    {
      "epoch": 6.975555419921876e-05,
      "step": 73144,
      "training_step_time": 6.820807218551636
    },
    {
      "epoch": 6.975650787353515e-05,
      "model_forward_time": 1.7397658824920654,
      "step": 73145
    },
    {
      "epoch": 6.975650787353515e-05,
      "step": 73145,
      "training_step_time": 6.7948644161224365
    },
    {
      "epoch": 6.975746154785156e-05,
      "model_forward_time": 1.7345449924468994,
      "step": 73146
    },
    {
      "epoch": 6.975746154785156e-05,
      "step": 73146,
      "training_step_time": 6.833301782608032
    },
    {
      "epoch": 6.975841522216797e-05,
      "model_forward_time": 1.7693712711334229,
      "step": 73147
    },
    {
      "epoch": 6.975841522216797e-05,
      "step": 73147,
      "training_step_time": 7.566046237945557
    },
    {
      "epoch": 6.975936889648437e-05,
      "model_forward_time": 2.314084053039551,
      "step": 73148
    },
    {
      "epoch": 6.975936889648437e-05,
      "step": 73148,
      "training_step_time": 9.29149842262268
    },
    {
      "epoch": 6.976032257080078e-05,
      "model_forward_time": 1.7263343334197998,
      "step": 73149
    },
    {
      "epoch": 6.976032257080078e-05,
      "step": 73149,
      "training_step_time": 11.013932228088379
    },
    {
      "epoch": 6.976127624511719e-05,
      "grad_norm": 0.13164417445659637,
      "learning_rate": 1.8449673866632378e-06,
      "loss": 0.0545,
      "step": 73150
    },
    {
      "epoch": 6.976127624511719e-05,
      "model_forward_time": 1.7441842555999756,
      "step": 73150
    },
    {
      "dynamics_loss_avg": 0.05280048716813326,
      "epoch": 6.976127624511719e-05,
      "step": 73150
    },
    {
      "action_loss_avg": 0.0011905121442396193,
      "epoch": 6.976127624511719e-05,
      "step": 73150
    },
    {
      "epoch": 6.976127624511719e-05,
      "step": 73150,
      "training_step_time": 7.709900140762329
    },
    {
      "epoch": 6.976222991943359e-05,
      "model_forward_time": 1.7271227836608887,
      "step": 73151
    },
    {
      "epoch": 6.976222991943359e-05,
      "step": 73151,
      "training_step_time": 6.793067932128906
    },
    {
      "epoch": 6.976318359375e-05,
      "model_forward_time": 1.735382080078125,
      "step": 73152
    },
    {
      "epoch": 6.976318359375e-05,
      "step": 73152,
      "training_step_time": 7.031708002090454
    },
    {
      "epoch": 6.976413726806641e-05,
      "model_forward_time": 1.3703835010528564,
      "step": 73153
    },
    {
      "epoch": 6.976413726806641e-05,
      "step": 73153,
      "training_step_time": 6.785020589828491
    },
    {
      "epoch": 6.976509094238281e-05,
      "model_forward_time": 1.7255916595458984,
      "step": 73154
    },
    {
      "epoch": 6.976509094238281e-05,
      "step": 73154,
      "training_step_time": 6.75586462020874
    },
    {
      "epoch": 6.976604461669922e-05,
      "model_forward_time": 1.7323055267333984,
      "step": 73155
    },
    {
      "epoch": 6.976604461669922e-05,
      "step": 73155,
      "training_step_time": 6.854750871658325
    },
    {
      "epoch": 6.976699829101563e-05,
      "model_forward_time": 1.7531225681304932,
      "step": 73156
    },
    {
      "epoch": 6.976699829101563e-05,
      "step": 73156,
      "training_step_time": 7.589383125305176
    },
    {
      "epoch": 6.976795196533203e-05,
      "model_forward_time": 2.331725835800171,
      "step": 73157
    },
    {
      "epoch": 6.976795196533203e-05,
      "step": 73157,
      "training_step_time": 8.238499641418457
    },
    {
      "epoch": 6.976890563964844e-05,
      "model_forward_time": 1.740492820739746,
      "step": 73158
    },
    {
      "epoch": 6.976890563964844e-05,
      "step": 73158,
      "training_step_time": 10.66341495513916
    },
    {
      "epoch": 6.976985931396485e-05,
      "model_forward_time": 1.7449426651000977,
      "step": 73159
    },
    {
      "epoch": 6.976985931396485e-05,
      "step": 73159,
      "training_step_time": 8.052747249603271
    },
    {
      "epoch": 6.977081298828125e-05,
      "grad_norm": 0.12527695298194885,
      "learning_rate": 1.8436848344785474e-06,
      "loss": 0.0565,
      "step": 73160
    },
    {
      "epoch": 6.977081298828125e-05,
      "model_forward_time": 1.7399609088897705,
      "step": 73160
    },
    {
      "dynamics_loss_avg": 0.054107763804495335,
      "epoch": 6.977081298828125e-05,
      "step": 73160
    },
    {
      "action_loss_avg": 0.0011748641903977842,
      "epoch": 6.977081298828125e-05,
      "step": 73160
    },
    {
      "epoch": 6.977081298828125e-05,
      "step": 73160,
      "training_step_time": 7.4514477252960205
    },
    {
      "epoch": 6.977176666259766e-05,
      "model_forward_time": 1.7414577007293701,
      "step": 73161
    },
    {
      "epoch": 6.977176666259766e-05,
      "step": 73161,
      "training_step_time": 7.104043006896973
    },
    {
      "epoch": 6.977272033691407e-05,
      "model_forward_time": 1.7381947040557861,
      "step": 73162
    },
    {
      "epoch": 6.977272033691407e-05,
      "step": 73162,
      "training_step_time": 6.884175062179565
    },
    {
      "epoch": 6.977367401123047e-05,
      "model_forward_time": 1.7503890991210938,
      "step": 73163
    },
    {
      "epoch": 6.977367401123047e-05,
      "step": 73163,
      "training_step_time": 6.9216413497924805
    },
    {
      "epoch": 6.977462768554688e-05,
      "model_forward_time": 1.747624397277832,
      "step": 73164
    },
    {
      "epoch": 6.977462768554688e-05,
      "step": 73164,
      "training_step_time": 6.8599467277526855
    },
    {
      "epoch": 6.977558135986329e-05,
      "model_forward_time": 1.7417254447937012,
      "step": 73165
    },
    {
      "epoch": 6.977558135986329e-05,
      "step": 73165,
      "training_step_time": 7.642306089401245
    },
    {
      "epoch": 6.977653503417968e-05,
      "model_forward_time": 2.3047430515289307,
      "step": 73166
    },
    {
      "epoch": 6.977653503417968e-05,
      "step": 73166,
      "training_step_time": 8.355361223220825
    },
    {
      "epoch": 6.97774887084961e-05,
      "model_forward_time": 1.7195873260498047,
      "step": 73167
    },
    {
      "epoch": 6.97774887084961e-05,
      "step": 73167,
      "training_step_time": 12.341967821121216
    },
    {
      "epoch": 6.97784423828125e-05,
      "model_forward_time": 1.7439424991607666,
      "step": 73168
    },
    {
      "epoch": 6.97784423828125e-05,
      "step": 73168,
      "training_step_time": 8.001606941223145
    },
    {
      "epoch": 6.97793960571289e-05,
      "model_forward_time": 1.7560157775878906,
      "step": 73169
    },
    {
      "epoch": 6.97793960571289e-05,
      "step": 73169,
      "training_step_time": 7.84106183052063
    },
    {
      "epoch": 6.978034973144531e-05,
      "grad_norm": 0.08093187212944031,
      "learning_rate": 1.842402627463738e-06,
      "loss": 0.055,
      "step": 73170
    },
    {
      "epoch": 6.978034973144531e-05,
      "model_forward_time": 1.742628812789917,
      "step": 73170
    },
    {
      "dynamics_loss_avg": 0.05559604167938233,
      "epoch": 6.978034973144531e-05,
      "step": 73170
    },
    {
      "action_loss_avg": 0.001894196990178898,
      "epoch": 6.978034973144531e-05,
      "step": 73170
    },
    {
      "epoch": 6.978034973144531e-05,
      "step": 73170,
      "training_step_time": 7.1829833984375
    },
    {
      "epoch": 6.978130340576172e-05,
      "model_forward_time": 1.7437477111816406,
      "step": 73171
    },
    {
      "epoch": 6.978130340576172e-05,
      "step": 73171,
      "training_step_time": 6.971292018890381
    },
    {
      "epoch": 6.978225708007812e-05,
      "model_forward_time": 1.7385766506195068,
      "step": 73172
    },
    {
      "epoch": 6.978225708007812e-05,
      "step": 73172,
      "training_step_time": 6.958153963088989
    },
    {
      "epoch": 6.978321075439453e-05,
      "model_forward_time": 1.7429871559143066,
      "step": 73173
    },
    {
      "epoch": 6.978321075439453e-05,
      "step": 73173,
      "training_step_time": 6.980543375015259
    },
    {
      "epoch": 6.978416442871094e-05,
      "model_forward_time": 1.7306983470916748,
      "step": 73174
    },
    {
      "epoch": 6.978416442871094e-05,
      "step": 73174,
      "training_step_time": 7.708967208862305
    },
    {
      "epoch": 6.978511810302734e-05,
      "model_forward_time": 2.307199478149414,
      "step": 73175
    },
    {
      "epoch": 6.978511810302734e-05,
      "step": 73175,
      "training_step_time": 8.799196720123291
    },
    {
      "epoch": 6.978607177734375e-05,
      "model_forward_time": 1.71785306930542,
      "step": 73176
    },
    {
      "epoch": 6.978607177734375e-05,
      "step": 73176,
      "training_step_time": 10.506328344345093
    },
    {
      "epoch": 6.978702545166016e-05,
      "model_forward_time": 1.738635540008545,
      "step": 73177
    },
    {
      "epoch": 6.978702545166016e-05,
      "step": 73177,
      "training_step_time": 9.526803016662598
    },
    {
      "epoch": 6.978797912597656e-05,
      "model_forward_time": 1.7317945957183838,
      "step": 73178
    },
    {
      "epoch": 6.978797912597656e-05,
      "step": 73178,
      "training_step_time": 7.821975231170654
    },
    {
      "epoch": 6.978893280029297e-05,
      "model_forward_time": 1.7323269844055176,
      "step": 73179
    },
    {
      "epoch": 6.978893280029297e-05,
      "step": 73179,
      "training_step_time": 7.1486735343933105
    },
    {
      "epoch": 6.978988647460938e-05,
      "grad_norm": 0.10214610397815704,
      "learning_rate": 1.8411207657590314e-06,
      "loss": 0.056,
      "step": 73180
    },
    {
      "epoch": 6.978988647460938e-05,
      "model_forward_time": 1.7350776195526123,
      "step": 73180
    },
    {
      "dynamics_loss_avg": 0.059256904758512974,
      "epoch": 6.978988647460938e-05,
      "step": 73180
    },
    {
      "action_loss_avg": 0.001480271539185196,
      "epoch": 6.978988647460938e-05,
      "step": 73180
    },
    {
      "epoch": 6.978988647460938e-05,
      "step": 73180,
      "training_step_time": 6.884594917297363
    },
    {
      "epoch": 6.979084014892578e-05,
      "model_forward_time": 1.7319049835205078,
      "step": 73181
    },
    {
      "epoch": 6.979084014892578e-05,
      "step": 73181,
      "training_step_time": 6.883065700531006
    },
    {
      "epoch": 6.979179382324219e-05,
      "model_forward_time": 1.74220871925354,
      "step": 73182
    },
    {
      "epoch": 6.979179382324219e-05,
      "step": 73182,
      "training_step_time": 6.858780860900879
    },
    {
      "epoch": 6.97927474975586e-05,
      "model_forward_time": 1.728010892868042,
      "step": 73183
    },
    {
      "epoch": 6.97927474975586e-05,
      "step": 73183,
      "training_step_time": 7.6914427280426025
    },
    {
      "epoch": 6.9793701171875e-05,
      "model_forward_time": 1.7379381656646729,
      "step": 73184
    },
    {
      "epoch": 6.9793701171875e-05,
      "step": 73184,
      "training_step_time": 8.781560182571411
    },
    {
      "epoch": 6.979465484619141e-05,
      "model_forward_time": 1.760101556777954,
      "step": 73185
    },
    {
      "epoch": 6.979465484619141e-05,
      "step": 73185,
      "training_step_time": 8.677447319030762
    },
    {
      "epoch": 6.979560852050782e-05,
      "model_forward_time": 1.7366292476654053,
      "step": 73186
    },
    {
      "epoch": 6.979560852050782e-05,
      "step": 73186,
      "training_step_time": 9.012214422225952
    },
    {
      "epoch": 6.979656219482422e-05,
      "model_forward_time": 1.7338740825653076,
      "step": 73187
    },
    {
      "epoch": 6.979656219482422e-05,
      "step": 73187,
      "training_step_time": 8.671759366989136
    },
    {
      "epoch": 6.979751586914063e-05,
      "model_forward_time": 1.7337255477905273,
      "step": 73188
    },
    {
      "epoch": 6.979751586914063e-05,
      "step": 73188,
      "training_step_time": 7.160641431808472
    },
    {
      "epoch": 6.979846954345702e-05,
      "model_forward_time": 1.7284867763519287,
      "step": 73189
    },
    {
      "epoch": 6.979846954345702e-05,
      "step": 73189,
      "training_step_time": 6.89452338218689
    },
    {
      "epoch": 6.979942321777344e-05,
      "grad_norm": 0.11245446652173996,
      "learning_rate": 1.8398392495046069e-06,
      "loss": 0.0565,
      "step": 73190
    },
    {
      "epoch": 6.979942321777344e-05,
      "model_forward_time": 1.7366821765899658,
      "step": 73190
    },
    {
      "dynamics_loss_avg": 0.05209393287077546,
      "epoch": 6.979942321777344e-05,
      "step": 73190
    },
    {
      "action_loss_avg": 0.0013561414903961121,
      "epoch": 6.979942321777344e-05,
      "step": 73190
    },
    {
      "epoch": 6.979942321777344e-05,
      "step": 73190,
      "training_step_time": 6.946281671524048
    },
    {
      "epoch": 6.980037689208985e-05,
      "model_forward_time": 1.7337121963500977,
      "step": 73191
    },
    {
      "epoch": 6.980037689208985e-05,
      "step": 73191,
      "training_step_time": 6.8531787395477295
    },
    {
      "epoch": 6.980133056640624e-05,
      "model_forward_time": 1.7365717887878418,
      "step": 73192
    },
    {
      "epoch": 6.980133056640624e-05,
      "step": 73192,
      "training_step_time": 7.650720596313477
    },
    {
      "epoch": 6.980228424072265e-05,
      "model_forward_time": 1.73445463180542,
      "step": 73193
    },
    {
      "epoch": 6.980228424072265e-05,
      "step": 73193,
      "training_step_time": 7.7786545753479
    },
    {
      "epoch": 6.980323791503906e-05,
      "model_forward_time": 2.3372271060943604,
      "step": 73194
    },
    {
      "epoch": 6.980323791503906e-05,
      "step": 73194,
      "training_step_time": 8.684479236602783
    },
    {
      "epoch": 6.980419158935546e-05,
      "model_forward_time": 1.7393181324005127,
      "step": 73195
    },
    {
      "epoch": 6.980419158935546e-05,
      "step": 73195,
      "training_step_time": 8.499572277069092
    },
    {
      "epoch": 6.980514526367187e-05,
      "model_forward_time": 1.745445966720581,
      "step": 73196
    },
    {
      "epoch": 6.980514526367187e-05,
      "step": 73196,
      "training_step_time": 7.821781873703003
    },
    {
      "epoch": 6.980609893798828e-05,
      "model_forward_time": 1.7443199157714844,
      "step": 73197
    },
    {
      "epoch": 6.980609893798828e-05,
      "step": 73197,
      "training_step_time": 7.9801342487335205
    },
    {
      "epoch": 6.980705261230468e-05,
      "model_forward_time": 1.7447144985198975,
      "step": 73198
    },
    {
      "epoch": 6.980705261230468e-05,
      "step": 73198,
      "training_step_time": 6.842196702957153
    },
    {
      "epoch": 6.980800628662109e-05,
      "model_forward_time": 1.7437412738800049,
      "step": 73199
    },
    {
      "epoch": 6.980800628662109e-05,
      "step": 73199,
      "training_step_time": 6.849077463150024
    },
    {
      "epoch": 6.98089599609375e-05,
      "grad_norm": 0.10108952969312668,
      "learning_rate": 1.838558078840611e-06,
      "loss": 0.0585,
      "step": 73200
    },
    {
      "epoch": 6.98089599609375e-05,
      "model_forward_time": 1.7564387321472168,
      "step": 73200
    },
    {
      "dynamics_loss_avg": 0.05522976946085691,
      "epoch": 6.98089599609375e-05,
      "step": 73200
    },
    {
      "action_loss_avg": 0.0009519282277324237,
      "epoch": 6.98089599609375e-05,
      "step": 73200
    },
    {
      "epoch": 6.98089599609375e-05,
      "step": 73200,
      "training_step_time": 6.8326733112335205
    },
    {
      "epoch": 6.98099136352539e-05,
      "model_forward_time": 1.7437043190002441,
      "step": 73201
    },
    {
      "epoch": 6.98099136352539e-05,
      "step": 73201,
      "training_step_time": 7.596465349197388
    },
    {
      "epoch": 6.981086730957031e-05,
      "model_forward_time": 1.7572863101959229,
      "step": 73202
    },
    {
      "epoch": 6.981086730957031e-05,
      "step": 73202,
      "training_step_time": 7.818264961242676
    },
    {
      "epoch": 6.981182098388672e-05,
      "model_forward_time": 2.358643054962158,
      "step": 73203
    },
    {
      "epoch": 6.981182098388672e-05,
      "step": 73203,
      "training_step_time": 7.716919660568237
    },
    {
      "epoch": 6.981277465820312e-05,
      "model_forward_time": 1.757049560546875,
      "step": 73204
    },
    {
      "epoch": 6.981277465820312e-05,
      "step": 73204,
      "training_step_time": 8.928313493728638
    },
    {
      "epoch": 6.981372833251953e-05,
      "model_forward_time": 1.7487077713012695,
      "step": 73205
    },
    {
      "epoch": 6.981372833251953e-05,
      "step": 73205,
      "training_step_time": 8.0464346408844
    },
    {
      "epoch": 6.981468200683594e-05,
      "model_forward_time": 1.7488813400268555,
      "step": 73206
    },
    {
      "epoch": 6.981468200683594e-05,
      "step": 73206,
      "training_step_time": 7.416259288787842
    },
    {
      "epoch": 6.981563568115234e-05,
      "model_forward_time": 1.7483623027801514,
      "step": 73207
    },
    {
      "epoch": 6.981563568115234e-05,
      "step": 73207,
      "training_step_time": 7.123214960098267
    },
    {
      "epoch": 6.981658935546875e-05,
      "model_forward_time": 1.7512428760528564,
      "step": 73208
    },
    {
      "epoch": 6.981658935546875e-05,
      "step": 73208,
      "training_step_time": 6.89120078086853
    },
    {
      "epoch": 6.981754302978516e-05,
      "model_forward_time": 1.7509663105010986,
      "step": 73209
    },
    {
      "epoch": 6.981754302978516e-05,
      "step": 73209,
      "training_step_time": 6.888881206512451
    },
    {
      "epoch": 6.981849670410156e-05,
      "grad_norm": 0.09271898865699768,
      "learning_rate": 1.837277253907152e-06,
      "loss": 0.057,
      "step": 73210
    },
    {
      "epoch": 6.981849670410156e-05,
      "model_forward_time": 1.7595970630645752,
      "step": 73210
    },
    {
      "dynamics_loss_avg": 0.06439084373414516,
      "epoch": 6.981849670410156e-05,
      "step": 73210
    },
    {
      "action_loss_avg": 0.0013921331614255906,
      "epoch": 6.981849670410156e-05,
      "step": 73210
    },
    {
      "epoch": 6.981849670410156e-05,
      "step": 73210,
      "training_step_time": 7.644100904464722
    },
    {
      "epoch": 6.981945037841797e-05,
      "model_forward_time": 1.7452037334442139,
      "step": 73211
    },
    {
      "epoch": 6.981945037841797e-05,
      "step": 73211,
      "training_step_time": 7.632641553878784
    },
    {
      "epoch": 6.982040405273438e-05,
      "model_forward_time": 2.340980052947998,
      "step": 73212
    },
    {
      "epoch": 6.982040405273438e-05,
      "step": 73212,
      "training_step_time": 7.763270139694214
    },
    {
      "epoch": 6.982135772705078e-05,
      "model_forward_time": 1.7478749752044678,
      "step": 73213
    },
    {
      "epoch": 6.982135772705078e-05,
      "step": 73213,
      "training_step_time": 8.783659934997559
    },
    {
      "epoch": 6.982231140136719e-05,
      "model_forward_time": 1.7570819854736328,
      "step": 73214
    },
    {
      "epoch": 6.982231140136719e-05,
      "step": 73214,
      "training_step_time": 8.951796054840088
    },
    {
      "epoch": 6.98232650756836e-05,
      "model_forward_time": 1.749558925628662,
      "step": 73215
    },
    {
      "epoch": 6.98232650756836e-05,
      "step": 73215,
      "training_step_time": 7.785780906677246
    },
    {
      "epoch": 6.982421875e-05,
      "model_forward_time": 1.7520990371704102,
      "step": 73216
    },
    {
      "epoch": 6.982421875e-05,
      "step": 73216,
      "training_step_time": 6.870505094528198
    },
    {
      "epoch": 6.98251724243164e-05,
      "model_forward_time": 1.7584424018859863,
      "step": 73217
    },
    {
      "epoch": 6.98251724243164e-05,
      "step": 73217,
      "training_step_time": 6.862076759338379
    },
    {
      "epoch": 6.982612609863282e-05,
      "model_forward_time": 1.758997917175293,
      "step": 73218
    },
    {
      "epoch": 6.982612609863282e-05,
      "step": 73218,
      "training_step_time": 6.8182213306427
    },
    {
      "epoch": 6.982707977294921e-05,
      "model_forward_time": 1.3858883380889893,
      "step": 73219
    },
    {
      "epoch": 6.982707977294921e-05,
      "step": 73219,
      "training_step_time": 7.610302686691284
    },
    {
      "epoch": 6.982803344726562e-05,
      "grad_norm": 0.11120583862066269,
      "learning_rate": 1.8359967748442958e-06,
      "loss": 0.0544,
      "step": 73220
    },
    {
      "epoch": 6.982803344726562e-05,
      "model_forward_time": 1.7469379901885986,
      "step": 73220
    },
    {
      "dynamics_loss_avg": 0.04691618559882045,
      "epoch": 6.982803344726562e-05,
      "step": 73220
    },
    {
      "action_loss_avg": 0.0005888607265660539,
      "epoch": 6.982803344726562e-05,
      "step": 73220
    },
    {
      "epoch": 6.982803344726562e-05,
      "step": 73220,
      "training_step_time": 7.625976324081421
    },
    {
      "epoch": 6.982898712158203e-05,
      "model_forward_time": 2.3448691368103027,
      "step": 73221
    },
    {
      "epoch": 6.982898712158203e-05,
      "step": 73221,
      "training_step_time": 8.63474154472351
    },
    {
      "epoch": 6.982994079589843e-05,
      "model_forward_time": 1.7438299655914307,
      "step": 73222
    },
    {
      "epoch": 6.982994079589843e-05,
      "step": 73222,
      "training_step_time": 10.286890268325806
    },
    {
      "epoch": 6.983089447021484e-05,
      "model_forward_time": 1.7546188831329346,
      "step": 73223
    },
    {
      "epoch": 6.983089447021484e-05,
      "step": 73223,
      "training_step_time": 7.95082950592041
    },
    {
      "epoch": 6.983184814453125e-05,
      "model_forward_time": 1.7442307472229004,
      "step": 73224
    },
    {
      "epoch": 6.983184814453125e-05,
      "step": 73224,
      "training_step_time": 7.836594581604004
    },
    {
      "epoch": 6.983280181884765e-05,
      "model_forward_time": 1.7545349597930908,
      "step": 73225
    },
    {
      "epoch": 6.983280181884765e-05,
      "step": 73225,
      "training_step_time": 7.4201719760894775
    },
    {
      "epoch": 6.983375549316406e-05,
      "model_forward_time": 1.743696689605713,
      "step": 73226
    },
    {
      "epoch": 6.983375549316406e-05,
      "step": 73226,
      "training_step_time": 6.944682598114014
    },
    {
      "epoch": 6.983470916748047e-05,
      "model_forward_time": 1.7355530261993408,
      "step": 73227
    },
    {
      "epoch": 6.983470916748047e-05,
      "step": 73227,
      "training_step_time": 6.914997100830078
    },
    {
      "epoch": 6.983566284179687e-05,
      "model_forward_time": 1.7430169582366943,
      "step": 73228
    },
    {
      "epoch": 6.983566284179687e-05,
      "step": 73228,
      "training_step_time": 7.638745069503784
    },
    {
      "epoch": 6.983661651611328e-05,
      "model_forward_time": 1.3711490631103516,
      "step": 73229
    },
    {
      "epoch": 6.983661651611328e-05,
      "step": 73229,
      "training_step_time": 7.603722810745239
    },
    {
      "epoch": 6.983757019042969e-05,
      "grad_norm": 0.10049110651016235,
      "learning_rate": 1.8347166417920776e-06,
      "loss": 0.0578,
      "step": 73230
    },
    {
      "epoch": 6.983757019042969e-05,
      "model_forward_time": 2.3125555515289307,
      "step": 73230
    },
    {
      "dynamics_loss_avg": 0.04906728398054838,
      "epoch": 6.983757019042969e-05,
      "step": 73230
    },
    {
      "action_loss_avg": 0.0009370166837470606,
      "epoch": 6.983757019042969e-05,
      "step": 73230
    },
    {
      "epoch": 6.983757019042969e-05,
      "step": 73230,
      "training_step_time": 7.64139723777771
    },
    {
      "epoch": 6.983852386474609e-05,
      "model_forward_time": 1.814427137374878,
      "step": 73231
    },
    {
      "epoch": 6.983852386474609e-05,
      "step": 73231,
      "training_step_time": 9.142418622970581
    },
    {
      "epoch": 6.98394775390625e-05,
      "model_forward_time": 1.7275750637054443,
      "step": 73232
    },
    {
      "epoch": 6.98394775390625e-05,
      "step": 73232,
      "training_step_time": 8.962884426116943
    },
    {
      "epoch": 6.984043121337891e-05,
      "model_forward_time": 1.7285823822021484,
      "step": 73233
    },
    {
      "epoch": 6.984043121337891e-05,
      "step": 73233,
      "training_step_time": 7.988613843917847
    },
    {
      "epoch": 6.984138488769531e-05,
      "model_forward_time": 1.7280254364013672,
      "step": 73234
    },
    {
      "epoch": 6.984138488769531e-05,
      "step": 73234,
      "training_step_time": 7.768855333328247
    },
    {
      "epoch": 6.984233856201172e-05,
      "model_forward_time": 1.7728016376495361,
      "step": 73235
    },
    {
      "epoch": 6.984233856201172e-05,
      "step": 73235,
      "training_step_time": 6.871330261230469
    },
    {
      "epoch": 6.984329223632813e-05,
      "model_forward_time": 1.7167425155639648,
      "step": 73236
    },
    {
      "epoch": 6.984329223632813e-05,
      "step": 73236,
      "training_step_time": 6.8892881870269775
    },
    {
      "epoch": 6.984424591064453e-05,
      "model_forward_time": 1.730313777923584,
      "step": 73237
    },
    {
      "epoch": 6.984424591064453e-05,
      "step": 73237,
      "training_step_time": 7.594274997711182
    },
    {
      "epoch": 6.984519958496094e-05,
      "model_forward_time": 1.7319228649139404,
      "step": 73238
    },
    {
      "epoch": 6.984519958496094e-05,
      "step": 73238,
      "training_step_time": 7.555347204208374
    },
    {
      "epoch": 6.984615325927735e-05,
      "model_forward_time": 2.2790687084198,
      "step": 73239
    },
    {
      "epoch": 6.984615325927735e-05,
      "step": 73239,
      "training_step_time": 8.059017181396484
    },
    {
      "epoch": 6.984710693359375e-05,
      "grad_norm": 0.08924499154090881,
      "learning_rate": 1.833436854890484e-06,
      "loss": 0.0577,
      "step": 73240
    },
    {
      "epoch": 6.984710693359375e-05,
      "model_forward_time": 1.756887435913086,
      "step": 73240
    },
    {
      "dynamics_loss_avg": 0.05563581055030227,
      "epoch": 6.984710693359375e-05,
      "step": 73240
    },
    {
      "action_loss_avg": 0.0015276774705853314,
      "epoch": 6.984710693359375e-05,
      "step": 73240
    },
    {
      "epoch": 6.984710693359375e-05,
      "step": 73240,
      "training_step_time": 7.716817617416382
    },
    {
      "epoch": 6.984806060791016e-05,
      "model_forward_time": 1.7242059707641602,
      "step": 73241
    },
    {
      "epoch": 6.984806060791016e-05,
      "step": 73241,
      "training_step_time": 11.121201753616333
    },
    {
      "epoch": 6.984901428222657e-05,
      "model_forward_time": 1.7263929843902588,
      "step": 73242
    },
    {
      "epoch": 6.984901428222657e-05,
      "step": 73242,
      "training_step_time": 8.091652154922485
    },
    {
      "epoch": 6.984996795654296e-05,
      "model_forward_time": 1.733241319656372,
      "step": 73243
    },
    {
      "epoch": 6.984996795654296e-05,
      "step": 73243,
      "training_step_time": 7.798325061798096
    },
    {
      "epoch": 6.985092163085937e-05,
      "model_forward_time": 1.738048791885376,
      "step": 73244
    },
    {
      "epoch": 6.985092163085937e-05,
      "step": 73244,
      "training_step_time": 6.870903253555298
    },
    {
      "epoch": 6.985187530517579e-05,
      "model_forward_time": 1.7338180541992188,
      "step": 73245
    },
    {
      "epoch": 6.985187530517579e-05,
      "step": 73245,
      "training_step_time": 6.868778705596924
    },
    {
      "epoch": 6.985282897949218e-05,
      "model_forward_time": 1.7301907539367676,
      "step": 73246
    },
    {
      "epoch": 6.985282897949218e-05,
      "step": 73246,
      "training_step_time": 7.587648868560791
    },
    {
      "epoch": 6.98537826538086e-05,
      "model_forward_time": 1.723463535308838,
      "step": 73247
    },
    {
      "epoch": 6.98537826538086e-05,
      "step": 73247,
      "training_step_time": 7.398864269256592
    },
    {
      "epoch": 6.9854736328125e-05,
      "model_forward_time": 1.722475528717041,
      "step": 73248
    },
    {
      "epoch": 6.9854736328125e-05,
      "step": 73248,
      "training_step_time": 7.5622639656066895
    },
    {
      "epoch": 6.98556900024414e-05,
      "model_forward_time": 2.2541325092315674,
      "step": 73249
    },
    {
      "epoch": 6.98556900024414e-05,
      "step": 73249,
      "training_step_time": 9.272681474685669
    },
    {
      "epoch": 6.985664367675781e-05,
      "grad_norm": 0.12036741524934769,
      "learning_rate": 1.8321574142794795e-06,
      "loss": 0.061,
      "step": 73250
    },
    {
      "epoch": 6.985664367675781e-05,
      "model_forward_time": 1.725080966949463,
      "step": 73250
    },
    {
      "dynamics_loss_avg": 0.054308864381164314,
      "epoch": 6.985664367675781e-05,
      "step": 73250
    },
    {
      "action_loss_avg": 0.0008358338411198929,
      "epoch": 6.985664367675781e-05,
      "step": 73250
    },
    {
      "epoch": 6.985664367675781e-05,
      "step": 73250,
      "training_step_time": 8.326920986175537
    },
    {
      "epoch": 6.985759735107422e-05,
      "model_forward_time": 1.728754997253418,
      "step": 73251
    },
    {
      "epoch": 6.985759735107422e-05,
      "step": 73251,
      "training_step_time": 8.71971321105957
    },
    {
      "epoch": 6.985855102539062e-05,
      "model_forward_time": 1.7251310348510742,
      "step": 73252
    },
    {
      "epoch": 6.985855102539062e-05,
      "step": 73252,
      "training_step_time": 7.764021396636963
    },
    {
      "epoch": 6.985950469970703e-05,
      "model_forward_time": 1.7342779636383057,
      "step": 73253
    },
    {
      "epoch": 6.985950469970703e-05,
      "step": 73253,
      "training_step_time": 7.08081841468811
    },
    {
      "epoch": 6.986045837402344e-05,
      "model_forward_time": 1.7234530448913574,
      "step": 73254
    },
    {
      "epoch": 6.986045837402344e-05,
      "step": 73254,
      "training_step_time": 6.8758909702301025
    },
    {
      "epoch": 6.986141204833984e-05,
      "model_forward_time": 1.7351248264312744,
      "step": 73255
    },
    {
      "epoch": 6.986141204833984e-05,
      "step": 73255,
      "training_step_time": 7.631100177764893
    },
    {
      "epoch": 6.986236572265625e-05,
      "model_forward_time": 1.7331535816192627,
      "step": 73256
    },
    {
      "epoch": 6.986236572265625e-05,
      "step": 73256,
      "training_step_time": 7.157653331756592
    },
    {
      "epoch": 6.986331939697266e-05,
      "model_forward_time": 1.733426809310913,
      "step": 73257
    },
    {
      "epoch": 6.986331939697266e-05,
      "step": 73257,
      "training_step_time": 7.57305908203125
    },
    {
      "epoch": 6.986427307128906e-05,
      "model_forward_time": 2.267660140991211,
      "step": 73258
    },
    {
      "epoch": 6.986427307128906e-05,
      "step": 73258,
      "training_step_time": 10.400139331817627
    },
    {
      "epoch": 6.986522674560547e-05,
      "model_forward_time": 1.806535005569458,
      "step": 73259
    },
    {
      "epoch": 6.986522674560547e-05,
      "step": 73259,
      "training_step_time": 9.068569421768188
    },
    {
      "epoch": 6.986618041992188e-05,
      "grad_norm": 0.09595947712659836,
      "learning_rate": 1.830878320098975e-06,
      "loss": 0.0601,
      "step": 73260
    },
    {
      "epoch": 6.986618041992188e-05,
      "model_forward_time": 1.7258317470550537,
      "step": 73260
    },
    {
      "dynamics_loss_avg": 0.05421741437166929,
      "epoch": 6.986618041992188e-05,
      "step": 73260
    },
    {
      "action_loss_avg": 0.001137416617712006,
      "epoch": 6.986618041992188e-05,
      "step": 73260
    },
    {
      "epoch": 6.986618041992188e-05,
      "step": 73260,
      "training_step_time": 8.456348419189453
    },
    {
      "epoch": 6.986713409423828e-05,
      "model_forward_time": 1.7319209575653076,
      "step": 73261
    },
    {
      "epoch": 6.986713409423828e-05,
      "step": 73261,
      "training_step_time": 9.141838550567627
    },
    {
      "epoch": 6.986808776855469e-05,
      "model_forward_time": 1.7218868732452393,
      "step": 73262
    },
    {
      "epoch": 6.986808776855469e-05,
      "step": 73262,
      "training_step_time": 7.436082601547241
    },
    {
      "epoch": 6.98690414428711e-05,
      "model_forward_time": 1.7269904613494873,
      "step": 73263
    },
    {
      "epoch": 6.98690414428711e-05,
      "step": 73263,
      "training_step_time": 6.898818254470825
    },
    {
      "epoch": 6.98699951171875e-05,
      "model_forward_time": 1.7314884662628174,
      "step": 73264
    },
    {
      "epoch": 6.98699951171875e-05,
      "step": 73264,
      "training_step_time": 7.6485915184021
    },
    {
      "epoch": 6.987094879150391e-05,
      "model_forward_time": 1.7176542282104492,
      "step": 73265
    },
    {
      "epoch": 6.987094879150391e-05,
      "step": 73265,
      "training_step_time": 7.222164630889893
    },
    {
      "epoch": 6.987190246582032e-05,
      "model_forward_time": 1.7175755500793457,
      "step": 73266
    },
    {
      "epoch": 6.987190246582032e-05,
      "step": 73266,
      "training_step_time": 7.688173055648804
    },
    {
      "epoch": 6.987285614013671e-05,
      "model_forward_time": 2.2955169677734375,
      "step": 73267
    },
    {
      "epoch": 6.987285614013671e-05,
      "step": 73267,
      "training_step_time": 9.841145753860474
    },
    {
      "epoch": 6.987380981445313e-05,
      "model_forward_time": 1.7861616611480713,
      "step": 73268
    },
    {
      "epoch": 6.987380981445313e-05,
      "step": 73268,
      "training_step_time": 10.032493114471436
    },
    {
      "epoch": 6.987476348876954e-05,
      "model_forward_time": 1.7278733253479004,
      "step": 73269
    },
    {
      "epoch": 6.987476348876954e-05,
      "step": 73269,
      "training_step_time": 8.703569889068604
    },
    {
      "epoch": 6.987571716308593e-05,
      "grad_norm": 0.08476828038692474,
      "learning_rate": 1.8295995724888549e-06,
      "loss": 0.0554,
      "step": 73270
    },
    {
      "epoch": 6.987571716308593e-05,
      "model_forward_time": 1.7393684387207031,
      "step": 73270
    },
    {
      "dynamics_loss_avg": 0.07032495196908713,
      "epoch": 6.987571716308593e-05,
      "step": 73270
    },
    {
      "action_loss_avg": 0.0016168115922482683,
      "epoch": 6.987571716308593e-05,
      "step": 73270
    },
    {
      "epoch": 6.987571716308593e-05,
      "step": 73270,
      "training_step_time": 7.744096755981445
    },
    {
      "epoch": 6.987667083740234e-05,
      "model_forward_time": 1.7294588088989258,
      "step": 73271
    },
    {
      "epoch": 6.987667083740234e-05,
      "step": 73271,
      "training_step_time": 7.835733652114868
    },
    {
      "epoch": 6.987762451171876e-05,
      "model_forward_time": 1.7329609394073486,
      "step": 73272
    },
    {
      "epoch": 6.987762451171876e-05,
      "step": 73272,
      "training_step_time": 6.94819188117981
    },
    {
      "epoch": 6.987857818603515e-05,
      "model_forward_time": 1.7311363220214844,
      "step": 73273
    },
    {
      "epoch": 6.987857818603515e-05,
      "step": 73273,
      "training_step_time": 7.336656093597412
    },
    {
      "epoch": 6.987953186035156e-05,
      "model_forward_time": 1.7359306812286377,
      "step": 73274
    },
    {
      "epoch": 6.987953186035156e-05,
      "step": 73274,
      "training_step_time": 6.8961827754974365
    },
    {
      "epoch": 6.988048553466797e-05,
      "model_forward_time": 1.7288272380828857,
      "step": 73275
    },
    {
      "epoch": 6.988048553466797e-05,
      "step": 73275,
      "training_step_time": 8.384031534194946
    },
    {
      "epoch": 6.988143920898437e-05,
      "model_forward_time": 2.2835805416107178,
      "step": 73276
    },
    {
      "epoch": 6.988143920898437e-05,
      "step": 73276,
      "training_step_time": 7.7153942584991455
    },
    {
      "epoch": 6.988239288330078e-05,
      "model_forward_time": 1.7689690589904785,
      "step": 73277
    },
    {
      "epoch": 6.988239288330078e-05,
      "step": 73277,
      "training_step_time": 8.692219734191895
    },
    {
      "epoch": 6.988334655761719e-05,
      "model_forward_time": 1.3643312454223633,
      "step": 73278
    },
    {
      "epoch": 6.988334655761719e-05,
      "step": 73278,
      "training_step_time": 8.41805386543274
    },
    {
      "epoch": 6.988430023193359e-05,
      "model_forward_time": 1.73264479637146,
      "step": 73279
    },
    {
      "epoch": 6.988430023193359e-05,
      "step": 73279,
      "training_step_time": 7.624297142028809
    },
    {
      "epoch": 6.988525390625e-05,
      "grad_norm": 0.07657516747713089,
      "learning_rate": 1.828321171588957e-06,
      "loss": 0.0606,
      "step": 73280
    },
    {
      "epoch": 6.988525390625e-05,
      "model_forward_time": 1.72902512550354,
      "step": 73280
    },
    {
      "dynamics_loss_avg": 0.05551529042422772,
      "epoch": 6.988525390625e-05,
      "step": 73280
    },
    {
      "action_loss_avg": 0.000890275090932846,
      "epoch": 6.988525390625e-05,
      "step": 73280
    },
    {
      "epoch": 6.988525390625e-05,
      "step": 73280,
      "training_step_time": 7.797553777694702
    },
    {
      "epoch": 6.988620758056641e-05,
      "model_forward_time": 1.3666977882385254,
      "step": 73281
    },
    {
      "epoch": 6.988620758056641e-05,
      "step": 73281,
      "training_step_time": 6.907431125640869
    },
    {
      "epoch": 6.988716125488281e-05,
      "model_forward_time": 1.7278506755828857,
      "step": 73282
    },
    {
      "epoch": 6.988716125488281e-05,
      "step": 73282,
      "training_step_time": 7.624183893203735
    },
    {
      "epoch": 6.988811492919922e-05,
      "model_forward_time": 1.7288458347320557,
      "step": 73283
    },
    {
      "epoch": 6.988811492919922e-05,
      "step": 73283,
      "training_step_time": 6.8695032596588135
    },
    {
      "epoch": 6.988906860351563e-05,
      "model_forward_time": 1.7300708293914795,
      "step": 73284
    },
    {
      "epoch": 6.988906860351563e-05,
      "step": 73284,
      "training_step_time": 7.622348070144653
    },
    {
      "epoch": 6.989002227783203e-05,
      "model_forward_time": 1.731652021408081,
      "step": 73285
    },
    {
      "epoch": 6.989002227783203e-05,
      "step": 73285,
      "training_step_time": 9.198315143585205
    },
    {
      "epoch": 6.989097595214844e-05,
      "model_forward_time": 1.7623858451843262,
      "step": 73286
    },
    {
      "epoch": 6.989097595214844e-05,
      "step": 73286,
      "training_step_time": 9.29192066192627
    },
    {
      "epoch": 6.989192962646485e-05,
      "model_forward_time": 1.7315614223480225,
      "step": 73287
    },
    {
      "epoch": 6.989192962646485e-05,
      "step": 73287,
      "training_step_time": 9.411610126495361
    },
    {
      "epoch": 6.989288330078125e-05,
      "model_forward_time": 1.7273657321929932,
      "step": 73288
    },
    {
      "epoch": 6.989288330078125e-05,
      "step": 73288,
      "training_step_time": 8.774054765701294
    },
    {
      "epoch": 6.989383697509766e-05,
      "model_forward_time": 1.731111764907837,
      "step": 73289
    },
    {
      "epoch": 6.989383697509766e-05,
      "step": 73289,
      "training_step_time": 8.57229208946228
    },
    {
      "epoch": 6.989479064941407e-05,
      "grad_norm": 0.09656091034412384,
      "learning_rate": 1.827043117539089e-06,
      "loss": 0.0576,
      "step": 73290
    },
    {
      "epoch": 6.989479064941407e-05,
      "model_forward_time": 1.7294063568115234,
      "step": 73290
    },
    {
      "dynamics_loss_avg": 0.07322675306349993,
      "epoch": 6.989479064941407e-05,
      "step": 73290
    },
    {
      "action_loss_avg": 0.0016800450393930078,
      "epoch": 6.989479064941407e-05,
      "step": 73290
    },
    {
      "epoch": 6.989479064941407e-05,
      "step": 73290,
      "training_step_time": 6.867422819137573
    },
    {
      "epoch": 6.989574432373047e-05,
      "model_forward_time": 1.7334179878234863,
      "step": 73291
    },
    {
      "epoch": 6.989574432373047e-05,
      "step": 73291,
      "training_step_time": 6.88759970664978
    },
    {
      "epoch": 6.989669799804688e-05,
      "model_forward_time": 1.7289292812347412,
      "step": 73292
    },
    {
      "epoch": 6.989669799804688e-05,
      "step": 73292,
      "training_step_time": 7.652653932571411
    },
    {
      "epoch": 6.989765167236329e-05,
      "model_forward_time": 1.7395069599151611,
      "step": 73293
    },
    {
      "epoch": 6.989765167236329e-05,
      "step": 73293,
      "training_step_time": 7.5966150760650635
    },
    {
      "epoch": 6.989860534667968e-05,
      "model_forward_time": 1.7190525531768799,
      "step": 73294
    },
    {
      "epoch": 6.989860534667968e-05,
      "step": 73294,
      "training_step_time": 7.625178098678589
    },
    {
      "epoch": 6.98995590209961e-05,
      "model_forward_time": 2.308908700942993,
      "step": 73295
    },
    {
      "epoch": 6.98995590209961e-05,
      "step": 73295,
      "training_step_time": 8.7599618434906
    },
    {
      "epoch": 6.99005126953125e-05,
      "model_forward_time": 1.7296581268310547,
      "step": 73296
    },
    {
      "epoch": 6.99005126953125e-05,
      "step": 73296,
      "training_step_time": 8.881550550460815
    },
    {
      "epoch": 6.99014663696289e-05,
      "model_forward_time": 1.3660690784454346,
      "step": 73297
    },
    {
      "epoch": 6.99014663696289e-05,
      "step": 73297,
      "training_step_time": 8.951452732086182
    },
    {
      "epoch": 6.990242004394531e-05,
      "model_forward_time": 1.7160511016845703,
      "step": 73298
    },
    {
      "epoch": 6.990242004394531e-05,
      "step": 73298,
      "training_step_time": 7.701172113418579
    },
    {
      "epoch": 6.990337371826173e-05,
      "model_forward_time": 1.7399585247039795,
      "step": 73299
    },
    {
      "epoch": 6.990337371826173e-05,
      "step": 73299,
      "training_step_time": 7.729642868041992
    },
    {
      "epoch": 6.990432739257812e-05,
      "grad_norm": 0.0762130469083786,
      "learning_rate": 1.8257654104790118e-06,
      "loss": 0.056,
      "step": 73300
    },
    {
      "epoch": 6.990432739257812e-05,
      "model_forward_time": 1.7369046211242676,
      "step": 73300
    },
    {
      "dynamics_loss_avg": 0.05887891836464405,
      "epoch": 6.990432739257812e-05,
      "step": 73300
    },
    {
      "action_loss_avg": 0.000866672323900275,
      "epoch": 6.990432739257812e-05,
      "step": 73300
    },
    {
      "epoch": 6.990432739257812e-05,
      "step": 73300,
      "training_step_time": 6.9332804679870605
    },
    {
      "epoch": 6.990528106689453e-05,
      "model_forward_time": 1.7372493743896484,
      "step": 73301
    },
    {
      "epoch": 6.990528106689453e-05,
      "step": 73301,
      "training_step_time": 7.678492307662964
    },
    {
      "epoch": 6.990623474121094e-05,
      "model_forward_time": 1.7418773174285889,
      "step": 73302
    },
    {
      "epoch": 6.990623474121094e-05,
      "step": 73302,
      "training_step_time": 8.415122509002686
    },
    {
      "epoch": 6.990718841552734e-05,
      "model_forward_time": 1.727914571762085,
      "step": 73303
    },
    {
      "epoch": 6.990718841552734e-05,
      "step": 73303,
      "training_step_time": 7.510977745056152
    },
    {
      "epoch": 6.990814208984375e-05,
      "model_forward_time": 2.2789008617401123,
      "step": 73304
    },
    {
      "epoch": 6.990814208984375e-05,
      "step": 73304,
      "training_step_time": 8.403874158859253
    },
    {
      "epoch": 6.990909576416016e-05,
      "model_forward_time": 1.8072640895843506,
      "step": 73305
    },
    {
      "epoch": 6.990909576416016e-05,
      "step": 73305,
      "training_step_time": 8.567907571792603
    },
    {
      "epoch": 6.991004943847656e-05,
      "model_forward_time": 1.7404096126556396,
      "step": 73306
    },
    {
      "epoch": 6.991004943847656e-05,
      "step": 73306,
      "training_step_time": 7.880116939544678
    },
    {
      "epoch": 6.991100311279297e-05,
      "model_forward_time": 1.7322874069213867,
      "step": 73307
    },
    {
      "epoch": 6.991100311279297e-05,
      "step": 73307,
      "training_step_time": 8.622692823410034
    },
    {
      "epoch": 6.991195678710938e-05,
      "model_forward_time": 1.7402088642120361,
      "step": 73308
    },
    {
      "epoch": 6.991195678710938e-05,
      "step": 73308,
      "training_step_time": 7.8605852127075195
    },
    {
      "epoch": 6.991291046142578e-05,
      "model_forward_time": 1.75811767578125,
      "step": 73309
    },
    {
      "epoch": 6.991291046142578e-05,
      "step": 73309,
      "training_step_time": 6.833444595336914
    },
    {
      "epoch": 6.991386413574219e-05,
      "grad_norm": 0.08405067026615143,
      "learning_rate": 1.8244880505484603e-06,
      "loss": 0.0556,
      "step": 73310
    },
    {
      "epoch": 6.991386413574219e-05,
      "model_forward_time": 1.7390406131744385,
      "step": 73310
    },
    {
      "dynamics_loss_avg": 0.05879669580608606,
      "epoch": 6.991386413574219e-05,
      "step": 73310
    },
    {
      "action_loss_avg": 0.0015675253642257303,
      "epoch": 6.991386413574219e-05,
      "step": 73310
    },
    {
      "epoch": 6.991386413574219e-05,
      "step": 73310,
      "training_step_time": 7.65518045425415
    },
    {
      "epoch": 6.99148178100586e-05,
      "model_forward_time": 1.7349002361297607,
      "step": 73311
    },
    {
      "epoch": 6.99148178100586e-05,
      "step": 73311,
      "training_step_time": 7.320332288742065
    },
    {
      "epoch": 6.9915771484375e-05,
      "model_forward_time": 1.7391042709350586,
      "step": 73312
    },
    {
      "epoch": 6.9915771484375e-05,
      "step": 73312,
      "training_step_time": 7.467767715454102
    },
    {
      "epoch": 6.991672515869141e-05,
      "model_forward_time": 2.281773567199707,
      "step": 73313
    },
    {
      "epoch": 6.991672515869141e-05,
      "step": 73313,
      "training_step_time": 7.467402458190918
    },
    {
      "epoch": 6.99176788330078e-05,
      "model_forward_time": 1.8005003929138184,
      "step": 73314
    },
    {
      "epoch": 6.99176788330078e-05,
      "step": 73314,
      "training_step_time": 9.531641006469727
    },
    {
      "epoch": 6.991863250732422e-05,
      "model_forward_time": 1.7465741634368896,
      "step": 73315
    },
    {
      "epoch": 6.991863250732422e-05,
      "step": 73315,
      "training_step_time": 7.770416736602783
    },
    {
      "epoch": 6.991958618164063e-05,
      "model_forward_time": 1.7268297672271729,
      "step": 73316
    },
    {
      "epoch": 6.991958618164063e-05,
      "step": 73316,
      "training_step_time": 7.624748229980469
    },
    {
      "epoch": 6.992053985595702e-05,
      "model_forward_time": 1.7429735660552979,
      "step": 73317
    },
    {
      "epoch": 6.992053985595702e-05,
      "step": 73317,
      "training_step_time": 7.79288125038147
    },
    {
      "epoch": 6.992149353027344e-05,
      "model_forward_time": 1.7416343688964844,
      "step": 73318
    },
    {
      "epoch": 6.992149353027344e-05,
      "step": 73318,
      "training_step_time": 6.75098443031311
    },
    {
      "epoch": 6.992244720458985e-05,
      "model_forward_time": 1.7386951446533203,
      "step": 73319
    },
    {
      "epoch": 6.992244720458985e-05,
      "step": 73319,
      "training_step_time": 7.638267755508423
    },
    {
      "epoch": 6.992340087890624e-05,
      "grad_norm": 0.10443899780511856,
      "learning_rate": 1.8232110378871193e-06,
      "loss": 0.0512,
      "step": 73320
    },
    {
      "epoch": 6.992340087890624e-05,
      "model_forward_time": 1.736586332321167,
      "step": 73320
    },
    {
      "dynamics_loss_avg": 0.05386307220906019,
      "epoch": 6.992340087890624e-05,
      "step": 73320
    },
    {
      "action_loss_avg": 0.0012287152378121391,
      "epoch": 6.992340087890624e-05,
      "step": 73320
    },
    {
      "epoch": 6.992340087890624e-05,
      "step": 73320,
      "training_step_time": 7.125788688659668
    },
    {
      "epoch": 6.992435455322265e-05,
      "model_forward_time": 1.733546257019043,
      "step": 73321
    },
    {
      "epoch": 6.992435455322265e-05,
      "step": 73321,
      "training_step_time": 7.613313674926758
    },
    {
      "epoch": 6.992530822753907e-05,
      "model_forward_time": 2.2900640964508057,
      "step": 73322
    },
    {
      "epoch": 6.992530822753907e-05,
      "step": 73322,
      "training_step_time": 8.621839761734009
    },
    {
      "epoch": 6.992626190185546e-05,
      "model_forward_time": 1.7889735698699951,
      "step": 73323
    },
    {
      "epoch": 6.992626190185546e-05,
      "step": 73323,
      "training_step_time": 9.25809621810913
    },
    {
      "epoch": 6.992721557617187e-05,
      "model_forward_time": 1.747063159942627,
      "step": 73324
    },
    {
      "epoch": 6.992721557617187e-05,
      "step": 73324,
      "training_step_time": 8.321260452270508
    },
    {
      "epoch": 6.992816925048828e-05,
      "model_forward_time": 1.7188935279846191,
      "step": 73325
    },
    {
      "epoch": 6.992816925048828e-05,
      "step": 73325,
      "training_step_time": 7.53984260559082
    },
    {
      "epoch": 6.992912292480468e-05,
      "model_forward_time": 1.739682912826538,
      "step": 73326
    },
    {
      "epoch": 6.992912292480468e-05,
      "step": 73326,
      "training_step_time": 7.680020332336426
    },
    {
      "epoch": 6.993007659912109e-05,
      "model_forward_time": 1.7575981616973877,
      "step": 73327
    },
    {
      "epoch": 6.993007659912109e-05,
      "step": 73327,
      "training_step_time": 6.705408573150635
    },
    {
      "epoch": 6.99310302734375e-05,
      "model_forward_time": 1.7479212284088135,
      "step": 73328
    },
    {
      "epoch": 6.99310302734375e-05,
      "step": 73328,
      "training_step_time": 7.59782338142395
    },
    {
      "epoch": 6.99319839477539e-05,
      "model_forward_time": 1.7373771667480469,
      "step": 73329
    },
    {
      "epoch": 6.99319839477539e-05,
      "step": 73329,
      "training_step_time": 7.0792412757873535
    },
    {
      "epoch": 6.993293762207031e-05,
      "grad_norm": 0.08357458561658859,
      "learning_rate": 1.821934372634645e-06,
      "loss": 0.0539,
      "step": 73330
    },
    {
      "epoch": 6.993293762207031e-05,
      "model_forward_time": 1.7354354858398438,
      "step": 73330
    },
    {
      "dynamics_loss_avg": 0.049626709707081315,
      "epoch": 6.993293762207031e-05,
      "step": 73330
    },
    {
      "action_loss_avg": 0.0014448436297243461,
      "epoch": 6.993293762207031e-05,
      "step": 73330
    },
    {
      "epoch": 6.993293762207031e-05,
      "step": 73330,
      "training_step_time": 7.569889307022095
    },
    {
      "epoch": 6.993389129638672e-05,
      "model_forward_time": 2.277381181716919,
      "step": 73331
    },
    {
      "epoch": 6.993389129638672e-05,
      "step": 73331,
      "training_step_time": 7.56566047668457
    },
    {
      "epoch": 6.993484497070312e-05,
      "model_forward_time": 1.7761645317077637,
      "step": 73332
    },
    {
      "epoch": 6.993484497070312e-05,
      "step": 73332,
      "training_step_time": 7.914045333862305
    },
    {
      "epoch": 6.993579864501953e-05,
      "model_forward_time": 1.7321550846099854,
      "step": 73333
    },
    {
      "epoch": 6.993579864501953e-05,
      "step": 73333,
      "training_step_time": 8.208623170852661
    },
    {
      "epoch": 6.993675231933594e-05,
      "model_forward_time": 1.7425343990325928,
      "step": 73334
    },
    {
      "epoch": 6.993675231933594e-05,
      "step": 73334,
      "training_step_time": 9.28097915649414
    },
    {
      "epoch": 6.993770599365234e-05,
      "model_forward_time": 1.733865737915039,
      "step": 73335
    },
    {
      "epoch": 6.993770599365234e-05,
      "step": 73335,
      "training_step_time": 8.219596147537231
    },
    {
      "epoch": 6.993865966796875e-05,
      "model_forward_time": 1.739689826965332,
      "step": 73336
    },
    {
      "epoch": 6.993865966796875e-05,
      "step": 73336,
      "training_step_time": 7.579294681549072
    },
    {
      "epoch": 6.993961334228516e-05,
      "model_forward_time": 1.7153940200805664,
      "step": 73337
    },
    {
      "epoch": 6.993961334228516e-05,
      "step": 73337,
      "training_step_time": 7.515026092529297
    },
    {
      "epoch": 6.994056701660156e-05,
      "model_forward_time": 1.7183935642242432,
      "step": 73338
    },
    {
      "epoch": 6.994056701660156e-05,
      "step": 73338,
      "training_step_time": 7.018467426300049
    },
    {
      "epoch": 6.994152069091797e-05,
      "model_forward_time": 1.7262046337127686,
      "step": 73339
    },
    {
      "epoch": 6.994152069091797e-05,
      "step": 73339,
      "training_step_time": 7.518510818481445
    },
    {
      "epoch": 6.994247436523438e-05,
      "grad_norm": 0.0987592414021492,
      "learning_rate": 1.8206580549306475e-06,
      "loss": 0.0568,
      "step": 73340
    },
    {
      "epoch": 6.994247436523438e-05,
      "model_forward_time": 2.2509474754333496,
      "step": 73340
    },
    {
      "dynamics_loss_avg": 0.06658868566155433,
      "epoch": 6.994247436523438e-05,
      "step": 73340
    },
    {
      "action_loss_avg": 0.0008774787056609057,
      "epoch": 6.994247436523438e-05,
      "step": 73340
    },
    {
      "epoch": 6.994247436523438e-05,
      "step": 73340,
      "training_step_time": 8.5733802318573
    },
    {
      "epoch": 6.994342803955078e-05,
      "model_forward_time": 1.7324459552764893,
      "step": 73341
    },
    {
      "epoch": 6.994342803955078e-05,
      "step": 73341,
      "training_step_time": 8.677727222442627
    },
    {
      "epoch": 6.994438171386719e-05,
      "model_forward_time": 1.723259449005127,
      "step": 73342
    },
    {
      "epoch": 6.994438171386719e-05,
      "step": 73342,
      "training_step_time": 8.765154838562012
    },
    {
      "epoch": 6.99453353881836e-05,
      "model_forward_time": 1.7180683612823486,
      "step": 73343
    },
    {
      "epoch": 6.99453353881836e-05,
      "step": 73343,
      "training_step_time": 7.7241737842559814
    },
    {
      "epoch": 6.99462890625e-05,
      "model_forward_time": 1.7193024158477783,
      "step": 73344
    },
    {
      "epoch": 6.99462890625e-05,
      "step": 73344,
      "training_step_time": 7.664225339889526
    },
    {
      "epoch": 6.99472427368164e-05,
      "model_forward_time": 1.7262253761291504,
      "step": 73345
    },
    {
      "epoch": 6.99472427368164e-05,
      "step": 73345,
      "training_step_time": 7.592103719711304
    },
    {
      "epoch": 6.994819641113282e-05,
      "model_forward_time": 1.7207930088043213,
      "step": 73346
    },
    {
      "epoch": 6.994819641113282e-05,
      "step": 73346,
      "training_step_time": 7.459906578063965
    },
    {
      "epoch": 6.994915008544921e-05,
      "model_forward_time": 1.7308108806610107,
      "step": 73347
    },
    {
      "epoch": 6.994915008544921e-05,
      "step": 73347,
      "training_step_time": 7.015727519989014
    },
    {
      "epoch": 6.995010375976562e-05,
      "model_forward_time": 1.6991240978240967,
      "step": 73348
    },
    {
      "epoch": 6.995010375976562e-05,
      "step": 73348,
      "training_step_time": 7.299885034561157
    },
    {
      "epoch": 6.995105743408203e-05,
      "model_forward_time": 1.7092711925506592,
      "step": 73349
    },
    {
      "epoch": 6.995105743408203e-05,
      "step": 73349,
      "training_step_time": 7.562074184417725
    },
    {
      "epoch": 6.995201110839843e-05,
      "grad_norm": 0.08478917181491852,
      "learning_rate": 1.819382084914707e-06,
      "loss": 0.0533,
      "step": 73350
    },
    {
      "epoch": 6.995201110839843e-05,
      "model_forward_time": 2.261087656021118,
      "step": 73350
    },
    {
      "dynamics_loss_avg": 0.050304565019905566,
      "epoch": 6.995201110839843e-05,
      "step": 73350
    },
    {
      "action_loss_avg": 0.000938997115008533,
      "epoch": 6.995201110839843e-05,
      "step": 73350
    },
    {
      "epoch": 6.995201110839843e-05,
      "step": 73350,
      "training_step_time": 8.639670848846436
    },
    {
      "epoch": 6.995296478271484e-05,
      "model_forward_time": 1.7286226749420166,
      "step": 73351
    },
    {
      "epoch": 6.995296478271484e-05,
      "step": 73351,
      "training_step_time": 9.350634813308716
    },
    {
      "epoch": 6.995391845703125e-05,
      "model_forward_time": 1.3698742389678955,
      "step": 73352
    },
    {
      "epoch": 6.995391845703125e-05,
      "step": 73352,
      "training_step_time": 7.786693572998047
    },
    {
      "epoch": 6.995487213134765e-05,
      "model_forward_time": 1.7519721984863281,
      "step": 73353
    },
    {
      "epoch": 6.995487213134765e-05,
      "step": 73353,
      "training_step_time": 7.7448859214782715
    },
    {
      "epoch": 6.995582580566406e-05,
      "model_forward_time": 1.730726957321167,
      "step": 73354
    },
    {
      "epoch": 6.995582580566406e-05,
      "step": 73354,
      "training_step_time": 7.7251176834106445
    },
    {
      "epoch": 6.995677947998047e-05,
      "model_forward_time": 1.7559940814971924,
      "step": 73355
    },
    {
      "epoch": 6.995677947998047e-05,
      "step": 73355,
      "training_step_time": 6.847141981124878
    },
    {
      "epoch": 6.995773315429687e-05,
      "model_forward_time": 1.730691909790039,
      "step": 73356
    },
    {
      "epoch": 6.995773315429687e-05,
      "step": 73356,
      "training_step_time": 7.581288576126099
    },
    {
      "epoch": 6.995868682861328e-05,
      "model_forward_time": 1.7296421527862549,
      "step": 73357
    },
    {
      "epoch": 6.995868682861328e-05,
      "step": 73357,
      "training_step_time": 7.484375715255737
    },
    {
      "epoch": 6.995964050292969e-05,
      "model_forward_time": 1.7384214401245117,
      "step": 73358
    },
    {
      "epoch": 6.995964050292969e-05,
      "step": 73358,
      "training_step_time": 8.415334701538086
    },
    {
      "epoch": 6.996059417724609e-05,
      "model_forward_time": 2.281561851501465,
      "step": 73359
    },
    {
      "epoch": 6.996059417724609e-05,
      "step": 73359,
      "training_step_time": 9.444738388061523
    },
    {
      "epoch": 6.99615478515625e-05,
      "grad_norm": 0.11875060200691223,
      "learning_rate": 1.8181064627263578e-06,
      "loss": 0.058,
      "step": 73360
    },
    {
      "epoch": 6.99615478515625e-05,
      "model_forward_time": 1.4125986099243164,
      "step": 73360
    },
    {
      "dynamics_loss_avg": 0.06256305724382401,
      "epoch": 6.99615478515625e-05,
      "step": 73360
    },
    {
      "action_loss_avg": 0.0010164555977098643,
      "epoch": 6.99615478515625e-05,
      "step": 73360
    },
    {
      "epoch": 6.99615478515625e-05,
      "step": 73360,
      "training_step_time": 9.570581674575806
    },
    {
      "epoch": 6.996250152587891e-05,
      "model_forward_time": 1.7422635555267334,
      "step": 73361
    },
    {
      "epoch": 6.996250152587891e-05,
      "step": 73361,
      "training_step_time": 8.50106143951416
    },
    {
      "epoch": 6.996345520019531e-05,
      "model_forward_time": 1.7314965724945068,
      "step": 73362
    },
    {
      "epoch": 6.996345520019531e-05,
      "step": 73362,
      "training_step_time": 7.521403789520264
    },
    {
      "epoch": 6.996440887451172e-05,
      "model_forward_time": 1.7339556217193604,
      "step": 73363
    },
    {
      "epoch": 6.996440887451172e-05,
      "step": 73363,
      "training_step_time": 7.7284135818481445
    },
    {
      "epoch": 6.996536254882813e-05,
      "model_forward_time": 1.7427513599395752,
      "step": 73364
    },
    {
      "epoch": 6.996536254882813e-05,
      "step": 73364,
      "training_step_time": 6.826714038848877
    },
    {
      "epoch": 6.996631622314453e-05,
      "model_forward_time": 1.366924524307251,
      "step": 73365
    },
    {
      "epoch": 6.996631622314453e-05,
      "step": 73365,
      "training_step_time": 7.62112021446228
    },
    {
      "epoch": 6.996726989746094e-05,
      "model_forward_time": 1.747680902481079,
      "step": 73366
    },
    {
      "epoch": 6.996726989746094e-05,
      "step": 73366,
      "training_step_time": 7.3405256271362305
    },
    {
      "epoch": 6.996822357177735e-05,
      "model_forward_time": 1.7350122928619385,
      "step": 73367
    },
    {
      "epoch": 6.996822357177735e-05,
      "step": 73367,
      "training_step_time": 8.170532941818237
    },
    {
      "epoch": 6.996917724609375e-05,
      "model_forward_time": 2.2353339195251465,
      "step": 73368
    },
    {
      "epoch": 6.996917724609375e-05,
      "step": 73368,
      "training_step_time": 7.519267797470093
    },
    {
      "epoch": 6.997013092041016e-05,
      "model_forward_time": 1.7705090045928955,
      "step": 73369
    },
    {
      "epoch": 6.997013092041016e-05,
      "step": 73369,
      "training_step_time": 10.206347465515137
    },
    {
      "epoch": 6.997108459472657e-05,
      "grad_norm": 0.09624461829662323,
      "learning_rate": 1.816831188505102e-06,
      "loss": 0.0604,
      "step": 73370
    },
    {
      "epoch": 6.997108459472657e-05,
      "model_forward_time": 1.741257667541504,
      "step": 73370
    },
    {
      "dynamics_loss_avg": 0.07081740070134401,
      "epoch": 6.997108459472657e-05,
      "step": 73370
    },
    {
      "action_loss_avg": 0.0011015023977961392,
      "epoch": 6.997108459472657e-05,
      "step": 73370
    },
    {
      "epoch": 6.997108459472657e-05,
      "step": 73370,
      "training_step_time": 7.8889381885528564
    },
    {
      "epoch": 6.997203826904296e-05,
      "model_forward_time": 1.7300903797149658,
      "step": 73371
    },
    {
      "epoch": 6.997203826904296e-05,
      "step": 73371,
      "training_step_time": 9.700753688812256
    },
    {
      "epoch": 6.997299194335937e-05,
      "model_forward_time": 1.7361314296722412,
      "step": 73372
    },
    {
      "epoch": 6.997299194335937e-05,
      "step": 73372,
      "training_step_time": 7.827882289886475
    },
    {
      "epoch": 6.997394561767579e-05,
      "model_forward_time": 1.7366316318511963,
      "step": 73373
    },
    {
      "epoch": 6.997394561767579e-05,
      "step": 73373,
      "training_step_time": 7.433139324188232
    },
    {
      "epoch": 6.997489929199218e-05,
      "model_forward_time": 1.7341907024383545,
      "step": 73374
    },
    {
      "epoch": 6.997489929199218e-05,
      "step": 73374,
      "training_step_time": 7.639919996261597
    },
    {
      "epoch": 6.99758529663086e-05,
      "model_forward_time": 1.7379803657531738,
      "step": 73375
    },
    {
      "epoch": 6.99758529663086e-05,
      "step": 73375,
      "training_step_time": 7.421133041381836
    },
    {
      "epoch": 6.9976806640625e-05,
      "model_forward_time": 1.7308802604675293,
      "step": 73376
    },
    {
      "epoch": 6.9976806640625e-05,
      "step": 73376,
      "training_step_time": 7.616653203964233
    },
    {
      "epoch": 6.99777603149414e-05,
      "model_forward_time": 1.7380881309509277,
      "step": 73377
    },
    {
      "epoch": 6.99777603149414e-05,
      "step": 73377,
      "training_step_time": 8.412249565124512
    },
    {
      "epoch": 6.997871398925781e-05,
      "model_forward_time": 2.354713201522827,
      "step": 73378
    },
    {
      "epoch": 6.997871398925781e-05,
      "step": 73378,
      "training_step_time": 8.776058912277222
    },
    {
      "epoch": 6.997966766357422e-05,
      "model_forward_time": 1.7417991161346436,
      "step": 73379
    },
    {
      "epoch": 6.997966766357422e-05,
      "step": 73379,
      "training_step_time": 9.498881816864014
    },
    {
      "epoch": 6.998062133789062e-05,
      "grad_norm": 0.0766676515340805,
      "learning_rate": 1.8155562623904021e-06,
      "loss": 0.0588,
      "step": 73380
    },
    {
      "epoch": 6.998062133789062e-05,
      "model_forward_time": 1.7369096279144287,
      "step": 73380
    },
    {
      "dynamics_loss_avg": 0.049972367100417615,
      "epoch": 6.998062133789062e-05,
      "step": 73380
    },
    {
      "action_loss_avg": 0.002161454930319451,
      "epoch": 6.998062133789062e-05,
      "step": 73380
    },
    {
      "epoch": 6.998062133789062e-05,
      "step": 73380,
      "training_step_time": 8.66175127029419
    },
    {
      "epoch": 6.998157501220703e-05,
      "model_forward_time": 1.737297534942627,
      "step": 73381
    },
    {
      "epoch": 6.998157501220703e-05,
      "step": 73381,
      "training_step_time": 7.6972362995147705
    },
    {
      "epoch": 6.998252868652344e-05,
      "model_forward_time": 1.7399680614471436,
      "step": 73382
    },
    {
      "epoch": 6.998252868652344e-05,
      "step": 73382,
      "training_step_time": 7.603195667266846
    },
    {
      "epoch": 6.998348236083984e-05,
      "model_forward_time": 1.72723388671875,
      "step": 73383
    },
    {
      "epoch": 6.998348236083984e-05,
      "step": 73383,
      "training_step_time": 7.549839496612549
    },
    {
      "epoch": 6.998443603515625e-05,
      "model_forward_time": 1.7341878414154053,
      "step": 73384
    },
    {
      "epoch": 6.998443603515625e-05,
      "step": 73384,
      "training_step_time": 7.386751890182495
    },
    {
      "epoch": 6.998538970947266e-05,
      "model_forward_time": 1.7189571857452393,
      "step": 73385
    },
    {
      "epoch": 6.998538970947266e-05,
      "step": 73385,
      "training_step_time": 7.618744850158691
    },
    {
      "epoch": 6.998634338378906e-05,
      "model_forward_time": 1.725271224975586,
      "step": 73386
    },
    {
      "epoch": 6.998634338378906e-05,
      "step": 73386,
      "training_step_time": 7.547892093658447
    },
    {
      "epoch": 6.998729705810547e-05,
      "model_forward_time": 2.2582907676696777,
      "step": 73387
    },
    {
      "epoch": 6.998729705810547e-05,
      "step": 73387,
      "training_step_time": 8.487215280532837
    },
    {
      "epoch": 6.998825073242188e-05,
      "model_forward_time": 1.8005273342132568,
      "step": 73388
    },
    {
      "epoch": 6.998825073242188e-05,
      "step": 73388,
      "training_step_time": 7.955030918121338
    },
    {
      "epoch": 6.998920440673828e-05,
      "model_forward_time": 1.7174670696258545,
      "step": 73389
    },
    {
      "epoch": 6.998920440673828e-05,
      "step": 73389,
      "training_step_time": 7.79661750793457
    },
    {
      "epoch": 6.999015808105469e-05,
      "grad_norm": 0.0855938121676445,
      "learning_rate": 1.8142816845216827e-06,
      "loss": 0.0566,
      "step": 73390
    },
    {
      "epoch": 6.999015808105469e-05,
      "model_forward_time": 1.735034465789795,
      "step": 73390
    },
    {
      "dynamics_loss_avg": 0.0432166425511241,
      "epoch": 6.999015808105469e-05,
      "step": 73390
    },
    {
      "action_loss_avg": 0.0009510891512036323,
      "epoch": 6.999015808105469e-05,
      "step": 73390
    },
    {
      "epoch": 6.999015808105469e-05,
      "step": 73390,
      "training_step_time": 7.73865270614624
    },
    {
      "epoch": 6.99911117553711e-05,
      "model_forward_time": 1.7283720970153809,
      "step": 73391
    },
    {
      "epoch": 6.99911117553711e-05,
      "step": 73391,
      "training_step_time": 7.839488983154297
    },
    {
      "epoch": 6.99920654296875e-05,
      "model_forward_time": 1.7121655941009521,
      "step": 73392
    },
    {
      "epoch": 6.99920654296875e-05,
      "step": 73392,
      "training_step_time": 7.881548643112183
    },
    {
      "epoch": 6.999301910400391e-05,
      "model_forward_time": 1.7236974239349365,
      "step": 73393
    },
    {
      "epoch": 6.999301910400391e-05,
      "step": 73393,
      "training_step_time": 7.5455451011657715
    },
    {
      "epoch": 6.999397277832032e-05,
      "model_forward_time": 1.715522050857544,
      "step": 73394
    },
    {
      "epoch": 6.999397277832032e-05,
      "step": 73394,
      "training_step_time": 7.80009913444519
    },
    {
      "epoch": 6.999492645263671e-05,
      "model_forward_time": 1.717655897140503,
      "step": 73395
    },
    {
      "epoch": 6.999492645263671e-05,
      "step": 73395,
      "training_step_time": 7.687285661697388
    },
    {
      "epoch": 6.999588012695313e-05,
      "model_forward_time": 2.2634799480438232,
      "step": 73396
    },
    {
      "epoch": 6.999588012695313e-05,
      "step": 73396,
      "training_step_time": 8.551062822341919
    },
    {
      "epoch": 6.999683380126954e-05,
      "model_forward_time": 1.8003771305084229,
      "step": 73397
    },
    {
      "epoch": 6.999683380126954e-05,
      "step": 73397,
      "training_step_time": 9.198590517044067
    },
    {
      "epoch": 6.999778747558593e-05,
      "model_forward_time": 1.727768898010254,
      "step": 73398
    },
    {
      "epoch": 6.999778747558593e-05,
      "step": 73398,
      "training_step_time": 7.81131386756897
    },
    {
      "epoch": 6.999874114990234e-05,
      "model_forward_time": 1.7205028533935547,
      "step": 73399
    },
    {
      "epoch": 6.999874114990234e-05,
      "step": 73399,
      "training_step_time": 7.629488706588745
    },
    {
      "epoch": 6.999969482421876e-05,
      "grad_norm": 0.09658090770244598,
      "learning_rate": 1.8130074550383275e-06,
      "loss": 0.0572,
      "step": 73400
    },
    {
      "epoch": 6.999969482421876e-05,
      "model_forward_time": 1.7151196002960205,
      "step": 73400
    },
    {
      "dynamics_loss_avg": 0.047885117772966626,
      "epoch": 6.999969482421876e-05,
      "step": 73400
    },
    {
      "action_loss_avg": 0.001090997461869847,
      "epoch": 6.999969482421876e-05,
      "step": 73400
    },
    {
      "epoch": 6.999969482421876e-05,
      "step": 73400,
      "training_step_time": 8.775346040725708
    },
    {
      "epoch": 7.000064849853515e-05,
      "model_forward_time": 1.690917730331421,
      "step": 73401
    },
    {
      "epoch": 7.000064849853515e-05,
      "step": 73401,
      "training_step_time": 7.702501058578491
    },
    {
      "epoch": 7.000160217285156e-05,
      "model_forward_time": 1.7168703079223633,
      "step": 73402
    },
    {
      "epoch": 7.000160217285156e-05,
      "step": 73402,
      "training_step_time": 7.460392236709595
    },
    {
      "epoch": 7.000255584716797e-05,
      "model_forward_time": 1.3483026027679443,
      "step": 73403
    },
    {
      "epoch": 7.000255584716797e-05,
      "step": 73403,
      "training_step_time": 7.691295385360718
    },
    {
      "epoch": 7.000350952148437e-05,
      "model_forward_time": 1.7117550373077393,
      "step": 73404
    },
    {
      "epoch": 7.000350952148437e-05,
      "step": 73404,
      "training_step_time": 7.630878448486328
    },
    {
      "epoch": 7.000446319580078e-05,
      "model_forward_time": 2.245129108428955,
      "step": 73405
    },
    {
      "epoch": 7.000446319580078e-05,
      "step": 73405,
      "training_step_time": 7.807474136352539
    },
    {
      "epoch": 7.000541687011719e-05,
      "model_forward_time": 1.7974624633789062,
      "step": 73406
    },
    {
      "epoch": 7.000541687011719e-05,
      "step": 73406,
      "training_step_time": 8.516129970550537
    },
    {
      "epoch": 7.000637054443359e-05,
      "model_forward_time": 1.7117078304290771,
      "step": 73407
    },
    {
      "epoch": 7.000637054443359e-05,
      "step": 73407,
      "training_step_time": 7.631085157394409
    },
    {
      "epoch": 7.000732421875e-05,
      "model_forward_time": 1.7320890426635742,
      "step": 73408
    },
    {
      "epoch": 7.000732421875e-05,
      "step": 73408,
      "training_step_time": 8.029048681259155
    },
    {
      "epoch": 7.000827789306641e-05,
      "model_forward_time": 1.7171547412872314,
      "step": 73409
    },
    {
      "epoch": 7.000827789306641e-05,
      "step": 73409,
      "training_step_time": 7.684038400650024
    },
    {
      "epoch": 7.000923156738281e-05,
      "grad_norm": 0.1365341693162918,
      "learning_rate": 1.8117335740796848e-06,
      "loss": 0.0585,
      "step": 73410
    },
    {
      "epoch": 7.000923156738281e-05,
      "model_forward_time": 1.7138793468475342,
      "step": 73410
    },
    {
      "dynamics_loss_avg": 0.053664210811257364,
      "epoch": 7.000923156738281e-05,
      "step": 73410
    },
    {
      "action_loss_avg": 0.0018941004585940392,
      "epoch": 7.000923156738281e-05,
      "step": 73410
    },
    {
      "epoch": 7.000923156738281e-05,
      "step": 73410,
      "training_step_time": 7.647434949874878
    },
    {
      "epoch": 7.001018524169922e-05,
      "model_forward_time": 1.7205758094787598,
      "step": 73411
    },
    {
      "epoch": 7.001018524169922e-05,
      "step": 73411,
      "training_step_time": 7.36972713470459
    },
    {
      "epoch": 7.001113891601563e-05,
      "model_forward_time": 1.7137620449066162,
      "step": 73412
    },
    {
      "epoch": 7.001113891601563e-05,
      "step": 73412,
      "training_step_time": 8.24755573272705
    },
    {
      "epoch": 7.001209259033203e-05,
      "model_forward_time": 1.7127149105072021,
      "step": 73413
    },
    {
      "epoch": 7.001209259033203e-05,
      "step": 73413,
      "training_step_time": 7.468235731124878
    },
    {
      "epoch": 7.001304626464844e-05,
      "model_forward_time": 1.723743200302124,
      "step": 73414
    },
    {
      "epoch": 7.001304626464844e-05,
      "step": 73414,
      "training_step_time": 7.86079216003418
    },
    {
      "epoch": 7.001399993896485e-05,
      "model_forward_time": 2.34541916847229,
      "step": 73415
    },
    {
      "epoch": 7.001399993896485e-05,
      "step": 73415,
      "training_step_time": 7.6523756980896
    },
    {
      "epoch": 7.001495361328125e-05,
      "model_forward_time": 1.728524923324585,
      "step": 73416
    },
    {
      "epoch": 7.001495361328125e-05,
      "step": 73416,
      "training_step_time": 8.500088214874268
    },
    {
      "epoch": 7.001590728759766e-05,
      "model_forward_time": 1.715926170349121,
      "step": 73417
    },
    {
      "epoch": 7.001590728759766e-05,
      "step": 73417,
      "training_step_time": 7.707826614379883
    },
    {
      "epoch": 7.001686096191407e-05,
      "model_forward_time": 1.7086329460144043,
      "step": 73418
    },
    {
      "epoch": 7.001686096191407e-05,
      "step": 73418,
      "training_step_time": 7.6796674728393555
    },
    {
      "epoch": 7.001781463623047e-05,
      "model_forward_time": 1.7299954891204834,
      "step": 73419
    },
    {
      "epoch": 7.001781463623047e-05,
      "step": 73419,
      "training_step_time": 8.517594337463379
    },
    {
      "epoch": 7.001876831054688e-05,
      "grad_norm": 0.09396631270647049,
      "learning_rate": 1.810460041785067e-06,
      "loss": 0.0628,
      "step": 73420
    },
    {
      "epoch": 7.001876831054688e-05,
      "model_forward_time": 1.359332799911499,
      "step": 73420
    },
    {
      "dynamics_loss_avg": 0.05614164117723704,
      "epoch": 7.001876831054688e-05,
      "step": 73420
    },
    {
      "action_loss_avg": 0.001162545138504356,
      "epoch": 7.001876831054688e-05,
      "step": 73420
    },
    {
      "epoch": 7.001876831054688e-05,
      "step": 73420,
      "training_step_time": 7.3886518478393555
    },
    {
      "epoch": 7.001972198486329e-05,
      "model_forward_time": 1.7227787971496582,
      "step": 73421
    },
    {
      "epoch": 7.001972198486329e-05,
      "step": 73421,
      "training_step_time": 7.398874998092651
    },
    {
      "epoch": 7.002067565917968e-05,
      "model_forward_time": 1.7038416862487793,
      "step": 73422
    },
    {
      "epoch": 7.002067565917968e-05,
      "step": 73422,
      "training_step_time": 7.561497211456299
    },
    {
      "epoch": 7.00216293334961e-05,
      "model_forward_time": 1.7214252948760986,
      "step": 73423
    },
    {
      "epoch": 7.00216293334961e-05,
      "step": 73423,
      "training_step_time": 7.7900707721710205
    },
    {
      "epoch": 7.00225830078125e-05,
      "model_forward_time": 2.3142893314361572,
      "step": 73424
    },
    {
      "epoch": 7.00225830078125e-05,
      "step": 73424,
      "training_step_time": 8.649204730987549
    },
    {
      "epoch": 7.00235366821289e-05,
      "model_forward_time": 1.7184298038482666,
      "step": 73425
    },
    {
      "epoch": 7.00235366821289e-05,
      "step": 73425,
      "training_step_time": 9.541759490966797
    },
    {
      "epoch": 7.002449035644531e-05,
      "model_forward_time": 1.7354068756103516,
      "step": 73426
    },
    {
      "epoch": 7.002449035644531e-05,
      "step": 73426,
      "training_step_time": 7.741899013519287
    },
    {
      "epoch": 7.002544403076173e-05,
      "model_forward_time": 1.7144784927368164,
      "step": 73427
    },
    {
      "epoch": 7.002544403076173e-05,
      "step": 73427,
      "training_step_time": 7.646640062332153
    },
    {
      "epoch": 7.002639770507812e-05,
      "model_forward_time": 1.7178196907043457,
      "step": 73428
    },
    {
      "epoch": 7.002639770507812e-05,
      "step": 73428,
      "training_step_time": 7.684894323348999
    },
    {
      "epoch": 7.002735137939453e-05,
      "model_forward_time": 1.721045970916748,
      "step": 73429
    },
    {
      "epoch": 7.002735137939453e-05,
      "step": 73429,
      "training_step_time": 7.518683195114136
    },
    {
      "epoch": 7.002830505371094e-05,
      "grad_norm": 0.09901213645935059,
      "learning_rate": 1.8091868582937423e-06,
      "loss": 0.0594,
      "step": 73430
    },
    {
      "epoch": 7.002830505371094e-05,
      "model_forward_time": 1.7076239585876465,
      "step": 73430
    },
    {
      "dynamics_loss_avg": 0.045684156380593775,
      "epoch": 7.002830505371094e-05,
      "step": 73430
    },
    {
      "action_loss_avg": 0.0015924489125609398,
      "epoch": 7.002830505371094e-05,
      "step": 73430
    },
    {
      "epoch": 7.002830505371094e-05,
      "step": 73430,
      "training_step_time": 7.359469413757324
    },
    {
      "epoch": 7.002925872802734e-05,
      "model_forward_time": 1.7132360935211182,
      "step": 73431
    },
    {
      "epoch": 7.002925872802734e-05,
      "step": 73431,
      "training_step_time": 7.700248718261719
    },
    {
      "epoch": 7.003021240234375e-05,
      "model_forward_time": 1.712460994720459,
      "step": 73432
    },
    {
      "epoch": 7.003021240234375e-05,
      "step": 73432,
      "training_step_time": 8.800601959228516
    },
    {
      "epoch": 7.003116607666016e-05,
      "model_forward_time": 2.31992244720459,
      "step": 73433
    },
    {
      "epoch": 7.003116607666016e-05,
      "step": 73433,
      "training_step_time": 8.728482007980347
    },
    {
      "epoch": 7.003211975097656e-05,
      "model_forward_time": 1.7182142734527588,
      "step": 73434
    },
    {
      "epoch": 7.003211975097656e-05,
      "step": 73434,
      "training_step_time": 10.43524980545044
    },
    {
      "epoch": 7.003307342529297e-05,
      "model_forward_time": 1.723210334777832,
      "step": 73435
    },
    {
      "epoch": 7.003307342529297e-05,
      "step": 73435,
      "training_step_time": 9.322274446487427
    },
    {
      "epoch": 7.003402709960938e-05,
      "model_forward_time": 1.352989912033081,
      "step": 73436
    },
    {
      "epoch": 7.003402709960938e-05,
      "step": 73436,
      "training_step_time": 8.408043384552002
    },
    {
      "epoch": 7.003498077392578e-05,
      "model_forward_time": 1.7145779132843018,
      "step": 73437
    },
    {
      "epoch": 7.003498077392578e-05,
      "step": 73437,
      "training_step_time": 7.7772216796875
    },
    {
      "epoch": 7.003593444824219e-05,
      "model_forward_time": 1.728731393814087,
      "step": 73438
    },
    {
      "epoch": 7.003593444824219e-05,
      "step": 73438,
      "training_step_time": 7.462795734405518
    },
    {
      "epoch": 7.003688812255859e-05,
      "model_forward_time": 1.3434147834777832,
      "step": 73439
    },
    {
      "epoch": 7.003688812255859e-05,
      "step": 73439,
      "training_step_time": 7.3910298347473145
    },
    {
      "epoch": 7.0037841796875e-05,
      "grad_norm": 0.1285575032234192,
      "learning_rate": 1.807914023744945e-06,
      "loss": 0.0586,
      "step": 73440
    },
    {
      "epoch": 7.0037841796875e-05,
      "model_forward_time": 1.718430757522583,
      "step": 73440
    },
    {
      "dynamics_loss_avg": 0.047006319276988506,
      "epoch": 7.0037841796875e-05,
      "step": 73440
    },
    {
      "action_loss_avg": 0.000898186193080619,
      "epoch": 7.0037841796875e-05,
      "step": 73440
    },
    {
      "epoch": 7.0037841796875e-05,
      "step": 73440,
      "training_step_time": 7.711166858673096
    },
    {
      "epoch": 7.003879547119141e-05,
      "model_forward_time": 1.7231271266937256,
      "step": 73441
    },
    {
      "epoch": 7.003879547119141e-05,
      "step": 73441,
      "training_step_time": 7.712160587310791
    },
    {
      "epoch": 7.00397491455078e-05,
      "model_forward_time": 2.2851836681365967,
      "step": 73442
    },
    {
      "epoch": 7.00397491455078e-05,
      "step": 73442,
      "training_step_time": 8.75991439819336
    },
    {
      "epoch": 7.004070281982422e-05,
      "model_forward_time": 1.7213585376739502,
      "step": 73443
    },
    {
      "epoch": 7.004070281982422e-05,
      "step": 73443,
      "training_step_time": 11.43522334098816
    },
    {
      "epoch": 7.004165649414063e-05,
      "model_forward_time": 1.7136387825012207,
      "step": 73444
    },
    {
      "epoch": 7.004165649414063e-05,
      "step": 73444,
      "training_step_time": 7.926257610321045
    },
    {
      "epoch": 7.004261016845702e-05,
      "model_forward_time": 1.714784860610962,
      "step": 73445
    },
    {
      "epoch": 7.004261016845702e-05,
      "step": 73445,
      "training_step_time": 8.969969749450684
    },
    {
      "epoch": 7.004356384277344e-05,
      "model_forward_time": 1.7142186164855957,
      "step": 73446
    },
    {
      "epoch": 7.004356384277344e-05,
      "step": 73446,
      "training_step_time": 7.708935499191284
    },
    {
      "epoch": 7.004451751708985e-05,
      "model_forward_time": 1.714287281036377,
      "step": 73447
    },
    {
      "epoch": 7.004451751708985e-05,
      "step": 73447,
      "training_step_time": 7.500398635864258
    },
    {
      "epoch": 7.004547119140624e-05,
      "model_forward_time": 1.7201848030090332,
      "step": 73448
    },
    {
      "epoch": 7.004547119140624e-05,
      "step": 73448,
      "training_step_time": 6.772014856338501
    },
    {
      "epoch": 7.004642486572265e-05,
      "model_forward_time": 1.708418846130371,
      "step": 73449
    },
    {
      "epoch": 7.004642486572265e-05,
      "step": 73449,
      "training_step_time": 7.651645660400391
    },
    {
      "epoch": 7.004737854003907e-05,
      "grad_norm": 0.10145602375268936,
      "learning_rate": 1.8066415382778708e-06,
      "loss": 0.0589,
      "step": 73450
    },
    {
      "epoch": 7.004737854003907e-05,
      "model_forward_time": 1.7080824375152588,
      "step": 73450
    },
    {
      "dynamics_loss_avg": 0.0624227256514132,
      "epoch": 7.004737854003907e-05,
      "step": 73450
    },
    {
      "action_loss_avg": 0.0011978786264080555,
      "epoch": 7.004737854003907e-05,
      "step": 73450
    },
    {
      "epoch": 7.004737854003907e-05,
      "step": 73450,
      "training_step_time": 7.439018964767456
    },
    {
      "epoch": 7.004833221435546e-05,
      "model_forward_time": 1.7469711303710938,
      "step": 73451
    },
    {
      "epoch": 7.004833221435546e-05,
      "step": 73451,
      "training_step_time": 8.758971929550171
    },
    {
      "epoch": 7.004928588867187e-05,
      "model_forward_time": 1.715827465057373,
      "step": 73452
    },
    {
      "epoch": 7.004928588867187e-05,
      "step": 73452,
      "training_step_time": 9.303508043289185
    },
    {
      "epoch": 7.005023956298828e-05,
      "model_forward_time": 1.7178442478179932,
      "step": 73453
    },
    {
      "epoch": 7.005023956298828e-05,
      "step": 73453,
      "training_step_time": 10.312533617019653
    },
    {
      "epoch": 7.005119323730468e-05,
      "model_forward_time": 1.717332363128662,
      "step": 73454
    },
    {
      "epoch": 7.005119323730468e-05,
      "step": 73454,
      "training_step_time": 8.493096590042114
    },
    {
      "epoch": 7.005214691162109e-05,
      "model_forward_time": 1.7072501182556152,
      "step": 73455
    },
    {
      "epoch": 7.005214691162109e-05,
      "step": 73455,
      "training_step_time": 7.754395961761475
    },
    {
      "epoch": 7.00531005859375e-05,
      "model_forward_time": 1.7276725769042969,
      "step": 73456
    },
    {
      "epoch": 7.00531005859375e-05,
      "step": 73456,
      "training_step_time": 8.241213083267212
    },
    {
      "epoch": 7.00540542602539e-05,
      "model_forward_time": 1.7220020294189453,
      "step": 73457
    },
    {
      "epoch": 7.00540542602539e-05,
      "step": 73457,
      "training_step_time": 6.995758056640625
    },
    {
      "epoch": 7.005500793457031e-05,
      "model_forward_time": 1.7405834197998047,
      "step": 73458
    },
    {
      "epoch": 7.005500793457031e-05,
      "step": 73458,
      "training_step_time": 7.595016956329346
    },
    {
      "epoch": 7.005596160888672e-05,
      "model_forward_time": 1.7317099571228027,
      "step": 73459
    },
    {
      "epoch": 7.005596160888672e-05,
      "step": 73459,
      "training_step_time": 7.409031867980957
    },
    {
      "epoch": 7.005691528320312e-05,
      "grad_norm": 0.08505523204803467,
      "learning_rate": 1.8053694020316787e-06,
      "loss": 0.0583,
      "step": 73460
    },
    {
      "epoch": 7.005691528320312e-05,
      "model_forward_time": 1.7483665943145752,
      "step": 73460
    },
    {
      "dynamics_loss_avg": 0.0673702135682106,
      "epoch": 7.005691528320312e-05,
      "step": 73460
    },
    {
      "action_loss_avg": 0.001622324244817719,
      "epoch": 7.005691528320312e-05,
      "step": 73460
    },
    {
      "epoch": 7.005691528320312e-05,
      "step": 73460,
      "training_step_time": 8.224308490753174
    },
    {
      "epoch": 7.005786895751953e-05,
      "model_forward_time": 2.279395580291748,
      "step": 73461
    },
    {
      "epoch": 7.005786895751953e-05,
      "step": 73461,
      "training_step_time": 10.460740327835083
    },
    {
      "epoch": 7.005882263183594e-05,
      "model_forward_time": 1.725273609161377,
      "step": 73462
    },
    {
      "epoch": 7.005882263183594e-05,
      "step": 73462,
      "training_step_time": 8.533738374710083
    },
    {
      "epoch": 7.005977630615234e-05,
      "model_forward_time": 1.739943504333496,
      "step": 73463
    },
    {
      "epoch": 7.005977630615234e-05,
      "step": 73463,
      "training_step_time": 8.204122066497803
    },
    {
      "epoch": 7.006072998046875e-05,
      "model_forward_time": 1.728330135345459,
      "step": 73464
    },
    {
      "epoch": 7.006072998046875e-05,
      "step": 73464,
      "training_step_time": 7.694559574127197
    },
    {
      "epoch": 7.006168365478516e-05,
      "model_forward_time": 1.73453688621521,
      "step": 73465
    },
    {
      "epoch": 7.006168365478516e-05,
      "step": 73465,
      "training_step_time": 7.747631549835205
    },
    {
      "epoch": 7.006263732910156e-05,
      "model_forward_time": 1.7276740074157715,
      "step": 73466
    },
    {
      "epoch": 7.006263732910156e-05,
      "step": 73466,
      "training_step_time": 7.491098642349243
    },
    {
      "epoch": 7.006359100341797e-05,
      "model_forward_time": 1.731929063796997,
      "step": 73467
    },
    {
      "epoch": 7.006359100341797e-05,
      "step": 73467,
      "training_step_time": 7.490058898925781
    },
    {
      "epoch": 7.006454467773438e-05,
      "model_forward_time": 1.74491548538208,
      "step": 73468
    },
    {
      "epoch": 7.006454467773438e-05,
      "step": 73468,
      "training_step_time": 7.504154920578003
    },
    {
      "epoch": 7.006549835205078e-05,
      "model_forward_time": 1.7368316650390625,
      "step": 73469
    },
    {
      "epoch": 7.006549835205078e-05,
      "step": 73469,
      "training_step_time": 9.11902666091919
    },
    {
      "epoch": 7.006645202636719e-05,
      "grad_norm": 0.12163842469453812,
      "learning_rate": 1.8040976151454836e-06,
      "loss": 0.0596,
      "step": 73470
    },
    {
      "epoch": 7.006645202636719e-05,
      "model_forward_time": 2.3018686771392822,
      "step": 73470
    },
    {
      "dynamics_loss_avg": 0.057082387432456014,
      "epoch": 7.006645202636719e-05,
      "step": 73470
    },
    {
      "action_loss_avg": 0.0008640698011731729,
      "epoch": 7.006645202636719e-05,
      "step": 73470
    },
    {
      "epoch": 7.006645202636719e-05,
      "step": 73470,
      "training_step_time": 7.575806379318237
    },
    {
      "epoch": 7.00674057006836e-05,
      "model_forward_time": 1.6968224048614502,
      "step": 73471
    },
    {
      "epoch": 7.00674057006836e-05,
      "step": 73471,
      "training_step_time": 8.832051277160645
    },
    {
      "epoch": 7.0068359375e-05,
      "model_forward_time": 1.7160217761993408,
      "step": 73472
    },
    {
      "epoch": 7.0068359375e-05,
      "step": 73472,
      "training_step_time": 7.8012707233428955
    },
    {
      "epoch": 7.00693130493164e-05,
      "model_forward_time": 1.731309413909912,
      "step": 73473
    },
    {
      "epoch": 7.00693130493164e-05,
      "step": 73473,
      "training_step_time": 7.629706144332886
    },
    {
      "epoch": 7.007026672363282e-05,
      "model_forward_time": 1.7350597381591797,
      "step": 73474
    },
    {
      "epoch": 7.007026672363282e-05,
      "step": 73474,
      "training_step_time": 7.734962463378906
    },
    {
      "epoch": 7.007122039794921e-05,
      "model_forward_time": 1.7117640972137451,
      "step": 73475
    },
    {
      "epoch": 7.007122039794921e-05,
      "step": 73475,
      "training_step_time": 7.533027410507202
    },
    {
      "epoch": 7.007217407226562e-05,
      "model_forward_time": 1.7140963077545166,
      "step": 73476
    },
    {
      "epoch": 7.007217407226562e-05,
      "step": 73476,
      "training_step_time": 7.93188214302063
    },
    {
      "epoch": 7.007312774658203e-05,
      "model_forward_time": 1.7146620750427246,
      "step": 73477
    },
    {
      "epoch": 7.007312774658203e-05,
      "step": 73477,
      "training_step_time": 7.527529954910278
    },
    {
      "epoch": 7.007408142089843e-05,
      "model_forward_time": 1.7192251682281494,
      "step": 73478
    },
    {
      "epoch": 7.007408142089843e-05,
      "step": 73478,
      "training_step_time": 7.646810293197632
    },
    {
      "epoch": 7.007503509521484e-05,
      "model_forward_time": 2.3116378784179688,
      "step": 73479
    },
    {
      "epoch": 7.007503509521484e-05,
      "step": 73479,
      "training_step_time": 8.594249486923218
    },
    {
      "epoch": 7.007598876953125e-05,
      "grad_norm": 0.09500854462385178,
      "learning_rate": 1.8028261777583705e-06,
      "loss": 0.0567,
      "step": 73480
    },
    {
      "epoch": 7.007598876953125e-05,
      "model_forward_time": 1.7055485248565674,
      "step": 73480
    },
    {
      "dynamics_loss_avg": 0.04927767887711525,
      "epoch": 7.007598876953125e-05,
      "step": 73480
    },
    {
      "action_loss_avg": 0.0008069027273450047,
      "epoch": 7.007598876953125e-05,
      "step": 73480
    },
    {
      "epoch": 7.007598876953125e-05,
      "step": 73480,
      "training_step_time": 7.761599540710449
    },
    {
      "epoch": 7.007694244384765e-05,
      "model_forward_time": 1.736370325088501,
      "step": 73481
    },
    {
      "epoch": 7.007694244384765e-05,
      "step": 73481,
      "training_step_time": 8.002402544021606
    },
    {
      "epoch": 7.007789611816406e-05,
      "model_forward_time": 1.7121212482452393,
      "step": 73482
    },
    {
      "epoch": 7.007789611816406e-05,
      "step": 73482,
      "training_step_time": 8.501965761184692
    },
    {
      "epoch": 7.007884979248047e-05,
      "model_forward_time": 1.7199106216430664,
      "step": 73483
    },
    {
      "epoch": 7.007884979248047e-05,
      "step": 73483,
      "training_step_time": 7.76764988899231
    },
    {
      "epoch": 7.007980346679687e-05,
      "model_forward_time": 1.7150652408599854,
      "step": 73484
    },
    {
      "epoch": 7.007980346679687e-05,
      "step": 73484,
      "training_step_time": 7.499127626419067
    },
    {
      "epoch": 7.008075714111328e-05,
      "model_forward_time": 1.713623285293579,
      "step": 73485
    },
    {
      "epoch": 7.008075714111328e-05,
      "step": 73485,
      "training_step_time": 7.585103273391724
    },
    {
      "epoch": 7.008171081542969e-05,
      "model_forward_time": 1.7219505310058594,
      "step": 73486
    },
    {
      "epoch": 7.008171081542969e-05,
      "step": 73486,
      "training_step_time": 8.291714191436768
    },
    {
      "epoch": 7.008266448974609e-05,
      "model_forward_time": 1.7250123023986816,
      "step": 73487
    },
    {
      "epoch": 7.008266448974609e-05,
      "step": 73487,
      "training_step_time": 7.604701042175293
    },
    {
      "epoch": 7.00836181640625e-05,
      "model_forward_time": 2.287933826446533,
      "step": 73488
    },
    {
      "epoch": 7.00836181640625e-05,
      "step": 73488,
      "training_step_time": 8.210783243179321
    },
    {
      "epoch": 7.008457183837891e-05,
      "model_forward_time": 1.7136635780334473,
      "step": 73489
    },
    {
      "epoch": 7.008457183837891e-05,
      "step": 73489,
      "training_step_time": 8.554672718048096
    },
    {
      "epoch": 7.008552551269531e-05,
      "grad_norm": 0.09531907737255096,
      "learning_rate": 1.801555090009376e-06,
      "loss": 0.0564,
      "step": 73490
    },
    {
      "epoch": 7.008552551269531e-05,
      "model_forward_time": 1.7187490463256836,
      "step": 73490
    },
    {
      "dynamics_loss_avg": 0.0656980885192752,
      "epoch": 7.008552551269531e-05,
      "step": 73490
    },
    {
      "action_loss_avg": 0.0011874177114805206,
      "epoch": 7.008552551269531e-05,
      "step": 73490
    },
    {
      "epoch": 7.008552551269531e-05,
      "step": 73490,
      "training_step_time": 7.907644987106323
    },
    {
      "epoch": 7.008647918701172e-05,
      "model_forward_time": 1.7188365459442139,
      "step": 73491
    },
    {
      "epoch": 7.008647918701172e-05,
      "step": 73491,
      "training_step_time": 7.60503625869751
    },
    {
      "epoch": 7.008743286132813e-05,
      "model_forward_time": 1.721909999847412,
      "step": 73492
    },
    {
      "epoch": 7.008743286132813e-05,
      "step": 73492,
      "training_step_time": 7.698690891265869
    },
    {
      "epoch": 7.008838653564453e-05,
      "model_forward_time": 1.3622734546661377,
      "step": 73493
    },
    {
      "epoch": 7.008838653564453e-05,
      "step": 73493,
      "training_step_time": 7.288961887359619
    },
    {
      "epoch": 7.008934020996094e-05,
      "model_forward_time": 1.7206318378448486,
      "step": 73494
    },
    {
      "epoch": 7.008934020996094e-05,
      "step": 73494,
      "training_step_time": 8.36873984336853
    },
    {
      "epoch": 7.009029388427735e-05,
      "model_forward_time": 1.7206134796142578,
      "step": 73495
    },
    {
      "epoch": 7.009029388427735e-05,
      "step": 73495,
      "training_step_time": 8.417735815048218
    },
    {
      "epoch": 7.009124755859375e-05,
      "model_forward_time": 1.7099597454071045,
      "step": 73496
    },
    {
      "epoch": 7.009124755859375e-05,
      "step": 73496,
      "training_step_time": 7.468001365661621
    },
    {
      "epoch": 7.009220123291016e-05,
      "model_forward_time": 2.270658254623413,
      "step": 73497
    },
    {
      "epoch": 7.009220123291016e-05,
      "step": 73497,
      "training_step_time": 9.320646524429321
    },
    {
      "epoch": 7.009315490722657e-05,
      "model_forward_time": 1.7341299057006836,
      "step": 73498
    },
    {
      "epoch": 7.009315490722657e-05,
      "step": 73498,
      "training_step_time": 8.12718677520752
    },
    {
      "epoch": 7.009410858154296e-05,
      "model_forward_time": 1.7136623859405518,
      "step": 73499
    },
    {
      "epoch": 7.009410858154296e-05,
      "step": 73499,
      "training_step_time": 7.837477684020996
    },
    {
      "epoch": 7.009506225585938e-05,
      "grad_norm": 0.07329171895980835,
      "learning_rate": 1.8002843520375118e-06,
      "loss": 0.0522,
      "step": 73500
    },
    {
      "epoch": 7.009506225585938e-05,
      "model_forward_time": 1.7270779609680176,
      "step": 73500
    },
    {
      "dynamics_loss_avg": 0.0724986655637622,
      "epoch": 7.009506225585938e-05,
      "step": 73500
    },
    {
      "action_loss_avg": 0.0010475266972207464,
      "epoch": 7.009506225585938e-05,
      "step": 73500
    },
    {
      "epoch": 7.009506225585938e-05,
      "step": 73500,
      "training_step_time": 8.200642347335815
    },
    {
      "epoch": 7.009601593017579e-05,
      "model_forward_time": 1.7343459129333496,
      "step": 73501
    },
    {
      "epoch": 7.009601593017579e-05,
      "step": 73501,
      "training_step_time": 7.731978178024292
    },
    {
      "epoch": 7.009696960449218e-05,
      "model_forward_time": 1.7178122997283936,
      "step": 73502
    },
    {
      "epoch": 7.009696960449218e-05,
      "step": 73502,
      "training_step_time": 7.883915901184082
    },
    {
      "epoch": 7.00979232788086e-05,
      "model_forward_time": 1.3330743312835693,
      "step": 73503
    },
    {
      "epoch": 7.00979232788086e-05,
      "step": 73503,
      "training_step_time": 7.633458852767944
    },
    {
      "epoch": 7.0098876953125e-05,
      "model_forward_time": 1.7159724235534668,
      "step": 73504
    },
    {
      "epoch": 7.0098876953125e-05,
      "step": 73504,
      "training_step_time": 7.7878735065460205
    },
    {
      "epoch": 7.00998306274414e-05,
      "model_forward_time": 1.7211685180664062,
      "step": 73505
    },
    {
      "epoch": 7.00998306274414e-05,
      "step": 73505,
      "training_step_time": 7.64634895324707
    },
    {
      "epoch": 7.010078430175781e-05,
      "model_forward_time": 1.727534532546997,
      "step": 73506
    },
    {
      "epoch": 7.010078430175781e-05,
      "step": 73506,
      "training_step_time": 9.72121000289917
    },
    {
      "epoch": 7.010173797607422e-05,
      "model_forward_time": 2.2718141078948975,
      "step": 73507
    },
    {
      "epoch": 7.010173797607422e-05,
      "step": 73507,
      "training_step_time": 8.735987901687622
    },
    {
      "epoch": 7.010269165039062e-05,
      "model_forward_time": 1.723191261291504,
      "step": 73508
    },
    {
      "epoch": 7.010269165039062e-05,
      "step": 73508,
      "training_step_time": 8.00940489768982
    },
    {
      "epoch": 7.010364532470703e-05,
      "model_forward_time": 1.7258269786834717,
      "step": 73509
    },
    {
      "epoch": 7.010364532470703e-05,
      "step": 73509,
      "training_step_time": 8.371540784835815
    },
    {
      "epoch": 7.010459899902344e-05,
      "grad_norm": 0.12114518135786057,
      "learning_rate": 1.7990139639817383e-06,
      "loss": 0.0572,
      "step": 73510
    },
    {
      "epoch": 7.010459899902344e-05,
      "model_forward_time": 1.7266476154327393,
      "step": 73510
    },
    {
      "dynamics_loss_avg": 0.048324756883084775,
      "epoch": 7.010459899902344e-05,
      "step": 73510
    },
    {
      "action_loss_avg": 0.0011134981119539588,
      "epoch": 7.010459899902344e-05,
      "step": 73510
    },
    {
      "epoch": 7.010459899902344e-05,
      "step": 73510,
      "training_step_time": 7.736989259719849
    },
    {
      "epoch": 7.010555267333984e-05,
      "model_forward_time": 1.71832275390625,
      "step": 73511
    },
    {
      "epoch": 7.010555267333984e-05,
      "step": 73511,
      "training_step_time": 7.868514776229858
    },
    {
      "epoch": 7.010650634765625e-05,
      "model_forward_time": 1.715116024017334,
      "step": 73512
    },
    {
      "epoch": 7.010650634765625e-05,
      "step": 73512,
      "training_step_time": 8.482719898223877
    },
    {
      "epoch": 7.010746002197266e-05,
      "model_forward_time": 1.7402737140655518,
      "step": 73513
    },
    {
      "epoch": 7.010746002197266e-05,
      "step": 73513,
      "training_step_time": 7.775559425354004
    },
    {
      "epoch": 7.010841369628906e-05,
      "model_forward_time": 1.7041270732879639,
      "step": 73514
    },
    {
      "epoch": 7.010841369628906e-05,
      "step": 73514,
      "training_step_time": 7.626871109008789
    },
    {
      "epoch": 7.010936737060547e-05,
      "model_forward_time": 1.7302145957946777,
      "step": 73515
    },
    {
      "epoch": 7.010936737060547e-05,
      "step": 73515,
      "training_step_time": 8.609912157058716
    },
    {
      "epoch": 7.011032104492188e-05,
      "model_forward_time": 2.314312219619751,
      "step": 73516
    },
    {
      "epoch": 7.011032104492188e-05,
      "step": 73516,
      "training_step_time": 9.487295866012573
    },
    {
      "epoch": 7.011127471923828e-05,
      "model_forward_time": 1.3473708629608154,
      "step": 73517
    },
    {
      "epoch": 7.011127471923828e-05,
      "step": 73517,
      "training_step_time": 8.543826580047607
    },
    {
      "epoch": 7.011222839355469e-05,
      "model_forward_time": 1.7122523784637451,
      "step": 73518
    },
    {
      "epoch": 7.011222839355469e-05,
      "step": 73518,
      "training_step_time": 7.882142066955566
    },
    {
      "epoch": 7.01131820678711e-05,
      "model_forward_time": 1.7158186435699463,
      "step": 73519
    },
    {
      "epoch": 7.01131820678711e-05,
      "step": 73519,
      "training_step_time": 8.207422494888306
    },
    {
      "epoch": 7.01141357421875e-05,
      "grad_norm": 0.11610573530197144,
      "learning_rate": 1.797743925980987e-06,
      "loss": 0.0538,
      "step": 73520
    },
    {
      "epoch": 7.01141357421875e-05,
      "model_forward_time": 1.7169122695922852,
      "step": 73520
    },
    {
      "dynamics_loss_avg": 0.047407443076372145,
      "epoch": 7.01141357421875e-05,
      "step": 73520
    },
    {
      "action_loss_avg": 0.001398801984032616,
      "epoch": 7.01141357421875e-05,
      "step": 73520
    },
    {
      "epoch": 7.01141357421875e-05,
      "step": 73520,
      "training_step_time": 7.81653094291687
    },
    {
      "epoch": 7.011508941650391e-05,
      "model_forward_time": 1.7139322757720947,
      "step": 73521
    },
    {
      "epoch": 7.011508941650391e-05,
      "step": 73521,
      "training_step_time": 7.5490641593933105
    },
    {
      "epoch": 7.011604309082032e-05,
      "model_forward_time": 1.7120370864868164,
      "step": 73522
    },
    {
      "epoch": 7.011604309082032e-05,
      "step": 73522,
      "training_step_time": 7.629644155502319
    },
    {
      "epoch": 7.011699676513672e-05,
      "model_forward_time": 1.7323410511016846,
      "step": 73523
    },
    {
      "epoch": 7.011699676513672e-05,
      "step": 73523,
      "training_step_time": 7.582030296325684
    },
    {
      "epoch": 7.011795043945313e-05,
      "model_forward_time": 1.7224891185760498,
      "step": 73524
    },
    {
      "epoch": 7.011795043945313e-05,
      "step": 73524,
      "training_step_time": 8.64656686782837
    },
    {
      "epoch": 7.011890411376954e-05,
      "model_forward_time": 2.3230443000793457,
      "step": 73525
    },
    {
      "epoch": 7.011890411376954e-05,
      "step": 73525,
      "training_step_time": 8.216851949691772
    },
    {
      "epoch": 7.011985778808593e-05,
      "model_forward_time": 1.720881700515747,
      "step": 73526
    },
    {
      "epoch": 7.011985778808593e-05,
      "step": 73526,
      "training_step_time": 8.965954303741455
    },
    {
      "epoch": 7.012081146240234e-05,
      "model_forward_time": 1.7261254787445068,
      "step": 73527
    },
    {
      "epoch": 7.012081146240234e-05,
      "step": 73527,
      "training_step_time": 8.77522873878479
    },
    {
      "epoch": 7.012176513671876e-05,
      "model_forward_time": 1.7281882762908936,
      "step": 73528
    },
    {
      "epoch": 7.012176513671876e-05,
      "step": 73528,
      "training_step_time": 8.409040451049805
    },
    {
      "epoch": 7.012271881103515e-05,
      "model_forward_time": 1.70845365524292,
      "step": 73529
    },
    {
      "epoch": 7.012271881103515e-05,
      "step": 73529,
      "training_step_time": 7.647254943847656
    },
    {
      "epoch": 7.012367248535156e-05,
      "grad_norm": 0.1054491475224495,
      "learning_rate": 1.7964742381741434e-06,
      "loss": 0.0554,
      "step": 73530
    },
    {
      "epoch": 7.012367248535156e-05,
      "model_forward_time": 1.7140460014343262,
      "step": 73530
    },
    {
      "dynamics_loss_avg": 0.06430321037769318,
      "epoch": 7.012367248535156e-05,
      "step": 73530
    },
    {
      "action_loss_avg": 0.0016150357434526086,
      "epoch": 7.012367248535156e-05,
      "step": 73530
    },
    {
      "epoch": 7.012367248535156e-05,
      "step": 73530,
      "training_step_time": 8.676753759384155
    },
    {
      "epoch": 7.012462615966797e-05,
      "model_forward_time": 1.7195100784301758,
      "step": 73531
    },
    {
      "epoch": 7.012462615966797e-05,
      "step": 73531,
      "training_step_time": 8.279025077819824
    },
    {
      "epoch": 7.012557983398437e-05,
      "model_forward_time": 1.701960563659668,
      "step": 73532
    },
    {
      "epoch": 7.012557983398437e-05,
      "step": 73532,
      "training_step_time": 7.63038182258606
    },
    {
      "epoch": 7.012653350830078e-05,
      "model_forward_time": 1.7084987163543701,
      "step": 73533
    },
    {
      "epoch": 7.012653350830078e-05,
      "step": 73533,
      "training_step_time": 7.428916692733765
    },
    {
      "epoch": 7.01274871826172e-05,
      "model_forward_time": 1.744866132736206,
      "step": 73534
    },
    {
      "epoch": 7.01274871826172e-05,
      "step": 73534,
      "training_step_time": 8.713142395019531
    },
    {
      "epoch": 7.012844085693359e-05,
      "model_forward_time": 1.7124860286712646,
      "step": 73535
    },
    {
      "epoch": 7.012844085693359e-05,
      "step": 73535,
      "training_step_time": 10.772793531417847
    },
    {
      "epoch": 7.012939453125e-05,
      "model_forward_time": 1.7116374969482422,
      "step": 73536
    },
    {
      "epoch": 7.012939453125e-05,
      "step": 73536,
      "training_step_time": 9.727768898010254
    },
    {
      "epoch": 7.013034820556641e-05,
      "model_forward_time": 1.7190029621124268,
      "step": 73537
    },
    {
      "epoch": 7.013034820556641e-05,
      "step": 73537,
      "training_step_time": 8.256421327590942
    },
    {
      "epoch": 7.013130187988281e-05,
      "model_forward_time": 1.7055251598358154,
      "step": 73538
    },
    {
      "epoch": 7.013130187988281e-05,
      "step": 73538,
      "training_step_time": 7.692111253738403
    },
    {
      "epoch": 7.013225555419922e-05,
      "model_forward_time": 1.7203090190887451,
      "step": 73539
    },
    {
      "epoch": 7.013225555419922e-05,
      "step": 73539,
      "training_step_time": 7.778872728347778
    },
    {
      "epoch": 7.013320922851563e-05,
      "grad_norm": 0.08342093229293823,
      "learning_rate": 1.7952049007000627e-06,
      "loss": 0.0592,
      "step": 73540
    },
    {
      "epoch": 7.013320922851563e-05,
      "model_forward_time": 1.7176876068115234,
      "step": 73540
    },
    {
      "dynamics_loss_avg": 0.06405065096914768,
      "epoch": 7.013320922851563e-05,
      "step": 73540
    },
    {
      "action_loss_avg": 0.0013889870751881972,
      "epoch": 7.013320922851563e-05,
      "step": 73540
    },
    {
      "epoch": 7.013320922851563e-05,
      "step": 73540,
      "training_step_time": 7.601969480514526
    },
    {
      "epoch": 7.013416290283203e-05,
      "model_forward_time": 1.7167081832885742,
      "step": 73541
    },
    {
      "epoch": 7.013416290283203e-05,
      "step": 73541,
      "training_step_time": 7.683734893798828
    },
    {
      "epoch": 7.013511657714844e-05,
      "model_forward_time": 1.7298431396484375,
      "step": 73542
    },
    {
      "epoch": 7.013511657714844e-05,
      "step": 73542,
      "training_step_time": 8.130915403366089
    },
    {
      "epoch": 7.013607025146485e-05,
      "model_forward_time": 1.7343323230743408,
      "step": 73543
    },
    {
      "epoch": 7.013607025146485e-05,
      "step": 73543,
      "training_step_time": 8.233712911605835
    },
    {
      "epoch": 7.013702392578125e-05,
      "model_forward_time": 2.2456459999084473,
      "step": 73544
    },
    {
      "epoch": 7.013702392578125e-05,
      "step": 73544,
      "training_step_time": 8.707831144332886
    },
    {
      "epoch": 7.013797760009766e-05,
      "model_forward_time": 1.7170164585113525,
      "step": 73545
    },
    {
      "epoch": 7.013797760009766e-05,
      "step": 73545,
      "training_step_time": 8.448252201080322
    },
    {
      "epoch": 7.013893127441407e-05,
      "model_forward_time": 1.7198336124420166,
      "step": 73546
    },
    {
      "epoch": 7.013893127441407e-05,
      "step": 73546,
      "training_step_time": 8.540226221084595
    },
    {
      "epoch": 7.013988494873047e-05,
      "model_forward_time": 1.7256979942321777,
      "step": 73547
    },
    {
      "epoch": 7.013988494873047e-05,
      "step": 73547,
      "training_step_time": 7.68973708152771
    },
    {
      "epoch": 7.014083862304688e-05,
      "model_forward_time": 1.7262423038482666,
      "step": 73548
    },
    {
      "epoch": 7.014083862304688e-05,
      "step": 73548,
      "training_step_time": 7.815212249755859
    },
    {
      "epoch": 7.014179229736329e-05,
      "model_forward_time": 1.7308011054992676,
      "step": 73549
    },
    {
      "epoch": 7.014179229736329e-05,
      "step": 73549,
      "training_step_time": 7.677370548248291
    },
    {
      "epoch": 7.014274597167968e-05,
      "grad_norm": 0.10754332691431046,
      "learning_rate": 1.7939359136975537e-06,
      "loss": 0.0539,
      "step": 73550
    },
    {
      "epoch": 7.014274597167968e-05,
      "model_forward_time": 1.7124061584472656,
      "step": 73550
    },
    {
      "dynamics_loss_avg": 0.05594562077894807,
      "epoch": 7.014274597167968e-05,
      "step": 73550
    },
    {
      "action_loss_avg": 0.001292619583546184,
      "epoch": 7.014274597167968e-05,
      "step": 73550
    },
    {
      "epoch": 7.014274597167968e-05,
      "step": 73550,
      "training_step_time": 7.512719631195068
    },
    {
      "epoch": 7.01436996459961e-05,
      "model_forward_time": 1.7090740203857422,
      "step": 73551
    },
    {
      "epoch": 7.01436996459961e-05,
      "step": 73551,
      "training_step_time": 7.627137899398804
    },
    {
      "epoch": 7.01446533203125e-05,
      "model_forward_time": 1.7200243473052979,
      "step": 73552
    },
    {
      "epoch": 7.01446533203125e-05,
      "step": 73552,
      "training_step_time": 8.652427196502686
    },
    {
      "epoch": 7.01456069946289e-05,
      "model_forward_time": 2.2719085216522217,
      "step": 73553
    },
    {
      "epoch": 7.01456069946289e-05,
      "step": 73553,
      "training_step_time": 8.469457864761353
    },
    {
      "epoch": 7.014656066894531e-05,
      "model_forward_time": 1.7314634323120117,
      "step": 73554
    },
    {
      "epoch": 7.014656066894531e-05,
      "step": 73554,
      "training_step_time": 8.790993452072144
    },
    {
      "epoch": 7.014751434326173e-05,
      "model_forward_time": 1.7278077602386475,
      "step": 73555
    },
    {
      "epoch": 7.014751434326173e-05,
      "step": 73555,
      "training_step_time": 8.73276424407959
    },
    {
      "epoch": 7.014846801757812e-05,
      "model_forward_time": 1.7326998710632324,
      "step": 73556
    },
    {
      "epoch": 7.014846801757812e-05,
      "step": 73556,
      "training_step_time": 8.716946363449097
    },
    {
      "epoch": 7.014942169189453e-05,
      "model_forward_time": 1.7371270656585693,
      "step": 73557
    },
    {
      "epoch": 7.014942169189453e-05,
      "step": 73557,
      "training_step_time": 7.376895904541016
    },
    {
      "epoch": 7.015037536621094e-05,
      "model_forward_time": 1.7262234687805176,
      "step": 73558
    },
    {
      "epoch": 7.015037536621094e-05,
      "step": 73558,
      "training_step_time": 7.8245720863342285
    },
    {
      "epoch": 7.015132904052734e-05,
      "model_forward_time": 1.7341842651367188,
      "step": 73559
    },
    {
      "epoch": 7.015132904052734e-05,
      "step": 73559,
      "training_step_time": 9.754382848739624
    },
    {
      "epoch": 7.015228271484375e-05,
      "grad_norm": 0.09331478923559189,
      "learning_rate": 1.7926672773053933e-06,
      "loss": 0.0534,
      "step": 73560
    },
    {
      "epoch": 7.015228271484375e-05,
      "model_forward_time": 1.7212011814117432,
      "step": 73560
    },
    {
      "dynamics_loss_avg": 0.0424564647488296,
      "epoch": 7.015228271484375e-05,
      "step": 73560
    },
    {
      "action_loss_avg": 0.0010159472207305954,
      "epoch": 7.015228271484375e-05,
      "step": 73560
    },
    {
      "epoch": 7.015228271484375e-05,
      "step": 73560,
      "training_step_time": 7.749883651733398
    },
    {
      "epoch": 7.015323638916016e-05,
      "model_forward_time": 1.7403266429901123,
      "step": 73561
    },
    {
      "epoch": 7.015323638916016e-05,
      "step": 73561,
      "training_step_time": 8.702914237976074
    },
    {
      "epoch": 7.015419006347656e-05,
      "model_forward_time": 2.3422892093658447,
      "step": 73562
    },
    {
      "epoch": 7.015419006347656e-05,
      "step": 73562,
      "training_step_time": 8.74726414680481
    },
    {
      "epoch": 7.015514373779297e-05,
      "model_forward_time": 1.7384450435638428,
      "step": 73563
    },
    {
      "epoch": 7.015514373779297e-05,
      "step": 73563,
      "training_step_time": 9.316856384277344
    },
    {
      "epoch": 7.015609741210937e-05,
      "model_forward_time": 1.7320845127105713,
      "step": 73564
    },
    {
      "epoch": 7.015609741210937e-05,
      "step": 73564,
      "training_step_time": 7.749181747436523
    },
    {
      "epoch": 7.015705108642578e-05,
      "model_forward_time": 1.727128505706787,
      "step": 73565
    },
    {
      "epoch": 7.015705108642578e-05,
      "step": 73565,
      "training_step_time": 9.006701469421387
    },
    {
      "epoch": 7.015800476074219e-05,
      "model_forward_time": 1.724778175354004,
      "step": 73566
    },
    {
      "epoch": 7.015800476074219e-05,
      "step": 73566,
      "training_step_time": 7.705397367477417
    },
    {
      "epoch": 7.015895843505859e-05,
      "model_forward_time": 1.7026591300964355,
      "step": 73567
    },
    {
      "epoch": 7.015895843505859e-05,
      "step": 73567,
      "training_step_time": 7.80617880821228
    },
    {
      "epoch": 7.0159912109375e-05,
      "model_forward_time": 1.7066898345947266,
      "step": 73568
    },
    {
      "epoch": 7.0159912109375e-05,
      "step": 73568,
      "training_step_time": 8.415553092956543
    },
    {
      "epoch": 7.016086578369141e-05,
      "model_forward_time": 1.7071316242218018,
      "step": 73569
    },
    {
      "epoch": 7.016086578369141e-05,
      "step": 73569,
      "training_step_time": 7.626855373382568
    },
    {
      "epoch": 7.01618194580078e-05,
      "grad_norm": 0.3434258699417114,
      "learning_rate": 1.791398991662317e-06,
      "loss": 0.0596,
      "step": 73570
    },
    {
      "epoch": 7.01618194580078e-05,
      "model_forward_time": 1.7195978164672852,
      "step": 73570
    },
    {
      "dynamics_loss_avg": 0.04998506186529994,
      "epoch": 7.01618194580078e-05,
      "step": 73570
    },
    {
      "action_loss_avg": 0.0010955046993331053,
      "epoch": 7.01618194580078e-05,
      "step": 73570
    },
    {
      "epoch": 7.01618194580078e-05,
      "step": 73570,
      "training_step_time": 8.646088361740112
    },
    {
      "epoch": 7.016277313232422e-05,
      "model_forward_time": 2.858509063720703,
      "step": 73571
    },
    {
      "epoch": 7.016277313232422e-05,
      "step": 73571,
      "training_step_time": 7.715949058532715
    },
    {
      "epoch": 7.016372680664063e-05,
      "model_forward_time": 1.3567938804626465,
      "step": 73572
    },
    {
      "epoch": 7.016372680664063e-05,
      "step": 73572,
      "training_step_time": 8.726390361785889
    },
    {
      "epoch": 7.016468048095702e-05,
      "model_forward_time": 1.734339714050293,
      "step": 73573
    },
    {
      "epoch": 7.016468048095702e-05,
      "step": 73573,
      "training_step_time": 9.653706073760986
    },
    {
      "epoch": 7.016563415527344e-05,
      "model_forward_time": 1.7170193195343018,
      "step": 73574
    },
    {
      "epoch": 7.016563415527344e-05,
      "step": 73574,
      "training_step_time": 8.42031979560852
    },
    {
      "epoch": 7.016658782958985e-05,
      "model_forward_time": 1.7180907726287842,
      "step": 73575
    },
    {
      "epoch": 7.016658782958985e-05,
      "step": 73575,
      "training_step_time": 8.779667615890503
    },
    {
      "epoch": 7.016754150390624e-05,
      "model_forward_time": 1.7220733165740967,
      "step": 73576
    },
    {
      "epoch": 7.016754150390624e-05,
      "step": 73576,
      "training_step_time": 7.800732135772705
    },
    {
      "epoch": 7.016849517822265e-05,
      "model_forward_time": 1.7175896167755127,
      "step": 73577
    },
    {
      "epoch": 7.016849517822265e-05,
      "step": 73577,
      "training_step_time": 8.242390394210815
    },
    {
      "epoch": 7.016944885253907e-05,
      "model_forward_time": 1.7363002300262451,
      "step": 73578
    },
    {
      "epoch": 7.016944885253907e-05,
      "step": 73578,
      "training_step_time": 7.545090675354004
    },
    {
      "epoch": 7.017040252685546e-05,
      "model_forward_time": 1.7481837272644043,
      "step": 73579
    },
    {
      "epoch": 7.017040252685546e-05,
      "step": 73579,
      "training_step_time": 8.580217361450195
    },
    {
      "epoch": 7.017135620117187e-05,
      "grad_norm": 0.08314000070095062,
      "learning_rate": 1.7901310569070245e-06,
      "loss": 0.0541,
      "step": 73580
    },
    {
      "epoch": 7.017135620117187e-05,
      "model_forward_time": 2.8821868896484375,
      "step": 73580
    },
    {
      "dynamics_loss_avg": 0.05511084403842688,
      "epoch": 7.017135620117187e-05,
      "step": 73580
    },
    {
      "action_loss_avg": 0.0008241350005846471,
      "epoch": 7.017135620117187e-05,
      "step": 73580
    },
    {
      "epoch": 7.017135620117187e-05,
      "step": 73580,
      "training_step_time": 8.44459581375122
    },
    {
      "epoch": 7.017230987548828e-05,
      "model_forward_time": 1.7255687713623047,
      "step": 73581
    },
    {
      "epoch": 7.017230987548828e-05,
      "step": 73581,
      "training_step_time": 7.92048454284668
    },
    {
      "epoch": 7.017326354980468e-05,
      "model_forward_time": 1.7276496887207031,
      "step": 73582
    },
    {
      "epoch": 7.017326354980468e-05,
      "step": 73582,
      "training_step_time": 8.804694652557373
    },
    {
      "epoch": 7.017421722412109e-05,
      "model_forward_time": 1.7315380573272705,
      "step": 73583
    },
    {
      "epoch": 7.017421722412109e-05,
      "step": 73583,
      "training_step_time": 7.6508824825286865
    },
    {
      "epoch": 7.01751708984375e-05,
      "model_forward_time": 1.7247886657714844,
      "step": 73584
    },
    {
      "epoch": 7.01751708984375e-05,
      "step": 73584,
      "training_step_time": 8.546468019485474
    },
    {
      "epoch": 7.01761245727539e-05,
      "model_forward_time": 1.7265191078186035,
      "step": 73585
    },
    {
      "epoch": 7.01761245727539e-05,
      "step": 73585,
      "training_step_time": 7.783125400543213
    },
    {
      "epoch": 7.017707824707031e-05,
      "model_forward_time": 1.7185485363006592,
      "step": 73586
    },
    {
      "epoch": 7.017707824707031e-05,
      "step": 73586,
      "training_step_time": 7.584024667739868
    },
    {
      "epoch": 7.017803192138672e-05,
      "model_forward_time": 1.7170522212982178,
      "step": 73587
    },
    {
      "epoch": 7.017803192138672e-05,
      "step": 73587,
      "training_step_time": 7.67794132232666
    },
    {
      "epoch": 7.017898559570312e-05,
      "model_forward_time": 1.8351030349731445,
      "step": 73588
    },
    {
      "epoch": 7.017898559570312e-05,
      "step": 73588,
      "training_step_time": 7.71730899810791
    },
    {
      "epoch": 7.017993927001953e-05,
      "model_forward_time": 2.8135695457458496,
      "step": 73589
    },
    {
      "epoch": 7.017993927001953e-05,
      "step": 73589,
      "training_step_time": 7.745930194854736
    },
    {
      "epoch": 7.018089294433594e-05,
      "grad_norm": 0.07585379481315613,
      "learning_rate": 1.7888634731781724e-06,
      "loss": 0.0579,
      "step": 73590
    },
    {
      "epoch": 7.018089294433594e-05,
      "model_forward_time": 1.7129135131835938,
      "step": 73590
    },
    {
      "dynamics_loss_avg": 0.046935173869133,
      "epoch": 7.018089294433594e-05,
      "step": 73590
    },
    {
      "action_loss_avg": 0.0007848128952900879,
      "epoch": 7.018089294433594e-05,
      "step": 73590
    },
    {
      "epoch": 7.018089294433594e-05,
      "step": 73590,
      "training_step_time": 7.974686145782471
    },
    {
      "epoch": 7.018184661865234e-05,
      "model_forward_time": 1.7216808795928955,
      "step": 73591
    },
    {
      "epoch": 7.018184661865234e-05,
      "step": 73591,
      "training_step_time": 9.40310287475586
    },
    {
      "epoch": 7.018280029296875e-05,
      "model_forward_time": 1.7239153385162354,
      "step": 73592
    },
    {
      "epoch": 7.018280029296875e-05,
      "step": 73592,
      "training_step_time": 7.7438414096832275
    },
    {
      "epoch": 7.018375396728516e-05,
      "model_forward_time": 1.7411186695098877,
      "step": 73593
    },
    {
      "epoch": 7.018375396728516e-05,
      "step": 73593,
      "training_step_time": 7.679380416870117
    },
    {
      "epoch": 7.018470764160156e-05,
      "model_forward_time": 1.7433698177337646,
      "step": 73594
    },
    {
      "epoch": 7.018470764160156e-05,
      "step": 73594,
      "training_step_time": 7.387815952301025
    },
    {
      "epoch": 7.018566131591797e-05,
      "model_forward_time": 1.751631498336792,
      "step": 73595
    },
    {
      "epoch": 7.018566131591797e-05,
      "step": 73595,
      "training_step_time": 8.57396149635315
    },
    {
      "epoch": 7.018661499023438e-05,
      "model_forward_time": 1.7446167469024658,
      "step": 73596
    },
    {
      "epoch": 7.018661499023438e-05,
      "step": 73596,
      "training_step_time": 7.644227027893066
    },
    {
      "epoch": 7.018756866455078e-05,
      "model_forward_time": 1.8029451370239258,
      "step": 73597
    },
    {
      "epoch": 7.018756866455078e-05,
      "step": 73597,
      "training_step_time": 9.371549129486084
    },
    {
      "epoch": 7.018852233886719e-05,
      "model_forward_time": 2.8708760738372803,
      "step": 73598
    },
    {
      "epoch": 7.018852233886719e-05,
      "step": 73598,
      "training_step_time": 8.7367525100708
    },
    {
      "epoch": 7.01894760131836e-05,
      "model_forward_time": 1.747683048248291,
      "step": 73599
    },
    {
      "epoch": 7.01894760131836e-05,
      "step": 73599,
      "training_step_time": 10.421303510665894
    },
    {
      "epoch": 7.01904296875e-05,
      "grad_norm": 0.10199659317731857,
      "learning_rate": 1.7875962406143843e-06,
      "loss": 0.058,
      "step": 73600
    },
    {
      "epoch": 7.01904296875e-05,
      "model_forward_time": 1.3585588932037354,
      "step": 73600
    },
    {
      "dynamics_loss_avg": 0.04042428620159626,
      "epoch": 7.01904296875e-05,
      "step": 73600
    },
    {
      "action_loss_avg": 0.0008809388265945018,
      "epoch": 7.01904296875e-05,
      "step": 73600
    },
    {
      "epoch": 7.01904296875e-05,
      "step": 73600,
      "training_step_time": 8.334691524505615
    },
    {
      "epoch": 7.01913833618164e-05,
      "model_forward_time": 1.7343113422393799,
      "step": 73601
    },
    {
      "epoch": 7.01913833618164e-05,
      "step": 73601,
      "training_step_time": 7.783041000366211
    },
    {
      "epoch": 7.019233703613282e-05,
      "model_forward_time": 1.7486844062805176,
      "step": 73602
    },
    {
      "epoch": 7.019233703613282e-05,
      "step": 73602,
      "training_step_time": 7.672456502914429
    },
    {
      "epoch": 7.019329071044921e-05,
      "model_forward_time": 1.7476866245269775,
      "step": 73603
    },
    {
      "epoch": 7.019329071044921e-05,
      "step": 73603,
      "training_step_time": 7.79982852935791
    },
    {
      "epoch": 7.019424438476562e-05,
      "model_forward_time": 1.746269702911377,
      "step": 73604
    },
    {
      "epoch": 7.019424438476562e-05,
      "step": 73604,
      "training_step_time": 7.8481361865997314
    },
    {
      "epoch": 7.019519805908204e-05,
      "model_forward_time": 1.7492003440856934,
      "step": 73605
    },
    {
      "epoch": 7.019519805908204e-05,
      "step": 73605,
      "training_step_time": 7.713651418685913
    },
    {
      "epoch": 7.019615173339843e-05,
      "model_forward_time": 1.7806882858276367,
      "step": 73606
    },
    {
      "epoch": 7.019615173339843e-05,
      "step": 73606,
      "training_step_time": 7.647320032119751
    },
    {
      "epoch": 7.019710540771484e-05,
      "model_forward_time": 1.7534339427947998,
      "step": 73607
    },
    {
      "epoch": 7.019710540771484e-05,
      "step": 73607,
      "training_step_time": 8.73590350151062
    },
    {
      "epoch": 7.019805908203125e-05,
      "model_forward_time": 1.7390761375427246,
      "step": 73608
    },
    {
      "epoch": 7.019805908203125e-05,
      "step": 73608,
      "training_step_time": 8.393479585647583
    },
    {
      "epoch": 7.019901275634765e-05,
      "model_forward_time": 1.7422816753387451,
      "step": 73609
    },
    {
      "epoch": 7.019901275634765e-05,
      "step": 73609,
      "training_step_time": 8.655352354049683
    },
    {
      "epoch": 7.019996643066406e-05,
      "grad_norm": 0.07690032571554184,
      "learning_rate": 1.7863293593542402e-06,
      "loss": 0.0588,
      "step": 73610
    },
    {
      "epoch": 7.019996643066406e-05,
      "model_forward_time": 1.72515869140625,
      "step": 73610
    },
    {
      "dynamics_loss_avg": 0.06282221963629127,
      "epoch": 7.019996643066406e-05,
      "step": 73610
    },
    {
      "action_loss_avg": 0.0012838271621149033,
      "epoch": 7.019996643066406e-05,
      "step": 73610
    },
    {
      "epoch": 7.019996643066406e-05,
      "step": 73610,
      "training_step_time": 7.778640031814575
    },
    {
      "epoch": 7.020092010498047e-05,
      "model_forward_time": 1.7429871559143066,
      "step": 73611
    },
    {
      "epoch": 7.020092010498047e-05,
      "step": 73611,
      "training_step_time": 7.516674041748047
    },
    {
      "epoch": 7.020187377929687e-05,
      "model_forward_time": 1.372502088546753,
      "step": 73612
    },
    {
      "epoch": 7.020187377929687e-05,
      "step": 73612,
      "training_step_time": 7.843141317367554
    },
    {
      "epoch": 7.020282745361328e-05,
      "model_forward_time": 1.7426259517669678,
      "step": 73613
    },
    {
      "epoch": 7.020282745361328e-05,
      "step": 73613,
      "training_step_time": 7.888330459594727
    },
    {
      "epoch": 7.020378112792969e-05,
      "model_forward_time": 1.7343997955322266,
      "step": 73614
    },
    {
      "epoch": 7.020378112792969e-05,
      "step": 73614,
      "training_step_time": 7.785380601882935
    },
    {
      "epoch": 7.020473480224609e-05,
      "model_forward_time": 1.755338191986084,
      "step": 73615
    },
    {
      "epoch": 7.020473480224609e-05,
      "step": 73615,
      "training_step_time": 8.20014476776123
    },
    {
      "epoch": 7.02056884765625e-05,
      "model_forward_time": 1.7470061779022217,
      "step": 73616
    },
    {
      "epoch": 7.02056884765625e-05,
      "step": 73616,
      "training_step_time": 8.569052934646606
    },
    {
      "epoch": 7.020664215087891e-05,
      "model_forward_time": 2.935530662536621,
      "step": 73617
    },
    {
      "epoch": 7.020664215087891e-05,
      "step": 73617,
      "training_step_time": 7.923565149307251
    },
    {
      "epoch": 7.020759582519531e-05,
      "model_forward_time": 1.745487928390503,
      "step": 73618
    },
    {
      "epoch": 7.020759582519531e-05,
      "step": 73618,
      "training_step_time": 9.08681058883667
    },
    {
      "epoch": 7.020854949951172e-05,
      "model_forward_time": 1.7333335876464844,
      "step": 73619
    },
    {
      "epoch": 7.020854949951172e-05,
      "step": 73619,
      "training_step_time": 7.884605169296265
    },
    {
      "epoch": 7.020950317382813e-05,
      "grad_norm": 0.08898506313562393,
      "learning_rate": 1.7850628295362854e-06,
      "loss": 0.0564,
      "step": 73620
    },
    {
      "epoch": 7.020950317382813e-05,
      "model_forward_time": 1.7290656566619873,
      "step": 73620
    },
    {
      "dynamics_loss_avg": 0.035915364231914285,
      "epoch": 7.020950317382813e-05,
      "step": 73620
    },
    {
      "action_loss_avg": 0.0011117398491478526,
      "epoch": 7.020950317382813e-05,
      "step": 73620
    },
    {
      "epoch": 7.020950317382813e-05,
      "step": 73620,
      "training_step_time": 7.628422498703003
    },
    {
      "epoch": 7.021045684814453e-05,
      "model_forward_time": 1.7441916465759277,
      "step": 73621
    },
    {
      "epoch": 7.021045684814453e-05,
      "step": 73621,
      "training_step_time": 7.888434648513794
    },
    {
      "epoch": 7.021141052246094e-05,
      "model_forward_time": 1.7474822998046875,
      "step": 73622
    },
    {
      "epoch": 7.021141052246094e-05,
      "step": 73622,
      "training_step_time": 7.965409517288208
    },
    {
      "epoch": 7.021236419677735e-05,
      "model_forward_time": 1.745147943496704,
      "step": 73623
    },
    {
      "epoch": 7.021236419677735e-05,
      "step": 73623,
      "training_step_time": 8.71142578125
    },
    {
      "epoch": 7.021331787109375e-05,
      "model_forward_time": 1.7419989109039307,
      "step": 73624
    },
    {
      "epoch": 7.021331787109375e-05,
      "step": 73624,
      "training_step_time": 7.891737937927246
    },
    {
      "epoch": 7.021427154541016e-05,
      "model_forward_time": 1.8641302585601807,
      "step": 73625
    },
    {
      "epoch": 7.021427154541016e-05,
      "step": 73625,
      "training_step_time": 7.811603307723999
    },
    {
      "epoch": 7.021522521972657e-05,
      "model_forward_time": 2.9142491817474365,
      "step": 73626
    },
    {
      "epoch": 7.021522521972657e-05,
      "step": 73626,
      "training_step_time": 8.867746829986572
    },
    {
      "epoch": 7.021617889404296e-05,
      "model_forward_time": 1.7465074062347412,
      "step": 73627
    },
    {
      "epoch": 7.021617889404296e-05,
      "step": 73627,
      "training_step_time": 10.687463521957397
    },
    {
      "epoch": 7.021713256835938e-05,
      "model_forward_time": 1.741157054901123,
      "step": 73628
    },
    {
      "epoch": 7.021713256835938e-05,
      "step": 73628,
      "training_step_time": 10.302451610565186
    },
    {
      "epoch": 7.021808624267579e-05,
      "model_forward_time": 1.7393817901611328,
      "step": 73629
    },
    {
      "epoch": 7.021808624267579e-05,
      "step": 73629,
      "training_step_time": 7.836353302001953
    },
    {
      "epoch": 7.021903991699218e-05,
      "grad_norm": 0.08540794998407364,
      "learning_rate": 1.7837966512990261e-06,
      "loss": 0.0564,
      "step": 73630
    },
    {
      "epoch": 7.021903991699218e-05,
      "model_forward_time": 1.7274341583251953,
      "step": 73630
    },
    {
      "dynamics_loss_avg": 0.05655696783214807,
      "epoch": 7.021903991699218e-05,
      "step": 73630
    },
    {
      "action_loss_avg": 0.002313010120997205,
      "epoch": 7.021903991699218e-05,
      "step": 73630
    },
    {
      "epoch": 7.021903991699218e-05,
      "step": 73630,
      "training_step_time": 7.722501516342163
    },
    {
      "epoch": 7.02199935913086e-05,
      "model_forward_time": 1.7474114894866943,
      "step": 73631
    },
    {
      "epoch": 7.02199935913086e-05,
      "step": 73631,
      "training_step_time": 7.931427001953125
    },
    {
      "epoch": 7.0220947265625e-05,
      "model_forward_time": 1.7461261749267578,
      "step": 73632
    },
    {
      "epoch": 7.0220947265625e-05,
      "step": 73632,
      "training_step_time": 7.743157148361206
    },
    {
      "epoch": 7.02219009399414e-05,
      "model_forward_time": 1.7403349876403809,
      "step": 73633
    },
    {
      "epoch": 7.02219009399414e-05,
      "step": 73633,
      "training_step_time": 7.7367236614227295
    },
    {
      "epoch": 7.022285461425781e-05,
      "model_forward_time": 1.8627026081085205,
      "step": 73634
    },
    {
      "epoch": 7.022285461425781e-05,
      "step": 73634,
      "training_step_time": 8.359463214874268
    },
    {
      "epoch": 7.022380828857422e-05,
      "model_forward_time": 2.8592300415039062,
      "step": 73635
    },
    {
      "epoch": 7.022380828857422e-05,
      "step": 73635,
      "training_step_time": 7.787342309951782
    },
    {
      "epoch": 7.022476196289062e-05,
      "model_forward_time": 1.7264032363891602,
      "step": 73636
    },
    {
      "epoch": 7.022476196289062e-05,
      "step": 73636,
      "training_step_time": 8.556065082550049
    },
    {
      "epoch": 7.022571563720703e-05,
      "model_forward_time": 1.7583305835723877,
      "step": 73637
    },
    {
      "epoch": 7.022571563720703e-05,
      "step": 73637,
      "training_step_time": 8.726130723953247
    },
    {
      "epoch": 7.022666931152344e-05,
      "model_forward_time": 1.3895654678344727,
      "step": 73638
    },
    {
      "epoch": 7.022666931152344e-05,
      "step": 73638,
      "training_step_time": 9.493096828460693
    },
    {
      "epoch": 7.022762298583984e-05,
      "model_forward_time": 1.7434768676757812,
      "step": 73639
    },
    {
      "epoch": 7.022762298583984e-05,
      "step": 73639,
      "training_step_time": 7.697933197021484
    },
    {
      "epoch": 7.022857666015625e-05,
      "grad_norm": 0.08818075805902481,
      "learning_rate": 1.7825308247809308e-06,
      "loss": 0.0577,
      "step": 73640
    },
    {
      "epoch": 7.022857666015625e-05,
      "model_forward_time": 1.7226293087005615,
      "step": 73640
    },
    {
      "dynamics_loss_avg": 0.0641168924048543,
      "epoch": 7.022857666015625e-05,
      "step": 73640
    },
    {
      "action_loss_avg": 0.0012485793442465365,
      "epoch": 7.022857666015625e-05,
      "step": 73640
    },
    {
      "epoch": 7.022857666015625e-05,
      "step": 73640,
      "training_step_time": 7.504175186157227
    },
    {
      "epoch": 7.022953033447266e-05,
      "model_forward_time": 1.7433228492736816,
      "step": 73641
    },
    {
      "epoch": 7.022953033447266e-05,
      "step": 73641,
      "training_step_time": 8.597753286361694
    },
    {
      "epoch": 7.023048400878906e-05,
      "model_forward_time": 1.7345798015594482,
      "step": 73642
    },
    {
      "epoch": 7.023048400878906e-05,
      "step": 73642,
      "training_step_time": 7.723542928695679
    },
    {
      "epoch": 7.023143768310547e-05,
      "model_forward_time": 1.833785057067871,
      "step": 73643
    },
    {
      "epoch": 7.023143768310547e-05,
      "step": 73643,
      "training_step_time": 7.722488641738892
    },
    {
      "epoch": 7.023239135742188e-05,
      "model_forward_time": 2.8368420600891113,
      "step": 73644
    },
    {
      "epoch": 7.023239135742188e-05,
      "step": 73644,
      "training_step_time": 8.71859097480774
    },
    {
      "epoch": 7.023334503173828e-05,
      "model_forward_time": 1.741649866104126,
      "step": 73645
    },
    {
      "epoch": 7.023334503173828e-05,
      "step": 73645,
      "training_step_time": 8.715574026107788
    },
    {
      "epoch": 7.023429870605469e-05,
      "model_forward_time": 1.744682788848877,
      "step": 73646
    },
    {
      "epoch": 7.023429870605469e-05,
      "step": 73646,
      "training_step_time": 7.909710884094238
    },
    {
      "epoch": 7.02352523803711e-05,
      "model_forward_time": 1.7425458431243896,
      "step": 73647
    },
    {
      "epoch": 7.02352523803711e-05,
      "step": 73647,
      "training_step_time": 7.7713303565979
    },
    {
      "epoch": 7.02362060546875e-05,
      "model_forward_time": 1.744377851486206,
      "step": 73648
    },
    {
      "epoch": 7.02362060546875e-05,
      "step": 73648,
      "training_step_time": 8.11624002456665
    },
    {
      "epoch": 7.023715972900391e-05,
      "model_forward_time": 1.742832899093628,
      "step": 73649
    },
    {
      "epoch": 7.023715972900391e-05,
      "step": 73649,
      "training_step_time": 7.751353979110718
    },
    {
      "epoch": 7.023811340332032e-05,
      "grad_norm": 0.08472229540348053,
      "learning_rate": 1.7812653501204264e-06,
      "loss": 0.0582,
      "step": 73650
    },
    {
      "epoch": 7.023811340332032e-05,
      "model_forward_time": 1.7250032424926758,
      "step": 73650
    },
    {
      "dynamics_loss_avg": 0.054153294302523136,
      "epoch": 7.023811340332032e-05,
      "step": 73650
    },
    {
      "action_loss_avg": 0.0007555763440905139,
      "epoch": 7.023811340332032e-05,
      "step": 73650
    },
    {
      "epoch": 7.023811340332032e-05,
      "step": 73650,
      "training_step_time": 7.861637830734253
    },
    {
      "epoch": 7.023906707763672e-05,
      "model_forward_time": 1.7427027225494385,
      "step": 73651
    },
    {
      "epoch": 7.023906707763672e-05,
      "step": 73651,
      "training_step_time": 8.370561599731445
    },
    {
      "epoch": 7.024002075195313e-05,
      "model_forward_time": 1.7971415519714355,
      "step": 73652
    },
    {
      "epoch": 7.024002075195313e-05,
      "step": 73652,
      "training_step_time": 7.662527084350586
    },
    {
      "epoch": 7.024097442626954e-05,
      "model_forward_time": 1.741898536682129,
      "step": 73653
    },
    {
      "epoch": 7.024097442626954e-05,
      "step": 73653,
      "training_step_time": 8.40221643447876
    },
    {
      "epoch": 7.024192810058593e-05,
      "model_forward_time": 2.850398302078247,
      "step": 73654
    },
    {
      "epoch": 7.024192810058593e-05,
      "step": 73654,
      "training_step_time": 8.488216876983643
    },
    {
      "epoch": 7.024288177490234e-05,
      "model_forward_time": 1.7398557662963867,
      "step": 73655
    },
    {
      "epoch": 7.024288177490234e-05,
      "step": 73655,
      "training_step_time": 8.40871262550354
    },
    {
      "epoch": 7.024383544921876e-05,
      "model_forward_time": 1.7413256168365479,
      "step": 73656
    },
    {
      "epoch": 7.024383544921876e-05,
      "step": 73656,
      "training_step_time": 7.808757543563843
    },
    {
      "epoch": 7.024478912353515e-05,
      "model_forward_time": 1.7293682098388672,
      "step": 73657
    },
    {
      "epoch": 7.024478912353515e-05,
      "step": 73657,
      "training_step_time": 7.572384357452393
    },
    {
      "epoch": 7.024574279785156e-05,
      "model_forward_time": 1.7449135780334473,
      "step": 73658
    },
    {
      "epoch": 7.024574279785156e-05,
      "step": 73658,
      "training_step_time": 7.720054864883423
    },
    {
      "epoch": 7.024669647216797e-05,
      "model_forward_time": 1.7403178215026855,
      "step": 73659
    },
    {
      "epoch": 7.024669647216797e-05,
      "step": 73659,
      "training_step_time": 7.8560755252838135
    },
    {
      "epoch": 7.024765014648437e-05,
      "grad_norm": 0.08553773909807205,
      "learning_rate": 1.780000227455903e-06,
      "loss": 0.056,
      "step": 73660
    },
    {
      "epoch": 7.024765014648437e-05,
      "model_forward_time": 1.7261009216308594,
      "step": 73660
    },
    {
      "dynamics_loss_avg": 0.0585188053548336,
      "epoch": 7.024765014648437e-05,
      "step": 73660
    },
    {
      "action_loss_avg": 0.0010633331577992067,
      "epoch": 7.024765014648437e-05,
      "step": 73660
    },
    {
      "epoch": 7.024765014648437e-05,
      "step": 73660,
      "training_step_time": 7.6442601680755615
    },
    {
      "epoch": 7.024860382080078e-05,
      "model_forward_time": 1.765976905822754,
      "step": 73661
    },
    {
      "epoch": 7.024860382080078e-05,
      "step": 73661,
      "training_step_time": 7.655909299850464
    },
    {
      "epoch": 7.02495574951172e-05,
      "model_forward_time": 1.747499942779541,
      "step": 73662
    },
    {
      "epoch": 7.02495574951172e-05,
      "step": 73662,
      "training_step_time": 7.649757623672485
    },
    {
      "epoch": 7.025051116943359e-05,
      "model_forward_time": 2.8181207180023193,
      "step": 73663
    },
    {
      "epoch": 7.025051116943359e-05,
      "step": 73663,
      "training_step_time": 7.726751327514648
    },
    {
      "epoch": 7.025146484375e-05,
      "model_forward_time": 1.7458767890930176,
      "step": 73664
    },
    {
      "epoch": 7.025146484375e-05,
      "step": 73664,
      "training_step_time": 7.723788738250732
    },
    {
      "epoch": 7.025241851806641e-05,
      "model_forward_time": 1.7370941638946533,
      "step": 73665
    },
    {
      "epoch": 7.025241851806641e-05,
      "step": 73665,
      "training_step_time": 7.8565592765808105
    },
    {
      "epoch": 7.025337219238281e-05,
      "model_forward_time": 1.733769416809082,
      "step": 73666
    },
    {
      "epoch": 7.025337219238281e-05,
      "step": 73666,
      "training_step_time": 7.708444833755493
    },
    {
      "epoch": 7.025432586669922e-05,
      "model_forward_time": 1.746467113494873,
      "step": 73667
    },
    {
      "epoch": 7.025432586669922e-05,
      "step": 73667,
      "training_step_time": 7.734434127807617
    },
    {
      "epoch": 7.025527954101563e-05,
      "model_forward_time": 1.7440321445465088,
      "step": 73668
    },
    {
      "epoch": 7.025527954101563e-05,
      "step": 73668,
      "training_step_time": 7.835617542266846
    },
    {
      "epoch": 7.025623321533203e-05,
      "model_forward_time": 1.7288744449615479,
      "step": 73669
    },
    {
      "epoch": 7.025623321533203e-05,
      "step": 73669,
      "training_step_time": 7.71978497505188
    },
    {
      "epoch": 7.025718688964844e-05,
      "grad_norm": 0.0995432510972023,
      "learning_rate": 1.7787354569257158e-06,
      "loss": 0.0586,
      "step": 73670
    },
    {
      "epoch": 7.025718688964844e-05,
      "model_forward_time": 1.7263994216918945,
      "step": 73670
    },
    {
      "dynamics_loss_avg": 0.06264709532260895,
      "epoch": 7.025718688964844e-05,
      "step": 73670
    },
    {
      "action_loss_avg": 0.0009545765955408569,
      "epoch": 7.025718688964844e-05,
      "step": 73670
    },
    {
      "epoch": 7.025718688964844e-05,
      "step": 73670,
      "training_step_time": 8.759952783584595
    },
    {
      "epoch": 7.025814056396485e-05,
      "model_forward_time": 1.7456867694854736,
      "step": 73671
    },
    {
      "epoch": 7.025814056396485e-05,
      "step": 73671,
      "training_step_time": 8.49699330329895
    },
    {
      "epoch": 7.025909423828125e-05,
      "model_forward_time": 2.8991150856018066,
      "step": 73672
    },
    {
      "epoch": 7.025909423828125e-05,
      "step": 73672,
      "training_step_time": 8.769630432128906
    },
    {
      "epoch": 7.026004791259766e-05,
      "model_forward_time": 1.733677864074707,
      "step": 73673
    },
    {
      "epoch": 7.026004791259766e-05,
      "step": 73673,
      "training_step_time": 7.7065277099609375
    },
    {
      "epoch": 7.026100158691407e-05,
      "model_forward_time": 1.7396388053894043,
      "step": 73674
    },
    {
      "epoch": 7.026100158691407e-05,
      "step": 73674,
      "training_step_time": 7.846813917160034
    },
    {
      "epoch": 7.026195526123047e-05,
      "model_forward_time": 1.3723022937774658,
      "step": 73675
    },
    {
      "epoch": 7.026195526123047e-05,
      "step": 73675,
      "training_step_time": 7.740213394165039
    },
    {
      "epoch": 7.026290893554688e-05,
      "model_forward_time": 1.7384426593780518,
      "step": 73676
    },
    {
      "epoch": 7.026290893554688e-05,
      "step": 73676,
      "training_step_time": 7.764897346496582
    },
    {
      "epoch": 7.026386260986329e-05,
      "model_forward_time": 1.7377378940582275,
      "step": 73677
    },
    {
      "epoch": 7.026386260986329e-05,
      "step": 73677,
      "training_step_time": 8.334062814712524
    },
    {
      "epoch": 7.026481628417968e-05,
      "model_forward_time": 1.7351441383361816,
      "step": 73678
    },
    {
      "epoch": 7.026481628417968e-05,
      "step": 73678,
      "training_step_time": 7.642377853393555
    },
    {
      "epoch": 7.02657699584961e-05,
      "model_forward_time": 1.7405991554260254,
      "step": 73679
    },
    {
      "epoch": 7.02657699584961e-05,
      "step": 73679,
      "training_step_time": 7.666273832321167
    },
    {
      "epoch": 7.02667236328125e-05,
      "grad_norm": 0.10838475823402405,
      "learning_rate": 1.7774710386681738e-06,
      "loss": 0.0538,
      "step": 73680
    },
    {
      "epoch": 7.02667236328125e-05,
      "model_forward_time": 1.8494606018066406,
      "step": 73680
    },
    {
      "dynamics_loss_avg": 0.06404632441699505,
      "epoch": 7.02667236328125e-05,
      "step": 73680
    },
    {
      "action_loss_avg": 0.0016510676505276933,
      "epoch": 7.02667236328125e-05,
      "step": 73680
    },
    {
      "epoch": 7.02667236328125e-05,
      "step": 73680,
      "training_step_time": 8.674494981765747
    },
    {
      "epoch": 7.02676773071289e-05,
      "model_forward_time": 2.905440092086792,
      "step": 73681
    },
    {
      "epoch": 7.02676773071289e-05,
      "step": 73681,
      "training_step_time": 7.866974353790283
    },
    {
      "epoch": 7.026863098144531e-05,
      "model_forward_time": 1.7361090183258057,
      "step": 73682
    },
    {
      "epoch": 7.026863098144531e-05,
      "step": 73682,
      "training_step_time": 9.815505027770996
    },
    {
      "epoch": 7.026958465576173e-05,
      "model_forward_time": 1.7376089096069336,
      "step": 73683
    },
    {
      "epoch": 7.026958465576173e-05,
      "step": 73683,
      "training_step_time": 9.562312126159668
    },
    {
      "epoch": 7.027053833007812e-05,
      "model_forward_time": 1.7047019004821777,
      "step": 73684
    },
    {
      "epoch": 7.027053833007812e-05,
      "step": 73684,
      "training_step_time": 7.704448699951172
    },
    {
      "epoch": 7.027149200439453e-05,
      "model_forward_time": 1.7326056957244873,
      "step": 73685
    },
    {
      "epoch": 7.027149200439453e-05,
      "step": 73685,
      "training_step_time": 8.257978200912476
    },
    {
      "epoch": 7.027244567871094e-05,
      "model_forward_time": 1.7258992195129395,
      "step": 73686
    },
    {
      "epoch": 7.027244567871094e-05,
      "step": 73686,
      "training_step_time": 7.466091156005859
    },
    {
      "epoch": 7.027339935302734e-05,
      "model_forward_time": 1.7286524772644043,
      "step": 73687
    },
    {
      "epoch": 7.027339935302734e-05,
      "step": 73687,
      "training_step_time": 8.828458786010742
    },
    {
      "epoch": 7.027435302734375e-05,
      "model_forward_time": 1.7330784797668457,
      "step": 73688
    },
    {
      "epoch": 7.027435302734375e-05,
      "step": 73688,
      "training_step_time": 8.300647020339966
    },
    {
      "epoch": 7.027530670166015e-05,
      "model_forward_time": 1.8519175052642822,
      "step": 73689
    },
    {
      "epoch": 7.027530670166015e-05,
      "step": 73689,
      "training_step_time": 7.613964796066284
    },
    {
      "epoch": 7.027626037597656e-05,
      "grad_norm": 0.10902971774339676,
      "learning_rate": 1.7762069728215547e-06,
      "loss": 0.0524,
      "step": 73690
    },
    {
      "epoch": 7.027626037597656e-05,
      "model_forward_time": 2.8757119178771973,
      "step": 73690
    },
    {
      "dynamics_loss_avg": 0.048832297697663306,
      "epoch": 7.027626037597656e-05,
      "step": 73690
    },
    {
      "action_loss_avg": 0.0012676680053118616,
      "epoch": 7.027626037597656e-05,
      "step": 73690
    },
    {
      "epoch": 7.027626037597656e-05,
      "step": 73690,
      "training_step_time": 8.31613039970398
    },
    {
      "epoch": 7.027721405029297e-05,
      "model_forward_time": 1.7233309745788574,
      "step": 73691
    },
    {
      "epoch": 7.027721405029297e-05,
      "step": 73691,
      "training_step_time": 8.359256982803345
    },
    {
      "epoch": 7.027816772460937e-05,
      "model_forward_time": 1.7313315868377686,
      "step": 73692
    },
    {
      "epoch": 7.027816772460937e-05,
      "step": 73692,
      "training_step_time": 7.655163049697876
    },
    {
      "epoch": 7.027912139892578e-05,
      "model_forward_time": 1.7316553592681885,
      "step": 73693
    },
    {
      "epoch": 7.027912139892578e-05,
      "step": 73693,
      "training_step_time": 9.792369365692139
    },
    {
      "epoch": 7.028007507324219e-05,
      "model_forward_time": 1.7385170459747314,
      "step": 73694
    },
    {
      "epoch": 7.028007507324219e-05,
      "step": 73694,
      "training_step_time": 8.039592981338501
    },
    {
      "epoch": 7.028102874755859e-05,
      "model_forward_time": 1.7421469688415527,
      "step": 73695
    },
    {
      "epoch": 7.028102874755859e-05,
      "step": 73695,
      "training_step_time": 7.577431678771973
    },
    {
      "epoch": 7.0281982421875e-05,
      "model_forward_time": 1.7378594875335693,
      "step": 73696
    },
    {
      "epoch": 7.0281982421875e-05,
      "step": 73696,
      "training_step_time": 7.416443109512329
    },
    {
      "epoch": 7.028293609619141e-05,
      "model_forward_time": 1.7383852005004883,
      "step": 73697
    },
    {
      "epoch": 7.028293609619141e-05,
      "step": 73697,
      "training_step_time": 7.529393196105957
    },
    {
      "epoch": 7.02838897705078e-05,
      "model_forward_time": 1.839503526687622,
      "step": 73698
    },
    {
      "epoch": 7.02838897705078e-05,
      "step": 73698,
      "training_step_time": 7.593594312667847
    },
    {
      "epoch": 7.028484344482422e-05,
      "model_forward_time": 2.885127067565918,
      "step": 73699
    },
    {
      "epoch": 7.028484344482422e-05,
      "step": 73699,
      "training_step_time": 8.394709825515747
    },
    {
      "epoch": 7.028579711914063e-05,
      "grad_norm": 0.09497787803411484,
      "learning_rate": 1.7749432595240934e-06,
      "loss": 0.0584,
      "step": 73700
    },
    {
      "epoch": 7.028579711914063e-05,
      "model_forward_time": 1.7311944961547852,
      "step": 73700
    },
    {
      "dynamics_loss_avg": 0.06032837461680174,
      "epoch": 7.028579711914063e-05,
      "step": 73700
    },
    {
      "action_loss_avg": 0.0017314155760686845,
      "epoch": 7.028579711914063e-05,
      "step": 73700
    },
    {
      "epoch": 7.028579711914063e-05,
      "step": 73700,
      "training_step_time": 7.66445255279541
    },
    {
      "epoch": 7.028675079345702e-05,
      "model_forward_time": 1.7371718883514404,
      "step": 73701
    },
    {
      "epoch": 7.028675079345702e-05,
      "step": 73701,
      "training_step_time": 7.726873159408569
    },
    {
      "epoch": 7.028770446777344e-05,
      "model_forward_time": 1.7266476154327393,
      "step": 73702
    },
    {
      "epoch": 7.028770446777344e-05,
      "step": 73702,
      "training_step_time": 8.900123119354248
    },
    {
      "epoch": 7.028865814208985e-05,
      "model_forward_time": 1.727604866027832,
      "step": 73703
    },
    {
      "epoch": 7.028865814208985e-05,
      "step": 73703,
      "training_step_time": 7.545965671539307
    },
    {
      "epoch": 7.028961181640624e-05,
      "model_forward_time": 1.7260117530822754,
      "step": 73704
    },
    {
      "epoch": 7.028961181640624e-05,
      "step": 73704,
      "training_step_time": 7.768102169036865
    },
    {
      "epoch": 7.029056549072265e-05,
      "model_forward_time": 1.7291662693023682,
      "step": 73705
    },
    {
      "epoch": 7.029056549072265e-05,
      "step": 73705,
      "training_step_time": 7.7410569190979
    },
    {
      "epoch": 7.029151916503907e-05,
      "model_forward_time": 1.7430675029754639,
      "step": 73706
    },
    {
      "epoch": 7.029151916503907e-05,
      "step": 73706,
      "training_step_time": 7.667033672332764
    },
    {
      "epoch": 7.029247283935546e-05,
      "model_forward_time": 1.7554965019226074,
      "step": 73707
    },
    {
      "epoch": 7.029247283935546e-05,
      "step": 73707,
      "training_step_time": 8.52993655204773
    },
    {
      "epoch": 7.029342651367187e-05,
      "model_forward_time": 1.7450408935546875,
      "step": 73708
    },
    {
      "epoch": 7.029342651367187e-05,
      "step": 73708,
      "training_step_time": 9.471086502075195
    },
    {
      "epoch": 7.029438018798828e-05,
      "model_forward_time": 1.7479517459869385,
      "step": 73709
    },
    {
      "epoch": 7.029438018798828e-05,
      "step": 73709,
      "training_step_time": 7.7025158405303955
    },
    {
      "epoch": 7.029533386230468e-05,
      "grad_norm": 0.13045881688594818,
      "learning_rate": 1.7736798989139909e-06,
      "loss": 0.0543,
      "step": 73710
    },
    {
      "epoch": 7.029533386230468e-05,
      "model_forward_time": 1.7394764423370361,
      "step": 73710
    },
    {
      "dynamics_loss_avg": 0.05536324679851532,
      "epoch": 7.029533386230468e-05,
      "step": 73710
    },
    {
      "action_loss_avg": 0.001394525017531123,
      "epoch": 7.029533386230468e-05,
      "step": 73710
    },
    {
      "epoch": 7.029533386230468e-05,
      "step": 73710,
      "training_step_time": 7.687249183654785
    },
    {
      "epoch": 7.029628753662109e-05,
      "model_forward_time": 1.7376549243927002,
      "step": 73711
    },
    {
      "epoch": 7.029628753662109e-05,
      "step": 73711,
      "training_step_time": 8.439839363098145
    },
    {
      "epoch": 7.02972412109375e-05,
      "model_forward_time": 1.7423346042633057,
      "step": 73712
    },
    {
      "epoch": 7.02972412109375e-05,
      "step": 73712,
      "training_step_time": 8.154239892959595
    },
    {
      "epoch": 7.02981948852539e-05,
      "model_forward_time": 1.7493679523468018,
      "step": 73713
    },
    {
      "epoch": 7.02981948852539e-05,
      "step": 73713,
      "training_step_time": 8.447243928909302
    },
    {
      "epoch": 7.029914855957031e-05,
      "model_forward_time": 1.7511188983917236,
      "step": 73714
    },
    {
      "epoch": 7.029914855957031e-05,
      "step": 73714,
      "training_step_time": 7.596282005310059
    },
    {
      "epoch": 7.030010223388672e-05,
      "model_forward_time": 1.737196445465088,
      "step": 73715
    },
    {
      "epoch": 7.030010223388672e-05,
      "step": 73715,
      "training_step_time": 7.701282978057861
    },
    {
      "epoch": 7.030105590820312e-05,
      "model_forward_time": 1.7442352771759033,
      "step": 73716
    },
    {
      "epoch": 7.030105590820312e-05,
      "step": 73716,
      "training_step_time": 7.564487934112549
    },
    {
      "epoch": 7.030200958251953e-05,
      "model_forward_time": 1.7420194149017334,
      "step": 73717
    },
    {
      "epoch": 7.030200958251953e-05,
      "step": 73717,
      "training_step_time": 7.664318323135376
    },
    {
      "epoch": 7.030296325683594e-05,
      "model_forward_time": 2.288623332977295,
      "step": 73718
    },
    {
      "epoch": 7.030296325683594e-05,
      "step": 73718,
      "training_step_time": 8.732564687728882
    },
    {
      "epoch": 7.030391693115234e-05,
      "model_forward_time": 1.7409040927886963,
      "step": 73719
    },
    {
      "epoch": 7.030391693115234e-05,
      "step": 73719,
      "training_step_time": 8.639190912246704
    },
    {
      "epoch": 7.030487060546875e-05,
      "grad_norm": 0.10903217643499374,
      "learning_rate": 1.7724168911294016e-06,
      "loss": 0.0563,
      "step": 73720
    },
    {
      "epoch": 7.030487060546875e-05,
      "model_forward_time": 1.7370083332061768,
      "step": 73720
    },
    {
      "dynamics_loss_avg": 0.05333578577265143,
      "epoch": 7.030487060546875e-05,
      "step": 73720
    },
    {
      "action_loss_avg": 0.001396474108332768,
      "epoch": 7.030487060546875e-05,
      "step": 73720
    },
    {
      "epoch": 7.030487060546875e-05,
      "step": 73720,
      "training_step_time": 7.882474422454834
    },
    {
      "epoch": 7.030582427978516e-05,
      "model_forward_time": 1.7292494773864746,
      "step": 73721
    },
    {
      "epoch": 7.030582427978516e-05,
      "step": 73721,
      "training_step_time": 7.832837820053101
    },
    {
      "epoch": 7.030677795410156e-05,
      "model_forward_time": 1.7359488010406494,
      "step": 73722
    },
    {
      "epoch": 7.030677795410156e-05,
      "step": 73722,
      "training_step_time": 7.740266799926758
    },
    {
      "epoch": 7.030773162841797e-05,
      "model_forward_time": 1.7579472064971924,
      "step": 73723
    },
    {
      "epoch": 7.030773162841797e-05,
      "step": 73723,
      "training_step_time": 7.788414716720581
    },
    {
      "epoch": 7.030868530273438e-05,
      "model_forward_time": 1.745431661605835,
      "step": 73724
    },
    {
      "epoch": 7.030868530273438e-05,
      "step": 73724,
      "training_step_time": 7.814757585525513
    },
    {
      "epoch": 7.030963897705078e-05,
      "model_forward_time": 1.7468860149383545,
      "step": 73725
    },
    {
      "epoch": 7.030963897705078e-05,
      "step": 73725,
      "training_step_time": 7.65981912612915
    },
    {
      "epoch": 7.031059265136719e-05,
      "model_forward_time": 1.7311546802520752,
      "step": 73726
    },
    {
      "epoch": 7.031059265136719e-05,
      "step": 73726,
      "training_step_time": 8.760375499725342
    },
    {
      "epoch": 7.03115463256836e-05,
      "model_forward_time": 2.3102476596832275,
      "step": 73727
    },
    {
      "epoch": 7.03115463256836e-05,
      "step": 73727,
      "training_step_time": 8.740979433059692
    },
    {
      "epoch": 7.03125e-05,
      "model_forward_time": 1.7154078483581543,
      "step": 73728
    },
    {
      "epoch": 7.03125e-05,
      "step": 73728,
      "training_step_time": 8.39145040512085
    },
    {
      "epoch": 7.03134536743164e-05,
      "model_forward_time": 1.7472343444824219,
      "step": 73729
    },
    {
      "epoch": 7.03134536743164e-05,
      "step": 73729,
      "training_step_time": 10.2649507522583
    },
    {
      "epoch": 7.031440734863282e-05,
      "grad_norm": 0.07263018935918808,
      "learning_rate": 1.7711542363084505e-06,
      "loss": 0.0547,
      "step": 73730
    },
    {
      "epoch": 7.031440734863282e-05,
      "model_forward_time": 2.8734781742095947,
      "step": 73730
    },
    {
      "dynamics_loss_avg": 0.053959820047020914,
      "epoch": 7.031440734863282e-05,
      "step": 73730
    },
    {
      "action_loss_avg": 0.0009949210740160196,
      "epoch": 7.031440734863282e-05,
      "step": 73730
    },
    {
      "epoch": 7.031440734863282e-05,
      "step": 73730,
      "training_step_time": 11.652827739715576
    },
    {
      "epoch": 7.031536102294921e-05,
      "model_forward_time": 1.7238154411315918,
      "step": 73731
    },
    {
      "epoch": 7.031536102294921e-05,
      "step": 73731,
      "training_step_time": 6.917338848114014
    },
    {
      "epoch": 7.031631469726562e-05,
      "model_forward_time": 1.7396655082702637,
      "step": 73732
    },
    {
      "epoch": 7.031631469726562e-05,
      "step": 73732,
      "training_step_time": 6.869346380233765
    },
    {
      "epoch": 7.031726837158204e-05,
      "model_forward_time": 1.7306318283081055,
      "step": 73733
    },
    {
      "epoch": 7.031726837158204e-05,
      "step": 73733,
      "training_step_time": 6.890404939651489
    },
    {
      "epoch": 7.031822204589843e-05,
      "model_forward_time": 1.7337119579315186,
      "step": 73734
    },
    {
      "epoch": 7.031822204589843e-05,
      "step": 73734,
      "training_step_time": 6.895963191986084
    },
    {
      "epoch": 7.031917572021484e-05,
      "model_forward_time": 1.743074655532837,
      "step": 73735
    },
    {
      "epoch": 7.031917572021484e-05,
      "step": 73735,
      "training_step_time": 6.910996675491333
    },
    {
      "epoch": 7.032012939453125e-05,
      "model_forward_time": 1.7377161979675293,
      "step": 73736
    },
    {
      "epoch": 7.032012939453125e-05,
      "step": 73736,
      "training_step_time": 6.8965113162994385
    },
    {
      "epoch": 7.032108306884765e-05,
      "model_forward_time": 1.7580084800720215,
      "step": 73737
    },
    {
      "epoch": 7.032108306884765e-05,
      "step": 73737,
      "training_step_time": 6.877130031585693
    },
    {
      "epoch": 7.032203674316406e-05,
      "model_forward_time": 1.3792712688446045,
      "step": 73738
    },
    {
      "epoch": 7.032203674316406e-05,
      "step": 73738,
      "training_step_time": 6.857908487319946
    },
    {
      "epoch": 7.032299041748047e-05,
      "model_forward_time": 2.3159475326538086,
      "step": 73739
    },
    {
      "epoch": 7.032299041748047e-05,
      "step": 73739,
      "training_step_time": 7.483210563659668
    },
    {
      "epoch": 7.032394409179687e-05,
      "grad_norm": 0.08995814621448517,
      "learning_rate": 1.7698919345892158e-06,
      "loss": 0.0566,
      "step": 73740
    },
    {
      "epoch": 7.032394409179687e-05,
      "model_forward_time": 1.757434368133545,
      "step": 73740
    },
    {
      "dynamics_loss_avg": 0.04651615722104907,
      "epoch": 7.032394409179687e-05,
      "step": 73740
    },
    {
      "action_loss_avg": 0.0012023967312416062,
      "epoch": 7.032394409179687e-05,
      "step": 73740
    },
    {
      "epoch": 7.032394409179687e-05,
      "step": 73740,
      "training_step_time": 7.872121810913086
    },
    {
      "epoch": 7.032489776611328e-05,
      "model_forward_time": 1.7547190189361572,
      "step": 73741
    },
    {
      "epoch": 7.032489776611328e-05,
      "step": 73741,
      "training_step_time": 6.865377902984619
    },
    {
      "epoch": 7.032585144042969e-05,
      "model_forward_time": 1.7593042850494385,
      "step": 73742
    },
    {
      "epoch": 7.032585144042969e-05,
      "step": 73742,
      "training_step_time": 6.857767105102539
    },
    {
      "epoch": 7.032680511474609e-05,
      "model_forward_time": 1.7612957954406738,
      "step": 73743
    },
    {
      "epoch": 7.032680511474609e-05,
      "step": 73743,
      "training_step_time": 6.8796703815460205
    },
    {
      "epoch": 7.03277587890625e-05,
      "model_forward_time": 1.7555780410766602,
      "step": 73744
    },
    {
      "epoch": 7.03277587890625e-05,
      "step": 73744,
      "training_step_time": 6.8496599197387695
    },
    {
      "epoch": 7.032871246337891e-05,
      "model_forward_time": 1.752110242843628,
      "step": 73745
    },
    {
      "epoch": 7.032871246337891e-05,
      "step": 73745,
      "training_step_time": 6.870858907699585
    },
    {
      "epoch": 7.032966613769531e-05,
      "model_forward_time": 1.754209041595459,
      "step": 73746
    },
    {
      "epoch": 7.032966613769531e-05,
      "step": 73746,
      "training_step_time": 6.852160453796387
    },
    {
      "epoch": 7.033061981201172e-05,
      "model_forward_time": 1.750070571899414,
      "step": 73747
    },
    {
      "epoch": 7.033061981201172e-05,
      "step": 73747,
      "training_step_time": 6.9368345737457275
    },
    {
      "epoch": 7.033157348632813e-05,
      "model_forward_time": 1.7545273303985596,
      "step": 73748
    },
    {
      "epoch": 7.033157348632813e-05,
      "step": 73748,
      "training_step_time": 9.459161758422852
    },
    {
      "epoch": 7.033252716064453e-05,
      "model_forward_time": 1.7504253387451172,
      "step": 73749
    },
    {
      "epoch": 7.033252716064453e-05,
      "step": 73749,
      "training_step_time": 7.843558311462402
    },
    {
      "epoch": 7.033348083496094e-05,
      "grad_norm": 0.10183939337730408,
      "learning_rate": 1.7686299861097429e-06,
      "loss": 0.053,
      "step": 73750
    },
    {
      "epoch": 7.033348083496094e-05,
      "model_forward_time": 1.7489008903503418,
      "step": 73750
    },
    {
      "dynamics_loss_avg": 0.056693229172378776,
      "epoch": 7.033348083496094e-05,
      "step": 73750
    },
    {
      "action_loss_avg": 0.0009132525330642239,
      "epoch": 7.033348083496094e-05,
      "step": 73750
    },
    {
      "epoch": 7.033348083496094e-05,
      "step": 73750,
      "training_step_time": 6.865706920623779
    },
    {
      "epoch": 7.033443450927735e-05,
      "model_forward_time": 1.7624092102050781,
      "step": 73751
    },
    {
      "epoch": 7.033443450927735e-05,
      "step": 73751,
      "training_step_time": 6.8956732749938965
    },
    {
      "epoch": 7.033538818359375e-05,
      "model_forward_time": 1.8266832828521729,
      "step": 73752
    },
    {
      "epoch": 7.033538818359375e-05,
      "step": 73752,
      "training_step_time": 6.892553806304932
    },
    {
      "epoch": 7.033634185791016e-05,
      "model_forward_time": 1.8498117923736572,
      "step": 73753
    },
    {
      "epoch": 7.033634185791016e-05,
      "step": 73753,
      "training_step_time": 6.875981569290161
    },
    {
      "epoch": 7.033729553222657e-05,
      "model_forward_time": 1.8456852436065674,
      "step": 73754
    },
    {
      "epoch": 7.033729553222657e-05,
      "step": 73754,
      "training_step_time": 6.895792007446289
    },
    {
      "epoch": 7.033824920654296e-05,
      "model_forward_time": 1.8343558311462402,
      "step": 73755
    },
    {
      "epoch": 7.033824920654296e-05,
      "step": 73755,
      "training_step_time": 6.880073547363281
    },
    {
      "epoch": 7.033920288085938e-05,
      "model_forward_time": 1.851100206375122,
      "step": 73756
    },
    {
      "epoch": 7.033920288085938e-05,
      "step": 73756,
      "training_step_time": 6.872690677642822
    },
    {
      "epoch": 7.034015655517579e-05,
      "model_forward_time": 1.8503987789154053,
      "step": 73757
    },
    {
      "epoch": 7.034015655517579e-05,
      "step": 73757,
      "training_step_time": 8.28937292098999
    },
    {
      "epoch": 7.034111022949218e-05,
      "model_forward_time": 2.394792318344116,
      "step": 73758
    },
    {
      "epoch": 7.034111022949218e-05,
      "step": 73758,
      "training_step_time": 8.964384078979492
    },
    {
      "epoch": 7.03420639038086e-05,
      "model_forward_time": 1.7146146297454834,
      "step": 73759
    },
    {
      "epoch": 7.03420639038086e-05,
      "step": 73759,
      "training_step_time": 7.645066022872925
    },
    {
      "epoch": 7.0343017578125e-05,
      "grad_norm": 0.06788798421621323,
      "learning_rate": 1.7673683910080358e-06,
      "loss": 0.055,
      "step": 73760
    },
    {
      "epoch": 7.0343017578125e-05,
      "model_forward_time": 1.7081995010375977,
      "step": 73760
    },
    {
      "dynamics_loss_avg": 0.050017814710736276,
      "epoch": 7.0343017578125e-05,
      "step": 73760
    },
    {
      "action_loss_avg": 0.0013485673727700487,
      "epoch": 7.0343017578125e-05,
      "step": 73760
    },
    {
      "epoch": 7.0343017578125e-05,
      "step": 73760,
      "training_step_time": 6.840426683425903
    },
    {
      "epoch": 7.03439712524414e-05,
      "model_forward_time": 1.731137752532959,
      "step": 73761
    },
    {
      "epoch": 7.03439712524414e-05,
      "step": 73761,
      "training_step_time": 6.893446922302246
    },
    {
      "epoch": 7.034492492675781e-05,
      "model_forward_time": 1.8278605937957764,
      "step": 73762
    },
    {
      "epoch": 7.034492492675781e-05,
      "step": 73762,
      "training_step_time": 6.874740839004517
    },
    {
      "epoch": 7.034587860107422e-05,
      "model_forward_time": 1.8286664485931396,
      "step": 73763
    },
    {
      "epoch": 7.034587860107422e-05,
      "step": 73763,
      "training_step_time": 6.85013747215271
    },
    {
      "epoch": 7.034683227539062e-05,
      "model_forward_time": 1.85853910446167,
      "step": 73764
    },
    {
      "epoch": 7.034683227539062e-05,
      "step": 73764,
      "training_step_time": 6.868724346160889
    },
    {
      "epoch": 7.034778594970703e-05,
      "model_forward_time": 1.8453364372253418,
      "step": 73765
    },
    {
      "epoch": 7.034778594970703e-05,
      "step": 73765,
      "training_step_time": 6.857372522354126
    },
    {
      "epoch": 7.034873962402344e-05,
      "model_forward_time": 1.8498592376708984,
      "step": 73766
    },
    {
      "epoch": 7.034873962402344e-05,
      "step": 73766,
      "training_step_time": 6.938171148300171
    },
    {
      "epoch": 7.034969329833984e-05,
      "model_forward_time": 2.4036264419555664,
      "step": 73767
    },
    {
      "epoch": 7.034969329833984e-05,
      "step": 73767,
      "training_step_time": 8.386876344680786
    },
    {
      "epoch": 7.035064697265625e-05,
      "model_forward_time": 1.8262519836425781,
      "step": 73768
    },
    {
      "epoch": 7.035064697265625e-05,
      "step": 73768,
      "training_step_time": 7.872680425643921
    },
    {
      "epoch": 7.035160064697266e-05,
      "model_forward_time": 1.8493375778198242,
      "step": 73769
    },
    {
      "epoch": 7.035160064697266e-05,
      "step": 73769,
      "training_step_time": 6.885735511779785
    },
    {
      "epoch": 7.035255432128906e-05,
      "grad_norm": 0.10178744047880173,
      "learning_rate": 1.766107149422064e-06,
      "loss": 0.051,
      "step": 73770
    },
    {
      "epoch": 7.035255432128906e-05,
      "model_forward_time": 1.8502087593078613,
      "step": 73770
    },
    {
      "dynamics_loss_avg": 0.0463144576177001,
      "epoch": 7.035255432128906e-05,
      "step": 73770
    },
    {
      "action_loss_avg": 0.0014161549450363963,
      "epoch": 7.035255432128906e-05,
      "step": 73770
    },
    {
      "epoch": 7.035255432128906e-05,
      "step": 73770,
      "training_step_time": 6.881076812744141
    },
    {
      "epoch": 7.035350799560547e-05,
      "model_forward_time": 1.8527052402496338,
      "step": 73771
    },
    {
      "epoch": 7.035350799560547e-05,
      "step": 73771,
      "training_step_time": 6.87251353263855
    },
    {
      "epoch": 7.035446166992188e-05,
      "model_forward_time": 1.8497107028961182,
      "step": 73772
    },
    {
      "epoch": 7.035446166992188e-05,
      "step": 73772,
      "training_step_time": 6.865640640258789
    },
    {
      "epoch": 7.035541534423828e-05,
      "model_forward_time": 1.8439595699310303,
      "step": 73773
    },
    {
      "epoch": 7.035541534423828e-05,
      "step": 73773,
      "training_step_time": 6.923344850540161
    },
    {
      "epoch": 7.035636901855469e-05,
      "model_forward_time": 1.8463692665100098,
      "step": 73774
    },
    {
      "epoch": 7.035636901855469e-05,
      "step": 73774,
      "training_step_time": 6.869468450546265
    },
    {
      "epoch": 7.03573226928711e-05,
      "model_forward_time": 1.840475082397461,
      "step": 73775
    },
    {
      "epoch": 7.03573226928711e-05,
      "step": 73775,
      "training_step_time": 6.880943775177002
    },
    {
      "epoch": 7.03582763671875e-05,
      "model_forward_time": 2.414085626602173,
      "step": 73776
    },
    {
      "epoch": 7.03582763671875e-05,
      "step": 73776,
      "training_step_time": 9.520585775375366
    },
    {
      "epoch": 7.035923004150391e-05,
      "model_forward_time": 1.8504149913787842,
      "step": 73777
    },
    {
      "epoch": 7.035923004150391e-05,
      "step": 73777,
      "training_step_time": 8.997297286987305
    },
    {
      "epoch": 7.036018371582032e-05,
      "model_forward_time": 1.852626085281372,
      "step": 73778
    },
    {
      "epoch": 7.036018371582032e-05,
      "step": 73778,
      "training_step_time": 6.895720720291138
    },
    {
      "epoch": 7.036113739013672e-05,
      "model_forward_time": 1.8141424655914307,
      "step": 73779
    },
    {
      "epoch": 7.036113739013672e-05,
      "step": 73779,
      "training_step_time": 6.801098585128784
    },
    {
      "epoch": 7.036209106445313e-05,
      "grad_norm": 0.11295495927333832,
      "learning_rate": 1.7648462614897498e-06,
      "loss": 0.0573,
      "step": 73780
    },
    {
      "epoch": 7.036209106445313e-05,
      "model_forward_time": 1.8507204055786133,
      "step": 73780
    },
    {
      "dynamics_loss_avg": 0.04728631936013698,
      "epoch": 7.036209106445313e-05,
      "step": 73780
    },
    {
      "action_loss_avg": 0.0012644576607272028,
      "epoch": 7.036209106445313e-05,
      "step": 73780
    },
    {
      "epoch": 7.036209106445313e-05,
      "step": 73780,
      "training_step_time": 6.906156778335571
    },
    {
      "epoch": 7.036304473876954e-05,
      "model_forward_time": 1.8444883823394775,
      "step": 73781
    },
    {
      "epoch": 7.036304473876954e-05,
      "step": 73781,
      "training_step_time": 6.927510976791382
    },
    {
      "epoch": 7.036399841308593e-05,
      "model_forward_time": 1.847684621810913,
      "step": 73782
    },
    {
      "epoch": 7.036399841308593e-05,
      "step": 73782,
      "training_step_time": 6.878206729888916
    },
    {
      "epoch": 7.036495208740234e-05,
      "model_forward_time": 1.8531506061553955,
      "step": 73783
    },
    {
      "epoch": 7.036495208740234e-05,
      "step": 73783,
      "training_step_time": 6.861945629119873
    },
    {
      "epoch": 7.036590576171876e-05,
      "model_forward_time": 1.853562831878662,
      "step": 73784
    },
    {
      "epoch": 7.036590576171876e-05,
      "step": 73784,
      "training_step_time": 6.984391689300537
    },
    {
      "epoch": 7.036685943603515e-05,
      "model_forward_time": 2.407504081726074,
      "step": 73785
    },
    {
      "epoch": 7.036685943603515e-05,
      "step": 73785,
      "training_step_time": 8.796374559402466
    },
    {
      "epoch": 7.036781311035156e-05,
      "model_forward_time": 1.8374571800231934,
      "step": 73786
    },
    {
      "epoch": 7.036781311035156e-05,
      "step": 73786,
      "training_step_time": 7.962653636932373
    },
    {
      "epoch": 7.036876678466797e-05,
      "model_forward_time": 1.848113775253296,
      "step": 73787
    },
    {
      "epoch": 7.036876678466797e-05,
      "step": 73787,
      "training_step_time": 7.6242804527282715
    },
    {
      "epoch": 7.036972045898437e-05,
      "model_forward_time": 1.852492094039917,
      "step": 73788
    },
    {
      "epoch": 7.036972045898437e-05,
      "step": 73788,
      "training_step_time": 6.790877819061279
    },
    {
      "epoch": 7.037067413330078e-05,
      "model_forward_time": 1.8444476127624512,
      "step": 73789
    },
    {
      "epoch": 7.037067413330078e-05,
      "step": 73789,
      "training_step_time": 6.891579866409302
    },
    {
      "epoch": 7.03716278076172e-05,
      "grad_norm": 0.09931709617376328,
      "learning_rate": 1.7635857273489864e-06,
      "loss": 0.0578,
      "step": 73790
    },
    {
      "epoch": 7.03716278076172e-05,
      "model_forward_time": 1.488051414489746,
      "step": 73790
    },
    {
      "dynamics_loss_avg": 0.06534935384988785,
      "epoch": 7.03716278076172e-05,
      "step": 73790
    },
    {
      "action_loss_avg": 0.0012885444157291204,
      "epoch": 7.03716278076172e-05,
      "step": 73790
    },
    {
      "epoch": 7.03716278076172e-05,
      "step": 73790,
      "training_step_time": 6.849169731140137
    },
    {
      "epoch": 7.037258148193359e-05,
      "model_forward_time": 1.8444786071777344,
      "step": 73791
    },
    {
      "epoch": 7.037258148193359e-05,
      "step": 73791,
      "training_step_time": 6.867962121963501
    },
    {
      "epoch": 7.037353515625e-05,
      "model_forward_time": 1.8448607921600342,
      "step": 73792
    },
    {
      "epoch": 7.037353515625e-05,
      "step": 73792,
      "training_step_time": 6.919568061828613
    },
    {
      "epoch": 7.037448883056641e-05,
      "model_forward_time": 1.8480746746063232,
      "step": 73793
    },
    {
      "epoch": 7.037448883056641e-05,
      "step": 73793,
      "training_step_time": 6.855197191238403
    },
    {
      "epoch": 7.037544250488281e-05,
      "model_forward_time": 2.3567986488342285,
      "step": 73794
    },
    {
      "epoch": 7.037544250488281e-05,
      "step": 73794,
      "training_step_time": 8.928943157196045
    },
    {
      "epoch": 7.037639617919922e-05,
      "model_forward_time": 1.8419697284698486,
      "step": 73795
    },
    {
      "epoch": 7.037639617919922e-05,
      "step": 73795,
      "training_step_time": 7.964290380477905
    },
    {
      "epoch": 7.037734985351563e-05,
      "model_forward_time": 1.8557302951812744,
      "step": 73796
    },
    {
      "epoch": 7.037734985351563e-05,
      "step": 73796,
      "training_step_time": 7.664597272872925
    },
    {
      "epoch": 7.037830352783203e-05,
      "model_forward_time": 1.8521175384521484,
      "step": 73797
    },
    {
      "epoch": 7.037830352783203e-05,
      "step": 73797,
      "training_step_time": 6.882381439208984
    },
    {
      "epoch": 7.037925720214844e-05,
      "model_forward_time": 1.852325677871704,
      "step": 73798
    },
    {
      "epoch": 7.037925720214844e-05,
      "step": 73798,
      "training_step_time": 6.8677918910980225
    },
    {
      "epoch": 7.038021087646485e-05,
      "model_forward_time": 1.8418517112731934,
      "step": 73799
    },
    {
      "epoch": 7.038021087646485e-05,
      "step": 73799,
      "training_step_time": 6.8508665561676025
    },
    {
      "epoch": 7.038116455078125e-05,
      "grad_norm": 0.11931363493204117,
      "learning_rate": 1.7623255471376199e-06,
      "loss": 0.0523,
      "step": 73800
    },
    {
      "epoch": 7.038116455078125e-05,
      "model_forward_time": 1.3682546615600586,
      "step": 73800
    },
    {
      "dynamics_loss_avg": 0.05498772710561752,
      "epoch": 7.038116455078125e-05,
      "step": 73800
    },
    {
      "action_loss_avg": 0.0011910048604477196,
      "epoch": 7.038116455078125e-05,
      "step": 73800
    },
    {
      "epoch": 7.038116455078125e-05,
      "step": 73800,
      "training_step_time": 6.8590357303619385
    },
    {
      "epoch": 7.038211822509766e-05,
      "model_forward_time": 1.746565580368042,
      "step": 73801
    },
    {
      "epoch": 7.038211822509766e-05,
      "step": 73801,
      "training_step_time": 6.8890907764434814
    },
    {
      "epoch": 7.038307189941407e-05,
      "model_forward_time": 1.7514054775238037,
      "step": 73802
    },
    {
      "epoch": 7.038307189941407e-05,
      "step": 73802,
      "training_step_time": 6.876040697097778
    },
    {
      "epoch": 7.038402557373047e-05,
      "model_forward_time": 1.73929762840271,
      "step": 73803
    },
    {
      "epoch": 7.038402557373047e-05,
      "step": 73803,
      "training_step_time": 9.00734281539917
    },
    {
      "epoch": 7.038497924804688e-05,
      "model_forward_time": 2.2900893688201904,
      "step": 73804
    },
    {
      "epoch": 7.038497924804688e-05,
      "step": 73804,
      "training_step_time": 8.517458438873291
    },
    {
      "epoch": 7.038593292236329e-05,
      "model_forward_time": 1.750983476638794,
      "step": 73805
    },
    {
      "epoch": 7.038593292236329e-05,
      "step": 73805,
      "training_step_time": 7.784323453903198
    },
    {
      "epoch": 7.038688659667969e-05,
      "model_forward_time": 1.7545561790466309,
      "step": 73806
    },
    {
      "epoch": 7.038688659667969e-05,
      "step": 73806,
      "training_step_time": 6.943140745162964
    },
    {
      "epoch": 7.03878402709961e-05,
      "model_forward_time": 1.7465691566467285,
      "step": 73807
    },
    {
      "epoch": 7.03878402709961e-05,
      "step": 73807,
      "training_step_time": 6.88225793838501
    },
    {
      "epoch": 7.03887939453125e-05,
      "model_forward_time": 1.7417666912078857,
      "step": 73808
    },
    {
      "epoch": 7.03887939453125e-05,
      "step": 73808,
      "training_step_time": 6.921942234039307
    },
    {
      "epoch": 7.03897476196289e-05,
      "model_forward_time": 1.7632386684417725,
      "step": 73809
    },
    {
      "epoch": 7.03897476196289e-05,
      "step": 73809,
      "training_step_time": 6.914202690124512
    },
    {
      "epoch": 7.039070129394531e-05,
      "grad_norm": 0.09178231656551361,
      "learning_rate": 1.7610657209934645e-06,
      "loss": 0.0525,
      "step": 73810
    },
    {
      "epoch": 7.039070129394531e-05,
      "model_forward_time": 1.7461957931518555,
      "step": 73810
    },
    {
      "dynamics_loss_avg": 0.05839743372052908,
      "epoch": 7.039070129394531e-05,
      "step": 73810
    },
    {
      "action_loss_avg": 0.0011462976224720478,
      "epoch": 7.039070129394531e-05,
      "step": 73810
    },
    {
      "epoch": 7.039070129394531e-05,
      "step": 73810,
      "training_step_time": 6.889282464981079
    },
    {
      "epoch": 7.039165496826173e-05,
      "model_forward_time": 1.7363815307617188,
      "step": 73811
    },
    {
      "epoch": 7.039165496826173e-05,
      "step": 73811,
      "training_step_time": 6.913160800933838
    },
    {
      "epoch": 7.039260864257812e-05,
      "model_forward_time": 1.742875576019287,
      "step": 73812
    },
    {
      "epoch": 7.039260864257812e-05,
      "step": 73812,
      "training_step_time": 7.8530895709991455
    },
    {
      "epoch": 7.039356231689453e-05,
      "model_forward_time": 1.972001075744629,
      "step": 73813
    },
    {
      "epoch": 7.039356231689453e-05,
      "step": 73813,
      "training_step_time": 9.828175067901611
    },
    {
      "epoch": 7.039451599121093e-05,
      "model_forward_time": 1.7580797672271729,
      "step": 73814
    },
    {
      "epoch": 7.039451599121093e-05,
      "step": 73814,
      "training_step_time": 8.051244497299194
    },
    {
      "epoch": 7.039546966552734e-05,
      "model_forward_time": 1.7622790336608887,
      "step": 73815
    },
    {
      "epoch": 7.039546966552734e-05,
      "step": 73815,
      "training_step_time": 7.637782573699951
    },
    {
      "epoch": 7.039642333984375e-05,
      "model_forward_time": 1.7628822326660156,
      "step": 73816
    },
    {
      "epoch": 7.039642333984375e-05,
      "step": 73816,
      "training_step_time": 6.833659410476685
    },
    {
      "epoch": 7.039737701416015e-05,
      "model_forward_time": 1.7503910064697266,
      "step": 73817
    },
    {
      "epoch": 7.039737701416015e-05,
      "step": 73817,
      "training_step_time": 6.902459621429443
    },
    {
      "epoch": 7.039833068847656e-05,
      "model_forward_time": 1.748133897781372,
      "step": 73818
    },
    {
      "epoch": 7.039833068847656e-05,
      "step": 73818,
      "training_step_time": 6.86890983581543
    },
    {
      "epoch": 7.039928436279297e-05,
      "model_forward_time": 1.7600891590118408,
      "step": 73819
    },
    {
      "epoch": 7.039928436279297e-05,
      "step": 73819,
      "training_step_time": 6.8919761180877686
    },
    {
      "epoch": 7.040023803710937e-05,
      "grad_norm": 0.11806685477495193,
      "learning_rate": 1.7598062490542916e-06,
      "loss": 0.0564,
      "step": 73820
    },
    {
      "epoch": 7.040023803710937e-05,
      "model_forward_time": 1.7522082328796387,
      "step": 73820
    },
    {
      "dynamics_loss_avg": 0.04622307810932398,
      "epoch": 7.040023803710937e-05,
      "step": 73820
    },
    {
      "action_loss_avg": 0.000635986564157065,
      "epoch": 7.040023803710937e-05,
      "step": 73820
    },
    {
      "epoch": 7.040023803710937e-05,
      "step": 73820,
      "training_step_time": 6.90538477897644
    },
    {
      "epoch": 7.040119171142578e-05,
      "model_forward_time": 1.7480981349945068,
      "step": 73821
    },
    {
      "epoch": 7.040119171142578e-05,
      "step": 73821,
      "training_step_time": 7.753559112548828
    },
    {
      "epoch": 7.040214538574219e-05,
      "model_forward_time": 2.328246593475342,
      "step": 73822
    },
    {
      "epoch": 7.040214538574219e-05,
      "step": 73822,
      "training_step_time": 9.400861501693726
    },
    {
      "epoch": 7.040309906005859e-05,
      "model_forward_time": 1.773177146911621,
      "step": 73823
    },
    {
      "epoch": 7.040309906005859e-05,
      "step": 73823,
      "training_step_time": 9.854363441467285
    },
    {
      "epoch": 7.0404052734375e-05,
      "model_forward_time": 1.7449884414672852,
      "step": 73824
    },
    {
      "epoch": 7.0404052734375e-05,
      "step": 73824,
      "training_step_time": 7.690776824951172
    },
    {
      "epoch": 7.040500640869141e-05,
      "model_forward_time": 1.7609074115753174,
      "step": 73825
    },
    {
      "epoch": 7.040500640869141e-05,
      "step": 73825,
      "training_step_time": 6.843034029006958
    },
    {
      "epoch": 7.04059600830078e-05,
      "model_forward_time": 1.7288539409637451,
      "step": 73826
    },
    {
      "epoch": 7.04059600830078e-05,
      "step": 73826,
      "training_step_time": 6.889665126800537
    },
    {
      "epoch": 7.040691375732422e-05,
      "model_forward_time": 1.7572870254516602,
      "step": 73827
    },
    {
      "epoch": 7.040691375732422e-05,
      "step": 73827,
      "training_step_time": 6.901214122772217
    },
    {
      "epoch": 7.040786743164063e-05,
      "model_forward_time": 1.735008955001831,
      "step": 73828
    },
    {
      "epoch": 7.040786743164063e-05,
      "step": 73828,
      "training_step_time": 6.883754730224609
    },
    {
      "epoch": 7.040882110595703e-05,
      "model_forward_time": 1.7519419193267822,
      "step": 73829
    },
    {
      "epoch": 7.040882110595703e-05,
      "step": 73829,
      "training_step_time": 6.842157363891602
    },
    {
      "epoch": 7.040977478027344e-05,
      "grad_norm": 0.08173631131649017,
      "learning_rate": 1.7585471314578378e-06,
      "loss": 0.0546,
      "step": 73830
    },
    {
      "epoch": 7.040977478027344e-05,
      "model_forward_time": 1.3739049434661865,
      "step": 73830
    },
    {
      "dynamics_loss_avg": 0.05841235667467117,
      "epoch": 7.040977478027344e-05,
      "step": 73830
    },
    {
      "action_loss_avg": 0.0009964587516151369,
      "epoch": 7.040977478027344e-05,
      "step": 73830
    },
    {
      "epoch": 7.040977478027344e-05,
      "step": 73830,
      "training_step_time": 7.692622900009155
    },
    {
      "epoch": 7.041072845458985e-05,
      "model_forward_time": 1.7824950218200684,
      "step": 73831
    },
    {
      "epoch": 7.041072845458985e-05,
      "step": 73831,
      "training_step_time": 10.321349620819092
    },
    {
      "epoch": 7.041168212890624e-05,
      "model_forward_time": 2.3085150718688965,
      "step": 73832
    },
    {
      "epoch": 7.041168212890624e-05,
      "step": 73832,
      "training_step_time": 8.45006799697876
    },
    {
      "epoch": 7.041263580322265e-05,
      "model_forward_time": 1.7481234073638916,
      "step": 73833
    },
    {
      "epoch": 7.041263580322265e-05,
      "step": 73833,
      "training_step_time": 7.710255861282349
    },
    {
      "epoch": 7.041358947753907e-05,
      "model_forward_time": 1.7499802112579346,
      "step": 73834
    },
    {
      "epoch": 7.041358947753907e-05,
      "step": 73834,
      "training_step_time": 6.92340350151062
    },
    {
      "epoch": 7.041454315185546e-05,
      "model_forward_time": 1.7565877437591553,
      "step": 73835
    },
    {
      "epoch": 7.041454315185546e-05,
      "step": 73835,
      "training_step_time": 6.928847551345825
    },
    {
      "epoch": 7.041549682617187e-05,
      "model_forward_time": 1.751023292541504,
      "step": 73836
    },
    {
      "epoch": 7.041549682617187e-05,
      "step": 73836,
      "training_step_time": 6.89385724067688
    },
    {
      "epoch": 7.041645050048828e-05,
      "model_forward_time": 1.7427785396575928,
      "step": 73837
    },
    {
      "epoch": 7.041645050048828e-05,
      "step": 73837,
      "training_step_time": 6.950781583786011
    },
    {
      "epoch": 7.041740417480468e-05,
      "model_forward_time": 1.7404913902282715,
      "step": 73838
    },
    {
      "epoch": 7.041740417480468e-05,
      "step": 73838,
      "training_step_time": 6.90273642539978
    },
    {
      "epoch": 7.041835784912109e-05,
      "model_forward_time": 1.7508463859558105,
      "step": 73839
    },
    {
      "epoch": 7.041835784912109e-05,
      "step": 73839,
      "training_step_time": 8.818761110305786
    },
    {
      "epoch": 7.04193115234375e-05,
      "grad_norm": 0.17932596802711487,
      "learning_rate": 1.7572883683417946e-06,
      "loss": 0.0573,
      "step": 73840
    },
    {
      "epoch": 7.04193115234375e-05,
      "model_forward_time": 1.7521443367004395,
      "step": 73840
    },
    {
      "dynamics_loss_avg": 0.04797334335744381,
      "epoch": 7.04193115234375e-05,
      "step": 73840
    },
    {
      "action_loss_avg": 0.0010206576465861873,
      "epoch": 7.04193115234375e-05,
      "step": 73840
    },
    {
      "epoch": 7.04193115234375e-05,
      "step": 73840,
      "training_step_time": 8.856169700622559
    },
    {
      "epoch": 7.04202651977539e-05,
      "model_forward_time": 2.298755645751953,
      "step": 73841
    },
    {
      "epoch": 7.04202651977539e-05,
      "step": 73841,
      "training_step_time": 8.73360276222229
    },
    {
      "epoch": 7.042121887207031e-05,
      "model_forward_time": 1.749647617340088,
      "step": 73842
    },
    {
      "epoch": 7.042121887207031e-05,
      "step": 73842,
      "training_step_time": 8.126166105270386
    },
    {
      "epoch": 7.042217254638672e-05,
      "model_forward_time": 1.7445313930511475,
      "step": 73843
    },
    {
      "epoch": 7.042217254638672e-05,
      "step": 73843,
      "training_step_time": 7.681433200836182
    },
    {
      "epoch": 7.042312622070312e-05,
      "model_forward_time": 1.746131181716919,
      "step": 73844
    },
    {
      "epoch": 7.042312622070312e-05,
      "step": 73844,
      "training_step_time": 6.905992746353149
    },
    {
      "epoch": 7.042407989501953e-05,
      "model_forward_time": 1.7532145977020264,
      "step": 73845
    },
    {
      "epoch": 7.042407989501953e-05,
      "step": 73845,
      "training_step_time": 6.916885852813721
    },
    {
      "epoch": 7.042503356933594e-05,
      "model_forward_time": 1.744666337966919,
      "step": 73846
    },
    {
      "epoch": 7.042503356933594e-05,
      "step": 73846,
      "training_step_time": 6.985444068908691
    },
    {
      "epoch": 7.042598724365234e-05,
      "model_forward_time": 1.7659189701080322,
      "step": 73847
    },
    {
      "epoch": 7.042598724365234e-05,
      "step": 73847,
      "training_step_time": 6.9167845249176025
    },
    {
      "epoch": 7.042694091796875e-05,
      "model_forward_time": 1.7488062381744385,
      "step": 73848
    },
    {
      "epoch": 7.042694091796875e-05,
      "step": 73848,
      "training_step_time": 7.794656276702881
    },
    {
      "epoch": 7.042789459228516e-05,
      "model_forward_time": 1.7377793788909912,
      "step": 73849
    },
    {
      "epoch": 7.042789459228516e-05,
      "step": 73849,
      "training_step_time": 9.890673160552979
    },
    {
      "epoch": 7.042884826660156e-05,
      "grad_norm": 0.10683520138263702,
      "learning_rate": 1.7560299598438214e-06,
      "loss": 0.0547,
      "step": 73850
    },
    {
      "epoch": 7.042884826660156e-05,
      "model_forward_time": 2.300105333328247,
      "step": 73850
    },
    {
      "dynamics_loss_avg": 0.060197440721094606,
      "epoch": 7.042884826660156e-05,
      "step": 73850
    },
    {
      "action_loss_avg": 0.0011037670774385333,
      "epoch": 7.042884826660156e-05,
      "step": 73850
    },
    {
      "epoch": 7.042884826660156e-05,
      "step": 73850,
      "training_step_time": 9.033652782440186
    },
    {
      "epoch": 7.042980194091797e-05,
      "model_forward_time": 1.7545595169067383,
      "step": 73851
    },
    {
      "epoch": 7.042980194091797e-05,
      "step": 73851,
      "training_step_time": 8.123902797698975
    },
    {
      "epoch": 7.043075561523438e-05,
      "model_forward_time": 1.3724513053894043,
      "step": 73852
    },
    {
      "epoch": 7.043075561523438e-05,
      "step": 73852,
      "training_step_time": 7.6770124435424805
    },
    {
      "epoch": 7.043170928955078e-05,
      "model_forward_time": 1.7447798252105713,
      "step": 73853
    },
    {
      "epoch": 7.043170928955078e-05,
      "step": 73853,
      "training_step_time": 6.943883657455444
    },
    {
      "epoch": 7.043266296386719e-05,
      "model_forward_time": 1.7467989921569824,
      "step": 73854
    },
    {
      "epoch": 7.043266296386719e-05,
      "step": 73854,
      "training_step_time": 6.915513753890991
    },
    {
      "epoch": 7.04336166381836e-05,
      "model_forward_time": 1.767491340637207,
      "step": 73855
    },
    {
      "epoch": 7.04336166381836e-05,
      "step": 73855,
      "training_step_time": 6.97795295715332
    },
    {
      "epoch": 7.04345703125e-05,
      "model_forward_time": 1.7538902759552002,
      "step": 73856
    },
    {
      "epoch": 7.04345703125e-05,
      "step": 73856,
      "training_step_time": 6.9321489334106445
    },
    {
      "epoch": 7.04355239868164e-05,
      "model_forward_time": 1.7561440467834473,
      "step": 73857
    },
    {
      "epoch": 7.04355239868164e-05,
      "step": 73857,
      "training_step_time": 7.7902703285217285
    },
    {
      "epoch": 7.043647766113282e-05,
      "model_forward_time": 1.7348535060882568,
      "step": 73858
    },
    {
      "epoch": 7.043647766113282e-05,
      "step": 73858,
      "training_step_time": 10.402666330337524
    },
    {
      "epoch": 7.043743133544921e-05,
      "model_forward_time": 2.335289716720581,
      "step": 73859
    },
    {
      "epoch": 7.043743133544921e-05,
      "step": 73859,
      "training_step_time": 8.59906268119812
    },
    {
      "epoch": 7.043838500976562e-05,
      "grad_norm": 0.1580716073513031,
      "learning_rate": 1.7547719061015333e-06,
      "loss": 0.0572,
      "step": 73860
    },
    {
      "epoch": 7.043838500976562e-05,
      "model_forward_time": 1.7464840412139893,
      "step": 73860
    },
    {
      "dynamics_loss_avg": 0.05884998720139265,
      "epoch": 7.043838500976562e-05,
      "step": 73860
    },
    {
      "action_loss_avg": 0.0013552349235396832,
      "epoch": 7.043838500976562e-05,
      "step": 73860
    },
    {
      "epoch": 7.043838500976562e-05,
      "step": 73860,
      "training_step_time": 8.074158191680908
    },
    {
      "epoch": 7.043933868408204e-05,
      "model_forward_time": 1.7506887912750244,
      "step": 73861
    },
    {
      "epoch": 7.043933868408204e-05,
      "step": 73861,
      "training_step_time": 7.705642938613892
    },
    {
      "epoch": 7.044029235839843e-05,
      "model_forward_time": 1.7478835582733154,
      "step": 73862
    },
    {
      "epoch": 7.044029235839843e-05,
      "step": 73862,
      "training_step_time": 6.971330404281616
    },
    {
      "epoch": 7.044124603271484e-05,
      "model_forward_time": 1.7424607276916504,
      "step": 73863
    },
    {
      "epoch": 7.044124603271484e-05,
      "step": 73863,
      "training_step_time": 6.980433940887451
    },
    {
      "epoch": 7.044219970703125e-05,
      "model_forward_time": 1.7483868598937988,
      "step": 73864
    },
    {
      "epoch": 7.044219970703125e-05,
      "step": 73864,
      "training_step_time": 6.931909799575806
    },
    {
      "epoch": 7.044315338134765e-05,
      "model_forward_time": 1.7511792182922363,
      "step": 73865
    },
    {
      "epoch": 7.044315338134765e-05,
      "step": 73865,
      "training_step_time": 6.967880010604858
    },
    {
      "epoch": 7.044410705566406e-05,
      "model_forward_time": 1.7304959297180176,
      "step": 73866
    },
    {
      "epoch": 7.044410705566406e-05,
      "step": 73866,
      "training_step_time": 7.723446846008301
    },
    {
      "epoch": 7.044506072998047e-05,
      "model_forward_time": 1.7465214729309082,
      "step": 73867
    },
    {
      "epoch": 7.044506072998047e-05,
      "step": 73867,
      "training_step_time": 7.648629665374756
    },
    {
      "epoch": 7.044601440429687e-05,
      "model_forward_time": 1.7913682460784912,
      "step": 73868
    },
    {
      "epoch": 7.044601440429687e-05,
      "step": 73868,
      "training_step_time": 9.456470251083374
    },
    {
      "epoch": 7.044696807861328e-05,
      "model_forward_time": 2.30549955368042,
      "step": 73869
    },
    {
      "epoch": 7.044696807861328e-05,
      "step": 73869,
      "training_step_time": 11.127813577651978
    },
    {
      "epoch": 7.044792175292969e-05,
      "grad_norm": 0.07138552516698837,
      "learning_rate": 1.7535142072525103e-06,
      "loss": 0.0548,
      "step": 73870
    },
    {
      "epoch": 7.044792175292969e-05,
      "model_forward_time": 1.7541301250457764,
      "step": 73870
    },
    {
      "dynamics_loss_avg": 0.05373512227088213,
      "epoch": 7.044792175292969e-05,
      "step": 73870
    },
    {
      "action_loss_avg": 0.0011574237360036932,
      "epoch": 7.044792175292969e-05,
      "step": 73870
    },
    {
      "epoch": 7.044792175292969e-05,
      "step": 73870,
      "training_step_time": 7.504216194152832
    },
    {
      "epoch": 7.044887542724609e-05,
      "model_forward_time": 1.7550206184387207,
      "step": 73871
    },
    {
      "epoch": 7.044887542724609e-05,
      "step": 73871,
      "training_step_time": 6.829010725021362
    },
    {
      "epoch": 7.04498291015625e-05,
      "model_forward_time": 1.7647883892059326,
      "step": 73872
    },
    {
      "epoch": 7.04498291015625e-05,
      "step": 73872,
      "training_step_time": 6.757153749465942
    },
    {
      "epoch": 7.045078277587891e-05,
      "model_forward_time": 1.769676685333252,
      "step": 73873
    },
    {
      "epoch": 7.045078277587891e-05,
      "step": 73873,
      "training_step_time": 6.764974594116211
    },
    {
      "epoch": 7.045173645019531e-05,
      "model_forward_time": 1.7409768104553223,
      "step": 73874
    },
    {
      "epoch": 7.045173645019531e-05,
      "step": 73874,
      "training_step_time": 6.747954845428467
    },
    {
      "epoch": 7.045269012451172e-05,
      "model_forward_time": 1.7550017833709717,
      "step": 73875
    },
    {
      "epoch": 7.045269012451172e-05,
      "step": 73875,
      "training_step_time": 7.64545202255249
    },
    {
      "epoch": 7.045364379882813e-05,
      "model_forward_time": 1.7627553939819336,
      "step": 73876
    },
    {
      "epoch": 7.045364379882813e-05,
      "step": 73876,
      "training_step_time": 9.330162048339844
    },
    {
      "epoch": 7.045459747314453e-05,
      "model_forward_time": 1.7969117164611816,
      "step": 73877
    },
    {
      "epoch": 7.045459747314453e-05,
      "step": 73877,
      "training_step_time": 9.429090738296509
    },
    {
      "epoch": 7.045555114746094e-05,
      "model_forward_time": 2.325659990310669,
      "step": 73878
    },
    {
      "epoch": 7.045555114746094e-05,
      "step": 73878,
      "training_step_time": 7.853992223739624
    },
    {
      "epoch": 7.045650482177735e-05,
      "model_forward_time": 1.7662999629974365,
      "step": 73879
    },
    {
      "epoch": 7.045650482177735e-05,
      "step": 73879,
      "training_step_time": 8.706574201583862
    },
    {
      "epoch": 7.045745849609375e-05,
      "grad_norm": 0.08755390346050262,
      "learning_rate": 1.7522568634342923e-06,
      "loss": 0.0536,
      "step": 73880
    },
    {
      "epoch": 7.045745849609375e-05,
      "model_forward_time": 1.7429509162902832,
      "step": 73880
    },
    {
      "dynamics_loss_avg": 0.0490759938955307,
      "epoch": 7.045745849609375e-05,
      "step": 73880
    },
    {
      "action_loss_avg": 0.0012887640128610655,
      "epoch": 7.045745849609375e-05,
      "step": 73880
    },
    {
      "epoch": 7.045745849609375e-05,
      "step": 73880,
      "training_step_time": 6.841379404067993
    },
    {
      "epoch": 7.045841217041016e-05,
      "model_forward_time": 1.763749599456787,
      "step": 73881
    },
    {
      "epoch": 7.045841217041016e-05,
      "step": 73881,
      "training_step_time": 6.75260329246521
    },
    {
      "epoch": 7.045936584472657e-05,
      "model_forward_time": 1.382533073425293,
      "step": 73882
    },
    {
      "epoch": 7.045936584472657e-05,
      "step": 73882,
      "training_step_time": 6.746413469314575
    },
    {
      "epoch": 7.046031951904296e-05,
      "model_forward_time": 1.7528724670410156,
      "step": 73883
    },
    {
      "epoch": 7.046031951904296e-05,
      "step": 73883,
      "training_step_time": 6.894619703292847
    },
    {
      "epoch": 7.046127319335938e-05,
      "model_forward_time": 1.7418251037597656,
      "step": 73884
    },
    {
      "epoch": 7.046127319335938e-05,
      "step": 73884,
      "training_step_time": 8.757482767105103
    },
    {
      "epoch": 7.046222686767579e-05,
      "model_forward_time": 1.7671196460723877,
      "step": 73885
    },
    {
      "epoch": 7.046222686767579e-05,
      "step": 73885,
      "training_step_time": 8.68342924118042
    },
    {
      "epoch": 7.046318054199218e-05,
      "model_forward_time": 1.7582519054412842,
      "step": 73886
    },
    {
      "epoch": 7.046318054199218e-05,
      "step": 73886,
      "training_step_time": 9.968770742416382
    },
    {
      "epoch": 7.04641342163086e-05,
      "model_forward_time": 2.329280376434326,
      "step": 73887
    },
    {
      "epoch": 7.04641342163086e-05,
      "step": 73887,
      "training_step_time": 10.351050853729248
    },
    {
      "epoch": 7.0465087890625e-05,
      "model_forward_time": 1.7653462886810303,
      "step": 73888
    },
    {
      "epoch": 7.0465087890625e-05,
      "step": 73888,
      "training_step_time": 8.454147577285767
    },
    {
      "epoch": 7.04660415649414e-05,
      "model_forward_time": 1.7649364471435547,
      "step": 73889
    },
    {
      "epoch": 7.04660415649414e-05,
      "step": 73889,
      "training_step_time": 7.543835878372192
    },
    {
      "epoch": 7.046699523925781e-05,
      "grad_norm": 0.09363917261362076,
      "learning_rate": 1.7509998747843826e-06,
      "loss": 0.0549,
      "step": 73890
    },
    {
      "epoch": 7.046699523925781e-05,
      "model_forward_time": 1.7574408054351807,
      "step": 73890
    },
    {
      "dynamics_loss_avg": 0.04957606699317694,
      "epoch": 7.046699523925781e-05,
      "step": 73890
    },
    {
      "action_loss_avg": 0.001330204209079966,
      "epoch": 7.046699523925781e-05,
      "step": 73890
    },
    {
      "epoch": 7.046699523925781e-05,
      "step": 73890,
      "training_step_time": 6.786609888076782
    },
    {
      "epoch": 7.046794891357422e-05,
      "model_forward_time": 1.741638422012329,
      "step": 73891
    },
    {
      "epoch": 7.046794891357422e-05,
      "step": 73891,
      "training_step_time": 6.792456150054932
    },
    {
      "epoch": 7.046890258789062e-05,
      "model_forward_time": 1.7491734027862549,
      "step": 73892
    },
    {
      "epoch": 7.046890258789062e-05,
      "step": 73892,
      "training_step_time": 6.798673629760742
    },
    {
      "epoch": 7.046985626220703e-05,
      "model_forward_time": 1.7428474426269531,
      "step": 73893
    },
    {
      "epoch": 7.046985626220703e-05,
      "step": 73893,
      "training_step_time": 8.242727518081665
    },
    {
      "epoch": 7.047080993652344e-05,
      "model_forward_time": 1.7341878414154053,
      "step": 73894
    },
    {
      "epoch": 7.047080993652344e-05,
      "step": 73894,
      "training_step_time": 7.678793668746948
    },
    {
      "epoch": 7.047176361083984e-05,
      "model_forward_time": 1.7379937171936035,
      "step": 73895
    },
    {
      "epoch": 7.047176361083984e-05,
      "step": 73895,
      "training_step_time": 8.75442385673523
    },
    {
      "epoch": 7.047271728515625e-05,
      "model_forward_time": 2.3284895420074463,
      "step": 73896
    },
    {
      "epoch": 7.047271728515625e-05,
      "step": 73896,
      "training_step_time": 9.077035665512085
    },
    {
      "epoch": 7.047367095947266e-05,
      "model_forward_time": 1.749828577041626,
      "step": 73897
    },
    {
      "epoch": 7.047367095947266e-05,
      "step": 73897,
      "training_step_time": 8.398850679397583
    },
    {
      "epoch": 7.047462463378906e-05,
      "model_forward_time": 1.7435073852539062,
      "step": 73898
    },
    {
      "epoch": 7.047462463378906e-05,
      "step": 73898,
      "training_step_time": 7.538222789764404
    },
    {
      "epoch": 7.047557830810547e-05,
      "model_forward_time": 1.7450926303863525,
      "step": 73899
    },
    {
      "epoch": 7.047557830810547e-05,
      "step": 73899,
      "training_step_time": 6.777952671051025
    },
    {
      "epoch": 7.047653198242188e-05,
      "grad_norm": 0.0915575847029686,
      "learning_rate": 1.7497432414402405e-06,
      "loss": 0.0547,
      "step": 73900
    },
    {
      "epoch": 7.047653198242188e-05,
      "model_forward_time": 1.7563226222991943,
      "step": 73900
    },
    {
      "dynamics_loss_avg": 0.053576180897653106,
      "epoch": 7.047653198242188e-05,
      "step": 73900
    },
    {
      "action_loss_avg": 0.0010011130987550132,
      "epoch": 7.047653198242188e-05,
      "step": 73900
    },
    {
      "epoch": 7.047653198242188e-05,
      "step": 73900,
      "training_step_time": 6.792596340179443
    },
    {
      "epoch": 7.047748565673828e-05,
      "model_forward_time": 1.7502272129058838,
      "step": 73901
    },
    {
      "epoch": 7.047748565673828e-05,
      "step": 73901,
      "training_step_time": 6.775122165679932
    },
    {
      "epoch": 7.047843933105469e-05,
      "model_forward_time": 1.7460503578186035,
      "step": 73902
    },
    {
      "epoch": 7.047843933105469e-05,
      "step": 73902,
      "training_step_time": 7.668500900268555
    },
    {
      "epoch": 7.04793930053711e-05,
      "model_forward_time": 1.7324416637420654,
      "step": 73903
    },
    {
      "epoch": 7.04793930053711e-05,
      "step": 73903,
      "training_step_time": 8.676340103149414
    },
    {
      "epoch": 7.04803466796875e-05,
      "model_forward_time": 1.7339935302734375,
      "step": 73904
    },
    {
      "epoch": 7.04803466796875e-05,
      "step": 73904,
      "training_step_time": 7.72087287902832
    },
    {
      "epoch": 7.048130035400391e-05,
      "model_forward_time": 1.7889091968536377,
      "step": 73905
    },
    {
      "epoch": 7.048130035400391e-05,
      "step": 73905,
      "training_step_time": 10.108453035354614
    },
    {
      "epoch": 7.048225402832032e-05,
      "model_forward_time": 1.7445414066314697,
      "step": 73906
    },
    {
      "epoch": 7.048225402832032e-05,
      "step": 73906,
      "training_step_time": 7.930933475494385
    },
    {
      "epoch": 7.048320770263672e-05,
      "model_forward_time": 1.3763670921325684,
      "step": 73907
    },
    {
      "epoch": 7.048320770263672e-05,
      "step": 73907,
      "training_step_time": 7.5835840702056885
    },
    {
      "epoch": 7.048416137695313e-05,
      "model_forward_time": 1.754237174987793,
      "step": 73908
    },
    {
      "epoch": 7.048416137695313e-05,
      "step": 73908,
      "training_step_time": 6.824358940124512
    },
    {
      "epoch": 7.048511505126954e-05,
      "model_forward_time": 1.7609221935272217,
      "step": 73909
    },
    {
      "epoch": 7.048511505126954e-05,
      "step": 73909,
      "training_step_time": 6.8075830936431885
    },
    {
      "epoch": 7.048606872558593e-05,
      "grad_norm": 0.0819423720240593,
      "learning_rate": 1.7484869635392903e-06,
      "loss": 0.0575,
      "step": 73910
    },
    {
      "epoch": 7.048606872558593e-05,
      "model_forward_time": 1.758974552154541,
      "step": 73910
    },
    {
      "dynamics_loss_avg": 0.04317616522312164,
      "epoch": 7.048606872558593e-05,
      "step": 73910
    },
    {
      "action_loss_avg": 0.0009261928993510082,
      "epoch": 7.048606872558593e-05,
      "step": 73910
    },
    {
      "epoch": 7.048606872558593e-05,
      "step": 73910,
      "training_step_time": 6.776143312454224
    },
    {
      "epoch": 7.048702239990235e-05,
      "model_forward_time": 1.7287333011627197,
      "step": 73911
    },
    {
      "epoch": 7.048702239990235e-05,
      "step": 73911,
      "training_step_time": 7.602045059204102
    },
    {
      "epoch": 7.048797607421876e-05,
      "model_forward_time": 1.733461618423462,
      "step": 73912
    },
    {
      "epoch": 7.048797607421876e-05,
      "step": 73912,
      "training_step_time": 7.624038934707642
    },
    {
      "epoch": 7.048892974853515e-05,
      "model_forward_time": 1.7422692775726318,
      "step": 73913
    },
    {
      "epoch": 7.048892974853515e-05,
      "step": 73913,
      "training_step_time": 8.789257287979126
    },
    {
      "epoch": 7.048988342285156e-05,
      "model_forward_time": 1.7846617698669434,
      "step": 73914
    },
    {
      "epoch": 7.048988342285156e-05,
      "step": 73914,
      "training_step_time": 9.123859882354736
    },
    {
      "epoch": 7.049083709716797e-05,
      "model_forward_time": 2.290259838104248,
      "step": 73915
    },
    {
      "epoch": 7.049083709716797e-05,
      "step": 73915,
      "training_step_time": 8.39888072013855
    },
    {
      "epoch": 7.049179077148437e-05,
      "model_forward_time": 1.7468235492706299,
      "step": 73916
    },
    {
      "epoch": 7.049179077148437e-05,
      "step": 73916,
      "training_step_time": 7.8375279903411865
    },
    {
      "epoch": 7.049274444580078e-05,
      "model_forward_time": 1.7484588623046875,
      "step": 73917
    },
    {
      "epoch": 7.049274444580078e-05,
      "step": 73917,
      "training_step_time": 6.77143931388855
    },
    {
      "epoch": 7.04936981201172e-05,
      "model_forward_time": 1.7440803050994873,
      "step": 73918
    },
    {
      "epoch": 7.04936981201172e-05,
      "step": 73918,
      "training_step_time": 6.8174567222595215
    },
    {
      "epoch": 7.049465179443359e-05,
      "model_forward_time": 1.742605447769165,
      "step": 73919
    },
    {
      "epoch": 7.049465179443359e-05,
      "step": 73919,
      "training_step_time": 6.832020998001099
    },
    {
      "epoch": 7.049560546875e-05,
      "grad_norm": 0.10210084170103073,
      "learning_rate": 1.747231041218919e-06,
      "loss": 0.0563,
      "step": 73920
    },
    {
      "epoch": 7.049560546875e-05,
      "model_forward_time": 1.7388603687286377,
      "step": 73920
    },
    {
      "dynamics_loss_avg": 0.047311934642493725,
      "epoch": 7.049560546875e-05,
      "step": 73920
    },
    {
      "action_loss_avg": 0.0011459424000349826,
      "epoch": 7.049560546875e-05,
      "step": 73920
    },
    {
      "epoch": 7.049560546875e-05,
      "step": 73920,
      "training_step_time": 7.629053115844727
    },
    {
      "epoch": 7.049655914306641e-05,
      "model_forward_time": 1.7437758445739746,
      "step": 73921
    },
    {
      "epoch": 7.049655914306641e-05,
      "step": 73921,
      "training_step_time": 8.518287897109985
    },
    {
      "epoch": 7.049751281738281e-05,
      "model_forward_time": 1.7340924739837646,
      "step": 73922
    },
    {
      "epoch": 7.049751281738281e-05,
      "step": 73922,
      "training_step_time": 8.531229734420776
    },
    {
      "epoch": 7.049846649169922e-05,
      "model_forward_time": 1.7795004844665527,
      "step": 73923
    },
    {
      "epoch": 7.049846649169922e-05,
      "step": 73923,
      "training_step_time": 9.451494932174683
    },
    {
      "epoch": 7.049942016601563e-05,
      "model_forward_time": 2.308438301086426,
      "step": 73924
    },
    {
      "epoch": 7.049942016601563e-05,
      "step": 73924,
      "training_step_time": 7.745966911315918
    },
    {
      "epoch": 7.050037384033203e-05,
      "model_forward_time": 1.7475171089172363,
      "step": 73925
    },
    {
      "epoch": 7.050037384033203e-05,
      "step": 73925,
      "training_step_time": 7.794442653656006
    },
    {
      "epoch": 7.050132751464844e-05,
      "model_forward_time": 1.7496159076690674,
      "step": 73926
    },
    {
      "epoch": 7.050132751464844e-05,
      "step": 73926,
      "training_step_time": 7.164994716644287
    },
    {
      "epoch": 7.050228118896485e-05,
      "model_forward_time": 1.7397823333740234,
      "step": 73927
    },
    {
      "epoch": 7.050228118896485e-05,
      "step": 73927,
      "training_step_time": 6.780462026596069
    },
    {
      "epoch": 7.050323486328125e-05,
      "model_forward_time": 1.7448487281799316,
      "step": 73928
    },
    {
      "epoch": 7.050323486328125e-05,
      "step": 73928,
      "training_step_time": 6.747540473937988
    },
    {
      "epoch": 7.050418853759766e-05,
      "model_forward_time": 1.7381348609924316,
      "step": 73929
    },
    {
      "epoch": 7.050418853759766e-05,
      "step": 73929,
      "training_step_time": 7.5874011516571045
    },
    {
      "epoch": 7.050514221191407e-05,
      "grad_norm": 0.0817706510424614,
      "learning_rate": 1.7459754746164692e-06,
      "loss": 0.0561,
      "step": 73930
    },
    {
      "epoch": 7.050514221191407e-05,
      "model_forward_time": 1.7582743167877197,
      "step": 73930
    },
    {
      "dynamics_loss_avg": 0.04287466276437044,
      "epoch": 7.050514221191407e-05,
      "step": 73930
    },
    {
      "action_loss_avg": 0.0013354643888305873,
      "epoch": 7.050514221191407e-05,
      "step": 73930
    },
    {
      "epoch": 7.050514221191407e-05,
      "step": 73930,
      "training_step_time": 7.757089853286743
    },
    {
      "epoch": 7.050609588623047e-05,
      "model_forward_time": 1.735816478729248,
      "step": 73931
    },
    {
      "epoch": 7.050609588623047e-05,
      "step": 73931,
      "training_step_time": 7.644177436828613
    },
    {
      "epoch": 7.050704956054688e-05,
      "model_forward_time": 1.7941782474517822,
      "step": 73932
    },
    {
      "epoch": 7.050704956054688e-05,
      "step": 73932,
      "training_step_time": 9.894093036651611
    },
    {
      "epoch": 7.050800323486329e-05,
      "model_forward_time": 2.320324420928955,
      "step": 73933
    },
    {
      "epoch": 7.050800323486329e-05,
      "step": 73933,
      "training_step_time": 9.5003981590271
    },
    {
      "epoch": 7.050895690917969e-05,
      "model_forward_time": 1.7624070644378662,
      "step": 73934
    },
    {
      "epoch": 7.050895690917969e-05,
      "step": 73934,
      "training_step_time": 7.740001916885376
    },
    {
      "epoch": 7.05099105834961e-05,
      "model_forward_time": 1.7445924282073975,
      "step": 73935
    },
    {
      "epoch": 7.05099105834961e-05,
      "step": 73935,
      "training_step_time": 7.261826515197754
    },
    {
      "epoch": 7.05108642578125e-05,
      "model_forward_time": 1.7529587745666504,
      "step": 73936
    },
    {
      "epoch": 7.05108642578125e-05,
      "step": 73936,
      "training_step_time": 6.847494840621948
    },
    {
      "epoch": 7.05118179321289e-05,
      "model_forward_time": 1.742875099182129,
      "step": 73937
    },
    {
      "epoch": 7.05118179321289e-05,
      "step": 73937,
      "training_step_time": 6.876642465591431
    },
    {
      "epoch": 7.051277160644531e-05,
      "model_forward_time": 1.7491042613983154,
      "step": 73938
    },
    {
      "epoch": 7.051277160644531e-05,
      "step": 73938,
      "training_step_time": 7.7387309074401855
    },
    {
      "epoch": 7.051372528076171e-05,
      "model_forward_time": 1.7407853603363037,
      "step": 73939
    },
    {
      "epoch": 7.051372528076171e-05,
      "step": 73939,
      "training_step_time": 8.16412901878357
    },
    {
      "epoch": 7.051467895507812e-05,
      "grad_norm": 0.08836839348077774,
      "learning_rate": 1.744720263869249e-06,
      "loss": 0.0536,
      "step": 73940
    },
    {
      "epoch": 7.051467895507812e-05,
      "model_forward_time": 1.7353110313415527,
      "step": 73940
    },
    {
      "dynamics_loss_avg": 0.046840973943471906,
      "epoch": 7.051467895507812e-05,
      "step": 73940
    },
    {
      "action_loss_avg": 0.0014922186150215567,
      "epoch": 7.051467895507812e-05,
      "step": 73940
    },
    {
      "epoch": 7.051467895507812e-05,
      "step": 73940,
      "training_step_time": 8.47688341140747
    },
    {
      "epoch": 7.051563262939453e-05,
      "model_forward_time": 1.7805190086364746,
      "step": 73941
    },
    {
      "epoch": 7.051563262939453e-05,
      "step": 73941,
      "training_step_time": 9.031362771987915
    },
    {
      "epoch": 7.051658630371093e-05,
      "model_forward_time": 2.301259756088257,
      "step": 73942
    },
    {
      "epoch": 7.051658630371093e-05,
      "step": 73942,
      "training_step_time": 8.88941502571106
    },
    {
      "epoch": 7.051753997802734e-05,
      "model_forward_time": 1.742471694946289,
      "step": 73943
    },
    {
      "epoch": 7.051753997802734e-05,
      "step": 73943,
      "training_step_time": 7.9665749073028564
    },
    {
      "epoch": 7.051849365234375e-05,
      "model_forward_time": 1.748593807220459,
      "step": 73944
    },
    {
      "epoch": 7.051849365234375e-05,
      "step": 73944,
      "training_step_time": 7.176481485366821
    },
    {
      "epoch": 7.051944732666015e-05,
      "model_forward_time": 1.7348804473876953,
      "step": 73945
    },
    {
      "epoch": 7.051944732666015e-05,
      "step": 73945,
      "training_step_time": 6.780574798583984
    },
    {
      "epoch": 7.052040100097656e-05,
      "model_forward_time": 1.7489948272705078,
      "step": 73946
    },
    {
      "epoch": 7.052040100097656e-05,
      "step": 73946,
      "training_step_time": 6.830235481262207
    },
    {
      "epoch": 7.052135467529297e-05,
      "model_forward_time": 1.7422261238098145,
      "step": 73947
    },
    {
      "epoch": 7.052135467529297e-05,
      "step": 73947,
      "training_step_time": 7.646591901779175
    },
    {
      "epoch": 7.052230834960937e-05,
      "model_forward_time": 1.7274038791656494,
      "step": 73948
    },
    {
      "epoch": 7.052230834960937e-05,
      "step": 73948,
      "training_step_time": 7.653048992156982
    },
    {
      "epoch": 7.052326202392578e-05,
      "model_forward_time": 1.7511539459228516,
      "step": 73949
    },
    {
      "epoch": 7.052326202392578e-05,
      "step": 73949,
      "training_step_time": 7.558789491653442
    },
    {
      "epoch": 7.052421569824219e-05,
      "grad_norm": 0.08382149040699005,
      "learning_rate": 1.7434654091145264e-06,
      "loss": 0.0526,
      "step": 73950
    },
    {
      "epoch": 7.052421569824219e-05,
      "model_forward_time": 1.7408528327941895,
      "step": 73950
    },
    {
      "dynamics_loss_avg": 0.0586523536592722,
      "epoch": 7.052421569824219e-05,
      "step": 73950
    },
    {
      "action_loss_avg": 0.0017676370160188525,
      "epoch": 7.052421569824219e-05,
      "step": 73950
    },
    {
      "epoch": 7.052421569824219e-05,
      "step": 73950,
      "training_step_time": 8.614976167678833
    },
    {
      "epoch": 7.052516937255859e-05,
      "model_forward_time": 1.7281131744384766,
      "step": 73951
    },
    {
      "epoch": 7.052516937255859e-05,
      "step": 73951,
      "training_step_time": 9.582645654678345
    },
    {
      "epoch": 7.0526123046875e-05,
      "model_forward_time": 1.727701187133789,
      "step": 73952
    },
    {
      "epoch": 7.0526123046875e-05,
      "step": 73952,
      "training_step_time": 8.66426968574524
    },
    {
      "epoch": 7.052707672119141e-05,
      "model_forward_time": 1.7257604598999023,
      "step": 73953
    },
    {
      "epoch": 7.052707672119141e-05,
      "step": 73953,
      "training_step_time": 7.6555681228637695
    },
    {
      "epoch": 7.05280303955078e-05,
      "model_forward_time": 1.7251644134521484,
      "step": 73954
    },
    {
      "epoch": 7.05280303955078e-05,
      "step": 73954,
      "training_step_time": 6.781432151794434
    },
    {
      "epoch": 7.052898406982422e-05,
      "model_forward_time": 1.729064702987671,
      "step": 73955
    },
    {
      "epoch": 7.052898406982422e-05,
      "step": 73955,
      "training_step_time": 6.846442937850952
    },
    {
      "epoch": 7.052993774414063e-05,
      "model_forward_time": 1.7123222351074219,
      "step": 73956
    },
    {
      "epoch": 7.052993774414063e-05,
      "step": 73956,
      "training_step_time": 7.717018365859985
    },
    {
      "epoch": 7.053089141845703e-05,
      "model_forward_time": 1.7294697761535645,
      "step": 73957
    },
    {
      "epoch": 7.053089141845703e-05,
      "step": 73957,
      "training_step_time": 7.629462242126465
    },
    {
      "epoch": 7.053184509277344e-05,
      "model_forward_time": 1.7504160404205322,
      "step": 73958
    },
    {
      "epoch": 7.053184509277344e-05,
      "step": 73958,
      "training_step_time": 7.671256065368652
    },
    {
      "epoch": 7.053279876708985e-05,
      "model_forward_time": 1.740356683731079,
      "step": 73959
    },
    {
      "epoch": 7.053279876708985e-05,
      "step": 73959,
      "training_step_time": 9.937717914581299
    },
    {
      "epoch": 7.053375244140624e-05,
      "grad_norm": 0.08516013622283936,
      "learning_rate": 1.7422109104895319e-06,
      "loss": 0.0514,
      "step": 73960
    },
    {
      "epoch": 7.053375244140624e-05,
      "model_forward_time": 1.7334904670715332,
      "step": 73960
    },
    {
      "dynamics_loss_avg": 0.05508131599053741,
      "epoch": 7.053375244140624e-05,
      "step": 73960
    },
    {
      "action_loss_avg": 0.001684637792641297,
      "epoch": 7.053375244140624e-05,
      "step": 73960
    },
    {
      "epoch": 7.053375244140624e-05,
      "step": 73960,
      "training_step_time": 8.531660318374634
    },
    {
      "epoch": 7.053470611572265e-05,
      "model_forward_time": 2.9347355365753174,
      "step": 73961
    },
    {
      "epoch": 7.053470611572265e-05,
      "step": 73961,
      "training_step_time": 7.82625150680542
    },
    {
      "epoch": 7.053565979003907e-05,
      "model_forward_time": 1.7256824970245361,
      "step": 73962
    },
    {
      "epoch": 7.053565979003907e-05,
      "step": 73962,
      "training_step_time": 8.102275371551514
    },
    {
      "epoch": 7.053661346435546e-05,
      "model_forward_time": 1.726593017578125,
      "step": 73963
    },
    {
      "epoch": 7.053661346435546e-05,
      "step": 73963,
      "training_step_time": 7.345621347427368
    },
    {
      "epoch": 7.053756713867187e-05,
      "model_forward_time": 1.7284581661224365,
      "step": 73964
    },
    {
      "epoch": 7.053756713867187e-05,
      "step": 73964,
      "training_step_time": 6.939271688461304
    },
    {
      "epoch": 7.053852081298828e-05,
      "model_forward_time": 1.3579084873199463,
      "step": 73965
    },
    {
      "epoch": 7.053852081298828e-05,
      "step": 73965,
      "training_step_time": 7.7014336585998535
    },
    {
      "epoch": 7.053947448730468e-05,
      "model_forward_time": 1.7254135608673096,
      "step": 73966
    },
    {
      "epoch": 7.053947448730468e-05,
      "step": 73966,
      "training_step_time": 8.419636487960815
    },
    {
      "epoch": 7.054042816162109e-05,
      "model_forward_time": 1.7486670017242432,
      "step": 73967
    },
    {
      "epoch": 7.054042816162109e-05,
      "step": 73967,
      "training_step_time": 7.693587779998779
    },
    {
      "epoch": 7.05413818359375e-05,
      "model_forward_time": 1.750495195388794,
      "step": 73968
    },
    {
      "epoch": 7.05413818359375e-05,
      "step": 73968,
      "training_step_time": 8.724947214126587
    },
    {
      "epoch": 7.05423355102539e-05,
      "model_forward_time": 1.7245042324066162,
      "step": 73969
    },
    {
      "epoch": 7.05423355102539e-05,
      "step": 73969,
      "training_step_time": 10.525550365447998
    },
    {
      "epoch": 7.054328918457031e-05,
      "grad_norm": 0.10257910192012787,
      "learning_rate": 1.7409567681314532e-06,
      "loss": 0.0573,
      "step": 73970
    },
    {
      "epoch": 7.054328918457031e-05,
      "model_forward_time": 2.969330310821533,
      "step": 73970
    },
    {
      "dynamics_loss_avg": 0.05536047574132681,
      "epoch": 7.054328918457031e-05,
      "step": 73970
    },
    {
      "action_loss_avg": 0.0009935835347278043,
      "epoch": 7.054328918457031e-05,
      "step": 73970
    },
    {
      "epoch": 7.054328918457031e-05,
      "step": 73970,
      "training_step_time": 8.459724426269531
    },
    {
      "epoch": 7.054424285888672e-05,
      "model_forward_time": 1.7113463878631592,
      "step": 73971
    },
    {
      "epoch": 7.054424285888672e-05,
      "step": 73971,
      "training_step_time": 9.056953191757202
    },
    {
      "epoch": 7.054519653320312e-05,
      "model_forward_time": 1.7217798233032227,
      "step": 73972
    },
    {
      "epoch": 7.054519653320312e-05,
      "step": 73972,
      "training_step_time": 7.95222020149231
    },
    {
      "epoch": 7.054615020751953e-05,
      "model_forward_time": 1.7206921577453613,
      "step": 73973
    },
    {
      "epoch": 7.054615020751953e-05,
      "step": 73973,
      "training_step_time": 7.007267475128174
    },
    {
      "epoch": 7.054710388183594e-05,
      "model_forward_time": 1.7225029468536377,
      "step": 73974
    },
    {
      "epoch": 7.054710388183594e-05,
      "step": 73974,
      "training_step_time": 7.818631649017334
    },
    {
      "epoch": 7.054805755615234e-05,
      "model_forward_time": 1.7215476036071777,
      "step": 73975
    },
    {
      "epoch": 7.054805755615234e-05,
      "step": 73975,
      "training_step_time": 7.472288131713867
    },
    {
      "epoch": 7.054901123046875e-05,
      "model_forward_time": 1.751326084136963,
      "step": 73976
    },
    {
      "epoch": 7.054901123046875e-05,
      "step": 73976,
      "training_step_time": 7.639436721801758
    },
    {
      "epoch": 7.054996490478516e-05,
      "model_forward_time": 1.7484030723571777,
      "step": 73977
    },
    {
      "epoch": 7.054996490478516e-05,
      "step": 73977,
      "training_step_time": 7.62938928604126
    },
    {
      "epoch": 7.055091857910156e-05,
      "model_forward_time": 1.718808889389038,
      "step": 73978
    },
    {
      "epoch": 7.055091857910156e-05,
      "step": 73978,
      "training_step_time": 9.25347900390625
    },
    {
      "epoch": 7.055187225341797e-05,
      "model_forward_time": 2.540539503097534,
      "step": 73979
    },
    {
      "epoch": 7.055187225341797e-05,
      "step": 73979,
      "training_step_time": 9.405649185180664
    },
    {
      "epoch": 7.055282592773438e-05,
      "grad_norm": 0.1009674146771431,
      "learning_rate": 1.7397029821774435e-06,
      "loss": 0.0539,
      "step": 73980
    },
    {
      "epoch": 7.055282592773438e-05,
      "model_forward_time": 1.720914602279663,
      "step": 73980
    },
    {
      "dynamics_loss_avg": 0.04946730826050043,
      "epoch": 7.055282592773438e-05,
      "step": 73980
    },
    {
      "action_loss_avg": 0.0010120458988239988,
      "epoch": 7.055282592773438e-05,
      "step": 73980
    },
    {
      "epoch": 7.055282592773438e-05,
      "step": 73980,
      "training_step_time": 8.092981100082397
    },
    {
      "epoch": 7.055377960205078e-05,
      "model_forward_time": 1.72568678855896,
      "step": 73981
    },
    {
      "epoch": 7.055377960205078e-05,
      "step": 73981,
      "training_step_time": 7.763535022735596
    },
    {
      "epoch": 7.055473327636719e-05,
      "model_forward_time": 1.720750093460083,
      "step": 73982
    },
    {
      "epoch": 7.055473327636719e-05,
      "step": 73982,
      "training_step_time": 6.863189697265625
    },
    {
      "epoch": 7.05556869506836e-05,
      "model_forward_time": 1.722435474395752,
      "step": 73983
    },
    {
      "epoch": 7.05556869506836e-05,
      "step": 73983,
      "training_step_time": 7.625635147094727
    },
    {
      "epoch": 7.0556640625e-05,
      "model_forward_time": 1.7204396724700928,
      "step": 73984
    },
    {
      "epoch": 7.0556640625e-05,
      "step": 73984,
      "training_step_time": 7.371722936630249
    },
    {
      "epoch": 7.05575942993164e-05,
      "model_forward_time": 1.7441835403442383,
      "step": 73985
    },
    {
      "epoch": 7.05575942993164e-05,
      "step": 73985,
      "training_step_time": 7.640862941741943
    },
    {
      "epoch": 7.055854797363282e-05,
      "model_forward_time": 1.739919900894165,
      "step": 73986
    },
    {
      "epoch": 7.055854797363282e-05,
      "step": 73986,
      "training_step_time": 7.632964372634888
    },
    {
      "epoch": 7.055950164794921e-05,
      "model_forward_time": 1.7236313819885254,
      "step": 73987
    },
    {
      "epoch": 7.055950164794921e-05,
      "step": 73987,
      "training_step_time": 8.580624103546143
    },
    {
      "epoch": 7.056045532226562e-05,
      "model_forward_time": 2.8606932163238525,
      "step": 73988
    },
    {
      "epoch": 7.056045532226562e-05,
      "step": 73988,
      "training_step_time": 7.806031703948975
    },
    {
      "epoch": 7.056140899658204e-05,
      "model_forward_time": 1.7284517288208008,
      "step": 73989
    },
    {
      "epoch": 7.056140899658204e-05,
      "step": 73989,
      "training_step_time": 8.970962047576904
    },
    {
      "epoch": 7.056236267089843e-05,
      "grad_norm": 0.08967694640159607,
      "learning_rate": 1.7384495527646127e-06,
      "loss": 0.0571,
      "step": 73990
    },
    {
      "epoch": 7.056236267089843e-05,
      "model_forward_time": 1.7257461547851562,
      "step": 73990
    },
    {
      "dynamics_loss_avg": 0.04995300117880106,
      "epoch": 7.056236267089843e-05,
      "step": 73990
    },
    {
      "action_loss_avg": 0.0006615658247028477,
      "epoch": 7.056236267089843e-05,
      "step": 73990
    },
    {
      "epoch": 7.056236267089843e-05,
      "step": 73990,
      "training_step_time": 7.78869366645813
    },
    {
      "epoch": 7.056331634521484e-05,
      "model_forward_time": 1.7206721305847168,
      "step": 73991
    },
    {
      "epoch": 7.056331634521484e-05,
      "step": 73991,
      "training_step_time": 6.7991108894348145
    },
    {
      "epoch": 7.056427001953125e-05,
      "model_forward_time": 1.7218079566955566,
      "step": 73992
    },
    {
      "epoch": 7.056427001953125e-05,
      "step": 73992,
      "training_step_time": 7.6350791454315186
    },
    {
      "epoch": 7.056522369384765e-05,
      "model_forward_time": 1.7207772731781006,
      "step": 73993
    },
    {
      "epoch": 7.056522369384765e-05,
      "step": 73993,
      "training_step_time": 7.280531644821167
    },
    {
      "epoch": 7.056617736816406e-05,
      "model_forward_time": 1.7477667331695557,
      "step": 73994
    },
    {
      "epoch": 7.056617736816406e-05,
      "step": 73994,
      "training_step_time": 8.345593929290771
    },
    {
      "epoch": 7.056713104248047e-05,
      "model_forward_time": 1.7226009368896484,
      "step": 73995
    },
    {
      "epoch": 7.056713104248047e-05,
      "step": 73995,
      "training_step_time": 8.175208330154419
    },
    {
      "epoch": 7.056808471679687e-05,
      "model_forward_time": 1.7286512851715088,
      "step": 73996
    },
    {
      "epoch": 7.056808471679687e-05,
      "step": 73996,
      "training_step_time": 8.447578430175781
    },
    {
      "epoch": 7.056903839111328e-05,
      "model_forward_time": 2.827831983566284,
      "step": 73997
    },
    {
      "epoch": 7.056903839111328e-05,
      "step": 73997,
      "training_step_time": 8.584634780883789
    },
    {
      "epoch": 7.056999206542969e-05,
      "model_forward_time": 1.7686586380004883,
      "step": 73998
    },
    {
      "epoch": 7.056999206542969e-05,
      "step": 73998,
      "training_step_time": 8.733521223068237
    },
    {
      "epoch": 7.057094573974609e-05,
      "model_forward_time": 1.3493824005126953,
      "step": 73999
    },
    {
      "epoch": 7.057094573974609e-05,
      "step": 73999,
      "training_step_time": 7.747524261474609
    },
    {
      "epoch": 7.05718994140625e-05,
      "grad_norm": 0.09563440829515457,
      "learning_rate": 1.737196480030035e-06,
      "loss": 0.0545,
      "step": 74000
    },
    {
      "epoch": 7.05718994140625e-05,
      "model_forward_time": 5.016217947006226,
      "step": 74000
    },
    {
      "dynamics_loss_avg": 0.10365644097328186,
      "epoch": 7.05718994140625e-05,
      "step": 74000
    },
    {
      "action_loss_avg": 0.0007067194092087448,
      "epoch": 7.05718994140625e-05,
      "step": 74000
    },
    {
      "epoch": 7.05718994140625e-05,
      "step": 74000,
      "training_step_time": 15.19653868675232
    },
    {
      "epoch": 9.5367431640625e-10,
      "model_forward_time": 1.7649779319763184,
      "step": 74001
    },
    {
      "epoch": 9.5367431640625e-10,
      "step": 74001,
      "training_step_time": 6.849234104156494
    },
    {
      "epoch": 1.9073486328125e-09,
      "model_forward_time": 1.713805913925171,
      "step": 74002
    },
    {
      "epoch": 1.9073486328125e-09,
      "step": 74002,
      "training_step_time": 6.984179973602295
    },
    {
      "epoch": 2.86102294921875e-09,
      "model_forward_time": 1.7311131954193115,
      "step": 74003
    },
    {
      "epoch": 2.86102294921875e-09,
      "step": 74003,
      "training_step_time": 6.94157075881958
    },
    {
      "epoch": 3.814697265625e-09,
      "model_forward_time": 1.7287638187408447,
      "step": 74004
    },
    {
      "epoch": 3.814697265625e-09,
      "step": 74004,
      "training_step_time": 6.958518743515015
    },
    {
      "epoch": 4.76837158203125e-09,
      "model_forward_time": 1.7316944599151611,
      "step": 74005
    },
    {
      "epoch": 4.76837158203125e-09,
      "step": 74005,
      "training_step_time": 6.828009843826294
    },
    {
      "epoch": 5.7220458984375e-09,
      "model_forward_time": 1.728895902633667,
      "step": 74006
    },
    {
      "epoch": 5.7220458984375e-09,
      "step": 74006,
      "training_step_time": 6.914621114730835
    },
    {
      "epoch": 6.67572021484375e-09,
      "model_forward_time": 2.9319679737091064,
      "step": 74007
    },
    {
      "epoch": 6.67572021484375e-09,
      "step": 74007,
      "training_step_time": 9.361904382705688
    },
    {
      "epoch": 7.62939453125e-09,
      "model_forward_time": 1.7227234840393066,
      "step": 74008
    },
    {
      "epoch": 7.62939453125e-09,
      "step": 74008,
      "training_step_time": 6.9989118576049805
    },
    {
      "epoch": 8.58306884765625e-09,
      "model_forward_time": 1.7382102012634277,
      "step": 74009
    },
    {
      "epoch": 8.58306884765625e-09,
      "step": 74009,
      "training_step_time": 6.962563514709473
    },
    {
      "epoch": 9.5367431640625e-09,
      "grad_norm": 0.09466995298862457,
      "learning_rate": 1.7359437641107446e-06,
      "loss": 0.0558,
      "step": 74010
    },
    {
      "epoch": 9.5367431640625e-09,
      "model_forward_time": 1.7291831970214844,
      "step": 74010
    },
    {
      "dynamics_loss_avg": 0.0609141212888062,
      "epoch": 9.5367431640625e-09,
      "step": 74010
    },
    {
      "action_loss_avg": 0.0025483501653070562,
      "epoch": 9.5367431640625e-09,
      "step": 74010
    },
    {
      "epoch": 9.5367431640625e-09,
      "step": 74010,
      "training_step_time": 6.954914808273315
    },
    {
      "epoch": 1.049041748046875e-08,
      "model_forward_time": 1.739668369293213,
      "step": 74011
    },
    {
      "epoch": 1.049041748046875e-08,
      "step": 74011,
      "training_step_time": 7.064329385757446
    },
    {
      "epoch": 1.1444091796875e-08,
      "model_forward_time": 1.7388975620269775,
      "step": 74012
    },
    {
      "epoch": 1.1444091796875e-08,
      "step": 74012,
      "training_step_time": 6.9909348487854
    },
    {
      "epoch": 1.239776611328125e-08,
      "model_forward_time": 1.7425081729888916,
      "step": 74013
    },
    {
      "epoch": 1.239776611328125e-08,
      "step": 74013,
      "training_step_time": 6.996542453765869
    },
    {
      "epoch": 1.33514404296875e-08,
      "model_forward_time": 1.7448854446411133,
      "step": 74014
    },
    {
      "epoch": 1.33514404296875e-08,
      "step": 74014,
      "training_step_time": 6.988260269165039
    },
    {
      "epoch": 1.430511474609375e-08,
      "model_forward_time": 1.7392911911010742,
      "step": 74015
    },
    {
      "epoch": 1.430511474609375e-08,
      "step": 74015,
      "training_step_time": 6.986462354660034
    },
    {
      "epoch": 1.52587890625e-08,
      "model_forward_time": 1.7767512798309326,
      "step": 74016
    },
    {
      "epoch": 1.52587890625e-08,
      "step": 74016,
      "training_step_time": 11.326351642608643
    },
    {
      "epoch": 1.621246337890625e-08,
      "model_forward_time": 1.7410361766815186,
      "step": 74017
    },
    {
      "epoch": 1.621246337890625e-08,
      "step": 74017,
      "training_step_time": 7.394773244857788
    },
    {
      "epoch": 1.71661376953125e-08,
      "model_forward_time": 1.740002155303955,
      "step": 74018
    },
    {
      "epoch": 1.71661376953125e-08,
      "step": 74018,
      "training_step_time": 6.887346982955933
    },
    {
      "epoch": 1.811981201171875e-08,
      "model_forward_time": 1.74772047996521,
      "step": 74019
    },
    {
      "epoch": 1.811981201171875e-08,
      "step": 74019,
      "training_step_time": 6.956576347351074
    },
    {
      "epoch": 1.9073486328125e-08,
      "grad_norm": 0.07621399313211441,
      "learning_rate": 1.7346914051437387e-06,
      "loss": 0.0545,
      "step": 74020
    },
    {
      "epoch": 1.9073486328125e-08,
      "model_forward_time": 1.735213041305542,
      "step": 74020
    },
    {
      "dynamics_loss_avg": 0.051042168587446216,
      "epoch": 1.9073486328125e-08,
      "step": 74020
    },
    {
      "action_loss_avg": 0.0023970015172380954,
      "epoch": 1.9073486328125e-08,
      "step": 74020
    },
    {
      "epoch": 1.9073486328125e-08,
      "step": 74020,
      "training_step_time": 6.995993614196777
    },
    {
      "epoch": 2.002716064453125e-08,
      "model_forward_time": 1.744809865951538,
      "step": 74021
    },
    {
      "epoch": 2.002716064453125e-08,
      "step": 74021,
      "training_step_time": 6.834853887557983
    },
    {
      "epoch": 2.09808349609375e-08,
      "model_forward_time": 1.737922191619873,
      "step": 74022
    },
    {
      "epoch": 2.09808349609375e-08,
      "step": 74022,
      "training_step_time": 6.846528768539429
    },
    {
      "epoch": 2.193450927734375e-08,
      "model_forward_time": 1.737381935119629,
      "step": 74023
    },
    {
      "epoch": 2.193450927734375e-08,
      "step": 74023,
      "training_step_time": 6.869318246841431
    },
    {
      "epoch": 2.288818359375e-08,
      "model_forward_time": 1.742856740951538,
      "step": 74024
    },
    {
      "epoch": 2.288818359375e-08,
      "step": 74024,
      "training_step_time": 6.90291166305542
    },
    {
      "epoch": 2.384185791015625e-08,
      "model_forward_time": 1.760416030883789,
      "step": 74025
    },
    {
      "epoch": 2.384185791015625e-08,
      "step": 74025,
      "training_step_time": 12.334285497665405
    },
    {
      "epoch": 2.47955322265625e-08,
      "model_forward_time": 2.35597562789917,
      "step": 74026
    },
    {
      "epoch": 2.47955322265625e-08,
      "step": 74026,
      "training_step_time": 7.796460866928101
    },
    {
      "epoch": 2.574920654296875e-08,
      "model_forward_time": 1.7445955276489258,
      "step": 74027
    },
    {
      "epoch": 2.574920654296875e-08,
      "step": 74027,
      "training_step_time": 6.946976900100708
    },
    {
      "epoch": 2.6702880859375e-08,
      "model_forward_time": 1.7304346561431885,
      "step": 74028
    },
    {
      "epoch": 2.6702880859375e-08,
      "step": 74028,
      "training_step_time": 6.925827264785767
    },
    {
      "epoch": 2.765655517578125e-08,
      "model_forward_time": 1.7218029499053955,
      "step": 74029
    },
    {
      "epoch": 2.765655517578125e-08,
      "step": 74029,
      "training_step_time": 6.938302755355835
    },
    {
      "epoch": 2.86102294921875e-08,
      "grad_norm": 0.09524060785770416,
      "learning_rate": 1.733439403265969e-06,
      "loss": 0.0552,
      "step": 74030
    },
    {
      "epoch": 2.86102294921875e-08,
      "model_forward_time": 1.7338171005249023,
      "step": 74030
    },
    {
      "dynamics_loss_avg": 0.0465045053511858,
      "epoch": 2.86102294921875e-08,
      "step": 74030
    },
    {
      "action_loss_avg": 0.0012700129693257623,
      "epoch": 2.86102294921875e-08,
      "step": 74030
    },
    {
      "epoch": 2.86102294921875e-08,
      "step": 74030,
      "training_step_time": 6.932214260101318
    },
    {
      "epoch": 2.956390380859375e-08,
      "model_forward_time": 1.7245993614196777,
      "step": 74031
    },
    {
      "epoch": 2.956390380859375e-08,
      "step": 74031,
      "training_step_time": 7.0177130699157715
    },
    {
      "epoch": 3.0517578125e-08,
      "model_forward_time": 1.7429637908935547,
      "step": 74032
    },
    {
      "epoch": 3.0517578125e-08,
      "step": 74032,
      "training_step_time": 6.961080551147461
    },
    {
      "epoch": 3.147125244140625e-08,
      "model_forward_time": 1.7404284477233887,
      "step": 74033
    },
    {
      "epoch": 3.147125244140625e-08,
      "step": 74033,
      "training_step_time": 6.93918251991272
    },
    {
      "epoch": 3.24249267578125e-08,
      "model_forward_time": 1.7415125370025635,
      "step": 74034
    },
    {
      "epoch": 3.24249267578125e-08,
      "step": 74034,
      "training_step_time": 8.8381929397583
    },
    {
      "epoch": 3.337860107421875e-08,
      "model_forward_time": 2.3558692932128906,
      "step": 74035
    },
    {
      "epoch": 3.337860107421875e-08,
      "step": 74035,
      "training_step_time": 7.880184173583984
    },
    {
      "epoch": 3.4332275390625e-08,
      "model_forward_time": 1.7380177974700928,
      "step": 74036
    },
    {
      "epoch": 3.4332275390625e-08,
      "step": 74036,
      "training_step_time": 7.477871656417847
    },
    {
      "epoch": 3.528594970703125e-08,
      "model_forward_time": 1.746030569076538,
      "step": 74037
    },
    {
      "epoch": 3.528594970703125e-08,
      "step": 74037,
      "training_step_time": 6.992622375488281
    },
    {
      "epoch": 3.62396240234375e-08,
      "model_forward_time": 1.7424850463867188,
      "step": 74038
    },
    {
      "epoch": 3.62396240234375e-08,
      "step": 74038,
      "training_step_time": 6.907994985580444
    },
    {
      "epoch": 3.719329833984375e-08,
      "model_forward_time": 1.7436413764953613,
      "step": 74039
    },
    {
      "epoch": 3.719329833984375e-08,
      "step": 74039,
      "training_step_time": 7.036832094192505
    },
    {
      "epoch": 3.814697265625e-08,
      "grad_norm": 0.10571271181106567,
      "learning_rate": 1.732187758614357e-06,
      "loss": 0.056,
      "step": 74040
    },
    {
      "epoch": 3.814697265625e-08,
      "model_forward_time": 1.7447035312652588,
      "step": 74040
    },
    {
      "dynamics_loss_avg": 0.0586027305573225,
      "epoch": 3.814697265625e-08,
      "step": 74040
    },
    {
      "action_loss_avg": 0.0016754188982304186,
      "epoch": 3.814697265625e-08,
      "step": 74040
    },
    {
      "epoch": 3.814697265625e-08,
      "step": 74040,
      "training_step_time": 6.980261564254761
    },
    {
      "epoch": 3.910064697265625e-08,
      "model_forward_time": 1.747018575668335,
      "step": 74041
    },
    {
      "epoch": 3.910064697265625e-08,
      "step": 74041,
      "training_step_time": 7.020944833755493
    },
    {
      "epoch": 4.00543212890625e-08,
      "model_forward_time": 1.744504451751709,
      "step": 74042
    },
    {
      "epoch": 4.00543212890625e-08,
      "step": 74042,
      "training_step_time": 6.980841398239136
    },
    {
      "epoch": 4.100799560546875e-08,
      "model_forward_time": 1.7473857402801514,
      "step": 74043
    },
    {
      "epoch": 4.100799560546875e-08,
      "step": 74043,
      "training_step_time": 7.03584623336792
    },
    {
      "epoch": 4.1961669921875e-08,
      "model_forward_time": 2.3825020790100098,
      "step": 74044
    },
    {
      "epoch": 4.1961669921875e-08,
      "step": 74044,
      "training_step_time": 8.746682167053223
    },
    {
      "epoch": 4.291534423828125e-08,
      "model_forward_time": 1.7545785903930664,
      "step": 74045
    },
    {
      "epoch": 4.291534423828125e-08,
      "step": 74045,
      "training_step_time": 7.422708988189697
    },
    {
      "epoch": 4.38690185546875e-08,
      "model_forward_time": 1.746394395828247,
      "step": 74046
    },
    {
      "epoch": 4.38690185546875e-08,
      "step": 74046,
      "training_step_time": 6.934187412261963
    },
    {
      "epoch": 4.482269287109375e-08,
      "model_forward_time": 1.387479305267334,
      "step": 74047
    },
    {
      "epoch": 4.482269287109375e-08,
      "step": 74047,
      "training_step_time": 6.971286058425903
    },
    {
      "epoch": 4.57763671875e-08,
      "model_forward_time": 1.744321584701538,
      "step": 74048
    },
    {
      "epoch": 4.57763671875e-08,
      "step": 74048,
      "training_step_time": 6.9142420291900635
    },
    {
      "epoch": 4.673004150390625e-08,
      "model_forward_time": 1.7432334423065186,
      "step": 74049
    },
    {
      "epoch": 4.673004150390625e-08,
      "step": 74049,
      "training_step_time": 6.910727262496948
    },
    {
      "epoch": 4.76837158203125e-08,
      "grad_norm": 0.12935243546962738,
      "learning_rate": 1.730936471325777e-06,
      "loss": 0.0556,
      "step": 74050
    },
    {
      "epoch": 4.76837158203125e-08,
      "model_forward_time": 1.749617338180542,
      "step": 74050
    },
    {
      "dynamics_loss_avg": 0.06606370359659194,
      "epoch": 4.76837158203125e-08,
      "step": 74050
    },
    {
      "action_loss_avg": 0.0010491935769096016,
      "epoch": 4.76837158203125e-08,
      "step": 74050
    },
    {
      "epoch": 4.76837158203125e-08,
      "step": 74050,
      "training_step_time": 6.956241607666016
    },
    {
      "epoch": 4.863739013671875e-08,
      "model_forward_time": 1.7196450233459473,
      "step": 74051
    },
    {
      "epoch": 4.863739013671875e-08,
      "step": 74051,
      "training_step_time": 6.999098300933838
    },
    {
      "epoch": 4.9591064453125e-08,
      "model_forward_time": 1.7444579601287842,
      "step": 74052
    },
    {
      "epoch": 4.9591064453125e-08,
      "step": 74052,
      "training_step_time": 7.930657148361206
    },
    {
      "epoch": 5.054473876953125e-08,
      "model_forward_time": 2.3831067085266113,
      "step": 74053
    },
    {
      "epoch": 5.054473876953125e-08,
      "step": 74053,
      "training_step_time": 9.477396965026855
    },
    {
      "epoch": 5.14984130859375e-08,
      "model_forward_time": 1.7479870319366455,
      "step": 74054
    },
    {
      "epoch": 5.14984130859375e-08,
      "step": 74054,
      "training_step_time": 8.38476276397705
    },
    {
      "epoch": 5.245208740234375e-08,
      "model_forward_time": 1.7431495189666748,
      "step": 74055
    },
    {
      "epoch": 5.245208740234375e-08,
      "step": 74055,
      "training_step_time": 6.962852239608765
    },
    {
      "epoch": 5.340576171875e-08,
      "model_forward_time": 1.742710828781128,
      "step": 74056
    },
    {
      "epoch": 5.340576171875e-08,
      "step": 74056,
      "training_step_time": 6.918966293334961
    },
    {
      "epoch": 5.435943603515625e-08,
      "model_forward_time": 1.7440729141235352,
      "step": 74057
    },
    {
      "epoch": 5.435943603515625e-08,
      "step": 74057,
      "training_step_time": 6.906028985977173
    },
    {
      "epoch": 5.53131103515625e-08,
      "model_forward_time": 1.7416999340057373,
      "step": 74058
    },
    {
      "epoch": 5.53131103515625e-08,
      "step": 74058,
      "training_step_time": 6.941385269165039
    },
    {
      "epoch": 5.626678466796875e-08,
      "model_forward_time": 1.7353482246398926,
      "step": 74059
    },
    {
      "epoch": 5.626678466796875e-08,
      "step": 74059,
      "training_step_time": 6.877087354660034
    },
    {
      "epoch": 5.7220458984375e-08,
      "grad_norm": 0.07722190022468567,
      "learning_rate": 1.7296855415370717e-06,
      "loss": 0.0561,
      "step": 74060
    },
    {
      "epoch": 5.7220458984375e-08,
      "model_forward_time": 1.7432661056518555,
      "step": 74060
    },
    {
      "dynamics_loss_avg": 0.060714025236666205,
      "epoch": 5.7220458984375e-08,
      "step": 74060
    },
    {
      "action_loss_avg": 0.0009606811960111372,
      "epoch": 5.7220458984375e-08,
      "step": 74060
    },
    {
      "epoch": 5.7220458984375e-08,
      "step": 74060,
      "training_step_time": 6.914443492889404
    },
    {
      "epoch": 5.817413330078125e-08,
      "model_forward_time": 1.7433688640594482,
      "step": 74061
    },
    {
      "epoch": 5.817413330078125e-08,
      "step": 74061,
      "training_step_time": 6.932187795639038
    },
    {
      "epoch": 5.91278076171875e-08,
      "model_forward_time": 1.7576894760131836,
      "step": 74062
    },
    {
      "epoch": 5.91278076171875e-08,
      "step": 74062,
      "training_step_time": 9.714230298995972
    },
    {
      "epoch": 6.008148193359375e-08,
      "model_forward_time": 1.7475502490997314,
      "step": 74063
    },
    {
      "epoch": 6.008148193359375e-08,
      "step": 74063,
      "training_step_time": 7.52284574508667
    },
    {
      "epoch": 6.103515625e-08,
      "model_forward_time": 1.7429838180541992,
      "step": 74064
    },
    {
      "epoch": 6.103515625e-08,
      "step": 74064,
      "training_step_time": 7.398822069168091
    },
    {
      "epoch": 6.198883056640625e-08,
      "model_forward_time": 1.750042200088501,
      "step": 74065
    },
    {
      "epoch": 6.198883056640625e-08,
      "step": 74065,
      "training_step_time": 6.890443325042725
    },
    {
      "epoch": 6.29425048828125e-08,
      "model_forward_time": 1.7459557056427002,
      "step": 74066
    },
    {
      "epoch": 6.29425048828125e-08,
      "step": 74066,
      "training_step_time": 6.887782335281372
    },
    {
      "epoch": 6.389617919921875e-08,
      "model_forward_time": 1.7415974140167236,
      "step": 74067
    },
    {
      "epoch": 6.389617919921875e-08,
      "step": 74067,
      "training_step_time": 6.902914047241211
    },
    {
      "epoch": 6.4849853515625e-08,
      "model_forward_time": 1.7557177543640137,
      "step": 74068
    },
    {
      "epoch": 6.4849853515625e-08,
      "step": 74068,
      "training_step_time": 6.880215406417847
    },
    {
      "epoch": 6.580352783203126e-08,
      "model_forward_time": 1.75929856300354,
      "step": 74069
    },
    {
      "epoch": 6.580352783203126e-08,
      "step": 74069,
      "training_step_time": 6.8727641105651855
    },
    {
      "epoch": 6.67572021484375e-08,
      "grad_norm": 0.11753105372190475,
      "learning_rate": 1.7284349693850349e-06,
      "loss": 0.0551,
      "step": 74070
    },
    {
      "epoch": 6.67572021484375e-08,
      "model_forward_time": 1.7647299766540527,
      "step": 74070
    },
    {
      "dynamics_loss_avg": 0.049406425841152665,
      "epoch": 6.67572021484375e-08,
      "step": 74070
    },
    {
      "action_loss_avg": 0.0016224425402469934,
      "epoch": 6.67572021484375e-08,
      "step": 74070
    },
    {
      "epoch": 6.67572021484375e-08,
      "step": 74070,
      "training_step_time": 6.856297731399536
    },
    {
      "epoch": 6.771087646484374e-08,
      "model_forward_time": 1.7870962619781494,
      "step": 74071
    },
    {
      "epoch": 6.771087646484374e-08,
      "step": 74071,
      "training_step_time": 8.92074990272522
    },
    {
      "epoch": 6.866455078125e-08,
      "model_forward_time": 2.3792712688446045,
      "step": 74072
    },
    {
      "epoch": 6.866455078125e-08,
      "step": 74072,
      "training_step_time": 7.776511192321777
    },
    {
      "epoch": 6.961822509765625e-08,
      "model_forward_time": 1.7654085159301758,
      "step": 74073
    },
    {
      "epoch": 6.961822509765625e-08,
      "step": 74073,
      "training_step_time": 7.603655099868774
    },
    {
      "epoch": 7.05718994140625e-08,
      "model_forward_time": 1.7588298320770264,
      "step": 74074
    },
    {
      "epoch": 7.05718994140625e-08,
      "step": 74074,
      "training_step_time": 6.875911474227905
    },
    {
      "epoch": 7.152557373046875e-08,
      "model_forward_time": 1.7581629753112793,
      "step": 74075
    },
    {
      "epoch": 7.152557373046875e-08,
      "step": 74075,
      "training_step_time": 6.895657062530518
    },
    {
      "epoch": 7.2479248046875e-08,
      "model_forward_time": 1.7415313720703125,
      "step": 74076
    },
    {
      "epoch": 7.2479248046875e-08,
      "step": 74076,
      "training_step_time": 6.956426620483398
    },
    {
      "epoch": 7.343292236328125e-08,
      "model_forward_time": 1.7615244388580322,
      "step": 74077
    },
    {
      "epoch": 7.343292236328125e-08,
      "step": 74077,
      "training_step_time": 6.852077960968018
    },
    {
      "epoch": 7.43865966796875e-08,
      "model_forward_time": 1.7503268718719482,
      "step": 74078
    },
    {
      "epoch": 7.43865966796875e-08,
      "step": 74078,
      "training_step_time": 6.958487510681152
    },
    {
      "epoch": 7.534027099609375e-08,
      "model_forward_time": 1.7644448280334473,
      "step": 74079
    },
    {
      "epoch": 7.534027099609375e-08,
      "step": 74079,
      "training_step_time": 6.891690254211426
    },
    {
      "epoch": 7.62939453125e-08,
      "grad_norm": 0.08300715684890747,
      "learning_rate": 1.7271847550064347e-06,
      "loss": 0.0543,
      "step": 74080
    },
    {
      "epoch": 7.62939453125e-08,
      "model_forward_time": 1.7626276016235352,
      "step": 74080
    },
    {
      "dynamics_loss_avg": 0.06586529109627008,
      "epoch": 7.62939453125e-08,
      "step": 74080
    },
    {
      "action_loss_avg": 0.0017046931068762205,
      "epoch": 7.62939453125e-08,
      "step": 74080
    },
    {
      "epoch": 7.62939453125e-08,
      "step": 74080,
      "training_step_time": 9.477668523788452
    },
    {
      "epoch": 7.724761962890625e-08,
      "model_forward_time": 2.387854814529419,
      "step": 74081
    },
    {
      "epoch": 7.724761962890625e-08,
      "step": 74081,
      "training_step_time": 7.9352991580963135
    },
    {
      "epoch": 7.82012939453125e-08,
      "model_forward_time": 1.751030683517456,
      "step": 74082
    },
    {
      "epoch": 7.82012939453125e-08,
      "step": 74082,
      "training_step_time": 7.650322914123535
    },
    {
      "epoch": 7.915496826171875e-08,
      "model_forward_time": 1.77008056640625,
      "step": 74083
    },
    {
      "epoch": 7.915496826171875e-08,
      "step": 74083,
      "training_step_time": 6.901135683059692
    },
    {
      "epoch": 8.0108642578125e-08,
      "model_forward_time": 1.7796120643615723,
      "step": 74084
    },
    {
      "epoch": 8.0108642578125e-08,
      "step": 74084,
      "training_step_time": 7.026342391967773
    },
    {
      "epoch": 8.106231689453126e-08,
      "model_forward_time": 1.7650229930877686,
      "step": 74085
    },
    {
      "epoch": 8.106231689453126e-08,
      "step": 74085,
      "training_step_time": 6.9038612842559814
    },
    {
      "epoch": 8.20159912109375e-08,
      "model_forward_time": 1.7601945400238037,
      "step": 74086
    },
    {
      "epoch": 8.20159912109375e-08,
      "step": 74086,
      "training_step_time": 6.872761487960815
    },
    {
      "epoch": 8.296966552734374e-08,
      "model_forward_time": 1.7590584754943848,
      "step": 74087
    },
    {
      "epoch": 8.296966552734374e-08,
      "step": 74087,
      "training_step_time": 6.846457004547119
    },
    {
      "epoch": 8.392333984375e-08,
      "model_forward_time": 1.3968148231506348,
      "step": 74088
    },
    {
      "epoch": 8.392333984375e-08,
      "step": 74088,
      "training_step_time": 6.8255774974823
    },
    {
      "epoch": 8.487701416015625e-08,
      "model_forward_time": 1.7537505626678467,
      "step": 74089
    },
    {
      "epoch": 8.487701416015625e-08,
      "step": 74089,
      "training_step_time": 9.047337055206299
    },
    {
      "epoch": 8.58306884765625e-08,
      "grad_norm": 0.08906611800193787,
      "learning_rate": 1.725934898537987e-06,
      "loss": 0.0548,
      "step": 74090
    },
    {
      "epoch": 8.58306884765625e-08,
      "model_forward_time": 2.3853020668029785,
      "step": 74090
    },
    {
      "dynamics_loss_avg": 0.05178071232512593,
      "epoch": 8.58306884765625e-08,
      "step": 74090
    },
    {
      "action_loss_avg": 0.0018001695512793957,
      "epoch": 8.58306884765625e-08,
      "step": 74090
    },
    {
      "epoch": 8.58306884765625e-08,
      "step": 74090,
      "training_step_time": 8.66395902633667
    },
    {
      "epoch": 8.678436279296875e-08,
      "model_forward_time": 1.7560253143310547,
      "step": 74091
    },
    {
      "epoch": 8.678436279296875e-08,
      "step": 74091,
      "training_step_time": 7.5615105628967285
    },
    {
      "epoch": 8.7738037109375e-08,
      "model_forward_time": 1.757993221282959,
      "step": 74092
    },
    {
      "epoch": 8.7738037109375e-08,
      "step": 74092,
      "training_step_time": 6.816742181777954
    },
    {
      "epoch": 8.869171142578125e-08,
      "model_forward_time": 1.7583084106445312,
      "step": 74093
    },
    {
      "epoch": 8.869171142578125e-08,
      "step": 74093,
      "training_step_time": 6.886046409606934
    },
    {
      "epoch": 8.96453857421875e-08,
      "model_forward_time": 1.759986400604248,
      "step": 74094
    },
    {
      "epoch": 8.96453857421875e-08,
      "step": 74094,
      "training_step_time": 6.859655380249023
    },
    {
      "epoch": 9.059906005859375e-08,
      "model_forward_time": 1.7609734535217285,
      "step": 74095
    },
    {
      "epoch": 9.059906005859375e-08,
      "step": 74095,
      "training_step_time": 6.911166429519653
    },
    {
      "epoch": 9.1552734375e-08,
      "model_forward_time": 1.7587637901306152,
      "step": 74096
    },
    {
      "epoch": 9.1552734375e-08,
      "step": 74096,
      "training_step_time": 6.852007627487183
    },
    {
      "epoch": 9.250640869140625e-08,
      "model_forward_time": 1.7542481422424316,
      "step": 74097
    },
    {
      "epoch": 9.250640869140625e-08,
      "step": 74097,
      "training_step_time": 6.841985464096069
    },
    {
      "epoch": 9.34600830078125e-08,
      "model_forward_time": 1.7629461288452148,
      "step": 74098
    },
    {
      "epoch": 9.34600830078125e-08,
      "step": 74098,
      "training_step_time": 8.676295757293701
    },
    {
      "epoch": 9.441375732421875e-08,
      "model_forward_time": 2.383087635040283,
      "step": 74099
    },
    {
      "epoch": 9.441375732421875e-08,
      "step": 74099,
      "training_step_time": 11.284281253814697
    },
    {
      "epoch": 9.5367431640625e-08,
      "grad_norm": 0.09693622589111328,
      "learning_rate": 1.7246854001163783e-06,
      "loss": 0.0556,
      "step": 74100
    },
    {
      "epoch": 9.5367431640625e-08,
      "model_forward_time": 1.7628676891326904,
      "step": 74100
    },
    {
      "dynamics_loss_avg": 0.05746258907020092,
      "epoch": 9.5367431640625e-08,
      "step": 74100
    },
    {
      "action_loss_avg": 0.0018245726998429746,
      "epoch": 9.5367431640625e-08,
      "step": 74100
    },
    {
      "epoch": 9.5367431640625e-08,
      "step": 74100,
      "training_step_time": 7.56299090385437
    },
    {
      "epoch": 9.632110595703126e-08,
      "model_forward_time": 1.7638981342315674,
      "step": 74101
    },
    {
      "epoch": 9.632110595703126e-08,
      "step": 74101,
      "training_step_time": 7.384275436401367
    },
    {
      "epoch": 9.72747802734375e-08,
      "model_forward_time": 1.7641687393188477,
      "step": 74102
    },
    {
      "epoch": 9.72747802734375e-08,
      "step": 74102,
      "training_step_time": 6.835719347000122
    },
    {
      "epoch": 9.822845458984374e-08,
      "model_forward_time": 1.761305809020996,
      "step": 74103
    },
    {
      "epoch": 9.822845458984374e-08,
      "step": 74103,
      "training_step_time": 6.863634824752808
    },
    {
      "epoch": 9.918212890625e-08,
      "model_forward_time": 1.7620131969451904,
      "step": 74104
    },
    {
      "epoch": 9.918212890625e-08,
      "step": 74104,
      "training_step_time": 6.876785039901733
    },
    {
      "epoch": 1.0013580322265625e-07,
      "model_forward_time": 1.7628061771392822,
      "step": 74105
    },
    {
      "epoch": 1.0013580322265625e-07,
      "step": 74105,
      "training_step_time": 6.859950304031372
    },
    {
      "epoch": 1.010894775390625e-07,
      "model_forward_time": 1.7649972438812256,
      "step": 74106
    },
    {
      "epoch": 1.010894775390625e-07,
      "step": 74106,
      "training_step_time": 6.925130367279053
    },
    {
      "epoch": 1.0204315185546875e-07,
      "model_forward_time": 1.7568981647491455,
      "step": 74107
    },
    {
      "epoch": 1.0204315185546875e-07,
      "step": 74107,
      "training_step_time": 8.84152102470398
    },
    {
      "epoch": 1.02996826171875e-07,
      "model_forward_time": 2.3791801929473877,
      "step": 74108
    },
    {
      "epoch": 1.02996826171875e-07,
      "step": 74108,
      "training_step_time": 11.420795679092407
    },
    {
      "epoch": 1.0395050048828125e-07,
      "model_forward_time": 1.7646782398223877,
      "step": 74109
    },
    {
      "epoch": 1.0395050048828125e-07,
      "step": 74109,
      "training_step_time": 7.810619354248047
    },
    {
      "epoch": 1.049041748046875e-07,
      "grad_norm": 0.07659485936164856,
      "learning_rate": 1.7234362598782478e-06,
      "loss": 0.0556,
      "step": 74110
    },
    {
      "epoch": 1.049041748046875e-07,
      "model_forward_time": 1.7640495300292969,
      "step": 74110
    },
    {
      "dynamics_loss_avg": 0.06944075059145689,
      "epoch": 1.049041748046875e-07,
      "step": 74110
    },
    {
      "action_loss_avg": 0.002012657525483519,
      "epoch": 1.049041748046875e-07,
      "step": 74110
    },
    {
      "epoch": 1.049041748046875e-07,
      "step": 74110,
      "training_step_time": 7.390333414077759
    },
    {
      "epoch": 1.0585784912109375e-07,
      "model_forward_time": 1.763375997543335,
      "step": 74111
    },
    {
      "epoch": 1.0585784912109375e-07,
      "step": 74111,
      "training_step_time": 6.827928304672241
    },
    {
      "epoch": 1.068115234375e-07,
      "model_forward_time": 1.7622389793395996,
      "step": 74112
    },
    {
      "epoch": 1.068115234375e-07,
      "step": 74112,
      "training_step_time": 6.763119220733643
    },
    {
      "epoch": 1.0776519775390625e-07,
      "model_forward_time": 1.7576940059661865,
      "step": 74113
    },
    {
      "epoch": 1.0776519775390625e-07,
      "step": 74113,
      "training_step_time": 10.859322547912598
    },
    {
      "epoch": 1.087188720703125e-07,
      "model_forward_time": 1.7620036602020264,
      "step": 74114
    },
    {
      "epoch": 1.087188720703125e-07,
      "step": 74114,
      "training_step_time": 6.839407444000244
    },
    {
      "epoch": 1.0967254638671875e-07,
      "model_forward_time": 1.76336669921875,
      "step": 74115
    },
    {
      "epoch": 1.0967254638671875e-07,
      "step": 74115,
      "training_step_time": 7.717462062835693
    },
    {
      "epoch": 1.10626220703125e-07,
      "model_forward_time": 1.7544622421264648,
      "step": 74116
    },
    {
      "epoch": 1.10626220703125e-07,
      "step": 74116,
      "training_step_time": 8.061158418655396
    },
    {
      "epoch": 1.1157989501953126e-07,
      "model_forward_time": 1.7577614784240723,
      "step": 74117
    },
    {
      "epoch": 1.1157989501953126e-07,
      "step": 74117,
      "training_step_time": 7.516031503677368
    },
    {
      "epoch": 1.125335693359375e-07,
      "model_forward_time": 1.3911652565002441,
      "step": 74118
    },
    {
      "epoch": 1.125335693359375e-07,
      "step": 74118,
      "training_step_time": 6.82114052772522
    },
    {
      "epoch": 1.1348724365234374e-07,
      "model_forward_time": 1.7576963901519775,
      "step": 74119
    },
    {
      "epoch": 1.1348724365234374e-07,
      "step": 74119,
      "training_step_time": 6.900336027145386
    },
    {
      "epoch": 1.1444091796875e-07,
      "grad_norm": 0.094448022544384,
      "learning_rate": 1.722187477960202e-06,
      "loss": 0.0554,
      "step": 74120
    },
    {
      "epoch": 1.1444091796875e-07,
      "model_forward_time": 1.7601356506347656,
      "step": 74120
    },
    {
      "dynamics_loss_avg": 0.05327912159264088,
      "epoch": 1.1444091796875e-07,
      "step": 74120
    },
    {
      "action_loss_avg": 0.001724347920389846,
      "epoch": 1.1444091796875e-07,
      "step": 74120
    },
    {
      "epoch": 1.1444091796875e-07,
      "step": 74120,
      "training_step_time": 6.7996954917907715
    },
    {
      "epoch": 1.1539459228515625e-07,
      "model_forward_time": 1.7609508037567139,
      "step": 74121
    },
    {
      "epoch": 1.1539459228515625e-07,
      "step": 74121,
      "training_step_time": 6.825437068939209
    },
    {
      "epoch": 1.163482666015625e-07,
      "model_forward_time": 1.7638163566589355,
      "step": 74122
    },
    {
      "epoch": 1.163482666015625e-07,
      "step": 74122,
      "training_step_time": 6.8467302322387695
    },
    {
      "epoch": 1.1730194091796875e-07,
      "model_forward_time": 2.3779726028442383,
      "step": 74123
    },
    {
      "epoch": 1.1730194091796875e-07,
      "step": 74123,
      "training_step_time": 7.843914270401001
    },
    {
      "epoch": 1.18255615234375e-07,
      "model_forward_time": 1.7390060424804688,
      "step": 74124
    },
    {
      "epoch": 1.18255615234375e-07,
      "step": 74124,
      "training_step_time": 7.826309680938721
    },
    {
      "epoch": 1.1920928955078125e-07,
      "model_forward_time": 1.759263515472412,
      "step": 74125
    },
    {
      "epoch": 1.1920928955078125e-07,
      "step": 74125,
      "training_step_time": 6.8209593296051025
    },
    {
      "epoch": 1.201629638671875e-07,
      "model_forward_time": 1.763552188873291,
      "step": 74126
    },
    {
      "epoch": 1.201629638671875e-07,
      "step": 74126,
      "training_step_time": 7.406881332397461
    },
    {
      "epoch": 1.2111663818359375e-07,
      "model_forward_time": 1.759169578552246,
      "step": 74127
    },
    {
      "epoch": 1.2111663818359375e-07,
      "step": 74127,
      "training_step_time": 7.243485689163208
    },
    {
      "epoch": 1.220703125e-07,
      "model_forward_time": 1.7559897899627686,
      "step": 74128
    },
    {
      "epoch": 1.220703125e-07,
      "step": 74128,
      "training_step_time": 6.827133655548096
    },
    {
      "epoch": 1.2302398681640625e-07,
      "model_forward_time": 1.3883752822875977,
      "step": 74129
    },
    {
      "epoch": 1.2302398681640625e-07,
      "step": 74129,
      "training_step_time": 6.871173143386841
    },
    {
      "epoch": 1.239776611328125e-07,
      "grad_norm": 0.10329660028219223,
      "learning_rate": 1.7209390544988058e-06,
      "loss": 0.0542,
      "step": 74130
    },
    {
      "epoch": 1.239776611328125e-07,
      "model_forward_time": 1.761655569076538,
      "step": 74130
    },
    {
      "dynamics_loss_avg": 0.07410207018256187,
      "epoch": 1.239776611328125e-07,
      "step": 74130
    },
    {
      "action_loss_avg": 0.0032324457541108133,
      "epoch": 1.239776611328125e-07,
      "step": 74130
    },
    {
      "epoch": 1.239776611328125e-07,
      "step": 74130,
      "training_step_time": 6.771740674972534
    },
    {
      "epoch": 1.2493133544921876e-07,
      "model_forward_time": 1.761120319366455,
      "step": 74131
    },
    {
      "epoch": 1.2493133544921876e-07,
      "step": 74131,
      "training_step_time": 6.858545303344727
    },
    {
      "epoch": 1.25885009765625e-07,
      "model_forward_time": 2.3980016708374023,
      "step": 74132
    },
    {
      "epoch": 1.25885009765625e-07,
      "step": 74132,
      "training_step_time": 7.831819534301758
    },
    {
      "epoch": 1.2683868408203126e-07,
      "model_forward_time": 1.7609620094299316,
      "step": 74133
    },
    {
      "epoch": 1.2683868408203126e-07,
      "step": 74133,
      "training_step_time": 7.79953670501709
    },
    {
      "epoch": 1.277923583984375e-07,
      "model_forward_time": 1.3883841037750244,
      "step": 74134
    },
    {
      "epoch": 1.277923583984375e-07,
      "step": 74134,
      "training_step_time": 7.185949325561523
    },
    {
      "epoch": 1.2874603271484376e-07,
      "model_forward_time": 1.7607307434082031,
      "step": 74135
    },
    {
      "epoch": 1.2874603271484376e-07,
      "step": 74135,
      "training_step_time": 7.435179948806763
    },
    {
      "epoch": 1.2969970703125e-07,
      "model_forward_time": 1.7604918479919434,
      "step": 74136
    },
    {
      "epoch": 1.2969970703125e-07,
      "step": 74136,
      "training_step_time": 7.356898546218872
    },
    {
      "epoch": 1.3065338134765626e-07,
      "model_forward_time": 1.7632639408111572,
      "step": 74137
    },
    {
      "epoch": 1.3065338134765626e-07,
      "step": 74137,
      "training_step_time": 6.945505380630493
    },
    {
      "epoch": 1.316070556640625e-07,
      "model_forward_time": 1.7517600059509277,
      "step": 74138
    },
    {
      "epoch": 1.316070556640625e-07,
      "step": 74138,
      "training_step_time": 6.896456956863403
    },
    {
      "epoch": 1.3256072998046876e-07,
      "model_forward_time": 1.7558555603027344,
      "step": 74139
    },
    {
      "epoch": 1.3256072998046876e-07,
      "step": 74139,
      "training_step_time": 6.9590795040130615
    },
    {
      "epoch": 1.33514404296875e-07,
      "grad_norm": 0.2399897426366806,
      "learning_rate": 1.7196909896305858e-06,
      "loss": 0.0551,
      "step": 74140
    },
    {
      "epoch": 1.33514404296875e-07,
      "model_forward_time": 1.7606921195983887,
      "step": 74140
    },
    {
      "dynamics_loss_avg": 0.05325095355510712,
      "epoch": 1.33514404296875e-07,
      "step": 74140
    },
    {
      "action_loss_avg": 0.0014958843355998396,
      "epoch": 1.33514404296875e-07,
      "step": 74140
    },
    {
      "epoch": 1.33514404296875e-07,
      "step": 74140,
      "training_step_time": 6.9146294593811035
    },
    {
      "epoch": 1.3446807861328124e-07,
      "model_forward_time": 2.0099785327911377,
      "step": 74141
    },
    {
      "epoch": 1.3446807861328124e-07,
      "step": 74141,
      "training_step_time": 9.710561990737915
    },
    {
      "epoch": 1.354217529296875e-07,
      "model_forward_time": 1.7540218830108643,
      "step": 74142
    },
    {
      "epoch": 1.354217529296875e-07,
      "step": 74142,
      "training_step_time": 7.194972991943359
    },
    {
      "epoch": 1.3637542724609374e-07,
      "model_forward_time": 1.7566101551055908,
      "step": 74143
    },
    {
      "epoch": 1.3637542724609374e-07,
      "step": 74143,
      "training_step_time": 7.8416924476623535
    },
    {
      "epoch": 1.373291015625e-07,
      "model_forward_time": 1.7545766830444336,
      "step": 74144
    },
    {
      "epoch": 1.373291015625e-07,
      "step": 74144,
      "training_step_time": 7.507335424423218
    },
    {
      "epoch": 1.3828277587890624e-07,
      "model_forward_time": 1.7600986957550049,
      "step": 74145
    },
    {
      "epoch": 1.3828277587890624e-07,
      "step": 74145,
      "training_step_time": 7.396924734115601
    },
    {
      "epoch": 1.392364501953125e-07,
      "model_forward_time": 1.764235258102417,
      "step": 74146
    },
    {
      "epoch": 1.392364501953125e-07,
      "step": 74146,
      "training_step_time": 7.367124557495117
    },
    {
      "epoch": 1.4019012451171874e-07,
      "model_forward_time": 1.754929780960083,
      "step": 74147
    },
    {
      "epoch": 1.4019012451171874e-07,
      "step": 74147,
      "training_step_time": 6.932980060577393
    },
    {
      "epoch": 1.41143798828125e-07,
      "model_forward_time": 1.7612614631652832,
      "step": 74148
    },
    {
      "epoch": 1.41143798828125e-07,
      "step": 74148,
      "training_step_time": 6.876148223876953
    },
    {
      "epoch": 1.4209747314453124e-07,
      "model_forward_time": 1.3882577419281006,
      "step": 74149
    },
    {
      "epoch": 1.4209747314453124e-07,
      "step": 74149,
      "training_step_time": 6.893905878067017
    },
    {
      "epoch": 1.430511474609375e-07,
      "grad_norm": 0.08186884969472885,
      "learning_rate": 1.7184432834920261e-06,
      "loss": 0.054,
      "step": 74150
    },
    {
      "epoch": 1.430511474609375e-07,
      "model_forward_time": 1.8141160011291504,
      "step": 74150
    },
    {
      "dynamics_loss_avg": 0.0508305705152452,
      "epoch": 1.430511474609375e-07,
      "step": 74150
    },
    {
      "action_loss_avg": 0.00126087115786504,
      "epoch": 1.430511474609375e-07,
      "step": 74150
    },
    {
      "epoch": 1.430511474609375e-07,
      "step": 74150,
      "training_step_time": 9.425052165985107
    },
    {
      "epoch": 1.4400482177734375e-07,
      "model_forward_time": 2.382333517074585,
      "step": 74151
    },
    {
      "epoch": 1.4400482177734375e-07,
      "step": 74151,
      "training_step_time": 7.823451042175293
    },
    {
      "epoch": 1.4495849609375e-07,
      "model_forward_time": 1.7604146003723145,
      "step": 74152
    },
    {
      "epoch": 1.4495849609375e-07,
      "step": 74152,
      "training_step_time": 7.831478595733643
    },
    {
      "epoch": 1.4591217041015625e-07,
      "model_forward_time": 1.753857135772705,
      "step": 74153
    },
    {
      "epoch": 1.4591217041015625e-07,
      "step": 74153,
      "training_step_time": 7.4018635749816895
    },
    {
      "epoch": 1.468658447265625e-07,
      "model_forward_time": 1.7651479244232178,
      "step": 74154
    },
    {
      "epoch": 1.468658447265625e-07,
      "step": 74154,
      "training_step_time": 6.878979206085205
    },
    {
      "epoch": 1.4781951904296875e-07,
      "model_forward_time": 1.3852384090423584,
      "step": 74155
    },
    {
      "epoch": 1.4781951904296875e-07,
      "step": 74155,
      "training_step_time": 7.50896143913269
    },
    {
      "epoch": 1.48773193359375e-07,
      "model_forward_time": 1.7603471279144287,
      "step": 74156
    },
    {
      "epoch": 1.48773193359375e-07,
      "step": 74156,
      "training_step_time": 6.8716044425964355
    },
    {
      "epoch": 1.4972686767578125e-07,
      "model_forward_time": 1.767716646194458,
      "step": 74157
    },
    {
      "epoch": 1.4972686767578125e-07,
      "step": 74157,
      "training_step_time": 6.841866731643677
    },
    {
      "epoch": 1.506805419921875e-07,
      "model_forward_time": 1.7608461380004883,
      "step": 74158
    },
    {
      "epoch": 1.506805419921875e-07,
      "step": 74158,
      "training_step_time": 6.8683202266693115
    },
    {
      "epoch": 1.5163421630859375e-07,
      "model_forward_time": 1.7900748252868652,
      "step": 74159
    },
    {
      "epoch": 1.5163421630859375e-07,
      "step": 74159,
      "training_step_time": 9.627285718917847
    },
    {
      "epoch": 1.52587890625e-07,
      "grad_norm": 0.08724091202020645,
      "learning_rate": 1.7171959362195746e-06,
      "loss": 0.0549,
      "step": 74160
    },
    {
      "epoch": 1.52587890625e-07,
      "model_forward_time": 2.412094831466675,
      "step": 74160
    },
    {
      "dynamics_loss_avg": 0.04869120754301548,
      "epoch": 1.52587890625e-07,
      "step": 74160
    },
    {
      "action_loss_avg": 0.001110522449016571,
      "epoch": 1.52587890625e-07,
      "step": 74160
    },
    {
      "epoch": 1.52587890625e-07,
      "step": 74160,
      "training_step_time": 7.7897257804870605
    },
    {
      "epoch": 1.5354156494140625e-07,
      "model_forward_time": 1.7589702606201172,
      "step": 74161
    },
    {
      "epoch": 1.5354156494140625e-07,
      "step": 74161,
      "training_step_time": 7.7173912525177
    },
    {
      "epoch": 1.544952392578125e-07,
      "model_forward_time": 1.7696197032928467,
      "step": 74162
    },
    {
      "epoch": 1.544952392578125e-07,
      "step": 74162,
      "training_step_time": 7.451232433319092
    },
    {
      "epoch": 1.5544891357421876e-07,
      "model_forward_time": 1.7530748844146729,
      "step": 74163
    },
    {
      "epoch": 1.5544891357421876e-07,
      "step": 74163,
      "training_step_time": 6.78757643699646
    },
    {
      "epoch": 1.56402587890625e-07,
      "model_forward_time": 1.763197422027588,
      "step": 74164
    },
    {
      "epoch": 1.56402587890625e-07,
      "step": 74164,
      "training_step_time": 7.38516640663147
    },
    {
      "epoch": 1.5735626220703126e-07,
      "model_forward_time": 1.7648885250091553,
      "step": 74165
    },
    {
      "epoch": 1.5735626220703126e-07,
      "step": 74165,
      "training_step_time": 6.8319008350372314
    },
    {
      "epoch": 1.583099365234375e-07,
      "model_forward_time": 1.7647175788879395,
      "step": 74166
    },
    {
      "epoch": 1.583099365234375e-07,
      "step": 74166,
      "training_step_time": 6.795602083206177
    },
    {
      "epoch": 1.5926361083984376e-07,
      "model_forward_time": 1.760669469833374,
      "step": 74167
    },
    {
      "epoch": 1.5926361083984376e-07,
      "step": 74167,
      "training_step_time": 6.858819246292114
    },
    {
      "epoch": 1.6021728515625e-07,
      "model_forward_time": 1.7620766162872314,
      "step": 74168
    },
    {
      "epoch": 1.6021728515625e-07,
      "step": 74168,
      "training_step_time": 10.244178771972656
    },
    {
      "epoch": 1.6117095947265626e-07,
      "model_forward_time": 2.392615556716919,
      "step": 74169
    },
    {
      "epoch": 1.6117095947265626e-07,
      "step": 74169,
      "training_step_time": 7.74548602104187
    },
    {
      "epoch": 1.621246337890625e-07,
      "grad_norm": 0.0965917780995369,
      "learning_rate": 1.7159489479496423e-06,
      "loss": 0.0553,
      "step": 74170
    },
    {
      "epoch": 1.621246337890625e-07,
      "model_forward_time": 1.770998239517212,
      "step": 74170
    },
    {
      "dynamics_loss_avg": 0.04985441910102963,
      "epoch": 1.621246337890625e-07,
      "step": 74170
    },
    {
      "action_loss_avg": 0.002246423449832946,
      "epoch": 1.621246337890625e-07,
      "step": 74170
    },
    {
      "epoch": 1.621246337890625e-07,
      "step": 74170,
      "training_step_time": 7.5734779834747314
    },
    {
      "epoch": 1.6307830810546876e-07,
      "model_forward_time": 1.7605910301208496,
      "step": 74171
    },
    {
      "epoch": 1.6307830810546876e-07,
      "step": 74171,
      "training_step_time": 8.706538677215576
    },
    {
      "epoch": 1.64031982421875e-07,
      "model_forward_time": 1.759676218032837,
      "step": 74172
    },
    {
      "epoch": 1.64031982421875e-07,
      "step": 74172,
      "training_step_time": 7.408948183059692
    },
    {
      "epoch": 1.6498565673828124e-07,
      "model_forward_time": 1.7763679027557373,
      "step": 74173
    },
    {
      "epoch": 1.6498565673828124e-07,
      "step": 74173,
      "training_step_time": 7.330415725708008
    },
    {
      "epoch": 1.659393310546875e-07,
      "model_forward_time": 1.4029748439788818,
      "step": 74174
    },
    {
      "epoch": 1.659393310546875e-07,
      "step": 74174,
      "training_step_time": 7.274073839187622
    },
    {
      "epoch": 1.6689300537109374e-07,
      "model_forward_time": 1.762763500213623,
      "step": 74175
    },
    {
      "epoch": 1.6689300537109374e-07,
      "step": 74175,
      "training_step_time": 6.825962543487549
    },
    {
      "epoch": 1.678466796875e-07,
      "model_forward_time": 1.7642765045166016,
      "step": 74176
    },
    {
      "epoch": 1.678466796875e-07,
      "step": 74176,
      "training_step_time": 6.832348585128784
    },
    {
      "epoch": 1.6880035400390624e-07,
      "model_forward_time": 1.7689659595489502,
      "step": 74177
    },
    {
      "epoch": 1.6880035400390624e-07,
      "step": 74177,
      "training_step_time": 9.255152702331543
    },
    {
      "epoch": 1.697540283203125e-07,
      "model_forward_time": 1.8203845024108887,
      "step": 74178
    },
    {
      "epoch": 1.697540283203125e-07,
      "step": 74178,
      "training_step_time": 8.63803482055664
    },
    {
      "epoch": 1.7070770263671874e-07,
      "model_forward_time": 2.3887181282043457,
      "step": 74179
    },
    {
      "epoch": 1.7070770263671874e-07,
      "step": 74179,
      "training_step_time": 7.6519060134887695
    },
    {
      "epoch": 1.71661376953125e-07,
      "grad_norm": 0.08439751714468002,
      "learning_rate": 1.714702318818593e-06,
      "loss": 0.0537,
      "step": 74180
    },
    {
      "epoch": 1.71661376953125e-07,
      "model_forward_time": 1.7644314765930176,
      "step": 74180
    },
    {
      "dynamics_loss_avg": 0.056547570135444404,
      "epoch": 1.71661376953125e-07,
      "step": 74180
    },
    {
      "action_loss_avg": 0.0012952186109032482,
      "epoch": 1.71661376953125e-07,
      "step": 74180
    },
    {
      "epoch": 1.71661376953125e-07,
      "step": 74180,
      "training_step_time": 7.74990177154541
    },
    {
      "epoch": 1.7261505126953124e-07,
      "model_forward_time": 1.7619428634643555,
      "step": 74181
    },
    {
      "epoch": 1.7261505126953124e-07,
      "step": 74181,
      "training_step_time": 7.372519254684448
    },
    {
      "epoch": 1.735687255859375e-07,
      "model_forward_time": 1.7624046802520752,
      "step": 74182
    },
    {
      "epoch": 1.735687255859375e-07,
      "step": 74182,
      "training_step_time": 7.384761095046997
    },
    {
      "epoch": 1.7452239990234375e-07,
      "model_forward_time": 1.76641845703125,
      "step": 74183
    },
    {
      "epoch": 1.7452239990234375e-07,
      "step": 74183,
      "training_step_time": 7.270989418029785
    },
    {
      "epoch": 1.7547607421875e-07,
      "model_forward_time": 1.7655038833618164,
      "step": 74184
    },
    {
      "epoch": 1.7547607421875e-07,
      "step": 74184,
      "training_step_time": 6.861706256866455
    },
    {
      "epoch": 1.7642974853515625e-07,
      "model_forward_time": 1.7614076137542725,
      "step": 74185
    },
    {
      "epoch": 1.7642974853515625e-07,
      "step": 74185,
      "training_step_time": 6.828965187072754
    },
    {
      "epoch": 1.773834228515625e-07,
      "model_forward_time": 1.7736291885375977,
      "step": 74186
    },
    {
      "epoch": 1.773834228515625e-07,
      "step": 74186,
      "training_step_time": 8.30953049659729
    },
    {
      "epoch": 1.7833709716796875e-07,
      "model_forward_time": 1.824143409729004,
      "step": 74187
    },
    {
      "epoch": 1.7833709716796875e-07,
      "step": 74187,
      "training_step_time": 8.348230600357056
    },
    {
      "epoch": 1.79290771484375e-07,
      "model_forward_time": 2.3744046688079834,
      "step": 74188
    },
    {
      "epoch": 1.79290771484375e-07,
      "step": 74188,
      "training_step_time": 7.7715020179748535
    },
    {
      "epoch": 1.8024444580078125e-07,
      "model_forward_time": 1.7702057361602783,
      "step": 74189
    },
    {
      "epoch": 1.8024444580078125e-07,
      "step": 74189,
      "training_step_time": 7.760380268096924
    },
    {
      "epoch": 1.811981201171875e-07,
      "grad_norm": 0.08687973767518997,
      "learning_rate": 1.7134560489627595e-06,
      "loss": 0.0537,
      "step": 74190
    },
    {
      "epoch": 1.811981201171875e-07,
      "model_forward_time": 1.7708022594451904,
      "step": 74190
    },
    {
      "dynamics_loss_avg": 0.05554140098392964,
      "epoch": 1.811981201171875e-07,
      "step": 74190
    },
    {
      "action_loss_avg": 0.0010982377367326989,
      "epoch": 1.811981201171875e-07,
      "step": 74190
    },
    {
      "epoch": 1.811981201171875e-07,
      "step": 74190,
      "training_step_time": 7.487647294998169
    },
    {
      "epoch": 1.8215179443359375e-07,
      "model_forward_time": 1.7576498985290527,
      "step": 74191
    },
    {
      "epoch": 1.8215179443359375e-07,
      "step": 74191,
      "training_step_time": 7.425354957580566
    },
    {
      "epoch": 1.8310546875e-07,
      "model_forward_time": 1.7642555236816406,
      "step": 74192
    },
    {
      "epoch": 1.8310546875e-07,
      "step": 74192,
      "training_step_time": 7.375056028366089
    },
    {
      "epoch": 1.8405914306640625e-07,
      "model_forward_time": 1.7648789882659912,
      "step": 74193
    },
    {
      "epoch": 1.8405914306640625e-07,
      "step": 74193,
      "training_step_time": 6.914428949356079
    },
    {
      "epoch": 1.850128173828125e-07,
      "model_forward_time": 1.7651047706604004,
      "step": 74194
    },
    {
      "epoch": 1.850128173828125e-07,
      "step": 74194,
      "training_step_time": 6.849532604217529
    },
    {
      "epoch": 1.8596649169921876e-07,
      "model_forward_time": 1.7589879035949707,
      "step": 74195
    },
    {
      "epoch": 1.8596649169921876e-07,
      "step": 74195,
      "training_step_time": 8.654364109039307
    },
    {
      "epoch": 1.86920166015625e-07,
      "model_forward_time": 1.8245062828063965,
      "step": 74196
    },
    {
      "epoch": 1.86920166015625e-07,
      "step": 74196,
      "training_step_time": 10.896419048309326
    },
    {
      "epoch": 1.8787384033203126e-07,
      "model_forward_time": 2.423065662384033,
      "step": 74197
    },
    {
      "epoch": 1.8787384033203126e-07,
      "step": 74197,
      "training_step_time": 7.829477310180664
    },
    {
      "epoch": 1.888275146484375e-07,
      "model_forward_time": 1.7614631652832031,
      "step": 74198
    },
    {
      "epoch": 1.888275146484375e-07,
      "step": 74198,
      "training_step_time": 7.807691335678101
    },
    {
      "epoch": 1.8978118896484376e-07,
      "model_forward_time": 1.766671895980835,
      "step": 74199
    },
    {
      "epoch": 1.8978118896484376e-07,
      "step": 74199,
      "training_step_time": 7.604781627655029
    },
    {
      "epoch": 1.9073486328125e-07,
      "grad_norm": 0.08595755696296692,
      "learning_rate": 1.7122101385184308e-06,
      "loss": 0.0557,
      "step": 74200
    },
    {
      "epoch": 1.9073486328125e-07,
      "model_forward_time": 1.7648863792419434,
      "step": 74200
    },
    {
      "dynamics_loss_avg": 0.05797701142728329,
      "epoch": 1.9073486328125e-07,
      "step": 74200
    },
    {
      "action_loss_avg": 0.0013215898536145688,
      "epoch": 1.9073486328125e-07,
      "step": 74200
    },
    {
      "epoch": 1.9073486328125e-07,
      "step": 74200,
      "training_step_time": 7.334057569503784
    },
    {
      "epoch": 1.9168853759765626e-07,
      "model_forward_time": 1.7651033401489258,
      "step": 74201
    },
    {
      "epoch": 1.9168853759765626e-07,
      "step": 74201,
      "training_step_time": 7.375870227813721
    },
    {
      "epoch": 1.926422119140625e-07,
      "model_forward_time": 1.7507708072662354,
      "step": 74202
    },
    {
      "epoch": 1.926422119140625e-07,
      "step": 74202,
      "training_step_time": 7.298864841461182
    },
    {
      "epoch": 1.9359588623046876e-07,
      "model_forward_time": 1.7545166015625,
      "step": 74203
    },
    {
      "epoch": 1.9359588623046876e-07,
      "step": 74203,
      "training_step_time": 6.838271617889404
    },
    {
      "epoch": 1.94549560546875e-07,
      "model_forward_time": 1.7517614364624023,
      "step": 74204
    },
    {
      "epoch": 1.94549560546875e-07,
      "step": 74204,
      "training_step_time": 7.739441394805908
    },
    {
      "epoch": 1.9550323486328124e-07,
      "model_forward_time": 1.3893299102783203,
      "step": 74205
    },
    {
      "epoch": 1.9550323486328124e-07,
      "step": 74205,
      "training_step_time": 10.781397104263306
    },
    {
      "epoch": 1.964569091796875e-07,
      "model_forward_time": 2.3769493103027344,
      "step": 74206
    },
    {
      "epoch": 1.964569091796875e-07,
      "step": 74206,
      "training_step_time": 7.77726149559021
    },
    {
      "epoch": 1.9741058349609374e-07,
      "model_forward_time": 1.7509160041809082,
      "step": 74207
    },
    {
      "epoch": 1.9741058349609374e-07,
      "step": 74207,
      "training_step_time": 8.226079940795898
    },
    {
      "epoch": 1.983642578125e-07,
      "model_forward_time": 1.7711615562438965,
      "step": 74208
    },
    {
      "epoch": 1.983642578125e-07,
      "step": 74208,
      "training_step_time": 8.122591257095337
    },
    {
      "epoch": 1.9931793212890624e-07,
      "model_forward_time": 1.764244794845581,
      "step": 74209
    },
    {
      "epoch": 1.9931793212890624e-07,
      "step": 74209,
      "training_step_time": 6.805503606796265
    },
    {
      "epoch": 2.002716064453125e-07,
      "grad_norm": 0.10995853692293167,
      "learning_rate": 1.71096458762186e-06,
      "loss": 0.0532,
      "step": 74210
    },
    {
      "epoch": 2.002716064453125e-07,
      "model_forward_time": 1.7675397396087646,
      "step": 74210
    },
    {
      "dynamics_loss_avg": 0.05656270179897547,
      "epoch": 2.002716064453125e-07,
      "step": 74210
    },
    {
      "action_loss_avg": 0.002213593010674231,
      "epoch": 2.002716064453125e-07,
      "step": 74210
    },
    {
      "epoch": 2.002716064453125e-07,
      "step": 74210,
      "training_step_time": 7.341550827026367
    },
    {
      "epoch": 2.0122528076171874e-07,
      "model_forward_time": 1.7719507217407227,
      "step": 74211
    },
    {
      "epoch": 2.0122528076171874e-07,
      "step": 74211,
      "training_step_time": 7.374698877334595
    },
    {
      "epoch": 2.02178955078125e-07,
      "model_forward_time": 1.7708990573883057,
      "step": 74212
    },
    {
      "epoch": 2.02178955078125e-07,
      "step": 74212,
      "training_step_time": 6.808459520339966
    },
    {
      "epoch": 2.0313262939453125e-07,
      "model_forward_time": 1.7463080883026123,
      "step": 74213
    },
    {
      "epoch": 2.0313262939453125e-07,
      "step": 74213,
      "training_step_time": 8.577674388885498
    },
    {
      "epoch": 2.040863037109375e-07,
      "model_forward_time": 1.7588553428649902,
      "step": 74214
    },
    {
      "epoch": 2.040863037109375e-07,
      "step": 74214,
      "training_step_time": 8.496041536331177
    },
    {
      "epoch": 2.0503997802734375e-07,
      "model_forward_time": 1.7443411350250244,
      "step": 74215
    },
    {
      "epoch": 2.0503997802734375e-07,
      "step": 74215,
      "training_step_time": 7.7495739459991455
    },
    {
      "epoch": 2.0599365234375e-07,
      "model_forward_time": 2.34991192817688,
      "step": 74216
    },
    {
      "epoch": 2.0599365234375e-07,
      "step": 74216,
      "training_step_time": 7.434621095657349
    },
    {
      "epoch": 2.0694732666015625e-07,
      "model_forward_time": 1.3819775581359863,
      "step": 74217
    },
    {
      "epoch": 2.0694732666015625e-07,
      "step": 74217,
      "training_step_time": 8.80295467376709
    },
    {
      "epoch": 2.079010009765625e-07,
      "model_forward_time": 1.7612667083740234,
      "step": 74218
    },
    {
      "epoch": 2.079010009765625e-07,
      "step": 74218,
      "training_step_time": 7.512028932571411
    },
    {
      "epoch": 2.0885467529296875e-07,
      "model_forward_time": 1.755690336227417,
      "step": 74219
    },
    {
      "epoch": 2.0885467529296875e-07,
      "step": 74219,
      "training_step_time": 7.453569173812866
    },
    {
      "epoch": 2.09808349609375e-07,
      "grad_norm": 0.36701178550720215,
      "learning_rate": 1.7097193964092552e-06,
      "loss": 0.0526,
      "step": 74220
    },
    {
      "epoch": 2.09808349609375e-07,
      "model_forward_time": 1.7587542533874512,
      "step": 74220
    },
    {
      "dynamics_loss_avg": 0.05970663353800774,
      "epoch": 2.09808349609375e-07,
      "step": 74220
    },
    {
      "action_loss_avg": 0.0014613148232456297,
      "epoch": 2.09808349609375e-07,
      "step": 74220
    },
    {
      "epoch": 2.09808349609375e-07,
      "step": 74220,
      "training_step_time": 7.3095703125
    },
    {
      "epoch": 2.1076202392578125e-07,
      "model_forward_time": 1.7454016208648682,
      "step": 74221
    },
    {
      "epoch": 2.1076202392578125e-07,
      "step": 74221,
      "training_step_time": 6.811261892318726
    },
    {
      "epoch": 2.117156982421875e-07,
      "model_forward_time": 1.761582612991333,
      "step": 74222
    },
    {
      "epoch": 2.117156982421875e-07,
      "step": 74222,
      "training_step_time": 7.780357360839844
    },
    {
      "epoch": 2.1266937255859375e-07,
      "model_forward_time": 1.7543878555297852,
      "step": 74223
    },
    {
      "epoch": 2.1266937255859375e-07,
      "step": 74223,
      "training_step_time": 9.255351066589355
    },
    {
      "epoch": 2.13623046875e-07,
      "model_forward_time": 1.800140142440796,
      "step": 74224
    },
    {
      "epoch": 2.13623046875e-07,
      "step": 74224,
      "training_step_time": 8.251538753509521
    },
    {
      "epoch": 2.1457672119140626e-07,
      "model_forward_time": 2.3666412830352783,
      "step": 74225
    },
    {
      "epoch": 2.1457672119140626e-07,
      "step": 74225,
      "training_step_time": 7.726492166519165
    },
    {
      "epoch": 2.155303955078125e-07,
      "model_forward_time": 1.7503137588500977,
      "step": 74226
    },
    {
      "epoch": 2.155303955078125e-07,
      "step": 74226,
      "training_step_time": 7.697528123855591
    },
    {
      "epoch": 2.1648406982421876e-07,
      "model_forward_time": 1.7305407524108887,
      "step": 74227
    },
    {
      "epoch": 2.1648406982421876e-07,
      "step": 74227,
      "training_step_time": 7.520939111709595
    },
    {
      "epoch": 2.17437744140625e-07,
      "model_forward_time": 1.753253698348999,
      "step": 74228
    },
    {
      "epoch": 2.17437744140625e-07,
      "step": 74228,
      "training_step_time": 7.995843887329102
    },
    {
      "epoch": 2.1839141845703126e-07,
      "model_forward_time": 1.7493078708648682,
      "step": 74229
    },
    {
      "epoch": 2.1839141845703126e-07,
      "step": 74229,
      "training_step_time": 7.335771799087524
    },
    {
      "epoch": 2.193450927734375e-07,
      "grad_norm": 0.08329284191131592,
      "learning_rate": 1.7084745650167922e-06,
      "loss": 0.0518,
      "step": 74230
    },
    {
      "epoch": 2.193450927734375e-07,
      "model_forward_time": 1.7441413402557373,
      "step": 74230
    },
    {
      "dynamics_loss_avg": 0.05125271771103144,
      "epoch": 2.193450927734375e-07,
      "step": 74230
    },
    {
      "action_loss_avg": 0.001342500449391082,
      "epoch": 2.193450927734375e-07,
      "step": 74230
    },
    {
      "epoch": 2.193450927734375e-07,
      "step": 74230,
      "training_step_time": 7.292135000228882
    },
    {
      "epoch": 2.2029876708984376e-07,
      "model_forward_time": 1.7540459632873535,
      "step": 74231
    },
    {
      "epoch": 2.2029876708984376e-07,
      "step": 74231,
      "training_step_time": 7.738416910171509
    },
    {
      "epoch": 2.2125244140625e-07,
      "model_forward_time": 1.746838092803955,
      "step": 74232
    },
    {
      "epoch": 2.2125244140625e-07,
      "step": 74232,
      "training_step_time": 9.626466751098633
    },
    {
      "epoch": 2.2220611572265626e-07,
      "model_forward_time": 1.79146409034729,
      "step": 74233
    },
    {
      "epoch": 2.2220611572265626e-07,
      "step": 74233,
      "training_step_time": 8.442055702209473
    },
    {
      "epoch": 2.231597900390625e-07,
      "model_forward_time": 2.3752102851867676,
      "step": 74234
    },
    {
      "epoch": 2.231597900390625e-07,
      "step": 74234,
      "training_step_time": 8.729825496673584
    },
    {
      "epoch": 2.2411346435546874e-07,
      "model_forward_time": 1.7531883716583252,
      "step": 74235
    },
    {
      "epoch": 2.2411346435546874e-07,
      "step": 74235,
      "training_step_time": 7.330087423324585
    },
    {
      "epoch": 2.25067138671875e-07,
      "model_forward_time": 1.757326364517212,
      "step": 74236
    },
    {
      "epoch": 2.25067138671875e-07,
      "step": 74236,
      "training_step_time": 7.76453971862793
    },
    {
      "epoch": 2.2602081298828124e-07,
      "model_forward_time": 1.7533543109893799,
      "step": 74237
    },
    {
      "epoch": 2.2602081298828124e-07,
      "step": 74237,
      "training_step_time": 6.832763910293579
    },
    {
      "epoch": 2.269744873046875e-07,
      "model_forward_time": 1.7576138973236084,
      "step": 74238
    },
    {
      "epoch": 2.269744873046875e-07,
      "step": 74238,
      "training_step_time": 7.376572370529175
    },
    {
      "epoch": 2.2792816162109374e-07,
      "model_forward_time": 1.7596938610076904,
      "step": 74239
    },
    {
      "epoch": 2.2792816162109374e-07,
      "step": 74239,
      "training_step_time": 7.324039459228516
    },
    {
      "epoch": 2.288818359375e-07,
      "grad_norm": 0.08926114439964294,
      "learning_rate": 1.7072300935806002e-06,
      "loss": 0.0523,
      "step": 74240
    },
    {
      "epoch": 2.288818359375e-07,
      "model_forward_time": 1.7322819232940674,
      "step": 74240
    },
    {
      "dynamics_loss_avg": 0.05507109891623259,
      "epoch": 2.288818359375e-07,
      "step": 74240
    },
    {
      "action_loss_avg": 0.0014584314718376844,
      "epoch": 2.288818359375e-07,
      "step": 74240
    },
    {
      "epoch": 2.288818359375e-07,
      "step": 74240,
      "training_step_time": 7.773869752883911
    },
    {
      "epoch": 2.2983551025390624e-07,
      "model_forward_time": 1.7595372200012207,
      "step": 74241
    },
    {
      "epoch": 2.2983551025390624e-07,
      "step": 74241,
      "training_step_time": 9.115867614746094
    },
    {
      "epoch": 2.307891845703125e-07,
      "model_forward_time": 1.7978661060333252,
      "step": 74242
    },
    {
      "epoch": 2.307891845703125e-07,
      "step": 74242,
      "training_step_time": 7.700909852981567
    },
    {
      "epoch": 2.3174285888671874e-07,
      "model_forward_time": 2.4231560230255127,
      "step": 74243
    },
    {
      "epoch": 2.3174285888671874e-07,
      "step": 74243,
      "training_step_time": 9.248265504837036
    },
    {
      "epoch": 2.32696533203125e-07,
      "model_forward_time": 1.7588260173797607,
      "step": 74244
    },
    {
      "epoch": 2.32696533203125e-07,
      "step": 74244,
      "training_step_time": 7.628566265106201
    },
    {
      "epoch": 2.3365020751953125e-07,
      "model_forward_time": 1.7591564655303955,
      "step": 74245
    },
    {
      "epoch": 2.3365020751953125e-07,
      "step": 74245,
      "training_step_time": 7.737846374511719
    },
    {
      "epoch": 2.346038818359375e-07,
      "model_forward_time": 1.3985495567321777,
      "step": 74246
    },
    {
      "epoch": 2.346038818359375e-07,
      "step": 74246,
      "training_step_time": 6.850850343704224
    },
    {
      "epoch": 2.3555755615234375e-07,
      "model_forward_time": 1.7734675407409668,
      "step": 74247
    },
    {
      "epoch": 2.3555755615234375e-07,
      "step": 74247,
      "training_step_time": 7.445597171783447
    },
    {
      "epoch": 2.3651123046875e-07,
      "model_forward_time": 1.7651801109313965,
      "step": 74248
    },
    {
      "epoch": 2.3651123046875e-07,
      "step": 74248,
      "training_step_time": 7.276506185531616
    },
    {
      "epoch": 2.3746490478515625e-07,
      "model_forward_time": 1.7642016410827637,
      "step": 74249
    },
    {
      "epoch": 2.3746490478515625e-07,
      "step": 74249,
      "training_step_time": 7.697545051574707
    },
    {
      "epoch": 2.384185791015625e-07,
      "grad_norm": 0.098216213285923,
      "learning_rate": 1.7059859822367764e-06,
      "loss": 0.0541,
      "step": 74250
    },
    {
      "epoch": 2.384185791015625e-07,
      "model_forward_time": 1.7402398586273193,
      "step": 74250
    },
    {
      "dynamics_loss_avg": 0.04345298893749714,
      "epoch": 2.384185791015625e-07,
      "step": 74250
    },
    {
      "action_loss_avg": 0.0014263347533415073,
      "epoch": 2.384185791015625e-07,
      "step": 74250
    },
    {
      "epoch": 2.384185791015625e-07,
      "step": 74250,
      "training_step_time": 7.964717626571655
    },
    {
      "epoch": 2.3937225341796875e-07,
      "model_forward_time": 1.8164691925048828,
      "step": 74251
    },
    {
      "epoch": 2.3937225341796875e-07,
      "step": 74251,
      "training_step_time": 9.55503535270691
    },
    {
      "epoch": 2.40325927734375e-07,
      "model_forward_time": 2.387491226196289,
      "step": 74252
    },
    {
      "epoch": 2.40325927734375e-07,
      "step": 74252,
      "training_step_time": 7.696661472320557
    },
    {
      "epoch": 2.4127960205078125e-07,
      "model_forward_time": 1.7760746479034424,
      "step": 74253
    },
    {
      "epoch": 2.4127960205078125e-07,
      "step": 74253,
      "training_step_time": 7.752622604370117
    },
    {
      "epoch": 2.422332763671875e-07,
      "model_forward_time": 1.7678308486938477,
      "step": 74254
    },
    {
      "epoch": 2.422332763671875e-07,
      "step": 74254,
      "training_step_time": 7.708288192749023
    },
    {
      "epoch": 2.4318695068359375e-07,
      "model_forward_time": 1.7725744247436523,
      "step": 74255
    },
    {
      "epoch": 2.4318695068359375e-07,
      "step": 74255,
      "training_step_time": 7.455191135406494
    },
    {
      "epoch": 2.44140625e-07,
      "model_forward_time": 1.7715392112731934,
      "step": 74256
    },
    {
      "epoch": 2.44140625e-07,
      "step": 74256,
      "training_step_time": 7.372934818267822
    },
    {
      "epoch": 2.4509429931640626e-07,
      "model_forward_time": 1.7675986289978027,
      "step": 74257
    },
    {
      "epoch": 2.4509429931640626e-07,
      "step": 74257,
      "training_step_time": 7.339796304702759
    },
    {
      "epoch": 2.460479736328125e-07,
      "model_forward_time": 1.7692246437072754,
      "step": 74258
    },
    {
      "epoch": 2.460479736328125e-07,
      "step": 74258,
      "training_step_time": 8.784083843231201
    },
    {
      "epoch": 2.4700164794921876e-07,
      "model_forward_time": 1.769040584564209,
      "step": 74259
    },
    {
      "epoch": 2.4700164794921876e-07,
      "step": 74259,
      "training_step_time": 8.515943765640259
    },
    {
      "epoch": 2.47955322265625e-07,
      "grad_norm": 0.10499842464923859,
      "learning_rate": 1.7047422311213696e-06,
      "loss": 0.0531,
      "step": 74260
    },
    {
      "epoch": 2.47955322265625e-07,
      "model_forward_time": 1.7914326190948486,
      "step": 74260
    },
    {
      "dynamics_loss_avg": 0.03772897543385625,
      "epoch": 2.47955322265625e-07,
      "step": 74260
    },
    {
      "action_loss_avg": 0.0011331747053191065,
      "epoch": 2.47955322265625e-07,
      "step": 74260
    },
    {
      "epoch": 2.47955322265625e-07,
      "step": 74260,
      "training_step_time": 7.731450796127319
    },
    {
      "epoch": 2.4890899658203126e-07,
      "model_forward_time": 1.7658138275146484,
      "step": 74261
    },
    {
      "epoch": 2.4890899658203126e-07,
      "step": 74261,
      "training_step_time": 8.74226999282837
    },
    {
      "epoch": 2.498626708984375e-07,
      "model_forward_time": 2.386054754257202,
      "step": 74262
    },
    {
      "epoch": 2.498626708984375e-07,
      "step": 74262,
      "training_step_time": 7.840100526809692
    },
    {
      "epoch": 2.5081634521484376e-07,
      "model_forward_time": 1.7666239738464355,
      "step": 74263
    },
    {
      "epoch": 2.5081634521484376e-07,
      "step": 74263,
      "training_step_time": 10.217797994613647
    },
    {
      "epoch": 2.5177001953125e-07,
      "model_forward_time": 1.7693836688995361,
      "step": 74264
    },
    {
      "epoch": 2.5177001953125e-07,
      "step": 74264,
      "training_step_time": 7.674584627151489
    },
    {
      "epoch": 2.5272369384765626e-07,
      "model_forward_time": 1.766275405883789,
      "step": 74265
    },
    {
      "epoch": 2.5272369384765626e-07,
      "step": 74265,
      "training_step_time": 7.2977824211120605
    },
    {
      "epoch": 2.536773681640625e-07,
      "model_forward_time": 1.7720580101013184,
      "step": 74266
    },
    {
      "epoch": 2.536773681640625e-07,
      "step": 74266,
      "training_step_time": 7.29174280166626
    },
    {
      "epoch": 2.5463104248046876e-07,
      "model_forward_time": 1.7665002346038818,
      "step": 74267
    },
    {
      "epoch": 2.5463104248046876e-07,
      "step": 74267,
      "training_step_time": 7.708961725234985
    },
    {
      "epoch": 2.55584716796875e-07,
      "model_forward_time": 1.765084981918335,
      "step": 74268
    },
    {
      "epoch": 2.55584716796875e-07,
      "step": 74268,
      "training_step_time": 9.375048160552979
    },
    {
      "epoch": 2.5653839111328127e-07,
      "model_forward_time": 1.7645118236541748,
      "step": 74269
    },
    {
      "epoch": 2.5653839111328127e-07,
      "step": 74269,
      "training_step_time": 9.26590609550476
    },
    {
      "epoch": 2.574920654296875e-07,
      "grad_norm": 0.0896041989326477,
      "learning_rate": 1.703498840370401e-06,
      "loss": 0.0525,
      "step": 74270
    },
    {
      "epoch": 2.574920654296875e-07,
      "model_forward_time": 1.7655794620513916,
      "step": 74270
    },
    {
      "dynamics_loss_avg": 0.061905781552195546,
      "epoch": 2.574920654296875e-07,
      "step": 74270
    },
    {
      "action_loss_avg": 0.0017653999908361583,
      "epoch": 2.574920654296875e-07,
      "step": 74270
    },
    {
      "epoch": 2.574920654296875e-07,
      "step": 74270,
      "training_step_time": 7.840497255325317
    },
    {
      "epoch": 2.5844573974609377e-07,
      "model_forward_time": 2.3932762145996094,
      "step": 74271
    },
    {
      "epoch": 2.5844573974609377e-07,
      "step": 74271,
      "training_step_time": 7.891313076019287
    },
    {
      "epoch": 2.593994140625e-07,
      "model_forward_time": 1.769211769104004,
      "step": 74272
    },
    {
      "epoch": 2.593994140625e-07,
      "step": 74272,
      "training_step_time": 7.82307243347168
    },
    {
      "epoch": 2.6035308837890627e-07,
      "model_forward_time": 1.7637434005737305,
      "step": 74273
    },
    {
      "epoch": 2.6035308837890627e-07,
      "step": 74273,
      "training_step_time": 8.392142295837402
    },
    {
      "epoch": 2.613067626953125e-07,
      "model_forward_time": 1.7716617584228516,
      "step": 74274
    },
    {
      "epoch": 2.613067626953125e-07,
      "step": 74274,
      "training_step_time": 6.938746690750122
    },
    {
      "epoch": 2.6226043701171877e-07,
      "model_forward_time": 1.7624282836914062,
      "step": 74275
    },
    {
      "epoch": 2.6226043701171877e-07,
      "step": 74275,
      "training_step_time": 7.588726043701172
    },
    {
      "epoch": 2.63214111328125e-07,
      "model_forward_time": 1.7683601379394531,
      "step": 74276
    },
    {
      "epoch": 2.63214111328125e-07,
      "step": 74276,
      "training_step_time": 7.845536708831787
    },
    {
      "epoch": 2.6416778564453127e-07,
      "model_forward_time": 1.770674705505371,
      "step": 74277
    },
    {
      "epoch": 2.6416778564453127e-07,
      "step": 74277,
      "training_step_time": 7.86781907081604
    },
    {
      "epoch": 2.651214599609375e-07,
      "model_forward_time": 1.7692787647247314,
      "step": 74278
    },
    {
      "epoch": 2.651214599609375e-07,
      "step": 74278,
      "training_step_time": 7.826226711273193
    },
    {
      "epoch": 2.660751342773438e-07,
      "model_forward_time": 1.8220295906066895,
      "step": 74279
    },
    {
      "epoch": 2.660751342773438e-07,
      "step": 74279,
      "training_step_time": 9.815377235412598
    },
    {
      "epoch": 2.6702880859375e-07,
      "grad_norm": 0.1009543165564537,
      "learning_rate": 1.7022558101198422e-06,
      "loss": 0.0538,
      "step": 74280
    },
    {
      "epoch": 2.6702880859375e-07,
      "model_forward_time": 2.4024860858917236,
      "step": 74280
    },
    {
      "dynamics_loss_avg": 0.05097811836749315,
      "epoch": 2.6702880859375e-07,
      "step": 74280
    },
    {
      "action_loss_avg": 0.0016100230306619779,
      "epoch": 2.6702880859375e-07,
      "step": 74280
    },
    {
      "epoch": 2.6702880859375e-07,
      "step": 74280,
      "training_step_time": 8.90431809425354
    },
    {
      "epoch": 2.679824829101563e-07,
      "model_forward_time": 1.7670042514801025,
      "step": 74281
    },
    {
      "epoch": 2.679824829101563e-07,
      "step": 74281,
      "training_step_time": 7.804243564605713
    },
    {
      "epoch": 2.689361572265625e-07,
      "model_forward_time": 1.7659902572631836,
      "step": 74282
    },
    {
      "epoch": 2.689361572265625e-07,
      "step": 74282,
      "training_step_time": 7.569039344787598
    },
    {
      "epoch": 2.698898315429687e-07,
      "model_forward_time": 1.7702720165252686,
      "step": 74283
    },
    {
      "epoch": 2.698898315429687e-07,
      "step": 74283,
      "training_step_time": 7.7208943367004395
    },
    {
      "epoch": 2.70843505859375e-07,
      "model_forward_time": 1.7711024284362793,
      "step": 74284
    },
    {
      "epoch": 2.70843505859375e-07,
      "step": 74284,
      "training_step_time": 7.666517972946167
    },
    {
      "epoch": 2.7179718017578123e-07,
      "model_forward_time": 1.764655590057373,
      "step": 74285
    },
    {
      "epoch": 2.7179718017578123e-07,
      "step": 74285,
      "training_step_time": 7.802048921585083
    },
    {
      "epoch": 2.727508544921875e-07,
      "model_forward_time": 1.7708308696746826,
      "step": 74286
    },
    {
      "epoch": 2.727508544921875e-07,
      "step": 74286,
      "training_step_time": 9.228466033935547
    },
    {
      "epoch": 2.7370452880859373e-07,
      "model_forward_time": 1.772773265838623,
      "step": 74287
    },
    {
      "epoch": 2.7370452880859373e-07,
      "step": 74287,
      "training_step_time": 9.012146949768066
    },
    {
      "epoch": 2.74658203125e-07,
      "model_forward_time": 1.827059030532837,
      "step": 74288
    },
    {
      "epoch": 2.74658203125e-07,
      "step": 74288,
      "training_step_time": 7.918078899383545
    },
    {
      "epoch": 2.7561187744140623e-07,
      "model_forward_time": 2.3574256896972656,
      "step": 74289
    },
    {
      "epoch": 2.7561187744140623e-07,
      "step": 74289,
      "training_step_time": 8.404621839523315
    },
    {
      "epoch": 2.765655517578125e-07,
      "grad_norm": 0.09139018505811691,
      "learning_rate": 1.7010131405056307e-06,
      "loss": 0.0533,
      "step": 74290
    },
    {
      "epoch": 2.765655517578125e-07,
      "model_forward_time": 1.765805959701538,
      "step": 74290
    },
    {
      "dynamics_loss_avg": 0.06051693968474865,
      "epoch": 2.765655517578125e-07,
      "step": 74290
    },
    {
      "action_loss_avg": 0.0019025615300051868,
      "epoch": 2.765655517578125e-07,
      "step": 74290
    },
    {
      "epoch": 2.765655517578125e-07,
      "step": 74290,
      "training_step_time": 7.769017219543457
    },
    {
      "epoch": 2.7751922607421873e-07,
      "model_forward_time": 1.7739276885986328,
      "step": 74291
    },
    {
      "epoch": 2.7751922607421873e-07,
      "step": 74291,
      "training_step_time": 7.591285467147827
    },
    {
      "epoch": 2.78472900390625e-07,
      "model_forward_time": 1.7620880603790283,
      "step": 74292
    },
    {
      "epoch": 2.78472900390625e-07,
      "step": 74292,
      "training_step_time": 7.7975592613220215
    },
    {
      "epoch": 2.7942657470703123e-07,
      "model_forward_time": 1.3792588710784912,
      "step": 74293
    },
    {
      "epoch": 2.7942657470703123e-07,
      "step": 74293,
      "training_step_time": 7.514421224594116
    },
    {
      "epoch": 2.803802490234375e-07,
      "model_forward_time": 1.768892526626587,
      "step": 74294
    },
    {
      "epoch": 2.803802490234375e-07,
      "step": 74294,
      "training_step_time": 7.852593898773193
    },
    {
      "epoch": 2.8133392333984374e-07,
      "model_forward_time": 1.7667381763458252,
      "step": 74295
    },
    {
      "epoch": 2.8133392333984374e-07,
      "step": 74295,
      "training_step_time": 8.648512125015259
    },
    {
      "epoch": 2.8228759765625e-07,
      "model_forward_time": 1.782212257385254,
      "step": 74296
    },
    {
      "epoch": 2.8228759765625e-07,
      "step": 74296,
      "training_step_time": 7.542677164077759
    },
    {
      "epoch": 2.8324127197265624e-07,
      "model_forward_time": 1.821563720703125,
      "step": 74297
    },
    {
      "epoch": 2.8324127197265624e-07,
      "step": 74297,
      "training_step_time": 8.255226373672485
    },
    {
      "epoch": 2.841949462890625e-07,
      "model_forward_time": 2.363349437713623,
      "step": 74298
    },
    {
      "epoch": 2.841949462890625e-07,
      "step": 74298,
      "training_step_time": 8.866555213928223
    },
    {
      "epoch": 2.8514862060546874e-07,
      "model_forward_time": 1.7672128677368164,
      "step": 74299
    },
    {
      "epoch": 2.8514862060546874e-07,
      "step": 74299,
      "training_step_time": 7.731722354888916
    },
    {
      "epoch": 2.86102294921875e-07,
      "grad_norm": 0.08006642013788223,
      "learning_rate": 1.6997708316636613e-06,
      "loss": 0.0523,
      "step": 74300
    },
    {
      "epoch": 2.86102294921875e-07,
      "model_forward_time": 1.7659711837768555,
      "step": 74300
    },
    {
      "dynamics_loss_avg": 0.038970598578453065,
      "epoch": 2.86102294921875e-07,
      "step": 74300
    },
    {
      "action_loss_avg": 0.0013275518780574203,
      "epoch": 2.86102294921875e-07,
      "step": 74300
    },
    {
      "epoch": 2.86102294921875e-07,
      "step": 74300,
      "training_step_time": 7.707574367523193
    },
    {
      "epoch": 2.8705596923828124e-07,
      "model_forward_time": 1.765226125717163,
      "step": 74301
    },
    {
      "epoch": 2.8705596923828124e-07,
      "step": 74301,
      "training_step_time": 7.6449174880981445
    },
    {
      "epoch": 2.880096435546875e-07,
      "model_forward_time": 1.7663111686706543,
      "step": 74302
    },
    {
      "epoch": 2.880096435546875e-07,
      "step": 74302,
      "training_step_time": 7.519342422485352
    },
    {
      "epoch": 2.8896331787109374e-07,
      "model_forward_time": 1.7706389427185059,
      "step": 74303
    },
    {
      "epoch": 2.8896331787109374e-07,
      "step": 74303,
      "training_step_time": 7.804487228393555
    },
    {
      "epoch": 2.899169921875e-07,
      "model_forward_time": 1.760115146636963,
      "step": 74304
    },
    {
      "epoch": 2.899169921875e-07,
      "step": 74304,
      "training_step_time": 7.726101875305176
    },
    {
      "epoch": 2.9087066650390624e-07,
      "model_forward_time": 1.7617547512054443,
      "step": 74305
    },
    {
      "epoch": 2.9087066650390624e-07,
      "step": 74305,
      "training_step_time": 7.6369006633758545
    },
    {
      "epoch": 2.918243408203125e-07,
      "model_forward_time": 1.4076316356658936,
      "step": 74306
    },
    {
      "epoch": 2.918243408203125e-07,
      "step": 74306,
      "training_step_time": 9.693879127502441
    },
    {
      "epoch": 2.9277801513671875e-07,
      "model_forward_time": 2.41972279548645,
      "step": 74307
    },
    {
      "epoch": 2.9277801513671875e-07,
      "step": 74307,
      "training_step_time": 8.50705337524414
    },
    {
      "epoch": 2.93731689453125e-07,
      "model_forward_time": 1.761695146560669,
      "step": 74308
    },
    {
      "epoch": 2.93731689453125e-07,
      "step": 74308,
      "training_step_time": 7.763108253479004
    },
    {
      "epoch": 2.9468536376953125e-07,
      "model_forward_time": 1.7590622901916504,
      "step": 74309
    },
    {
      "epoch": 2.9468536376953125e-07,
      "step": 74309,
      "training_step_time": 7.8520941734313965
    },
    {
      "epoch": 2.956390380859375e-07,
      "grad_norm": 0.10502828657627106,
      "learning_rate": 1.698528883729793e-06,
      "loss": 0.0543,
      "step": 74310
    },
    {
      "epoch": 2.956390380859375e-07,
      "model_forward_time": 1.7947523593902588,
      "step": 74310
    },
    {
      "dynamics_loss_avg": 0.047674904391169545,
      "epoch": 2.956390380859375e-07,
      "step": 74310
    },
    {
      "action_loss_avg": 0.0018567004735814407,
      "epoch": 2.956390380859375e-07,
      "step": 74310
    },
    {
      "epoch": 2.956390380859375e-07,
      "step": 74310,
      "training_step_time": 7.5011091232299805
    },
    {
      "epoch": 2.9659271240234375e-07,
      "model_forward_time": 1.7682678699493408,
      "step": 74311
    },
    {
      "epoch": 2.9659271240234375e-07,
      "step": 74311,
      "training_step_time": 8.55312728881836
    },
    {
      "epoch": 2.9754638671875e-07,
      "model_forward_time": 1.7724552154541016,
      "step": 74312
    },
    {
      "epoch": 2.9754638671875e-07,
      "step": 74312,
      "training_step_time": 7.774932622909546
    },
    {
      "epoch": 2.9850006103515625e-07,
      "model_forward_time": 1.7609193325042725,
      "step": 74313
    },
    {
      "epoch": 2.9850006103515625e-07,
      "step": 74313,
      "training_step_time": 7.4407126903533936
    },
    {
      "epoch": 2.994537353515625e-07,
      "model_forward_time": 1.7637810707092285,
      "step": 74314
    },
    {
      "epoch": 2.994537353515625e-07,
      "step": 74314,
      "training_step_time": 7.862314462661743
    },
    {
      "epoch": 3.0040740966796875e-07,
      "model_forward_time": 1.7854185104370117,
      "step": 74315
    },
    {
      "epoch": 3.0040740966796875e-07,
      "step": 74315,
      "training_step_time": 8.34952425956726
    },
    {
      "epoch": 3.01361083984375e-07,
      "model_forward_time": 2.416161298751831,
      "step": 74316
    },
    {
      "epoch": 3.01361083984375e-07,
      "step": 74316,
      "training_step_time": 7.972400426864624
    },
    {
      "epoch": 3.0231475830078125e-07,
      "model_forward_time": 1.769768238067627,
      "step": 74317
    },
    {
      "epoch": 3.0231475830078125e-07,
      "step": 74317,
      "training_step_time": 9.488957166671753
    },
    {
      "epoch": 3.032684326171875e-07,
      "model_forward_time": 1.7673640251159668,
      "step": 74318
    },
    {
      "epoch": 3.032684326171875e-07,
      "step": 74318,
      "training_step_time": 7.852628946304321
    },
    {
      "epoch": 3.0422210693359376e-07,
      "model_forward_time": 1.7764935493469238,
      "step": 74319
    },
    {
      "epoch": 3.0422210693359376e-07,
      "step": 74319,
      "training_step_time": 7.532552003860474
    },
    {
      "epoch": 3.0517578125e-07,
      "grad_norm": 0.07497823238372803,
      "learning_rate": 1.6972872968398396e-06,
      "loss": 0.0523,
      "step": 74320
    },
    {
      "epoch": 3.0517578125e-07,
      "model_forward_time": 1.7643816471099854,
      "step": 74320
    },
    {
      "dynamics_loss_avg": 0.040660457033663985,
      "epoch": 3.0517578125e-07,
      "step": 74320
    },
    {
      "action_loss_avg": 0.001043772665434517,
      "epoch": 3.0517578125e-07,
      "step": 74320
    },
    {
      "epoch": 3.0517578125e-07,
      "step": 74320,
      "training_step_time": 7.793166875839233
    },
    {
      "epoch": 3.0612945556640626e-07,
      "model_forward_time": 1.7550561428070068,
      "step": 74321
    },
    {
      "epoch": 3.0612945556640626e-07,
      "step": 74321,
      "training_step_time": 8.772705793380737
    },
    {
      "epoch": 3.070831298828125e-07,
      "model_forward_time": 1.7560856342315674,
      "step": 74322
    },
    {
      "epoch": 3.070831298828125e-07,
      "step": 74322,
      "training_step_time": 7.818942546844482
    },
    {
      "epoch": 3.0803680419921876e-07,
      "model_forward_time": 1.7576267719268799,
      "step": 74323
    },
    {
      "epoch": 3.0803680419921876e-07,
      "step": 74323,
      "training_step_time": 7.8089919090271
    },
    {
      "epoch": 3.08990478515625e-07,
      "model_forward_time": 1.7553112506866455,
      "step": 74324
    },
    {
      "epoch": 3.08990478515625e-07,
      "step": 74324,
      "training_step_time": 7.766467332839966
    },
    {
      "epoch": 3.0994415283203126e-07,
      "model_forward_time": 2.401259660720825,
      "step": 74325
    },
    {
      "epoch": 3.0994415283203126e-07,
      "step": 74325,
      "training_step_time": 10.337908267974854
    },
    {
      "epoch": 3.108978271484375e-07,
      "model_forward_time": 1.7643542289733887,
      "step": 74326
    },
    {
      "epoch": 3.108978271484375e-07,
      "step": 74326,
      "training_step_time": 7.859539270401001
    },
    {
      "epoch": 3.1185150146484376e-07,
      "model_forward_time": 1.7690637111663818,
      "step": 74327
    },
    {
      "epoch": 3.1185150146484376e-07,
      "step": 74327,
      "training_step_time": 7.8391642570495605
    },
    {
      "epoch": 3.1280517578125e-07,
      "model_forward_time": 1.7654190063476562,
      "step": 74328
    },
    {
      "epoch": 3.1280517578125e-07,
      "step": 74328,
      "training_step_time": 7.609490394592285
    },
    {
      "epoch": 3.1375885009765626e-07,
      "model_forward_time": 1.7549145221710205,
      "step": 74329
    },
    {
      "epoch": 3.1375885009765626e-07,
      "step": 74329,
      "training_step_time": 7.742008447647095
    },
    {
      "epoch": 3.147125244140625e-07,
      "grad_norm": 0.074447400867939,
      "learning_rate": 1.6960460711295845e-06,
      "loss": 0.0548,
      "step": 74330
    },
    {
      "epoch": 3.147125244140625e-07,
      "model_forward_time": 1.7666428089141846,
      "step": 74330
    },
    {
      "dynamics_loss_avg": 0.06019444279372692,
      "epoch": 3.147125244140625e-07,
      "step": 74330
    },
    {
      "action_loss_avg": 0.0015039786812849343,
      "epoch": 3.147125244140625e-07,
      "step": 74330
    },
    {
      "epoch": 3.147125244140625e-07,
      "step": 74330,
      "training_step_time": 7.663143634796143
    },
    {
      "epoch": 3.1566619873046877e-07,
      "model_forward_time": 1.7588531970977783,
      "step": 74331
    },
    {
      "epoch": 3.1566619873046877e-07,
      "step": 74331,
      "training_step_time": 7.733596563339233
    },
    {
      "epoch": 3.16619873046875e-07,
      "model_forward_time": 1.7700490951538086,
      "step": 74332
    },
    {
      "epoch": 3.16619873046875e-07,
      "step": 74332,
      "training_step_time": 7.821150779724121
    },
    {
      "epoch": 3.1757354736328127e-07,
      "model_forward_time": 1.758455753326416,
      "step": 74333
    },
    {
      "epoch": 3.1757354736328127e-07,
      "step": 74333,
      "training_step_time": 8.396650791168213
    },
    {
      "epoch": 3.185272216796875e-07,
      "model_forward_time": 1.8106040954589844,
      "step": 74334
    },
    {
      "epoch": 3.185272216796875e-07,
      "step": 74334,
      "training_step_time": 9.878628253936768
    },
    {
      "epoch": 3.1948089599609377e-07,
      "model_forward_time": 2.3794870376586914,
      "step": 74335
    },
    {
      "epoch": 3.1948089599609377e-07,
      "step": 74335,
      "training_step_time": 8.484517574310303
    },
    {
      "epoch": 3.204345703125e-07,
      "model_forward_time": 1.758040189743042,
      "step": 74336
    },
    {
      "epoch": 3.204345703125e-07,
      "step": 74336,
      "training_step_time": 8.269851684570312
    },
    {
      "epoch": 3.2138824462890627e-07,
      "model_forward_time": 1.7608411312103271,
      "step": 74337
    },
    {
      "epoch": 3.2138824462890627e-07,
      "step": 74337,
      "training_step_time": 7.464854717254639
    },
    {
      "epoch": 3.223419189453125e-07,
      "model_forward_time": 1.7662463188171387,
      "step": 74338
    },
    {
      "epoch": 3.223419189453125e-07,
      "step": 74338,
      "training_step_time": 7.655603885650635
    },
    {
      "epoch": 3.2329559326171877e-07,
      "model_forward_time": 1.7529892921447754,
      "step": 74339
    },
    {
      "epoch": 3.2329559326171877e-07,
      "step": 74339,
      "training_step_time": 9.193999290466309
    },
    {
      "epoch": 3.24249267578125e-07,
      "grad_norm": 0.07683195918798447,
      "learning_rate": 1.6948052067347626e-06,
      "loss": 0.0536,
      "step": 74340
    },
    {
      "epoch": 3.24249267578125e-07,
      "model_forward_time": 1.7615983486175537,
      "step": 74340
    },
    {
      "dynamics_loss_avg": 0.04905868088826537,
      "epoch": 3.24249267578125e-07,
      "step": 74340
    },
    {
      "action_loss_avg": 0.003536001016618684,
      "epoch": 3.24249267578125e-07,
      "step": 74340
    },
    {
      "epoch": 3.24249267578125e-07,
      "step": 74340,
      "training_step_time": 7.675878524780273
    },
    {
      "epoch": 3.252029418945313e-07,
      "model_forward_time": 1.7666597366333008,
      "step": 74341
    },
    {
      "epoch": 3.252029418945313e-07,
      "step": 74341,
      "training_step_time": 7.692224025726318
    },
    {
      "epoch": 3.261566162109375e-07,
      "model_forward_time": 1.7608566284179688,
      "step": 74342
    },
    {
      "epoch": 3.261566162109375e-07,
      "step": 74342,
      "training_step_time": 8.638916730880737
    },
    {
      "epoch": 3.271102905273438e-07,
      "model_forward_time": 1.813533067703247,
      "step": 74343
    },
    {
      "epoch": 3.271102905273438e-07,
      "step": 74343,
      "training_step_time": 9.320146322250366
    },
    {
      "epoch": 3.2806396484375e-07,
      "model_forward_time": 2.3887109756469727,
      "step": 74344
    },
    {
      "epoch": 3.2806396484375e-07,
      "step": 74344,
      "training_step_time": 7.719916582107544
    },
    {
      "epoch": 3.290176391601562e-07,
      "model_forward_time": 1.7586121559143066,
      "step": 74345
    },
    {
      "epoch": 3.290176391601562e-07,
      "step": 74345,
      "training_step_time": 8.941742897033691
    },
    {
      "epoch": 3.299713134765625e-07,
      "model_forward_time": 1.751352071762085,
      "step": 74346
    },
    {
      "epoch": 3.299713134765625e-07,
      "step": 74346,
      "training_step_time": 7.778537034988403
    },
    {
      "epoch": 3.309249877929687e-07,
      "model_forward_time": 1.7579338550567627,
      "step": 74347
    },
    {
      "epoch": 3.309249877929687e-07,
      "step": 74347,
      "training_step_time": 7.761325359344482
    },
    {
      "epoch": 3.31878662109375e-07,
      "model_forward_time": 1.766653299331665,
      "step": 74348
    },
    {
      "epoch": 3.31878662109375e-07,
      "step": 74348,
      "training_step_time": 7.492048501968384
    },
    {
      "epoch": 3.3283233642578123e-07,
      "model_forward_time": 1.7600328922271729,
      "step": 74349
    },
    {
      "epoch": 3.3283233642578123e-07,
      "step": 74349,
      "training_step_time": 7.730658769607544
    },
    {
      "epoch": 3.337860107421875e-07,
      "grad_norm": 0.10538841784000397,
      "learning_rate": 1.6935647037910751e-06,
      "loss": 0.0531,
      "step": 74350
    },
    {
      "epoch": 3.337860107421875e-07,
      "model_forward_time": 1.7514734268188477,
      "step": 74350
    },
    {
      "dynamics_loss_avg": 0.06019543297588825,
      "epoch": 3.337860107421875e-07,
      "step": 74350
    },
    {
      "action_loss_avg": 0.001364537375047803,
      "epoch": 3.337860107421875e-07,
      "step": 74350
    },
    {
      "epoch": 3.337860107421875e-07,
      "step": 74350,
      "training_step_time": 8.208849668502808
    },
    {
      "epoch": 3.3473968505859373e-07,
      "model_forward_time": 1.7574124336242676,
      "step": 74351
    },
    {
      "epoch": 3.3473968505859373e-07,
      "step": 74351,
      "training_step_time": 7.774036407470703
    },
    {
      "epoch": 3.35693359375e-07,
      "model_forward_time": 1.7818620204925537,
      "step": 74352
    },
    {
      "epoch": 3.35693359375e-07,
      "step": 74352,
      "training_step_time": 9.070347309112549
    },
    {
      "epoch": 3.3664703369140623e-07,
      "model_forward_time": 2.3866069316864014,
      "step": 74353
    },
    {
      "epoch": 3.3664703369140623e-07,
      "step": 74353,
      "training_step_time": 7.997342824935913
    },
    {
      "epoch": 3.376007080078125e-07,
      "model_forward_time": 1.7605345249176025,
      "step": 74354
    },
    {
      "epoch": 3.376007080078125e-07,
      "step": 74354,
      "training_step_time": 7.834130764007568
    },
    {
      "epoch": 3.3855438232421873e-07,
      "model_forward_time": 1.758380651473999,
      "step": 74355
    },
    {
      "epoch": 3.3855438232421873e-07,
      "step": 74355,
      "training_step_time": 7.828122615814209
    },
    {
      "epoch": 3.39508056640625e-07,
      "model_forward_time": 1.7619976997375488,
      "step": 74356
    },
    {
      "epoch": 3.39508056640625e-07,
      "step": 74356,
      "training_step_time": 7.568983793258667
    },
    {
      "epoch": 3.4046173095703123e-07,
      "model_forward_time": 1.761265516281128,
      "step": 74357
    },
    {
      "epoch": 3.4046173095703123e-07,
      "step": 74357,
      "training_step_time": 7.76398777961731
    },
    {
      "epoch": 3.414154052734375e-07,
      "model_forward_time": 1.765855073928833,
      "step": 74358
    },
    {
      "epoch": 3.414154052734375e-07,
      "step": 74358,
      "training_step_time": 7.698767900466919
    },
    {
      "epoch": 3.4236907958984374e-07,
      "model_forward_time": 1.758345365524292,
      "step": 74359
    },
    {
      "epoch": 3.4236907958984374e-07,
      "step": 74359,
      "training_step_time": 7.296933889389038
    },
    {
      "epoch": 3.4332275390625e-07,
      "grad_norm": 0.09354958683252335,
      "learning_rate": 1.6923245624341794e-06,
      "loss": 0.0523,
      "step": 74360
    },
    {
      "epoch": 3.4332275390625e-07,
      "model_forward_time": 1.7602250576019287,
      "step": 74360
    },
    {
      "dynamics_loss_avg": 0.046251314971596,
      "epoch": 3.4332275390625e-07,
      "step": 74360
    },
    {
      "action_loss_avg": 0.0012508026615250857,
      "epoch": 3.4332275390625e-07,
      "step": 74360
    },
    {
      "epoch": 3.4332275390625e-07,
      "step": 74360,
      "training_step_time": 7.730182886123657
    },
    {
      "epoch": 3.4427642822265624e-07,
      "model_forward_time": 1.7650363445281982,
      "step": 74361
    },
    {
      "epoch": 3.4427642822265624e-07,
      "step": 74361,
      "training_step_time": 9.18380618095398
    },
    {
      "epoch": 3.452301025390625e-07,
      "model_forward_time": 2.4176909923553467,
      "step": 74362
    },
    {
      "epoch": 3.452301025390625e-07,
      "step": 74362,
      "training_step_time": 8.811461687088013
    },
    {
      "epoch": 3.4618377685546874e-07,
      "model_forward_time": 1.763444423675537,
      "step": 74363
    },
    {
      "epoch": 3.4618377685546874e-07,
      "step": 74363,
      "training_step_time": 7.7844579219818115
    },
    {
      "epoch": 3.47137451171875e-07,
      "model_forward_time": 1.7593252658843994,
      "step": 74364
    },
    {
      "epoch": 3.47137451171875e-07,
      "step": 74364,
      "training_step_time": 7.770063638687134
    },
    {
      "epoch": 3.4809112548828124e-07,
      "model_forward_time": 1.762416124343872,
      "step": 74365
    },
    {
      "epoch": 3.4809112548828124e-07,
      "step": 74365,
      "training_step_time": 7.5330650806427
    },
    {
      "epoch": 3.490447998046875e-07,
      "model_forward_time": 1.737379789352417,
      "step": 74366
    },
    {
      "epoch": 3.490447998046875e-07,
      "step": 74366,
      "training_step_time": 7.667133092880249
    },
    {
      "epoch": 3.4999847412109374e-07,
      "model_forward_time": 1.7711803913116455,
      "step": 74367
    },
    {
      "epoch": 3.4999847412109374e-07,
      "step": 74367,
      "training_step_time": 7.734864711761475
    },
    {
      "epoch": 3.509521484375e-07,
      "model_forward_time": 1.7657232284545898,
      "step": 74368
    },
    {
      "epoch": 3.509521484375e-07,
      "step": 74368,
      "training_step_time": 7.272374868392944
    },
    {
      "epoch": 3.5190582275390624e-07,
      "model_forward_time": 1.7568795680999756,
      "step": 74369
    },
    {
      "epoch": 3.5190582275390624e-07,
      "step": 74369,
      "training_step_time": 8.351958513259888
    },
    {
      "epoch": 3.528594970703125e-07,
      "grad_norm": 0.09700512140989304,
      "learning_rate": 1.6910847827996961e-06,
      "loss": 0.0551,
      "step": 74370
    },
    {
      "epoch": 3.528594970703125e-07,
      "model_forward_time": 1.7628297805786133,
      "step": 74370
    },
    {
      "dynamics_loss_avg": 0.04657223392277956,
      "epoch": 3.528594970703125e-07,
      "step": 74370
    },
    {
      "action_loss_avg": 0.0013902386286645196,
      "epoch": 3.528594970703125e-07,
      "step": 74370
    },
    {
      "epoch": 3.528594970703125e-07,
      "step": 74370,
      "training_step_time": 8.851418495178223
    },
    {
      "epoch": 3.5381317138671875e-07,
      "model_forward_time": 2.4116909503936768,
      "step": 74371
    },
    {
      "epoch": 3.5381317138671875e-07,
      "step": 74371,
      "training_step_time": 7.795544624328613
    },
    {
      "epoch": 3.54766845703125e-07,
      "model_forward_time": 1.7583024501800537,
      "step": 74372
    },
    {
      "epoch": 3.54766845703125e-07,
      "step": 74372,
      "training_step_time": 9.06793212890625
    },
    {
      "epoch": 3.5572052001953125e-07,
      "model_forward_time": 1.7634978294372559,
      "step": 74373
    },
    {
      "epoch": 3.5572052001953125e-07,
      "step": 74373,
      "training_step_time": 8.510564804077148
    },
    {
      "epoch": 3.566741943359375e-07,
      "model_forward_time": 1.7656302452087402,
      "step": 74374
    },
    {
      "epoch": 3.566741943359375e-07,
      "step": 74374,
      "training_step_time": 7.780029058456421
    },
    {
      "epoch": 3.5762786865234375e-07,
      "model_forward_time": 1.7616252899169922,
      "step": 74375
    },
    {
      "epoch": 3.5762786865234375e-07,
      "step": 74375,
      "training_step_time": 7.676974534988403
    },
    {
      "epoch": 3.5858154296875e-07,
      "model_forward_time": 1.7583096027374268,
      "step": 74376
    },
    {
      "epoch": 3.5858154296875e-07,
      "step": 74376,
      "training_step_time": 7.696087837219238
    },
    {
      "epoch": 3.5953521728515625e-07,
      "model_forward_time": 1.7611656188964844,
      "step": 74377
    },
    {
      "epoch": 3.5953521728515625e-07,
      "step": 74377,
      "training_step_time": 8.202210187911987
    },
    {
      "epoch": 3.604888916015625e-07,
      "model_forward_time": 1.7654345035552979,
      "step": 74378
    },
    {
      "epoch": 3.604888916015625e-07,
      "step": 74378,
      "training_step_time": 8.598449945449829
    },
    {
      "epoch": 3.6144256591796875e-07,
      "model_forward_time": 1.7623751163482666,
      "step": 74379
    },
    {
      "epoch": 3.6144256591796875e-07,
      "step": 74379,
      "training_step_time": 9.341749906539917
    },
    {
      "epoch": 3.62396240234375e-07,
      "grad_norm": 0.10285954177379608,
      "learning_rate": 1.689845365023206e-06,
      "loss": 0.0545,
      "step": 74380
    },
    {
      "epoch": 3.62396240234375e-07,
      "model_forward_time": 1.8267185688018799,
      "step": 74380
    },
    {
      "dynamics_loss_avg": 0.04242052864283323,
      "epoch": 3.62396240234375e-07,
      "step": 74380
    },
    {
      "action_loss_avg": 0.0012672554148593917,
      "epoch": 3.62396240234375e-07,
      "step": 74380
    },
    {
      "epoch": 3.62396240234375e-07,
      "step": 74380,
      "training_step_time": 9.314517259597778
    },
    {
      "epoch": 3.6334991455078126e-07,
      "model_forward_time": 1.7538471221923828,
      "step": 74381
    },
    {
      "epoch": 3.6334991455078126e-07,
      "step": 74381,
      "training_step_time": 9.463229656219482
    },
    {
      "epoch": 3.643035888671875e-07,
      "model_forward_time": 1.760082483291626,
      "step": 74382
    },
    {
      "epoch": 3.643035888671875e-07,
      "step": 74382,
      "training_step_time": 7.797282934188843
    },
    {
      "epoch": 3.6525726318359376e-07,
      "model_forward_time": 1.7658402919769287,
      "step": 74383
    },
    {
      "epoch": 3.6525726318359376e-07,
      "step": 74383,
      "training_step_time": 8.559506893157959
    },
    {
      "epoch": 3.662109375e-07,
      "model_forward_time": 1.7578413486480713,
      "step": 74384
    },
    {
      "epoch": 3.662109375e-07,
      "step": 74384,
      "training_step_time": 7.757054805755615
    },
    {
      "epoch": 3.6716461181640626e-07,
      "model_forward_time": 1.7694854736328125,
      "step": 74385
    },
    {
      "epoch": 3.6716461181640626e-07,
      "step": 74385,
      "training_step_time": 7.744564056396484
    },
    {
      "epoch": 3.681182861328125e-07,
      "model_forward_time": 1.7610483169555664,
      "step": 74386
    },
    {
      "epoch": 3.681182861328125e-07,
      "step": 74386,
      "training_step_time": 7.217195749282837
    },
    {
      "epoch": 3.6907196044921876e-07,
      "model_forward_time": 1.7648820877075195,
      "step": 74387
    },
    {
      "epoch": 3.6907196044921876e-07,
      "step": 74387,
      "training_step_time": 8.570202589035034
    },
    {
      "epoch": 3.70025634765625e-07,
      "model_forward_time": 1.7645704746246338,
      "step": 74388
    },
    {
      "epoch": 3.70025634765625e-07,
      "step": 74388,
      "training_step_time": 8.440451860427856
    },
    {
      "epoch": 3.7097930908203126e-07,
      "model_forward_time": 1.825200080871582,
      "step": 74389
    },
    {
      "epoch": 3.7097930908203126e-07,
      "step": 74389,
      "training_step_time": 8.971522092819214
    },
    {
      "epoch": 3.719329833984375e-07,
      "grad_norm": 0.11675937473773956,
      "learning_rate": 1.6886063092402504e-06,
      "loss": 0.0549,
      "step": 74390
    },
    {
      "epoch": 3.719329833984375e-07,
      "model_forward_time": 2.3882699012756348,
      "step": 74390
    },
    {
      "dynamics_loss_avg": 0.05869293212890625,
      "epoch": 3.719329833984375e-07,
      "step": 74390
    },
    {
      "action_loss_avg": 0.0014538268413161858,
      "epoch": 3.719329833984375e-07,
      "step": 74390
    },
    {
      "epoch": 3.719329833984375e-07,
      "step": 74390,
      "training_step_time": 7.686936378479004
    },
    {
      "epoch": 3.7288665771484376e-07,
      "model_forward_time": 1.7647697925567627,
      "step": 74391
    },
    {
      "epoch": 3.7288665771484376e-07,
      "step": 74391,
      "training_step_time": 8.723273277282715
    },
    {
      "epoch": 3.7384033203125e-07,
      "model_forward_time": 1.76423978805542,
      "step": 74392
    },
    {
      "epoch": 3.7384033203125e-07,
      "step": 74392,
      "training_step_time": 7.869205474853516
    },
    {
      "epoch": 3.7479400634765627e-07,
      "model_forward_time": 1.7637560367584229,
      "step": 74393
    },
    {
      "epoch": 3.7479400634765627e-07,
      "step": 74393,
      "training_step_time": 7.5279340744018555
    },
    {
      "epoch": 3.757476806640625e-07,
      "model_forward_time": 1.7606959342956543,
      "step": 74394
    },
    {
      "epoch": 3.757476806640625e-07,
      "step": 74394,
      "training_step_time": 7.721481084823608
    },
    {
      "epoch": 3.7670135498046877e-07,
      "model_forward_time": 1.757755994796753,
      "step": 74395
    },
    {
      "epoch": 3.7670135498046877e-07,
      "step": 74395,
      "training_step_time": 8.010474681854248
    },
    {
      "epoch": 3.77655029296875e-07,
      "model_forward_time": 1.7634892463684082,
      "step": 74396
    },
    {
      "epoch": 3.77655029296875e-07,
      "step": 74396,
      "training_step_time": 7.737263202667236
    },
    {
      "epoch": 3.7860870361328127e-07,
      "model_forward_time": 1.7571518421173096,
      "step": 74397
    },
    {
      "epoch": 3.7860870361328127e-07,
      "step": 74397,
      "training_step_time": 8.301241159439087
    },
    {
      "epoch": 3.795623779296875e-07,
      "model_forward_time": 1.8246212005615234,
      "step": 74398
    },
    {
      "epoch": 3.795623779296875e-07,
      "step": 74398,
      "training_step_time": 8.687795639038086
    },
    {
      "epoch": 3.8051605224609377e-07,
      "model_forward_time": 2.3893630504608154,
      "step": 74399
    },
    {
      "epoch": 3.8051605224609377e-07,
      "step": 74399,
      "training_step_time": 8.427101612091064
    },
    {
      "epoch": 3.814697265625e-07,
      "grad_norm": 0.15826933085918427,
      "learning_rate": 1.6873676155863317e-06,
      "loss": 0.0553,
      "step": 74400
    },
    {
      "epoch": 3.814697265625e-07,
      "model_forward_time": 1.7603731155395508,
      "step": 74400
    },
    {
      "dynamics_loss_avg": 0.05030187275260687,
      "epoch": 3.814697265625e-07,
      "step": 74400
    },
    {
      "action_loss_avg": 0.0016417332109995187,
      "epoch": 3.814697265625e-07,
      "step": 74400
    },
    {
      "epoch": 3.814697265625e-07,
      "step": 74400,
      "training_step_time": 7.763450622558594
    },
    {
      "epoch": 3.8242340087890627e-07,
      "model_forward_time": 1.7623519897460938,
      "step": 74401
    },
    {
      "epoch": 3.8242340087890627e-07,
      "step": 74401,
      "training_step_time": 8.043869972229004
    },
    {
      "epoch": 3.833770751953125e-07,
      "model_forward_time": 1.7610838413238525,
      "step": 74402
    },
    {
      "epoch": 3.833770751953125e-07,
      "step": 74402,
      "training_step_time": 7.4736855030059814
    },
    {
      "epoch": 3.843307495117188e-07,
      "model_forward_time": 1.7602169513702393,
      "step": 74403
    },
    {
      "epoch": 3.843307495117188e-07,
      "step": 74403,
      "training_step_time": 7.717968702316284
    },
    {
      "epoch": 3.85284423828125e-07,
      "model_forward_time": 1.7491827011108398,
      "step": 74404
    },
    {
      "epoch": 3.85284423828125e-07,
      "step": 74404,
      "training_step_time": 7.812719106674194
    },
    {
      "epoch": 3.862380981445313e-07,
      "model_forward_time": 1.7670598030090332,
      "step": 74405
    },
    {
      "epoch": 3.862380981445313e-07,
      "step": 74405,
      "training_step_time": 7.723358392715454
    },
    {
      "epoch": 3.871917724609375e-07,
      "model_forward_time": 1.7675492763519287,
      "step": 74406
    },
    {
      "epoch": 3.871917724609375e-07,
      "step": 74406,
      "training_step_time": 8.803852558135986
    },
    {
      "epoch": 3.881454467773437e-07,
      "model_forward_time": 1.805417776107788,
      "step": 74407
    },
    {
      "epoch": 3.881454467773437e-07,
      "step": 74407,
      "training_step_time": 8.385571956634521
    },
    {
      "epoch": 3.8909912109375e-07,
      "model_forward_time": 2.4141790866851807,
      "step": 74408
    },
    {
      "epoch": 3.8909912109375e-07,
      "step": 74408,
      "training_step_time": 7.986083745956421
    },
    {
      "epoch": 3.900527954101562e-07,
      "model_forward_time": 1.7602190971374512,
      "step": 74409
    },
    {
      "epoch": 3.900527954101562e-07,
      "step": 74409,
      "training_step_time": 8.741835594177246
    },
    {
      "epoch": 3.910064697265625e-07,
      "grad_norm": 0.14143235981464386,
      "learning_rate": 1.6861292841969084e-06,
      "loss": 0.0561,
      "step": 74410
    },
    {
      "epoch": 3.910064697265625e-07,
      "model_forward_time": 1.76310396194458,
      "step": 74410
    },
    {
      "dynamics_loss_avg": 0.05348493764176965,
      "epoch": 3.910064697265625e-07,
      "step": 74410
    },
    {
      "action_loss_avg": 0.0011102533317171037,
      "epoch": 3.910064697265625e-07,
      "step": 74410
    },
    {
      "epoch": 3.910064697265625e-07,
      "step": 74410,
      "training_step_time": 8.7066068649292
    },
    {
      "epoch": 3.9196014404296873e-07,
      "model_forward_time": 1.779731273651123,
      "step": 74411
    },
    {
      "epoch": 3.9196014404296873e-07,
      "step": 74411,
      "training_step_time": 9.37430191040039
    },
    {
      "epoch": 3.92913818359375e-07,
      "model_forward_time": 1.761906385421753,
      "step": 74412
    },
    {
      "epoch": 3.92913818359375e-07,
      "step": 74412,
      "training_step_time": 7.728772878646851
    },
    {
      "epoch": 3.9386749267578123e-07,
      "model_forward_time": 1.7650277614593506,
      "step": 74413
    },
    {
      "epoch": 3.9386749267578123e-07,
      "step": 74413,
      "training_step_time": 7.431575775146484
    },
    {
      "epoch": 3.948211669921875e-07,
      "model_forward_time": 1.76615309715271,
      "step": 74414
    },
    {
      "epoch": 3.948211669921875e-07,
      "step": 74414,
      "training_step_time": 9.656683444976807
    },
    {
      "epoch": 3.9577484130859373e-07,
      "model_forward_time": 1.7689156532287598,
      "step": 74415
    },
    {
      "epoch": 3.9577484130859373e-07,
      "step": 74415,
      "training_step_time": 8.550506353378296
    },
    {
      "epoch": 3.96728515625e-07,
      "model_forward_time": 1.7623026371002197,
      "step": 74416
    },
    {
      "epoch": 3.96728515625e-07,
      "step": 74416,
      "training_step_time": 7.753204584121704
    },
    {
      "epoch": 3.9768218994140623e-07,
      "model_forward_time": 2.4119536876678467,
      "step": 74417
    },
    {
      "epoch": 3.9768218994140623e-07,
      "step": 74417,
      "training_step_time": 9.311936616897583
    },
    {
      "epoch": 3.986358642578125e-07,
      "model_forward_time": 1.3915455341339111,
      "step": 74418
    },
    {
      "epoch": 3.986358642578125e-07,
      "step": 74418,
      "training_step_time": 7.693597078323364
    },
    {
      "epoch": 3.9958953857421873e-07,
      "model_forward_time": 1.7691004276275635,
      "step": 74419
    },
    {
      "epoch": 3.9958953857421873e-07,
      "step": 74419,
      "training_step_time": 7.737454175949097
    },
    {
      "epoch": 4.00543212890625e-07,
      "grad_norm": 0.10130546987056732,
      "learning_rate": 1.6848913152074059e-06,
      "loss": 0.0561,
      "step": 74420
    },
    {
      "epoch": 4.00543212890625e-07,
      "model_forward_time": 1.770193338394165,
      "step": 74420
    },
    {
      "dynamics_loss_avg": 0.04149841628968716,
      "epoch": 4.00543212890625e-07,
      "step": 74420
    },
    {
      "action_loss_avg": 0.0009501908672973514,
      "epoch": 4.00543212890625e-07,
      "step": 74420
    },
    {
      "epoch": 4.00543212890625e-07,
      "step": 74420,
      "training_step_time": 7.7309746742248535
    },
    {
      "epoch": 4.0149688720703124e-07,
      "model_forward_time": 1.7683050632476807,
      "step": 74421
    },
    {
      "epoch": 4.0149688720703124e-07,
      "step": 74421,
      "training_step_time": 7.724136829376221
    },
    {
      "epoch": 4.024505615234375e-07,
      "model_forward_time": 1.762840986251831,
      "step": 74422
    },
    {
      "epoch": 4.024505615234375e-07,
      "step": 74422,
      "training_step_time": 7.362412214279175
    },
    {
      "epoch": 4.0340423583984374e-07,
      "model_forward_time": 1.7652881145477295,
      "step": 74423
    },
    {
      "epoch": 4.0340423583984374e-07,
      "step": 74423,
      "training_step_time": 8.932780981063843
    },
    {
      "epoch": 4.0435791015625e-07,
      "model_forward_time": 1.783104419708252,
      "step": 74424
    },
    {
      "epoch": 4.0435791015625e-07,
      "step": 74424,
      "training_step_time": 7.7994608879089355
    },
    {
      "epoch": 4.0531158447265624e-07,
      "model_forward_time": 1.763204574584961,
      "step": 74425
    },
    {
      "epoch": 4.0531158447265624e-07,
      "step": 74425,
      "training_step_time": 7.724831581115723
    },
    {
      "epoch": 4.062652587890625e-07,
      "model_forward_time": 2.415229082107544,
      "step": 74426
    },
    {
      "epoch": 4.062652587890625e-07,
      "step": 74426,
      "training_step_time": 8.415161848068237
    },
    {
      "epoch": 4.0721893310546874e-07,
      "model_forward_time": 1.391782283782959,
      "step": 74427
    },
    {
      "epoch": 4.0721893310546874e-07,
      "step": 74427,
      "training_step_time": 9.787297010421753
    },
    {
      "epoch": 4.08172607421875e-07,
      "model_forward_time": 1.7643547058105469,
      "step": 74428
    },
    {
      "epoch": 4.08172607421875e-07,
      "step": 74428,
      "training_step_time": 7.849077939987183
    },
    {
      "epoch": 4.0912628173828124e-07,
      "model_forward_time": 1.7692952156066895,
      "step": 74429
    },
    {
      "epoch": 4.0912628173828124e-07,
      "step": 74429,
      "training_step_time": 7.724478483200073
    },
    {
      "epoch": 4.100799560546875e-07,
      "grad_norm": 0.10138796269893646,
      "learning_rate": 1.6836537087532034e-06,
      "loss": 0.0582,
      "step": 74430
    },
    {
      "epoch": 4.100799560546875e-07,
      "model_forward_time": 1.7661547660827637,
      "step": 74430
    },
    {
      "dynamics_loss_avg": 0.05433425307273865,
      "epoch": 4.100799560546875e-07,
      "step": 74430
    },
    {
      "action_loss_avg": 0.0018588343926239758,
      "epoch": 4.100799560546875e-07,
      "step": 74430
    },
    {
      "epoch": 4.100799560546875e-07,
      "step": 74430,
      "training_step_time": 7.72368860244751
    },
    {
      "epoch": 4.1103363037109374e-07,
      "model_forward_time": 1.769662618637085,
      "step": 74431
    },
    {
      "epoch": 4.1103363037109374e-07,
      "step": 74431,
      "training_step_time": 7.653076171875
    },
    {
      "epoch": 4.119873046875e-07,
      "model_forward_time": 1.7641303539276123,
      "step": 74432
    },
    {
      "epoch": 4.119873046875e-07,
      "step": 74432,
      "training_step_time": 7.969770669937134
    },
    {
      "epoch": 4.1294097900390625e-07,
      "model_forward_time": 1.7702257633209229,
      "step": 74433
    },
    {
      "epoch": 4.1294097900390625e-07,
      "step": 74433,
      "training_step_time": 8.407315492630005
    },
    {
      "epoch": 4.138946533203125e-07,
      "model_forward_time": 1.7675576210021973,
      "step": 74434
    },
    {
      "epoch": 4.138946533203125e-07,
      "step": 74434,
      "training_step_time": 7.780525207519531
    },
    {
      "epoch": 4.1484832763671875e-07,
      "model_forward_time": 1.8302900791168213,
      "step": 74435
    },
    {
      "epoch": 4.1484832763671875e-07,
      "step": 74435,
      "training_step_time": 9.343103408813477
    },
    {
      "epoch": 4.15802001953125e-07,
      "model_forward_time": 2.411102533340454,
      "step": 74436
    },
    {
      "epoch": 4.15802001953125e-07,
      "step": 74436,
      "training_step_time": 7.770972013473511
    },
    {
      "epoch": 4.1675567626953125e-07,
      "model_forward_time": 1.7662744522094727,
      "step": 74437
    },
    {
      "epoch": 4.1675567626953125e-07,
      "step": 74437,
      "training_step_time": 7.730329513549805
    },
    {
      "epoch": 4.177093505859375e-07,
      "model_forward_time": 1.7703964710235596,
      "step": 74438
    },
    {
      "epoch": 4.177093505859375e-07,
      "step": 74438,
      "training_step_time": 7.622446298599243
    },
    {
      "epoch": 4.1866302490234375e-07,
      "model_forward_time": 1.772150993347168,
      "step": 74439
    },
    {
      "epoch": 4.1866302490234375e-07,
      "step": 74439,
      "training_step_time": 7.717161655426025
    },
    {
      "epoch": 4.1961669921875e-07,
      "grad_norm": 0.09101293236017227,
      "learning_rate": 1.6824164649696467e-06,
      "loss": 0.057,
      "step": 74440
    },
    {
      "epoch": 4.1961669921875e-07,
      "model_forward_time": 1.7664704322814941,
      "step": 74440
    },
    {
      "dynamics_loss_avg": 0.046269687451422216,
      "epoch": 4.1961669921875e-07,
      "step": 74440
    },
    {
      "action_loss_avg": 0.001113288738997653,
      "epoch": 4.1961669921875e-07,
      "step": 74440
    },
    {
      "epoch": 4.1961669921875e-07,
      "step": 74440,
      "training_step_time": 7.698343515396118
    },
    {
      "epoch": 4.2057037353515625e-07,
      "model_forward_time": 1.7760958671569824,
      "step": 74441
    },
    {
      "epoch": 4.2057037353515625e-07,
      "step": 74441,
      "training_step_time": 7.956595420837402
    },
    {
      "epoch": 4.215240478515625e-07,
      "model_forward_time": 1.7704851627349854,
      "step": 74442
    },
    {
      "epoch": 4.215240478515625e-07,
      "step": 74442,
      "training_step_time": 8.766438722610474
    },
    {
      "epoch": 4.2247772216796875e-07,
      "model_forward_time": 1.767730712890625,
      "step": 74443
    },
    {
      "epoch": 4.2247772216796875e-07,
      "step": 74443,
      "training_step_time": 7.774701118469238
    },
    {
      "epoch": 4.23431396484375e-07,
      "model_forward_time": 1.8259611129760742,
      "step": 74444
    },
    {
      "epoch": 4.23431396484375e-07,
      "step": 74444,
      "training_step_time": 9.68600583076477
    },
    {
      "epoch": 4.2438507080078126e-07,
      "model_forward_time": 2.4006080627441406,
      "step": 74445
    },
    {
      "epoch": 4.2438507080078126e-07,
      "step": 74445,
      "training_step_time": 7.96947717666626
    },
    {
      "epoch": 4.253387451171875e-07,
      "model_forward_time": 1.741565465927124,
      "step": 74446
    },
    {
      "epoch": 4.253387451171875e-07,
      "step": 74446,
      "training_step_time": 7.72564697265625
    },
    {
      "epoch": 4.2629241943359376e-07,
      "model_forward_time": 1.7690966129302979,
      "step": 74447
    },
    {
      "epoch": 4.2629241943359376e-07,
      "step": 74447,
      "training_step_time": 7.74445652961731
    },
    {
      "epoch": 4.2724609375e-07,
      "model_forward_time": 1.7575817108154297,
      "step": 74448
    },
    {
      "epoch": 4.2724609375e-07,
      "step": 74448,
      "training_step_time": 8.547119379043579
    },
    {
      "epoch": 4.2819976806640626e-07,
      "model_forward_time": 1.7654945850372314,
      "step": 74449
    },
    {
      "epoch": 4.2819976806640626e-07,
      "step": 74449,
      "training_step_time": 7.349802494049072
    },
    {
      "epoch": 4.291534423828125e-07,
      "grad_norm": 0.08159007877111435,
      "learning_rate": 1.6811795839920342e-06,
      "loss": 0.0569,
      "step": 74450
    },
    {
      "epoch": 4.291534423828125e-07,
      "model_forward_time": 1.7836639881134033,
      "step": 74450
    },
    {
      "dynamics_loss_avg": 0.06700487248599529,
      "epoch": 4.291534423828125e-07,
      "step": 74450
    },
    {
      "action_loss_avg": 0.0018167336471378802,
      "epoch": 4.291534423828125e-07,
      "step": 74450
    },
    {
      "epoch": 4.291534423828125e-07,
      "step": 74450,
      "training_step_time": 7.913376331329346
    },
    {
      "epoch": 4.3010711669921876e-07,
      "model_forward_time": 1.7624883651733398,
      "step": 74451
    },
    {
      "epoch": 4.3010711669921876e-07,
      "step": 74451,
      "training_step_time": 7.721351146697998
    },
    {
      "epoch": 4.31060791015625e-07,
      "model_forward_time": 1.761540174484253,
      "step": 74452
    },
    {
      "epoch": 4.31060791015625e-07,
      "step": 74452,
      "training_step_time": 8.413862466812134
    },
    {
      "epoch": 4.3201446533203126e-07,
      "model_forward_time": 1.7681448459625244,
      "step": 74453
    },
    {
      "epoch": 4.3201446533203126e-07,
      "step": 74453,
      "training_step_time": 9.402828454971313
    },
    {
      "epoch": 4.329681396484375e-07,
      "model_forward_time": 2.4252402782440186,
      "step": 74454
    },
    {
      "epoch": 4.329681396484375e-07,
      "step": 74454,
      "training_step_time": 8.468953371047974
    },
    {
      "epoch": 4.3392181396484376e-07,
      "model_forward_time": 1.7580976486206055,
      "step": 74455
    },
    {
      "epoch": 4.3392181396484376e-07,
      "step": 74455,
      "training_step_time": 8.421485185623169
    },
    {
      "epoch": 4.3487548828125e-07,
      "model_forward_time": 1.7422945499420166,
      "step": 74456
    },
    {
      "epoch": 4.3487548828125e-07,
      "step": 74456,
      "training_step_time": 8.506961822509766
    },
    {
      "epoch": 4.3582916259765627e-07,
      "model_forward_time": 1.7901349067687988,
      "step": 74457
    },
    {
      "epoch": 4.3582916259765627e-07,
      "step": 74457,
      "training_step_time": 7.75493860244751
    },
    {
      "epoch": 4.367828369140625e-07,
      "model_forward_time": 1.768702507019043,
      "step": 74458
    },
    {
      "epoch": 4.367828369140625e-07,
      "step": 74458,
      "training_step_time": 7.571136236190796
    },
    {
      "epoch": 4.3773651123046877e-07,
      "model_forward_time": 1.767749547958374,
      "step": 74459
    },
    {
      "epoch": 4.3773651123046877e-07,
      "step": 74459,
      "training_step_time": 7.564734220504761
    },
    {
      "epoch": 4.38690185546875e-07,
      "grad_norm": 0.06792973726987839,
      "learning_rate": 1.679943065955636e-06,
      "loss": 0.0539,
      "step": 74460
    },
    {
      "epoch": 4.38690185546875e-07,
      "model_forward_time": 1.7644360065460205,
      "step": 74460
    },
    {
      "dynamics_loss_avg": 0.05443116500973701,
      "epoch": 4.38690185546875e-07,
      "step": 74460
    },
    {
      "action_loss_avg": 0.0018631274258950725,
      "epoch": 4.38690185546875e-07,
      "step": 74460
    },
    {
      "epoch": 4.38690185546875e-07,
      "step": 74460,
      "training_step_time": 7.907123327255249
    },
    {
      "epoch": 4.3964385986328127e-07,
      "model_forward_time": 1.7648060321807861,
      "step": 74461
    },
    {
      "epoch": 4.3964385986328127e-07,
      "step": 74461,
      "training_step_time": 8.283167600631714
    },
    {
      "epoch": 4.405975341796875e-07,
      "model_forward_time": 1.771712303161621,
      "step": 74462
    },
    {
      "epoch": 4.405975341796875e-07,
      "step": 74462,
      "training_step_time": 7.792665481567383
    },
    {
      "epoch": 4.4155120849609377e-07,
      "model_forward_time": 2.4241299629211426,
      "step": 74463
    },
    {
      "epoch": 4.4155120849609377e-07,
      "step": 74463,
      "training_step_time": 8.96031904220581
    },
    {
      "epoch": 4.425048828125e-07,
      "model_forward_time": 1.761274814605713,
      "step": 74464
    },
    {
      "epoch": 4.425048828125e-07,
      "step": 74464,
      "training_step_time": 7.731014966964722
    },
    {
      "epoch": 4.4345855712890627e-07,
      "model_forward_time": 1.7369115352630615,
      "step": 74465
    },
    {
      "epoch": 4.4345855712890627e-07,
      "step": 74465,
      "training_step_time": 8.881257057189941
    },
    {
      "epoch": 4.444122314453125e-07,
      "model_forward_time": 1.7622222900390625,
      "step": 74466
    },
    {
      "epoch": 4.444122314453125e-07,
      "step": 74466,
      "training_step_time": 8.280651807785034
    },
    {
      "epoch": 4.453659057617188e-07,
      "model_forward_time": 1.7777316570281982,
      "step": 74467
    },
    {
      "epoch": 4.453659057617188e-07,
      "step": 74467,
      "training_step_time": 8.178665399551392
    },
    {
      "epoch": 4.46319580078125e-07,
      "model_forward_time": 1.764697551727295,
      "step": 74468
    },
    {
      "epoch": 4.46319580078125e-07,
      "step": 74468,
      "training_step_time": 7.598659038543701
    },
    {
      "epoch": 4.472732543945312e-07,
      "model_forward_time": 1.7620482444763184,
      "step": 74469
    },
    {
      "epoch": 4.472732543945312e-07,
      "step": 74469,
      "training_step_time": 9.620031833648682
    },
    {
      "epoch": 4.482269287109375e-07,
      "grad_norm": 0.10387734323740005,
      "learning_rate": 1.6787069109956711e-06,
      "loss": 0.0573,
      "step": 74470
    },
    {
      "epoch": 4.482269287109375e-07,
      "model_forward_time": 1.7616024017333984,
      "step": 74470
    },
    {
      "dynamics_loss_avg": 0.035505492612719534,
      "epoch": 4.482269287109375e-07,
      "step": 74470
    },
    {
      "action_loss_avg": 0.0009222144915838725,
      "epoch": 4.482269287109375e-07,
      "step": 74470
    },
    {
      "epoch": 4.482269287109375e-07,
      "step": 74470,
      "training_step_time": 8.16321325302124
    },
    {
      "epoch": 4.491806030273437e-07,
      "model_forward_time": 1.7637436389923096,
      "step": 74471
    },
    {
      "epoch": 4.491806030273437e-07,
      "step": 74471,
      "training_step_time": 8.190420389175415
    },
    {
      "epoch": 4.5013427734375e-07,
      "model_forward_time": 1.826685905456543,
      "step": 74472
    },
    {
      "epoch": 4.5013427734375e-07,
      "step": 74472,
      "training_step_time": 10.443259477615356
    },
    {
      "epoch": 4.5108795166015623e-07,
      "model_forward_time": 1.7577893733978271,
      "step": 74473
    },
    {
      "epoch": 4.5108795166015623e-07,
      "step": 74473,
      "training_step_time": 9.38380241394043
    },
    {
      "epoch": 4.520416259765625e-07,
      "model_forward_time": 1.7398419380187988,
      "step": 74474
    },
    {
      "epoch": 4.520416259765625e-07,
      "step": 74474,
      "training_step_time": 8.582139492034912
    },
    {
      "epoch": 4.5299530029296873e-07,
      "model_forward_time": 1.7678229808807373,
      "step": 74475
    },
    {
      "epoch": 4.5299530029296873e-07,
      "step": 74475,
      "training_step_time": 7.625405550003052
    },
    {
      "epoch": 4.53948974609375e-07,
      "model_forward_time": 1.756767749786377,
      "step": 74476
    },
    {
      "epoch": 4.53948974609375e-07,
      "step": 74476,
      "training_step_time": 7.449428081512451
    },
    {
      "epoch": 4.5490264892578123e-07,
      "model_forward_time": 1.76593017578125,
      "step": 74477
    },
    {
      "epoch": 4.5490264892578123e-07,
      "step": 74477,
      "training_step_time": 7.598244905471802
    },
    {
      "epoch": 4.558563232421875e-07,
      "model_forward_time": 1.7600929737091064,
      "step": 74478
    },
    {
      "epoch": 4.558563232421875e-07,
      "step": 74478,
      "training_step_time": 7.700016021728516
    },
    {
      "epoch": 4.5680999755859373e-07,
      "model_forward_time": 1.7610118389129639,
      "step": 74479
    },
    {
      "epoch": 4.5680999755859373e-07,
      "step": 74479,
      "training_step_time": 8.237850904464722
    },
    {
      "epoch": 4.57763671875e-07,
      "grad_norm": 0.10656479746103287,
      "learning_rate": 1.6774711192473264e-06,
      "loss": 0.0553,
      "step": 74480
    },
    {
      "epoch": 4.57763671875e-07,
      "model_forward_time": 1.7710916996002197,
      "step": 74480
    },
    {
      "dynamics_loss_avg": 0.03999928757548332,
      "epoch": 4.57763671875e-07,
      "step": 74480
    },
    {
      "action_loss_avg": 0.0016457162710139528,
      "epoch": 4.57763671875e-07,
      "step": 74480
    },
    {
      "epoch": 4.57763671875e-07,
      "step": 74480,
      "training_step_time": 9.474669933319092
    },
    {
      "epoch": 4.5871734619140623e-07,
      "model_forward_time": 1.8071496486663818,
      "step": 74481
    },
    {
      "epoch": 4.5871734619140623e-07,
      "step": 74481,
      "training_step_time": 9.46727705001831
    },
    {
      "epoch": 4.596710205078125e-07,
      "model_forward_time": 2.399935483932495,
      "step": 74482
    },
    {
      "epoch": 4.596710205078125e-07,
      "step": 74482,
      "training_step_time": 7.887208700180054
    },
    {
      "epoch": 4.6062469482421874e-07,
      "model_forward_time": 1.766038179397583,
      "step": 74483
    },
    {
      "epoch": 4.6062469482421874e-07,
      "step": 74483,
      "training_step_time": 9.255799531936646
    },
    {
      "epoch": 4.61578369140625e-07,
      "model_forward_time": 1.7568039894104004,
      "step": 74484
    },
    {
      "epoch": 4.61578369140625e-07,
      "step": 74484,
      "training_step_time": 7.644225835800171
    },
    {
      "epoch": 4.6253204345703124e-07,
      "model_forward_time": 1.7642560005187988,
      "step": 74485
    },
    {
      "epoch": 4.6253204345703124e-07,
      "step": 74485,
      "training_step_time": 7.382092714309692
    },
    {
      "epoch": 4.634857177734375e-07,
      "model_forward_time": 1.762176513671875,
      "step": 74486
    },
    {
      "epoch": 4.634857177734375e-07,
      "step": 74486,
      "training_step_time": 7.538370847702026
    },
    {
      "epoch": 4.6443939208984374e-07,
      "model_forward_time": 1.7674102783203125,
      "step": 74487
    },
    {
      "epoch": 4.6443939208984374e-07,
      "step": 74487,
      "training_step_time": 7.566808462142944
    },
    {
      "epoch": 4.6539306640625e-07,
      "model_forward_time": 1.7566394805908203,
      "step": 74488
    },
    {
      "epoch": 4.6539306640625e-07,
      "step": 74488,
      "training_step_time": 7.527997970581055
    },
    {
      "epoch": 4.6634674072265624e-07,
      "model_forward_time": 1.762937068939209,
      "step": 74489
    },
    {
      "epoch": 4.6634674072265624e-07,
      "step": 74489,
      "training_step_time": 8.340039491653442
    },
    {
      "epoch": 4.673004150390625e-07,
      "grad_norm": 0.08825695514678955,
      "learning_rate": 1.6762356908457427e-06,
      "loss": 0.057,
      "step": 74490
    },
    {
      "epoch": 4.673004150390625e-07,
      "model_forward_time": 1.7747151851654053,
      "step": 74490
    },
    {
      "dynamics_loss_avg": 0.059182648174464704,
      "epoch": 4.673004150390625e-07,
      "step": 74490
    },
    {
      "action_loss_avg": 0.0009267782326787711,
      "epoch": 4.673004150390625e-07,
      "step": 74490
    },
    {
      "epoch": 4.673004150390625e-07,
      "step": 74490,
      "training_step_time": 7.531753063201904
    },
    {
      "epoch": 4.6825408935546874e-07,
      "model_forward_time": 2.3951268196105957,
      "step": 74491
    },
    {
      "epoch": 4.6825408935546874e-07,
      "step": 74491,
      "training_step_time": 7.880807161331177
    },
    {
      "epoch": 4.69207763671875e-07,
      "model_forward_time": 1.7640082836151123,
      "step": 74492
    },
    {
      "epoch": 4.69207763671875e-07,
      "step": 74492,
      "training_step_time": 7.675911903381348
    },
    {
      "epoch": 4.7016143798828124e-07,
      "model_forward_time": 1.7777721881866455,
      "step": 74493
    },
    {
      "epoch": 4.7016143798828124e-07,
      "step": 74493,
      "training_step_time": 7.74214243888855
    },
    {
      "epoch": 4.711151123046875e-07,
      "model_forward_time": 1.7615351676940918,
      "step": 74494
    },
    {
      "epoch": 4.711151123046875e-07,
      "step": 74494,
      "training_step_time": 7.484612941741943
    },
    {
      "epoch": 4.7206878662109375e-07,
      "model_forward_time": 1.7787628173828125,
      "step": 74495
    },
    {
      "epoch": 4.7206878662109375e-07,
      "step": 74495,
      "training_step_time": 7.503716230392456
    },
    {
      "epoch": 4.730224609375e-07,
      "model_forward_time": 1.7678802013397217,
      "step": 74496
    },
    {
      "epoch": 4.730224609375e-07,
      "step": 74496,
      "training_step_time": 8.559236764907837
    },
    {
      "epoch": 4.7397613525390625e-07,
      "model_forward_time": 1.7718734741210938,
      "step": 74497
    },
    {
      "epoch": 4.7397613525390625e-07,
      "step": 74497,
      "training_step_time": 8.29991364479065
    },
    {
      "epoch": 4.749298095703125e-07,
      "model_forward_time": 1.765834093093872,
      "step": 74498
    },
    {
      "epoch": 4.749298095703125e-07,
      "step": 74498,
      "training_step_time": 7.679334878921509
    },
    {
      "epoch": 4.7588348388671875e-07,
      "model_forward_time": 1.7693989276885986,
      "step": 74499
    },
    {
      "epoch": 4.7588348388671875e-07,
      "step": 74499,
      "training_step_time": 8.400989532470703
    },
    {
      "epoch": 4.76837158203125e-07,
      "grad_norm": 0.20754562318325043,
      "learning_rate": 1.6750006259260288e-06,
      "loss": 0.0545,
      "step": 74500
    },
    {
      "epoch": 4.76837158203125e-07,
      "model_forward_time": 2.417839765548706,
      "step": 74500
    },
    {
      "dynamics_loss_avg": 0.04202274437993765,
      "epoch": 4.76837158203125e-07,
      "step": 74500
    },
    {
      "action_loss_avg": 0.0014134539727820084,
      "epoch": 4.76837158203125e-07,
      "step": 74500
    },
    {
      "epoch": 4.76837158203125e-07,
      "step": 74500,
      "training_step_time": 7.823357820510864
    },
    {
      "epoch": 4.777908325195313e-07,
      "model_forward_time": 1.758436918258667,
      "step": 74501
    },
    {
      "epoch": 4.777908325195313e-07,
      "step": 74501,
      "training_step_time": 7.693717956542969
    },
    {
      "epoch": 4.787445068359375e-07,
      "model_forward_time": 1.7666430473327637,
      "step": 74502
    },
    {
      "epoch": 4.787445068359375e-07,
      "step": 74502,
      "training_step_time": 7.712785005569458
    },
    {
      "epoch": 4.796981811523438e-07,
      "model_forward_time": 1.743042230606079,
      "step": 74503
    },
    {
      "epoch": 4.796981811523438e-07,
      "step": 74503,
      "training_step_time": 8.521835088729858
    },
    {
      "epoch": 4.8065185546875e-07,
      "model_forward_time": 1.7730398178100586,
      "step": 74504
    },
    {
      "epoch": 4.8065185546875e-07,
      "step": 74504,
      "training_step_time": 7.5102317333221436
    },
    {
      "epoch": 4.816055297851563e-07,
      "model_forward_time": 1.7613863945007324,
      "step": 74505
    },
    {
      "epoch": 4.816055297851563e-07,
      "step": 74505,
      "training_step_time": 7.70073127746582
    },
    {
      "epoch": 4.825592041015625e-07,
      "model_forward_time": 1.7661292552947998,
      "step": 74506
    },
    {
      "epoch": 4.825592041015625e-07,
      "step": 74506,
      "training_step_time": 8.764671802520752
    },
    {
      "epoch": 4.835128784179688e-07,
      "model_forward_time": 1.7675299644470215,
      "step": 74507
    },
    {
      "epoch": 4.835128784179688e-07,
      "step": 74507,
      "training_step_time": 8.22907829284668
    },
    {
      "epoch": 4.84466552734375e-07,
      "model_forward_time": 1.7669312953948975,
      "step": 74508
    },
    {
      "epoch": 4.84466552734375e-07,
      "step": 74508,
      "training_step_time": 8.182339668273926
    },
    {
      "epoch": 4.854202270507813e-07,
      "model_forward_time": 2.4273338317871094,
      "step": 74509
    },
    {
      "epoch": 4.854202270507813e-07,
      "step": 74509,
      "training_step_time": 9.686338901519775
    },
    {
      "epoch": 4.863739013671875e-07,
      "grad_norm": 0.08488593995571136,
      "learning_rate": 1.6737659246232441e-06,
      "loss": 0.0559,
      "step": 74510
    },
    {
      "epoch": 4.863739013671875e-07,
      "model_forward_time": 1.7555975914001465,
      "step": 74510
    },
    {
      "dynamics_loss_avg": 0.051524842716753484,
      "epoch": 4.863739013671875e-07,
      "step": 74510
    },
    {
      "action_loss_avg": 0.0009085219207918271,
      "epoch": 4.863739013671875e-07,
      "step": 74510
    },
    {
      "epoch": 4.863739013671875e-07,
      "step": 74510,
      "training_step_time": 9.394383192062378
    },
    {
      "epoch": 4.873275756835938e-07,
      "model_forward_time": 1.768763542175293,
      "step": 74511
    },
    {
      "epoch": 4.873275756835938e-07,
      "step": 74511,
      "training_step_time": 7.782974720001221
    },
    {
      "epoch": 4.8828125e-07,
      "model_forward_time": 1.758772611618042,
      "step": 74512
    },
    {
      "epoch": 4.8828125e-07,
      "step": 74512,
      "training_step_time": 8.205632448196411
    },
    {
      "epoch": 4.892349243164063e-07,
      "model_forward_time": 1.7527265548706055,
      "step": 74513
    },
    {
      "epoch": 4.892349243164063e-07,
      "step": 74513,
      "training_step_time": 8.188678503036499
    },
    {
      "epoch": 4.901885986328125e-07,
      "model_forward_time": 1.7788147926330566,
      "step": 74514
    },
    {
      "epoch": 4.901885986328125e-07,
      "step": 74514,
      "training_step_time": 7.611074209213257
    },
    {
      "epoch": 4.911422729492188e-07,
      "model_forward_time": 1.7897741794586182,
      "step": 74515
    },
    {
      "epoch": 4.911422729492188e-07,
      "step": 74515,
      "training_step_time": 7.803735256195068
    },
    {
      "epoch": 4.92095947265625e-07,
      "model_forward_time": 1.7632901668548584,
      "step": 74516
    },
    {
      "epoch": 4.92095947265625e-07,
      "step": 74516,
      "training_step_time": 7.591926097869873
    },
    {
      "epoch": 4.930496215820313e-07,
      "model_forward_time": 1.772024393081665,
      "step": 74517
    },
    {
      "epoch": 4.930496215820313e-07,
      "step": 74517,
      "training_step_time": 9.9722421169281
    },
    {
      "epoch": 4.940032958984375e-07,
      "model_forward_time": 2.4109880924224854,
      "step": 74518
    },
    {
      "epoch": 4.940032958984375e-07,
      "step": 74518,
      "training_step_time": 8.476779699325562
    },
    {
      "epoch": 4.949569702148438e-07,
      "model_forward_time": 1.7641160488128662,
      "step": 74519
    },
    {
      "epoch": 4.949569702148438e-07,
      "step": 74519,
      "training_step_time": 9.00753116607666
    },
    {
      "epoch": 4.9591064453125e-07,
      "grad_norm": 0.1347823292016983,
      "learning_rate": 1.67253158707242e-06,
      "loss": 0.0564,
      "step": 74520
    },
    {
      "epoch": 4.9591064453125e-07,
      "model_forward_time": 1.7707200050354004,
      "step": 74520
    },
    {
      "dynamics_loss_avg": 0.0696068562567234,
      "epoch": 4.9591064453125e-07,
      "step": 74520
    },
    {
      "action_loss_avg": 0.0015857052931096405,
      "epoch": 4.9591064453125e-07,
      "step": 74520
    },
    {
      "epoch": 4.9591064453125e-07,
      "step": 74520,
      "training_step_time": 7.7414538860321045
    },
    {
      "epoch": 4.968643188476563e-07,
      "model_forward_time": 1.7606964111328125,
      "step": 74521
    },
    {
      "epoch": 4.968643188476563e-07,
      "step": 74521,
      "training_step_time": 8.460459232330322
    },
    {
      "epoch": 4.978179931640625e-07,
      "model_forward_time": 1.7422432899475098,
      "step": 74522
    },
    {
      "epoch": 4.978179931640625e-07,
      "step": 74522,
      "training_step_time": 8.34864354133606
    },
    {
      "epoch": 4.987716674804688e-07,
      "model_forward_time": 1.7591915130615234,
      "step": 74523
    },
    {
      "epoch": 4.987716674804688e-07,
      "step": 74523,
      "training_step_time": 7.594683885574341
    },
    {
      "epoch": 4.99725341796875e-07,
      "model_forward_time": 1.7649753093719482,
      "step": 74524
    },
    {
      "epoch": 4.99725341796875e-07,
      "step": 74524,
      "training_step_time": 7.816630601882935
    },
    {
      "epoch": 5.006790161132813e-07,
      "model_forward_time": 1.3974404335021973,
      "step": 74525
    },
    {
      "epoch": 5.006790161132813e-07,
      "step": 74525,
      "training_step_time": 7.60799503326416
    },
    {
      "epoch": 5.016326904296875e-07,
      "model_forward_time": 1.76564621925354,
      "step": 74526
    },
    {
      "epoch": 5.016326904296875e-07,
      "step": 74526,
      "training_step_time": 8.515867948532104
    },
    {
      "epoch": 5.025863647460938e-07,
      "model_forward_time": 1.7858142852783203,
      "step": 74527
    },
    {
      "epoch": 5.025863647460938e-07,
      "step": 74527,
      "training_step_time": 7.670474529266357
    },
    {
      "epoch": 5.035400390625e-07,
      "model_forward_time": 2.3969736099243164,
      "step": 74528
    },
    {
      "epoch": 5.035400390625e-07,
      "step": 74528,
      "training_step_time": 8.689758539199829
    },
    {
      "epoch": 5.044937133789063e-07,
      "model_forward_time": 1.751417636871338,
      "step": 74529
    },
    {
      "epoch": 5.044937133789063e-07,
      "step": 74529,
      "training_step_time": 7.683022975921631
    },
    {
      "epoch": 5.054473876953125e-07,
      "grad_norm": 0.08605457842350006,
      "learning_rate": 1.6712976134085375e-06,
      "loss": 0.0576,
      "step": 74530
    },
    {
      "epoch": 5.054473876953125e-07,
      "model_forward_time": 1.760378122329712,
      "step": 74530
    },
    {
      "dynamics_loss_avg": 0.05616516005247831,
      "epoch": 5.054473876953125e-07,
      "step": 74530
    },
    {
      "action_loss_avg": 0.001373964291997254,
      "epoch": 5.054473876953125e-07,
      "step": 74530
    },
    {
      "epoch": 5.054473876953125e-07,
      "step": 74530,
      "training_step_time": 7.644581317901611
    },
    {
      "epoch": 5.064010620117188e-07,
      "model_forward_time": 1.760664939880371,
      "step": 74531
    },
    {
      "epoch": 5.064010620117188e-07,
      "step": 74531,
      "training_step_time": 7.4051525592803955
    },
    {
      "epoch": 5.07354736328125e-07,
      "model_forward_time": 1.7585535049438477,
      "step": 74532
    },
    {
      "epoch": 5.07354736328125e-07,
      "step": 74532,
      "training_step_time": 7.532542943954468
    },
    {
      "epoch": 5.083084106445313e-07,
      "model_forward_time": 1.7565891742706299,
      "step": 74533
    },
    {
      "epoch": 5.083084106445313e-07,
      "step": 74533,
      "training_step_time": 7.662888526916504
    },
    {
      "epoch": 5.092620849609375e-07,
      "model_forward_time": 1.7552073001861572,
      "step": 74534
    },
    {
      "epoch": 5.092620849609375e-07,
      "step": 74534,
      "training_step_time": 8.866666078567505
    },
    {
      "epoch": 5.102157592773438e-07,
      "model_forward_time": 1.7547657489776611,
      "step": 74535
    },
    {
      "epoch": 5.102157592773438e-07,
      "step": 74535,
      "training_step_time": 8.416852951049805
    },
    {
      "epoch": 5.1116943359375e-07,
      "model_forward_time": 1.7499818801879883,
      "step": 74536
    },
    {
      "epoch": 5.1116943359375e-07,
      "step": 74536,
      "training_step_time": 8.234723806381226
    },
    {
      "epoch": 5.121231079101563e-07,
      "model_forward_time": 2.408991575241089,
      "step": 74537
    },
    {
      "epoch": 5.121231079101563e-07,
      "step": 74537,
      "training_step_time": 9.491237878799438
    },
    {
      "epoch": 5.130767822265625e-07,
      "model_forward_time": 1.752479076385498,
      "step": 74538
    },
    {
      "epoch": 5.130767822265625e-07,
      "step": 74538,
      "training_step_time": 8.216726064682007
    },
    {
      "epoch": 5.140304565429688e-07,
      "model_forward_time": 1.7637431621551514,
      "step": 74539
    },
    {
      "epoch": 5.140304565429688e-07,
      "step": 74539,
      "training_step_time": 8.56749153137207
    },
    {
      "epoch": 5.14984130859375e-07,
      "grad_norm": 0.09972307831048965,
      "learning_rate": 1.6700640037665444e-06,
      "loss": 0.0562,
      "step": 74540
    },
    {
      "epoch": 5.14984130859375e-07,
      "model_forward_time": 1.754098892211914,
      "step": 74540
    },
    {
      "dynamics_loss_avg": 0.04195242114365101,
      "epoch": 5.14984130859375e-07,
      "step": 74540
    },
    {
      "action_loss_avg": 0.0012478527380153537,
      "epoch": 5.14984130859375e-07,
      "step": 74540
    },
    {
      "epoch": 5.14984130859375e-07,
      "step": 74540,
      "training_step_time": 7.42467188835144
    },
    {
      "epoch": 5.159378051757813e-07,
      "model_forward_time": 1.7594704627990723,
      "step": 74541
    },
    {
      "epoch": 5.159378051757813e-07,
      "step": 74541,
      "training_step_time": 7.533817768096924
    },
    {
      "epoch": 5.168914794921875e-07,
      "model_forward_time": 1.7661662101745605,
      "step": 74542
    },
    {
      "epoch": 5.168914794921875e-07,
      "step": 74542,
      "training_step_time": 7.704458475112915
    },
    {
      "epoch": 5.178451538085938e-07,
      "model_forward_time": 1.7619805335998535,
      "step": 74543
    },
    {
      "epoch": 5.178451538085938e-07,
      "step": 74543,
      "training_step_time": 7.673240900039673
    },
    {
      "epoch": 5.18798828125e-07,
      "model_forward_time": 1.7631144523620605,
      "step": 74544
    },
    {
      "epoch": 5.18798828125e-07,
      "step": 74544,
      "training_step_time": 8.545945882797241
    },
    {
      "epoch": 5.197525024414063e-07,
      "model_forward_time": 1.7632687091827393,
      "step": 74545
    },
    {
      "epoch": 5.197525024414063e-07,
      "step": 74545,
      "training_step_time": 7.680706262588501
    },
    {
      "epoch": 5.207061767578125e-07,
      "model_forward_time": 2.4090325832366943,
      "step": 74546
    },
    {
      "epoch": 5.207061767578125e-07,
      "step": 74546,
      "training_step_time": 8.699050188064575
    },
    {
      "epoch": 5.216598510742188e-07,
      "model_forward_time": 1.7624375820159912,
      "step": 74547
    },
    {
      "epoch": 5.216598510742188e-07,
      "step": 74547,
      "training_step_time": 7.74324107170105
    },
    {
      "epoch": 5.22613525390625e-07,
      "model_forward_time": 1.7651934623718262,
      "step": 74548
    },
    {
      "epoch": 5.22613525390625e-07,
      "step": 74548,
      "training_step_time": 8.240496635437012
    },
    {
      "epoch": 5.235671997070313e-07,
      "model_forward_time": 1.7609038352966309,
      "step": 74549
    },
    {
      "epoch": 5.235671997070313e-07,
      "step": 74549,
      "training_step_time": 7.651619911193848
    },
    {
      "epoch": 5.245208740234375e-07,
      "grad_norm": 0.10823147743940353,
      "learning_rate": 1.668830758281344e-06,
      "loss": 0.0566,
      "step": 74550
    },
    {
      "epoch": 5.245208740234375e-07,
      "model_forward_time": 1.7625072002410889,
      "step": 74550
    },
    {
      "dynamics_loss_avg": 0.04954561712220311,
      "epoch": 5.245208740234375e-07,
      "step": 74550
    },
    {
      "action_loss_avg": 0.001679909093945753,
      "epoch": 5.245208740234375e-07,
      "step": 74550
    },
    {
      "epoch": 5.245208740234375e-07,
      "step": 74550,
      "training_step_time": 8.244232416152954
    },
    {
      "epoch": 5.254745483398438e-07,
      "model_forward_time": 1.7610886096954346,
      "step": 74551
    },
    {
      "epoch": 5.254745483398438e-07,
      "step": 74551,
      "training_step_time": 7.694144010543823
    },
    {
      "epoch": 5.2642822265625e-07,
      "model_forward_time": 1.7752435207366943,
      "step": 74552
    },
    {
      "epoch": 5.2642822265625e-07,
      "step": 74552,
      "training_step_time": 7.539579391479492
    },
    {
      "epoch": 5.273818969726563e-07,
      "model_forward_time": 1.745981216430664,
      "step": 74553
    },
    {
      "epoch": 5.273818969726563e-07,
      "step": 74553,
      "training_step_time": 9.462945938110352
    },
    {
      "epoch": 5.283355712890625e-07,
      "model_forward_time": 1.7524995803833008,
      "step": 74554
    },
    {
      "epoch": 5.283355712890625e-07,
      "step": 74554,
      "training_step_time": 8.568696737289429
    },
    {
      "epoch": 5.292892456054688e-07,
      "model_forward_time": 2.402960777282715,
      "step": 74555
    },
    {
      "epoch": 5.292892456054688e-07,
      "step": 74555,
      "training_step_time": 9.362072706222534
    },
    {
      "epoch": 5.30242919921875e-07,
      "model_forward_time": 1.7738451957702637,
      "step": 74556
    },
    {
      "epoch": 5.30242919921875e-07,
      "step": 74556,
      "training_step_time": 7.6333208084106445
    },
    {
      "epoch": 5.311965942382813e-07,
      "model_forward_time": 1.765573263168335,
      "step": 74557
    },
    {
      "epoch": 5.311965942382813e-07,
      "step": 74557,
      "training_step_time": 7.708919525146484
    },
    {
      "epoch": 5.321502685546875e-07,
      "model_forward_time": 1.7507710456848145,
      "step": 74558
    },
    {
      "epoch": 5.321502685546875e-07,
      "step": 74558,
      "training_step_time": 7.623473167419434
    },
    {
      "epoch": 5.331039428710938e-07,
      "model_forward_time": 1.754363775253296,
      "step": 74559
    },
    {
      "epoch": 5.331039428710938e-07,
      "step": 74559,
      "training_step_time": 7.475987195968628
    },
    {
      "epoch": 5.340576171875e-07,
      "grad_norm": 0.13456198573112488,
      "learning_rate": 1.6675978770878055e-06,
      "loss": 0.0559,
      "step": 74560
    },
    {
      "epoch": 5.340576171875e-07,
      "model_forward_time": 1.7591667175292969,
      "step": 74560
    },
    {
      "dynamics_loss_avg": 0.0658625633455813,
      "epoch": 5.340576171875e-07,
      "step": 74560
    },
    {
      "action_loss_avg": 0.001277987749199383,
      "epoch": 5.340576171875e-07,
      "step": 74560
    },
    {
      "epoch": 5.340576171875e-07,
      "step": 74560,
      "training_step_time": 8.163951396942139
    },
    {
      "epoch": 5.350112915039063e-07,
      "model_forward_time": 1.764221429824829,
      "step": 74561
    },
    {
      "epoch": 5.350112915039063e-07,
      "step": 74561,
      "training_step_time": 8.587692260742188
    },
    {
      "epoch": 5.359649658203126e-07,
      "model_forward_time": 1.7582497596740723,
      "step": 74562
    },
    {
      "epoch": 5.359649658203126e-07,
      "step": 74562,
      "training_step_time": 7.553972482681274
    },
    {
      "epoch": 5.369186401367187e-07,
      "model_forward_time": 1.7598912715911865,
      "step": 74563
    },
    {
      "epoch": 5.369186401367187e-07,
      "step": 74563,
      "training_step_time": 7.643142223358154
    },
    {
      "epoch": 5.37872314453125e-07,
      "model_forward_time": 1.8141748905181885,
      "step": 74564
    },
    {
      "epoch": 5.37872314453125e-07,
      "step": 74564,
      "training_step_time": 9.522597312927246
    },
    {
      "epoch": 5.388259887695312e-07,
      "model_forward_time": 1.7575292587280273,
      "step": 74565
    },
    {
      "epoch": 5.388259887695312e-07,
      "step": 74565,
      "training_step_time": 7.624356269836426
    },
    {
      "epoch": 5.397796630859375e-07,
      "model_forward_time": 1.7622921466827393,
      "step": 74566
    },
    {
      "epoch": 5.397796630859375e-07,
      "step": 74566,
      "training_step_time": 8.235551595687866
    },
    {
      "epoch": 5.407333374023437e-07,
      "model_forward_time": 1.7646186351776123,
      "step": 74567
    },
    {
      "epoch": 5.407333374023437e-07,
      "step": 74567,
      "training_step_time": 7.603221654891968
    },
    {
      "epoch": 5.4168701171875e-07,
      "model_forward_time": 1.7335340976715088,
      "step": 74568
    },
    {
      "epoch": 5.4168701171875e-07,
      "step": 74568,
      "training_step_time": 8.288774490356445
    },
    {
      "epoch": 5.426406860351562e-07,
      "model_forward_time": 1.7595276832580566,
      "step": 74569
    },
    {
      "epoch": 5.426406860351562e-07,
      "step": 74569,
      "training_step_time": 7.624639987945557
    },
    {
      "epoch": 5.435943603515625e-07,
      "grad_norm": 0.10787417739629745,
      "learning_rate": 1.6663653603207508e-06,
      "loss": 0.0551,
      "step": 74570
    },
    {
      "epoch": 5.435943603515625e-07,
      "model_forward_time": 1.7422022819519043,
      "step": 74570
    },
    {
      "dynamics_loss_avg": 0.06107930680736899,
      "epoch": 5.435943603515625e-07,
      "step": 74570
    },
    {
      "action_loss_avg": 0.001914455002406612,
      "epoch": 5.435943603515625e-07,
      "step": 74570
    },
    {
      "epoch": 5.435943603515625e-07,
      "step": 74570,
      "training_step_time": 7.629506826400757
    },
    {
      "epoch": 5.445480346679687e-07,
      "model_forward_time": 1.7354629039764404,
      "step": 74571
    },
    {
      "epoch": 5.445480346679687e-07,
      "step": 74571,
      "training_step_time": 8.492585182189941
    },
    {
      "epoch": 5.45501708984375e-07,
      "model_forward_time": 1.7541899681091309,
      "step": 74572
    },
    {
      "epoch": 5.45501708984375e-07,
      "step": 74572,
      "training_step_time": 8.422931432723999
    },
    {
      "epoch": 5.464553833007812e-07,
      "model_forward_time": 1.8210821151733398,
      "step": 74573
    },
    {
      "epoch": 5.464553833007812e-07,
      "step": 74573,
      "training_step_time": 8.220052242279053
    },
    {
      "epoch": 5.474090576171875e-07,
      "model_forward_time": 2.3967669010162354,
      "step": 74574
    },
    {
      "epoch": 5.474090576171875e-07,
      "step": 74574,
      "training_step_time": 8.593868017196655
    },
    {
      "epoch": 5.483627319335937e-07,
      "model_forward_time": 1.7560336589813232,
      "step": 74575
    },
    {
      "epoch": 5.483627319335937e-07,
      "step": 74575,
      "training_step_time": 7.686280965805054
    },
    {
      "epoch": 5.4931640625e-07,
      "model_forward_time": 1.764014482498169,
      "step": 74576
    },
    {
      "epoch": 5.4931640625e-07,
      "step": 74576,
      "training_step_time": 7.726137638092041
    },
    {
      "epoch": 5.502700805664062e-07,
      "model_forward_time": 1.744520902633667,
      "step": 74577
    },
    {
      "epoch": 5.502700805664062e-07,
      "step": 74577,
      "training_step_time": 7.669230222702026
    },
    {
      "epoch": 5.512237548828125e-07,
      "model_forward_time": 1.7632288932800293,
      "step": 74578
    },
    {
      "epoch": 5.512237548828125e-07,
      "step": 74578,
      "training_step_time": 8.518428802490234
    },
    {
      "epoch": 5.521774291992187e-07,
      "model_forward_time": 1.7464408874511719,
      "step": 74579
    },
    {
      "epoch": 5.521774291992187e-07,
      "step": 74579,
      "training_step_time": 8.29471468925476
    },
    {
      "epoch": 5.53131103515625e-07,
      "grad_norm": 0.0764179602265358,
      "learning_rate": 1.6651332081149685e-06,
      "loss": 0.056,
      "step": 74580
    },
    {
      "epoch": 5.53131103515625e-07,
      "model_forward_time": 1.7550122737884521,
      "step": 74580
    },
    {
      "dynamics_loss_avg": 0.05452694408595562,
      "epoch": 5.53131103515625e-07,
      "step": 74580
    },
    {
      "action_loss_avg": 0.0014833789726253598,
      "epoch": 5.53131103515625e-07,
      "step": 74580
    },
    {
      "epoch": 5.53131103515625e-07,
      "step": 74580,
      "training_step_time": 7.445909023284912
    },
    {
      "epoch": 5.540847778320312e-07,
      "model_forward_time": 1.7540473937988281,
      "step": 74581
    },
    {
      "epoch": 5.540847778320312e-07,
      "step": 74581,
      "training_step_time": 9.18004298210144
    },
    {
      "epoch": 5.550384521484375e-07,
      "model_forward_time": 1.8052582740783691,
      "step": 74582
    },
    {
      "epoch": 5.550384521484375e-07,
      "step": 74582,
      "training_step_time": 7.71390175819397
    },
    {
      "epoch": 5.559921264648437e-07,
      "model_forward_time": 2.41921329498291,
      "step": 74583
    },
    {
      "epoch": 5.559921264648437e-07,
      "step": 74583,
      "training_step_time": 7.713581562042236
    },
    {
      "epoch": 5.5694580078125e-07,
      "model_forward_time": 1.7640657424926758,
      "step": 74584
    },
    {
      "epoch": 5.5694580078125e-07,
      "step": 74584,
      "training_step_time": 7.67839789390564
    },
    {
      "epoch": 5.578994750976562e-07,
      "model_forward_time": 1.7383313179016113,
      "step": 74585
    },
    {
      "epoch": 5.578994750976562e-07,
      "step": 74585,
      "training_step_time": 7.660259008407593
    },
    {
      "epoch": 5.588531494140625e-07,
      "model_forward_time": 1.72652268409729,
      "step": 74586
    },
    {
      "epoch": 5.588531494140625e-07,
      "step": 74586,
      "training_step_time": 9.206500053405762
    },
    {
      "epoch": 5.598068237304687e-07,
      "model_forward_time": 1.7491850852966309,
      "step": 74587
    },
    {
      "epoch": 5.598068237304687e-07,
      "step": 74587,
      "training_step_time": 7.570279836654663
    },
    {
      "epoch": 5.60760498046875e-07,
      "model_forward_time": 1.744530200958252,
      "step": 74588
    },
    {
      "epoch": 5.60760498046875e-07,
      "step": 74588,
      "training_step_time": 8.324018955230713
    },
    {
      "epoch": 5.617141723632812e-07,
      "model_forward_time": 1.7463102340698242,
      "step": 74589
    },
    {
      "epoch": 5.617141723632812e-07,
      "step": 74589,
      "training_step_time": 7.418850421905518
    },
    {
      "epoch": 5.626678466796875e-07,
      "grad_norm": 0.09340396523475647,
      "learning_rate": 1.6639014206052045e-06,
      "loss": 0.0561,
      "step": 74590
    },
    {
      "epoch": 5.626678466796875e-07,
      "model_forward_time": 1.7406890392303467,
      "step": 74590
    },
    {
      "dynamics_loss_avg": 0.04548671543598175,
      "epoch": 5.626678466796875e-07,
      "step": 74590
    },
    {
      "action_loss_avg": 0.0010873854800593108,
      "epoch": 5.626678466796875e-07,
      "step": 74590
    },
    {
      "epoch": 5.626678466796875e-07,
      "step": 74590,
      "training_step_time": 7.594877004623413
    },
    {
      "epoch": 5.636215209960937e-07,
      "model_forward_time": 1.7441368103027344,
      "step": 74591
    },
    {
      "epoch": 5.636215209960937e-07,
      "step": 74591,
      "training_step_time": 7.675762176513672
    },
    {
      "epoch": 5.645751953125e-07,
      "model_forward_time": 2.3941781520843506,
      "step": 74592
    },
    {
      "epoch": 5.645751953125e-07,
      "step": 74592,
      "training_step_time": 8.720267534255981
    },
    {
      "epoch": 5.655288696289062e-07,
      "model_forward_time": 1.7539172172546387,
      "step": 74593
    },
    {
      "epoch": 5.655288696289062e-07,
      "step": 74593,
      "training_step_time": 8.144181966781616
    },
    {
      "epoch": 5.664825439453125e-07,
      "model_forward_time": 1.7455644607543945,
      "step": 74594
    },
    {
      "epoch": 5.664825439453125e-07,
      "step": 74594,
      "training_step_time": 8.77678632736206
    },
    {
      "epoch": 5.674362182617187e-07,
      "model_forward_time": 1.7458162307739258,
      "step": 74595
    },
    {
      "epoch": 5.674362182617187e-07,
      "step": 74595,
      "training_step_time": 7.443624496459961
    },
    {
      "epoch": 5.68389892578125e-07,
      "model_forward_time": 1.733630657196045,
      "step": 74596
    },
    {
      "epoch": 5.68389892578125e-07,
      "step": 74596,
      "training_step_time": 7.496530294418335
    },
    {
      "epoch": 5.693435668945312e-07,
      "model_forward_time": 1.7456815242767334,
      "step": 74597
    },
    {
      "epoch": 5.693435668945312e-07,
      "step": 74597,
      "training_step_time": 7.6474714279174805
    },
    {
      "epoch": 5.702972412109375e-07,
      "model_forward_time": 1.7472045421600342,
      "step": 74598
    },
    {
      "epoch": 5.702972412109375e-07,
      "step": 74598,
      "training_step_time": 8.168781280517578
    },
    {
      "epoch": 5.712509155273437e-07,
      "model_forward_time": 1.7376015186309814,
      "step": 74599
    },
    {
      "epoch": 5.712509155273437e-07,
      "step": 74599,
      "training_step_time": 7.590555667877197
    },
    {
      "epoch": 5.7220458984375e-07,
      "grad_norm": 0.09183189272880554,
      "learning_rate": 1.6626699979261656e-06,
      "loss": 0.0573,
      "step": 74600
    },
    {
      "epoch": 5.7220458984375e-07,
      "model_forward_time": 1.7371296882629395,
      "step": 74600
    },
    {
      "dynamics_loss_avg": 0.06710370071232319,
      "epoch": 5.7220458984375e-07,
      "step": 74600
    },
    {
      "action_loss_avg": 0.0016579611808992922,
      "epoch": 5.7220458984375e-07,
      "step": 74600
    },
    {
      "epoch": 5.7220458984375e-07,
      "step": 74600,
      "training_step_time": 8.676831007003784
    },
    {
      "epoch": 5.731582641601562e-07,
      "model_forward_time": 2.3835222721099854,
      "step": 74601
    },
    {
      "epoch": 5.731582641601562e-07,
      "step": 74601,
      "training_step_time": 10.341301441192627
    },
    {
      "epoch": 5.741119384765625e-07,
      "model_forward_time": 1.736023187637329,
      "step": 74602
    },
    {
      "epoch": 5.741119384765625e-07,
      "step": 74602,
      "training_step_time": 7.608374834060669
    },
    {
      "epoch": 5.750656127929687e-07,
      "model_forward_time": 1.7352004051208496,
      "step": 74603
    },
    {
      "epoch": 5.750656127929687e-07,
      "step": 74603,
      "training_step_time": 7.664115905761719
    },
    {
      "epoch": 5.76019287109375e-07,
      "model_forward_time": 1.7454283237457275,
      "step": 74604
    },
    {
      "epoch": 5.76019287109375e-07,
      "step": 74604,
      "training_step_time": 7.377020597457886
    },
    {
      "epoch": 5.769729614257812e-07,
      "model_forward_time": 1.7389757633209229,
      "step": 74605
    },
    {
      "epoch": 5.769729614257812e-07,
      "step": 74605,
      "training_step_time": 7.5994343757629395
    },
    {
      "epoch": 5.779266357421875e-07,
      "model_forward_time": 1.738417387008667,
      "step": 74606
    },
    {
      "epoch": 5.779266357421875e-07,
      "step": 74606,
      "training_step_time": 7.621185064315796
    },
    {
      "epoch": 5.788803100585937e-07,
      "model_forward_time": 1.742311954498291,
      "step": 74607
    },
    {
      "epoch": 5.788803100585937e-07,
      "step": 74607,
      "training_step_time": 7.499455690383911
    },
    {
      "epoch": 5.79833984375e-07,
      "model_forward_time": 1.7414839267730713,
      "step": 74608
    },
    {
      "epoch": 5.79833984375e-07,
      "step": 74608,
      "training_step_time": 7.669471502304077
    },
    {
      "epoch": 5.807876586914062e-07,
      "model_forward_time": 1.7452151775360107,
      "step": 74609
    },
    {
      "epoch": 5.807876586914062e-07,
      "step": 74609,
      "training_step_time": 9.82263708114624
    },
    {
      "epoch": 5.817413330078125e-07,
      "grad_norm": 0.11046954989433289,
      "learning_rate": 1.6614389402125198e-06,
      "loss": 0.0542,
      "step": 74610
    },
    {
      "epoch": 5.817413330078125e-07,
      "model_forward_time": 2.4100353717803955,
      "step": 74610
    },
    {
      "dynamics_loss_avg": 0.0661909998394549,
      "epoch": 5.817413330078125e-07,
      "step": 74610
    },
    {
      "action_loss_avg": 0.0028251856958377173,
      "epoch": 5.817413330078125e-07,
      "step": 74610
    },
    {
      "epoch": 5.817413330078125e-07,
      "step": 74610,
      "training_step_time": 8.27472186088562
    },
    {
      "epoch": 5.826950073242187e-07,
      "model_forward_time": 1.7434351444244385,
      "step": 74611
    },
    {
      "epoch": 5.826950073242187e-07,
      "step": 74611,
      "training_step_time": 7.659921884536743
    },
    {
      "epoch": 5.83648681640625e-07,
      "model_forward_time": 1.7439134120941162,
      "step": 74612
    },
    {
      "epoch": 5.83648681640625e-07,
      "step": 74612,
      "training_step_time": 9.534438371658325
    },
    {
      "epoch": 5.846023559570312e-07,
      "model_forward_time": 1.7433717250823975,
      "step": 74613
    },
    {
      "epoch": 5.846023559570312e-07,
      "step": 74613,
      "training_step_time": 8.369454622268677
    },
    {
      "epoch": 5.855560302734375e-07,
      "model_forward_time": 1.7408092021942139,
      "step": 74614
    },
    {
      "epoch": 5.855560302734375e-07,
      "step": 74614,
      "training_step_time": 7.648136138916016
    },
    {
      "epoch": 5.865097045898437e-07,
      "model_forward_time": 1.7342889308929443,
      "step": 74615
    },
    {
      "epoch": 5.865097045898437e-07,
      "step": 74615,
      "training_step_time": 8.235647916793823
    },
    {
      "epoch": 5.8746337890625e-07,
      "model_forward_time": 1.7361531257629395,
      "step": 74616
    },
    {
      "epoch": 5.8746337890625e-07,
      "step": 74616,
      "training_step_time": 7.537519454956055
    },
    {
      "epoch": 5.884170532226562e-07,
      "model_forward_time": 1.369291067123413,
      "step": 74617
    },
    {
      "epoch": 5.884170532226562e-07,
      "step": 74617,
      "training_step_time": 7.653843402862549
    },
    {
      "epoch": 5.893707275390625e-07,
      "model_forward_time": 1.7431530952453613,
      "step": 74618
    },
    {
      "epoch": 5.893707275390625e-07,
      "step": 74618,
      "training_step_time": 7.613895893096924
    },
    {
      "epoch": 5.903244018554687e-07,
      "model_forward_time": 1.8511106967926025,
      "step": 74619
    },
    {
      "epoch": 5.903244018554687e-07,
      "step": 74619,
      "training_step_time": 8.278501749038696
    },
    {
      "epoch": 5.91278076171875e-07,
      "grad_norm": 0.09013744443655014,
      "learning_rate": 1.6602082475988902e-06,
      "loss": 0.0553,
      "step": 74620
    },
    {
      "epoch": 5.91278076171875e-07,
      "model_forward_time": 2.34409761428833,
      "step": 74620
    },
    {
      "dynamics_loss_avg": 0.0580287029966712,
      "epoch": 5.91278076171875e-07,
      "step": 74620
    },
    {
      "action_loss_avg": 0.001627913155243732,
      "epoch": 5.91278076171875e-07,
      "step": 74620
    },
    {
      "epoch": 5.91278076171875e-07,
      "step": 74620,
      "training_step_time": 7.793413877487183
    },
    {
      "epoch": 5.922317504882812e-07,
      "model_forward_time": 1.7400405406951904,
      "step": 74621
    },
    {
      "epoch": 5.922317504882812e-07,
      "step": 74621,
      "training_step_time": 7.658824682235718
    },
    {
      "epoch": 5.931854248046875e-07,
      "model_forward_time": 1.738584280014038,
      "step": 74622
    },
    {
      "epoch": 5.931854248046875e-07,
      "step": 74622,
      "training_step_time": 8.657977104187012
    },
    {
      "epoch": 5.941390991210937e-07,
      "model_forward_time": 1.7396645545959473,
      "step": 74623
    },
    {
      "epoch": 5.941390991210937e-07,
      "step": 74623,
      "training_step_time": 8.597025632858276
    },
    {
      "epoch": 5.950927734375e-07,
      "model_forward_time": 1.7442569732666016,
      "step": 74624
    },
    {
      "epoch": 5.950927734375e-07,
      "step": 74624,
      "training_step_time": 7.631698846817017
    },
    {
      "epoch": 5.960464477539062e-07,
      "model_forward_time": 1.7409813404083252,
      "step": 74625
    },
    {
      "epoch": 5.960464477539062e-07,
      "step": 74625,
      "training_step_time": 8.319844484329224
    },
    {
      "epoch": 5.970001220703125e-07,
      "model_forward_time": 1.3861515522003174,
      "step": 74626
    },
    {
      "epoch": 5.970001220703125e-07,
      "step": 74626,
      "training_step_time": 8.229422807693481
    },
    {
      "epoch": 5.979537963867188e-07,
      "model_forward_time": 1.7548797130584717,
      "step": 74627
    },
    {
      "epoch": 5.979537963867188e-07,
      "step": 74627,
      "training_step_time": 8.591663122177124
    },
    {
      "epoch": 5.98907470703125e-07,
      "model_forward_time": 1.869701623916626,
      "step": 74628
    },
    {
      "epoch": 5.98907470703125e-07,
      "step": 74628,
      "training_step_time": 8.32609748840332
    },
    {
      "epoch": 5.998611450195313e-07,
      "model_forward_time": 2.3906335830688477,
      "step": 74629
    },
    {
      "epoch": 5.998611450195313e-07,
      "step": 74629,
      "training_step_time": 7.68694543838501
    },
    {
      "epoch": 6.008148193359375e-07,
      "grad_norm": 0.08485811203718185,
      "learning_rate": 1.6589779202198674e-06,
      "loss": 0.0538,
      "step": 74630
    },
    {
      "epoch": 6.008148193359375e-07,
      "model_forward_time": 1.7512853145599365,
      "step": 74630
    },
    {
      "dynamics_loss_avg": 0.06812723241746425,
      "epoch": 6.008148193359375e-07,
      "step": 74630
    },
    {
      "action_loss_avg": 0.001712174993008375,
      "epoch": 6.008148193359375e-07,
      "step": 74630
    },
    {
      "epoch": 6.008148193359375e-07,
      "step": 74630,
      "training_step_time": 8.163862705230713
    },
    {
      "epoch": 6.017684936523438e-07,
      "model_forward_time": 1.7636725902557373,
      "step": 74631
    },
    {
      "epoch": 6.017684936523438e-07,
      "step": 74631,
      "training_step_time": 8.693570137023926
    },
    {
      "epoch": 6.0272216796875e-07,
      "model_forward_time": 1.7681679725646973,
      "step": 74632
    },
    {
      "epoch": 6.0272216796875e-07,
      "step": 74632,
      "training_step_time": 7.995787143707275
    },
    {
      "epoch": 6.036758422851563e-07,
      "model_forward_time": 1.7534356117248535,
      "step": 74633
    },
    {
      "epoch": 6.036758422851563e-07,
      "step": 74633,
      "training_step_time": 8.577146530151367
    },
    {
      "epoch": 6.046295166015625e-07,
      "model_forward_time": 1.7439556121826172,
      "step": 74634
    },
    {
      "epoch": 6.046295166015625e-07,
      "step": 74634,
      "training_step_time": 7.632449626922607
    },
    {
      "epoch": 6.055831909179688e-07,
      "model_forward_time": 1.7595458030700684,
      "step": 74635
    },
    {
      "epoch": 6.055831909179688e-07,
      "step": 74635,
      "training_step_time": 7.503317594528198
    },
    {
      "epoch": 6.06536865234375e-07,
      "model_forward_time": 1.7584259510040283,
      "step": 74636
    },
    {
      "epoch": 6.06536865234375e-07,
      "step": 74636,
      "training_step_time": 7.554335594177246
    },
    {
      "epoch": 6.074905395507813e-07,
      "model_forward_time": 1.4869167804718018,
      "step": 74637
    },
    {
      "epoch": 6.074905395507813e-07,
      "step": 74637,
      "training_step_time": 8.62098217010498
    },
    {
      "epoch": 6.084442138671875e-07,
      "model_forward_time": 2.317345380783081,
      "step": 74638
    },
    {
      "epoch": 6.084442138671875e-07,
      "step": 74638,
      "training_step_time": 7.685095310211182
    },
    {
      "epoch": 6.093978881835938e-07,
      "model_forward_time": 1.7386915683746338,
      "step": 74639
    },
    {
      "epoch": 6.093978881835938e-07,
      "step": 74639,
      "training_step_time": 7.6455912590026855
    },
    {
      "epoch": 6.103515625e-07,
      "grad_norm": 0.08609066158533096,
      "learning_rate": 1.6577479582099925e-06,
      "loss": 0.0538,
      "step": 74640
    },
    {
      "epoch": 6.103515625e-07,
      "model_forward_time": 1.7444016933441162,
      "step": 74640
    },
    {
      "dynamics_loss_avg": 0.05180294560268521,
      "epoch": 6.103515625e-07,
      "step": 74640
    },
    {
      "action_loss_avg": 0.0014664794318377972,
      "epoch": 6.103515625e-07,
      "step": 74640
    },
    {
      "epoch": 6.103515625e-07,
      "step": 74640,
      "training_step_time": 7.617175579071045
    },
    {
      "epoch": 6.113052368164063e-07,
      "model_forward_time": 1.7353127002716064,
      "step": 74641
    },
    {
      "epoch": 6.113052368164063e-07,
      "step": 74641,
      "training_step_time": 7.406336545944214
    },
    {
      "epoch": 6.122589111328125e-07,
      "model_forward_time": 1.7570946216583252,
      "step": 74642
    },
    {
      "epoch": 6.122589111328125e-07,
      "step": 74642,
      "training_step_time": 8.554044723510742
    },
    {
      "epoch": 6.132125854492188e-07,
      "model_forward_time": 1.7617430686950684,
      "step": 74643
    },
    {
      "epoch": 6.132125854492188e-07,
      "step": 74643,
      "training_step_time": 7.621302843093872
    },
    {
      "epoch": 6.14166259765625e-07,
      "model_forward_time": 1.7559425830841064,
      "step": 74644
    },
    {
      "epoch": 6.14166259765625e-07,
      "step": 74644,
      "training_step_time": 8.098596334457397
    },
    {
      "epoch": 6.151199340820313e-07,
      "model_forward_time": 1.7593073844909668,
      "step": 74645
    },
    {
      "epoch": 6.151199340820313e-07,
      "step": 74645,
      "training_step_time": 9.658048868179321
    },
    {
      "epoch": 6.160736083984375e-07,
      "model_forward_time": 1.8524353504180908,
      "step": 74646
    },
    {
      "epoch": 6.160736083984375e-07,
      "step": 74646,
      "training_step_time": 8.490113735198975
    },
    {
      "epoch": 6.170272827148438e-07,
      "model_forward_time": 1.7526891231536865,
      "step": 74647
    },
    {
      "epoch": 6.170272827148438e-07,
      "step": 74647,
      "training_step_time": 8.611514806747437
    },
    {
      "epoch": 6.1798095703125e-07,
      "model_forward_time": 2.383918046951294,
      "step": 74648
    },
    {
      "epoch": 6.1798095703125e-07,
      "step": 74648,
      "training_step_time": 7.676484107971191
    },
    {
      "epoch": 6.189346313476563e-07,
      "model_forward_time": 1.7816638946533203,
      "step": 74649
    },
    {
      "epoch": 6.189346313476563e-07,
      "step": 74649,
      "training_step_time": 8.621070146560669
    },
    {
      "epoch": 6.198883056640625e-07,
      "grad_norm": 0.09266464412212372,
      "learning_rate": 1.6565183617037789e-06,
      "loss": 0.0539,
      "step": 74650
    },
    {
      "epoch": 6.198883056640625e-07,
      "model_forward_time": 1.750952959060669,
      "step": 74650
    },
    {
      "dynamics_loss_avg": 0.037337626051157716,
      "epoch": 6.198883056640625e-07,
      "step": 74650
    },
    {
      "action_loss_avg": 0.0025900528708007187,
      "epoch": 6.198883056640625e-07,
      "step": 74650
    },
    {
      "epoch": 6.198883056640625e-07,
      "step": 74650,
      "training_step_time": 7.391300439834595
    },
    {
      "epoch": 6.208419799804688e-07,
      "model_forward_time": 1.7544224262237549,
      "step": 74651
    },
    {
      "epoch": 6.208419799804688e-07,
      "step": 74651,
      "training_step_time": 8.792019367218018
    },
    {
      "epoch": 6.21795654296875e-07,
      "model_forward_time": 1.7672595977783203,
      "step": 74652
    },
    {
      "epoch": 6.21795654296875e-07,
      "step": 74652,
      "training_step_time": 7.6239542961120605
    },
    {
      "epoch": 6.227493286132813e-07,
      "model_forward_time": 1.7587201595306396,
      "step": 74653
    },
    {
      "epoch": 6.227493286132813e-07,
      "step": 74653,
      "training_step_time": 7.46570086479187
    },
    {
      "epoch": 6.237030029296875e-07,
      "model_forward_time": 1.745755910873413,
      "step": 74654
    },
    {
      "epoch": 6.237030029296875e-07,
      "step": 74654,
      "training_step_time": 7.673417329788208
    },
    {
      "epoch": 6.246566772460938e-07,
      "model_forward_time": 1.8619747161865234,
      "step": 74655
    },
    {
      "epoch": 6.246566772460938e-07,
      "step": 74655,
      "training_step_time": 7.788219213485718
    },
    {
      "epoch": 6.256103515625e-07,
      "model_forward_time": 1.7646758556365967,
      "step": 74656
    },
    {
      "epoch": 6.256103515625e-07,
      "step": 74656,
      "training_step_time": 7.83398699760437
    },
    {
      "epoch": 6.265640258789063e-07,
      "model_forward_time": 2.4079935550689697,
      "step": 74657
    },
    {
      "epoch": 6.265640258789063e-07,
      "step": 74657,
      "training_step_time": 7.851447582244873
    },
    {
      "epoch": 6.275177001953125e-07,
      "model_forward_time": 1.3857123851776123,
      "step": 74658
    },
    {
      "epoch": 6.275177001953125e-07,
      "step": 74658,
      "training_step_time": 7.79438042640686
    },
    {
      "epoch": 6.284713745117188e-07,
      "model_forward_time": 1.75862455368042,
      "step": 74659
    },
    {
      "epoch": 6.284713745117188e-07,
      "step": 74659,
      "training_step_time": 7.573358058929443
    },
    {
      "epoch": 6.29425048828125e-07,
      "grad_norm": 0.09684521704912186,
      "learning_rate": 1.6552891308356895e-06,
      "loss": 0.0527,
      "step": 74660
    },
    {
      "epoch": 6.29425048828125e-07,
      "model_forward_time": 1.754927158355713,
      "step": 74660
    },
    {
      "dynamics_loss_avg": 0.03979745525866747,
      "epoch": 6.29425048828125e-07,
      "step": 74660
    },
    {
      "action_loss_avg": 0.0011624322301940992,
      "epoch": 6.29425048828125e-07,
      "step": 74660
    },
    {
      "epoch": 6.29425048828125e-07,
      "step": 74660,
      "training_step_time": 8.29227328300476
    },
    {
      "epoch": 6.303787231445313e-07,
      "model_forward_time": 1.771575689315796,
      "step": 74661
    },
    {
      "epoch": 6.303787231445313e-07,
      "step": 74661,
      "training_step_time": 9.058232069015503
    },
    {
      "epoch": 6.313323974609375e-07,
      "model_forward_time": 1.7588815689086914,
      "step": 74662
    },
    {
      "epoch": 6.313323974609375e-07,
      "step": 74662,
      "training_step_time": 7.554997205734253
    },
    {
      "epoch": 6.322860717773438e-07,
      "model_forward_time": 1.7689340114593506,
      "step": 74663
    },
    {
      "epoch": 6.322860717773438e-07,
      "step": 74663,
      "training_step_time": 7.5611793994903564
    },
    {
      "epoch": 6.3323974609375e-07,
      "model_forward_time": 1.8183116912841797,
      "step": 74664
    },
    {
      "epoch": 6.3323974609375e-07,
      "step": 74664,
      "training_step_time": 8.29039454460144
    },
    {
      "epoch": 6.341934204101563e-07,
      "model_forward_time": 1.753523826599121,
      "step": 74665
    },
    {
      "epoch": 6.341934204101563e-07,
      "step": 74665,
      "training_step_time": 8.335523128509521
    },
    {
      "epoch": 6.351470947265625e-07,
      "model_forward_time": 2.407963991165161,
      "step": 74666
    },
    {
      "epoch": 6.351470947265625e-07,
      "step": 74666,
      "training_step_time": 8.608685731887817
    },
    {
      "epoch": 6.361007690429688e-07,
      "model_forward_time": 1.392838478088379,
      "step": 74667
    },
    {
      "epoch": 6.361007690429688e-07,
      "step": 74667,
      "training_step_time": 8.748628616333008
    },
    {
      "epoch": 6.37054443359375e-07,
      "model_forward_time": 1.7497754096984863,
      "step": 74668
    },
    {
      "epoch": 6.37054443359375e-07,
      "step": 74668,
      "training_step_time": 7.826305389404297
    },
    {
      "epoch": 6.380081176757813e-07,
      "model_forward_time": 1.7576868534088135,
      "step": 74669
    },
    {
      "epoch": 6.380081176757813e-07,
      "step": 74669,
      "training_step_time": 7.717327833175659
    },
    {
      "epoch": 6.389617919921875e-07,
      "grad_norm": 0.07809167355298996,
      "learning_rate": 1.6540602657401528e-06,
      "loss": 0.0539,
      "step": 74670
    },
    {
      "epoch": 6.389617919921875e-07,
      "model_forward_time": 1.7550559043884277,
      "step": 74670
    },
    {
      "dynamics_loss_avg": 0.06697643958032132,
      "epoch": 6.389617919921875e-07,
      "step": 74670
    },
    {
      "action_loss_avg": 0.0017762096860678867,
      "epoch": 6.389617919921875e-07,
      "step": 74670
    },
    {
      "epoch": 6.389617919921875e-07,
      "step": 74670,
      "training_step_time": 7.720229148864746
    },
    {
      "epoch": 6.399154663085938e-07,
      "model_forward_time": 1.7311747074127197,
      "step": 74671
    },
    {
      "epoch": 6.399154663085938e-07,
      "step": 74671,
      "training_step_time": 9.244785785675049
    },
    {
      "epoch": 6.40869140625e-07,
      "model_forward_time": 1.7439653873443604,
      "step": 74672
    },
    {
      "epoch": 6.40869140625e-07,
      "step": 74672,
      "training_step_time": 7.644173860549927
    },
    {
      "epoch": 6.418228149414063e-07,
      "model_forward_time": 1.7678377628326416,
      "step": 74673
    },
    {
      "epoch": 6.418228149414063e-07,
      "step": 74673,
      "training_step_time": 8.433400869369507
    },
    {
      "epoch": 6.427764892578125e-07,
      "model_forward_time": 1.7361717224121094,
      "step": 74674
    },
    {
      "epoch": 6.427764892578125e-07,
      "step": 74674,
      "training_step_time": 8.407560348510742
    },
    {
      "epoch": 6.437301635742188e-07,
      "model_forward_time": 2.3057448863983154,
      "step": 74675
    },
    {
      "epoch": 6.437301635742188e-07,
      "step": 74675,
      "training_step_time": 8.814184427261353
    },
    {
      "epoch": 6.44683837890625e-07,
      "model_forward_time": 1.7427639961242676,
      "step": 74676
    },
    {
      "epoch": 6.44683837890625e-07,
      "step": 74676,
      "training_step_time": 8.317687034606934
    },
    {
      "epoch": 6.456375122070313e-07,
      "model_forward_time": 1.7407348155975342,
      "step": 74677
    },
    {
      "epoch": 6.456375122070313e-07,
      "step": 74677,
      "training_step_time": 8.757081985473633
    },
    {
      "epoch": 6.465911865234375e-07,
      "model_forward_time": 1.7412140369415283,
      "step": 74678
    },
    {
      "epoch": 6.465911865234375e-07,
      "step": 74678,
      "training_step_time": 7.5742573738098145
    },
    {
      "epoch": 6.475448608398438e-07,
      "model_forward_time": 1.7442541122436523,
      "step": 74679
    },
    {
      "epoch": 6.475448608398438e-07,
      "step": 74679,
      "training_step_time": 7.68643045425415
    },
    {
      "epoch": 6.4849853515625e-07,
      "grad_norm": 0.16850608587265015,
      "learning_rate": 1.6528317665515537e-06,
      "loss": 0.0553,
      "step": 74680
    },
    {
      "epoch": 6.4849853515625e-07,
      "model_forward_time": 1.7351069450378418,
      "step": 74680
    },
    {
      "dynamics_loss_avg": 0.05617031529545784,
      "epoch": 6.4849853515625e-07,
      "step": 74680
    },
    {
      "action_loss_avg": 0.0012914897539303637,
      "epoch": 6.4849853515625e-07,
      "step": 74680
    },
    {
      "epoch": 6.4849853515625e-07,
      "step": 74680,
      "training_step_time": 8.477270126342773
    },
    {
      "epoch": 6.494522094726563e-07,
      "model_forward_time": 1.367969036102295,
      "step": 74681
    },
    {
      "epoch": 6.494522094726563e-07,
      "step": 74681,
      "training_step_time": 8.043028831481934
    },
    {
      "epoch": 6.504058837890625e-07,
      "model_forward_time": 1.7318601608276367,
      "step": 74682
    },
    {
      "epoch": 6.504058837890625e-07,
      "step": 74682,
      "training_step_time": 9.797268629074097
    },
    {
      "epoch": 6.513595581054688e-07,
      "model_forward_time": 1.7393434047698975,
      "step": 74683
    },
    {
      "epoch": 6.513595581054688e-07,
      "step": 74683,
      "training_step_time": 8.7578763961792
    },
    {
      "epoch": 6.52313232421875e-07,
      "model_forward_time": 1.7384095191955566,
      "step": 74684
    },
    {
      "epoch": 6.52313232421875e-07,
      "step": 74684,
      "training_step_time": 8.83229374885559
    },
    {
      "epoch": 6.532669067382813e-07,
      "model_forward_time": 2.357234477996826,
      "step": 74685
    },
    {
      "epoch": 6.532669067382813e-07,
      "step": 74685,
      "training_step_time": 7.8211140632629395
    },
    {
      "epoch": 6.542205810546876e-07,
      "model_forward_time": 1.7400391101837158,
      "step": 74686
    },
    {
      "epoch": 6.542205810546876e-07,
      "step": 74686,
      "training_step_time": 7.763960838317871
    },
    {
      "epoch": 6.551742553710938e-07,
      "model_forward_time": 1.7366943359375,
      "step": 74687
    },
    {
      "epoch": 6.551742553710938e-07,
      "step": 74687,
      "training_step_time": 7.5472636222839355
    },
    {
      "epoch": 6.561279296875e-07,
      "model_forward_time": 1.7384357452392578,
      "step": 74688
    },
    {
      "epoch": 6.561279296875e-07,
      "step": 74688,
      "training_step_time": 7.659458637237549
    },
    {
      "epoch": 6.570816040039062e-07,
      "model_forward_time": 1.7392144203186035,
      "step": 74689
    },
    {
      "epoch": 6.570816040039062e-07,
      "step": 74689,
      "training_step_time": 8.82492184638977
    },
    {
      "epoch": 6.580352783203124e-07,
      "grad_norm": 0.09452574700117111,
      "learning_rate": 1.6516036334042418e-06,
      "loss": 0.053,
      "step": 74690
    },
    {
      "epoch": 6.580352783203124e-07,
      "model_forward_time": 1.7414920330047607,
      "step": 74690
    },
    {
      "dynamics_loss_avg": 0.05840310547500849,
      "epoch": 6.580352783203124e-07,
      "step": 74690
    },
    {
      "action_loss_avg": 0.0010403040650999174,
      "epoch": 6.580352783203124e-07,
      "step": 74690
    },
    {
      "epoch": 6.580352783203124e-07,
      "step": 74690,
      "training_step_time": 7.496673107147217
    },
    {
      "epoch": 6.589889526367187e-07,
      "model_forward_time": 1.742460012435913,
      "step": 74691
    },
    {
      "epoch": 6.589889526367187e-07,
      "step": 74691,
      "training_step_time": 8.555006265640259
    },
    {
      "epoch": 6.59942626953125e-07,
      "model_forward_time": 1.8527438640594482,
      "step": 74692
    },
    {
      "epoch": 6.59942626953125e-07,
      "step": 74692,
      "training_step_time": 7.700169324874878
    },
    {
      "epoch": 6.608963012695312e-07,
      "model_forward_time": 1.7410130500793457,
      "step": 74693
    },
    {
      "epoch": 6.608963012695312e-07,
      "step": 74693,
      "training_step_time": 8.220573902130127
    },
    {
      "epoch": 6.618499755859375e-07,
      "model_forward_time": 1.9561779499053955,
      "step": 74694
    },
    {
      "epoch": 6.618499755859375e-07,
      "step": 74694,
      "training_step_time": 9.170417308807373
    },
    {
      "epoch": 6.628036499023437e-07,
      "model_forward_time": 1.7429988384246826,
      "step": 74695
    },
    {
      "epoch": 6.628036499023437e-07,
      "step": 74695,
      "training_step_time": 9.659369468688965
    },
    {
      "epoch": 6.6375732421875e-07,
      "model_forward_time": 1.8809010982513428,
      "step": 74696
    },
    {
      "epoch": 6.6375732421875e-07,
      "step": 74696,
      "training_step_time": 7.624972581863403
    },
    {
      "epoch": 6.647109985351562e-07,
      "model_forward_time": 1.8931000232696533,
      "step": 74697
    },
    {
      "epoch": 6.647109985351562e-07,
      "step": 74697,
      "training_step_time": 7.569130182266235
    },
    {
      "epoch": 6.656646728515625e-07,
      "model_forward_time": 1.8776111602783203,
      "step": 74698
    },
    {
      "epoch": 6.656646728515625e-07,
      "step": 74698,
      "training_step_time": 7.660346031188965
    },
    {
      "epoch": 6.666183471679687e-07,
      "model_forward_time": 1.8812053203582764,
      "step": 74699
    },
    {
      "epoch": 6.666183471679687e-07,
      "step": 74699,
      "training_step_time": 8.39997148513794
    },
    {
      "epoch": 6.67572021484375e-07,
      "grad_norm": 0.078237384557724,
      "learning_rate": 1.6503758664325192e-06,
      "loss": 0.0551,
      "step": 74700
    },
    {
      "epoch": 6.67572021484375e-07,
      "model_forward_time": 1.885152816772461,
      "step": 74700
    },
    {
      "dynamics_loss_avg": 0.046098354086279866,
      "epoch": 6.67572021484375e-07,
      "step": 74700
    },
    {
      "action_loss_avg": 0.0014942604291718453,
      "epoch": 6.67572021484375e-07,
      "step": 74700
    },
    {
      "epoch": 6.67572021484375e-07,
      "step": 74700,
      "training_step_time": 8.723496437072754
    },
    {
      "epoch": 6.685256958007812e-07,
      "model_forward_time": 1.880934238433838,
      "step": 74701
    },
    {
      "epoch": 6.685256958007812e-07,
      "step": 74701,
      "training_step_time": 8.57351279258728
    },
    {
      "epoch": 6.694793701171875e-07,
      "model_forward_time": 1.8769586086273193,
      "step": 74702
    },
    {
      "epoch": 6.694793701171875e-07,
      "step": 74702,
      "training_step_time": 8.870680809020996
    },
    {
      "epoch": 6.704330444335937e-07,
      "model_forward_time": 2.470127582550049,
      "step": 74703
    },
    {
      "epoch": 6.704330444335937e-07,
      "step": 74703,
      "training_step_time": 7.734859228134155
    },
    {
      "epoch": 6.7138671875e-07,
      "model_forward_time": 1.8918616771697998,
      "step": 74704
    },
    {
      "epoch": 6.7138671875e-07,
      "step": 74704,
      "training_step_time": 8.672724723815918
    },
    {
      "epoch": 6.723403930664062e-07,
      "model_forward_time": 1.8795132637023926,
      "step": 74705
    },
    {
      "epoch": 6.723403930664062e-07,
      "step": 74705,
      "training_step_time": 7.753616809844971
    },
    {
      "epoch": 6.732940673828125e-07,
      "model_forward_time": 1.8879303932189941,
      "step": 74706
    },
    {
      "epoch": 6.732940673828125e-07,
      "step": 74706,
      "training_step_time": 9.141660451889038
    },
    {
      "epoch": 6.742477416992187e-07,
      "model_forward_time": 1.8755276203155518,
      "step": 74707
    },
    {
      "epoch": 6.742477416992187e-07,
      "step": 74707,
      "training_step_time": 7.638214349746704
    },
    {
      "epoch": 6.75201416015625e-07,
      "model_forward_time": 1.8830044269561768,
      "step": 74708
    },
    {
      "epoch": 6.75201416015625e-07,
      "step": 74708,
      "training_step_time": 7.37507963180542
    },
    {
      "epoch": 6.761550903320312e-07,
      "model_forward_time": 1.8911323547363281,
      "step": 74709
    },
    {
      "epoch": 6.761550903320312e-07,
      "step": 74709,
      "training_step_time": 7.408227443695068
    },
    {
      "epoch": 6.771087646484375e-07,
      "grad_norm": 0.09501665085554123,
      "learning_rate": 1.6491484657706591e-06,
      "loss": 0.0549,
      "step": 74710
    },
    {
      "epoch": 6.771087646484375e-07,
      "model_forward_time": 1.8760857582092285,
      "step": 74710
    },
    {
      "dynamics_loss_avg": 0.06226175390183926,
      "epoch": 6.771087646484375e-07,
      "step": 74710
    },
    {
      "action_loss_avg": 0.0018893379252403974,
      "epoch": 6.771087646484375e-07,
      "step": 74710
    },
    {
      "epoch": 6.771087646484375e-07,
      "step": 74710,
      "training_step_time": 7.577813386917114
    },
    {
      "epoch": 6.780624389648437e-07,
      "model_forward_time": 1.867081642150879,
      "step": 74711
    },
    {
      "epoch": 6.780624389648437e-07,
      "step": 74711,
      "training_step_time": 8.204779624938965
    },
    {
      "epoch": 6.7901611328125e-07,
      "model_forward_time": 1.8930106163024902,
      "step": 74712
    },
    {
      "epoch": 6.7901611328125e-07,
      "step": 74712,
      "training_step_time": 7.731133222579956
    },
    {
      "epoch": 6.799697875976562e-07,
      "model_forward_time": 2.4776089191436768,
      "step": 74713
    },
    {
      "epoch": 6.799697875976562e-07,
      "step": 74713,
      "training_step_time": 8.426688194274902
    },
    {
      "epoch": 6.809234619140625e-07,
      "model_forward_time": 1.8821725845336914,
      "step": 74714
    },
    {
      "epoch": 6.809234619140625e-07,
      "step": 74714,
      "training_step_time": 7.5407538414001465
    },
    {
      "epoch": 6.818771362304687e-07,
      "model_forward_time": 1.8778886795043945,
      "step": 74715
    },
    {
      "epoch": 6.818771362304687e-07,
      "step": 74715,
      "training_step_time": 7.589081525802612
    },
    {
      "epoch": 6.82830810546875e-07,
      "model_forward_time": 1.8813621997833252,
      "step": 74716
    },
    {
      "epoch": 6.82830810546875e-07,
      "step": 74716,
      "training_step_time": 8.594035625457764
    },
    {
      "epoch": 6.837844848632812e-07,
      "model_forward_time": 1.8770325183868408,
      "step": 74717
    },
    {
      "epoch": 6.837844848632812e-07,
      "step": 74717,
      "training_step_time": 9.411181449890137
    },
    {
      "epoch": 6.847381591796875e-07,
      "model_forward_time": 1.9098844528198242,
      "step": 74718
    },
    {
      "epoch": 6.847381591796875e-07,
      "step": 74718,
      "training_step_time": 7.27590537071228
    },
    {
      "epoch": 6.856918334960937e-07,
      "model_forward_time": 1.9041924476623535,
      "step": 74719
    },
    {
      "epoch": 6.856918334960937e-07,
      "step": 74719,
      "training_step_time": 7.730173349380493
    },
    {
      "epoch": 6.866455078125e-07,
      "grad_norm": 0.07961604744195938,
      "learning_rate": 1.647921431552883e-06,
      "loss": 0.0552,
      "step": 74720
    },
    {
      "epoch": 6.866455078125e-07,
      "model_forward_time": 1.9072778224945068,
      "step": 74720
    },
    {
      "dynamics_loss_avg": 0.06172051355242729,
      "epoch": 6.866455078125e-07,
      "step": 74720
    },
    {
      "action_loss_avg": 0.002498171851038933,
      "epoch": 6.866455078125e-07,
      "step": 74720
    },
    {
      "epoch": 6.866455078125e-07,
      "step": 74720,
      "training_step_time": 9.295116186141968
    },
    {
      "epoch": 6.875991821289062e-07,
      "model_forward_time": 1.9033684730529785,
      "step": 74721
    },
    {
      "epoch": 6.875991821289062e-07,
      "step": 74721,
      "training_step_time": 8.606656074523926
    },
    {
      "epoch": 6.885528564453125e-07,
      "model_forward_time": 2.4833762645721436,
      "step": 74722
    },
    {
      "epoch": 6.885528564453125e-07,
      "step": 74722,
      "training_step_time": 7.622930526733398
    },
    {
      "epoch": 6.895065307617187e-07,
      "model_forward_time": 1.92097806930542,
      "step": 74723
    },
    {
      "epoch": 6.895065307617187e-07,
      "step": 74723,
      "training_step_time": 7.729268312454224
    },
    {
      "epoch": 6.90460205078125e-07,
      "model_forward_time": 1.90494966506958,
      "step": 74724
    },
    {
      "epoch": 6.90460205078125e-07,
      "step": 74724,
      "training_step_time": 7.74904203414917
    },
    {
      "epoch": 6.914138793945312e-07,
      "model_forward_time": 1.9080009460449219,
      "step": 74725
    },
    {
      "epoch": 6.914138793945312e-07,
      "step": 74725,
      "training_step_time": 7.782463550567627
    },
    {
      "epoch": 6.923675537109375e-07,
      "model_forward_time": 1.904911756515503,
      "step": 74726
    },
    {
      "epoch": 6.923675537109375e-07,
      "step": 74726,
      "training_step_time": 7.767857551574707
    },
    {
      "epoch": 6.933212280273437e-07,
      "model_forward_time": 1.8993184566497803,
      "step": 74727
    },
    {
      "epoch": 6.933212280273437e-07,
      "step": 74727,
      "training_step_time": 8.16967248916626
    },
    {
      "epoch": 6.9427490234375e-07,
      "model_forward_time": 1.9134626388549805,
      "step": 74728
    },
    {
      "epoch": 6.9427490234375e-07,
      "step": 74728,
      "training_step_time": 7.833211183547974
    },
    {
      "epoch": 6.952285766601562e-07,
      "model_forward_time": 1.9108219146728516,
      "step": 74729
    },
    {
      "epoch": 6.952285766601562e-07,
      "step": 74729,
      "training_step_time": 8.888039588928223
    },
    {
      "epoch": 6.961822509765625e-07,
      "grad_norm": 0.06764831393957138,
      "learning_rate": 1.6466947639133813e-06,
      "loss": 0.0536,
      "step": 74730
    },
    {
      "epoch": 6.961822509765625e-07,
      "model_forward_time": 1.8998064994812012,
      "step": 74730
    },
    {
      "dynamics_loss_avg": 0.03527452563866973,
      "epoch": 6.961822509765625e-07,
      "step": 74730
    },
    {
      "action_loss_avg": 0.0009046576742548496,
      "epoch": 6.961822509765625e-07,
      "step": 74730
    },
    {
      "epoch": 6.961822509765625e-07,
      "step": 74730,
      "training_step_time": 7.852823972702026
    },
    {
      "epoch": 6.971359252929687e-07,
      "model_forward_time": 2.4740116596221924,
      "step": 74731
    },
    {
      "epoch": 6.971359252929687e-07,
      "step": 74731,
      "training_step_time": 7.7493062019348145
    },
    {
      "epoch": 6.98089599609375e-07,
      "model_forward_time": 1.9033031463623047,
      "step": 74732
    },
    {
      "epoch": 6.98089599609375e-07,
      "step": 74732,
      "training_step_time": 7.8250463008880615
    },
    {
      "epoch": 6.990432739257812e-07,
      "model_forward_time": 1.9098398685455322,
      "step": 74733
    },
    {
      "epoch": 6.990432739257812e-07,
      "step": 74733,
      "training_step_time": 7.691943883895874
    },
    {
      "epoch": 6.999969482421875e-07,
      "model_forward_time": 1.908747911453247,
      "step": 74734
    },
    {
      "epoch": 6.999969482421875e-07,
      "step": 74734,
      "training_step_time": 8.326279640197754
    },
    {
      "epoch": 7.009506225585937e-07,
      "model_forward_time": 1.9060485363006592,
      "step": 74735
    },
    {
      "epoch": 7.009506225585937e-07,
      "step": 74735,
      "training_step_time": 7.731256484985352
    },
    {
      "epoch": 7.01904296875e-07,
      "model_forward_time": 1.8982839584350586,
      "step": 74736
    },
    {
      "epoch": 7.01904296875e-07,
      "step": 74736,
      "training_step_time": 7.424434423446655
    },
    {
      "epoch": 7.028579711914062e-07,
      "model_forward_time": 1.9048843383789062,
      "step": 74737
    },
    {
      "epoch": 7.028579711914062e-07,
      "step": 74737,
      "training_step_time": 7.734079122543335
    },
    {
      "epoch": 7.038116455078125e-07,
      "model_forward_time": 1.9096298217773438,
      "step": 74738
    },
    {
      "epoch": 7.038116455078125e-07,
      "step": 74738,
      "training_step_time": 8.269649267196655
    },
    {
      "epoch": 7.047653198242187e-07,
      "model_forward_time": 1.8967106342315674,
      "step": 74739
    },
    {
      "epoch": 7.047653198242187e-07,
      "step": 74739,
      "training_step_time": 8.604186296463013
    },
    {
      "epoch": 7.05718994140625e-07,
      "grad_norm": 0.09301933646202087,
      "learning_rate": 1.6454684629862965e-06,
      "loss": 0.0573,
      "step": 74740
    },
    {
      "epoch": 7.05718994140625e-07,
      "model_forward_time": 2.5082952976226807,
      "step": 74740
    },
    {
      "dynamics_loss_avg": 0.05152714010328054,
      "epoch": 7.05718994140625e-07,
      "step": 74740
    },
    {
      "action_loss_avg": 0.0015073633228894323,
      "epoch": 7.05718994140625e-07,
      "step": 74740
    },
    {
      "epoch": 7.05718994140625e-07,
      "step": 74740,
      "training_step_time": 7.829996109008789
    },
    {
      "epoch": 7.066726684570312e-07,
      "model_forward_time": 1.9050705432891846,
      "step": 74741
    },
    {
      "epoch": 7.066726684570312e-07,
      "step": 74741,
      "training_step_time": 7.724292278289795
    },
    {
      "epoch": 7.076263427734375e-07,
      "model_forward_time": 1.899125099182129,
      "step": 74742
    },
    {
      "epoch": 7.076263427734375e-07,
      "step": 74742,
      "training_step_time": 7.687071800231934
    },
    {
      "epoch": 7.085800170898437e-07,
      "model_forward_time": 1.9095227718353271,
      "step": 74743
    },
    {
      "epoch": 7.085800170898437e-07,
      "step": 74743,
      "training_step_time": 7.711141586303711
    },
    {
      "epoch": 7.0953369140625e-07,
      "model_forward_time": 1.9031257629394531,
      "step": 74744
    },
    {
      "epoch": 7.0953369140625e-07,
      "step": 74744,
      "training_step_time": 7.791488170623779
    },
    {
      "epoch": 7.104873657226562e-07,
      "model_forward_time": 1.910144567489624,
      "step": 74745
    },
    {
      "epoch": 7.104873657226562e-07,
      "step": 74745,
      "training_step_time": 8.240705728530884
    },
    {
      "epoch": 7.114410400390625e-07,
      "model_forward_time": 1.9084491729736328,
      "step": 74746
    },
    {
      "epoch": 7.114410400390625e-07,
      "step": 74746,
      "training_step_time": 8.287912368774414
    },
    {
      "epoch": 7.123947143554687e-07,
      "model_forward_time": 1.8987610340118408,
      "step": 74747
    },
    {
      "epoch": 7.123947143554687e-07,
      "step": 74747,
      "training_step_time": 9.797735691070557
    },
    {
      "epoch": 7.13348388671875e-07,
      "model_forward_time": 1.892634630203247,
      "step": 74748
    },
    {
      "epoch": 7.13348388671875e-07,
      "step": 74748,
      "training_step_time": 7.86881160736084
    },
    {
      "epoch": 7.143020629882812e-07,
      "model_forward_time": 2.514219284057617,
      "step": 74749
    },
    {
      "epoch": 7.143020629882812e-07,
      "step": 74749,
      "training_step_time": 7.860793828964233
    },
    {
      "epoch": 7.152557373046875e-07,
      "grad_norm": 0.07606254518032074,
      "learning_rate": 1.644242528905739e-06,
      "loss": 0.058,
      "step": 74750
    },
    {
      "epoch": 7.152557373046875e-07,
      "model_forward_time": 1.8971235752105713,
      "step": 74750
    },
    {
      "dynamics_loss_avg": 0.058728696405887605,
      "epoch": 7.152557373046875e-07,
      "step": 74750
    },
    {
      "action_loss_avg": 0.0015766383381560444,
      "epoch": 7.152557373046875e-07,
      "step": 74750
    },
    {
      "epoch": 7.152557373046875e-07,
      "step": 74750,
      "training_step_time": 7.741297721862793
    },
    {
      "epoch": 7.162094116210938e-07,
      "model_forward_time": 1.9094858169555664,
      "step": 74751
    },
    {
      "epoch": 7.162094116210938e-07,
      "step": 74751,
      "training_step_time": 9.220830202102661
    },
    {
      "epoch": 7.171630859375e-07,
      "model_forward_time": 1.9027397632598877,
      "step": 74752
    },
    {
      "epoch": 7.171630859375e-07,
      "step": 74752,
      "training_step_time": 7.679258584976196
    },
    {
      "epoch": 7.181167602539063e-07,
      "model_forward_time": 1.895613670349121,
      "step": 74753
    },
    {
      "epoch": 7.181167602539063e-07,
      "step": 74753,
      "training_step_time": 7.785752058029175
    },
    {
      "epoch": 7.190704345703125e-07,
      "model_forward_time": 1.8967444896697998,
      "step": 74754
    },
    {
      "epoch": 7.190704345703125e-07,
      "step": 74754,
      "training_step_time": 7.431953191757202
    },
    {
      "epoch": 7.200241088867188e-07,
      "model_forward_time": 1.9058654308319092,
      "step": 74755
    },
    {
      "epoch": 7.200241088867188e-07,
      "step": 74755,
      "training_step_time": 8.394026041030884
    },
    {
      "epoch": 7.20977783203125e-07,
      "model_forward_time": 1.9107611179351807,
      "step": 74756
    },
    {
      "epoch": 7.20977783203125e-07,
      "step": 74756,
      "training_step_time": 7.754375457763672
    },
    {
      "epoch": 7.219314575195313e-07,
      "model_forward_time": 1.897179126739502,
      "step": 74757
    },
    {
      "epoch": 7.219314575195313e-07,
      "step": 74757,
      "training_step_time": 7.8793206214904785
    },
    {
      "epoch": 7.228851318359375e-07,
      "model_forward_time": 2.4879558086395264,
      "step": 74758
    },
    {
      "epoch": 7.228851318359375e-07,
      "step": 74758,
      "training_step_time": 7.892178297042847
    },
    {
      "epoch": 7.238388061523438e-07,
      "model_forward_time": 1.8979451656341553,
      "step": 74759
    },
    {
      "epoch": 7.238388061523438e-07,
      "step": 74759,
      "training_step_time": 7.778828859329224
    },
    {
      "epoch": 7.2479248046875e-07,
      "grad_norm": 0.09450609236955643,
      "learning_rate": 1.6430169618057718e-06,
      "loss": 0.0575,
      "step": 74760
    },
    {
      "epoch": 7.2479248046875e-07,
      "model_forward_time": 1.9022619724273682,
      "step": 74760
    },
    {
      "dynamics_loss_avg": 0.06532001793384552,
      "epoch": 7.2479248046875e-07,
      "step": 74760
    },
    {
      "action_loss_avg": 0.0018710113188717514,
      "epoch": 7.2479248046875e-07,
      "step": 74760
    },
    {
      "epoch": 7.2479248046875e-07,
      "step": 74760,
      "training_step_time": 7.850418567657471
    },
    {
      "epoch": 7.257461547851563e-07,
      "model_forward_time": 1.9130961894989014,
      "step": 74761
    },
    {
      "epoch": 7.257461547851563e-07,
      "step": 74761,
      "training_step_time": 8.572277307510376
    },
    {
      "epoch": 7.266998291015625e-07,
      "model_forward_time": 1.907348394393921,
      "step": 74762
    },
    {
      "epoch": 7.266998291015625e-07,
      "step": 74762,
      "training_step_time": 8.638004779815674
    },
    {
      "epoch": 7.276535034179688e-07,
      "model_forward_time": 1.883117437362671,
      "step": 74763
    },
    {
      "epoch": 7.276535034179688e-07,
      "step": 74763,
      "training_step_time": 7.772918462753296
    },
    {
      "epoch": 7.28607177734375e-07,
      "model_forward_time": 1.9029345512390137,
      "step": 74764
    },
    {
      "epoch": 7.28607177734375e-07,
      "step": 74764,
      "training_step_time": 7.611628532409668
    },
    {
      "epoch": 7.295608520507813e-07,
      "model_forward_time": 1.9010491371154785,
      "step": 74765
    },
    {
      "epoch": 7.295608520507813e-07,
      "step": 74765,
      "training_step_time": 8.726357460021973
    },
    {
      "epoch": 7.305145263671875e-07,
      "model_forward_time": 1.9113221168518066,
      "step": 74766
    },
    {
      "epoch": 7.305145263671875e-07,
      "step": 74766,
      "training_step_time": 9.673177242279053
    },
    {
      "epoch": 7.314682006835938e-07,
      "model_forward_time": 1.9007830619812012,
      "step": 74767
    },
    {
      "epoch": 7.314682006835938e-07,
      "step": 74767,
      "training_step_time": 9.35616946220398
    },
    {
      "epoch": 7.32421875e-07,
      "model_forward_time": 2.519822597503662,
      "step": 74768
    },
    {
      "epoch": 7.32421875e-07,
      "step": 74768,
      "training_step_time": 7.907102823257446
    },
    {
      "epoch": 7.333755493164063e-07,
      "model_forward_time": 1.9061000347137451,
      "step": 74769
    },
    {
      "epoch": 7.333755493164063e-07,
      "step": 74769,
      "training_step_time": 7.844418525695801
    },
    {
      "epoch": 7.343292236328125e-07,
      "grad_norm": 0.07811173051595688,
      "learning_rate": 1.641791761820422e-06,
      "loss": 0.0561,
      "step": 74770
    },
    {
      "epoch": 7.343292236328125e-07,
      "model_forward_time": 1.7662174701690674,
      "step": 74770
    },
    {
      "dynamics_loss_avg": 0.050677837990224364,
      "epoch": 7.343292236328125e-07,
      "step": 74770
    },
    {
      "action_loss_avg": 0.0012980137835256755,
      "epoch": 7.343292236328125e-07,
      "step": 74770
    },
    {
      "epoch": 7.343292236328125e-07,
      "step": 74770,
      "training_step_time": 7.732705593109131
    },
    {
      "epoch": 7.352828979492188e-07,
      "model_forward_time": 1.7495238780975342,
      "step": 74771
    },
    {
      "epoch": 7.352828979492188e-07,
      "step": 74771,
      "training_step_time": 7.808648586273193
    },
    {
      "epoch": 7.36236572265625e-07,
      "model_forward_time": 1.7221200466156006,
      "step": 74772
    },
    {
      "epoch": 7.36236572265625e-07,
      "step": 74772,
      "training_step_time": 8.689324855804443
    },
    {
      "epoch": 7.371902465820313e-07,
      "model_forward_time": 1.7512195110321045,
      "step": 74773
    },
    {
      "epoch": 7.371902465820313e-07,
      "step": 74773,
      "training_step_time": 7.56722354888916
    },
    {
      "epoch": 7.381439208984375e-07,
      "model_forward_time": 1.7238085269927979,
      "step": 74774
    },
    {
      "epoch": 7.381439208984375e-07,
      "step": 74774,
      "training_step_time": 8.362118244171143
    },
    {
      "epoch": 7.390975952148438e-07,
      "model_forward_time": 1.7848830223083496,
      "step": 74775
    },
    {
      "epoch": 7.390975952148438e-07,
      "step": 74775,
      "training_step_time": 8.716755628585815
    },
    {
      "epoch": 7.4005126953125e-07,
      "model_forward_time": 1.737919569015503,
      "step": 74776
    },
    {
      "epoch": 7.4005126953125e-07,
      "step": 74776,
      "training_step_time": 7.80603289604187
    },
    {
      "epoch": 7.410049438476563e-07,
      "model_forward_time": 1.9602289199829102,
      "step": 74777
    },
    {
      "epoch": 7.410049438476563e-07,
      "step": 74777,
      "training_step_time": 9.293486833572388
    },
    {
      "epoch": 7.419586181640625e-07,
      "model_forward_time": 1.7353520393371582,
      "step": 74778
    },
    {
      "epoch": 7.419586181640625e-07,
      "step": 74778,
      "training_step_time": 8.323203802108765
    },
    {
      "epoch": 7.429122924804688e-07,
      "model_forward_time": 1.7293074131011963,
      "step": 74779
    },
    {
      "epoch": 7.429122924804688e-07,
      "step": 74779,
      "training_step_time": 7.686190366744995
    },
    {
      "epoch": 7.43865966796875e-07,
      "grad_norm": 0.10933371633291245,
      "learning_rate": 1.6405669290836763e-06,
      "loss": 0.0572,
      "step": 74780
    },
    {
      "epoch": 7.43865966796875e-07,
      "model_forward_time": 1.743044137954712,
      "step": 74780
    },
    {
      "dynamics_loss_avg": 0.06365194041281938,
      "epoch": 7.43865966796875e-07,
      "step": 74780
    },
    {
      "action_loss_avg": 0.0009454417682718486,
      "epoch": 7.43865966796875e-07,
      "step": 74780
    },
    {
      "epoch": 7.43865966796875e-07,
      "step": 74780,
      "training_step_time": 8.586726427078247
    },
    {
      "epoch": 7.448196411132813e-07,
      "model_forward_time": 1.7242021560668945,
      "step": 74781
    },
    {
      "epoch": 7.448196411132813e-07,
      "step": 74781,
      "training_step_time": 7.705749750137329
    },
    {
      "epoch": 7.457733154296875e-07,
      "model_forward_time": 1.7353646755218506,
      "step": 74782
    },
    {
      "epoch": 7.457733154296875e-07,
      "step": 74782,
      "training_step_time": 7.436413764953613
    },
    {
      "epoch": 7.467269897460938e-07,
      "model_forward_time": 1.726247787475586,
      "step": 74783
    },
    {
      "epoch": 7.467269897460938e-07,
      "step": 74783,
      "training_step_time": 11.954500198364258
    },
    {
      "epoch": 7.476806640625e-07,
      "model_forward_time": 1.725839614868164,
      "step": 74784
    },
    {
      "epoch": 7.476806640625e-07,
      "step": 74784,
      "training_step_time": 6.8116374015808105
    },
    {
      "epoch": 7.486343383789063e-07,
      "model_forward_time": 1.7275910377502441,
      "step": 74785
    },
    {
      "epoch": 7.486343383789063e-07,
      "step": 74785,
      "training_step_time": 8.237638711929321
    },
    {
      "epoch": 7.495880126953125e-07,
      "model_forward_time": 1.7455272674560547,
      "step": 74786
    },
    {
      "epoch": 7.495880126953125e-07,
      "step": 74786,
      "training_step_time": 7.606448173522949
    },
    {
      "epoch": 7.505416870117188e-07,
      "model_forward_time": 1.7267019748687744,
      "step": 74787
    },
    {
      "epoch": 7.505416870117188e-07,
      "step": 74787,
      "training_step_time": 8.42690110206604
    },
    {
      "epoch": 7.51495361328125e-07,
      "model_forward_time": 1.7311155796051025,
      "step": 74788
    },
    {
      "epoch": 7.51495361328125e-07,
      "step": 74788,
      "training_step_time": 6.797645330429077
    },
    {
      "epoch": 7.524490356445313e-07,
      "model_forward_time": 1.7488443851470947,
      "step": 74789
    },
    {
      "epoch": 7.524490356445313e-07,
      "step": 74789,
      "training_step_time": 6.811314582824707
    },
    {
      "epoch": 7.534027099609375e-07,
      "grad_norm": 0.08603759109973907,
      "learning_rate": 1.6393424637294825e-06,
      "loss": 0.0566,
      "step": 74790
    },
    {
      "epoch": 7.534027099609375e-07,
      "model_forward_time": 1.7291219234466553,
      "step": 74790
    },
    {
      "dynamics_loss_avg": 0.06289950842037798,
      "epoch": 7.534027099609375e-07,
      "step": 74790
    },
    {
      "action_loss_avg": 0.00166998888598755,
      "epoch": 7.534027099609375e-07,
      "step": 74790
    },
    {
      "epoch": 7.534027099609375e-07,
      "step": 74790,
      "training_step_time": 6.805810213088989
    },
    {
      "epoch": 7.543563842773438e-07,
      "model_forward_time": 1.7311735153198242,
      "step": 74791
    },
    {
      "epoch": 7.543563842773438e-07,
      "step": 74791,
      "training_step_time": 6.765746593475342
    },
    {
      "epoch": 7.5531005859375e-07,
      "model_forward_time": 1.7293152809143066,
      "step": 74792
    },
    {
      "epoch": 7.5531005859375e-07,
      "step": 74792,
      "training_step_time": 6.708024263381958
    },
    {
      "epoch": 7.562637329101563e-07,
      "model_forward_time": 2.3350512981414795,
      "step": 74793
    },
    {
      "epoch": 7.562637329101563e-07,
      "step": 74793,
      "training_step_time": 7.675555467605591
    },
    {
      "epoch": 7.572174072265625e-07,
      "model_forward_time": 1.7332704067230225,
      "step": 74794
    },
    {
      "epoch": 7.572174072265625e-07,
      "step": 74794,
      "training_step_time": 7.583463430404663
    },
    {
      "epoch": 7.581710815429688e-07,
      "model_forward_time": 1.7348875999450684,
      "step": 74795
    },
    {
      "epoch": 7.581710815429688e-07,
      "step": 74795,
      "training_step_time": 6.700251340866089
    },
    {
      "epoch": 7.59124755859375e-07,
      "model_forward_time": 1.7357757091522217,
      "step": 74796
    },
    {
      "epoch": 7.59124755859375e-07,
      "step": 74796,
      "training_step_time": 7.61012077331543
    },
    {
      "epoch": 7.600784301757813e-07,
      "model_forward_time": 1.7290332317352295,
      "step": 74797
    },
    {
      "epoch": 7.600784301757813e-07,
      "step": 74797,
      "training_step_time": 7.609053134918213
    },
    {
      "epoch": 7.610321044921875e-07,
      "model_forward_time": 1.7281954288482666,
      "step": 74798
    },
    {
      "epoch": 7.610321044921875e-07,
      "step": 74798,
      "training_step_time": 6.74283766746521
    },
    {
      "epoch": 7.619857788085938e-07,
      "model_forward_time": 1.734773874282837,
      "step": 74799
    },
    {
      "epoch": 7.619857788085938e-07,
      "step": 74799,
      "training_step_time": 6.680995225906372
    },
    {
      "epoch": 7.62939453125e-07,
      "grad_norm": 0.17149944603443146,
      "learning_rate": 1.6381183658917422e-06,
      "loss": 0.0551,
      "step": 74800
    },
    {
      "epoch": 7.62939453125e-07,
      "model_forward_time": 1.758265495300293,
      "step": 74800
    },
    {
      "dynamics_loss_avg": 0.048105405643582344,
      "epoch": 7.62939453125e-07,
      "step": 74800
    },
    {
      "action_loss_avg": 0.001305501681054011,
      "epoch": 7.62939453125e-07,
      "step": 74800
    },
    {
      "epoch": 7.62939453125e-07,
      "step": 74800,
      "training_step_time": 6.728962421417236
    },
    {
      "epoch": 7.638931274414063e-07,
      "model_forward_time": 1.7384302616119385,
      "step": 74801
    },
    {
      "epoch": 7.638931274414063e-07,
      "step": 74801,
      "training_step_time": 6.705377578735352
    },
    {
      "epoch": 7.648468017578125e-07,
      "model_forward_time": 2.3468902111053467,
      "step": 74802
    },
    {
      "epoch": 7.648468017578125e-07,
      "step": 74802,
      "training_step_time": 8.427665948867798
    },
    {
      "epoch": 7.658004760742188e-07,
      "model_forward_time": 1.7450003623962402,
      "step": 74803
    },
    {
      "epoch": 7.658004760742188e-07,
      "step": 74803,
      "training_step_time": 7.51847243309021
    },
    {
      "epoch": 7.66754150390625e-07,
      "model_forward_time": 1.7462575435638428,
      "step": 74804
    },
    {
      "epoch": 7.66754150390625e-07,
      "step": 74804,
      "training_step_time": 7.079921245574951
    },
    {
      "epoch": 7.677078247070313e-07,
      "model_forward_time": 1.748387098312378,
      "step": 74805
    },
    {
      "epoch": 7.677078247070313e-07,
      "step": 74805,
      "training_step_time": 8.058602809906006
    },
    {
      "epoch": 7.686614990234375e-07,
      "model_forward_time": 1.739959955215454,
      "step": 74806
    },
    {
      "epoch": 7.686614990234375e-07,
      "step": 74806,
      "training_step_time": 7.606008768081665
    },
    {
      "epoch": 7.696151733398438e-07,
      "model_forward_time": 1.7430758476257324,
      "step": 74807
    },
    {
      "epoch": 7.696151733398438e-07,
      "step": 74807,
      "training_step_time": 6.778570890426636
    },
    {
      "epoch": 7.7056884765625e-07,
      "model_forward_time": 1.380497932434082,
      "step": 74808
    },
    {
      "epoch": 7.7056884765625e-07,
      "step": 74808,
      "training_step_time": 6.701698303222656
    },
    {
      "epoch": 7.715225219726563e-07,
      "model_forward_time": 1.7459940910339355,
      "step": 74809
    },
    {
      "epoch": 7.715225219726563e-07,
      "step": 74809,
      "training_step_time": 6.7185187339782715
    },
    {
      "epoch": 7.724761962890626e-07,
      "grad_norm": 0.10379960387945175,
      "learning_rate": 1.636894635704323e-06,
      "loss": 0.0546,
      "step": 74810
    },
    {
      "epoch": 7.724761962890626e-07,
      "model_forward_time": 1.748657464981079,
      "step": 74810
    },
    {
      "dynamics_loss_avg": 0.05296435821801424,
      "epoch": 7.724761962890626e-07,
      "step": 74810
    },
    {
      "action_loss_avg": 0.001313028429285623,
      "epoch": 7.724761962890626e-07,
      "step": 74810
    },
    {
      "epoch": 7.724761962890626e-07,
      "step": 74810,
      "training_step_time": 6.747274875640869
    },
    {
      "epoch": 7.734298706054688e-07,
      "model_forward_time": 2.35076904296875,
      "step": 74811
    },
    {
      "epoch": 7.734298706054688e-07,
      "step": 74811,
      "training_step_time": 9.414779424667358
    },
    {
      "epoch": 7.74383544921875e-07,
      "model_forward_time": 1.743490219116211,
      "step": 74812
    },
    {
      "epoch": 7.74383544921875e-07,
      "step": 74812,
      "training_step_time": 7.565500736236572
    },
    {
      "epoch": 7.753372192382812e-07,
      "model_forward_time": 1.753638505935669,
      "step": 74813
    },
    {
      "epoch": 7.753372192382812e-07,
      "step": 74813,
      "training_step_time": 7.565686225891113
    },
    {
      "epoch": 7.762908935546874e-07,
      "model_forward_time": 1.7458651065826416,
      "step": 74814
    },
    {
      "epoch": 7.762908935546874e-07,
      "step": 74814,
      "training_step_time": 7.342894792556763
    },
    {
      "epoch": 7.772445678710937e-07,
      "model_forward_time": 1.7391445636749268,
      "step": 74815
    },
    {
      "epoch": 7.772445678710937e-07,
      "step": 74815,
      "training_step_time": 7.58910346031189
    },
    {
      "epoch": 7.781982421875e-07,
      "model_forward_time": 1.7483365535736084,
      "step": 74816
    },
    {
      "epoch": 7.781982421875e-07,
      "step": 74816,
      "training_step_time": 6.72220778465271
    },
    {
      "epoch": 7.791519165039062e-07,
      "model_forward_time": 1.7459123134613037,
      "step": 74817
    },
    {
      "epoch": 7.791519165039062e-07,
      "step": 74817,
      "training_step_time": 6.7582643032073975
    },
    {
      "epoch": 7.801055908203125e-07,
      "model_forward_time": 1.7476353645324707,
      "step": 74818
    },
    {
      "epoch": 7.801055908203125e-07,
      "step": 74818,
      "training_step_time": 6.747278928756714
    },
    {
      "epoch": 7.810592651367187e-07,
      "model_forward_time": 1.742830753326416,
      "step": 74819
    },
    {
      "epoch": 7.810592651367187e-07,
      "step": 74819,
      "training_step_time": 6.759099721908569
    },
    {
      "epoch": 7.82012939453125e-07,
      "grad_norm": 0.09031300246715546,
      "learning_rate": 1.6356712733010515e-06,
      "loss": 0.0556,
      "step": 74820
    },
    {
      "epoch": 7.82012939453125e-07,
      "model_forward_time": 2.333235025405884,
      "step": 74820
    },
    {
      "dynamics_loss_avg": 0.0442313689738512,
      "epoch": 7.82012939453125e-07,
      "step": 74820
    },
    {
      "action_loss_avg": 0.0010400334576843306,
      "epoch": 7.82012939453125e-07,
      "step": 74820
    },
    {
      "epoch": 7.82012939453125e-07,
      "step": 74820,
      "training_step_time": 11.615668773651123
    },
    {
      "epoch": 7.829666137695312e-07,
      "model_forward_time": 1.745466709136963,
      "step": 74821
    },
    {
      "epoch": 7.829666137695312e-07,
      "step": 74821,
      "training_step_time": 7.589585304260254
    },
    {
      "epoch": 7.839202880859375e-07,
      "model_forward_time": 1.3740239143371582,
      "step": 74822
    },
    {
      "epoch": 7.839202880859375e-07,
      "step": 74822,
      "training_step_time": 8.104948282241821
    },
    {
      "epoch": 7.848739624023437e-07,
      "model_forward_time": 1.7452099323272705,
      "step": 74823
    },
    {
      "epoch": 7.848739624023437e-07,
      "step": 74823,
      "training_step_time": 7.18690824508667
    },
    {
      "epoch": 7.8582763671875e-07,
      "model_forward_time": 1.7532391548156738,
      "step": 74824
    },
    {
      "epoch": 7.8582763671875e-07,
      "step": 74824,
      "training_step_time": 7.6285810470581055
    },
    {
      "epoch": 7.867813110351562e-07,
      "model_forward_time": 1.77060866355896,
      "step": 74825
    },
    {
      "epoch": 7.867813110351562e-07,
      "step": 74825,
      "training_step_time": 6.751272678375244
    },
    {
      "epoch": 7.877349853515625e-07,
      "model_forward_time": 1.7542541027069092,
      "step": 74826
    },
    {
      "epoch": 7.877349853515625e-07,
      "step": 74826,
      "training_step_time": 6.729177236557007
    },
    {
      "epoch": 7.886886596679687e-07,
      "model_forward_time": 1.7400882244110107,
      "step": 74827
    },
    {
      "epoch": 7.886886596679687e-07,
      "step": 74827,
      "training_step_time": 6.76140832901001
    },
    {
      "epoch": 7.89642333984375e-07,
      "model_forward_time": 1.369706630706787,
      "step": 74828
    },
    {
      "epoch": 7.89642333984375e-07,
      "step": 74828,
      "training_step_time": 6.749921798706055
    },
    {
      "epoch": 7.905960083007812e-07,
      "model_forward_time": 1.7977068424224854,
      "step": 74829
    },
    {
      "epoch": 7.905960083007812e-07,
      "step": 74829,
      "training_step_time": 10.411157131195068
    },
    {
      "epoch": 7.915496826171875e-07,
      "grad_norm": 0.09754105657339096,
      "learning_rate": 1.6344482788157096e-06,
      "loss": 0.0543,
      "step": 74830
    },
    {
      "epoch": 7.915496826171875e-07,
      "model_forward_time": 2.38154935836792,
      "step": 74830
    },
    {
      "dynamics_loss_avg": 0.05653099454939366,
      "epoch": 7.915496826171875e-07,
      "step": 74830
    },
    {
      "action_loss_avg": 0.001270056638168171,
      "epoch": 7.915496826171875e-07,
      "step": 74830
    },
    {
      "epoch": 7.915496826171875e-07,
      "step": 74830,
      "training_step_time": 7.76722526550293
    },
    {
      "epoch": 7.925033569335937e-07,
      "model_forward_time": 1.7204833030700684,
      "step": 74831
    },
    {
      "epoch": 7.925033569335937e-07,
      "step": 74831,
      "training_step_time": 7.440450429916382
    },
    {
      "epoch": 7.9345703125e-07,
      "model_forward_time": 1.733884334564209,
      "step": 74832
    },
    {
      "epoch": 7.9345703125e-07,
      "step": 74832,
      "training_step_time": 7.6643452644348145
    },
    {
      "epoch": 7.944107055664062e-07,
      "model_forward_time": 1.7342753410339355,
      "step": 74833
    },
    {
      "epoch": 7.944107055664062e-07,
      "step": 74833,
      "training_step_time": 7.662606716156006
    },
    {
      "epoch": 7.953643798828125e-07,
      "model_forward_time": 1.7351911067962646,
      "step": 74834
    },
    {
      "epoch": 7.953643798828125e-07,
      "step": 74834,
      "training_step_time": 7.675877809524536
    },
    {
      "epoch": 7.963180541992187e-07,
      "model_forward_time": 1.7393858432769775,
      "step": 74835
    },
    {
      "epoch": 7.963180541992187e-07,
      "step": 74835,
      "training_step_time": 6.748417854309082
    },
    {
      "epoch": 7.97271728515625e-07,
      "model_forward_time": 1.7244682312011719,
      "step": 74836
    },
    {
      "epoch": 7.97271728515625e-07,
      "step": 74836,
      "training_step_time": 6.762484073638916
    },
    {
      "epoch": 7.982254028320312e-07,
      "model_forward_time": 1.7313921451568604,
      "step": 74837
    },
    {
      "epoch": 7.982254028320312e-07,
      "step": 74837,
      "training_step_time": 6.734803199768066
    },
    {
      "epoch": 7.991790771484375e-07,
      "model_forward_time": 1.807239055633545,
      "step": 74838
    },
    {
      "epoch": 7.991790771484375e-07,
      "step": 74838,
      "training_step_time": 8.673209190368652
    },
    {
      "epoch": 8.001327514648437e-07,
      "model_forward_time": 2.370338201522827,
      "step": 74839
    },
    {
      "epoch": 8.001327514648437e-07,
      "step": 74839,
      "training_step_time": 8.727469682693481
    },
    {
      "epoch": 8.0108642578125e-07,
      "grad_norm": 0.08840437233448029,
      "learning_rate": 1.6332256523820444e-06,
      "loss": 0.053,
      "step": 74840
    },
    {
      "epoch": 8.0108642578125e-07,
      "model_forward_time": 1.7449393272399902,
      "step": 74840
    },
    {
      "dynamics_loss_avg": 0.06853796523064375,
      "epoch": 8.0108642578125e-07,
      "step": 74840
    },
    {
      "action_loss_avg": 0.001625244179740548,
      "epoch": 8.0108642578125e-07,
      "step": 74840
    },
    {
      "epoch": 8.0108642578125e-07,
      "step": 74840,
      "training_step_time": 7.59470272064209
    },
    {
      "epoch": 8.020401000976562e-07,
      "model_forward_time": 1.3766148090362549,
      "step": 74841
    },
    {
      "epoch": 8.020401000976562e-07,
      "step": 74841,
      "training_step_time": 7.075690746307373
    },
    {
      "epoch": 8.029937744140625e-07,
      "model_forward_time": 1.7327499389648438,
      "step": 74842
    },
    {
      "epoch": 8.029937744140625e-07,
      "step": 74842,
      "training_step_time": 7.654015302658081
    },
    {
      "epoch": 8.039474487304687e-07,
      "model_forward_time": 1.735569715499878,
      "step": 74843
    },
    {
      "epoch": 8.039474487304687e-07,
      "step": 74843,
      "training_step_time": 7.596690893173218
    },
    {
      "epoch": 8.04901123046875e-07,
      "model_forward_time": 1.7304368019104004,
      "step": 74844
    },
    {
      "epoch": 8.04901123046875e-07,
      "step": 74844,
      "training_step_time": 6.7556562423706055
    },
    {
      "epoch": 8.058547973632812e-07,
      "model_forward_time": 1.7340974807739258,
      "step": 74845
    },
    {
      "epoch": 8.058547973632812e-07,
      "step": 74845,
      "training_step_time": 6.728445529937744
    },
    {
      "epoch": 8.068084716796875e-07,
      "model_forward_time": 1.7387070655822754,
      "step": 74846
    },
    {
      "epoch": 8.068084716796875e-07,
      "step": 74846,
      "training_step_time": 6.742655992507935
    },
    {
      "epoch": 8.077621459960937e-07,
      "model_forward_time": 1.7564442157745361,
      "step": 74847
    },
    {
      "epoch": 8.077621459960937e-07,
      "step": 74847,
      "training_step_time": 9.15511178970337
    },
    {
      "epoch": 8.087158203125e-07,
      "model_forward_time": 2.3929550647735596,
      "step": 74848
    },
    {
      "epoch": 8.087158203125e-07,
      "step": 74848,
      "training_step_time": 8.651514768600464
    },
    {
      "epoch": 8.096694946289062e-07,
      "model_forward_time": 1.7338008880615234,
      "step": 74849
    },
    {
      "epoch": 8.096694946289062e-07,
      "step": 74849,
      "training_step_time": 7.56573486328125
    },
    {
      "epoch": 8.106231689453125e-07,
      "grad_norm": 0.2225760817527771,
      "learning_rate": 1.6320033941337599e-06,
      "loss": 0.053,
      "step": 74850
    },
    {
      "epoch": 8.106231689453125e-07,
      "model_forward_time": 1.7308411598205566,
      "step": 74850
    },
    {
      "dynamics_loss_avg": 0.05141722299158573,
      "epoch": 8.106231689453125e-07,
      "step": 74850
    },
    {
      "action_loss_avg": 0.0015207149554044008,
      "epoch": 8.106231689453125e-07,
      "step": 74850
    },
    {
      "epoch": 8.106231689453125e-07,
      "step": 74850,
      "training_step_time": 7.447504758834839
    },
    {
      "epoch": 8.115768432617187e-07,
      "model_forward_time": 1.7329084873199463,
      "step": 74851
    },
    {
      "epoch": 8.115768432617187e-07,
      "step": 74851,
      "training_step_time": 7.755608081817627
    },
    {
      "epoch": 8.12530517578125e-07,
      "model_forward_time": 1.7384068965911865,
      "step": 74852
    },
    {
      "epoch": 8.12530517578125e-07,
      "step": 74852,
      "training_step_time": 7.636606216430664
    },
    {
      "epoch": 8.134841918945312e-07,
      "model_forward_time": 1.7286081314086914,
      "step": 74853
    },
    {
      "epoch": 8.134841918945312e-07,
      "step": 74853,
      "training_step_time": 6.826473236083984
    },
    {
      "epoch": 8.144378662109375e-07,
      "model_forward_time": 1.7394862174987793,
      "step": 74854
    },
    {
      "epoch": 8.144378662109375e-07,
      "step": 74854,
      "training_step_time": 6.733474254608154
    },
    {
      "epoch": 8.153915405273437e-07,
      "model_forward_time": 1.7385187149047852,
      "step": 74855
    },
    {
      "epoch": 8.153915405273437e-07,
      "step": 74855,
      "training_step_time": 6.789657831192017
    },
    {
      "epoch": 8.1634521484375e-07,
      "model_forward_time": 1.7415046691894531,
      "step": 74856
    },
    {
      "epoch": 8.1634521484375e-07,
      "step": 74856,
      "training_step_time": 9.821897268295288
    },
    {
      "epoch": 8.172988891601562e-07,
      "model_forward_time": 2.378755807876587,
      "step": 74857
    },
    {
      "epoch": 8.172988891601562e-07,
      "step": 74857,
      "training_step_time": 8.62861180305481
    },
    {
      "epoch": 8.182525634765625e-07,
      "model_forward_time": 1.7281882762908936,
      "step": 74858
    },
    {
      "epoch": 8.182525634765625e-07,
      "step": 74858,
      "training_step_time": 7.634908199310303
    },
    {
      "epoch": 8.192062377929687e-07,
      "model_forward_time": 1.7341747283935547,
      "step": 74859
    },
    {
      "epoch": 8.192062377929687e-07,
      "step": 74859,
      "training_step_time": 7.760869264602661
    },
    {
      "epoch": 8.20159912109375e-07,
      "grad_norm": 0.10072387009859085,
      "learning_rate": 1.6307815042045216e-06,
      "loss": 0.0525,
      "step": 74860
    },
    {
      "epoch": 8.20159912109375e-07,
      "model_forward_time": 1.7272510528564453,
      "step": 74860
    },
    {
      "dynamics_loss_avg": 0.04870504662394524,
      "epoch": 8.20159912109375e-07,
      "step": 74860
    },
    {
      "action_loss_avg": 0.0009561825514538214,
      "epoch": 8.20159912109375e-07,
      "step": 74860
    },
    {
      "epoch": 8.20159912109375e-07,
      "step": 74860,
      "training_step_time": 8.264117002487183
    },
    {
      "epoch": 8.211135864257812e-07,
      "model_forward_time": 1.37611985206604,
      "step": 74861
    },
    {
      "epoch": 8.211135864257812e-07,
      "step": 74861,
      "training_step_time": 7.677606105804443
    },
    {
      "epoch": 8.220672607421875e-07,
      "model_forward_time": 1.7372479438781738,
      "step": 74862
    },
    {
      "epoch": 8.220672607421875e-07,
      "step": 74862,
      "training_step_time": 6.853761911392212
    },
    {
      "epoch": 8.230209350585937e-07,
      "model_forward_time": 1.7457191944122314,
      "step": 74863
    },
    {
      "epoch": 8.230209350585937e-07,
      "step": 74863,
      "training_step_time": 6.864365100860596
    },
    {
      "epoch": 8.23974609375e-07,
      "model_forward_time": 1.744694709777832,
      "step": 74864
    },
    {
      "epoch": 8.23974609375e-07,
      "step": 74864,
      "training_step_time": 6.8210976123809814
    },
    {
      "epoch": 8.249282836914062e-07,
      "model_forward_time": 1.7379286289215088,
      "step": 74865
    },
    {
      "epoch": 8.249282836914062e-07,
      "step": 74865,
      "training_step_time": 8.291075468063354
    },
    {
      "epoch": 8.258819580078125e-07,
      "model_forward_time": 1.7965819835662842,
      "step": 74866
    },
    {
      "epoch": 8.258819580078125e-07,
      "step": 74866,
      "training_step_time": 10.242923259735107
    },
    {
      "epoch": 8.268356323242187e-07,
      "model_forward_time": 1.6389734745025635,
      "step": 74867
    },
    {
      "epoch": 8.268356323242187e-07,
      "step": 74867,
      "training_step_time": 7.791528224945068
    },
    {
      "epoch": 8.27789306640625e-07,
      "model_forward_time": 1.7428956031799316,
      "step": 74868
    },
    {
      "epoch": 8.27789306640625e-07,
      "step": 74868,
      "training_step_time": 7.783235549926758
    },
    {
      "epoch": 8.287429809570312e-07,
      "model_forward_time": 1.7412314414978027,
      "step": 74869
    },
    {
      "epoch": 8.287429809570312e-07,
      "step": 74869,
      "training_step_time": 7.774263143539429
    },
    {
      "epoch": 8.296966552734375e-07,
      "grad_norm": 0.12851651012897491,
      "learning_rate": 1.6295599827279513e-06,
      "loss": 0.0519,
      "step": 74870
    },
    {
      "epoch": 8.296966552734375e-07,
      "model_forward_time": 1.740255355834961,
      "step": 74870
    },
    {
      "dynamics_loss_avg": 0.04766724845394492,
      "epoch": 8.296966552734375e-07,
      "step": 74870
    },
    {
      "action_loss_avg": 0.0015726770710898564,
      "epoch": 8.296966552734375e-07,
      "step": 74870
    },
    {
      "epoch": 8.296966552734375e-07,
      "step": 74870,
      "training_step_time": 7.641690254211426
    },
    {
      "epoch": 8.306503295898437e-07,
      "model_forward_time": 1.7472257614135742,
      "step": 74871
    },
    {
      "epoch": 8.306503295898437e-07,
      "step": 74871,
      "training_step_time": 7.7109925746917725
    },
    {
      "epoch": 8.3160400390625e-07,
      "model_forward_time": 1.7391936779022217,
      "step": 74872
    },
    {
      "epoch": 8.3160400390625e-07,
      "step": 74872,
      "training_step_time": 6.809882879257202
    },
    {
      "epoch": 8.325576782226562e-07,
      "model_forward_time": 1.7409656047821045,
      "step": 74873
    },
    {
      "epoch": 8.325576782226562e-07,
      "step": 74873,
      "training_step_time": 6.828749418258667
    },
    {
      "epoch": 8.335113525390625e-07,
      "model_forward_time": 1.3751649856567383,
      "step": 74874
    },
    {
      "epoch": 8.335113525390625e-07,
      "step": 74874,
      "training_step_time": 9.239542961120605
    },
    {
      "epoch": 8.344650268554688e-07,
      "model_forward_time": 1.7997844219207764,
      "step": 74875
    },
    {
      "epoch": 8.344650268554688e-07,
      "step": 74875,
      "training_step_time": 9.05318307876587
    },
    {
      "epoch": 8.35418701171875e-07,
      "model_forward_time": 2.364150285720825,
      "step": 74876
    },
    {
      "epoch": 8.35418701171875e-07,
      "step": 74876,
      "training_step_time": 7.756943225860596
    },
    {
      "epoch": 8.363723754882813e-07,
      "model_forward_time": 1.748661756515503,
      "step": 74877
    },
    {
      "epoch": 8.363723754882813e-07,
      "step": 74877,
      "training_step_time": 7.64174222946167
    },
    {
      "epoch": 8.373260498046875e-07,
      "model_forward_time": 1.73502779006958,
      "step": 74878
    },
    {
      "epoch": 8.373260498046875e-07,
      "step": 74878,
      "training_step_time": 7.679912328720093
    },
    {
      "epoch": 8.382797241210938e-07,
      "model_forward_time": 1.7495813369750977,
      "step": 74879
    },
    {
      "epoch": 8.382797241210938e-07,
      "step": 74879,
      "training_step_time": 7.658513069152832
    },
    {
      "epoch": 8.392333984375e-07,
      "grad_norm": 0.10503847897052765,
      "learning_rate": 1.6283388298376356e-06,
      "loss": 0.0533,
      "step": 74880
    },
    {
      "epoch": 8.392333984375e-07,
      "model_forward_time": 1.7482624053955078,
      "step": 74880
    },
    {
      "dynamics_loss_avg": 0.05464425263926387,
      "epoch": 8.392333984375e-07,
      "step": 74880
    },
    {
      "action_loss_avg": 0.001078847273311112,
      "epoch": 8.392333984375e-07,
      "step": 74880
    },
    {
      "epoch": 8.392333984375e-07,
      "step": 74880,
      "training_step_time": 7.670036792755127
    },
    {
      "epoch": 8.401870727539063e-07,
      "model_forward_time": 1.7435510158538818,
      "step": 74881
    },
    {
      "epoch": 8.401870727539063e-07,
      "step": 74881,
      "training_step_time": 6.788772106170654
    },
    {
      "epoch": 8.411407470703125e-07,
      "model_forward_time": 1.7514591217041016,
      "step": 74882
    },
    {
      "epoch": 8.411407470703125e-07,
      "step": 74882,
      "training_step_time": 6.865292549133301
    },
    {
      "epoch": 8.420944213867188e-07,
      "model_forward_time": 1.744020700454712,
      "step": 74883
    },
    {
      "epoch": 8.420944213867188e-07,
      "step": 74883,
      "training_step_time": 8.575769186019897
    },
    {
      "epoch": 8.43048095703125e-07,
      "model_forward_time": 1.8039829730987549,
      "step": 74884
    },
    {
      "epoch": 8.43048095703125e-07,
      "step": 74884,
      "training_step_time": 8.847169637680054
    },
    {
      "epoch": 8.440017700195313e-07,
      "model_forward_time": 1.7443060874938965,
      "step": 74885
    },
    {
      "epoch": 8.440017700195313e-07,
      "step": 74885,
      "training_step_time": 9.87843632698059
    },
    {
      "epoch": 8.449554443359375e-07,
      "model_forward_time": 2.3687212467193604,
      "step": 74886
    },
    {
      "epoch": 8.449554443359375e-07,
      "step": 74886,
      "training_step_time": 7.568547248840332
    },
    {
      "epoch": 8.459091186523438e-07,
      "model_forward_time": 1.7636773586273193,
      "step": 74887
    },
    {
      "epoch": 8.459091186523438e-07,
      "step": 74887,
      "training_step_time": 7.175881385803223
    },
    {
      "epoch": 8.4686279296875e-07,
      "model_forward_time": 1.7429003715515137,
      "step": 74888
    },
    {
      "epoch": 8.4686279296875e-07,
      "step": 74888,
      "training_step_time": 9.209512710571289
    },
    {
      "epoch": 8.478164672851563e-07,
      "model_forward_time": 1.7474033832550049,
      "step": 74889
    },
    {
      "epoch": 8.478164672851563e-07,
      "step": 74889,
      "training_step_time": 7.625918865203857
    },
    {
      "epoch": 8.487701416015625e-07,
      "grad_norm": 0.12027500569820404,
      "learning_rate": 1.6271180456671125e-06,
      "loss": 0.0527,
      "step": 74890
    },
    {
      "epoch": 8.487701416015625e-07,
      "model_forward_time": 1.0440189838409424,
      "step": 74890
    },
    {
      "dynamics_loss_avg": 0.04959824027027935,
      "epoch": 8.487701416015625e-07,
      "step": 74890
    },
    {
      "action_loss_avg": 0.0018264165584696457,
      "epoch": 8.487701416015625e-07,
      "step": 74890
    },
    {
      "epoch": 8.487701416015625e-07,
      "step": 74890,
      "training_step_time": 6.780028581619263
    },
    {
      "epoch": 8.497238159179688e-07,
      "model_forward_time": 1.7445716857910156,
      "step": 74891
    },
    {
      "epoch": 8.497238159179688e-07,
      "step": 74891,
      "training_step_time": 6.7642662525177
    },
    {
      "epoch": 8.50677490234375e-07,
      "model_forward_time": 1.7406671047210693,
      "step": 74892
    },
    {
      "epoch": 8.50677490234375e-07,
      "step": 74892,
      "training_step_time": 9.424649953842163
    },
    {
      "epoch": 8.516311645507813e-07,
      "model_forward_time": 1.7821638584136963,
      "step": 74893
    },
    {
      "epoch": 8.516311645507813e-07,
      "step": 74893,
      "training_step_time": 9.487833738327026
    },
    {
      "epoch": 8.525848388671875e-07,
      "model_forward_time": 1.7518949508666992,
      "step": 74894
    },
    {
      "epoch": 8.525848388671875e-07,
      "step": 74894,
      "training_step_time": 8.901905536651611
    },
    {
      "epoch": 8.535385131835938e-07,
      "model_forward_time": 2.397618293762207,
      "step": 74895
    },
    {
      "epoch": 8.535385131835938e-07,
      "step": 74895,
      "training_step_time": 7.767330646514893
    },
    {
      "epoch": 8.544921875e-07,
      "model_forward_time": 1.7614083290100098,
      "step": 74896
    },
    {
      "epoch": 8.544921875e-07,
      "step": 74896,
      "training_step_time": 7.63177227973938
    },
    {
      "epoch": 8.554458618164063e-07,
      "model_forward_time": 1.750565767288208,
      "step": 74897
    },
    {
      "epoch": 8.554458618164063e-07,
      "step": 74897,
      "training_step_time": 7.785345554351807
    },
    {
      "epoch": 8.563995361328125e-07,
      "model_forward_time": 1.748650312423706,
      "step": 74898
    },
    {
      "epoch": 8.563995361328125e-07,
      "step": 74898,
      "training_step_time": 7.631541728973389
    },
    {
      "epoch": 8.573532104492188e-07,
      "model_forward_time": 1.7509279251098633,
      "step": 74899
    },
    {
      "epoch": 8.573532104492188e-07,
      "step": 74899,
      "training_step_time": 6.79264760017395
    },
    {
      "epoch": 8.58306884765625e-07,
      "grad_norm": 0.08010297268629074,
      "learning_rate": 1.625897630349893e-06,
      "loss": 0.053,
      "step": 74900
    },
    {
      "epoch": 8.58306884765625e-07,
      "model_forward_time": 1.7492179870605469,
      "step": 74900
    },
    {
      "dynamics_loss_avg": 0.04788671433925629,
      "epoch": 8.58306884765625e-07,
      "step": 74900
    },
    {
      "action_loss_avg": 0.0014745205371582415,
      "epoch": 8.58306884765625e-07,
      "step": 74900
    },
    {
      "epoch": 8.58306884765625e-07,
      "step": 74900,
      "training_step_time": 6.849148988723755
    },
    {
      "epoch": 8.592605590820313e-07,
      "model_forward_time": 1.7474584579467773,
      "step": 74901
    },
    {
      "epoch": 8.592605590820313e-07,
      "step": 74901,
      "training_step_time": 7.743598222732544
    },
    {
      "epoch": 8.602142333984375e-07,
      "model_forward_time": 1.7544679641723633,
      "step": 74902
    },
    {
      "epoch": 8.602142333984375e-07,
      "step": 74902,
      "training_step_time": 10.547847032546997
    },
    {
      "epoch": 8.611679077148438e-07,
      "model_forward_time": 1.749553918838501,
      "step": 74903
    },
    {
      "epoch": 8.611679077148438e-07,
      "step": 74903,
      "training_step_time": 8.96066403388977
    },
    {
      "epoch": 8.6212158203125e-07,
      "model_forward_time": 2.378019332885742,
      "step": 74904
    },
    {
      "epoch": 8.6212158203125e-07,
      "step": 74904,
      "training_step_time": 7.786600351333618
    },
    {
      "epoch": 8.630752563476563e-07,
      "model_forward_time": 1.7448441982269287,
      "step": 74905
    },
    {
      "epoch": 8.630752563476563e-07,
      "step": 74905,
      "training_step_time": 7.664104461669922
    },
    {
      "epoch": 8.640289306640625e-07,
      "model_forward_time": 1.752697229385376,
      "step": 74906
    },
    {
      "epoch": 8.640289306640625e-07,
      "step": 74906,
      "training_step_time": 8.389531373977661
    },
    {
      "epoch": 8.649826049804688e-07,
      "model_forward_time": 1.7448844909667969,
      "step": 74907
    },
    {
      "epoch": 8.649826049804688e-07,
      "step": 74907,
      "training_step_time": 7.64233922958374
    },
    {
      "epoch": 8.65936279296875e-07,
      "model_forward_time": 1.7473516464233398,
      "step": 74908
    },
    {
      "epoch": 8.65936279296875e-07,
      "step": 74908,
      "training_step_time": 7.641297101974487
    },
    {
      "epoch": 8.668899536132813e-07,
      "model_forward_time": 1.7446424961090088,
      "step": 74909
    },
    {
      "epoch": 8.668899536132813e-07,
      "step": 74909,
      "training_step_time": 6.796288013458252
    },
    {
      "epoch": 8.678436279296875e-07,
      "grad_norm": 0.08828981965780258,
      "learning_rate": 1.6246775840194334e-06,
      "loss": 0.0549,
      "step": 74910
    },
    {
      "epoch": 8.678436279296875e-07,
      "model_forward_time": 1.7430319786071777,
      "step": 74910
    },
    {
      "dynamics_loss_avg": 0.04275258649140597,
      "epoch": 8.678436279296875e-07,
      "step": 74910
    },
    {
      "action_loss_avg": 0.0008569914585677907,
      "epoch": 8.678436279296875e-07,
      "step": 74910
    },
    {
      "epoch": 8.678436279296875e-07,
      "step": 74910,
      "training_step_time": 7.633553743362427
    },
    {
      "epoch": 8.687973022460938e-07,
      "model_forward_time": 1.7450201511383057,
      "step": 74911
    },
    {
      "epoch": 8.687973022460938e-07,
      "step": 74911,
      "training_step_time": 8.427542686462402
    },
    {
      "epoch": 8.697509765625e-07,
      "model_forward_time": 1.8132431507110596,
      "step": 74912
    },
    {
      "epoch": 8.697509765625e-07,
      "step": 74912,
      "training_step_time": 9.403450965881348
    },
    {
      "epoch": 8.707046508789063e-07,
      "model_forward_time": 2.3325769901275635,
      "step": 74913
    },
    {
      "epoch": 8.707046508789063e-07,
      "step": 74913,
      "training_step_time": 7.718556642532349
    },
    {
      "epoch": 8.716583251953125e-07,
      "model_forward_time": 1.7419917583465576,
      "step": 74914
    },
    {
      "epoch": 8.716583251953125e-07,
      "step": 74914,
      "training_step_time": 7.7443389892578125
    },
    {
      "epoch": 8.726119995117188e-07,
      "model_forward_time": 1.748058795928955,
      "step": 74915
    },
    {
      "epoch": 8.726119995117188e-07,
      "step": 74915,
      "training_step_time": 7.71106743812561
    },
    {
      "epoch": 8.73565673828125e-07,
      "model_forward_time": 1.7399344444274902,
      "step": 74916
    },
    {
      "epoch": 8.73565673828125e-07,
      "step": 74916,
      "training_step_time": 7.619809150695801
    },
    {
      "epoch": 8.745193481445313e-07,
      "model_forward_time": 1.7432026863098145,
      "step": 74917
    },
    {
      "epoch": 8.745193481445313e-07,
      "step": 74917,
      "training_step_time": 7.659157991409302
    },
    {
      "epoch": 8.754730224609375e-07,
      "model_forward_time": 1.748004674911499,
      "step": 74918
    },
    {
      "epoch": 8.754730224609375e-07,
      "step": 74918,
      "training_step_time": 6.7959325313568115
    },
    {
      "epoch": 8.764266967773438e-07,
      "model_forward_time": 1.7462973594665527,
      "step": 74919
    },
    {
      "epoch": 8.764266967773438e-07,
      "step": 74919,
      "training_step_time": 7.583540916442871
    },
    {
      "epoch": 8.7738037109375e-07,
      "grad_norm": 0.09041577577590942,
      "learning_rate": 1.6234579068091604e-06,
      "loss": 0.056,
      "step": 74920
    },
    {
      "epoch": 8.7738037109375e-07,
      "model_forward_time": 1.7435894012451172,
      "step": 74920
    },
    {
      "dynamics_loss_avg": 0.051154729910194874,
      "epoch": 8.7738037109375e-07,
      "step": 74920
    },
    {
      "action_loss_avg": 0.0017201152048073709,
      "epoch": 8.7738037109375e-07,
      "step": 74920
    },
    {
      "epoch": 8.7738037109375e-07,
      "step": 74920,
      "training_step_time": 8.364520788192749
    },
    {
      "epoch": 8.783340454101563e-07,
      "model_forward_time": 1.8078303337097168,
      "step": 74921
    },
    {
      "epoch": 8.783340454101563e-07,
      "step": 74921,
      "training_step_time": 8.300973653793335
    },
    {
      "epoch": 8.792877197265625e-07,
      "model_forward_time": 1.7507071495056152,
      "step": 74922
    },
    {
      "epoch": 8.792877197265625e-07,
      "step": 74922,
      "training_step_time": 7.774077653884888
    },
    {
      "epoch": 8.802413940429688e-07,
      "model_forward_time": 2.3695967197418213,
      "step": 74923
    },
    {
      "epoch": 8.802413940429688e-07,
      "step": 74923,
      "training_step_time": 7.793577432632446
    },
    {
      "epoch": 8.81195068359375e-07,
      "model_forward_time": 1.3788111209869385,
      "step": 74924
    },
    {
      "epoch": 8.81195068359375e-07,
      "step": 74924,
      "training_step_time": 7.75218391418457
    },
    {
      "epoch": 8.821487426757813e-07,
      "model_forward_time": 1.7438759803771973,
      "step": 74925
    },
    {
      "epoch": 8.821487426757813e-07,
      "step": 74925,
      "training_step_time": 7.366000652313232
    },
    {
      "epoch": 8.831024169921875e-07,
      "model_forward_time": 1.765132188796997,
      "step": 74926
    },
    {
      "epoch": 8.831024169921875e-07,
      "step": 74926,
      "training_step_time": 8.556267976760864
    },
    {
      "epoch": 8.840560913085938e-07,
      "model_forward_time": 1.7444558143615723,
      "step": 74927
    },
    {
      "epoch": 8.840560913085938e-07,
      "step": 74927,
      "training_step_time": 6.779886484146118
    },
    {
      "epoch": 8.85009765625e-07,
      "model_forward_time": 1.7408552169799805,
      "step": 74928
    },
    {
      "epoch": 8.85009765625e-07,
      "step": 74928,
      "training_step_time": 7.620476245880127
    },
    {
      "epoch": 8.859634399414063e-07,
      "model_forward_time": 1.7428419589996338,
      "step": 74929
    },
    {
      "epoch": 8.859634399414063e-07,
      "step": 74929,
      "training_step_time": 9.451741456985474
    },
    {
      "epoch": 8.869171142578125e-07,
      "grad_norm": 0.2565977871417999,
      "learning_rate": 1.6222385988524524e-06,
      "loss": 0.0556,
      "step": 74930
    },
    {
      "epoch": 8.869171142578125e-07,
      "model_forward_time": 1.80482816696167,
      "step": 74930
    },
    {
      "dynamics_loss_avg": 0.048563157208263874,
      "epoch": 8.869171142578125e-07,
      "step": 74930
    },
    {
      "action_loss_avg": 0.0014137612102786078,
      "epoch": 8.869171142578125e-07,
      "step": 74930
    },
    {
      "epoch": 8.869171142578125e-07,
      "step": 74930,
      "training_step_time": 9.242241859436035
    },
    {
      "epoch": 8.878707885742188e-07,
      "model_forward_time": 1.7417631149291992,
      "step": 74931
    },
    {
      "epoch": 8.878707885742188e-07,
      "step": 74931,
      "training_step_time": 8.526255130767822
    },
    {
      "epoch": 8.88824462890625e-07,
      "model_forward_time": 2.362945556640625,
      "step": 74932
    },
    {
      "epoch": 8.88824462890625e-07,
      "step": 74932,
      "training_step_time": 7.687429428100586
    },
    {
      "epoch": 8.897781372070313e-07,
      "model_forward_time": 1.7533445358276367,
      "step": 74933
    },
    {
      "epoch": 8.897781372070313e-07,
      "step": 74933,
      "training_step_time": 7.64446234703064
    },
    {
      "epoch": 8.907318115234375e-07,
      "model_forward_time": 1.7348525524139404,
      "step": 74934
    },
    {
      "epoch": 8.907318115234375e-07,
      "step": 74934,
      "training_step_time": 7.2498533725738525
    },
    {
      "epoch": 8.916854858398438e-07,
      "model_forward_time": 1.7488937377929688,
      "step": 74935
    },
    {
      "epoch": 8.916854858398438e-07,
      "step": 74935,
      "training_step_time": 7.616159915924072
    },
    {
      "epoch": 8.9263916015625e-07,
      "model_forward_time": 1.7418303489685059,
      "step": 74936
    },
    {
      "epoch": 8.9263916015625e-07,
      "step": 74936,
      "training_step_time": 7.646177291870117
    },
    {
      "epoch": 8.935928344726563e-07,
      "model_forward_time": 1.745384931564331,
      "step": 74937
    },
    {
      "epoch": 8.935928344726563e-07,
      "step": 74937,
      "training_step_time": 7.593401908874512
    },
    {
      "epoch": 8.945465087890624e-07,
      "model_forward_time": 1.7415297031402588,
      "step": 74938
    },
    {
      "epoch": 8.945465087890624e-07,
      "step": 74938,
      "training_step_time": 8.557471990585327
    },
    {
      "epoch": 8.955001831054687e-07,
      "model_forward_time": 1.7788636684417725,
      "step": 74939
    },
    {
      "epoch": 8.955001831054687e-07,
      "step": 74939,
      "training_step_time": 9.386152029037476
    },
    {
      "epoch": 8.96453857421875e-07,
      "grad_norm": 0.09650492668151855,
      "learning_rate": 1.6210196602826544e-06,
      "loss": 0.0559,
      "step": 74940
    },
    {
      "epoch": 8.96453857421875e-07,
      "model_forward_time": 1.7409813404083252,
      "step": 74940
    },
    {
      "dynamics_loss_avg": 0.03715554177761078,
      "epoch": 8.96453857421875e-07,
      "step": 74940
    },
    {
      "action_loss_avg": 0.0018289016734343022,
      "epoch": 8.96453857421875e-07,
      "step": 74940
    },
    {
      "epoch": 8.96453857421875e-07,
      "step": 74940,
      "training_step_time": 8.773088693618774
    },
    {
      "epoch": 8.974075317382812e-07,
      "model_forward_time": 2.3719704151153564,
      "step": 74941
    },
    {
      "epoch": 8.974075317382812e-07,
      "step": 74941,
      "training_step_time": 7.681536674499512
    },
    {
      "epoch": 8.983612060546875e-07,
      "model_forward_time": 1.743074655532837,
      "step": 74942
    },
    {
      "epoch": 8.983612060546875e-07,
      "step": 74942,
      "training_step_time": 7.647295236587524
    },
    {
      "epoch": 8.993148803710937e-07,
      "model_forward_time": 1.7497565746307373,
      "step": 74943
    },
    {
      "epoch": 8.993148803710937e-07,
      "step": 74943,
      "training_step_time": 7.621384382247925
    },
    {
      "epoch": 9.002685546875e-07,
      "model_forward_time": 1.7399320602416992,
      "step": 74944
    },
    {
      "epoch": 9.002685546875e-07,
      "step": 74944,
      "training_step_time": 7.5766518115997314
    },
    {
      "epoch": 9.012222290039062e-07,
      "model_forward_time": 1.7358295917510986,
      "step": 74945
    },
    {
      "epoch": 9.012222290039062e-07,
      "step": 74945,
      "training_step_time": 7.617154598236084
    },
    {
      "epoch": 9.021759033203125e-07,
      "model_forward_time": 1.7473163604736328,
      "step": 74946
    },
    {
      "epoch": 9.021759033203125e-07,
      "step": 74946,
      "training_step_time": 7.521629095077515
    },
    {
      "epoch": 9.031295776367187e-07,
      "model_forward_time": 1.7444379329681396,
      "step": 74947
    },
    {
      "epoch": 9.031295776367187e-07,
      "step": 74947,
      "training_step_time": 7.748023509979248
    },
    {
      "epoch": 9.04083251953125e-07,
      "model_forward_time": 1.758509874343872,
      "step": 74948
    },
    {
      "epoch": 9.04083251953125e-07,
      "step": 74948,
      "training_step_time": 8.240398168563843
    },
    {
      "epoch": 9.050369262695312e-07,
      "model_forward_time": 1.744724988937378,
      "step": 74949
    },
    {
      "epoch": 9.050369262695312e-07,
      "step": 74949,
      "training_step_time": 9.152587652206421
    },
    {
      "epoch": 9.059906005859375e-07,
      "grad_norm": 0.0833103358745575,
      "learning_rate": 1.6198010912330647e-06,
      "loss": 0.0554,
      "step": 74950
    },
    {
      "epoch": 9.059906005859375e-07,
      "model_forward_time": 2.353943347930908,
      "step": 74950
    },
    {
      "dynamics_loss_avg": 0.02750176899135113,
      "epoch": 9.059906005859375e-07,
      "step": 74950
    },
    {
      "action_loss_avg": 0.0006693065646686591,
      "epoch": 9.059906005859375e-07,
      "step": 74950
    },
    {
      "epoch": 9.059906005859375e-07,
      "step": 74950,
      "training_step_time": 7.725571393966675
    },
    {
      "epoch": 9.069442749023437e-07,
      "model_forward_time": 1.7546663284301758,
      "step": 74951
    },
    {
      "epoch": 9.069442749023437e-07,
      "step": 74951,
      "training_step_time": 8.236977577209473
    },
    {
      "epoch": 9.0789794921875e-07,
      "model_forward_time": 1.7538831233978271,
      "step": 74952
    },
    {
      "epoch": 9.0789794921875e-07,
      "step": 74952,
      "training_step_time": 7.61400580406189
    },
    {
      "epoch": 9.088516235351562e-07,
      "model_forward_time": 1.752422571182251,
      "step": 74953
    },
    {
      "epoch": 9.088516235351562e-07,
      "step": 74953,
      "training_step_time": 7.583975791931152
    },
    {
      "epoch": 9.098052978515625e-07,
      "model_forward_time": 1.7496037483215332,
      "step": 74954
    },
    {
      "epoch": 9.098052978515625e-07,
      "step": 74954,
      "training_step_time": 7.659704208374023
    },
    {
      "epoch": 9.107589721679687e-07,
      "model_forward_time": 1.7571370601654053,
      "step": 74955
    },
    {
      "epoch": 9.107589721679687e-07,
      "step": 74955,
      "training_step_time": 7.486621618270874
    },
    {
      "epoch": 9.11712646484375e-07,
      "model_forward_time": 1.7674274444580078,
      "step": 74956
    },
    {
      "epoch": 9.11712646484375e-07,
      "step": 74956,
      "training_step_time": 7.688030242919922
    },
    {
      "epoch": 9.126663208007812e-07,
      "model_forward_time": 1.7440485954284668,
      "step": 74957
    },
    {
      "epoch": 9.126663208007812e-07,
      "step": 74957,
      "training_step_time": 8.315490484237671
    },
    {
      "epoch": 9.136199951171875e-07,
      "model_forward_time": 1.8033041954040527,
      "step": 74958
    },
    {
      "epoch": 9.136199951171875e-07,
      "step": 74958,
      "training_step_time": 9.612688779830933
    },
    {
      "epoch": 9.145736694335937e-07,
      "model_forward_time": 1.75978422164917,
      "step": 74959
    },
    {
      "epoch": 9.145736694335937e-07,
      "step": 74959,
      "training_step_time": 8.388801336288452
    },
    {
      "epoch": 9.1552734375e-07,
      "grad_norm": 0.10755462199449539,
      "learning_rate": 1.6185828918369456e-06,
      "loss": 0.0563,
      "step": 74960
    },
    {
      "epoch": 9.1552734375e-07,
      "model_forward_time": 1.7488524913787842,
      "step": 74960
    },
    {
      "dynamics_loss_avg": 0.045436022244393824,
      "epoch": 9.1552734375e-07,
      "step": 74960
    },
    {
      "action_loss_avg": 0.002202381344977766,
      "epoch": 9.1552734375e-07,
      "step": 74960
    },
    {
      "epoch": 9.1552734375e-07,
      "step": 74960,
      "training_step_time": 7.80947208404541
    },
    {
      "epoch": 9.164810180664062e-07,
      "model_forward_time": 1.7557129859924316,
      "step": 74961
    },
    {
      "epoch": 9.164810180664062e-07,
      "step": 74961,
      "training_step_time": 7.675013065338135
    },
    {
      "epoch": 9.174346923828125e-07,
      "model_forward_time": 1.7523212432861328,
      "step": 74962
    },
    {
      "epoch": 9.174346923828125e-07,
      "step": 74962,
      "training_step_time": 7.603050947189331
    },
    {
      "epoch": 9.183883666992187e-07,
      "model_forward_time": 1.7598991394042969,
      "step": 74963
    },
    {
      "epoch": 9.183883666992187e-07,
      "step": 74963,
      "training_step_time": 7.563811779022217
    },
    {
      "epoch": 9.19342041015625e-07,
      "model_forward_time": 1.7453374862670898,
      "step": 74964
    },
    {
      "epoch": 9.19342041015625e-07,
      "step": 74964,
      "training_step_time": 7.128663539886475
    },
    {
      "epoch": 9.202957153320312e-07,
      "model_forward_time": 1.7537095546722412,
      "step": 74965
    },
    {
      "epoch": 9.202957153320312e-07,
      "step": 74965,
      "training_step_time": 8.533896207809448
    },
    {
      "epoch": 9.212493896484375e-07,
      "model_forward_time": 1.7408897876739502,
      "step": 74966
    },
    {
      "epoch": 9.212493896484375e-07,
      "step": 74966,
      "training_step_time": 8.635900259017944
    },
    {
      "epoch": 9.222030639648437e-07,
      "model_forward_time": 1.8008830547332764,
      "step": 74967
    },
    {
      "epoch": 9.222030639648437e-07,
      "step": 74967,
      "training_step_time": 8.943372964859009
    },
    {
      "epoch": 9.2315673828125e-07,
      "model_forward_time": 1.743471622467041,
      "step": 74968
    },
    {
      "epoch": 9.2315673828125e-07,
      "step": 74968,
      "training_step_time": 9.30858039855957
    },
    {
      "epoch": 9.241104125976562e-07,
      "model_forward_time": 2.370668888092041,
      "step": 74969
    },
    {
      "epoch": 9.241104125976562e-07,
      "step": 74969,
      "training_step_time": 8.683351755142212
    },
    {
      "epoch": 9.250640869140625e-07,
      "grad_norm": 0.11189208924770355,
      "learning_rate": 1.6173650622275171e-06,
      "loss": 0.0557,
      "step": 74970
    },
    {
      "epoch": 9.250640869140625e-07,
      "model_forward_time": 1.7477943897247314,
      "step": 74970
    },
    {
      "dynamics_loss_avg": 0.05433170292526483,
      "epoch": 9.250640869140625e-07,
      "step": 74970
    },
    {
      "action_loss_avg": 0.001611333666369319,
      "epoch": 9.250640869140625e-07,
      "step": 74970
    },
    {
      "epoch": 9.250640869140625e-07,
      "step": 74970,
      "training_step_time": 7.639743089675903
    },
    {
      "epoch": 9.260177612304687e-07,
      "model_forward_time": 1.7482311725616455,
      "step": 74971
    },
    {
      "epoch": 9.260177612304687e-07,
      "step": 74971,
      "training_step_time": 7.543014287948608
    },
    {
      "epoch": 9.26971435546875e-07,
      "model_forward_time": 1.7485430240631104,
      "step": 74972
    },
    {
      "epoch": 9.26971435546875e-07,
      "step": 74972,
      "training_step_time": 7.654631614685059
    },
    {
      "epoch": 9.279251098632812e-07,
      "model_forward_time": 1.74064302444458,
      "step": 74973
    },
    {
      "epoch": 9.279251098632812e-07,
      "step": 74973,
      "training_step_time": 7.180504083633423
    },
    {
      "epoch": 9.288787841796875e-07,
      "model_forward_time": 1.7510104179382324,
      "step": 74974
    },
    {
      "epoch": 9.288787841796875e-07,
      "step": 74974,
      "training_step_time": 7.626399755477905
    },
    {
      "epoch": 9.298324584960937e-07,
      "model_forward_time": 1.7414007186889648,
      "step": 74975
    },
    {
      "epoch": 9.298324584960937e-07,
      "step": 74975,
      "training_step_time": 9.879475831985474
    },
    {
      "epoch": 9.307861328125e-07,
      "model_forward_time": 1.7963194847106934,
      "step": 74976
    },
    {
      "epoch": 9.307861328125e-07,
      "step": 74976,
      "training_step_time": 10.385048627853394
    },
    {
      "epoch": 9.317398071289062e-07,
      "model_forward_time": 1.7382023334503174,
      "step": 74977
    },
    {
      "epoch": 9.317398071289062e-07,
      "step": 74977,
      "training_step_time": 7.754735946655273
    },
    {
      "epoch": 9.326934814453125e-07,
      "model_forward_time": 2.3481998443603516,
      "step": 74978
    },
    {
      "epoch": 9.326934814453125e-07,
      "step": 74978,
      "training_step_time": 7.7564377784729
    },
    {
      "epoch": 9.336471557617187e-07,
      "model_forward_time": 1.379127025604248,
      "step": 74979
    },
    {
      "epoch": 9.336471557617187e-07,
      "step": 74979,
      "training_step_time": 7.743438005447388
    },
    {
      "epoch": 9.34600830078125e-07,
      "grad_norm": 0.11363331973552704,
      "learning_rate": 1.6161476025379618e-06,
      "loss": 0.0562,
      "step": 74980
    },
    {
      "epoch": 9.34600830078125e-07,
      "model_forward_time": 1.7693796157836914,
      "step": 74980
    },
    {
      "dynamics_loss_avg": 0.05102218203246593,
      "epoch": 9.34600830078125e-07,
      "step": 74980
    },
    {
      "action_loss_avg": 0.0010433284915052354,
      "epoch": 9.34600830078125e-07,
      "step": 74980
    },
    {
      "epoch": 9.34600830078125e-07,
      "step": 74980,
      "training_step_time": 7.790276288986206
    },
    {
      "epoch": 9.355545043945312e-07,
      "model_forward_time": 1.7444510459899902,
      "step": 74981
    },
    {
      "epoch": 9.355545043945312e-07,
      "step": 74981,
      "training_step_time": 7.55344557762146
    },
    {
      "epoch": 9.365081787109375e-07,
      "model_forward_time": 1.7446129322052002,
      "step": 74982
    },
    {
      "epoch": 9.365081787109375e-07,
      "step": 74982,
      "training_step_time": 7.643125772476196
    },
    {
      "epoch": 9.374618530273437e-07,
      "model_forward_time": 1.769867181777954,
      "step": 74983
    },
    {
      "epoch": 9.374618530273437e-07,
      "step": 74983,
      "training_step_time": 7.4834043979644775
    },
    {
      "epoch": 9.3841552734375e-07,
      "model_forward_time": 1.738940954208374,
      "step": 74984
    },
    {
      "epoch": 9.3841552734375e-07,
      "step": 74984,
      "training_step_time": 8.480608701705933
    },
    {
      "epoch": 9.393692016601562e-07,
      "model_forward_time": 1.752558946609497,
      "step": 74985
    },
    {
      "epoch": 9.393692016601562e-07,
      "step": 74985,
      "training_step_time": 9.457518815994263
    },
    {
      "epoch": 9.403228759765625e-07,
      "model_forward_time": 1.4449446201324463,
      "step": 74986
    },
    {
      "epoch": 9.403228759765625e-07,
      "step": 74986,
      "training_step_time": 7.717686176300049
    },
    {
      "epoch": 9.412765502929687e-07,
      "model_forward_time": 2.4642155170440674,
      "step": 74987
    },
    {
      "epoch": 9.412765502929687e-07,
      "step": 74987,
      "training_step_time": 7.67400050163269
    },
    {
      "epoch": 9.42230224609375e-07,
      "model_forward_time": 1.895479440689087,
      "step": 74988
    },
    {
      "epoch": 9.42230224609375e-07,
      "step": 74988,
      "training_step_time": 7.715343713760376
    },
    {
      "epoch": 9.431838989257812e-07,
      "model_forward_time": 1.888890027999878,
      "step": 74989
    },
    {
      "epoch": 9.431838989257812e-07,
      "step": 74989,
      "training_step_time": 7.667074918746948
    },
    {
      "epoch": 9.441375732421875e-07,
      "grad_norm": 0.09385721385478973,
      "learning_rate": 1.6149305129014153e-06,
      "loss": 0.0565,
      "step": 74990
    },
    {
      "epoch": 9.441375732421875e-07,
      "model_forward_time": 1.8873779773712158,
      "step": 74990
    },
    {
      "dynamics_loss_avg": 0.05555632896721363,
      "epoch": 9.441375732421875e-07,
      "step": 74990
    },
    {
      "action_loss_avg": 0.001487569833989255,
      "epoch": 9.441375732421875e-07,
      "step": 74990
    },
    {
      "epoch": 9.441375732421875e-07,
      "step": 74990,
      "training_step_time": 7.5655457973480225
    },
    {
      "epoch": 9.450912475585937e-07,
      "model_forward_time": 1.885589361190796,
      "step": 74991
    },
    {
      "epoch": 9.450912475585937e-07,
      "step": 74991,
      "training_step_time": 7.656175374984741
    },
    {
      "epoch": 9.46044921875e-07,
      "model_forward_time": 1.8832151889801025,
      "step": 74992
    },
    {
      "epoch": 9.46044921875e-07,
      "step": 74992,
      "training_step_time": 8.172587394714355
    },
    {
      "epoch": 9.469985961914062e-07,
      "model_forward_time": 1.883906364440918,
      "step": 74993
    },
    {
      "epoch": 9.469985961914062e-07,
      "step": 74993,
      "training_step_time": 8.634155750274658
    },
    {
      "epoch": 9.479522705078125e-07,
      "model_forward_time": 1.8902335166931152,
      "step": 74994
    },
    {
      "epoch": 9.479522705078125e-07,
      "step": 74994,
      "training_step_time": 8.584008932113647
    },
    {
      "epoch": 9.489059448242187e-07,
      "model_forward_time": 1.8911573886871338,
      "step": 74995
    },
    {
      "epoch": 9.489059448242187e-07,
      "step": 74995,
      "training_step_time": 8.668570041656494
    },
    {
      "epoch": 9.49859619140625e-07,
      "model_forward_time": 2.491452693939209,
      "step": 74996
    },
    {
      "epoch": 9.49859619140625e-07,
      "step": 74996,
      "training_step_time": 7.694271802902222
    },
    {
      "epoch": 9.508132934570312e-07,
      "model_forward_time": 1.8953847885131836,
      "step": 74997
    },
    {
      "epoch": 9.508132934570312e-07,
      "step": 74997,
      "training_step_time": 8.180875539779663
    },
    {
      "epoch": 9.517669677734375e-07,
      "model_forward_time": 1.9029247760772705,
      "step": 74998
    },
    {
      "epoch": 9.517669677734375e-07,
      "step": 74998,
      "training_step_time": 7.901479959487915
    },
    {
      "epoch": 9.527206420898437e-07,
      "model_forward_time": 1.9370899200439453,
      "step": 74999
    },
    {
      "epoch": 9.527206420898437e-07,
      "step": 74999,
      "training_step_time": 8.192966938018799
    },
    {
      "epoch": 9.5367431640625e-07,
      "grad_norm": 0.08469463884830475,
      "learning_rate": 1.6137137934509805e-06,
      "loss": 0.0554,
      "step": 75000
    },
    {
      "epoch": 9.5367431640625e-07,
      "model_forward_time": 1.7742302417755127,
      "step": 75000
    },
    {
      "dynamics_loss_avg": 0.04583727288991213,
      "epoch": 9.5367431640625e-07,
      "step": 75000
    },
    {
      "action_loss_avg": 0.0015159653092268854,
      "epoch": 9.5367431640625e-07,
      "step": 75000
    },
    {
      "epoch": 9.5367431640625e-07,
      "step": 75000,
      "training_step_time": 6.646155118942261
    },
    {
      "epoch": 9.546279907226561e-07,
      "model_forward_time": 1.7617924213409424,
      "step": 75001
    },
    {
      "epoch": 9.546279907226561e-07,
      "step": 75001,
      "training_step_time": 8.261914730072021
    },
    {
      "epoch": 9.555816650390625e-07,
      "model_forward_time": 1.7601797580718994,
      "step": 75002
    },
    {
      "epoch": 9.555816650390625e-07,
      "step": 75002,
      "training_step_time": 8.439842462539673
    },
    {
      "epoch": 9.565353393554686e-07,
      "model_forward_time": 1.768979787826538,
      "step": 75003
    },
    {
      "epoch": 9.565353393554686e-07,
      "step": 75003,
      "training_step_time": 10.597193717956543
    },
    {
      "epoch": 9.57489013671875e-07,
      "model_forward_time": 1.7672579288482666,
      "step": 75004
    },
    {
      "epoch": 9.57489013671875e-07,
      "step": 75004,
      "training_step_time": 10.74822998046875
    },
    {
      "epoch": 9.584426879882811e-07,
      "model_forward_time": 3.4706871509552,
      "step": 75005
    },
    {
      "epoch": 9.584426879882811e-07,
      "step": 75005,
      "training_step_time": 8.3186776638031
    },
    {
      "epoch": 9.593963623046875e-07,
      "model_forward_time": 1.415308952331543,
      "step": 75006
    },
    {
      "epoch": 9.593963623046875e-07,
      "step": 75006,
      "training_step_time": 7.729202032089233
    },
    {
      "epoch": 9.603500366210937e-07,
      "model_forward_time": 1.7595140933990479,
      "step": 75007
    },
    {
      "epoch": 9.603500366210937e-07,
      "step": 75007,
      "training_step_time": 7.6288909912109375
    },
    {
      "epoch": 9.613037109375e-07,
      "model_forward_time": 1.76582932472229,
      "step": 75008
    },
    {
      "epoch": 9.613037109375e-07,
      "step": 75008,
      "training_step_time": 7.266707420349121
    },
    {
      "epoch": 9.622573852539062e-07,
      "model_forward_time": 1.7642366886138916,
      "step": 75009
    },
    {
      "epoch": 9.622573852539062e-07,
      "step": 75009,
      "training_step_time": 7.62205171585083
    },
    {
      "epoch": 9.632110595703125e-07,
      "grad_norm": 0.11875688284635544,
      "learning_rate": 1.6124974443197122e-06,
      "loss": 0.0582,
      "step": 75010
    },
    {
      "epoch": 9.632110595703125e-07,
      "model_forward_time": 1.7718088626861572,
      "step": 75010
    },
    {
      "dynamics_loss_avg": 0.058293452858924864,
      "epoch": 9.632110595703125e-07,
      "step": 75010
    },
    {
      "action_loss_avg": 0.0014359978027641772,
      "epoch": 9.632110595703125e-07,
      "step": 75010
    },
    {
      "epoch": 9.632110595703125e-07,
      "step": 75010,
      "training_step_time": 7.304761648178101
    },
    {
      "epoch": 9.641647338867187e-07,
      "model_forward_time": 1.7677998542785645,
      "step": 75011
    },
    {
      "epoch": 9.641647338867187e-07,
      "step": 75011,
      "training_step_time": 7.707321405410767
    },
    {
      "epoch": 9.65118408203125e-07,
      "model_forward_time": 1.7547681331634521,
      "step": 75012
    },
    {
      "epoch": 9.65118408203125e-07,
      "step": 75012,
      "training_step_time": 8.533113718032837
    },
    {
      "epoch": 9.660720825195312e-07,
      "model_forward_time": 1.8846795558929443,
      "step": 75013
    },
    {
      "epoch": 9.660720825195312e-07,
      "step": 75013,
      "training_step_time": 10.659116744995117
    },
    {
      "epoch": 9.670257568359375e-07,
      "model_forward_time": 1.755582332611084,
      "step": 75014
    },
    {
      "epoch": 9.670257568359375e-07,
      "step": 75014,
      "training_step_time": 9.374238967895508
    },
    {
      "epoch": 9.679794311523437e-07,
      "model_forward_time": 1.757889986038208,
      "step": 75015
    },
    {
      "epoch": 9.679794311523437e-07,
      "step": 75015,
      "training_step_time": 7.719613790512085
    },
    {
      "epoch": 9.6893310546875e-07,
      "model_forward_time": 1.7656443119049072,
      "step": 75016
    },
    {
      "epoch": 9.6893310546875e-07,
      "step": 75016,
      "training_step_time": 7.715615510940552
    },
    {
      "epoch": 9.698867797851562e-07,
      "model_forward_time": 1.7720580101013184,
      "step": 75017
    },
    {
      "epoch": 9.698867797851562e-07,
      "step": 75017,
      "training_step_time": 7.408484220504761
    },
    {
      "epoch": 9.708404541015625e-07,
      "model_forward_time": 1.410393476486206,
      "step": 75018
    },
    {
      "epoch": 9.708404541015625e-07,
      "step": 75018,
      "training_step_time": 7.672795057296753
    },
    {
      "epoch": 9.717941284179687e-07,
      "model_forward_time": 1.7711269855499268,
      "step": 75019
    },
    {
      "epoch": 9.717941284179687e-07,
      "step": 75019,
      "training_step_time": 7.426335573196411
    },
    {
      "epoch": 9.72747802734375e-07,
      "grad_norm": 0.09590895473957062,
      "learning_rate": 1.6112814656406306e-06,
      "loss": 0.0553,
      "step": 75020
    },
    {
      "epoch": 9.72747802734375e-07,
      "model_forward_time": 1.7648935317993164,
      "step": 75020
    },
    {
      "dynamics_loss_avg": 0.04936673007905483,
      "epoch": 9.72747802734375e-07,
      "step": 75020
    },
    {
      "action_loss_avg": 0.0013259797357022761,
      "epoch": 9.72747802734375e-07,
      "step": 75020
    },
    {
      "epoch": 9.72747802734375e-07,
      "step": 75020,
      "training_step_time": 7.768072605133057
    },
    {
      "epoch": 9.737014770507812e-07,
      "model_forward_time": 1.7706470489501953,
      "step": 75021
    },
    {
      "epoch": 9.737014770507812e-07,
      "step": 75021,
      "training_step_time": 9.561994075775146
    },
    {
      "epoch": 9.746551513671875e-07,
      "model_forward_time": 1.8242366313934326,
      "step": 75022
    },
    {
      "epoch": 9.746551513671875e-07,
      "step": 75022,
      "training_step_time": 8.4679536819458
    },
    {
      "epoch": 9.756088256835937e-07,
      "model_forward_time": 1.7693262100219727,
      "step": 75023
    },
    {
      "epoch": 9.756088256835937e-07,
      "step": 75023,
      "training_step_time": 8.589405298233032
    },
    {
      "epoch": 9.765625e-07,
      "model_forward_time": 2.8573086261749268,
      "step": 75024
    },
    {
      "epoch": 9.765625e-07,
      "step": 75024,
      "training_step_time": 7.816930294036865
    },
    {
      "epoch": 9.775161743164062e-07,
      "model_forward_time": 1.404289722442627,
      "step": 75025
    },
    {
      "epoch": 9.775161743164062e-07,
      "step": 75025,
      "training_step_time": 7.788175106048584
    },
    {
      "epoch": 9.784698486328125e-07,
      "model_forward_time": 1.775575876235962,
      "step": 75026
    },
    {
      "epoch": 9.784698486328125e-07,
      "step": 75026,
      "training_step_time": 7.830090284347534
    },
    {
      "epoch": 9.794235229492187e-07,
      "model_forward_time": 1.7681868076324463,
      "step": 75027
    },
    {
      "epoch": 9.794235229492187e-07,
      "step": 75027,
      "training_step_time": 7.74244499206543
    },
    {
      "epoch": 9.80377197265625e-07,
      "model_forward_time": 1.7700705528259277,
      "step": 75028
    },
    {
      "epoch": 9.80377197265625e-07,
      "step": 75028,
      "training_step_time": 7.713948965072632
    },
    {
      "epoch": 9.813308715820312e-07,
      "model_forward_time": 1.7784347534179688,
      "step": 75029
    },
    {
      "epoch": 9.813308715820312e-07,
      "step": 75029,
      "training_step_time": 8.643491744995117
    },
    {
      "epoch": 9.822845458984375e-07,
      "grad_norm": 0.11028487235307693,
      "learning_rate": 1.6100658575467127e-06,
      "loss": 0.0576,
      "step": 75030
    },
    {
      "epoch": 9.822845458984375e-07,
      "model_forward_time": 1.7669665813446045,
      "step": 75030
    },
    {
      "dynamics_loss_avg": 0.0446965679526329,
      "epoch": 9.822845458984375e-07,
      "step": 75030
    },
    {
      "action_loss_avg": 0.0009460038185352459,
      "epoch": 9.822845458984375e-07,
      "step": 75030
    },
    {
      "epoch": 9.822845458984375e-07,
      "step": 75030,
      "training_step_time": 7.84385085105896
    },
    {
      "epoch": 9.832382202148437e-07,
      "model_forward_time": 1.8052551746368408,
      "step": 75031
    },
    {
      "epoch": 9.832382202148437e-07,
      "step": 75031,
      "training_step_time": 9.244937419891357
    },
    {
      "epoch": 9.8419189453125e-07,
      "model_forward_time": 1.7679967880249023,
      "step": 75032
    },
    {
      "epoch": 9.8419189453125e-07,
      "step": 75032,
      "training_step_time": 7.897881507873535
    },
    {
      "epoch": 9.851455688476562e-07,
      "model_forward_time": 2.8435981273651123,
      "step": 75033
    },
    {
      "epoch": 9.851455688476562e-07,
      "step": 75033,
      "training_step_time": 8.463793992996216
    },
    {
      "epoch": 9.860992431640625e-07,
      "model_forward_time": 1.7815814018249512,
      "step": 75034
    },
    {
      "epoch": 9.860992431640625e-07,
      "step": 75034,
      "training_step_time": 7.479452848434448
    },
    {
      "epoch": 9.870529174804687e-07,
      "model_forward_time": 1.7715308666229248,
      "step": 75035
    },
    {
      "epoch": 9.870529174804687e-07,
      "step": 75035,
      "training_step_time": 8.35840392112732
    },
    {
      "epoch": 9.88006591796875e-07,
      "model_forward_time": 1.7715144157409668,
      "step": 75036
    },
    {
      "epoch": 9.88006591796875e-07,
      "step": 75036,
      "training_step_time": 7.600210189819336
    },
    {
      "epoch": 9.889602661132812e-07,
      "model_forward_time": 1.7725422382354736,
      "step": 75037
    },
    {
      "epoch": 9.889602661132812e-07,
      "step": 75037,
      "training_step_time": 7.647735118865967
    },
    {
      "epoch": 9.899139404296875e-07,
      "model_forward_time": 1.7720646858215332,
      "step": 75038
    },
    {
      "epoch": 9.899139404296875e-07,
      "step": 75038,
      "training_step_time": 7.5448830127716064
    },
    {
      "epoch": 9.908676147460937e-07,
      "model_forward_time": 1.7633137702941895,
      "step": 75039
    },
    {
      "epoch": 9.908676147460937e-07,
      "step": 75039,
      "training_step_time": 8.30954384803772
    },
    {
      "epoch": 9.918212890625e-07,
      "grad_norm": 0.11211476475000381,
      "learning_rate": 1.6088506201708976e-06,
      "loss": 0.0573,
      "step": 75040
    },
    {
      "epoch": 9.918212890625e-07,
      "model_forward_time": 1.7731077671051025,
      "step": 75040
    },
    {
      "dynamics_loss_avg": 0.05968810878694057,
      "epoch": 9.918212890625e-07,
      "step": 75040
    },
    {
      "action_loss_avg": 0.0015499895031098276,
      "epoch": 9.918212890625e-07,
      "step": 75040
    },
    {
      "epoch": 9.918212890625e-07,
      "step": 75040,
      "training_step_time": 8.449570655822754
    },
    {
      "epoch": 9.927749633789062e-07,
      "model_forward_time": 1.760801076889038,
      "step": 75041
    },
    {
      "epoch": 9.927749633789062e-07,
      "step": 75041,
      "training_step_time": 9.612491846084595
    },
    {
      "epoch": 9.937286376953125e-07,
      "model_forward_time": 2.863497734069824,
      "step": 75042
    },
    {
      "epoch": 9.937286376953125e-07,
      "step": 75042,
      "training_step_time": 7.857651710510254
    },
    {
      "epoch": 9.946823120117187e-07,
      "model_forward_time": 1.7517249584197998,
      "step": 75043
    },
    {
      "epoch": 9.946823120117187e-07,
      "step": 75043,
      "training_step_time": 9.346012115478516
    },
    {
      "epoch": 9.95635986328125e-07,
      "model_forward_time": 1.7290019989013672,
      "step": 75044
    },
    {
      "epoch": 9.95635986328125e-07,
      "step": 75044,
      "training_step_time": 7.671864986419678
    },
    {
      "epoch": 9.965896606445312e-07,
      "model_forward_time": 1.7568233013153076,
      "step": 75045
    },
    {
      "epoch": 9.965896606445312e-07,
      "step": 75045,
      "training_step_time": 8.334240674972534
    },
    {
      "epoch": 9.975433349609375e-07,
      "model_forward_time": 1.7462594509124756,
      "step": 75046
    },
    {
      "epoch": 9.975433349609375e-07,
      "step": 75046,
      "training_step_time": 7.597563743591309
    },
    {
      "epoch": 9.984970092773437e-07,
      "model_forward_time": 1.7481215000152588,
      "step": 75047
    },
    {
      "epoch": 9.984970092773437e-07,
      "step": 75047,
      "training_step_time": 8.288116931915283
    },
    {
      "epoch": 9.9945068359375e-07,
      "model_forward_time": 1.7582643032073975,
      "step": 75048
    },
    {
      "epoch": 9.9945068359375e-07,
      "step": 75048,
      "training_step_time": 7.72814154624939
    },
    {
      "epoch": 1.0004043579101562e-06,
      "model_forward_time": 1.7427418231964111,
      "step": 75049
    },
    {
      "epoch": 1.0004043579101562e-06,
      "step": 75049,
      "training_step_time": 8.413575172424316
    },
    {
      "epoch": 1.0013580322265625e-06,
      "grad_norm": 0.10819634795188904,
      "learning_rate": 1.6076357536460785e-06,
      "loss": 0.057,
      "step": 75050
    },
    {
      "epoch": 1.0013580322265625e-06,
      "model_forward_time": 1.8379309177398682,
      "step": 75050
    },
    {
      "dynamics_loss_avg": 0.05354289300739765,
      "epoch": 1.0013580322265625e-06,
      "step": 75050
    },
    {
      "action_loss_avg": 0.0011722127674147487,
      "epoch": 1.0013580322265625e-06,
      "step": 75050
    },
    {
      "epoch": 1.0013580322265625e-06,
      "step": 75050,
      "training_step_time": 10.08428692817688
    },
    {
      "epoch": 1.0023117065429687e-06,
      "model_forward_time": 2.772284507751465,
      "step": 75051
    },
    {
      "epoch": 1.0023117065429687e-06,
      "step": 75051,
      "training_step_time": 8.493439197540283
    },
    {
      "epoch": 1.003265380859375e-06,
      "model_forward_time": 1.7509043216705322,
      "step": 75052
    },
    {
      "epoch": 1.003265380859375e-06,
      "step": 75052,
      "training_step_time": 7.880397081375122
    },
    {
      "epoch": 1.0042190551757812e-06,
      "model_forward_time": 1.7466294765472412,
      "step": 75053
    },
    {
      "epoch": 1.0042190551757812e-06,
      "step": 75053,
      "training_step_time": 7.686124324798584
    },
    {
      "epoch": 1.0051727294921875e-06,
      "model_forward_time": 1.7457988262176514,
      "step": 75054
    },
    {
      "epoch": 1.0051727294921875e-06,
      "step": 75054,
      "training_step_time": 8.074261903762817
    },
    {
      "epoch": 1.0061264038085937e-06,
      "model_forward_time": 1.7485334873199463,
      "step": 75055
    },
    {
      "epoch": 1.0061264038085937e-06,
      "step": 75055,
      "training_step_time": 7.670424938201904
    },
    {
      "epoch": 1.007080078125e-06,
      "model_forward_time": 1.7524008750915527,
      "step": 75056
    },
    {
      "epoch": 1.007080078125e-06,
      "step": 75056,
      "training_step_time": 7.368131399154663
    },
    {
      "epoch": 1.0080337524414062e-06,
      "model_forward_time": 1.74774169921875,
      "step": 75057
    },
    {
      "epoch": 1.0080337524414062e-06,
      "step": 75057,
      "training_step_time": 7.7540810108184814
    },
    {
      "epoch": 1.0089874267578125e-06,
      "model_forward_time": 1.7410297393798828,
      "step": 75058
    },
    {
      "epoch": 1.0089874267578125e-06,
      "step": 75058,
      "training_step_time": 9.645386695861816
    },
    {
      "epoch": 1.0099411010742187e-06,
      "model_forward_time": 1.8494873046875,
      "step": 75059
    },
    {
      "epoch": 1.0099411010742187e-06,
      "step": 75059,
      "training_step_time": 9.33319640159607
    },
    {
      "epoch": 1.010894775390625e-06,
      "grad_norm": 0.1008223444223404,
      "learning_rate": 1.606421258105113e-06,
      "loss": 0.0577,
      "step": 75060
    },
    {
      "epoch": 1.010894775390625e-06,
      "model_forward_time": 1.7433052062988281,
      "step": 75060
    },
    {
      "dynamics_loss_avg": 0.06054929308593273,
      "epoch": 1.010894775390625e-06,
      "step": 75060
    },
    {
      "action_loss_avg": 0.002781419039820321,
      "epoch": 1.010894775390625e-06,
      "step": 75060
    },
    {
      "epoch": 1.010894775390625e-06,
      "step": 75060,
      "training_step_time": 7.88878870010376
    },
    {
      "epoch": 1.0118484497070312e-06,
      "model_forward_time": 2.7988107204437256,
      "step": 75061
    },
    {
      "epoch": 1.0118484497070312e-06,
      "step": 75061,
      "training_step_time": 7.884413003921509
    },
    {
      "epoch": 1.0128021240234376e-06,
      "model_forward_time": 1.7362008094787598,
      "step": 75062
    },
    {
      "epoch": 1.0128021240234376e-06,
      "step": 75062,
      "training_step_time": 7.6047351360321045
    },
    {
      "epoch": 1.0137557983398437e-06,
      "model_forward_time": 1.3626961708068848,
      "step": 75063
    },
    {
      "epoch": 1.0137557983398437e-06,
      "step": 75063,
      "training_step_time": 7.8000328540802
    },
    {
      "epoch": 1.01470947265625e-06,
      "model_forward_time": 1.7416572570800781,
      "step": 75064
    },
    {
      "epoch": 1.01470947265625e-06,
      "step": 75064,
      "training_step_time": 7.620584011077881
    },
    {
      "epoch": 1.0156631469726562e-06,
      "model_forward_time": 1.7362794876098633,
      "step": 75065
    },
    {
      "epoch": 1.0156631469726562e-06,
      "step": 75065,
      "training_step_time": 7.664179086685181
    },
    {
      "epoch": 1.0166168212890626e-06,
      "model_forward_time": 1.7350668907165527,
      "step": 75066
    },
    {
      "epoch": 1.0166168212890626e-06,
      "step": 75066,
      "training_step_time": 8.77042007446289
    },
    {
      "epoch": 1.0175704956054687e-06,
      "model_forward_time": 1.7417271137237549,
      "step": 75067
    },
    {
      "epoch": 1.0175704956054687e-06,
      "step": 75067,
      "training_step_time": 7.89156174659729
    },
    {
      "epoch": 1.018524169921875e-06,
      "model_forward_time": 1.7551085948944092,
      "step": 75068
    },
    {
      "epoch": 1.018524169921875e-06,
      "step": 75068,
      "training_step_time": 7.778044700622559
    },
    {
      "epoch": 1.0194778442382812e-06,
      "model_forward_time": 1.7578237056732178,
      "step": 75069
    },
    {
      "epoch": 1.0194778442382812e-06,
      "step": 75069,
      "training_step_time": 8.876042604446411
    },
    {
      "epoch": 1.0204315185546876e-06,
      "grad_norm": 0.12940901517868042,
      "learning_rate": 1.6052071336808184e-06,
      "loss": 0.057,
      "step": 75070
    },
    {
      "epoch": 1.0204315185546876e-06,
      "model_forward_time": 2.821479558944702,
      "step": 75070
    },
    {
      "dynamics_loss_avg": 0.04498484265059233,
      "epoch": 1.0204315185546876e-06,
      "step": 75070
    },
    {
      "action_loss_avg": 0.0008897577790776268,
      "epoch": 1.0204315185546876e-06,
      "step": 75070
    },
    {
      "epoch": 1.0204315185546876e-06,
      "step": 75070,
      "training_step_time": 7.869141578674316
    },
    {
      "epoch": 1.0213851928710937e-06,
      "model_forward_time": 1.7413561344146729,
      "step": 75071
    },
    {
      "epoch": 1.0213851928710937e-06,
      "step": 75071,
      "training_step_time": 7.63190221786499
    },
    {
      "epoch": 1.0223388671875e-06,
      "model_forward_time": 1.741215705871582,
      "step": 75072
    },
    {
      "epoch": 1.0223388671875e-06,
      "step": 75072,
      "training_step_time": 8.496421337127686
    },
    {
      "epoch": 1.0232925415039062e-06,
      "model_forward_time": 1.733156442642212,
      "step": 75073
    },
    {
      "epoch": 1.0232925415039062e-06,
      "step": 75073,
      "training_step_time": 7.587612628936768
    },
    {
      "epoch": 1.0242462158203126e-06,
      "model_forward_time": 1.7322537899017334,
      "step": 75074
    },
    {
      "epoch": 1.0242462158203126e-06,
      "step": 75074,
      "training_step_time": 8.397122383117676
    },
    {
      "epoch": 1.0251998901367187e-06,
      "model_forward_time": 1.7434196472167969,
      "step": 75075
    },
    {
      "epoch": 1.0251998901367187e-06,
      "step": 75075,
      "training_step_time": 8.732793092727661
    },
    {
      "epoch": 1.026153564453125e-06,
      "model_forward_time": 1.7372174263000488,
      "step": 75076
    },
    {
      "epoch": 1.026153564453125e-06,
      "step": 75076,
      "training_step_time": 8.622140169143677
    },
    {
      "epoch": 1.0271072387695312e-06,
      "model_forward_time": 1.3664848804473877,
      "step": 75077
    },
    {
      "epoch": 1.0271072387695312e-06,
      "step": 75077,
      "training_step_time": 10.332739114761353
    },
    {
      "epoch": 1.0280609130859376e-06,
      "model_forward_time": 1.7582385540008545,
      "step": 75078
    },
    {
      "epoch": 1.0280609130859376e-06,
      "step": 75078,
      "training_step_time": 8.381450891494751
    },
    {
      "epoch": 1.0290145874023437e-06,
      "model_forward_time": 2.828861713409424,
      "step": 75079
    },
    {
      "epoch": 1.0290145874023437e-06,
      "step": 75079,
      "training_step_time": 8.411674737930298
    },
    {
      "epoch": 1.02996826171875e-06,
      "grad_norm": 0.09996945410966873,
      "learning_rate": 1.603993380505966e-06,
      "loss": 0.0571,
      "step": 75080
    },
    {
      "epoch": 1.02996826171875e-06,
      "model_forward_time": 1.7343833446502686,
      "step": 75080
    },
    {
      "dynamics_loss_avg": 0.05082710590213537,
      "epoch": 1.02996826171875e-06,
      "step": 75080
    },
    {
      "action_loss_avg": 0.0014382582885446027,
      "epoch": 1.02996826171875e-06,
      "step": 75080
    },
    {
      "epoch": 1.02996826171875e-06,
      "step": 75080,
      "training_step_time": 7.831781625747681
    },
    {
      "epoch": 1.0309219360351562e-06,
      "model_forward_time": 1.7299046516418457,
      "step": 75081
    },
    {
      "epoch": 1.0309219360351562e-06,
      "step": 75081,
      "training_step_time": 7.770925760269165
    },
    {
      "epoch": 1.0318756103515626e-06,
      "model_forward_time": 1.7418622970581055,
      "step": 75082
    },
    {
      "epoch": 1.0318756103515626e-06,
      "step": 75082,
      "training_step_time": 7.5551323890686035
    },
    {
      "epoch": 1.0328292846679687e-06,
      "model_forward_time": 1.7440297603607178,
      "step": 75083
    },
    {
      "epoch": 1.0328292846679687e-06,
      "step": 75083,
      "training_step_time": 7.30321478843689
    },
    {
      "epoch": 1.033782958984375e-06,
      "model_forward_time": 1.7507212162017822,
      "step": 75084
    },
    {
      "epoch": 1.033782958984375e-06,
      "step": 75084,
      "training_step_time": 9.202745914459229
    },
    {
      "epoch": 1.0347366333007812e-06,
      "model_forward_time": 1.742002248764038,
      "step": 75085
    },
    {
      "epoch": 1.0347366333007812e-06,
      "step": 75085,
      "training_step_time": 8.672822952270508
    },
    {
      "epoch": 1.0356903076171876e-06,
      "model_forward_time": 1.7649967670440674,
      "step": 75086
    },
    {
      "epoch": 1.0356903076171876e-06,
      "step": 75086,
      "training_step_time": 8.215555667877197
    },
    {
      "epoch": 1.0366439819335937e-06,
      "model_forward_time": 1.7567329406738281,
      "step": 75087
    },
    {
      "epoch": 1.0366439819335937e-06,
      "step": 75087,
      "training_step_time": 8.893044710159302
    },
    {
      "epoch": 1.03759765625e-06,
      "model_forward_time": 2.8299131393432617,
      "step": 75088
    },
    {
      "epoch": 1.03759765625e-06,
      "step": 75088,
      "training_step_time": 7.88555645942688
    },
    {
      "epoch": 1.0385513305664062e-06,
      "model_forward_time": 1.7398755550384521,
      "step": 75089
    },
    {
      "epoch": 1.0385513305664062e-06,
      "step": 75089,
      "training_step_time": 8.689841508865356
    },
    {
      "epoch": 1.0395050048828126e-06,
      "grad_norm": 0.11681541800498962,
      "learning_rate": 1.6027799987132913e-06,
      "loss": 0.0555,
      "step": 75090
    },
    {
      "epoch": 1.0395050048828126e-06,
      "model_forward_time": 1.7391619682312012,
      "step": 75090
    },
    {
      "dynamics_loss_avg": 0.05628131702542305,
      "epoch": 1.0395050048828126e-06,
      "step": 75090
    },
    {
      "action_loss_avg": 0.0029479248390998692,
      "epoch": 1.0395050048828126e-06,
      "step": 75090
    },
    {
      "epoch": 1.0395050048828126e-06,
      "step": 75090,
      "training_step_time": 8.28211498260498
    },
    {
      "epoch": 1.0404586791992187e-06,
      "model_forward_time": 1.734786033630371,
      "step": 75091
    },
    {
      "epoch": 1.0404586791992187e-06,
      "step": 75091,
      "training_step_time": 7.456470489501953
    },
    {
      "epoch": 1.041412353515625e-06,
      "model_forward_time": 1.7253103256225586,
      "step": 75092
    },
    {
      "epoch": 1.041412353515625e-06,
      "step": 75092,
      "training_step_time": 7.573421955108643
    },
    {
      "epoch": 1.0423660278320312e-06,
      "model_forward_time": 1.7488465309143066,
      "step": 75093
    },
    {
      "epoch": 1.0423660278320312e-06,
      "step": 75093,
      "training_step_time": 8.127911567687988
    },
    {
      "epoch": 1.0433197021484376e-06,
      "model_forward_time": 1.735689401626587,
      "step": 75094
    },
    {
      "epoch": 1.0433197021484376e-06,
      "step": 75094,
      "training_step_time": 7.8314528465271
    },
    {
      "epoch": 1.0442733764648437e-06,
      "model_forward_time": 1.7381863594055176,
      "step": 75095
    },
    {
      "epoch": 1.0442733764648437e-06,
      "step": 75095,
      "training_step_time": 8.608996152877808
    },
    {
      "epoch": 1.04522705078125e-06,
      "model_forward_time": 1.8024349212646484,
      "step": 75096
    },
    {
      "epoch": 1.04522705078125e-06,
      "step": 75096,
      "training_step_time": 11.170558452606201
    },
    {
      "epoch": 1.0461807250976562e-06,
      "model_forward_time": 1.7476251125335693,
      "step": 75097
    },
    {
      "epoch": 1.0461807250976562e-06,
      "step": 75097,
      "training_step_time": 11.359524250030518
    },
    {
      "epoch": 1.0471343994140626e-06,
      "model_forward_time": 1.7321038246154785,
      "step": 75098
    },
    {
      "epoch": 1.0471343994140626e-06,
      "step": 75098,
      "training_step_time": 7.925966501235962
    },
    {
      "epoch": 1.0480880737304687e-06,
      "model_forward_time": 1.7357170581817627,
      "step": 75099
    },
    {
      "epoch": 1.0480880737304687e-06,
      "step": 75099,
      "training_step_time": 8.467934846878052
    },
    {
      "epoch": 1.049041748046875e-06,
      "grad_norm": 0.10008391737937927,
      "learning_rate": 1.601566988435489e-06,
      "loss": 0.0573,
      "step": 75100
    },
    {
      "epoch": 1.049041748046875e-06,
      "model_forward_time": 1.7401924133300781,
      "step": 75100
    },
    {
      "dynamics_loss_avg": 0.04652857575565576,
      "epoch": 1.049041748046875e-06,
      "step": 75100
    },
    {
      "action_loss_avg": 0.001316216739360243,
      "epoch": 1.049041748046875e-06,
      "step": 75100
    },
    {
      "epoch": 1.049041748046875e-06,
      "step": 75100,
      "training_step_time": 7.804813623428345
    },
    {
      "epoch": 1.0499954223632812e-06,
      "model_forward_time": 1.746873378753662,
      "step": 75101
    },
    {
      "epoch": 1.0499954223632812e-06,
      "step": 75101,
      "training_step_time": 7.59900975227356
    },
    {
      "epoch": 1.0509490966796876e-06,
      "model_forward_time": 1.7397584915161133,
      "step": 75102
    },
    {
      "epoch": 1.0509490966796876e-06,
      "step": 75102,
      "training_step_time": 7.707086086273193
    },
    {
      "epoch": 1.0519027709960937e-06,
      "model_forward_time": 1.738682746887207,
      "step": 75103
    },
    {
      "epoch": 1.0519027709960937e-06,
      "step": 75103,
      "training_step_time": 7.770745038986206
    },
    {
      "epoch": 1.0528564453125e-06,
      "model_forward_time": 1.7390947341918945,
      "step": 75104
    },
    {
      "epoch": 1.0528564453125e-06,
      "step": 75104,
      "training_step_time": 7.856287479400635
    },
    {
      "epoch": 1.0538101196289062e-06,
      "model_forward_time": 1.7552156448364258,
      "step": 75105
    },
    {
      "epoch": 1.0538101196289062e-06,
      "step": 75105,
      "training_step_time": 9.156914949417114
    },
    {
      "epoch": 1.0547637939453126e-06,
      "model_forward_time": 1.7406702041625977,
      "step": 75106
    },
    {
      "epoch": 1.0547637939453126e-06,
      "step": 75106,
      "training_step_time": 10.73376989364624
    },
    {
      "epoch": 1.0557174682617187e-06,
      "model_forward_time": 2.803771495819092,
      "step": 75107
    },
    {
      "epoch": 1.0557174682617187e-06,
      "step": 75107,
      "training_step_time": 9.336734771728516
    },
    {
      "epoch": 1.056671142578125e-06,
      "model_forward_time": 1.740767478942871,
      "step": 75108
    },
    {
      "epoch": 1.056671142578125e-06,
      "step": 75108,
      "training_step_time": 8.35347032546997
    },
    {
      "epoch": 1.0576248168945312e-06,
      "model_forward_time": 1.3661863803863525,
      "step": 75109
    },
    {
      "epoch": 1.0576248168945312e-06,
      "step": 75109,
      "training_step_time": 7.723431587219238
    },
    {
      "epoch": 1.0585784912109376e-06,
      "grad_norm": 0.07767841964960098,
      "learning_rate": 1.6003543498052127e-06,
      "loss": 0.0562,
      "step": 75110
    },
    {
      "epoch": 1.0585784912109376e-06,
      "model_forward_time": 1.74373197555542,
      "step": 75110
    },
    {
      "dynamics_loss_avg": 0.06033066902309656,
      "epoch": 1.0585784912109376e-06,
      "step": 75110
    },
    {
      "action_loss_avg": 0.0015125814883504064,
      "epoch": 1.0585784912109376e-06,
      "step": 75110
    },
    {
      "epoch": 1.0585784912109376e-06,
      "step": 75110,
      "training_step_time": 7.562849044799805
    },
    {
      "epoch": 1.0595321655273437e-06,
      "model_forward_time": 1.748227596282959,
      "step": 75111
    },
    {
      "epoch": 1.0595321655273437e-06,
      "step": 75111,
      "training_step_time": 7.646013259887695
    },
    {
      "epoch": 1.06048583984375e-06,
      "model_forward_time": 1.7358648777008057,
      "step": 75112
    },
    {
      "epoch": 1.06048583984375e-06,
      "step": 75112,
      "training_step_time": 8.274730205535889
    },
    {
      "epoch": 1.0614395141601562e-06,
      "model_forward_time": 1.7340748310089111,
      "step": 75113
    },
    {
      "epoch": 1.0614395141601562e-06,
      "step": 75113,
      "training_step_time": 7.872522354125977
    },
    {
      "epoch": 1.0623931884765626e-06,
      "model_forward_time": 1.7599296569824219,
      "step": 75114
    },
    {
      "epoch": 1.0623931884765626e-06,
      "step": 75114,
      "training_step_time": 8.433480978012085
    },
    {
      "epoch": 1.0633468627929687e-06,
      "model_forward_time": 1.3993422985076904,
      "step": 75115
    },
    {
      "epoch": 1.0633468627929687e-06,
      "step": 75115,
      "training_step_time": 7.819934844970703
    },
    {
      "epoch": 1.064300537109375e-06,
      "model_forward_time": 2.848074436187744,
      "step": 75116
    },
    {
      "epoch": 1.064300537109375e-06,
      "step": 75116,
      "training_step_time": 7.816889762878418
    },
    {
      "epoch": 1.0652542114257812e-06,
      "model_forward_time": 1.7454471588134766,
      "step": 75117
    },
    {
      "epoch": 1.0652542114257812e-06,
      "step": 75117,
      "training_step_time": 7.809642314910889
    },
    {
      "epoch": 1.0662078857421876e-06,
      "model_forward_time": 1.7398912906646729,
      "step": 75118
    },
    {
      "epoch": 1.0662078857421876e-06,
      "step": 75118,
      "training_step_time": 7.780502796173096
    },
    {
      "epoch": 1.0671615600585937e-06,
      "model_forward_time": 1.7427268028259277,
      "step": 75119
    },
    {
      "epoch": 1.0671615600585937e-06,
      "step": 75119,
      "training_step_time": 7.534387111663818
    },
    {
      "epoch": 1.068115234375e-06,
      "grad_norm": 0.08590659499168396,
      "learning_rate": 1.5991420829550714e-06,
      "loss": 0.056,
      "step": 75120
    },
    {
      "epoch": 1.068115234375e-06,
      "model_forward_time": 1.7438421249389648,
      "step": 75120
    },
    {
      "dynamics_loss_avg": 0.048332329466938974,
      "epoch": 1.068115234375e-06,
      "step": 75120
    },
    {
      "action_loss_avg": 0.0011859572201501577,
      "epoch": 1.068115234375e-06,
      "step": 75120
    },
    {
      "epoch": 1.068115234375e-06,
      "step": 75120,
      "training_step_time": 7.7419750690460205
    },
    {
      "epoch": 1.0690689086914062e-06,
      "model_forward_time": 1.7457411289215088,
      "step": 75121
    },
    {
      "epoch": 1.0690689086914062e-06,
      "step": 75121,
      "training_step_time": 7.586758136749268
    },
    {
      "epoch": 1.0700225830078126e-06,
      "model_forward_time": 1.7365095615386963,
      "step": 75122
    },
    {
      "epoch": 1.0700225830078126e-06,
      "step": 75122,
      "training_step_time": 7.778063058853149
    },
    {
      "epoch": 1.0709762573242187e-06,
      "model_forward_time": 1.7574658393859863,
      "step": 75123
    },
    {
      "epoch": 1.0709762573242187e-06,
      "step": 75123,
      "training_step_time": 9.120271444320679
    },
    {
      "epoch": 1.071929931640625e-06,
      "model_forward_time": 1.3863816261291504,
      "step": 75124
    },
    {
      "epoch": 1.071929931640625e-06,
      "step": 75124,
      "training_step_time": 7.914004802703857
    },
    {
      "epoch": 1.0728836059570312e-06,
      "model_forward_time": 2.817445755004883,
      "step": 75125
    },
    {
      "epoch": 1.0728836059570312e-06,
      "step": 75125,
      "training_step_time": 8.468369722366333
    },
    {
      "epoch": 1.0738372802734374e-06,
      "model_forward_time": 1.7470378875732422,
      "step": 75126
    },
    {
      "epoch": 1.0738372802734374e-06,
      "step": 75126,
      "training_step_time": 7.762986421585083
    },
    {
      "epoch": 1.0747909545898438e-06,
      "model_forward_time": 1.7427971363067627,
      "step": 75127
    },
    {
      "epoch": 1.0747909545898438e-06,
      "step": 75127,
      "training_step_time": 7.704903841018677
    },
    {
      "epoch": 1.07574462890625e-06,
      "model_forward_time": 1.742612600326538,
      "step": 75128
    },
    {
      "epoch": 1.07574462890625e-06,
      "step": 75128,
      "training_step_time": 8.487011194229126
    },
    {
      "epoch": 1.0766983032226563e-06,
      "model_forward_time": 1.7469046115875244,
      "step": 75129
    },
    {
      "epoch": 1.0766983032226563e-06,
      "step": 75129,
      "training_step_time": 8.927071332931519
    },
    {
      "epoch": 1.0776519775390624e-06,
      "grad_norm": 0.08062265068292618,
      "learning_rate": 1.5979301880176407e-06,
      "loss": 0.0579,
      "step": 75130
    },
    {
      "epoch": 1.0776519775390624e-06,
      "model_forward_time": 1.7346723079681396,
      "step": 75130
    },
    {
      "dynamics_loss_avg": 0.0699064951390028,
      "epoch": 1.0776519775390624e-06,
      "step": 75130
    },
    {
      "action_loss_avg": 0.0018527881940826774,
      "epoch": 1.0776519775390624e-06,
      "step": 75130
    },
    {
      "epoch": 1.0776519775390624e-06,
      "step": 75130,
      "training_step_time": 7.556048393249512
    },
    {
      "epoch": 1.0786056518554688e-06,
      "model_forward_time": 1.7360363006591797,
      "step": 75131
    },
    {
      "epoch": 1.0786056518554688e-06,
      "step": 75131,
      "training_step_time": 8.355929851531982
    },
    {
      "epoch": 1.079559326171875e-06,
      "model_forward_time": 1.7426176071166992,
      "step": 75132
    },
    {
      "epoch": 1.079559326171875e-06,
      "step": 75132,
      "training_step_time": 9.803688764572144
    },
    {
      "epoch": 1.0805130004882813e-06,
      "model_forward_time": 1.7534453868865967,
      "step": 75133
    },
    {
      "epoch": 1.0805130004882813e-06,
      "step": 75133,
      "training_step_time": 8.823434114456177
    },
    {
      "epoch": 1.0814666748046874e-06,
      "model_forward_time": 1.7341859340667725,
      "step": 75134
    },
    {
      "epoch": 1.0814666748046874e-06,
      "step": 75134,
      "training_step_time": 7.925796747207642
    },
    {
      "epoch": 1.0824203491210938e-06,
      "model_forward_time": 2.8763070106506348,
      "step": 75135
    },
    {
      "epoch": 1.0824203491210938e-06,
      "step": 75135,
      "training_step_time": 8.708708047866821
    },
    {
      "epoch": 1.0833740234375e-06,
      "model_forward_time": 1.7396154403686523,
      "step": 75136
    },
    {
      "epoch": 1.0833740234375e-06,
      "step": 75136,
      "training_step_time": 7.710968017578125
    },
    {
      "epoch": 1.0843276977539063e-06,
      "model_forward_time": 1.7424507141113281,
      "step": 75137
    },
    {
      "epoch": 1.0843276977539063e-06,
      "step": 75137,
      "training_step_time": 7.731529235839844
    },
    {
      "epoch": 1.0852813720703124e-06,
      "model_forward_time": 1.742488145828247,
      "step": 75138
    },
    {
      "epoch": 1.0852813720703124e-06,
      "step": 75138,
      "training_step_time": 7.7592995166778564
    },
    {
      "epoch": 1.0862350463867188e-06,
      "model_forward_time": 1.7414743900299072,
      "step": 75139
    },
    {
      "epoch": 1.0862350463867188e-06,
      "step": 75139,
      "training_step_time": 7.703280448913574
    },
    {
      "epoch": 1.087188720703125e-06,
      "grad_norm": 0.06191341206431389,
      "learning_rate": 1.5967186651254485e-06,
      "loss": 0.0551,
      "step": 75140
    },
    {
      "epoch": 1.087188720703125e-06,
      "model_forward_time": 1.7352790832519531,
      "step": 75140
    },
    {
      "dynamics_loss_avg": 0.06133315470069647,
      "epoch": 1.087188720703125e-06,
      "step": 75140
    },
    {
      "action_loss_avg": 0.002240313548827544,
      "epoch": 1.087188720703125e-06,
      "step": 75140
    },
    {
      "epoch": 1.087188720703125e-06,
      "step": 75140,
      "training_step_time": 8.927351713180542
    },
    {
      "epoch": 1.0881423950195313e-06,
      "model_forward_time": 1.73793625831604,
      "step": 75141
    },
    {
      "epoch": 1.0881423950195313e-06,
      "step": 75141,
      "training_step_time": 8.790466785430908
    },
    {
      "epoch": 1.0890960693359374e-06,
      "model_forward_time": 1.7753658294677734,
      "step": 75142
    },
    {
      "epoch": 1.0890960693359374e-06,
      "step": 75142,
      "training_step_time": 9.250213146209717
    },
    {
      "epoch": 1.0900497436523438e-06,
      "model_forward_time": 1.7392144203186035,
      "step": 75143
    },
    {
      "epoch": 1.0900497436523438e-06,
      "step": 75143,
      "training_step_time": 7.869203567504883
    },
    {
      "epoch": 1.09100341796875e-06,
      "model_forward_time": 2.8464739322662354,
      "step": 75144
    },
    {
      "epoch": 1.09100341796875e-06,
      "step": 75144,
      "training_step_time": 7.774280309677124
    },
    {
      "epoch": 1.0919570922851563e-06,
      "model_forward_time": 1.7390964031219482,
      "step": 75145
    },
    {
      "epoch": 1.0919570922851563e-06,
      "step": 75145,
      "training_step_time": 7.695152282714844
    },
    {
      "epoch": 1.0929107666015624e-06,
      "model_forward_time": 1.7328441143035889,
      "step": 75146
    },
    {
      "epoch": 1.0929107666015624e-06,
      "step": 75146,
      "training_step_time": 7.747934579849243
    },
    {
      "epoch": 1.0938644409179688e-06,
      "model_forward_time": 1.7447731494903564,
      "step": 75147
    },
    {
      "epoch": 1.0938644409179688e-06,
      "step": 75147,
      "training_step_time": 7.77295708656311
    },
    {
      "epoch": 1.094818115234375e-06,
      "model_forward_time": 1.7329685688018799,
      "step": 75148
    },
    {
      "epoch": 1.094818115234375e-06,
      "step": 75148,
      "training_step_time": 7.678439140319824
    },
    {
      "epoch": 1.0957717895507813e-06,
      "model_forward_time": 1.7373476028442383,
      "step": 75149
    },
    {
      "epoch": 1.0957717895507813e-06,
      "step": 75149,
      "training_step_time": 8.248181819915771
    },
    {
      "epoch": 1.0967254638671874e-06,
      "grad_norm": 0.1001359298825264,
      "learning_rate": 1.5955075144109854e-06,
      "loss": 0.0554,
      "step": 75150
    },
    {
      "epoch": 1.0967254638671874e-06,
      "model_forward_time": 1.7538042068481445,
      "step": 75150
    },
    {
      "dynamics_loss_avg": 0.03758047744631767,
      "epoch": 1.0967254638671874e-06,
      "step": 75150
    },
    {
      "action_loss_avg": 0.00148202782147564,
      "epoch": 1.0967254638671874e-06,
      "step": 75150
    },
    {
      "epoch": 1.0967254638671874e-06,
      "step": 75150,
      "training_step_time": 7.876628637313843
    },
    {
      "epoch": 1.0976791381835938e-06,
      "model_forward_time": 1.7495677471160889,
      "step": 75151
    },
    {
      "epoch": 1.0976791381835938e-06,
      "step": 75151,
      "training_step_time": 9.124595642089844
    },
    {
      "epoch": 1.0986328125e-06,
      "model_forward_time": 1.7499995231628418,
      "step": 75152
    },
    {
      "epoch": 1.0986328125e-06,
      "step": 75152,
      "training_step_time": 7.954903841018677
    },
    {
      "epoch": 1.0995864868164063e-06,
      "model_forward_time": 2.829284191131592,
      "step": 75153
    },
    {
      "epoch": 1.0995864868164063e-06,
      "step": 75153,
      "training_step_time": 8.290162086486816
    },
    {
      "epoch": 1.1005401611328124e-06,
      "model_forward_time": 1.7381267547607422,
      "step": 75154
    },
    {
      "epoch": 1.1005401611328124e-06,
      "step": 75154,
      "training_step_time": 8.796366453170776
    },
    {
      "epoch": 1.1014938354492188e-06,
      "model_forward_time": 1.7411394119262695,
      "step": 75155
    },
    {
      "epoch": 1.1014938354492188e-06,
      "step": 75155,
      "training_step_time": 7.750688791275024
    },
    {
      "epoch": 1.102447509765625e-06,
      "model_forward_time": 1.7325797080993652,
      "step": 75156
    },
    {
      "epoch": 1.102447509765625e-06,
      "step": 75156,
      "training_step_time": 7.692413568496704
    },
    {
      "epoch": 1.1034011840820313e-06,
      "model_forward_time": 1.7332720756530762,
      "step": 75157
    },
    {
      "epoch": 1.1034011840820313e-06,
      "step": 75157,
      "training_step_time": 7.569879770278931
    },
    {
      "epoch": 1.1043548583984374e-06,
      "model_forward_time": 1.740981101989746,
      "step": 75158
    },
    {
      "epoch": 1.1043548583984374e-06,
      "step": 75158,
      "training_step_time": 7.6361083984375
    },
    {
      "epoch": 1.1053085327148438e-06,
      "model_forward_time": 1.7349634170532227,
      "step": 75159
    },
    {
      "epoch": 1.1053085327148438e-06,
      "step": 75159,
      "training_step_time": 8.556435585021973
    },
    {
      "epoch": 1.10626220703125e-06,
      "grad_norm": 0.1127888560295105,
      "learning_rate": 1.594296736006703e-06,
      "loss": 0.0569,
      "step": 75160
    },
    {
      "epoch": 1.10626220703125e-06,
      "model_forward_time": 1.7372586727142334,
      "step": 75160
    },
    {
      "dynamics_loss_avg": 0.05702425483614206,
      "epoch": 1.10626220703125e-06,
      "step": 75160
    },
    {
      "action_loss_avg": 0.0019338880956638604,
      "epoch": 1.10626220703125e-06,
      "step": 75160
    },
    {
      "epoch": 1.10626220703125e-06,
      "step": 75160,
      "training_step_time": 8.910927772521973
    },
    {
      "epoch": 1.1072158813476563e-06,
      "model_forward_time": 1.7648036479949951,
      "step": 75161
    },
    {
      "epoch": 1.1072158813476563e-06,
      "step": 75161,
      "training_step_time": 8.867023468017578
    },
    {
      "epoch": 1.1081695556640624e-06,
      "model_forward_time": 2.8231325149536133,
      "step": 75162
    },
    {
      "epoch": 1.1081695556640624e-06,
      "step": 75162,
      "training_step_time": 9.363040685653687
    },
    {
      "epoch": 1.1091232299804688e-06,
      "model_forward_time": 1.7377583980560303,
      "step": 75163
    },
    {
      "epoch": 1.1091232299804688e-06,
      "step": 75163,
      "training_step_time": 7.78350305557251
    },
    {
      "epoch": 1.110076904296875e-06,
      "model_forward_time": 1.7474439144134521,
      "step": 75164
    },
    {
      "epoch": 1.110076904296875e-06,
      "step": 75164,
      "training_step_time": 7.7403295040130615
    },
    {
      "epoch": 1.1110305786132813e-06,
      "model_forward_time": 1.7334392070770264,
      "step": 75165
    },
    {
      "epoch": 1.1110305786132813e-06,
      "step": 75165,
      "training_step_time": 7.52630615234375
    },
    {
      "epoch": 1.1119842529296874e-06,
      "model_forward_time": 1.743734359741211,
      "step": 75166
    },
    {
      "epoch": 1.1119842529296874e-06,
      "step": 75166,
      "training_step_time": 7.767320394515991
    },
    {
      "epoch": 1.1129379272460938e-06,
      "model_forward_time": 1.7377245426177979,
      "step": 75167
    },
    {
      "epoch": 1.1129379272460938e-06,
      "step": 75167,
      "training_step_time": 7.766662359237671
    },
    {
      "epoch": 1.1138916015625e-06,
      "model_forward_time": 1.7354302406311035,
      "step": 75168
    },
    {
      "epoch": 1.1138916015625e-06,
      "step": 75168,
      "training_step_time": 7.902252912521362
    },
    {
      "epoch": 1.1148452758789063e-06,
      "model_forward_time": 1.7425525188446045,
      "step": 75169
    },
    {
      "epoch": 1.1148452758789063e-06,
      "step": 75169,
      "training_step_time": 7.902387857437134
    },
    {
      "epoch": 1.1157989501953124e-06,
      "grad_norm": 0.08332500606775284,
      "learning_rate": 1.59308633004501e-06,
      "loss": 0.0563,
      "step": 75170
    },
    {
      "epoch": 1.1157989501953124e-06,
      "model_forward_time": 1.8269615173339844,
      "step": 75170
    },
    {
      "dynamics_loss_avg": 0.048638013750314714,
      "epoch": 1.1157989501953124e-06,
      "step": 75170
    },
    {
      "action_loss_avg": 0.0022856324212625623,
      "epoch": 1.1157989501953124e-06,
      "step": 75170
    },
    {
      "epoch": 1.1157989501953124e-06,
      "step": 75170,
      "training_step_time": 7.936530828475952
    },
    {
      "epoch": 1.1167526245117188e-06,
      "model_forward_time": 2.798030138015747,
      "step": 75171
    },
    {
      "epoch": 1.1167526245117188e-06,
      "step": 75171,
      "training_step_time": 7.878510236740112
    },
    {
      "epoch": 1.117706298828125e-06,
      "model_forward_time": 1.3684420585632324,
      "step": 75172
    },
    {
      "epoch": 1.117706298828125e-06,
      "step": 75172,
      "training_step_time": 8.711815357208252
    },
    {
      "epoch": 1.1186599731445313e-06,
      "model_forward_time": 1.735640287399292,
      "step": 75173
    },
    {
      "epoch": 1.1186599731445313e-06,
      "step": 75173,
      "training_step_time": 7.787843704223633
    },
    {
      "epoch": 1.1196136474609374e-06,
      "model_forward_time": 1.7392299175262451,
      "step": 75174
    },
    {
      "epoch": 1.1196136474609374e-06,
      "step": 75174,
      "training_step_time": 7.655738592147827
    },
    {
      "epoch": 1.1205673217773438e-06,
      "model_forward_time": 1.7580697536468506,
      "step": 75175
    },
    {
      "epoch": 1.1205673217773438e-06,
      "step": 75175,
      "training_step_time": 7.698283433914185
    },
    {
      "epoch": 1.12152099609375e-06,
      "model_forward_time": 1.7402560710906982,
      "step": 75176
    },
    {
      "epoch": 1.12152099609375e-06,
      "step": 75176,
      "training_step_time": 7.536976099014282
    },
    {
      "epoch": 1.1224746704101563e-06,
      "model_forward_time": 1.7322602272033691,
      "step": 75177
    },
    {
      "epoch": 1.1224746704101563e-06,
      "step": 75177,
      "training_step_time": 7.8432817459106445
    },
    {
      "epoch": 1.1234283447265624e-06,
      "model_forward_time": 1.7410228252410889,
      "step": 75178
    },
    {
      "epoch": 1.1234283447265624e-06,
      "step": 75178,
      "training_step_time": 7.780808448791504
    },
    {
      "epoch": 1.1243820190429688e-06,
      "model_forward_time": 1.3838717937469482,
      "step": 75179
    },
    {
      "epoch": 1.1243820190429688e-06,
      "step": 75179,
      "training_step_time": 8.292365789413452
    },
    {
      "epoch": 1.125335693359375e-06,
      "grad_norm": 0.11405713856220245,
      "learning_rate": 1.5918762966582724e-06,
      "loss": 0.0552,
      "step": 75180
    },
    {
      "epoch": 1.125335693359375e-06,
      "model_forward_time": 1.7348952293395996,
      "step": 75180
    },
    {
      "dynamics_loss_avg": 0.04777029044926166,
      "epoch": 1.125335693359375e-06,
      "step": 75180
    },
    {
      "action_loss_avg": 0.0015565189998596907,
      "epoch": 1.125335693359375e-06,
      "step": 75180
    },
    {
      "epoch": 1.125335693359375e-06,
      "step": 75180,
      "training_step_time": 7.837945222854614
    },
    {
      "epoch": 1.1262893676757813e-06,
      "model_forward_time": 2.8882088661193848,
      "step": 75181
    },
    {
      "epoch": 1.1262893676757813e-06,
      "step": 75181,
      "training_step_time": 7.834786653518677
    },
    {
      "epoch": 1.1272430419921874e-06,
      "model_forward_time": 1.7406706809997559,
      "step": 75182
    },
    {
      "epoch": 1.1272430419921874e-06,
      "step": 75182,
      "training_step_time": 7.6919846534729
    },
    {
      "epoch": 1.1281967163085938e-06,
      "model_forward_time": 1.7456212043762207,
      "step": 75183
    },
    {
      "epoch": 1.1281967163085938e-06,
      "step": 75183,
      "training_step_time": 7.71971583366394
    },
    {
      "epoch": 1.129150390625e-06,
      "model_forward_time": 1.7401061058044434,
      "step": 75184
    },
    {
      "epoch": 1.129150390625e-06,
      "step": 75184,
      "training_step_time": 7.727156400680542
    },
    {
      "epoch": 1.1301040649414063e-06,
      "model_forward_time": 1.7470409870147705,
      "step": 75185
    },
    {
      "epoch": 1.1301040649414063e-06,
      "step": 75185,
      "training_step_time": 7.736422061920166
    },
    {
      "epoch": 1.1310577392578124e-06,
      "model_forward_time": 1.7527623176574707,
      "step": 75186
    },
    {
      "epoch": 1.1310577392578124e-06,
      "step": 75186,
      "training_step_time": 8.442855596542358
    },
    {
      "epoch": 1.1320114135742188e-06,
      "model_forward_time": 1.7416207790374756,
      "step": 75187
    },
    {
      "epoch": 1.1320114135742188e-06,
      "step": 75187,
      "training_step_time": 8.239065885543823
    },
    {
      "epoch": 1.132965087890625e-06,
      "model_forward_time": 1.7822177410125732,
      "step": 75188
    },
    {
      "epoch": 1.132965087890625e-06,
      "step": 75188,
      "training_step_time": 9.024478912353516
    },
    {
      "epoch": 1.1339187622070313e-06,
      "model_forward_time": 1.7430403232574463,
      "step": 75189
    },
    {
      "epoch": 1.1339187622070313e-06,
      "step": 75189,
      "training_step_time": 8.351413488388062
    },
    {
      "epoch": 1.1348724365234374e-06,
      "grad_norm": 0.12627655267715454,
      "learning_rate": 1.5906666359788203e-06,
      "loss": 0.0556,
      "step": 75190
    },
    {
      "epoch": 1.1348724365234374e-06,
      "model_forward_time": 2.840639352798462,
      "step": 75190
    },
    {
      "dynamics_loss_avg": 0.055026021972298625,
      "epoch": 1.1348724365234374e-06,
      "step": 75190
    },
    {
      "action_loss_avg": 0.000867633888265118,
      "epoch": 1.1348724365234374e-06,
      "step": 75190
    },
    {
      "epoch": 1.1348724365234374e-06,
      "step": 75190,
      "training_step_time": 7.7593162059783936
    },
    {
      "epoch": 1.1358261108398438e-06,
      "model_forward_time": 1.7341794967651367,
      "step": 75191
    },
    {
      "epoch": 1.1358261108398438e-06,
      "step": 75191,
      "training_step_time": 8.423698663711548
    },
    {
      "epoch": 1.13677978515625e-06,
      "model_forward_time": 1.7446284294128418,
      "step": 75192
    },
    {
      "epoch": 1.13677978515625e-06,
      "step": 75192,
      "training_step_time": 7.7671959400177
    },
    {
      "epoch": 1.1377334594726563e-06,
      "model_forward_time": 1.743884801864624,
      "step": 75193
    },
    {
      "epoch": 1.1377334594726563e-06,
      "step": 75193,
      "training_step_time": 7.684564113616943
    },
    {
      "epoch": 1.1386871337890625e-06,
      "model_forward_time": 1.7396183013916016,
      "step": 75194
    },
    {
      "epoch": 1.1386871337890625e-06,
      "step": 75194,
      "training_step_time": 7.688462734222412
    },
    {
      "epoch": 1.1396408081054688e-06,
      "model_forward_time": 1.7396507263183594,
      "step": 75195
    },
    {
      "epoch": 1.1396408081054688e-06,
      "step": 75195,
      "training_step_time": 8.176147222518921
    },
    {
      "epoch": 1.140594482421875e-06,
      "model_forward_time": 1.737083911895752,
      "step": 75196
    },
    {
      "epoch": 1.140594482421875e-06,
      "step": 75196,
      "training_step_time": 8.275263547897339
    },
    {
      "epoch": 1.1415481567382813e-06,
      "model_forward_time": 1.7412559986114502,
      "step": 75197
    },
    {
      "epoch": 1.1415481567382813e-06,
      "step": 75197,
      "training_step_time": 9.955480098724365
    },
    {
      "epoch": 1.1425018310546875e-06,
      "model_forward_time": 1.3959510326385498,
      "step": 75198
    },
    {
      "epoch": 1.1425018310546875e-06,
      "step": 75198,
      "training_step_time": 8.324133157730103
    },
    {
      "epoch": 1.1434555053710938e-06,
      "model_forward_time": 2.7739884853363037,
      "step": 75199
    },
    {
      "epoch": 1.1434555053710938e-06,
      "step": 75199,
      "training_step_time": 8.412612676620483
    },
    {
      "epoch": 1.1444091796875e-06,
      "grad_norm": 0.08826008439064026,
      "learning_rate": 1.589457348138938e-06,
      "loss": 0.0565,
      "step": 75200
    },
    {
      "epoch": 1.1444091796875e-06,
      "model_forward_time": 1.7358431816101074,
      "step": 75200
    },
    {
      "dynamics_loss_avg": 0.04799574818462134,
      "epoch": 1.1444091796875e-06,
      "step": 75200
    },
    {
      "action_loss_avg": 0.0008920579450204969,
      "epoch": 1.1444091796875e-06,
      "step": 75200
    },
    {
      "epoch": 1.1444091796875e-06,
      "step": 75200,
      "training_step_time": 7.545938014984131
    },
    {
      "epoch": 1.1453628540039063e-06,
      "model_forward_time": 1.7640092372894287,
      "step": 75201
    },
    {
      "epoch": 1.1453628540039063e-06,
      "step": 75201,
      "training_step_time": 7.638857841491699
    },
    {
      "epoch": 1.1463165283203125e-06,
      "model_forward_time": 1.724565029144287,
      "step": 75202
    },
    {
      "epoch": 1.1463165283203125e-06,
      "step": 75202,
      "training_step_time": 7.574870347976685
    },
    {
      "epoch": 1.1472702026367188e-06,
      "model_forward_time": 1.72361421585083,
      "step": 75203
    },
    {
      "epoch": 1.1472702026367188e-06,
      "step": 75203,
      "training_step_time": 8.70283842086792
    },
    {
      "epoch": 1.148223876953125e-06,
      "model_forward_time": 1.7512543201446533,
      "step": 75204
    },
    {
      "epoch": 1.148223876953125e-06,
      "step": 75204,
      "training_step_time": 7.528492450714111
    },
    {
      "epoch": 1.1491775512695313e-06,
      "model_forward_time": 1.7246613502502441,
      "step": 75205
    },
    {
      "epoch": 1.1491775512695313e-06,
      "step": 75205,
      "training_step_time": 7.852555274963379
    },
    {
      "epoch": 1.1501312255859375e-06,
      "model_forward_time": 1.740825891494751,
      "step": 75206
    },
    {
      "epoch": 1.1501312255859375e-06,
      "step": 75206,
      "training_step_time": 8.468910217285156
    },
    {
      "epoch": 1.1510848999023438e-06,
      "model_forward_time": 1.8235931396484375,
      "step": 75207
    },
    {
      "epoch": 1.1510848999023438e-06,
      "step": 75207,
      "training_step_time": 7.833263158798218
    },
    {
      "epoch": 1.15203857421875e-06,
      "model_forward_time": 1.7431485652923584,
      "step": 75208
    },
    {
      "epoch": 1.15203857421875e-06,
      "step": 75208,
      "training_step_time": 8.316044807434082
    },
    {
      "epoch": 1.1529922485351563e-06,
      "model_forward_time": 2.821462392807007,
      "step": 75209
    },
    {
      "epoch": 1.1529922485351563e-06,
      "step": 75209,
      "training_step_time": 7.7027812004089355
    },
    {
      "epoch": 1.1539459228515625e-06,
      "grad_norm": 0.07271457463502884,
      "learning_rate": 1.588248433270872e-06,
      "loss": 0.056,
      "step": 75210
    },
    {
      "epoch": 1.1539459228515625e-06,
      "model_forward_time": 1.7394487857818604,
      "step": 75210
    },
    {
      "dynamics_loss_avg": 0.04456307403743267,
      "epoch": 1.1539459228515625e-06,
      "step": 75210
    },
    {
      "action_loss_avg": 0.0008880715118721128,
      "epoch": 1.1539459228515625e-06,
      "step": 75210
    },
    {
      "epoch": 1.1539459228515625e-06,
      "step": 75210,
      "training_step_time": 8.525628328323364
    },
    {
      "epoch": 1.1548995971679688e-06,
      "model_forward_time": 1.7364952564239502,
      "step": 75211
    },
    {
      "epoch": 1.1548995971679688e-06,
      "step": 75211,
      "training_step_time": 7.708738088607788
    },
    {
      "epoch": 1.155853271484375e-06,
      "model_forward_time": 1.738816499710083,
      "step": 75212
    },
    {
      "epoch": 1.155853271484375e-06,
      "step": 75212,
      "training_step_time": 7.674025535583496
    },
    {
      "epoch": 1.1568069458007813e-06,
      "model_forward_time": 1.7460966110229492,
      "step": 75213
    },
    {
      "epoch": 1.1568069458007813e-06,
      "step": 75213,
      "training_step_time": 8.409372329711914
    },
    {
      "epoch": 1.1577606201171875e-06,
      "model_forward_time": 1.363691806793213,
      "step": 75214
    },
    {
      "epoch": 1.1577606201171875e-06,
      "step": 75214,
      "training_step_time": 8.839055299758911
    },
    {
      "epoch": 1.1587142944335938e-06,
      "model_forward_time": 1.7473154067993164,
      "step": 75215
    },
    {
      "epoch": 1.1587142944335938e-06,
      "step": 75215,
      "training_step_time": 7.773121118545532
    },
    {
      "epoch": 1.15966796875e-06,
      "model_forward_time": 1.8258907794952393,
      "step": 75216
    },
    {
      "epoch": 1.15966796875e-06,
      "step": 75216,
      "training_step_time": 7.768709659576416
    },
    {
      "epoch": 1.1606216430664063e-06,
      "model_forward_time": 1.7443408966064453,
      "step": 75217
    },
    {
      "epoch": 1.1606216430664063e-06,
      "step": 75217,
      "training_step_time": 9.32611608505249
    },
    {
      "epoch": 1.1615753173828125e-06,
      "model_forward_time": 2.8742501735687256,
      "step": 75218
    },
    {
      "epoch": 1.1615753173828125e-06,
      "step": 75218,
      "training_step_time": 7.73891544342041
    },
    {
      "epoch": 1.1625289916992188e-06,
      "model_forward_time": 1.7383277416229248,
      "step": 75219
    },
    {
      "epoch": 1.1625289916992188e-06,
      "step": 75219,
      "training_step_time": 7.671255826950073
    },
    {
      "epoch": 1.163482666015625e-06,
      "grad_norm": 0.091863252222538,
      "learning_rate": 1.587039891506828e-06,
      "loss": 0.0557,
      "step": 75220
    },
    {
      "epoch": 1.163482666015625e-06,
      "model_forward_time": 1.7423372268676758,
      "step": 75220
    },
    {
      "dynamics_loss_avg": 0.058429947309196,
      "epoch": 1.163482666015625e-06,
      "step": 75220
    },
    {
      "action_loss_avg": 0.00119234450685326,
      "epoch": 1.163482666015625e-06,
      "step": 75220
    },
    {
      "epoch": 1.163482666015625e-06,
      "step": 75220,
      "training_step_time": 7.764755010604858
    },
    {
      "epoch": 1.1644363403320313e-06,
      "model_forward_time": 1.7409236431121826,
      "step": 75221
    },
    {
      "epoch": 1.1644363403320313e-06,
      "step": 75221,
      "training_step_time": 8.422260761260986
    },
    {
      "epoch": 1.1653900146484375e-06,
      "model_forward_time": 1.742518663406372,
      "step": 75222
    },
    {
      "epoch": 1.1653900146484375e-06,
      "step": 75222,
      "training_step_time": 7.691134452819824
    },
    {
      "epoch": 1.1663436889648438e-06,
      "model_forward_time": 1.7364709377288818,
      "step": 75223
    },
    {
      "epoch": 1.1663436889648438e-06,
      "step": 75223,
      "training_step_time": 9.685685634613037
    },
    {
      "epoch": 1.16729736328125e-06,
      "model_forward_time": 1.7431654930114746,
      "step": 75224
    },
    {
      "epoch": 1.16729736328125e-06,
      "step": 75224,
      "training_step_time": 7.729562759399414
    },
    {
      "epoch": 1.1682510375976563e-06,
      "model_forward_time": 1.8021435737609863,
      "step": 75225
    },
    {
      "epoch": 1.1682510375976563e-06,
      "step": 75225,
      "training_step_time": 9.576557636260986
    },
    {
      "epoch": 1.1692047119140625e-06,
      "model_forward_time": 1.7569327354431152,
      "step": 75226
    },
    {
      "epoch": 1.1692047119140625e-06,
      "step": 75226,
      "training_step_time": 7.752021074295044
    },
    {
      "epoch": 1.1701583862304688e-06,
      "model_forward_time": 2.8401458263397217,
      "step": 75227
    },
    {
      "epoch": 1.1701583862304688e-06,
      "step": 75227,
      "training_step_time": 7.819811105728149
    },
    {
      "epoch": 1.171112060546875e-06,
      "model_forward_time": 1.7415666580200195,
      "step": 75228
    },
    {
      "epoch": 1.171112060546875e-06,
      "step": 75228,
      "training_step_time": 7.703203439712524
    },
    {
      "epoch": 1.1720657348632813e-06,
      "model_forward_time": 1.7528398036956787,
      "step": 75229
    },
    {
      "epoch": 1.1720657348632813e-06,
      "step": 75229,
      "training_step_time": 7.347675085067749
    },
    {
      "epoch": 1.1730194091796875e-06,
      "grad_norm": 0.08334074169397354,
      "learning_rate": 1.5858317229789717e-06,
      "loss": 0.0563,
      "step": 75230
    },
    {
      "epoch": 1.1730194091796875e-06,
      "model_forward_time": 1.739978551864624,
      "step": 75230
    },
    {
      "dynamics_loss_avg": 0.048902394343167545,
      "epoch": 1.1730194091796875e-06,
      "step": 75230
    },
    {
      "action_loss_avg": 0.0009537162899505347,
      "epoch": 1.1730194091796875e-06,
      "step": 75230
    },
    {
      "epoch": 1.1730194091796875e-06,
      "step": 75230,
      "training_step_time": 7.570291757583618
    },
    {
      "epoch": 1.1739730834960938e-06,
      "model_forward_time": 1.7341055870056152,
      "step": 75231
    },
    {
      "epoch": 1.1739730834960938e-06,
      "step": 75231,
      "training_step_time": 8.162729263305664
    },
    {
      "epoch": 1.1749267578125e-06,
      "model_forward_time": 1.7406370639801025,
      "step": 75232
    },
    {
      "epoch": 1.1749267578125e-06,
      "step": 75232,
      "training_step_time": 7.751680850982666
    },
    {
      "epoch": 1.1758804321289063e-06,
      "model_forward_time": 1.7516801357269287,
      "step": 75233
    },
    {
      "epoch": 1.1758804321289063e-06,
      "step": 75233,
      "training_step_time": 9.257049798965454
    },
    {
      "epoch": 1.1768341064453125e-06,
      "model_forward_time": 1.753441572189331,
      "step": 75234
    },
    {
      "epoch": 1.1768341064453125e-06,
      "step": 75234,
      "training_step_time": 9.216548442840576
    },
    {
      "epoch": 1.1777877807617188e-06,
      "model_forward_time": 1.7547857761383057,
      "step": 75235
    },
    {
      "epoch": 1.1777877807617188e-06,
      "step": 75235,
      "training_step_time": 8.518192052841187
    },
    {
      "epoch": 1.178741455078125e-06,
      "model_forward_time": 2.8260486125946045,
      "step": 75236
    },
    {
      "epoch": 1.178741455078125e-06,
      "step": 75236,
      "training_step_time": 7.823822498321533
    },
    {
      "epoch": 1.1796951293945313e-06,
      "model_forward_time": 1.7427637577056885,
      "step": 75237
    },
    {
      "epoch": 1.1796951293945313e-06,
      "step": 75237,
      "training_step_time": 8.531612396240234
    },
    {
      "epoch": 1.1806488037109375e-06,
      "model_forward_time": 1.7442421913146973,
      "step": 75238
    },
    {
      "epoch": 1.1806488037109375e-06,
      "step": 75238,
      "training_step_time": 7.426543951034546
    },
    {
      "epoch": 1.1816024780273438e-06,
      "model_forward_time": 1.7373697757720947,
      "step": 75239
    },
    {
      "epoch": 1.1816024780273438e-06,
      "step": 75239,
      "training_step_time": 8.677788257598877
    },
    {
      "epoch": 1.18255615234375e-06,
      "grad_norm": 0.10517356544733047,
      "learning_rate": 1.5846239278194232e-06,
      "loss": 0.0557,
      "step": 75240
    },
    {
      "epoch": 1.18255615234375e-06,
      "model_forward_time": 1.7557506561279297,
      "step": 75240
    },
    {
      "dynamics_loss_avg": 0.04504483956843615,
      "epoch": 1.18255615234375e-06,
      "step": 75240
    },
    {
      "action_loss_avg": 0.0007390267739538104,
      "epoch": 1.18255615234375e-06,
      "step": 75240
    },
    {
      "epoch": 1.18255615234375e-06,
      "step": 75240,
      "training_step_time": 8.71674633026123
    },
    {
      "epoch": 1.1835098266601563e-06,
      "model_forward_time": 1.7357008457183838,
      "step": 75241
    },
    {
      "epoch": 1.1835098266601563e-06,
      "step": 75241,
      "training_step_time": 7.814680576324463
    },
    {
      "epoch": 1.1844635009765625e-06,
      "model_forward_time": 1.7421338558197021,
      "step": 75242
    },
    {
      "epoch": 1.1844635009765625e-06,
      "step": 75242,
      "training_step_time": 7.772722244262695
    },
    {
      "epoch": 1.1854171752929688e-06,
      "model_forward_time": 1.7421014308929443,
      "step": 75243
    },
    {
      "epoch": 1.1854171752929688e-06,
      "step": 75243,
      "training_step_time": 8.367100477218628
    },
    {
      "epoch": 1.186370849609375e-06,
      "model_forward_time": 1.4571871757507324,
      "step": 75244
    },
    {
      "epoch": 1.186370849609375e-06,
      "step": 75244,
      "training_step_time": 7.778048515319824
    },
    {
      "epoch": 1.1873245239257814e-06,
      "model_forward_time": 2.793764114379883,
      "step": 75245
    },
    {
      "epoch": 1.1873245239257814e-06,
      "step": 75245,
      "training_step_time": 7.744412422180176
    },
    {
      "epoch": 1.1882781982421875e-06,
      "model_forward_time": 1.7381141185760498,
      "step": 75246
    },
    {
      "epoch": 1.1882781982421875e-06,
      "step": 75246,
      "training_step_time": 7.691933870315552
    },
    {
      "epoch": 1.1892318725585939e-06,
      "model_forward_time": 1.731537103652954,
      "step": 75247
    },
    {
      "epoch": 1.1892318725585939e-06,
      "step": 75247,
      "training_step_time": 7.693540334701538
    },
    {
      "epoch": 1.190185546875e-06,
      "model_forward_time": 1.7285397052764893,
      "step": 75248
    },
    {
      "epoch": 1.190185546875e-06,
      "step": 75248,
      "training_step_time": 7.744081735610962
    },
    {
      "epoch": 1.1911392211914064e-06,
      "model_forward_time": 1.7335996627807617,
      "step": 75249
    },
    {
      "epoch": 1.1911392211914064e-06,
      "step": 75249,
      "training_step_time": 7.746994972229004
    },
    {
      "epoch": 1.1920928955078125e-06,
      "grad_norm": 0.09621651470661163,
      "learning_rate": 1.5834165061602691e-06,
      "loss": 0.057,
      "step": 75250
    },
    {
      "epoch": 1.1920928955078125e-06,
      "model_forward_time": 1.723712682723999,
      "step": 75250
    },
    {
      "dynamics_loss_avg": 0.052562884986400604,
      "epoch": 1.1920928955078125e-06,
      "step": 75250
    },
    {
      "action_loss_avg": 0.0010645174043020234,
      "epoch": 1.1920928955078125e-06,
      "step": 75250
    },
    {
      "epoch": 1.1920928955078125e-06,
      "step": 75250,
      "training_step_time": 7.952063083648682
    },
    {
      "epoch": 1.1930465698242186e-06,
      "model_forward_time": 1.7265305519104004,
      "step": 75251
    },
    {
      "epoch": 1.1930465698242186e-06,
      "step": 75251,
      "training_step_time": 9.178743124008179
    },
    {
      "epoch": 1.194000244140625e-06,
      "model_forward_time": 1.7324743270874023,
      "step": 75252
    },
    {
      "epoch": 1.194000244140625e-06,
      "step": 75252,
      "training_step_time": 9.301552772521973
    },
    {
      "epoch": 1.1949539184570311e-06,
      "model_forward_time": 1.3917479515075684,
      "step": 75253
    },
    {
      "epoch": 1.1949539184570311e-06,
      "step": 75253,
      "training_step_time": 8.362889528274536
    },
    {
      "epoch": 1.1959075927734375e-06,
      "model_forward_time": 1.7367117404937744,
      "step": 75254
    },
    {
      "epoch": 1.1959075927734375e-06,
      "step": 75254,
      "training_step_time": 8.559507608413696
    },
    {
      "epoch": 1.1968612670898436e-06,
      "model_forward_time": 2.8247029781341553,
      "step": 75255
    },
    {
      "epoch": 1.1968612670898436e-06,
      "step": 75255,
      "training_step_time": 8.455260515213013
    },
    {
      "epoch": 1.19781494140625e-06,
      "model_forward_time": 1.7312676906585693,
      "step": 75256
    },
    {
      "epoch": 1.19781494140625e-06,
      "step": 75256,
      "training_step_time": 7.737082004547119
    },
    {
      "epoch": 1.1987686157226562e-06,
      "model_forward_time": 1.7375388145446777,
      "step": 75257
    },
    {
      "epoch": 1.1987686157226562e-06,
      "step": 75257,
      "training_step_time": 8.020163536071777
    },
    {
      "epoch": 1.1997222900390625e-06,
      "model_forward_time": 1.7290503978729248,
      "step": 75258
    },
    {
      "epoch": 1.1997222900390625e-06,
      "step": 75258,
      "training_step_time": 7.650548458099365
    },
    {
      "epoch": 1.2006759643554687e-06,
      "model_forward_time": 1.7248196601867676,
      "step": 75259
    },
    {
      "epoch": 1.2006759643554687e-06,
      "step": 75259,
      "training_step_time": 8.44068169593811
    },
    {
      "epoch": 1.201629638671875e-06,
      "grad_norm": 0.0945136770606041,
      "learning_rate": 1.5822094581335473e-06,
      "loss": 0.0553,
      "step": 75260
    },
    {
      "epoch": 1.201629638671875e-06,
      "model_forward_time": 1.7330055236816406,
      "step": 75260
    },
    {
      "dynamics_loss_avg": 0.055982212722301486,
      "epoch": 1.201629638671875e-06,
      "step": 75260
    },
    {
      "action_loss_avg": 0.0010628387331962585,
      "epoch": 1.201629638671875e-06,
      "step": 75260
    },
    {
      "epoch": 1.201629638671875e-06,
      "step": 75260,
      "training_step_time": 8.492737531661987
    },
    {
      "epoch": 1.2025833129882812e-06,
      "model_forward_time": 1.725144863128662,
      "step": 75261
    },
    {
      "epoch": 1.2025833129882812e-06,
      "step": 75261,
      "training_step_time": 9.37342119216919
    },
    {
      "epoch": 1.2035369873046875e-06,
      "model_forward_time": 1.7786617279052734,
      "step": 75262
    },
    {
      "epoch": 1.2035369873046875e-06,
      "step": 75262,
      "training_step_time": 9.451726913452148
    },
    {
      "epoch": 1.2044906616210937e-06,
      "model_forward_time": 1.7278459072113037,
      "step": 75263
    },
    {
      "epoch": 1.2044906616210937e-06,
      "step": 75263,
      "training_step_time": 7.777391672134399
    },
    {
      "epoch": 1.2054443359375e-06,
      "model_forward_time": 2.826716661453247,
      "step": 75264
    },
    {
      "epoch": 1.2054443359375e-06,
      "step": 75264,
      "training_step_time": 8.577042579650879
    },
    {
      "epoch": 1.2063980102539062e-06,
      "model_forward_time": 1.7249650955200195,
      "step": 75265
    },
    {
      "epoch": 1.2063980102539062e-06,
      "step": 75265,
      "training_step_time": 7.780775785446167
    },
    {
      "epoch": 1.2073516845703125e-06,
      "model_forward_time": 1.7227027416229248,
      "step": 75266
    },
    {
      "epoch": 1.2073516845703125e-06,
      "step": 75266,
      "training_step_time": 7.610761404037476
    },
    {
      "epoch": 1.2083053588867187e-06,
      "model_forward_time": 1.7384090423583984,
      "step": 75267
    },
    {
      "epoch": 1.2083053588867187e-06,
      "step": 75267,
      "training_step_time": 8.892914295196533
    },
    {
      "epoch": 1.209259033203125e-06,
      "model_forward_time": 1.7233960628509521,
      "step": 75268
    },
    {
      "epoch": 1.209259033203125e-06,
      "step": 75268,
      "training_step_time": 7.887156248092651
    },
    {
      "epoch": 1.2102127075195312e-06,
      "model_forward_time": 1.7336020469665527,
      "step": 75269
    },
    {
      "epoch": 1.2102127075195312e-06,
      "step": 75269,
      "training_step_time": 7.82179856300354
    },
    {
      "epoch": 1.2111663818359375e-06,
      "grad_norm": 0.10291004180908203,
      "learning_rate": 1.5810027838712604e-06,
      "loss": 0.0545,
      "step": 75270
    },
    {
      "epoch": 1.2111663818359375e-06,
      "model_forward_time": 1.7233555316925049,
      "step": 75270
    },
    {
      "dynamics_loss_avg": 0.051092969998717305,
      "epoch": 1.2111663818359375e-06,
      "step": 75270
    },
    {
      "action_loss_avg": 0.001231519499560818,
      "epoch": 1.2111663818359375e-06,
      "step": 75270
    },
    {
      "epoch": 1.2111663818359375e-06,
      "step": 75270,
      "training_step_time": 8.599886894226074
    },
    {
      "epoch": 1.2121200561523437e-06,
      "model_forward_time": 1.7341420650482178,
      "step": 75271
    },
    {
      "epoch": 1.2121200561523437e-06,
      "step": 75271,
      "training_step_time": 8.419297218322754
    },
    {
      "epoch": 1.21307373046875e-06,
      "model_forward_time": 1.7592058181762695,
      "step": 75272
    },
    {
      "epoch": 1.21307373046875e-06,
      "step": 75272,
      "training_step_time": 7.8149309158325195
    },
    {
      "epoch": 1.2140274047851562e-06,
      "model_forward_time": 2.8021047115325928,
      "step": 75273
    },
    {
      "epoch": 1.2140274047851562e-06,
      "step": 75273,
      "training_step_time": 7.834394454956055
    },
    {
      "epoch": 1.2149810791015625e-06,
      "model_forward_time": 1.7252564430236816,
      "step": 75274
    },
    {
      "epoch": 1.2149810791015625e-06,
      "step": 75274,
      "training_step_time": 7.628839492797852
    },
    {
      "epoch": 1.2159347534179687e-06,
      "model_forward_time": 1.7407207489013672,
      "step": 75275
    },
    {
      "epoch": 1.2159347534179687e-06,
      "step": 75275,
      "training_step_time": 8.18781042098999
    },
    {
      "epoch": 1.216888427734375e-06,
      "model_forward_time": 1.7311666011810303,
      "step": 75276
    },
    {
      "epoch": 1.216888427734375e-06,
      "step": 75276,
      "training_step_time": 7.775230646133423
    },
    {
      "epoch": 1.2178421020507812e-06,
      "model_forward_time": 1.7225329875946045,
      "step": 75277
    },
    {
      "epoch": 1.2178421020507812e-06,
      "step": 75277,
      "training_step_time": 7.734943628311157
    },
    {
      "epoch": 1.2187957763671875e-06,
      "model_forward_time": 1.7213609218597412,
      "step": 75278
    },
    {
      "epoch": 1.2187957763671875e-06,
      "step": 75278,
      "training_step_time": 8.47320294380188
    },
    {
      "epoch": 1.2197494506835937e-06,
      "model_forward_time": 1.7195773124694824,
      "step": 75279
    },
    {
      "epoch": 1.2197494506835937e-06,
      "step": 75279,
      "training_step_time": 7.832015037536621
    },
    {
      "epoch": 1.220703125e-06,
      "grad_norm": 0.07589025050401688,
      "learning_rate": 1.5797964835053686e-06,
      "loss": 0.0552,
      "step": 75280
    },
    {
      "epoch": 1.220703125e-06,
      "model_forward_time": 1.7229361534118652,
      "step": 75280
    },
    {
      "dynamics_loss_avg": 0.05018764771521091,
      "epoch": 1.220703125e-06,
      "step": 75280
    },
    {
      "action_loss_avg": 0.001136380067327991,
      "epoch": 1.220703125e-06,
      "step": 75280
    },
    {
      "epoch": 1.220703125e-06,
      "step": 75280,
      "training_step_time": 8.320040941238403
    },
    {
      "epoch": 1.2216567993164062e-06,
      "model_forward_time": 1.805509328842163,
      "step": 75281
    },
    {
      "epoch": 1.2216567993164062e-06,
      "step": 75281,
      "training_step_time": 8.384690284729004
    },
    {
      "epoch": 1.2226104736328125e-06,
      "model_forward_time": 2.7782466411590576,
      "step": 75282
    },
    {
      "epoch": 1.2226104736328125e-06,
      "step": 75282,
      "training_step_time": 8.836873531341553
    },
    {
      "epoch": 1.2235641479492187e-06,
      "model_forward_time": 1.7154459953308105,
      "step": 75283
    },
    {
      "epoch": 1.2235641479492187e-06,
      "step": 75283,
      "training_step_time": 7.657189607620239
    },
    {
      "epoch": 1.224517822265625e-06,
      "model_forward_time": 1.7169740200042725,
      "step": 75284
    },
    {
      "epoch": 1.224517822265625e-06,
      "step": 75284,
      "training_step_time": 7.3854827880859375
    },
    {
      "epoch": 1.2254714965820312e-06,
      "model_forward_time": 1.7197275161743164,
      "step": 75285
    },
    {
      "epoch": 1.2254714965820312e-06,
      "step": 75285,
      "training_step_time": 7.626253128051758
    },
    {
      "epoch": 1.2264251708984375e-06,
      "model_forward_time": 1.7130637168884277,
      "step": 75286
    },
    {
      "epoch": 1.2264251708984375e-06,
      "step": 75286,
      "training_step_time": 7.784581422805786
    },
    {
      "epoch": 1.2273788452148437e-06,
      "model_forward_time": 1.7213234901428223,
      "step": 75287
    },
    {
      "epoch": 1.2273788452148437e-06,
      "step": 75287,
      "training_step_time": 7.628365516662598
    },
    {
      "epoch": 1.22833251953125e-06,
      "model_forward_time": 1.71779465675354,
      "step": 75288
    },
    {
      "epoch": 1.22833251953125e-06,
      "step": 75288,
      "training_step_time": 8.355916023254395
    },
    {
      "epoch": 1.2292861938476562e-06,
      "model_forward_time": 1.7156014442443848,
      "step": 75289
    },
    {
      "epoch": 1.2292861938476562e-06,
      "step": 75289,
      "training_step_time": 8.77720308303833
    },
    {
      "epoch": 1.2302398681640625e-06,
      "grad_norm": 0.09831216186285019,
      "learning_rate": 1.5785905571677934e-06,
      "loss": 0.0545,
      "step": 75290
    },
    {
      "epoch": 1.2302398681640625e-06,
      "model_forward_time": 1.8076581954956055,
      "step": 75290
    },
    {
      "dynamics_loss_avg": 0.057333352603018285,
      "epoch": 1.2302398681640625e-06,
      "step": 75290
    },
    {
      "action_loss_avg": 0.0017649898945819587,
      "epoch": 1.2302398681640625e-06,
      "step": 75290
    },
    {
      "epoch": 1.2302398681640625e-06,
      "step": 75290,
      "training_step_time": 8.231774091720581
    },
    {
      "epoch": 1.2311935424804687e-06,
      "model_forward_time": 1.7183940410614014,
      "step": 75291
    },
    {
      "epoch": 1.2311935424804687e-06,
      "step": 75291,
      "training_step_time": 8.633999347686768
    },
    {
      "epoch": 1.232147216796875e-06,
      "model_forward_time": 1.7196121215820312,
      "step": 75292
    },
    {
      "epoch": 1.232147216796875e-06,
      "step": 75292,
      "training_step_time": 8.508993864059448
    },
    {
      "epoch": 1.2331008911132812e-06,
      "model_forward_time": 1.7118315696716309,
      "step": 75293
    },
    {
      "epoch": 1.2331008911132812e-06,
      "step": 75293,
      "training_step_time": 7.338341236114502
    },
    {
      "epoch": 1.2340545654296875e-06,
      "model_forward_time": 1.7292327880859375,
      "step": 75294
    },
    {
      "epoch": 1.2340545654296875e-06,
      "step": 75294,
      "training_step_time": 8.428519487380981
    },
    {
      "epoch": 1.2350082397460937e-06,
      "model_forward_time": 1.7253446578979492,
      "step": 75295
    },
    {
      "epoch": 1.2350082397460937e-06,
      "step": 75295,
      "training_step_time": 7.820156574249268
    },
    {
      "epoch": 1.2359619140625e-06,
      "model_forward_time": 1.724656581878662,
      "step": 75296
    },
    {
      "epoch": 1.2359619140625e-06,
      "step": 75296,
      "training_step_time": 8.290298700332642
    },
    {
      "epoch": 1.2369155883789062e-06,
      "model_forward_time": 1.710529088973999,
      "step": 75297
    },
    {
      "epoch": 1.2369155883789062e-06,
      "step": 75297,
      "training_step_time": 7.718772649765015
    },
    {
      "epoch": 1.2378692626953125e-06,
      "model_forward_time": 1.726773738861084,
      "step": 75298
    },
    {
      "epoch": 1.2378692626953125e-06,
      "step": 75298,
      "training_step_time": 8.203032970428467
    },
    {
      "epoch": 1.2388229370117187e-06,
      "model_forward_time": 1.769890308380127,
      "step": 75299
    },
    {
      "epoch": 1.2388229370117187e-06,
      "step": 75299,
      "training_step_time": 9.06232237815857
    },
    {
      "epoch": 1.239776611328125e-06,
      "grad_norm": 0.0877816379070282,
      "learning_rate": 1.5773850049904089e-06,
      "loss": 0.0546,
      "step": 75300
    },
    {
      "epoch": 1.239776611328125e-06,
      "model_forward_time": 1.7253899574279785,
      "step": 75300
    },
    {
      "dynamics_loss_avg": 0.057529675960540774,
      "epoch": 1.239776611328125e-06,
      "step": 75300
    },
    {
      "action_loss_avg": 0.0015163865842623635,
      "epoch": 1.239776611328125e-06,
      "step": 75300
    },
    {
      "epoch": 1.239776611328125e-06,
      "step": 75300,
      "training_step_time": 10.132008075714111
    },
    {
      "epoch": 1.2407302856445312e-06,
      "model_forward_time": 2.858640670776367,
      "step": 75301
    },
    {
      "epoch": 1.2407302856445312e-06,
      "step": 75301,
      "training_step_time": 7.749935865402222
    },
    {
      "epoch": 1.2416839599609375e-06,
      "model_forward_time": 1.729614496231079,
      "step": 75302
    },
    {
      "epoch": 1.2416839599609375e-06,
      "step": 75302,
      "training_step_time": 7.5467798709869385
    },
    {
      "epoch": 1.2426376342773437e-06,
      "model_forward_time": 1.7305583953857422,
      "step": 75303
    },
    {
      "epoch": 1.2426376342773437e-06,
      "step": 75303,
      "training_step_time": 7.480973243713379
    },
    {
      "epoch": 1.24359130859375e-06,
      "model_forward_time": 1.7176547050476074,
      "step": 75304
    },
    {
      "epoch": 1.24359130859375e-06,
      "step": 75304,
      "training_step_time": 7.7797582149505615
    },
    {
      "epoch": 1.2445449829101562e-06,
      "model_forward_time": 1.7203240394592285,
      "step": 75305
    },
    {
      "epoch": 1.2445449829101562e-06,
      "step": 75305,
      "training_step_time": 8.49785590171814
    },
    {
      "epoch": 1.2454986572265625e-06,
      "model_forward_time": 1.7229478359222412,
      "step": 75306
    },
    {
      "epoch": 1.2454986572265625e-06,
      "step": 75306,
      "training_step_time": 8.640167236328125
    },
    {
      "epoch": 1.2464523315429687e-06,
      "model_forward_time": 1.7222216129302979,
      "step": 75307
    },
    {
      "epoch": 1.2464523315429687e-06,
      "step": 75307,
      "training_step_time": 7.746058464050293
    },
    {
      "epoch": 1.247406005859375e-06,
      "model_forward_time": 1.7203648090362549,
      "step": 75308
    },
    {
      "epoch": 1.247406005859375e-06,
      "step": 75308,
      "training_step_time": 7.69721794128418
    },
    {
      "epoch": 1.2483596801757812e-06,
      "model_forward_time": 1.7430784702301025,
      "step": 75309
    },
    {
      "epoch": 1.2483596801757812e-06,
      "step": 75309,
      "training_step_time": 8.677836179733276
    },
    {
      "epoch": 1.2493133544921876e-06,
      "grad_norm": 0.09990815073251724,
      "learning_rate": 1.5761798271050544e-06,
      "loss": 0.0555,
      "step": 75310
    },
    {
      "epoch": 1.2493133544921876e-06,
      "model_forward_time": 2.8338186740875244,
      "step": 75310
    },
    {
      "dynamics_loss_avg": 0.0515028964728117,
      "epoch": 1.2493133544921876e-06,
      "step": 75310
    },
    {
      "action_loss_avg": 0.0011925291502848268,
      "epoch": 1.2493133544921876e-06,
      "step": 75310
    },
    {
      "epoch": 1.2493133544921876e-06,
      "step": 75310,
      "training_step_time": 8.394836664199829
    },
    {
      "epoch": 1.2502670288085937e-06,
      "model_forward_time": 1.7149522304534912,
      "step": 75311
    },
    {
      "epoch": 1.2502670288085937e-06,
      "step": 75311,
      "training_step_time": 7.4872002601623535
    },
    {
      "epoch": 1.251220703125e-06,
      "model_forward_time": 1.7260193824768066,
      "step": 75312
    },
    {
      "epoch": 1.251220703125e-06,
      "step": 75312,
      "training_step_time": 7.48431396484375
    },
    {
      "epoch": 1.2521743774414062e-06,
      "model_forward_time": 1.722449541091919,
      "step": 75313
    },
    {
      "epoch": 1.2521743774414062e-06,
      "step": 75313,
      "training_step_time": 7.6666579246521
    },
    {
      "epoch": 1.2531280517578126e-06,
      "model_forward_time": 1.7197368144989014,
      "step": 75314
    },
    {
      "epoch": 1.2531280517578126e-06,
      "step": 75314,
      "training_step_time": 8.766509056091309
    },
    {
      "epoch": 1.2540817260742187e-06,
      "model_forward_time": 1.721562147140503,
      "step": 75315
    },
    {
      "epoch": 1.2540817260742187e-06,
      "step": 75315,
      "training_step_time": 7.776707172393799
    },
    {
      "epoch": 1.255035400390625e-06,
      "model_forward_time": 1.7185015678405762,
      "step": 75316
    },
    {
      "epoch": 1.255035400390625e-06,
      "step": 75316,
      "training_step_time": 9.063392877578735
    },
    {
      "epoch": 1.2559890747070312e-06,
      "model_forward_time": 1.3488569259643555,
      "step": 75317
    },
    {
      "epoch": 1.2559890747070312e-06,
      "step": 75317,
      "training_step_time": 9.63777208328247
    },
    {
      "epoch": 1.2569427490234376e-06,
      "model_forward_time": 1.7570087909698486,
      "step": 75318
    },
    {
      "epoch": 1.2569427490234376e-06,
      "step": 75318,
      "training_step_time": 8.379386186599731
    },
    {
      "epoch": 1.2578964233398437e-06,
      "model_forward_time": 2.772512912750244,
      "step": 75319
    },
    {
      "epoch": 1.2578964233398437e-06,
      "step": 75319,
      "training_step_time": 8.61554217338562
    },
    {
      "epoch": 1.25885009765625e-06,
      "grad_norm": 0.09132245182991028,
      "learning_rate": 1.5749750236435279e-06,
      "loss": 0.0536,
      "step": 75320
    },
    {
      "epoch": 1.25885009765625e-06,
      "model_forward_time": 1.7297279834747314,
      "step": 75320
    },
    {
      "dynamics_loss_avg": 0.06173109505325556,
      "epoch": 1.25885009765625e-06,
      "step": 75320
    },
    {
      "action_loss_avg": 0.0015591300500091166,
      "epoch": 1.25885009765625e-06,
      "step": 75320
    },
    {
      "epoch": 1.25885009765625e-06,
      "step": 75320,
      "training_step_time": 8.343336582183838
    },
    {
      "epoch": 1.2598037719726562e-06,
      "model_forward_time": 1.729780673980713,
      "step": 75321
    },
    {
      "epoch": 1.2598037719726562e-06,
      "step": 75321,
      "training_step_time": 7.53232216835022
    },
    {
      "epoch": 1.2607574462890626e-06,
      "model_forward_time": 1.7248404026031494,
      "step": 75322
    },
    {
      "epoch": 1.2607574462890626e-06,
      "step": 75322,
      "training_step_time": 7.770059108734131
    },
    {
      "epoch": 1.2617111206054687e-06,
      "model_forward_time": 1.7145941257476807,
      "step": 75323
    },
    {
      "epoch": 1.2617111206054687e-06,
      "step": 75323,
      "training_step_time": 7.533544063568115
    },
    {
      "epoch": 1.262664794921875e-06,
      "model_forward_time": 1.7220900058746338,
      "step": 75324
    },
    {
      "epoch": 1.262664794921875e-06,
      "step": 75324,
      "training_step_time": 7.6668994426727295
    },
    {
      "epoch": 1.2636184692382812e-06,
      "model_forward_time": 1.7219748497009277,
      "step": 75325
    },
    {
      "epoch": 1.2636184692382812e-06,
      "step": 75325,
      "training_step_time": 8.316918134689331
    },
    {
      "epoch": 1.2645721435546876e-06,
      "model_forward_time": 1.7288413047790527,
      "step": 75326
    },
    {
      "epoch": 1.2645721435546876e-06,
      "step": 75326,
      "training_step_time": 8.64646053314209
    },
    {
      "epoch": 1.2655258178710937e-06,
      "model_forward_time": 1.789954423904419,
      "step": 75327
    },
    {
      "epoch": 1.2655258178710937e-06,
      "step": 75327,
      "training_step_time": 8.791756868362427
    },
    {
      "epoch": 1.2664794921875e-06,
      "model_forward_time": 2.772535562515259,
      "step": 75328
    },
    {
      "epoch": 1.2664794921875e-06,
      "step": 75328,
      "training_step_time": 7.743708848953247
    },
    {
      "epoch": 1.2674331665039062e-06,
      "model_forward_time": 1.7067897319793701,
      "step": 75329
    },
    {
      "epoch": 1.2674331665039062e-06,
      "step": 75329,
      "training_step_time": 8.513309955596924
    },
    {
      "epoch": 1.2683868408203126e-06,
      "grad_norm": 0.09181276708841324,
      "learning_rate": 1.5737705947375814e-06,
      "loss": 0.0537,
      "step": 75330
    },
    {
      "epoch": 1.2683868408203126e-06,
      "model_forward_time": 1.7096657752990723,
      "step": 75330
    },
    {
      "dynamics_loss_avg": 0.055622640810906884,
      "epoch": 1.2683868408203126e-06,
      "step": 75330
    },
    {
      "action_loss_avg": 0.0009639857104048133,
      "epoch": 1.2683868408203126e-06,
      "step": 75330
    },
    {
      "epoch": 1.2683868408203126e-06,
      "step": 75330,
      "training_step_time": 7.427016019821167
    },
    {
      "epoch": 1.2693405151367187e-06,
      "model_forward_time": 1.709904670715332,
      "step": 75331
    },
    {
      "epoch": 1.2693405151367187e-06,
      "step": 75331,
      "training_step_time": 7.7017786502838135
    },
    {
      "epoch": 1.270294189453125e-06,
      "model_forward_time": 1.7087724208831787,
      "step": 75332
    },
    {
      "epoch": 1.270294189453125e-06,
      "step": 75332,
      "training_step_time": 7.801983833312988
    },
    {
      "epoch": 1.2712478637695312e-06,
      "model_forward_time": 1.711991548538208,
      "step": 75333
    },
    {
      "epoch": 1.2712478637695312e-06,
      "step": 75333,
      "training_step_time": 8.174046277999878
    },
    {
      "epoch": 1.2722015380859376e-06,
      "model_forward_time": 1.7117393016815186,
      "step": 75334
    },
    {
      "epoch": 1.2722015380859376e-06,
      "step": 75334,
      "training_step_time": 8.269969463348389
    },
    {
      "epoch": 1.2731552124023437e-06,
      "model_forward_time": 1.7107634544372559,
      "step": 75335
    },
    {
      "epoch": 1.2731552124023437e-06,
      "step": 75335,
      "training_step_time": 9.629245042800903
    },
    {
      "epoch": 1.27410888671875e-06,
      "model_forward_time": 1.7683420181274414,
      "step": 75336
    },
    {
      "epoch": 1.27410888671875e-06,
      "step": 75336,
      "training_step_time": 8.395771980285645
    },
    {
      "epoch": 1.2750625610351562e-06,
      "model_forward_time": 1.7110188007354736,
      "step": 75337
    },
    {
      "epoch": 1.2750625610351562e-06,
      "step": 75337,
      "training_step_time": 7.860126972198486
    },
    {
      "epoch": 1.2760162353515626e-06,
      "model_forward_time": 2.8199622631073,
      "step": 75338
    },
    {
      "epoch": 1.2760162353515626e-06,
      "step": 75338,
      "training_step_time": 8.416487693786621
    },
    {
      "epoch": 1.2769699096679687e-06,
      "model_forward_time": 1.7112846374511719,
      "step": 75339
    },
    {
      "epoch": 1.2769699096679687e-06,
      "step": 75339,
      "training_step_time": 7.471618175506592
    },
    {
      "epoch": 1.277923583984375e-06,
      "grad_norm": 0.08205409348011017,
      "learning_rate": 1.5725665405189321e-06,
      "loss": 0.0542,
      "step": 75340
    },
    {
      "epoch": 1.277923583984375e-06,
      "model_forward_time": 1.708024263381958,
      "step": 75340
    },
    {
      "dynamics_loss_avg": 0.046644015982747075,
      "epoch": 1.277923583984375e-06,
      "step": 75340
    },
    {
      "action_loss_avg": 0.0012174769421108068,
      "epoch": 1.277923583984375e-06,
      "step": 75340
    },
    {
      "epoch": 1.277923583984375e-06,
      "step": 75340,
      "training_step_time": 7.650103330612183
    },
    {
      "epoch": 1.2788772583007812e-06,
      "model_forward_time": 1.7026546001434326,
      "step": 75341
    },
    {
      "epoch": 1.2788772583007812e-06,
      "step": 75341,
      "training_step_time": 8.836773157119751
    },
    {
      "epoch": 1.2798309326171876e-06,
      "model_forward_time": 1.7054717540740967,
      "step": 75342
    },
    {
      "epoch": 1.2798309326171876e-06,
      "step": 75342,
      "training_step_time": 8.534468650817871
    },
    {
      "epoch": 1.2807846069335937e-06,
      "model_forward_time": 1.7125084400177002,
      "step": 75343
    },
    {
      "epoch": 1.2807846069335937e-06,
      "step": 75343,
      "training_step_time": 7.726530075073242
    },
    {
      "epoch": 1.28173828125e-06,
      "model_forward_time": 1.7057008743286133,
      "step": 75344
    },
    {
      "epoch": 1.28173828125e-06,
      "step": 75344,
      "training_step_time": 7.863450765609741
    },
    {
      "epoch": 1.2826919555664062e-06,
      "model_forward_time": 1.7281513214111328,
      "step": 75345
    },
    {
      "epoch": 1.2826919555664062e-06,
      "step": 75345,
      "training_step_time": 8.892077922821045
    },
    {
      "epoch": 1.2836456298828126e-06,
      "model_forward_time": 1.71073317527771,
      "step": 75346
    },
    {
      "epoch": 1.2836456298828126e-06,
      "step": 75346,
      "training_step_time": 7.834749221801758
    },
    {
      "epoch": 1.2845993041992187e-06,
      "model_forward_time": 2.8467092514038086,
      "step": 75347
    },
    {
      "epoch": 1.2845993041992187e-06,
      "step": 75347,
      "training_step_time": 8.297847747802734
    },
    {
      "epoch": 1.285552978515625e-06,
      "model_forward_time": 1.7229220867156982,
      "step": 75348
    },
    {
      "epoch": 1.285552978515625e-06,
      "step": 75348,
      "training_step_time": 8.50158977508545
    },
    {
      "epoch": 1.2865066528320312e-06,
      "model_forward_time": 1.7093331813812256,
      "step": 75349
    },
    {
      "epoch": 1.2865066528320312e-06,
      "step": 75349,
      "training_step_time": 7.56708025932312
    },
    {
      "epoch": 1.2874603271484376e-06,
      "grad_norm": 0.11482048779726028,
      "learning_rate": 1.5713628611192527e-06,
      "loss": 0.0562,
      "step": 75350
    },
    {
      "epoch": 1.2874603271484376e-06,
      "model_forward_time": 1.7067594528198242,
      "step": 75350
    },
    {
      "dynamics_loss_avg": 0.05264951754361391,
      "epoch": 1.2874603271484376e-06,
      "step": 75350
    },
    {
      "action_loss_avg": 0.0011597345641348512,
      "epoch": 1.2874603271484376e-06,
      "step": 75350
    },
    {
      "epoch": 1.2874603271484376e-06,
      "step": 75350,
      "training_step_time": 7.788164138793945
    },
    {
      "epoch": 1.2884140014648437e-06,
      "model_forward_time": 1.7210156917572021,
      "step": 75351
    },
    {
      "epoch": 1.2884140014648437e-06,
      "step": 75351,
      "training_step_time": 10.192695379257202
    },
    {
      "epoch": 1.28936767578125e-06,
      "model_forward_time": 1.7103326320648193,
      "step": 75352
    },
    {
      "epoch": 1.28936767578125e-06,
      "step": 75352,
      "training_step_time": 8.611382007598877
    },
    {
      "epoch": 1.2903213500976562e-06,
      "model_forward_time": 1.7205028533935547,
      "step": 75353
    },
    {
      "epoch": 1.2903213500976562e-06,
      "step": 75353,
      "training_step_time": 9.350817441940308
    },
    {
      "epoch": 1.2912750244140626e-06,
      "model_forward_time": 1.7084040641784668,
      "step": 75354
    },
    {
      "epoch": 1.2912750244140626e-06,
      "step": 75354,
      "training_step_time": 7.825998783111572
    },
    {
      "epoch": 1.2922286987304687e-06,
      "model_forward_time": 1.792891025543213,
      "step": 75355
    },
    {
      "epoch": 1.2922286987304687e-06,
      "step": 75355,
      "training_step_time": 7.750200271606445
    },
    {
      "epoch": 1.293182373046875e-06,
      "model_forward_time": 2.8383255004882812,
      "step": 75356
    },
    {
      "epoch": 1.293182373046875e-06,
      "step": 75356,
      "training_step_time": 7.630084753036499
    },
    {
      "epoch": 1.2941360473632812e-06,
      "model_forward_time": 1.327364444732666,
      "step": 75357
    },
    {
      "epoch": 1.2941360473632812e-06,
      "step": 75357,
      "training_step_time": 8.337486267089844
    },
    {
      "epoch": 1.2950897216796876e-06,
      "model_forward_time": 1.7235114574432373,
      "step": 75358
    },
    {
      "epoch": 1.2950897216796876e-06,
      "step": 75358,
      "training_step_time": 7.165057182312012
    },
    {
      "epoch": 1.2960433959960937e-06,
      "model_forward_time": 1.7096986770629883,
      "step": 75359
    },
    {
      "epoch": 1.2960433959960937e-06,
      "step": 75359,
      "training_step_time": 7.673710823059082
    },
    {
      "epoch": 1.2969970703125e-06,
      "grad_norm": 0.11125899106264114,
      "learning_rate": 1.5701595566701772e-06,
      "loss": 0.0548,
      "step": 75360
    },
    {
      "epoch": 1.2969970703125e-06,
      "model_forward_time": 1.7106873989105225,
      "step": 75360
    },
    {
      "dynamics_loss_avg": 0.051767689548432826,
      "epoch": 1.2969970703125e-06,
      "step": 75360
    },
    {
      "action_loss_avg": 0.0013516685139620677,
      "epoch": 1.2969970703125e-06,
      "step": 75360
    },
    {
      "epoch": 1.2969970703125e-06,
      "step": 75360,
      "training_step_time": 7.554895639419556
    },
    {
      "epoch": 1.2979507446289062e-06,
      "model_forward_time": 1.7089190483093262,
      "step": 75361
    },
    {
      "epoch": 1.2979507446289062e-06,
      "step": 75361,
      "training_step_time": 8.375656127929688
    },
    {
      "epoch": 1.2989044189453126e-06,
      "model_forward_time": 1.7129104137420654,
      "step": 75362
    },
    {
      "epoch": 1.2989044189453126e-06,
      "step": 75362,
      "training_step_time": 8.40802526473999
    },
    {
      "epoch": 1.2998580932617187e-06,
      "model_forward_time": 1.7097301483154297,
      "step": 75363
    },
    {
      "epoch": 1.2998580932617187e-06,
      "step": 75363,
      "training_step_time": 7.690754175186157
    },
    {
      "epoch": 1.300811767578125e-06,
      "model_forward_time": 1.7540473937988281,
      "step": 75364
    },
    {
      "epoch": 1.300811767578125e-06,
      "step": 75364,
      "training_step_time": 8.12691354751587
    },
    {
      "epoch": 1.3017654418945312e-06,
      "model_forward_time": 2.7998430728912354,
      "step": 75365
    },
    {
      "epoch": 1.3017654418945312e-06,
      "step": 75365,
      "training_step_time": 8.285659313201904
    },
    {
      "epoch": 1.3027191162109376e-06,
      "model_forward_time": 1.713625431060791,
      "step": 75366
    },
    {
      "epoch": 1.3027191162109376e-06,
      "step": 75366,
      "training_step_time": 7.608005523681641
    },
    {
      "epoch": 1.3036727905273437e-06,
      "model_forward_time": 1.726073980331421,
      "step": 75367
    },
    {
      "epoch": 1.3036727905273437e-06,
      "step": 75367,
      "training_step_time": 7.155938148498535
    },
    {
      "epoch": 1.30462646484375e-06,
      "model_forward_time": 1.7157130241394043,
      "step": 75368
    },
    {
      "epoch": 1.30462646484375e-06,
      "step": 75368,
      "training_step_time": 8.50470232963562
    },
    {
      "epoch": 1.3055801391601562e-06,
      "model_forward_time": 1.7133762836456299,
      "step": 75369
    },
    {
      "epoch": 1.3055801391601562e-06,
      "step": 75369,
      "training_step_time": 7.414263963699341
    },
    {
      "epoch": 1.3065338134765626e-06,
      "grad_norm": 0.07798006385564804,
      "learning_rate": 1.5689566273032946e-06,
      "loss": 0.0548,
      "step": 75370
    },
    {
      "epoch": 1.3065338134765626e-06,
      "model_forward_time": 1.7119929790496826,
      "step": 75370
    },
    {
      "dynamics_loss_avg": 0.039796097483485934,
      "epoch": 1.3065338134765626e-06,
      "step": 75370
    },
    {
      "action_loss_avg": 0.0009608048218069598,
      "epoch": 1.3065338134765626e-06,
      "step": 75370
    },
    {
      "epoch": 1.3065338134765626e-06,
      "step": 75370,
      "training_step_time": 7.577247858047485
    },
    {
      "epoch": 1.3074874877929687e-06,
      "model_forward_time": 1.7249867916107178,
      "step": 75371
    },
    {
      "epoch": 1.3074874877929687e-06,
      "step": 75371,
      "training_step_time": 8.412834882736206
    },
    {
      "epoch": 1.308441162109375e-06,
      "model_forward_time": 1.715994119644165,
      "step": 75372
    },
    {
      "epoch": 1.308441162109375e-06,
      "step": 75372,
      "training_step_time": 7.799685955047607
    },
    {
      "epoch": 1.3093948364257812e-06,
      "model_forward_time": 1.7504639625549316,
      "step": 75373
    },
    {
      "epoch": 1.3093948364257812e-06,
      "step": 75373,
      "training_step_time": 8.354526996612549
    },
    {
      "epoch": 1.3103485107421876e-06,
      "model_forward_time": 1.7197470664978027,
      "step": 75374
    },
    {
      "epoch": 1.3103485107421876e-06,
      "step": 75374,
      "training_step_time": 10.159302473068237
    },
    {
      "epoch": 1.3113021850585938e-06,
      "model_forward_time": 1.70745849609375,
      "step": 75375
    },
    {
      "epoch": 1.3113021850585938e-06,
      "step": 75375,
      "training_step_time": 7.78230357170105
    },
    {
      "epoch": 1.312255859375e-06,
      "model_forward_time": 1.7184436321258545,
      "step": 75376
    },
    {
      "epoch": 1.312255859375e-06,
      "step": 75376,
      "training_step_time": 7.504334926605225
    },
    {
      "epoch": 1.3132095336914063e-06,
      "model_forward_time": 1.7239704132080078,
      "step": 75377
    },
    {
      "epoch": 1.3132095336914063e-06,
      "step": 75377,
      "training_step_time": 9.273571729660034
    },
    {
      "epoch": 1.3141632080078124e-06,
      "model_forward_time": 1.7234365940093994,
      "step": 75378
    },
    {
      "epoch": 1.3141632080078124e-06,
      "step": 75378,
      "training_step_time": 8.720480918884277
    },
    {
      "epoch": 1.3151168823242188e-06,
      "model_forward_time": 1.746267557144165,
      "step": 75379
    },
    {
      "epoch": 1.3151168823242188e-06,
      "step": 75379,
      "training_step_time": 7.724548578262329
    },
    {
      "epoch": 1.316070556640625e-06,
      "grad_norm": 0.16459481418132782,
      "learning_rate": 1.5677540731501584e-06,
      "loss": 0.0557,
      "step": 75380
    },
    {
      "epoch": 1.316070556640625e-06,
      "model_forward_time": 1.7142295837402344,
      "step": 75380
    },
    {
      "dynamics_loss_avg": 0.0549642089754343,
      "epoch": 1.316070556640625e-06,
      "step": 75380
    },
    {
      "action_loss_avg": 0.0011517538805492222,
      "epoch": 1.316070556640625e-06,
      "step": 75380
    },
    {
      "epoch": 1.316070556640625e-06,
      "step": 75380,
      "training_step_time": 8.323379039764404
    },
    {
      "epoch": 1.3170242309570313e-06,
      "model_forward_time": 1.7210438251495361,
      "step": 75381
    },
    {
      "epoch": 1.3170242309570313e-06,
      "step": 75381,
      "training_step_time": 8.649385690689087
    },
    {
      "epoch": 1.3179779052734374e-06,
      "model_forward_time": 1.7382299900054932,
      "step": 75382
    },
    {
      "epoch": 1.3179779052734374e-06,
      "step": 75382,
      "training_step_time": 10.521336793899536
    },
    {
      "epoch": 1.3189315795898438e-06,
      "model_forward_time": 1.7090792655944824,
      "step": 75383
    },
    {
      "epoch": 1.3189315795898438e-06,
      "step": 75383,
      "training_step_time": 8.797773599624634
    },
    {
      "epoch": 1.31988525390625e-06,
      "model_forward_time": 2.893585681915283,
      "step": 75384
    },
    {
      "epoch": 1.31988525390625e-06,
      "step": 75384,
      "training_step_time": 7.783443927764893
    },
    {
      "epoch": 1.3208389282226563e-06,
      "model_forward_time": 1.7235169410705566,
      "step": 75385
    },
    {
      "epoch": 1.3208389282226563e-06,
      "step": 75385,
      "training_step_time": 7.805596828460693
    },
    {
      "epoch": 1.3217926025390624e-06,
      "model_forward_time": 1.7141315937042236,
      "step": 75386
    },
    {
      "epoch": 1.3217926025390624e-06,
      "step": 75386,
      "training_step_time": 8.48708200454712
    },
    {
      "epoch": 1.3227462768554688e-06,
      "model_forward_time": 1.717752456665039,
      "step": 75387
    },
    {
      "epoch": 1.3227462768554688e-06,
      "step": 75387,
      "training_step_time": 8.52140998840332
    },
    {
      "epoch": 1.323699951171875e-06,
      "model_forward_time": 1.70346999168396,
      "step": 75388
    },
    {
      "epoch": 1.323699951171875e-06,
      "step": 75388,
      "training_step_time": 8.570799350738525
    },
    {
      "epoch": 1.3246536254882813e-06,
      "model_forward_time": 1.719095230102539,
      "step": 75389
    },
    {
      "epoch": 1.3246536254882813e-06,
      "step": 75389,
      "training_step_time": 8.666204929351807
    },
    {
      "epoch": 1.3256072998046874e-06,
      "grad_norm": 0.10313784331083298,
      "learning_rate": 1.5665518943422747e-06,
      "loss": 0.0551,
      "step": 75390
    },
    {
      "epoch": 1.3256072998046874e-06,
      "model_forward_time": 1.7211201190948486,
      "step": 75390
    },
    {
      "dynamics_loss_avg": 0.05013261148706079,
      "epoch": 1.3256072998046874e-06,
      "step": 75390
    },
    {
      "action_loss_avg": 0.0016991105308989062,
      "epoch": 1.3256072998046874e-06,
      "step": 75390
    },
    {
      "epoch": 1.3256072998046874e-06,
      "step": 75390,
      "training_step_time": 7.805758476257324
    },
    {
      "epoch": 1.3265609741210938e-06,
      "model_forward_time": 1.753516435623169,
      "step": 75391
    },
    {
      "epoch": 1.3265609741210938e-06,
      "step": 75391,
      "training_step_time": 7.806471824645996
    },
    {
      "epoch": 1.3275146484375e-06,
      "model_forward_time": 1.7513554096221924,
      "step": 75392
    },
    {
      "epoch": 1.3275146484375e-06,
      "step": 75392,
      "training_step_time": 7.755385875701904
    },
    {
      "epoch": 1.3284683227539063e-06,
      "model_forward_time": 2.8876891136169434,
      "step": 75393
    },
    {
      "epoch": 1.3284683227539063e-06,
      "step": 75393,
      "training_step_time": 7.773055553436279
    },
    {
      "epoch": 1.3294219970703124e-06,
      "model_forward_time": 1.7183723449707031,
      "step": 75394
    },
    {
      "epoch": 1.3294219970703124e-06,
      "step": 75394,
      "training_step_time": 8.486685514450073
    },
    {
      "epoch": 1.3303756713867188e-06,
      "model_forward_time": 1.7150654792785645,
      "step": 75395
    },
    {
      "epoch": 1.3303756713867188e-06,
      "step": 75395,
      "training_step_time": 6.7200844287872314
    },
    {
      "epoch": 1.331329345703125e-06,
      "model_forward_time": 1.7268695831298828,
      "step": 75396
    },
    {
      "epoch": 1.331329345703125e-06,
      "step": 75396,
      "training_step_time": 7.71183705329895
    },
    {
      "epoch": 1.3322830200195313e-06,
      "model_forward_time": 1.7253022193908691,
      "step": 75397
    },
    {
      "epoch": 1.3322830200195313e-06,
      "step": 75397,
      "training_step_time": 7.560984134674072
    },
    {
      "epoch": 1.3332366943359374e-06,
      "model_forward_time": 1.718127965927124,
      "step": 75398
    },
    {
      "epoch": 1.3332366943359374e-06,
      "step": 75398,
      "training_step_time": 7.669875144958496
    },
    {
      "epoch": 1.3341903686523438e-06,
      "model_forward_time": 1.7251760959625244,
      "step": 75399
    },
    {
      "epoch": 1.3341903686523438e-06,
      "step": 75399,
      "training_step_time": 9.194429874420166
    },
    {
      "epoch": 1.33514404296875e-06,
      "grad_norm": 0.08429248631000519,
      "learning_rate": 1.5653500910111135e-06,
      "loss": 0.0566,
      "step": 75400
    },
    {
      "epoch": 1.33514404296875e-06,
      "model_forward_time": 1.7395007610321045,
      "step": 75400
    },
    {
      "dynamics_loss_avg": 0.055584772303700444,
      "epoch": 1.33514404296875e-06,
      "step": 75400
    },
    {
      "action_loss_avg": 0.0020550928777083753,
      "epoch": 1.33514404296875e-06,
      "step": 75400
    },
    {
      "epoch": 1.33514404296875e-06,
      "step": 75400,
      "training_step_time": 7.826835870742798
    },
    {
      "epoch": 1.3360977172851563e-06,
      "model_forward_time": 1.7441210746765137,
      "step": 75401
    },
    {
      "epoch": 1.3360977172851563e-06,
      "step": 75401,
      "training_step_time": 9.63382339477539
    },
    {
      "epoch": 1.3370513916015624e-06,
      "model_forward_time": 2.846700668334961,
      "step": 75402
    },
    {
      "epoch": 1.3370513916015624e-06,
      "step": 75402,
      "training_step_time": 7.7053210735321045
    },
    {
      "epoch": 1.3380050659179688e-06,
      "model_forward_time": 1.7168068885803223,
      "step": 75403
    },
    {
      "epoch": 1.3380050659179688e-06,
      "step": 75403,
      "training_step_time": 8.349747657775879
    },
    {
      "epoch": 1.338958740234375e-06,
      "model_forward_time": 1.73323392868042,
      "step": 75404
    },
    {
      "epoch": 1.338958740234375e-06,
      "step": 75404,
      "training_step_time": 7.2775280475616455
    },
    {
      "epoch": 1.3399124145507813e-06,
      "model_forward_time": 1.728748083114624,
      "step": 75405
    },
    {
      "epoch": 1.3399124145507813e-06,
      "step": 75405,
      "training_step_time": 7.771374464035034
    },
    {
      "epoch": 1.3408660888671874e-06,
      "model_forward_time": 1.7187910079956055,
      "step": 75406
    },
    {
      "epoch": 1.3408660888671874e-06,
      "step": 75406,
      "training_step_time": 7.598886013031006
    },
    {
      "epoch": 1.3418197631835938e-06,
      "model_forward_time": 1.7227318286895752,
      "step": 75407
    },
    {
      "epoch": 1.3418197631835938e-06,
      "step": 75407,
      "training_step_time": 7.664687395095825
    },
    {
      "epoch": 1.3427734375e-06,
      "model_forward_time": 1.731332778930664,
      "step": 75408
    },
    {
      "epoch": 1.3427734375e-06,
      "step": 75408,
      "training_step_time": 8.5024995803833
    },
    {
      "epoch": 1.3437271118164063e-06,
      "model_forward_time": 1.7277333736419678,
      "step": 75409
    },
    {
      "epoch": 1.3437271118164063e-06,
      "step": 75409,
      "training_step_time": 7.735609531402588
    },
    {
      "epoch": 1.3446807861328124e-06,
      "grad_norm": 0.07911242544651031,
      "learning_rate": 1.5641486632881026e-06,
      "loss": 0.0554,
      "step": 75410
    },
    {
      "epoch": 1.3446807861328124e-06,
      "model_forward_time": 1.734954833984375,
      "step": 75410
    },
    {
      "dynamics_loss_avg": 0.04461330072954297,
      "epoch": 1.3446807861328124e-06,
      "step": 75410
    },
    {
      "action_loss_avg": 0.0015213143284199758,
      "epoch": 1.3446807861328124e-06,
      "step": 75410
    },
    {
      "epoch": 1.3446807861328124e-06,
      "step": 75410,
      "training_step_time": 8.611348152160645
    },
    {
      "epoch": 1.3456344604492188e-06,
      "model_forward_time": 2.820096492767334,
      "step": 75411
    },
    {
      "epoch": 1.3456344604492188e-06,
      "step": 75411,
      "training_step_time": 8.302178144454956
    },
    {
      "epoch": 1.346588134765625e-06,
      "model_forward_time": 1.714594841003418,
      "step": 75412
    },
    {
      "epoch": 1.346588134765625e-06,
      "step": 75412,
      "training_step_time": 8.237659931182861
    },
    {
      "epoch": 1.3475418090820313e-06,
      "model_forward_time": 1.733165979385376,
      "step": 75413
    },
    {
      "epoch": 1.3475418090820313e-06,
      "step": 75413,
      "training_step_time": 7.298729658126831
    },
    {
      "epoch": 1.3484954833984374e-06,
      "model_forward_time": 1.7259433269500732,
      "step": 75414
    },
    {
      "epoch": 1.3484954833984374e-06,
      "step": 75414,
      "training_step_time": 8.648377418518066
    },
    {
      "epoch": 1.3494491577148438e-06,
      "model_forward_time": 1.7184031009674072,
      "step": 75415
    },
    {
      "epoch": 1.3494491577148438e-06,
      "step": 75415,
      "training_step_time": 7.676382303237915
    },
    {
      "epoch": 1.35040283203125e-06,
      "model_forward_time": 1.7297189235687256,
      "step": 75416
    },
    {
      "epoch": 1.35040283203125e-06,
      "step": 75416,
      "training_step_time": 10.550905704498291
    },
    {
      "epoch": 1.3513565063476563e-06,
      "model_forward_time": 1.7151808738708496,
      "step": 75417
    },
    {
      "epoch": 1.3513565063476563e-06,
      "step": 75417,
      "training_step_time": 9.413551330566406
    },
    {
      "epoch": 1.3523101806640624e-06,
      "model_forward_time": 1.725856065750122,
      "step": 75418
    },
    {
      "epoch": 1.3523101806640624e-06,
      "step": 75418,
      "training_step_time": 7.638504505157471
    },
    {
      "epoch": 1.3532638549804688e-06,
      "model_forward_time": 1.7547791004180908,
      "step": 75419
    },
    {
      "epoch": 1.3532638549804688e-06,
      "step": 75419,
      "training_step_time": 7.853914976119995
    },
    {
      "epoch": 1.354217529296875e-06,
      "grad_norm": 0.13730210065841675,
      "learning_rate": 1.5629476113046298e-06,
      "loss": 0.0563,
      "step": 75420
    },
    {
      "epoch": 1.354217529296875e-06,
      "model_forward_time": 2.796736240386963,
      "step": 75420
    },
    {
      "dynamics_loss_avg": 0.052564213238656524,
      "epoch": 1.354217529296875e-06,
      "step": 75420
    },
    {
      "action_loss_avg": 0.0020366237207781524,
      "epoch": 1.354217529296875e-06,
      "step": 75420
    },
    {
      "epoch": 1.354217529296875e-06,
      "step": 75420,
      "training_step_time": 8.986706733703613
    },
    {
      "epoch": 1.3551712036132813e-06,
      "model_forward_time": 1.71897292137146,
      "step": 75421
    },
    {
      "epoch": 1.3551712036132813e-06,
      "step": 75421,
      "training_step_time": 7.685434818267822
    },
    {
      "epoch": 1.3561248779296874e-06,
      "model_forward_time": 1.7172818183898926,
      "step": 75422
    },
    {
      "epoch": 1.3561248779296874e-06,
      "step": 75422,
      "training_step_time": 8.339837551116943
    },
    {
      "epoch": 1.3570785522460938e-06,
      "model_forward_time": 1.778205394744873,
      "step": 75423
    },
    {
      "epoch": 1.3570785522460938e-06,
      "step": 75423,
      "training_step_time": 8.283470153808594
    },
    {
      "epoch": 1.3580322265625e-06,
      "model_forward_time": 1.798933744430542,
      "step": 75424
    },
    {
      "epoch": 1.3580322265625e-06,
      "step": 75424,
      "training_step_time": 8.780797719955444
    },
    {
      "epoch": 1.3589859008789063e-06,
      "model_forward_time": 1.8173046112060547,
      "step": 75425
    },
    {
      "epoch": 1.3589859008789063e-06,
      "step": 75425,
      "training_step_time": 6.76734471321106
    },
    {
      "epoch": 1.3599395751953124e-06,
      "model_forward_time": 1.7990524768829346,
      "step": 75426
    },
    {
      "epoch": 1.3599395751953124e-06,
      "step": 75426,
      "training_step_time": 8.63266897201538
    },
    {
      "epoch": 1.3608932495117188e-06,
      "model_forward_time": 1.7876574993133545,
      "step": 75427
    },
    {
      "epoch": 1.3608932495117188e-06,
      "step": 75427,
      "training_step_time": 6.74759316444397
    },
    {
      "epoch": 1.361846923828125e-06,
      "model_forward_time": 1.8094723224639893,
      "step": 75428
    },
    {
      "epoch": 1.361846923828125e-06,
      "step": 75428,
      "training_step_time": 6.714829444885254
    },
    {
      "epoch": 1.3628005981445313e-06,
      "model_forward_time": 1.8896548748016357,
      "step": 75429
    },
    {
      "epoch": 1.3628005981445313e-06,
      "step": 75429,
      "training_step_time": 6.726739168167114
    },
    {
      "epoch": 1.3637542724609374e-06,
      "grad_norm": 0.10096176713705063,
      "learning_rate": 1.5617469351920373e-06,
      "loss": 0.0565,
      "step": 75430
    },
    {
      "epoch": 1.3637542724609374e-06,
      "model_forward_time": 1.9548940658569336,
      "step": 75430
    },
    {
      "dynamics_loss_avg": 0.056418617442250255,
      "epoch": 1.3637542724609374e-06,
      "step": 75430
    },
    {
      "action_loss_avg": 0.0018583752593258397,
      "epoch": 1.3637542724609374e-06,
      "step": 75430
    },
    {
      "epoch": 1.3637542724609374e-06,
      "step": 75430,
      "training_step_time": 6.777622699737549
    },
    {
      "epoch": 1.3647079467773438e-06,
      "model_forward_time": 1.9382612705230713,
      "step": 75431
    },
    {
      "epoch": 1.3647079467773438e-06,
      "step": 75431,
      "training_step_time": 7.806925296783447
    },
    {
      "epoch": 1.36566162109375e-06,
      "model_forward_time": 1.9048092365264893,
      "step": 75432
    },
    {
      "epoch": 1.36566162109375e-06,
      "step": 75432,
      "training_step_time": 8.261050701141357
    },
    {
      "epoch": 1.3666152954101563e-06,
      "model_forward_time": 1.941920280456543,
      "step": 75433
    },
    {
      "epoch": 1.3666152954101563e-06,
      "step": 75433,
      "training_step_time": 8.58279538154602
    },
    {
      "epoch": 1.3675689697265624e-06,
      "model_forward_time": 1.9343407154083252,
      "step": 75434
    },
    {
      "epoch": 1.3675689697265624e-06,
      "step": 75434,
      "training_step_time": 8.49079155921936
    },
    {
      "epoch": 1.3685226440429688e-06,
      "model_forward_time": 1.9571805000305176,
      "step": 75435
    },
    {
      "epoch": 1.3685226440429688e-06,
      "step": 75435,
      "training_step_time": 8.656855821609497
    },
    {
      "epoch": 1.369476318359375e-06,
      "model_forward_time": 1.9361708164215088,
      "step": 75436
    },
    {
      "epoch": 1.369476318359375e-06,
      "step": 75436,
      "training_step_time": 7.237972974777222
    },
    {
      "epoch": 1.3704299926757813e-06,
      "model_forward_time": 1.9376907348632812,
      "step": 75437
    },
    {
      "epoch": 1.3704299926757813e-06,
      "step": 75437,
      "training_step_time": 6.781834602355957
    },
    {
      "epoch": 1.3713836669921874e-06,
      "model_forward_time": 1.9173219203948975,
      "step": 75438
    },
    {
      "epoch": 1.3713836669921874e-06,
      "step": 75438,
      "training_step_time": 6.765534162521362
    },
    {
      "epoch": 1.3723373413085938e-06,
      "model_forward_time": 1.899616003036499,
      "step": 75439
    },
    {
      "epoch": 1.3723373413085938e-06,
      "step": 75439,
      "training_step_time": 6.745962619781494
    },
    {
      "epoch": 1.373291015625e-06,
      "grad_norm": 0.12031341344118118,
      "learning_rate": 1.5605466350816317e-06,
      "loss": 0.0557,
      "step": 75440
    },
    {
      "epoch": 1.373291015625e-06,
      "model_forward_time": 1.9201045036315918,
      "step": 75440
    },
    {
      "dynamics_loss_avg": 0.053148532286286354,
      "epoch": 1.373291015625e-06,
      "step": 75440
    },
    {
      "action_loss_avg": 0.0009091002168133855,
      "epoch": 1.373291015625e-06,
      "step": 75440
    },
    {
      "epoch": 1.373291015625e-06,
      "step": 75440,
      "training_step_time": 6.677061557769775
    },
    {
      "epoch": 1.3742446899414063e-06,
      "model_forward_time": 2.5652737617492676,
      "step": 75441
    },
    {
      "epoch": 1.3742446899414063e-06,
      "step": 75441,
      "training_step_time": 9.186241626739502
    },
    {
      "epoch": 1.3751983642578125e-06,
      "model_forward_time": 1.9235715866088867,
      "step": 75442
    },
    {
      "epoch": 1.3751983642578125e-06,
      "step": 75442,
      "training_step_time": 8.625314712524414
    },
    {
      "epoch": 1.3761520385742188e-06,
      "model_forward_time": 1.9608752727508545,
      "step": 75443
    },
    {
      "epoch": 1.3761520385742188e-06,
      "step": 75443,
      "training_step_time": 7.719601154327393
    },
    {
      "epoch": 1.377105712890625e-06,
      "model_forward_time": 1.9337623119354248,
      "step": 75444
    },
    {
      "epoch": 1.377105712890625e-06,
      "step": 75444,
      "training_step_time": 7.63526725769043
    },
    {
      "epoch": 1.3780593872070313e-06,
      "model_forward_time": 1.921854019165039,
      "step": 75445
    },
    {
      "epoch": 1.3780593872070313e-06,
      "step": 75445,
      "training_step_time": 7.622596979141235
    },
    {
      "epoch": 1.3790130615234375e-06,
      "model_forward_time": 1.9503743648529053,
      "step": 75446
    },
    {
      "epoch": 1.3790130615234375e-06,
      "step": 75446,
      "training_step_time": 6.776439666748047
    },
    {
      "epoch": 1.3799667358398438e-06,
      "model_forward_time": 1.9350364208221436,
      "step": 75447
    },
    {
      "epoch": 1.3799667358398438e-06,
      "step": 75447,
      "training_step_time": 6.769999742507935
    },
    {
      "epoch": 1.38092041015625e-06,
      "model_forward_time": 1.9287121295928955,
      "step": 75448
    },
    {
      "epoch": 1.38092041015625e-06,
      "step": 75448,
      "training_step_time": 6.748759984970093
    },
    {
      "epoch": 1.3818740844726563e-06,
      "model_forward_time": 1.9441332817077637,
      "step": 75449
    },
    {
      "epoch": 1.3818740844726563e-06,
      "step": 75449,
      "training_step_time": 6.798208475112915
    },
    {
      "epoch": 1.3828277587890625e-06,
      "grad_norm": 0.07999000698328018,
      "learning_rate": 1.559346711104674e-06,
      "loss": 0.0563,
      "step": 75450
    },
    {
      "epoch": 1.3828277587890625e-06,
      "model_forward_time": 2.5791964530944824,
      "step": 75450
    },
    {
      "dynamics_loss_avg": 0.0625394569709897,
      "epoch": 1.3828277587890625e-06,
      "step": 75450
    },
    {
      "action_loss_avg": 0.0011872747680172324,
      "epoch": 1.3828277587890625e-06,
      "step": 75450
    },
    {
      "epoch": 1.3828277587890625e-06,
      "step": 75450,
      "training_step_time": 9.083817481994629
    },
    {
      "epoch": 1.3837814331054688e-06,
      "model_forward_time": 1.9441378116607666,
      "step": 75451
    },
    {
      "epoch": 1.3837814331054688e-06,
      "step": 75451,
      "training_step_time": 9.365757942199707
    },
    {
      "epoch": 1.384735107421875e-06,
      "model_forward_time": 1.9199562072753906,
      "step": 75452
    },
    {
      "epoch": 1.384735107421875e-06,
      "step": 75452,
      "training_step_time": 7.69916033744812
    },
    {
      "epoch": 1.3856887817382813e-06,
      "model_forward_time": 1.9497249126434326,
      "step": 75453
    },
    {
      "epoch": 1.3856887817382813e-06,
      "step": 75453,
      "training_step_time": 7.603118896484375
    },
    {
      "epoch": 1.3866424560546875e-06,
      "model_forward_time": 1.920297622680664,
      "step": 75454
    },
    {
      "epoch": 1.3866424560546875e-06,
      "step": 75454,
      "training_step_time": 7.583690166473389
    },
    {
      "epoch": 1.3875961303710938e-06,
      "model_forward_time": 1.876424789428711,
      "step": 75455
    },
    {
      "epoch": 1.3875961303710938e-06,
      "step": 75455,
      "training_step_time": 6.7282397747039795
    },
    {
      "epoch": 1.3885498046875e-06,
      "model_forward_time": 1.9124014377593994,
      "step": 75456
    },
    {
      "epoch": 1.3885498046875e-06,
      "step": 75456,
      "training_step_time": 6.742508411407471
    },
    {
      "epoch": 1.3895034790039063e-06,
      "model_forward_time": 1.8703596591949463,
      "step": 75457
    },
    {
      "epoch": 1.3895034790039063e-06,
      "step": 75457,
      "training_step_time": 6.701273679733276
    },
    {
      "epoch": 1.3904571533203125e-06,
      "model_forward_time": 1.9169540405273438,
      "step": 75458
    },
    {
      "epoch": 1.3904571533203125e-06,
      "step": 75458,
      "training_step_time": 6.77565336227417
    },
    {
      "epoch": 1.3914108276367188e-06,
      "model_forward_time": 2.5515284538269043,
      "step": 75459
    },
    {
      "epoch": 1.3914108276367188e-06,
      "step": 75459,
      "training_step_time": 8.428120851516724
    },
    {
      "epoch": 1.392364501953125e-06,
      "grad_norm": 0.08194152265787125,
      "learning_rate": 1.5581471633923883e-06,
      "loss": 0.0546,
      "step": 75460
    },
    {
      "epoch": 1.392364501953125e-06,
      "model_forward_time": 1.905890941619873,
      "step": 75460
    },
    {
      "dynamics_loss_avg": 0.039167836867272855,
      "epoch": 1.392364501953125e-06,
      "step": 75460
    },
    {
      "action_loss_avg": 0.0010116350604221225,
      "epoch": 1.392364501953125e-06,
      "step": 75460
    },
    {
      "epoch": 1.392364501953125e-06,
      "step": 75460,
      "training_step_time": 9.835402488708496
    },
    {
      "epoch": 1.3933181762695313e-06,
      "model_forward_time": 1.9163963794708252,
      "step": 75461
    },
    {
      "epoch": 1.3933181762695313e-06,
      "step": 75461,
      "training_step_time": 8.783857107162476
    },
    {
      "epoch": 1.3942718505859375e-06,
      "model_forward_time": 1.8891842365264893,
      "step": 75462
    },
    {
      "epoch": 1.3942718505859375e-06,
      "step": 75462,
      "training_step_time": 8.43922233581543
    },
    {
      "epoch": 1.3952255249023438e-06,
      "model_forward_time": 1.9202520847320557,
      "step": 75463
    },
    {
      "epoch": 1.3952255249023438e-06,
      "step": 75463,
      "training_step_time": 7.642047882080078
    },
    {
      "epoch": 1.39617919921875e-06,
      "model_forward_time": 1.9203102588653564,
      "step": 75464
    },
    {
      "epoch": 1.39617919921875e-06,
      "step": 75464,
      "training_step_time": 7.20400333404541
    },
    {
      "epoch": 1.3971328735351563e-06,
      "model_forward_time": 1.9225685596466064,
      "step": 75465
    },
    {
      "epoch": 1.3971328735351563e-06,
      "step": 75465,
      "training_step_time": 6.725036382675171
    },
    {
      "epoch": 1.3980865478515625e-06,
      "model_forward_time": 1.908613681793213,
      "step": 75466
    },
    {
      "epoch": 1.3980865478515625e-06,
      "step": 75466,
      "training_step_time": 6.642107009887695
    },
    {
      "epoch": 1.3990402221679688e-06,
      "model_forward_time": 1.920095443725586,
      "step": 75467
    },
    {
      "epoch": 1.3990402221679688e-06,
      "step": 75467,
      "training_step_time": 6.654119968414307
    },
    {
      "epoch": 1.399993896484375e-06,
      "model_forward_time": 2.5035324096679688,
      "step": 75468
    },
    {
      "epoch": 1.399993896484375e-06,
      "step": 75468,
      "training_step_time": 8.130255460739136
    },
    {
      "epoch": 1.4009475708007813e-06,
      "model_forward_time": 1.9070782661437988,
      "step": 75469
    },
    {
      "epoch": 1.4009475708007813e-06,
      "step": 75469,
      "training_step_time": 10.021636009216309
    },
    {
      "epoch": 1.4019012451171875e-06,
      "grad_norm": 0.10505262762308121,
      "learning_rate": 1.5569479920759506e-06,
      "loss": 0.0552,
      "step": 75470
    },
    {
      "epoch": 1.4019012451171875e-06,
      "model_forward_time": 1.9065937995910645,
      "step": 75470
    },
    {
      "dynamics_loss_avg": 0.053437888994812964,
      "epoch": 1.4019012451171875e-06,
      "step": 75470
    },
    {
      "action_loss_avg": 0.0015970335283782334,
      "epoch": 1.4019012451171875e-06,
      "step": 75470
    },
    {
      "epoch": 1.4019012451171875e-06,
      "step": 75470,
      "training_step_time": 7.841733455657959
    },
    {
      "epoch": 1.4028549194335938e-06,
      "model_forward_time": 1.9078595638275146,
      "step": 75471
    },
    {
      "epoch": 1.4028549194335938e-06,
      "step": 75471,
      "training_step_time": 7.709172248840332
    },
    {
      "epoch": 1.40380859375e-06,
      "model_forward_time": 1.9153101444244385,
      "step": 75472
    },
    {
      "epoch": 1.40380859375e-06,
      "step": 75472,
      "training_step_time": 7.653980016708374
    },
    {
      "epoch": 1.4047622680664063e-06,
      "model_forward_time": 1.9073975086212158,
      "step": 75473
    },
    {
      "epoch": 1.4047622680664063e-06,
      "step": 75473,
      "training_step_time": 7.35841965675354
    },
    {
      "epoch": 1.4057159423828125e-06,
      "model_forward_time": 1.9105479717254639,
      "step": 75474
    },
    {
      "epoch": 1.4057159423828125e-06,
      "step": 75474,
      "training_step_time": 6.810359477996826
    },
    {
      "epoch": 1.4066696166992188e-06,
      "model_forward_time": 1.9138236045837402,
      "step": 75475
    },
    {
      "epoch": 1.4066696166992188e-06,
      "step": 75475,
      "training_step_time": 6.802520036697388
    },
    {
      "epoch": 1.407623291015625e-06,
      "model_forward_time": 1.9219636917114258,
      "step": 75476
    },
    {
      "epoch": 1.407623291015625e-06,
      "step": 75476,
      "training_step_time": 6.841522216796875
    },
    {
      "epoch": 1.4085769653320313e-06,
      "model_forward_time": 2.5206515789031982,
      "step": 75477
    },
    {
      "epoch": 1.4085769653320313e-06,
      "step": 75477,
      "training_step_time": 8.440847635269165
    },
    {
      "epoch": 1.4095306396484375e-06,
      "model_forward_time": 1.9282162189483643,
      "step": 75478
    },
    {
      "epoch": 1.4095306396484375e-06,
      "step": 75478,
      "training_step_time": 8.74001932144165
    },
    {
      "epoch": 1.4104843139648438e-06,
      "model_forward_time": 1.9131085872650146,
      "step": 75479
    },
    {
      "epoch": 1.4104843139648438e-06,
      "step": 75479,
      "training_step_time": 8.470357418060303
    },
    {
      "epoch": 1.41143798828125e-06,
      "grad_norm": 0.09461347758769989,
      "learning_rate": 1.5557491972865075e-06,
      "loss": 0.0552,
      "step": 75480
    },
    {
      "epoch": 1.41143798828125e-06,
      "model_forward_time": 1.9104363918304443,
      "step": 75480
    },
    {
      "dynamics_loss_avg": 0.05405491068959236,
      "epoch": 1.41143798828125e-06,
      "step": 75480
    },
    {
      "action_loss_avg": 0.0010263243282679468,
      "epoch": 1.41143798828125e-06,
      "step": 75480
    },
    {
      "epoch": 1.41143798828125e-06,
      "step": 75480,
      "training_step_time": 7.6597771644592285
    },
    {
      "epoch": 1.4123916625976563e-06,
      "model_forward_time": 1.9270656108856201,
      "step": 75481
    },
    {
      "epoch": 1.4123916625976563e-06,
      "step": 75481,
      "training_step_time": 7.836225509643555
    },
    {
      "epoch": 1.4133453369140625e-06,
      "model_forward_time": 1.9180223941802979,
      "step": 75482
    },
    {
      "epoch": 1.4133453369140625e-06,
      "step": 75482,
      "training_step_time": 7.3702404499053955
    },
    {
      "epoch": 1.4142990112304688e-06,
      "model_forward_time": 1.9127655029296875,
      "step": 75483
    },
    {
      "epoch": 1.4142990112304688e-06,
      "step": 75483,
      "training_step_time": 6.840351104736328
    },
    {
      "epoch": 1.415252685546875e-06,
      "model_forward_time": 1.8852431774139404,
      "step": 75484
    },
    {
      "epoch": 1.415252685546875e-06,
      "step": 75484,
      "training_step_time": 6.851110935211182
    },
    {
      "epoch": 1.4162063598632813e-06,
      "model_forward_time": 1.8966515064239502,
      "step": 75485
    },
    {
      "epoch": 1.4162063598632813e-06,
      "step": 75485,
      "training_step_time": 6.872944355010986
    },
    {
      "epoch": 1.4171600341796875e-06,
      "model_forward_time": 1.9154431819915771,
      "step": 75486
    },
    {
      "epoch": 1.4171600341796875e-06,
      "step": 75486,
      "training_step_time": 7.80706262588501
    },
    {
      "epoch": 1.4181137084960938e-06,
      "model_forward_time": 1.8744122982025146,
      "step": 75487
    },
    {
      "epoch": 1.4181137084960938e-06,
      "step": 75487,
      "training_step_time": 8.613527774810791
    },
    {
      "epoch": 1.4190673828125e-06,
      "model_forward_time": 1.915593147277832,
      "step": 75488
    },
    {
      "epoch": 1.4190673828125e-06,
      "step": 75488,
      "training_step_time": 10.159485340118408
    },
    {
      "epoch": 1.4200210571289063e-06,
      "model_forward_time": 1.900620460510254,
      "step": 75489
    },
    {
      "epoch": 1.4200210571289063e-06,
      "step": 75489,
      "training_step_time": 8.451061964035034
    },
    {
      "epoch": 1.4209747314453125e-06,
      "grad_norm": 0.09017148613929749,
      "learning_rate": 1.5545507791551523e-06,
      "loss": 0.0559,
      "step": 75490
    },
    {
      "epoch": 1.4209747314453125e-06,
      "model_forward_time": 1.9111676216125488,
      "step": 75490
    },
    {
      "dynamics_loss_avg": 0.06025780010968447,
      "epoch": 1.4209747314453125e-06,
      "step": 75490
    },
    {
      "action_loss_avg": 0.001811849107616581,
      "epoch": 1.4209747314453125e-06,
      "step": 75490
    },
    {
      "epoch": 1.4209747314453125e-06,
      "step": 75490,
      "training_step_time": 7.7340569496154785
    },
    {
      "epoch": 1.4219284057617188e-06,
      "model_forward_time": 1.908970594406128,
      "step": 75491
    },
    {
      "epoch": 1.4219284057617188e-06,
      "step": 75491,
      "training_step_time": 7.756549835205078
    },
    {
      "epoch": 1.422882080078125e-06,
      "model_forward_time": 1.9146771430969238,
      "step": 75492
    },
    {
      "epoch": 1.422882080078125e-06,
      "step": 75492,
      "training_step_time": 6.775726318359375
    },
    {
      "epoch": 1.4238357543945313e-06,
      "model_forward_time": 1.9110844135284424,
      "step": 75493
    },
    {
      "epoch": 1.4238357543945313e-06,
      "step": 75493,
      "training_step_time": 6.784595251083374
    },
    {
      "epoch": 1.4247894287109375e-06,
      "model_forward_time": 1.9056289196014404,
      "step": 75494
    },
    {
      "epoch": 1.4247894287109375e-06,
      "step": 75494,
      "training_step_time": 6.817546606063843
    },
    {
      "epoch": 1.4257431030273439e-06,
      "model_forward_time": 1.913888692855835,
      "step": 75495
    },
    {
      "epoch": 1.4257431030273439e-06,
      "step": 75495,
      "training_step_time": 6.799251556396484
    },
    {
      "epoch": 1.42669677734375e-06,
      "model_forward_time": 2.4810967445373535,
      "step": 75496
    },
    {
      "epoch": 1.42669677734375e-06,
      "step": 75496,
      "training_step_time": 7.7595884799957275
    },
    {
      "epoch": 1.4276504516601564e-06,
      "model_forward_time": 1.9142744541168213,
      "step": 75497
    },
    {
      "epoch": 1.4276504516601564e-06,
      "step": 75497,
      "training_step_time": 9.29520297050476
    },
    {
      "epoch": 1.4286041259765625e-06,
      "model_forward_time": 1.9065730571746826,
      "step": 75498
    },
    {
      "epoch": 1.4286041259765625e-06,
      "step": 75498,
      "training_step_time": 7.782045602798462
    },
    {
      "epoch": 1.4295578002929689e-06,
      "model_forward_time": 1.9210162162780762,
      "step": 75499
    },
    {
      "epoch": 1.4295578002929689e-06,
      "step": 75499,
      "training_step_time": 7.717021465301514
    },
    {
      "epoch": 1.430511474609375e-06,
      "grad_norm": 0.06718150526285172,
      "learning_rate": 1.5533527378129454e-06,
      "loss": 0.0549,
      "step": 75500
    },
    {
      "epoch": 1.430511474609375e-06,
      "model_forward_time": 1.9087998867034912,
      "step": 75500
    },
    {
      "dynamics_loss_avg": 0.04440638665109873,
      "epoch": 1.430511474609375e-06,
      "step": 75500
    },
    {
      "action_loss_avg": 0.0011442031944170594,
      "epoch": 1.430511474609375e-06,
      "step": 75500
    },
    {
      "epoch": 1.430511474609375e-06,
      "step": 75500,
      "training_step_time": 7.489699602127075
    },
    {
      "epoch": 1.4314651489257811e-06,
      "model_forward_time": 1.9216008186340332,
      "step": 75501
    },
    {
      "epoch": 1.4314651489257811e-06,
      "step": 75501,
      "training_step_time": 6.810356616973877
    },
    {
      "epoch": 1.4324188232421875e-06,
      "model_forward_time": 1.9160280227661133,
      "step": 75502
    },
    {
      "epoch": 1.4324188232421875e-06,
      "step": 75502,
      "training_step_time": 6.870922327041626
    },
    {
      "epoch": 1.4333724975585936e-06,
      "model_forward_time": 1.9005248546600342,
      "step": 75503
    },
    {
      "epoch": 1.4333724975585936e-06,
      "step": 75503,
      "training_step_time": 6.7040464878082275
    },
    {
      "epoch": 1.434326171875e-06,
      "model_forward_time": 1.9132564067840576,
      "step": 75504
    },
    {
      "epoch": 1.434326171875e-06,
      "step": 75504,
      "training_step_time": 6.695303201675415
    },
    {
      "epoch": 1.4352798461914061e-06,
      "model_forward_time": 2.5212955474853516,
      "step": 75505
    },
    {
      "epoch": 1.4352798461914061e-06,
      "step": 75505,
      "training_step_time": 7.632437467575073
    },
    {
      "epoch": 1.4362335205078125e-06,
      "model_forward_time": 1.9150176048278809,
      "step": 75506
    },
    {
      "epoch": 1.4362335205078125e-06,
      "step": 75506,
      "training_step_time": 11.124671936035156
    },
    {
      "epoch": 1.4371871948242187e-06,
      "model_forward_time": 1.906367540359497,
      "step": 75507
    },
    {
      "epoch": 1.4371871948242187e-06,
      "step": 75507,
      "training_step_time": 9.002819776535034
    },
    {
      "epoch": 1.438140869140625e-06,
      "model_forward_time": 1.917910099029541,
      "step": 75508
    },
    {
      "epoch": 1.438140869140625e-06,
      "step": 75508,
      "training_step_time": 7.605721473693848
    },
    {
      "epoch": 1.4390945434570312e-06,
      "model_forward_time": 1.9132201671600342,
      "step": 75509
    },
    {
      "epoch": 1.4390945434570312e-06,
      "step": 75509,
      "training_step_time": 8.384310483932495
    },
    {
      "epoch": 1.4400482177734375e-06,
      "grad_norm": 0.07942579686641693,
      "learning_rate": 1.552155073390899e-06,
      "loss": 0.055,
      "step": 75510
    },
    {
      "epoch": 1.4400482177734375e-06,
      "model_forward_time": 1.899644136428833,
      "step": 75510
    },
    {
      "dynamics_loss_avg": 0.04967900272458792,
      "epoch": 1.4400482177734375e-06,
      "step": 75510
    },
    {
      "action_loss_avg": 0.0010151063557714223,
      "epoch": 1.4400482177734375e-06,
      "step": 75510
    },
    {
      "epoch": 1.4400482177734375e-06,
      "step": 75510,
      "training_step_time": 6.763999938964844
    },
    {
      "epoch": 1.4410018920898437e-06,
      "model_forward_time": 1.9206573963165283,
      "step": 75511
    },
    {
      "epoch": 1.4410018920898437e-06,
      "step": 75511,
      "training_step_time": 6.799027442932129
    },
    {
      "epoch": 1.44195556640625e-06,
      "model_forward_time": 1.93031907081604,
      "step": 75512
    },
    {
      "epoch": 1.44195556640625e-06,
      "step": 75512,
      "training_step_time": 6.780996561050415
    },
    {
      "epoch": 1.4429092407226562e-06,
      "model_forward_time": 1.9200944900512695,
      "step": 75513
    },
    {
      "epoch": 1.4429092407226562e-06,
      "step": 75513,
      "training_step_time": 6.800853967666626
    },
    {
      "epoch": 1.4438629150390625e-06,
      "model_forward_time": 2.492192506790161,
      "step": 75514
    },
    {
      "epoch": 1.4438629150390625e-06,
      "step": 75514,
      "training_step_time": 7.715075969696045
    },
    {
      "epoch": 1.4448165893554687e-06,
      "model_forward_time": 1.8861920833587646,
      "step": 75515
    },
    {
      "epoch": 1.4448165893554687e-06,
      "step": 75515,
      "training_step_time": 11.39967679977417
    },
    {
      "epoch": 1.445770263671875e-06,
      "model_forward_time": 1.9252784252166748,
      "step": 75516
    },
    {
      "epoch": 1.445770263671875e-06,
      "step": 75516,
      "training_step_time": 8.541073560714722
    },
    {
      "epoch": 1.4467239379882812e-06,
      "model_forward_time": 1.912926197052002,
      "step": 75517
    },
    {
      "epoch": 1.4467239379882812e-06,
      "step": 75517,
      "training_step_time": 7.607503890991211
    },
    {
      "epoch": 1.4476776123046875e-06,
      "model_forward_time": 1.9140543937683105,
      "step": 75518
    },
    {
      "epoch": 1.4476776123046875e-06,
      "step": 75518,
      "training_step_time": 7.6119704246521
    },
    {
      "epoch": 1.4486312866210937e-06,
      "model_forward_time": 1.9048912525177002,
      "step": 75519
    },
    {
      "epoch": 1.4486312866210937e-06,
      "step": 75519,
      "training_step_time": 7.275588274002075
    },
    {
      "epoch": 1.4495849609375e-06,
      "grad_norm": 0.11072708666324615,
      "learning_rate": 1.5509577860199904e-06,
      "loss": 0.0557,
      "step": 75520
    },
    {
      "epoch": 1.4495849609375e-06,
      "model_forward_time": 1.9222891330718994,
      "step": 75520
    },
    {
      "dynamics_loss_avg": 0.07281572744250298,
      "epoch": 1.4495849609375e-06,
      "step": 75520
    },
    {
      "action_loss_avg": 0.001576655195094645,
      "epoch": 1.4495849609375e-06,
      "step": 75520
    },
    {
      "epoch": 1.4495849609375e-06,
      "step": 75520,
      "training_step_time": 6.792738199234009
    },
    {
      "epoch": 1.4505386352539062e-06,
      "model_forward_time": 1.9154505729675293,
      "step": 75521
    },
    {
      "epoch": 1.4505386352539062e-06,
      "step": 75521,
      "training_step_time": 6.805202960968018
    },
    {
      "epoch": 1.4514923095703125e-06,
      "model_forward_time": 1.9199011325836182,
      "step": 75522
    },
    {
      "epoch": 1.4514923095703125e-06,
      "step": 75522,
      "training_step_time": 6.817276239395142
    },
    {
      "epoch": 1.4524459838867187e-06,
      "model_forward_time": 2.5391457080841064,
      "step": 75523
    },
    {
      "epoch": 1.4524459838867187e-06,
      "step": 75523,
      "training_step_time": 7.644193172454834
    },
    {
      "epoch": 1.453399658203125e-06,
      "model_forward_time": 1.9176311492919922,
      "step": 75524
    },
    {
      "epoch": 1.453399658203125e-06,
      "step": 75524,
      "training_step_time": 9.236621618270874
    },
    {
      "epoch": 1.4543533325195312e-06,
      "model_forward_time": 1.9162800312042236,
      "step": 75525
    },
    {
      "epoch": 1.4543533325195312e-06,
      "step": 75525,
      "training_step_time": 8.56201982498169
    },
    {
      "epoch": 1.4553070068359375e-06,
      "model_forward_time": 1.7828068733215332,
      "step": 75526
    },
    {
      "epoch": 1.4553070068359375e-06,
      "step": 75526,
      "training_step_time": 7.614923715591431
    },
    {
      "epoch": 1.4562606811523437e-06,
      "model_forward_time": 1.7824022769927979,
      "step": 75527
    },
    {
      "epoch": 1.4562606811523437e-06,
      "step": 75527,
      "training_step_time": 8.59244418144226
    },
    {
      "epoch": 1.45721435546875e-06,
      "model_forward_time": 1.7881286144256592,
      "step": 75528
    },
    {
      "epoch": 1.45721435546875e-06,
      "step": 75528,
      "training_step_time": 7.339592218399048
    },
    {
      "epoch": 1.4581680297851562e-06,
      "model_forward_time": 1.7839136123657227,
      "step": 75529
    },
    {
      "epoch": 1.4581680297851562e-06,
      "step": 75529,
      "training_step_time": 6.787773132324219
    },
    {
      "epoch": 1.4591217041015625e-06,
      "grad_norm": 0.09582237899303436,
      "learning_rate": 1.5497608758311528e-06,
      "loss": 0.0552,
      "step": 75530
    },
    {
      "epoch": 1.4591217041015625e-06,
      "model_forward_time": 1.774355173110962,
      "step": 75530
    },
    {
      "dynamics_loss_avg": 0.03904153648763895,
      "epoch": 1.4591217041015625e-06,
      "step": 75530
    },
    {
      "action_loss_avg": 0.0009251449257135392,
      "epoch": 1.4591217041015625e-06,
      "step": 75530
    },
    {
      "epoch": 1.4591217041015625e-06,
      "step": 75530,
      "training_step_time": 6.802402496337891
    },
    {
      "epoch": 1.4600753784179687e-06,
      "model_forward_time": 1.7787137031555176,
      "step": 75531
    },
    {
      "epoch": 1.4600753784179687e-06,
      "step": 75531,
      "training_step_time": 6.777240753173828
    },
    {
      "epoch": 1.461029052734375e-06,
      "model_forward_time": 2.4327378273010254,
      "step": 75532
    },
    {
      "epoch": 1.461029052734375e-06,
      "step": 75532,
      "training_step_time": 7.638882398605347
    },
    {
      "epoch": 1.4619827270507812e-06,
      "model_forward_time": 1.8311398029327393,
      "step": 75533
    },
    {
      "epoch": 1.4619827270507812e-06,
      "step": 75533,
      "training_step_time": 9.16060996055603
    },
    {
      "epoch": 1.4629364013671875e-06,
      "model_forward_time": 1.7794842720031738,
      "step": 75534
    },
    {
      "epoch": 1.4629364013671875e-06,
      "step": 75534,
      "training_step_time": 8.38059139251709
    },
    {
      "epoch": 1.4638900756835937e-06,
      "model_forward_time": 1.7782399654388428,
      "step": 75535
    },
    {
      "epoch": 1.4638900756835937e-06,
      "step": 75535,
      "training_step_time": 8.680538177490234
    },
    {
      "epoch": 1.46484375e-06,
      "model_forward_time": 1.7797932624816895,
      "step": 75536
    },
    {
      "epoch": 1.46484375e-06,
      "step": 75536,
      "training_step_time": 7.6199986934661865
    },
    {
      "epoch": 1.4657974243164062e-06,
      "model_forward_time": 1.7870585918426514,
      "step": 75537
    },
    {
      "epoch": 1.4657974243164062e-06,
      "step": 75537,
      "training_step_time": 7.632271766662598
    },
    {
      "epoch": 1.4667510986328125e-06,
      "model_forward_time": 1.7899937629699707,
      "step": 75538
    },
    {
      "epoch": 1.4667510986328125e-06,
      "step": 75538,
      "training_step_time": 6.762333393096924
    },
    {
      "epoch": 1.4677047729492187e-06,
      "model_forward_time": 1.7610366344451904,
      "step": 75539
    },
    {
      "epoch": 1.4677047729492187e-06,
      "step": 75539,
      "training_step_time": 6.760649681091309
    },
    {
      "epoch": 1.468658447265625e-06,
      "grad_norm": 0.08601561188697815,
      "learning_rate": 1.5485643429552799e-06,
      "loss": 0.0561,
      "step": 75540
    },
    {
      "epoch": 1.468658447265625e-06,
      "model_forward_time": 1.7756421566009521,
      "step": 75540
    },
    {
      "dynamics_loss_avg": 0.03817512225359678,
      "epoch": 1.468658447265625e-06,
      "step": 75540
    },
    {
      "action_loss_avg": 0.001407846505753696,
      "epoch": 1.468658447265625e-06,
      "step": 75540
    },
    {
      "epoch": 1.468658447265625e-06,
      "step": 75540,
      "training_step_time": 6.759409189224243
    },
    {
      "epoch": 1.4696121215820312e-06,
      "model_forward_time": 1.7691361904144287,
      "step": 75541
    },
    {
      "epoch": 1.4696121215820312e-06,
      "step": 75541,
      "training_step_time": 9.43183970451355
    },
    {
      "epoch": 1.4705657958984375e-06,
      "model_forward_time": 1.7966527938842773,
      "step": 75542
    },
    {
      "epoch": 1.4705657958984375e-06,
      "step": 75542,
      "training_step_time": 7.80312705039978
    },
    {
      "epoch": 1.4715194702148437e-06,
      "model_forward_time": 1.7793703079223633,
      "step": 75543
    },
    {
      "epoch": 1.4715194702148437e-06,
      "step": 75543,
      "training_step_time": 9.763944149017334
    },
    {
      "epoch": 1.47247314453125e-06,
      "model_forward_time": 1.7776639461517334,
      "step": 75544
    },
    {
      "epoch": 1.47247314453125e-06,
      "step": 75544,
      "training_step_time": 7.885643243789673
    },
    {
      "epoch": 1.4734268188476562e-06,
      "model_forward_time": 1.7856645584106445,
      "step": 75545
    },
    {
      "epoch": 1.4734268188476562e-06,
      "step": 75545,
      "training_step_time": 7.6994733810424805
    },
    {
      "epoch": 1.4743804931640625e-06,
      "model_forward_time": 1.7681872844696045,
      "step": 75546
    },
    {
      "epoch": 1.4743804931640625e-06,
      "step": 75546,
      "training_step_time": 7.771615982055664
    },
    {
      "epoch": 1.4753341674804687e-06,
      "model_forward_time": 1.7735424041748047,
      "step": 75547
    },
    {
      "epoch": 1.4753341674804687e-06,
      "step": 75547,
      "training_step_time": 7.2317054271698
    },
    {
      "epoch": 1.476287841796875e-06,
      "model_forward_time": 1.4106276035308838,
      "step": 75548
    },
    {
      "epoch": 1.476287841796875e-06,
      "step": 75548,
      "training_step_time": 6.775413274765015
    },
    {
      "epoch": 1.4772415161132812e-06,
      "model_forward_time": 1.7692368030548096,
      "step": 75549
    },
    {
      "epoch": 1.4772415161132812e-06,
      "step": 75549,
      "training_step_time": 6.795668601989746
    },
    {
      "epoch": 1.4781951904296875e-06,
      "grad_norm": 0.07751932740211487,
      "learning_rate": 1.54736818752322e-06,
      "loss": 0.0567,
      "step": 75550
    },
    {
      "epoch": 1.4781951904296875e-06,
      "model_forward_time": 1.7639474868774414,
      "step": 75550
    },
    {
      "dynamics_loss_avg": 0.04429041389375925,
      "epoch": 1.4781951904296875e-06,
      "step": 75550
    },
    {
      "action_loss_avg": 0.0016609230893664061,
      "epoch": 1.4781951904296875e-06,
      "step": 75550
    },
    {
      "epoch": 1.4781951904296875e-06,
      "step": 75550,
      "training_step_time": 7.595944404602051
    },
    {
      "epoch": 1.4791488647460937e-06,
      "model_forward_time": 2.3792929649353027,
      "step": 75551
    },
    {
      "epoch": 1.4791488647460937e-06,
      "step": 75551,
      "training_step_time": 8.623084545135498
    },
    {
      "epoch": 1.4801025390625e-06,
      "model_forward_time": 1.8344886302947998,
      "step": 75552
    },
    {
      "epoch": 1.4801025390625e-06,
      "step": 75552,
      "training_step_time": 8.42159914970398
    },
    {
      "epoch": 1.4810562133789062e-06,
      "model_forward_time": 1.7721481323242188,
      "step": 75553
    },
    {
      "epoch": 1.4810562133789062e-06,
      "step": 75553,
      "training_step_time": 9.842949867248535
    },
    {
      "epoch": 1.4820098876953125e-06,
      "model_forward_time": 1.7762901782989502,
      "step": 75554
    },
    {
      "epoch": 1.4820098876953125e-06,
      "step": 75554,
      "training_step_time": 7.776516437530518
    },
    {
      "epoch": 1.4829635620117187e-06,
      "model_forward_time": 1.7745616436004639,
      "step": 75555
    },
    {
      "epoch": 1.4829635620117187e-06,
      "step": 75555,
      "training_step_time": 7.719675779342651
    },
    {
      "epoch": 1.483917236328125e-06,
      "model_forward_time": 1.7708005905151367,
      "step": 75556
    },
    {
      "epoch": 1.483917236328125e-06,
      "step": 75556,
      "training_step_time": 7.234121322631836
    },
    {
      "epoch": 1.4848709106445312e-06,
      "model_forward_time": 1.7725646495819092,
      "step": 75557
    },
    {
      "epoch": 1.4848709106445312e-06,
      "step": 75557,
      "training_step_time": 6.815695762634277
    },
    {
      "epoch": 1.4858245849609375e-06,
      "model_forward_time": 1.7681059837341309,
      "step": 75558
    },
    {
      "epoch": 1.4858245849609375e-06,
      "step": 75558,
      "training_step_time": 6.824651479721069
    },
    {
      "epoch": 1.4867782592773437e-06,
      "model_forward_time": 1.766442060470581,
      "step": 75559
    },
    {
      "epoch": 1.4867782592773437e-06,
      "step": 75559,
      "training_step_time": 7.285816431045532
    },
    {
      "epoch": 1.48773193359375e-06,
      "grad_norm": 0.0877600610256195,
      "learning_rate": 1.5461724096657837e-06,
      "loss": 0.0561,
      "step": 75560
    },
    {
      "epoch": 1.48773193359375e-06,
      "model_forward_time": 2.427921772003174,
      "step": 75560
    },
    {
      "dynamics_loss_avg": 0.04550656452775002,
      "epoch": 1.48773193359375e-06,
      "step": 75560
    },
    {
      "action_loss_avg": 0.0012842269672546536,
      "epoch": 1.48773193359375e-06,
      "step": 75560
    },
    {
      "epoch": 1.48773193359375e-06,
      "step": 75560,
      "training_step_time": 8.798535108566284
    },
    {
      "epoch": 1.4886856079101562e-06,
      "model_forward_time": 1.8096437454223633,
      "step": 75561
    },
    {
      "epoch": 1.4886856079101562e-06,
      "step": 75561,
      "training_step_time": 10.957884073257446
    },
    {
      "epoch": 1.4896392822265626e-06,
      "model_forward_time": 1.7510387897491455,
      "step": 75562
    },
    {
      "epoch": 1.4896392822265626e-06,
      "step": 75562,
      "training_step_time": 8.530806064605713
    },
    {
      "epoch": 1.4905929565429687e-06,
      "model_forward_time": 1.7618069648742676,
      "step": 75563
    },
    {
      "epoch": 1.4905929565429687e-06,
      "step": 75563,
      "training_step_time": 7.730596542358398
    },
    {
      "epoch": 1.491546630859375e-06,
      "model_forward_time": 1.7587339878082275,
      "step": 75564
    },
    {
      "epoch": 1.491546630859375e-06,
      "step": 75564,
      "training_step_time": 8.59549069404602
    },
    {
      "epoch": 1.4925003051757812e-06,
      "model_forward_time": 1.760498046875,
      "step": 75565
    },
    {
      "epoch": 1.4925003051757812e-06,
      "step": 75565,
      "training_step_time": 7.333035469055176
    },
    {
      "epoch": 1.4934539794921876e-06,
      "model_forward_time": 1.7732677459716797,
      "step": 75566
    },
    {
      "epoch": 1.4934539794921876e-06,
      "step": 75566,
      "training_step_time": 6.902746915817261
    },
    {
      "epoch": 1.4944076538085937e-06,
      "model_forward_time": 1.7600455284118652,
      "step": 75567
    },
    {
      "epoch": 1.4944076538085937e-06,
      "step": 75567,
      "training_step_time": 6.820919513702393
    },
    {
      "epoch": 1.495361328125e-06,
      "model_forward_time": 1.75980544090271,
      "step": 75568
    },
    {
      "epoch": 1.495361328125e-06,
      "step": 75568,
      "training_step_time": 7.722904682159424
    },
    {
      "epoch": 1.4963150024414062e-06,
      "model_forward_time": 2.427361249923706,
      "step": 75569
    },
    {
      "epoch": 1.4963150024414062e-06,
      "step": 75569,
      "training_step_time": 7.845301628112793
    },
    {
      "epoch": 1.4972686767578126e-06,
      "grad_norm": 0.09315545856952667,
      "learning_rate": 1.5449770095137417e-06,
      "loss": 0.0574,
      "step": 75570
    },
    {
      "epoch": 1.4972686767578126e-06,
      "model_forward_time": 1.8332321643829346,
      "step": 75570
    },
    {
      "dynamics_loss_avg": 0.05416497513651848,
      "epoch": 1.4972686767578126e-06,
      "step": 75570
    },
    {
      "action_loss_avg": 0.0014829883293714374,
      "epoch": 1.4972686767578126e-06,
      "step": 75570
    },
    {
      "epoch": 1.4972686767578126e-06,
      "step": 75570,
      "training_step_time": 9.61579155921936
    },
    {
      "epoch": 1.4982223510742187e-06,
      "model_forward_time": 1.7641985416412354,
      "step": 75571
    },
    {
      "epoch": 1.4982223510742187e-06,
      "step": 75571,
      "training_step_time": 7.855909585952759
    },
    {
      "epoch": 1.499176025390625e-06,
      "model_forward_time": 1.7760207653045654,
      "step": 75572
    },
    {
      "epoch": 1.499176025390625e-06,
      "step": 75572,
      "training_step_time": 7.620495319366455
    },
    {
      "epoch": 1.5001296997070312e-06,
      "model_forward_time": 1.7796258926391602,
      "step": 75573
    },
    {
      "epoch": 1.5001296997070312e-06,
      "step": 75573,
      "training_step_time": 7.7300074100494385
    },
    {
      "epoch": 1.5010833740234376e-06,
      "model_forward_time": 1.769181251525879,
      "step": 75574
    },
    {
      "epoch": 1.5010833740234376e-06,
      "step": 75574,
      "training_step_time": 7.218076229095459
    },
    {
      "epoch": 1.5020370483398437e-06,
      "model_forward_time": 1.7648406028747559,
      "step": 75575
    },
    {
      "epoch": 1.5020370483398437e-06,
      "step": 75575,
      "training_step_time": 6.81234335899353
    },
    {
      "epoch": 1.50299072265625e-06,
      "model_forward_time": 1.764350414276123,
      "step": 75576
    },
    {
      "epoch": 1.50299072265625e-06,
      "step": 75576,
      "training_step_time": 6.792388200759888
    },
    {
      "epoch": 1.5039443969726562e-06,
      "model_forward_time": 1.770136833190918,
      "step": 75577
    },
    {
      "epoch": 1.5039443969726562e-06,
      "step": 75577,
      "training_step_time": 7.702345609664917
    },
    {
      "epoch": 1.5048980712890626e-06,
      "model_forward_time": 2.4139292240142822,
      "step": 75578
    },
    {
      "epoch": 1.5048980712890626e-06,
      "step": 75578,
      "training_step_time": 7.7963666915893555
    },
    {
      "epoch": 1.5058517456054687e-06,
      "model_forward_time": 1.828132152557373,
      "step": 75579
    },
    {
      "epoch": 1.5058517456054687e-06,
      "step": 75579,
      "training_step_time": 7.8487255573272705
    },
    {
      "epoch": 1.506805419921875e-06,
      "grad_norm": 0.10719102621078491,
      "learning_rate": 1.5437819871978177e-06,
      "loss": 0.0572,
      "step": 75580
    },
    {
      "epoch": 1.506805419921875e-06,
      "model_forward_time": 1.7626354694366455,
      "step": 75580
    },
    {
      "dynamics_loss_avg": 0.0543232187628746,
      "epoch": 1.506805419921875e-06,
      "step": 75580
    },
    {
      "action_loss_avg": 0.001662776037119329,
      "epoch": 1.506805419921875e-06,
      "step": 75580
    },
    {
      "epoch": 1.506805419921875e-06,
      "step": 75580,
      "training_step_time": 9.586477518081665
    },
    {
      "epoch": 1.5077590942382812e-06,
      "model_forward_time": 1.7710137367248535,
      "step": 75581
    },
    {
      "epoch": 1.5077590942382812e-06,
      "step": 75581,
      "training_step_time": 9.403130769729614
    },
    {
      "epoch": 1.5087127685546876e-06,
      "model_forward_time": 1.7765185832977295,
      "step": 75582
    },
    {
      "epoch": 1.5087127685546876e-06,
      "step": 75582,
      "training_step_time": 7.6731178760528564
    },
    {
      "epoch": 1.5096664428710937e-06,
      "model_forward_time": 1.7720921039581299,
      "step": 75583
    },
    {
      "epoch": 1.5096664428710937e-06,
      "step": 75583,
      "training_step_time": 7.69000244140625
    },
    {
      "epoch": 1.5106201171875e-06,
      "model_forward_time": 1.7759201526641846,
      "step": 75584
    },
    {
      "epoch": 1.5106201171875e-06,
      "step": 75584,
      "training_step_time": 6.8301331996917725
    },
    {
      "epoch": 1.5115737915039062e-06,
      "model_forward_time": 1.7645087242126465,
      "step": 75585
    },
    {
      "epoch": 1.5115737915039062e-06,
      "step": 75585,
      "training_step_time": 6.831259727478027
    },
    {
      "epoch": 1.5125274658203126e-06,
      "model_forward_time": 1.7757861614227295,
      "step": 75586
    },
    {
      "epoch": 1.5125274658203126e-06,
      "step": 75586,
      "training_step_time": 7.6756932735443115
    },
    {
      "epoch": 1.5134811401367187e-06,
      "model_forward_time": 1.7710211277008057,
      "step": 75587
    },
    {
      "epoch": 1.5134811401367187e-06,
      "step": 75587,
      "training_step_time": 9.219472408294678
    },
    {
      "epoch": 1.514434814453125e-06,
      "model_forward_time": 1.8108675479888916,
      "step": 75588
    },
    {
      "epoch": 1.514434814453125e-06,
      "step": 75588,
      "training_step_time": 8.656530380249023
    },
    {
      "epoch": 1.5153884887695312e-06,
      "model_forward_time": 1.7661709785461426,
      "step": 75589
    },
    {
      "epoch": 1.5153884887695312e-06,
      "step": 75589,
      "training_step_time": 8.335038900375366
    },
    {
      "epoch": 1.5163421630859376e-06,
      "grad_norm": 0.09281141310930252,
      "learning_rate": 1.5425873428486994e-06,
      "loss": 0.0573,
      "step": 75590
    },
    {
      "epoch": 1.5163421630859376e-06,
      "model_forward_time": 1.770411729812622,
      "step": 75590
    },
    {
      "dynamics_loss_avg": 0.04146452900022268,
      "epoch": 1.5163421630859376e-06,
      "step": 75590
    },
    {
      "action_loss_avg": 0.0015455640386790037,
      "epoch": 1.5163421630859376e-06,
      "step": 75590
    },
    {
      "epoch": 1.5163421630859376e-06,
      "step": 75590,
      "training_step_time": 9.242840051651001
    },
    {
      "epoch": 1.5172958374023437e-06,
      "model_forward_time": 1.767383337020874,
      "step": 75591
    },
    {
      "epoch": 1.5172958374023437e-06,
      "step": 75591,
      "training_step_time": 7.673362493515015
    },
    {
      "epoch": 1.51824951171875e-06,
      "model_forward_time": 1.7692017555236816,
      "step": 75592
    },
    {
      "epoch": 1.51824951171875e-06,
      "step": 75592,
      "training_step_time": 7.385143756866455
    },
    {
      "epoch": 1.5192031860351562e-06,
      "model_forward_time": 1.7661018371582031,
      "step": 75593
    },
    {
      "epoch": 1.5192031860351562e-06,
      "step": 75593,
      "training_step_time": 7.179265260696411
    },
    {
      "epoch": 1.5201568603515626e-06,
      "model_forward_time": 1.7654938697814941,
      "step": 75594
    },
    {
      "epoch": 1.5201568603515626e-06,
      "step": 75594,
      "training_step_time": 6.823463439941406
    },
    {
      "epoch": 1.5211105346679687e-06,
      "model_forward_time": 1.3988027572631836,
      "step": 75595
    },
    {
      "epoch": 1.5211105346679687e-06,
      "step": 75595,
      "training_step_time": 7.5991528034210205
    },
    {
      "epoch": 1.522064208984375e-06,
      "model_forward_time": 1.8400611877441406,
      "step": 75596
    },
    {
      "epoch": 1.522064208984375e-06,
      "step": 75596,
      "training_step_time": 7.717493772506714
    },
    {
      "epoch": 1.5230178833007812e-06,
      "model_forward_time": 2.4062962532043457,
      "step": 75597
    },
    {
      "epoch": 1.5230178833007812e-06,
      "step": 75597,
      "training_step_time": 8.433861255645752
    },
    {
      "epoch": 1.5239715576171876e-06,
      "model_forward_time": 1.7670705318450928,
      "step": 75598
    },
    {
      "epoch": 1.5239715576171876e-06,
      "step": 75598,
      "training_step_time": 8.792094707489014
    },
    {
      "epoch": 1.5249252319335937e-06,
      "model_forward_time": 1.7552361488342285,
      "step": 75599
    },
    {
      "epoch": 1.5249252319335937e-06,
      "step": 75599,
      "training_step_time": 8.517115831375122
    },
    {
      "epoch": 1.52587890625e-06,
      "grad_norm": 0.10316798835992813,
      "learning_rate": 1.5413930765970314e-06,
      "loss": 0.0575,
      "step": 75600
    },
    {
      "epoch": 1.52587890625e-06,
      "model_forward_time": 1.7556235790252686,
      "step": 75600
    },
    {
      "dynamics_loss_avg": 0.05272432491183281,
      "epoch": 1.52587890625e-06,
      "step": 75600
    },
    {
      "action_loss_avg": 0.001716957305325195,
      "epoch": 1.52587890625e-06,
      "step": 75600
    },
    {
      "epoch": 1.52587890625e-06,
      "step": 75600,
      "training_step_time": 7.675093650817871
    },
    {
      "epoch": 1.5268325805664062e-06,
      "model_forward_time": 1.7671775817871094,
      "step": 75601
    },
    {
      "epoch": 1.5268325805664062e-06,
      "step": 75601,
      "training_step_time": 7.347577095031738
    },
    {
      "epoch": 1.5277862548828126e-06,
      "model_forward_time": 1.7707970142364502,
      "step": 75602
    },
    {
      "epoch": 1.5277862548828126e-06,
      "step": 75602,
      "training_step_time": 7.214577913284302
    },
    {
      "epoch": 1.5287399291992187e-06,
      "model_forward_time": 1.7614660263061523,
      "step": 75603
    },
    {
      "epoch": 1.5287399291992187e-06,
      "step": 75603,
      "training_step_time": 6.750943422317505
    },
    {
      "epoch": 1.529693603515625e-06,
      "model_forward_time": 1.7736291885375977,
      "step": 75604
    },
    {
      "epoch": 1.529693603515625e-06,
      "step": 75604,
      "training_step_time": 6.694939374923706
    },
    {
      "epoch": 1.5306472778320312e-06,
      "model_forward_time": 1.7800145149230957,
      "step": 75605
    },
    {
      "epoch": 1.5306472778320312e-06,
      "step": 75605,
      "training_step_time": 7.780342102050781
    },
    {
      "epoch": 1.5316009521484376e-06,
      "model_forward_time": 2.3873016834259033,
      "step": 75606
    },
    {
      "epoch": 1.5316009521484376e-06,
      "step": 75606,
      "training_step_time": 8.889126777648926
    },
    {
      "epoch": 1.5325546264648437e-06,
      "model_forward_time": 1.8193109035491943,
      "step": 75607
    },
    {
      "epoch": 1.5325546264648437e-06,
      "step": 75607,
      "training_step_time": 7.716259717941284
    },
    {
      "epoch": 1.53350830078125e-06,
      "model_forward_time": 1.7658207416534424,
      "step": 75608
    },
    {
      "epoch": 1.53350830078125e-06,
      "step": 75608,
      "training_step_time": 8.838446378707886
    },
    {
      "epoch": 1.5344619750976562e-06,
      "model_forward_time": 1.7620389461517334,
      "step": 75609
    },
    {
      "epoch": 1.5344619750976562e-06,
      "step": 75609,
      "training_step_time": 7.585080623626709
    },
    {
      "epoch": 1.5354156494140626e-06,
      "grad_norm": 0.0923752412199974,
      "learning_rate": 1.5401991885734175e-06,
      "loss": 0.0558,
      "step": 75610
    },
    {
      "epoch": 1.5354156494140626e-06,
      "model_forward_time": 1.7563157081604004,
      "step": 75610
    },
    {
      "dynamics_loss_avg": 0.05654667392373085,
      "epoch": 1.5354156494140626e-06,
      "step": 75610
    },
    {
      "action_loss_avg": 0.0011577036639209837,
      "epoch": 1.5354156494140626e-06,
      "step": 75610
    },
    {
      "epoch": 1.5354156494140626e-06,
      "step": 75610,
      "training_step_time": 7.268825054168701
    },
    {
      "epoch": 1.5363693237304687e-06,
      "model_forward_time": 1.7604575157165527,
      "step": 75611
    },
    {
      "epoch": 1.5363693237304687e-06,
      "step": 75611,
      "training_step_time": 7.221964120864868
    },
    {
      "epoch": 1.537322998046875e-06,
      "model_forward_time": 1.7654759883880615,
      "step": 75612
    },
    {
      "epoch": 1.537322998046875e-06,
      "step": 75612,
      "training_step_time": 6.7145912647247314
    },
    {
      "epoch": 1.5382766723632812e-06,
      "model_forward_time": 1.769240140914917,
      "step": 75613
    },
    {
      "epoch": 1.5382766723632812e-06,
      "step": 75613,
      "training_step_time": 6.738190412521362
    },
    {
      "epoch": 1.5392303466796876e-06,
      "model_forward_time": 1.7485108375549316,
      "step": 75614
    },
    {
      "epoch": 1.5392303466796876e-06,
      "step": 75614,
      "training_step_time": 7.595963716506958
    },
    {
      "epoch": 1.5401840209960937e-06,
      "model_forward_time": 2.4284424781799316,
      "step": 75615
    },
    {
      "epoch": 1.5401840209960937e-06,
      "step": 75615,
      "training_step_time": 8.566498279571533
    },
    {
      "epoch": 1.5411376953125e-06,
      "model_forward_time": 1.804161548614502,
      "step": 75616
    },
    {
      "epoch": 1.5411376953125e-06,
      "step": 75616,
      "training_step_time": 8.373699188232422
    },
    {
      "epoch": 1.5420913696289062e-06,
      "model_forward_time": 1.7545816898345947,
      "step": 75617
    },
    {
      "epoch": 1.5420913696289062e-06,
      "step": 75617,
      "training_step_time": 10.118006706237793
    },
    {
      "epoch": 1.5430450439453126e-06,
      "model_forward_time": 1.7503337860107422,
      "step": 75618
    },
    {
      "epoch": 1.5430450439453126e-06,
      "step": 75618,
      "training_step_time": 8.688379764556885
    },
    {
      "epoch": 1.5439987182617187e-06,
      "model_forward_time": 1.7665998935699463,
      "step": 75619
    },
    {
      "epoch": 1.5439987182617187e-06,
      "step": 75619,
      "training_step_time": 7.639055490493774
    },
    {
      "epoch": 1.544952392578125e-06,
      "grad_norm": 0.08816166967153549,
      "learning_rate": 1.5390056789084173e-06,
      "loss": 0.0577,
      "step": 75620
    },
    {
      "epoch": 1.544952392578125e-06,
      "model_forward_time": 1.767564296722412,
      "step": 75620
    },
    {
      "dynamics_loss_avg": 0.04657849296927452,
      "epoch": 1.544952392578125e-06,
      "step": 75620
    },
    {
      "action_loss_avg": 0.001197584776673466,
      "epoch": 1.544952392578125e-06,
      "step": 75620
    },
    {
      "epoch": 1.544952392578125e-06,
      "step": 75620,
      "training_step_time": 7.968286514282227
    },
    {
      "epoch": 1.5459060668945312e-06,
      "model_forward_time": 1.7706897258758545,
      "step": 75621
    },
    {
      "epoch": 1.5459060668945312e-06,
      "step": 75621,
      "training_step_time": 7.1077797412872314
    },
    {
      "epoch": 1.5468597412109376e-06,
      "model_forward_time": 1.765920877456665,
      "step": 75622
    },
    {
      "epoch": 1.5468597412109376e-06,
      "step": 75622,
      "training_step_time": 6.746456146240234
    },
    {
      "epoch": 1.5478134155273437e-06,
      "model_forward_time": 1.7588446140289307,
      "step": 75623
    },
    {
      "epoch": 1.5478134155273437e-06,
      "step": 75623,
      "training_step_time": 7.574357509613037
    },
    {
      "epoch": 1.54876708984375e-06,
      "model_forward_time": 2.41829252243042,
      "step": 75624
    },
    {
      "epoch": 1.54876708984375e-06,
      "step": 75624,
      "training_step_time": 8.034980058670044
    },
    {
      "epoch": 1.5497207641601562e-06,
      "model_forward_time": 1.7773549556732178,
      "step": 75625
    },
    {
      "epoch": 1.5497207641601562e-06,
      "step": 75625,
      "training_step_time": 9.268962383270264
    },
    {
      "epoch": 1.5506744384765624e-06,
      "model_forward_time": 1.8153364658355713,
      "step": 75626
    },
    {
      "epoch": 1.5506744384765624e-06,
      "step": 75626,
      "training_step_time": 8.892130136489868
    },
    {
      "epoch": 1.5516281127929688e-06,
      "model_forward_time": 1.7664527893066406,
      "step": 75627
    },
    {
      "epoch": 1.5516281127929688e-06,
      "step": 75627,
      "training_step_time": 9.80200481414795
    },
    {
      "epoch": 1.552581787109375e-06,
      "model_forward_time": 1.7713556289672852,
      "step": 75628
    },
    {
      "epoch": 1.552581787109375e-06,
      "step": 75628,
      "training_step_time": 7.623563051223755
    },
    {
      "epoch": 1.5535354614257813e-06,
      "model_forward_time": 1.7708375453948975,
      "step": 75629
    },
    {
      "epoch": 1.5535354614257813e-06,
      "step": 75629,
      "training_step_time": 7.38765287399292
    },
    {
      "epoch": 1.5544891357421874e-06,
      "grad_norm": 0.14114555716514587,
      "learning_rate": 1.5378125477325545e-06,
      "loss": 0.0572,
      "step": 75630
    },
    {
      "epoch": 1.5544891357421874e-06,
      "model_forward_time": 1.7747225761413574,
      "step": 75630
    },
    {
      "dynamics_loss_avg": 0.044398941285908225,
      "epoch": 1.5544891357421874e-06,
      "step": 75630
    },
    {
      "action_loss_avg": 0.0011619373166467995,
      "epoch": 1.5544891357421874e-06,
      "step": 75630
    },
    {
      "epoch": 1.5544891357421874e-06,
      "step": 75630,
      "training_step_time": 7.249570369720459
    },
    {
      "epoch": 1.5554428100585938e-06,
      "model_forward_time": 1.7669150829315186,
      "step": 75631
    },
    {
      "epoch": 1.5554428100585938e-06,
      "step": 75631,
      "training_step_time": 6.829696178436279
    },
    {
      "epoch": 1.556396484375e-06,
      "model_forward_time": 1.7600784301757812,
      "step": 75632
    },
    {
      "epoch": 1.556396484375e-06,
      "step": 75632,
      "training_step_time": 7.286427736282349
    },
    {
      "epoch": 1.5573501586914063e-06,
      "model_forward_time": 1.771789789199829,
      "step": 75633
    },
    {
      "epoch": 1.5573501586914063e-06,
      "step": 75633,
      "training_step_time": 8.710937738418579
    },
    {
      "epoch": 1.5583038330078124e-06,
      "model_forward_time": 1.7826769351959229,
      "step": 75634
    },
    {
      "epoch": 1.5583038330078124e-06,
      "step": 75634,
      "training_step_time": 8.592663764953613
    },
    {
      "epoch": 1.5592575073242188e-06,
      "model_forward_time": 1.8123631477355957,
      "step": 75635
    },
    {
      "epoch": 1.5592575073242188e-06,
      "step": 75635,
      "training_step_time": 9.431816816329956
    },
    {
      "epoch": 1.560211181640625e-06,
      "model_forward_time": 1.8082091808319092,
      "step": 75636
    },
    {
      "epoch": 1.560211181640625e-06,
      "step": 75636,
      "training_step_time": 8.209919691085815
    },
    {
      "epoch": 1.5611648559570313e-06,
      "model_forward_time": 1.763843059539795,
      "step": 75637
    },
    {
      "epoch": 1.5611648559570313e-06,
      "step": 75637,
      "training_step_time": 7.623779296875
    },
    {
      "epoch": 1.5621185302734374e-06,
      "model_forward_time": 1.7701809406280518,
      "step": 75638
    },
    {
      "epoch": 1.5621185302734374e-06,
      "step": 75638,
      "training_step_time": 7.300132751464844
    },
    {
      "epoch": 1.5630722045898438e-06,
      "model_forward_time": 1.7754809856414795,
      "step": 75639
    },
    {
      "epoch": 1.5630722045898438e-06,
      "step": 75639,
      "training_step_time": 7.158090353012085
    },
    {
      "epoch": 1.56402587890625e-06,
      "grad_norm": 0.08170062303543091,
      "learning_rate": 1.5366197951763046e-06,
      "loss": 0.0578,
      "step": 75640
    },
    {
      "epoch": 1.56402587890625e-06,
      "model_forward_time": 1.7687008380889893,
      "step": 75640
    },
    {
      "dynamics_loss_avg": 0.05903224032372236,
      "epoch": 1.56402587890625e-06,
      "step": 75640
    },
    {
      "action_loss_avg": 0.002321089553879574,
      "epoch": 1.56402587890625e-06,
      "step": 75640
    },
    {
      "epoch": 1.56402587890625e-06,
      "step": 75640,
      "training_step_time": 6.74665379524231
    },
    {
      "epoch": 1.5649795532226563e-06,
      "model_forward_time": 1.7743778228759766,
      "step": 75641
    },
    {
      "epoch": 1.5649795532226563e-06,
      "step": 75641,
      "training_step_time": 6.735596656799316
    },
    {
      "epoch": 1.5659332275390624e-06,
      "model_forward_time": 1.7745106220245361,
      "step": 75642
    },
    {
      "epoch": 1.5659332275390624e-06,
      "step": 75642,
      "training_step_time": 7.565647602081299
    },
    {
      "epoch": 1.5668869018554688e-06,
      "model_forward_time": 2.396989583969116,
      "step": 75643
    },
    {
      "epoch": 1.5668869018554688e-06,
      "step": 75643,
      "training_step_time": 8.431007862091064
    },
    {
      "epoch": 1.567840576171875e-06,
      "model_forward_time": 1.833021879196167,
      "step": 75644
    },
    {
      "epoch": 1.567840576171875e-06,
      "step": 75644,
      "training_step_time": 8.317566394805908
    },
    {
      "epoch": 1.5687942504882813e-06,
      "model_forward_time": 1.7681715488433838,
      "step": 75645
    },
    {
      "epoch": 1.5687942504882813e-06,
      "step": 75645,
      "training_step_time": 8.709774255752563
    },
    {
      "epoch": 1.5697479248046874e-06,
      "model_forward_time": 1.7714087963104248,
      "step": 75646
    },
    {
      "epoch": 1.5697479248046874e-06,
      "step": 75646,
      "training_step_time": 7.57084059715271
    },
    {
      "epoch": 1.5707015991210938e-06,
      "model_forward_time": 1.7661092281341553,
      "step": 75647
    },
    {
      "epoch": 1.5707015991210938e-06,
      "step": 75647,
      "training_step_time": 8.296398639678955
    },
    {
      "epoch": 1.5716552734375e-06,
      "model_forward_time": 1.7728221416473389,
      "step": 75648
    },
    {
      "epoch": 1.5716552734375e-06,
      "step": 75648,
      "training_step_time": 7.199658155441284
    },
    {
      "epoch": 1.5726089477539063e-06,
      "model_forward_time": 1.7712233066558838,
      "step": 75649
    },
    {
      "epoch": 1.5726089477539063e-06,
      "step": 75649,
      "training_step_time": 6.662592649459839
    },
    {
      "epoch": 1.5735626220703124e-06,
      "grad_norm": 0.10941673815250397,
      "learning_rate": 1.5354274213701077e-06,
      "loss": 0.0585,
      "step": 75650
    },
    {
      "epoch": 1.5735626220703124e-06,
      "model_forward_time": 1.7721922397613525,
      "step": 75650
    },
    {
      "dynamics_loss_avg": 0.05629558861255646,
      "epoch": 1.5735626220703124e-06,
      "step": 75650
    },
    {
      "action_loss_avg": 0.0029927333351224663,
      "epoch": 1.5735626220703124e-06,
      "step": 75650
    },
    {
      "epoch": 1.5735626220703124e-06,
      "step": 75650,
      "training_step_time": 6.71573281288147
    },
    {
      "epoch": 1.5745162963867188e-06,
      "model_forward_time": 1.7669482231140137,
      "step": 75651
    },
    {
      "epoch": 1.5745162963867188e-06,
      "step": 75651,
      "training_step_time": 7.55526876449585
    },
    {
      "epoch": 1.575469970703125e-06,
      "model_forward_time": 2.403881311416626,
      "step": 75652
    },
    {
      "epoch": 1.575469970703125e-06,
      "step": 75652,
      "training_step_time": 7.5158162117004395
    },
    {
      "epoch": 1.5764236450195313e-06,
      "model_forward_time": 1.8144567012786865,
      "step": 75653
    },
    {
      "epoch": 1.5764236450195313e-06,
      "step": 75653,
      "training_step_time": 10.666868686676025
    },
    {
      "epoch": 1.5773773193359374e-06,
      "model_forward_time": 1.767745018005371,
      "step": 75654
    },
    {
      "epoch": 1.5773773193359374e-06,
      "step": 75654,
      "training_step_time": 7.711293935775757
    },
    {
      "epoch": 1.5783309936523438e-06,
      "model_forward_time": 1.7660918235778809,
      "step": 75655
    },
    {
      "epoch": 1.5783309936523438e-06,
      "step": 75655,
      "training_step_time": 8.145963191986084
    },
    {
      "epoch": 1.57928466796875e-06,
      "model_forward_time": 1.7696995735168457,
      "step": 75656
    },
    {
      "epoch": 1.57928466796875e-06,
      "step": 75656,
      "training_step_time": 7.557995557785034
    },
    {
      "epoch": 1.5802383422851563e-06,
      "model_forward_time": 1.7677397727966309,
      "step": 75657
    },
    {
      "epoch": 1.5802383422851563e-06,
      "step": 75657,
      "training_step_time": 7.215331077575684
    },
    {
      "epoch": 1.5811920166015624e-06,
      "model_forward_time": 1.771190881729126,
      "step": 75658
    },
    {
      "epoch": 1.5811920166015624e-06,
      "step": 75658,
      "training_step_time": 6.774150133132935
    },
    {
      "epoch": 1.5821456909179688e-06,
      "model_forward_time": 1.771355390548706,
      "step": 75659
    },
    {
      "epoch": 1.5821456909179688e-06,
      "step": 75659,
      "training_step_time": 6.854798793792725
    },
    {
      "epoch": 1.583099365234375e-06,
      "grad_norm": 0.09108349680900574,
      "learning_rate": 1.5342354264443565e-06,
      "loss": 0.0574,
      "step": 75660
    },
    {
      "epoch": 1.583099365234375e-06,
      "model_forward_time": 1.7732951641082764,
      "step": 75660
    },
    {
      "dynamics_loss_avg": 0.06051923893392086,
      "epoch": 1.583099365234375e-06,
      "step": 75660
    },
    {
      "action_loss_avg": 0.0012657388404477387,
      "epoch": 1.583099365234375e-06,
      "step": 75660
    },
    {
      "epoch": 1.583099365234375e-06,
      "step": 75660,
      "training_step_time": 8.425735473632812
    },
    {
      "epoch": 1.5840530395507813e-06,
      "model_forward_time": 2.4110355377197266,
      "step": 75661
    },
    {
      "epoch": 1.5840530395507813e-06,
      "step": 75661,
      "training_step_time": 7.6061952114105225
    },
    {
      "epoch": 1.5850067138671874e-06,
      "model_forward_time": 1.7800912857055664,
      "step": 75662
    },
    {
      "epoch": 1.5850067138671874e-06,
      "step": 75662,
      "training_step_time": 8.138686895370483
    },
    {
      "epoch": 1.5859603881835938e-06,
      "model_forward_time": 1.7702007293701172,
      "step": 75663
    },
    {
      "epoch": 1.5859603881835938e-06,
      "step": 75663,
      "training_step_time": 10.291244983673096
    },
    {
      "epoch": 1.5869140625e-06,
      "model_forward_time": 1.7699754238128662,
      "step": 75664
    },
    {
      "epoch": 1.5869140625e-06,
      "step": 75664,
      "training_step_time": 9.277160167694092
    },
    {
      "epoch": 1.5878677368164063e-06,
      "model_forward_time": 1.7641165256500244,
      "step": 75665
    },
    {
      "epoch": 1.5878677368164063e-06,
      "step": 75665,
      "training_step_time": 8.435037851333618
    },
    {
      "epoch": 1.5888214111328124e-06,
      "model_forward_time": 1.748791217803955,
      "step": 75666
    },
    {
      "epoch": 1.5888214111328124e-06,
      "step": 75666,
      "training_step_time": 7.34865403175354
    },
    {
      "epoch": 1.5897750854492188e-06,
      "model_forward_time": 1.7648365497589111,
      "step": 75667
    },
    {
      "epoch": 1.5897750854492188e-06,
      "step": 75667,
      "training_step_time": 7.267788887023926
    },
    {
      "epoch": 1.590728759765625e-06,
      "model_forward_time": 1.4069240093231201,
      "step": 75668
    },
    {
      "epoch": 1.590728759765625e-06,
      "step": 75668,
      "training_step_time": 6.789460182189941
    },
    {
      "epoch": 1.5916824340820313e-06,
      "model_forward_time": 1.7974567413330078,
      "step": 75669
    },
    {
      "epoch": 1.5916824340820313e-06,
      "step": 75669,
      "training_step_time": 8.317001581192017
    },
    {
      "epoch": 1.5926361083984374e-06,
      "grad_norm": 0.09475730359554291,
      "learning_rate": 1.533043810529411e-06,
      "loss": 0.058,
      "step": 75670
    },
    {
      "epoch": 1.5926361083984374e-06,
      "model_forward_time": 2.415895700454712,
      "step": 75670
    },
    {
      "dynamics_loss_avg": 0.05873367860913277,
      "epoch": 1.5926361083984374e-06,
      "step": 75670
    },
    {
      "action_loss_avg": 0.001444305683253333,
      "epoch": 1.5926361083984374e-06,
      "step": 75670
    },
    {
      "epoch": 1.5926361083984374e-06,
      "step": 75670,
      "training_step_time": 8.148502588272095
    },
    {
      "epoch": 1.5935897827148438e-06,
      "model_forward_time": 1.759012222290039,
      "step": 75671
    },
    {
      "epoch": 1.5935897827148438e-06,
      "step": 75671,
      "training_step_time": 8.997689485549927
    },
    {
      "epoch": 1.59454345703125e-06,
      "model_forward_time": 1.8318803310394287,
      "step": 75672
    },
    {
      "epoch": 1.59454345703125e-06,
      "step": 75672,
      "training_step_time": 8.311239957809448
    },
    {
      "epoch": 1.5954971313476563e-06,
      "model_forward_time": 1.7584702968597412,
      "step": 75673
    },
    {
      "epoch": 1.5954971313476563e-06,
      "step": 75673,
      "training_step_time": 8.89913535118103
    },
    {
      "epoch": 1.5964508056640624e-06,
      "model_forward_time": 1.7669987678527832,
      "step": 75674
    },
    {
      "epoch": 1.5964508056640624e-06,
      "step": 75674,
      "training_step_time": 8.20181918144226
    },
    {
      "epoch": 1.5974044799804688e-06,
      "model_forward_time": 1.8070261478424072,
      "step": 75675
    },
    {
      "epoch": 1.5974044799804688e-06,
      "step": 75675,
      "training_step_time": 7.393513441085815
    },
    {
      "epoch": 1.598358154296875e-06,
      "model_forward_time": 1.7762236595153809,
      "step": 75676
    },
    {
      "epoch": 1.598358154296875e-06,
      "step": 75676,
      "training_step_time": 7.173069715499878
    },
    {
      "epoch": 1.5993118286132813e-06,
      "model_forward_time": 1.7738821506500244,
      "step": 75677
    },
    {
      "epoch": 1.5993118286132813e-06,
      "step": 75677,
      "training_step_time": 7.190034866333008
    },
    {
      "epoch": 1.6002655029296874e-06,
      "model_forward_time": 1.767338752746582,
      "step": 75678
    },
    {
      "epoch": 1.6002655029296874e-06,
      "step": 75678,
      "training_step_time": 7.173006772994995
    },
    {
      "epoch": 1.6012191772460938e-06,
      "model_forward_time": 1.7692368030548096,
      "step": 75679
    },
    {
      "epoch": 1.6012191772460938e-06,
      "step": 75679,
      "training_step_time": 9.538863897323608
    },
    {
      "epoch": 1.6021728515625e-06,
      "grad_norm": 0.09498482197523117,
      "learning_rate": 1.5318525737555806e-06,
      "loss": 0.0573,
      "step": 75680
    },
    {
      "epoch": 1.6021728515625e-06,
      "model_forward_time": 1.7707223892211914,
      "step": 75680
    },
    {
      "dynamics_loss_avg": 0.04949312582612038,
      "epoch": 1.6021728515625e-06,
      "step": 75680
    },
    {
      "action_loss_avg": 0.001894941064529121,
      "epoch": 1.6021728515625e-06,
      "step": 75680
    },
    {
      "epoch": 1.6021728515625e-06,
      "step": 75680,
      "training_step_time": 8.344107627868652
    },
    {
      "epoch": 1.6031265258789063e-06,
      "model_forward_time": 1.4209339618682861,
      "step": 75681
    },
    {
      "epoch": 1.6031265258789063e-06,
      "step": 75681,
      "training_step_time": 8.573306322097778
    },
    {
      "epoch": 1.6040802001953124e-06,
      "model_forward_time": 1.7689063549041748,
      "step": 75682
    },
    {
      "epoch": 1.6040802001953124e-06,
      "step": 75682,
      "training_step_time": 7.634071350097656
    },
    {
      "epoch": 1.6050338745117188e-06,
      "model_forward_time": 1.7653656005859375,
      "step": 75683
    },
    {
      "epoch": 1.6050338745117188e-06,
      "step": 75683,
      "training_step_time": 7.633137464523315
    },
    {
      "epoch": 1.605987548828125e-06,
      "model_forward_time": 1.7771999835968018,
      "step": 75684
    },
    {
      "epoch": 1.605987548828125e-06,
      "step": 75684,
      "training_step_time": 7.581000566482544
    },
    {
      "epoch": 1.6069412231445313e-06,
      "model_forward_time": 1.778719425201416,
      "step": 75685
    },
    {
      "epoch": 1.6069412231445313e-06,
      "step": 75685,
      "training_step_time": 7.169559955596924
    },
    {
      "epoch": 1.6078948974609374e-06,
      "model_forward_time": 1.7506473064422607,
      "step": 75686
    },
    {
      "epoch": 1.6078948974609374e-06,
      "step": 75686,
      "training_step_time": 7.23115611076355
    },
    {
      "epoch": 1.6088485717773438e-06,
      "model_forward_time": 1.7419707775115967,
      "step": 75687
    },
    {
      "epoch": 1.6088485717773438e-06,
      "step": 75687,
      "training_step_time": 7.121644973754883
    },
    {
      "epoch": 1.60980224609375e-06,
      "model_forward_time": 1.3696343898773193,
      "step": 75688
    },
    {
      "epoch": 1.60980224609375e-06,
      "step": 75688,
      "training_step_time": 8.210333824157715
    },
    {
      "epoch": 1.6107559204101563e-06,
      "model_forward_time": 2.3651411533355713,
      "step": 75689
    },
    {
      "epoch": 1.6107559204101563e-06,
      "step": 75689,
      "training_step_time": 8.063071489334106
    },
    {
      "epoch": 1.6117095947265624e-06,
      "grad_norm": 0.10025801509618759,
      "learning_rate": 1.53066171625314e-06,
      "loss": 0.0566,
      "step": 75690
    },
    {
      "epoch": 1.6117095947265624e-06,
      "model_forward_time": 1.3590364456176758,
      "step": 75690
    },
    {
      "dynamics_loss_avg": 0.052635755762457846,
      "epoch": 1.6117095947265624e-06,
      "step": 75690
    },
    {
      "action_loss_avg": 0.0018076247681165114,
      "epoch": 1.6117095947265624e-06,
      "step": 75690
    },
    {
      "epoch": 1.6117095947265624e-06,
      "step": 75690,
      "training_step_time": 10.89239501953125
    },
    {
      "epoch": 1.6126632690429688e-06,
      "model_forward_time": 1.7509174346923828,
      "step": 75691
    },
    {
      "epoch": 1.6126632690429688e-06,
      "step": 75691,
      "training_step_time": 7.686145782470703
    },
    {
      "epoch": 1.613616943359375e-06,
      "model_forward_time": 1.7435948848724365,
      "step": 75692
    },
    {
      "epoch": 1.613616943359375e-06,
      "step": 75692,
      "training_step_time": 9.333409070968628
    },
    {
      "epoch": 1.6145706176757813e-06,
      "model_forward_time": 1.7438108921051025,
      "step": 75693
    },
    {
      "epoch": 1.6145706176757813e-06,
      "step": 75693,
      "training_step_time": 8.265038013458252
    },
    {
      "epoch": 1.6155242919921875e-06,
      "model_forward_time": 1.7448360919952393,
      "step": 75694
    },
    {
      "epoch": 1.6155242919921875e-06,
      "step": 75694,
      "training_step_time": 7.00378942489624
    },
    {
      "epoch": 1.6164779663085938e-06,
      "model_forward_time": 1.7627770900726318,
      "step": 75695
    },
    {
      "epoch": 1.6164779663085938e-06,
      "step": 75695,
      "training_step_time": 7.05258846282959
    },
    {
      "epoch": 1.617431640625e-06,
      "model_forward_time": 1.7544994354248047,
      "step": 75696
    },
    {
      "epoch": 1.617431640625e-06,
      "step": 75696,
      "training_step_time": 7.217508554458618
    },
    {
      "epoch": 1.6183853149414063e-06,
      "model_forward_time": 1.7743911743164062,
      "step": 75697
    },
    {
      "epoch": 1.6183853149414063e-06,
      "step": 75697,
      "training_step_time": 7.5715577602386475
    },
    {
      "epoch": 1.6193389892578125e-06,
      "model_forward_time": 2.4044010639190674,
      "step": 75698
    },
    {
      "epoch": 1.6193389892578125e-06,
      "step": 75698,
      "training_step_time": 9.66231918334961
    },
    {
      "epoch": 1.6202926635742188e-06,
      "model_forward_time": 1.7621865272521973,
      "step": 75699
    },
    {
      "epoch": 1.6202926635742188e-06,
      "step": 75699,
      "training_step_time": 8.249378681182861
    },
    {
      "epoch": 1.621246337890625e-06,
      "grad_norm": 0.2829221189022064,
      "learning_rate": 1.5294712381523168e-06,
      "loss": 0.0564,
      "step": 75700
    },
    {
      "epoch": 1.621246337890625e-06,
      "model_forward_time": 1.5025572776794434,
      "step": 75700
    },
    {
      "dynamics_loss_avg": 0.04412332735955715,
      "epoch": 1.621246337890625e-06,
      "step": 75700
    },
    {
      "action_loss_avg": 0.0013573754054959863,
      "epoch": 1.621246337890625e-06,
      "step": 75700
    },
    {
      "epoch": 1.621246337890625e-06,
      "step": 75700,
      "training_step_time": 8.489843368530273
    },
    {
      "epoch": 1.6222000122070313e-06,
      "model_forward_time": 1.7672572135925293,
      "step": 75701
    },
    {
      "epoch": 1.6222000122070313e-06,
      "step": 75701,
      "training_step_time": 10.544586420059204
    },
    {
      "epoch": 1.6231536865234375e-06,
      "model_forward_time": 1.7612740993499756,
      "step": 75702
    },
    {
      "epoch": 1.6231536865234375e-06,
      "step": 75702,
      "training_step_time": 7.713496923446655
    },
    {
      "epoch": 1.6241073608398438e-06,
      "model_forward_time": 1.7602710723876953,
      "step": 75703
    },
    {
      "epoch": 1.6241073608398438e-06,
      "step": 75703,
      "training_step_time": 7.260517120361328
    },
    {
      "epoch": 1.62506103515625e-06,
      "model_forward_time": 1.7608373165130615,
      "step": 75704
    },
    {
      "epoch": 1.62506103515625e-06,
      "step": 75704,
      "training_step_time": 7.178117513656616
    },
    {
      "epoch": 1.6260147094726563e-06,
      "model_forward_time": 1.7476067543029785,
      "step": 75705
    },
    {
      "epoch": 1.6260147094726563e-06,
      "step": 75705,
      "training_step_time": 7.268701791763306
    },
    {
      "epoch": 1.6269683837890625e-06,
      "model_forward_time": 1.7614820003509521,
      "step": 75706
    },
    {
      "epoch": 1.6269683837890625e-06,
      "step": 75706,
      "training_step_time": 8.536332845687866
    },
    {
      "epoch": 1.6279220581054688e-06,
      "model_forward_time": 1.7605793476104736,
      "step": 75707
    },
    {
      "epoch": 1.6279220581054688e-06,
      "step": 75707,
      "training_step_time": 8.904606103897095
    },
    {
      "epoch": 1.628875732421875e-06,
      "model_forward_time": 2.392834424972534,
      "step": 75708
    },
    {
      "epoch": 1.628875732421875e-06,
      "step": 75708,
      "training_step_time": 9.629501104354858
    },
    {
      "epoch": 1.6298294067382813e-06,
      "model_forward_time": 1.853961706161499,
      "step": 75709
    },
    {
      "epoch": 1.6298294067382813e-06,
      "step": 75709,
      "training_step_time": 8.264694452285767
    },
    {
      "epoch": 1.6307830810546875e-06,
      "grad_norm": 0.09176350384950638,
      "learning_rate": 1.5282811395833024e-06,
      "loss": 0.0566,
      "step": 75710
    },
    {
      "epoch": 1.6307830810546875e-06,
      "model_forward_time": 1.7700042724609375,
      "step": 75710
    },
    {
      "dynamics_loss_avg": 0.05504191163927317,
      "epoch": 1.6307830810546875e-06,
      "step": 75710
    },
    {
      "action_loss_avg": 0.001106630335561931,
      "epoch": 1.6307830810546875e-06,
      "step": 75710
    },
    {
      "epoch": 1.6307830810546875e-06,
      "step": 75710,
      "training_step_time": 7.751532554626465
    },
    {
      "epoch": 1.6317367553710938e-06,
      "model_forward_time": 1.7592079639434814,
      "step": 75711
    },
    {
      "epoch": 1.6317367553710938e-06,
      "step": 75711,
      "training_step_time": 8.144630193710327
    },
    {
      "epoch": 1.6326904296875e-06,
      "model_forward_time": 1.778153657913208,
      "step": 75712
    },
    {
      "epoch": 1.6326904296875e-06,
      "step": 75712,
      "training_step_time": 7.318439483642578
    },
    {
      "epoch": 1.6336441040039063e-06,
      "model_forward_time": 1.7699904441833496,
      "step": 75713
    },
    {
      "epoch": 1.6336441040039063e-06,
      "step": 75713,
      "training_step_time": 7.161183595657349
    },
    {
      "epoch": 1.6345977783203125e-06,
      "model_forward_time": 1.7553160190582275,
      "step": 75714
    },
    {
      "epoch": 1.6345977783203125e-06,
      "step": 75714,
      "training_step_time": 7.294389486312866
    },
    {
      "epoch": 1.6355514526367188e-06,
      "model_forward_time": 1.7531065940856934,
      "step": 75715
    },
    {
      "epoch": 1.6355514526367188e-06,
      "step": 75715,
      "training_step_time": 8.762431621551514
    },
    {
      "epoch": 1.636505126953125e-06,
      "model_forward_time": 1.7938799858093262,
      "step": 75716
    },
    {
      "epoch": 1.636505126953125e-06,
      "step": 75716,
      "training_step_time": 7.575380325317383
    },
    {
      "epoch": 1.6374588012695313e-06,
      "model_forward_time": 2.390291213989258,
      "step": 75717
    },
    {
      "epoch": 1.6374588012695313e-06,
      "step": 75717,
      "training_step_time": 7.575801134109497
    },
    {
      "epoch": 1.6384124755859375e-06,
      "model_forward_time": 1.8379580974578857,
      "step": 75718
    },
    {
      "epoch": 1.6384124755859375e-06,
      "step": 75718,
      "training_step_time": 9.080399513244629
    },
    {
      "epoch": 1.6393661499023438e-06,
      "model_forward_time": 1.7690622806549072,
      "step": 75719
    },
    {
      "epoch": 1.6393661499023438e-06,
      "step": 75719,
      "training_step_time": 8.496225357055664
    },
    {
      "epoch": 1.64031982421875e-06,
      "grad_norm": 0.12242221087217331,
      "learning_rate": 1.52709142067624e-06,
      "loss": 0.0565,
      "step": 75720
    },
    {
      "epoch": 1.64031982421875e-06,
      "model_forward_time": 1.7680928707122803,
      "step": 75720
    },
    {
      "dynamics_loss_avg": 0.04289272353053093,
      "epoch": 1.64031982421875e-06,
      "step": 75720
    },
    {
      "action_loss_avg": 0.0009399677248438821,
      "epoch": 1.64031982421875e-06,
      "step": 75720
    },
    {
      "epoch": 1.64031982421875e-06,
      "step": 75720,
      "training_step_time": 8.097946882247925
    },
    {
      "epoch": 1.6412734985351563e-06,
      "model_forward_time": 1.766880750656128,
      "step": 75721
    },
    {
      "epoch": 1.6412734985351563e-06,
      "step": 75721,
      "training_step_time": 7.578342914581299
    },
    {
      "epoch": 1.6422271728515625e-06,
      "model_forward_time": 1.773268699645996,
      "step": 75722
    },
    {
      "epoch": 1.6422271728515625e-06,
      "step": 75722,
      "training_step_time": 7.075466871261597
    },
    {
      "epoch": 1.6431808471679688e-06,
      "model_forward_time": 1.7701916694641113,
      "step": 75723
    },
    {
      "epoch": 1.6431808471679688e-06,
      "step": 75723,
      "training_step_time": 7.182671308517456
    },
    {
      "epoch": 1.644134521484375e-06,
      "model_forward_time": 1.758044719696045,
      "step": 75724
    },
    {
      "epoch": 1.644134521484375e-06,
      "step": 75724,
      "training_step_time": 7.535268545150757
    },
    {
      "epoch": 1.6450881958007813e-06,
      "model_forward_time": 1.7757651805877686,
      "step": 75725
    },
    {
      "epoch": 1.6450881958007813e-06,
      "step": 75725,
      "training_step_time": 7.7660369873046875
    },
    {
      "epoch": 1.6460418701171875e-06,
      "model_forward_time": 2.3907201290130615,
      "step": 75726
    },
    {
      "epoch": 1.6460418701171875e-06,
      "step": 75726,
      "training_step_time": 7.547354698181152
    },
    {
      "epoch": 1.6469955444335938e-06,
      "model_forward_time": 1.8190081119537354,
      "step": 75727
    },
    {
      "epoch": 1.6469955444335938e-06,
      "step": 75727,
      "training_step_time": 9.289604663848877
    },
    {
      "epoch": 1.64794921875e-06,
      "model_forward_time": 1.7779741287231445,
      "step": 75728
    },
    {
      "epoch": 1.64794921875e-06,
      "step": 75728,
      "training_step_time": 8.630768060684204
    },
    {
      "epoch": 1.6489028930664063e-06,
      "model_forward_time": 1.7819883823394775,
      "step": 75729
    },
    {
      "epoch": 1.6489028930664063e-06,
      "step": 75729,
      "training_step_time": 8.514126062393188
    },
    {
      "epoch": 1.6498565673828125e-06,
      "grad_norm": 0.11115901917219162,
      "learning_rate": 1.5259020815612414e-06,
      "loss": 0.0558,
      "step": 75730
    },
    {
      "epoch": 1.6498565673828125e-06,
      "model_forward_time": 1.7707359790802002,
      "step": 75730
    },
    {
      "dynamics_loss_avg": 0.0502338569611311,
      "epoch": 1.6498565673828125e-06,
      "step": 75730
    },
    {
      "action_loss_avg": 0.0010443698876770214,
      "epoch": 1.6498565673828125e-06,
      "step": 75730
    },
    {
      "epoch": 1.6498565673828125e-06,
      "step": 75730,
      "training_step_time": 7.559092998504639
    },
    {
      "epoch": 1.6508102416992188e-06,
      "model_forward_time": 1.7598206996917725,
      "step": 75731
    },
    {
      "epoch": 1.6508102416992188e-06,
      "step": 75731,
      "training_step_time": 7.709117412567139
    },
    {
      "epoch": 1.651763916015625e-06,
      "model_forward_time": 1.782602310180664,
      "step": 75732
    },
    {
      "epoch": 1.651763916015625e-06,
      "step": 75732,
      "training_step_time": 7.14015007019043
    },
    {
      "epoch": 1.6527175903320313e-06,
      "model_forward_time": 1.7699792385101318,
      "step": 75733
    },
    {
      "epoch": 1.6527175903320313e-06,
      "step": 75733,
      "training_step_time": 8.26484227180481
    },
    {
      "epoch": 1.6536712646484375e-06,
      "model_forward_time": 1.7776787281036377,
      "step": 75734
    },
    {
      "epoch": 1.6536712646484375e-06,
      "step": 75734,
      "training_step_time": 7.594567775726318
    },
    {
      "epoch": 1.6546249389648438e-06,
      "model_forward_time": 2.4025204181671143,
      "step": 75735
    },
    {
      "epoch": 1.6546249389648438e-06,
      "step": 75735,
      "training_step_time": 9.53942608833313
    },
    {
      "epoch": 1.65557861328125e-06,
      "model_forward_time": 1.8240759372711182,
      "step": 75736
    },
    {
      "epoch": 1.65557861328125e-06,
      "step": 75736,
      "training_step_time": 9.801073789596558
    },
    {
      "epoch": 1.6565322875976563e-06,
      "model_forward_time": 1.7754154205322266,
      "step": 75737
    },
    {
      "epoch": 1.6565322875976563e-06,
      "step": 75737,
      "training_step_time": 7.621121406555176
    },
    {
      "epoch": 1.6574859619140625e-06,
      "model_forward_time": 1.7775871753692627,
      "step": 75738
    },
    {
      "epoch": 1.6574859619140625e-06,
      "step": 75738,
      "training_step_time": 8.689810276031494
    },
    {
      "epoch": 1.6584396362304688e-06,
      "model_forward_time": 1.774655818939209,
      "step": 75739
    },
    {
      "epoch": 1.6584396362304688e-06,
      "step": 75739,
      "training_step_time": 7.603383302688599
    },
    {
      "epoch": 1.659393310546875e-06,
      "grad_norm": 0.08087807893753052,
      "learning_rate": 1.5247131223683665e-06,
      "loss": 0.0546,
      "step": 75740
    },
    {
      "epoch": 1.659393310546875e-06,
      "model_forward_time": 1.781092882156372,
      "step": 75740
    },
    {
      "dynamics_loss_avg": 0.04997056564316153,
      "epoch": 1.659393310546875e-06,
      "step": 75740
    },
    {
      "action_loss_avg": 0.0012053475191351026,
      "epoch": 1.659393310546875e-06,
      "step": 75740
    },
    {
      "epoch": 1.659393310546875e-06,
      "step": 75740,
      "training_step_time": 6.984429121017456
    },
    {
      "epoch": 1.6603469848632813e-06,
      "model_forward_time": 1.7723591327667236,
      "step": 75741
    },
    {
      "epoch": 1.6603469848632813e-06,
      "step": 75741,
      "training_step_time": 7.140228033065796
    },
    {
      "epoch": 1.6613006591796875e-06,
      "model_forward_time": 1.7761170864105225,
      "step": 75742
    },
    {
      "epoch": 1.6613006591796875e-06,
      "step": 75742,
      "training_step_time": 8.210422992706299
    },
    {
      "epoch": 1.6622543334960938e-06,
      "model_forward_time": 1.7724039554595947,
      "step": 75743
    },
    {
      "epoch": 1.6622543334960938e-06,
      "step": 75743,
      "training_step_time": 7.577992916107178
    },
    {
      "epoch": 1.6632080078125e-06,
      "model_forward_time": 2.3972535133361816,
      "step": 75744
    },
    {
      "epoch": 1.6632080078125e-06,
      "step": 75744,
      "training_step_time": 7.585288763046265
    },
    {
      "epoch": 1.6641616821289064e-06,
      "model_forward_time": 1.7898290157318115,
      "step": 75745
    },
    {
      "epoch": 1.6641616821289064e-06,
      "step": 75745,
      "training_step_time": 7.792723178863525
    },
    {
      "epoch": 1.6651153564453125e-06,
      "model_forward_time": 1.7801964282989502,
      "step": 75746
    },
    {
      "epoch": 1.6651153564453125e-06,
      "step": 75746,
      "training_step_time": 8.593547582626343
    },
    {
      "epoch": 1.6660690307617189e-06,
      "model_forward_time": 1.7738420963287354,
      "step": 75747
    },
    {
      "epoch": 1.6660690307617189e-06,
      "step": 75747,
      "training_step_time": 8.850838661193848
    },
    {
      "epoch": 1.667022705078125e-06,
      "model_forward_time": 1.4175949096679688,
      "step": 75748
    },
    {
      "epoch": 1.667022705078125e-06,
      "step": 75748,
      "training_step_time": 7.732845067977905
    },
    {
      "epoch": 1.6679763793945314e-06,
      "model_forward_time": 1.7795214653015137,
      "step": 75749
    },
    {
      "epoch": 1.6679763793945314e-06,
      "step": 75749,
      "training_step_time": 7.69262957572937
    },
    {
      "epoch": 1.6689300537109375e-06,
      "grad_norm": 0.0820622518658638,
      "learning_rate": 1.5235245432276408e-06,
      "loss": 0.0543,
      "step": 75750
    },
    {
      "epoch": 1.6689300537109375e-06,
      "model_forward_time": 1.7726786136627197,
      "step": 75750
    },
    {
      "dynamics_loss_avg": 0.04856782238930464,
      "epoch": 1.6689300537109375e-06,
      "step": 75750
    },
    {
      "action_loss_avg": 0.0014676014427095652,
      "epoch": 1.6689300537109375e-06,
      "step": 75750
    },
    {
      "epoch": 1.6689300537109375e-06,
      "step": 75750,
      "training_step_time": 7.233967065811157
    },
    {
      "epoch": 1.6698837280273436e-06,
      "model_forward_time": 1.773317813873291,
      "step": 75751
    },
    {
      "epoch": 1.6698837280273436e-06,
      "step": 75751,
      "training_step_time": 7.6822333335876465
    },
    {
      "epoch": 1.67083740234375e-06,
      "model_forward_time": 1.4140856266021729,
      "step": 75752
    },
    {
      "epoch": 1.67083740234375e-06,
      "step": 75752,
      "training_step_time": 9.08216905593872
    },
    {
      "epoch": 1.6717910766601561e-06,
      "model_forward_time": 1.7631950378417969,
      "step": 75753
    },
    {
      "epoch": 1.6717910766601561e-06,
      "step": 75753,
      "training_step_time": 7.582368850708008
    },
    {
      "epoch": 1.6727447509765625e-06,
      "model_forward_time": 2.4326915740966797,
      "step": 75754
    },
    {
      "epoch": 1.6727447509765625e-06,
      "step": 75754,
      "training_step_time": 9.270216941833496
    },
    {
      "epoch": 1.6736984252929686e-06,
      "model_forward_time": 1.7807950973510742,
      "step": 75755
    },
    {
      "epoch": 1.6736984252929686e-06,
      "step": 75755,
      "training_step_time": 9.851557970046997
    },
    {
      "epoch": 1.674652099609375e-06,
      "model_forward_time": 1.7753660678863525,
      "step": 75756
    },
    {
      "epoch": 1.674652099609375e-06,
      "step": 75756,
      "training_step_time": 7.716302156448364
    },
    {
      "epoch": 1.6756057739257812e-06,
      "model_forward_time": 1.775705337524414,
      "step": 75757
    },
    {
      "epoch": 1.6756057739257812e-06,
      "step": 75757,
      "training_step_time": 7.689164161682129
    },
    {
      "epoch": 1.6765594482421875e-06,
      "model_forward_time": 1.7841506004333496,
      "step": 75758
    },
    {
      "epoch": 1.6765594482421875e-06,
      "step": 75758,
      "training_step_time": 7.788778781890869
    },
    {
      "epoch": 1.6775131225585937e-06,
      "model_forward_time": 1.7782032489776611,
      "step": 75759
    },
    {
      "epoch": 1.6775131225585937e-06,
      "step": 75759,
      "training_step_time": 7.169186592102051
    },
    {
      "epoch": 1.678466796875e-06,
      "grad_norm": 0.206559419631958,
      "learning_rate": 1.5223363442690426e-06,
      "loss": 0.0563,
      "step": 75760
    },
    {
      "epoch": 1.678466796875e-06,
      "model_forward_time": 1.7590398788452148,
      "step": 75760
    },
    {
      "dynamics_loss_avg": 0.055405815877020356,
      "epoch": 1.678466796875e-06,
      "step": 75760
    },
    {
      "action_loss_avg": 0.0009852943883743137,
      "epoch": 1.678466796875e-06,
      "step": 75760
    },
    {
      "epoch": 1.678466796875e-06,
      "step": 75760,
      "training_step_time": 7.656938791275024
    },
    {
      "epoch": 1.6794204711914062e-06,
      "model_forward_time": 1.761995792388916,
      "step": 75761
    },
    {
      "epoch": 1.6794204711914062e-06,
      "step": 75761,
      "training_step_time": 8.16226053237915
    },
    {
      "epoch": 1.6803741455078125e-06,
      "model_forward_time": 1.7617089748382568,
      "step": 75762
    },
    {
      "epoch": 1.6803741455078125e-06,
      "step": 75762,
      "training_step_time": 9.00145173072815
    },
    {
      "epoch": 1.6813278198242187e-06,
      "model_forward_time": 2.395012855529785,
      "step": 75763
    },
    {
      "epoch": 1.6813278198242187e-06,
      "step": 75763,
      "training_step_time": 8.28133225440979
    },
    {
      "epoch": 1.682281494140625e-06,
      "model_forward_time": 1.8438687324523926,
      "step": 75764
    },
    {
      "epoch": 1.682281494140625e-06,
      "step": 75764,
      "training_step_time": 7.6421098709106445
    },
    {
      "epoch": 1.6832351684570312e-06,
      "model_forward_time": 1.7578842639923096,
      "step": 75765
    },
    {
      "epoch": 1.6832351684570312e-06,
      "step": 75765,
      "training_step_time": 8.579625368118286
    },
    {
      "epoch": 1.6841888427734375e-06,
      "model_forward_time": 1.7626879215240479,
      "step": 75766
    },
    {
      "epoch": 1.6841888427734375e-06,
      "step": 75766,
      "training_step_time": 8.542996644973755
    },
    {
      "epoch": 1.6851425170898437e-06,
      "model_forward_time": 1.759824275970459,
      "step": 75767
    },
    {
      "epoch": 1.6851425170898437e-06,
      "step": 75767,
      "training_step_time": 7.6376588344573975
    },
    {
      "epoch": 1.68609619140625e-06,
      "model_forward_time": 1.7641849517822266,
      "step": 75768
    },
    {
      "epoch": 1.68609619140625e-06,
      "step": 75768,
      "training_step_time": 6.718365669250488
    },
    {
      "epoch": 1.6870498657226562e-06,
      "model_forward_time": 1.7629377841949463,
      "step": 75769
    },
    {
      "epoch": 1.6870498657226562e-06,
      "step": 75769,
      "training_step_time": 7.6704490184783936
    },
    {
      "epoch": 1.6880035400390625e-06,
      "grad_norm": 0.14594529569149017,
      "learning_rate": 1.5211485256225134e-06,
      "loss": 0.0552,
      "step": 75770
    },
    {
      "epoch": 1.6880035400390625e-06,
      "model_forward_time": 1.752263069152832,
      "step": 75770
    },
    {
      "dynamics_loss_avg": 0.04873644020408392,
      "epoch": 1.6880035400390625e-06,
      "step": 75770
    },
    {
      "action_loss_avg": 0.0016223831684328617,
      "epoch": 1.6880035400390625e-06,
      "step": 75770
    },
    {
      "epoch": 1.6880035400390625e-06,
      "step": 75770,
      "training_step_time": 7.334912061691284
    },
    {
      "epoch": 1.6889572143554687e-06,
      "model_forward_time": 1.7673611640930176,
      "step": 75771
    },
    {
      "epoch": 1.6889572143554687e-06,
      "step": 75771,
      "training_step_time": 9.092660903930664
    },
    {
      "epoch": 1.689910888671875e-06,
      "model_forward_time": 2.381000280380249,
      "step": 75772
    },
    {
      "epoch": 1.689910888671875e-06,
      "step": 75772,
      "training_step_time": 7.643504858016968
    },
    {
      "epoch": 1.6908645629882812e-06,
      "model_forward_time": 1.853417158126831,
      "step": 75773
    },
    {
      "epoch": 1.6908645629882812e-06,
      "step": 75773,
      "training_step_time": 9.495010614395142
    },
    {
      "epoch": 1.6918182373046875e-06,
      "model_forward_time": 1.760188102722168,
      "step": 75774
    },
    {
      "epoch": 1.6918182373046875e-06,
      "step": 75774,
      "training_step_time": 8.281813621520996
    },
    {
      "epoch": 1.6927719116210937e-06,
      "model_forward_time": 1.7581443786621094,
      "step": 75775
    },
    {
      "epoch": 1.6927719116210937e-06,
      "step": 75775,
      "training_step_time": 7.697827577590942
    },
    {
      "epoch": 1.6937255859375e-06,
      "model_forward_time": 1.7636923789978027,
      "step": 75776
    },
    {
      "epoch": 1.6937255859375e-06,
      "step": 75776,
      "training_step_time": 7.743000745773315
    },
    {
      "epoch": 1.6946792602539062e-06,
      "model_forward_time": 1.7633323669433594,
      "step": 75777
    },
    {
      "epoch": 1.6946792602539062e-06,
      "step": 75777,
      "training_step_time": 7.1967384815216064
    },
    {
      "epoch": 1.6956329345703125e-06,
      "model_forward_time": 1.7602219581604004,
      "step": 75778
    },
    {
      "epoch": 1.6956329345703125e-06,
      "step": 75778,
      "training_step_time": 7.730966806411743
    },
    {
      "epoch": 1.6965866088867187e-06,
      "model_forward_time": 1.758070707321167,
      "step": 75779
    },
    {
      "epoch": 1.6965866088867187e-06,
      "step": 75779,
      "training_step_time": 7.244929552078247
    },
    {
      "epoch": 1.697540283203125e-06,
      "grad_norm": 0.15208245813846588,
      "learning_rate": 1.519961087417951e-06,
      "loss": 0.0554,
      "step": 75780
    },
    {
      "epoch": 1.697540283203125e-06,
      "model_forward_time": 1.7558999061584473,
      "step": 75780
    },
    {
      "dynamics_loss_avg": 0.05574692208319902,
      "epoch": 1.697540283203125e-06,
      "step": 75780
    },
    {
      "action_loss_avg": 0.0017424071847926825,
      "epoch": 1.697540283203125e-06,
      "step": 75780
    },
    {
      "epoch": 1.697540283203125e-06,
      "step": 75780,
      "training_step_time": 7.682823181152344
    },
    {
      "epoch": 1.6984939575195312e-06,
      "model_forward_time": 1.7653110027313232,
      "step": 75781
    },
    {
      "epoch": 1.6984939575195312e-06,
      "step": 75781,
      "training_step_time": 9.360947847366333
    },
    {
      "epoch": 1.6994476318359375e-06,
      "model_forward_time": 1.7998015880584717,
      "step": 75782
    },
    {
      "epoch": 1.6994476318359375e-06,
      "step": 75782,
      "training_step_time": 7.717930316925049
    },
    {
      "epoch": 1.7004013061523437e-06,
      "model_forward_time": 1.7580857276916504,
      "step": 75783
    },
    {
      "epoch": 1.7004013061523437e-06,
      "step": 75783,
      "training_step_time": 8.824042081832886
    },
    {
      "epoch": 1.70135498046875e-06,
      "model_forward_time": 1.7505214214324951,
      "step": 75784
    },
    {
      "epoch": 1.70135498046875e-06,
      "step": 75784,
      "training_step_time": 8.46689224243164
    },
    {
      "epoch": 1.7023086547851562e-06,
      "model_forward_time": 1.7655832767486572,
      "step": 75785
    },
    {
      "epoch": 1.7023086547851562e-06,
      "step": 75785,
      "training_step_time": 7.7388529777526855
    },
    {
      "epoch": 1.7032623291015625e-06,
      "model_forward_time": 1.7604734897613525,
      "step": 75786
    },
    {
      "epoch": 1.7032623291015625e-06,
      "step": 75786,
      "training_step_time": 7.679900646209717
    },
    {
      "epoch": 1.7042160034179687e-06,
      "model_forward_time": 1.7579853534698486,
      "step": 75787
    },
    {
      "epoch": 1.7042160034179687e-06,
      "step": 75787,
      "training_step_time": 8.407469034194946
    },
    {
      "epoch": 1.705169677734375e-06,
      "model_forward_time": 1.7594993114471436,
      "step": 75788
    },
    {
      "epoch": 1.705169677734375e-06,
      "step": 75788,
      "training_step_time": 7.245800495147705
    },
    {
      "epoch": 1.7061233520507812e-06,
      "model_forward_time": 1.7603979110717773,
      "step": 75789
    },
    {
      "epoch": 1.7061233520507812e-06,
      "step": 75789,
      "training_step_time": 7.713658809661865
    },
    {
      "epoch": 1.7070770263671875e-06,
      "grad_norm": 0.0943574383854866,
      "learning_rate": 1.5187740297852138e-06,
      "loss": 0.0562,
      "step": 75790
    },
    {
      "epoch": 1.7070770263671875e-06,
      "model_forward_time": 1.750462532043457,
      "step": 75790
    },
    {
      "dynamics_loss_avg": 0.05269568637013435,
      "epoch": 1.7070770263671875e-06,
      "step": 75790
    },
    {
      "action_loss_avg": 0.00158513113274239,
      "epoch": 1.7070770263671875e-06,
      "step": 75790
    },
    {
      "epoch": 1.7070770263671875e-06,
      "step": 75790,
      "training_step_time": 7.627649307250977
    },
    {
      "epoch": 1.7080307006835937e-06,
      "model_forward_time": 2.4413399696350098,
      "step": 75791
    },
    {
      "epoch": 1.7080307006835937e-06,
      "step": 75791,
      "training_step_time": 8.761404275894165
    },
    {
      "epoch": 1.708984375e-06,
      "model_forward_time": 1.7552142143249512,
      "step": 75792
    },
    {
      "epoch": 1.708984375e-06,
      "step": 75792,
      "training_step_time": 8.699989318847656
    },
    {
      "epoch": 1.7099380493164062e-06,
      "model_forward_time": 1.7632019519805908,
      "step": 75793
    },
    {
      "epoch": 1.7099380493164062e-06,
      "step": 75793,
      "training_step_time": 8.68836236000061
    },
    {
      "epoch": 1.7108917236328125e-06,
      "model_forward_time": 1.7593483924865723,
      "step": 75794
    },
    {
      "epoch": 1.7108917236328125e-06,
      "step": 75794,
      "training_step_time": 9.064960479736328
    },
    {
      "epoch": 1.7118453979492187e-06,
      "model_forward_time": 1.7516629695892334,
      "step": 75795
    },
    {
      "epoch": 1.7118453979492187e-06,
      "step": 75795,
      "training_step_time": 7.348135471343994
    },
    {
      "epoch": 1.712799072265625e-06,
      "model_forward_time": 1.7791111469268799,
      "step": 75796
    },
    {
      "epoch": 1.712799072265625e-06,
      "step": 75796,
      "training_step_time": 7.708086252212524
    },
    {
      "epoch": 1.7137527465820312e-06,
      "model_forward_time": 1.7609951496124268,
      "step": 75797
    },
    {
      "epoch": 1.7137527465820312e-06,
      "step": 75797,
      "training_step_time": 7.841626882553101
    },
    {
      "epoch": 1.7147064208984375e-06,
      "model_forward_time": 1.7794785499572754,
      "step": 75798
    },
    {
      "epoch": 1.7147064208984375e-06,
      "step": 75798,
      "training_step_time": 7.477471828460693
    },
    {
      "epoch": 1.7156600952148437e-06,
      "model_forward_time": 1.77439546585083,
      "step": 75799
    },
    {
      "epoch": 1.7156600952148437e-06,
      "step": 75799,
      "training_step_time": 8.205273389816284
    },
    {
      "epoch": 1.71661376953125e-06,
      "grad_norm": 0.08143939822912216,
      "learning_rate": 1.517587352854113e-06,
      "loss": 0.0555,
      "step": 75800
    },
    {
      "epoch": 1.71661376953125e-06,
      "model_forward_time": 2.4202723503112793,
      "step": 75800
    },
    {
      "dynamics_loss_avg": 0.051401442289352416,
      "epoch": 1.71661376953125e-06,
      "step": 75800
    },
    {
      "action_loss_avg": 0.0008249991224147379,
      "epoch": 1.71661376953125e-06,
      "step": 75800
    },
    {
      "epoch": 1.71661376953125e-06,
      "step": 75800,
      "training_step_time": 7.743167877197266
    },
    {
      "epoch": 1.7175674438476562e-06,
      "model_forward_time": 1.768763542175293,
      "step": 75801
    },
    {
      "epoch": 1.7175674438476562e-06,
      "step": 75801,
      "training_step_time": 8.496783256530762
    },
    {
      "epoch": 1.7185211181640625e-06,
      "model_forward_time": 1.7566545009613037,
      "step": 75802
    },
    {
      "epoch": 1.7185211181640625e-06,
      "step": 75802,
      "training_step_time": 8.472588539123535
    },
    {
      "epoch": 1.7194747924804687e-06,
      "model_forward_time": 1.7493672370910645,
      "step": 75803
    },
    {
      "epoch": 1.7194747924804687e-06,
      "step": 75803,
      "training_step_time": 8.531373739242554
    },
    {
      "epoch": 1.720428466796875e-06,
      "model_forward_time": 1.7541005611419678,
      "step": 75804
    },
    {
      "epoch": 1.720428466796875e-06,
      "step": 75804,
      "training_step_time": 7.314876317977905
    },
    {
      "epoch": 1.7213821411132812e-06,
      "model_forward_time": 1.7860772609710693,
      "step": 75805
    },
    {
      "epoch": 1.7213821411132812e-06,
      "step": 75805,
      "training_step_time": 7.65772557258606
    },
    {
      "epoch": 1.7223358154296875e-06,
      "model_forward_time": 1.7693750858306885,
      "step": 75806
    },
    {
      "epoch": 1.7223358154296875e-06,
      "step": 75806,
      "training_step_time": 7.186442136764526
    },
    {
      "epoch": 1.7232894897460937e-06,
      "model_forward_time": 1.7698161602020264,
      "step": 75807
    },
    {
      "epoch": 1.7232894897460937e-06,
      "step": 75807,
      "training_step_time": 8.465598344802856
    },
    {
      "epoch": 1.7242431640625e-06,
      "model_forward_time": 1.7761876583099365,
      "step": 75808
    },
    {
      "epoch": 1.7242431640625e-06,
      "step": 75808,
      "training_step_time": 8.905843734741211
    },
    {
      "epoch": 1.7251968383789062e-06,
      "model_forward_time": 2.4153692722320557,
      "step": 75809
    },
    {
      "epoch": 1.7251968383789062e-06,
      "step": 75809,
      "training_step_time": 7.582346439361572
    },
    {
      "epoch": 1.7261505126953126e-06,
      "grad_norm": 0.07570500671863556,
      "learning_rate": 1.516401056754423e-06,
      "loss": 0.0567,
      "step": 75810
    },
    {
      "epoch": 1.7261505126953126e-06,
      "model_forward_time": 1.8297054767608643,
      "step": 75810
    },
    {
      "dynamics_loss_avg": 0.04881545398384333,
      "epoch": 1.7261505126953126e-06,
      "step": 75810
    },
    {
      "action_loss_avg": 0.0017171396582853049,
      "epoch": 1.7261505126953126e-06,
      "step": 75810
    },
    {
      "epoch": 1.7261505126953126e-06,
      "step": 75810,
      "training_step_time": 7.856818675994873
    },
    {
      "epoch": 1.7271041870117187e-06,
      "model_forward_time": 1.774888038635254,
      "step": 75811
    },
    {
      "epoch": 1.7271041870117187e-06,
      "step": 75811,
      "training_step_time": 7.7967071533203125
    },
    {
      "epoch": 1.728057861328125e-06,
      "model_forward_time": 1.782050371170044,
      "step": 75812
    },
    {
      "epoch": 1.728057861328125e-06,
      "step": 75812,
      "training_step_time": 7.711670637130737
    },
    {
      "epoch": 1.7290115356445312e-06,
      "model_forward_time": 1.771822214126587,
      "step": 75813
    },
    {
      "epoch": 1.7290115356445312e-06,
      "step": 75813,
      "training_step_time": 7.742079734802246
    },
    {
      "epoch": 1.7299652099609376e-06,
      "model_forward_time": 1.7756669521331787,
      "step": 75814
    },
    {
      "epoch": 1.7299652099609376e-06,
      "step": 75814,
      "training_step_time": 7.6098949909210205
    },
    {
      "epoch": 1.7309188842773437e-06,
      "model_forward_time": 1.7837588787078857,
      "step": 75815
    },
    {
      "epoch": 1.7309188842773437e-06,
      "step": 75815,
      "training_step_time": 7.2279417514801025
    },
    {
      "epoch": 1.73187255859375e-06,
      "model_forward_time": 1.775524616241455,
      "step": 75816
    },
    {
      "epoch": 1.73187255859375e-06,
      "step": 75816,
      "training_step_time": 7.3289031982421875
    },
    {
      "epoch": 1.7328262329101562e-06,
      "model_forward_time": 1.7864105701446533,
      "step": 75817
    },
    {
      "epoch": 1.7328262329101562e-06,
      "step": 75817,
      "training_step_time": 9.310970306396484
    },
    {
      "epoch": 1.7337799072265626e-06,
      "model_forward_time": 2.422506093978882,
      "step": 75818
    },
    {
      "epoch": 1.7337799072265626e-06,
      "step": 75818,
      "training_step_time": 7.511136054992676
    },
    {
      "epoch": 1.7347335815429687e-06,
      "model_forward_time": 1.8505158424377441,
      "step": 75819
    },
    {
      "epoch": 1.7347335815429687e-06,
      "step": 75819,
      "training_step_time": 9.758172035217285
    },
    {
      "epoch": 1.735687255859375e-06,
      "grad_norm": 0.08894181996583939,
      "learning_rate": 1.5152151416158784e-06,
      "loss": 0.0569,
      "step": 75820
    },
    {
      "epoch": 1.735687255859375e-06,
      "model_forward_time": 1.7731678485870361,
      "step": 75820
    },
    {
      "dynamics_loss_avg": 0.038117899745702746,
      "epoch": 1.735687255859375e-06,
      "step": 75820
    },
    {
      "action_loss_avg": 0.001439713075524196,
      "epoch": 1.735687255859375e-06,
      "step": 75820
    },
    {
      "epoch": 1.735687255859375e-06,
      "step": 75820,
      "training_step_time": 9.862584829330444
    },
    {
      "epoch": 1.7366409301757812e-06,
      "model_forward_time": 1.7725281715393066,
      "step": 75821
    },
    {
      "epoch": 1.7366409301757812e-06,
      "step": 75821,
      "training_step_time": 9.163258790969849
    },
    {
      "epoch": 1.7375946044921876e-06,
      "model_forward_time": 1.767305612564087,
      "step": 75822
    },
    {
      "epoch": 1.7375946044921876e-06,
      "step": 75822,
      "training_step_time": 7.656236886978149
    },
    {
      "epoch": 1.7385482788085937e-06,
      "model_forward_time": 1.7635829448699951,
      "step": 75823
    },
    {
      "epoch": 1.7385482788085937e-06,
      "step": 75823,
      "training_step_time": 8.674572467803955
    },
    {
      "epoch": 1.739501953125e-06,
      "model_forward_time": 1.7631003856658936,
      "step": 75824
    },
    {
      "epoch": 1.739501953125e-06,
      "step": 75824,
      "training_step_time": 7.096746206283569
    },
    {
      "epoch": 1.7404556274414062e-06,
      "model_forward_time": 1.7552237510681152,
      "step": 75825
    },
    {
      "epoch": 1.7404556274414062e-06,
      "step": 75825,
      "training_step_time": 7.279927730560303
    },
    {
      "epoch": 1.7414093017578126e-06,
      "model_forward_time": 1.7626349925994873,
      "step": 75826
    },
    {
      "epoch": 1.7414093017578126e-06,
      "step": 75826,
      "training_step_time": 7.586188316345215
    },
    {
      "epoch": 1.7423629760742187e-06,
      "model_forward_time": 2.3786628246307373,
      "step": 75827
    },
    {
      "epoch": 1.7423629760742187e-06,
      "step": 75827,
      "training_step_time": 9.78285264968872
    },
    {
      "epoch": 1.743316650390625e-06,
      "model_forward_time": 1.8143982887268066,
      "step": 75828
    },
    {
      "epoch": 1.743316650390625e-06,
      "step": 75828,
      "training_step_time": 8.216788530349731
    },
    {
      "epoch": 1.7442703247070312e-06,
      "model_forward_time": 1.7551560401916504,
      "step": 75829
    },
    {
      "epoch": 1.7442703247070312e-06,
      "step": 75829,
      "training_step_time": 8.24967646598816
    },
    {
      "epoch": 1.7452239990234376e-06,
      "grad_norm": 0.08930224180221558,
      "learning_rate": 1.5140296075681643e-06,
      "loss": 0.0567,
      "step": 75830
    },
    {
      "epoch": 1.7452239990234376e-06,
      "model_forward_time": 1.7650237083435059,
      "step": 75830
    },
    {
      "dynamics_loss_avg": 0.04636847451329231,
      "epoch": 1.7452239990234376e-06,
      "step": 75830
    },
    {
      "action_loss_avg": 0.0009753129328601062,
      "epoch": 1.7452239990234376e-06,
      "step": 75830
    },
    {
      "epoch": 1.7452239990234376e-06,
      "step": 75830,
      "training_step_time": 7.72226095199585
    },
    {
      "epoch": 1.7461776733398437e-06,
      "model_forward_time": 1.7708511352539062,
      "step": 75831
    },
    {
      "epoch": 1.7461776733398437e-06,
      "step": 75831,
      "training_step_time": 7.690165996551514
    },
    {
      "epoch": 1.74713134765625e-06,
      "model_forward_time": 1.7657880783081055,
      "step": 75832
    },
    {
      "epoch": 1.74713134765625e-06,
      "step": 75832,
      "training_step_time": 7.7016730308532715
    },
    {
      "epoch": 1.7480850219726562e-06,
      "model_forward_time": 1.7637550830841064,
      "step": 75833
    },
    {
      "epoch": 1.7480850219726562e-06,
      "step": 75833,
      "training_step_time": 7.104067802429199
    },
    {
      "epoch": 1.7490386962890626e-06,
      "model_forward_time": 1.7748308181762695,
      "step": 75834
    },
    {
      "epoch": 1.7490386962890626e-06,
      "step": 75834,
      "training_step_time": 7.485764265060425
    },
    {
      "epoch": 1.7499923706054687e-06,
      "model_forward_time": 1.7689287662506104,
      "step": 75835
    },
    {
      "epoch": 1.7499923706054687e-06,
      "step": 75835,
      "training_step_time": 8.359929084777832
    },
    {
      "epoch": 1.750946044921875e-06,
      "model_forward_time": 1.7684471607208252,
      "step": 75836
    },
    {
      "epoch": 1.750946044921875e-06,
      "step": 75836,
      "training_step_time": 7.638400077819824
    },
    {
      "epoch": 1.7518997192382812e-06,
      "model_forward_time": 2.425224542617798,
      "step": 75837
    },
    {
      "epoch": 1.7518997192382812e-06,
      "step": 75837,
      "training_step_time": 10.961697101593018
    },
    {
      "epoch": 1.7528533935546876e-06,
      "model_forward_time": 1.7718448638916016,
      "step": 75838
    },
    {
      "epoch": 1.7528533935546876e-06,
      "step": 75838,
      "training_step_time": 8.586726188659668
    },
    {
      "epoch": 1.7538070678710937e-06,
      "model_forward_time": 1.7648766040802002,
      "step": 75839
    },
    {
      "epoch": 1.7538070678710937e-06,
      "step": 75839,
      "training_step_time": 7.771486520767212
    },
    {
      "epoch": 1.7547607421875e-06,
      "grad_norm": 0.09026887267827988,
      "learning_rate": 1.5128444547409332e-06,
      "loss": 0.0557,
      "step": 75840
    },
    {
      "epoch": 1.7547607421875e-06,
      "model_forward_time": 1.7680039405822754,
      "step": 75840
    },
    {
      "dynamics_loss_avg": 0.036509291268885136,
      "epoch": 1.7547607421875e-06,
      "step": 75840
    },
    {
      "action_loss_avg": 0.0012494070804677903,
      "epoch": 1.7547607421875e-06,
      "step": 75840
    },
    {
      "epoch": 1.7547607421875e-06,
      "step": 75840,
      "training_step_time": 7.699493408203125
    },
    {
      "epoch": 1.7557144165039062e-06,
      "model_forward_time": 1.7555904388427734,
      "step": 75841
    },
    {
      "epoch": 1.7557144165039062e-06,
      "step": 75841,
      "training_step_time": 7.698923587799072
    },
    {
      "epoch": 1.7566680908203126e-06,
      "model_forward_time": 1.7427990436553955,
      "step": 75842
    },
    {
      "epoch": 1.7566680908203126e-06,
      "step": 75842,
      "training_step_time": 7.28182315826416
    },
    {
      "epoch": 1.7576217651367187e-06,
      "model_forward_time": 1.750281572341919,
      "step": 75843
    },
    {
      "epoch": 1.7576217651367187e-06,
      "step": 75843,
      "training_step_time": 7.969334602355957
    },
    {
      "epoch": 1.758575439453125e-06,
      "model_forward_time": 1.757699966430664,
      "step": 75844
    },
    {
      "epoch": 1.758575439453125e-06,
      "step": 75844,
      "training_step_time": 8.465900421142578
    },
    {
      "epoch": 1.7595291137695312e-06,
      "model_forward_time": 1.7616004943847656,
      "step": 75845
    },
    {
      "epoch": 1.7595291137695312e-06,
      "step": 75845,
      "training_step_time": 7.644204378128052
    },
    {
      "epoch": 1.7604827880859376e-06,
      "model_forward_time": 2.404309034347534,
      "step": 75846
    },
    {
      "epoch": 1.7604827880859376e-06,
      "step": 75846,
      "training_step_time": 9.3786141872406
    },
    {
      "epoch": 1.7614364624023437e-06,
      "model_forward_time": 1.756059169769287,
      "step": 75847
    },
    {
      "epoch": 1.7614364624023437e-06,
      "step": 75847,
      "training_step_time": 7.743166446685791
    },
    {
      "epoch": 1.76239013671875e-06,
      "model_forward_time": 1.7493045330047607,
      "step": 75848
    },
    {
      "epoch": 1.76239013671875e-06,
      "step": 75848,
      "training_step_time": 7.659616231918335
    },
    {
      "epoch": 1.7633438110351562e-06,
      "model_forward_time": 1.7546441555023193,
      "step": 75849
    },
    {
      "epoch": 1.7633438110351562e-06,
      "step": 75849,
      "training_step_time": 8.703024864196777
    },
    {
      "epoch": 1.7642974853515626e-06,
      "grad_norm": 0.10277043282985687,
      "learning_rate": 1.5116596832637864e-06,
      "loss": 0.0542,
      "step": 75850
    },
    {
      "epoch": 1.7642974853515626e-06,
      "model_forward_time": 1.7566709518432617,
      "step": 75850
    },
    {
      "dynamics_loss_avg": 0.053810260631144045,
      "epoch": 1.7642974853515626e-06,
      "step": 75850
    },
    {
      "action_loss_avg": 0.0011442891554906964,
      "epoch": 1.7642974853515626e-06,
      "step": 75850
    },
    {
      "epoch": 1.7642974853515626e-06,
      "step": 75850,
      "training_step_time": 7.747910976409912
    },
    {
      "epoch": 1.7652511596679687e-06,
      "model_forward_time": 1.7497241497039795,
      "step": 75851
    },
    {
      "epoch": 1.7652511596679687e-06,
      "step": 75851,
      "training_step_time": 7.3440468311309814
    },
    {
      "epoch": 1.766204833984375e-06,
      "model_forward_time": 1.7702360153198242,
      "step": 75852
    },
    {
      "epoch": 1.766204833984375e-06,
      "step": 75852,
      "training_step_time": 7.387054681777954
    },
    {
      "epoch": 1.7671585083007812e-06,
      "model_forward_time": 1.7665517330169678,
      "step": 75853
    },
    {
      "epoch": 1.7671585083007812e-06,
      "step": 75853,
      "training_step_time": 7.574180603027344
    },
    {
      "epoch": 1.7681121826171876e-06,
      "model_forward_time": 1.788139820098877,
      "step": 75854
    },
    {
      "epoch": 1.7681121826171876e-06,
      "step": 75854,
      "training_step_time": 7.689569711685181
    },
    {
      "epoch": 1.7690658569335937e-06,
      "model_forward_time": 2.3984766006469727,
      "step": 75855
    },
    {
      "epoch": 1.7690658569335937e-06,
      "step": 75855,
      "training_step_time": 8.395307779312134
    },
    {
      "epoch": 1.77001953125e-06,
      "model_forward_time": 1.4604308605194092,
      "step": 75856
    },
    {
      "epoch": 1.77001953125e-06,
      "step": 75856,
      "training_step_time": 8.271158695220947
    },
    {
      "epoch": 1.7709732055664062e-06,
      "model_forward_time": 1.763138771057129,
      "step": 75857
    },
    {
      "epoch": 1.7709732055664062e-06,
      "step": 75857,
      "training_step_time": 9.570781469345093
    },
    {
      "epoch": 1.7719268798828126e-06,
      "model_forward_time": 1.7579236030578613,
      "step": 75858
    },
    {
      "epoch": 1.7719268798828126e-06,
      "step": 75858,
      "training_step_time": 7.570499658584595
    },
    {
      "epoch": 1.7728805541992187e-06,
      "model_forward_time": 1.764861822128296,
      "step": 75859
    },
    {
      "epoch": 1.7728805541992187e-06,
      "step": 75859,
      "training_step_time": 7.778979063034058
    },
    {
      "epoch": 1.773834228515625e-06,
      "grad_norm": 0.12070901691913605,
      "learning_rate": 1.5104752932662953e-06,
      "loss": 0.0533,
      "step": 75860
    },
    {
      "epoch": 1.773834228515625e-06,
      "model_forward_time": 1.7650885581970215,
      "step": 75860
    },
    {
      "dynamics_loss_avg": 0.04379510944709182,
      "epoch": 1.773834228515625e-06,
      "step": 75860
    },
    {
      "action_loss_avg": 0.0009306464111432433,
      "epoch": 1.773834228515625e-06,
      "step": 75860
    },
    {
      "epoch": 1.773834228515625e-06,
      "step": 75860,
      "training_step_time": 7.32586669921875
    },
    {
      "epoch": 1.7747879028320312e-06,
      "model_forward_time": 1.7928454875946045,
      "step": 75861
    },
    {
      "epoch": 1.7747879028320312e-06,
      "step": 75861,
      "training_step_time": 7.343705892562866
    },
    {
      "epoch": 1.7757415771484376e-06,
      "model_forward_time": 1.761579990386963,
      "step": 75862
    },
    {
      "epoch": 1.7757415771484376e-06,
      "step": 75862,
      "training_step_time": 7.537856578826904
    },
    {
      "epoch": 1.7766952514648437e-06,
      "model_forward_time": 1.7592194080352783,
      "step": 75863
    },
    {
      "epoch": 1.7766952514648437e-06,
      "step": 75863,
      "training_step_time": 7.638686656951904
    },
    {
      "epoch": 1.77764892578125e-06,
      "model_forward_time": 2.4251649379730225,
      "step": 75864
    },
    {
      "epoch": 1.77764892578125e-06,
      "step": 75864,
      "training_step_time": 8.494948863983154
    },
    {
      "epoch": 1.7786026000976562e-06,
      "model_forward_time": 1.809065341949463,
      "step": 75865
    },
    {
      "epoch": 1.7786026000976562e-06,
      "step": 75865,
      "training_step_time": 9.632039070129395
    },
    {
      "epoch": 1.7795562744140626e-06,
      "model_forward_time": 1.762840747833252,
      "step": 75866
    },
    {
      "epoch": 1.7795562744140626e-06,
      "step": 75866,
      "training_step_time": 9.761030673980713
    },
    {
      "epoch": 1.7805099487304687e-06,
      "model_forward_time": 1.7674403190612793,
      "step": 75867
    },
    {
      "epoch": 1.7805099487304687e-06,
      "step": 75867,
      "training_step_time": 7.6350014209747314
    },
    {
      "epoch": 1.781463623046875e-06,
      "model_forward_time": 1.7658522129058838,
      "step": 75868
    },
    {
      "epoch": 1.781463623046875e-06,
      "step": 75868,
      "training_step_time": 7.695417881011963
    },
    {
      "epoch": 1.7824172973632812e-06,
      "model_forward_time": 1.7668828964233398,
      "step": 75869
    },
    {
      "epoch": 1.7824172973632812e-06,
      "step": 75869,
      "training_step_time": 7.656290531158447
    },
    {
      "epoch": 1.7833709716796876e-06,
      "grad_norm": 0.09226822108030319,
      "learning_rate": 1.5092912848779778e-06,
      "loss": 0.0534,
      "step": 75870
    },
    {
      "epoch": 1.7833709716796876e-06,
      "model_forward_time": 1.7330284118652344,
      "step": 75870
    },
    {
      "dynamics_loss_avg": 0.056908915378153326,
      "epoch": 1.7833709716796876e-06,
      "step": 75870
    },
    {
      "action_loss_avg": 0.0015076949261128903,
      "epoch": 1.7833709716796876e-06,
      "step": 75870
    },
    {
      "epoch": 1.7833709716796876e-06,
      "step": 75870,
      "training_step_time": 7.337172031402588
    },
    {
      "epoch": 1.7843246459960937e-06,
      "model_forward_time": 1.7565720081329346,
      "step": 75871
    },
    {
      "epoch": 1.7843246459960937e-06,
      "step": 75871,
      "training_step_time": 8.896555185317993
    },
    {
      "epoch": 1.7852783203125e-06,
      "model_forward_time": 1.7491800785064697,
      "step": 75872
    },
    {
      "epoch": 1.7852783203125e-06,
      "step": 75872,
      "training_step_time": 7.674035310745239
    },
    {
      "epoch": 1.7862319946289062e-06,
      "model_forward_time": 2.354337692260742,
      "step": 75873
    },
    {
      "epoch": 1.7862319946289062e-06,
      "step": 75873,
      "training_step_time": 8.241624355316162
    },
    {
      "epoch": 1.7871856689453126e-06,
      "model_forward_time": 1.757462501525879,
      "step": 75874
    },
    {
      "epoch": 1.7871856689453126e-06,
      "step": 75874,
      "training_step_time": 10.227030754089355
    },
    {
      "epoch": 1.7881393432617188e-06,
      "model_forward_time": 1.7469935417175293,
      "step": 75875
    },
    {
      "epoch": 1.7881393432617188e-06,
      "step": 75875,
      "training_step_time": 10.511671781539917
    },
    {
      "epoch": 1.789093017578125e-06,
      "model_forward_time": 1.7621769905090332,
      "step": 75876
    },
    {
      "epoch": 1.789093017578125e-06,
      "step": 75876,
      "training_step_time": 8.234007120132446
    },
    {
      "epoch": 1.7900466918945313e-06,
      "model_forward_time": 1.7567996978759766,
      "step": 75877
    },
    {
      "epoch": 1.7900466918945313e-06,
      "step": 75877,
      "training_step_time": 8.40662431716919
    },
    {
      "epoch": 1.7910003662109374e-06,
      "model_forward_time": 1.7472515106201172,
      "step": 75878
    },
    {
      "epoch": 1.7910003662109374e-06,
      "step": 75878,
      "training_step_time": 7.6667094230651855
    },
    {
      "epoch": 1.7919540405273438e-06,
      "model_forward_time": 1.7557625770568848,
      "step": 75879
    },
    {
      "epoch": 1.7919540405273438e-06,
      "step": 75879,
      "training_step_time": 8.546933889389038
    },
    {
      "epoch": 1.79290771484375e-06,
      "grad_norm": 0.09215682744979858,
      "learning_rate": 1.5081076582283195e-06,
      "loss": 0.0532,
      "step": 75880
    },
    {
      "epoch": 1.79290771484375e-06,
      "model_forward_time": 1.743398904800415,
      "step": 75880
    },
    {
      "dynamics_loss_avg": 0.05433524399995804,
      "epoch": 1.79290771484375e-06,
      "step": 75880
    },
    {
      "action_loss_avg": 0.0014560323965270073,
      "epoch": 1.79290771484375e-06,
      "step": 75880
    },
    {
      "epoch": 1.79290771484375e-06,
      "step": 75880,
      "training_step_time": 7.35764479637146
    },
    {
      "epoch": 1.7938613891601563e-06,
      "model_forward_time": 1.7509419918060303,
      "step": 75881
    },
    {
      "epoch": 1.7938613891601563e-06,
      "step": 75881,
      "training_step_time": 8.171663045883179
    },
    {
      "epoch": 1.7948150634765624e-06,
      "model_forward_time": 1.7543761730194092,
      "step": 75882
    },
    {
      "epoch": 1.7948150634765624e-06,
      "step": 75882,
      "training_step_time": 9.295747756958008
    },
    {
      "epoch": 1.7957687377929688e-06,
      "model_forward_time": 2.3756701946258545,
      "step": 75883
    },
    {
      "epoch": 1.7957687377929688e-06,
      "step": 75883,
      "training_step_time": 7.656982898712158
    },
    {
      "epoch": 1.796722412109375e-06,
      "model_forward_time": 1.830353021621704,
      "step": 75884
    },
    {
      "epoch": 1.796722412109375e-06,
      "step": 75884,
      "training_step_time": 9.30760669708252
    },
    {
      "epoch": 1.7976760864257813e-06,
      "model_forward_time": 1.7600700855255127,
      "step": 75885
    },
    {
      "epoch": 1.7976760864257813e-06,
      "step": 75885,
      "training_step_time": 7.719679117202759
    },
    {
      "epoch": 1.7986297607421874e-06,
      "model_forward_time": 1.749924898147583,
      "step": 75886
    },
    {
      "epoch": 1.7986297607421874e-06,
      "step": 75886,
      "training_step_time": 7.950812101364136
    },
    {
      "epoch": 1.7995834350585938e-06,
      "model_forward_time": 1.766883373260498,
      "step": 75887
    },
    {
      "epoch": 1.7995834350585938e-06,
      "step": 75887,
      "training_step_time": 7.644093036651611
    },
    {
      "epoch": 1.800537109375e-06,
      "model_forward_time": 1.7417240142822266,
      "step": 75888
    },
    {
      "epoch": 1.800537109375e-06,
      "step": 75888,
      "training_step_time": 8.32715654373169
    },
    {
      "epoch": 1.8014907836914063e-06,
      "model_forward_time": 1.754610300064087,
      "step": 75889
    },
    {
      "epoch": 1.8014907836914063e-06,
      "step": 75889,
      "training_step_time": 7.92934513092041
    },
    {
      "epoch": 1.8024444580078124e-06,
      "grad_norm": 0.0989026427268982,
      "learning_rate": 1.5069244134467553e-06,
      "loss": 0.0535,
      "step": 75890
    },
    {
      "epoch": 1.8024444580078124e-06,
      "model_forward_time": 1.7501447200775146,
      "step": 75890
    },
    {
      "dynamics_loss_avg": 0.042577506089583037,
      "epoch": 1.8024444580078124e-06,
      "step": 75890
    },
    {
      "action_loss_avg": 0.0011528324030223303,
      "epoch": 1.8024444580078124e-06,
      "step": 75890
    },
    {
      "epoch": 1.8024444580078124e-06,
      "step": 75890,
      "training_step_time": 7.655864953994751
    },
    {
      "epoch": 1.8033981323242188e-06,
      "model_forward_time": 1.7444775104522705,
      "step": 75891
    },
    {
      "epoch": 1.8033981323242188e-06,
      "step": 75891,
      "training_step_time": 7.705465316772461
    },
    {
      "epoch": 1.804351806640625e-06,
      "model_forward_time": 2.37998628616333,
      "step": 75892
    },
    {
      "epoch": 1.804351806640625e-06,
      "step": 75892,
      "training_step_time": 7.588682413101196
    },
    {
      "epoch": 1.8053054809570313e-06,
      "model_forward_time": 1.8620414733886719,
      "step": 75893
    },
    {
      "epoch": 1.8053054809570313e-06,
      "step": 75893,
      "training_step_time": 7.769517183303833
    },
    {
      "epoch": 1.8062591552734374e-06,
      "model_forward_time": 1.7529375553131104,
      "step": 75894
    },
    {
      "epoch": 1.8062591552734374e-06,
      "step": 75894,
      "training_step_time": 8.42544674873352
    },
    {
      "epoch": 1.8072128295898438e-06,
      "model_forward_time": 1.7558624744415283,
      "step": 75895
    },
    {
      "epoch": 1.8072128295898438e-06,
      "step": 75895,
      "training_step_time": 8.379463195800781
    },
    {
      "epoch": 1.80816650390625e-06,
      "model_forward_time": 1.7598938941955566,
      "step": 75896
    },
    {
      "epoch": 1.80816650390625e-06,
      "step": 75896,
      "training_step_time": 7.789093971252441
    },
    {
      "epoch": 1.8091201782226563e-06,
      "model_forward_time": 1.7571518421173096,
      "step": 75897
    },
    {
      "epoch": 1.8091201782226563e-06,
      "step": 75897,
      "training_step_time": 7.122253656387329
    },
    {
      "epoch": 1.8100738525390624e-06,
      "model_forward_time": 1.7447478771209717,
      "step": 75898
    },
    {
      "epoch": 1.8100738525390624e-06,
      "step": 75898,
      "training_step_time": 7.298437833786011
    },
    {
      "epoch": 1.8110275268554688e-06,
      "model_forward_time": 1.7464706897735596,
      "step": 75899
    },
    {
      "epoch": 1.8110275268554688e-06,
      "step": 75899,
      "training_step_time": 8.163918495178223
    },
    {
      "epoch": 1.811981201171875e-06,
      "grad_norm": 0.1011364758014679,
      "learning_rate": 1.5057415506626877e-06,
      "loss": 0.0522,
      "step": 75900
    },
    {
      "epoch": 1.811981201171875e-06,
      "model_forward_time": 1.7522571086883545,
      "step": 75900
    },
    {
      "dynamics_loss_avg": 0.05477075073868036,
      "epoch": 1.811981201171875e-06,
      "step": 75900
    },
    {
      "action_loss_avg": 0.0019186737481504679,
      "epoch": 1.811981201171875e-06,
      "step": 75900
    },
    {
      "epoch": 1.811981201171875e-06,
      "step": 75900,
      "training_step_time": 7.616369962692261
    },
    {
      "epoch": 1.8129348754882813e-06,
      "model_forward_time": 2.378946542739868,
      "step": 75901
    },
    {
      "epoch": 1.8129348754882813e-06,
      "step": 75901,
      "training_step_time": 7.616015195846558
    },
    {
      "epoch": 1.8138885498046874e-06,
      "model_forward_time": 1.8271644115447998,
      "step": 75902
    },
    {
      "epoch": 1.8138885498046874e-06,
      "step": 75902,
      "training_step_time": 8.759386539459229
    },
    {
      "epoch": 1.8148422241210938e-06,
      "model_forward_time": 1.7517154216766357,
      "step": 75903
    },
    {
      "epoch": 1.8148422241210938e-06,
      "step": 75903,
      "training_step_time": 8.549379348754883
    },
    {
      "epoch": 1.8157958984375e-06,
      "model_forward_time": 1.7578177452087402,
      "step": 75904
    },
    {
      "epoch": 1.8157958984375e-06,
      "step": 75904,
      "training_step_time": 7.880943059921265
    },
    {
      "epoch": 1.8167495727539063e-06,
      "model_forward_time": 1.749582290649414,
      "step": 75905
    },
    {
      "epoch": 1.8167495727539063e-06,
      "step": 75905,
      "training_step_time": 7.77020788192749
    },
    {
      "epoch": 1.8177032470703124e-06,
      "model_forward_time": 1.748203992843628,
      "step": 75906
    },
    {
      "epoch": 1.8177032470703124e-06,
      "step": 75906,
      "training_step_time": 7.671117782592773
    },
    {
      "epoch": 1.8186569213867188e-06,
      "model_forward_time": 1.7544009685516357,
      "step": 75907
    },
    {
      "epoch": 1.8186569213867188e-06,
      "step": 75907,
      "training_step_time": 7.953160762786865
    },
    {
      "epoch": 1.819610595703125e-06,
      "model_forward_time": 1.3875606060028076,
      "step": 75908
    },
    {
      "epoch": 1.819610595703125e-06,
      "step": 75908,
      "training_step_time": 7.717245578765869
    },
    {
      "epoch": 1.8205642700195313e-06,
      "model_forward_time": 1.7505948543548584,
      "step": 75909
    },
    {
      "epoch": 1.8205642700195313e-06,
      "step": 75909,
      "training_step_time": 8.528154373168945
    },
    {
      "epoch": 1.8215179443359374e-06,
      "grad_norm": 0.0856572836637497,
      "learning_rate": 1.5045590700054669e-06,
      "loss": 0.053,
      "step": 75910
    },
    {
      "epoch": 1.8215179443359374e-06,
      "model_forward_time": 2.362617015838623,
      "step": 75910
    },
    {
      "dynamics_loss_avg": 0.04049165137112141,
      "epoch": 1.8215179443359374e-06,
      "step": 75910
    },
    {
      "action_loss_avg": 0.0013175370084354654,
      "epoch": 1.8215179443359374e-06,
      "step": 75910
    },
    {
      "epoch": 1.8215179443359374e-06,
      "step": 75910,
      "training_step_time": 8.287380456924438
    },
    {
      "epoch": 1.8224716186523438e-06,
      "model_forward_time": 1.8009858131408691,
      "step": 75911
    },
    {
      "epoch": 1.8224716186523438e-06,
      "step": 75911,
      "training_step_time": 7.711628675460815
    },
    {
      "epoch": 1.82342529296875e-06,
      "model_forward_time": 1.7450711727142334,
      "step": 75912
    },
    {
      "epoch": 1.82342529296875e-06,
      "step": 75912,
      "training_step_time": 10.068912506103516
    },
    {
      "epoch": 1.8243789672851563e-06,
      "model_forward_time": 1.7583842277526855,
      "step": 75913
    },
    {
      "epoch": 1.8243789672851563e-06,
      "step": 75913,
      "training_step_time": 8.996719360351562
    },
    {
      "epoch": 1.8253326416015624e-06,
      "model_forward_time": 1.7595796585083008,
      "step": 75914
    },
    {
      "epoch": 1.8253326416015624e-06,
      "step": 75914,
      "training_step_time": 7.777684688568115
    },
    {
      "epoch": 1.8262863159179688e-06,
      "model_forward_time": 1.7477390766143799,
      "step": 75915
    },
    {
      "epoch": 1.8262863159179688e-06,
      "step": 75915,
      "training_step_time": 7.743937730789185
    },
    {
      "epoch": 1.827239990234375e-06,
      "model_forward_time": 1.7730226516723633,
      "step": 75916
    },
    {
      "epoch": 1.827239990234375e-06,
      "step": 75916,
      "training_step_time": 7.394261837005615
    },
    {
      "epoch": 1.8281936645507813e-06,
      "model_forward_time": 1.758732795715332,
      "step": 75917
    },
    {
      "epoch": 1.8281936645507813e-06,
      "step": 75917,
      "training_step_time": 9.398077964782715
    },
    {
      "epoch": 1.8291473388671874e-06,
      "model_forward_time": 1.7516064643859863,
      "step": 75918
    },
    {
      "epoch": 1.8291473388671874e-06,
      "step": 75918,
      "training_step_time": 8.534097671508789
    },
    {
      "epoch": 1.8301010131835938e-06,
      "model_forward_time": 1.7514653205871582,
      "step": 75919
    },
    {
      "epoch": 1.8301010131835938e-06,
      "step": 75919,
      "training_step_time": 9.628633260726929
    },
    {
      "epoch": 1.8310546875e-06,
      "grad_norm": 0.07612788677215576,
      "learning_rate": 1.5033769716044144e-06,
      "loss": 0.0531,
      "step": 75920
    },
    {
      "epoch": 1.8310546875e-06,
      "model_forward_time": 1.7757172584533691,
      "step": 75920
    },
    {
      "dynamics_loss_avg": 0.04249912798404694,
      "epoch": 1.8310546875e-06,
      "step": 75920
    },
    {
      "action_loss_avg": 0.000899275676056277,
      "epoch": 1.8310546875e-06,
      "step": 75920
    },
    {
      "epoch": 1.8310546875e-06,
      "step": 75920,
      "training_step_time": 8.279126644134521
    },
    {
      "epoch": 1.8320083618164063e-06,
      "model_forward_time": 1.7538857460021973,
      "step": 75921
    },
    {
      "epoch": 1.8320083618164063e-06,
      "step": 75921,
      "training_step_time": 8.747223615646362
    },
    {
      "epoch": 1.8329620361328124e-06,
      "model_forward_time": 1.7597992420196533,
      "step": 75922
    },
    {
      "epoch": 1.8329620361328124e-06,
      "step": 75922,
      "training_step_time": 7.955560922622681
    },
    {
      "epoch": 1.8339157104492188e-06,
      "model_forward_time": 1.7636840343475342,
      "step": 75923
    },
    {
      "epoch": 1.8339157104492188e-06,
      "step": 75923,
      "training_step_time": 8.659723043441772
    },
    {
      "epoch": 1.834869384765625e-06,
      "model_forward_time": 1.755051612854004,
      "step": 75924
    },
    {
      "epoch": 1.834869384765625e-06,
      "step": 75924,
      "training_step_time": 7.693299055099487
    },
    {
      "epoch": 1.8358230590820313e-06,
      "model_forward_time": 1.7529857158660889,
      "step": 75925
    },
    {
      "epoch": 1.8358230590820313e-06,
      "step": 75925,
      "training_step_time": 7.299546241760254
    },
    {
      "epoch": 1.8367767333984374e-06,
      "model_forward_time": 1.758819818496704,
      "step": 75926
    },
    {
      "epoch": 1.8367767333984374e-06,
      "step": 75926,
      "training_step_time": 7.963204383850098
    },
    {
      "epoch": 1.8377304077148438e-06,
      "model_forward_time": 1.7432360649108887,
      "step": 75927
    },
    {
      "epoch": 1.8377304077148438e-06,
      "step": 75927,
      "training_step_time": 7.547124147415161
    },
    {
      "epoch": 1.83868408203125e-06,
      "model_forward_time": 1.7513148784637451,
      "step": 75928
    },
    {
      "epoch": 1.83868408203125e-06,
      "step": 75928,
      "training_step_time": 7.6610424518585205
    },
    {
      "epoch": 1.8396377563476563e-06,
      "model_forward_time": 2.380662441253662,
      "step": 75929
    },
    {
      "epoch": 1.8396377563476563e-06,
      "step": 75929,
      "training_step_time": 10.60633373260498
    },
    {
      "epoch": 1.8405914306640624e-06,
      "grad_norm": 0.07478971779346466,
      "learning_rate": 1.5021952555887976e-06,
      "loss": 0.0524,
      "step": 75930
    },
    {
      "epoch": 1.8405914306640624e-06,
      "model_forward_time": 1.74284029006958,
      "step": 75930
    },
    {
      "dynamics_loss_avg": 0.042906501516699794,
      "epoch": 1.8405914306640624e-06,
      "step": 75930
    },
    {
      "action_loss_avg": 0.0009253785683540627,
      "epoch": 1.8405914306640624e-06,
      "step": 75930
    },
    {
      "epoch": 1.8405914306640624e-06,
      "step": 75930,
      "training_step_time": 7.689321756362915
    },
    {
      "epoch": 1.8415451049804688e-06,
      "model_forward_time": 1.759690284729004,
      "step": 75931
    },
    {
      "epoch": 1.8415451049804688e-06,
      "step": 75931,
      "training_step_time": 10.330858707427979
    },
    {
      "epoch": 1.842498779296875e-06,
      "model_forward_time": 1.7507166862487793,
      "step": 75932
    },
    {
      "epoch": 1.842498779296875e-06,
      "step": 75932,
      "training_step_time": 8.51794147491455
    },
    {
      "epoch": 1.8434524536132813e-06,
      "model_forward_time": 1.752763271331787,
      "step": 75933
    },
    {
      "epoch": 1.8434524536132813e-06,
      "step": 75933,
      "training_step_time": 7.693418025970459
    },
    {
      "epoch": 1.8444061279296874e-06,
      "model_forward_time": 1.7529630661010742,
      "step": 75934
    },
    {
      "epoch": 1.8444061279296874e-06,
      "step": 75934,
      "training_step_time": 7.638431787490845
    },
    {
      "epoch": 1.8453598022460938e-06,
      "model_forward_time": 1.7534599304199219,
      "step": 75935
    },
    {
      "epoch": 1.8453598022460938e-06,
      "step": 75935,
      "training_step_time": 7.528084993362427
    },
    {
      "epoch": 1.8463134765625e-06,
      "model_forward_time": 1.7491121292114258,
      "step": 75936
    },
    {
      "epoch": 1.8463134765625e-06,
      "step": 75936,
      "training_step_time": 8.389524459838867
    },
    {
      "epoch": 1.8472671508789063e-06,
      "model_forward_time": 1.7547991275787354,
      "step": 75937
    },
    {
      "epoch": 1.8472671508789063e-06,
      "step": 75937,
      "training_step_time": 7.676074743270874
    },
    {
      "epoch": 1.8482208251953124e-06,
      "model_forward_time": 2.381275177001953,
      "step": 75938
    },
    {
      "epoch": 1.8482208251953124e-06,
      "step": 75938,
      "training_step_time": 9.160597085952759
    },
    {
      "epoch": 1.8491744995117188e-06,
      "model_forward_time": 1.8552114963531494,
      "step": 75939
    },
    {
      "epoch": 1.8491744995117188e-06,
      "step": 75939,
      "training_step_time": 7.706868648529053
    },
    {
      "epoch": 1.850128173828125e-06,
      "grad_norm": 0.10480114072561264,
      "learning_rate": 1.50101392208785e-06,
      "loss": 0.0532,
      "step": 75940
    },
    {
      "epoch": 1.850128173828125e-06,
      "model_forward_time": 1.7502446174621582,
      "step": 75940
    },
    {
      "dynamics_loss_avg": 0.03957509491592646,
      "epoch": 1.850128173828125e-06,
      "step": 75940
    },
    {
      "action_loss_avg": 0.0013409183302428573,
      "epoch": 1.850128173828125e-06,
      "step": 75940
    },
    {
      "epoch": 1.850128173828125e-06,
      "step": 75940,
      "training_step_time": 7.749515533447266
    },
    {
      "epoch": 1.8510818481445313e-06,
      "model_forward_time": 1.755906105041504,
      "step": 75941
    },
    {
      "epoch": 1.8510818481445313e-06,
      "step": 75941,
      "training_step_time": 7.8954503536224365
    },
    {
      "epoch": 1.8520355224609375e-06,
      "model_forward_time": 1.7556707859039307,
      "step": 75942
    },
    {
      "epoch": 1.8520355224609375e-06,
      "step": 75942,
      "training_step_time": 7.704737186431885
    },
    {
      "epoch": 1.8529891967773438e-06,
      "model_forward_time": 1.7591536045074463,
      "step": 75943
    },
    {
      "epoch": 1.8529891967773438e-06,
      "step": 75943,
      "training_step_time": 7.672070503234863
    },
    {
      "epoch": 1.85394287109375e-06,
      "model_forward_time": 1.7668559551239014,
      "step": 75944
    },
    {
      "epoch": 1.85394287109375e-06,
      "step": 75944,
      "training_step_time": 7.525862693786621
    },
    {
      "epoch": 1.8548965454101563e-06,
      "model_forward_time": 1.7472004890441895,
      "step": 75945
    },
    {
      "epoch": 1.8548965454101563e-06,
      "step": 75945,
      "training_step_time": 8.089410305023193
    },
    {
      "epoch": 1.8558502197265625e-06,
      "model_forward_time": 1.7564570903778076,
      "step": 75946
    },
    {
      "epoch": 1.8558502197265625e-06,
      "step": 75946,
      "training_step_time": 10.122762680053711
    },
    {
      "epoch": 1.8568038940429688e-06,
      "model_forward_time": 2.3853702545166016,
      "step": 75947
    },
    {
      "epoch": 1.8568038940429688e-06,
      "step": 75947,
      "training_step_time": 7.623550653457642
    },
    {
      "epoch": 1.857757568359375e-06,
      "model_forward_time": 1.8115730285644531,
      "step": 75948
    },
    {
      "epoch": 1.857757568359375e-06,
      "step": 75948,
      "training_step_time": 8.2099130153656
    },
    {
      "epoch": 1.8587112426757813e-06,
      "model_forward_time": 1.7485151290893555,
      "step": 75949
    },
    {
      "epoch": 1.8587112426757813e-06,
      "step": 75949,
      "training_step_time": 7.728060483932495
    },
    {
      "epoch": 1.8596649169921875e-06,
      "grad_norm": 0.0850323811173439,
      "learning_rate": 1.499832971230758e-06,
      "loss": 0.0524,
      "step": 75950
    },
    {
      "epoch": 1.8596649169921875e-06,
      "model_forward_time": 1.7490673065185547,
      "step": 75950
    },
    {
      "dynamics_loss_avg": 0.04954294115304947,
      "epoch": 1.8596649169921875e-06,
      "step": 75950
    },
    {
      "action_loss_avg": 0.0008822786097880453,
      "epoch": 1.8596649169921875e-06,
      "step": 75950
    },
    {
      "epoch": 1.8596649169921875e-06,
      "step": 75950,
      "training_step_time": 8.831716299057007
    },
    {
      "epoch": 1.8606185913085938e-06,
      "model_forward_time": 1.7527832984924316,
      "step": 75951
    },
    {
      "epoch": 1.8606185913085938e-06,
      "step": 75951,
      "training_step_time": 7.71433162689209
    },
    {
      "epoch": 1.861572265625e-06,
      "model_forward_time": 1.7517890930175781,
      "step": 75952
    },
    {
      "epoch": 1.861572265625e-06,
      "step": 75952,
      "training_step_time": 7.434497594833374
    },
    {
      "epoch": 1.8625259399414063e-06,
      "model_forward_time": 1.7504057884216309,
      "step": 75953
    },
    {
      "epoch": 1.8625259399414063e-06,
      "step": 75953,
      "training_step_time": 7.7525904178619385
    },
    {
      "epoch": 1.8634796142578125e-06,
      "model_forward_time": 1.7581918239593506,
      "step": 75954
    },
    {
      "epoch": 1.8634796142578125e-06,
      "step": 75954,
      "training_step_time": 9.074503898620605
    },
    {
      "epoch": 1.8644332885742188e-06,
      "model_forward_time": 1.756190299987793,
      "step": 75955
    },
    {
      "epoch": 1.8644332885742188e-06,
      "step": 75955,
      "training_step_time": 7.683339357376099
    },
    {
      "epoch": 1.865386962890625e-06,
      "model_forward_time": 2.4113080501556396,
      "step": 75956
    },
    {
      "epoch": 1.865386962890625e-06,
      "step": 75956,
      "training_step_time": 7.705276727676392
    },
    {
      "epoch": 1.8663406372070313e-06,
      "model_forward_time": 1.8018608093261719,
      "step": 75957
    },
    {
      "epoch": 1.8663406372070313e-06,
      "step": 75957,
      "training_step_time": 7.6663432121276855
    },
    {
      "epoch": 1.8672943115234375e-06,
      "model_forward_time": 1.75541090965271,
      "step": 75958
    },
    {
      "epoch": 1.8672943115234375e-06,
      "step": 75958,
      "training_step_time": 7.750897169113159
    },
    {
      "epoch": 1.8682479858398438e-06,
      "model_forward_time": 1.7573745250701904,
      "step": 75959
    },
    {
      "epoch": 1.8682479858398438e-06,
      "step": 75959,
      "training_step_time": 8.378190279006958
    },
    {
      "epoch": 1.86920166015625e-06,
      "grad_norm": 0.08489461243152618,
      "learning_rate": 1.498652403146671e-06,
      "loss": 0.0534,
      "step": 75960
    },
    {
      "epoch": 1.86920166015625e-06,
      "model_forward_time": 1.7543585300445557,
      "step": 75960
    },
    {
      "dynamics_loss_avg": 0.04271984044462442,
      "epoch": 1.86920166015625e-06,
      "step": 75960
    },
    {
      "action_loss_avg": 0.0013832023076247423,
      "epoch": 1.86920166015625e-06,
      "step": 75960
    },
    {
      "epoch": 1.86920166015625e-06,
      "step": 75960,
      "training_step_time": 7.851528167724609
    },
    {
      "epoch": 1.8701553344726563e-06,
      "model_forward_time": 1.75545072555542,
      "step": 75961
    },
    {
      "epoch": 1.8701553344726563e-06,
      "step": 75961,
      "training_step_time": 7.805037975311279
    },
    {
      "epoch": 1.8711090087890625e-06,
      "model_forward_time": 1.7645440101623535,
      "step": 75962
    },
    {
      "epoch": 1.8711090087890625e-06,
      "step": 75962,
      "training_step_time": 7.551800012588501
    },
    {
      "epoch": 1.8720626831054688e-06,
      "model_forward_time": 1.7537775039672852,
      "step": 75963
    },
    {
      "epoch": 1.8720626831054688e-06,
      "step": 75963,
      "training_step_time": 8.391714334487915
    },
    {
      "epoch": 1.873016357421875e-06,
      "model_forward_time": 1.7550115585327148,
      "step": 75964
    },
    {
      "epoch": 1.873016357421875e-06,
      "step": 75964,
      "training_step_time": 7.696727752685547
    },
    {
      "epoch": 1.8739700317382813e-06,
      "model_forward_time": 2.3739664554595947,
      "step": 75965
    },
    {
      "epoch": 1.8739700317382813e-06,
      "step": 75965,
      "training_step_time": 8.526849508285522
    },
    {
      "epoch": 1.8749237060546875e-06,
      "model_forward_time": 1.7890799045562744,
      "step": 75966
    },
    {
      "epoch": 1.8749237060546875e-06,
      "step": 75966,
      "training_step_time": 9.679170370101929
    },
    {
      "epoch": 1.8758773803710938e-06,
      "model_forward_time": 1.7525222301483154,
      "step": 75967
    },
    {
      "epoch": 1.8758773803710938e-06,
      "step": 75967,
      "training_step_time": 8.6584153175354
    },
    {
      "epoch": 1.8768310546875e-06,
      "model_forward_time": 1.7670917510986328,
      "step": 75968
    },
    {
      "epoch": 1.8768310546875e-06,
      "step": 75968,
      "training_step_time": 7.59236478805542
    },
    {
      "epoch": 1.8777847290039063e-06,
      "model_forward_time": 1.756227970123291,
      "step": 75969
    },
    {
      "epoch": 1.8777847290039063e-06,
      "step": 75969,
      "training_step_time": 9.20960021018982
    },
    {
      "epoch": 1.8787384033203125e-06,
      "grad_norm": 0.11380461603403091,
      "learning_rate": 1.4974722179646917e-06,
      "loss": 0.052,
      "step": 75970
    },
    {
      "epoch": 1.8787384033203125e-06,
      "model_forward_time": 1.7567131519317627,
      "step": 75970
    },
    {
      "dynamics_loss_avg": 0.045859864354133605,
      "epoch": 1.8787384033203125e-06,
      "step": 75970
    },
    {
      "action_loss_avg": 0.0010549644444836303,
      "epoch": 1.8787384033203125e-06,
      "step": 75970
    },
    {
      "epoch": 1.8787384033203125e-06,
      "step": 75970,
      "training_step_time": 8.836999416351318
    },
    {
      "epoch": 1.8796920776367188e-06,
      "model_forward_time": 1.7545533180236816,
      "step": 75971
    },
    {
      "epoch": 1.8796920776367188e-06,
      "step": 75971,
      "training_step_time": 8.180946350097656
    },
    {
      "epoch": 1.880645751953125e-06,
      "model_forward_time": 1.7515513896942139,
      "step": 75972
    },
    {
      "epoch": 1.880645751953125e-06,
      "step": 75972,
      "training_step_time": 8.161184072494507
    },
    {
      "epoch": 1.8815994262695313e-06,
      "model_forward_time": 1.392247200012207,
      "step": 75973
    },
    {
      "epoch": 1.8815994262695313e-06,
      "step": 75973,
      "training_step_time": 7.641768932342529
    },
    {
      "epoch": 1.8825531005859375e-06,
      "model_forward_time": 1.759756326675415,
      "step": 75974
    },
    {
      "epoch": 1.8825531005859375e-06,
      "step": 75974,
      "training_step_time": 8.489440202713013
    },
    {
      "epoch": 1.8835067749023438e-06,
      "model_forward_time": 2.407168388366699,
      "step": 75975
    },
    {
      "epoch": 1.8835067749023438e-06,
      "step": 75975,
      "training_step_time": 9.268753051757812
    },
    {
      "epoch": 1.88446044921875e-06,
      "model_forward_time": 1.8598213195800781,
      "step": 75976
    },
    {
      "epoch": 1.88446044921875e-06,
      "step": 75976,
      "training_step_time": 7.72870135307312
    },
    {
      "epoch": 1.8854141235351563e-06,
      "model_forward_time": 1.754953145980835,
      "step": 75977
    },
    {
      "epoch": 1.8854141235351563e-06,
      "step": 75977,
      "training_step_time": 9.266774654388428
    },
    {
      "epoch": 1.8863677978515625e-06,
      "model_forward_time": 1.7536425590515137,
      "step": 75978
    },
    {
      "epoch": 1.8863677978515625e-06,
      "step": 75978,
      "training_step_time": 7.888859748840332
    },
    {
      "epoch": 1.8873214721679688e-06,
      "model_forward_time": 1.753929615020752,
      "step": 75979
    },
    {
      "epoch": 1.8873214721679688e-06,
      "step": 75979,
      "training_step_time": 7.685402154922485
    },
    {
      "epoch": 1.888275146484375e-06,
      "grad_norm": 0.12147250771522522,
      "learning_rate": 1.4962924158138848e-06,
      "loss": 0.053,
      "step": 75980
    },
    {
      "epoch": 1.888275146484375e-06,
      "model_forward_time": 1.7515106201171875,
      "step": 75980
    },
    {
      "dynamics_loss_avg": 0.05491095613688231,
      "epoch": 1.888275146484375e-06,
      "step": 75980
    },
    {
      "action_loss_avg": 0.0016201687743887306,
      "epoch": 1.888275146484375e-06,
      "step": 75980
    },
    {
      "epoch": 1.888275146484375e-06,
      "step": 75980,
      "training_step_time": 7.759572267532349
    },
    {
      "epoch": 1.8892288208007813e-06,
      "model_forward_time": 1.753274917602539,
      "step": 75981
    },
    {
      "epoch": 1.8892288208007813e-06,
      "step": 75981,
      "training_step_time": 7.443205118179321
    },
    {
      "epoch": 1.8901824951171875e-06,
      "model_forward_time": 1.7536308765411377,
      "step": 75982
    },
    {
      "epoch": 1.8901824951171875e-06,
      "step": 75982,
      "training_step_time": 7.612470865249634
    },
    {
      "epoch": 1.8911361694335938e-06,
      "model_forward_time": 1.7508373260498047,
      "step": 75983
    },
    {
      "epoch": 1.8911361694335938e-06,
      "step": 75983,
      "training_step_time": 9.069775581359863
    },
    {
      "epoch": 1.89208984375e-06,
      "model_forward_time": 2.3876662254333496,
      "step": 75984
    },
    {
      "epoch": 1.89208984375e-06,
      "step": 75984,
      "training_step_time": 7.584134578704834
    },
    {
      "epoch": 1.8930435180664063e-06,
      "model_forward_time": 1.8697073459625244,
      "step": 75985
    },
    {
      "epoch": 1.8930435180664063e-06,
      "step": 75985,
      "training_step_time": 8.268176078796387
    },
    {
      "epoch": 1.8939971923828125e-06,
      "model_forward_time": 1.7498092651367188,
      "step": 75986
    },
    {
      "epoch": 1.8939971923828125e-06,
      "step": 75986,
      "training_step_time": 7.656287670135498
    },
    {
      "epoch": 1.8949508666992188e-06,
      "model_forward_time": 1.755847692489624,
      "step": 75987
    },
    {
      "epoch": 1.8949508666992188e-06,
      "step": 75987,
      "training_step_time": 8.546583890914917
    },
    {
      "epoch": 1.895904541015625e-06,
      "model_forward_time": 1.776982307434082,
      "step": 75988
    },
    {
      "epoch": 1.895904541015625e-06,
      "step": 75988,
      "training_step_time": 8.636473655700684
    },
    {
      "epoch": 1.8968582153320313e-06,
      "model_forward_time": 1.7564277648925781,
      "step": 75989
    },
    {
      "epoch": 1.8968582153320313e-06,
      "step": 75989,
      "training_step_time": 7.404210329055786
    },
    {
      "epoch": 1.8978118896484375e-06,
      "grad_norm": 0.10159394145011902,
      "learning_rate": 1.4951129968232714e-06,
      "loss": 0.0535,
      "step": 75990
    },
    {
      "epoch": 1.8978118896484375e-06,
      "model_forward_time": 1.7484526634216309,
      "step": 75990
    },
    {
      "dynamics_loss_avg": 0.060818524844944474,
      "epoch": 1.8978118896484375e-06,
      "step": 75990
    },
    {
      "action_loss_avg": 0.0013139430986484513,
      "epoch": 1.8978118896484375e-06,
      "step": 75990
    },
    {
      "epoch": 1.8978118896484375e-06,
      "step": 75990,
      "training_step_time": 7.432272911071777
    },
    {
      "epoch": 1.8987655639648438e-06,
      "model_forward_time": 1.7538349628448486,
      "step": 75991
    },
    {
      "epoch": 1.8987655639648438e-06,
      "step": 75991,
      "training_step_time": 7.3834288120269775
    },
    {
      "epoch": 1.89971923828125e-06,
      "model_forward_time": 1.7623398303985596,
      "step": 75992
    },
    {
      "epoch": 1.89971923828125e-06,
      "step": 75992,
      "training_step_time": 7.615016937255859
    },
    {
      "epoch": 1.9006729125976563e-06,
      "model_forward_time": 2.421313762664795,
      "step": 75993
    },
    {
      "epoch": 1.9006729125976563e-06,
      "step": 75993,
      "training_step_time": 8.317948579788208
    },
    {
      "epoch": 1.9016265869140625e-06,
      "model_forward_time": 1.8433187007904053,
      "step": 75994
    },
    {
      "epoch": 1.9016265869140625e-06,
      "step": 75994,
      "training_step_time": 7.614936828613281
    },
    {
      "epoch": 1.9025802612304689e-06,
      "model_forward_time": 1.7618615627288818,
      "step": 75995
    },
    {
      "epoch": 1.9025802612304689e-06,
      "step": 75995,
      "training_step_time": 8.457190752029419
    },
    {
      "epoch": 1.903533935546875e-06,
      "model_forward_time": 1.7651407718658447,
      "step": 75996
    },
    {
      "epoch": 1.903533935546875e-06,
      "step": 75996,
      "training_step_time": 7.534549713134766
    },
    {
      "epoch": 1.9044876098632814e-06,
      "model_forward_time": 1.7615432739257812,
      "step": 75997
    },
    {
      "epoch": 1.9044876098632814e-06,
      "step": 75997,
      "training_step_time": 7.735843181610107
    },
    {
      "epoch": 1.9054412841796875e-06,
      "model_forward_time": 1.7767412662506104,
      "step": 75998
    },
    {
      "epoch": 1.9054412841796875e-06,
      "step": 75998,
      "training_step_time": 7.896059274673462
    },
    {
      "epoch": 1.9063949584960939e-06,
      "model_forward_time": 1.8152976036071777,
      "step": 75999
    },
    {
      "epoch": 1.9063949584960939e-06,
      "step": 75999,
      "training_step_time": 7.670195817947388
    },
    {
      "epoch": 1.9073486328125e-06,
      "grad_norm": 0.09167490154504776,
      "learning_rate": 1.4939339611218328e-06,
      "loss": 0.053,
      "step": 76000
    },
    {
      "epoch": 1.9073486328125e-06,
      "model_forward_time": 1.9592242240905762,
      "step": 76000
    },
    {
      "dynamics_loss_avg": 0.04122888995334506,
      "epoch": 1.9073486328125e-06,
      "step": 76000
    },
    {
      "action_loss_avg": 0.0017721438664011658,
      "epoch": 1.9073486328125e-06,
      "step": 76000
    },
    {
      "epoch": 1.9073486328125e-06,
      "step": 76000,
      "training_step_time": 9.446238040924072
    },
    {
      "epoch": 1.908302307128906e-06,
      "model_forward_time": 1.7459502220153809,
      "step": 76001
    },
    {
      "epoch": 1.908302307128906e-06,
      "step": 76001,
      "training_step_time": 8.375311613082886
    },
    {
      "epoch": 1.9092559814453123e-06,
      "model_forward_time": 3.5641300678253174,
      "step": 76002
    },
    {
      "epoch": 1.9092559814453123e-06,
      "step": 76002,
      "training_step_time": 10.466495990753174
    },
    {
      "epoch": 1.910209655761719e-06,
      "model_forward_time": 1.7446599006652832,
      "step": 76003
    },
    {
      "epoch": 1.910209655761719e-06,
      "step": 76003,
      "training_step_time": 9.502293109893799
    },
    {
      "epoch": 1.911163330078125e-06,
      "model_forward_time": 1.8075499534606934,
      "step": 76004
    },
    {
      "epoch": 1.911163330078125e-06,
      "step": 76004,
      "training_step_time": 9.186521291732788
    },
    {
      "epoch": 1.912117004394531e-06,
      "model_forward_time": 1.7470598220825195,
      "step": 76005
    },
    {
      "epoch": 1.912117004394531e-06,
      "step": 76005,
      "training_step_time": 9.453530311584473
    },
    {
      "epoch": 1.9130706787109373e-06,
      "model_forward_time": 1.746629238128662,
      "step": 76006
    },
    {
      "epoch": 1.9130706787109373e-06,
      "step": 76006,
      "training_step_time": 7.893258810043335
    },
    {
      "epoch": 1.914024353027344e-06,
      "model_forward_time": 1.7524259090423584,
      "step": 76007
    },
    {
      "epoch": 1.914024353027344e-06,
      "step": 76007,
      "training_step_time": 9.022999286651611
    },
    {
      "epoch": 1.91497802734375e-06,
      "model_forward_time": 1.7499735355377197,
      "step": 76008
    },
    {
      "epoch": 1.91497802734375e-06,
      "step": 76008,
      "training_step_time": 8.464581966400146
    },
    {
      "epoch": 1.915931701660156e-06,
      "model_forward_time": 1.7508645057678223,
      "step": 76009
    },
    {
      "epoch": 1.915931701660156e-06,
      "step": 76009,
      "training_step_time": 8.2770414352417
    },
    {
      "epoch": 1.9168853759765623e-06,
      "grad_norm": 0.08144884556531906,
      "learning_rate": 1.4927553088385033e-06,
      "loss": 0.0515,
      "step": 76010
    },
    {
      "epoch": 1.9168853759765623e-06,
      "model_forward_time": 1.7522895336151123,
      "step": 76010
    },
    {
      "dynamics_loss_avg": 0.048505171574652194,
      "epoch": 1.9168853759765623e-06,
      "step": 76010
    },
    {
      "action_loss_avg": 0.000982579024275765,
      "epoch": 1.9168853759765623e-06,
      "step": 76010
    },
    {
      "epoch": 1.9168853759765623e-06,
      "step": 76010,
      "training_step_time": 7.835153102874756
    },
    {
      "epoch": 1.917839050292969e-06,
      "model_forward_time": 2.8966832160949707,
      "step": 76011
    },
    {
      "epoch": 1.917839050292969e-06,
      "step": 76011,
      "training_step_time": 8.656504392623901
    },
    {
      "epoch": 1.918792724609375e-06,
      "model_forward_time": 1.7534925937652588,
      "step": 76012
    },
    {
      "epoch": 1.918792724609375e-06,
      "step": 76012,
      "training_step_time": 7.78526759147644
    },
    {
      "epoch": 1.919746398925781e-06,
      "model_forward_time": 1.7672624588012695,
      "step": 76013
    },
    {
      "epoch": 1.919746398925781e-06,
      "step": 76013,
      "training_step_time": 8.318908214569092
    },
    {
      "epoch": 1.9207000732421873e-06,
      "model_forward_time": 1.7703862190246582,
      "step": 76014
    },
    {
      "epoch": 1.9207000732421873e-06,
      "step": 76014,
      "training_step_time": 7.752771854400635
    },
    {
      "epoch": 1.921653747558594e-06,
      "model_forward_time": 1.760230541229248,
      "step": 76015
    },
    {
      "epoch": 1.921653747558594e-06,
      "step": 76015,
      "training_step_time": 7.828434705734253
    },
    {
      "epoch": 1.922607421875e-06,
      "model_forward_time": 1.7546415328979492,
      "step": 76016
    },
    {
      "epoch": 1.922607421875e-06,
      "step": 76016,
      "training_step_time": 7.791916847229004
    },
    {
      "epoch": 1.923561096191406e-06,
      "model_forward_time": 1.7573566436767578,
      "step": 76017
    },
    {
      "epoch": 1.923561096191406e-06,
      "step": 76017,
      "training_step_time": 8.812841415405273
    },
    {
      "epoch": 1.9245147705078123e-06,
      "model_forward_time": 1.7557299137115479,
      "step": 76018
    },
    {
      "epoch": 1.9245147705078123e-06,
      "step": 76018,
      "training_step_time": 7.970907211303711
    },
    {
      "epoch": 1.925468444824219e-06,
      "model_forward_time": 1.7469491958618164,
      "step": 76019
    },
    {
      "epoch": 1.925468444824219e-06,
      "step": 76019,
      "training_step_time": 7.718013525009155
    },
    {
      "epoch": 1.926422119140625e-06,
      "grad_norm": 0.08640152215957642,
      "learning_rate": 1.49157704010218e-06,
      "loss": 0.0554,
      "step": 76020
    },
    {
      "epoch": 1.926422119140625e-06,
      "model_forward_time": 2.843920946121216,
      "step": 76020
    },
    {
      "dynamics_loss_avg": 0.04411669094115496,
      "epoch": 1.926422119140625e-06,
      "step": 76020
    },
    {
      "action_loss_avg": 0.0008567579585360363,
      "epoch": 1.926422119140625e-06,
      "step": 76020
    },
    {
      "epoch": 1.926422119140625e-06,
      "step": 76020,
      "training_step_time": 7.756794452667236
    },
    {
      "epoch": 1.927375793457031e-06,
      "model_forward_time": 1.7636518478393555,
      "step": 76021
    },
    {
      "epoch": 1.927375793457031e-06,
      "step": 76021,
      "training_step_time": 8.291200399398804
    },
    {
      "epoch": 1.9283294677734373e-06,
      "model_forward_time": 1.7606573104858398,
      "step": 76022
    },
    {
      "epoch": 1.9283294677734373e-06,
      "step": 76022,
      "training_step_time": 8.83700680732727
    },
    {
      "epoch": 1.929283142089844e-06,
      "model_forward_time": 1.7806761264801025,
      "step": 76023
    },
    {
      "epoch": 1.929283142089844e-06,
      "step": 76023,
      "training_step_time": 8.419258832931519
    },
    {
      "epoch": 1.93023681640625e-06,
      "model_forward_time": 1.746978759765625,
      "step": 76024
    },
    {
      "epoch": 1.93023681640625e-06,
      "step": 76024,
      "training_step_time": 7.892339468002319
    },
    {
      "epoch": 1.931190490722656e-06,
      "model_forward_time": 1.7542662620544434,
      "step": 76025
    },
    {
      "epoch": 1.931190490722656e-06,
      "step": 76025,
      "training_step_time": 7.800407886505127
    },
    {
      "epoch": 1.9321441650390623e-06,
      "model_forward_time": 1.752906084060669,
      "step": 76026
    },
    {
      "epoch": 1.9321441650390623e-06,
      "step": 76026,
      "training_step_time": 7.781074047088623
    },
    {
      "epoch": 1.933097839355469e-06,
      "model_forward_time": 1.7491683959960938,
      "step": 76027
    },
    {
      "epoch": 1.933097839355469e-06,
      "step": 76027,
      "training_step_time": 7.326802968978882
    },
    {
      "epoch": 1.934051513671875e-06,
      "model_forward_time": 1.7637791633605957,
      "step": 76028
    },
    {
      "epoch": 1.934051513671875e-06,
      "step": 76028,
      "training_step_time": 8.548793315887451
    },
    {
      "epoch": 1.935005187988281e-06,
      "model_forward_time": 2.815284252166748,
      "step": 76029
    },
    {
      "epoch": 1.935005187988281e-06,
      "step": 76029,
      "training_step_time": 8.314968585968018
    },
    {
      "epoch": 1.9359588623046873e-06,
      "grad_norm": 0.07440022379159927,
      "learning_rate": 1.4903991550417184e-06,
      "loss": 0.0521,
      "step": 76030
    },
    {
      "epoch": 1.9359588623046873e-06,
      "model_forward_time": 1.7629261016845703,
      "step": 76030
    },
    {
      "dynamics_loss_avg": 0.05789593830704689,
      "epoch": 1.9359588623046873e-06,
      "step": 76030
    },
    {
      "action_loss_avg": 0.001953836268512532,
      "epoch": 1.9359588623046873e-06,
      "step": 76030
    },
    {
      "epoch": 1.9359588623046873e-06,
      "step": 76030,
      "training_step_time": 7.731504440307617
    },
    {
      "epoch": 1.936912536621094e-06,
      "model_forward_time": 1.7602674961090088,
      "step": 76031
    },
    {
      "epoch": 1.936912536621094e-06,
      "step": 76031,
      "training_step_time": 7.884252309799194
    },
    {
      "epoch": 1.9378662109375e-06,
      "model_forward_time": 1.854010820388794,
      "step": 76032
    },
    {
      "epoch": 1.9378662109375e-06,
      "step": 76032,
      "training_step_time": 7.828952312469482
    },
    {
      "epoch": 1.938819885253906e-06,
      "model_forward_time": 1.764559030532837,
      "step": 76033
    },
    {
      "epoch": 1.938819885253906e-06,
      "step": 76033,
      "training_step_time": 8.01721978187561
    },
    {
      "epoch": 1.9397735595703123e-06,
      "model_forward_time": 1.744938850402832,
      "step": 76034
    },
    {
      "epoch": 1.9397735595703123e-06,
      "step": 76034,
      "training_step_time": 8.88952088356018
    },
    {
      "epoch": 1.940727233886719e-06,
      "model_forward_time": 1.7636795043945312,
      "step": 76035
    },
    {
      "epoch": 1.940727233886719e-06,
      "step": 76035,
      "training_step_time": 7.814042806625366
    },
    {
      "epoch": 1.941680908203125e-06,
      "model_forward_time": 1.3923256397247314,
      "step": 76036
    },
    {
      "epoch": 1.941680908203125e-06,
      "step": 76036,
      "training_step_time": 7.725600719451904
    },
    {
      "epoch": 1.942634582519531e-06,
      "model_forward_time": 1.7521350383758545,
      "step": 76037
    },
    {
      "epoch": 1.942634582519531e-06,
      "step": 76037,
      "training_step_time": 8.302481651306152
    },
    {
      "epoch": 1.9435882568359373e-06,
      "model_forward_time": 1.7621068954467773,
      "step": 76038
    },
    {
      "epoch": 1.9435882568359373e-06,
      "step": 76038,
      "training_step_time": 8.74106240272522
    },
    {
      "epoch": 1.944541931152344e-06,
      "model_forward_time": 2.9187498092651367,
      "step": 76039
    },
    {
      "epoch": 1.944541931152344e-06,
      "step": 76039,
      "training_step_time": 9.433419704437256
    },
    {
      "epoch": 1.94549560546875e-06,
      "grad_norm": 0.08255300670862198,
      "learning_rate": 1.4892216537859277e-06,
      "loss": 0.0539,
      "step": 76040
    },
    {
      "epoch": 1.94549560546875e-06,
      "model_forward_time": 1.7510483264923096,
      "step": 76040
    },
    {
      "dynamics_loss_avg": 0.049942909926176074,
      "epoch": 1.94549560546875e-06,
      "step": 76040
    },
    {
      "action_loss_avg": 0.0009105675737373531,
      "epoch": 1.94549560546875e-06,
      "step": 76040
    },
    {
      "epoch": 1.94549560546875e-06,
      "step": 76040,
      "training_step_time": 7.854976415634155
    },
    {
      "epoch": 1.946449279785156e-06,
      "model_forward_time": 1.839869737625122,
      "step": 76041
    },
    {
      "epoch": 1.946449279785156e-06,
      "step": 76041,
      "training_step_time": 11.668676853179932
    },
    {
      "epoch": 1.9474029541015623e-06,
      "model_forward_time": 1.7553040981292725,
      "step": 76042
    },
    {
      "epoch": 1.9474029541015623e-06,
      "step": 76042,
      "training_step_time": 8.757023811340332
    },
    {
      "epoch": 1.948356628417969e-06,
      "model_forward_time": 1.7614951133728027,
      "step": 76043
    },
    {
      "epoch": 1.948356628417969e-06,
      "step": 76043,
      "training_step_time": 7.812803030014038
    },
    {
      "epoch": 1.949310302734375e-06,
      "model_forward_time": 1.763155221939087,
      "step": 76044
    },
    {
      "epoch": 1.949310302734375e-06,
      "step": 76044,
      "training_step_time": 7.880466938018799
    },
    {
      "epoch": 1.950263977050781e-06,
      "model_forward_time": 1.7571375370025635,
      "step": 76045
    },
    {
      "epoch": 1.950263977050781e-06,
      "step": 76045,
      "training_step_time": 7.771611928939819
    },
    {
      "epoch": 1.9512176513671873e-06,
      "model_forward_time": 1.7583165168762207,
      "step": 76046
    },
    {
      "epoch": 1.9512176513671873e-06,
      "step": 76046,
      "training_step_time": 7.73211145401001
    },
    {
      "epoch": 1.952171325683594e-06,
      "model_forward_time": 1.7640526294708252,
      "step": 76047
    },
    {
      "epoch": 1.952171325683594e-06,
      "step": 76047,
      "training_step_time": 8.774802684783936
    },
    {
      "epoch": 1.953125e-06,
      "model_forward_time": 2.840394973754883,
      "step": 76048
    },
    {
      "epoch": 1.953125e-06,
      "step": 76048,
      "training_step_time": 8.743627786636353
    },
    {
      "epoch": 1.954078674316406e-06,
      "model_forward_time": 1.7569565773010254,
      "step": 76049
    },
    {
      "epoch": 1.954078674316406e-06,
      "step": 76049,
      "training_step_time": 8.692614316940308
    },
    {
      "epoch": 1.9550323486328123e-06,
      "grad_norm": 0.16564440727233887,
      "learning_rate": 1.4880445364635777e-06,
      "loss": 0.0528,
      "step": 76050
    },
    {
      "epoch": 1.9550323486328123e-06,
      "model_forward_time": 1.7594611644744873,
      "step": 76050
    },
    {
      "dynamics_loss_avg": 0.03894822876900435,
      "epoch": 1.9550323486328123e-06,
      "step": 76050
    },
    {
      "action_loss_avg": 0.0013184106413973496,
      "epoch": 1.9550323486328123e-06,
      "step": 76050
    },
    {
      "epoch": 1.9550323486328123e-06,
      "step": 76050,
      "training_step_time": 8.67911958694458
    },
    {
      "epoch": 1.955986022949219e-06,
      "model_forward_time": 1.7562742233276367,
      "step": 76051
    },
    {
      "epoch": 1.955986022949219e-06,
      "step": 76051,
      "training_step_time": 8.673295259475708
    },
    {
      "epoch": 1.956939697265625e-06,
      "model_forward_time": 1.7509818077087402,
      "step": 76052
    },
    {
      "epoch": 1.956939697265625e-06,
      "step": 76052,
      "training_step_time": 8.356083393096924
    },
    {
      "epoch": 1.957893371582031e-06,
      "model_forward_time": 1.7595810890197754,
      "step": 76053
    },
    {
      "epoch": 1.957893371582031e-06,
      "step": 76053,
      "training_step_time": 7.801722049713135
    },
    {
      "epoch": 1.9588470458984373e-06,
      "model_forward_time": 1.747446060180664,
      "step": 76054
    },
    {
      "epoch": 1.9588470458984373e-06,
      "step": 76054,
      "training_step_time": 7.799377202987671
    },
    {
      "epoch": 1.959800720214844e-06,
      "model_forward_time": 1.7511231899261475,
      "step": 76055
    },
    {
      "epoch": 1.959800720214844e-06,
      "step": 76055,
      "training_step_time": 7.61960506439209
    },
    {
      "epoch": 1.96075439453125e-06,
      "model_forward_time": 1.7479138374328613,
      "step": 76056
    },
    {
      "epoch": 1.96075439453125e-06,
      "step": 76056,
      "training_step_time": 7.763231039047241
    },
    {
      "epoch": 1.961708068847656e-06,
      "model_forward_time": 2.819659948348999,
      "step": 76057
    },
    {
      "epoch": 1.961708068847656e-06,
      "step": 76057,
      "training_step_time": 7.720484256744385
    },
    {
      "epoch": 1.9626617431640623e-06,
      "model_forward_time": 1.7463414669036865,
      "step": 76058
    },
    {
      "epoch": 1.9626617431640623e-06,
      "step": 76058,
      "training_step_time": 9.172747135162354
    },
    {
      "epoch": 1.963615417480469e-06,
      "model_forward_time": 1.753978967666626,
      "step": 76059
    },
    {
      "epoch": 1.963615417480469e-06,
      "step": 76059,
      "training_step_time": 8.54787826538086
    },
    {
      "epoch": 1.964569091796875e-06,
      "grad_norm": 0.0996025949716568,
      "learning_rate": 1.4868678032033973e-06,
      "loss": 0.0533,
      "step": 76060
    },
    {
      "epoch": 1.964569091796875e-06,
      "model_forward_time": 1.767169713973999,
      "step": 76060
    },
    {
      "dynamics_loss_avg": 0.05750864390283823,
      "epoch": 1.964569091796875e-06,
      "step": 76060
    },
    {
      "action_loss_avg": 0.001653520905529149,
      "epoch": 1.964569091796875e-06,
      "step": 76060
    },
    {
      "epoch": 1.964569091796875e-06,
      "step": 76060,
      "training_step_time": 7.944923639297485
    },
    {
      "epoch": 1.965522766113281e-06,
      "model_forward_time": 1.750291109085083,
      "step": 76061
    },
    {
      "epoch": 1.965522766113281e-06,
      "step": 76061,
      "training_step_time": 7.802732944488525
    },
    {
      "epoch": 1.9664764404296873e-06,
      "model_forward_time": 1.7625417709350586,
      "step": 76062
    },
    {
      "epoch": 1.9664764404296873e-06,
      "step": 76062,
      "training_step_time": 8.802034139633179
    },
    {
      "epoch": 1.967430114746094e-06,
      "model_forward_time": 1.748340368270874,
      "step": 76063
    },
    {
      "epoch": 1.967430114746094e-06,
      "step": 76063,
      "training_step_time": 7.743981599807739
    },
    {
      "epoch": 1.9683837890625e-06,
      "model_forward_time": 1.7532789707183838,
      "step": 76064
    },
    {
      "epoch": 1.9683837890625e-06,
      "step": 76064,
      "training_step_time": 7.503976345062256
    },
    {
      "epoch": 1.969337463378906e-06,
      "model_forward_time": 1.762577772140503,
      "step": 76065
    },
    {
      "epoch": 1.969337463378906e-06,
      "step": 76065,
      "training_step_time": 7.743643045425415
    },
    {
      "epoch": 1.9702911376953123e-06,
      "model_forward_time": 2.841956615447998,
      "step": 76066
    },
    {
      "epoch": 1.9702911376953123e-06,
      "step": 76066,
      "training_step_time": 7.830644607543945
    },
    {
      "epoch": 1.971244812011719e-06,
      "model_forward_time": 1.7520091533660889,
      "step": 76067
    },
    {
      "epoch": 1.971244812011719e-06,
      "step": 76067,
      "training_step_time": 8.462992429733276
    },
    {
      "epoch": 1.972198486328125e-06,
      "model_forward_time": 1.7429049015045166,
      "step": 76068
    },
    {
      "epoch": 1.972198486328125e-06,
      "step": 76068,
      "training_step_time": 7.844161510467529
    },
    {
      "epoch": 1.973152160644531e-06,
      "model_forward_time": 1.7887752056121826,
      "step": 76069
    },
    {
      "epoch": 1.973152160644531e-06,
      "step": 76069,
      "training_step_time": 7.908070087432861
    },
    {
      "epoch": 1.9741058349609373e-06,
      "grad_norm": 0.0897301733493805,
      "learning_rate": 1.4856914541340738e-06,
      "loss": 0.0537,
      "step": 76070
    },
    {
      "epoch": 1.9741058349609373e-06,
      "model_forward_time": 1.7334935665130615,
      "step": 76070
    },
    {
      "dynamics_loss_avg": 0.03436277378350496,
      "epoch": 1.9741058349609373e-06,
      "step": 76070
    },
    {
      "action_loss_avg": 0.0009446355223190039,
      "epoch": 1.9741058349609373e-06,
      "step": 76070
    },
    {
      "epoch": 1.9741058349609373e-06,
      "step": 76070,
      "training_step_time": 7.672125816345215
    },
    {
      "epoch": 1.975059509277344e-06,
      "model_forward_time": 1.7436432838439941,
      "step": 76071
    },
    {
      "epoch": 1.975059509277344e-06,
      "step": 76071,
      "training_step_time": 7.701066017150879
    },
    {
      "epoch": 1.97601318359375e-06,
      "model_forward_time": 1.7515912055969238,
      "step": 76072
    },
    {
      "epoch": 1.97601318359375e-06,
      "step": 76072,
      "training_step_time": 7.724154233932495
    },
    {
      "epoch": 1.976966857910156e-06,
      "model_forward_time": 1.7539165019989014,
      "step": 76073
    },
    {
      "epoch": 1.976966857910156e-06,
      "step": 76073,
      "training_step_time": 7.538494825363159
    },
    {
      "epoch": 1.9779205322265624e-06,
      "model_forward_time": 1.7458913326263428,
      "step": 76074
    },
    {
      "epoch": 1.9779205322265624e-06,
      "step": 76074,
      "training_step_time": 7.668658971786499
    },
    {
      "epoch": 1.978874206542969e-06,
      "model_forward_time": 2.806414842605591,
      "step": 76075
    },
    {
      "epoch": 1.978874206542969e-06,
      "step": 76075,
      "training_step_time": 8.36128044128418
    },
    {
      "epoch": 1.979827880859375e-06,
      "model_forward_time": 1.7728996276855469,
      "step": 76076
    },
    {
      "epoch": 1.979827880859375e-06,
      "step": 76076,
      "training_step_time": 11.421289443969727
    },
    {
      "epoch": 1.980781555175781e-06,
      "model_forward_time": 1.7471246719360352,
      "step": 76077
    },
    {
      "epoch": 1.980781555175781e-06,
      "step": 76077,
      "training_step_time": 6.766489028930664
    },
    {
      "epoch": 1.9817352294921874e-06,
      "model_forward_time": 1.7472341060638428,
      "step": 76078
    },
    {
      "epoch": 1.9817352294921874e-06,
      "step": 76078,
      "training_step_time": 7.8029797077178955
    },
    {
      "epoch": 1.982688903808594e-06,
      "model_forward_time": 1.7506399154663086,
      "step": 76079
    },
    {
      "epoch": 1.982688903808594e-06,
      "step": 76079,
      "training_step_time": 7.680762767791748
    },
    {
      "epoch": 1.983642578125e-06,
      "grad_norm": 0.0823831707239151,
      "learning_rate": 1.4845154893842473e-06,
      "loss": 0.0548,
      "step": 76080
    },
    {
      "epoch": 1.983642578125e-06,
      "model_forward_time": 1.7461938858032227,
      "step": 76080
    },
    {
      "dynamics_loss_avg": 0.03584080608561635,
      "epoch": 1.983642578125e-06,
      "step": 76080
    },
    {
      "action_loss_avg": 0.0010451765556354076,
      "epoch": 1.983642578125e-06,
      "step": 76080
    },
    {
      "epoch": 1.983642578125e-06,
      "step": 76080,
      "training_step_time": 7.6578192710876465
    },
    {
      "epoch": 1.9845962524414062e-06,
      "model_forward_time": 1.760169506072998,
      "step": 76081
    },
    {
      "epoch": 1.9845962524414062e-06,
      "step": 76081,
      "training_step_time": 6.730136156082153
    },
    {
      "epoch": 1.9855499267578124e-06,
      "model_forward_time": 1.764758586883545,
      "step": 76082
    },
    {
      "epoch": 1.9855499267578124e-06,
      "step": 76082,
      "training_step_time": 6.800849676132202
    },
    {
      "epoch": 1.986503601074219e-06,
      "model_forward_time": 1.7721681594848633,
      "step": 76083
    },
    {
      "epoch": 1.986503601074219e-06,
      "step": 76083,
      "training_step_time": 6.772937536239624
    },
    {
      "epoch": 1.987457275390625e-06,
      "model_forward_time": 1.7616562843322754,
      "step": 76084
    },
    {
      "epoch": 1.987457275390625e-06,
      "step": 76084,
      "training_step_time": 6.786146640777588
    },
    {
      "epoch": 1.9884109497070312e-06,
      "model_forward_time": 1.7586779594421387,
      "step": 76085
    },
    {
      "epoch": 1.9884109497070312e-06,
      "step": 76085,
      "training_step_time": 6.807325839996338
    },
    {
      "epoch": 1.9893646240234374e-06,
      "model_forward_time": 2.8741021156311035,
      "step": 76086
    },
    {
      "epoch": 1.9893646240234374e-06,
      "step": 76086,
      "training_step_time": 8.742205381393433
    },
    {
      "epoch": 1.990318298339844e-06,
      "model_forward_time": 1.7643916606903076,
      "step": 76087
    },
    {
      "epoch": 1.990318298339844e-06,
      "step": 76087,
      "training_step_time": 7.117833614349365
    },
    {
      "epoch": 1.99127197265625e-06,
      "model_forward_time": 1.750126600265503,
      "step": 76088
    },
    {
      "epoch": 1.99127197265625e-06,
      "step": 76088,
      "training_step_time": 7.70209527015686
    },
    {
      "epoch": 1.9922256469726562e-06,
      "model_forward_time": 1.7686214447021484,
      "step": 76089
    },
    {
      "epoch": 1.9922256469726562e-06,
      "step": 76089,
      "training_step_time": 7.620576858520508
    },
    {
      "epoch": 1.9931793212890624e-06,
      "grad_norm": 0.09172072261571884,
      "learning_rate": 1.4833399090825235e-06,
      "loss": 0.0538,
      "step": 76090
    },
    {
      "epoch": 1.9931793212890624e-06,
      "model_forward_time": 1.7692561149597168,
      "step": 76090
    },
    {
      "dynamics_loss_avg": 0.06851621624082327,
      "epoch": 1.9931793212890624e-06,
      "step": 76090
    },
    {
      "action_loss_avg": 0.0022398166969651357,
      "epoch": 1.9931793212890624e-06,
      "step": 76090
    },
    {
      "epoch": 1.9931793212890624e-06,
      "step": 76090,
      "training_step_time": 6.752685546875
    },
    {
      "epoch": 1.994132995605469e-06,
      "model_forward_time": 1.756929874420166,
      "step": 76091
    },
    {
      "epoch": 1.994132995605469e-06,
      "step": 76091,
      "training_step_time": 6.549036264419556
    },
    {
      "epoch": 1.995086669921875e-06,
      "model_forward_time": 1.7620232105255127,
      "step": 76092
    },
    {
      "epoch": 1.995086669921875e-06,
      "step": 76092,
      "training_step_time": 6.705261468887329
    },
    {
      "epoch": 1.9960403442382812e-06,
      "model_forward_time": 1.7681593894958496,
      "step": 76093
    },
    {
      "epoch": 1.9960403442382812e-06,
      "step": 76093,
      "training_step_time": 6.764057874679565
    },
    {
      "epoch": 1.9969940185546874e-06,
      "model_forward_time": 1.7586150169372559,
      "step": 76094
    },
    {
      "epoch": 1.9969940185546874e-06,
      "step": 76094,
      "training_step_time": 6.7761452198028564
    },
    {
      "epoch": 1.997947692871094e-06,
      "model_forward_time": 2.867459774017334,
      "step": 76095
    },
    {
      "epoch": 1.997947692871094e-06,
      "step": 76095,
      "training_step_time": 7.841789722442627
    },
    {
      "epoch": 1.9989013671875e-06,
      "model_forward_time": 1.7705714702606201,
      "step": 76096
    },
    {
      "epoch": 1.9989013671875e-06,
      "step": 76096,
      "training_step_time": 7.629787921905518
    },
    {
      "epoch": 1.9998550415039062e-06,
      "model_forward_time": 1.7584586143493652,
      "step": 76097
    },
    {
      "epoch": 1.9998550415039062e-06,
      "step": 76097,
      "training_step_time": 8.42552924156189
    },
    {
      "epoch": 2.0008087158203124e-06,
      "model_forward_time": 1.7518930435180664,
      "step": 76098
    },
    {
      "epoch": 2.0008087158203124e-06,
      "step": 76098,
      "training_step_time": 7.7213294506073
    },
    {
      "epoch": 2.001762390136719e-06,
      "model_forward_time": 1.7432339191436768,
      "step": 76099
    },
    {
      "epoch": 2.001762390136719e-06,
      "step": 76099,
      "training_step_time": 6.787668228149414
    },
    {
      "epoch": 2.002716064453125e-06,
      "grad_norm": 0.09606065601110458,
      "learning_rate": 1.4821647133574563e-06,
      "loss": 0.0542,
      "step": 76100
    },
    {
      "epoch": 2.002716064453125e-06,
      "model_forward_time": 1.7423968315124512,
      "step": 76100
    },
    {
      "dynamics_loss_avg": 0.05923498952761293,
      "epoch": 2.002716064453125e-06,
      "step": 76100
    },
    {
      "action_loss_avg": 0.0021319382125511766,
      "epoch": 2.002716064453125e-06,
      "step": 76100
    },
    {
      "epoch": 2.002716064453125e-06,
      "step": 76100,
      "training_step_time": 6.798123359680176
    },
    {
      "epoch": 2.0036697387695312e-06,
      "model_forward_time": 1.7494523525238037,
      "step": 76101
    },
    {
      "epoch": 2.0036697387695312e-06,
      "step": 76101,
      "training_step_time": 6.78492283821106
    },
    {
      "epoch": 2.0046234130859374e-06,
      "model_forward_time": 1.7481584548950195,
      "step": 76102
    },
    {
      "epoch": 2.0046234130859374e-06,
      "step": 76102,
      "training_step_time": 6.743043899536133
    },
    {
      "epoch": 2.005577087402344e-06,
      "model_forward_time": 1.7527542114257812,
      "step": 76103
    },
    {
      "epoch": 2.005577087402344e-06,
      "step": 76103,
      "training_step_time": 6.774668216705322
    },
    {
      "epoch": 2.00653076171875e-06,
      "model_forward_time": 2.9127047061920166,
      "step": 76104
    },
    {
      "epoch": 2.00653076171875e-06,
      "step": 76104,
      "training_step_time": 9.359837293624878
    },
    {
      "epoch": 2.0074844360351562e-06,
      "model_forward_time": 1.7465643882751465,
      "step": 76105
    },
    {
      "epoch": 2.0074844360351562e-06,
      "step": 76105,
      "training_step_time": 7.573042631149292
    },
    {
      "epoch": 2.0084381103515624e-06,
      "model_forward_time": 1.3810431957244873,
      "step": 76106
    },
    {
      "epoch": 2.0084381103515624e-06,
      "step": 76106,
      "training_step_time": 7.362044095993042
    },
    {
      "epoch": 2.009391784667969e-06,
      "model_forward_time": 1.7419705390930176,
      "step": 76107
    },
    {
      "epoch": 2.009391784667969e-06,
      "step": 76107,
      "training_step_time": 7.677572727203369
    },
    {
      "epoch": 2.010345458984375e-06,
      "model_forward_time": 1.7438695430755615,
      "step": 76108
    },
    {
      "epoch": 2.010345458984375e-06,
      "step": 76108,
      "training_step_time": 6.782293081283569
    },
    {
      "epoch": 2.0112991333007812e-06,
      "model_forward_time": 1.7524454593658447,
      "step": 76109
    },
    {
      "epoch": 2.0112991333007812e-06,
      "step": 76109,
      "training_step_time": 6.730347156524658
    },
    {
      "epoch": 2.0122528076171874e-06,
      "grad_norm": 0.11486092954874039,
      "learning_rate": 1.4809899023375707e-06,
      "loss": 0.0545,
      "step": 76110
    },
    {
      "epoch": 2.0122528076171874e-06,
      "model_forward_time": 1.743753433227539,
      "step": 76110
    },
    {
      "dynamics_loss_avg": 0.04167138859629631,
      "epoch": 2.0122528076171874e-06,
      "step": 76110
    },
    {
      "action_loss_avg": 0.0017685504135442898,
      "epoch": 2.0122528076171874e-06,
      "step": 76110
    },
    {
      "epoch": 2.0122528076171874e-06,
      "step": 76110,
      "training_step_time": 6.818101406097412
    },
    {
      "epoch": 2.013206481933594e-06,
      "model_forward_time": 1.750025987625122,
      "step": 76111
    },
    {
      "epoch": 2.013206481933594e-06,
      "step": 76111,
      "training_step_time": 6.7620179653167725
    },
    {
      "epoch": 2.01416015625e-06,
      "model_forward_time": 1.7535016536712646,
      "step": 76112
    },
    {
      "epoch": 2.01416015625e-06,
      "step": 76112,
      "training_step_time": 6.765156030654907
    },
    {
      "epoch": 2.0151138305664062e-06,
      "model_forward_time": 2.7982876300811768,
      "step": 76113
    },
    {
      "epoch": 2.0151138305664062e-06,
      "step": 76113,
      "training_step_time": 11.426512718200684
    },
    {
      "epoch": 2.0160675048828124e-06,
      "model_forward_time": 1.7881672382354736,
      "step": 76114
    },
    {
      "epoch": 2.0160675048828124e-06,
      "step": 76114,
      "training_step_time": 7.778292179107666
    },
    {
      "epoch": 2.017021179199219e-06,
      "model_forward_time": 1.758493423461914,
      "step": 76115
    },
    {
      "epoch": 2.017021179199219e-06,
      "step": 76115,
      "training_step_time": 6.747965097427368
    },
    {
      "epoch": 2.017974853515625e-06,
      "model_forward_time": 1.7606611251831055,
      "step": 76116
    },
    {
      "epoch": 2.017974853515625e-06,
      "step": 76116,
      "training_step_time": 9.042855501174927
    },
    {
      "epoch": 2.0189285278320312e-06,
      "model_forward_time": 1.7579131126403809,
      "step": 76117
    },
    {
      "epoch": 2.0189285278320312e-06,
      "step": 76117,
      "training_step_time": 6.739378213882446
    },
    {
      "epoch": 2.0198822021484374e-06,
      "model_forward_time": 1.753143310546875,
      "step": 76118
    },
    {
      "epoch": 2.0198822021484374e-06,
      "step": 76118,
      "training_step_time": 6.737287521362305
    },
    {
      "epoch": 2.020835876464844e-06,
      "model_forward_time": 1.7545530796051025,
      "step": 76119
    },
    {
      "epoch": 2.020835876464844e-06,
      "step": 76119,
      "training_step_time": 6.75327205657959
    },
    {
      "epoch": 2.02178955078125e-06,
      "grad_norm": 0.11396725475788116,
      "learning_rate": 1.4798154761513366e-06,
      "loss": 0.0539,
      "step": 76120
    },
    {
      "epoch": 2.02178955078125e-06,
      "model_forward_time": 1.3954088687896729,
      "step": 76120
    },
    {
      "dynamics_loss_avg": 0.05169652979820967,
      "epoch": 2.02178955078125e-06,
      "step": 76120
    },
    {
      "action_loss_avg": 0.001740888273343444,
      "epoch": 2.02178955078125e-06,
      "step": 76120
    },
    {
      "epoch": 2.02178955078125e-06,
      "step": 76120,
      "training_step_time": 6.734551906585693
    },
    {
      "epoch": 2.0227432250976562e-06,
      "model_forward_time": 1.7524068355560303,
      "step": 76121
    },
    {
      "epoch": 2.0227432250976562e-06,
      "step": 76121,
      "training_step_time": 6.746334552764893
    },
    {
      "epoch": 2.0236968994140624e-06,
      "model_forward_time": 1.7582614421844482,
      "step": 76122
    },
    {
      "epoch": 2.0236968994140624e-06,
      "step": 76122,
      "training_step_time": 9.78705358505249
    },
    {
      "epoch": 2.024650573730469e-06,
      "model_forward_time": 2.9100303649902344,
      "step": 76123
    },
    {
      "epoch": 2.024650573730469e-06,
      "step": 76123,
      "training_step_time": 7.75636887550354
    },
    {
      "epoch": 2.025604248046875e-06,
      "model_forward_time": 1.7636828422546387,
      "step": 76124
    },
    {
      "epoch": 2.025604248046875e-06,
      "step": 76124,
      "training_step_time": 6.825446367263794
    },
    {
      "epoch": 2.0265579223632812e-06,
      "model_forward_time": 1.7663242816925049,
      "step": 76125
    },
    {
      "epoch": 2.0265579223632812e-06,
      "step": 76125,
      "training_step_time": 7.73000693321228
    },
    {
      "epoch": 2.0275115966796874e-06,
      "model_forward_time": 1.7553579807281494,
      "step": 76126
    },
    {
      "epoch": 2.0275115966796874e-06,
      "step": 76126,
      "training_step_time": 7.709015607833862
    },
    {
      "epoch": 2.0284652709960935e-06,
      "model_forward_time": 1.751774787902832,
      "step": 76127
    },
    {
      "epoch": 2.0284652709960935e-06,
      "step": 76127,
      "training_step_time": 6.826303958892822
    },
    {
      "epoch": 2.0294189453125e-06,
      "model_forward_time": 1.7719395160675049,
      "step": 76128
    },
    {
      "epoch": 2.0294189453125e-06,
      "step": 76128,
      "training_step_time": 6.8158087730407715
    },
    {
      "epoch": 2.0303726196289063e-06,
      "model_forward_time": 1.7489643096923828,
      "step": 76129
    },
    {
      "epoch": 2.0303726196289063e-06,
      "step": 76129,
      "training_step_time": 6.777327060699463
    },
    {
      "epoch": 2.0313262939453124e-06,
      "grad_norm": 0.07648614048957825,
      "learning_rate": 1.4786414349271916e-06,
      "loss": 0.0543,
      "step": 76130
    },
    {
      "epoch": 2.0313262939453124e-06,
      "model_forward_time": 1.7538409233093262,
      "step": 76130
    },
    {
      "dynamics_loss_avg": 0.043422481324523685,
      "epoch": 2.0313262939453124e-06,
      "step": 76130
    },
    {
      "action_loss_avg": 0.0011709835460351314,
      "epoch": 2.0313262939453124e-06,
      "step": 76130
    },
    {
      "epoch": 2.0313262939453124e-06,
      "step": 76130,
      "training_step_time": 6.79086709022522
    },
    {
      "epoch": 2.0322799682617185e-06,
      "model_forward_time": 1.7614490985870361,
      "step": 76131
    },
    {
      "epoch": 2.0322799682617185e-06,
      "step": 76131,
      "training_step_time": 8.70345687866211
    },
    {
      "epoch": 2.033233642578125e-06,
      "model_forward_time": 2.9284377098083496,
      "step": 76132
    },
    {
      "epoch": 2.033233642578125e-06,
      "step": 76132,
      "training_step_time": 7.831358194351196
    },
    {
      "epoch": 2.0341873168945313e-06,
      "model_forward_time": 1.7723660469055176,
      "step": 76133
    },
    {
      "epoch": 2.0341873168945313e-06,
      "step": 76133,
      "training_step_time": 7.758943319320679
    },
    {
      "epoch": 2.0351409912109374e-06,
      "model_forward_time": 1.7689156532287598,
      "step": 76134
    },
    {
      "epoch": 2.0351409912109374e-06,
      "step": 76134,
      "training_step_time": 7.7216105461120605
    },
    {
      "epoch": 2.0360946655273435e-06,
      "model_forward_time": 1.7674951553344727,
      "step": 76135
    },
    {
      "epoch": 2.0360946655273435e-06,
      "step": 76135,
      "training_step_time": 7.312610387802124
    },
    {
      "epoch": 2.03704833984375e-06,
      "model_forward_time": 1.7567417621612549,
      "step": 76136
    },
    {
      "epoch": 2.03704833984375e-06,
      "step": 76136,
      "training_step_time": 6.796738862991333
    },
    {
      "epoch": 2.0380020141601563e-06,
      "model_forward_time": 1.7555663585662842,
      "step": 76137
    },
    {
      "epoch": 2.0380020141601563e-06,
      "step": 76137,
      "training_step_time": 6.7873475551605225
    },
    {
      "epoch": 2.0389556884765624e-06,
      "model_forward_time": 1.7597770690917969,
      "step": 76138
    },
    {
      "epoch": 2.0389556884765624e-06,
      "step": 76138,
      "training_step_time": 6.77461838722229
    },
    {
      "epoch": 2.0399093627929686e-06,
      "model_forward_time": 1.7523188591003418,
      "step": 76139
    },
    {
      "epoch": 2.0399093627929686e-06,
      "step": 76139,
      "training_step_time": 6.783335447311401
    },
    {
      "epoch": 2.040863037109375e-06,
      "grad_norm": 0.08343934267759323,
      "learning_rate": 1.4774677787935227e-06,
      "loss": 0.0555,
      "step": 76140
    },
    {
      "epoch": 2.040863037109375e-06,
      "model_forward_time": 1.7590909004211426,
      "step": 76140
    },
    {
      "dynamics_loss_avg": 0.05167240314185619,
      "epoch": 2.040863037109375e-06,
      "step": 76140
    },
    {
      "action_loss_avg": 0.0014456127188168465,
      "epoch": 2.040863037109375e-06,
      "step": 76140
    },
    {
      "epoch": 2.040863037109375e-06,
      "step": 76140,
      "training_step_time": 8.784717559814453
    },
    {
      "epoch": 2.0418167114257813e-06,
      "model_forward_time": 2.9277877807617188,
      "step": 76141
    },
    {
      "epoch": 2.0418167114257813e-06,
      "step": 76141,
      "training_step_time": 10.100844383239746
    },
    {
      "epoch": 2.0427703857421874e-06,
      "model_forward_time": 1.7540984153747559,
      "step": 76142
    },
    {
      "epoch": 2.0427703857421874e-06,
      "step": 76142,
      "training_step_time": 8.451586723327637
    },
    {
      "epoch": 2.0437240600585936e-06,
      "model_forward_time": 1.7603051662445068,
      "step": 76143
    },
    {
      "epoch": 2.0437240600585936e-06,
      "step": 76143,
      "training_step_time": 7.674375534057617
    },
    {
      "epoch": 2.044677734375e-06,
      "model_forward_time": 1.3965258598327637,
      "step": 76144
    },
    {
      "epoch": 2.044677734375e-06,
      "step": 76144,
      "training_step_time": 7.7112298011779785
    },
    {
      "epoch": 2.0456314086914063e-06,
      "model_forward_time": 1.3855760097503662,
      "step": 76145
    },
    {
      "epoch": 2.0456314086914063e-06,
      "step": 76145,
      "training_step_time": 6.774248361587524
    },
    {
      "epoch": 2.0465850830078124e-06,
      "model_forward_time": 1.7556867599487305,
      "step": 76146
    },
    {
      "epoch": 2.0465850830078124e-06,
      "step": 76146,
      "training_step_time": 6.719427585601807
    },
    {
      "epoch": 2.0475387573242186e-06,
      "model_forward_time": 1.7606687545776367,
      "step": 76147
    },
    {
      "epoch": 2.0475387573242186e-06,
      "step": 76147,
      "training_step_time": 6.757633686065674
    },
    {
      "epoch": 2.048492431640625e-06,
      "model_forward_time": 1.7719755172729492,
      "step": 76148
    },
    {
      "epoch": 2.048492431640625e-06,
      "step": 76148,
      "training_step_time": 6.7820000648498535
    },
    {
      "epoch": 2.0494461059570313e-06,
      "model_forward_time": 1.764869213104248,
      "step": 76149
    },
    {
      "epoch": 2.0494461059570313e-06,
      "step": 76149,
      "training_step_time": 9.362464666366577
    },
    {
      "epoch": 2.0503997802734374e-06,
      "grad_norm": 0.12727104127407074,
      "learning_rate": 1.4762945078786839e-06,
      "loss": 0.0549,
      "step": 76150
    },
    {
      "epoch": 2.0503997802734374e-06,
      "model_forward_time": 1.7707562446594238,
      "step": 76150
    },
    {
      "dynamics_loss_avg": 0.06719514150172472,
      "epoch": 2.0503997802734374e-06,
      "step": 76150
    },
    {
      "action_loss_avg": 0.0012414501863531769,
      "epoch": 2.0503997802734374e-06,
      "step": 76150
    },
    {
      "epoch": 2.0503997802734374e-06,
      "step": 76150,
      "training_step_time": 9.445543766021729
    },
    {
      "epoch": 2.0513534545898436e-06,
      "model_forward_time": 2.845273494720459,
      "step": 76151
    },
    {
      "epoch": 2.0513534545898436e-06,
      "step": 76151,
      "training_step_time": 7.777908563613892
    },
    {
      "epoch": 2.05230712890625e-06,
      "model_forward_time": 1.75286865234375,
      "step": 76152
    },
    {
      "epoch": 2.05230712890625e-06,
      "step": 76152,
      "training_step_time": 7.332163572311401
    },
    {
      "epoch": 2.0532608032226563e-06,
      "model_forward_time": 1.7436573505401611,
      "step": 76153
    },
    {
      "epoch": 2.0532608032226563e-06,
      "step": 76153,
      "training_step_time": 7.756500482559204
    },
    {
      "epoch": 2.0542144775390624e-06,
      "model_forward_time": 1.750720739364624,
      "step": 76154
    },
    {
      "epoch": 2.0542144775390624e-06,
      "step": 76154,
      "training_step_time": 7.380888223648071
    },
    {
      "epoch": 2.0551681518554686e-06,
      "model_forward_time": 1.7448465824127197,
      "step": 76155
    },
    {
      "epoch": 2.0551681518554686e-06,
      "step": 76155,
      "training_step_time": 6.709638833999634
    },
    {
      "epoch": 2.056121826171875e-06,
      "model_forward_time": 1.7422630786895752,
      "step": 76156
    },
    {
      "epoch": 2.056121826171875e-06,
      "step": 76156,
      "training_step_time": 6.644399404525757
    },
    {
      "epoch": 2.0570755004882813e-06,
      "model_forward_time": 1.7480390071868896,
      "step": 76157
    },
    {
      "epoch": 2.0570755004882813e-06,
      "step": 76157,
      "training_step_time": 6.730785608291626
    },
    {
      "epoch": 2.0580291748046874e-06,
      "model_forward_time": 1.7415361404418945,
      "step": 76158
    },
    {
      "epoch": 2.0580291748046874e-06,
      "step": 76158,
      "training_step_time": 7.636686325073242
    },
    {
      "epoch": 2.0589828491210936e-06,
      "model_forward_time": 1.7509267330169678,
      "step": 76159
    },
    {
      "epoch": 2.0589828491210936e-06,
      "step": 76159,
      "training_step_time": 10.157503128051758
    },
    {
      "epoch": 2.0599365234375e-06,
      "grad_norm": 0.07281886786222458,
      "learning_rate": 1.4751216223109777e-06,
      "loss": 0.0555,
      "step": 76160
    },
    {
      "epoch": 2.0599365234375e-06,
      "model_forward_time": 2.8514490127563477,
      "step": 76160
    },
    {
      "dynamics_loss_avg": 0.056693424470722675,
      "epoch": 2.0599365234375e-06,
      "step": 76160
    },
    {
      "action_loss_avg": 0.0012538707698695362,
      "epoch": 2.0599365234375e-06,
      "step": 76160
    },
    {
      "epoch": 2.0599365234375e-06,
      "step": 76160,
      "training_step_time": 7.739433526992798
    },
    {
      "epoch": 2.0608901977539063e-06,
      "model_forward_time": 1.744917392730713,
      "step": 76161
    },
    {
      "epoch": 2.0608901977539063e-06,
      "step": 76161,
      "training_step_time": 7.451558351516724
    },
    {
      "epoch": 2.0618438720703124e-06,
      "model_forward_time": 1.7416057586669922,
      "step": 76162
    },
    {
      "epoch": 2.0618438720703124e-06,
      "step": 76162,
      "training_step_time": 8.331419944763184
    },
    {
      "epoch": 2.0627975463867186e-06,
      "model_forward_time": 1.7432670593261719,
      "step": 76163
    },
    {
      "epoch": 2.0627975463867186e-06,
      "step": 76163,
      "training_step_time": 7.619210243225098
    },
    {
      "epoch": 2.063751220703125e-06,
      "model_forward_time": 1.7335681915283203,
      "step": 76164
    },
    {
      "epoch": 2.063751220703125e-06,
      "step": 76164,
      "training_step_time": 6.640482425689697
    },
    {
      "epoch": 2.0647048950195313e-06,
      "model_forward_time": 1.7325336933135986,
      "step": 76165
    },
    {
      "epoch": 2.0647048950195313e-06,
      "step": 76165,
      "training_step_time": 6.731621026992798
    },
    {
      "epoch": 2.0656585693359374e-06,
      "model_forward_time": 1.734206199645996,
      "step": 76166
    },
    {
      "epoch": 2.0656585693359374e-06,
      "step": 76166,
      "training_step_time": 6.715410947799683
    },
    {
      "epoch": 2.0666122436523436e-06,
      "model_forward_time": 1.7368035316467285,
      "step": 76167
    },
    {
      "epoch": 2.0666122436523436e-06,
      "step": 76167,
      "training_step_time": 8.587826490402222
    },
    {
      "epoch": 2.06756591796875e-06,
      "model_forward_time": 1.7278990745544434,
      "step": 76168
    },
    {
      "epoch": 2.06756591796875e-06,
      "step": 76168,
      "training_step_time": 9.871110200881958
    },
    {
      "epoch": 2.0685195922851563e-06,
      "model_forward_time": 2.8569016456604004,
      "step": 76169
    },
    {
      "epoch": 2.0685195922851563e-06,
      "step": 76169,
      "training_step_time": 9.575732231140137
    },
    {
      "epoch": 2.0694732666015624e-06,
      "grad_norm": 0.08231459558010101,
      "learning_rate": 1.4739491222186714e-06,
      "loss": 0.0564,
      "step": 76170
    },
    {
      "epoch": 2.0694732666015624e-06,
      "model_forward_time": 1.7312202453613281,
      "step": 76170
    },
    {
      "dynamics_loss_avg": 0.04576635770499706,
      "epoch": 2.0694732666015624e-06,
      "step": 76170
    },
    {
      "action_loss_avg": 0.0009968841361114756,
      "epoch": 2.0694732666015624e-06,
      "step": 76170
    },
    {
      "epoch": 2.0694732666015624e-06,
      "step": 76170,
      "training_step_time": 7.643801689147949
    },
    {
      "epoch": 2.0704269409179686e-06,
      "model_forward_time": 1.7329397201538086,
      "step": 76171
    },
    {
      "epoch": 2.0704269409179686e-06,
      "step": 76171,
      "training_step_time": 7.706706762313843
    },
    {
      "epoch": 2.071380615234375e-06,
      "model_forward_time": 1.7227671146392822,
      "step": 76172
    },
    {
      "epoch": 2.071380615234375e-06,
      "step": 76172,
      "training_step_time": 7.646050214767456
    },
    {
      "epoch": 2.0723342895507813e-06,
      "model_forward_time": 1.7406644821166992,
      "step": 76173
    },
    {
      "epoch": 2.0723342895507813e-06,
      "step": 76173,
      "training_step_time": 6.811449289321899
    },
    {
      "epoch": 2.0732879638671874e-06,
      "model_forward_time": 1.733865737915039,
      "step": 76174
    },
    {
      "epoch": 2.0732879638671874e-06,
      "step": 76174,
      "training_step_time": 6.8604652881622314
    },
    {
      "epoch": 2.0742416381835936e-06,
      "model_forward_time": 1.7314050197601318,
      "step": 76175
    },
    {
      "epoch": 2.0742416381835936e-06,
      "step": 76175,
      "training_step_time": 6.78363561630249
    },
    {
      "epoch": 2.0751953125e-06,
      "model_forward_time": 1.7367520332336426,
      "step": 76176
    },
    {
      "epoch": 2.0751953125e-06,
      "step": 76176,
      "training_step_time": 7.6851019859313965
    },
    {
      "epoch": 2.0761489868164063e-06,
      "model_forward_time": 1.7343332767486572,
      "step": 76177
    },
    {
      "epoch": 2.0761489868164063e-06,
      "step": 76177,
      "training_step_time": 8.65975284576416
    },
    {
      "epoch": 2.0771026611328124e-06,
      "model_forward_time": 2.8623361587524414,
      "step": 76178
    },
    {
      "epoch": 2.0771026611328124e-06,
      "step": 76178,
      "training_step_time": 9.206021547317505
    },
    {
      "epoch": 2.0780563354492186e-06,
      "model_forward_time": 1.7288804054260254,
      "step": 76179
    },
    {
      "epoch": 2.0780563354492186e-06,
      "step": 76179,
      "training_step_time": 7.804387092590332
    },
    {
      "epoch": 2.079010009765625e-06,
      "grad_norm": 0.08042871952056885,
      "learning_rate": 1.472777007729988e-06,
      "loss": 0.0565,
      "step": 76180
    },
    {
      "epoch": 2.079010009765625e-06,
      "model_forward_time": 1.7427611351013184,
      "step": 76180
    },
    {
      "dynamics_loss_avg": 0.04864615835249424,
      "epoch": 2.079010009765625e-06,
      "step": 76180
    },
    {
      "action_loss_avg": 0.0010070225776871665,
      "epoch": 2.079010009765625e-06,
      "step": 76180
    },
    {
      "epoch": 2.079010009765625e-06,
      "step": 76180,
      "training_step_time": 8.200286626815796
    },
    {
      "epoch": 2.0799636840820313e-06,
      "model_forward_time": 1.7257697582244873,
      "step": 76181
    },
    {
      "epoch": 2.0799636840820313e-06,
      "step": 76181,
      "training_step_time": 7.821073770523071
    },
    {
      "epoch": 2.0809173583984374e-06,
      "model_forward_time": 1.7341814041137695,
      "step": 76182
    },
    {
      "epoch": 2.0809173583984374e-06,
      "step": 76182,
      "training_step_time": 6.8367273807525635
    },
    {
      "epoch": 2.0818710327148436e-06,
      "model_forward_time": 1.7369799613952637,
      "step": 76183
    },
    {
      "epoch": 2.0818710327148436e-06,
      "step": 76183,
      "training_step_time": 6.855069875717163
    },
    {
      "epoch": 2.08282470703125e-06,
      "model_forward_time": 1.7364637851715088,
      "step": 76184
    },
    {
      "epoch": 2.08282470703125e-06,
      "step": 76184,
      "training_step_time": 6.8302130699157715
    },
    {
      "epoch": 2.0837783813476563e-06,
      "model_forward_time": 1.735762357711792,
      "step": 76185
    },
    {
      "epoch": 2.0837783813476563e-06,
      "step": 76185,
      "training_step_time": 7.695924997329712
    },
    {
      "epoch": 2.0847320556640624e-06,
      "model_forward_time": 1.7306489944458008,
      "step": 76186
    },
    {
      "epoch": 2.0847320556640624e-06,
      "step": 76186,
      "training_step_time": 9.420037031173706
    },
    {
      "epoch": 2.0856857299804686e-06,
      "model_forward_time": 2.838880777359009,
      "step": 76187
    },
    {
      "epoch": 2.0856857299804686e-06,
      "step": 76187,
      "training_step_time": 10.923587322235107
    },
    {
      "epoch": 2.086639404296875e-06,
      "model_forward_time": 1.7294111251831055,
      "step": 76188
    },
    {
      "epoch": 2.086639404296875e-06,
      "step": 76188,
      "training_step_time": 7.800089120864868
    },
    {
      "epoch": 2.0875930786132813e-06,
      "model_forward_time": 1.7401645183563232,
      "step": 76189
    },
    {
      "epoch": 2.0875930786132813e-06,
      "step": 76189,
      "training_step_time": 8.18952202796936
    },
    {
      "epoch": 2.0885467529296874e-06,
      "grad_norm": 0.09701598435640335,
      "learning_rate": 1.4716052789731095e-06,
      "loss": 0.0576,
      "step": 76190
    },
    {
      "epoch": 2.0885467529296874e-06,
      "model_forward_time": 1.7296717166900635,
      "step": 76190
    },
    {
      "dynamics_loss_avg": 0.05092001873999834,
      "epoch": 2.0885467529296874e-06,
      "step": 76190
    },
    {
      "action_loss_avg": 0.001234552990354132,
      "epoch": 2.0885467529296874e-06,
      "step": 76190
    },
    {
      "epoch": 2.0885467529296874e-06,
      "step": 76190,
      "training_step_time": 7.818845748901367
    },
    {
      "epoch": 2.0895004272460936e-06,
      "model_forward_time": 1.7315001487731934,
      "step": 76191
    },
    {
      "epoch": 2.0895004272460936e-06,
      "step": 76191,
      "training_step_time": 7.257982492446899
    },
    {
      "epoch": 2.0904541015625e-06,
      "model_forward_time": 1.741997241973877,
      "step": 76192
    },
    {
      "epoch": 2.0904541015625e-06,
      "step": 76192,
      "training_step_time": 6.840596914291382
    },
    {
      "epoch": 2.0914077758789063e-06,
      "model_forward_time": 1.7344472408294678,
      "step": 76193
    },
    {
      "epoch": 2.0914077758789063e-06,
      "step": 76193,
      "training_step_time": 6.9235968589782715
    },
    {
      "epoch": 2.0923614501953125e-06,
      "model_forward_time": 1.72940993309021,
      "step": 76194
    },
    {
      "epoch": 2.0923614501953125e-06,
      "step": 76194,
      "training_step_time": 7.686683177947998
    },
    {
      "epoch": 2.0933151245117186e-06,
      "model_forward_time": 1.7340407371520996,
      "step": 76195
    },
    {
      "epoch": 2.0933151245117186e-06,
      "step": 76195,
      "training_step_time": 8.67891788482666
    },
    {
      "epoch": 2.094268798828125e-06,
      "model_forward_time": 1.7400765419006348,
      "step": 76196
    },
    {
      "epoch": 2.094268798828125e-06,
      "step": 76196,
      "training_step_time": 9.382689952850342
    },
    {
      "epoch": 2.0952224731445313e-06,
      "model_forward_time": 1.8086843490600586,
      "step": 76197
    },
    {
      "epoch": 2.0952224731445313e-06,
      "step": 76197,
      "training_step_time": 8.823609352111816
    },
    {
      "epoch": 2.0961761474609375e-06,
      "model_forward_time": 1.7400460243225098,
      "step": 76198
    },
    {
      "epoch": 2.0961761474609375e-06,
      "step": 76198,
      "training_step_time": 7.575676202774048
    },
    {
      "epoch": 2.0971298217773436e-06,
      "model_forward_time": 1.738591194152832,
      "step": 76199
    },
    {
      "epoch": 2.0971298217773436e-06,
      "step": 76199,
      "training_step_time": 7.810546875
    },
    {
      "epoch": 2.09808349609375e-06,
      "grad_norm": 0.12138501554727554,
      "learning_rate": 1.470433936076171e-06,
      "loss": 0.0571,
      "step": 76200
    },
    {
      "epoch": 2.09808349609375e-06,
      "model_forward_time": 1.7293097972869873,
      "step": 76200
    },
    {
      "dynamics_loss_avg": 0.06110919611528516,
      "epoch": 2.09808349609375e-06,
      "step": 76200
    },
    {
      "action_loss_avg": 0.001594004180515185,
      "epoch": 2.09808349609375e-06,
      "step": 76200
    },
    {
      "epoch": 2.09808349609375e-06,
      "step": 76200,
      "training_step_time": 7.194394826889038
    },
    {
      "epoch": 2.0990371704101563e-06,
      "model_forward_time": 1.7386062145233154,
      "step": 76201
    },
    {
      "epoch": 2.0990371704101563e-06,
      "step": 76201,
      "training_step_time": 6.805906772613525
    },
    {
      "epoch": 2.0999908447265625e-06,
      "model_forward_time": 1.7399342060089111,
      "step": 76202
    },
    {
      "epoch": 2.0999908447265625e-06,
      "step": 76202,
      "training_step_time": 6.7955641746521
    },
    {
      "epoch": 2.1009445190429686e-06,
      "model_forward_time": 1.7342453002929688,
      "step": 76203
    },
    {
      "epoch": 2.1009445190429686e-06,
      "step": 76203,
      "training_step_time": 7.564861059188843
    },
    {
      "epoch": 2.101898193359375e-06,
      "model_forward_time": 1.7305564880371094,
      "step": 76204
    },
    {
      "epoch": 2.101898193359375e-06,
      "step": 76204,
      "training_step_time": 8.227969884872437
    },
    {
      "epoch": 2.1028518676757813e-06,
      "model_forward_time": 1.7344131469726562,
      "step": 76205
    },
    {
      "epoch": 2.1028518676757813e-06,
      "step": 76205,
      "training_step_time": 11.053192853927612
    },
    {
      "epoch": 2.1038055419921875e-06,
      "model_forward_time": 2.8608295917510986,
      "step": 76206
    },
    {
      "epoch": 2.1038055419921875e-06,
      "step": 76206,
      "training_step_time": 10.469920635223389
    },
    {
      "epoch": 2.1047592163085936e-06,
      "model_forward_time": 1.7373089790344238,
      "step": 76207
    },
    {
      "epoch": 2.1047592163085936e-06,
      "step": 76207,
      "training_step_time": 7.47141170501709
    },
    {
      "epoch": 2.105712890625e-06,
      "model_forward_time": 1.7343614101409912,
      "step": 76208
    },
    {
      "epoch": 2.105712890625e-06,
      "step": 76208,
      "training_step_time": 7.723229885101318
    },
    {
      "epoch": 2.1066665649414063e-06,
      "model_forward_time": 1.7351176738739014,
      "step": 76209
    },
    {
      "epoch": 2.1066665649414063e-06,
      "step": 76209,
      "training_step_time": 7.890793085098267
    },
    {
      "epoch": 2.1076202392578125e-06,
      "grad_norm": 0.07305040955543518,
      "learning_rate": 1.469262979167272e-06,
      "loss": 0.0603,
      "step": 76210
    },
    {
      "epoch": 2.1076202392578125e-06,
      "model_forward_time": 1.7364659309387207,
      "step": 76210
    },
    {
      "dynamics_loss_avg": 0.05503323674201965,
      "epoch": 2.1076202392578125e-06,
      "step": 76210
    },
    {
      "action_loss_avg": 0.0010497079900233075,
      "epoch": 2.1076202392578125e-06,
      "step": 76210
    },
    {
      "epoch": 2.1076202392578125e-06,
      "step": 76210,
      "training_step_time": 6.737586498260498
    },
    {
      "epoch": 2.1085739135742186e-06,
      "model_forward_time": 1.7395339012145996,
      "step": 76211
    },
    {
      "epoch": 2.1085739135742186e-06,
      "step": 76211,
      "training_step_time": 6.769297122955322
    },
    {
      "epoch": 2.109527587890625e-06,
      "model_forward_time": 1.7366220951080322,
      "step": 76212
    },
    {
      "epoch": 2.109527587890625e-06,
      "step": 76212,
      "training_step_time": 7.620718240737915
    },
    {
      "epoch": 2.1104812622070313e-06,
      "model_forward_time": 1.7420740127563477,
      "step": 76213
    },
    {
      "epoch": 2.1104812622070313e-06,
      "step": 76213,
      "training_step_time": 7.704704999923706
    },
    {
      "epoch": 2.1114349365234375e-06,
      "model_forward_time": 1.7575321197509766,
      "step": 76214
    },
    {
      "epoch": 2.1114349365234375e-06,
      "step": 76214,
      "training_step_time": 8.740970134735107
    },
    {
      "epoch": 2.1123886108398436e-06,
      "model_forward_time": 2.403524160385132,
      "step": 76215
    },
    {
      "epoch": 2.1123886108398436e-06,
      "step": 76215,
      "training_step_time": 7.792565584182739
    },
    {
      "epoch": 2.11334228515625e-06,
      "model_forward_time": 1.8851120471954346,
      "step": 76216
    },
    {
      "epoch": 2.11334228515625e-06,
      "step": 76216,
      "training_step_time": 7.6459949016571045
    },
    {
      "epoch": 2.1142959594726563e-06,
      "model_forward_time": 1.7718253135681152,
      "step": 76217
    },
    {
      "epoch": 2.1142959594726563e-06,
      "step": 76217,
      "training_step_time": 7.663599967956543
    },
    {
      "epoch": 2.1152496337890625e-06,
      "model_forward_time": 1.7632246017456055,
      "step": 76218
    },
    {
      "epoch": 2.1152496337890625e-06,
      "step": 76218,
      "training_step_time": 7.6706671714782715
    },
    {
      "epoch": 2.1162033081054686e-06,
      "model_forward_time": 1.7709674835205078,
      "step": 76219
    },
    {
      "epoch": 2.1162033081054686e-06,
      "step": 76219,
      "training_step_time": 7.121611595153809
    },
    {
      "epoch": 2.117156982421875e-06,
      "grad_norm": 0.1014997661113739,
      "learning_rate": 1.4680924083744641e-06,
      "loss": 0.058,
      "step": 76220
    },
    {
      "epoch": 2.117156982421875e-06,
      "model_forward_time": 1.7694237232208252,
      "step": 76220
    },
    {
      "dynamics_loss_avg": 0.05196390990167856,
      "epoch": 2.117156982421875e-06,
      "step": 76220
    },
    {
      "action_loss_avg": 0.0018045453674858437,
      "epoch": 2.117156982421875e-06,
      "step": 76220
    },
    {
      "epoch": 2.117156982421875e-06,
      "step": 76220,
      "training_step_time": 6.748698949813843
    },
    {
      "epoch": 2.1181106567382813e-06,
      "model_forward_time": 1.7721765041351318,
      "step": 76221
    },
    {
      "epoch": 2.1181106567382813e-06,
      "step": 76221,
      "training_step_time": 7.612119913101196
    },
    {
      "epoch": 2.1190643310546875e-06,
      "model_forward_time": 1.7775464057922363,
      "step": 76222
    },
    {
      "epoch": 2.1190643310546875e-06,
      "step": 76222,
      "training_step_time": 8.600464344024658
    },
    {
      "epoch": 2.1200180053710936e-06,
      "model_forward_time": 1.7713286876678467,
      "step": 76223
    },
    {
      "epoch": 2.1200180053710936e-06,
      "step": 76223,
      "training_step_time": 10.518150568008423
    },
    {
      "epoch": 2.1209716796875e-06,
      "model_forward_time": 2.397899627685547,
      "step": 76224
    },
    {
      "epoch": 2.1209716796875e-06,
      "step": 76224,
      "training_step_time": 9.238826751708984
    },
    {
      "epoch": 2.1219253540039063e-06,
      "model_forward_time": 1.8717961311340332,
      "step": 76225
    },
    {
      "epoch": 2.1219253540039063e-06,
      "step": 76225,
      "training_step_time": 8.240546941757202
    },
    {
      "epoch": 2.1228790283203125e-06,
      "model_forward_time": 1.7718472480773926,
      "step": 76226
    },
    {
      "epoch": 2.1228790283203125e-06,
      "step": 76226,
      "training_step_time": 7.745777130126953
    },
    {
      "epoch": 2.1238327026367186e-06,
      "model_forward_time": 1.7699594497680664,
      "step": 76227
    },
    {
      "epoch": 2.1238327026367186e-06,
      "step": 76227,
      "training_step_time": 7.753116130828857
    },
    {
      "epoch": 2.124786376953125e-06,
      "model_forward_time": 1.7721450328826904,
      "step": 76228
    },
    {
      "epoch": 2.124786376953125e-06,
      "step": 76228,
      "training_step_time": 7.194902658462524
    },
    {
      "epoch": 2.1257400512695313e-06,
      "model_forward_time": 1.7757625579833984,
      "step": 76229
    },
    {
      "epoch": 2.1257400512695313e-06,
      "step": 76229,
      "training_step_time": 6.769673585891724
    },
    {
      "epoch": 2.1266937255859375e-06,
      "grad_norm": 0.09077318012714386,
      "learning_rate": 1.4669222238257603e-06,
      "loss": 0.06,
      "step": 76230
    },
    {
      "epoch": 2.1266937255859375e-06,
      "model_forward_time": 1.7722618579864502,
      "step": 76230
    },
    {
      "dynamics_loss_avg": 0.056779488921165466,
      "epoch": 2.1266937255859375e-06,
      "step": 76230
    },
    {
      "action_loss_avg": 0.0015349483757745475,
      "epoch": 2.1266937255859375e-06,
      "step": 76230
    },
    {
      "epoch": 2.1266937255859375e-06,
      "step": 76230,
      "training_step_time": 7.56967568397522
    },
    {
      "epoch": 2.1276473999023436e-06,
      "model_forward_time": 1.7744331359863281,
      "step": 76231
    },
    {
      "epoch": 2.1276473999023436e-06,
      "step": 76231,
      "training_step_time": 7.981773853302002
    },
    {
      "epoch": 2.12860107421875e-06,
      "model_forward_time": 1.7773425579071045,
      "step": 76232
    },
    {
      "epoch": 2.12860107421875e-06,
      "step": 76232,
      "training_step_time": 9.569682121276855
    },
    {
      "epoch": 2.1295547485351563e-06,
      "model_forward_time": 2.382739543914795,
      "step": 76233
    },
    {
      "epoch": 2.1295547485351563e-06,
      "step": 76233,
      "training_step_time": 9.329959392547607
    },
    {
      "epoch": 2.1305084228515625e-06,
      "model_forward_time": 1.8601150512695312,
      "step": 76234
    },
    {
      "epoch": 2.1305084228515625e-06,
      "step": 76234,
      "training_step_time": 8.408800840377808
    },
    {
      "epoch": 2.1314620971679686e-06,
      "model_forward_time": 1.788844347000122,
      "step": 76235
    },
    {
      "epoch": 2.1314620971679686e-06,
      "step": 76235,
      "training_step_time": 7.66083836555481
    },
    {
      "epoch": 2.132415771484375e-06,
      "model_forward_time": 1.407785177230835,
      "step": 76236
    },
    {
      "epoch": 2.132415771484375e-06,
      "step": 76236,
      "training_step_time": 7.7739622592926025
    },
    {
      "epoch": 2.1333694458007813e-06,
      "model_forward_time": 1.7914495468139648,
      "step": 76237
    },
    {
      "epoch": 2.1333694458007813e-06,
      "step": 76237,
      "training_step_time": 7.157614707946777
    },
    {
      "epoch": 2.1343231201171875e-06,
      "model_forward_time": 1.771493911743164,
      "step": 76238
    },
    {
      "epoch": 2.1343231201171875e-06,
      "step": 76238,
      "training_step_time": 6.7858288288116455
    },
    {
      "epoch": 2.1352767944335936e-06,
      "model_forward_time": 1.77195405960083,
      "step": 76239
    },
    {
      "epoch": 2.1352767944335936e-06,
      "step": 76239,
      "training_step_time": 7.618466377258301
    },
    {
      "epoch": 2.13623046875e-06,
      "grad_norm": 0.21674181520938873,
      "learning_rate": 1.4657524256491302e-06,
      "loss": 0.0594,
      "step": 76240
    },
    {
      "epoch": 2.13623046875e-06,
      "model_forward_time": 1.7762553691864014,
      "step": 76240
    },
    {
      "dynamics_loss_avg": 0.06983838621526957,
      "epoch": 2.13623046875e-06,
      "step": 76240
    },
    {
      "action_loss_avg": 0.0016797719086753204,
      "epoch": 2.13623046875e-06,
      "step": 76240
    },
    {
      "epoch": 2.13623046875e-06,
      "step": 76240,
      "training_step_time": 7.262404918670654
    },
    {
      "epoch": 2.1371841430664063e-06,
      "model_forward_time": 1.7642228603363037,
      "step": 76241
    },
    {
      "epoch": 2.1371841430664063e-06,
      "step": 76241,
      "training_step_time": 7.746479749679565
    },
    {
      "epoch": 2.1381378173828125e-06,
      "model_forward_time": 2.3985071182250977,
      "step": 76242
    },
    {
      "epoch": 2.1381378173828125e-06,
      "step": 76242,
      "training_step_time": 10.168840169906616
    },
    {
      "epoch": 2.1390914916992186e-06,
      "model_forward_time": 1.79496169090271,
      "step": 76243
    },
    {
      "epoch": 2.1390914916992186e-06,
      "step": 76243,
      "training_step_time": 7.6308698654174805
    },
    {
      "epoch": 2.140045166015625e-06,
      "model_forward_time": 1.7757225036621094,
      "step": 76244
    },
    {
      "epoch": 2.140045166015625e-06,
      "step": 76244,
      "training_step_time": 9.039332389831543
    },
    {
      "epoch": 2.1409988403320314e-06,
      "model_forward_time": 1.7698290348052979,
      "step": 76245
    },
    {
      "epoch": 2.1409988403320314e-06,
      "step": 76245,
      "training_step_time": 7.564752101898193
    },
    {
      "epoch": 2.1419525146484375e-06,
      "model_forward_time": 1.7673945426940918,
      "step": 76246
    },
    {
      "epoch": 2.1419525146484375e-06,
      "step": 76246,
      "training_step_time": 8.026138067245483
    },
    {
      "epoch": 2.1429061889648436e-06,
      "model_forward_time": 1.7711098194122314,
      "step": 76247
    },
    {
      "epoch": 2.1429061889648436e-06,
      "step": 76247,
      "training_step_time": 6.724788427352905
    },
    {
      "epoch": 2.14385986328125e-06,
      "model_forward_time": 1.7656328678131104,
      "step": 76248
    },
    {
      "epoch": 2.14385986328125e-06,
      "step": 76248,
      "training_step_time": 8.018235445022583
    },
    {
      "epoch": 2.1448135375976564e-06,
      "model_forward_time": 1.7700293064117432,
      "step": 76249
    },
    {
      "epoch": 2.1448135375976564e-06,
      "step": 76249,
      "training_step_time": 7.234831094741821
    },
    {
      "epoch": 2.1457672119140625e-06,
      "grad_norm": 0.08967757970094681,
      "learning_rate": 1.464583013972502e-06,
      "loss": 0.0583,
      "step": 76250
    },
    {
      "epoch": 2.1457672119140625e-06,
      "model_forward_time": 1.762340784072876,
      "step": 76250
    },
    {
      "dynamics_loss_avg": 0.050334295630455016,
      "epoch": 2.1457672119140625e-06,
      "step": 76250
    },
    {
      "action_loss_avg": 0.0009103780437726527,
      "epoch": 2.1457672119140625e-06,
      "step": 76250
    },
    {
      "epoch": 2.1457672119140625e-06,
      "step": 76250,
      "training_step_time": 8.412448644638062
    },
    {
      "epoch": 2.1467208862304686e-06,
      "model_forward_time": 2.4711642265319824,
      "step": 76251
    },
    {
      "epoch": 2.1467208862304686e-06,
      "step": 76251,
      "training_step_time": 10.784178972244263
    },
    {
      "epoch": 2.147674560546875e-06,
      "model_forward_time": 1.7654285430908203,
      "step": 76252
    },
    {
      "epoch": 2.147674560546875e-06,
      "step": 76252,
      "training_step_time": 8.796771049499512
    },
    {
      "epoch": 2.1486282348632814e-06,
      "model_forward_time": 1.8744864463806152,
      "step": 76253
    },
    {
      "epoch": 2.1486282348632814e-06,
      "step": 76253,
      "training_step_time": 7.624524354934692
    },
    {
      "epoch": 2.1495819091796875e-06,
      "model_forward_time": 1.7679409980773926,
      "step": 76254
    },
    {
      "epoch": 2.1495819091796875e-06,
      "step": 76254,
      "training_step_time": 8.534518957138062
    },
    {
      "epoch": 2.1505355834960936e-06,
      "model_forward_time": 1.7631337642669678,
      "step": 76255
    },
    {
      "epoch": 2.1505355834960936e-06,
      "step": 76255,
      "training_step_time": 7.854236364364624
    },
    {
      "epoch": 2.1514892578125e-06,
      "model_forward_time": 1.76812744140625,
      "step": 76256
    },
    {
      "epoch": 2.1514892578125e-06,
      "step": 76256,
      "training_step_time": 7.290470123291016
    },
    {
      "epoch": 2.1524429321289064e-06,
      "model_forward_time": 1.7685651779174805,
      "step": 76257
    },
    {
      "epoch": 2.1524429321289064e-06,
      "step": 76257,
      "training_step_time": 7.5606348514556885
    },
    {
      "epoch": 2.1533966064453125e-06,
      "model_forward_time": 1.7764015197753906,
      "step": 76258
    },
    {
      "epoch": 2.1533966064453125e-06,
      "step": 76258,
      "training_step_time": 7.540817022323608
    },
    {
      "epoch": 2.1543502807617187e-06,
      "model_forward_time": 1.7682898044586182,
      "step": 76259
    },
    {
      "epoch": 2.1543502807617187e-06,
      "step": 76259,
      "training_step_time": 10.021681308746338
    },
    {
      "epoch": 2.155303955078125e-06,
      "grad_norm": 0.09023867547512054,
      "learning_rate": 1.4634139889237593e-06,
      "loss": 0.0572,
      "step": 76260
    },
    {
      "epoch": 2.155303955078125e-06,
      "model_forward_time": 2.3921759128570557,
      "step": 76260
    },
    {
      "dynamics_loss_avg": 0.06476421467959881,
      "epoch": 2.155303955078125e-06,
      "step": 76260
    },
    {
      "action_loss_avg": 0.001377805179799907,
      "epoch": 2.155303955078125e-06,
      "step": 76260
    },
    {
      "epoch": 2.155303955078125e-06,
      "step": 76260,
      "training_step_time": 8.20417070388794
    },
    {
      "epoch": 2.1562576293945314e-06,
      "model_forward_time": 1.7722530364990234,
      "step": 76261
    },
    {
      "epoch": 2.1562576293945314e-06,
      "step": 76261,
      "training_step_time": 7.708385705947876
    },
    {
      "epoch": 2.1572113037109375e-06,
      "model_forward_time": 1.8666729927062988,
      "step": 76262
    },
    {
      "epoch": 2.1572113037109375e-06,
      "step": 76262,
      "training_step_time": 8.31415581703186
    },
    {
      "epoch": 2.1581649780273437e-06,
      "model_forward_time": 1.771827220916748,
      "step": 76263
    },
    {
      "epoch": 2.1581649780273437e-06,
      "step": 76263,
      "training_step_time": 7.5589680671691895
    },
    {
      "epoch": 2.15911865234375e-06,
      "model_forward_time": 1.769191026687622,
      "step": 76264
    },
    {
      "epoch": 2.15911865234375e-06,
      "step": 76264,
      "training_step_time": 7.599179267883301
    },
    {
      "epoch": 2.1600723266601564e-06,
      "model_forward_time": 1.7640526294708252,
      "step": 76265
    },
    {
      "epoch": 2.1600723266601564e-06,
      "step": 76265,
      "training_step_time": 7.393214225769043
    },
    {
      "epoch": 2.1610260009765625e-06,
      "model_forward_time": 1.7670786380767822,
      "step": 76266
    },
    {
      "epoch": 2.1610260009765625e-06,
      "step": 76266,
      "training_step_time": 7.663274765014648
    },
    {
      "epoch": 2.1619796752929687e-06,
      "model_forward_time": 1.7818663120269775,
      "step": 76267
    },
    {
      "epoch": 2.1619796752929687e-06,
      "step": 76267,
      "training_step_time": 7.747203826904297
    },
    {
      "epoch": 2.162933349609375e-06,
      "model_forward_time": 1.7720634937286377,
      "step": 76268
    },
    {
      "epoch": 2.162933349609375e-06,
      "step": 76268,
      "training_step_time": 7.819346189498901
    },
    {
      "epoch": 2.1638870239257814e-06,
      "model_forward_time": 2.4079504013061523,
      "step": 76269
    },
    {
      "epoch": 2.1638870239257814e-06,
      "step": 76269,
      "training_step_time": 10.928642272949219
    },
    {
      "epoch": 2.1648406982421875e-06,
      "grad_norm": 0.10862971842288971,
      "learning_rate": 1.462245350630745e-06,
      "loss": 0.0556,
      "step": 76270
    },
    {
      "epoch": 2.1648406982421875e-06,
      "model_forward_time": 1.7717230319976807,
      "step": 76270
    },
    {
      "dynamics_loss_avg": 0.05609486615285277,
      "epoch": 2.1648406982421875e-06,
      "step": 76270
    },
    {
      "action_loss_avg": 0.0011786391783971339,
      "epoch": 2.1648406982421875e-06,
      "step": 76270
    },
    {
      "epoch": 2.1648406982421875e-06,
      "step": 76270,
      "training_step_time": 9.424419641494751
    },
    {
      "epoch": 2.1657943725585937e-06,
      "model_forward_time": 1.811934471130371,
      "step": 76271
    },
    {
      "epoch": 2.1657943725585937e-06,
      "step": 76271,
      "training_step_time": 7.861817836761475
    },
    {
      "epoch": 2.166748046875e-06,
      "model_forward_time": 1.7757086753845215,
      "step": 76272
    },
    {
      "epoch": 2.166748046875e-06,
      "step": 76272,
      "training_step_time": 7.765952825546265
    },
    {
      "epoch": 2.1677017211914064e-06,
      "model_forward_time": 1.7724263668060303,
      "step": 76273
    },
    {
      "epoch": 2.1677017211914064e-06,
      "step": 76273,
      "training_step_time": 7.763758659362793
    },
    {
      "epoch": 2.1686553955078125e-06,
      "model_forward_time": 1.7737228870391846,
      "step": 76274
    },
    {
      "epoch": 2.1686553955078125e-06,
      "step": 76274,
      "training_step_time": 7.419609308242798
    },
    {
      "epoch": 2.1696090698242187e-06,
      "model_forward_time": 1.7697010040283203,
      "step": 76275
    },
    {
      "epoch": 2.1696090698242187e-06,
      "step": 76275,
      "training_step_time": 7.6569225788116455
    },
    {
      "epoch": 2.170562744140625e-06,
      "model_forward_time": 1.7683277130126953,
      "step": 76276
    },
    {
      "epoch": 2.170562744140625e-06,
      "step": 76276,
      "training_step_time": 7.803913593292236
    },
    {
      "epoch": 2.1715164184570314e-06,
      "model_forward_time": 1.7711784839630127,
      "step": 76277
    },
    {
      "epoch": 2.1715164184570314e-06,
      "step": 76277,
      "training_step_time": 7.887507915496826
    },
    {
      "epoch": 2.1724700927734375e-06,
      "model_forward_time": 2.3745219707489014,
      "step": 76278
    },
    {
      "epoch": 2.1724700927734375e-06,
      "step": 76278,
      "training_step_time": 7.880459547042847
    },
    {
      "epoch": 2.1734237670898437e-06,
      "model_forward_time": 1.7650816440582275,
      "step": 76279
    },
    {
      "epoch": 2.1734237670898437e-06,
      "step": 76279,
      "training_step_time": 9.978129386901855
    },
    {
      "epoch": 2.17437744140625e-06,
      "grad_norm": 0.13249267637729645,
      "learning_rate": 1.4610770992212619e-06,
      "loss": 0.0557,
      "step": 76280
    },
    {
      "epoch": 2.17437744140625e-06,
      "model_forward_time": 1.7826528549194336,
      "step": 76280
    },
    {
      "dynamics_loss_avg": 0.04460132271051407,
      "epoch": 2.17437744140625e-06,
      "step": 76280
    },
    {
      "action_loss_avg": 0.001347757552866824,
      "epoch": 2.17437744140625e-06,
      "step": 76280
    },
    {
      "epoch": 2.17437744140625e-06,
      "step": 76280,
      "training_step_time": 7.867152214050293
    },
    {
      "epoch": 2.1753311157226564e-06,
      "model_forward_time": 1.7624931335449219,
      "step": 76281
    },
    {
      "epoch": 2.1753311157226564e-06,
      "step": 76281,
      "training_step_time": 7.655589818954468
    },
    {
      "epoch": 2.1762847900390625e-06,
      "model_forward_time": 1.7731475830078125,
      "step": 76282
    },
    {
      "epoch": 2.1762847900390625e-06,
      "step": 76282,
      "training_step_time": 7.847963333129883
    },
    {
      "epoch": 2.1772384643554687e-06,
      "model_forward_time": 1.7922093868255615,
      "step": 76283
    },
    {
      "epoch": 2.1772384643554687e-06,
      "step": 76283,
      "training_step_time": 7.825607061386108
    },
    {
      "epoch": 2.178192138671875e-06,
      "model_forward_time": 1.7690043449401855,
      "step": 76284
    },
    {
      "epoch": 2.178192138671875e-06,
      "step": 76284,
      "training_step_time": 7.459467649459839
    },
    {
      "epoch": 2.1791458129882814e-06,
      "model_forward_time": 1.766772985458374,
      "step": 76285
    },
    {
      "epoch": 2.1791458129882814e-06,
      "step": 76285,
      "training_step_time": 7.546886920928955
    },
    {
      "epoch": 2.1800994873046875e-06,
      "model_forward_time": 1.7677249908447266,
      "step": 76286
    },
    {
      "epoch": 2.1800994873046875e-06,
      "step": 76286,
      "training_step_time": 7.63280987739563
    },
    {
      "epoch": 2.1810531616210937e-06,
      "model_forward_time": 1.7708373069763184,
      "step": 76287
    },
    {
      "epoch": 2.1810531616210937e-06,
      "step": 76287,
      "training_step_time": 10.35104250907898
    },
    {
      "epoch": 2.1820068359375e-06,
      "model_forward_time": 2.3851027488708496,
      "step": 76288
    },
    {
      "epoch": 2.1820068359375e-06,
      "step": 76288,
      "training_step_time": 7.65495228767395
    },
    {
      "epoch": 2.1829605102539064e-06,
      "model_forward_time": 1.772562026977539,
      "step": 76289
    },
    {
      "epoch": 2.1829605102539064e-06,
      "step": 76289,
      "training_step_time": 9.153693199157715
    },
    {
      "epoch": 2.1839141845703125e-06,
      "grad_norm": 0.10237224400043488,
      "learning_rate": 1.4599092348230653e-06,
      "loss": 0.0554,
      "step": 76290
    },
    {
      "epoch": 2.1839141845703125e-06,
      "model_forward_time": 1.8744895458221436,
      "step": 76290
    },
    {
      "dynamics_loss_avg": 0.07183095775544643,
      "epoch": 2.1839141845703125e-06,
      "step": 76290
    },
    {
      "action_loss_avg": 0.0022281033336184917,
      "epoch": 2.1839141845703125e-06,
      "step": 76290
    },
    {
      "epoch": 2.1839141845703125e-06,
      "step": 76290,
      "training_step_time": 7.61640477180481
    },
    {
      "epoch": 2.1848678588867187e-06,
      "model_forward_time": 1.773061990737915,
      "step": 76291
    },
    {
      "epoch": 2.1848678588867187e-06,
      "step": 76291,
      "training_step_time": 7.34956431388855
    },
    {
      "epoch": 2.185821533203125e-06,
      "model_forward_time": 1.768601655960083,
      "step": 76292
    },
    {
      "epoch": 2.185821533203125e-06,
      "step": 76292,
      "training_step_time": 7.700417757034302
    },
    {
      "epoch": 2.1867752075195314e-06,
      "model_forward_time": 1.7677245140075684,
      "step": 76293
    },
    {
      "epoch": 2.1867752075195314e-06,
      "step": 76293,
      "training_step_time": 7.159084796905518
    },
    {
      "epoch": 2.1877288818359375e-06,
      "model_forward_time": 1.7619903087615967,
      "step": 76294
    },
    {
      "epoch": 2.1877288818359375e-06,
      "step": 76294,
      "training_step_time": 7.981580018997192
    },
    {
      "epoch": 2.1886825561523437e-06,
      "model_forward_time": 1.7753727436065674,
      "step": 76295
    },
    {
      "epoch": 2.1886825561523437e-06,
      "step": 76295,
      "training_step_time": 7.625619411468506
    },
    {
      "epoch": 2.18963623046875e-06,
      "model_forward_time": 1.7853872776031494,
      "step": 76296
    },
    {
      "epoch": 2.18963623046875e-06,
      "step": 76296,
      "training_step_time": 7.8078649044036865
    },
    {
      "epoch": 2.1905899047851564e-06,
      "model_forward_time": 2.392425537109375,
      "step": 76297
    },
    {
      "epoch": 2.1905899047851564e-06,
      "step": 76297,
      "training_step_time": 9.941458225250244
    },
    {
      "epoch": 2.1915435791015625e-06,
      "model_forward_time": 1.788243293762207,
      "step": 76298
    },
    {
      "epoch": 2.1915435791015625e-06,
      "step": 76298,
      "training_step_time": 7.638636112213135
    },
    {
      "epoch": 2.1924972534179687e-06,
      "model_forward_time": 1.8057332038879395,
      "step": 76299
    },
    {
      "epoch": 2.1924972534179687e-06,
      "step": 76299,
      "training_step_time": 9.242230415344238
    },
    {
      "epoch": 2.193450927734375e-06,
      "grad_norm": 0.10971119999885559,
      "learning_rate": 1.4587417575638724e-06,
      "loss": 0.054,
      "step": 76300
    },
    {
      "epoch": 2.193450927734375e-06,
      "model_forward_time": 1.7685577869415283,
      "step": 76300
    },
    {
      "dynamics_loss_avg": 0.061884854920208454,
      "epoch": 2.193450927734375e-06,
      "step": 76300
    },
    {
      "action_loss_avg": 0.0010372242890298367,
      "epoch": 2.193450927734375e-06,
      "step": 76300
    },
    {
      "epoch": 2.193450927734375e-06,
      "step": 76300,
      "training_step_time": 7.624094009399414
    },
    {
      "epoch": 2.1944046020507814e-06,
      "model_forward_time": 1.7677357196807861,
      "step": 76301
    },
    {
      "epoch": 2.1944046020507814e-06,
      "step": 76301,
      "training_step_time": 7.3764121532440186
    },
    {
      "epoch": 2.1953582763671875e-06,
      "model_forward_time": 1.770334005355835,
      "step": 76302
    },
    {
      "epoch": 2.1953582763671875e-06,
      "step": 76302,
      "training_step_time": 7.1981282234191895
    },
    {
      "epoch": 2.1963119506835937e-06,
      "model_forward_time": 1.770052194595337,
      "step": 76303
    },
    {
      "epoch": 2.1963119506835937e-06,
      "step": 76303,
      "training_step_time": 7.462799072265625
    },
    {
      "epoch": 2.197265625e-06,
      "model_forward_time": 1.7754578590393066,
      "step": 76304
    },
    {
      "epoch": 2.197265625e-06,
      "step": 76304,
      "training_step_time": 7.571494102478027
    },
    {
      "epoch": 2.1982192993164064e-06,
      "model_forward_time": 1.7847845554351807,
      "step": 76305
    },
    {
      "epoch": 2.1982192993164064e-06,
      "step": 76305,
      "training_step_time": 7.7183310985565186
    },
    {
      "epoch": 2.1991729736328125e-06,
      "model_forward_time": 2.3753883838653564,
      "step": 76306
    },
    {
      "epoch": 2.1991729736328125e-06,
      "step": 76306,
      "training_step_time": 8.797089338302612
    },
    {
      "epoch": 2.2001266479492187e-06,
      "model_forward_time": 1.768049955368042,
      "step": 76307
    },
    {
      "epoch": 2.2001266479492187e-06,
      "step": 76307,
      "training_step_time": 8.676734447479248
    },
    {
      "epoch": 2.201080322265625e-06,
      "model_forward_time": 1.4090368747711182,
      "step": 76308
    },
    {
      "epoch": 2.201080322265625e-06,
      "step": 76308,
      "training_step_time": 8.584056854248047
    },
    {
      "epoch": 2.2020339965820314e-06,
      "model_forward_time": 1.7530760765075684,
      "step": 76309
    },
    {
      "epoch": 2.2020339965820314e-06,
      "step": 76309,
      "training_step_time": 7.61190938949585
    },
    {
      "epoch": 2.2029876708984376e-06,
      "grad_norm": 0.07386387139558792,
      "learning_rate": 1.4575746675713565e-06,
      "loss": 0.0535,
      "step": 76310
    },
    {
      "epoch": 2.2029876708984376e-06,
      "model_forward_time": 1.7656896114349365,
      "step": 76310
    },
    {
      "dynamics_loss_avg": 0.048988984897732735,
      "epoch": 2.2029876708984376e-06,
      "step": 76310
    },
    {
      "action_loss_avg": 0.0008799435570836067,
      "epoch": 2.2029876708984376e-06,
      "step": 76310
    },
    {
      "epoch": 2.2029876708984376e-06,
      "step": 76310,
      "training_step_time": 7.342363595962524
    },
    {
      "epoch": 2.2039413452148437e-06,
      "model_forward_time": 1.7632050514221191,
      "step": 76311
    },
    {
      "epoch": 2.2039413452148437e-06,
      "step": 76311,
      "training_step_time": 7.781455993652344
    },
    {
      "epoch": 2.20489501953125e-06,
      "model_forward_time": 1.7751448154449463,
      "step": 76312
    },
    {
      "epoch": 2.20489501953125e-06,
      "step": 76312,
      "training_step_time": 7.371283054351807
    },
    {
      "epoch": 2.2058486938476564e-06,
      "model_forward_time": 1.7752301692962646,
      "step": 76313
    },
    {
      "epoch": 2.2058486938476564e-06,
      "step": 76313,
      "training_step_time": 7.512167692184448
    },
    {
      "epoch": 2.2068023681640626e-06,
      "model_forward_time": 1.7665719985961914,
      "step": 76314
    },
    {
      "epoch": 2.2068023681640626e-06,
      "step": 76314,
      "training_step_time": 8.708782434463501
    },
    {
      "epoch": 2.2077560424804687e-06,
      "model_forward_time": 2.4032583236694336,
      "step": 76315
    },
    {
      "epoch": 2.2077560424804687e-06,
      "step": 76315,
      "training_step_time": 8.74935531616211
    },
    {
      "epoch": 2.208709716796875e-06,
      "model_forward_time": 1.7684521675109863,
      "step": 76316
    },
    {
      "epoch": 2.208709716796875e-06,
      "step": 76316,
      "training_step_time": 10.073447704315186
    },
    {
      "epoch": 2.2096633911132814e-06,
      "model_forward_time": 1.7486217021942139,
      "step": 76317
    },
    {
      "epoch": 2.2096633911132814e-06,
      "step": 76317,
      "training_step_time": 8.74432921409607
    },
    {
      "epoch": 2.2106170654296876e-06,
      "model_forward_time": 1.8585567474365234,
      "step": 76318
    },
    {
      "epoch": 2.2106170654296876e-06,
      "step": 76318,
      "training_step_time": 7.657155275344849
    },
    {
      "epoch": 2.2115707397460937e-06,
      "model_forward_time": 1.7874336242675781,
      "step": 76319
    },
    {
      "epoch": 2.2115707397460937e-06,
      "step": 76319,
      "training_step_time": 8.160601139068604
    },
    {
      "epoch": 2.2125244140625e-06,
      "grad_norm": 0.13114479184150696,
      "learning_rate": 1.4564079649731505e-06,
      "loss": 0.0539,
      "step": 76320
    },
    {
      "epoch": 2.2125244140625e-06,
      "model_forward_time": 1.77083420753479,
      "step": 76320
    },
    {
      "dynamics_loss_avg": 0.05501849539577961,
      "epoch": 2.2125244140625e-06,
      "step": 76320
    },
    {
      "action_loss_avg": 0.001181869499851018,
      "epoch": 2.2125244140625e-06,
      "step": 76320
    },
    {
      "epoch": 2.2125244140625e-06,
      "step": 76320,
      "training_step_time": 7.6226115226745605
    },
    {
      "epoch": 2.2134780883789064e-06,
      "model_forward_time": 1.7696573734283447,
      "step": 76321
    },
    {
      "epoch": 2.2134780883789064e-06,
      "step": 76321,
      "training_step_time": 7.36532187461853
    },
    {
      "epoch": 2.2144317626953126e-06,
      "model_forward_time": 1.7614498138427734,
      "step": 76322
    },
    {
      "epoch": 2.2144317626953126e-06,
      "step": 76322,
      "training_step_time": 7.489911317825317
    },
    {
      "epoch": 2.2153854370117187e-06,
      "model_forward_time": 1.409562587738037,
      "step": 76323
    },
    {
      "epoch": 2.2153854370117187e-06,
      "step": 76323,
      "training_step_time": 8.471158266067505
    },
    {
      "epoch": 2.216339111328125e-06,
      "model_forward_time": 1.767185926437378,
      "step": 76324
    },
    {
      "epoch": 2.216339111328125e-06,
      "step": 76324,
      "training_step_time": 7.615450143814087
    },
    {
      "epoch": 2.2172927856445314e-06,
      "model_forward_time": 2.3857269287109375,
      "step": 76325
    },
    {
      "epoch": 2.2172927856445314e-06,
      "step": 76325,
      "training_step_time": 7.633261680603027
    },
    {
      "epoch": 2.2182464599609376e-06,
      "model_forward_time": 1.7647573947906494,
      "step": 76326
    },
    {
      "epoch": 2.2182464599609376e-06,
      "step": 76326,
      "training_step_time": 8.604568481445312
    },
    {
      "epoch": 2.2192001342773437e-06,
      "model_forward_time": 1.8688023090362549,
      "step": 76327
    },
    {
      "epoch": 2.2192001342773437e-06,
      "step": 76327,
      "training_step_time": 7.622948884963989
    },
    {
      "epoch": 2.22015380859375e-06,
      "model_forward_time": 1.7682697772979736,
      "step": 76328
    },
    {
      "epoch": 2.22015380859375e-06,
      "step": 76328,
      "training_step_time": 7.599305152893066
    },
    {
      "epoch": 2.2211074829101564e-06,
      "model_forward_time": 1.7641334533691406,
      "step": 76329
    },
    {
      "epoch": 2.2211074829101564e-06,
      "step": 76329,
      "training_step_time": 7.645738124847412
    },
    {
      "epoch": 2.2220611572265626e-06,
      "grad_norm": 0.16123844683170319,
      "learning_rate": 1.4552416498968402e-06,
      "loss": 0.0549,
      "step": 76330
    },
    {
      "epoch": 2.2220611572265626e-06,
      "model_forward_time": 1.4079911708831787,
      "step": 76330
    },
    {
      "dynamics_loss_avg": 0.0667981312610209,
      "epoch": 2.2220611572265626e-06,
      "step": 76330
    },
    {
      "action_loss_avg": 0.0021112642833031716,
      "epoch": 2.2220611572265626e-06,
      "step": 76330
    },
    {
      "epoch": 2.2220611572265626e-06,
      "step": 76330,
      "training_step_time": 7.353291273117065
    },
    {
      "epoch": 2.2230148315429687e-06,
      "model_forward_time": 1.775155782699585,
      "step": 76331
    },
    {
      "epoch": 2.2230148315429687e-06,
      "step": 76331,
      "training_step_time": 7.4844396114349365
    },
    {
      "epoch": 2.223968505859375e-06,
      "model_forward_time": 1.7629454135894775,
      "step": 76332
    },
    {
      "epoch": 2.223968505859375e-06,
      "step": 76332,
      "training_step_time": 7.630271911621094
    },
    {
      "epoch": 2.2249221801757814e-06,
      "model_forward_time": 1.7894814014434814,
      "step": 76333
    },
    {
      "epoch": 2.2249221801757814e-06,
      "step": 76333,
      "training_step_time": 8.311243057250977
    },
    {
      "epoch": 2.2258758544921876e-06,
      "model_forward_time": 2.3942902088165283,
      "step": 76334
    },
    {
      "epoch": 2.2258758544921876e-06,
      "step": 76334,
      "training_step_time": 10.224620342254639
    },
    {
      "epoch": 2.2268295288085937e-06,
      "model_forward_time": 1.762049674987793,
      "step": 76335
    },
    {
      "epoch": 2.2268295288085937e-06,
      "step": 76335,
      "training_step_time": 7.881278991699219
    },
    {
      "epoch": 2.227783203125e-06,
      "model_forward_time": 1.8464031219482422,
      "step": 76336
    },
    {
      "epoch": 2.227783203125e-06,
      "step": 76336,
      "training_step_time": 7.780780076980591
    },
    {
      "epoch": 2.2287368774414064e-06,
      "model_forward_time": 1.7710037231445312,
      "step": 76337
    },
    {
      "epoch": 2.2287368774414064e-06,
      "step": 76337,
      "training_step_time": 8.365734577178955
    },
    {
      "epoch": 2.2296905517578126e-06,
      "model_forward_time": 1.7709836959838867,
      "step": 76338
    },
    {
      "epoch": 2.2296905517578126e-06,
      "step": 76338,
      "training_step_time": 7.630802869796753
    },
    {
      "epoch": 2.2306442260742187e-06,
      "model_forward_time": 1.766967535018921,
      "step": 76339
    },
    {
      "epoch": 2.2306442260742187e-06,
      "step": 76339,
      "training_step_time": 7.475619554519653
    },
    {
      "epoch": 2.231597900390625e-06,
      "grad_norm": 0.10835415869951248,
      "learning_rate": 1.4540757224699747e-06,
      "loss": 0.0546,
      "step": 76340
    },
    {
      "epoch": 2.231597900390625e-06,
      "model_forward_time": 1.7680397033691406,
      "step": 76340
    },
    {
      "dynamics_loss_avg": 0.052689403854310514,
      "epoch": 2.231597900390625e-06,
      "step": 76340
    },
    {
      "action_loss_avg": 0.0008201185817597433,
      "epoch": 2.231597900390625e-06,
      "step": 76340
    },
    {
      "epoch": 2.231597900390625e-06,
      "step": 76340,
      "training_step_time": 7.601105690002441
    },
    {
      "epoch": 2.2325515747070314e-06,
      "model_forward_time": 1.7691290378570557,
      "step": 76341
    },
    {
      "epoch": 2.2325515747070314e-06,
      "step": 76341,
      "training_step_time": 7.65855598449707
    },
    {
      "epoch": 2.2335052490234376e-06,
      "model_forward_time": 1.7730712890625,
      "step": 76342
    },
    {
      "epoch": 2.2335052490234376e-06,
      "step": 76342,
      "training_step_time": 9.353236675262451
    },
    {
      "epoch": 2.2344589233398437e-06,
      "model_forward_time": 2.3750622272491455,
      "step": 76343
    },
    {
      "epoch": 2.2344589233398437e-06,
      "step": 76343,
      "training_step_time": 8.608688592910767
    },
    {
      "epoch": 2.23541259765625e-06,
      "model_forward_time": 1.7747039794921875,
      "step": 76344
    },
    {
      "epoch": 2.23541259765625e-06,
      "step": 76344,
      "training_step_time": 9.195502996444702
    },
    {
      "epoch": 2.2363662719726564e-06,
      "model_forward_time": 1.8175265789031982,
      "step": 76345
    },
    {
      "epoch": 2.2363662719726564e-06,
      "step": 76345,
      "training_step_time": 7.721466064453125
    },
    {
      "epoch": 2.2373199462890626e-06,
      "model_forward_time": 1.7758615016937256,
      "step": 76346
    },
    {
      "epoch": 2.2373199462890626e-06,
      "step": 76346,
      "training_step_time": 7.535699844360352
    },
    {
      "epoch": 2.2382736206054687e-06,
      "model_forward_time": 1.7930257320404053,
      "step": 76347
    },
    {
      "epoch": 2.2382736206054687e-06,
      "step": 76347,
      "training_step_time": 7.270308971405029
    },
    {
      "epoch": 2.239227294921875e-06,
      "model_forward_time": 1.766075611114502,
      "step": 76348
    },
    {
      "epoch": 2.239227294921875e-06,
      "step": 76348,
      "training_step_time": 7.355417013168335
    },
    {
      "epoch": 2.2401809692382814e-06,
      "model_forward_time": 1.7703971862792969,
      "step": 76349
    },
    {
      "epoch": 2.2401809692382814e-06,
      "step": 76349,
      "training_step_time": 7.125102519989014
    },
    {
      "epoch": 2.2411346435546876e-06,
      "grad_norm": 0.09079916030168533,
      "learning_rate": 1.452910182820056e-06,
      "loss": 0.0543,
      "step": 76350
    },
    {
      "epoch": 2.2411346435546876e-06,
      "model_forward_time": 1.7701036930084229,
      "step": 76350
    },
    {
      "dynamics_loss_avg": 0.04882164988666773,
      "epoch": 2.2411346435546876e-06,
      "step": 76350
    },
    {
      "action_loss_avg": 0.0012268043181393295,
      "epoch": 2.2411346435546876e-06,
      "step": 76350
    },
    {
      "epoch": 2.2411346435546876e-06,
      "step": 76350,
      "training_step_time": 7.551753520965576
    },
    {
      "epoch": 2.2420883178710937e-06,
      "model_forward_time": 1.7749645709991455,
      "step": 76351
    },
    {
      "epoch": 2.2420883178710937e-06,
      "step": 76351,
      "training_step_time": 8.512275218963623
    },
    {
      "epoch": 2.2430419921875e-06,
      "model_forward_time": 2.396894693374634,
      "step": 76352
    },
    {
      "epoch": 2.2430419921875e-06,
      "step": 76352,
      "training_step_time": 8.664592266082764
    },
    {
      "epoch": 2.2439956665039064e-06,
      "model_forward_time": 1.7676916122436523,
      "step": 76353
    },
    {
      "epoch": 2.2439956665039064e-06,
      "step": 76353,
      "training_step_time": 8.518924236297607
    },
    {
      "epoch": 2.2449493408203126e-06,
      "model_forward_time": 1.7770116329193115,
      "step": 76354
    },
    {
      "epoch": 2.2449493408203126e-06,
      "step": 76354,
      "training_step_time": 7.734110593795776
    },
    {
      "epoch": 2.2459030151367187e-06,
      "model_forward_time": 1.772843837738037,
      "step": 76355
    },
    {
      "epoch": 2.2459030151367187e-06,
      "step": 76355,
      "training_step_time": 7.508015871047974
    },
    {
      "epoch": 2.246856689453125e-06,
      "model_forward_time": 1.7566955089569092,
      "step": 76356
    },
    {
      "epoch": 2.246856689453125e-06,
      "step": 76356,
      "training_step_time": 7.385878801345825
    },
    {
      "epoch": 2.2478103637695314e-06,
      "model_forward_time": 1.772749423980713,
      "step": 76357
    },
    {
      "epoch": 2.2478103637695314e-06,
      "step": 76357,
      "training_step_time": 7.444323301315308
    },
    {
      "epoch": 2.2487640380859376e-06,
      "model_forward_time": 1.786499261856079,
      "step": 76358
    },
    {
      "epoch": 2.2487640380859376e-06,
      "step": 76358,
      "training_step_time": 6.751251935958862
    },
    {
      "epoch": 2.2497177124023437e-06,
      "model_forward_time": 1.7664191722869873,
      "step": 76359
    },
    {
      "epoch": 2.2497177124023437e-06,
      "step": 76359,
      "training_step_time": 8.2108154296875
    },
    {
      "epoch": 2.25067138671875e-06,
      "grad_norm": 0.10153217613697052,
      "learning_rate": 1.4517450310745467e-06,
      "loss": 0.054,
      "step": 76360
    },
    {
      "epoch": 2.25067138671875e-06,
      "model_forward_time": 1.7657532691955566,
      "step": 76360
    },
    {
      "dynamics_loss_avg": 0.047692294046282765,
      "epoch": 2.25067138671875e-06,
      "step": 76360
    },
    {
      "action_loss_avg": 0.0015915946365566923,
      "epoch": 2.25067138671875e-06,
      "step": 76360
    },
    {
      "epoch": 2.25067138671875e-06,
      "step": 76360,
      "training_step_time": 7.641659498214722
    },
    {
      "epoch": 2.2516250610351564e-06,
      "model_forward_time": 2.4211955070495605,
      "step": 76361
    },
    {
      "epoch": 2.2516250610351564e-06,
      "step": 76361,
      "training_step_time": 9.826472520828247
    },
    {
      "epoch": 2.2525787353515626e-06,
      "model_forward_time": 1.7668991088867188,
      "step": 76362
    },
    {
      "epoch": 2.2525787353515626e-06,
      "step": 76362,
      "training_step_time": 9.30565857887268
    },
    {
      "epoch": 2.2535324096679687e-06,
      "model_forward_time": 1.7606110572814941,
      "step": 76363
    },
    {
      "epoch": 2.2535324096679687e-06,
      "step": 76363,
      "training_step_time": 7.798309087753296
    },
    {
      "epoch": 2.254486083984375e-06,
      "model_forward_time": 1.8706421852111816,
      "step": 76364
    },
    {
      "epoch": 2.254486083984375e-06,
      "step": 76364,
      "training_step_time": 8.118180513381958
    },
    {
      "epoch": 2.2554397583007815e-06,
      "model_forward_time": 1.7781195640563965,
      "step": 76365
    },
    {
      "epoch": 2.2554397583007815e-06,
      "step": 76365,
      "training_step_time": 8.11395001411438
    },
    {
      "epoch": 2.2563934326171876e-06,
      "model_forward_time": 1.7706756591796875,
      "step": 76366
    },
    {
      "epoch": 2.2563934326171876e-06,
      "step": 76366,
      "training_step_time": 8.302916526794434
    },
    {
      "epoch": 2.2573471069335937e-06,
      "model_forward_time": 1.7643013000488281,
      "step": 76367
    },
    {
      "epoch": 2.2573471069335937e-06,
      "step": 76367,
      "training_step_time": 6.689430475234985
    },
    {
      "epoch": 2.25830078125e-06,
      "model_forward_time": 1.767336130142212,
      "step": 76368
    },
    {
      "epoch": 2.25830078125e-06,
      "step": 76368,
      "training_step_time": 7.532355308532715
    },
    {
      "epoch": 2.2592544555664065e-06,
      "model_forward_time": 1.7737629413604736,
      "step": 76369
    },
    {
      "epoch": 2.2592544555664065e-06,
      "step": 76369,
      "training_step_time": 8.088014125823975
    },
    {
      "epoch": 2.2602081298828126e-06,
      "grad_norm": 0.08268992602825165,
      "learning_rate": 1.450580267360866e-06,
      "loss": 0.0542,
      "step": 76370
    },
    {
      "epoch": 2.2602081298828126e-06,
      "model_forward_time": 2.400033950805664,
      "step": 76370
    },
    {
      "dynamics_loss_avg": 0.05588083723559976,
      "epoch": 2.2602081298828126e-06,
      "step": 76370
    },
    {
      "action_loss_avg": 0.0019645912245323414,
      "epoch": 2.2602081298828126e-06,
      "step": 76370
    },
    {
      "epoch": 2.2602081298828126e-06,
      "step": 76370,
      "training_step_time": 8.98556113243103
    },
    {
      "epoch": 2.2611618041992187e-06,
      "model_forward_time": 1.7897062301635742,
      "step": 76371
    },
    {
      "epoch": 2.2611618041992187e-06,
      "step": 76371,
      "training_step_time": 11.37338376045227
    },
    {
      "epoch": 2.262115478515625e-06,
      "model_forward_time": 1.7616312503814697,
      "step": 76372
    },
    {
      "epoch": 2.262115478515625e-06,
      "step": 76372,
      "training_step_time": 7.671279191970825
    },
    {
      "epoch": 2.2630691528320315e-06,
      "model_forward_time": 1.8262221813201904,
      "step": 76373
    },
    {
      "epoch": 2.2630691528320315e-06,
      "step": 76373,
      "training_step_time": 8.186304092407227
    },
    {
      "epoch": 2.2640228271484376e-06,
      "model_forward_time": 1.768226146697998,
      "step": 76374
    },
    {
      "epoch": 2.2640228271484376e-06,
      "step": 76374,
      "training_step_time": 9.058131694793701
    },
    {
      "epoch": 2.2649765014648438e-06,
      "model_forward_time": 1.7725298404693604,
      "step": 76375
    },
    {
      "epoch": 2.2649765014648438e-06,
      "step": 76375,
      "training_step_time": 7.278946399688721
    },
    {
      "epoch": 2.26593017578125e-06,
      "model_forward_time": 1.772291660308838,
      "step": 76376
    },
    {
      "epoch": 2.26593017578125e-06,
      "step": 76376,
      "training_step_time": 7.04114031791687
    },
    {
      "epoch": 2.266883850097656e-06,
      "model_forward_time": 1.7681684494018555,
      "step": 76377
    },
    {
      "epoch": 2.266883850097656e-06,
      "step": 76377,
      "training_step_time": 7.533172607421875
    },
    {
      "epoch": 2.2678375244140626e-06,
      "model_forward_time": 1.7604517936706543,
      "step": 76378
    },
    {
      "epoch": 2.2678375244140626e-06,
      "step": 76378,
      "training_step_time": 8.638778448104858
    },
    {
      "epoch": 2.2687911987304688e-06,
      "model_forward_time": 1.771385669708252,
      "step": 76379
    },
    {
      "epoch": 2.2687911987304688e-06,
      "step": 76379,
      "training_step_time": 10.279122352600098
    },
    {
      "epoch": 2.269744873046875e-06,
      "grad_norm": 0.08120062947273254,
      "learning_rate": 1.4494158918063923e-06,
      "loss": 0.0533,
      "step": 76380
    },
    {
      "epoch": 2.269744873046875e-06,
      "model_forward_time": 1.7922286987304688,
      "step": 76380
    },
    {
      "dynamics_loss_avg": 0.04512011259794235,
      "epoch": 2.269744873046875e-06,
      "step": 76380
    },
    {
      "action_loss_avg": 0.0015317121549742296,
      "epoch": 2.269744873046875e-06,
      "step": 76380
    },
    {
      "epoch": 2.269744873046875e-06,
      "step": 76380,
      "training_step_time": 7.618421316146851
    },
    {
      "epoch": 2.270698547363281e-06,
      "model_forward_time": 1.7563269138336182,
      "step": 76381
    },
    {
      "epoch": 2.270698547363281e-06,
      "step": 76381,
      "training_step_time": 8.301602602005005
    },
    {
      "epoch": 2.2716522216796876e-06,
      "model_forward_time": 1.8243632316589355,
      "step": 76382
    },
    {
      "epoch": 2.2716522216796876e-06,
      "step": 76382,
      "training_step_time": 7.639556407928467
    },
    {
      "epoch": 2.2726058959960938e-06,
      "model_forward_time": 1.79872727394104,
      "step": 76383
    },
    {
      "epoch": 2.2726058959960938e-06,
      "step": 76383,
      "training_step_time": 7.5510358810424805
    },
    {
      "epoch": 2.2735595703125e-06,
      "model_forward_time": 1.7606003284454346,
      "step": 76384
    },
    {
      "epoch": 2.2735595703125e-06,
      "step": 76384,
      "training_step_time": 7.369636297225952
    },
    {
      "epoch": 2.274513244628906e-06,
      "model_forward_time": 1.7720999717712402,
      "step": 76385
    },
    {
      "epoch": 2.274513244628906e-06,
      "step": 76385,
      "training_step_time": 7.342207908630371
    },
    {
      "epoch": 2.2754669189453126e-06,
      "model_forward_time": 1.7705285549163818,
      "step": 76386
    },
    {
      "epoch": 2.2754669189453126e-06,
      "step": 76386,
      "training_step_time": 8.14322018623352
    },
    {
      "epoch": 2.2764205932617188e-06,
      "model_forward_time": 1.774747371673584,
      "step": 76387
    },
    {
      "epoch": 2.2764205932617188e-06,
      "step": 76387,
      "training_step_time": 7.182011604309082
    },
    {
      "epoch": 2.277374267578125e-06,
      "model_forward_time": 1.777134656906128,
      "step": 76388
    },
    {
      "epoch": 2.277374267578125e-06,
      "step": 76388,
      "training_step_time": 7.638234853744507
    },
    {
      "epoch": 2.278327941894531e-06,
      "model_forward_time": 1.9840519428253174,
      "step": 76389
    },
    {
      "epoch": 2.278327941894531e-06,
      "step": 76389,
      "training_step_time": 9.303475856781006
    },
    {
      "epoch": 2.2792816162109376e-06,
      "grad_norm": 0.09483892470598221,
      "learning_rate": 1.448251904538457e-06,
      "loss": 0.0562,
      "step": 76390
    },
    {
      "epoch": 2.2792816162109376e-06,
      "model_forward_time": 1.7655508518218994,
      "step": 76390
    },
    {
      "dynamics_loss_avg": 0.052596263214945795,
      "epoch": 2.2792816162109376e-06,
      "step": 76390
    },
    {
      "action_loss_avg": 0.0016240237397141754,
      "epoch": 2.2792816162109376e-06,
      "step": 76390
    },
    {
      "epoch": 2.2792816162109376e-06,
      "step": 76390,
      "training_step_time": 8.849497079849243
    },
    {
      "epoch": 2.2802352905273438e-06,
      "model_forward_time": 1.7673907279968262,
      "step": 76391
    },
    {
      "epoch": 2.2802352905273438e-06,
      "step": 76391,
      "training_step_time": 8.619359254837036
    },
    {
      "epoch": 2.28118896484375e-06,
      "model_forward_time": 1.8703489303588867,
      "step": 76392
    },
    {
      "epoch": 2.28118896484375e-06,
      "step": 76392,
      "training_step_time": 7.549937963485718
    },
    {
      "epoch": 2.282142639160156e-06,
      "model_forward_time": 1.7652397155761719,
      "step": 76393
    },
    {
      "epoch": 2.282142639160156e-06,
      "step": 76393,
      "training_step_time": 7.268333911895752
    },
    {
      "epoch": 2.2830963134765626e-06,
      "model_forward_time": 1.769946813583374,
      "step": 76394
    },
    {
      "epoch": 2.2830963134765626e-06,
      "step": 76394,
      "training_step_time": 7.37978720664978
    },
    {
      "epoch": 2.2840499877929688e-06,
      "model_forward_time": 1.4081788063049316,
      "step": 76395
    },
    {
      "epoch": 2.2840499877929688e-06,
      "step": 76395,
      "training_step_time": 7.5610010623931885
    },
    {
      "epoch": 2.285003662109375e-06,
      "model_forward_time": 1.770228624343872,
      "step": 76396
    },
    {
      "epoch": 2.285003662109375e-06,
      "step": 76396,
      "training_step_time": 7.480467081069946
    },
    {
      "epoch": 2.285957336425781e-06,
      "model_forward_time": 1.7909212112426758,
      "step": 76397
    },
    {
      "epoch": 2.285957336425781e-06,
      "step": 76397,
      "training_step_time": 7.700897693634033
    },
    {
      "epoch": 2.2869110107421876e-06,
      "model_forward_time": 2.3884572982788086,
      "step": 76398
    },
    {
      "epoch": 2.2869110107421876e-06,
      "step": 76398,
      "training_step_time": 8.726899862289429
    },
    {
      "epoch": 2.2878646850585938e-06,
      "model_forward_time": 1.7685484886169434,
      "step": 76399
    },
    {
      "epoch": 2.2878646850585938e-06,
      "step": 76399,
      "training_step_time": 8.776034593582153
    },
    {
      "epoch": 2.288818359375e-06,
      "grad_norm": 0.08214206993579865,
      "learning_rate": 1.4470883056843549e-06,
      "loss": 0.0559,
      "step": 76400
    },
    {
      "epoch": 2.288818359375e-06,
      "model_forward_time": 1.7635374069213867,
      "step": 76400
    },
    {
      "dynamics_loss_avg": 0.0383527398109436,
      "epoch": 2.288818359375e-06,
      "step": 76400
    },
    {
      "action_loss_avg": 0.0009060296637471765,
      "epoch": 2.288818359375e-06,
      "step": 76400
    },
    {
      "epoch": 2.288818359375e-06,
      "step": 76400,
      "training_step_time": 9.318118572235107
    },
    {
      "epoch": 2.289772033691406e-06,
      "model_forward_time": 1.8457024097442627,
      "step": 76401
    },
    {
      "epoch": 2.289772033691406e-06,
      "step": 76401,
      "training_step_time": 7.5753257274627686
    },
    {
      "epoch": 2.2907257080078126e-06,
      "model_forward_time": 1.767099380493164,
      "step": 76402
    },
    {
      "epoch": 2.2907257080078126e-06,
      "step": 76402,
      "training_step_time": 7.976744890213013
    },
    {
      "epoch": 2.2916793823242188e-06,
      "model_forward_time": 1.771378993988037,
      "step": 76403
    },
    {
      "epoch": 2.2916793823242188e-06,
      "step": 76403,
      "training_step_time": 7.313287734985352
    },
    {
      "epoch": 2.292633056640625e-06,
      "model_forward_time": 1.7653510570526123,
      "step": 76404
    },
    {
      "epoch": 2.292633056640625e-06,
      "step": 76404,
      "training_step_time": 7.556655406951904
    },
    {
      "epoch": 2.293586730957031e-06,
      "model_forward_time": 1.7625324726104736,
      "step": 76405
    },
    {
      "epoch": 2.293586730957031e-06,
      "step": 76405,
      "training_step_time": 7.489777088165283
    },
    {
      "epoch": 2.2945404052734376e-06,
      "model_forward_time": 1.775517463684082,
      "step": 76406
    },
    {
      "epoch": 2.2945404052734376e-06,
      "step": 76406,
      "training_step_time": 7.7702226638793945
    },
    {
      "epoch": 2.2954940795898438e-06,
      "model_forward_time": 2.4074952602386475,
      "step": 76407
    },
    {
      "epoch": 2.2954940795898438e-06,
      "step": 76407,
      "training_step_time": 7.78584098815918
    },
    {
      "epoch": 2.29644775390625e-06,
      "model_forward_time": 1.7711124420166016,
      "step": 76408
    },
    {
      "epoch": 2.29644775390625e-06,
      "step": 76408,
      "training_step_time": 10.44139575958252
    },
    {
      "epoch": 2.297401428222656e-06,
      "model_forward_time": 1.77431321144104,
      "step": 76409
    },
    {
      "epoch": 2.297401428222656e-06,
      "step": 76409,
      "training_step_time": 8.078377723693848
    },
    {
      "epoch": 2.2983551025390626e-06,
      "grad_norm": 0.09573321044445038,
      "learning_rate": 1.4459250953713321e-06,
      "loss": 0.0553,
      "step": 76410
    },
    {
      "epoch": 2.2983551025390626e-06,
      "model_forward_time": 1.8199386596679688,
      "step": 76410
    },
    {
      "dynamics_loss_avg": 0.04958254555240273,
      "epoch": 2.2983551025390626e-06,
      "step": 76410
    },
    {
      "action_loss_avg": 0.001163580981665291,
      "epoch": 2.2983551025390626e-06,
      "step": 76410
    },
    {
      "epoch": 2.2983551025390626e-06,
      "step": 76410,
      "training_step_time": 7.660593748092651
    },
    {
      "epoch": 2.2993087768554688e-06,
      "model_forward_time": 1.7731070518493652,
      "step": 76411
    },
    {
      "epoch": 2.2993087768554688e-06,
      "step": 76411,
      "training_step_time": 8.235317707061768
    },
    {
      "epoch": 2.300262451171875e-06,
      "model_forward_time": 1.7718842029571533,
      "step": 76412
    },
    {
      "epoch": 2.300262451171875e-06,
      "step": 76412,
      "training_step_time": 7.334044933319092
    },
    {
      "epoch": 2.301216125488281e-06,
      "model_forward_time": 1.7682912349700928,
      "step": 76413
    },
    {
      "epoch": 2.301216125488281e-06,
      "step": 76413,
      "training_step_time": 7.615183115005493
    },
    {
      "epoch": 2.3021697998046876e-06,
      "model_forward_time": 1.7591454982757568,
      "step": 76414
    },
    {
      "epoch": 2.3021697998046876e-06,
      "step": 76414,
      "training_step_time": 7.450731992721558
    },
    {
      "epoch": 2.3031234741210938e-06,
      "model_forward_time": 1.770885705947876,
      "step": 76415
    },
    {
      "epoch": 2.3031234741210938e-06,
      "step": 76415,
      "training_step_time": 8.1025230884552
    },
    {
      "epoch": 2.3040771484375e-06,
      "model_forward_time": 2.399533271789551,
      "step": 76416
    },
    {
      "epoch": 2.3040771484375e-06,
      "step": 76416,
      "training_step_time": 8.59041452407837
    },
    {
      "epoch": 2.305030822753906e-06,
      "model_forward_time": 1.7686634063720703,
      "step": 76417
    },
    {
      "epoch": 2.305030822753906e-06,
      "step": 76417,
      "training_step_time": 9.239643812179565
    },
    {
      "epoch": 2.3059844970703126e-06,
      "model_forward_time": 1.7680084705352783,
      "step": 76418
    },
    {
      "epoch": 2.3059844970703126e-06,
      "step": 76418,
      "training_step_time": 8.306901216506958
    },
    {
      "epoch": 2.3069381713867188e-06,
      "model_forward_time": 1.7940270900726318,
      "step": 76419
    },
    {
      "epoch": 2.3069381713867188e-06,
      "step": 76419,
      "training_step_time": 7.949188470840454
    },
    {
      "epoch": 2.307891845703125e-06,
      "grad_norm": 0.1060512363910675,
      "learning_rate": 1.4447622737265982e-06,
      "loss": 0.0551,
      "step": 76420
    },
    {
      "epoch": 2.307891845703125e-06,
      "model_forward_time": 1.7677197456359863,
      "step": 76420
    },
    {
      "dynamics_loss_avg": 0.06099812807515263,
      "epoch": 2.307891845703125e-06,
      "step": 76420
    },
    {
      "action_loss_avg": 0.0019148491701344028,
      "epoch": 2.307891845703125e-06,
      "step": 76420
    },
    {
      "epoch": 2.307891845703125e-06,
      "step": 76420,
      "training_step_time": 7.680434703826904
    },
    {
      "epoch": 2.308845520019531e-06,
      "model_forward_time": 1.7699930667877197,
      "step": 76421
    },
    {
      "epoch": 2.308845520019531e-06,
      "step": 76421,
      "training_step_time": 7.458395004272461
    },
    {
      "epoch": 2.3097991943359376e-06,
      "model_forward_time": 1.765808343887329,
      "step": 76422
    },
    {
      "epoch": 2.3097991943359376e-06,
      "step": 76422,
      "training_step_time": 7.696298837661743
    },
    {
      "epoch": 2.310752868652344e-06,
      "model_forward_time": 1.76741623878479,
      "step": 76423
    },
    {
      "epoch": 2.310752868652344e-06,
      "step": 76423,
      "training_step_time": 7.575801849365234
    },
    {
      "epoch": 2.31170654296875e-06,
      "model_forward_time": 1.7980613708496094,
      "step": 76424
    },
    {
      "epoch": 2.31170654296875e-06,
      "step": 76424,
      "training_step_time": 7.734748601913452
    },
    {
      "epoch": 2.312660217285156e-06,
      "model_forward_time": 2.397270679473877,
      "step": 76425
    },
    {
      "epoch": 2.312660217285156e-06,
      "step": 76425,
      "training_step_time": 8.770066738128662
    },
    {
      "epoch": 2.3136138916015626e-06,
      "model_forward_time": 1.7720043659210205,
      "step": 76426
    },
    {
      "epoch": 2.3136138916015626e-06,
      "step": 76426,
      "training_step_time": 8.79090929031372
    },
    {
      "epoch": 2.314567565917969e-06,
      "model_forward_time": 1.769026756286621,
      "step": 76427
    },
    {
      "epoch": 2.314567565917969e-06,
      "step": 76427,
      "training_step_time": 7.7595720291137695
    },
    {
      "epoch": 2.315521240234375e-06,
      "model_forward_time": 1.7786822319030762,
      "step": 76428
    },
    {
      "epoch": 2.315521240234375e-06,
      "step": 76428,
      "training_step_time": 8.781941652297974
    },
    {
      "epoch": 2.316474914550781e-06,
      "model_forward_time": 1.889509916305542,
      "step": 76429
    },
    {
      "epoch": 2.316474914550781e-06,
      "step": 76429,
      "training_step_time": 7.755558013916016
    },
    {
      "epoch": 2.3174285888671877e-06,
      "grad_norm": 0.08399456739425659,
      "learning_rate": 1.4435998408773162e-06,
      "loss": 0.0549,
      "step": 76430
    },
    {
      "epoch": 2.3174285888671877e-06,
      "model_forward_time": 1.761993646621704,
      "step": 76430
    },
    {
      "dynamics_loss_avg": 0.07599551752209663,
      "epoch": 2.3174285888671877e-06,
      "step": 76430
    },
    {
      "action_loss_avg": 0.001957182481419295,
      "epoch": 2.3174285888671877e-06,
      "step": 76430
    },
    {
      "epoch": 2.3174285888671877e-06,
      "step": 76430,
      "training_step_time": 7.501455068588257
    },
    {
      "epoch": 2.318382263183594e-06,
      "model_forward_time": 1.7785778045654297,
      "step": 76431
    },
    {
      "epoch": 2.318382263183594e-06,
      "step": 76431,
      "training_step_time": 7.797752857208252
    },
    {
      "epoch": 2.3193359375e-06,
      "model_forward_time": 1.7724084854125977,
      "step": 76432
    },
    {
      "epoch": 2.3193359375e-06,
      "step": 76432,
      "training_step_time": 7.673639297485352
    },
    {
      "epoch": 2.320289611816406e-06,
      "model_forward_time": 1.4142887592315674,
      "step": 76433
    },
    {
      "epoch": 2.320289611816406e-06,
      "step": 76433,
      "training_step_time": 7.758272171020508
    },
    {
      "epoch": 2.3212432861328127e-06,
      "model_forward_time": 1.7760841846466064,
      "step": 76434
    },
    {
      "epoch": 2.3212432861328127e-06,
      "step": 76434,
      "training_step_time": 8.564802408218384
    },
    {
      "epoch": 2.322196960449219e-06,
      "model_forward_time": 2.4026246070861816,
      "step": 76435
    },
    {
      "epoch": 2.322196960449219e-06,
      "step": 76435,
      "training_step_time": 10.402227640151978
    },
    {
      "epoch": 2.323150634765625e-06,
      "model_forward_time": 1.7680432796478271,
      "step": 76436
    },
    {
      "epoch": 2.323150634765625e-06,
      "step": 76436,
      "training_step_time": 8.46993374824524
    },
    {
      "epoch": 2.324104309082031e-06,
      "model_forward_time": 1.767399787902832,
      "step": 76437
    },
    {
      "epoch": 2.324104309082031e-06,
      "step": 76437,
      "training_step_time": 9.243709325790405
    },
    {
      "epoch": 2.3250579833984377e-06,
      "model_forward_time": 1.8613877296447754,
      "step": 76438
    },
    {
      "epoch": 2.3250579833984377e-06,
      "step": 76438,
      "training_step_time": 7.691319942474365
    },
    {
      "epoch": 2.326011657714844e-06,
      "model_forward_time": 1.795128583908081,
      "step": 76439
    },
    {
      "epoch": 2.326011657714844e-06,
      "step": 76439,
      "training_step_time": 7.32088303565979
    },
    {
      "epoch": 2.32696533203125e-06,
      "grad_norm": 0.07015351951122284,
      "learning_rate": 1.4424377969506103e-06,
      "loss": 0.056,
      "step": 76440
    },
    {
      "epoch": 2.32696533203125e-06,
      "model_forward_time": 1.766974687576294,
      "step": 76440
    },
    {
      "dynamics_loss_avg": 0.04728306252509355,
      "epoch": 2.32696533203125e-06,
      "step": 76440
    },
    {
      "action_loss_avg": 0.0011213626858079806,
      "epoch": 2.32696533203125e-06,
      "step": 76440
    },
    {
      "epoch": 2.32696533203125e-06,
      "step": 76440,
      "training_step_time": 7.58478856086731
    },
    {
      "epoch": 2.327919006347656e-06,
      "model_forward_time": 1.771547794342041,
      "step": 76441
    },
    {
      "epoch": 2.327919006347656e-06,
      "step": 76441,
      "training_step_time": 7.3739564418792725
    },
    {
      "epoch": 2.3288726806640627e-06,
      "model_forward_time": 1.7730979919433594,
      "step": 76442
    },
    {
      "epoch": 2.3288726806640627e-06,
      "step": 76442,
      "training_step_time": 8.239140272140503
    },
    {
      "epoch": 2.329826354980469e-06,
      "model_forward_time": 1.8192744255065918,
      "step": 76443
    },
    {
      "epoch": 2.329826354980469e-06,
      "step": 76443,
      "training_step_time": 7.645193099975586
    },
    {
      "epoch": 2.330780029296875e-06,
      "model_forward_time": 2.401075601577759,
      "step": 76444
    },
    {
      "epoch": 2.330780029296875e-06,
      "step": 76444,
      "training_step_time": 8.457146406173706
    },
    {
      "epoch": 2.331733703613281e-06,
      "model_forward_time": 1.7735741138458252,
      "step": 76445
    },
    {
      "epoch": 2.331733703613281e-06,
      "step": 76445,
      "training_step_time": 9.702628374099731
    },
    {
      "epoch": 2.3326873779296877e-06,
      "model_forward_time": 1.7696781158447266,
      "step": 76446
    },
    {
      "epoch": 2.3326873779296877e-06,
      "step": 76446,
      "training_step_time": 8.413316011428833
    },
    {
      "epoch": 2.333641052246094e-06,
      "model_forward_time": 1.775702953338623,
      "step": 76447
    },
    {
      "epoch": 2.333641052246094e-06,
      "step": 76447,
      "training_step_time": 7.725043058395386
    },
    {
      "epoch": 2.3345947265625e-06,
      "model_forward_time": 1.7750787734985352,
      "step": 76448
    },
    {
      "epoch": 2.3345947265625e-06,
      "step": 76448,
      "training_step_time": 7.379528045654297
    },
    {
      "epoch": 2.335548400878906e-06,
      "model_forward_time": 1.7687349319458008,
      "step": 76449
    },
    {
      "epoch": 2.335548400878906e-06,
      "step": 76449,
      "training_step_time": 7.637158393859863
    },
    {
      "epoch": 2.3365020751953127e-06,
      "grad_norm": 0.09776084870100021,
      "learning_rate": 1.4412761420735567e-06,
      "loss": 0.0558,
      "step": 76450
    },
    {
      "epoch": 2.3365020751953127e-06,
      "model_forward_time": 1.770566701889038,
      "step": 76450
    },
    {
      "dynamics_loss_avg": 0.05345573648810387,
      "epoch": 2.3365020751953127e-06,
      "step": 76450
    },
    {
      "action_loss_avg": 0.0014267589780502022,
      "epoch": 2.3365020751953127e-06,
      "step": 76450
    },
    {
      "epoch": 2.3365020751953127e-06,
      "step": 76450,
      "training_step_time": 7.631020784378052
    },
    {
      "epoch": 2.337455749511719e-06,
      "model_forward_time": 1.7751593589782715,
      "step": 76451
    },
    {
      "epoch": 2.337455749511719e-06,
      "step": 76451,
      "training_step_time": 8.31797170639038
    },
    {
      "epoch": 2.338409423828125e-06,
      "model_forward_time": 1.7743580341339111,
      "step": 76452
    },
    {
      "epoch": 2.338409423828125e-06,
      "step": 76452,
      "training_step_time": 8.133278131484985
    },
    {
      "epoch": 2.339363098144531e-06,
      "model_forward_time": 2.4050745964050293,
      "step": 76453
    },
    {
      "epoch": 2.339363098144531e-06,
      "step": 76453,
      "training_step_time": 10.164267301559448
    },
    {
      "epoch": 2.3403167724609377e-06,
      "model_forward_time": 1.7674424648284912,
      "step": 76454
    },
    {
      "epoch": 2.3403167724609377e-06,
      "step": 76454,
      "training_step_time": 7.591355085372925
    },
    {
      "epoch": 2.341270446777344e-06,
      "model_forward_time": 1.7713816165924072,
      "step": 76455
    },
    {
      "epoch": 2.341270446777344e-06,
      "step": 76455,
      "training_step_time": 7.70027494430542
    },
    {
      "epoch": 2.34222412109375e-06,
      "model_forward_time": 1.7729167938232422,
      "step": 76456
    },
    {
      "epoch": 2.34222412109375e-06,
      "step": 76456,
      "training_step_time": 8.238732099533081
    },
    {
      "epoch": 2.343177795410156e-06,
      "model_forward_time": 1.8741698265075684,
      "step": 76457
    },
    {
      "epoch": 2.343177795410156e-06,
      "step": 76457,
      "training_step_time": 8.460641145706177
    },
    {
      "epoch": 2.3441314697265627e-06,
      "model_forward_time": 1.7678101062774658,
      "step": 76458
    },
    {
      "epoch": 2.3441314697265627e-06,
      "step": 76458,
      "training_step_time": 8.093535423278809
    },
    {
      "epoch": 2.345085144042969e-06,
      "model_forward_time": 1.7571516036987305,
      "step": 76459
    },
    {
      "epoch": 2.345085144042969e-06,
      "step": 76459,
      "training_step_time": 7.134209394454956
    },
    {
      "epoch": 2.346038818359375e-06,
      "grad_norm": 0.09625986963510513,
      "learning_rate": 1.4401148763731953e-06,
      "loss": 0.0545,
      "step": 76460
    },
    {
      "epoch": 2.346038818359375e-06,
      "model_forward_time": 1.7616782188415527,
      "step": 76460
    },
    {
      "dynamics_loss_avg": 0.03806668445467949,
      "epoch": 2.346038818359375e-06,
      "step": 76460
    },
    {
      "action_loss_avg": 0.0010791862441692501,
      "epoch": 2.346038818359375e-06,
      "step": 76460
    },
    {
      "epoch": 2.346038818359375e-06,
      "step": 76460,
      "training_step_time": 7.664810657501221
    },
    {
      "epoch": 2.346992492675781e-06,
      "model_forward_time": 1.7774908542633057,
      "step": 76461
    },
    {
      "epoch": 2.346992492675781e-06,
      "step": 76461,
      "training_step_time": 8.148887872695923
    },
    {
      "epoch": 2.3479461669921877e-06,
      "model_forward_time": 2.4339675903320312,
      "step": 76462
    },
    {
      "epoch": 2.3479461669921877e-06,
      "step": 76462,
      "training_step_time": 7.761765003204346
    },
    {
      "epoch": 2.348899841308594e-06,
      "model_forward_time": 1.7767157554626465,
      "step": 76463
    },
    {
      "epoch": 2.348899841308594e-06,
      "step": 76463,
      "training_step_time": 8.08191967010498
    },
    {
      "epoch": 2.349853515625e-06,
      "model_forward_time": 1.8378956317901611,
      "step": 76464
    },
    {
      "epoch": 2.349853515625e-06,
      "step": 76464,
      "training_step_time": 8.219220638275146
    },
    {
      "epoch": 2.350807189941406e-06,
      "model_forward_time": 1.7630040645599365,
      "step": 76465
    },
    {
      "epoch": 2.350807189941406e-06,
      "step": 76465,
      "training_step_time": 7.7815306186676025
    },
    {
      "epoch": 2.3517608642578127e-06,
      "model_forward_time": 1.78269624710083,
      "step": 76466
    },
    {
      "epoch": 2.3517608642578127e-06,
      "step": 76466,
      "training_step_time": 7.376300573348999
    },
    {
      "epoch": 2.352714538574219e-06,
      "model_forward_time": 1.7789440155029297,
      "step": 76467
    },
    {
      "epoch": 2.352714538574219e-06,
      "step": 76467,
      "training_step_time": 8.275575160980225
    },
    {
      "epoch": 2.353668212890625e-06,
      "model_forward_time": 1.7847204208374023,
      "step": 76468
    },
    {
      "epoch": 2.353668212890625e-06,
      "step": 76468,
      "training_step_time": 7.070664405822754
    },
    {
      "epoch": 2.354621887207031e-06,
      "model_forward_time": 1.789489984512329,
      "step": 76469
    },
    {
      "epoch": 2.354621887207031e-06,
      "step": 76469,
      "training_step_time": 7.565990447998047
    },
    {
      "epoch": 2.3555755615234377e-06,
      "grad_norm": 0.107643261551857,
      "learning_rate": 1.4389539999765174e-06,
      "loss": 0.0546,
      "step": 76470
    },
    {
      "epoch": 2.3555755615234377e-06,
      "model_forward_time": 1.7880547046661377,
      "step": 76470
    },
    {
      "dynamics_loss_avg": 0.07226790655404329,
      "epoch": 2.3555755615234377e-06,
      "step": 76470
    },
    {
      "action_loss_avg": 0.0015893880045041442,
      "epoch": 2.3555755615234377e-06,
      "step": 76470
    },
    {
      "epoch": 2.3555755615234377e-06,
      "step": 76470,
      "training_step_time": 8.111281633377075
    },
    {
      "epoch": 2.356529235839844e-06,
      "model_forward_time": 1.7820076942443848,
      "step": 76471
    },
    {
      "epoch": 2.356529235839844e-06,
      "step": 76471,
      "training_step_time": 8.797863006591797
    },
    {
      "epoch": 2.35748291015625e-06,
      "model_forward_time": 1.7967844009399414,
      "step": 76472
    },
    {
      "epoch": 2.35748291015625e-06,
      "step": 76472,
      "training_step_time": 7.540526866912842
    },
    {
      "epoch": 2.358436584472656e-06,
      "model_forward_time": 1.8194549083709717,
      "step": 76473
    },
    {
      "epoch": 2.358436584472656e-06,
      "step": 76473,
      "training_step_time": 8.117457866668701
    },
    {
      "epoch": 2.3593902587890627e-06,
      "model_forward_time": 1.7929449081420898,
      "step": 76474
    },
    {
      "epoch": 2.3593902587890627e-06,
      "step": 76474,
      "training_step_time": 8.754226446151733
    },
    {
      "epoch": 2.360343933105469e-06,
      "model_forward_time": 1.79341721534729,
      "step": 76475
    },
    {
      "epoch": 2.360343933105469e-06,
      "step": 76475,
      "training_step_time": 7.5292065143585205
    },
    {
      "epoch": 2.361297607421875e-06,
      "model_forward_time": 1.783998727798462,
      "step": 76476
    },
    {
      "epoch": 2.361297607421875e-06,
      "step": 76476,
      "training_step_time": 8.332332611083984
    },
    {
      "epoch": 2.362251281738281e-06,
      "model_forward_time": 1.7870674133300781,
      "step": 76477
    },
    {
      "epoch": 2.362251281738281e-06,
      "step": 76477,
      "training_step_time": 7.367329835891724
    },
    {
      "epoch": 2.3632049560546877e-06,
      "model_forward_time": 1.7732343673706055,
      "step": 76478
    },
    {
      "epoch": 2.3632049560546877e-06,
      "step": 76478,
      "training_step_time": 8.219512701034546
    },
    {
      "epoch": 2.364158630371094e-06,
      "model_forward_time": 1.7717053890228271,
      "step": 76479
    },
    {
      "epoch": 2.364158630371094e-06,
      "step": 76479,
      "training_step_time": 8.64565134048462
    },
    {
      "epoch": 2.3651123046875e-06,
      "grad_norm": 0.07205642014741898,
      "learning_rate": 1.4377935130104758e-06,
      "loss": 0.0552,
      "step": 76480
    },
    {
      "epoch": 2.3651123046875e-06,
      "model_forward_time": 1.7809514999389648,
      "step": 76480
    },
    {
      "dynamics_loss_avg": 0.05725397057831287,
      "epoch": 2.3651123046875e-06,
      "step": 76480
    },
    {
      "action_loss_avg": 0.001045966413221322,
      "epoch": 2.3651123046875e-06,
      "step": 76480
    },
    {
      "epoch": 2.3651123046875e-06,
      "step": 76480,
      "training_step_time": 7.884386301040649
    },
    {
      "epoch": 2.366065979003906e-06,
      "model_forward_time": 2.3896305561065674,
      "step": 76481
    },
    {
      "epoch": 2.366065979003906e-06,
      "step": 76481,
      "training_step_time": 9.205905199050903
    },
    {
      "epoch": 2.3670196533203127e-06,
      "model_forward_time": 1.7868332862854004,
      "step": 76482
    },
    {
      "epoch": 2.3670196533203127e-06,
      "step": 76482,
      "training_step_time": 9.003363609313965
    },
    {
      "epoch": 2.367973327636719e-06,
      "model_forward_time": 1.768622636795044,
      "step": 76483
    },
    {
      "epoch": 2.367973327636719e-06,
      "step": 76483,
      "training_step_time": 8.707815647125244
    },
    {
      "epoch": 2.368927001953125e-06,
      "model_forward_time": 1.7657313346862793,
      "step": 76484
    },
    {
      "epoch": 2.368927001953125e-06,
      "step": 76484,
      "training_step_time": 7.63983678817749
    },
    {
      "epoch": 2.369880676269531e-06,
      "model_forward_time": 1.7641899585723877,
      "step": 76485
    },
    {
      "epoch": 2.369880676269531e-06,
      "step": 76485,
      "training_step_time": 7.881316184997559
    },
    {
      "epoch": 2.3708343505859377e-06,
      "model_forward_time": 1.7652127742767334,
      "step": 76486
    },
    {
      "epoch": 2.3708343505859377e-06,
      "step": 76486,
      "training_step_time": 7.43259072303772
    },
    {
      "epoch": 2.371788024902344e-06,
      "model_forward_time": 1.7585082054138184,
      "step": 76487
    },
    {
      "epoch": 2.371788024902344e-06,
      "step": 76487,
      "training_step_time": 7.659494400024414
    },
    {
      "epoch": 2.37274169921875e-06,
      "model_forward_time": 1.76096773147583,
      "step": 76488
    },
    {
      "epoch": 2.37274169921875e-06,
      "step": 76488,
      "training_step_time": 7.656757354736328
    },
    {
      "epoch": 2.373695373535156e-06,
      "model_forward_time": 1.7533659934997559,
      "step": 76489
    },
    {
      "epoch": 2.373695373535156e-06,
      "step": 76489,
      "training_step_time": 8.348970174789429
    },
    {
      "epoch": 2.3746490478515627e-06,
      "grad_norm": 0.07781995087862015,
      "learning_rate": 1.43663341560198e-06,
      "loss": 0.0543,
      "step": 76490
    },
    {
      "epoch": 2.3746490478515627e-06,
      "model_forward_time": 2.376840353012085,
      "step": 76490
    },
    {
      "dynamics_loss_avg": 0.05912400912493467,
      "epoch": 2.3746490478515627e-06,
      "step": 76490
    },
    {
      "action_loss_avg": 0.002609576046233997,
      "epoch": 2.3746490478515627e-06,
      "step": 76490
    },
    {
      "epoch": 2.3746490478515627e-06,
      "step": 76490,
      "training_step_time": 7.588852405548096
    },
    {
      "epoch": 2.375602722167969e-06,
      "model_forward_time": 1.7560696601867676,
      "step": 76491
    },
    {
      "epoch": 2.375602722167969e-06,
      "step": 76491,
      "training_step_time": 8.389348983764648
    },
    {
      "epoch": 2.376556396484375e-06,
      "model_forward_time": 1.7472493648529053,
      "step": 76492
    },
    {
      "epoch": 2.376556396484375e-06,
      "step": 76492,
      "training_step_time": 7.765285968780518
    },
    {
      "epoch": 2.377510070800781e-06,
      "model_forward_time": 1.7766969203948975,
      "step": 76493
    },
    {
      "epoch": 2.377510070800781e-06,
      "step": 76493,
      "training_step_time": 8.624781131744385
    },
    {
      "epoch": 2.3784637451171877e-06,
      "model_forward_time": 1.7449142932891846,
      "step": 76494
    },
    {
      "epoch": 2.3784637451171877e-06,
      "step": 76494,
      "training_step_time": 7.154428720474243
    },
    {
      "epoch": 2.379417419433594e-06,
      "model_forward_time": 1.7544662952423096,
      "step": 76495
    },
    {
      "epoch": 2.379417419433594e-06,
      "step": 76495,
      "training_step_time": 7.588946104049683
    },
    {
      "epoch": 2.38037109375e-06,
      "model_forward_time": 1.404167890548706,
      "step": 76496
    },
    {
      "epoch": 2.38037109375e-06,
      "step": 76496,
      "training_step_time": 7.59980320930481
    },
    {
      "epoch": 2.381324768066406e-06,
      "model_forward_time": 1.770575761795044,
      "step": 76497
    },
    {
      "epoch": 2.381324768066406e-06,
      "step": 76497,
      "training_step_time": 8.118571281433105
    },
    {
      "epoch": 2.3822784423828127e-06,
      "model_forward_time": 1.7662630081176758,
      "step": 76498
    },
    {
      "epoch": 2.3822784423828127e-06,
      "step": 76498,
      "training_step_time": 8.389195442199707
    },
    {
      "epoch": 2.383232116699219e-06,
      "model_forward_time": 2.4240469932556152,
      "step": 76499
    },
    {
      "epoch": 2.383232116699219e-06,
      "step": 76499,
      "training_step_time": 7.54873251914978
    },
    {
      "epoch": 2.384185791015625e-06,
      "grad_norm": 0.07777248322963715,
      "learning_rate": 1.435473707877898e-06,
      "loss": 0.0558,
      "step": 76500
    },
    {
      "epoch": 2.384185791015625e-06,
      "model_forward_time": 1.7597875595092773,
      "step": 76500
    },
    {
      "dynamics_loss_avg": 0.04364988603629172,
      "epoch": 2.384185791015625e-06,
      "step": 76500
    },
    {
      "action_loss_avg": 0.0011609017266891897,
      "epoch": 2.384185791015625e-06,
      "step": 76500
    },
    {
      "epoch": 2.384185791015625e-06,
      "step": 76500,
      "training_step_time": 8.578372240066528
    },
    {
      "epoch": 2.385139465332031e-06,
      "model_forward_time": 1.8303649425506592,
      "step": 76501
    },
    {
      "epoch": 2.385139465332031e-06,
      "step": 76501,
      "training_step_time": 7.637309789657593
    },
    {
      "epoch": 2.3860931396484373e-06,
      "model_forward_time": 1.7816531658172607,
      "step": 76502
    },
    {
      "epoch": 2.3860931396484373e-06,
      "step": 76502,
      "training_step_time": 7.62414026260376
    },
    {
      "epoch": 2.387046813964844e-06,
      "model_forward_time": 1.7744956016540527,
      "step": 76503
    },
    {
      "epoch": 2.387046813964844e-06,
      "step": 76503,
      "training_step_time": 8.392600536346436
    },
    {
      "epoch": 2.38800048828125e-06,
      "model_forward_time": 1.7760138511657715,
      "step": 76504
    },
    {
      "epoch": 2.38800048828125e-06,
      "step": 76504,
      "training_step_time": 7.607206344604492
    },
    {
      "epoch": 2.388954162597656e-06,
      "model_forward_time": 1.7766599655151367,
      "step": 76505
    },
    {
      "epoch": 2.388954162597656e-06,
      "step": 76505,
      "training_step_time": 7.683237314224243
    },
    {
      "epoch": 2.3899078369140623e-06,
      "model_forward_time": 1.7717235088348389,
      "step": 76506
    },
    {
      "epoch": 2.3899078369140623e-06,
      "step": 76506,
      "training_step_time": 7.516604900360107
    },
    {
      "epoch": 2.390861511230469e-06,
      "model_forward_time": 1.778857946395874,
      "step": 76507
    },
    {
      "epoch": 2.390861511230469e-06,
      "step": 76507,
      "training_step_time": 8.757089376449585
    },
    {
      "epoch": 2.391815185546875e-06,
      "model_forward_time": 2.4199061393737793,
      "step": 76508
    },
    {
      "epoch": 2.391815185546875e-06,
      "step": 76508,
      "training_step_time": 8.456891775131226
    },
    {
      "epoch": 2.392768859863281e-06,
      "model_forward_time": 1.7713704109191895,
      "step": 76509
    },
    {
      "epoch": 2.392768859863281e-06,
      "step": 76509,
      "training_step_time": 8.848693132400513
    },
    {
      "epoch": 2.3937225341796873e-06,
      "grad_norm": 0.09853758662939072,
      "learning_rate": 1.4343143899650509e-06,
      "loss": 0.0569,
      "step": 76510
    },
    {
      "epoch": 2.3937225341796873e-06,
      "model_forward_time": 1.8313097953796387,
      "step": 76510
    },
    {
      "dynamics_loss_avg": 0.046832555532455446,
      "epoch": 2.3937225341796873e-06,
      "step": 76510
    },
    {
      "action_loss_avg": 0.0014762435166630894,
      "epoch": 2.3937225341796873e-06,
      "step": 76510
    },
    {
      "epoch": 2.3937225341796873e-06,
      "step": 76510,
      "training_step_time": 7.5557661056518555
    },
    {
      "epoch": 2.394676208496094e-06,
      "model_forward_time": 1.7713935375213623,
      "step": 76511
    },
    {
      "epoch": 2.394676208496094e-06,
      "step": 76511,
      "training_step_time": 8.591842651367188
    },
    {
      "epoch": 2.3956298828125e-06,
      "model_forward_time": 1.7649881839752197,
      "step": 76512
    },
    {
      "epoch": 2.3956298828125e-06,
      "step": 76512,
      "training_step_time": 7.599665403366089
    },
    {
      "epoch": 2.396583557128906e-06,
      "model_forward_time": 1.753739356994629,
      "step": 76513
    },
    {
      "epoch": 2.396583557128906e-06,
      "step": 76513,
      "training_step_time": 7.647106170654297
    },
    {
      "epoch": 2.3975372314453123e-06,
      "model_forward_time": 1.7670583724975586,
      "step": 76514
    },
    {
      "epoch": 2.3975372314453123e-06,
      "step": 76514,
      "training_step_time": 7.58180046081543
    },
    {
      "epoch": 2.398490905761719e-06,
      "model_forward_time": 1.7894039154052734,
      "step": 76515
    },
    {
      "epoch": 2.398490905761719e-06,
      "step": 76515,
      "training_step_time": 8.795215606689453
    },
    {
      "epoch": 2.399444580078125e-06,
      "model_forward_time": 1.7567617893218994,
      "step": 76516
    },
    {
      "epoch": 2.399444580078125e-06,
      "step": 76516,
      "training_step_time": 8.343065977096558
    },
    {
      "epoch": 2.400398254394531e-06,
      "model_forward_time": 1.7777769565582275,
      "step": 76517
    },
    {
      "epoch": 2.400398254394531e-06,
      "step": 76517,
      "training_step_time": 9.001823663711548
    },
    {
      "epoch": 2.4013519287109373e-06,
      "model_forward_time": 1.7593698501586914,
      "step": 76518
    },
    {
      "epoch": 2.4013519287109373e-06,
      "step": 76518,
      "training_step_time": 8.393851280212402
    },
    {
      "epoch": 2.402305603027344e-06,
      "model_forward_time": 1.8024132251739502,
      "step": 76519
    },
    {
      "epoch": 2.402305603027344e-06,
      "step": 76519,
      "training_step_time": 9.085707426071167
    },
    {
      "epoch": 2.40325927734375e-06,
      "grad_norm": 0.09757578372955322,
      "learning_rate": 1.4331554619902205e-06,
      "loss": 0.0565,
      "step": 76520
    },
    {
      "epoch": 2.40325927734375e-06,
      "model_forward_time": 1.7635087966918945,
      "step": 76520
    },
    {
      "dynamics_loss_avg": 0.07666178755462169,
      "epoch": 2.40325927734375e-06,
      "step": 76520
    },
    {
      "action_loss_avg": 0.0019153186818584799,
      "epoch": 2.40325927734375e-06,
      "step": 76520
    },
    {
      "epoch": 2.40325927734375e-06,
      "step": 76520,
      "training_step_time": 7.819947242736816
    },
    {
      "epoch": 2.404212951660156e-06,
      "model_forward_time": 1.7668561935424805,
      "step": 76521
    },
    {
      "epoch": 2.404212951660156e-06,
      "step": 76521,
      "training_step_time": 7.53320050239563
    },
    {
      "epoch": 2.4051666259765623e-06,
      "model_forward_time": 1.757493495941162,
      "step": 76522
    },
    {
      "epoch": 2.4051666259765623e-06,
      "step": 76522,
      "training_step_time": 7.600449562072754
    },
    {
      "epoch": 2.406120300292969e-06,
      "model_forward_time": 1.7652599811553955,
      "step": 76523
    },
    {
      "epoch": 2.406120300292969e-06,
      "step": 76523,
      "training_step_time": 7.612451791763306
    },
    {
      "epoch": 2.407073974609375e-06,
      "model_forward_time": 1.7614364624023438,
      "step": 76524
    },
    {
      "epoch": 2.407073974609375e-06,
      "step": 76524,
      "training_step_time": 7.394442319869995
    },
    {
      "epoch": 2.408027648925781e-06,
      "model_forward_time": 1.7610046863555908,
      "step": 76525
    },
    {
      "epoch": 2.408027648925781e-06,
      "step": 76525,
      "training_step_time": 7.574363946914673
    },
    {
      "epoch": 2.4089813232421873e-06,
      "model_forward_time": 1.7803332805633545,
      "step": 76526
    },
    {
      "epoch": 2.4089813232421873e-06,
      "step": 76526,
      "training_step_time": 8.677258014678955
    },
    {
      "epoch": 2.409934997558594e-06,
      "model_forward_time": 2.392929792404175,
      "step": 76527
    },
    {
      "epoch": 2.409934997558594e-06,
      "step": 76527,
      "training_step_time": 9.45888614654541
    },
    {
      "epoch": 2.410888671875e-06,
      "model_forward_time": 1.763443946838379,
      "step": 76528
    },
    {
      "epoch": 2.410888671875e-06,
      "step": 76528,
      "training_step_time": 9.207736730575562
    },
    {
      "epoch": 2.411842346191406e-06,
      "model_forward_time": 1.7468886375427246,
      "step": 76529
    },
    {
      "epoch": 2.411842346191406e-06,
      "step": 76529,
      "training_step_time": 8.269794702529907
    },
    {
      "epoch": 2.4127960205078123e-06,
      "grad_norm": 0.10200507938861847,
      "learning_rate": 1.4319969240801485e-06,
      "loss": 0.0552,
      "step": 76530
    },
    {
      "epoch": 2.4127960205078123e-06,
      "model_forward_time": 1.7415096759796143,
      "step": 76530
    },
    {
      "dynamics_loss_avg": 0.058766202069818974,
      "epoch": 2.4127960205078123e-06,
      "step": 76530
    },
    {
      "action_loss_avg": 0.0014686746639199556,
      "epoch": 2.4127960205078123e-06,
      "step": 76530
    },
    {
      "epoch": 2.4127960205078123e-06,
      "step": 76530,
      "training_step_time": 7.603378534317017
    },
    {
      "epoch": 2.413749694824219e-06,
      "model_forward_time": 1.7515439987182617,
      "step": 76531
    },
    {
      "epoch": 2.413749694824219e-06,
      "step": 76531,
      "training_step_time": 7.562809705734253
    },
    {
      "epoch": 2.414703369140625e-06,
      "model_forward_time": 1.7810583114624023,
      "step": 76532
    },
    {
      "epoch": 2.414703369140625e-06,
      "step": 76532,
      "training_step_time": 7.781387567520142
    },
    {
      "epoch": 2.415657043457031e-06,
      "model_forward_time": 1.389683485031128,
      "step": 76533
    },
    {
      "epoch": 2.415657043457031e-06,
      "step": 76533,
      "training_step_time": 8.192246913909912
    },
    {
      "epoch": 2.4166107177734373e-06,
      "model_forward_time": 1.7549231052398682,
      "step": 76534
    },
    {
      "epoch": 2.4166107177734373e-06,
      "step": 76534,
      "training_step_time": 7.48970627784729
    },
    {
      "epoch": 2.417564392089844e-06,
      "model_forward_time": 1.7515158653259277,
      "step": 76535
    },
    {
      "epoch": 2.417564392089844e-06,
      "step": 76535,
      "training_step_time": 7.90958571434021
    },
    {
      "epoch": 2.41851806640625e-06,
      "model_forward_time": 2.377040386199951,
      "step": 76536
    },
    {
      "epoch": 2.41851806640625e-06,
      "step": 76536,
      "training_step_time": 7.691282510757446
    },
    {
      "epoch": 2.419471740722656e-06,
      "model_forward_time": 1.7687709331512451,
      "step": 76537
    },
    {
      "epoch": 2.419471740722656e-06,
      "step": 76537,
      "training_step_time": 8.869091033935547
    },
    {
      "epoch": 2.4204254150390623e-06,
      "model_forward_time": 1.8218603134155273,
      "step": 76538
    },
    {
      "epoch": 2.4204254150390623e-06,
      "step": 76538,
      "training_step_time": 7.645098447799683
    },
    {
      "epoch": 2.421379089355469e-06,
      "model_forward_time": 1.7519519329071045,
      "step": 76539
    },
    {
      "epoch": 2.421379089355469e-06,
      "step": 76539,
      "training_step_time": 7.677922248840332
    },
    {
      "epoch": 2.422332763671875e-06,
      "grad_norm": 0.13802781701087952,
      "learning_rate": 1.4308387763615261e-06,
      "loss": 0.0559,
      "step": 76540
    },
    {
      "epoch": 2.422332763671875e-06,
      "model_forward_time": 1.7538566589355469,
      "step": 76540
    },
    {
      "dynamics_loss_avg": 0.059663832746446134,
      "epoch": 2.422332763671875e-06,
      "step": 76540
    },
    {
      "action_loss_avg": 0.0014402829547179863,
      "epoch": 2.422332763671875e-06,
      "step": 76540
    },
    {
      "epoch": 2.422332763671875e-06,
      "step": 76540,
      "training_step_time": 7.513043403625488
    },
    {
      "epoch": 2.423286437988281e-06,
      "model_forward_time": 1.7495086193084717,
      "step": 76541
    },
    {
      "epoch": 2.423286437988281e-06,
      "step": 76541,
      "training_step_time": 7.613582372665405
    },
    {
      "epoch": 2.4242401123046873e-06,
      "model_forward_time": 1.7481722831726074,
      "step": 76542
    },
    {
      "epoch": 2.4242401123046873e-06,
      "step": 76542,
      "training_step_time": 7.141744136810303
    },
    {
      "epoch": 2.425193786621094e-06,
      "model_forward_time": 1.7437465190887451,
      "step": 76543
    },
    {
      "epoch": 2.425193786621094e-06,
      "step": 76543,
      "training_step_time": 8.124070167541504
    },
    {
      "epoch": 2.4261474609375e-06,
      "model_forward_time": 1.7418718338012695,
      "step": 76544
    },
    {
      "epoch": 2.4261474609375e-06,
      "step": 76544,
      "training_step_time": 8.37674593925476
    },
    {
      "epoch": 2.427101135253906e-06,
      "model_forward_time": 2.375753879547119,
      "step": 76545
    },
    {
      "epoch": 2.427101135253906e-06,
      "step": 76545,
      "training_step_time": 8.45395827293396
    },
    {
      "epoch": 2.4280548095703123e-06,
      "model_forward_time": 1.732483148574829,
      "step": 76546
    },
    {
      "epoch": 2.4280548095703123e-06,
      "step": 76546,
      "training_step_time": 7.465027332305908
    },
    {
      "epoch": 2.429008483886719e-06,
      "model_forward_time": 1.8191490173339844,
      "step": 76547
    },
    {
      "epoch": 2.429008483886719e-06,
      "step": 76547,
      "training_step_time": 9.586732864379883
    },
    {
      "epoch": 2.429962158203125e-06,
      "model_forward_time": 1.7561736106872559,
      "step": 76548
    },
    {
      "epoch": 2.429962158203125e-06,
      "step": 76548,
      "training_step_time": 8.647414445877075
    },
    {
      "epoch": 2.430915832519531e-06,
      "model_forward_time": 1.7514891624450684,
      "step": 76549
    },
    {
      "epoch": 2.430915832519531e-06,
      "step": 76549,
      "training_step_time": 7.576324224472046
    },
    {
      "epoch": 2.4318695068359373e-06,
      "grad_norm": 0.0940469354391098,
      "learning_rate": 1.4296810189610095e-06,
      "loss": 0.0537,
      "step": 76550
    },
    {
      "epoch": 2.4318695068359373e-06,
      "model_forward_time": 1.7496094703674316,
      "step": 76550
    },
    {
      "dynamics_loss_avg": 0.05904540680348873,
      "epoch": 2.4318695068359373e-06,
      "step": 76550
    },
    {
      "action_loss_avg": 0.002647293795598671,
      "epoch": 2.4318695068359373e-06,
      "step": 76550
    },
    {
      "epoch": 2.4318695068359373e-06,
      "step": 76550,
      "training_step_time": 9.393888473510742
    },
    {
      "epoch": 2.432823181152344e-06,
      "model_forward_time": 1.7653858661651611,
      "step": 76551
    },
    {
      "epoch": 2.432823181152344e-06,
      "step": 76551,
      "training_step_time": 7.133838415145874
    },
    {
      "epoch": 2.43377685546875e-06,
      "model_forward_time": 1.7574095726013184,
      "step": 76552
    },
    {
      "epoch": 2.43377685546875e-06,
      "step": 76552,
      "training_step_time": 7.534096956253052
    },
    {
      "epoch": 2.434730529785156e-06,
      "model_forward_time": 1.7464356422424316,
      "step": 76553
    },
    {
      "epoch": 2.434730529785156e-06,
      "step": 76553,
      "training_step_time": 8.3234703540802
    },
    {
      "epoch": 2.4356842041015623e-06,
      "model_forward_time": 1.751988410949707,
      "step": 76554
    },
    {
      "epoch": 2.4356842041015623e-06,
      "step": 76554,
      "training_step_time": 8.797590017318726
    },
    {
      "epoch": 2.436637878417969e-06,
      "model_forward_time": 1.7506020069122314,
      "step": 76555
    },
    {
      "epoch": 2.436637878417969e-06,
      "step": 76555,
      "training_step_time": 8.372504472732544
    },
    {
      "epoch": 2.437591552734375e-06,
      "model_forward_time": 1.7974185943603516,
      "step": 76556
    },
    {
      "epoch": 2.437591552734375e-06,
      "step": 76556,
      "training_step_time": 9.08034610748291
    },
    {
      "epoch": 2.438545227050781e-06,
      "model_forward_time": 1.7490949630737305,
      "step": 76557
    },
    {
      "epoch": 2.438545227050781e-06,
      "step": 76557,
      "training_step_time": 8.551733016967773
    },
    {
      "epoch": 2.4394989013671873e-06,
      "model_forward_time": 1.7535827159881592,
      "step": 76558
    },
    {
      "epoch": 2.4394989013671873e-06,
      "step": 76558,
      "training_step_time": 7.510535955429077
    },
    {
      "epoch": 2.440452575683594e-06,
      "model_forward_time": 1.72855544090271,
      "step": 76559
    },
    {
      "epoch": 2.440452575683594e-06,
      "step": 76559,
      "training_step_time": 8.968122005462646
    },
    {
      "epoch": 2.44140625e-06,
      "grad_norm": 0.11353211849927902,
      "learning_rate": 1.4285236520052086e-06,
      "loss": 0.0572,
      "step": 76560
    },
    {
      "epoch": 2.44140625e-06,
      "model_forward_time": 1.7536628246307373,
      "step": 76560
    },
    {
      "dynamics_loss_avg": 0.054851151444017886,
      "epoch": 2.44140625e-06,
      "step": 76560
    },
    {
      "action_loss_avg": 0.0019501986855175347,
      "epoch": 2.44140625e-06,
      "step": 76560
    },
    {
      "epoch": 2.44140625e-06,
      "step": 76560,
      "training_step_time": 7.590195894241333
    },
    {
      "epoch": 2.442359924316406e-06,
      "model_forward_time": 1.392801284790039,
      "step": 76561
    },
    {
      "epoch": 2.442359924316406e-06,
      "step": 76561,
      "training_step_time": 7.963951110839844
    },
    {
      "epoch": 2.4433135986328123e-06,
      "model_forward_time": 1.7575151920318604,
      "step": 76562
    },
    {
      "epoch": 2.4433135986328123e-06,
      "step": 76562,
      "training_step_time": 9.503504991531372
    },
    {
      "epoch": 2.444267272949219e-06,
      "model_forward_time": 1.7660531997680664,
      "step": 76563
    },
    {
      "epoch": 2.444267272949219e-06,
      "step": 76563,
      "training_step_time": 8.828604936599731
    },
    {
      "epoch": 2.445220947265625e-06,
      "model_forward_time": 2.4130117893218994,
      "step": 76564
    },
    {
      "epoch": 2.445220947265625e-06,
      "step": 76564,
      "training_step_time": 7.515619516372681
    },
    {
      "epoch": 2.446174621582031e-06,
      "model_forward_time": 1.7845592498779297,
      "step": 76565
    },
    {
      "epoch": 2.446174621582031e-06,
      "step": 76565,
      "training_step_time": 8.322875499725342
    },
    {
      "epoch": 2.4471282958984373e-06,
      "model_forward_time": 1.7573156356811523,
      "step": 76566
    },
    {
      "epoch": 2.4471282958984373e-06,
      "step": 76566,
      "training_step_time": 9.183456659317017
    },
    {
      "epoch": 2.448081970214844e-06,
      "model_forward_time": 1.7554950714111328,
      "step": 76567
    },
    {
      "epoch": 2.448081970214844e-06,
      "step": 76567,
      "training_step_time": 7.509100914001465
    },
    {
      "epoch": 2.44903564453125e-06,
      "model_forward_time": 1.763688325881958,
      "step": 76568
    },
    {
      "epoch": 2.44903564453125e-06,
      "step": 76568,
      "training_step_time": 7.55932879447937
    },
    {
      "epoch": 2.449989318847656e-06,
      "model_forward_time": 1.7541139125823975,
      "step": 76569
    },
    {
      "epoch": 2.449989318847656e-06,
      "step": 76569,
      "training_step_time": 7.5831685066223145
    },
    {
      "epoch": 2.4509429931640623e-06,
      "grad_norm": 0.1155330166220665,
      "learning_rate": 1.4273666756206933e-06,
      "loss": 0.0556,
      "step": 76570
    },
    {
      "epoch": 2.4509429931640623e-06,
      "model_forward_time": 1.7514545917510986,
      "step": 76570
    },
    {
      "dynamics_loss_avg": 0.05527336932718754,
      "epoch": 2.4509429931640623e-06,
      "step": 76570
    },
    {
      "action_loss_avg": 0.001128377292479854,
      "epoch": 2.4509429931640623e-06,
      "step": 76570
    },
    {
      "epoch": 2.4509429931640623e-06,
      "step": 76570,
      "training_step_time": 7.354939699172974
    },
    {
      "epoch": 2.451896667480469e-06,
      "model_forward_time": 1.7623751163482666,
      "step": 76571
    },
    {
      "epoch": 2.451896667480469e-06,
      "step": 76571,
      "training_step_time": 8.593210697174072
    },
    {
      "epoch": 2.452850341796875e-06,
      "model_forward_time": 1.7493863105773926,
      "step": 76572
    },
    {
      "epoch": 2.452850341796875e-06,
      "step": 76572,
      "training_step_time": 8.613834381103516
    },
    {
      "epoch": 2.453804016113281e-06,
      "model_forward_time": 2.4048566818237305,
      "step": 76573
    },
    {
      "epoch": 2.453804016113281e-06,
      "step": 76573,
      "training_step_time": 8.5202054977417
    },
    {
      "epoch": 2.4547576904296874e-06,
      "model_forward_time": 1.404644250869751,
      "step": 76574
    },
    {
      "epoch": 2.4547576904296874e-06,
      "step": 76574,
      "training_step_time": 10.118735551834106
    },
    {
      "epoch": 2.455711364746094e-06,
      "model_forward_time": 1.7584130764007568,
      "step": 76575
    },
    {
      "epoch": 2.455711364746094e-06,
      "step": 76575,
      "training_step_time": 8.585670232772827
    },
    {
      "epoch": 2.4566650390625e-06,
      "model_forward_time": 1.7565724849700928,
      "step": 76576
    },
    {
      "epoch": 2.4566650390625e-06,
      "step": 76576,
      "training_step_time": 7.475185394287109
    },
    {
      "epoch": 2.457618713378906e-06,
      "model_forward_time": 1.7618207931518555,
      "step": 76577
    },
    {
      "epoch": 2.457618713378906e-06,
      "step": 76577,
      "training_step_time": 7.813781023025513
    },
    {
      "epoch": 2.4585723876953124e-06,
      "model_forward_time": 1.7532885074615479,
      "step": 76578
    },
    {
      "epoch": 2.4585723876953124e-06,
      "step": 76578,
      "training_step_time": 7.630810976028442
    },
    {
      "epoch": 2.459526062011719e-06,
      "model_forward_time": 1.7602605819702148,
      "step": 76579
    },
    {
      "epoch": 2.459526062011719e-06,
      "step": 76579,
      "training_step_time": 7.491466045379639
    },
    {
      "epoch": 2.460479736328125e-06,
      "grad_norm": 0.10944806039333344,
      "learning_rate": 1.4262100899339848e-06,
      "loss": 0.0542,
      "step": 76580
    },
    {
      "epoch": 2.460479736328125e-06,
      "model_forward_time": 1.7558367252349854,
      "step": 76580
    },
    {
      "dynamics_loss_avg": 0.06600573789328337,
      "epoch": 2.460479736328125e-06,
      "step": 76580
    },
    {
      "action_loss_avg": 0.0011760843946831301,
      "epoch": 2.460479736328125e-06,
      "step": 76580
    },
    {
      "epoch": 2.460479736328125e-06,
      "step": 76580,
      "training_step_time": 8.26331639289856
    },
    {
      "epoch": 2.4614334106445312e-06,
      "model_forward_time": 1.762432336807251,
      "step": 76581
    },
    {
      "epoch": 2.4614334106445312e-06,
      "step": 76581,
      "training_step_time": 7.806850433349609
    },
    {
      "epoch": 2.4623870849609374e-06,
      "model_forward_time": 2.3977911472320557,
      "step": 76582
    },
    {
      "epoch": 2.4623870849609374e-06,
      "step": 76582,
      "training_step_time": 8.337172746658325
    },
    {
      "epoch": 2.463340759277344e-06,
      "model_forward_time": 1.7537238597869873,
      "step": 76583
    },
    {
      "epoch": 2.463340759277344e-06,
      "step": 76583,
      "training_step_time": 8.322264671325684
    },
    {
      "epoch": 2.46429443359375e-06,
      "model_forward_time": 1.825011968612671,
      "step": 76584
    },
    {
      "epoch": 2.46429443359375e-06,
      "step": 76584,
      "training_step_time": 8.262289047241211
    },
    {
      "epoch": 2.4652481079101562e-06,
      "model_forward_time": 1.771939992904663,
      "step": 76585
    },
    {
      "epoch": 2.4652481079101562e-06,
      "step": 76585,
      "training_step_time": 7.529443025588989
    },
    {
      "epoch": 2.4662017822265624e-06,
      "model_forward_time": 1.768951654434204,
      "step": 76586
    },
    {
      "epoch": 2.4662017822265624e-06,
      "step": 76586,
      "training_step_time": 8.2089684009552
    },
    {
      "epoch": 2.467155456542969e-06,
      "model_forward_time": 1.7638473510742188,
      "step": 76587
    },
    {
      "epoch": 2.467155456542969e-06,
      "step": 76587,
      "training_step_time": 7.63307785987854
    },
    {
      "epoch": 2.468109130859375e-06,
      "model_forward_time": 1.7662785053253174,
      "step": 76588
    },
    {
      "epoch": 2.468109130859375e-06,
      "step": 76588,
      "training_step_time": 8.286170959472656
    },
    {
      "epoch": 2.4690628051757812e-06,
      "model_forward_time": 1.7525906562805176,
      "step": 76589
    },
    {
      "epoch": 2.4690628051757812e-06,
      "step": 76589,
      "training_step_time": 7.549875736236572
    },
    {
      "epoch": 2.4700164794921874e-06,
      "grad_norm": 0.1012912467122078,
      "learning_rate": 1.4250538950715697e-06,
      "loss": 0.0543,
      "step": 76590
    },
    {
      "epoch": 2.4700164794921874e-06,
      "model_forward_time": 1.7459580898284912,
      "step": 76590
    },
    {
      "dynamics_loss_avg": 0.0638681024312973,
      "epoch": 2.4700164794921874e-06,
      "step": 76590
    },
    {
      "action_loss_avg": 0.001302547607338056,
      "epoch": 2.4700164794921874e-06,
      "step": 76590
    },
    {
      "epoch": 2.4700164794921874e-06,
      "step": 76590,
      "training_step_time": 8.235479831695557
    },
    {
      "epoch": 2.470970153808594e-06,
      "model_forward_time": 2.341550588607788,
      "step": 76591
    },
    {
      "epoch": 2.470970153808594e-06,
      "step": 76591,
      "training_step_time": 7.5426836013793945
    },
    {
      "epoch": 2.471923828125e-06,
      "model_forward_time": 1.7733440399169922,
      "step": 76592
    },
    {
      "epoch": 2.471923828125e-06,
      "step": 76592,
      "training_step_time": 8.328081607818604
    },
    {
      "epoch": 2.4728775024414062e-06,
      "model_forward_time": 1.7563259601593018,
      "step": 76593
    },
    {
      "epoch": 2.4728775024414062e-06,
      "step": 76593,
      "training_step_time": 7.541676759719849
    },
    {
      "epoch": 2.4738311767578124e-06,
      "model_forward_time": 1.7434351444244385,
      "step": 76594
    },
    {
      "epoch": 2.4738311767578124e-06,
      "step": 76594,
      "training_step_time": 9.42289686203003
    },
    {
      "epoch": 2.474784851074219e-06,
      "model_forward_time": 1.7451510429382324,
      "step": 76595
    },
    {
      "epoch": 2.474784851074219e-06,
      "step": 76595,
      "training_step_time": 8.363391637802124
    },
    {
      "epoch": 2.475738525390625e-06,
      "model_forward_time": 1.7454767227172852,
      "step": 76596
    },
    {
      "epoch": 2.475738525390625e-06,
      "step": 76596,
      "training_step_time": 7.612134695053101
    },
    {
      "epoch": 2.4766921997070312e-06,
      "model_forward_time": 1.741011142730713,
      "step": 76597
    },
    {
      "epoch": 2.4766921997070312e-06,
      "step": 76597,
      "training_step_time": 7.369657039642334
    },
    {
      "epoch": 2.4776458740234374e-06,
      "model_forward_time": 1.7471935749053955,
      "step": 76598
    },
    {
      "epoch": 2.4776458740234374e-06,
      "step": 76598,
      "training_step_time": 7.617896795272827
    },
    {
      "epoch": 2.478599548339844e-06,
      "model_forward_time": 1.7425768375396729,
      "step": 76599
    },
    {
      "epoch": 2.478599548339844e-06,
      "step": 76599,
      "training_step_time": 7.566067934036255
    },
    {
      "epoch": 2.47955322265625e-06,
      "grad_norm": 0.10115236788988113,
      "learning_rate": 1.4238980911598842e-06,
      "loss": 0.0539,
      "step": 76600
    },
    {
      "epoch": 2.47955322265625e-06,
      "model_forward_time": 1.745182991027832,
      "step": 76600
    },
    {
      "dynamics_loss_avg": 0.05194485615938902,
      "epoch": 2.47955322265625e-06,
      "step": 76600
    },
    {
      "action_loss_avg": 0.0011830257280962542,
      "epoch": 2.47955322265625e-06,
      "step": 76600
    },
    {
      "epoch": 2.47955322265625e-06,
      "step": 76600,
      "training_step_time": 8.325439929962158
    },
    {
      "epoch": 2.4805068969726562e-06,
      "model_forward_time": 2.409201145172119,
      "step": 76601
    },
    {
      "epoch": 2.4805068969726562e-06,
      "step": 76601,
      "training_step_time": 7.788378477096558
    },
    {
      "epoch": 2.4814605712890624e-06,
      "model_forward_time": 1.7649483680725098,
      "step": 76602
    },
    {
      "epoch": 2.4814605712890624e-06,
      "step": 76602,
      "training_step_time": 8.194669008255005
    },
    {
      "epoch": 2.482414245605469e-06,
      "model_forward_time": 1.7459752559661865,
      "step": 76603
    },
    {
      "epoch": 2.482414245605469e-06,
      "step": 76603,
      "training_step_time": 8.342084407806396
    },
    {
      "epoch": 2.483367919921875e-06,
      "model_forward_time": 1.727919101715088,
      "step": 76604
    },
    {
      "epoch": 2.483367919921875e-06,
      "step": 76604,
      "training_step_time": 7.684751749038696
    },
    {
      "epoch": 2.4843215942382812e-06,
      "model_forward_time": 1.7343716621398926,
      "step": 76605
    },
    {
      "epoch": 2.4843215942382812e-06,
      "step": 76605,
      "training_step_time": 9.299520254135132
    },
    {
      "epoch": 2.4852752685546874e-06,
      "model_forward_time": 1.7223639488220215,
      "step": 76606
    },
    {
      "epoch": 2.4852752685546874e-06,
      "step": 76606,
      "training_step_time": 7.585351228713989
    },
    {
      "epoch": 2.486228942871094e-06,
      "model_forward_time": 1.7296257019042969,
      "step": 76607
    },
    {
      "epoch": 2.486228942871094e-06,
      "step": 76607,
      "training_step_time": 8.778592109680176
    },
    {
      "epoch": 2.4871826171875e-06,
      "model_forward_time": 1.3615057468414307,
      "step": 76608
    },
    {
      "epoch": 2.4871826171875e-06,
      "step": 76608,
      "training_step_time": 7.575257062911987
    },
    {
      "epoch": 2.4881362915039062e-06,
      "model_forward_time": 1.7305655479431152,
      "step": 76609
    },
    {
      "epoch": 2.4881362915039062e-06,
      "step": 76609,
      "training_step_time": 7.952389478683472
    },
    {
      "epoch": 2.4890899658203124e-06,
      "grad_norm": 0.12309082597494125,
      "learning_rate": 1.4227426783253267e-06,
      "loss": 0.054,
      "step": 76610
    },
    {
      "epoch": 2.4890899658203124e-06,
      "model_forward_time": 2.409252882003784,
      "step": 76610
    },
    {
      "dynamics_loss_avg": 0.06328333169221878,
      "epoch": 2.4890899658203124e-06,
      "step": 76610
    },
    {
      "action_loss_avg": 0.001449952260009013,
      "epoch": 2.4890899658203124e-06,
      "step": 76610
    },
    {
      "epoch": 2.4890899658203124e-06,
      "step": 76610,
      "training_step_time": 8.422286987304688
    },
    {
      "epoch": 2.490043640136719e-06,
      "model_forward_time": 1.7452819347381592,
      "step": 76611
    },
    {
      "epoch": 2.490043640136719e-06,
      "step": 76611,
      "training_step_time": 10.252972602844238
    },
    {
      "epoch": 2.490997314453125e-06,
      "model_forward_time": 1.7334089279174805,
      "step": 76612
    },
    {
      "epoch": 2.490997314453125e-06,
      "step": 76612,
      "training_step_time": 8.431768894195557
    },
    {
      "epoch": 2.4919509887695312e-06,
      "model_forward_time": 1.3666713237762451,
      "step": 76613
    },
    {
      "epoch": 2.4919509887695312e-06,
      "step": 76613,
      "training_step_time": 7.741286516189575
    },
    {
      "epoch": 2.4929046630859374e-06,
      "model_forward_time": 1.7244200706481934,
      "step": 76614
    },
    {
      "epoch": 2.4929046630859374e-06,
      "step": 76614,
      "training_step_time": 8.184009313583374
    },
    {
      "epoch": 2.493858337402344e-06,
      "model_forward_time": 1.7323729991912842,
      "step": 76615
    },
    {
      "epoch": 2.493858337402344e-06,
      "step": 76615,
      "training_step_time": 7.8139448165893555
    },
    {
      "epoch": 2.49481201171875e-06,
      "model_forward_time": 1.7314658164978027,
      "step": 76616
    },
    {
      "epoch": 2.49481201171875e-06,
      "step": 76616,
      "training_step_time": 7.73794150352478
    },
    {
      "epoch": 2.4957656860351562e-06,
      "model_forward_time": 1.7276511192321777,
      "step": 76617
    },
    {
      "epoch": 2.4957656860351562e-06,
      "step": 76617,
      "training_step_time": 7.781540155410767
    },
    {
      "epoch": 2.4967193603515624e-06,
      "model_forward_time": 1.7311146259307861,
      "step": 76618
    },
    {
      "epoch": 2.4967193603515624e-06,
      "step": 76618,
      "training_step_time": 8.076258182525635
    },
    {
      "epoch": 2.497673034667969e-06,
      "model_forward_time": 2.4059271812438965,
      "step": 76619
    },
    {
      "epoch": 2.497673034667969e-06,
      "step": 76619,
      "training_step_time": 7.824313640594482
    },
    {
      "epoch": 2.498626708984375e-06,
      "grad_norm": 0.08890226483345032,
      "learning_rate": 1.4215876566942521e-06,
      "loss": 0.0531,
      "step": 76620
    },
    {
      "epoch": 2.498626708984375e-06,
      "model_forward_time": 1.7453646659851074,
      "step": 76620
    },
    {
      "dynamics_loss_avg": 0.05768363848328591,
      "epoch": 2.498626708984375e-06,
      "step": 76620
    },
    {
      "action_loss_avg": 0.00109854098118376,
      "epoch": 2.498626708984375e-06,
      "step": 76620
    },
    {
      "epoch": 2.498626708984375e-06,
      "step": 76620,
      "training_step_time": 7.796310186386108
    },
    {
      "epoch": 2.4995803833007812e-06,
      "model_forward_time": 1.7266266345977783,
      "step": 76621
    },
    {
      "epoch": 2.4995803833007812e-06,
      "step": 76621,
      "training_step_time": 7.823326587677002
    },
    {
      "epoch": 2.5005340576171874e-06,
      "model_forward_time": 1.7374169826507568,
      "step": 76622
    },
    {
      "epoch": 2.5005340576171874e-06,
      "step": 76622,
      "training_step_time": 7.764330148696899
    },
    {
      "epoch": 2.501487731933594e-06,
      "model_forward_time": 1.7282533645629883,
      "step": 76623
    },
    {
      "epoch": 2.501487731933594e-06,
      "step": 76623,
      "training_step_time": 7.776201486587524
    },
    {
      "epoch": 2.50244140625e-06,
      "model_forward_time": 1.7337143421173096,
      "step": 76624
    },
    {
      "epoch": 2.50244140625e-06,
      "step": 76624,
      "training_step_time": 8.490121364593506
    },
    {
      "epoch": 2.5033950805664062e-06,
      "model_forward_time": 1.7360811233520508,
      "step": 76625
    },
    {
      "epoch": 2.5033950805664062e-06,
      "step": 76625,
      "training_step_time": 7.762599945068359
    },
    {
      "epoch": 2.5043487548828124e-06,
      "model_forward_time": 1.7312753200531006,
      "step": 76626
    },
    {
      "epoch": 2.5043487548828124e-06,
      "step": 76626,
      "training_step_time": 8.624209642410278
    },
    {
      "epoch": 2.5053024291992185e-06,
      "model_forward_time": 1.7313401699066162,
      "step": 76627
    },
    {
      "epoch": 2.5053024291992185e-06,
      "step": 76627,
      "training_step_time": 7.932755708694458
    },
    {
      "epoch": 2.506256103515625e-06,
      "model_forward_time": 1.7383358478546143,
      "step": 76628
    },
    {
      "epoch": 2.506256103515625e-06,
      "step": 76628,
      "training_step_time": 8.771424055099487
    },
    {
      "epoch": 2.5072097778320313e-06,
      "model_forward_time": 1.8034403324127197,
      "step": 76629
    },
    {
      "epoch": 2.5072097778320313e-06,
      "step": 76629,
      "training_step_time": 7.81751012802124
    },
    {
      "epoch": 2.5081634521484374e-06,
      "grad_norm": 0.1170157715678215,
      "learning_rate": 1.4204330263929722e-06,
      "loss": 0.0538,
      "step": 76630
    },
    {
      "epoch": 2.5081634521484374e-06,
      "model_forward_time": 1.7284140586853027,
      "step": 76630
    },
    {
      "dynamics_loss_avg": 0.06794670466333627,
      "epoch": 2.5081634521484374e-06,
      "step": 76630
    },
    {
      "action_loss_avg": 0.0016500981175340713,
      "epoch": 2.5081634521484374e-06,
      "step": 76630
    },
    {
      "epoch": 2.5081634521484374e-06,
      "step": 76630,
      "training_step_time": 8.503293752670288
    },
    {
      "epoch": 2.5091171264648435e-06,
      "model_forward_time": 1.7289834022521973,
      "step": 76631
    },
    {
      "epoch": 2.5091171264648435e-06,
      "step": 76631,
      "training_step_time": 8.543948888778687
    },
    {
      "epoch": 2.51007080078125e-06,
      "model_forward_time": 1.735217571258545,
      "step": 76632
    },
    {
      "epoch": 2.51007080078125e-06,
      "step": 76632,
      "training_step_time": 8.536600351333618
    },
    {
      "epoch": 2.5110244750976563e-06,
      "model_forward_time": 1.7441790103912354,
      "step": 76633
    },
    {
      "epoch": 2.5110244750976563e-06,
      "step": 76633,
      "training_step_time": 8.413314580917358
    },
    {
      "epoch": 2.5119781494140624e-06,
      "model_forward_time": 1.7346532344818115,
      "step": 76634
    },
    {
      "epoch": 2.5119781494140624e-06,
      "step": 76634,
      "training_step_time": 7.785577774047852
    },
    {
      "epoch": 2.5129318237304685e-06,
      "model_forward_time": 1.7365760803222656,
      "step": 76635
    },
    {
      "epoch": 2.5129318237304685e-06,
      "step": 76635,
      "training_step_time": 7.686691045761108
    },
    {
      "epoch": 2.513885498046875e-06,
      "model_forward_time": 1.7434041500091553,
      "step": 76636
    },
    {
      "epoch": 2.513885498046875e-06,
      "step": 76636,
      "training_step_time": 8.903173685073853
    },
    {
      "epoch": 2.5148391723632813e-06,
      "model_forward_time": 1.742799997329712,
      "step": 76637
    },
    {
      "epoch": 2.5148391723632813e-06,
      "step": 76637,
      "training_step_time": 8.551334142684937
    },
    {
      "epoch": 2.5157928466796874e-06,
      "model_forward_time": 2.418226718902588,
      "step": 76638
    },
    {
      "epoch": 2.5157928466796874e-06,
      "step": 76638,
      "training_step_time": 9.87175989151001
    },
    {
      "epoch": 2.5167465209960936e-06,
      "model_forward_time": 1.7353360652923584,
      "step": 76639
    },
    {
      "epoch": 2.5167465209960936e-06,
      "step": 76639,
      "training_step_time": 7.957936525344849
    },
    {
      "epoch": 2.5177001953125e-06,
      "grad_norm": 0.18970073759555817,
      "learning_rate": 1.4192787875477537e-06,
      "loss": 0.0527,
      "step": 76640
    },
    {
      "epoch": 2.5177001953125e-06,
      "model_forward_time": 1.7327089309692383,
      "step": 76640
    },
    {
      "dynamics_loss_avg": 0.04547125492244959,
      "epoch": 2.5177001953125e-06,
      "step": 76640
    },
    {
      "action_loss_avg": 0.0012921320594614372,
      "epoch": 2.5177001953125e-06,
      "step": 76640
    },
    {
      "epoch": 2.5177001953125e-06,
      "step": 76640,
      "training_step_time": 8.512119054794312
    },
    {
      "epoch": 2.5186538696289063e-06,
      "model_forward_time": 1.7401576042175293,
      "step": 76641
    },
    {
      "epoch": 2.5186538696289063e-06,
      "step": 76641,
      "training_step_time": 8.070971965789795
    },
    {
      "epoch": 2.5196075439453124e-06,
      "model_forward_time": 1.7405056953430176,
      "step": 76642
    },
    {
      "epoch": 2.5196075439453124e-06,
      "step": 76642,
      "training_step_time": 7.630678176879883
    },
    {
      "epoch": 2.5205612182617186e-06,
      "model_forward_time": 1.7257330417633057,
      "step": 76643
    },
    {
      "epoch": 2.5205612182617186e-06,
      "step": 76643,
      "training_step_time": 7.6419196128845215
    },
    {
      "epoch": 2.521514892578125e-06,
      "model_forward_time": 1.733461618423462,
      "step": 76644
    },
    {
      "epoch": 2.521514892578125e-06,
      "step": 76644,
      "training_step_time": 7.647250413894653
    },
    {
      "epoch": 2.5224685668945313e-06,
      "model_forward_time": 1.738539218902588,
      "step": 76645
    },
    {
      "epoch": 2.5224685668945313e-06,
      "step": 76645,
      "training_step_time": 8.162373065948486
    },
    {
      "epoch": 2.5234222412109374e-06,
      "model_forward_time": 1.7348873615264893,
      "step": 76646
    },
    {
      "epoch": 2.5234222412109374e-06,
      "step": 76646,
      "training_step_time": 8.478033304214478
    },
    {
      "epoch": 2.5243759155273436e-06,
      "model_forward_time": 2.410423755645752,
      "step": 76647
    },
    {
      "epoch": 2.5243759155273436e-06,
      "step": 76647,
      "training_step_time": 7.564364671707153
    },
    {
      "epoch": 2.52532958984375e-06,
      "model_forward_time": 1.7537579536437988,
      "step": 76648
    },
    {
      "epoch": 2.52532958984375e-06,
      "step": 76648,
      "training_step_time": 8.254982709884644
    },
    {
      "epoch": 2.5262832641601563e-06,
      "model_forward_time": 1.7922358512878418,
      "step": 76649
    },
    {
      "epoch": 2.5262832641601563e-06,
      "step": 76649,
      "training_step_time": 7.638217449188232
    },
    {
      "epoch": 2.5272369384765624e-06,
      "grad_norm": 0.06950689852237701,
      "learning_rate": 1.4181249402848246e-06,
      "loss": 0.0535,
      "step": 76650
    },
    {
      "epoch": 2.5272369384765624e-06,
      "model_forward_time": 1.7651615142822266,
      "step": 76650
    },
    {
      "dynamics_loss_avg": 0.052588658034801485,
      "epoch": 2.5272369384765624e-06,
      "step": 76650
    },
    {
      "action_loss_avg": 0.001603063730726717,
      "epoch": 2.5272369384765624e-06,
      "step": 76650
    },
    {
      "epoch": 2.5272369384765624e-06,
      "step": 76650,
      "training_step_time": 9.27724289894104
    },
    {
      "epoch": 2.5281906127929686e-06,
      "model_forward_time": 1.7678487300872803,
      "step": 76651
    },
    {
      "epoch": 2.5281906127929686e-06,
      "step": 76651,
      "training_step_time": 7.625278949737549
    },
    {
      "epoch": 2.529144287109375e-06,
      "model_forward_time": 1.8382911682128906,
      "step": 76652
    },
    {
      "epoch": 2.529144287109375e-06,
      "step": 76652,
      "training_step_time": 7.677014112472534
    },
    {
      "epoch": 2.5300979614257813e-06,
      "model_forward_time": 1.7612550258636475,
      "step": 76653
    },
    {
      "epoch": 2.5300979614257813e-06,
      "step": 76653,
      "training_step_time": 8.673821210861206
    },
    {
      "epoch": 2.5310516357421874e-06,
      "model_forward_time": 1.7585783004760742,
      "step": 76654
    },
    {
      "epoch": 2.5310516357421874e-06,
      "step": 76654,
      "training_step_time": 8.27428674697876
    },
    {
      "epoch": 2.5320053100585936e-06,
      "model_forward_time": 1.7597088813781738,
      "step": 76655
    },
    {
      "epoch": 2.5320053100585936e-06,
      "step": 76655,
      "training_step_time": 7.758502244949341
    },
    {
      "epoch": 2.532958984375e-06,
      "model_forward_time": 2.8649744987487793,
      "step": 76656
    },
    {
      "epoch": 2.532958984375e-06,
      "step": 76656,
      "training_step_time": 7.618175268173218
    },
    {
      "epoch": 2.5339126586914063e-06,
      "model_forward_time": 1.4110090732574463,
      "step": 76657
    },
    {
      "epoch": 2.5339126586914063e-06,
      "step": 76657,
      "training_step_time": 7.669104337692261
    },
    {
      "epoch": 2.5348663330078124e-06,
      "model_forward_time": 1.76615309715271,
      "step": 76658
    },
    {
      "epoch": 2.5348663330078124e-06,
      "step": 76658,
      "training_step_time": 8.642044305801392
    },
    {
      "epoch": 2.5358200073242186e-06,
      "model_forward_time": 1.7626399993896484,
      "step": 76659
    },
    {
      "epoch": 2.5358200073242186e-06,
      "step": 76659,
      "training_step_time": 7.584892988204956
    },
    {
      "epoch": 2.536773681640625e-06,
      "grad_norm": 0.14035198092460632,
      "learning_rate": 1.4169714847303656e-06,
      "loss": 0.0549,
      "step": 76660
    },
    {
      "epoch": 2.536773681640625e-06,
      "model_forward_time": 1.4120264053344727,
      "step": 76660
    },
    {
      "dynamics_loss_avg": 0.05386938201263547,
      "epoch": 2.536773681640625e-06,
      "step": 76660
    },
    {
      "action_loss_avg": 0.0017522113863378762,
      "epoch": 2.536773681640625e-06,
      "step": 76660
    },
    {
      "epoch": 2.536773681640625e-06,
      "step": 76660,
      "training_step_time": 7.5998215675354
    },
    {
      "epoch": 2.5377273559570313e-06,
      "model_forward_time": 1.8283352851867676,
      "step": 76661
    },
    {
      "epoch": 2.5377273559570313e-06,
      "step": 76661,
      "training_step_time": 7.698862791061401
    },
    {
      "epoch": 2.5386810302734374e-06,
      "model_forward_time": 1.7508869171142578,
      "step": 76662
    },
    {
      "epoch": 2.5386810302734374e-06,
      "step": 76662,
      "training_step_time": 7.627047777175903
    },
    {
      "epoch": 2.5396347045898436e-06,
      "model_forward_time": 1.776923656463623,
      "step": 76663
    },
    {
      "epoch": 2.5396347045898436e-06,
      "step": 76663,
      "training_step_time": 8.336787223815918
    },
    {
      "epoch": 2.54058837890625e-06,
      "model_forward_time": 1.79864501953125,
      "step": 76664
    },
    {
      "epoch": 2.54058837890625e-06,
      "step": 76664,
      "training_step_time": 7.780716419219971
    },
    {
      "epoch": 2.5415420532226563e-06,
      "model_forward_time": 1.7566561698913574,
      "step": 76665
    },
    {
      "epoch": 2.5415420532226563e-06,
      "step": 76665,
      "training_step_time": 8.633856534957886
    },
    {
      "epoch": 2.5424957275390624e-06,
      "model_forward_time": 1.7759990692138672,
      "step": 76666
    },
    {
      "epoch": 2.5424957275390624e-06,
      "step": 76666,
      "training_step_time": 8.54483938217163
    },
    {
      "epoch": 2.5434494018554686e-06,
      "model_forward_time": 1.791623830795288,
      "step": 76667
    },
    {
      "epoch": 2.5434494018554686e-06,
      "step": 76667,
      "training_step_time": 8.285134553909302
    },
    {
      "epoch": 2.544403076171875e-06,
      "model_forward_time": 1.7587158679962158,
      "step": 76668
    },
    {
      "epoch": 2.544403076171875e-06,
      "step": 76668,
      "training_step_time": 7.644165515899658
    },
    {
      "epoch": 2.5453567504882813e-06,
      "model_forward_time": 1.7774527072906494,
      "step": 76669
    },
    {
      "epoch": 2.5453567504882813e-06,
      "step": 76669,
      "training_step_time": 8.535000085830688
    },
    {
      "epoch": 2.5463104248046874e-06,
      "grad_norm": 0.07451843470335007,
      "learning_rate": 1.4158184210105196e-06,
      "loss": 0.0547,
      "step": 76670
    },
    {
      "epoch": 2.5463104248046874e-06,
      "model_forward_time": 1.7894668579101562,
      "step": 76670
    },
    {
      "dynamics_loss_avg": 0.05411020154133439,
      "epoch": 2.5463104248046874e-06,
      "step": 76670
    },
    {
      "action_loss_avg": 0.0015224307775497437,
      "epoch": 2.5463104248046874e-06,
      "step": 76670
    },
    {
      "epoch": 2.5463104248046874e-06,
      "step": 76670,
      "training_step_time": 7.680088520050049
    },
    {
      "epoch": 2.5472640991210936e-06,
      "model_forward_time": 1.7674729824066162,
      "step": 76671
    },
    {
      "epoch": 2.5472640991210936e-06,
      "step": 76671,
      "training_step_time": 7.6672141551971436
    },
    {
      "epoch": 2.5482177734375e-06,
      "model_forward_time": 1.7734096050262451,
      "step": 76672
    },
    {
      "epoch": 2.5482177734375e-06,
      "step": 76672,
      "training_step_time": 7.69171667098999
    },
    {
      "epoch": 2.5491714477539063e-06,
      "model_forward_time": 1.7723841667175293,
      "step": 76673
    },
    {
      "epoch": 2.5491714477539063e-06,
      "step": 76673,
      "training_step_time": 8.717713594436646
    },
    {
      "epoch": 2.5501251220703124e-06,
      "model_forward_time": 1.743976354598999,
      "step": 76674
    },
    {
      "epoch": 2.5501251220703124e-06,
      "step": 76674,
      "training_step_time": 7.632091999053955
    },
    {
      "epoch": 2.5510787963867186e-06,
      "model_forward_time": 2.870994806289673,
      "step": 76675
    },
    {
      "epoch": 2.5510787963867186e-06,
      "step": 76675,
      "training_step_time": 8.522422075271606
    },
    {
      "epoch": 2.552032470703125e-06,
      "model_forward_time": 1.772446632385254,
      "step": 76676
    },
    {
      "epoch": 2.552032470703125e-06,
      "step": 76676,
      "training_step_time": 8.497148990631104
    },
    {
      "epoch": 2.5529861450195313e-06,
      "model_forward_time": 1.7718548774719238,
      "step": 76677
    },
    {
      "epoch": 2.5529861450195313e-06,
      "step": 76677,
      "training_step_time": 7.734983921051025
    },
    {
      "epoch": 2.5539398193359374e-06,
      "model_forward_time": 1.7709596157073975,
      "step": 76678
    },
    {
      "epoch": 2.5539398193359374e-06,
      "step": 76678,
      "training_step_time": 8.776843309402466
    },
    {
      "epoch": 2.5548934936523436e-06,
      "model_forward_time": 1.7511484622955322,
      "step": 76679
    },
    {
      "epoch": 2.5548934936523436e-06,
      "step": 76679,
      "training_step_time": 7.6431190967559814
    },
    {
      "epoch": 2.55584716796875e-06,
      "grad_norm": 0.10492487251758575,
      "learning_rate": 1.4146657492513788e-06,
      "loss": 0.0536,
      "step": 76680
    },
    {
      "epoch": 2.55584716796875e-06,
      "model_forward_time": 1.8372113704681396,
      "step": 76680
    },
    {
      "dynamics_loss_avg": 0.0503408707678318,
      "epoch": 2.55584716796875e-06,
      "step": 76680
    },
    {
      "action_loss_avg": 0.0009414779051439837,
      "epoch": 2.55584716796875e-06,
      "step": 76680
    },
    {
      "epoch": 2.55584716796875e-06,
      "step": 76680,
      "training_step_time": 7.609888076782227
    },
    {
      "epoch": 2.5568008422851563e-06,
      "model_forward_time": 1.7698924541473389,
      "step": 76681
    },
    {
      "epoch": 2.5568008422851563e-06,
      "step": 76681,
      "training_step_time": 7.694280385971069
    },
    {
      "epoch": 2.5577545166015624e-06,
      "model_forward_time": 1.7693507671356201,
      "step": 76682
    },
    {
      "epoch": 2.5577545166015624e-06,
      "step": 76682,
      "training_step_time": 8.462287187576294
    },
    {
      "epoch": 2.5587081909179686e-06,
      "model_forward_time": 1.7717907428741455,
      "step": 76683
    },
    {
      "epoch": 2.5587081909179686e-06,
      "step": 76683,
      "training_step_time": 8.486404418945312
    },
    {
      "epoch": 2.559661865234375e-06,
      "model_forward_time": 2.898167848587036,
      "step": 76684
    },
    {
      "epoch": 2.559661865234375e-06,
      "step": 76684,
      "training_step_time": 8.716363430023193
    },
    {
      "epoch": 2.5606155395507813e-06,
      "model_forward_time": 1.766507625579834,
      "step": 76685
    },
    {
      "epoch": 2.5606155395507813e-06,
      "step": 76685,
      "training_step_time": 8.438202619552612
    },
    {
      "epoch": 2.5615692138671874e-06,
      "model_forward_time": 1.7726402282714844,
      "step": 76686
    },
    {
      "epoch": 2.5615692138671874e-06,
      "step": 76686,
      "training_step_time": 8.619591474533081
    },
    {
      "epoch": 2.5625228881835936e-06,
      "model_forward_time": 1.7693631649017334,
      "step": 76687
    },
    {
      "epoch": 2.5625228881835936e-06,
      "step": 76687,
      "training_step_time": 7.581625699996948
    },
    {
      "epoch": 2.5634765625e-06,
      "model_forward_time": 1.785510540008545,
      "step": 76688
    },
    {
      "epoch": 2.5634765625e-06,
      "step": 76688,
      "training_step_time": 8.640418529510498
    },
    {
      "epoch": 2.5644302368164063e-06,
      "model_forward_time": 1.8420300483703613,
      "step": 76689
    },
    {
      "epoch": 2.5644302368164063e-06,
      "step": 76689,
      "training_step_time": 8.414482355117798
    },
    {
      "epoch": 2.5653839111328124e-06,
      "grad_norm": 0.10407542437314987,
      "learning_rate": 1.4135134695790048e-06,
      "loss": 0.054,
      "step": 76690
    },
    {
      "epoch": 2.5653839111328124e-06,
      "model_forward_time": 1.7694849967956543,
      "step": 76690
    },
    {
      "dynamics_loss_avg": 0.047757469583302736,
      "epoch": 2.5653839111328124e-06,
      "step": 76690
    },
    {
      "action_loss_avg": 0.0009574830532073974,
      "epoch": 2.5653839111328124e-06,
      "step": 76690
    },
    {
      "epoch": 2.5653839111328124e-06,
      "step": 76690,
      "training_step_time": 7.760023355484009
    },
    {
      "epoch": 2.5663375854492186e-06,
      "model_forward_time": 1.7760579586029053,
      "step": 76691
    },
    {
      "epoch": 2.5663375854492186e-06,
      "step": 76691,
      "training_step_time": 8.145035982131958
    },
    {
      "epoch": 2.567291259765625e-06,
      "model_forward_time": 1.7739121913909912,
      "step": 76692
    },
    {
      "epoch": 2.567291259765625e-06,
      "step": 76692,
      "training_step_time": 8.754441261291504
    },
    {
      "epoch": 2.5682449340820313e-06,
      "model_forward_time": 2.9102365970611572,
      "step": 76693
    },
    {
      "epoch": 2.5682449340820313e-06,
      "step": 76693,
      "training_step_time": 7.645002365112305
    },
    {
      "epoch": 2.5691986083984375e-06,
      "model_forward_time": 1.7752366065979004,
      "step": 76694
    },
    {
      "epoch": 2.5691986083984375e-06,
      "step": 76694,
      "training_step_time": 9.200738668441772
    },
    {
      "epoch": 2.5701522827148436e-06,
      "model_forward_time": 1.7683115005493164,
      "step": 76695
    },
    {
      "epoch": 2.5701522827148436e-06,
      "step": 76695,
      "training_step_time": 7.770761251449585
    },
    {
      "epoch": 2.57110595703125e-06,
      "model_forward_time": 1.7855207920074463,
      "step": 76696
    },
    {
      "epoch": 2.57110595703125e-06,
      "step": 76696,
      "training_step_time": 7.675320386886597
    },
    {
      "epoch": 2.5720596313476563e-06,
      "model_forward_time": 1.7774393558502197,
      "step": 76697
    },
    {
      "epoch": 2.5720596313476563e-06,
      "step": 76697,
      "training_step_time": 7.772464036941528
    },
    {
      "epoch": 2.5730133056640625e-06,
      "model_forward_time": 1.8406827449798584,
      "step": 76698
    },
    {
      "epoch": 2.5730133056640625e-06,
      "step": 76698,
      "training_step_time": 7.72892951965332
    },
    {
      "epoch": 2.5739669799804686e-06,
      "model_forward_time": 1.7797155380249023,
      "step": 76699
    },
    {
      "epoch": 2.5739669799804686e-06,
      "step": 76699,
      "training_step_time": 8.322757482528687
    },
    {
      "epoch": 2.574920654296875e-06,
      "grad_norm": 0.08049264550209045,
      "learning_rate": 1.4123615821194037e-06,
      "loss": 0.0546,
      "step": 76700
    },
    {
      "epoch": 2.574920654296875e-06,
      "model_forward_time": 1.757849931716919,
      "step": 76700
    },
    {
      "dynamics_loss_avg": 0.05578067284077406,
      "epoch": 2.574920654296875e-06,
      "step": 76700
    },
    {
      "action_loss_avg": 0.00125964573235251,
      "epoch": 2.574920654296875e-06,
      "step": 76700
    },
    {
      "epoch": 2.574920654296875e-06,
      "step": 76700,
      "training_step_time": 8.902564287185669
    },
    {
      "epoch": 2.5758743286132813e-06,
      "model_forward_time": 1.776996374130249,
      "step": 76701
    },
    {
      "epoch": 2.5758743286132813e-06,
      "step": 76701,
      "training_step_time": 8.71713376045227
    },
    {
      "epoch": 2.5768280029296875e-06,
      "model_forward_time": 2.8365399837493896,
      "step": 76702
    },
    {
      "epoch": 2.5768280029296875e-06,
      "step": 76702,
      "training_step_time": 8.579968214035034
    },
    {
      "epoch": 2.5777816772460936e-06,
      "model_forward_time": 1.766331434249878,
      "step": 76703
    },
    {
      "epoch": 2.5777816772460936e-06,
      "step": 76703,
      "training_step_time": 7.685525894165039
    },
    {
      "epoch": 2.5787353515625e-06,
      "model_forward_time": 1.7711822986602783,
      "step": 76704
    },
    {
      "epoch": 2.5787353515625e-06,
      "step": 76704,
      "training_step_time": 7.808732271194458
    },
    {
      "epoch": 2.5796890258789063e-06,
      "model_forward_time": 1.7681405544281006,
      "step": 76705
    },
    {
      "epoch": 2.5796890258789063e-06,
      "step": 76705,
      "training_step_time": 7.69235634803772
    },
    {
      "epoch": 2.5806427001953125e-06,
      "model_forward_time": 1.7647099494934082,
      "step": 76706
    },
    {
      "epoch": 2.5806427001953125e-06,
      "step": 76706,
      "training_step_time": 7.73449182510376
    },
    {
      "epoch": 2.5815963745117186e-06,
      "model_forward_time": 1.8290839195251465,
      "step": 76707
    },
    {
      "epoch": 2.5815963745117186e-06,
      "step": 76707,
      "training_step_time": 7.692455768585205
    },
    {
      "epoch": 2.582550048828125e-06,
      "model_forward_time": 1.7830290794372559,
      "step": 76708
    },
    {
      "epoch": 2.582550048828125e-06,
      "step": 76708,
      "training_step_time": 7.740366697311401
    },
    {
      "epoch": 2.5835037231445313e-06,
      "model_forward_time": 1.7508559226989746,
      "step": 76709
    },
    {
      "epoch": 2.5835037231445313e-06,
      "step": 76709,
      "training_step_time": 7.649442195892334
    },
    {
      "epoch": 2.5844573974609375e-06,
      "grad_norm": 0.09534484893083572,
      "learning_rate": 1.4112100869985473e-06,
      "loss": 0.0552,
      "step": 76710
    },
    {
      "epoch": 2.5844573974609375e-06,
      "model_forward_time": 1.7600476741790771,
      "step": 76710
    },
    {
      "dynamics_loss_avg": 0.06171145346015692,
      "epoch": 2.5844573974609375e-06,
      "step": 76710
    },
    {
      "action_loss_avg": 0.0017172925174236298,
      "epoch": 2.5844573974609375e-06,
      "step": 76710
    },
    {
      "epoch": 2.5844573974609375e-06,
      "step": 76710,
      "training_step_time": 7.696245193481445
    },
    {
      "epoch": 2.5854110717773436e-06,
      "model_forward_time": 2.8895297050476074,
      "step": 76711
    },
    {
      "epoch": 2.5854110717773436e-06,
      "step": 76711,
      "training_step_time": 8.479825258255005
    },
    {
      "epoch": 2.58636474609375e-06,
      "model_forward_time": 1.7608706951141357,
      "step": 76712
    },
    {
      "epoch": 2.58636474609375e-06,
      "step": 76712,
      "training_step_time": 7.6299333572387695
    },
    {
      "epoch": 2.5873184204101563e-06,
      "model_forward_time": 1.7705881595611572,
      "step": 76713
    },
    {
      "epoch": 2.5873184204101563e-06,
      "step": 76713,
      "training_step_time": 8.564616918563843
    },
    {
      "epoch": 2.5882720947265625e-06,
      "model_forward_time": 1.7797625064849854,
      "step": 76714
    },
    {
      "epoch": 2.5882720947265625e-06,
      "step": 76714,
      "training_step_time": 7.75738000869751
    },
    {
      "epoch": 2.5892257690429686e-06,
      "model_forward_time": 1.7723448276519775,
      "step": 76715
    },
    {
      "epoch": 2.5892257690429686e-06,
      "step": 76715,
      "training_step_time": 7.738152503967285
    },
    {
      "epoch": 2.590179443359375e-06,
      "model_forward_time": 1.7668893337249756,
      "step": 76716
    },
    {
      "epoch": 2.590179443359375e-06,
      "step": 76716,
      "training_step_time": 7.732797622680664
    },
    {
      "epoch": 2.5911331176757813e-06,
      "model_forward_time": 1.847785234451294,
      "step": 76717
    },
    {
      "epoch": 2.5911331176757813e-06,
      "step": 76717,
      "training_step_time": 7.761091709136963
    },
    {
      "epoch": 2.5920867919921875e-06,
      "model_forward_time": 1.7831993103027344,
      "step": 76718
    },
    {
      "epoch": 2.5920867919921875e-06,
      "step": 76718,
      "training_step_time": 7.75048565864563
    },
    {
      "epoch": 2.5930404663085936e-06,
      "model_forward_time": 1.782106637954712,
      "step": 76719
    },
    {
      "epoch": 2.5930404663085936e-06,
      "step": 76719,
      "training_step_time": 9.815572738647461
    },
    {
      "epoch": 2.593994140625e-06,
      "grad_norm": 0.10322042554616928,
      "learning_rate": 1.410058984342359e-06,
      "loss": 0.0543,
      "step": 76720
    },
    {
      "epoch": 2.593994140625e-06,
      "model_forward_time": 1.7445847988128662,
      "step": 76720
    },
    {
      "dynamics_loss_avg": 0.04982158914208412,
      "epoch": 2.593994140625e-06,
      "step": 76720
    },
    {
      "action_loss_avg": 0.0016411913791671395,
      "epoch": 2.593994140625e-06,
      "step": 76720
    },
    {
      "epoch": 2.593994140625e-06,
      "step": 76720,
      "training_step_time": 10.558851480484009
    },
    {
      "epoch": 2.5949478149414063e-06,
      "model_forward_time": 1.7614154815673828,
      "step": 76721
    },
    {
      "epoch": 2.5949478149414063e-06,
      "step": 76721,
      "training_step_time": 8.379653453826904
    },
    {
      "epoch": 2.5959014892578125e-06,
      "model_forward_time": 1.7539184093475342,
      "step": 76722
    },
    {
      "epoch": 2.5959014892578125e-06,
      "step": 76722,
      "training_step_time": 7.792969465255737
    },
    {
      "epoch": 2.5968551635742186e-06,
      "model_forward_time": 1.7547402381896973,
      "step": 76723
    },
    {
      "epoch": 2.5968551635742186e-06,
      "step": 76723,
      "training_step_time": 9.39870810508728
    },
    {
      "epoch": 2.597808837890625e-06,
      "model_forward_time": 1.778604507446289,
      "step": 76724
    },
    {
      "epoch": 2.597808837890625e-06,
      "step": 76724,
      "training_step_time": 7.680161476135254
    },
    {
      "epoch": 2.5987625122070313e-06,
      "model_forward_time": 1.7603332996368408,
      "step": 76725
    },
    {
      "epoch": 2.5987625122070313e-06,
      "step": 76725,
      "training_step_time": 7.7486326694488525
    },
    {
      "epoch": 2.5997161865234375e-06,
      "model_forward_time": 1.8295915126800537,
      "step": 76726
    },
    {
      "epoch": 2.5997161865234375e-06,
      "step": 76726,
      "training_step_time": 7.746158838272095
    },
    {
      "epoch": 2.6006698608398436e-06,
      "model_forward_time": 1.7661404609680176,
      "step": 76727
    },
    {
      "epoch": 2.6006698608398436e-06,
      "step": 76727,
      "training_step_time": 7.701977252960205
    },
    {
      "epoch": 2.60162353515625e-06,
      "model_forward_time": 1.7492475509643555,
      "step": 76728
    },
    {
      "epoch": 2.60162353515625e-06,
      "step": 76728,
      "training_step_time": 8.303622007369995
    },
    {
      "epoch": 2.6025772094726563e-06,
      "model_forward_time": 1.7590365409851074,
      "step": 76729
    },
    {
      "epoch": 2.6025772094726563e-06,
      "step": 76729,
      "training_step_time": 7.831042528152466
    },
    {
      "epoch": 2.6035308837890625e-06,
      "grad_norm": 0.11933735013008118,
      "learning_rate": 1.408908274276723e-06,
      "loss": 0.0558,
      "step": 76730
    },
    {
      "epoch": 2.6035308837890625e-06,
      "model_forward_time": 2.886016607284546,
      "step": 76730
    },
    {
      "dynamics_loss_avg": 0.0661294573917985,
      "epoch": 2.6035308837890625e-06,
      "step": 76730
    },
    {
      "action_loss_avg": 0.001956402463838458,
      "epoch": 2.6035308837890625e-06,
      "step": 76730
    },
    {
      "epoch": 2.6035308837890625e-06,
      "step": 76730,
      "training_step_time": 8.742146492004395
    },
    {
      "epoch": 2.6044845581054686e-06,
      "model_forward_time": 1.7645769119262695,
      "step": 76731
    },
    {
      "epoch": 2.6044845581054686e-06,
      "step": 76731,
      "training_step_time": 10.177652835845947
    },
    {
      "epoch": 2.605438232421875e-06,
      "model_forward_time": 1.7849621772766113,
      "step": 76732
    },
    {
      "epoch": 2.605438232421875e-06,
      "step": 76732,
      "training_step_time": 7.676375389099121
    },
    {
      "epoch": 2.6063919067382813e-06,
      "model_forward_time": 1.7582426071166992,
      "step": 76733
    },
    {
      "epoch": 2.6063919067382813e-06,
      "step": 76733,
      "training_step_time": 7.621723651885986
    },
    {
      "epoch": 2.6073455810546875e-06,
      "model_forward_time": 1.7336235046386719,
      "step": 76734
    },
    {
      "epoch": 2.6073455810546875e-06,
      "step": 76734,
      "training_step_time": 7.726232051849365
    },
    {
      "epoch": 2.6082992553710936e-06,
      "model_forward_time": 1.802570104598999,
      "step": 76735
    },
    {
      "epoch": 2.6082992553710936e-06,
      "step": 76735,
      "training_step_time": 7.7580249309539795
    },
    {
      "epoch": 2.6092529296875e-06,
      "model_forward_time": 1.7552094459533691,
      "step": 76736
    },
    {
      "epoch": 2.6092529296875e-06,
      "step": 76736,
      "training_step_time": 7.690005779266357
    },
    {
      "epoch": 2.6102066040039063e-06,
      "model_forward_time": 1.7475507259368896,
      "step": 76737
    },
    {
      "epoch": 2.6102066040039063e-06,
      "step": 76737,
      "training_step_time": 8.507330417633057
    },
    {
      "epoch": 2.6111602783203125e-06,
      "model_forward_time": 1.7594935894012451,
      "step": 76738
    },
    {
      "epoch": 2.6111602783203125e-06,
      "step": 76738,
      "training_step_time": 7.7542030811309814
    },
    {
      "epoch": 2.6121139526367186e-06,
      "model_forward_time": 2.860971212387085,
      "step": 76739
    },
    {
      "epoch": 2.6121139526367186e-06,
      "step": 76739,
      "training_step_time": 7.779635429382324
    },
    {
      "epoch": 2.613067626953125e-06,
      "grad_norm": 0.07594957202672958,
      "learning_rate": 1.4077579569274785e-06,
      "loss": 0.0542,
      "step": 76740
    },
    {
      "epoch": 2.613067626953125e-06,
      "model_forward_time": 1.7490234375,
      "step": 76740
    },
    {
      "dynamics_loss_avg": 0.06984787825495005,
      "epoch": 2.613067626953125e-06,
      "step": 76740
    },
    {
      "action_loss_avg": 0.0021885408903472127,
      "epoch": 2.613067626953125e-06,
      "step": 76740
    },
    {
      "epoch": 2.613067626953125e-06,
      "step": 76740,
      "training_step_time": 8.572548866271973
    },
    {
      "epoch": 2.6140213012695313e-06,
      "model_forward_time": 1.7550179958343506,
      "step": 76741
    },
    {
      "epoch": 2.6140213012695313e-06,
      "step": 76741,
      "training_step_time": 8.623941898345947
    },
    {
      "epoch": 2.6149749755859375e-06,
      "model_forward_time": 1.7524399757385254,
      "step": 76742
    },
    {
      "epoch": 2.6149749755859375e-06,
      "step": 76742,
      "training_step_time": 8.765288829803467
    },
    {
      "epoch": 2.6159286499023436e-06,
      "model_forward_time": 1.769470453262329,
      "step": 76743
    },
    {
      "epoch": 2.6159286499023436e-06,
      "step": 76743,
      "training_step_time": 9.292575597763062
    },
    {
      "epoch": 2.61688232421875e-06,
      "model_forward_time": 1.7838504314422607,
      "step": 76744
    },
    {
      "epoch": 2.61688232421875e-06,
      "step": 76744,
      "training_step_time": 7.798477411270142
    },
    {
      "epoch": 2.6178359985351564e-06,
      "model_forward_time": 1.8244028091430664,
      "step": 76745
    },
    {
      "epoch": 2.6178359985351564e-06,
      "step": 76745,
      "training_step_time": 8.456875562667847
    },
    {
      "epoch": 2.6187896728515625e-06,
      "model_forward_time": 1.7538251876831055,
      "step": 76746
    },
    {
      "epoch": 2.6187896728515625e-06,
      "step": 76746,
      "training_step_time": 9.221237659454346
    },
    {
      "epoch": 2.6197433471679686e-06,
      "model_forward_time": 1.7565569877624512,
      "step": 76747
    },
    {
      "epoch": 2.6197433471679686e-06,
      "step": 76747,
      "training_step_time": 8.414415836334229
    },
    {
      "epoch": 2.620697021484375e-06,
      "model_forward_time": 3.4856133460998535,
      "step": 76748
    },
    {
      "epoch": 2.620697021484375e-06,
      "step": 76748,
      "training_step_time": 10.022690057754517
    },
    {
      "epoch": 2.6216506958007814e-06,
      "model_forward_time": 1.7541277408599854,
      "step": 76749
    },
    {
      "epoch": 2.6216506958007814e-06,
      "step": 76749,
      "training_step_time": 6.78718113899231
    },
    {
      "epoch": 2.6226043701171875e-06,
      "grad_norm": 0.09143088012933731,
      "learning_rate": 1.4066080324204245e-06,
      "loss": 0.0549,
      "step": 76750
    },
    {
      "epoch": 2.6226043701171875e-06,
      "model_forward_time": 1.7534751892089844,
      "step": 76750
    },
    {
      "dynamics_loss_avg": 0.036906130425632,
      "epoch": 2.6226043701171875e-06,
      "step": 76750
    },
    {
      "action_loss_avg": 0.0007498492923332378,
      "epoch": 2.6226043701171875e-06,
      "step": 76750
    },
    {
      "epoch": 2.6226043701171875e-06,
      "step": 76750,
      "training_step_time": 7.630099296569824
    },
    {
      "epoch": 2.6235580444335936e-06,
      "model_forward_time": 1.7551686763763428,
      "step": 76751
    },
    {
      "epoch": 2.6235580444335936e-06,
      "step": 76751,
      "training_step_time": 8.263306379318237
    },
    {
      "epoch": 2.62451171875e-06,
      "model_forward_time": 1.7539772987365723,
      "step": 76752
    },
    {
      "epoch": 2.62451171875e-06,
      "step": 76752,
      "training_step_time": 7.635836362838745
    },
    {
      "epoch": 2.6254653930664064e-06,
      "model_forward_time": 1.7580060958862305,
      "step": 76753
    },
    {
      "epoch": 2.6254653930664064e-06,
      "step": 76753,
      "training_step_time": 6.740877151489258
    },
    {
      "epoch": 2.6264190673828125e-06,
      "model_forward_time": 1.7553000450134277,
      "step": 76754
    },
    {
      "epoch": 2.6264190673828125e-06,
      "step": 76754,
      "training_step_time": 6.749438047409058
    },
    {
      "epoch": 2.6273727416992186e-06,
      "model_forward_time": 1.7518038749694824,
      "step": 76755
    },
    {
      "epoch": 2.6273727416992186e-06,
      "step": 76755,
      "training_step_time": 6.879013538360596
    },
    {
      "epoch": 2.628326416015625e-06,
      "model_forward_time": 1.757767915725708,
      "step": 76756
    },
    {
      "epoch": 2.628326416015625e-06,
      "step": 76756,
      "training_step_time": 6.852781295776367
    },
    {
      "epoch": 2.6292800903320314e-06,
      "model_forward_time": 2.459872245788574,
      "step": 76757
    },
    {
      "epoch": 2.6292800903320314e-06,
      "step": 76757,
      "training_step_time": 8.586844444274902
    },
    {
      "epoch": 2.6302337646484375e-06,
      "model_forward_time": 1.754305362701416,
      "step": 76758
    },
    {
      "epoch": 2.6302337646484375e-06,
      "step": 76758,
      "training_step_time": 8.91731071472168
    },
    {
      "epoch": 2.6311874389648437e-06,
      "model_forward_time": 1.754082202911377,
      "step": 76759
    },
    {
      "epoch": 2.6311874389648437e-06,
      "step": 76759,
      "training_step_time": 7.422438859939575
    },
    {
      "epoch": 2.63214111328125e-06,
      "grad_norm": 0.10079989582300186,
      "learning_rate": 1.405458500881312e-06,
      "loss": 0.0537,
      "step": 76760
    },
    {
      "epoch": 2.63214111328125e-06,
      "model_forward_time": 1.7539772987365723,
      "step": 76760
    },
    {
      "dynamics_loss_avg": 0.048429861478507516,
      "epoch": 2.63214111328125e-06,
      "step": 76760
    },
    {
      "action_loss_avg": 0.000999108934774995,
      "epoch": 2.63214111328125e-06,
      "step": 76760
    },
    {
      "epoch": 2.63214111328125e-06,
      "step": 76760,
      "training_step_time": 9.332801103591919
    },
    {
      "epoch": 2.6330947875976564e-06,
      "model_forward_time": 1.75941801071167,
      "step": 76761
    },
    {
      "epoch": 2.6330947875976564e-06,
      "step": 76761,
      "training_step_time": 7.563421726226807
    },
    {
      "epoch": 2.6340484619140625e-06,
      "model_forward_time": 1.7559313774108887,
      "step": 76762
    },
    {
      "epoch": 2.6340484619140625e-06,
      "step": 76762,
      "training_step_time": 6.835149526596069
    },
    {
      "epoch": 2.6350021362304687e-06,
      "model_forward_time": 1.7650761604309082,
      "step": 76763
    },
    {
      "epoch": 2.6350021362304687e-06,
      "step": 76763,
      "training_step_time": 6.82990026473999
    },
    {
      "epoch": 2.635955810546875e-06,
      "model_forward_time": 1.3883137702941895,
      "step": 76764
    },
    {
      "epoch": 2.635955810546875e-06,
      "step": 76764,
      "training_step_time": 6.718823432922363
    },
    {
      "epoch": 2.6369094848632814e-06,
      "model_forward_time": 1.7470684051513672,
      "step": 76765
    },
    {
      "epoch": 2.6369094848632814e-06,
      "step": 76765,
      "training_step_time": 6.847712516784668
    },
    {
      "epoch": 2.6378631591796875e-06,
      "model_forward_time": 1.7551512718200684,
      "step": 76766
    },
    {
      "epoch": 2.6378631591796875e-06,
      "step": 76766,
      "training_step_time": 7.860372304916382
    },
    {
      "epoch": 2.6388168334960937e-06,
      "model_forward_time": 2.3587331771850586,
      "step": 76767
    },
    {
      "epoch": 2.6388168334960937e-06,
      "step": 76767,
      "training_step_time": 8.601358652114868
    },
    {
      "epoch": 2.6397705078125e-06,
      "model_forward_time": 1.7569785118103027,
      "step": 76768
    },
    {
      "epoch": 2.6397705078125e-06,
      "step": 76768,
      "training_step_time": 7.486809492111206
    },
    {
      "epoch": 2.6407241821289064e-06,
      "model_forward_time": 1.7594377994537354,
      "step": 76769
    },
    {
      "epoch": 2.6407241821289064e-06,
      "step": 76769,
      "training_step_time": 7.622209072113037
    },
    {
      "epoch": 2.6416778564453125e-06,
      "grad_norm": 0.08886408805847168,
      "learning_rate": 1.4043093624358538e-06,
      "loss": 0.0552,
      "step": 76770
    },
    {
      "epoch": 2.6416778564453125e-06,
      "model_forward_time": 1.7594611644744873,
      "step": 76770
    },
    {
      "dynamics_loss_avg": 0.05875269956886768,
      "epoch": 2.6416778564453125e-06,
      "step": 76770
    },
    {
      "action_loss_avg": 0.0011075204703956843,
      "epoch": 2.6416778564453125e-06,
      "step": 76770
    },
    {
      "epoch": 2.6416778564453125e-06,
      "step": 76770,
      "training_step_time": 7.620955228805542
    },
    {
      "epoch": 2.6426315307617187e-06,
      "model_forward_time": 1.762765884399414,
      "step": 76771
    },
    {
      "epoch": 2.6426315307617187e-06,
      "step": 76771,
      "training_step_time": 6.795296907424927
    },
    {
      "epoch": 2.643585205078125e-06,
      "model_forward_time": 1.754012107849121,
      "step": 76772
    },
    {
      "epoch": 2.643585205078125e-06,
      "step": 76772,
      "training_step_time": 6.8364787101745605
    },
    {
      "epoch": 2.6445388793945314e-06,
      "model_forward_time": 1.7672533988952637,
      "step": 76773
    },
    {
      "epoch": 2.6445388793945314e-06,
      "step": 76773,
      "training_step_time": 6.800465106964111
    },
    {
      "epoch": 2.6454925537109375e-06,
      "model_forward_time": 1.7599270343780518,
      "step": 76774
    },
    {
      "epoch": 2.6454925537109375e-06,
      "step": 76774,
      "training_step_time": 6.775455951690674
    },
    {
      "epoch": 2.6464462280273437e-06,
      "model_forward_time": 1.760051965713501,
      "step": 76775
    },
    {
      "epoch": 2.6464462280273437e-06,
      "step": 76775,
      "training_step_time": 6.897256851196289
    },
    {
      "epoch": 2.64739990234375e-06,
      "model_forward_time": 2.4268319606781006,
      "step": 76776
    },
    {
      "epoch": 2.64739990234375e-06,
      "step": 76776,
      "training_step_time": 9.694566249847412
    },
    {
      "epoch": 2.6483535766601564e-06,
      "model_forward_time": 1.7562756538391113,
      "step": 76777
    },
    {
      "epoch": 2.6483535766601564e-06,
      "step": 76777,
      "training_step_time": 7.649472713470459
    },
    {
      "epoch": 2.6493072509765625e-06,
      "model_forward_time": 1.7420141696929932,
      "step": 76778
    },
    {
      "epoch": 2.6493072509765625e-06,
      "step": 76778,
      "training_step_time": 7.680917263031006
    },
    {
      "epoch": 2.6502609252929687e-06,
      "model_forward_time": 1.754941701889038,
      "step": 76779
    },
    {
      "epoch": 2.6502609252929687e-06,
      "step": 76779,
      "training_step_time": 7.729876518249512
    },
    {
      "epoch": 2.651214599609375e-06,
      "grad_norm": 0.08622915297746658,
      "learning_rate": 1.403160617209719e-06,
      "loss": 0.0523,
      "step": 76780
    },
    {
      "epoch": 2.651214599609375e-06,
      "model_forward_time": 1.7468600273132324,
      "step": 76780
    },
    {
      "dynamics_loss_avg": 0.043229483999311924,
      "epoch": 2.651214599609375e-06,
      "step": 76780
    },
    {
      "action_loss_avg": 0.001203880220418796,
      "epoch": 2.651214599609375e-06,
      "step": 76780
    },
    {
      "epoch": 2.651214599609375e-06,
      "step": 76780,
      "training_step_time": 6.898184776306152
    },
    {
      "epoch": 2.6521682739257814e-06,
      "model_forward_time": 1.7520265579223633,
      "step": 76781
    },
    {
      "epoch": 2.6521682739257814e-06,
      "step": 76781,
      "training_step_time": 6.854578971862793
    },
    {
      "epoch": 2.6531219482421875e-06,
      "model_forward_time": 1.7483720779418945,
      "step": 76782
    },
    {
      "epoch": 2.6531219482421875e-06,
      "step": 76782,
      "training_step_time": 6.817509651184082
    },
    {
      "epoch": 2.6540756225585937e-06,
      "model_forward_time": 1.755894422531128,
      "step": 76783
    },
    {
      "epoch": 2.6540756225585937e-06,
      "step": 76783,
      "training_step_time": 6.9083476066589355
    },
    {
      "epoch": 2.655029296875e-06,
      "model_forward_time": 1.7539739608764648,
      "step": 76784
    },
    {
      "epoch": 2.655029296875e-06,
      "step": 76784,
      "training_step_time": 6.853098154067993
    },
    {
      "epoch": 2.6559829711914064e-06,
      "model_forward_time": 2.432671070098877,
      "step": 76785
    },
    {
      "epoch": 2.6559829711914064e-06,
      "step": 76785,
      "training_step_time": 12.03713583946228
    },
    {
      "epoch": 2.6569366455078125e-06,
      "model_forward_time": 1.7513294219970703,
      "step": 76786
    },
    {
      "epoch": 2.6569366455078125e-06,
      "step": 76786,
      "training_step_time": 8.658845901489258
    },
    {
      "epoch": 2.6578903198242187e-06,
      "model_forward_time": 1.7520029544830322,
      "step": 76787
    },
    {
      "epoch": 2.6578903198242187e-06,
      "step": 76787,
      "training_step_time": 7.671604633331299
    },
    {
      "epoch": 2.658843994140625e-06,
      "model_forward_time": 1.755368709564209,
      "step": 76788
    },
    {
      "epoch": 2.658843994140625e-06,
      "step": 76788,
      "training_step_time": 7.688736438751221
    },
    {
      "epoch": 2.6597976684570314e-06,
      "model_forward_time": 1.7504544258117676,
      "step": 76789
    },
    {
      "epoch": 2.6597976684570314e-06,
      "step": 76789,
      "training_step_time": 7.477841854095459
    },
    {
      "epoch": 2.6607513427734375e-06,
      "grad_norm": 0.06863854825496674,
      "learning_rate": 1.40201226532853e-06,
      "loss": 0.0543,
      "step": 76790
    },
    {
      "epoch": 2.6607513427734375e-06,
      "model_forward_time": 1.7531726360321045,
      "step": 76790
    },
    {
      "dynamics_loss_avg": 0.060302617587149145,
      "epoch": 2.6607513427734375e-06,
      "step": 76790
    },
    {
      "action_loss_avg": 0.0011793637910159304,
      "epoch": 2.6607513427734375e-06,
      "step": 76790
    },
    {
      "epoch": 2.6607513427734375e-06,
      "step": 76790,
      "training_step_time": 6.820889472961426
    },
    {
      "epoch": 2.6617050170898437e-06,
      "model_forward_time": 1.750943660736084,
      "step": 76791
    },
    {
      "epoch": 2.6617050170898437e-06,
      "step": 76791,
      "training_step_time": 6.834306478500366
    },
    {
      "epoch": 2.66265869140625e-06,
      "model_forward_time": 1.7313640117645264,
      "step": 76792
    },
    {
      "epoch": 2.66265869140625e-06,
      "step": 76792,
      "training_step_time": 6.760419607162476
    },
    {
      "epoch": 2.6636123657226564e-06,
      "model_forward_time": 1.7524478435516357,
      "step": 76793
    },
    {
      "epoch": 2.6636123657226564e-06,
      "step": 76793,
      "training_step_time": 6.755988836288452
    },
    {
      "epoch": 2.6645660400390625e-06,
      "model_forward_time": 2.4372072219848633,
      "step": 76794
    },
    {
      "epoch": 2.6645660400390625e-06,
      "step": 76794,
      "training_step_time": 8.32655930519104
    },
    {
      "epoch": 2.6655197143554687e-06,
      "model_forward_time": 1.746717929840088,
      "step": 76795
    },
    {
      "epoch": 2.6655197143554687e-06,
      "step": 76795,
      "training_step_time": 7.8568010330200195
    },
    {
      "epoch": 2.666473388671875e-06,
      "model_forward_time": 1.7496364116668701,
      "step": 76796
    },
    {
      "epoch": 2.666473388671875e-06,
      "step": 76796,
      "training_step_time": 7.659989356994629
    },
    {
      "epoch": 2.6674270629882814e-06,
      "model_forward_time": 1.7538132667541504,
      "step": 76797
    },
    {
      "epoch": 2.6674270629882814e-06,
      "step": 76797,
      "training_step_time": 7.739026784896851
    },
    {
      "epoch": 2.6683807373046875e-06,
      "model_forward_time": 1.7582876682281494,
      "step": 76798
    },
    {
      "epoch": 2.6683807373046875e-06,
      "step": 76798,
      "training_step_time": 7.4823102951049805
    },
    {
      "epoch": 2.6693344116210937e-06,
      "model_forward_time": 1.7464179992675781,
      "step": 76799
    },
    {
      "epoch": 2.6693344116210937e-06,
      "step": 76799,
      "training_step_time": 6.921572923660278
    },
    {
      "epoch": 2.6702880859375e-06,
      "grad_norm": 0.11570536345243454,
      "learning_rate": 1.4008643069178702e-06,
      "loss": 0.0545,
      "step": 76800
    },
    {
      "epoch": 2.6702880859375e-06,
      "model_forward_time": 1.764303207397461,
      "step": 76800
    },
    {
      "dynamics_loss_avg": 0.05405820207670331,
      "epoch": 2.6702880859375e-06,
      "step": 76800
    },
    {
      "action_loss_avg": 0.0013497900901711546,
      "epoch": 2.6702880859375e-06,
      "step": 76800
    },
    {
      "epoch": 2.6702880859375e-06,
      "step": 76800,
      "training_step_time": 6.831284999847412
    },
    {
      "epoch": 2.6712417602539064e-06,
      "model_forward_time": 1.7527751922607422,
      "step": 76801
    },
    {
      "epoch": 2.6712417602539064e-06,
      "step": 76801,
      "training_step_time": 6.797190427780151
    },
    {
      "epoch": 2.6721954345703125e-06,
      "model_forward_time": 1.7513034343719482,
      "step": 76802
    },
    {
      "epoch": 2.6721954345703125e-06,
      "step": 76802,
      "training_step_time": 6.7754740715026855
    },
    {
      "epoch": 2.6731491088867187e-06,
      "model_forward_time": 2.3664865493774414,
      "step": 76803
    },
    {
      "epoch": 2.6731491088867187e-06,
      "step": 76803,
      "training_step_time": 7.6579694747924805
    },
    {
      "epoch": 2.674102783203125e-06,
      "model_forward_time": 1.8281941413879395,
      "step": 76804
    },
    {
      "epoch": 2.674102783203125e-06,
      "step": 76804,
      "training_step_time": 9.430480718612671
    },
    {
      "epoch": 2.6750564575195314e-06,
      "model_forward_time": 1.7541728019714355,
      "step": 76805
    },
    {
      "epoch": 2.6750564575195314e-06,
      "step": 76805,
      "training_step_time": 7.6731038093566895
    },
    {
      "epoch": 2.6760101318359375e-06,
      "model_forward_time": 1.3832447528839111,
      "step": 76806
    },
    {
      "epoch": 2.6760101318359375e-06,
      "step": 76806,
      "training_step_time": 9.282777547836304
    },
    {
      "epoch": 2.6769638061523437e-06,
      "model_forward_time": 1.7568538188934326,
      "step": 76807
    },
    {
      "epoch": 2.6769638061523437e-06,
      "step": 76807,
      "training_step_time": 7.543414354324341
    },
    {
      "epoch": 2.67791748046875e-06,
      "model_forward_time": 1.755082607269287,
      "step": 76808
    },
    {
      "epoch": 2.67791748046875e-06,
      "step": 76808,
      "training_step_time": 6.836809158325195
    },
    {
      "epoch": 2.6788711547851564e-06,
      "model_forward_time": 1.752610683441162,
      "step": 76809
    },
    {
      "epoch": 2.6788711547851564e-06,
      "step": 76809,
      "training_step_time": 6.875163316726685
    },
    {
      "epoch": 2.6798248291015626e-06,
      "grad_norm": 0.09839882701635361,
      "learning_rate": 1.3997167421032782e-06,
      "loss": 0.0553,
      "step": 76810
    },
    {
      "epoch": 2.6798248291015626e-06,
      "model_forward_time": 1.7466378211975098,
      "step": 76810
    },
    {
      "dynamics_loss_avg": 0.06240738332271576,
      "epoch": 2.6798248291015626e-06,
      "step": 76810
    },
    {
      "action_loss_avg": 0.0020593998109688983,
      "epoch": 2.6798248291015626e-06,
      "step": 76810
    },
    {
      "epoch": 2.6798248291015626e-06,
      "step": 76810,
      "training_step_time": 6.814960479736328
    },
    {
      "epoch": 2.6807785034179687e-06,
      "model_forward_time": 1.3861496448516846,
      "step": 76811
    },
    {
      "epoch": 2.6807785034179687e-06,
      "step": 76811,
      "training_step_time": 6.800203084945679
    },
    {
      "epoch": 2.681732177734375e-06,
      "model_forward_time": 1.749499797821045,
      "step": 76812
    },
    {
      "epoch": 2.681732177734375e-06,
      "step": 76812,
      "training_step_time": 7.660028696060181
    },
    {
      "epoch": 2.6826858520507814e-06,
      "model_forward_time": 2.402982711791992,
      "step": 76813
    },
    {
      "epoch": 2.6826858520507814e-06,
      "step": 76813,
      "training_step_time": 9.443394184112549
    },
    {
      "epoch": 2.6836395263671876e-06,
      "model_forward_time": 1.7510666847229004,
      "step": 76814
    },
    {
      "epoch": 2.6836395263671876e-06,
      "step": 76814,
      "training_step_time": 7.614694356918335
    },
    {
      "epoch": 2.6845932006835937e-06,
      "model_forward_time": 1.7600822448730469,
      "step": 76815
    },
    {
      "epoch": 2.6845932006835937e-06,
      "step": 76815,
      "training_step_time": 7.586238861083984
    },
    {
      "epoch": 2.685546875e-06,
      "model_forward_time": 1.7532768249511719,
      "step": 76816
    },
    {
      "epoch": 2.685546875e-06,
      "step": 76816,
      "training_step_time": 8.636441946029663
    },
    {
      "epoch": 2.6865005493164064e-06,
      "model_forward_time": 1.7591683864593506,
      "step": 76817
    },
    {
      "epoch": 2.6865005493164064e-06,
      "step": 76817,
      "training_step_time": 6.797076225280762
    },
    {
      "epoch": 2.6874542236328126e-06,
      "model_forward_time": 1.7449591159820557,
      "step": 76818
    },
    {
      "epoch": 2.6874542236328126e-06,
      "step": 76818,
      "training_step_time": 6.799915075302124
    },
    {
      "epoch": 2.6884078979492187e-06,
      "model_forward_time": 1.749086856842041,
      "step": 76819
    },
    {
      "epoch": 2.6884078979492187e-06,
      "step": 76819,
      "training_step_time": 6.798814535140991
    },
    {
      "epoch": 2.689361572265625e-06,
      "grad_norm": 0.08693161606788635,
      "learning_rate": 1.398569571010252e-06,
      "loss": 0.0545,
      "step": 76820
    },
    {
      "epoch": 2.689361572265625e-06,
      "model_forward_time": 1.750690221786499,
      "step": 76820
    },
    {
      "dynamics_loss_avg": 0.04363012947142124,
      "epoch": 2.689361572265625e-06,
      "step": 76820
    },
    {
      "action_loss_avg": 0.0008466140483506024,
      "epoch": 2.689361572265625e-06,
      "step": 76820
    },
    {
      "epoch": 2.689361572265625e-06,
      "step": 76820,
      "training_step_time": 6.8251447677612305
    },
    {
      "epoch": 2.6903152465820314e-06,
      "model_forward_time": 1.7504198551177979,
      "step": 76821
    },
    {
      "epoch": 2.6903152465820314e-06,
      "step": 76821,
      "training_step_time": 8.316328763961792
    },
    {
      "epoch": 2.6912689208984376e-06,
      "model_forward_time": 2.4051084518432617,
      "step": 76822
    },
    {
      "epoch": 2.6912689208984376e-06,
      "step": 76822,
      "training_step_time": 8.348845720291138
    },
    {
      "epoch": 2.6922225952148437e-06,
      "model_forward_time": 1.752429723739624,
      "step": 76823
    },
    {
      "epoch": 2.6922225952148437e-06,
      "step": 76823,
      "training_step_time": 8.358253717422485
    },
    {
      "epoch": 2.69317626953125e-06,
      "model_forward_time": 1.7507450580596924,
      "step": 76824
    },
    {
      "epoch": 2.69317626953125e-06,
      "step": 76824,
      "training_step_time": 7.501469850540161
    },
    {
      "epoch": 2.6941299438476564e-06,
      "model_forward_time": 1.7488377094268799,
      "step": 76825
    },
    {
      "epoch": 2.6941299438476564e-06,
      "step": 76825,
      "training_step_time": 7.617992877960205
    },
    {
      "epoch": 2.6950836181640626e-06,
      "model_forward_time": 1.7550427913665771,
      "step": 76826
    },
    {
      "epoch": 2.6950836181640626e-06,
      "step": 76826,
      "training_step_time": 7.316067934036255
    },
    {
      "epoch": 2.6960372924804687e-06,
      "model_forward_time": 1.757211446762085,
      "step": 76827
    },
    {
      "epoch": 2.6960372924804687e-06,
      "step": 76827,
      "training_step_time": 6.742856979370117
    },
    {
      "epoch": 2.696990966796875e-06,
      "model_forward_time": 1.756760835647583,
      "step": 76828
    },
    {
      "epoch": 2.696990966796875e-06,
      "step": 76828,
      "training_step_time": 6.748377799987793
    },
    {
      "epoch": 2.6979446411132814e-06,
      "model_forward_time": 1.7404558658599854,
      "step": 76829
    },
    {
      "epoch": 2.6979446411132814e-06,
      "step": 76829,
      "training_step_time": 6.758462190628052
    },
    {
      "epoch": 2.6988983154296876e-06,
      "grad_norm": 0.12202449142932892,
      "learning_rate": 1.3974227937642416e-06,
      "loss": 0.054,
      "step": 76830
    },
    {
      "epoch": 2.6988983154296876e-06,
      "model_forward_time": 1.76690673828125,
      "step": 76830
    },
    {
      "dynamics_loss_avg": 0.042476820200681685,
      "epoch": 2.6988983154296876e-06,
      "step": 76830
    },
    {
      "action_loss_avg": 0.0010883440816542133,
      "epoch": 2.6988983154296876e-06,
      "step": 76830
    },
    {
      "epoch": 2.6988983154296876e-06,
      "step": 76830,
      "training_step_time": 6.745108127593994
    },
    {
      "epoch": 2.6998519897460937e-06,
      "model_forward_time": 2.4085729122161865,
      "step": 76831
    },
    {
      "epoch": 2.6998519897460937e-06,
      "step": 76831,
      "training_step_time": 9.488556385040283
    },
    {
      "epoch": 2.7008056640625e-06,
      "model_forward_time": 1.7549917697906494,
      "step": 76832
    },
    {
      "epoch": 2.7008056640625e-06,
      "step": 76832,
      "training_step_time": 8.327857732772827
    },
    {
      "epoch": 2.7017593383789064e-06,
      "model_forward_time": 1.749783992767334,
      "step": 76833
    },
    {
      "epoch": 2.7017593383789064e-06,
      "step": 76833,
      "training_step_time": 8.353848934173584
    },
    {
      "epoch": 2.7027130126953126e-06,
      "model_forward_time": 1.747112512588501,
      "step": 76834
    },
    {
      "epoch": 2.7027130126953126e-06,
      "step": 76834,
      "training_step_time": 8.255177736282349
    },
    {
      "epoch": 2.7036666870117187e-06,
      "model_forward_time": 1.7486863136291504,
      "step": 76835
    },
    {
      "epoch": 2.7036666870117187e-06,
      "step": 76835,
      "training_step_time": 7.144214868545532
    },
    {
      "epoch": 2.704620361328125e-06,
      "model_forward_time": 1.752641201019287,
      "step": 76836
    },
    {
      "epoch": 2.704620361328125e-06,
      "step": 76836,
      "training_step_time": 6.725924015045166
    },
    {
      "epoch": 2.7055740356445314e-06,
      "model_forward_time": 1.7523303031921387,
      "step": 76837
    },
    {
      "epoch": 2.7055740356445314e-06,
      "step": 76837,
      "training_step_time": 6.7443554401397705
    },
    {
      "epoch": 2.7065277099609376e-06,
      "model_forward_time": 1.7558224201202393,
      "step": 76838
    },
    {
      "epoch": 2.7065277099609376e-06,
      "step": 76838,
      "training_step_time": 6.715627670288086
    },
    {
      "epoch": 2.7074813842773437e-06,
      "model_forward_time": 1.7359936237335205,
      "step": 76839
    },
    {
      "epoch": 2.7074813842773437e-06,
      "step": 76839,
      "training_step_time": 6.716498374938965
    },
    {
      "epoch": 2.70843505859375e-06,
      "grad_norm": 0.07776050269603729,
      "learning_rate": 1.3962764104906596e-06,
      "loss": 0.0539,
      "step": 76840
    },
    {
      "epoch": 2.70843505859375e-06,
      "model_forward_time": 2.0099408626556396,
      "step": 76840
    },
    {
      "dynamics_loss_avg": 0.051039747148752215,
      "epoch": 2.70843505859375e-06,
      "step": 76840
    },
    {
      "action_loss_avg": 0.0011401836120057852,
      "epoch": 2.70843505859375e-06,
      "step": 76840
    },
    {
      "epoch": 2.70843505859375e-06,
      "step": 76840,
      "training_step_time": 9.448416709899902
    },
    {
      "epoch": 2.7093887329101564e-06,
      "model_forward_time": 1.7666821479797363,
      "step": 76841
    },
    {
      "epoch": 2.7093887329101564e-06,
      "step": 76841,
      "training_step_time": 8.557401657104492
    },
    {
      "epoch": 2.7103424072265626e-06,
      "model_forward_time": 1.7512469291687012,
      "step": 76842
    },
    {
      "epoch": 2.7103424072265626e-06,
      "step": 76842,
      "training_step_time": 7.606292724609375
    },
    {
      "epoch": 2.7112960815429687e-06,
      "model_forward_time": 1.7501447200775146,
      "step": 76843
    },
    {
      "epoch": 2.7112960815429687e-06,
      "step": 76843,
      "training_step_time": 7.695774078369141
    },
    {
      "epoch": 2.712249755859375e-06,
      "model_forward_time": 1.7512779235839844,
      "step": 76844
    },
    {
      "epoch": 2.712249755859375e-06,
      "step": 76844,
      "training_step_time": 7.467695951461792
    },
    {
      "epoch": 2.7132034301757814e-06,
      "model_forward_time": 1.7388899326324463,
      "step": 76845
    },
    {
      "epoch": 2.7132034301757814e-06,
      "step": 76845,
      "training_step_time": 6.805519104003906
    },
    {
      "epoch": 2.7141571044921876e-06,
      "model_forward_time": 1.7578420639038086,
      "step": 76846
    },
    {
      "epoch": 2.7141571044921876e-06,
      "step": 76846,
      "training_step_time": 6.92969274520874
    },
    {
      "epoch": 2.7151107788085937e-06,
      "model_forward_time": 1.7572603225708008,
      "step": 76847
    },
    {
      "epoch": 2.7151107788085937e-06,
      "step": 76847,
      "training_step_time": 6.892622232437134
    },
    {
      "epoch": 2.716064453125e-06,
      "model_forward_time": 1.385695457458496,
      "step": 76848
    },
    {
      "epoch": 2.716064453125e-06,
      "step": 76848,
      "training_step_time": 6.951297760009766
    },
    {
      "epoch": 2.7170181274414064e-06,
      "model_forward_time": 1.7518157958984375,
      "step": 76849
    },
    {
      "epoch": 2.7170181274414064e-06,
      "step": 76849,
      "training_step_time": 8.800079822540283
    },
    {
      "epoch": 2.7179718017578126e-06,
      "grad_norm": 0.0971066877245903,
      "learning_rate": 1.3951304213148698e-06,
      "loss": 0.0546,
      "step": 76850
    },
    {
      "epoch": 2.7179718017578126e-06,
      "model_forward_time": 2.467237949371338,
      "step": 76850
    },
    {
      "dynamics_loss_avg": 0.06250914633274078,
      "epoch": 2.7179718017578126e-06,
      "step": 76850
    },
    {
      "action_loss_avg": 0.0015359558310592547,
      "epoch": 2.7179718017578126e-06,
      "step": 76850
    },
    {
      "epoch": 2.7179718017578126e-06,
      "step": 76850,
      "training_step_time": 9.582409143447876
    },
    {
      "epoch": 2.7189254760742187e-06,
      "model_forward_time": 1.7521750926971436,
      "step": 76851
    },
    {
      "epoch": 2.7189254760742187e-06,
      "step": 76851,
      "training_step_time": 7.7531328201293945
    },
    {
      "epoch": 2.719879150390625e-06,
      "model_forward_time": 1.752530574798584,
      "step": 76852
    },
    {
      "epoch": 2.719879150390625e-06,
      "step": 76852,
      "training_step_time": 7.598806619644165
    },
    {
      "epoch": 2.7208328247070314e-06,
      "model_forward_time": 1.7540078163146973,
      "step": 76853
    },
    {
      "epoch": 2.7208328247070314e-06,
      "step": 76853,
      "training_step_time": 7.628425598144531
    },
    {
      "epoch": 2.7217864990234376e-06,
      "model_forward_time": 1.756380319595337,
      "step": 76854
    },
    {
      "epoch": 2.7217864990234376e-06,
      "step": 76854,
      "training_step_time": 7.2408599853515625
    },
    {
      "epoch": 2.7227401733398437e-06,
      "model_forward_time": 1.7533197402954102,
      "step": 76855
    },
    {
      "epoch": 2.7227401733398437e-06,
      "step": 76855,
      "training_step_time": 6.773788690567017
    },
    {
      "epoch": 2.72369384765625e-06,
      "model_forward_time": 1.756603717803955,
      "step": 76856
    },
    {
      "epoch": 2.72369384765625e-06,
      "step": 76856,
      "training_step_time": 6.783403635025024
    },
    {
      "epoch": 2.7246475219726564e-06,
      "model_forward_time": 1.7487750053405762,
      "step": 76857
    },
    {
      "epoch": 2.7246475219726564e-06,
      "step": 76857,
      "training_step_time": 6.658128499984741
    },
    {
      "epoch": 2.7256011962890626e-06,
      "model_forward_time": 1.748673439025879,
      "step": 76858
    },
    {
      "epoch": 2.7256011962890626e-06,
      "step": 76858,
      "training_step_time": 9.564000606536865
    },
    {
      "epoch": 2.7265548706054687e-06,
      "model_forward_time": 2.362584352493286,
      "step": 76859
    },
    {
      "epoch": 2.7265548706054687e-06,
      "step": 76859,
      "training_step_time": 11.532755851745605
    },
    {
      "epoch": 2.727508544921875e-06,
      "grad_norm": 0.08801265805959702,
      "learning_rate": 1.393984826362198e-06,
      "loss": 0.0545,
      "step": 76860
    },
    {
      "epoch": 2.727508544921875e-06,
      "model_forward_time": 1.778852939605713,
      "step": 76860
    },
    {
      "dynamics_loss_avg": 0.052899541519582274,
      "epoch": 2.727508544921875e-06,
      "step": 76860
    },
    {
      "action_loss_avg": 0.0015304554457543419,
      "epoch": 2.727508544921875e-06,
      "step": 76860
    },
    {
      "epoch": 2.727508544921875e-06,
      "step": 76860,
      "training_step_time": 7.717624187469482
    },
    {
      "epoch": 2.7284622192382814e-06,
      "model_forward_time": 1.751471757888794,
      "step": 76861
    },
    {
      "epoch": 2.7284622192382814e-06,
      "step": 76861,
      "training_step_time": 7.572577953338623
    },
    {
      "epoch": 2.7294158935546876e-06,
      "model_forward_time": 1.7518219947814941,
      "step": 76862
    },
    {
      "epoch": 2.7294158935546876e-06,
      "step": 76862,
      "training_step_time": 7.682873010635376
    },
    {
      "epoch": 2.7303695678710937e-06,
      "model_forward_time": 1.7516522407531738,
      "step": 76863
    },
    {
      "epoch": 2.7303695678710937e-06,
      "step": 76863,
      "training_step_time": 7.208000183105469
    },
    {
      "epoch": 2.7313232421875e-06,
      "model_forward_time": 1.74969482421875,
      "step": 76864
    },
    {
      "epoch": 2.7313232421875e-06,
      "step": 76864,
      "training_step_time": 6.804925441741943
    },
    {
      "epoch": 2.7322769165039065e-06,
      "model_forward_time": 1.7454638481140137,
      "step": 76865
    },
    {
      "epoch": 2.7322769165039065e-06,
      "step": 76865,
      "training_step_time": 6.7586259841918945
    },
    {
      "epoch": 2.7332305908203126e-06,
      "model_forward_time": 1.3792319297790527,
      "step": 76866
    },
    {
      "epoch": 2.7332305908203126e-06,
      "step": 76866,
      "training_step_time": 6.739631652832031
    },
    {
      "epoch": 2.7341842651367187e-06,
      "model_forward_time": 1.7601139545440674,
      "step": 76867
    },
    {
      "epoch": 2.7341842651367187e-06,
      "step": 76867,
      "training_step_time": 8.713546752929688
    },
    {
      "epoch": 2.735137939453125e-06,
      "model_forward_time": 2.385625123977661,
      "step": 76868
    },
    {
      "epoch": 2.735137939453125e-06,
      "step": 76868,
      "training_step_time": 10.43443512916565
    },
    {
      "epoch": 2.7360916137695315e-06,
      "model_forward_time": 1.7796878814697266,
      "step": 76869
    },
    {
      "epoch": 2.7360916137695315e-06,
      "step": 76869,
      "training_step_time": 8.509905099868774
    },
    {
      "epoch": 2.7370452880859376e-06,
      "grad_norm": 0.08555246144533157,
      "learning_rate": 1.3928396257579218e-06,
      "loss": 0.0547,
      "step": 76870
    },
    {
      "epoch": 2.7370452880859376e-06,
      "model_forward_time": 1.7452826499938965,
      "step": 76870
    },
    {
      "dynamics_loss_avg": 0.05242985282093286,
      "epoch": 2.7370452880859376e-06,
      "step": 76870
    },
    {
      "action_loss_avg": 0.0017400776065187528,
      "epoch": 2.7370452880859376e-06,
      "step": 76870
    },
    {
      "epoch": 2.7370452880859376e-06,
      "step": 76870,
      "training_step_time": 7.5902955532073975
    },
    {
      "epoch": 2.7379989624023437e-06,
      "model_forward_time": 1.75,
      "step": 76871
    },
    {
      "epoch": 2.7379989624023437e-06,
      "step": 76871,
      "training_step_time": 7.171858787536621
    },
    {
      "epoch": 2.73895263671875e-06,
      "model_forward_time": 1.377241611480713,
      "step": 76872
    },
    {
      "epoch": 2.73895263671875e-06,
      "step": 76872,
      "training_step_time": 8.23845911026001
    },
    {
      "epoch": 2.7399063110351565e-06,
      "model_forward_time": 1.7500319480895996,
      "step": 76873
    },
    {
      "epoch": 2.7399063110351565e-06,
      "step": 76873,
      "training_step_time": 6.751825332641602
    },
    {
      "epoch": 2.7408599853515626e-06,
      "model_forward_time": 1.7463743686676025,
      "step": 76874
    },
    {
      "epoch": 2.7408599853515626e-06,
      "step": 76874,
      "training_step_time": 6.752960205078125
    },
    {
      "epoch": 2.7418136596679688e-06,
      "model_forward_time": 1.3788888454437256,
      "step": 76875
    },
    {
      "epoch": 2.7418136596679688e-06,
      "step": 76875,
      "training_step_time": 6.802184820175171
    },
    {
      "epoch": 2.742767333984375e-06,
      "model_forward_time": 1.7550978660583496,
      "step": 76876
    },
    {
      "epoch": 2.742767333984375e-06,
      "step": 76876,
      "training_step_time": 8.175944328308105
    },
    {
      "epoch": 2.743721008300781e-06,
      "model_forward_time": 1.7767822742462158,
      "step": 76877
    },
    {
      "epoch": 2.743721008300781e-06,
      "step": 76877,
      "training_step_time": 9.490068197250366
    },
    {
      "epoch": 2.7446746826171876e-06,
      "model_forward_time": 2.424767017364502,
      "step": 76878
    },
    {
      "epoch": 2.7446746826171876e-06,
      "step": 76878,
      "training_step_time": 7.7339768409729
    },
    {
      "epoch": 2.7456283569335938e-06,
      "model_forward_time": 1.7498664855957031,
      "step": 76879
    },
    {
      "epoch": 2.7456283569335938e-06,
      "step": 76879,
      "training_step_time": 7.565210819244385
    },
    {
      "epoch": 2.74658203125e-06,
      "grad_norm": 0.15085045993328094,
      "learning_rate": 1.3916948196272833e-06,
      "loss": 0.0555,
      "step": 76880
    },
    {
      "epoch": 2.74658203125e-06,
      "model_forward_time": 1.7457573413848877,
      "step": 76880
    },
    {
      "dynamics_loss_avg": 0.0570290545001626,
      "epoch": 2.74658203125e-06,
      "step": 76880
    },
    {
      "action_loss_avg": 0.0017178298527142032,
      "epoch": 2.74658203125e-06,
      "step": 76880
    },
    {
      "epoch": 2.74658203125e-06,
      "step": 76880,
      "training_step_time": 7.589087963104248
    },
    {
      "epoch": 2.747535705566406e-06,
      "model_forward_time": 1.7489278316497803,
      "step": 76881
    },
    {
      "epoch": 2.747535705566406e-06,
      "step": 76881,
      "training_step_time": 7.49466609954834
    },
    {
      "epoch": 2.7484893798828126e-06,
      "model_forward_time": 1.75514554977417,
      "step": 76882
    },
    {
      "epoch": 2.7484893798828126e-06,
      "step": 76882,
      "training_step_time": 7.205058574676514
    },
    {
      "epoch": 2.7494430541992188e-06,
      "model_forward_time": 1.748915672302246,
      "step": 76883
    },
    {
      "epoch": 2.7494430541992188e-06,
      "step": 76883,
      "training_step_time": 6.868541240692139
    },
    {
      "epoch": 2.750396728515625e-06,
      "model_forward_time": 1.7526195049285889,
      "step": 76884
    },
    {
      "epoch": 2.750396728515625e-06,
      "step": 76884,
      "training_step_time": 6.715842247009277
    },
    {
      "epoch": 2.751350402832031e-06,
      "model_forward_time": 1.7541515827178955,
      "step": 76885
    },
    {
      "epoch": 2.751350402832031e-06,
      "step": 76885,
      "training_step_time": 8.759440422058105
    },
    {
      "epoch": 2.7523040771484376e-06,
      "model_forward_time": 1.3817672729492188,
      "step": 76886
    },
    {
      "epoch": 2.7523040771484376e-06,
      "step": 76886,
      "training_step_time": 8.261543035507202
    },
    {
      "epoch": 2.7532577514648438e-06,
      "model_forward_time": 2.4607245922088623,
      "step": 76887
    },
    {
      "epoch": 2.7532577514648438e-06,
      "step": 76887,
      "training_step_time": 7.736087799072266
    },
    {
      "epoch": 2.75421142578125e-06,
      "model_forward_time": 1.7495701313018799,
      "step": 76888
    },
    {
      "epoch": 2.75421142578125e-06,
      "step": 76888,
      "training_step_time": 7.604865312576294
    },
    {
      "epoch": 2.755165100097656e-06,
      "model_forward_time": 1.7514193058013916,
      "step": 76889
    },
    {
      "epoch": 2.755165100097656e-06,
      "step": 76889,
      "training_step_time": 7.5701868534088135
    },
    {
      "epoch": 2.7561187744140626e-06,
      "grad_norm": 0.1564263105392456,
      "learning_rate": 1.3905504080954735e-06,
      "loss": 0.0541,
      "step": 76890
    },
    {
      "epoch": 2.7561187744140626e-06,
      "model_forward_time": 1.3687539100646973,
      "step": 76890
    },
    {
      "dynamics_loss_avg": 0.04372951928526163,
      "epoch": 2.7561187744140626e-06,
      "step": 76890
    },
    {
      "action_loss_avg": 0.0010616166458930821,
      "epoch": 2.7561187744140626e-06,
      "step": 76890
    },
    {
      "epoch": 2.7561187744140626e-06,
      "step": 76890,
      "training_step_time": 7.5387609004974365
    },
    {
      "epoch": 2.7570724487304688e-06,
      "model_forward_time": 1.7764177322387695,
      "step": 76891
    },
    {
      "epoch": 2.7570724487304688e-06,
      "step": 76891,
      "training_step_time": 7.214615106582642
    },
    {
      "epoch": 2.758026123046875e-06,
      "model_forward_time": 1.7655699253082275,
      "step": 76892
    },
    {
      "epoch": 2.758026123046875e-06,
      "step": 76892,
      "training_step_time": 6.744435787200928
    },
    {
      "epoch": 2.758979797363281e-06,
      "model_forward_time": 1.7692556381225586,
      "step": 76893
    },
    {
      "epoch": 2.758979797363281e-06,
      "step": 76893,
      "training_step_time": 6.758198022842407
    },
    {
      "epoch": 2.7599334716796876e-06,
      "model_forward_time": 1.7823119163513184,
      "step": 76894
    },
    {
      "epoch": 2.7599334716796876e-06,
      "step": 76894,
      "training_step_time": 7.418285846710205
    },
    {
      "epoch": 2.7608871459960938e-06,
      "model_forward_time": 1.7723309993743896,
      "step": 76895
    },
    {
      "epoch": 2.7608871459960938e-06,
      "step": 76895,
      "training_step_time": 9.196507453918457
    },
    {
      "epoch": 2.7618408203125e-06,
      "model_forward_time": 2.9174909591674805,
      "step": 76896
    },
    {
      "epoch": 2.7618408203125e-06,
      "step": 76896,
      "training_step_time": 10.44512677192688
    },
    {
      "epoch": 2.762794494628906e-06,
      "model_forward_time": 1.7625908851623535,
      "step": 76897
    },
    {
      "epoch": 2.762794494628906e-06,
      "step": 76897,
      "training_step_time": 7.727485656738281
    },
    {
      "epoch": 2.7637481689453126e-06,
      "model_forward_time": 1.7746217250823975,
      "step": 76898
    },
    {
      "epoch": 2.7637481689453126e-06,
      "step": 76898,
      "training_step_time": 7.560816049575806
    },
    {
      "epoch": 2.7647018432617188e-06,
      "model_forward_time": 1.8099815845489502,
      "step": 76899
    },
    {
      "epoch": 2.7647018432617188e-06,
      "step": 76899,
      "training_step_time": 8.409867286682129
    },
    {
      "epoch": 2.765655517578125e-06,
      "grad_norm": 0.09581544995307922,
      "learning_rate": 1.3894063912876454e-06,
      "loss": 0.0541,
      "step": 76900
    },
    {
      "epoch": 2.765655517578125e-06,
      "model_forward_time": 1.7495992183685303,
      "step": 76900
    },
    {
      "dynamics_loss_avg": 0.04468934033066034,
      "epoch": 2.765655517578125e-06,
      "step": 76900
    },
    {
      "action_loss_avg": 0.0012341328721959144,
      "epoch": 2.765655517578125e-06,
      "step": 76900
    },
    {
      "epoch": 2.765655517578125e-06,
      "step": 76900,
      "training_step_time": 7.1916773319244385
    },
    {
      "epoch": 2.766609191894531e-06,
      "model_forward_time": 1.8012540340423584,
      "step": 76901
    },
    {
      "epoch": 2.766609191894531e-06,
      "step": 76901,
      "training_step_time": 6.761747598648071
    },
    {
      "epoch": 2.7675628662109376e-06,
      "model_forward_time": 1.794177532196045,
      "step": 76902
    },
    {
      "epoch": 2.7675628662109376e-06,
      "step": 76902,
      "training_step_time": 6.74315071105957
    },
    {
      "epoch": 2.7685165405273438e-06,
      "model_forward_time": 1.4208438396453857,
      "step": 76903
    },
    {
      "epoch": 2.7685165405273438e-06,
      "step": 76903,
      "training_step_time": 7.440229415893555
    },
    {
      "epoch": 2.76947021484375e-06,
      "model_forward_time": 1.769550085067749,
      "step": 76904
    },
    {
      "epoch": 2.76947021484375e-06,
      "step": 76904,
      "training_step_time": 9.626161098480225
    },
    {
      "epoch": 2.770423889160156e-06,
      "model_forward_time": 1.7571029663085938,
      "step": 76905
    },
    {
      "epoch": 2.770423889160156e-06,
      "step": 76905,
      "training_step_time": 8.258280754089355
    },
    {
      "epoch": 2.7713775634765626e-06,
      "model_forward_time": 2.884286403656006,
      "step": 76906
    },
    {
      "epoch": 2.7713775634765626e-06,
      "step": 76906,
      "training_step_time": 7.698497772216797
    },
    {
      "epoch": 2.7723312377929688e-06,
      "model_forward_time": 1.7623677253723145,
      "step": 76907
    },
    {
      "epoch": 2.7723312377929688e-06,
      "step": 76907,
      "training_step_time": 7.588355541229248
    },
    {
      "epoch": 2.773284912109375e-06,
      "model_forward_time": 1.7732293605804443,
      "step": 76908
    },
    {
      "epoch": 2.773284912109375e-06,
      "step": 76908,
      "training_step_time": 7.619922876358032
    },
    {
      "epoch": 2.774238586425781e-06,
      "model_forward_time": 1.7812488079071045,
      "step": 76909
    },
    {
      "epoch": 2.774238586425781e-06,
      "step": 76909,
      "training_step_time": 7.488414525985718
    },
    {
      "epoch": 2.7751922607421876e-06,
      "grad_norm": 0.08099982142448425,
      "learning_rate": 1.3882627693289046e-06,
      "loss": 0.0552,
      "step": 76910
    },
    {
      "epoch": 2.7751922607421876e-06,
      "model_forward_time": 1.7645149230957031,
      "step": 76910
    },
    {
      "dynamics_loss_avg": 0.05435297228395939,
      "epoch": 2.7751922607421876e-06,
      "step": 76910
    },
    {
      "action_loss_avg": 0.0014956368366256356,
      "epoch": 2.7751922607421876e-06,
      "step": 76910
    },
    {
      "epoch": 2.7751922607421876e-06,
      "step": 76910,
      "training_step_time": 7.194201469421387
    },
    {
      "epoch": 2.7761459350585938e-06,
      "model_forward_time": 1.7400000095367432,
      "step": 76911
    },
    {
      "epoch": 2.7761459350585938e-06,
      "step": 76911,
      "training_step_time": 6.7320051193237305
    },
    {
      "epoch": 2.777099609375e-06,
      "model_forward_time": 1.7661471366882324,
      "step": 76912
    },
    {
      "epoch": 2.777099609375e-06,
      "step": 76912,
      "training_step_time": 8.002209901809692
    },
    {
      "epoch": 2.778053283691406e-06,
      "model_forward_time": 1.7710177898406982,
      "step": 76913
    },
    {
      "epoch": 2.778053283691406e-06,
      "step": 76913,
      "training_step_time": 7.644237279891968
    },
    {
      "epoch": 2.7790069580078126e-06,
      "model_forward_time": 1.7648537158966064,
      "step": 76914
    },
    {
      "epoch": 2.7790069580078126e-06,
      "step": 76914,
      "training_step_time": 9.240638256072998
    },
    {
      "epoch": 2.7799606323242188e-06,
      "model_forward_time": 2.943341016769409,
      "step": 76915
    },
    {
      "epoch": 2.7799606323242188e-06,
      "step": 76915,
      "training_step_time": 8.41766095161438
    },
    {
      "epoch": 2.780914306640625e-06,
      "model_forward_time": 1.768204689025879,
      "step": 76916
    },
    {
      "epoch": 2.780914306640625e-06,
      "step": 76916,
      "training_step_time": 8.511892557144165
    },
    {
      "epoch": 2.781867980957031e-06,
      "model_forward_time": 1.7607061862945557,
      "step": 76917
    },
    {
      "epoch": 2.781867980957031e-06,
      "step": 76917,
      "training_step_time": 7.791807413101196
    },
    {
      "epoch": 2.7828216552734376e-06,
      "model_forward_time": 1.782623052597046,
      "step": 76918
    },
    {
      "epoch": 2.7828216552734376e-06,
      "step": 76918,
      "training_step_time": 7.863463640213013
    },
    {
      "epoch": 2.7837753295898438e-06,
      "model_forward_time": 1.7645678520202637,
      "step": 76919
    },
    {
      "epoch": 2.7837753295898438e-06,
      "step": 76919,
      "training_step_time": 7.297199487686157
    },
    {
      "epoch": 2.78472900390625e-06,
      "grad_norm": 0.11592309176921844,
      "learning_rate": 1.3871195423443184e-06,
      "loss": 0.0534,
      "step": 76920
    },
    {
      "epoch": 2.78472900390625e-06,
      "model_forward_time": 1.7714931964874268,
      "step": 76920
    },
    {
      "dynamics_loss_avg": 0.046879427134990694,
      "epoch": 2.78472900390625e-06,
      "step": 76920
    },
    {
      "action_loss_avg": 0.000976981726125814,
      "epoch": 2.78472900390625e-06,
      "step": 76920
    },
    {
      "epoch": 2.78472900390625e-06,
      "step": 76920,
      "training_step_time": 6.779281139373779
    },
    {
      "epoch": 2.785682678222656e-06,
      "model_forward_time": 1.7913472652435303,
      "step": 76921
    },
    {
      "epoch": 2.785682678222656e-06,
      "step": 76921,
      "training_step_time": 7.3697874546051025
    },
    {
      "epoch": 2.7866363525390626e-06,
      "model_forward_time": 1.7712419033050537,
      "step": 76922
    },
    {
      "epoch": 2.7866363525390626e-06,
      "step": 76922,
      "training_step_time": 7.706902503967285
    },
    {
      "epoch": 2.787590026855469e-06,
      "model_forward_time": 1.770533561706543,
      "step": 76923
    },
    {
      "epoch": 2.787590026855469e-06,
      "step": 76923,
      "training_step_time": 11.677196264266968
    },
    {
      "epoch": 2.788543701171875e-06,
      "model_forward_time": 2.9263389110565186,
      "step": 76924
    },
    {
      "epoch": 2.788543701171875e-06,
      "step": 76924,
      "training_step_time": 9.223320484161377
    },
    {
      "epoch": 2.789497375488281e-06,
      "model_forward_time": 1.7657935619354248,
      "step": 76925
    },
    {
      "epoch": 2.789497375488281e-06,
      "step": 76925,
      "training_step_time": 7.533595561981201
    },
    {
      "epoch": 2.7904510498046876e-06,
      "model_forward_time": 1.7497212886810303,
      "step": 76926
    },
    {
      "epoch": 2.7904510498046876e-06,
      "step": 76926,
      "training_step_time": 7.692078113555908
    },
    {
      "epoch": 2.791404724121094e-06,
      "model_forward_time": 1.7517316341400146,
      "step": 76927
    },
    {
      "epoch": 2.791404724121094e-06,
      "step": 76927,
      "training_step_time": 7.704740047454834
    },
    {
      "epoch": 2.7923583984375e-06,
      "model_forward_time": 1.74741792678833,
      "step": 76928
    },
    {
      "epoch": 2.7923583984375e-06,
      "step": 76928,
      "training_step_time": 7.147898197174072
    },
    {
      "epoch": 2.793312072753906e-06,
      "model_forward_time": 1.7380826473236084,
      "step": 76929
    },
    {
      "epoch": 2.793312072753906e-06,
      "step": 76929,
      "training_step_time": 6.77241063117981
    },
    {
      "epoch": 2.7942657470703127e-06,
      "grad_norm": 0.07958915829658508,
      "learning_rate": 1.3859767104589044e-06,
      "loss": 0.0545,
      "step": 76930
    },
    {
      "epoch": 2.7942657470703127e-06,
      "model_forward_time": 1.7568755149841309,
      "step": 76930
    },
    {
      "dynamics_loss_avg": 0.04213296035304666,
      "epoch": 2.7942657470703127e-06,
      "step": 76930
    },
    {
      "action_loss_avg": 0.0012133637035731226,
      "epoch": 2.7942657470703127e-06,
      "step": 76930
    },
    {
      "epoch": 2.7942657470703127e-06,
      "step": 76930,
      "training_step_time": 7.3780927658081055
    },
    {
      "epoch": 2.795219421386719e-06,
      "model_forward_time": 1.7653937339782715,
      "step": 76931
    },
    {
      "epoch": 2.795219421386719e-06,
      "step": 76931,
      "training_step_time": 7.702219247817993
    },
    {
      "epoch": 2.796173095703125e-06,
      "model_forward_time": 1.784435510635376,
      "step": 76932
    },
    {
      "epoch": 2.796173095703125e-06,
      "step": 76932,
      "training_step_time": 10.062373876571655
    },
    {
      "epoch": 2.797126770019531e-06,
      "model_forward_time": 2.8527724742889404,
      "step": 76933
    },
    {
      "epoch": 2.797126770019531e-06,
      "step": 76933,
      "training_step_time": 9.027101516723633
    },
    {
      "epoch": 2.7980804443359377e-06,
      "model_forward_time": 1.7529363632202148,
      "step": 76934
    },
    {
      "epoch": 2.7980804443359377e-06,
      "step": 76934,
      "training_step_time": 7.725324630737305
    },
    {
      "epoch": 2.799034118652344e-06,
      "model_forward_time": 1.752692699432373,
      "step": 76935
    },
    {
      "epoch": 2.799034118652344e-06,
      "step": 76935,
      "training_step_time": 7.739239931106567
    },
    {
      "epoch": 2.79998779296875e-06,
      "model_forward_time": 1.7585809230804443,
      "step": 76936
    },
    {
      "epoch": 2.79998779296875e-06,
      "step": 76936,
      "training_step_time": 7.634636402130127
    },
    {
      "epoch": 2.800941467285156e-06,
      "model_forward_time": 1.3892834186553955,
      "step": 76937
    },
    {
      "epoch": 2.800941467285156e-06,
      "step": 76937,
      "training_step_time": 7.14158821105957
    },
    {
      "epoch": 2.8018951416015627e-06,
      "model_forward_time": 1.7558033466339111,
      "step": 76938
    },
    {
      "epoch": 2.8018951416015627e-06,
      "step": 76938,
      "training_step_time": 6.729461193084717
    },
    {
      "epoch": 2.802848815917969e-06,
      "model_forward_time": 1.7541444301605225,
      "step": 76939
    },
    {
      "epoch": 2.802848815917969e-06,
      "step": 76939,
      "training_step_time": 7.491038084030151
    },
    {
      "epoch": 2.803802490234375e-06,
      "grad_norm": 0.08616809546947479,
      "learning_rate": 1.3848342737976467e-06,
      "loss": 0.0533,
      "step": 76940
    },
    {
      "epoch": 2.803802490234375e-06,
      "model_forward_time": 1.7596638202667236,
      "step": 76940
    },
    {
      "dynamics_loss_avg": 0.07338017467409372,
      "epoch": 2.803802490234375e-06,
      "step": 76940
    },
    {
      "action_loss_avg": 0.0019081031787209213,
      "epoch": 2.803802490234375e-06,
      "step": 76940
    },
    {
      "epoch": 2.803802490234375e-06,
      "step": 76940,
      "training_step_time": 7.677947044372559
    },
    {
      "epoch": 2.804756164550781e-06,
      "model_forward_time": 1.7672276496887207,
      "step": 76941
    },
    {
      "epoch": 2.804756164550781e-06,
      "step": 76941,
      "training_step_time": 10.308118104934692
    },
    {
      "epoch": 2.8057098388671877e-06,
      "model_forward_time": 1.7789201736450195,
      "step": 76942
    },
    {
      "epoch": 2.8057098388671877e-06,
      "step": 76942,
      "training_step_time": 9.213515996932983
    },
    {
      "epoch": 2.806663513183594e-06,
      "model_forward_time": 2.882190465927124,
      "step": 76943
    },
    {
      "epoch": 2.806663513183594e-06,
      "step": 76943,
      "training_step_time": 8.750991344451904
    },
    {
      "epoch": 2.8076171875e-06,
      "model_forward_time": 1.7696967124938965,
      "step": 76944
    },
    {
      "epoch": 2.8076171875e-06,
      "step": 76944,
      "training_step_time": 7.735270977020264
    },
    {
      "epoch": 2.808570861816406e-06,
      "model_forward_time": 1.7566120624542236,
      "step": 76945
    },
    {
      "epoch": 2.808570861816406e-06,
      "step": 76945,
      "training_step_time": 7.58290958404541
    },
    {
      "epoch": 2.8095245361328127e-06,
      "model_forward_time": 1.754943609237671,
      "step": 76946
    },
    {
      "epoch": 2.8095245361328127e-06,
      "step": 76946,
      "training_step_time": 7.401358127593994
    },
    {
      "epoch": 2.810478210449219e-06,
      "model_forward_time": 1.761843204498291,
      "step": 76947
    },
    {
      "epoch": 2.810478210449219e-06,
      "step": 76947,
      "training_step_time": 6.807507753372192
    },
    {
      "epoch": 2.811431884765625e-06,
      "model_forward_time": 1.7617263793945312,
      "step": 76948
    },
    {
      "epoch": 2.811431884765625e-06,
      "step": 76948,
      "training_step_time": 7.444966077804565
    },
    {
      "epoch": 2.812385559082031e-06,
      "model_forward_time": 1.7584505081176758,
      "step": 76949
    },
    {
      "epoch": 2.812385559082031e-06,
      "step": 76949,
      "training_step_time": 7.09302020072937
    },
    {
      "epoch": 2.8133392333984377e-06,
      "grad_norm": 0.7279530167579651,
      "learning_rate": 1.3836922324854756e-06,
      "loss": 0.0555,
      "step": 76950
    },
    {
      "epoch": 2.8133392333984377e-06,
      "model_forward_time": 1.7620055675506592,
      "step": 76950
    },
    {
      "dynamics_loss_avg": 0.05526974704116583,
      "epoch": 2.8133392333984377e-06,
      "step": 76950
    },
    {
      "action_loss_avg": 0.0009595691750291735,
      "epoch": 2.8133392333984377e-06,
      "step": 76950
    },
    {
      "epoch": 2.8133392333984377e-06,
      "step": 76950,
      "training_step_time": 9.509066343307495
    },
    {
      "epoch": 2.814292907714844e-06,
      "model_forward_time": 1.7555932998657227,
      "step": 76951
    },
    {
      "epoch": 2.814292907714844e-06,
      "step": 76951,
      "training_step_time": 8.221461057662964
    },
    {
      "epoch": 2.81524658203125e-06,
      "model_forward_time": 2.884032726287842,
      "step": 76952
    },
    {
      "epoch": 2.81524658203125e-06,
      "step": 76952,
      "training_step_time": 8.172055959701538
    },
    {
      "epoch": 2.816200256347656e-06,
      "model_forward_time": 1.7305185794830322,
      "step": 76953
    },
    {
      "epoch": 2.816200256347656e-06,
      "step": 76953,
      "training_step_time": 7.518967390060425
    },
    {
      "epoch": 2.8171539306640627e-06,
      "model_forward_time": 1.758704662322998,
      "step": 76954
    },
    {
      "epoch": 2.8171539306640627e-06,
      "step": 76954,
      "training_step_time": 7.633306264877319
    },
    {
      "epoch": 2.818107604980469e-06,
      "model_forward_time": 1.7559154033660889,
      "step": 76955
    },
    {
      "epoch": 2.818107604980469e-06,
      "step": 76955,
      "training_step_time": 7.588741302490234
    },
    {
      "epoch": 2.819061279296875e-06,
      "model_forward_time": 1.7489221096038818,
      "step": 76956
    },
    {
      "epoch": 2.819061279296875e-06,
      "step": 76956,
      "training_step_time": 7.100234508514404
    },
    {
      "epoch": 2.820014953613281e-06,
      "model_forward_time": 1.7615516185760498,
      "step": 76957
    },
    {
      "epoch": 2.820014953613281e-06,
      "step": 76957,
      "training_step_time": 7.347299814224243
    },
    {
      "epoch": 2.8209686279296877e-06,
      "model_forward_time": 1.758610725402832,
      "step": 76958
    },
    {
      "epoch": 2.8209686279296877e-06,
      "step": 76958,
      "training_step_time": 7.040083408355713
    },
    {
      "epoch": 2.821922302246094e-06,
      "model_forward_time": 1.7508125305175781,
      "step": 76959
    },
    {
      "epoch": 2.821922302246094e-06,
      "step": 76959,
      "training_step_time": 7.780523061752319
    },
    {
      "epoch": 2.8228759765625e-06,
      "grad_norm": 0.10297705978155136,
      "learning_rate": 1.3825505866472871e-06,
      "loss": 0.055,
      "step": 76960
    },
    {
      "epoch": 2.8228759765625e-06,
      "model_forward_time": 1.782698631286621,
      "step": 76960
    },
    {
      "dynamics_loss_avg": 0.04668052345514297,
      "epoch": 2.8228759765625e-06,
      "step": 76960
    },
    {
      "action_loss_avg": 0.001736658107256517,
      "epoch": 2.8228759765625e-06,
      "step": 76960
    },
    {
      "epoch": 2.8228759765625e-06,
      "step": 76960,
      "training_step_time": 8.69465970993042
    },
    {
      "epoch": 2.823829650878906e-06,
      "model_forward_time": 2.9416205883026123,
      "step": 76961
    },
    {
      "epoch": 2.823829650878906e-06,
      "step": 76961,
      "training_step_time": 7.6817567348480225
    },
    {
      "epoch": 2.8247833251953127e-06,
      "model_forward_time": 1.7513463497161865,
      "step": 76962
    },
    {
      "epoch": 2.8247833251953127e-06,
      "step": 76962,
      "training_step_time": 7.5032453536987305
    },
    {
      "epoch": 2.825736999511719e-06,
      "model_forward_time": 1.74690842628479,
      "step": 76963
    },
    {
      "epoch": 2.825736999511719e-06,
      "step": 76963,
      "training_step_time": 7.600010633468628
    },
    {
      "epoch": 2.826690673828125e-06,
      "model_forward_time": 1.7437140941619873,
      "step": 76964
    },
    {
      "epoch": 2.826690673828125e-06,
      "step": 76964,
      "training_step_time": 7.571065664291382
    },
    {
      "epoch": 2.827644348144531e-06,
      "model_forward_time": 1.7559208869934082,
      "step": 76965
    },
    {
      "epoch": 2.827644348144531e-06,
      "step": 76965,
      "training_step_time": 7.149000406265259
    },
    {
      "epoch": 2.8285980224609377e-06,
      "model_forward_time": 1.7572963237762451,
      "step": 76966
    },
    {
      "epoch": 2.8285980224609377e-06,
      "step": 76966,
      "training_step_time": 8.28146243095398
    },
    {
      "epoch": 2.829551696777344e-06,
      "model_forward_time": 1.7437305450439453,
      "step": 76967
    },
    {
      "epoch": 2.829551696777344e-06,
      "step": 76967,
      "training_step_time": 7.555154323577881
    },
    {
      "epoch": 2.83050537109375e-06,
      "model_forward_time": 1.7497761249542236,
      "step": 76968
    },
    {
      "epoch": 2.83050537109375e-06,
      "step": 76968,
      "training_step_time": 8.106275081634521
    },
    {
      "epoch": 2.831459045410156e-06,
      "model_forward_time": 1.781419038772583,
      "step": 76969
    },
    {
      "epoch": 2.831459045410156e-06,
      "step": 76969,
      "training_step_time": 11.040151834487915
    },
    {
      "epoch": 2.8324127197265627e-06,
      "grad_norm": 0.07847770303487778,
      "learning_rate": 1.3814093364079256e-06,
      "loss": 0.054,
      "step": 76970
    },
    {
      "epoch": 2.8324127197265627e-06,
      "model_forward_time": 2.8432748317718506,
      "step": 76970
    },
    {
      "dynamics_loss_avg": 0.056155132781714204,
      "epoch": 2.8324127197265627e-06,
      "step": 76970
    },
    {
      "action_loss_avg": 0.0010498756455490366,
      "epoch": 2.8324127197265627e-06,
      "step": 76970
    },
    {
      "epoch": 2.8324127197265627e-06,
      "step": 76970,
      "training_step_time": 9.156232118606567
    },
    {
      "epoch": 2.833366394042969e-06,
      "model_forward_time": 1.7456958293914795,
      "step": 76971
    },
    {
      "epoch": 2.833366394042969e-06,
      "step": 76971,
      "training_step_time": 9.6134352684021
    },
    {
      "epoch": 2.834320068359375e-06,
      "model_forward_time": 1.7662456035614014,
      "step": 76972
    },
    {
      "epoch": 2.834320068359375e-06,
      "step": 76972,
      "training_step_time": 7.48229718208313
    },
    {
      "epoch": 2.835273742675781e-06,
      "model_forward_time": 1.748248815536499,
      "step": 76973
    },
    {
      "epoch": 2.835273742675781e-06,
      "step": 76973,
      "training_step_time": 7.485750198364258
    },
    {
      "epoch": 2.8362274169921877e-06,
      "model_forward_time": 1.7532298564910889,
      "step": 76974
    },
    {
      "epoch": 2.8362274169921877e-06,
      "step": 76974,
      "training_step_time": 7.095742464065552
    },
    {
      "epoch": 2.837181091308594e-06,
      "model_forward_time": 1.7476236820220947,
      "step": 76975
    },
    {
      "epoch": 2.837181091308594e-06,
      "step": 76975,
      "training_step_time": 7.3205695152282715
    },
    {
      "epoch": 2.838134765625e-06,
      "model_forward_time": 1.758591651916504,
      "step": 76976
    },
    {
      "epoch": 2.838134765625e-06,
      "step": 76976,
      "training_step_time": 7.332611799240112
    },
    {
      "epoch": 2.839088439941406e-06,
      "model_forward_time": 1.7533886432647705,
      "step": 76977
    },
    {
      "epoch": 2.839088439941406e-06,
      "step": 76977,
      "training_step_time": 7.51192045211792
    },
    {
      "epoch": 2.8400421142578127e-06,
      "model_forward_time": 1.7510733604431152,
      "step": 76978
    },
    {
      "epoch": 2.8400421142578127e-06,
      "step": 76978,
      "training_step_time": 7.794693470001221
    },
    {
      "epoch": 2.840995788574219e-06,
      "model_forward_time": 2.863901376724243,
      "step": 76979
    },
    {
      "epoch": 2.840995788574219e-06,
      "step": 76979,
      "training_step_time": 9.131409406661987
    },
    {
      "epoch": 2.841949462890625e-06,
      "grad_norm": 0.08615120500326157,
      "learning_rate": 1.3802684818921991e-06,
      "loss": 0.0542,
      "step": 76980
    },
    {
      "epoch": 2.841949462890625e-06,
      "model_forward_time": 1.736879587173462,
      "step": 76980
    },
    {
      "dynamics_loss_avg": 0.03850695136934519,
      "epoch": 2.841949462890625e-06,
      "step": 76980
    },
    {
      "action_loss_avg": 0.0011519192077685148,
      "epoch": 2.841949462890625e-06,
      "step": 76980
    },
    {
      "epoch": 2.841949462890625e-06,
      "step": 76980,
      "training_step_time": 8.244706153869629
    },
    {
      "epoch": 2.842903137207031e-06,
      "model_forward_time": 1.752021312713623,
      "step": 76981
    },
    {
      "epoch": 2.842903137207031e-06,
      "step": 76981,
      "training_step_time": 8.385177850723267
    },
    {
      "epoch": 2.8438568115234377e-06,
      "model_forward_time": 1.7516803741455078,
      "step": 76982
    },
    {
      "epoch": 2.8438568115234377e-06,
      "step": 76982,
      "training_step_time": 7.532461881637573
    },
    {
      "epoch": 2.844810485839844e-06,
      "model_forward_time": 1.7481882572174072,
      "step": 76983
    },
    {
      "epoch": 2.844810485839844e-06,
      "step": 76983,
      "training_step_time": 7.50551700592041
    },
    {
      "epoch": 2.84576416015625e-06,
      "model_forward_time": 1.7451622486114502,
      "step": 76984
    },
    {
      "epoch": 2.84576416015625e-06,
      "step": 76984,
      "training_step_time": 7.302798748016357
    },
    {
      "epoch": 2.846717834472656e-06,
      "model_forward_time": 1.7499637603759766,
      "step": 76985
    },
    {
      "epoch": 2.846717834472656e-06,
      "step": 76985,
      "training_step_time": 7.428180694580078
    },
    {
      "epoch": 2.8476715087890627e-06,
      "model_forward_time": 1.7533724308013916,
      "step": 76986
    },
    {
      "epoch": 2.8476715087890627e-06,
      "step": 76986,
      "training_step_time": 7.526738405227661
    },
    {
      "epoch": 2.848625183105469e-06,
      "model_forward_time": 1.748380184173584,
      "step": 76987
    },
    {
      "epoch": 2.848625183105469e-06,
      "step": 76987,
      "training_step_time": 10.294805765151978
    },
    {
      "epoch": 2.849578857421875e-06,
      "model_forward_time": 1.7492039203643799,
      "step": 76988
    },
    {
      "epoch": 2.849578857421875e-06,
      "step": 76988,
      "training_step_time": 9.21245551109314
    },
    {
      "epoch": 2.850532531738281e-06,
      "model_forward_time": 2.920454502105713,
      "step": 76989
    },
    {
      "epoch": 2.850532531738281e-06,
      "step": 76989,
      "training_step_time": 8.570555448532104
    },
    {
      "epoch": 2.8514862060546877e-06,
      "grad_norm": 0.08756006509065628,
      "learning_rate": 1.3791280232248688e-06,
      "loss": 0.0549,
      "step": 76990
    },
    {
      "epoch": 2.8514862060546877e-06,
      "model_forward_time": 1.7538783550262451,
      "step": 76990
    },
    {
      "dynamics_loss_avg": 0.05933083277195692,
      "epoch": 2.8514862060546877e-06,
      "step": 76990
    },
    {
      "action_loss_avg": 0.0020504953019553794,
      "epoch": 2.8514862060546877e-06,
      "step": 76990
    },
    {
      "epoch": 2.8514862060546877e-06,
      "step": 76990,
      "training_step_time": 7.575259685516357
    },
    {
      "epoch": 2.852439880371094e-06,
      "model_forward_time": 1.743987798690796,
      "step": 76991
    },
    {
      "epoch": 2.852439880371094e-06,
      "step": 76991,
      "training_step_time": 7.563647747039795
    },
    {
      "epoch": 2.8533935546875e-06,
      "model_forward_time": 1.7482492923736572,
      "step": 76992
    },
    {
      "epoch": 2.8533935546875e-06,
      "step": 76992,
      "training_step_time": 7.489946126937866
    },
    {
      "epoch": 2.854347229003906e-06,
      "model_forward_time": 1.7636325359344482,
      "step": 76993
    },
    {
      "epoch": 2.854347229003906e-06,
      "step": 76993,
      "training_step_time": 7.922471523284912
    },
    {
      "epoch": 2.8553009033203127e-06,
      "model_forward_time": 1.7342162132263184,
      "step": 76994
    },
    {
      "epoch": 2.8553009033203127e-06,
      "step": 76994,
      "training_step_time": 7.2327775955200195
    },
    {
      "epoch": 2.856254577636719e-06,
      "model_forward_time": 1.7592639923095703,
      "step": 76995
    },
    {
      "epoch": 2.856254577636719e-06,
      "step": 76995,
      "training_step_time": 7.469265937805176
    },
    {
      "epoch": 2.857208251953125e-06,
      "model_forward_time": 1.7546768188476562,
      "step": 76996
    },
    {
      "epoch": 2.857208251953125e-06,
      "step": 76996,
      "training_step_time": 8.27710771560669
    },
    {
      "epoch": 2.858161926269531e-06,
      "model_forward_time": 1.7859888076782227,
      "step": 76997
    },
    {
      "epoch": 2.858161926269531e-06,
      "step": 76997,
      "training_step_time": 9.015413284301758
    },
    {
      "epoch": 2.8591156005859377e-06,
      "model_forward_time": 2.99812650680542,
      "step": 76998
    },
    {
      "epoch": 2.8591156005859377e-06,
      "step": 76998,
      "training_step_time": 7.9980080127716064
    },
    {
      "epoch": 2.860069274902344e-06,
      "model_forward_time": 1.766465663909912,
      "step": 76999
    },
    {
      "epoch": 2.860069274902344e-06,
      "step": 76999,
      "training_step_time": 7.8651227951049805
    },
    {
      "epoch": 2.86102294921875e-06,
      "grad_norm": 0.10241471230983734,
      "learning_rate": 1.3779879605306557e-06,
      "loss": 0.056,
      "step": 77000
    },
    {
      "epoch": 2.86102294921875e-06,
      "model_forward_time": 1.7237560749053955,
      "step": 77000
    },
    {
      "dynamics_loss_avg": 0.05182122383266687,
      "epoch": 2.86102294921875e-06,
      "step": 77000
    },
    {
      "action_loss_avg": 0.0009474935417529195,
      "epoch": 2.86102294921875e-06,
      "step": 77000
    },
    {
      "epoch": 2.86102294921875e-06,
      "step": 77000,
      "training_step_time": 6.583348274230957
    },
    {
      "epoch": 2.861976623535156e-06,
      "model_forward_time": 1.7290472984313965,
      "step": 77001
    },
    {
      "epoch": 2.861976623535156e-06,
      "step": 77001,
      "training_step_time": 7.704238414764404
    },
    {
      "epoch": 2.8629302978515623e-06,
      "model_forward_time": 1.7296733856201172,
      "step": 77002
    },
    {
      "epoch": 2.8629302978515623e-06,
      "step": 77002,
      "training_step_time": 8.665669441223145
    },
    {
      "epoch": 2.863883972167969e-06,
      "model_forward_time": 1.7250056266784668,
      "step": 77003
    },
    {
      "epoch": 2.863883972167969e-06,
      "step": 77003,
      "training_step_time": 8.36801791191101
    },
    {
      "epoch": 2.864837646484375e-06,
      "model_forward_time": 1.7403218746185303,
      "step": 77004
    },
    {
      "epoch": 2.864837646484375e-06,
      "step": 77004,
      "training_step_time": 7.754287958145142
    },
    {
      "epoch": 2.865791320800781e-06,
      "model_forward_time": 1.733882188796997,
      "step": 77005
    },
    {
      "epoch": 2.865791320800781e-06,
      "step": 77005,
      "training_step_time": 10.962592363357544
    },
    {
      "epoch": 2.8667449951171873e-06,
      "model_forward_time": 2.5268633365631104,
      "step": 77006
    },
    {
      "epoch": 2.8667449951171873e-06,
      "step": 77006,
      "training_step_time": 12.107329607009888
    },
    {
      "epoch": 2.867698669433594e-06,
      "model_forward_time": 1.8503668308258057,
      "step": 77007
    },
    {
      "epoch": 2.867698669433594e-06,
      "step": 77007,
      "training_step_time": 8.703420400619507
    },
    {
      "epoch": 2.86865234375e-06,
      "model_forward_time": 1.7312748432159424,
      "step": 77008
    },
    {
      "epoch": 2.86865234375e-06,
      "step": 77008,
      "training_step_time": 8.476454973220825
    },
    {
      "epoch": 2.869606018066406e-06,
      "model_forward_time": 1.7364251613616943,
      "step": 77009
    },
    {
      "epoch": 2.869606018066406e-06,
      "step": 77009,
      "training_step_time": 7.531804084777832
    },
    {
      "epoch": 2.8705596923828123e-06,
      "grad_norm": 0.10220266878604889,
      "learning_rate": 1.376848293934232e-06,
      "loss": 0.0561,
      "step": 77010
    },
    {
      "epoch": 2.8705596923828123e-06,
      "model_forward_time": 1.725163459777832,
      "step": 77010
    },
    {
      "dynamics_loss_avg": 0.048494668677449226,
      "epoch": 2.8705596923828123e-06,
      "step": 77010
    },
    {
      "action_loss_avg": 0.00201782651420217,
      "epoch": 2.8705596923828123e-06,
      "step": 77010
    },
    {
      "epoch": 2.8705596923828123e-06,
      "step": 77010,
      "training_step_time": 7.475835800170898
    },
    {
      "epoch": 2.871513366699219e-06,
      "model_forward_time": 1.7271924018859863,
      "step": 77011
    },
    {
      "epoch": 2.871513366699219e-06,
      "step": 77011,
      "training_step_time": 7.433228254318237
    },
    {
      "epoch": 2.872467041015625e-06,
      "model_forward_time": 1.725351095199585,
      "step": 77012
    },
    {
      "epoch": 2.872467041015625e-06,
      "step": 77012,
      "training_step_time": 8.207181692123413
    },
    {
      "epoch": 2.873420715332031e-06,
      "model_forward_time": 1.722611427307129,
      "step": 77013
    },
    {
      "epoch": 2.873420715332031e-06,
      "step": 77013,
      "training_step_time": 7.640723705291748
    },
    {
      "epoch": 2.8743743896484373e-06,
      "model_forward_time": 1.724891185760498,
      "step": 77014
    },
    {
      "epoch": 2.8743743896484373e-06,
      "step": 77014,
      "training_step_time": 8.539244413375854
    },
    {
      "epoch": 2.875328063964844e-06,
      "model_forward_time": 1.7317888736724854,
      "step": 77015
    },
    {
      "epoch": 2.875328063964844e-06,
      "step": 77015,
      "training_step_time": 10.66465973854065
    },
    {
      "epoch": 2.87628173828125e-06,
      "model_forward_time": 2.4773647785186768,
      "step": 77016
    },
    {
      "epoch": 2.87628173828125e-06,
      "step": 77016,
      "training_step_time": 10.266585350036621
    },
    {
      "epoch": 2.877235412597656e-06,
      "model_forward_time": 1.7413520812988281,
      "step": 77017
    },
    {
      "epoch": 2.877235412597656e-06,
      "step": 77017,
      "training_step_time": 8.561946630477905
    },
    {
      "epoch": 2.8781890869140623e-06,
      "model_forward_time": 1.7314767837524414,
      "step": 77018
    },
    {
      "epoch": 2.8781890869140623e-06,
      "step": 77018,
      "training_step_time": 7.683474779129028
    },
    {
      "epoch": 2.879142761230469e-06,
      "model_forward_time": 1.7345590591430664,
      "step": 77019
    },
    {
      "epoch": 2.879142761230469e-06,
      "step": 77019,
      "training_step_time": 7.430795907974243
    },
    {
      "epoch": 2.880096435546875e-06,
      "grad_norm": 0.18944086134433746,
      "learning_rate": 1.3757090235602316e-06,
      "loss": 0.0556,
      "step": 77020
    },
    {
      "epoch": 2.880096435546875e-06,
      "model_forward_time": 1.7225420475006104,
      "step": 77020
    },
    {
      "dynamics_loss_avg": 0.04436060804873705,
      "epoch": 2.880096435546875e-06,
      "step": 77020
    },
    {
      "action_loss_avg": 0.001479188268422149,
      "epoch": 2.880096435546875e-06,
      "step": 77020
    },
    {
      "epoch": 2.880096435546875e-06,
      "step": 77020,
      "training_step_time": 8.417608499526978
    },
    {
      "epoch": 2.881050109863281e-06,
      "model_forward_time": 1.7365612983703613,
      "step": 77021
    },
    {
      "epoch": 2.881050109863281e-06,
      "step": 77021,
      "training_step_time": 7.365118980407715
    },
    {
      "epoch": 2.8820037841796873e-06,
      "model_forward_time": 1.7346880435943604,
      "step": 77022
    },
    {
      "epoch": 2.8820037841796873e-06,
      "step": 77022,
      "training_step_time": 7.4568352699279785
    },
    {
      "epoch": 2.882957458496094e-06,
      "model_forward_time": 1.7254416942596436,
      "step": 77023
    },
    {
      "epoch": 2.882957458496094e-06,
      "step": 77023,
      "training_step_time": 7.480553388595581
    },
    {
      "epoch": 2.8839111328125e-06,
      "model_forward_time": 1.7394111156463623,
      "step": 77024
    },
    {
      "epoch": 2.8839111328125e-06,
      "step": 77024,
      "training_step_time": 8.883785963058472
    },
    {
      "epoch": 2.884864807128906e-06,
      "model_forward_time": 2.426652669906616,
      "step": 77025
    },
    {
      "epoch": 2.884864807128906e-06,
      "step": 77025,
      "training_step_time": 7.608015060424805
    },
    {
      "epoch": 2.8858184814453123e-06,
      "model_forward_time": 1.7427854537963867,
      "step": 77026
    },
    {
      "epoch": 2.8858184814453123e-06,
      "step": 77026,
      "training_step_time": 8.603288888931274
    },
    {
      "epoch": 2.886772155761719e-06,
      "model_forward_time": 1.7351930141448975,
      "step": 77027
    },
    {
      "epoch": 2.886772155761719e-06,
      "step": 77027,
      "training_step_time": 8.220527172088623
    },
    {
      "epoch": 2.887725830078125e-06,
      "model_forward_time": 1.7361557483673096,
      "step": 77028
    },
    {
      "epoch": 2.887725830078125e-06,
      "step": 77028,
      "training_step_time": 7.603447675704956
    },
    {
      "epoch": 2.888679504394531e-06,
      "model_forward_time": 1.7415947914123535,
      "step": 77029
    },
    {
      "epoch": 2.888679504394531e-06,
      "step": 77029,
      "training_step_time": 7.613781690597534
    },
    {
      "epoch": 2.8896331787109373e-06,
      "grad_norm": 0.12127687782049179,
      "learning_rate": 1.3745701495332447e-06,
      "loss": 0.0558,
      "step": 77030
    },
    {
      "epoch": 2.8896331787109373e-06,
      "model_forward_time": 1.730614423751831,
      "step": 77030
    },
    {
      "dynamics_loss_avg": 0.03586185593158007,
      "epoch": 2.8896331787109373e-06,
      "step": 77030
    },
    {
      "action_loss_avg": 0.0008539066417142749,
      "epoch": 2.8896331787109373e-06,
      "step": 77030
    },
    {
      "epoch": 2.8896331787109373e-06,
      "step": 77030,
      "training_step_time": 7.2723705768585205
    },
    {
      "epoch": 2.890586853027344e-06,
      "model_forward_time": 1.7367610931396484,
      "step": 77031
    },
    {
      "epoch": 2.890586853027344e-06,
      "step": 77031,
      "training_step_time": 7.171010971069336
    },
    {
      "epoch": 2.89154052734375e-06,
      "model_forward_time": 1.7356469631195068,
      "step": 77032
    },
    {
      "epoch": 2.89154052734375e-06,
      "step": 77032,
      "training_step_time": 7.753344774246216
    },
    {
      "epoch": 2.892494201660156e-06,
      "model_forward_time": 1.7396149635314941,
      "step": 77033
    },
    {
      "epoch": 2.892494201660156e-06,
      "step": 77033,
      "training_step_time": 7.738872051239014
    },
    {
      "epoch": 2.8934478759765623e-06,
      "model_forward_time": 2.395758628845215,
      "step": 77034
    },
    {
      "epoch": 2.8934478759765623e-06,
      "step": 77034,
      "training_step_time": 7.758187532424927
    },
    {
      "epoch": 2.894401550292969e-06,
      "model_forward_time": 1.7376511096954346,
      "step": 77035
    },
    {
      "epoch": 2.894401550292969e-06,
      "step": 77035,
      "training_step_time": 7.7715489864349365
    },
    {
      "epoch": 2.895355224609375e-06,
      "model_forward_time": 1.733367681503296,
      "step": 77036
    },
    {
      "epoch": 2.895355224609375e-06,
      "step": 77036,
      "training_step_time": 7.634801149368286
    },
    {
      "epoch": 2.896308898925781e-06,
      "model_forward_time": 1.7247111797332764,
      "step": 77037
    },
    {
      "epoch": 2.896308898925781e-06,
      "step": 77037,
      "training_step_time": 7.795165061950684
    },
    {
      "epoch": 2.8972625732421873e-06,
      "model_forward_time": 1.7227604389190674,
      "step": 77038
    },
    {
      "epoch": 2.8972625732421873e-06,
      "step": 77038,
      "training_step_time": 7.541487693786621
    },
    {
      "epoch": 2.898216247558594e-06,
      "model_forward_time": 1.727670669555664,
      "step": 77039
    },
    {
      "epoch": 2.898216247558594e-06,
      "step": 77039,
      "training_step_time": 6.9713733196258545
    },
    {
      "epoch": 2.899169921875e-06,
      "grad_norm": 0.13788092136383057,
      "learning_rate": 1.3734316719778145e-06,
      "loss": 0.0566,
      "step": 77040
    },
    {
      "epoch": 2.899169921875e-06,
      "model_forward_time": 1.7303893566131592,
      "step": 77040
    },
    {
      "dynamics_loss_avg": 0.05117080621421337,
      "epoch": 2.899169921875e-06,
      "step": 77040
    },
    {
      "action_loss_avg": 0.0012792081601219253,
      "epoch": 2.899169921875e-06,
      "step": 77040
    },
    {
      "epoch": 2.899169921875e-06,
      "step": 77040,
      "training_step_time": 7.408331632614136
    },
    {
      "epoch": 2.900123596191406e-06,
      "model_forward_time": 1.723510980606079,
      "step": 77041
    },
    {
      "epoch": 2.900123596191406e-06,
      "step": 77041,
      "training_step_time": 7.378936290740967
    },
    {
      "epoch": 2.9010772705078123e-06,
      "model_forward_time": 1.7218661308288574,
      "step": 77042
    },
    {
      "epoch": 2.9010772705078123e-06,
      "step": 77042,
      "training_step_time": 8.608947992324829
    },
    {
      "epoch": 2.902030944824219e-06,
      "model_forward_time": 2.3756520748138428,
      "step": 77043
    },
    {
      "epoch": 2.902030944824219e-06,
      "step": 77043,
      "training_step_time": 9.084446907043457
    },
    {
      "epoch": 2.902984619140625e-06,
      "model_forward_time": 1.7587203979492188,
      "step": 77044
    },
    {
      "epoch": 2.902984619140625e-06,
      "step": 77044,
      "training_step_time": 7.768473863601685
    },
    {
      "epoch": 2.903938293457031e-06,
      "model_forward_time": 1.7361423969268799,
      "step": 77045
    },
    {
      "epoch": 2.903938293457031e-06,
      "step": 77045,
      "training_step_time": 7.687676668167114
    },
    {
      "epoch": 2.9048919677734373e-06,
      "model_forward_time": 1.7440919876098633,
      "step": 77046
    },
    {
      "epoch": 2.9048919677734373e-06,
      "step": 77046,
      "training_step_time": 7.753203868865967
    },
    {
      "epoch": 2.905845642089844e-06,
      "model_forward_time": 1.7246484756469727,
      "step": 77047
    },
    {
      "epoch": 2.905845642089844e-06,
      "step": 77047,
      "training_step_time": 7.465577125549316
    },
    {
      "epoch": 2.90679931640625e-06,
      "model_forward_time": 1.7364671230316162,
      "step": 77048
    },
    {
      "epoch": 2.90679931640625e-06,
      "step": 77048,
      "training_step_time": 7.004770278930664
    },
    {
      "epoch": 2.907752990722656e-06,
      "model_forward_time": 1.7360928058624268,
      "step": 77049
    },
    {
      "epoch": 2.907752990722656e-06,
      "step": 77049,
      "training_step_time": 8.198963165283203
    },
    {
      "epoch": 2.9087066650390623e-06,
      "grad_norm": 0.10164163261651993,
      "learning_rate": 1.3722935910184453e-06,
      "loss": 0.0553,
      "step": 77050
    },
    {
      "epoch": 2.9087066650390623e-06,
      "model_forward_time": 1.7345302104949951,
      "step": 77050
    },
    {
      "dynamics_loss_avg": 0.07113046534359455,
      "epoch": 2.9087066650390623e-06,
      "step": 77050
    },
    {
      "action_loss_avg": 0.0019434072630247102,
      "epoch": 2.9087066650390623e-06,
      "step": 77050
    },
    {
      "epoch": 2.9087066650390623e-06,
      "step": 77050,
      "training_step_time": 7.4399497509002686
    },
    {
      "epoch": 2.909660339355469e-06,
      "model_forward_time": 1.736025094985962,
      "step": 77051
    },
    {
      "epoch": 2.909660339355469e-06,
      "step": 77051,
      "training_step_time": 8.449237585067749
    },
    {
      "epoch": 2.910614013671875e-06,
      "model_forward_time": 2.3532469272613525,
      "step": 77052
    },
    {
      "epoch": 2.910614013671875e-06,
      "step": 77052,
      "training_step_time": 10.399879932403564
    },
    {
      "epoch": 2.911567687988281e-06,
      "model_forward_time": 1.7706291675567627,
      "step": 77053
    },
    {
      "epoch": 2.911567687988281e-06,
      "step": 77053,
      "training_step_time": 8.272599935531616
    },
    {
      "epoch": 2.9125213623046873e-06,
      "model_forward_time": 1.7382533550262451,
      "step": 77054
    },
    {
      "epoch": 2.9125213623046873e-06,
      "step": 77054,
      "training_step_time": 7.609908819198608
    },
    {
      "epoch": 2.913475036621094e-06,
      "model_forward_time": 1.7363252639770508,
      "step": 77055
    },
    {
      "epoch": 2.913475036621094e-06,
      "step": 77055,
      "training_step_time": 7.7404186725616455
    },
    {
      "epoch": 2.9144287109375e-06,
      "model_forward_time": 1.7394413948059082,
      "step": 77056
    },
    {
      "epoch": 2.9144287109375e-06,
      "step": 77056,
      "training_step_time": 7.347797155380249
    },
    {
      "epoch": 2.915382385253906e-06,
      "model_forward_time": 1.7415850162506104,
      "step": 77057
    },
    {
      "epoch": 2.915382385253906e-06,
      "step": 77057,
      "training_step_time": 7.543612957000732
    },
    {
      "epoch": 2.9163360595703123e-06,
      "model_forward_time": 1.7340054512023926,
      "step": 77058
    },
    {
      "epoch": 2.9163360595703123e-06,
      "step": 77058,
      "training_step_time": 7.34082555770874
    },
    {
      "epoch": 2.917289733886719e-06,
      "model_forward_time": 1.7417986392974854,
      "step": 77059
    },
    {
      "epoch": 2.917289733886719e-06,
      "step": 77059,
      "training_step_time": 7.3772313594818115
    },
    {
      "epoch": 2.918243408203125e-06,
      "grad_norm": 0.12223358452320099,
      "learning_rate": 1.371155906779592e-06,
      "loss": 0.0552,
      "step": 77060
    },
    {
      "epoch": 2.918243408203125e-06,
      "model_forward_time": 1.3554742336273193,
      "step": 77060
    },
    {
      "dynamics_loss_avg": 0.05719516109675169,
      "epoch": 2.918243408203125e-06,
      "step": 77060
    },
    {
      "action_loss_avg": 0.0011273871990852059,
      "epoch": 2.918243408203125e-06,
      "step": 77060
    },
    {
      "epoch": 2.918243408203125e-06,
      "step": 77060,
      "training_step_time": 7.857293605804443
    },
    {
      "epoch": 2.919197082519531e-06,
      "model_forward_time": 1.7433593273162842,
      "step": 77061
    },
    {
      "epoch": 2.919197082519531e-06,
      "step": 77061,
      "training_step_time": 9.004920721054077
    },
    {
      "epoch": 2.9201507568359373e-06,
      "model_forward_time": 2.3911588191986084,
      "step": 77062
    },
    {
      "epoch": 2.9201507568359373e-06,
      "step": 77062,
      "training_step_time": 8.530439138412476
    },
    {
      "epoch": 2.921104431152344e-06,
      "model_forward_time": 1.7349388599395752,
      "step": 77063
    },
    {
      "epoch": 2.921104431152344e-06,
      "step": 77063,
      "training_step_time": 7.626243352890015
    },
    {
      "epoch": 2.92205810546875e-06,
      "model_forward_time": 1.7377326488494873,
      "step": 77064
    },
    {
      "epoch": 2.92205810546875e-06,
      "step": 77064,
      "training_step_time": 7.6062095165252686
    },
    {
      "epoch": 2.923011779785156e-06,
      "model_forward_time": 1.7376599311828613,
      "step": 77065
    },
    {
      "epoch": 2.923011779785156e-06,
      "step": 77065,
      "training_step_time": 7.33271861076355
    },
    {
      "epoch": 2.9239654541015623e-06,
      "model_forward_time": 1.7320585250854492,
      "step": 77066
    },
    {
      "epoch": 2.9239654541015623e-06,
      "step": 77066,
      "training_step_time": 8.678995370864868
    },
    {
      "epoch": 2.924919128417969e-06,
      "model_forward_time": 1.738471508026123,
      "step": 77067
    },
    {
      "epoch": 2.924919128417969e-06,
      "step": 77067,
      "training_step_time": 7.313634872436523
    },
    {
      "epoch": 2.925872802734375e-06,
      "model_forward_time": 1.73829984664917,
      "step": 77068
    },
    {
      "epoch": 2.925872802734375e-06,
      "step": 77068,
      "training_step_time": 7.159272909164429
    },
    {
      "epoch": 2.926826477050781e-06,
      "model_forward_time": 1.7400636672973633,
      "step": 77069
    },
    {
      "epoch": 2.926826477050781e-06,
      "step": 77069,
      "training_step_time": 9.684350967407227
    },
    {
      "epoch": 2.9277801513671873e-06,
      "grad_norm": 0.12471328675746918,
      "learning_rate": 1.3700186193856758e-06,
      "loss": 0.055,
      "step": 77070
    },
    {
      "epoch": 2.9277801513671873e-06,
      "model_forward_time": 1.7365748882293701,
      "step": 77070
    },
    {
      "dynamics_loss_avg": 0.045390515215694906,
      "epoch": 2.9277801513671873e-06,
      "step": 77070
    },
    {
      "action_loss_avg": 0.0011783355366787873,
      "epoch": 2.9277801513671873e-06,
      "step": 77070
    },
    {
      "epoch": 2.9277801513671873e-06,
      "step": 77070,
      "training_step_time": 7.549433708190918
    },
    {
      "epoch": 2.928733825683594e-06,
      "model_forward_time": 2.3733954429626465,
      "step": 77071
    },
    {
      "epoch": 2.928733825683594e-06,
      "step": 77071,
      "training_step_time": 9.115580320358276
    },
    {
      "epoch": 2.9296875e-06,
      "model_forward_time": 1.707524061203003,
      "step": 77072
    },
    {
      "epoch": 2.9296875e-06,
      "step": 77072,
      "training_step_time": 8.723047494888306
    },
    {
      "epoch": 2.930641174316406e-06,
      "model_forward_time": 1.7215924263000488,
      "step": 77073
    },
    {
      "epoch": 2.930641174316406e-06,
      "step": 77073,
      "training_step_time": 8.067708492279053
    },
    {
      "epoch": 2.9315948486328124e-06,
      "model_forward_time": 1.7463853359222412,
      "step": 77074
    },
    {
      "epoch": 2.9315948486328124e-06,
      "step": 77074,
      "training_step_time": 8.460249423980713
    },
    {
      "epoch": 2.932548522949219e-06,
      "model_forward_time": 1.742976188659668,
      "step": 77075
    },
    {
      "epoch": 2.932548522949219e-06,
      "step": 77075,
      "training_step_time": 7.503947734832764
    },
    {
      "epoch": 2.933502197265625e-06,
      "model_forward_time": 1.7396624088287354,
      "step": 77076
    },
    {
      "epoch": 2.933502197265625e-06,
      "step": 77076,
      "training_step_time": 7.279015302658081
    },
    {
      "epoch": 2.934455871582031e-06,
      "model_forward_time": 1.7389445304870605,
      "step": 77077
    },
    {
      "epoch": 2.934455871582031e-06,
      "step": 77077,
      "training_step_time": 7.202534437179565
    },
    {
      "epoch": 2.9354095458984374e-06,
      "model_forward_time": 1.7395191192626953,
      "step": 77078
    },
    {
      "epoch": 2.9354095458984374e-06,
      "step": 77078,
      "training_step_time": 8.974578142166138
    },
    {
      "epoch": 2.936363220214844e-06,
      "model_forward_time": 1.7569575309753418,
      "step": 77079
    },
    {
      "epoch": 2.936363220214844e-06,
      "step": 77079,
      "training_step_time": 8.814146280288696
    },
    {
      "epoch": 2.93731689453125e-06,
      "grad_norm": 0.11498578637838364,
      "learning_rate": 1.368881728961064e-06,
      "loss": 0.0555,
      "step": 77080
    },
    {
      "epoch": 2.93731689453125e-06,
      "model_forward_time": 2.400390625,
      "step": 77080
    },
    {
      "dynamics_loss_avg": 0.05174216255545616,
      "epoch": 2.93731689453125e-06,
      "step": 77080
    },
    {
      "action_loss_avg": 0.0009266723296605051,
      "epoch": 2.93731689453125e-06,
      "step": 77080
    },
    {
      "epoch": 2.93731689453125e-06,
      "step": 77080,
      "training_step_time": 9.407819271087646
    },
    {
      "epoch": 2.9382705688476562e-06,
      "model_forward_time": 1.761810541152954,
      "step": 77081
    },
    {
      "epoch": 2.9382705688476562e-06,
      "step": 77081,
      "training_step_time": 8.556665420532227
    },
    {
      "epoch": 2.9392242431640624e-06,
      "model_forward_time": 1.7348625659942627,
      "step": 77082
    },
    {
      "epoch": 2.9392242431640624e-06,
      "step": 77082,
      "training_step_time": 7.623729467391968
    },
    {
      "epoch": 2.940177917480469e-06,
      "model_forward_time": 1.740440845489502,
      "step": 77083
    },
    {
      "epoch": 2.940177917480469e-06,
      "step": 77083,
      "training_step_time": 8.369143962860107
    },
    {
      "epoch": 2.941131591796875e-06,
      "model_forward_time": 1.7432992458343506,
      "step": 77084
    },
    {
      "epoch": 2.941131591796875e-06,
      "step": 77084,
      "training_step_time": 7.520374774932861
    },
    {
      "epoch": 2.9420852661132812e-06,
      "model_forward_time": 1.7340290546417236,
      "step": 77085
    },
    {
      "epoch": 2.9420852661132812e-06,
      "step": 77085,
      "training_step_time": 6.950584173202515
    },
    {
      "epoch": 2.9430389404296874e-06,
      "model_forward_time": 1.7412164211273193,
      "step": 77086
    },
    {
      "epoch": 2.9430389404296874e-06,
      "step": 77086,
      "training_step_time": 7.288899898529053
    },
    {
      "epoch": 2.943992614746094e-06,
      "model_forward_time": 1.7382848262786865,
      "step": 77087
    },
    {
      "epoch": 2.943992614746094e-06,
      "step": 77087,
      "training_step_time": 7.649725914001465
    },
    {
      "epoch": 2.9449462890625e-06,
      "model_forward_time": 1.750335454940796,
      "step": 77088
    },
    {
      "epoch": 2.9449462890625e-06,
      "step": 77088,
      "training_step_time": 10.008516788482666
    },
    {
      "epoch": 2.9458999633789062e-06,
      "model_forward_time": 2.401092052459717,
      "step": 77089
    },
    {
      "epoch": 2.9458999633789062e-06,
      "step": 77089,
      "training_step_time": 8.681180000305176
    },
    {
      "epoch": 2.9468536376953124e-06,
      "grad_norm": 0.08748997002840042,
      "learning_rate": 1.3677452356300885e-06,
      "loss": 0.0553,
      "step": 77090
    },
    {
      "epoch": 2.9468536376953124e-06,
      "model_forward_time": 1.7618772983551025,
      "step": 77090
    },
    {
      "dynamics_loss_avg": 0.05694174524396658,
      "epoch": 2.9468536376953124e-06,
      "step": 77090
    },
    {
      "action_loss_avg": 0.001628265855833888,
      "epoch": 2.9468536376953124e-06,
      "step": 77090
    },
    {
      "epoch": 2.9468536376953124e-06,
      "step": 77090,
      "training_step_time": 8.581312417984009
    },
    {
      "epoch": 2.947807312011719e-06,
      "model_forward_time": 1.7381482124328613,
      "step": 77091
    },
    {
      "epoch": 2.947807312011719e-06,
      "step": 77091,
      "training_step_time": 7.798740386962891
    },
    {
      "epoch": 2.948760986328125e-06,
      "model_forward_time": 1.7457208633422852,
      "step": 77092
    },
    {
      "epoch": 2.948760986328125e-06,
      "step": 77092,
      "training_step_time": 7.824916124343872
    },
    {
      "epoch": 2.9497146606445312e-06,
      "model_forward_time": 1.7359018325805664,
      "step": 77093
    },
    {
      "epoch": 2.9497146606445312e-06,
      "step": 77093,
      "training_step_time": 8.69455337524414
    },
    {
      "epoch": 2.9506683349609374e-06,
      "model_forward_time": 1.7478892803192139,
      "step": 77094
    },
    {
      "epoch": 2.9506683349609374e-06,
      "step": 77094,
      "training_step_time": 7.112636566162109
    },
    {
      "epoch": 2.951622009277344e-06,
      "model_forward_time": 1.7379138469696045,
      "step": 77095
    },
    {
      "epoch": 2.951622009277344e-06,
      "step": 77095,
      "training_step_time": 7.5070929527282715
    },
    {
      "epoch": 2.95257568359375e-06,
      "model_forward_time": 1.7412786483764648,
      "step": 77096
    },
    {
      "epoch": 2.95257568359375e-06,
      "step": 77096,
      "training_step_time": 8.629738807678223
    },
    {
      "epoch": 2.9535293579101562e-06,
      "model_forward_time": 1.7411301136016846,
      "step": 77097
    },
    {
      "epoch": 2.9535293579101562e-06,
      "step": 77097,
      "training_step_time": 10.283091306686401
    },
    {
      "epoch": 2.9544830322265624e-06,
      "model_forward_time": 1.737823247909546,
      "step": 77098
    },
    {
      "epoch": 2.9544830322265624e-06,
      "step": 77098,
      "training_step_time": 8.80552339553833
    },
    {
      "epoch": 2.955436706542969e-06,
      "model_forward_time": 1.8021273612976074,
      "step": 77099
    },
    {
      "epoch": 2.955436706542969e-06,
      "step": 77099,
      "training_step_time": 9.218101739883423
    },
    {
      "epoch": 2.956390380859375e-06,
      "grad_norm": 0.09264623373746872,
      "learning_rate": 1.3666091395170323e-06,
      "loss": 0.0535,
      "step": 77100
    },
    {
      "epoch": 2.956390380859375e-06,
      "model_forward_time": 1.730623722076416,
      "step": 77100
    },
    {
      "dynamics_loss_avg": 0.06725246980786323,
      "epoch": 2.956390380859375e-06,
      "step": 77100
    },
    {
      "action_loss_avg": 0.0011724954412784428,
      "epoch": 2.956390380859375e-06,
      "step": 77100
    },
    {
      "epoch": 2.956390380859375e-06,
      "step": 77100,
      "training_step_time": 8.30353593826294
    },
    {
      "epoch": 2.9573440551757812e-06,
      "model_forward_time": 1.7344787120819092,
      "step": 77101
    },
    {
      "epoch": 2.9573440551757812e-06,
      "step": 77101,
      "training_step_time": 7.706888198852539
    },
    {
      "epoch": 2.9582977294921874e-06,
      "model_forward_time": 1.739058494567871,
      "step": 77102
    },
    {
      "epoch": 2.9582977294921874e-06,
      "step": 77102,
      "training_step_time": 7.826083421707153
    },
    {
      "epoch": 2.959251403808594e-06,
      "model_forward_time": 1.715193271636963,
      "step": 77103
    },
    {
      "epoch": 2.959251403808594e-06,
      "step": 77103,
      "training_step_time": 7.37027645111084
    },
    {
      "epoch": 2.960205078125e-06,
      "model_forward_time": 1.7226877212524414,
      "step": 77104
    },
    {
      "epoch": 2.960205078125e-06,
      "step": 77104,
      "training_step_time": 7.507059812545776
    },
    {
      "epoch": 2.9611587524414062e-06,
      "model_forward_time": 1.7172572612762451,
      "step": 77105
    },
    {
      "epoch": 2.9611587524414062e-06,
      "step": 77105,
      "training_step_time": 7.729487895965576
    },
    {
      "epoch": 2.9621124267578124e-06,
      "model_forward_time": 1.7198898792266846,
      "step": 77106
    },
    {
      "epoch": 2.9621124267578124e-06,
      "step": 77106,
      "training_step_time": 8.749591588973999
    },
    {
      "epoch": 2.963066101074219e-06,
      "model_forward_time": 1.7297191619873047,
      "step": 77107
    },
    {
      "epoch": 2.963066101074219e-06,
      "step": 77107,
      "training_step_time": 7.7043256759643555
    },
    {
      "epoch": 2.964019775390625e-06,
      "model_forward_time": 2.4332263469696045,
      "step": 77108
    },
    {
      "epoch": 2.964019775390625e-06,
      "step": 77108,
      "training_step_time": 7.764733552932739
    },
    {
      "epoch": 2.9649734497070312e-06,
      "model_forward_time": 1.733389139175415,
      "step": 77109
    },
    {
      "epoch": 2.9649734497070312e-06,
      "step": 77109,
      "training_step_time": 7.772533655166626
    },
    {
      "epoch": 2.9659271240234374e-06,
      "grad_norm": 0.09796147048473358,
      "learning_rate": 1.3654734407461385e-06,
      "loss": 0.053,
      "step": 77110
    },
    {
      "epoch": 2.9659271240234374e-06,
      "model_forward_time": 1.738541841506958,
      "step": 77110
    },
    {
      "dynamics_loss_avg": 0.0478690842166543,
      "epoch": 2.9659271240234374e-06,
      "step": 77110
    },
    {
      "action_loss_avg": 0.001349807798396796,
      "epoch": 2.9659271240234374e-06,
      "step": 77110
    },
    {
      "epoch": 2.9659271240234374e-06,
      "step": 77110,
      "training_step_time": 7.830766201019287
    },
    {
      "epoch": 2.966880798339844e-06,
      "model_forward_time": 1.751692295074463,
      "step": 77111
    },
    {
      "epoch": 2.966880798339844e-06,
      "step": 77111,
      "training_step_time": 7.802241563796997
    },
    {
      "epoch": 2.96783447265625e-06,
      "model_forward_time": 1.7696833610534668,
      "step": 77112
    },
    {
      "epoch": 2.96783447265625e-06,
      "step": 77112,
      "training_step_time": 7.45468807220459
    },
    {
      "epoch": 2.9687881469726562e-06,
      "model_forward_time": 1.7488226890563965,
      "step": 77113
    },
    {
      "epoch": 2.9687881469726562e-06,
      "step": 77113,
      "training_step_time": 7.420164346694946
    },
    {
      "epoch": 2.9697418212890624e-06,
      "model_forward_time": 1.7462620735168457,
      "step": 77114
    },
    {
      "epoch": 2.9697418212890624e-06,
      "step": 77114,
      "training_step_time": 7.594928503036499
    },
    {
      "epoch": 2.970695495605469e-06,
      "model_forward_time": 1.7437493801116943,
      "step": 77115
    },
    {
      "epoch": 2.970695495605469e-06,
      "step": 77115,
      "training_step_time": 8.492360591888428
    },
    {
      "epoch": 2.971649169921875e-06,
      "model_forward_time": 1.7410824298858643,
      "step": 77116
    },
    {
      "epoch": 2.971649169921875e-06,
      "step": 77116,
      "training_step_time": 7.65271782875061
    },
    {
      "epoch": 2.9726028442382812e-06,
      "model_forward_time": 2.4583799839019775,
      "step": 77117
    },
    {
      "epoch": 2.9726028442382812e-06,
      "step": 77117,
      "training_step_time": 9.389575481414795
    },
    {
      "epoch": 2.9735565185546874e-06,
      "model_forward_time": 1.7424087524414062,
      "step": 77118
    },
    {
      "epoch": 2.9735565185546874e-06,
      "step": 77118,
      "training_step_time": 7.672675848007202
    },
    {
      "epoch": 2.974510192871094e-06,
      "model_forward_time": 1.7364437580108643,
      "step": 77119
    },
    {
      "epoch": 2.974510192871094e-06,
      "step": 77119,
      "training_step_time": 7.677130937576294
    },
    {
      "epoch": 2.9754638671875e-06,
      "grad_norm": 0.09801837056875229,
      "learning_rate": 1.3643381394416023e-06,
      "loss": 0.0548,
      "step": 77120
    },
    {
      "epoch": 2.9754638671875e-06,
      "model_forward_time": 1.7374513149261475,
      "step": 77120
    },
    {
      "dynamics_loss_avg": 0.04931232053786516,
      "epoch": 2.9754638671875e-06,
      "step": 77120
    },
    {
      "action_loss_avg": 0.0017096208757720888,
      "epoch": 2.9754638671875e-06,
      "step": 77120
    },
    {
      "epoch": 2.9754638671875e-06,
      "step": 77120,
      "training_step_time": 7.673837900161743
    },
    {
      "epoch": 2.9764175415039062e-06,
      "model_forward_time": 1.7329201698303223,
      "step": 77121
    },
    {
      "epoch": 2.9764175415039062e-06,
      "step": 77121,
      "training_step_time": 7.416200637817383
    },
    {
      "epoch": 2.9773712158203124e-06,
      "model_forward_time": 1.7367470264434814,
      "step": 77122
    },
    {
      "epoch": 2.9773712158203124e-06,
      "step": 77122,
      "training_step_time": 7.856069326400757
    },
    {
      "epoch": 2.978324890136719e-06,
      "model_forward_time": 1.7201526165008545,
      "step": 77123
    },
    {
      "epoch": 2.978324890136719e-06,
      "step": 77123,
      "training_step_time": 7.236104965209961
    },
    {
      "epoch": 2.979278564453125e-06,
      "model_forward_time": 1.722477912902832,
      "step": 77124
    },
    {
      "epoch": 2.979278564453125e-06,
      "step": 77124,
      "training_step_time": 7.6756911277771
    },
    {
      "epoch": 2.9802322387695312e-06,
      "model_forward_time": 1.7401039600372314,
      "step": 77125
    },
    {
      "epoch": 2.9802322387695312e-06,
      "step": 77125,
      "training_step_time": 9.23102593421936
    },
    {
      "epoch": 2.9811859130859374e-06,
      "model_forward_time": 2.4551920890808105,
      "step": 77126
    },
    {
      "epoch": 2.9811859130859374e-06,
      "step": 77126,
      "training_step_time": 9.260033130645752
    },
    {
      "epoch": 2.9821395874023435e-06,
      "model_forward_time": 1.738776445388794,
      "step": 77127
    },
    {
      "epoch": 2.9821395874023435e-06,
      "step": 77127,
      "training_step_time": 8.63599157333374
    },
    {
      "epoch": 2.98309326171875e-06,
      "model_forward_time": 1.7553479671478271,
      "step": 77128
    },
    {
      "epoch": 2.98309326171875e-06,
      "step": 77128,
      "training_step_time": 7.657722473144531
    },
    {
      "epoch": 2.9840469360351563e-06,
      "model_forward_time": 1.7381508350372314,
      "step": 77129
    },
    {
      "epoch": 2.9840469360351563e-06,
      "step": 77129,
      "training_step_time": 8.505494594573975
    },
    {
      "epoch": 2.9850006103515624e-06,
      "grad_norm": 0.09434165805578232,
      "learning_rate": 1.3632032357275836e-06,
      "loss": 0.0547,
      "step": 77130
    },
    {
      "epoch": 2.9850006103515624e-06,
      "model_forward_time": 1.735992431640625,
      "step": 77130
    },
    {
      "dynamics_loss_avg": 0.053562537953257564,
      "epoch": 2.9850006103515624e-06,
      "step": 77130
    },
    {
      "action_loss_avg": 0.001975364953977987,
      "epoch": 2.9850006103515624e-06,
      "step": 77130
    },
    {
      "epoch": 2.9850006103515624e-06,
      "step": 77130,
      "training_step_time": 7.976765394210815
    },
    {
      "epoch": 2.9859542846679685e-06,
      "model_forward_time": 1.7045714855194092,
      "step": 77131
    },
    {
      "epoch": 2.9859542846679685e-06,
      "step": 77131,
      "training_step_time": 6.9747114181518555
    },
    {
      "epoch": 2.986907958984375e-06,
      "model_forward_time": 1.713987112045288,
      "step": 77132
    },
    {
      "epoch": 2.986907958984375e-06,
      "step": 77132,
      "training_step_time": 7.337410926818848
    },
    {
      "epoch": 2.9878616333007813e-06,
      "model_forward_time": 1.7176158428192139,
      "step": 77133
    },
    {
      "epoch": 2.9878616333007813e-06,
      "step": 77133,
      "training_step_time": 7.74923038482666
    },
    {
      "epoch": 2.9888153076171874e-06,
      "model_forward_time": 1.341012954711914,
      "step": 77134
    },
    {
      "epoch": 2.9888153076171874e-06,
      "step": 77134,
      "training_step_time": 7.670712232589722
    },
    {
      "epoch": 2.9897689819335935e-06,
      "model_forward_time": 2.316840648651123,
      "step": 77135
    },
    {
      "epoch": 2.9897689819335935e-06,
      "step": 77135,
      "training_step_time": 9.974425315856934
    },
    {
      "epoch": 2.99072265625e-06,
      "model_forward_time": 1.7237522602081299,
      "step": 77136
    },
    {
      "epoch": 2.99072265625e-06,
      "step": 77136,
      "training_step_time": 8.534297227859497
    },
    {
      "epoch": 2.9916763305664063e-06,
      "model_forward_time": 1.7374253273010254,
      "step": 77137
    },
    {
      "epoch": 2.9916763305664063e-06,
      "step": 77137,
      "training_step_time": 7.614729642868042
    },
    {
      "epoch": 2.9926300048828124e-06,
      "model_forward_time": 1.719954013824463,
      "step": 77138
    },
    {
      "epoch": 2.9926300048828124e-06,
      "step": 77138,
      "training_step_time": 8.63327145576477
    },
    {
      "epoch": 2.9935836791992186e-06,
      "model_forward_time": 1.7219231128692627,
      "step": 77139
    },
    {
      "epoch": 2.9935836791992186e-06,
      "step": 77139,
      "training_step_time": 7.66564154624939
    },
    {
      "epoch": 2.994537353515625e-06,
      "grad_norm": 0.24579977989196777,
      "learning_rate": 1.3620687297281898e-06,
      "loss": 0.0556,
      "step": 77140
    },
    {
      "epoch": 2.994537353515625e-06,
      "model_forward_time": 1.7219815254211426,
      "step": 77140
    },
    {
      "dynamics_loss_avg": 0.04644244574010372,
      "epoch": 2.994537353515625e-06,
      "step": 77140
    },
    {
      "action_loss_avg": 0.0012856394401751458,
      "epoch": 2.994537353515625e-06,
      "step": 77140
    },
    {
      "epoch": 2.994537353515625e-06,
      "step": 77140,
      "training_step_time": 7.430462598800659
    },
    {
      "epoch": 2.9954910278320313e-06,
      "model_forward_time": 1.7191262245178223,
      "step": 77141
    },
    {
      "epoch": 2.9954910278320313e-06,
      "step": 77141,
      "training_step_time": 8.966947555541992
    },
    {
      "epoch": 2.9964447021484374e-06,
      "model_forward_time": 1.7192931175231934,
      "step": 77142
    },
    {
      "epoch": 2.9964447021484374e-06,
      "step": 77142,
      "training_step_time": 7.667372703552246
    },
    {
      "epoch": 2.9973983764648436e-06,
      "model_forward_time": 1.7150051593780518,
      "step": 77143
    },
    {
      "epoch": 2.9973983764648436e-06,
      "step": 77143,
      "training_step_time": 8.251746892929077
    },
    {
      "epoch": 2.99835205078125e-06,
      "model_forward_time": 1.7141790390014648,
      "step": 77144
    },
    {
      "epoch": 2.99835205078125e-06,
      "step": 77144,
      "training_step_time": 9.62164568901062
    },
    {
      "epoch": 2.9993057250976563e-06,
      "model_forward_time": 2.35280179977417,
      "step": 77145
    },
    {
      "epoch": 2.9993057250976563e-06,
      "step": 77145,
      "training_step_time": 8.297409296035767
    },
    {
      "epoch": 3.0002593994140624e-06,
      "model_forward_time": 1.7390697002410889,
      "step": 77146
    },
    {
      "epoch": 3.0002593994140624e-06,
      "step": 77146,
      "training_step_time": 8.38467812538147
    },
    {
      "epoch": 3.0012130737304686e-06,
      "model_forward_time": 1.7226014137268066,
      "step": 77147
    },
    {
      "epoch": 3.0012130737304686e-06,
      "step": 77147,
      "training_step_time": 8.17358660697937
    },
    {
      "epoch": 3.002166748046875e-06,
      "model_forward_time": 1.723374366760254,
      "step": 77148
    },
    {
      "epoch": 3.002166748046875e-06,
      "step": 77148,
      "training_step_time": 7.732079267501831
    },
    {
      "epoch": 3.0031204223632813e-06,
      "model_forward_time": 1.7288639545440674,
      "step": 77149
    },
    {
      "epoch": 3.0031204223632813e-06,
      "step": 77149,
      "training_step_time": 8.501091241836548
    },
    {
      "epoch": 3.0040740966796874e-06,
      "grad_norm": 0.12332724034786224,
      "learning_rate": 1.3609346215674918e-06,
      "loss": 0.0542,
      "step": 77150
    },
    {
      "epoch": 3.0040740966796874e-06,
      "model_forward_time": 1.7353253364562988,
      "step": 77150
    },
    {
      "dynamics_loss_avg": 0.044185145013034346,
      "epoch": 3.0040740966796874e-06,
      "step": 77150
    },
    {
      "action_loss_avg": 0.001143102120840922,
      "epoch": 3.0040740966796874e-06,
      "step": 77150
    },
    {
      "epoch": 3.0040740966796874e-06,
      "step": 77150,
      "training_step_time": 8.087278842926025
    },
    {
      "epoch": 3.0050277709960936e-06,
      "model_forward_time": 1.733471155166626,
      "step": 77151
    },
    {
      "epoch": 3.0050277709960936e-06,
      "step": 77151,
      "training_step_time": 8.209967136383057
    },
    {
      "epoch": 3.0059814453125e-06,
      "model_forward_time": 1.7470879554748535,
      "step": 77152
    },
    {
      "epoch": 3.0059814453125e-06,
      "step": 77152,
      "training_step_time": 8.404545307159424
    },
    {
      "epoch": 3.0069351196289063e-06,
      "model_forward_time": 1.749565839767456,
      "step": 77153
    },
    {
      "epoch": 3.0069351196289063e-06,
      "step": 77153,
      "training_step_time": 9.079472541809082
    },
    {
      "epoch": 3.0078887939453124e-06,
      "model_forward_time": 2.4198756217956543,
      "step": 77154
    },
    {
      "epoch": 3.0078887939453124e-06,
      "step": 77154,
      "training_step_time": 7.684382915496826
    },
    {
      "epoch": 3.0088424682617186e-06,
      "model_forward_time": 1.7595930099487305,
      "step": 77155
    },
    {
      "epoch": 3.0088424682617186e-06,
      "step": 77155,
      "training_step_time": 9.58499264717102
    },
    {
      "epoch": 3.009796142578125e-06,
      "model_forward_time": 1.749046802520752,
      "step": 77156
    },
    {
      "epoch": 3.009796142578125e-06,
      "step": 77156,
      "training_step_time": 7.696047782897949
    },
    {
      "epoch": 3.0107498168945313e-06,
      "model_forward_time": 1.7492444515228271,
      "step": 77157
    },
    {
      "epoch": 3.0107498168945313e-06,
      "step": 77157,
      "training_step_time": 7.728585958480835
    },
    {
      "epoch": 3.0117034912109374e-06,
      "model_forward_time": 1.751894235610962,
      "step": 77158
    },
    {
      "epoch": 3.0117034912109374e-06,
      "step": 77158,
      "training_step_time": 7.4668169021606445
    },
    {
      "epoch": 3.0126571655273436e-06,
      "model_forward_time": 1.7415494918823242,
      "step": 77159
    },
    {
      "epoch": 3.0126571655273436e-06,
      "step": 77159,
      "training_step_time": 7.1598992347717285
    },
    {
      "epoch": 3.01361083984375e-06,
      "grad_norm": 0.08302994817495346,
      "learning_rate": 1.359800911369511e-06,
      "loss": 0.056,
      "step": 77160
    },
    {
      "epoch": 3.01361083984375e-06,
      "model_forward_time": 1.7481307983398438,
      "step": 77160
    },
    {
      "dynamics_loss_avg": 0.047707568109035495,
      "epoch": 3.01361083984375e-06,
      "step": 77160
    },
    {
      "action_loss_avg": 0.0020558467833325265,
      "epoch": 3.01361083984375e-06,
      "step": 77160
    },
    {
      "epoch": 3.01361083984375e-06,
      "step": 77160,
      "training_step_time": 7.684723854064941
    },
    {
      "epoch": 3.0145645141601563e-06,
      "model_forward_time": 1.7596287727355957,
      "step": 77161
    },
    {
      "epoch": 3.0145645141601563e-06,
      "step": 77161,
      "training_step_time": 7.624721050262451
    },
    {
      "epoch": 3.0155181884765624e-06,
      "model_forward_time": 1.748758316040039,
      "step": 77162
    },
    {
      "epoch": 3.0155181884765624e-06,
      "step": 77162,
      "training_step_time": 8.653881311416626
    },
    {
      "epoch": 3.0164718627929686e-06,
      "model_forward_time": 2.4873366355895996,
      "step": 77163
    },
    {
      "epoch": 3.0164718627929686e-06,
      "step": 77163,
      "training_step_time": 7.662123680114746
    },
    {
      "epoch": 3.017425537109375e-06,
      "model_forward_time": 1.7503039836883545,
      "step": 77164
    },
    {
      "epoch": 3.017425537109375e-06,
      "step": 77164,
      "training_step_time": 7.724299192428589
    },
    {
      "epoch": 3.0183792114257813e-06,
      "model_forward_time": 1.7640962600708008,
      "step": 77165
    },
    {
      "epoch": 3.0183792114257813e-06,
      "step": 77165,
      "training_step_time": 7.680886268615723
    },
    {
      "epoch": 3.0193328857421874e-06,
      "model_forward_time": 1.7466201782226562,
      "step": 77166
    },
    {
      "epoch": 3.0193328857421874e-06,
      "step": 77166,
      "training_step_time": 7.692291259765625
    },
    {
      "epoch": 3.0202865600585936e-06,
      "model_forward_time": 1.7533233165740967,
      "step": 77167
    },
    {
      "epoch": 3.0202865600585936e-06,
      "step": 77167,
      "training_step_time": 7.50989031791687
    },
    {
      "epoch": 3.021240234375e-06,
      "model_forward_time": 1.7519700527191162,
      "step": 77168
    },
    {
      "epoch": 3.021240234375e-06,
      "step": 77168,
      "training_step_time": 7.130055665969849
    },
    {
      "epoch": 3.0221939086914063e-06,
      "model_forward_time": 1.7500412464141846,
      "step": 77169
    },
    {
      "epoch": 3.0221939086914063e-06,
      "step": 77169,
      "training_step_time": 7.6655473709106445
    },
    {
      "epoch": 3.0231475830078124e-06,
      "grad_norm": 0.1010175421833992,
      "learning_rate": 1.358667599258231e-06,
      "loss": 0.0553,
      "step": 77170
    },
    {
      "epoch": 3.0231475830078124e-06,
      "model_forward_time": 1.7516443729400635,
      "step": 77170
    },
    {
      "dynamics_loss_avg": 0.05683375783264637,
      "epoch": 3.0231475830078124e-06,
      "step": 77170
    },
    {
      "action_loss_avg": 0.001505506335524842,
      "epoch": 3.0231475830078124e-06,
      "step": 77170
    },
    {
      "epoch": 3.0231475830078124e-06,
      "step": 77170,
      "training_step_time": 9.127256393432617
    },
    {
      "epoch": 3.0241012573242186e-06,
      "model_forward_time": 1.7521514892578125,
      "step": 77171
    },
    {
      "epoch": 3.0241012573242186e-06,
      "step": 77171,
      "training_step_time": 8.249567985534668
    },
    {
      "epoch": 3.025054931640625e-06,
      "model_forward_time": 2.474900960922241,
      "step": 77172
    },
    {
      "epoch": 3.025054931640625e-06,
      "step": 77172,
      "training_step_time": 7.678367614746094
    },
    {
      "epoch": 3.0260086059570313e-06,
      "model_forward_time": 1.7567412853240967,
      "step": 77173
    },
    {
      "epoch": 3.0260086059570313e-06,
      "step": 77173,
      "training_step_time": 7.802797794342041
    },
    {
      "epoch": 3.0269622802734374e-06,
      "model_forward_time": 1.7669644355773926,
      "step": 77174
    },
    {
      "epoch": 3.0269622802734374e-06,
      "step": 77174,
      "training_step_time": 7.8110737800598145
    },
    {
      "epoch": 3.0279159545898436e-06,
      "model_forward_time": 1.7522687911987305,
      "step": 77175
    },
    {
      "epoch": 3.0279159545898436e-06,
      "step": 77175,
      "training_step_time": 7.760640859603882
    },
    {
      "epoch": 3.02886962890625e-06,
      "model_forward_time": 1.749495029449463,
      "step": 77176
    },
    {
      "epoch": 3.02886962890625e-06,
      "step": 77176,
      "training_step_time": 7.788537979125977
    },
    {
      "epoch": 3.0298233032226563e-06,
      "model_forward_time": 1.7456707954406738,
      "step": 77177
    },
    {
      "epoch": 3.0298233032226563e-06,
      "step": 77177,
      "training_step_time": 7.246002197265625
    },
    {
      "epoch": 3.0307769775390624e-06,
      "model_forward_time": 1.7476723194122314,
      "step": 77178
    },
    {
      "epoch": 3.0307769775390624e-06,
      "step": 77178,
      "training_step_time": 7.807133436203003
    },
    {
      "epoch": 3.0317306518554686e-06,
      "model_forward_time": 1.38187837600708,
      "step": 77179
    },
    {
      "epoch": 3.0317306518554686e-06,
      "step": 77179,
      "training_step_time": 7.756957292556763
    },
    {
      "epoch": 3.032684326171875e-06,
      "grad_norm": 0.07133149355649948,
      "learning_rate": 1.357534685357586e-06,
      "loss": 0.0566,
      "step": 77180
    },
    {
      "epoch": 3.032684326171875e-06,
      "model_forward_time": 1.7493095397949219,
      "step": 77180
    },
    {
      "dynamics_loss_avg": 0.05003401711583137,
      "epoch": 3.032684326171875e-06,
      "step": 77180
    },
    {
      "action_loss_avg": 0.0011930781387491153,
      "epoch": 3.032684326171875e-06,
      "step": 77180
    },
    {
      "epoch": 3.032684326171875e-06,
      "step": 77180,
      "training_step_time": 8.569885492324829
    },
    {
      "epoch": 3.0336380004882813e-06,
      "model_forward_time": 2.363802433013916,
      "step": 77181
    },
    {
      "epoch": 3.0336380004882813e-06,
      "step": 77181,
      "training_step_time": 8.911176204681396
    },
    {
      "epoch": 3.0345916748046874e-06,
      "model_forward_time": 1.7781968116760254,
      "step": 77182
    },
    {
      "epoch": 3.0345916748046874e-06,
      "step": 77182,
      "training_step_time": 8.475053310394287
    },
    {
      "epoch": 3.0355453491210936e-06,
      "model_forward_time": 1.7695693969726562,
      "step": 77183
    },
    {
      "epoch": 3.0355453491210936e-06,
      "step": 77183,
      "training_step_time": 7.741639137268066
    },
    {
      "epoch": 3.0364990234375e-06,
      "model_forward_time": 1.7468056678771973,
      "step": 77184
    },
    {
      "epoch": 3.0364990234375e-06,
      "step": 77184,
      "training_step_time": 7.6767919063568115
    },
    {
      "epoch": 3.0374526977539063e-06,
      "model_forward_time": 1.7521259784698486,
      "step": 77185
    },
    {
      "epoch": 3.0374526977539063e-06,
      "step": 77185,
      "training_step_time": 7.831835508346558
    },
    {
      "epoch": 3.0384063720703124e-06,
      "model_forward_time": 1.7468593120574951,
      "step": 77186
    },
    {
      "epoch": 3.0384063720703124e-06,
      "step": 77186,
      "training_step_time": 7.579006195068359
    },
    {
      "epoch": 3.0393600463867186e-06,
      "model_forward_time": 1.747239589691162,
      "step": 77187
    },
    {
      "epoch": 3.0393600463867186e-06,
      "step": 77187,
      "training_step_time": 7.685145378112793
    },
    {
      "epoch": 3.040313720703125e-06,
      "model_forward_time": 1.7516429424285889,
      "step": 77188
    },
    {
      "epoch": 3.040313720703125e-06,
      "step": 77188,
      "training_step_time": 7.679743766784668
    },
    {
      "epoch": 3.0412673950195313e-06,
      "model_forward_time": 1.748685598373413,
      "step": 77189
    },
    {
      "epoch": 3.0412673950195313e-06,
      "step": 77189,
      "training_step_time": 7.7048187255859375
    },
    {
      "epoch": 3.0422210693359374e-06,
      "grad_norm": 0.12170764803886414,
      "learning_rate": 1.3564021697914714e-06,
      "loss": 0.0572,
      "step": 77190
    },
    {
      "epoch": 3.0422210693359374e-06,
      "model_forward_time": 1.7470080852508545,
      "step": 77190
    },
    {
      "dynamics_loss_avg": 0.05268766237422824,
      "epoch": 3.0422210693359374e-06,
      "step": 77190
    },
    {
      "action_loss_avg": 0.0021398736513219773,
      "epoch": 3.0422210693359374e-06,
      "step": 77190
    },
    {
      "epoch": 3.0422210693359374e-06,
      "step": 77190,
      "training_step_time": 7.765633583068848
    },
    {
      "epoch": 3.0431747436523436e-06,
      "model_forward_time": 2.4217772483825684,
      "step": 77191
    },
    {
      "epoch": 3.0431747436523436e-06,
      "step": 77191,
      "training_step_time": 10.636136531829834
    },
    {
      "epoch": 3.04412841796875e-06,
      "model_forward_time": 1.7575232982635498,
      "step": 77192
    },
    {
      "epoch": 3.04412841796875e-06,
      "step": 77192,
      "training_step_time": 7.719194412231445
    },
    {
      "epoch": 3.0450820922851563e-06,
      "model_forward_time": 1.7303392887115479,
      "step": 77193
    },
    {
      "epoch": 3.0450820922851563e-06,
      "step": 77193,
      "training_step_time": 8.619438886642456
    },
    {
      "epoch": 3.0460357666015625e-06,
      "model_forward_time": 1.735424280166626,
      "step": 77194
    },
    {
      "epoch": 3.0460357666015625e-06,
      "step": 77194,
      "training_step_time": 7.826382875442505
    },
    {
      "epoch": 3.0469894409179686e-06,
      "model_forward_time": 1.729703426361084,
      "step": 77195
    },
    {
      "epoch": 3.0469894409179686e-06,
      "step": 77195,
      "training_step_time": 7.682626724243164
    },
    {
      "epoch": 3.047943115234375e-06,
      "model_forward_time": 1.7318432331085205,
      "step": 77196
    },
    {
      "epoch": 3.047943115234375e-06,
      "step": 77196,
      "training_step_time": 8.123582363128662
    },
    {
      "epoch": 3.0488967895507813e-06,
      "model_forward_time": 1.7296695709228516,
      "step": 77197
    },
    {
      "epoch": 3.0488967895507813e-06,
      "step": 77197,
      "training_step_time": 7.632870674133301
    },
    {
      "epoch": 3.0498504638671875e-06,
      "model_forward_time": 1.7334513664245605,
      "step": 77198
    },
    {
      "epoch": 3.0498504638671875e-06,
      "step": 77198,
      "training_step_time": 7.792484283447266
    },
    {
      "epoch": 3.0508041381835936e-06,
      "model_forward_time": 1.727536678314209,
      "step": 77199
    },
    {
      "epoch": 3.0508041381835936e-06,
      "step": 77199,
      "training_step_time": 9.104270935058594
    },
    {
      "epoch": 3.0517578125e-06,
      "grad_norm": 0.0977572575211525,
      "learning_rate": 1.355270052683738e-06,
      "loss": 0.0562,
      "step": 77200
    },
    {
      "epoch": 3.0517578125e-06,
      "model_forward_time": 2.4122231006622314,
      "step": 77200
    },
    {
      "dynamics_loss_avg": 0.04917205683887005,
      "epoch": 3.0517578125e-06,
      "step": 77200
    },
    {
      "action_loss_avg": 0.0009796916230698117,
      "epoch": 3.0517578125e-06,
      "step": 77200
    },
    {
      "epoch": 3.0517578125e-06,
      "step": 77200,
      "training_step_time": 9.767877101898193
    },
    {
      "epoch": 3.0527114868164063e-06,
      "model_forward_time": 1.7321434020996094,
      "step": 77201
    },
    {
      "epoch": 3.0527114868164063e-06,
      "step": 77201,
      "training_step_time": 7.828031301498413
    },
    {
      "epoch": 3.0536651611328125e-06,
      "model_forward_time": 1.735978126525879,
      "step": 77202
    },
    {
      "epoch": 3.0536651611328125e-06,
      "step": 77202,
      "training_step_time": 7.7093305587768555
    },
    {
      "epoch": 3.0546188354492186e-06,
      "model_forward_time": 1.7279958724975586,
      "step": 77203
    },
    {
      "epoch": 3.0546188354492186e-06,
      "step": 77203,
      "training_step_time": 7.532222032546997
    },
    {
      "epoch": 3.055572509765625e-06,
      "model_forward_time": 1.736161708831787,
      "step": 77204
    },
    {
      "epoch": 3.055572509765625e-06,
      "step": 77204,
      "training_step_time": 7.747079849243164
    },
    {
      "epoch": 3.0565261840820313e-06,
      "model_forward_time": 1.7298603057861328,
      "step": 77205
    },
    {
      "epoch": 3.0565261840820313e-06,
      "step": 77205,
      "training_step_time": 7.588744640350342
    },
    {
      "epoch": 3.0574798583984375e-06,
      "model_forward_time": 1.7227859497070312,
      "step": 77206
    },
    {
      "epoch": 3.0574798583984375e-06,
      "step": 77206,
      "training_step_time": 7.8197643756866455
    },
    {
      "epoch": 3.0584335327148436e-06,
      "model_forward_time": 1.7290308475494385,
      "step": 77207
    },
    {
      "epoch": 3.0584335327148436e-06,
      "step": 77207,
      "training_step_time": 8.595393896102905
    },
    {
      "epoch": 3.05938720703125e-06,
      "model_forward_time": 1.7246251106262207,
      "step": 77208
    },
    {
      "epoch": 3.05938720703125e-06,
      "step": 77208,
      "training_step_time": 9.33870267868042
    },
    {
      "epoch": 3.0603408813476563e-06,
      "model_forward_time": 2.428473949432373,
      "step": 77209
    },
    {
      "epoch": 3.0603408813476563e-06,
      "step": 77209,
      "training_step_time": 8.321158647537231
    },
    {
      "epoch": 3.0612945556640625e-06,
      "grad_norm": 0.09910403192043304,
      "learning_rate": 1.3541383341581925e-06,
      "loss": 0.0564,
      "step": 77210
    },
    {
      "epoch": 3.0612945556640625e-06,
      "model_forward_time": 1.7335007190704346,
      "step": 77210
    },
    {
      "dynamics_loss_avg": 0.06032207938842475,
      "epoch": 3.0612945556640625e-06,
      "step": 77210
    },
    {
      "action_loss_avg": 0.0013161824899725616,
      "epoch": 3.0612945556640625e-06,
      "step": 77210
    },
    {
      "epoch": 3.0612945556640625e-06,
      "step": 77210,
      "training_step_time": 8.559312582015991
    },
    {
      "epoch": 3.0622482299804686e-06,
      "model_forward_time": 1.7401704788208008,
      "step": 77211
    },
    {
      "epoch": 3.0622482299804686e-06,
      "step": 77211,
      "training_step_time": 7.597710609436035
    },
    {
      "epoch": 3.063201904296875e-06,
      "model_forward_time": 1.731898546218872,
      "step": 77212
    },
    {
      "epoch": 3.063201904296875e-06,
      "step": 77212,
      "training_step_time": 7.552998781204224
    },
    {
      "epoch": 3.0641555786132813e-06,
      "model_forward_time": 1.7475500106811523,
      "step": 77213
    },
    {
      "epoch": 3.0641555786132813e-06,
      "step": 77213,
      "training_step_time": 7.611706495285034
    },
    {
      "epoch": 3.0651092529296875e-06,
      "model_forward_time": 1.750885009765625,
      "step": 77214
    },
    {
      "epoch": 3.0651092529296875e-06,
      "step": 77214,
      "training_step_time": 7.564860820770264
    },
    {
      "epoch": 3.0660629272460936e-06,
      "model_forward_time": 1.7497313022613525,
      "step": 77215
    },
    {
      "epoch": 3.0660629272460936e-06,
      "step": 77215,
      "training_step_time": 8.167778730392456
    },
    {
      "epoch": 3.0670166015625e-06,
      "model_forward_time": 1.7470357418060303,
      "step": 77216
    },
    {
      "epoch": 3.0670166015625e-06,
      "step": 77216,
      "training_step_time": 8.337968111038208
    },
    {
      "epoch": 3.0679702758789063e-06,
      "model_forward_time": 1.748594045639038,
      "step": 77217
    },
    {
      "epoch": 3.0679702758789063e-06,
      "step": 77217,
      "training_step_time": 8.649270057678223
    },
    {
      "epoch": 3.0689239501953125e-06,
      "model_forward_time": 2.417172431945801,
      "step": 77218
    },
    {
      "epoch": 3.0689239501953125e-06,
      "step": 77218,
      "training_step_time": 10.120007038116455
    },
    {
      "epoch": 3.0698776245117186e-06,
      "model_forward_time": 1.7509093284606934,
      "step": 77219
    },
    {
      "epoch": 3.0698776245117186e-06,
      "step": 77219,
      "training_step_time": 8.44259262084961
    },
    {
      "epoch": 3.070831298828125e-06,
      "grad_norm": 0.07471908628940582,
      "learning_rate": 1.3530070143385966e-06,
      "loss": 0.0567,
      "step": 77220
    },
    {
      "epoch": 3.070831298828125e-06,
      "model_forward_time": 1.762160301208496,
      "step": 77220
    },
    {
      "dynamics_loss_avg": 0.047050094418227675,
      "epoch": 3.070831298828125e-06,
      "step": 77220
    },
    {
      "action_loss_avg": 0.0015651844587409869,
      "epoch": 3.070831298828125e-06,
      "step": 77220
    },
    {
      "epoch": 3.070831298828125e-06,
      "step": 77220,
      "training_step_time": 7.477641820907593
    },
    {
      "epoch": 3.0717849731445313e-06,
      "model_forward_time": 1.7433936595916748,
      "step": 77221
    },
    {
      "epoch": 3.0717849731445313e-06,
      "step": 77221,
      "training_step_time": 7.628920316696167
    },
    {
      "epoch": 3.0727386474609375e-06,
      "model_forward_time": 1.7509255409240723,
      "step": 77222
    },
    {
      "epoch": 3.0727386474609375e-06,
      "step": 77222,
      "training_step_time": 7.653900623321533
    },
    {
      "epoch": 3.0736923217773436e-06,
      "model_forward_time": 1.7490198612213135,
      "step": 77223
    },
    {
      "epoch": 3.0736923217773436e-06,
      "step": 77223,
      "training_step_time": 7.547735691070557
    },
    {
      "epoch": 3.07464599609375e-06,
      "model_forward_time": 1.7518649101257324,
      "step": 77224
    },
    {
      "epoch": 3.07464599609375e-06,
      "step": 77224,
      "training_step_time": 7.789446592330933
    },
    {
      "epoch": 3.0755996704101563e-06,
      "model_forward_time": 1.7469828128814697,
      "step": 77225
    },
    {
      "epoch": 3.0755996704101563e-06,
      "step": 77225,
      "training_step_time": 8.772422075271606
    },
    {
      "epoch": 3.0765533447265625e-06,
      "model_forward_time": 1.7501194477081299,
      "step": 77226
    },
    {
      "epoch": 3.0765533447265625e-06,
      "step": 77226,
      "training_step_time": 9.27303671836853
    },
    {
      "epoch": 3.0775070190429686e-06,
      "model_forward_time": 2.359779119491577,
      "step": 77227
    },
    {
      "epoch": 3.0775070190429686e-06,
      "step": 77227,
      "training_step_time": 8.655708312988281
    },
    {
      "epoch": 3.078460693359375e-06,
      "model_forward_time": 1.7715165615081787,
      "step": 77228
    },
    {
      "epoch": 3.078460693359375e-06,
      "step": 77228,
      "training_step_time": 9.482837915420532
    },
    {
      "epoch": 3.0794143676757813e-06,
      "model_forward_time": 1.3870086669921875,
      "step": 77229
    },
    {
      "epoch": 3.0794143676757813e-06,
      "step": 77229,
      "training_step_time": 8.427602529525757
    },
    {
      "epoch": 3.0803680419921875e-06,
      "grad_norm": 0.09906396269798279,
      "learning_rate": 1.3518760933486702e-06,
      "loss": 0.0573,
      "step": 77230
    },
    {
      "epoch": 3.0803680419921875e-06,
      "model_forward_time": 1.3744778633117676,
      "step": 77230
    },
    {
      "dynamics_loss_avg": 0.0648004688322544,
      "epoch": 3.0803680419921875e-06,
      "step": 77230
    },
    {
      "action_loss_avg": 0.001601710938848555,
      "epoch": 3.0803680419921875e-06,
      "step": 77230
    },
    {
      "epoch": 3.0803680419921875e-06,
      "step": 77230,
      "training_step_time": 8.300551176071167
    },
    {
      "epoch": 3.0813217163085936e-06,
      "model_forward_time": 1.7402470111846924,
      "step": 77231
    },
    {
      "epoch": 3.0813217163085936e-06,
      "step": 77231,
      "training_step_time": 7.833920240402222
    },
    {
      "epoch": 3.082275390625e-06,
      "model_forward_time": 1.744185447692871,
      "step": 77232
    },
    {
      "epoch": 3.082275390625e-06,
      "step": 77232,
      "training_step_time": 7.632946252822876
    },
    {
      "epoch": 3.0832290649414063e-06,
      "model_forward_time": 1.752032995223999,
      "step": 77233
    },
    {
      "epoch": 3.0832290649414063e-06,
      "step": 77233,
      "training_step_time": 7.773086071014404
    },
    {
      "epoch": 3.0841827392578125e-06,
      "model_forward_time": 1.7460567951202393,
      "step": 77234
    },
    {
      "epoch": 3.0841827392578125e-06,
      "step": 77234,
      "training_step_time": 8.106622695922852
    },
    {
      "epoch": 3.0851364135742186e-06,
      "model_forward_time": 1.7413856983184814,
      "step": 77235
    },
    {
      "epoch": 3.0851364135742186e-06,
      "step": 77235,
      "training_step_time": 9.917335987091064
    },
    {
      "epoch": 3.086090087890625e-06,
      "model_forward_time": 1.7483711242675781,
      "step": 77236
    },
    {
      "epoch": 3.086090087890625e-06,
      "step": 77236,
      "training_step_time": 9.44479751586914
    },
    {
      "epoch": 3.0870437622070313e-06,
      "model_forward_time": 2.112410545349121,
      "step": 77237
    },
    {
      "epoch": 3.0870437622070313e-06,
      "step": 77237,
      "training_step_time": 8.620440483093262
    },
    {
      "epoch": 3.0879974365234375e-06,
      "model_forward_time": 1.758971929550171,
      "step": 77238
    },
    {
      "epoch": 3.0879974365234375e-06,
      "step": 77238,
      "training_step_time": 8.693150997161865
    },
    {
      "epoch": 3.0889511108398436e-06,
      "model_forward_time": 1.7482080459594727,
      "step": 77239
    },
    {
      "epoch": 3.0889511108398436e-06,
      "step": 77239,
      "training_step_time": 8.287471294403076
    },
    {
      "epoch": 3.08990478515625e-06,
      "grad_norm": 0.1379166543483734,
      "learning_rate": 1.3507455713120908e-06,
      "loss": 0.0572,
      "step": 77240
    },
    {
      "epoch": 3.08990478515625e-06,
      "model_forward_time": 1.7468516826629639,
      "step": 77240
    },
    {
      "dynamics_loss_avg": 0.04277694588527083,
      "epoch": 3.08990478515625e-06,
      "step": 77240
    },
    {
      "action_loss_avg": 0.0010624718910548837,
      "epoch": 3.08990478515625e-06,
      "step": 77240
    },
    {
      "epoch": 3.08990478515625e-06,
      "step": 77240,
      "training_step_time": 8.28372573852539
    },
    {
      "epoch": 3.0908584594726563e-06,
      "model_forward_time": 1.7459867000579834,
      "step": 77241
    },
    {
      "epoch": 3.0908584594726563e-06,
      "step": 77241,
      "training_step_time": 8.473392248153687
    },
    {
      "epoch": 3.0918121337890625e-06,
      "model_forward_time": 1.7490944862365723,
      "step": 77242
    },
    {
      "epoch": 3.0918121337890625e-06,
      "step": 77242,
      "training_step_time": 8.508445739746094
    },
    {
      "epoch": 3.0927658081054686e-06,
      "model_forward_time": 1.7363483905792236,
      "step": 77243
    },
    {
      "epoch": 3.0927658081054686e-06,
      "step": 77243,
      "training_step_time": 8.194247007369995
    },
    {
      "epoch": 3.093719482421875e-06,
      "model_forward_time": 1.750457763671875,
      "step": 77244
    },
    {
      "epoch": 3.093719482421875e-06,
      "step": 77244,
      "training_step_time": 9.44218897819519
    },
    {
      "epoch": 3.0946731567382814e-06,
      "model_forward_time": 1.7533059120178223,
      "step": 77245
    },
    {
      "epoch": 3.0946731567382814e-06,
      "step": 77245,
      "training_step_time": 8.244953870773315
    },
    {
      "epoch": 3.0956268310546875e-06,
      "model_forward_time": 2.4613518714904785,
      "step": 77246
    },
    {
      "epoch": 3.0956268310546875e-06,
      "step": 77246,
      "training_step_time": 7.746080636978149
    },
    {
      "epoch": 3.0965805053710936e-06,
      "model_forward_time": 1.7458302974700928,
      "step": 77247
    },
    {
      "epoch": 3.0965805053710936e-06,
      "step": 77247,
      "training_step_time": 8.81855297088623
    },
    {
      "epoch": 3.0975341796875e-06,
      "model_forward_time": 1.765732765197754,
      "step": 77248
    },
    {
      "epoch": 3.0975341796875e-06,
      "step": 77248,
      "training_step_time": 8.083582162857056
    },
    {
      "epoch": 3.0984878540039064e-06,
      "model_forward_time": 1.7467029094696045,
      "step": 77249
    },
    {
      "epoch": 3.0984878540039064e-06,
      "step": 77249,
      "training_step_time": 7.74101710319519
    },
    {
      "epoch": 3.0994415283203125e-06,
      "grad_norm": 0.10163131356239319,
      "learning_rate": 1.3496154483524876e-06,
      "loss": 0.0545,
      "step": 77250
    },
    {
      "epoch": 3.0994415283203125e-06,
      "model_forward_time": 1.7460525035858154,
      "step": 77250
    },
    {
      "dynamics_loss_avg": 0.05604975074529648,
      "epoch": 3.0994415283203125e-06,
      "step": 77250
    },
    {
      "action_loss_avg": 0.00176316594297532,
      "epoch": 3.0994415283203125e-06,
      "step": 77250
    },
    {
      "epoch": 3.0994415283203125e-06,
      "step": 77250,
      "training_step_time": 7.837018013000488
    },
    {
      "epoch": 3.1003952026367186e-06,
      "model_forward_time": 1.7514712810516357,
      "step": 77251
    },
    {
      "epoch": 3.1003952026367186e-06,
      "step": 77251,
      "training_step_time": 7.77346134185791
    },
    {
      "epoch": 3.101348876953125e-06,
      "model_forward_time": 1.747856616973877,
      "step": 77252
    },
    {
      "epoch": 3.101348876953125e-06,
      "step": 77252,
      "training_step_time": 8.508856058120728
    },
    {
      "epoch": 3.1023025512695314e-06,
      "model_forward_time": 1.7514193058013916,
      "step": 77253
    },
    {
      "epoch": 3.1023025512695314e-06,
      "step": 77253,
      "training_step_time": 7.798373222351074
    },
    {
      "epoch": 3.1032562255859375e-06,
      "model_forward_time": 1.7461063861846924,
      "step": 77254
    },
    {
      "epoch": 3.1032562255859375e-06,
      "step": 77254,
      "training_step_time": 8.567606210708618
    },
    {
      "epoch": 3.1042098999023436e-06,
      "model_forward_time": 2.3941688537597656,
      "step": 77255
    },
    {
      "epoch": 3.1042098999023436e-06,
      "step": 77255,
      "training_step_time": 10.789755582809448
    },
    {
      "epoch": 3.10516357421875e-06,
      "model_forward_time": 1.7516076564788818,
      "step": 77256
    },
    {
      "epoch": 3.10516357421875e-06,
      "step": 77256,
      "training_step_time": 7.763937711715698
    },
    {
      "epoch": 3.1061172485351564e-06,
      "model_forward_time": 1.7787814140319824,
      "step": 77257
    },
    {
      "epoch": 3.1061172485351564e-06,
      "step": 77257,
      "training_step_time": 8.39905333518982
    },
    {
      "epoch": 3.1070709228515625e-06,
      "model_forward_time": 1.7496330738067627,
      "step": 77258
    },
    {
      "epoch": 3.1070709228515625e-06,
      "step": 77258,
      "training_step_time": 8.36711859703064
    },
    {
      "epoch": 3.1080245971679687e-06,
      "model_forward_time": 1.753490686416626,
      "step": 77259
    },
    {
      "epoch": 3.1080245971679687e-06,
      "step": 77259,
      "training_step_time": 8.80804443359375
    },
    {
      "epoch": 3.108978271484375e-06,
      "grad_norm": 0.08353716880083084,
      "learning_rate": 1.3484857245934507e-06,
      "loss": 0.0546,
      "step": 77260
    },
    {
      "epoch": 3.108978271484375e-06,
      "model_forward_time": 1.7374598979949951,
      "step": 77260
    },
    {
      "dynamics_loss_avg": 0.05160830710083246,
      "epoch": 3.108978271484375e-06,
      "step": 77260
    },
    {
      "action_loss_avg": 0.001635884292772971,
      "epoch": 3.108978271484375e-06,
      "step": 77260
    },
    {
      "epoch": 3.108978271484375e-06,
      "step": 77260,
      "training_step_time": 7.770057678222656
    },
    {
      "epoch": 3.1099319458007814e-06,
      "model_forward_time": 1.7526817321777344,
      "step": 77261
    },
    {
      "epoch": 3.1099319458007814e-06,
      "step": 77261,
      "training_step_time": 8.862822532653809
    },
    {
      "epoch": 3.1108856201171875e-06,
      "model_forward_time": 1.7538785934448242,
      "step": 77262
    },
    {
      "epoch": 3.1108856201171875e-06,
      "step": 77262,
      "training_step_time": 7.748746395111084
    },
    {
      "epoch": 3.1118392944335937e-06,
      "model_forward_time": 1.7511110305786133,
      "step": 77263
    },
    {
      "epoch": 3.1118392944335937e-06,
      "step": 77263,
      "training_step_time": 7.691039562225342
    },
    {
      "epoch": 3.11279296875e-06,
      "model_forward_time": 1.749248743057251,
      "step": 77264
    },
    {
      "epoch": 3.11279296875e-06,
      "step": 77264,
      "training_step_time": 9.255770444869995
    },
    {
      "epoch": 3.1137466430664064e-06,
      "model_forward_time": 1.774764060974121,
      "step": 77265
    },
    {
      "epoch": 3.1137466430664064e-06,
      "step": 77265,
      "training_step_time": 8.4753258228302
    },
    {
      "epoch": 3.1147003173828125e-06,
      "model_forward_time": 1.7692327499389648,
      "step": 77266
    },
    {
      "epoch": 3.1147003173828125e-06,
      "step": 77266,
      "training_step_time": 7.596280097961426
    },
    {
      "epoch": 3.1156539916992187e-06,
      "model_forward_time": 1.7532975673675537,
      "step": 77267
    },
    {
      "epoch": 3.1156539916992187e-06,
      "step": 77267,
      "training_step_time": 7.7378623485565186
    },
    {
      "epoch": 3.116607666015625e-06,
      "model_forward_time": 1.7502861022949219,
      "step": 77268
    },
    {
      "epoch": 3.116607666015625e-06,
      "step": 77268,
      "training_step_time": 7.800156354904175
    },
    {
      "epoch": 3.1175613403320314e-06,
      "model_forward_time": 1.7456309795379639,
      "step": 77269
    },
    {
      "epoch": 3.1175613403320314e-06,
      "step": 77269,
      "training_step_time": 7.626919746398926
    },
    {
      "epoch": 3.1185150146484375e-06,
      "grad_norm": 0.07258866727352142,
      "learning_rate": 1.3473564001585248e-06,
      "loss": 0.0547,
      "step": 77270
    },
    {
      "epoch": 3.1185150146484375e-06,
      "model_forward_time": 1.7417936325073242,
      "step": 77270
    },
    {
      "dynamics_loss_avg": 0.06207552663981915,
      "epoch": 3.1185150146484375e-06,
      "step": 77270
    },
    {
      "action_loss_avg": 0.0015052336500957608,
      "epoch": 3.1185150146484375e-06,
      "step": 77270
    },
    {
      "epoch": 3.1185150146484375e-06,
      "step": 77270,
      "training_step_time": 8.556887865066528
    },
    {
      "epoch": 3.1194686889648437e-06,
      "model_forward_time": 1.7479243278503418,
      "step": 77271
    },
    {
      "epoch": 3.1194686889648437e-06,
      "step": 77271,
      "training_step_time": 8.66557002067566
    },
    {
      "epoch": 3.12042236328125e-06,
      "model_forward_time": 1.750760793685913,
      "step": 77272
    },
    {
      "epoch": 3.12042236328125e-06,
      "step": 77272,
      "training_step_time": 7.623958110809326
    },
    {
      "epoch": 3.1213760375976564e-06,
      "model_forward_time": 1.7455201148986816,
      "step": 77273
    },
    {
      "epoch": 3.1213760375976564e-06,
      "step": 77273,
      "training_step_time": 8.245587825775146
    },
    {
      "epoch": 3.1223297119140625e-06,
      "model_forward_time": 2.3824708461761475,
      "step": 77274
    },
    {
      "epoch": 3.1223297119140625e-06,
      "step": 77274,
      "training_step_time": 7.768258333206177
    },
    {
      "epoch": 3.1232833862304687e-06,
      "model_forward_time": 1.7671570777893066,
      "step": 77275
    },
    {
      "epoch": 3.1232833862304687e-06,
      "step": 77275,
      "training_step_time": 8.471419334411621
    },
    {
      "epoch": 3.124237060546875e-06,
      "model_forward_time": 1.743427038192749,
      "step": 77276
    },
    {
      "epoch": 3.124237060546875e-06,
      "step": 77276,
      "training_step_time": 7.715238571166992
    },
    {
      "epoch": 3.1251907348632814e-06,
      "model_forward_time": 1.7472178936004639,
      "step": 77277
    },
    {
      "epoch": 3.1251907348632814e-06,
      "step": 77277,
      "training_step_time": 7.736009359359741
    },
    {
      "epoch": 3.1261444091796875e-06,
      "model_forward_time": 1.750171184539795,
      "step": 77278
    },
    {
      "epoch": 3.1261444091796875e-06,
      "step": 77278,
      "training_step_time": 8.761467695236206
    },
    {
      "epoch": 3.1270980834960937e-06,
      "model_forward_time": 1.7408523559570312,
      "step": 77279
    },
    {
      "epoch": 3.1270980834960937e-06,
      "step": 77279,
      "training_step_time": 7.7458415031433105
    },
    {
      "epoch": 3.1280517578125e-06,
      "grad_norm": 0.08993323892354965,
      "learning_rate": 1.3462274751712128e-06,
      "loss": 0.0543,
      "step": 77280
    },
    {
      "epoch": 3.1280517578125e-06,
      "model_forward_time": 1.739048719406128,
      "step": 77280
    },
    {
      "dynamics_loss_avg": 0.051995206810534,
      "epoch": 3.1280517578125e-06,
      "step": 77280
    },
    {
      "action_loss_avg": 0.0011968588136369363,
      "epoch": 3.1280517578125e-06,
      "step": 77280
    },
    {
      "epoch": 3.1280517578125e-06,
      "step": 77280,
      "training_step_time": 7.6741907596588135
    },
    {
      "epoch": 3.1290054321289064e-06,
      "model_forward_time": 1.7296698093414307,
      "step": 77281
    },
    {
      "epoch": 3.1290054321289064e-06,
      "step": 77281,
      "training_step_time": 7.798471212387085
    },
    {
      "epoch": 3.1299591064453125e-06,
      "model_forward_time": 1.7530150413513184,
      "step": 77282
    },
    {
      "epoch": 3.1299591064453125e-06,
      "step": 77282,
      "training_step_time": 8.719297647476196
    },
    {
      "epoch": 3.1309127807617187e-06,
      "model_forward_time": 2.4282262325286865,
      "step": 77283
    },
    {
      "epoch": 3.1309127807617187e-06,
      "step": 77283,
      "training_step_time": 9.717220544815063
    },
    {
      "epoch": 3.131866455078125e-06,
      "model_forward_time": 1.7454025745391846,
      "step": 77284
    },
    {
      "epoch": 3.131866455078125e-06,
      "step": 77284,
      "training_step_time": 8.821090459823608
    },
    {
      "epoch": 3.1328201293945314e-06,
      "model_forward_time": 1.7501115798950195,
      "step": 77285
    },
    {
      "epoch": 3.1328201293945314e-06,
      "step": 77285,
      "training_step_time": 9.260913372039795
    },
    {
      "epoch": 3.1337738037109375e-06,
      "model_forward_time": 1.7474071979522705,
      "step": 77286
    },
    {
      "epoch": 3.1337738037109375e-06,
      "step": 77286,
      "training_step_time": 7.716688394546509
    },
    {
      "epoch": 3.1347274780273437e-06,
      "model_forward_time": 1.750091552734375,
      "step": 77287
    },
    {
      "epoch": 3.1347274780273437e-06,
      "step": 77287,
      "training_step_time": 7.7910425662994385
    },
    {
      "epoch": 3.13568115234375e-06,
      "model_forward_time": 1.7465102672576904,
      "step": 77288
    },
    {
      "epoch": 3.13568115234375e-06,
      "step": 77288,
      "training_step_time": 7.604572296142578
    },
    {
      "epoch": 3.1366348266601564e-06,
      "model_forward_time": 1.745225191116333,
      "step": 77289
    },
    {
      "epoch": 3.1366348266601564e-06,
      "step": 77289,
      "training_step_time": 7.688873767852783
    },
    {
      "epoch": 3.1375885009765625e-06,
      "grad_norm": 0.09217964857816696,
      "learning_rate": 1.34509894975497e-06,
      "loss": 0.0539,
      "step": 77290
    },
    {
      "epoch": 3.1375885009765625e-06,
      "model_forward_time": 1.748788595199585,
      "step": 77290
    },
    {
      "dynamics_loss_avg": 0.05878002047538757,
      "epoch": 3.1375885009765625e-06,
      "step": 77290
    },
    {
      "action_loss_avg": 0.0018278502859175206,
      "epoch": 3.1375885009765625e-06,
      "step": 77290
    },
    {
      "epoch": 3.1375885009765625e-06,
      "step": 77290,
      "training_step_time": 7.698300838470459
    },
    {
      "epoch": 3.1385421752929687e-06,
      "model_forward_time": 1.738732099533081,
      "step": 77291
    },
    {
      "epoch": 3.1385421752929687e-06,
      "step": 77291,
      "training_step_time": 8.803270816802979
    },
    {
      "epoch": 3.139495849609375e-06,
      "model_forward_time": 2.4507639408111572,
      "step": 77292
    },
    {
      "epoch": 3.139495849609375e-06,
      "step": 77292,
      "training_step_time": 8.755503177642822
    },
    {
      "epoch": 3.1404495239257814e-06,
      "model_forward_time": 1.736772060394287,
      "step": 77293
    },
    {
      "epoch": 3.1404495239257814e-06,
      "step": 77293,
      "training_step_time": 7.715014457702637
    },
    {
      "epoch": 3.1414031982421875e-06,
      "model_forward_time": 1.7675955295562744,
      "step": 77294
    },
    {
      "epoch": 3.1414031982421875e-06,
      "step": 77294,
      "training_step_time": 8.94866681098938
    },
    {
      "epoch": 3.1423568725585937e-06,
      "model_forward_time": 1.754492998123169,
      "step": 77295
    },
    {
      "epoch": 3.1423568725585937e-06,
      "step": 77295,
      "training_step_time": 8.251901388168335
    },
    {
      "epoch": 3.143310546875e-06,
      "model_forward_time": 1.749370813369751,
      "step": 77296
    },
    {
      "epoch": 3.143310546875e-06,
      "step": 77296,
      "training_step_time": 8.611098527908325
    },
    {
      "epoch": 3.1442642211914064e-06,
      "model_forward_time": 1.742433786392212,
      "step": 77297
    },
    {
      "epoch": 3.1442642211914064e-06,
      "step": 77297,
      "training_step_time": 7.867574214935303
    },
    {
      "epoch": 3.1452178955078125e-06,
      "model_forward_time": 1.7520599365234375,
      "step": 77298
    },
    {
      "epoch": 3.1452178955078125e-06,
      "step": 77298,
      "training_step_time": 7.681899070739746
    },
    {
      "epoch": 3.1461715698242187e-06,
      "model_forward_time": 1.7456917762756348,
      "step": 77299
    },
    {
      "epoch": 3.1461715698242187e-06,
      "step": 77299,
      "training_step_time": 8.22974419593811
    },
    {
      "epoch": 3.147125244140625e-06,
      "grad_norm": 0.09005212783813477,
      "learning_rate": 1.3439708240332117e-06,
      "loss": 0.0544,
      "step": 77300
    },
    {
      "epoch": 3.147125244140625e-06,
      "model_forward_time": 1.7412152290344238,
      "step": 77300
    },
    {
      "dynamics_loss_avg": 0.04248839747160673,
      "epoch": 3.147125244140625e-06,
      "step": 77300
    },
    {
      "action_loss_avg": 0.0011343706049956382,
      "epoch": 3.147125244140625e-06,
      "step": 77300
    },
    {
      "epoch": 3.147125244140625e-06,
      "step": 77300,
      "training_step_time": 10.423201560974121
    },
    {
      "epoch": 3.1480789184570314e-06,
      "model_forward_time": 2.453505039215088,
      "step": 77301
    },
    {
      "epoch": 3.1480789184570314e-06,
      "step": 77301,
      "training_step_time": 8.339295387268066
    },
    {
      "epoch": 3.1490325927734375e-06,
      "model_forward_time": 1.7670783996582031,
      "step": 77302
    },
    {
      "epoch": 3.1490325927734375e-06,
      "step": 77302,
      "training_step_time": 8.714553594589233
    },
    {
      "epoch": 3.1499862670898437e-06,
      "model_forward_time": 1.7658729553222656,
      "step": 77303
    },
    {
      "epoch": 3.1499862670898437e-06,
      "step": 77303,
      "training_step_time": 9.244884729385376
    },
    {
      "epoch": 3.15093994140625e-06,
      "model_forward_time": 1.750342845916748,
      "step": 77304
    },
    {
      "epoch": 3.15093994140625e-06,
      "step": 77304,
      "training_step_time": 7.811669826507568
    },
    {
      "epoch": 3.1518936157226564e-06,
      "model_forward_time": 1.7428503036499023,
      "step": 77305
    },
    {
      "epoch": 3.1518936157226564e-06,
      "step": 77305,
      "training_step_time": 7.86850905418396
    },
    {
      "epoch": 3.1528472900390625e-06,
      "model_forward_time": 1.750720739364624,
      "step": 77306
    },
    {
      "epoch": 3.1528472900390625e-06,
      "step": 77306,
      "training_step_time": 7.554250240325928
    },
    {
      "epoch": 3.1538009643554687e-06,
      "model_forward_time": 1.74125075340271,
      "step": 77307
    },
    {
      "epoch": 3.1538009643554687e-06,
      "step": 77307,
      "training_step_time": 7.746511697769165
    },
    {
      "epoch": 3.154754638671875e-06,
      "model_forward_time": 1.7497308254241943,
      "step": 77308
    },
    {
      "epoch": 3.154754638671875e-06,
      "step": 77308,
      "training_step_time": 8.239544153213501
    },
    {
      "epoch": 3.1557083129882814e-06,
      "model_forward_time": 1.7456316947937012,
      "step": 77309
    },
    {
      "epoch": 3.1557083129882814e-06,
      "step": 77309,
      "training_step_time": 8.546967029571533
    },
    {
      "epoch": 3.1566619873046876e-06,
      "grad_norm": 0.06818767637014389,
      "learning_rate": 1.3428430981293044e-06,
      "loss": 0.0549,
      "step": 77310
    },
    {
      "epoch": 3.1566619873046876e-06,
      "model_forward_time": 2.4000799655914307,
      "step": 77310
    },
    {
      "dynamics_loss_avg": 0.07538283318281173,
      "epoch": 3.1566619873046876e-06,
      "step": 77310
    },
    {
      "action_loss_avg": 0.001804791419999674,
      "epoch": 3.1566619873046876e-06,
      "step": 77310
    },
    {
      "epoch": 3.1566619873046876e-06,
      "step": 77310,
      "training_step_time": 8.511584758758545
    },
    {
      "epoch": 3.1576156616210937e-06,
      "model_forward_time": 1.772557258605957,
      "step": 77311
    },
    {
      "epoch": 3.1576156616210937e-06,
      "step": 77311,
      "training_step_time": 7.7988386154174805
    },
    {
      "epoch": 3.1585693359375e-06,
      "model_forward_time": 1.7613012790679932,
      "step": 77312
    },
    {
      "epoch": 3.1585693359375e-06,
      "step": 77312,
      "training_step_time": 7.785877704620361
    },
    {
      "epoch": 3.1595230102539064e-06,
      "model_forward_time": 1.7475636005401611,
      "step": 77313
    },
    {
      "epoch": 3.1595230102539064e-06,
      "step": 77313,
      "training_step_time": 8.352835655212402
    },
    {
      "epoch": 3.1604766845703126e-06,
      "model_forward_time": 1.752769947052002,
      "step": 77314
    },
    {
      "epoch": 3.1604766845703126e-06,
      "step": 77314,
      "training_step_time": 7.802489280700684
    },
    {
      "epoch": 3.1614303588867187e-06,
      "model_forward_time": 1.7491185665130615,
      "step": 77315
    },
    {
      "epoch": 3.1614303588867187e-06,
      "step": 77315,
      "training_step_time": 8.409852027893066
    },
    {
      "epoch": 3.162384033203125e-06,
      "model_forward_time": 1.7475183010101318,
      "step": 77316
    },
    {
      "epoch": 3.162384033203125e-06,
      "step": 77316,
      "training_step_time": 7.759596824645996
    },
    {
      "epoch": 3.1633377075195314e-06,
      "model_forward_time": 1.7520906925201416,
      "step": 77317
    },
    {
      "epoch": 3.1633377075195314e-06,
      "step": 77317,
      "training_step_time": 8.566588163375854
    },
    {
      "epoch": 3.1642913818359376e-06,
      "model_forward_time": 1.7478148937225342,
      "step": 77318
    },
    {
      "epoch": 3.1642913818359376e-06,
      "step": 77318,
      "training_step_time": 9.511343717575073
    },
    {
      "epoch": 3.1652450561523437e-06,
      "model_forward_time": 1.745600938796997,
      "step": 77319
    },
    {
      "epoch": 3.1652450561523437e-06,
      "step": 77319,
      "training_step_time": 9.586215734481812
    },
    {
      "epoch": 3.16619873046875e-06,
      "grad_norm": 0.07675794512033463,
      "learning_rate": 1.3417157721665807e-06,
      "loss": 0.0546,
      "step": 77320
    },
    {
      "epoch": 3.16619873046875e-06,
      "model_forward_time": 1.752734899520874,
      "step": 77320
    },
    {
      "dynamics_loss_avg": 0.05012369565665722,
      "epoch": 3.16619873046875e-06,
      "step": 77320
    },
    {
      "action_loss_avg": 0.0013036335323704407,
      "epoch": 3.16619873046875e-06,
      "step": 77320
    },
    {
      "epoch": 3.16619873046875e-06,
      "step": 77320,
      "training_step_time": 8.63805103302002
    },
    {
      "epoch": 3.1671524047851564e-06,
      "model_forward_time": 1.7526297569274902,
      "step": 77321
    },
    {
      "epoch": 3.1671524047851564e-06,
      "step": 77321,
      "training_step_time": 8.608605146408081
    },
    {
      "epoch": 3.1681060791015626e-06,
      "model_forward_time": 1.7484965324401855,
      "step": 77322
    },
    {
      "epoch": 3.1681060791015626e-06,
      "step": 77322,
      "training_step_time": 7.709228515625
    },
    {
      "epoch": 3.1690597534179687e-06,
      "model_forward_time": 1.3704266548156738,
      "step": 77323
    },
    {
      "epoch": 3.1690597534179687e-06,
      "step": 77323,
      "training_step_time": 7.850446701049805
    },
    {
      "epoch": 3.170013427734375e-06,
      "model_forward_time": 1.754960298538208,
      "step": 77324
    },
    {
      "epoch": 3.170013427734375e-06,
      "step": 77324,
      "training_step_time": 8.802932500839233
    },
    {
      "epoch": 3.1709671020507814e-06,
      "model_forward_time": 1.7547647953033447,
      "step": 77325
    },
    {
      "epoch": 3.1709671020507814e-06,
      "step": 77325,
      "training_step_time": 7.7328200340271
    },
    {
      "epoch": 3.1719207763671876e-06,
      "model_forward_time": 1.7336204051971436,
      "step": 77326
    },
    {
      "epoch": 3.1719207763671876e-06,
      "step": 77326,
      "training_step_time": 7.759511470794678
    },
    {
      "epoch": 3.1728744506835937e-06,
      "model_forward_time": 1.7523369789123535,
      "step": 77327
    },
    {
      "epoch": 3.1728744506835937e-06,
      "step": 77327,
      "training_step_time": 7.7348127365112305
    },
    {
      "epoch": 3.173828125e-06,
      "model_forward_time": 1.758134126663208,
      "step": 77328
    },
    {
      "epoch": 3.173828125e-06,
      "step": 77328,
      "training_step_time": 7.7996556758880615
    },
    {
      "epoch": 3.1747817993164064e-06,
      "model_forward_time": 2.4175848960876465,
      "step": 77329
    },
    {
      "epoch": 3.1747817993164064e-06,
      "step": 77329,
      "training_step_time": 7.75598669052124
    },
    {
      "epoch": 3.1757354736328126e-06,
      "grad_norm": 0.10607819259166718,
      "learning_rate": 1.3405888462683186e-06,
      "loss": 0.055,
      "step": 77330
    },
    {
      "epoch": 3.1757354736328126e-06,
      "model_forward_time": 1.752901315689087,
      "step": 77330
    },
    {
      "dynamics_loss_avg": 0.05042745769023895,
      "epoch": 3.1757354736328126e-06,
      "step": 77330
    },
    {
      "action_loss_avg": 0.002314000259502791,
      "epoch": 3.1757354736328126e-06,
      "step": 77330
    },
    {
      "epoch": 3.1757354736328126e-06,
      "step": 77330,
      "training_step_time": 8.500646591186523
    },
    {
      "epoch": 3.1766891479492187e-06,
      "model_forward_time": 1.7800004482269287,
      "step": 77331
    },
    {
      "epoch": 3.1766891479492187e-06,
      "step": 77331,
      "training_step_time": 7.677499294281006
    },
    {
      "epoch": 3.177642822265625e-06,
      "model_forward_time": 1.7674980163574219,
      "step": 77332
    },
    {
      "epoch": 3.177642822265625e-06,
      "step": 77332,
      "training_step_time": 7.790306568145752
    },
    {
      "epoch": 3.1785964965820314e-06,
      "model_forward_time": 1.7826945781707764,
      "step": 77333
    },
    {
      "epoch": 3.1785964965820314e-06,
      "step": 77333,
      "training_step_time": 7.791571855545044
    },
    {
      "epoch": 3.1795501708984376e-06,
      "model_forward_time": 1.7750580310821533,
      "step": 77334
    },
    {
      "epoch": 3.1795501708984376e-06,
      "step": 77334,
      "training_step_time": 7.844037771224976
    },
    {
      "epoch": 3.1805038452148437e-06,
      "model_forward_time": 1.7737927436828613,
      "step": 77335
    },
    {
      "epoch": 3.1805038452148437e-06,
      "step": 77335,
      "training_step_time": 7.791327476501465
    },
    {
      "epoch": 3.18145751953125e-06,
      "model_forward_time": 1.7812201976776123,
      "step": 77336
    },
    {
      "epoch": 3.18145751953125e-06,
      "step": 77336,
      "training_step_time": 9.35948371887207
    },
    {
      "epoch": 3.1824111938476564e-06,
      "model_forward_time": 1.771791934967041,
      "step": 77337
    },
    {
      "epoch": 3.1824111938476564e-06,
      "step": 77337,
      "training_step_time": 9.783241510391235
    },
    {
      "epoch": 3.1833648681640626e-06,
      "model_forward_time": 2.478823184967041,
      "step": 77338
    },
    {
      "epoch": 3.1833648681640626e-06,
      "step": 77338,
      "training_step_time": 9.026211977005005
    },
    {
      "epoch": 3.1843185424804687e-06,
      "model_forward_time": 1.7802023887634277,
      "step": 77339
    },
    {
      "epoch": 3.1843185424804687e-06,
      "step": 77339,
      "training_step_time": 8.673479080200195
    },
    {
      "epoch": 3.185272216796875e-06,
      "grad_norm": 0.39847543835639954,
      "learning_rate": 1.3394623205577595e-06,
      "loss": 0.0534,
      "step": 77340
    },
    {
      "epoch": 3.185272216796875e-06,
      "model_forward_time": 1.78513503074646,
      "step": 77340
    },
    {
      "dynamics_loss_avg": 0.046502945758402345,
      "epoch": 3.185272216796875e-06,
      "step": 77340
    },
    {
      "action_loss_avg": 0.0019440068645053543,
      "epoch": 3.185272216796875e-06,
      "step": 77340
    },
    {
      "epoch": 3.185272216796875e-06,
      "step": 77340,
      "training_step_time": 8.798717975616455
    },
    {
      "epoch": 3.1862258911132814e-06,
      "model_forward_time": 1.7809231281280518,
      "step": 77341
    },
    {
      "epoch": 3.1862258911132814e-06,
      "step": 77341,
      "training_step_time": 7.941455364227295
    },
    {
      "epoch": 3.1871795654296876e-06,
      "model_forward_time": 1.7807555198669434,
      "step": 77342
    },
    {
      "epoch": 3.1871795654296876e-06,
      "step": 77342,
      "training_step_time": 7.9060094356536865
    },
    {
      "epoch": 3.1881332397460937e-06,
      "model_forward_time": 1.7734434604644775,
      "step": 77343
    },
    {
      "epoch": 3.1881332397460937e-06,
      "step": 77343,
      "training_step_time": 8.719668865203857
    },
    {
      "epoch": 3.1890869140625e-06,
      "model_forward_time": 1.7779641151428223,
      "step": 77344
    },
    {
      "epoch": 3.1890869140625e-06,
      "step": 77344,
      "training_step_time": 7.816536903381348
    },
    {
      "epoch": 3.1900405883789064e-06,
      "model_forward_time": 1.780731201171875,
      "step": 77345
    },
    {
      "epoch": 3.1900405883789064e-06,
      "step": 77345,
      "training_step_time": 7.769708156585693
    },
    {
      "epoch": 3.1909942626953126e-06,
      "model_forward_time": 1.7672040462493896,
      "step": 77346
    },
    {
      "epoch": 3.1909942626953126e-06,
      "step": 77346,
      "training_step_time": 7.87174654006958
    },
    {
      "epoch": 3.1919479370117187e-06,
      "model_forward_time": 2.49613094329834,
      "step": 77347
    },
    {
      "epoch": 3.1919479370117187e-06,
      "step": 77347,
      "training_step_time": 7.926984786987305
    },
    {
      "epoch": 3.192901611328125e-06,
      "model_forward_time": 1.7804522514343262,
      "step": 77348
    },
    {
      "epoch": 3.192901611328125e-06,
      "step": 77348,
      "training_step_time": 8.432787895202637
    },
    {
      "epoch": 3.1938552856445314e-06,
      "model_forward_time": 1.8032848834991455,
      "step": 77349
    },
    {
      "epoch": 3.1938552856445314e-06,
      "step": 77349,
      "training_step_time": 7.864462614059448
    },
    {
      "epoch": 3.1948089599609376e-06,
      "grad_norm": 0.09561228007078171,
      "learning_rate": 1.3383361951580964e-06,
      "loss": 0.0553,
      "step": 77350
    },
    {
      "epoch": 3.1948089599609376e-06,
      "model_forward_time": 1.7725591659545898,
      "step": 77350
    },
    {
      "dynamics_loss_avg": 0.04797342531383038,
      "epoch": 3.1948089599609376e-06,
      "step": 77350
    },
    {
      "action_loss_avg": 0.001607955084182322,
      "epoch": 3.1948089599609376e-06,
      "step": 77350
    },
    {
      "epoch": 3.1948089599609376e-06,
      "step": 77350,
      "training_step_time": 7.838132858276367
    },
    {
      "epoch": 3.1957626342773437e-06,
      "model_forward_time": 1.7849788665771484,
      "step": 77351
    },
    {
      "epoch": 3.1957626342773437e-06,
      "step": 77351,
      "training_step_time": 7.9740211963653564
    },
    {
      "epoch": 3.19671630859375e-06,
      "model_forward_time": 1.7639200687408447,
      "step": 77352
    },
    {
      "epoch": 3.19671630859375e-06,
      "step": 77352,
      "training_step_time": 8.82690954208374
    },
    {
      "epoch": 3.1976699829101564e-06,
      "model_forward_time": 1.778442144393921,
      "step": 77353
    },
    {
      "epoch": 3.1976699829101564e-06,
      "step": 77353,
      "training_step_time": 7.844136714935303
    },
    {
      "epoch": 3.1986236572265626e-06,
      "model_forward_time": 1.7749683856964111,
      "step": 77354
    },
    {
      "epoch": 3.1986236572265626e-06,
      "step": 77354,
      "training_step_time": 7.964659690856934
    },
    {
      "epoch": 3.1995773315429687e-06,
      "model_forward_time": 1.779005765914917,
      "step": 77355
    },
    {
      "epoch": 3.1995773315429687e-06,
      "step": 77355,
      "training_step_time": 7.851221084594727
    },
    {
      "epoch": 3.200531005859375e-06,
      "model_forward_time": 2.450254201889038,
      "step": 77356
    },
    {
      "epoch": 3.200531005859375e-06,
      "step": 77356,
      "training_step_time": 7.882629156112671
    },
    {
      "epoch": 3.2014846801757814e-06,
      "model_forward_time": 1.7789764404296875,
      "step": 77357
    },
    {
      "epoch": 3.2014846801757814e-06,
      "step": 77357,
      "training_step_time": 7.937205791473389
    },
    {
      "epoch": 3.2024383544921876e-06,
      "model_forward_time": 1.7856433391571045,
      "step": 77358
    },
    {
      "epoch": 3.2024383544921876e-06,
      "step": 77358,
      "training_step_time": 8.529094219207764
    },
    {
      "epoch": 3.2033920288085937e-06,
      "model_forward_time": 1.811769723892212,
      "step": 77359
    },
    {
      "epoch": 3.2033920288085937e-06,
      "step": 77359,
      "training_step_time": 8.245727300643921
    },
    {
      "epoch": 3.204345703125e-06,
      "grad_norm": 0.07886933535337448,
      "learning_rate": 1.3372104701924831e-06,
      "loss": 0.056,
      "step": 77360
    },
    {
      "epoch": 3.204345703125e-06,
      "model_forward_time": 1.7840354442596436,
      "step": 77360
    },
    {
      "dynamics_loss_avg": 0.05132445180788636,
      "epoch": 3.204345703125e-06,
      "step": 77360
    },
    {
      "action_loss_avg": 0.0008473322377540171,
      "epoch": 3.204345703125e-06,
      "step": 77360
    },
    {
      "epoch": 3.204345703125e-06,
      "step": 77360,
      "training_step_time": 7.830787897109985
    },
    {
      "epoch": 3.2052993774414064e-06,
      "model_forward_time": 1.7705504894256592,
      "step": 77361
    },
    {
      "epoch": 3.2052993774414064e-06,
      "step": 77361,
      "training_step_time": 7.8395397663116455
    },
    {
      "epoch": 3.2062530517578126e-06,
      "model_forward_time": 1.78361177444458,
      "step": 77362
    },
    {
      "epoch": 3.2062530517578126e-06,
      "step": 77362,
      "training_step_time": 7.529961585998535
    },
    {
      "epoch": 3.2072067260742187e-06,
      "model_forward_time": 1.779902696609497,
      "step": 77363
    },
    {
      "epoch": 3.2072067260742187e-06,
      "step": 77363,
      "training_step_time": 7.751178979873657
    },
    {
      "epoch": 3.208160400390625e-06,
      "model_forward_time": 1.7746641635894775,
      "step": 77364
    },
    {
      "epoch": 3.208160400390625e-06,
      "step": 77364,
      "training_step_time": 7.78291392326355
    },
    {
      "epoch": 3.2091140747070315e-06,
      "model_forward_time": 1.7784416675567627,
      "step": 77365
    },
    {
      "epoch": 3.2091140747070315e-06,
      "step": 77365,
      "training_step_time": 8.317176103591919
    },
    {
      "epoch": 3.2100677490234376e-06,
      "model_forward_time": 2.456491708755493,
      "step": 77366
    },
    {
      "epoch": 3.2100677490234376e-06,
      "step": 77366,
      "training_step_time": 8.97719669342041
    },
    {
      "epoch": 3.2110214233398437e-06,
      "model_forward_time": 1.777341365814209,
      "step": 77367
    },
    {
      "epoch": 3.2110214233398437e-06,
      "step": 77367,
      "training_step_time": 7.639275312423706
    },
    {
      "epoch": 3.21197509765625e-06,
      "model_forward_time": 1.790797233581543,
      "step": 77368
    },
    {
      "epoch": 3.21197509765625e-06,
      "step": 77368,
      "training_step_time": 8.269025087356567
    },
    {
      "epoch": 3.2129287719726565e-06,
      "model_forward_time": 1.7644197940826416,
      "step": 77369
    },
    {
      "epoch": 3.2129287719726565e-06,
      "step": 77369,
      "training_step_time": 7.6551673412323
    },
    {
      "epoch": 3.2138824462890626e-06,
      "grad_norm": 0.09439194202423096,
      "learning_rate": 1.3360851457840245e-06,
      "loss": 0.056,
      "step": 77370
    },
    {
      "epoch": 3.2138824462890626e-06,
      "model_forward_time": 1.7722010612487793,
      "step": 77370
    },
    {
      "dynamics_loss_avg": 0.04983677193522453,
      "epoch": 3.2138824462890626e-06,
      "step": 77370
    },
    {
      "action_loss_avg": 0.0013994856970384717,
      "epoch": 3.2138824462890626e-06,
      "step": 77370
    },
    {
      "epoch": 3.2138824462890626e-06,
      "step": 77370,
      "training_step_time": 7.914781808853149
    },
    {
      "epoch": 3.2148361206054687e-06,
      "model_forward_time": 1.7644920349121094,
      "step": 77371
    },
    {
      "epoch": 3.2148361206054687e-06,
      "step": 77371,
      "training_step_time": 7.667524576187134
    },
    {
      "epoch": 3.215789794921875e-06,
      "model_forward_time": 1.769364833831787,
      "step": 77372
    },
    {
      "epoch": 3.215789794921875e-06,
      "step": 77372,
      "training_step_time": 7.820479869842529
    },
    {
      "epoch": 3.2167434692382815e-06,
      "model_forward_time": 1.7628562450408936,
      "step": 77373
    },
    {
      "epoch": 3.2167434692382815e-06,
      "step": 77373,
      "training_step_time": 8.27340316772461
    },
    {
      "epoch": 3.2176971435546876e-06,
      "model_forward_time": 1.7699134349822998,
      "step": 77374
    },
    {
      "epoch": 3.2176971435546876e-06,
      "step": 77374,
      "training_step_time": 7.7784342765808105
    },
    {
      "epoch": 3.2186508178710938e-06,
      "model_forward_time": 2.467219352722168,
      "step": 77375
    },
    {
      "epoch": 3.2186508178710938e-06,
      "step": 77375,
      "training_step_time": 10.246836423873901
    },
    {
      "epoch": 3.2196044921875e-06,
      "model_forward_time": 1.7739036083221436,
      "step": 77376
    },
    {
      "epoch": 3.2196044921875e-06,
      "step": 77376,
      "training_step_time": 9.4288010597229
    },
    {
      "epoch": 3.220558166503906e-06,
      "model_forward_time": 1.7875781059265137,
      "step": 77377
    },
    {
      "epoch": 3.220558166503906e-06,
      "step": 77377,
      "training_step_time": 7.719799518585205
    },
    {
      "epoch": 3.2215118408203126e-06,
      "model_forward_time": 1.776453971862793,
      "step": 77378
    },
    {
      "epoch": 3.2215118408203126e-06,
      "step": 77378,
      "training_step_time": 7.598205804824829
    },
    {
      "epoch": 3.2224655151367188e-06,
      "model_forward_time": 1.7614283561706543,
      "step": 77379
    },
    {
      "epoch": 3.2224655151367188e-06,
      "step": 77379,
      "training_step_time": 7.657273292541504
    },
    {
      "epoch": 3.223419189453125e-06,
      "grad_norm": 0.09281063824892044,
      "learning_rate": 1.3349602220557856e-06,
      "loss": 0.0556,
      "step": 77380
    },
    {
      "epoch": 3.223419189453125e-06,
      "model_forward_time": 1.768742322921753,
      "step": 77380
    },
    {
      "dynamics_loss_avg": 0.044948953296989204,
      "epoch": 3.223419189453125e-06,
      "step": 77380
    },
    {
      "action_loss_avg": 0.0013551085401559248,
      "epoch": 3.223419189453125e-06,
      "step": 77380
    },
    {
      "epoch": 3.223419189453125e-06,
      "step": 77380,
      "training_step_time": 7.596564054489136
    },
    {
      "epoch": 3.224372863769531e-06,
      "model_forward_time": 1.771979808807373,
      "step": 77381
    },
    {
      "epoch": 3.224372863769531e-06,
      "step": 77381,
      "training_step_time": 7.755882024765015
    },
    {
      "epoch": 3.2253265380859376e-06,
      "model_forward_time": 1.7715461254119873,
      "step": 77382
    },
    {
      "epoch": 3.2253265380859376e-06,
      "step": 77382,
      "training_step_time": 9.443831443786621
    },
    {
      "epoch": 3.2262802124023438e-06,
      "model_forward_time": 1.768613576889038,
      "step": 77383
    },
    {
      "epoch": 3.2262802124023438e-06,
      "step": 77383,
      "training_step_time": 9.939551830291748
    },
    {
      "epoch": 3.22723388671875e-06,
      "model_forward_time": 2.410200834274292,
      "step": 77384
    },
    {
      "epoch": 3.22723388671875e-06,
      "step": 77384,
      "training_step_time": 8.719924926757812
    },
    {
      "epoch": 3.228187561035156e-06,
      "model_forward_time": 1.774963617324829,
      "step": 77385
    },
    {
      "epoch": 3.228187561035156e-06,
      "step": 77385,
      "training_step_time": 7.816823244094849
    },
    {
      "epoch": 3.2291412353515626e-06,
      "model_forward_time": 1.790921926498413,
      "step": 77386
    },
    {
      "epoch": 3.2291412353515626e-06,
      "step": 77386,
      "training_step_time": 8.916902780532837
    },
    {
      "epoch": 3.2300949096679688e-06,
      "model_forward_time": 1.7749216556549072,
      "step": 77387
    },
    {
      "epoch": 3.2300949096679688e-06,
      "step": 77387,
      "training_step_time": 7.591044664382935
    },
    {
      "epoch": 3.231048583984375e-06,
      "model_forward_time": 1.7708244323730469,
      "step": 77388
    },
    {
      "epoch": 3.231048583984375e-06,
      "step": 77388,
      "training_step_time": 7.850686550140381
    },
    {
      "epoch": 3.232002258300781e-06,
      "model_forward_time": 1.774700403213501,
      "step": 77389
    },
    {
      "epoch": 3.232002258300781e-06,
      "step": 77389,
      "training_step_time": 7.686963319778442
    },
    {
      "epoch": 3.2329559326171876e-06,
      "grad_norm": 0.10346733778715134,
      "learning_rate": 1.3338356991307866e-06,
      "loss": 0.0556,
      "step": 77390
    },
    {
      "epoch": 3.2329559326171876e-06,
      "model_forward_time": 1.7673108577728271,
      "step": 77390
    },
    {
      "dynamics_loss_avg": 0.06836574673652648,
      "epoch": 3.2329559326171876e-06,
      "step": 77390
    },
    {
      "action_loss_avg": 0.0012352389836451038,
      "epoch": 3.2329559326171876e-06,
      "step": 77390
    },
    {
      "epoch": 3.2329559326171876e-06,
      "step": 77390,
      "training_step_time": 7.760785341262817
    },
    {
      "epoch": 3.2339096069335938e-06,
      "model_forward_time": 1.7744827270507812,
      "step": 77391
    },
    {
      "epoch": 3.2339096069335938e-06,
      "step": 77391,
      "training_step_time": 8.49778413772583
    },
    {
      "epoch": 3.23486328125e-06,
      "model_forward_time": 1.7709100246429443,
      "step": 77392
    },
    {
      "epoch": 3.23486328125e-06,
      "step": 77392,
      "training_step_time": 9.22898530960083
    },
    {
      "epoch": 3.235816955566406e-06,
      "model_forward_time": 2.4015142917633057,
      "step": 77393
    },
    {
      "epoch": 3.235816955566406e-06,
      "step": 77393,
      "training_step_time": 8.325522184371948
    },
    {
      "epoch": 3.2367706298828126e-06,
      "model_forward_time": 1.8070080280303955,
      "step": 77394
    },
    {
      "epoch": 3.2367706298828126e-06,
      "step": 77394,
      "training_step_time": 9.237109422683716
    },
    {
      "epoch": 3.2377243041992188e-06,
      "model_forward_time": 1.7853684425354004,
      "step": 77395
    },
    {
      "epoch": 3.2377243041992188e-06,
      "step": 77395,
      "training_step_time": 7.728025913238525
    },
    {
      "epoch": 3.238677978515625e-06,
      "model_forward_time": 1.7666010856628418,
      "step": 77396
    },
    {
      "epoch": 3.238677978515625e-06,
      "step": 77396,
      "training_step_time": 7.672517538070679
    },
    {
      "epoch": 3.239631652832031e-06,
      "model_forward_time": 1.773789882659912,
      "step": 77397
    },
    {
      "epoch": 3.239631652832031e-06,
      "step": 77397,
      "training_step_time": 8.953865766525269
    },
    {
      "epoch": 3.2405853271484376e-06,
      "model_forward_time": 1.7747313976287842,
      "step": 77398
    },
    {
      "epoch": 3.2405853271484376e-06,
      "step": 77398,
      "training_step_time": 7.747675657272339
    },
    {
      "epoch": 3.2415390014648438e-06,
      "model_forward_time": 1.3782570362091064,
      "step": 77399
    },
    {
      "epoch": 3.2415390014648438e-06,
      "step": 77399,
      "training_step_time": 7.662141799926758
    },
    {
      "epoch": 3.24249267578125e-06,
      "grad_norm": 0.10459090769290924,
      "learning_rate": 1.332711577132005e-06,
      "loss": 0.0572,
      "step": 77400
    },
    {
      "epoch": 3.24249267578125e-06,
      "model_forward_time": 1.7683186531066895,
      "step": 77400
    },
    {
      "dynamics_loss_avg": 0.04974572770297527,
      "epoch": 3.24249267578125e-06,
      "step": 77400
    },
    {
      "action_loss_avg": 0.0009742677619215101,
      "epoch": 3.24249267578125e-06,
      "step": 77400
    },
    {
      "epoch": 3.24249267578125e-06,
      "step": 77400,
      "training_step_time": 11.68943476676941
    },
    {
      "epoch": 3.243446350097656e-06,
      "model_forward_time": 1.3903570175170898,
      "step": 77401
    },
    {
      "epoch": 3.243446350097656e-06,
      "step": 77401,
      "training_step_time": 7.989727735519409
    },
    {
      "epoch": 3.2444000244140626e-06,
      "model_forward_time": 1.7787182331085205,
      "step": 77402
    },
    {
      "epoch": 3.2444000244140626e-06,
      "step": 77402,
      "training_step_time": 7.653731822967529
    },
    {
      "epoch": 3.2453536987304688e-06,
      "model_forward_time": 1.0469958782196045,
      "step": 77403
    },
    {
      "epoch": 3.2453536987304688e-06,
      "step": 77403,
      "training_step_time": 7.689043760299683
    },
    {
      "epoch": 3.246307373046875e-06,
      "model_forward_time": 1.7705538272857666,
      "step": 77404
    },
    {
      "epoch": 3.246307373046875e-06,
      "step": 77404,
      "training_step_time": 7.713201522827148
    },
    {
      "epoch": 3.247261047363281e-06,
      "model_forward_time": 1.7737696170806885,
      "step": 77405
    },
    {
      "epoch": 3.247261047363281e-06,
      "step": 77405,
      "training_step_time": 6.828151226043701
    },
    {
      "epoch": 3.2482147216796876e-06,
      "model_forward_time": 1.773221731185913,
      "step": 77406
    },
    {
      "epoch": 3.2482147216796876e-06,
      "step": 77406,
      "training_step_time": 6.828500270843506
    },
    {
      "epoch": 3.2491683959960938e-06,
      "model_forward_time": 1.7362408638000488,
      "step": 77407
    },
    {
      "epoch": 3.2491683959960938e-06,
      "step": 77407,
      "training_step_time": 6.824891567230225
    },
    {
      "epoch": 3.2501220703125e-06,
      "model_forward_time": 1.7302508354187012,
      "step": 77408
    },
    {
      "epoch": 3.2501220703125e-06,
      "step": 77408,
      "training_step_time": 6.802490472793579
    },
    {
      "epoch": 3.251075744628906e-06,
      "model_forward_time": 1.7321772575378418,
      "step": 77409
    },
    {
      "epoch": 3.251075744628906e-06,
      "step": 77409,
      "training_step_time": 6.814992904663086
    },
    {
      "epoch": 3.2520294189453126e-06,
      "grad_norm": 0.10042781382799149,
      "learning_rate": 1.3315878561823697e-06,
      "loss": 0.0573,
      "step": 77410
    },
    {
      "epoch": 3.2520294189453126e-06,
      "model_forward_time": 2.3636162281036377,
      "step": 77410
    },
    {
      "dynamics_loss_avg": 0.060144266486167906,
      "epoch": 3.2520294189453126e-06,
      "step": 77410
    },
    {
      "action_loss_avg": 0.0013135014334693551,
      "epoch": 3.2520294189453126e-06,
      "step": 77410
    },
    {
      "epoch": 3.2520294189453126e-06,
      "step": 77410,
      "training_step_time": 7.783948659896851
    },
    {
      "epoch": 3.2529830932617188e-06,
      "model_forward_time": 1.729285478591919,
      "step": 77411
    },
    {
      "epoch": 3.2529830932617188e-06,
      "step": 77411,
      "training_step_time": 7.342317342758179
    },
    {
      "epoch": 3.253936767578125e-06,
      "model_forward_time": 1.7291197776794434,
      "step": 77412
    },
    {
      "epoch": 3.253936767578125e-06,
      "step": 77412,
      "training_step_time": 6.775119781494141
    },
    {
      "epoch": 3.254890441894531e-06,
      "model_forward_time": 1.7273564338684082,
      "step": 77413
    },
    {
      "epoch": 3.254890441894531e-06,
      "step": 77413,
      "training_step_time": 8.52074646949768
    },
    {
      "epoch": 3.2558441162109376e-06,
      "model_forward_time": 1.7301757335662842,
      "step": 77414
    },
    {
      "epoch": 3.2558441162109376e-06,
      "step": 77414,
      "training_step_time": 6.801696538925171
    },
    {
      "epoch": 3.2567977905273438e-06,
      "model_forward_time": 1.7277004718780518,
      "step": 77415
    },
    {
      "epoch": 3.2567977905273438e-06,
      "step": 77415,
      "training_step_time": 6.774046421051025
    },
    {
      "epoch": 3.25775146484375e-06,
      "model_forward_time": 1.7355060577392578,
      "step": 77416
    },
    {
      "epoch": 3.25775146484375e-06,
      "step": 77416,
      "training_step_time": 6.7973175048828125
    },
    {
      "epoch": 3.258705139160156e-06,
      "model_forward_time": 1.7303757667541504,
      "step": 77417
    },
    {
      "epoch": 3.258705139160156e-06,
      "step": 77417,
      "training_step_time": 6.819936275482178
    },
    {
      "epoch": 3.2596588134765626e-06,
      "model_forward_time": 1.7244353294372559,
      "step": 77418
    },
    {
      "epoch": 3.2596588134765626e-06,
      "step": 77418,
      "training_step_time": 6.81492018699646
    },
    {
      "epoch": 3.2606124877929688e-06,
      "model_forward_time": 2.357057809829712,
      "step": 77419
    },
    {
      "epoch": 3.2606124877929688e-06,
      "step": 77419,
      "training_step_time": 8.662171840667725
    },
    {
      "epoch": 3.261566162109375e-06,
      "grad_norm": 0.07599309086799622,
      "learning_rate": 1.330464536404773e-06,
      "loss": 0.0565,
      "step": 77420
    },
    {
      "epoch": 3.261566162109375e-06,
      "model_forward_time": 1.7244253158569336,
      "step": 77420
    },
    {
      "dynamics_loss_avg": 0.05865898095071316,
      "epoch": 3.261566162109375e-06,
      "step": 77420
    },
    {
      "action_loss_avg": 0.001811859739245847,
      "epoch": 3.261566162109375e-06,
      "step": 77420
    },
    {
      "epoch": 3.261566162109375e-06,
      "step": 77420,
      "training_step_time": 7.448540449142456
    },
    {
      "epoch": 3.262519836425781e-06,
      "model_forward_time": 1.7338616847991943,
      "step": 77421
    },
    {
      "epoch": 3.262519836425781e-06,
      "step": 77421,
      "training_step_time": 6.811112403869629
    },
    {
      "epoch": 3.2634735107421876e-06,
      "model_forward_time": 1.727027177810669,
      "step": 77422
    },
    {
      "epoch": 3.2634735107421876e-06,
      "step": 77422,
      "training_step_time": 8.412514209747314
    },
    {
      "epoch": 3.264427185058594e-06,
      "model_forward_time": 1.7250690460205078,
      "step": 77423
    },
    {
      "epoch": 3.264427185058594e-06,
      "step": 77423,
      "training_step_time": 7.535297155380249
    },
    {
      "epoch": 3.265380859375e-06,
      "model_forward_time": 1.7309186458587646,
      "step": 77424
    },
    {
      "epoch": 3.265380859375e-06,
      "step": 77424,
      "training_step_time": 6.860947847366333
    },
    {
      "epoch": 3.266334533691406e-06,
      "model_forward_time": 1.7259366512298584,
      "step": 77425
    },
    {
      "epoch": 3.266334533691406e-06,
      "step": 77425,
      "training_step_time": 6.845324993133545
    },
    {
      "epoch": 3.2672882080078126e-06,
      "model_forward_time": 1.7217350006103516,
      "step": 77426
    },
    {
      "epoch": 3.2672882080078126e-06,
      "step": 77426,
      "training_step_time": 6.9008002281188965
    },
    {
      "epoch": 3.268241882324219e-06,
      "model_forward_time": 1.7348182201385498,
      "step": 77427
    },
    {
      "epoch": 3.268241882324219e-06,
      "step": 77427,
      "training_step_time": 6.849228143692017
    },
    {
      "epoch": 3.269195556640625e-06,
      "model_forward_time": 1.747887134552002,
      "step": 77428
    },
    {
      "epoch": 3.269195556640625e-06,
      "step": 77428,
      "training_step_time": 9.861530303955078
    },
    {
      "epoch": 3.270149230957031e-06,
      "model_forward_time": 1.7323405742645264,
      "step": 77429
    },
    {
      "epoch": 3.270149230957031e-06,
      "step": 77429,
      "training_step_time": 7.572999000549316
    },
    {
      "epoch": 3.2711029052734377e-06,
      "grad_norm": 0.07498399168252945,
      "learning_rate": 1.3293416179220564e-06,
      "loss": 0.0559,
      "step": 77430
    },
    {
      "epoch": 3.2711029052734377e-06,
      "model_forward_time": 1.723738431930542,
      "step": 77430
    },
    {
      "dynamics_loss_avg": 0.04258159901946783,
      "epoch": 3.2711029052734377e-06,
      "step": 77430
    },
    {
      "action_loss_avg": 0.0014167928777169437,
      "epoch": 3.2711029052734377e-06,
      "step": 77430
    },
    {
      "epoch": 3.2711029052734377e-06,
      "step": 77430,
      "training_step_time": 6.825246810913086
    },
    {
      "epoch": 3.272056579589844e-06,
      "model_forward_time": 1.724921464920044,
      "step": 77431
    },
    {
      "epoch": 3.272056579589844e-06,
      "step": 77431,
      "training_step_time": 7.537922143936157
    },
    {
      "epoch": 3.27301025390625e-06,
      "model_forward_time": 1.731372356414795,
      "step": 77432
    },
    {
      "epoch": 3.27301025390625e-06,
      "step": 77432,
      "training_step_time": 7.808306932449341
    },
    {
      "epoch": 3.273963928222656e-06,
      "model_forward_time": 1.7281780242919922,
      "step": 77433
    },
    {
      "epoch": 3.273963928222656e-06,
      "step": 77433,
      "training_step_time": 6.905484437942505
    },
    {
      "epoch": 3.2749176025390627e-06,
      "model_forward_time": 1.7250871658325195,
      "step": 77434
    },
    {
      "epoch": 3.2749176025390627e-06,
      "step": 77434,
      "training_step_time": 6.860600471496582
    },
    {
      "epoch": 3.275871276855469e-06,
      "model_forward_time": 1.7227675914764404,
      "step": 77435
    },
    {
      "epoch": 3.275871276855469e-06,
      "step": 77435,
      "training_step_time": 6.855721473693848
    },
    {
      "epoch": 3.276824951171875e-06,
      "model_forward_time": 1.72214674949646,
      "step": 77436
    },
    {
      "epoch": 3.276824951171875e-06,
      "step": 77436,
      "training_step_time": 6.844669580459595
    },
    {
      "epoch": 3.277778625488281e-06,
      "model_forward_time": 1.753349781036377,
      "step": 77437
    },
    {
      "epoch": 3.277778625488281e-06,
      "step": 77437,
      "training_step_time": 10.713716745376587
    },
    {
      "epoch": 3.2787322998046877e-06,
      "model_forward_time": 2.4344468116760254,
      "step": 77438
    },
    {
      "epoch": 3.2787322998046877e-06,
      "step": 77438,
      "training_step_time": 7.829110860824585
    },
    {
      "epoch": 3.279685974121094e-06,
      "model_forward_time": 1.7342414855957031,
      "step": 77439
    },
    {
      "epoch": 3.279685974121094e-06,
      "step": 77439,
      "training_step_time": 6.8271472454071045
    },
    {
      "epoch": 3.2806396484375e-06,
      "grad_norm": 0.07299261540174484,
      "learning_rate": 1.328219100857021e-06,
      "loss": 0.0564,
      "step": 77440
    },
    {
      "epoch": 3.2806396484375e-06,
      "model_forward_time": 1.7271864414215088,
      "step": 77440
    },
    {
      "dynamics_loss_avg": 0.07041938677430153,
      "epoch": 3.2806396484375e-06,
      "step": 77440
    },
    {
      "action_loss_avg": 0.002235600628773682,
      "epoch": 3.2806396484375e-06,
      "step": 77440
    },
    {
      "epoch": 3.2806396484375e-06,
      "step": 77440,
      "training_step_time": 7.488826274871826
    },
    {
      "epoch": 3.281593322753906e-06,
      "model_forward_time": 1.730665683746338,
      "step": 77441
    },
    {
      "epoch": 3.281593322753906e-06,
      "step": 77441,
      "training_step_time": 7.814234018325806
    },
    {
      "epoch": 3.2825469970703127e-06,
      "model_forward_time": 1.7268433570861816,
      "step": 77442
    },
    {
      "epoch": 3.2825469970703127e-06,
      "step": 77442,
      "training_step_time": 6.884431600570679
    },
    {
      "epoch": 3.283500671386719e-06,
      "model_forward_time": 1.7245523929595947,
      "step": 77443
    },
    {
      "epoch": 3.283500671386719e-06,
      "step": 77443,
      "training_step_time": 6.834301948547363
    },
    {
      "epoch": 3.284454345703125e-06,
      "model_forward_time": 1.7322726249694824,
      "step": 77444
    },
    {
      "epoch": 3.284454345703125e-06,
      "step": 77444,
      "training_step_time": 6.84235954284668
    },
    {
      "epoch": 3.285408020019531e-06,
      "model_forward_time": 1.7220025062561035,
      "step": 77445
    },
    {
      "epoch": 3.285408020019531e-06,
      "step": 77445,
      "training_step_time": 6.751513957977295
    },
    {
      "epoch": 3.2863616943359377e-06,
      "model_forward_time": 1.737541675567627,
      "step": 77446
    },
    {
      "epoch": 3.2863616943359377e-06,
      "step": 77446,
      "training_step_time": 9.217223405838013
    },
    {
      "epoch": 3.287315368652344e-06,
      "model_forward_time": 2.0730950832366943,
      "step": 77447
    },
    {
      "epoch": 3.287315368652344e-06,
      "step": 77447,
      "training_step_time": 7.817306995391846
    },
    {
      "epoch": 3.28826904296875e-06,
      "model_forward_time": 1.7306525707244873,
      "step": 77448
    },
    {
      "epoch": 3.28826904296875e-06,
      "step": 77448,
      "training_step_time": 7.403869390487671
    },
    {
      "epoch": 3.289222717285156e-06,
      "model_forward_time": 1.7310783863067627,
      "step": 77449
    },
    {
      "epoch": 3.289222717285156e-06,
      "step": 77449,
      "training_step_time": 7.457658290863037
    },
    {
      "epoch": 3.2901763916015627e-06,
      "grad_norm": 0.1602763533592224,
      "learning_rate": 1.327096985332425e-06,
      "loss": 0.0565,
      "step": 77450
    },
    {
      "epoch": 3.2901763916015627e-06,
      "model_forward_time": 1.7235383987426758,
      "step": 77450
    },
    {
      "dynamics_loss_avg": 0.054005471616983415,
      "epoch": 3.2901763916015627e-06,
      "step": 77450
    },
    {
      "action_loss_avg": 0.0015038557990919798,
      "epoch": 3.2901763916015627e-06,
      "step": 77450
    },
    {
      "epoch": 3.2901763916015627e-06,
      "step": 77450,
      "training_step_time": 8.388643741607666
    },
    {
      "epoch": 3.291130065917969e-06,
      "model_forward_time": 1.7321662902832031,
      "step": 77451
    },
    {
      "epoch": 3.291130065917969e-06,
      "step": 77451,
      "training_step_time": 6.723418474197388
    },
    {
      "epoch": 3.292083740234375e-06,
      "model_forward_time": 1.73089599609375,
      "step": 77452
    },
    {
      "epoch": 3.292083740234375e-06,
      "step": 77452,
      "training_step_time": 6.757599830627441
    },
    {
      "epoch": 3.293037414550781e-06,
      "model_forward_time": 1.7188961505889893,
      "step": 77453
    },
    {
      "epoch": 3.293037414550781e-06,
      "step": 77453,
      "training_step_time": 6.795858383178711
    },
    {
      "epoch": 3.2939910888671877e-06,
      "model_forward_time": 1.7283580303192139,
      "step": 77454
    },
    {
      "epoch": 3.2939910888671877e-06,
      "step": 77454,
      "training_step_time": 6.75243067741394
    },
    {
      "epoch": 3.294944763183594e-06,
      "model_forward_time": 1.7286384105682373,
      "step": 77455
    },
    {
      "epoch": 3.294944763183594e-06,
      "step": 77455,
      "training_step_time": 8.627480745315552
    },
    {
      "epoch": 3.2958984375e-06,
      "model_forward_time": 2.3805630207061768,
      "step": 77456
    },
    {
      "epoch": 3.2958984375e-06,
      "step": 77456,
      "training_step_time": 8.705371379852295
    },
    {
      "epoch": 3.296852111816406e-06,
      "model_forward_time": 1.7274408340454102,
      "step": 77457
    },
    {
      "epoch": 3.296852111816406e-06,
      "step": 77457,
      "training_step_time": 8.064452648162842
    },
    {
      "epoch": 3.2978057861328127e-06,
      "model_forward_time": 1.7262120246887207,
      "step": 77458
    },
    {
      "epoch": 3.2978057861328127e-06,
      "step": 77458,
      "training_step_time": 6.762368440628052
    },
    {
      "epoch": 3.298759460449219e-06,
      "model_forward_time": 1.723703145980835,
      "step": 77459
    },
    {
      "epoch": 3.298759460449219e-06,
      "step": 77459,
      "training_step_time": 9.74003005027771
    },
    {
      "epoch": 3.299713134765625e-06,
      "grad_norm": 0.1239999532699585,
      "learning_rate": 1.3259752714709817e-06,
      "loss": 0.0558,
      "step": 77460
    },
    {
      "epoch": 3.299713134765625e-06,
      "model_forward_time": 1.7355754375457764,
      "step": 77460
    },
    {
      "dynamics_loss_avg": 0.06411026380956172,
      "epoch": 3.299713134765625e-06,
      "step": 77460
    },
    {
      "action_loss_avg": 0.0012537056463770568,
      "epoch": 3.299713134765625e-06,
      "step": 77460
    },
    {
      "epoch": 3.299713134765625e-06,
      "step": 77460,
      "training_step_time": 7.487698554992676
    },
    {
      "epoch": 3.300666809082031e-06,
      "model_forward_time": 1.7424468994140625,
      "step": 77461
    },
    {
      "epoch": 3.300666809082031e-06,
      "step": 77461,
      "training_step_time": 6.734551668167114
    },
    {
      "epoch": 3.3016204833984377e-06,
      "model_forward_time": 1.7372426986694336,
      "step": 77462
    },
    {
      "epoch": 3.3016204833984377e-06,
      "step": 77462,
      "training_step_time": 6.80189323425293
    },
    {
      "epoch": 3.302574157714844e-06,
      "model_forward_time": 1.7406823635101318,
      "step": 77463
    },
    {
      "epoch": 3.302574157714844e-06,
      "step": 77463,
      "training_step_time": 6.697315692901611
    },
    {
      "epoch": 3.30352783203125e-06,
      "model_forward_time": 1.741260290145874,
      "step": 77464
    },
    {
      "epoch": 3.30352783203125e-06,
      "step": 77464,
      "training_step_time": 7.460387229919434
    },
    {
      "epoch": 3.304481506347656e-06,
      "model_forward_time": 2.389564037322998,
      "step": 77465
    },
    {
      "epoch": 3.304481506347656e-06,
      "step": 77465,
      "training_step_time": 9.889210224151611
    },
    {
      "epoch": 3.3054351806640627e-06,
      "model_forward_time": 1.7208340167999268,
      "step": 77466
    },
    {
      "epoch": 3.3054351806640627e-06,
      "step": 77466,
      "training_step_time": 7.718976974487305
    },
    {
      "epoch": 3.306388854980469e-06,
      "model_forward_time": 1.7281970977783203,
      "step": 77467
    },
    {
      "epoch": 3.306388854980469e-06,
      "step": 77467,
      "training_step_time": 7.299484729766846
    },
    {
      "epoch": 3.307342529296875e-06,
      "model_forward_time": 1.7300441265106201,
      "step": 77468
    },
    {
      "epoch": 3.307342529296875e-06,
      "step": 77468,
      "training_step_time": 8.869094133377075
    },
    {
      "epoch": 3.308296203613281e-06,
      "model_forward_time": 1.7266924381256104,
      "step": 77469
    },
    {
      "epoch": 3.308296203613281e-06,
      "step": 77469,
      "training_step_time": 7.786611795425415
    },
    {
      "epoch": 3.3092498779296877e-06,
      "grad_norm": 0.08940491825342178,
      "learning_rate": 1.3248539593953568e-06,
      "loss": 0.0559,
      "step": 77470
    },
    {
      "epoch": 3.3092498779296877e-06,
      "model_forward_time": 1.7361934185028076,
      "step": 77470
    },
    {
      "dynamics_loss_avg": 0.05236510057002306,
      "epoch": 3.3092498779296877e-06,
      "step": 77470
    },
    {
      "action_loss_avg": 0.0009955466259270907,
      "epoch": 3.3092498779296877e-06,
      "step": 77470
    },
    {
      "epoch": 3.3092498779296877e-06,
      "step": 77470,
      "training_step_time": 6.839542627334595
    },
    {
      "epoch": 3.310203552246094e-06,
      "model_forward_time": 1.7354357242584229,
      "step": 77471
    },
    {
      "epoch": 3.310203552246094e-06,
      "step": 77471,
      "training_step_time": 6.88566255569458
    },
    {
      "epoch": 3.3111572265625e-06,
      "model_forward_time": 1.7353570461273193,
      "step": 77472
    },
    {
      "epoch": 3.3111572265625e-06,
      "step": 77472,
      "training_step_time": 6.772163391113281
    },
    {
      "epoch": 3.312110900878906e-06,
      "model_forward_time": 1.730520248413086,
      "step": 77473
    },
    {
      "epoch": 3.312110900878906e-06,
      "step": 77473,
      "training_step_time": 7.296136140823364
    },
    {
      "epoch": 3.3130645751953127e-06,
      "model_forward_time": 1.739501714706421,
      "step": 77474
    },
    {
      "epoch": 3.3130645751953127e-06,
      "step": 77474,
      "training_step_time": 9.342083930969238
    },
    {
      "epoch": 3.314018249511719e-06,
      "model_forward_time": 1.7443828582763672,
      "step": 77475
    },
    {
      "epoch": 3.314018249511719e-06,
      "step": 77475,
      "training_step_time": 7.621241331100464
    },
    {
      "epoch": 3.314971923828125e-06,
      "model_forward_time": 1.7355399131774902,
      "step": 77476
    },
    {
      "epoch": 3.314971923828125e-06,
      "step": 77476,
      "training_step_time": 7.302242040634155
    },
    {
      "epoch": 3.315925598144531e-06,
      "model_forward_time": 1.7217626571655273,
      "step": 77477
    },
    {
      "epoch": 3.315925598144531e-06,
      "step": 77477,
      "training_step_time": 7.391162395477295
    },
    {
      "epoch": 3.3168792724609377e-06,
      "model_forward_time": 1.735119104385376,
      "step": 77478
    },
    {
      "epoch": 3.3168792724609377e-06,
      "step": 77478,
      "training_step_time": 7.705645799636841
    },
    {
      "epoch": 3.317832946777344e-06,
      "model_forward_time": 1.7297947406768799,
      "step": 77479
    },
    {
      "epoch": 3.317832946777344e-06,
      "step": 77479,
      "training_step_time": 6.757880449295044
    },
    {
      "epoch": 3.31878662109375e-06,
      "grad_norm": 0.074160136282444,
      "learning_rate": 1.323733049228178e-06,
      "loss": 0.0542,
      "step": 77480
    },
    {
      "epoch": 3.31878662109375e-06,
      "model_forward_time": 1.725172519683838,
      "step": 77480
    },
    {
      "dynamics_loss_avg": 0.050109193474054334,
      "epoch": 3.31878662109375e-06,
      "step": 77480
    },
    {
      "action_loss_avg": 0.0011497957166284323,
      "epoch": 3.31878662109375e-06,
      "step": 77480
    },
    {
      "epoch": 3.31878662109375e-06,
      "step": 77480,
      "training_step_time": 6.72276496887207
    },
    {
      "epoch": 3.319740295410156e-06,
      "model_forward_time": 1.7313096523284912,
      "step": 77481
    },
    {
      "epoch": 3.319740295410156e-06,
      "step": 77481,
      "training_step_time": 6.7533347606658936
    },
    {
      "epoch": 3.3206939697265627e-06,
      "model_forward_time": 1.7292218208312988,
      "step": 77482
    },
    {
      "epoch": 3.3206939697265627e-06,
      "step": 77482,
      "training_step_time": 7.291578769683838
    },
    {
      "epoch": 3.321647644042969e-06,
      "model_forward_time": 1.7559595108032227,
      "step": 77483
    },
    {
      "epoch": 3.321647644042969e-06,
      "step": 77483,
      "training_step_time": 10.134503841400146
    },
    {
      "epoch": 3.322601318359375e-06,
      "model_forward_time": 2.449667453765869,
      "step": 77484
    },
    {
      "epoch": 3.322601318359375e-06,
      "step": 77484,
      "training_step_time": 8.395515203475952
    },
    {
      "epoch": 3.323554992675781e-06,
      "model_forward_time": 1.7277791500091553,
      "step": 77485
    },
    {
      "epoch": 3.323554992675781e-06,
      "step": 77485,
      "training_step_time": 7.406843423843384
    },
    {
      "epoch": 3.3245086669921877e-06,
      "model_forward_time": 1.7317101955413818,
      "step": 77486
    },
    {
      "epoch": 3.3245086669921877e-06,
      "step": 77486,
      "training_step_time": 8.407660484313965
    },
    {
      "epoch": 3.325462341308594e-06,
      "model_forward_time": 1.7307374477386475,
      "step": 77487
    },
    {
      "epoch": 3.325462341308594e-06,
      "step": 77487,
      "training_step_time": 8.370181560516357
    },
    {
      "epoch": 3.326416015625e-06,
      "model_forward_time": 1.7345190048217773,
      "step": 77488
    },
    {
      "epoch": 3.326416015625e-06,
      "step": 77488,
      "training_step_time": 6.735108852386475
    },
    {
      "epoch": 3.327369689941406e-06,
      "model_forward_time": 1.7265615463256836,
      "step": 77489
    },
    {
      "epoch": 3.327369689941406e-06,
      "step": 77489,
      "training_step_time": 6.718430995941162
    },
    {
      "epoch": 3.3283233642578127e-06,
      "grad_norm": 0.12266243249177933,
      "learning_rate": 1.3226125410920277e-06,
      "loss": 0.056,
      "step": 77490
    },
    {
      "epoch": 3.3283233642578127e-06,
      "model_forward_time": 1.727876901626587,
      "step": 77490
    },
    {
      "dynamics_loss_avg": 0.07062105759978295,
      "epoch": 3.3283233642578127e-06,
      "step": 77490
    },
    {
      "action_loss_avg": 0.002263529156334698,
      "epoch": 3.3283233642578127e-06,
      "step": 77490
    },
    {
      "epoch": 3.3283233642578127e-06,
      "step": 77490,
      "training_step_time": 6.655325412750244
    },
    {
      "epoch": 3.329277038574219e-06,
      "model_forward_time": 1.7296829223632812,
      "step": 77491
    },
    {
      "epoch": 3.329277038574219e-06,
      "step": 77491,
      "training_step_time": 6.6530067920684814
    },
    {
      "epoch": 3.330230712890625e-06,
      "model_forward_time": 1.7234086990356445,
      "step": 77492
    },
    {
      "epoch": 3.330230712890625e-06,
      "step": 77492,
      "training_step_time": 9.177409172058105
    },
    {
      "epoch": 3.331184387207031e-06,
      "model_forward_time": 2.372333526611328,
      "step": 77493
    },
    {
      "epoch": 3.331184387207031e-06,
      "step": 77493,
      "training_step_time": 8.476221561431885
    },
    {
      "epoch": 3.3321380615234377e-06,
      "model_forward_time": 1.729736328125,
      "step": 77494
    },
    {
      "epoch": 3.3321380615234377e-06,
      "step": 77494,
      "training_step_time": 7.517257928848267
    },
    {
      "epoch": 3.333091735839844e-06,
      "model_forward_time": 1.7219159603118896,
      "step": 77495
    },
    {
      "epoch": 3.333091735839844e-06,
      "step": 77495,
      "training_step_time": 7.380932569503784
    },
    {
      "epoch": 3.33404541015625e-06,
      "model_forward_time": 1.7258689403533936,
      "step": 77496
    },
    {
      "epoch": 3.33404541015625e-06,
      "step": 77496,
      "training_step_time": 7.732657194137573
    },
    {
      "epoch": 3.334999084472656e-06,
      "model_forward_time": 1.7382009029388428,
      "step": 77497
    },
    {
      "epoch": 3.334999084472656e-06,
      "step": 77497,
      "training_step_time": 7.565540075302124
    },
    {
      "epoch": 3.3359527587890627e-06,
      "model_forward_time": 1.7319188117980957,
      "step": 77498
    },
    {
      "epoch": 3.3359527587890627e-06,
      "step": 77498,
      "training_step_time": 6.7961037158966064
    },
    {
      "epoch": 3.336906433105469e-06,
      "model_forward_time": 1.727271556854248,
      "step": 77499
    },
    {
      "epoch": 3.336906433105469e-06,
      "step": 77499,
      "training_step_time": 6.809756517410278
    },
    {
      "epoch": 3.337860107421875e-06,
      "grad_norm": 0.08545807749032974,
      "learning_rate": 1.3214924351094387e-06,
      "loss": 0.0556,
      "step": 77500
    },
    {
      "epoch": 3.337860107421875e-06,
      "model_forward_time": 1.7270944118499756,
      "step": 77500
    },
    {
      "dynamics_loss_avg": 0.05611191764473915,
      "epoch": 3.337860107421875e-06,
      "step": 77500
    },
    {
      "action_loss_avg": 0.0010926574643235654,
      "epoch": 3.337860107421875e-06,
      "step": 77500
    },
    {
      "epoch": 3.337860107421875e-06,
      "step": 77500,
      "training_step_time": 6.816865921020508
    },
    {
      "epoch": 3.338813781738281e-06,
      "model_forward_time": 1.7294039726257324,
      "step": 77501
    },
    {
      "epoch": 3.338813781738281e-06,
      "step": 77501,
      "training_step_time": 9.105599403381348
    },
    {
      "epoch": 3.3397674560546873e-06,
      "model_forward_time": 2.3820693492889404,
      "step": 77502
    },
    {
      "epoch": 3.3397674560546873e-06,
      "step": 77502,
      "training_step_time": 9.323389291763306
    },
    {
      "epoch": 3.340721130371094e-06,
      "model_forward_time": 1.7270729541778564,
      "step": 77503
    },
    {
      "epoch": 3.340721130371094e-06,
      "step": 77503,
      "training_step_time": 8.441886901855469
    },
    {
      "epoch": 3.3416748046875e-06,
      "model_forward_time": 1.744593858718872,
      "step": 77504
    },
    {
      "epoch": 3.3416748046875e-06,
      "step": 77504,
      "training_step_time": 8.125068426132202
    },
    {
      "epoch": 3.342628479003906e-06,
      "model_forward_time": 1.750481128692627,
      "step": 77505
    },
    {
      "epoch": 3.342628479003906e-06,
      "step": 77505,
      "training_step_time": 7.844017267227173
    },
    {
      "epoch": 3.3435821533203123e-06,
      "model_forward_time": 1.7379696369171143,
      "step": 77506
    },
    {
      "epoch": 3.3435821533203123e-06,
      "step": 77506,
      "training_step_time": 7.649301052093506
    },
    {
      "epoch": 3.344535827636719e-06,
      "model_forward_time": 1.7386443614959717,
      "step": 77507
    },
    {
      "epoch": 3.344535827636719e-06,
      "step": 77507,
      "training_step_time": 6.789021968841553
    },
    {
      "epoch": 3.345489501953125e-06,
      "model_forward_time": 1.7247035503387451,
      "step": 77508
    },
    {
      "epoch": 3.345489501953125e-06,
      "step": 77508,
      "training_step_time": 6.793842792510986
    },
    {
      "epoch": 3.346443176269531e-06,
      "model_forward_time": 1.7371032238006592,
      "step": 77509
    },
    {
      "epoch": 3.346443176269531e-06,
      "step": 77509,
      "training_step_time": 6.7881433963775635
    },
    {
      "epoch": 3.3473968505859373e-06,
      "grad_norm": 0.08355623483657837,
      "learning_rate": 1.3203727314029069e-06,
      "loss": 0.0561,
      "step": 77510
    },
    {
      "epoch": 3.3473968505859373e-06,
      "model_forward_time": 1.7544996738433838,
      "step": 77510
    },
    {
      "dynamics_loss_avg": 0.04770140470936894,
      "epoch": 3.3473968505859373e-06,
      "step": 77510
    },
    {
      "action_loss_avg": 0.0013315997552126647,
      "epoch": 3.3473968505859373e-06,
      "step": 77510
    },
    {
      "epoch": 3.3473968505859373e-06,
      "step": 77510,
      "training_step_time": 8.713191986083984
    },
    {
      "epoch": 3.348350524902344e-06,
      "model_forward_time": 2.489680528640747,
      "step": 77511
    },
    {
      "epoch": 3.348350524902344e-06,
      "step": 77511,
      "training_step_time": 9.638677597045898
    },
    {
      "epoch": 3.34930419921875e-06,
      "model_forward_time": 1.7310309410095215,
      "step": 77512
    },
    {
      "epoch": 3.34930419921875e-06,
      "step": 77512,
      "training_step_time": 7.612975597381592
    },
    {
      "epoch": 3.350257873535156e-06,
      "model_forward_time": 1.7379233837127686,
      "step": 77513
    },
    {
      "epoch": 3.350257873535156e-06,
      "step": 77513,
      "training_step_time": 7.416313409805298
    },
    {
      "epoch": 3.3512115478515623e-06,
      "model_forward_time": 1.740081548690796,
      "step": 77514
    },
    {
      "epoch": 3.3512115478515623e-06,
      "step": 77514,
      "training_step_time": 7.8397376537323
    },
    {
      "epoch": 3.352165222167969e-06,
      "model_forward_time": 1.748584270477295,
      "step": 77515
    },
    {
      "epoch": 3.352165222167969e-06,
      "step": 77515,
      "training_step_time": 7.646681547164917
    },
    {
      "epoch": 3.353118896484375e-06,
      "model_forward_time": 1.7376363277435303,
      "step": 77516
    },
    {
      "epoch": 3.353118896484375e-06,
      "step": 77516,
      "training_step_time": 6.778863430023193
    },
    {
      "epoch": 3.354072570800781e-06,
      "model_forward_time": 1.712092399597168,
      "step": 77517
    },
    {
      "epoch": 3.354072570800781e-06,
      "step": 77517,
      "training_step_time": 6.803171634674072
    },
    {
      "epoch": 3.3550262451171873e-06,
      "model_forward_time": 1.740922212600708,
      "step": 77518
    },
    {
      "epoch": 3.3550262451171873e-06,
      "step": 77518,
      "training_step_time": 6.77972149848938
    },
    {
      "epoch": 3.355979919433594e-06,
      "model_forward_time": 1.7305047512054443,
      "step": 77519
    },
    {
      "epoch": 3.355979919433594e-06,
      "step": 77519,
      "training_step_time": 7.779218673706055
    },
    {
      "epoch": 3.35693359375e-06,
      "grad_norm": 0.11296842992305756,
      "learning_rate": 1.319253430094881e-06,
      "loss": 0.0552,
      "step": 77520
    },
    {
      "epoch": 3.35693359375e-06,
      "model_forward_time": 2.388324499130249,
      "step": 77520
    },
    {
      "dynamics_loss_avg": 0.04897285345941782,
      "epoch": 3.35693359375e-06,
      "step": 77520
    },
    {
      "action_loss_avg": 0.0019382811442483217,
      "epoch": 3.35693359375e-06,
      "step": 77520
    },
    {
      "epoch": 3.35693359375e-06,
      "step": 77520,
      "training_step_time": 11.496090650558472
    },
    {
      "epoch": 3.357887268066406e-06,
      "model_forward_time": 1.7441551685333252,
      "step": 77521
    },
    {
      "epoch": 3.357887268066406e-06,
      "step": 77521,
      "training_step_time": 8.563566207885742
    },
    {
      "epoch": 3.3588409423828123e-06,
      "model_forward_time": 1.7420012950897217,
      "step": 77522
    },
    {
      "epoch": 3.3588409423828123e-06,
      "step": 77522,
      "training_step_time": 7.485394477844238
    },
    {
      "epoch": 3.359794616699219e-06,
      "model_forward_time": 1.746610403060913,
      "step": 77523
    },
    {
      "epoch": 3.359794616699219e-06,
      "step": 77523,
      "training_step_time": 7.515656471252441
    },
    {
      "epoch": 3.360748291015625e-06,
      "model_forward_time": 1.7371103763580322,
      "step": 77524
    },
    {
      "epoch": 3.360748291015625e-06,
      "step": 77524,
      "training_step_time": 7.817547559738159
    },
    {
      "epoch": 3.361701965332031e-06,
      "model_forward_time": 1.7443029880523682,
      "step": 77525
    },
    {
      "epoch": 3.361701965332031e-06,
      "step": 77525,
      "training_step_time": 6.841853857040405
    },
    {
      "epoch": 3.3626556396484373e-06,
      "model_forward_time": 1.7474772930145264,
      "step": 77526
    },
    {
      "epoch": 3.3626556396484373e-06,
      "step": 77526,
      "training_step_time": 6.867486000061035
    },
    {
      "epoch": 3.363609313964844e-06,
      "model_forward_time": 1.7432615756988525,
      "step": 77527
    },
    {
      "epoch": 3.363609313964844e-06,
      "step": 77527,
      "training_step_time": 6.7581377029418945
    },
    {
      "epoch": 3.36456298828125e-06,
      "model_forward_time": 1.7268688678741455,
      "step": 77528
    },
    {
      "epoch": 3.36456298828125e-06,
      "step": 77528,
      "training_step_time": 7.763390302658081
    },
    {
      "epoch": 3.365516662597656e-06,
      "model_forward_time": 2.346731185913086,
      "step": 77529
    },
    {
      "epoch": 3.365516662597656e-06,
      "step": 77529,
      "training_step_time": 11.17910647392273
    },
    {
      "epoch": 3.3664703369140623e-06,
      "grad_norm": 0.07621411234140396,
      "learning_rate": 1.3181345313077677e-06,
      "loss": 0.0557,
      "step": 77530
    },
    {
      "epoch": 3.3664703369140623e-06,
      "model_forward_time": 1.7727596759796143,
      "step": 77530
    },
    {
      "dynamics_loss_avg": 0.05871540531516075,
      "epoch": 3.3664703369140623e-06,
      "step": 77530
    },
    {
      "action_loss_avg": 0.001885061332723126,
      "epoch": 3.3664703369140623e-06,
      "step": 77530
    },
    {
      "epoch": 3.3664703369140623e-06,
      "step": 77530,
      "training_step_time": 8.231456756591797
    },
    {
      "epoch": 3.367424011230469e-06,
      "model_forward_time": 1.7449653148651123,
      "step": 77531
    },
    {
      "epoch": 3.367424011230469e-06,
      "step": 77531,
      "training_step_time": 7.357789516448975
    },
    {
      "epoch": 3.368377685546875e-06,
      "model_forward_time": 1.7473044395446777,
      "step": 77532
    },
    {
      "epoch": 3.368377685546875e-06,
      "step": 77532,
      "training_step_time": 7.340883255004883
    },
    {
      "epoch": 3.369331359863281e-06,
      "model_forward_time": 1.7335772514343262,
      "step": 77533
    },
    {
      "epoch": 3.369331359863281e-06,
      "step": 77533,
      "training_step_time": 7.695401668548584
    },
    {
      "epoch": 3.3702850341796873e-06,
      "model_forward_time": 1.7429053783416748,
      "step": 77534
    },
    {
      "epoch": 3.3702850341796873e-06,
      "step": 77534,
      "training_step_time": 7.4607179164886475
    },
    {
      "epoch": 3.371238708496094e-06,
      "model_forward_time": 1.737370491027832,
      "step": 77535
    },
    {
      "epoch": 3.371238708496094e-06,
      "step": 77535,
      "training_step_time": 6.761983633041382
    },
    {
      "epoch": 3.3721923828125e-06,
      "model_forward_time": 1.738081455230713,
      "step": 77536
    },
    {
      "epoch": 3.3721923828125e-06,
      "step": 77536,
      "training_step_time": 6.778440475463867
    },
    {
      "epoch": 3.373146057128906e-06,
      "model_forward_time": 1.7311937808990479,
      "step": 77537
    },
    {
      "epoch": 3.373146057128906e-06,
      "step": 77537,
      "training_step_time": 7.7676732540130615
    },
    {
      "epoch": 3.3740997314453123e-06,
      "model_forward_time": 2.3774335384368896,
      "step": 77538
    },
    {
      "epoch": 3.3740997314453123e-06,
      "step": 77538,
      "training_step_time": 9.727958679199219
    },
    {
      "epoch": 3.375053405761719e-06,
      "model_forward_time": 1.8027324676513672,
      "step": 77539
    },
    {
      "epoch": 3.375053405761719e-06,
      "step": 77539,
      "training_step_time": 8.812869310379028
    },
    {
      "epoch": 3.376007080078125e-06,
      "grad_norm": 0.42678818106651306,
      "learning_rate": 1.3170160351639245e-06,
      "loss": 0.0558,
      "step": 77540
    },
    {
      "epoch": 3.376007080078125e-06,
      "model_forward_time": 1.7284140586853027,
      "step": 77540
    },
    {
      "dynamics_loss_avg": 0.050958033092319964,
      "epoch": 3.376007080078125e-06,
      "step": 77540
    },
    {
      "action_loss_avg": 0.0015513889957219362,
      "epoch": 3.376007080078125e-06,
      "step": 77540
    },
    {
      "epoch": 3.376007080078125e-06,
      "step": 77540,
      "training_step_time": 7.418292284011841
    },
    {
      "epoch": 3.376960754394531e-06,
      "model_forward_time": 1.737928867340088,
      "step": 77541
    },
    {
      "epoch": 3.376960754394531e-06,
      "step": 77541,
      "training_step_time": 9.221349954605103
    },
    {
      "epoch": 3.3779144287109373e-06,
      "model_forward_time": 1.7335190773010254,
      "step": 77542
    },
    {
      "epoch": 3.3779144287109373e-06,
      "step": 77542,
      "training_step_time": 7.748036623001099
    },
    {
      "epoch": 3.378868103027344e-06,
      "model_forward_time": 1.7335939407348633,
      "step": 77543
    },
    {
      "epoch": 3.378868103027344e-06,
      "step": 77543,
      "training_step_time": 7.521661043167114
    },
    {
      "epoch": 3.37982177734375e-06,
      "model_forward_time": 1.741481065750122,
      "step": 77544
    },
    {
      "epoch": 3.37982177734375e-06,
      "step": 77544,
      "training_step_time": 6.815470933914185
    },
    {
      "epoch": 3.380775451660156e-06,
      "model_forward_time": 1.737610101699829,
      "step": 77545
    },
    {
      "epoch": 3.380775451660156e-06,
      "step": 77545,
      "training_step_time": 6.754992246627808
    },
    {
      "epoch": 3.3817291259765623e-06,
      "model_forward_time": 1.7460317611694336,
      "step": 77546
    },
    {
      "epoch": 3.3817291259765623e-06,
      "step": 77546,
      "training_step_time": 7.814501762390137
    },
    {
      "epoch": 3.382682800292969e-06,
      "model_forward_time": 2.362264633178711,
      "step": 77547
    },
    {
      "epoch": 3.382682800292969e-06,
      "step": 77547,
      "training_step_time": 10.666174411773682
    },
    {
      "epoch": 3.383636474609375e-06,
      "model_forward_time": 1.7980144023895264,
      "step": 77548
    },
    {
      "epoch": 3.383636474609375e-06,
      "step": 77548,
      "training_step_time": 9.447483539581299
    },
    {
      "epoch": 3.384590148925781e-06,
      "model_forward_time": 1.7296326160430908,
      "step": 77549
    },
    {
      "epoch": 3.384590148925781e-06,
      "step": 77549,
      "training_step_time": 10.082777738571167
    },
    {
      "epoch": 3.3855438232421873e-06,
      "grad_norm": 0.08346530795097351,
      "learning_rate": 1.315897941785672e-06,
      "loss": 0.0561,
      "step": 77550
    },
    {
      "epoch": 3.3855438232421873e-06,
      "model_forward_time": 1.7265563011169434,
      "step": 77550
    },
    {
      "dynamics_loss_avg": 0.06615468598902226,
      "epoch": 3.3855438232421873e-06,
      "step": 77550
    },
    {
      "action_loss_avg": 0.0020521987753454596,
      "epoch": 3.3855438232421873e-06,
      "step": 77550
    },
    {
      "epoch": 3.3855438232421873e-06,
      "step": 77550,
      "training_step_time": 7.510580539703369
    },
    {
      "epoch": 3.386497497558594e-06,
      "model_forward_time": 1.73476243019104,
      "step": 77551
    },
    {
      "epoch": 3.386497497558594e-06,
      "step": 77551,
      "training_step_time": 7.8333001136779785
    },
    {
      "epoch": 3.387451171875e-06,
      "model_forward_time": 1.7253403663635254,
      "step": 77552
    },
    {
      "epoch": 3.387451171875e-06,
      "step": 77552,
      "training_step_time": 7.598703145980835
    },
    {
      "epoch": 3.388404846191406e-06,
      "model_forward_time": 1.7335569858551025,
      "step": 77553
    },
    {
      "epoch": 3.388404846191406e-06,
      "step": 77553,
      "training_step_time": 6.631862640380859
    },
    {
      "epoch": 3.3893585205078123e-06,
      "model_forward_time": 1.737656593322754,
      "step": 77554
    },
    {
      "epoch": 3.3893585205078123e-06,
      "step": 77554,
      "training_step_time": 6.7375829219818115
    },
    {
      "epoch": 3.390312194824219e-06,
      "model_forward_time": 1.7341856956481934,
      "step": 77555
    },
    {
      "epoch": 3.390312194824219e-06,
      "step": 77555,
      "training_step_time": 7.81737208366394
    },
    {
      "epoch": 3.391265869140625e-06,
      "model_forward_time": 1.9722466468811035,
      "step": 77556
    },
    {
      "epoch": 3.391265869140625e-06,
      "step": 77556,
      "training_step_time": 8.789667129516602
    },
    {
      "epoch": 3.392219543457031e-06,
      "model_forward_time": 1.839195728302002,
      "step": 77557
    },
    {
      "epoch": 3.392219543457031e-06,
      "step": 77557,
      "training_step_time": 8.608101606369019
    },
    {
      "epoch": 3.3931732177734373e-06,
      "model_forward_time": 1.7389168739318848,
      "step": 77558
    },
    {
      "epoch": 3.3931732177734373e-06,
      "step": 77558,
      "training_step_time": 8.7059485912323
    },
    {
      "epoch": 3.394126892089844e-06,
      "model_forward_time": 1.7295024394989014,
      "step": 77559
    },
    {
      "epoch": 3.394126892089844e-06,
      "step": 77559,
      "training_step_time": 7.5391318798065186
    },
    {
      "epoch": 3.39508056640625e-06,
      "grad_norm": 0.1148054301738739,
      "learning_rate": 1.31478025129528e-06,
      "loss": 0.0562,
      "step": 77560
    },
    {
      "epoch": 3.39508056640625e-06,
      "model_forward_time": 1.7305073738098145,
      "step": 77560
    },
    {
      "dynamics_loss_avg": 0.053113738633692266,
      "epoch": 3.39508056640625e-06,
      "step": 77560
    },
    {
      "action_loss_avg": 0.0017079139419365675,
      "epoch": 3.39508056640625e-06,
      "step": 77560
    },
    {
      "epoch": 3.39508056640625e-06,
      "step": 77560,
      "training_step_time": 7.544910669326782
    },
    {
      "epoch": 3.396034240722656e-06,
      "model_forward_time": 1.7317309379577637,
      "step": 77561
    },
    {
      "epoch": 3.396034240722656e-06,
      "step": 77561,
      "training_step_time": 8.436014890670776
    },
    {
      "epoch": 3.3969879150390623e-06,
      "model_forward_time": 1.7343993186950684,
      "step": 77562
    },
    {
      "epoch": 3.3969879150390623e-06,
      "step": 77562,
      "training_step_time": 6.727713584899902
    },
    {
      "epoch": 3.397941589355469e-06,
      "model_forward_time": 1.7319951057434082,
      "step": 77563
    },
    {
      "epoch": 3.397941589355469e-06,
      "step": 77563,
      "training_step_time": 6.717729568481445
    },
    {
      "epoch": 3.398895263671875e-06,
      "model_forward_time": 1.7306735515594482,
      "step": 77564
    },
    {
      "epoch": 3.398895263671875e-06,
      "step": 77564,
      "training_step_time": 7.6677916049957275
    },
    {
      "epoch": 3.399848937988281e-06,
      "model_forward_time": 2.354898691177368,
      "step": 77565
    },
    {
      "epoch": 3.399848937988281e-06,
      "step": 77565,
      "training_step_time": 7.5450599193573
    },
    {
      "epoch": 3.4008026123046873e-06,
      "model_forward_time": 1.734893560409546,
      "step": 77566
    },
    {
      "epoch": 3.4008026123046873e-06,
      "step": 77566,
      "training_step_time": 7.667907953262329
    },
    {
      "epoch": 3.401756286621094e-06,
      "model_forward_time": 1.8002064228057861,
      "step": 77567
    },
    {
      "epoch": 3.401756286621094e-06,
      "step": 77567,
      "training_step_time": 7.724606990814209
    },
    {
      "epoch": 3.4027099609375e-06,
      "model_forward_time": 1.7319278717041016,
      "step": 77568
    },
    {
      "epoch": 3.4027099609375e-06,
      "step": 77568,
      "training_step_time": 7.41675066947937
    },
    {
      "epoch": 3.403663635253906e-06,
      "model_forward_time": 1.7338645458221436,
      "step": 77569
    },
    {
      "epoch": 3.403663635253906e-06,
      "step": 77569,
      "training_step_time": 7.4429051876068115
    },
    {
      "epoch": 3.4046173095703123e-06,
      "grad_norm": 0.06903260946273804,
      "learning_rate": 1.3136629638149796e-06,
      "loss": 0.0554,
      "step": 77570
    },
    {
      "epoch": 3.4046173095703123e-06,
      "model_forward_time": 1.7460715770721436,
      "step": 77570
    },
    {
      "dynamics_loss_avg": 0.04113508351147175,
      "epoch": 3.4046173095703123e-06,
      "step": 77570
    },
    {
      "action_loss_avg": 0.0023268189164809884,
      "epoch": 3.4046173095703123e-06,
      "step": 77570
    },
    {
      "epoch": 3.4046173095703123e-06,
      "step": 77570,
      "training_step_time": 8.229101181030273
    },
    {
      "epoch": 3.405570983886719e-06,
      "model_forward_time": 1.7517123222351074,
      "step": 77571
    },
    {
      "epoch": 3.405570983886719e-06,
      "step": 77571,
      "training_step_time": 7.421912431716919
    },
    {
      "epoch": 3.406524658203125e-06,
      "model_forward_time": 1.727529764175415,
      "step": 77572
    },
    {
      "epoch": 3.406524658203125e-06,
      "step": 77572,
      "training_step_time": 6.743400812149048
    },
    {
      "epoch": 3.407478332519531e-06,
      "model_forward_time": 1.7350783348083496,
      "step": 77573
    },
    {
      "epoch": 3.407478332519531e-06,
      "step": 77573,
      "training_step_time": 7.6690733432769775
    },
    {
      "epoch": 3.4084320068359374e-06,
      "model_forward_time": 2.355558156967163,
      "step": 77574
    },
    {
      "epoch": 3.4084320068359374e-06,
      "step": 77574,
      "training_step_time": 7.567711353302002
    },
    {
      "epoch": 3.409385681152344e-06,
      "model_forward_time": 1.7288105487823486,
      "step": 77575
    },
    {
      "epoch": 3.409385681152344e-06,
      "step": 77575,
      "training_step_time": 7.732804298400879
    },
    {
      "epoch": 3.41033935546875e-06,
      "model_forward_time": 1.7991254329681396,
      "step": 77576
    },
    {
      "epoch": 3.41033935546875e-06,
      "step": 77576,
      "training_step_time": 8.66437554359436
    },
    {
      "epoch": 3.411293029785156e-06,
      "model_forward_time": 1.7339739799499512,
      "step": 77577
    },
    {
      "epoch": 3.411293029785156e-06,
      "step": 77577,
      "training_step_time": 7.430711507797241
    },
    {
      "epoch": 3.4122467041015624e-06,
      "model_forward_time": 1.7350082397460938,
      "step": 77578
    },
    {
      "epoch": 3.4122467041015624e-06,
      "step": 77578,
      "training_step_time": 7.342047452926636
    },
    {
      "epoch": 3.413200378417969e-06,
      "model_forward_time": 1.7465782165527344,
      "step": 77579
    },
    {
      "epoch": 3.413200378417969e-06,
      "step": 77579,
      "training_step_time": 7.7301201820373535
    },
    {
      "epoch": 3.414154052734375e-06,
      "grad_norm": 0.08393318951129913,
      "learning_rate": 1.3125460794669548e-06,
      "loss": 0.0559,
      "step": 77580
    },
    {
      "epoch": 3.414154052734375e-06,
      "model_forward_time": 1.737131118774414,
      "step": 77580
    },
    {
      "dynamics_loss_avg": 0.04803024847060442,
      "epoch": 3.414154052734375e-06,
      "step": 77580
    },
    {
      "action_loss_avg": 0.0016320142807671801,
      "epoch": 3.414154052734375e-06,
      "step": 77580
    },
    {
      "epoch": 3.414154052734375e-06,
      "step": 77580,
      "training_step_time": 7.406650066375732
    },
    {
      "epoch": 3.4151077270507812e-06,
      "model_forward_time": 1.7349085807800293,
      "step": 77581
    },
    {
      "epoch": 3.4151077270507812e-06,
      "step": 77581,
      "training_step_time": 6.719797134399414
    },
    {
      "epoch": 3.4160614013671874e-06,
      "model_forward_time": 1.7455792427062988,
      "step": 77582
    },
    {
      "epoch": 3.4160614013671874e-06,
      "step": 77582,
      "training_step_time": 7.696464776992798
    },
    {
      "epoch": 3.417015075683594e-06,
      "model_forward_time": 1.0262548923492432,
      "step": 77583
    },
    {
      "epoch": 3.417015075683594e-06,
      "step": 77583,
      "training_step_time": 7.59070348739624
    },
    {
      "epoch": 3.41796875e-06,
      "model_forward_time": 1.737830400466919,
      "step": 77584
    },
    {
      "epoch": 3.41796875e-06,
      "step": 77584,
      "training_step_time": 8.48567008972168
    },
    {
      "epoch": 3.4189224243164062e-06,
      "model_forward_time": 1.7835149765014648,
      "step": 77585
    },
    {
      "epoch": 3.4189224243164062e-06,
      "step": 77585,
      "training_step_time": 8.383285522460938
    },
    {
      "epoch": 3.4198760986328124e-06,
      "model_forward_time": 1.7342448234558105,
      "step": 77586
    },
    {
      "epoch": 3.4198760986328124e-06,
      "step": 77586,
      "training_step_time": 8.930922269821167
    },
    {
      "epoch": 3.420829772949219e-06,
      "model_forward_time": 1.7312970161437988,
      "step": 77587
    },
    {
      "epoch": 3.420829772949219e-06,
      "step": 77587,
      "training_step_time": 7.5293474197387695
    },
    {
      "epoch": 3.421783447265625e-06,
      "model_forward_time": 1.735236644744873,
      "step": 77588
    },
    {
      "epoch": 3.421783447265625e-06,
      "step": 77588,
      "training_step_time": 7.480811357498169
    },
    {
      "epoch": 3.4227371215820312e-06,
      "model_forward_time": 1.729585886001587,
      "step": 77589
    },
    {
      "epoch": 3.4227371215820312e-06,
      "step": 77589,
      "training_step_time": 7.86602258682251
    },
    {
      "epoch": 3.4236907958984374e-06,
      "grad_norm": 0.08577455580234528,
      "learning_rate": 1.3114295983733483e-06,
      "loss": 0.056,
      "step": 77590
    },
    {
      "epoch": 3.4236907958984374e-06,
      "model_forward_time": 1.7326104640960693,
      "step": 77590
    },
    {
      "dynamics_loss_avg": 0.05592119470238686,
      "epoch": 3.4236907958984374e-06,
      "step": 77590
    },
    {
      "action_loss_avg": 0.0008081159903667867,
      "epoch": 3.4236907958984374e-06,
      "step": 77590
    },
    {
      "epoch": 3.4236907958984374e-06,
      "step": 77590,
      "training_step_time": 6.802441835403442
    },
    {
      "epoch": 3.424644470214844e-06,
      "model_forward_time": 1.7246410846710205,
      "step": 77591
    },
    {
      "epoch": 3.424644470214844e-06,
      "step": 77591,
      "training_step_time": 7.612093210220337
    },
    {
      "epoch": 3.42559814453125e-06,
      "model_forward_time": 1.7463572025299072,
      "step": 77592
    },
    {
      "epoch": 3.42559814453125e-06,
      "step": 77592,
      "training_step_time": 8.829878330230713
    },
    {
      "epoch": 3.4265518188476562e-06,
      "model_forward_time": 2.4092695713043213,
      "step": 77593
    },
    {
      "epoch": 3.4265518188476562e-06,
      "step": 77593,
      "training_step_time": 8.369622230529785
    },
    {
      "epoch": 3.4275054931640624e-06,
      "model_forward_time": 1.8094241619110107,
      "step": 77594
    },
    {
      "epoch": 3.4275054931640624e-06,
      "step": 77594,
      "training_step_time": 7.805628299713135
    },
    {
      "epoch": 3.428459167480469e-06,
      "model_forward_time": 1.7890856266021729,
      "step": 77595
    },
    {
      "epoch": 3.428459167480469e-06,
      "step": 77595,
      "training_step_time": 7.645563840866089
    },
    {
      "epoch": 3.429412841796875e-06,
      "model_forward_time": 1.7485899925231934,
      "step": 77596
    },
    {
      "epoch": 3.429412841796875e-06,
      "step": 77596,
      "training_step_time": 7.552007675170898
    },
    {
      "epoch": 3.4303665161132812e-06,
      "model_forward_time": 1.7576582431793213,
      "step": 77597
    },
    {
      "epoch": 3.4303665161132812e-06,
      "step": 77597,
      "training_step_time": 7.412508010864258
    },
    {
      "epoch": 3.4313201904296874e-06,
      "model_forward_time": 1.7560710906982422,
      "step": 77598
    },
    {
      "epoch": 3.4313201904296874e-06,
      "step": 77598,
      "training_step_time": 7.749350309371948
    },
    {
      "epoch": 3.432273864746094e-06,
      "model_forward_time": 1.7692806720733643,
      "step": 77599
    },
    {
      "epoch": 3.432273864746094e-06,
      "step": 77599,
      "training_step_time": 6.7720561027526855
    },
    {
      "epoch": 3.4332275390625e-06,
      "grad_norm": 0.09503167122602463,
      "learning_rate": 1.3103135206562535e-06,
      "loss": 0.0544,
      "step": 77600
    },
    {
      "epoch": 3.4332275390625e-06,
      "model_forward_time": 1.7719104290008545,
      "step": 77600
    },
    {
      "dynamics_loss_avg": 0.06310877446085214,
      "epoch": 3.4332275390625e-06,
      "step": 77600
    },
    {
      "action_loss_avg": 0.0017620819911826403,
      "epoch": 3.4332275390625e-06,
      "step": 77600
    },
    {
      "epoch": 3.4332275390625e-06,
      "step": 77600,
      "training_step_time": 7.810068130493164
    },
    {
      "epoch": 3.4341812133789062e-06,
      "model_forward_time": 1.7742195129394531,
      "step": 77601
    },
    {
      "epoch": 3.4341812133789062e-06,
      "step": 77601,
      "training_step_time": 7.788172006607056
    },
    {
      "epoch": 3.4351348876953124e-06,
      "model_forward_time": 2.3997089862823486,
      "step": 77602
    },
    {
      "epoch": 3.4351348876953124e-06,
      "step": 77602,
      "training_step_time": 9.28276252746582
    },
    {
      "epoch": 3.436088562011719e-06,
      "model_forward_time": 1.7758245468139648,
      "step": 77603
    },
    {
      "epoch": 3.436088562011719e-06,
      "step": 77603,
      "training_step_time": 8.672912836074829
    },
    {
      "epoch": 3.437042236328125e-06,
      "model_forward_time": 1.7591474056243896,
      "step": 77604
    },
    {
      "epoch": 3.437042236328125e-06,
      "step": 77604,
      "training_step_time": 8.5624418258667
    },
    {
      "epoch": 3.4379959106445312e-06,
      "model_forward_time": 1.7665181159973145,
      "step": 77605
    },
    {
      "epoch": 3.4379959106445312e-06,
      "step": 77605,
      "training_step_time": 7.52598762512207
    },
    {
      "epoch": 3.4389495849609374e-06,
      "model_forward_time": 1.7721316814422607,
      "step": 77606
    },
    {
      "epoch": 3.4389495849609374e-06,
      "step": 77606,
      "training_step_time": 7.504628419876099
    },
    {
      "epoch": 3.439903259277344e-06,
      "model_forward_time": 1.761129379272461,
      "step": 77607
    },
    {
      "epoch": 3.439903259277344e-06,
      "step": 77607,
      "training_step_time": 7.696717023849487
    },
    {
      "epoch": 3.44085693359375e-06,
      "model_forward_time": 1.7715396881103516,
      "step": 77608
    },
    {
      "epoch": 3.44085693359375e-06,
      "step": 77608,
      "training_step_time": 6.83724045753479
    },
    {
      "epoch": 3.4418106079101562e-06,
      "model_forward_time": 1.7624950408935547,
      "step": 77609
    },
    {
      "epoch": 3.4418106079101562e-06,
      "step": 77609,
      "training_step_time": 7.7921669483184814
    },
    {
      "epoch": 3.4427642822265624e-06,
      "grad_norm": 0.08709512650966644,
      "learning_rate": 1.3091978464377264e-06,
      "loss": 0.056,
      "step": 77610
    },
    {
      "epoch": 3.4427642822265624e-06,
      "model_forward_time": 1.7663028240203857,
      "step": 77610
    },
    {
      "dynamics_loss_avg": 0.04307122118771076,
      "epoch": 3.4427642822265624e-06,
      "step": 77610
    },
    {
      "action_loss_avg": 0.0010785927530378102,
      "epoch": 3.4427642822265624e-06,
      "step": 77610
    },
    {
      "epoch": 3.4427642822265624e-06,
      "step": 77610,
      "training_step_time": 8.763629913330078
    },
    {
      "epoch": 3.443717956542969e-06,
      "model_forward_time": 2.4224958419799805,
      "step": 77611
    },
    {
      "epoch": 3.443717956542969e-06,
      "step": 77611,
      "training_step_time": 9.621302366256714
    },
    {
      "epoch": 3.444671630859375e-06,
      "model_forward_time": 1.7737047672271729,
      "step": 77612
    },
    {
      "epoch": 3.444671630859375e-06,
      "step": 77612,
      "training_step_time": 8.69750690460205
    },
    {
      "epoch": 3.4456253051757812e-06,
      "model_forward_time": 1.821648120880127,
      "step": 77613
    },
    {
      "epoch": 3.4456253051757812e-06,
      "step": 77613,
      "training_step_time": 7.818565130233765
    },
    {
      "epoch": 3.4465789794921874e-06,
      "model_forward_time": 1.7677884101867676,
      "step": 77614
    },
    {
      "epoch": 3.4465789794921874e-06,
      "step": 77614,
      "training_step_time": 7.539878606796265
    },
    {
      "epoch": 3.447532653808594e-06,
      "model_forward_time": 1.7622802257537842,
      "step": 77615
    },
    {
      "epoch": 3.447532653808594e-06,
      "step": 77615,
      "training_step_time": 7.506852149963379
    },
    {
      "epoch": 3.448486328125e-06,
      "model_forward_time": 1.7601683139801025,
      "step": 77616
    },
    {
      "epoch": 3.448486328125e-06,
      "step": 77616,
      "training_step_time": 7.668860673904419
    },
    {
      "epoch": 3.4494400024414062e-06,
      "model_forward_time": 1.763113021850586,
      "step": 77617
    },
    {
      "epoch": 3.4494400024414062e-06,
      "step": 77617,
      "training_step_time": 7.566062688827515
    },
    {
      "epoch": 3.4503936767578124e-06,
      "model_forward_time": 1.7645697593688965,
      "step": 77618
    },
    {
      "epoch": 3.4503936767578124e-06,
      "step": 77618,
      "training_step_time": 7.751169443130493
    },
    {
      "epoch": 3.451347351074219e-06,
      "model_forward_time": 1.7616536617279053,
      "step": 77619
    },
    {
      "epoch": 3.451347351074219e-06,
      "step": 77619,
      "training_step_time": 8.59336805343628
    },
    {
      "epoch": 3.452301025390625e-06,
      "grad_norm": 0.08926679939031601,
      "learning_rate": 1.3080825758397724e-06,
      "loss": 0.0555,
      "step": 77620
    },
    {
      "epoch": 3.452301025390625e-06,
      "model_forward_time": 2.3953006267547607,
      "step": 77620
    },
    {
      "dynamics_loss_avg": 0.05643975678831339,
      "epoch": 3.452301025390625e-06,
      "step": 77620
    },
    {
      "action_loss_avg": 0.0009204499481711536,
      "epoch": 3.452301025390625e-06,
      "step": 77620
    },
    {
      "epoch": 3.452301025390625e-06,
      "step": 77620,
      "training_step_time": 8.282573461532593
    },
    {
      "epoch": 3.4532546997070312e-06,
      "model_forward_time": 1.3890178203582764,
      "step": 77621
    },
    {
      "epoch": 3.4532546997070312e-06,
      "step": 77621,
      "training_step_time": 8.937482118606567
    },
    {
      "epoch": 3.4542083740234374e-06,
      "model_forward_time": 1.8293688297271729,
      "step": 77622
    },
    {
      "epoch": 3.4542083740234374e-06,
      "step": 77622,
      "training_step_time": 8.558761358261108
    },
    {
      "epoch": 3.455162048339844e-06,
      "model_forward_time": 1.7699427604675293,
      "step": 77623
    },
    {
      "epoch": 3.455162048339844e-06,
      "step": 77623,
      "training_step_time": 7.550699234008789
    },
    {
      "epoch": 3.45611572265625e-06,
      "model_forward_time": 1.7653026580810547,
      "step": 77624
    },
    {
      "epoch": 3.45611572265625e-06,
      "step": 77624,
      "training_step_time": 8.219779014587402
    },
    {
      "epoch": 3.4570693969726562e-06,
      "model_forward_time": 1.7677323818206787,
      "step": 77625
    },
    {
      "epoch": 3.4570693969726562e-06,
      "step": 77625,
      "training_step_time": 7.572450160980225
    },
    {
      "epoch": 3.4580230712890624e-06,
      "model_forward_time": 1.776498794555664,
      "step": 77626
    },
    {
      "epoch": 3.4580230712890624e-06,
      "step": 77626,
      "training_step_time": 7.8006134033203125
    },
    {
      "epoch": 3.4589767456054685e-06,
      "model_forward_time": 1.7737677097320557,
      "step": 77627
    },
    {
      "epoch": 3.4589767456054685e-06,
      "step": 77627,
      "training_step_time": 7.743121862411499
    },
    {
      "epoch": 3.459930419921875e-06,
      "model_forward_time": 1.7678725719451904,
      "step": 77628
    },
    {
      "epoch": 3.459930419921875e-06,
      "step": 77628,
      "training_step_time": 7.806255578994751
    },
    {
      "epoch": 3.4608840942382813e-06,
      "model_forward_time": 1.7575061321258545,
      "step": 77629
    },
    {
      "epoch": 3.4608840942382813e-06,
      "step": 77629,
      "training_step_time": 7.758235454559326
    },
    {
      "epoch": 3.4618377685546874e-06,
      "grad_norm": 0.08423551172018051,
      "learning_rate": 1.3069677089843568e-06,
      "loss": 0.0545,
      "step": 77630
    },
    {
      "epoch": 3.4618377685546874e-06,
      "model_forward_time": 2.432135581970215,
      "step": 77630
    },
    {
      "dynamics_loss_avg": 0.06450701747089624,
      "epoch": 3.4618377685546874e-06,
      "step": 77630
    },
    {
      "action_loss_avg": 0.0020511827955488116,
      "epoch": 3.4618377685546874e-06,
      "step": 77630
    },
    {
      "epoch": 3.4618377685546874e-06,
      "step": 77630,
      "training_step_time": 9.651187181472778
    },
    {
      "epoch": 3.4627914428710935e-06,
      "model_forward_time": 1.853173017501831,
      "step": 77631
    },
    {
      "epoch": 3.4627914428710935e-06,
      "step": 77631,
      "training_step_time": 10.271989822387695
    },
    {
      "epoch": 3.4637451171875e-06,
      "model_forward_time": 1.7645161151885986,
      "step": 77632
    },
    {
      "epoch": 3.4637451171875e-06,
      "step": 77632,
      "training_step_time": 7.586335897445679
    },
    {
      "epoch": 3.4646987915039063e-06,
      "model_forward_time": 1.7775623798370361,
      "step": 77633
    },
    {
      "epoch": 3.4646987915039063e-06,
      "step": 77633,
      "training_step_time": 8.18418574333191
    },
    {
      "epoch": 3.4656524658203124e-06,
      "model_forward_time": 1.8093695640563965,
      "step": 77634
    },
    {
      "epoch": 3.4656524658203124e-06,
      "step": 77634,
      "training_step_time": 8.42438817024231
    },
    {
      "epoch": 3.4666061401367185e-06,
      "model_forward_time": 1.771585464477539,
      "step": 77635
    },
    {
      "epoch": 3.4666061401367185e-06,
      "step": 77635,
      "training_step_time": 7.868336200714111
    },
    {
      "epoch": 3.467559814453125e-06,
      "model_forward_time": 1.7614350318908691,
      "step": 77636
    },
    {
      "epoch": 3.467559814453125e-06,
      "step": 77636,
      "training_step_time": 8.478267192840576
    },
    {
      "epoch": 3.4685134887695313e-06,
      "model_forward_time": 1.784026861190796,
      "step": 77637
    },
    {
      "epoch": 3.4685134887695313e-06,
      "step": 77637,
      "training_step_time": 7.543492078781128
    },
    {
      "epoch": 3.4694671630859374e-06,
      "model_forward_time": 1.8345065116882324,
      "step": 77638
    },
    {
      "epoch": 3.4694671630859374e-06,
      "step": 77638,
      "training_step_time": 7.770326375961304
    },
    {
      "epoch": 3.4704208374023436e-06,
      "model_forward_time": 2.4352786540985107,
      "step": 77639
    },
    {
      "epoch": 3.4704208374023436e-06,
      "step": 77639,
      "training_step_time": 9.761260509490967
    },
    {
      "epoch": 3.47137451171875e-06,
      "grad_norm": 0.08852456510066986,
      "learning_rate": 1.3058532459934009e-06,
      "loss": 0.0532,
      "step": 77640
    },
    {
      "epoch": 3.47137451171875e-06,
      "model_forward_time": 1.3962550163269043,
      "step": 77640
    },
    {
      "dynamics_loss_avg": 0.05290789492428303,
      "epoch": 3.47137451171875e-06,
      "step": 77640
    },
    {
      "action_loss_avg": 0.003775741334538907,
      "epoch": 3.47137451171875e-06,
      "step": 77640
    },
    {
      "epoch": 3.47137451171875e-06,
      "step": 77640,
      "training_step_time": 8.605232954025269
    },
    {
      "epoch": 3.4723281860351563e-06,
      "model_forward_time": 1.796327829360962,
      "step": 77641
    },
    {
      "epoch": 3.4723281860351563e-06,
      "step": 77641,
      "training_step_time": 7.81243109703064
    },
    {
      "epoch": 3.4732818603515624e-06,
      "model_forward_time": 1.7608106136322021,
      "step": 77642
    },
    {
      "epoch": 3.4732818603515624e-06,
      "step": 77642,
      "training_step_time": 7.5639612674713135
    },
    {
      "epoch": 3.4742355346679686e-06,
      "model_forward_time": 1.757720947265625,
      "step": 77643
    },
    {
      "epoch": 3.4742355346679686e-06,
      "step": 77643,
      "training_step_time": 7.660516023635864
    },
    {
      "epoch": 3.475189208984375e-06,
      "model_forward_time": 1.7830512523651123,
      "step": 77644
    },
    {
      "epoch": 3.475189208984375e-06,
      "step": 77644,
      "training_step_time": 7.710214614868164
    },
    {
      "epoch": 3.4761428833007813e-06,
      "model_forward_time": 1.8077635765075684,
      "step": 77645
    },
    {
      "epoch": 3.4761428833007813e-06,
      "step": 77645,
      "training_step_time": 7.816571235656738
    },
    {
      "epoch": 3.4770965576171874e-06,
      "model_forward_time": 1.7735261917114258,
      "step": 77646
    },
    {
      "epoch": 3.4770965576171874e-06,
      "step": 77646,
      "training_step_time": 7.550265550613403
    },
    {
      "epoch": 3.4780502319335936e-06,
      "model_forward_time": 1.7720773220062256,
      "step": 77647
    },
    {
      "epoch": 3.4780502319335936e-06,
      "step": 77647,
      "training_step_time": 7.870420455932617
    },
    {
      "epoch": 3.47900390625e-06,
      "model_forward_time": 2.4574151039123535,
      "step": 77648
    },
    {
      "epoch": 3.47900390625e-06,
      "step": 77648,
      "training_step_time": 7.893362522125244
    },
    {
      "epoch": 3.4799575805664063e-06,
      "model_forward_time": 1.7762396335601807,
      "step": 77649
    },
    {
      "epoch": 3.4799575805664063e-06,
      "step": 77649,
      "training_step_time": 9.304179906845093
    },
    {
      "epoch": 3.4809112548828124e-06,
      "grad_norm": 0.07331332564353943,
      "learning_rate": 1.304739186988781e-06,
      "loss": 0.0527,
      "step": 77650
    },
    {
      "epoch": 3.4809112548828124e-06,
      "model_forward_time": 1.7324295043945312,
      "step": 77650
    },
    {
      "dynamics_loss_avg": 0.05004212893545627,
      "epoch": 3.4809112548828124e-06,
      "step": 77650
    },
    {
      "action_loss_avg": 0.0014902333670761437,
      "epoch": 3.4809112548828124e-06,
      "step": 77650
    },
    {
      "epoch": 3.4809112548828124e-06,
      "step": 77650,
      "training_step_time": 10.569074630737305
    },
    {
      "epoch": 3.4818649291992186e-06,
      "model_forward_time": 1.8047370910644531,
      "step": 77651
    },
    {
      "epoch": 3.4818649291992186e-06,
      "step": 77651,
      "training_step_time": 8.295275926589966
    },
    {
      "epoch": 3.482818603515625e-06,
      "model_forward_time": 1.7569916248321533,
      "step": 77652
    },
    {
      "epoch": 3.482818603515625e-06,
      "step": 77652,
      "training_step_time": 8.093990802764893
    },
    {
      "epoch": 3.4837722778320313e-06,
      "model_forward_time": 1.759026288986206,
      "step": 77653
    },
    {
      "epoch": 3.4837722778320313e-06,
      "step": 77653,
      "training_step_time": 7.652558088302612
    },
    {
      "epoch": 3.4847259521484374e-06,
      "model_forward_time": 1.7485504150390625,
      "step": 77654
    },
    {
      "epoch": 3.4847259521484374e-06,
      "step": 77654,
      "training_step_time": 7.841365575790405
    },
    {
      "epoch": 3.4856796264648436e-06,
      "model_forward_time": 1.7450003623962402,
      "step": 77655
    },
    {
      "epoch": 3.4856796264648436e-06,
      "step": 77655,
      "training_step_time": 7.544863224029541
    },
    {
      "epoch": 3.48663330078125e-06,
      "model_forward_time": 1.7516744136810303,
      "step": 77656
    },
    {
      "epoch": 3.48663330078125e-06,
      "step": 77656,
      "training_step_time": 7.7421040534973145
    },
    {
      "epoch": 3.4875869750976563e-06,
      "model_forward_time": 2.3931877613067627,
      "step": 77657
    },
    {
      "epoch": 3.4875869750976563e-06,
      "step": 77657,
      "training_step_time": 8.590912342071533
    },
    {
      "epoch": 3.4885406494140624e-06,
      "model_forward_time": 1.7506954669952393,
      "step": 77658
    },
    {
      "epoch": 3.4885406494140624e-06,
      "step": 77658,
      "training_step_time": 9.450085401535034
    },
    {
      "epoch": 3.4894943237304686e-06,
      "model_forward_time": 1.745588779449463,
      "step": 77659
    },
    {
      "epoch": 3.4894943237304686e-06,
      "step": 77659,
      "training_step_time": 7.944663047790527
    },
    {
      "epoch": 3.490447998046875e-06,
      "grad_norm": 0.18019604682922363,
      "learning_rate": 1.3036255320923264e-06,
      "loss": 0.0548,
      "step": 77660
    },
    {
      "epoch": 3.490447998046875e-06,
      "model_forward_time": 1.7991855144500732,
      "step": 77660
    },
    {
      "dynamics_loss_avg": 0.05789326895028353,
      "epoch": 3.490447998046875e-06,
      "step": 77660
    },
    {
      "action_loss_avg": 0.0011866397690027952,
      "epoch": 3.490447998046875e-06,
      "step": 77660
    },
    {
      "epoch": 3.490447998046875e-06,
      "step": 77660,
      "training_step_time": 7.659895420074463
    },
    {
      "epoch": 3.4914016723632813e-06,
      "model_forward_time": 1.7401745319366455,
      "step": 77661
    },
    {
      "epoch": 3.4914016723632813e-06,
      "step": 77661,
      "training_step_time": 7.521576166152954
    },
    {
      "epoch": 3.4923553466796874e-06,
      "model_forward_time": 1.7561993598937988,
      "step": 77662
    },
    {
      "epoch": 3.4923553466796874e-06,
      "step": 77662,
      "training_step_time": 7.700204610824585
    },
    {
      "epoch": 3.4933090209960936e-06,
      "model_forward_time": 1.7848877906799316,
      "step": 77663
    },
    {
      "epoch": 3.4933090209960936e-06,
      "step": 77663,
      "training_step_time": 7.849618673324585
    },
    {
      "epoch": 3.4942626953125e-06,
      "model_forward_time": 1.7514431476593018,
      "step": 77664
    },
    {
      "epoch": 3.4942626953125e-06,
      "step": 77664,
      "training_step_time": 7.541255712509155
    },
    {
      "epoch": 3.4952163696289063e-06,
      "model_forward_time": 1.755422830581665,
      "step": 77665
    },
    {
      "epoch": 3.4952163696289063e-06,
      "step": 77665,
      "training_step_time": 7.785053730010986
    },
    {
      "epoch": 3.4961700439453124e-06,
      "model_forward_time": 1.75,
      "step": 77666
    },
    {
      "epoch": 3.4961700439453124e-06,
      "step": 77666,
      "training_step_time": 10.029997825622559
    },
    {
      "epoch": 3.4971237182617186e-06,
      "model_forward_time": 1.7524726390838623,
      "step": 77667
    },
    {
      "epoch": 3.4971237182617186e-06,
      "step": 77667,
      "training_step_time": 8.809149503707886
    },
    {
      "epoch": 3.498077392578125e-06,
      "model_forward_time": 1.745086908340454,
      "step": 77668
    },
    {
      "epoch": 3.498077392578125e-06,
      "step": 77668,
      "training_step_time": 8.611888885498047
    },
    {
      "epoch": 3.4990310668945313e-06,
      "model_forward_time": 1.8097598552703857,
      "step": 77669
    },
    {
      "epoch": 3.4990310668945313e-06,
      "step": 77669,
      "training_step_time": 8.604768753051758
    },
    {
      "epoch": 3.4999847412109374e-06,
      "grad_norm": 0.09135432541370392,
      "learning_rate": 1.3025122814258274e-06,
      "loss": 0.0536,
      "step": 77670
    },
    {
      "epoch": 3.4999847412109374e-06,
      "model_forward_time": 1.737626552581787,
      "step": 77670
    },
    {
      "dynamics_loss_avg": 0.05109630562365055,
      "epoch": 3.4999847412109374e-06,
      "step": 77670
    },
    {
      "action_loss_avg": 0.002233921765582636,
      "epoch": 3.4999847412109374e-06,
      "step": 77670
    },
    {
      "epoch": 3.4999847412109374e-06,
      "step": 77670,
      "training_step_time": 7.564779281616211
    },
    {
      "epoch": 3.5009384155273436e-06,
      "model_forward_time": 1.7453808784484863,
      "step": 77671
    },
    {
      "epoch": 3.5009384155273436e-06,
      "step": 77671,
      "training_step_time": 7.739705801010132
    },
    {
      "epoch": 3.50189208984375e-06,
      "model_forward_time": 1.7417476177215576,
      "step": 77672
    },
    {
      "epoch": 3.50189208984375e-06,
      "step": 77672,
      "training_step_time": 7.797618865966797
    },
    {
      "epoch": 3.5028457641601563e-06,
      "model_forward_time": 1.7775259017944336,
      "step": 77673
    },
    {
      "epoch": 3.5028457641601563e-06,
      "step": 77673,
      "training_step_time": 7.743573427200317
    },
    {
      "epoch": 3.5037994384765624e-06,
      "model_forward_time": 1.7390694618225098,
      "step": 77674
    },
    {
      "epoch": 3.5037994384765624e-06,
      "step": 77674,
      "training_step_time": 7.75004768371582
    },
    {
      "epoch": 3.5047531127929686e-06,
      "model_forward_time": 1.7312400341033936,
      "step": 77675
    },
    {
      "epoch": 3.5047531127929686e-06,
      "step": 77675,
      "training_step_time": 7.816834211349487
    },
    {
      "epoch": 3.505706787109375e-06,
      "model_forward_time": 2.3686821460723877,
      "step": 77676
    },
    {
      "epoch": 3.505706787109375e-06,
      "step": 77676,
      "training_step_time": 8.942079544067383
    },
    {
      "epoch": 3.5066604614257813e-06,
      "model_forward_time": 1.728097915649414,
      "step": 77677
    },
    {
      "epoch": 3.5066604614257813e-06,
      "step": 77677,
      "training_step_time": 7.777944803237915
    },
    {
      "epoch": 3.5076141357421874e-06,
      "model_forward_time": 1.771691083908081,
      "step": 77678
    },
    {
      "epoch": 3.5076141357421874e-06,
      "step": 77678,
      "training_step_time": 7.904221296310425
    },
    {
      "epoch": 3.5085678100585936e-06,
      "model_forward_time": 1.7414882183074951,
      "step": 77679
    },
    {
      "epoch": 3.5085678100585936e-06,
      "step": 77679,
      "training_step_time": 7.555018663406372
    },
    {
      "epoch": 3.509521484375e-06,
      "grad_norm": 0.11011781543493271,
      "learning_rate": 1.3013994351110247e-06,
      "loss": 0.0536,
      "step": 77680
    },
    {
      "epoch": 3.509521484375e-06,
      "model_forward_time": 1.7487232685089111,
      "step": 77680
    },
    {
      "dynamics_loss_avg": 0.061845910549163816,
      "epoch": 3.509521484375e-06,
      "step": 77680
    },
    {
      "action_loss_avg": 0.0016909707454033196,
      "epoch": 3.509521484375e-06,
      "step": 77680
    },
    {
      "epoch": 3.509521484375e-06,
      "step": 77680,
      "training_step_time": 7.425191402435303
    },
    {
      "epoch": 3.5104751586914063e-06,
      "model_forward_time": 1.7387568950653076,
      "step": 77681
    },
    {
      "epoch": 3.5104751586914063e-06,
      "step": 77681,
      "training_step_time": 7.48263955116272
    },
    {
      "epoch": 3.5114288330078124e-06,
      "model_forward_time": 1.7348098754882812,
      "step": 77682
    },
    {
      "epoch": 3.5114288330078124e-06,
      "step": 77682,
      "training_step_time": 7.939878940582275
    },
    {
      "epoch": 3.5123825073242186e-06,
      "model_forward_time": 1.7457554340362549,
      "step": 77683
    },
    {
      "epoch": 3.5123825073242186e-06,
      "step": 77683,
      "training_step_time": 7.803558111190796
    },
    {
      "epoch": 3.513336181640625e-06,
      "model_forward_time": 1.744201421737671,
      "step": 77684
    },
    {
      "epoch": 3.513336181640625e-06,
      "step": 77684,
      "training_step_time": 8.624602556228638
    },
    {
      "epoch": 3.5142898559570313e-06,
      "model_forward_time": 2.398676872253418,
      "step": 77685
    },
    {
      "epoch": 3.5142898559570313e-06,
      "step": 77685,
      "training_step_time": 8.876010417938232
    },
    {
      "epoch": 3.5152435302734374e-06,
      "model_forward_time": 1.7595751285552979,
      "step": 77686
    },
    {
      "epoch": 3.5152435302734374e-06,
      "step": 77686,
      "training_step_time": 10.6622953414917
    },
    {
      "epoch": 3.5161972045898436e-06,
      "model_forward_time": 1.7644586563110352,
      "step": 77687
    },
    {
      "epoch": 3.5161972045898436e-06,
      "step": 77687,
      "training_step_time": 10.254288673400879
    },
    {
      "epoch": 3.51715087890625e-06,
      "model_forward_time": 1.7626888751983643,
      "step": 77688
    },
    {
      "epoch": 3.51715087890625e-06,
      "step": 77688,
      "training_step_time": 7.420117378234863
    },
    {
      "epoch": 3.5181045532226563e-06,
      "model_forward_time": 1.7687463760375977,
      "step": 77689
    },
    {
      "epoch": 3.5181045532226563e-06,
      "step": 77689,
      "training_step_time": 7.790533065795898
    },
    {
      "epoch": 3.5190582275390624e-06,
      "grad_norm": 0.1425512433052063,
      "learning_rate": 1.3002869932696195e-06,
      "loss": 0.0536,
      "step": 77690
    },
    {
      "epoch": 3.5190582275390624e-06,
      "model_forward_time": 1.7708678245544434,
      "step": 77690
    },
    {
      "dynamics_loss_avg": 0.060441809520125386,
      "epoch": 3.5190582275390624e-06,
      "step": 77690
    },
    {
      "action_loss_avg": 0.0019162472803145648,
      "epoch": 3.5190582275390624e-06,
      "step": 77690
    },
    {
      "epoch": 3.5190582275390624e-06,
      "step": 77690,
      "training_step_time": 7.465592622756958
    },
    {
      "epoch": 3.5200119018554686e-06,
      "model_forward_time": 1.765002965927124,
      "step": 77691
    },
    {
      "epoch": 3.5200119018554686e-06,
      "step": 77691,
      "training_step_time": 7.696698427200317
    },
    {
      "epoch": 3.520965576171875e-06,
      "model_forward_time": 1.7639796733856201,
      "step": 77692
    },
    {
      "epoch": 3.520965576171875e-06,
      "step": 77692,
      "training_step_time": 7.759496450424194
    },
    {
      "epoch": 3.5219192504882813e-06,
      "model_forward_time": 1.7556018829345703,
      "step": 77693
    },
    {
      "epoch": 3.5219192504882813e-06,
      "step": 77693,
      "training_step_time": 8.407831907272339
    },
    {
      "epoch": 3.5228729248046875e-06,
      "model_forward_time": 2.3911612033843994,
      "step": 77694
    },
    {
      "epoch": 3.5228729248046875e-06,
      "step": 77694,
      "training_step_time": 10.170360803604126
    },
    {
      "epoch": 3.5238265991210936e-06,
      "model_forward_time": 1.771946668624878,
      "step": 77695
    },
    {
      "epoch": 3.5238265991210936e-06,
      "step": 77695,
      "training_step_time": 10.424588918685913
    },
    {
      "epoch": 3.5247802734375e-06,
      "model_forward_time": 1.7603051662445068,
      "step": 77696
    },
    {
      "epoch": 3.5247802734375e-06,
      "step": 77696,
      "training_step_time": 9.734052419662476
    },
    {
      "epoch": 3.5257339477539063e-06,
      "model_forward_time": 1.8263671398162842,
      "step": 77697
    },
    {
      "epoch": 3.5257339477539063e-06,
      "step": 77697,
      "training_step_time": 9.211886882781982
    },
    {
      "epoch": 3.5266876220703125e-06,
      "model_forward_time": 1.7633898258209229,
      "step": 77698
    },
    {
      "epoch": 3.5266876220703125e-06,
      "step": 77698,
      "training_step_time": 7.7145676612854
    },
    {
      "epoch": 3.5276412963867186e-06,
      "model_forward_time": 1.7691502571105957,
      "step": 77699
    },
    {
      "epoch": 3.5276412963867186e-06,
      "step": 77699,
      "training_step_time": 7.720178842544556
    },
    {
      "epoch": 3.528594970703125e-06,
      "grad_norm": 0.09045659750699997,
      "learning_rate": 1.2991749560232663e-06,
      "loss": 0.0551,
      "step": 77700
    },
    {
      "epoch": 3.528594970703125e-06,
      "model_forward_time": 1.7516505718231201,
      "step": 77700
    },
    {
      "dynamics_loss_avg": 0.04567582048475742,
      "epoch": 3.528594970703125e-06,
      "step": 77700
    },
    {
      "action_loss_avg": 0.0009225854941178113,
      "epoch": 3.528594970703125e-06,
      "step": 77700
    },
    {
      "epoch": 3.528594970703125e-06,
      "step": 77700,
      "training_step_time": 7.695025682449341
    },
    {
      "epoch": 3.5295486450195313e-06,
      "model_forward_time": 1.771533727645874,
      "step": 77701
    },
    {
      "epoch": 3.5295486450195313e-06,
      "step": 77701,
      "training_step_time": 7.588956117630005
    },
    {
      "epoch": 3.5305023193359375e-06,
      "model_forward_time": 1.8009400367736816,
      "step": 77702
    },
    {
      "epoch": 3.5305023193359375e-06,
      "step": 77702,
      "training_step_time": 7.670129299163818
    },
    {
      "epoch": 3.5314559936523436e-06,
      "model_forward_time": 2.3940742015838623,
      "step": 77703
    },
    {
      "epoch": 3.5314559936523436e-06,
      "step": 77703,
      "training_step_time": 7.595725059509277
    },
    {
      "epoch": 3.53240966796875e-06,
      "model_forward_time": 1.750837802886963,
      "step": 77704
    },
    {
      "epoch": 3.53240966796875e-06,
      "step": 77704,
      "training_step_time": 7.7475104331970215
    },
    {
      "epoch": 3.5333633422851563e-06,
      "model_forward_time": 1.7556016445159912,
      "step": 77705
    },
    {
      "epoch": 3.5333633422851563e-06,
      "step": 77705,
      "training_step_time": 8.548637390136719
    },
    {
      "epoch": 3.5343170166015625e-06,
      "model_forward_time": 1.8191261291503906,
      "step": 77706
    },
    {
      "epoch": 3.5343170166015625e-06,
      "step": 77706,
      "training_step_time": 8.377176761627197
    },
    {
      "epoch": 3.5352706909179686e-06,
      "model_forward_time": 1.7565033435821533,
      "step": 77707
    },
    {
      "epoch": 3.5352706909179686e-06,
      "step": 77707,
      "training_step_time": 7.427960395812988
    },
    {
      "epoch": 3.536224365234375e-06,
      "model_forward_time": 1.767521858215332,
      "step": 77708
    },
    {
      "epoch": 3.536224365234375e-06,
      "step": 77708,
      "training_step_time": 7.43555212020874
    },
    {
      "epoch": 3.5371780395507813e-06,
      "model_forward_time": 1.7302932739257812,
      "step": 77709
    },
    {
      "epoch": 3.5371780395507813e-06,
      "step": 77709,
      "training_step_time": 7.847210884094238
    },
    {
      "epoch": 3.5381317138671875e-06,
      "grad_norm": 0.0841473862528801,
      "learning_rate": 1.2980633234935768e-06,
      "loss": 0.056,
      "step": 77710
    },
    {
      "epoch": 3.5381317138671875e-06,
      "model_forward_time": 1.742784023284912,
      "step": 77710
    },
    {
      "dynamics_loss_avg": 0.047523456439375875,
      "epoch": 3.5381317138671875e-06,
      "step": 77710
    },
    {
      "action_loss_avg": 0.0032088722713524477,
      "epoch": 3.5381317138671875e-06,
      "step": 77710
    },
    {
      "epoch": 3.5381317138671875e-06,
      "step": 77710,
      "training_step_time": 7.510568618774414
    },
    {
      "epoch": 3.5390853881835936e-06,
      "model_forward_time": 1.732640266418457,
      "step": 77711
    },
    {
      "epoch": 3.5390853881835936e-06,
      "step": 77711,
      "training_step_time": 8.327876806259155
    },
    {
      "epoch": 3.5400390625e-06,
      "model_forward_time": 2.3734898567199707,
      "step": 77712
    },
    {
      "epoch": 3.5400390625e-06,
      "step": 77712,
      "training_step_time": 7.582112073898315
    },
    {
      "epoch": 3.5409927368164063e-06,
      "model_forward_time": 1.7470221519470215,
      "step": 77713
    },
    {
      "epoch": 3.5409927368164063e-06,
      "step": 77713,
      "training_step_time": 7.791598081588745
    },
    {
      "epoch": 3.5419464111328125e-06,
      "model_forward_time": 1.730210781097412,
      "step": 77714
    },
    {
      "epoch": 3.5419464111328125e-06,
      "step": 77714,
      "training_step_time": 7.73529839515686
    },
    {
      "epoch": 3.5429000854492186e-06,
      "model_forward_time": 1.7846331596374512,
      "step": 77715
    },
    {
      "epoch": 3.5429000854492186e-06,
      "step": 77715,
      "training_step_time": 7.683927536010742
    },
    {
      "epoch": 3.543853759765625e-06,
      "model_forward_time": 1.740347146987915,
      "step": 77716
    },
    {
      "epoch": 3.543853759765625e-06,
      "step": 77716,
      "training_step_time": 7.408934831619263
    },
    {
      "epoch": 3.5448074340820313e-06,
      "model_forward_time": 1.7410883903503418,
      "step": 77717
    },
    {
      "epoch": 3.5448074340820313e-06,
      "step": 77717,
      "training_step_time": 7.405598402023315
    },
    {
      "epoch": 3.5457611083984375e-06,
      "model_forward_time": 1.7451252937316895,
      "step": 77718
    },
    {
      "epoch": 3.5457611083984375e-06,
      "step": 77718,
      "training_step_time": 7.285533428192139
    },
    {
      "epoch": 3.5467147827148436e-06,
      "model_forward_time": 1.7352886199951172,
      "step": 77719
    },
    {
      "epoch": 3.5467147827148436e-06,
      "step": 77719,
      "training_step_time": 7.6753151416778564
    },
    {
      "epoch": 3.54766845703125e-06,
      "grad_norm": 0.1507658213376999,
      "learning_rate": 1.2969520958021154e-06,
      "loss": 0.0541,
      "step": 77720
    },
    {
      "epoch": 3.54766845703125e-06,
      "model_forward_time": 1.7427091598510742,
      "step": 77720
    },
    {
      "dynamics_loss_avg": 0.05723901204764843,
      "epoch": 3.54766845703125e-06,
      "step": 77720
    },
    {
      "action_loss_avg": 0.001251291620428674,
      "epoch": 3.54766845703125e-06,
      "step": 77720
    },
    {
      "epoch": 3.54766845703125e-06,
      "step": 77720,
      "training_step_time": 8.700625658035278
    },
    {
      "epoch": 3.5486221313476563e-06,
      "model_forward_time": 1.6466317176818848,
      "step": 77721
    },
    {
      "epoch": 3.5486221313476563e-06,
      "step": 77721,
      "training_step_time": 7.5963475704193115
    },
    {
      "epoch": 3.5495758056640625e-06,
      "model_forward_time": 1.7535481452941895,
      "step": 77722
    },
    {
      "epoch": 3.5495758056640625e-06,
      "step": 77722,
      "training_step_time": 8.515186786651611
    },
    {
      "epoch": 3.5505294799804686e-06,
      "model_forward_time": 1.743546485900879,
      "step": 77723
    },
    {
      "epoch": 3.5505294799804686e-06,
      "step": 77723,
      "training_step_time": 9.770275592803955
    },
    {
      "epoch": 3.551483154296875e-06,
      "model_forward_time": 1.7523574829101562,
      "step": 77724
    },
    {
      "epoch": 3.551483154296875e-06,
      "step": 77724,
      "training_step_time": 7.5190935134887695
    },
    {
      "epoch": 3.5524368286132813e-06,
      "model_forward_time": 1.7375400066375732,
      "step": 77725
    },
    {
      "epoch": 3.5524368286132813e-06,
      "step": 77725,
      "training_step_time": 8.23005723953247
    },
    {
      "epoch": 3.5533905029296875e-06,
      "model_forward_time": 1.7357792854309082,
      "step": 77726
    },
    {
      "epoch": 3.5533905029296875e-06,
      "step": 77726,
      "training_step_time": 8.07982063293457
    },
    {
      "epoch": 3.5543441772460936e-06,
      "model_forward_time": 1.7439253330230713,
      "step": 77727
    },
    {
      "epoch": 3.5543441772460936e-06,
      "step": 77727,
      "training_step_time": 7.210470914840698
    },
    {
      "epoch": 3.5552978515625e-06,
      "model_forward_time": 1.7385039329528809,
      "step": 77728
    },
    {
      "epoch": 3.5552978515625e-06,
      "step": 77728,
      "training_step_time": 7.6612865924835205
    },
    {
      "epoch": 3.5562515258789063e-06,
      "model_forward_time": 1.7480123043060303,
      "step": 77729
    },
    {
      "epoch": 3.5562515258789063e-06,
      "step": 77729,
      "training_step_time": 7.66897177696228
    },
    {
      "epoch": 3.5572052001953125e-06,
      "grad_norm": 0.07081535458564758,
      "learning_rate": 1.295841273070405e-06,
      "loss": 0.0544,
      "step": 77730
    },
    {
      "epoch": 3.5572052001953125e-06,
      "model_forward_time": 2.3357176780700684,
      "step": 77730
    },
    {
      "dynamics_loss_avg": 0.04914683699607849,
      "epoch": 3.5572052001953125e-06,
      "step": 77730
    },
    {
      "action_loss_avg": 0.0012426504516042769,
      "epoch": 3.5572052001953125e-06,
      "step": 77730
    },
    {
      "epoch": 3.5572052001953125e-06,
      "step": 77730,
      "training_step_time": 7.775810480117798
    },
    {
      "epoch": 3.5581588745117186e-06,
      "model_forward_time": 1.7407398223876953,
      "step": 77731
    },
    {
      "epoch": 3.5581588745117186e-06,
      "step": 77731,
      "training_step_time": 8.3724045753479
    },
    {
      "epoch": 3.559112548828125e-06,
      "model_forward_time": 1.7387614250183105,
      "step": 77732
    },
    {
      "epoch": 3.559112548828125e-06,
      "step": 77732,
      "training_step_time": 9.352283954620361
    },
    {
      "epoch": 3.5600662231445313e-06,
      "model_forward_time": 1.736725091934204,
      "step": 77733
    },
    {
      "epoch": 3.5600662231445313e-06,
      "step": 77733,
      "training_step_time": 8.572598218917847
    },
    {
      "epoch": 3.5610198974609375e-06,
      "model_forward_time": 1.8035869598388672,
      "step": 77734
    },
    {
      "epoch": 3.5610198974609375e-06,
      "step": 77734,
      "training_step_time": 7.655641794204712
    },
    {
      "epoch": 3.5619735717773436e-06,
      "model_forward_time": 1.8181371688842773,
      "step": 77735
    },
    {
      "epoch": 3.5619735717773436e-06,
      "step": 77735,
      "training_step_time": 7.487523317337036
    },
    {
      "epoch": 3.56292724609375e-06,
      "model_forward_time": 1.7393381595611572,
      "step": 77736
    },
    {
      "epoch": 3.56292724609375e-06,
      "step": 77736,
      "training_step_time": 7.14654541015625
    },
    {
      "epoch": 3.5638809204101563e-06,
      "model_forward_time": 1.7457821369171143,
      "step": 77737
    },
    {
      "epoch": 3.5638809204101563e-06,
      "step": 77737,
      "training_step_time": 7.610287666320801
    },
    {
      "epoch": 3.5648345947265625e-06,
      "model_forward_time": 1.7299814224243164,
      "step": 77738
    },
    {
      "epoch": 3.5648345947265625e-06,
      "step": 77738,
      "training_step_time": 7.568188905715942
    },
    {
      "epoch": 3.5657882690429686e-06,
      "model_forward_time": 1.983618974685669,
      "step": 77739
    },
    {
      "epoch": 3.5657882690429686e-06,
      "step": 77739,
      "training_step_time": 8.116460084915161
    },
    {
      "epoch": 3.566741943359375e-06,
      "grad_norm": 0.07454437762498856,
      "learning_rate": 1.2947308554199256e-06,
      "loss": 0.0536,
      "step": 77740
    },
    {
      "epoch": 3.566741943359375e-06,
      "model_forward_time": 1.747293472290039,
      "step": 77740
    },
    {
      "dynamics_loss_avg": 0.04707322958856821,
      "epoch": 3.566741943359375e-06,
      "step": 77740
    },
    {
      "action_loss_avg": 0.0010840994422324003,
      "epoch": 3.566741943359375e-06,
      "step": 77740
    },
    {
      "epoch": 3.566741943359375e-06,
      "step": 77740,
      "training_step_time": 8.983018159866333
    },
    {
      "epoch": 3.5676956176757813e-06,
      "model_forward_time": 1.7695832252502441,
      "step": 77741
    },
    {
      "epoch": 3.5676956176757813e-06,
      "step": 77741,
      "training_step_time": 9.618176937103271
    },
    {
      "epoch": 3.5686492919921875e-06,
      "model_forward_time": 1.7723608016967773,
      "step": 77742
    },
    {
      "epoch": 3.5686492919921875e-06,
      "step": 77742,
      "training_step_time": 9.344541549682617
    },
    {
      "epoch": 3.5696029663085936e-06,
      "model_forward_time": 1.7900028228759766,
      "step": 77743
    },
    {
      "epoch": 3.5696029663085936e-06,
      "step": 77743,
      "training_step_time": 7.5789196491241455
    },
    {
      "epoch": 3.570556640625e-06,
      "model_forward_time": 1.7684402465820312,
      "step": 77744
    },
    {
      "epoch": 3.570556640625e-06,
      "step": 77744,
      "training_step_time": 7.657907247543335
    },
    {
      "epoch": 3.5715103149414064e-06,
      "model_forward_time": 1.7604625225067139,
      "step": 77745
    },
    {
      "epoch": 3.5715103149414064e-06,
      "step": 77745,
      "training_step_time": 8.36675500869751
    },
    {
      "epoch": 3.5724639892578125e-06,
      "model_forward_time": 1.766608476638794,
      "step": 77746
    },
    {
      "epoch": 3.5724639892578125e-06,
      "step": 77746,
      "training_step_time": 7.727768659591675
    },
    {
      "epoch": 3.5734176635742186e-06,
      "model_forward_time": 1.7678983211517334,
      "step": 77747
    },
    {
      "epoch": 3.5734176635742186e-06,
      "step": 77747,
      "training_step_time": 8.359272241592407
    },
    {
      "epoch": 3.574371337890625e-06,
      "model_forward_time": 2.394455909729004,
      "step": 77748
    },
    {
      "epoch": 3.574371337890625e-06,
      "step": 77748,
      "training_step_time": 7.655423879623413
    },
    {
      "epoch": 3.5753250122070314e-06,
      "model_forward_time": 1.7725660800933838,
      "step": 77749
    },
    {
      "epoch": 3.5753250122070314e-06,
      "step": 77749,
      "training_step_time": 8.252010345458984
    },
    {
      "epoch": 3.5762786865234375e-06,
      "grad_norm": 0.09626195579767227,
      "learning_rate": 1.2936208429721075e-06,
      "loss": 0.0537,
      "step": 77750
    },
    {
      "epoch": 3.5762786865234375e-06,
      "model_forward_time": 1.7576830387115479,
      "step": 77750
    },
    {
      "dynamics_loss_avg": 0.05333479819819331,
      "epoch": 3.5762786865234375e-06,
      "step": 77750
    },
    {
      "action_loss_avg": 0.0014978381979744882,
      "epoch": 3.5762786865234375e-06,
      "step": 77750
    },
    {
      "epoch": 3.5762786865234375e-06,
      "step": 77750,
      "training_step_time": 8.614193677902222
    },
    {
      "epoch": 3.5772323608398436e-06,
      "model_forward_time": 1.7723870277404785,
      "step": 77751
    },
    {
      "epoch": 3.5772323608398436e-06,
      "step": 77751,
      "training_step_time": 8.691859483718872
    },
    {
      "epoch": 3.57818603515625e-06,
      "model_forward_time": 1.778921365737915,
      "step": 77752
    },
    {
      "epoch": 3.57818603515625e-06,
      "step": 77752,
      "training_step_time": 8.48791241645813
    },
    {
      "epoch": 3.5791397094726564e-06,
      "model_forward_time": 1.7665839195251465,
      "step": 77753
    },
    {
      "epoch": 3.5791397094726564e-06,
      "step": 77753,
      "training_step_time": 7.842200517654419
    },
    {
      "epoch": 3.5800933837890625e-06,
      "model_forward_time": 1.7730672359466553,
      "step": 77754
    },
    {
      "epoch": 3.5800933837890625e-06,
      "step": 77754,
      "training_step_time": 7.985527992248535
    },
    {
      "epoch": 3.5810470581054686e-06,
      "model_forward_time": 1.778221845626831,
      "step": 77755
    },
    {
      "epoch": 3.5810470581054686e-06,
      "step": 77755,
      "training_step_time": 7.375336647033691
    },
    {
      "epoch": 3.582000732421875e-06,
      "model_forward_time": 1.7744088172912598,
      "step": 77756
    },
    {
      "epoch": 3.582000732421875e-06,
      "step": 77756,
      "training_step_time": 8.835918426513672
    },
    {
      "epoch": 3.5829544067382814e-06,
      "model_forward_time": 1.7689366340637207,
      "step": 77757
    },
    {
      "epoch": 3.5829544067382814e-06,
      "step": 77757,
      "training_step_time": 9.161004781723022
    },
    {
      "epoch": 3.5839080810546875e-06,
      "model_forward_time": 2.4046854972839355,
      "step": 77758
    },
    {
      "epoch": 3.5839080810546875e-06,
      "step": 77758,
      "training_step_time": 8.711976289749146
    },
    {
      "epoch": 3.5848617553710937e-06,
      "model_forward_time": 1.7704501152038574,
      "step": 77759
    },
    {
      "epoch": 3.5848617553710937e-06,
      "step": 77759,
      "training_step_time": 9.418108701705933
    },
    {
      "epoch": 3.5858154296875e-06,
      "grad_norm": 0.09529657661914825,
      "learning_rate": 1.2925112358483415e-06,
      "loss": 0.0556,
      "step": 77760
    },
    {
      "epoch": 3.5858154296875e-06,
      "model_forward_time": 1.7606837749481201,
      "step": 77760
    },
    {
      "dynamics_loss_avg": 0.04401363916695118,
      "epoch": 3.5858154296875e-06,
      "step": 77760
    },
    {
      "action_loss_avg": 0.0018942978698760272,
      "epoch": 3.5858154296875e-06,
      "step": 77760
    },
    {
      "epoch": 3.5858154296875e-06,
      "step": 77760,
      "training_step_time": 7.913339138031006
    },
    {
      "epoch": 3.5867691040039064e-06,
      "model_forward_time": 1.7609257698059082,
      "step": 77761
    },
    {
      "epoch": 3.5867691040039064e-06,
      "step": 77761,
      "training_step_time": 7.759668827056885
    },
    {
      "epoch": 3.5877227783203125e-06,
      "model_forward_time": 1.8315701484680176,
      "step": 77762
    },
    {
      "epoch": 3.5877227783203125e-06,
      "step": 77762,
      "training_step_time": 7.817561864852905
    },
    {
      "epoch": 3.5886764526367187e-06,
      "model_forward_time": 1.7720341682434082,
      "step": 77763
    },
    {
      "epoch": 3.5886764526367187e-06,
      "step": 77763,
      "training_step_time": 8.29647421836853
    },
    {
      "epoch": 3.589630126953125e-06,
      "model_forward_time": 1.7571775913238525,
      "step": 77764
    },
    {
      "epoch": 3.589630126953125e-06,
      "step": 77764,
      "training_step_time": 7.299718618392944
    },
    {
      "epoch": 3.5905838012695314e-06,
      "model_forward_time": 1.7576239109039307,
      "step": 77765
    },
    {
      "epoch": 3.5905838012695314e-06,
      "step": 77765,
      "training_step_time": 7.781105041503906
    },
    {
      "epoch": 3.5915374755859375e-06,
      "model_forward_time": 1.7634656429290771,
      "step": 77766
    },
    {
      "epoch": 3.5915374755859375e-06,
      "step": 77766,
      "training_step_time": 8.196725368499756
    },
    {
      "epoch": 3.5924911499023437e-06,
      "model_forward_time": 2.4555742740631104,
      "step": 77767
    },
    {
      "epoch": 3.5924911499023437e-06,
      "step": 77767,
      "training_step_time": 7.6341872215271
    },
    {
      "epoch": 3.59344482421875e-06,
      "model_forward_time": 1.7571308612823486,
      "step": 77768
    },
    {
      "epoch": 3.59344482421875e-06,
      "step": 77768,
      "training_step_time": 8.669426918029785
    },
    {
      "epoch": 3.5943984985351564e-06,
      "model_forward_time": 1.061523199081421,
      "step": 77769
    },
    {
      "epoch": 3.5943984985351564e-06,
      "step": 77769,
      "training_step_time": 7.681232452392578
    },
    {
      "epoch": 3.5953521728515625e-06,
      "grad_norm": 0.09596090018749237,
      "learning_rate": 1.2914020341699728e-06,
      "loss": 0.054,
      "step": 77770
    },
    {
      "epoch": 3.5953521728515625e-06,
      "model_forward_time": 1.7603349685668945,
      "step": 77770
    },
    {
      "dynamics_loss_avg": 0.03906373456120491,
      "epoch": 3.5953521728515625e-06,
      "step": 77770
    },
    {
      "action_loss_avg": 0.002025981480255723,
      "epoch": 3.5953521728515625e-06,
      "step": 77770
    },
    {
      "epoch": 3.5953521728515625e-06,
      "step": 77770,
      "training_step_time": 7.659510374069214
    },
    {
      "epoch": 3.5963058471679687e-06,
      "model_forward_time": 1.8193597793579102,
      "step": 77771
    },
    {
      "epoch": 3.5963058471679687e-06,
      "step": 77771,
      "training_step_time": 7.834382057189941
    },
    {
      "epoch": 3.597259521484375e-06,
      "model_forward_time": 1.7636103630065918,
      "step": 77772
    },
    {
      "epoch": 3.597259521484375e-06,
      "step": 77772,
      "training_step_time": 7.43359112739563
    },
    {
      "epoch": 3.5982131958007814e-06,
      "model_forward_time": 1.765002965927124,
      "step": 77773
    },
    {
      "epoch": 3.5982131958007814e-06,
      "step": 77773,
      "training_step_time": 7.574474334716797
    },
    {
      "epoch": 3.5991668701171875e-06,
      "model_forward_time": 1.7579030990600586,
      "step": 77774
    },
    {
      "epoch": 3.5991668701171875e-06,
      "step": 77774,
      "training_step_time": 7.710623502731323
    },
    {
      "epoch": 3.6001205444335937e-06,
      "model_forward_time": 1.7606496810913086,
      "step": 77775
    },
    {
      "epoch": 3.6001205444335937e-06,
      "step": 77775,
      "training_step_time": 7.580222845077515
    },
    {
      "epoch": 3.60107421875e-06,
      "model_forward_time": 2.419931173324585,
      "step": 77776
    },
    {
      "epoch": 3.60107421875e-06,
      "step": 77776,
      "training_step_time": 8.167575597763062
    },
    {
      "epoch": 3.6020278930664064e-06,
      "model_forward_time": 1.7732689380645752,
      "step": 77777
    },
    {
      "epoch": 3.6020278930664064e-06,
      "step": 77777,
      "training_step_time": 8.353416442871094
    },
    {
      "epoch": 3.6029815673828125e-06,
      "model_forward_time": 1.7666583061218262,
      "step": 77778
    },
    {
      "epoch": 3.6029815673828125e-06,
      "step": 77778,
      "training_step_time": 8.246227979660034
    },
    {
      "epoch": 3.6039352416992187e-06,
      "model_forward_time": 1.4057927131652832,
      "step": 77779
    },
    {
      "epoch": 3.6039352416992187e-06,
      "step": 77779,
      "training_step_time": 7.711928844451904
    },
    {
      "epoch": 3.604888916015625e-06,
      "grad_norm": 0.11083768308162689,
      "learning_rate": 1.2902932380583034e-06,
      "loss": 0.0532,
      "step": 77780
    },
    {
      "epoch": 3.604888916015625e-06,
      "model_forward_time": 1.8144350051879883,
      "step": 77780
    },
    {
      "dynamics_loss_avg": 0.0607242913916707,
      "epoch": 3.604888916015625e-06,
      "step": 77780
    },
    {
      "action_loss_avg": 0.0015399896074086428,
      "epoch": 3.604888916015625e-06,
      "step": 77780
    },
    {
      "epoch": 3.604888916015625e-06,
      "step": 77780,
      "training_step_time": 8.31547212600708
    },
    {
      "epoch": 3.6058425903320314e-06,
      "model_forward_time": 1.7764408588409424,
      "step": 77781
    },
    {
      "epoch": 3.6058425903320314e-06,
      "step": 77781,
      "training_step_time": 7.3620171546936035
    },
    {
      "epoch": 3.6067962646484375e-06,
      "model_forward_time": 1.7639086246490479,
      "step": 77782
    },
    {
      "epoch": 3.6067962646484375e-06,
      "step": 77782,
      "training_step_time": 7.466222763061523
    },
    {
      "epoch": 3.6077499389648437e-06,
      "model_forward_time": 1.764094591140747,
      "step": 77783
    },
    {
      "epoch": 3.6077499389648437e-06,
      "step": 77783,
      "training_step_time": 7.454366683959961
    },
    {
      "epoch": 3.60870361328125e-06,
      "model_forward_time": 1.7646448612213135,
      "step": 77784
    },
    {
      "epoch": 3.60870361328125e-06,
      "step": 77784,
      "training_step_time": 7.6601479053497314
    },
    {
      "epoch": 3.6096572875976564e-06,
      "model_forward_time": 1.7638962268829346,
      "step": 77785
    },
    {
      "epoch": 3.6096572875976564e-06,
      "step": 77785,
      "training_step_time": 7.607347726821899
    },
    {
      "epoch": 3.6106109619140625e-06,
      "model_forward_time": 2.3966448307037354,
      "step": 77786
    },
    {
      "epoch": 3.6106109619140625e-06,
      "step": 77786,
      "training_step_time": 8.594407796859741
    },
    {
      "epoch": 3.6115646362304687e-06,
      "model_forward_time": 1.7571775913238525,
      "step": 77787
    },
    {
      "epoch": 3.6115646362304687e-06,
      "step": 77787,
      "training_step_time": 8.270442485809326
    },
    {
      "epoch": 3.612518310546875e-06,
      "model_forward_time": 1.4017760753631592,
      "step": 77788
    },
    {
      "epoch": 3.612518310546875e-06,
      "step": 77788,
      "training_step_time": 8.355985164642334
    },
    {
      "epoch": 3.6134719848632814e-06,
      "model_forward_time": 1.404932975769043,
      "step": 77789
    },
    {
      "epoch": 3.6134719848632814e-06,
      "step": 77789,
      "training_step_time": 8.06548523902893
    },
    {
      "epoch": 3.6144256591796875e-06,
      "grad_norm": 0.06818879395723343,
      "learning_rate": 1.2891848476345864e-06,
      "loss": 0.0537,
      "step": 77790
    },
    {
      "epoch": 3.6144256591796875e-06,
      "model_forward_time": 1.7638113498687744,
      "step": 77790
    },
    {
      "dynamics_loss_avg": 0.052556426450610164,
      "epoch": 3.6144256591796875e-06,
      "step": 77790
    },
    {
      "action_loss_avg": 0.0008884063368896023,
      "epoch": 3.6144256591796875e-06,
      "step": 77790
    },
    {
      "epoch": 3.6144256591796875e-06,
      "step": 77790,
      "training_step_time": 7.556278228759766
    },
    {
      "epoch": 3.6153793334960937e-06,
      "model_forward_time": 1.7696154117584229,
      "step": 77791
    },
    {
      "epoch": 3.6153793334960937e-06,
      "step": 77791,
      "training_step_time": 7.727067947387695
    },
    {
      "epoch": 3.6163330078125e-06,
      "model_forward_time": 1.7435054779052734,
      "step": 77792
    },
    {
      "epoch": 3.6163330078125e-06,
      "step": 77792,
      "training_step_time": 7.257054805755615
    },
    {
      "epoch": 3.6172866821289064e-06,
      "model_forward_time": 1.747648000717163,
      "step": 77793
    },
    {
      "epoch": 3.6172866821289064e-06,
      "step": 77793,
      "training_step_time": 7.844149589538574
    },
    {
      "epoch": 3.6182403564453125e-06,
      "model_forward_time": 1.7490754127502441,
      "step": 77794
    },
    {
      "epoch": 3.6182403564453125e-06,
      "step": 77794,
      "training_step_time": 7.741478204727173
    },
    {
      "epoch": 3.6191940307617187e-06,
      "model_forward_time": 2.393280267715454,
      "step": 77795
    },
    {
      "epoch": 3.6191940307617187e-06,
      "step": 77795,
      "training_step_time": 9.756214618682861
    },
    {
      "epoch": 3.620147705078125e-06,
      "model_forward_time": 1.7482860088348389,
      "step": 77796
    },
    {
      "epoch": 3.620147705078125e-06,
      "step": 77796,
      "training_step_time": 7.701025009155273
    },
    {
      "epoch": 3.6211013793945314e-06,
      "model_forward_time": 1.7482240200042725,
      "step": 77797
    },
    {
      "epoch": 3.6211013793945314e-06,
      "step": 77797,
      "training_step_time": 7.689010143280029
    },
    {
      "epoch": 3.6220550537109375e-06,
      "model_forward_time": 1.7390153408050537,
      "step": 77798
    },
    {
      "epoch": 3.6220550537109375e-06,
      "step": 77798,
      "training_step_time": 8.551619052886963
    },
    {
      "epoch": 3.6230087280273437e-06,
      "model_forward_time": 1.7467668056488037,
      "step": 77799
    },
    {
      "epoch": 3.6230087280273437e-06,
      "step": 77799,
      "training_step_time": 8.611454963684082
    },
    {
      "epoch": 3.62396240234375e-06,
      "grad_norm": 0.07226872444152832,
      "learning_rate": 1.2880768630200368e-06,
      "loss": 0.0537,
      "step": 77800
    },
    {
      "epoch": 3.62396240234375e-06,
      "model_forward_time": 1.741405963897705,
      "step": 77800
    },
    {
      "dynamics_loss_avg": 0.06501347199082375,
      "epoch": 3.62396240234375e-06,
      "step": 77800
    },
    {
      "action_loss_avg": 0.0015822533750906587,
      "epoch": 3.62396240234375e-06,
      "step": 77800
    },
    {
      "epoch": 3.62396240234375e-06,
      "step": 77800,
      "training_step_time": 7.186218976974487
    },
    {
      "epoch": 3.6249160766601564e-06,
      "model_forward_time": 1.7403168678283691,
      "step": 77801
    },
    {
      "epoch": 3.6249160766601564e-06,
      "step": 77801,
      "training_step_time": 7.143139362335205
    },
    {
      "epoch": 3.6258697509765625e-06,
      "model_forward_time": 1.7495903968811035,
      "step": 77802
    },
    {
      "epoch": 3.6258697509765625e-06,
      "step": 77802,
      "training_step_time": 7.6597795486450195
    },
    {
      "epoch": 3.6268234252929687e-06,
      "model_forward_time": 1.7465908527374268,
      "step": 77803
    },
    {
      "epoch": 3.6268234252929687e-06,
      "step": 77803,
      "training_step_time": 7.6017279624938965
    },
    {
      "epoch": 3.627777099609375e-06,
      "model_forward_time": 2.3805789947509766,
      "step": 77804
    },
    {
      "epoch": 3.627777099609375e-06,
      "step": 77804,
      "training_step_time": 8.549614667892456
    },
    {
      "epoch": 3.6287307739257814e-06,
      "model_forward_time": 1.747260570526123,
      "step": 77805
    },
    {
      "epoch": 3.6287307739257814e-06,
      "step": 77805,
      "training_step_time": 7.801666975021362
    },
    {
      "epoch": 3.6296844482421875e-06,
      "model_forward_time": 1.7537903785705566,
      "step": 77806
    },
    {
      "epoch": 3.6296844482421875e-06,
      "step": 77806,
      "training_step_time": 7.769260883331299
    },
    {
      "epoch": 3.6306381225585937e-06,
      "model_forward_time": 1.7512941360473633,
      "step": 77807
    },
    {
      "epoch": 3.6306381225585937e-06,
      "step": 77807,
      "training_step_time": 8.505404472351074
    },
    {
      "epoch": 3.631591796875e-06,
      "model_forward_time": 1.8008010387420654,
      "step": 77808
    },
    {
      "epoch": 3.631591796875e-06,
      "step": 77808,
      "training_step_time": 8.520678997039795
    },
    {
      "epoch": 3.6325454711914064e-06,
      "model_forward_time": 1.7602744102478027,
      "step": 77809
    },
    {
      "epoch": 3.6325454711914064e-06,
      "step": 77809,
      "training_step_time": 7.447595834732056
    },
    {
      "epoch": 3.6334991455078126e-06,
      "grad_norm": 0.07220055907964706,
      "learning_rate": 1.286969284335819e-06,
      "loss": 0.055,
      "step": 77810
    },
    {
      "epoch": 3.6334991455078126e-06,
      "model_forward_time": 1.7508292198181152,
      "step": 77810
    },
    {
      "dynamics_loss_avg": 0.04855682272464037,
      "epoch": 3.6334991455078126e-06,
      "step": 77810
    },
    {
      "action_loss_avg": 0.00232868755410891,
      "epoch": 3.6334991455078126e-06,
      "step": 77810
    },
    {
      "epoch": 3.6334991455078126e-06,
      "step": 77810,
      "training_step_time": 7.452779769897461
    },
    {
      "epoch": 3.6344528198242187e-06,
      "model_forward_time": 1.7452661991119385,
      "step": 77811
    },
    {
      "epoch": 3.6344528198242187e-06,
      "step": 77811,
      "training_step_time": 7.810354948043823
    },
    {
      "epoch": 3.635406494140625e-06,
      "model_forward_time": 1.7500393390655518,
      "step": 77812
    },
    {
      "epoch": 3.635406494140625e-06,
      "step": 77812,
      "training_step_time": 8.291248083114624
    },
    {
      "epoch": 3.6363601684570314e-06,
      "model_forward_time": 2.3800742626190186,
      "step": 77813
    },
    {
      "epoch": 3.6363601684570314e-06,
      "step": 77813,
      "training_step_time": 8.627426624298096
    },
    {
      "epoch": 3.6373138427734376e-06,
      "model_forward_time": 1.748415231704712,
      "step": 77814
    },
    {
      "epoch": 3.6373138427734376e-06,
      "step": 77814,
      "training_step_time": 8.762803554534912
    },
    {
      "epoch": 3.6382675170898437e-06,
      "model_forward_time": 1.7809815406799316,
      "step": 77815
    },
    {
      "epoch": 3.6382675170898437e-06,
      "step": 77815,
      "training_step_time": 7.795321941375732
    },
    {
      "epoch": 3.63922119140625e-06,
      "model_forward_time": 1.7442216873168945,
      "step": 77816
    },
    {
      "epoch": 3.63922119140625e-06,
      "step": 77816,
      "training_step_time": 9.605504035949707
    },
    {
      "epoch": 3.6401748657226564e-06,
      "model_forward_time": 1.8048481941223145,
      "step": 77817
    },
    {
      "epoch": 3.6401748657226564e-06,
      "step": 77817,
      "training_step_time": 7.869703769683838
    },
    {
      "epoch": 3.6411285400390626e-06,
      "model_forward_time": 1.7351629734039307,
      "step": 77818
    },
    {
      "epoch": 3.6411285400390626e-06,
      "step": 77818,
      "training_step_time": 7.407366037368774
    },
    {
      "epoch": 3.6420822143554687e-06,
      "model_forward_time": 1.7457165718078613,
      "step": 77819
    },
    {
      "epoch": 3.6420822143554687e-06,
      "step": 77819,
      "training_step_time": 7.941469430923462
    },
    {
      "epoch": 3.643035888671875e-06,
      "grad_norm": 0.09318480640649796,
      "learning_rate": 1.2858621117030578e-06,
      "loss": 0.053,
      "step": 77820
    },
    {
      "epoch": 3.643035888671875e-06,
      "model_forward_time": 1.7392547130584717,
      "step": 77820
    },
    {
      "dynamics_loss_avg": 0.04611305668950081,
      "epoch": 3.643035888671875e-06,
      "step": 77820
    },
    {
      "action_loss_avg": 0.0015870997274760156,
      "epoch": 3.643035888671875e-06,
      "step": 77820
    },
    {
      "epoch": 3.643035888671875e-06,
      "step": 77820,
      "training_step_time": 7.510293483734131
    },
    {
      "epoch": 3.6439895629882814e-06,
      "model_forward_time": 1.751713752746582,
      "step": 77821
    },
    {
      "epoch": 3.6439895629882814e-06,
      "step": 77821,
      "training_step_time": 8.507556438446045
    },
    {
      "epoch": 3.6449432373046876e-06,
      "model_forward_time": 2.350036144256592,
      "step": 77822
    },
    {
      "epoch": 3.6449432373046876e-06,
      "step": 77822,
      "training_step_time": 7.617426872253418
    },
    {
      "epoch": 3.6458969116210937e-06,
      "model_forward_time": 1.7454264163970947,
      "step": 77823
    },
    {
      "epoch": 3.6458969116210937e-06,
      "step": 77823,
      "training_step_time": 7.799098014831543
    },
    {
      "epoch": 3.6468505859375e-06,
      "model_forward_time": 1.738943338394165,
      "step": 77824
    },
    {
      "epoch": 3.6468505859375e-06,
      "step": 77824,
      "training_step_time": 8.140649557113647
    },
    {
      "epoch": 3.6478042602539064e-06,
      "model_forward_time": 1.7467761039733887,
      "step": 77825
    },
    {
      "epoch": 3.6478042602539064e-06,
      "step": 77825,
      "training_step_time": 8.702552318572998
    },
    {
      "epoch": 3.6487579345703126e-06,
      "model_forward_time": 1.8172664642333984,
      "step": 77826
    },
    {
      "epoch": 3.6487579345703126e-06,
      "step": 77826,
      "training_step_time": 7.828831911087036
    },
    {
      "epoch": 3.6497116088867187e-06,
      "model_forward_time": 1.7342455387115479,
      "step": 77827
    },
    {
      "epoch": 3.6497116088867187e-06,
      "step": 77827,
      "training_step_time": 7.4693922996521
    },
    {
      "epoch": 3.650665283203125e-06,
      "model_forward_time": 1.738391637802124,
      "step": 77828
    },
    {
      "epoch": 3.650665283203125e-06,
      "step": 77828,
      "training_step_time": 8.482502937316895
    },
    {
      "epoch": 3.6516189575195314e-06,
      "model_forward_time": 1.7435405254364014,
      "step": 77829
    },
    {
      "epoch": 3.6516189575195314e-06,
      "step": 77829,
      "training_step_time": 7.648531675338745
    },
    {
      "epoch": 3.6525726318359376e-06,
      "grad_norm": 0.08012822270393372,
      "learning_rate": 1.2847553452428324e-06,
      "loss": 0.0531,
      "step": 77830
    },
    {
      "epoch": 3.6525726318359376e-06,
      "model_forward_time": 1.7419049739837646,
      "step": 77830
    },
    {
      "dynamics_loss_avg": 0.049242631439119576,
      "epoch": 3.6525726318359376e-06,
      "step": 77830
    },
    {
      "action_loss_avg": 0.0012081326916813851,
      "epoch": 3.6525726318359376e-06,
      "step": 77830
    },
    {
      "epoch": 3.6525726318359376e-06,
      "step": 77830,
      "training_step_time": 7.626177072525024
    },
    {
      "epoch": 3.6535263061523437e-06,
      "model_forward_time": 1.7799131870269775,
      "step": 77831
    },
    {
      "epoch": 3.6535263061523437e-06,
      "step": 77831,
      "training_step_time": 7.5680317878723145
    },
    {
      "epoch": 3.65447998046875e-06,
      "model_forward_time": 2.3979411125183105,
      "step": 77832
    },
    {
      "epoch": 3.65447998046875e-06,
      "step": 77832,
      "training_step_time": 9.090430974960327
    },
    {
      "epoch": 3.6554336547851564e-06,
      "model_forward_time": 1.7698554992675781,
      "step": 77833
    },
    {
      "epoch": 3.6554336547851564e-06,
      "step": 77833,
      "training_step_time": 7.6197075843811035
    },
    {
      "epoch": 3.6563873291015626e-06,
      "model_forward_time": 1.403459072113037,
      "step": 77834
    },
    {
      "epoch": 3.6563873291015626e-06,
      "step": 77834,
      "training_step_time": 9.19254469871521
    },
    {
      "epoch": 3.6573410034179687e-06,
      "model_forward_time": 1.8087663650512695,
      "step": 77835
    },
    {
      "epoch": 3.6573410034179687e-06,
      "step": 77835,
      "training_step_time": 8.521874904632568
    },
    {
      "epoch": 3.658294677734375e-06,
      "model_forward_time": 1.7710292339324951,
      "step": 77836
    },
    {
      "epoch": 3.658294677734375e-06,
      "step": 77836,
      "training_step_time": 7.574061393737793
    },
    {
      "epoch": 3.6592483520507814e-06,
      "model_forward_time": 1.7817857265472412,
      "step": 77837
    },
    {
      "epoch": 3.6592483520507814e-06,
      "step": 77837,
      "training_step_time": 7.420613050460815
    },
    {
      "epoch": 3.6602020263671876e-06,
      "model_forward_time": 1.7671842575073242,
      "step": 77838
    },
    {
      "epoch": 3.6602020263671876e-06,
      "step": 77838,
      "training_step_time": 7.857999324798584
    },
    {
      "epoch": 3.6611557006835937e-06,
      "model_forward_time": 1.7755374908447266,
      "step": 77839
    },
    {
      "epoch": 3.6611557006835937e-06,
      "step": 77839,
      "training_step_time": 8.548016786575317
    },
    {
      "epoch": 3.662109375e-06,
      "grad_norm": 0.0734146386384964,
      "learning_rate": 1.283648985076178e-06,
      "loss": 0.0525,
      "step": 77840
    },
    {
      "epoch": 3.662109375e-06,
      "model_forward_time": 1.7659568786621094,
      "step": 77840
    },
    {
      "dynamics_loss_avg": 0.04105635788291693,
      "epoch": 3.662109375e-06,
      "step": 77840
    },
    {
      "action_loss_avg": 0.0013836423400789498,
      "epoch": 3.662109375e-06,
      "step": 77840
    },
    {
      "epoch": 3.662109375e-06,
      "step": 77840,
      "training_step_time": 9.731351613998413
    },
    {
      "epoch": 3.6630630493164064e-06,
      "model_forward_time": 2.387033462524414,
      "step": 77841
    },
    {
      "epoch": 3.6630630493164064e-06,
      "step": 77841,
      "training_step_time": 9.984683513641357
    },
    {
      "epoch": 3.6640167236328126e-06,
      "model_forward_time": 1.7651410102844238,
      "step": 77842
    },
    {
      "epoch": 3.6640167236328126e-06,
      "step": 77842,
      "training_step_time": 8.640538930892944
    },
    {
      "epoch": 3.6649703979492187e-06,
      "model_forward_time": 1.804678201675415,
      "step": 77843
    },
    {
      "epoch": 3.6649703979492187e-06,
      "step": 77843,
      "training_step_time": 7.871915817260742
    },
    {
      "epoch": 3.665924072265625e-06,
      "model_forward_time": 1.778193712234497,
      "step": 77844
    },
    {
      "epoch": 3.665924072265625e-06,
      "step": 77844,
      "training_step_time": 10.39761209487915
    },
    {
      "epoch": 3.6668777465820314e-06,
      "model_forward_time": 1.771867275238037,
      "step": 77845
    },
    {
      "epoch": 3.6668777465820314e-06,
      "step": 77845,
      "training_step_time": 7.841041564941406
    },
    {
      "epoch": 3.6678314208984376e-06,
      "model_forward_time": 1.7677600383758545,
      "step": 77846
    },
    {
      "epoch": 3.6678314208984376e-06,
      "step": 77846,
      "training_step_time": 7.471714496612549
    },
    {
      "epoch": 3.6687850952148437e-06,
      "model_forward_time": 1.7643325328826904,
      "step": 77847
    },
    {
      "epoch": 3.6687850952148437e-06,
      "step": 77847,
      "training_step_time": 8.421256065368652
    },
    {
      "epoch": 3.66973876953125e-06,
      "model_forward_time": 1.7365102767944336,
      "step": 77848
    },
    {
      "epoch": 3.66973876953125e-06,
      "step": 77848,
      "training_step_time": 7.613399505615234
    },
    {
      "epoch": 3.6706924438476564e-06,
      "model_forward_time": 1.7478222846984863,
      "step": 77849
    },
    {
      "epoch": 3.6706924438476564e-06,
      "step": 77849,
      "training_step_time": 7.639136791229248
    },
    {
      "epoch": 3.6716461181640626e-06,
      "grad_norm": 0.13984858989715576,
      "learning_rate": 1.2825430313240817e-06,
      "loss": 0.0535,
      "step": 77850
    },
    {
      "epoch": 3.6716461181640626e-06,
      "model_forward_time": 2.387579917907715,
      "step": 77850
    },
    {
      "dynamics_loss_avg": 0.05411556102335453,
      "epoch": 3.6716461181640626e-06,
      "step": 77850
    },
    {
      "action_loss_avg": 0.0011793785728514195,
      "epoch": 3.6716461181640626e-06,
      "step": 77850
    },
    {
      "epoch": 3.6716461181640626e-06,
      "step": 77850,
      "training_step_time": 7.642131805419922
    },
    {
      "epoch": 3.6725997924804687e-06,
      "model_forward_time": 1.7429909706115723,
      "step": 77851
    },
    {
      "epoch": 3.6725997924804687e-06,
      "step": 77851,
      "training_step_time": 7.636458158493042
    },
    {
      "epoch": 3.673553466796875e-06,
      "model_forward_time": 1.736138105392456,
      "step": 77852
    },
    {
      "epoch": 3.673553466796875e-06,
      "step": 77852,
      "training_step_time": 8.531128406524658
    },
    {
      "epoch": 3.6745071411132814e-06,
      "model_forward_time": 1.745429515838623,
      "step": 77853
    },
    {
      "epoch": 3.6745071411132814e-06,
      "step": 77853,
      "training_step_time": 9.55630350112915
    },
    {
      "epoch": 3.6754608154296876e-06,
      "model_forward_time": 1.7809438705444336,
      "step": 77854
    },
    {
      "epoch": 3.6754608154296876e-06,
      "step": 77854,
      "training_step_time": 7.756021738052368
    },
    {
      "epoch": 3.6764144897460937e-06,
      "model_forward_time": 1.7403266429901123,
      "step": 77855
    },
    {
      "epoch": 3.6764144897460937e-06,
      "step": 77855,
      "training_step_time": 7.501158952713013
    },
    {
      "epoch": 3.6773681640625e-06,
      "model_forward_time": 1.7568943500518799,
      "step": 77856
    },
    {
      "epoch": 3.6773681640625e-06,
      "step": 77856,
      "training_step_time": 7.24619197845459
    },
    {
      "epoch": 3.6783218383789064e-06,
      "model_forward_time": 1.7806684970855713,
      "step": 77857
    },
    {
      "epoch": 3.6783218383789064e-06,
      "step": 77857,
      "training_step_time": 7.590605735778809
    },
    {
      "epoch": 3.6792755126953126e-06,
      "model_forward_time": 1.74312162399292,
      "step": 77858
    },
    {
      "epoch": 3.6792755126953126e-06,
      "step": 77858,
      "training_step_time": 8.087157726287842
    },
    {
      "epoch": 3.6802291870117187e-06,
      "model_forward_time": 2.3865745067596436,
      "step": 77859
    },
    {
      "epoch": 3.6802291870117187e-06,
      "step": 77859,
      "training_step_time": 10.79043197631836
    },
    {
      "epoch": 3.681182861328125e-06,
      "grad_norm": 0.09113598614931107,
      "learning_rate": 1.2814374841074912e-06,
      "loss": 0.0552,
      "step": 77860
    },
    {
      "epoch": 3.681182861328125e-06,
      "model_forward_time": 1.7447690963745117,
      "step": 77860
    },
    {
      "dynamics_loss_avg": 0.05445367433130741,
      "epoch": 3.681182861328125e-06,
      "step": 77860
    },
    {
      "action_loss_avg": 0.0019211212085792795,
      "epoch": 3.681182861328125e-06,
      "step": 77860
    },
    {
      "epoch": 3.681182861328125e-06,
      "step": 77860,
      "training_step_time": 8.550737380981445
    },
    {
      "epoch": 3.6821365356445314e-06,
      "model_forward_time": 1.7475745677947998,
      "step": 77861
    },
    {
      "epoch": 3.6821365356445314e-06,
      "step": 77861,
      "training_step_time": 8.491442203521729
    },
    {
      "epoch": 3.6830902099609376e-06,
      "model_forward_time": 1.7463858127593994,
      "step": 77862
    },
    {
      "epoch": 3.6830902099609376e-06,
      "step": 77862,
      "training_step_time": 7.578933000564575
    },
    {
      "epoch": 3.6840438842773437e-06,
      "model_forward_time": 1.8111333847045898,
      "step": 77863
    },
    {
      "epoch": 3.6840438842773437e-06,
      "step": 77863,
      "training_step_time": 7.773314476013184
    },
    {
      "epoch": 3.68499755859375e-06,
      "model_forward_time": 1.7530262470245361,
      "step": 77864
    },
    {
      "epoch": 3.68499755859375e-06,
      "step": 77864,
      "training_step_time": 8.790209770202637
    },
    {
      "epoch": 3.6859512329101565e-06,
      "model_forward_time": 1.0530600547790527,
      "step": 77865
    },
    {
      "epoch": 3.6859512329101565e-06,
      "step": 77865,
      "training_step_time": 7.422476530075073
    },
    {
      "epoch": 3.6869049072265626e-06,
      "model_forward_time": 1.7527337074279785,
      "step": 77866
    },
    {
      "epoch": 3.6869049072265626e-06,
      "step": 77866,
      "training_step_time": 7.733718156814575
    },
    {
      "epoch": 3.6878585815429687e-06,
      "model_forward_time": 1.7563071250915527,
      "step": 77867
    },
    {
      "epoch": 3.6878585815429687e-06,
      "step": 77867,
      "training_step_time": 7.903342247009277
    },
    {
      "epoch": 3.688812255859375e-06,
      "model_forward_time": 1.7528924942016602,
      "step": 77868
    },
    {
      "epoch": 3.688812255859375e-06,
      "step": 77868,
      "training_step_time": 8.299419164657593
    },
    {
      "epoch": 3.6897659301757815e-06,
      "model_forward_time": 2.3647658824920654,
      "step": 77869
    },
    {
      "epoch": 3.6897659301757815e-06,
      "step": 77869,
      "training_step_time": 9.743021488189697
    },
    {
      "epoch": 3.6907196044921876e-06,
      "grad_norm": 0.0875263586640358,
      "learning_rate": 1.2803323435473058e-06,
      "loss": 0.0539,
      "step": 77870
    },
    {
      "epoch": 3.6907196044921876e-06,
      "model_forward_time": 1.3827171325683594,
      "step": 77870
    },
    {
      "dynamics_loss_avg": 0.041467562317848206,
      "epoch": 3.6907196044921876e-06,
      "step": 77870
    },
    {
      "action_loss_avg": 0.0010878743953071535,
      "epoch": 3.6907196044921876e-06,
      "step": 77870
    },
    {
      "epoch": 3.6907196044921876e-06,
      "step": 77870,
      "training_step_time": 7.759963512420654
    },
    {
      "epoch": 3.6916732788085937e-06,
      "model_forward_time": 1.7549452781677246,
      "step": 77871
    },
    {
      "epoch": 3.6916732788085937e-06,
      "step": 77871,
      "training_step_time": 8.485248804092407
    },
    {
      "epoch": 3.692626953125e-06,
      "model_forward_time": 1.8179357051849365,
      "step": 77872
    },
    {
      "epoch": 3.692626953125e-06,
      "step": 77872,
      "training_step_time": 8.308882713317871
    },
    {
      "epoch": 3.6935806274414065e-06,
      "model_forward_time": 1.7533667087554932,
      "step": 77873
    },
    {
      "epoch": 3.6935806274414065e-06,
      "step": 77873,
      "training_step_time": 7.793784856796265
    },
    {
      "epoch": 3.6945343017578126e-06,
      "model_forward_time": 1.7552423477172852,
      "step": 77874
    },
    {
      "epoch": 3.6945343017578126e-06,
      "step": 77874,
      "training_step_time": 7.495337247848511
    },
    {
      "epoch": 3.6954879760742188e-06,
      "model_forward_time": 1.758039951324463,
      "step": 77875
    },
    {
      "epoch": 3.6954879760742188e-06,
      "step": 77875,
      "training_step_time": 7.815096139907837
    },
    {
      "epoch": 3.696441650390625e-06,
      "model_forward_time": 1.7451579570770264,
      "step": 77876
    },
    {
      "epoch": 3.696441650390625e-06,
      "step": 77876,
      "training_step_time": 7.890020847320557
    },
    {
      "epoch": 3.697395324707031e-06,
      "model_forward_time": 1.7585067749023438,
      "step": 77877
    },
    {
      "epoch": 3.697395324707031e-06,
      "step": 77877,
      "training_step_time": 10.379257917404175
    },
    {
      "epoch": 3.6983489990234376e-06,
      "model_forward_time": 2.377211093902588,
      "step": 77878
    },
    {
      "epoch": 3.6983489990234376e-06,
      "step": 77878,
      "training_step_time": 7.911174058914185
    },
    {
      "epoch": 3.6993026733398438e-06,
      "model_forward_time": 1.7414000034332275,
      "step": 77879
    },
    {
      "epoch": 3.6993026733398438e-06,
      "step": 77879,
      "training_step_time": 8.35882043838501
    },
    {
      "epoch": 3.70025634765625e-06,
      "grad_norm": 0.07870622724294662,
      "learning_rate": 1.2792276097643842e-06,
      "loss": 0.0541,
      "step": 77880
    },
    {
      "epoch": 3.70025634765625e-06,
      "model_forward_time": 1.7383267879486084,
      "step": 77880
    },
    {
      "dynamics_loss_avg": 0.05046053733676672,
      "epoch": 3.70025634765625e-06,
      "step": 77880
    },
    {
      "action_loss_avg": 0.0008624773850897327,
      "epoch": 3.70025634765625e-06,
      "step": 77880
    },
    {
      "epoch": 3.70025634765625e-06,
      "step": 77880,
      "training_step_time": 9.525330305099487
    },
    {
      "epoch": 3.701210021972656e-06,
      "model_forward_time": 1.8088781833648682,
      "step": 77881
    },
    {
      "epoch": 3.701210021972656e-06,
      "step": 77881,
      "training_step_time": 10.31850790977478
    },
    {
      "epoch": 3.7021636962890626e-06,
      "model_forward_time": 1.7400875091552734,
      "step": 77882
    },
    {
      "epoch": 3.7021636962890626e-06,
      "step": 77882,
      "training_step_time": 7.733762979507446
    },
    {
      "epoch": 3.7031173706054688e-06,
      "model_forward_time": 1.7402591705322266,
      "step": 77883
    },
    {
      "epoch": 3.7031173706054688e-06,
      "step": 77883,
      "training_step_time": 7.461604118347168
    },
    {
      "epoch": 3.704071044921875e-06,
      "model_forward_time": 1.7481462955474854,
      "step": 77884
    },
    {
      "epoch": 3.704071044921875e-06,
      "step": 77884,
      "training_step_time": 7.6599485874176025
    },
    {
      "epoch": 3.705024719238281e-06,
      "model_forward_time": 1.7485363483428955,
      "step": 77885
    },
    {
      "epoch": 3.705024719238281e-06,
      "step": 77885,
      "training_step_time": 7.771159648895264
    },
    {
      "epoch": 3.7059783935546876e-06,
      "model_forward_time": 1.746577262878418,
      "step": 77886
    },
    {
      "epoch": 3.7059783935546876e-06,
      "step": 77886,
      "training_step_time": 7.8773417472839355
    },
    {
      "epoch": 3.7069320678710938e-06,
      "model_forward_time": 2.393799066543579,
      "step": 77887
    },
    {
      "epoch": 3.7069320678710938e-06,
      "step": 77887,
      "training_step_time": 8.959043502807617
    },
    {
      "epoch": 3.7078857421875e-06,
      "model_forward_time": 1.7428829669952393,
      "step": 77888
    },
    {
      "epoch": 3.7078857421875e-06,
      "step": 77888,
      "training_step_time": 9.153955459594727
    },
    {
      "epoch": 3.708839416503906e-06,
      "model_forward_time": 1.7432599067687988,
      "step": 77889
    },
    {
      "epoch": 3.708839416503906e-06,
      "step": 77889,
      "training_step_time": 7.860420227050781
    },
    {
      "epoch": 3.7097930908203126e-06,
      "grad_norm": 0.0771259292960167,
      "learning_rate": 1.2781232828795343e-06,
      "loss": 0.0561,
      "step": 77890
    },
    {
      "epoch": 3.7097930908203126e-06,
      "model_forward_time": 1.777282953262329,
      "step": 77890
    },
    {
      "dynamics_loss_avg": 0.05160847660154104,
      "epoch": 3.7097930908203126e-06,
      "step": 77890
    },
    {
      "action_loss_avg": 0.0007333164830924943,
      "epoch": 3.7097930908203126e-06,
      "step": 77890
    },
    {
      "epoch": 3.7097930908203126e-06,
      "step": 77890,
      "training_step_time": 9.180006980895996
    },
    {
      "epoch": 3.7107467651367188e-06,
      "model_forward_time": 1.7491636276245117,
      "step": 77891
    },
    {
      "epoch": 3.7107467651367188e-06,
      "step": 77891,
      "training_step_time": 7.802984714508057
    },
    {
      "epoch": 3.711700439453125e-06,
      "model_forward_time": 1.7532949447631836,
      "step": 77892
    },
    {
      "epoch": 3.711700439453125e-06,
      "step": 77892,
      "training_step_time": 7.461414098739624
    },
    {
      "epoch": 3.712654113769531e-06,
      "model_forward_time": 1.74702787399292,
      "step": 77893
    },
    {
      "epoch": 3.712654113769531e-06,
      "step": 77893,
      "training_step_time": 7.296248435974121
    },
    {
      "epoch": 3.7136077880859376e-06,
      "model_forward_time": 1.7465758323669434,
      "step": 77894
    },
    {
      "epoch": 3.7136077880859376e-06,
      "step": 77894,
      "training_step_time": 7.692314147949219
    },
    {
      "epoch": 3.7145614624023438e-06,
      "model_forward_time": 1.7543613910675049,
      "step": 77895
    },
    {
      "epoch": 3.7145614624023438e-06,
      "step": 77895,
      "training_step_time": 8.680664777755737
    },
    {
      "epoch": 3.71551513671875e-06,
      "model_forward_time": 2.389893054962158,
      "step": 77896
    },
    {
      "epoch": 3.71551513671875e-06,
      "step": 77896,
      "training_step_time": 9.18326473236084
    },
    {
      "epoch": 3.716468811035156e-06,
      "model_forward_time": 1.7428674697875977,
      "step": 77897
    },
    {
      "epoch": 3.716468811035156e-06,
      "step": 77897,
      "training_step_time": 8.573847532272339
    },
    {
      "epoch": 3.7174224853515626e-06,
      "model_forward_time": 1.7444748878479004,
      "step": 77898
    },
    {
      "epoch": 3.7174224853515626e-06,
      "step": 77898,
      "training_step_time": 7.652734041213989
    },
    {
      "epoch": 3.7183761596679688e-06,
      "model_forward_time": 1.7488987445831299,
      "step": 77899
    },
    {
      "epoch": 3.7183761596679688e-06,
      "step": 77899,
      "training_step_time": 8.403645277023315
    },
    {
      "epoch": 3.719329833984375e-06,
      "grad_norm": 0.08418887853622437,
      "learning_rate": 1.277019363013529e-06,
      "loss": 0.0559,
      "step": 77900
    },
    {
      "epoch": 3.719329833984375e-06,
      "model_forward_time": 1.8496108055114746,
      "step": 77900
    },
    {
      "dynamics_loss_avg": 0.054324017092585566,
      "epoch": 3.719329833984375e-06,
      "step": 77900
    },
    {
      "action_loss_avg": 0.0014515837654471397,
      "epoch": 3.719329833984375e-06,
      "step": 77900
    },
    {
      "epoch": 3.719329833984375e-06,
      "step": 77900,
      "training_step_time": 7.756850957870483
    },
    {
      "epoch": 3.720283508300781e-06,
      "model_forward_time": 1.8121066093444824,
      "step": 77901
    },
    {
      "epoch": 3.720283508300781e-06,
      "step": 77901,
      "training_step_time": 7.845722913742065
    },
    {
      "epoch": 3.7212371826171876e-06,
      "model_forward_time": 1.7726011276245117,
      "step": 77902
    },
    {
      "epoch": 3.7212371826171876e-06,
      "step": 77902,
      "training_step_time": 7.216832637786865
    },
    {
      "epoch": 3.7221908569335938e-06,
      "model_forward_time": 1.7636845111846924,
      "step": 77903
    },
    {
      "epoch": 3.7221908569335938e-06,
      "step": 77903,
      "training_step_time": 7.730055332183838
    },
    {
      "epoch": 3.72314453125e-06,
      "model_forward_time": 1.7689487934112549,
      "step": 77904
    },
    {
      "epoch": 3.72314453125e-06,
      "step": 77904,
      "training_step_time": 7.653902769088745
    },
    {
      "epoch": 3.724098205566406e-06,
      "model_forward_time": 1.7744085788726807,
      "step": 77905
    },
    {
      "epoch": 3.724098205566406e-06,
      "step": 77905,
      "training_step_time": 9.375070810317993
    },
    {
      "epoch": 3.7250518798828126e-06,
      "model_forward_time": 1.772718906402588,
      "step": 77906
    },
    {
      "epoch": 3.7250518798828126e-06,
      "step": 77906,
      "training_step_time": 9.466098546981812
    },
    {
      "epoch": 3.7260055541992188e-06,
      "model_forward_time": 1.7704689502716064,
      "step": 77907
    },
    {
      "epoch": 3.7260055541992188e-06,
      "step": 77907,
      "training_step_time": 9.357803344726562
    },
    {
      "epoch": 3.726959228515625e-06,
      "model_forward_time": 1.770477056503296,
      "step": 77908
    },
    {
      "epoch": 3.726959228515625e-06,
      "step": 77908,
      "training_step_time": 7.563167333602905
    },
    {
      "epoch": 3.727912902832031e-06,
      "model_forward_time": 1.7941391468048096,
      "step": 77909
    },
    {
      "epoch": 3.727912902832031e-06,
      "step": 77909,
      "training_step_time": 7.668888092041016
    },
    {
      "epoch": 3.7288665771484376e-06,
      "grad_norm": 0.08131759613752365,
      "learning_rate": 1.2759158502870877e-06,
      "loss": 0.0546,
      "step": 77910
    },
    {
      "epoch": 3.7288665771484376e-06,
      "model_forward_time": 1.7666709423065186,
      "step": 77910
    },
    {
      "dynamics_loss_avg": 0.0330367224290967,
      "epoch": 3.7288665771484376e-06,
      "step": 77910
    },
    {
      "action_loss_avg": 0.0009365833102492616,
      "epoch": 3.7288665771484376e-06,
      "step": 77910
    },
    {
      "epoch": 3.7288665771484376e-06,
      "step": 77910,
      "training_step_time": 7.2131993770599365
    },
    {
      "epoch": 3.7298202514648438e-06,
      "model_forward_time": 1.7663474082946777,
      "step": 77911
    },
    {
      "epoch": 3.7298202514648438e-06,
      "step": 77911,
      "training_step_time": 7.827290058135986
    },
    {
      "epoch": 3.73077392578125e-06,
      "model_forward_time": 1.7656564712524414,
      "step": 77912
    },
    {
      "epoch": 3.73077392578125e-06,
      "step": 77912,
      "training_step_time": 9.14239239692688
    },
    {
      "epoch": 3.731727600097656e-06,
      "model_forward_time": 1.772646188735962,
      "step": 77913
    },
    {
      "epoch": 3.731727600097656e-06,
      "step": 77913,
      "training_step_time": 8.64441967010498
    },
    {
      "epoch": 3.7326812744140626e-06,
      "model_forward_time": 1.7645998001098633,
      "step": 77914
    },
    {
      "epoch": 3.7326812744140626e-06,
      "step": 77914,
      "training_step_time": 7.69757342338562
    },
    {
      "epoch": 3.7336349487304688e-06,
      "model_forward_time": 2.409257173538208,
      "step": 77915
    },
    {
      "epoch": 3.7336349487304688e-06,
      "step": 77915,
      "training_step_time": 8.545297622680664
    },
    {
      "epoch": 3.734588623046875e-06,
      "model_forward_time": 1.7612056732177734,
      "step": 77916
    },
    {
      "epoch": 3.734588623046875e-06,
      "step": 77916,
      "training_step_time": 9.382354259490967
    },
    {
      "epoch": 3.735542297363281e-06,
      "model_forward_time": 1.7608811855316162,
      "step": 77917
    },
    {
      "epoch": 3.735542297363281e-06,
      "step": 77917,
      "training_step_time": 7.632577896118164
    },
    {
      "epoch": 3.7364959716796876e-06,
      "model_forward_time": 1.776306390762329,
      "step": 77918
    },
    {
      "epoch": 3.7364959716796876e-06,
      "step": 77918,
      "training_step_time": 7.80583930015564
    },
    {
      "epoch": 3.7374496459960938e-06,
      "model_forward_time": 1.765578269958496,
      "step": 77919
    },
    {
      "epoch": 3.7374496459960938e-06,
      "step": 77919,
      "training_step_time": 7.445878028869629
    },
    {
      "epoch": 3.7384033203125e-06,
      "grad_norm": 0.16888907551765442,
      "learning_rate": 1.2748127448208913e-06,
      "loss": 0.0569,
      "step": 77920
    },
    {
      "epoch": 3.7384033203125e-06,
      "model_forward_time": 1.7711668014526367,
      "step": 77920
    },
    {
      "dynamics_loss_avg": 0.03977508801035583,
      "epoch": 3.7384033203125e-06,
      "step": 77920
    },
    {
      "action_loss_avg": 0.0009369905485073105,
      "epoch": 3.7384033203125e-06,
      "step": 77920
    },
    {
      "epoch": 3.7384033203125e-06,
      "step": 77920,
      "training_step_time": 7.949068784713745
    },
    {
      "epoch": 3.739356994628906e-06,
      "model_forward_time": 1.772869348526001,
      "step": 77921
    },
    {
      "epoch": 3.739356994628906e-06,
      "step": 77921,
      "training_step_time": 7.567147970199585
    },
    {
      "epoch": 3.7403106689453126e-06,
      "model_forward_time": 1.7716870307922363,
      "step": 77922
    },
    {
      "epoch": 3.7403106689453126e-06,
      "step": 77922,
      "training_step_time": 7.657526969909668
    },
    {
      "epoch": 3.741264343261719e-06,
      "model_forward_time": 1.764253854751587,
      "step": 77923
    },
    {
      "epoch": 3.741264343261719e-06,
      "step": 77923,
      "training_step_time": 7.885005950927734
    },
    {
      "epoch": 3.742218017578125e-06,
      "model_forward_time": 2.396308422088623,
      "step": 77924
    },
    {
      "epoch": 3.742218017578125e-06,
      "step": 77924,
      "training_step_time": 7.763917922973633
    },
    {
      "epoch": 3.743171691894531e-06,
      "model_forward_time": 1.769784688949585,
      "step": 77925
    },
    {
      "epoch": 3.743171691894531e-06,
      "step": 77925,
      "training_step_time": 8.27539849281311
    },
    {
      "epoch": 3.7441253662109376e-06,
      "model_forward_time": 1.765665054321289,
      "step": 77926
    },
    {
      "epoch": 3.7441253662109376e-06,
      "step": 77926,
      "training_step_time": 7.574031352996826
    },
    {
      "epoch": 3.745079040527344e-06,
      "model_forward_time": 1.7689180374145508,
      "step": 77927
    },
    {
      "epoch": 3.745079040527344e-06,
      "step": 77927,
      "training_step_time": 8.555341958999634
    },
    {
      "epoch": 3.74603271484375e-06,
      "model_forward_time": 1.825836420059204,
      "step": 77928
    },
    {
      "epoch": 3.74603271484375e-06,
      "step": 77928,
      "training_step_time": 8.345290184020996
    },
    {
      "epoch": 3.746986389160156e-06,
      "model_forward_time": 1.7680299282073975,
      "step": 77929
    },
    {
      "epoch": 3.746986389160156e-06,
      "step": 77929,
      "training_step_time": 7.492904186248779
    },
    {
      "epoch": 3.7479400634765627e-06,
      "grad_norm": 0.10981376469135284,
      "learning_rate": 1.2737100467355706e-06,
      "loss": 0.0555,
      "step": 77930
    },
    {
      "epoch": 3.7479400634765627e-06,
      "model_forward_time": 1.7664079666137695,
      "step": 77930
    },
    {
      "dynamics_loss_avg": 0.04453814346343279,
      "epoch": 3.7479400634765627e-06,
      "step": 77930
    },
    {
      "action_loss_avg": 0.0009475029452005401,
      "epoch": 3.7479400634765627e-06,
      "step": 77930
    },
    {
      "epoch": 3.7479400634765627e-06,
      "step": 77930,
      "training_step_time": 8.568316221237183
    },
    {
      "epoch": 3.748893737792969e-06,
      "model_forward_time": 1.7696774005889893,
      "step": 77931
    },
    {
      "epoch": 3.748893737792969e-06,
      "step": 77931,
      "training_step_time": 7.516563892364502
    },
    {
      "epoch": 3.749847412109375e-06,
      "model_forward_time": 1.7302682399749756,
      "step": 77932
    },
    {
      "epoch": 3.749847412109375e-06,
      "step": 77932,
      "training_step_time": 7.799441337585449
    },
    {
      "epoch": 3.750801086425781e-06,
      "model_forward_time": 2.357335329055786,
      "step": 77933
    },
    {
      "epoch": 3.750801086425781e-06,
      "step": 77933,
      "training_step_time": 9.355124235153198
    },
    {
      "epoch": 3.7517547607421877e-06,
      "model_forward_time": 1.7794029712677002,
      "step": 77934
    },
    {
      "epoch": 3.7517547607421877e-06,
      "step": 77934,
      "training_step_time": 7.651290416717529
    },
    {
      "epoch": 3.752708435058594e-06,
      "model_forward_time": 1.7420270442962646,
      "step": 77935
    },
    {
      "epoch": 3.752708435058594e-06,
      "step": 77935,
      "training_step_time": 7.718207597732544
    },
    {
      "epoch": 3.753662109375e-06,
      "model_forward_time": 1.7470455169677734,
      "step": 77936
    },
    {
      "epoch": 3.753662109375e-06,
      "step": 77936,
      "training_step_time": 9.129647254943848
    },
    {
      "epoch": 3.754615783691406e-06,
      "model_forward_time": 1.7991359233856201,
      "step": 77937
    },
    {
      "epoch": 3.754615783691406e-06,
      "step": 77937,
      "training_step_time": 7.536943435668945
    },
    {
      "epoch": 3.7555694580078127e-06,
      "model_forward_time": 1.748769760131836,
      "step": 77938
    },
    {
      "epoch": 3.7555694580078127e-06,
      "step": 77938,
      "training_step_time": 7.50018310546875
    },
    {
      "epoch": 3.756523132324219e-06,
      "model_forward_time": 1.7515933513641357,
      "step": 77939
    },
    {
      "epoch": 3.756523132324219e-06,
      "step": 77939,
      "training_step_time": 8.634530782699585
    },
    {
      "epoch": 3.757476806640625e-06,
      "grad_norm": 0.07518331706523895,
      "learning_rate": 1.272607756151717e-06,
      "loss": 0.0577,
      "step": 77940
    },
    {
      "epoch": 3.757476806640625e-06,
      "model_forward_time": 1.7592425346374512,
      "step": 77940
    },
    {
      "dynamics_loss_avg": 0.04672315418720245,
      "epoch": 3.757476806640625e-06,
      "step": 77940
    },
    {
      "action_loss_avg": 0.0011956918839132412,
      "epoch": 3.757476806640625e-06,
      "step": 77940
    },
    {
      "epoch": 3.757476806640625e-06,
      "step": 77940,
      "training_step_time": 7.595070123672485
    },
    {
      "epoch": 3.758430480957031e-06,
      "model_forward_time": 1.744419813156128,
      "step": 77941
    },
    {
      "epoch": 3.758430480957031e-06,
      "step": 77941,
      "training_step_time": 8.282676696777344
    },
    {
      "epoch": 3.7593841552734377e-06,
      "model_forward_time": 2.41406512260437,
      "step": 77942
    },
    {
      "epoch": 3.7593841552734377e-06,
      "step": 77942,
      "training_step_time": 7.662088394165039
    },
    {
      "epoch": 3.760337829589844e-06,
      "model_forward_time": 1.7455005645751953,
      "step": 77943
    },
    {
      "epoch": 3.760337829589844e-06,
      "step": 77943,
      "training_step_time": 7.720947742462158
    },
    {
      "epoch": 3.76129150390625e-06,
      "model_forward_time": 1.7539801597595215,
      "step": 77944
    },
    {
      "epoch": 3.76129150390625e-06,
      "step": 77944,
      "training_step_time": 7.800488471984863
    },
    {
      "epoch": 3.762245178222656e-06,
      "model_forward_time": 1.7426574230194092,
      "step": 77945
    },
    {
      "epoch": 3.762245178222656e-06,
      "step": 77945,
      "training_step_time": 11.320905208587646
    },
    {
      "epoch": 3.7631988525390627e-06,
      "model_forward_time": 1.8031742572784424,
      "step": 77946
    },
    {
      "epoch": 3.7631988525390627e-06,
      "step": 77946,
      "training_step_time": 7.692701578140259
    },
    {
      "epoch": 3.764152526855469e-06,
      "model_forward_time": 1.7508556842803955,
      "step": 77947
    },
    {
      "epoch": 3.764152526855469e-06,
      "step": 77947,
      "training_step_time": 7.648342609405518
    },
    {
      "epoch": 3.765106201171875e-06,
      "model_forward_time": 1.748901128768921,
      "step": 77948
    },
    {
      "epoch": 3.765106201171875e-06,
      "step": 77948,
      "training_step_time": 8.74927568435669
    },
    {
      "epoch": 3.766059875488281e-06,
      "model_forward_time": 1.739591121673584,
      "step": 77949
    },
    {
      "epoch": 3.766059875488281e-06,
      "step": 77949,
      "training_step_time": 7.668421268463135
    },
    {
      "epoch": 3.7670135498046877e-06,
      "grad_norm": 0.10953100025653839,
      "learning_rate": 1.2715058731898756e-06,
      "loss": 0.0551,
      "step": 77950
    },
    {
      "epoch": 3.7670135498046877e-06,
      "model_forward_time": 1.7491097450256348,
      "step": 77950
    },
    {
      "dynamics_loss_avg": 0.06437421906739474,
      "epoch": 3.7670135498046877e-06,
      "step": 77950
    },
    {
      "action_loss_avg": 0.0016118606785312296,
      "epoch": 3.7670135498046877e-06,
      "step": 77950
    },
    {
      "epoch": 3.7670135498046877e-06,
      "step": 77950,
      "training_step_time": 7.776614189147949
    },
    {
      "epoch": 3.767967224121094e-06,
      "model_forward_time": 2.3818278312683105,
      "step": 77951
    },
    {
      "epoch": 3.767967224121094e-06,
      "step": 77951,
      "training_step_time": 8.380059719085693
    },
    {
      "epoch": 3.7689208984375e-06,
      "model_forward_time": 1.7472114562988281,
      "step": 77952
    },
    {
      "epoch": 3.7689208984375e-06,
      "step": 77952,
      "training_step_time": 8.68704891204834
    },
    {
      "epoch": 3.769874572753906e-06,
      "model_forward_time": 1.751354455947876,
      "step": 77953
    },
    {
      "epoch": 3.769874572753906e-06,
      "step": 77953,
      "training_step_time": 8.751008749008179
    },
    {
      "epoch": 3.7708282470703127e-06,
      "model_forward_time": 1.739405632019043,
      "step": 77954
    },
    {
      "epoch": 3.7708282470703127e-06,
      "step": 77954,
      "training_step_time": 7.896538019180298
    },
    {
      "epoch": 3.771781921386719e-06,
      "model_forward_time": 1.782932996749878,
      "step": 77955
    },
    {
      "epoch": 3.771781921386719e-06,
      "step": 77955,
      "training_step_time": 7.829362392425537
    },
    {
      "epoch": 3.772735595703125e-06,
      "model_forward_time": 1.753493309020996,
      "step": 77956
    },
    {
      "epoch": 3.772735595703125e-06,
      "step": 77956,
      "training_step_time": 7.665332317352295
    },
    {
      "epoch": 3.773689270019531e-06,
      "model_forward_time": 1.7423577308654785,
      "step": 77957
    },
    {
      "epoch": 3.773689270019531e-06,
      "step": 77957,
      "training_step_time": 7.8059163093566895
    },
    {
      "epoch": 3.7746429443359377e-06,
      "model_forward_time": 1.379901647567749,
      "step": 77958
    },
    {
      "epoch": 3.7746429443359377e-06,
      "step": 77958,
      "training_step_time": 8.558467864990234
    },
    {
      "epoch": 3.775596618652344e-06,
      "model_forward_time": 1.7939906120300293,
      "step": 77959
    },
    {
      "epoch": 3.775596618652344e-06,
      "step": 77959,
      "training_step_time": 7.883653402328491
    },
    {
      "epoch": 3.77655029296875e-06,
      "grad_norm": 0.10966094583272934,
      "learning_rate": 1.2704043979705472e-06,
      "loss": 0.0559,
      "step": 77960
    },
    {
      "epoch": 3.77655029296875e-06,
      "model_forward_time": 1.7405097484588623,
      "step": 77960
    },
    {
      "dynamics_loss_avg": 0.041949070151895286,
      "epoch": 3.77655029296875e-06,
      "step": 77960
    },
    {
      "action_loss_avg": 0.0014018363610375673,
      "epoch": 3.77655029296875e-06,
      "step": 77960
    },
    {
      "epoch": 3.77655029296875e-06,
      "step": 77960,
      "training_step_time": 8.639731645584106
    },
    {
      "epoch": 3.777503967285156e-06,
      "model_forward_time": 2.359915256500244,
      "step": 77961
    },
    {
      "epoch": 3.777503967285156e-06,
      "step": 77961,
      "training_step_time": 8.619598150253296
    },
    {
      "epoch": 3.7784576416015627e-06,
      "model_forward_time": 1.7641654014587402,
      "step": 77962
    },
    {
      "epoch": 3.7784576416015627e-06,
      "step": 77962,
      "training_step_time": 8.827846765518188
    },
    {
      "epoch": 3.779411315917969e-06,
      "model_forward_time": 1.8077833652496338,
      "step": 77963
    },
    {
      "epoch": 3.779411315917969e-06,
      "step": 77963,
      "training_step_time": 8.499305725097656
    },
    {
      "epoch": 3.780364990234375e-06,
      "model_forward_time": 1.7698338031768799,
      "step": 77964
    },
    {
      "epoch": 3.780364990234375e-06,
      "step": 77964,
      "training_step_time": 7.771660566329956
    },
    {
      "epoch": 3.781318664550781e-06,
      "model_forward_time": 1.817469596862793,
      "step": 77965
    },
    {
      "epoch": 3.781318664550781e-06,
      "step": 77965,
      "training_step_time": 7.81400990486145
    },
    {
      "epoch": 3.7822723388671877e-06,
      "model_forward_time": 1.7630195617675781,
      "step": 77966
    },
    {
      "epoch": 3.7822723388671877e-06,
      "step": 77966,
      "training_step_time": 8.390440702438354
    },
    {
      "epoch": 3.783226013183594e-06,
      "model_forward_time": 1.7765042781829834,
      "step": 77967
    },
    {
      "epoch": 3.783226013183594e-06,
      "step": 77967,
      "training_step_time": 8.730394124984741
    },
    {
      "epoch": 3.7841796875e-06,
      "model_forward_time": 1.7702968120574951,
      "step": 77968
    },
    {
      "epoch": 3.7841796875e-06,
      "step": 77968,
      "training_step_time": 8.388942003250122
    },
    {
      "epoch": 3.785133361816406e-06,
      "model_forward_time": 1.7656497955322266,
      "step": 77969
    },
    {
      "epoch": 3.785133361816406e-06,
      "step": 77969,
      "training_step_time": 8.741560220718384
    },
    {
      "epoch": 3.7860870361328127e-06,
      "grad_norm": 0.10902424156665802,
      "learning_rate": 1.269303330614185e-06,
      "loss": 0.0562,
      "step": 77970
    },
    {
      "epoch": 3.7860870361328127e-06,
      "model_forward_time": 2.4017558097839355,
      "step": 77970
    },
    {
      "dynamics_loss_avg": 0.06344123855233193,
      "epoch": 3.7860870361328127e-06,
      "step": 77970
    },
    {
      "action_loss_avg": 0.001862521842122078,
      "epoch": 3.7860870361328127e-06,
      "step": 77970
    },
    {
      "epoch": 3.7860870361328127e-06,
      "step": 77970,
      "training_step_time": 8.690776824951172
    },
    {
      "epoch": 3.787040710449219e-06,
      "model_forward_time": 1.7638444900512695,
      "step": 77971
    },
    {
      "epoch": 3.787040710449219e-06,
      "step": 77971,
      "training_step_time": 9.280227184295654
    },
    {
      "epoch": 3.787994384765625e-06,
      "model_forward_time": 1.767747402191162,
      "step": 77972
    },
    {
      "epoch": 3.787994384765625e-06,
      "step": 77972,
      "training_step_time": 8.392749309539795
    },
    {
      "epoch": 3.788948059082031e-06,
      "model_forward_time": 1.764911413192749,
      "step": 77973
    },
    {
      "epoch": 3.788948059082031e-06,
      "step": 77973,
      "training_step_time": 10.62299656867981
    },
    {
      "epoch": 3.7899017333984377e-06,
      "model_forward_time": 1.8186562061309814,
      "step": 77974
    },
    {
      "epoch": 3.7899017333984377e-06,
      "step": 77974,
      "training_step_time": 7.751564979553223
    },
    {
      "epoch": 3.790855407714844e-06,
      "model_forward_time": 1.7662901878356934,
      "step": 77975
    },
    {
      "epoch": 3.790855407714844e-06,
      "step": 77975,
      "training_step_time": 7.83318305015564
    },
    {
      "epoch": 3.79180908203125e-06,
      "model_forward_time": 1.779266595840454,
      "step": 77976
    },
    {
      "epoch": 3.79180908203125e-06,
      "step": 77976,
      "training_step_time": 7.869816541671753
    },
    {
      "epoch": 3.792762756347656e-06,
      "model_forward_time": 1.7650816440582275,
      "step": 77977
    },
    {
      "epoch": 3.792762756347656e-06,
      "step": 77977,
      "training_step_time": 7.808313369750977
    },
    {
      "epoch": 3.7937164306640627e-06,
      "model_forward_time": 1.7529094219207764,
      "step": 77978
    },
    {
      "epoch": 3.7937164306640627e-06,
      "step": 77978,
      "training_step_time": 8.890163660049438
    },
    {
      "epoch": 3.794670104980469e-06,
      "model_forward_time": 2.4137918949127197,
      "step": 77979
    },
    {
      "epoch": 3.794670104980469e-06,
      "step": 77979,
      "training_step_time": 7.971073150634766
    },
    {
      "epoch": 3.795623779296875e-06,
      "grad_norm": 0.07208361476659775,
      "learning_rate": 1.2682026712412016e-06,
      "loss": 0.0572,
      "step": 77980
    },
    {
      "epoch": 3.795623779296875e-06,
      "model_forward_time": 1.7622110843658447,
      "step": 77980
    },
    {
      "dynamics_loss_avg": 0.05711114164441824,
      "epoch": 3.795623779296875e-06,
      "step": 77980
    },
    {
      "action_loss_avg": 0.0012992616626434027,
      "epoch": 3.795623779296875e-06,
      "step": 77980
    },
    {
      "epoch": 3.795623779296875e-06,
      "step": 77980,
      "training_step_time": 7.891193151473999
    },
    {
      "epoch": 3.796577453613281e-06,
      "model_forward_time": 1.7649941444396973,
      "step": 77981
    },
    {
      "epoch": 3.796577453613281e-06,
      "step": 77981,
      "training_step_time": 9.304216623306274
    },
    {
      "epoch": 3.7975311279296877e-06,
      "model_forward_time": 1.7657489776611328,
      "step": 77982
    },
    {
      "epoch": 3.7975311279296877e-06,
      "step": 77982,
      "training_step_time": 7.821057319641113
    },
    {
      "epoch": 3.798484802246094e-06,
      "model_forward_time": 1.8276066780090332,
      "step": 77983
    },
    {
      "epoch": 3.798484802246094e-06,
      "step": 77983,
      "training_step_time": 8.306915760040283
    },
    {
      "epoch": 3.7994384765625e-06,
      "model_forward_time": 1.4108290672302246,
      "step": 77984
    },
    {
      "epoch": 3.7994384765625e-06,
      "step": 77984,
      "training_step_time": 7.75761342048645
    },
    {
      "epoch": 3.800392150878906e-06,
      "model_forward_time": 1.7661595344543457,
      "step": 77985
    },
    {
      "epoch": 3.800392150878906e-06,
      "step": 77985,
      "training_step_time": 8.62134075164795
    },
    {
      "epoch": 3.8013458251953127e-06,
      "model_forward_time": 1.7649803161621094,
      "step": 77986
    },
    {
      "epoch": 3.8013458251953127e-06,
      "step": 77986,
      "training_step_time": 7.733560800552368
    },
    {
      "epoch": 3.802299499511719e-06,
      "model_forward_time": 1.7881169319152832,
      "step": 77987
    },
    {
      "epoch": 3.802299499511719e-06,
      "step": 77987,
      "training_step_time": 8.324982166290283
    },
    {
      "epoch": 3.803253173828125e-06,
      "model_forward_time": 2.4170825481414795,
      "step": 77988
    },
    {
      "epoch": 3.803253173828125e-06,
      "step": 77988,
      "training_step_time": 8.47274374961853
    },
    {
      "epoch": 3.804206848144531e-06,
      "model_forward_time": 1.7652642726898193,
      "step": 77989
    },
    {
      "epoch": 3.804206848144531e-06,
      "step": 77989,
      "training_step_time": 8.727116584777832
    },
    {
      "epoch": 3.8051605224609377e-06,
      "grad_norm": 0.08595152944326401,
      "learning_rate": 1.2671024199719645e-06,
      "loss": 0.0569,
      "step": 77990
    },
    {
      "epoch": 3.8051605224609377e-06,
      "model_forward_time": 1.7688515186309814,
      "step": 77990
    },
    {
      "dynamics_loss_avg": 0.051652162708342074,
      "epoch": 3.8051605224609377e-06,
      "step": 77990
    },
    {
      "action_loss_avg": 0.0021769973856862633,
      "epoch": 3.8051605224609377e-06,
      "step": 77990
    },
    {
      "epoch": 3.8051605224609377e-06,
      "step": 77990,
      "training_step_time": 7.964405059814453
    },
    {
      "epoch": 3.806114196777344e-06,
      "model_forward_time": 1.7643322944641113,
      "step": 77991
    },
    {
      "epoch": 3.806114196777344e-06,
      "step": 77991,
      "training_step_time": 7.815558195114136
    },
    {
      "epoch": 3.80706787109375e-06,
      "model_forward_time": 1.8074798583984375,
      "step": 77992
    },
    {
      "epoch": 3.80706787109375e-06,
      "step": 77992,
      "training_step_time": 7.832455396652222
    },
    {
      "epoch": 3.808021545410156e-06,
      "model_forward_time": 1.770559310913086,
      "step": 77993
    },
    {
      "epoch": 3.808021545410156e-06,
      "step": 77993,
      "training_step_time": 7.790351629257202
    },
    {
      "epoch": 3.8089752197265627e-06,
      "model_forward_time": 1.7692234516143799,
      "step": 77994
    },
    {
      "epoch": 3.8089752197265627e-06,
      "step": 77994,
      "training_step_time": 7.728822708129883
    },
    {
      "epoch": 3.809928894042969e-06,
      "model_forward_time": 1.7575795650482178,
      "step": 77995
    },
    {
      "epoch": 3.809928894042969e-06,
      "step": 77995,
      "training_step_time": 7.76220440864563
    },
    {
      "epoch": 3.810882568359375e-06,
      "model_forward_time": 1.7712993621826172,
      "step": 77996
    },
    {
      "epoch": 3.810882568359375e-06,
      "step": 77996,
      "training_step_time": 7.853219270706177
    },
    {
      "epoch": 3.811836242675781e-06,
      "model_forward_time": 1.795271396636963,
      "step": 77997
    },
    {
      "epoch": 3.811836242675781e-06,
      "step": 77997,
      "training_step_time": 8.955219745635986
    },
    {
      "epoch": 3.8127899169921877e-06,
      "model_forward_time": 1.8014264106750488,
      "step": 77998
    },
    {
      "epoch": 3.8127899169921877e-06,
      "step": 77998,
      "training_step_time": 8.008533954620361
    },
    {
      "epoch": 3.813743591308594e-06,
      "model_forward_time": 1.8129780292510986,
      "step": 77999
    },
    {
      "epoch": 3.813743591308594e-06,
      "step": 77999,
      "training_step_time": 9.326100587844849
    },
    {
      "epoch": 3.814697265625e-06,
      "grad_norm": 0.0806136205792427,
      "learning_rate": 1.2660025769267937e-06,
      "loss": 0.0561,
      "step": 78000
    },
    {
      "epoch": 3.814697265625e-06,
      "model_forward_time": 1.7664036750793457,
      "step": 78000
    },
    {
      "dynamics_loss_avg": 0.05393628533929586,
      "epoch": 3.814697265625e-06,
      "step": 78000
    },
    {
      "action_loss_avg": 0.00106092213827651,
      "epoch": 3.814697265625e-06,
      "step": 78000
    },
    {
      "epoch": 3.814697265625e-06,
      "step": 78000,
      "training_step_time": 7.9723732471466064
    },
    {
      "epoch": 3.8156509399414066e-06,
      "model_forward_time": 1.7414984703063965,
      "step": 78001
    },
    {
      "epoch": 3.8156509399414066e-06,
      "step": 78001,
      "training_step_time": 8.864608526229858
    },
    {
      "epoch": 3.816604614257812e-06,
      "model_forward_time": 1.741647481918335,
      "step": 78002
    },
    {
      "epoch": 3.816604614257812e-06,
      "step": 78002,
      "training_step_time": 7.7745139598846436
    },
    {
      "epoch": 3.817558288574219e-06,
      "model_forward_time": 1.7440838813781738,
      "step": 78003
    },
    {
      "epoch": 3.817558288574219e-06,
      "step": 78003,
      "training_step_time": 7.94013237953186
    },
    {
      "epoch": 3.818511962890625e-06,
      "model_forward_time": 1.7798895835876465,
      "step": 78004
    },
    {
      "epoch": 3.818511962890625e-06,
      "step": 78004,
      "training_step_time": 7.625704050064087
    },
    {
      "epoch": 3.819465637207031e-06,
      "model_forward_time": 1.7456693649291992,
      "step": 78005
    },
    {
      "epoch": 3.819465637207031e-06,
      "step": 78005,
      "training_step_time": 9.24766993522644
    },
    {
      "epoch": 3.820419311523438e-06,
      "model_forward_time": 3.445016860961914,
      "step": 78006
    },
    {
      "epoch": 3.820419311523438e-06,
      "step": 78006,
      "training_step_time": 11.160894632339478
    },
    {
      "epoch": 3.8213729858398434e-06,
      "model_forward_time": 1.8297398090362549,
      "step": 78007
    },
    {
      "epoch": 3.8213729858398434e-06,
      "step": 78007,
      "training_step_time": 8.156058549880981
    },
    {
      "epoch": 3.82232666015625e-06,
      "model_forward_time": 1.7483141422271729,
      "step": 78008
    },
    {
      "epoch": 3.82232666015625e-06,
      "step": 78008,
      "training_step_time": 10.329160928726196
    },
    {
      "epoch": 3.8232803344726566e-06,
      "model_forward_time": 1.786586046218872,
      "step": 78009
    },
    {
      "epoch": 3.8232803344726566e-06,
      "step": 78009,
      "training_step_time": 9.455305576324463
    },
    {
      "epoch": 3.824234008789062e-06,
      "grad_norm": 0.08731292933225632,
      "learning_rate": 1.2649031422259667e-06,
      "loss": 0.0582,
      "step": 78010
    },
    {
      "epoch": 3.824234008789062e-06,
      "model_forward_time": 1.7543516159057617,
      "step": 78010
    },
    {
      "dynamics_loss_avg": 0.061338812857866284,
      "epoch": 3.824234008789062e-06,
      "step": 78010
    },
    {
      "action_loss_avg": 0.0011826424364699052,
      "epoch": 3.824234008789062e-06,
      "step": 78010
    },
    {
      "epoch": 3.824234008789062e-06,
      "step": 78010,
      "training_step_time": 9.152021408081055
    },
    {
      "epoch": 3.825187683105469e-06,
      "model_forward_time": 1.7449982166290283,
      "step": 78011
    },
    {
      "epoch": 3.825187683105469e-06,
      "step": 78011,
      "training_step_time": 7.585848808288574
    },
    {
      "epoch": 3.826141357421875e-06,
      "model_forward_time": 1.751094102859497,
      "step": 78012
    },
    {
      "epoch": 3.826141357421875e-06,
      "step": 78012,
      "training_step_time": 7.826723337173462
    },
    {
      "epoch": 3.827095031738281e-06,
      "model_forward_time": 1.7727012634277344,
      "step": 78013
    },
    {
      "epoch": 3.827095031738281e-06,
      "step": 78013,
      "training_step_time": 7.60183572769165
    },
    {
      "epoch": 3.828048706054688e-06,
      "model_forward_time": 1.7629156112670898,
      "step": 78014
    },
    {
      "epoch": 3.828048706054688e-06,
      "step": 78014,
      "training_step_time": 7.58035683631897
    },
    {
      "epoch": 3.8290023803710934e-06,
      "model_forward_time": 2.8422818183898926,
      "step": 78015
    },
    {
      "epoch": 3.8290023803710934e-06,
      "step": 78015,
      "training_step_time": 7.634709596633911
    },
    {
      "epoch": 3.8299560546875e-06,
      "model_forward_time": 1.7514266967773438,
      "step": 78016
    },
    {
      "epoch": 3.8299560546875e-06,
      "step": 78016,
      "training_step_time": 8.639449834823608
    },
    {
      "epoch": 3.830909729003907e-06,
      "model_forward_time": 1.7543249130249023,
      "step": 78017
    },
    {
      "epoch": 3.830909729003907e-06,
      "step": 78017,
      "training_step_time": 9.053214311599731
    },
    {
      "epoch": 3.831863403320312e-06,
      "model_forward_time": 1.7711410522460938,
      "step": 78018
    },
    {
      "epoch": 3.831863403320312e-06,
      "step": 78018,
      "training_step_time": 7.94861102104187
    },
    {
      "epoch": 3.832817077636719e-06,
      "model_forward_time": 1.7551579475402832,
      "step": 78019
    },
    {
      "epoch": 3.832817077636719e-06,
      "step": 78019,
      "training_step_time": 8.29575228691101
    },
    {
      "epoch": 3.833770751953125e-06,
      "grad_norm": 0.08211508393287659,
      "learning_rate": 1.2638041159897164e-06,
      "loss": 0.0583,
      "step": 78020
    },
    {
      "epoch": 3.833770751953125e-06,
      "model_forward_time": 1.7432632446289062,
      "step": 78020
    },
    {
      "dynamics_loss_avg": 0.04399599730968475,
      "epoch": 3.833770751953125e-06,
      "step": 78020
    },
    {
      "action_loss_avg": 0.0018574785179225727,
      "epoch": 3.833770751953125e-06,
      "step": 78020
    },
    {
      "epoch": 3.833770751953125e-06,
      "step": 78020,
      "training_step_time": 7.6699864864349365
    },
    {
      "epoch": 3.834724426269531e-06,
      "model_forward_time": 1.7517955303192139,
      "step": 78021
    },
    {
      "epoch": 3.834724426269531e-06,
      "step": 78021,
      "training_step_time": 8.228891372680664
    },
    {
      "epoch": 3.835678100585938e-06,
      "model_forward_time": 1.7749640941619873,
      "step": 78022
    },
    {
      "epoch": 3.835678100585938e-06,
      "step": 78022,
      "training_step_time": 7.747974157333374
    },
    {
      "epoch": 3.8366317749023435e-06,
      "model_forward_time": 1.746009349822998,
      "step": 78023
    },
    {
      "epoch": 3.8366317749023435e-06,
      "step": 78023,
      "training_step_time": 7.681846380233765
    },
    {
      "epoch": 3.83758544921875e-06,
      "model_forward_time": 2.8017256259918213,
      "step": 78024
    },
    {
      "epoch": 3.83758544921875e-06,
      "step": 78024,
      "training_step_time": 7.764683723449707
    },
    {
      "epoch": 3.838539123535157e-06,
      "model_forward_time": 1.754488229751587,
      "step": 78025
    },
    {
      "epoch": 3.838539123535157e-06,
      "step": 78025,
      "training_step_time": 7.7720277309417725
    },
    {
      "epoch": 3.839492797851562e-06,
      "model_forward_time": 1.7386415004730225,
      "step": 78026
    },
    {
      "epoch": 3.839492797851562e-06,
      "step": 78026,
      "training_step_time": 7.742310285568237
    },
    {
      "epoch": 3.840446472167969e-06,
      "model_forward_time": 1.7440228462219238,
      "step": 78027
    },
    {
      "epoch": 3.840446472167969e-06,
      "step": 78027,
      "training_step_time": 8.688394784927368
    },
    {
      "epoch": 3.841400146484375e-06,
      "model_forward_time": 1.7568774223327637,
      "step": 78028
    },
    {
      "epoch": 3.841400146484375e-06,
      "step": 78028,
      "training_step_time": 8.33082628250122
    },
    {
      "epoch": 3.842353820800781e-06,
      "model_forward_time": 1.7314453125,
      "step": 78029
    },
    {
      "epoch": 3.842353820800781e-06,
      "step": 78029,
      "training_step_time": 7.7115747928619385
    },
    {
      "epoch": 3.843307495117188e-06,
      "grad_norm": 0.09155160933732986,
      "learning_rate": 1.262705498338232e-06,
      "loss": 0.0566,
      "step": 78030
    },
    {
      "epoch": 3.843307495117188e-06,
      "model_forward_time": 1.7453529834747314,
      "step": 78030
    },
    {
      "dynamics_loss_avg": 0.050928945373743775,
      "epoch": 3.843307495117188e-06,
      "step": 78030
    },
    {
      "action_loss_avg": 0.0017206916614668443,
      "epoch": 3.843307495117188e-06,
      "step": 78030
    },
    {
      "epoch": 3.843307495117188e-06,
      "step": 78030,
      "training_step_time": 7.7301225662231445
    },
    {
      "epoch": 3.8442611694335935e-06,
      "model_forward_time": 1.7536797523498535,
      "step": 78031
    },
    {
      "epoch": 3.8442611694335935e-06,
      "step": 78031,
      "training_step_time": 7.751053810119629
    },
    {
      "epoch": 3.84521484375e-06,
      "model_forward_time": 1.7464025020599365,
      "step": 78032
    },
    {
      "epoch": 3.84521484375e-06,
      "step": 78032,
      "training_step_time": 7.7574756145477295
    },
    {
      "epoch": 3.846168518066407e-06,
      "model_forward_time": 1.7439942359924316,
      "step": 78033
    },
    {
      "epoch": 3.846168518066407e-06,
      "step": 78033,
      "training_step_time": 8.345881462097168
    },
    {
      "epoch": 3.847122192382812e-06,
      "model_forward_time": 2.8479838371276855,
      "step": 78034
    },
    {
      "epoch": 3.847122192382812e-06,
      "step": 78034,
      "training_step_time": 8.534228563308716
    },
    {
      "epoch": 3.848075866699219e-06,
      "model_forward_time": 1.7436609268188477,
      "step": 78035
    },
    {
      "epoch": 3.848075866699219e-06,
      "step": 78035,
      "training_step_time": 8.583308458328247
    },
    {
      "epoch": 3.849029541015625e-06,
      "model_forward_time": 1.7453587055206299,
      "step": 78036
    },
    {
      "epoch": 3.849029541015625e-06,
      "step": 78036,
      "training_step_time": 11.231571197509766
    },
    {
      "epoch": 3.849983215332031e-06,
      "model_forward_time": 1.7613520622253418,
      "step": 78037
    },
    {
      "epoch": 3.849983215332031e-06,
      "step": 78037,
      "training_step_time": 9.267560005187988
    },
    {
      "epoch": 3.850936889648438e-06,
      "model_forward_time": 1.74723219871521,
      "step": 78038
    },
    {
      "epoch": 3.850936889648438e-06,
      "step": 78038,
      "training_step_time": 8.713250398635864
    },
    {
      "epoch": 3.8518905639648435e-06,
      "model_forward_time": 1.755079746246338,
      "step": 78039
    },
    {
      "epoch": 3.8518905639648435e-06,
      "step": 78039,
      "training_step_time": 8.277801275253296
    },
    {
      "epoch": 3.85284423828125e-06,
      "grad_norm": 0.11377472430467606,
      "learning_rate": 1.2616072893916543e-06,
      "loss": 0.0572,
      "step": 78040
    },
    {
      "epoch": 3.85284423828125e-06,
      "model_forward_time": 1.7496860027313232,
      "step": 78040
    },
    {
      "dynamics_loss_avg": 0.05989234484732151,
      "epoch": 3.85284423828125e-06,
      "step": 78040
    },
    {
      "action_loss_avg": 0.0018260872340761126,
      "epoch": 3.85284423828125e-06,
      "step": 78040
    },
    {
      "epoch": 3.85284423828125e-06,
      "step": 78040,
      "training_step_time": 7.77082896232605
    },
    {
      "epoch": 3.853797912597657e-06,
      "model_forward_time": 1.772085189819336,
      "step": 78041
    },
    {
      "epoch": 3.853797912597657e-06,
      "step": 78041,
      "training_step_time": 7.810386896133423
    },
    {
      "epoch": 3.854751586914062e-06,
      "model_forward_time": 1.7453186511993408,
      "step": 78042
    },
    {
      "epoch": 3.854751586914062e-06,
      "step": 78042,
      "training_step_time": 8.569658756256104
    },
    {
      "epoch": 3.855705261230469e-06,
      "model_forward_time": 2.858071804046631,
      "step": 78043
    },
    {
      "epoch": 3.855705261230469e-06,
      "step": 78043,
      "training_step_time": 7.82381534576416
    },
    {
      "epoch": 3.856658935546875e-06,
      "model_forward_time": 1.7489581108093262,
      "step": 78044
    },
    {
      "epoch": 3.856658935546875e-06,
      "step": 78044,
      "training_step_time": 9.414445877075195
    },
    {
      "epoch": 3.857612609863281e-06,
      "model_forward_time": 1.7461931705474854,
      "step": 78045
    },
    {
      "epoch": 3.857612609863281e-06,
      "step": 78045,
      "training_step_time": 7.853598356246948
    },
    {
      "epoch": 3.858566284179688e-06,
      "model_forward_time": 1.7779712677001953,
      "step": 78046
    },
    {
      "epoch": 3.858566284179688e-06,
      "step": 78046,
      "training_step_time": 7.97553014755249
    },
    {
      "epoch": 3.8595199584960935e-06,
      "model_forward_time": 1.7451932430267334,
      "step": 78047
    },
    {
      "epoch": 3.8595199584960935e-06,
      "step": 78047,
      "training_step_time": 7.783677816390991
    },
    {
      "epoch": 3.8604736328125e-06,
      "model_forward_time": 1.7455708980560303,
      "step": 78048
    },
    {
      "epoch": 3.8604736328125e-06,
      "step": 78048,
      "training_step_time": 7.686093091964722
    },
    {
      "epoch": 3.861427307128907e-06,
      "model_forward_time": 1.749864101409912,
      "step": 78049
    },
    {
      "epoch": 3.861427307128907e-06,
      "step": 78049,
      "training_step_time": 8.712560892105103
    },
    {
      "epoch": 3.862380981445312e-06,
      "grad_norm": 0.0915137305855751,
      "learning_rate": 1.2605094892700837e-06,
      "loss": 0.0578,
      "step": 78050
    },
    {
      "epoch": 3.862380981445312e-06,
      "model_forward_time": 1.7642498016357422,
      "step": 78050
    },
    {
      "dynamics_loss_avg": 0.061212115548551084,
      "epoch": 3.862380981445312e-06,
      "step": 78050
    },
    {
      "action_loss_avg": 0.0024285178660647944,
      "epoch": 3.862380981445312e-06,
      "step": 78050
    },
    {
      "epoch": 3.862380981445312e-06,
      "step": 78050,
      "training_step_time": 8.355139255523682
    },
    {
      "epoch": 3.863334655761719e-06,
      "model_forward_time": 1.7512764930725098,
      "step": 78051
    },
    {
      "epoch": 3.863334655761719e-06,
      "step": 78051,
      "training_step_time": 12.316020488739014
    },
    {
      "epoch": 3.864288330078125e-06,
      "model_forward_time": 1.7575042247772217,
      "step": 78052
    },
    {
      "epoch": 3.864288330078125e-06,
      "step": 78052,
      "training_step_time": 7.669002294540405
    },
    {
      "epoch": 3.865242004394531e-06,
      "model_forward_time": 1.7612361907958984,
      "step": 78053
    },
    {
      "epoch": 3.865242004394531e-06,
      "step": 78053,
      "training_step_time": 7.5640175342559814
    },
    {
      "epoch": 3.866195678710938e-06,
      "model_forward_time": 1.7580902576446533,
      "step": 78054
    },
    {
      "epoch": 3.866195678710938e-06,
      "step": 78054,
      "training_step_time": 7.876702308654785
    },
    {
      "epoch": 3.8671493530273435e-06,
      "model_forward_time": 1.757941484451294,
      "step": 78055
    },
    {
      "epoch": 3.8671493530273435e-06,
      "step": 78055,
      "training_step_time": 7.737807035446167
    },
    {
      "epoch": 3.86810302734375e-06,
      "model_forward_time": 1.755958080291748,
      "step": 78056
    },
    {
      "epoch": 3.86810302734375e-06,
      "step": 78056,
      "training_step_time": 6.839358329772949
    },
    {
      "epoch": 3.869056701660157e-06,
      "model_forward_time": 1.7595140933990479,
      "step": 78057
    },
    {
      "epoch": 3.869056701660157e-06,
      "step": 78057,
      "training_step_time": 6.851035118103027
    },
    {
      "epoch": 3.870010375976562e-06,
      "model_forward_time": 1.7615141868591309,
      "step": 78058
    },
    {
      "epoch": 3.870010375976562e-06,
      "step": 78058,
      "training_step_time": 6.846614122390747
    },
    {
      "epoch": 3.870964050292969e-06,
      "model_forward_time": 1.048799991607666,
      "step": 78059
    },
    {
      "epoch": 3.870964050292969e-06,
      "step": 78059,
      "training_step_time": 6.875559091567993
    },
    {
      "epoch": 3.871917724609375e-06,
      "grad_norm": 0.0889035165309906,
      "learning_rate": 1.259412098093572e-06,
      "loss": 0.0577,
      "step": 78060
    },
    {
      "epoch": 3.871917724609375e-06,
      "model_forward_time": 1.7535951137542725,
      "step": 78060
    },
    {
      "dynamics_loss_avg": 0.0704385619610548,
      "epoch": 3.871917724609375e-06,
      "step": 78060
    },
    {
      "action_loss_avg": 0.0014827236765995622,
      "epoch": 3.871917724609375e-06,
      "step": 78060
    },
    {
      "epoch": 3.871917724609375e-06,
      "step": 78060,
      "training_step_time": 6.879063844680786
    },
    {
      "epoch": 3.872871398925781e-06,
      "model_forward_time": 2.836193561553955,
      "step": 78061
    },
    {
      "epoch": 3.872871398925781e-06,
      "step": 78061,
      "training_step_time": 7.935096263885498
    },
    {
      "epoch": 3.873825073242188e-06,
      "model_forward_time": 1.7522552013397217,
      "step": 78062
    },
    {
      "epoch": 3.873825073242188e-06,
      "step": 78062,
      "training_step_time": 7.390606164932251
    },
    {
      "epoch": 3.8747787475585935e-06,
      "model_forward_time": 1.7541985511779785,
      "step": 78063
    },
    {
      "epoch": 3.8747787475585935e-06,
      "step": 78063,
      "training_step_time": 6.860540866851807
    },
    {
      "epoch": 3.875732421875e-06,
      "model_forward_time": 1.7628309726715088,
      "step": 78064
    },
    {
      "epoch": 3.875732421875e-06,
      "step": 78064,
      "training_step_time": 7.726160049438477
    },
    {
      "epoch": 3.876686096191407e-06,
      "model_forward_time": 1.7197952270507812,
      "step": 78065
    },
    {
      "epoch": 3.876686096191407e-06,
      "step": 78065,
      "training_step_time": 7.53015398979187
    },
    {
      "epoch": 3.877639770507812e-06,
      "model_forward_time": 1.746100664138794,
      "step": 78066
    },
    {
      "epoch": 3.877639770507812e-06,
      "step": 78066,
      "training_step_time": 6.908030986785889
    },
    {
      "epoch": 3.878593444824219e-06,
      "model_forward_time": 1.7388195991516113,
      "step": 78067
    },
    {
      "epoch": 3.878593444824219e-06,
      "step": 78067,
      "training_step_time": 6.86638331413269
    },
    {
      "epoch": 3.879547119140625e-06,
      "model_forward_time": 1.7575078010559082,
      "step": 78068
    },
    {
      "epoch": 3.879547119140625e-06,
      "step": 78068,
      "training_step_time": 6.862075567245483
    },
    {
      "epoch": 3.880500793457031e-06,
      "model_forward_time": 1.7555711269378662,
      "step": 78069
    },
    {
      "epoch": 3.880500793457031e-06,
      "step": 78069,
      "training_step_time": 6.862239837646484
    },
    {
      "epoch": 3.881454467773438e-06,
      "grad_norm": 0.10952860116958618,
      "learning_rate": 1.258315115982131e-06,
      "loss": 0.0553,
      "step": 78070
    },
    {
      "epoch": 3.881454467773438e-06,
      "model_forward_time": 2.3844926357269287,
      "step": 78070
    },
    {
      "dynamics_loss_avg": 0.046413285098969935,
      "epoch": 3.881454467773438e-06,
      "step": 78070
    },
    {
      "action_loss_avg": 0.0016539473173907027,
      "epoch": 3.881454467773438e-06,
      "step": 78070
    },
    {
      "epoch": 3.881454467773438e-06,
      "step": 78070,
      "training_step_time": 7.908214569091797
    },
    {
      "epoch": 3.8824081420898435e-06,
      "model_forward_time": 1.7571935653686523,
      "step": 78071
    },
    {
      "epoch": 3.8824081420898435e-06,
      "step": 78071,
      "training_step_time": 7.401586294174194
    },
    {
      "epoch": 3.88336181640625e-06,
      "model_forward_time": 1.7571208477020264,
      "step": 78072
    },
    {
      "epoch": 3.88336181640625e-06,
      "step": 78072,
      "training_step_time": 6.873864412307739
    },
    {
      "epoch": 3.884315490722657e-06,
      "model_forward_time": 1.7665586471557617,
      "step": 78073
    },
    {
      "epoch": 3.884315490722657e-06,
      "step": 78073,
      "training_step_time": 7.72251296043396
    },
    {
      "epoch": 3.885269165039062e-06,
      "model_forward_time": 1.7578015327453613,
      "step": 78074
    },
    {
      "epoch": 3.885269165039062e-06,
      "step": 78074,
      "training_step_time": 7.578369140625
    },
    {
      "epoch": 3.886222839355469e-06,
      "model_forward_time": 1.7615931034088135,
      "step": 78075
    },
    {
      "epoch": 3.886222839355469e-06,
      "step": 78075,
      "training_step_time": 6.863605260848999
    },
    {
      "epoch": 3.887176513671875e-06,
      "model_forward_time": 1.7614004611968994,
      "step": 78076
    },
    {
      "epoch": 3.887176513671875e-06,
      "step": 78076,
      "training_step_time": 6.861095666885376
    },
    {
      "epoch": 3.888130187988281e-06,
      "model_forward_time": 1.7597391605377197,
      "step": 78077
    },
    {
      "epoch": 3.888130187988281e-06,
      "step": 78077,
      "training_step_time": 6.880282640457153
    },
    {
      "epoch": 3.889083862304688e-06,
      "model_forward_time": 1.765364408493042,
      "step": 78078
    },
    {
      "epoch": 3.889083862304688e-06,
      "step": 78078,
      "training_step_time": 6.848395347595215
    },
    {
      "epoch": 3.8900375366210935e-06,
      "model_forward_time": 2.4039347171783447,
      "step": 78079
    },
    {
      "epoch": 3.8900375366210935e-06,
      "step": 78079,
      "training_step_time": 7.903689622879028
    },
    {
      "epoch": 3.8909912109375e-06,
      "grad_norm": 0.515521228313446,
      "learning_rate": 1.2572185430557204e-06,
      "loss": 0.0584,
      "step": 78080
    },
    {
      "epoch": 3.8909912109375e-06,
      "model_forward_time": 1.7600796222686768,
      "step": 78080
    },
    {
      "dynamics_loss_avg": 0.0662108886986971,
      "epoch": 3.8909912109375e-06,
      "step": 78080
    },
    {
      "action_loss_avg": 0.0018364737101364882,
      "epoch": 3.8909912109375e-06,
      "step": 78080
    },
    {
      "epoch": 3.8909912109375e-06,
      "step": 78080,
      "training_step_time": 7.91744589805603
    },
    {
      "epoch": 3.891944885253907e-06,
      "model_forward_time": 1.7603535652160645,
      "step": 78081
    },
    {
      "epoch": 3.891944885253907e-06,
      "step": 78081,
      "training_step_time": 7.432142972946167
    },
    {
      "epoch": 3.892898559570312e-06,
      "model_forward_time": 1.7691552639007568,
      "step": 78082
    },
    {
      "epoch": 3.892898559570312e-06,
      "step": 78082,
      "training_step_time": 7.71507453918457
    },
    {
      "epoch": 3.893852233886719e-06,
      "model_forward_time": 1.7583997249603271,
      "step": 78083
    },
    {
      "epoch": 3.893852233886719e-06,
      "step": 78083,
      "training_step_time": 7.594512462615967
    },
    {
      "epoch": 3.894805908203125e-06,
      "model_forward_time": 1.7581360340118408,
      "step": 78084
    },
    {
      "epoch": 3.894805908203125e-06,
      "step": 78084,
      "training_step_time": 6.920639514923096
    },
    {
      "epoch": 3.895759582519531e-06,
      "model_forward_time": 1.7535247802734375,
      "step": 78085
    },
    {
      "epoch": 3.895759582519531e-06,
      "step": 78085,
      "training_step_time": 6.88899564743042
    },
    {
      "epoch": 3.896713256835938e-06,
      "model_forward_time": 1.394641637802124,
      "step": 78086
    },
    {
      "epoch": 3.896713256835938e-06,
      "step": 78086,
      "training_step_time": 6.89482569694519
    },
    {
      "epoch": 3.8976669311523435e-06,
      "model_forward_time": 1.768610954284668,
      "step": 78087
    },
    {
      "epoch": 3.8976669311523435e-06,
      "step": 78087,
      "training_step_time": 6.911337852478027
    },
    {
      "epoch": 3.89862060546875e-06,
      "model_forward_time": 1.767085075378418,
      "step": 78088
    },
    {
      "epoch": 3.89862060546875e-06,
      "step": 78088,
      "training_step_time": 9.93199348449707
    },
    {
      "epoch": 3.899574279785157e-06,
      "model_forward_time": 2.3842554092407227,
      "step": 78089
    },
    {
      "epoch": 3.899574279785157e-06,
      "step": 78089,
      "training_step_time": 7.9120728969573975
    },
    {
      "epoch": 3.900527954101562e-06,
      "grad_norm": 0.08856307715177536,
      "learning_rate": 1.2561223794342659e-06,
      "loss": 0.0572,
      "step": 78090
    },
    {
      "epoch": 3.900527954101562e-06,
      "model_forward_time": 1.767974853515625,
      "step": 78090
    },
    {
      "dynamics_loss_avg": 0.07655945159494877,
      "epoch": 3.900527954101562e-06,
      "step": 78090
    },
    {
      "action_loss_avg": 0.002877891264506616,
      "epoch": 3.900527954101562e-06,
      "step": 78090
    },
    {
      "epoch": 3.900527954101562e-06,
      "step": 78090,
      "training_step_time": 7.5017640590667725
    },
    {
      "epoch": 3.901481628417969e-06,
      "model_forward_time": 1.7706151008605957,
      "step": 78091
    },
    {
      "epoch": 3.901481628417969e-06,
      "step": 78091,
      "training_step_time": 7.781924486160278
    },
    {
      "epoch": 3.902435302734375e-06,
      "model_forward_time": 1.7612168788909912,
      "step": 78092
    },
    {
      "epoch": 3.902435302734375e-06,
      "step": 78092,
      "training_step_time": 8.173159122467041
    },
    {
      "epoch": 3.903388977050781e-06,
      "model_forward_time": 1.767826795578003,
      "step": 78093
    },
    {
      "epoch": 3.903388977050781e-06,
      "step": 78093,
      "training_step_time": 6.923007011413574
    },
    {
      "epoch": 3.904342651367188e-06,
      "model_forward_time": 1.7649667263031006,
      "step": 78094
    },
    {
      "epoch": 3.904342651367188e-06,
      "step": 78094,
      "training_step_time": 6.943335771560669
    },
    {
      "epoch": 3.9052963256835935e-06,
      "model_forward_time": 1.776332139968872,
      "step": 78095
    },
    {
      "epoch": 3.9052963256835935e-06,
      "step": 78095,
      "training_step_time": 6.902334690093994
    },
    {
      "epoch": 3.90625e-06,
      "model_forward_time": 1.7665483951568604,
      "step": 78096
    },
    {
      "epoch": 3.90625e-06,
      "step": 78096,
      "training_step_time": 6.871562719345093
    },
    {
      "epoch": 3.907203674316407e-06,
      "model_forward_time": 1.7675344944000244,
      "step": 78097
    },
    {
      "epoch": 3.907203674316407e-06,
      "step": 78097,
      "training_step_time": 10.6828293800354
    },
    {
      "epoch": 3.908157348632812e-06,
      "model_forward_time": 2.413362979888916,
      "step": 78098
    },
    {
      "epoch": 3.908157348632812e-06,
      "step": 78098,
      "training_step_time": 7.9095399379730225
    },
    {
      "epoch": 3.909111022949219e-06,
      "model_forward_time": 1.759638786315918,
      "step": 78099
    },
    {
      "epoch": 3.909111022949219e-06,
      "step": 78099,
      "training_step_time": 7.459650754928589
    },
    {
      "epoch": 3.910064697265625e-06,
      "grad_norm": 0.08169720321893692,
      "learning_rate": 1.255026625237637e-06,
      "loss": 0.0562,
      "step": 78100
    },
    {
      "epoch": 3.910064697265625e-06,
      "model_forward_time": 1.7767367362976074,
      "step": 78100
    },
    {
      "dynamics_loss_avg": 0.05426853494718671,
      "epoch": 3.910064697265625e-06,
      "step": 78100
    },
    {
      "action_loss_avg": 0.002861135092098266,
      "epoch": 3.910064697265625e-06,
      "step": 78100
    },
    {
      "epoch": 3.910064697265625e-06,
      "step": 78100,
      "training_step_time": 7.705286979675293
    },
    {
      "epoch": 3.911018371582031e-06,
      "model_forward_time": 1.7691891193389893,
      "step": 78101
    },
    {
      "epoch": 3.911018371582031e-06,
      "step": 78101,
      "training_step_time": 7.525550365447998
    },
    {
      "epoch": 3.911972045898438e-06,
      "model_forward_time": 1.747774600982666,
      "step": 78102
    },
    {
      "epoch": 3.911972045898438e-06,
      "step": 78102,
      "training_step_time": 7.552156925201416
    },
    {
      "epoch": 3.9129257202148435e-06,
      "model_forward_time": 1.7568261623382568,
      "step": 78103
    },
    {
      "epoch": 3.9129257202148435e-06,
      "step": 78103,
      "training_step_time": 6.851192474365234
    },
    {
      "epoch": 3.91387939453125e-06,
      "model_forward_time": 1.7605836391448975,
      "step": 78104
    },
    {
      "epoch": 3.91387939453125e-06,
      "step": 78104,
      "training_step_time": 6.831634759902954
    },
    {
      "epoch": 3.914833068847657e-06,
      "model_forward_time": 1.7656888961791992,
      "step": 78105
    },
    {
      "epoch": 3.914833068847657e-06,
      "step": 78105,
      "training_step_time": 6.889769792556763
    },
    {
      "epoch": 3.915786743164062e-06,
      "model_forward_time": 1.7636983394622803,
      "step": 78106
    },
    {
      "epoch": 3.915786743164062e-06,
      "step": 78106,
      "training_step_time": 9.227718591690063
    },
    {
      "epoch": 3.916740417480469e-06,
      "model_forward_time": 2.3976354598999023,
      "step": 78107
    },
    {
      "epoch": 3.916740417480469e-06,
      "step": 78107,
      "training_step_time": 7.865893602371216
    },
    {
      "epoch": 3.917694091796875e-06,
      "model_forward_time": 1.7702205181121826,
      "step": 78108
    },
    {
      "epoch": 3.917694091796875e-06,
      "step": 78108,
      "training_step_time": 7.2701826095581055
    },
    {
      "epoch": 3.918647766113281e-06,
      "model_forward_time": 1.7619991302490234,
      "step": 78109
    },
    {
      "epoch": 3.918647766113281e-06,
      "step": 78109,
      "training_step_time": 7.702594518661499
    },
    {
      "epoch": 3.919601440429688e-06,
      "grad_norm": 0.07341407239437103,
      "learning_rate": 1.253931280585668e-06,
      "loss": 0.056,
      "step": 78110
    },
    {
      "epoch": 3.919601440429688e-06,
      "model_forward_time": 1.7650585174560547,
      "step": 78110
    },
    {
      "dynamics_loss_avg": 0.044707605987787245,
      "epoch": 3.919601440429688e-06,
      "step": 78110
    },
    {
      "action_loss_avg": 0.0012326808646321297,
      "epoch": 3.919601440429688e-06,
      "step": 78110
    },
    {
      "epoch": 3.919601440429688e-06,
      "step": 78110,
      "training_step_time": 7.54646897315979
    },
    {
      "epoch": 3.9205551147460935e-06,
      "model_forward_time": 1.7510793209075928,
      "step": 78111
    },
    {
      "epoch": 3.9205551147460935e-06,
      "step": 78111,
      "training_step_time": 7.594406604766846
    },
    {
      "epoch": 3.9215087890625e-06,
      "model_forward_time": 1.7631330490112305,
      "step": 78112
    },
    {
      "epoch": 3.9215087890625e-06,
      "step": 78112,
      "training_step_time": 6.872121095657349
    },
    {
      "epoch": 3.922462463378907e-06,
      "model_forward_time": 1.7632224559783936,
      "step": 78113
    },
    {
      "epoch": 3.922462463378907e-06,
      "step": 78113,
      "training_step_time": 6.902487277984619
    },
    {
      "epoch": 3.923416137695312e-06,
      "model_forward_time": 1.7714543342590332,
      "step": 78114
    },
    {
      "epoch": 3.923416137695312e-06,
      "step": 78114,
      "training_step_time": 6.890296697616577
    },
    {
      "epoch": 3.924369812011719e-06,
      "model_forward_time": 1.75593900680542,
      "step": 78115
    },
    {
      "epoch": 3.924369812011719e-06,
      "step": 78115,
      "training_step_time": 8.547492265701294
    },
    {
      "epoch": 3.925323486328125e-06,
      "model_forward_time": 2.422968864440918,
      "step": 78116
    },
    {
      "epoch": 3.925323486328125e-06,
      "step": 78116,
      "training_step_time": 10.551531314849854
    },
    {
      "epoch": 3.926277160644531e-06,
      "model_forward_time": 1.7644660472869873,
      "step": 78117
    },
    {
      "epoch": 3.926277160644531e-06,
      "step": 78117,
      "training_step_time": 7.402011871337891
    },
    {
      "epoch": 3.927230834960938e-06,
      "model_forward_time": 1.756495475769043,
      "step": 78118
    },
    {
      "epoch": 3.927230834960938e-06,
      "step": 78118,
      "training_step_time": 7.5071845054626465
    },
    {
      "epoch": 3.9281845092773435e-06,
      "model_forward_time": 1.7689154148101807,
      "step": 78119
    },
    {
      "epoch": 3.9281845092773435e-06,
      "step": 78119,
      "training_step_time": 7.7404255867004395
    },
    {
      "epoch": 3.92913818359375e-06,
      "grad_norm": 0.13924291729927063,
      "learning_rate": 1.25283634559814e-06,
      "loss": 0.0564,
      "step": 78120
    },
    {
      "epoch": 3.92913818359375e-06,
      "model_forward_time": 1.7474563121795654,
      "step": 78120
    },
    {
      "dynamics_loss_avg": 0.045018930546939374,
      "epoch": 3.92913818359375e-06,
      "step": 78120
    },
    {
      "action_loss_avg": 0.00201518940157257,
      "epoch": 3.92913818359375e-06,
      "step": 78120
    },
    {
      "epoch": 3.92913818359375e-06,
      "step": 78120,
      "training_step_time": 7.636329889297485
    },
    {
      "epoch": 3.930091857910157e-06,
      "model_forward_time": 1.7599995136260986,
      "step": 78121
    },
    {
      "epoch": 3.930091857910157e-06,
      "step": 78121,
      "training_step_time": 6.890097618103027
    },
    {
      "epoch": 3.931045532226562e-06,
      "model_forward_time": 1.7524662017822266,
      "step": 78122
    },
    {
      "epoch": 3.931045532226562e-06,
      "step": 78122,
      "training_step_time": 6.881590366363525
    },
    {
      "epoch": 3.931999206542969e-06,
      "model_forward_time": 1.7568554878234863,
      "step": 78123
    },
    {
      "epoch": 3.931999206542969e-06,
      "step": 78123,
      "training_step_time": 6.897619962692261
    },
    {
      "epoch": 3.932952880859375e-06,
      "model_forward_time": 1.7664973735809326,
      "step": 78124
    },
    {
      "epoch": 3.932952880859375e-06,
      "step": 78124,
      "training_step_time": 8.446310758590698
    },
    {
      "epoch": 3.933906555175781e-06,
      "model_forward_time": 2.43428111076355,
      "step": 78125
    },
    {
      "epoch": 3.933906555175781e-06,
      "step": 78125,
      "training_step_time": 8.738381147384644
    },
    {
      "epoch": 3.934860229492188e-06,
      "model_forward_time": 1.7632298469543457,
      "step": 78126
    },
    {
      "epoch": 3.934860229492188e-06,
      "step": 78126,
      "training_step_time": 7.911977529525757
    },
    {
      "epoch": 3.9358139038085935e-06,
      "model_forward_time": 1.7621991634368896,
      "step": 78127
    },
    {
      "epoch": 3.9358139038085935e-06,
      "step": 78127,
      "training_step_time": 7.51521110534668
    },
    {
      "epoch": 3.936767578125e-06,
      "model_forward_time": 1.763319492340088,
      "step": 78128
    },
    {
      "epoch": 3.936767578125e-06,
      "step": 78128,
      "training_step_time": 7.774595022201538
    },
    {
      "epoch": 3.937721252441406e-06,
      "model_forward_time": 1.7593886852264404,
      "step": 78129
    },
    {
      "epoch": 3.937721252441406e-06,
      "step": 78129,
      "training_step_time": 7.628371953964233
    },
    {
      "epoch": 3.938674926757812e-06,
      "grad_norm": 0.07955671846866608,
      "learning_rate": 1.2517418203947973e-06,
      "loss": 0.0572,
      "step": 78130
    },
    {
      "epoch": 3.938674926757812e-06,
      "model_forward_time": 1.7654531002044678,
      "step": 78130
    },
    {
      "dynamics_loss_avg": 0.06380402240902186,
      "epoch": 3.938674926757812e-06,
      "step": 78130
    },
    {
      "action_loss_avg": 0.0016035931301303207,
      "epoch": 3.938674926757812e-06,
      "step": 78130
    },
    {
      "epoch": 3.938674926757812e-06,
      "step": 78130,
      "training_step_time": 6.858352184295654
    },
    {
      "epoch": 3.939628601074219e-06,
      "model_forward_time": 1.7679383754730225,
      "step": 78131
    },
    {
      "epoch": 3.939628601074219e-06,
      "step": 78131,
      "training_step_time": 6.905855894088745
    },
    {
      "epoch": 3.940582275390625e-06,
      "model_forward_time": 1.7647593021392822,
      "step": 78132
    },
    {
      "epoch": 3.940582275390625e-06,
      "step": 78132,
      "training_step_time": 6.906733989715576
    },
    {
      "epoch": 3.941535949707031e-06,
      "model_forward_time": 1.7499966621398926,
      "step": 78133
    },
    {
      "epoch": 3.941535949707031e-06,
      "step": 78133,
      "training_step_time": 6.837087154388428
    },
    {
      "epoch": 3.942489624023438e-06,
      "model_forward_time": 1.8413317203521729,
      "step": 78134
    },
    {
      "epoch": 3.942489624023438e-06,
      "step": 78134,
      "training_step_time": 8.359206438064575
    },
    {
      "epoch": 3.9434432983398435e-06,
      "model_forward_time": 2.4112110137939453,
      "step": 78135
    },
    {
      "epoch": 3.9434432983398435e-06,
      "step": 78135,
      "training_step_time": 8.537672281265259
    },
    {
      "epoch": 3.94439697265625e-06,
      "model_forward_time": 1.7667243480682373,
      "step": 78136
    },
    {
      "epoch": 3.94439697265625e-06,
      "step": 78136,
      "training_step_time": 7.492871284484863
    },
    {
      "epoch": 3.945350646972656e-06,
      "model_forward_time": 1.7672135829925537,
      "step": 78137
    },
    {
      "epoch": 3.945350646972656e-06,
      "step": 78137,
      "training_step_time": 7.824228525161743
    },
    {
      "epoch": 3.946304321289062e-06,
      "model_forward_time": 1.7720870971679688,
      "step": 78138
    },
    {
      "epoch": 3.946304321289062e-06,
      "step": 78138,
      "training_step_time": 9.087177276611328
    },
    {
      "epoch": 3.947257995605469e-06,
      "model_forward_time": 1.7644147872924805,
      "step": 78139
    },
    {
      "epoch": 3.947257995605469e-06,
      "step": 78139,
      "training_step_time": 6.869526386260986
    },
    {
      "epoch": 3.948211669921875e-06,
      "grad_norm": 0.09683243185281754,
      "learning_rate": 1.2506477050953309e-06,
      "loss": 0.0563,
      "step": 78140
    },
    {
      "epoch": 3.948211669921875e-06,
      "model_forward_time": 1.7550547122955322,
      "step": 78140
    },
    {
      "dynamics_loss_avg": 0.04306529788300395,
      "epoch": 3.948211669921875e-06,
      "step": 78140
    },
    {
      "action_loss_avg": 0.0012704861990641802,
      "epoch": 3.948211669921875e-06,
      "step": 78140
    },
    {
      "epoch": 3.948211669921875e-06,
      "step": 78140,
      "training_step_time": 6.773270845413208
    },
    {
      "epoch": 3.949165344238281e-06,
      "model_forward_time": 1.7613000869750977,
      "step": 78141
    },
    {
      "epoch": 3.949165344238281e-06,
      "step": 78141,
      "training_step_time": 6.789773225784302
    },
    {
      "epoch": 3.950119018554688e-06,
      "model_forward_time": 1.765537977218628,
      "step": 78142
    },
    {
      "epoch": 3.950119018554688e-06,
      "step": 78142,
      "training_step_time": 6.7651002407073975
    },
    {
      "epoch": 3.9510726928710936e-06,
      "model_forward_time": 1.8150062561035156,
      "step": 78143
    },
    {
      "epoch": 3.9510726928710936e-06,
      "step": 78143,
      "training_step_time": 10.428119659423828
    },
    {
      "epoch": 3.9520263671875e-06,
      "model_forward_time": 2.40006947517395,
      "step": 78144
    },
    {
      "epoch": 3.9520263671875e-06,
      "step": 78144,
      "training_step_time": 7.874294281005859
    },
    {
      "epoch": 3.952980041503906e-06,
      "model_forward_time": 1.7685534954071045,
      "step": 78145
    },
    {
      "epoch": 3.952980041503906e-06,
      "step": 78145,
      "training_step_time": 8.594286680221558
    },
    {
      "epoch": 3.953933715820312e-06,
      "model_forward_time": 1.7478876113891602,
      "step": 78146
    },
    {
      "epoch": 3.953933715820312e-06,
      "step": 78146,
      "training_step_time": 7.733003616333008
    },
    {
      "epoch": 3.954887390136719e-06,
      "model_forward_time": 1.7605252265930176,
      "step": 78147
    },
    {
      "epoch": 3.954887390136719e-06,
      "step": 78147,
      "training_step_time": 8.619513750076294
    },
    {
      "epoch": 3.955841064453125e-06,
      "model_forward_time": 1.7671713829040527,
      "step": 78148
    },
    {
      "epoch": 3.955841064453125e-06,
      "step": 78148,
      "training_step_time": 7.586346626281738
    },
    {
      "epoch": 3.956794738769531e-06,
      "model_forward_time": 1.7603743076324463,
      "step": 78149
    },
    {
      "epoch": 3.956794738769531e-06,
      "step": 78149,
      "training_step_time": 6.794768333435059
    },
    {
      "epoch": 3.957748413085938e-06,
      "grad_norm": 0.12243335694074631,
      "learning_rate": 1.2495539998193972e-06,
      "loss": 0.0546,
      "step": 78150
    },
    {
      "epoch": 3.957748413085938e-06,
      "model_forward_time": 1.752716302871704,
      "step": 78150
    },
    {
      "dynamics_loss_avg": 0.056090245768427846,
      "epoch": 3.957748413085938e-06,
      "step": 78150
    },
    {
      "action_loss_avg": 0.0019816670974250883,
      "epoch": 3.957748413085938e-06,
      "step": 78150
    },
    {
      "epoch": 3.957748413085938e-06,
      "step": 78150,
      "training_step_time": 6.7699525356292725
    },
    {
      "epoch": 3.9587020874023436e-06,
      "model_forward_time": 1.7603952884674072,
      "step": 78151
    },
    {
      "epoch": 3.9587020874023436e-06,
      "step": 78151,
      "training_step_time": 6.841885805130005
    },
    {
      "epoch": 3.95965576171875e-06,
      "model_forward_time": 1.778458833694458,
      "step": 78152
    },
    {
      "epoch": 3.95965576171875e-06,
      "step": 78152,
      "training_step_time": 9.265633821487427
    },
    {
      "epoch": 3.960609436035156e-06,
      "model_forward_time": 2.39922833442688,
      "step": 78153
    },
    {
      "epoch": 3.960609436035156e-06,
      "step": 78153,
      "training_step_time": 11.52116084098816
    },
    {
      "epoch": 3.961563110351562e-06,
      "model_forward_time": 1.7614490985870361,
      "step": 78154
    },
    {
      "epoch": 3.961563110351562e-06,
      "step": 78154,
      "training_step_time": 7.771410703659058
    },
    {
      "epoch": 3.962516784667969e-06,
      "model_forward_time": 1.7561235427856445,
      "step": 78155
    },
    {
      "epoch": 3.962516784667969e-06,
      "step": 78155,
      "training_step_time": 7.72061014175415
    },
    {
      "epoch": 3.963470458984375e-06,
      "model_forward_time": 1.7562994956970215,
      "step": 78156
    },
    {
      "epoch": 3.963470458984375e-06,
      "step": 78156,
      "training_step_time": 7.551260709762573
    },
    {
      "epoch": 3.964424133300781e-06,
      "model_forward_time": 1.7629530429840088,
      "step": 78157
    },
    {
      "epoch": 3.964424133300781e-06,
      "step": 78157,
      "training_step_time": 7.534175634384155
    },
    {
      "epoch": 3.965377807617188e-06,
      "model_forward_time": 1.7834889888763428,
      "step": 78158
    },
    {
      "epoch": 3.965377807617188e-06,
      "step": 78158,
      "training_step_time": 6.813711643218994
    },
    {
      "epoch": 3.9663314819335936e-06,
      "model_forward_time": 1.7568514347076416,
      "step": 78159
    },
    {
      "epoch": 3.9663314819335936e-06,
      "step": 78159,
      "training_step_time": 6.793789863586426
    },
    {
      "epoch": 3.96728515625e-06,
      "grad_norm": 0.08390940725803375,
      "learning_rate": 1.2484607046865977e-06,
      "loss": 0.0555,
      "step": 78160
    },
    {
      "epoch": 3.96728515625e-06,
      "model_forward_time": 1.7556076049804688,
      "step": 78160
    },
    {
      "dynamics_loss_avg": 0.07157150749117136,
      "epoch": 3.96728515625e-06,
      "step": 78160
    },
    {
      "action_loss_avg": 0.0016564829333219678,
      "epoch": 3.96728515625e-06,
      "step": 78160
    },
    {
      "epoch": 3.96728515625e-06,
      "step": 78160,
      "training_step_time": 6.788878440856934
    },
    {
      "epoch": 3.968238830566406e-06,
      "model_forward_time": 1.763974666595459,
      "step": 78161
    },
    {
      "epoch": 3.968238830566406e-06,
      "step": 78161,
      "training_step_time": 8.603407621383667
    },
    {
      "epoch": 3.9691925048828124e-06,
      "model_forward_time": 2.423811674118042,
      "step": 78162
    },
    {
      "epoch": 3.9691925048828124e-06,
      "step": 78162,
      "training_step_time": 8.428433656692505
    },
    {
      "epoch": 3.970146179199219e-06,
      "model_forward_time": 1.7607369422912598,
      "step": 78163
    },
    {
      "epoch": 3.970146179199219e-06,
      "step": 78163,
      "training_step_time": 7.818417310714722
    },
    {
      "epoch": 3.971099853515625e-06,
      "model_forward_time": 1.768735647201538,
      "step": 78164
    },
    {
      "epoch": 3.971099853515625e-06,
      "step": 78164,
      "training_step_time": 7.4842448234558105
    },
    {
      "epoch": 3.972053527832031e-06,
      "model_forward_time": 1.7750895023345947,
      "step": 78165
    },
    {
      "epoch": 3.972053527832031e-06,
      "step": 78165,
      "training_step_time": 7.754631519317627
    },
    {
      "epoch": 3.973007202148438e-06,
      "model_forward_time": 1.778897762298584,
      "step": 78166
    },
    {
      "epoch": 3.973007202148438e-06,
      "step": 78166,
      "training_step_time": 7.65670919418335
    },
    {
      "epoch": 3.9739608764648436e-06,
      "model_forward_time": 1.775874137878418,
      "step": 78167
    },
    {
      "epoch": 3.9739608764648436e-06,
      "step": 78167,
      "training_step_time": 6.856872797012329
    },
    {
      "epoch": 3.97491455078125e-06,
      "model_forward_time": 1.7766902446746826,
      "step": 78168
    },
    {
      "epoch": 3.97491455078125e-06,
      "step": 78168,
      "training_step_time": 6.8076183795928955
    },
    {
      "epoch": 3.975868225097656e-06,
      "model_forward_time": 1.7809059619903564,
      "step": 78169
    },
    {
      "epoch": 3.975868225097656e-06,
      "step": 78169,
      "training_step_time": 6.824652910232544
    },
    {
      "epoch": 3.9768218994140624e-06,
      "grad_norm": 0.1763111650943756,
      "learning_rate": 1.2473678198164967e-06,
      "loss": 0.0564,
      "step": 78170
    },
    {
      "epoch": 3.9768218994140624e-06,
      "model_forward_time": 1.772204875946045,
      "step": 78170
    },
    {
      "dynamics_loss_avg": 0.06068713515996933,
      "epoch": 3.9768218994140624e-06,
      "step": 78170
    },
    {
      "action_loss_avg": 0.0016916184511501342,
      "epoch": 3.9768218994140624e-06,
      "step": 78170
    },
    {
      "epoch": 3.9768218994140624e-06,
      "step": 78170,
      "training_step_time": 7.806156396865845
    },
    {
      "epoch": 3.977775573730469e-06,
      "model_forward_time": 2.46685791015625,
      "step": 78171
    },
    {
      "epoch": 3.977775573730469e-06,
      "step": 78171,
      "training_step_time": 8.94184136390686
    },
    {
      "epoch": 3.978729248046875e-06,
      "model_forward_time": 1.7851297855377197,
      "step": 78172
    },
    {
      "epoch": 3.978729248046875e-06,
      "step": 78172,
      "training_step_time": 8.506948471069336
    },
    {
      "epoch": 3.979682922363281e-06,
      "model_forward_time": 1.7784392833709717,
      "step": 78173
    },
    {
      "epoch": 3.979682922363281e-06,
      "step": 78173,
      "training_step_time": 7.8321239948272705
    },
    {
      "epoch": 3.980636596679688e-06,
      "model_forward_time": 1.776867151260376,
      "step": 78174
    },
    {
      "epoch": 3.980636596679688e-06,
      "step": 78174,
      "training_step_time": 7.738160133361816
    },
    {
      "epoch": 3.981590270996094e-06,
      "model_forward_time": 1.7787647247314453,
      "step": 78175
    },
    {
      "epoch": 3.981590270996094e-06,
      "step": 78175,
      "training_step_time": 7.581916809082031
    },
    {
      "epoch": 3.9825439453125e-06,
      "model_forward_time": 1.7747101783752441,
      "step": 78176
    },
    {
      "epoch": 3.9825439453125e-06,
      "step": 78176,
      "training_step_time": 6.8133227825164795
    },
    {
      "epoch": 3.983497619628906e-06,
      "model_forward_time": 1.788853645324707,
      "step": 78177
    },
    {
      "epoch": 3.983497619628906e-06,
      "step": 78177,
      "training_step_time": 6.805668830871582
    },
    {
      "epoch": 3.9844512939453124e-06,
      "model_forward_time": 1.7852163314819336,
      "step": 78178
    },
    {
      "epoch": 3.9844512939453124e-06,
      "step": 78178,
      "training_step_time": 6.79915976524353
    },
    {
      "epoch": 3.985404968261719e-06,
      "model_forward_time": 1.7748589515686035,
      "step": 78179
    },
    {
      "epoch": 3.985404968261719e-06,
      "step": 78179,
      "training_step_time": 8.665885925292969
    },
    {
      "epoch": 3.986358642578125e-06,
      "grad_norm": 0.0877692848443985,
      "learning_rate": 1.2462753453286076e-06,
      "loss": 0.0558,
      "step": 78180
    },
    {
      "epoch": 3.986358642578125e-06,
      "model_forward_time": 1.8573877811431885,
      "step": 78180
    },
    {
      "dynamics_loss_avg": 0.05589690040796995,
      "epoch": 3.986358642578125e-06,
      "step": 78180
    },
    {
      "action_loss_avg": 0.001755662594223395,
      "epoch": 3.986358642578125e-06,
      "step": 78180
    },
    {
      "epoch": 3.986358642578125e-06,
      "step": 78180,
      "training_step_time": 11.809242010116577
    },
    {
      "epoch": 3.987312316894531e-06,
      "model_forward_time": 1.775242805480957,
      "step": 78181
    },
    {
      "epoch": 3.987312316894531e-06,
      "step": 78181,
      "training_step_time": 7.96744441986084
    },
    {
      "epoch": 3.988265991210938e-06,
      "model_forward_time": 1.7839088439941406,
      "step": 78182
    },
    {
      "epoch": 3.988265991210938e-06,
      "step": 78182,
      "training_step_time": 7.8666770458221436
    },
    {
      "epoch": 3.989219665527344e-06,
      "model_forward_time": 1.781510353088379,
      "step": 78183
    },
    {
      "epoch": 3.989219665527344e-06,
      "step": 78183,
      "training_step_time": 7.767378330230713
    },
    {
      "epoch": 3.99017333984375e-06,
      "model_forward_time": 1.784423589706421,
      "step": 78184
    },
    {
      "epoch": 3.99017333984375e-06,
      "step": 78184,
      "training_step_time": 7.636573314666748
    },
    {
      "epoch": 3.991127014160156e-06,
      "model_forward_time": 1.4152050018310547,
      "step": 78185
    },
    {
      "epoch": 3.991127014160156e-06,
      "step": 78185,
      "training_step_time": 7.226556062698364
    },
    {
      "epoch": 3.9920806884765624e-06,
      "model_forward_time": 1.7760841846466064,
      "step": 78186
    },
    {
      "epoch": 3.9920806884765624e-06,
      "step": 78186,
      "training_step_time": 6.840075969696045
    },
    {
      "epoch": 3.993034362792969e-06,
      "model_forward_time": 1.7674760818481445,
      "step": 78187
    },
    {
      "epoch": 3.993034362792969e-06,
      "step": 78187,
      "training_step_time": 6.877390146255493
    },
    {
      "epoch": 3.993988037109375e-06,
      "model_forward_time": 1.770538568496704,
      "step": 78188
    },
    {
      "epoch": 3.993988037109375e-06,
      "step": 78188,
      "training_step_time": 9.672272682189941
    },
    {
      "epoch": 3.994941711425781e-06,
      "model_forward_time": 1.818387746810913,
      "step": 78189
    },
    {
      "epoch": 3.994941711425781e-06,
      "step": 78189,
      "training_step_time": 12.259685516357422
    },
    {
      "epoch": 3.995895385742188e-06,
      "grad_norm": 0.08778425306081772,
      "learning_rate": 1.245183281342403e-06,
      "loss": 0.0545,
      "step": 78190
    },
    {
      "epoch": 3.995895385742188e-06,
      "model_forward_time": 2.4728119373321533,
      "step": 78190
    },
    {
      "dynamics_loss_avg": 0.05323302075266838,
      "epoch": 3.995895385742188e-06,
      "step": 78190
    },
    {
      "action_loss_avg": 0.0012637043808354064,
      "epoch": 3.995895385742188e-06,
      "step": 78190
    },
    {
      "epoch": 3.995895385742188e-06,
      "step": 78190,
      "training_step_time": 8.640256404876709
    },
    {
      "epoch": 3.996849060058594e-06,
      "model_forward_time": 1.751692295074463,
      "step": 78191
    },
    {
      "epoch": 3.996849060058594e-06,
      "step": 78191,
      "training_step_time": 7.812252759933472
    },
    {
      "epoch": 3.997802734375e-06,
      "model_forward_time": 1.7427701950073242,
      "step": 78192
    },
    {
      "epoch": 3.997802734375e-06,
      "step": 78192,
      "training_step_time": 7.71432638168335
    },
    {
      "epoch": 3.998756408691406e-06,
      "model_forward_time": 1.7701468467712402,
      "step": 78193
    },
    {
      "epoch": 3.998756408691406e-06,
      "step": 78193,
      "training_step_time": 8.50972056388855
    },
    {
      "epoch": 3.9997100830078125e-06,
      "model_forward_time": 1.765763521194458,
      "step": 78194
    },
    {
      "epoch": 3.9997100830078125e-06,
      "step": 78194,
      "training_step_time": 7.576785326004028
    },
    {
      "epoch": 4.000663757324219e-06,
      "model_forward_time": 1.765199899673462,
      "step": 78195
    },
    {
      "epoch": 4.000663757324219e-06,
      "step": 78195,
      "training_step_time": 6.816451787948608
    },
    {
      "epoch": 4.001617431640625e-06,
      "model_forward_time": 1.7717418670654297,
      "step": 78196
    },
    {
      "epoch": 4.001617431640625e-06,
      "step": 78196,
      "training_step_time": 6.822771310806274
    },
    {
      "epoch": 4.002571105957031e-06,
      "model_forward_time": 1.7769653797149658,
      "step": 78197
    },
    {
      "epoch": 4.002571105957031e-06,
      "step": 78197,
      "training_step_time": 7.587305784225464
    },
    {
      "epoch": 4.003524780273438e-06,
      "model_forward_time": 1.773036241531372,
      "step": 78198
    },
    {
      "epoch": 4.003524780273438e-06,
      "step": 78198,
      "training_step_time": 12.000550508499146
    },
    {
      "epoch": 4.004478454589844e-06,
      "model_forward_time": 2.4448437690734863,
      "step": 78199
    },
    {
      "epoch": 4.004478454589844e-06,
      "step": 78199,
      "training_step_time": 9.993883848190308
    },
    {
      "epoch": 4.00543212890625e-06,
      "grad_norm": 0.10220418870449066,
      "learning_rate": 1.2440916279773097e-06,
      "loss": 0.0559,
      "step": 78200
    },
    {
      "epoch": 4.00543212890625e-06,
      "model_forward_time": 1.7688634395599365,
      "step": 78200
    },
    {
      "dynamics_loss_avg": 0.05173906134441495,
      "epoch": 4.00543212890625e-06,
      "step": 78200
    },
    {
      "action_loss_avg": 0.0013480977911967784,
      "epoch": 4.00543212890625e-06,
      "step": 78200
    },
    {
      "epoch": 4.00543212890625e-06,
      "step": 78200,
      "training_step_time": 7.761271238327026
    },
    {
      "epoch": 4.006385803222656e-06,
      "model_forward_time": 1.752023458480835,
      "step": 78201
    },
    {
      "epoch": 4.006385803222656e-06,
      "step": 78201,
      "training_step_time": 8.213789224624634
    },
    {
      "epoch": 4.0073394775390625e-06,
      "model_forward_time": 1.7672581672668457,
      "step": 78202
    },
    {
      "epoch": 4.0073394775390625e-06,
      "step": 78202,
      "training_step_time": 7.727354049682617
    },
    {
      "epoch": 4.008293151855469e-06,
      "model_forward_time": 1.771446943283081,
      "step": 78203
    },
    {
      "epoch": 4.008293151855469e-06,
      "step": 78203,
      "training_step_time": 7.609750270843506
    },
    {
      "epoch": 4.009246826171875e-06,
      "model_forward_time": 1.7654874324798584,
      "step": 78204
    },
    {
      "epoch": 4.009246826171875e-06,
      "step": 78204,
      "training_step_time": 6.90065860748291
    },
    {
      "epoch": 4.010200500488281e-06,
      "model_forward_time": 1.7649016380310059,
      "step": 78205
    },
    {
      "epoch": 4.010200500488281e-06,
      "step": 78205,
      "training_step_time": 6.924215316772461
    },
    {
      "epoch": 4.011154174804688e-06,
      "model_forward_time": 1.7643680572509766,
      "step": 78206
    },
    {
      "epoch": 4.011154174804688e-06,
      "step": 78206,
      "training_step_time": 7.567065954208374
    },
    {
      "epoch": 4.012107849121094e-06,
      "model_forward_time": 1.7737505435943604,
      "step": 78207
    },
    {
      "epoch": 4.012107849121094e-06,
      "step": 78207,
      "training_step_time": 10.312283277511597
    },
    {
      "epoch": 4.0130615234375e-06,
      "model_forward_time": 2.4536895751953125,
      "step": 78208
    },
    {
      "epoch": 4.0130615234375e-06,
      "step": 78208,
      "training_step_time": 8.586210012435913
    },
    {
      "epoch": 4.014015197753906e-06,
      "model_forward_time": 1.7606189250946045,
      "step": 78209
    },
    {
      "epoch": 4.014015197753906e-06,
      "step": 78209,
      "training_step_time": 9.20027756690979
    },
    {
      "epoch": 4.0149688720703125e-06,
      "grad_norm": 0.0979473665356636,
      "learning_rate": 1.24300038535271e-06,
      "loss": 0.0563,
      "step": 78210
    },
    {
      "epoch": 4.0149688720703125e-06,
      "model_forward_time": 1.7586824893951416,
      "step": 78210
    },
    {
      "dynamics_loss_avg": 0.06218502540141344,
      "epoch": 4.0149688720703125e-06,
      "step": 78210
    },
    {
      "action_loss_avg": 0.0014141238032607361,
      "epoch": 4.0149688720703125e-06,
      "step": 78210
    },
    {
      "epoch": 4.0149688720703125e-06,
      "step": 78210,
      "training_step_time": 7.792075872421265
    },
    {
      "epoch": 4.015922546386719e-06,
      "model_forward_time": 1.7651429176330566,
      "step": 78211
    },
    {
      "epoch": 4.015922546386719e-06,
      "step": 78211,
      "training_step_time": 7.668621063232422
    },
    {
      "epoch": 4.016876220703125e-06,
      "model_forward_time": 1.7625792026519775,
      "step": 78212
    },
    {
      "epoch": 4.016876220703125e-06,
      "step": 78212,
      "training_step_time": 7.606484889984131
    },
    {
      "epoch": 4.017829895019531e-06,
      "model_forward_time": 1.779677391052246,
      "step": 78213
    },
    {
      "epoch": 4.017829895019531e-06,
      "step": 78213,
      "training_step_time": 6.898504018783569
    },
    {
      "epoch": 4.018783569335938e-06,
      "model_forward_time": 1.7677154541015625,
      "step": 78214
    },
    {
      "epoch": 4.018783569335938e-06,
      "step": 78214,
      "training_step_time": 6.848100662231445
    },
    {
      "epoch": 4.019737243652344e-06,
      "model_forward_time": 1.7709119319915771,
      "step": 78215
    },
    {
      "epoch": 4.019737243652344e-06,
      "step": 78215,
      "training_step_time": 7.5839385986328125
    },
    {
      "epoch": 4.02069091796875e-06,
      "model_forward_time": 1.7709195613861084,
      "step": 78216
    },
    {
      "epoch": 4.02069091796875e-06,
      "step": 78216,
      "training_step_time": 7.712034463882446
    },
    {
      "epoch": 4.021644592285156e-06,
      "model_forward_time": 2.473893642425537,
      "step": 78217
    },
    {
      "epoch": 4.021644592285156e-06,
      "step": 78217,
      "training_step_time": 8.790262460708618
    },
    {
      "epoch": 4.0225982666015625e-06,
      "model_forward_time": 1.7652461528778076,
      "step": 78218
    },
    {
      "epoch": 4.0225982666015625e-06,
      "step": 78218,
      "training_step_time": 7.8218865394592285
    },
    {
      "epoch": 4.023551940917969e-06,
      "model_forward_time": 1.7690770626068115,
      "step": 78219
    },
    {
      "epoch": 4.023551940917969e-06,
      "step": 78219,
      "training_step_time": 8.597167730331421
    },
    {
      "epoch": 4.024505615234375e-06,
      "grad_norm": 0.08492806553840637,
      "learning_rate": 1.2419095535879382e-06,
      "loss": 0.0561,
      "step": 78220
    },
    {
      "epoch": 4.024505615234375e-06,
      "model_forward_time": 1.7718732357025146,
      "step": 78220
    },
    {
      "dynamics_loss_avg": 0.06814237423241139,
      "epoch": 4.024505615234375e-06,
      "step": 78220
    },
    {
      "action_loss_avg": 0.00239265191485174,
      "epoch": 4.024505615234375e-06,
      "step": 78220
    },
    {
      "epoch": 4.024505615234375e-06,
      "step": 78220,
      "training_step_time": 7.674960613250732
    },
    {
      "epoch": 4.025459289550781e-06,
      "model_forward_time": 1.766420602798462,
      "step": 78221
    },
    {
      "epoch": 4.025459289550781e-06,
      "step": 78221,
      "training_step_time": 7.585324764251709
    },
    {
      "epoch": 4.026412963867188e-06,
      "model_forward_time": 1.7719659805297852,
      "step": 78222
    },
    {
      "epoch": 4.026412963867188e-06,
      "step": 78222,
      "training_step_time": 7.209612607955933
    },
    {
      "epoch": 4.027366638183594e-06,
      "model_forward_time": 1.769822597503662,
      "step": 78223
    },
    {
      "epoch": 4.027366638183594e-06,
      "step": 78223,
      "training_step_time": 6.952070713043213
    },
    {
      "epoch": 4.0283203125e-06,
      "model_forward_time": 1.7515826225280762,
      "step": 78224
    },
    {
      "epoch": 4.0283203125e-06,
      "step": 78224,
      "training_step_time": 8.326542139053345
    },
    {
      "epoch": 4.029273986816406e-06,
      "model_forward_time": 1.7542212009429932,
      "step": 78225
    },
    {
      "epoch": 4.029273986816406e-06,
      "step": 78225,
      "training_step_time": 7.716807842254639
    },
    {
      "epoch": 4.0302276611328125e-06,
      "model_forward_time": 1.8450994491577148,
      "step": 78226
    },
    {
      "epoch": 4.0302276611328125e-06,
      "step": 78226,
      "training_step_time": 9.662741899490356
    },
    {
      "epoch": 4.031181335449219e-06,
      "model_forward_time": 2.39789080619812,
      "step": 78227
    },
    {
      "epoch": 4.031181335449219e-06,
      "step": 78227,
      "training_step_time": 9.136563539505005
    },
    {
      "epoch": 4.032135009765625e-06,
      "model_forward_time": 1.7535524368286133,
      "step": 78228
    },
    {
      "epoch": 4.032135009765625e-06,
      "step": 78228,
      "training_step_time": 7.7583208084106445
    },
    {
      "epoch": 4.033088684082031e-06,
      "model_forward_time": 1.7573785781860352,
      "step": 78229
    },
    {
      "epoch": 4.033088684082031e-06,
      "step": 78229,
      "training_step_time": 7.719062566757202
    },
    {
      "epoch": 4.034042358398438e-06,
      "grad_norm": 0.0825146809220314,
      "learning_rate": 1.2408191328022878e-06,
      "loss": 0.0559,
      "step": 78230
    },
    {
      "epoch": 4.034042358398438e-06,
      "model_forward_time": 1.7443761825561523,
      "step": 78230
    },
    {
      "dynamics_loss_avg": 0.0437663946300745,
      "epoch": 4.034042358398438e-06,
      "step": 78230
    },
    {
      "action_loss_avg": 0.0018532899324782194,
      "epoch": 4.034042358398438e-06,
      "step": 78230
    },
    {
      "epoch": 4.034042358398438e-06,
      "step": 78230,
      "training_step_time": 8.267855167388916
    },
    {
      "epoch": 4.034996032714844e-06,
      "model_forward_time": 1.7496492862701416,
      "step": 78231
    },
    {
      "epoch": 4.034996032714844e-06,
      "step": 78231,
      "training_step_time": 7.1922383308410645
    },
    {
      "epoch": 4.03594970703125e-06,
      "model_forward_time": 1.7569177150726318,
      "step": 78232
    },
    {
      "epoch": 4.03594970703125e-06,
      "step": 78232,
      "training_step_time": 6.81537389755249
    },
    {
      "epoch": 4.036903381347656e-06,
      "model_forward_time": 1.7630603313446045,
      "step": 78233
    },
    {
      "epoch": 4.036903381347656e-06,
      "step": 78233,
      "training_step_time": 7.600907802581787
    },
    {
      "epoch": 4.0378570556640625e-06,
      "model_forward_time": 1.7515463829040527,
      "step": 78234
    },
    {
      "epoch": 4.0378570556640625e-06,
      "step": 78234,
      "training_step_time": 8.777628660202026
    },
    {
      "epoch": 4.038810729980469e-06,
      "model_forward_time": 1.85843825340271,
      "step": 78235
    },
    {
      "epoch": 4.038810729980469e-06,
      "step": 78235,
      "training_step_time": 8.308396577835083
    },
    {
      "epoch": 4.039764404296875e-06,
      "model_forward_time": 2.4015800952911377,
      "step": 78236
    },
    {
      "epoch": 4.039764404296875e-06,
      "step": 78236,
      "training_step_time": 10.523027658462524
    },
    {
      "epoch": 4.040718078613281e-06,
      "model_forward_time": 1.766235589981079,
      "step": 78237
    },
    {
      "epoch": 4.040718078613281e-06,
      "step": 78237,
      "training_step_time": 7.799952507019043
    },
    {
      "epoch": 4.041671752929688e-06,
      "model_forward_time": 1.3940765857696533,
      "step": 78238
    },
    {
      "epoch": 4.041671752929688e-06,
      "step": 78238,
      "training_step_time": 8.644306898117065
    },
    {
      "epoch": 4.042625427246094e-06,
      "model_forward_time": 1.7691261768341064,
      "step": 78239
    },
    {
      "epoch": 4.042625427246094e-06,
      "step": 78239,
      "training_step_time": 7.662978649139404
    },
    {
      "epoch": 4.0435791015625e-06,
      "grad_norm": 0.12280596047639847,
      "learning_rate": 1.2397291231150066e-06,
      "loss": 0.0555,
      "step": 78240
    },
    {
      "epoch": 4.0435791015625e-06,
      "model_forward_time": 1.7559471130371094,
      "step": 78240
    },
    {
      "dynamics_loss_avg": 0.047814366035163404,
      "epoch": 4.0435791015625e-06,
      "step": 78240
    },
    {
      "action_loss_avg": 0.0020993121492210776,
      "epoch": 4.0435791015625e-06,
      "step": 78240
    },
    {
      "epoch": 4.0435791015625e-06,
      "step": 78240,
      "training_step_time": 7.588536262512207
    },
    {
      "epoch": 4.044532775878906e-06,
      "model_forward_time": 1.4019434452056885,
      "step": 78241
    },
    {
      "epoch": 4.044532775878906e-06,
      "step": 78241,
      "training_step_time": 6.79731011390686
    },
    {
      "epoch": 4.0454864501953125e-06,
      "model_forward_time": 1.7549171447753906,
      "step": 78242
    },
    {
      "epoch": 4.0454864501953125e-06,
      "step": 78242,
      "training_step_time": 7.66948390007019
    },
    {
      "epoch": 4.046440124511719e-06,
      "model_forward_time": 1.7950713634490967,
      "step": 78243
    },
    {
      "epoch": 4.046440124511719e-06,
      "step": 78243,
      "training_step_time": 7.65938663482666
    },
    {
      "epoch": 4.047393798828125e-06,
      "model_forward_time": 1.8285846710205078,
      "step": 78244
    },
    {
      "epoch": 4.047393798828125e-06,
      "step": 78244,
      "training_step_time": 9.467161417007446
    },
    {
      "epoch": 4.048347473144531e-06,
      "model_forward_time": 2.527225971221924,
      "step": 78245
    },
    {
      "epoch": 4.048347473144531e-06,
      "step": 78245,
      "training_step_time": 11.189019441604614
    },
    {
      "epoch": 4.049301147460938e-06,
      "model_forward_time": 1.881612777709961,
      "step": 78246
    },
    {
      "epoch": 4.049301147460938e-06,
      "step": 78246,
      "training_step_time": 9.233141422271729
    },
    {
      "epoch": 4.050254821777344e-06,
      "model_forward_time": 1.901172161102295,
      "step": 78247
    },
    {
      "epoch": 4.050254821777344e-06,
      "step": 78247,
      "training_step_time": 7.658190011978149
    },
    {
      "epoch": 4.05120849609375e-06,
      "model_forward_time": 1.5226008892059326,
      "step": 78248
    },
    {
      "epoch": 4.05120849609375e-06,
      "step": 78248,
      "training_step_time": 7.734657287597656
    },
    {
      "epoch": 4.052162170410156e-06,
      "model_forward_time": 1.9000170230865479,
      "step": 78249
    },
    {
      "epoch": 4.052162170410156e-06,
      "step": 78249,
      "training_step_time": 7.64751410484314
    },
    {
      "epoch": 4.0531158447265625e-06,
      "grad_norm": 0.08693145215511322,
      "learning_rate": 1.2386395246452937e-06,
      "loss": 0.0558,
      "step": 78250
    },
    {
      "epoch": 4.0531158447265625e-06,
      "model_forward_time": 1.5319130420684814,
      "step": 78250
    },
    {
      "dynamics_loss_avg": 0.04024803098291159,
      "epoch": 4.0531158447265625e-06,
      "step": 78250
    },
    {
      "action_loss_avg": 0.0009290158690419048,
      "epoch": 4.0531158447265625e-06,
      "step": 78250
    },
    {
      "epoch": 4.0531158447265625e-06,
      "step": 78250,
      "training_step_time": 6.810080051422119
    },
    {
      "epoch": 4.054069519042969e-06,
      "model_forward_time": 1.8984463214874268,
      "step": 78251
    },
    {
      "epoch": 4.054069519042969e-06,
      "step": 78251,
      "training_step_time": 7.650192737579346
    },
    {
      "epoch": 4.055023193359375e-06,
      "model_forward_time": 1.8799681663513184,
      "step": 78252
    },
    {
      "epoch": 4.055023193359375e-06,
      "step": 78252,
      "training_step_time": 7.5815534591674805
    },
    {
      "epoch": 4.055976867675781e-06,
      "model_forward_time": 1.887176513671875,
      "step": 78253
    },
    {
      "epoch": 4.055976867675781e-06,
      "step": 78253,
      "training_step_time": 8.788316011428833
    },
    {
      "epoch": 4.056930541992187e-06,
      "model_forward_time": 1.9127638339996338,
      "step": 78254
    },
    {
      "epoch": 4.056930541992187e-06,
      "step": 78254,
      "training_step_time": 8.265634536743164
    },
    {
      "epoch": 4.057884216308594e-06,
      "model_forward_time": 1.9309844970703125,
      "step": 78255
    },
    {
      "epoch": 4.057884216308594e-06,
      "step": 78255,
      "training_step_time": 7.8678412437438965
    },
    {
      "epoch": 4.058837890625e-06,
      "model_forward_time": 1.9259746074676514,
      "step": 78256
    },
    {
      "epoch": 4.058837890625e-06,
      "step": 78256,
      "training_step_time": 7.711164951324463
    },
    {
      "epoch": 4.059791564941406e-06,
      "model_forward_time": 1.9266180992126465,
      "step": 78257
    },
    {
      "epoch": 4.059791564941406e-06,
      "step": 78257,
      "training_step_time": 7.7626097202301025
    },
    {
      "epoch": 4.0607452392578125e-06,
      "model_forward_time": 1.9167892932891846,
      "step": 78258
    },
    {
      "epoch": 4.0607452392578125e-06,
      "step": 78258,
      "training_step_time": 7.633860349655151
    },
    {
      "epoch": 4.061698913574219e-06,
      "model_forward_time": 1.929417371749878,
      "step": 78259
    },
    {
      "epoch": 4.061698913574219e-06,
      "step": 78259,
      "training_step_time": 7.224912405014038
    },
    {
      "epoch": 4.062652587890625e-06,
      "grad_norm": 0.07452443987131119,
      "learning_rate": 1.2375503375123082e-06,
      "loss": 0.0559,
      "step": 78260
    },
    {
      "epoch": 4.062652587890625e-06,
      "model_forward_time": 1.911156415939331,
      "step": 78260
    },
    {
      "dynamics_loss_avg": 0.05456014098599553,
      "epoch": 4.062652587890625e-06,
      "step": 78260
    },
    {
      "action_loss_avg": 0.0014375779777765275,
      "epoch": 4.062652587890625e-06,
      "step": 78260
    },
    {
      "epoch": 4.062652587890625e-06,
      "step": 78260,
      "training_step_time": 7.660150766372681
    },
    {
      "epoch": 4.063606262207031e-06,
      "model_forward_time": 1.9089407920837402,
      "step": 78261
    },
    {
      "epoch": 4.063606262207031e-06,
      "step": 78261,
      "training_step_time": 8.114235401153564
    },
    {
      "epoch": 4.064559936523437e-06,
      "model_forward_time": 1.9175257682800293,
      "step": 78262
    },
    {
      "epoch": 4.064559936523437e-06,
      "step": 78262,
      "training_step_time": 8.759751558303833
    },
    {
      "epoch": 4.065513610839844e-06,
      "model_forward_time": 1.9126780033111572,
      "step": 78263
    },
    {
      "epoch": 4.065513610839844e-06,
      "step": 78263,
      "training_step_time": 9.279269933700562
    },
    {
      "epoch": 4.06646728515625e-06,
      "model_forward_time": 2.5719821453094482,
      "step": 78264
    },
    {
      "epoch": 4.06646728515625e-06,
      "step": 78264,
      "training_step_time": 8.555326461791992
    },
    {
      "epoch": 4.067420959472656e-06,
      "model_forward_time": 1.917598009109497,
      "step": 78265
    },
    {
      "epoch": 4.067420959472656e-06,
      "step": 78265,
      "training_step_time": 8.664665460586548
    },
    {
      "epoch": 4.0683746337890625e-06,
      "model_forward_time": 1.9205515384674072,
      "step": 78266
    },
    {
      "epoch": 4.0683746337890625e-06,
      "step": 78266,
      "training_step_time": 7.771684169769287
    },
    {
      "epoch": 4.069328308105469e-06,
      "model_forward_time": 1.9147353172302246,
      "step": 78267
    },
    {
      "epoch": 4.069328308105469e-06,
      "step": 78267,
      "training_step_time": 7.592808723449707
    },
    {
      "epoch": 4.070281982421875e-06,
      "model_forward_time": 1.914083480834961,
      "step": 78268
    },
    {
      "epoch": 4.070281982421875e-06,
      "step": 78268,
      "training_step_time": 7.222108364105225
    },
    {
      "epoch": 4.071235656738281e-06,
      "model_forward_time": 1.9150917530059814,
      "step": 78269
    },
    {
      "epoch": 4.071235656738281e-06,
      "step": 78269,
      "training_step_time": 7.649592399597168
    },
    {
      "epoch": 4.072189331054687e-06,
      "grad_norm": 0.08016740530729294,
      "learning_rate": 1.2364615618351583e-06,
      "loss": 0.0545,
      "step": 78270
    },
    {
      "epoch": 4.072189331054687e-06,
      "model_forward_time": 1.915933609008789,
      "step": 78270
    },
    {
      "dynamics_loss_avg": 0.08049969412386418,
      "epoch": 4.072189331054687e-06,
      "step": 78270
    },
    {
      "action_loss_avg": 0.003642018185928464,
      "epoch": 4.072189331054687e-06,
      "step": 78270
    },
    {
      "epoch": 4.072189331054687e-06,
      "step": 78270,
      "training_step_time": 7.626727104187012
    },
    {
      "epoch": 4.073143005371094e-06,
      "model_forward_time": 1.9257140159606934,
      "step": 78271
    },
    {
      "epoch": 4.073143005371094e-06,
      "step": 78271,
      "training_step_time": 8.33338713645935
    },
    {
      "epoch": 4.0740966796875e-06,
      "model_forward_time": 1.906975507736206,
      "step": 78272
    },
    {
      "epoch": 4.0740966796875e-06,
      "step": 78272,
      "training_step_time": 10.234207391738892
    },
    {
      "epoch": 4.075050354003906e-06,
      "model_forward_time": 2.5610461235046387,
      "step": 78273
    },
    {
      "epoch": 4.075050354003906e-06,
      "step": 78273,
      "training_step_time": 8.500960350036621
    },
    {
      "epoch": 4.0760040283203125e-06,
      "model_forward_time": 1.9163002967834473,
      "step": 78274
    },
    {
      "epoch": 4.0760040283203125e-06,
      "step": 78274,
      "training_step_time": 7.786155462265015
    },
    {
      "epoch": 4.076957702636719e-06,
      "model_forward_time": 1.9140284061431885,
      "step": 78275
    },
    {
      "epoch": 4.076957702636719e-06,
      "step": 78275,
      "training_step_time": 7.8391149044036865
    },
    {
      "epoch": 4.077911376953125e-06,
      "model_forward_time": 1.9115409851074219,
      "step": 78276
    },
    {
      "epoch": 4.077911376953125e-06,
      "step": 78276,
      "training_step_time": 7.667124509811401
    },
    {
      "epoch": 4.078865051269531e-06,
      "model_forward_time": 1.9080569744110107,
      "step": 78277
    },
    {
      "epoch": 4.078865051269531e-06,
      "step": 78277,
      "training_step_time": 7.604593276977539
    },
    {
      "epoch": 4.079818725585937e-06,
      "model_forward_time": 1.9054152965545654,
      "step": 78278
    },
    {
      "epoch": 4.079818725585937e-06,
      "step": 78278,
      "training_step_time": 7.676909923553467
    },
    {
      "epoch": 4.080772399902344e-06,
      "model_forward_time": 1.9213223457336426,
      "step": 78279
    },
    {
      "epoch": 4.080772399902344e-06,
      "step": 78279,
      "training_step_time": 7.772894620895386
    },
    {
      "epoch": 4.08172607421875e-06,
      "grad_norm": 0.09343667328357697,
      "learning_rate": 1.235373197732917e-06,
      "loss": 0.0543,
      "step": 78280
    },
    {
      "epoch": 4.08172607421875e-06,
      "model_forward_time": 1.916893720626831,
      "step": 78280
    },
    {
      "dynamics_loss_avg": 0.05809793099761009,
      "epoch": 4.08172607421875e-06,
      "step": 78280
    },
    {
      "action_loss_avg": 0.0020304484642110765,
      "epoch": 4.08172607421875e-06,
      "step": 78280
    },
    {
      "epoch": 4.08172607421875e-06,
      "step": 78280,
      "training_step_time": 7.684531927108765
    },
    {
      "epoch": 4.082679748535156e-06,
      "model_forward_time": 1.904270887374878,
      "step": 78281
    },
    {
      "epoch": 4.082679748535156e-06,
      "step": 78281,
      "training_step_time": 10.36693787574768
    },
    {
      "epoch": 4.0836334228515625e-06,
      "model_forward_time": 2.5488243103027344,
      "step": 78282
    },
    {
      "epoch": 4.0836334228515625e-06,
      "step": 78282,
      "training_step_time": 7.801732540130615
    },
    {
      "epoch": 4.084587097167969e-06,
      "model_forward_time": 1.5389885902404785,
      "step": 78283
    },
    {
      "epoch": 4.084587097167969e-06,
      "step": 78283,
      "training_step_time": 7.7654876708984375
    },
    {
      "epoch": 4.085540771484375e-06,
      "model_forward_time": 1.9038972854614258,
      "step": 78284
    },
    {
      "epoch": 4.085540771484375e-06,
      "step": 78284,
      "training_step_time": 7.747532606124878
    },
    {
      "epoch": 4.086494445800781e-06,
      "model_forward_time": 1.888655424118042,
      "step": 78285
    },
    {
      "epoch": 4.086494445800781e-06,
      "step": 78285,
      "training_step_time": 7.647192478179932
    },
    {
      "epoch": 4.087448120117187e-06,
      "model_forward_time": 1.9061839580535889,
      "step": 78286
    },
    {
      "epoch": 4.087448120117187e-06,
      "step": 78286,
      "training_step_time": 7.289280414581299
    },
    {
      "epoch": 4.088401794433594e-06,
      "model_forward_time": 1.9027957916259766,
      "step": 78287
    },
    {
      "epoch": 4.088401794433594e-06,
      "step": 78287,
      "training_step_time": 7.4782774448394775
    },
    {
      "epoch": 4.08935546875e-06,
      "model_forward_time": 1.910475254058838,
      "step": 78288
    },
    {
      "epoch": 4.08935546875e-06,
      "step": 78288,
      "training_step_time": 7.673081636428833
    },
    {
      "epoch": 4.090309143066406e-06,
      "model_forward_time": 1.9171786308288574,
      "step": 78289
    },
    {
      "epoch": 4.090309143066406e-06,
      "step": 78289,
      "training_step_time": 9.167745113372803
    },
    {
      "epoch": 4.0912628173828125e-06,
      "grad_norm": 0.07105357199907303,
      "learning_rate": 1.2342852453246007e-06,
      "loss": 0.0537,
      "step": 78290
    },
    {
      "epoch": 4.0912628173828125e-06,
      "model_forward_time": 1.9135456085205078,
      "step": 78290
    },
    {
      "dynamics_loss_avg": 0.059829764254391196,
      "epoch": 4.0912628173828125e-06,
      "step": 78290
    },
    {
      "action_loss_avg": 0.0019727309670997785,
      "epoch": 4.0912628173828125e-06,
      "step": 78290
    },
    {
      "epoch": 4.0912628173828125e-06,
      "step": 78290,
      "training_step_time": 8.617878437042236
    },
    {
      "epoch": 4.092216491699219e-06,
      "model_forward_time": 2.5151641368865967,
      "step": 78291
    },
    {
      "epoch": 4.092216491699219e-06,
      "step": 78291,
      "training_step_time": 9.37586784362793
    },
    {
      "epoch": 4.093170166015625e-06,
      "model_forward_time": 1.9147350788116455,
      "step": 78292
    },
    {
      "epoch": 4.093170166015625e-06,
      "step": 78292,
      "training_step_time": 8.502564907073975
    },
    {
      "epoch": 4.094123840332031e-06,
      "model_forward_time": 1.918203592300415,
      "step": 78293
    },
    {
      "epoch": 4.094123840332031e-06,
      "step": 78293,
      "training_step_time": 7.624399185180664
    },
    {
      "epoch": 4.095077514648437e-06,
      "model_forward_time": 1.9102308750152588,
      "step": 78294
    },
    {
      "epoch": 4.095077514648437e-06,
      "step": 78294,
      "training_step_time": 7.537602424621582
    },
    {
      "epoch": 4.096031188964844e-06,
      "model_forward_time": 1.9084687232971191,
      "step": 78295
    },
    {
      "epoch": 4.096031188964844e-06,
      "step": 78295,
      "training_step_time": 7.157721281051636
    },
    {
      "epoch": 4.09698486328125e-06,
      "model_forward_time": 1.9041564464569092,
      "step": 78296
    },
    {
      "epoch": 4.09698486328125e-06,
      "step": 78296,
      "training_step_time": 7.400456428527832
    },
    {
      "epoch": 4.097938537597656e-06,
      "model_forward_time": 1.9054467678070068,
      "step": 78297
    },
    {
      "epoch": 4.097938537597656e-06,
      "step": 78297,
      "training_step_time": 7.5423829555511475
    },
    {
      "epoch": 4.0988922119140625e-06,
      "model_forward_time": 1.8982319831848145,
      "step": 78298
    },
    {
      "epoch": 4.0988922119140625e-06,
      "step": 78298,
      "training_step_time": 7.57277512550354
    },
    {
      "epoch": 4.099845886230469e-06,
      "model_forward_time": 1.8978180885314941,
      "step": 78299
    },
    {
      "epoch": 4.099845886230469e-06,
      "step": 78299,
      "training_step_time": 10.364019632339478
    },
    {
      "epoch": 4.100799560546875e-06,
      "grad_norm": 0.08717712759971619,
      "learning_rate": 1.23319770472919e-06,
      "loss": 0.0536,
      "step": 78300
    },
    {
      "epoch": 4.100799560546875e-06,
      "model_forward_time": 1.9102580547332764,
      "step": 78300
    },
    {
      "dynamics_loss_avg": 0.05417481251060963,
      "epoch": 4.100799560546875e-06,
      "step": 78300
    },
    {
      "action_loss_avg": 0.0018990329321241007,
      "epoch": 4.100799560546875e-06,
      "step": 78300
    },
    {
      "epoch": 4.100799560546875e-06,
      "step": 78300,
      "training_step_time": 9.321666479110718
    },
    {
      "epoch": 4.101753234863281e-06,
      "model_forward_time": 2.526951789855957,
      "step": 78301
    },
    {
      "epoch": 4.101753234863281e-06,
      "step": 78301,
      "training_step_time": 7.643974781036377
    },
    {
      "epoch": 4.102706909179687e-06,
      "model_forward_time": 1.9057044982910156,
      "step": 78302
    },
    {
      "epoch": 4.102706909179687e-06,
      "step": 78302,
      "training_step_time": 7.658343553543091
    },
    {
      "epoch": 4.103660583496094e-06,
      "model_forward_time": 1.8934745788574219,
      "step": 78303
    },
    {
      "epoch": 4.103660583496094e-06,
      "step": 78303,
      "training_step_time": 8.616400718688965
    },
    {
      "epoch": 4.1046142578125e-06,
      "model_forward_time": 1.8943774700164795,
      "step": 78304
    },
    {
      "epoch": 4.1046142578125e-06,
      "step": 78304,
      "training_step_time": 7.186046361923218
    },
    {
      "epoch": 4.105567932128906e-06,
      "model_forward_time": 1.8969271183013916,
      "step": 78305
    },
    {
      "epoch": 4.105567932128906e-06,
      "step": 78305,
      "training_step_time": 7.479008674621582
    },
    {
      "epoch": 4.1065216064453125e-06,
      "model_forward_time": 1.881659746170044,
      "step": 78306
    },
    {
      "epoch": 4.1065216064453125e-06,
      "step": 78306,
      "training_step_time": 7.401752471923828
    },
    {
      "epoch": 4.107475280761719e-06,
      "model_forward_time": 1.873894214630127,
      "step": 78307
    },
    {
      "epoch": 4.107475280761719e-06,
      "step": 78307,
      "training_step_time": 8.07653284072876
    },
    {
      "epoch": 4.108428955078125e-06,
      "model_forward_time": 1.8893802165985107,
      "step": 78308
    },
    {
      "epoch": 4.108428955078125e-06,
      "step": 78308,
      "training_step_time": 8.081431865692139
    },
    {
      "epoch": 4.109382629394531e-06,
      "model_forward_time": 1.9053988456726074,
      "step": 78309
    },
    {
      "epoch": 4.109382629394531e-06,
      "step": 78309,
      "training_step_time": 9.00589919090271
    },
    {
      "epoch": 4.110336303710937e-06,
      "grad_norm": 0.10367171466350555,
      "learning_rate": 1.2321105760656138e-06,
      "loss": 0.0525,
      "step": 78310
    },
    {
      "epoch": 4.110336303710937e-06,
      "model_forward_time": 2.488568067550659,
      "step": 78310
    },
    {
      "dynamics_loss_avg": 0.058479813672602174,
      "epoch": 4.110336303710937e-06,
      "step": 78310
    },
    {
      "action_loss_avg": 0.002328012965153903,
      "epoch": 4.110336303710937e-06,
      "step": 78310
    },
    {
      "epoch": 4.110336303710937e-06,
      "step": 78310,
      "training_step_time": 7.702381610870361
    },
    {
      "epoch": 4.111289978027344e-06,
      "model_forward_time": 1.8993456363677979,
      "step": 78311
    },
    {
      "epoch": 4.111289978027344e-06,
      "step": 78311,
      "training_step_time": 7.56400203704834
    },
    {
      "epoch": 4.11224365234375e-06,
      "model_forward_time": 1.905362844467163,
      "step": 78312
    },
    {
      "epoch": 4.11224365234375e-06,
      "step": 78312,
      "training_step_time": 7.532933235168457
    },
    {
      "epoch": 4.113197326660156e-06,
      "model_forward_time": 1.8966064453125,
      "step": 78313
    },
    {
      "epoch": 4.113197326660156e-06,
      "step": 78313,
      "training_step_time": 8.230553388595581
    },
    {
      "epoch": 4.1141510009765626e-06,
      "model_forward_time": 1.895841360092163,
      "step": 78314
    },
    {
      "epoch": 4.1141510009765626e-06,
      "step": 78314,
      "training_step_time": 7.362670660018921
    },
    {
      "epoch": 4.115104675292969e-06,
      "model_forward_time": 1.9088404178619385,
      "step": 78315
    },
    {
      "epoch": 4.115104675292969e-06,
      "step": 78315,
      "training_step_time": 7.517637014389038
    },
    {
      "epoch": 4.116058349609375e-06,
      "model_forward_time": 1.9077222347259521,
      "step": 78316
    },
    {
      "epoch": 4.116058349609375e-06,
      "step": 78316,
      "training_step_time": 7.626840353012085
    },
    {
      "epoch": 4.117012023925781e-06,
      "model_forward_time": 1.886277198791504,
      "step": 78317
    },
    {
      "epoch": 4.117012023925781e-06,
      "step": 78317,
      "training_step_time": 8.079763650894165
    },
    {
      "epoch": 4.117965698242187e-06,
      "model_forward_time": 1.8973324298858643,
      "step": 78318
    },
    {
      "epoch": 4.117965698242187e-06,
      "step": 78318,
      "training_step_time": 8.841683626174927
    },
    {
      "epoch": 4.118919372558594e-06,
      "model_forward_time": 2.4637367725372314,
      "step": 78319
    },
    {
      "epoch": 4.118919372558594e-06,
      "step": 78319,
      "training_step_time": 8.449492692947388
    },
    {
      "epoch": 4.119873046875e-06,
      "grad_norm": 0.23157791793346405,
      "learning_rate": 1.2310238594527613e-06,
      "loss": 0.0531,
      "step": 78320
    },
    {
      "epoch": 4.119873046875e-06,
      "model_forward_time": 1.8988697528839111,
      "step": 78320
    },
    {
      "dynamics_loss_avg": 0.054843501187860966,
      "epoch": 4.119873046875e-06,
      "step": 78320
    },
    {
      "action_loss_avg": 0.0009703075716970489,
      "epoch": 4.119873046875e-06,
      "step": 78320
    },
    {
      "epoch": 4.119873046875e-06,
      "step": 78320,
      "training_step_time": 8.090861320495605
    },
    {
      "epoch": 4.120826721191406e-06,
      "model_forward_time": 1.5401389598846436,
      "step": 78321
    },
    {
      "epoch": 4.120826721191406e-06,
      "step": 78321,
      "training_step_time": 7.809979200363159
    },
    {
      "epoch": 4.1217803955078126e-06,
      "model_forward_time": 1.9110023975372314,
      "step": 78322
    },
    {
      "epoch": 4.1217803955078126e-06,
      "step": 78322,
      "training_step_time": 7.706636190414429
    },
    {
      "epoch": 4.122734069824219e-06,
      "model_forward_time": 1.901472806930542,
      "step": 78323
    },
    {
      "epoch": 4.122734069824219e-06,
      "step": 78323,
      "training_step_time": 8.13592791557312
    },
    {
      "epoch": 4.123687744140625e-06,
      "model_forward_time": 1.9113669395446777,
      "step": 78324
    },
    {
      "epoch": 4.123687744140625e-06,
      "step": 78324,
      "training_step_time": 7.287122011184692
    },
    {
      "epoch": 4.124641418457031e-06,
      "model_forward_time": 1.9022386074066162,
      "step": 78325
    },
    {
      "epoch": 4.124641418457031e-06,
      "step": 78325,
      "training_step_time": 8.435745477676392
    },
    {
      "epoch": 4.125595092773437e-06,
      "model_forward_time": 1.9060990810394287,
      "step": 78326
    },
    {
      "epoch": 4.125595092773437e-06,
      "step": 78326,
      "training_step_time": 7.587900638580322
    },
    {
      "epoch": 4.126548767089844e-06,
      "model_forward_time": 1.9119255542755127,
      "step": 78327
    },
    {
      "epoch": 4.126548767089844e-06,
      "step": 78327,
      "training_step_time": 9.653580904006958
    },
    {
      "epoch": 4.12750244140625e-06,
      "model_forward_time": 2.525153160095215,
      "step": 78328
    },
    {
      "epoch": 4.12750244140625e-06,
      "step": 78328,
      "training_step_time": 8.345462322235107
    },
    {
      "epoch": 4.128456115722656e-06,
      "model_forward_time": 1.8830554485321045,
      "step": 78329
    },
    {
      "epoch": 4.128456115722656e-06,
      "step": 78329,
      "training_step_time": 8.830873489379883
    },
    {
      "epoch": 4.1294097900390626e-06,
      "grad_norm": 0.07365187257528305,
      "learning_rate": 1.2299375550094694e-06,
      "loss": 0.0537,
      "step": 78330
    },
    {
      "epoch": 4.1294097900390626e-06,
      "model_forward_time": 1.8971683979034424,
      "step": 78330
    },
    {
      "dynamics_loss_avg": 0.06479844637215137,
      "epoch": 4.1294097900390626e-06,
      "step": 78330
    },
    {
      "action_loss_avg": 0.0018790951115079224,
      "epoch": 4.1294097900390626e-06,
      "step": 78330
    },
    {
      "epoch": 4.1294097900390626e-06,
      "step": 78330,
      "training_step_time": 7.726085186004639
    },
    {
      "epoch": 4.130363464355469e-06,
      "model_forward_time": 1.8922827243804932,
      "step": 78331
    },
    {
      "epoch": 4.130363464355469e-06,
      "step": 78331,
      "training_step_time": 8.734776258468628
    },
    {
      "epoch": 4.131317138671875e-06,
      "model_forward_time": 1.897554636001587,
      "step": 78332
    },
    {
      "epoch": 4.131317138671875e-06,
      "step": 78332,
      "training_step_time": 7.597696542739868
    },
    {
      "epoch": 4.1322708129882814e-06,
      "model_forward_time": 1.894054889678955,
      "step": 78333
    },
    {
      "epoch": 4.1322708129882814e-06,
      "step": 78333,
      "training_step_time": 7.301961898803711
    },
    {
      "epoch": 4.133224487304687e-06,
      "model_forward_time": 1.8859977722167969,
      "step": 78334
    },
    {
      "epoch": 4.133224487304687e-06,
      "step": 78334,
      "training_step_time": 7.730010509490967
    },
    {
      "epoch": 4.134178161621094e-06,
      "model_forward_time": 1.8943498134613037,
      "step": 78335
    },
    {
      "epoch": 4.134178161621094e-06,
      "step": 78335,
      "training_step_time": 8.5780348777771
    },
    {
      "epoch": 4.1351318359375e-06,
      "model_forward_time": 1.892188310623169,
      "step": 78336
    },
    {
      "epoch": 4.1351318359375e-06,
      "step": 78336,
      "training_step_time": 10.502596855163574
    },
    {
      "epoch": 4.136085510253906e-06,
      "model_forward_time": 2.4927847385406494,
      "step": 78337
    },
    {
      "epoch": 4.136085510253906e-06,
      "step": 78337,
      "training_step_time": 10.005335807800293
    },
    {
      "epoch": 4.1370391845703126e-06,
      "model_forward_time": 1.893697738647461,
      "step": 78338
    },
    {
      "epoch": 4.1370391845703126e-06,
      "step": 78338,
      "training_step_time": 7.900590419769287
    },
    {
      "epoch": 4.137992858886719e-06,
      "model_forward_time": 1.9000592231750488,
      "step": 78339
    },
    {
      "epoch": 4.137992858886719e-06,
      "step": 78339,
      "training_step_time": 7.825505256652832
    },
    {
      "epoch": 4.138946533203125e-06,
      "grad_norm": 0.13670110702514648,
      "learning_rate": 1.228851662854542e-06,
      "loss": 0.0542,
      "step": 78340
    },
    {
      "epoch": 4.138946533203125e-06,
      "model_forward_time": 1.8911609649658203,
      "step": 78340
    },
    {
      "dynamics_loss_avg": 0.05264761997386813,
      "epoch": 4.138946533203125e-06,
      "step": 78340
    },
    {
      "action_loss_avg": 0.001468001282773912,
      "epoch": 4.138946533203125e-06,
      "step": 78340
    },
    {
      "epoch": 4.138946533203125e-06,
      "step": 78340,
      "training_step_time": 7.819689750671387
    },
    {
      "epoch": 4.1399002075195314e-06,
      "model_forward_time": 1.8860492706298828,
      "step": 78341
    },
    {
      "epoch": 4.1399002075195314e-06,
      "step": 78341,
      "training_step_time": 7.6878437995910645
    },
    {
      "epoch": 4.140853881835937e-06,
      "model_forward_time": 1.8729181289672852,
      "step": 78342
    },
    {
      "epoch": 4.140853881835937e-06,
      "step": 78342,
      "training_step_time": 7.407481670379639
    },
    {
      "epoch": 4.141807556152344e-06,
      "model_forward_time": 1.8972301483154297,
      "step": 78343
    },
    {
      "epoch": 4.141807556152344e-06,
      "step": 78343,
      "training_step_time": 7.716158151626587
    },
    {
      "epoch": 4.14276123046875e-06,
      "model_forward_time": 1.8942549228668213,
      "step": 78344
    },
    {
      "epoch": 4.14276123046875e-06,
      "step": 78344,
      "training_step_time": 8.532969951629639
    },
    {
      "epoch": 4.143714904785156e-06,
      "model_forward_time": 1.900787115097046,
      "step": 78345
    },
    {
      "epoch": 4.143714904785156e-06,
      "step": 78345,
      "training_step_time": 8.42484188079834
    },
    {
      "epoch": 4.144668579101563e-06,
      "model_forward_time": 1.8877897262573242,
      "step": 78346
    },
    {
      "epoch": 4.144668579101563e-06,
      "step": 78346,
      "training_step_time": 8.414368152618408
    },
    {
      "epoch": 4.145622253417969e-06,
      "model_forward_time": 2.5102851390838623,
      "step": 78347
    },
    {
      "epoch": 4.145622253417969e-06,
      "step": 78347,
      "training_step_time": 7.862207889556885
    },
    {
      "epoch": 4.146575927734375e-06,
      "model_forward_time": 1.8904080390930176,
      "step": 78348
    },
    {
      "epoch": 4.146575927734375e-06,
      "step": 78348,
      "training_step_time": 7.765548229217529
    },
    {
      "epoch": 4.1475296020507814e-06,
      "model_forward_time": 1.8903770446777344,
      "step": 78349
    },
    {
      "epoch": 4.1475296020507814e-06,
      "step": 78349,
      "training_step_time": 7.883151292800903
    },
    {
      "epoch": 4.148483276367187e-06,
      "grad_norm": 0.08217055350542068,
      "learning_rate": 1.2277661831067255e-06,
      "loss": 0.0531,
      "step": 78350
    },
    {
      "epoch": 4.148483276367187e-06,
      "model_forward_time": 1.893664836883545,
      "step": 78350
    },
    {
      "dynamics_loss_avg": 0.045593665540218355,
      "epoch": 4.148483276367187e-06,
      "step": 78350
    },
    {
      "action_loss_avg": 0.0012960804364411161,
      "epoch": 4.148483276367187e-06,
      "step": 78350
    },
    {
      "epoch": 4.148483276367187e-06,
      "step": 78350,
      "training_step_time": 7.723947286605835
    },
    {
      "epoch": 4.149436950683594e-06,
      "model_forward_time": 1.9065062999725342,
      "step": 78351
    },
    {
      "epoch": 4.149436950683594e-06,
      "step": 78351,
      "training_step_time": 7.461576223373413
    },
    {
      "epoch": 4.150390625e-06,
      "model_forward_time": 1.9003832340240479,
      "step": 78352
    },
    {
      "epoch": 4.150390625e-06,
      "step": 78352,
      "training_step_time": 7.536873817443848
    },
    {
      "epoch": 4.151344299316406e-06,
      "model_forward_time": 1.8891899585723877,
      "step": 78353
    },
    {
      "epoch": 4.151344299316406e-06,
      "step": 78353,
      "training_step_time": 8.606032848358154
    },
    {
      "epoch": 4.152297973632813e-06,
      "model_forward_time": 1.9061625003814697,
      "step": 78354
    },
    {
      "epoch": 4.152297973632813e-06,
      "step": 78354,
      "training_step_time": 7.750831127166748
    },
    {
      "epoch": 4.153251647949219e-06,
      "model_forward_time": 1.8995347023010254,
      "step": 78355
    },
    {
      "epoch": 4.153251647949219e-06,
      "step": 78355,
      "training_step_time": 9.086292505264282
    },
    {
      "epoch": 4.154205322265625e-06,
      "model_forward_time": 2.4922332763671875,
      "step": 78356
    },
    {
      "epoch": 4.154205322265625e-06,
      "step": 78356,
      "training_step_time": 8.631349325180054
    },
    {
      "epoch": 4.1551589965820314e-06,
      "model_forward_time": 1.8851814270019531,
      "step": 78357
    },
    {
      "epoch": 4.1551589965820314e-06,
      "step": 78357,
      "training_step_time": 7.889357328414917
    },
    {
      "epoch": 4.156112670898437e-06,
      "model_forward_time": 1.9016597270965576,
      "step": 78358
    },
    {
      "epoch": 4.156112670898437e-06,
      "step": 78358,
      "training_step_time": 7.907781600952148
    },
    {
      "epoch": 4.157066345214844e-06,
      "model_forward_time": 1.9092328548431396,
      "step": 78359
    },
    {
      "epoch": 4.157066345214844e-06,
      "step": 78359,
      "training_step_time": 7.711922645568848
    },
    {
      "epoch": 4.15802001953125e-06,
      "grad_norm": 0.08615840971469879,
      "learning_rate": 1.2266811158847285e-06,
      "loss": 0.0535,
      "step": 78360
    },
    {
      "epoch": 4.15802001953125e-06,
      "model_forward_time": 1.8792643547058105,
      "step": 78360
    },
    {
      "dynamics_loss_avg": 0.061679930239915845,
      "epoch": 4.15802001953125e-06,
      "step": 78360
    },
    {
      "action_loss_avg": 0.0013445140386465937,
      "epoch": 4.15802001953125e-06,
      "step": 78360
    },
    {
      "epoch": 4.15802001953125e-06,
      "step": 78360,
      "training_step_time": 7.259449243545532
    },
    {
      "epoch": 4.158973693847656e-06,
      "model_forward_time": 1.9004170894622803,
      "step": 78361
    },
    {
      "epoch": 4.158973693847656e-06,
      "step": 78361,
      "training_step_time": 7.576743125915527
    },
    {
      "epoch": 4.159927368164063e-06,
      "model_forward_time": 1.893681526184082,
      "step": 78362
    },
    {
      "epoch": 4.159927368164063e-06,
      "step": 78362,
      "training_step_time": 7.717540740966797
    },
    {
      "epoch": 4.160881042480469e-06,
      "model_forward_time": 1.900646686553955,
      "step": 78363
    },
    {
      "epoch": 4.160881042480469e-06,
      "step": 78363,
      "training_step_time": 7.687222957611084
    },
    {
      "epoch": 4.161834716796875e-06,
      "model_forward_time": 1.90138578414917,
      "step": 78364
    },
    {
      "epoch": 4.161834716796875e-06,
      "step": 78364,
      "training_step_time": 8.777109384536743
    },
    {
      "epoch": 4.1627883911132815e-06,
      "model_forward_time": 2.4873287677764893,
      "step": 78365
    },
    {
      "epoch": 4.1627883911132815e-06,
      "step": 78365,
      "training_step_time": 8.811956882476807
    },
    {
      "epoch": 4.163742065429687e-06,
      "model_forward_time": 1.8945224285125732,
      "step": 78366
    },
    {
      "epoch": 4.163742065429687e-06,
      "step": 78366,
      "training_step_time": 10.255516767501831
    },
    {
      "epoch": 4.164695739746094e-06,
      "model_forward_time": 1.8991219997406006,
      "step": 78367
    },
    {
      "epoch": 4.164695739746094e-06,
      "step": 78367,
      "training_step_time": 7.760254383087158
    },
    {
      "epoch": 4.1656494140625e-06,
      "model_forward_time": 1.9053466320037842,
      "step": 78368
    },
    {
      "epoch": 4.1656494140625e-06,
      "step": 78368,
      "training_step_time": 8.627007961273193
    },
    {
      "epoch": 4.166603088378906e-06,
      "model_forward_time": 1.900406837463379,
      "step": 78369
    },
    {
      "epoch": 4.166603088378906e-06,
      "step": 78369,
      "training_step_time": 7.273448944091797
    },
    {
      "epoch": 4.167556762695313e-06,
      "grad_norm": 0.10727603733539581,
      "learning_rate": 1.2255964613072107e-06,
      "loss": 0.054,
      "step": 78370
    },
    {
      "epoch": 4.167556762695313e-06,
      "model_forward_time": 1.8963799476623535,
      "step": 78370
    },
    {
      "dynamics_loss_avg": 0.05605711154639721,
      "epoch": 4.167556762695313e-06,
      "step": 78370
    },
    {
      "action_loss_avg": 0.002114084135973826,
      "epoch": 4.167556762695313e-06,
      "step": 78370
    },
    {
      "epoch": 4.167556762695313e-06,
      "step": 78370,
      "training_step_time": 7.4346604347229
    },
    {
      "epoch": 4.168510437011719e-06,
      "model_forward_time": 1.9006223678588867,
      "step": 78371
    },
    {
      "epoch": 4.168510437011719e-06,
      "step": 78371,
      "training_step_time": 7.648310422897339
    },
    {
      "epoch": 4.169464111328125e-06,
      "model_forward_time": 1.9094007015228271,
      "step": 78372
    },
    {
      "epoch": 4.169464111328125e-06,
      "step": 78372,
      "training_step_time": 8.718440055847168
    },
    {
      "epoch": 4.1704177856445315e-06,
      "model_forward_time": 1.8924636840820312,
      "step": 78373
    },
    {
      "epoch": 4.1704177856445315e-06,
      "step": 78373,
      "training_step_time": 8.569232702255249
    },
    {
      "epoch": 4.171371459960937e-06,
      "model_forward_time": 2.5407462120056152,
      "step": 78374
    },
    {
      "epoch": 4.171371459960937e-06,
      "step": 78374,
      "training_step_time": 8.673735857009888
    },
    {
      "epoch": 4.172325134277344e-06,
      "model_forward_time": 1.895751714706421,
      "step": 78375
    },
    {
      "epoch": 4.172325134277344e-06,
      "step": 78375,
      "training_step_time": 8.736245155334473
    },
    {
      "epoch": 4.17327880859375e-06,
      "model_forward_time": 1.8887641429901123,
      "step": 78376
    },
    {
      "epoch": 4.17327880859375e-06,
      "step": 78376,
      "training_step_time": 7.710953235626221
    },
    {
      "epoch": 4.174232482910156e-06,
      "model_forward_time": 1.8989083766937256,
      "step": 78377
    },
    {
      "epoch": 4.174232482910156e-06,
      "step": 78377,
      "training_step_time": 7.613832235336304
    },
    {
      "epoch": 4.175186157226563e-06,
      "model_forward_time": 1.8987925052642822,
      "step": 78378
    },
    {
      "epoch": 4.175186157226563e-06,
      "step": 78378,
      "training_step_time": 7.61620020866394
    },
    {
      "epoch": 4.176139831542968e-06,
      "model_forward_time": 1.8955349922180176,
      "step": 78379
    },
    {
      "epoch": 4.176139831542968e-06,
      "step": 78379,
      "training_step_time": 7.296236753463745
    },
    {
      "epoch": 4.177093505859375e-06,
      "grad_norm": 0.07238050550222397,
      "learning_rate": 1.2245122194927905e-06,
      "loss": 0.0544,
      "step": 78380
    },
    {
      "epoch": 4.177093505859375e-06,
      "model_forward_time": 1.527306318283081,
      "step": 78380
    },
    {
      "dynamics_loss_avg": 0.0475923964753747,
      "epoch": 4.177093505859375e-06,
      "step": 78380
    },
    {
      "action_loss_avg": 0.001465549785643816,
      "epoch": 4.177093505859375e-06,
      "step": 78380
    },
    {
      "epoch": 4.177093505859375e-06,
      "step": 78380,
      "training_step_time": 8.333804607391357
    },
    {
      "epoch": 4.1780471801757815e-06,
      "model_forward_time": 1.9056737422943115,
      "step": 78381
    },
    {
      "epoch": 4.1780471801757815e-06,
      "step": 78381,
      "training_step_time": 9.055308103561401
    },
    {
      "epoch": 4.179000854492187e-06,
      "model_forward_time": 1.8934481143951416,
      "step": 78382
    },
    {
      "epoch": 4.179000854492187e-06,
      "step": 78382,
      "training_step_time": 8.144838333129883
    },
    {
      "epoch": 4.179954528808594e-06,
      "model_forward_time": 1.901350975036621,
      "step": 78383
    },
    {
      "epoch": 4.179954528808594e-06,
      "step": 78383,
      "training_step_time": 9.122657537460327
    },
    {
      "epoch": 4.180908203125e-06,
      "model_forward_time": 1.8844349384307861,
      "step": 78384
    },
    {
      "epoch": 4.180908203125e-06,
      "step": 78384,
      "training_step_time": 7.788161277770996
    },
    {
      "epoch": 4.181861877441406e-06,
      "model_forward_time": 1.8829569816589355,
      "step": 78385
    },
    {
      "epoch": 4.181861877441406e-06,
      "step": 78385,
      "training_step_time": 7.733640432357788
    },
    {
      "epoch": 4.182815551757813e-06,
      "model_forward_time": 1.9041309356689453,
      "step": 78386
    },
    {
      "epoch": 4.182815551757813e-06,
      "step": 78386,
      "training_step_time": 7.631203889846802
    },
    {
      "epoch": 4.183769226074218e-06,
      "model_forward_time": 1.8953790664672852,
      "step": 78387
    },
    {
      "epoch": 4.183769226074218e-06,
      "step": 78387,
      "training_step_time": 7.658754110336304
    },
    {
      "epoch": 4.184722900390625e-06,
      "model_forward_time": 1.8942701816558838,
      "step": 78388
    },
    {
      "epoch": 4.184722900390625e-06,
      "step": 78388,
      "training_step_time": 7.344422340393066
    },
    {
      "epoch": 4.1856765747070315e-06,
      "model_forward_time": 1.8784775733947754,
      "step": 78389
    },
    {
      "epoch": 4.1856765747070315e-06,
      "step": 78389,
      "training_step_time": 8.230712652206421
    },
    {
      "epoch": 4.186630249023437e-06,
      "grad_norm": 0.08275086432695389,
      "learning_rate": 1.2234283905600365e-06,
      "loss": 0.0568,
      "step": 78390
    },
    {
      "epoch": 4.186630249023437e-06,
      "model_forward_time": 1.8919191360473633,
      "step": 78390
    },
    {
      "dynamics_loss_avg": 0.058557191677391526,
      "epoch": 4.186630249023437e-06,
      "step": 78390
    },
    {
      "action_loss_avg": 0.0021686088817659765,
      "epoch": 4.186630249023437e-06,
      "step": 78390
    },
    {
      "epoch": 4.186630249023437e-06,
      "step": 78390,
      "training_step_time": 7.5973992347717285
    },
    {
      "epoch": 4.187583923339844e-06,
      "model_forward_time": 1.8930022716522217,
      "step": 78391
    },
    {
      "epoch": 4.187583923339844e-06,
      "step": 78391,
      "training_step_time": 8.696455001831055
    },
    {
      "epoch": 4.18853759765625e-06,
      "model_forward_time": 1.892524003982544,
      "step": 78392
    },
    {
      "epoch": 4.18853759765625e-06,
      "step": 78392,
      "training_step_time": 7.741546154022217
    },
    {
      "epoch": 4.189491271972656e-06,
      "model_forward_time": 2.490821599960327,
      "step": 78393
    },
    {
      "epoch": 4.189491271972656e-06,
      "step": 78393,
      "training_step_time": 7.553088188171387
    },
    {
      "epoch": 4.190444946289063e-06,
      "model_forward_time": 1.8906159400939941,
      "step": 78394
    },
    {
      "epoch": 4.190444946289063e-06,
      "step": 78394,
      "training_step_time": 8.160560369491577
    },
    {
      "epoch": 4.191398620605468e-06,
      "model_forward_time": 1.9013090133666992,
      "step": 78395
    },
    {
      "epoch": 4.191398620605468e-06,
      "step": 78395,
      "training_step_time": 8.474159002304077
    },
    {
      "epoch": 4.192352294921875e-06,
      "model_forward_time": 1.8895461559295654,
      "step": 78396
    },
    {
      "epoch": 4.192352294921875e-06,
      "step": 78396,
      "training_step_time": 7.512383460998535
    },
    {
      "epoch": 4.1933059692382815e-06,
      "model_forward_time": 1.9028637409210205,
      "step": 78397
    },
    {
      "epoch": 4.1933059692382815e-06,
      "step": 78397,
      "training_step_time": 7.494965553283691
    },
    {
      "epoch": 4.194259643554687e-06,
      "model_forward_time": 1.8762226104736328,
      "step": 78398
    },
    {
      "epoch": 4.194259643554687e-06,
      "step": 78398,
      "training_step_time": 7.492915630340576
    },
    {
      "epoch": 4.195213317871094e-06,
      "model_forward_time": 1.9131324291229248,
      "step": 78399
    },
    {
      "epoch": 4.195213317871094e-06,
      "step": 78399,
      "training_step_time": 8.587191820144653
    },
    {
      "epoch": 4.1961669921875e-06,
      "grad_norm": 0.07330282777547836,
      "learning_rate": 1.2223449746274752e-06,
      "loss": 0.0545,
      "step": 78400
    },
    {
      "epoch": 4.1961669921875e-06,
      "model_forward_time": 1.8961122035980225,
      "step": 78400
    },
    {
      "dynamics_loss_avg": 0.07546585239470005,
      "epoch": 4.1961669921875e-06,
      "step": 78400
    },
    {
      "action_loss_avg": 0.0021481041563674807,
      "epoch": 4.1961669921875e-06,
      "step": 78400
    },
    {
      "epoch": 4.1961669921875e-06,
      "step": 78400,
      "training_step_time": 8.52664303779602
    },
    {
      "epoch": 4.197120666503906e-06,
      "model_forward_time": 1.8988792896270752,
      "step": 78401
    },
    {
      "epoch": 4.197120666503906e-06,
      "step": 78401,
      "training_step_time": 8.157591342926025
    },
    {
      "epoch": 4.198074340820313e-06,
      "model_forward_time": 2.4652984142303467,
      "step": 78402
    },
    {
      "epoch": 4.198074340820313e-06,
      "step": 78402,
      "training_step_time": 8.317267656326294
    },
    {
      "epoch": 4.199028015136718e-06,
      "model_forward_time": 1.904515027999878,
      "step": 78403
    },
    {
      "epoch": 4.199028015136718e-06,
      "step": 78403,
      "training_step_time": 7.599337100982666
    },
    {
      "epoch": 4.199981689453125e-06,
      "model_forward_time": 1.8932220935821533,
      "step": 78404
    },
    {
      "epoch": 4.199981689453125e-06,
      "step": 78404,
      "training_step_time": 7.620831489562988
    },
    {
      "epoch": 4.2009353637695315e-06,
      "model_forward_time": 1.903184413909912,
      "step": 78405
    },
    {
      "epoch": 4.2009353637695315e-06,
      "step": 78405,
      "training_step_time": 9.14658498764038
    },
    {
      "epoch": 4.201889038085937e-06,
      "model_forward_time": 1.8997511863708496,
      "step": 78406
    },
    {
      "epoch": 4.201889038085937e-06,
      "step": 78406,
      "training_step_time": 7.485659122467041
    },
    {
      "epoch": 4.202842712402344e-06,
      "model_forward_time": 1.9102087020874023,
      "step": 78407
    },
    {
      "epoch": 4.202842712402344e-06,
      "step": 78407,
      "training_step_time": 7.278125524520874
    },
    {
      "epoch": 4.20379638671875e-06,
      "model_forward_time": 1.8974955081939697,
      "step": 78408
    },
    {
      "epoch": 4.20379638671875e-06,
      "step": 78408,
      "training_step_time": 7.628661155700684
    },
    {
      "epoch": 4.204750061035156e-06,
      "model_forward_time": 1.897993803024292,
      "step": 78409
    },
    {
      "epoch": 4.204750061035156e-06,
      "step": 78409,
      "training_step_time": 7.646478176116943
    },
    {
      "epoch": 4.205703735351563e-06,
      "grad_norm": 0.0876973569393158,
      "learning_rate": 1.2212619718135877e-06,
      "loss": 0.0551,
      "step": 78410
    },
    {
      "epoch": 4.205703735351563e-06,
      "model_forward_time": 1.8916311264038086,
      "step": 78410
    },
    {
      "dynamics_loss_avg": 0.05664450246840715,
      "epoch": 4.205703735351563e-06,
      "step": 78410
    },
    {
      "action_loss_avg": 0.0015249351621605457,
      "epoch": 4.205703735351563e-06,
      "step": 78410
    },
    {
      "epoch": 4.205703735351563e-06,
      "step": 78410,
      "training_step_time": 8.56061840057373
    },
    {
      "epoch": 4.206657409667968e-06,
      "model_forward_time": 2.536992073059082,
      "step": 78411
    },
    {
      "epoch": 4.206657409667968e-06,
      "step": 78411,
      "training_step_time": 9.591766595840454
    },
    {
      "epoch": 4.207611083984375e-06,
      "model_forward_time": 1.894327163696289,
      "step": 78412
    },
    {
      "epoch": 4.207611083984375e-06,
      "step": 78412,
      "training_step_time": 7.640043020248413
    },
    {
      "epoch": 4.2085647583007815e-06,
      "model_forward_time": 1.9001388549804688,
      "step": 78413
    },
    {
      "epoch": 4.2085647583007815e-06,
      "step": 78413,
      "training_step_time": 7.6351423263549805
    },
    {
      "epoch": 4.209518432617187e-06,
      "model_forward_time": 1.8895065784454346,
      "step": 78414
    },
    {
      "epoch": 4.209518432617187e-06,
      "step": 78414,
      "training_step_time": 7.421327829360962
    },
    {
      "epoch": 4.210472106933594e-06,
      "model_forward_time": 1.8825325965881348,
      "step": 78415
    },
    {
      "epoch": 4.210472106933594e-06,
      "step": 78415,
      "training_step_time": 7.454015493392944
    },
    {
      "epoch": 4.21142578125e-06,
      "model_forward_time": 1.887843370437622,
      "step": 78416
    },
    {
      "epoch": 4.21142578125e-06,
      "step": 78416,
      "training_step_time": 7.236691474914551
    },
    {
      "epoch": 4.212379455566406e-06,
      "model_forward_time": 1.9165873527526855,
      "step": 78417
    },
    {
      "epoch": 4.212379455566406e-06,
      "step": 78417,
      "training_step_time": 7.653982877731323
    },
    {
      "epoch": 4.213333129882813e-06,
      "model_forward_time": 1.891833782196045,
      "step": 78418
    },
    {
      "epoch": 4.213333129882813e-06,
      "step": 78418,
      "training_step_time": 7.639509439468384
    },
    {
      "epoch": 4.214286804199218e-06,
      "model_forward_time": 1.8949224948883057,
      "step": 78419
    },
    {
      "epoch": 4.214286804199218e-06,
      "step": 78419,
      "training_step_time": 10.367894649505615
    },
    {
      "epoch": 4.215240478515625e-06,
      "grad_norm": 0.09369181096553802,
      "learning_rate": 1.220179382236812e-06,
      "loss": 0.0565,
      "step": 78420
    },
    {
      "epoch": 4.215240478515625e-06,
      "model_forward_time": 2.532888889312744,
      "step": 78420
    },
    {
      "dynamics_loss_avg": 0.0413274584338069,
      "epoch": 4.215240478515625e-06,
      "step": 78420
    },
    {
      "action_loss_avg": 0.0009913120709825307,
      "epoch": 4.215240478515625e-06,
      "step": 78420
    },
    {
      "epoch": 4.215240478515625e-06,
      "step": 78420,
      "training_step_time": 8.36424994468689
    },
    {
      "epoch": 4.2161941528320315e-06,
      "model_forward_time": 1.5206611156463623,
      "step": 78421
    },
    {
      "epoch": 4.2161941528320315e-06,
      "step": 78421,
      "training_step_time": 7.732634544372559
    },
    {
      "epoch": 4.217147827148437e-06,
      "model_forward_time": 1.8991012573242188,
      "step": 78422
    },
    {
      "epoch": 4.217147827148437e-06,
      "step": 78422,
      "training_step_time": 7.706672430038452
    },
    {
      "epoch": 4.218101501464844e-06,
      "model_forward_time": 1.905961275100708,
      "step": 78423
    },
    {
      "epoch": 4.218101501464844e-06,
      "step": 78423,
      "training_step_time": 9.486219882965088
    },
    {
      "epoch": 4.21905517578125e-06,
      "model_forward_time": 1.8964893817901611,
      "step": 78424
    },
    {
      "epoch": 4.21905517578125e-06,
      "step": 78424,
      "training_step_time": 7.532435894012451
    },
    {
      "epoch": 4.220008850097656e-06,
      "model_forward_time": 1.8958892822265625,
      "step": 78425
    },
    {
      "epoch": 4.220008850097656e-06,
      "step": 78425,
      "training_step_time": 7.194223880767822
    },
    {
      "epoch": 4.220962524414063e-06,
      "model_forward_time": 1.9058754444122314,
      "step": 78426
    },
    {
      "epoch": 4.220962524414063e-06,
      "step": 78426,
      "training_step_time": 7.690221309661865
    },
    {
      "epoch": 4.221916198730468e-06,
      "model_forward_time": 1.8995280265808105,
      "step": 78427
    },
    {
      "epoch": 4.221916198730468e-06,
      "step": 78427,
      "training_step_time": 7.629805088043213
    },
    {
      "epoch": 4.222869873046875e-06,
      "model_forward_time": 1.8950605392456055,
      "step": 78428
    },
    {
      "epoch": 4.222869873046875e-06,
      "step": 78428,
      "training_step_time": 8.422417163848877
    },
    {
      "epoch": 4.2238235473632815e-06,
      "model_forward_time": 1.895275592803955,
      "step": 78429
    },
    {
      "epoch": 4.2238235473632815e-06,
      "step": 78429,
      "training_step_time": 9.29265809059143
    },
    {
      "epoch": 4.224777221679687e-06,
      "grad_norm": 0.10078011453151703,
      "learning_rate": 1.2190972060155337e-06,
      "loss": 0.0556,
      "step": 78430
    },
    {
      "epoch": 4.224777221679687e-06,
      "model_forward_time": 2.505366325378418,
      "step": 78430
    },
    {
      "dynamics_loss_avg": 0.058095873519778254,
      "epoch": 4.224777221679687e-06,
      "step": 78430
    },
    {
      "action_loss_avg": 0.0013673092937096954,
      "epoch": 4.224777221679687e-06,
      "step": 78430
    },
    {
      "epoch": 4.224777221679687e-06,
      "step": 78430,
      "training_step_time": 8.877074956893921
    },
    {
      "epoch": 4.225730895996094e-06,
      "model_forward_time": 1.903801441192627,
      "step": 78431
    },
    {
      "epoch": 4.225730895996094e-06,
      "step": 78431,
      "training_step_time": 9.470019340515137
    },
    {
      "epoch": 4.2266845703125e-06,
      "model_forward_time": 1.9033174514770508,
      "step": 78432
    },
    {
      "epoch": 4.2266845703125e-06,
      "step": 78432,
      "training_step_time": 7.511321783065796
    },
    {
      "epoch": 4.227638244628906e-06,
      "model_forward_time": 1.9011955261230469,
      "step": 78433
    },
    {
      "epoch": 4.227638244628906e-06,
      "step": 78433,
      "training_step_time": 7.544338226318359
    },
    {
      "epoch": 4.228591918945313e-06,
      "model_forward_time": 1.744281530380249,
      "step": 78434
    },
    {
      "epoch": 4.228591918945313e-06,
      "step": 78434,
      "training_step_time": 7.545448541641235
    },
    {
      "epoch": 4.229545593261718e-06,
      "model_forward_time": 1.7537412643432617,
      "step": 78435
    },
    {
      "epoch": 4.229545593261718e-06,
      "step": 78435,
      "training_step_time": 8.756103038787842
    },
    {
      "epoch": 4.230499267578125e-06,
      "model_forward_time": 1.7590646743774414,
      "step": 78436
    },
    {
      "epoch": 4.230499267578125e-06,
      "step": 78436,
      "training_step_time": 8.581045150756836
    },
    {
      "epoch": 4.2314529418945315e-06,
      "model_forward_time": 1.745896577835083,
      "step": 78437
    },
    {
      "epoch": 4.2314529418945315e-06,
      "step": 78437,
      "training_step_time": 7.5843236446380615
    },
    {
      "epoch": 4.232406616210937e-06,
      "model_forward_time": 1.765984296798706,
      "step": 78438
    },
    {
      "epoch": 4.232406616210937e-06,
      "step": 78438,
      "training_step_time": 7.679464101791382
    },
    {
      "epoch": 4.233360290527344e-06,
      "model_forward_time": 2.8456811904907227,
      "step": 78439
    },
    {
      "epoch": 4.233360290527344e-06,
      "step": 78439,
      "training_step_time": 7.657440900802612
    },
    {
      "epoch": 4.23431396484375e-06,
      "grad_norm": 0.09110567718744278,
      "learning_rate": 1.2180154432681012e-06,
      "loss": 0.0554,
      "step": 78440
    },
    {
      "epoch": 4.23431396484375e-06,
      "model_forward_time": 1.7525346279144287,
      "step": 78440
    },
    {
      "dynamics_loss_avg": 0.04641491658985615,
      "epoch": 4.23431396484375e-06,
      "step": 78440
    },
    {
      "action_loss_avg": 0.0017485106131061912,
      "epoch": 4.23431396484375e-06,
      "step": 78440
    },
    {
      "epoch": 4.23431396484375e-06,
      "step": 78440,
      "training_step_time": 7.676482677459717
    },
    {
      "epoch": 4.235267639160156e-06,
      "model_forward_time": 1.7498421669006348,
      "step": 78441
    },
    {
      "epoch": 4.235267639160156e-06,
      "step": 78441,
      "training_step_time": 8.111408710479736
    },
    {
      "epoch": 4.236221313476563e-06,
      "model_forward_time": 1.7555699348449707,
      "step": 78442
    },
    {
      "epoch": 4.236221313476563e-06,
      "step": 78442,
      "training_step_time": 7.495938539505005
    },
    {
      "epoch": 4.237174987792968e-06,
      "model_forward_time": 1.4111478328704834,
      "step": 78443
    },
    {
      "epoch": 4.237174987792968e-06,
      "step": 78443,
      "training_step_time": 7.486231803894043
    },
    {
      "epoch": 4.238128662109375e-06,
      "model_forward_time": 1.7537429332733154,
      "step": 78444
    },
    {
      "epoch": 4.238128662109375e-06,
      "step": 78444,
      "training_step_time": 7.80028772354126
    },
    {
      "epoch": 4.2390823364257815e-06,
      "model_forward_time": 1.7501039505004883,
      "step": 78445
    },
    {
      "epoch": 4.2390823364257815e-06,
      "step": 78445,
      "training_step_time": 8.518123865127563
    },
    {
      "epoch": 4.240036010742187e-06,
      "model_forward_time": 1.7506253719329834,
      "step": 78446
    },
    {
      "epoch": 4.240036010742187e-06,
      "step": 78446,
      "training_step_time": 9.530903339385986
    },
    {
      "epoch": 4.240989685058594e-06,
      "model_forward_time": 1.7558140754699707,
      "step": 78447
    },
    {
      "epoch": 4.240989685058594e-06,
      "step": 78447,
      "training_step_time": 7.712070941925049
    },
    {
      "epoch": 4.241943359375e-06,
      "model_forward_time": 2.8012866973876953,
      "step": 78448
    },
    {
      "epoch": 4.241943359375e-06,
      "step": 78448,
      "training_step_time": 8.216417789459229
    },
    {
      "epoch": 4.242897033691406e-06,
      "model_forward_time": 1.7425239086151123,
      "step": 78449
    },
    {
      "epoch": 4.242897033691406e-06,
      "step": 78449,
      "training_step_time": 8.18725848197937
    },
    {
      "epoch": 4.243850708007813e-06,
      "grad_norm": 0.0720066949725151,
      "learning_rate": 1.216934094112815e-06,
      "loss": 0.0559,
      "step": 78450
    },
    {
      "epoch": 4.243850708007813e-06,
      "model_forward_time": 1.752713918685913,
      "step": 78450
    },
    {
      "dynamics_loss_avg": 0.054960431344807145,
      "epoch": 4.243850708007813e-06,
      "step": 78450
    },
    {
      "action_loss_avg": 0.0012090357893612235,
      "epoch": 4.243850708007813e-06,
      "step": 78450
    },
    {
      "epoch": 4.243850708007813e-06,
      "step": 78450,
      "training_step_time": 7.61739444732666
    },
    {
      "epoch": 4.244804382324218e-06,
      "model_forward_time": 1.7526624202728271,
      "step": 78451
    },
    {
      "epoch": 4.244804382324218e-06,
      "step": 78451,
      "training_step_time": 7.5817649364471436
    },
    {
      "epoch": 4.245758056640625e-06,
      "model_forward_time": 1.7732152938842773,
      "step": 78452
    },
    {
      "epoch": 4.245758056640625e-06,
      "step": 78452,
      "training_step_time": 7.503206968307495
    },
    {
      "epoch": 4.2467117309570315e-06,
      "model_forward_time": 1.7497997283935547,
      "step": 78453
    },
    {
      "epoch": 4.2467117309570315e-06,
      "step": 78453,
      "training_step_time": 7.325653791427612
    },
    {
      "epoch": 4.247665405273437e-06,
      "model_forward_time": 1.7547826766967773,
      "step": 78454
    },
    {
      "epoch": 4.247665405273437e-06,
      "step": 78454,
      "training_step_time": 7.737228155136108
    },
    {
      "epoch": 4.248619079589844e-06,
      "model_forward_time": 1.7703919410705566,
      "step": 78455
    },
    {
      "epoch": 4.248619079589844e-06,
      "step": 78455,
      "training_step_time": 8.626892328262329
    },
    {
      "epoch": 4.24957275390625e-06,
      "model_forward_time": 1.7795851230621338,
      "step": 78456
    },
    {
      "epoch": 4.24957275390625e-06,
      "step": 78456,
      "training_step_time": 7.674142360687256
    },
    {
      "epoch": 4.250526428222656e-06,
      "model_forward_time": 2.847656488418579,
      "step": 78457
    },
    {
      "epoch": 4.250526428222656e-06,
      "step": 78457,
      "training_step_time": 7.722971200942993
    },
    {
      "epoch": 4.251480102539063e-06,
      "model_forward_time": 1.7746655941009521,
      "step": 78458
    },
    {
      "epoch": 4.251480102539063e-06,
      "step": 78458,
      "training_step_time": 8.672011137008667
    },
    {
      "epoch": 4.252433776855468e-06,
      "model_forward_time": 1.7602260112762451,
      "step": 78459
    },
    {
      "epoch": 4.252433776855468e-06,
      "step": 78459,
      "training_step_time": 7.547423362731934
    },
    {
      "epoch": 4.253387451171875e-06,
      "grad_norm": 0.09821741282939911,
      "learning_rate": 1.2158531586679268e-06,
      "loss": 0.0563,
      "step": 78460
    },
    {
      "epoch": 4.253387451171875e-06,
      "model_forward_time": 1.7550745010375977,
      "step": 78460
    },
    {
      "dynamics_loss_avg": 0.05996857462450862,
      "epoch": 4.253387451171875e-06,
      "step": 78460
    },
    {
      "action_loss_avg": 0.002405697153881192,
      "epoch": 4.253387451171875e-06,
      "step": 78460
    },
    {
      "epoch": 4.253387451171875e-06,
      "step": 78460,
      "training_step_time": 7.55622935295105
    },
    {
      "epoch": 4.2543411254882815e-06,
      "model_forward_time": 1.7859854698181152,
      "step": 78461
    },
    {
      "epoch": 4.2543411254882815e-06,
      "step": 78461,
      "training_step_time": 7.2242395877838135
    },
    {
      "epoch": 4.255294799804687e-06,
      "model_forward_time": 1.7582335472106934,
      "step": 78462
    },
    {
      "epoch": 4.255294799804687e-06,
      "step": 78462,
      "training_step_time": 7.353347301483154
    },
    {
      "epoch": 4.256248474121094e-06,
      "model_forward_time": 1.3918757438659668,
      "step": 78463
    },
    {
      "epoch": 4.256248474121094e-06,
      "step": 78463,
      "training_step_time": 8.432204723358154
    },
    {
      "epoch": 4.2572021484375e-06,
      "model_forward_time": 1.7628583908081055,
      "step": 78464
    },
    {
      "epoch": 4.2572021484375e-06,
      "step": 78464,
      "training_step_time": 8.242053508758545
    },
    {
      "epoch": 4.258155822753906e-06,
      "model_forward_time": 1.7629458904266357,
      "step": 78465
    },
    {
      "epoch": 4.258155822753906e-06,
      "step": 78465,
      "training_step_time": 9.479369878768921
    },
    {
      "epoch": 4.259109497070313e-06,
      "model_forward_time": 1.7982447147369385,
      "step": 78466
    },
    {
      "epoch": 4.259109497070313e-06,
      "step": 78466,
      "training_step_time": 7.638645648956299
    },
    {
      "epoch": 4.260063171386718e-06,
      "model_forward_time": 2.855705499649048,
      "step": 78467
    },
    {
      "epoch": 4.260063171386718e-06,
      "step": 78467,
      "training_step_time": 9.240221500396729
    },
    {
      "epoch": 4.261016845703125e-06,
      "model_forward_time": 1.7680895328521729,
      "step": 78468
    },
    {
      "epoch": 4.261016845703125e-06,
      "step": 78468,
      "training_step_time": 8.40744137763977
    },
    {
      "epoch": 4.2619705200195315e-06,
      "model_forward_time": 1.7673399448394775,
      "step": 78469
    },
    {
      "epoch": 4.2619705200195315e-06,
      "step": 78469,
      "training_step_time": 7.496006727218628
    },
    {
      "epoch": 4.262924194335937e-06,
      "grad_norm": 0.08151800185441971,
      "learning_rate": 1.2147726370516478e-06,
      "loss": 0.0565,
      "step": 78470
    },
    {
      "epoch": 4.262924194335937e-06,
      "model_forward_time": 1.7735252380371094,
      "step": 78470
    },
    {
      "dynamics_loss_avg": 0.050035362225025894,
      "epoch": 4.262924194335937e-06,
      "step": 78470
    },
    {
      "action_loss_avg": 0.0012792939553037286,
      "epoch": 4.262924194335937e-06,
      "step": 78470
    },
    {
      "epoch": 4.262924194335937e-06,
      "step": 78470,
      "training_step_time": 7.579840898513794
    },
    {
      "epoch": 4.263877868652344e-06,
      "model_forward_time": 1.7606518268585205,
      "step": 78471
    },
    {
      "epoch": 4.263877868652344e-06,
      "step": 78471,
      "training_step_time": 7.677658796310425
    },
    {
      "epoch": 4.26483154296875e-06,
      "model_forward_time": 1.7660996913909912,
      "step": 78472
    },
    {
      "epoch": 4.26483154296875e-06,
      "step": 78472,
      "training_step_time": 7.700350761413574
    },
    {
      "epoch": 4.265785217285156e-06,
      "model_forward_time": 1.7628142833709717,
      "step": 78473
    },
    {
      "epoch": 4.265785217285156e-06,
      "step": 78473,
      "training_step_time": 8.290571212768555
    },
    {
      "epoch": 4.266738891601563e-06,
      "model_forward_time": 1.7560081481933594,
      "step": 78474
    },
    {
      "epoch": 4.266738891601563e-06,
      "step": 78474,
      "training_step_time": 8.237245798110962
    },
    {
      "epoch": 4.267692565917968e-06,
      "model_forward_time": 1.7857434749603271,
      "step": 78475
    },
    {
      "epoch": 4.267692565917968e-06,
      "step": 78475,
      "training_step_time": 7.639786958694458
    },
    {
      "epoch": 4.268646240234375e-06,
      "model_forward_time": 2.835144281387329,
      "step": 78476
    },
    {
      "epoch": 4.268646240234375e-06,
      "step": 78476,
      "training_step_time": 7.634521961212158
    },
    {
      "epoch": 4.2695999145507815e-06,
      "model_forward_time": 1.761260986328125,
      "step": 78477
    },
    {
      "epoch": 4.2695999145507815e-06,
      "step": 78477,
      "training_step_time": 8.089539527893066
    },
    {
      "epoch": 4.270553588867187e-06,
      "model_forward_time": 1.760650396347046,
      "step": 78478
    },
    {
      "epoch": 4.270553588867187e-06,
      "step": 78478,
      "training_step_time": 7.527569532394409
    },
    {
      "epoch": 4.271507263183594e-06,
      "model_forward_time": 1.7765917778015137,
      "step": 78479
    },
    {
      "epoch": 4.271507263183594e-06,
      "step": 78479,
      "training_step_time": 7.507484197616577
    },
    {
      "epoch": 4.2724609375e-06,
      "grad_norm": 0.08124538511037827,
      "learning_rate": 1.2136925293821421e-06,
      "loss": 0.056,
      "step": 78480
    },
    {
      "epoch": 4.2724609375e-06,
      "model_forward_time": 1.75887131690979,
      "step": 78480
    },
    {
      "dynamics_loss_avg": 0.045421839226037264,
      "epoch": 4.2724609375e-06,
      "step": 78480
    },
    {
      "action_loss_avg": 0.0016411613265518098,
      "epoch": 4.2724609375e-06,
      "step": 78480
    },
    {
      "epoch": 4.2724609375e-06,
      "step": 78480,
      "training_step_time": 7.4985105991363525
    },
    {
      "epoch": 4.273414611816406e-06,
      "model_forward_time": 1.7629191875457764,
      "step": 78481
    },
    {
      "epoch": 4.273414611816406e-06,
      "step": 78481,
      "training_step_time": 7.7229697704315186
    },
    {
      "epoch": 4.274368286132813e-06,
      "model_forward_time": 1.7616996765136719,
      "step": 78482
    },
    {
      "epoch": 4.274368286132813e-06,
      "step": 78482,
      "training_step_time": 8.500571727752686
    },
    {
      "epoch": 4.275321960449218e-06,
      "model_forward_time": 1.7626993656158447,
      "step": 78483
    },
    {
      "epoch": 4.275321960449218e-06,
      "step": 78483,
      "training_step_time": 9.315535545349121
    },
    {
      "epoch": 4.276275634765625e-06,
      "model_forward_time": 1.7733254432678223,
      "step": 78484
    },
    {
      "epoch": 4.276275634765625e-06,
      "step": 78484,
      "training_step_time": 11.040456771850586
    },
    {
      "epoch": 4.2772293090820316e-06,
      "model_forward_time": 2.8631200790405273,
      "step": 78485
    },
    {
      "epoch": 4.2772293090820316e-06,
      "step": 78485,
      "training_step_time": 9.743539333343506
    },
    {
      "epoch": 4.278182983398437e-06,
      "model_forward_time": 1.7597026824951172,
      "step": 78486
    },
    {
      "epoch": 4.278182983398437e-06,
      "step": 78486,
      "training_step_time": 7.576646089553833
    },
    {
      "epoch": 4.279136657714844e-06,
      "model_forward_time": 1.750669002532959,
      "step": 78487
    },
    {
      "epoch": 4.279136657714844e-06,
      "step": 78487,
      "training_step_time": 7.4798972606658936
    },
    {
      "epoch": 4.28009033203125e-06,
      "model_forward_time": 1.7680895328521729,
      "step": 78488
    },
    {
      "epoch": 4.28009033203125e-06,
      "step": 78488,
      "training_step_time": 7.501126766204834
    },
    {
      "epoch": 4.281044006347656e-06,
      "model_forward_time": 1.7851228713989258,
      "step": 78489
    },
    {
      "epoch": 4.281044006347656e-06,
      "step": 78489,
      "training_step_time": 7.495489120483398
    },
    {
      "epoch": 4.281997680664063e-06,
      "grad_norm": 0.08751820027828217,
      "learning_rate": 1.2126128357775308e-06,
      "loss": 0.0564,
      "step": 78490
    },
    {
      "epoch": 4.281997680664063e-06,
      "model_forward_time": 1.7597992420196533,
      "step": 78490
    },
    {
      "dynamics_loss_avg": 0.05245187319815159,
      "epoch": 4.281997680664063e-06,
      "step": 78490
    },
    {
      "action_loss_avg": 0.0011027685104636475,
      "epoch": 4.281997680664063e-06,
      "step": 78490
    },
    {
      "epoch": 4.281997680664063e-06,
      "step": 78490,
      "training_step_time": 8.3375563621521
    },
    {
      "epoch": 4.2829513549804684e-06,
      "model_forward_time": 1.7561564445495605,
      "step": 78491
    },
    {
      "epoch": 4.2829513549804684e-06,
      "step": 78491,
      "training_step_time": 7.691661834716797
    },
    {
      "epoch": 4.283905029296875e-06,
      "model_forward_time": 1.7596352100372314,
      "step": 78492
    },
    {
      "epoch": 4.283905029296875e-06,
      "step": 78492,
      "training_step_time": 8.374970436096191
    },
    {
      "epoch": 4.2848587036132816e-06,
      "model_forward_time": 1.776996374130249,
      "step": 78493
    },
    {
      "epoch": 4.2848587036132816e-06,
      "step": 78493,
      "training_step_time": 7.630521059036255
    },
    {
      "epoch": 4.285812377929687e-06,
      "model_forward_time": 2.8567750453948975,
      "step": 78494
    },
    {
      "epoch": 4.285812377929687e-06,
      "step": 78494,
      "training_step_time": 7.751901626586914
    },
    {
      "epoch": 4.286766052246094e-06,
      "model_forward_time": 1.7724041938781738,
      "step": 78495
    },
    {
      "epoch": 4.286766052246094e-06,
      "step": 78495,
      "training_step_time": 7.656351327896118
    },
    {
      "epoch": 4.2877197265625e-06,
      "model_forward_time": 1.7629189491271973,
      "step": 78496
    },
    {
      "epoch": 4.2877197265625e-06,
      "step": 78496,
      "training_step_time": 8.629219055175781
    },
    {
      "epoch": 4.288673400878906e-06,
      "model_forward_time": 1.764089822769165,
      "step": 78497
    },
    {
      "epoch": 4.288673400878906e-06,
      "step": 78497,
      "training_step_time": 7.546071767807007
    },
    {
      "epoch": 4.289627075195313e-06,
      "model_forward_time": 1.779294490814209,
      "step": 78498
    },
    {
      "epoch": 4.289627075195313e-06,
      "step": 78498,
      "training_step_time": 7.1807825565338135
    },
    {
      "epoch": 4.2905807495117184e-06,
      "model_forward_time": 1.7599475383758545,
      "step": 78499
    },
    {
      "epoch": 4.2905807495117184e-06,
      "step": 78499,
      "training_step_time": 7.481597900390625
    },
    {
      "epoch": 4.291534423828125e-06,
      "grad_norm": 0.07626417279243469,
      "learning_rate": 1.2115335563558838e-06,
      "loss": 0.0576,
      "step": 78500
    },
    {
      "epoch": 4.291534423828125e-06,
      "model_forward_time": 1.7559278011322021,
      "step": 78500
    },
    {
      "dynamics_loss_avg": 0.08682603128254414,
      "epoch": 4.291534423828125e-06,
      "step": 78500
    },
    {
      "action_loss_avg": 0.0025965935230487956,
      "epoch": 4.291534423828125e-06,
      "step": 78500
    },
    {
      "epoch": 4.291534423828125e-06,
      "step": 78500,
      "training_step_time": 8.240474224090576
    },
    {
      "epoch": 4.2924880981445316e-06,
      "model_forward_time": 1.7587230205535889,
      "step": 78501
    },
    {
      "epoch": 4.2924880981445316e-06,
      "step": 78501,
      "training_step_time": 8.60145616531372
    },
    {
      "epoch": 4.293441772460937e-06,
      "model_forward_time": 1.7655510902404785,
      "step": 78502
    },
    {
      "epoch": 4.293441772460937e-06,
      "step": 78502,
      "training_step_time": 8.111905813217163
    },
    {
      "epoch": 4.294395446777344e-06,
      "model_forward_time": 2.8469724655151367,
      "step": 78503
    },
    {
      "epoch": 4.294395446777344e-06,
      "step": 78503,
      "training_step_time": 10.517356872558594
    },
    {
      "epoch": 4.29534912109375e-06,
      "model_forward_time": 1.7583346366882324,
      "step": 78504
    },
    {
      "epoch": 4.29534912109375e-06,
      "step": 78504,
      "training_step_time": 8.76573395729065
    },
    {
      "epoch": 4.296302795410156e-06,
      "model_forward_time": 1.7583861351013184,
      "step": 78505
    },
    {
      "epoch": 4.296302795410156e-06,
      "step": 78505,
      "training_step_time": 9.257097959518433
    },
    {
      "epoch": 4.297256469726563e-06,
      "model_forward_time": 1.7582499980926514,
      "step": 78506
    },
    {
      "epoch": 4.297256469726563e-06,
      "step": 78506,
      "training_step_time": 7.555994749069214
    },
    {
      "epoch": 4.2982101440429684e-06,
      "model_forward_time": 1.7909495830535889,
      "step": 78507
    },
    {
      "epoch": 4.2982101440429684e-06,
      "step": 78507,
      "training_step_time": 7.299173831939697
    },
    {
      "epoch": 4.299163818359375e-06,
      "model_forward_time": 1.378537893295288,
      "step": 78508
    },
    {
      "epoch": 4.299163818359375e-06,
      "step": 78508,
      "training_step_time": 7.434099435806274
    },
    {
      "epoch": 4.3001174926757816e-06,
      "model_forward_time": 1.75008225440979,
      "step": 78509
    },
    {
      "epoch": 4.3001174926757816e-06,
      "step": 78509,
      "training_step_time": 7.6679160594940186
    },
    {
      "epoch": 4.301071166992187e-06,
      "grad_norm": 0.13477779924869537,
      "learning_rate": 1.2104546912352327e-06,
      "loss": 0.0574,
      "step": 78510
    },
    {
      "epoch": 4.301071166992187e-06,
      "model_forward_time": 1.7564289569854736,
      "step": 78510
    },
    {
      "dynamics_loss_avg": 0.05254716537892819,
      "epoch": 4.301071166992187e-06,
      "step": 78510
    },
    {
      "action_loss_avg": 0.0017472556384745986,
      "epoch": 4.301071166992187e-06,
      "step": 78510
    },
    {
      "epoch": 4.301071166992187e-06,
      "step": 78510,
      "training_step_time": 8.958639860153198
    },
    {
      "epoch": 4.302024841308594e-06,
      "model_forward_time": 1.7454817295074463,
      "step": 78511
    },
    {
      "epoch": 4.302024841308594e-06,
      "step": 78511,
      "training_step_time": 8.087508916854858
    },
    {
      "epoch": 4.302978515625e-06,
      "model_forward_time": 1.787391185760498,
      "step": 78512
    },
    {
      "epoch": 4.302978515625e-06,
      "step": 78512,
      "training_step_time": 8.61506700515747
    },
    {
      "epoch": 4.303932189941406e-06,
      "model_forward_time": 2.9001142978668213,
      "step": 78513
    },
    {
      "epoch": 4.303932189941406e-06,
      "step": 78513,
      "training_step_time": 8.629347324371338
    },
    {
      "epoch": 4.304885864257813e-06,
      "model_forward_time": 1.7719461917877197,
      "step": 78514
    },
    {
      "epoch": 4.304885864257813e-06,
      "step": 78514,
      "training_step_time": 8.650841474533081
    },
    {
      "epoch": 4.3058395385742184e-06,
      "model_forward_time": 1.756960153579712,
      "step": 78515
    },
    {
      "epoch": 4.3058395385742184e-06,
      "step": 78515,
      "training_step_time": 7.555742502212524
    },
    {
      "epoch": 4.306793212890625e-06,
      "model_forward_time": 1.7590758800506592,
      "step": 78516
    },
    {
      "epoch": 4.306793212890625e-06,
      "step": 78516,
      "training_step_time": 7.507362604141235
    },
    {
      "epoch": 4.307746887207032e-06,
      "model_forward_time": 1.7584302425384521,
      "step": 78517
    },
    {
      "epoch": 4.307746887207032e-06,
      "step": 78517,
      "training_step_time": 7.480990648269653
    },
    {
      "epoch": 4.308700561523437e-06,
      "model_forward_time": 1.7625083923339844,
      "step": 78518
    },
    {
      "epoch": 4.308700561523437e-06,
      "step": 78518,
      "training_step_time": 8.57752776145935
    },
    {
      "epoch": 4.309654235839844e-06,
      "model_forward_time": 1.7657837867736816,
      "step": 78519
    },
    {
      "epoch": 4.309654235839844e-06,
      "step": 78519,
      "training_step_time": 7.530024528503418
    },
    {
      "epoch": 4.31060791015625e-06,
      "grad_norm": 0.13792718946933746,
      "learning_rate": 1.2093762405335558e-06,
      "loss": 0.0586,
      "step": 78520
    },
    {
      "epoch": 4.31060791015625e-06,
      "model_forward_time": 1.7570147514343262,
      "step": 78520
    },
    {
      "dynamics_loss_avg": 0.06392325693741441,
      "epoch": 4.31060791015625e-06,
      "step": 78520
    },
    {
      "action_loss_avg": 0.0018071789236273617,
      "epoch": 4.31060791015625e-06,
      "step": 78520
    },
    {
      "epoch": 4.31060791015625e-06,
      "step": 78520,
      "training_step_time": 7.78275728225708
    },
    {
      "epoch": 4.311561584472656e-06,
      "model_forward_time": 1.7889719009399414,
      "step": 78521
    },
    {
      "epoch": 4.311561584472656e-06,
      "step": 78521,
      "training_step_time": 7.673610210418701
    },
    {
      "epoch": 4.312515258789063e-06,
      "model_forward_time": 2.8546624183654785,
      "step": 78522
    },
    {
      "epoch": 4.312515258789063e-06,
      "step": 78522,
      "training_step_time": 9.04381275177002
    },
    {
      "epoch": 4.3134689331054685e-06,
      "model_forward_time": 1.766545295715332,
      "step": 78523
    },
    {
      "epoch": 4.3134689331054685e-06,
      "step": 78523,
      "training_step_time": 8.665472745895386
    },
    {
      "epoch": 4.314422607421875e-06,
      "model_forward_time": 1.7672111988067627,
      "step": 78524
    },
    {
      "epoch": 4.314422607421875e-06,
      "step": 78524,
      "training_step_time": 8.489361047744751
    },
    {
      "epoch": 4.315376281738282e-06,
      "model_forward_time": 1.77616548538208,
      "step": 78525
    },
    {
      "epoch": 4.315376281738282e-06,
      "step": 78525,
      "training_step_time": 7.533545017242432
    },
    {
      "epoch": 4.316329956054687e-06,
      "model_forward_time": 1.7614555358886719,
      "step": 78526
    },
    {
      "epoch": 4.316329956054687e-06,
      "step": 78526,
      "training_step_time": 7.287837743759155
    },
    {
      "epoch": 4.317283630371094e-06,
      "model_forward_time": 1.757364273071289,
      "step": 78527
    },
    {
      "epoch": 4.317283630371094e-06,
      "step": 78527,
      "training_step_time": 8.614482641220093
    },
    {
      "epoch": 4.3182373046875e-06,
      "model_forward_time": 1.7660791873931885,
      "step": 78528
    },
    {
      "epoch": 4.3182373046875e-06,
      "step": 78528,
      "training_step_time": 7.563245058059692
    },
    {
      "epoch": 4.319190979003906e-06,
      "model_forward_time": 1.7634685039520264,
      "step": 78529
    },
    {
      "epoch": 4.319190979003906e-06,
      "step": 78529,
      "training_step_time": 8.469937324523926
    },
    {
      "epoch": 4.320144653320313e-06,
      "grad_norm": 0.10864029079675674,
      "learning_rate": 1.2082982043687975e-06,
      "loss": 0.0572,
      "step": 78530
    },
    {
      "epoch": 4.320144653320313e-06,
      "model_forward_time": 1.7587313652038574,
      "step": 78530
    },
    {
      "dynamics_loss_avg": 0.052760566025972365,
      "epoch": 4.320144653320313e-06,
      "step": 78530
    },
    {
      "action_loss_avg": 0.0023016969265881927,
      "epoch": 4.320144653320313e-06,
      "step": 78530
    },
    {
      "epoch": 4.320144653320313e-06,
      "step": 78530,
      "training_step_time": 8.730316162109375
    },
    {
      "epoch": 4.3210983276367185e-06,
      "model_forward_time": 2.8691227436065674,
      "step": 78531
    },
    {
      "epoch": 4.3210983276367185e-06,
      "step": 78531,
      "training_step_time": 7.774070739746094
    },
    {
      "epoch": 4.322052001953125e-06,
      "model_forward_time": 1.767333745956421,
      "step": 78532
    },
    {
      "epoch": 4.322052001953125e-06,
      "step": 78532,
      "training_step_time": 7.657603025436401
    },
    {
      "epoch": 4.323005676269532e-06,
      "model_forward_time": 1.7573823928833008,
      "step": 78533
    },
    {
      "epoch": 4.323005676269532e-06,
      "step": 78533,
      "training_step_time": 8.531593799591064
    },
    {
      "epoch": 4.323959350585937e-06,
      "model_forward_time": 1.7640364170074463,
      "step": 78534
    },
    {
      "epoch": 4.323959350585937e-06,
      "step": 78534,
      "training_step_time": 8.230997323989868
    },
    {
      "epoch": 4.324913024902344e-06,
      "model_forward_time": 1.7919518947601318,
      "step": 78535
    },
    {
      "epoch": 4.324913024902344e-06,
      "step": 78535,
      "training_step_time": 8.232731819152832
    },
    {
      "epoch": 4.32586669921875e-06,
      "model_forward_time": 1.7676193714141846,
      "step": 78536
    },
    {
      "epoch": 4.32586669921875e-06,
      "step": 78536,
      "training_step_time": 7.525905132293701
    },
    {
      "epoch": 4.326820373535156e-06,
      "model_forward_time": 1.7602574825286865,
      "step": 78537
    },
    {
      "epoch": 4.326820373535156e-06,
      "step": 78537,
      "training_step_time": 8.253634929656982
    },
    {
      "epoch": 4.327774047851563e-06,
      "model_forward_time": 1.7636868953704834,
      "step": 78538
    },
    {
      "epoch": 4.327774047851563e-06,
      "step": 78538,
      "training_step_time": 8.434764623641968
    },
    {
      "epoch": 4.3287277221679685e-06,
      "model_forward_time": 1.743340015411377,
      "step": 78539
    },
    {
      "epoch": 4.3287277221679685e-06,
      "step": 78539,
      "training_step_time": 10.936302661895752
    },
    {
      "epoch": 4.329681396484375e-06,
      "grad_norm": 0.0847475454211235,
      "learning_rate": 1.2072205828588452e-06,
      "loss": 0.0573,
      "step": 78540
    },
    {
      "epoch": 4.329681396484375e-06,
      "model_forward_time": 2.8274638652801514,
      "step": 78540
    },
    {
      "dynamics_loss_avg": 0.05733916461467743,
      "epoch": 4.329681396484375e-06,
      "step": 78540
    },
    {
      "action_loss_avg": 0.002882565651088953,
      "epoch": 4.329681396484375e-06,
      "step": 78540
    },
    {
      "epoch": 4.329681396484375e-06,
      "step": 78540,
      "training_step_time": 8.705927610397339
    },
    {
      "epoch": 4.330635070800782e-06,
      "model_forward_time": 1.7663578987121582,
      "step": 78541
    },
    {
      "epoch": 4.330635070800782e-06,
      "step": 78541,
      "training_step_time": 8.383087396621704
    },
    {
      "epoch": 4.331588745117187e-06,
      "model_forward_time": 1.7617511749267578,
      "step": 78542
    },
    {
      "epoch": 4.331588745117187e-06,
      "step": 78542,
      "training_step_time": 8.500218629837036
    },
    {
      "epoch": 4.332542419433594e-06,
      "model_forward_time": 1.7563164234161377,
      "step": 78543
    },
    {
      "epoch": 4.332542419433594e-06,
      "step": 78543,
      "training_step_time": 8.413797616958618
    },
    {
      "epoch": 4.33349609375e-06,
      "model_forward_time": 1.7885730266571045,
      "step": 78544
    },
    {
      "epoch": 4.33349609375e-06,
      "step": 78544,
      "training_step_time": 7.194704055786133
    },
    {
      "epoch": 4.334449768066406e-06,
      "model_forward_time": 1.7628908157348633,
      "step": 78545
    },
    {
      "epoch": 4.334449768066406e-06,
      "step": 78545,
      "training_step_time": 7.507935523986816
    },
    {
      "epoch": 4.335403442382813e-06,
      "model_forward_time": 1.758516550064087,
      "step": 78546
    },
    {
      "epoch": 4.335403442382813e-06,
      "step": 78546,
      "training_step_time": 7.733507394790649
    },
    {
      "epoch": 4.3363571166992185e-06,
      "model_forward_time": 1.7511589527130127,
      "step": 78547
    },
    {
      "epoch": 4.3363571166992185e-06,
      "step": 78547,
      "training_step_time": 8.686954736709595
    },
    {
      "epoch": 4.337310791015625e-06,
      "model_forward_time": 1.7633728981018066,
      "step": 78548
    },
    {
      "epoch": 4.337310791015625e-06,
      "step": 78548,
      "training_step_time": 8.096616744995117
    },
    {
      "epoch": 4.338264465332032e-06,
      "model_forward_time": 1.7865281105041504,
      "step": 78549
    },
    {
      "epoch": 4.338264465332032e-06,
      "step": 78549,
      "training_step_time": 8.85091757774353
    },
    {
      "epoch": 4.339218139648437e-06,
      "grad_norm": 0.12248505651950836,
      "learning_rate": 1.206143376121549e-06,
      "loss": 0.0584,
      "step": 78550
    },
    {
      "epoch": 4.339218139648437e-06,
      "model_forward_time": 1.7570745944976807,
      "step": 78550
    },
    {
      "dynamics_loss_avg": 0.06421670261770487,
      "epoch": 4.339218139648437e-06,
      "step": 78550
    },
    {
      "action_loss_avg": 0.0017753446707502007,
      "epoch": 4.339218139648437e-06,
      "step": 78550
    },
    {
      "epoch": 4.339218139648437e-06,
      "step": 78550,
      "training_step_time": 9.22097396850586
    },
    {
      "epoch": 4.340171813964844e-06,
      "model_forward_time": 1.7654926776885986,
      "step": 78551
    },
    {
      "epoch": 4.340171813964844e-06,
      "step": 78551,
      "training_step_time": 8.489144802093506
    },
    {
      "epoch": 4.34112548828125e-06,
      "model_forward_time": 1.756976842880249,
      "step": 78552
    },
    {
      "epoch": 4.34112548828125e-06,
      "step": 78552,
      "training_step_time": 7.5530686378479
    },
    {
      "epoch": 4.342079162597656e-06,
      "model_forward_time": 1.7858390808105469,
      "step": 78553
    },
    {
      "epoch": 4.342079162597656e-06,
      "step": 78553,
      "training_step_time": 7.375807762145996
    },
    {
      "epoch": 4.343032836914063e-06,
      "model_forward_time": 1.7555468082427979,
      "step": 78554
    },
    {
      "epoch": 4.343032836914063e-06,
      "step": 78554,
      "training_step_time": 7.413811206817627
    },
    {
      "epoch": 4.3439865112304685e-06,
      "model_forward_time": 1.7552061080932617,
      "step": 78555
    },
    {
      "epoch": 4.3439865112304685e-06,
      "step": 78555,
      "training_step_time": 9.567955732345581
    },
    {
      "epoch": 4.344940185546875e-06,
      "model_forward_time": 1.7655165195465088,
      "step": 78556
    },
    {
      "epoch": 4.344940185546875e-06,
      "step": 78556,
      "training_step_time": 8.573911666870117
    },
    {
      "epoch": 4.345893859863282e-06,
      "model_forward_time": 1.7644078731536865,
      "step": 78557
    },
    {
      "epoch": 4.345893859863282e-06,
      "step": 78557,
      "training_step_time": 8.159064769744873
    },
    {
      "epoch": 4.346847534179687e-06,
      "model_forward_time": 1.7803735733032227,
      "step": 78558
    },
    {
      "epoch": 4.346847534179687e-06,
      "step": 78558,
      "training_step_time": 8.526800394058228
    },
    {
      "epoch": 4.347801208496094e-06,
      "model_forward_time": 2.8677618503570557,
      "step": 78559
    },
    {
      "epoch": 4.347801208496094e-06,
      "step": 78559,
      "training_step_time": 8.193722486495972
    },
    {
      "epoch": 4.3487548828125e-06,
      "grad_norm": 0.08307702839374542,
      "learning_rate": 1.2050665842747078e-06,
      "loss": 0.0568,
      "step": 78560
    },
    {
      "epoch": 4.3487548828125e-06,
      "model_forward_time": 1.7560913562774658,
      "step": 78560
    },
    {
      "dynamics_loss_avg": 0.05827477630227804,
      "epoch": 4.3487548828125e-06,
      "step": 78560
    },
    {
      "action_loss_avg": 0.0012845055694924667,
      "epoch": 4.3487548828125e-06,
      "step": 78560
    },
    {
      "epoch": 4.3487548828125e-06,
      "step": 78560,
      "training_step_time": 7.629091739654541
    },
    {
      "epoch": 4.349708557128906e-06,
      "model_forward_time": 1.3919496536254883,
      "step": 78561
    },
    {
      "epoch": 4.349708557128906e-06,
      "step": 78561,
      "training_step_time": 8.114680290222168
    },
    {
      "epoch": 4.350662231445313e-06,
      "model_forward_time": 1.7737057209014893,
      "step": 78562
    },
    {
      "epoch": 4.350662231445313e-06,
      "step": 78562,
      "training_step_time": 7.531278610229492
    },
    {
      "epoch": 4.3516159057617185e-06,
      "model_forward_time": 1.7661335468292236,
      "step": 78563
    },
    {
      "epoch": 4.3516159057617185e-06,
      "step": 78563,
      "training_step_time": 7.272935390472412
    },
    {
      "epoch": 4.352569580078125e-06,
      "model_forward_time": 1.7601754665374756,
      "step": 78564
    },
    {
      "epoch": 4.352569580078125e-06,
      "step": 78564,
      "training_step_time": 7.69957971572876
    },
    {
      "epoch": 4.353523254394532e-06,
      "model_forward_time": 1.763364315032959,
      "step": 78565
    },
    {
      "epoch": 4.353523254394532e-06,
      "step": 78565,
      "training_step_time": 7.5777342319488525
    },
    {
      "epoch": 4.354476928710937e-06,
      "model_forward_time": 1.757443904876709,
      "step": 78566
    },
    {
      "epoch": 4.354476928710937e-06,
      "step": 78566,
      "training_step_time": 7.5665740966796875
    },
    {
      "epoch": 4.355430603027344e-06,
      "model_forward_time": 1.775916576385498,
      "step": 78567
    },
    {
      "epoch": 4.355430603027344e-06,
      "step": 78567,
      "training_step_time": 8.447927236557007
    },
    {
      "epoch": 4.35638427734375e-06,
      "model_forward_time": 2.8699095249176025,
      "step": 78568
    },
    {
      "epoch": 4.35638427734375e-06,
      "step": 78568,
      "training_step_time": 9.190372228622437
    },
    {
      "epoch": 4.357337951660156e-06,
      "model_forward_time": 1.7647790908813477,
      "step": 78569
    },
    {
      "epoch": 4.357337951660156e-06,
      "step": 78569,
      "training_step_time": 7.63437557220459
    },
    {
      "epoch": 4.358291625976563e-06,
      "grad_norm": 0.0713568925857544,
      "learning_rate": 1.2039902074360805e-06,
      "loss": 0.0565,
      "step": 78570
    },
    {
      "epoch": 4.358291625976563e-06,
      "model_forward_time": 1.7581310272216797,
      "step": 78570
    },
    {
      "dynamics_loss_avg": 0.07724339393898845,
      "epoch": 4.358291625976563e-06,
      "step": 78570
    },
    {
      "action_loss_avg": 0.0016347596189007162,
      "epoch": 4.358291625976563e-06,
      "step": 78570
    },
    {
      "epoch": 4.358291625976563e-06,
      "step": 78570,
      "training_step_time": 7.523920774459839
    },
    {
      "epoch": 4.3592453002929685e-06,
      "model_forward_time": 1.7923688888549805,
      "step": 78571
    },
    {
      "epoch": 4.3592453002929685e-06,
      "step": 78571,
      "training_step_time": 7.464991331100464
    },
    {
      "epoch": 4.360198974609375e-06,
      "model_forward_time": 1.7582242488861084,
      "step": 78572
    },
    {
      "epoch": 4.360198974609375e-06,
      "step": 78572,
      "training_step_time": 7.911048412322998
    },
    {
      "epoch": 4.361152648925782e-06,
      "model_forward_time": 1.7589282989501953,
      "step": 78573
    },
    {
      "epoch": 4.361152648925782e-06,
      "step": 78573,
      "training_step_time": 8.520541667938232
    },
    {
      "epoch": 4.362106323242187e-06,
      "model_forward_time": 1.7662949562072754,
      "step": 78574
    },
    {
      "epoch": 4.362106323242187e-06,
      "step": 78574,
      "training_step_time": 7.675912857055664
    },
    {
      "epoch": 4.363059997558594e-06,
      "model_forward_time": 1.7596850395202637,
      "step": 78575
    },
    {
      "epoch": 4.363059997558594e-06,
      "step": 78575,
      "training_step_time": 8.386067628860474
    },
    {
      "epoch": 4.364013671875e-06,
      "model_forward_time": 1.770226240158081,
      "step": 78576
    },
    {
      "epoch": 4.364013671875e-06,
      "step": 78576,
      "training_step_time": 9.97515320777893
    },
    {
      "epoch": 4.364967346191406e-06,
      "model_forward_time": 2.8258044719696045,
      "step": 78577
    },
    {
      "epoch": 4.364967346191406e-06,
      "step": 78577,
      "training_step_time": 7.671404838562012
    },
    {
      "epoch": 4.365921020507813e-06,
      "model_forward_time": 1.4004225730895996,
      "step": 78578
    },
    {
      "epoch": 4.365921020507813e-06,
      "step": 78578,
      "training_step_time": 8.471421241760254
    },
    {
      "epoch": 4.3668746948242185e-06,
      "model_forward_time": 1.763596773147583,
      "step": 78579
    },
    {
      "epoch": 4.3668746948242185e-06,
      "step": 78579,
      "training_step_time": 7.5498669147491455
    },
    {
      "epoch": 4.367828369140625e-06,
      "grad_norm": 0.08099476248025894,
      "learning_rate": 1.2029142457233755e-06,
      "loss": 0.0579,
      "step": 78580
    },
    {
      "epoch": 4.367828369140625e-06,
      "model_forward_time": 1.7662022113800049,
      "step": 78580
    },
    {
      "dynamics_loss_avg": 0.05645835567265749,
      "epoch": 4.367828369140625e-06,
      "step": 78580
    },
    {
      "action_loss_avg": 0.0015731775260064752,
      "epoch": 4.367828369140625e-06,
      "step": 78580
    },
    {
      "epoch": 4.367828369140625e-06,
      "step": 78580,
      "training_step_time": 7.549776315689087
    },
    {
      "epoch": 4.368782043457032e-06,
      "model_forward_time": 1.7331693172454834,
      "step": 78581
    },
    {
      "epoch": 4.368782043457032e-06,
      "step": 78581,
      "training_step_time": 7.302351236343384
    },
    {
      "epoch": 4.369735717773437e-06,
      "model_forward_time": 1.740405797958374,
      "step": 78582
    },
    {
      "epoch": 4.369735717773437e-06,
      "step": 78582,
      "training_step_time": 7.553805351257324
    },
    {
      "epoch": 4.370689392089844e-06,
      "model_forward_time": 1.7359919548034668,
      "step": 78583
    },
    {
      "epoch": 4.370689392089844e-06,
      "step": 78583,
      "training_step_time": 8.665279150009155
    },
    {
      "epoch": 4.37164306640625e-06,
      "model_forward_time": 1.7456130981445312,
      "step": 78584
    },
    {
      "epoch": 4.37164306640625e-06,
      "step": 78584,
      "training_step_time": 8.620182275772095
    },
    {
      "epoch": 4.372596740722656e-06,
      "model_forward_time": 1.7427735328674316,
      "step": 78585
    },
    {
      "epoch": 4.372596740722656e-06,
      "step": 78585,
      "training_step_time": 7.7258830070495605
    },
    {
      "epoch": 4.373550415039063e-06,
      "model_forward_time": 1.791794776916504,
      "step": 78586
    },
    {
      "epoch": 4.373550415039063e-06,
      "step": 78586,
      "training_step_time": 8.281931161880493
    },
    {
      "epoch": 4.3745040893554685e-06,
      "model_forward_time": 1.7668111324310303,
      "step": 78587
    },
    {
      "epoch": 4.3745040893554685e-06,
      "step": 78587,
      "training_step_time": 8.545096158981323
    },
    {
      "epoch": 4.375457763671875e-06,
      "model_forward_time": 1.7654190063476562,
      "step": 78588
    },
    {
      "epoch": 4.375457763671875e-06,
      "step": 78588,
      "training_step_time": 8.529192447662354
    },
    {
      "epoch": 4.376411437988282e-06,
      "model_forward_time": 1.7693417072296143,
      "step": 78589
    },
    {
      "epoch": 4.376411437988282e-06,
      "step": 78589,
      "training_step_time": 7.510796308517456
    },
    {
      "epoch": 4.377365112304687e-06,
      "grad_norm": 0.0825551226735115,
      "learning_rate": 1.2018386992542602e-06,
      "loss": 0.0558,
      "step": 78590
    },
    {
      "epoch": 4.377365112304687e-06,
      "model_forward_time": 1.765225887298584,
      "step": 78590
    },
    {
      "dynamics_loss_avg": 0.054198031686246394,
      "epoch": 4.377365112304687e-06,
      "step": 78590
    },
    {
      "action_loss_avg": 0.0023623473243787885,
      "epoch": 4.377365112304687e-06,
      "step": 78590
    },
    {
      "epoch": 4.377365112304687e-06,
      "step": 78590,
      "training_step_time": 7.608552694320679
    },
    {
      "epoch": 4.378318786621094e-06,
      "model_forward_time": 1.7616374492645264,
      "step": 78591
    },
    {
      "epoch": 4.378318786621094e-06,
      "step": 78591,
      "training_step_time": 7.51686429977417
    },
    {
      "epoch": 4.3792724609375e-06,
      "model_forward_time": 1.7502696514129639,
      "step": 78592
    },
    {
      "epoch": 4.3792724609375e-06,
      "step": 78592,
      "training_step_time": 9.040327787399292
    },
    {
      "epoch": 4.380226135253906e-06,
      "model_forward_time": 1.7634730339050293,
      "step": 78593
    },
    {
      "epoch": 4.380226135253906e-06,
      "step": 78593,
      "training_step_time": 9.374364614486694
    },
    {
      "epoch": 4.381179809570313e-06,
      "model_forward_time": 1.7568247318267822,
      "step": 78594
    },
    {
      "epoch": 4.381179809570313e-06,
      "step": 78594,
      "training_step_time": 8.19483494758606
    },
    {
      "epoch": 4.3821334838867185e-06,
      "model_forward_time": 1.7886590957641602,
      "step": 78595
    },
    {
      "epoch": 4.3821334838867185e-06,
      "step": 78595,
      "training_step_time": 8.590781450271606
    },
    {
      "epoch": 4.383087158203125e-06,
      "model_forward_time": 2.893385171890259,
      "step": 78596
    },
    {
      "epoch": 4.383087158203125e-06,
      "step": 78596,
      "training_step_time": 8.339710474014282
    },
    {
      "epoch": 4.384040832519532e-06,
      "model_forward_time": 1.7603530883789062,
      "step": 78597
    },
    {
      "epoch": 4.384040832519532e-06,
      "step": 78597,
      "training_step_time": 7.656353712081909
    },
    {
      "epoch": 4.384994506835937e-06,
      "model_forward_time": 1.758702278137207,
      "step": 78598
    },
    {
      "epoch": 4.384994506835937e-06,
      "step": 78598,
      "training_step_time": 7.5363383293151855
    },
    {
      "epoch": 4.385948181152344e-06,
      "model_forward_time": 1.7718300819396973,
      "step": 78599
    },
    {
      "epoch": 4.385948181152344e-06,
      "step": 78599,
      "training_step_time": 7.526062250137329
    },
    {
      "epoch": 4.38690185546875e-06,
      "grad_norm": 0.09980820119380951,
      "learning_rate": 1.2007635681463542e-06,
      "loss": 0.0571,
      "step": 78600
    },
    {
      "epoch": 4.38690185546875e-06,
      "model_forward_time": 1.7586750984191895,
      "step": 78600
    },
    {
      "dynamics_loss_avg": 0.08370815068483353,
      "epoch": 4.38690185546875e-06,
      "step": 78600
    },
    {
      "action_loss_avg": 0.0018806938402121886,
      "epoch": 4.38690185546875e-06,
      "step": 78600
    },
    {
      "epoch": 4.38690185546875e-06,
      "step": 78600,
      "training_step_time": 7.300966739654541
    },
    {
      "epoch": 4.387855529785156e-06,
      "model_forward_time": 1.7586824893951416,
      "step": 78601
    },
    {
      "epoch": 4.387855529785156e-06,
      "step": 78601,
      "training_step_time": 7.789642810821533
    },
    {
      "epoch": 4.388809204101563e-06,
      "model_forward_time": 1.7694075107574463,
      "step": 78602
    },
    {
      "epoch": 4.388809204101563e-06,
      "step": 78602,
      "training_step_time": 7.667651176452637
    },
    {
      "epoch": 4.3897628784179685e-06,
      "model_forward_time": 1.7675080299377441,
      "step": 78603
    },
    {
      "epoch": 4.3897628784179685e-06,
      "step": 78603,
      "training_step_time": 8.915090322494507
    },
    {
      "epoch": 4.390716552734375e-06,
      "model_forward_time": 1.7673423290252686,
      "step": 78604
    },
    {
      "epoch": 4.390716552734375e-06,
      "step": 78604,
      "training_step_time": 8.687886953353882
    },
    {
      "epoch": 4.391670227050782e-06,
      "model_forward_time": 2.8711445331573486,
      "step": 78605
    },
    {
      "epoch": 4.391670227050782e-06,
      "step": 78605,
      "training_step_time": 9.73566484451294
    },
    {
      "epoch": 4.392623901367187e-06,
      "model_forward_time": 1.7422518730163574,
      "step": 78606
    },
    {
      "epoch": 4.392623901367187e-06,
      "step": 78606,
      "training_step_time": 7.7168238162994385
    },
    {
      "epoch": 4.393577575683594e-06,
      "model_forward_time": 1.7487478256225586,
      "step": 78607
    },
    {
      "epoch": 4.393577575683594e-06,
      "step": 78607,
      "training_step_time": 8.406086206436157
    },
    {
      "epoch": 4.39453125e-06,
      "model_forward_time": 1.7723112106323242,
      "step": 78608
    },
    {
      "epoch": 4.39453125e-06,
      "step": 78608,
      "training_step_time": 7.54507040977478
    },
    {
      "epoch": 4.395484924316406e-06,
      "model_forward_time": 1.7380187511444092,
      "step": 78609
    },
    {
      "epoch": 4.395484924316406e-06,
      "step": 78609,
      "training_step_time": 7.572296380996704
    },
    {
      "epoch": 4.396438598632813e-06,
      "grad_norm": 0.09844087809324265,
      "learning_rate": 1.1996888525172334e-06,
      "loss": 0.0555,
      "step": 78610
    },
    {
      "epoch": 4.396438598632813e-06,
      "model_forward_time": 1.7382264137268066,
      "step": 78610
    },
    {
      "dynamics_loss_avg": 0.049515325017273425,
      "epoch": 4.396438598632813e-06,
      "step": 78610
    },
    {
      "action_loss_avg": 0.0015309690264984966,
      "epoch": 4.396438598632813e-06,
      "step": 78610
    },
    {
      "epoch": 4.396438598632813e-06,
      "step": 78610,
      "training_step_time": 7.805962800979614
    },
    {
      "epoch": 4.3973922729492185e-06,
      "model_forward_time": 1.7380826473236084,
      "step": 78611
    },
    {
      "epoch": 4.3973922729492185e-06,
      "step": 78611,
      "training_step_time": 7.6094934940338135
    },
    {
      "epoch": 4.398345947265625e-06,
      "model_forward_time": 1.7479889392852783,
      "step": 78612
    },
    {
      "epoch": 4.398345947265625e-06,
      "step": 78612,
      "training_step_time": 7.669533967971802
    },
    {
      "epoch": 4.399299621582032e-06,
      "model_forward_time": 1.7531280517578125,
      "step": 78613
    },
    {
      "epoch": 4.399299621582032e-06,
      "step": 78613,
      "training_step_time": 7.736469745635986
    },
    {
      "epoch": 4.400253295898437e-06,
      "model_forward_time": 2.83758544921875,
      "step": 78614
    },
    {
      "epoch": 4.400253295898437e-06,
      "step": 78614,
      "training_step_time": 7.695589542388916
    },
    {
      "epoch": 4.401206970214844e-06,
      "model_forward_time": 1.7311303615570068,
      "step": 78615
    },
    {
      "epoch": 4.401206970214844e-06,
      "step": 78615,
      "training_step_time": 8.680249691009521
    },
    {
      "epoch": 4.40216064453125e-06,
      "model_forward_time": 1.75105881690979,
      "step": 78616
    },
    {
      "epoch": 4.40216064453125e-06,
      "step": 78616,
      "training_step_time": 7.6205713748931885
    },
    {
      "epoch": 4.403114318847656e-06,
      "model_forward_time": 1.7735567092895508,
      "step": 78617
    },
    {
      "epoch": 4.403114318847656e-06,
      "step": 78617,
      "training_step_time": 7.564198970794678
    },
    {
      "epoch": 4.404067993164063e-06,
      "model_forward_time": 1.740126371383667,
      "step": 78618
    },
    {
      "epoch": 4.404067993164063e-06,
      "step": 78618,
      "training_step_time": 7.5647499561309814
    },
    {
      "epoch": 4.4050216674804685e-06,
      "model_forward_time": 1.7555723190307617,
      "step": 78619
    },
    {
      "epoch": 4.4050216674804685e-06,
      "step": 78619,
      "training_step_time": 9.323771715164185
    },
    {
      "epoch": 4.405975341796875e-06,
      "grad_norm": 0.09483321756124496,
      "learning_rate": 1.198614552484425e-06,
      "loss": 0.0542,
      "step": 78620
    },
    {
      "epoch": 4.405975341796875e-06,
      "model_forward_time": 1.7514560222625732,
      "step": 78620
    },
    {
      "dynamics_loss_avg": 0.04710912127047777,
      "epoch": 4.405975341796875e-06,
      "step": 78620
    },
    {
      "action_loss_avg": 0.0021000497916247697,
      "epoch": 4.405975341796875e-06,
      "step": 78620
    },
    {
      "epoch": 4.405975341796875e-06,
      "step": 78620,
      "training_step_time": 7.6433141231536865
    },
    {
      "epoch": 4.406929016113282e-06,
      "model_forward_time": 1.7473886013031006,
      "step": 78621
    },
    {
      "epoch": 4.406929016113282e-06,
      "step": 78621,
      "training_step_time": 8.90186357498169
    },
    {
      "epoch": 4.407882690429687e-06,
      "model_forward_time": 1.3843610286712646,
      "step": 78622
    },
    {
      "epoch": 4.407882690429687e-06,
      "step": 78622,
      "training_step_time": 7.649704217910767
    },
    {
      "epoch": 4.408836364746094e-06,
      "model_forward_time": 2.801881790161133,
      "step": 78623
    },
    {
      "epoch": 4.408836364746094e-06,
      "step": 78623,
      "training_step_time": 8.663226842880249
    },
    {
      "epoch": 4.4097900390625e-06,
      "model_forward_time": 1.7636103630065918,
      "step": 78624
    },
    {
      "epoch": 4.4097900390625e-06,
      "step": 78624,
      "training_step_time": 7.802473306655884
    },
    {
      "epoch": 4.410743713378906e-06,
      "model_forward_time": 1.7529029846191406,
      "step": 78625
    },
    {
      "epoch": 4.410743713378906e-06,
      "step": 78625,
      "training_step_time": 7.682826995849609
    },
    {
      "epoch": 4.411697387695313e-06,
      "model_forward_time": 1.7798261642456055,
      "step": 78626
    },
    {
      "epoch": 4.411697387695313e-06,
      "step": 78626,
      "training_step_time": 7.522494316101074
    },
    {
      "epoch": 4.4126510620117185e-06,
      "model_forward_time": 1.759852409362793,
      "step": 78627
    },
    {
      "epoch": 4.4126510620117185e-06,
      "step": 78627,
      "training_step_time": 7.545186281204224
    },
    {
      "epoch": 4.413604736328125e-06,
      "model_forward_time": 1.7565655708312988,
      "step": 78628
    },
    {
      "epoch": 4.413604736328125e-06,
      "step": 78628,
      "training_step_time": 7.686455726623535
    },
    {
      "epoch": 4.414558410644531e-06,
      "model_forward_time": 1.7419688701629639,
      "step": 78629
    },
    {
      "epoch": 4.414558410644531e-06,
      "step": 78629,
      "training_step_time": 7.795248508453369
    },
    {
      "epoch": 4.415512084960937e-06,
      "grad_norm": 0.09188639372587204,
      "learning_rate": 1.197540668165415e-06,
      "loss": 0.0542,
      "step": 78630
    },
    {
      "epoch": 4.415512084960937e-06,
      "model_forward_time": 1.7445130348205566,
      "step": 78630
    },
    {
      "dynamics_loss_avg": 0.04993595033884048,
      "epoch": 4.415512084960937e-06,
      "step": 78630
    },
    {
      "action_loss_avg": 0.001087093402747996,
      "epoch": 4.415512084960937e-06,
      "step": 78630
    },
    {
      "epoch": 4.415512084960937e-06,
      "step": 78630,
      "training_step_time": 9.319042921066284
    },
    {
      "epoch": 4.416465759277344e-06,
      "model_forward_time": 1.7301602363586426,
      "step": 78631
    },
    {
      "epoch": 4.416465759277344e-06,
      "step": 78631,
      "training_step_time": 7.71533989906311
    },
    {
      "epoch": 4.41741943359375e-06,
      "model_forward_time": 1.7746179103851318,
      "step": 78632
    },
    {
      "epoch": 4.41741943359375e-06,
      "step": 78632,
      "training_step_time": 9.243102073669434
    },
    {
      "epoch": 4.418373107910156e-06,
      "model_forward_time": 2.8793811798095703,
      "step": 78633
    },
    {
      "epoch": 4.418373107910156e-06,
      "step": 78633,
      "training_step_time": 9.502301216125488
    },
    {
      "epoch": 4.419326782226563e-06,
      "model_forward_time": 1.7495238780975342,
      "step": 78634
    },
    {
      "epoch": 4.419326782226563e-06,
      "step": 78634,
      "training_step_time": 7.620340824127197
    },
    {
      "epoch": 4.4202804565429685e-06,
      "model_forward_time": 1.7472882270812988,
      "step": 78635
    },
    {
      "epoch": 4.4202804565429685e-06,
      "step": 78635,
      "training_step_time": 9.041225671768188
    },
    {
      "epoch": 4.421234130859375e-06,
      "model_forward_time": 1.7435567378997803,
      "step": 78636
    },
    {
      "epoch": 4.421234130859375e-06,
      "step": 78636,
      "training_step_time": 7.594406604766846
    },
    {
      "epoch": 4.422187805175781e-06,
      "model_forward_time": 1.742443323135376,
      "step": 78637
    },
    {
      "epoch": 4.422187805175781e-06,
      "step": 78637,
      "training_step_time": 7.519413709640503
    },
    {
      "epoch": 4.423141479492187e-06,
      "model_forward_time": 1.7388601303100586,
      "step": 78638
    },
    {
      "epoch": 4.423141479492187e-06,
      "step": 78638,
      "training_step_time": 8.712185144424438
    },
    {
      "epoch": 4.424095153808594e-06,
      "model_forward_time": 1.7433016300201416,
      "step": 78639
    },
    {
      "epoch": 4.424095153808594e-06,
      "step": 78639,
      "training_step_time": 10.119192361831665
    },
    {
      "epoch": 4.425048828125e-06,
      "grad_norm": 0.11052068322896957,
      "learning_rate": 1.1964671996776395e-06,
      "loss": 0.0554,
      "step": 78640
    },
    {
      "epoch": 4.425048828125e-06,
      "model_forward_time": 1.7533860206604004,
      "step": 78640
    },
    {
      "dynamics_loss_avg": 0.06420105192810296,
      "epoch": 4.425048828125e-06,
      "step": 78640
    },
    {
      "action_loss_avg": 0.001777168776607141,
      "epoch": 4.425048828125e-06,
      "step": 78640
    },
    {
      "epoch": 4.425048828125e-06,
      "step": 78640,
      "training_step_time": 7.719703912734985
    },
    {
      "epoch": 4.426002502441406e-06,
      "model_forward_time": 1.7667129039764404,
      "step": 78641
    },
    {
      "epoch": 4.426002502441406e-06,
      "step": 78641,
      "training_step_time": 8.336405992507935
    },
    {
      "epoch": 4.426956176757813e-06,
      "model_forward_time": 2.8700428009033203,
      "step": 78642
    },
    {
      "epoch": 4.426956176757813e-06,
      "step": 78642,
      "training_step_time": 7.861007213592529
    },
    {
      "epoch": 4.4279098510742186e-06,
      "model_forward_time": 1.7452712059020996,
      "step": 78643
    },
    {
      "epoch": 4.4279098510742186e-06,
      "step": 78643,
      "training_step_time": 7.82092022895813
    },
    {
      "epoch": 4.428863525390625e-06,
      "model_forward_time": 1.7435944080352783,
      "step": 78644
    },
    {
      "epoch": 4.428863525390625e-06,
      "step": 78644,
      "training_step_time": 8.703913450241089
    },
    {
      "epoch": 4.429817199707031e-06,
      "model_forward_time": 1.7640173435211182,
      "step": 78645
    },
    {
      "epoch": 4.429817199707031e-06,
      "step": 78645,
      "training_step_time": 7.708421230316162
    },
    {
      "epoch": 4.430770874023437e-06,
      "model_forward_time": 1.7476699352264404,
      "step": 78646
    },
    {
      "epoch": 4.430770874023437e-06,
      "step": 78646,
      "training_step_time": 7.678421974182129
    },
    {
      "epoch": 4.431724548339844e-06,
      "model_forward_time": 1.741042137145996,
      "step": 78647
    },
    {
      "epoch": 4.431724548339844e-06,
      "step": 78647,
      "training_step_time": 7.869199752807617
    },
    {
      "epoch": 4.43267822265625e-06,
      "model_forward_time": 1.7444679737091064,
      "step": 78648
    },
    {
      "epoch": 4.43267822265625e-06,
      "step": 78648,
      "training_step_time": 8.790971517562866
    },
    {
      "epoch": 4.433631896972656e-06,
      "model_forward_time": 1.7414133548736572,
      "step": 78649
    },
    {
      "epoch": 4.433631896972656e-06,
      "step": 78649,
      "training_step_time": 8.839909791946411
    },
    {
      "epoch": 4.434585571289063e-06,
      "grad_norm": 0.06714611500501633,
      "learning_rate": 1.1953941471384923e-06,
      "loss": 0.0549,
      "step": 78650
    },
    {
      "epoch": 4.434585571289063e-06,
      "model_forward_time": 1.380925178527832,
      "step": 78650
    },
    {
      "dynamics_loss_avg": 0.04623640365898609,
      "epoch": 4.434585571289063e-06,
      "step": 78650
    },
    {
      "action_loss_avg": 0.0010890020639635623,
      "epoch": 4.434585571289063e-06,
      "step": 78650
    },
    {
      "epoch": 4.434585571289063e-06,
      "step": 78650,
      "training_step_time": 7.945614337921143
    },
    {
      "epoch": 4.4355392456054686e-06,
      "model_forward_time": 2.8504302501678467,
      "step": 78651
    },
    {
      "epoch": 4.4355392456054686e-06,
      "step": 78651,
      "training_step_time": 7.8998918533325195
    },
    {
      "epoch": 4.436492919921875e-06,
      "model_forward_time": 1.744673252105713,
      "step": 78652
    },
    {
      "epoch": 4.436492919921875e-06,
      "step": 78652,
      "training_step_time": 7.826575517654419
    },
    {
      "epoch": 4.437446594238281e-06,
      "model_forward_time": 1.7477045059204102,
      "step": 78653
    },
    {
      "epoch": 4.437446594238281e-06,
      "step": 78653,
      "training_step_time": 7.657677412033081
    },
    {
      "epoch": 4.438400268554687e-06,
      "model_forward_time": 1.7659626007080078,
      "step": 78654
    },
    {
      "epoch": 4.438400268554687e-06,
      "step": 78654,
      "training_step_time": 7.7480788230896
    },
    {
      "epoch": 4.439353942871094e-06,
      "model_forward_time": 1.7488162517547607,
      "step": 78655
    },
    {
      "epoch": 4.439353942871094e-06,
      "step": 78655,
      "training_step_time": 7.748883247375488
    },
    {
      "epoch": 4.4403076171875e-06,
      "model_forward_time": 1.741847038269043,
      "step": 78656
    },
    {
      "epoch": 4.4403076171875e-06,
      "step": 78656,
      "training_step_time": 7.8371665477752686
    },
    {
      "epoch": 4.441261291503906e-06,
      "model_forward_time": 1.7412834167480469,
      "step": 78657
    },
    {
      "epoch": 4.441261291503906e-06,
      "step": 78657,
      "training_step_time": 10.112137079238892
    },
    {
      "epoch": 4.442214965820313e-06,
      "model_forward_time": 1.3839917182922363,
      "step": 78658
    },
    {
      "epoch": 4.442214965820313e-06,
      "step": 78658,
      "training_step_time": 8.917545318603516
    },
    {
      "epoch": 4.4431686401367186e-06,
      "model_forward_time": 1.7410273551940918,
      "step": 78659
    },
    {
      "epoch": 4.4431686401367186e-06,
      "step": 78659,
      "training_step_time": 8.825219869613647
    },
    {
      "epoch": 4.444122314453125e-06,
      "grad_norm": 0.1001010611653328,
      "learning_rate": 1.1943215106653205e-06,
      "loss": 0.0556,
      "step": 78660
    },
    {
      "epoch": 4.444122314453125e-06,
      "model_forward_time": 2.7797539234161377,
      "step": 78660
    },
    {
      "dynamics_loss_avg": 0.053808814659714696,
      "epoch": 4.444122314453125e-06,
      "step": 78660
    },
    {
      "action_loss_avg": 0.002408449089853093,
      "epoch": 4.444122314453125e-06,
      "step": 78660
    },
    {
      "epoch": 4.444122314453125e-06,
      "step": 78660,
      "training_step_time": 7.746694803237915
    },
    {
      "epoch": 4.445075988769531e-06,
      "model_forward_time": 1.7441987991333008,
      "step": 78661
    },
    {
      "epoch": 4.445075988769531e-06,
      "step": 78661,
      "training_step_time": 8.534966468811035
    },
    {
      "epoch": 4.4460296630859374e-06,
      "model_forward_time": 1.7425825595855713,
      "step": 78662
    },
    {
      "epoch": 4.4460296630859374e-06,
      "step": 78662,
      "training_step_time": 8.58054232597351
    },
    {
      "epoch": 4.446983337402344e-06,
      "model_forward_time": 1.7659366130828857,
      "step": 78663
    },
    {
      "epoch": 4.446983337402344e-06,
      "step": 78663,
      "training_step_time": 8.208666801452637
    },
    {
      "epoch": 4.44793701171875e-06,
      "model_forward_time": 1.7464056015014648,
      "step": 78664
    },
    {
      "epoch": 4.44793701171875e-06,
      "step": 78664,
      "training_step_time": 7.640857696533203
    },
    {
      "epoch": 4.448890686035156e-06,
      "model_forward_time": 1.0294170379638672,
      "step": 78665
    },
    {
      "epoch": 4.448890686035156e-06,
      "step": 78665,
      "training_step_time": 7.836715459823608
    },
    {
      "epoch": 4.449844360351563e-06,
      "model_forward_time": 1.7422099113464355,
      "step": 78666
    },
    {
      "epoch": 4.449844360351563e-06,
      "step": 78666,
      "training_step_time": 9.600041389465332
    },
    {
      "epoch": 4.4507980346679686e-06,
      "model_forward_time": 1.7441864013671875,
      "step": 78667
    },
    {
      "epoch": 4.4507980346679686e-06,
      "step": 78667,
      "training_step_time": 7.714181900024414
    },
    {
      "epoch": 4.451751708984375e-06,
      "model_forward_time": 1.740567922592163,
      "step": 78668
    },
    {
      "epoch": 4.451751708984375e-06,
      "step": 78668,
      "training_step_time": 10.617629051208496
    },
    {
      "epoch": 4.452705383300781e-06,
      "model_forward_time": 1.7765429019927979,
      "step": 78669
    },
    {
      "epoch": 4.452705383300781e-06,
      "step": 78669,
      "training_step_time": 9.357988834381104
    },
    {
      "epoch": 4.4536590576171874e-06,
      "grad_norm": 0.09113206714391708,
      "learning_rate": 1.1932492903754289e-06,
      "loss": 0.0547,
      "step": 78670
    },
    {
      "epoch": 4.4536590576171874e-06,
      "model_forward_time": 2.9026193618774414,
      "step": 78670
    },
    {
      "dynamics_loss_avg": 0.05833288431167603,
      "epoch": 4.4536590576171874e-06,
      "step": 78670
    },
    {
      "action_loss_avg": 0.0020239751145709308,
      "epoch": 4.4536590576171874e-06,
      "step": 78670
    },
    {
      "epoch": 4.4536590576171874e-06,
      "step": 78670,
      "training_step_time": 7.786741256713867
    },
    {
      "epoch": 4.454612731933594e-06,
      "model_forward_time": 1.7627618312835693,
      "step": 78671
    },
    {
      "epoch": 4.454612731933594e-06,
      "step": 78671,
      "training_step_time": 7.698853969573975
    },
    {
      "epoch": 4.45556640625e-06,
      "model_forward_time": 1.776412010192871,
      "step": 78672
    },
    {
      "epoch": 4.45556640625e-06,
      "step": 78672,
      "training_step_time": 8.52764105796814
    },
    {
      "epoch": 4.456520080566406e-06,
      "model_forward_time": 1.7636160850524902,
      "step": 78673
    },
    {
      "epoch": 4.456520080566406e-06,
      "step": 78673,
      "training_step_time": 7.7867431640625
    },
    {
      "epoch": 4.457473754882813e-06,
      "model_forward_time": 1.749953031539917,
      "step": 78674
    },
    {
      "epoch": 4.457473754882813e-06,
      "step": 78674,
      "training_step_time": 7.808364391326904
    },
    {
      "epoch": 4.458427429199219e-06,
      "model_forward_time": 1.7622225284576416,
      "step": 78675
    },
    {
      "epoch": 4.458427429199219e-06,
      "step": 78675,
      "training_step_time": 9.503453969955444
    },
    {
      "epoch": 4.459381103515625e-06,
      "model_forward_time": 1.7596075534820557,
      "step": 78676
    },
    {
      "epoch": 4.459381103515625e-06,
      "step": 78676,
      "training_step_time": 7.6685471534729
    },
    {
      "epoch": 4.460334777832031e-06,
      "model_forward_time": 1.7565960884094238,
      "step": 78677
    },
    {
      "epoch": 4.460334777832031e-06,
      "step": 78677,
      "training_step_time": 9.069484949111938
    },
    {
      "epoch": 4.4612884521484374e-06,
      "model_forward_time": 1.7908949851989746,
      "step": 78678
    },
    {
      "epoch": 4.4612884521484374e-06,
      "step": 78678,
      "training_step_time": 7.737116813659668
    },
    {
      "epoch": 4.462242126464844e-06,
      "model_forward_time": 2.851255416870117,
      "step": 78679
    },
    {
      "epoch": 4.462242126464844e-06,
      "step": 78679,
      "training_step_time": 9.183925151824951
    },
    {
      "epoch": 4.46319580078125e-06,
      "grad_norm": 0.07810873538255692,
      "learning_rate": 1.1921774863860696e-06,
      "loss": 0.0552,
      "step": 78680
    },
    {
      "epoch": 4.46319580078125e-06,
      "model_forward_time": 1.7606241703033447,
      "step": 78680
    },
    {
      "dynamics_loss_avg": 0.06488803792744875,
      "epoch": 4.46319580078125e-06,
      "step": 78680
    },
    {
      "action_loss_avg": 0.0023111938557121904,
      "epoch": 4.46319580078125e-06,
      "step": 78680
    },
    {
      "epoch": 4.46319580078125e-06,
      "step": 78680,
      "training_step_time": 7.784879207611084
    },
    {
      "epoch": 4.464149475097656e-06,
      "model_forward_time": 1.7570304870605469,
      "step": 78681
    },
    {
      "epoch": 4.464149475097656e-06,
      "step": 78681,
      "training_step_time": 7.725311517715454
    },
    {
      "epoch": 4.465103149414063e-06,
      "model_forward_time": 1.755096435546875,
      "step": 78682
    },
    {
      "epoch": 4.465103149414063e-06,
      "step": 78682,
      "training_step_time": 7.548688650131226
    },
    {
      "epoch": 4.466056823730469e-06,
      "model_forward_time": 1.7536075115203857,
      "step": 78683
    },
    {
      "epoch": 4.466056823730469e-06,
      "step": 78683,
      "training_step_time": 7.768988370895386
    },
    {
      "epoch": 4.467010498046875e-06,
      "model_forward_time": 1.7475788593292236,
      "step": 78684
    },
    {
      "epoch": 4.467010498046875e-06,
      "step": 78684,
      "training_step_time": 8.832098007202148
    },
    {
      "epoch": 4.467964172363281e-06,
      "model_forward_time": 1.7428467273712158,
      "step": 78685
    },
    {
      "epoch": 4.467964172363281e-06,
      "step": 78685,
      "training_step_time": 8.19832706451416
    },
    {
      "epoch": 4.4689178466796874e-06,
      "model_forward_time": 1.747809886932373,
      "step": 78686
    },
    {
      "epoch": 4.4689178466796874e-06,
      "step": 78686,
      "training_step_time": 7.645730018615723
    },
    {
      "epoch": 4.469871520996094e-06,
      "model_forward_time": 1.7520151138305664,
      "step": 78687
    },
    {
      "epoch": 4.469871520996094e-06,
      "step": 78687,
      "training_step_time": 9.215914487838745
    },
    {
      "epoch": 4.4708251953125e-06,
      "model_forward_time": 2.371065855026245,
      "step": 78688
    },
    {
      "epoch": 4.4708251953125e-06,
      "step": 78688,
      "training_step_time": 8.480252027511597
    },
    {
      "epoch": 4.471778869628906e-06,
      "model_forward_time": 1.7474052906036377,
      "step": 78689
    },
    {
      "epoch": 4.471778869628906e-06,
      "step": 78689,
      "training_step_time": 7.610838413238525
    },
    {
      "epoch": 4.472732543945313e-06,
      "grad_norm": 0.08925710618495941,
      "learning_rate": 1.1911060988144556e-06,
      "loss": 0.0558,
      "step": 78690
    },
    {
      "epoch": 4.472732543945313e-06,
      "model_forward_time": 1.740473985671997,
      "step": 78690
    },
    {
      "dynamics_loss_avg": 0.04639212340116501,
      "epoch": 4.472732543945313e-06,
      "step": 78690
    },
    {
      "action_loss_avg": 0.0012743668368784711,
      "epoch": 4.472732543945313e-06,
      "step": 78690
    },
    {
      "epoch": 4.472732543945313e-06,
      "step": 78690,
      "training_step_time": 7.685350656509399
    },
    {
      "epoch": 4.473686218261719e-06,
      "model_forward_time": 1.042982816696167,
      "step": 78691
    },
    {
      "epoch": 4.473686218261719e-06,
      "step": 78691,
      "training_step_time": 7.539536714553833
    },
    {
      "epoch": 4.474639892578125e-06,
      "model_forward_time": 1.85786771774292,
      "step": 78692
    },
    {
      "epoch": 4.474639892578125e-06,
      "step": 78692,
      "training_step_time": 7.6330835819244385
    },
    {
      "epoch": 4.475593566894531e-06,
      "model_forward_time": 1.746981143951416,
      "step": 78693
    },
    {
      "epoch": 4.475593566894531e-06,
      "step": 78693,
      "training_step_time": 7.485586404800415
    },
    {
      "epoch": 4.4765472412109375e-06,
      "model_forward_time": 1.7377660274505615,
      "step": 78694
    },
    {
      "epoch": 4.4765472412109375e-06,
      "step": 78694,
      "training_step_time": 7.668674945831299
    },
    {
      "epoch": 4.477500915527344e-06,
      "model_forward_time": 1.746764898300171,
      "step": 78695
    },
    {
      "epoch": 4.477500915527344e-06,
      "step": 78695,
      "training_step_time": 7.647700548171997
    },
    {
      "epoch": 4.47845458984375e-06,
      "model_forward_time": 1.7532050609588623,
      "step": 78696
    },
    {
      "epoch": 4.47845458984375e-06,
      "step": 78696,
      "training_step_time": 8.346978425979614
    },
    {
      "epoch": 4.479408264160156e-06,
      "model_forward_time": 1.7420051097869873,
      "step": 78697
    },
    {
      "epoch": 4.479408264160156e-06,
      "step": 78697,
      "training_step_time": 9.066565752029419
    },
    {
      "epoch": 4.480361938476563e-06,
      "model_forward_time": 2.359553813934326,
      "step": 78698
    },
    {
      "epoch": 4.480361938476563e-06,
      "step": 78698,
      "training_step_time": 8.748438119888306
    },
    {
      "epoch": 4.481315612792969e-06,
      "model_forward_time": 1.7502269744873047,
      "step": 78699
    },
    {
      "epoch": 4.481315612792969e-06,
      "step": 78699,
      "training_step_time": 8.18154501914978
    },
    {
      "epoch": 4.482269287109375e-06,
      "grad_norm": 0.13504625856876373,
      "learning_rate": 1.1900351277777534e-06,
      "loss": 0.055,
      "step": 78700
    },
    {
      "epoch": 4.482269287109375e-06,
      "model_forward_time": 1.7394013404846191,
      "step": 78700
    },
    {
      "dynamics_loss_avg": 0.05707190996035934,
      "epoch": 4.482269287109375e-06,
      "step": 78700
    },
    {
      "action_loss_avg": 0.0024470019445288925,
      "epoch": 4.482269287109375e-06,
      "step": 78700
    },
    {
      "epoch": 4.482269287109375e-06,
      "step": 78700,
      "training_step_time": 7.479355573654175
    },
    {
      "epoch": 4.483222961425781e-06,
      "model_forward_time": 1.8609559535980225,
      "step": 78701
    },
    {
      "epoch": 4.483222961425781e-06,
      "step": 78701,
      "training_step_time": 7.570819616317749
    },
    {
      "epoch": 4.4841766357421875e-06,
      "model_forward_time": 1.7455344200134277,
      "step": 78702
    },
    {
      "epoch": 4.4841766357421875e-06,
      "step": 78702,
      "training_step_time": 7.472898006439209
    },
    {
      "epoch": 4.485130310058594e-06,
      "model_forward_time": 1.745629072189331,
      "step": 78703
    },
    {
      "epoch": 4.485130310058594e-06,
      "step": 78703,
      "training_step_time": 7.747709035873413
    },
    {
      "epoch": 4.486083984375e-06,
      "model_forward_time": 1.745182752609253,
      "step": 78704
    },
    {
      "epoch": 4.486083984375e-06,
      "step": 78704,
      "training_step_time": 8.89789366722107
    },
    {
      "epoch": 4.487037658691406e-06,
      "model_forward_time": 1.7455837726593018,
      "step": 78705
    },
    {
      "epoch": 4.487037658691406e-06,
      "step": 78705,
      "training_step_time": 8.607193946838379
    },
    {
      "epoch": 4.487991333007813e-06,
      "model_forward_time": 1.7387313842773438,
      "step": 78706
    },
    {
      "epoch": 4.487991333007813e-06,
      "step": 78706,
      "training_step_time": 7.701398134231567
    },
    {
      "epoch": 4.488945007324219e-06,
      "model_forward_time": 2.37068247795105,
      "step": 78707
    },
    {
      "epoch": 4.488945007324219e-06,
      "step": 78707,
      "training_step_time": 7.677978515625
    },
    {
      "epoch": 4.489898681640625e-06,
      "model_forward_time": 1.741685390472412,
      "step": 78708
    },
    {
      "epoch": 4.489898681640625e-06,
      "step": 78708,
      "training_step_time": 7.631758451461792
    },
    {
      "epoch": 4.490852355957031e-06,
      "model_forward_time": 1.747812032699585,
      "step": 78709
    },
    {
      "epoch": 4.490852355957031e-06,
      "step": 78709,
      "training_step_time": 7.526141166687012
    },
    {
      "epoch": 4.4918060302734375e-06,
      "grad_norm": 0.13240797817707062,
      "learning_rate": 1.18896457339308e-06,
      "loss": 0.0547,
      "step": 78710
    },
    {
      "epoch": 4.4918060302734375e-06,
      "model_forward_time": 1.8428480625152588,
      "step": 78710
    },
    {
      "dynamics_loss_avg": 0.06261067725718021,
      "epoch": 4.4918060302734375e-06,
      "step": 78710
    },
    {
      "action_loss_avg": 0.002814323944039643,
      "epoch": 4.4918060302734375e-06,
      "step": 78710
    },
    {
      "epoch": 4.4918060302734375e-06,
      "step": 78710,
      "training_step_time": 7.603371620178223
    },
    {
      "epoch": 4.492759704589844e-06,
      "model_forward_time": 1.754122018814087,
      "step": 78711
    },
    {
      "epoch": 4.492759704589844e-06,
      "step": 78711,
      "training_step_time": 7.4926958084106445
    },
    {
      "epoch": 4.49371337890625e-06,
      "model_forward_time": 1.743579626083374,
      "step": 78712
    },
    {
      "epoch": 4.49371337890625e-06,
      "step": 78712,
      "training_step_time": 9.420058965682983
    },
    {
      "epoch": 4.494667053222656e-06,
      "model_forward_time": 1.7494471073150635,
      "step": 78713
    },
    {
      "epoch": 4.494667053222656e-06,
      "step": 78713,
      "training_step_time": 7.580154895782471
    },
    {
      "epoch": 4.495620727539063e-06,
      "model_forward_time": 1.7525978088378906,
      "step": 78714
    },
    {
      "epoch": 4.495620727539063e-06,
      "step": 78714,
      "training_step_time": 8.650007247924805
    },
    {
      "epoch": 4.496574401855469e-06,
      "model_forward_time": 1.7452654838562012,
      "step": 78715
    },
    {
      "epoch": 4.496574401855469e-06,
      "step": 78715,
      "training_step_time": 7.627850294113159
    },
    {
      "epoch": 4.497528076171875e-06,
      "model_forward_time": 2.365217447280884,
      "step": 78716
    },
    {
      "epoch": 4.497528076171875e-06,
      "step": 78716,
      "training_step_time": 8.267328262329102
    },
    {
      "epoch": 4.498481750488281e-06,
      "model_forward_time": 1.7569127082824707,
      "step": 78717
    },
    {
      "epoch": 4.498481750488281e-06,
      "step": 78717,
      "training_step_time": 8.201691627502441
    },
    {
      "epoch": 4.4994354248046875e-06,
      "model_forward_time": 1.7388954162597656,
      "step": 78718
    },
    {
      "epoch": 4.4994354248046875e-06,
      "step": 78718,
      "training_step_time": 8.09337854385376
    },
    {
      "epoch": 4.500389099121094e-06,
      "model_forward_time": 1.856215000152588,
      "step": 78719
    },
    {
      "epoch": 4.500389099121094e-06,
      "step": 78719,
      "training_step_time": 7.634504556655884
    },
    {
      "epoch": 4.5013427734375e-06,
      "grad_norm": 0.09982742369174957,
      "learning_rate": 1.1878944357775107e-06,
      "loss": 0.0569,
      "step": 78720
    },
    {
      "epoch": 4.5013427734375e-06,
      "model_forward_time": 1.7525691986083984,
      "step": 78720
    },
    {
      "dynamics_loss_avg": 0.05402557048946619,
      "epoch": 4.5013427734375e-06,
      "step": 78720
    },
    {
      "action_loss_avg": 0.0018902413838077336,
      "epoch": 4.5013427734375e-06,
      "step": 78720
    },
    {
      "epoch": 4.5013427734375e-06,
      "step": 78720,
      "training_step_time": 8.406280279159546
    },
    {
      "epoch": 4.502296447753906e-06,
      "model_forward_time": 1.7450120449066162,
      "step": 78721
    },
    {
      "epoch": 4.502296447753906e-06,
      "step": 78721,
      "training_step_time": 8.065223455429077
    },
    {
      "epoch": 4.503250122070313e-06,
      "model_forward_time": 1.3801512718200684,
      "step": 78722
    },
    {
      "epoch": 4.503250122070313e-06,
      "step": 78722,
      "training_step_time": 11.731804847717285
    },
    {
      "epoch": 4.504203796386719e-06,
      "model_forward_time": 1.7482569217681885,
      "step": 78723
    },
    {
      "epoch": 4.504203796386719e-06,
      "step": 78723,
      "training_step_time": 8.406991958618164
    },
    {
      "epoch": 4.505157470703125e-06,
      "model_forward_time": 1.7590265274047852,
      "step": 78724
    },
    {
      "epoch": 4.505157470703125e-06,
      "step": 78724,
      "training_step_time": 7.562069416046143
    },
    {
      "epoch": 4.506111145019531e-06,
      "model_forward_time": 1.749474048614502,
      "step": 78725
    },
    {
      "epoch": 4.506111145019531e-06,
      "step": 78725,
      "training_step_time": 7.533606052398682
    },
    {
      "epoch": 4.5070648193359375e-06,
      "model_forward_time": 1.7519493103027344,
      "step": 78726
    },
    {
      "epoch": 4.5070648193359375e-06,
      "step": 78726,
      "training_step_time": 7.499398946762085
    },
    {
      "epoch": 4.508018493652344e-06,
      "model_forward_time": 1.743903398513794,
      "step": 78727
    },
    {
      "epoch": 4.508018493652344e-06,
      "step": 78727,
      "training_step_time": 6.753514289855957
    },
    {
      "epoch": 4.50897216796875e-06,
      "model_forward_time": 1.767087697982788,
      "step": 78728
    },
    {
      "epoch": 4.50897216796875e-06,
      "step": 78728,
      "training_step_time": 6.786682605743408
    },
    {
      "epoch": 4.509925842285156e-06,
      "model_forward_time": 1.7576570510864258,
      "step": 78729
    },
    {
      "epoch": 4.509925842285156e-06,
      "step": 78729,
      "training_step_time": 6.797754764556885
    },
    {
      "epoch": 4.510879516601563e-06,
      "grad_norm": 0.09690529853105545,
      "learning_rate": 1.186824715048075e-06,
      "loss": 0.0561,
      "step": 78730
    },
    {
      "epoch": 4.510879516601563e-06,
      "model_forward_time": 1.7498812675476074,
      "step": 78730
    },
    {
      "dynamics_loss_avg": 0.06593457497656345,
      "epoch": 4.510879516601563e-06,
      "step": 78730
    },
    {
      "action_loss_avg": 0.0015153106418438257,
      "epoch": 4.510879516601563e-06,
      "step": 78730
    },
    {
      "epoch": 4.510879516601563e-06,
      "step": 78730,
      "training_step_time": 6.799810171127319
    },
    {
      "epoch": 4.511833190917969e-06,
      "model_forward_time": 1.755037546157837,
      "step": 78731
    },
    {
      "epoch": 4.511833190917969e-06,
      "step": 78731,
      "training_step_time": 6.742707252502441
    },
    {
      "epoch": 4.512786865234375e-06,
      "model_forward_time": 2.415499687194824,
      "step": 78732
    },
    {
      "epoch": 4.512786865234375e-06,
      "step": 78732,
      "training_step_time": 8.462130308151245
    },
    {
      "epoch": 4.513740539550781e-06,
      "model_forward_time": 1.7331743240356445,
      "step": 78733
    },
    {
      "epoch": 4.513740539550781e-06,
      "step": 78733,
      "training_step_time": 8.063665628433228
    },
    {
      "epoch": 4.5146942138671875e-06,
      "model_forward_time": 1.7459666728973389,
      "step": 78734
    },
    {
      "epoch": 4.5146942138671875e-06,
      "step": 78734,
      "training_step_time": 8.702741146087646
    },
    {
      "epoch": 4.515647888183594e-06,
      "model_forward_time": 1.7451801300048828,
      "step": 78735
    },
    {
      "epoch": 4.515647888183594e-06,
      "step": 78735,
      "training_step_time": 6.771498680114746
    },
    {
      "epoch": 4.5166015625e-06,
      "model_forward_time": 1.7439088821411133,
      "step": 78736
    },
    {
      "epoch": 4.5166015625e-06,
      "step": 78736,
      "training_step_time": 6.745352745056152
    },
    {
      "epoch": 4.517555236816406e-06,
      "model_forward_time": 1.7466824054718018,
      "step": 78737
    },
    {
      "epoch": 4.517555236816406e-06,
      "step": 78737,
      "training_step_time": 6.748361587524414
    },
    {
      "epoch": 4.518508911132813e-06,
      "model_forward_time": 1.7503211498260498,
      "step": 78738
    },
    {
      "epoch": 4.518508911132813e-06,
      "step": 78738,
      "training_step_time": 6.822728872299194
    },
    {
      "epoch": 4.519462585449219e-06,
      "model_forward_time": 1.7490017414093018,
      "step": 78739
    },
    {
      "epoch": 4.519462585449219e-06,
      "step": 78739,
      "training_step_time": 6.752641916275024
    },
    {
      "epoch": 4.520416259765625e-06,
      "grad_norm": 0.26940956711769104,
      "learning_rate": 1.1857554113217568e-06,
      "loss": 0.0548,
      "step": 78740
    },
    {
      "epoch": 4.520416259765625e-06,
      "model_forward_time": 1.7486851215362549,
      "step": 78740
    },
    {
      "dynamics_loss_avg": 0.05545233506709337,
      "epoch": 4.520416259765625e-06,
      "step": 78740
    },
    {
      "action_loss_avg": 0.0015080304816365241,
      "epoch": 4.520416259765625e-06,
      "step": 78740
    },
    {
      "epoch": 4.520416259765625e-06,
      "step": 78740,
      "training_step_time": 6.743427276611328
    },
    {
      "epoch": 4.521369934082031e-06,
      "model_forward_time": 2.4298648834228516,
      "step": 78741
    },
    {
      "epoch": 4.521369934082031e-06,
      "step": 78741,
      "training_step_time": 8.761277675628662
    },
    {
      "epoch": 4.5223236083984375e-06,
      "model_forward_time": 1.7437045574188232,
      "step": 78742
    },
    {
      "epoch": 4.5223236083984375e-06,
      "step": 78742,
      "training_step_time": 7.433323860168457
    },
    {
      "epoch": 4.523277282714844e-06,
      "model_forward_time": 1.7594325542449951,
      "step": 78743
    },
    {
      "epoch": 4.523277282714844e-06,
      "step": 78743,
      "training_step_time": 7.603640079498291
    },
    {
      "epoch": 4.52423095703125e-06,
      "model_forward_time": 1.7419421672821045,
      "step": 78744
    },
    {
      "epoch": 4.52423095703125e-06,
      "step": 78744,
      "training_step_time": 7.423396110534668
    },
    {
      "epoch": 4.525184631347656e-06,
      "model_forward_time": 1.742281436920166,
      "step": 78745
    },
    {
      "epoch": 4.525184631347656e-06,
      "step": 78745,
      "training_step_time": 6.739047288894653
    },
    {
      "epoch": 4.526138305664063e-06,
      "model_forward_time": 1.7463278770446777,
      "step": 78746
    },
    {
      "epoch": 4.526138305664063e-06,
      "step": 78746,
      "training_step_time": 6.769416093826294
    },
    {
      "epoch": 4.527091979980469e-06,
      "model_forward_time": 1.7457225322723389,
      "step": 78747
    },
    {
      "epoch": 4.527091979980469e-06,
      "step": 78747,
      "training_step_time": 6.786507606506348
    },
    {
      "epoch": 4.528045654296875e-06,
      "model_forward_time": 1.7479071617126465,
      "step": 78748
    },
    {
      "epoch": 4.528045654296875e-06,
      "step": 78748,
      "training_step_time": 6.776252508163452
    },
    {
      "epoch": 4.528999328613281e-06,
      "model_forward_time": 1.7711751461029053,
      "step": 78749
    },
    {
      "epoch": 4.528999328613281e-06,
      "step": 78749,
      "training_step_time": 6.821488618850708
    },
    {
      "epoch": 4.5299530029296875e-06,
      "grad_norm": 0.08458047360181808,
      "learning_rate": 1.1846865247154903e-06,
      "loss": 0.0559,
      "step": 78750
    },
    {
      "epoch": 4.5299530029296875e-06,
      "model_forward_time": 2.3676371574401855,
      "step": 78750
    },
    {
      "dynamics_loss_avg": 0.04113137889653444,
      "epoch": 4.5299530029296875e-06,
      "step": 78750
    },
    {
      "action_loss_avg": 0.0014090118755120784,
      "epoch": 4.5299530029296875e-06,
      "step": 78750
    },
    {
      "epoch": 4.5299530029296875e-06,
      "step": 78750,
      "training_step_time": 8.717406988143921
    },
    {
      "epoch": 4.530906677246094e-06,
      "model_forward_time": 1.7480602264404297,
      "step": 78751
    },
    {
      "epoch": 4.530906677246094e-06,
      "step": 78751,
      "training_step_time": 7.46984338760376
    },
    {
      "epoch": 4.5318603515625e-06,
      "model_forward_time": 1.7544591426849365,
      "step": 78752
    },
    {
      "epoch": 4.5318603515625e-06,
      "step": 78752,
      "training_step_time": 7.528415203094482
    },
    {
      "epoch": 4.532814025878906e-06,
      "model_forward_time": 1.7512786388397217,
      "step": 78753
    },
    {
      "epoch": 4.532814025878906e-06,
      "step": 78753,
      "training_step_time": 7.3908607959747314
    },
    {
      "epoch": 4.533767700195312e-06,
      "model_forward_time": 1.7394707202911377,
      "step": 78754
    },
    {
      "epoch": 4.533767700195312e-06,
      "step": 78754,
      "training_step_time": 6.7395946979522705
    },
    {
      "epoch": 4.534721374511719e-06,
      "model_forward_time": 1.7472491264343262,
      "step": 78755
    },
    {
      "epoch": 4.534721374511719e-06,
      "step": 78755,
      "training_step_time": 6.712083101272583
    },
    {
      "epoch": 4.535675048828125e-06,
      "model_forward_time": 1.737823247909546,
      "step": 78756
    },
    {
      "epoch": 4.535675048828125e-06,
      "step": 78756,
      "training_step_time": 6.7737438678741455
    },
    {
      "epoch": 4.536628723144531e-06,
      "model_forward_time": 1.747424840927124,
      "step": 78757
    },
    {
      "epoch": 4.536628723144531e-06,
      "step": 78757,
      "training_step_time": 6.77541446685791
    },
    {
      "epoch": 4.5375823974609375e-06,
      "model_forward_time": 1.7433741092681885,
      "step": 78758
    },
    {
      "epoch": 4.5375823974609375e-06,
      "step": 78758,
      "training_step_time": 6.772572994232178
    },
    {
      "epoch": 4.538536071777344e-06,
      "model_forward_time": 2.4011950492858887,
      "step": 78759
    },
    {
      "epoch": 4.538536071777344e-06,
      "step": 78759,
      "training_step_time": 9.26913070678711
    },
    {
      "epoch": 4.53948974609375e-06,
      "grad_norm": 0.09656805545091629,
      "learning_rate": 1.1836180553461713e-06,
      "loss": 0.0565,
      "step": 78760
    },
    {
      "epoch": 4.53948974609375e-06,
      "model_forward_time": 1.7353339195251465,
      "step": 78760
    },
    {
      "dynamics_loss_avg": 0.06109911799430847,
      "epoch": 4.53948974609375e-06,
      "step": 78760
    },
    {
      "action_loss_avg": 0.0019251503894338385,
      "epoch": 4.53948974609375e-06,
      "step": 78760
    },
    {
      "epoch": 4.53948974609375e-06,
      "step": 78760,
      "training_step_time": 7.6323888301849365
    },
    {
      "epoch": 4.540443420410156e-06,
      "model_forward_time": 1.740896463394165,
      "step": 78761
    },
    {
      "epoch": 4.540443420410156e-06,
      "step": 78761,
      "training_step_time": 7.517848014831543
    },
    {
      "epoch": 4.541397094726562e-06,
      "model_forward_time": 1.7501497268676758,
      "step": 78762
    },
    {
      "epoch": 4.541397094726562e-06,
      "step": 78762,
      "training_step_time": 7.493653297424316
    },
    {
      "epoch": 4.542350769042969e-06,
      "model_forward_time": 1.7477881908416748,
      "step": 78763
    },
    {
      "epoch": 4.542350769042969e-06,
      "step": 78763,
      "training_step_time": 6.72182297706604
    },
    {
      "epoch": 4.543304443359375e-06,
      "model_forward_time": 1.7696828842163086,
      "step": 78764
    },
    {
      "epoch": 4.543304443359375e-06,
      "step": 78764,
      "training_step_time": 6.785463571548462
    },
    {
      "epoch": 4.544258117675781e-06,
      "model_forward_time": 1.759232759475708,
      "step": 78765
    },
    {
      "epoch": 4.544258117675781e-06,
      "step": 78765,
      "training_step_time": 6.710309982299805
    },
    {
      "epoch": 4.5452117919921875e-06,
      "model_forward_time": 1.7692008018493652,
      "step": 78766
    },
    {
      "epoch": 4.5452117919921875e-06,
      "step": 78766,
      "training_step_time": 6.724150896072388
    },
    {
      "epoch": 4.546165466308594e-06,
      "model_forward_time": 1.7649741172790527,
      "step": 78767
    },
    {
      "epoch": 4.546165466308594e-06,
      "step": 78767,
      "training_step_time": 6.723768949508667
    },
    {
      "epoch": 4.547119140625e-06,
      "model_forward_time": 2.818708896636963,
      "step": 78768
    },
    {
      "epoch": 4.547119140625e-06,
      "step": 78768,
      "training_step_time": 10.193259000778198
    },
    {
      "epoch": 4.548072814941406e-06,
      "model_forward_time": 1.7627735137939453,
      "step": 78769
    },
    {
      "epoch": 4.548072814941406e-06,
      "step": 78769,
      "training_step_time": 7.6326189041137695
    },
    {
      "epoch": 4.549026489257812e-06,
      "grad_norm": 0.0819050595164299,
      "learning_rate": 1.182550003330642e-06,
      "loss": 0.0564,
      "step": 78770
    },
    {
      "epoch": 4.549026489257812e-06,
      "model_forward_time": 1.7544541358947754,
      "step": 78770
    },
    {
      "dynamics_loss_avg": 0.06550265531986951,
      "epoch": 4.549026489257812e-06,
      "step": 78770
    },
    {
      "action_loss_avg": 0.002287847502157092,
      "epoch": 4.549026489257812e-06,
      "step": 78770
    },
    {
      "epoch": 4.549026489257812e-06,
      "step": 78770,
      "training_step_time": 7.496302604675293
    },
    {
      "epoch": 4.549980163574219e-06,
      "model_forward_time": 1.7534115314483643,
      "step": 78771
    },
    {
      "epoch": 4.549980163574219e-06,
      "step": 78771,
      "training_step_time": 7.49033522605896
    },
    {
      "epoch": 4.550933837890625e-06,
      "model_forward_time": 1.7891032695770264,
      "step": 78772
    },
    {
      "epoch": 4.550933837890625e-06,
      "step": 78772,
      "training_step_time": 7.0462281703948975
    },
    {
      "epoch": 4.551887512207031e-06,
      "model_forward_time": 1.763293981552124,
      "step": 78773
    },
    {
      "epoch": 4.551887512207031e-06,
      "step": 78773,
      "training_step_time": 6.726104021072388
    },
    {
      "epoch": 4.5528411865234375e-06,
      "model_forward_time": 1.767000675201416,
      "step": 78774
    },
    {
      "epoch": 4.5528411865234375e-06,
      "step": 78774,
      "training_step_time": 6.71818470954895
    },
    {
      "epoch": 4.553794860839844e-06,
      "model_forward_time": 1.7646450996398926,
      "step": 78775
    },
    {
      "epoch": 4.553794860839844e-06,
      "step": 78775,
      "training_step_time": 6.702417373657227
    },
    {
      "epoch": 4.55474853515625e-06,
      "model_forward_time": 1.7590878009796143,
      "step": 78776
    },
    {
      "epoch": 4.55474853515625e-06,
      "step": 78776,
      "training_step_time": 6.690904855728149
    },
    {
      "epoch": 4.555702209472656e-06,
      "model_forward_time": 2.7826015949249268,
      "step": 78777
    },
    {
      "epoch": 4.555702209472656e-06,
      "step": 78777,
      "training_step_time": 10.355712652206421
    },
    {
      "epoch": 4.556655883789062e-06,
      "model_forward_time": 1.7821271419525146,
      "step": 78778
    },
    {
      "epoch": 4.556655883789062e-06,
      "step": 78778,
      "training_step_time": 8.628607511520386
    },
    {
      "epoch": 4.557609558105469e-06,
      "model_forward_time": 1.7714359760284424,
      "step": 78779
    },
    {
      "epoch": 4.557609558105469e-06,
      "step": 78779,
      "training_step_time": 7.936557292938232
    },
    {
      "epoch": 4.558563232421875e-06,
      "grad_norm": 0.08364217728376389,
      "learning_rate": 1.1814823687857052e-06,
      "loss": 0.0549,
      "step": 78780
    },
    {
      "epoch": 4.558563232421875e-06,
      "model_forward_time": 1.7565069198608398,
      "step": 78780
    },
    {
      "dynamics_loss_avg": 0.04453402105718851,
      "epoch": 4.558563232421875e-06,
      "step": 78780
    },
    {
      "action_loss_avg": 0.0010568627389147879,
      "epoch": 4.558563232421875e-06,
      "step": 78780
    },
    {
      "epoch": 4.558563232421875e-06,
      "step": 78780,
      "training_step_time": 7.5362279415130615
    },
    {
      "epoch": 4.559516906738281e-06,
      "model_forward_time": 1.3942532539367676,
      "step": 78781
    },
    {
      "epoch": 4.559516906738281e-06,
      "step": 78781,
      "training_step_time": 7.075544357299805
    },
    {
      "epoch": 4.5604705810546875e-06,
      "model_forward_time": 1.7777860164642334,
      "step": 78782
    },
    {
      "epoch": 4.5604705810546875e-06,
      "step": 78782,
      "training_step_time": 6.781332015991211
    },
    {
      "epoch": 4.561424255371094e-06,
      "model_forward_time": 1.7663393020629883,
      "step": 78783
    },
    {
      "epoch": 4.561424255371094e-06,
      "step": 78783,
      "training_step_time": 6.6905272006988525
    },
    {
      "epoch": 4.5623779296875e-06,
      "model_forward_time": 1.7676036357879639,
      "step": 78784
    },
    {
      "epoch": 4.5623779296875e-06,
      "step": 78784,
      "training_step_time": 6.76185417175293
    },
    {
      "epoch": 4.563331604003906e-06,
      "model_forward_time": 1.7672722339630127,
      "step": 78785
    },
    {
      "epoch": 4.563331604003906e-06,
      "step": 78785,
      "training_step_time": 6.723620653152466
    },
    {
      "epoch": 4.564285278320312e-06,
      "model_forward_time": 1.7770042419433594,
      "step": 78786
    },
    {
      "epoch": 4.564285278320312e-06,
      "step": 78786,
      "training_step_time": 8.640571117401123
    },
    {
      "epoch": 4.565238952636719e-06,
      "model_forward_time": 2.888880968093872,
      "step": 78787
    },
    {
      "epoch": 4.565238952636719e-06,
      "step": 78787,
      "training_step_time": 8.396726846694946
    },
    {
      "epoch": 4.566192626953125e-06,
      "model_forward_time": 1.7882893085479736,
      "step": 78788
    },
    {
      "epoch": 4.566192626953125e-06,
      "step": 78788,
      "training_step_time": 7.489015579223633
    },
    {
      "epoch": 4.567146301269531e-06,
      "model_forward_time": 1.3941888809204102,
      "step": 78789
    },
    {
      "epoch": 4.567146301269531e-06,
      "step": 78789,
      "training_step_time": 8.578104257583618
    },
    {
      "epoch": 4.5680999755859375e-06,
      "grad_norm": 0.08937378227710724,
      "learning_rate": 1.1804151518281155e-06,
      "loss": 0.0555,
      "step": 78790
    },
    {
      "epoch": 4.5680999755859375e-06,
      "model_forward_time": 1.7630999088287354,
      "step": 78790
    },
    {
      "dynamics_loss_avg": 0.06412756480276585,
      "epoch": 4.5680999755859375e-06,
      "step": 78790
    },
    {
      "action_loss_avg": 0.0026566949672997,
      "epoch": 4.5680999755859375e-06,
      "step": 78790
    },
    {
      "epoch": 4.5680999755859375e-06,
      "step": 78790,
      "training_step_time": 7.2167277336120605
    },
    {
      "epoch": 4.569053649902344e-06,
      "model_forward_time": 1.7696735858917236,
      "step": 78791
    },
    {
      "epoch": 4.569053649902344e-06,
      "step": 78791,
      "training_step_time": 6.705489158630371
    },
    {
      "epoch": 4.57000732421875e-06,
      "model_forward_time": 1.7739882469177246,
      "step": 78792
    },
    {
      "epoch": 4.57000732421875e-06,
      "step": 78792,
      "training_step_time": 6.753493309020996
    },
    {
      "epoch": 4.570960998535156e-06,
      "model_forward_time": 1.761192798614502,
      "step": 78793
    },
    {
      "epoch": 4.570960998535156e-06,
      "step": 78793,
      "training_step_time": 6.7022881507873535
    },
    {
      "epoch": 4.571914672851562e-06,
      "model_forward_time": 1.7622160911560059,
      "step": 78794
    },
    {
      "epoch": 4.571914672851562e-06,
      "step": 78794,
      "training_step_time": 6.695611476898193
    },
    {
      "epoch": 4.572868347167969e-06,
      "model_forward_time": 1.7626843452453613,
      "step": 78795
    },
    {
      "epoch": 4.572868347167969e-06,
      "step": 78795,
      "training_step_time": 9.606491088867188
    },
    {
      "epoch": 4.573822021484375e-06,
      "model_forward_time": 2.850594997406006,
      "step": 78796
    },
    {
      "epoch": 4.573822021484375e-06,
      "step": 78796,
      "training_step_time": 10.526977062225342
    },
    {
      "epoch": 4.574775695800781e-06,
      "model_forward_time": 1.792729377746582,
      "step": 78797
    },
    {
      "epoch": 4.574775695800781e-06,
      "step": 78797,
      "training_step_time": 8.118261575698853
    },
    {
      "epoch": 4.5757293701171875e-06,
      "model_forward_time": 1.7576532363891602,
      "step": 78798
    },
    {
      "epoch": 4.5757293701171875e-06,
      "step": 78798,
      "training_step_time": 7.528390169143677
    },
    {
      "epoch": 4.576683044433594e-06,
      "model_forward_time": 1.7656004428863525,
      "step": 78799
    },
    {
      "epoch": 4.576683044433594e-06,
      "step": 78799,
      "training_step_time": 7.43548059463501
    },
    {
      "epoch": 4.57763671875e-06,
      "grad_norm": 0.2195649892091751,
      "learning_rate": 1.179348352574583e-06,
      "loss": 0.0558,
      "step": 78800
    },
    {
      "epoch": 4.57763671875e-06,
      "model_forward_time": 1.4012975692749023,
      "step": 78800
    },
    {
      "dynamics_loss_avg": 0.05913312342017889,
      "epoch": 4.57763671875e-06,
      "step": 78800
    },
    {
      "action_loss_avg": 0.0022926651115994902,
      "epoch": 4.57763671875e-06,
      "step": 78800
    },
    {
      "epoch": 4.57763671875e-06,
      "step": 78800,
      "training_step_time": 6.704450368881226
    },
    {
      "epoch": 4.578590393066406e-06,
      "model_forward_time": 1.7634773254394531,
      "step": 78801
    },
    {
      "epoch": 4.578590393066406e-06,
      "step": 78801,
      "training_step_time": 6.71356201171875
    },
    {
      "epoch": 4.579544067382812e-06,
      "model_forward_time": 1.7635324001312256,
      "step": 78802
    },
    {
      "epoch": 4.579544067382812e-06,
      "step": 78802,
      "training_step_time": 6.679811239242554
    },
    {
      "epoch": 4.580497741699219e-06,
      "model_forward_time": 1.7656364440917969,
      "step": 78803
    },
    {
      "epoch": 4.580497741699219e-06,
      "step": 78803,
      "training_step_time": 6.6331093311309814
    },
    {
      "epoch": 4.581451416015625e-06,
      "model_forward_time": 1.7506062984466553,
      "step": 78804
    },
    {
      "epoch": 4.581451416015625e-06,
      "step": 78804,
      "training_step_time": 8.561008214950562
    },
    {
      "epoch": 4.582405090332031e-06,
      "model_forward_time": 2.7937655448913574,
      "step": 78805
    },
    {
      "epoch": 4.582405090332031e-06,
      "step": 78805,
      "training_step_time": 10.10810899734497
    },
    {
      "epoch": 4.5833587646484375e-06,
      "model_forward_time": 1.8058552742004395,
      "step": 78806
    },
    {
      "epoch": 4.5833587646484375e-06,
      "step": 78806,
      "training_step_time": 8.272341966629028
    },
    {
      "epoch": 4.584312438964844e-06,
      "model_forward_time": 1.7662098407745361,
      "step": 78807
    },
    {
      "epoch": 4.584312438964844e-06,
      "step": 78807,
      "training_step_time": 7.51765251159668
    },
    {
      "epoch": 4.58526611328125e-06,
      "model_forward_time": 1.778221607208252,
      "step": 78808
    },
    {
      "epoch": 4.58526611328125e-06,
      "step": 78808,
      "training_step_time": 7.5071070194244385
    },
    {
      "epoch": 4.586219787597656e-06,
      "model_forward_time": 1.7704265117645264,
      "step": 78809
    },
    {
      "epoch": 4.586219787597656e-06,
      "step": 78809,
      "training_step_time": 6.712173938751221
    },
    {
      "epoch": 4.587173461914062e-06,
      "grad_norm": 0.08219947665929794,
      "learning_rate": 1.1782819711417697e-06,
      "loss": 0.0538,
      "step": 78810
    },
    {
      "epoch": 4.587173461914062e-06,
      "model_forward_time": 1.7739317417144775,
      "step": 78810
    },
    {
      "dynamics_loss_avg": 0.04975398536771536,
      "epoch": 4.587173461914062e-06,
      "step": 78810
    },
    {
      "action_loss_avg": 0.001587787241442129,
      "epoch": 4.587173461914062e-06,
      "step": 78810
    },
    {
      "epoch": 4.587173461914062e-06,
      "step": 78810,
      "training_step_time": 6.751542329788208
    },
    {
      "epoch": 4.588127136230469e-06,
      "model_forward_time": 1.7677290439605713,
      "step": 78811
    },
    {
      "epoch": 4.588127136230469e-06,
      "step": 78811,
      "training_step_time": 6.687608242034912
    },
    {
      "epoch": 4.589080810546875e-06,
      "model_forward_time": 1.069835901260376,
      "step": 78812
    },
    {
      "epoch": 4.589080810546875e-06,
      "step": 78812,
      "training_step_time": 6.7023797035217285
    },
    {
      "epoch": 4.590034484863281e-06,
      "model_forward_time": 1.7718162536621094,
      "step": 78813
    },
    {
      "epoch": 4.590034484863281e-06,
      "step": 78813,
      "training_step_time": 8.29070258140564
    },
    {
      "epoch": 4.5909881591796876e-06,
      "model_forward_time": 1.7766859531402588,
      "step": 78814
    },
    {
      "epoch": 4.5909881591796876e-06,
      "step": 78814,
      "training_step_time": 11.030213117599487
    },
    {
      "epoch": 4.591941833496094e-06,
      "model_forward_time": 2.505563735961914,
      "step": 78815
    },
    {
      "epoch": 4.591941833496094e-06,
      "step": 78815,
      "training_step_time": 8.493062734603882
    },
    {
      "epoch": 4.5928955078125e-06,
      "model_forward_time": 1.7774467468261719,
      "step": 78816
    },
    {
      "epoch": 4.5928955078125e-06,
      "step": 78816,
      "training_step_time": 7.613839149475098
    },
    {
      "epoch": 4.593849182128906e-06,
      "model_forward_time": 1.7675089836120605,
      "step": 78817
    },
    {
      "epoch": 4.593849182128906e-06,
      "step": 78817,
      "training_step_time": 7.543083667755127
    },
    {
      "epoch": 4.594802856445312e-06,
      "model_forward_time": 1.782733678817749,
      "step": 78818
    },
    {
      "epoch": 4.594802856445312e-06,
      "step": 78818,
      "training_step_time": 6.745131492614746
    },
    {
      "epoch": 4.595756530761719e-06,
      "model_forward_time": 1.7757432460784912,
      "step": 78819
    },
    {
      "epoch": 4.595756530761719e-06,
      "step": 78819,
      "training_step_time": 6.755360841751099
    },
    {
      "epoch": 4.596710205078125e-06,
      "grad_norm": 0.0931968241930008,
      "learning_rate": 1.1772160076462951e-06,
      "loss": 0.0534,
      "step": 78820
    },
    {
      "epoch": 4.596710205078125e-06,
      "model_forward_time": 1.745190143585205,
      "step": 78820
    },
    {
      "dynamics_loss_avg": 0.06400051303207874,
      "epoch": 4.596710205078125e-06,
      "step": 78820
    },
    {
      "action_loss_avg": 0.0029939208674477414,
      "epoch": 4.596710205078125e-06,
      "step": 78820
    },
    {
      "epoch": 4.596710205078125e-06,
      "step": 78820,
      "training_step_time": 6.683530807495117
    },
    {
      "epoch": 4.597663879394531e-06,
      "model_forward_time": 1.7584307193756104,
      "step": 78821
    },
    {
      "epoch": 4.597663879394531e-06,
      "step": 78821,
      "training_step_time": 6.747761249542236
    },
    {
      "epoch": 4.5986175537109376e-06,
      "model_forward_time": 1.7599785327911377,
      "step": 78822
    },
    {
      "epoch": 4.5986175537109376e-06,
      "step": 78822,
      "training_step_time": 8.999947309494019
    },
    {
      "epoch": 4.599571228027344e-06,
      "model_forward_time": 1.759378433227539,
      "step": 78823
    },
    {
      "epoch": 4.599571228027344e-06,
      "step": 78823,
      "training_step_time": 9.348616361618042
    },
    {
      "epoch": 4.60052490234375e-06,
      "model_forward_time": 2.8822896480560303,
      "step": 78824
    },
    {
      "epoch": 4.60052490234375e-06,
      "step": 78824,
      "training_step_time": 8.715074300765991
    },
    {
      "epoch": 4.601478576660156e-06,
      "model_forward_time": 1.76554274559021,
      "step": 78825
    },
    {
      "epoch": 4.601478576660156e-06,
      "step": 78825,
      "training_step_time": 7.723585367202759
    },
    {
      "epoch": 4.602432250976562e-06,
      "model_forward_time": 1.7575769424438477,
      "step": 78826
    },
    {
      "epoch": 4.602432250976562e-06,
      "step": 78826,
      "training_step_time": 7.599367618560791
    },
    {
      "epoch": 4.603385925292969e-06,
      "model_forward_time": 1.7741632461547852,
      "step": 78827
    },
    {
      "epoch": 4.603385925292969e-06,
      "step": 78827,
      "training_step_time": 7.208528280258179
    },
    {
      "epoch": 4.604339599609375e-06,
      "model_forward_time": 1.7824428081512451,
      "step": 78828
    },
    {
      "epoch": 4.604339599609375e-06,
      "step": 78828,
      "training_step_time": 6.754565715789795
    },
    {
      "epoch": 4.605293273925781e-06,
      "model_forward_time": 1.404937744140625,
      "step": 78829
    },
    {
      "epoch": 4.605293273925781e-06,
      "step": 78829,
      "training_step_time": 6.7215940952301025
    },
    {
      "epoch": 4.6062469482421876e-06,
      "grad_norm": 0.09963957220315933,
      "learning_rate": 1.1761504622047293e-06,
      "loss": 0.0528,
      "step": 78830
    },
    {
      "epoch": 4.6062469482421876e-06,
      "model_forward_time": 1.7712256908416748,
      "step": 78830
    },
    {
      "dynamics_loss_avg": 0.0756878487765789,
      "epoch": 4.6062469482421876e-06,
      "step": 78830
    },
    {
      "action_loss_avg": 0.0012062949186656623,
      "epoch": 4.6062469482421876e-06,
      "step": 78830
    },
    {
      "epoch": 4.6062469482421876e-06,
      "step": 78830,
      "training_step_time": 6.765912055969238
    },
    {
      "epoch": 4.607200622558594e-06,
      "model_forward_time": 1.7810349464416504,
      "step": 78831
    },
    {
      "epoch": 4.607200622558594e-06,
      "step": 78831,
      "training_step_time": 7.572383403778076
    },
    {
      "epoch": 4.608154296875e-06,
      "model_forward_time": 1.776369333267212,
      "step": 78832
    },
    {
      "epoch": 4.608154296875e-06,
      "step": 78832,
      "training_step_time": 8.220114946365356
    },
    {
      "epoch": 4.6091079711914064e-06,
      "model_forward_time": 1.840111494064331,
      "step": 78833
    },
    {
      "epoch": 4.6091079711914064e-06,
      "step": 78833,
      "training_step_time": 9.159448385238647
    },
    {
      "epoch": 4.610061645507812e-06,
      "model_forward_time": 2.9126830101013184,
      "step": 78834
    },
    {
      "epoch": 4.610061645507812e-06,
      "step": 78834,
      "training_step_time": 7.7919676303863525
    },
    {
      "epoch": 4.611015319824219e-06,
      "model_forward_time": 1.7756311893463135,
      "step": 78835
    },
    {
      "epoch": 4.611015319824219e-06,
      "step": 78835,
      "training_step_time": 7.601345062255859
    },
    {
      "epoch": 4.611968994140625e-06,
      "model_forward_time": 1.775613784790039,
      "step": 78836
    },
    {
      "epoch": 4.611968994140625e-06,
      "step": 78836,
      "training_step_time": 7.251353740692139
    },
    {
      "epoch": 4.612922668457031e-06,
      "model_forward_time": 1.7778754234313965,
      "step": 78837
    },
    {
      "epoch": 4.612922668457031e-06,
      "step": 78837,
      "training_step_time": 6.7578771114349365
    },
    {
      "epoch": 4.6138763427734376e-06,
      "model_forward_time": 1.7806344032287598,
      "step": 78838
    },
    {
      "epoch": 4.6138763427734376e-06,
      "step": 78838,
      "training_step_time": 6.723849058151245
    },
    {
      "epoch": 4.614830017089844e-06,
      "model_forward_time": 1.766331672668457,
      "step": 78839
    },
    {
      "epoch": 4.614830017089844e-06,
      "step": 78839,
      "training_step_time": 6.725275278091431
    },
    {
      "epoch": 4.61578369140625e-06,
      "grad_norm": 0.08656689524650574,
      "learning_rate": 1.1750853349336e-06,
      "loss": 0.0546,
      "step": 78840
    },
    {
      "epoch": 4.61578369140625e-06,
      "model_forward_time": 1.773284912109375,
      "step": 78840
    },
    {
      "dynamics_loss_avg": 0.05410129390656948,
      "epoch": 4.61578369140625e-06,
      "step": 78840
    },
    {
      "action_loss_avg": 0.0019759753588004967,
      "epoch": 4.61578369140625e-06,
      "step": 78840
    },
    {
      "epoch": 4.61578369140625e-06,
      "step": 78840,
      "training_step_time": 8.504385948181152
    },
    {
      "epoch": 4.6167373657226564e-06,
      "model_forward_time": 1.7699975967407227,
      "step": 78841
    },
    {
      "epoch": 4.6167373657226564e-06,
      "step": 78841,
      "training_step_time": 8.331562042236328
    },
    {
      "epoch": 4.617691040039062e-06,
      "model_forward_time": 1.8204193115234375,
      "step": 78842
    },
    {
      "epoch": 4.617691040039062e-06,
      "step": 78842,
      "training_step_time": 10.658169507980347
    },
    {
      "epoch": 4.618644714355469e-06,
      "model_forward_time": 2.880066394805908,
      "step": 78843
    },
    {
      "epoch": 4.618644714355469e-06,
      "step": 78843,
      "training_step_time": 7.696523189544678
    },
    {
      "epoch": 4.619598388671875e-06,
      "model_forward_time": 1.778834342956543,
      "step": 78844
    },
    {
      "epoch": 4.619598388671875e-06,
      "step": 78844,
      "training_step_time": 7.648285627365112
    },
    {
      "epoch": 4.620552062988281e-06,
      "model_forward_time": 1.7771081924438477,
      "step": 78845
    },
    {
      "epoch": 4.620552062988281e-06,
      "step": 78845,
      "training_step_time": 7.17994236946106
    },
    {
      "epoch": 4.621505737304688e-06,
      "model_forward_time": 1.7677063941955566,
      "step": 78846
    },
    {
      "epoch": 4.621505737304688e-06,
      "step": 78846,
      "training_step_time": 6.7761359214782715
    },
    {
      "epoch": 4.622459411621094e-06,
      "model_forward_time": 1.7724037170410156,
      "step": 78847
    },
    {
      "epoch": 4.622459411621094e-06,
      "step": 78847,
      "training_step_time": 6.725635766983032
    },
    {
      "epoch": 4.6234130859375e-06,
      "model_forward_time": 1.7751657962799072,
      "step": 78848
    },
    {
      "epoch": 4.6234130859375e-06,
      "step": 78848,
      "training_step_time": 6.736375570297241
    },
    {
      "epoch": 4.6243667602539064e-06,
      "model_forward_time": 1.7830145359039307,
      "step": 78849
    },
    {
      "epoch": 4.6243667602539064e-06,
      "step": 78849,
      "training_step_time": 9.30812382698059
    },
    {
      "epoch": 4.625320434570312e-06,
      "grad_norm": 0.07830589264631271,
      "learning_rate": 1.1740206259493874e-06,
      "loss": 0.0551,
      "step": 78850
    },
    {
      "epoch": 4.625320434570312e-06,
      "model_forward_time": 1.76788330078125,
      "step": 78850
    },
    {
      "dynamics_loss_avg": 0.05808621067553758,
      "epoch": 4.625320434570312e-06,
      "step": 78850
    },
    {
      "action_loss_avg": 0.0014518086914904415,
      "epoch": 4.625320434570312e-06,
      "step": 78850
    },
    {
      "epoch": 4.625320434570312e-06,
      "step": 78850,
      "training_step_time": 7.558865308761597
    },
    {
      "epoch": 4.626274108886719e-06,
      "model_forward_time": 1.8459739685058594,
      "step": 78851
    },
    {
      "epoch": 4.626274108886719e-06,
      "step": 78851,
      "training_step_time": 9.773617506027222
    },
    {
      "epoch": 4.627227783203125e-06,
      "model_forward_time": 2.5104589462280273,
      "step": 78852
    },
    {
      "epoch": 4.627227783203125e-06,
      "step": 78852,
      "training_step_time": 7.604150295257568
    },
    {
      "epoch": 4.628181457519531e-06,
      "model_forward_time": 1.7654836177825928,
      "step": 78853
    },
    {
      "epoch": 4.628181457519531e-06,
      "step": 78853,
      "training_step_time": 7.553875684738159
    },
    {
      "epoch": 4.629135131835938e-06,
      "model_forward_time": 1.7750885486602783,
      "step": 78854
    },
    {
      "epoch": 4.629135131835938e-06,
      "step": 78854,
      "training_step_time": 7.482621192932129
    },
    {
      "epoch": 4.630088806152344e-06,
      "model_forward_time": 1.7841637134552002,
      "step": 78855
    },
    {
      "epoch": 4.630088806152344e-06,
      "step": 78855,
      "training_step_time": 7.066190242767334
    },
    {
      "epoch": 4.63104248046875e-06,
      "model_forward_time": 1.7762954235076904,
      "step": 78856
    },
    {
      "epoch": 4.63104248046875e-06,
      "step": 78856,
      "training_step_time": 6.7180399894714355
    },
    {
      "epoch": 4.6319961547851564e-06,
      "model_forward_time": 1.7780961990356445,
      "step": 78857
    },
    {
      "epoch": 4.6319961547851564e-06,
      "step": 78857,
      "training_step_time": 6.6610634326934814
    },
    {
      "epoch": 4.632949829101562e-06,
      "model_forward_time": 1.0827105045318604,
      "step": 78858
    },
    {
      "epoch": 4.632949829101562e-06,
      "step": 78858,
      "training_step_time": 7.618107795715332
    },
    {
      "epoch": 4.633903503417969e-06,
      "model_forward_time": 1.7686614990234375,
      "step": 78859
    },
    {
      "epoch": 4.633903503417969e-06,
      "step": 78859,
      "training_step_time": 8.4473295211792
    },
    {
      "epoch": 4.634857177734375e-06,
      "grad_norm": 0.09584763646125793,
      "learning_rate": 1.1729563353685285e-06,
      "loss": 0.0546,
      "step": 78860
    },
    {
      "epoch": 4.634857177734375e-06,
      "model_forward_time": 1.8473012447357178,
      "step": 78860
    },
    {
      "dynamics_loss_avg": 0.07339915335178375,
      "epoch": 4.634857177734375e-06,
      "step": 78860
    },
    {
      "action_loss_avg": 0.002171519404510036,
      "epoch": 4.634857177734375e-06,
      "step": 78860
    },
    {
      "epoch": 4.634857177734375e-06,
      "step": 78860,
      "training_step_time": 9.665340185165405
    },
    {
      "epoch": 4.635810852050781e-06,
      "model_forward_time": 1.7831647396087646,
      "step": 78861
    },
    {
      "epoch": 4.635810852050781e-06,
      "step": 78861,
      "training_step_time": 8.488429546356201
    },
    {
      "epoch": 4.636764526367188e-06,
      "model_forward_time": 2.8583755493164062,
      "step": 78862
    },
    {
      "epoch": 4.636764526367188e-06,
      "step": 78862,
      "training_step_time": 7.632470607757568
    },
    {
      "epoch": 4.637718200683594e-06,
      "model_forward_time": 1.7880663871765137,
      "step": 78863
    },
    {
      "epoch": 4.637718200683594e-06,
      "step": 78863,
      "training_step_time": 7.6912806034088135
    },
    {
      "epoch": 4.638671875e-06,
      "model_forward_time": 1.7856082916259766,
      "step": 78864
    },
    {
      "epoch": 4.638671875e-06,
      "step": 78864,
      "training_step_time": 7.398485422134399
    },
    {
      "epoch": 4.6396255493164065e-06,
      "model_forward_time": 1.774674654006958,
      "step": 78865
    },
    {
      "epoch": 4.6396255493164065e-06,
      "step": 78865,
      "training_step_time": 6.885965824127197
    },
    {
      "epoch": 4.640579223632812e-06,
      "model_forward_time": 1.7733068466186523,
      "step": 78866
    },
    {
      "epoch": 4.640579223632812e-06,
      "step": 78866,
      "training_step_time": 6.991728067398071
    },
    {
      "epoch": 4.641532897949219e-06,
      "model_forward_time": 1.777151107788086,
      "step": 78867
    },
    {
      "epoch": 4.641532897949219e-06,
      "step": 78867,
      "training_step_time": 8.213606834411621
    },
    {
      "epoch": 4.642486572265625e-06,
      "model_forward_time": 1.7752647399902344,
      "step": 78868
    },
    {
      "epoch": 4.642486572265625e-06,
      "step": 78868,
      "training_step_time": 7.856911897659302
    },
    {
      "epoch": 4.643440246582031e-06,
      "model_forward_time": 1.8454995155334473,
      "step": 78869
    },
    {
      "epoch": 4.643440246582031e-06,
      "step": 78869,
      "training_step_time": 9.897071123123169
    },
    {
      "epoch": 4.644393920898438e-06,
      "grad_norm": 0.08469534665346146,
      "learning_rate": 1.171892463307409e-06,
      "loss": 0.0548,
      "step": 78870
    },
    {
      "epoch": 4.644393920898438e-06,
      "model_forward_time": 1.7656893730163574,
      "step": 78870
    },
    {
      "dynamics_loss_avg": 0.05287825260311365,
      "epoch": 4.644393920898438e-06,
      "step": 78870
    },
    {
      "action_loss_avg": 0.0014873752748826518,
      "epoch": 4.644393920898438e-06,
      "step": 78870
    },
    {
      "epoch": 4.644393920898438e-06,
      "step": 78870,
      "training_step_time": 7.823441505432129
    },
    {
      "epoch": 4.645347595214844e-06,
      "model_forward_time": 2.892409086227417,
      "step": 78871
    },
    {
      "epoch": 4.645347595214844e-06,
      "step": 78871,
      "training_step_time": 8.818077087402344
    },
    {
      "epoch": 4.64630126953125e-06,
      "model_forward_time": 1.7750139236450195,
      "step": 78872
    },
    {
      "epoch": 4.64630126953125e-06,
      "step": 78872,
      "training_step_time": 7.809570074081421
    },
    {
      "epoch": 4.6472549438476565e-06,
      "model_forward_time": 1.7727186679840088,
      "step": 78873
    },
    {
      "epoch": 4.6472549438476565e-06,
      "step": 78873,
      "training_step_time": 7.483709096908569
    },
    {
      "epoch": 4.648208618164062e-06,
      "model_forward_time": 1.7744865417480469,
      "step": 78874
    },
    {
      "epoch": 4.648208618164062e-06,
      "step": 78874,
      "training_step_time": 6.884875297546387
    },
    {
      "epoch": 4.649162292480469e-06,
      "model_forward_time": 1.768096923828125,
      "step": 78875
    },
    {
      "epoch": 4.649162292480469e-06,
      "step": 78875,
      "training_step_time": 6.839952707290649
    },
    {
      "epoch": 4.650115966796875e-06,
      "model_forward_time": 1.7809405326843262,
      "step": 78876
    },
    {
      "epoch": 4.650115966796875e-06,
      "step": 78876,
      "training_step_time": 7.6748082637786865
    },
    {
      "epoch": 4.651069641113281e-06,
      "model_forward_time": 1.7848360538482666,
      "step": 78877
    },
    {
      "epoch": 4.651069641113281e-06,
      "step": 78877,
      "training_step_time": 7.830918312072754
    },
    {
      "epoch": 4.652023315429688e-06,
      "model_forward_time": 1.8609652519226074,
      "step": 78878
    },
    {
      "epoch": 4.652023315429688e-06,
      "step": 78878,
      "training_step_time": 9.385496377944946
    },
    {
      "epoch": 4.652976989746093e-06,
      "model_forward_time": 1.7720491886138916,
      "step": 78879
    },
    {
      "epoch": 4.652976989746093e-06,
      "step": 78879,
      "training_step_time": 10.377500057220459
    },
    {
      "epoch": 4.6539306640625e-06,
      "grad_norm": 0.08645207434892654,
      "learning_rate": 1.1708290098823766e-06,
      "loss": 0.053,
      "step": 78880
    },
    {
      "epoch": 4.6539306640625e-06,
      "model_forward_time": 2.866027593612671,
      "step": 78880
    },
    {
      "dynamics_loss_avg": 0.050737081095576286,
      "epoch": 4.6539306640625e-06,
      "step": 78880
    },
    {
      "action_loss_avg": 0.002132396516390145,
      "epoch": 4.6539306640625e-06,
      "step": 78880
    },
    {
      "epoch": 4.6539306640625e-06,
      "step": 78880,
      "training_step_time": 7.82187032699585
    },
    {
      "epoch": 4.6548843383789065e-06,
      "model_forward_time": 1.7722187042236328,
      "step": 78881
    },
    {
      "epoch": 4.6548843383789065e-06,
      "step": 78881,
      "training_step_time": 8.73310136795044
    },
    {
      "epoch": 4.655838012695312e-06,
      "model_forward_time": 1.7722203731536865,
      "step": 78882
    },
    {
      "epoch": 4.655838012695312e-06,
      "step": 78882,
      "training_step_time": 7.632765293121338
    },
    {
      "epoch": 4.656791687011719e-06,
      "model_forward_time": 1.768031358718872,
      "step": 78883
    },
    {
      "epoch": 4.656791687011719e-06,
      "step": 78883,
      "training_step_time": 6.950910806655884
    },
    {
      "epoch": 4.657745361328125e-06,
      "model_forward_time": 1.7871837615966797,
      "step": 78884
    },
    {
      "epoch": 4.657745361328125e-06,
      "step": 78884,
      "training_step_time": 6.8722615242004395
    },
    {
      "epoch": 4.658699035644531e-06,
      "model_forward_time": 1.7752070426940918,
      "step": 78885
    },
    {
      "epoch": 4.658699035644531e-06,
      "step": 78885,
      "training_step_time": 7.649691820144653
    },
    {
      "epoch": 4.659652709960938e-06,
      "model_forward_time": 1.7746291160583496,
      "step": 78886
    },
    {
      "epoch": 4.659652709960938e-06,
      "step": 78886,
      "training_step_time": 8.596831321716309
    },
    {
      "epoch": 4.660606384277343e-06,
      "model_forward_time": 1.848275899887085,
      "step": 78887
    },
    {
      "epoch": 4.660606384277343e-06,
      "step": 78887,
      "training_step_time": 9.380118370056152
    },
    {
      "epoch": 4.66156005859375e-06,
      "model_forward_time": 1.7822086811065674,
      "step": 78888
    },
    {
      "epoch": 4.66156005859375e-06,
      "step": 78888,
      "training_step_time": 10.185993909835815
    },
    {
      "epoch": 4.6625137329101565e-06,
      "model_forward_time": 2.413764476776123,
      "step": 78889
    },
    {
      "epoch": 4.6625137329101565e-06,
      "step": 78889,
      "training_step_time": 7.791642665863037
    },
    {
      "epoch": 4.663467407226562e-06,
      "grad_norm": 0.0757802203297615,
      "learning_rate": 1.1697659752097247e-06,
      "loss": 0.0549,
      "step": 78890
    },
    {
      "epoch": 4.663467407226562e-06,
      "model_forward_time": 1.7790911197662354,
      "step": 78890
    },
    {
      "dynamics_loss_avg": 0.05356054119765759,
      "epoch": 4.663467407226562e-06,
      "step": 78890
    },
    {
      "action_loss_avg": 0.0025358737213537095,
      "epoch": 4.663467407226562e-06,
      "step": 78890
    },
    {
      "epoch": 4.663467407226562e-06,
      "step": 78890,
      "training_step_time": 7.740354299545288
    },
    {
      "epoch": 4.664421081542969e-06,
      "model_forward_time": 1.7587952613830566,
      "step": 78891
    },
    {
      "epoch": 4.664421081542969e-06,
      "step": 78891,
      "training_step_time": 7.608381271362305
    },
    {
      "epoch": 4.665374755859375e-06,
      "model_forward_time": 1.7506351470947266,
      "step": 78892
    },
    {
      "epoch": 4.665374755859375e-06,
      "step": 78892,
      "training_step_time": 6.874800443649292
    },
    {
      "epoch": 4.666328430175781e-06,
      "model_forward_time": 1.3873636722564697,
      "step": 78893
    },
    {
      "epoch": 4.666328430175781e-06,
      "step": 78893,
      "training_step_time": 6.8840203285217285
    },
    {
      "epoch": 4.667282104492188e-06,
      "model_forward_time": 1.7647147178649902,
      "step": 78894
    },
    {
      "epoch": 4.667282104492188e-06,
      "step": 78894,
      "training_step_time": 7.639181137084961
    },
    {
      "epoch": 4.668235778808593e-06,
      "model_forward_time": 1.7627592086791992,
      "step": 78895
    },
    {
      "epoch": 4.668235778808593e-06,
      "step": 78895,
      "training_step_time": 7.795166015625
    },
    {
      "epoch": 4.669189453125e-06,
      "model_forward_time": 1.813171625137329,
      "step": 78896
    },
    {
      "epoch": 4.669189453125e-06,
      "step": 78896,
      "training_step_time": 7.798332214355469
    },
    {
      "epoch": 4.6701431274414065e-06,
      "model_forward_time": 1.7642085552215576,
      "step": 78897
    },
    {
      "epoch": 4.6701431274414065e-06,
      "step": 78897,
      "training_step_time": 10.04065465927124
    },
    {
      "epoch": 4.671096801757812e-06,
      "model_forward_time": 1.760188341140747,
      "step": 78898
    },
    {
      "epoch": 4.671096801757812e-06,
      "step": 78898,
      "training_step_time": 8.961599588394165
    },
    {
      "epoch": 4.672050476074219e-06,
      "model_forward_time": 1.3998100757598877,
      "step": 78899
    },
    {
      "epoch": 4.672050476074219e-06,
      "step": 78899,
      "training_step_time": 8.307240962982178
    },
    {
      "epoch": 4.673004150390625e-06,
      "grad_norm": 0.10973329097032547,
      "learning_rate": 1.1687033594057067e-06,
      "loss": 0.0561,
      "step": 78900
    },
    {
      "epoch": 4.673004150390625e-06,
      "model_forward_time": 1.7631959915161133,
      "step": 78900
    },
    {
      "dynamics_loss_avg": 0.048197932355105874,
      "epoch": 4.673004150390625e-06,
      "step": 78900
    },
    {
      "action_loss_avg": 0.0010120722436113283,
      "epoch": 4.673004150390625e-06,
      "step": 78900
    },
    {
      "epoch": 4.673004150390625e-06,
      "step": 78900,
      "training_step_time": 7.676113128662109
    },
    {
      "epoch": 4.673957824707031e-06,
      "model_forward_time": 1.7525429725646973,
      "step": 78901
    },
    {
      "epoch": 4.673957824707031e-06,
      "step": 78901,
      "training_step_time": 6.943814754486084
    },
    {
      "epoch": 4.674911499023438e-06,
      "model_forward_time": 1.7688384056091309,
      "step": 78902
    },
    {
      "epoch": 4.674911499023438e-06,
      "step": 78902,
      "training_step_time": 6.904800176620483
    },
    {
      "epoch": 4.675865173339843e-06,
      "model_forward_time": 1.771132469177246,
      "step": 78903
    },
    {
      "epoch": 4.675865173339843e-06,
      "step": 78903,
      "training_step_time": 7.675642728805542
    },
    {
      "epoch": 4.67681884765625e-06,
      "model_forward_time": 1.7628793716430664,
      "step": 78904
    },
    {
      "epoch": 4.67681884765625e-06,
      "step": 78904,
      "training_step_time": 7.899618625640869
    },
    {
      "epoch": 4.6777725219726565e-06,
      "model_forward_time": 1.8477280139923096,
      "step": 78905
    },
    {
      "epoch": 4.6777725219726565e-06,
      "step": 78905,
      "training_step_time": 10.141449213027954
    },
    {
      "epoch": 4.678726196289062e-06,
      "model_forward_time": 1.7646939754486084,
      "step": 78906
    },
    {
      "epoch": 4.678726196289062e-06,
      "step": 78906,
      "training_step_time": 8.54111623764038
    },
    {
      "epoch": 4.679679870605469e-06,
      "model_forward_time": 1.7596626281738281,
      "step": 78907
    },
    {
      "epoch": 4.679679870605469e-06,
      "step": 78907,
      "training_step_time": 8.804119110107422
    },
    {
      "epoch": 4.680633544921875e-06,
      "model_forward_time": 2.8773951530456543,
      "step": 78908
    },
    {
      "epoch": 4.680633544921875e-06,
      "step": 78908,
      "training_step_time": 8.501110553741455
    },
    {
      "epoch": 4.681587219238281e-06,
      "model_forward_time": 1.7506709098815918,
      "step": 78909
    },
    {
      "epoch": 4.681587219238281e-06,
      "step": 78909,
      "training_step_time": 7.631471872329712
    },
    {
      "epoch": 4.682540893554688e-06,
      "grad_norm": 0.07749420404434204,
      "learning_rate": 1.1676411625865297e-06,
      "loss": 0.0545,
      "step": 78910
    },
    {
      "epoch": 4.682540893554688e-06,
      "model_forward_time": 1.7634849548339844,
      "step": 78910
    },
    {
      "dynamics_loss_avg": 0.04346529506146908,
      "epoch": 4.682540893554688e-06,
      "step": 78910
    },
    {
      "action_loss_avg": 0.0013469428289681674,
      "epoch": 4.682540893554688e-06,
      "step": 78910
    },
    {
      "epoch": 4.682540893554688e-06,
      "step": 78910,
      "training_step_time": 7.505692958831787
    },
    {
      "epoch": 4.683494567871093e-06,
      "model_forward_time": 1.76708984375,
      "step": 78911
    },
    {
      "epoch": 4.683494567871093e-06,
      "step": 78911,
      "training_step_time": 6.897464036941528
    },
    {
      "epoch": 4.6844482421875e-06,
      "model_forward_time": 1.7712175846099854,
      "step": 78912
    },
    {
      "epoch": 4.6844482421875e-06,
      "step": 78912,
      "training_step_time": 7.616897821426392
    },
    {
      "epoch": 4.6854019165039065e-06,
      "model_forward_time": 1.7751717567443848,
      "step": 78913
    },
    {
      "epoch": 4.6854019165039065e-06,
      "step": 78913,
      "training_step_time": 7.775951147079468
    },
    {
      "epoch": 4.686355590820312e-06,
      "model_forward_time": 1.8172025680541992,
      "step": 78914
    },
    {
      "epoch": 4.686355590820312e-06,
      "step": 78914,
      "training_step_time": 8.5330650806427
    },
    {
      "epoch": 4.687309265136719e-06,
      "model_forward_time": 1.7802314758300781,
      "step": 78915
    },
    {
      "epoch": 4.687309265136719e-06,
      "step": 78915,
      "training_step_time": 8.220646381378174
    },
    {
      "epoch": 4.688262939453125e-06,
      "model_forward_time": 1.7678356170654297,
      "step": 78916
    },
    {
      "epoch": 4.688262939453125e-06,
      "step": 78916,
      "training_step_time": 8.30774211883545
    },
    {
      "epoch": 4.689216613769531e-06,
      "model_forward_time": 2.893602132797241,
      "step": 78917
    },
    {
      "epoch": 4.689216613769531e-06,
      "step": 78917,
      "training_step_time": 7.804941654205322
    },
    {
      "epoch": 4.690170288085938e-06,
      "model_forward_time": 1.780029058456421,
      "step": 78918
    },
    {
      "epoch": 4.690170288085938e-06,
      "step": 78918,
      "training_step_time": 7.600866317749023
    },
    {
      "epoch": 4.691123962402343e-06,
      "model_forward_time": 1.773162841796875,
      "step": 78919
    },
    {
      "epoch": 4.691123962402343e-06,
      "step": 78919,
      "training_step_time": 7.380242824554443
    },
    {
      "epoch": 4.69207763671875e-06,
      "grad_norm": 0.08492111414670944,
      "learning_rate": 1.1665793848683544e-06,
      "loss": 0.0542,
      "step": 78920
    },
    {
      "epoch": 4.69207763671875e-06,
      "model_forward_time": 1.7837297916412354,
      "step": 78920
    },
    {
      "dynamics_loss_avg": 0.04695876073092222,
      "epoch": 4.69207763671875e-06,
      "step": 78920
    },
    {
      "action_loss_avg": 0.0020662087830714882,
      "epoch": 4.69207763671875e-06,
      "step": 78920
    },
    {
      "epoch": 4.69207763671875e-06,
      "step": 78920,
      "training_step_time": 6.95530104637146
    },
    {
      "epoch": 4.6930313110351565e-06,
      "model_forward_time": 1.7751121520996094,
      "step": 78921
    },
    {
      "epoch": 4.6930313110351565e-06,
      "step": 78921,
      "training_step_time": 7.672267913818359
    },
    {
      "epoch": 4.693984985351562e-06,
      "model_forward_time": 1.7794220447540283,
      "step": 78922
    },
    {
      "epoch": 4.693984985351562e-06,
      "step": 78922,
      "training_step_time": 7.78530216217041
    },
    {
      "epoch": 4.694938659667969e-06,
      "model_forward_time": 1.793511152267456,
      "step": 78923
    },
    {
      "epoch": 4.694938659667969e-06,
      "step": 78923,
      "training_step_time": 9.15648865699768
    },
    {
      "epoch": 4.695892333984375e-06,
      "model_forward_time": 1.7786829471588135,
      "step": 78924
    },
    {
      "epoch": 4.695892333984375e-06,
      "step": 78924,
      "training_step_time": 8.952753782272339
    },
    {
      "epoch": 4.696846008300781e-06,
      "model_forward_time": 1.7667980194091797,
      "step": 78925
    },
    {
      "epoch": 4.696846008300781e-06,
      "step": 78925,
      "training_step_time": 8.512407064437866
    },
    {
      "epoch": 4.697799682617188e-06,
      "model_forward_time": 2.82521390914917,
      "step": 78926
    },
    {
      "epoch": 4.697799682617188e-06,
      "step": 78926,
      "training_step_time": 7.812284231185913
    },
    {
      "epoch": 4.698753356933593e-06,
      "model_forward_time": 1.7657697200775146,
      "step": 78927
    },
    {
      "epoch": 4.698753356933593e-06,
      "step": 78927,
      "training_step_time": 7.668682813644409
    },
    {
      "epoch": 4.69970703125e-06,
      "model_forward_time": 1.7706129550933838,
      "step": 78928
    },
    {
      "epoch": 4.69970703125e-06,
      "step": 78928,
      "training_step_time": 7.672352075576782
    },
    {
      "epoch": 4.7006607055664065e-06,
      "model_forward_time": 1.780505657196045,
      "step": 78929
    },
    {
      "epoch": 4.7006607055664065e-06,
      "step": 78929,
      "training_step_time": 6.957827806472778
    },
    {
      "epoch": 4.701614379882812e-06,
      "grad_norm": 0.08559627085924149,
      "learning_rate": 1.1655180263672928e-06,
      "loss": 0.0551,
      "step": 78930
    },
    {
      "epoch": 4.701614379882812e-06,
      "model_forward_time": 1.7693030834197998,
      "step": 78930
    },
    {
      "dynamics_loss_avg": 0.053655190765857695,
      "epoch": 4.701614379882812e-06,
      "step": 78930
    },
    {
      "action_loss_avg": 0.0021640265884343536,
      "epoch": 4.701614379882812e-06,
      "step": 78930
    },
    {
      "epoch": 4.701614379882812e-06,
      "step": 78930,
      "training_step_time": 7.621978044509888
    },
    {
      "epoch": 4.702568054199219e-06,
      "model_forward_time": 1.7629144191741943,
      "step": 78931
    },
    {
      "epoch": 4.702568054199219e-06,
      "step": 78931,
      "training_step_time": 7.778131484985352
    },
    {
      "epoch": 4.703521728515625e-06,
      "model_forward_time": 1.7689845561981201,
      "step": 78932
    },
    {
      "epoch": 4.703521728515625e-06,
      "step": 78932,
      "training_step_time": 8.598668098449707
    },
    {
      "epoch": 4.704475402832031e-06,
      "model_forward_time": 1.8213872909545898,
      "step": 78933
    },
    {
      "epoch": 4.704475402832031e-06,
      "step": 78933,
      "training_step_time": 9.52510404586792
    },
    {
      "epoch": 4.705429077148438e-06,
      "model_forward_time": 1.7464137077331543,
      "step": 78934
    },
    {
      "epoch": 4.705429077148438e-06,
      "step": 78934,
      "training_step_time": 10.501331329345703
    },
    {
      "epoch": 4.706382751464843e-06,
      "model_forward_time": 2.8083624839782715,
      "step": 78935
    },
    {
      "epoch": 4.706382751464843e-06,
      "step": 78935,
      "training_step_time": 7.8296263217926025
    },
    {
      "epoch": 4.70733642578125e-06,
      "model_forward_time": 1.7674925327301025,
      "step": 78936
    },
    {
      "epoch": 4.70733642578125e-06,
      "step": 78936,
      "training_step_time": 7.697340726852417
    },
    {
      "epoch": 4.7082901000976565e-06,
      "model_forward_time": 1.7632372379302979,
      "step": 78937
    },
    {
      "epoch": 4.7082901000976565e-06,
      "step": 78937,
      "training_step_time": 7.615604400634766
    },
    {
      "epoch": 4.709243774414062e-06,
      "model_forward_time": 1.7595133781433105,
      "step": 78938
    },
    {
      "epoch": 4.709243774414062e-06,
      "step": 78938,
      "training_step_time": 6.926072597503662
    },
    {
      "epoch": 4.710197448730469e-06,
      "model_forward_time": 1.771468162536621,
      "step": 78939
    },
    {
      "epoch": 4.710197448730469e-06,
      "step": 78939,
      "training_step_time": 7.613657236099243
    },
    {
      "epoch": 4.711151123046875e-06,
      "grad_norm": 0.0904877632856369,
      "learning_rate": 1.1644570871994143e-06,
      "loss": 0.0534,
      "step": 78940
    },
    {
      "epoch": 4.711151123046875e-06,
      "model_forward_time": 1.7553400993347168,
      "step": 78940
    },
    {
      "dynamics_loss_avg": 0.04572046548128128,
      "epoch": 4.711151123046875e-06,
      "step": 78940
    },
    {
      "action_loss_avg": 0.001709654845762998,
      "epoch": 4.711151123046875e-06,
      "step": 78940
    },
    {
      "epoch": 4.711151123046875e-06,
      "step": 78940,
      "training_step_time": 7.684494256973267
    },
    {
      "epoch": 4.712104797363281e-06,
      "model_forward_time": 1.7728862762451172,
      "step": 78941
    },
    {
      "epoch": 4.712104797363281e-06,
      "step": 78941,
      "training_step_time": 7.761190891265869
    },
    {
      "epoch": 4.713058471679688e-06,
      "model_forward_time": 1.78371262550354,
      "step": 78942
    },
    {
      "epoch": 4.713058471679688e-06,
      "step": 78942,
      "training_step_time": 8.775625467300415
    },
    {
      "epoch": 4.714012145996093e-06,
      "model_forward_time": 1.758263111114502,
      "step": 78943
    },
    {
      "epoch": 4.714012145996093e-06,
      "step": 78943,
      "training_step_time": 8.402806043624878
    },
    {
      "epoch": 4.7149658203125e-06,
      "model_forward_time": 1.7994952201843262,
      "step": 78944
    },
    {
      "epoch": 4.7149658203125e-06,
      "step": 78944,
      "training_step_time": 8.41045069694519
    },
    {
      "epoch": 4.7159194946289065e-06,
      "model_forward_time": 2.909045934677124,
      "step": 78945
    },
    {
      "epoch": 4.7159194946289065e-06,
      "step": 78945,
      "training_step_time": 8.697404623031616
    },
    {
      "epoch": 4.716873168945312e-06,
      "model_forward_time": 1.761570930480957,
      "step": 78946
    },
    {
      "epoch": 4.716873168945312e-06,
      "step": 78946,
      "training_step_time": 7.525109052658081
    },
    {
      "epoch": 4.717826843261719e-06,
      "model_forward_time": 1.7604913711547852,
      "step": 78947
    },
    {
      "epoch": 4.717826843261719e-06,
      "step": 78947,
      "training_step_time": 7.28621244430542
    },
    {
      "epoch": 4.718780517578125e-06,
      "model_forward_time": 1.7634742259979248,
      "step": 78948
    },
    {
      "epoch": 4.718780517578125e-06,
      "step": 78948,
      "training_step_time": 7.434306621551514
    },
    {
      "epoch": 4.719734191894531e-06,
      "model_forward_time": 1.7549958229064941,
      "step": 78949
    },
    {
      "epoch": 4.719734191894531e-06,
      "step": 78949,
      "training_step_time": 7.474950790405273
    },
    {
      "epoch": 4.720687866210938e-06,
      "grad_norm": 0.10379205644130707,
      "learning_rate": 1.163396567480744e-06,
      "loss": 0.0533,
      "step": 78950
    },
    {
      "epoch": 4.720687866210938e-06,
      "model_forward_time": 1.774465560913086,
      "step": 78950
    },
    {
      "dynamics_loss_avg": 0.05588631965219974,
      "epoch": 4.720687866210938e-06,
      "step": 78950
    },
    {
      "action_loss_avg": 0.0023909779672976585,
      "epoch": 4.720687866210938e-06,
      "step": 78950
    },
    {
      "epoch": 4.720687866210938e-06,
      "step": 78950,
      "training_step_time": 8.480518817901611
    },
    {
      "epoch": 4.721641540527343e-06,
      "model_forward_time": 1.7808139324188232,
      "step": 78951
    },
    {
      "epoch": 4.721641540527343e-06,
      "step": 78951,
      "training_step_time": 9.668636083602905
    },
    {
      "epoch": 4.72259521484375e-06,
      "model_forward_time": 1.7755541801452637,
      "step": 78952
    },
    {
      "epoch": 4.72259521484375e-06,
      "step": 78952,
      "training_step_time": 8.390294313430786
    },
    {
      "epoch": 4.7235488891601565e-06,
      "model_forward_time": 1.7982165813446045,
      "step": 78953
    },
    {
      "epoch": 4.7235488891601565e-06,
      "step": 78953,
      "training_step_time": 8.243870258331299
    },
    {
      "epoch": 4.724502563476562e-06,
      "model_forward_time": 2.8500871658325195,
      "step": 78954
    },
    {
      "epoch": 4.724502563476562e-06,
      "step": 78954,
      "training_step_time": 7.804619789123535
    },
    {
      "epoch": 4.725456237792969e-06,
      "model_forward_time": 1.758481502532959,
      "step": 78955
    },
    {
      "epoch": 4.725456237792969e-06,
      "step": 78955,
      "training_step_time": 7.525209426879883
    },
    {
      "epoch": 4.726409912109375e-06,
      "model_forward_time": 1.770118236541748,
      "step": 78956
    },
    {
      "epoch": 4.726409912109375e-06,
      "step": 78956,
      "training_step_time": 7.286656856536865
    },
    {
      "epoch": 4.727363586425781e-06,
      "model_forward_time": 1.7710058689117432,
      "step": 78957
    },
    {
      "epoch": 4.727363586425781e-06,
      "step": 78957,
      "training_step_time": 7.530601263046265
    },
    {
      "epoch": 4.728317260742188e-06,
      "model_forward_time": 1.7637650966644287,
      "step": 78958
    },
    {
      "epoch": 4.728317260742188e-06,
      "step": 78958,
      "training_step_time": 7.588952302932739
    },
    {
      "epoch": 4.729270935058593e-06,
      "model_forward_time": 1.769374132156372,
      "step": 78959
    },
    {
      "epoch": 4.729270935058593e-06,
      "step": 78959,
      "training_step_time": 7.613389015197754
    },
    {
      "epoch": 4.730224609375e-06,
      "grad_norm": 0.1199081614613533,
      "learning_rate": 1.1623364673272553e-06,
      "loss": 0.0541,
      "step": 78960
    },
    {
      "epoch": 4.730224609375e-06,
      "model_forward_time": 1.7738280296325684,
      "step": 78960
    },
    {
      "dynamics_loss_avg": 0.05769589561969042,
      "epoch": 4.730224609375e-06,
      "step": 78960
    },
    {
      "action_loss_avg": 0.0014167227782309055,
      "epoch": 4.730224609375e-06,
      "step": 78960
    },
    {
      "epoch": 4.730224609375e-06,
      "step": 78960,
      "training_step_time": 8.71210527420044
    },
    {
      "epoch": 4.7311782836914065e-06,
      "model_forward_time": 1.389188528060913,
      "step": 78961
    },
    {
      "epoch": 4.7311782836914065e-06,
      "step": 78961,
      "training_step_time": 9.503955602645874
    },
    {
      "epoch": 4.732131958007812e-06,
      "model_forward_time": 1.7876579761505127,
      "step": 78962
    },
    {
      "epoch": 4.732131958007812e-06,
      "step": 78962,
      "training_step_time": 9.134228229522705
    },
    {
      "epoch": 4.733085632324219e-06,
      "model_forward_time": 2.8712973594665527,
      "step": 78963
    },
    {
      "epoch": 4.733085632324219e-06,
      "step": 78963,
      "training_step_time": 8.198217630386353
    },
    {
      "epoch": 4.734039306640625e-06,
      "model_forward_time": 1.7664480209350586,
      "step": 78964
    },
    {
      "epoch": 4.734039306640625e-06,
      "step": 78964,
      "training_step_time": 7.564795732498169
    },
    {
      "epoch": 4.734992980957031e-06,
      "model_forward_time": 1.7690296173095703,
      "step": 78965
    },
    {
      "epoch": 4.734992980957031e-06,
      "step": 78965,
      "training_step_time": 7.262113571166992
    },
    {
      "epoch": 4.735946655273438e-06,
      "model_forward_time": 1.7488372325897217,
      "step": 78966
    },
    {
      "epoch": 4.735946655273438e-06,
      "step": 78966,
      "training_step_time": 7.4163124561309814
    },
    {
      "epoch": 4.736900329589843e-06,
      "model_forward_time": 1.7615911960601807,
      "step": 78967
    },
    {
      "epoch": 4.736900329589843e-06,
      "step": 78967,
      "training_step_time": 7.558403730392456
    },
    {
      "epoch": 4.73785400390625e-06,
      "model_forward_time": 1.7631735801696777,
      "step": 78968
    },
    {
      "epoch": 4.73785400390625e-06,
      "step": 78968,
      "training_step_time": 7.601298809051514
    },
    {
      "epoch": 4.7388076782226565e-06,
      "model_forward_time": 1.781590223312378,
      "step": 78969
    },
    {
      "epoch": 4.7388076782226565e-06,
      "step": 78969,
      "training_step_time": 10.023520231246948
    },
    {
      "epoch": 4.739761352539062e-06,
      "grad_norm": 0.12896135449409485,
      "learning_rate": 1.1612767868548808e-06,
      "loss": 0.0528,
      "step": 78970
    },
    {
      "epoch": 4.739761352539062e-06,
      "model_forward_time": 1.770531177520752,
      "step": 78970
    },
    {
      "dynamics_loss_avg": 0.05586824389174581,
      "epoch": 4.739761352539062e-06,
      "step": 78970
    },
    {
      "action_loss_avg": 0.0021261321584461258,
      "epoch": 4.739761352539062e-06,
      "step": 78970
    },
    {
      "epoch": 4.739761352539062e-06,
      "step": 78970,
      "training_step_time": 8.45300579071045
    },
    {
      "epoch": 4.740715026855469e-06,
      "model_forward_time": 1.784555196762085,
      "step": 78971
    },
    {
      "epoch": 4.740715026855469e-06,
      "step": 78971,
      "training_step_time": 9.549005031585693
    },
    {
      "epoch": 4.741668701171875e-06,
      "model_forward_time": 2.8458032608032227,
      "step": 78972
    },
    {
      "epoch": 4.741668701171875e-06,
      "step": 78972,
      "training_step_time": 8.268490552902222
    },
    {
      "epoch": 4.742622375488281e-06,
      "model_forward_time": 1.775125503540039,
      "step": 78973
    },
    {
      "epoch": 4.742622375488281e-06,
      "step": 78973,
      "training_step_time": 7.669736385345459
    },
    {
      "epoch": 4.743576049804688e-06,
      "model_forward_time": 1.7518932819366455,
      "step": 78974
    },
    {
      "epoch": 4.743576049804688e-06,
      "step": 78974,
      "training_step_time": 7.547524452209473
    },
    {
      "epoch": 4.744529724121093e-06,
      "model_forward_time": 1.7504644393920898,
      "step": 78975
    },
    {
      "epoch": 4.744529724121093e-06,
      "step": 78975,
      "training_step_time": 7.483403921127319
    },
    {
      "epoch": 4.7454833984375e-06,
      "model_forward_time": 1.7497549057006836,
      "step": 78976
    },
    {
      "epoch": 4.7454833984375e-06,
      "step": 78976,
      "training_step_time": 7.2662787437438965
    },
    {
      "epoch": 4.7464370727539065e-06,
      "model_forward_time": 1.748913288116455,
      "step": 78977
    },
    {
      "epoch": 4.7464370727539065e-06,
      "step": 78977,
      "training_step_time": 7.422808885574341
    },
    {
      "epoch": 4.747390747070312e-06,
      "model_forward_time": 1.7455227375030518,
      "step": 78978
    },
    {
      "epoch": 4.747390747070312e-06,
      "step": 78978,
      "training_step_time": 8.161922693252563
    },
    {
      "epoch": 4.748344421386719e-06,
      "model_forward_time": 1.774465560913086,
      "step": 78979
    },
    {
      "epoch": 4.748344421386719e-06,
      "step": 78979,
      "training_step_time": 7.721780300140381
    },
    {
      "epoch": 4.749298095703125e-06,
      "grad_norm": 0.11619497090578079,
      "learning_rate": 1.1602175261795057e-06,
      "loss": 0.0554,
      "step": 78980
    },
    {
      "epoch": 4.749298095703125e-06,
      "model_forward_time": 1.75614595413208,
      "step": 78980
    },
    {
      "dynamics_loss_avg": 0.05614629853516817,
      "epoch": 4.749298095703125e-06,
      "step": 78980
    },
    {
      "action_loss_avg": 0.0013652149296831339,
      "epoch": 4.749298095703125e-06,
      "step": 78980
    },
    {
      "epoch": 4.749298095703125e-06,
      "step": 78980,
      "training_step_time": 8.968534469604492
    },
    {
      "epoch": 4.750251770019531e-06,
      "model_forward_time": 1.7869346141815186,
      "step": 78981
    },
    {
      "epoch": 4.750251770019531e-06,
      "step": 78981,
      "training_step_time": 8.331312656402588
    },
    {
      "epoch": 4.751205444335938e-06,
      "model_forward_time": 1.7577009201049805,
      "step": 78982
    },
    {
      "epoch": 4.751205444335938e-06,
      "step": 78982,
      "training_step_time": 7.5752112865448
    },
    {
      "epoch": 4.752159118652343e-06,
      "model_forward_time": 1.7511615753173828,
      "step": 78983
    },
    {
      "epoch": 4.752159118652343e-06,
      "step": 78983,
      "training_step_time": 7.498539447784424
    },
    {
      "epoch": 4.75311279296875e-06,
      "model_forward_time": 1.771453857421875,
      "step": 78984
    },
    {
      "epoch": 4.75311279296875e-06,
      "step": 78984,
      "training_step_time": 7.484149932861328
    },
    {
      "epoch": 4.7540664672851566e-06,
      "model_forward_time": 1.7437989711761475,
      "step": 78985
    },
    {
      "epoch": 4.7540664672851566e-06,
      "step": 78985,
      "training_step_time": 6.713025093078613
    },
    {
      "epoch": 4.755020141601562e-06,
      "model_forward_time": 1.7628343105316162,
      "step": 78986
    },
    {
      "epoch": 4.755020141601562e-06,
      "step": 78986,
      "training_step_time": 7.685388803482056
    },
    {
      "epoch": 4.755973815917969e-06,
      "model_forward_time": 1.766066551208496,
      "step": 78987
    },
    {
      "epoch": 4.755973815917969e-06,
      "step": 78987,
      "training_step_time": 8.23086404800415
    },
    {
      "epoch": 4.756927490234375e-06,
      "model_forward_time": 1.7749176025390625,
      "step": 78988
    },
    {
      "epoch": 4.756927490234375e-06,
      "step": 78988,
      "training_step_time": 9.772648096084595
    },
    {
      "epoch": 4.757881164550781e-06,
      "model_forward_time": 1.7718193531036377,
      "step": 78989
    },
    {
      "epoch": 4.757881164550781e-06,
      "step": 78989,
      "training_step_time": 10.225135564804077
    },
    {
      "epoch": 4.758834838867188e-06,
      "grad_norm": 0.11911365389823914,
      "learning_rate": 1.1591586854169696e-06,
      "loss": 0.0549,
      "step": 78990
    },
    {
      "epoch": 4.758834838867188e-06,
      "model_forward_time": 1.7730402946472168,
      "step": 78990
    },
    {
      "dynamics_loss_avg": 0.05329022612422705,
      "epoch": 4.758834838867188e-06,
      "step": 78990
    },
    {
      "action_loss_avg": 0.001214766368502751,
      "epoch": 4.758834838867188e-06,
      "step": 78990
    },
    {
      "epoch": 4.758834838867188e-06,
      "step": 78990,
      "training_step_time": 7.746943712234497
    },
    {
      "epoch": 4.7597885131835934e-06,
      "model_forward_time": 2.912661552429199,
      "step": 78991
    },
    {
      "epoch": 4.7597885131835934e-06,
      "step": 78991,
      "training_step_time": 8.786609649658203
    },
    {
      "epoch": 4.7607421875e-06,
      "model_forward_time": 1.7640247344970703,
      "step": 78992
    },
    {
      "epoch": 4.7607421875e-06,
      "step": 78992,
      "training_step_time": 7.432555675506592
    },
    {
      "epoch": 4.7616958618164066e-06,
      "model_forward_time": 1.7660651206970215,
      "step": 78993
    },
    {
      "epoch": 4.7616958618164066e-06,
      "step": 78993,
      "training_step_time": 7.485229969024658
    },
    {
      "epoch": 4.762649536132812e-06,
      "model_forward_time": 1.774430274963379,
      "step": 78994
    },
    {
      "epoch": 4.762649536132812e-06,
      "step": 78994,
      "training_step_time": 6.76170539855957
    },
    {
      "epoch": 4.763603210449219e-06,
      "model_forward_time": 1.771117925643921,
      "step": 78995
    },
    {
      "epoch": 4.763603210449219e-06,
      "step": 78995,
      "training_step_time": 7.762927532196045
    },
    {
      "epoch": 4.764556884765625e-06,
      "model_forward_time": 1.7581946849822998,
      "step": 78996
    },
    {
      "epoch": 4.764556884765625e-06,
      "step": 78996,
      "training_step_time": 7.3864946365356445
    },
    {
      "epoch": 4.765510559082031e-06,
      "model_forward_time": 1.7739193439483643,
      "step": 78997
    },
    {
      "epoch": 4.765510559082031e-06,
      "step": 78997,
      "training_step_time": 8.43872880935669
    },
    {
      "epoch": 4.766464233398438e-06,
      "model_forward_time": 1.7895450592041016,
      "step": 78998
    },
    {
      "epoch": 4.766464233398438e-06,
      "step": 78998,
      "training_step_time": 9.214152812957764
    },
    {
      "epoch": 4.7674179077148434e-06,
      "model_forward_time": 1.752471923828125,
      "step": 78999
    },
    {
      "epoch": 4.7674179077148434e-06,
      "step": 78999,
      "training_step_time": 7.8045899868011475
    },
    {
      "epoch": 4.76837158203125e-06,
      "grad_norm": 0.08177686482667923,
      "learning_rate": 1.1581002646830642e-06,
      "loss": 0.0543,
      "step": 79000
    },
    {
      "epoch": 4.76837158203125e-06,
      "model_forward_time": 1.8348112106323242,
      "step": 79000
    },
    {
      "dynamics_loss_avg": 0.05747305639088154,
      "epoch": 4.76837158203125e-06,
      "step": 79000
    },
    {
      "action_loss_avg": 0.001652690613991581,
      "epoch": 4.76837158203125e-06,
      "step": 79000
    },
    {
      "epoch": 4.76837158203125e-06,
      "step": 79000,
      "training_step_time": 6.6918933391571045
    },
    {
      "epoch": 4.7693252563476566e-06,
      "model_forward_time": 1.7540428638458252,
      "step": 79001
    },
    {
      "epoch": 4.7693252563476566e-06,
      "step": 79001,
      "training_step_time": 8.490073204040527
    },
    {
      "epoch": 4.770278930664062e-06,
      "model_forward_time": 1.7555580139160156,
      "step": 79002
    },
    {
      "epoch": 4.770278930664062e-06,
      "step": 79002,
      "training_step_time": 7.62663197517395
    },
    {
      "epoch": 4.771232604980469e-06,
      "model_forward_time": 1.756727933883667,
      "step": 79003
    },
    {
      "epoch": 4.771232604980469e-06,
      "step": 79003,
      "training_step_time": 6.846874713897705
    },
    {
      "epoch": 4.772186279296875e-06,
      "model_forward_time": 1.7505900859832764,
      "step": 79004
    },
    {
      "epoch": 4.772186279296875e-06,
      "step": 79004,
      "training_step_time": 7.70670485496521
    },
    {
      "epoch": 4.773139953613281e-06,
      "model_forward_time": 1.7538516521453857,
      "step": 79005
    },
    {
      "epoch": 4.773139953613281e-06,
      "step": 79005,
      "training_step_time": 8.313578367233276
    },
    {
      "epoch": 4.774093627929688e-06,
      "model_forward_time": 1.754533052444458,
      "step": 79006
    },
    {
      "epoch": 4.774093627929688e-06,
      "step": 79006,
      "training_step_time": 11.086525917053223
    },
    {
      "epoch": 4.7750473022460934e-06,
      "model_forward_time": 1.7737045288085938,
      "step": 79007
    },
    {
      "epoch": 4.7750473022460934e-06,
      "step": 79007,
      "training_step_time": 11.554832935333252
    },
    {
      "epoch": 4.7760009765625e-06,
      "model_forward_time": 1.7593235969543457,
      "step": 79008
    },
    {
      "epoch": 4.7760009765625e-06,
      "step": 79008,
      "training_step_time": 12.59873914718628
    },
    {
      "epoch": 4.7769546508789066e-06,
      "model_forward_time": 1.7648379802703857,
      "step": 79009
    },
    {
      "epoch": 4.7769546508789066e-06,
      "step": 79009,
      "training_step_time": 7.720138072967529
    },
    {
      "epoch": 4.777908325195312e-06,
      "grad_norm": 0.09852913022041321,
      "learning_rate": 1.1570422640935386e-06,
      "loss": 0.0565,
      "step": 79010
    },
    {
      "epoch": 4.777908325195312e-06,
      "model_forward_time": 1.750166654586792,
      "step": 79010
    },
    {
      "dynamics_loss_avg": 0.05708868699148297,
      "epoch": 4.777908325195312e-06,
      "step": 79010
    },
    {
      "action_loss_avg": 0.0013798936386592686,
      "epoch": 4.777908325195312e-06,
      "step": 79010
    },
    {
      "epoch": 4.777908325195312e-06,
      "step": 79010,
      "training_step_time": 7.539793252944946
    },
    {
      "epoch": 4.778861999511719e-06,
      "model_forward_time": 1.7530732154846191,
      "step": 79011
    },
    {
      "epoch": 4.778861999511719e-06,
      "step": 79011,
      "training_step_time": 7.58122706413269
    },
    {
      "epoch": 4.779815673828125e-06,
      "model_forward_time": 1.738206148147583,
      "step": 79012
    },
    {
      "epoch": 4.779815673828125e-06,
      "step": 79012,
      "training_step_time": 6.752114295959473
    },
    {
      "epoch": 4.780769348144531e-06,
      "model_forward_time": 1.7392916679382324,
      "step": 79013
    },
    {
      "epoch": 4.780769348144531e-06,
      "step": 79013,
      "training_step_time": 7.563930034637451
    },
    {
      "epoch": 4.781723022460938e-06,
      "model_forward_time": 1.7429132461547852,
      "step": 79014
    },
    {
      "epoch": 4.781723022460938e-06,
      "step": 79014,
      "training_step_time": 7.238809823989868
    },
    {
      "epoch": 4.7826766967773434e-06,
      "model_forward_time": 1.7377710342407227,
      "step": 79015
    },
    {
      "epoch": 4.7826766967773434e-06,
      "step": 79015,
      "training_step_time": 8.449481964111328
    },
    {
      "epoch": 4.78363037109375e-06,
      "model_forward_time": 1.7601103782653809,
      "step": 79016
    },
    {
      "epoch": 4.78363037109375e-06,
      "step": 79016,
      "training_step_time": 8.574796676635742
    },
    {
      "epoch": 4.784584045410157e-06,
      "model_forward_time": 1.737217903137207,
      "step": 79017
    },
    {
      "epoch": 4.784584045410157e-06,
      "step": 79017,
      "training_step_time": 8.71461009979248
    },
    {
      "epoch": 4.785537719726562e-06,
      "model_forward_time": 2.8430490493774414,
      "step": 79018
    },
    {
      "epoch": 4.785537719726562e-06,
      "step": 79018,
      "training_step_time": 8.783997535705566
    },
    {
      "epoch": 4.786491394042969e-06,
      "model_forward_time": 1.7557837963104248,
      "step": 79019
    },
    {
      "epoch": 4.786491394042969e-06,
      "step": 79019,
      "training_step_time": 7.548682451248169
    },
    {
      "epoch": 4.787445068359375e-06,
      "grad_norm": 0.07520181685686111,
      "learning_rate": 1.155984683764092e-06,
      "loss": 0.0553,
      "step": 79020
    },
    {
      "epoch": 4.787445068359375e-06,
      "model_forward_time": 1.7461063861846924,
      "step": 79020
    },
    {
      "dynamics_loss_avg": 0.04564960747957229,
      "epoch": 4.787445068359375e-06,
      "step": 79020
    },
    {
      "action_loss_avg": 0.0015166648488957434,
      "epoch": 4.787445068359375e-06,
      "step": 79020
    },
    {
      "epoch": 4.787445068359375e-06,
      "step": 79020,
      "training_step_time": 7.56907320022583
    },
    {
      "epoch": 4.788398742675781e-06,
      "model_forward_time": 1.7397880554199219,
      "step": 79021
    },
    {
      "epoch": 4.788398742675781e-06,
      "step": 79021,
      "training_step_time": 6.770437002182007
    },
    {
      "epoch": 4.789352416992188e-06,
      "model_forward_time": 1.7530715465545654,
      "step": 79022
    },
    {
      "epoch": 4.789352416992188e-06,
      "step": 79022,
      "training_step_time": 7.585490942001343
    },
    {
      "epoch": 4.7903060913085935e-06,
      "model_forward_time": 1.7538702487945557,
      "step": 79023
    },
    {
      "epoch": 4.7903060913085935e-06,
      "step": 79023,
      "training_step_time": 7.147669553756714
    },
    {
      "epoch": 4.791259765625e-06,
      "model_forward_time": 1.747284173965454,
      "step": 79024
    },
    {
      "epoch": 4.791259765625e-06,
      "step": 79024,
      "training_step_time": 7.619415998458862
    },
    {
      "epoch": 4.792213439941407e-06,
      "model_forward_time": 1.7614998817443848,
      "step": 79025
    },
    {
      "epoch": 4.792213439941407e-06,
      "step": 79025,
      "training_step_time": 8.521990060806274
    },
    {
      "epoch": 4.793167114257812e-06,
      "model_forward_time": 1.7382090091705322,
      "step": 79026
    },
    {
      "epoch": 4.793167114257812e-06,
      "step": 79026,
      "training_step_time": 8.366693019866943
    },
    {
      "epoch": 4.794120788574219e-06,
      "model_forward_time": 2.7914488315582275,
      "step": 79027
    },
    {
      "epoch": 4.794120788574219e-06,
      "step": 79027,
      "training_step_time": 8.355998039245605
    },
    {
      "epoch": 4.795074462890625e-06,
      "model_forward_time": 1.746478796005249,
      "step": 79028
    },
    {
      "epoch": 4.795074462890625e-06,
      "step": 79028,
      "training_step_time": 7.552696704864502
    },
    {
      "epoch": 4.796028137207031e-06,
      "model_forward_time": 1.7363874912261963,
      "step": 79029
    },
    {
      "epoch": 4.796028137207031e-06,
      "step": 79029,
      "training_step_time": 7.594958066940308
    },
    {
      "epoch": 4.796981811523438e-06,
      "grad_norm": 0.09255070984363556,
      "learning_rate": 1.1549275238103813e-06,
      "loss": 0.0549,
      "step": 79030
    },
    {
      "epoch": 4.796981811523438e-06,
      "model_forward_time": 1.7344989776611328,
      "step": 79030
    },
    {
      "dynamics_loss_avg": 0.050258179381489756,
      "epoch": 4.796981811523438e-06,
      "step": 79030
    },
    {
      "action_loss_avg": 0.0013381426397245377,
      "epoch": 4.796981811523438e-06,
      "step": 79030
    },
    {
      "epoch": 4.796981811523438e-06,
      "step": 79030,
      "training_step_time": 7.407029628753662
    },
    {
      "epoch": 4.7979354858398435e-06,
      "model_forward_time": 1.736576795578003,
      "step": 79031
    },
    {
      "epoch": 4.7979354858398435e-06,
      "step": 79031,
      "training_step_time": 7.406213760375977
    },
    {
      "epoch": 4.79888916015625e-06,
      "model_forward_time": 1.7426726818084717,
      "step": 79032
    },
    {
      "epoch": 4.79888916015625e-06,
      "step": 79032,
      "training_step_time": 7.673809051513672
    },
    {
      "epoch": 4.799842834472657e-06,
      "model_forward_time": 1.744819164276123,
      "step": 79033
    },
    {
      "epoch": 4.799842834472657e-06,
      "step": 79033,
      "training_step_time": 7.466498374938965
    },
    {
      "epoch": 4.800796508789062e-06,
      "model_forward_time": 1.7602465152740479,
      "step": 79034
    },
    {
      "epoch": 4.800796508789062e-06,
      "step": 79034,
      "training_step_time": 8.437158346176147
    },
    {
      "epoch": 4.801750183105469e-06,
      "model_forward_time": 1.7450456619262695,
      "step": 79035
    },
    {
      "epoch": 4.801750183105469e-06,
      "step": 79035,
      "training_step_time": 10.39036250114441
    },
    {
      "epoch": 4.802703857421875e-06,
      "model_forward_time": 2.8282904624938965,
      "step": 79036
    },
    {
      "epoch": 4.802703857421875e-06,
      "step": 79036,
      "training_step_time": 9.72174620628357
    },
    {
      "epoch": 4.803657531738281e-06,
      "model_forward_time": 1.7516281604766846,
      "step": 79037
    },
    {
      "epoch": 4.803657531738281e-06,
      "step": 79037,
      "training_step_time": 8.622653484344482
    },
    {
      "epoch": 4.804611206054688e-06,
      "model_forward_time": 1.7533206939697266,
      "step": 79038
    },
    {
      "epoch": 4.804611206054688e-06,
      "step": 79038,
      "training_step_time": 7.535770893096924
    },
    {
      "epoch": 4.8055648803710935e-06,
      "model_forward_time": 1.755183219909668,
      "step": 79039
    },
    {
      "epoch": 4.8055648803710935e-06,
      "step": 79039,
      "training_step_time": 7.383844375610352
    },
    {
      "epoch": 4.8065185546875e-06,
      "grad_norm": 0.10750314593315125,
      "learning_rate": 1.1538707843480151e-06,
      "loss": 0.0559,
      "step": 79040
    },
    {
      "epoch": 4.8065185546875e-06,
      "model_forward_time": 1.752485990524292,
      "step": 79040
    },
    {
      "dynamics_loss_avg": 0.035870675928890705,
      "epoch": 4.8065185546875e-06,
      "step": 79040
    },
    {
      "action_loss_avg": 0.001008574542356655,
      "epoch": 4.8065185546875e-06,
      "step": 79040
    },
    {
      "epoch": 4.8065185546875e-06,
      "step": 79040,
      "training_step_time": 6.8367390632629395
    },
    {
      "epoch": 4.807472229003907e-06,
      "model_forward_time": 1.7603850364685059,
      "step": 79041
    },
    {
      "epoch": 4.807472229003907e-06,
      "step": 79041,
      "training_step_time": 8.18609094619751
    },
    {
      "epoch": 4.808425903320312e-06,
      "model_forward_time": 1.7530226707458496,
      "step": 79042
    },
    {
      "epoch": 4.808425903320312e-06,
      "step": 79042,
      "training_step_time": 7.467502593994141
    },
    {
      "epoch": 4.809379577636719e-06,
      "model_forward_time": 1.7603843212127686,
      "step": 79043
    },
    {
      "epoch": 4.809379577636719e-06,
      "step": 79043,
      "training_step_time": 7.7081544399261475
    },
    {
      "epoch": 4.810333251953125e-06,
      "model_forward_time": 1.7562856674194336,
      "step": 79044
    },
    {
      "epoch": 4.810333251953125e-06,
      "step": 79044,
      "training_step_time": 8.486181497573853
    },
    {
      "epoch": 4.811286926269531e-06,
      "model_forward_time": 2.8926076889038086,
      "step": 79045
    },
    {
      "epoch": 4.811286926269531e-06,
      "step": 79045,
      "training_step_time": 9.546302556991577
    },
    {
      "epoch": 4.812240600585938e-06,
      "model_forward_time": 1.7815475463867188,
      "step": 79046
    },
    {
      "epoch": 4.812240600585938e-06,
      "step": 79046,
      "training_step_time": 7.756446361541748
    },
    {
      "epoch": 4.8131942749023435e-06,
      "model_forward_time": 1.7536580562591553,
      "step": 79047
    },
    {
      "epoch": 4.8131942749023435e-06,
      "step": 79047,
      "training_step_time": 8.497477293014526
    },
    {
      "epoch": 4.81414794921875e-06,
      "model_forward_time": 1.7654404640197754,
      "step": 79048
    },
    {
      "epoch": 4.81414794921875e-06,
      "step": 79048,
      "training_step_time": 7.582967042922974
    },
    {
      "epoch": 4.815101623535157e-06,
      "model_forward_time": 1.7562077045440674,
      "step": 79049
    },
    {
      "epoch": 4.815101623535157e-06,
      "step": 79049,
      "training_step_time": 6.744530200958252
    },
    {
      "epoch": 4.816055297851562e-06,
      "grad_norm": 0.08953655511140823,
      "learning_rate": 1.1528144654925595e-06,
      "loss": 0.0574,
      "step": 79050
    },
    {
      "epoch": 4.816055297851562e-06,
      "model_forward_time": 1.7452349662780762,
      "step": 79050
    },
    {
      "dynamics_loss_avg": 0.06334654912352562,
      "epoch": 4.816055297851562e-06,
      "step": 79050
    },
    {
      "action_loss_avg": 0.001307600032305345,
      "epoch": 4.816055297851562e-06,
      "step": 79050
    },
    {
      "epoch": 4.816055297851562e-06,
      "step": 79050,
      "training_step_time": 8.116785526275635
    },
    {
      "epoch": 4.817008972167969e-06,
      "model_forward_time": 1.7439181804656982,
      "step": 79051
    },
    {
      "epoch": 4.817008972167969e-06,
      "step": 79051,
      "training_step_time": 7.299621105194092
    },
    {
      "epoch": 4.817962646484375e-06,
      "model_forward_time": 1.7452595233917236,
      "step": 79052
    },
    {
      "epoch": 4.817962646484375e-06,
      "step": 79052,
      "training_step_time": 7.645675420761108
    },
    {
      "epoch": 4.818916320800781e-06,
      "model_forward_time": 1.7695279121398926,
      "step": 79053
    },
    {
      "epoch": 4.818916320800781e-06,
      "step": 79053,
      "training_step_time": 8.182455778121948
    },
    {
      "epoch": 4.819869995117188e-06,
      "model_forward_time": 2.868304967880249,
      "step": 79054
    },
    {
      "epoch": 4.819869995117188e-06,
      "step": 79054,
      "training_step_time": 8.533481359481812
    },
    {
      "epoch": 4.8208236694335935e-06,
      "model_forward_time": 1.7769017219543457,
      "step": 79055
    },
    {
      "epoch": 4.8208236694335935e-06,
      "step": 79055,
      "training_step_time": 10.00527834892273
    },
    {
      "epoch": 4.82177734375e-06,
      "model_forward_time": 1.7621369361877441,
      "step": 79056
    },
    {
      "epoch": 4.82177734375e-06,
      "step": 79056,
      "training_step_time": 7.513294696807861
    },
    {
      "epoch": 4.822731018066407e-06,
      "model_forward_time": 1.7592658996582031,
      "step": 79057
    },
    {
      "epoch": 4.822731018066407e-06,
      "step": 79057,
      "training_step_time": 8.277871370315552
    },
    {
      "epoch": 4.823684692382812e-06,
      "model_forward_time": 1.7605981826782227,
      "step": 79058
    },
    {
      "epoch": 4.823684692382812e-06,
      "step": 79058,
      "training_step_time": 6.7087273597717285
    },
    {
      "epoch": 4.824638366699219e-06,
      "model_forward_time": 1.7597112655639648,
      "step": 79059
    },
    {
      "epoch": 4.824638366699219e-06,
      "step": 79059,
      "training_step_time": 7.616954565048218
    },
    {
      "epoch": 4.825592041015625e-06,
      "grad_norm": 0.10752739012241364,
      "learning_rate": 1.1517585673595277e-06,
      "loss": 0.0566,
      "step": 79060
    },
    {
      "epoch": 4.825592041015625e-06,
      "model_forward_time": 1.745572566986084,
      "step": 79060
    },
    {
      "dynamics_loss_avg": 0.046606828551739456,
      "epoch": 4.825592041015625e-06,
      "step": 79060
    },
    {
      "action_loss_avg": 0.0009408457553945482,
      "epoch": 4.825592041015625e-06,
      "step": 79060
    },
    {
      "epoch": 4.825592041015625e-06,
      "step": 79060,
      "training_step_time": 7.526957988739014
    },
    {
      "epoch": 4.826545715332031e-06,
      "model_forward_time": 1.7482450008392334,
      "step": 79061
    },
    {
      "epoch": 4.826545715332031e-06,
      "step": 79061,
      "training_step_time": 8.220962524414062
    },
    {
      "epoch": 4.827499389648438e-06,
      "model_forward_time": 1.780489444732666,
      "step": 79062
    },
    {
      "epoch": 4.827499389648438e-06,
      "step": 79062,
      "training_step_time": 7.652283430099487
    },
    {
      "epoch": 4.8284530639648435e-06,
      "model_forward_time": 2.826233148574829,
      "step": 79063
    },
    {
      "epoch": 4.8284530639648435e-06,
      "step": 79063,
      "training_step_time": 9.176691055297852
    },
    {
      "epoch": 4.82940673828125e-06,
      "model_forward_time": 1.7569773197174072,
      "step": 79064
    },
    {
      "epoch": 4.82940673828125e-06,
      "step": 79064,
      "training_step_time": 9.158926010131836
    },
    {
      "epoch": 4.830360412597657e-06,
      "model_forward_time": 1.7542974948883057,
      "step": 79065
    },
    {
      "epoch": 4.830360412597657e-06,
      "step": 79065,
      "training_step_time": 7.5726025104522705
    },
    {
      "epoch": 4.831314086914062e-06,
      "model_forward_time": 1.7471191883087158,
      "step": 79066
    },
    {
      "epoch": 4.831314086914062e-06,
      "step": 79066,
      "training_step_time": 7.506960153579712
    },
    {
      "epoch": 4.832267761230469e-06,
      "model_forward_time": 1.7456130981445312,
      "step": 79067
    },
    {
      "epoch": 4.832267761230469e-06,
      "step": 79067,
      "training_step_time": 7.259487867355347
    },
    {
      "epoch": 4.833221435546875e-06,
      "model_forward_time": 1.7451646327972412,
      "step": 79068
    },
    {
      "epoch": 4.833221435546875e-06,
      "step": 79068,
      "training_step_time": 7.628874778747559
    },
    {
      "epoch": 4.834175109863281e-06,
      "model_forward_time": 1.7264928817749023,
      "step": 79069
    },
    {
      "epoch": 4.834175109863281e-06,
      "step": 79069,
      "training_step_time": 8.321195363998413
    },
    {
      "epoch": 4.835128784179688e-06,
      "grad_norm": 0.19839753210544586,
      "learning_rate": 1.150703090064395e-06,
      "loss": 0.056,
      "step": 79070
    },
    {
      "epoch": 4.835128784179688e-06,
      "model_forward_time": 1.7399537563323975,
      "step": 79070
    },
    {
      "dynamics_loss_avg": 0.049390055052936074,
      "epoch": 4.835128784179688e-06,
      "step": 79070
    },
    {
      "action_loss_avg": 0.0009478354069869966,
      "epoch": 4.835128784179688e-06,
      "step": 79070
    },
    {
      "epoch": 4.835128784179688e-06,
      "step": 79070,
      "training_step_time": 8.32752537727356
    },
    {
      "epoch": 4.8360824584960935e-06,
      "model_forward_time": 1.7711679935455322,
      "step": 79071
    },
    {
      "epoch": 4.8360824584960935e-06,
      "step": 79071,
      "training_step_time": 8.287270784378052
    },
    {
      "epoch": 4.8370361328125e-06,
      "model_forward_time": 2.8086915016174316,
      "step": 79072
    },
    {
      "epoch": 4.8370361328125e-06,
      "step": 79072,
      "training_step_time": 8.495137214660645
    },
    {
      "epoch": 4.837989807128907e-06,
      "model_forward_time": 1.7638721466064453,
      "step": 79073
    },
    {
      "epoch": 4.837989807128907e-06,
      "step": 79073,
      "training_step_time": 8.384385108947754
    },
    {
      "epoch": 4.838943481445312e-06,
      "model_forward_time": 1.7464592456817627,
      "step": 79074
    },
    {
      "epoch": 4.838943481445312e-06,
      "step": 79074,
      "training_step_time": 7.710122346878052
    },
    {
      "epoch": 4.839897155761719e-06,
      "model_forward_time": 1.7534871101379395,
      "step": 79075
    },
    {
      "epoch": 4.839897155761719e-06,
      "step": 79075,
      "training_step_time": 7.497676849365234
    },
    {
      "epoch": 4.840850830078125e-06,
      "model_forward_time": 1.7498602867126465,
      "step": 79076
    },
    {
      "epoch": 4.840850830078125e-06,
      "step": 79076,
      "training_step_time": 7.593033075332642
    },
    {
      "epoch": 4.841804504394531e-06,
      "model_forward_time": 1.7487208843231201,
      "step": 79077
    },
    {
      "epoch": 4.841804504394531e-06,
      "step": 79077,
      "training_step_time": 8.51087498664856
    },
    {
      "epoch": 4.842758178710938e-06,
      "model_forward_time": 1.748741626739502,
      "step": 79078
    },
    {
      "epoch": 4.842758178710938e-06,
      "step": 79078,
      "training_step_time": 7.394985675811768
    },
    {
      "epoch": 4.8437118530273435e-06,
      "model_forward_time": 1.3733553886413574,
      "step": 79079
    },
    {
      "epoch": 4.8437118530273435e-06,
      "step": 79079,
      "training_step_time": 8.3731369972229
    },
    {
      "epoch": 4.84466552734375e-06,
      "grad_norm": 0.10494940727949142,
      "learning_rate": 1.1496480337225835e-06,
      "loss": 0.0578,
      "step": 79080
    },
    {
      "epoch": 4.84466552734375e-06,
      "model_forward_time": 1.7458224296569824,
      "step": 79080
    },
    {
      "dynamics_loss_avg": 0.05732770506292582,
      "epoch": 4.84466552734375e-06,
      "step": 79080
    },
    {
      "action_loss_avg": 0.001540913403732702,
      "epoch": 4.84466552734375e-06,
      "step": 79080
    },
    {
      "epoch": 4.84466552734375e-06,
      "step": 79080,
      "training_step_time": 8.564510583877563
    },
    {
      "epoch": 4.845619201660157e-06,
      "model_forward_time": 1.757969856262207,
      "step": 79081
    },
    {
      "epoch": 4.845619201660157e-06,
      "step": 79081,
      "training_step_time": 9.627257585525513
    },
    {
      "epoch": 4.846572875976562e-06,
      "model_forward_time": 2.9012205600738525,
      "step": 79082
    },
    {
      "epoch": 4.846572875976562e-06,
      "step": 79082,
      "training_step_time": 10.415894269943237
    },
    {
      "epoch": 4.847526550292969e-06,
      "model_forward_time": 1.7425625324249268,
      "step": 79083
    },
    {
      "epoch": 4.847526550292969e-06,
      "step": 79083,
      "training_step_time": 7.768850326538086
    },
    {
      "epoch": 4.848480224609375e-06,
      "model_forward_time": 1.7486188411712646,
      "step": 79084
    },
    {
      "epoch": 4.848480224609375e-06,
      "step": 79084,
      "training_step_time": 7.537087440490723
    },
    {
      "epoch": 4.849433898925781e-06,
      "model_forward_time": 1.7579150199890137,
      "step": 79085
    },
    {
      "epoch": 4.849433898925781e-06,
      "step": 79085,
      "training_step_time": 7.610287189483643
    },
    {
      "epoch": 4.850387573242188e-06,
      "model_forward_time": 1.7537109851837158,
      "step": 79086
    },
    {
      "epoch": 4.850387573242188e-06,
      "step": 79086,
      "training_step_time": 7.588826656341553
    },
    {
      "epoch": 4.8513412475585935e-06,
      "model_forward_time": 1.759474754333496,
      "step": 79087
    },
    {
      "epoch": 4.8513412475585935e-06,
      "step": 79087,
      "training_step_time": 7.273508310317993
    },
    {
      "epoch": 4.852294921875e-06,
      "model_forward_time": 1.7516651153564453,
      "step": 79088
    },
    {
      "epoch": 4.852294921875e-06,
      "step": 79088,
      "training_step_time": 7.605015754699707
    },
    {
      "epoch": 4.853248596191407e-06,
      "model_forward_time": 1.7601287364959717,
      "step": 79089
    },
    {
      "epoch": 4.853248596191407e-06,
      "step": 79089,
      "training_step_time": 7.587615013122559
    },
    {
      "epoch": 4.854202270507812e-06,
      "grad_norm": 0.09718692302703857,
      "learning_rate": 1.1485933984494752e-06,
      "loss": 0.0564,
      "step": 79090
    },
    {
      "epoch": 4.854202270507812e-06,
      "model_forward_time": 1.7750341892242432,
      "step": 79090
    },
    {
      "dynamics_loss_avg": 0.05358566008508205,
      "epoch": 4.854202270507812e-06,
      "step": 79090
    },
    {
      "action_loss_avg": 0.0014965233887778596,
      "epoch": 4.854202270507812e-06,
      "step": 79090
    },
    {
      "epoch": 4.854202270507812e-06,
      "step": 79090,
      "training_step_time": 8.619394063949585
    },
    {
      "epoch": 4.855155944824219e-06,
      "model_forward_time": 2.903040885925293,
      "step": 79091
    },
    {
      "epoch": 4.855155944824219e-06,
      "step": 79091,
      "training_step_time": 9.627072095870972
    },
    {
      "epoch": 4.856109619140625e-06,
      "model_forward_time": 1.7444589138031006,
      "step": 79092
    },
    {
      "epoch": 4.856109619140625e-06,
      "step": 79092,
      "training_step_time": 8.246644020080566
    },
    {
      "epoch": 4.857063293457031e-06,
      "model_forward_time": 1.7572665214538574,
      "step": 79093
    },
    {
      "epoch": 4.857063293457031e-06,
      "step": 79093,
      "training_step_time": 7.536074876785278
    },
    {
      "epoch": 4.858016967773438e-06,
      "model_forward_time": 1.7500584125518799,
      "step": 79094
    },
    {
      "epoch": 4.858016967773438e-06,
      "step": 79094,
      "training_step_time": 7.540268182754517
    },
    {
      "epoch": 4.8589706420898435e-06,
      "model_forward_time": 1.7596049308776855,
      "step": 79095
    },
    {
      "epoch": 4.8589706420898435e-06,
      "step": 79095,
      "training_step_time": 6.7497639656066895
    },
    {
      "epoch": 4.85992431640625e-06,
      "model_forward_time": 1.7639241218566895,
      "step": 79096
    },
    {
      "epoch": 4.85992431640625e-06,
      "step": 79096,
      "training_step_time": 7.619115591049194
    },
    {
      "epoch": 4.860877990722657e-06,
      "model_forward_time": 1.7576758861541748,
      "step": 79097
    },
    {
      "epoch": 4.860877990722657e-06,
      "step": 79097,
      "training_step_time": 8.153520822525024
    },
    {
      "epoch": 4.861831665039062e-06,
      "model_forward_time": 1.7509005069732666,
      "step": 79098
    },
    {
      "epoch": 4.861831665039062e-06,
      "step": 79098,
      "training_step_time": 7.606914758682251
    },
    {
      "epoch": 4.862785339355469e-06,
      "model_forward_time": 1.7845256328582764,
      "step": 79099
    },
    {
      "epoch": 4.862785339355469e-06,
      "step": 79099,
      "training_step_time": 8.358175277709961
    },
    {
      "epoch": 4.863739013671875e-06,
      "grad_norm": 0.12441367655992508,
      "learning_rate": 1.1475391843604e-06,
      "loss": 0.0555,
      "step": 79100
    },
    {
      "epoch": 4.863739013671875e-06,
      "model_forward_time": 2.8413949012756348,
      "step": 79100
    },
    {
      "dynamics_loss_avg": 0.04542007585987449,
      "epoch": 4.863739013671875e-06,
      "step": 79100
    },
    {
      "action_loss_avg": 0.0013462234259350226,
      "epoch": 4.863739013671875e-06,
      "step": 79100
    },
    {
      "epoch": 4.863739013671875e-06,
      "step": 79100,
      "training_step_time": 8.421054363250732
    },
    {
      "epoch": 4.864692687988281e-06,
      "model_forward_time": 1.7832911014556885,
      "step": 79101
    },
    {
      "epoch": 4.864692687988281e-06,
      "step": 79101,
      "training_step_time": 9.681880950927734
    },
    {
      "epoch": 4.865646362304688e-06,
      "model_forward_time": 1.3790202140808105,
      "step": 79102
    },
    {
      "epoch": 4.865646362304688e-06,
      "step": 79102,
      "training_step_time": 7.591829299926758
    },
    {
      "epoch": 4.8666000366210935e-06,
      "model_forward_time": 1.7427136898040771,
      "step": 79103
    },
    {
      "epoch": 4.8666000366210935e-06,
      "step": 79103,
      "training_step_time": 7.472548961639404
    },
    {
      "epoch": 4.8675537109375e-06,
      "model_forward_time": 1.7574646472930908,
      "step": 79104
    },
    {
      "epoch": 4.8675537109375e-06,
      "step": 79104,
      "training_step_time": 7.568761587142944
    },
    {
      "epoch": 4.868507385253907e-06,
      "model_forward_time": 1.7635610103607178,
      "step": 79105
    },
    {
      "epoch": 4.868507385253907e-06,
      "step": 79105,
      "training_step_time": 7.663917779922485
    },
    {
      "epoch": 4.869461059570312e-06,
      "model_forward_time": 1.7519707679748535,
      "step": 79106
    },
    {
      "epoch": 4.869461059570312e-06,
      "step": 79106,
      "training_step_time": 7.628328561782837
    },
    {
      "epoch": 4.870414733886719e-06,
      "model_forward_time": 1.7557423114776611,
      "step": 79107
    },
    {
      "epoch": 4.870414733886719e-06,
      "step": 79107,
      "training_step_time": 8.173336505889893
    },
    {
      "epoch": 4.871368408203125e-06,
      "model_forward_time": 1.7663147449493408,
      "step": 79108
    },
    {
      "epoch": 4.871368408203125e-06,
      "step": 79108,
      "training_step_time": 8.910446166992188
    },
    {
      "epoch": 4.872322082519531e-06,
      "model_forward_time": 2.821150064468384,
      "step": 79109
    },
    {
      "epoch": 4.872322082519531e-06,
      "step": 79109,
      "training_step_time": 7.682591438293457
    },
    {
      "epoch": 4.873275756835938e-06,
      "grad_norm": 0.08752629160881042,
      "learning_rate": 1.1464853915706503e-06,
      "loss": 0.0557,
      "step": 79110
    },
    {
      "epoch": 4.873275756835938e-06,
      "model_forward_time": 1.7735943794250488,
      "step": 79110
    },
    {
      "dynamics_loss_avg": 0.06697317119687796,
      "epoch": 4.873275756835938e-06,
      "step": 79110
    },
    {
      "action_loss_avg": 0.002536811277968809,
      "epoch": 4.873275756835938e-06,
      "step": 79110
    },
    {
      "epoch": 4.873275756835938e-06,
      "step": 79110,
      "training_step_time": 8.88002896308899
    },
    {
      "epoch": 4.8742294311523435e-06,
      "model_forward_time": 1.7483415603637695,
      "step": 79111
    },
    {
      "epoch": 4.8742294311523435e-06,
      "step": 79111,
      "training_step_time": 7.578366279602051
    },
    {
      "epoch": 4.87518310546875e-06,
      "model_forward_time": 1.75518798828125,
      "step": 79112
    },
    {
      "epoch": 4.87518310546875e-06,
      "step": 79112,
      "training_step_time": 7.366138458251953
    },
    {
      "epoch": 4.876136779785157e-06,
      "model_forward_time": 1.747560977935791,
      "step": 79113
    },
    {
      "epoch": 4.876136779785157e-06,
      "step": 79113,
      "training_step_time": 7.5797059535980225
    },
    {
      "epoch": 4.877090454101562e-06,
      "model_forward_time": 1.7492520809173584,
      "step": 79114
    },
    {
      "epoch": 4.877090454101562e-06,
      "step": 79114,
      "training_step_time": 7.681213617324829
    },
    {
      "epoch": 4.878044128417969e-06,
      "model_forward_time": 1.7639334201812744,
      "step": 79115
    },
    {
      "epoch": 4.878044128417969e-06,
      "step": 79115,
      "training_step_time": 8.167640209197998
    },
    {
      "epoch": 4.878997802734375e-06,
      "model_forward_time": 1.757786750793457,
      "step": 79116
    },
    {
      "epoch": 4.878997802734375e-06,
      "step": 79116,
      "training_step_time": 8.575294971466064
    },
    {
      "epoch": 4.879951477050781e-06,
      "model_forward_time": 1.7580945491790771,
      "step": 79117
    },
    {
      "epoch": 4.879951477050781e-06,
      "step": 79117,
      "training_step_time": 7.730697154998779
    },
    {
      "epoch": 4.880905151367188e-06,
      "model_forward_time": 1.7527949810028076,
      "step": 79118
    },
    {
      "epoch": 4.880905151367188e-06,
      "step": 79118,
      "training_step_time": 8.835209608078003
    },
    {
      "epoch": 4.8818588256835935e-06,
      "model_forward_time": 1.7757582664489746,
      "step": 79119
    },
    {
      "epoch": 4.8818588256835935e-06,
      "step": 79119,
      "training_step_time": 9.104573726654053
    },
    {
      "epoch": 4.8828125e-06,
      "grad_norm": 0.08022917062044144,
      "learning_rate": 1.1454320201954628e-06,
      "loss": 0.0561,
      "step": 79120
    },
    {
      "epoch": 4.8828125e-06,
      "model_forward_time": 1.7327232360839844,
      "step": 79120
    },
    {
      "dynamics_loss_avg": 0.06337912231683732,
      "epoch": 4.8828125e-06,
      "step": 79120
    },
    {
      "action_loss_avg": 0.0012677953054662794,
      "epoch": 4.8828125e-06,
      "step": 79120
    },
    {
      "epoch": 4.8828125e-06,
      "step": 79120,
      "training_step_time": 7.841080904006958
    },
    {
      "epoch": 4.883766174316407e-06,
      "model_forward_time": 1.732971429824829,
      "step": 79121
    },
    {
      "epoch": 4.883766174316407e-06,
      "step": 79121,
      "training_step_time": 8.148873805999756
    },
    {
      "epoch": 4.884719848632812e-06,
      "model_forward_time": 1.7521300315856934,
      "step": 79122
    },
    {
      "epoch": 4.884719848632812e-06,
      "step": 79122,
      "training_step_time": 7.589933633804321
    },
    {
      "epoch": 4.885673522949219e-06,
      "model_forward_time": 1.7666871547698975,
      "step": 79123
    },
    {
      "epoch": 4.885673522949219e-06,
      "step": 79123,
      "training_step_time": 7.727831602096558
    },
    {
      "epoch": 4.886627197265625e-06,
      "model_forward_time": 1.7597198486328125,
      "step": 79124
    },
    {
      "epoch": 4.886627197265625e-06,
      "step": 79124,
      "training_step_time": 7.704516649246216
    },
    {
      "epoch": 4.887580871582031e-06,
      "model_forward_time": 1.757544994354248,
      "step": 79125
    },
    {
      "epoch": 4.887580871582031e-06,
      "step": 79125,
      "training_step_time": 7.7050416469573975
    },
    {
      "epoch": 4.888534545898438e-06,
      "model_forward_time": 1.7581403255462646,
      "step": 79126
    },
    {
      "epoch": 4.888534545898438e-06,
      "step": 79126,
      "training_step_time": 7.690169095993042
    },
    {
      "epoch": 4.8894882202148435e-06,
      "model_forward_time": 1.7839806079864502,
      "step": 79127
    },
    {
      "epoch": 4.8894882202148435e-06,
      "step": 79127,
      "training_step_time": 8.250365495681763
    },
    {
      "epoch": 4.89044189453125e-06,
      "model_forward_time": 2.9607856273651123,
      "step": 79128
    },
    {
      "epoch": 4.89044189453125e-06,
      "step": 79128,
      "training_step_time": 9.59295916557312
    },
    {
      "epoch": 4.891395568847656e-06,
      "model_forward_time": 1.7587652206420898,
      "step": 79129
    },
    {
      "epoch": 4.891395568847656e-06,
      "step": 79129,
      "training_step_time": 8.847318887710571
    },
    {
      "epoch": 4.892349243164062e-06,
      "grad_norm": 0.11226209998130798,
      "learning_rate": 1.1443790703500356e-06,
      "loss": 0.0558,
      "step": 79130
    },
    {
      "epoch": 4.892349243164062e-06,
      "model_forward_time": 1.7521874904632568,
      "step": 79130
    },
    {
      "dynamics_loss_avg": 0.05476128477603197,
      "epoch": 4.892349243164062e-06,
      "step": 79130
    },
    {
      "action_loss_avg": 0.001621957973111421,
      "epoch": 4.892349243164062e-06,
      "step": 79130
    },
    {
      "epoch": 4.892349243164062e-06,
      "step": 79130,
      "training_step_time": 7.412278652191162
    },
    {
      "epoch": 4.893302917480469e-06,
      "model_forward_time": 1.7545628547668457,
      "step": 79131
    },
    {
      "epoch": 4.893302917480469e-06,
      "step": 79131,
      "training_step_time": 7.5959999561309814
    },
    {
      "epoch": 4.894256591796875e-06,
      "model_forward_time": 1.7443804740905762,
      "step": 79132
    },
    {
      "epoch": 4.894256591796875e-06,
      "step": 79132,
      "training_step_time": 8.43159532546997
    },
    {
      "epoch": 4.895210266113281e-06,
      "model_forward_time": 1.7541155815124512,
      "step": 79133
    },
    {
      "epoch": 4.895210266113281e-06,
      "step": 79133,
      "training_step_time": 7.716005325317383
    },
    {
      "epoch": 4.896163940429688e-06,
      "model_forward_time": 1.75240159034729,
      "step": 79134
    },
    {
      "epoch": 4.896163940429688e-06,
      "step": 79134,
      "training_step_time": 7.667670726776123
    },
    {
      "epoch": 4.8971176147460935e-06,
      "model_forward_time": 1.7575922012329102,
      "step": 79135
    },
    {
      "epoch": 4.8971176147460935e-06,
      "step": 79135,
      "training_step_time": 7.719871520996094
    },
    {
      "epoch": 4.8980712890625e-06,
      "model_forward_time": 1.7673850059509277,
      "step": 79136
    },
    {
      "epoch": 4.8980712890625e-06,
      "step": 79136,
      "training_step_time": 9.42728304862976
    },
    {
      "epoch": 4.899024963378906e-06,
      "model_forward_time": 2.8657214641571045,
      "step": 79137
    },
    {
      "epoch": 4.899024963378906e-06,
      "step": 79137,
      "training_step_time": 8.807376384735107
    },
    {
      "epoch": 4.899978637695312e-06,
      "model_forward_time": 1.7652668952941895,
      "step": 79138
    },
    {
      "epoch": 4.899978637695312e-06,
      "step": 79138,
      "training_step_time": 8.9033522605896
    },
    {
      "epoch": 4.900932312011719e-06,
      "model_forward_time": 1.7556519508361816,
      "step": 79139
    },
    {
      "epoch": 4.900932312011719e-06,
      "step": 79139,
      "training_step_time": 7.693424463272095
    },
    {
      "epoch": 4.901885986328125e-06,
      "grad_norm": 0.09163809567689896,
      "learning_rate": 1.1433265421495154e-06,
      "loss": 0.0549,
      "step": 79140
    },
    {
      "epoch": 4.901885986328125e-06,
      "model_forward_time": 1.7539153099060059,
      "step": 79140
    },
    {
      "dynamics_loss_avg": 0.05447404142469168,
      "epoch": 4.901885986328125e-06,
      "step": 79140
    },
    {
      "action_loss_avg": 0.001188416621880606,
      "epoch": 4.901885986328125e-06,
      "step": 79140
    },
    {
      "epoch": 4.901885986328125e-06,
      "step": 79140,
      "training_step_time": 7.669677257537842
    },
    {
      "epoch": 4.902839660644531e-06,
      "model_forward_time": 1.7532682418823242,
      "step": 79141
    },
    {
      "epoch": 4.902839660644531e-06,
      "step": 79141,
      "training_step_time": 7.377699613571167
    },
    {
      "epoch": 4.903793334960938e-06,
      "model_forward_time": 1.7531406879425049,
      "step": 79142
    },
    {
      "epoch": 4.903793334960938e-06,
      "step": 79142,
      "training_step_time": 7.75574517250061
    },
    {
      "epoch": 4.9047470092773436e-06,
      "model_forward_time": 1.7551336288452148,
      "step": 79143
    },
    {
      "epoch": 4.9047470092773436e-06,
      "step": 79143,
      "training_step_time": 8.58022141456604
    },
    {
      "epoch": 4.90570068359375e-06,
      "model_forward_time": 1.7538816928863525,
      "step": 79144
    },
    {
      "epoch": 4.90570068359375e-06,
      "step": 79144,
      "training_step_time": 8.861498594284058
    },
    {
      "epoch": 4.906654357910156e-06,
      "model_forward_time": 1.7886934280395508,
      "step": 79145
    },
    {
      "epoch": 4.906654357910156e-06,
      "step": 79145,
      "training_step_time": 9.595278978347778
    },
    {
      "epoch": 4.907608032226562e-06,
      "model_forward_time": 2.839179039001465,
      "step": 79146
    },
    {
      "epoch": 4.907608032226562e-06,
      "step": 79146,
      "training_step_time": 7.899675130844116
    },
    {
      "epoch": 4.908561706542969e-06,
      "model_forward_time": 1.7743818759918213,
      "step": 79147
    },
    {
      "epoch": 4.908561706542969e-06,
      "step": 79147,
      "training_step_time": 7.895164728164673
    },
    {
      "epoch": 4.909515380859375e-06,
      "model_forward_time": 1.7491192817687988,
      "step": 79148
    },
    {
      "epoch": 4.909515380859375e-06,
      "step": 79148,
      "training_step_time": 7.677782773971558
    },
    {
      "epoch": 4.910469055175781e-06,
      "model_forward_time": 1.7460963726043701,
      "step": 79149
    },
    {
      "epoch": 4.910469055175781e-06,
      "step": 79149,
      "training_step_time": 7.694365978240967
    },
    {
      "epoch": 4.911422729492188e-06,
      "grad_norm": 0.09806432574987411,
      "learning_rate": 1.1422744357090066e-06,
      "loss": 0.0551,
      "step": 79150
    },
    {
      "epoch": 4.911422729492188e-06,
      "model_forward_time": 1.38877534866333,
      "step": 79150
    },
    {
      "dynamics_loss_avg": 0.04418032914400101,
      "epoch": 4.911422729492188e-06,
      "step": 79150
    },
    {
      "action_loss_avg": 0.0007925721234641969,
      "epoch": 4.911422729492188e-06,
      "step": 79150
    },
    {
      "epoch": 4.911422729492188e-06,
      "step": 79150,
      "training_step_time": 7.690207004547119
    },
    {
      "epoch": 4.9123764038085936e-06,
      "model_forward_time": 1.7400360107421875,
      "step": 79151
    },
    {
      "epoch": 4.9123764038085936e-06,
      "step": 79151,
      "training_step_time": 7.741094350814819
    },
    {
      "epoch": 4.913330078125e-06,
      "model_forward_time": 1.7593929767608643,
      "step": 79152
    },
    {
      "epoch": 4.913330078125e-06,
      "step": 79152,
      "training_step_time": 8.656599283218384
    },
    {
      "epoch": 4.914283752441406e-06,
      "model_forward_time": 1.7686052322387695,
      "step": 79153
    },
    {
      "epoch": 4.914283752441406e-06,
      "step": 79153,
      "training_step_time": 7.76828145980835
    },
    {
      "epoch": 4.915237426757812e-06,
      "model_forward_time": 1.7717392444610596,
      "step": 79154
    },
    {
      "epoch": 4.915237426757812e-06,
      "step": 79154,
      "training_step_time": 8.409169673919678
    },
    {
      "epoch": 4.916191101074219e-06,
      "model_forward_time": 2.8033111095428467,
      "step": 79155
    },
    {
      "epoch": 4.916191101074219e-06,
      "step": 79155,
      "training_step_time": 8.487660884857178
    },
    {
      "epoch": 4.917144775390625e-06,
      "model_forward_time": 1.7829320430755615,
      "step": 79156
    },
    {
      "epoch": 4.917144775390625e-06,
      "step": 79156,
      "training_step_time": 8.39603567123413
    },
    {
      "epoch": 4.918098449707031e-06,
      "model_forward_time": 1.7464334964752197,
      "step": 79157
    },
    {
      "epoch": 4.918098449707031e-06,
      "step": 79157,
      "training_step_time": 7.86523175239563
    },
    {
      "epoch": 4.919052124023438e-06,
      "model_forward_time": 1.7540793418884277,
      "step": 79158
    },
    {
      "epoch": 4.919052124023438e-06,
      "step": 79158,
      "training_step_time": 7.549804449081421
    },
    {
      "epoch": 4.9200057983398436e-06,
      "model_forward_time": 1.7528600692749023,
      "step": 79159
    },
    {
      "epoch": 4.9200057983398436e-06,
      "step": 79159,
      "training_step_time": 7.631201982498169
    },
    {
      "epoch": 4.92095947265625e-06,
      "grad_norm": 0.11484555900096893,
      "learning_rate": 1.1412227511435647e-06,
      "loss": 0.0569,
      "step": 79160
    },
    {
      "epoch": 4.92095947265625e-06,
      "model_forward_time": 1.7505922317504883,
      "step": 79160
    },
    {
      "dynamics_loss_avg": 0.06164322514086962,
      "epoch": 4.92095947265625e-06,
      "step": 79160
    },
    {
      "action_loss_avg": 0.0011709468904882669,
      "epoch": 4.92095947265625e-06,
      "step": 79160
    },
    {
      "epoch": 4.92095947265625e-06,
      "step": 79160,
      "training_step_time": 7.7606987953186035
    },
    {
      "epoch": 4.921913146972656e-06,
      "model_forward_time": 1.3871192932128906,
      "step": 79161
    },
    {
      "epoch": 4.921913146972656e-06,
      "step": 79161,
      "training_step_time": 9.651892185211182
    },
    {
      "epoch": 4.9228668212890624e-06,
      "model_forward_time": 1.7522695064544678,
      "step": 79162
    },
    {
      "epoch": 4.9228668212890624e-06,
      "step": 79162,
      "training_step_time": 7.787399768829346
    },
    {
      "epoch": 4.923820495605469e-06,
      "model_forward_time": 1.7601492404937744,
      "step": 79163
    },
    {
      "epoch": 4.923820495605469e-06,
      "step": 79163,
      "training_step_time": 8.297487735748291
    },
    {
      "epoch": 4.924774169921875e-06,
      "model_forward_time": 1.7511990070343018,
      "step": 79164
    },
    {
      "epoch": 4.924774169921875e-06,
      "step": 79164,
      "training_step_time": 9.632142305374146
    },
    {
      "epoch": 4.925727844238281e-06,
      "model_forward_time": 2.9383535385131836,
      "step": 79165
    },
    {
      "epoch": 4.925727844238281e-06,
      "step": 79165,
      "training_step_time": 7.851395130157471
    },
    {
      "epoch": 4.926681518554688e-06,
      "model_forward_time": 1.7442336082458496,
      "step": 79166
    },
    {
      "epoch": 4.926681518554688e-06,
      "step": 79166,
      "training_step_time": 8.835941076278687
    },
    {
      "epoch": 4.9276351928710936e-06,
      "model_forward_time": 1.7665143013000488,
      "step": 79167
    },
    {
      "epoch": 4.9276351928710936e-06,
      "step": 79167,
      "training_step_time": 7.489929437637329
    },
    {
      "epoch": 4.9285888671875e-06,
      "model_forward_time": 1.7563252449035645,
      "step": 79168
    },
    {
      "epoch": 4.9285888671875e-06,
      "step": 79168,
      "training_step_time": 7.526159048080444
    },
    {
      "epoch": 4.929542541503906e-06,
      "model_forward_time": 1.7452683448791504,
      "step": 79169
    },
    {
      "epoch": 4.929542541503906e-06,
      "step": 79169,
      "training_step_time": 7.485451936721802
    },
    {
      "epoch": 4.9304962158203124e-06,
      "grad_norm": 0.08151701837778091,
      "learning_rate": 1.1401714885682024e-06,
      "loss": 0.0561,
      "step": 79170
    },
    {
      "epoch": 4.9304962158203124e-06,
      "model_forward_time": 1.7519495487213135,
      "step": 79170
    },
    {
      "dynamics_loss_avg": 0.05242730155587196,
      "epoch": 4.9304962158203124e-06,
      "step": 79170
    },
    {
      "action_loss_avg": 0.0018218761193566023,
      "epoch": 4.9304962158203124e-06,
      "step": 79170
    },
    {
      "epoch": 4.9304962158203124e-06,
      "step": 79170,
      "training_step_time": 7.682593822479248
    },
    {
      "epoch": 4.931449890136719e-06,
      "model_forward_time": 1.7560346126556396,
      "step": 79171
    },
    {
      "epoch": 4.931449890136719e-06,
      "step": 79171,
      "training_step_time": 9.443200588226318
    },
    {
      "epoch": 4.932403564453125e-06,
      "model_forward_time": 1.751176357269287,
      "step": 79172
    },
    {
      "epoch": 4.932403564453125e-06,
      "step": 79172,
      "training_step_time": 8.491828203201294
    },
    {
      "epoch": 4.933357238769531e-06,
      "model_forward_time": 1.7780413627624512,
      "step": 79173
    },
    {
      "epoch": 4.933357238769531e-06,
      "step": 79173,
      "training_step_time": 10.127274513244629
    },
    {
      "epoch": 4.934310913085938e-06,
      "model_forward_time": 2.877631664276123,
      "step": 79174
    },
    {
      "epoch": 4.934310913085938e-06,
      "step": 79174,
      "training_step_time": 7.729474306106567
    },
    {
      "epoch": 4.935264587402344e-06,
      "model_forward_time": 1.7800147533416748,
      "step": 79175
    },
    {
      "epoch": 4.935264587402344e-06,
      "step": 79175,
      "training_step_time": 9.18557858467102
    },
    {
      "epoch": 4.93621826171875e-06,
      "model_forward_time": 1.7568845748901367,
      "step": 79176
    },
    {
      "epoch": 4.93621826171875e-06,
      "step": 79176,
      "training_step_time": 7.456788778305054
    },
    {
      "epoch": 4.937171936035156e-06,
      "model_forward_time": 1.756622552871704,
      "step": 79177
    },
    {
      "epoch": 4.937171936035156e-06,
      "step": 79177,
      "training_step_time": 7.570044040679932
    },
    {
      "epoch": 4.9381256103515624e-06,
      "model_forward_time": 1.7596211433410645,
      "step": 79178
    },
    {
      "epoch": 4.9381256103515624e-06,
      "step": 79178,
      "training_step_time": 7.49036979675293
    },
    {
      "epoch": 4.939079284667969e-06,
      "model_forward_time": 1.755338191986084,
      "step": 79179
    },
    {
      "epoch": 4.939079284667969e-06,
      "step": 79179,
      "training_step_time": 7.583991527557373
    },
    {
      "epoch": 4.940032958984375e-06,
      "grad_norm": 0.4896470308303833,
      "learning_rate": 1.1391206480978822e-06,
      "loss": 0.0554,
      "step": 79180
    },
    {
      "epoch": 4.940032958984375e-06,
      "model_forward_time": 1.7448854446411133,
      "step": 79180
    },
    {
      "dynamics_loss_avg": 0.054448891151696446,
      "epoch": 4.940032958984375e-06,
      "step": 79180
    },
    {
      "action_loss_avg": 0.0011525086738402024,
      "epoch": 4.940032958984375e-06,
      "step": 79180
    },
    {
      "epoch": 4.940032958984375e-06,
      "step": 79180,
      "training_step_time": 7.739419937133789
    },
    {
      "epoch": 4.940986633300781e-06,
      "model_forward_time": 1.7590949535369873,
      "step": 79181
    },
    {
      "epoch": 4.940986633300781e-06,
      "step": 79181,
      "training_step_time": 7.72566556930542
    },
    {
      "epoch": 4.941940307617188e-06,
      "model_forward_time": 1.773608922958374,
      "step": 79182
    },
    {
      "epoch": 4.941940307617188e-06,
      "step": 79182,
      "training_step_time": 9.074743032455444
    },
    {
      "epoch": 4.942893981933594e-06,
      "model_forward_time": 2.8335165977478027,
      "step": 79183
    },
    {
      "epoch": 4.942893981933594e-06,
      "step": 79183,
      "training_step_time": 8.80164361000061
    },
    {
      "epoch": 4.94384765625e-06,
      "model_forward_time": 1.7849113941192627,
      "step": 79184
    },
    {
      "epoch": 4.94384765625e-06,
      "step": 79184,
      "training_step_time": 7.900924205780029
    },
    {
      "epoch": 4.944801330566406e-06,
      "model_forward_time": 1.758211612701416,
      "step": 79185
    },
    {
      "epoch": 4.944801330566406e-06,
      "step": 79185,
      "training_step_time": 7.689748048782349
    },
    {
      "epoch": 4.9457550048828124e-06,
      "model_forward_time": 1.7457280158996582,
      "step": 79186
    },
    {
      "epoch": 4.9457550048828124e-06,
      "step": 79186,
      "training_step_time": 7.6205267906188965
    },
    {
      "epoch": 4.946708679199219e-06,
      "model_forward_time": 1.7537124156951904,
      "step": 79187
    },
    {
      "epoch": 4.946708679199219e-06,
      "step": 79187,
      "training_step_time": 7.547208547592163
    },
    {
      "epoch": 4.947662353515625e-06,
      "model_forward_time": 1.7522845268249512,
      "step": 79188
    },
    {
      "epoch": 4.947662353515625e-06,
      "step": 79188,
      "training_step_time": 8.65019965171814
    },
    {
      "epoch": 4.948616027832031e-06,
      "model_forward_time": 1.7428300380706787,
      "step": 79189
    },
    {
      "epoch": 4.948616027832031e-06,
      "step": 79189,
      "training_step_time": 8.661208391189575
    },
    {
      "epoch": 4.949569702148438e-06,
      "grad_norm": 0.07233237475156784,
      "learning_rate": 1.1380702298475226e-06,
      "loss": 0.0553,
      "step": 79190
    },
    {
      "epoch": 4.949569702148438e-06,
      "model_forward_time": 1.749523401260376,
      "step": 79190
    },
    {
      "dynamics_loss_avg": 0.05356861967593431,
      "epoch": 4.949569702148438e-06,
      "step": 79190
    },
    {
      "action_loss_avg": 0.0012996397505048662,
      "epoch": 4.949569702148438e-06,
      "step": 79190
    },
    {
      "epoch": 4.949569702148438e-06,
      "step": 79190,
      "training_step_time": 8.648826122283936
    },
    {
      "epoch": 4.950523376464844e-06,
      "model_forward_time": 1.7772166728973389,
      "step": 79191
    },
    {
      "epoch": 4.950523376464844e-06,
      "step": 79191,
      "training_step_time": 9.406555414199829
    },
    {
      "epoch": 4.95147705078125e-06,
      "model_forward_time": 2.810947895050049,
      "step": 79192
    },
    {
      "epoch": 4.95147705078125e-06,
      "step": 79192,
      "training_step_time": 7.716940879821777
    },
    {
      "epoch": 4.952430725097656e-06,
      "model_forward_time": 1.7791669368743896,
      "step": 79193
    },
    {
      "epoch": 4.952430725097656e-06,
      "step": 79193,
      "training_step_time": 9.713406085968018
    },
    {
      "epoch": 4.9533843994140625e-06,
      "model_forward_time": 1.7549471855163574,
      "step": 79194
    },
    {
      "epoch": 4.9533843994140625e-06,
      "step": 79194,
      "training_step_time": 7.656247138977051
    },
    {
      "epoch": 4.954338073730469e-06,
      "model_forward_time": 1.7506225109100342,
      "step": 79195
    },
    {
      "epoch": 4.954338073730469e-06,
      "step": 79195,
      "training_step_time": 7.494431972503662
    },
    {
      "epoch": 4.955291748046875e-06,
      "model_forward_time": 1.7602458000183105,
      "step": 79196
    },
    {
      "epoch": 4.955291748046875e-06,
      "step": 79196,
      "training_step_time": 7.553018093109131
    },
    {
      "epoch": 4.956245422363281e-06,
      "model_forward_time": 1.7579944133758545,
      "step": 79197
    },
    {
      "epoch": 4.956245422363281e-06,
      "step": 79197,
      "training_step_time": 7.704352140426636
    },
    {
      "epoch": 4.957199096679688e-06,
      "model_forward_time": 1.7496869564056396,
      "step": 79198
    },
    {
      "epoch": 4.957199096679688e-06,
      "step": 79198,
      "training_step_time": 9.564735651016235
    },
    {
      "epoch": 4.958152770996094e-06,
      "model_forward_time": 1.7463359832763672,
      "step": 79199
    },
    {
      "epoch": 4.958152770996094e-06,
      "step": 79199,
      "training_step_time": 9.431015491485596
    },
    {
      "epoch": 4.9591064453125e-06,
      "grad_norm": 0.10807795077562332,
      "learning_rate": 1.1370202339319985e-06,
      "loss": 0.0562,
      "step": 79200
    },
    {
      "epoch": 4.9591064453125e-06,
      "model_forward_time": 1.7506537437438965,
      "step": 79200
    },
    {
      "dynamics_loss_avg": 0.0517869345843792,
      "epoch": 4.9591064453125e-06,
      "step": 79200
    },
    {
      "action_loss_avg": 0.0009594556642696262,
      "epoch": 4.9591064453125e-06,
      "step": 79200
    },
    {
      "epoch": 4.9591064453125e-06,
      "step": 79200,
      "training_step_time": 8.302772045135498
    },
    {
      "epoch": 4.960060119628906e-06,
      "model_forward_time": 2.8226406574249268,
      "step": 79201
    },
    {
      "epoch": 4.960060119628906e-06,
      "step": 79201,
      "training_step_time": 8.699609994888306
    },
    {
      "epoch": 4.9610137939453125e-06,
      "model_forward_time": 1.7648565769195557,
      "step": 79202
    },
    {
      "epoch": 4.9610137939453125e-06,
      "step": 79202,
      "training_step_time": 9.11678695678711
    },
    {
      "epoch": 4.961967468261719e-06,
      "model_forward_time": 1.755624532699585,
      "step": 79203
    },
    {
      "epoch": 4.961967468261719e-06,
      "step": 79203,
      "training_step_time": 7.821683883666992
    },
    {
      "epoch": 4.962921142578125e-06,
      "model_forward_time": 1.7501931190490723,
      "step": 79204
    },
    {
      "epoch": 4.962921142578125e-06,
      "step": 79204,
      "training_step_time": 8.258671760559082
    },
    {
      "epoch": 4.963874816894531e-06,
      "model_forward_time": 1.770061731338501,
      "step": 79205
    },
    {
      "epoch": 4.963874816894531e-06,
      "step": 79205,
      "training_step_time": 7.477845907211304
    },
    {
      "epoch": 4.964828491210938e-06,
      "model_forward_time": 1.0579140186309814,
      "step": 79206
    },
    {
      "epoch": 4.964828491210938e-06,
      "step": 79206,
      "training_step_time": 7.5888776779174805
    },
    {
      "epoch": 4.965782165527344e-06,
      "model_forward_time": 1.7541053295135498,
      "step": 79207
    },
    {
      "epoch": 4.965782165527344e-06,
      "step": 79207,
      "training_step_time": 7.7121663093566895
    },
    {
      "epoch": 4.96673583984375e-06,
      "model_forward_time": 1.7514476776123047,
      "step": 79208
    },
    {
      "epoch": 4.96673583984375e-06,
      "step": 79208,
      "training_step_time": 8.458065271377563
    },
    {
      "epoch": 4.967689514160156e-06,
      "model_forward_time": 1.753838062286377,
      "step": 79209
    },
    {
      "epoch": 4.967689514160156e-06,
      "step": 79209,
      "training_step_time": 9.134470224380493
    },
    {
      "epoch": 4.9686431884765625e-06,
      "grad_norm": 0.10243552178144455,
      "learning_rate": 1.1359706604661319e-06,
      "loss": 0.0577,
      "step": 79210
    },
    {
      "epoch": 4.9686431884765625e-06,
      "model_forward_time": 1.763638973236084,
      "step": 79210
    },
    {
      "dynamics_loss_avg": 0.040665939636528495,
      "epoch": 4.9686431884765625e-06,
      "step": 79210
    },
    {
      "action_loss_avg": 0.0008496811700752005,
      "epoch": 4.9686431884765625e-06,
      "step": 79210
    },
    {
      "epoch": 4.9686431884765625e-06,
      "step": 79210,
      "training_step_time": 8.173284769058228
    },
    {
      "epoch": 4.969596862792969e-06,
      "model_forward_time": 2.867345094680786,
      "step": 79211
    },
    {
      "epoch": 4.969596862792969e-06,
      "step": 79211,
      "training_step_time": 8.679056882858276
    },
    {
      "epoch": 4.970550537109375e-06,
      "model_forward_time": 1.783149003982544,
      "step": 79212
    },
    {
      "epoch": 4.970550537109375e-06,
      "step": 79212,
      "training_step_time": 8.216239213943481
    },
    {
      "epoch": 4.971504211425781e-06,
      "model_forward_time": 1.7240703105926514,
      "step": 79213
    },
    {
      "epoch": 4.971504211425781e-06,
      "step": 79213,
      "training_step_time": 7.396883726119995
    },
    {
      "epoch": 4.972457885742188e-06,
      "model_forward_time": 1.7504701614379883,
      "step": 79214
    },
    {
      "epoch": 4.972457885742188e-06,
      "step": 79214,
      "training_step_time": 7.512251377105713
    },
    {
      "epoch": 4.973411560058594e-06,
      "model_forward_time": 1.7371630668640137,
      "step": 79215
    },
    {
      "epoch": 4.973411560058594e-06,
      "step": 79215,
      "training_step_time": 7.553212881088257
    },
    {
      "epoch": 4.974365234375e-06,
      "model_forward_time": 1.7454249858856201,
      "step": 79216
    },
    {
      "epoch": 4.974365234375e-06,
      "step": 79216,
      "training_step_time": 7.661732912063599
    },
    {
      "epoch": 4.975318908691406e-06,
      "model_forward_time": 1.7516074180603027,
      "step": 79217
    },
    {
      "epoch": 4.975318908691406e-06,
      "step": 79217,
      "training_step_time": 7.597527265548706
    },
    {
      "epoch": 4.9762725830078125e-06,
      "model_forward_time": 1.744995355606079,
      "step": 79218
    },
    {
      "epoch": 4.9762725830078125e-06,
      "step": 79218,
      "training_step_time": 8.360848665237427
    },
    {
      "epoch": 4.977226257324219e-06,
      "model_forward_time": 1.7681126594543457,
      "step": 79219
    },
    {
      "epoch": 4.977226257324219e-06,
      "step": 79219,
      "training_step_time": 8.164947271347046
    },
    {
      "epoch": 4.978179931640625e-06,
      "grad_norm": 0.09718304127454758,
      "learning_rate": 1.1349215095647043e-06,
      "loss": 0.0575,
      "step": 79220
    },
    {
      "epoch": 4.978179931640625e-06,
      "model_forward_time": 2.8172731399536133,
      "step": 79220
    },
    {
      "dynamics_loss_avg": 0.05152628016658127,
      "epoch": 4.978179931640625e-06,
      "step": 79220
    },
    {
      "action_loss_avg": 0.0011612248461460694,
      "epoch": 4.978179931640625e-06,
      "step": 79220
    },
    {
      "epoch": 4.978179931640625e-06,
      "step": 79220,
      "training_step_time": 7.686281204223633
    },
    {
      "epoch": 4.979133605957031e-06,
      "model_forward_time": 1.7800531387329102,
      "step": 79221
    },
    {
      "epoch": 4.979133605957031e-06,
      "step": 79221,
      "training_step_time": 7.676207065582275
    },
    {
      "epoch": 4.980087280273438e-06,
      "model_forward_time": 1.731858253479004,
      "step": 79222
    },
    {
      "epoch": 4.980087280273438e-06,
      "step": 79222,
      "training_step_time": 7.69768214225769
    },
    {
      "epoch": 4.981040954589844e-06,
      "model_forward_time": 1.7447483539581299,
      "step": 79223
    },
    {
      "epoch": 4.981040954589844e-06,
      "step": 79223,
      "training_step_time": 8.367516040802002
    },
    {
      "epoch": 4.98199462890625e-06,
      "model_forward_time": 1.7518773078918457,
      "step": 79224
    },
    {
      "epoch": 4.98199462890625e-06,
      "step": 79224,
      "training_step_time": 8.099386215209961
    },
    {
      "epoch": 4.982948303222656e-06,
      "model_forward_time": 1.7508323192596436,
      "step": 79225
    },
    {
      "epoch": 4.982948303222656e-06,
      "step": 79225,
      "training_step_time": 8.65768575668335
    },
    {
      "epoch": 4.9839019775390625e-06,
      "model_forward_time": 1.7623004913330078,
      "step": 79226
    },
    {
      "epoch": 4.9839019775390625e-06,
      "step": 79226,
      "training_step_time": 8.402388572692871
    },
    {
      "epoch": 4.984855651855469e-06,
      "model_forward_time": 1.743788480758667,
      "step": 79227
    },
    {
      "epoch": 4.984855651855469e-06,
      "step": 79227,
      "training_step_time": 8.551116704940796
    },
    {
      "epoch": 4.985809326171875e-06,
      "model_forward_time": 1.7683501243591309,
      "step": 79228
    },
    {
      "epoch": 4.985809326171875e-06,
      "step": 79228,
      "training_step_time": 7.62223482131958
    },
    {
      "epoch": 4.986763000488281e-06,
      "model_forward_time": 2.821833372116089,
      "step": 79229
    },
    {
      "epoch": 4.986763000488281e-06,
      "step": 79229,
      "training_step_time": 7.773510694503784
    },
    {
      "epoch": 4.987716674804688e-06,
      "grad_norm": 0.08772369474172592,
      "learning_rate": 1.1338727813424494e-06,
      "loss": 0.0572,
      "step": 79230
    },
    {
      "epoch": 4.987716674804688e-06,
      "model_forward_time": 1.7518000602722168,
      "step": 79230
    },
    {
      "dynamics_loss_avg": 0.05232951212674379,
      "epoch": 4.987716674804688e-06,
      "step": 79230
    },
    {
      "action_loss_avg": 0.001677965489216149,
      "epoch": 4.987716674804688e-06,
      "step": 79230
    },
    {
      "epoch": 4.987716674804688e-06,
      "step": 79230,
      "training_step_time": 9.110458850860596
    },
    {
      "epoch": 4.988670349121094e-06,
      "model_forward_time": 1.743952751159668,
      "step": 79231
    },
    {
      "epoch": 4.988670349121094e-06,
      "step": 79231,
      "training_step_time": 7.692628383636475
    },
    {
      "epoch": 4.9896240234375e-06,
      "model_forward_time": 1.7467732429504395,
      "step": 79232
    },
    {
      "epoch": 4.9896240234375e-06,
      "step": 79232,
      "training_step_time": 7.605459213256836
    },
    {
      "epoch": 4.990577697753906e-06,
      "model_forward_time": 1.7367329597473145,
      "step": 79233
    },
    {
      "epoch": 4.990577697753906e-06,
      "step": 79233,
      "training_step_time": 8.456613540649414
    },
    {
      "epoch": 4.9915313720703125e-06,
      "model_forward_time": 1.7474250793457031,
      "step": 79234
    },
    {
      "epoch": 4.9915313720703125e-06,
      "step": 79234,
      "training_step_time": 8.657752990722656
    },
    {
      "epoch": 4.992485046386719e-06,
      "model_forward_time": 1.750565528869629,
      "step": 79235
    },
    {
      "epoch": 4.992485046386719e-06,
      "step": 79235,
      "training_step_time": 7.6366260051727295
    },
    {
      "epoch": 4.993438720703125e-06,
      "model_forward_time": 1.74322509765625,
      "step": 79236
    },
    {
      "epoch": 4.993438720703125e-06,
      "step": 79236,
      "training_step_time": 7.688940525054932
    },
    {
      "epoch": 4.994392395019531e-06,
      "model_forward_time": 1.7627971172332764,
      "step": 79237
    },
    {
      "epoch": 4.994392395019531e-06,
      "step": 79237,
      "training_step_time": 9.200413703918457
    },
    {
      "epoch": 4.995346069335938e-06,
      "model_forward_time": 1.735013484954834,
      "step": 79238
    },
    {
      "epoch": 4.995346069335938e-06,
      "step": 79238,
      "training_step_time": 9.385625839233398
    },
    {
      "epoch": 4.996299743652344e-06,
      "model_forward_time": 1.743438482284546,
      "step": 79239
    },
    {
      "epoch": 4.996299743652344e-06,
      "step": 79239,
      "training_step_time": 8.204090595245361
    },
    {
      "epoch": 4.99725341796875e-06,
      "grad_norm": 0.08957196772098541,
      "learning_rate": 1.132824475914056e-06,
      "loss": 0.0576,
      "step": 79240
    },
    {
      "epoch": 4.99725341796875e-06,
      "model_forward_time": 1.7413272857666016,
      "step": 79240
    },
    {
      "dynamics_loss_avg": 0.0536583399400115,
      "epoch": 4.99725341796875e-06,
      "step": 79240
    },
    {
      "action_loss_avg": 0.0010586124320980162,
      "epoch": 4.99725341796875e-06,
      "step": 79240
    },
    {
      "epoch": 4.99725341796875e-06,
      "step": 79240,
      "training_step_time": 7.786401748657227
    },
    {
      "epoch": 4.998207092285156e-06,
      "model_forward_time": 1.7585110664367676,
      "step": 79241
    },
    {
      "epoch": 4.998207092285156e-06,
      "step": 79241,
      "training_step_time": 8.284581422805786
    },
    {
      "epoch": 4.9991607666015625e-06,
      "model_forward_time": 1.749122142791748,
      "step": 79242
    },
    {
      "epoch": 4.9991607666015625e-06,
      "step": 79242,
      "training_step_time": 7.476034164428711
    },
    {
      "epoch": 5.000114440917969e-06,
      "model_forward_time": 1.7560229301452637,
      "step": 79243
    },
    {
      "epoch": 5.000114440917969e-06,
      "step": 79243,
      "training_step_time": 8.6731595993042
    },
    {
      "epoch": 5.001068115234375e-06,
      "model_forward_time": 1.7633953094482422,
      "step": 79244
    },
    {
      "epoch": 5.001068115234375e-06,
      "step": 79244,
      "training_step_time": 8.479261875152588
    },
    {
      "epoch": 5.002021789550781e-06,
      "model_forward_time": 1.7555701732635498,
      "step": 79245
    },
    {
      "epoch": 5.002021789550781e-06,
      "step": 79245,
      "training_step_time": 7.689646005630493
    },
    {
      "epoch": 5.002975463867188e-06,
      "model_forward_time": 1.7556531429290771,
      "step": 79246
    },
    {
      "epoch": 5.002975463867188e-06,
      "step": 79246,
      "training_step_time": 7.702772378921509
    },
    {
      "epoch": 5.003929138183594e-06,
      "model_forward_time": 1.760298490524292,
      "step": 79247
    },
    {
      "epoch": 5.003929138183594e-06,
      "step": 79247,
      "training_step_time": 8.470819234848022
    },
    {
      "epoch": 5.0048828125e-06,
      "model_forward_time": 2.8326921463012695,
      "step": 79248
    },
    {
      "epoch": 5.0048828125e-06,
      "step": 79248,
      "training_step_time": 8.207978963851929
    },
    {
      "epoch": 5.005836486816406e-06,
      "model_forward_time": 1.7693238258361816,
      "step": 79249
    },
    {
      "epoch": 5.005836486816406e-06,
      "step": 79249,
      "training_step_time": 7.780681371688843
    },
    {
      "epoch": 5.0067901611328125e-06,
      "grad_norm": 0.0932002067565918,
      "learning_rate": 1.1317765933941621e-06,
      "loss": 0.0575,
      "step": 79250
    },
    {
      "epoch": 5.0067901611328125e-06,
      "model_forward_time": 1.76735520362854,
      "step": 79250
    },
    {
      "dynamics_loss_avg": 0.05983761418610811,
      "epoch": 5.0067901611328125e-06,
      "step": 79250
    },
    {
      "action_loss_avg": 0.0012854819011408835,
      "epoch": 5.0067901611328125e-06,
      "step": 79250
    },
    {
      "epoch": 5.0067901611328125e-06,
      "step": 79250,
      "training_step_time": 7.653151273727417
    },
    {
      "epoch": 5.007743835449219e-06,
      "model_forward_time": 1.7602510452270508,
      "step": 79251
    },
    {
      "epoch": 5.007743835449219e-06,
      "step": 79251,
      "training_step_time": 7.570094108581543
    },
    {
      "epoch": 5.008697509765625e-06,
      "model_forward_time": 1.7621524333953857,
      "step": 79252
    },
    {
      "epoch": 5.008697509765625e-06,
      "step": 79252,
      "training_step_time": 7.654745817184448
    },
    {
      "epoch": 5.009651184082031e-06,
      "model_forward_time": 1.7612042427062988,
      "step": 79253
    },
    {
      "epoch": 5.009651184082031e-06,
      "step": 79253,
      "training_step_time": 7.763999938964844
    },
    {
      "epoch": 5.010604858398437e-06,
      "model_forward_time": 1.7561018466949463,
      "step": 79254
    },
    {
      "epoch": 5.010604858398437e-06,
      "step": 79254,
      "training_step_time": 8.575645446777344
    },
    {
      "epoch": 5.011558532714844e-06,
      "model_forward_time": 1.7752339839935303,
      "step": 79255
    },
    {
      "epoch": 5.011558532714844e-06,
      "step": 79255,
      "training_step_time": 7.794721364974976
    },
    {
      "epoch": 5.01251220703125e-06,
      "model_forward_time": 1.7739191055297852,
      "step": 79256
    },
    {
      "epoch": 5.01251220703125e-06,
      "step": 79256,
      "training_step_time": 7.757844686508179
    },
    {
      "epoch": 5.013465881347656e-06,
      "model_forward_time": 2.8616514205932617,
      "step": 79257
    },
    {
      "epoch": 5.013465881347656e-06,
      "step": 79257,
      "training_step_time": 8.475276708602905
    },
    {
      "epoch": 5.0144195556640625e-06,
      "model_forward_time": 1.7742640972137451,
      "step": 79258
    },
    {
      "epoch": 5.0144195556640625e-06,
      "step": 79258,
      "training_step_time": 8.82779836654663
    },
    {
      "epoch": 5.015373229980469e-06,
      "model_forward_time": 1.7557473182678223,
      "step": 79259
    },
    {
      "epoch": 5.015373229980469e-06,
      "step": 79259,
      "training_step_time": 7.767719745635986
    },
    {
      "epoch": 5.016326904296875e-06,
      "grad_norm": 0.07492702454328537,
      "learning_rate": 1.130729133897366e-06,
      "loss": 0.0562,
      "step": 79260
    },
    {
      "epoch": 5.016326904296875e-06,
      "model_forward_time": 1.7545750141143799,
      "step": 79260
    },
    {
      "dynamics_loss_avg": 0.04461510749533772,
      "epoch": 5.016326904296875e-06,
      "step": 79260
    },
    {
      "action_loss_avg": 0.001291074586333707,
      "epoch": 5.016326904296875e-06,
      "step": 79260
    },
    {
      "epoch": 5.016326904296875e-06,
      "step": 79260,
      "training_step_time": 7.5888543128967285
    },
    {
      "epoch": 5.017280578613281e-06,
      "model_forward_time": 1.3902862071990967,
      "step": 79261
    },
    {
      "epoch": 5.017280578613281e-06,
      "step": 79261,
      "training_step_time": 7.6887524127960205
    },
    {
      "epoch": 5.018234252929687e-06,
      "model_forward_time": 1.7592604160308838,
      "step": 79262
    },
    {
      "epoch": 5.018234252929687e-06,
      "step": 79262,
      "training_step_time": 7.698901414871216
    },
    {
      "epoch": 5.019187927246094e-06,
      "model_forward_time": 1.7636668682098389,
      "step": 79263
    },
    {
      "epoch": 5.019187927246094e-06,
      "step": 79263,
      "training_step_time": 9.971125841140747
    },
    {
      "epoch": 5.0201416015625e-06,
      "model_forward_time": 1.7576773166656494,
      "step": 79264
    },
    {
      "epoch": 5.0201416015625e-06,
      "step": 79264,
      "training_step_time": 7.732895374298096
    },
    {
      "epoch": 5.021095275878906e-06,
      "model_forward_time": 1.7831037044525146,
      "step": 79265
    },
    {
      "epoch": 5.021095275878906e-06,
      "step": 79265,
      "training_step_time": 7.647434234619141
    },
    {
      "epoch": 5.0220489501953125e-06,
      "model_forward_time": 2.845366954803467,
      "step": 79266
    },
    {
      "epoch": 5.0220489501953125e-06,
      "step": 79266,
      "training_step_time": 7.686813592910767
    },
    {
      "epoch": 5.023002624511719e-06,
      "model_forward_time": 1.7496395111083984,
      "step": 79267
    },
    {
      "epoch": 5.023002624511719e-06,
      "step": 79267,
      "training_step_time": 8.188235998153687
    },
    {
      "epoch": 5.023956298828125e-06,
      "model_forward_time": 1.382131576538086,
      "step": 79268
    },
    {
      "epoch": 5.023956298828125e-06,
      "step": 79268,
      "training_step_time": 7.701884746551514
    },
    {
      "epoch": 5.024909973144531e-06,
      "model_forward_time": 1.7558748722076416,
      "step": 79269
    },
    {
      "epoch": 5.024909973144531e-06,
      "step": 79269,
      "training_step_time": 8.551161289215088
    },
    {
      "epoch": 5.025863647460937e-06,
      "grad_norm": 0.06810788810253143,
      "learning_rate": 1.1296820975382123e-06,
      "loss": 0.0562,
      "step": 79270
    },
    {
      "epoch": 5.025863647460937e-06,
      "model_forward_time": 1.3887906074523926,
      "step": 79270
    },
    {
      "dynamics_loss_avg": 0.056051872111856936,
      "epoch": 5.025863647460937e-06,
      "step": 79270
    },
    {
      "action_loss_avg": 0.0008878016727976501,
      "epoch": 5.025863647460937e-06,
      "step": 79270
    },
    {
      "epoch": 5.025863647460937e-06,
      "step": 79270,
      "training_step_time": 7.665168762207031
    },
    {
      "epoch": 5.026817321777344e-06,
      "model_forward_time": 1.7575359344482422,
      "step": 79271
    },
    {
      "epoch": 5.026817321777344e-06,
      "step": 79271,
      "training_step_time": 7.709490537643433
    },
    {
      "epoch": 5.02777099609375e-06,
      "model_forward_time": 1.778261423110962,
      "step": 79272
    },
    {
      "epoch": 5.02777099609375e-06,
      "step": 79272,
      "training_step_time": 8.709249258041382
    },
    {
      "epoch": 5.028724670410156e-06,
      "model_forward_time": 1.7823805809020996,
      "step": 79273
    },
    {
      "epoch": 5.028724670410156e-06,
      "step": 79273,
      "training_step_time": 8.573867082595825
    },
    {
      "epoch": 5.0296783447265625e-06,
      "model_forward_time": 1.8673861026763916,
      "step": 79274
    },
    {
      "epoch": 5.0296783447265625e-06,
      "step": 79274,
      "training_step_time": 7.65504002571106
    },
    {
      "epoch": 5.030632019042969e-06,
      "model_forward_time": 2.8583803176879883,
      "step": 79275
    },
    {
      "epoch": 5.030632019042969e-06,
      "step": 79275,
      "training_step_time": 8.660689115524292
    },
    {
      "epoch": 5.031585693359375e-06,
      "model_forward_time": 1.7816612720489502,
      "step": 79276
    },
    {
      "epoch": 5.031585693359375e-06,
      "step": 79276,
      "training_step_time": 7.641260862350464
    },
    {
      "epoch": 5.032539367675781e-06,
      "model_forward_time": 1.4155924320220947,
      "step": 79277
    },
    {
      "epoch": 5.032539367675781e-06,
      "step": 79277,
      "training_step_time": 8.109687805175781
    },
    {
      "epoch": 5.033493041992187e-06,
      "model_forward_time": 1.7933480739593506,
      "step": 79278
    },
    {
      "epoch": 5.033493041992187e-06,
      "step": 79278,
      "training_step_time": 7.602111339569092
    },
    {
      "epoch": 5.034446716308594e-06,
      "model_forward_time": 1.7949392795562744,
      "step": 79279
    },
    {
      "epoch": 5.034446716308594e-06,
      "step": 79279,
      "training_step_time": 7.5084147453308105
    },
    {
      "epoch": 5.035400390625e-06,
      "grad_norm": 0.10452084988355637,
      "learning_rate": 1.1286354844312076e-06,
      "loss": 0.0564,
      "step": 79280
    },
    {
      "epoch": 5.035400390625e-06,
      "model_forward_time": 1.796053409576416,
      "step": 79280
    },
    {
      "dynamics_loss_avg": 0.05683509949594736,
      "epoch": 5.035400390625e-06,
      "step": 79280
    },
    {
      "action_loss_avg": 0.002299600458354689,
      "epoch": 5.035400390625e-06,
      "step": 79280
    },
    {
      "epoch": 5.035400390625e-06,
      "step": 79280,
      "training_step_time": 7.854923725128174
    },
    {
      "epoch": 5.036354064941406e-06,
      "model_forward_time": 1.7990055084228516,
      "step": 79281
    },
    {
      "epoch": 5.036354064941406e-06,
      "step": 79281,
      "training_step_time": 8.999074935913086
    },
    {
      "epoch": 5.0373077392578125e-06,
      "model_forward_time": 1.7960047721862793,
      "step": 79282
    },
    {
      "epoch": 5.0373077392578125e-06,
      "step": 79282,
      "training_step_time": 7.644387483596802
    },
    {
      "epoch": 5.038261413574219e-06,
      "model_forward_time": 1.8544270992279053,
      "step": 79283
    },
    {
      "epoch": 5.038261413574219e-06,
      "step": 79283,
      "training_step_time": 8.327810049057007
    },
    {
      "epoch": 5.039215087890625e-06,
      "model_forward_time": 1.7903540134429932,
      "step": 79284
    },
    {
      "epoch": 5.039215087890625e-06,
      "step": 79284,
      "training_step_time": 8.115324974060059
    },
    {
      "epoch": 5.040168762207031e-06,
      "model_forward_time": 2.8699326515197754,
      "step": 79285
    },
    {
      "epoch": 5.040168762207031e-06,
      "step": 79285,
      "training_step_time": 8.506132364273071
    },
    {
      "epoch": 5.041122436523437e-06,
      "model_forward_time": 1.8163528442382812,
      "step": 79286
    },
    {
      "epoch": 5.041122436523437e-06,
      "step": 79286,
      "training_step_time": 7.515173673629761
    },
    {
      "epoch": 5.042076110839844e-06,
      "model_forward_time": 1.799910306930542,
      "step": 79287
    },
    {
      "epoch": 5.042076110839844e-06,
      "step": 79287,
      "training_step_time": 7.603055477142334
    },
    {
      "epoch": 5.04302978515625e-06,
      "model_forward_time": 1.793572187423706,
      "step": 79288
    },
    {
      "epoch": 5.04302978515625e-06,
      "step": 79288,
      "training_step_time": 7.495551109313965
    },
    {
      "epoch": 5.043983459472656e-06,
      "model_forward_time": 1.7922248840332031,
      "step": 79289
    },
    {
      "epoch": 5.043983459472656e-06,
      "step": 79289,
      "training_step_time": 7.592622518539429
    },
    {
      "epoch": 5.0449371337890625e-06,
      "grad_norm": 0.0884542465209961,
      "learning_rate": 1.1275892946908023e-06,
      "loss": 0.0552,
      "step": 79290
    },
    {
      "epoch": 5.0449371337890625e-06,
      "model_forward_time": 1.797621488571167,
      "step": 79290
    },
    {
      "dynamics_loss_avg": 0.048868057876825334,
      "epoch": 5.0449371337890625e-06,
      "step": 79290
    },
    {
      "action_loss_avg": 0.0012308826408116147,
      "epoch": 5.0449371337890625e-06,
      "step": 79290
    },
    {
      "epoch": 5.0449371337890625e-06,
      "step": 79290,
      "training_step_time": 7.817730903625488
    },
    {
      "epoch": 5.045890808105469e-06,
      "model_forward_time": 1.8017821311950684,
      "step": 79291
    },
    {
      "epoch": 5.045890808105469e-06,
      "step": 79291,
      "training_step_time": 8.655355453491211
    },
    {
      "epoch": 5.046844482421875e-06,
      "model_forward_time": 1.86098313331604,
      "step": 79292
    },
    {
      "epoch": 5.046844482421875e-06,
      "step": 79292,
      "training_step_time": 8.653451919555664
    },
    {
      "epoch": 5.047798156738281e-06,
      "model_forward_time": 1.4309515953063965,
      "step": 79293
    },
    {
      "epoch": 5.047798156738281e-06,
      "step": 79293,
      "training_step_time": 8.945815324783325
    },
    {
      "epoch": 5.048751831054687e-06,
      "model_forward_time": 2.896268606185913,
      "step": 79294
    },
    {
      "epoch": 5.048751831054687e-06,
      "step": 79294,
      "training_step_time": 9.058608770370483
    },
    {
      "epoch": 5.049705505371094e-06,
      "model_forward_time": 1.7838735580444336,
      "step": 79295
    },
    {
      "epoch": 5.049705505371094e-06,
      "step": 79295,
      "training_step_time": 7.51885986328125
    },
    {
      "epoch": 5.0506591796875e-06,
      "model_forward_time": 1.7931931018829346,
      "step": 79296
    },
    {
      "epoch": 5.0506591796875e-06,
      "step": 79296,
      "training_step_time": 7.597957134246826
    },
    {
      "epoch": 5.051612854003906e-06,
      "model_forward_time": 1.7992579936981201,
      "step": 79297
    },
    {
      "epoch": 5.051612854003906e-06,
      "step": 79297,
      "training_step_time": 7.5367584228515625
    },
    {
      "epoch": 5.0525665283203125e-06,
      "model_forward_time": 1.798969030380249,
      "step": 79298
    },
    {
      "epoch": 5.0525665283203125e-06,
      "step": 79298,
      "training_step_time": 7.632277965545654
    },
    {
      "epoch": 5.053520202636719e-06,
      "model_forward_time": 1.8009257316589355,
      "step": 79299
    },
    {
      "epoch": 5.053520202636719e-06,
      "step": 79299,
      "training_step_time": 8.68770432472229
    },
    {
      "epoch": 5.054473876953125e-06,
      "grad_norm": 0.16457459330558777,
      "learning_rate": 1.126543528431413e-06,
      "loss": 0.0554,
      "step": 79300
    },
    {
      "epoch": 5.054473876953125e-06,
      "model_forward_time": 1.7997045516967773,
      "step": 79300
    },
    {
      "dynamics_loss_avg": 0.07488773204386234,
      "epoch": 5.054473876953125e-06,
      "step": 79300
    },
    {
      "action_loss_avg": 0.0024391196406213568,
      "epoch": 5.054473876953125e-06,
      "step": 79300
    },
    {
      "epoch": 5.054473876953125e-06,
      "step": 79300,
      "training_step_time": 8.386358261108398
    },
    {
      "epoch": 5.055427551269531e-06,
      "model_forward_time": 1.865739345550537,
      "step": 79301
    },
    {
      "epoch": 5.055427551269531e-06,
      "step": 79301,
      "training_step_time": 7.619511604309082
    },
    {
      "epoch": 5.056381225585937e-06,
      "model_forward_time": 1.80023193359375,
      "step": 79302
    },
    {
      "epoch": 5.056381225585937e-06,
      "step": 79302,
      "training_step_time": 9.05482292175293
    },
    {
      "epoch": 5.057334899902344e-06,
      "model_forward_time": 2.895285129547119,
      "step": 79303
    },
    {
      "epoch": 5.057334899902344e-06,
      "step": 79303,
      "training_step_time": 8.570132493972778
    },
    {
      "epoch": 5.05828857421875e-06,
      "model_forward_time": 1.779165267944336,
      "step": 79304
    },
    {
      "epoch": 5.05828857421875e-06,
      "step": 79304,
      "training_step_time": 7.521745443344116
    },
    {
      "epoch": 5.059242248535156e-06,
      "model_forward_time": 1.8000130653381348,
      "step": 79305
    },
    {
      "epoch": 5.059242248535156e-06,
      "step": 79305,
      "training_step_time": 7.551997661590576
    },
    {
      "epoch": 5.0601959228515625e-06,
      "model_forward_time": 1.7972438335418701,
      "step": 79306
    },
    {
      "epoch": 5.0601959228515625e-06,
      "step": 79306,
      "training_step_time": 7.585237503051758
    },
    {
      "epoch": 5.061149597167969e-06,
      "model_forward_time": 1.8040969371795654,
      "step": 79307
    },
    {
      "epoch": 5.061149597167969e-06,
      "step": 79307,
      "training_step_time": 7.531543493270874
    },
    {
      "epoch": 5.062103271484375e-06,
      "model_forward_time": 1.8058209419250488,
      "step": 79308
    },
    {
      "epoch": 5.062103271484375e-06,
      "step": 79308,
      "training_step_time": 7.75018310546875
    },
    {
      "epoch": 5.063056945800781e-06,
      "model_forward_time": 1.798884630203247,
      "step": 79309
    },
    {
      "epoch": 5.063056945800781e-06,
      "step": 79309,
      "training_step_time": 7.603487968444824
    },
    {
      "epoch": 5.064010620117187e-06,
      "grad_norm": 0.35346531867980957,
      "learning_rate": 1.125498185767398e-06,
      "loss": 0.0545,
      "step": 79310
    },
    {
      "epoch": 5.064010620117187e-06,
      "model_forward_time": 1.8420536518096924,
      "step": 79310
    },
    {
      "dynamics_loss_avg": 0.04689024314284325,
      "epoch": 5.064010620117187e-06,
      "step": 79310
    },
    {
      "action_loss_avg": 0.0016686018789187074,
      "epoch": 5.064010620117187e-06,
      "step": 79310
    },
    {
      "epoch": 5.064010620117187e-06,
      "step": 79310,
      "training_step_time": 7.5930869579315186
    },
    {
      "epoch": 5.064964294433594e-06,
      "model_forward_time": 1.813605546951294,
      "step": 79311
    },
    {
      "epoch": 5.064964294433594e-06,
      "step": 79311,
      "training_step_time": 8.378827333450317
    },
    {
      "epoch": 5.06591796875e-06,
      "model_forward_time": 2.867631196975708,
      "step": 79312
    },
    {
      "epoch": 5.06591796875e-06,
      "step": 79312,
      "training_step_time": 8.291630744934082
    },
    {
      "epoch": 5.066871643066406e-06,
      "model_forward_time": 1.796626329421997,
      "step": 79313
    },
    {
      "epoch": 5.066871643066406e-06,
      "step": 79313,
      "training_step_time": 8.49291706085205
    },
    {
      "epoch": 5.0678253173828126e-06,
      "model_forward_time": 1.7953827381134033,
      "step": 79314
    },
    {
      "epoch": 5.0678253173828126e-06,
      "step": 79314,
      "training_step_time": 7.6323065757751465
    },
    {
      "epoch": 5.068778991699219e-06,
      "model_forward_time": 1.803576946258545,
      "step": 79315
    },
    {
      "epoch": 5.068778991699219e-06,
      "step": 79315,
      "training_step_time": 7.5385448932647705
    },
    {
      "epoch": 5.069732666015625e-06,
      "model_forward_time": 1.7944819927215576,
      "step": 79316
    },
    {
      "epoch": 5.069732666015625e-06,
      "step": 79316,
      "training_step_time": 7.587970018386841
    },
    {
      "epoch": 5.070686340332031e-06,
      "model_forward_time": 1.8182406425476074,
      "step": 79317
    },
    {
      "epoch": 5.070686340332031e-06,
      "step": 79317,
      "training_step_time": 7.807737350463867
    },
    {
      "epoch": 5.071640014648437e-06,
      "model_forward_time": 1.796050786972046,
      "step": 79318
    },
    {
      "epoch": 5.071640014648437e-06,
      "step": 79318,
      "training_step_time": 8.732207298278809
    },
    {
      "epoch": 5.072593688964844e-06,
      "model_forward_time": 1.8214454650878906,
      "step": 79319
    },
    {
      "epoch": 5.072593688964844e-06,
      "step": 79319,
      "training_step_time": 7.701290845870972
    },
    {
      "epoch": 5.07354736328125e-06,
      "grad_norm": 0.13772687315940857,
      "learning_rate": 1.124453266813078e-06,
      "loss": 0.0527,
      "step": 79320
    },
    {
      "epoch": 5.07354736328125e-06,
      "model_forward_time": 1.8042373657226562,
      "step": 79320
    },
    {
      "dynamics_loss_avg": 0.05620056390762329,
      "epoch": 5.07354736328125e-06,
      "step": 79320
    },
    {
      "action_loss_avg": 0.0016420345316873863,
      "epoch": 5.07354736328125e-06,
      "step": 79320
    },
    {
      "epoch": 5.07354736328125e-06,
      "step": 79320,
      "training_step_time": 7.572488069534302
    },
    {
      "epoch": 5.074501037597656e-06,
      "model_forward_time": 1.7986390590667725,
      "step": 79321
    },
    {
      "epoch": 5.074501037597656e-06,
      "step": 79321,
      "training_step_time": 8.150336265563965
    },
    {
      "epoch": 5.0754547119140626e-06,
      "model_forward_time": 1.787325143814087,
      "step": 79322
    },
    {
      "epoch": 5.0754547119140626e-06,
      "step": 79322,
      "training_step_time": 9.471987009048462
    },
    {
      "epoch": 5.076408386230469e-06,
      "model_forward_time": 1.8156051635742188,
      "step": 79323
    },
    {
      "epoch": 5.076408386230469e-06,
      "step": 79323,
      "training_step_time": 8.31009578704834
    },
    {
      "epoch": 5.077362060546875e-06,
      "model_forward_time": 1.81465482711792,
      "step": 79324
    },
    {
      "epoch": 5.077362060546875e-06,
      "step": 79324,
      "training_step_time": 7.563321828842163
    },
    {
      "epoch": 5.078315734863281e-06,
      "model_forward_time": 1.8233776092529297,
      "step": 79325
    },
    {
      "epoch": 5.078315734863281e-06,
      "step": 79325,
      "training_step_time": 8.42699909210205
    },
    {
      "epoch": 5.079269409179687e-06,
      "model_forward_time": 1.829803705215454,
      "step": 79326
    },
    {
      "epoch": 5.079269409179687e-06,
      "step": 79326,
      "training_step_time": 8.371577501296997
    },
    {
      "epoch": 5.080223083496094e-06,
      "model_forward_time": 1.8160772323608398,
      "step": 79327
    },
    {
      "epoch": 5.080223083496094e-06,
      "step": 79327,
      "training_step_time": 8.769492626190186
    },
    {
      "epoch": 5.0811767578125e-06,
      "model_forward_time": 1.812438726425171,
      "step": 79328
    },
    {
      "epoch": 5.0811767578125e-06,
      "step": 79328,
      "training_step_time": 7.65989875793457
    },
    {
      "epoch": 5.082130432128906e-06,
      "model_forward_time": 1.8870413303375244,
      "step": 79329
    },
    {
      "epoch": 5.082130432128906e-06,
      "step": 79329,
      "training_step_time": 10.701072692871094
    },
    {
      "epoch": 5.0830841064453126e-06,
      "grad_norm": 0.09714716672897339,
      "learning_rate": 1.1234087716827203e-06,
      "loss": 0.0531,
      "step": 79330
    },
    {
      "epoch": 5.0830841064453126e-06,
      "model_forward_time": 1.8158860206604004,
      "step": 79330
    },
    {
      "dynamics_loss_avg": 0.03900568503886461,
      "epoch": 5.0830841064453126e-06,
      "step": 79330
    },
    {
      "action_loss_avg": 0.0007670998777030036,
      "epoch": 5.0830841064453126e-06,
      "step": 79330
    },
    {
      "epoch": 5.0830841064453126e-06,
      "step": 79330,
      "training_step_time": 8.643173217773438
    },
    {
      "epoch": 5.084037780761719e-06,
      "model_forward_time": 2.96250581741333,
      "step": 79331
    },
    {
      "epoch": 5.084037780761719e-06,
      "step": 79331,
      "training_step_time": 7.720545291900635
    },
    {
      "epoch": 5.084991455078125e-06,
      "model_forward_time": 1.811180591583252,
      "step": 79332
    },
    {
      "epoch": 5.084991455078125e-06,
      "step": 79332,
      "training_step_time": 7.627100229263306
    },
    {
      "epoch": 5.0859451293945314e-06,
      "model_forward_time": 1.8165185451507568,
      "step": 79333
    },
    {
      "epoch": 5.0859451293945314e-06,
      "step": 79333,
      "training_step_time": 7.539200782775879
    },
    {
      "epoch": 5.086898803710937e-06,
      "model_forward_time": 1.81315016746521,
      "step": 79334
    },
    {
      "epoch": 5.086898803710937e-06,
      "step": 79334,
      "training_step_time": 7.61431884765625
    },
    {
      "epoch": 5.087852478027344e-06,
      "model_forward_time": 1.8192474842071533,
      "step": 79335
    },
    {
      "epoch": 5.087852478027344e-06,
      "step": 79335,
      "training_step_time": 7.6796534061431885
    },
    {
      "epoch": 5.08880615234375e-06,
      "model_forward_time": 1.8255352973937988,
      "step": 79336
    },
    {
      "epoch": 5.08880615234375e-06,
      "step": 79336,
      "training_step_time": 8.504665851593018
    },
    {
      "epoch": 5.089759826660156e-06,
      "model_forward_time": 1.8094911575317383,
      "step": 79337
    },
    {
      "epoch": 5.089759826660156e-06,
      "step": 79337,
      "training_step_time": 8.935763359069824
    },
    {
      "epoch": 5.0907135009765626e-06,
      "model_forward_time": 1.8792402744293213,
      "step": 79338
    },
    {
      "epoch": 5.0907135009765626e-06,
      "step": 79338,
      "training_step_time": 8.312736988067627
    },
    {
      "epoch": 5.091667175292969e-06,
      "model_forward_time": 1.8110501766204834,
      "step": 79339
    },
    {
      "epoch": 5.091667175292969e-06,
      "step": 79339,
      "training_step_time": 7.57261323928833
    },
    {
      "epoch": 5.092620849609375e-06,
      "grad_norm": 0.08333420008420944,
      "learning_rate": 1.1223647004905525e-06,
      "loss": 0.0537,
      "step": 79340
    },
    {
      "epoch": 5.092620849609375e-06,
      "model_forward_time": 2.9476265907287598,
      "step": 79340
    },
    {
      "dynamics_loss_avg": 0.05037194490432739,
      "epoch": 5.092620849609375e-06,
      "step": 79340
    },
    {
      "action_loss_avg": 0.0008881923829903826,
      "epoch": 5.092620849609375e-06,
      "step": 79340
    },
    {
      "epoch": 5.092620849609375e-06,
      "step": 79340,
      "training_step_time": 8.43594765663147
    },
    {
      "epoch": 5.0935745239257814e-06,
      "model_forward_time": 1.810978651046753,
      "step": 79341
    },
    {
      "epoch": 5.0935745239257814e-06,
      "step": 79341,
      "training_step_time": 8.962665557861328
    },
    {
      "epoch": 5.094528198242187e-06,
      "model_forward_time": 1.814272403717041,
      "step": 79342
    },
    {
      "epoch": 5.094528198242187e-06,
      "step": 79342,
      "training_step_time": 7.5946738719940186
    },
    {
      "epoch": 5.095481872558594e-06,
      "model_forward_time": 1.8127079010009766,
      "step": 79343
    },
    {
      "epoch": 5.095481872558594e-06,
      "step": 79343,
      "training_step_time": 7.5675294399261475
    },
    {
      "epoch": 5.096435546875e-06,
      "model_forward_time": 1.8185293674468994,
      "step": 79344
    },
    {
      "epoch": 5.096435546875e-06,
      "step": 79344,
      "training_step_time": 8.29268741607666
    },
    {
      "epoch": 5.097389221191406e-06,
      "model_forward_time": 1.8094871044158936,
      "step": 79345
    },
    {
      "epoch": 5.097389221191406e-06,
      "step": 79345,
      "training_step_time": 7.809601068496704
    },
    {
      "epoch": 5.098342895507813e-06,
      "model_forward_time": 1.8085224628448486,
      "step": 79346
    },
    {
      "epoch": 5.098342895507813e-06,
      "step": 79346,
      "training_step_time": 8.369082689285278
    },
    {
      "epoch": 5.099296569824219e-06,
      "model_forward_time": 1.8535711765289307,
      "step": 79347
    },
    {
      "epoch": 5.099296569824219e-06,
      "step": 79347,
      "training_step_time": 10.018508434295654
    },
    {
      "epoch": 5.100250244140625e-06,
      "model_forward_time": 1.8090496063232422,
      "step": 79348
    },
    {
      "epoch": 5.100250244140625e-06,
      "step": 79348,
      "training_step_time": 8.30691123008728
    },
    {
      "epoch": 5.1012039184570314e-06,
      "model_forward_time": 2.933832883834839,
      "step": 79349
    },
    {
      "epoch": 5.1012039184570314e-06,
      "step": 79349,
      "training_step_time": 8.522262334823608
    },
    {
      "epoch": 5.102157592773437e-06,
      "grad_norm": 0.11914949864149094,
      "learning_rate": 1.1213210533507485e-06,
      "loss": 0.0538,
      "step": 79350
    },
    {
      "epoch": 5.102157592773437e-06,
      "model_forward_time": 1.8075170516967773,
      "step": 79350
    },
    {
      "dynamics_loss_avg": 0.042014497146010396,
      "epoch": 5.102157592773437e-06,
      "step": 79350
    },
    {
      "action_loss_avg": 0.001555049663875252,
      "epoch": 5.102157592773437e-06,
      "step": 79350
    },
    {
      "epoch": 5.102157592773437e-06,
      "step": 79350,
      "training_step_time": 8.510309934616089
    },
    {
      "epoch": 5.103111267089844e-06,
      "model_forward_time": 1.8330204486846924,
      "step": 79351
    },
    {
      "epoch": 5.103111267089844e-06,
      "step": 79351,
      "training_step_time": 7.583794832229614
    },
    {
      "epoch": 5.10406494140625e-06,
      "model_forward_time": 1.801497459411621,
      "step": 79352
    },
    {
      "epoch": 5.10406494140625e-06,
      "step": 79352,
      "training_step_time": 8.383618831634521
    },
    {
      "epoch": 5.105018615722656e-06,
      "model_forward_time": 1.81561279296875,
      "step": 79353
    },
    {
      "epoch": 5.105018615722656e-06,
      "step": 79353,
      "training_step_time": 7.6188130378723145
    },
    {
      "epoch": 5.105972290039063e-06,
      "model_forward_time": 1.8284218311309814,
      "step": 79354
    },
    {
      "epoch": 5.105972290039063e-06,
      "step": 79354,
      "training_step_time": 8.598158359527588
    },
    {
      "epoch": 5.106925964355469e-06,
      "model_forward_time": 1.8102507591247559,
      "step": 79355
    },
    {
      "epoch": 5.106925964355469e-06,
      "step": 79355,
      "training_step_time": 8.484992742538452
    },
    {
      "epoch": 5.107879638671875e-06,
      "model_forward_time": 1.8283934593200684,
      "step": 79356
    },
    {
      "epoch": 5.107879638671875e-06,
      "step": 79356,
      "training_step_time": 8.278218030929565
    },
    {
      "epoch": 5.1088333129882814e-06,
      "model_forward_time": 1.8081324100494385,
      "step": 79357
    },
    {
      "epoch": 5.1088333129882814e-06,
      "step": 79357,
      "training_step_time": 8.915675401687622
    },
    {
      "epoch": 5.109786987304687e-06,
      "model_forward_time": 2.9050934314727783,
      "step": 79358
    },
    {
      "epoch": 5.109786987304687e-06,
      "step": 79358,
      "training_step_time": 7.702026844024658
    },
    {
      "epoch": 5.110740661621094e-06,
      "model_forward_time": 1.822054386138916,
      "step": 79359
    },
    {
      "epoch": 5.110740661621094e-06,
      "step": 79359,
      "training_step_time": 9.64353322982788
    },
    {
      "epoch": 5.1116943359375e-06,
      "grad_norm": 0.11505139619112015,
      "learning_rate": 1.1202778303774448e-06,
      "loss": 0.052,
      "step": 79360
    },
    {
      "epoch": 5.1116943359375e-06,
      "model_forward_time": 1.8189365863800049,
      "step": 79360
    },
    {
      "dynamics_loss_avg": 0.04233893807977438,
      "epoch": 5.1116943359375e-06,
      "step": 79360
    },
    {
      "action_loss_avg": 0.0009023762133438141,
      "epoch": 5.1116943359375e-06,
      "step": 79360
    },
    {
      "epoch": 5.1116943359375e-06,
      "step": 79360,
      "training_step_time": 8.332983493804932
    },
    {
      "epoch": 5.112648010253906e-06,
      "model_forward_time": 1.8224592208862305,
      "step": 79361
    },
    {
      "epoch": 5.112648010253906e-06,
      "step": 79361,
      "training_step_time": 7.5305845737457275
    },
    {
      "epoch": 5.113601684570313e-06,
      "model_forward_time": 1.8206584453582764,
      "step": 79362
    },
    {
      "epoch": 5.113601684570313e-06,
      "step": 79362,
      "training_step_time": 7.61120343208313
    },
    {
      "epoch": 5.114555358886719e-06,
      "model_forward_time": 1.8184609413146973,
      "step": 79363
    },
    {
      "epoch": 5.114555358886719e-06,
      "step": 79363,
      "training_step_time": 7.583863973617554
    },
    {
      "epoch": 5.115509033203125e-06,
      "model_forward_time": 1.7958757877349854,
      "step": 79364
    },
    {
      "epoch": 5.115509033203125e-06,
      "step": 79364,
      "training_step_time": 9.742879629135132
    },
    {
      "epoch": 5.1164627075195315e-06,
      "model_forward_time": 1.782358169555664,
      "step": 79365
    },
    {
      "epoch": 5.1164627075195315e-06,
      "step": 79365,
      "training_step_time": 7.641439199447632
    },
    {
      "epoch": 5.117416381835937e-06,
      "model_forward_time": 1.8617093563079834,
      "step": 79366
    },
    {
      "epoch": 5.117416381835937e-06,
      "step": 79366,
      "training_step_time": 7.59243106842041
    },
    {
      "epoch": 5.118370056152344e-06,
      "model_forward_time": 2.879230260848999,
      "step": 79367
    },
    {
      "epoch": 5.118370056152344e-06,
      "step": 79367,
      "training_step_time": 8.178194284439087
    },
    {
      "epoch": 5.11932373046875e-06,
      "model_forward_time": 1.784264087677002,
      "step": 79368
    },
    {
      "epoch": 5.11932373046875e-06,
      "step": 79368,
      "training_step_time": 7.604354381561279
    },
    {
      "epoch": 5.120277404785156e-06,
      "model_forward_time": 1.1015582084655762,
      "step": 79369
    },
    {
      "epoch": 5.120277404785156e-06,
      "step": 79369,
      "training_step_time": 7.604513883590698
    },
    {
      "epoch": 5.121231079101563e-06,
      "grad_norm": 0.0774388536810875,
      "learning_rate": 1.1192350316847234e-06,
      "loss": 0.0534,
      "step": 79370
    },
    {
      "epoch": 5.121231079101563e-06,
      "model_forward_time": 1.7833764553070068,
      "step": 79370
    },
    {
      "dynamics_loss_avg": 0.06141720190644264,
      "epoch": 5.121231079101563e-06,
      "step": 79370
    },
    {
      "action_loss_avg": 0.001578907057410106,
      "epoch": 5.121231079101563e-06,
      "step": 79370
    },
    {
      "epoch": 5.121231079101563e-06,
      "step": 79370,
      "training_step_time": 7.532227516174316
    },
    {
      "epoch": 5.122184753417969e-06,
      "model_forward_time": 1.796149730682373,
      "step": 79371
    },
    {
      "epoch": 5.122184753417969e-06,
      "step": 79371,
      "training_step_time": 7.610323429107666
    },
    {
      "epoch": 5.123138427734375e-06,
      "model_forward_time": 1.7865309715270996,
      "step": 79372
    },
    {
      "epoch": 5.123138427734375e-06,
      "step": 79372,
      "training_step_time": 7.558985948562622
    },
    {
      "epoch": 5.1240921020507815e-06,
      "model_forward_time": 1.7938570976257324,
      "step": 79373
    },
    {
      "epoch": 5.1240921020507815e-06,
      "step": 79373,
      "training_step_time": 7.5093114376068115
    },
    {
      "epoch": 5.125045776367187e-06,
      "model_forward_time": 1.7936291694641113,
      "step": 79374
    },
    {
      "epoch": 5.125045776367187e-06,
      "step": 79374,
      "training_step_time": 8.99722671508789
    },
    {
      "epoch": 5.125999450683594e-06,
      "model_forward_time": 1.8412866592407227,
      "step": 79375
    },
    {
      "epoch": 5.125999450683594e-06,
      "step": 79375,
      "training_step_time": 8.070120811462402
    },
    {
      "epoch": 5.126953125e-06,
      "model_forward_time": 2.8848912715911865,
      "step": 79376
    },
    {
      "epoch": 5.126953125e-06,
      "step": 79376,
      "training_step_time": 9.13549542427063
    },
    {
      "epoch": 5.127906799316406e-06,
      "model_forward_time": 1.7901084423065186,
      "step": 79377
    },
    {
      "epoch": 5.127906799316406e-06,
      "step": 79377,
      "training_step_time": 6.674986362457275
    },
    {
      "epoch": 5.128860473632813e-06,
      "model_forward_time": 1.7937145233154297,
      "step": 79378
    },
    {
      "epoch": 5.128860473632813e-06,
      "step": 79378,
      "training_step_time": 7.516936779022217
    },
    {
      "epoch": 5.129814147949218e-06,
      "model_forward_time": 1.7993526458740234,
      "step": 79379
    },
    {
      "epoch": 5.129814147949218e-06,
      "step": 79379,
      "training_step_time": 7.492908477783203
    },
    {
      "epoch": 5.130767822265625e-06,
      "grad_norm": 0.12490297853946686,
      "learning_rate": 1.118192657386626e-06,
      "loss": 0.0527,
      "step": 79380
    },
    {
      "epoch": 5.130767822265625e-06,
      "model_forward_time": 1.799675464630127,
      "step": 79380
    },
    {
      "dynamics_loss_avg": 0.04691696371883154,
      "epoch": 5.130767822265625e-06,
      "step": 79380
    },
    {
      "action_loss_avg": 0.0009003128769109025,
      "epoch": 5.130767822265625e-06,
      "step": 79380
    },
    {
      "epoch": 5.130767822265625e-06,
      "step": 79380,
      "training_step_time": 8.644927024841309
    },
    {
      "epoch": 5.1317214965820315e-06,
      "model_forward_time": 1.7925608158111572,
      "step": 79381
    },
    {
      "epoch": 5.1317214965820315e-06,
      "step": 79381,
      "training_step_time": 6.682759523391724
    },
    {
      "epoch": 5.132675170898437e-06,
      "model_forward_time": 1.7962908744812012,
      "step": 79382
    },
    {
      "epoch": 5.132675170898437e-06,
      "step": 79382,
      "training_step_time": 6.639505624771118
    },
    {
      "epoch": 5.133628845214844e-06,
      "model_forward_time": 1.7987747192382812,
      "step": 79383
    },
    {
      "epoch": 5.133628845214844e-06,
      "step": 79383,
      "training_step_time": 6.667605876922607
    },
    {
      "epoch": 5.13458251953125e-06,
      "model_forward_time": 1.7892379760742188,
      "step": 79384
    },
    {
      "epoch": 5.13458251953125e-06,
      "step": 79384,
      "training_step_time": 8.288215398788452
    },
    {
      "epoch": 5.135536193847656e-06,
      "model_forward_time": 1.8115370273590088,
      "step": 79385
    },
    {
      "epoch": 5.135536193847656e-06,
      "step": 79385,
      "training_step_time": 11.203350067138672
    },
    {
      "epoch": 5.136489868164063e-06,
      "model_forward_time": 1.7969350814819336,
      "step": 79386
    },
    {
      "epoch": 5.136489868164063e-06,
      "step": 79386,
      "training_step_time": 8.518254518508911
    },
    {
      "epoch": 5.137443542480468e-06,
      "model_forward_time": 1.8000657558441162,
      "step": 79387
    },
    {
      "epoch": 5.137443542480468e-06,
      "step": 79387,
      "training_step_time": 7.052721261978149
    },
    {
      "epoch": 5.138397216796875e-06,
      "model_forward_time": 1.7992067337036133,
      "step": 79388
    },
    {
      "epoch": 5.138397216796875e-06,
      "step": 79388,
      "training_step_time": 7.681647300720215
    },
    {
      "epoch": 5.1393508911132815e-06,
      "model_forward_time": 1.798370122909546,
      "step": 79389
    },
    {
      "epoch": 5.1393508911132815e-06,
      "step": 79389,
      "training_step_time": 7.712230920791626
    },
    {
      "epoch": 5.140304565429687e-06,
      "grad_norm": 0.07667025178670883,
      "learning_rate": 1.1171507075971416e-06,
      "loss": 0.053,
      "step": 79390
    },
    {
      "epoch": 5.140304565429687e-06,
      "model_forward_time": 1.805497407913208,
      "step": 79390
    },
    {
      "dynamics_loss_avg": 0.04939762828871608,
      "epoch": 5.140304565429687e-06,
      "step": 79390
    },
    {
      "action_loss_avg": 0.002354168155579828,
      "epoch": 5.140304565429687e-06,
      "step": 79390
    },
    {
      "epoch": 5.140304565429687e-06,
      "step": 79390,
      "training_step_time": 6.783899545669556
    },
    {
      "epoch": 5.141258239746094e-06,
      "model_forward_time": 1.8024697303771973,
      "step": 79391
    },
    {
      "epoch": 5.141258239746094e-06,
      "step": 79391,
      "training_step_time": 6.774524927139282
    },
    {
      "epoch": 5.1422119140625e-06,
      "model_forward_time": 1.7935693264007568,
      "step": 79392
    },
    {
      "epoch": 5.1422119140625e-06,
      "step": 79392,
      "training_step_time": 6.818793773651123
    },
    {
      "epoch": 5.143165588378906e-06,
      "model_forward_time": 1.7882444858551025,
      "step": 79393
    },
    {
      "epoch": 5.143165588378906e-06,
      "step": 79393,
      "training_step_time": 6.7541022300720215
    },
    {
      "epoch": 5.144119262695313e-06,
      "model_forward_time": 2.8513243198394775,
      "step": 79394
    },
    {
      "epoch": 5.144119262695313e-06,
      "step": 79394,
      "training_step_time": 10.385090351104736
    },
    {
      "epoch": 5.145072937011718e-06,
      "model_forward_time": 1.7958550453186035,
      "step": 79395
    },
    {
      "epoch": 5.145072937011718e-06,
      "step": 79395,
      "training_step_time": 7.830795049667358
    },
    {
      "epoch": 5.146026611328125e-06,
      "model_forward_time": 1.7904505729675293,
      "step": 79396
    },
    {
      "epoch": 5.146026611328125e-06,
      "step": 79396,
      "training_step_time": 7.372555732727051
    },
    {
      "epoch": 5.1469802856445315e-06,
      "model_forward_time": 1.7978534698486328,
      "step": 79397
    },
    {
      "epoch": 5.1469802856445315e-06,
      "step": 79397,
      "training_step_time": 6.826977968215942
    },
    {
      "epoch": 5.147933959960937e-06,
      "model_forward_time": 1.8098835945129395,
      "step": 79398
    },
    {
      "epoch": 5.147933959960937e-06,
      "step": 79398,
      "training_step_time": 7.6401121616363525
    },
    {
      "epoch": 5.148887634277344e-06,
      "model_forward_time": 1.7891204357147217,
      "step": 79399
    },
    {
      "epoch": 5.148887634277344e-06,
      "step": 79399,
      "training_step_time": 7.158635854721069
    },
    {
      "epoch": 5.14984130859375e-06,
      "grad_norm": 0.07775761187076569,
      "learning_rate": 1.1161091824302178e-06,
      "loss": 0.0531,
      "step": 79400
    },
    {
      "epoch": 5.14984130859375e-06,
      "model_forward_time": 1.7904491424560547,
      "step": 79400
    },
    {
      "dynamics_loss_avg": 0.03902620170265436,
      "epoch": 5.14984130859375e-06,
      "step": 79400
    },
    {
      "action_loss_avg": 0.0006100069949752651,
      "epoch": 5.14984130859375e-06,
      "step": 79400
    },
    {
      "epoch": 5.14984130859375e-06,
      "step": 79400,
      "training_step_time": 6.796144962310791
    },
    {
      "epoch": 5.150794982910156e-06,
      "model_forward_time": 1.7948439121246338,
      "step": 79401
    },
    {
      "epoch": 5.150794982910156e-06,
      "step": 79401,
      "training_step_time": 6.7835633754730225
    },
    {
      "epoch": 5.151748657226563e-06,
      "model_forward_time": 1.7930033206939697,
      "step": 79402
    },
    {
      "epoch": 5.151748657226563e-06,
      "step": 79402,
      "training_step_time": 6.799030780792236
    },
    {
      "epoch": 5.152702331542968e-06,
      "model_forward_time": 2.920280933380127,
      "step": 79403
    },
    {
      "epoch": 5.152702331542968e-06,
      "step": 79403,
      "training_step_time": 7.81354284286499
    },
    {
      "epoch": 5.153656005859375e-06,
      "model_forward_time": 1.797558307647705,
      "step": 79404
    },
    {
      "epoch": 5.153656005859375e-06,
      "step": 79404,
      "training_step_time": 9.594825029373169
    },
    {
      "epoch": 5.1546096801757815e-06,
      "model_forward_time": 1.8023688793182373,
      "step": 79405
    },
    {
      "epoch": 5.1546096801757815e-06,
      "step": 79405,
      "training_step_time": 7.758995532989502
    },
    {
      "epoch": 5.155563354492187e-06,
      "model_forward_time": 1.808091163635254,
      "step": 79406
    },
    {
      "epoch": 5.155563354492187e-06,
      "step": 79406,
      "training_step_time": 7.501383304595947
    },
    {
      "epoch": 5.156517028808594e-06,
      "model_forward_time": 1.7978956699371338,
      "step": 79407
    },
    {
      "epoch": 5.156517028808594e-06,
      "step": 79407,
      "training_step_time": 8.686614036560059
    },
    {
      "epoch": 5.157470703125e-06,
      "model_forward_time": 1.798097848892212,
      "step": 79408
    },
    {
      "epoch": 5.157470703125e-06,
      "step": 79408,
      "training_step_time": 7.601335525512695
    },
    {
      "epoch": 5.158424377441406e-06,
      "model_forward_time": 1.8096363544464111,
      "step": 79409
    },
    {
      "epoch": 5.158424377441406e-06,
      "step": 79409,
      "training_step_time": 6.830810070037842
    },
    {
      "epoch": 5.159378051757813e-06,
      "grad_norm": 0.08884535729885101,
      "learning_rate": 1.115068081999755e-06,
      "loss": 0.0544,
      "step": 79410
    },
    {
      "epoch": 5.159378051757813e-06,
      "model_forward_time": 1.784621000289917,
      "step": 79410
    },
    {
      "dynamics_loss_avg": 0.04669261928647757,
      "epoch": 5.159378051757813e-06,
      "step": 79410
    },
    {
      "action_loss_avg": 0.0012819446244975552,
      "epoch": 5.159378051757813e-06,
      "step": 79410
    },
    {
      "epoch": 5.159378051757813e-06,
      "step": 79410,
      "training_step_time": 6.7702107429504395
    },
    {
      "epoch": 5.160331726074218e-06,
      "model_forward_time": 1.7892959117889404,
      "step": 79411
    },
    {
      "epoch": 5.160331726074218e-06,
      "step": 79411,
      "training_step_time": 6.850135803222656
    },
    {
      "epoch": 5.161285400390625e-06,
      "model_forward_time": 2.8522212505340576,
      "step": 79412
    },
    {
      "epoch": 5.161285400390625e-06,
      "step": 79412,
      "training_step_time": 8.356334447860718
    },
    {
      "epoch": 5.1622390747070315e-06,
      "model_forward_time": 1.819498062133789,
      "step": 79413
    },
    {
      "epoch": 5.1622390747070315e-06,
      "step": 79413,
      "training_step_time": 9.49308705329895
    },
    {
      "epoch": 5.163192749023437e-06,
      "model_forward_time": 1.8010916709899902,
      "step": 79414
    },
    {
      "epoch": 5.163192749023437e-06,
      "step": 79414,
      "training_step_time": 7.666640520095825
    },
    {
      "epoch": 5.164146423339844e-06,
      "model_forward_time": 1.8025989532470703,
      "step": 79415
    },
    {
      "epoch": 5.164146423339844e-06,
      "step": 79415,
      "training_step_time": 7.4133734703063965
    },
    {
      "epoch": 5.16510009765625e-06,
      "model_forward_time": 1.8182237148284912,
      "step": 79416
    },
    {
      "epoch": 5.16510009765625e-06,
      "step": 79416,
      "training_step_time": 7.541064500808716
    },
    {
      "epoch": 5.166053771972656e-06,
      "model_forward_time": 1.798471212387085,
      "step": 79417
    },
    {
      "epoch": 5.166053771972656e-06,
      "step": 79417,
      "training_step_time": 7.583575963973999
    },
    {
      "epoch": 5.167007446289063e-06,
      "model_forward_time": 1.8079144954681396,
      "step": 79418
    },
    {
      "epoch": 5.167007446289063e-06,
      "step": 79418,
      "training_step_time": 6.646416902542114
    },
    {
      "epoch": 5.167961120605468e-06,
      "model_forward_time": 1.804856777191162,
      "step": 79419
    },
    {
      "epoch": 5.167961120605468e-06,
      "step": 79419,
      "training_step_time": 6.658278465270996
    },
    {
      "epoch": 5.168914794921875e-06,
      "grad_norm": 0.17603379487991333,
      "learning_rate": 1.1140274064196071e-06,
      "loss": 0.0556,
      "step": 79420
    },
    {
      "epoch": 5.168914794921875e-06,
      "model_forward_time": 1.7894046306610107,
      "step": 79420
    },
    {
      "dynamics_loss_avg": 0.05536258853971958,
      "epoch": 5.168914794921875e-06,
      "step": 79420
    },
    {
      "action_loss_avg": 0.0015708065155195072,
      "epoch": 5.168914794921875e-06,
      "step": 79420
    },
    {
      "epoch": 5.168914794921875e-06,
      "step": 79420,
      "training_step_time": 6.709717512130737
    },
    {
      "epoch": 5.1698684692382815e-06,
      "model_forward_time": 2.9771580696105957,
      "step": 79421
    },
    {
      "epoch": 5.1698684692382815e-06,
      "step": 79421,
      "training_step_time": 7.744930982589722
    },
    {
      "epoch": 5.170822143554687e-06,
      "model_forward_time": 1.809568166732788,
      "step": 79422
    },
    {
      "epoch": 5.170822143554687e-06,
      "step": 79422,
      "training_step_time": 8.204938650131226
    },
    {
      "epoch": 5.171775817871094e-06,
      "model_forward_time": 1.7922139167785645,
      "step": 79423
    },
    {
      "epoch": 5.171775817871094e-06,
      "step": 79423,
      "training_step_time": 7.719732761383057
    },
    {
      "epoch": 5.1727294921875e-06,
      "model_forward_time": 1.8147640228271484,
      "step": 79424
    },
    {
      "epoch": 5.1727294921875e-06,
      "step": 79424,
      "training_step_time": 7.5884881019592285
    },
    {
      "epoch": 5.173683166503906e-06,
      "model_forward_time": 1.806183099746704,
      "step": 79425
    },
    {
      "epoch": 5.173683166503906e-06,
      "step": 79425,
      "training_step_time": 7.638336658477783
    },
    {
      "epoch": 5.174636840820313e-06,
      "model_forward_time": 1.7948777675628662,
      "step": 79426
    },
    {
      "epoch": 5.174636840820313e-06,
      "step": 79426,
      "training_step_time": 7.61313533782959
    },
    {
      "epoch": 5.175590515136718e-06,
      "model_forward_time": 1.7933290004730225,
      "step": 79427
    },
    {
      "epoch": 5.175590515136718e-06,
      "step": 79427,
      "training_step_time": 7.092965126037598
    },
    {
      "epoch": 5.176544189453125e-06,
      "model_forward_time": 1.8040964603424072,
      "step": 79428
    },
    {
      "epoch": 5.176544189453125e-06,
      "step": 79428,
      "training_step_time": 6.772484064102173
    },
    {
      "epoch": 5.1774978637695315e-06,
      "model_forward_time": 1.815476417541504,
      "step": 79429
    },
    {
      "epoch": 5.1774978637695315e-06,
      "step": 79429,
      "training_step_time": 6.7872960567474365
    },
    {
      "epoch": 5.178451538085937e-06,
      "grad_norm": 0.1352090984582901,
      "learning_rate": 1.1129871558035781e-06,
      "loss": 0.0537,
      "step": 79430
    },
    {
      "epoch": 5.178451538085937e-06,
      "model_forward_time": 2.8695907592773438,
      "step": 79430
    },
    {
      "dynamics_loss_avg": 0.04978228285908699,
      "epoch": 5.178451538085937e-06,
      "step": 79430
    },
    {
      "action_loss_avg": 0.0007983698684256524,
      "epoch": 5.178451538085937e-06,
      "step": 79430
    },
    {
      "epoch": 5.178451538085937e-06,
      "step": 79430,
      "training_step_time": 7.729033946990967
    },
    {
      "epoch": 5.179405212402344e-06,
      "model_forward_time": 1.8397488594055176,
      "step": 79431
    },
    {
      "epoch": 5.179405212402344e-06,
      "step": 79431,
      "training_step_time": 9.56932520866394
    },
    {
      "epoch": 5.18035888671875e-06,
      "model_forward_time": 1.448115348815918,
      "step": 79432
    },
    {
      "epoch": 5.18035888671875e-06,
      "step": 79432,
      "training_step_time": 7.6847243309021
    },
    {
      "epoch": 5.181312561035156e-06,
      "model_forward_time": 1.7787301540374756,
      "step": 79433
    },
    {
      "epoch": 5.181312561035156e-06,
      "step": 79433,
      "training_step_time": 7.656795501708984
    },
    {
      "epoch": 5.182266235351563e-06,
      "model_forward_time": 1.7964999675750732,
      "step": 79434
    },
    {
      "epoch": 5.182266235351563e-06,
      "step": 79434,
      "training_step_time": 7.588318109512329
    },
    {
      "epoch": 5.183219909667968e-06,
      "model_forward_time": 1.7905325889587402,
      "step": 79435
    },
    {
      "epoch": 5.183219909667968e-06,
      "step": 79435,
      "training_step_time": 7.553753852844238
    },
    {
      "epoch": 5.184173583984375e-06,
      "model_forward_time": 1.774749755859375,
      "step": 79436
    },
    {
      "epoch": 5.184173583984375e-06,
      "step": 79436,
      "training_step_time": 7.109169960021973
    },
    {
      "epoch": 5.1851272583007815e-06,
      "model_forward_time": 1.7804574966430664,
      "step": 79437
    },
    {
      "epoch": 5.1851272583007815e-06,
      "step": 79437,
      "training_step_time": 6.685132026672363
    },
    {
      "epoch": 5.186080932617187e-06,
      "model_forward_time": 1.7881975173950195,
      "step": 79438
    },
    {
      "epoch": 5.186080932617187e-06,
      "step": 79438,
      "training_step_time": 6.717620134353638
    },
    {
      "epoch": 5.187034606933594e-06,
      "model_forward_time": 1.7875618934631348,
      "step": 79439
    },
    {
      "epoch": 5.187034606933594e-06,
      "step": 79439,
      "training_step_time": 6.69441556930542
    },
    {
      "epoch": 5.18798828125e-06,
      "grad_norm": 0.09625083953142166,
      "learning_rate": 1.11194733026543e-06,
      "loss": 0.0561,
      "step": 79440
    },
    {
      "epoch": 5.18798828125e-06,
      "model_forward_time": 2.892108678817749,
      "step": 79440
    },
    {
      "dynamics_loss_avg": 0.0511199040338397,
      "epoch": 5.18798828125e-06,
      "step": 79440
    },
    {
      "action_loss_avg": 0.0012722849496640265,
      "epoch": 5.18798828125e-06,
      "step": 79440
    },
    {
      "epoch": 5.18798828125e-06,
      "step": 79440,
      "training_step_time": 8.628365516662598
    },
    {
      "epoch": 5.188941955566406e-06,
      "model_forward_time": 1.7957549095153809,
      "step": 79441
    },
    {
      "epoch": 5.188941955566406e-06,
      "step": 79441,
      "training_step_time": 7.617704629898071
    },
    {
      "epoch": 5.189895629882813e-06,
      "model_forward_time": 1.8064520359039307,
      "step": 79442
    },
    {
      "epoch": 5.189895629882813e-06,
      "step": 79442,
      "training_step_time": 7.693323373794556
    },
    {
      "epoch": 5.190849304199218e-06,
      "model_forward_time": 1.7960999011993408,
      "step": 79443
    },
    {
      "epoch": 5.190849304199218e-06,
      "step": 79443,
      "training_step_time": 8.415888547897339
    },
    {
      "epoch": 5.191802978515625e-06,
      "model_forward_time": 1.8110086917877197,
      "step": 79444
    },
    {
      "epoch": 5.191802978515625e-06,
      "step": 79444,
      "training_step_time": 7.617060422897339
    },
    {
      "epoch": 5.1927566528320315e-06,
      "model_forward_time": 1.806372880935669,
      "step": 79445
    },
    {
      "epoch": 5.1927566528320315e-06,
      "step": 79445,
      "training_step_time": 7.323880195617676
    },
    {
      "epoch": 5.193710327148437e-06,
      "model_forward_time": 1.8042049407958984,
      "step": 79446
    },
    {
      "epoch": 5.193710327148437e-06,
      "step": 79446,
      "training_step_time": 6.738477468490601
    },
    {
      "epoch": 5.194664001464844e-06,
      "model_forward_time": 1.8013150691986084,
      "step": 79447
    },
    {
      "epoch": 5.194664001464844e-06,
      "step": 79447,
      "training_step_time": 6.754980802536011
    },
    {
      "epoch": 5.19561767578125e-06,
      "model_forward_time": 1.808656930923462,
      "step": 79448
    },
    {
      "epoch": 5.19561767578125e-06,
      "step": 79448,
      "training_step_time": 6.666894435882568
    },
    {
      "epoch": 5.196571350097656e-06,
      "model_forward_time": 2.9004225730895996,
      "step": 79449
    },
    {
      "epoch": 5.196571350097656e-06,
      "step": 79449,
      "training_step_time": 8.454851865768433
    },
    {
      "epoch": 5.197525024414063e-06,
      "grad_norm": 0.09773878753185272,
      "learning_rate": 1.110907929918877e-06,
      "loss": 0.0555,
      "step": 79450
    },
    {
      "epoch": 5.197525024414063e-06,
      "model_forward_time": 1.7946059703826904,
      "step": 79450
    },
    {
      "dynamics_loss_avg": 0.05502491146326065,
      "epoch": 5.197525024414063e-06,
      "step": 79450
    },
    {
      "action_loss_avg": 0.0013844767876435072,
      "epoch": 5.197525024414063e-06,
      "step": 79450
    },
    {
      "epoch": 5.197525024414063e-06,
      "step": 79450,
      "training_step_time": 8.548929691314697
    },
    {
      "epoch": 5.198478698730468e-06,
      "model_forward_time": 1.8081111907958984,
      "step": 79451
    },
    {
      "epoch": 5.198478698730468e-06,
      "step": 79451,
      "training_step_time": 8.70976710319519
    },
    {
      "epoch": 5.199432373046875e-06,
      "model_forward_time": 1.801654577255249,
      "step": 79452
    },
    {
      "epoch": 5.199432373046875e-06,
      "step": 79452,
      "training_step_time": 7.623610258102417
    },
    {
      "epoch": 5.2003860473632815e-06,
      "model_forward_time": 1.8047170639038086,
      "step": 79453
    },
    {
      "epoch": 5.2003860473632815e-06,
      "step": 79453,
      "training_step_time": 7.581671714782715
    },
    {
      "epoch": 5.201339721679687e-06,
      "model_forward_time": 1.7710366249084473,
      "step": 79454
    },
    {
      "epoch": 5.201339721679687e-06,
      "step": 79454,
      "training_step_time": 7.187741756439209
    },
    {
      "epoch": 5.202293395996094e-06,
      "model_forward_time": 1.8078083992004395,
      "step": 79455
    },
    {
      "epoch": 5.202293395996094e-06,
      "step": 79455,
      "training_step_time": 6.835479021072388
    },
    {
      "epoch": 5.2032470703125e-06,
      "model_forward_time": 1.7873456478118896,
      "step": 79456
    },
    {
      "epoch": 5.2032470703125e-06,
      "step": 79456,
      "training_step_time": 6.705448389053345
    },
    {
      "epoch": 5.204200744628906e-06,
      "model_forward_time": 1.776838779449463,
      "step": 79457
    },
    {
      "epoch": 5.204200744628906e-06,
      "step": 79457,
      "training_step_time": 6.686668872833252
    },
    {
      "epoch": 5.205154418945313e-06,
      "model_forward_time": 2.974745273590088,
      "step": 79458
    },
    {
      "epoch": 5.205154418945313e-06,
      "step": 79458,
      "training_step_time": 7.785515069961548
    },
    {
      "epoch": 5.206108093261718e-06,
      "model_forward_time": 1.8098328113555908,
      "step": 79459
    },
    {
      "epoch": 5.206108093261718e-06,
      "step": 79459,
      "training_step_time": 9.643322944641113
    },
    {
      "epoch": 5.207061767578125e-06,
      "grad_norm": 0.09436481446027756,
      "learning_rate": 1.1098689548775848e-06,
      "loss": 0.0566,
      "step": 79460
    },
    {
      "epoch": 5.207061767578125e-06,
      "model_forward_time": 1.7923195362091064,
      "step": 79460
    },
    {
      "dynamics_loss_avg": 0.05542600471526384,
      "epoch": 5.207061767578125e-06,
      "step": 79460
    },
    {
      "action_loss_avg": 0.0009687707497505471,
      "epoch": 5.207061767578125e-06,
      "step": 79460
    },
    {
      "epoch": 5.207061767578125e-06,
      "step": 79460,
      "training_step_time": 7.729122161865234
    },
    {
      "epoch": 5.2080154418945315e-06,
      "model_forward_time": 1.808725357055664,
      "step": 79461
    },
    {
      "epoch": 5.2080154418945315e-06,
      "step": 79461,
      "training_step_time": 7.641217470169067
    },
    {
      "epoch": 5.208969116210937e-06,
      "model_forward_time": 1.80759859085083,
      "step": 79462
    },
    {
      "epoch": 5.208969116210937e-06,
      "step": 79462,
      "training_step_time": 7.58893609046936
    },
    {
      "epoch": 5.209922790527344e-06,
      "model_forward_time": 1.8188257217407227,
      "step": 79463
    },
    {
      "epoch": 5.209922790527344e-06,
      "step": 79463,
      "training_step_time": 8.174416303634644
    },
    {
      "epoch": 5.21087646484375e-06,
      "model_forward_time": 1.8026764392852783,
      "step": 79464
    },
    {
      "epoch": 5.21087646484375e-06,
      "step": 79464,
      "training_step_time": 6.807736873626709
    },
    {
      "epoch": 5.211830139160156e-06,
      "model_forward_time": 1.8100037574768066,
      "step": 79465
    },
    {
      "epoch": 5.211830139160156e-06,
      "step": 79465,
      "training_step_time": 6.7216503620147705
    },
    {
      "epoch": 5.212783813476563e-06,
      "model_forward_time": 1.8035242557525635,
      "step": 79466
    },
    {
      "epoch": 5.212783813476563e-06,
      "step": 79466,
      "training_step_time": 6.682829856872559
    },
    {
      "epoch": 5.213737487792968e-06,
      "model_forward_time": 2.9077327251434326,
      "step": 79467
    },
    {
      "epoch": 5.213737487792968e-06,
      "step": 79467,
      "training_step_time": 7.645715236663818
    },
    {
      "epoch": 5.214691162109375e-06,
      "model_forward_time": 1.793414831161499,
      "step": 79468
    },
    {
      "epoch": 5.214691162109375e-06,
      "step": 79468,
      "training_step_time": 10.59544849395752
    },
    {
      "epoch": 5.2156448364257815e-06,
      "model_forward_time": 1.803349494934082,
      "step": 79469
    },
    {
      "epoch": 5.2156448364257815e-06,
      "step": 79469,
      "training_step_time": 7.679981231689453
    },
    {
      "epoch": 5.216598510742187e-06,
      "grad_norm": 0.087913878262043,
      "learning_rate": 1.1088304052551762e-06,
      "loss": 0.0569,
      "step": 79470
    },
    {
      "epoch": 5.216598510742187e-06,
      "model_forward_time": 1.7868223190307617,
      "step": 79470
    },
    {
      "dynamics_loss_avg": 0.059118013456463815,
      "epoch": 5.216598510742187e-06,
      "step": 79470
    },
    {
      "action_loss_avg": 0.0015384657424874603,
      "epoch": 5.216598510742187e-06,
      "step": 79470
    },
    {
      "epoch": 5.216598510742187e-06,
      "step": 79470,
      "training_step_time": 8.531618356704712
    },
    {
      "epoch": 5.217552185058594e-06,
      "model_forward_time": 1.8096058368682861,
      "step": 79471
    },
    {
      "epoch": 5.217552185058594e-06,
      "step": 79471,
      "training_step_time": 7.582171201705933
    },
    {
      "epoch": 5.218505859375e-06,
      "model_forward_time": 1.815340518951416,
      "step": 79472
    },
    {
      "epoch": 5.218505859375e-06,
      "step": 79472,
      "training_step_time": 7.558361053466797
    },
    {
      "epoch": 5.219459533691406e-06,
      "model_forward_time": 1.8165593147277832,
      "step": 79473
    },
    {
      "epoch": 5.219459533691406e-06,
      "step": 79473,
      "training_step_time": 7.115177392959595
    },
    {
      "epoch": 5.220413208007813e-06,
      "model_forward_time": 1.785459041595459,
      "step": 79474
    },
    {
      "epoch": 5.220413208007813e-06,
      "step": 79474,
      "training_step_time": 6.732154607772827
    },
    {
      "epoch": 5.221366882324218e-06,
      "model_forward_time": 1.8044538497924805,
      "step": 79475
    },
    {
      "epoch": 5.221366882324218e-06,
      "step": 79475,
      "training_step_time": 6.6974568367004395
    },
    {
      "epoch": 5.222320556640625e-06,
      "model_forward_time": 2.9019386768341064,
      "step": 79476
    },
    {
      "epoch": 5.222320556640625e-06,
      "step": 79476,
      "training_step_time": 7.756318807601929
    },
    {
      "epoch": 5.2232742309570315e-06,
      "model_forward_time": 1.8044068813323975,
      "step": 79477
    },
    {
      "epoch": 5.2232742309570315e-06,
      "step": 79477,
      "training_step_time": 8.512229681015015
    },
    {
      "epoch": 5.224227905273437e-06,
      "model_forward_time": 1.7876036167144775,
      "step": 79478
    },
    {
      "epoch": 5.224227905273437e-06,
      "step": 79478,
      "training_step_time": 8.213014364242554
    },
    {
      "epoch": 5.225181579589844e-06,
      "model_forward_time": 1.8100440502166748,
      "step": 79479
    },
    {
      "epoch": 5.225181579589844e-06,
      "step": 79479,
      "training_step_time": 7.572187185287476
    },
    {
      "epoch": 5.22613525390625e-06,
      "grad_norm": 0.0908816009759903,
      "learning_rate": 1.1077922811652215e-06,
      "loss": 0.0577,
      "step": 79480
    },
    {
      "epoch": 5.22613525390625e-06,
      "model_forward_time": 1.8089025020599365,
      "step": 79480
    },
    {
      "dynamics_loss_avg": 0.04032346135936678,
      "epoch": 5.22613525390625e-06,
      "step": 79480
    },
    {
      "action_loss_avg": 0.001257199255633168,
      "epoch": 5.22613525390625e-06,
      "step": 79480
    },
    {
      "epoch": 5.22613525390625e-06,
      "step": 79480,
      "training_step_time": 7.6042115688323975
    },
    {
      "epoch": 5.227088928222656e-06,
      "model_forward_time": 1.7979938983917236,
      "step": 79481
    },
    {
      "epoch": 5.227088928222656e-06,
      "step": 79481,
      "training_step_time": 7.5426552295684814
    },
    {
      "epoch": 5.228042602539063e-06,
      "model_forward_time": 1.802091360092163,
      "step": 79482
    },
    {
      "epoch": 5.228042602539063e-06,
      "step": 79482,
      "training_step_time": 6.614877939224243
    },
    {
      "epoch": 5.228996276855468e-06,
      "model_forward_time": 1.8061771392822266,
      "step": 79483
    },
    {
      "epoch": 5.228996276855468e-06,
      "step": 79483,
      "training_step_time": 7.045389175415039
    },
    {
      "epoch": 5.229949951171875e-06,
      "model_forward_time": 1.447188138961792,
      "step": 79484
    },
    {
      "epoch": 5.229949951171875e-06,
      "step": 79484,
      "training_step_time": 6.660214424133301
    },
    {
      "epoch": 5.2309036254882816e-06,
      "model_forward_time": 1.7987160682678223,
      "step": 79485
    },
    {
      "epoch": 5.2309036254882816e-06,
      "step": 79485,
      "training_step_time": 7.560400724411011
    },
    {
      "epoch": 5.231857299804687e-06,
      "model_forward_time": 2.8469948768615723,
      "step": 79486
    },
    {
      "epoch": 5.231857299804687e-06,
      "step": 79486,
      "training_step_time": 8.140302181243896
    },
    {
      "epoch": 5.232810974121094e-06,
      "model_forward_time": 1.7905290126800537,
      "step": 79487
    },
    {
      "epoch": 5.232810974121094e-06,
      "step": 79487,
      "training_step_time": 9.328532457351685
    },
    {
      "epoch": 5.2337646484375e-06,
      "model_forward_time": 1.4478230476379395,
      "step": 79488
    },
    {
      "epoch": 5.2337646484375e-06,
      "step": 79488,
      "training_step_time": 7.604379653930664
    },
    {
      "epoch": 5.234718322753906e-06,
      "model_forward_time": 1.789856195449829,
      "step": 79489
    },
    {
      "epoch": 5.234718322753906e-06,
      "step": 79489,
      "training_step_time": 8.164963006973267
    },
    {
      "epoch": 5.235671997070313e-06,
      "grad_norm": 0.07463110983371735,
      "learning_rate": 1.1067545827212534e-06,
      "loss": 0.0565,
      "step": 79490
    },
    {
      "epoch": 5.235671997070313e-06,
      "model_forward_time": 1.7788288593292236,
      "step": 79490
    },
    {
      "dynamics_loss_avg": 0.04708126336336136,
      "epoch": 5.235671997070313e-06,
      "step": 79490
    },
    {
      "action_loss_avg": 0.0011557255493244156,
      "epoch": 5.235671997070313e-06,
      "step": 79490
    },
    {
      "epoch": 5.235671997070313e-06,
      "step": 79490,
      "training_step_time": 7.556800603866577
    },
    {
      "epoch": 5.2366256713867184e-06,
      "model_forward_time": 1.7327873706817627,
      "step": 79491
    },
    {
      "epoch": 5.2366256713867184e-06,
      "step": 79491,
      "training_step_time": 7.217311859130859
    },
    {
      "epoch": 5.237579345703125e-06,
      "model_forward_time": 1.752608060836792,
      "step": 79492
    },
    {
      "epoch": 5.237579345703125e-06,
      "step": 79492,
      "training_step_time": 7.060000658035278
    },
    {
      "epoch": 5.2385330200195316e-06,
      "model_forward_time": 1.742849588394165,
      "step": 79493
    },
    {
      "epoch": 5.2385330200195316e-06,
      "step": 79493,
      "training_step_time": 6.612206697463989
    },
    {
      "epoch": 5.239486694335937e-06,
      "model_forward_time": 1.7464039325714111,
      "step": 79494
    },
    {
      "epoch": 5.239486694335937e-06,
      "step": 79494,
      "training_step_time": 7.199843645095825
    },
    {
      "epoch": 5.240440368652344e-06,
      "model_forward_time": 2.419466972351074,
      "step": 79495
    },
    {
      "epoch": 5.240440368652344e-06,
      "step": 79495,
      "training_step_time": 8.634120464324951
    },
    {
      "epoch": 5.24139404296875e-06,
      "model_forward_time": 1.753070592880249,
      "step": 79496
    },
    {
      "epoch": 5.24139404296875e-06,
      "step": 79496,
      "training_step_time": 8.504909992218018
    },
    {
      "epoch": 5.242347717285156e-06,
      "model_forward_time": 1.7541933059692383,
      "step": 79497
    },
    {
      "epoch": 5.242347717285156e-06,
      "step": 79497,
      "training_step_time": 7.665485143661499
    },
    {
      "epoch": 5.243301391601563e-06,
      "model_forward_time": 1.7474515438079834,
      "step": 79498
    },
    {
      "epoch": 5.243301391601563e-06,
      "step": 79498,
      "training_step_time": 7.688226699829102
    },
    {
      "epoch": 5.2442550659179684e-06,
      "model_forward_time": 1.7398457527160645,
      "step": 79499
    },
    {
      "epoch": 5.2442550659179684e-06,
      "step": 79499,
      "training_step_time": 7.630754470825195
    },
    {
      "epoch": 5.245208740234375e-06,
      "grad_norm": 0.06898784637451172,
      "learning_rate": 1.1057173100367495e-06,
      "loss": 0.057,
      "step": 79500
    },
    {
      "epoch": 5.245208740234375e-06,
      "model_forward_time": 1.7493770122528076,
      "step": 79500
    },
    {
      "dynamics_loss_avg": 0.04989135740324855,
      "epoch": 5.245208740234375e-06,
      "step": 79500
    },
    {
      "action_loss_avg": 0.0014373183454154058,
      "epoch": 5.245208740234375e-06,
      "step": 79500
    },
    {
      "epoch": 5.245208740234375e-06,
      "step": 79500,
      "training_step_time": 7.2861292362213135
    },
    {
      "epoch": 5.2461624145507816e-06,
      "model_forward_time": 1.7416434288024902,
      "step": 79501
    },
    {
      "epoch": 5.2461624145507816e-06,
      "step": 79501,
      "training_step_time": 7.146068096160889
    },
    {
      "epoch": 5.247116088867187e-06,
      "model_forward_time": 1.8074827194213867,
      "step": 79502
    },
    {
      "epoch": 5.247116088867187e-06,
      "step": 79502,
      "training_step_time": 6.72909951210022
    },
    {
      "epoch": 5.248069763183594e-06,
      "model_forward_time": 1.7434091567993164,
      "step": 79503
    },
    {
      "epoch": 5.248069763183594e-06,
      "step": 79503,
      "training_step_time": 7.431447982788086
    },
    {
      "epoch": 5.2490234375e-06,
      "model_forward_time": 2.4081194400787354,
      "step": 79504
    },
    {
      "epoch": 5.2490234375e-06,
      "step": 79504,
      "training_step_time": 8.476824045181274
    },
    {
      "epoch": 5.249977111816406e-06,
      "model_forward_time": 1.846428632736206,
      "step": 79505
    },
    {
      "epoch": 5.249977111816406e-06,
      "step": 79505,
      "training_step_time": 9.837644338607788
    },
    {
      "epoch": 5.250930786132813e-06,
      "model_forward_time": 1.7595863342285156,
      "step": 79506
    },
    {
      "epoch": 5.250930786132813e-06,
      "step": 79506,
      "training_step_time": 9.8981454372406
    },
    {
      "epoch": 5.2518844604492184e-06,
      "model_forward_time": 1.7551186084747314,
      "step": 79507
    },
    {
      "epoch": 5.2518844604492184e-06,
      "step": 79507,
      "training_step_time": 7.66968560218811
    },
    {
      "epoch": 5.252838134765625e-06,
      "model_forward_time": 1.755002498626709,
      "step": 79508
    },
    {
      "epoch": 5.252838134765625e-06,
      "step": 79508,
      "training_step_time": 7.587006568908691
    },
    {
      "epoch": 5.2537918090820316e-06,
      "model_forward_time": 1.758739709854126,
      "step": 79509
    },
    {
      "epoch": 5.2537918090820316e-06,
      "step": 79509,
      "training_step_time": 8.286915302276611
    },
    {
      "epoch": 5.254745483398437e-06,
      "grad_norm": 0.09368132799863815,
      "learning_rate": 1.1046804632251472e-06,
      "loss": 0.0573,
      "step": 79510
    },
    {
      "epoch": 5.254745483398437e-06,
      "model_forward_time": 1.7327699661254883,
      "step": 79510
    },
    {
      "dynamics_loss_avg": 0.057435692846775056,
      "epoch": 5.254745483398437e-06,
      "step": 79510
    },
    {
      "action_loss_avg": 0.0011584355728700758,
      "epoch": 5.254745483398437e-06,
      "step": 79510
    },
    {
      "epoch": 5.254745483398437e-06,
      "step": 79510,
      "training_step_time": 7.14674711227417
    },
    {
      "epoch": 5.255699157714844e-06,
      "model_forward_time": 1.756697416305542,
      "step": 79511
    },
    {
      "epoch": 5.255699157714844e-06,
      "step": 79511,
      "training_step_time": 6.720073699951172
    },
    {
      "epoch": 5.25665283203125e-06,
      "model_forward_time": 1.7291338443756104,
      "step": 79512
    },
    {
      "epoch": 5.25665283203125e-06,
      "step": 79512,
      "training_step_time": 6.700080156326294
    },
    {
      "epoch": 5.257606506347656e-06,
      "model_forward_time": 2.373624086380005,
      "step": 79513
    },
    {
      "epoch": 5.257606506347656e-06,
      "step": 79513,
      "training_step_time": 7.666309833526611
    },
    {
      "epoch": 5.258560180664063e-06,
      "model_forward_time": 1.748504638671875,
      "step": 79514
    },
    {
      "epoch": 5.258560180664063e-06,
      "step": 79514,
      "training_step_time": 9.268611431121826
    },
    {
      "epoch": 5.2595138549804684e-06,
      "model_forward_time": 1.3538398742675781,
      "step": 79515
    },
    {
      "epoch": 5.2595138549804684e-06,
      "step": 79515,
      "training_step_time": 9.485382080078125
    },
    {
      "epoch": 5.260467529296875e-06,
      "model_forward_time": 1.7326719760894775,
      "step": 79516
    },
    {
      "epoch": 5.260467529296875e-06,
      "step": 79516,
      "training_step_time": 8.629636526107788
    },
    {
      "epoch": 5.261421203613282e-06,
      "model_forward_time": 1.732945203781128,
      "step": 79517
    },
    {
      "epoch": 5.261421203613282e-06,
      "step": 79517,
      "training_step_time": 7.632852792739868
    },
    {
      "epoch": 5.262374877929687e-06,
      "model_forward_time": 1.7374818325042725,
      "step": 79518
    },
    {
      "epoch": 5.262374877929687e-06,
      "step": 79518,
      "training_step_time": 7.548915147781372
    },
    {
      "epoch": 5.263328552246094e-06,
      "model_forward_time": 1.729604959487915,
      "step": 79519
    },
    {
      "epoch": 5.263328552246094e-06,
      "step": 79519,
      "training_step_time": 7.19998025894165
    },
    {
      "epoch": 5.2642822265625e-06,
      "grad_norm": 0.08178652077913284,
      "learning_rate": 1.1036440423998317e-06,
      "loss": 0.0557,
      "step": 79520
    },
    {
      "epoch": 5.2642822265625e-06,
      "model_forward_time": 1.724336862564087,
      "step": 79520
    },
    {
      "dynamics_loss_avg": 0.045280213840305804,
      "epoch": 5.2642822265625e-06,
      "step": 79520
    },
    {
      "action_loss_avg": 0.0009453101112740115,
      "epoch": 5.2642822265625e-06,
      "step": 79520
    },
    {
      "epoch": 5.2642822265625e-06,
      "step": 79520,
      "training_step_time": 6.648727178573608
    },
    {
      "epoch": 5.265235900878906e-06,
      "model_forward_time": 1.7294151782989502,
      "step": 79521
    },
    {
      "epoch": 5.265235900878906e-06,
      "step": 79521,
      "training_step_time": 6.646944522857666
    },
    {
      "epoch": 5.266189575195313e-06,
      "model_forward_time": 1.7288212776184082,
      "step": 79522
    },
    {
      "epoch": 5.266189575195313e-06,
      "step": 79522,
      "training_step_time": 8.605870246887207
    },
    {
      "epoch": 5.2671432495117185e-06,
      "model_forward_time": 1.9925565719604492,
      "step": 79523
    },
    {
      "epoch": 5.2671432495117185e-06,
      "step": 79523,
      "training_step_time": 10.940587759017944
    },
    {
      "epoch": 5.268096923828125e-06,
      "model_forward_time": 1.734875202178955,
      "step": 79524
    },
    {
      "epoch": 5.268096923828125e-06,
      "step": 79524,
      "training_step_time": 8.742838144302368
    },
    {
      "epoch": 5.269050598144532e-06,
      "model_forward_time": 1.7244222164154053,
      "step": 79525
    },
    {
      "epoch": 5.269050598144532e-06,
      "step": 79525,
      "training_step_time": 8.286410093307495
    },
    {
      "epoch": 5.270004272460937e-06,
      "model_forward_time": 1.7283308506011963,
      "step": 79526
    },
    {
      "epoch": 5.270004272460937e-06,
      "step": 79526,
      "training_step_time": 7.617197275161743
    },
    {
      "epoch": 5.270957946777344e-06,
      "model_forward_time": 1.7371644973754883,
      "step": 79527
    },
    {
      "epoch": 5.270957946777344e-06,
      "step": 79527,
      "training_step_time": 7.507763624191284
    },
    {
      "epoch": 5.27191162109375e-06,
      "model_forward_time": 1.7278153896331787,
      "step": 79528
    },
    {
      "epoch": 5.27191162109375e-06,
      "step": 79528,
      "training_step_time": 7.1725077629089355
    },
    {
      "epoch": 5.272865295410156e-06,
      "model_forward_time": 1.7246370315551758,
      "step": 79529
    },
    {
      "epoch": 5.272865295410156e-06,
      "step": 79529,
      "training_step_time": 7.109197616577148
    },
    {
      "epoch": 5.273818969726563e-06,
      "grad_norm": 0.1000613197684288,
      "learning_rate": 1.1026080476741469e-06,
      "loss": 0.0566,
      "step": 79530
    },
    {
      "epoch": 5.273818969726563e-06,
      "model_forward_time": 1.7354762554168701,
      "step": 79530
    },
    {
      "dynamics_loss_avg": 0.0628737412393093,
      "epoch": 5.273818969726563e-06,
      "step": 79530
    },
    {
      "action_loss_avg": 0.0015219192195218057,
      "epoch": 5.273818969726563e-06,
      "step": 79530
    },
    {
      "epoch": 5.273818969726563e-06,
      "step": 79530,
      "training_step_time": 6.743465900421143
    },
    {
      "epoch": 5.2747726440429685e-06,
      "model_forward_time": 1.7297382354736328,
      "step": 79531
    },
    {
      "epoch": 5.2747726440429685e-06,
      "step": 79531,
      "training_step_time": 7.597599506378174
    },
    {
      "epoch": 5.275726318359375e-06,
      "model_forward_time": 2.3781678676605225,
      "step": 79532
    },
    {
      "epoch": 5.275726318359375e-06,
      "step": 79532,
      "training_step_time": 7.619042158126831
    },
    {
      "epoch": 5.276679992675782e-06,
      "model_forward_time": 1.7383909225463867,
      "step": 79533
    },
    {
      "epoch": 5.276679992675782e-06,
      "step": 79533,
      "training_step_time": 8.336873292922974
    },
    {
      "epoch": 5.277633666992187e-06,
      "model_forward_time": 1.7246882915496826,
      "step": 79534
    },
    {
      "epoch": 5.277633666992187e-06,
      "step": 79534,
      "training_step_time": 7.727373361587524
    },
    {
      "epoch": 5.278587341308594e-06,
      "model_forward_time": 1.7507753372192383,
      "step": 79535
    },
    {
      "epoch": 5.278587341308594e-06,
      "step": 79535,
      "training_step_time": 7.680706024169922
    },
    {
      "epoch": 5.279541015625e-06,
      "model_forward_time": 1.725738525390625,
      "step": 79536
    },
    {
      "epoch": 5.279541015625e-06,
      "step": 79536,
      "training_step_time": 7.590510368347168
    },
    {
      "epoch": 5.280494689941406e-06,
      "model_forward_time": 1.7326629161834717,
      "step": 79537
    },
    {
      "epoch": 5.280494689941406e-06,
      "step": 79537,
      "training_step_time": 7.57518744468689
    },
    {
      "epoch": 5.281448364257813e-06,
      "model_forward_time": 1.7338042259216309,
      "step": 79538
    },
    {
      "epoch": 5.281448364257813e-06,
      "step": 79538,
      "training_step_time": 7.158535957336426
    },
    {
      "epoch": 5.2824020385742185e-06,
      "model_forward_time": 1.7398996353149414,
      "step": 79539
    },
    {
      "epoch": 5.2824020385742185e-06,
      "step": 79539,
      "training_step_time": 6.749921798706055
    },
    {
      "epoch": 5.283355712890625e-06,
      "grad_norm": 0.08547331392765045,
      "learning_rate": 1.1015724791613835e-06,
      "loss": 0.0577,
      "step": 79540
    },
    {
      "epoch": 5.283355712890625e-06,
      "model_forward_time": 1.72957444190979,
      "step": 79540
    },
    {
      "dynamics_loss_avg": 0.05683341370895505,
      "epoch": 5.283355712890625e-06,
      "step": 79540
    },
    {
      "action_loss_avg": 0.0014663878930150532,
      "epoch": 5.283355712890625e-06,
      "step": 79540
    },
    {
      "epoch": 5.283355712890625e-06,
      "step": 79540,
      "training_step_time": 7.596889495849609
    },
    {
      "epoch": 5.284309387207032e-06,
      "model_forward_time": 2.3768444061279297,
      "step": 79541
    },
    {
      "epoch": 5.284309387207032e-06,
      "step": 79541,
      "training_step_time": 9.112611532211304
    },
    {
      "epoch": 5.285263061523437e-06,
      "model_forward_time": 1.7292590141296387,
      "step": 79542
    },
    {
      "epoch": 5.285263061523437e-06,
      "step": 79542,
      "training_step_time": 9.378562450408936
    },
    {
      "epoch": 5.286216735839844e-06,
      "model_forward_time": 1.7576837539672852,
      "step": 79543
    },
    {
      "epoch": 5.286216735839844e-06,
      "step": 79543,
      "training_step_time": 8.630347967147827
    },
    {
      "epoch": 5.28717041015625e-06,
      "model_forward_time": 1.7414920330047607,
      "step": 79544
    },
    {
      "epoch": 5.28717041015625e-06,
      "step": 79544,
      "training_step_time": 8.393210887908936
    },
    {
      "epoch": 5.288124084472656e-06,
      "model_forward_time": 1.7327733039855957,
      "step": 79545
    },
    {
      "epoch": 5.288124084472656e-06,
      "step": 79545,
      "training_step_time": 7.640955924987793
    },
    {
      "epoch": 5.289077758789063e-06,
      "model_forward_time": 1.726203441619873,
      "step": 79546
    },
    {
      "epoch": 5.289077758789063e-06,
      "step": 79546,
      "training_step_time": 7.625556945800781
    },
    {
      "epoch": 5.2900314331054685e-06,
      "model_forward_time": 1.7298898696899414,
      "step": 79547
    },
    {
      "epoch": 5.2900314331054685e-06,
      "step": 79547,
      "training_step_time": 7.144866228103638
    },
    {
      "epoch": 5.290985107421875e-06,
      "model_forward_time": 1.7357933521270752,
      "step": 79548
    },
    {
      "epoch": 5.290985107421875e-06,
      "step": 79548,
      "training_step_time": 6.825986862182617
    },
    {
      "epoch": 5.291938781738282e-06,
      "model_forward_time": 1.734499454498291,
      "step": 79549
    },
    {
      "epoch": 5.291938781738282e-06,
      "step": 79549,
      "training_step_time": 8.52905011177063
    },
    {
      "epoch": 5.292892456054687e-06,
      "grad_norm": 0.0899738222360611,
      "learning_rate": 1.1005373369747952e-06,
      "loss": 0.0573,
      "step": 79550
    },
    {
      "epoch": 5.292892456054687e-06,
      "model_forward_time": 2.360903739929199,
      "step": 79550
    },
    {
      "dynamics_loss_avg": 0.06023464687168598,
      "epoch": 5.292892456054687e-06,
      "step": 79550
    },
    {
      "action_loss_avg": 0.00195089805056341,
      "epoch": 5.292892456054687e-06,
      "step": 79550
    },
    {
      "epoch": 5.292892456054687e-06,
      "step": 79550,
      "training_step_time": 8.63142466545105
    },
    {
      "epoch": 5.293846130371094e-06,
      "model_forward_time": 1.7346603870391846,
      "step": 79551
    },
    {
      "epoch": 5.293846130371094e-06,
      "step": 79551,
      "training_step_time": 10.359405755996704
    },
    {
      "epoch": 5.2947998046875e-06,
      "model_forward_time": 1.754739761352539,
      "step": 79552
    },
    {
      "epoch": 5.2947998046875e-06,
      "step": 79552,
      "training_step_time": 8.745627880096436
    },
    {
      "epoch": 5.295753479003906e-06,
      "model_forward_time": 1.73006272315979,
      "step": 79553
    },
    {
      "epoch": 5.295753479003906e-06,
      "step": 79553,
      "training_step_time": 7.5967888832092285
    },
    {
      "epoch": 5.296707153320313e-06,
      "model_forward_time": 1.7308313846588135,
      "step": 79554
    },
    {
      "epoch": 5.296707153320313e-06,
      "step": 79554,
      "training_step_time": 7.576969861984253
    },
    {
      "epoch": 5.2976608276367185e-06,
      "model_forward_time": 1.3558180332183838,
      "step": 79555
    },
    {
      "epoch": 5.2976608276367185e-06,
      "step": 79555,
      "training_step_time": 7.544445037841797
    },
    {
      "epoch": 5.298614501953125e-06,
      "model_forward_time": 1.742053508758545,
      "step": 79556
    },
    {
      "epoch": 5.298614501953125e-06,
      "step": 79556,
      "training_step_time": 6.730258464813232
    },
    {
      "epoch": 5.299568176269532e-06,
      "model_forward_time": 1.7806510925292969,
      "step": 79557
    },
    {
      "epoch": 5.299568176269532e-06,
      "step": 79557,
      "training_step_time": 6.7934792041778564
    },
    {
      "epoch": 5.300521850585937e-06,
      "model_forward_time": 1.8062233924865723,
      "step": 79558
    },
    {
      "epoch": 5.300521850585937e-06,
      "step": 79558,
      "training_step_time": 7.3826470375061035
    },
    {
      "epoch": 5.301475524902344e-06,
      "model_forward_time": 1.819018840789795,
      "step": 79559
    },
    {
      "epoch": 5.301475524902344e-06,
      "step": 79559,
      "training_step_time": 8.726133108139038
    },
    {
      "epoch": 5.30242919921875e-06,
      "grad_norm": 0.08774694800376892,
      "learning_rate": 1.09950262122758e-06,
      "loss": 0.0557,
      "step": 79560
    },
    {
      "epoch": 5.30242919921875e-06,
      "model_forward_time": 2.9152939319610596,
      "step": 79560
    },
    {
      "dynamics_loss_avg": 0.05836743153631687,
      "epoch": 5.30242919921875e-06,
      "step": 79560
    },
    {
      "action_loss_avg": 0.001496087241685018,
      "epoch": 5.30242919921875e-06,
      "step": 79560
    },
    {
      "epoch": 5.30242919921875e-06,
      "step": 79560,
      "training_step_time": 9.800015211105347
    },
    {
      "epoch": 5.303382873535156e-06,
      "model_forward_time": 1.8143954277038574,
      "step": 79561
    },
    {
      "epoch": 5.303382873535156e-06,
      "step": 79561,
      "training_step_time": 10.17067575454712
    },
    {
      "epoch": 5.304336547851563e-06,
      "model_forward_time": 1.799344778060913,
      "step": 79562
    },
    {
      "epoch": 5.304336547851563e-06,
      "step": 79562,
      "training_step_time": 8.224255561828613
    },
    {
      "epoch": 5.3052902221679685e-06,
      "model_forward_time": 1.8031272888183594,
      "step": 79563
    },
    {
      "epoch": 5.3052902221679685e-06,
      "step": 79563,
      "training_step_time": 7.513830184936523
    },
    {
      "epoch": 5.306243896484375e-06,
      "model_forward_time": 1.8211123943328857,
      "step": 79564
    },
    {
      "epoch": 5.306243896484375e-06,
      "step": 79564,
      "training_step_time": 7.602159023284912
    },
    {
      "epoch": 5.307197570800782e-06,
      "model_forward_time": 1.800337791442871,
      "step": 79565
    },
    {
      "epoch": 5.307197570800782e-06,
      "step": 79565,
      "training_step_time": 7.20284366607666
    },
    {
      "epoch": 5.308151245117187e-06,
      "model_forward_time": 1.8121654987335205,
      "step": 79566
    },
    {
      "epoch": 5.308151245117187e-06,
      "step": 79566,
      "training_step_time": 7.1641011238098145
    },
    {
      "epoch": 5.309104919433594e-06,
      "model_forward_time": 1.810727596282959,
      "step": 79567
    },
    {
      "epoch": 5.309104919433594e-06,
      "step": 79567,
      "training_step_time": 7.078553199768066
    },
    {
      "epoch": 5.31005859375e-06,
      "model_forward_time": 1.8048579692840576,
      "step": 79568
    },
    {
      "epoch": 5.31005859375e-06,
      "step": 79568,
      "training_step_time": 7.796004772186279
    },
    {
      "epoch": 5.311012268066406e-06,
      "model_forward_time": 2.9198050498962402,
      "step": 79569
    },
    {
      "epoch": 5.311012268066406e-06,
      "step": 79569,
      "training_step_time": 9.458285570144653
    },
    {
      "epoch": 5.311965942382813e-06,
      "grad_norm": 0.07979466766119003,
      "learning_rate": 1.0984683320328948e-06,
      "loss": 0.0558,
      "step": 79570
    },
    {
      "epoch": 5.311965942382813e-06,
      "model_forward_time": 1.808004379272461,
      "step": 79570
    },
    {
      "dynamics_loss_avg": 0.049193456023931506,
      "epoch": 5.311965942382813e-06,
      "step": 79570
    },
    {
      "action_loss_avg": 0.0011274815304204821,
      "epoch": 5.311965942382813e-06,
      "step": 79570
    },
    {
      "epoch": 5.311965942382813e-06,
      "step": 79570,
      "training_step_time": 8.701648712158203
    },
    {
      "epoch": 5.3129196166992185e-06,
      "model_forward_time": 1.7943620681762695,
      "step": 79571
    },
    {
      "epoch": 5.3129196166992185e-06,
      "step": 79571,
      "training_step_time": 7.766777515411377
    },
    {
      "epoch": 5.313873291015625e-06,
      "model_forward_time": 1.448737382888794,
      "step": 79572
    },
    {
      "epoch": 5.313873291015625e-06,
      "step": 79572,
      "training_step_time": 7.704489707946777
    },
    {
      "epoch": 5.314826965332032e-06,
      "model_forward_time": 1.8111779689788818,
      "step": 79573
    },
    {
      "epoch": 5.314826965332032e-06,
      "step": 79573,
      "training_step_time": 7.6562159061431885
    },
    {
      "epoch": 5.315780639648437e-06,
      "model_forward_time": 1.7866997718811035,
      "step": 79574
    },
    {
      "epoch": 5.315780639648437e-06,
      "step": 79574,
      "training_step_time": 7.309730291366577
    },
    {
      "epoch": 5.316734313964844e-06,
      "model_forward_time": 1.8009965419769287,
      "step": 79575
    },
    {
      "epoch": 5.316734313964844e-06,
      "step": 79575,
      "training_step_time": 7.169735431671143
    },
    {
      "epoch": 5.31768798828125e-06,
      "model_forward_time": 1.8016059398651123,
      "step": 79576
    },
    {
      "epoch": 5.31768798828125e-06,
      "step": 79576,
      "training_step_time": 7.163213491439819
    },
    {
      "epoch": 5.318641662597656e-06,
      "model_forward_time": 1.8045806884765625,
      "step": 79577
    },
    {
      "epoch": 5.318641662597656e-06,
      "step": 79577,
      "training_step_time": 8.470831394195557
    },
    {
      "epoch": 5.319595336914063e-06,
      "model_forward_time": 2.8469011783599854,
      "step": 79578
    },
    {
      "epoch": 5.319595336914063e-06,
      "step": 79578,
      "training_step_time": 8.196576118469238
    },
    {
      "epoch": 5.3205490112304685e-06,
      "model_forward_time": 1.8057913780212402,
      "step": 79579
    },
    {
      "epoch": 5.3205490112304685e-06,
      "step": 79579,
      "training_step_time": 9.220094442367554
    },
    {
      "epoch": 5.321502685546875e-06,
      "grad_norm": 0.10375075787305832,
      "learning_rate": 1.0974344695038458e-06,
      "loss": 0.0547,
      "step": 79580
    },
    {
      "epoch": 5.321502685546875e-06,
      "model_forward_time": 1.785597562789917,
      "step": 79580
    },
    {
      "dynamics_loss_avg": 0.05446208734065294,
      "epoch": 5.321502685546875e-06,
      "step": 79580
    },
    {
      "action_loss_avg": 0.0011654636589810252,
      "epoch": 5.321502685546875e-06,
      "step": 79580
    },
    {
      "epoch": 5.321502685546875e-06,
      "step": 79580,
      "training_step_time": 8.373648881912231
    },
    {
      "epoch": 5.322456359863282e-06,
      "model_forward_time": 1.7919573783874512,
      "step": 79581
    },
    {
      "epoch": 5.322456359863282e-06,
      "step": 79581,
      "training_step_time": 7.643961191177368
    },
    {
      "epoch": 5.323410034179687e-06,
      "model_forward_time": 1.806246280670166,
      "step": 79582
    },
    {
      "epoch": 5.323410034179687e-06,
      "step": 79582,
      "training_step_time": 7.857795476913452
    },
    {
      "epoch": 5.324363708496094e-06,
      "model_forward_time": 1.8033852577209473,
      "step": 79583
    },
    {
      "epoch": 5.324363708496094e-06,
      "step": 79583,
      "training_step_time": 7.395411968231201
    },
    {
      "epoch": 5.3253173828125e-06,
      "model_forward_time": 1.7985031604766846,
      "step": 79584
    },
    {
      "epoch": 5.3253173828125e-06,
      "step": 79584,
      "training_step_time": 7.203501224517822
    },
    {
      "epoch": 5.326271057128906e-06,
      "model_forward_time": 1.8011853694915771,
      "step": 79585
    },
    {
      "epoch": 5.326271057128906e-06,
      "step": 79585,
      "training_step_time": 7.1472954750061035
    },
    {
      "epoch": 5.327224731445313e-06,
      "model_forward_time": 1.796013593673706,
      "step": 79586
    },
    {
      "epoch": 5.327224731445313e-06,
      "step": 79586,
      "training_step_time": 7.890017509460449
    },
    {
      "epoch": 5.3281784057617185e-06,
      "model_forward_time": 2.855703353881836,
      "step": 79587
    },
    {
      "epoch": 5.3281784057617185e-06,
      "step": 79587,
      "training_step_time": 8.81777310371399
    },
    {
      "epoch": 5.329132080078125e-06,
      "model_forward_time": 1.8016107082366943,
      "step": 79588
    },
    {
      "epoch": 5.329132080078125e-06,
      "step": 79588,
      "training_step_time": 8.43110728263855
    },
    {
      "epoch": 5.330085754394532e-06,
      "model_forward_time": 1.7923023700714111,
      "step": 79589
    },
    {
      "epoch": 5.330085754394532e-06,
      "step": 79589,
      "training_step_time": 7.783600330352783
    },
    {
      "epoch": 5.331039428710937e-06,
      "grad_norm": 0.08664385229349136,
      "learning_rate": 1.0964010337534968e-06,
      "loss": 0.0547,
      "step": 79590
    },
    {
      "epoch": 5.331039428710937e-06,
      "model_forward_time": 1.8064987659454346,
      "step": 79590
    },
    {
      "dynamics_loss_avg": 0.05252054817974568,
      "epoch": 5.331039428710937e-06,
      "step": 79590
    },
    {
      "action_loss_avg": 0.0009436379972612485,
      "epoch": 5.331039428710937e-06,
      "step": 79590
    },
    {
      "epoch": 5.331039428710937e-06,
      "step": 79590,
      "training_step_time": 7.65804123878479
    },
    {
      "epoch": 5.331993103027344e-06,
      "model_forward_time": 1.7928154468536377,
      "step": 79591
    },
    {
      "epoch": 5.331993103027344e-06,
      "step": 79591,
      "training_step_time": 7.906252384185791
    },
    {
      "epoch": 5.33294677734375e-06,
      "model_forward_time": 1.7942633628845215,
      "step": 79592
    },
    {
      "epoch": 5.33294677734375e-06,
      "step": 79592,
      "training_step_time": 8.31838870048523
    },
    {
      "epoch": 5.333900451660156e-06,
      "model_forward_time": 1.810610055923462,
      "step": 79593
    },
    {
      "epoch": 5.333900451660156e-06,
      "step": 79593,
      "training_step_time": 7.149653434753418
    },
    {
      "epoch": 5.334854125976563e-06,
      "model_forward_time": 1.792877197265625,
      "step": 79594
    },
    {
      "epoch": 5.334854125976563e-06,
      "step": 79594,
      "training_step_time": 7.176676988601685
    },
    {
      "epoch": 5.3358078002929685e-06,
      "model_forward_time": 1.804997205734253,
      "step": 79595
    },
    {
      "epoch": 5.3358078002929685e-06,
      "step": 79595,
      "training_step_time": 8.675918340682983
    },
    {
      "epoch": 5.336761474609375e-06,
      "model_forward_time": 1.7859151363372803,
      "step": 79596
    },
    {
      "epoch": 5.336761474609375e-06,
      "step": 79596,
      "training_step_time": 8.121168375015259
    },
    {
      "epoch": 5.337715148925782e-06,
      "model_forward_time": 1.8689215183258057,
      "step": 79597
    },
    {
      "epoch": 5.337715148925782e-06,
      "step": 79597,
      "training_step_time": 8.562453269958496
    },
    {
      "epoch": 5.338668823242187e-06,
      "model_forward_time": 1.7868967056274414,
      "step": 79598
    },
    {
      "epoch": 5.338668823242187e-06,
      "step": 79598,
      "training_step_time": 7.715397119522095
    },
    {
      "epoch": 5.339622497558594e-06,
      "model_forward_time": 1.8015353679656982,
      "step": 79599
    },
    {
      "epoch": 5.339622497558594e-06,
      "step": 79599,
      "training_step_time": 8.284574747085571
    },
    {
      "epoch": 5.340576171875e-06,
      "grad_norm": 0.12000038474798203,
      "learning_rate": 1.095368024894861e-06,
      "loss": 0.0549,
      "step": 79600
    },
    {
      "epoch": 5.340576171875e-06,
      "model_forward_time": 1.8026843070983887,
      "step": 79600
    },
    {
      "dynamics_loss_avg": 0.04682100722566247,
      "epoch": 5.340576171875e-06,
      "step": 79600
    },
    {
      "action_loss_avg": 0.001032576966099441,
      "epoch": 5.340576171875e-06,
      "step": 79600
    },
    {
      "epoch": 5.340576171875e-06,
      "step": 79600,
      "training_step_time": 7.872398138046265
    },
    {
      "epoch": 5.341529846191406e-06,
      "model_forward_time": 1.8118083477020264,
      "step": 79601
    },
    {
      "epoch": 5.341529846191406e-06,
      "step": 79601,
      "training_step_time": 7.593042373657227
    },
    {
      "epoch": 5.342483520507813e-06,
      "model_forward_time": 1.8051667213439941,
      "step": 79602
    },
    {
      "epoch": 5.342483520507813e-06,
      "step": 79602,
      "training_step_time": 7.24920916557312
    },
    {
      "epoch": 5.3434371948242185e-06,
      "model_forward_time": 1.8040554523468018,
      "step": 79603
    },
    {
      "epoch": 5.3434371948242185e-06,
      "step": 79603,
      "training_step_time": 7.12372088432312
    },
    {
      "epoch": 5.344390869140625e-06,
      "model_forward_time": 1.8006041049957275,
      "step": 79604
    },
    {
      "epoch": 5.344390869140625e-06,
      "step": 79604,
      "training_step_time": 7.591052770614624
    },
    {
      "epoch": 5.345344543457032e-06,
      "model_forward_time": 1.794069766998291,
      "step": 79605
    },
    {
      "epoch": 5.345344543457032e-06,
      "step": 79605,
      "training_step_time": 8.888394832611084
    },
    {
      "epoch": 5.346298217773437e-06,
      "model_forward_time": 2.844613552093506,
      "step": 79606
    },
    {
      "epoch": 5.346298217773437e-06,
      "step": 79606,
      "training_step_time": 8.5085129737854
    },
    {
      "epoch": 5.347251892089844e-06,
      "model_forward_time": 1.7960994243621826,
      "step": 79607
    },
    {
      "epoch": 5.347251892089844e-06,
      "step": 79607,
      "training_step_time": 9.221094608306885
    },
    {
      "epoch": 5.34820556640625e-06,
      "model_forward_time": 1.8210952281951904,
      "step": 79608
    },
    {
      "epoch": 5.34820556640625e-06,
      "step": 79608,
      "training_step_time": 8.552843809127808
    },
    {
      "epoch": 5.349159240722656e-06,
      "model_forward_time": 1.8000996112823486,
      "step": 79609
    },
    {
      "epoch": 5.349159240722656e-06,
      "step": 79609,
      "training_step_time": 7.801894903182983
    },
    {
      "epoch": 5.350112915039063e-06,
      "grad_norm": 0.09290959686040878,
      "learning_rate": 1.094335443040907e-06,
      "loss": 0.0531,
      "step": 79610
    },
    {
      "epoch": 5.350112915039063e-06,
      "model_forward_time": 1.793717861175537,
      "step": 79610
    },
    {
      "dynamics_loss_avg": 0.07024589050561189,
      "epoch": 5.350112915039063e-06,
      "step": 79610
    },
    {
      "action_loss_avg": 0.0013086957042105496,
      "epoch": 5.350112915039063e-06,
      "step": 79610
    },
    {
      "epoch": 5.350112915039063e-06,
      "step": 79610,
      "training_step_time": 7.888597249984741
    },
    {
      "epoch": 5.3510665893554685e-06,
      "model_forward_time": 1.8099033832550049,
      "step": 79611
    },
    {
      "epoch": 5.3510665893554685e-06,
      "step": 79611,
      "training_step_time": 7.375004529953003
    },
    {
      "epoch": 5.352020263671875e-06,
      "model_forward_time": 1.8026039600372314,
      "step": 79612
    },
    {
      "epoch": 5.352020263671875e-06,
      "step": 79612,
      "training_step_time": 8.295709609985352
    },
    {
      "epoch": 5.352973937988282e-06,
      "model_forward_time": 1.8241550922393799,
      "step": 79613
    },
    {
      "epoch": 5.352973937988282e-06,
      "step": 79613,
      "training_step_time": 7.442869424819946
    },
    {
      "epoch": 5.353927612304687e-06,
      "model_forward_time": 1.791043758392334,
      "step": 79614
    },
    {
      "epoch": 5.353927612304687e-06,
      "step": 79614,
      "training_step_time": 8.077404975891113
    },
    {
      "epoch": 5.354881286621094e-06,
      "model_forward_time": 2.880056142807007,
      "step": 79615
    },
    {
      "epoch": 5.354881286621094e-06,
      "step": 79615,
      "training_step_time": 7.8134119510650635
    },
    {
      "epoch": 5.3558349609375e-06,
      "model_forward_time": 1.812697410583496,
      "step": 79616
    },
    {
      "epoch": 5.3558349609375e-06,
      "step": 79616,
      "training_step_time": 8.50846266746521
    },
    {
      "epoch": 5.356788635253906e-06,
      "model_forward_time": 1.789245367050171,
      "step": 79617
    },
    {
      "epoch": 5.356788635253906e-06,
      "step": 79617,
      "training_step_time": 7.774622917175293
    },
    {
      "epoch": 5.357742309570313e-06,
      "model_forward_time": 1.801414966583252,
      "step": 79618
    },
    {
      "epoch": 5.357742309570313e-06,
      "step": 79618,
      "training_step_time": 8.23765516281128
    },
    {
      "epoch": 5.3586959838867185e-06,
      "model_forward_time": 1.7967216968536377,
      "step": 79619
    },
    {
      "epoch": 5.3586959838867185e-06,
      "step": 79619,
      "training_step_time": 7.739960193634033
    },
    {
      "epoch": 5.359649658203125e-06,
      "grad_norm": 0.12443523108959198,
      "learning_rate": 1.0933032883045575e-06,
      "loss": 0.0538,
      "step": 79620
    },
    {
      "epoch": 5.359649658203125e-06,
      "model_forward_time": 1.8075015544891357,
      "step": 79620
    },
    {
      "dynamics_loss_avg": 0.04125964445993304,
      "epoch": 5.359649658203125e-06,
      "step": 79620
    },
    {
      "action_loss_avg": 0.0009321661869762466,
      "epoch": 5.359649658203125e-06,
      "step": 79620
    },
    {
      "epoch": 5.359649658203125e-06,
      "step": 79620,
      "training_step_time": 7.652332305908203
    },
    {
      "epoch": 5.360603332519532e-06,
      "model_forward_time": 1.8153927326202393,
      "step": 79621
    },
    {
      "epoch": 5.360603332519532e-06,
      "step": 79621,
      "training_step_time": 7.1590235233306885
    },
    {
      "epoch": 5.361557006835937e-06,
      "model_forward_time": 1.8051118850708008,
      "step": 79622
    },
    {
      "epoch": 5.361557006835937e-06,
      "step": 79622,
      "training_step_time": 7.038263320922852
    },
    {
      "epoch": 5.362510681152344e-06,
      "model_forward_time": 1.7997496128082275,
      "step": 79623
    },
    {
      "epoch": 5.362510681152344e-06,
      "step": 79623,
      "training_step_time": 7.915979623794556
    },
    {
      "epoch": 5.36346435546875e-06,
      "model_forward_time": 2.8785786628723145,
      "step": 79624
    },
    {
      "epoch": 5.36346435546875e-06,
      "step": 79624,
      "training_step_time": 9.255739688873291
    },
    {
      "epoch": 5.364418029785156e-06,
      "model_forward_time": 1.8158247470855713,
      "step": 79625
    },
    {
      "epoch": 5.364418029785156e-06,
      "step": 79625,
      "training_step_time": 11.005195379257202
    },
    {
      "epoch": 5.365371704101563e-06,
      "model_forward_time": 1.797957420349121,
      "step": 79626
    },
    {
      "epoch": 5.365371704101563e-06,
      "step": 79626,
      "training_step_time": 9.212693691253662
    },
    {
      "epoch": 5.3663253784179685e-06,
      "model_forward_time": 1.8135910034179688,
      "step": 79627
    },
    {
      "epoch": 5.3663253784179685e-06,
      "step": 79627,
      "training_step_time": 7.64203143119812
    },
    {
      "epoch": 5.367279052734375e-06,
      "model_forward_time": 1.8077070713043213,
      "step": 79628
    },
    {
      "epoch": 5.367279052734375e-06,
      "step": 79628,
      "training_step_time": 7.711921215057373
    },
    {
      "epoch": 5.368232727050781e-06,
      "model_forward_time": 1.8060574531555176,
      "step": 79629
    },
    {
      "epoch": 5.368232727050781e-06,
      "step": 79629,
      "training_step_time": 7.550482511520386
    },
    {
      "epoch": 5.369186401367187e-06,
      "grad_norm": 0.07217291742563248,
      "learning_rate": 1.0922715607986871e-06,
      "loss": 0.0539,
      "step": 79630
    },
    {
      "epoch": 5.369186401367187e-06,
      "model_forward_time": 1.8078224658966064,
      "step": 79630
    },
    {
      "dynamics_loss_avg": 0.0644446088001132,
      "epoch": 5.369186401367187e-06,
      "step": 79630
    },
    {
      "action_loss_avg": 0.0014194591727573424,
      "epoch": 5.369186401367187e-06,
      "step": 79630
    },
    {
      "epoch": 5.369186401367187e-06,
      "step": 79630,
      "training_step_time": 7.140256881713867
    },
    {
      "epoch": 5.370140075683594e-06,
      "model_forward_time": 1.802185297012329,
      "step": 79631
    },
    {
      "epoch": 5.370140075683594e-06,
      "step": 79631,
      "training_step_time": 7.016314506530762
    },
    {
      "epoch": 5.37109375e-06,
      "model_forward_time": 1.8098015785217285,
      "step": 79632
    },
    {
      "epoch": 5.37109375e-06,
      "step": 79632,
      "training_step_time": 8.432641506195068
    },
    {
      "epoch": 5.372047424316406e-06,
      "model_forward_time": 2.85697340965271,
      "step": 79633
    },
    {
      "epoch": 5.372047424316406e-06,
      "step": 79633,
      "training_step_time": 8.714625120162964
    },
    {
      "epoch": 5.373001098632813e-06,
      "model_forward_time": 1.8037750720977783,
      "step": 79634
    },
    {
      "epoch": 5.373001098632813e-06,
      "step": 79634,
      "training_step_time": 8.722609758377075
    },
    {
      "epoch": 5.3739547729492185e-06,
      "model_forward_time": 1.7821331024169922,
      "step": 79635
    },
    {
      "epoch": 5.3739547729492185e-06,
      "step": 79635,
      "training_step_time": 8.539610147476196
    },
    {
      "epoch": 5.374908447265625e-06,
      "model_forward_time": 1.7968738079071045,
      "step": 79636
    },
    {
      "epoch": 5.374908447265625e-06,
      "step": 79636,
      "training_step_time": 8.69407320022583
    },
    {
      "epoch": 5.375862121582031e-06,
      "model_forward_time": 1.8081769943237305,
      "step": 79637
    },
    {
      "epoch": 5.375862121582031e-06,
      "step": 79637,
      "training_step_time": 8.689297914505005
    },
    {
      "epoch": 5.376815795898437e-06,
      "model_forward_time": 1.7925937175750732,
      "step": 79638
    },
    {
      "epoch": 5.376815795898437e-06,
      "step": 79638,
      "training_step_time": 7.612070560455322
    },
    {
      "epoch": 5.377769470214844e-06,
      "model_forward_time": 1.7979259490966797,
      "step": 79639
    },
    {
      "epoch": 5.377769470214844e-06,
      "step": 79639,
      "training_step_time": 7.102663278579712
    },
    {
      "epoch": 5.37872314453125e-06,
      "grad_norm": 0.1922176033258438,
      "learning_rate": 1.0912402606361234e-06,
      "loss": 0.0545,
      "step": 79640
    },
    {
      "epoch": 5.37872314453125e-06,
      "model_forward_time": 1.8373689651489258,
      "step": 79640
    },
    {
      "dynamics_loss_avg": 0.0408048490062356,
      "epoch": 5.37872314453125e-06,
      "step": 79640
    },
    {
      "action_loss_avg": 0.0014491465757600964,
      "epoch": 5.37872314453125e-06,
      "step": 79640
    },
    {
      "epoch": 5.37872314453125e-06,
      "step": 79640,
      "training_step_time": 7.0946044921875
    },
    {
      "epoch": 5.379676818847656e-06,
      "model_forward_time": 1.8062973022460938,
      "step": 79641
    },
    {
      "epoch": 5.379676818847656e-06,
      "step": 79641,
      "training_step_time": 7.625912666320801
    },
    {
      "epoch": 5.380630493164063e-06,
      "model_forward_time": 2.8794257640838623,
      "step": 79642
    },
    {
      "epoch": 5.380630493164063e-06,
      "step": 79642,
      "training_step_time": 7.841593980789185
    },
    {
      "epoch": 5.3815841674804686e-06,
      "model_forward_time": 1.8453242778778076,
      "step": 79643
    },
    {
      "epoch": 5.3815841674804686e-06,
      "step": 79643,
      "training_step_time": 8.552297115325928
    },
    {
      "epoch": 5.382537841796875e-06,
      "model_forward_time": 1.7999842166900635,
      "step": 79644
    },
    {
      "epoch": 5.382537841796875e-06,
      "step": 79644,
      "training_step_time": 8.261888980865479
    },
    {
      "epoch": 5.383491516113281e-06,
      "model_forward_time": 1.813859462738037,
      "step": 79645
    },
    {
      "epoch": 5.383491516113281e-06,
      "step": 79645,
      "training_step_time": 8.810987949371338
    },
    {
      "epoch": 5.384445190429687e-06,
      "model_forward_time": 1.806086540222168,
      "step": 79646
    },
    {
      "epoch": 5.384445190429687e-06,
      "step": 79646,
      "training_step_time": 8.29041576385498
    },
    {
      "epoch": 5.385398864746094e-06,
      "model_forward_time": 1.8228819370269775,
      "step": 79647
    },
    {
      "epoch": 5.385398864746094e-06,
      "step": 79647,
      "training_step_time": 7.654189348220825
    },
    {
      "epoch": 5.3863525390625e-06,
      "model_forward_time": 1.8036718368530273,
      "step": 79648
    },
    {
      "epoch": 5.3863525390625e-06,
      "step": 79648,
      "training_step_time": 7.308264493942261
    },
    {
      "epoch": 5.387306213378906e-06,
      "model_forward_time": 1.8151895999908447,
      "step": 79649
    },
    {
      "epoch": 5.387306213378906e-06,
      "step": 79649,
      "training_step_time": 7.153268575668335
    },
    {
      "epoch": 5.388259887695313e-06,
      "grad_norm": 0.09009188413619995,
      "learning_rate": 1.090209387929647e-06,
      "loss": 0.0555,
      "step": 79650
    },
    {
      "epoch": 5.388259887695313e-06,
      "model_forward_time": 1.8048255443572998,
      "step": 79650
    },
    {
      "dynamics_loss_avg": 0.0561626797541976,
      "epoch": 5.388259887695313e-06,
      "step": 79650
    },
    {
      "action_loss_avg": 0.001573345295037143,
      "epoch": 5.388259887695313e-06,
      "step": 79650
    },
    {
      "epoch": 5.388259887695313e-06,
      "step": 79650,
      "training_step_time": 7.657835006713867
    },
    {
      "epoch": 5.3892135620117186e-06,
      "model_forward_time": 1.8130943775177002,
      "step": 79651
    },
    {
      "epoch": 5.3892135620117186e-06,
      "step": 79651,
      "training_step_time": 8.449546813964844
    },
    {
      "epoch": 5.390167236328125e-06,
      "model_forward_time": 1.8172039985656738,
      "step": 79652
    },
    {
      "epoch": 5.390167236328125e-06,
      "step": 79652,
      "training_step_time": 7.645021438598633
    },
    {
      "epoch": 5.391120910644531e-06,
      "model_forward_time": 1.7822363376617432,
      "step": 79653
    },
    {
      "epoch": 5.391120910644531e-06,
      "step": 79653,
      "training_step_time": 10.631678819656372
    },
    {
      "epoch": 5.392074584960937e-06,
      "model_forward_time": 1.7971117496490479,
      "step": 79654
    },
    {
      "epoch": 5.392074584960937e-06,
      "step": 79654,
      "training_step_time": 9.108152389526367
    },
    {
      "epoch": 5.393028259277344e-06,
      "model_forward_time": 1.7847836017608643,
      "step": 79655
    },
    {
      "epoch": 5.393028259277344e-06,
      "step": 79655,
      "training_step_time": 8.779817581176758
    },
    {
      "epoch": 5.39398193359375e-06,
      "model_forward_time": 1.7909011840820312,
      "step": 79656
    },
    {
      "epoch": 5.39398193359375e-06,
      "step": 79656,
      "training_step_time": 7.581892728805542
    },
    {
      "epoch": 5.394935607910156e-06,
      "model_forward_time": 1.7906239032745361,
      "step": 79657
    },
    {
      "epoch": 5.394935607910156e-06,
      "step": 79657,
      "training_step_time": 7.296117305755615
    },
    {
      "epoch": 5.395889282226563e-06,
      "model_forward_time": 1.8031742572784424,
      "step": 79658
    },
    {
      "epoch": 5.395889282226563e-06,
      "step": 79658,
      "training_step_time": 7.152142763137817
    },
    {
      "epoch": 5.3968429565429686e-06,
      "model_forward_time": 1.8001306056976318,
      "step": 79659
    },
    {
      "epoch": 5.3968429565429686e-06,
      "step": 79659,
      "training_step_time": 7.490084886550903
    },
    {
      "epoch": 5.397796630859375e-06,
      "grad_norm": 0.17363551259040833,
      "learning_rate": 1.0891789427919959e-06,
      "loss": 0.0554,
      "step": 79660
    },
    {
      "epoch": 5.397796630859375e-06,
      "model_forward_time": 1.7800352573394775,
      "step": 79660
    },
    {
      "dynamics_loss_avg": 0.048617614433169365,
      "epoch": 5.397796630859375e-06,
      "step": 79660
    },
    {
      "action_loss_avg": 0.0013288530462887138,
      "epoch": 5.397796630859375e-06,
      "step": 79660
    },
    {
      "epoch": 5.397796630859375e-06,
      "step": 79660,
      "training_step_time": 8.677565336227417
    },
    {
      "epoch": 5.398750305175781e-06,
      "model_forward_time": 2.927325487136841,
      "step": 79661
    },
    {
      "epoch": 5.398750305175781e-06,
      "step": 79661,
      "training_step_time": 9.475070476531982
    },
    {
      "epoch": 5.3997039794921874e-06,
      "model_forward_time": 1.7836222648620605,
      "step": 79662
    },
    {
      "epoch": 5.3997039794921874e-06,
      "step": 79662,
      "training_step_time": 11.526213884353638
    },
    {
      "epoch": 5.400657653808594e-06,
      "model_forward_time": 1.7916345596313477,
      "step": 79663
    },
    {
      "epoch": 5.400657653808594e-06,
      "step": 79663,
      "training_step_time": 8.844595193862915
    },
    {
      "epoch": 5.401611328125e-06,
      "model_forward_time": 1.775742769241333,
      "step": 79664
    },
    {
      "epoch": 5.401611328125e-06,
      "step": 79664,
      "training_step_time": 7.807222843170166
    },
    {
      "epoch": 5.402565002441406e-06,
      "model_forward_time": 1.7987966537475586,
      "step": 79665
    },
    {
      "epoch": 5.402565002441406e-06,
      "step": 79665,
      "training_step_time": 8.74133586883545
    },
    {
      "epoch": 5.403518676757813e-06,
      "model_forward_time": 1.7910792827606201,
      "step": 79666
    },
    {
      "epoch": 5.403518676757813e-06,
      "step": 79666,
      "training_step_time": 7.824273586273193
    },
    {
      "epoch": 5.4044723510742186e-06,
      "model_forward_time": 1.7924902439117432,
      "step": 79667
    },
    {
      "epoch": 5.4044723510742186e-06,
      "step": 79667,
      "training_step_time": 7.136861324310303
    },
    {
      "epoch": 5.405426025390625e-06,
      "model_forward_time": 1.8011994361877441,
      "step": 79668
    },
    {
      "epoch": 5.405426025390625e-06,
      "step": 79668,
      "training_step_time": 7.484407901763916
    },
    {
      "epoch": 5.406379699707031e-06,
      "model_forward_time": 1.7757060527801514,
      "step": 79669
    },
    {
      "epoch": 5.406379699707031e-06,
      "step": 79669,
      "training_step_time": 8.780239582061768
    },
    {
      "epoch": 5.4073333740234374e-06,
      "grad_norm": 0.10155151784420013,
      "learning_rate": 1.088148925335854e-06,
      "loss": 0.0554,
      "step": 79670
    },
    {
      "epoch": 5.4073333740234374e-06,
      "model_forward_time": 2.938985586166382,
      "step": 79670
    },
    {
      "dynamics_loss_avg": 0.055484073050320146,
      "epoch": 5.4073333740234374e-06,
      "step": 79670
    },
    {
      "action_loss_avg": 0.0014776999887544662,
      "epoch": 5.4073333740234374e-06,
      "step": 79670
    },
    {
      "epoch": 5.4073333740234374e-06,
      "step": 79670,
      "training_step_time": 8.673859119415283
    },
    {
      "epoch": 5.408287048339844e-06,
      "model_forward_time": 1.7919230461120605,
      "step": 79671
    },
    {
      "epoch": 5.408287048339844e-06,
      "step": 79671,
      "training_step_time": 9.406280040740967
    },
    {
      "epoch": 5.40924072265625e-06,
      "model_forward_time": 1.7970860004425049,
      "step": 79672
    },
    {
      "epoch": 5.40924072265625e-06,
      "step": 79672,
      "training_step_time": 8.479209184646606
    },
    {
      "epoch": 5.410194396972656e-06,
      "model_forward_time": 1.8005051612854004,
      "step": 79673
    },
    {
      "epoch": 5.410194396972656e-06,
      "step": 79673,
      "training_step_time": 7.711709260940552
    },
    {
      "epoch": 5.411148071289063e-06,
      "model_forward_time": 1.7971622943878174,
      "step": 79674
    },
    {
      "epoch": 5.411148071289063e-06,
      "step": 79674,
      "training_step_time": 8.778465986251831
    },
    {
      "epoch": 5.412101745605469e-06,
      "model_forward_time": 1.7930958271026611,
      "step": 79675
    },
    {
      "epoch": 5.412101745605469e-06,
      "step": 79675,
      "training_step_time": 7.641936779022217
    },
    {
      "epoch": 5.413055419921875e-06,
      "model_forward_time": 1.805776596069336,
      "step": 79676
    },
    {
      "epoch": 5.413055419921875e-06,
      "step": 79676,
      "training_step_time": 7.278526544570923
    },
    {
      "epoch": 5.414009094238281e-06,
      "model_forward_time": 1.8007843494415283,
      "step": 79677
    },
    {
      "epoch": 5.414009094238281e-06,
      "step": 79677,
      "training_step_time": 7.505715608596802
    },
    {
      "epoch": 5.4149627685546874e-06,
      "model_forward_time": 1.7946887016296387,
      "step": 79678
    },
    {
      "epoch": 5.4149627685546874e-06,
      "step": 79678,
      "training_step_time": 8.78099250793457
    },
    {
      "epoch": 5.415916442871094e-06,
      "model_forward_time": 2.8821420669555664,
      "step": 79679
    },
    {
      "epoch": 5.415916442871094e-06,
      "step": 79679,
      "training_step_time": 8.643329858779907
    },
    {
      "epoch": 5.4168701171875e-06,
      "grad_norm": 0.10350614786148071,
      "learning_rate": 1.0871193356738641e-06,
      "loss": 0.0539,
      "step": 79680
    },
    {
      "epoch": 5.4168701171875e-06,
      "model_forward_time": 1.852616786956787,
      "step": 79680
    },
    {
      "dynamics_loss_avg": 0.053486670926213266,
      "epoch": 5.4168701171875e-06,
      "step": 79680
    },
    {
      "action_loss_avg": 0.0011471570207504556,
      "epoch": 5.4168701171875e-06,
      "step": 79680
    },
    {
      "epoch": 5.4168701171875e-06,
      "step": 79680,
      "training_step_time": 8.12026834487915
    },
    {
      "epoch": 5.417823791503906e-06,
      "model_forward_time": 1.7876782417297363,
      "step": 79681
    },
    {
      "epoch": 5.417823791503906e-06,
      "step": 79681,
      "training_step_time": 7.773037672042847
    },
    {
      "epoch": 5.418777465820313e-06,
      "model_forward_time": 1.793165922164917,
      "step": 79682
    },
    {
      "epoch": 5.418777465820313e-06,
      "step": 79682,
      "training_step_time": 8.35269021987915
    },
    {
      "epoch": 5.419731140136719e-06,
      "model_forward_time": 1.8001563549041748,
      "step": 79683
    },
    {
      "epoch": 5.419731140136719e-06,
      "step": 79683,
      "training_step_time": 7.735549449920654
    },
    {
      "epoch": 5.420684814453125e-06,
      "model_forward_time": 1.7830116748809814,
      "step": 79684
    },
    {
      "epoch": 5.420684814453125e-06,
      "step": 79684,
      "training_step_time": 7.644845485687256
    },
    {
      "epoch": 5.421638488769531e-06,
      "model_forward_time": 1.8027997016906738,
      "step": 79685
    },
    {
      "epoch": 5.421638488769531e-06,
      "step": 79685,
      "training_step_time": 7.31328821182251
    },
    {
      "epoch": 5.4225921630859374e-06,
      "model_forward_time": 1.7827520370483398,
      "step": 79686
    },
    {
      "epoch": 5.4225921630859374e-06,
      "step": 79686,
      "training_step_time": 7.499565839767456
    },
    {
      "epoch": 5.423545837402344e-06,
      "model_forward_time": 1.7856087684631348,
      "step": 79687
    },
    {
      "epoch": 5.423545837402344e-06,
      "step": 79687,
      "training_step_time": 8.689200162887573
    },
    {
      "epoch": 5.42449951171875e-06,
      "model_forward_time": 2.8391659259796143,
      "step": 79688
    },
    {
      "epoch": 5.42449951171875e-06,
      "step": 79688,
      "training_step_time": 8.611459970474243
    },
    {
      "epoch": 5.425453186035156e-06,
      "model_forward_time": 1.4619402885437012,
      "step": 79689
    },
    {
      "epoch": 5.425453186035156e-06,
      "step": 79689,
      "training_step_time": 11.13847017288208
    },
    {
      "epoch": 5.426406860351563e-06,
      "grad_norm": 0.11764764040708542,
      "learning_rate": 1.0860901739186209e-06,
      "loss": 0.0551,
      "step": 79690
    },
    {
      "epoch": 5.426406860351563e-06,
      "model_forward_time": 1.7931926250457764,
      "step": 79690
    },
    {
      "dynamics_loss_avg": 0.044152820110321046,
      "epoch": 5.426406860351563e-06,
      "step": 79690
    },
    {
      "action_loss_avg": 0.0007401397844660096,
      "epoch": 5.426406860351563e-06,
      "step": 79690
    },
    {
      "epoch": 5.426406860351563e-06,
      "step": 79690,
      "training_step_time": 9.23175573348999
    },
    {
      "epoch": 5.427360534667969e-06,
      "model_forward_time": 1.803304672241211,
      "step": 79691
    },
    {
      "epoch": 5.427360534667969e-06,
      "step": 79691,
      "training_step_time": 9.323474884033203
    },
    {
      "epoch": 5.428314208984375e-06,
      "model_forward_time": 1.785001516342163,
      "step": 79692
    },
    {
      "epoch": 5.428314208984375e-06,
      "step": 79692,
      "training_step_time": 7.7795915603637695
    },
    {
      "epoch": 5.429267883300781e-06,
      "model_forward_time": 1.808821201324463,
      "step": 79693
    },
    {
      "epoch": 5.429267883300781e-06,
      "step": 79693,
      "training_step_time": 7.67591404914856
    },
    {
      "epoch": 5.4302215576171875e-06,
      "model_forward_time": 1.7962603569030762,
      "step": 79694
    },
    {
      "epoch": 5.4302215576171875e-06,
      "step": 79694,
      "training_step_time": 7.310270547866821
    },
    {
      "epoch": 5.431175231933594e-06,
      "model_forward_time": 1.7974398136138916,
      "step": 79695
    },
    {
      "epoch": 5.431175231933594e-06,
      "step": 79695,
      "training_step_time": 7.4424004554748535
    },
    {
      "epoch": 5.43212890625e-06,
      "model_forward_time": 1.7815330028533936,
      "step": 79696
    },
    {
      "epoch": 5.43212890625e-06,
      "step": 79696,
      "training_step_time": 7.77912163734436
    },
    {
      "epoch": 5.433082580566406e-06,
      "model_forward_time": 1.763016700744629,
      "step": 79697
    },
    {
      "epoch": 5.433082580566406e-06,
      "step": 79697,
      "training_step_time": 8.459980010986328
    },
    {
      "epoch": 5.434036254882813e-06,
      "model_forward_time": 1.82193922996521,
      "step": 79698
    },
    {
      "epoch": 5.434036254882813e-06,
      "step": 79698,
      "training_step_time": 9.316501379013062
    },
    {
      "epoch": 5.434989929199219e-06,
      "model_forward_time": 1.7723631858825684,
      "step": 79699
    },
    {
      "epoch": 5.434989929199219e-06,
      "step": 79699,
      "training_step_time": 7.817149639129639
    },
    {
      "epoch": 5.435943603515625e-06,
      "grad_norm": 0.10087523609399796,
      "learning_rate": 1.0850614401826731e-06,
      "loss": 0.0553,
      "step": 79700
    },
    {
      "epoch": 5.435943603515625e-06,
      "model_forward_time": 1.7807607650756836,
      "step": 79700
    },
    {
      "dynamics_loss_avg": 0.04557347148656845,
      "epoch": 5.435943603515625e-06,
      "step": 79700
    },
    {
      "action_loss_avg": 0.001464550441596657,
      "epoch": 5.435943603515625e-06,
      "step": 79700
    },
    {
      "epoch": 5.435943603515625e-06,
      "step": 79700,
      "training_step_time": 8.748237609863281
    },
    {
      "epoch": 5.436897277832031e-06,
      "model_forward_time": 1.7897608280181885,
      "step": 79701
    },
    {
      "epoch": 5.436897277832031e-06,
      "step": 79701,
      "training_step_time": 7.704496383666992
    },
    {
      "epoch": 5.4378509521484375e-06,
      "model_forward_time": 1.7818572521209717,
      "step": 79702
    },
    {
      "epoch": 5.4378509521484375e-06,
      "step": 79702,
      "training_step_time": 7.687726736068726
    },
    {
      "epoch": 5.438804626464844e-06,
      "model_forward_time": 1.783540964126587,
      "step": 79703
    },
    {
      "epoch": 5.438804626464844e-06,
      "step": 79703,
      "training_step_time": 7.311892747879028
    },
    {
      "epoch": 5.43975830078125e-06,
      "model_forward_time": 1.7949252128601074,
      "step": 79704
    },
    {
      "epoch": 5.43975830078125e-06,
      "step": 79704,
      "training_step_time": 7.459813117980957
    },
    {
      "epoch": 5.440711975097656e-06,
      "model_forward_time": 1.7851743698120117,
      "step": 79705
    },
    {
      "epoch": 5.440711975097656e-06,
      "step": 79705,
      "training_step_time": 7.474764823913574
    },
    {
      "epoch": 5.441665649414063e-06,
      "model_forward_time": 1.7754366397857666,
      "step": 79706
    },
    {
      "epoch": 5.441665649414063e-06,
      "step": 79706,
      "training_step_time": 7.827349424362183
    },
    {
      "epoch": 5.442619323730469e-06,
      "model_forward_time": 2.8894734382629395,
      "step": 79707
    },
    {
      "epoch": 5.442619323730469e-06,
      "step": 79707,
      "training_step_time": 8.405513048171997
    },
    {
      "epoch": 5.443572998046875e-06,
      "model_forward_time": 1.7724483013153076,
      "step": 79708
    },
    {
      "epoch": 5.443572998046875e-06,
      "step": 79708,
      "training_step_time": 8.673311710357666
    },
    {
      "epoch": 5.444526672363281e-06,
      "model_forward_time": 1.785898208618164,
      "step": 79709
    },
    {
      "epoch": 5.444526672363281e-06,
      "step": 79709,
      "training_step_time": 8.692736148834229
    },
    {
      "epoch": 5.4454803466796875e-06,
      "grad_norm": 0.10767978429794312,
      "learning_rate": 1.0840331345785193e-06,
      "loss": 0.0546,
      "step": 79710
    },
    {
      "epoch": 5.4454803466796875e-06,
      "model_forward_time": 1.783597707748413,
      "step": 79710
    },
    {
      "dynamics_loss_avg": 0.03654338885098696,
      "epoch": 5.4454803466796875e-06,
      "step": 79710
    },
    {
      "action_loss_avg": 0.0008453617949271575,
      "epoch": 5.4454803466796875e-06,
      "step": 79710
    },
    {
      "epoch": 5.4454803466796875e-06,
      "step": 79710,
      "training_step_time": 7.683308839797974
    },
    {
      "epoch": 5.446434020996094e-06,
      "model_forward_time": 1.8000373840332031,
      "step": 79711
    },
    {
      "epoch": 5.446434020996094e-06,
      "step": 79711,
      "training_step_time": 7.7622785568237305
    },
    {
      "epoch": 5.4473876953125e-06,
      "model_forward_time": 1.7915287017822266,
      "step": 79712
    },
    {
      "epoch": 5.4473876953125e-06,
      "step": 79712,
      "training_step_time": 7.605758428573608
    },
    {
      "epoch": 5.448341369628906e-06,
      "model_forward_time": 1.7886791229248047,
      "step": 79713
    },
    {
      "epoch": 5.448341369628906e-06,
      "step": 79713,
      "training_step_time": 7.699991226196289
    },
    {
      "epoch": 5.449295043945313e-06,
      "model_forward_time": 1.7905583381652832,
      "step": 79714
    },
    {
      "epoch": 5.449295043945313e-06,
      "step": 79714,
      "training_step_time": 7.495467185974121
    },
    {
      "epoch": 5.450248718261719e-06,
      "model_forward_time": 1.7839057445526123,
      "step": 79715
    },
    {
      "epoch": 5.450248718261719e-06,
      "step": 79715,
      "training_step_time": 8.531279563903809
    },
    {
      "epoch": 5.451202392578125e-06,
      "model_forward_time": 2.8830926418304443,
      "step": 79716
    },
    {
      "epoch": 5.451202392578125e-06,
      "step": 79716,
      "training_step_time": 8.492524862289429
    },
    {
      "epoch": 5.452156066894531e-06,
      "model_forward_time": 1.8465938568115234,
      "step": 79717
    },
    {
      "epoch": 5.452156066894531e-06,
      "step": 79717,
      "training_step_time": 7.846368074417114
    },
    {
      "epoch": 5.4531097412109375e-06,
      "model_forward_time": 1.7905433177947998,
      "step": 79718
    },
    {
      "epoch": 5.4531097412109375e-06,
      "step": 79718,
      "training_step_time": 10.19859528541565
    },
    {
      "epoch": 5.454063415527344e-06,
      "model_forward_time": 1.7833800315856934,
      "step": 79719
    },
    {
      "epoch": 5.454063415527344e-06,
      "step": 79719,
      "training_step_time": 7.747950553894043
    },
    {
      "epoch": 5.45501708984375e-06,
      "grad_norm": 0.07905023545026779,
      "learning_rate": 1.0830052572186151e-06,
      "loss": 0.0538,
      "step": 79720
    },
    {
      "epoch": 5.45501708984375e-06,
      "model_forward_time": 1.7780663967132568,
      "step": 79720
    },
    {
      "dynamics_loss_avg": 0.05428135879337788,
      "epoch": 5.45501708984375e-06,
      "step": 79720
    },
    {
      "action_loss_avg": 0.0010779710923088715,
      "epoch": 5.45501708984375e-06,
      "step": 79720
    },
    {
      "epoch": 5.45501708984375e-06,
      "step": 79720,
      "training_step_time": 7.843492269515991
    },
    {
      "epoch": 5.455970764160156e-06,
      "model_forward_time": 1.7968523502349854,
      "step": 79721
    },
    {
      "epoch": 5.455970764160156e-06,
      "step": 79721,
      "training_step_time": 7.622145652770996
    },
    {
      "epoch": 5.456924438476563e-06,
      "model_forward_time": 1.7883639335632324,
      "step": 79722
    },
    {
      "epoch": 5.456924438476563e-06,
      "step": 79722,
      "training_step_time": 6.909643650054932
    },
    {
      "epoch": 5.457878112792969e-06,
      "model_forward_time": 1.7954716682434082,
      "step": 79723
    },
    {
      "epoch": 5.457878112792969e-06,
      "step": 79723,
      "training_step_time": 7.505900144577026
    },
    {
      "epoch": 5.458831787109375e-06,
      "model_forward_time": 1.7883706092834473,
      "step": 79724
    },
    {
      "epoch": 5.458831787109375e-06,
      "step": 79724,
      "training_step_time": 8.45291805267334
    },
    {
      "epoch": 5.459785461425781e-06,
      "model_forward_time": 2.844982147216797,
      "step": 79725
    },
    {
      "epoch": 5.459785461425781e-06,
      "step": 79725,
      "training_step_time": 8.788955688476562
    },
    {
      "epoch": 5.4607391357421875e-06,
      "model_forward_time": 1.8458261489868164,
      "step": 79726
    },
    {
      "epoch": 5.4607391357421875e-06,
      "step": 79726,
      "training_step_time": 9.83866834640503
    },
    {
      "epoch": 5.461692810058594e-06,
      "model_forward_time": 1.7863292694091797,
      "step": 79727
    },
    {
      "epoch": 5.461692810058594e-06,
      "step": 79727,
      "training_step_time": 9.69428277015686
    },
    {
      "epoch": 5.462646484375e-06,
      "model_forward_time": 1.8028240203857422,
      "step": 79728
    },
    {
      "epoch": 5.462646484375e-06,
      "step": 79728,
      "training_step_time": 7.778301239013672
    },
    {
      "epoch": 5.463600158691406e-06,
      "model_forward_time": 1.7896249294281006,
      "step": 79729
    },
    {
      "epoch": 5.463600158691406e-06,
      "step": 79729,
      "training_step_time": 7.854517698287964
    },
    {
      "epoch": 5.464553833007813e-06,
      "grad_norm": 0.09257442504167557,
      "learning_rate": 1.0819778082153642e-06,
      "loss": 0.055,
      "step": 79730
    },
    {
      "epoch": 5.464553833007813e-06,
      "model_forward_time": 1.7887840270996094,
      "step": 79730
    },
    {
      "dynamics_loss_avg": 0.04525790885090828,
      "epoch": 5.464553833007813e-06,
      "step": 79730
    },
    {
      "action_loss_avg": 0.001045406819321215,
      "epoch": 5.464553833007813e-06,
      "step": 79730
    },
    {
      "epoch": 5.464553833007813e-06,
      "step": 79730,
      "training_step_time": 7.625569105148315
    },
    {
      "epoch": 5.465507507324219e-06,
      "model_forward_time": 1.8055264949798584,
      "step": 79731
    },
    {
      "epoch": 5.465507507324219e-06,
      "step": 79731,
      "training_step_time": 7.347686529159546
    },
    {
      "epoch": 5.466461181640625e-06,
      "model_forward_time": 1.7957608699798584,
      "step": 79732
    },
    {
      "epoch": 5.466461181640625e-06,
      "step": 79732,
      "training_step_time": 7.487251281738281
    },
    {
      "epoch": 5.467414855957031e-06,
      "model_forward_time": 1.0989465713500977,
      "step": 79733
    },
    {
      "epoch": 5.467414855957031e-06,
      "step": 79733,
      "training_step_time": 8.723647832870483
    },
    {
      "epoch": 5.4683685302734375e-06,
      "model_forward_time": 1.4262659549713135,
      "step": 79734
    },
    {
      "epoch": 5.4683685302734375e-06,
      "step": 79734,
      "training_step_time": 8.787205696105957
    },
    {
      "epoch": 5.469322204589844e-06,
      "model_forward_time": 2.87237548828125,
      "step": 79735
    },
    {
      "epoch": 5.469322204589844e-06,
      "step": 79735,
      "training_step_time": 7.735834360122681
    },
    {
      "epoch": 5.47027587890625e-06,
      "model_forward_time": 1.7788009643554688,
      "step": 79736
    },
    {
      "epoch": 5.47027587890625e-06,
      "step": 79736,
      "training_step_time": 8.63802719116211
    },
    {
      "epoch": 5.471229553222656e-06,
      "model_forward_time": 1.7808411121368408,
      "step": 79737
    },
    {
      "epoch": 5.471229553222656e-06,
      "step": 79737,
      "training_step_time": 8.647796154022217
    },
    {
      "epoch": 5.472183227539063e-06,
      "model_forward_time": 1.7793874740600586,
      "step": 79738
    },
    {
      "epoch": 5.472183227539063e-06,
      "step": 79738,
      "training_step_time": 7.743001461029053
    },
    {
      "epoch": 5.473136901855469e-06,
      "model_forward_time": 1.7869758605957031,
      "step": 79739
    },
    {
      "epoch": 5.473136901855469e-06,
      "step": 79739,
      "training_step_time": 8.608922243118286
    },
    {
      "epoch": 5.474090576171875e-06,
      "grad_norm": 0.153306245803833,
      "learning_rate": 1.0809507876811326e-06,
      "loss": 0.0557,
      "step": 79740
    },
    {
      "epoch": 5.474090576171875e-06,
      "model_forward_time": 1.7844254970550537,
      "step": 79740
    },
    {
      "dynamics_loss_avg": 0.07434055712074042,
      "epoch": 5.474090576171875e-06,
      "step": 79740
    },
    {
      "action_loss_avg": 0.0019498242443660274,
      "epoch": 5.474090576171875e-06,
      "step": 79740
    },
    {
      "epoch": 5.474090576171875e-06,
      "step": 79740,
      "training_step_time": 7.579213619232178
    },
    {
      "epoch": 5.475044250488281e-06,
      "model_forward_time": 1.7882294654846191,
      "step": 79741
    },
    {
      "epoch": 5.475044250488281e-06,
      "step": 79741,
      "training_step_time": 7.564235210418701
    },
    {
      "epoch": 5.4759979248046875e-06,
      "model_forward_time": 1.7891173362731934,
      "step": 79742
    },
    {
      "epoch": 5.4759979248046875e-06,
      "step": 79742,
      "training_step_time": 7.723970413208008
    },
    {
      "epoch": 5.476951599121094e-06,
      "model_forward_time": 1.7784736156463623,
      "step": 79743
    },
    {
      "epoch": 5.476951599121094e-06,
      "step": 79743,
      "training_step_time": 8.684877872467041
    },
    {
      "epoch": 5.4779052734375e-06,
      "model_forward_time": 2.8008060455322266,
      "step": 79744
    },
    {
      "epoch": 5.4779052734375e-06,
      "step": 79744,
      "training_step_time": 7.7265825271606445
    },
    {
      "epoch": 5.478858947753906e-06,
      "model_forward_time": 1.8308851718902588,
      "step": 79745
    },
    {
      "epoch": 5.478858947753906e-06,
      "step": 79745,
      "training_step_time": 7.812842607498169
    },
    {
      "epoch": 5.479812622070313e-06,
      "model_forward_time": 1.7847106456756592,
      "step": 79746
    },
    {
      "epoch": 5.479812622070313e-06,
      "step": 79746,
      "training_step_time": 10.440539360046387
    },
    {
      "epoch": 5.480766296386719e-06,
      "model_forward_time": 1.7793605327606201,
      "step": 79747
    },
    {
      "epoch": 5.480766296386719e-06,
      "step": 79747,
      "training_step_time": 7.799694776535034
    },
    {
      "epoch": 5.481719970703125e-06,
      "model_forward_time": 1.7879748344421387,
      "step": 79748
    },
    {
      "epoch": 5.481719970703125e-06,
      "step": 79748,
      "training_step_time": 8.299634218215942
    },
    {
      "epoch": 5.482673645019531e-06,
      "model_forward_time": 1.7844698429107666,
      "step": 79749
    },
    {
      "epoch": 5.482673645019531e-06,
      "step": 79749,
      "training_step_time": 7.643009424209595
    },
    {
      "epoch": 5.4836273193359375e-06,
      "grad_norm": 0.16033749282360077,
      "learning_rate": 1.0799241957282292e-06,
      "loss": 0.0528,
      "step": 79750
    },
    {
      "epoch": 5.4836273193359375e-06,
      "model_forward_time": 1.7832584381103516,
      "step": 79750
    },
    {
      "dynamics_loss_avg": 0.04593048468232155,
      "epoch": 5.4836273193359375e-06,
      "step": 79750
    },
    {
      "action_loss_avg": 0.0011995178560027854,
      "epoch": 5.4836273193359375e-06,
      "step": 79750
    },
    {
      "epoch": 5.4836273193359375e-06,
      "step": 79750,
      "training_step_time": 7.5081822872161865
    },
    {
      "epoch": 5.484580993652344e-06,
      "model_forward_time": 1.783285140991211,
      "step": 79751
    },
    {
      "epoch": 5.484580993652344e-06,
      "step": 79751,
      "training_step_time": 8.058788776397705
    },
    {
      "epoch": 5.48553466796875e-06,
      "model_forward_time": 1.7754638195037842,
      "step": 79752
    },
    {
      "epoch": 5.48553466796875e-06,
      "step": 79752,
      "training_step_time": 7.725017547607422
    },
    {
      "epoch": 5.486488342285156e-06,
      "model_forward_time": 2.8577094078063965,
      "step": 79753
    },
    {
      "epoch": 5.486488342285156e-06,
      "step": 79753,
      "training_step_time": 8.781795740127563
    },
    {
      "epoch": 5.487442016601562e-06,
      "model_forward_time": 1.848984718322754,
      "step": 79754
    },
    {
      "epoch": 5.487442016601562e-06,
      "step": 79754,
      "training_step_time": 9.367213726043701
    },
    {
      "epoch": 5.488395690917969e-06,
      "model_forward_time": 1.7831976413726807,
      "step": 79755
    },
    {
      "epoch": 5.488395690917969e-06,
      "step": 79755,
      "training_step_time": 8.300881147384644
    },
    {
      "epoch": 5.489349365234375e-06,
      "model_forward_time": 1.7881817817687988,
      "step": 79756
    },
    {
      "epoch": 5.489349365234375e-06,
      "step": 79756,
      "training_step_time": 8.70484972000122
    },
    {
      "epoch": 5.490303039550781e-06,
      "model_forward_time": 1.7890729904174805,
      "step": 79757
    },
    {
      "epoch": 5.490303039550781e-06,
      "step": 79757,
      "training_step_time": 7.61626935005188
    },
    {
      "epoch": 5.4912567138671875e-06,
      "model_forward_time": 1.790536642074585,
      "step": 79758
    },
    {
      "epoch": 5.4912567138671875e-06,
      "step": 79758,
      "training_step_time": 7.607811212539673
    },
    {
      "epoch": 5.492210388183594e-06,
      "model_forward_time": 1.793407917022705,
      "step": 79759
    },
    {
      "epoch": 5.492210388183594e-06,
      "step": 79759,
      "training_step_time": 8.038602590560913
    },
    {
      "epoch": 5.4931640625e-06,
      "grad_norm": 0.12638446688652039,
      "learning_rate": 1.078898032468923e-06,
      "loss": 0.0552,
      "step": 79760
    },
    {
      "epoch": 5.4931640625e-06,
      "model_forward_time": 1.7861006259918213,
      "step": 79760
    },
    {
      "dynamics_loss_avg": 0.05911631230264902,
      "epoch": 5.4931640625e-06,
      "step": 79760
    },
    {
      "action_loss_avg": 0.0009536153898807243,
      "epoch": 5.4931640625e-06,
      "step": 79760
    },
    {
      "epoch": 5.4931640625e-06,
      "step": 79760,
      "training_step_time": 7.518784523010254
    },
    {
      "epoch": 5.494117736816406e-06,
      "model_forward_time": 1.78489089012146,
      "step": 79761
    },
    {
      "epoch": 5.494117736816406e-06,
      "step": 79761,
      "training_step_time": 8.424464702606201
    },
    {
      "epoch": 5.495071411132812e-06,
      "model_forward_time": 2.8582491874694824,
      "step": 79762
    },
    {
      "epoch": 5.495071411132812e-06,
      "step": 79762,
      "training_step_time": 7.8675925731658936
    },
    {
      "epoch": 5.496025085449219e-06,
      "model_forward_time": 1.8559081554412842,
      "step": 79763
    },
    {
      "epoch": 5.496025085449219e-06,
      "step": 79763,
      "training_step_time": 7.773770332336426
    },
    {
      "epoch": 5.496978759765625e-06,
      "model_forward_time": 1.7811307907104492,
      "step": 79764
    },
    {
      "epoch": 5.496978759765625e-06,
      "step": 79764,
      "training_step_time": 7.899361848831177
    },
    {
      "epoch": 5.497932434082031e-06,
      "model_forward_time": 1.785688877105713,
      "step": 79765
    },
    {
      "epoch": 5.497932434082031e-06,
      "step": 79765,
      "training_step_time": 8.839865922927856
    },
    {
      "epoch": 5.4988861083984375e-06,
      "model_forward_time": 1.7901599407196045,
      "step": 79766
    },
    {
      "epoch": 5.4988861083984375e-06,
      "step": 79766,
      "training_step_time": 7.698160648345947
    },
    {
      "epoch": 5.499839782714844e-06,
      "model_forward_time": 1.784724235534668,
      "step": 79767
    },
    {
      "epoch": 5.499839782714844e-06,
      "step": 79767,
      "training_step_time": 8.714938163757324
    },
    {
      "epoch": 5.50079345703125e-06,
      "model_forward_time": 1.775601863861084,
      "step": 79768
    },
    {
      "epoch": 5.50079345703125e-06,
      "step": 79768,
      "training_step_time": 7.5529351234436035
    },
    {
      "epoch": 5.501747131347656e-06,
      "model_forward_time": 1.8007309436798096,
      "step": 79769
    },
    {
      "epoch": 5.501747131347656e-06,
      "step": 79769,
      "training_step_time": 7.809084415435791
    },
    {
      "epoch": 5.502700805664062e-06,
      "grad_norm": 0.07408448308706284,
      "learning_rate": 1.0778722980154316e-06,
      "loss": 0.0552,
      "step": 79770
    },
    {
      "epoch": 5.502700805664062e-06,
      "model_forward_time": 1.0932340621948242,
      "step": 79770
    },
    {
      "dynamics_loss_avg": 0.06077552456408739,
      "epoch": 5.502700805664062e-06,
      "step": 79770
    },
    {
      "action_loss_avg": 0.0021495116292499007,
      "epoch": 5.502700805664062e-06,
      "step": 79770
    },
    {
      "epoch": 5.502700805664062e-06,
      "step": 79770,
      "training_step_time": 9.364222764968872
    },
    {
      "epoch": 5.503654479980469e-06,
      "model_forward_time": 2.8849565982818604,
      "step": 79771
    },
    {
      "epoch": 5.503654479980469e-06,
      "step": 79771,
      "training_step_time": 8.636303424835205
    },
    {
      "epoch": 5.504608154296875e-06,
      "model_forward_time": 1.850442886352539,
      "step": 79772
    },
    {
      "epoch": 5.504608154296875e-06,
      "step": 79772,
      "training_step_time": 10.020323753356934
    },
    {
      "epoch": 5.505561828613281e-06,
      "model_forward_time": 1.7746760845184326,
      "step": 79773
    },
    {
      "epoch": 5.505561828613281e-06,
      "step": 79773,
      "training_step_time": 8.255445003509521
    },
    {
      "epoch": 5.5065155029296875e-06,
      "model_forward_time": 1.7845556735992432,
      "step": 79774
    },
    {
      "epoch": 5.5065155029296875e-06,
      "step": 79774,
      "training_step_time": 8.383597373962402
    },
    {
      "epoch": 5.507469177246094e-06,
      "model_forward_time": 1.7859251499176025,
      "step": 79775
    },
    {
      "epoch": 5.507469177246094e-06,
      "step": 79775,
      "training_step_time": 7.742957592010498
    },
    {
      "epoch": 5.5084228515625e-06,
      "model_forward_time": 1.7838256359100342,
      "step": 79776
    },
    {
      "epoch": 5.5084228515625e-06,
      "step": 79776,
      "training_step_time": 7.856644868850708
    },
    {
      "epoch": 5.509376525878906e-06,
      "model_forward_time": 1.7610447406768799,
      "step": 79777
    },
    {
      "epoch": 5.509376525878906e-06,
      "step": 79777,
      "training_step_time": 7.391275882720947
    },
    {
      "epoch": 5.510330200195312e-06,
      "model_forward_time": 1.7775135040283203,
      "step": 79778
    },
    {
      "epoch": 5.510330200195312e-06,
      "step": 79778,
      "training_step_time": 8.281584978103638
    },
    {
      "epoch": 5.511283874511719e-06,
      "model_forward_time": 1.7835805416107178,
      "step": 79779
    },
    {
      "epoch": 5.511283874511719e-06,
      "step": 79779,
      "training_step_time": 8.57169222831726
    },
    {
      "epoch": 5.512237548828125e-06,
      "grad_norm": 0.07968234270811081,
      "learning_rate": 1.0768469924799301e-06,
      "loss": 0.0546,
      "step": 79780
    },
    {
      "epoch": 5.512237548828125e-06,
      "model_forward_time": 2.8248116970062256,
      "step": 79780
    },
    {
      "dynamics_loss_avg": 0.05011096969246864,
      "epoch": 5.512237548828125e-06,
      "step": 79780
    },
    {
      "action_loss_avg": 0.0012304349918849766,
      "epoch": 5.512237548828125e-06,
      "step": 79780
    },
    {
      "epoch": 5.512237548828125e-06,
      "step": 79780,
      "training_step_time": 9.452314376831055
    },
    {
      "epoch": 5.513191223144531e-06,
      "model_forward_time": 1.8392300605773926,
      "step": 79781
    },
    {
      "epoch": 5.513191223144531e-06,
      "step": 79781,
      "training_step_time": 8.596277475357056
    },
    {
      "epoch": 5.5141448974609375e-06,
      "model_forward_time": 1.7844789028167725,
      "step": 79782
    },
    {
      "epoch": 5.5141448974609375e-06,
      "step": 79782,
      "training_step_time": 7.765789270401001
    },
    {
      "epoch": 5.515098571777344e-06,
      "model_forward_time": 1.7864360809326172,
      "step": 79783
    },
    {
      "epoch": 5.515098571777344e-06,
      "step": 79783,
      "training_step_time": 8.230388402938843
    },
    {
      "epoch": 5.51605224609375e-06,
      "model_forward_time": 1.7840964794158936,
      "step": 79784
    },
    {
      "epoch": 5.51605224609375e-06,
      "step": 79784,
      "training_step_time": 7.732942342758179
    },
    {
      "epoch": 5.517005920410156e-06,
      "model_forward_time": 1.780120611190796,
      "step": 79785
    },
    {
      "epoch": 5.517005920410156e-06,
      "step": 79785,
      "training_step_time": 7.6713690757751465
    },
    {
      "epoch": 5.517959594726562e-06,
      "model_forward_time": 1.7815890312194824,
      "step": 79786
    },
    {
      "epoch": 5.517959594726562e-06,
      "step": 79786,
      "training_step_time": 7.355506658554077
    },
    {
      "epoch": 5.518913269042969e-06,
      "model_forward_time": 1.786146640777588,
      "step": 79787
    },
    {
      "epoch": 5.518913269042969e-06,
      "step": 79787,
      "training_step_time": 7.640547275543213
    },
    {
      "epoch": 5.519866943359375e-06,
      "model_forward_time": 1.7866613864898682,
      "step": 79788
    },
    {
      "epoch": 5.519866943359375e-06,
      "step": 79788,
      "training_step_time": 9.221067667007446
    },
    {
      "epoch": 5.520820617675781e-06,
      "model_forward_time": 2.83662748336792,
      "step": 79789
    },
    {
      "epoch": 5.520820617675781e-06,
      "step": 79789,
      "training_step_time": 8.3024320602417
    },
    {
      "epoch": 5.5217742919921875e-06,
      "grad_norm": 0.09248729795217514,
      "learning_rate": 1.075822115974542e-06,
      "loss": 0.0555,
      "step": 79790
    },
    {
      "epoch": 5.5217742919921875e-06,
      "model_forward_time": 1.8058385848999023,
      "step": 79790
    },
    {
      "dynamics_loss_avg": 0.0407210567034781,
      "epoch": 5.5217742919921875e-06,
      "step": 79790
    },
    {
      "action_loss_avg": 0.0008901902474462986,
      "epoch": 5.5217742919921875e-06,
      "step": 79790
    },
    {
      "epoch": 5.5217742919921875e-06,
      "step": 79790,
      "training_step_time": 9.171338319778442
    },
    {
      "epoch": 5.522727966308594e-06,
      "model_forward_time": 1.7814245223999023,
      "step": 79791
    },
    {
      "epoch": 5.522727966308594e-06,
      "step": 79791,
      "training_step_time": 7.715687990188599
    },
    {
      "epoch": 5.523681640625e-06,
      "model_forward_time": 1.803673267364502,
      "step": 79792
    },
    {
      "epoch": 5.523681640625e-06,
      "step": 79792,
      "training_step_time": 8.702347040176392
    },
    {
      "epoch": 5.524635314941406e-06,
      "model_forward_time": 1.7743301391601562,
      "step": 79793
    },
    {
      "epoch": 5.524635314941406e-06,
      "step": 79793,
      "training_step_time": 7.889953374862671
    },
    {
      "epoch": 5.525588989257812e-06,
      "model_forward_time": 1.7837369441986084,
      "step": 79794
    },
    {
      "epoch": 5.525588989257812e-06,
      "step": 79794,
      "training_step_time": 8.48261308670044
    },
    {
      "epoch": 5.526542663574219e-06,
      "model_forward_time": 1.802687168121338,
      "step": 79795
    },
    {
      "epoch": 5.526542663574219e-06,
      "step": 79795,
      "training_step_time": 7.758436441421509
    },
    {
      "epoch": 5.527496337890625e-06,
      "model_forward_time": 1.7969138622283936,
      "step": 79796
    },
    {
      "epoch": 5.527496337890625e-06,
      "step": 79796,
      "training_step_time": 7.6269004344940186
    },
    {
      "epoch": 5.528450012207031e-06,
      "model_forward_time": 1.7826550006866455,
      "step": 79797
    },
    {
      "epoch": 5.528450012207031e-06,
      "step": 79797,
      "training_step_time": 8.437212705612183
    },
    {
      "epoch": 5.5294036865234375e-06,
      "model_forward_time": 1.7947649955749512,
      "step": 79798
    },
    {
      "epoch": 5.5294036865234375e-06,
      "step": 79798,
      "training_step_time": 9.407912254333496
    },
    {
      "epoch": 5.530357360839844e-06,
      "model_forward_time": 1.8080289363861084,
      "step": 79799
    },
    {
      "epoch": 5.530357360839844e-06,
      "step": 79799,
      "training_step_time": 7.810994863510132
    },
    {
      "epoch": 5.53131103515625e-06,
      "grad_norm": 0.09737693518400192,
      "learning_rate": 1.074797668611347e-06,
      "loss": 0.0551,
      "step": 79800
    },
    {
      "epoch": 5.53131103515625e-06,
      "model_forward_time": 1.855722427368164,
      "step": 79800
    },
    {
      "dynamics_loss_avg": 0.05452194083482027,
      "epoch": 5.53131103515625e-06,
      "step": 79800
    },
    {
      "action_loss_avg": 0.000963516000774689,
      "epoch": 5.53131103515625e-06,
      "step": 79800
    },
    {
      "epoch": 5.53131103515625e-06,
      "step": 79800,
      "training_step_time": 7.755577325820923
    },
    {
      "epoch": 5.532264709472656e-06,
      "model_forward_time": 1.7800254821777344,
      "step": 79801
    },
    {
      "epoch": 5.532264709472656e-06,
      "step": 79801,
      "training_step_time": 8.442358255386353
    },
    {
      "epoch": 5.533218383789062e-06,
      "model_forward_time": 1.759178876876831,
      "step": 79802
    },
    {
      "epoch": 5.533218383789062e-06,
      "step": 79802,
      "training_step_time": 8.263015985488892
    },
    {
      "epoch": 5.534172058105469e-06,
      "model_forward_time": 1.7859270572662354,
      "step": 79803
    },
    {
      "epoch": 5.534172058105469e-06,
      "step": 79803,
      "training_step_time": 7.667446613311768
    },
    {
      "epoch": 5.535125732421875e-06,
      "model_forward_time": 1.7882134914398193,
      "step": 79804
    },
    {
      "epoch": 5.535125732421875e-06,
      "step": 79804,
      "training_step_time": 7.796185255050659
    },
    {
      "epoch": 5.536079406738281e-06,
      "model_forward_time": 1.8112456798553467,
      "step": 79805
    },
    {
      "epoch": 5.536079406738281e-06,
      "step": 79805,
      "training_step_time": 8.101260423660278
    },
    {
      "epoch": 5.5370330810546875e-06,
      "model_forward_time": 1.800837755203247,
      "step": 79806
    },
    {
      "epoch": 5.5370330810546875e-06,
      "step": 79806,
      "training_step_time": 7.437300682067871
    },
    {
      "epoch": 5.537986755371094e-06,
      "model_forward_time": 1.8210084438323975,
      "step": 79807
    },
    {
      "epoch": 5.537986755371094e-06,
      "step": 79807,
      "training_step_time": 7.7458977699279785
    },
    {
      "epoch": 5.5389404296875e-06,
      "model_forward_time": 2.9115188121795654,
      "step": 79808
    },
    {
      "epoch": 5.5389404296875e-06,
      "step": 79808,
      "training_step_time": 9.463228702545166
    },
    {
      "epoch": 5.539894104003906e-06,
      "model_forward_time": 1.4377450942993164,
      "step": 79809
    },
    {
      "epoch": 5.539894104003906e-06,
      "step": 79809,
      "training_step_time": 9.301351308822632
    },
    {
      "epoch": 5.540847778320312e-06,
      "grad_norm": 0.11735723167657852,
      "learning_rate": 1.0737736505023777e-06,
      "loss": 0.0564,
      "step": 79810
    },
    {
      "epoch": 5.540847778320312e-06,
      "model_forward_time": 1.7876648902893066,
      "step": 79810
    },
    {
      "dynamics_loss_avg": 0.05710989274084568,
      "epoch": 5.540847778320312e-06,
      "step": 79810
    },
    {
      "action_loss_avg": 0.00092717177758459,
      "epoch": 5.540847778320312e-06,
      "step": 79810
    },
    {
      "epoch": 5.540847778320312e-06,
      "step": 79810,
      "training_step_time": 8.55077314376831
    },
    {
      "epoch": 5.541801452636719e-06,
      "model_forward_time": 1.805323600769043,
      "step": 79811
    },
    {
      "epoch": 5.541801452636719e-06,
      "step": 79811,
      "training_step_time": 8.364146947860718
    },
    {
      "epoch": 5.542755126953125e-06,
      "model_forward_time": 1.8031256198883057,
      "step": 79812
    },
    {
      "epoch": 5.542755126953125e-06,
      "step": 79812,
      "training_step_time": 7.613136529922485
    },
    {
      "epoch": 5.543708801269531e-06,
      "model_forward_time": 1.8014678955078125,
      "step": 79813
    },
    {
      "epoch": 5.543708801269531e-06,
      "step": 79813,
      "training_step_time": 8.566136598587036
    },
    {
      "epoch": 5.5446624755859376e-06,
      "model_forward_time": 1.8007333278656006,
      "step": 79814
    },
    {
      "epoch": 5.5446624755859376e-06,
      "step": 79814,
      "training_step_time": 7.580281496047974
    },
    {
      "epoch": 5.545616149902344e-06,
      "model_forward_time": 1.7993366718292236,
      "step": 79815
    },
    {
      "epoch": 5.545616149902344e-06,
      "step": 79815,
      "training_step_time": 7.427133798599243
    },
    {
      "epoch": 5.54656982421875e-06,
      "model_forward_time": 1.7986299991607666,
      "step": 79816
    },
    {
      "epoch": 5.54656982421875e-06,
      "step": 79816,
      "training_step_time": 7.742145299911499
    },
    {
      "epoch": 5.547523498535156e-06,
      "model_forward_time": 2.869978904724121,
      "step": 79817
    },
    {
      "epoch": 5.547523498535156e-06,
      "step": 79817,
      "training_step_time": 8.527830839157104
    },
    {
      "epoch": 5.548477172851562e-06,
      "model_forward_time": 1.438291311264038,
      "step": 79818
    },
    {
      "epoch": 5.548477172851562e-06,
      "step": 79818,
      "training_step_time": 8.748997926712036
    },
    {
      "epoch": 5.549430847167969e-06,
      "model_forward_time": 1.7973191738128662,
      "step": 79819
    },
    {
      "epoch": 5.549430847167969e-06,
      "step": 79819,
      "training_step_time": 8.422135353088379
    },
    {
      "epoch": 5.550384521484375e-06,
      "grad_norm": 0.09791331738233566,
      "learning_rate": 1.0727500617596203e-06,
      "loss": 0.0561,
      "step": 79820
    },
    {
      "epoch": 5.550384521484375e-06,
      "model_forward_time": 1.7978031635284424,
      "step": 79820
    },
    {
      "dynamics_loss_avg": 0.05487489253282547,
      "epoch": 5.550384521484375e-06,
      "step": 79820
    },
    {
      "action_loss_avg": 0.0014758711564354598,
      "epoch": 5.550384521484375e-06,
      "step": 79820
    },
    {
      "epoch": 5.550384521484375e-06,
      "step": 79820,
      "training_step_time": 7.621738910675049
    },
    {
      "epoch": 5.551338195800781e-06,
      "model_forward_time": 1.8091285228729248,
      "step": 79821
    },
    {
      "epoch": 5.551338195800781e-06,
      "step": 79821,
      "training_step_time": 7.706657409667969
    },
    {
      "epoch": 5.5522918701171876e-06,
      "model_forward_time": 1.8054211139678955,
      "step": 79822
    },
    {
      "epoch": 5.5522918701171876e-06,
      "step": 79822,
      "training_step_time": 7.861420392990112
    },
    {
      "epoch": 5.553245544433594e-06,
      "model_forward_time": 1.7967674732208252,
      "step": 79823
    },
    {
      "epoch": 5.553245544433594e-06,
      "step": 79823,
      "training_step_time": 7.605484485626221
    },
    {
      "epoch": 5.55419921875e-06,
      "model_forward_time": 1.7997705936431885,
      "step": 79824
    },
    {
      "epoch": 5.55419921875e-06,
      "step": 79824,
      "training_step_time": 7.67351770401001
    },
    {
      "epoch": 5.555152893066406e-06,
      "model_forward_time": 1.8472046852111816,
      "step": 79825
    },
    {
      "epoch": 5.555152893066406e-06,
      "step": 79825,
      "training_step_time": 7.815480470657349
    },
    {
      "epoch": 5.556106567382812e-06,
      "model_forward_time": 2.853278160095215,
      "step": 79826
    },
    {
      "epoch": 5.556106567382812e-06,
      "step": 79826,
      "training_step_time": 8.26735806465149
    },
    {
      "epoch": 5.557060241699219e-06,
      "model_forward_time": 1.8096113204956055,
      "step": 79827
    },
    {
      "epoch": 5.557060241699219e-06,
      "step": 79827,
      "training_step_time": 8.624763011932373
    },
    {
      "epoch": 5.558013916015625e-06,
      "model_forward_time": 1.8635711669921875,
      "step": 79828
    },
    {
      "epoch": 5.558013916015625e-06,
      "step": 79828,
      "training_step_time": 8.706244945526123
    },
    {
      "epoch": 5.558967590332031e-06,
      "model_forward_time": 1.7929308414459229,
      "step": 79829
    },
    {
      "epoch": 5.558967590332031e-06,
      "step": 79829,
      "training_step_time": 7.902332544326782
    },
    {
      "epoch": 5.5599212646484376e-06,
      "grad_norm": 0.11083363741636276,
      "learning_rate": 1.0717269024950105e-06,
      "loss": 0.0565,
      "step": 79830
    },
    {
      "epoch": 5.5599212646484376e-06,
      "model_forward_time": 1.7957439422607422,
      "step": 79830
    },
    {
      "dynamics_loss_avg": 0.06107557211071253,
      "epoch": 5.5599212646484376e-06,
      "step": 79830
    },
    {
      "action_loss_avg": 0.0011221061693504452,
      "epoch": 5.5599212646484376e-06,
      "step": 79830
    },
    {
      "epoch": 5.5599212646484376e-06,
      "step": 79830,
      "training_step_time": 7.887583494186401
    },
    {
      "epoch": 5.560874938964844e-06,
      "model_forward_time": 1.8165185451507568,
      "step": 79831
    },
    {
      "epoch": 5.560874938964844e-06,
      "step": 79831,
      "training_step_time": 7.785820484161377
    },
    {
      "epoch": 5.56182861328125e-06,
      "model_forward_time": 1.797130823135376,
      "step": 79832
    },
    {
      "epoch": 5.56182861328125e-06,
      "step": 79832,
      "training_step_time": 8.008769512176514
    },
    {
      "epoch": 5.5627822875976564e-06,
      "model_forward_time": 1.800572156906128,
      "step": 79833
    },
    {
      "epoch": 5.5627822875976564e-06,
      "step": 79833,
      "training_step_time": 7.797497510910034
    },
    {
      "epoch": 5.563735961914062e-06,
      "model_forward_time": 1.8004028797149658,
      "step": 79834
    },
    {
      "epoch": 5.563735961914062e-06,
      "step": 79834,
      "training_step_time": 8.793872356414795
    },
    {
      "epoch": 5.564689636230469e-06,
      "model_forward_time": 1.8017044067382812,
      "step": 79835
    },
    {
      "epoch": 5.564689636230469e-06,
      "step": 79835,
      "training_step_time": 8.438596487045288
    },
    {
      "epoch": 5.565643310546875e-06,
      "model_forward_time": 1.8132574558258057,
      "step": 79836
    },
    {
      "epoch": 5.565643310546875e-06,
      "step": 79836,
      "training_step_time": 8.984260082244873
    },
    {
      "epoch": 5.566596984863281e-06,
      "model_forward_time": 1.8499464988708496,
      "step": 79837
    },
    {
      "epoch": 5.566596984863281e-06,
      "step": 79837,
      "training_step_time": 7.86889386177063
    },
    {
      "epoch": 5.5675506591796876e-06,
      "model_forward_time": 1.7622559070587158,
      "step": 79838
    },
    {
      "epoch": 5.5675506591796876e-06,
      "step": 79838,
      "training_step_time": 7.952582836151123
    },
    {
      "epoch": 5.568504333496094e-06,
      "model_forward_time": 1.7901010513305664,
      "step": 79839
    },
    {
      "epoch": 5.568504333496094e-06,
      "step": 79839,
      "training_step_time": 8.522873640060425
    },
    {
      "epoch": 5.5694580078125e-06,
      "grad_norm": 0.07419116050004959,
      "learning_rate": 1.0707041728204421e-06,
      "loss": 0.0573,
      "step": 79840
    },
    {
      "epoch": 5.5694580078125e-06,
      "model_forward_time": 1.799692153930664,
      "step": 79840
    },
    {
      "dynamics_loss_avg": 0.041537365317344664,
      "epoch": 5.5694580078125e-06,
      "step": 79840
    },
    {
      "action_loss_avg": 0.0014060220040846617,
      "epoch": 5.5694580078125e-06,
      "step": 79840
    },
    {
      "epoch": 5.5694580078125e-06,
      "step": 79840,
      "training_step_time": 7.6416144371032715
    },
    {
      "epoch": 5.5704116821289064e-06,
      "model_forward_time": 1.7939958572387695,
      "step": 79841
    },
    {
      "epoch": 5.5704116821289064e-06,
      "step": 79841,
      "training_step_time": 8.424651622772217
    },
    {
      "epoch": 5.571365356445312e-06,
      "model_forward_time": 1.787001132965088,
      "step": 79842
    },
    {
      "epoch": 5.571365356445312e-06,
      "step": 79842,
      "training_step_time": 7.629047632217407
    },
    {
      "epoch": 5.572319030761719e-06,
      "model_forward_time": 1.786827564239502,
      "step": 79843
    },
    {
      "epoch": 5.572319030761719e-06,
      "step": 79843,
      "training_step_time": 7.603148937225342
    },
    {
      "epoch": 5.573272705078125e-06,
      "model_forward_time": 1.7903716564178467,
      "step": 79844
    },
    {
      "epoch": 5.573272705078125e-06,
      "step": 79844,
      "training_step_time": 7.732203006744385
    },
    {
      "epoch": 5.574226379394531e-06,
      "model_forward_time": 2.479921340942383,
      "step": 79845
    },
    {
      "epoch": 5.574226379394531e-06,
      "step": 79845,
      "training_step_time": 8.600265979766846
    },
    {
      "epoch": 5.575180053710938e-06,
      "model_forward_time": 1.814432144165039,
      "step": 79846
    },
    {
      "epoch": 5.575180053710938e-06,
      "step": 79846,
      "training_step_time": 9.66894006729126
    },
    {
      "epoch": 5.576133728027344e-06,
      "model_forward_time": 1.7746548652648926,
      "step": 79847
    },
    {
      "epoch": 5.576133728027344e-06,
      "step": 79847,
      "training_step_time": 7.876088380813599
    },
    {
      "epoch": 5.57708740234375e-06,
      "model_forward_time": 1.7854156494140625,
      "step": 79848
    },
    {
      "epoch": 5.57708740234375e-06,
      "step": 79848,
      "training_step_time": 7.760931491851807
    },
    {
      "epoch": 5.5780410766601564e-06,
      "model_forward_time": 1.8002429008483887,
      "step": 79849
    },
    {
      "epoch": 5.5780410766601564e-06,
      "step": 79849,
      "training_step_time": 7.6931891441345215
    },
    {
      "epoch": 5.578994750976562e-06,
      "grad_norm": 0.11398763954639435,
      "learning_rate": 1.0696818728477559e-06,
      "loss": 0.0554,
      "step": 79850
    },
    {
      "epoch": 5.578994750976562e-06,
      "model_forward_time": 1.7994108200073242,
      "step": 79850
    },
    {
      "dynamics_loss_avg": 0.05757092013955116,
      "epoch": 5.578994750976562e-06,
      "step": 79850
    },
    {
      "action_loss_avg": 0.0013630702465889043,
      "epoch": 5.578994750976562e-06,
      "step": 79850
    },
    {
      "epoch": 5.578994750976562e-06,
      "step": 79850,
      "training_step_time": 8.860282897949219
    },
    {
      "epoch": 5.579948425292969e-06,
      "model_forward_time": 1.7997984886169434,
      "step": 79851
    },
    {
      "epoch": 5.579948425292969e-06,
      "step": 79851,
      "training_step_time": 7.712472438812256
    },
    {
      "epoch": 5.580902099609375e-06,
      "model_forward_time": 1.7971084117889404,
      "step": 79852
    },
    {
      "epoch": 5.580902099609375e-06,
      "step": 79852,
      "training_step_time": 8.212148904800415
    },
    {
      "epoch": 5.581855773925781e-06,
      "model_forward_time": 1.796241283416748,
      "step": 79853
    },
    {
      "epoch": 5.581855773925781e-06,
      "step": 79853,
      "training_step_time": 8.826480150222778
    },
    {
      "epoch": 5.582809448242188e-06,
      "model_forward_time": 2.8624675273895264,
      "step": 79854
    },
    {
      "epoch": 5.582809448242188e-06,
      "step": 79854,
      "training_step_time": 9.670708656311035
    },
    {
      "epoch": 5.583763122558594e-06,
      "model_forward_time": 1.789503574371338,
      "step": 79855
    },
    {
      "epoch": 5.583763122558594e-06,
      "step": 79855,
      "training_step_time": 9.672560453414917
    },
    {
      "epoch": 5.584716796875e-06,
      "model_forward_time": 1.818659782409668,
      "step": 79856
    },
    {
      "epoch": 5.584716796875e-06,
      "step": 79856,
      "training_step_time": 7.7073352336883545
    },
    {
      "epoch": 5.5856704711914064e-06,
      "model_forward_time": 1.7461628913879395,
      "step": 79857
    },
    {
      "epoch": 5.5856704711914064e-06,
      "step": 79857,
      "training_step_time": 8.255767345428467
    },
    {
      "epoch": 5.586624145507812e-06,
      "model_forward_time": 1.7391648292541504,
      "step": 79858
    },
    {
      "epoch": 5.586624145507812e-06,
      "step": 79858,
      "training_step_time": 7.81296968460083
    },
    {
      "epoch": 5.587577819824219e-06,
      "model_forward_time": 1.7393620014190674,
      "step": 79859
    },
    {
      "epoch": 5.587577819824219e-06,
      "step": 79859,
      "training_step_time": 9.33388876914978
    },
    {
      "epoch": 5.588531494140625e-06,
      "grad_norm": 0.10624927282333374,
      "learning_rate": 1.068660002688751e-06,
      "loss": 0.0556,
      "step": 79860
    },
    {
      "epoch": 5.588531494140625e-06,
      "model_forward_time": 1.7462079524993896,
      "step": 79860
    },
    {
      "dynamics_loss_avg": 0.056001237407326696,
      "epoch": 5.588531494140625e-06,
      "step": 79860
    },
    {
      "action_loss_avg": 0.0011743858602130785,
      "epoch": 5.588531494140625e-06,
      "step": 79860
    },
    {
      "epoch": 5.588531494140625e-06,
      "step": 79860,
      "training_step_time": 7.831635236740112
    },
    {
      "epoch": 5.589485168457031e-06,
      "model_forward_time": 1.7410457134246826,
      "step": 79861
    },
    {
      "epoch": 5.589485168457031e-06,
      "step": 79861,
      "training_step_time": 7.584646463394165
    },
    {
      "epoch": 5.590438842773438e-06,
      "model_forward_time": 1.742119312286377,
      "step": 79862
    },
    {
      "epoch": 5.590438842773438e-06,
      "step": 79862,
      "training_step_time": 7.877290725708008
    },
    {
      "epoch": 5.591392517089844e-06,
      "model_forward_time": 2.832219123840332,
      "step": 79863
    },
    {
      "epoch": 5.591392517089844e-06,
      "step": 79863,
      "training_step_time": 9.409903287887573
    },
    {
      "epoch": 5.59234619140625e-06,
      "model_forward_time": 1.7530879974365234,
      "step": 79864
    },
    {
      "epoch": 5.59234619140625e-06,
      "step": 79864,
      "training_step_time": 8.815346956253052
    },
    {
      "epoch": 5.5932998657226565e-06,
      "model_forward_time": 1.8097002506256104,
      "step": 79865
    },
    {
      "epoch": 5.5932998657226565e-06,
      "step": 79865,
      "training_step_time": 7.853837490081787
    },
    {
      "epoch": 5.594253540039062e-06,
      "model_forward_time": 1.7430057525634766,
      "step": 79866
    },
    {
      "epoch": 5.594253540039062e-06,
      "step": 79866,
      "training_step_time": 7.678743124008179
    },
    {
      "epoch": 5.595207214355469e-06,
      "model_forward_time": 1.745873212814331,
      "step": 79867
    },
    {
      "epoch": 5.595207214355469e-06,
      "step": 79867,
      "training_step_time": 7.798162460327148
    },
    {
      "epoch": 5.596160888671875e-06,
      "model_forward_time": 1.7492563724517822,
      "step": 79868
    },
    {
      "epoch": 5.596160888671875e-06,
      "step": 79868,
      "training_step_time": 8.274543762207031
    },
    {
      "epoch": 5.597114562988281e-06,
      "model_forward_time": 1.7474427223205566,
      "step": 79869
    },
    {
      "epoch": 5.597114562988281e-06,
      "step": 79869,
      "training_step_time": 8.5130136013031
    },
    {
      "epoch": 5.598068237304688e-06,
      "grad_norm": 0.0944543406367302,
      "learning_rate": 1.0676385624551778e-06,
      "loss": 0.0562,
      "step": 79870
    },
    {
      "epoch": 5.598068237304688e-06,
      "model_forward_time": 1.749748945236206,
      "step": 79870
    },
    {
      "dynamics_loss_avg": 0.04959332440048456,
      "epoch": 5.598068237304688e-06,
      "step": 79870
    },
    {
      "action_loss_avg": 0.0013679422670975327,
      "epoch": 5.598068237304688e-06,
      "step": 79870
    },
    {
      "epoch": 5.598068237304688e-06,
      "step": 79870,
      "training_step_time": 7.71728515625
    },
    {
      "epoch": 5.599021911621094e-06,
      "model_forward_time": 1.7365872859954834,
      "step": 79871
    },
    {
      "epoch": 5.599021911621094e-06,
      "step": 79871,
      "training_step_time": 7.879190921783447
    },
    {
      "epoch": 5.5999755859375e-06,
      "model_forward_time": 2.7832844257354736,
      "step": 79872
    },
    {
      "epoch": 5.5999755859375e-06,
      "step": 79872,
      "training_step_time": 8.202820062637329
    },
    {
      "epoch": 5.6009292602539065e-06,
      "model_forward_time": 1.7415082454681396,
      "step": 79873
    },
    {
      "epoch": 5.6009292602539065e-06,
      "step": 79873,
      "training_step_time": 8.683250427246094
    },
    {
      "epoch": 5.601882934570312e-06,
      "model_forward_time": 1.8219876289367676,
      "step": 79874
    },
    {
      "epoch": 5.601882934570312e-06,
      "step": 79874,
      "training_step_time": 8.655121326446533
    },
    {
      "epoch": 5.602836608886719e-06,
      "model_forward_time": 1.7400715351104736,
      "step": 79875
    },
    {
      "epoch": 5.602836608886719e-06,
      "step": 79875,
      "training_step_time": 8.830949306488037
    },
    {
      "epoch": 5.603790283203125e-06,
      "model_forward_time": 1.3637759685516357,
      "step": 79876
    },
    {
      "epoch": 5.603790283203125e-06,
      "step": 79876,
      "training_step_time": 8.342974424362183
    },
    {
      "epoch": 5.604743957519531e-06,
      "model_forward_time": 1.7445154190063477,
      "step": 79877
    },
    {
      "epoch": 5.604743957519531e-06,
      "step": 79877,
      "training_step_time": 7.776709079742432
    },
    {
      "epoch": 5.605697631835938e-06,
      "model_forward_time": 1.7463271617889404,
      "step": 79878
    },
    {
      "epoch": 5.605697631835938e-06,
      "step": 79878,
      "training_step_time": 7.52292013168335
    },
    {
      "epoch": 5.606651306152343e-06,
      "model_forward_time": 1.7376911640167236,
      "step": 79879
    },
    {
      "epoch": 5.606651306152343e-06,
      "step": 79879,
      "training_step_time": 7.753737688064575
    },
    {
      "epoch": 5.60760498046875e-06,
      "grad_norm": 0.07872488349676132,
      "learning_rate": 1.0666175522587402e-06,
      "loss": 0.0553,
      "step": 79880
    },
    {
      "epoch": 5.60760498046875e-06,
      "model_forward_time": 1.742292881011963,
      "step": 79880
    },
    {
      "dynamics_loss_avg": 0.0578638568520546,
      "epoch": 5.60760498046875e-06,
      "step": 79880
    },
    {
      "action_loss_avg": 0.0014876299537718296,
      "epoch": 5.60760498046875e-06,
      "step": 79880
    },
    {
      "epoch": 5.60760498046875e-06,
      "step": 79880,
      "training_step_time": 7.756345510482788
    },
    {
      "epoch": 5.6085586547851565e-06,
      "model_forward_time": 1.7423479557037354,
      "step": 79881
    },
    {
      "epoch": 5.6085586547851565e-06,
      "step": 79881,
      "training_step_time": 9.285969495773315
    },
    {
      "epoch": 5.609512329101562e-06,
      "model_forward_time": 2.8320488929748535,
      "step": 79882
    },
    {
      "epoch": 5.609512329101562e-06,
      "step": 79882,
      "training_step_time": 8.415241956710815
    },
    {
      "epoch": 5.610466003417969e-06,
      "model_forward_time": 1.8273680210113525,
      "step": 79883
    },
    {
      "epoch": 5.610466003417969e-06,
      "step": 79883,
      "training_step_time": 8.297656297683716
    },
    {
      "epoch": 5.611419677734375e-06,
      "model_forward_time": 1.739915132522583,
      "step": 79884
    },
    {
      "epoch": 5.611419677734375e-06,
      "step": 79884,
      "training_step_time": 7.709882497787476
    },
    {
      "epoch": 5.612373352050781e-06,
      "model_forward_time": 1.7455854415893555,
      "step": 79885
    },
    {
      "epoch": 5.612373352050781e-06,
      "step": 79885,
      "training_step_time": 7.708580732345581
    },
    {
      "epoch": 5.613327026367188e-06,
      "model_forward_time": 1.7411670684814453,
      "step": 79886
    },
    {
      "epoch": 5.613327026367188e-06,
      "step": 79886,
      "training_step_time": 8.571828365325928
    },
    {
      "epoch": 5.614280700683593e-06,
      "model_forward_time": 1.7534830570220947,
      "step": 79887
    },
    {
      "epoch": 5.614280700683593e-06,
      "step": 79887,
      "training_step_time": 7.565417289733887
    },
    {
      "epoch": 5.615234375e-06,
      "model_forward_time": 1.7452232837677002,
      "step": 79888
    },
    {
      "epoch": 5.615234375e-06,
      "step": 79888,
      "training_step_time": 7.7241294384002686
    },
    {
      "epoch": 5.6161880493164065e-06,
      "model_forward_time": 1.7598860263824463,
      "step": 79889
    },
    {
      "epoch": 5.6161880493164065e-06,
      "step": 79889,
      "training_step_time": 8.415651082992554
    },
    {
      "epoch": 5.617141723632812e-06,
      "grad_norm": 0.10142850875854492,
      "learning_rate": 1.0655969722110925e-06,
      "loss": 0.0535,
      "step": 79890
    },
    {
      "epoch": 5.617141723632812e-06,
      "model_forward_time": 1.7546517848968506,
      "step": 79890
    },
    {
      "dynamics_loss_avg": 0.050148706510663034,
      "epoch": 5.617141723632812e-06,
      "step": 79890
    },
    {
      "action_loss_avg": 0.0012710576760582627,
      "epoch": 5.617141723632812e-06,
      "step": 79890
    },
    {
      "epoch": 5.617141723632812e-06,
      "step": 79890,
      "training_step_time": 8.737182855606079
    },
    {
      "epoch": 5.618095397949219e-06,
      "model_forward_time": 2.840160369873047,
      "step": 79891
    },
    {
      "epoch": 5.618095397949219e-06,
      "step": 79891,
      "training_step_time": 7.666037082672119
    },
    {
      "epoch": 5.619049072265625e-06,
      "model_forward_time": 1.8186852931976318,
      "step": 79892
    },
    {
      "epoch": 5.619049072265625e-06,
      "step": 79892,
      "training_step_time": 8.881110668182373
    },
    {
      "epoch": 5.620002746582031e-06,
      "model_forward_time": 1.7405526638031006,
      "step": 79893
    },
    {
      "epoch": 5.620002746582031e-06,
      "step": 79893,
      "training_step_time": 8.597862720489502
    },
    {
      "epoch": 5.620956420898438e-06,
      "model_forward_time": 1.7427761554718018,
      "step": 79894
    },
    {
      "epoch": 5.620956420898438e-06,
      "step": 79894,
      "training_step_time": 8.458117961883545
    },
    {
      "epoch": 5.621910095214843e-06,
      "model_forward_time": 1.7424497604370117,
      "step": 79895
    },
    {
      "epoch": 5.621910095214843e-06,
      "step": 79895,
      "training_step_time": 7.730749845504761
    },
    {
      "epoch": 5.62286376953125e-06,
      "model_forward_time": 1.761575698852539,
      "step": 79896
    },
    {
      "epoch": 5.62286376953125e-06,
      "step": 79896,
      "training_step_time": 8.268138885498047
    },
    {
      "epoch": 5.6238174438476565e-06,
      "model_forward_time": 1.774381160736084,
      "step": 79897
    },
    {
      "epoch": 5.6238174438476565e-06,
      "step": 79897,
      "training_step_time": 8.373830795288086
    },
    {
      "epoch": 5.624771118164062e-06,
      "model_forward_time": 1.773660659790039,
      "step": 79898
    },
    {
      "epoch": 5.624771118164062e-06,
      "step": 79898,
      "training_step_time": 7.562854290008545
    },
    {
      "epoch": 5.625724792480469e-06,
      "model_forward_time": 1.772979497909546,
      "step": 79899
    },
    {
      "epoch": 5.625724792480469e-06,
      "step": 79899,
      "training_step_time": 9.246086120605469
    },
    {
      "epoch": 5.626678466796875e-06,
      "grad_norm": 0.14800240099430084,
      "learning_rate": 1.064576822423844e-06,
      "loss": 0.0541,
      "step": 79900
    },
    {
      "epoch": 5.626678466796875e-06,
      "model_forward_time": 2.8188347816467285,
      "step": 79900
    },
    {
      "dynamics_loss_avg": 0.04889802262187004,
      "epoch": 5.626678466796875e-06,
      "step": 79900
    },
    {
      "action_loss_avg": 0.0013116463320329784,
      "epoch": 5.626678466796875e-06,
      "step": 79900
    },
    {
      "epoch": 5.626678466796875e-06,
      "step": 79900,
      "training_step_time": 7.6494879722595215
    },
    {
      "epoch": 5.627632141113281e-06,
      "model_forward_time": 1.8310339450836182,
      "step": 79901
    },
    {
      "epoch": 5.627632141113281e-06,
      "step": 79901,
      "training_step_time": 8.611317157745361
    },
    {
      "epoch": 5.628585815429688e-06,
      "model_forward_time": 1.7639317512512207,
      "step": 79902
    },
    {
      "epoch": 5.628585815429688e-06,
      "step": 79902,
      "training_step_time": 7.717400312423706
    },
    {
      "epoch": 5.629539489746093e-06,
      "model_forward_time": 1.7657172679901123,
      "step": 79903
    },
    {
      "epoch": 5.629539489746093e-06,
      "step": 79903,
      "training_step_time": 8.44122314453125
    },
    {
      "epoch": 5.6304931640625e-06,
      "model_forward_time": 1.7532098293304443,
      "step": 79904
    },
    {
      "epoch": 5.6304931640625e-06,
      "step": 79904,
      "training_step_time": 8.941185235977173
    },
    {
      "epoch": 5.6314468383789065e-06,
      "model_forward_time": 1.7655613422393799,
      "step": 79905
    },
    {
      "epoch": 5.6314468383789065e-06,
      "step": 79905,
      "training_step_time": 7.6043102741241455
    },
    {
      "epoch": 5.632400512695312e-06,
      "model_forward_time": 1.767202615737915,
      "step": 79906
    },
    {
      "epoch": 5.632400512695312e-06,
      "step": 79906,
      "training_step_time": 8.517462730407715
    },
    {
      "epoch": 5.633354187011719e-06,
      "model_forward_time": 1.7561001777648926,
      "step": 79907
    },
    {
      "epoch": 5.633354187011719e-06,
      "step": 79907,
      "training_step_time": 8.341953992843628
    },
    {
      "epoch": 5.634307861328125e-06,
      "model_forward_time": 1.76318359375,
      "step": 79908
    },
    {
      "epoch": 5.634307861328125e-06,
      "step": 79908,
      "training_step_time": 7.7570929527282715
    },
    {
      "epoch": 5.635261535644531e-06,
      "model_forward_time": 2.8522884845733643,
      "step": 79909
    },
    {
      "epoch": 5.635261535644531e-06,
      "step": 79909,
      "training_step_time": 7.67771053314209
    },
    {
      "epoch": 5.636215209960938e-06,
      "grad_norm": 0.09627711772918701,
      "learning_rate": 1.063557103008559e-06,
      "loss": 0.0532,
      "step": 79910
    },
    {
      "epoch": 5.636215209960938e-06,
      "model_forward_time": 1.8308947086334229,
      "step": 79910
    },
    {
      "dynamics_loss_avg": 0.05682094991207123,
      "epoch": 5.636215209960938e-06,
      "step": 79910
    },
    {
      "action_loss_avg": 0.0010810856358148158,
      "epoch": 5.636215209960938e-06,
      "step": 79910
    },
    {
      "epoch": 5.636215209960938e-06,
      "step": 79910,
      "training_step_time": 9.39096736907959
    },
    {
      "epoch": 5.637168884277343e-06,
      "model_forward_time": 1.7647230625152588,
      "step": 79911
    },
    {
      "epoch": 5.637168884277343e-06,
      "step": 79911,
      "training_step_time": 9.991643190383911
    },
    {
      "epoch": 5.63812255859375e-06,
      "model_forward_time": 1.7603583335876465,
      "step": 79912
    },
    {
      "epoch": 5.63812255859375e-06,
      "step": 79912,
      "training_step_time": 8.82020616531372
    },
    {
      "epoch": 5.6390762329101565e-06,
      "model_forward_time": 1.7644376754760742,
      "step": 79913
    },
    {
      "epoch": 5.6390762329101565e-06,
      "step": 79913,
      "training_step_time": 8.452165365219116
    },
    {
      "epoch": 5.640029907226562e-06,
      "model_forward_time": 1.7554314136505127,
      "step": 79914
    },
    {
      "epoch": 5.640029907226562e-06,
      "step": 79914,
      "training_step_time": 7.722369194030762
    },
    {
      "epoch": 5.640983581542969e-06,
      "model_forward_time": 1.7651383876800537,
      "step": 79915
    },
    {
      "epoch": 5.640983581542969e-06,
      "step": 79915,
      "training_step_time": 7.434203386306763
    },
    {
      "epoch": 5.641937255859375e-06,
      "model_forward_time": 1.7683172225952148,
      "step": 79916
    },
    {
      "epoch": 5.641937255859375e-06,
      "step": 79916,
      "training_step_time": 7.627402067184448
    },
    {
      "epoch": 5.642890930175781e-06,
      "model_forward_time": 1.7803032398223877,
      "step": 79917
    },
    {
      "epoch": 5.642890930175781e-06,
      "step": 79917,
      "training_step_time": 8.159879207611084
    },
    {
      "epoch": 5.643844604492188e-06,
      "model_forward_time": 1.7605853080749512,
      "step": 79918
    },
    {
      "epoch": 5.643844604492188e-06,
      "step": 79918,
      "training_step_time": 8.102302312850952
    },
    {
      "epoch": 5.644798278808593e-06,
      "model_forward_time": 1.8357150554656982,
      "step": 79919
    },
    {
      "epoch": 5.644798278808593e-06,
      "step": 79919,
      "training_step_time": 8.26937460899353
    },
    {
      "epoch": 5.645751953125e-06,
      "grad_norm": 0.09736558794975281,
      "learning_rate": 1.0625378140767496e-06,
      "loss": 0.0538,
      "step": 79920
    },
    {
      "epoch": 5.645751953125e-06,
      "model_forward_time": 1.7635748386383057,
      "step": 79920
    },
    {
      "dynamics_loss_avg": 0.06142841745167971,
      "epoch": 5.645751953125e-06,
      "step": 79920
    },
    {
      "action_loss_avg": 0.0015732738364022225,
      "epoch": 5.645751953125e-06,
      "step": 79920
    },
    {
      "epoch": 5.645751953125e-06,
      "step": 79920,
      "training_step_time": 7.6852076053619385
    },
    {
      "epoch": 5.6467056274414065e-06,
      "model_forward_time": 1.7639586925506592,
      "step": 79921
    },
    {
      "epoch": 5.6467056274414065e-06,
      "step": 79921,
      "training_step_time": 7.8277647495269775
    },
    {
      "epoch": 5.647659301757812e-06,
      "model_forward_time": 1.7652108669281006,
      "step": 79922
    },
    {
      "epoch": 5.647659301757812e-06,
      "step": 79922,
      "training_step_time": 7.708729982376099
    },
    {
      "epoch": 5.648612976074219e-06,
      "model_forward_time": 1.7716217041015625,
      "step": 79923
    },
    {
      "epoch": 5.648612976074219e-06,
      "step": 79923,
      "training_step_time": 8.173753023147583
    },
    {
      "epoch": 5.649566650390625e-06,
      "model_forward_time": 1.7652568817138672,
      "step": 79924
    },
    {
      "epoch": 5.649566650390625e-06,
      "step": 79924,
      "training_step_time": 8.097429752349854
    },
    {
      "epoch": 5.650520324707031e-06,
      "model_forward_time": 1.7696194648742676,
      "step": 79925
    },
    {
      "epoch": 5.650520324707031e-06,
      "step": 79925,
      "training_step_time": 8.389873027801514
    },
    {
      "epoch": 5.651473999023438e-06,
      "model_forward_time": 1.7611148357391357,
      "step": 79926
    },
    {
      "epoch": 5.651473999023438e-06,
      "step": 79926,
      "training_step_time": 7.628223180770874
    },
    {
      "epoch": 5.652427673339843e-06,
      "model_forward_time": 1.7645618915557861,
      "step": 79927
    },
    {
      "epoch": 5.652427673339843e-06,
      "step": 79927,
      "training_step_time": 7.6468236446380615
    },
    {
      "epoch": 5.65338134765625e-06,
      "model_forward_time": 2.834378242492676,
      "step": 79928
    },
    {
      "epoch": 5.65338134765625e-06,
      "step": 79928,
      "training_step_time": 9.47557520866394
    },
    {
      "epoch": 5.6543350219726565e-06,
      "model_forward_time": 1.7247052192687988,
      "step": 79929
    },
    {
      "epoch": 5.6543350219726565e-06,
      "step": 79929,
      "training_step_time": 8.436720609664917
    },
    {
      "epoch": 5.655288696289062e-06,
      "grad_norm": 0.09722970426082611,
      "learning_rate": 1.0615189557398847e-06,
      "loss": 0.054,
      "step": 79930
    },
    {
      "epoch": 5.655288696289062e-06,
      "model_forward_time": 1.7442493438720703,
      "step": 79930
    },
    {
      "dynamics_loss_avg": 0.04876975910738111,
      "epoch": 5.655288696289062e-06,
      "step": 79930
    },
    {
      "action_loss_avg": 0.0011372346867574378,
      "epoch": 5.655288696289062e-06,
      "step": 79930
    },
    {
      "epoch": 5.655288696289062e-06,
      "step": 79930,
      "training_step_time": 7.827772855758667
    },
    {
      "epoch": 5.656242370605469e-06,
      "model_forward_time": 1.747971773147583,
      "step": 79931
    },
    {
      "epoch": 5.656242370605469e-06,
      "step": 79931,
      "training_step_time": 8.174008131027222
    },
    {
      "epoch": 5.657196044921875e-06,
      "model_forward_time": 1.7420754432678223,
      "step": 79932
    },
    {
      "epoch": 5.657196044921875e-06,
      "step": 79932,
      "training_step_time": 7.702149391174316
    },
    {
      "epoch": 5.658149719238281e-06,
      "model_forward_time": 1.747314691543579,
      "step": 79933
    },
    {
      "epoch": 5.658149719238281e-06,
      "step": 79933,
      "training_step_time": 7.5754053592681885
    },
    {
      "epoch": 5.659103393554688e-06,
      "model_forward_time": 1.742551326751709,
      "step": 79934
    },
    {
      "epoch": 5.659103393554688e-06,
      "step": 79934,
      "training_step_time": 8.224302291870117
    },
    {
      "epoch": 5.660057067871093e-06,
      "model_forward_time": 1.7415120601654053,
      "step": 79935
    },
    {
      "epoch": 5.660057067871093e-06,
      "step": 79935,
      "training_step_time": 7.615488767623901
    },
    {
      "epoch": 5.6610107421875e-06,
      "model_forward_time": 1.7436738014221191,
      "step": 79936
    },
    {
      "epoch": 5.6610107421875e-06,
      "step": 79936,
      "training_step_time": 7.648327350616455
    },
    {
      "epoch": 5.6619644165039065e-06,
      "model_forward_time": 2.8500101566314697,
      "step": 79937
    },
    {
      "epoch": 5.6619644165039065e-06,
      "step": 79937,
      "training_step_time": 7.6051130294799805
    },
    {
      "epoch": 5.662918090820312e-06,
      "model_forward_time": 1.7495629787445068,
      "step": 79938
    },
    {
      "epoch": 5.662918090820312e-06,
      "step": 79938,
      "training_step_time": 8.650465726852417
    },
    {
      "epoch": 5.663871765136719e-06,
      "model_forward_time": 1.7533538341522217,
      "step": 79939
    },
    {
      "epoch": 5.663871765136719e-06,
      "step": 79939,
      "training_step_time": 8.529658079147339
    },
    {
      "epoch": 5.664825439453125e-06,
      "grad_norm": 0.08197728544473648,
      "learning_rate": 1.0605005281093861e-06,
      "loss": 0.0526,
      "step": 79940
    },
    {
      "epoch": 5.664825439453125e-06,
      "model_forward_time": 1.7561726570129395,
      "step": 79940
    },
    {
      "dynamics_loss_avg": 0.059163106977939604,
      "epoch": 5.664825439453125e-06,
      "step": 79940
    },
    {
      "action_loss_avg": 0.0011182280373759567,
      "epoch": 5.664825439453125e-06,
      "step": 79940
    },
    {
      "epoch": 5.664825439453125e-06,
      "step": 79940,
      "training_step_time": 9.334980249404907
    },
    {
      "epoch": 5.665779113769531e-06,
      "model_forward_time": 1.7647161483764648,
      "step": 79941
    },
    {
      "epoch": 5.665779113769531e-06,
      "step": 79941,
      "training_step_time": 8.202294826507568
    },
    {
      "epoch": 5.666732788085938e-06,
      "model_forward_time": 1.7640252113342285,
      "step": 79942
    },
    {
      "epoch": 5.666732788085938e-06,
      "step": 79942,
      "training_step_time": 8.150577306747437
    },
    {
      "epoch": 5.667686462402343e-06,
      "model_forward_time": 1.7884719371795654,
      "step": 79943
    },
    {
      "epoch": 5.667686462402343e-06,
      "step": 79943,
      "training_step_time": 7.5481274127960205
    },
    {
      "epoch": 5.66864013671875e-06,
      "model_forward_time": 1.7833166122436523,
      "step": 79944
    },
    {
      "epoch": 5.66864013671875e-06,
      "step": 79944,
      "training_step_time": 7.613206624984741
    },
    {
      "epoch": 5.6695938110351565e-06,
      "model_forward_time": 1.7776212692260742,
      "step": 79945
    },
    {
      "epoch": 5.6695938110351565e-06,
      "step": 79945,
      "training_step_time": 7.624651670455933
    },
    {
      "epoch": 5.670547485351562e-06,
      "model_forward_time": 2.912541389465332,
      "step": 79946
    },
    {
      "epoch": 5.670547485351562e-06,
      "step": 79946,
      "training_step_time": 8.475334882736206
    },
    {
      "epoch": 5.671501159667969e-06,
      "model_forward_time": 1.7735800743103027,
      "step": 79947
    },
    {
      "epoch": 5.671501159667969e-06,
      "step": 79947,
      "training_step_time": 10.689110040664673
    },
    {
      "epoch": 5.672454833984375e-06,
      "model_forward_time": 1.7814881801605225,
      "step": 79948
    },
    {
      "epoch": 5.672454833984375e-06,
      "step": 79948,
      "training_step_time": 9.494450569152832
    },
    {
      "epoch": 5.673408508300781e-06,
      "model_forward_time": 1.7648789882659912,
      "step": 79949
    },
    {
      "epoch": 5.673408508300781e-06,
      "step": 79949,
      "training_step_time": 7.613297939300537
    },
    {
      "epoch": 5.674362182617188e-06,
      "grad_norm": 0.08567382395267487,
      "learning_rate": 1.0594825312966272e-06,
      "loss": 0.052,
      "step": 79950
    },
    {
      "epoch": 5.674362182617188e-06,
      "model_forward_time": 1.7924087047576904,
      "step": 79950
    },
    {
      "dynamics_loss_avg": 0.05072796028107405,
      "epoch": 5.674362182617188e-06,
      "step": 79950
    },
    {
      "action_loss_avg": 0.0014858470676699653,
      "epoch": 5.674362182617188e-06,
      "step": 79950
    },
    {
      "epoch": 5.674362182617188e-06,
      "step": 79950,
      "training_step_time": 7.78063702583313
    },
    {
      "epoch": 5.675315856933593e-06,
      "model_forward_time": 1.4038987159729004,
      "step": 79951
    },
    {
      "epoch": 5.675315856933593e-06,
      "step": 79951,
      "training_step_time": 7.6041100025177
    },
    {
      "epoch": 5.67626953125e-06,
      "model_forward_time": 1.7724964618682861,
      "step": 79952
    },
    {
      "epoch": 5.67626953125e-06,
      "step": 79952,
      "training_step_time": 7.5450475215911865
    },
    {
      "epoch": 5.6772232055664065e-06,
      "model_forward_time": 1.7792401313781738,
      "step": 79953
    },
    {
      "epoch": 5.6772232055664065e-06,
      "step": 79953,
      "training_step_time": 8.599223136901855
    },
    {
      "epoch": 5.678176879882812e-06,
      "model_forward_time": 1.7872889041900635,
      "step": 79954
    },
    {
      "epoch": 5.678176879882812e-06,
      "step": 79954,
      "training_step_time": 7.704570770263672
    },
    {
      "epoch": 5.679130554199219e-06,
      "model_forward_time": 2.8964803218841553,
      "step": 79955
    },
    {
      "epoch": 5.679130554199219e-06,
      "step": 79955,
      "training_step_time": 8.212268114089966
    },
    {
      "epoch": 5.680084228515625e-06,
      "model_forward_time": 1.774742841720581,
      "step": 79956
    },
    {
      "epoch": 5.680084228515625e-06,
      "step": 79956,
      "training_step_time": 7.693894863128662
    },
    {
      "epoch": 5.681037902832031e-06,
      "model_forward_time": 1.776177167892456,
      "step": 79957
    },
    {
      "epoch": 5.681037902832031e-06,
      "step": 79957,
      "training_step_time": 7.671919822692871
    },
    {
      "epoch": 5.681991577148438e-06,
      "model_forward_time": 1.7663404941558838,
      "step": 79958
    },
    {
      "epoch": 5.681991577148438e-06,
      "step": 79958,
      "training_step_time": 7.706119775772095
    },
    {
      "epoch": 5.682945251464843e-06,
      "model_forward_time": 1.7682228088378906,
      "step": 79959
    },
    {
      "epoch": 5.682945251464843e-06,
      "step": 79959,
      "training_step_time": 7.7108681201934814
    },
    {
      "epoch": 5.68389892578125e-06,
      "grad_norm": 0.0786411464214325,
      "learning_rate": 1.0584649654129336e-06,
      "loss": 0.0522,
      "step": 79960
    },
    {
      "epoch": 5.68389892578125e-06,
      "model_forward_time": 1.7756309509277344,
      "step": 79960
    },
    {
      "dynamics_loss_avg": 0.03998813852667808,
      "epoch": 5.68389892578125e-06,
      "step": 79960
    },
    {
      "action_loss_avg": 0.0011474925820948556,
      "epoch": 5.68389892578125e-06,
      "step": 79960
    },
    {
      "epoch": 5.68389892578125e-06,
      "step": 79960,
      "training_step_time": 7.660321474075317
    },
    {
      "epoch": 5.6848526000976565e-06,
      "model_forward_time": 1.7690300941467285,
      "step": 79961
    },
    {
      "epoch": 5.6848526000976565e-06,
      "step": 79961,
      "training_step_time": 8.554524898529053
    },
    {
      "epoch": 5.685806274414062e-06,
      "model_forward_time": 1.7795922756195068,
      "step": 79962
    },
    {
      "epoch": 5.685806274414062e-06,
      "step": 79962,
      "training_step_time": 8.201026916503906
    },
    {
      "epoch": 5.686759948730469e-06,
      "model_forward_time": 1.765746831893921,
      "step": 79963
    },
    {
      "epoch": 5.686759948730469e-06,
      "step": 79963,
      "training_step_time": 7.6123528480529785
    },
    {
      "epoch": 5.687713623046875e-06,
      "model_forward_time": 1.7937464714050293,
      "step": 79964
    },
    {
      "epoch": 5.687713623046875e-06,
      "step": 79964,
      "training_step_time": 9.030872344970703
    },
    {
      "epoch": 5.688667297363281e-06,
      "model_forward_time": 2.7797963619232178,
      "step": 79965
    },
    {
      "epoch": 5.688667297363281e-06,
      "step": 79965,
      "training_step_time": 10.054097890853882
    },
    {
      "epoch": 5.689620971679688e-06,
      "model_forward_time": 1.382934331893921,
      "step": 79966
    },
    {
      "epoch": 5.689620971679688e-06,
      "step": 79966,
      "training_step_time": 7.680452585220337
    },
    {
      "epoch": 5.690574645996093e-06,
      "model_forward_time": 1.7491679191589355,
      "step": 79967
    },
    {
      "epoch": 5.690574645996093e-06,
      "step": 79967,
      "training_step_time": 8.648941278457642
    },
    {
      "epoch": 5.6915283203125e-06,
      "model_forward_time": 1.7541985511779785,
      "step": 79968
    },
    {
      "epoch": 5.6915283203125e-06,
      "step": 79968,
      "training_step_time": 7.633540630340576
    },
    {
      "epoch": 5.6924819946289065e-06,
      "model_forward_time": 1.7540395259857178,
      "step": 79969
    },
    {
      "epoch": 5.6924819946289065e-06,
      "step": 79969,
      "training_step_time": 7.582751035690308
    },
    {
      "epoch": 5.693435668945312e-06,
      "grad_norm": 0.11105693131685257,
      "learning_rate": 1.0574478305695861e-06,
      "loss": 0.0513,
      "step": 79970
    },
    {
      "epoch": 5.693435668945312e-06,
      "model_forward_time": 1.760582685470581,
      "step": 79970
    },
    {
      "dynamics_loss_avg": 0.05118354968726635,
      "epoch": 5.693435668945312e-06,
      "step": 79970
    },
    {
      "action_loss_avg": 0.0010000920476159081,
      "epoch": 5.693435668945312e-06,
      "step": 79970
    },
    {
      "epoch": 5.693435668945312e-06,
      "step": 79970,
      "training_step_time": 8.486258029937744
    },
    {
      "epoch": 5.694389343261719e-06,
      "model_forward_time": 1.3706340789794922,
      "step": 79971
    },
    {
      "epoch": 5.694389343261719e-06,
      "step": 79971,
      "training_step_time": 8.477998733520508
    },
    {
      "epoch": 5.695343017578125e-06,
      "model_forward_time": 1.7550239562988281,
      "step": 79972
    },
    {
      "epoch": 5.695343017578125e-06,
      "step": 79972,
      "training_step_time": 7.733299970626831
    },
    {
      "epoch": 5.696296691894531e-06,
      "model_forward_time": 1.0548412799835205,
      "step": 79973
    },
    {
      "epoch": 5.696296691894531e-06,
      "step": 79973,
      "training_step_time": 7.785918951034546
    },
    {
      "epoch": 5.697250366210938e-06,
      "model_forward_time": 2.811967372894287,
      "step": 79974
    },
    {
      "epoch": 5.697250366210938e-06,
      "step": 79974,
      "training_step_time": 9.486875057220459
    },
    {
      "epoch": 5.698204040527343e-06,
      "model_forward_time": 1.777466058731079,
      "step": 79975
    },
    {
      "epoch": 5.698204040527343e-06,
      "step": 79975,
      "training_step_time": 8.722172498703003
    },
    {
      "epoch": 5.69915771484375e-06,
      "model_forward_time": 1.7854945659637451,
      "step": 79976
    },
    {
      "epoch": 5.69915771484375e-06,
      "step": 79976,
      "training_step_time": 7.6668641567230225
    },
    {
      "epoch": 5.7001113891601565e-06,
      "model_forward_time": 1.770991563796997,
      "step": 79977
    },
    {
      "epoch": 5.7001113891601565e-06,
      "step": 79977,
      "training_step_time": 8.509544134140015
    },
    {
      "epoch": 5.701065063476562e-06,
      "model_forward_time": 1.7694787979125977,
      "step": 79978
    },
    {
      "epoch": 5.701065063476562e-06,
      "step": 79978,
      "training_step_time": 8.581491708755493
    },
    {
      "epoch": 5.702018737792969e-06,
      "model_forward_time": 1.7935194969177246,
      "step": 79979
    },
    {
      "epoch": 5.702018737792969e-06,
      "step": 79979,
      "training_step_time": 8.581341981887817
    },
    {
      "epoch": 5.702972412109375e-06,
      "grad_norm": 0.10511285066604614,
      "learning_rate": 1.0564311268778154e-06,
      "loss": 0.0513,
      "step": 79980
    },
    {
      "epoch": 5.702972412109375e-06,
      "model_forward_time": 1.7782597541809082,
      "step": 79980
    },
    {
      "dynamics_loss_avg": 0.02708194744773209,
      "epoch": 5.702972412109375e-06,
      "step": 79980
    },
    {
      "action_loss_avg": 0.0010090396302985028,
      "epoch": 5.702972412109375e-06,
      "step": 79980
    },
    {
      "epoch": 5.702972412109375e-06,
      "step": 79980,
      "training_step_time": 8.194996356964111
    },
    {
      "epoch": 5.703926086425781e-06,
      "model_forward_time": 1.7722196578979492,
      "step": 79981
    },
    {
      "epoch": 5.703926086425781e-06,
      "step": 79981,
      "training_step_time": 7.651392459869385
    },
    {
      "epoch": 5.704879760742188e-06,
      "model_forward_time": 1.8106234073638916,
      "step": 79982
    },
    {
      "epoch": 5.704879760742188e-06,
      "step": 79982,
      "training_step_time": 7.554394721984863
    },
    {
      "epoch": 5.705833435058593e-06,
      "model_forward_time": 1.7954299449920654,
      "step": 79983
    },
    {
      "epoch": 5.705833435058593e-06,
      "step": 79983,
      "training_step_time": 8.486302614212036
    },
    {
      "epoch": 5.706787109375e-06,
      "model_forward_time": 2.868391990661621,
      "step": 79984
    },
    {
      "epoch": 5.706787109375e-06,
      "step": 79984,
      "training_step_time": 8.180044889450073
    },
    {
      "epoch": 5.7077407836914066e-06,
      "model_forward_time": 1.7749824523925781,
      "step": 79985
    },
    {
      "epoch": 5.7077407836914066e-06,
      "step": 79985,
      "training_step_time": 7.559439659118652
    },
    {
      "epoch": 5.708694458007812e-06,
      "model_forward_time": 1.7853400707244873,
      "step": 79986
    },
    {
      "epoch": 5.708694458007812e-06,
      "step": 79986,
      "training_step_time": 7.661391973495483
    },
    {
      "epoch": 5.709648132324219e-06,
      "model_forward_time": 1.7721962928771973,
      "step": 79987
    },
    {
      "epoch": 5.709648132324219e-06,
      "step": 79987,
      "training_step_time": 7.682605028152466
    },
    {
      "epoch": 5.710601806640625e-06,
      "model_forward_time": 1.7845344543457031,
      "step": 79988
    },
    {
      "epoch": 5.710601806640625e-06,
      "step": 79988,
      "training_step_time": 7.553886413574219
    },
    {
      "epoch": 5.711555480957031e-06,
      "model_forward_time": 1.4096732139587402,
      "step": 79989
    },
    {
      "epoch": 5.711555480957031e-06,
      "step": 79989,
      "training_step_time": 7.580159425735474
    },
    {
      "epoch": 5.712509155273438e-06,
      "grad_norm": 0.06956903636455536,
      "learning_rate": 1.0554148544488074e-06,
      "loss": 0.0503,
      "step": 79990
    },
    {
      "epoch": 5.712509155273438e-06,
      "model_forward_time": 1.7701585292816162,
      "step": 79990
    },
    {
      "dynamics_loss_avg": 0.045607464574277404,
      "epoch": 5.712509155273438e-06,
      "step": 79990
    },
    {
      "action_loss_avg": 0.0010605072951875628,
      "epoch": 5.712509155273438e-06,
      "step": 79990
    },
    {
      "epoch": 5.712509155273438e-06,
      "step": 79990,
      "training_step_time": 7.635572671890259
    },
    {
      "epoch": 5.7134628295898434e-06,
      "model_forward_time": 1.7721121311187744,
      "step": 79991
    },
    {
      "epoch": 5.7134628295898434e-06,
      "step": 79991,
      "training_step_time": 9.157074213027954
    },
    {
      "epoch": 5.71441650390625e-06,
      "model_forward_time": 1.495513677597046,
      "step": 79992
    },
    {
      "epoch": 5.71441650390625e-06,
      "step": 79992,
      "training_step_time": 7.5886147022247314
    },
    {
      "epoch": 5.7153701782226566e-06,
      "model_forward_time": 2.87724232673645,
      "step": 79993
    },
    {
      "epoch": 5.7153701782226566e-06,
      "step": 79993,
      "training_step_time": 8.682526111602783
    },
    {
      "epoch": 5.716323852539062e-06,
      "model_forward_time": 1.7876639366149902,
      "step": 79994
    },
    {
      "epoch": 5.716323852539062e-06,
      "step": 79994,
      "training_step_time": 8.524967193603516
    },
    {
      "epoch": 5.717277526855469e-06,
      "model_forward_time": 1.7607135772705078,
      "step": 79995
    },
    {
      "epoch": 5.717277526855469e-06,
      "step": 79995,
      "training_step_time": 9.257779359817505
    },
    {
      "epoch": 5.718231201171875e-06,
      "model_forward_time": 1.7780284881591797,
      "step": 79996
    },
    {
      "epoch": 5.718231201171875e-06,
      "step": 79996,
      "training_step_time": 7.587889194488525
    },
    {
      "epoch": 5.719184875488281e-06,
      "model_forward_time": 1.775698184967041,
      "step": 79997
    },
    {
      "epoch": 5.719184875488281e-06,
      "step": 79997,
      "training_step_time": 7.601258754730225
    },
    {
      "epoch": 5.720138549804688e-06,
      "model_forward_time": 1.8217952251434326,
      "step": 79998
    },
    {
      "epoch": 5.720138549804688e-06,
      "step": 79998,
      "training_step_time": 7.886819362640381
    },
    {
      "epoch": 5.7210922241210934e-06,
      "model_forward_time": 1.867546558380127,
      "step": 79999
    },
    {
      "epoch": 5.7210922241210934e-06,
      "step": 79999,
      "training_step_time": 8.75196385383606
    },
    {
      "epoch": 5.7220458984375e-06,
      "grad_norm": 0.07752849161624908,
      "learning_rate": 1.0543990133937004e-06,
      "loss": 0.0518,
      "step": 80000
    },
    {
      "epoch": 5.7220458984375e-06,
      "model_forward_time": 1.817929983139038,
      "step": 80000
    },
    {
      "dynamics_loss_avg": 0.05441694129258394,
      "epoch": 5.7220458984375e-06,
      "step": 80000
    },
    {
      "action_loss_avg": 0.0012472164235077798,
      "epoch": 5.7220458984375e-06,
      "step": 80000
    },
    {
      "epoch": 5.7220458984375e-06,
      "step": 80000,
      "training_step_time": 9.9651780128479
    },
    {
      "epoch": 5.7229995727539066e-06,
      "model_forward_time": 3.516993761062622,
      "step": 80001
    },
    {
      "epoch": 5.7229995727539066e-06,
      "step": 80001,
      "training_step_time": 12.077656269073486
    },
    {
      "epoch": 5.723953247070312e-06,
      "model_forward_time": 1.8309352397918701,
      "step": 80002
    },
    {
      "epoch": 5.723953247070312e-06,
      "step": 80002,
      "training_step_time": 9.687427997589111
    },
    {
      "epoch": 5.724906921386719e-06,
      "model_forward_time": 1.7703301906585693,
      "step": 80003
    },
    {
      "epoch": 5.724906921386719e-06,
      "step": 80003,
      "training_step_time": 9.551625490188599
    },
    {
      "epoch": 5.725860595703125e-06,
      "model_forward_time": 1.767528772354126,
      "step": 80004
    },
    {
      "epoch": 5.725860595703125e-06,
      "step": 80004,
      "training_step_time": 9.35060739517212
    },
    {
      "epoch": 5.726814270019531e-06,
      "model_forward_time": 1.779139757156372,
      "step": 80005
    },
    {
      "epoch": 5.726814270019531e-06,
      "step": 80005,
      "training_step_time": 8.372849225997925
    },
    {
      "epoch": 5.727767944335938e-06,
      "model_forward_time": 1.771867036819458,
      "step": 80006
    },
    {
      "epoch": 5.727767944335938e-06,
      "step": 80006,
      "training_step_time": 9.204945802688599
    },
    {
      "epoch": 5.7287216186523434e-06,
      "model_forward_time": 1.7830839157104492,
      "step": 80007
    },
    {
      "epoch": 5.7287216186523434e-06,
      "step": 80007,
      "training_step_time": 9.88358211517334
    },
    {
      "epoch": 5.72967529296875e-06,
      "model_forward_time": 1.7639918327331543,
      "step": 80008
    },
    {
      "epoch": 5.72967529296875e-06,
      "step": 80008,
      "training_step_time": 7.629992723464966
    },
    {
      "epoch": 5.7306289672851566e-06,
      "model_forward_time": 1.7741875648498535,
      "step": 80009
    },
    {
      "epoch": 5.7306289672851566e-06,
      "step": 80009,
      "training_step_time": 8.424159049987793
    },
    {
      "epoch": 5.731582641601562e-06,
      "grad_norm": 0.0868564248085022,
      "learning_rate": 1.0533836038235867e-06,
      "loss": 0.0524,
      "step": 80010
    },
    {
      "epoch": 5.731582641601562e-06,
      "model_forward_time": 1.8316373825073242,
      "step": 80010
    },
    {
      "dynamics_loss_avg": 0.04631927348673344,
      "epoch": 5.731582641601562e-06,
      "step": 80010
    },
    {
      "action_loss_avg": 0.0009333333815447986,
      "epoch": 5.731582641601562e-06,
      "step": 80010
    },
    {
      "epoch": 5.731582641601562e-06,
      "step": 80010,
      "training_step_time": 9.399860143661499
    },
    {
      "epoch": 5.732536315917969e-06,
      "model_forward_time": 2.8239126205444336,
      "step": 80011
    },
    {
      "epoch": 5.732536315917969e-06,
      "step": 80011,
      "training_step_time": 8.667238712310791
    },
    {
      "epoch": 5.733489990234375e-06,
      "model_forward_time": 1.7615363597869873,
      "step": 80012
    },
    {
      "epoch": 5.733489990234375e-06,
      "step": 80012,
      "training_step_time": 8.174069166183472
    },
    {
      "epoch": 5.734443664550781e-06,
      "model_forward_time": 1.7674646377563477,
      "step": 80013
    },
    {
      "epoch": 5.734443664550781e-06,
      "step": 80013,
      "training_step_time": 7.643088340759277
    },
    {
      "epoch": 5.735397338867188e-06,
      "model_forward_time": 1.7716028690338135,
      "step": 80014
    },
    {
      "epoch": 5.735397338867188e-06,
      "step": 80014,
      "training_step_time": 8.155835390090942
    },
    {
      "epoch": 5.7363510131835934e-06,
      "model_forward_time": 1.7665541172027588,
      "step": 80015
    },
    {
      "epoch": 5.7363510131835934e-06,
      "step": 80015,
      "training_step_time": 9.584898948669434
    },
    {
      "epoch": 5.7373046875e-06,
      "model_forward_time": 1.772657871246338,
      "step": 80016
    },
    {
      "epoch": 5.7373046875e-06,
      "step": 80016,
      "training_step_time": 7.605120420455933
    },
    {
      "epoch": 5.738258361816407e-06,
      "model_forward_time": 1.7582144737243652,
      "step": 80017
    },
    {
      "epoch": 5.738258361816407e-06,
      "step": 80017,
      "training_step_time": 7.573926687240601
    },
    {
      "epoch": 5.739212036132812e-06,
      "model_forward_time": 1.7602949142456055,
      "step": 80018
    },
    {
      "epoch": 5.739212036132812e-06,
      "step": 80018,
      "training_step_time": 7.605677843093872
    },
    {
      "epoch": 5.740165710449219e-06,
      "model_forward_time": 1.8274571895599365,
      "step": 80019
    },
    {
      "epoch": 5.740165710449219e-06,
      "step": 80019,
      "training_step_time": 8.091867685317993
    },
    {
      "epoch": 5.741119384765625e-06,
      "grad_norm": 0.0814398005604744,
      "learning_rate": 1.052368625849507e-06,
      "loss": 0.0526,
      "step": 80020
    },
    {
      "epoch": 5.741119384765625e-06,
      "model_forward_time": 2.8919317722320557,
      "step": 80020
    },
    {
      "dynamics_loss_avg": 0.06072060223668814,
      "epoch": 5.741119384765625e-06,
      "step": 80020
    },
    {
      "action_loss_avg": 0.0009737114043673501,
      "epoch": 5.741119384765625e-06,
      "step": 80020
    },
    {
      "epoch": 5.741119384765625e-06,
      "step": 80020,
      "training_step_time": 8.593534469604492
    },
    {
      "epoch": 5.742073059082031e-06,
      "model_forward_time": 1.0782356262207031,
      "step": 80021
    },
    {
      "epoch": 5.742073059082031e-06,
      "step": 80021,
      "training_step_time": 8.755549907684326
    },
    {
      "epoch": 5.743026733398438e-06,
      "model_forward_time": 1.7821581363677979,
      "step": 80022
    },
    {
      "epoch": 5.743026733398438e-06,
      "step": 80022,
      "training_step_time": 8.241579294204712
    },
    {
      "epoch": 5.7439804077148435e-06,
      "model_forward_time": 1.7690010070800781,
      "step": 80023
    },
    {
      "epoch": 5.7439804077148435e-06,
      "step": 80023,
      "training_step_time": 7.657333135604858
    },
    {
      "epoch": 5.74493408203125e-06,
      "model_forward_time": 1.7747950553894043,
      "step": 80024
    },
    {
      "epoch": 5.74493408203125e-06,
      "step": 80024,
      "training_step_time": 11.525720596313477
    },
    {
      "epoch": 5.745887756347657e-06,
      "model_forward_time": 1.771850824356079,
      "step": 80025
    },
    {
      "epoch": 5.745887756347657e-06,
      "step": 80025,
      "training_step_time": 7.5539209842681885
    },
    {
      "epoch": 5.746841430664062e-06,
      "model_forward_time": 1.7862660884857178,
      "step": 80026
    },
    {
      "epoch": 5.746841430664062e-06,
      "step": 80026,
      "training_step_time": 7.515819549560547
    },
    {
      "epoch": 5.747795104980469e-06,
      "model_forward_time": 1.770071029663086,
      "step": 80027
    },
    {
      "epoch": 5.747795104980469e-06,
      "step": 80027,
      "training_step_time": 7.869213581085205
    },
    {
      "epoch": 5.748748779296875e-06,
      "model_forward_time": 1.766345739364624,
      "step": 80028
    },
    {
      "epoch": 5.748748779296875e-06,
      "step": 80028,
      "training_step_time": 7.454125642776489
    },
    {
      "epoch": 5.749702453613281e-06,
      "model_forward_time": 1.7706425189971924,
      "step": 80029
    },
    {
      "epoch": 5.749702453613281e-06,
      "step": 80029,
      "training_step_time": 6.621532440185547
    },
    {
      "epoch": 5.750656127929688e-06,
      "grad_norm": 0.07465948164463043,
      "learning_rate": 1.05135407958246e-06,
      "loss": 0.0527,
      "step": 80030
    },
    {
      "epoch": 5.750656127929688e-06,
      "model_forward_time": 1.7757513523101807,
      "step": 80030
    },
    {
      "dynamics_loss_avg": 0.03671453343704343,
      "epoch": 5.750656127929688e-06,
      "step": 80030
    },
    {
      "action_loss_avg": 0.001514170423615724,
      "epoch": 5.750656127929688e-06,
      "step": 80030
    },
    {
      "epoch": 5.750656127929688e-06,
      "step": 80030,
      "training_step_time": 6.596248388290405
    },
    {
      "epoch": 5.7516098022460935e-06,
      "model_forward_time": 1.7655808925628662,
      "step": 80031
    },
    {
      "epoch": 5.7516098022460935e-06,
      "step": 80031,
      "training_step_time": 6.664283752441406
    },
    {
      "epoch": 5.7525634765625e-06,
      "model_forward_time": 1.7679507732391357,
      "step": 80032
    },
    {
      "epoch": 5.7525634765625e-06,
      "step": 80032,
      "training_step_time": 6.642999887466431
    },
    {
      "epoch": 5.753517150878907e-06,
      "model_forward_time": 1.776848554611206,
      "step": 80033
    },
    {
      "epoch": 5.753517150878907e-06,
      "step": 80033,
      "training_step_time": 6.6089770793914795
    },
    {
      "epoch": 5.754470825195312e-06,
      "model_forward_time": 2.817004680633545,
      "step": 80034
    },
    {
      "epoch": 5.754470825195312e-06,
      "step": 80034,
      "training_step_time": 7.602149724960327
    },
    {
      "epoch": 5.755424499511719e-06,
      "model_forward_time": 1.7610492706298828,
      "step": 80035
    },
    {
      "epoch": 5.755424499511719e-06,
      "step": 80035,
      "training_step_time": 7.597605228424072
    },
    {
      "epoch": 5.756378173828125e-06,
      "model_forward_time": 1.7621893882751465,
      "step": 80036
    },
    {
      "epoch": 5.756378173828125e-06,
      "step": 80036,
      "training_step_time": 6.622533321380615
    },
    {
      "epoch": 5.757331848144531e-06,
      "model_forward_time": 1.7554731369018555,
      "step": 80037
    },
    {
      "epoch": 5.757331848144531e-06,
      "step": 80037,
      "training_step_time": 7.48111629486084
    },
    {
      "epoch": 5.758285522460938e-06,
      "model_forward_time": 1.7635350227355957,
      "step": 80038
    },
    {
      "epoch": 5.758285522460938e-06,
      "step": 80038,
      "training_step_time": 7.335486650466919
    },
    {
      "epoch": 5.7592391967773435e-06,
      "model_forward_time": 1.7664387226104736,
      "step": 80039
    },
    {
      "epoch": 5.7592391967773435e-06,
      "step": 80039,
      "training_step_time": 6.6342668533325195
    },
    {
      "epoch": 5.76019287109375e-06,
      "grad_norm": 0.1224457249045372,
      "learning_rate": 1.0503399651333934e-06,
      "loss": 0.0531,
      "step": 80040
    },
    {
      "epoch": 5.76019287109375e-06,
      "model_forward_time": 1.7275238037109375,
      "step": 80040
    },
    {
      "dynamics_loss_avg": 0.03518057744950056,
      "epoch": 5.76019287109375e-06,
      "step": 80040
    },
    {
      "action_loss_avg": 0.0010268561891280115,
      "epoch": 5.76019287109375e-06,
      "step": 80040
    },
    {
      "epoch": 5.76019287109375e-06,
      "step": 80040,
      "training_step_time": 6.607854843139648
    },
    {
      "epoch": 5.761146545410157e-06,
      "model_forward_time": 1.7458202838897705,
      "step": 80041
    },
    {
      "epoch": 5.761146545410157e-06,
      "step": 80041,
      "training_step_time": 6.606245756149292
    },
    {
      "epoch": 5.762100219726562e-06,
      "model_forward_time": 1.7357113361358643,
      "step": 80042
    },
    {
      "epoch": 5.762100219726562e-06,
      "step": 80042,
      "training_step_time": 6.605984926223755
    },
    {
      "epoch": 5.763053894042969e-06,
      "model_forward_time": 2.3909029960632324,
      "step": 80043
    },
    {
      "epoch": 5.763053894042969e-06,
      "step": 80043,
      "training_step_time": 7.6263957023620605
    },
    {
      "epoch": 5.764007568359375e-06,
      "model_forward_time": 1.737614393234253,
      "step": 80044
    },
    {
      "epoch": 5.764007568359375e-06,
      "step": 80044,
      "training_step_time": 7.6178672313690186
    },
    {
      "epoch": 5.764961242675781e-06,
      "model_forward_time": 1.7364237308502197,
      "step": 80045
    },
    {
      "epoch": 5.764961242675781e-06,
      "step": 80045,
      "training_step_time": 6.59691047668457
    },
    {
      "epoch": 5.765914916992188e-06,
      "model_forward_time": 1.7426588535308838,
      "step": 80046
    },
    {
      "epoch": 5.765914916992188e-06,
      "step": 80046,
      "training_step_time": 7.489466905593872
    },
    {
      "epoch": 5.7668685913085935e-06,
      "model_forward_time": 1.7341179847717285,
      "step": 80047
    },
    {
      "epoch": 5.7668685913085935e-06,
      "step": 80047,
      "training_step_time": 7.342470407485962
    },
    {
      "epoch": 5.767822265625e-06,
      "model_forward_time": 1.7385210990905762,
      "step": 80048
    },
    {
      "epoch": 5.767822265625e-06,
      "step": 80048,
      "training_step_time": 6.597292184829712
    },
    {
      "epoch": 5.768775939941407e-06,
      "model_forward_time": 1.7407982349395752,
      "step": 80049
    },
    {
      "epoch": 5.768775939941407e-06,
      "step": 80049,
      "training_step_time": 6.621973752975464
    },
    {
      "epoch": 5.769729614257812e-06,
      "grad_norm": 0.10640759766101837,
      "learning_rate": 1.04932628261321e-06,
      "loss": 0.0545,
      "step": 80050
    },
    {
      "epoch": 5.769729614257812e-06,
      "model_forward_time": 1.7398285865783691,
      "step": 80050
    },
    {
      "dynamics_loss_avg": 0.04844520352780819,
      "epoch": 5.769729614257812e-06,
      "step": 80050
    },
    {
      "action_loss_avg": 0.0009387838304974139,
      "epoch": 5.769729614257812e-06,
      "step": 80050
    },
    {
      "epoch": 5.769729614257812e-06,
      "step": 80050,
      "training_step_time": 6.590822458267212
    },
    {
      "epoch": 5.770683288574219e-06,
      "model_forward_time": 1.7347490787506104,
      "step": 80051
    },
    {
      "epoch": 5.770683288574219e-06,
      "step": 80051,
      "training_step_time": 6.6233580112457275
    },
    {
      "epoch": 5.771636962890625e-06,
      "model_forward_time": 2.013483762741089,
      "step": 80052
    },
    {
      "epoch": 5.771636962890625e-06,
      "step": 80052,
      "training_step_time": 7.65727424621582
    },
    {
      "epoch": 5.772590637207031e-06,
      "model_forward_time": 1.7332079410552979,
      "step": 80053
    },
    {
      "epoch": 5.772590637207031e-06,
      "step": 80053,
      "training_step_time": 7.626170635223389
    },
    {
      "epoch": 5.773544311523438e-06,
      "model_forward_time": 1.7383530139923096,
      "step": 80054
    },
    {
      "epoch": 5.773544311523438e-06,
      "step": 80054,
      "training_step_time": 6.614802360534668
    },
    {
      "epoch": 5.7744979858398435e-06,
      "model_forward_time": 1.738901138305664,
      "step": 80055
    },
    {
      "epoch": 5.7744979858398435e-06,
      "step": 80055,
      "training_step_time": 8.342985153198242
    },
    {
      "epoch": 5.77545166015625e-06,
      "model_forward_time": 1.7439477443695068,
      "step": 80056
    },
    {
      "epoch": 5.77545166015625e-06,
      "step": 80056,
      "training_step_time": 7.417367219924927
    },
    {
      "epoch": 5.776405334472657e-06,
      "model_forward_time": 1.7389588356018066,
      "step": 80057
    },
    {
      "epoch": 5.776405334472657e-06,
      "step": 80057,
      "training_step_time": 6.597723722457886
    },
    {
      "epoch": 5.777359008789062e-06,
      "model_forward_time": 1.7471842765808105,
      "step": 80058
    },
    {
      "epoch": 5.777359008789062e-06,
      "step": 80058,
      "training_step_time": 6.613466262817383
    },
    {
      "epoch": 5.778312683105469e-06,
      "model_forward_time": 1.742008924484253,
      "step": 80059
    },
    {
      "epoch": 5.778312683105469e-06,
      "step": 80059,
      "training_step_time": 6.666777610778809
    },
    {
      "epoch": 5.779266357421875e-06,
      "grad_norm": 0.07099978625774384,
      "learning_rate": 1.0483130321327644e-06,
      "loss": 0.0545,
      "step": 80060
    },
    {
      "epoch": 5.779266357421875e-06,
      "model_forward_time": 1.7384355068206787,
      "step": 80060
    },
    {
      "dynamics_loss_avg": 0.03074677465483546,
      "epoch": 5.779266357421875e-06,
      "step": 80060
    },
    {
      "action_loss_avg": 0.0007791613694280386,
      "epoch": 5.779266357421875e-06,
      "step": 80060
    },
    {
      "epoch": 5.779266357421875e-06,
      "step": 80060,
      "training_step_time": 6.603266954421997
    },
    {
      "epoch": 5.780220031738281e-06,
      "model_forward_time": 2.4247725009918213,
      "step": 80061
    },
    {
      "epoch": 5.780220031738281e-06,
      "step": 80061,
      "training_step_time": 10.161782503128052
    },
    {
      "epoch": 5.781173706054688e-06,
      "model_forward_time": 1.7393112182617188,
      "step": 80062
    },
    {
      "epoch": 5.781173706054688e-06,
      "step": 80062,
      "training_step_time": 7.546501159667969
    },
    {
      "epoch": 5.7821273803710935e-06,
      "model_forward_time": 1.7381868362426758,
      "step": 80063
    },
    {
      "epoch": 5.7821273803710935e-06,
      "step": 80063,
      "training_step_time": 7.627074718475342
    },
    {
      "epoch": 5.7830810546875e-06,
      "model_forward_time": 1.7397491931915283,
      "step": 80064
    },
    {
      "epoch": 5.7830810546875e-06,
      "step": 80064,
      "training_step_time": 7.515238046646118
    },
    {
      "epoch": 5.784034729003907e-06,
      "model_forward_time": 1.738896131515503,
      "step": 80065
    },
    {
      "epoch": 5.784034729003907e-06,
      "step": 80065,
      "training_step_time": 7.538883686065674
    },
    {
      "epoch": 5.784988403320312e-06,
      "model_forward_time": 1.740297555923462,
      "step": 80066
    },
    {
      "epoch": 5.784988403320312e-06,
      "step": 80066,
      "training_step_time": 7.424611568450928
    },
    {
      "epoch": 5.785942077636719e-06,
      "model_forward_time": 1.734527349472046,
      "step": 80067
    },
    {
      "epoch": 5.785942077636719e-06,
      "step": 80067,
      "training_step_time": 6.6534037590026855
    },
    {
      "epoch": 5.786895751953125e-06,
      "model_forward_time": 1.735539436340332,
      "step": 80068
    },
    {
      "epoch": 5.786895751953125e-06,
      "step": 80068,
      "training_step_time": 6.635409116744995
    },
    {
      "epoch": 5.787849426269531e-06,
      "model_forward_time": 1.7380273342132568,
      "step": 80069
    },
    {
      "epoch": 5.787849426269531e-06,
      "step": 80069,
      "training_step_time": 6.639900207519531
    },
    {
      "epoch": 5.788803100585938e-06,
      "grad_norm": 0.08859258890151978,
      "learning_rate": 1.0473002138028654e-06,
      "loss": 0.0544,
      "step": 80070
    },
    {
      "epoch": 5.788803100585938e-06,
      "model_forward_time": 2.3450214862823486,
      "step": 80070
    },
    {
      "dynamics_loss_avg": 0.056737757101655004,
      "epoch": 5.788803100585938e-06,
      "step": 80070
    },
    {
      "action_loss_avg": 0.001740040711592883,
      "epoch": 5.788803100585938e-06,
      "step": 80070
    },
    {
      "epoch": 5.788803100585938e-06,
      "step": 80070,
      "training_step_time": 10.384206056594849
    },
    {
      "epoch": 5.7897567749023435e-06,
      "model_forward_time": 1.391758918762207,
      "step": 80071
    },
    {
      "epoch": 5.7897567749023435e-06,
      "step": 80071,
      "training_step_time": 7.695218324661255
    },
    {
      "epoch": 5.79071044921875e-06,
      "model_forward_time": 1.7407171726226807,
      "step": 80072
    },
    {
      "epoch": 5.79071044921875e-06,
      "step": 80072,
      "training_step_time": 7.652293682098389
    },
    {
      "epoch": 5.791664123535157e-06,
      "model_forward_time": 1.742870807647705,
      "step": 80073
    },
    {
      "epoch": 5.791664123535157e-06,
      "step": 80073,
      "training_step_time": 7.486301898956299
    },
    {
      "epoch": 5.792617797851562e-06,
      "model_forward_time": 1.7311267852783203,
      "step": 80074
    },
    {
      "epoch": 5.792617797851562e-06,
      "step": 80074,
      "training_step_time": 7.537650108337402
    },
    {
      "epoch": 5.793571472167969e-06,
      "model_forward_time": 1.7317488193511963,
      "step": 80075
    },
    {
      "epoch": 5.793571472167969e-06,
      "step": 80075,
      "training_step_time": 7.429338693618774
    },
    {
      "epoch": 5.794525146484375e-06,
      "model_forward_time": 1.7346501350402832,
      "step": 80076
    },
    {
      "epoch": 5.794525146484375e-06,
      "step": 80076,
      "training_step_time": 6.67271089553833
    },
    {
      "epoch": 5.795478820800781e-06,
      "model_forward_time": 1.7378320693969727,
      "step": 80077
    },
    {
      "epoch": 5.795478820800781e-06,
      "step": 80077,
      "training_step_time": 6.699524641036987
    },
    {
      "epoch": 5.796432495117188e-06,
      "model_forward_time": 1.7448396682739258,
      "step": 80078
    },
    {
      "epoch": 5.796432495117188e-06,
      "step": 80078,
      "training_step_time": 6.667128562927246
    },
    {
      "epoch": 5.7973861694335935e-06,
      "model_forward_time": 1.7424304485321045,
      "step": 80079
    },
    {
      "epoch": 5.7973861694335935e-06,
      "step": 80079,
      "training_step_time": 11.331403255462646
    },
    {
      "epoch": 5.79833984375e-06,
      "grad_norm": 0.08705577999353409,
      "learning_rate": 1.0462878277342713e-06,
      "loss": 0.0561,
      "step": 80080
    },
    {
      "epoch": 5.79833984375e-06,
      "model_forward_time": 2.383362293243408,
      "step": 80080
    },
    {
      "dynamics_loss_avg": 0.06318862549960613,
      "epoch": 5.79833984375e-06,
      "step": 80080
    },
    {
      "action_loss_avg": 0.0016598968184553086,
      "epoch": 5.79833984375e-06,
      "step": 80080
    },
    {
      "epoch": 5.79833984375e-06,
      "step": 80080,
      "training_step_time": 8.4151029586792
    },
    {
      "epoch": 5.799293518066407e-06,
      "model_forward_time": 1.734618902206421,
      "step": 80081
    },
    {
      "epoch": 5.799293518066407e-06,
      "step": 80081,
      "training_step_time": 7.69002628326416
    },
    {
      "epoch": 5.800247192382812e-06,
      "model_forward_time": 1.730515480041504,
      "step": 80082
    },
    {
      "epoch": 5.800247192382812e-06,
      "step": 80082,
      "training_step_time": 7.452332496643066
    },
    {
      "epoch": 5.801200866699219e-06,
      "model_forward_time": 1.7439205646514893,
      "step": 80083
    },
    {
      "epoch": 5.801200866699219e-06,
      "step": 80083,
      "training_step_time": 7.6518096923828125
    },
    {
      "epoch": 5.802154541015625e-06,
      "model_forward_time": 1.7424745559692383,
      "step": 80084
    },
    {
      "epoch": 5.802154541015625e-06,
      "step": 80084,
      "training_step_time": 7.494742155075073
    },
    {
      "epoch": 5.803108215332031e-06,
      "model_forward_time": 1.741635799407959,
      "step": 80085
    },
    {
      "epoch": 5.803108215332031e-06,
      "step": 80085,
      "training_step_time": 6.694830656051636
    },
    {
      "epoch": 5.804061889648438e-06,
      "model_forward_time": 1.7365574836730957,
      "step": 80086
    },
    {
      "epoch": 5.804061889648438e-06,
      "step": 80086,
      "training_step_time": 6.670454263687134
    },
    {
      "epoch": 5.8050155639648435e-06,
      "model_forward_time": 1.7462997436523438,
      "step": 80087
    },
    {
      "epoch": 5.8050155639648435e-06,
      "step": 80087,
      "training_step_time": 6.671785116195679
    },
    {
      "epoch": 5.80596923828125e-06,
      "model_forward_time": 1.7419145107269287,
      "step": 80088
    },
    {
      "epoch": 5.80596923828125e-06,
      "step": 80088,
      "training_step_time": 9.723889827728271
    },
    {
      "epoch": 5.806922912597657e-06,
      "model_forward_time": 2.380622386932373,
      "step": 80089
    },
    {
      "epoch": 5.806922912597657e-06,
      "step": 80089,
      "training_step_time": 9.453901529312134
    },
    {
      "epoch": 5.807876586914062e-06,
      "grad_norm": 0.3562670350074768,
      "learning_rate": 1.0452758740376973e-06,
      "loss": 0.0576,
      "step": 80090
    },
    {
      "epoch": 5.807876586914062e-06,
      "model_forward_time": 1.7398715019226074,
      "step": 80090
    },
    {
      "dynamics_loss_avg": 0.04784928010776639,
      "epoch": 5.807876586914062e-06,
      "step": 80090
    },
    {
      "action_loss_avg": 0.0010065835318528115,
      "epoch": 5.807876586914062e-06,
      "step": 80090
    },
    {
      "epoch": 5.807876586914062e-06,
      "step": 80090,
      "training_step_time": 7.631947994232178
    },
    {
      "epoch": 5.808830261230469e-06,
      "model_forward_time": 1.739227294921875,
      "step": 80091
    },
    {
      "epoch": 5.808830261230469e-06,
      "step": 80091,
      "training_step_time": 7.404630899429321
    },
    {
      "epoch": 5.809783935546875e-06,
      "model_forward_time": 1.7367396354675293,
      "step": 80092
    },
    {
      "epoch": 5.809783935546875e-06,
      "step": 80092,
      "training_step_time": 7.6153833866119385
    },
    {
      "epoch": 5.810737609863281e-06,
      "model_forward_time": 1.7350456714630127,
      "step": 80093
    },
    {
      "epoch": 5.810737609863281e-06,
      "step": 80093,
      "training_step_time": 7.409759521484375
    },
    {
      "epoch": 5.811691284179688e-06,
      "model_forward_time": 1.7474663257598877,
      "step": 80094
    },
    {
      "epoch": 5.811691284179688e-06,
      "step": 80094,
      "training_step_time": 6.6437060832977295
    },
    {
      "epoch": 5.8126449584960935e-06,
      "model_forward_time": 1.7427191734313965,
      "step": 80095
    },
    {
      "epoch": 5.8126449584960935e-06,
      "step": 80095,
      "training_step_time": 6.641961574554443
    },
    {
      "epoch": 5.8135986328125e-06,
      "model_forward_time": 1.7437279224395752,
      "step": 80096
    },
    {
      "epoch": 5.8135986328125e-06,
      "step": 80096,
      "training_step_time": 6.742868185043335
    },
    {
      "epoch": 5.814552307128907e-06,
      "model_forward_time": 1.7330341339111328,
      "step": 80097
    },
    {
      "epoch": 5.814552307128907e-06,
      "step": 80097,
      "training_step_time": 8.673245429992676
    },
    {
      "epoch": 5.815505981445312e-06,
      "model_forward_time": 2.3785812854766846,
      "step": 80098
    },
    {
      "epoch": 5.815505981445312e-06,
      "step": 80098,
      "training_step_time": 9.127280712127686
    },
    {
      "epoch": 5.816459655761719e-06,
      "model_forward_time": 1.7332193851470947,
      "step": 80099
    },
    {
      "epoch": 5.816459655761719e-06,
      "step": 80099,
      "training_step_time": 7.7132110595703125
    },
    {
      "epoch": 5.817413330078125e-06,
      "grad_norm": 0.19986461102962494,
      "learning_rate": 1.0442643528238066e-06,
      "loss": 0.056,
      "step": 80100
    },
    {
      "epoch": 5.817413330078125e-06,
      "model_forward_time": 1.7360882759094238,
      "step": 80100
    },
    {
      "dynamics_loss_avg": 0.05955933500081301,
      "epoch": 5.817413330078125e-06,
      "step": 80100
    },
    {
      "action_loss_avg": 0.001066940149758011,
      "epoch": 5.817413330078125e-06,
      "step": 80100
    },
    {
      "epoch": 5.817413330078125e-06,
      "step": 80100,
      "training_step_time": 8.154648542404175
    },
    {
      "epoch": 5.818367004394531e-06,
      "model_forward_time": 1.7370679378509521,
      "step": 80101
    },
    {
      "epoch": 5.818367004394531e-06,
      "step": 80101,
      "training_step_time": 7.545556545257568
    },
    {
      "epoch": 5.819320678710938e-06,
      "model_forward_time": 1.7364625930786133,
      "step": 80102
    },
    {
      "epoch": 5.819320678710938e-06,
      "step": 80102,
      "training_step_time": 7.9571373462677
    },
    {
      "epoch": 5.8202743530273435e-06,
      "model_forward_time": 1.740173578262329,
      "step": 80103
    },
    {
      "epoch": 5.8202743530273435e-06,
      "step": 80103,
      "training_step_time": 6.65201473236084
    },
    {
      "epoch": 5.82122802734375e-06,
      "model_forward_time": 1.7421433925628662,
      "step": 80104
    },
    {
      "epoch": 5.82122802734375e-06,
      "step": 80104,
      "training_step_time": 6.678769588470459
    },
    {
      "epoch": 5.822181701660157e-06,
      "model_forward_time": 1.7412827014923096,
      "step": 80105
    },
    {
      "epoch": 5.822181701660157e-06,
      "step": 80105,
      "training_step_time": 6.652657747268677
    },
    {
      "epoch": 5.823135375976562e-06,
      "model_forward_time": 1.738651990890503,
      "step": 80106
    },
    {
      "epoch": 5.823135375976562e-06,
      "step": 80106,
      "training_step_time": 7.708693981170654
    },
    {
      "epoch": 5.824089050292969e-06,
      "model_forward_time": 2.3807826042175293,
      "step": 80107
    },
    {
      "epoch": 5.824089050292969e-06,
      "step": 80107,
      "training_step_time": 10.574450254440308
    },
    {
      "epoch": 5.825042724609375e-06,
      "model_forward_time": 1.7393848896026611,
      "step": 80108
    },
    {
      "epoch": 5.825042724609375e-06,
      "step": 80108,
      "training_step_time": 9.663257360458374
    },
    {
      "epoch": 5.825996398925781e-06,
      "model_forward_time": 1.7359991073608398,
      "step": 80109
    },
    {
      "epoch": 5.825996398925781e-06,
      "step": 80109,
      "training_step_time": 7.60564923286438
    },
    {
      "epoch": 5.826950073242188e-06,
      "grad_norm": 0.10163046419620514,
      "learning_rate": 1.043253264203219e-06,
      "loss": 0.0563,
      "step": 80110
    },
    {
      "epoch": 5.826950073242188e-06,
      "model_forward_time": 1.7364492416381836,
      "step": 80110
    },
    {
      "dynamics_loss_avg": 0.05195640157908201,
      "epoch": 5.826950073242188e-06,
      "step": 80110
    },
    {
      "action_loss_avg": 0.0007890495529863983,
      "epoch": 5.826950073242188e-06,
      "step": 80110
    },
    {
      "epoch": 5.826950073242188e-06,
      "step": 80110,
      "training_step_time": 7.461356163024902
    },
    {
      "epoch": 5.8279037475585935e-06,
      "model_forward_time": 1.7463655471801758,
      "step": 80111
    },
    {
      "epoch": 5.8279037475585935e-06,
      "step": 80111,
      "training_step_time": 7.58133864402771
    },
    {
      "epoch": 5.828857421875e-06,
      "model_forward_time": 1.7461693286895752,
      "step": 80112
    },
    {
      "epoch": 5.828857421875e-06,
      "step": 80112,
      "training_step_time": 7.3526999950408936
    },
    {
      "epoch": 5.829811096191407e-06,
      "model_forward_time": 1.038527488708496,
      "step": 80113
    },
    {
      "epoch": 5.829811096191407e-06,
      "step": 80113,
      "training_step_time": 6.667402029037476
    },
    {
      "epoch": 5.830764770507812e-06,
      "model_forward_time": 1.736767292022705,
      "step": 80114
    },
    {
      "epoch": 5.830764770507812e-06,
      "step": 80114,
      "training_step_time": 6.663185119628906
    },
    {
      "epoch": 5.831718444824219e-06,
      "model_forward_time": 1.7425792217254639,
      "step": 80115
    },
    {
      "epoch": 5.831718444824219e-06,
      "step": 80115,
      "training_step_time": 7.664088249206543
    },
    {
      "epoch": 5.832672119140625e-06,
      "model_forward_time": 1.7671089172363281,
      "step": 80116
    },
    {
      "epoch": 5.832672119140625e-06,
      "step": 80116,
      "training_step_time": 9.692392587661743
    },
    {
      "epoch": 5.833625793457031e-06,
      "model_forward_time": 2.3677940368652344,
      "step": 80117
    },
    {
      "epoch": 5.833625793457031e-06,
      "step": 80117,
      "training_step_time": 7.750956058502197
    },
    {
      "epoch": 5.834579467773438e-06,
      "model_forward_time": 1.7444429397583008,
      "step": 80118
    },
    {
      "epoch": 5.834579467773438e-06,
      "step": 80118,
      "training_step_time": 7.685774564743042
    },
    {
      "epoch": 5.8355331420898435e-06,
      "model_forward_time": 1.7355716228485107,
      "step": 80119
    },
    {
      "epoch": 5.8355331420898435e-06,
      "step": 80119,
      "training_step_time": 7.429813861846924
    },
    {
      "epoch": 5.83648681640625e-06,
      "grad_norm": 0.1143006980419159,
      "learning_rate": 1.042242608286505e-06,
      "loss": 0.0578,
      "step": 80120
    },
    {
      "epoch": 5.83648681640625e-06,
      "model_forward_time": 1.7283074855804443,
      "step": 80120
    },
    {
      "dynamics_loss_avg": 0.044250420108437535,
      "epoch": 5.83648681640625e-06,
      "step": 80120
    },
    {
      "action_loss_avg": 0.0005958922149147839,
      "epoch": 5.83648681640625e-06,
      "step": 80120
    },
    {
      "epoch": 5.83648681640625e-06,
      "step": 80120,
      "training_step_time": 8.11025595664978
    },
    {
      "epoch": 5.837440490722657e-06,
      "model_forward_time": 1.7494440078735352,
      "step": 80121
    },
    {
      "epoch": 5.837440490722657e-06,
      "step": 80121,
      "training_step_time": 7.419697523117065
    },
    {
      "epoch": 5.838394165039062e-06,
      "model_forward_time": 1.7472126483917236,
      "step": 80122
    },
    {
      "epoch": 5.838394165039062e-06,
      "step": 80122,
      "training_step_time": 6.786925554275513
    },
    {
      "epoch": 5.839347839355469e-06,
      "model_forward_time": 1.7597098350524902,
      "step": 80123
    },
    {
      "epoch": 5.839347839355469e-06,
      "step": 80123,
      "training_step_time": 6.793720722198486
    },
    {
      "epoch": 5.840301513671875e-06,
      "model_forward_time": 1.7395286560058594,
      "step": 80124
    },
    {
      "epoch": 5.840301513671875e-06,
      "step": 80124,
      "training_step_time": 7.711819171905518
    },
    {
      "epoch": 5.841255187988281e-06,
      "model_forward_time": 1.7346746921539307,
      "step": 80125
    },
    {
      "epoch": 5.841255187988281e-06,
      "step": 80125,
      "training_step_time": 12.949897050857544
    },
    {
      "epoch": 5.842208862304688e-06,
      "model_forward_time": 2.374713897705078,
      "step": 80126
    },
    {
      "epoch": 5.842208862304688e-06,
      "step": 80126,
      "training_step_time": 8.679305076599121
    },
    {
      "epoch": 5.8431625366210935e-06,
      "model_forward_time": 1.3872008323669434,
      "step": 80127
    },
    {
      "epoch": 5.8431625366210935e-06,
      "step": 80127,
      "training_step_time": 8.473555326461792
    },
    {
      "epoch": 5.8441162109375e-06,
      "model_forward_time": 1.750267505645752,
      "step": 80128
    },
    {
      "epoch": 5.8441162109375e-06,
      "step": 80128,
      "training_step_time": 7.613616466522217
    },
    {
      "epoch": 5.845069885253906e-06,
      "model_forward_time": 1.741349220275879,
      "step": 80129
    },
    {
      "epoch": 5.845069885253906e-06,
      "step": 80129,
      "training_step_time": 7.663796901702881
    },
    {
      "epoch": 5.846023559570312e-06,
      "grad_norm": 0.0725010558962822,
      "learning_rate": 1.041232385184191e-06,
      "loss": 0.0574,
      "step": 80130
    },
    {
      "epoch": 5.846023559570312e-06,
      "model_forward_time": 1.777637243270874,
      "step": 80130
    },
    {
      "dynamics_loss_avg": 0.04964908268302679,
      "epoch": 5.846023559570312e-06,
      "step": 80130
    },
    {
      "action_loss_avg": 0.0010485176055226475,
      "epoch": 5.846023559570312e-06,
      "step": 80130
    },
    {
      "epoch": 5.846023559570312e-06,
      "step": 80130,
      "training_step_time": 7.464228868484497
    },
    {
      "epoch": 5.846977233886719e-06,
      "model_forward_time": 1.060410499572754,
      "step": 80131
    },
    {
      "epoch": 5.846977233886719e-06,
      "step": 80131,
      "training_step_time": 6.757846117019653
    },
    {
      "epoch": 5.847930908203125e-06,
      "model_forward_time": 1.7566218376159668,
      "step": 80132
    },
    {
      "epoch": 5.847930908203125e-06,
      "step": 80132,
      "training_step_time": 6.789396524429321
    },
    {
      "epoch": 5.848884582519531e-06,
      "model_forward_time": 1.7541170120239258,
      "step": 80133
    },
    {
      "epoch": 5.848884582519531e-06,
      "step": 80133,
      "training_step_time": 7.747115135192871
    },
    {
      "epoch": 5.849838256835938e-06,
      "model_forward_time": 1.7643136978149414,
      "step": 80134
    },
    {
      "epoch": 5.849838256835938e-06,
      "step": 80134,
      "training_step_time": 8.845509052276611
    },
    {
      "epoch": 5.8507919311523435e-06,
      "model_forward_time": 1.8098759651184082,
      "step": 80135
    },
    {
      "epoch": 5.8507919311523435e-06,
      "step": 80135,
      "training_step_time": 8.50835919380188
    },
    {
      "epoch": 5.85174560546875e-06,
      "model_forward_time": 2.4288673400878906,
      "step": 80136
    },
    {
      "epoch": 5.85174560546875e-06,
      "step": 80136,
      "training_step_time": 7.688389778137207
    },
    {
      "epoch": 5.852699279785156e-06,
      "model_forward_time": 1.774404764175415,
      "step": 80137
    },
    {
      "epoch": 5.852699279785156e-06,
      "step": 80137,
      "training_step_time": 7.6364076137542725
    },
    {
      "epoch": 5.853652954101562e-06,
      "model_forward_time": 1.7596611976623535,
      "step": 80138
    },
    {
      "epoch": 5.853652954101562e-06,
      "step": 80138,
      "training_step_time": 7.482724905014038
    },
    {
      "epoch": 5.854606628417969e-06,
      "model_forward_time": 1.7644093036651611,
      "step": 80139
    },
    {
      "epoch": 5.854606628417969e-06,
      "step": 80139,
      "training_step_time": 7.494531869888306
    },
    {
      "epoch": 5.855560302734375e-06,
      "grad_norm": 0.08315277844667435,
      "learning_rate": 1.0402225950067495e-06,
      "loss": 0.0584,
      "step": 80140
    },
    {
      "epoch": 5.855560302734375e-06,
      "model_forward_time": 1.7572731971740723,
      "step": 80140
    },
    {
      "dynamics_loss_avg": 0.05243992768228054,
      "epoch": 5.855560302734375e-06,
      "step": 80140
    },
    {
      "action_loss_avg": 0.0019144865014823153,
      "epoch": 5.855560302734375e-06,
      "step": 80140
    },
    {
      "epoch": 5.855560302734375e-06,
      "step": 80140,
      "training_step_time": 6.7482709884643555
    },
    {
      "epoch": 5.856513977050781e-06,
      "model_forward_time": 1.8001413345336914,
      "step": 80141
    },
    {
      "epoch": 5.856513977050781e-06,
      "step": 80141,
      "training_step_time": 6.736357688903809
    },
    {
      "epoch": 5.857467651367188e-06,
      "model_forward_time": 1.7652935981750488,
      "step": 80142
    },
    {
      "epoch": 5.857467651367188e-06,
      "step": 80142,
      "training_step_time": 7.658543586730957
    },
    {
      "epoch": 5.8584213256835936e-06,
      "model_forward_time": 1.7601768970489502,
      "step": 80143
    },
    {
      "epoch": 5.8584213256835936e-06,
      "step": 80143,
      "training_step_time": 8.049423694610596
    },
    {
      "epoch": 5.859375e-06,
      "model_forward_time": 1.8303227424621582,
      "step": 80144
    },
    {
      "epoch": 5.859375e-06,
      "step": 80144,
      "training_step_time": 8.92651915550232
    },
    {
      "epoch": 5.860328674316406e-06,
      "model_forward_time": 2.866889238357544,
      "step": 80145
    },
    {
      "epoch": 5.860328674316406e-06,
      "step": 80145,
      "training_step_time": 8.210688352584839
    },
    {
      "epoch": 5.861282348632812e-06,
      "model_forward_time": 1.761472463607788,
      "step": 80146
    },
    {
      "epoch": 5.861282348632812e-06,
      "step": 80146,
      "training_step_time": 7.601685285568237
    },
    {
      "epoch": 5.862236022949219e-06,
      "model_forward_time": 1.7588272094726562,
      "step": 80147
    },
    {
      "epoch": 5.862236022949219e-06,
      "step": 80147,
      "training_step_time": 7.530162334442139
    },
    {
      "epoch": 5.863189697265625e-06,
      "model_forward_time": 1.756500244140625,
      "step": 80148
    },
    {
      "epoch": 5.863189697265625e-06,
      "step": 80148,
      "training_step_time": 7.49699330329895
    },
    {
      "epoch": 5.864143371582031e-06,
      "model_forward_time": 1.763878583908081,
      "step": 80149
    },
    {
      "epoch": 5.864143371582031e-06,
      "step": 80149,
      "training_step_time": 6.744637727737427
    },
    {
      "epoch": 5.865097045898438e-06,
      "grad_norm": 0.0855356752872467,
      "learning_rate": 1.0392132378646118e-06,
      "loss": 0.0578,
      "step": 80150
    },
    {
      "epoch": 5.865097045898438e-06,
      "model_forward_time": 1.7607793807983398,
      "step": 80150
    },
    {
      "dynamics_loss_avg": 0.05871920622885227,
      "epoch": 5.865097045898438e-06,
      "step": 80150
    },
    {
      "action_loss_avg": 0.0014856593974400313,
      "epoch": 5.865097045898438e-06,
      "step": 80150
    },
    {
      "epoch": 5.865097045898438e-06,
      "step": 80150,
      "training_step_time": 6.7578277587890625
    },
    {
      "epoch": 5.8660507202148436e-06,
      "model_forward_time": 1.3981950283050537,
      "step": 80151
    },
    {
      "epoch": 5.8660507202148436e-06,
      "step": 80151,
      "training_step_time": 7.558766603469849
    },
    {
      "epoch": 5.86700439453125e-06,
      "model_forward_time": 1.7637314796447754,
      "step": 80152
    },
    {
      "epoch": 5.86700439453125e-06,
      "step": 80152,
      "training_step_time": 7.497140645980835
    },
    {
      "epoch": 5.867958068847656e-06,
      "model_forward_time": 1.8011512756347656,
      "step": 80153
    },
    {
      "epoch": 5.867958068847656e-06,
      "step": 80153,
      "training_step_time": 10.270545959472656
    },
    {
      "epoch": 5.868911743164062e-06,
      "model_forward_time": 2.8144006729125977,
      "step": 80154
    },
    {
      "epoch": 5.868911743164062e-06,
      "step": 80154,
      "training_step_time": 7.710097789764404
    },
    {
      "epoch": 5.869865417480469e-06,
      "model_forward_time": 1.7678313255310059,
      "step": 80155
    },
    {
      "epoch": 5.869865417480469e-06,
      "step": 80155,
      "training_step_time": 7.801180124282837
    },
    {
      "epoch": 5.870819091796875e-06,
      "model_forward_time": 1.7627127170562744,
      "step": 80156
    },
    {
      "epoch": 5.870819091796875e-06,
      "step": 80156,
      "training_step_time": 7.624889850616455
    },
    {
      "epoch": 5.871772766113281e-06,
      "model_forward_time": 1.7697830200195312,
      "step": 80157
    },
    {
      "epoch": 5.871772766113281e-06,
      "step": 80157,
      "training_step_time": 7.72607946395874
    },
    {
      "epoch": 5.872726440429688e-06,
      "model_forward_time": 1.7618532180786133,
      "step": 80158
    },
    {
      "epoch": 5.872726440429688e-06,
      "step": 80158,
      "training_step_time": 7.568812847137451
    },
    {
      "epoch": 5.8736801147460936e-06,
      "model_forward_time": 1.749305248260498,
      "step": 80159
    },
    {
      "epoch": 5.8736801147460936e-06,
      "step": 80159,
      "training_step_time": 6.90886116027832
    },
    {
      "epoch": 5.8746337890625e-06,
      "grad_norm": 0.07656758278608322,
      "learning_rate": 1.0382043138681609e-06,
      "loss": 0.0578,
      "step": 80160
    },
    {
      "epoch": 5.8746337890625e-06,
      "model_forward_time": 1.7666380405426025,
      "step": 80160
    },
    {
      "dynamics_loss_avg": 0.06476310677826405,
      "epoch": 5.8746337890625e-06,
      "step": 80160
    },
    {
      "action_loss_avg": 0.0010920162429101765,
      "epoch": 5.8746337890625e-06,
      "step": 80160
    },
    {
      "epoch": 5.8746337890625e-06,
      "step": 80160,
      "training_step_time": 8.751078605651855
    },
    {
      "epoch": 5.875587463378906e-06,
      "model_forward_time": 1.761777639389038,
      "step": 80161
    },
    {
      "epoch": 5.875587463378906e-06,
      "step": 80161,
      "training_step_time": 7.608844995498657
    },
    {
      "epoch": 5.8765411376953124e-06,
      "model_forward_time": 1.7567839622497559,
      "step": 80162
    },
    {
      "epoch": 5.8765411376953124e-06,
      "step": 80162,
      "training_step_time": 10.050635576248169
    },
    {
      "epoch": 5.877494812011719e-06,
      "model_forward_time": 1.761702299118042,
      "step": 80163
    },
    {
      "epoch": 5.877494812011719e-06,
      "step": 80163,
      "training_step_time": 9.54750108718872
    },
    {
      "epoch": 5.878448486328125e-06,
      "model_forward_time": 2.8526389598846436,
      "step": 80164
    },
    {
      "epoch": 5.878448486328125e-06,
      "step": 80164,
      "training_step_time": 7.817486047744751
    },
    {
      "epoch": 5.879402160644531e-06,
      "model_forward_time": 1.7616472244262695,
      "step": 80165
    },
    {
      "epoch": 5.879402160644531e-06,
      "step": 80165,
      "training_step_time": 7.725358009338379
    },
    {
      "epoch": 5.880355834960938e-06,
      "model_forward_time": 1.756316900253296,
      "step": 80166
    },
    {
      "epoch": 5.880355834960938e-06,
      "step": 80166,
      "training_step_time": 7.697655439376831
    },
    {
      "epoch": 5.8813095092773436e-06,
      "model_forward_time": 1.7563693523406982,
      "step": 80167
    },
    {
      "epoch": 5.8813095092773436e-06,
      "step": 80167,
      "training_step_time": 7.505223751068115
    },
    {
      "epoch": 5.88226318359375e-06,
      "model_forward_time": 1.7336208820343018,
      "step": 80168
    },
    {
      "epoch": 5.88226318359375e-06,
      "step": 80168,
      "training_step_time": 6.804333448410034
    },
    {
      "epoch": 5.883216857910156e-06,
      "model_forward_time": 1.037670612335205,
      "step": 80169
    },
    {
      "epoch": 5.883216857910156e-06,
      "step": 80169,
      "training_step_time": 7.687268257141113
    },
    {
      "epoch": 5.8841705322265624e-06,
      "grad_norm": 0.1018715500831604,
      "learning_rate": 1.0371958231277285e-06,
      "loss": 0.0584,
      "step": 80170
    },
    {
      "epoch": 5.8841705322265624e-06,
      "model_forward_time": 1.7552592754364014,
      "step": 80170
    },
    {
      "dynamics_loss_avg": 0.05639857212081552,
      "epoch": 5.8841705322265624e-06,
      "step": 80170
    },
    {
      "action_loss_avg": 0.0014772211259696632,
      "epoch": 5.8841705322265624e-06,
      "step": 80170
    },
    {
      "epoch": 5.8841705322265624e-06,
      "step": 80170,
      "training_step_time": 7.803797245025635
    },
    {
      "epoch": 5.885124206542969e-06,
      "model_forward_time": 1.7465050220489502,
      "step": 80171
    },
    {
      "epoch": 5.885124206542969e-06,
      "step": 80171,
      "training_step_time": 10.962560892105103
    },
    {
      "epoch": 5.886077880859375e-06,
      "model_forward_time": 1.8205127716064453,
      "step": 80172
    },
    {
      "epoch": 5.886077880859375e-06,
      "step": 80172,
      "training_step_time": 8.460068941116333
    },
    {
      "epoch": 5.887031555175781e-06,
      "model_forward_time": 2.8280882835388184,
      "step": 80173
    },
    {
      "epoch": 5.887031555175781e-06,
      "step": 80173,
      "training_step_time": 7.791425466537476
    },
    {
      "epoch": 5.887985229492188e-06,
      "model_forward_time": 1.7395930290222168,
      "step": 80174
    },
    {
      "epoch": 5.887985229492188e-06,
      "step": 80174,
      "training_step_time": 7.914690017700195
    },
    {
      "epoch": 5.888938903808594e-06,
      "model_forward_time": 1.755150556564331,
      "step": 80175
    },
    {
      "epoch": 5.888938903808594e-06,
      "step": 80175,
      "training_step_time": 7.495048999786377
    },
    {
      "epoch": 5.889892578125e-06,
      "model_forward_time": 1.7562389373779297,
      "step": 80176
    },
    {
      "epoch": 5.889892578125e-06,
      "step": 80176,
      "training_step_time": 7.50936484336853
    },
    {
      "epoch": 5.890846252441406e-06,
      "model_forward_time": 1.750234603881836,
      "step": 80177
    },
    {
      "epoch": 5.890846252441406e-06,
      "step": 80177,
      "training_step_time": 6.82236123085022
    },
    {
      "epoch": 5.8917999267578124e-06,
      "model_forward_time": 1.3928625583648682,
      "step": 80178
    },
    {
      "epoch": 5.8917999267578124e-06,
      "step": 80178,
      "training_step_time": 6.811653137207031
    },
    {
      "epoch": 5.892753601074219e-06,
      "model_forward_time": 1.7465522289276123,
      "step": 80179
    },
    {
      "epoch": 5.892753601074219e-06,
      "step": 80179,
      "training_step_time": 7.779462814331055
    },
    {
      "epoch": 5.893707275390625e-06,
      "grad_norm": 0.14485280215740204,
      "learning_rate": 1.0361877657536024e-06,
      "loss": 0.0587,
      "step": 80180
    },
    {
      "epoch": 5.893707275390625e-06,
      "model_forward_time": 1.7458267211914062,
      "step": 80180
    },
    {
      "dynamics_loss_avg": 0.059878173470497134,
      "epoch": 5.893707275390625e-06,
      "step": 80180
    },
    {
      "action_loss_avg": 0.0021312448428943754,
      "epoch": 5.893707275390625e-06,
      "step": 80180
    },
    {
      "epoch": 5.893707275390625e-06,
      "step": 80180,
      "training_step_time": 10.324180603027344
    },
    {
      "epoch": 5.894660949707031e-06,
      "model_forward_time": 1.8250067234039307,
      "step": 80181
    },
    {
      "epoch": 5.894660949707031e-06,
      "step": 80181,
      "training_step_time": 9.516707420349121
    },
    {
      "epoch": 5.895614624023438e-06,
      "model_forward_time": 2.7828314304351807,
      "step": 80182
    },
    {
      "epoch": 5.895614624023438e-06,
      "step": 80182,
      "training_step_time": 7.865271091461182
    },
    {
      "epoch": 5.896568298339844e-06,
      "model_forward_time": 1.7514827251434326,
      "step": 80183
    },
    {
      "epoch": 5.896568298339844e-06,
      "step": 80183,
      "training_step_time": 7.860854864120483
    },
    {
      "epoch": 5.89752197265625e-06,
      "model_forward_time": 1.7536816596984863,
      "step": 80184
    },
    {
      "epoch": 5.89752197265625e-06,
      "step": 80184,
      "training_step_time": 7.538714647293091
    },
    {
      "epoch": 5.898475646972656e-06,
      "model_forward_time": 1.749995231628418,
      "step": 80185
    },
    {
      "epoch": 5.898475646972656e-06,
      "step": 80185,
      "training_step_time": 7.7025721073150635
    },
    {
      "epoch": 5.8994293212890624e-06,
      "model_forward_time": 1.7430856227874756,
      "step": 80186
    },
    {
      "epoch": 5.8994293212890624e-06,
      "step": 80186,
      "training_step_time": 6.760366678237915
    },
    {
      "epoch": 5.900382995605469e-06,
      "model_forward_time": 1.7451300621032715,
      "step": 80187
    },
    {
      "epoch": 5.900382995605469e-06,
      "step": 80187,
      "training_step_time": 6.8724377155303955
    },
    {
      "epoch": 5.901336669921875e-06,
      "model_forward_time": 1.7503464221954346,
      "step": 80188
    },
    {
      "epoch": 5.901336669921875e-06,
      "step": 80188,
      "training_step_time": 7.780832529067993
    },
    {
      "epoch": 5.902290344238281e-06,
      "model_forward_time": 1.7548184394836426,
      "step": 80189
    },
    {
      "epoch": 5.902290344238281e-06,
      "step": 80189,
      "training_step_time": 8.16561508178711
    },
    {
      "epoch": 5.903244018554688e-06,
      "grad_norm": 0.07258966565132141,
      "learning_rate": 1.035180141856023e-06,
      "loss": 0.0578,
      "step": 80190
    },
    {
      "epoch": 5.903244018554688e-06,
      "model_forward_time": 1.8016061782836914,
      "step": 80190
    },
    {
      "dynamics_loss_avg": 0.04591122604906559,
      "epoch": 5.903244018554688e-06,
      "step": 80190
    },
    {
      "action_loss_avg": 0.0009825763539993205,
      "epoch": 5.903244018554688e-06,
      "step": 80190
    },
    {
      "epoch": 5.903244018554688e-06,
      "step": 80190,
      "training_step_time": 11.447223901748657
    },
    {
      "epoch": 5.904197692871094e-06,
      "model_forward_time": 1.7591590881347656,
      "step": 80191
    },
    {
      "epoch": 5.904197692871094e-06,
      "step": 80191,
      "training_step_time": 7.861138343811035
    },
    {
      "epoch": 5.9051513671875e-06,
      "model_forward_time": 2.8547277450561523,
      "step": 80192
    },
    {
      "epoch": 5.9051513671875e-06,
      "step": 80192,
      "training_step_time": 7.728926420211792
    },
    {
      "epoch": 5.906105041503906e-06,
      "model_forward_time": 1.7671947479248047,
      "step": 80193
    },
    {
      "epoch": 5.906105041503906e-06,
      "step": 80193,
      "training_step_time": 7.733892917633057
    },
    {
      "epoch": 5.9070587158203125e-06,
      "model_forward_time": 1.7664463520050049,
      "step": 80194
    },
    {
      "epoch": 5.9070587158203125e-06,
      "step": 80194,
      "training_step_time": 7.784733295440674
    },
    {
      "epoch": 5.908012390136719e-06,
      "model_forward_time": 1.7585089206695557,
      "step": 80195
    },
    {
      "epoch": 5.908012390136719e-06,
      "step": 80195,
      "training_step_time": 6.822391986846924
    },
    {
      "epoch": 5.908966064453125e-06,
      "model_forward_time": 1.7627079486846924,
      "step": 80196
    },
    {
      "epoch": 5.908966064453125e-06,
      "step": 80196,
      "training_step_time": 6.790438652038574
    },
    {
      "epoch": 5.909919738769531e-06,
      "model_forward_time": 1.7548305988311768,
      "step": 80197
    },
    {
      "epoch": 5.909919738769531e-06,
      "step": 80197,
      "training_step_time": 7.736471176147461
    },
    {
      "epoch": 5.910873413085938e-06,
      "model_forward_time": 1.7585783004760742,
      "step": 80198
    },
    {
      "epoch": 5.910873413085938e-06,
      "step": 80198,
      "training_step_time": 7.639014720916748
    },
    {
      "epoch": 5.911827087402344e-06,
      "model_forward_time": 1.830681562423706,
      "step": 80199
    },
    {
      "epoch": 5.911827087402344e-06,
      "step": 80199,
      "training_step_time": 10.439475774765015
    },
    {
      "epoch": 5.91278076171875e-06,
      "grad_norm": 0.07832058519124985,
      "learning_rate": 1.0341729515451837e-06,
      "loss": 0.0571,
      "step": 80200
    },
    {
      "epoch": 5.91278076171875e-06,
      "model_forward_time": 1.7372865676879883,
      "step": 80200
    },
    {
      "dynamics_loss_avg": 0.057083146460354325,
      "epoch": 5.91278076171875e-06,
      "step": 80200
    },
    {
      "action_loss_avg": 0.0022528870962560178,
      "epoch": 5.91278076171875e-06,
      "step": 80200
    },
    {
      "epoch": 5.91278076171875e-06,
      "step": 80200,
      "training_step_time": 9.435757875442505
    },
    {
      "epoch": 5.913734436035156e-06,
      "model_forward_time": 2.8610568046569824,
      "step": 80201
    },
    {
      "epoch": 5.913734436035156e-06,
      "step": 80201,
      "training_step_time": 7.77204704284668
    },
    {
      "epoch": 5.9146881103515625e-06,
      "model_forward_time": 1.755957841873169,
      "step": 80202
    },
    {
      "epoch": 5.9146881103515625e-06,
      "step": 80202,
      "training_step_time": 7.755520343780518
    },
    {
      "epoch": 5.915641784667969e-06,
      "model_forward_time": 1.7543964385986328,
      "step": 80203
    },
    {
      "epoch": 5.915641784667969e-06,
      "step": 80203,
      "training_step_time": 8.432082891464233
    },
    {
      "epoch": 5.916595458984375e-06,
      "model_forward_time": 1.752464771270752,
      "step": 80204
    },
    {
      "epoch": 5.916595458984375e-06,
      "step": 80204,
      "training_step_time": 7.51604437828064
    },
    {
      "epoch": 5.917549133300781e-06,
      "model_forward_time": 1.7525267601013184,
      "step": 80205
    },
    {
      "epoch": 5.917549133300781e-06,
      "step": 80205,
      "training_step_time": 6.6653454303741455
    },
    {
      "epoch": 5.918502807617188e-06,
      "model_forward_time": 1.7323718070983887,
      "step": 80206
    },
    {
      "epoch": 5.918502807617188e-06,
      "step": 80206,
      "training_step_time": 7.619002103805542
    },
    {
      "epoch": 5.919456481933594e-06,
      "model_forward_time": 1.7529072761535645,
      "step": 80207
    },
    {
      "epoch": 5.919456481933594e-06,
      "step": 80207,
      "training_step_time": 9.468427658081055
    },
    {
      "epoch": 5.92041015625e-06,
      "model_forward_time": 1.8093593120574951,
      "step": 80208
    },
    {
      "epoch": 5.92041015625e-06,
      "step": 80208,
      "training_step_time": 9.120290994644165
    },
    {
      "epoch": 5.921363830566406e-06,
      "model_forward_time": 1.7564871311187744,
      "step": 80209
    },
    {
      "epoch": 5.921363830566406e-06,
      "step": 80209,
      "training_step_time": 10.46420955657959
    },
    {
      "epoch": 5.9223175048828125e-06,
      "grad_norm": 0.3129681646823883,
      "learning_rate": 1.0331661949312266e-06,
      "loss": 0.0574,
      "step": 80210
    },
    {
      "epoch": 5.9223175048828125e-06,
      "model_forward_time": 2.830700397491455,
      "step": 80210
    },
    {
      "dynamics_loss_avg": 0.0723204119130969,
      "epoch": 5.9223175048828125e-06,
      "step": 80210
    },
    {
      "action_loss_avg": 0.0012398939667036756,
      "epoch": 5.9223175048828125e-06,
      "step": 80210
    },
    {
      "epoch": 5.9223175048828125e-06,
      "step": 80210,
      "training_step_time": 7.957502841949463
    },
    {
      "epoch": 5.923271179199219e-06,
      "model_forward_time": 1.7511444091796875,
      "step": 80211
    },
    {
      "epoch": 5.923271179199219e-06,
      "step": 80211,
      "training_step_time": 7.778180837631226
    },
    {
      "epoch": 5.924224853515625e-06,
      "model_forward_time": 1.753051519393921,
      "step": 80212
    },
    {
      "epoch": 5.924224853515625e-06,
      "step": 80212,
      "training_step_time": 7.7560670375823975
    },
    {
      "epoch": 5.925178527832031e-06,
      "model_forward_time": 1.756786823272705,
      "step": 80213
    },
    {
      "epoch": 5.925178527832031e-06,
      "step": 80213,
      "training_step_time": 7.583390235900879
    },
    {
      "epoch": 5.926132202148438e-06,
      "model_forward_time": 1.759556770324707,
      "step": 80214
    },
    {
      "epoch": 5.926132202148438e-06,
      "step": 80214,
      "training_step_time": 6.856865406036377
    },
    {
      "epoch": 5.927085876464844e-06,
      "model_forward_time": 1.756115198135376,
      "step": 80215
    },
    {
      "epoch": 5.927085876464844e-06,
      "step": 80215,
      "training_step_time": 7.19731879234314
    },
    {
      "epoch": 5.92803955078125e-06,
      "model_forward_time": 1.7672860622406006,
      "step": 80216
    },
    {
      "epoch": 5.92803955078125e-06,
      "step": 80216,
      "training_step_time": 7.794183969497681
    },
    {
      "epoch": 5.928993225097656e-06,
      "model_forward_time": 1.812798261642456,
      "step": 80217
    },
    {
      "epoch": 5.928993225097656e-06,
      "step": 80217,
      "training_step_time": 9.043826341629028
    },
    {
      "epoch": 5.9299468994140625e-06,
      "model_forward_time": 1.7501437664031982,
      "step": 80218
    },
    {
      "epoch": 5.9299468994140625e-06,
      "step": 80218,
      "training_step_time": 8.903539896011353
    },
    {
      "epoch": 5.930900573730469e-06,
      "model_forward_time": 2.8437435626983643,
      "step": 80219
    },
    {
      "epoch": 5.930900573730469e-06,
      "step": 80219,
      "training_step_time": 7.890995502471924
    },
    {
      "epoch": 5.931854248046875e-06,
      "grad_norm": 0.12366598099470139,
      "learning_rate": 1.0321598721242514e-06,
      "loss": 0.0572,
      "step": 80220
    },
    {
      "epoch": 5.931854248046875e-06,
      "model_forward_time": 1.7459406852722168,
      "step": 80220
    },
    {
      "dynamics_loss_avg": 0.06178374299779534,
      "epoch": 5.931854248046875e-06,
      "step": 80220
    },
    {
      "action_loss_avg": 0.0021075256750918924,
      "epoch": 5.931854248046875e-06,
      "step": 80220
    },
    {
      "epoch": 5.931854248046875e-06,
      "step": 80220,
      "training_step_time": 8.276020288467407
    },
    {
      "epoch": 5.932807922363281e-06,
      "model_forward_time": 1.7637526988983154,
      "step": 80221
    },
    {
      "epoch": 5.932807922363281e-06,
      "step": 80221,
      "training_step_time": 7.678241729736328
    },
    {
      "epoch": 5.933761596679688e-06,
      "model_forward_time": 1.75740385055542,
      "step": 80222
    },
    {
      "epoch": 5.933761596679688e-06,
      "step": 80222,
      "training_step_time": 7.528148651123047
    },
    {
      "epoch": 5.934715270996094e-06,
      "model_forward_time": 1.7722253799438477,
      "step": 80223
    },
    {
      "epoch": 5.934715270996094e-06,
      "step": 80223,
      "training_step_time": 6.784686803817749
    },
    {
      "epoch": 5.9356689453125e-06,
      "model_forward_time": 1.7634449005126953,
      "step": 80224
    },
    {
      "epoch": 5.9356689453125e-06,
      "step": 80224,
      "training_step_time": 7.075297594070435
    },
    {
      "epoch": 5.936622619628906e-06,
      "model_forward_time": 1.7473978996276855,
      "step": 80225
    },
    {
      "epoch": 5.936622619628906e-06,
      "step": 80225,
      "training_step_time": 7.7988762855529785
    },
    {
      "epoch": 5.9375762939453125e-06,
      "model_forward_time": 1.800286054611206,
      "step": 80226
    },
    {
      "epoch": 5.9375762939453125e-06,
      "step": 80226,
      "training_step_time": 8.43457579612732
    },
    {
      "epoch": 5.938529968261719e-06,
      "model_forward_time": 1.7496092319488525,
      "step": 80227
    },
    {
      "epoch": 5.938529968261719e-06,
      "step": 80227,
      "training_step_time": 8.367297172546387
    },
    {
      "epoch": 5.939483642578125e-06,
      "model_forward_time": 2.801490306854248,
      "step": 80228
    },
    {
      "epoch": 5.939483642578125e-06,
      "step": 80228,
      "training_step_time": 7.667876243591309
    },
    {
      "epoch": 5.940437316894531e-06,
      "model_forward_time": 1.7579014301300049,
      "step": 80229
    },
    {
      "epoch": 5.940437316894531e-06,
      "step": 80229,
      "training_step_time": 7.823548793792725
    },
    {
      "epoch": 5.941390991210938e-06,
      "grad_norm": 0.14649374783039093,
      "learning_rate": 1.0311539832343065e-06,
      "loss": 0.0562,
      "step": 80230
    },
    {
      "epoch": 5.941390991210938e-06,
      "model_forward_time": 1.7589871883392334,
      "step": 80230
    },
    {
      "dynamics_loss_avg": 0.05809423476457596,
      "epoch": 5.941390991210938e-06,
      "step": 80230
    },
    {
      "action_loss_avg": 0.0011040767072699963,
      "epoch": 5.941390991210938e-06,
      "step": 80230
    },
    {
      "epoch": 5.941390991210938e-06,
      "step": 80230,
      "training_step_time": 7.726038217544556
    },
    {
      "epoch": 5.942344665527344e-06,
      "model_forward_time": 1.7550063133239746,
      "step": 80231
    },
    {
      "epoch": 5.942344665527344e-06,
      "step": 80231,
      "training_step_time": 7.770743370056152
    },
    {
      "epoch": 5.94329833984375e-06,
      "model_forward_time": 1.7717182636260986,
      "step": 80232
    },
    {
      "epoch": 5.94329833984375e-06,
      "step": 80232,
      "training_step_time": 6.764625072479248
    },
    {
      "epoch": 5.944252014160156e-06,
      "model_forward_time": 1.7573554515838623,
      "step": 80233
    },
    {
      "epoch": 5.944252014160156e-06,
      "step": 80233,
      "training_step_time": 6.827422618865967
    },
    {
      "epoch": 5.9452056884765625e-06,
      "model_forward_time": 1.7548539638519287,
      "step": 80234
    },
    {
      "epoch": 5.9452056884765625e-06,
      "step": 80234,
      "training_step_time": 7.771499156951904
    },
    {
      "epoch": 5.946159362792969e-06,
      "model_forward_time": 1.756988525390625,
      "step": 80235
    },
    {
      "epoch": 5.946159362792969e-06,
      "step": 80235,
      "training_step_time": 10.355468034744263
    },
    {
      "epoch": 5.947113037109375e-06,
      "model_forward_time": 1.7456653118133545,
      "step": 80236
    },
    {
      "epoch": 5.947113037109375e-06,
      "step": 80236,
      "training_step_time": 8.91525673866272
    },
    {
      "epoch": 5.948066711425781e-06,
      "model_forward_time": 1.8007206916809082,
      "step": 80237
    },
    {
      "epoch": 5.948066711425781e-06,
      "step": 80237,
      "training_step_time": 9.332668542861938
    },
    {
      "epoch": 5.949020385742188e-06,
      "model_forward_time": 1.7487173080444336,
      "step": 80238
    },
    {
      "epoch": 5.949020385742188e-06,
      "step": 80238,
      "training_step_time": 7.769695520401001
    },
    {
      "epoch": 5.949974060058594e-06,
      "model_forward_time": 1.7512037754058838,
      "step": 80239
    },
    {
      "epoch": 5.949974060058594e-06,
      "step": 80239,
      "training_step_time": 7.693075180053711
    },
    {
      "epoch": 5.950927734375e-06,
      "grad_norm": 0.14963126182556152,
      "learning_rate": 1.030148528371394e-06,
      "loss": 0.0557,
      "step": 80240
    },
    {
      "epoch": 5.950927734375e-06,
      "model_forward_time": 1.7621550559997559,
      "step": 80240
    },
    {
      "dynamics_loss_avg": 0.05451902262866497,
      "epoch": 5.950927734375e-06,
      "step": 80240
    },
    {
      "action_loss_avg": 0.0014995343750342726,
      "epoch": 5.950927734375e-06,
      "step": 80240
    },
    {
      "epoch": 5.950927734375e-06,
      "step": 80240,
      "training_step_time": 8.487403154373169
    },
    {
      "epoch": 5.951881408691406e-06,
      "model_forward_time": 1.7526981830596924,
      "step": 80241
    },
    {
      "epoch": 5.951881408691406e-06,
      "step": 80241,
      "training_step_time": 7.634296178817749
    },
    {
      "epoch": 5.9528350830078125e-06,
      "model_forward_time": 1.7606990337371826,
      "step": 80242
    },
    {
      "epoch": 5.9528350830078125e-06,
      "step": 80242,
      "training_step_time": 6.811161756515503
    },
    {
      "epoch": 5.953788757324219e-06,
      "model_forward_time": 1.751688003540039,
      "step": 80243
    },
    {
      "epoch": 5.953788757324219e-06,
      "step": 80243,
      "training_step_time": 8.485721588134766
    },
    {
      "epoch": 5.954742431640625e-06,
      "model_forward_time": 1.7560040950775146,
      "step": 80244
    },
    {
      "epoch": 5.954742431640625e-06,
      "step": 80244,
      "training_step_time": 9.498137474060059
    },
    {
      "epoch": 5.955696105957031e-06,
      "model_forward_time": 1.8163936138153076,
      "step": 80245
    },
    {
      "epoch": 5.955696105957031e-06,
      "step": 80245,
      "training_step_time": 8.627965688705444
    },
    {
      "epoch": 5.956649780273438e-06,
      "model_forward_time": 1.747251033782959,
      "step": 80246
    },
    {
      "epoch": 5.956649780273438e-06,
      "step": 80246,
      "training_step_time": 8.670270681381226
    },
    {
      "epoch": 5.957603454589844e-06,
      "model_forward_time": 2.8153045177459717,
      "step": 80247
    },
    {
      "epoch": 5.957603454589844e-06,
      "step": 80247,
      "training_step_time": 7.899240255355835
    },
    {
      "epoch": 5.95855712890625e-06,
      "model_forward_time": 1.7530581951141357,
      "step": 80248
    },
    {
      "epoch": 5.95855712890625e-06,
      "step": 80248,
      "training_step_time": 7.734065532684326
    },
    {
      "epoch": 5.959510803222656e-06,
      "model_forward_time": 1.7532107830047607,
      "step": 80249
    },
    {
      "epoch": 5.959510803222656e-06,
      "step": 80249,
      "training_step_time": 7.699352979660034
    },
    {
      "epoch": 5.9604644775390625e-06,
      "grad_norm": 0.1435108333826065,
      "learning_rate": 1.0291435076454704e-06,
      "loss": 0.0556,
      "step": 80250
    },
    {
      "epoch": 5.9604644775390625e-06,
      "model_forward_time": 1.749074935913086,
      "step": 80250
    },
    {
      "dynamics_loss_avg": 0.05584575831890106,
      "epoch": 5.9604644775390625e-06,
      "step": 80250
    },
    {
      "action_loss_avg": 0.0010548243968514725,
      "epoch": 5.9604644775390625e-06,
      "step": 80250
    },
    {
      "epoch": 5.9604644775390625e-06,
      "step": 80250,
      "training_step_time": 7.515099287033081
    },
    {
      "epoch": 5.961418151855469e-06,
      "model_forward_time": 1.7525017261505127,
      "step": 80251
    },
    {
      "epoch": 5.961418151855469e-06,
      "step": 80251,
      "training_step_time": 6.8035502433776855
    },
    {
      "epoch": 5.962371826171875e-06,
      "model_forward_time": 1.777716875076294,
      "step": 80252
    },
    {
      "epoch": 5.962371826171875e-06,
      "step": 80252,
      "training_step_time": 7.722261190414429
    },
    {
      "epoch": 5.963325500488281e-06,
      "model_forward_time": 1.754063367843628,
      "step": 80253
    },
    {
      "epoch": 5.963325500488281e-06,
      "step": 80253,
      "training_step_time": 8.408799648284912
    },
    {
      "epoch": 5.964279174804687e-06,
      "model_forward_time": 1.7599947452545166,
      "step": 80254
    },
    {
      "epoch": 5.964279174804687e-06,
      "step": 80254,
      "training_step_time": 9.310100793838501
    },
    {
      "epoch": 5.965232849121094e-06,
      "model_forward_time": 1.7501130104064941,
      "step": 80255
    },
    {
      "epoch": 5.965232849121094e-06,
      "step": 80255,
      "training_step_time": 10.275848627090454
    },
    {
      "epoch": 5.9661865234375e-06,
      "model_forward_time": 2.8297040462493896,
      "step": 80256
    },
    {
      "epoch": 5.9661865234375e-06,
      "step": 80256,
      "training_step_time": 8.935809850692749
    },
    {
      "epoch": 5.967140197753906e-06,
      "model_forward_time": 1.7517635822296143,
      "step": 80257
    },
    {
      "epoch": 5.967140197753906e-06,
      "step": 80257,
      "training_step_time": 7.7591776847839355
    },
    {
      "epoch": 5.9680938720703125e-06,
      "model_forward_time": 1.379701852798462,
      "step": 80258
    },
    {
      "epoch": 5.9680938720703125e-06,
      "step": 80258,
      "training_step_time": 8.759490251541138
    },
    {
      "epoch": 5.969047546386719e-06,
      "model_forward_time": 1.7350034713745117,
      "step": 80259
    },
    {
      "epoch": 5.969047546386719e-06,
      "step": 80259,
      "training_step_time": 7.454361200332642
    },
    {
      "epoch": 5.970001220703125e-06,
      "grad_norm": 0.11746614426374435,
      "learning_rate": 1.0281389211664439e-06,
      "loss": 0.0545,
      "step": 80260
    },
    {
      "epoch": 5.970001220703125e-06,
      "model_forward_time": 1.747042179107666,
      "step": 80260
    },
    {
      "dynamics_loss_avg": 0.06416839100420475,
      "epoch": 5.970001220703125e-06,
      "step": 80260
    },
    {
      "action_loss_avg": 0.001200008878367953,
      "epoch": 5.970001220703125e-06,
      "step": 80260
    },
    {
      "epoch": 5.970001220703125e-06,
      "step": 80260,
      "training_step_time": 6.850458383560181
    },
    {
      "epoch": 5.970954895019531e-06,
      "model_forward_time": 1.750058650970459,
      "step": 80261
    },
    {
      "epoch": 5.970954895019531e-06,
      "step": 80261,
      "training_step_time": 8.72340202331543
    },
    {
      "epoch": 5.971908569335937e-06,
      "model_forward_time": 1.4033946990966797,
      "step": 80262
    },
    {
      "epoch": 5.971908569335937e-06,
      "step": 80262,
      "training_step_time": 8.631460428237915
    },
    {
      "epoch": 5.972862243652344e-06,
      "model_forward_time": 1.7443976402282715,
      "step": 80263
    },
    {
      "epoch": 5.972862243652344e-06,
      "step": 80263,
      "training_step_time": 9.685367822647095
    },
    {
      "epoch": 5.97381591796875e-06,
      "model_forward_time": 1.8295836448669434,
      "step": 80264
    },
    {
      "epoch": 5.97381591796875e-06,
      "step": 80264,
      "training_step_time": 7.919465780258179
    },
    {
      "epoch": 5.974769592285156e-06,
      "model_forward_time": 2.8410990238189697,
      "step": 80265
    },
    {
      "epoch": 5.974769592285156e-06,
      "step": 80265,
      "training_step_time": 7.732774019241333
    },
    {
      "epoch": 5.9757232666015625e-06,
      "model_forward_time": 1.7569448947906494,
      "step": 80266
    },
    {
      "epoch": 5.9757232666015625e-06,
      "step": 80266,
      "training_step_time": 7.876591920852661
    },
    {
      "epoch": 5.976676940917969e-06,
      "model_forward_time": 1.7660877704620361,
      "step": 80267
    },
    {
      "epoch": 5.976676940917969e-06,
      "step": 80267,
      "training_step_time": 7.771654367446899
    },
    {
      "epoch": 5.977630615234375e-06,
      "model_forward_time": 1.7455158233642578,
      "step": 80268
    },
    {
      "epoch": 5.977630615234375e-06,
      "step": 80268,
      "training_step_time": 7.846337795257568
    },
    {
      "epoch": 5.978584289550781e-06,
      "model_forward_time": 1.7461047172546387,
      "step": 80269
    },
    {
      "epoch": 5.978584289550781e-06,
      "step": 80269,
      "training_step_time": 6.853283882141113
    },
    {
      "epoch": 5.979537963867187e-06,
      "grad_norm": 0.08839594572782516,
      "learning_rate": 1.0271347690441718e-06,
      "loss": 0.0549,
      "step": 80270
    },
    {
      "epoch": 5.979537963867187e-06,
      "model_forward_time": 1.7523863315582275,
      "step": 80270
    },
    {
      "dynamics_loss_avg": 0.05363254398107529,
      "epoch": 5.979537963867187e-06,
      "step": 80270
    },
    {
      "action_loss_avg": 0.001365366211393848,
      "epoch": 5.979537963867187e-06,
      "step": 80270
    },
    {
      "epoch": 5.979537963867187e-06,
      "step": 80270,
      "training_step_time": 7.687725782394409
    },
    {
      "epoch": 5.980491638183594e-06,
      "model_forward_time": 1.7832624912261963,
      "step": 80271
    },
    {
      "epoch": 5.980491638183594e-06,
      "step": 80271,
      "training_step_time": 8.498811960220337
    },
    {
      "epoch": 5.9814453125e-06,
      "model_forward_time": 1.752364158630371,
      "step": 80272
    },
    {
      "epoch": 5.9814453125e-06,
      "step": 80272,
      "training_step_time": 8.638602018356323
    },
    {
      "epoch": 5.982398986816406e-06,
      "model_forward_time": 1.835552453994751,
      "step": 80273
    },
    {
      "epoch": 5.982398986816406e-06,
      "step": 80273,
      "training_step_time": 10.259174585342407
    },
    {
      "epoch": 5.9833526611328125e-06,
      "model_forward_time": 2.8164303302764893,
      "step": 80274
    },
    {
      "epoch": 5.9833526611328125e-06,
      "step": 80274,
      "training_step_time": 8.265411615371704
    },
    {
      "epoch": 5.984306335449219e-06,
      "model_forward_time": 1.7747700214385986,
      "step": 80275
    },
    {
      "epoch": 5.984306335449219e-06,
      "step": 80275,
      "training_step_time": 7.604170322418213
    },
    {
      "epoch": 5.985260009765625e-06,
      "model_forward_time": 1.7549078464508057,
      "step": 80276
    },
    {
      "epoch": 5.985260009765625e-06,
      "step": 80276,
      "training_step_time": 7.494415998458862
    },
    {
      "epoch": 5.986213684082031e-06,
      "model_forward_time": 1.753755807876587,
      "step": 80277
    },
    {
      "epoch": 5.986213684082031e-06,
      "step": 80277,
      "training_step_time": 7.631832599639893
    },
    {
      "epoch": 5.987167358398437e-06,
      "model_forward_time": 1.7686119079589844,
      "step": 80278
    },
    {
      "epoch": 5.987167358398437e-06,
      "step": 80278,
      "training_step_time": 6.726625442504883
    },
    {
      "epoch": 5.988121032714844e-06,
      "model_forward_time": 1.7684686183929443,
      "step": 80279
    },
    {
      "epoch": 5.988121032714844e-06,
      "step": 80279,
      "training_step_time": 7.508294343948364
    },
    {
      "epoch": 5.98907470703125e-06,
      "grad_norm": 0.11810369044542313,
      "learning_rate": 1.0261310513884697e-06,
      "loss": 0.0538,
      "step": 80280
    },
    {
      "epoch": 5.98907470703125e-06,
      "model_forward_time": 1.7559716701507568,
      "step": 80280
    },
    {
      "dynamics_loss_avg": 0.04938650615513325,
      "epoch": 5.98907470703125e-06,
      "step": 80280
    },
    {
      "action_loss_avg": 0.0008837870409479365,
      "epoch": 5.98907470703125e-06,
      "step": 80280
    },
    {
      "epoch": 5.98907470703125e-06,
      "step": 80280,
      "training_step_time": 7.644735336303711
    },
    {
      "epoch": 5.990028381347656e-06,
      "model_forward_time": 1.7553887367248535,
      "step": 80281
    },
    {
      "epoch": 5.990028381347656e-06,
      "step": 80281,
      "training_step_time": 8.771250486373901
    },
    {
      "epoch": 5.9909820556640625e-06,
      "model_forward_time": 1.817521572113037,
      "step": 80282
    },
    {
      "epoch": 5.9909820556640625e-06,
      "step": 80282,
      "training_step_time": 9.314886093139648
    },
    {
      "epoch": 5.991935729980469e-06,
      "model_forward_time": 2.814256429672241,
      "step": 80283
    },
    {
      "epoch": 5.991935729980469e-06,
      "step": 80283,
      "training_step_time": 8.667553424835205
    },
    {
      "epoch": 5.992889404296875e-06,
      "model_forward_time": 1.7726340293884277,
      "step": 80284
    },
    {
      "epoch": 5.992889404296875e-06,
      "step": 80284,
      "training_step_time": 7.681247711181641
    },
    {
      "epoch": 5.993843078613281e-06,
      "model_forward_time": 1.7552590370178223,
      "step": 80285
    },
    {
      "epoch": 5.993843078613281e-06,
      "step": 80285,
      "training_step_time": 8.055590629577637
    },
    {
      "epoch": 5.994796752929687e-06,
      "model_forward_time": 1.7513401508331299,
      "step": 80286
    },
    {
      "epoch": 5.994796752929687e-06,
      "step": 80286,
      "training_step_time": 7.5573647022247314
    },
    {
      "epoch": 5.995750427246094e-06,
      "model_forward_time": 1.7638537883758545,
      "step": 80287
    },
    {
      "epoch": 5.995750427246094e-06,
      "step": 80287,
      "training_step_time": 7.013725280761719
    },
    {
      "epoch": 5.9967041015625e-06,
      "model_forward_time": 1.7583141326904297,
      "step": 80288
    },
    {
      "epoch": 5.9967041015625e-06,
      "step": 80288,
      "training_step_time": 7.315538167953491
    },
    {
      "epoch": 5.997657775878906e-06,
      "model_forward_time": 1.7517249584197998,
      "step": 80289
    },
    {
      "epoch": 5.997657775878906e-06,
      "step": 80289,
      "training_step_time": 7.828741550445557
    },
    {
      "epoch": 5.9986114501953125e-06,
      "grad_norm": 0.07593341916799545,
      "learning_rate": 1.0251277683091e-06,
      "loss": 0.0541,
      "step": 80290
    },
    {
      "epoch": 5.9986114501953125e-06,
      "model_forward_time": 1.7842228412628174,
      "step": 80290
    },
    {
      "dynamics_loss_avg": 0.03966755960136652,
      "epoch": 5.9986114501953125e-06,
      "step": 80290
    },
    {
      "action_loss_avg": 0.0009986099146772176,
      "epoch": 5.9986114501953125e-06,
      "step": 80290
    },
    {
      "epoch": 5.9986114501953125e-06,
      "step": 80290,
      "training_step_time": 7.9457221031188965
    },
    {
      "epoch": 5.999565124511719e-06,
      "model_forward_time": 1.774014949798584,
      "step": 80291
    },
    {
      "epoch": 5.999565124511719e-06,
      "step": 80291,
      "training_step_time": 9.249762058258057
    },
    {
      "epoch": 6.000518798828125e-06,
      "model_forward_time": 1.7584228515625,
      "step": 80292
    },
    {
      "epoch": 6.000518798828125e-06,
      "step": 80292,
      "training_step_time": 10.460844993591309
    },
    {
      "epoch": 6.001472473144531e-06,
      "model_forward_time": 2.837160587310791,
      "step": 80293
    },
    {
      "epoch": 6.001472473144531e-06,
      "step": 80293,
      "training_step_time": 7.743758201599121
    },
    {
      "epoch": 6.002426147460937e-06,
      "model_forward_time": 1.7504358291625977,
      "step": 80294
    },
    {
      "epoch": 6.002426147460937e-06,
      "step": 80294,
      "training_step_time": 7.634620428085327
    },
    {
      "epoch": 6.003379821777344e-06,
      "model_forward_time": 1.7527797222137451,
      "step": 80295
    },
    {
      "epoch": 6.003379821777344e-06,
      "step": 80295,
      "training_step_time": 7.572650671005249
    },
    {
      "epoch": 6.00433349609375e-06,
      "model_forward_time": 1.7732536792755127,
      "step": 80296
    },
    {
      "epoch": 6.00433349609375e-06,
      "step": 80296,
      "training_step_time": 7.567029237747192
    },
    {
      "epoch": 6.005287170410156e-06,
      "model_forward_time": 1.75657320022583,
      "step": 80297
    },
    {
      "epoch": 6.005287170410156e-06,
      "step": 80297,
      "training_step_time": 7.132998466491699
    },
    {
      "epoch": 6.0062408447265625e-06,
      "model_forward_time": 1.7582061290740967,
      "step": 80298
    },
    {
      "epoch": 6.0062408447265625e-06,
      "step": 80298,
      "training_step_time": 8.509727954864502
    },
    {
      "epoch": 6.007194519042969e-06,
      "model_forward_time": 1.7644484043121338,
      "step": 80299
    },
    {
      "epoch": 6.007194519042969e-06,
      "step": 80299,
      "training_step_time": 9.048194885253906
    },
    {
      "epoch": 6.008148193359375e-06,
      "grad_norm": 0.12302936613559723,
      "learning_rate": 1.0241249199157827e-06,
      "loss": 0.0544,
      "step": 80300
    },
    {
      "epoch": 6.008148193359375e-06,
      "model_forward_time": 1.7564842700958252,
      "step": 80300
    },
    {
      "dynamics_loss_avg": 0.05045007802546024,
      "epoch": 6.008148193359375e-06,
      "step": 80300
    },
    {
      "action_loss_avg": 0.0010330295277526601,
      "epoch": 6.008148193359375e-06,
      "step": 80300
    },
    {
      "epoch": 6.008148193359375e-06,
      "step": 80300,
      "training_step_time": 8.645027160644531
    },
    {
      "epoch": 6.009101867675781e-06,
      "model_forward_time": 1.8263049125671387,
      "step": 80301
    },
    {
      "epoch": 6.009101867675781e-06,
      "step": 80301,
      "training_step_time": 8.594690561294556
    },
    {
      "epoch": 6.010055541992187e-06,
      "model_forward_time": 2.861203670501709,
      "step": 80302
    },
    {
      "epoch": 6.010055541992187e-06,
      "step": 80302,
      "training_step_time": 7.7699134349823
    },
    {
      "epoch": 6.011009216308594e-06,
      "model_forward_time": 1.7561793327331543,
      "step": 80303
    },
    {
      "epoch": 6.011009216308594e-06,
      "step": 80303,
      "training_step_time": 7.690100908279419
    },
    {
      "epoch": 6.011962890625e-06,
      "model_forward_time": 1.762162208557129,
      "step": 80304
    },
    {
      "epoch": 6.011962890625e-06,
      "step": 80304,
      "training_step_time": 7.5566229820251465
    },
    {
      "epoch": 6.012916564941406e-06,
      "model_forward_time": 1.7467167377471924,
      "step": 80305
    },
    {
      "epoch": 6.012916564941406e-06,
      "step": 80305,
      "training_step_time": 7.516493558883667
    },
    {
      "epoch": 6.0138702392578125e-06,
      "model_forward_time": 1.7528796195983887,
      "step": 80306
    },
    {
      "epoch": 6.0138702392578125e-06,
      "step": 80306,
      "training_step_time": 7.27747917175293
    },
    {
      "epoch": 6.014823913574219e-06,
      "model_forward_time": 1.7651028633117676,
      "step": 80307
    },
    {
      "epoch": 6.014823913574219e-06,
      "step": 80307,
      "training_step_time": 7.674260377883911
    },
    {
      "epoch": 6.015777587890625e-06,
      "model_forward_time": 1.770279884338379,
      "step": 80308
    },
    {
      "epoch": 6.015777587890625e-06,
      "step": 80308,
      "training_step_time": 8.213758945465088
    },
    {
      "epoch": 6.016731262207031e-06,
      "model_forward_time": 1.763087511062622,
      "step": 80309
    },
    {
      "epoch": 6.016731262207031e-06,
      "step": 80309,
      "training_step_time": 8.232537269592285
    },
    {
      "epoch": 6.017684936523437e-06,
      "grad_norm": 0.10644137859344482,
      "learning_rate": 1.0231225063181838e-06,
      "loss": 0.0556,
      "step": 80310
    },
    {
      "epoch": 6.017684936523437e-06,
      "model_forward_time": 1.8315341472625732,
      "step": 80310
    },
    {
      "dynamics_loss_avg": 0.05557174552232027,
      "epoch": 6.017684936523437e-06,
      "step": 80310
    },
    {
      "action_loss_avg": 0.001717395184095949,
      "epoch": 6.017684936523437e-06,
      "step": 80310
    },
    {
      "epoch": 6.017684936523437e-06,
      "step": 80310,
      "training_step_time": 7.753094911575317
    },
    {
      "epoch": 6.018638610839844e-06,
      "model_forward_time": 2.8543403148651123,
      "step": 80311
    },
    {
      "epoch": 6.018638610839844e-06,
      "step": 80311,
      "training_step_time": 8.272104740142822
    },
    {
      "epoch": 6.01959228515625e-06,
      "model_forward_time": 1.7510592937469482,
      "step": 80312
    },
    {
      "epoch": 6.01959228515625e-06,
      "step": 80312,
      "training_step_time": 8.230581998825073
    },
    {
      "epoch": 6.020545959472656e-06,
      "model_forward_time": 1.7595853805541992,
      "step": 80313
    },
    {
      "epoch": 6.020545959472656e-06,
      "step": 80313,
      "training_step_time": 7.562379360198975
    },
    {
      "epoch": 6.0214996337890626e-06,
      "model_forward_time": 1.4113333225250244,
      "step": 80314
    },
    {
      "epoch": 6.0214996337890626e-06,
      "step": 80314,
      "training_step_time": 7.648333311080933
    },
    {
      "epoch": 6.022453308105469e-06,
      "model_forward_time": 1.74603271484375,
      "step": 80315
    },
    {
      "epoch": 6.022453308105469e-06,
      "step": 80315,
      "training_step_time": 6.659168720245361
    },
    {
      "epoch": 6.023406982421875e-06,
      "model_forward_time": 1.7680058479309082,
      "step": 80316
    },
    {
      "epoch": 6.023406982421875e-06,
      "step": 80316,
      "training_step_time": 7.675404071807861
    },
    {
      "epoch": 6.024360656738281e-06,
      "model_forward_time": 1.7721128463745117,
      "step": 80317
    },
    {
      "epoch": 6.024360656738281e-06,
      "step": 80317,
      "training_step_time": 8.514049053192139
    },
    {
      "epoch": 6.025314331054687e-06,
      "model_forward_time": 1.7770929336547852,
      "step": 80318
    },
    {
      "epoch": 6.025314331054687e-06,
      "step": 80318,
      "training_step_time": 9.647094964981079
    },
    {
      "epoch": 6.026268005371094e-06,
      "model_forward_time": 1.8423333168029785,
      "step": 80319
    },
    {
      "epoch": 6.026268005371094e-06,
      "step": 80319,
      "training_step_time": 8.716169834136963
    },
    {
      "epoch": 6.0272216796875e-06,
      "grad_norm": 0.12564517557621002,
      "learning_rate": 1.0221205276259317e-06,
      "loss": 0.0538,
      "step": 80320
    },
    {
      "epoch": 6.0272216796875e-06,
      "model_forward_time": 2.852947235107422,
      "step": 80320
    },
    {
      "dynamics_loss_avg": 0.050309980660676955,
      "epoch": 6.0272216796875e-06,
      "step": 80320
    },
    {
      "action_loss_avg": 0.001236098530353047,
      "epoch": 6.0272216796875e-06,
      "step": 80320
    },
    {
      "epoch": 6.0272216796875e-06,
      "step": 80320,
      "training_step_time": 8.621699571609497
    },
    {
      "epoch": 6.028175354003906e-06,
      "model_forward_time": 1.7748959064483643,
      "step": 80321
    },
    {
      "epoch": 6.028175354003906e-06,
      "step": 80321,
      "training_step_time": 7.631473064422607
    },
    {
      "epoch": 6.0291290283203126e-06,
      "model_forward_time": 1.7647042274475098,
      "step": 80322
    },
    {
      "epoch": 6.0291290283203126e-06,
      "step": 80322,
      "training_step_time": 7.522165536880493
    },
    {
      "epoch": 6.030082702636719e-06,
      "model_forward_time": 1.7758347988128662,
      "step": 80323
    },
    {
      "epoch": 6.030082702636719e-06,
      "step": 80323,
      "training_step_time": 7.558549165725708
    },
    {
      "epoch": 6.031036376953125e-06,
      "model_forward_time": 1.7686591148376465,
      "step": 80324
    },
    {
      "epoch": 6.031036376953125e-06,
      "step": 80324,
      "training_step_time": 7.36163854598999
    },
    {
      "epoch": 6.031990051269531e-06,
      "model_forward_time": 1.7688689231872559,
      "step": 80325
    },
    {
      "epoch": 6.031990051269531e-06,
      "step": 80325,
      "training_step_time": 7.646061420440674
    },
    {
      "epoch": 6.032943725585937e-06,
      "model_forward_time": 1.7789089679718018,
      "step": 80326
    },
    {
      "epoch": 6.032943725585937e-06,
      "step": 80326,
      "training_step_time": 7.607945203781128
    },
    {
      "epoch": 6.033897399902344e-06,
      "model_forward_time": 1.7678744792938232,
      "step": 80327
    },
    {
      "epoch": 6.033897399902344e-06,
      "step": 80327,
      "training_step_time": 8.69729208946228
    },
    {
      "epoch": 6.03485107421875e-06,
      "model_forward_time": 1.8003032207489014,
      "step": 80328
    },
    {
      "epoch": 6.03485107421875e-06,
      "step": 80328,
      "training_step_time": 7.593770265579224
    },
    {
      "epoch": 6.035804748535156e-06,
      "model_forward_time": 1.7802894115447998,
      "step": 80329
    },
    {
      "epoch": 6.035804748535156e-06,
      "step": 80329,
      "training_step_time": 8.338951110839844
    },
    {
      "epoch": 6.0367584228515626e-06,
      "grad_norm": 0.07382896542549133,
      "learning_rate": 1.0211189839485958e-06,
      "loss": 0.0547,
      "step": 80330
    },
    {
      "epoch": 6.0367584228515626e-06,
      "model_forward_time": 1.7560021877288818,
      "step": 80330
    },
    {
      "dynamics_loss_avg": 0.058708783332258464,
      "epoch": 6.0367584228515626e-06,
      "step": 80330
    },
    {
      "action_loss_avg": 0.0012051906407577917,
      "epoch": 6.0367584228515626e-06,
      "step": 80330
    },
    {
      "epoch": 6.0367584228515626e-06,
      "step": 80330,
      "training_step_time": 8.71838116645813
    },
    {
      "epoch": 6.037712097167969e-06,
      "model_forward_time": 1.7692322731018066,
      "step": 80331
    },
    {
      "epoch": 6.037712097167969e-06,
      "step": 80331,
      "training_step_time": 7.5124592781066895
    },
    {
      "epoch": 6.038665771484375e-06,
      "model_forward_time": 1.7799763679504395,
      "step": 80332
    },
    {
      "epoch": 6.038665771484375e-06,
      "step": 80332,
      "training_step_time": 7.50214958190918
    },
    {
      "epoch": 6.0396194458007814e-06,
      "model_forward_time": 1.77201509475708,
      "step": 80333
    },
    {
      "epoch": 6.0396194458007814e-06,
      "step": 80333,
      "training_step_time": 7.274442434310913
    },
    {
      "epoch": 6.040573120117187e-06,
      "model_forward_time": 1.758692979812622,
      "step": 80334
    },
    {
      "epoch": 6.040573120117187e-06,
      "step": 80334,
      "training_step_time": 8.542319536209106
    },
    {
      "epoch": 6.041526794433594e-06,
      "model_forward_time": 1.7651331424713135,
      "step": 80335
    },
    {
      "epoch": 6.041526794433594e-06,
      "step": 80335,
      "training_step_time": 7.560442924499512
    },
    {
      "epoch": 6.04248046875e-06,
      "model_forward_time": 1.7650032043457031,
      "step": 80336
    },
    {
      "epoch": 6.04248046875e-06,
      "step": 80336,
      "training_step_time": 8.061059713363647
    },
    {
      "epoch": 6.043434143066406e-06,
      "model_forward_time": 1.7744622230529785,
      "step": 80337
    },
    {
      "epoch": 6.043434143066406e-06,
      "step": 80337,
      "training_step_time": 8.822571992874146
    },
    {
      "epoch": 6.0443878173828126e-06,
      "model_forward_time": 1.7729332447052002,
      "step": 80338
    },
    {
      "epoch": 6.0443878173828126e-06,
      "step": 80338,
      "training_step_time": 7.626073837280273
    },
    {
      "epoch": 6.045341491699219e-06,
      "model_forward_time": 2.861021041870117,
      "step": 80339
    },
    {
      "epoch": 6.045341491699219e-06,
      "step": 80339,
      "training_step_time": 8.622312068939209
    },
    {
      "epoch": 6.046295166015625e-06,
      "grad_norm": 0.09344739466905594,
      "learning_rate": 1.0201178753957075e-06,
      "loss": 0.0558,
      "step": 80340
    },
    {
      "epoch": 6.046295166015625e-06,
      "model_forward_time": 1.76499605178833,
      "step": 80340
    },
    {
      "dynamics_loss_avg": 0.05494726244360208,
      "epoch": 6.046295166015625e-06,
      "step": 80340
    },
    {
      "action_loss_avg": 0.0018504166509956121,
      "epoch": 6.046295166015625e-06,
      "step": 80340
    },
    {
      "epoch": 6.046295166015625e-06,
      "step": 80340,
      "training_step_time": 7.698115825653076
    },
    {
      "epoch": 6.0472488403320314e-06,
      "model_forward_time": 1.778487205505371,
      "step": 80341
    },
    {
      "epoch": 6.0472488403320314e-06,
      "step": 80341,
      "training_step_time": 7.488234758377075
    },
    {
      "epoch": 6.048202514648437e-06,
      "model_forward_time": 1.7702655792236328,
      "step": 80342
    },
    {
      "epoch": 6.048202514648437e-06,
      "step": 80342,
      "training_step_time": 7.506357908248901
    },
    {
      "epoch": 6.049156188964844e-06,
      "model_forward_time": 1.761775016784668,
      "step": 80343
    },
    {
      "epoch": 6.049156188964844e-06,
      "step": 80343,
      "training_step_time": 7.532733678817749
    },
    {
      "epoch": 6.05010986328125e-06,
      "model_forward_time": 1.765286922454834,
      "step": 80344
    },
    {
      "epoch": 6.05010986328125e-06,
      "step": 80344,
      "training_step_time": 7.532508850097656
    },
    {
      "epoch": 6.051063537597656e-06,
      "model_forward_time": 1.7665107250213623,
      "step": 80345
    },
    {
      "epoch": 6.051063537597656e-06,
      "step": 80345,
      "training_step_time": 8.08263611793518
    },
    {
      "epoch": 6.052017211914063e-06,
      "model_forward_time": 1.0402088165283203,
      "step": 80346
    },
    {
      "epoch": 6.052017211914063e-06,
      "step": 80346,
      "training_step_time": 10.001119613647461
    },
    {
      "epoch": 6.052970886230469e-06,
      "model_forward_time": 1.840752124786377,
      "step": 80347
    },
    {
      "epoch": 6.052970886230469e-06,
      "step": 80347,
      "training_step_time": 8.546121835708618
    },
    {
      "epoch": 6.053924560546875e-06,
      "model_forward_time": 2.8333442211151123,
      "step": 80348
    },
    {
      "epoch": 6.053924560546875e-06,
      "step": 80348,
      "training_step_time": 8.266568422317505
    },
    {
      "epoch": 6.0548782348632814e-06,
      "model_forward_time": 1.772376298904419,
      "step": 80349
    },
    {
      "epoch": 6.0548782348632814e-06,
      "step": 80349,
      "training_step_time": 8.487869501113892
    },
    {
      "epoch": 6.055831909179687e-06,
      "grad_norm": 0.1019010841846466,
      "learning_rate": 1.0191172020767431e-06,
      "loss": 0.0553,
      "step": 80350
    },
    {
      "epoch": 6.055831909179687e-06,
      "model_forward_time": 1.7477991580963135,
      "step": 80350
    },
    {
      "dynamics_loss_avg": 0.035803813487291336,
      "epoch": 6.055831909179687e-06,
      "step": 80350
    },
    {
      "action_loss_avg": 0.0011883859289810062,
      "epoch": 6.055831909179687e-06,
      "step": 80350
    },
    {
      "epoch": 6.055831909179687e-06,
      "step": 80350,
      "training_step_time": 8.155641317367554
    },
    {
      "epoch": 6.056785583496094e-06,
      "model_forward_time": 1.7616567611694336,
      "step": 80351
    },
    {
      "epoch": 6.056785583496094e-06,
      "step": 80351,
      "training_step_time": 7.491653680801392
    },
    {
      "epoch": 6.0577392578125e-06,
      "model_forward_time": 1.8585634231567383,
      "step": 80352
    },
    {
      "epoch": 6.0577392578125e-06,
      "step": 80352,
      "training_step_time": 7.4912495613098145
    },
    {
      "epoch": 6.058692932128906e-06,
      "model_forward_time": 1.760253667831421,
      "step": 80353
    },
    {
      "epoch": 6.058692932128906e-06,
      "step": 80353,
      "training_step_time": 7.5477635860443115
    },
    {
      "epoch": 6.059646606445313e-06,
      "model_forward_time": 1.7553682327270508,
      "step": 80354
    },
    {
      "epoch": 6.059646606445313e-06,
      "step": 80354,
      "training_step_time": 8.437758684158325
    },
    {
      "epoch": 6.060600280761719e-06,
      "model_forward_time": 1.7635776996612549,
      "step": 80355
    },
    {
      "epoch": 6.060600280761719e-06,
      "step": 80355,
      "training_step_time": 8.26098084449768
    },
    {
      "epoch": 6.061553955078125e-06,
      "model_forward_time": 1.7293915748596191,
      "step": 80356
    },
    {
      "epoch": 6.061553955078125e-06,
      "step": 80356,
      "training_step_time": 8.182553052902222
    },
    {
      "epoch": 6.0625076293945314e-06,
      "model_forward_time": 2.3689045906066895,
      "step": 80357
    },
    {
      "epoch": 6.0625076293945314e-06,
      "step": 80357,
      "training_step_time": 10.349277973175049
    },
    {
      "epoch": 6.063461303710937e-06,
      "model_forward_time": 1.735086441040039,
      "step": 80358
    },
    {
      "epoch": 6.063461303710937e-06,
      "step": 80358,
      "training_step_time": 7.6885528564453125
    },
    {
      "epoch": 6.064414978027344e-06,
      "model_forward_time": 1.7378160953521729,
      "step": 80359
    },
    {
      "epoch": 6.064414978027344e-06,
      "step": 80359,
      "training_step_time": 7.463748931884766
    },
    {
      "epoch": 6.06536865234375e-06,
      "grad_norm": 0.07365267723798752,
      "learning_rate": 1.0181169641011363e-06,
      "loss": 0.0553,
      "step": 80360
    },
    {
      "epoch": 6.06536865234375e-06,
      "model_forward_time": 1.7573151588439941,
      "step": 80360
    },
    {
      "dynamics_loss_avg": 0.04866334665566683,
      "epoch": 6.06536865234375e-06,
      "step": 80360
    },
    {
      "action_loss_avg": 0.0012768155604135246,
      "epoch": 6.06536865234375e-06,
      "step": 80360
    },
    {
      "epoch": 6.06536865234375e-06,
      "step": 80360,
      "training_step_time": 7.506645679473877
    },
    {
      "epoch": 6.066322326660156e-06,
      "model_forward_time": 1.7454805374145508,
      "step": 80361
    },
    {
      "epoch": 6.066322326660156e-06,
      "step": 80361,
      "training_step_time": 7.566845655441284
    },
    {
      "epoch": 6.067276000976563e-06,
      "model_forward_time": 1.7616550922393799,
      "step": 80362
    },
    {
      "epoch": 6.067276000976563e-06,
      "step": 80362,
      "training_step_time": 7.666250705718994
    },
    {
      "epoch": 6.068229675292969e-06,
      "model_forward_time": 1.848839521408081,
      "step": 80363
    },
    {
      "epoch": 6.068229675292969e-06,
      "step": 80363,
      "training_step_time": 9.90021800994873
    },
    {
      "epoch": 6.069183349609375e-06,
      "model_forward_time": 1.764606237411499,
      "step": 80364
    },
    {
      "epoch": 6.069183349609375e-06,
      "step": 80364,
      "training_step_time": 8.655352354049683
    },
    {
      "epoch": 6.0701370239257815e-06,
      "model_forward_time": 1.7580699920654297,
      "step": 80365
    },
    {
      "epoch": 6.0701370239257815e-06,
      "step": 80365,
      "training_step_time": 7.622046709060669
    },
    {
      "epoch": 6.071090698242187e-06,
      "model_forward_time": 1.754218578338623,
      "step": 80366
    },
    {
      "epoch": 6.071090698242187e-06,
      "step": 80366,
      "training_step_time": 8.824784755706787
    },
    {
      "epoch": 6.072044372558594e-06,
      "model_forward_time": 1.749927043914795,
      "step": 80367
    },
    {
      "epoch": 6.072044372558594e-06,
      "step": 80367,
      "training_step_time": 7.601625919342041
    },
    {
      "epoch": 6.072998046875e-06,
      "model_forward_time": 1.7661941051483154,
      "step": 80368
    },
    {
      "epoch": 6.072998046875e-06,
      "step": 80368,
      "training_step_time": 7.598343133926392
    },
    {
      "epoch": 6.073951721191406e-06,
      "model_forward_time": 1.7538695335388184,
      "step": 80369
    },
    {
      "epoch": 6.073951721191406e-06,
      "step": 80369,
      "training_step_time": 7.636702060699463
    },
    {
      "epoch": 6.074905395507813e-06,
      "grad_norm": 0.08344408124685287,
      "learning_rate": 1.0171171615782722e-06,
      "loss": 0.057,
      "step": 80370
    },
    {
      "epoch": 6.074905395507813e-06,
      "model_forward_time": 1.7423312664031982,
      "step": 80370
    },
    {
      "dynamics_loss_avg": 0.05372702032327652,
      "epoch": 6.074905395507813e-06,
      "step": 80370
    },
    {
      "action_loss_avg": 0.0011150582737172953,
      "epoch": 6.074905395507813e-06,
      "step": 80370
    },
    {
      "epoch": 6.074905395507813e-06,
      "step": 80370,
      "training_step_time": 7.4353883266448975
    },
    {
      "epoch": 6.075859069824219e-06,
      "model_forward_time": 1.7622909545898438,
      "step": 80371
    },
    {
      "epoch": 6.075859069824219e-06,
      "step": 80371,
      "training_step_time": 7.588294506072998
    },
    {
      "epoch": 6.076812744140625e-06,
      "model_forward_time": 1.8198521137237549,
      "step": 80372
    },
    {
      "epoch": 6.076812744140625e-06,
      "step": 80372,
      "training_step_time": 8.282251119613647
    },
    {
      "epoch": 6.0777664184570315e-06,
      "model_forward_time": 1.7661809921264648,
      "step": 80373
    },
    {
      "epoch": 6.0777664184570315e-06,
      "step": 80373,
      "training_step_time": 9.689493179321289
    },
    {
      "epoch": 6.078720092773437e-06,
      "model_forward_time": 1.763962984085083,
      "step": 80374
    },
    {
      "epoch": 6.078720092773437e-06,
      "step": 80374,
      "training_step_time": 8.681393384933472
    },
    {
      "epoch": 6.079673767089844e-06,
      "model_forward_time": 1.7558834552764893,
      "step": 80375
    },
    {
      "epoch": 6.079673767089844e-06,
      "step": 80375,
      "training_step_time": 8.52260422706604
    },
    {
      "epoch": 6.08062744140625e-06,
      "model_forward_time": 2.370413303375244,
      "step": 80376
    },
    {
      "epoch": 6.08062744140625e-06,
      "step": 80376,
      "training_step_time": 7.803824186325073
    },
    {
      "epoch": 6.081581115722656e-06,
      "model_forward_time": 1.758643388748169,
      "step": 80377
    },
    {
      "epoch": 6.081581115722656e-06,
      "step": 80377,
      "training_step_time": 8.559381246566772
    },
    {
      "epoch": 6.082534790039063e-06,
      "model_forward_time": 1.7488069534301758,
      "step": 80378
    },
    {
      "epoch": 6.082534790039063e-06,
      "step": 80378,
      "training_step_time": 7.6438963413238525
    },
    {
      "epoch": 6.083488464355468e-06,
      "model_forward_time": 1.7489354610443115,
      "step": 80379
    },
    {
      "epoch": 6.083488464355468e-06,
      "step": 80379,
      "training_step_time": 7.100722789764404
    },
    {
      "epoch": 6.084442138671875e-06,
      "grad_norm": 0.07172462344169617,
      "learning_rate": 1.0161177946174867e-06,
      "loss": 0.0575,
      "step": 80380
    },
    {
      "epoch": 6.084442138671875e-06,
      "model_forward_time": 1.7619514465332031,
      "step": 80380
    },
    {
      "dynamics_loss_avg": 0.04644989911466837,
      "epoch": 6.084442138671875e-06,
      "step": 80380
    },
    {
      "action_loss_avg": 0.0017241720459423959,
      "epoch": 6.084442138671875e-06,
      "step": 80380
    },
    {
      "epoch": 6.084442138671875e-06,
      "step": 80380,
      "training_step_time": 7.729056358337402
    },
    {
      "epoch": 6.0853958129882815e-06,
      "model_forward_time": 1.7885408401489258,
      "step": 80381
    },
    {
      "epoch": 6.0853958129882815e-06,
      "step": 80381,
      "training_step_time": 7.606761932373047
    },
    {
      "epoch": 6.086349487304687e-06,
      "model_forward_time": 1.7673039436340332,
      "step": 80382
    },
    {
      "epoch": 6.086349487304687e-06,
      "step": 80382,
      "training_step_time": 8.598410606384277
    },
    {
      "epoch": 6.087303161621094e-06,
      "model_forward_time": 1.7551767826080322,
      "step": 80383
    },
    {
      "epoch": 6.087303161621094e-06,
      "step": 80383,
      "training_step_time": 7.795648574829102
    },
    {
      "epoch": 6.0882568359375e-06,
      "model_forward_time": 1.7633686065673828,
      "step": 80384
    },
    {
      "epoch": 6.0882568359375e-06,
      "step": 80384,
      "training_step_time": 10.31592583656311
    },
    {
      "epoch": 6.089210510253906e-06,
      "model_forward_time": 2.372990846633911,
      "step": 80385
    },
    {
      "epoch": 6.089210510253906e-06,
      "step": 80385,
      "training_step_time": 8.718357563018799
    },
    {
      "epoch": 6.090164184570313e-06,
      "model_forward_time": 1.7669827938079834,
      "step": 80386
    },
    {
      "epoch": 6.090164184570313e-06,
      "step": 80386,
      "training_step_time": 7.696639776229858
    },
    {
      "epoch": 6.091117858886718e-06,
      "model_forward_time": 1.7597732543945312,
      "step": 80387
    },
    {
      "epoch": 6.091117858886718e-06,
      "step": 80387,
      "training_step_time": 7.649986267089844
    },
    {
      "epoch": 6.092071533203125e-06,
      "model_forward_time": 1.7661123275756836,
      "step": 80388
    },
    {
      "epoch": 6.092071533203125e-06,
      "step": 80388,
      "training_step_time": 7.379333972930908
    },
    {
      "epoch": 6.0930252075195315e-06,
      "model_forward_time": 1.7474207878112793,
      "step": 80389
    },
    {
      "epoch": 6.0930252075195315e-06,
      "step": 80389,
      "training_step_time": 7.475049734115601
    },
    {
      "epoch": 6.093978881835937e-06,
      "grad_norm": 0.11326317489147186,
      "learning_rate": 1.0151188633280706e-06,
      "loss": 0.0573,
      "step": 80390
    },
    {
      "epoch": 6.093978881835937e-06,
      "model_forward_time": 1.7649362087249756,
      "step": 80390
    },
    {
      "dynamics_loss_avg": 0.03510162178426981,
      "epoch": 6.093978881835937e-06,
      "step": 80390
    },
    {
      "action_loss_avg": 0.001454358102637343,
      "epoch": 6.093978881835937e-06,
      "step": 80390
    },
    {
      "epoch": 6.093978881835937e-06,
      "step": 80390,
      "training_step_time": 7.608933210372925
    },
    {
      "epoch": 6.094932556152344e-06,
      "model_forward_time": 1.7626643180847168,
      "step": 80391
    },
    {
      "epoch": 6.094932556152344e-06,
      "step": 80391,
      "training_step_time": 8.153026819229126
    },
    {
      "epoch": 6.09588623046875e-06,
      "model_forward_time": 1.7553369998931885,
      "step": 80392
    },
    {
      "epoch": 6.09588623046875e-06,
      "step": 80392,
      "training_step_time": 9.382926940917969
    },
    {
      "epoch": 6.096839904785156e-06,
      "model_forward_time": 1.7611331939697266,
      "step": 80393
    },
    {
      "epoch": 6.096839904785156e-06,
      "step": 80393,
      "training_step_time": 8.783783674240112
    },
    {
      "epoch": 6.097793579101563e-06,
      "model_forward_time": 2.3781003952026367,
      "step": 80394
    },
    {
      "epoch": 6.097793579101563e-06,
      "step": 80394,
      "training_step_time": 7.785739183425903
    },
    {
      "epoch": 6.098747253417968e-06,
      "model_forward_time": 1.3906104564666748,
      "step": 80395
    },
    {
      "epoch": 6.098747253417968e-06,
      "step": 80395,
      "training_step_time": 8.676942586898804
    },
    {
      "epoch": 6.099700927734375e-06,
      "model_forward_time": 1.7672748565673828,
      "step": 80396
    },
    {
      "epoch": 6.099700927734375e-06,
      "step": 80396,
      "training_step_time": 7.511754989624023
    },
    {
      "epoch": 6.1006546020507815e-06,
      "model_forward_time": 1.7574033737182617,
      "step": 80397
    },
    {
      "epoch": 6.1006546020507815e-06,
      "step": 80397,
      "training_step_time": 7.414261102676392
    },
    {
      "epoch": 6.101608276367187e-06,
      "model_forward_time": 1.7533447742462158,
      "step": 80398
    },
    {
      "epoch": 6.101608276367187e-06,
      "step": 80398,
      "training_step_time": 7.550314664840698
    },
    {
      "epoch": 6.102561950683594e-06,
      "model_forward_time": 1.7703070640563965,
      "step": 80399
    },
    {
      "epoch": 6.102561950683594e-06,
      "step": 80399,
      "training_step_time": 8.65664005279541
    },
    {
      "epoch": 6.103515625e-06,
      "grad_norm": 0.10541640222072601,
      "learning_rate": 1.0141203678192634e-06,
      "loss": 0.0572,
      "step": 80400
    },
    {
      "epoch": 6.103515625e-06,
      "model_forward_time": 1.8644769191741943,
      "step": 80400
    },
    {
      "dynamics_loss_avg": 0.04669463559985161,
      "epoch": 6.103515625e-06,
      "step": 80400
    },
    {
      "action_loss_avg": 0.0012478303571697325,
      "epoch": 6.103515625e-06,
      "step": 80400
    },
    {
      "epoch": 6.103515625e-06,
      "step": 80400,
      "training_step_time": 7.654395818710327
    },
    {
      "epoch": 6.104469299316406e-06,
      "model_forward_time": 1.7565381526947021,
      "step": 80401
    },
    {
      "epoch": 6.104469299316406e-06,
      "step": 80401,
      "training_step_time": 7.7536091804504395
    },
    {
      "epoch": 6.105422973632813e-06,
      "model_forward_time": 1.7628297805786133,
      "step": 80402
    },
    {
      "epoch": 6.105422973632813e-06,
      "step": 80402,
      "training_step_time": 9.519910335540771
    },
    {
      "epoch": 6.106376647949218e-06,
      "model_forward_time": 2.384054183959961,
      "step": 80403
    },
    {
      "epoch": 6.106376647949218e-06,
      "step": 80403,
      "training_step_time": 8.656906604766846
    },
    {
      "epoch": 6.107330322265625e-06,
      "model_forward_time": 1.7664010524749756,
      "step": 80404
    },
    {
      "epoch": 6.107330322265625e-06,
      "step": 80404,
      "training_step_time": 9.272995948791504
    },
    {
      "epoch": 6.1082839965820315e-06,
      "model_forward_time": 1.761164665222168,
      "step": 80405
    },
    {
      "epoch": 6.1082839965820315e-06,
      "step": 80405,
      "training_step_time": 7.537390232086182
    },
    {
      "epoch": 6.109237670898437e-06,
      "model_forward_time": 1.7587969303131104,
      "step": 80406
    },
    {
      "epoch": 6.109237670898437e-06,
      "step": 80406,
      "training_step_time": 7.669016122817993
    },
    {
      "epoch": 6.110191345214844e-06,
      "model_forward_time": 1.7777125835418701,
      "step": 80407
    },
    {
      "epoch": 6.110191345214844e-06,
      "step": 80407,
      "training_step_time": 7.5679051876068115
    },
    {
      "epoch": 6.11114501953125e-06,
      "model_forward_time": 1.744499921798706,
      "step": 80408
    },
    {
      "epoch": 6.11114501953125e-06,
      "step": 80408,
      "training_step_time": 7.697472333908081
    },
    {
      "epoch": 6.112098693847656e-06,
      "model_forward_time": 1.8450751304626465,
      "step": 80409
    },
    {
      "epoch": 6.112098693847656e-06,
      "step": 80409,
      "training_step_time": 7.710919380187988
    },
    {
      "epoch": 6.113052368164063e-06,
      "grad_norm": 0.20637451112270355,
      "learning_rate": 1.0131223082002612e-06,
      "loss": 0.0584,
      "step": 80410
    },
    {
      "epoch": 6.113052368164063e-06,
      "model_forward_time": 1.7472107410430908,
      "step": 80410
    },
    {
      "dynamics_loss_avg": 0.055326235853135584,
      "epoch": 6.113052368164063e-06,
      "step": 80410
    },
    {
      "action_loss_avg": 0.001313346516690217,
      "epoch": 6.113052368164063e-06,
      "step": 80410
    },
    {
      "epoch": 6.113052368164063e-06,
      "step": 80410,
      "training_step_time": 8.223514556884766
    },
    {
      "epoch": 6.114006042480468e-06,
      "model_forward_time": 1.7579092979431152,
      "step": 80411
    },
    {
      "epoch": 6.114006042480468e-06,
      "step": 80411,
      "training_step_time": 9.219481229782104
    },
    {
      "epoch": 6.114959716796875e-06,
      "model_forward_time": 2.398754596710205,
      "step": 80412
    },
    {
      "epoch": 6.114959716796875e-06,
      "step": 80412,
      "training_step_time": 8.169368505477905
    },
    {
      "epoch": 6.1159133911132815e-06,
      "model_forward_time": 1.7621524333953857,
      "step": 80413
    },
    {
      "epoch": 6.1159133911132815e-06,
      "step": 80413,
      "training_step_time": 8.60529351234436
    },
    {
      "epoch": 6.116867065429687e-06,
      "model_forward_time": 1.7670598030090332,
      "step": 80414
    },
    {
      "epoch": 6.116867065429687e-06,
      "step": 80414,
      "training_step_time": 7.731151103973389
    },
    {
      "epoch": 6.117820739746094e-06,
      "model_forward_time": 1.7705233097076416,
      "step": 80415
    },
    {
      "epoch": 6.117820739746094e-06,
      "step": 80415,
      "training_step_time": 7.66055703163147
    },
    {
      "epoch": 6.1187744140625e-06,
      "model_forward_time": 1.7627208232879639,
      "step": 80416
    },
    {
      "epoch": 6.1187744140625e-06,
      "step": 80416,
      "training_step_time": 7.691978216171265
    },
    {
      "epoch": 6.119728088378906e-06,
      "model_forward_time": 1.7520015239715576,
      "step": 80417
    },
    {
      "epoch": 6.119728088378906e-06,
      "step": 80417,
      "training_step_time": 7.684095859527588
    },
    {
      "epoch": 6.120681762695313e-06,
      "model_forward_time": 1.8110661506652832,
      "step": 80418
    },
    {
      "epoch": 6.120681762695313e-06,
      "step": 80418,
      "training_step_time": 7.6939897537231445
    },
    {
      "epoch": 6.121635437011718e-06,
      "model_forward_time": 1.7629244327545166,
      "step": 80419
    },
    {
      "epoch": 6.121635437011718e-06,
      "step": 80419,
      "training_step_time": 7.74066424369812
    },
    {
      "epoch": 6.122589111328125e-06,
      "grad_norm": 0.09525305777788162,
      "learning_rate": 1.012124684580208e-06,
      "loss": 0.0592,
      "step": 80420
    },
    {
      "epoch": 6.122589111328125e-06,
      "model_forward_time": 1.7427401542663574,
      "step": 80420
    },
    {
      "dynamics_loss_avg": 0.06737311594188214,
      "epoch": 6.122589111328125e-06,
      "step": 80420
    },
    {
      "action_loss_avg": 0.001235671600443311,
      "epoch": 6.122589111328125e-06,
      "step": 80420
    },
    {
      "epoch": 6.122589111328125e-06,
      "step": 80420,
      "training_step_time": 9.080798387527466
    },
    {
      "epoch": 6.1235427856445315e-06,
      "model_forward_time": 2.4146437644958496,
      "step": 80421
    },
    {
      "epoch": 6.1235427856445315e-06,
      "step": 80421,
      "training_step_time": 9.552908897399902
    },
    {
      "epoch": 6.124496459960937e-06,
      "model_forward_time": 1.7668991088867188,
      "step": 80422
    },
    {
      "epoch": 6.124496459960937e-06,
      "step": 80422,
      "training_step_time": 9.023682117462158
    },
    {
      "epoch": 6.125450134277344e-06,
      "model_forward_time": 1.7512052059173584,
      "step": 80423
    },
    {
      "epoch": 6.125450134277344e-06,
      "step": 80423,
      "training_step_time": 7.610803127288818
    },
    {
      "epoch": 6.12640380859375e-06,
      "model_forward_time": 1.7597246170043945,
      "step": 80424
    },
    {
      "epoch": 6.12640380859375e-06,
      "step": 80424,
      "training_step_time": 7.58534049987793
    },
    {
      "epoch": 6.127357482910156e-06,
      "model_forward_time": 1.7623391151428223,
      "step": 80425
    },
    {
      "epoch": 6.127357482910156e-06,
      "step": 80425,
      "training_step_time": 7.043551206588745
    },
    {
      "epoch": 6.128311157226563e-06,
      "model_forward_time": 1.7654199600219727,
      "step": 80426
    },
    {
      "epoch": 6.128311157226563e-06,
      "step": 80426,
      "training_step_time": 9.43720293045044
    },
    {
      "epoch": 6.129264831542968e-06,
      "model_forward_time": 1.7854983806610107,
      "step": 80427
    },
    {
      "epoch": 6.129264831542968e-06,
      "step": 80427,
      "training_step_time": 7.566262245178223
    },
    {
      "epoch": 6.130218505859375e-06,
      "model_forward_time": 1.7442872524261475,
      "step": 80428
    },
    {
      "epoch": 6.130218505859375e-06,
      "step": 80428,
      "training_step_time": 7.609551191329956
    },
    {
      "epoch": 6.1311721801757815e-06,
      "model_forward_time": 1.7627074718475342,
      "step": 80429
    },
    {
      "epoch": 6.1311721801757815e-06,
      "step": 80429,
      "training_step_time": 9.23563265800476
    },
    {
      "epoch": 6.132125854492187e-06,
      "grad_norm": 0.07614831626415253,
      "learning_rate": 1.011127497068204e-06,
      "loss": 0.0577,
      "step": 80430
    },
    {
      "epoch": 6.132125854492187e-06,
      "model_forward_time": 2.3522186279296875,
      "step": 80430
    },
    {
      "dynamics_loss_avg": 0.04783917814493179,
      "epoch": 6.132125854492187e-06,
      "step": 80430
    },
    {
      "action_loss_avg": 0.0010166299412958323,
      "epoch": 6.132125854492187e-06,
      "step": 80430
    },
    {
      "epoch": 6.132125854492187e-06,
      "step": 80430,
      "training_step_time": 11.689039707183838
    },
    {
      "epoch": 6.133079528808594e-06,
      "model_forward_time": 1.763190746307373,
      "step": 80431
    },
    {
      "epoch": 6.133079528808594e-06,
      "step": 80431,
      "training_step_time": 8.453952312469482
    },
    {
      "epoch": 6.134033203125e-06,
      "model_forward_time": 1.3931465148925781,
      "step": 80432
    },
    {
      "epoch": 6.134033203125e-06,
      "step": 80432,
      "training_step_time": 7.711507797241211
    },
    {
      "epoch": 6.134986877441406e-06,
      "model_forward_time": 1.7659471035003662,
      "step": 80433
    },
    {
      "epoch": 6.134986877441406e-06,
      "step": 80433,
      "training_step_time": 7.63051176071167
    },
    {
      "epoch": 6.135940551757813e-06,
      "model_forward_time": 1.757239818572998,
      "step": 80434
    },
    {
      "epoch": 6.135940551757813e-06,
      "step": 80434,
      "training_step_time": 7.637087106704712
    },
    {
      "epoch": 6.136894226074218e-06,
      "model_forward_time": 1.759453296661377,
      "step": 80435
    },
    {
      "epoch": 6.136894226074218e-06,
      "step": 80435,
      "training_step_time": 7.6991260051727295
    },
    {
      "epoch": 6.137847900390625e-06,
      "model_forward_time": 1.7646522521972656,
      "step": 80436
    },
    {
      "epoch": 6.137847900390625e-06,
      "step": 80436,
      "training_step_time": 9.540709018707275
    },
    {
      "epoch": 6.1388015747070315e-06,
      "model_forward_time": 1.76584792137146,
      "step": 80437
    },
    {
      "epoch": 6.1388015747070315e-06,
      "step": 80437,
      "training_step_time": 7.611264228820801
    },
    {
      "epoch": 6.139755249023437e-06,
      "model_forward_time": 1.7656755447387695,
      "step": 80438
    },
    {
      "epoch": 6.139755249023437e-06,
      "step": 80438,
      "training_step_time": 9.504638910293579
    },
    {
      "epoch": 6.140708923339844e-06,
      "model_forward_time": 1.393681526184082,
      "step": 80439
    },
    {
      "epoch": 6.140708923339844e-06,
      "step": 80439,
      "training_step_time": 8.723624229431152
    },
    {
      "epoch": 6.14166259765625e-06,
      "grad_norm": 0.08056896924972534,
      "learning_rate": 1.0101307457732988e-06,
      "loss": 0.0567,
      "step": 80440
    },
    {
      "epoch": 6.14166259765625e-06,
      "model_forward_time": 2.3786706924438477,
      "step": 80440
    },
    {
      "dynamics_loss_avg": 0.05503906104713678,
      "epoch": 6.14166259765625e-06,
      "step": 80440
    },
    {
      "action_loss_avg": 0.001078051351942122,
      "epoch": 6.14166259765625e-06,
      "step": 80440
    },
    {
      "epoch": 6.14166259765625e-06,
      "step": 80440,
      "training_step_time": 8.658236026763916
    },
    {
      "epoch": 6.142616271972656e-06,
      "model_forward_time": 1.766662836074829,
      "step": 80441
    },
    {
      "epoch": 6.142616271972656e-06,
      "step": 80441,
      "training_step_time": 7.637606859207153
    },
    {
      "epoch": 6.143569946289063e-06,
      "model_forward_time": 1.7720224857330322,
      "step": 80442
    },
    {
      "epoch": 6.143569946289063e-06,
      "step": 80442,
      "training_step_time": 8.159026861190796
    },
    {
      "epoch": 6.144523620605468e-06,
      "model_forward_time": 1.7663962841033936,
      "step": 80443
    },
    {
      "epoch": 6.144523620605468e-06,
      "step": 80443,
      "training_step_time": 7.510403633117676
    },
    {
      "epoch": 6.145477294921875e-06,
      "model_forward_time": 1.7685599327087402,
      "step": 80444
    },
    {
      "epoch": 6.145477294921875e-06,
      "step": 80444,
      "training_step_time": 7.5688087940216064
    },
    {
      "epoch": 6.1464309692382815e-06,
      "model_forward_time": 1.7659883499145508,
      "step": 80445
    },
    {
      "epoch": 6.1464309692382815e-06,
      "step": 80445,
      "training_step_time": 8.896764278411865
    },
    {
      "epoch": 6.147384643554687e-06,
      "model_forward_time": 1.8755533695220947,
      "step": 80446
    },
    {
      "epoch": 6.147384643554687e-06,
      "step": 80446,
      "training_step_time": 7.553020000457764
    },
    {
      "epoch": 6.148338317871094e-06,
      "model_forward_time": 1.7662158012390137,
      "step": 80447
    },
    {
      "epoch": 6.148338317871094e-06,
      "step": 80447,
      "training_step_time": 8.652642488479614
    },
    {
      "epoch": 6.1492919921875e-06,
      "model_forward_time": 1.7472035884857178,
      "step": 80448
    },
    {
      "epoch": 6.1492919921875e-06,
      "step": 80448,
      "training_step_time": 9.227288007736206
    },
    {
      "epoch": 6.150245666503906e-06,
      "model_forward_time": 2.395944833755493,
      "step": 80449
    },
    {
      "epoch": 6.150245666503906e-06,
      "step": 80449,
      "training_step_time": 7.6206371784210205
    },
    {
      "epoch": 6.151199340820313e-06,
      "grad_norm": 0.13818685710430145,
      "learning_rate": 1.0091344308044987e-06,
      "loss": 0.0575,
      "step": 80450
    },
    {
      "epoch": 6.151199340820313e-06,
      "model_forward_time": 1.7537555694580078,
      "step": 80450
    },
    {
      "dynamics_loss_avg": 0.06535364501178265,
      "epoch": 6.151199340820313e-06,
      "step": 80450
    },
    {
      "action_loss_avg": 0.0017557941959239543,
      "epoch": 6.151199340820313e-06,
      "step": 80450
    },
    {
      "epoch": 6.151199340820313e-06,
      "step": 80450,
      "training_step_time": 7.534502267837524
    },
    {
      "epoch": 6.152153015136718e-06,
      "model_forward_time": 1.754673957824707,
      "step": 80451
    },
    {
      "epoch": 6.152153015136718e-06,
      "step": 80451,
      "training_step_time": 7.642949104309082
    },
    {
      "epoch": 6.153106689453125e-06,
      "model_forward_time": 1.7432124614715576,
      "step": 80452
    },
    {
      "epoch": 6.153106689453125e-06,
      "step": 80452,
      "training_step_time": 7.567389488220215
    },
    {
      "epoch": 6.1540603637695315e-06,
      "model_forward_time": 1.7662851810455322,
      "step": 80453
    },
    {
      "epoch": 6.1540603637695315e-06,
      "step": 80453,
      "training_step_time": 7.597122669219971
    },
    {
      "epoch": 6.155014038085937e-06,
      "model_forward_time": 1.7648882865905762,
      "step": 80454
    },
    {
      "epoch": 6.155014038085937e-06,
      "step": 80454,
      "training_step_time": 8.128497838973999
    },
    {
      "epoch": 6.155967712402344e-06,
      "model_forward_time": 1.8475983142852783,
      "step": 80455
    },
    {
      "epoch": 6.155967712402344e-06,
      "step": 80455,
      "training_step_time": 7.6855809688568115
    },
    {
      "epoch": 6.15692138671875e-06,
      "model_forward_time": 1.7679409980773926,
      "step": 80456
    },
    {
      "epoch": 6.15692138671875e-06,
      "step": 80456,
      "training_step_time": 8.684577465057373
    },
    {
      "epoch": 6.157875061035156e-06,
      "model_forward_time": 1.7699501514434814,
      "step": 80457
    },
    {
      "epoch": 6.157875061035156e-06,
      "step": 80457,
      "training_step_time": 9.055197954177856
    },
    {
      "epoch": 6.158828735351563e-06,
      "model_forward_time": 2.41837215423584,
      "step": 80458
    },
    {
      "epoch": 6.158828735351563e-06,
      "step": 80458,
      "training_step_time": 8.916036367416382
    },
    {
      "epoch": 6.159782409667968e-06,
      "model_forward_time": 1.7711973190307617,
      "step": 80459
    },
    {
      "epoch": 6.159782409667968e-06,
      "step": 80459,
      "training_step_time": 7.661715984344482
    },
    {
      "epoch": 6.160736083984375e-06,
      "grad_norm": 0.10054177790880203,
      "learning_rate": 1.0081385522707554e-06,
      "loss": 0.0575,
      "step": 80460
    },
    {
      "epoch": 6.160736083984375e-06,
      "model_forward_time": 1.758277416229248,
      "step": 80460
    },
    {
      "dynamics_loss_avg": 0.05180224608629942,
      "epoch": 6.160736083984375e-06,
      "step": 80460
    },
    {
      "action_loss_avg": 0.0011374997295206413,
      "epoch": 6.160736083984375e-06,
      "step": 80460
    },
    {
      "epoch": 6.160736083984375e-06,
      "step": 80460,
      "training_step_time": 7.811001777648926
    },
    {
      "epoch": 6.1616897583007815e-06,
      "model_forward_time": 1.7648417949676514,
      "step": 80461
    },
    {
      "epoch": 6.1616897583007815e-06,
      "step": 80461,
      "training_step_time": 7.65190315246582
    },
    {
      "epoch": 6.162643432617187e-06,
      "model_forward_time": 1.7458345890045166,
      "step": 80462
    },
    {
      "epoch": 6.162643432617187e-06,
      "step": 80462,
      "training_step_time": 7.696384906768799
    },
    {
      "epoch": 6.163597106933594e-06,
      "model_forward_time": 1.7650549411773682,
      "step": 80463
    },
    {
      "epoch": 6.163597106933594e-06,
      "step": 80463,
      "training_step_time": 7.604111909866333
    },
    {
      "epoch": 6.16455078125e-06,
      "model_forward_time": 1.823990821838379,
      "step": 80464
    },
    {
      "epoch": 6.16455078125e-06,
      "step": 80464,
      "training_step_time": 8.587179899215698
    },
    {
      "epoch": 6.165504455566406e-06,
      "model_forward_time": 1.7700135707855225,
      "step": 80465
    },
    {
      "epoch": 6.165504455566406e-06,
      "step": 80465,
      "training_step_time": 8.449286937713623
    },
    {
      "epoch": 6.166458129882813e-06,
      "model_forward_time": 1.7664718627929688,
      "step": 80466
    },
    {
      "epoch": 6.166458129882813e-06,
      "step": 80466,
      "training_step_time": 9.290110349655151
    },
    {
      "epoch": 6.167411804199218e-06,
      "model_forward_time": 2.356560468673706,
      "step": 80467
    },
    {
      "epoch": 6.167411804199218e-06,
      "step": 80467,
      "training_step_time": 9.499773502349854
    },
    {
      "epoch": 6.168365478515625e-06,
      "model_forward_time": 1.7608850002288818,
      "step": 80468
    },
    {
      "epoch": 6.168365478515625e-06,
      "step": 80468,
      "training_step_time": 8.685664176940918
    },
    {
      "epoch": 6.1693191528320315e-06,
      "model_forward_time": 1.742408275604248,
      "step": 80469
    },
    {
      "epoch": 6.1693191528320315e-06,
      "step": 80469,
      "training_step_time": 7.727508544921875
    },
    {
      "epoch": 6.170272827148437e-06,
      "grad_norm": 0.19858978688716888,
      "learning_rate": 1.0071431102809803e-06,
      "loss": 0.0573,
      "step": 80470
    },
    {
      "epoch": 6.170272827148437e-06,
      "model_forward_time": 1.7396674156188965,
      "step": 80470
    },
    {
      "dynamics_loss_avg": 0.061525576002895835,
      "epoch": 6.170272827148437e-06,
      "step": 80470
    },
    {
      "action_loss_avg": 0.001789660006761551,
      "epoch": 6.170272827148437e-06,
      "step": 80470
    },
    {
      "epoch": 6.170272827148437e-06,
      "step": 80470,
      "training_step_time": 7.54772424697876
    },
    {
      "epoch": 6.171226501464844e-06,
      "model_forward_time": 1.7703471183776855,
      "step": 80471
    },
    {
      "epoch": 6.171226501464844e-06,
      "step": 80471,
      "training_step_time": 7.653048038482666
    },
    {
      "epoch": 6.17218017578125e-06,
      "model_forward_time": 1.7733261585235596,
      "step": 80472
    },
    {
      "epoch": 6.17218017578125e-06,
      "step": 80472,
      "training_step_time": 7.541305780410767
    },
    {
      "epoch": 6.173133850097656e-06,
      "model_forward_time": 1.8067271709442139,
      "step": 80473
    },
    {
      "epoch": 6.173133850097656e-06,
      "step": 80473,
      "training_step_time": 7.6386024951934814
    },
    {
      "epoch": 6.174087524414063e-06,
      "model_forward_time": 1.7570316791534424,
      "step": 80474
    },
    {
      "epoch": 6.174087524414063e-06,
      "step": 80474,
      "training_step_time": 10.02785611152649
    },
    {
      "epoch": 6.175041198730468e-06,
      "model_forward_time": 1.7743275165557861,
      "step": 80475
    },
    {
      "epoch": 6.175041198730468e-06,
      "step": 80475,
      "training_step_time": 7.703320503234863
    },
    {
      "epoch": 6.175994873046875e-06,
      "model_forward_time": 1.7624638080596924,
      "step": 80476
    },
    {
      "epoch": 6.175994873046875e-06,
      "step": 80476,
      "training_step_time": 8.603326082229614
    },
    {
      "epoch": 6.1769485473632815e-06,
      "model_forward_time": 2.386612892150879,
      "step": 80477
    },
    {
      "epoch": 6.1769485473632815e-06,
      "step": 80477,
      "training_step_time": 9.003509521484375
    },
    {
      "epoch": 6.177902221679687e-06,
      "model_forward_time": 1.7452387809753418,
      "step": 80478
    },
    {
      "epoch": 6.177902221679687e-06,
      "step": 80478,
      "training_step_time": 7.599347114562988
    },
    {
      "epoch": 6.178855895996094e-06,
      "model_forward_time": 1.7488930225372314,
      "step": 80479
    },
    {
      "epoch": 6.178855895996094e-06,
      "step": 80479,
      "training_step_time": 7.601714372634888
    },
    {
      "epoch": 6.1798095703125e-06,
      "grad_norm": 0.15789194405078888,
      "learning_rate": 1.0061481049440297e-06,
      "loss": 0.0573,
      "step": 80480
    },
    {
      "epoch": 6.1798095703125e-06,
      "model_forward_time": 1.7494513988494873,
      "step": 80480
    },
    {
      "dynamics_loss_avg": 0.07366898022592068,
      "epoch": 6.1798095703125e-06,
      "step": 80480
    },
    {
      "action_loss_avg": 0.0013325318461284042,
      "epoch": 6.1798095703125e-06,
      "step": 80480
    },
    {
      "epoch": 6.1798095703125e-06,
      "step": 80480,
      "training_step_time": 8.399599552154541
    },
    {
      "epoch": 6.180763244628906e-06,
      "model_forward_time": 1.3875150680541992,
      "step": 80481
    },
    {
      "epoch": 6.180763244628906e-06,
      "step": 80481,
      "training_step_time": 8.753436088562012
    },
    {
      "epoch": 6.181716918945313e-06,
      "model_forward_time": 1.7530457973480225,
      "step": 80482
    },
    {
      "epoch": 6.181716918945313e-06,
      "step": 80482,
      "training_step_time": 8.071430444717407
    },
    {
      "epoch": 6.182670593261718e-06,
      "model_forward_time": 1.7626407146453857,
      "step": 80483
    },
    {
      "epoch": 6.182670593261718e-06,
      "step": 80483,
      "training_step_time": 8.490359544754028
    },
    {
      "epoch": 6.183624267578125e-06,
      "model_forward_time": 1.7662856578826904,
      "step": 80484
    },
    {
      "epoch": 6.183624267578125e-06,
      "step": 80484,
      "training_step_time": 8.259809494018555
    },
    {
      "epoch": 6.1845779418945316e-06,
      "model_forward_time": 1.7468225955963135,
      "step": 80485
    },
    {
      "epoch": 6.1845779418945316e-06,
      "step": 80485,
      "training_step_time": 8.406823635101318
    },
    {
      "epoch": 6.185531616210937e-06,
      "model_forward_time": 2.3507585525512695,
      "step": 80486
    },
    {
      "epoch": 6.185531616210937e-06,
      "step": 80486,
      "training_step_time": 8.593627452850342
    },
    {
      "epoch": 6.186485290527344e-06,
      "model_forward_time": 1.7694790363311768,
      "step": 80487
    },
    {
      "epoch": 6.186485290527344e-06,
      "step": 80487,
      "training_step_time": 7.651482582092285
    },
    {
      "epoch": 6.18743896484375e-06,
      "model_forward_time": 1.7574739456176758,
      "step": 80488
    },
    {
      "epoch": 6.18743896484375e-06,
      "step": 80488,
      "training_step_time": 7.536450624465942
    },
    {
      "epoch": 6.188392639160156e-06,
      "model_forward_time": 1.765324592590332,
      "step": 80489
    },
    {
      "epoch": 6.188392639160156e-06,
      "step": 80489,
      "training_step_time": 7.599142789840698
    },
    {
      "epoch": 6.189346313476563e-06,
      "grad_norm": 0.09070506691932678,
      "learning_rate": 1.0051535363687187e-06,
      "loss": 0.0562,
      "step": 80490
    },
    {
      "epoch": 6.189346313476563e-06,
      "model_forward_time": 1.754530668258667,
      "step": 80490
    },
    {
      "dynamics_loss_avg": 0.054472333379089834,
      "epoch": 6.189346313476563e-06,
      "step": 80490
    },
    {
      "action_loss_avg": 0.0011756007064832375,
      "epoch": 6.189346313476563e-06,
      "step": 80490
    },
    {
      "epoch": 6.189346313476563e-06,
      "step": 80490,
      "training_step_time": 7.283908128738403
    },
    {
      "epoch": 6.1902999877929684e-06,
      "model_forward_time": 1.7513034343719482,
      "step": 80491
    },
    {
      "epoch": 6.1902999877929684e-06,
      "step": 80491,
      "training_step_time": 7.547873497009277
    },
    {
      "epoch": 6.191253662109375e-06,
      "model_forward_time": 1.8839035034179688,
      "step": 80492
    },
    {
      "epoch": 6.191253662109375e-06,
      "step": 80492,
      "training_step_time": 7.645518064498901
    },
    {
      "epoch": 6.1922073364257816e-06,
      "model_forward_time": 1.7507290840148926,
      "step": 80493
    },
    {
      "epoch": 6.1922073364257816e-06,
      "step": 80493,
      "training_step_time": 7.59652853012085
    },
    {
      "epoch": 6.193161010742187e-06,
      "model_forward_time": 1.763315200805664,
      "step": 80494
    },
    {
      "epoch": 6.193161010742187e-06,
      "step": 80494,
      "training_step_time": 9.282860040664673
    },
    {
      "epoch": 6.194114685058594e-06,
      "model_forward_time": 1.9930291175842285,
      "step": 80495
    },
    {
      "epoch": 6.194114685058594e-06,
      "step": 80495,
      "training_step_time": 8.63363242149353
    },
    {
      "epoch": 6.195068359375e-06,
      "model_forward_time": 1.7687022686004639,
      "step": 80496
    },
    {
      "epoch": 6.195068359375e-06,
      "step": 80496,
      "training_step_time": 8.396148681640625
    },
    {
      "epoch": 6.196022033691406e-06,
      "model_forward_time": 1.760840654373169,
      "step": 80497
    },
    {
      "epoch": 6.196022033691406e-06,
      "step": 80497,
      "training_step_time": 7.728399276733398
    },
    {
      "epoch": 6.196975708007813e-06,
      "model_forward_time": 1.7725334167480469,
      "step": 80498
    },
    {
      "epoch": 6.196975708007813e-06,
      "step": 80498,
      "training_step_time": 7.64488410949707
    },
    {
      "epoch": 6.1979293823242184e-06,
      "model_forward_time": 1.3875813484191895,
      "step": 80499
    },
    {
      "epoch": 6.1979293823242184e-06,
      "step": 80499,
      "training_step_time": 7.367160320281982
    },
    {
      "epoch": 6.198883056640625e-06,
      "grad_norm": 0.0932876318693161,
      "learning_rate": 1.0041594046638087e-06,
      "loss": 0.0572,
      "step": 80500
    },
    {
      "epoch": 6.198883056640625e-06,
      "model_forward_time": 1.7516071796417236,
      "step": 80500
    },
    {
      "dynamics_loss_avg": 0.042321886867284775,
      "epoch": 6.198883056640625e-06,
      "step": 80500
    },
    {
      "action_loss_avg": 0.0010280140850227326,
      "epoch": 6.198883056640625e-06,
      "step": 80500
    },
    {
      "epoch": 6.198883056640625e-06,
      "step": 80500,
      "training_step_time": 7.623440504074097
    },
    {
      "epoch": 6.1998367309570316e-06,
      "model_forward_time": 1.8644533157348633,
      "step": 80501
    },
    {
      "epoch": 6.1998367309570316e-06,
      "step": 80501,
      "training_step_time": 8.765862941741943
    },
    {
      "epoch": 6.200790405273437e-06,
      "model_forward_time": 1.7674140930175781,
      "step": 80502
    },
    {
      "epoch": 6.200790405273437e-06,
      "step": 80502,
      "training_step_time": 8.676779985427856
    },
    {
      "epoch": 6.201744079589844e-06,
      "model_forward_time": 1.7600157260894775,
      "step": 80503
    },
    {
      "epoch": 6.201744079589844e-06,
      "step": 80503,
      "training_step_time": 7.751538038253784
    },
    {
      "epoch": 6.20269775390625e-06,
      "model_forward_time": 2.367788791656494,
      "step": 80504
    },
    {
      "epoch": 6.20269775390625e-06,
      "step": 80504,
      "training_step_time": 7.6195902824401855
    },
    {
      "epoch": 6.203651428222656e-06,
      "model_forward_time": 1.7587432861328125,
      "step": 80505
    },
    {
      "epoch": 6.203651428222656e-06,
      "step": 80505,
      "training_step_time": 7.680248260498047
    },
    {
      "epoch": 6.204605102539063e-06,
      "model_forward_time": 1.760948657989502,
      "step": 80506
    },
    {
      "epoch": 6.204605102539063e-06,
      "step": 80506,
      "training_step_time": 8.139988422393799
    },
    {
      "epoch": 6.2055587768554684e-06,
      "model_forward_time": 1.7594702243804932,
      "step": 80507
    },
    {
      "epoch": 6.2055587768554684e-06,
      "step": 80507,
      "training_step_time": 7.586369752883911
    },
    {
      "epoch": 6.206512451171875e-06,
      "model_forward_time": 1.7635135650634766,
      "step": 80508
    },
    {
      "epoch": 6.206512451171875e-06,
      "step": 80508,
      "training_step_time": 7.857626676559448
    },
    {
      "epoch": 6.2074661254882816e-06,
      "model_forward_time": 1.7679967880249023,
      "step": 80509
    },
    {
      "epoch": 6.2074661254882816e-06,
      "step": 80509,
      "training_step_time": 8.383221864700317
    },
    {
      "epoch": 6.208419799804687e-06,
      "grad_norm": 0.16425779461860657,
      "learning_rate": 1.0031657099380204e-06,
      "loss": 0.0568,
      "step": 80510
    },
    {
      "epoch": 6.208419799804687e-06,
      "model_forward_time": 1.8320810794830322,
      "step": 80510
    },
    {
      "dynamics_loss_avg": 0.056511367298662664,
      "epoch": 6.208419799804687e-06,
      "step": 80510
    },
    {
      "action_loss_avg": 0.0013682335673365742,
      "epoch": 6.208419799804687e-06,
      "step": 80510
    },
    {
      "epoch": 6.208419799804687e-06,
      "step": 80510,
      "training_step_time": 8.500249862670898
    },
    {
      "epoch": 6.209373474121094e-06,
      "model_forward_time": 1.7596328258514404,
      "step": 80511
    },
    {
      "epoch": 6.209373474121094e-06,
      "step": 80511,
      "training_step_time": 8.35784125328064
    },
    {
      "epoch": 6.2103271484375e-06,
      "model_forward_time": 1.7577149868011475,
      "step": 80512
    },
    {
      "epoch": 6.2103271484375e-06,
      "step": 80512,
      "training_step_time": 8.492059230804443
    },
    {
      "epoch": 6.211280822753906e-06,
      "model_forward_time": 1.751732587814331,
      "step": 80513
    },
    {
      "epoch": 6.211280822753906e-06,
      "step": 80513,
      "training_step_time": 10.31647777557373
    },
    {
      "epoch": 6.212234497070313e-06,
      "model_forward_time": 2.368527889251709,
      "step": 80514
    },
    {
      "epoch": 6.212234497070313e-06,
      "step": 80514,
      "training_step_time": 8.394125938415527
    },
    {
      "epoch": 6.2131881713867184e-06,
      "model_forward_time": 1.7645914554595947,
      "step": 80515
    },
    {
      "epoch": 6.2131881713867184e-06,
      "step": 80515,
      "training_step_time": 7.614019155502319
    },
    {
      "epoch": 6.214141845703125e-06,
      "model_forward_time": 1.7595808506011963,
      "step": 80516
    },
    {
      "epoch": 6.214141845703125e-06,
      "step": 80516,
      "training_step_time": 7.547581672668457
    },
    {
      "epoch": 6.215095520019532e-06,
      "model_forward_time": 1.7695674896240234,
      "step": 80517
    },
    {
      "epoch": 6.215095520019532e-06,
      "step": 80517,
      "training_step_time": 7.586444616317749
    },
    {
      "epoch": 6.216049194335937e-06,
      "model_forward_time": 1.7487690448760986,
      "step": 80518
    },
    {
      "epoch": 6.216049194335937e-06,
      "step": 80518,
      "training_step_time": 7.439588308334351
    },
    {
      "epoch": 6.217002868652344e-06,
      "model_forward_time": 1.8057951927185059,
      "step": 80519
    },
    {
      "epoch": 6.217002868652344e-06,
      "step": 80519,
      "training_step_time": 10.087061643600464
    },
    {
      "epoch": 6.21795654296875e-06,
      "grad_norm": 0.07134094089269638,
      "learning_rate": 1.0021724523000203e-06,
      "loss": 0.0544,
      "step": 80520
    },
    {
      "epoch": 6.21795654296875e-06,
      "model_forward_time": 1.762648105621338,
      "step": 80520
    },
    {
      "dynamics_loss_avg": 0.05079492591321468,
      "epoch": 6.21795654296875e-06,
      "step": 80520
    },
    {
      "action_loss_avg": 0.0011978818220086396,
      "epoch": 6.21795654296875e-06,
      "step": 80520
    },
    {
      "epoch": 6.21795654296875e-06,
      "step": 80520,
      "training_step_time": 9.083670616149902
    },
    {
      "epoch": 6.218910217285156e-06,
      "model_forward_time": 1.7650046348571777,
      "step": 80521
    },
    {
      "epoch": 6.218910217285156e-06,
      "step": 80521,
      "training_step_time": 7.6180195808410645
    },
    {
      "epoch": 6.219863891601563e-06,
      "model_forward_time": 1.7599108219146729,
      "step": 80522
    },
    {
      "epoch": 6.219863891601563e-06,
      "step": 80522,
      "training_step_time": 8.219791173934937
    },
    {
      "epoch": 6.2208175659179685e-06,
      "model_forward_time": 2.377267599105835,
      "step": 80523
    },
    {
      "epoch": 6.2208175659179685e-06,
      "step": 80523,
      "training_step_time": 8.472929000854492
    },
    {
      "epoch": 6.221771240234375e-06,
      "model_forward_time": 1.7593214511871338,
      "step": 80524
    },
    {
      "epoch": 6.221771240234375e-06,
      "step": 80524,
      "training_step_time": 8.668405294418335
    },
    {
      "epoch": 6.222724914550782e-06,
      "model_forward_time": 1.7764065265655518,
      "step": 80525
    },
    {
      "epoch": 6.222724914550782e-06,
      "step": 80525,
      "training_step_time": 8.187767505645752
    },
    {
      "epoch": 6.223678588867187e-06,
      "model_forward_time": 1.7460825443267822,
      "step": 80526
    },
    {
      "epoch": 6.223678588867187e-06,
      "step": 80526,
      "training_step_time": 7.535083293914795
    },
    {
      "epoch": 6.224632263183594e-06,
      "model_forward_time": 1.7606267929077148,
      "step": 80527
    },
    {
      "epoch": 6.224632263183594e-06,
      "step": 80527,
      "training_step_time": 7.4641432762146
    },
    {
      "epoch": 6.2255859375e-06,
      "model_forward_time": 1.7573180198669434,
      "step": 80528
    },
    {
      "epoch": 6.2255859375e-06,
      "step": 80528,
      "training_step_time": 9.120574474334717
    },
    {
      "epoch": 6.226539611816406e-06,
      "model_forward_time": 1.7564303874969482,
      "step": 80529
    },
    {
      "epoch": 6.226539611816406e-06,
      "step": 80529,
      "training_step_time": 9.518774509429932
    },
    {
      "epoch": 6.227493286132813e-06,
      "grad_norm": 0.0843738242983818,
      "learning_rate": 1.001179631858431e-06,
      "loss": 0.0563,
      "step": 80530
    },
    {
      "epoch": 6.227493286132813e-06,
      "model_forward_time": 1.7487661838531494,
      "step": 80530
    },
    {
      "dynamics_loss_avg": 0.056912788096815346,
      "epoch": 6.227493286132813e-06,
      "step": 80530
    },
    {
      "action_loss_avg": 0.000937473337398842,
      "epoch": 6.227493286132813e-06,
      "step": 80530
    },
    {
      "epoch": 6.227493286132813e-06,
      "step": 80530,
      "training_step_time": 8.41969347000122
    },
    {
      "epoch": 6.2284469604492185e-06,
      "model_forward_time": 1.7560737133026123,
      "step": 80531
    },
    {
      "epoch": 6.2284469604492185e-06,
      "step": 80531,
      "training_step_time": 7.709598541259766
    },
    {
      "epoch": 6.229400634765625e-06,
      "model_forward_time": 2.3894855976104736,
      "step": 80532
    },
    {
      "epoch": 6.229400634765625e-06,
      "step": 80532,
      "training_step_time": 8.142009735107422
    },
    {
      "epoch": 6.230354309082032e-06,
      "model_forward_time": 1.7638435363769531,
      "step": 80533
    },
    {
      "epoch": 6.230354309082032e-06,
      "step": 80533,
      "training_step_time": 8.372016191482544
    },
    {
      "epoch": 6.231307983398437e-06,
      "model_forward_time": 1.7650375366210938,
      "step": 80534
    },
    {
      "epoch": 6.231307983398437e-06,
      "step": 80534,
      "training_step_time": 7.562970399856567
    },
    {
      "epoch": 6.232261657714844e-06,
      "model_forward_time": 1.767927885055542,
      "step": 80535
    },
    {
      "epoch": 6.232261657714844e-06,
      "step": 80535,
      "training_step_time": 7.512024402618408
    },
    {
      "epoch": 6.23321533203125e-06,
      "model_forward_time": 1.7634961605072021,
      "step": 80536
    },
    {
      "epoch": 6.23321533203125e-06,
      "step": 80536,
      "training_step_time": 8.209816694259644
    },
    {
      "epoch": 6.234169006347656e-06,
      "model_forward_time": 1.7600610256195068,
      "step": 80537
    },
    {
      "epoch": 6.234169006347656e-06,
      "step": 80537,
      "training_step_time": 8.049921989440918
    },
    {
      "epoch": 6.235122680664063e-06,
      "model_forward_time": 1.8705964088439941,
      "step": 80538
    },
    {
      "epoch": 6.235122680664063e-06,
      "step": 80538,
      "training_step_time": 8.618505716323853
    },
    {
      "epoch": 6.2360763549804685e-06,
      "model_forward_time": 1.7585721015930176,
      "step": 80539
    },
    {
      "epoch": 6.2360763549804685e-06,
      "step": 80539,
      "training_step_time": 8.449128150939941
    },
    {
      "epoch": 6.237030029296875e-06,
      "grad_norm": 0.09220417588949203,
      "learning_rate": 1.0001872487218244e-06,
      "loss": 0.0564,
      "step": 80540
    },
    {
      "epoch": 6.237030029296875e-06,
      "model_forward_time": 1.7569818496704102,
      "step": 80540
    },
    {
      "dynamics_loss_avg": 0.04527933970093727,
      "epoch": 6.237030029296875e-06,
      "step": 80540
    },
    {
      "action_loss_avg": 0.0016465234803035855,
      "epoch": 6.237030029296875e-06,
      "step": 80540
    },
    {
      "epoch": 6.237030029296875e-06,
      "step": 80540,
      "training_step_time": 9.122958660125732
    },
    {
      "epoch": 6.237983703613282e-06,
      "model_forward_time": 2.4151811599731445,
      "step": 80541
    },
    {
      "epoch": 6.237983703613282e-06,
      "step": 80541,
      "training_step_time": 8.139870166778564
    },
    {
      "epoch": 6.238937377929687e-06,
      "model_forward_time": 1.7620320320129395,
      "step": 80542
    },
    {
      "epoch": 6.238937377929687e-06,
      "step": 80542,
      "training_step_time": 7.682149171829224
    },
    {
      "epoch": 6.239891052246094e-06,
      "model_forward_time": 1.7694284915924072,
      "step": 80543
    },
    {
      "epoch": 6.239891052246094e-06,
      "step": 80543,
      "training_step_time": 7.662998676300049
    },
    {
      "epoch": 6.2408447265625e-06,
      "model_forward_time": 1.7680675983428955,
      "step": 80544
    },
    {
      "epoch": 6.2408447265625e-06,
      "step": 80544,
      "training_step_time": 8.603663206100464
    },
    {
      "epoch": 6.241798400878906e-06,
      "model_forward_time": 1.7631118297576904,
      "step": 80545
    },
    {
      "epoch": 6.241798400878906e-06,
      "step": 80545,
      "training_step_time": 7.2939231395721436
    },
    {
      "epoch": 6.242752075195313e-06,
      "model_forward_time": 1.756570816040039,
      "step": 80546
    },
    {
      "epoch": 6.242752075195313e-06,
      "step": 80546,
      "training_step_time": 8.168299913406372
    },
    {
      "epoch": 6.2437057495117185e-06,
      "model_forward_time": 1.8450405597686768,
      "step": 80547
    },
    {
      "epoch": 6.2437057495117185e-06,
      "step": 80547,
      "training_step_time": 7.737610578536987
    },
    {
      "epoch": 6.244659423828125e-06,
      "model_forward_time": 1.7748689651489258,
      "step": 80548
    },
    {
      "epoch": 6.244659423828125e-06,
      "step": 80548,
      "training_step_time": 8.721373081207275
    },
    {
      "epoch": 6.245613098144532e-06,
      "model_forward_time": 1.7662014961242676,
      "step": 80549
    },
    {
      "epoch": 6.245613098144532e-06,
      "step": 80549,
      "training_step_time": 8.519222021102905
    },
    {
      "epoch": 6.246566772460937e-06,
      "grad_norm": 0.08898569643497467,
      "learning_rate": 9.991953029987272e-07,
      "loss": 0.0541,
      "step": 80550
    },
    {
      "epoch": 6.246566772460937e-06,
      "model_forward_time": 2.3801183700561523,
      "step": 80550
    },
    {
      "dynamics_loss_avg": 0.05768390148878098,
      "epoch": 6.246566772460937e-06,
      "step": 80550
    },
    {
      "action_loss_avg": 0.0010804256657138468,
      "epoch": 6.246566772460937e-06,
      "step": 80550
    },
    {
      "epoch": 6.246566772460937e-06,
      "step": 80550,
      "training_step_time": 10.620617628097534
    },
    {
      "epoch": 6.247520446777344e-06,
      "model_forward_time": 1.7730488777160645,
      "step": 80551
    },
    {
      "epoch": 6.247520446777344e-06,
      "step": 80551,
      "training_step_time": 7.675461530685425
    },
    {
      "epoch": 6.24847412109375e-06,
      "model_forward_time": 1.7701265811920166,
      "step": 80552
    },
    {
      "epoch": 6.24847412109375e-06,
      "step": 80552,
      "training_step_time": 7.644971132278442
    },
    {
      "epoch": 6.249427795410156e-06,
      "model_forward_time": 1.7799253463745117,
      "step": 80553
    },
    {
      "epoch": 6.249427795410156e-06,
      "step": 80553,
      "training_step_time": 7.61984658241272
    },
    {
      "epoch": 6.250381469726563e-06,
      "model_forward_time": 1.7693920135498047,
      "step": 80554
    },
    {
      "epoch": 6.250381469726563e-06,
      "step": 80554,
      "training_step_time": 7.5425121784210205
    },
    {
      "epoch": 6.2513351440429685e-06,
      "model_forward_time": 1.7795813083648682,
      "step": 80555
    },
    {
      "epoch": 6.2513351440429685e-06,
      "step": 80555,
      "training_step_time": 8.629715204238892
    },
    {
      "epoch": 6.252288818359375e-06,
      "model_forward_time": 1.4077601432800293,
      "step": 80556
    },
    {
      "epoch": 6.252288818359375e-06,
      "step": 80556,
      "training_step_time": 10.071166276931763
    },
    {
      "epoch": 6.253242492675782e-06,
      "model_forward_time": 1.7648296356201172,
      "step": 80557
    },
    {
      "epoch": 6.253242492675782e-06,
      "step": 80557,
      "training_step_time": 7.764273405075073
    },
    {
      "epoch": 6.254196166992187e-06,
      "model_forward_time": 1.7782642841339111,
      "step": 80558
    },
    {
      "epoch": 6.254196166992187e-06,
      "step": 80558,
      "training_step_time": 7.718090534210205
    },
    {
      "epoch": 6.255149841308594e-06,
      "model_forward_time": 1.7864816188812256,
      "step": 80559
    },
    {
      "epoch": 6.255149841308594e-06,
      "step": 80559,
      "training_step_time": 8.601104974746704
    },
    {
      "epoch": 6.256103515625e-06,
      "grad_norm": 0.08668650686740875,
      "learning_rate": 9.982037947976147e-07,
      "loss": 0.0552,
      "step": 80560
    },
    {
      "epoch": 6.256103515625e-06,
      "model_forward_time": 2.395193099975586,
      "step": 80560
    },
    {
      "dynamics_loss_avg": 0.04610422514379024,
      "epoch": 6.256103515625e-06,
      "step": 80560
    },
    {
      "action_loss_avg": 0.001525513594970107,
      "epoch": 6.256103515625e-06,
      "step": 80560
    },
    {
      "epoch": 6.256103515625e-06,
      "step": 80560,
      "training_step_time": 8.550631999969482
    },
    {
      "epoch": 6.257057189941406e-06,
      "model_forward_time": 1.7709481716156006,
      "step": 80561
    },
    {
      "epoch": 6.257057189941406e-06,
      "step": 80561,
      "training_step_time": 7.628057479858398
    },
    {
      "epoch": 6.258010864257813e-06,
      "model_forward_time": 1.7594780921936035,
      "step": 80562
    },
    {
      "epoch": 6.258010864257813e-06,
      "step": 80562,
      "training_step_time": 10.117698431015015
    },
    {
      "epoch": 6.2589645385742185e-06,
      "model_forward_time": 1.7754123210906982,
      "step": 80563
    },
    {
      "epoch": 6.2589645385742185e-06,
      "step": 80563,
      "training_step_time": 8.205731391906738
    },
    {
      "epoch": 6.259918212890625e-06,
      "model_forward_time": 1.7700562477111816,
      "step": 80564
    },
    {
      "epoch": 6.259918212890625e-06,
      "step": 80564,
      "training_step_time": 7.577171325683594
    },
    {
      "epoch": 6.260871887207032e-06,
      "model_forward_time": 1.7764122486114502,
      "step": 80565
    },
    {
      "epoch": 6.260871887207032e-06,
      "step": 80565,
      "training_step_time": 7.584631681442261
    },
    {
      "epoch": 6.261825561523437e-06,
      "model_forward_time": 1.7587623596191406,
      "step": 80566
    },
    {
      "epoch": 6.261825561523437e-06,
      "step": 80566,
      "training_step_time": 7.676204681396484
    },
    {
      "epoch": 6.262779235839844e-06,
      "model_forward_time": 1.7530415058135986,
      "step": 80567
    },
    {
      "epoch": 6.262779235839844e-06,
      "step": 80567,
      "training_step_time": 8.102272748947144
    },
    {
      "epoch": 6.26373291015625e-06,
      "model_forward_time": 1.777280330657959,
      "step": 80568
    },
    {
      "epoch": 6.26373291015625e-06,
      "step": 80568,
      "training_step_time": 9.526137113571167
    },
    {
      "epoch": 6.264686584472656e-06,
      "model_forward_time": 2.4649670124053955,
      "step": 80569
    },
    {
      "epoch": 6.264686584472656e-06,
      "step": 80569,
      "training_step_time": 7.642829895019531
    },
    {
      "epoch": 6.265640258789063e-06,
      "grad_norm": 0.09205569326877594,
      "learning_rate": 9.97212724226921e-07,
      "loss": 0.0557,
      "step": 80570
    },
    {
      "epoch": 6.265640258789063e-06,
      "model_forward_time": 1.9109861850738525,
      "step": 80570
    },
    {
      "dynamics_loss_avg": 0.06245055012404919,
      "epoch": 6.265640258789063e-06,
      "step": 80570
    },
    {
      "action_loss_avg": 0.0014445634093135594,
      "epoch": 6.265640258789063e-06,
      "step": 80570
    },
    {
      "epoch": 6.265640258789063e-06,
      "step": 80570,
      "training_step_time": 7.636824131011963
    },
    {
      "epoch": 6.2665939331054685e-06,
      "model_forward_time": 1.9073374271392822,
      "step": 80571
    },
    {
      "epoch": 6.2665939331054685e-06,
      "step": 80571,
      "training_step_time": 8.434762477874756
    },
    {
      "epoch": 6.267547607421875e-06,
      "model_forward_time": 1.9097263813018799,
      "step": 80572
    },
    {
      "epoch": 6.267547607421875e-06,
      "step": 80572,
      "training_step_time": 8.405347347259521
    },
    {
      "epoch": 6.268501281738282e-06,
      "model_forward_time": 1.9049139022827148,
      "step": 80573
    },
    {
      "epoch": 6.268501281738282e-06,
      "step": 80573,
      "training_step_time": 9.058853387832642
    },
    {
      "epoch": 6.269454956054687e-06,
      "model_forward_time": 1.9120399951934814,
      "step": 80574
    },
    {
      "epoch": 6.269454956054687e-06,
      "step": 80574,
      "training_step_time": 7.492674827575684
    },
    {
      "epoch": 6.270408630371094e-06,
      "model_forward_time": 1.528433084487915,
      "step": 80575
    },
    {
      "epoch": 6.270408630371094e-06,
      "step": 80575,
      "training_step_time": 7.836742162704468
    },
    {
      "epoch": 6.2713623046875e-06,
      "model_forward_time": 1.8971655368804932,
      "step": 80576
    },
    {
      "epoch": 6.2713623046875e-06,
      "step": 80576,
      "training_step_time": 8.135241508483887
    },
    {
      "epoch": 6.272315979003906e-06,
      "model_forward_time": 1.894874095916748,
      "step": 80577
    },
    {
      "epoch": 6.272315979003906e-06,
      "step": 80577,
      "training_step_time": 7.7013726234436035
    },
    {
      "epoch": 6.273269653320313e-06,
      "model_forward_time": 2.5285251140594482,
      "step": 80578
    },
    {
      "epoch": 6.273269653320313e-06,
      "step": 80578,
      "training_step_time": 8.19996976852417
    },
    {
      "epoch": 6.2742233276367185e-06,
      "model_forward_time": 1.9057352542877197,
      "step": 80579
    },
    {
      "epoch": 6.2742233276367185e-06,
      "step": 80579,
      "training_step_time": 8.550679445266724
    },
    {
      "epoch": 6.275177001953125e-06,
      "grad_norm": 0.07513415068387985,
      "learning_rate": 9.962220913950238e-07,
      "loss": 0.0563,
      "step": 80580
    },
    {
      "epoch": 6.275177001953125e-06,
      "model_forward_time": 1.913538932800293,
      "step": 80580
    },
    {
      "dynamics_loss_avg": 0.046006770618259904,
      "epoch": 6.275177001953125e-06,
      "step": 80580
    },
    {
      "action_loss_avg": 0.0012866193836089223,
      "epoch": 6.275177001953125e-06,
      "step": 80580
    },
    {
      "epoch": 6.275177001953125e-06,
      "step": 80580,
      "training_step_time": 7.589894533157349
    },
    {
      "epoch": 6.276130676269532e-06,
      "model_forward_time": 1.1657426357269287,
      "step": 80581
    },
    {
      "epoch": 6.276130676269532e-06,
      "step": 80581,
      "training_step_time": 7.559812784194946
    },
    {
      "epoch": 6.277084350585937e-06,
      "model_forward_time": 1.9068026542663574,
      "step": 80582
    },
    {
      "epoch": 6.277084350585937e-06,
      "step": 80582,
      "training_step_time": 8.660679340362549
    },
    {
      "epoch": 6.278038024902344e-06,
      "model_forward_time": 1.9122314453125,
      "step": 80583
    },
    {
      "epoch": 6.278038024902344e-06,
      "step": 80583,
      "training_step_time": 8.293198347091675
    },
    {
      "epoch": 6.27899169921875e-06,
      "model_forward_time": 1.9148120880126953,
      "step": 80584
    },
    {
      "epoch": 6.27899169921875e-06,
      "step": 80584,
      "training_step_time": 8.299423694610596
    },
    {
      "epoch": 6.279945373535156e-06,
      "model_forward_time": 1.908005714416504,
      "step": 80585
    },
    {
      "epoch": 6.279945373535156e-06,
      "step": 80585,
      "training_step_time": 7.578320741653442
    },
    {
      "epoch": 6.280899047851563e-06,
      "model_forward_time": 1.9061734676361084,
      "step": 80586
    },
    {
      "epoch": 6.280899047851563e-06,
      "step": 80586,
      "training_step_time": 7.7337965965271
    },
    {
      "epoch": 6.2818527221679685e-06,
      "model_forward_time": 1.9000213146209717,
      "step": 80587
    },
    {
      "epoch": 6.2818527221679685e-06,
      "step": 80587,
      "training_step_time": 8.658124923706055
    },
    {
      "epoch": 6.282806396484375e-06,
      "model_forward_time": 2.505833387374878,
      "step": 80588
    },
    {
      "epoch": 6.282806396484375e-06,
      "step": 80588,
      "training_step_time": 7.717416524887085
    },
    {
      "epoch": 6.283760070800782e-06,
      "model_forward_time": 1.5374040603637695,
      "step": 80589
    },
    {
      "epoch": 6.283760070800782e-06,
      "step": 80589,
      "training_step_time": 9.006572484970093
    },
    {
      "epoch": 6.284713745117187e-06,
      "grad_norm": 0.09138251841068268,
      "learning_rate": 9.952318964102592e-07,
      "loss": 0.0549,
      "step": 80590
    },
    {
      "epoch": 6.284713745117187e-06,
      "model_forward_time": 1.9052543640136719,
      "step": 80590
    },
    {
      "dynamics_loss_avg": 0.048715056665241716,
      "epoch": 6.284713745117187e-06,
      "step": 80590
    },
    {
      "action_loss_avg": 0.001085441728355363,
      "epoch": 6.284713745117187e-06,
      "step": 80590
    },
    {
      "epoch": 6.284713745117187e-06,
      "step": 80590,
      "training_step_time": 8.075353384017944
    },
    {
      "epoch": 6.285667419433594e-06,
      "model_forward_time": 1.904005527496338,
      "step": 80591
    },
    {
      "epoch": 6.285667419433594e-06,
      "step": 80591,
      "training_step_time": 8.606064319610596
    },
    {
      "epoch": 6.28662109375e-06,
      "model_forward_time": 1.9010896682739258,
      "step": 80592
    },
    {
      "epoch": 6.28662109375e-06,
      "step": 80592,
      "training_step_time": 7.717855453491211
    },
    {
      "epoch": 6.287574768066406e-06,
      "model_forward_time": 1.5185613632202148,
      "step": 80593
    },
    {
      "epoch": 6.287574768066406e-06,
      "step": 80593,
      "training_step_time": 8.319871664047241
    },
    {
      "epoch": 6.288528442382813e-06,
      "model_forward_time": 1.9068055152893066,
      "step": 80594
    },
    {
      "epoch": 6.288528442382813e-06,
      "step": 80594,
      "training_step_time": 7.644942760467529
    },
    {
      "epoch": 6.2894821166992185e-06,
      "model_forward_time": 1.917283296585083,
      "step": 80595
    },
    {
      "epoch": 6.2894821166992185e-06,
      "step": 80595,
      "training_step_time": 11.48863434791565
    },
    {
      "epoch": 6.290435791015625e-06,
      "model_forward_time": 1.8950743675231934,
      "step": 80596
    },
    {
      "epoch": 6.290435791015625e-06,
      "step": 80596,
      "training_step_time": 8.479259967803955
    },
    {
      "epoch": 6.291389465332032e-06,
      "model_forward_time": 2.5200629234313965,
      "step": 80597
    },
    {
      "epoch": 6.291389465332032e-06,
      "step": 80597,
      "training_step_time": 8.569284200668335
    },
    {
      "epoch": 6.292343139648437e-06,
      "model_forward_time": 1.8826053142547607,
      "step": 80598
    },
    {
      "epoch": 6.292343139648437e-06,
      "step": 80598,
      "training_step_time": 7.511656761169434
    },
    {
      "epoch": 6.293296813964844e-06,
      "model_forward_time": 1.8864710330963135,
      "step": 80599
    },
    {
      "epoch": 6.293296813964844e-06,
      "step": 80599,
      "training_step_time": 7.536538124084473
    },
    {
      "epoch": 6.29425048828125e-06,
      "grad_norm": 0.06444597989320755,
      "learning_rate": 9.942421393809148e-07,
      "loss": 0.055,
      "step": 80600
    },
    {
      "epoch": 6.29425048828125e-06,
      "model_forward_time": 1.897245168685913,
      "step": 80600
    },
    {
      "dynamics_loss_avg": 0.05589064955711365,
      "epoch": 6.29425048828125e-06,
      "step": 80600
    },
    {
      "action_loss_avg": 0.0011023320286767558,
      "epoch": 6.29425048828125e-06,
      "step": 80600
    },
    {
      "epoch": 6.29425048828125e-06,
      "step": 80600,
      "training_step_time": 8.202329874038696
    },
    {
      "epoch": 6.295204162597656e-06,
      "model_forward_time": 1.9154000282287598,
      "step": 80601
    },
    {
      "epoch": 6.295204162597656e-06,
      "step": 80601,
      "training_step_time": 7.6570885181427
    },
    {
      "epoch": 6.296157836914063e-06,
      "model_forward_time": 1.9345872402191162,
      "step": 80602
    },
    {
      "epoch": 6.296157836914063e-06,
      "step": 80602,
      "training_step_time": 8.541463851928711
    },
    {
      "epoch": 6.2971115112304685e-06,
      "model_forward_time": 1.8950655460357666,
      "step": 80603
    },
    {
      "epoch": 6.2971115112304685e-06,
      "step": 80603,
      "training_step_time": 7.58031439781189
    },
    {
      "epoch": 6.298065185546875e-06,
      "model_forward_time": 1.9237449169158936,
      "step": 80604
    },
    {
      "epoch": 6.298065185546875e-06,
      "step": 80604,
      "training_step_time": 7.561284065246582
    },
    {
      "epoch": 6.299018859863282e-06,
      "model_forward_time": 1.9271204471588135,
      "step": 80605
    },
    {
      "epoch": 6.299018859863282e-06,
      "step": 80605,
      "training_step_time": 7.655148506164551
    },
    {
      "epoch": 6.299972534179687e-06,
      "model_forward_time": 2.5383493900299072,
      "step": 80606
    },
    {
      "epoch": 6.299972534179687e-06,
      "step": 80606,
      "training_step_time": 8.733258247375488
    },
    {
      "epoch": 6.300926208496094e-06,
      "model_forward_time": 1.9448583126068115,
      "step": 80607
    },
    {
      "epoch": 6.300926208496094e-06,
      "step": 80607,
      "training_step_time": 7.63892126083374
    },
    {
      "epoch": 6.3018798828125e-06,
      "model_forward_time": 1.9245893955230713,
      "step": 80608
    },
    {
      "epoch": 6.3018798828125e-06,
      "step": 80608,
      "training_step_time": 7.588683605194092
    },
    {
      "epoch": 6.302833557128906e-06,
      "model_forward_time": 1.9358837604522705,
      "step": 80609
    },
    {
      "epoch": 6.302833557128906e-06,
      "step": 80609,
      "training_step_time": 7.7264769077301025
    },
    {
      "epoch": 6.303787231445313e-06,
      "grad_norm": 0.08587110042572021,
      "learning_rate": 9.932528204152264e-07,
      "loss": 0.0563,
      "step": 80610
    },
    {
      "epoch": 6.303787231445313e-06,
      "model_forward_time": 1.9292576313018799,
      "step": 80610
    },
    {
      "dynamics_loss_avg": 0.049603459797799586,
      "epoch": 6.303787231445313e-06,
      "step": 80610
    },
    {
      "action_loss_avg": 0.0010925532769761048,
      "epoch": 6.303787231445313e-06,
      "step": 80610
    },
    {
      "epoch": 6.303787231445313e-06,
      "step": 80610,
      "training_step_time": 7.629430294036865
    },
    {
      "epoch": 6.3047409057617185e-06,
      "model_forward_time": 1.94566011428833,
      "step": 80611
    },
    {
      "epoch": 6.3047409057617185e-06,
      "step": 80611,
      "training_step_time": 7.480005264282227
    },
    {
      "epoch": 6.305694580078125e-06,
      "model_forward_time": 1.9181389808654785,
      "step": 80612
    },
    {
      "epoch": 6.305694580078125e-06,
      "step": 80612,
      "training_step_time": 8.509818315505981
    },
    {
      "epoch": 6.306648254394532e-06,
      "model_forward_time": 1.9177277088165283,
      "step": 80613
    },
    {
      "epoch": 6.306648254394532e-06,
      "step": 80613,
      "training_step_time": 8.172003984451294
    },
    {
      "epoch": 6.307601928710937e-06,
      "model_forward_time": 1.9190387725830078,
      "step": 80614
    },
    {
      "epoch": 6.307601928710937e-06,
      "step": 80614,
      "training_step_time": 7.850744724273682
    },
    {
      "epoch": 6.308555603027344e-06,
      "model_forward_time": 1.9060039520263672,
      "step": 80615
    },
    {
      "epoch": 6.308555603027344e-06,
      "step": 80615,
      "training_step_time": 8.241640567779541
    },
    {
      "epoch": 6.30950927734375e-06,
      "model_forward_time": 2.534175157546997,
      "step": 80616
    },
    {
      "epoch": 6.30950927734375e-06,
      "step": 80616,
      "training_step_time": 8.792055368423462
    },
    {
      "epoch": 6.310462951660156e-06,
      "model_forward_time": 1.9262619018554688,
      "step": 80617
    },
    {
      "epoch": 6.310462951660156e-06,
      "step": 80617,
      "training_step_time": 7.691366672515869
    },
    {
      "epoch": 6.311416625976563e-06,
      "model_forward_time": 1.934812307357788,
      "step": 80618
    },
    {
      "epoch": 6.311416625976563e-06,
      "step": 80618,
      "training_step_time": 8.84539270401001
    },
    {
      "epoch": 6.3123703002929685e-06,
      "model_forward_time": 1.9202206134796143,
      "step": 80619
    },
    {
      "epoch": 6.3123703002929685e-06,
      "step": 80619,
      "training_step_time": 8.67484188079834
    },
    {
      "epoch": 6.313323974609375e-06,
      "grad_norm": 0.09468834847211838,
      "learning_rate": 9.922639396213857e-07,
      "loss": 0.0561,
      "step": 80620
    },
    {
      "epoch": 6.313323974609375e-06,
      "model_forward_time": 1.9231057167053223,
      "step": 80620
    },
    {
      "dynamics_loss_avg": 0.052476892434060576,
      "epoch": 6.313323974609375e-06,
      "step": 80620
    },
    {
      "action_loss_avg": 0.0008758970448980108,
      "epoch": 6.313323974609375e-06,
      "step": 80620
    },
    {
      "epoch": 6.313323974609375e-06,
      "step": 80620,
      "training_step_time": 8.626638650894165
    },
    {
      "epoch": 6.314277648925782e-06,
      "model_forward_time": 1.9246327877044678,
      "step": 80621
    },
    {
      "epoch": 6.314277648925782e-06,
      "step": 80621,
      "training_step_time": 8.696935892105103
    },
    {
      "epoch": 6.315231323242187e-06,
      "model_forward_time": 1.9175326824188232,
      "step": 80622
    },
    {
      "epoch": 6.315231323242187e-06,
      "step": 80622,
      "training_step_time": 7.695164442062378
    },
    {
      "epoch": 6.316184997558594e-06,
      "model_forward_time": 1.9220657348632812,
      "step": 80623
    },
    {
      "epoch": 6.316184997558594e-06,
      "step": 80623,
      "training_step_time": 8.41042685508728
    },
    {
      "epoch": 6.317138671875e-06,
      "model_forward_time": 1.5458931922912598,
      "step": 80624
    },
    {
      "epoch": 6.317138671875e-06,
      "step": 80624,
      "training_step_time": 7.771460294723511
    },
    {
      "epoch": 6.318092346191406e-06,
      "model_forward_time": 2.5336849689483643,
      "step": 80625
    },
    {
      "epoch": 6.318092346191406e-06,
      "step": 80625,
      "training_step_time": 8.415912628173828
    },
    {
      "epoch": 6.319046020507813e-06,
      "model_forward_time": 1.9268059730529785,
      "step": 80626
    },
    {
      "epoch": 6.319046020507813e-06,
      "step": 80626,
      "training_step_time": 7.678206205368042
    },
    {
      "epoch": 6.3199996948242185e-06,
      "model_forward_time": 1.9192752838134766,
      "step": 80627
    },
    {
      "epoch": 6.3199996948242185e-06,
      "step": 80627,
      "training_step_time": 8.44095516204834
    },
    {
      "epoch": 6.320953369140625e-06,
      "model_forward_time": 1.9200146198272705,
      "step": 80628
    },
    {
      "epoch": 6.320953369140625e-06,
      "step": 80628,
      "training_step_time": 8.687841892242432
    },
    {
      "epoch": 6.321907043457031e-06,
      "model_forward_time": 1.9214937686920166,
      "step": 80629
    },
    {
      "epoch": 6.321907043457031e-06,
      "step": 80629,
      "training_step_time": 8.872925996780396
    },
    {
      "epoch": 6.322860717773437e-06,
      "grad_norm": 0.1243942454457283,
      "learning_rate": 9.912754971075345e-07,
      "loss": 0.0556,
      "step": 80630
    },
    {
      "epoch": 6.322860717773437e-06,
      "model_forward_time": 1.9366974830627441,
      "step": 80630
    },
    {
      "dynamics_loss_avg": 0.05926228072494268,
      "epoch": 6.322860717773437e-06,
      "step": 80630
    },
    {
      "action_loss_avg": 0.0012241639662534,
      "epoch": 6.322860717773437e-06,
      "step": 80630
    },
    {
      "epoch": 6.322860717773437e-06,
      "step": 80630,
      "training_step_time": 9.327589988708496
    },
    {
      "epoch": 6.323814392089844e-06,
      "model_forward_time": 1.9163057804107666,
      "step": 80631
    },
    {
      "epoch": 6.323814392089844e-06,
      "step": 80631,
      "training_step_time": 7.611455678939819
    },
    {
      "epoch": 6.32476806640625e-06,
      "model_forward_time": 1.9224929809570312,
      "step": 80632
    },
    {
      "epoch": 6.32476806640625e-06,
      "step": 80632,
      "training_step_time": 7.800230979919434
    },
    {
      "epoch": 6.325721740722656e-06,
      "model_forward_time": 1.905318260192871,
      "step": 80633
    },
    {
      "epoch": 6.325721740722656e-06,
      "step": 80633,
      "training_step_time": 7.605082988739014
    },
    {
      "epoch": 6.326675415039063e-06,
      "model_forward_time": 2.5032825469970703,
      "step": 80634
    },
    {
      "epoch": 6.326675415039063e-06,
      "step": 80634,
      "training_step_time": 8.614848852157593
    },
    {
      "epoch": 6.3276290893554685e-06,
      "model_forward_time": 1.9182178974151611,
      "step": 80635
    },
    {
      "epoch": 6.3276290893554685e-06,
      "step": 80635,
      "training_step_time": 7.592718601226807
    },
    {
      "epoch": 6.328582763671875e-06,
      "model_forward_time": 1.9137940406799316,
      "step": 80636
    },
    {
      "epoch": 6.328582763671875e-06,
      "step": 80636,
      "training_step_time": 7.314455032348633
    },
    {
      "epoch": 6.329536437988281e-06,
      "model_forward_time": 1.9101715087890625,
      "step": 80637
    },
    {
      "epoch": 6.329536437988281e-06,
      "step": 80637,
      "training_step_time": 8.510705709457397
    },
    {
      "epoch": 6.330490112304687e-06,
      "model_forward_time": 1.9234466552734375,
      "step": 80638
    },
    {
      "epoch": 6.330490112304687e-06,
      "step": 80638,
      "training_step_time": 7.550275087356567
    },
    {
      "epoch": 6.331443786621094e-06,
      "model_forward_time": 1.949765920639038,
      "step": 80639
    },
    {
      "epoch": 6.331443786621094e-06,
      "step": 80639,
      "training_step_time": 8.722743034362793
    },
    {
      "epoch": 6.3323974609375e-06,
      "grad_norm": 0.09957005828619003,
      "learning_rate": 9.902874929817696e-07,
      "loss": 0.055,
      "step": 80640
    },
    {
      "epoch": 6.3323974609375e-06,
      "model_forward_time": 1.9258267879486084,
      "step": 80640
    },
    {
      "dynamics_loss_avg": 0.06186313703656197,
      "epoch": 6.3323974609375e-06,
      "step": 80640
    },
    {
      "action_loss_avg": 0.0011501880653668195,
      "epoch": 6.3323974609375e-06,
      "step": 80640
    },
    {
      "epoch": 6.3323974609375e-06,
      "step": 80640,
      "training_step_time": 9.052733898162842
    },
    {
      "epoch": 6.333351135253906e-06,
      "model_forward_time": 1.9253804683685303,
      "step": 80641
    },
    {
      "epoch": 6.333351135253906e-06,
      "step": 80641,
      "training_step_time": 7.7164483070373535
    },
    {
      "epoch": 6.334304809570313e-06,
      "model_forward_time": 1.9286589622497559,
      "step": 80642
    },
    {
      "epoch": 6.334304809570313e-06,
      "step": 80642,
      "training_step_time": 8.322804689407349
    },
    {
      "epoch": 6.3352584838867186e-06,
      "model_forward_time": 2.5782954692840576,
      "step": 80643
    },
    {
      "epoch": 6.3352584838867186e-06,
      "step": 80643,
      "training_step_time": 7.7966148853302
    },
    {
      "epoch": 6.336212158203125e-06,
      "model_forward_time": 1.8954975605010986,
      "step": 80644
    },
    {
      "epoch": 6.336212158203125e-06,
      "step": 80644,
      "training_step_time": 7.796874761581421
    },
    {
      "epoch": 6.337165832519531e-06,
      "model_forward_time": 1.9155125617980957,
      "step": 80645
    },
    {
      "epoch": 6.337165832519531e-06,
      "step": 80645,
      "training_step_time": 7.572084665298462
    },
    {
      "epoch": 6.338119506835937e-06,
      "model_forward_time": 1.8997514247894287,
      "step": 80646
    },
    {
      "epoch": 6.338119506835937e-06,
      "step": 80646,
      "training_step_time": 8.73539662361145
    },
    {
      "epoch": 6.339073181152344e-06,
      "model_forward_time": 1.910304069519043,
      "step": 80647
    },
    {
      "epoch": 6.339073181152344e-06,
      "step": 80647,
      "training_step_time": 7.674510955810547
    },
    {
      "epoch": 6.34002685546875e-06,
      "model_forward_time": 1.91302490234375,
      "step": 80648
    },
    {
      "epoch": 6.34002685546875e-06,
      "step": 80648,
      "training_step_time": 8.386438131332397
    },
    {
      "epoch": 6.340980529785156e-06,
      "model_forward_time": 1.8913013935089111,
      "step": 80649
    },
    {
      "epoch": 6.340980529785156e-06,
      "step": 80649,
      "training_step_time": 9.09726881980896
    },
    {
      "epoch": 6.341934204101563e-06,
      "grad_norm": 0.08707739412784576,
      "learning_rate": 9.892999273521343e-07,
      "loss": 0.0564,
      "step": 80650
    },
    {
      "epoch": 6.341934204101563e-06,
      "model_forward_time": 1.9064133167266846,
      "step": 80650
    },
    {
      "dynamics_loss_avg": 0.05795743828639388,
      "epoch": 6.341934204101563e-06,
      "step": 80650
    },
    {
      "action_loss_avg": 0.0008683043328346684,
      "epoch": 6.341934204101563e-06,
      "step": 80650
    },
    {
      "epoch": 6.341934204101563e-06,
      "step": 80650,
      "training_step_time": 8.36573600769043
    },
    {
      "epoch": 6.3428878784179686e-06,
      "model_forward_time": 1.8978843688964844,
      "step": 80651
    },
    {
      "epoch": 6.3428878784179686e-06,
      "step": 80651,
      "training_step_time": 7.7505717277526855
    },
    {
      "epoch": 6.343841552734375e-06,
      "model_forward_time": 2.5195250511169434,
      "step": 80652
    },
    {
      "epoch": 6.343841552734375e-06,
      "step": 80652,
      "training_step_time": 8.548915147781372
    },
    {
      "epoch": 6.344795227050781e-06,
      "model_forward_time": 1.9033408164978027,
      "step": 80653
    },
    {
      "epoch": 6.344795227050781e-06,
      "step": 80653,
      "training_step_time": 8.425948143005371
    },
    {
      "epoch": 6.345748901367187e-06,
      "model_forward_time": 1.9043200016021729,
      "step": 80654
    },
    {
      "epoch": 6.345748901367187e-06,
      "step": 80654,
      "training_step_time": 7.610609531402588
    },
    {
      "epoch": 6.346702575683594e-06,
      "model_forward_time": 1.9066815376281738,
      "step": 80655
    },
    {
      "epoch": 6.346702575683594e-06,
      "step": 80655,
      "training_step_time": 7.678586721420288
    },
    {
      "epoch": 6.34765625e-06,
      "model_forward_time": 1.9062540531158447,
      "step": 80656
    },
    {
      "epoch": 6.34765625e-06,
      "step": 80656,
      "training_step_time": 7.548868656158447
    },
    {
      "epoch": 6.348609924316406e-06,
      "model_forward_time": 1.9101958274841309,
      "step": 80657
    },
    {
      "epoch": 6.348609924316406e-06,
      "step": 80657,
      "training_step_time": 7.560748815536499
    },
    {
      "epoch": 6.349563598632813e-06,
      "model_forward_time": 1.7404584884643555,
      "step": 80658
    },
    {
      "epoch": 6.349563598632813e-06,
      "step": 80658,
      "training_step_time": 7.7659993171691895
    },
    {
      "epoch": 6.3505172729492186e-06,
      "model_forward_time": 1.3667795658111572,
      "step": 80659
    },
    {
      "epoch": 6.3505172729492186e-06,
      "step": 80659,
      "training_step_time": 8.231626272201538
    },
    {
      "epoch": 6.351470947265625e-06,
      "grad_norm": 0.12241565436124802,
      "learning_rate": 9.8831280032663e-07,
      "loss": 0.0547,
      "step": 80660
    },
    {
      "epoch": 6.351470947265625e-06,
      "model_forward_time": 1.7417314052581787,
      "step": 80660
    },
    {
      "dynamics_loss_avg": 0.06879500597715378,
      "epoch": 6.351470947265625e-06,
      "step": 80660
    },
    {
      "action_loss_avg": 0.0012115017569158226,
      "epoch": 6.351470947265625e-06,
      "step": 80660
    },
    {
      "epoch": 6.351470947265625e-06,
      "step": 80660,
      "training_step_time": 8.727050304412842
    },
    {
      "epoch": 6.352424621582031e-06,
      "model_forward_time": 1.3654391765594482,
      "step": 80661
    },
    {
      "epoch": 6.352424621582031e-06,
      "step": 80661,
      "training_step_time": 8.579199075698853
    },
    {
      "epoch": 6.3533782958984374e-06,
      "model_forward_time": 2.3336122035980225,
      "step": 80662
    },
    {
      "epoch": 6.3533782958984374e-06,
      "step": 80662,
      "training_step_time": 8.578888416290283
    },
    {
      "epoch": 6.354331970214844e-06,
      "model_forward_time": 1.7302179336547852,
      "step": 80663
    },
    {
      "epoch": 6.354331970214844e-06,
      "step": 80663,
      "training_step_time": 8.365974426269531
    },
    {
      "epoch": 6.35528564453125e-06,
      "model_forward_time": 1.7294788360595703,
      "step": 80664
    },
    {
      "epoch": 6.35528564453125e-06,
      "step": 80664,
      "training_step_time": 7.675929546356201
    },
    {
      "epoch": 6.356239318847656e-06,
      "model_forward_time": 1.7280652523040771,
      "step": 80665
    },
    {
      "epoch": 6.356239318847656e-06,
      "step": 80665,
      "training_step_time": 7.606341123580933
    },
    {
      "epoch": 6.357192993164063e-06,
      "model_forward_time": 1.738459587097168,
      "step": 80666
    },
    {
      "epoch": 6.357192993164063e-06,
      "step": 80666,
      "training_step_time": 7.617452621459961
    },
    {
      "epoch": 6.3581466674804686e-06,
      "model_forward_time": 1.7269158363342285,
      "step": 80667
    },
    {
      "epoch": 6.3581466674804686e-06,
      "step": 80667,
      "training_step_time": 7.732167959213257
    },
    {
      "epoch": 6.359100341796875e-06,
      "model_forward_time": 1.7318954467773438,
      "step": 80668
    },
    {
      "epoch": 6.359100341796875e-06,
      "step": 80668,
      "training_step_time": 7.654320478439331
    },
    {
      "epoch": 6.360054016113281e-06,
      "model_forward_time": 1.7299315929412842,
      "step": 80669
    },
    {
      "epoch": 6.360054016113281e-06,
      "step": 80669,
      "training_step_time": 7.761240482330322
    },
    {
      "epoch": 6.3610076904296874e-06,
      "grad_norm": 0.11801949888467789,
      "learning_rate": 9.87326112013206e-07,
      "loss": 0.0564,
      "step": 80670
    },
    {
      "epoch": 6.3610076904296874e-06,
      "model_forward_time": 1.7460072040557861,
      "step": 80670
    },
    {
      "dynamics_loss_avg": 0.041800304874777795,
      "epoch": 6.3610076904296874e-06,
      "step": 80670
    },
    {
      "action_loss_avg": 0.0007463482135790401,
      "epoch": 6.3610076904296874e-06,
      "step": 80670
    },
    {
      "epoch": 6.3610076904296874e-06,
      "step": 80670,
      "training_step_time": 7.739902019500732
    },
    {
      "epoch": 6.361961364746094e-06,
      "model_forward_time": 2.3868422508239746,
      "step": 80671
    },
    {
      "epoch": 6.361961364746094e-06,
      "step": 80671,
      "training_step_time": 8.5901038646698
    },
    {
      "epoch": 6.3629150390625e-06,
      "model_forward_time": 1.731785774230957,
      "step": 80672
    },
    {
      "epoch": 6.3629150390625e-06,
      "step": 80672,
      "training_step_time": 8.270916938781738
    },
    {
      "epoch": 6.363868713378906e-06,
      "model_forward_time": 1.7476608753204346,
      "step": 80673
    },
    {
      "epoch": 6.363868713378906e-06,
      "step": 80673,
      "training_step_time": 8.44276237487793
    },
    {
      "epoch": 6.364822387695313e-06,
      "model_forward_time": 1.746328353881836,
      "step": 80674
    },
    {
      "epoch": 6.364822387695313e-06,
      "step": 80674,
      "training_step_time": 7.65505313873291
    },
    {
      "epoch": 6.365776062011719e-06,
      "model_forward_time": 1.7469308376312256,
      "step": 80675
    },
    {
      "epoch": 6.365776062011719e-06,
      "step": 80675,
      "training_step_time": 9.797396421432495
    },
    {
      "epoch": 6.366729736328125e-06,
      "model_forward_time": 1.7458937168121338,
      "step": 80676
    },
    {
      "epoch": 6.366729736328125e-06,
      "step": 80676,
      "training_step_time": 7.797112464904785
    },
    {
      "epoch": 6.367683410644531e-06,
      "model_forward_time": 1.743013620376587,
      "step": 80677
    },
    {
      "epoch": 6.367683410644531e-06,
      "step": 80677,
      "training_step_time": 8.136183023452759
    },
    {
      "epoch": 6.3686370849609374e-06,
      "model_forward_time": 1.7459306716918945,
      "step": 80678
    },
    {
      "epoch": 6.3686370849609374e-06,
      "step": 80678,
      "training_step_time": 8.25917387008667
    },
    {
      "epoch": 6.369590759277344e-06,
      "model_forward_time": 1.758939266204834,
      "step": 80679
    },
    {
      "epoch": 6.369590759277344e-06,
      "step": 80679,
      "training_step_time": 9.255312442779541
    },
    {
      "epoch": 6.37054443359375e-06,
      "grad_norm": 0.10642913728952408,
      "learning_rate": 9.863398625197663e-07,
      "loss": 0.0562,
      "step": 80680
    },
    {
      "epoch": 6.37054443359375e-06,
      "model_forward_time": 2.4224822521209717,
      "step": 80680
    },
    {
      "dynamics_loss_avg": 0.05894840657711029,
      "epoch": 6.37054443359375e-06,
      "step": 80680
    },
    {
      "action_loss_avg": 0.0010439775040140375,
      "epoch": 6.37054443359375e-06,
      "step": 80680
    },
    {
      "epoch": 6.37054443359375e-06,
      "step": 80680,
      "training_step_time": 11.198519706726074
    },
    {
      "epoch": 6.371498107910156e-06,
      "model_forward_time": 1.7305099964141846,
      "step": 80681
    },
    {
      "epoch": 6.371498107910156e-06,
      "step": 80681,
      "training_step_time": 7.824695587158203
    },
    {
      "epoch": 6.372451782226563e-06,
      "model_forward_time": 1.0368525981903076,
      "step": 80682
    },
    {
      "epoch": 6.372451782226563e-06,
      "step": 80682,
      "training_step_time": 8.604676485061646
    },
    {
      "epoch": 6.373405456542969e-06,
      "model_forward_time": 1.7435922622680664,
      "step": 80683
    },
    {
      "epoch": 6.373405456542969e-06,
      "step": 80683,
      "training_step_time": 8.574201583862305
    },
    {
      "epoch": 6.374359130859375e-06,
      "model_forward_time": 1.738572597503662,
      "step": 80684
    },
    {
      "epoch": 6.374359130859375e-06,
      "step": 80684,
      "training_step_time": 7.570355176925659
    },
    {
      "epoch": 6.375312805175781e-06,
      "model_forward_time": 1.7732040882110596,
      "step": 80685
    },
    {
      "epoch": 6.375312805175781e-06,
      "step": 80685,
      "training_step_time": 8.254873991012573
    },
    {
      "epoch": 6.3762664794921874e-06,
      "model_forward_time": 1.7417998313903809,
      "step": 80686
    },
    {
      "epoch": 6.3762664794921874e-06,
      "step": 80686,
      "training_step_time": 8.433290004730225
    },
    {
      "epoch": 6.377220153808594e-06,
      "model_forward_time": 1.7420830726623535,
      "step": 80687
    },
    {
      "epoch": 6.377220153808594e-06,
      "step": 80687,
      "training_step_time": 9.522754430770874
    },
    {
      "epoch": 6.378173828125e-06,
      "model_forward_time": 1.777116298675537,
      "step": 80688
    },
    {
      "epoch": 6.378173828125e-06,
      "step": 80688,
      "training_step_time": 8.502090692520142
    },
    {
      "epoch": 6.379127502441406e-06,
      "model_forward_time": 1.7386374473571777,
      "step": 80689
    },
    {
      "epoch": 6.379127502441406e-06,
      "step": 80689,
      "training_step_time": 8.226070880889893
    },
    {
      "epoch": 6.380081176757813e-06,
      "grad_norm": 0.114024817943573,
      "learning_rate": 9.853540519541616e-07,
      "loss": 0.0568,
      "step": 80690
    },
    {
      "epoch": 6.380081176757813e-06,
      "model_forward_time": 1.624129056930542,
      "step": 80690
    },
    {
      "dynamics_loss_avg": 0.048547607660293576,
      "epoch": 6.380081176757813e-06,
      "step": 80690
    },
    {
      "action_loss_avg": 0.00153623422520468,
      "epoch": 6.380081176757813e-06,
      "step": 80690
    },
    {
      "epoch": 6.380081176757813e-06,
      "step": 80690,
      "training_step_time": 7.601660490036011
    },
    {
      "epoch": 6.381034851074219e-06,
      "model_forward_time": 1.74163818359375,
      "step": 80691
    },
    {
      "epoch": 6.381034851074219e-06,
      "step": 80691,
      "training_step_time": 7.950093984603882
    },
    {
      "epoch": 6.381988525390625e-06,
      "model_forward_time": 1.7387750148773193,
      "step": 80692
    },
    {
      "epoch": 6.381988525390625e-06,
      "step": 80692,
      "training_step_time": 7.5652008056640625
    },
    {
      "epoch": 6.382942199707031e-06,
      "model_forward_time": 1.7385673522949219,
      "step": 80693
    },
    {
      "epoch": 6.382942199707031e-06,
      "step": 80693,
      "training_step_time": 11.524340391159058
    },
    {
      "epoch": 6.3838958740234375e-06,
      "model_forward_time": 1.7384040355682373,
      "step": 80694
    },
    {
      "epoch": 6.3838958740234375e-06,
      "step": 80694,
      "training_step_time": 6.708908557891846
    },
    {
      "epoch": 6.384849548339844e-06,
      "model_forward_time": 1.7407753467559814,
      "step": 80695
    },
    {
      "epoch": 6.384849548339844e-06,
      "step": 80695,
      "training_step_time": 7.7804951667785645
    },
    {
      "epoch": 6.38580322265625e-06,
      "model_forward_time": 1.0337340831756592,
      "step": 80696
    },
    {
      "epoch": 6.38580322265625e-06,
      "step": 80696,
      "training_step_time": 8.017163515090942
    },
    {
      "epoch": 6.386756896972656e-06,
      "model_forward_time": 1.720726728439331,
      "step": 80697
    },
    {
      "epoch": 6.386756896972656e-06,
      "step": 80697,
      "training_step_time": 8.423078298568726
    },
    {
      "epoch": 6.387710571289063e-06,
      "model_forward_time": 1.7343924045562744,
      "step": 80698
    },
    {
      "epoch": 6.387710571289063e-06,
      "step": 80698,
      "training_step_time": 6.738277912139893
    },
    {
      "epoch": 6.388664245605469e-06,
      "model_forward_time": 1.7362518310546875,
      "step": 80699
    },
    {
      "epoch": 6.388664245605469e-06,
      "step": 80699,
      "training_step_time": 6.725221633911133
    },
    {
      "epoch": 6.389617919921875e-06,
      "grad_norm": 0.10028517246246338,
      "learning_rate": 9.843686804242053e-07,
      "loss": 0.0557,
      "step": 80700
    },
    {
      "epoch": 6.389617919921875e-06,
      "model_forward_time": 1.3610341548919678,
      "step": 80700
    },
    {
      "dynamics_loss_avg": 0.06822581738233566,
      "epoch": 6.389617919921875e-06,
      "step": 80700
    },
    {
      "action_loss_avg": 0.001330585213145241,
      "epoch": 6.389617919921875e-06,
      "step": 80700
    },
    {
      "epoch": 6.389617919921875e-06,
      "step": 80700,
      "training_step_time": 6.7338221073150635
    },
    {
      "epoch": 6.390571594238281e-06,
      "model_forward_time": 1.7385447025299072,
      "step": 80701
    },
    {
      "epoch": 6.390571594238281e-06,
      "step": 80701,
      "training_step_time": 6.727138519287109
    },
    {
      "epoch": 6.3915252685546875e-06,
      "model_forward_time": 1.7360749244689941,
      "step": 80702
    },
    {
      "epoch": 6.3915252685546875e-06,
      "step": 80702,
      "training_step_time": 9.89540147781372
    },
    {
      "epoch": 6.392478942871094e-06,
      "model_forward_time": 2.4051506519317627,
      "step": 80703
    },
    {
      "epoch": 6.392478942871094e-06,
      "step": 80703,
      "training_step_time": 7.688963413238525
    },
    {
      "epoch": 6.3934326171875e-06,
      "model_forward_time": 1.7417924404144287,
      "step": 80704
    },
    {
      "epoch": 6.3934326171875e-06,
      "step": 80704,
      "training_step_time": 6.748811483383179
    },
    {
      "epoch": 6.394386291503906e-06,
      "model_forward_time": 1.7347698211669922,
      "step": 80705
    },
    {
      "epoch": 6.394386291503906e-06,
      "step": 80705,
      "training_step_time": 7.319094657897949
    },
    {
      "epoch": 6.395339965820313e-06,
      "model_forward_time": 1.7380971908569336,
      "step": 80706
    },
    {
      "epoch": 6.395339965820313e-06,
      "step": 80706,
      "training_step_time": 7.412736415863037
    },
    {
      "epoch": 6.396293640136719e-06,
      "model_forward_time": 1.737217664718628,
      "step": 80707
    },
    {
      "epoch": 6.396293640136719e-06,
      "step": 80707,
      "training_step_time": 7.598496198654175
    },
    {
      "epoch": 6.397247314453125e-06,
      "model_forward_time": 1.7441906929016113,
      "step": 80708
    },
    {
      "epoch": 6.397247314453125e-06,
      "step": 80708,
      "training_step_time": 6.739759683609009
    },
    {
      "epoch": 6.398200988769531e-06,
      "model_forward_time": 1.7373015880584717,
      "step": 80709
    },
    {
      "epoch": 6.398200988769531e-06,
      "step": 80709,
      "training_step_time": 6.656956195831299
    },
    {
      "epoch": 6.3991546630859375e-06,
      "grad_norm": 0.11696737259626389,
      "learning_rate": 9.833837480376507e-07,
      "loss": 0.058,
      "step": 80710
    },
    {
      "epoch": 6.3991546630859375e-06,
      "model_forward_time": 1.743163824081421,
      "step": 80710
    },
    {
      "dynamics_loss_avg": 0.052712715417146686,
      "epoch": 6.3991546630859375e-06,
      "step": 80710
    },
    {
      "action_loss_avg": 0.0012912575126392766,
      "epoch": 6.3991546630859375e-06,
      "step": 80710
    },
    {
      "epoch": 6.3991546630859375e-06,
      "step": 80710,
      "training_step_time": 6.629260540008545
    },
    {
      "epoch": 6.400108337402344e-06,
      "model_forward_time": 1.7378950119018555,
      "step": 80711
    },
    {
      "epoch": 6.400108337402344e-06,
      "step": 80711,
      "training_step_time": 7.684881687164307
    },
    {
      "epoch": 6.40106201171875e-06,
      "model_forward_time": 2.411463499069214,
      "step": 80712
    },
    {
      "epoch": 6.40106201171875e-06,
      "step": 80712,
      "training_step_time": 8.336632490158081
    },
    {
      "epoch": 6.402015686035156e-06,
      "model_forward_time": 1.757657766342163,
      "step": 80713
    },
    {
      "epoch": 6.402015686035156e-06,
      "step": 80713,
      "training_step_time": 6.664008617401123
    },
    {
      "epoch": 6.402969360351563e-06,
      "model_forward_time": 1.7498767375946045,
      "step": 80714
    },
    {
      "epoch": 6.402969360351563e-06,
      "step": 80714,
      "training_step_time": 7.31661319732666
    },
    {
      "epoch": 6.403923034667969e-06,
      "model_forward_time": 1.7512760162353516,
      "step": 80715
    },
    {
      "epoch": 6.403923034667969e-06,
      "step": 80715,
      "training_step_time": 7.612144947052002
    },
    {
      "epoch": 6.404876708984375e-06,
      "model_forward_time": 1.7540483474731445,
      "step": 80716
    },
    {
      "epoch": 6.404876708984375e-06,
      "step": 80716,
      "training_step_time": 7.627979755401611
    },
    {
      "epoch": 6.405830383300781e-06,
      "model_forward_time": 1.7572340965270996,
      "step": 80717
    },
    {
      "epoch": 6.405830383300781e-06,
      "step": 80717,
      "training_step_time": 6.609894752502441
    },
    {
      "epoch": 6.4067840576171875e-06,
      "model_forward_time": 1.7678353786468506,
      "step": 80718
    },
    {
      "epoch": 6.4067840576171875e-06,
      "step": 80718,
      "training_step_time": 6.627079486846924
    },
    {
      "epoch": 6.407737731933594e-06,
      "model_forward_time": 1.7576870918273926,
      "step": 80719
    },
    {
      "epoch": 6.407737731933594e-06,
      "step": 80719,
      "training_step_time": 6.653987646102905
    },
    {
      "epoch": 6.40869140625e-06,
      "grad_norm": 0.08648121356964111,
      "learning_rate": 9.82399254902212e-07,
      "loss": 0.0571,
      "step": 80720
    },
    {
      "epoch": 6.40869140625e-06,
      "model_forward_time": 1.747779369354248,
      "step": 80720
    },
    {
      "dynamics_loss_avg": 0.05650492925196886,
      "epoch": 6.40869140625e-06,
      "step": 80720
    },
    {
      "action_loss_avg": 0.0010939889878500253,
      "epoch": 6.40869140625e-06,
      "step": 80720
    },
    {
      "epoch": 6.40869140625e-06,
      "step": 80720,
      "training_step_time": 7.674258708953857
    },
    {
      "epoch": 6.409645080566406e-06,
      "model_forward_time": 2.402043581008911,
      "step": 80721
    },
    {
      "epoch": 6.409645080566406e-06,
      "step": 80721,
      "training_step_time": 7.702884197235107
    },
    {
      "epoch": 6.410598754882813e-06,
      "model_forward_time": 1.7557947635650635,
      "step": 80722
    },
    {
      "epoch": 6.410598754882813e-06,
      "step": 80722,
      "training_step_time": 7.234186172485352
    },
    {
      "epoch": 6.411552429199219e-06,
      "model_forward_time": 1.7601380348205566,
      "step": 80723
    },
    {
      "epoch": 6.411552429199219e-06,
      "step": 80723,
      "training_step_time": 7.394177436828613
    },
    {
      "epoch": 6.412506103515625e-06,
      "model_forward_time": 1.7570598125457764,
      "step": 80724
    },
    {
      "epoch": 6.412506103515625e-06,
      "step": 80724,
      "training_step_time": 7.316756010055542
    },
    {
      "epoch": 6.413459777832031e-06,
      "model_forward_time": 1.7584292888641357,
      "step": 80725
    },
    {
      "epoch": 6.413459777832031e-06,
      "step": 80725,
      "training_step_time": 7.508015871047974
    },
    {
      "epoch": 6.4144134521484375e-06,
      "model_forward_time": 1.7614092826843262,
      "step": 80726
    },
    {
      "epoch": 6.4144134521484375e-06,
      "step": 80726,
      "training_step_time": 7.049437999725342
    },
    {
      "epoch": 6.415367126464844e-06,
      "model_forward_time": 1.762103796005249,
      "step": 80727
    },
    {
      "epoch": 6.415367126464844e-06,
      "step": 80727,
      "training_step_time": 6.610926866531372
    },
    {
      "epoch": 6.41632080078125e-06,
      "model_forward_time": 1.757833480834961,
      "step": 80728
    },
    {
      "epoch": 6.41632080078125e-06,
      "step": 80728,
      "training_step_time": 6.700503587722778
    },
    {
      "epoch": 6.417274475097656e-06,
      "model_forward_time": 1.762610673904419,
      "step": 80729
    },
    {
      "epoch": 6.417274475097656e-06,
      "step": 80729,
      "training_step_time": 6.646674156188965
    },
    {
      "epoch": 6.418228149414063e-06,
      "grad_norm": 0.0916815772652626,
      "learning_rate": 9.814152011255496e-07,
      "loss": 0.0567,
      "step": 80730
    },
    {
      "epoch": 6.418228149414063e-06,
      "model_forward_time": 1.844310998916626,
      "step": 80730
    },
    {
      "dynamics_loss_avg": 0.0593757813796401,
      "epoch": 6.418228149414063e-06,
      "step": 80730
    },
    {
      "action_loss_avg": 0.0010300674592144787,
      "epoch": 6.418228149414063e-06,
      "step": 80730
    },
    {
      "epoch": 6.418228149414063e-06,
      "step": 80730,
      "training_step_time": 8.419053554534912
    },
    {
      "epoch": 6.419181823730469e-06,
      "model_forward_time": 2.392855167388916,
      "step": 80731
    },
    {
      "epoch": 6.419181823730469e-06,
      "step": 80731,
      "training_step_time": 7.590981483459473
    },
    {
      "epoch": 6.420135498046875e-06,
      "model_forward_time": 1.7605068683624268,
      "step": 80732
    },
    {
      "epoch": 6.420135498046875e-06,
      "step": 80732,
      "training_step_time": 7.109093189239502
    },
    {
      "epoch": 6.421089172363281e-06,
      "model_forward_time": 1.7628049850463867,
      "step": 80733
    },
    {
      "epoch": 6.421089172363281e-06,
      "step": 80733,
      "training_step_time": 7.196303844451904
    },
    {
      "epoch": 6.4220428466796875e-06,
      "model_forward_time": 1.7574577331542969,
      "step": 80734
    },
    {
      "epoch": 6.4220428466796875e-06,
      "step": 80734,
      "training_step_time": 7.553930044174194
    },
    {
      "epoch": 6.422996520996094e-06,
      "model_forward_time": 1.7613260746002197,
      "step": 80735
    },
    {
      "epoch": 6.422996520996094e-06,
      "step": 80735,
      "training_step_time": 7.086172580718994
    },
    {
      "epoch": 6.4239501953125e-06,
      "model_forward_time": 1.7695190906524658,
      "step": 80736
    },
    {
      "epoch": 6.4239501953125e-06,
      "step": 80736,
      "training_step_time": 6.633693695068359
    },
    {
      "epoch": 6.424903869628906e-06,
      "model_forward_time": 1.765442132949829,
      "step": 80737
    },
    {
      "epoch": 6.424903869628906e-06,
      "step": 80737,
      "training_step_time": 6.613525629043579
    },
    {
      "epoch": 6.425857543945313e-06,
      "model_forward_time": 1.761704683303833,
      "step": 80738
    },
    {
      "epoch": 6.425857543945313e-06,
      "step": 80738,
      "training_step_time": 6.603121280670166
    },
    {
      "epoch": 6.426811218261719e-06,
      "model_forward_time": 1.8569371700286865,
      "step": 80739
    },
    {
      "epoch": 6.426811218261719e-06,
      "step": 80739,
      "training_step_time": 10.592996597290039
    },
    {
      "epoch": 6.427764892578125e-06,
      "grad_norm": 0.10590144246816635,
      "learning_rate": 9.804315868152798e-07,
      "loss": 0.0576,
      "step": 80740
    },
    {
      "epoch": 6.427764892578125e-06,
      "model_forward_time": 2.383962869644165,
      "step": 80740
    },
    {
      "dynamics_loss_avg": 0.06797017604112625,
      "epoch": 6.427764892578125e-06,
      "step": 80740
    },
    {
      "action_loss_avg": 0.0019249916076660156,
      "epoch": 6.427764892578125e-06,
      "step": 80740
    },
    {
      "epoch": 6.427764892578125e-06,
      "step": 80740,
      "training_step_time": 7.678386688232422
    },
    {
      "epoch": 6.428718566894531e-06,
      "model_forward_time": 1.770493745803833,
      "step": 80741
    },
    {
      "epoch": 6.428718566894531e-06,
      "step": 80741,
      "training_step_time": 7.3945090770721436
    },
    {
      "epoch": 6.4296722412109375e-06,
      "model_forward_time": 1.7657086849212646,
      "step": 80742
    },
    {
      "epoch": 6.4296722412109375e-06,
      "step": 80742,
      "training_step_time": 7.247785568237305
    },
    {
      "epoch": 6.430625915527344e-06,
      "model_forward_time": 1.7696523666381836,
      "step": 80743
    },
    {
      "epoch": 6.430625915527344e-06,
      "step": 80743,
      "training_step_time": 7.598219871520996
    },
    {
      "epoch": 6.43157958984375e-06,
      "model_forward_time": 1.7699174880981445,
      "step": 80744
    },
    {
      "epoch": 6.43157958984375e-06,
      "step": 80744,
      "training_step_time": 7.153881072998047
    },
    {
      "epoch": 6.432533264160156e-06,
      "model_forward_time": 1.7753996849060059,
      "step": 80745
    },
    {
      "epoch": 6.432533264160156e-06,
      "step": 80745,
      "training_step_time": 6.6787192821502686
    },
    {
      "epoch": 6.433486938476563e-06,
      "model_forward_time": 1.7508282661437988,
      "step": 80746
    },
    {
      "epoch": 6.433486938476563e-06,
      "step": 80746,
      "training_step_time": 6.726199150085449
    },
    {
      "epoch": 6.434440612792969e-06,
      "model_forward_time": 1.7730753421783447,
      "step": 80747
    },
    {
      "epoch": 6.434440612792969e-06,
      "step": 80747,
      "training_step_time": 6.753896713256836
    },
    {
      "epoch": 6.435394287109375e-06,
      "model_forward_time": 1.7974116802215576,
      "step": 80748
    },
    {
      "epoch": 6.435394287109375e-06,
      "step": 80748,
      "training_step_time": 11.384954690933228
    },
    {
      "epoch": 6.436347961425781e-06,
      "model_forward_time": 2.416045904159546,
      "step": 80749
    },
    {
      "epoch": 6.436347961425781e-06,
      "step": 80749,
      "training_step_time": 9.472349882125854
    },
    {
      "epoch": 6.4373016357421875e-06,
      "grad_norm": 0.10047784447669983,
      "learning_rate": 9.79448412078966e-07,
      "loss": 0.0583,
      "step": 80750
    },
    {
      "epoch": 6.4373016357421875e-06,
      "model_forward_time": 1.7636044025421143,
      "step": 80750
    },
    {
      "dynamics_loss_avg": 0.0675439864397049,
      "epoch": 6.4373016357421875e-06,
      "step": 80750
    },
    {
      "action_loss_avg": 0.0012669525865931063,
      "epoch": 6.4373016357421875e-06,
      "step": 80750
    },
    {
      "epoch": 6.4373016357421875e-06,
      "step": 80750,
      "training_step_time": 7.689709663391113
    },
    {
      "epoch": 6.438255310058594e-06,
      "model_forward_time": 1.7731411457061768,
      "step": 80751
    },
    {
      "epoch": 6.438255310058594e-06,
      "step": 80751,
      "training_step_time": 7.274943828582764
    },
    {
      "epoch": 6.439208984375e-06,
      "model_forward_time": 1.7756187915802002,
      "step": 80752
    },
    {
      "epoch": 6.439208984375e-06,
      "step": 80752,
      "training_step_time": 8.457768678665161
    },
    {
      "epoch": 6.440162658691406e-06,
      "model_forward_time": 1.7723848819732666,
      "step": 80753
    },
    {
      "epoch": 6.440162658691406e-06,
      "step": 80753,
      "training_step_time": 7.602069616317749
    },
    {
      "epoch": 6.441116333007812e-06,
      "model_forward_time": 1.7778127193450928,
      "step": 80754
    },
    {
      "epoch": 6.441116333007812e-06,
      "step": 80754,
      "training_step_time": 7.120814800262451
    },
    {
      "epoch": 6.442070007324219e-06,
      "model_forward_time": 1.7691009044647217,
      "step": 80755
    },
    {
      "epoch": 6.442070007324219e-06,
      "step": 80755,
      "training_step_time": 6.720734119415283
    },
    {
      "epoch": 6.443023681640625e-06,
      "model_forward_time": 1.7505161762237549,
      "step": 80756
    },
    {
      "epoch": 6.443023681640625e-06,
      "step": 80756,
      "training_step_time": 6.712390661239624
    },
    {
      "epoch": 6.443977355957031e-06,
      "model_forward_time": 1.7704074382781982,
      "step": 80757
    },
    {
      "epoch": 6.443977355957031e-06,
      "step": 80757,
      "training_step_time": 10.331032752990723
    },
    {
      "epoch": 6.4449310302734375e-06,
      "model_forward_time": 2.4155333042144775,
      "step": 80758
    },
    {
      "epoch": 6.4449310302734375e-06,
      "step": 80758,
      "training_step_time": 7.787023544311523
    },
    {
      "epoch": 6.445884704589844e-06,
      "model_forward_time": 1.7829289436340332,
      "step": 80759
    },
    {
      "epoch": 6.445884704589844e-06,
      "step": 80759,
      "training_step_time": 7.6953864097595215
    },
    {
      "epoch": 6.44683837890625e-06,
      "grad_norm": 0.12716814875602722,
      "learning_rate": 9.784656770241313e-07,
      "loss": 0.0572,
      "step": 80760
    },
    {
      "epoch": 6.44683837890625e-06,
      "model_forward_time": 1.752526044845581,
      "step": 80760
    },
    {
      "dynamics_loss_avg": 0.05461692102253437,
      "epoch": 6.44683837890625e-06,
      "step": 80760
    },
    {
      "action_loss_avg": 0.0008717053540749475,
      "epoch": 6.44683837890625e-06,
      "step": 80760
    },
    {
      "epoch": 6.44683837890625e-06,
      "step": 80760,
      "training_step_time": 7.279064178466797
    },
    {
      "epoch": 6.447792053222656e-06,
      "model_forward_time": 1.7545013427734375,
      "step": 80761
    },
    {
      "epoch": 6.447792053222656e-06,
      "step": 80761,
      "training_step_time": 7.6138811111450195
    },
    {
      "epoch": 6.448745727539062e-06,
      "model_forward_time": 1.3921794891357422,
      "step": 80762
    },
    {
      "epoch": 6.448745727539062e-06,
      "step": 80762,
      "training_step_time": 7.5963969230651855
    },
    {
      "epoch": 6.449699401855469e-06,
      "model_forward_time": 1.7795531749725342,
      "step": 80763
    },
    {
      "epoch": 6.449699401855469e-06,
      "step": 80763,
      "training_step_time": 7.156573534011841
    },
    {
      "epoch": 6.450653076171875e-06,
      "model_forward_time": 1.7725191116333008,
      "step": 80764
    },
    {
      "epoch": 6.450653076171875e-06,
      "step": 80764,
      "training_step_time": 6.736239433288574
    },
    {
      "epoch": 6.451606750488281e-06,
      "model_forward_time": 1.7693023681640625,
      "step": 80765
    },
    {
      "epoch": 6.451606750488281e-06,
      "step": 80765,
      "training_step_time": 6.700451374053955
    },
    {
      "epoch": 6.4525604248046875e-06,
      "model_forward_time": 1.7664825916290283,
      "step": 80766
    },
    {
      "epoch": 6.4525604248046875e-06,
      "step": 80766,
      "training_step_time": 9.417206525802612
    },
    {
      "epoch": 6.453514099121094e-06,
      "model_forward_time": 2.4150280952453613,
      "step": 80767
    },
    {
      "epoch": 6.453514099121094e-06,
      "step": 80767,
      "training_step_time": 8.478774070739746
    },
    {
      "epoch": 6.4544677734375e-06,
      "model_forward_time": 1.7680079936981201,
      "step": 80768
    },
    {
      "epoch": 6.4544677734375e-06,
      "step": 80768,
      "training_step_time": 7.652425765991211
    },
    {
      "epoch": 6.455421447753906e-06,
      "model_forward_time": 1.7670834064483643,
      "step": 80769
    },
    {
      "epoch": 6.455421447753906e-06,
      "step": 80769,
      "training_step_time": 7.362834692001343
    },
    {
      "epoch": 6.456375122070312e-06,
      "grad_norm": 0.10238327085971832,
      "learning_rate": 9.77483381758243e-07,
      "loss": 0.057,
      "step": 80770
    },
    {
      "epoch": 6.456375122070312e-06,
      "model_forward_time": 1.755312442779541,
      "step": 80770
    },
    {
      "dynamics_loss_avg": 0.05444038212299347,
      "epoch": 6.456375122070312e-06,
      "step": 80770
    },
    {
      "action_loss_avg": 0.0010357659630244598,
      "epoch": 6.456375122070312e-06,
      "step": 80770
    },
    {
      "epoch": 6.456375122070312e-06,
      "step": 80770,
      "training_step_time": 7.556816101074219
    },
    {
      "epoch": 6.457328796386719e-06,
      "model_forward_time": 1.7665979862213135,
      "step": 80771
    },
    {
      "epoch": 6.457328796386719e-06,
      "step": 80771,
      "training_step_time": 7.611165285110474
    },
    {
      "epoch": 6.458282470703125e-06,
      "model_forward_time": 1.771658182144165,
      "step": 80772
    },
    {
      "epoch": 6.458282470703125e-06,
      "step": 80772,
      "training_step_time": 7.113530874252319
    },
    {
      "epoch": 6.459236145019531e-06,
      "model_forward_time": 1.764535903930664,
      "step": 80773
    },
    {
      "epoch": 6.459236145019531e-06,
      "step": 80773,
      "training_step_time": 6.654372453689575
    },
    {
      "epoch": 6.4601898193359375e-06,
      "model_forward_time": 1.7698299884796143,
      "step": 80774
    },
    {
      "epoch": 6.4601898193359375e-06,
      "step": 80774,
      "training_step_time": 6.630198240280151
    },
    {
      "epoch": 6.461143493652344e-06,
      "model_forward_time": 1.767935037612915,
      "step": 80775
    },
    {
      "epoch": 6.461143493652344e-06,
      "step": 80775,
      "training_step_time": 8.123392581939697
    },
    {
      "epoch": 6.46209716796875e-06,
      "model_forward_time": 1.8265316486358643,
      "step": 80776
    },
    {
      "epoch": 6.46209716796875e-06,
      "step": 80776,
      "training_step_time": 9.192957162857056
    },
    {
      "epoch": 6.463050842285156e-06,
      "model_forward_time": 2.374159097671509,
      "step": 80777
    },
    {
      "epoch": 6.463050842285156e-06,
      "step": 80777,
      "training_step_time": 7.6302330493927
    },
    {
      "epoch": 6.464004516601562e-06,
      "model_forward_time": 1.747882604598999,
      "step": 80778
    },
    {
      "epoch": 6.464004516601562e-06,
      "step": 80778,
      "training_step_time": 8.549481630325317
    },
    {
      "epoch": 6.464958190917969e-06,
      "model_forward_time": 1.7687413692474365,
      "step": 80779
    },
    {
      "epoch": 6.464958190917969e-06,
      "step": 80779,
      "training_step_time": 7.551563501358032
    },
    {
      "epoch": 6.465911865234375e-06,
      "grad_norm": 0.08590564131736755,
      "learning_rate": 9.765015263887251e-07,
      "loss": 0.0565,
      "step": 80780
    },
    {
      "epoch": 6.465911865234375e-06,
      "model_forward_time": 1.7591493129730225,
      "step": 80780
    },
    {
      "dynamics_loss_avg": 0.0680348202586174,
      "epoch": 6.465911865234375e-06,
      "step": 80780
    },
    {
      "action_loss_avg": 0.001978789665736258,
      "epoch": 6.465911865234375e-06,
      "step": 80780
    },
    {
      "epoch": 6.465911865234375e-06,
      "step": 80780,
      "training_step_time": 7.551100015640259
    },
    {
      "epoch": 6.466865539550781e-06,
      "model_forward_time": 1.7624568939208984,
      "step": 80781
    },
    {
      "epoch": 6.466865539550781e-06,
      "step": 80781,
      "training_step_time": 7.568018198013306
    },
    {
      "epoch": 6.4678192138671875e-06,
      "model_forward_time": 1.7631523609161377,
      "step": 80782
    },
    {
      "epoch": 6.4678192138671875e-06,
      "step": 80782,
      "training_step_time": 7.11389684677124
    },
    {
      "epoch": 6.468772888183594e-06,
      "model_forward_time": 1.3855667114257812,
      "step": 80783
    },
    {
      "epoch": 6.468772888183594e-06,
      "step": 80783,
      "training_step_time": 6.626051664352417
    },
    {
      "epoch": 6.4697265625e-06,
      "model_forward_time": 1.7664663791656494,
      "step": 80784
    },
    {
      "epoch": 6.4697265625e-06,
      "step": 80784,
      "training_step_time": 7.491868257522583
    },
    {
      "epoch": 6.470680236816406e-06,
      "model_forward_time": 1.3899803161621094,
      "step": 80785
    },
    {
      "epoch": 6.470680236816406e-06,
      "step": 80785,
      "training_step_time": 9.75434160232544
    },
    {
      "epoch": 6.471633911132812e-06,
      "model_forward_time": 2.414729595184326,
      "step": 80786
    },
    {
      "epoch": 6.471633911132812e-06,
      "step": 80786,
      "training_step_time": 7.676792621612549
    },
    {
      "epoch": 6.472587585449219e-06,
      "model_forward_time": 1.766124963760376,
      "step": 80787
    },
    {
      "epoch": 6.472587585449219e-06,
      "step": 80787,
      "training_step_time": 7.599714517593384
    },
    {
      "epoch": 6.473541259765625e-06,
      "model_forward_time": 1.7711679935455322,
      "step": 80788
    },
    {
      "epoch": 6.473541259765625e-06,
      "step": 80788,
      "training_step_time": 7.506103277206421
    },
    {
      "epoch": 6.474494934082031e-06,
      "model_forward_time": 1.7756452560424805,
      "step": 80789
    },
    {
      "epoch": 6.474494934082031e-06,
      "step": 80789,
      "training_step_time": 7.629349946975708
    },
    {
      "epoch": 6.4754486083984375e-06,
      "grad_norm": 0.08177026361227036,
      "learning_rate": 9.755201110229502e-07,
      "loss": 0.0563,
      "step": 80790
    },
    {
      "epoch": 6.4754486083984375e-06,
      "model_forward_time": 1.3739991188049316,
      "step": 80790
    },
    {
      "dynamics_loss_avg": 0.05332117415964603,
      "epoch": 6.4754486083984375e-06,
      "step": 80790
    },
    {
      "action_loss_avg": 0.000948203397274483,
      "epoch": 6.4754486083984375e-06,
      "step": 80790
    },
    {
      "epoch": 6.4754486083984375e-06,
      "step": 80790,
      "training_step_time": 7.539984226226807
    },
    {
      "epoch": 6.476402282714844e-06,
      "model_forward_time": 1.7714929580688477,
      "step": 80791
    },
    {
      "epoch": 6.476402282714844e-06,
      "step": 80791,
      "training_step_time": 7.097180604934692
    },
    {
      "epoch": 6.47735595703125e-06,
      "model_forward_time": 1.7680857181549072,
      "step": 80792
    },
    {
      "epoch": 6.47735595703125e-06,
      "step": 80792,
      "training_step_time": 6.642239332199097
    },
    {
      "epoch": 6.478309631347656e-06,
      "model_forward_time": 1.7573580741882324,
      "step": 80793
    },
    {
      "epoch": 6.478309631347656e-06,
      "step": 80793,
      "training_step_time": 7.556382656097412
    },
    {
      "epoch": 6.479263305664062e-06,
      "model_forward_time": 1.779853343963623,
      "step": 80794
    },
    {
      "epoch": 6.479263305664062e-06,
      "step": 80794,
      "training_step_time": 10.19538140296936
    },
    {
      "epoch": 6.480216979980469e-06,
      "model_forward_time": 2.3375940322875977,
      "step": 80795
    },
    {
      "epoch": 6.480216979980469e-06,
      "step": 80795,
      "training_step_time": 7.68774151802063
    },
    {
      "epoch": 6.481170654296875e-06,
      "model_forward_time": 1.7755751609802246,
      "step": 80796
    },
    {
      "epoch": 6.481170654296875e-06,
      "step": 80796,
      "training_step_time": 7.662698984146118
    },
    {
      "epoch": 6.482124328613281e-06,
      "model_forward_time": 1.7703089714050293,
      "step": 80797
    },
    {
      "epoch": 6.482124328613281e-06,
      "step": 80797,
      "training_step_time": 7.208792209625244
    },
    {
      "epoch": 6.4830780029296875e-06,
      "model_forward_time": 1.7662737369537354,
      "step": 80798
    },
    {
      "epoch": 6.4830780029296875e-06,
      "step": 80798,
      "training_step_time": 7.643643140792847
    },
    {
      "epoch": 6.484031677246094e-06,
      "model_forward_time": 1.7689464092254639,
      "step": 80799
    },
    {
      "epoch": 6.484031677246094e-06,
      "step": 80799,
      "training_step_time": 7.5709733963012695
    },
    {
      "epoch": 6.4849853515625e-06,
      "grad_norm": 0.19075927138328552,
      "learning_rate": 9.745391357682448e-07,
      "loss": 0.0558,
      "step": 80800
    },
    {
      "epoch": 6.4849853515625e-06,
      "model_forward_time": 1.7241697311401367,
      "step": 80800
    },
    {
      "dynamics_loss_avg": 0.04663215894252062,
      "epoch": 6.4849853515625e-06,
      "step": 80800
    },
    {
      "action_loss_avg": 0.0008622382738394663,
      "epoch": 6.4849853515625e-06,
      "step": 80800
    },
    {
      "epoch": 6.4849853515625e-06,
      "step": 80800,
      "training_step_time": 7.0960187911987305
    },
    {
      "epoch": 6.485939025878906e-06,
      "model_forward_time": 1.7440927028656006,
      "step": 80801
    },
    {
      "epoch": 6.485939025878906e-06,
      "step": 80801,
      "training_step_time": 6.635693311691284
    },
    {
      "epoch": 6.486892700195312e-06,
      "model_forward_time": 1.741548776626587,
      "step": 80802
    },
    {
      "epoch": 6.486892700195312e-06,
      "step": 80802,
      "training_step_time": 7.530025243759155
    },
    {
      "epoch": 6.487846374511719e-06,
      "model_forward_time": 1.7344787120819092,
      "step": 80803
    },
    {
      "epoch": 6.487846374511719e-06,
      "step": 80803,
      "training_step_time": 9.345237731933594
    },
    {
      "epoch": 6.488800048828125e-06,
      "model_forward_time": 1.7489831447601318,
      "step": 80804
    },
    {
      "epoch": 6.488800048828125e-06,
      "step": 80804,
      "training_step_time": 9.481937408447266
    },
    {
      "epoch": 6.489753723144531e-06,
      "model_forward_time": 2.0082623958587646,
      "step": 80805
    },
    {
      "epoch": 6.489753723144531e-06,
      "step": 80805,
      "training_step_time": 8.513341665267944
    },
    {
      "epoch": 6.4907073974609375e-06,
      "model_forward_time": 1.7416605949401855,
      "step": 80806
    },
    {
      "epoch": 6.4907073974609375e-06,
      "step": 80806,
      "training_step_time": 7.6493752002716064
    },
    {
      "epoch": 6.491661071777344e-06,
      "model_forward_time": 1.739619493484497,
      "step": 80807
    },
    {
      "epoch": 6.491661071777344e-06,
      "step": 80807,
      "training_step_time": 8.594687461853027
    },
    {
      "epoch": 6.49261474609375e-06,
      "model_forward_time": 1.7459335327148438,
      "step": 80808
    },
    {
      "epoch": 6.49261474609375e-06,
      "step": 80808,
      "training_step_time": 7.430043935775757
    },
    {
      "epoch": 6.493568420410156e-06,
      "model_forward_time": 1.739133596420288,
      "step": 80809
    },
    {
      "epoch": 6.493568420410156e-06,
      "step": 80809,
      "training_step_time": 7.651609659194946
    },
    {
      "epoch": 6.494522094726562e-06,
      "grad_norm": 0.1168706864118576,
      "learning_rate": 9.735586007318875e-07,
      "loss": 0.0554,
      "step": 80810
    },
    {
      "epoch": 6.494522094726562e-06,
      "model_forward_time": 1.7354326248168945,
      "step": 80810
    },
    {
      "dynamics_loss_avg": 0.04524890119209886,
      "epoch": 6.494522094726562e-06,
      "step": 80810
    },
    {
      "action_loss_avg": 0.0014867810561554507,
      "epoch": 6.494522094726562e-06,
      "step": 80810
    },
    {
      "epoch": 6.494522094726562e-06,
      "step": 80810,
      "training_step_time": 6.67076849937439
    },
    {
      "epoch": 6.495475769042969e-06,
      "model_forward_time": 1.7406835556030273,
      "step": 80811
    },
    {
      "epoch": 6.495475769042969e-06,
      "step": 80811,
      "training_step_time": 8.135056257247925
    },
    {
      "epoch": 6.496429443359375e-06,
      "model_forward_time": 1.7422420978546143,
      "step": 80812
    },
    {
      "epoch": 6.496429443359375e-06,
      "step": 80812,
      "training_step_time": 10.679774761199951
    },
    {
      "epoch": 6.497383117675781e-06,
      "model_forward_time": 1.7549715042114258,
      "step": 80813
    },
    {
      "epoch": 6.497383117675781e-06,
      "step": 80813,
      "training_step_time": 8.381680727005005
    },
    {
      "epoch": 6.4983367919921876e-06,
      "model_forward_time": 2.385672092437744,
      "step": 80814
    },
    {
      "epoch": 6.4983367919921876e-06,
      "step": 80814,
      "training_step_time": 7.846731662750244
    },
    {
      "epoch": 6.499290466308594e-06,
      "model_forward_time": 1.7405931949615479,
      "step": 80815
    },
    {
      "epoch": 6.499290466308594e-06,
      "step": 80815,
      "training_step_time": 7.706372976303101
    },
    {
      "epoch": 6.500244140625e-06,
      "model_forward_time": 1.7350976467132568,
      "step": 80816
    },
    {
      "epoch": 6.500244140625e-06,
      "step": 80816,
      "training_step_time": 7.724137306213379
    },
    {
      "epoch": 6.501197814941406e-06,
      "model_forward_time": 1.7356364727020264,
      "step": 80817
    },
    {
      "epoch": 6.501197814941406e-06,
      "step": 80817,
      "training_step_time": 7.536418199539185
    },
    {
      "epoch": 6.502151489257812e-06,
      "model_forward_time": 1.7407877445220947,
      "step": 80818
    },
    {
      "epoch": 6.502151489257812e-06,
      "step": 80818,
      "training_step_time": 7.651562690734863
    },
    {
      "epoch": 6.503105163574219e-06,
      "model_forward_time": 1.739776372909546,
      "step": 80819
    },
    {
      "epoch": 6.503105163574219e-06,
      "step": 80819,
      "training_step_time": 6.786729574203491
    },
    {
      "epoch": 6.504058837890625e-06,
      "grad_norm": 0.09836052358150482,
      "learning_rate": 9.725785060211095e-07,
      "loss": 0.0567,
      "step": 80820
    },
    {
      "epoch": 6.504058837890625e-06,
      "model_forward_time": 1.734724998474121,
      "step": 80820
    },
    {
      "dynamics_loss_avg": 0.06610736977308988,
      "epoch": 6.504058837890625e-06,
      "step": 80820
    },
    {
      "action_loss_avg": 0.0013290474773384631,
      "epoch": 6.504058837890625e-06,
      "step": 80820
    },
    {
      "epoch": 6.504058837890625e-06,
      "step": 80820,
      "training_step_time": 7.620455503463745
    },
    {
      "epoch": 6.505012512207031e-06,
      "model_forward_time": 1.7392911911010742,
      "step": 80821
    },
    {
      "epoch": 6.505012512207031e-06,
      "step": 80821,
      "training_step_time": 9.655029535293579
    },
    {
      "epoch": 6.5059661865234376e-06,
      "model_forward_time": 1.751826286315918,
      "step": 80822
    },
    {
      "epoch": 6.5059661865234376e-06,
      "step": 80822,
      "training_step_time": 8.438892602920532
    },
    {
      "epoch": 6.506919860839844e-06,
      "model_forward_time": 2.420203685760498,
      "step": 80823
    },
    {
      "epoch": 6.506919860839844e-06,
      "step": 80823,
      "training_step_time": 9.02287483215332
    },
    {
      "epoch": 6.50787353515625e-06,
      "model_forward_time": 1.741098165512085,
      "step": 80824
    },
    {
      "epoch": 6.50787353515625e-06,
      "step": 80824,
      "training_step_time": 7.719668626785278
    },
    {
      "epoch": 6.508827209472656e-06,
      "model_forward_time": 1.7371604442596436,
      "step": 80825
    },
    {
      "epoch": 6.508827209472656e-06,
      "step": 80825,
      "training_step_time": 7.763488054275513
    },
    {
      "epoch": 6.509780883789062e-06,
      "model_forward_time": 1.7379117012023926,
      "step": 80826
    },
    {
      "epoch": 6.509780883789062e-06,
      "step": 80826,
      "training_step_time": 8.366366624832153
    },
    {
      "epoch": 6.510734558105469e-06,
      "model_forward_time": 1.7391417026519775,
      "step": 80827
    },
    {
      "epoch": 6.510734558105469e-06,
      "step": 80827,
      "training_step_time": 7.772016763687134
    },
    {
      "epoch": 6.511688232421875e-06,
      "model_forward_time": 1.738781213760376,
      "step": 80828
    },
    {
      "epoch": 6.511688232421875e-06,
      "step": 80828,
      "training_step_time": 6.882873296737671
    },
    {
      "epoch": 6.512641906738281e-06,
      "model_forward_time": 1.7356560230255127,
      "step": 80829
    },
    {
      "epoch": 6.512641906738281e-06,
      "step": 80829,
      "training_step_time": 7.746235609054565
    },
    {
      "epoch": 6.5135955810546876e-06,
      "grad_norm": 0.09669485688209534,
      "learning_rate": 9.715988517430896e-07,
      "loss": 0.0551,
      "step": 80830
    },
    {
      "epoch": 6.5135955810546876e-06,
      "model_forward_time": 1.7366321086883545,
      "step": 80830
    },
    {
      "dynamics_loss_avg": 0.06519688088446855,
      "epoch": 6.5135955810546876e-06,
      "step": 80830
    },
    {
      "action_loss_avg": 0.0018918262387160212,
      "epoch": 6.5135955810546876e-06,
      "step": 80830
    },
    {
      "epoch": 6.5135955810546876e-06,
      "step": 80830,
      "training_step_time": 7.697367191314697
    },
    {
      "epoch": 6.514549255371094e-06,
      "model_forward_time": 1.750396728515625,
      "step": 80831
    },
    {
      "epoch": 6.514549255371094e-06,
      "step": 80831,
      "training_step_time": 11.007478952407837
    },
    {
      "epoch": 6.5155029296875e-06,
      "model_forward_time": 2.377866744995117,
      "step": 80832
    },
    {
      "epoch": 6.5155029296875e-06,
      "step": 80832,
      "training_step_time": 8.318148136138916
    },
    {
      "epoch": 6.5164566040039064e-06,
      "model_forward_time": 1.7290236949920654,
      "step": 80833
    },
    {
      "epoch": 6.5164566040039064e-06,
      "step": 80833,
      "training_step_time": 7.634016275405884
    },
    {
      "epoch": 6.517410278320312e-06,
      "model_forward_time": 1.7211556434631348,
      "step": 80834
    },
    {
      "epoch": 6.517410278320312e-06,
      "step": 80834,
      "training_step_time": 7.612859725952148
    },
    {
      "epoch": 6.518363952636719e-06,
      "model_forward_time": 1.7237703800201416,
      "step": 80835
    },
    {
      "epoch": 6.518363952636719e-06,
      "step": 80835,
      "training_step_time": 8.336597681045532
    },
    {
      "epoch": 6.519317626953125e-06,
      "model_forward_time": 1.7245185375213623,
      "step": 80836
    },
    {
      "epoch": 6.519317626953125e-06,
      "step": 80836,
      "training_step_time": 7.607630252838135
    },
    {
      "epoch": 6.520271301269531e-06,
      "model_forward_time": 1.7399592399597168,
      "step": 80837
    },
    {
      "epoch": 6.520271301269531e-06,
      "step": 80837,
      "training_step_time": 6.769792795181274
    },
    {
      "epoch": 6.5212249755859376e-06,
      "model_forward_time": 1.7464570999145508,
      "step": 80838
    },
    {
      "epoch": 6.5212249755859376e-06,
      "step": 80838,
      "training_step_time": 7.556232929229736
    },
    {
      "epoch": 6.522178649902344e-06,
      "model_forward_time": 1.738351821899414,
      "step": 80839
    },
    {
      "epoch": 6.522178649902344e-06,
      "step": 80839,
      "training_step_time": 8.732317209243774
    },
    {
      "epoch": 6.52313232421875e-06,
      "grad_norm": 0.1691252738237381,
      "learning_rate": 9.706196380049627e-07,
      "loss": 0.0563,
      "step": 80840
    },
    {
      "epoch": 6.52313232421875e-06,
      "model_forward_time": 1.7292141914367676,
      "step": 80840
    },
    {
      "dynamics_loss_avg": 0.0645242003723979,
      "epoch": 6.52313232421875e-06,
      "step": 80840
    },
    {
      "action_loss_avg": 0.0016419513151049614,
      "epoch": 6.52313232421875e-06,
      "step": 80840
    },
    {
      "epoch": 6.52313232421875e-06,
      "step": 80840,
      "training_step_time": 8.270256280899048
    },
    {
      "epoch": 6.5240859985351564e-06,
      "model_forward_time": 1.7350022792816162,
      "step": 80841
    },
    {
      "epoch": 6.5240859985351564e-06,
      "step": 80841,
      "training_step_time": 8.86186146736145
    },
    {
      "epoch": 6.525039672851562e-06,
      "model_forward_time": 1.7398490905761719,
      "step": 80842
    },
    {
      "epoch": 6.525039672851562e-06,
      "step": 80842,
      "training_step_time": 8.463974237442017
    },
    {
      "epoch": 6.525993347167969e-06,
      "model_forward_time": 1.740457534790039,
      "step": 80843
    },
    {
      "epoch": 6.525993347167969e-06,
      "step": 80843,
      "training_step_time": 7.868094444274902
    },
    {
      "epoch": 6.526947021484375e-06,
      "model_forward_time": 1.741955041885376,
      "step": 80844
    },
    {
      "epoch": 6.526947021484375e-06,
      "step": 80844,
      "training_step_time": 8.572198867797852
    },
    {
      "epoch": 6.527900695800781e-06,
      "model_forward_time": 1.7360966205596924,
      "step": 80845
    },
    {
      "epoch": 6.527900695800781e-06,
      "step": 80845,
      "training_step_time": 7.582228183746338
    },
    {
      "epoch": 6.528854370117188e-06,
      "model_forward_time": 1.740807294845581,
      "step": 80846
    },
    {
      "epoch": 6.528854370117188e-06,
      "step": 80846,
      "training_step_time": 7.632812261581421
    },
    {
      "epoch": 6.529808044433594e-06,
      "model_forward_time": 1.7501778602600098,
      "step": 80847
    },
    {
      "epoch": 6.529808044433594e-06,
      "step": 80847,
      "training_step_time": 7.618056774139404
    },
    {
      "epoch": 6.53076171875e-06,
      "model_forward_time": 1.749506950378418,
      "step": 80848
    },
    {
      "epoch": 6.53076171875e-06,
      "step": 80848,
      "training_step_time": 7.663161993026733
    },
    {
      "epoch": 6.5317153930664064e-06,
      "model_forward_time": 1.743931770324707,
      "step": 80849
    },
    {
      "epoch": 6.5317153930664064e-06,
      "step": 80849,
      "training_step_time": 10.601776123046875
    },
    {
      "epoch": 6.532669067382812e-06,
      "grad_norm": 0.10607165098190308,
      "learning_rate": 9.69640864913815e-07,
      "loss": 0.0562,
      "step": 80850
    },
    {
      "epoch": 6.532669067382812e-06,
      "model_forward_time": 1.8489656448364258,
      "step": 80850
    },
    {
      "dynamics_loss_avg": 0.0547101529315114,
      "epoch": 6.532669067382812e-06,
      "step": 80850
    },
    {
      "action_loss_avg": 0.0012718560348730535,
      "epoch": 6.532669067382812e-06,
      "step": 80850
    },
    {
      "epoch": 6.532669067382812e-06,
      "step": 80850,
      "training_step_time": 8.31316590309143
    },
    {
      "epoch": 6.533622741699219e-06,
      "model_forward_time": 2.3785407543182373,
      "step": 80851
    },
    {
      "epoch": 6.533622741699219e-06,
      "step": 80851,
      "training_step_time": 7.714247941970825
    },
    {
      "epoch": 6.534576416015625e-06,
      "model_forward_time": 1.7716286182403564,
      "step": 80852
    },
    {
      "epoch": 6.534576416015625e-06,
      "step": 80852,
      "training_step_time": 7.319198846817017
    },
    {
      "epoch": 6.535530090332031e-06,
      "model_forward_time": 1.7413458824157715,
      "step": 80853
    },
    {
      "epoch": 6.535530090332031e-06,
      "step": 80853,
      "training_step_time": 7.570925235748291
    },
    {
      "epoch": 6.536483764648438e-06,
      "model_forward_time": 1.7415494918823242,
      "step": 80854
    },
    {
      "epoch": 6.536483764648438e-06,
      "step": 80854,
      "training_step_time": 7.485453128814697
    },
    {
      "epoch": 6.537437438964844e-06,
      "model_forward_time": 1.7415447235107422,
      "step": 80855
    },
    {
      "epoch": 6.537437438964844e-06,
      "step": 80855,
      "training_step_time": 7.6639580726623535
    },
    {
      "epoch": 6.53839111328125e-06,
      "model_forward_time": 1.7476353645324707,
      "step": 80856
    },
    {
      "epoch": 6.53839111328125e-06,
      "step": 80856,
      "training_step_time": 7.512321710586548
    },
    {
      "epoch": 6.5393447875976564e-06,
      "model_forward_time": 1.7453818321228027,
      "step": 80857
    },
    {
      "epoch": 6.5393447875976564e-06,
      "step": 80857,
      "training_step_time": 8.509598016738892
    },
    {
      "epoch": 6.540298461914062e-06,
      "model_forward_time": 1.7372636795043945,
      "step": 80858
    },
    {
      "epoch": 6.540298461914062e-06,
      "step": 80858,
      "training_step_time": 10.831573247909546
    },
    {
      "epoch": 6.541252136230469e-06,
      "model_forward_time": 1.8230030536651611,
      "step": 80859
    },
    {
      "epoch": 6.541252136230469e-06,
      "step": 80859,
      "training_step_time": 8.68158745765686
    },
    {
      "epoch": 6.542205810546875e-06,
      "grad_norm": 0.08872339874505997,
      "learning_rate": 9.686625325766818e-07,
      "loss": 0.0545,
      "step": 80860
    },
    {
      "epoch": 6.542205810546875e-06,
      "model_forward_time": 2.397249221801758,
      "step": 80860
    },
    {
      "dynamics_loss_avg": 0.050500281527638434,
      "epoch": 6.542205810546875e-06,
      "step": 80860
    },
    {
      "action_loss_avg": 0.0013362899131607263,
      "epoch": 6.542205810546875e-06,
      "step": 80860
    },
    {
      "epoch": 6.542205810546875e-06,
      "step": 80860,
      "training_step_time": 8.746342897415161
    },
    {
      "epoch": 6.543159484863281e-06,
      "model_forward_time": 1.3932669162750244,
      "step": 80861
    },
    {
      "epoch": 6.543159484863281e-06,
      "step": 80861,
      "training_step_time": 7.457519054412842
    },
    {
      "epoch": 6.544113159179688e-06,
      "model_forward_time": 1.7686238288879395,
      "step": 80862
    },
    {
      "epoch": 6.544113159179688e-06,
      "step": 80862,
      "training_step_time": 7.626035213470459
    },
    {
      "epoch": 6.545066833496094e-06,
      "model_forward_time": 1.7368175983428955,
      "step": 80863
    },
    {
      "epoch": 6.545066833496094e-06,
      "step": 80863,
      "training_step_time": 7.609286308288574
    },
    {
      "epoch": 6.5460205078125e-06,
      "model_forward_time": 1.7411210536956787,
      "step": 80864
    },
    {
      "epoch": 6.5460205078125e-06,
      "step": 80864,
      "training_step_time": 7.592676877975464
    },
    {
      "epoch": 6.5469741821289065e-06,
      "model_forward_time": 1.7434520721435547,
      "step": 80865
    },
    {
      "epoch": 6.5469741821289065e-06,
      "step": 80865,
      "training_step_time": 7.548578977584839
    },
    {
      "epoch": 6.547927856445312e-06,
      "model_forward_time": 1.7658677101135254,
      "step": 80866
    },
    {
      "epoch": 6.547927856445312e-06,
      "step": 80866,
      "training_step_time": 7.622033357620239
    },
    {
      "epoch": 6.548881530761719e-06,
      "model_forward_time": 1.736574411392212,
      "step": 80867
    },
    {
      "epoch": 6.548881530761719e-06,
      "step": 80867,
      "training_step_time": 8.20823621749878
    },
    {
      "epoch": 6.549835205078125e-06,
      "model_forward_time": 1.7832729816436768,
      "step": 80868
    },
    {
      "epoch": 6.549835205078125e-06,
      "step": 80868,
      "training_step_time": 9.161405324935913
    },
    {
      "epoch": 6.550788879394531e-06,
      "model_forward_time": 2.386828660964966,
      "step": 80869
    },
    {
      "epoch": 6.550788879394531e-06,
      "step": 80869,
      "training_step_time": 8.64046049118042
    },
    {
      "epoch": 6.551742553710938e-06,
      "grad_norm": 0.1133955717086792,
      "learning_rate": 9.676846411005542e-07,
      "loss": 0.0541,
      "step": 80870
    },
    {
      "epoch": 6.551742553710938e-06,
      "model_forward_time": 1.7374560832977295,
      "step": 80870
    },
    {
      "dynamics_loss_avg": 0.04198133405297995,
      "epoch": 6.551742553710938e-06,
      "step": 80870
    },
    {
      "action_loss_avg": 0.0014077407628064974,
      "epoch": 6.551742553710938e-06,
      "step": 80870
    },
    {
      "epoch": 6.551742553710938e-06,
      "step": 80870,
      "training_step_time": 7.881872653961182
    },
    {
      "epoch": 6.552696228027344e-06,
      "model_forward_time": 1.7422668933868408,
      "step": 80871
    },
    {
      "epoch": 6.552696228027344e-06,
      "step": 80871,
      "training_step_time": 7.861818790435791
    },
    {
      "epoch": 6.55364990234375e-06,
      "model_forward_time": 1.7449119091033936,
      "step": 80872
    },
    {
      "epoch": 6.55364990234375e-06,
      "step": 80872,
      "training_step_time": 7.570979833602905
    },
    {
      "epoch": 6.5546035766601565e-06,
      "model_forward_time": 1.7405807971954346,
      "step": 80873
    },
    {
      "epoch": 6.5546035766601565e-06,
      "step": 80873,
      "training_step_time": 7.736553907394409
    },
    {
      "epoch": 6.555557250976562e-06,
      "model_forward_time": 1.7401962280273438,
      "step": 80874
    },
    {
      "epoch": 6.555557250976562e-06,
      "step": 80874,
      "training_step_time": 7.7519690990448
    },
    {
      "epoch": 6.556510925292969e-06,
      "model_forward_time": 1.7463641166687012,
      "step": 80875
    },
    {
      "epoch": 6.556510925292969e-06,
      "step": 80875,
      "training_step_time": 7.815800666809082
    },
    {
      "epoch": 6.557464599609375e-06,
      "model_forward_time": 1.7380595207214355,
      "step": 80876
    },
    {
      "epoch": 6.557464599609375e-06,
      "step": 80876,
      "training_step_time": 8.481525659561157
    },
    {
      "epoch": 6.558418273925781e-06,
      "model_forward_time": 1.7544348239898682,
      "step": 80877
    },
    {
      "epoch": 6.558418273925781e-06,
      "step": 80877,
      "training_step_time": 11.678189992904663
    },
    {
      "epoch": 6.559371948242188e-06,
      "model_forward_time": 2.4048550128936768,
      "step": 80878
    },
    {
      "epoch": 6.559371948242188e-06,
      "step": 80878,
      "training_step_time": 7.813215970993042
    },
    {
      "epoch": 6.560325622558593e-06,
      "model_forward_time": 1.7374212741851807,
      "step": 80879
    },
    {
      "epoch": 6.560325622558593e-06,
      "step": 80879,
      "training_step_time": 8.691402673721313
    },
    {
      "epoch": 6.561279296875e-06,
      "grad_norm": 0.09748625010251999,
      "learning_rate": 9.667071905923691e-07,
      "loss": 0.0567,
      "step": 80880
    },
    {
      "epoch": 6.561279296875e-06,
      "model_forward_time": 1.7407159805297852,
      "step": 80880
    },
    {
      "dynamics_loss_avg": 0.05997310169041157,
      "epoch": 6.561279296875e-06,
      "step": 80880
    },
    {
      "action_loss_avg": 0.0012049006967572496,
      "epoch": 6.561279296875e-06,
      "step": 80880
    },
    {
      "epoch": 6.561279296875e-06,
      "step": 80880,
      "training_step_time": 7.740819215774536
    },
    {
      "epoch": 6.5622329711914065e-06,
      "model_forward_time": 1.7529897689819336,
      "step": 80881
    },
    {
      "epoch": 6.5622329711914065e-06,
      "step": 80881,
      "training_step_time": 7.56198525428772
    },
    {
      "epoch": 6.563186645507812e-06,
      "model_forward_time": 1.7570545673370361,
      "step": 80882
    },
    {
      "epoch": 6.563186645507812e-06,
      "step": 80882,
      "training_step_time": 7.787115812301636
    },
    {
      "epoch": 6.564140319824219e-06,
      "model_forward_time": 1.7577135562896729,
      "step": 80883
    },
    {
      "epoch": 6.564140319824219e-06,
      "step": 80883,
      "training_step_time": 7.745211601257324
    },
    {
      "epoch": 6.565093994140625e-06,
      "model_forward_time": 1.756014347076416,
      "step": 80884
    },
    {
      "epoch": 6.565093994140625e-06,
      "step": 80884,
      "training_step_time": 8.778954982757568
    },
    {
      "epoch": 6.566047668457031e-06,
      "model_forward_time": 1.7502868175506592,
      "step": 80885
    },
    {
      "epoch": 6.566047668457031e-06,
      "step": 80885,
      "training_step_time": 9.369136095046997
    },
    {
      "epoch": 6.567001342773438e-06,
      "model_forward_time": 1.7495477199554443,
      "step": 80886
    },
    {
      "epoch": 6.567001342773438e-06,
      "step": 80886,
      "training_step_time": 8.766340494155884
    },
    {
      "epoch": 6.567955017089843e-06,
      "model_forward_time": 1.8606908321380615,
      "step": 80887
    },
    {
      "epoch": 6.567955017089843e-06,
      "step": 80887,
      "training_step_time": 8.326287984848022
    },
    {
      "epoch": 6.56890869140625e-06,
      "model_forward_time": 1.7452290058135986,
      "step": 80888
    },
    {
      "epoch": 6.56890869140625e-06,
      "step": 80888,
      "training_step_time": 8.678797006607056
    },
    {
      "epoch": 6.5698623657226565e-06,
      "model_forward_time": 1.3744516372680664,
      "step": 80889
    },
    {
      "epoch": 6.5698623657226565e-06,
      "step": 80889,
      "training_step_time": 7.74473237991333
    },
    {
      "epoch": 6.570816040039062e-06,
      "grad_norm": 0.0889086052775383,
      "learning_rate": 9.657301811590241e-07,
      "loss": 0.0545,
      "step": 80890
    },
    {
      "epoch": 6.570816040039062e-06,
      "model_forward_time": 1.742391586303711,
      "step": 80890
    },
    {
      "dynamics_loss_avg": 0.04206105805933476,
      "epoch": 6.570816040039062e-06,
      "step": 80890
    },
    {
      "action_loss_avg": 0.0010693727992475033,
      "epoch": 6.570816040039062e-06,
      "step": 80890
    },
    {
      "epoch": 6.570816040039062e-06,
      "step": 80890,
      "training_step_time": 7.476372241973877
    },
    {
      "epoch": 6.571769714355469e-06,
      "model_forward_time": 1.7295379638671875,
      "step": 80891
    },
    {
      "epoch": 6.571769714355469e-06,
      "step": 80891,
      "training_step_time": 7.516077280044556
    },
    {
      "epoch": 6.572723388671875e-06,
      "model_forward_time": 1.7557170391082764,
      "step": 80892
    },
    {
      "epoch": 6.572723388671875e-06,
      "step": 80892,
      "training_step_time": 7.742452621459961
    },
    {
      "epoch": 6.573677062988281e-06,
      "model_forward_time": 1.036353588104248,
      "step": 80893
    },
    {
      "epoch": 6.573677062988281e-06,
      "step": 80893,
      "training_step_time": 7.623469591140747
    },
    {
      "epoch": 6.574630737304688e-06,
      "model_forward_time": 1.7507128715515137,
      "step": 80894
    },
    {
      "epoch": 6.574630737304688e-06,
      "step": 80894,
      "training_step_time": 7.577472925186157
    },
    {
      "epoch": 6.575584411621093e-06,
      "model_forward_time": 1.3801817893981934,
      "step": 80895
    },
    {
      "epoch": 6.575584411621093e-06,
      "step": 80895,
      "training_step_time": 10.245547771453857
    },
    {
      "epoch": 6.5765380859375e-06,
      "model_forward_time": 1.8584811687469482,
      "step": 80896
    },
    {
      "epoch": 6.5765380859375e-06,
      "step": 80896,
      "training_step_time": 10.725812196731567
    },
    {
      "epoch": 6.5774917602539065e-06,
      "model_forward_time": 2.4086527824401855,
      "step": 80897
    },
    {
      "epoch": 6.5774917602539065e-06,
      "step": 80897,
      "training_step_time": 7.702200174331665
    },
    {
      "epoch": 6.578445434570312e-06,
      "model_forward_time": 1.7685604095458984,
      "step": 80898
    },
    {
      "epoch": 6.578445434570312e-06,
      "step": 80898,
      "training_step_time": 8.25440001487732
    },
    {
      "epoch": 6.579399108886719e-06,
      "model_forward_time": 1.7680420875549316,
      "step": 80899
    },
    {
      "epoch": 6.579399108886719e-06,
      "step": 80899,
      "training_step_time": 7.412917852401733
    },
    {
      "epoch": 6.580352783203125e-06,
      "grad_norm": 0.07244224101305008,
      "learning_rate": 9.647536129073603e-07,
      "loss": 0.0537,
      "step": 80900
    },
    {
      "epoch": 6.580352783203125e-06,
      "model_forward_time": 1.7407832145690918,
      "step": 80900
    },
    {
      "dynamics_loss_avg": 0.05515577681362629,
      "epoch": 6.580352783203125e-06,
      "step": 80900
    },
    {
      "action_loss_avg": 0.0009174084465485066,
      "epoch": 6.580352783203125e-06,
      "step": 80900
    },
    {
      "epoch": 6.580352783203125e-06,
      "step": 80900,
      "training_step_time": 7.541042327880859
    },
    {
      "epoch": 6.581306457519531e-06,
      "model_forward_time": 1.763805866241455,
      "step": 80901
    },
    {
      "epoch": 6.581306457519531e-06,
      "step": 80901,
      "training_step_time": 7.509028434753418
    },
    {
      "epoch": 6.582260131835938e-06,
      "model_forward_time": 1.753166675567627,
      "step": 80902
    },
    {
      "epoch": 6.582260131835938e-06,
      "step": 80902,
      "training_step_time": 7.521341323852539
    },
    {
      "epoch": 6.583213806152343e-06,
      "model_forward_time": 1.7584354877471924,
      "step": 80903
    },
    {
      "epoch": 6.583213806152343e-06,
      "step": 80903,
      "training_step_time": 8.301684141159058
    },
    {
      "epoch": 6.58416748046875e-06,
      "model_forward_time": 1.755915880203247,
      "step": 80904
    },
    {
      "epoch": 6.58416748046875e-06,
      "step": 80904,
      "training_step_time": 8.895197629928589
    },
    {
      "epoch": 6.5851211547851565e-06,
      "model_forward_time": 1.8211195468902588,
      "step": 80905
    },
    {
      "epoch": 6.5851211547851565e-06,
      "step": 80905,
      "training_step_time": 8.630630016326904
    },
    {
      "epoch": 6.586074829101562e-06,
      "model_forward_time": 2.343177556991577,
      "step": 80906
    },
    {
      "epoch": 6.586074829101562e-06,
      "step": 80906,
      "training_step_time": 8.52448844909668
    },
    {
      "epoch": 6.587028503417969e-06,
      "model_forward_time": 1.7645094394683838,
      "step": 80907
    },
    {
      "epoch": 6.587028503417969e-06,
      "step": 80907,
      "training_step_time": 8.144492387771606
    },
    {
      "epoch": 6.587982177734375e-06,
      "model_forward_time": 1.7572417259216309,
      "step": 80908
    },
    {
      "epoch": 6.587982177734375e-06,
      "step": 80908,
      "training_step_time": 7.360518217086792
    },
    {
      "epoch": 6.588935852050781e-06,
      "model_forward_time": 1.751316785812378,
      "step": 80909
    },
    {
      "epoch": 6.588935852050781e-06,
      "step": 80909,
      "training_step_time": 7.522662162780762
    },
    {
      "epoch": 6.589889526367188e-06,
      "grad_norm": 0.0871925950050354,
      "learning_rate": 9.63777485944175e-07,
      "loss": 0.0537,
      "step": 80910
    },
    {
      "epoch": 6.589889526367188e-06,
      "model_forward_time": 1.7498300075531006,
      "step": 80910
    },
    {
      "dynamics_loss_avg": 0.0568210918456316,
      "epoch": 6.589889526367188e-06,
      "step": 80910
    },
    {
      "action_loss_avg": 0.0011107948754215612,
      "epoch": 6.589889526367188e-06,
      "step": 80910
    },
    {
      "epoch": 6.589889526367188e-06,
      "step": 80910,
      "training_step_time": 7.500338792800903
    },
    {
      "epoch": 6.590843200683593e-06,
      "model_forward_time": 1.3805198669433594,
      "step": 80911
    },
    {
      "epoch": 6.590843200683593e-06,
      "step": 80911,
      "training_step_time": 7.495961904525757
    },
    {
      "epoch": 6.591796875e-06,
      "model_forward_time": 1.7429263591766357,
      "step": 80912
    },
    {
      "epoch": 6.591796875e-06,
      "step": 80912,
      "training_step_time": 9.222979068756104
    },
    {
      "epoch": 6.5927505493164065e-06,
      "model_forward_time": 1.7430846691131592,
      "step": 80913
    },
    {
      "epoch": 6.5927505493164065e-06,
      "step": 80913,
      "training_step_time": 9.05877947807312
    },
    {
      "epoch": 6.593704223632812e-06,
      "model_forward_time": 1.7922983169555664,
      "step": 80914
    },
    {
      "epoch": 6.593704223632812e-06,
      "step": 80914,
      "training_step_time": 8.54626727104187
    },
    {
      "epoch": 6.594657897949219e-06,
      "model_forward_time": 1.7464165687561035,
      "step": 80915
    },
    {
      "epoch": 6.594657897949219e-06,
      "step": 80915,
      "training_step_time": 8.128844976425171
    },
    {
      "epoch": 6.595611572265625e-06,
      "model_forward_time": 2.375786066055298,
      "step": 80916
    },
    {
      "epoch": 6.595611572265625e-06,
      "step": 80916,
      "training_step_time": 7.600959062576294
    },
    {
      "epoch": 6.596565246582031e-06,
      "model_forward_time": 1.746352195739746,
      "step": 80917
    },
    {
      "epoch": 6.596565246582031e-06,
      "step": 80917,
      "training_step_time": 7.73825216293335
    },
    {
      "epoch": 6.597518920898438e-06,
      "model_forward_time": 1.740309715270996,
      "step": 80918
    },
    {
      "epoch": 6.597518920898438e-06,
      "step": 80918,
      "training_step_time": 7.713537693023682
    },
    {
      "epoch": 6.598472595214843e-06,
      "model_forward_time": 1.7417261600494385,
      "step": 80919
    },
    {
      "epoch": 6.598472595214843e-06,
      "step": 80919,
      "training_step_time": 7.477505445480347
    },
    {
      "epoch": 6.59942626953125e-06,
      "grad_norm": 0.10076877474784851,
      "learning_rate": 9.628018003762153e-07,
      "loss": 0.0537,
      "step": 80920
    },
    {
      "epoch": 6.59942626953125e-06,
      "model_forward_time": 1.747624397277832,
      "step": 80920
    },
    {
      "dynamics_loss_avg": 0.04935130290687084,
      "epoch": 6.59942626953125e-06,
      "step": 80920
    },
    {
      "action_loss_avg": 0.000917253823718056,
      "epoch": 6.59942626953125e-06,
      "step": 80920
    },
    {
      "epoch": 6.59942626953125e-06,
      "step": 80920,
      "training_step_time": 7.578794717788696
    },
    {
      "epoch": 6.6003799438476565e-06,
      "model_forward_time": 1.7417078018188477,
      "step": 80921
    },
    {
      "epoch": 6.6003799438476565e-06,
      "step": 80921,
      "training_step_time": 7.661845922470093
    },
    {
      "epoch": 6.601333618164062e-06,
      "model_forward_time": 1.757610559463501,
      "step": 80922
    },
    {
      "epoch": 6.601333618164062e-06,
      "step": 80922,
      "training_step_time": 7.608661413192749
    },
    {
      "epoch": 6.602287292480469e-06,
      "model_forward_time": 1.7717368602752686,
      "step": 80923
    },
    {
      "epoch": 6.602287292480469e-06,
      "step": 80923,
      "training_step_time": 9.2333402633667
    },
    {
      "epoch": 6.603240966796875e-06,
      "model_forward_time": 1.7548046112060547,
      "step": 80924
    },
    {
      "epoch": 6.603240966796875e-06,
      "step": 80924,
      "training_step_time": 8.157873153686523
    },
    {
      "epoch": 6.604194641113281e-06,
      "model_forward_time": 2.3668105602264404,
      "step": 80925
    },
    {
      "epoch": 6.604194641113281e-06,
      "step": 80925,
      "training_step_time": 9.077629566192627
    },
    {
      "epoch": 6.605148315429688e-06,
      "model_forward_time": 1.7580013275146484,
      "step": 80926
    },
    {
      "epoch": 6.605148315429688e-06,
      "step": 80926,
      "training_step_time": 7.600205421447754
    },
    {
      "epoch": 6.606101989746093e-06,
      "model_forward_time": 1.757439136505127,
      "step": 80927
    },
    {
      "epoch": 6.606101989746093e-06,
      "step": 80927,
      "training_step_time": 6.759096145629883
    },
    {
      "epoch": 6.6070556640625e-06,
      "model_forward_time": 1.7464792728424072,
      "step": 80928
    },
    {
      "epoch": 6.6070556640625e-06,
      "step": 80928,
      "training_step_time": 7.684231996536255
    },
    {
      "epoch": 6.6080093383789065e-06,
      "model_forward_time": 1.7512166500091553,
      "step": 80929
    },
    {
      "epoch": 6.6080093383789065e-06,
      "step": 80929,
      "training_step_time": 7.594422340393066
    },
    {
      "epoch": 6.608963012695312e-06,
      "grad_norm": 0.09739166498184204,
      "learning_rate": 9.618265563101814e-07,
      "loss": 0.0545,
      "step": 80930
    },
    {
      "epoch": 6.608963012695312e-06,
      "model_forward_time": 1.7389955520629883,
      "step": 80930
    },
    {
      "dynamics_loss_avg": 0.0593705365434289,
      "epoch": 6.608963012695312e-06,
      "step": 80930
    },
    {
      "action_loss_avg": 0.0014699383114930241,
      "epoch": 6.608963012695312e-06,
      "step": 80930
    },
    {
      "epoch": 6.608963012695312e-06,
      "step": 80930,
      "training_step_time": 7.597756624221802
    },
    {
      "epoch": 6.609916687011719e-06,
      "model_forward_time": 1.7499496936798096,
      "step": 80931
    },
    {
      "epoch": 6.609916687011719e-06,
      "step": 80931,
      "training_step_time": 9.747803926467896
    },
    {
      "epoch": 6.610870361328125e-06,
      "model_forward_time": 1.7465465068817139,
      "step": 80932
    },
    {
      "epoch": 6.610870361328125e-06,
      "step": 80932,
      "training_step_time": 9.688799858093262
    },
    {
      "epoch": 6.611824035644531e-06,
      "model_forward_time": 1.764066457748413,
      "step": 80933
    },
    {
      "epoch": 6.611824035644531e-06,
      "step": 80933,
      "training_step_time": 9.427592754364014
    },
    {
      "epoch": 6.612777709960938e-06,
      "model_forward_time": 2.4171221256256104,
      "step": 80934
    },
    {
      "epoch": 6.612777709960938e-06,
      "step": 80934,
      "training_step_time": 9.883765697479248
    },
    {
      "epoch": 6.613731384277343e-06,
      "model_forward_time": 1.7511987686157227,
      "step": 80935
    },
    {
      "epoch": 6.613731384277343e-06,
      "step": 80935,
      "training_step_time": 7.685551404953003
    },
    {
      "epoch": 6.61468505859375e-06,
      "model_forward_time": 1.7466249465942383,
      "step": 80936
    },
    {
      "epoch": 6.61468505859375e-06,
      "step": 80936,
      "training_step_time": 6.871848106384277
    },
    {
      "epoch": 6.6156387329101565e-06,
      "model_forward_time": 1.7535536289215088,
      "step": 80937
    },
    {
      "epoch": 6.6156387329101565e-06,
      "step": 80937,
      "training_step_time": 7.611361026763916
    },
    {
      "epoch": 6.616592407226562e-06,
      "model_forward_time": 1.75882887840271,
      "step": 80938
    },
    {
      "epoch": 6.616592407226562e-06,
      "step": 80938,
      "training_step_time": 7.660660028457642
    },
    {
      "epoch": 6.617546081542969e-06,
      "model_forward_time": 1.7378668785095215,
      "step": 80939
    },
    {
      "epoch": 6.617546081542969e-06,
      "step": 80939,
      "training_step_time": 7.744802951812744
    },
    {
      "epoch": 6.618499755859375e-06,
      "grad_norm": 0.09464708715677261,
      "learning_rate": 9.608517538527212e-07,
      "loss": 0.0544,
      "step": 80940
    },
    {
      "epoch": 6.618499755859375e-06,
      "model_forward_time": 1.7338407039642334,
      "step": 80940
    },
    {
      "dynamics_loss_avg": 0.07149922251701354,
      "epoch": 6.618499755859375e-06,
      "step": 80940
    },
    {
      "action_loss_avg": 0.002094135357765481,
      "epoch": 6.618499755859375e-06,
      "step": 80940
    },
    {
      "epoch": 6.618499755859375e-06,
      "step": 80940,
      "training_step_time": 7.523029327392578
    },
    {
      "epoch": 6.619453430175781e-06,
      "model_forward_time": 1.7495474815368652,
      "step": 80941
    },
    {
      "epoch": 6.619453430175781e-06,
      "step": 80941,
      "training_step_time": 9.209423065185547
    },
    {
      "epoch": 6.620407104492188e-06,
      "model_forward_time": 1.8415334224700928,
      "step": 80942
    },
    {
      "epoch": 6.620407104492188e-06,
      "step": 80942,
      "training_step_time": 8.569571495056152
    },
    {
      "epoch": 6.621360778808593e-06,
      "model_forward_time": 2.3802490234375,
      "step": 80943
    },
    {
      "epoch": 6.621360778808593e-06,
      "step": 80943,
      "training_step_time": 7.759320497512817
    },
    {
      "epoch": 6.622314453125e-06,
      "model_forward_time": 1.7379233837127686,
      "step": 80944
    },
    {
      "epoch": 6.622314453125e-06,
      "step": 80944,
      "training_step_time": 8.451757669448853
    },
    {
      "epoch": 6.6232681274414065e-06,
      "model_forward_time": 1.741180658340454,
      "step": 80945
    },
    {
      "epoch": 6.6232681274414065e-06,
      "step": 80945,
      "training_step_time": 7.764634847640991
    },
    {
      "epoch": 6.624221801757812e-06,
      "model_forward_time": 1.0379862785339355,
      "step": 80946
    },
    {
      "epoch": 6.624221801757812e-06,
      "step": 80946,
      "training_step_time": 7.707495450973511
    },
    {
      "epoch": 6.625175476074219e-06,
      "model_forward_time": 1.7522330284118652,
      "step": 80947
    },
    {
      "epoch": 6.625175476074219e-06,
      "step": 80947,
      "training_step_time": 7.691367864608765
    },
    {
      "epoch": 6.626129150390625e-06,
      "model_forward_time": 1.7437353134155273,
      "step": 80948
    },
    {
      "epoch": 6.626129150390625e-06,
      "step": 80948,
      "training_step_time": 7.724298477172852
    },
    {
      "epoch": 6.627082824707031e-06,
      "model_forward_time": 1.7809903621673584,
      "step": 80949
    },
    {
      "epoch": 6.627082824707031e-06,
      "step": 80949,
      "training_step_time": 7.534498453140259
    },
    {
      "epoch": 6.628036499023438e-06,
      "grad_norm": 0.0957651361823082,
      "learning_rate": 9.598773931104433e-07,
      "loss": 0.0536,
      "step": 80950
    },
    {
      "epoch": 6.628036499023438e-06,
      "model_forward_time": 1.7431492805480957,
      "step": 80950
    },
    {
      "dynamics_loss_avg": 0.06176703497767448,
      "epoch": 6.628036499023438e-06,
      "step": 80950
    },
    {
      "action_loss_avg": 0.0009189521137159318,
      "epoch": 6.628036499023438e-06,
      "step": 80950
    },
    {
      "epoch": 6.628036499023438e-06,
      "step": 80950,
      "training_step_time": 9.787916421890259
    },
    {
      "epoch": 6.628990173339843e-06,
      "model_forward_time": 1.8177909851074219,
      "step": 80951
    },
    {
      "epoch": 6.628990173339843e-06,
      "step": 80951,
      "training_step_time": 9.077605724334717
    },
    {
      "epoch": 6.62994384765625e-06,
      "model_forward_time": 1.7495546340942383,
      "step": 80952
    },
    {
      "epoch": 6.62994384765625e-06,
      "step": 80952,
      "training_step_time": 8.297582864761353
    },
    {
      "epoch": 6.6308975219726565e-06,
      "model_forward_time": 2.380281925201416,
      "step": 80953
    },
    {
      "epoch": 6.6308975219726565e-06,
      "step": 80953,
      "training_step_time": 8.364085912704468
    },
    {
      "epoch": 6.631851196289062e-06,
      "model_forward_time": 1.7548060417175293,
      "step": 80954
    },
    {
      "epoch": 6.631851196289062e-06,
      "step": 80954,
      "training_step_time": 7.41346001625061
    },
    {
      "epoch": 6.632804870605469e-06,
      "model_forward_time": 1.7515206336975098,
      "step": 80955
    },
    {
      "epoch": 6.632804870605469e-06,
      "step": 80955,
      "training_step_time": 7.602541446685791
    },
    {
      "epoch": 6.633758544921875e-06,
      "model_forward_time": 1.7544751167297363,
      "step": 80956
    },
    {
      "epoch": 6.633758544921875e-06,
      "step": 80956,
      "training_step_time": 7.691870927810669
    },
    {
      "epoch": 6.634712219238281e-06,
      "model_forward_time": 1.753478765487671,
      "step": 80957
    },
    {
      "epoch": 6.634712219238281e-06,
      "step": 80957,
      "training_step_time": 7.741771221160889
    },
    {
      "epoch": 6.635665893554688e-06,
      "model_forward_time": 1.7552611827850342,
      "step": 80958
    },
    {
      "epoch": 6.635665893554688e-06,
      "step": 80958,
      "training_step_time": 7.723365783691406
    },
    {
      "epoch": 6.636619567871093e-06,
      "model_forward_time": 1.7554943561553955,
      "step": 80959
    },
    {
      "epoch": 6.636619567871093e-06,
      "step": 80959,
      "training_step_time": 8.641978979110718
    },
    {
      "epoch": 6.6375732421875e-06,
      "grad_norm": 0.08279059082269669,
      "learning_rate": 9.589034741898984e-07,
      "loss": 0.0553,
      "step": 80960
    },
    {
      "epoch": 6.6375732421875e-06,
      "model_forward_time": 1.7797422409057617,
      "step": 80960
    },
    {
      "dynamics_loss_avg": 0.0621841823682189,
      "epoch": 6.6375732421875e-06,
      "step": 80960
    },
    {
      "action_loss_avg": 0.0009832646901486441,
      "epoch": 6.6375732421875e-06,
      "step": 80960
    },
    {
      "epoch": 6.6375732421875e-06,
      "step": 80960,
      "training_step_time": 9.459479570388794
    },
    {
      "epoch": 6.6385269165039065e-06,
      "model_forward_time": 1.7519328594207764,
      "step": 80961
    },
    {
      "epoch": 6.6385269165039065e-06,
      "step": 80961,
      "training_step_time": 7.817787170410156
    },
    {
      "epoch": 6.639480590820312e-06,
      "model_forward_time": 2.366593837738037,
      "step": 80962
    },
    {
      "epoch": 6.639480590820312e-06,
      "step": 80962,
      "training_step_time": 8.383745908737183
    },
    {
      "epoch": 6.640434265136719e-06,
      "model_forward_time": 1.7492461204528809,
      "step": 80963
    },
    {
      "epoch": 6.640434265136719e-06,
      "step": 80963,
      "training_step_time": 7.767490386962891
    },
    {
      "epoch": 6.641387939453125e-06,
      "model_forward_time": 1.7499303817749023,
      "step": 80964
    },
    {
      "epoch": 6.641387939453125e-06,
      "step": 80964,
      "training_step_time": 7.647409200668335
    },
    {
      "epoch": 6.642341613769531e-06,
      "model_forward_time": 1.7328050136566162,
      "step": 80965
    },
    {
      "epoch": 6.642341613769531e-06,
      "step": 80965,
      "training_step_time": 7.72822380065918
    },
    {
      "epoch": 6.643295288085938e-06,
      "model_forward_time": 1.7537384033203125,
      "step": 80966
    },
    {
      "epoch": 6.643295288085938e-06,
      "step": 80966,
      "training_step_time": 7.703462362289429
    },
    {
      "epoch": 6.644248962402343e-06,
      "model_forward_time": 1.7534470558166504,
      "step": 80967
    },
    {
      "epoch": 6.644248962402343e-06,
      "step": 80967,
      "training_step_time": 8.416032075881958
    },
    {
      "epoch": 6.64520263671875e-06,
      "model_forward_time": 1.7553951740264893,
      "step": 80968
    },
    {
      "epoch": 6.64520263671875e-06,
      "step": 80968,
      "training_step_time": 8.40616512298584
    },
    {
      "epoch": 6.6461563110351565e-06,
      "model_forward_time": 1.7677297592163086,
      "step": 80969
    },
    {
      "epoch": 6.6461563110351565e-06,
      "step": 80969,
      "training_step_time": 10.685036420822144
    },
    {
      "epoch": 6.647109985351562e-06,
      "grad_norm": 0.06640291959047318,
      "learning_rate": 9.579299971975946e-07,
      "loss": 0.0544,
      "step": 80970
    },
    {
      "epoch": 6.647109985351562e-06,
      "model_forward_time": 1.7342033386230469,
      "step": 80970
    },
    {
      "dynamics_loss_avg": 0.04642095100134611,
      "epoch": 6.647109985351562e-06,
      "step": 80970
    },
    {
      "action_loss_avg": 0.0016193940275115893,
      "epoch": 6.647109985351562e-06,
      "step": 80970
    },
    {
      "epoch": 6.647109985351562e-06,
      "step": 80970,
      "training_step_time": 7.780442714691162
    },
    {
      "epoch": 6.648063659667969e-06,
      "model_forward_time": 2.4188294410705566,
      "step": 80971
    },
    {
      "epoch": 6.648063659667969e-06,
      "step": 80971,
      "training_step_time": 7.794692516326904
    },
    {
      "epoch": 6.649017333984375e-06,
      "model_forward_time": 1.7522954940795898,
      "step": 80972
    },
    {
      "epoch": 6.649017333984375e-06,
      "step": 80972,
      "training_step_time": 7.750676870346069
    },
    {
      "epoch": 6.649971008300781e-06,
      "model_forward_time": 1.753371238708496,
      "step": 80973
    },
    {
      "epoch": 6.649971008300781e-06,
      "step": 80973,
      "training_step_time": 7.680866003036499
    },
    {
      "epoch": 6.650924682617188e-06,
      "model_forward_time": 1.7538657188415527,
      "step": 80974
    },
    {
      "epoch": 6.650924682617188e-06,
      "step": 80974,
      "training_step_time": 7.687250852584839
    },
    {
      "epoch": 6.651878356933593e-06,
      "model_forward_time": 1.74436354637146,
      "step": 80975
    },
    {
      "epoch": 6.651878356933593e-06,
      "step": 80975,
      "training_step_time": 7.6644370555877686
    },
    {
      "epoch": 6.65283203125e-06,
      "model_forward_time": 1.757810115814209,
      "step": 80976
    },
    {
      "epoch": 6.65283203125e-06,
      "step": 80976,
      "training_step_time": 7.697993993759155
    },
    {
      "epoch": 6.6537857055664065e-06,
      "model_forward_time": 1.746385097503662,
      "step": 80977
    },
    {
      "epoch": 6.6537857055664065e-06,
      "step": 80977,
      "training_step_time": 9.142844438552856
    },
    {
      "epoch": 6.654739379882812e-06,
      "model_forward_time": 1.7456080913543701,
      "step": 80978
    },
    {
      "epoch": 6.654739379882812e-06,
      "step": 80978,
      "training_step_time": 7.807554483413696
    },
    {
      "epoch": 6.655693054199219e-06,
      "model_forward_time": 1.864311933517456,
      "step": 80979
    },
    {
      "epoch": 6.655693054199219e-06,
      "step": 80979,
      "training_step_time": 7.631489038467407
    },
    {
      "epoch": 6.656646728515625e-06,
      "grad_norm": 0.08156483620405197,
      "learning_rate": 9.569569622399882e-07,
      "loss": 0.0564,
      "step": 80980
    },
    {
      "epoch": 6.656646728515625e-06,
      "model_forward_time": 1.737595796585083,
      "step": 80980
    },
    {
      "dynamics_loss_avg": 0.04557910393923521,
      "epoch": 6.656646728515625e-06,
      "step": 80980
    },
    {
      "action_loss_avg": 0.0006798688846174628,
      "epoch": 6.656646728515625e-06,
      "step": 80980
    },
    {
      "epoch": 6.656646728515625e-06,
      "step": 80980,
      "training_step_time": 8.166193723678589
    },
    {
      "epoch": 6.657600402832031e-06,
      "model_forward_time": 1.7512471675872803,
      "step": 80981
    },
    {
      "epoch": 6.657600402832031e-06,
      "step": 80981,
      "training_step_time": 8.16972279548645
    },
    {
      "epoch": 6.658554077148438e-06,
      "model_forward_time": 1.745194673538208,
      "step": 80982
    },
    {
      "epoch": 6.658554077148438e-06,
      "step": 80982,
      "training_step_time": 7.533344984054565
    },
    {
      "epoch": 6.659507751464843e-06,
      "model_forward_time": 1.747323989868164,
      "step": 80983
    },
    {
      "epoch": 6.659507751464843e-06,
      "step": 80983,
      "training_step_time": 7.5394463539123535
    },
    {
      "epoch": 6.66046142578125e-06,
      "model_forward_time": 1.7535035610198975,
      "step": 80984
    },
    {
      "epoch": 6.66046142578125e-06,
      "step": 80984,
      "training_step_time": 7.554821968078613
    },
    {
      "epoch": 6.6614151000976566e-06,
      "model_forward_time": 1.7521274089813232,
      "step": 80985
    },
    {
      "epoch": 6.6614151000976566e-06,
      "step": 80985,
      "training_step_time": 7.6351706981658936
    },
    {
      "epoch": 6.662368774414062e-06,
      "model_forward_time": 1.7454593181610107,
      "step": 80986
    },
    {
      "epoch": 6.662368774414062e-06,
      "step": 80986,
      "training_step_time": 8.511523485183716
    },
    {
      "epoch": 6.663322448730469e-06,
      "model_forward_time": 1.7476074695587158,
      "step": 80987
    },
    {
      "epoch": 6.663322448730469e-06,
      "step": 80987,
      "training_step_time": 8.728487968444824
    },
    {
      "epoch": 6.664276123046875e-06,
      "model_forward_time": 1.8460428714752197,
      "step": 80988
    },
    {
      "epoch": 6.664276123046875e-06,
      "step": 80988,
      "training_step_time": 10.291700839996338
    },
    {
      "epoch": 6.665229797363281e-06,
      "model_forward_time": 1.7427074909210205,
      "step": 80989
    },
    {
      "epoch": 6.665229797363281e-06,
      "step": 80989,
      "training_step_time": 7.653475284576416
    },
    {
      "epoch": 6.666183471679688e-06,
      "grad_norm": 0.08047696202993393,
      "learning_rate": 9.559843694234917e-07,
      "loss": 0.0553,
      "step": 80990
    },
    {
      "epoch": 6.666183471679688e-06,
      "model_forward_time": 2.3607566356658936,
      "step": 80990
    },
    {
      "dynamics_loss_avg": 0.0414616959169507,
      "epoch": 6.666183471679688e-06,
      "step": 80990
    },
    {
      "action_loss_avg": 0.0013191177771659569,
      "epoch": 6.666183471679688e-06,
      "step": 80990
    },
    {
      "epoch": 6.666183471679688e-06,
      "step": 80990,
      "training_step_time": 7.65541672706604
    },
    {
      "epoch": 6.6671371459960934e-06,
      "model_forward_time": 1.7507116794586182,
      "step": 80991
    },
    {
      "epoch": 6.6671371459960934e-06,
      "step": 80991,
      "training_step_time": 7.588668346405029
    },
    {
      "epoch": 6.6680908203125e-06,
      "model_forward_time": 1.7606945037841797,
      "step": 80992
    },
    {
      "epoch": 6.6680908203125e-06,
      "step": 80992,
      "training_step_time": 6.755884885787964
    },
    {
      "epoch": 6.6690444946289066e-06,
      "model_forward_time": 1.760910987854004,
      "step": 80993
    },
    {
      "epoch": 6.6690444946289066e-06,
      "step": 80993,
      "training_step_time": 7.5801544189453125
    },
    {
      "epoch": 6.669998168945312e-06,
      "model_forward_time": 1.761756181716919,
      "step": 80994
    },
    {
      "epoch": 6.669998168945312e-06,
      "step": 80994,
      "training_step_time": 7.6235339641571045
    },
    {
      "epoch": 6.670951843261719e-06,
      "model_forward_time": 1.7637653350830078,
      "step": 80995
    },
    {
      "epoch": 6.670951843261719e-06,
      "step": 80995,
      "training_step_time": 7.558459281921387
    },
    {
      "epoch": 6.671905517578125e-06,
      "model_forward_time": 1.7676780223846436,
      "step": 80996
    },
    {
      "epoch": 6.671905517578125e-06,
      "step": 80996,
      "training_step_time": 9.227839708328247
    },
    {
      "epoch": 6.672859191894531e-06,
      "model_forward_time": 1.835465431213379,
      "step": 80997
    },
    {
      "epoch": 6.672859191894531e-06,
      "step": 80997,
      "training_step_time": 10.931894063949585
    },
    {
      "epoch": 6.673812866210938e-06,
      "model_forward_time": 1.8190572261810303,
      "step": 80998
    },
    {
      "epoch": 6.673812866210938e-06,
      "step": 80998,
      "training_step_time": 7.99826455116272
    },
    {
      "epoch": 6.6747665405273434e-06,
      "model_forward_time": 2.428798198699951,
      "step": 80999
    },
    {
      "epoch": 6.6747665405273434e-06,
      "step": 80999,
      "training_step_time": 7.7445549964904785
    },
    {
      "epoch": 6.67572021484375e-06,
      "grad_norm": 0.06838000565767288,
      "learning_rate": 9.550122188544625e-07,
      "loss": 0.0553,
      "step": 81000
    },
    {
      "epoch": 6.67572021484375e-06,
      "model_forward_time": 1.8303184509277344,
      "step": 81000
    },
    {
      "dynamics_loss_avg": 0.05345008447766304,
      "epoch": 6.67572021484375e-06,
      "step": 81000
    },
    {
      "action_loss_avg": 0.0007621395168825984,
      "epoch": 6.67572021484375e-06,
      "step": 81000
    },
    {
      "epoch": 6.67572021484375e-06,
      "step": 81000,
      "training_step_time": 6.508072853088379
    },
    {
      "epoch": 6.6766738891601566e-06,
      "model_forward_time": 1.7550809383392334,
      "step": 81001
    },
    {
      "epoch": 6.6766738891601566e-06,
      "step": 81001,
      "training_step_time": 7.942431926727295
    },
    {
      "epoch": 6.677627563476562e-06,
      "model_forward_time": 1.76003098487854,
      "step": 81002
    },
    {
      "epoch": 6.677627563476562e-06,
      "step": 81002,
      "training_step_time": 7.505420923233032
    },
    {
      "epoch": 6.678581237792969e-06,
      "model_forward_time": 1.7591679096221924,
      "step": 81003
    },
    {
      "epoch": 6.678581237792969e-06,
      "step": 81003,
      "training_step_time": 7.594151496887207
    },
    {
      "epoch": 6.679534912109375e-06,
      "model_forward_time": 1.7598295211791992,
      "step": 81004
    },
    {
      "epoch": 6.679534912109375e-06,
      "step": 81004,
      "training_step_time": 7.445757150650024
    },
    {
      "epoch": 6.680488586425781e-06,
      "model_forward_time": 1.7587082386016846,
      "step": 81005
    },
    {
      "epoch": 6.680488586425781e-06,
      "step": 81005,
      "training_step_time": 10.809625387191772
    },
    {
      "epoch": 6.681442260742188e-06,
      "model_forward_time": 1.8227815628051758,
      "step": 81006
    },
    {
      "epoch": 6.681442260742188e-06,
      "step": 81006,
      "training_step_time": 9.651376724243164
    },
    {
      "epoch": 6.6823959350585934e-06,
      "model_forward_time": 3.0033109188079834,
      "step": 81007
    },
    {
      "epoch": 6.6823959350585934e-06,
      "step": 81007,
      "training_step_time": 7.680883169174194
    },
    {
      "epoch": 6.683349609375e-06,
      "model_forward_time": 1.756378412246704,
      "step": 81008
    },
    {
      "epoch": 6.683349609375e-06,
      "step": 81008,
      "training_step_time": 8.381415605545044
    },
    {
      "epoch": 6.6843032836914066e-06,
      "model_forward_time": 1.7574331760406494,
      "step": 81009
    },
    {
      "epoch": 6.6843032836914066e-06,
      "step": 81009,
      "training_step_time": 7.6638829708099365
    },
    {
      "epoch": 6.685256958007812e-06,
      "grad_norm": 0.09166109561920166,
      "learning_rate": 9.540405106392154e-07,
      "loss": 0.0548,
      "step": 81010
    },
    {
      "epoch": 6.685256958007812e-06,
      "model_forward_time": 1.3908417224884033,
      "step": 81010
    },
    {
      "dynamics_loss_avg": 0.04706033850088716,
      "epoch": 6.685256958007812e-06,
      "step": 81010
    },
    {
      "action_loss_avg": 0.0014153486583381892,
      "epoch": 6.685256958007812e-06,
      "step": 81010
    },
    {
      "epoch": 6.685256958007812e-06,
      "step": 81010,
      "training_step_time": 7.4342052936553955
    },
    {
      "epoch": 6.686210632324219e-06,
      "model_forward_time": 1.7482595443725586,
      "step": 81011
    },
    {
      "epoch": 6.686210632324219e-06,
      "step": 81011,
      "training_step_time": 7.537227392196655
    },
    {
      "epoch": 6.687164306640625e-06,
      "model_forward_time": 1.7546799182891846,
      "step": 81012
    },
    {
      "epoch": 6.687164306640625e-06,
      "step": 81012,
      "training_step_time": 9.02771544456482
    },
    {
      "epoch": 6.688117980957031e-06,
      "model_forward_time": 1.7600743770599365,
      "step": 81013
    },
    {
      "epoch": 6.688117980957031e-06,
      "step": 81013,
      "training_step_time": 8.218905448913574
    },
    {
      "epoch": 6.689071655273438e-06,
      "model_forward_time": 1.7842316627502441,
      "step": 81014
    },
    {
      "epoch": 6.689071655273438e-06,
      "step": 81014,
      "training_step_time": 8.415183305740356
    },
    {
      "epoch": 6.6900253295898434e-06,
      "model_forward_time": 1.7572567462921143,
      "step": 81015
    },
    {
      "epoch": 6.6900253295898434e-06,
      "step": 81015,
      "training_step_time": 9.410351037979126
    },
    {
      "epoch": 6.69097900390625e-06,
      "model_forward_time": 1.7550628185272217,
      "step": 81016
    },
    {
      "epoch": 6.69097900390625e-06,
      "step": 81016,
      "training_step_time": 7.617533445358276
    },
    {
      "epoch": 6.691932678222657e-06,
      "model_forward_time": 2.3703694343566895,
      "step": 81017
    },
    {
      "epoch": 6.691932678222657e-06,
      "step": 81017,
      "training_step_time": 8.680941581726074
    },
    {
      "epoch": 6.692886352539062e-06,
      "model_forward_time": 1.7547814846038818,
      "step": 81018
    },
    {
      "epoch": 6.692886352539062e-06,
      "step": 81018,
      "training_step_time": 7.602890253067017
    },
    {
      "epoch": 6.693840026855469e-06,
      "model_forward_time": 1.7565903663635254,
      "step": 81019
    },
    {
      "epoch": 6.693840026855469e-06,
      "step": 81019,
      "training_step_time": 7.4207751750946045
    },
    {
      "epoch": 6.694793701171875e-06,
      "grad_norm": 0.437485933303833,
      "learning_rate": 9.530692448840151e-07,
      "loss": 0.0551,
      "step": 81020
    },
    {
      "epoch": 6.694793701171875e-06,
      "model_forward_time": 1.7510085105895996,
      "step": 81020
    },
    {
      "dynamics_loss_avg": 0.05250319633632898,
      "epoch": 6.694793701171875e-06,
      "step": 81020
    },
    {
      "action_loss_avg": 0.0015061975078424438,
      "epoch": 6.694793701171875e-06,
      "step": 81020
    },
    {
      "epoch": 6.694793701171875e-06,
      "step": 81020,
      "training_step_time": 7.507773160934448
    },
    {
      "epoch": 6.695747375488281e-06,
      "model_forward_time": 1.7520086765289307,
      "step": 81021
    },
    {
      "epoch": 6.695747375488281e-06,
      "step": 81021,
      "training_step_time": 7.647002696990967
    },
    {
      "epoch": 6.696701049804688e-06,
      "model_forward_time": 1.759490966796875,
      "step": 81022
    },
    {
      "epoch": 6.696701049804688e-06,
      "step": 81022,
      "training_step_time": 7.63059139251709
    },
    {
      "epoch": 6.6976547241210935e-06,
      "model_forward_time": 1.7581548690795898,
      "step": 81023
    },
    {
      "epoch": 6.6976547241210935e-06,
      "step": 81023,
      "training_step_time": 9.31165599822998
    },
    {
      "epoch": 6.6986083984375e-06,
      "model_forward_time": 1.75144362449646,
      "step": 81024
    },
    {
      "epoch": 6.6986083984375e-06,
      "step": 81024,
      "training_step_time": 7.615753889083862
    },
    {
      "epoch": 6.699562072753907e-06,
      "model_forward_time": 1.8718500137329102,
      "step": 81025
    },
    {
      "epoch": 6.699562072753907e-06,
      "step": 81025,
      "training_step_time": 8.272622346878052
    },
    {
      "epoch": 6.700515747070312e-06,
      "model_forward_time": 2.3762147426605225,
      "step": 81026
    },
    {
      "epoch": 6.700515747070312e-06,
      "step": 81026,
      "training_step_time": 8.183450698852539
    },
    {
      "epoch": 6.701469421386719e-06,
      "model_forward_time": 1.75734281539917,
      "step": 81027
    },
    {
      "epoch": 6.701469421386719e-06,
      "step": 81027,
      "training_step_time": 8.115994930267334
    },
    {
      "epoch": 6.702423095703125e-06,
      "model_forward_time": 1.7504756450653076,
      "step": 81028
    },
    {
      "epoch": 6.702423095703125e-06,
      "step": 81028,
      "training_step_time": 7.584487676620483
    },
    {
      "epoch": 6.703376770019531e-06,
      "model_forward_time": 1.746546983718872,
      "step": 81029
    },
    {
      "epoch": 6.703376770019531e-06,
      "step": 81029,
      "training_step_time": 8.242807149887085
    },
    {
      "epoch": 6.704330444335938e-06,
      "grad_norm": 0.09967673569917679,
      "learning_rate": 9.52098421695079e-07,
      "loss": 0.0551,
      "step": 81030
    },
    {
      "epoch": 6.704330444335938e-06,
      "model_forward_time": 1.753455638885498,
      "step": 81030
    },
    {
      "dynamics_loss_avg": 0.0464170184917748,
      "epoch": 6.704330444335938e-06,
      "step": 81030
    },
    {
      "action_loss_avg": 0.001793875510338694,
      "epoch": 6.704330444335938e-06,
      "step": 81030
    },
    {
      "epoch": 6.704330444335938e-06,
      "step": 81030,
      "training_step_time": 8.045798778533936
    },
    {
      "epoch": 6.7052841186523435e-06,
      "model_forward_time": 1.7579305171966553,
      "step": 81031
    },
    {
      "epoch": 6.7052841186523435e-06,
      "step": 81031,
      "training_step_time": 7.6110124588012695
    },
    {
      "epoch": 6.70623779296875e-06,
      "model_forward_time": 1.75272798538208,
      "step": 81032
    },
    {
      "epoch": 6.70623779296875e-06,
      "step": 81032,
      "training_step_time": 7.6566383838653564
    },
    {
      "epoch": 6.707191467285157e-06,
      "model_forward_time": 1.7468225955963135,
      "step": 81033
    },
    {
      "epoch": 6.707191467285157e-06,
      "step": 81033,
      "training_step_time": 11.071749210357666
    },
    {
      "epoch": 6.708145141601562e-06,
      "model_forward_time": 1.8627104759216309,
      "step": 81034
    },
    {
      "epoch": 6.708145141601562e-06,
      "step": 81034,
      "training_step_time": 8.71350622177124
    },
    {
      "epoch": 6.709098815917969e-06,
      "model_forward_time": 2.4116857051849365,
      "step": 81035
    },
    {
      "epoch": 6.709098815917969e-06,
      "step": 81035,
      "training_step_time": 7.642704725265503
    },
    {
      "epoch": 6.710052490234375e-06,
      "model_forward_time": 1.7564187049865723,
      "step": 81036
    },
    {
      "epoch": 6.710052490234375e-06,
      "step": 81036,
      "training_step_time": 8.445377588272095
    },
    {
      "epoch": 6.711006164550781e-06,
      "model_forward_time": 1.753483533859253,
      "step": 81037
    },
    {
      "epoch": 6.711006164550781e-06,
      "step": 81037,
      "training_step_time": 7.535532236099243
    },
    {
      "epoch": 6.711959838867188e-06,
      "model_forward_time": 1.749281406402588,
      "step": 81038
    },
    {
      "epoch": 6.711959838867188e-06,
      "step": 81038,
      "training_step_time": 7.018451690673828
    },
    {
      "epoch": 6.7129135131835935e-06,
      "model_forward_time": 1.756166696548462,
      "step": 81039
    },
    {
      "epoch": 6.7129135131835935e-06,
      "step": 81039,
      "training_step_time": 7.537511825561523
    },
    {
      "epoch": 6.7138671875e-06,
      "grad_norm": 0.13763481378555298,
      "learning_rate": 9.511280411785728e-07,
      "loss": 0.0559,
      "step": 81040
    },
    {
      "epoch": 6.7138671875e-06,
      "model_forward_time": 1.7542452812194824,
      "step": 81040
    },
    {
      "dynamics_loss_avg": 0.047646537981927396,
      "epoch": 6.7138671875e-06,
      "step": 81040
    },
    {
      "action_loss_avg": 0.0009916649898514152,
      "epoch": 6.7138671875e-06,
      "step": 81040
    },
    {
      "epoch": 6.7138671875e-06,
      "step": 81040,
      "training_step_time": 7.553187131881714
    },
    {
      "epoch": 6.714820861816407e-06,
      "model_forward_time": 1.749136209487915,
      "step": 81041
    },
    {
      "epoch": 6.714820861816407e-06,
      "step": 81041,
      "training_step_time": 7.658830881118774
    },
    {
      "epoch": 6.715774536132812e-06,
      "model_forward_time": 1.760378122329712,
      "step": 81042
    },
    {
      "epoch": 6.715774536132812e-06,
      "step": 81042,
      "training_step_time": 8.990808248519897
    },
    {
      "epoch": 6.716728210449219e-06,
      "model_forward_time": 1.831831693649292,
      "step": 81043
    },
    {
      "epoch": 6.716728210449219e-06,
      "step": 81043,
      "training_step_time": 8.43982195854187
    },
    {
      "epoch": 6.717681884765625e-06,
      "model_forward_time": 2.4229178428649902,
      "step": 81044
    },
    {
      "epoch": 6.717681884765625e-06,
      "step": 81044,
      "training_step_time": 9.12100625038147
    },
    {
      "epoch": 6.718635559082031e-06,
      "model_forward_time": 1.7576985359191895,
      "step": 81045
    },
    {
      "epoch": 6.718635559082031e-06,
      "step": 81045,
      "training_step_time": 8.811788082122803
    },
    {
      "epoch": 6.719589233398438e-06,
      "model_forward_time": 1.759540319442749,
      "step": 81046
    },
    {
      "epoch": 6.719589233398438e-06,
      "step": 81046,
      "training_step_time": 7.647587299346924
    },
    {
      "epoch": 6.7205429077148435e-06,
      "model_forward_time": 1.7566733360290527,
      "step": 81047
    },
    {
      "epoch": 6.7205429077148435e-06,
      "step": 81047,
      "training_step_time": 7.44821834564209
    },
    {
      "epoch": 6.72149658203125e-06,
      "model_forward_time": 1.7532715797424316,
      "step": 81048
    },
    {
      "epoch": 6.72149658203125e-06,
      "step": 81048,
      "training_step_time": 7.52772331237793
    },
    {
      "epoch": 6.722450256347657e-06,
      "model_forward_time": 1.7487857341766357,
      "step": 81049
    },
    {
      "epoch": 6.722450256347657e-06,
      "step": 81049,
      "training_step_time": 7.554609298706055
    },
    {
      "epoch": 6.723403930664062e-06,
      "grad_norm": 0.0859859362244606,
      "learning_rate": 9.50158103440616e-07,
      "loss": 0.0547,
      "step": 81050
    },
    {
      "epoch": 6.723403930664062e-06,
      "model_forward_time": 1.7526845932006836,
      "step": 81050
    },
    {
      "dynamics_loss_avg": 0.06010578442364931,
      "epoch": 6.723403930664062e-06,
      "step": 81050
    },
    {
      "action_loss_avg": 0.0013130266161169857,
      "epoch": 6.723403930664062e-06,
      "step": 81050
    },
    {
      "epoch": 6.723403930664062e-06,
      "step": 81050,
      "training_step_time": 8.350692987442017
    },
    {
      "epoch": 6.724357604980469e-06,
      "model_forward_time": 1.7536640167236328,
      "step": 81051
    },
    {
      "epoch": 6.724357604980469e-06,
      "step": 81051,
      "training_step_time": 9.419501781463623
    },
    {
      "epoch": 6.725311279296875e-06,
      "model_forward_time": 1.8007819652557373,
      "step": 81052
    },
    {
      "epoch": 6.725311279296875e-06,
      "step": 81052,
      "training_step_time": 8.625510215759277
    },
    {
      "epoch": 6.726264953613281e-06,
      "model_forward_time": 1.756333589553833,
      "step": 81053
    },
    {
      "epoch": 6.726264953613281e-06,
      "step": 81053,
      "training_step_time": 8.929168939590454
    },
    {
      "epoch": 6.727218627929688e-06,
      "model_forward_time": 1.778402328491211,
      "step": 81054
    },
    {
      "epoch": 6.727218627929688e-06,
      "step": 81054,
      "training_step_time": 8.65304684638977
    },
    {
      "epoch": 6.7281723022460935e-06,
      "model_forward_time": 1.7632050514221191,
      "step": 81055
    },
    {
      "epoch": 6.7281723022460935e-06,
      "step": 81055,
      "training_step_time": 7.5869386196136475
    },
    {
      "epoch": 6.7291259765625e-06,
      "model_forward_time": 1.7573449611663818,
      "step": 81056
    },
    {
      "epoch": 6.7291259765625e-06,
      "step": 81056,
      "training_step_time": 7.417355298995972
    },
    {
      "epoch": 6.730079650878907e-06,
      "model_forward_time": 1.7592666149139404,
      "step": 81057
    },
    {
      "epoch": 6.730079650878907e-06,
      "step": 81057,
      "training_step_time": 7.499327659606934
    },
    {
      "epoch": 6.731033325195312e-06,
      "model_forward_time": 1.7580711841583252,
      "step": 81058
    },
    {
      "epoch": 6.731033325195312e-06,
      "step": 81058,
      "training_step_time": 8.039449691772461
    },
    {
      "epoch": 6.731986999511719e-06,
      "model_forward_time": 1.7597250938415527,
      "step": 81059
    },
    {
      "epoch": 6.731986999511719e-06,
      "step": 81059,
      "training_step_time": 7.549532413482666
    },
    {
      "epoch": 6.732940673828125e-06,
      "grad_norm": 0.15296772122383118,
      "learning_rate": 9.491886085872814e-07,
      "loss": 0.0542,
      "step": 81060
    },
    {
      "epoch": 6.732940673828125e-06,
      "model_forward_time": 1.7585294246673584,
      "step": 81060
    },
    {
      "dynamics_loss_avg": 0.060033674351871014,
      "epoch": 6.732940673828125e-06,
      "step": 81060
    },
    {
      "action_loss_avg": 0.001294134755153209,
      "epoch": 6.732940673828125e-06,
      "step": 81060
    },
    {
      "epoch": 6.732940673828125e-06,
      "step": 81060,
      "training_step_time": 8.575732231140137
    },
    {
      "epoch": 6.733894348144531e-06,
      "model_forward_time": 1.7723324298858643,
      "step": 81061
    },
    {
      "epoch": 6.733894348144531e-06,
      "step": 81061,
      "training_step_time": 9.009963989257812
    },
    {
      "epoch": 6.734848022460938e-06,
      "model_forward_time": 1.7556800842285156,
      "step": 81062
    },
    {
      "epoch": 6.734848022460938e-06,
      "step": 81062,
      "training_step_time": 7.61794376373291
    },
    {
      "epoch": 6.7358016967773435e-06,
      "model_forward_time": 2.3713791370391846,
      "step": 81063
    },
    {
      "epoch": 6.7358016967773435e-06,
      "step": 81063,
      "training_step_time": 8.261871099472046
    },
    {
      "epoch": 6.73675537109375e-06,
      "model_forward_time": 1.3874802589416504,
      "step": 81064
    },
    {
      "epoch": 6.73675537109375e-06,
      "step": 81064,
      "training_step_time": 8.37867283821106
    },
    {
      "epoch": 6.737709045410157e-06,
      "model_forward_time": 1.753058671951294,
      "step": 81065
    },
    {
      "epoch": 6.737709045410157e-06,
      "step": 81065,
      "training_step_time": 7.433600187301636
    },
    {
      "epoch": 6.738662719726562e-06,
      "model_forward_time": 1.7633659839630127,
      "step": 81066
    },
    {
      "epoch": 6.738662719726562e-06,
      "step": 81066,
      "training_step_time": 7.494746923446655
    },
    {
      "epoch": 6.739616394042969e-06,
      "model_forward_time": 1.7626631259918213,
      "step": 81067
    },
    {
      "epoch": 6.739616394042969e-06,
      "step": 81067,
      "training_step_time": 7.47765040397644
    },
    {
      "epoch": 6.740570068359375e-06,
      "model_forward_time": 1.754667043685913,
      "step": 81068
    },
    {
      "epoch": 6.740570068359375e-06,
      "step": 81068,
      "training_step_time": 7.611078262329102
    },
    {
      "epoch": 6.741523742675781e-06,
      "model_forward_time": 1.7583732604980469,
      "step": 81069
    },
    {
      "epoch": 6.741523742675781e-06,
      "step": 81069,
      "training_step_time": 8.148875713348389
    },
    {
      "epoch": 6.742477416992188e-06,
      "grad_norm": 0.09592849761247635,
      "learning_rate": 9.482195567245894e-07,
      "loss": 0.0555,
      "step": 81070
    },
    {
      "epoch": 6.742477416992188e-06,
      "model_forward_time": 1.7479770183563232,
      "step": 81070
    },
    {
      "dynamics_loss_avg": 0.06432962156832218,
      "epoch": 6.742477416992188e-06,
      "step": 81070
    },
    {
      "action_loss_avg": 0.001645803969586268,
      "epoch": 6.742477416992188e-06,
      "step": 81070
    },
    {
      "epoch": 6.742477416992188e-06,
      "step": 81070,
      "training_step_time": 9.759021282196045
    },
    {
      "epoch": 6.7434310913085935e-06,
      "model_forward_time": 1.7563865184783936,
      "step": 81071
    },
    {
      "epoch": 6.7434310913085935e-06,
      "step": 81071,
      "training_step_time": 7.60061502456665
    },
    {
      "epoch": 6.744384765625e-06,
      "model_forward_time": 2.484628677368164,
      "step": 81072
    },
    {
      "epoch": 6.744384765625e-06,
      "step": 81072,
      "training_step_time": 8.672365665435791
    },
    {
      "epoch": 6.745338439941407e-06,
      "model_forward_time": 1.7622582912445068,
      "step": 81073
    },
    {
      "epoch": 6.745338439941407e-06,
      "step": 81073,
      "training_step_time": 7.670799255371094
    },
    {
      "epoch": 6.746292114257812e-06,
      "model_forward_time": 1.7702696323394775,
      "step": 81074
    },
    {
      "epoch": 6.746292114257812e-06,
      "step": 81074,
      "training_step_time": 7.626889705657959
    },
    {
      "epoch": 6.747245788574219e-06,
      "model_forward_time": 1.7636871337890625,
      "step": 81075
    },
    {
      "epoch": 6.747245788574219e-06,
      "step": 81075,
      "training_step_time": 7.471405267715454
    },
    {
      "epoch": 6.748199462890625e-06,
      "model_forward_time": 1.7658987045288086,
      "step": 81076
    },
    {
      "epoch": 6.748199462890625e-06,
      "step": 81076,
      "training_step_time": 7.4944844245910645
    },
    {
      "epoch": 6.749153137207031e-06,
      "model_forward_time": 1.7607271671295166,
      "step": 81077
    },
    {
      "epoch": 6.749153137207031e-06,
      "step": 81077,
      "training_step_time": 7.602885484695435
    },
    {
      "epoch": 6.750106811523438e-06,
      "model_forward_time": 1.7591326236724854,
      "step": 81078
    },
    {
      "epoch": 6.750106811523438e-06,
      "step": 81078,
      "training_step_time": 8.495042324066162
    },
    {
      "epoch": 6.7510604858398435e-06,
      "model_forward_time": 1.7454109191894531,
      "step": 81079
    },
    {
      "epoch": 6.7510604858398435e-06,
      "step": 81079,
      "training_step_time": 9.315067291259766
    },
    {
      "epoch": 6.75201416015625e-06,
      "grad_norm": 0.08471303433179855,
      "learning_rate": 9.472509479585151e-07,
      "loss": 0.0561,
      "step": 81080
    },
    {
      "epoch": 6.75201416015625e-06,
      "model_forward_time": 1.8786146640777588,
      "step": 81080
    },
    {
      "dynamics_loss_avg": 0.05550457667559385,
      "epoch": 6.75201416015625e-06,
      "step": 81080
    },
    {
      "action_loss_avg": 0.0013545406283810735,
      "epoch": 6.75201416015625e-06,
      "step": 81080
    },
    {
      "epoch": 6.75201416015625e-06,
      "step": 81080,
      "training_step_time": 8.3280770778656
    },
    {
      "epoch": 6.752967834472657e-06,
      "model_forward_time": 2.4179446697235107,
      "step": 81081
    },
    {
      "epoch": 6.752967834472657e-06,
      "step": 81081,
      "training_step_time": 8.613779544830322
    },
    {
      "epoch": 6.753921508789062e-06,
      "model_forward_time": 1.7515320777893066,
      "step": 81082
    },
    {
      "epoch": 6.753921508789062e-06,
      "step": 81082,
      "training_step_time": 8.542338371276855
    },
    {
      "epoch": 6.754875183105469e-06,
      "model_forward_time": 1.7531721591949463,
      "step": 81083
    },
    {
      "epoch": 6.754875183105469e-06,
      "step": 81083,
      "training_step_time": 7.827483654022217
    },
    {
      "epoch": 6.755828857421875e-06,
      "model_forward_time": 1.7501928806304932,
      "step": 81084
    },
    {
      "epoch": 6.755828857421875e-06,
      "step": 81084,
      "training_step_time": 7.37919807434082
    },
    {
      "epoch": 6.756782531738281e-06,
      "model_forward_time": 1.7758474349975586,
      "step": 81085
    },
    {
      "epoch": 6.756782531738281e-06,
      "step": 81085,
      "training_step_time": 7.697964191436768
    },
    {
      "epoch": 6.757736206054688e-06,
      "model_forward_time": 1.7529866695404053,
      "step": 81086
    },
    {
      "epoch": 6.757736206054688e-06,
      "step": 81086,
      "training_step_time": 7.787316083908081
    },
    {
      "epoch": 6.7586898803710935e-06,
      "model_forward_time": 1.7554564476013184,
      "step": 81087
    },
    {
      "epoch": 6.7586898803710935e-06,
      "step": 81087,
      "training_step_time": 7.719292163848877
    },
    {
      "epoch": 6.7596435546875e-06,
      "model_forward_time": 1.3868868350982666,
      "step": 81088
    },
    {
      "epoch": 6.7596435546875e-06,
      "step": 81088,
      "training_step_time": 7.786717653274536
    },
    {
      "epoch": 6.760597229003907e-06,
      "model_forward_time": 1.8258309364318848,
      "step": 81089
    },
    {
      "epoch": 6.760597229003907e-06,
      "step": 81089,
      "training_step_time": 7.6542067527771
    },
    {
      "epoch": 6.761550903320312e-06,
      "grad_norm": 0.12367618083953857,
      "learning_rate": 9.462827823949833e-07,
      "loss": 0.0544,
      "step": 81090
    },
    {
      "epoch": 6.761550903320312e-06,
      "model_forward_time": 2.400714635848999,
      "step": 81090
    },
    {
      "dynamics_loss_avg": 0.0499614167958498,
      "epoch": 6.761550903320312e-06,
      "step": 81090
    },
    {
      "action_loss_avg": 0.0013234837853815407,
      "epoch": 6.761550903320312e-06,
      "step": 81090
    },
    {
      "epoch": 6.761550903320312e-06,
      "step": 81090,
      "training_step_time": 8.3547945022583
    },
    {
      "epoch": 6.762504577636719e-06,
      "model_forward_time": 1.7555482387542725,
      "step": 81091
    },
    {
      "epoch": 6.762504577636719e-06,
      "step": 81091,
      "training_step_time": 7.679727792739868
    },
    {
      "epoch": 6.763458251953125e-06,
      "model_forward_time": 1.384232521057129,
      "step": 81092
    },
    {
      "epoch": 6.763458251953125e-06,
      "step": 81092,
      "training_step_time": 7.703325271606445
    },
    {
      "epoch": 6.764411926269531e-06,
      "model_forward_time": 1.7556860446929932,
      "step": 81093
    },
    {
      "epoch": 6.764411926269531e-06,
      "step": 81093,
      "training_step_time": 7.225435018539429
    },
    {
      "epoch": 6.765365600585938e-06,
      "model_forward_time": 1.7539589405059814,
      "step": 81094
    },
    {
      "epoch": 6.765365600585938e-06,
      "step": 81094,
      "training_step_time": 7.562038898468018
    },
    {
      "epoch": 6.7663192749023435e-06,
      "model_forward_time": 1.7518010139465332,
      "step": 81095
    },
    {
      "epoch": 6.7663192749023435e-06,
      "step": 81095,
      "training_step_time": 7.665929794311523
    },
    {
      "epoch": 6.76727294921875e-06,
      "model_forward_time": 1.750279188156128,
      "step": 81096
    },
    {
      "epoch": 6.76727294921875e-06,
      "step": 81096,
      "training_step_time": 7.601610422134399
    },
    {
      "epoch": 6.768226623535157e-06,
      "model_forward_time": 1.7604906558990479,
      "step": 81097
    },
    {
      "epoch": 6.768226623535157e-06,
      "step": 81097,
      "training_step_time": 7.782665729522705
    },
    {
      "epoch": 6.769180297851562e-06,
      "model_forward_time": 1.7933177947998047,
      "step": 81098
    },
    {
      "epoch": 6.769180297851562e-06,
      "step": 81098,
      "training_step_time": 8.423421621322632
    },
    {
      "epoch": 6.770133972167969e-06,
      "model_forward_time": 1.7597134113311768,
      "step": 81099
    },
    {
      "epoch": 6.770133972167969e-06,
      "step": 81099,
      "training_step_time": 7.872273206710815
    },
    {
      "epoch": 6.771087646484375e-06,
      "grad_norm": 0.07516337931156158,
      "learning_rate": 9.453150601398736e-07,
      "loss": 0.0548,
      "step": 81100
    },
    {
      "epoch": 6.771087646484375e-06,
      "model_forward_time": 2.3657729625701904,
      "step": 81100
    },
    {
      "dynamics_loss_avg": 0.06302131293341517,
      "epoch": 6.771087646484375e-06,
      "step": 81100
    },
    {
      "action_loss_avg": 0.0013863500324077904,
      "epoch": 6.771087646484375e-06,
      "step": 81100
    },
    {
      "epoch": 6.771087646484375e-06,
      "step": 81100,
      "training_step_time": 7.777393579483032
    },
    {
      "epoch": 6.772041320800781e-06,
      "model_forward_time": 1.7566404342651367,
      "step": 81101
    },
    {
      "epoch": 6.772041320800781e-06,
      "step": 81101,
      "training_step_time": 8.715524673461914
    },
    {
      "epoch": 6.772994995117188e-06,
      "model_forward_time": 1.7779972553253174,
      "step": 81102
    },
    {
      "epoch": 6.772994995117188e-06,
      "step": 81102,
      "training_step_time": 7.875345945358276
    },
    {
      "epoch": 6.7739486694335935e-06,
      "model_forward_time": 1.7364730834960938,
      "step": 81103
    },
    {
      "epoch": 6.7739486694335935e-06,
      "step": 81103,
      "training_step_time": 7.706817626953125
    },
    {
      "epoch": 6.77490234375e-06,
      "model_forward_time": 1.7409706115722656,
      "step": 81104
    },
    {
      "epoch": 6.77490234375e-06,
      "step": 81104,
      "training_step_time": 7.9104225635528564
    },
    {
      "epoch": 6.775856018066407e-06,
      "model_forward_time": 1.7471692562103271,
      "step": 81105
    },
    {
      "epoch": 6.775856018066407e-06,
      "step": 81105,
      "training_step_time": 7.751205205917358
    },
    {
      "epoch": 6.776809692382812e-06,
      "model_forward_time": 1.749765157699585,
      "step": 81106
    },
    {
      "epoch": 6.776809692382812e-06,
      "step": 81106,
      "training_step_time": 7.86456823348999
    },
    {
      "epoch": 6.777763366699219e-06,
      "model_forward_time": 1.7511992454528809,
      "step": 81107
    },
    {
      "epoch": 6.777763366699219e-06,
      "step": 81107,
      "training_step_time": 8.852619647979736
    },
    {
      "epoch": 6.778717041015625e-06,
      "model_forward_time": 1.865250825881958,
      "step": 81108
    },
    {
      "epoch": 6.778717041015625e-06,
      "step": 81108,
      "training_step_time": 7.8698272705078125
    },
    {
      "epoch": 6.779670715332031e-06,
      "model_forward_time": 2.379075050354004,
      "step": 81109
    },
    {
      "epoch": 6.779670715332031e-06,
      "step": 81109,
      "training_step_time": 8.322513103485107
    },
    {
      "epoch": 6.780624389648438e-06,
      "grad_norm": 0.05942188575863838,
      "learning_rate": 9.443477812990109e-07,
      "loss": 0.0555,
      "step": 81110
    },
    {
      "epoch": 6.780624389648438e-06,
      "model_forward_time": 1.7326304912567139,
      "step": 81110
    },
    {
      "dynamics_loss_avg": 0.06048924345523119,
      "epoch": 6.780624389648438e-06,
      "step": 81110
    },
    {
      "action_loss_avg": 0.0014406687289010734,
      "epoch": 6.780624389648438e-06,
      "step": 81110
    },
    {
      "epoch": 6.780624389648438e-06,
      "step": 81110,
      "training_step_time": 7.830364465713501
    },
    {
      "epoch": 6.7815780639648435e-06,
      "model_forward_time": 1.7355725765228271,
      "step": 81111
    },
    {
      "epoch": 6.7815780639648435e-06,
      "step": 81111,
      "training_step_time": 7.816050052642822
    },
    {
      "epoch": 6.78253173828125e-06,
      "model_forward_time": 1.7502021789550781,
      "step": 81112
    },
    {
      "epoch": 6.78253173828125e-06,
      "step": 81112,
      "training_step_time": 7.336148500442505
    },
    {
      "epoch": 6.783485412597657e-06,
      "model_forward_time": 1.379467487335205,
      "step": 81113
    },
    {
      "epoch": 6.783485412597657e-06,
      "step": 81113,
      "training_step_time": 7.691027879714966
    },
    {
      "epoch": 6.784439086914062e-06,
      "model_forward_time": 1.7668211460113525,
      "step": 81114
    },
    {
      "epoch": 6.784439086914062e-06,
      "step": 81114,
      "training_step_time": 7.817499160766602
    },
    {
      "epoch": 6.785392761230469e-06,
      "model_forward_time": 1.7443885803222656,
      "step": 81115
    },
    {
      "epoch": 6.785392761230469e-06,
      "step": 81115,
      "training_step_time": 8.373789072036743
    },
    {
      "epoch": 6.786346435546875e-06,
      "model_forward_time": 1.7495510578155518,
      "step": 81116
    },
    {
      "epoch": 6.786346435546875e-06,
      "step": 81116,
      "training_step_time": 9.273716926574707
    },
    {
      "epoch": 6.787300109863281e-06,
      "model_forward_time": 1.8690159320831299,
      "step": 81117
    },
    {
      "epoch": 6.787300109863281e-06,
      "step": 81117,
      "training_step_time": 7.9373674392700195
    },
    {
      "epoch": 6.788253784179688e-06,
      "model_forward_time": 2.40339994430542,
      "step": 81118
    },
    {
      "epoch": 6.788253784179688e-06,
      "step": 81118,
      "training_step_time": 8.68981122970581
    },
    {
      "epoch": 6.7892074584960935e-06,
      "model_forward_time": 1.7486507892608643,
      "step": 81119
    },
    {
      "epoch": 6.7892074584960935e-06,
      "step": 81119,
      "training_step_time": 7.8488664627075195
    },
    {
      "epoch": 6.7901611328125e-06,
      "grad_norm": 0.07542449980974197,
      "learning_rate": 9.43380945978179e-07,
      "loss": 0.0567,
      "step": 81120
    },
    {
      "epoch": 6.7901611328125e-06,
      "model_forward_time": 1.7435662746429443,
      "step": 81120
    },
    {
      "dynamics_loss_avg": 0.045911460928618905,
      "epoch": 6.7901611328125e-06,
      "step": 81120
    },
    {
      "action_loss_avg": 0.0017227371732587925,
      "epoch": 6.7901611328125e-06,
      "step": 81120
    },
    {
      "epoch": 6.7901611328125e-06,
      "step": 81120,
      "training_step_time": 7.806098222732544
    },
    {
      "epoch": 6.791114807128907e-06,
      "model_forward_time": 1.7491698265075684,
      "step": 81121
    },
    {
      "epoch": 6.791114807128907e-06,
      "step": 81121,
      "training_step_time": 7.361889362335205
    },
    {
      "epoch": 6.792068481445312e-06,
      "model_forward_time": 1.7568917274475098,
      "step": 81122
    },
    {
      "epoch": 6.792068481445312e-06,
      "step": 81122,
      "training_step_time": 7.703110456466675
    },
    {
      "epoch": 6.793022155761719e-06,
      "model_forward_time": 1.7505574226379395,
      "step": 81123
    },
    {
      "epoch": 6.793022155761719e-06,
      "step": 81123,
      "training_step_time": 7.802657842636108
    },
    {
      "epoch": 6.793975830078125e-06,
      "model_forward_time": 1.747511863708496,
      "step": 81124
    },
    {
      "epoch": 6.793975830078125e-06,
      "step": 81124,
      "training_step_time": 7.80260968208313
    },
    {
      "epoch": 6.794929504394531e-06,
      "model_forward_time": 1.7454121112823486,
      "step": 81125
    },
    {
      "epoch": 6.794929504394531e-06,
      "step": 81125,
      "training_step_time": 10.779780387878418
    },
    {
      "epoch": 6.795883178710938e-06,
      "model_forward_time": 1.8391242027282715,
      "step": 81126
    },
    {
      "epoch": 6.795883178710938e-06,
      "step": 81126,
      "training_step_time": 9.327965259552002
    },
    {
      "epoch": 6.7968368530273435e-06,
      "model_forward_time": 2.3777108192443848,
      "step": 81127
    },
    {
      "epoch": 6.7968368530273435e-06,
      "step": 81127,
      "training_step_time": 7.696906089782715
    },
    {
      "epoch": 6.79779052734375e-06,
      "model_forward_time": 1.7364370822906494,
      "step": 81128
    },
    {
      "epoch": 6.79779052734375e-06,
      "step": 81128,
      "training_step_time": 7.698166608810425
    },
    {
      "epoch": 6.798744201660156e-06,
      "model_forward_time": 1.7491645812988281,
      "step": 81129
    },
    {
      "epoch": 6.798744201660156e-06,
      "step": 81129,
      "training_step_time": 7.672723770141602
    },
    {
      "epoch": 6.799697875976562e-06,
      "grad_norm": 0.08767000585794449,
      "learning_rate": 9.424145542831048e-07,
      "loss": 0.0564,
      "step": 81130
    },
    {
      "epoch": 6.799697875976562e-06,
      "model_forward_time": 1.7433276176452637,
      "step": 81130
    },
    {
      "dynamics_loss_avg": 0.06274758456274868,
      "epoch": 6.799697875976562e-06,
      "step": 81130
    },
    {
      "action_loss_avg": 0.0016016950248740613,
      "epoch": 6.799697875976562e-06,
      "step": 81130
    },
    {
      "epoch": 6.799697875976562e-06,
      "step": 81130,
      "training_step_time": 7.343263387680054
    },
    {
      "epoch": 6.800651550292969e-06,
      "model_forward_time": 1.743156909942627,
      "step": 81131
    },
    {
      "epoch": 6.800651550292969e-06,
      "step": 81131,
      "training_step_time": 7.516557216644287
    },
    {
      "epoch": 6.801605224609375e-06,
      "model_forward_time": 1.7364342212677002,
      "step": 81132
    },
    {
      "epoch": 6.801605224609375e-06,
      "step": 81132,
      "training_step_time": 7.787834644317627
    },
    {
      "epoch": 6.802558898925781e-06,
      "model_forward_time": 1.7401487827301025,
      "step": 81133
    },
    {
      "epoch": 6.802558898925781e-06,
      "step": 81133,
      "training_step_time": 7.756367921829224
    },
    {
      "epoch": 6.803512573242188e-06,
      "model_forward_time": 1.7371728420257568,
      "step": 81134
    },
    {
      "epoch": 6.803512573242188e-06,
      "step": 81134,
      "training_step_time": 8.864175081253052
    },
    {
      "epoch": 6.8044662475585935e-06,
      "model_forward_time": 1.7972474098205566,
      "step": 81135
    },
    {
      "epoch": 6.8044662475585935e-06,
      "step": 81135,
      "training_step_time": 10.898749828338623
    },
    {
      "epoch": 6.805419921875e-06,
      "model_forward_time": 1.7407774925231934,
      "step": 81136
    },
    {
      "epoch": 6.805419921875e-06,
      "step": 81136,
      "training_step_time": 8.305353879928589
    },
    {
      "epoch": 6.806373596191406e-06,
      "model_forward_time": 1.3723444938659668,
      "step": 81137
    },
    {
      "epoch": 6.806373596191406e-06,
      "step": 81137,
      "training_step_time": 8.611153841018677
    },
    {
      "epoch": 6.807327270507812e-06,
      "model_forward_time": 1.7472636699676514,
      "step": 81138
    },
    {
      "epoch": 6.807327270507812e-06,
      "step": 81138,
      "training_step_time": 8.467129230499268
    },
    {
      "epoch": 6.808280944824219e-06,
      "model_forward_time": 1.7469310760498047,
      "step": 81139
    },
    {
      "epoch": 6.808280944824219e-06,
      "step": 81139,
      "training_step_time": 7.4490132331848145
    },
    {
      "epoch": 6.809234619140625e-06,
      "grad_norm": 0.09295009821653366,
      "learning_rate": 9.414486063194778e-07,
      "loss": 0.0561,
      "step": 81140
    },
    {
      "epoch": 6.809234619140625e-06,
      "model_forward_time": 1.7395281791687012,
      "step": 81140
    },
    {
      "dynamics_loss_avg": 0.043084566667675975,
      "epoch": 6.809234619140625e-06,
      "step": 81140
    },
    {
      "action_loss_avg": 0.0013284020416904241,
      "epoch": 6.809234619140625e-06,
      "step": 81140
    },
    {
      "epoch": 6.809234619140625e-06,
      "step": 81140,
      "training_step_time": 8.532551288604736
    },
    {
      "epoch": 6.810188293457031e-06,
      "model_forward_time": 1.7560193538665771,
      "step": 81141
    },
    {
      "epoch": 6.810188293457031e-06,
      "step": 81141,
      "training_step_time": 7.830188989639282
    },
    {
      "epoch": 6.811141967773438e-06,
      "model_forward_time": 1.7506721019744873,
      "step": 81142
    },
    {
      "epoch": 6.811141967773438e-06,
      "step": 81142,
      "training_step_time": 8.752288579940796
    },
    {
      "epoch": 6.8120956420898436e-06,
      "model_forward_time": 1.7513651847839355,
      "step": 81143
    },
    {
      "epoch": 6.8120956420898436e-06,
      "step": 81143,
      "training_step_time": 9.92582082748413
    },
    {
      "epoch": 6.81304931640625e-06,
      "model_forward_time": 1.7909364700317383,
      "step": 81144
    },
    {
      "epoch": 6.81304931640625e-06,
      "step": 81144,
      "training_step_time": 9.140686988830566
    },
    {
      "epoch": 6.814002990722656e-06,
      "model_forward_time": 1.7471911907196045,
      "step": 81145
    },
    {
      "epoch": 6.814002990722656e-06,
      "step": 81145,
      "training_step_time": 7.925154209136963
    },
    {
      "epoch": 6.814956665039062e-06,
      "model_forward_time": 2.3532562255859375,
      "step": 81146
    },
    {
      "epoch": 6.814956665039062e-06,
      "step": 81146,
      "training_step_time": 7.77940034866333
    },
    {
      "epoch": 6.815910339355469e-06,
      "model_forward_time": 1.754453420639038,
      "step": 81147
    },
    {
      "epoch": 6.815910339355469e-06,
      "step": 81147,
      "training_step_time": 8.651599407196045
    },
    {
      "epoch": 6.816864013671875e-06,
      "model_forward_time": 1.758056879043579,
      "step": 81148
    },
    {
      "epoch": 6.816864013671875e-06,
      "step": 81148,
      "training_step_time": 7.750198125839233
    },
    {
      "epoch": 6.817817687988281e-06,
      "model_forward_time": 1.7572052478790283,
      "step": 81149
    },
    {
      "epoch": 6.817817687988281e-06,
      "step": 81149,
      "training_step_time": 7.505016565322876
    },
    {
      "epoch": 6.818771362304688e-06,
      "grad_norm": 0.07222339510917664,
      "learning_rate": 9.404831021929273e-07,
      "loss": 0.0563,
      "step": 81150
    },
    {
      "epoch": 6.818771362304688e-06,
      "model_forward_time": 1.7430496215820312,
      "step": 81150
    },
    {
      "dynamics_loss_avg": 0.06499361135065555,
      "epoch": 6.818771362304688e-06,
      "step": 81150
    },
    {
      "action_loss_avg": 0.00114281015121378,
      "epoch": 6.818771362304688e-06,
      "step": 81150
    },
    {
      "epoch": 6.818771362304688e-06,
      "step": 81150,
      "training_step_time": 7.707077741622925
    },
    {
      "epoch": 6.8197250366210936e-06,
      "model_forward_time": 1.7440435886383057,
      "step": 81151
    },
    {
      "epoch": 6.8197250366210936e-06,
      "step": 81151,
      "training_step_time": 8.226022720336914
    },
    {
      "epoch": 6.8206787109375e-06,
      "model_forward_time": 1.7591547966003418,
      "step": 81152
    },
    {
      "epoch": 6.8206787109375e-06,
      "step": 81152,
      "training_step_time": 10.23641324043274
    },
    {
      "epoch": 6.821632385253906e-06,
      "model_forward_time": 1.7564477920532227,
      "step": 81153
    },
    {
      "epoch": 6.821632385253906e-06,
      "step": 81153,
      "training_step_time": 9.073501348495483
    },
    {
      "epoch": 6.822586059570312e-06,
      "model_forward_time": 1.7606046199798584,
      "step": 81154
    },
    {
      "epoch": 6.822586059570312e-06,
      "step": 81154,
      "training_step_time": 7.9291863441467285
    },
    {
      "epoch": 6.823539733886719e-06,
      "model_forward_time": 2.343590497970581,
      "step": 81155
    },
    {
      "epoch": 6.823539733886719e-06,
      "step": 81155,
      "training_step_time": 7.838235855102539
    },
    {
      "epoch": 6.824493408203125e-06,
      "model_forward_time": 1.7547383308410645,
      "step": 81156
    },
    {
      "epoch": 6.824493408203125e-06,
      "step": 81156,
      "training_step_time": 9.007466077804565
    },
    {
      "epoch": 6.825447082519531e-06,
      "model_forward_time": 1.837615966796875,
      "step": 81157
    },
    {
      "epoch": 6.825447082519531e-06,
      "step": 81157,
      "training_step_time": 7.905583381652832
    },
    {
      "epoch": 6.826400756835938e-06,
      "model_forward_time": 1.7391993999481201,
      "step": 81158
    },
    {
      "epoch": 6.826400756835938e-06,
      "step": 81158,
      "training_step_time": 7.452667713165283
    },
    {
      "epoch": 6.8273544311523436e-06,
      "model_forward_time": 1.7490322589874268,
      "step": 81159
    },
    {
      "epoch": 6.8273544311523436e-06,
      "step": 81159,
      "training_step_time": 8.710634231567383
    },
    {
      "epoch": 6.82830810546875e-06,
      "grad_norm": 0.10703093558549881,
      "learning_rate": 9.395180420090427e-07,
      "loss": 0.0573,
      "step": 81160
    },
    {
      "epoch": 6.82830810546875e-06,
      "model_forward_time": 1.7524495124816895,
      "step": 81160
    },
    {
      "dynamics_loss_avg": 0.06457220781594515,
      "epoch": 6.82830810546875e-06,
      "step": 81160
    },
    {
      "action_loss_avg": 0.001238805751199834,
      "epoch": 6.82830810546875e-06,
      "step": 81160
    },
    {
      "epoch": 6.82830810546875e-06,
      "step": 81160,
      "training_step_time": 7.747644901275635
    },
    {
      "epoch": 6.829261779785156e-06,
      "model_forward_time": 1.7679362297058105,
      "step": 81161
    },
    {
      "epoch": 6.829261779785156e-06,
      "step": 81161,
      "training_step_time": 10.369155883789062
    },
    {
      "epoch": 6.8302154541015624e-06,
      "model_forward_time": 1.7503349781036377,
      "step": 81162
    },
    {
      "epoch": 6.8302154541015624e-06,
      "step": 81162,
      "training_step_time": 9.332173109054565
    },
    {
      "epoch": 6.831169128417969e-06,
      "model_forward_time": 1.7865571975708008,
      "step": 81163
    },
    {
      "epoch": 6.831169128417969e-06,
      "step": 81163,
      "training_step_time": 7.855413436889648
    },
    {
      "epoch": 6.832122802734375e-06,
      "model_forward_time": 2.3857316970825195,
      "step": 81164
    },
    {
      "epoch": 6.832122802734375e-06,
      "step": 81164,
      "training_step_time": 8.454744815826416
    },
    {
      "epoch": 6.833076477050781e-06,
      "model_forward_time": 1.7485835552215576,
      "step": 81165
    },
    {
      "epoch": 6.833076477050781e-06,
      "step": 81165,
      "training_step_time": 7.941438674926758
    },
    {
      "epoch": 6.834030151367188e-06,
      "model_forward_time": 1.7604315280914307,
      "step": 81166
    },
    {
      "epoch": 6.834030151367188e-06,
      "step": 81166,
      "training_step_time": 7.659494638442993
    },
    {
      "epoch": 6.8349838256835936e-06,
      "model_forward_time": 1.3835945129394531,
      "step": 81167
    },
    {
      "epoch": 6.8349838256835936e-06,
      "step": 81167,
      "training_step_time": 7.483211517333984
    },
    {
      "epoch": 6.8359375e-06,
      "model_forward_time": 1.7571418285369873,
      "step": 81168
    },
    {
      "epoch": 6.8359375e-06,
      "step": 81168,
      "training_step_time": 7.7257421016693115
    },
    {
      "epoch": 6.836891174316406e-06,
      "model_forward_time": 1.7474088668823242,
      "step": 81169
    },
    {
      "epoch": 6.836891174316406e-06,
      "step": 81169,
      "training_step_time": 7.7044172286987305
    },
    {
      "epoch": 6.8378448486328124e-06,
      "grad_norm": 0.08863082528114319,
      "learning_rate": 9.385534258733592e-07,
      "loss": 0.057,
      "step": 81170
    },
    {
      "epoch": 6.8378448486328124e-06,
      "model_forward_time": 1.7447474002838135,
      "step": 81170
    },
    {
      "dynamics_loss_avg": 0.05463625770062208,
      "epoch": 6.8378448486328124e-06,
      "step": 81170
    },
    {
      "action_loss_avg": 0.0016863348835613578,
      "epoch": 6.8378448486328124e-06,
      "step": 81170
    },
    {
      "epoch": 6.8378448486328124e-06,
      "step": 81170,
      "training_step_time": 9.339420795440674
    },
    {
      "epoch": 6.838798522949219e-06,
      "model_forward_time": 1.7603096961975098,
      "step": 81171
    },
    {
      "epoch": 6.838798522949219e-06,
      "step": 81171,
      "training_step_time": 9.374310493469238
    },
    {
      "epoch": 6.839752197265625e-06,
      "model_forward_time": 1.7502691745758057,
      "step": 81172
    },
    {
      "epoch": 6.839752197265625e-06,
      "step": 81172,
      "training_step_time": 7.8070127964019775
    },
    {
      "epoch": 6.840705871582031e-06,
      "model_forward_time": 2.3971123695373535,
      "step": 81173
    },
    {
      "epoch": 6.840705871582031e-06,
      "step": 81173,
      "training_step_time": 8.490176677703857
    },
    {
      "epoch": 6.841659545898438e-06,
      "model_forward_time": 1.748530626296997,
      "step": 81174
    },
    {
      "epoch": 6.841659545898438e-06,
      "step": 81174,
      "training_step_time": 8.562692880630493
    },
    {
      "epoch": 6.842613220214844e-06,
      "model_forward_time": 1.7522525787353516,
      "step": 81175
    },
    {
      "epoch": 6.842613220214844e-06,
      "step": 81175,
      "training_step_time": 7.855804681777954
    },
    {
      "epoch": 6.84356689453125e-06,
      "model_forward_time": 1.7459807395935059,
      "step": 81176
    },
    {
      "epoch": 6.84356689453125e-06,
      "step": 81176,
      "training_step_time": 7.442612409591675
    },
    {
      "epoch": 6.844520568847656e-06,
      "model_forward_time": 1.7535014152526855,
      "step": 81177
    },
    {
      "epoch": 6.844520568847656e-06,
      "step": 81177,
      "training_step_time": 7.5219457149505615
    },
    {
      "epoch": 6.8454742431640624e-06,
      "model_forward_time": 1.7513880729675293,
      "step": 81178
    },
    {
      "epoch": 6.8454742431640624e-06,
      "step": 81178,
      "training_step_time": 8.547898530960083
    },
    {
      "epoch": 6.846427917480469e-06,
      "model_forward_time": 1.7525012493133545,
      "step": 81179
    },
    {
      "epoch": 6.846427917480469e-06,
      "step": 81179,
      "training_step_time": 8.563400268554688
    },
    {
      "epoch": 6.847381591796875e-06,
      "grad_norm": 0.08773355185985565,
      "learning_rate": 9.375892538913678e-07,
      "loss": 0.0556,
      "step": 81180
    },
    {
      "epoch": 6.847381591796875e-06,
      "model_forward_time": 1.7423524856567383,
      "step": 81180
    },
    {
      "dynamics_loss_avg": 0.050672586634755136,
      "epoch": 6.847381591796875e-06,
      "step": 81180
    },
    {
      "action_loss_avg": 0.0013866293884348124,
      "epoch": 6.847381591796875e-06,
      "step": 81180
    },
    {
      "epoch": 6.847381591796875e-06,
      "step": 81180,
      "training_step_time": 8.827311038970947
    },
    {
      "epoch": 6.848335266113281e-06,
      "model_forward_time": 1.754209280014038,
      "step": 81181
    },
    {
      "epoch": 6.848335266113281e-06,
      "step": 81181,
      "training_step_time": 9.646470785140991
    },
    {
      "epoch": 6.849288940429688e-06,
      "model_forward_time": 1.8631165027618408,
      "step": 81182
    },
    {
      "epoch": 6.849288940429688e-06,
      "step": 81182,
      "training_step_time": 8.456714868545532
    },
    {
      "epoch": 6.850242614746094e-06,
      "model_forward_time": 1.7489631175994873,
      "step": 81183
    },
    {
      "epoch": 6.850242614746094e-06,
      "step": 81183,
      "training_step_time": 8.951175689697266
    },
    {
      "epoch": 6.8511962890625e-06,
      "model_forward_time": 1.7555737495422363,
      "step": 81184
    },
    {
      "epoch": 6.8511962890625e-06,
      "step": 81184,
      "training_step_time": 7.891995191574097
    },
    {
      "epoch": 6.852149963378906e-06,
      "model_forward_time": 1.7493202686309814,
      "step": 81185
    },
    {
      "epoch": 6.852149963378906e-06,
      "step": 81185,
      "training_step_time": 8.267614603042603
    },
    {
      "epoch": 6.8531036376953124e-06,
      "model_forward_time": 1.7514986991882324,
      "step": 81186
    },
    {
      "epoch": 6.8531036376953124e-06,
      "step": 81186,
      "training_step_time": 7.767122507095337
    },
    {
      "epoch": 6.854057312011719e-06,
      "model_forward_time": 1.749680995941162,
      "step": 81187
    },
    {
      "epoch": 6.854057312011719e-06,
      "step": 81187,
      "training_step_time": 7.8512492179870605
    },
    {
      "epoch": 6.855010986328125e-06,
      "model_forward_time": 1.7503347396850586,
      "step": 81188
    },
    {
      "epoch": 6.855010986328125e-06,
      "step": 81188,
      "training_step_time": 8.674715518951416
    },
    {
      "epoch": 6.855964660644531e-06,
      "model_forward_time": 1.7504911422729492,
      "step": 81189
    },
    {
      "epoch": 6.855964660644531e-06,
      "step": 81189,
      "training_step_time": 7.885144472122192
    },
    {
      "epoch": 6.856918334960938e-06,
      "grad_norm": 0.07547704130411148,
      "learning_rate": 9.36625526168507e-07,
      "loss": 0.0572,
      "step": 81190
    },
    {
      "epoch": 6.856918334960938e-06,
      "model_forward_time": 1.7401256561279297,
      "step": 81190
    },
    {
      "dynamics_loss_avg": 0.0614103052765131,
      "epoch": 6.856918334960938e-06,
      "step": 81190
    },
    {
      "action_loss_avg": 0.002118210564367473,
      "epoch": 6.856918334960938e-06,
      "step": 81190
    },
    {
      "epoch": 6.856918334960938e-06,
      "step": 81190,
      "training_step_time": 8.8333899974823
    },
    {
      "epoch": 6.857872009277344e-06,
      "model_forward_time": 1.8511743545532227,
      "step": 81191
    },
    {
      "epoch": 6.857872009277344e-06,
      "step": 81191,
      "training_step_time": 7.715948581695557
    },
    {
      "epoch": 6.85882568359375e-06,
      "model_forward_time": 2.4445419311523438,
      "step": 81192
    },
    {
      "epoch": 6.85882568359375e-06,
      "step": 81192,
      "training_step_time": 8.351133108139038
    },
    {
      "epoch": 6.859779357910156e-06,
      "model_forward_time": 1.743913173675537,
      "step": 81193
    },
    {
      "epoch": 6.859779357910156e-06,
      "step": 81193,
      "training_step_time": 7.755731105804443
    },
    {
      "epoch": 6.8607330322265625e-06,
      "model_forward_time": 1.7547173500061035,
      "step": 81194
    },
    {
      "epoch": 6.8607330322265625e-06,
      "step": 81194,
      "training_step_time": 7.80729079246521
    },
    {
      "epoch": 6.861686706542969e-06,
      "model_forward_time": 1.7491106986999512,
      "step": 81195
    },
    {
      "epoch": 6.861686706542969e-06,
      "step": 81195,
      "training_step_time": 7.426349639892578
    },
    {
      "epoch": 6.862640380859375e-06,
      "model_forward_time": 1.7548823356628418,
      "step": 81196
    },
    {
      "epoch": 6.862640380859375e-06,
      "step": 81196,
      "training_step_time": 8.780782461166382
    },
    {
      "epoch": 6.863594055175781e-06,
      "model_forward_time": 1.7572338581085205,
      "step": 81197
    },
    {
      "epoch": 6.863594055175781e-06,
      "step": 81197,
      "training_step_time": 7.768366098403931
    },
    {
      "epoch": 6.864547729492188e-06,
      "model_forward_time": 1.7497024536132812,
      "step": 81198
    },
    {
      "epoch": 6.864547729492188e-06,
      "step": 81198,
      "training_step_time": 8.487314939498901
    },
    {
      "epoch": 6.865501403808594e-06,
      "model_forward_time": 1.7310686111450195,
      "step": 81199
    },
    {
      "epoch": 6.865501403808594e-06,
      "step": 81199,
      "training_step_time": 8.38144302368164
    },
    {
      "epoch": 6.866455078125e-06,
      "grad_norm": 0.13657401502132416,
      "learning_rate": 9.356622428101686e-07,
      "loss": 0.0569,
      "step": 81200
    },
    {
      "epoch": 6.866455078125e-06,
      "model_forward_time": 1.7744767665863037,
      "step": 81200
    },
    {
      "dynamics_loss_avg": 0.06559937968850135,
      "epoch": 6.866455078125e-06,
      "step": 81200
    },
    {
      "action_loss_avg": 0.0021050427545560523,
      "epoch": 6.866455078125e-06,
      "step": 81200
    },
    {
      "epoch": 6.866455078125e-06,
      "step": 81200,
      "training_step_time": 8.722866535186768
    },
    {
      "epoch": 6.867408752441406e-06,
      "model_forward_time": 2.3860924243927,
      "step": 81201
    },
    {
      "epoch": 6.867408752441406e-06,
      "step": 81201,
      "training_step_time": 8.398168087005615
    },
    {
      "epoch": 6.8683624267578125e-06,
      "model_forward_time": 1.7493820190429688,
      "step": 81202
    },
    {
      "epoch": 6.8683624267578125e-06,
      "step": 81202,
      "training_step_time": 7.783005952835083
    },
    {
      "epoch": 6.869316101074219e-06,
      "model_forward_time": 1.7515366077423096,
      "step": 81203
    },
    {
      "epoch": 6.869316101074219e-06,
      "step": 81203,
      "training_step_time": 7.821120977401733
    },
    {
      "epoch": 6.870269775390625e-06,
      "model_forward_time": 1.747589111328125,
      "step": 81204
    },
    {
      "epoch": 6.870269775390625e-06,
      "step": 81204,
      "training_step_time": 7.4132936000823975
    },
    {
      "epoch": 6.871223449707031e-06,
      "model_forward_time": 1.7553794384002686,
      "step": 81205
    },
    {
      "epoch": 6.871223449707031e-06,
      "step": 81205,
      "training_step_time": 7.688878059387207
    },
    {
      "epoch": 6.872177124023438e-06,
      "model_forward_time": 1.7382457256317139,
      "step": 81206
    },
    {
      "epoch": 6.872177124023438e-06,
      "step": 81206,
      "training_step_time": 7.775214195251465
    },
    {
      "epoch": 6.873130798339844e-06,
      "model_forward_time": 1.749711036682129,
      "step": 81207
    },
    {
      "epoch": 6.873130798339844e-06,
      "step": 81207,
      "training_step_time": 8.354788780212402
    },
    {
      "epoch": 6.87408447265625e-06,
      "model_forward_time": 1.7388267517089844,
      "step": 81208
    },
    {
      "epoch": 6.87408447265625e-06,
      "step": 81208,
      "training_step_time": 10.59908127784729
    },
    {
      "epoch": 6.875038146972656e-06,
      "model_forward_time": 1.7585504055023193,
      "step": 81209
    },
    {
      "epoch": 6.875038146972656e-06,
      "step": 81209,
      "training_step_time": 8.503253698348999
    },
    {
      "epoch": 6.8759918212890625e-06,
      "grad_norm": 0.07994933426380157,
      "learning_rate": 9.346994039216972e-07,
      "loss": 0.0558,
      "step": 81210
    },
    {
      "epoch": 6.8759918212890625e-06,
      "model_forward_time": 2.377271890640259,
      "step": 81210
    },
    {
      "dynamics_loss_avg": 0.05443260502070189,
      "epoch": 6.8759918212890625e-06,
      "step": 81210
    },
    {
      "action_loss_avg": 0.0016947788244578987,
      "epoch": 6.8759918212890625e-06,
      "step": 81210
    },
    {
      "epoch": 6.8759918212890625e-06,
      "step": 81210,
      "training_step_time": 8.503525257110596
    },
    {
      "epoch": 6.876945495605469e-06,
      "model_forward_time": 1.7479748725891113,
      "step": 81211
    },
    {
      "epoch": 6.876945495605469e-06,
      "step": 81211,
      "training_step_time": 7.818554639816284
    },
    {
      "epoch": 6.877899169921875e-06,
      "model_forward_time": 1.7615256309509277,
      "step": 81212
    },
    {
      "epoch": 6.877899169921875e-06,
      "step": 81212,
      "training_step_time": 7.828147649765015
    },
    {
      "epoch": 6.878852844238281e-06,
      "model_forward_time": 1.7506566047668457,
      "step": 81213
    },
    {
      "epoch": 6.878852844238281e-06,
      "step": 81213,
      "training_step_time": 7.431502819061279
    },
    {
      "epoch": 6.879806518554688e-06,
      "model_forward_time": 1.7474706172943115,
      "step": 81214
    },
    {
      "epoch": 6.879806518554688e-06,
      "step": 81214,
      "training_step_time": 7.698797702789307
    },
    {
      "epoch": 6.880760192871094e-06,
      "model_forward_time": 1.7595763206481934,
      "step": 81215
    },
    {
      "epoch": 6.880760192871094e-06,
      "step": 81215,
      "training_step_time": 8.30861210823059
    },
    {
      "epoch": 6.8817138671875e-06,
      "model_forward_time": 1.7506482601165771,
      "step": 81216
    },
    {
      "epoch": 6.8817138671875e-06,
      "step": 81216,
      "training_step_time": 9.388088464736938
    },
    {
      "epoch": 6.882667541503906e-06,
      "model_forward_time": 1.7462458610534668,
      "step": 81217
    },
    {
      "epoch": 6.882667541503906e-06,
      "step": 81217,
      "training_step_time": 8.405369997024536
    },
    {
      "epoch": 6.8836212158203125e-06,
      "model_forward_time": 1.7590713500976562,
      "step": 81218
    },
    {
      "epoch": 6.8836212158203125e-06,
      "step": 81218,
      "training_step_time": 8.829713582992554
    },
    {
      "epoch": 6.884574890136719e-06,
      "model_forward_time": 2.4534380435943604,
      "step": 81219
    },
    {
      "epoch": 6.884574890136719e-06,
      "step": 81219,
      "training_step_time": 7.916753053665161
    },
    {
      "epoch": 6.885528564453125e-06,
      "grad_norm": 0.07690070569515228,
      "learning_rate": 9.337370096083887e-07,
      "loss": 0.0567,
      "step": 81220
    },
    {
      "epoch": 6.885528564453125e-06,
      "model_forward_time": 1.7467339038848877,
      "step": 81220
    },
    {
      "dynamics_loss_avg": 0.04955417523160577,
      "epoch": 6.885528564453125e-06,
      "step": 81220
    },
    {
      "action_loss_avg": 0.0011107908096164465,
      "epoch": 6.885528564453125e-06,
      "step": 81220
    },
    {
      "epoch": 6.885528564453125e-06,
      "step": 81220,
      "training_step_time": 7.79487943649292
    },
    {
      "epoch": 6.886482238769531e-06,
      "model_forward_time": 1.7510523796081543,
      "step": 81221
    },
    {
      "epoch": 6.886482238769531e-06,
      "step": 81221,
      "training_step_time": 7.803523302078247
    },
    {
      "epoch": 6.887435913085938e-06,
      "model_forward_time": 1.7612035274505615,
      "step": 81222
    },
    {
      "epoch": 6.887435913085938e-06,
      "step": 81222,
      "training_step_time": 7.429655313491821
    },
    {
      "epoch": 6.888389587402344e-06,
      "model_forward_time": 1.7603158950805664,
      "step": 81223
    },
    {
      "epoch": 6.888389587402344e-06,
      "step": 81223,
      "training_step_time": 7.742491006851196
    },
    {
      "epoch": 6.88934326171875e-06,
      "model_forward_time": 1.7532901763916016,
      "step": 81224
    },
    {
      "epoch": 6.88934326171875e-06,
      "step": 81224,
      "training_step_time": 8.656217098236084
    },
    {
      "epoch": 6.890296936035156e-06,
      "model_forward_time": 1.7694475650787354,
      "step": 81225
    },
    {
      "epoch": 6.890296936035156e-06,
      "step": 81225,
      "training_step_time": 9.310582876205444
    },
    {
      "epoch": 6.8912506103515625e-06,
      "model_forward_time": 1.7560672760009766,
      "step": 81226
    },
    {
      "epoch": 6.8912506103515625e-06,
      "step": 81226,
      "training_step_time": 8.898793458938599
    },
    {
      "epoch": 6.892204284667969e-06,
      "model_forward_time": 1.7639744281768799,
      "step": 81227
    },
    {
      "epoch": 6.892204284667969e-06,
      "step": 81227,
      "training_step_time": 7.850649118423462
    },
    {
      "epoch": 6.893157958984375e-06,
      "model_forward_time": 2.455713987350464,
      "step": 81228
    },
    {
      "epoch": 6.893157958984375e-06,
      "step": 81228,
      "training_step_time": 8.903531074523926
    },
    {
      "epoch": 6.894111633300781e-06,
      "model_forward_time": 1.7483599185943604,
      "step": 81229
    },
    {
      "epoch": 6.894111633300781e-06,
      "step": 81229,
      "training_step_time": 7.7534871101379395
    },
    {
      "epoch": 6.895065307617188e-06,
      "grad_norm": 0.09211570024490356,
      "learning_rate": 9.327750599754853e-07,
      "loss": 0.0549,
      "step": 81230
    },
    {
      "epoch": 6.895065307617188e-06,
      "model_forward_time": 1.756622076034546,
      "step": 81230
    },
    {
      "dynamics_loss_avg": 0.06196372667327523,
      "epoch": 6.895065307617188e-06,
      "step": 81230
    },
    {
      "action_loss_avg": 0.0020365520525956527,
      "epoch": 6.895065307617188e-06,
      "step": 81230
    },
    {
      "epoch": 6.895065307617188e-06,
      "step": 81230,
      "training_step_time": 7.564862966537476
    },
    {
      "epoch": 6.896018981933594e-06,
      "model_forward_time": 1.7634010314941406,
      "step": 81231
    },
    {
      "epoch": 6.896018981933594e-06,
      "step": 81231,
      "training_step_time": 8.546664953231812
    },
    {
      "epoch": 6.89697265625e-06,
      "model_forward_time": 1.74757981300354,
      "step": 81232
    },
    {
      "epoch": 6.89697265625e-06,
      "step": 81232,
      "training_step_time": 7.6761205196380615
    },
    {
      "epoch": 6.897926330566406e-06,
      "model_forward_time": 1.754533290863037,
      "step": 81233
    },
    {
      "epoch": 6.897926330566406e-06,
      "step": 81233,
      "training_step_time": 7.74817967414856
    },
    {
      "epoch": 6.8988800048828125e-06,
      "model_forward_time": 1.7517781257629395,
      "step": 81234
    },
    {
      "epoch": 6.8988800048828125e-06,
      "step": 81234,
      "training_step_time": 8.778210163116455
    },
    {
      "epoch": 6.899833679199219e-06,
      "model_forward_time": 1.7598683834075928,
      "step": 81235
    },
    {
      "epoch": 6.899833679199219e-06,
      "step": 81235,
      "training_step_time": 9.004128217697144
    },
    {
      "epoch": 6.900787353515625e-06,
      "model_forward_time": 1.7570250034332275,
      "step": 81236
    },
    {
      "epoch": 6.900787353515625e-06,
      "step": 81236,
      "training_step_time": 10.445842742919922
    },
    {
      "epoch": 6.901741027832031e-06,
      "model_forward_time": 1.8627233505249023,
      "step": 81237
    },
    {
      "epoch": 6.901741027832031e-06,
      "step": 81237,
      "training_step_time": 7.871471881866455
    },
    {
      "epoch": 6.902694702148438e-06,
      "model_forward_time": 2.3749330043792725,
      "step": 81238
    },
    {
      "epoch": 6.902694702148438e-06,
      "step": 81238,
      "training_step_time": 9.245400667190552
    },
    {
      "epoch": 6.903648376464844e-06,
      "model_forward_time": 1.7535829544067383,
      "step": 81239
    },
    {
      "epoch": 6.903648376464844e-06,
      "step": 81239,
      "training_step_time": 9.852695226669312
    },
    {
      "epoch": 6.90460205078125e-06,
      "grad_norm": 0.09591388702392578,
      "learning_rate": 9.318135551281876e-07,
      "loss": 0.0563,
      "step": 81240
    },
    {
      "epoch": 6.90460205078125e-06,
      "model_forward_time": 1.751380443572998,
      "step": 81240
    },
    {
      "dynamics_loss_avg": 0.0795988354831934,
      "epoch": 6.90460205078125e-06,
      "step": 81240
    },
    {
      "action_loss_avg": 0.0013164380012312903,
      "epoch": 6.90460205078125e-06,
      "step": 81240
    },
    {
      "epoch": 6.90460205078125e-06,
      "step": 81240,
      "training_step_time": 7.8077943325042725
    },
    {
      "epoch": 6.905555725097656e-06,
      "model_forward_time": 1.3867995738983154,
      "step": 81241
    },
    {
      "epoch": 6.905555725097656e-06,
      "step": 81241,
      "training_step_time": 7.704243183135986
    },
    {
      "epoch": 6.9065093994140625e-06,
      "model_forward_time": 1.7468276023864746,
      "step": 81242
    },
    {
      "epoch": 6.9065093994140625e-06,
      "step": 81242,
      "training_step_time": 8.383242130279541
    },
    {
      "epoch": 6.907463073730469e-06,
      "model_forward_time": 1.7589561939239502,
      "step": 81243
    },
    {
      "epoch": 6.907463073730469e-06,
      "step": 81243,
      "training_step_time": 7.730664491653442
    },
    {
      "epoch": 6.908416748046875e-06,
      "model_forward_time": 1.3784613609313965,
      "step": 81244
    },
    {
      "epoch": 6.908416748046875e-06,
      "step": 81244,
      "training_step_time": 7.8874125480651855
    },
    {
      "epoch": 6.909370422363281e-06,
      "model_forward_time": 1.7491424083709717,
      "step": 81245
    },
    {
      "epoch": 6.909370422363281e-06,
      "step": 81245,
      "training_step_time": 8.692167282104492
    },
    {
      "epoch": 6.910324096679688e-06,
      "model_forward_time": 1.8749516010284424,
      "step": 81246
    },
    {
      "epoch": 6.910324096679688e-06,
      "step": 81246,
      "training_step_time": 9.304456233978271
    },
    {
      "epoch": 6.911277770996094e-06,
      "model_forward_time": 2.409118890762329,
      "step": 81247
    },
    {
      "epoch": 6.911277770996094e-06,
      "step": 81247,
      "training_step_time": 7.68413519859314
    },
    {
      "epoch": 6.9122314453125e-06,
      "model_forward_time": 1.7443428039550781,
      "step": 81248
    },
    {
      "epoch": 6.9122314453125e-06,
      "step": 81248,
      "training_step_time": 9.313647985458374
    },
    {
      "epoch": 6.913185119628906e-06,
      "model_forward_time": 1.7496147155761719,
      "step": 81249
    },
    {
      "epoch": 6.913185119628906e-06,
      "step": 81249,
      "training_step_time": 7.774401664733887
    },
    {
      "epoch": 6.9141387939453125e-06,
      "grad_norm": 0.09915965795516968,
      "learning_rate": 9.308524951716424e-07,
      "loss": 0.0568,
      "step": 81250
    },
    {
      "epoch": 6.9141387939453125e-06,
      "model_forward_time": 1.762366771697998,
      "step": 81250
    },
    {
      "dynamics_loss_avg": 0.0678679347038269,
      "epoch": 6.9141387939453125e-06,
      "step": 81250
    },
    {
      "action_loss_avg": 0.0015410234802402556,
      "epoch": 6.9141387939453125e-06,
      "step": 81250
    },
    {
      "epoch": 6.9141387939453125e-06,
      "step": 81250,
      "training_step_time": 8.519319534301758
    },
    {
      "epoch": 6.915092468261719e-06,
      "model_forward_time": 1.7484180927276611,
      "step": 81251
    },
    {
      "epoch": 6.915092468261719e-06,
      "step": 81251,
      "training_step_time": 7.777137279510498
    },
    {
      "epoch": 6.916046142578125e-06,
      "model_forward_time": 1.746201753616333,
      "step": 81252
    },
    {
      "epoch": 6.916046142578125e-06,
      "step": 81252,
      "training_step_time": 7.796953439712524
    },
    {
      "epoch": 6.916999816894531e-06,
      "model_forward_time": 1.7444391250610352,
      "step": 81253
    },
    {
      "epoch": 6.916999816894531e-06,
      "step": 81253,
      "training_step_time": 8.462565898895264
    },
    {
      "epoch": 6.917953491210937e-06,
      "model_forward_time": 1.7477927207946777,
      "step": 81254
    },
    {
      "epoch": 6.917953491210937e-06,
      "step": 81254,
      "training_step_time": 7.801388263702393
    },
    {
      "epoch": 6.918907165527344e-06,
      "model_forward_time": 1.874861478805542,
      "step": 81255
    },
    {
      "epoch": 6.918907165527344e-06,
      "step": 81255,
      "training_step_time": 7.7317140102386475
    },
    {
      "epoch": 6.91986083984375e-06,
      "model_forward_time": 2.398591995239258,
      "step": 81256
    },
    {
      "epoch": 6.91986083984375e-06,
      "step": 81256,
      "training_step_time": 7.741766691207886
    },
    {
      "epoch": 6.920814514160156e-06,
      "model_forward_time": 1.37339186668396,
      "step": 81257
    },
    {
      "epoch": 6.920814514160156e-06,
      "step": 81257,
      "training_step_time": 7.847209930419922
    },
    {
      "epoch": 6.9217681884765625e-06,
      "model_forward_time": 1.7479870319366455,
      "step": 81258
    },
    {
      "epoch": 6.9217681884765625e-06,
      "step": 81258,
      "training_step_time": 7.779072999954224
    },
    {
      "epoch": 6.922721862792969e-06,
      "model_forward_time": 1.747668981552124,
      "step": 81259
    },
    {
      "epoch": 6.922721862792969e-06,
      "step": 81259,
      "training_step_time": 7.4796061515808105
    },
    {
      "epoch": 6.923675537109375e-06,
      "grad_norm": 0.07717638462781906,
      "learning_rate": 9.298918802109497e-07,
      "loss": 0.0571,
      "step": 81260
    },
    {
      "epoch": 6.923675537109375e-06,
      "model_forward_time": 1.7463572025299072,
      "step": 81260
    },
    {
      "dynamics_loss_avg": 0.053360197134315966,
      "epoch": 6.923675537109375e-06,
      "step": 81260
    },
    {
      "action_loss_avg": 0.0011213873600354417,
      "epoch": 6.923675537109375e-06,
      "step": 81260
    },
    {
      "epoch": 6.923675537109375e-06,
      "step": 81260,
      "training_step_time": 7.790210723876953
    },
    {
      "epoch": 6.924629211425781e-06,
      "model_forward_time": 1.7522170543670654,
      "step": 81261
    },
    {
      "epoch": 6.924629211425781e-06,
      "step": 81261,
      "training_step_time": 9.633244276046753
    },
    {
      "epoch": 6.925582885742187e-06,
      "model_forward_time": 1.7466306686401367,
      "step": 81262
    },
    {
      "epoch": 6.925582885742187e-06,
      "step": 81262,
      "training_step_time": 8.409117937088013
    },
    {
      "epoch": 6.926536560058594e-06,
      "model_forward_time": 1.7460682392120361,
      "step": 81263
    },
    {
      "epoch": 6.926536560058594e-06,
      "step": 81263,
      "training_step_time": 8.817500829696655
    },
    {
      "epoch": 6.927490234375e-06,
      "model_forward_time": 1.8749089241027832,
      "step": 81264
    },
    {
      "epoch": 6.927490234375e-06,
      "step": 81264,
      "training_step_time": 7.795631170272827
    },
    {
      "epoch": 6.928443908691406e-06,
      "model_forward_time": 1.7500874996185303,
      "step": 81265
    },
    {
      "epoch": 6.928443908691406e-06,
      "step": 81265,
      "training_step_time": 7.785009384155273
    },
    {
      "epoch": 6.9293975830078125e-06,
      "model_forward_time": 2.356215476989746,
      "step": 81266
    },
    {
      "epoch": 6.9293975830078125e-06,
      "step": 81266,
      "training_step_time": 9.508092880249023
    },
    {
      "epoch": 6.930351257324219e-06,
      "model_forward_time": 1.7560310363769531,
      "step": 81267
    },
    {
      "epoch": 6.930351257324219e-06,
      "step": 81267,
      "training_step_time": 7.733942031860352
    },
    {
      "epoch": 6.931304931640625e-06,
      "model_forward_time": 1.7455880641937256,
      "step": 81268
    },
    {
      "epoch": 6.931304931640625e-06,
      "step": 81268,
      "training_step_time": 7.714895963668823
    },
    {
      "epoch": 6.932258605957031e-06,
      "model_forward_time": 1.7476441860198975,
      "step": 81269
    },
    {
      "epoch": 6.932258605957031e-06,
      "step": 81269,
      "training_step_time": 7.736992835998535
    },
    {
      "epoch": 6.933212280273437e-06,
      "grad_norm": 0.08982431143522263,
      "learning_rate": 9.289317103511624e-07,
      "loss": 0.0556,
      "step": 81270
    },
    {
      "epoch": 6.933212280273437e-06,
      "model_forward_time": 1.3803553581237793,
      "step": 81270
    },
    {
      "dynamics_loss_avg": 0.0780945422127843,
      "epoch": 6.933212280273437e-06,
      "step": 81270
    },
    {
      "action_loss_avg": 0.0020654020248912276,
      "epoch": 6.933212280273437e-06,
      "step": 81270
    },
    {
      "epoch": 6.933212280273437e-06,
      "step": 81270,
      "training_step_time": 7.634609222412109
    },
    {
      "epoch": 6.934165954589844e-06,
      "model_forward_time": 1.7543981075286865,
      "step": 81271
    },
    {
      "epoch": 6.934165954589844e-06,
      "step": 81271,
      "training_step_time": 7.81885290145874
    },
    {
      "epoch": 6.93511962890625e-06,
      "model_forward_time": 1.7496297359466553,
      "step": 81272
    },
    {
      "epoch": 6.93511962890625e-06,
      "step": 81272,
      "training_step_time": 8.722121953964233
    },
    {
      "epoch": 6.936073303222656e-06,
      "model_forward_time": 1.861750602722168,
      "step": 81273
    },
    {
      "epoch": 6.936073303222656e-06,
      "step": 81273,
      "training_step_time": 9.612457513809204
    },
    {
      "epoch": 6.9370269775390625e-06,
      "model_forward_time": 1.757443904876709,
      "step": 81274
    },
    {
      "epoch": 6.9370269775390625e-06,
      "step": 81274,
      "training_step_time": 8.143753051757812
    },
    {
      "epoch": 6.937980651855469e-06,
      "model_forward_time": 2.348672389984131,
      "step": 81275
    },
    {
      "epoch": 6.937980651855469e-06,
      "step": 81275,
      "training_step_time": 8.632302045822144
    },
    {
      "epoch": 6.938934326171875e-06,
      "model_forward_time": 1.741175889968872,
      "step": 81276
    },
    {
      "epoch": 6.938934326171875e-06,
      "step": 81276,
      "training_step_time": 7.654506683349609
    },
    {
      "epoch": 6.939888000488281e-06,
      "model_forward_time": 1.7505607604980469,
      "step": 81277
    },
    {
      "epoch": 6.939888000488281e-06,
      "step": 81277,
      "training_step_time": 7.69977879524231
    },
    {
      "epoch": 6.940841674804687e-06,
      "model_forward_time": 1.764082908630371,
      "step": 81278
    },
    {
      "epoch": 6.940841674804687e-06,
      "step": 81278,
      "training_step_time": 7.608657360076904
    },
    {
      "epoch": 6.941795349121094e-06,
      "model_forward_time": 1.7532246112823486,
      "step": 81279
    },
    {
      "epoch": 6.941795349121094e-06,
      "step": 81279,
      "training_step_time": 7.240622282028198
    },
    {
      "epoch": 6.9427490234375e-06,
      "grad_norm": 0.08105218410491943,
      "learning_rate": 9.279719856972835e-07,
      "loss": 0.0574,
      "step": 81280
    },
    {
      "epoch": 6.9427490234375e-06,
      "model_forward_time": 1.742119550704956,
      "step": 81280
    },
    {
      "dynamics_loss_avg": 0.062175259739160535,
      "epoch": 6.9427490234375e-06,
      "step": 81280
    },
    {
      "action_loss_avg": 0.002614822203759104,
      "epoch": 6.9427490234375e-06,
      "step": 81280
    },
    {
      "epoch": 6.9427490234375e-06,
      "step": 81280,
      "training_step_time": 7.633427381515503
    },
    {
      "epoch": 6.943702697753906e-06,
      "model_forward_time": 1.769423007965088,
      "step": 81281
    },
    {
      "epoch": 6.943702697753906e-06,
      "step": 81281,
      "training_step_time": 7.740157127380371
    },
    {
      "epoch": 6.9446563720703125e-06,
      "model_forward_time": 1.8619027137756348,
      "step": 81282
    },
    {
      "epoch": 6.9446563720703125e-06,
      "step": 81282,
      "training_step_time": 7.659411430358887
    },
    {
      "epoch": 6.945610046386719e-06,
      "model_forward_time": 1.7551946640014648,
      "step": 81283
    },
    {
      "epoch": 6.945610046386719e-06,
      "step": 81283,
      "training_step_time": 7.62696647644043
    },
    {
      "epoch": 6.946563720703125e-06,
      "model_forward_time": 2.4193406105041504,
      "step": 81284
    },
    {
      "epoch": 6.946563720703125e-06,
      "step": 81284,
      "training_step_time": 9.226686716079712
    },
    {
      "epoch": 6.947517395019531e-06,
      "model_forward_time": 1.7273008823394775,
      "step": 81285
    },
    {
      "epoch": 6.947517395019531e-06,
      "step": 81285,
      "training_step_time": 7.637858152389526
    },
    {
      "epoch": 6.948471069335937e-06,
      "model_forward_time": 1.7549145221710205,
      "step": 81286
    },
    {
      "epoch": 6.948471069335937e-06,
      "step": 81286,
      "training_step_time": 7.558352708816528
    },
    {
      "epoch": 6.949424743652344e-06,
      "model_forward_time": 1.7496235370635986,
      "step": 81287
    },
    {
      "epoch": 6.949424743652344e-06,
      "step": 81287,
      "training_step_time": 7.622573137283325
    },
    {
      "epoch": 6.95037841796875e-06,
      "model_forward_time": 1.7530889511108398,
      "step": 81288
    },
    {
      "epoch": 6.95037841796875e-06,
      "step": 81288,
      "training_step_time": 7.306695222854614
    },
    {
      "epoch": 6.951332092285156e-06,
      "model_forward_time": 1.7465770244598389,
      "step": 81289
    },
    {
      "epoch": 6.951332092285156e-06,
      "step": 81289,
      "training_step_time": 8.610679149627686
    },
    {
      "epoch": 6.9522857666015625e-06,
      "grad_norm": 0.07248824834823608,
      "learning_rate": 9.270127063542644e-07,
      "loss": 0.0562,
      "step": 81290
    },
    {
      "epoch": 6.9522857666015625e-06,
      "model_forward_time": 1.7403194904327393,
      "step": 81290
    },
    {
      "dynamics_loss_avg": 0.055213738791644575,
      "epoch": 6.9522857666015625e-06,
      "step": 81290
    },
    {
      "action_loss_avg": 0.0009885271836537867,
      "epoch": 6.9522857666015625e-06,
      "step": 81290
    },
    {
      "epoch": 6.9522857666015625e-06,
      "step": 81290,
      "training_step_time": 8.519227266311646
    },
    {
      "epoch": 6.953239440917969e-06,
      "model_forward_time": 1.8321356773376465,
      "step": 81291
    },
    {
      "epoch": 6.953239440917969e-06,
      "step": 81291,
      "training_step_time": 10.921116828918457
    },
    {
      "epoch": 6.954193115234375e-06,
      "model_forward_time": 1.7808003425598145,
      "step": 81292
    },
    {
      "epoch": 6.954193115234375e-06,
      "step": 81292,
      "training_step_time": 7.623883485794067
    },
    {
      "epoch": 6.955146789550781e-06,
      "model_forward_time": 2.533308982849121,
      "step": 81293
    },
    {
      "epoch": 6.955146789550781e-06,
      "step": 81293,
      "training_step_time": 8.466975450515747
    },
    {
      "epoch": 6.956100463867187e-06,
      "model_forward_time": 1.5192127227783203,
      "step": 81294
    },
    {
      "epoch": 6.956100463867187e-06,
      "step": 81294,
      "training_step_time": 7.710751533508301
    },
    {
      "epoch": 6.957054138183594e-06,
      "model_forward_time": 1.8994712829589844,
      "step": 81295
    },
    {
      "epoch": 6.957054138183594e-06,
      "step": 81295,
      "training_step_time": 7.604664087295532
    },
    {
      "epoch": 6.9580078125e-06,
      "model_forward_time": 1.906242847442627,
      "step": 81296
    },
    {
      "epoch": 6.9580078125e-06,
      "step": 81296,
      "training_step_time": 7.6501524448394775
    },
    {
      "epoch": 6.958961486816406e-06,
      "model_forward_time": 1.9079616069793701,
      "step": 81297
    },
    {
      "epoch": 6.958961486816406e-06,
      "step": 81297,
      "training_step_time": 7.342176675796509
    },
    {
      "epoch": 6.9599151611328125e-06,
      "model_forward_time": 1.8940527439117432,
      "step": 81298
    },
    {
      "epoch": 6.9599151611328125e-06,
      "step": 81298,
      "training_step_time": 8.584124088287354
    },
    {
      "epoch": 6.960868835449219e-06,
      "model_forward_time": 1.9081251621246338,
      "step": 81299
    },
    {
      "epoch": 6.960868835449219e-06,
      "step": 81299,
      "training_step_time": 8.295655727386475
    },
    {
      "epoch": 6.961822509765625e-06,
      "grad_norm": 0.1186230257153511,
      "learning_rate": 9.260538724270124e-07,
      "loss": 0.0555,
      "step": 81300
    },
    {
      "epoch": 6.961822509765625e-06,
      "model_forward_time": 1.9071152210235596,
      "step": 81300
    },
    {
      "dynamics_loss_avg": 0.05305223213508725,
      "epoch": 6.961822509765625e-06,
      "step": 81300
    },
    {
      "action_loss_avg": 0.0010288788107573055,
      "epoch": 6.961822509765625e-06,
      "step": 81300
    },
    {
      "epoch": 6.961822509765625e-06,
      "step": 81300,
      "training_step_time": 9.283377408981323
    },
    {
      "epoch": 6.962776184082031e-06,
      "model_forward_time": 1.9064767360687256,
      "step": 81301
    },
    {
      "epoch": 6.962776184082031e-06,
      "step": 81301,
      "training_step_time": 7.668136358261108
    },
    {
      "epoch": 6.963729858398437e-06,
      "model_forward_time": 1.9036521911621094,
      "step": 81302
    },
    {
      "epoch": 6.963729858398437e-06,
      "step": 81302,
      "training_step_time": 7.64496111869812
    },
    {
      "epoch": 6.964683532714844e-06,
      "model_forward_time": 2.4931957721710205,
      "step": 81303
    },
    {
      "epoch": 6.964683532714844e-06,
      "step": 81303,
      "training_step_time": 7.726280450820923
    },
    {
      "epoch": 6.96563720703125e-06,
      "model_forward_time": 1.9142932891845703,
      "step": 81304
    },
    {
      "epoch": 6.96563720703125e-06,
      "step": 81304,
      "training_step_time": 7.680365085601807
    },
    {
      "epoch": 6.966590881347656e-06,
      "model_forward_time": 1.8935093879699707,
      "step": 81305
    },
    {
      "epoch": 6.966590881347656e-06,
      "step": 81305,
      "training_step_time": 7.6495201587677
    },
    {
      "epoch": 6.9675445556640625e-06,
      "model_forward_time": 1.9037630558013916,
      "step": 81306
    },
    {
      "epoch": 6.9675445556640625e-06,
      "step": 81306,
      "training_step_time": 7.58596134185791
    },
    {
      "epoch": 6.968498229980469e-06,
      "model_forward_time": 1.8949713706970215,
      "step": 81307
    },
    {
      "epoch": 6.968498229980469e-06,
      "step": 81307,
      "training_step_time": 8.56036639213562
    },
    {
      "epoch": 6.969451904296875e-06,
      "model_forward_time": 1.907137393951416,
      "step": 81308
    },
    {
      "epoch": 6.969451904296875e-06,
      "step": 81308,
      "training_step_time": 7.6253931522369385
    },
    {
      "epoch": 6.970405578613281e-06,
      "model_forward_time": 1.907015085220337,
      "step": 81309
    },
    {
      "epoch": 6.970405578613281e-06,
      "step": 81309,
      "training_step_time": 9.47182035446167
    },
    {
      "epoch": 6.971359252929687e-06,
      "grad_norm": 0.08103284239768982,
      "learning_rate": 9.250954840203852e-07,
      "loss": 0.055,
      "step": 81310
    },
    {
      "epoch": 6.971359252929687e-06,
      "model_forward_time": 1.9178149700164795,
      "step": 81310
    },
    {
      "dynamics_loss_avg": 0.0741963054984808,
      "epoch": 6.971359252929687e-06,
      "step": 81310
    },
    {
      "action_loss_avg": 0.0012917295010993258,
      "epoch": 6.971359252929687e-06,
      "step": 81310
    },
    {
      "epoch": 6.971359252929687e-06,
      "step": 81310,
      "training_step_time": 8.606905937194824
    },
    {
      "epoch": 6.972312927246094e-06,
      "model_forward_time": 1.9065089225769043,
      "step": 81311
    },
    {
      "epoch": 6.972312927246094e-06,
      "step": 81311,
      "training_step_time": 8.492509603500366
    },
    {
      "epoch": 6.9732666015625e-06,
      "model_forward_time": 2.4630563259124756,
      "step": 81312
    },
    {
      "epoch": 6.9732666015625e-06,
      "step": 81312,
      "training_step_time": 9.066459655761719
    },
    {
      "epoch": 6.974220275878906e-06,
      "model_forward_time": 1.9025774002075195,
      "step": 81313
    },
    {
      "epoch": 6.974220275878906e-06,
      "step": 81313,
      "training_step_time": 7.6654298305511475
    },
    {
      "epoch": 6.9751739501953126e-06,
      "model_forward_time": 1.9085109233856201,
      "step": 81314
    },
    {
      "epoch": 6.9751739501953126e-06,
      "step": 81314,
      "training_step_time": 7.673887491226196
    },
    {
      "epoch": 6.976127624511719e-06,
      "model_forward_time": 1.908874750137329,
      "step": 81315
    },
    {
      "epoch": 6.976127624511719e-06,
      "step": 81315,
      "training_step_time": 7.622545957565308
    },
    {
      "epoch": 6.977081298828125e-06,
      "model_forward_time": 1.8796639442443848,
      "step": 81316
    },
    {
      "epoch": 6.977081298828125e-06,
      "step": 81316,
      "training_step_time": 7.662481784820557
    },
    {
      "epoch": 6.978034973144531e-06,
      "model_forward_time": 1.9063975811004639,
      "step": 81317
    },
    {
      "epoch": 6.978034973144531e-06,
      "step": 81317,
      "training_step_time": 8.526358604431152
    },
    {
      "epoch": 6.978988647460937e-06,
      "model_forward_time": 1.9131808280944824,
      "step": 81318
    },
    {
      "epoch": 6.978988647460937e-06,
      "step": 81318,
      "training_step_time": 7.722614526748657
    },
    {
      "epoch": 6.979942321777344e-06,
      "model_forward_time": 1.9264435768127441,
      "step": 81319
    },
    {
      "epoch": 6.979942321777344e-06,
      "step": 81319,
      "training_step_time": 8.255769729614258
    },
    {
      "epoch": 6.98089599609375e-06,
      "grad_norm": 0.07536135613918304,
      "learning_rate": 9.241375412391879e-07,
      "loss": 0.0556,
      "step": 81320
    },
    {
      "epoch": 6.98089599609375e-06,
      "model_forward_time": 1.1677112579345703,
      "step": 81320
    },
    {
      "dynamics_loss_avg": 0.05643202718347311,
      "epoch": 6.98089599609375e-06,
      "step": 81320
    },
    {
      "action_loss_avg": 0.002086602902272716,
      "epoch": 6.98089599609375e-06,
      "step": 81320
    },
    {
      "epoch": 6.98089599609375e-06,
      "step": 81320,
      "training_step_time": 8.307431936264038
    },
    {
      "epoch": 6.981849670410156e-06,
      "model_forward_time": 1.9015183448791504,
      "step": 81321
    },
    {
      "epoch": 6.981849670410156e-06,
      "step": 81321,
      "training_step_time": 8.29539179801941
    },
    {
      "epoch": 6.9828033447265626e-06,
      "model_forward_time": 1.8886494636535645,
      "step": 81322
    },
    {
      "epoch": 6.9828033447265626e-06,
      "step": 81322,
      "training_step_time": 7.678597688674927
    },
    {
      "epoch": 6.983757019042969e-06,
      "model_forward_time": 1.9035522937774658,
      "step": 81323
    },
    {
      "epoch": 6.983757019042969e-06,
      "step": 81323,
      "training_step_time": 7.670152902603149
    },
    {
      "epoch": 6.984710693359375e-06,
      "model_forward_time": 1.910797119140625,
      "step": 81324
    },
    {
      "epoch": 6.984710693359375e-06,
      "step": 81324,
      "training_step_time": 7.6701743602752686
    },
    {
      "epoch": 6.985664367675781e-06,
      "model_forward_time": 1.902575969696045,
      "step": 81325
    },
    {
      "epoch": 6.985664367675781e-06,
      "step": 81325,
      "training_step_time": 8.264332294464111
    },
    {
      "epoch": 6.986618041992187e-06,
      "model_forward_time": 1.9021461009979248,
      "step": 81326
    },
    {
      "epoch": 6.986618041992187e-06,
      "step": 81326,
      "training_step_time": 7.603459119796753
    },
    {
      "epoch": 6.987571716308594e-06,
      "model_forward_time": 1.9085009098052979,
      "step": 81327
    },
    {
      "epoch": 6.987571716308594e-06,
      "step": 81327,
      "training_step_time": 9.111671686172485
    },
    {
      "epoch": 6.988525390625e-06,
      "model_forward_time": 1.9137053489685059,
      "step": 81328
    },
    {
      "epoch": 6.988525390625e-06,
      "step": 81328,
      "training_step_time": 7.666759252548218
    },
    {
      "epoch": 6.989479064941406e-06,
      "model_forward_time": 1.9152660369873047,
      "step": 81329
    },
    {
      "epoch": 6.989479064941406e-06,
      "step": 81329,
      "training_step_time": 7.673492670059204
    },
    {
      "epoch": 6.9904327392578126e-06,
      "grad_norm": 0.08188439160585403,
      "learning_rate": 9.231800441881811e-07,
      "loss": 0.0549,
      "step": 81330
    },
    {
      "epoch": 6.9904327392578126e-06,
      "model_forward_time": 1.9022176265716553,
      "step": 81330
    },
    {
      "dynamics_loss_avg": 0.07236591260880232,
      "epoch": 6.9904327392578126e-06,
      "step": 81330
    },
    {
      "action_loss_avg": 0.0021407799154985695,
      "epoch": 6.9904327392578126e-06,
      "step": 81330
    },
    {
      "epoch": 6.9904327392578126e-06,
      "step": 81330,
      "training_step_time": 8.218258619308472
    },
    {
      "epoch": 6.991386413574219e-06,
      "model_forward_time": 2.511892795562744,
      "step": 81331
    },
    {
      "epoch": 6.991386413574219e-06,
      "step": 81331,
      "training_step_time": 7.640957355499268
    },
    {
      "epoch": 6.992340087890625e-06,
      "model_forward_time": 1.9038081169128418,
      "step": 81332
    },
    {
      "epoch": 6.992340087890625e-06,
      "step": 81332,
      "training_step_time": 8.515960454940796
    },
    {
      "epoch": 6.9932937622070314e-06,
      "model_forward_time": 1.9118578433990479,
      "step": 81333
    },
    {
      "epoch": 6.9932937622070314e-06,
      "step": 81333,
      "training_step_time": 7.622677564620972
    },
    {
      "epoch": 6.994247436523437e-06,
      "model_forward_time": 1.9098460674285889,
      "step": 81334
    },
    {
      "epoch": 6.994247436523437e-06,
      "step": 81334,
      "training_step_time": 7.671587705612183
    },
    {
      "epoch": 6.995201110839844e-06,
      "model_forward_time": 1.9085533618927002,
      "step": 81335
    },
    {
      "epoch": 6.995201110839844e-06,
      "step": 81335,
      "training_step_time": 9.309645175933838
    },
    {
      "epoch": 6.99615478515625e-06,
      "model_forward_time": 1.903583288192749,
      "step": 81336
    },
    {
      "epoch": 6.99615478515625e-06,
      "step": 81336,
      "training_step_time": 7.6811683177948
    },
    {
      "epoch": 6.997108459472656e-06,
      "model_forward_time": 1.899928331375122,
      "step": 81337
    },
    {
      "epoch": 6.997108459472656e-06,
      "step": 81337,
      "training_step_time": 8.691746711730957
    },
    {
      "epoch": 6.9980621337890626e-06,
      "model_forward_time": 1.527297019958496,
      "step": 81338
    },
    {
      "epoch": 6.9980621337890626e-06,
      "step": 81338,
      "training_step_time": 7.632668733596802
    },
    {
      "epoch": 6.999015808105469e-06,
      "model_forward_time": 1.902069091796875,
      "step": 81339
    },
    {
      "epoch": 6.999015808105469e-06,
      "step": 81339,
      "training_step_time": 7.642855167388916
    },
    {
      "epoch": 6.999969482421875e-06,
      "grad_norm": 0.0832979828119278,
      "learning_rate": 9.222229929720756e-07,
      "loss": 0.0553,
      "step": 81340
    },
    {
      "epoch": 6.999969482421875e-06,
      "model_forward_time": 2.5545573234558105,
      "step": 81340
    },
    {
      "dynamics_loss_avg": 0.054819997400045395,
      "epoch": 6.999969482421875e-06,
      "step": 81340
    },
    {
      "action_loss_avg": 0.001207115116994828,
      "epoch": 6.999969482421875e-06,
      "step": 81340
    },
    {
      "epoch": 6.999969482421875e-06,
      "step": 81340,
      "training_step_time": 7.720112085342407
    },
    {
      "epoch": 7.0009231567382814e-06,
      "model_forward_time": 1.9110164642333984,
      "step": 81341
    },
    {
      "epoch": 7.0009231567382814e-06,
      "step": 81341,
      "training_step_time": 7.664447546005249
    },
    {
      "epoch": 7.001876831054687e-06,
      "model_forward_time": 1.9063763618469238,
      "step": 81342
    },
    {
      "epoch": 7.001876831054687e-06,
      "step": 81342,
      "training_step_time": 7.562060594558716
    },
    {
      "epoch": 7.002830505371094e-06,
      "model_forward_time": 2.9803900718688965,
      "step": 81343
    },
    {
      "epoch": 7.002830505371094e-06,
      "step": 81343,
      "training_step_time": 9.653550148010254
    },
    {
      "epoch": 7.0037841796875e-06,
      "model_forward_time": 1.9069328308105469,
      "step": 81344
    },
    {
      "epoch": 7.0037841796875e-06,
      "step": 81344,
      "training_step_time": 8.770956039428711
    },
    {
      "epoch": 7.004737854003906e-06,
      "model_forward_time": 1.919391393661499,
      "step": 81345
    },
    {
      "epoch": 7.004737854003906e-06,
      "step": 81345,
      "training_step_time": 7.216134548187256
    },
    {
      "epoch": 7.005691528320313e-06,
      "model_forward_time": 1.9047038555145264,
      "step": 81346
    },
    {
      "epoch": 7.005691528320313e-06,
      "step": 81346,
      "training_step_time": 7.42423939704895
    },
    {
      "epoch": 7.006645202636719e-06,
      "model_forward_time": 1.9050943851470947,
      "step": 81347
    },
    {
      "epoch": 7.006645202636719e-06,
      "step": 81347,
      "training_step_time": 7.437187433242798
    },
    {
      "epoch": 7.007598876953125e-06,
      "model_forward_time": 1.7405753135681152,
      "step": 81348
    },
    {
      "epoch": 7.007598876953125e-06,
      "step": 81348,
      "training_step_time": 6.710853576660156
    },
    {
      "epoch": 7.0085525512695314e-06,
      "model_forward_time": 1.7396459579467773,
      "step": 81349
    },
    {
      "epoch": 7.0085525512695314e-06,
      "step": 81349,
      "training_step_time": 6.718951463699341
    },
    {
      "epoch": 7.009506225585937e-06,
      "grad_norm": 0.10174538195133209,
      "learning_rate": 9.212663876955336e-07,
      "loss": 0.0547,
      "step": 81350
    },
    {
      "epoch": 7.009506225585937e-06,
      "model_forward_time": 1.7361528873443604,
      "step": 81350
    },
    {
      "dynamics_loss_avg": 0.056277240999042986,
      "epoch": 7.009506225585937e-06,
      "step": 81350
    },
    {
      "action_loss_avg": 0.0016503119870321826,
      "epoch": 7.009506225585937e-06,
      "step": 81350
    },
    {
      "epoch": 7.009506225585937e-06,
      "step": 81350,
      "training_step_time": 6.73244309425354
    },
    {
      "epoch": 7.010459899902344e-06,
      "model_forward_time": 1.7384111881256104,
      "step": 81351
    },
    {
      "epoch": 7.010459899902344e-06,
      "step": 81351,
      "training_step_time": 6.7163450717926025
    },
    {
      "epoch": 7.01141357421875e-06,
      "model_forward_time": 2.4157328605651855,
      "step": 81352
    },
    {
      "epoch": 7.01141357421875e-06,
      "step": 81352,
      "training_step_time": 7.424347162246704
    },
    {
      "epoch": 7.012367248535156e-06,
      "model_forward_time": 1.7450718879699707,
      "step": 81353
    },
    {
      "epoch": 7.012367248535156e-06,
      "step": 81353,
      "training_step_time": 7.7362048625946045
    },
    {
      "epoch": 7.013320922851563e-06,
      "model_forward_time": 1.7280793190002441,
      "step": 81354
    },
    {
      "epoch": 7.013320922851563e-06,
      "step": 81354,
      "training_step_time": 8.459494352340698
    },
    {
      "epoch": 7.014274597167969e-06,
      "model_forward_time": 1.7398946285247803,
      "step": 81355
    },
    {
      "epoch": 7.014274597167969e-06,
      "step": 81355,
      "training_step_time": 7.638081312179565
    },
    {
      "epoch": 7.015228271484375e-06,
      "model_forward_time": 1.7426977157592773,
      "step": 81356
    },
    {
      "epoch": 7.015228271484375e-06,
      "step": 81356,
      "training_step_time": 7.5820701122283936
    },
    {
      "epoch": 7.0161819458007814e-06,
      "model_forward_time": 1.7420504093170166,
      "step": 81357
    },
    {
      "epoch": 7.0161819458007814e-06,
      "step": 81357,
      "training_step_time": 6.694343328475952
    },
    {
      "epoch": 7.017135620117187e-06,
      "model_forward_time": 1.7312190532684326,
      "step": 81358
    },
    {
      "epoch": 7.017135620117187e-06,
      "step": 81358,
      "training_step_time": 6.663852214813232
    },
    {
      "epoch": 7.018089294433594e-06,
      "model_forward_time": 1.7400422096252441,
      "step": 81359
    },
    {
      "epoch": 7.018089294433594e-06,
      "step": 81359,
      "training_step_time": 6.625818729400635
    },
    {
      "epoch": 7.01904296875e-06,
      "grad_norm": 0.10099396854639053,
      "learning_rate": 9.20310228463166e-07,
      "loss": 0.0545,
      "step": 81360
    },
    {
      "epoch": 7.01904296875e-06,
      "model_forward_time": 1.741872787475586,
      "step": 81360
    },
    {
      "dynamics_loss_avg": 0.05110581032931805,
      "epoch": 7.01904296875e-06,
      "step": 81360
    },
    {
      "action_loss_avg": 0.0012792198423994706,
      "epoch": 7.01904296875e-06,
      "step": 81360
    },
    {
      "epoch": 7.01904296875e-06,
      "step": 81360,
      "training_step_time": 6.634013652801514
    },
    {
      "epoch": 7.019996643066406e-06,
      "model_forward_time": 2.385923385620117,
      "step": 81361
    },
    {
      "epoch": 7.019996643066406e-06,
      "step": 81361,
      "training_step_time": 7.947989463806152
    },
    {
      "epoch": 7.020950317382813e-06,
      "model_forward_time": 1.7450714111328125,
      "step": 81362
    },
    {
      "epoch": 7.020950317382813e-06,
      "step": 81362,
      "training_step_time": 7.741918325424194
    },
    {
      "epoch": 7.021903991699219e-06,
      "model_forward_time": 1.7461411952972412,
      "step": 81363
    },
    {
      "epoch": 7.021903991699219e-06,
      "step": 81363,
      "training_step_time": 7.477385759353638
    },
    {
      "epoch": 7.022857666015625e-06,
      "model_forward_time": 1.7522594928741455,
      "step": 81364
    },
    {
      "epoch": 7.022857666015625e-06,
      "step": 81364,
      "training_step_time": 7.569101333618164
    },
    {
      "epoch": 7.0238113403320315e-06,
      "model_forward_time": 1.7528314590454102,
      "step": 81365
    },
    {
      "epoch": 7.0238113403320315e-06,
      "step": 81365,
      "training_step_time": 7.629470586776733
    },
    {
      "epoch": 7.024765014648437e-06,
      "model_forward_time": 1.7417597770690918,
      "step": 81366
    },
    {
      "epoch": 7.024765014648437e-06,
      "step": 81366,
      "training_step_time": 6.732539653778076
    },
    {
      "epoch": 7.025718688964844e-06,
      "model_forward_time": 1.7445487976074219,
      "step": 81367
    },
    {
      "epoch": 7.025718688964844e-06,
      "step": 81367,
      "training_step_time": 6.709090948104858
    },
    {
      "epoch": 7.02667236328125e-06,
      "model_forward_time": 1.7566843032836914,
      "step": 81368
    },
    {
      "epoch": 7.02667236328125e-06,
      "step": 81368,
      "training_step_time": 6.732184410095215
    },
    {
      "epoch": 7.027626037597656e-06,
      "model_forward_time": 1.7429883480072021,
      "step": 81369
    },
    {
      "epoch": 7.027626037597656e-06,
      "step": 81369,
      "training_step_time": 6.736784219741821
    },
    {
      "epoch": 7.028579711914063e-06,
      "grad_norm": 0.0992678701877594,
      "learning_rate": 9.193545153795392e-07,
      "loss": 0.054,
      "step": 81370
    },
    {
      "epoch": 7.028579711914063e-06,
      "model_forward_time": 2.3868789672851562,
      "step": 81370
    },
    {
      "dynamics_loss_avg": 0.058523121476173404,
      "epoch": 7.028579711914063e-06,
      "step": 81370
    },
    {
      "action_loss_avg": 0.0019067403802182526,
      "epoch": 7.028579711914063e-06,
      "step": 81370
    },
    {
      "epoch": 7.028579711914063e-06,
      "step": 81370,
      "training_step_time": 7.163276672363281
    },
    {
      "epoch": 7.029533386230469e-06,
      "model_forward_time": 1.789860725402832,
      "step": 81371
    },
    {
      "epoch": 7.029533386230469e-06,
      "step": 81371,
      "training_step_time": 8.574399709701538
    },
    {
      "epoch": 7.030487060546875e-06,
      "model_forward_time": 1.3844516277313232,
      "step": 81372
    },
    {
      "epoch": 7.030487060546875e-06,
      "step": 81372,
      "training_step_time": 7.600353956222534
    },
    {
      "epoch": 7.0314407348632815e-06,
      "model_forward_time": 1.7446279525756836,
      "step": 81373
    },
    {
      "epoch": 7.0314407348632815e-06,
      "step": 81373,
      "training_step_time": 7.544694423675537
    },
    {
      "epoch": 7.032394409179687e-06,
      "model_forward_time": 1.745253562927246,
      "step": 81374
    },
    {
      "epoch": 7.032394409179687e-06,
      "step": 81374,
      "training_step_time": 7.605649471282959
    },
    {
      "epoch": 7.033348083496094e-06,
      "model_forward_time": 1.747401237487793,
      "step": 81375
    },
    {
      "epoch": 7.033348083496094e-06,
      "step": 81375,
      "training_step_time": 6.672428369522095
    },
    {
      "epoch": 7.0343017578125e-06,
      "model_forward_time": 1.3760578632354736,
      "step": 81376
    },
    {
      "epoch": 7.0343017578125e-06,
      "step": 81376,
      "training_step_time": 6.684425592422485
    },
    {
      "epoch": 7.035255432128906e-06,
      "model_forward_time": 1.7437937259674072,
      "step": 81377
    },
    {
      "epoch": 7.035255432128906e-06,
      "step": 81377,
      "training_step_time": 6.723430871963501
    },
    {
      "epoch": 7.036209106445313e-06,
      "model_forward_time": 1.7440645694732666,
      "step": 81378
    },
    {
      "epoch": 7.036209106445313e-06,
      "step": 81378,
      "training_step_time": 6.716461420059204
    },
    {
      "epoch": 7.037162780761718e-06,
      "model_forward_time": 1.7404606342315674,
      "step": 81379
    },
    {
      "epoch": 7.037162780761718e-06,
      "step": 81379,
      "training_step_time": 6.7487101554870605
    },
    {
      "epoch": 7.038116455078125e-06,
      "grad_norm": 0.3448861539363861,
      "learning_rate": 9.183992485491666e-07,
      "loss": 0.0535,
      "step": 81380
    },
    {
      "epoch": 7.038116455078125e-06,
      "model_forward_time": 2.377228021621704,
      "step": 81380
    },
    {
      "dynamics_loss_avg": 0.048666446655988696,
      "epoch": 7.038116455078125e-06,
      "step": 81380
    },
    {
      "action_loss_avg": 0.0012451922986656428,
      "epoch": 7.038116455078125e-06,
      "step": 81380
    },
    {
      "epoch": 7.038116455078125e-06,
      "step": 81380,
      "training_step_time": 10.929572105407715
    },
    {
      "epoch": 7.0390701293945315e-06,
      "model_forward_time": 1.7822513580322266,
      "step": 81381
    },
    {
      "epoch": 7.0390701293945315e-06,
      "step": 81381,
      "training_step_time": 7.7233357429504395
    },
    {
      "epoch": 7.040023803710937e-06,
      "model_forward_time": 1.7411441802978516,
      "step": 81382
    },
    {
      "epoch": 7.040023803710937e-06,
      "step": 81382,
      "training_step_time": 7.652918577194214
    },
    {
      "epoch": 7.040977478027344e-06,
      "model_forward_time": 1.7453022003173828,
      "step": 81383
    },
    {
      "epoch": 7.040977478027344e-06,
      "step": 81383,
      "training_step_time": 7.693294048309326
    },
    {
      "epoch": 7.04193115234375e-06,
      "model_forward_time": 1.7414591312408447,
      "step": 81384
    },
    {
      "epoch": 7.04193115234375e-06,
      "step": 81384,
      "training_step_time": 6.7259016036987305
    },
    {
      "epoch": 7.042884826660156e-06,
      "model_forward_time": 1.7613675594329834,
      "step": 81385
    },
    {
      "epoch": 7.042884826660156e-06,
      "step": 81385,
      "training_step_time": 6.674923658370972
    },
    {
      "epoch": 7.043838500976563e-06,
      "model_forward_time": 1.747635841369629,
      "step": 81386
    },
    {
      "epoch": 7.043838500976563e-06,
      "step": 81386,
      "training_step_time": 6.691380023956299
    },
    {
      "epoch": 7.044792175292968e-06,
      "model_forward_time": 1.7424333095550537,
      "step": 81387
    },
    {
      "epoch": 7.044792175292968e-06,
      "step": 81387,
      "training_step_time": 6.632911682128906
    },
    {
      "epoch": 7.045745849609375e-06,
      "model_forward_time": 1.7382259368896484,
      "step": 81388
    },
    {
      "epoch": 7.045745849609375e-06,
      "step": 81388,
      "training_step_time": 6.639399528503418
    },
    {
      "epoch": 7.0466995239257815e-06,
      "model_forward_time": 2.4245920181274414,
      "step": 81389
    },
    {
      "epoch": 7.0466995239257815e-06,
      "step": 81389,
      "training_step_time": 9.147315740585327
    },
    {
      "epoch": 7.047653198242187e-06,
      "grad_norm": 0.1408029943704605,
      "learning_rate": 9.174444280765155e-07,
      "loss": 0.0529,
      "step": 81390
    },
    {
      "epoch": 7.047653198242187e-06,
      "model_forward_time": 1.7446091175079346,
      "step": 81390
    },
    {
      "dynamics_loss_avg": 0.07061249185353517,
      "epoch": 7.047653198242187e-06,
      "step": 81390
    },
    {
      "action_loss_avg": 0.0017565803544130176,
      "epoch": 7.047653198242187e-06,
      "step": 81390
    },
    {
      "epoch": 7.047653198242187e-06,
      "step": 81390,
      "training_step_time": 7.737879514694214
    },
    {
      "epoch": 7.048606872558594e-06,
      "model_forward_time": 1.785430669784546,
      "step": 81391
    },
    {
      "epoch": 7.048606872558594e-06,
      "step": 81391,
      "training_step_time": 7.692266464233398
    },
    {
      "epoch": 7.049560546875e-06,
      "model_forward_time": 1.7391531467437744,
      "step": 81392
    },
    {
      "epoch": 7.049560546875e-06,
      "step": 81392,
      "training_step_time": 7.670503616333008
    },
    {
      "epoch": 7.050514221191406e-06,
      "model_forward_time": 1.7372004985809326,
      "step": 81393
    },
    {
      "epoch": 7.050514221191406e-06,
      "step": 81393,
      "training_step_time": 7.6140453815460205
    },
    {
      "epoch": 7.051467895507813e-06,
      "model_forward_time": 1.7433106899261475,
      "step": 81394
    },
    {
      "epoch": 7.051467895507813e-06,
      "step": 81394,
      "training_step_time": 6.757096290588379
    },
    {
      "epoch": 7.052421569824218e-06,
      "model_forward_time": 1.7427189350128174,
      "step": 81395
    },
    {
      "epoch": 7.052421569824218e-06,
      "step": 81395,
      "training_step_time": 6.777851104736328
    },
    {
      "epoch": 7.053375244140625e-06,
      "model_forward_time": 1.7474288940429688,
      "step": 81396
    },
    {
      "epoch": 7.053375244140625e-06,
      "step": 81396,
      "training_step_time": 6.787278175354004
    },
    {
      "epoch": 7.0543289184570315e-06,
      "model_forward_time": 1.7441225051879883,
      "step": 81397
    },
    {
      "epoch": 7.0543289184570315e-06,
      "step": 81397,
      "training_step_time": 6.790360689163208
    },
    {
      "epoch": 7.055282592773437e-06,
      "model_forward_time": 2.4065358638763428,
      "step": 81398
    },
    {
      "epoch": 7.055282592773437e-06,
      "step": 81398,
      "training_step_time": 7.681313753128052
    },
    {
      "epoch": 7.056236267089844e-06,
      "model_forward_time": 1.8125264644622803,
      "step": 81399
    },
    {
      "epoch": 7.056236267089844e-06,
      "step": 81399,
      "training_step_time": 9.189628601074219
    },
    {
      "epoch": 7.05718994140625e-06,
      "grad_norm": 0.10505364835262299,
      "learning_rate": 9.164900540660032e-07,
      "loss": 0.0525,
      "step": 81400
    },
    {
      "epoch": 7.05718994140625e-06,
      "model_forward_time": 1.744910717010498,
      "step": 81400
    },
    {
      "dynamics_loss_avg": 0.05785692408680916,
      "epoch": 7.05718994140625e-06,
      "step": 81400
    },
    {
      "action_loss_avg": 0.001743290259037167,
      "epoch": 7.05718994140625e-06,
      "step": 81400
    },
    {
      "epoch": 7.05718994140625e-06,
      "step": 81400,
      "training_step_time": 8.55814504623413
    },
    {
      "epoch": 7.058143615722656e-06,
      "model_forward_time": 1.7458221912384033,
      "step": 81401
    },
    {
      "epoch": 7.058143615722656e-06,
      "step": 81401,
      "training_step_time": 8.266983032226562
    },
    {
      "epoch": 7.059097290039063e-06,
      "model_forward_time": 1.74515700340271,
      "step": 81402
    },
    {
      "epoch": 7.059097290039063e-06,
      "step": 81402,
      "training_step_time": 7.587477207183838
    },
    {
      "epoch": 7.060050964355468e-06,
      "model_forward_time": 1.7540273666381836,
      "step": 81403
    },
    {
      "epoch": 7.060050964355468e-06,
      "step": 81403,
      "training_step_time": 6.685925245285034
    },
    {
      "epoch": 7.061004638671875e-06,
      "model_forward_time": 1.7483196258544922,
      "step": 81404
    },
    {
      "epoch": 7.061004638671875e-06,
      "step": 81404,
      "training_step_time": 6.685312986373901
    },
    {
      "epoch": 7.0619583129882815e-06,
      "model_forward_time": 1.7497804164886475,
      "step": 81405
    },
    {
      "epoch": 7.0619583129882815e-06,
      "step": 81405,
      "training_step_time": 6.700093030929565
    },
    {
      "epoch": 7.062911987304687e-06,
      "model_forward_time": 1.750302791595459,
      "step": 81406
    },
    {
      "epoch": 7.062911987304687e-06,
      "step": 81406,
      "training_step_time": 6.664127826690674
    },
    {
      "epoch": 7.063865661621094e-06,
      "model_forward_time": 2.353574275970459,
      "step": 81407
    },
    {
      "epoch": 7.063865661621094e-06,
      "step": 81407,
      "training_step_time": 7.095463037490845
    },
    {
      "epoch": 7.0648193359375e-06,
      "model_forward_time": 1.781581163406372,
      "step": 81408
    },
    {
      "epoch": 7.0648193359375e-06,
      "step": 81408,
      "training_step_time": 10.2848379611969
    },
    {
      "epoch": 7.065773010253906e-06,
      "model_forward_time": 1.7555568218231201,
      "step": 81409
    },
    {
      "epoch": 7.065773010253906e-06,
      "step": 81409,
      "training_step_time": 8.244140863418579
    },
    {
      "epoch": 7.066726684570313e-06,
      "grad_norm": 0.07523752748966217,
      "learning_rate": 9.155361266220009e-07,
      "loss": 0.0533,
      "step": 81410
    },
    {
      "epoch": 7.066726684570313e-06,
      "model_forward_time": 1.7461795806884766,
      "step": 81410
    },
    {
      "dynamics_loss_avg": 0.053783705458045006,
      "epoch": 7.066726684570313e-06,
      "step": 81410
    },
    {
      "action_loss_avg": 0.0014868960541207343,
      "epoch": 7.066726684570313e-06,
      "step": 81410
    },
    {
      "epoch": 7.066726684570313e-06,
      "step": 81410,
      "training_step_time": 7.510318040847778
    },
    {
      "epoch": 7.067680358886718e-06,
      "model_forward_time": 1.7557320594787598,
      "step": 81411
    },
    {
      "epoch": 7.067680358886718e-06,
      "step": 81411,
      "training_step_time": 7.5571393966674805
    },
    {
      "epoch": 7.068634033203125e-06,
      "model_forward_time": 1.7469110488891602,
      "step": 81412
    },
    {
      "epoch": 7.068634033203125e-06,
      "step": 81412,
      "training_step_time": 6.637145042419434
    },
    {
      "epoch": 7.0695877075195315e-06,
      "model_forward_time": 1.7474415302276611,
      "step": 81413
    },
    {
      "epoch": 7.0695877075195315e-06,
      "step": 81413,
      "training_step_time": 6.6188271045684814
    },
    {
      "epoch": 7.070541381835937e-06,
      "model_forward_time": 1.7471163272857666,
      "step": 81414
    },
    {
      "epoch": 7.070541381835937e-06,
      "step": 81414,
      "training_step_time": 6.649392604827881
    },
    {
      "epoch": 7.071495056152344e-06,
      "model_forward_time": 1.7415456771850586,
      "step": 81415
    },
    {
      "epoch": 7.071495056152344e-06,
      "step": 81415,
      "training_step_time": 6.621679306030273
    },
    {
      "epoch": 7.07244873046875e-06,
      "model_forward_time": 1.7444548606872559,
      "step": 81416
    },
    {
      "epoch": 7.07244873046875e-06,
      "step": 81416,
      "training_step_time": 6.626890659332275
    },
    {
      "epoch": 7.073402404785156e-06,
      "model_forward_time": 2.390223503112793,
      "step": 81417
    },
    {
      "epoch": 7.073402404785156e-06,
      "step": 81417,
      "training_step_time": 8.773494005203247
    },
    {
      "epoch": 7.074356079101563e-06,
      "model_forward_time": 1.7540879249572754,
      "step": 81418
    },
    {
      "epoch": 7.074356079101563e-06,
      "step": 81418,
      "training_step_time": 7.8028271198272705
    },
    {
      "epoch": 7.075309753417968e-06,
      "model_forward_time": 1.754765272140503,
      "step": 81419
    },
    {
      "epoch": 7.075309753417968e-06,
      "step": 81419,
      "training_step_time": 7.5897533893585205
    },
    {
      "epoch": 7.076263427734375e-06,
      "grad_norm": 0.07909616082906723,
      "learning_rate": 9.145826458488255e-07,
      "loss": 0.0542,
      "step": 81420
    },
    {
      "epoch": 7.076263427734375e-06,
      "model_forward_time": 1.3779659271240234,
      "step": 81420
    },
    {
      "dynamics_loss_avg": 0.057833429984748365,
      "epoch": 7.076263427734375e-06,
      "step": 81420
    },
    {
      "action_loss_avg": 0.001736067491583526,
      "epoch": 7.076263427734375e-06,
      "step": 81420
    },
    {
      "epoch": 7.076263427734375e-06,
      "step": 81420,
      "training_step_time": 7.683715581893921
    },
    {
      "epoch": 7.0772171020507815e-06,
      "model_forward_time": 1.756256341934204,
      "step": 81421
    },
    {
      "epoch": 7.0772171020507815e-06,
      "step": 81421,
      "training_step_time": 6.9014294147491455
    },
    {
      "epoch": 7.078170776367187e-06,
      "model_forward_time": 1.7526159286499023,
      "step": 81422
    },
    {
      "epoch": 7.078170776367187e-06,
      "step": 81422,
      "training_step_time": 6.874472379684448
    },
    {
      "epoch": 7.079124450683594e-06,
      "model_forward_time": 1.752793550491333,
      "step": 81423
    },
    {
      "epoch": 7.079124450683594e-06,
      "step": 81423,
      "training_step_time": 6.722524642944336
    },
    {
      "epoch": 7.080078125e-06,
      "model_forward_time": 1.7470149993896484,
      "step": 81424
    },
    {
      "epoch": 7.080078125e-06,
      "step": 81424,
      "training_step_time": 6.725095987319946
    },
    {
      "epoch": 7.081031799316406e-06,
      "model_forward_time": 1.7557122707366943,
      "step": 81425
    },
    {
      "epoch": 7.081031799316406e-06,
      "step": 81425,
      "training_step_time": 6.694281339645386
    },
    {
      "epoch": 7.081985473632813e-06,
      "model_forward_time": 2.428504467010498,
      "step": 81426
    },
    {
      "epoch": 7.081985473632813e-06,
      "step": 81426,
      "training_step_time": 11.167769432067871
    },
    {
      "epoch": 7.082939147949218e-06,
      "model_forward_time": 1.753312587738037,
      "step": 81427
    },
    {
      "epoch": 7.082939147949218e-06,
      "step": 81427,
      "training_step_time": 8.576662540435791
    },
    {
      "epoch": 7.083892822265625e-06,
      "model_forward_time": 1.7479195594787598,
      "step": 81428
    },
    {
      "epoch": 7.083892822265625e-06,
      "step": 81428,
      "training_step_time": 7.657790899276733
    },
    {
      "epoch": 7.0848464965820315e-06,
      "model_forward_time": 1.7472522258758545,
      "step": 81429
    },
    {
      "epoch": 7.0848464965820315e-06,
      "step": 81429,
      "training_step_time": 7.621668100357056
    },
    {
      "epoch": 7.085800170898437e-06,
      "grad_norm": 0.09776512533426285,
      "learning_rate": 9.13629611850751e-07,
      "loss": 0.0528,
      "step": 81430
    },
    {
      "epoch": 7.085800170898437e-06,
      "model_forward_time": 1.7345926761627197,
      "step": 81430
    },
    {
      "dynamics_loss_avg": 0.05684369541704655,
      "epoch": 7.085800170898437e-06,
      "step": 81430
    },
    {
      "action_loss_avg": 0.001817879028385505,
      "epoch": 7.085800170898437e-06,
      "step": 81430
    },
    {
      "epoch": 7.085800170898437e-06,
      "step": 81430,
      "training_step_time": 6.685961008071899
    },
    {
      "epoch": 7.086753845214844e-06,
      "model_forward_time": 1.7395148277282715,
      "step": 81431
    },
    {
      "epoch": 7.086753845214844e-06,
      "step": 81431,
      "training_step_time": 6.655145883560181
    },
    {
      "epoch": 7.08770751953125e-06,
      "model_forward_time": 1.7467050552368164,
      "step": 81432
    },
    {
      "epoch": 7.08770751953125e-06,
      "step": 81432,
      "training_step_time": 6.669055461883545
    },
    {
      "epoch": 7.088661193847656e-06,
      "model_forward_time": 1.7401220798492432,
      "step": 81433
    },
    {
      "epoch": 7.088661193847656e-06,
      "step": 81433,
      "training_step_time": 6.680180788040161
    },
    {
      "epoch": 7.089614868164063e-06,
      "model_forward_time": 1.7399802207946777,
      "step": 81434
    },
    {
      "epoch": 7.089614868164063e-06,
      "step": 81434,
      "training_step_time": 6.6791558265686035
    },
    {
      "epoch": 7.090568542480468e-06,
      "model_forward_time": 2.4277758598327637,
      "step": 81435
    },
    {
      "epoch": 7.090568542480468e-06,
      "step": 81435,
      "training_step_time": 11.350014448165894
    },
    {
      "epoch": 7.091522216796875e-06,
      "model_forward_time": 1.8157517910003662,
      "step": 81436
    },
    {
      "epoch": 7.091522216796875e-06,
      "step": 81436,
      "training_step_time": 8.7971510887146
    },
    {
      "epoch": 7.0924758911132815e-06,
      "model_forward_time": 1.7590031623840332,
      "step": 81437
    },
    {
      "epoch": 7.0924758911132815e-06,
      "step": 81437,
      "training_step_time": 7.653499126434326
    },
    {
      "epoch": 7.093429565429687e-06,
      "model_forward_time": 1.7453083992004395,
      "step": 81438
    },
    {
      "epoch": 7.093429565429687e-06,
      "step": 81438,
      "training_step_time": 8.624852657318115
    },
    {
      "epoch": 7.094383239746094e-06,
      "model_forward_time": 1.7453255653381348,
      "step": 81439
    },
    {
      "epoch": 7.094383239746094e-06,
      "step": 81439,
      "training_step_time": 7.419027328491211
    },
    {
      "epoch": 7.0953369140625e-06,
      "grad_norm": 0.08911553770303726,
      "learning_rate": 9.126770247319966e-07,
      "loss": 0.0524,
      "step": 81440
    },
    {
      "epoch": 7.0953369140625e-06,
      "model_forward_time": 1.7489228248596191,
      "step": 81440
    },
    {
      "dynamics_loss_avg": 0.05559980655089021,
      "epoch": 7.0953369140625e-06,
      "step": 81440
    },
    {
      "action_loss_avg": 0.001579208590555936,
      "epoch": 7.0953369140625e-06,
      "step": 81440
    },
    {
      "epoch": 7.0953369140625e-06,
      "step": 81440,
      "training_step_time": 6.684355020523071
    },
    {
      "epoch": 7.096290588378906e-06,
      "model_forward_time": 1.7412331104278564,
      "step": 81441
    },
    {
      "epoch": 7.096290588378906e-06,
      "step": 81441,
      "training_step_time": 6.680931568145752
    },
    {
      "epoch": 7.097244262695313e-06,
      "model_forward_time": 1.73848557472229,
      "step": 81442
    },
    {
      "epoch": 7.097244262695313e-06,
      "step": 81442,
      "training_step_time": 6.70235013961792
    },
    {
      "epoch": 7.098197937011718e-06,
      "model_forward_time": 1.7795588970184326,
      "step": 81443
    },
    {
      "epoch": 7.098197937011718e-06,
      "step": 81443,
      "training_step_time": 6.661391973495483
    },
    {
      "epoch": 7.099151611328125e-06,
      "model_forward_time": 2.364717483520508,
      "step": 81444
    },
    {
      "epoch": 7.099151611328125e-06,
      "step": 81444,
      "training_step_time": 7.754039764404297
    },
    {
      "epoch": 7.1001052856445315e-06,
      "model_forward_time": 1.8120477199554443,
      "step": 81445
    },
    {
      "epoch": 7.1001052856445315e-06,
      "step": 81445,
      "training_step_time": 9.66288685798645
    },
    {
      "epoch": 7.101058959960937e-06,
      "model_forward_time": 1.7483813762664795,
      "step": 81446
    },
    {
      "epoch": 7.101058959960937e-06,
      "step": 81446,
      "training_step_time": 7.772326469421387
    },
    {
      "epoch": 7.102012634277344e-06,
      "model_forward_time": 1.7470321655273438,
      "step": 81447
    },
    {
      "epoch": 7.102012634277344e-06,
      "step": 81447,
      "training_step_time": 7.636240005493164
    },
    {
      "epoch": 7.10296630859375e-06,
      "model_forward_time": 1.7439663410186768,
      "step": 81448
    },
    {
      "epoch": 7.10296630859375e-06,
      "step": 81448,
      "training_step_time": 7.6475794315338135
    },
    {
      "epoch": 7.103919982910156e-06,
      "model_forward_time": 1.8252441883087158,
      "step": 81449
    },
    {
      "epoch": 7.103919982910156e-06,
      "step": 81449,
      "training_step_time": 6.725404500961304
    },
    {
      "epoch": 7.104873657226563e-06,
      "grad_norm": 0.08744238317012787,
      "learning_rate": 9.11724884596738e-07,
      "loss": 0.0539,
      "step": 81450
    },
    {
      "epoch": 7.104873657226563e-06,
      "model_forward_time": 1.730034589767456,
      "step": 81450
    },
    {
      "dynamics_loss_avg": 0.05573758706450462,
      "epoch": 7.104873657226563e-06,
      "step": 81450
    },
    {
      "action_loss_avg": 0.001180953427683562,
      "epoch": 7.104873657226563e-06,
      "step": 81450
    },
    {
      "epoch": 7.104873657226563e-06,
      "step": 81450,
      "training_step_time": 6.7151854038238525
    },
    {
      "epoch": 7.105827331542968e-06,
      "model_forward_time": 1.7472434043884277,
      "step": 81451
    },
    {
      "epoch": 7.105827331542968e-06,
      "step": 81451,
      "training_step_time": 6.714811086654663
    },
    {
      "epoch": 7.106781005859375e-06,
      "model_forward_time": 1.380889654159546,
      "step": 81452
    },
    {
      "epoch": 7.106781005859375e-06,
      "step": 81452,
      "training_step_time": 6.705458641052246
    },
    {
      "epoch": 7.1077346801757815e-06,
      "model_forward_time": 1.7436633110046387,
      "step": 81453
    },
    {
      "epoch": 7.1077346801757815e-06,
      "step": 81453,
      "training_step_time": 9.444211959838867
    },
    {
      "epoch": 7.108688354492187e-06,
      "model_forward_time": 2.428060531616211,
      "step": 81454
    },
    {
      "epoch": 7.108688354492187e-06,
      "step": 81454,
      "training_step_time": 9.413691997528076
    },
    {
      "epoch": 7.109642028808594e-06,
      "model_forward_time": 1.7498724460601807,
      "step": 81455
    },
    {
      "epoch": 7.109642028808594e-06,
      "step": 81455,
      "training_step_time": 7.767988681793213
    },
    {
      "epoch": 7.110595703125e-06,
      "model_forward_time": 1.737483024597168,
      "step": 81456
    },
    {
      "epoch": 7.110595703125e-06,
      "step": 81456,
      "training_step_time": 7.579907655715942
    },
    {
      "epoch": 7.111549377441406e-06,
      "model_forward_time": 1.7392332553863525,
      "step": 81457
    },
    {
      "epoch": 7.111549377441406e-06,
      "step": 81457,
      "training_step_time": 7.6966283321380615
    },
    {
      "epoch": 7.112503051757813e-06,
      "model_forward_time": 1.3753459453582764,
      "step": 81458
    },
    {
      "epoch": 7.112503051757813e-06,
      "step": 81458,
      "training_step_time": 6.6998610496521
    },
    {
      "epoch": 7.113456726074218e-06,
      "model_forward_time": 1.7519447803497314,
      "step": 81459
    },
    {
      "epoch": 7.113456726074218e-06,
      "step": 81459,
      "training_step_time": 6.683725357055664
    },
    {
      "epoch": 7.114410400390625e-06,
      "grad_norm": 0.070311039686203,
      "learning_rate": 9.107731915490997e-07,
      "loss": 0.0551,
      "step": 81460
    },
    {
      "epoch": 7.114410400390625e-06,
      "model_forward_time": 1.7378499507904053,
      "step": 81460
    },
    {
      "dynamics_loss_avg": 0.046205036249011754,
      "epoch": 7.114410400390625e-06,
      "step": 81460
    },
    {
      "action_loss_avg": 0.0014860230003250763,
      "epoch": 7.114410400390625e-06,
      "step": 81460
    },
    {
      "epoch": 7.114410400390625e-06,
      "step": 81460,
      "training_step_time": 6.651740074157715
    },
    {
      "epoch": 7.1153640747070315e-06,
      "model_forward_time": 1.724804162979126,
      "step": 81461
    },
    {
      "epoch": 7.1153640747070315e-06,
      "step": 81461,
      "training_step_time": 6.571298837661743
    },
    {
      "epoch": 7.116317749023437e-06,
      "model_forward_time": 1.036102056503296,
      "step": 81462
    },
    {
      "epoch": 7.116317749023437e-06,
      "step": 81462,
      "training_step_time": 8.432544708251953
    },
    {
      "epoch": 7.117271423339844e-06,
      "model_forward_time": 1.755526065826416,
      "step": 81463
    },
    {
      "epoch": 7.117271423339844e-06,
      "step": 81463,
      "training_step_time": 10.063936471939087
    },
    {
      "epoch": 7.11822509765625e-06,
      "model_forward_time": 1.7738456726074219,
      "step": 81464
    },
    {
      "epoch": 7.11822509765625e-06,
      "step": 81464,
      "training_step_time": 9.080901622772217
    },
    {
      "epoch": 7.119178771972656e-06,
      "model_forward_time": 1.741856575012207,
      "step": 81465
    },
    {
      "epoch": 7.119178771972656e-06,
      "step": 81465,
      "training_step_time": 7.631871700286865
    },
    {
      "epoch": 7.120132446289063e-06,
      "model_forward_time": 1.3463470935821533,
      "step": 81466
    },
    {
      "epoch": 7.120132446289063e-06,
      "step": 81466,
      "training_step_time": 7.660958528518677
    },
    {
      "epoch": 7.121086120605468e-06,
      "model_forward_time": 1.3577995300292969,
      "step": 81467
    },
    {
      "epoch": 7.121086120605468e-06,
      "step": 81467,
      "training_step_time": 7.3256754875183105
    },
    {
      "epoch": 7.122039794921875e-06,
      "model_forward_time": 1.7353193759918213,
      "step": 81468
    },
    {
      "epoch": 7.122039794921875e-06,
      "step": 81468,
      "training_step_time": 6.782796382904053
    },
    {
      "epoch": 7.1229934692382815e-06,
      "model_forward_time": 1.7402310371398926,
      "step": 81469
    },
    {
      "epoch": 7.1229934692382815e-06,
      "step": 81469,
      "training_step_time": 6.693880558013916
    },
    {
      "epoch": 7.123947143554687e-06,
      "grad_norm": 0.08110008388757706,
      "learning_rate": 9.098219456931578e-07,
      "loss": 0.054,
      "step": 81470
    },
    {
      "epoch": 7.123947143554687e-06,
      "model_forward_time": 1.7260758876800537,
      "step": 81470
    },
    {
      "dynamics_loss_avg": 0.043525987025350335,
      "epoch": 7.123947143554687e-06,
      "step": 81470
    },
    {
      "action_loss_avg": 0.0008141473022988066,
      "epoch": 7.123947143554687e-06,
      "step": 81470
    },
    {
      "epoch": 7.123947143554687e-06,
      "step": 81470,
      "training_step_time": 6.687106370925903
    },
    {
      "epoch": 7.124900817871094e-06,
      "model_forward_time": 1.736966609954834,
      "step": 81471
    },
    {
      "epoch": 7.124900817871094e-06,
      "step": 81471,
      "training_step_time": 8.292349576950073
    },
    {
      "epoch": 7.1258544921875e-06,
      "model_forward_time": 1.7349214553833008,
      "step": 81472
    },
    {
      "epoch": 7.1258544921875e-06,
      "step": 81472,
      "training_step_time": 8.403563022613525
    },
    {
      "epoch": 7.126808166503906e-06,
      "model_forward_time": 2.3484115600585938,
      "step": 81473
    },
    {
      "epoch": 7.126808166503906e-06,
      "step": 81473,
      "training_step_time": 8.938298225402832
    },
    {
      "epoch": 7.127761840820313e-06,
      "model_forward_time": 1.7401976585388184,
      "step": 81474
    },
    {
      "epoch": 7.127761840820313e-06,
      "step": 81474,
      "training_step_time": 8.25195860862732
    },
    {
      "epoch": 7.128715515136718e-06,
      "model_forward_time": 1.784513235092163,
      "step": 81475
    },
    {
      "epoch": 7.128715515136718e-06,
      "step": 81475,
      "training_step_time": 7.672578811645508
    },
    {
      "epoch": 7.129669189453125e-06,
      "model_forward_time": 1.7447590827941895,
      "step": 81476
    },
    {
      "epoch": 7.129669189453125e-06,
      "step": 81476,
      "training_step_time": 7.630180597305298
    },
    {
      "epoch": 7.1306228637695315e-06,
      "model_forward_time": 1.7548792362213135,
      "step": 81477
    },
    {
      "epoch": 7.1306228637695315e-06,
      "step": 81477,
      "training_step_time": 6.728838682174683
    },
    {
      "epoch": 7.131576538085937e-06,
      "model_forward_time": 1.7542743682861328,
      "step": 81478
    },
    {
      "epoch": 7.131576538085937e-06,
      "step": 81478,
      "training_step_time": 6.716556072235107
    },
    {
      "epoch": 7.132530212402344e-06,
      "model_forward_time": 1.7511024475097656,
      "step": 81479
    },
    {
      "epoch": 7.132530212402344e-06,
      "step": 81479,
      "training_step_time": 6.745119094848633
    },
    {
      "epoch": 7.13348388671875e-06,
      "grad_norm": 0.12277711927890778,
      "learning_rate": 9.088711471329375e-07,
      "loss": 0.0545,
      "step": 81480
    },
    {
      "epoch": 7.13348388671875e-06,
      "model_forward_time": 1.766380786895752,
      "step": 81480
    },
    {
      "dynamics_loss_avg": 0.03904458619654179,
      "epoch": 7.13348388671875e-06,
      "step": 81480
    },
    {
      "action_loss_avg": 0.0016730285831727087,
      "epoch": 7.13348388671875e-06,
      "step": 81480
    },
    {
      "epoch": 7.13348388671875e-06,
      "step": 81480,
      "training_step_time": 7.251459121704102
    },
    {
      "epoch": 7.134437561035156e-06,
      "model_forward_time": 1.7447712421417236,
      "step": 81481
    },
    {
      "epoch": 7.134437561035156e-06,
      "step": 81481,
      "training_step_time": 8.278356313705444
    },
    {
      "epoch": 7.135391235351563e-06,
      "model_forward_time": 2.4441847801208496,
      "step": 81482
    },
    {
      "epoch": 7.135391235351563e-06,
      "step": 81482,
      "training_step_time": 9.57775616645813
    },
    {
      "epoch": 7.136344909667968e-06,
      "model_forward_time": 1.7557153701782227,
      "step": 81483
    },
    {
      "epoch": 7.136344909667968e-06,
      "step": 81483,
      "training_step_time": 7.702434778213501
    },
    {
      "epoch": 7.137298583984375e-06,
      "model_forward_time": 1.7511801719665527,
      "step": 81484
    },
    {
      "epoch": 7.137298583984375e-06,
      "step": 81484,
      "training_step_time": 7.651952743530273
    },
    {
      "epoch": 7.1382522583007816e-06,
      "model_forward_time": 1.748945713043213,
      "step": 81485
    },
    {
      "epoch": 7.1382522583007816e-06,
      "step": 81485,
      "training_step_time": 7.604334115982056
    },
    {
      "epoch": 7.139205932617187e-06,
      "model_forward_time": 1.7443342208862305,
      "step": 81486
    },
    {
      "epoch": 7.139205932617187e-06,
      "step": 81486,
      "training_step_time": 7.285592555999756
    },
    {
      "epoch": 7.140159606933594e-06,
      "model_forward_time": 1.7487602233886719,
      "step": 81487
    },
    {
      "epoch": 7.140159606933594e-06,
      "step": 81487,
      "training_step_time": 6.775240421295166
    },
    {
      "epoch": 7.14111328125e-06,
      "model_forward_time": 1.369743824005127,
      "step": 81488
    },
    {
      "epoch": 7.14111328125e-06,
      "step": 81488,
      "training_step_time": 6.718703031539917
    },
    {
      "epoch": 7.142066955566406e-06,
      "model_forward_time": 1.7496623992919922,
      "step": 81489
    },
    {
      "epoch": 7.142066955566406e-06,
      "step": 81489,
      "training_step_time": 7.047608375549316
    },
    {
      "epoch": 7.143020629882813e-06,
      "grad_norm": 0.06703974306583405,
      "learning_rate": 9.079207959724185e-07,
      "loss": 0.0567,
      "step": 81490
    },
    {
      "epoch": 7.143020629882813e-06,
      "model_forward_time": 1.7464931011199951,
      "step": 81490
    },
    {
      "dynamics_loss_avg": 0.05542413499206304,
      "epoch": 7.143020629882813e-06,
      "step": 81490
    },
    {
      "action_loss_avg": 0.0013510799501091242,
      "epoch": 7.143020629882813e-06,
      "step": 81490
    },
    {
      "epoch": 7.143020629882813e-06,
      "step": 81490,
      "training_step_time": 9.058964729309082
    },
    {
      "epoch": 7.1439743041992184e-06,
      "model_forward_time": 1.8620812892913818,
      "step": 81491
    },
    {
      "epoch": 7.1439743041992184e-06,
      "step": 81491,
      "training_step_time": 9.415326356887817
    },
    {
      "epoch": 7.144927978515625e-06,
      "model_forward_time": 1.749405860900879,
      "step": 81492
    },
    {
      "epoch": 7.144927978515625e-06,
      "step": 81492,
      "training_step_time": 9.33879566192627
    },
    {
      "epoch": 7.1458816528320316e-06,
      "model_forward_time": 1.7490668296813965,
      "step": 81493
    },
    {
      "epoch": 7.1458816528320316e-06,
      "step": 81493,
      "training_step_time": 7.674706935882568
    },
    {
      "epoch": 7.146835327148437e-06,
      "model_forward_time": 1.7598259449005127,
      "step": 81494
    },
    {
      "epoch": 7.146835327148437e-06,
      "step": 81494,
      "training_step_time": 7.623661279678345
    },
    {
      "epoch": 7.147789001464844e-06,
      "model_forward_time": 1.7773990631103516,
      "step": 81495
    },
    {
      "epoch": 7.147789001464844e-06,
      "step": 81495,
      "training_step_time": 7.482099294662476
    },
    {
      "epoch": 7.14874267578125e-06,
      "model_forward_time": 1.7659482955932617,
      "step": 81496
    },
    {
      "epoch": 7.14874267578125e-06,
      "step": 81496,
      "training_step_time": 6.738856792449951
    },
    {
      "epoch": 7.149696350097656e-06,
      "model_forward_time": 1.7408277988433838,
      "step": 81497
    },
    {
      "epoch": 7.149696350097656e-06,
      "step": 81497,
      "training_step_time": 6.785845518112183
    },
    {
      "epoch": 7.150650024414063e-06,
      "model_forward_time": 1.7776525020599365,
      "step": 81498
    },
    {
      "epoch": 7.150650024414063e-06,
      "step": 81498,
      "training_step_time": 7.04774022102356
    },
    {
      "epoch": 7.1516036987304684e-06,
      "model_forward_time": 1.7653605937957764,
      "step": 81499
    },
    {
      "epoch": 7.1516036987304684e-06,
      "step": 81499,
      "training_step_time": 11.342318773269653
    },
    {
      "epoch": 7.152557373046875e-06,
      "grad_norm": 0.08299895375967026,
      "learning_rate": 9.06970892315528e-07,
      "loss": 0.0553,
      "step": 81500
    },
    {
      "epoch": 7.152557373046875e-06,
      "model_forward_time": 1.7880659103393555,
      "step": 81500
    },
    {
      "dynamics_loss_avg": 0.06688877204433083,
      "epoch": 7.152557373046875e-06,
      "step": 81500
    },
    {
      "action_loss_avg": 0.0019605480891186746,
      "epoch": 7.152557373046875e-06,
      "step": 81500
    },
    {
      "epoch": 7.152557373046875e-06,
      "step": 81500,
      "training_step_time": 7.755184888839722
    },
    {
      "epoch": 7.1535110473632816e-06,
      "model_forward_time": 2.407228469848633,
      "step": 81501
    },
    {
      "epoch": 7.1535110473632816e-06,
      "step": 81501,
      "training_step_time": 7.792437791824341
    },
    {
      "epoch": 7.154464721679687e-06,
      "model_forward_time": 1.7433521747589111,
      "step": 81502
    },
    {
      "epoch": 7.154464721679687e-06,
      "step": 81502,
      "training_step_time": 7.716960430145264
    },
    {
      "epoch": 7.155418395996094e-06,
      "model_forward_time": 1.7477471828460693,
      "step": 81503
    },
    {
      "epoch": 7.155418395996094e-06,
      "step": 81503,
      "training_step_time": 7.589131116867065
    },
    {
      "epoch": 7.1563720703125e-06,
      "model_forward_time": 1.0385234355926514,
      "step": 81504
    },
    {
      "epoch": 7.1563720703125e-06,
      "step": 81504,
      "training_step_time": 7.1820032596588135
    },
    {
      "epoch": 7.157325744628906e-06,
      "model_forward_time": 1.7546272277832031,
      "step": 81505
    },
    {
      "epoch": 7.157325744628906e-06,
      "step": 81505,
      "training_step_time": 6.6790995597839355
    },
    {
      "epoch": 7.158279418945313e-06,
      "model_forward_time": 1.7485558986663818,
      "step": 81506
    },
    {
      "epoch": 7.158279418945313e-06,
      "step": 81506,
      "training_step_time": 6.690162897109985
    },
    {
      "epoch": 7.1592330932617184e-06,
      "model_forward_time": 1.7553164958953857,
      "step": 81507
    },
    {
      "epoch": 7.1592330932617184e-06,
      "step": 81507,
      "training_step_time": 6.984007120132446
    },
    {
      "epoch": 7.160186767578125e-06,
      "model_forward_time": 1.7565844058990479,
      "step": 81508
    },
    {
      "epoch": 7.160186767578125e-06,
      "step": 81508,
      "training_step_time": 8.080565690994263
    },
    {
      "epoch": 7.1611404418945316e-06,
      "model_forward_time": 1.7597053050994873,
      "step": 81509
    },
    {
      "epoch": 7.1611404418945316e-06,
      "step": 81509,
      "training_step_time": 10.635266304016113
    },
    {
      "epoch": 7.162094116210937e-06,
      "grad_norm": 0.08280788362026215,
      "learning_rate": 9.060214362661468e-07,
      "loss": 0.0564,
      "step": 81510
    },
    {
      "epoch": 7.162094116210937e-06,
      "model_forward_time": 1.0173025131225586,
      "step": 81510
    },
    {
      "dynamics_loss_avg": 0.03421791638247669,
      "epoch": 7.162094116210937e-06,
      "step": 81510
    },
    {
      "action_loss_avg": 0.001135676087869797,
      "epoch": 7.162094116210937e-06,
      "step": 81510
    },
    {
      "epoch": 7.162094116210937e-06,
      "step": 81510,
      "training_step_time": 9.234481811523438
    },
    {
      "epoch": 7.163047790527344e-06,
      "model_forward_time": 2.3998961448669434,
      "step": 81511
    },
    {
      "epoch": 7.163047790527344e-06,
      "step": 81511,
      "training_step_time": 7.591560125350952
    },
    {
      "epoch": 7.16400146484375e-06,
      "model_forward_time": 1.7434580326080322,
      "step": 81512
    },
    {
      "epoch": 7.16400146484375e-06,
      "step": 81512,
      "training_step_time": 7.553951263427734
    },
    {
      "epoch": 7.164955139160156e-06,
      "model_forward_time": 1.7393009662628174,
      "step": 81513
    },
    {
      "epoch": 7.164955139160156e-06,
      "step": 81513,
      "training_step_time": 8.135264158248901
    },
    {
      "epoch": 7.165908813476563e-06,
      "model_forward_time": 1.7494053840637207,
      "step": 81514
    },
    {
      "epoch": 7.165908813476563e-06,
      "step": 81514,
      "training_step_time": 6.702414512634277
    },
    {
      "epoch": 7.1668624877929684e-06,
      "model_forward_time": 1.3901185989379883,
      "step": 81515
    },
    {
      "epoch": 7.1668624877929684e-06,
      "step": 81515,
      "training_step_time": 6.698564529418945
    },
    {
      "epoch": 7.167816162109375e-06,
      "model_forward_time": 1.754591464996338,
      "step": 81516
    },
    {
      "epoch": 7.167816162109375e-06,
      "step": 81516,
      "training_step_time": 7.00302267074585
    },
    {
      "epoch": 7.168769836425782e-06,
      "model_forward_time": 1.745676040649414,
      "step": 81517
    },
    {
      "epoch": 7.168769836425782e-06,
      "step": 81517,
      "training_step_time": 7.23023533821106
    },
    {
      "epoch": 7.169723510742187e-06,
      "model_forward_time": 1.7408268451690674,
      "step": 81518
    },
    {
      "epoch": 7.169723510742187e-06,
      "step": 81518,
      "training_step_time": 9.319688320159912
    },
    {
      "epoch": 7.170677185058594e-06,
      "model_forward_time": 1.8581571578979492,
      "step": 81519
    },
    {
      "epoch": 7.170677185058594e-06,
      "step": 81519,
      "training_step_time": 9.350438117980957
    },
    {
      "epoch": 7.171630859375e-06,
      "grad_norm": 0.09169681370258331,
      "learning_rate": 9.050724279281064e-07,
      "loss": 0.0581,
      "step": 81520
    },
    {
      "epoch": 7.171630859375e-06,
      "model_forward_time": 2.343142509460449,
      "step": 81520
    },
    {
      "dynamics_loss_avg": 0.057044174522161484,
      "epoch": 7.171630859375e-06,
      "step": 81520
    },
    {
      "action_loss_avg": 0.0017603301326744258,
      "epoch": 7.171630859375e-06,
      "step": 81520
    },
    {
      "epoch": 7.171630859375e-06,
      "step": 81520,
      "training_step_time": 7.6595778465271
    },
    {
      "epoch": 7.172584533691406e-06,
      "model_forward_time": 1.7472023963928223,
      "step": 81521
    },
    {
      "epoch": 7.172584533691406e-06,
      "step": 81521,
      "training_step_time": 7.610592842102051
    },
    {
      "epoch": 7.173538208007813e-06,
      "model_forward_time": 1.744328260421753,
      "step": 81522
    },
    {
      "epoch": 7.173538208007813e-06,
      "step": 81522,
      "training_step_time": 8.448969602584839
    },
    {
      "epoch": 7.1744918823242185e-06,
      "model_forward_time": 1.7388761043548584,
      "step": 81523
    },
    {
      "epoch": 7.1744918823242185e-06,
      "step": 81523,
      "training_step_time": 7.257400989532471
    },
    {
      "epoch": 7.175445556640625e-06,
      "model_forward_time": 1.7489230632781982,
      "step": 81524
    },
    {
      "epoch": 7.175445556640625e-06,
      "step": 81524,
      "training_step_time": 6.722187280654907
    },
    {
      "epoch": 7.176399230957032e-06,
      "model_forward_time": 1.0281243324279785,
      "step": 81525
    },
    {
      "epoch": 7.176399230957032e-06,
      "step": 81525,
      "training_step_time": 6.991230010986328
    },
    {
      "epoch": 7.177352905273437e-06,
      "model_forward_time": 1.747833013534546,
      "step": 81526
    },
    {
      "epoch": 7.177352905273437e-06,
      "step": 81526,
      "training_step_time": 7.23140549659729
    },
    {
      "epoch": 7.178306579589844e-06,
      "model_forward_time": 1.3733420372009277,
      "step": 81527
    },
    {
      "epoch": 7.178306579589844e-06,
      "step": 81527,
      "training_step_time": 8.784327745437622
    },
    {
      "epoch": 7.17926025390625e-06,
      "model_forward_time": 1.861065149307251,
      "step": 81528
    },
    {
      "epoch": 7.17926025390625e-06,
      "step": 81528,
      "training_step_time": 9.709819555282593
    },
    {
      "epoch": 7.180213928222656e-06,
      "model_forward_time": 1.377450942993164,
      "step": 81529
    },
    {
      "epoch": 7.180213928222656e-06,
      "step": 81529,
      "training_step_time": 9.220714807510376
    },
    {
      "epoch": 7.181167602539063e-06,
      "grad_norm": 0.21551500260829926,
      "learning_rate": 9.041238674051895e-07,
      "loss": 0.0588,
      "step": 81530
    },
    {
      "epoch": 7.181167602539063e-06,
      "model_forward_time": 1.749372959136963,
      "step": 81530
    },
    {
      "dynamics_loss_avg": 0.07236929573118686,
      "epoch": 7.181167602539063e-06,
      "step": 81530
    },
    {
      "action_loss_avg": 0.0012310161779168992,
      "epoch": 7.181167602539063e-06,
      "step": 81530
    },
    {
      "epoch": 7.181167602539063e-06,
      "step": 81530,
      "training_step_time": 7.815776824951172
    },
    {
      "epoch": 7.1821212768554685e-06,
      "model_forward_time": 1.739999771118164,
      "step": 81531
    },
    {
      "epoch": 7.1821212768554685e-06,
      "step": 81531,
      "training_step_time": 7.6883087158203125
    },
    {
      "epoch": 7.183074951171875e-06,
      "model_forward_time": 1.743823528289795,
      "step": 81532
    },
    {
      "epoch": 7.183074951171875e-06,
      "step": 81532,
      "training_step_time": 7.671790599822998
    },
    {
      "epoch": 7.184028625488282e-06,
      "model_forward_time": 1.7405834197998047,
      "step": 81533
    },
    {
      "epoch": 7.184028625488282e-06,
      "step": 81533,
      "training_step_time": 6.826549768447876
    },
    {
      "epoch": 7.184982299804687e-06,
      "model_forward_time": 1.7400965690612793,
      "step": 81534
    },
    {
      "epoch": 7.184982299804687e-06,
      "step": 81534,
      "training_step_time": 7.152571678161621
    },
    {
      "epoch": 7.185935974121094e-06,
      "model_forward_time": 1.7388479709625244,
      "step": 81535
    },
    {
      "epoch": 7.185935974121094e-06,
      "step": 81535,
      "training_step_time": 7.342205286026001
    },
    {
      "epoch": 7.1868896484375e-06,
      "model_forward_time": 1.7434520721435547,
      "step": 81536
    },
    {
      "epoch": 7.1868896484375e-06,
      "step": 81536,
      "training_step_time": 8.458243370056152
    },
    {
      "epoch": 7.187843322753906e-06,
      "model_forward_time": 1.850039005279541,
      "step": 81537
    },
    {
      "epoch": 7.187843322753906e-06,
      "step": 81537,
      "training_step_time": 10.654672145843506
    },
    {
      "epoch": 7.188796997070313e-06,
      "model_forward_time": 1.7395002841949463,
      "step": 81538
    },
    {
      "epoch": 7.188796997070313e-06,
      "step": 81538,
      "training_step_time": 8.81038236618042
    },
    {
      "epoch": 7.1897506713867185e-06,
      "model_forward_time": 2.366286039352417,
      "step": 81539
    },
    {
      "epoch": 7.1897506713867185e-06,
      "step": 81539,
      "training_step_time": 7.7669243812561035
    },
    {
      "epoch": 7.190704345703125e-06,
      "grad_norm": 0.10272165387868881,
      "learning_rate": 9.031757548011277e-07,
      "loss": 0.0582,
      "step": 81540
    },
    {
      "epoch": 7.190704345703125e-06,
      "model_forward_time": 1.740384578704834,
      "step": 81540
    },
    {
      "dynamics_loss_avg": 0.04786751680076122,
      "epoch": 7.190704345703125e-06,
      "step": 81540
    },
    {
      "action_loss_avg": 0.00135555854649283,
      "epoch": 7.190704345703125e-06,
      "step": 81540
    },
    {
      "epoch": 7.190704345703125e-06,
      "step": 81540,
      "training_step_time": 7.535087585449219
    },
    {
      "epoch": 7.191658020019532e-06,
      "model_forward_time": 1.7417967319488525,
      "step": 81541
    },
    {
      "epoch": 7.191658020019532e-06,
      "step": 81541,
      "training_step_time": 8.227182626724243
    },
    {
      "epoch": 7.192611694335937e-06,
      "model_forward_time": 1.7401669025421143,
      "step": 81542
    },
    {
      "epoch": 7.192611694335937e-06,
      "step": 81542,
      "training_step_time": 6.813703298568726
    },
    {
      "epoch": 7.193565368652344e-06,
      "model_forward_time": 1.7458915710449219,
      "step": 81543
    },
    {
      "epoch": 7.193565368652344e-06,
      "step": 81543,
      "training_step_time": 7.72374701499939
    },
    {
      "epoch": 7.19451904296875e-06,
      "model_forward_time": 1.7499029636383057,
      "step": 81544
    },
    {
      "epoch": 7.19451904296875e-06,
      "step": 81544,
      "training_step_time": 7.393861770629883
    },
    {
      "epoch": 7.195472717285156e-06,
      "model_forward_time": 1.7471773624420166,
      "step": 81545
    },
    {
      "epoch": 7.195472717285156e-06,
      "step": 81545,
      "training_step_time": 8.421356678009033
    },
    {
      "epoch": 7.196426391601563e-06,
      "model_forward_time": 1.8081727027893066,
      "step": 81546
    },
    {
      "epoch": 7.196426391601563e-06,
      "step": 81546,
      "training_step_time": 9.888870239257812
    },
    {
      "epoch": 7.1973800659179685e-06,
      "model_forward_time": 1.7531476020812988,
      "step": 81547
    },
    {
      "epoch": 7.1973800659179685e-06,
      "step": 81547,
      "training_step_time": 8.504843711853027
    },
    {
      "epoch": 7.198333740234375e-06,
      "model_forward_time": 2.3740077018737793,
      "step": 81548
    },
    {
      "epoch": 7.198333740234375e-06,
      "step": 81548,
      "training_step_time": 7.785690546035767
    },
    {
      "epoch": 7.199287414550782e-06,
      "model_forward_time": 1.7464494705200195,
      "step": 81549
    },
    {
      "epoch": 7.199287414550782e-06,
      "step": 81549,
      "training_step_time": 7.669947862625122
    },
    {
      "epoch": 7.200241088867187e-06,
      "grad_norm": 0.12581734359264374,
      "learning_rate": 9.022280902196051e-07,
      "loss": 0.0557,
      "step": 81550
    },
    {
      "epoch": 7.200241088867187e-06,
      "model_forward_time": 1.3740556240081787,
      "step": 81550
    },
    {
      "dynamics_loss_avg": 0.037933307327330114,
      "epoch": 7.200241088867187e-06,
      "step": 81550
    },
    {
      "action_loss_avg": 0.0014613780367653817,
      "epoch": 7.200241088867187e-06,
      "step": 81550
    },
    {
      "epoch": 7.200241088867187e-06,
      "step": 81550,
      "training_step_time": 7.673132419586182
    },
    {
      "epoch": 7.201194763183594e-06,
      "model_forward_time": 1.7526535987854004,
      "step": 81551
    },
    {
      "epoch": 7.201194763183594e-06,
      "step": 81551,
      "training_step_time": 7.2832112312316895
    },
    {
      "epoch": 7.2021484375e-06,
      "model_forward_time": 1.750361680984497,
      "step": 81552
    },
    {
      "epoch": 7.2021484375e-06,
      "step": 81552,
      "training_step_time": 7.8165998458862305
    },
    {
      "epoch": 7.203102111816406e-06,
      "model_forward_time": 1.7498810291290283,
      "step": 81553
    },
    {
      "epoch": 7.203102111816406e-06,
      "step": 81553,
      "training_step_time": 7.370650291442871
    },
    {
      "epoch": 7.204055786132813e-06,
      "model_forward_time": 1.7500333786010742,
      "step": 81554
    },
    {
      "epoch": 7.204055786132813e-06,
      "step": 81554,
      "training_step_time": 8.535259246826172
    },
    {
      "epoch": 7.2050094604492185e-06,
      "model_forward_time": 1.7673394680023193,
      "step": 81555
    },
    {
      "epoch": 7.2050094604492185e-06,
      "step": 81555,
      "training_step_time": 8.803496360778809
    },
    {
      "epoch": 7.205963134765625e-06,
      "model_forward_time": 1.7443366050720215,
      "step": 81556
    },
    {
      "epoch": 7.205963134765625e-06,
      "step": 81556,
      "training_step_time": 9.423385620117188
    },
    {
      "epoch": 7.206916809082032e-06,
      "model_forward_time": 2.398137092590332,
      "step": 81557
    },
    {
      "epoch": 7.206916809082032e-06,
      "step": 81557,
      "training_step_time": 8.597408771514893
    },
    {
      "epoch": 7.207870483398437e-06,
      "model_forward_time": 1.3714754581451416,
      "step": 81558
    },
    {
      "epoch": 7.207870483398437e-06,
      "step": 81558,
      "training_step_time": 7.669657945632935
    },
    {
      "epoch": 7.208824157714844e-06,
      "model_forward_time": 1.7516593933105469,
      "step": 81559
    },
    {
      "epoch": 7.208824157714844e-06,
      "step": 81559,
      "training_step_time": 7.70002293586731
    },
    {
      "epoch": 7.20977783203125e-06,
      "grad_norm": 0.10279250890016556,
      "learning_rate": 9.012808737642592e-07,
      "loss": 0.0588,
      "step": 81560
    },
    {
      "epoch": 7.20977783203125e-06,
      "model_forward_time": 1.0358788967132568,
      "step": 81560
    },
    {
      "dynamics_loss_avg": 0.052560358215123414,
      "epoch": 7.20977783203125e-06,
      "step": 81560
    },
    {
      "action_loss_avg": 0.0012748990149702877,
      "epoch": 7.20977783203125e-06,
      "step": 81560
    },
    {
      "epoch": 7.20977783203125e-06,
      "step": 81560,
      "training_step_time": 7.47175931930542
    },
    {
      "epoch": 7.210731506347656e-06,
      "model_forward_time": 1.7479581832885742,
      "step": 81561
    },
    {
      "epoch": 7.210731506347656e-06,
      "step": 81561,
      "training_step_time": 7.787039756774902
    },
    {
      "epoch": 7.211685180664063e-06,
      "model_forward_time": 1.7493722438812256,
      "step": 81562
    },
    {
      "epoch": 7.211685180664063e-06,
      "step": 81562,
      "training_step_time": 6.7951979637146
    },
    {
      "epoch": 7.2126388549804685e-06,
      "model_forward_time": 1.367602825164795,
      "step": 81563
    },
    {
      "epoch": 7.2126388549804685e-06,
      "step": 81563,
      "training_step_time": 7.743706703186035
    },
    {
      "epoch": 7.213592529296875e-06,
      "model_forward_time": 1.7520534992218018,
      "step": 81564
    },
    {
      "epoch": 7.213592529296875e-06,
      "step": 81564,
      "training_step_time": 9.471561670303345
    },
    {
      "epoch": 7.214546203613282e-06,
      "model_forward_time": 1.382112979888916,
      "step": 81565
    },
    {
      "epoch": 7.214546203613282e-06,
      "step": 81565,
      "training_step_time": 8.74888014793396
    },
    {
      "epoch": 7.215499877929687e-06,
      "model_forward_time": 1.7492578029632568,
      "step": 81566
    },
    {
      "epoch": 7.215499877929687e-06,
      "step": 81566,
      "training_step_time": 8.434712171554565
    },
    {
      "epoch": 7.216453552246094e-06,
      "model_forward_time": 2.4029574394226074,
      "step": 81567
    },
    {
      "epoch": 7.216453552246094e-06,
      "step": 81567,
      "training_step_time": 7.741634845733643
    },
    {
      "epoch": 7.2174072265625e-06,
      "model_forward_time": 1.7452938556671143,
      "step": 81568
    },
    {
      "epoch": 7.2174072265625e-06,
      "step": 81568,
      "training_step_time": 7.723121166229248
    },
    {
      "epoch": 7.218360900878906e-06,
      "model_forward_time": 1.7503550052642822,
      "step": 81569
    },
    {
      "epoch": 7.218360900878906e-06,
      "step": 81569,
      "training_step_time": 7.686160087585449
    },
    {
      "epoch": 7.219314575195313e-06,
      "grad_norm": 0.08950908482074738,
      "learning_rate": 9.003341055386738e-07,
      "loss": 0.0577,
      "step": 81570
    },
    {
      "epoch": 7.219314575195313e-06,
      "model_forward_time": 1.7447848320007324,
      "step": 81570
    },
    {
      "dynamics_loss_avg": 0.05926300808787346,
      "epoch": 7.219314575195313e-06,
      "step": 81570
    },
    {
      "action_loss_avg": 0.0014651101140771062,
      "epoch": 7.219314575195313e-06,
      "step": 81570
    },
    {
      "epoch": 7.219314575195313e-06,
      "step": 81570,
      "training_step_time": 7.8021111488342285
    },
    {
      "epoch": 7.2202682495117185e-06,
      "model_forward_time": 1.7539162635803223,
      "step": 81571
    },
    {
      "epoch": 7.2202682495117185e-06,
      "step": 81571,
      "training_step_time": 6.9088451862335205
    },
    {
      "epoch": 7.221221923828125e-06,
      "model_forward_time": 1.7499494552612305,
      "step": 81572
    },
    {
      "epoch": 7.221221923828125e-06,
      "step": 81572,
      "training_step_time": 7.798013687133789
    },
    {
      "epoch": 7.222175598144532e-06,
      "model_forward_time": 1.7503015995025635,
      "step": 81573
    },
    {
      "epoch": 7.222175598144532e-06,
      "step": 81573,
      "training_step_time": 7.831171751022339
    },
    {
      "epoch": 7.223129272460937e-06,
      "model_forward_time": 1.8660058975219727,
      "step": 81574
    },
    {
      "epoch": 7.223129272460937e-06,
      "step": 81574,
      "training_step_time": 8.37007737159729
    },
    {
      "epoch": 7.224082946777344e-06,
      "model_forward_time": 1.7475006580352783,
      "step": 81575
    },
    {
      "epoch": 7.224082946777344e-06,
      "step": 81575,
      "training_step_time": 8.395800352096558
    },
    {
      "epoch": 7.22503662109375e-06,
      "model_forward_time": 2.394473075866699,
      "step": 81576
    },
    {
      "epoch": 7.22503662109375e-06,
      "step": 81576,
      "training_step_time": 7.815354824066162
    },
    {
      "epoch": 7.225990295410156e-06,
      "model_forward_time": 1.7421822547912598,
      "step": 81577
    },
    {
      "epoch": 7.225990295410156e-06,
      "step": 81577,
      "training_step_time": 7.7249603271484375
    },
    {
      "epoch": 7.226943969726563e-06,
      "model_forward_time": 1.3704097270965576,
      "step": 81578
    },
    {
      "epoch": 7.226943969726563e-06,
      "step": 81578,
      "training_step_time": 7.690041780471802
    },
    {
      "epoch": 7.2278976440429685e-06,
      "model_forward_time": 1.3699736595153809,
      "step": 81579
    },
    {
      "epoch": 7.2278976440429685e-06,
      "step": 81579,
      "training_step_time": 7.778842449188232
    },
    {
      "epoch": 7.228851318359375e-06,
      "grad_norm": 0.07568889111280441,
      "learning_rate": 8.993877856463862e-07,
      "loss": 0.0578,
      "step": 81580
    },
    {
      "epoch": 7.228851318359375e-06,
      "model_forward_time": 1.7451350688934326,
      "step": 81580
    },
    {
      "dynamics_loss_avg": 0.06566536743193865,
      "epoch": 7.228851318359375e-06,
      "step": 81580
    },
    {
      "action_loss_avg": 0.0012527752987807617,
      "epoch": 7.228851318359375e-06,
      "step": 81580
    },
    {
      "epoch": 7.228851318359375e-06,
      "step": 81580,
      "training_step_time": 6.856128454208374
    },
    {
      "epoch": 7.229804992675782e-06,
      "model_forward_time": 1.7462632656097412,
      "step": 81581
    },
    {
      "epoch": 7.229804992675782e-06,
      "step": 81581,
      "training_step_time": 7.650981187820435
    },
    {
      "epoch": 7.230758666992187e-06,
      "model_forward_time": 1.747166633605957,
      "step": 81582
    },
    {
      "epoch": 7.230758666992187e-06,
      "step": 81582,
      "training_step_time": 8.74225902557373
    },
    {
      "epoch": 7.231712341308594e-06,
      "model_forward_time": 1.8471307754516602,
      "step": 81583
    },
    {
      "epoch": 7.231712341308594e-06,
      "step": 81583,
      "training_step_time": 9.932286739349365
    },
    {
      "epoch": 7.232666015625e-06,
      "model_forward_time": 1.7498626708984375,
      "step": 81584
    },
    {
      "epoch": 7.232666015625e-06,
      "step": 81584,
      "training_step_time": 8.668576955795288
    },
    {
      "epoch": 7.233619689941406e-06,
      "model_forward_time": 1.7531723976135254,
      "step": 81585
    },
    {
      "epoch": 7.233619689941406e-06,
      "step": 81585,
      "training_step_time": 7.886417627334595
    },
    {
      "epoch": 7.234573364257813e-06,
      "model_forward_time": 1.7514464855194092,
      "step": 81586
    },
    {
      "epoch": 7.234573364257813e-06,
      "step": 81586,
      "training_step_time": 7.78074312210083
    },
    {
      "epoch": 7.2355270385742185e-06,
      "model_forward_time": 1.745572805404663,
      "step": 81587
    },
    {
      "epoch": 7.2355270385742185e-06,
      "step": 81587,
      "training_step_time": 7.657005786895752
    },
    {
      "epoch": 7.236480712890625e-06,
      "model_forward_time": 1.7466135025024414,
      "step": 81588
    },
    {
      "epoch": 7.236480712890625e-06,
      "step": 81588,
      "training_step_time": 7.753438711166382
    },
    {
      "epoch": 7.237434387207032e-06,
      "model_forward_time": 1.745964527130127,
      "step": 81589
    },
    {
      "epoch": 7.237434387207032e-06,
      "step": 81589,
      "training_step_time": 6.81769585609436
    },
    {
      "epoch": 7.238388061523437e-06,
      "grad_norm": 0.08542527258396149,
      "learning_rate": 8.984419141908857e-07,
      "loss": 0.0563,
      "step": 81590
    },
    {
      "epoch": 7.238388061523437e-06,
      "model_forward_time": 1.7430522441864014,
      "step": 81590
    },
    {
      "dynamics_loss_avg": 0.0797184195369482,
      "epoch": 7.238388061523437e-06,
      "step": 81590
    },
    {
      "action_loss_avg": 0.0012822332413634286,
      "epoch": 7.238388061523437e-06,
      "step": 81590
    },
    {
      "epoch": 7.238388061523437e-06,
      "step": 81590,
      "training_step_time": 7.596786022186279
    },
    {
      "epoch": 7.239341735839844e-06,
      "model_forward_time": 1.7490086555480957,
      "step": 81591
    },
    {
      "epoch": 7.239341735839844e-06,
      "step": 81591,
      "training_step_time": 9.334513902664185
    },
    {
      "epoch": 7.24029541015625e-06,
      "model_forward_time": 1.8132052421569824,
      "step": 81592
    },
    {
      "epoch": 7.24029541015625e-06,
      "step": 81592,
      "training_step_time": 8.692551612854004
    },
    {
      "epoch": 7.241249084472656e-06,
      "model_forward_time": 1.7471091747283936,
      "step": 81593
    },
    {
      "epoch": 7.241249084472656e-06,
      "step": 81593,
      "training_step_time": 7.819318532943726
    },
    {
      "epoch": 7.242202758789063e-06,
      "model_forward_time": 1.7564868927001953,
      "step": 81594
    },
    {
      "epoch": 7.242202758789063e-06,
      "step": 81594,
      "training_step_time": 7.787176609039307
    },
    {
      "epoch": 7.2431564331054685e-06,
      "model_forward_time": 2.3830857276916504,
      "step": 81595
    },
    {
      "epoch": 7.2431564331054685e-06,
      "step": 81595,
      "training_step_time": 7.767667055130005
    },
    {
      "epoch": 7.244110107421875e-06,
      "model_forward_time": 1.7492444515228271,
      "step": 81596
    },
    {
      "epoch": 7.244110107421875e-06,
      "step": 81596,
      "training_step_time": 7.736953020095825
    },
    {
      "epoch": 7.245063781738282e-06,
      "model_forward_time": 1.737008810043335,
      "step": 81597
    },
    {
      "epoch": 7.245063781738282e-06,
      "step": 81597,
      "training_step_time": 7.881760597229004
    },
    {
      "epoch": 7.246017456054687e-06,
      "model_forward_time": 1.7409751415252686,
      "step": 81598
    },
    {
      "epoch": 7.246017456054687e-06,
      "step": 81598,
      "training_step_time": 7.221814393997192
    },
    {
      "epoch": 7.246971130371094e-06,
      "model_forward_time": 1.7428348064422607,
      "step": 81599
    },
    {
      "epoch": 7.246971130371094e-06,
      "step": 81599,
      "training_step_time": 7.696003198623657
    },
    {
      "epoch": 7.2479248046875e-06,
      "grad_norm": 0.08939430862665176,
      "learning_rate": 8.974964912756118e-07,
      "loss": 0.0568,
      "step": 81600
    },
    {
      "epoch": 7.2479248046875e-06,
      "model_forward_time": 1.733412742614746,
      "step": 81600
    },
    {
      "dynamics_loss_avg": 0.05656124260276556,
      "epoch": 7.2479248046875e-06,
      "step": 81600
    },
    {
      "action_loss_avg": 0.002009366962010972,
      "epoch": 7.2479248046875e-06,
      "step": 81600
    },
    {
      "epoch": 7.2479248046875e-06,
      "step": 81600,
      "training_step_time": 8.922762393951416
    },
    {
      "epoch": 7.248878479003906e-06,
      "model_forward_time": 1.7747399806976318,
      "step": 81601
    },
    {
      "epoch": 7.248878479003906e-06,
      "step": 81601,
      "training_step_time": 10.325746297836304
    },
    {
      "epoch": 7.249832153320313e-06,
      "model_forward_time": 1.9016225337982178,
      "step": 81602
    },
    {
      "epoch": 7.249832153320313e-06,
      "step": 81602,
      "training_step_time": 7.863168954849243
    },
    {
      "epoch": 7.2507858276367185e-06,
      "model_forward_time": 1.8986687660217285,
      "step": 81603
    },
    {
      "epoch": 7.2507858276367185e-06,
      "step": 81603,
      "training_step_time": 8.53674840927124
    },
    {
      "epoch": 7.251739501953125e-06,
      "model_forward_time": 2.4719126224517822,
      "step": 81604
    },
    {
      "epoch": 7.251739501953125e-06,
      "step": 81604,
      "training_step_time": 7.773128271102905
    },
    {
      "epoch": 7.252693176269532e-06,
      "model_forward_time": 1.7469892501831055,
      "step": 81605
    },
    {
      "epoch": 7.252693176269532e-06,
      "step": 81605,
      "training_step_time": 7.768787145614624
    },
    {
      "epoch": 7.253646850585937e-06,
      "model_forward_time": 1.734623908996582,
      "step": 81606
    },
    {
      "epoch": 7.253646850585937e-06,
      "step": 81606,
      "training_step_time": 7.86491060256958
    },
    {
      "epoch": 7.254600524902344e-06,
      "model_forward_time": 1.7476084232330322,
      "step": 81607
    },
    {
      "epoch": 7.254600524902344e-06,
      "step": 81607,
      "training_step_time": 7.188279867172241
    },
    {
      "epoch": 7.25555419921875e-06,
      "model_forward_time": 1.746556282043457,
      "step": 81608
    },
    {
      "epoch": 7.25555419921875e-06,
      "step": 81608,
      "training_step_time": 7.657177209854126
    },
    {
      "epoch": 7.256507873535156e-06,
      "model_forward_time": 1.739943027496338,
      "step": 81609
    },
    {
      "epoch": 7.256507873535156e-06,
      "step": 81609,
      "training_step_time": 7.817106485366821
    },
    {
      "epoch": 7.257461547851563e-06,
      "grad_norm": 0.1140531525015831,
      "learning_rate": 8.965515170039524e-07,
      "loss": 0.0557,
      "step": 81610
    },
    {
      "epoch": 7.257461547851563e-06,
      "model_forward_time": 1.753307580947876,
      "step": 81610
    },
    {
      "dynamics_loss_avg": 0.07361241467297078,
      "epoch": 7.257461547851563e-06,
      "step": 81610
    },
    {
      "action_loss_avg": 0.001304835604969412,
      "epoch": 7.257461547851563e-06,
      "step": 81610
    },
    {
      "epoch": 7.257461547851563e-06,
      "step": 81610,
      "training_step_time": 8.645703554153442
    },
    {
      "epoch": 7.2584152221679685e-06,
      "model_forward_time": 1.7488186359405518,
      "step": 81611
    },
    {
      "epoch": 7.2584152221679685e-06,
      "step": 81611,
      "training_step_time": 8.825562477111816
    },
    {
      "epoch": 7.259368896484375e-06,
      "model_forward_time": 1.7533397674560547,
      "step": 81612
    },
    {
      "epoch": 7.259368896484375e-06,
      "step": 81612,
      "training_step_time": 7.830812454223633
    },
    {
      "epoch": 7.260322570800782e-06,
      "model_forward_time": 2.408923387527466,
      "step": 81613
    },
    {
      "epoch": 7.260322570800782e-06,
      "step": 81613,
      "training_step_time": 7.817003965377808
    },
    {
      "epoch": 7.261276245117187e-06,
      "model_forward_time": 1.748455286026001,
      "step": 81614
    },
    {
      "epoch": 7.261276245117187e-06,
      "step": 81614,
      "training_step_time": 7.753804445266724
    },
    {
      "epoch": 7.262229919433594e-06,
      "model_forward_time": 1.7462995052337646,
      "step": 81615
    },
    {
      "epoch": 7.262229919433594e-06,
      "step": 81615,
      "training_step_time": 7.883795738220215
    },
    {
      "epoch": 7.26318359375e-06,
      "model_forward_time": 1.7556822299957275,
      "step": 81616
    },
    {
      "epoch": 7.26318359375e-06,
      "step": 81616,
      "training_step_time": 7.668362379074097
    },
    {
      "epoch": 7.264137268066406e-06,
      "model_forward_time": 1.7500224113464355,
      "step": 81617
    },
    {
      "epoch": 7.264137268066406e-06,
      "step": 81617,
      "training_step_time": 7.714708566665649
    },
    {
      "epoch": 7.265090942382813e-06,
      "model_forward_time": 1.7500581741333008,
      "step": 81618
    },
    {
      "epoch": 7.265090942382813e-06,
      "step": 81618,
      "training_step_time": 8.732333660125732
    },
    {
      "epoch": 7.2660446166992185e-06,
      "model_forward_time": 1.7489628791809082,
      "step": 81619
    },
    {
      "epoch": 7.2660446166992185e-06,
      "step": 81619,
      "training_step_time": 8.450966119766235
    },
    {
      "epoch": 7.266998291015625e-06,
      "grad_norm": 0.12273703515529633,
      "learning_rate": 8.956069914792515e-07,
      "loss": 0.0552,
      "step": 81620
    },
    {
      "epoch": 7.266998291015625e-06,
      "model_forward_time": 1.7418763637542725,
      "step": 81620
    },
    {
      "dynamics_loss_avg": 0.05073599498718977,
      "epoch": 7.266998291015625e-06,
      "step": 81620
    },
    {
      "action_loss_avg": 0.0008875252024154179,
      "epoch": 7.266998291015625e-06,
      "step": 81620
    },
    {
      "epoch": 7.266998291015625e-06,
      "step": 81620,
      "training_step_time": 8.766771793365479
    },
    {
      "epoch": 7.267951965332032e-06,
      "model_forward_time": 1.7403717041015625,
      "step": 81621
    },
    {
      "epoch": 7.267951965332032e-06,
      "step": 81621,
      "training_step_time": 9.547783136367798
    },
    {
      "epoch": 7.268905639648437e-06,
      "model_forward_time": 2.400289297103882,
      "step": 81622
    },
    {
      "epoch": 7.268905639648437e-06,
      "step": 81622,
      "training_step_time": 7.915552139282227
    },
    {
      "epoch": 7.269859313964844e-06,
      "model_forward_time": 1.7424769401550293,
      "step": 81623
    },
    {
      "epoch": 7.269859313964844e-06,
      "step": 81623,
      "training_step_time": 7.788782358169556
    },
    {
      "epoch": 7.27081298828125e-06,
      "model_forward_time": 1.7470033168792725,
      "step": 81624
    },
    {
      "epoch": 7.27081298828125e-06,
      "step": 81624,
      "training_step_time": 7.946189880371094
    },
    {
      "epoch": 7.271766662597656e-06,
      "model_forward_time": 1.7419476509094238,
      "step": 81625
    },
    {
      "epoch": 7.271766662597656e-06,
      "step": 81625,
      "training_step_time": 7.6683104038238525
    },
    {
      "epoch": 7.272720336914063e-06,
      "model_forward_time": 1.7407872676849365,
      "step": 81626
    },
    {
      "epoch": 7.272720336914063e-06,
      "step": 81626,
      "training_step_time": 7.779716968536377
    },
    {
      "epoch": 7.2736740112304685e-06,
      "model_forward_time": 1.7488601207733154,
      "step": 81627
    },
    {
      "epoch": 7.2736740112304685e-06,
      "step": 81627,
      "training_step_time": 7.8656392097473145
    },
    {
      "epoch": 7.274627685546875e-06,
      "model_forward_time": 1.7365071773529053,
      "step": 81628
    },
    {
      "epoch": 7.274627685546875e-06,
      "step": 81628,
      "training_step_time": 7.898079872131348
    },
    {
      "epoch": 7.275581359863281e-06,
      "model_forward_time": 1.855318546295166,
      "step": 81629
    },
    {
      "epoch": 7.275581359863281e-06,
      "step": 81629,
      "training_step_time": 8.91050910949707
    },
    {
      "epoch": 7.276535034179687e-06,
      "grad_norm": 0.13597489893436432,
      "learning_rate": 8.94662914804797e-07,
      "loss": 0.0556,
      "step": 81630
    },
    {
      "epoch": 7.276535034179687e-06,
      "model_forward_time": 1.0255694389343262,
      "step": 81630
    },
    {
      "dynamics_loss_avg": 0.06537763774394989,
      "epoch": 7.276535034179687e-06,
      "step": 81630
    },
    {
      "action_loss_avg": 0.001276313458220102,
      "epoch": 7.276535034179687e-06,
      "step": 81630
    },
    {
      "epoch": 7.276535034179687e-06,
      "step": 81630,
      "training_step_time": 8.512160062789917
    },
    {
      "epoch": 7.277488708496094e-06,
      "model_forward_time": 1.7411484718322754,
      "step": 81631
    },
    {
      "epoch": 7.277488708496094e-06,
      "step": 81631,
      "training_step_time": 7.8239452838897705
    },
    {
      "epoch": 7.2784423828125e-06,
      "model_forward_time": 2.3693504333496094,
      "step": 81632
    },
    {
      "epoch": 7.2784423828125e-06,
      "step": 81632,
      "training_step_time": 7.751241683959961
    },
    {
      "epoch": 7.279396057128906e-06,
      "model_forward_time": 1.7454752922058105,
      "step": 81633
    },
    {
      "epoch": 7.279396057128906e-06,
      "step": 81633,
      "training_step_time": 7.815490245819092
    },
    {
      "epoch": 7.280349731445313e-06,
      "model_forward_time": 1.7435617446899414,
      "step": 81634
    },
    {
      "epoch": 7.280349731445313e-06,
      "step": 81634,
      "training_step_time": 7.613073825836182
    },
    {
      "epoch": 7.2813034057617185e-06,
      "model_forward_time": 1.7403817176818848,
      "step": 81635
    },
    {
      "epoch": 7.2813034057617185e-06,
      "step": 81635,
      "training_step_time": 7.663186073303223
    },
    {
      "epoch": 7.282257080078125e-06,
      "model_forward_time": 1.7420415878295898,
      "step": 81636
    },
    {
      "epoch": 7.282257080078125e-06,
      "step": 81636,
      "training_step_time": 7.6819167137146
    },
    {
      "epoch": 7.283210754394531e-06,
      "model_forward_time": 1.7391607761383057,
      "step": 81637
    },
    {
      "epoch": 7.283210754394531e-06,
      "step": 81637,
      "training_step_time": 7.971188068389893
    },
    {
      "epoch": 7.284164428710937e-06,
      "model_forward_time": 1.8519954681396484,
      "step": 81638
    },
    {
      "epoch": 7.284164428710937e-06,
      "step": 81638,
      "training_step_time": 9.474595785140991
    },
    {
      "epoch": 7.285118103027344e-06,
      "model_forward_time": 1.744560956954956,
      "step": 81639
    },
    {
      "epoch": 7.285118103027344e-06,
      "step": 81639,
      "training_step_time": 9.333328485488892
    },
    {
      "epoch": 7.28607177734375e-06,
      "grad_norm": 0.09646286070346832,
      "learning_rate": 8.93719287083834e-07,
      "loss": 0.0539,
      "step": 81640
    },
    {
      "epoch": 7.28607177734375e-06,
      "model_forward_time": 1.739656686782837,
      "step": 81640
    },
    {
      "dynamics_loss_avg": 0.06027219146490097,
      "epoch": 7.28607177734375e-06,
      "step": 81640
    },
    {
      "action_loss_avg": 0.0015094183734618127,
      "epoch": 7.28607177734375e-06,
      "step": 81640
    },
    {
      "epoch": 7.28607177734375e-06,
      "step": 81640,
      "training_step_time": 8.306748628616333
    },
    {
      "epoch": 7.287025451660156e-06,
      "model_forward_time": 2.3764443397521973,
      "step": 81641
    },
    {
      "epoch": 7.287025451660156e-06,
      "step": 81641,
      "training_step_time": 7.749081611633301
    },
    {
      "epoch": 7.287979125976563e-06,
      "model_forward_time": 1.7379000186920166,
      "step": 81642
    },
    {
      "epoch": 7.287979125976563e-06,
      "step": 81642,
      "training_step_time": 7.848351001739502
    },
    {
      "epoch": 7.2889328002929686e-06,
      "model_forward_time": 1.7382175922393799,
      "step": 81643
    },
    {
      "epoch": 7.2889328002929686e-06,
      "step": 81643,
      "training_step_time": 7.685623407363892
    },
    {
      "epoch": 7.289886474609375e-06,
      "model_forward_time": 1.7496209144592285,
      "step": 81644
    },
    {
      "epoch": 7.289886474609375e-06,
      "step": 81644,
      "training_step_time": 8.448154211044312
    },
    {
      "epoch": 7.290840148925781e-06,
      "model_forward_time": 1.7572474479675293,
      "step": 81645
    },
    {
      "epoch": 7.290840148925781e-06,
      "step": 81645,
      "training_step_time": 7.868929386138916
    },
    {
      "epoch": 7.291793823242187e-06,
      "model_forward_time": 1.754284143447876,
      "step": 81646
    },
    {
      "epoch": 7.291793823242187e-06,
      "step": 81646,
      "training_step_time": 8.61948823928833
    },
    {
      "epoch": 7.292747497558594e-06,
      "model_forward_time": 1.8544013500213623,
      "step": 81647
    },
    {
      "epoch": 7.292747497558594e-06,
      "step": 81647,
      "training_step_time": 10.074885606765747
    },
    {
      "epoch": 7.293701171875e-06,
      "model_forward_time": 1.7595698833465576,
      "step": 81648
    },
    {
      "epoch": 7.293701171875e-06,
      "step": 81648,
      "training_step_time": 9.63542127609253
    },
    {
      "epoch": 7.294654846191406e-06,
      "model_forward_time": 1.7561790943145752,
      "step": 81649
    },
    {
      "epoch": 7.294654846191406e-06,
      "step": 81649,
      "training_step_time": 7.7879862785339355
    },
    {
      "epoch": 7.295608520507813e-06,
      "grad_norm": 0.09197469055652618,
      "learning_rate": 8.927761084195569e-07,
      "loss": 0.0549,
      "step": 81650
    },
    {
      "epoch": 7.295608520507813e-06,
      "model_forward_time": 2.4461727142333984,
      "step": 81650
    },
    {
      "dynamics_loss_avg": 0.04962031804025173,
      "epoch": 7.295608520507813e-06,
      "step": 81650
    },
    {
      "action_loss_avg": 0.001160055579384789,
      "epoch": 7.295608520507813e-06,
      "step": 81650
    },
    {
      "epoch": 7.295608520507813e-06,
      "step": 81650,
      "training_step_time": 7.761152505874634
    },
    {
      "epoch": 7.2965621948242186e-06,
      "model_forward_time": 1.7563331127166748,
      "step": 81651
    },
    {
      "epoch": 7.2965621948242186e-06,
      "step": 81651,
      "training_step_time": 7.770115613937378
    },
    {
      "epoch": 7.297515869140625e-06,
      "model_forward_time": 1.7562315464019775,
      "step": 81652
    },
    {
      "epoch": 7.297515869140625e-06,
      "step": 81652,
      "training_step_time": 7.661766052246094
    },
    {
      "epoch": 7.298469543457031e-06,
      "model_forward_time": 1.818324089050293,
      "step": 81653
    },
    {
      "epoch": 7.298469543457031e-06,
      "step": 81653,
      "training_step_time": 8.349598169326782
    },
    {
      "epoch": 7.299423217773437e-06,
      "model_forward_time": 1.7609148025512695,
      "step": 81654
    },
    {
      "epoch": 7.299423217773437e-06,
      "step": 81654,
      "training_step_time": 7.733888387680054
    },
    {
      "epoch": 7.300376892089844e-06,
      "model_forward_time": 1.757394790649414,
      "step": 81655
    },
    {
      "epoch": 7.300376892089844e-06,
      "step": 81655,
      "training_step_time": 7.676079273223877
    },
    {
      "epoch": 7.30133056640625e-06,
      "model_forward_time": 1.7647526264190674,
      "step": 81656
    },
    {
      "epoch": 7.30133056640625e-06,
      "step": 81656,
      "training_step_time": 8.464044570922852
    },
    {
      "epoch": 7.302284240722656e-06,
      "model_forward_time": 1.767052173614502,
      "step": 81657
    },
    {
      "epoch": 7.302284240722656e-06,
      "step": 81657,
      "training_step_time": 8.64008378982544
    },
    {
      "epoch": 7.303237915039063e-06,
      "model_forward_time": 1.7864141464233398,
      "step": 81658
    },
    {
      "epoch": 7.303237915039063e-06,
      "step": 81658,
      "training_step_time": 8.378907203674316
    },
    {
      "epoch": 7.3041915893554686e-06,
      "model_forward_time": 2.417222499847412,
      "step": 81659
    },
    {
      "epoch": 7.3041915893554686e-06,
      "step": 81659,
      "training_step_time": 9.90664267539978
    },
    {
      "epoch": 7.305145263671875e-06,
      "grad_norm": 0.09430784732103348,
      "learning_rate": 8.918333789151096e-07,
      "loss": 0.0534,
      "step": 81660
    },
    {
      "epoch": 7.305145263671875e-06,
      "model_forward_time": 1.7358951568603516,
      "step": 81660
    },
    {
      "dynamics_loss_avg": 0.04613197930157185,
      "epoch": 7.305145263671875e-06,
      "step": 81660
    },
    {
      "action_loss_avg": 0.0013854260119842365,
      "epoch": 7.305145263671875e-06,
      "step": 81660
    },
    {
      "epoch": 7.305145263671875e-06,
      "step": 81660,
      "training_step_time": 7.809346675872803
    },
    {
      "epoch": 7.306098937988281e-06,
      "model_forward_time": 1.7570970058441162,
      "step": 81661
    },
    {
      "epoch": 7.306098937988281e-06,
      "step": 81661,
      "training_step_time": 7.723244667053223
    },
    {
      "epoch": 7.3070526123046874e-06,
      "model_forward_time": 1.7608516216278076,
      "step": 81662
    },
    {
      "epoch": 7.3070526123046874e-06,
      "step": 81662,
      "training_step_time": 7.461406230926514
    },
    {
      "epoch": 7.308006286621094e-06,
      "model_forward_time": 1.756721019744873,
      "step": 81663
    },
    {
      "epoch": 7.308006286621094e-06,
      "step": 81663,
      "training_step_time": 7.782490968704224
    },
    {
      "epoch": 7.3089599609375e-06,
      "model_forward_time": 1.7483525276184082,
      "step": 81664
    },
    {
      "epoch": 7.3089599609375e-06,
      "step": 81664,
      "training_step_time": 8.625343799591064
    },
    {
      "epoch": 7.309913635253906e-06,
      "model_forward_time": 1.7510015964508057,
      "step": 81665
    },
    {
      "epoch": 7.309913635253906e-06,
      "step": 81665,
      "training_step_time": 9.67411494255066
    },
    {
      "epoch": 7.310867309570313e-06,
      "model_forward_time": 1.383037805557251,
      "step": 81666
    },
    {
      "epoch": 7.310867309570313e-06,
      "step": 81666,
      "training_step_time": 7.797008514404297
    },
    {
      "epoch": 7.3118209838867186e-06,
      "model_forward_time": 1.7344613075256348,
      "step": 81667
    },
    {
      "epoch": 7.3118209838867186e-06,
      "step": 81667,
      "training_step_time": 9.057116746902466
    },
    {
      "epoch": 7.312774658203125e-06,
      "model_forward_time": 1.7475512027740479,
      "step": 81668
    },
    {
      "epoch": 7.312774658203125e-06,
      "step": 81668,
      "training_step_time": 7.800098657608032
    },
    {
      "epoch": 7.313728332519531e-06,
      "model_forward_time": 1.9644484519958496,
      "step": 81669
    },
    {
      "epoch": 7.313728332519531e-06,
      "step": 81669,
      "training_step_time": 7.75853157043457
    },
    {
      "epoch": 7.3146820068359374e-06,
      "grad_norm": 0.22596339881420135,
      "learning_rate": 8.908910986735869e-07,
      "loss": 0.0532,
      "step": 81670
    },
    {
      "epoch": 7.3146820068359374e-06,
      "model_forward_time": 1.7258903980255127,
      "step": 81670
    },
    {
      "dynamics_loss_avg": 0.06763606257736683,
      "epoch": 7.3146820068359374e-06,
      "step": 81670
    },
    {
      "action_loss_avg": 0.0009659905364969746,
      "epoch": 7.3146820068359374e-06,
      "step": 81670
    },
    {
      "epoch": 7.3146820068359374e-06,
      "step": 81670,
      "training_step_time": 7.692842960357666
    },
    {
      "epoch": 7.315635681152344e-06,
      "model_forward_time": 1.0251884460449219,
      "step": 81671
    },
    {
      "epoch": 7.315635681152344e-06,
      "step": 81671,
      "training_step_time": 7.423226356506348
    },
    {
      "epoch": 7.31658935546875e-06,
      "model_forward_time": 1.7356972694396973,
      "step": 81672
    },
    {
      "epoch": 7.31658935546875e-06,
      "step": 81672,
      "training_step_time": 7.722822904586792
    },
    {
      "epoch": 7.317543029785156e-06,
      "model_forward_time": 1.7342579364776611,
      "step": 81673
    },
    {
      "epoch": 7.317543029785156e-06,
      "step": 81673,
      "training_step_time": 8.274783372879028
    },
    {
      "epoch": 7.318496704101563e-06,
      "model_forward_time": 1.7519021034240723,
      "step": 81674
    },
    {
      "epoch": 7.318496704101563e-06,
      "step": 81674,
      "training_step_time": 10.27346158027649
    },
    {
      "epoch": 7.319450378417969e-06,
      "model_forward_time": 1.7463507652282715,
      "step": 81675
    },
    {
      "epoch": 7.319450378417969e-06,
      "step": 81675,
      "training_step_time": 8.300176858901978
    },
    {
      "epoch": 7.320404052734375e-06,
      "model_forward_time": 1.3598430156707764,
      "step": 81676
    },
    {
      "epoch": 7.320404052734375e-06,
      "step": 81676,
      "training_step_time": 9.439852237701416
    },
    {
      "epoch": 7.321357727050781e-06,
      "model_forward_time": 1.737999439239502,
      "step": 81677
    },
    {
      "epoch": 7.321357727050781e-06,
      "step": 81677,
      "training_step_time": 7.755616903305054
    },
    {
      "epoch": 7.3223114013671874e-06,
      "model_forward_time": 2.3276214599609375,
      "step": 81678
    },
    {
      "epoch": 7.3223114013671874e-06,
      "step": 81678,
      "training_step_time": 7.774098873138428
    },
    {
      "epoch": 7.323265075683594e-06,
      "model_forward_time": 1.737304925918579,
      "step": 81679
    },
    {
      "epoch": 7.323265075683594e-06,
      "step": 81679,
      "training_step_time": 7.652604103088379
    },
    {
      "epoch": 7.32421875e-06,
      "grad_norm": 0.07790408283472061,
      "learning_rate": 8.89949267798037e-07,
      "loss": 0.0531,
      "step": 81680
    },
    {
      "epoch": 7.32421875e-06,
      "model_forward_time": 1.731694221496582,
      "step": 81680
    },
    {
      "dynamics_loss_avg": 0.056758289039134976,
      "epoch": 7.32421875e-06,
      "step": 81680
    },
    {
      "action_loss_avg": 0.0011276656412519514,
      "epoch": 7.32421875e-06,
      "step": 81680
    },
    {
      "epoch": 7.32421875e-06,
      "step": 81680,
      "training_step_time": 7.522436141967773
    },
    {
      "epoch": 7.325172424316406e-06,
      "model_forward_time": 1.7480831146240234,
      "step": 81681
    },
    {
      "epoch": 7.325172424316406e-06,
      "step": 81681,
      "training_step_time": 7.7128331661224365
    },
    {
      "epoch": 7.326126098632813e-06,
      "model_forward_time": 1.746973991394043,
      "step": 81682
    },
    {
      "epoch": 7.326126098632813e-06,
      "step": 81682,
      "training_step_time": 7.620072841644287
    },
    {
      "epoch": 7.327079772949219e-06,
      "model_forward_time": 1.7529282569885254,
      "step": 81683
    },
    {
      "epoch": 7.327079772949219e-06,
      "step": 81683,
      "training_step_time": 9.642744779586792
    },
    {
      "epoch": 7.328033447265625e-06,
      "model_forward_time": 1.7653446197509766,
      "step": 81684
    },
    {
      "epoch": 7.328033447265625e-06,
      "step": 81684,
      "training_step_time": 9.8664710521698
    },
    {
      "epoch": 7.328987121582031e-06,
      "model_forward_time": 1.3783586025238037,
      "step": 81685
    },
    {
      "epoch": 7.328987121582031e-06,
      "step": 81685,
      "training_step_time": 7.789246320724487
    },
    {
      "epoch": 7.3299407958984374e-06,
      "model_forward_time": 1.7481544017791748,
      "step": 81686
    },
    {
      "epoch": 7.3299407958984374e-06,
      "step": 81686,
      "training_step_time": 8.754313468933105
    },
    {
      "epoch": 7.330894470214844e-06,
      "model_forward_time": 1.7485756874084473,
      "step": 81687
    },
    {
      "epoch": 7.330894470214844e-06,
      "step": 81687,
      "training_step_time": 8.34850811958313
    },
    {
      "epoch": 7.33184814453125e-06,
      "model_forward_time": 2.3937735557556152,
      "step": 81688
    },
    {
      "epoch": 7.33184814453125e-06,
      "step": 81688,
      "training_step_time": 7.6522345542907715
    },
    {
      "epoch": 7.332801818847656e-06,
      "model_forward_time": 1.7466800212860107,
      "step": 81689
    },
    {
      "epoch": 7.332801818847656e-06,
      "step": 81689,
      "training_step_time": 7.6303534507751465
    },
    {
      "epoch": 7.333755493164063e-06,
      "grad_norm": 0.09009962528944016,
      "learning_rate": 8.890078863914542e-07,
      "loss": 0.0533,
      "step": 81690
    },
    {
      "epoch": 7.333755493164063e-06,
      "model_forward_time": 1.7517342567443848,
      "step": 81690
    },
    {
      "dynamics_loss_avg": 0.047769094631075856,
      "epoch": 7.333755493164063e-06,
      "step": 81690
    },
    {
      "action_loss_avg": 0.0013870006921933963,
      "epoch": 7.333755493164063e-06,
      "step": 81690
    },
    {
      "epoch": 7.333755493164063e-06,
      "step": 81690,
      "training_step_time": 7.686238527297974
    },
    {
      "epoch": 7.334709167480469e-06,
      "model_forward_time": 1.7473790645599365,
      "step": 81691
    },
    {
      "epoch": 7.334709167480469e-06,
      "step": 81691,
      "training_step_time": 8.093615770339966
    },
    {
      "epoch": 7.335662841796875e-06,
      "model_forward_time": 1.7448410987854004,
      "step": 81692
    },
    {
      "epoch": 7.335662841796875e-06,
      "step": 81692,
      "training_step_time": 9.856685400009155
    },
    {
      "epoch": 7.336616516113281e-06,
      "model_forward_time": 1.7407889366149902,
      "step": 81693
    },
    {
      "epoch": 7.336616516113281e-06,
      "step": 81693,
      "training_step_time": 8.989813327789307
    },
    {
      "epoch": 7.3375701904296875e-06,
      "model_forward_time": 1.7394766807556152,
      "step": 81694
    },
    {
      "epoch": 7.3375701904296875e-06,
      "step": 81694,
      "training_step_time": 8.523210048675537
    },
    {
      "epoch": 7.338523864746094e-06,
      "model_forward_time": 1.7392382621765137,
      "step": 81695
    },
    {
      "epoch": 7.338523864746094e-06,
      "step": 81695,
      "training_step_time": 9.451225519180298
    },
    {
      "epoch": 7.3394775390625e-06,
      "model_forward_time": 1.7392163276672363,
      "step": 81696
    },
    {
      "epoch": 7.3394775390625e-06,
      "step": 81696,
      "training_step_time": 8.305150985717773
    },
    {
      "epoch": 7.340431213378906e-06,
      "model_forward_time": 2.3719139099121094,
      "step": 81697
    },
    {
      "epoch": 7.340431213378906e-06,
      "step": 81697,
      "training_step_time": 7.600250959396362
    },
    {
      "epoch": 7.341384887695313e-06,
      "model_forward_time": 1.7441487312316895,
      "step": 81698
    },
    {
      "epoch": 7.341384887695313e-06,
      "step": 81698,
      "training_step_time": 7.679814338684082
    },
    {
      "epoch": 7.342338562011719e-06,
      "model_forward_time": 1.740448236465454,
      "step": 81699
    },
    {
      "epoch": 7.342338562011719e-06,
      "step": 81699,
      "training_step_time": 7.763610124588013
    },
    {
      "epoch": 7.343292236328125e-06,
      "grad_norm": 0.2821180522441864,
      "learning_rate": 8.880669545567894e-07,
      "loss": 0.0529,
      "step": 81700
    },
    {
      "epoch": 7.343292236328125e-06,
      "model_forward_time": 1.7314245700836182,
      "step": 81700
    },
    {
      "dynamics_loss_avg": 0.05484428610652685,
      "epoch": 7.343292236328125e-06,
      "step": 81700
    },
    {
      "action_loss_avg": 0.0014196102827554568,
      "epoch": 7.343292236328125e-06,
      "step": 81700
    },
    {
      "epoch": 7.343292236328125e-06,
      "step": 81700,
      "training_step_time": 7.368656873703003
    },
    {
      "epoch": 7.344245910644531e-06,
      "model_forward_time": 1.7369511127471924,
      "step": 81701
    },
    {
      "epoch": 7.344245910644531e-06,
      "step": 81701,
      "training_step_time": 8.55554723739624
    },
    {
      "epoch": 7.3451995849609375e-06,
      "model_forward_time": 1.7339603900909424,
      "step": 81702
    },
    {
      "epoch": 7.3451995849609375e-06,
      "step": 81702,
      "training_step_time": 8.415209293365479
    },
    {
      "epoch": 7.346153259277344e-06,
      "model_forward_time": 1.7661480903625488,
      "step": 81703
    },
    {
      "epoch": 7.346153259277344e-06,
      "step": 81703,
      "training_step_time": 9.516914129257202
    },
    {
      "epoch": 7.34710693359375e-06,
      "model_forward_time": 1.7422873973846436,
      "step": 81704
    },
    {
      "epoch": 7.34710693359375e-06,
      "step": 81704,
      "training_step_time": 8.42712140083313
    },
    {
      "epoch": 7.348060607910156e-06,
      "model_forward_time": 1.73655366897583,
      "step": 81705
    },
    {
      "epoch": 7.348060607910156e-06,
      "step": 81705,
      "training_step_time": 7.738149881362915
    },
    {
      "epoch": 7.349014282226563e-06,
      "model_forward_time": 2.378730535507202,
      "step": 81706
    },
    {
      "epoch": 7.349014282226563e-06,
      "step": 81706,
      "training_step_time": 7.7680230140686035
    },
    {
      "epoch": 7.349967956542969e-06,
      "model_forward_time": 1.734205961227417,
      "step": 81707
    },
    {
      "epoch": 7.349967956542969e-06,
      "step": 81707,
      "training_step_time": 7.737149238586426
    },
    {
      "epoch": 7.350921630859375e-06,
      "model_forward_time": 1.7354240417480469,
      "step": 81708
    },
    {
      "epoch": 7.350921630859375e-06,
      "step": 81708,
      "training_step_time": 7.613692283630371
    },
    {
      "epoch": 7.351875305175781e-06,
      "model_forward_time": 1.7355594635009766,
      "step": 81709
    },
    {
      "epoch": 7.351875305175781e-06,
      "step": 81709,
      "training_step_time": 7.388817071914673
    },
    {
      "epoch": 7.3528289794921875e-06,
      "grad_norm": 0.07450243085622787,
      "learning_rate": 8.871264723969381e-07,
      "loss": 0.054,
      "step": 81710
    },
    {
      "epoch": 7.3528289794921875e-06,
      "model_forward_time": 1.7352962493896484,
      "step": 81710
    },
    {
      "dynamics_loss_avg": 0.034269675798714164,
      "epoch": 7.3528289794921875e-06,
      "step": 81710
    },
    {
      "action_loss_avg": 0.0007108179299393669,
      "epoch": 7.3528289794921875e-06,
      "step": 81710
    },
    {
      "epoch": 7.3528289794921875e-06,
      "step": 81710,
      "training_step_time": 7.7881481647491455
    },
    {
      "epoch": 7.353782653808594e-06,
      "model_forward_time": 1.7562735080718994,
      "step": 81711
    },
    {
      "epoch": 7.353782653808594e-06,
      "step": 81711,
      "training_step_time": 7.78889274597168
    },
    {
      "epoch": 7.354736328125e-06,
      "model_forward_time": 1.7529888153076172,
      "step": 81712
    },
    {
      "epoch": 7.354736328125e-06,
      "step": 81712,
      "training_step_time": 7.7950828075408936
    },
    {
      "epoch": 7.355690002441406e-06,
      "model_forward_time": 1.7422430515289307,
      "step": 81713
    },
    {
      "epoch": 7.355690002441406e-06,
      "step": 81713,
      "training_step_time": 7.829970359802246
    },
    {
      "epoch": 7.356643676757813e-06,
      "model_forward_time": 1.7445058822631836,
      "step": 81714
    },
    {
      "epoch": 7.356643676757813e-06,
      "step": 81714,
      "training_step_time": 10.380539894104004
    },
    {
      "epoch": 7.357597351074219e-06,
      "model_forward_time": 2.373262643814087,
      "step": 81715
    },
    {
      "epoch": 7.357597351074219e-06,
      "step": 81715,
      "training_step_time": 7.761700391769409
    },
    {
      "epoch": 7.358551025390625e-06,
      "model_forward_time": 1.7375397682189941,
      "step": 81716
    },
    {
      "epoch": 7.358551025390625e-06,
      "step": 81716,
      "training_step_time": 7.671776294708252
    },
    {
      "epoch": 7.359504699707031e-06,
      "model_forward_time": 1.0257515907287598,
      "step": 81717
    },
    {
      "epoch": 7.359504699707031e-06,
      "step": 81717,
      "training_step_time": 7.588587284088135
    },
    {
      "epoch": 7.3604583740234375e-06,
      "model_forward_time": 1.7373545169830322,
      "step": 81718
    },
    {
      "epoch": 7.3604583740234375e-06,
      "step": 81718,
      "training_step_time": 7.371924877166748
    },
    {
      "epoch": 7.361412048339844e-06,
      "model_forward_time": 1.7365336418151855,
      "step": 81719
    },
    {
      "epoch": 7.361412048339844e-06,
      "step": 81719,
      "training_step_time": 7.773697137832642
    },
    {
      "epoch": 7.36236572265625e-06,
      "grad_norm": 0.07951728999614716,
      "learning_rate": 8.861864400147541e-07,
      "loss": 0.0553,
      "step": 81720
    },
    {
      "epoch": 7.36236572265625e-06,
      "model_forward_time": 1.7507097721099854,
      "step": 81720
    },
    {
      "dynamics_loss_avg": 0.05176818147301674,
      "epoch": 7.36236572265625e-06,
      "step": 81720
    },
    {
      "action_loss_avg": 0.0013396790629485622,
      "epoch": 7.36236572265625e-06,
      "step": 81720
    },
    {
      "epoch": 7.36236572265625e-06,
      "step": 81720,
      "training_step_time": 8.772653818130493
    },
    {
      "epoch": 7.363319396972656e-06,
      "model_forward_time": 1.744236946105957,
      "step": 81721
    },
    {
      "epoch": 7.363319396972656e-06,
      "step": 81721,
      "training_step_time": 7.793054580688477
    },
    {
      "epoch": 7.364273071289063e-06,
      "model_forward_time": 1.7399787902832031,
      "step": 81722
    },
    {
      "epoch": 7.364273071289063e-06,
      "step": 81722,
      "training_step_time": 9.483776330947876
    },
    {
      "epoch": 7.365226745605469e-06,
      "model_forward_time": 1.734893560409546,
      "step": 81723
    },
    {
      "epoch": 7.365226745605469e-06,
      "step": 81723,
      "training_step_time": 7.790301084518433
    },
    {
      "epoch": 7.366180419921875e-06,
      "model_forward_time": 1.7459216117858887,
      "step": 81724
    },
    {
      "epoch": 7.366180419921875e-06,
      "step": 81724,
      "training_step_time": 7.796326398849487
    },
    {
      "epoch": 7.367134094238281e-06,
      "model_forward_time": 2.3671152591705322,
      "step": 81725
    },
    {
      "epoch": 7.367134094238281e-06,
      "step": 81725,
      "training_step_time": 7.533785104751587
    },
    {
      "epoch": 7.3680877685546875e-06,
      "model_forward_time": 1.748182773590088,
      "step": 81726
    },
    {
      "epoch": 7.3680877685546875e-06,
      "step": 81726,
      "training_step_time": 7.697430372238159
    },
    {
      "epoch": 7.369041442871094e-06,
      "model_forward_time": 1.7394607067108154,
      "step": 81727
    },
    {
      "epoch": 7.369041442871094e-06,
      "step": 81727,
      "training_step_time": 7.377342462539673
    },
    {
      "epoch": 7.3699951171875e-06,
      "model_forward_time": 1.7367830276489258,
      "step": 81728
    },
    {
      "epoch": 7.3699951171875e-06,
      "step": 81728,
      "training_step_time": 7.7799224853515625
    },
    {
      "epoch": 7.370948791503906e-06,
      "model_forward_time": 1.7363460063934326,
      "step": 81729
    },
    {
      "epoch": 7.370948791503906e-06,
      "step": 81729,
      "training_step_time": 9.801830768585205
    },
    {
      "epoch": 7.371902465820313e-06,
      "grad_norm": 0.08306168019771576,
      "learning_rate": 8.852468575130352e-07,
      "loss": 0.0537,
      "step": 81730
    },
    {
      "epoch": 7.371902465820313e-06,
      "model_forward_time": 1.7364234924316406,
      "step": 81730
    },
    {
      "dynamics_loss_avg": 0.052548780012875794,
      "epoch": 7.371902465820313e-06,
      "step": 81730
    },
    {
      "action_loss_avg": 0.002122730331029743,
      "epoch": 7.371902465820313e-06,
      "step": 81730
    },
    {
      "epoch": 7.371902465820313e-06,
      "step": 81730,
      "training_step_time": 9.790098190307617
    },
    {
      "epoch": 7.372856140136719e-06,
      "model_forward_time": 1.3627264499664307,
      "step": 81731
    },
    {
      "epoch": 7.372856140136719e-06,
      "step": 81731,
      "training_step_time": 9.523599624633789
    },
    {
      "epoch": 7.373809814453125e-06,
      "model_forward_time": 1.7333343029022217,
      "step": 81732
    },
    {
      "epoch": 7.373809814453125e-06,
      "step": 81732,
      "training_step_time": 7.774837493896484
    },
    {
      "epoch": 7.374763488769531e-06,
      "model_forward_time": 1.7356994152069092,
      "step": 81733
    },
    {
      "epoch": 7.374763488769531e-06,
      "step": 81733,
      "training_step_time": 7.811811208724976
    },
    {
      "epoch": 7.3757171630859375e-06,
      "model_forward_time": 2.3566837310791016,
      "step": 81734
    },
    {
      "epoch": 7.3757171630859375e-06,
      "step": 81734,
      "training_step_time": 7.734548091888428
    },
    {
      "epoch": 7.376670837402344e-06,
      "model_forward_time": 1.7373948097229004,
      "step": 81735
    },
    {
      "epoch": 7.376670837402344e-06,
      "step": 81735,
      "training_step_time": 8.530672311782837
    },
    {
      "epoch": 7.37762451171875e-06,
      "model_forward_time": 1.7406973838806152,
      "step": 81736
    },
    {
      "epoch": 7.37762451171875e-06,
      "step": 81736,
      "training_step_time": 7.377207279205322
    },
    {
      "epoch": 7.378578186035156e-06,
      "model_forward_time": 1.7319953441619873,
      "step": 81737
    },
    {
      "epoch": 7.378578186035156e-06,
      "step": 81737,
      "training_step_time": 8.72228717803955
    },
    {
      "epoch": 7.379531860351563e-06,
      "model_forward_time": 1.7472419738769531,
      "step": 81738
    },
    {
      "epoch": 7.379531860351563e-06,
      "step": 81738,
      "training_step_time": 8.212059020996094
    },
    {
      "epoch": 7.380485534667969e-06,
      "model_forward_time": 1.7613170146942139,
      "step": 81739
    },
    {
      "epoch": 7.380485534667969e-06,
      "step": 81739,
      "training_step_time": 8.448445081710815
    },
    {
      "epoch": 7.381439208984375e-06,
      "grad_norm": 0.09607750922441483,
      "learning_rate": 8.843077249945347e-07,
      "loss": 0.0539,
      "step": 81740
    },
    {
      "epoch": 7.381439208984375e-06,
      "model_forward_time": 1.7979695796966553,
      "step": 81740
    },
    {
      "dynamics_loss_avg": 0.05260561853647232,
      "epoch": 7.381439208984375e-06,
      "step": 81740
    },
    {
      "action_loss_avg": 0.001079771769582294,
      "epoch": 7.381439208984375e-06,
      "step": 81740
    },
    {
      "epoch": 7.381439208984375e-06,
      "step": 81740,
      "training_step_time": 7.743009567260742
    },
    {
      "epoch": 7.382392883300781e-06,
      "model_forward_time": 1.7475783824920654,
      "step": 81741
    },
    {
      "epoch": 7.382392883300781e-06,
      "step": 81741,
      "training_step_time": 8.500000238418579
    },
    {
      "epoch": 7.3833465576171875e-06,
      "model_forward_time": 1.7530181407928467,
      "step": 81742
    },
    {
      "epoch": 7.3833465576171875e-06,
      "step": 81742,
      "training_step_time": 7.717113494873047
    },
    {
      "epoch": 7.384300231933594e-06,
      "model_forward_time": 2.3707854747772217,
      "step": 81743
    },
    {
      "epoch": 7.384300231933594e-06,
      "step": 81743,
      "training_step_time": 7.705607891082764
    },
    {
      "epoch": 7.38525390625e-06,
      "model_forward_time": 1.7401278018951416,
      "step": 81744
    },
    {
      "epoch": 7.38525390625e-06,
      "step": 81744,
      "training_step_time": 8.395449876785278
    },
    {
      "epoch": 7.386207580566406e-06,
      "model_forward_time": 1.7459287643432617,
      "step": 81745
    },
    {
      "epoch": 7.386207580566406e-06,
      "step": 81745,
      "training_step_time": 7.5180628299713135
    },
    {
      "epoch": 7.387161254882813e-06,
      "model_forward_time": 1.0373141765594482,
      "step": 81746
    },
    {
      "epoch": 7.387161254882813e-06,
      "step": 81746,
      "training_step_time": 9.396398544311523
    },
    {
      "epoch": 7.388114929199219e-06,
      "model_forward_time": 1.751924753189087,
      "step": 81747
    },
    {
      "epoch": 7.388114929199219e-06,
      "step": 81747,
      "training_step_time": 9.356189250946045
    },
    {
      "epoch": 7.389068603515625e-06,
      "model_forward_time": 1.753342628479004,
      "step": 81748
    },
    {
      "epoch": 7.389068603515625e-06,
      "step": 81748,
      "training_step_time": 7.751096487045288
    },
    {
      "epoch": 7.390022277832031e-06,
      "model_forward_time": 1.7640507221221924,
      "step": 81749
    },
    {
      "epoch": 7.390022277832031e-06,
      "step": 81749,
      "training_step_time": 8.512948751449585
    },
    {
      "epoch": 7.3909759521484375e-06,
      "grad_norm": 0.08263212442398071,
      "learning_rate": 8.833690425619523e-07,
      "loss": 0.0536,
      "step": 81750
    },
    {
      "epoch": 7.3909759521484375e-06,
      "model_forward_time": 1.7464733123779297,
      "step": 81750
    },
    {
      "dynamics_loss_avg": 0.04727895334362984,
      "epoch": 7.3909759521484375e-06,
      "step": 81750
    },
    {
      "action_loss_avg": 0.0010360849148128182,
      "epoch": 7.3909759521484375e-06,
      "step": 81750
    },
    {
      "epoch": 7.3909759521484375e-06,
      "step": 81750,
      "training_step_time": 8.573782920837402
    },
    {
      "epoch": 7.391929626464844e-06,
      "model_forward_time": 1.7470579147338867,
      "step": 81751
    },
    {
      "epoch": 7.391929626464844e-06,
      "step": 81751,
      "training_step_time": 8.650967836380005
    },
    {
      "epoch": 7.39288330078125e-06,
      "model_forward_time": 1.7450644969940186,
      "step": 81752
    },
    {
      "epoch": 7.39288330078125e-06,
      "step": 81752,
      "training_step_time": 7.801520347595215
    },
    {
      "epoch": 7.393836975097656e-06,
      "model_forward_time": 2.3851137161254883,
      "step": 81753
    },
    {
      "epoch": 7.393836975097656e-06,
      "step": 81753,
      "training_step_time": 7.776490926742554
    },
    {
      "epoch": 7.394790649414062e-06,
      "model_forward_time": 1.7602522373199463,
      "step": 81754
    },
    {
      "epoch": 7.394790649414062e-06,
      "step": 81754,
      "training_step_time": 7.706891298294067
    },
    {
      "epoch": 7.395744323730469e-06,
      "model_forward_time": 1.764430284500122,
      "step": 81755
    },
    {
      "epoch": 7.395744323730469e-06,
      "step": 81755,
      "training_step_time": 7.7501490116119385
    },
    {
      "epoch": 7.396697998046875e-06,
      "model_forward_time": 1.7650048732757568,
      "step": 81756
    },
    {
      "epoch": 7.396697998046875e-06,
      "step": 81756,
      "training_step_time": 7.732276201248169
    },
    {
      "epoch": 7.397651672363281e-06,
      "model_forward_time": 1.7545270919799805,
      "step": 81757
    },
    {
      "epoch": 7.397651672363281e-06,
      "step": 81757,
      "training_step_time": 8.33942198753357
    },
    {
      "epoch": 7.3986053466796875e-06,
      "model_forward_time": 1.8815088272094727,
      "step": 81758
    },
    {
      "epoch": 7.3986053466796875e-06,
      "step": 81758,
      "training_step_time": 9.140685081481934
    },
    {
      "epoch": 7.399559020996094e-06,
      "model_forward_time": 1.775437593460083,
      "step": 81759
    },
    {
      "epoch": 7.399559020996094e-06,
      "step": 81759,
      "training_step_time": 7.8089234828948975
    },
    {
      "epoch": 7.4005126953125e-06,
      "grad_norm": 0.07409266382455826,
      "learning_rate": 8.824308103179418e-07,
      "loss": 0.055,
      "step": 81760
    },
    {
      "epoch": 7.4005126953125e-06,
      "model_forward_time": 1.058786392211914,
      "step": 81760
    },
    {
      "dynamics_loss_avg": 0.04242358487099409,
      "epoch": 7.4005126953125e-06,
      "step": 81760
    },
    {
      "action_loss_avg": 0.0009098595619434491,
      "epoch": 7.4005126953125e-06,
      "step": 81760
    },
    {
      "epoch": 7.4005126953125e-06,
      "step": 81760,
      "training_step_time": 8.307960033416748
    },
    {
      "epoch": 7.401466369628906e-06,
      "model_forward_time": 1.7689568996429443,
      "step": 81761
    },
    {
      "epoch": 7.401466369628906e-06,
      "step": 81761,
      "training_step_time": 7.837574243545532
    },
    {
      "epoch": 7.402420043945312e-06,
      "model_forward_time": 2.385664701461792,
      "step": 81762
    },
    {
      "epoch": 7.402420043945312e-06,
      "step": 81762,
      "training_step_time": 7.741949081420898
    },
    {
      "epoch": 7.403373718261719e-06,
      "model_forward_time": 1.768934965133667,
      "step": 81763
    },
    {
      "epoch": 7.403373718261719e-06,
      "step": 81763,
      "training_step_time": 7.736552953720093
    },
    {
      "epoch": 7.404327392578125e-06,
      "model_forward_time": 1.7716560363769531,
      "step": 81764
    },
    {
      "epoch": 7.404327392578125e-06,
      "step": 81764,
      "training_step_time": 7.7798473834991455
    },
    {
      "epoch": 7.405281066894531e-06,
      "model_forward_time": 1.7747654914855957,
      "step": 81765
    },
    {
      "epoch": 7.405281066894531e-06,
      "step": 81765,
      "training_step_time": 7.676123142242432
    },
    {
      "epoch": 7.4062347412109375e-06,
      "model_forward_time": 1.7685878276824951,
      "step": 81766
    },
    {
      "epoch": 7.4062347412109375e-06,
      "step": 81766,
      "training_step_time": 7.760441541671753
    },
    {
      "epoch": 7.407188415527344e-06,
      "model_forward_time": 1.8785181045532227,
      "step": 81767
    },
    {
      "epoch": 7.407188415527344e-06,
      "step": 81767,
      "training_step_time": 9.483100891113281
    },
    {
      "epoch": 7.40814208984375e-06,
      "model_forward_time": 1.7651000022888184,
      "step": 81768
    },
    {
      "epoch": 7.40814208984375e-06,
      "step": 81768,
      "training_step_time": 8.216769933700562
    },
    {
      "epoch": 7.409095764160156e-06,
      "model_forward_time": 1.0315666198730469,
      "step": 81769
    },
    {
      "epoch": 7.409095764160156e-06,
      "step": 81769,
      "training_step_time": 8.523662567138672
    },
    {
      "epoch": 7.410049438476562e-06,
      "grad_norm": 0.10961290448904037,
      "learning_rate": 8.814930283651074e-07,
      "loss": 0.055,
      "step": 81770
    },
    {
      "epoch": 7.410049438476562e-06,
      "model_forward_time": 1.754986047744751,
      "step": 81770
    },
    {
      "dynamics_loss_avg": 0.05626938184723258,
      "epoch": 7.410049438476562e-06,
      "step": 81770
    },
    {
      "action_loss_avg": 0.0010707482637371869,
      "epoch": 7.410049438476562e-06,
      "step": 81770
    },
    {
      "epoch": 7.410049438476562e-06,
      "step": 81770,
      "training_step_time": 8.362911939620972
    },
    {
      "epoch": 7.411003112792969e-06,
      "model_forward_time": 1.756089687347412,
      "step": 81771
    },
    {
      "epoch": 7.411003112792969e-06,
      "step": 81771,
      "training_step_time": 7.54956316947937
    },
    {
      "epoch": 7.411956787109375e-06,
      "model_forward_time": 2.3864710330963135,
      "step": 81772
    },
    {
      "epoch": 7.411956787109375e-06,
      "step": 81772,
      "training_step_time": 8.735109329223633
    },
    {
      "epoch": 7.412910461425781e-06,
      "model_forward_time": 1.7593016624450684,
      "step": 81773
    },
    {
      "epoch": 7.412910461425781e-06,
      "step": 81773,
      "training_step_time": 7.74701714515686
    },
    {
      "epoch": 7.4138641357421875e-06,
      "model_forward_time": 1.7574355602264404,
      "step": 81774
    },
    {
      "epoch": 7.4138641357421875e-06,
      "step": 81774,
      "training_step_time": 8.262452602386475
    },
    {
      "epoch": 7.414817810058594e-06,
      "model_forward_time": 1.7602119445800781,
      "step": 81775
    },
    {
      "epoch": 7.414817810058594e-06,
      "step": 81775,
      "training_step_time": 7.661271095275879
    },
    {
      "epoch": 7.415771484375e-06,
      "model_forward_time": 1.8376927375793457,
      "step": 81776
    },
    {
      "epoch": 7.415771484375e-06,
      "step": 81776,
      "training_step_time": 9.138267755508423
    },
    {
      "epoch": 7.416725158691406e-06,
      "model_forward_time": 1.7585177421569824,
      "step": 81777
    },
    {
      "epoch": 7.416725158691406e-06,
      "step": 81777,
      "training_step_time": 10.392163753509521
    },
    {
      "epoch": 7.417678833007812e-06,
      "model_forward_time": 1.7570700645446777,
      "step": 81778
    },
    {
      "epoch": 7.417678833007812e-06,
      "step": 81778,
      "training_step_time": 7.6541221141815186
    },
    {
      "epoch": 7.418632507324219e-06,
      "model_forward_time": 1.7551777362823486,
      "step": 81779
    },
    {
      "epoch": 7.418632507324219e-06,
      "step": 81779,
      "training_step_time": 7.69159460067749
    },
    {
      "epoch": 7.419586181640625e-06,
      "grad_norm": 0.07691488415002823,
      "learning_rate": 8.805556968060047e-07,
      "loss": 0.0566,
      "step": 81780
    },
    {
      "epoch": 7.419586181640625e-06,
      "model_forward_time": 1.7529315948486328,
      "step": 81780
    },
    {
      "dynamics_loss_avg": 0.04460018426179886,
      "epoch": 7.419586181640625e-06,
      "step": 81780
    },
    {
      "action_loss_avg": 0.0016708296665456147,
      "epoch": 7.419586181640625e-06,
      "step": 81780
    },
    {
      "epoch": 7.419586181640625e-06,
      "step": 81780,
      "training_step_time": 7.66729998588562
    },
    {
      "epoch": 7.420539855957031e-06,
      "model_forward_time": 2.426110029220581,
      "step": 81781
    },
    {
      "epoch": 7.420539855957031e-06,
      "step": 81781,
      "training_step_time": 8.479022026062012
    },
    {
      "epoch": 7.4214935302734375e-06,
      "model_forward_time": 1.7559444904327393,
      "step": 81782
    },
    {
      "epoch": 7.4214935302734375e-06,
      "step": 81782,
      "training_step_time": 7.673255443572998
    },
    {
      "epoch": 7.422447204589844e-06,
      "model_forward_time": 1.7544608116149902,
      "step": 81783
    },
    {
      "epoch": 7.422447204589844e-06,
      "step": 81783,
      "training_step_time": 7.494950294494629
    },
    {
      "epoch": 7.42340087890625e-06,
      "model_forward_time": 1.7540960311889648,
      "step": 81784
    },
    {
      "epoch": 7.42340087890625e-06,
      "step": 81784,
      "training_step_time": 7.609600305557251
    },
    {
      "epoch": 7.424354553222656e-06,
      "model_forward_time": 1.7881848812103271,
      "step": 81785
    },
    {
      "epoch": 7.424354553222656e-06,
      "step": 81785,
      "training_step_time": 7.645558595657349
    },
    {
      "epoch": 7.425308227539062e-06,
      "model_forward_time": 1.7712035179138184,
      "step": 81786
    },
    {
      "epoch": 7.425308227539062e-06,
      "step": 81786,
      "training_step_time": 7.722221612930298
    },
    {
      "epoch": 7.426261901855469e-06,
      "model_forward_time": 1.7586424350738525,
      "step": 81787
    },
    {
      "epoch": 7.426261901855469e-06,
      "step": 81787,
      "training_step_time": 7.845426082611084
    },
    {
      "epoch": 7.427215576171875e-06,
      "model_forward_time": 1.7554082870483398,
      "step": 81788
    },
    {
      "epoch": 7.427215576171875e-06,
      "step": 81788,
      "training_step_time": 7.743366479873657
    },
    {
      "epoch": 7.428169250488281e-06,
      "model_forward_time": 1.7542014122009277,
      "step": 81789
    },
    {
      "epoch": 7.428169250488281e-06,
      "step": 81789,
      "training_step_time": 7.835926055908203
    },
    {
      "epoch": 7.4291229248046875e-06,
      "grad_norm": 0.09974078834056854,
      "learning_rate": 8.796188157431363e-07,
      "loss": 0.0556,
      "step": 81790
    },
    {
      "epoch": 7.4291229248046875e-06,
      "model_forward_time": 2.355214834213257,
      "step": 81790
    },
    {
      "dynamics_loss_avg": 0.0552351557649672,
      "epoch": 7.4291229248046875e-06,
      "step": 81790
    },
    {
      "action_loss_avg": 0.0015761584276333452,
      "epoch": 7.4291229248046875e-06,
      "step": 81790
    },
    {
      "epoch": 7.4291229248046875e-06,
      "step": 81790,
      "training_step_time": 7.685270309448242
    },
    {
      "epoch": 7.430076599121094e-06,
      "model_forward_time": 1.7341351509094238,
      "step": 81791
    },
    {
      "epoch": 7.430076599121094e-06,
      "step": 81791,
      "training_step_time": 7.717668533325195
    },
    {
      "epoch": 7.4310302734375e-06,
      "model_forward_time": 1.0396459102630615,
      "step": 81792
    },
    {
      "epoch": 7.4310302734375e-06,
      "step": 81792,
      "training_step_time": 7.356103420257568
    },
    {
      "epoch": 7.431983947753906e-06,
      "model_forward_time": 1.7648522853851318,
      "step": 81793
    },
    {
      "epoch": 7.431983947753906e-06,
      "step": 81793,
      "training_step_time": 7.678579092025757
    },
    {
      "epoch": 7.432937622070312e-06,
      "model_forward_time": 1.7477645874023438,
      "step": 81794
    },
    {
      "epoch": 7.432937622070312e-06,
      "step": 81794,
      "training_step_time": 7.832866430282593
    },
    {
      "epoch": 7.433891296386719e-06,
      "model_forward_time": 1.780937671661377,
      "step": 81795
    },
    {
      "epoch": 7.433891296386719e-06,
      "step": 81795,
      "training_step_time": 8.346361875534058
    },
    {
      "epoch": 7.434844970703125e-06,
      "model_forward_time": 1.7523972988128662,
      "step": 81796
    },
    {
      "epoch": 7.434844970703125e-06,
      "step": 81796,
      "training_step_time": 7.999446392059326
    },
    {
      "epoch": 7.435798645019531e-06,
      "model_forward_time": 1.7502598762512207,
      "step": 81797
    },
    {
      "epoch": 7.435798645019531e-06,
      "step": 81797,
      "training_step_time": 7.8218042850494385
    },
    {
      "epoch": 7.4367523193359375e-06,
      "model_forward_time": 1.7531766891479492,
      "step": 81798
    },
    {
      "epoch": 7.4367523193359375e-06,
      "step": 81798,
      "training_step_time": 7.794147968292236
    },
    {
      "epoch": 7.437705993652344e-06,
      "model_forward_time": 1.7542204856872559,
      "step": 81799
    },
    {
      "epoch": 7.437705993652344e-06,
      "step": 81799,
      "training_step_time": 7.717482805252075
    },
    {
      "epoch": 7.43865966796875e-06,
      "grad_norm": 0.12361074984073639,
      "learning_rate": 8.786823852789594e-07,
      "loss": 0.0549,
      "step": 81800
    },
    {
      "epoch": 7.43865966796875e-06,
      "model_forward_time": 2.3848001956939697,
      "step": 81800
    },
    {
      "dynamics_loss_avg": 0.06388816609978676,
      "epoch": 7.43865966796875e-06,
      "step": 81800
    },
    {
      "action_loss_avg": 0.0015419736329931766,
      "epoch": 7.43865966796875e-06,
      "step": 81800
    },
    {
      "epoch": 7.43865966796875e-06,
      "step": 81800,
      "training_step_time": 8.463541507720947
    },
    {
      "epoch": 7.439613342285156e-06,
      "model_forward_time": 1.7483315467834473,
      "step": 81801
    },
    {
      "epoch": 7.439613342285156e-06,
      "step": 81801,
      "training_step_time": 7.520737171173096
    },
    {
      "epoch": 7.440567016601562e-06,
      "model_forward_time": 1.7491929531097412,
      "step": 81802
    },
    {
      "epoch": 7.440567016601562e-06,
      "step": 81802,
      "training_step_time": 8.261293649673462
    },
    {
      "epoch": 7.441520690917969e-06,
      "model_forward_time": 1.7492730617523193,
      "step": 81803
    },
    {
      "epoch": 7.441520690917969e-06,
      "step": 81803,
      "training_step_time": 8.28661036491394
    },
    {
      "epoch": 7.442474365234375e-06,
      "model_forward_time": 1.7479889392852783,
      "step": 81804
    },
    {
      "epoch": 7.442474365234375e-06,
      "step": 81804,
      "training_step_time": 8.861581325531006
    },
    {
      "epoch": 7.443428039550781e-06,
      "model_forward_time": 1.7489733695983887,
      "step": 81805
    },
    {
      "epoch": 7.443428039550781e-06,
      "step": 81805,
      "training_step_time": 7.811107635498047
    },
    {
      "epoch": 7.4443817138671875e-06,
      "model_forward_time": 1.7490670680999756,
      "step": 81806
    },
    {
      "epoch": 7.4443817138671875e-06,
      "step": 81806,
      "training_step_time": 8.470603704452515
    },
    {
      "epoch": 7.445335388183594e-06,
      "model_forward_time": 1.3771915435791016,
      "step": 81807
    },
    {
      "epoch": 7.445335388183594e-06,
      "step": 81807,
      "training_step_time": 9.129911661148071
    },
    {
      "epoch": 7.4462890625e-06,
      "model_forward_time": 1.746995210647583,
      "step": 81808
    },
    {
      "epoch": 7.4462890625e-06,
      "step": 81808,
      "training_step_time": 7.863384962081909
    },
    {
      "epoch": 7.447242736816406e-06,
      "model_forward_time": 2.4175968170166016,
      "step": 81809
    },
    {
      "epoch": 7.447242736816406e-06,
      "step": 81809,
      "training_step_time": 8.540054559707642
    },
    {
      "epoch": 7.448196411132812e-06,
      "grad_norm": 0.11082471907138824,
      "learning_rate": 8.777464055158814e-07,
      "loss": 0.0564,
      "step": 81810
    },
    {
      "epoch": 7.448196411132812e-06,
      "model_forward_time": 1.7419548034667969,
      "step": 81810
    },
    {
      "dynamics_loss_avg": 0.061400062404572964,
      "epoch": 7.448196411132812e-06,
      "step": 81810
    },
    {
      "action_loss_avg": 0.0022396741958800705,
      "epoch": 7.448196411132812e-06,
      "step": 81810
    },
    {
      "epoch": 7.448196411132812e-06,
      "step": 81810,
      "training_step_time": 7.997812509536743
    },
    {
      "epoch": 7.449150085449219e-06,
      "model_forward_time": 1.7496132850646973,
      "step": 81811
    },
    {
      "epoch": 7.449150085449219e-06,
      "step": 81811,
      "training_step_time": 7.69868278503418
    },
    {
      "epoch": 7.450103759765625e-06,
      "model_forward_time": 1.7720255851745605,
      "step": 81812
    },
    {
      "epoch": 7.450103759765625e-06,
      "step": 81812,
      "training_step_time": 7.784757137298584
    },
    {
      "epoch": 7.451057434082031e-06,
      "model_forward_time": 1.7518048286437988,
      "step": 81813
    },
    {
      "epoch": 7.451057434082031e-06,
      "step": 81813,
      "training_step_time": 7.855691432952881
    },
    {
      "epoch": 7.4520111083984376e-06,
      "model_forward_time": 1.7447431087493896,
      "step": 81814
    },
    {
      "epoch": 7.4520111083984376e-06,
      "step": 81814,
      "training_step_time": 8.690508127212524
    },
    {
      "epoch": 7.452964782714844e-06,
      "model_forward_time": 1.7451622486114502,
      "step": 81815
    },
    {
      "epoch": 7.452964782714844e-06,
      "step": 81815,
      "training_step_time": 8.990303754806519
    },
    {
      "epoch": 7.45391845703125e-06,
      "model_forward_time": 1.7481443881988525,
      "step": 81816
    },
    {
      "epoch": 7.45391845703125e-06,
      "step": 81816,
      "training_step_time": 7.901042699813843
    },
    {
      "epoch": 7.454872131347656e-06,
      "model_forward_time": 1.7523982524871826,
      "step": 81817
    },
    {
      "epoch": 7.454872131347656e-06,
      "step": 81817,
      "training_step_time": 7.748903036117554
    },
    {
      "epoch": 7.455825805664062e-06,
      "model_forward_time": 2.37627911567688,
      "step": 81818
    },
    {
      "epoch": 7.455825805664062e-06,
      "step": 81818,
      "training_step_time": 7.890005826950073
    },
    {
      "epoch": 7.456779479980469e-06,
      "model_forward_time": 1.7486572265625,
      "step": 81819
    },
    {
      "epoch": 7.456779479980469e-06,
      "step": 81819,
      "training_step_time": 7.382143974304199
    },
    {
      "epoch": 7.457733154296875e-06,
      "grad_norm": 0.09465285390615463,
      "learning_rate": 8.768108765562577e-07,
      "loss": 0.0563,
      "step": 81820
    },
    {
      "epoch": 7.457733154296875e-06,
      "model_forward_time": 1.7269937992095947,
      "step": 81820
    },
    {
      "dynamics_loss_avg": 0.05075051579624414,
      "epoch": 7.457733154296875e-06,
      "step": 81820
    },
    {
      "action_loss_avg": 0.0014385143818799405,
      "epoch": 7.457733154296875e-06,
      "step": 81820
    },
    {
      "epoch": 7.457733154296875e-06,
      "step": 81820,
      "training_step_time": 7.712318420410156
    },
    {
      "epoch": 7.458686828613281e-06,
      "model_forward_time": 1.7681090831756592,
      "step": 81821
    },
    {
      "epoch": 7.458686828613281e-06,
      "step": 81821,
      "training_step_time": 8.455342292785645
    },
    {
      "epoch": 7.4596405029296876e-06,
      "model_forward_time": 1.755734920501709,
      "step": 81822
    },
    {
      "epoch": 7.4596405029296876e-06,
      "step": 81822,
      "training_step_time": 7.824644327163696
    },
    {
      "epoch": 7.460594177246094e-06,
      "model_forward_time": 1.7447798252105713,
      "step": 81823
    },
    {
      "epoch": 7.460594177246094e-06,
      "step": 81823,
      "training_step_time": 9.094446897506714
    },
    {
      "epoch": 7.4615478515625e-06,
      "model_forward_time": 1.747471809387207,
      "step": 81824
    },
    {
      "epoch": 7.4615478515625e-06,
      "step": 81824,
      "training_step_time": 9.856996059417725
    },
    {
      "epoch": 7.462501525878906e-06,
      "model_forward_time": 1.7501282691955566,
      "step": 81825
    },
    {
      "epoch": 7.462501525878906e-06,
      "step": 81825,
      "training_step_time": 8.785936832427979
    },
    {
      "epoch": 7.463455200195312e-06,
      "model_forward_time": 1.7456824779510498,
      "step": 81826
    },
    {
      "epoch": 7.463455200195312e-06,
      "step": 81826,
      "training_step_time": 8.168353796005249
    },
    {
      "epoch": 7.464408874511719e-06,
      "model_forward_time": 1.7474451065063477,
      "step": 81827
    },
    {
      "epoch": 7.464408874511719e-06,
      "step": 81827,
      "training_step_time": 7.780798673629761
    },
    {
      "epoch": 7.465362548828125e-06,
      "model_forward_time": 2.413698434829712,
      "step": 81828
    },
    {
      "epoch": 7.465362548828125e-06,
      "step": 81828,
      "training_step_time": 7.6513755321502686
    },
    {
      "epoch": 7.466316223144531e-06,
      "model_forward_time": 1.7514266967773438,
      "step": 81829
    },
    {
      "epoch": 7.466316223144531e-06,
      "step": 81829,
      "training_step_time": 7.642272233963013
    },
    {
      "epoch": 7.4672698974609376e-06,
      "grad_norm": 0.08272869139909744,
      "learning_rate": 8.758757985023974e-07,
      "loss": 0.0572,
      "step": 81830
    },
    {
      "epoch": 7.4672698974609376e-06,
      "model_forward_time": 1.754929780960083,
      "step": 81830
    },
    {
      "dynamics_loss_avg": 0.06350968889892102,
      "epoch": 7.4672698974609376e-06,
      "step": 81830
    },
    {
      "action_loss_avg": 0.0018975895072799176,
      "epoch": 7.4672698974609376e-06,
      "step": 81830
    },
    {
      "epoch": 7.4672698974609376e-06,
      "step": 81830,
      "training_step_time": 7.70353627204895
    },
    {
      "epoch": 7.468223571777344e-06,
      "model_forward_time": 1.7620329856872559,
      "step": 81831
    },
    {
      "epoch": 7.468223571777344e-06,
      "step": 81831,
      "training_step_time": 9.796171188354492
    },
    {
      "epoch": 7.46917724609375e-06,
      "model_forward_time": 1.7470474243164062,
      "step": 81832
    },
    {
      "epoch": 7.46917724609375e-06,
      "step": 81832,
      "training_step_time": 8.296361923217773
    },
    {
      "epoch": 7.4701309204101564e-06,
      "model_forward_time": 1.7514405250549316,
      "step": 81833
    },
    {
      "epoch": 7.4701309204101564e-06,
      "step": 81833,
      "training_step_time": 8.345994710922241
    },
    {
      "epoch": 7.471084594726562e-06,
      "model_forward_time": 1.041818380355835,
      "step": 81834
    },
    {
      "epoch": 7.471084594726562e-06,
      "step": 81834,
      "training_step_time": 7.772886514663696
    },
    {
      "epoch": 7.472038269042969e-06,
      "model_forward_time": 1.7521302700042725,
      "step": 81835
    },
    {
      "epoch": 7.472038269042969e-06,
      "step": 81835,
      "training_step_time": 8.64520525932312
    },
    {
      "epoch": 7.472991943359375e-06,
      "model_forward_time": 1.7503724098205566,
      "step": 81836
    },
    {
      "epoch": 7.472991943359375e-06,
      "step": 81836,
      "training_step_time": 8.741260051727295
    },
    {
      "epoch": 7.473945617675781e-06,
      "model_forward_time": 2.426778554916382,
      "step": 81837
    },
    {
      "epoch": 7.473945617675781e-06,
      "step": 81837,
      "training_step_time": 8.560296297073364
    },
    {
      "epoch": 7.4748992919921876e-06,
      "model_forward_time": 1.7528443336486816,
      "step": 81838
    },
    {
      "epoch": 7.4748992919921876e-06,
      "step": 81838,
      "training_step_time": 7.774430513381958
    },
    {
      "epoch": 7.475852966308594e-06,
      "model_forward_time": 1.7610418796539307,
      "step": 81839
    },
    {
      "epoch": 7.475852966308594e-06,
      "step": 81839,
      "training_step_time": 8.639998435974121
    },
    {
      "epoch": 7.476806640625e-06,
      "grad_norm": 0.08102550357580185,
      "learning_rate": 8.749411714565598e-07,
      "loss": 0.0568,
      "step": 81840
    },
    {
      "epoch": 7.476806640625e-06,
      "model_forward_time": 1.7450363636016846,
      "step": 81840
    },
    {
      "dynamics_loss_avg": 0.059740543179214,
      "epoch": 7.476806640625e-06,
      "step": 81840
    },
    {
      "action_loss_avg": 0.001425482804188505,
      "epoch": 7.476806640625e-06,
      "step": 81840
    },
    {
      "epoch": 7.476806640625e-06,
      "step": 81840,
      "training_step_time": 8.333009958267212
    },
    {
      "epoch": 7.4777603149414064e-06,
      "model_forward_time": 1.7544190883636475,
      "step": 81841
    },
    {
      "epoch": 7.4777603149414064e-06,
      "step": 81841,
      "training_step_time": 8.395901441574097
    },
    {
      "epoch": 7.478713989257812e-06,
      "model_forward_time": 1.7403535842895508,
      "step": 81842
    },
    {
      "epoch": 7.478713989257812e-06,
      "step": 81842,
      "training_step_time": 8.509987354278564
    },
    {
      "epoch": 7.479667663574219e-06,
      "model_forward_time": 1.753521203994751,
      "step": 81843
    },
    {
      "epoch": 7.479667663574219e-06,
      "step": 81843,
      "training_step_time": 9.178407430648804
    },
    {
      "epoch": 7.480621337890625e-06,
      "model_forward_time": 1.7524149417877197,
      "step": 81844
    },
    {
      "epoch": 7.480621337890625e-06,
      "step": 81844,
      "training_step_time": 7.768734455108643
    },
    {
      "epoch": 7.481575012207031e-06,
      "model_forward_time": 1.7457993030548096,
      "step": 81845
    },
    {
      "epoch": 7.481575012207031e-06,
      "step": 81845,
      "training_step_time": 7.745212554931641
    },
    {
      "epoch": 7.482528686523438e-06,
      "model_forward_time": 2.4105734825134277,
      "step": 81846
    },
    {
      "epoch": 7.482528686523438e-06,
      "step": 81846,
      "training_step_time": 7.858380556106567
    },
    {
      "epoch": 7.483482360839844e-06,
      "model_forward_time": 1.753657579421997,
      "step": 81847
    },
    {
      "epoch": 7.483482360839844e-06,
      "step": 81847,
      "training_step_time": 7.711899518966675
    },
    {
      "epoch": 7.48443603515625e-06,
      "model_forward_time": 1.747837781906128,
      "step": 81848
    },
    {
      "epoch": 7.48443603515625e-06,
      "step": 81848,
      "training_step_time": 8.846429347991943
    },
    {
      "epoch": 7.4853897094726564e-06,
      "model_forward_time": 1.747663974761963,
      "step": 81849
    },
    {
      "epoch": 7.4853897094726564e-06,
      "step": 81849,
      "training_step_time": 8.536855220794678
    },
    {
      "epoch": 7.486343383789062e-06,
      "grad_norm": 0.13179849088191986,
      "learning_rate": 8.74006995520954e-07,
      "loss": 0.0576,
      "step": 81850
    },
    {
      "epoch": 7.486343383789062e-06,
      "model_forward_time": 1.753040075302124,
      "step": 81850
    },
    {
      "dynamics_loss_avg": 0.06260797679424286,
      "epoch": 7.486343383789062e-06,
      "step": 81850
    },
    {
      "action_loss_avg": 0.0009619578748242929,
      "epoch": 7.486343383789062e-06,
      "step": 81850
    },
    {
      "epoch": 7.486343383789062e-06,
      "step": 81850,
      "training_step_time": 10.662563800811768
    },
    {
      "epoch": 7.487297058105469e-06,
      "model_forward_time": 1.7461504936218262,
      "step": 81851
    },
    {
      "epoch": 7.487297058105469e-06,
      "step": 81851,
      "training_step_time": 7.824075937271118
    },
    {
      "epoch": 7.488250732421875e-06,
      "model_forward_time": 1.7457332611083984,
      "step": 81852
    },
    {
      "epoch": 7.488250732421875e-06,
      "step": 81852,
      "training_step_time": 7.809606552124023
    },
    {
      "epoch": 7.489204406738281e-06,
      "model_forward_time": 1.740107536315918,
      "step": 81853
    },
    {
      "epoch": 7.489204406738281e-06,
      "step": 81853,
      "training_step_time": 8.32435154914856
    },
    {
      "epoch": 7.490158081054688e-06,
      "model_forward_time": 1.744274616241455,
      "step": 81854
    },
    {
      "epoch": 7.490158081054688e-06,
      "step": 81854,
      "training_step_time": 7.673970699310303
    },
    {
      "epoch": 7.491111755371094e-06,
      "model_forward_time": 2.341789484024048,
      "step": 81855
    },
    {
      "epoch": 7.491111755371094e-06,
      "step": 81855,
      "training_step_time": 7.868942022323608
    },
    {
      "epoch": 7.4920654296875e-06,
      "model_forward_time": 1.7398097515106201,
      "step": 81856
    },
    {
      "epoch": 7.4920654296875e-06,
      "step": 81856,
      "training_step_time": 8.468190670013428
    },
    {
      "epoch": 7.4930191040039064e-06,
      "model_forward_time": 1.7423796653747559,
      "step": 81857
    },
    {
      "epoch": 7.4930191040039064e-06,
      "step": 81857,
      "training_step_time": 7.673016548156738
    },
    {
      "epoch": 7.493972778320312e-06,
      "model_forward_time": 1.7720983028411865,
      "step": 81858
    },
    {
      "epoch": 7.493972778320312e-06,
      "step": 81858,
      "training_step_time": 8.613372325897217
    },
    {
      "epoch": 7.494926452636719e-06,
      "model_forward_time": 1.7590551376342773,
      "step": 81859
    },
    {
      "epoch": 7.494926452636719e-06,
      "step": 81859,
      "training_step_time": 8.879657745361328
    },
    {
      "epoch": 7.495880126953125e-06,
      "grad_norm": 0.14005744457244873,
      "learning_rate": 8.730732707977396e-07,
      "loss": 0.0559,
      "step": 81860
    },
    {
      "epoch": 7.495880126953125e-06,
      "model_forward_time": 1.7443530559539795,
      "step": 81860
    },
    {
      "dynamics_loss_avg": 0.06038423143327236,
      "epoch": 7.495880126953125e-06,
      "step": 81860
    },
    {
      "action_loss_avg": 0.002016144129447639,
      "epoch": 7.495880126953125e-06,
      "step": 81860
    },
    {
      "epoch": 7.495880126953125e-06,
      "step": 81860,
      "training_step_time": 10.554991006851196
    },
    {
      "epoch": 7.496833801269531e-06,
      "model_forward_time": 1.7336583137512207,
      "step": 81861
    },
    {
      "epoch": 7.496833801269531e-06,
      "step": 81861,
      "training_step_time": 7.862251043319702
    },
    {
      "epoch": 7.497787475585938e-06,
      "model_forward_time": 1.739476203918457,
      "step": 81862
    },
    {
      "epoch": 7.497787475585938e-06,
      "step": 81862,
      "training_step_time": 8.310709714889526
    },
    {
      "epoch": 7.498741149902344e-06,
      "model_forward_time": 1.737006425857544,
      "step": 81863
    },
    {
      "epoch": 7.498741149902344e-06,
      "step": 81863,
      "training_step_time": 7.743380784988403
    },
    {
      "epoch": 7.49969482421875e-06,
      "model_forward_time": 1.727987289428711,
      "step": 81864
    },
    {
      "epoch": 7.49969482421875e-06,
      "step": 81864,
      "training_step_time": 7.809581756591797
    },
    {
      "epoch": 7.5006484985351565e-06,
      "model_forward_time": 2.3533124923706055,
      "step": 81865
    },
    {
      "epoch": 7.5006484985351565e-06,
      "step": 81865,
      "training_step_time": 7.673528671264648
    },
    {
      "epoch": 7.501602172851562e-06,
      "model_forward_time": 1.731990098953247,
      "step": 81866
    },
    {
      "epoch": 7.501602172851562e-06,
      "step": 81866,
      "training_step_time": 7.68270206451416
    },
    {
      "epoch": 7.502555847167969e-06,
      "model_forward_time": 1.7601759433746338,
      "step": 81867
    },
    {
      "epoch": 7.502555847167969e-06,
      "step": 81867,
      "training_step_time": 8.617811679840088
    },
    {
      "epoch": 7.503509521484375e-06,
      "model_forward_time": 1.7364940643310547,
      "step": 81868
    },
    {
      "epoch": 7.503509521484375e-06,
      "step": 81868,
      "training_step_time": 7.827824592590332
    },
    {
      "epoch": 7.504463195800781e-06,
      "model_forward_time": 1.734494686126709,
      "step": 81869
    },
    {
      "epoch": 7.504463195800781e-06,
      "step": 81869,
      "training_step_time": 8.750287771224976
    },
    {
      "epoch": 7.505416870117188e-06,
      "grad_norm": 0.0756366029381752,
      "learning_rate": 8.721399973890282e-07,
      "loss": 0.0564,
      "step": 81870
    },
    {
      "epoch": 7.505416870117188e-06,
      "model_forward_time": 1.7248094081878662,
      "step": 81870
    },
    {
      "dynamics_loss_avg": 0.06374387936666608,
      "epoch": 7.505416870117188e-06,
      "step": 81870
    },
    {
      "action_loss_avg": 0.0014763793267775327,
      "epoch": 7.505416870117188e-06,
      "step": 81870
    },
    {
      "epoch": 7.505416870117188e-06,
      "step": 81870,
      "training_step_time": 8.882081270217896
    },
    {
      "epoch": 7.506370544433594e-06,
      "model_forward_time": 1.7382166385650635,
      "step": 81871
    },
    {
      "epoch": 7.506370544433594e-06,
      "step": 81871,
      "training_step_time": 7.729491472244263
    },
    {
      "epoch": 7.50732421875e-06,
      "model_forward_time": 1.7346575260162354,
      "step": 81872
    },
    {
      "epoch": 7.50732421875e-06,
      "step": 81872,
      "training_step_time": 7.733773708343506
    },
    {
      "epoch": 7.5082778930664065e-06,
      "model_forward_time": 1.734243392944336,
      "step": 81873
    },
    {
      "epoch": 7.5082778930664065e-06,
      "step": 81873,
      "training_step_time": 7.8324294090271
    },
    {
      "epoch": 7.509231567382812e-06,
      "model_forward_time": 2.364694118499756,
      "step": 81874
    },
    {
      "epoch": 7.509231567382812e-06,
      "step": 81874,
      "training_step_time": 7.694497585296631
    },
    {
      "epoch": 7.510185241699219e-06,
      "model_forward_time": 1.7384228706359863,
      "step": 81875
    },
    {
      "epoch": 7.510185241699219e-06,
      "step": 81875,
      "training_step_time": 8.720863580703735
    },
    {
      "epoch": 7.511138916015625e-06,
      "model_forward_time": 1.759192943572998,
      "step": 81876
    },
    {
      "epoch": 7.511138916015625e-06,
      "step": 81876,
      "training_step_time": 7.751096487045288
    },
    {
      "epoch": 7.512092590332031e-06,
      "model_forward_time": 1.7370471954345703,
      "step": 81877
    },
    {
      "epoch": 7.512092590332031e-06,
      "step": 81877,
      "training_step_time": 7.822141647338867
    },
    {
      "epoch": 7.513046264648438e-06,
      "model_forward_time": 1.7354826927185059,
      "step": 81878
    },
    {
      "epoch": 7.513046264648438e-06,
      "step": 81878,
      "training_step_time": 7.749670028686523
    },
    {
      "epoch": 7.513999938964843e-06,
      "model_forward_time": 1.733159065246582,
      "step": 81879
    },
    {
      "epoch": 7.513999938964843e-06,
      "step": 81879,
      "training_step_time": 8.83571195602417
    },
    {
      "epoch": 7.51495361328125e-06,
      "grad_norm": 0.08318745344877243,
      "learning_rate": 8.712071753968793e-07,
      "loss": 0.0562,
      "step": 81880
    },
    {
      "epoch": 7.51495361328125e-06,
      "model_forward_time": 1.719712257385254,
      "step": 81880
    },
    {
      "dynamics_loss_avg": 0.046151093952357766,
      "epoch": 7.51495361328125e-06,
      "step": 81880
    },
    {
      "action_loss_avg": 0.0013776202787994408,
      "epoch": 7.51495361328125e-06,
      "step": 81880
    },
    {
      "epoch": 7.51495361328125e-06,
      "step": 81880,
      "training_step_time": 7.761155128479004
    },
    {
      "epoch": 7.5159072875976565e-06,
      "model_forward_time": 1.7382690906524658,
      "step": 81881
    },
    {
      "epoch": 7.5159072875976565e-06,
      "step": 81881,
      "training_step_time": 7.797331094741821
    },
    {
      "epoch": 7.516860961914062e-06,
      "model_forward_time": 1.3545916080474854,
      "step": 81882
    },
    {
      "epoch": 7.516860961914062e-06,
      "step": 81882,
      "training_step_time": 7.847262620925903
    },
    {
      "epoch": 7.517814636230469e-06,
      "model_forward_time": 2.4117796421051025,
      "step": 81883
    },
    {
      "epoch": 7.517814636230469e-06,
      "step": 81883,
      "training_step_time": 7.691426515579224
    },
    {
      "epoch": 7.518768310546875e-06,
      "model_forward_time": 1.731581687927246,
      "step": 81884
    },
    {
      "epoch": 7.518768310546875e-06,
      "step": 81884,
      "training_step_time": 9.328310489654541
    },
    {
      "epoch": 7.519721984863281e-06,
      "model_forward_time": 1.7423529624938965,
      "step": 81885
    },
    {
      "epoch": 7.519721984863281e-06,
      "step": 81885,
      "training_step_time": 7.6731908321380615
    },
    {
      "epoch": 7.520675659179688e-06,
      "model_forward_time": 1.7277095317840576,
      "step": 81886
    },
    {
      "epoch": 7.520675659179688e-06,
      "step": 81886,
      "training_step_time": 9.356871843338013
    },
    {
      "epoch": 7.521629333496093e-06,
      "model_forward_time": 1.7334885597229004,
      "step": 81887
    },
    {
      "epoch": 7.521629333496093e-06,
      "step": 81887,
      "training_step_time": 8.31981372833252
    },
    {
      "epoch": 7.5225830078125e-06,
      "model_forward_time": 1.3613369464874268,
      "step": 81888
    },
    {
      "epoch": 7.5225830078125e-06,
      "step": 81888,
      "training_step_time": 8.495165824890137
    },
    {
      "epoch": 7.5235366821289065e-06,
      "model_forward_time": 1.3585882186889648,
      "step": 81889
    },
    {
      "epoch": 7.5235366821289065e-06,
      "step": 81889,
      "training_step_time": 7.778823137283325
    },
    {
      "epoch": 7.524490356445312e-06,
      "grad_norm": 0.2747381031513214,
      "learning_rate": 8.702748049233073e-07,
      "loss": 0.0569,
      "step": 81890
    },
    {
      "epoch": 7.524490356445312e-06,
      "model_forward_time": 1.7323439121246338,
      "step": 81890
    },
    {
      "dynamics_loss_avg": 0.043462148495018484,
      "epoch": 7.524490356445312e-06,
      "step": 81890
    },
    {
      "action_loss_avg": 0.0012599165522260592,
      "epoch": 7.524490356445312e-06,
      "step": 81890
    },
    {
      "epoch": 7.524490356445312e-06,
      "step": 81890,
      "training_step_time": 8.323594570159912
    },
    {
      "epoch": 7.525444030761719e-06,
      "model_forward_time": 1.7329039573669434,
      "step": 81891
    },
    {
      "epoch": 7.525444030761719e-06,
      "step": 81891,
      "training_step_time": 9.31873869895935
    },
    {
      "epoch": 7.526397705078125e-06,
      "model_forward_time": 1.7339494228363037,
      "step": 81892
    },
    {
      "epoch": 7.526397705078125e-06,
      "step": 81892,
      "training_step_time": 7.669161796569824
    },
    {
      "epoch": 7.527351379394531e-06,
      "model_forward_time": 2.3610968589782715,
      "step": 81893
    },
    {
      "epoch": 7.527351379394531e-06,
      "step": 81893,
      "training_step_time": 7.5693089962005615
    },
    {
      "epoch": 7.528305053710938e-06,
      "model_forward_time": 1.7359778881072998,
      "step": 81894
    },
    {
      "epoch": 7.528305053710938e-06,
      "step": 81894,
      "training_step_time": 7.704397916793823
    },
    {
      "epoch": 7.529258728027343e-06,
      "model_forward_time": 1.7316572666168213,
      "step": 81895
    },
    {
      "epoch": 7.529258728027343e-06,
      "step": 81895,
      "training_step_time": 8.657192468643188
    },
    {
      "epoch": 7.53021240234375e-06,
      "model_forward_time": 1.7494964599609375,
      "step": 81896
    },
    {
      "epoch": 7.53021240234375e-06,
      "step": 81896,
      "training_step_time": 7.808294057846069
    },
    {
      "epoch": 7.5311660766601565e-06,
      "model_forward_time": 1.7364535331726074,
      "step": 81897
    },
    {
      "epoch": 7.5311660766601565e-06,
      "step": 81897,
      "training_step_time": 9.230181455612183
    },
    {
      "epoch": 7.532119750976562e-06,
      "model_forward_time": 1.7337794303894043,
      "step": 81898
    },
    {
      "epoch": 7.532119750976562e-06,
      "step": 81898,
      "training_step_time": 7.798677206039429
    },
    {
      "epoch": 7.533073425292969e-06,
      "model_forward_time": 1.7347662448883057,
      "step": 81899
    },
    {
      "epoch": 7.533073425292969e-06,
      "step": 81899,
      "training_step_time": 7.710303544998169
    },
    {
      "epoch": 7.534027099609375e-06,
      "grad_norm": 0.0921594426035881,
      "learning_rate": 8.693428860702719e-07,
      "loss": 0.0552,
      "step": 81900
    },
    {
      "epoch": 7.534027099609375e-06,
      "model_forward_time": 1.7240874767303467,
      "step": 81900
    },
    {
      "dynamics_loss_avg": 0.05025154538452625,
      "epoch": 7.534027099609375e-06,
      "step": 81900
    },
    {
      "action_loss_avg": 0.0017567256814800202,
      "epoch": 7.534027099609375e-06,
      "step": 81900
    },
    {
      "epoch": 7.534027099609375e-06,
      "step": 81900,
      "training_step_time": 7.761183500289917
    },
    {
      "epoch": 7.534980773925781e-06,
      "model_forward_time": 1.7351865768432617,
      "step": 81901
    },
    {
      "epoch": 7.534980773925781e-06,
      "step": 81901,
      "training_step_time": 8.573576927185059
    },
    {
      "epoch": 7.535934448242188e-06,
      "model_forward_time": 2.3788764476776123,
      "step": 81902
    },
    {
      "epoch": 7.535934448242188e-06,
      "step": 81902,
      "training_step_time": 7.612177848815918
    },
    {
      "epoch": 7.536888122558593e-06,
      "model_forward_time": 1.7335247993469238,
      "step": 81903
    },
    {
      "epoch": 7.536888122558593e-06,
      "step": 81903,
      "training_step_time": 8.787644863128662
    },
    {
      "epoch": 7.537841796875e-06,
      "model_forward_time": 1.7446963787078857,
      "step": 81904
    },
    {
      "epoch": 7.537841796875e-06,
      "step": 81904,
      "training_step_time": 7.862138748168945
    },
    {
      "epoch": 7.5387954711914065e-06,
      "model_forward_time": 1.7476389408111572,
      "step": 81905
    },
    {
      "epoch": 7.5387954711914065e-06,
      "step": 81905,
      "training_step_time": 8.415562868118286
    },
    {
      "epoch": 7.539749145507812e-06,
      "model_forward_time": 1.7369625568389893,
      "step": 81906
    },
    {
      "epoch": 7.539749145507812e-06,
      "step": 81906,
      "training_step_time": 8.569864511489868
    },
    {
      "epoch": 7.540702819824219e-06,
      "model_forward_time": 1.7341296672821045,
      "step": 81907
    },
    {
      "epoch": 7.540702819824219e-06,
      "step": 81907,
      "training_step_time": 8.90360140800476
    },
    {
      "epoch": 7.541656494140625e-06,
      "model_forward_time": 1.7344346046447754,
      "step": 81908
    },
    {
      "epoch": 7.541656494140625e-06,
      "step": 81908,
      "training_step_time": 9.269160509109497
    },
    {
      "epoch": 7.542610168457031e-06,
      "model_forward_time": 1.7384328842163086,
      "step": 81909
    },
    {
      "epoch": 7.542610168457031e-06,
      "step": 81909,
      "training_step_time": 7.915763854980469
    },
    {
      "epoch": 7.543563842773438e-06,
      "grad_norm": 0.11449377983808517,
      "learning_rate": 8.684114189396897e-07,
      "loss": 0.0552,
      "step": 81910
    },
    {
      "epoch": 7.543563842773438e-06,
      "model_forward_time": 1.7218658924102783,
      "step": 81910
    },
    {
      "dynamics_loss_avg": 0.06814859546720982,
      "epoch": 7.543563842773438e-06,
      "step": 81910
    },
    {
      "action_loss_avg": 0.0011919555021449924,
      "epoch": 7.543563842773438e-06,
      "step": 81910
    },
    {
      "epoch": 7.543563842773438e-06,
      "step": 81910,
      "training_step_time": 7.874502420425415
    },
    {
      "epoch": 7.544517517089843e-06,
      "model_forward_time": 2.3868119716644287,
      "step": 81911
    },
    {
      "epoch": 7.544517517089843e-06,
      "step": 81911,
      "training_step_time": 7.738844394683838
    },
    {
      "epoch": 7.54547119140625e-06,
      "model_forward_time": 1.0121443271636963,
      "step": 81912
    },
    {
      "epoch": 7.54547119140625e-06,
      "step": 81912,
      "training_step_time": 8.471311807632446
    },
    {
      "epoch": 7.5464248657226565e-06,
      "model_forward_time": 1.7562649250030518,
      "step": 81913
    },
    {
      "epoch": 7.5464248657226565e-06,
      "step": 81913,
      "training_step_time": 8.969707250595093
    },
    {
      "epoch": 7.547378540039062e-06,
      "model_forward_time": 1.7519009113311768,
      "step": 81914
    },
    {
      "epoch": 7.547378540039062e-06,
      "step": 81914,
      "training_step_time": 7.833612442016602
    },
    {
      "epoch": 7.548332214355469e-06,
      "model_forward_time": 1.737208366394043,
      "step": 81915
    },
    {
      "epoch": 7.548332214355469e-06,
      "step": 81915,
      "training_step_time": 9.89253282546997
    },
    {
      "epoch": 7.549285888671875e-06,
      "model_forward_time": 1.7282204627990723,
      "step": 81916
    },
    {
      "epoch": 7.549285888671875e-06,
      "step": 81916,
      "training_step_time": 8.471539974212646
    },
    {
      "epoch": 7.550239562988281e-06,
      "model_forward_time": 1.7430083751678467,
      "step": 81917
    },
    {
      "epoch": 7.550239562988281e-06,
      "step": 81917,
      "training_step_time": 8.404257774353027
    },
    {
      "epoch": 7.551193237304688e-06,
      "model_forward_time": 1.7523870468139648,
      "step": 81918
    },
    {
      "epoch": 7.551193237304688e-06,
      "step": 81918,
      "training_step_time": 7.940754413604736
    },
    {
      "epoch": 7.552146911621093e-06,
      "model_forward_time": 1.7584264278411865,
      "step": 81919
    },
    {
      "epoch": 7.552146911621093e-06,
      "step": 81919,
      "training_step_time": 7.902249574661255
    },
    {
      "epoch": 7.5531005859375e-06,
      "grad_norm": 0.1320681869983673,
      "learning_rate": 8.674804036334223e-07,
      "loss": 0.0558,
      "step": 81920
    },
    {
      "epoch": 7.5531005859375e-06,
      "model_forward_time": 1.3749990463256836,
      "step": 81920
    },
    {
      "dynamics_loss_avg": 0.04529331512749195,
      "epoch": 7.5531005859375e-06,
      "step": 81920
    },
    {
      "action_loss_avg": 0.001215430282172747,
      "epoch": 7.5531005859375e-06,
      "step": 81920
    },
    {
      "epoch": 7.5531005859375e-06,
      "step": 81920,
      "training_step_time": 8.87183952331543
    },
    {
      "epoch": 7.5540542602539065e-06,
      "model_forward_time": 1.7563714981079102,
      "step": 81921
    },
    {
      "epoch": 7.5540542602539065e-06,
      "step": 81921,
      "training_step_time": 8.463283777236938
    },
    {
      "epoch": 7.555007934570312e-06,
      "model_forward_time": 1.748581886291504,
      "step": 81922
    },
    {
      "epoch": 7.555007934570312e-06,
      "step": 81922,
      "training_step_time": 9.638179063796997
    },
    {
      "epoch": 7.555961608886719e-06,
      "model_forward_time": 1.7776107788085938,
      "step": 81923
    },
    {
      "epoch": 7.555961608886719e-06,
      "step": 81923,
      "training_step_time": 8.33408260345459
    },
    {
      "epoch": 7.556915283203125e-06,
      "model_forward_time": 1.7477965354919434,
      "step": 81924
    },
    {
      "epoch": 7.556915283203125e-06,
      "step": 81924,
      "training_step_time": 8.01832365989685
    },
    {
      "epoch": 7.557868957519531e-06,
      "model_forward_time": 1.7497594356536865,
      "step": 81925
    },
    {
      "epoch": 7.557868957519531e-06,
      "step": 81925,
      "training_step_time": 8.556910991668701
    },
    {
      "epoch": 7.558822631835938e-06,
      "model_forward_time": 1.376971960067749,
      "step": 81926
    },
    {
      "epoch": 7.558822631835938e-06,
      "step": 81926,
      "training_step_time": 8.759552717208862
    },
    {
      "epoch": 7.559776306152343e-06,
      "model_forward_time": 1.7549502849578857,
      "step": 81927
    },
    {
      "epoch": 7.559776306152343e-06,
      "step": 81927,
      "training_step_time": 8.666937828063965
    },
    {
      "epoch": 7.56072998046875e-06,
      "model_forward_time": 1.7582509517669678,
      "step": 81928
    },
    {
      "epoch": 7.56072998046875e-06,
      "step": 81928,
      "training_step_time": 8.759366512298584
    },
    {
      "epoch": 7.5616836547851565e-06,
      "model_forward_time": 1.7506027221679688,
      "step": 81929
    },
    {
      "epoch": 7.5616836547851565e-06,
      "step": 81929,
      "training_step_time": 7.816912889480591
    },
    {
      "epoch": 7.562637329101562e-06,
      "grad_norm": 0.10827012360095978,
      "learning_rate": 8.665498402532857e-07,
      "loss": 0.0551,
      "step": 81930
    },
    {
      "epoch": 7.562637329101562e-06,
      "model_forward_time": 2.3806567192077637,
      "step": 81930
    },
    {
      "dynamics_loss_avg": 0.05899054650217295,
      "epoch": 7.562637329101562e-06,
      "step": 81930
    },
    {
      "action_loss_avg": 0.0014875135733745993,
      "epoch": 7.562637329101562e-06,
      "step": 81930
    },
    {
      "epoch": 7.562637329101562e-06,
      "step": 81930,
      "training_step_time": 7.838451623916626
    },
    {
      "epoch": 7.563591003417969e-06,
      "model_forward_time": 1.7555384635925293,
      "step": 81931
    },
    {
      "epoch": 7.563591003417969e-06,
      "step": 81931,
      "training_step_time": 8.789762258529663
    },
    {
      "epoch": 7.564544677734375e-06,
      "model_forward_time": 1.7544119358062744,
      "step": 81932
    },
    {
      "epoch": 7.564544677734375e-06,
      "step": 81932,
      "training_step_time": 9.321656703948975
    },
    {
      "epoch": 7.565498352050781e-06,
      "model_forward_time": 1.7495789527893066,
      "step": 81933
    },
    {
      "epoch": 7.565498352050781e-06,
      "step": 81933,
      "training_step_time": 8.021888256072998
    },
    {
      "epoch": 7.566452026367188e-06,
      "model_forward_time": 1.7523579597473145,
      "step": 81934
    },
    {
      "epoch": 7.566452026367188e-06,
      "step": 81934,
      "training_step_time": 8.65231704711914
    },
    {
      "epoch": 7.567405700683593e-06,
      "model_forward_time": 1.760650396347046,
      "step": 81935
    },
    {
      "epoch": 7.567405700683593e-06,
      "step": 81935,
      "training_step_time": 7.732406854629517
    },
    {
      "epoch": 7.568359375e-06,
      "model_forward_time": 1.7517447471618652,
      "step": 81936
    },
    {
      "epoch": 7.568359375e-06,
      "step": 81936,
      "training_step_time": 8.450326681137085
    },
    {
      "epoch": 7.5693130493164065e-06,
      "model_forward_time": 1.7508745193481445,
      "step": 81937
    },
    {
      "epoch": 7.5693130493164065e-06,
      "step": 81937,
      "training_step_time": 7.232032775878906
    },
    {
      "epoch": 7.570266723632812e-06,
      "model_forward_time": 1.7557647228240967,
      "step": 81938
    },
    {
      "epoch": 7.570266723632812e-06,
      "step": 81938,
      "training_step_time": 7.882564067840576
    },
    {
      "epoch": 7.571220397949219e-06,
      "model_forward_time": 2.3736367225646973,
      "step": 81939
    },
    {
      "epoch": 7.571220397949219e-06,
      "step": 81939,
      "training_step_time": 8.655997514724731
    },
    {
      "epoch": 7.572174072265625e-06,
      "grad_norm": 0.11873084306716919,
      "learning_rate": 8.656197289010432e-07,
      "loss": 0.0548,
      "step": 81940
    },
    {
      "epoch": 7.572174072265625e-06,
      "model_forward_time": 1.7526137828826904,
      "step": 81940
    },
    {
      "dynamics_loss_avg": 0.059118757117539646,
      "epoch": 7.572174072265625e-06,
      "step": 81940
    },
    {
      "action_loss_avg": 0.0018220643687527627,
      "epoch": 7.572174072265625e-06,
      "step": 81940
    },
    {
      "epoch": 7.572174072265625e-06,
      "step": 81940,
      "training_step_time": 7.764722585678101
    },
    {
      "epoch": 7.573127746582031e-06,
      "model_forward_time": 1.7624647617340088,
      "step": 81941
    },
    {
      "epoch": 7.573127746582031e-06,
      "step": 81941,
      "training_step_time": 9.397521734237671
    },
    {
      "epoch": 7.574081420898438e-06,
      "model_forward_time": 1.838285207748413,
      "step": 81942
    },
    {
      "epoch": 7.574081420898438e-06,
      "step": 81942,
      "training_step_time": 8.896613121032715
    },
    {
      "epoch": 7.575035095214843e-06,
      "model_forward_time": 1.760072946548462,
      "step": 81943
    },
    {
      "epoch": 7.575035095214843e-06,
      "step": 81943,
      "training_step_time": 7.8209192752838135
    },
    {
      "epoch": 7.57598876953125e-06,
      "model_forward_time": 1.763765573501587,
      "step": 81944
    },
    {
      "epoch": 7.57598876953125e-06,
      "step": 81944,
      "training_step_time": 7.819641351699829
    },
    {
      "epoch": 7.5769424438476565e-06,
      "model_forward_time": 1.7397503852844238,
      "step": 81945
    },
    {
      "epoch": 7.5769424438476565e-06,
      "step": 81945,
      "training_step_time": 7.772501707077026
    },
    {
      "epoch": 7.577896118164062e-06,
      "model_forward_time": 1.7520413398742676,
      "step": 81946
    },
    {
      "epoch": 7.577896118164062e-06,
      "step": 81946,
      "training_step_time": 7.746986627578735
    },
    {
      "epoch": 7.578849792480469e-06,
      "model_forward_time": 1.0434343814849854,
      "step": 81947
    },
    {
      "epoch": 7.578849792480469e-06,
      "step": 81947,
      "training_step_time": 7.814124822616577
    },
    {
      "epoch": 7.579803466796875e-06,
      "model_forward_time": 2.407806634902954,
      "step": 81948
    },
    {
      "epoch": 7.579803466796875e-06,
      "step": 81948,
      "training_step_time": 8.514819860458374
    },
    {
      "epoch": 7.580757141113281e-06,
      "model_forward_time": 1.7558434009552002,
      "step": 81949
    },
    {
      "epoch": 7.580757141113281e-06,
      "step": 81949,
      "training_step_time": 7.838803052902222
    },
    {
      "epoch": 7.581710815429688e-06,
      "grad_norm": 0.0834319144487381,
      "learning_rate": 8.646900696784122e-07,
      "loss": 0.0545,
      "step": 81950
    },
    {
      "epoch": 7.581710815429688e-06,
      "model_forward_time": 1.7454595565795898,
      "step": 81950
    },
    {
      "dynamics_loss_avg": 0.055894849635660646,
      "epoch": 7.581710815429688e-06,
      "step": 81950
    },
    {
      "action_loss_avg": 0.0016002014715922996,
      "epoch": 7.581710815429688e-06,
      "step": 81950
    },
    {
      "epoch": 7.581710815429688e-06,
      "step": 81950,
      "training_step_time": 8.578007936477661
    },
    {
      "epoch": 7.582664489746093e-06,
      "model_forward_time": 1.740251064300537,
      "step": 81951
    },
    {
      "epoch": 7.582664489746093e-06,
      "step": 81951,
      "training_step_time": 10.13979458808899
    },
    {
      "epoch": 7.5836181640625e-06,
      "model_forward_time": 1.734194040298462,
      "step": 81952
    },
    {
      "epoch": 7.5836181640625e-06,
      "step": 81952,
      "training_step_time": 7.9083311557769775
    },
    {
      "epoch": 7.5845718383789065e-06,
      "model_forward_time": 1.7404274940490723,
      "step": 81953
    },
    {
      "epoch": 7.5845718383789065e-06,
      "step": 81953,
      "training_step_time": 7.858597993850708
    },
    {
      "epoch": 7.585525512695312e-06,
      "model_forward_time": 1.746607780456543,
      "step": 81954
    },
    {
      "epoch": 7.585525512695312e-06,
      "step": 81954,
      "training_step_time": 8.704859018325806
    },
    {
      "epoch": 7.586479187011719e-06,
      "model_forward_time": 1.7445321083068848,
      "step": 81955
    },
    {
      "epoch": 7.586479187011719e-06,
      "step": 81955,
      "training_step_time": 7.769755840301514
    },
    {
      "epoch": 7.587432861328125e-06,
      "model_forward_time": 1.7329888343811035,
      "step": 81956
    },
    {
      "epoch": 7.587432861328125e-06,
      "step": 81956,
      "training_step_time": 7.891224145889282
    },
    {
      "epoch": 7.588386535644531e-06,
      "model_forward_time": 1.7351229190826416,
      "step": 81957
    },
    {
      "epoch": 7.588386535644531e-06,
      "step": 81957,
      "training_step_time": 9.186433553695679
    },
    {
      "epoch": 7.589340209960938e-06,
      "model_forward_time": 1.3530211448669434,
      "step": 81958
    },
    {
      "epoch": 7.589340209960938e-06,
      "step": 81958,
      "training_step_time": 7.673387765884399
    },
    {
      "epoch": 7.590293884277343e-06,
      "model_forward_time": 1.7320129871368408,
      "step": 81959
    },
    {
      "epoch": 7.590293884277343e-06,
      "step": 81959,
      "training_step_time": 9.133653163909912
    },
    {
      "epoch": 7.59124755859375e-06,
      "grad_norm": 0.09296523034572601,
      "learning_rate": 8.637608626870553e-07,
      "loss": 0.0552,
      "step": 81960
    },
    {
      "epoch": 7.59124755859375e-06,
      "model_forward_time": 1.801581621170044,
      "step": 81960
    },
    {
      "dynamics_loss_avg": 0.06221039406955242,
      "epoch": 7.59124755859375e-06,
      "step": 81960
    },
    {
      "action_loss_avg": 0.0014275682042352855,
      "epoch": 7.59124755859375e-06,
      "step": 81960
    },
    {
      "epoch": 7.59124755859375e-06,
      "step": 81960,
      "training_step_time": 9.258028030395508
    },
    {
      "epoch": 7.5922012329101565e-06,
      "model_forward_time": 1.7387197017669678,
      "step": 81961
    },
    {
      "epoch": 7.5922012329101565e-06,
      "step": 81961,
      "training_step_time": 9.073936700820923
    },
    {
      "epoch": 7.593154907226562e-06,
      "model_forward_time": 1.7432935237884521,
      "step": 81962
    },
    {
      "epoch": 7.593154907226562e-06,
      "step": 81962,
      "training_step_time": 7.836201429367065
    },
    {
      "epoch": 7.594108581542969e-06,
      "model_forward_time": 1.7307746410369873,
      "step": 81963
    },
    {
      "epoch": 7.594108581542969e-06,
      "step": 81963,
      "training_step_time": 8.842264890670776
    },
    {
      "epoch": 7.595062255859375e-06,
      "model_forward_time": 1.7400434017181396,
      "step": 81964
    },
    {
      "epoch": 7.595062255859375e-06,
      "step": 81964,
      "training_step_time": 7.776960372924805
    },
    {
      "epoch": 7.596015930175781e-06,
      "model_forward_time": 1.0189244747161865,
      "step": 81965
    },
    {
      "epoch": 7.596015930175781e-06,
      "step": 81965,
      "training_step_time": 7.930158615112305
    },
    {
      "epoch": 7.596969604492188e-06,
      "model_forward_time": 1.7379591464996338,
      "step": 81966
    },
    {
      "epoch": 7.596969604492188e-06,
      "step": 81966,
      "training_step_time": 8.360457420349121
    },
    {
      "epoch": 7.597923278808593e-06,
      "model_forward_time": 2.389308214187622,
      "step": 81967
    },
    {
      "epoch": 7.597923278808593e-06,
      "step": 81967,
      "training_step_time": 7.89147686958313
    },
    {
      "epoch": 7.598876953125e-06,
      "model_forward_time": 1.7350666522979736,
      "step": 81968
    },
    {
      "epoch": 7.598876953125e-06,
      "step": 81968,
      "training_step_time": 8.977011680603027
    },
    {
      "epoch": 7.5998306274414065e-06,
      "model_forward_time": 1.773350715637207,
      "step": 81969
    },
    {
      "epoch": 7.5998306274414065e-06,
      "step": 81969,
      "training_step_time": 7.949204683303833
    },
    {
      "epoch": 7.600784301757812e-06,
      "grad_norm": 0.08261139690876007,
      "learning_rate": 8.628321080285945e-07,
      "loss": 0.0558,
      "step": 81970
    },
    {
      "epoch": 7.600784301757812e-06,
      "model_forward_time": 1.7309515476226807,
      "step": 81970
    },
    {
      "dynamics_loss_avg": 0.06940027046948671,
      "epoch": 7.600784301757812e-06,
      "step": 81970
    },
    {
      "action_loss_avg": 0.0015669762389734388,
      "epoch": 7.600784301757812e-06,
      "step": 81970
    },
    {
      "epoch": 7.600784301757812e-06,
      "step": 81970,
      "training_step_time": 8.030360221862793
    },
    {
      "epoch": 7.601737976074219e-06,
      "model_forward_time": 1.7360494136810303,
      "step": 81971
    },
    {
      "epoch": 7.601737976074219e-06,
      "step": 81971,
      "training_step_time": 7.853183031082153
    },
    {
      "epoch": 7.602691650390625e-06,
      "model_forward_time": 1.7414031028747559,
      "step": 81972
    },
    {
      "epoch": 7.602691650390625e-06,
      "step": 81972,
      "training_step_time": 7.779187202453613
    },
    {
      "epoch": 7.603645324707031e-06,
      "model_forward_time": 1.7427680492401123,
      "step": 81973
    },
    {
      "epoch": 7.603645324707031e-06,
      "step": 81973,
      "training_step_time": 7.987919092178345
    },
    {
      "epoch": 7.604598999023438e-06,
      "model_forward_time": 1.715193510055542,
      "step": 81974
    },
    {
      "epoch": 7.604598999023438e-06,
      "step": 81974,
      "training_step_time": 7.860309600830078
    },
    {
      "epoch": 7.605552673339843e-06,
      "model_forward_time": 1.7219641208648682,
      "step": 81975
    },
    {
      "epoch": 7.605552673339843e-06,
      "step": 81975,
      "training_step_time": 7.820422649383545
    },
    {
      "epoch": 7.60650634765625e-06,
      "model_forward_time": 2.397949457168579,
      "step": 81976
    },
    {
      "epoch": 7.60650634765625e-06,
      "step": 81976,
      "training_step_time": 8.48470687866211
    },
    {
      "epoch": 7.6074600219726565e-06,
      "model_forward_time": 1.7398934364318848,
      "step": 81977
    },
    {
      "epoch": 7.6074600219726565e-06,
      "step": 81977,
      "training_step_time": 7.948458433151245
    },
    {
      "epoch": 7.608413696289062e-06,
      "model_forward_time": 1.757277488708496,
      "step": 81978
    },
    {
      "epoch": 7.608413696289062e-06,
      "step": 81978,
      "training_step_time": 8.488414287567139
    },
    {
      "epoch": 7.609367370605469e-06,
      "model_forward_time": 1.734726905822754,
      "step": 81979
    },
    {
      "epoch": 7.609367370605469e-06,
      "step": 81979,
      "training_step_time": 8.391347885131836
    },
    {
      "epoch": 7.610321044921875e-06,
      "grad_norm": 0.127407044172287,
      "learning_rate": 8.619038058045926e-07,
      "loss": 0.054,
      "step": 81980
    },
    {
      "epoch": 7.610321044921875e-06,
      "model_forward_time": 1.7299466133117676,
      "step": 81980
    },
    {
      "dynamics_loss_avg": 0.056092833168804646,
      "epoch": 7.610321044921875e-06,
      "step": 81980
    },
    {
      "action_loss_avg": 0.0015021769839222543,
      "epoch": 7.610321044921875e-06,
      "step": 81980
    },
    {
      "epoch": 7.610321044921875e-06,
      "step": 81980,
      "training_step_time": 7.849000930786133
    },
    {
      "epoch": 7.611274719238281e-06,
      "model_forward_time": 1.7365424633026123,
      "step": 81981
    },
    {
      "epoch": 7.611274719238281e-06,
      "step": 81981,
      "training_step_time": 8.553625583648682
    },
    {
      "epoch": 7.612228393554688e-06,
      "model_forward_time": 1.7378077507019043,
      "step": 81982
    },
    {
      "epoch": 7.612228393554688e-06,
      "step": 81982,
      "training_step_time": 7.893599271774292
    },
    {
      "epoch": 7.613182067871093e-06,
      "model_forward_time": 1.3594799041748047,
      "step": 81983
    },
    {
      "epoch": 7.613182067871093e-06,
      "step": 81983,
      "training_step_time": 7.796706676483154
    },
    {
      "epoch": 7.6141357421875e-06,
      "model_forward_time": 1.738152265548706,
      "step": 81984
    },
    {
      "epoch": 7.6141357421875e-06,
      "step": 81984,
      "training_step_time": 8.377290487289429
    },
    {
      "epoch": 7.6150894165039066e-06,
      "model_forward_time": 1.7319509983062744,
      "step": 81985
    },
    {
      "epoch": 7.6150894165039066e-06,
      "step": 81985,
      "training_step_time": 7.879258632659912
    },
    {
      "epoch": 7.616043090820312e-06,
      "model_forward_time": 2.363212823867798,
      "step": 81986
    },
    {
      "epoch": 7.616043090820312e-06,
      "step": 81986,
      "training_step_time": 7.896002769470215
    },
    {
      "epoch": 7.616996765136719e-06,
      "model_forward_time": 1.742781639099121,
      "step": 81987
    },
    {
      "epoch": 7.616996765136719e-06,
      "step": 81987,
      "training_step_time": 7.925853967666626
    },
    {
      "epoch": 7.617950439453125e-06,
      "model_forward_time": 1.761030912399292,
      "step": 81988
    },
    {
      "epoch": 7.617950439453125e-06,
      "step": 81988,
      "training_step_time": 8.653510332107544
    },
    {
      "epoch": 7.618904113769531e-06,
      "model_forward_time": 1.7406280040740967,
      "step": 81989
    },
    {
      "epoch": 7.618904113769531e-06,
      "step": 81989,
      "training_step_time": 9.893565654754639
    },
    {
      "epoch": 7.619857788085938e-06,
      "grad_norm": 0.09524863958358765,
      "learning_rate": 8.609759561165704e-07,
      "loss": 0.0562,
      "step": 81990
    },
    {
      "epoch": 7.619857788085938e-06,
      "model_forward_time": 1.7359230518341064,
      "step": 81990
    },
    {
      "dynamics_loss_avg": 0.04472884014248848,
      "epoch": 7.619857788085938e-06,
      "step": 81990
    },
    {
      "action_loss_avg": 0.0012957488041138276,
      "epoch": 7.619857788085938e-06,
      "step": 81990
    },
    {
      "epoch": 7.619857788085938e-06,
      "step": 81990,
      "training_step_time": 8.546120405197144
    },
    {
      "epoch": 7.6208114624023434e-06,
      "model_forward_time": 1.7345092296600342,
      "step": 81991
    },
    {
      "epoch": 7.6208114624023434e-06,
      "step": 81991,
      "training_step_time": 8.71613073348999
    },
    {
      "epoch": 7.62176513671875e-06,
      "model_forward_time": 1.7393395900726318,
      "step": 81992
    },
    {
      "epoch": 7.62176513671875e-06,
      "step": 81992,
      "training_step_time": 7.623512268066406
    },
    {
      "epoch": 7.6227188110351566e-06,
      "model_forward_time": 1.7423977851867676,
      "step": 81993
    },
    {
      "epoch": 7.6227188110351566e-06,
      "step": 81993,
      "training_step_time": 9.285841941833496
    },
    {
      "epoch": 7.623672485351562e-06,
      "model_forward_time": 1.7294819355010986,
      "step": 81994
    },
    {
      "epoch": 7.623672485351562e-06,
      "step": 81994,
      "training_step_time": 8.752845764160156
    },
    {
      "epoch": 7.624626159667969e-06,
      "model_forward_time": 2.3447506427764893,
      "step": 81995
    },
    {
      "epoch": 7.624626159667969e-06,
      "step": 81995,
      "training_step_time": 7.924180746078491
    },
    {
      "epoch": 7.625579833984375e-06,
      "model_forward_time": 1.0264716148376465,
      "step": 81996
    },
    {
      "epoch": 7.625579833984375e-06,
      "step": 81996,
      "training_step_time": 8.399769067764282
    },
    {
      "epoch": 7.626533508300781e-06,
      "model_forward_time": 1.8623552322387695,
      "step": 81997
    },
    {
      "epoch": 7.626533508300781e-06,
      "step": 81997,
      "training_step_time": 7.888634204864502
    },
    {
      "epoch": 7.627487182617188e-06,
      "model_forward_time": 1.7499785423278809,
      "step": 81998
    },
    {
      "epoch": 7.627487182617188e-06,
      "step": 81998,
      "training_step_time": 8.143073797225952
    },
    {
      "epoch": 7.6284408569335934e-06,
      "model_forward_time": 1.7450966835021973,
      "step": 81999
    },
    {
      "epoch": 7.6284408569335934e-06,
      "step": 81999,
      "training_step_time": 7.7859110832214355
    },
    {
      "epoch": 7.62939453125e-06,
      "grad_norm": 0.09469035267829895,
      "learning_rate": 8.600485590659935e-07,
      "loss": 0.0567,
      "step": 82000
    },
    {
      "epoch": 7.62939453125e-06,
      "model_forward_time": 1.77785325050354,
      "step": 82000
    },
    {
      "dynamics_loss_avg": 0.08887077225372195,
      "epoch": 7.62939453125e-06,
      "step": 82000
    },
    {
      "action_loss_avg": 0.001924996988964267,
      "epoch": 7.62939453125e-06,
      "step": 82000
    },
    {
      "epoch": 7.62939453125e-06,
      "step": 82000,
      "training_step_time": 6.710098505020142
    },
    {
      "epoch": 7.630348205566406e-06,
      "model_forward_time": 1.7882447242736816,
      "step": 82001
    },
    {
      "epoch": 7.630348205566406e-06,
      "step": 82001,
      "training_step_time": 6.818922281265259
    },
    {
      "epoch": 7.631301879882813e-06,
      "model_forward_time": 1.801917552947998,
      "step": 82002
    },
    {
      "epoch": 7.631301879882813e-06,
      "step": 82002,
      "training_step_time": 6.924266815185547
    },
    {
      "epoch": 7.632255554199219e-06,
      "model_forward_time": 1.848440408706665,
      "step": 82003
    },
    {
      "epoch": 7.632255554199219e-06,
      "step": 82003,
      "training_step_time": 6.904350757598877
    },
    {
      "epoch": 7.633209228515625e-06,
      "model_forward_time": 1.7998342514038086,
      "step": 82004
    },
    {
      "epoch": 7.633209228515625e-06,
      "step": 82004,
      "training_step_time": 6.8231894969940186
    },
    {
      "epoch": 7.634162902832032e-06,
      "model_forward_time": 1.789771318435669,
      "step": 82005
    },
    {
      "epoch": 7.634162902832032e-06,
      "step": 82005,
      "training_step_time": 6.886790990829468
    },
    {
      "epoch": 7.635116577148438e-06,
      "model_forward_time": 1.7916300296783447,
      "step": 82006
    },
    {
      "epoch": 7.635116577148438e-06,
      "step": 82006,
      "training_step_time": 6.855855703353882
    },
    {
      "epoch": 7.636070251464843e-06,
      "model_forward_time": 1.7636940479278564,
      "step": 82007
    },
    {
      "epoch": 7.636070251464843e-06,
      "step": 82007,
      "training_step_time": 6.868210554122925
    },
    {
      "epoch": 7.63702392578125e-06,
      "model_forward_time": 1.80051589012146,
      "step": 82008
    },
    {
      "epoch": 7.63702392578125e-06,
      "step": 82008,
      "training_step_time": 8.03864049911499
    },
    {
      "epoch": 7.637977600097657e-06,
      "model_forward_time": 3.551225185394287,
      "step": 82009
    },
    {
      "epoch": 7.637977600097657e-06,
      "step": 82009,
      "training_step_time": 8.536729574203491
    },
    {
      "epoch": 7.638931274414062e-06,
      "grad_norm": 0.10779749602079391,
      "learning_rate": 8.591216147542814e-07,
      "loss": 0.0553,
      "step": 82010
    },
    {
      "epoch": 7.638931274414062e-06,
      "model_forward_time": 1.7605888843536377,
      "step": 82010
    },
    {
      "dynamics_loss_avg": 0.055788512155413625,
      "epoch": 7.638931274414062e-06,
      "step": 82010
    },
    {
      "action_loss_avg": 0.0017906344204675407,
      "epoch": 7.638931274414062e-06,
      "step": 82010
    },
    {
      "epoch": 7.638931274414062e-06,
      "step": 82010,
      "training_step_time": 6.9353554248809814
    },
    {
      "epoch": 7.639884948730468e-06,
      "model_forward_time": 1.7692716121673584,
      "step": 82011
    },
    {
      "epoch": 7.639884948730468e-06,
      "step": 82011,
      "training_step_time": 6.848424673080444
    },
    {
      "epoch": 7.640838623046875e-06,
      "model_forward_time": 1.7533900737762451,
      "step": 82012
    },
    {
      "epoch": 7.640838623046875e-06,
      "step": 82012,
      "training_step_time": 6.912410497665405
    },
    {
      "epoch": 7.641792297363281e-06,
      "model_forward_time": 1.7687203884124756,
      "step": 82013
    },
    {
      "epoch": 7.641792297363281e-06,
      "step": 82013,
      "training_step_time": 6.89191746711731
    },
    {
      "epoch": 7.642745971679687e-06,
      "model_forward_time": 1.770447015762329,
      "step": 82014
    },
    {
      "epoch": 7.642745971679687e-06,
      "step": 82014,
      "training_step_time": 6.885114908218384
    },
    {
      "epoch": 7.643699645996094e-06,
      "model_forward_time": 1.778993844985962,
      "step": 82015
    },
    {
      "epoch": 7.643699645996094e-06,
      "step": 82015,
      "training_step_time": 6.9028565883636475
    },
    {
      "epoch": 7.6446533203125e-06,
      "model_forward_time": 1.7625164985656738,
      "step": 82016
    },
    {
      "epoch": 7.6446533203125e-06,
      "step": 82016,
      "training_step_time": 6.849972486495972
    },
    {
      "epoch": 7.645606994628906e-06,
      "model_forward_time": 1.7643029689788818,
      "step": 82017
    },
    {
      "epoch": 7.645606994628906e-06,
      "step": 82017,
      "training_step_time": 6.800101280212402
    },
    {
      "epoch": 7.646560668945313e-06,
      "model_forward_time": 2.8947646617889404,
      "step": 82018
    },
    {
      "epoch": 7.646560668945313e-06,
      "step": 82018,
      "training_step_time": 7.978631496429443
    },
    {
      "epoch": 7.647514343261719e-06,
      "model_forward_time": 1.7776744365692139,
      "step": 82019
    },
    {
      "epoch": 7.647514343261719e-06,
      "step": 82019,
      "training_step_time": 7.719873905181885
    },
    {
      "epoch": 7.648468017578125e-06,
      "grad_norm": 0.08224993944168091,
      "learning_rate": 8.581951232828045e-07,
      "loss": 0.0548,
      "step": 82020
    },
    {
      "epoch": 7.648468017578125e-06,
      "model_forward_time": 1.7779595851898193,
      "step": 82020
    },
    {
      "dynamics_loss_avg": 0.05003878828138113,
      "epoch": 7.648468017578125e-06,
      "step": 82020
    },
    {
      "action_loss_avg": 0.0012044495844747872,
      "epoch": 7.648468017578125e-06,
      "step": 82020
    },
    {
      "epoch": 7.648468017578125e-06,
      "step": 82020,
      "training_step_time": 6.857011079788208
    },
    {
      "epoch": 7.649421691894532e-06,
      "model_forward_time": 1.7902522087097168,
      "step": 82021
    },
    {
      "epoch": 7.649421691894532e-06,
      "step": 82021,
      "training_step_time": 6.833188533782959
    },
    {
      "epoch": 7.650375366210938e-06,
      "model_forward_time": 1.770289421081543,
      "step": 82022
    },
    {
      "epoch": 7.650375366210938e-06,
      "step": 82022,
      "training_step_time": 6.843522310256958
    },
    {
      "epoch": 7.651329040527343e-06,
      "model_forward_time": 1.7684681415557861,
      "step": 82023
    },
    {
      "epoch": 7.651329040527343e-06,
      "step": 82023,
      "training_step_time": 6.893083810806274
    },
    {
      "epoch": 7.65228271484375e-06,
      "model_forward_time": 1.78416109085083,
      "step": 82024
    },
    {
      "epoch": 7.65228271484375e-06,
      "step": 82024,
      "training_step_time": 6.833481788635254
    },
    {
      "epoch": 7.653236389160157e-06,
      "model_forward_time": 1.7703194618225098,
      "step": 82025
    },
    {
      "epoch": 7.653236389160157e-06,
      "step": 82025,
      "training_step_time": 6.864378452301025
    },
    {
      "epoch": 7.654190063476562e-06,
      "model_forward_time": 1.7693195343017578,
      "step": 82026
    },
    {
      "epoch": 7.654190063476562e-06,
      "step": 82026,
      "training_step_time": 6.930033445358276
    },
    {
      "epoch": 7.655143737792968e-06,
      "model_forward_time": 2.9209935665130615,
      "step": 82027
    },
    {
      "epoch": 7.655143737792968e-06,
      "step": 82027,
      "training_step_time": 7.930736064910889
    },
    {
      "epoch": 7.656097412109375e-06,
      "model_forward_time": 1.781874656677246,
      "step": 82028
    },
    {
      "epoch": 7.656097412109375e-06,
      "step": 82028,
      "training_step_time": 7.845722913742065
    },
    {
      "epoch": 7.657051086425781e-06,
      "model_forward_time": 1.7927277088165283,
      "step": 82029
    },
    {
      "epoch": 7.657051086425781e-06,
      "step": 82029,
      "training_step_time": 6.869781732559204
    },
    {
      "epoch": 7.658004760742187e-06,
      "grad_norm": 0.10565677285194397,
      "learning_rate": 8.572690847528825e-07,
      "loss": 0.0551,
      "step": 82030
    },
    {
      "epoch": 7.658004760742187e-06,
      "model_forward_time": 1.7594797611236572,
      "step": 82030
    },
    {
      "dynamics_loss_avg": 0.060355335101485254,
      "epoch": 7.658004760742187e-06,
      "step": 82030
    },
    {
      "action_loss_avg": 0.0016391197219491004,
      "epoch": 7.658004760742187e-06,
      "step": 82030
    },
    {
      "epoch": 7.658004760742187e-06,
      "step": 82030,
      "training_step_time": 6.905679702758789
    },
    {
      "epoch": 7.658958435058594e-06,
      "model_forward_time": 1.7847790718078613,
      "step": 82031
    },
    {
      "epoch": 7.658958435058594e-06,
      "step": 82031,
      "training_step_time": 6.82212233543396
    },
    {
      "epoch": 7.659912109375e-06,
      "model_forward_time": 1.7774596214294434,
      "step": 82032
    },
    {
      "epoch": 7.659912109375e-06,
      "step": 82032,
      "training_step_time": 6.8342413902282715
    },
    {
      "epoch": 7.660865783691406e-06,
      "model_forward_time": 1.7752618789672852,
      "step": 82033
    },
    {
      "epoch": 7.660865783691406e-06,
      "step": 82033,
      "training_step_time": 6.884826898574829
    },
    {
      "epoch": 7.661819458007813e-06,
      "model_forward_time": 1.7613325119018555,
      "step": 82034
    },
    {
      "epoch": 7.661819458007813e-06,
      "step": 82034,
      "training_step_time": 6.814084053039551
    },
    {
      "epoch": 7.662773132324219e-06,
      "model_forward_time": 1.7678847312927246,
      "step": 82035
    },
    {
      "epoch": 7.662773132324219e-06,
      "step": 82035,
      "training_step_time": 6.8116233348846436
    },
    {
      "epoch": 7.663726806640625e-06,
      "model_forward_time": 2.920008659362793,
      "step": 82036
    },
    {
      "epoch": 7.663726806640625e-06,
      "step": 82036,
      "training_step_time": 9.955402135848999
    },
    {
      "epoch": 7.664680480957032e-06,
      "model_forward_time": 1.761582374572754,
      "step": 82037
    },
    {
      "epoch": 7.664680480957032e-06,
      "step": 82037,
      "training_step_time": 7.870459794998169
    },
    {
      "epoch": 7.665634155273438e-06,
      "model_forward_time": 1.7656211853027344,
      "step": 82038
    },
    {
      "epoch": 7.665634155273438e-06,
      "step": 82038,
      "training_step_time": 6.88094162940979
    },
    {
      "epoch": 7.666587829589843e-06,
      "model_forward_time": 1.7530128955841064,
      "step": 82039
    },
    {
      "epoch": 7.666587829589843e-06,
      "step": 82039,
      "training_step_time": 6.849518775939941
    },
    {
      "epoch": 7.66754150390625e-06,
      "grad_norm": 0.0947437658905983,
      "learning_rate": 8.563434992657838e-07,
      "loss": 0.0558,
      "step": 82040
    },
    {
      "epoch": 7.66754150390625e-06,
      "model_forward_time": 1.778496503829956,
      "step": 82040
    },
    {
      "dynamics_loss_avg": 0.059039504081010816,
      "epoch": 7.66754150390625e-06,
      "step": 82040
    },
    {
      "action_loss_avg": 0.002108006563503295,
      "epoch": 7.66754150390625e-06,
      "step": 82040
    },
    {
      "epoch": 7.66754150390625e-06,
      "step": 82040,
      "training_step_time": 6.896881341934204
    },
    {
      "epoch": 7.668495178222657e-06,
      "model_forward_time": 1.7492330074310303,
      "step": 82041
    },
    {
      "epoch": 7.668495178222657e-06,
      "step": 82041,
      "training_step_time": 6.764262914657593
    },
    {
      "epoch": 7.669448852539062e-06,
      "model_forward_time": 1.7707221508026123,
      "step": 82042
    },
    {
      "epoch": 7.669448852539062e-06,
      "step": 82042,
      "training_step_time": 6.798036336898804
    },
    {
      "epoch": 7.670402526855468e-06,
      "model_forward_time": 1.4021732807159424,
      "step": 82043
    },
    {
      "epoch": 7.670402526855468e-06,
      "step": 82043,
      "training_step_time": 6.817798137664795
    },
    {
      "epoch": 7.671356201171875e-06,
      "model_forward_time": 1.761566162109375,
      "step": 82044
    },
    {
      "epoch": 7.671356201171875e-06,
      "step": 82044,
      "training_step_time": 6.846439838409424
    },
    {
      "epoch": 7.672309875488281e-06,
      "model_forward_time": 1.7563791275024414,
      "step": 82045
    },
    {
      "epoch": 7.672309875488281e-06,
      "step": 82045,
      "training_step_time": 9.553564071655273
    },
    {
      "epoch": 7.673263549804687e-06,
      "model_forward_time": 1.8861641883850098,
      "step": 82046
    },
    {
      "epoch": 7.673263549804687e-06,
      "step": 82046,
      "training_step_time": 7.795257568359375
    },
    {
      "epoch": 7.674217224121094e-06,
      "model_forward_time": 1.767005443572998,
      "step": 82047
    },
    {
      "epoch": 7.674217224121094e-06,
      "step": 82047,
      "training_step_time": 6.764845371246338
    },
    {
      "epoch": 7.6751708984375e-06,
      "model_forward_time": 1.7805490493774414,
      "step": 82048
    },
    {
      "epoch": 7.6751708984375e-06,
      "step": 82048,
      "training_step_time": 6.833520889282227
    },
    {
      "epoch": 7.676124572753906e-06,
      "model_forward_time": 1.7732274532318115,
      "step": 82049
    },
    {
      "epoch": 7.676124572753906e-06,
      "step": 82049,
      "training_step_time": 6.8158814907073975
    },
    {
      "epoch": 7.677078247070313e-06,
      "grad_norm": 0.08950895816087723,
      "learning_rate": 8.554183669227306e-07,
      "loss": 0.0548,
      "step": 82050
    },
    {
      "epoch": 7.677078247070313e-06,
      "model_forward_time": 1.769216775894165,
      "step": 82050
    },
    {
      "dynamics_loss_avg": 0.056434928253293035,
      "epoch": 7.677078247070313e-06,
      "step": 82050
    },
    {
      "action_loss_avg": 0.0010700393584556877,
      "epoch": 7.677078247070313e-06,
      "step": 82050
    },
    {
      "epoch": 7.677078247070313e-06,
      "step": 82050,
      "training_step_time": 6.8396220207214355
    },
    {
      "epoch": 7.678031921386719e-06,
      "model_forward_time": 1.7897729873657227,
      "step": 82051
    },
    {
      "epoch": 7.678031921386719e-06,
      "step": 82051,
      "training_step_time": 6.816037654876709
    },
    {
      "epoch": 7.678985595703125e-06,
      "model_forward_time": 1.766319990158081,
      "step": 82052
    },
    {
      "epoch": 7.678985595703125e-06,
      "step": 82052,
      "training_step_time": 6.813997983932495
    },
    {
      "epoch": 7.679939270019532e-06,
      "model_forward_time": 1.7885873317718506,
      "step": 82053
    },
    {
      "epoch": 7.679939270019532e-06,
      "step": 82053,
      "training_step_time": 6.793043375015259
    },
    {
      "epoch": 7.680892944335938e-06,
      "model_forward_time": 1.7473926544189453,
      "step": 82054
    },
    {
      "epoch": 7.680892944335938e-06,
      "step": 82054,
      "training_step_time": 8.427901268005371
    },
    {
      "epoch": 7.681846618652343e-06,
      "model_forward_time": 2.864959955215454,
      "step": 82055
    },
    {
      "epoch": 7.681846618652343e-06,
      "step": 82055,
      "training_step_time": 7.928863525390625
    },
    {
      "epoch": 7.68280029296875e-06,
      "model_forward_time": 1.7789554595947266,
      "step": 82056
    },
    {
      "epoch": 7.68280029296875e-06,
      "step": 82056,
      "training_step_time": 6.836657762527466
    },
    {
      "epoch": 7.683753967285157e-06,
      "model_forward_time": 1.7557392120361328,
      "step": 82057
    },
    {
      "epoch": 7.683753967285157e-06,
      "step": 82057,
      "training_step_time": 6.8251259326934814
    },
    {
      "epoch": 7.684707641601562e-06,
      "model_forward_time": 1.789705514907837,
      "step": 82058
    },
    {
      "epoch": 7.684707641601562e-06,
      "step": 82058,
      "training_step_time": 6.79155421257019
    },
    {
      "epoch": 7.685661315917968e-06,
      "model_forward_time": 1.8041150569915771,
      "step": 82059
    },
    {
      "epoch": 7.685661315917968e-06,
      "step": 82059,
      "training_step_time": 6.826581001281738
    },
    {
      "epoch": 7.686614990234375e-06,
      "grad_norm": 0.1388651579618454,
      "learning_rate": 8.544936878248938e-07,
      "loss": 0.0547,
      "step": 82060
    },
    {
      "epoch": 7.686614990234375e-06,
      "model_forward_time": 1.8067083358764648,
      "step": 82060
    },
    {
      "dynamics_loss_avg": 0.06204805988818407,
      "epoch": 7.686614990234375e-06,
      "step": 82060
    },
    {
      "action_loss_avg": 0.0015433858236065135,
      "epoch": 7.686614990234375e-06,
      "step": 82060
    },
    {
      "epoch": 7.686614990234375e-06,
      "step": 82060,
      "training_step_time": 6.8129308223724365
    },
    {
      "epoch": 7.687568664550781e-06,
      "model_forward_time": 1.799638271331787,
      "step": 82061
    },
    {
      "epoch": 7.687568664550781e-06,
      "step": 82061,
      "training_step_time": 6.796051502227783
    },
    {
      "epoch": 7.688522338867187e-06,
      "model_forward_time": 1.797105073928833,
      "step": 82062
    },
    {
      "epoch": 7.688522338867187e-06,
      "step": 82062,
      "training_step_time": 6.861987829208374
    },
    {
      "epoch": 7.689476013183594e-06,
      "model_forward_time": 1.7761540412902832,
      "step": 82063
    },
    {
      "epoch": 7.689476013183594e-06,
      "step": 82063,
      "training_step_time": 7.837677240371704
    },
    {
      "epoch": 7.6904296875e-06,
      "model_forward_time": 2.9139187335968018,
      "step": 82064
    },
    {
      "epoch": 7.6904296875e-06,
      "step": 82064,
      "training_step_time": 8.575698852539062
    },
    {
      "epoch": 7.691383361816406e-06,
      "model_forward_time": 1.7923147678375244,
      "step": 82065
    },
    {
      "epoch": 7.691383361816406e-06,
      "step": 82065,
      "training_step_time": 7.824995517730713
    },
    {
      "epoch": 7.692337036132813e-06,
      "model_forward_time": 1.7940678596496582,
      "step": 82066
    },
    {
      "epoch": 7.692337036132813e-06,
      "step": 82066,
      "training_step_time": 6.864283561706543
    },
    {
      "epoch": 7.693290710449219e-06,
      "model_forward_time": 1.0790648460388184,
      "step": 82067
    },
    {
      "epoch": 7.693290710449219e-06,
      "step": 82067,
      "training_step_time": 6.829365968704224
    },
    {
      "epoch": 7.694244384765625e-06,
      "model_forward_time": 1.7771599292755127,
      "step": 82068
    },
    {
      "epoch": 7.694244384765625e-06,
      "step": 82068,
      "training_step_time": 6.824303388595581
    },
    {
      "epoch": 7.695198059082032e-06,
      "model_forward_time": 1.7972495555877686,
      "step": 82069
    },
    {
      "epoch": 7.695198059082032e-06,
      "step": 82069,
      "training_step_time": 6.839568853378296
    },
    {
      "epoch": 7.696151733398438e-06,
      "grad_norm": 0.10719548165798187,
      "learning_rate": 8.535694620733942e-07,
      "loss": 0.0545,
      "step": 82070
    },
    {
      "epoch": 7.696151733398438e-06,
      "model_forward_time": 1.7994136810302734,
      "step": 82070
    },
    {
      "dynamics_loss_avg": 0.06166211068630219,
      "epoch": 7.696151733398438e-06,
      "step": 82070
    },
    {
      "action_loss_avg": 0.0011622996651567518,
      "epoch": 7.696151733398438e-06,
      "step": 82070
    },
    {
      "epoch": 7.696151733398438e-06,
      "step": 82070,
      "training_step_time": 6.870496988296509
    },
    {
      "epoch": 7.697105407714843e-06,
      "model_forward_time": 1.7703981399536133,
      "step": 82071
    },
    {
      "epoch": 7.697105407714843e-06,
      "step": 82071,
      "training_step_time": 6.687745094299316
    },
    {
      "epoch": 7.69805908203125e-06,
      "model_forward_time": 1.7788801193237305,
      "step": 82072
    },
    {
      "epoch": 7.69805908203125e-06,
      "step": 82072,
      "training_step_time": 8.42695951461792
    },
    {
      "epoch": 7.699012756347657e-06,
      "model_forward_time": 2.8847930431365967,
      "step": 82073
    },
    {
      "epoch": 7.699012756347657e-06,
      "step": 82073,
      "training_step_time": 9.915885925292969
    },
    {
      "epoch": 7.699966430664062e-06,
      "model_forward_time": 1.796191930770874,
      "step": 82074
    },
    {
      "epoch": 7.699966430664062e-06,
      "step": 82074,
      "training_step_time": 8.392707109451294
    },
    {
      "epoch": 7.700920104980468e-06,
      "model_forward_time": 1.792872667312622,
      "step": 82075
    },
    {
      "epoch": 7.700920104980468e-06,
      "step": 82075,
      "training_step_time": 6.797602891921997
    },
    {
      "epoch": 7.701873779296875e-06,
      "model_forward_time": 1.7937648296356201,
      "step": 82076
    },
    {
      "epoch": 7.701873779296875e-06,
      "step": 82076,
      "training_step_time": 6.778872728347778
    },
    {
      "epoch": 7.702827453613281e-06,
      "model_forward_time": 1.8026885986328125,
      "step": 82077
    },
    {
      "epoch": 7.702827453613281e-06,
      "step": 82077,
      "training_step_time": 6.8306097984313965
    },
    {
      "epoch": 7.703781127929687e-06,
      "model_forward_time": 1.7954339981079102,
      "step": 82078
    },
    {
      "epoch": 7.703781127929687e-06,
      "step": 82078,
      "training_step_time": 6.785376071929932
    },
    {
      "epoch": 7.704734802246094e-06,
      "model_forward_time": 1.787172555923462,
      "step": 82079
    },
    {
      "epoch": 7.704734802246094e-06,
      "step": 82079,
      "training_step_time": 6.8442347049713135
    },
    {
      "epoch": 7.7056884765625e-06,
      "grad_norm": 0.09657876938581467,
      "learning_rate": 8.526456897693053e-07,
      "loss": 0.0542,
      "step": 82080
    },
    {
      "epoch": 7.7056884765625e-06,
      "model_forward_time": 1.771183729171753,
      "step": 82080
    },
    {
      "dynamics_loss_avg": 0.06097155120223761,
      "epoch": 7.7056884765625e-06,
      "step": 82080
    },
    {
      "action_loss_avg": 0.0017234461527550593,
      "epoch": 7.7056884765625e-06,
      "step": 82080
    },
    {
      "epoch": 7.7056884765625e-06,
      "step": 82080,
      "training_step_time": 6.861584186553955
    },
    {
      "epoch": 7.706642150878906e-06,
      "model_forward_time": 1.786470651626587,
      "step": 82081
    },
    {
      "epoch": 7.706642150878906e-06,
      "step": 82081,
      "training_step_time": 7.817662477493286
    },
    {
      "epoch": 7.707595825195313e-06,
      "model_forward_time": 2.877150774002075,
      "step": 82082
    },
    {
      "epoch": 7.707595825195313e-06,
      "step": 82082,
      "training_step_time": 10.105347394943237
    },
    {
      "epoch": 7.708549499511719e-06,
      "model_forward_time": 1.797738790512085,
      "step": 82083
    },
    {
      "epoch": 7.708549499511719e-06,
      "step": 82083,
      "training_step_time": 7.813903570175171
    },
    {
      "epoch": 7.709503173828125e-06,
      "model_forward_time": 1.7947726249694824,
      "step": 82084
    },
    {
      "epoch": 7.709503173828125e-06,
      "step": 82084,
      "training_step_time": 7.802747011184692
    },
    {
      "epoch": 7.710456848144532e-06,
      "model_forward_time": 1.7896559238433838,
      "step": 82085
    },
    {
      "epoch": 7.710456848144532e-06,
      "step": 82085,
      "training_step_time": 6.846913576126099
    },
    {
      "epoch": 7.711410522460938e-06,
      "model_forward_time": 1.7824797630310059,
      "step": 82086
    },
    {
      "epoch": 7.711410522460938e-06,
      "step": 82086,
      "training_step_time": 6.7828240394592285
    },
    {
      "epoch": 7.712364196777344e-06,
      "model_forward_time": 1.7872352600097656,
      "step": 82087
    },
    {
      "epoch": 7.712364196777344e-06,
      "step": 82087,
      "training_step_time": 6.817171335220337
    },
    {
      "epoch": 7.71331787109375e-06,
      "model_forward_time": 1.786529302597046,
      "step": 82088
    },
    {
      "epoch": 7.71331787109375e-06,
      "step": 82088,
      "training_step_time": 6.824324369430542
    },
    {
      "epoch": 7.714271545410157e-06,
      "model_forward_time": 1.7857944965362549,
      "step": 82089
    },
    {
      "epoch": 7.714271545410157e-06,
      "step": 82089,
      "training_step_time": 6.818470478057861
    },
    {
      "epoch": 7.715225219726562e-06,
      "grad_norm": 0.13122685253620148,
      "learning_rate": 8.517223710136458e-07,
      "loss": 0.0554,
      "step": 82090
    },
    {
      "epoch": 7.715225219726562e-06,
      "model_forward_time": 1.7967467308044434,
      "step": 82090
    },
    {
      "dynamics_loss_avg": 0.05628868732601404,
      "epoch": 7.715225219726562e-06,
      "step": 82090
    },
    {
      "action_loss_avg": 0.0015739798254799098,
      "epoch": 7.715225219726562e-06,
      "step": 82090
    },
    {
      "epoch": 7.715225219726562e-06,
      "step": 82090,
      "training_step_time": 7.891326665878296
    },
    {
      "epoch": 7.716178894042968e-06,
      "model_forward_time": 1.8201673030853271,
      "step": 82091
    },
    {
      "epoch": 7.716178894042968e-06,
      "step": 82091,
      "training_step_time": 11.176267385482788
    },
    {
      "epoch": 7.717132568359376e-06,
      "model_forward_time": 2.8846704959869385,
      "step": 82092
    },
    {
      "epoch": 7.717132568359376e-06,
      "step": 82092,
      "training_step_time": 8.693742990493774
    },
    {
      "epoch": 7.718086242675781e-06,
      "model_forward_time": 1.783130168914795,
      "step": 82093
    },
    {
      "epoch": 7.718086242675781e-06,
      "step": 82093,
      "training_step_time": 7.83250617980957
    },
    {
      "epoch": 7.719039916992187e-06,
      "model_forward_time": 1.7867951393127441,
      "step": 82094
    },
    {
      "epoch": 7.719039916992187e-06,
      "step": 82094,
      "training_step_time": 6.7887678146362305
    },
    {
      "epoch": 7.719993591308594e-06,
      "model_forward_time": 1.7952477931976318,
      "step": 82095
    },
    {
      "epoch": 7.719993591308594e-06,
      "step": 82095,
      "training_step_time": 6.825551509857178
    },
    {
      "epoch": 7.720947265625e-06,
      "model_forward_time": 1.7970190048217773,
      "step": 82096
    },
    {
      "epoch": 7.720947265625e-06,
      "step": 82096,
      "training_step_time": 6.759762287139893
    },
    {
      "epoch": 7.721900939941406e-06,
      "model_forward_time": 1.7945168018341064,
      "step": 82097
    },
    {
      "epoch": 7.721900939941406e-06,
      "step": 82097,
      "training_step_time": 6.768548965454102
    },
    {
      "epoch": 7.722854614257813e-06,
      "model_forward_time": 1.7872934341430664,
      "step": 82098
    },
    {
      "epoch": 7.722854614257813e-06,
      "step": 82098,
      "training_step_time": 6.836986303329468
    },
    {
      "epoch": 7.723808288574219e-06,
      "model_forward_time": 1.792396068572998,
      "step": 82099
    },
    {
      "epoch": 7.723808288574219e-06,
      "step": 82099,
      "training_step_time": 7.772263288497925
    },
    {
      "epoch": 7.724761962890625e-06,
      "grad_norm": 0.09970733523368835,
      "learning_rate": 8.507995059073931e-07,
      "loss": 0.054,
      "step": 82100
    },
    {
      "epoch": 7.724761962890625e-06,
      "model_forward_time": 1.7727725505828857,
      "step": 82100
    },
    {
      "dynamics_loss_avg": 0.05073163527995348,
      "epoch": 7.724761962890625e-06,
      "step": 82100
    },
    {
      "action_loss_avg": 0.0015174037049291655,
      "epoch": 7.724761962890625e-06,
      "step": 82100
    },
    {
      "epoch": 7.724761962890625e-06,
      "step": 82100,
      "training_step_time": 9.899094343185425
    },
    {
      "epoch": 7.725715637207032e-06,
      "model_forward_time": 2.9031012058258057,
      "step": 82101
    },
    {
      "epoch": 7.725715637207032e-06,
      "step": 82101,
      "training_step_time": 7.8754966259002686
    },
    {
      "epoch": 7.726669311523438e-06,
      "model_forward_time": 1.8026273250579834,
      "step": 82102
    },
    {
      "epoch": 7.726669311523438e-06,
      "step": 82102,
      "training_step_time": 7.771915912628174
    },
    {
      "epoch": 7.727622985839844e-06,
      "model_forward_time": 1.8023455142974854,
      "step": 82103
    },
    {
      "epoch": 7.727622985839844e-06,
      "step": 82103,
      "training_step_time": 6.877454996109009
    },
    {
      "epoch": 7.72857666015625e-06,
      "model_forward_time": 1.7745811939239502,
      "step": 82104
    },
    {
      "epoch": 7.72857666015625e-06,
      "step": 82104,
      "training_step_time": 6.773737192153931
    },
    {
      "epoch": 7.729530334472657e-06,
      "model_forward_time": 1.7874653339385986,
      "step": 82105
    },
    {
      "epoch": 7.729530334472657e-06,
      "step": 82105,
      "training_step_time": 6.858104705810547
    },
    {
      "epoch": 7.730484008789062e-06,
      "model_forward_time": 1.7863061428070068,
      "step": 82106
    },
    {
      "epoch": 7.730484008789062e-06,
      "step": 82106,
      "training_step_time": 6.870713949203491
    },
    {
      "epoch": 7.731437683105468e-06,
      "model_forward_time": 1.7916128635406494,
      "step": 82107
    },
    {
      "epoch": 7.731437683105468e-06,
      "step": 82107,
      "training_step_time": 6.836745023727417
    },
    {
      "epoch": 7.732391357421876e-06,
      "model_forward_time": 1.7845628261566162,
      "step": 82108
    },
    {
      "epoch": 7.732391357421876e-06,
      "step": 82108,
      "training_step_time": 7.820964574813843
    },
    {
      "epoch": 7.733345031738281e-06,
      "model_forward_time": 1.7872271537780762,
      "step": 82109
    },
    {
      "epoch": 7.733345031738281e-06,
      "step": 82109,
      "training_step_time": 10.387440204620361
    },
    {
      "epoch": 7.734298706054687e-06,
      "grad_norm": 0.18200375139713287,
      "learning_rate": 8.498770945514673e-07,
      "loss": 0.0553,
      "step": 82110
    },
    {
      "epoch": 7.734298706054687e-06,
      "model_forward_time": 2.9481091499328613,
      "step": 82110
    },
    {
      "dynamics_loss_avg": 0.052929897978901866,
      "epoch": 7.734298706054687e-06,
      "step": 82110
    },
    {
      "action_loss_avg": 0.001523427531355992,
      "epoch": 7.734298706054687e-06,
      "step": 82110
    },
    {
      "epoch": 7.734298706054687e-06,
      "step": 82110,
      "training_step_time": 10.408218383789062
    },
    {
      "epoch": 7.735252380371094e-06,
      "model_forward_time": 1.7875254154205322,
      "step": 82111
    },
    {
      "epoch": 7.735252380371094e-06,
      "step": 82111,
      "training_step_time": 7.830545425415039
    },
    {
      "epoch": 7.7362060546875e-06,
      "model_forward_time": 1.7907426357269287,
      "step": 82112
    },
    {
      "epoch": 7.7362060546875e-06,
      "step": 82112,
      "training_step_time": 6.770090341567993
    },
    {
      "epoch": 7.737159729003906e-06,
      "model_forward_time": 1.7989277839660645,
      "step": 82113
    },
    {
      "epoch": 7.737159729003906e-06,
      "step": 82113,
      "training_step_time": 6.833521604537964
    },
    {
      "epoch": 7.738113403320313e-06,
      "model_forward_time": 1.7886266708374023,
      "step": 82114
    },
    {
      "epoch": 7.738113403320313e-06,
      "step": 82114,
      "training_step_time": 6.79952597618103
    },
    {
      "epoch": 7.739067077636719e-06,
      "model_forward_time": 1.782663345336914,
      "step": 82115
    },
    {
      "epoch": 7.739067077636719e-06,
      "step": 82115,
      "training_step_time": 6.824226140975952
    },
    {
      "epoch": 7.740020751953125e-06,
      "model_forward_time": 1.7945201396942139,
      "step": 82116
    },
    {
      "epoch": 7.740020751953125e-06,
      "step": 82116,
      "training_step_time": 6.840325355529785
    },
    {
      "epoch": 7.740974426269532e-06,
      "model_forward_time": 1.7859704494476318,
      "step": 82117
    },
    {
      "epoch": 7.740974426269532e-06,
      "step": 82117,
      "training_step_time": 7.700669050216675
    },
    {
      "epoch": 7.741928100585938e-06,
      "model_forward_time": 1.7780051231384277,
      "step": 82118
    },
    {
      "epoch": 7.741928100585938e-06,
      "step": 82118,
      "training_step_time": 10.13782525062561
    },
    {
      "epoch": 7.742881774902344e-06,
      "model_forward_time": 2.910684823989868,
      "step": 82119
    },
    {
      "epoch": 7.742881774902344e-06,
      "step": 82119,
      "training_step_time": 8.381653308868408
    },
    {
      "epoch": 7.74383544921875e-06,
      "grad_norm": 0.10606105625629425,
      "learning_rate": 8.489551370467436e-07,
      "loss": 0.054,
      "step": 82120
    },
    {
      "epoch": 7.74383544921875e-06,
      "model_forward_time": 1.7724738121032715,
      "step": 82120
    },
    {
      "dynamics_loss_avg": 0.04424486942589283,
      "epoch": 7.74383544921875e-06,
      "step": 82120
    },
    {
      "action_loss_avg": 0.0015115996822714806,
      "epoch": 7.74383544921875e-06,
      "step": 82120
    },
    {
      "epoch": 7.74383544921875e-06,
      "step": 82120,
      "training_step_time": 7.766337633132935
    },
    {
      "epoch": 7.744789123535157e-06,
      "model_forward_time": 1.771327257156372,
      "step": 82121
    },
    {
      "epoch": 7.744789123535157e-06,
      "step": 82121,
      "training_step_time": 6.777426481246948
    },
    {
      "epoch": 7.745742797851562e-06,
      "model_forward_time": 1.789093255996704,
      "step": 82122
    },
    {
      "epoch": 7.745742797851562e-06,
      "step": 82122,
      "training_step_time": 6.835433483123779
    },
    {
      "epoch": 7.746696472167968e-06,
      "model_forward_time": 1.7879507541656494,
      "step": 82123
    },
    {
      "epoch": 7.746696472167968e-06,
      "step": 82123,
      "training_step_time": 6.840173721313477
    },
    {
      "epoch": 7.747650146484376e-06,
      "model_forward_time": 1.7901253700256348,
      "step": 82124
    },
    {
      "epoch": 7.747650146484376e-06,
      "step": 82124,
      "training_step_time": 6.898130655288696
    },
    {
      "epoch": 7.748603820800781e-06,
      "model_forward_time": 1.7832098007202148,
      "step": 82125
    },
    {
      "epoch": 7.748603820800781e-06,
      "step": 82125,
      "training_step_time": 6.810739040374756
    },
    {
      "epoch": 7.749557495117187e-06,
      "model_forward_time": 1.7946667671203613,
      "step": 82126
    },
    {
      "epoch": 7.749557495117187e-06,
      "step": 82126,
      "training_step_time": 7.66325569152832
    },
    {
      "epoch": 7.750511169433594e-06,
      "model_forward_time": 1.790177822113037,
      "step": 82127
    },
    {
      "epoch": 7.750511169433594e-06,
      "step": 82127,
      "training_step_time": 7.906722545623779
    },
    {
      "epoch": 7.75146484375e-06,
      "model_forward_time": 2.903703451156616,
      "step": 82128
    },
    {
      "epoch": 7.75146484375e-06,
      "step": 82128,
      "training_step_time": 9.531162977218628
    },
    {
      "epoch": 7.752418518066406e-06,
      "model_forward_time": 1.7939057350158691,
      "step": 82129
    },
    {
      "epoch": 7.752418518066406e-06,
      "step": 82129,
      "training_step_time": 8.49915337562561
    },
    {
      "epoch": 7.753372192382813e-06,
      "grad_norm": 0.08103719353675842,
      "learning_rate": 8.480336334940442e-07,
      "loss": 0.054,
      "step": 82130
    },
    {
      "epoch": 7.753372192382813e-06,
      "model_forward_time": 1.7844922542572021,
      "step": 82130
    },
    {
      "dynamics_loss_avg": 0.060093360021710396,
      "epoch": 7.753372192382813e-06,
      "step": 82130
    },
    {
      "action_loss_avg": 0.0010585102427285165,
      "epoch": 7.753372192382813e-06,
      "step": 82130
    },
    {
      "epoch": 7.753372192382813e-06,
      "step": 82130,
      "training_step_time": 7.547848463058472
    },
    {
      "epoch": 7.754325866699219e-06,
      "model_forward_time": 1.7849431037902832,
      "step": 82131
    },
    {
      "epoch": 7.754325866699219e-06,
      "step": 82131,
      "training_step_time": 6.861807346343994
    },
    {
      "epoch": 7.755279541015625e-06,
      "model_forward_time": 1.428271770477295,
      "step": 82132
    },
    {
      "epoch": 7.755279541015625e-06,
      "step": 82132,
      "training_step_time": 6.838990688323975
    },
    {
      "epoch": 7.75623321533203e-06,
      "model_forward_time": 1.7975289821624756,
      "step": 82133
    },
    {
      "epoch": 7.75623321533203e-06,
      "step": 82133,
      "training_step_time": 6.851542949676514
    },
    {
      "epoch": 7.757186889648438e-06,
      "model_forward_time": 1.819256067276001,
      "step": 82134
    },
    {
      "epoch": 7.757186889648438e-06,
      "step": 82134,
      "training_step_time": 6.820613622665405
    },
    {
      "epoch": 7.758140563964844e-06,
      "model_forward_time": 1.78767991065979,
      "step": 82135
    },
    {
      "epoch": 7.758140563964844e-06,
      "step": 82135,
      "training_step_time": 7.609410047531128
    },
    {
      "epoch": 7.75909423828125e-06,
      "model_forward_time": 1.7861926555633545,
      "step": 82136
    },
    {
      "epoch": 7.75909423828125e-06,
      "step": 82136,
      "training_step_time": 9.376818895339966
    },
    {
      "epoch": 7.760047912597657e-06,
      "model_forward_time": 1.8212971687316895,
      "step": 82137
    },
    {
      "epoch": 7.760047912597657e-06,
      "step": 82137,
      "training_step_time": 11.25278902053833
    },
    {
      "epoch": 7.761001586914062e-06,
      "model_forward_time": 2.8860349655151367,
      "step": 82138
    },
    {
      "epoch": 7.761001586914062e-06,
      "step": 82138,
      "training_step_time": 8.466286420822144
    },
    {
      "epoch": 7.761955261230468e-06,
      "model_forward_time": 1.7887940406799316,
      "step": 82139
    },
    {
      "epoch": 7.761955261230468e-06,
      "step": 82139,
      "training_step_time": 7.734609842300415
    },
    {
      "epoch": 7.762908935546876e-06,
      "grad_norm": 0.08299766480922699,
      "learning_rate": 8.471125839941458e-07,
      "loss": 0.0546,
      "step": 82140
    },
    {
      "epoch": 7.762908935546876e-06,
      "model_forward_time": 1.7719001770019531,
      "step": 82140
    },
    {
      "dynamics_loss_avg": 0.05986880790442228,
      "epoch": 7.762908935546876e-06,
      "step": 82140
    },
    {
      "action_loss_avg": 0.002312472739140503,
      "epoch": 7.762908935546876e-06,
      "step": 82140
    },
    {
      "epoch": 7.762908935546876e-06,
      "step": 82140,
      "training_step_time": 6.823620080947876
    },
    {
      "epoch": 7.763862609863281e-06,
      "model_forward_time": 1.7964427471160889,
      "step": 82141
    },
    {
      "epoch": 7.763862609863281e-06,
      "step": 82141,
      "training_step_time": 6.796503782272339
    },
    {
      "epoch": 7.764816284179687e-06,
      "model_forward_time": 1.8023345470428467,
      "step": 82142
    },
    {
      "epoch": 7.764816284179687e-06,
      "step": 82142,
      "training_step_time": 6.792367935180664
    },
    {
      "epoch": 7.765769958496094e-06,
      "model_forward_time": 1.7800168991088867,
      "step": 82143
    },
    {
      "epoch": 7.765769958496094e-06,
      "step": 82143,
      "training_step_time": 6.751040458679199
    },
    {
      "epoch": 7.7667236328125e-06,
      "model_forward_time": 1.7895228862762451,
      "step": 82144
    },
    {
      "epoch": 7.7667236328125e-06,
      "step": 82144,
      "training_step_time": 7.572760105133057
    },
    {
      "epoch": 7.767677307128906e-06,
      "model_forward_time": 1.7884769439697266,
      "step": 82145
    },
    {
      "epoch": 7.767677307128906e-06,
      "step": 82145,
      "training_step_time": 8.789868593215942
    },
    {
      "epoch": 7.768630981445313e-06,
      "model_forward_time": 1.7987651824951172,
      "step": 82146
    },
    {
      "epoch": 7.768630981445313e-06,
      "step": 82146,
      "training_step_time": 9.41650676727295
    },
    {
      "epoch": 7.769584655761719e-06,
      "model_forward_time": 2.912752151489258,
      "step": 82147
    },
    {
      "epoch": 7.769584655761719e-06,
      "step": 82147,
      "training_step_time": 7.822990655899048
    },
    {
      "epoch": 7.770538330078125e-06,
      "model_forward_time": 1.7863185405731201,
      "step": 82148
    },
    {
      "epoch": 7.770538330078125e-06,
      "step": 82148,
      "training_step_time": 7.7273619174957275
    },
    {
      "epoch": 7.77149200439453e-06,
      "model_forward_time": 1.7934010028839111,
      "step": 82149
    },
    {
      "epoch": 7.77149200439453e-06,
      "step": 82149,
      "training_step_time": 7.810796022415161
    },
    {
      "epoch": 7.772445678710938e-06,
      "grad_norm": 0.08478790521621704,
      "learning_rate": 8.461919886477682e-07,
      "loss": 0.0537,
      "step": 82150
    },
    {
      "epoch": 7.772445678710938e-06,
      "model_forward_time": 1.7816479206085205,
      "step": 82150
    },
    {
      "dynamics_loss_avg": 0.05014097578823566,
      "epoch": 7.772445678710938e-06,
      "step": 82150
    },
    {
      "action_loss_avg": 0.0017140741343609988,
      "epoch": 7.772445678710938e-06,
      "step": 82150
    },
    {
      "epoch": 7.772445678710938e-06,
      "step": 82150,
      "training_step_time": 6.823888778686523
    },
    {
      "epoch": 7.773399353027344e-06,
      "model_forward_time": 1.7892422676086426,
      "step": 82151
    },
    {
      "epoch": 7.773399353027344e-06,
      "step": 82151,
      "training_step_time": 6.786308288574219
    },
    {
      "epoch": 7.77435302734375e-06,
      "model_forward_time": 1.7912249565124512,
      "step": 82152
    },
    {
      "epoch": 7.77435302734375e-06,
      "step": 82152,
      "training_step_time": 6.821895360946655
    },
    {
      "epoch": 7.775306701660157e-06,
      "model_forward_time": 1.801194667816162,
      "step": 82153
    },
    {
      "epoch": 7.775306701660157e-06,
      "step": 82153,
      "training_step_time": 7.639990568161011
    },
    {
      "epoch": 7.776260375976562e-06,
      "model_forward_time": 1.7993602752685547,
      "step": 82154
    },
    {
      "epoch": 7.776260375976562e-06,
      "step": 82154,
      "training_step_time": 7.773394584655762
    },
    {
      "epoch": 7.777214050292968e-06,
      "model_forward_time": 1.7756762504577637,
      "step": 82155
    },
    {
      "epoch": 7.777214050292968e-06,
      "step": 82155,
      "training_step_time": 11.50084400177002
    },
    {
      "epoch": 7.778167724609376e-06,
      "model_forward_time": 2.8967196941375732,
      "step": 82156
    },
    {
      "epoch": 7.778167724609376e-06,
      "step": 82156,
      "training_step_time": 9.522975206375122
    },
    {
      "epoch": 7.779121398925781e-06,
      "model_forward_time": 1.8005313873291016,
      "step": 82157
    },
    {
      "epoch": 7.779121398925781e-06,
      "step": 82157,
      "training_step_time": 8.644541263580322
    },
    {
      "epoch": 7.780075073242187e-06,
      "model_forward_time": 1.7952885627746582,
      "step": 82158
    },
    {
      "epoch": 7.780075073242187e-06,
      "step": 82158,
      "training_step_time": 7.853257179260254
    },
    {
      "epoch": 7.781028747558594e-06,
      "model_forward_time": 1.800786018371582,
      "step": 82159
    },
    {
      "epoch": 7.781028747558594e-06,
      "step": 82159,
      "training_step_time": 6.843924045562744
    },
    {
      "epoch": 7.781982421875e-06,
      "grad_norm": 0.08354548364877701,
      "learning_rate": 8.452718475555927e-07,
      "loss": 0.055,
      "step": 82160
    },
    {
      "epoch": 7.781982421875e-06,
      "model_forward_time": 1.7912302017211914,
      "step": 82160
    },
    {
      "dynamics_loss_avg": 0.047362190950661895,
      "epoch": 7.781982421875e-06,
      "step": 82160
    },
    {
      "action_loss_avg": 0.0011748904536943883,
      "epoch": 7.781982421875e-06,
      "step": 82160
    },
    {
      "epoch": 7.781982421875e-06,
      "step": 82160,
      "training_step_time": 6.839401721954346
    },
    {
      "epoch": 7.782936096191406e-06,
      "model_forward_time": 1.7989599704742432,
      "step": 82161
    },
    {
      "epoch": 7.782936096191406e-06,
      "step": 82161,
      "training_step_time": 6.7997331619262695
    },
    {
      "epoch": 7.783889770507813e-06,
      "model_forward_time": 1.8127460479736328,
      "step": 82162
    },
    {
      "epoch": 7.783889770507813e-06,
      "step": 82162,
      "training_step_time": 7.613349437713623
    },
    {
      "epoch": 7.784843444824219e-06,
      "model_forward_time": 1.7946419715881348,
      "step": 82163
    },
    {
      "epoch": 7.784843444824219e-06,
      "step": 82163,
      "training_step_time": 8.603349208831787
    },
    {
      "epoch": 7.785797119140625e-06,
      "model_forward_time": 1.7748911380767822,
      "step": 82164
    },
    {
      "epoch": 7.785797119140625e-06,
      "step": 82164,
      "training_step_time": 7.876530647277832
    },
    {
      "epoch": 7.78675079345703e-06,
      "model_forward_time": 2.979754686355591,
      "step": 82165
    },
    {
      "epoch": 7.78675079345703e-06,
      "step": 82165,
      "training_step_time": 9.250912189483643
    },
    {
      "epoch": 7.787704467773438e-06,
      "model_forward_time": 1.792076826095581,
      "step": 82166
    },
    {
      "epoch": 7.787704467773438e-06,
      "step": 82166,
      "training_step_time": 7.732131242752075
    },
    {
      "epoch": 7.788658142089844e-06,
      "model_forward_time": 1.7946181297302246,
      "step": 82167
    },
    {
      "epoch": 7.788658142089844e-06,
      "step": 82167,
      "training_step_time": 7.7794599533081055
    },
    {
      "epoch": 7.78961181640625e-06,
      "model_forward_time": 1.7965264320373535,
      "step": 82168
    },
    {
      "epoch": 7.78961181640625e-06,
      "step": 82168,
      "training_step_time": 6.8039422035217285
    },
    {
      "epoch": 7.790565490722657e-06,
      "model_forward_time": 1.787111520767212,
      "step": 82169
    },
    {
      "epoch": 7.790565490722657e-06,
      "step": 82169,
      "training_step_time": 6.835856199264526
    },
    {
      "epoch": 7.791519165039062e-06,
      "grad_norm": 0.11516530811786652,
      "learning_rate": 8.443521608182398e-07,
      "loss": 0.0559,
      "step": 82170
    },
    {
      "epoch": 7.791519165039062e-06,
      "model_forward_time": 1.794968843460083,
      "step": 82170
    },
    {
      "dynamics_loss_avg": 0.049119516741484404,
      "epoch": 7.791519165039062e-06,
      "step": 82170
    },
    {
      "action_loss_avg": 0.0015329008921980858,
      "epoch": 7.791519165039062e-06,
      "step": 82170
    },
    {
      "epoch": 7.791519165039062e-06,
      "step": 82170,
      "training_step_time": 6.7926836013793945
    },
    {
      "epoch": 7.792472839355468e-06,
      "model_forward_time": 1.792635440826416,
      "step": 82171
    },
    {
      "epoch": 7.792472839355468e-06,
      "step": 82171,
      "training_step_time": 7.5646936893463135
    },
    {
      "epoch": 7.793426513671876e-06,
      "model_forward_time": 1.7955636978149414,
      "step": 82172
    },
    {
      "epoch": 7.793426513671876e-06,
      "step": 82172,
      "training_step_time": 7.711725234985352
    },
    {
      "epoch": 7.794380187988281e-06,
      "model_forward_time": 1.7795932292938232,
      "step": 82173
    },
    {
      "epoch": 7.794380187988281e-06,
      "step": 82173,
      "training_step_time": 9.505454063415527
    },
    {
      "epoch": 7.795333862304687e-06,
      "model_forward_time": 1.9156455993652344,
      "step": 82174
    },
    {
      "epoch": 7.795333862304687e-06,
      "step": 82174,
      "training_step_time": 9.23478651046753
    },
    {
      "epoch": 7.796287536621094e-06,
      "model_forward_time": 1.7961406707763672,
      "step": 82175
    },
    {
      "epoch": 7.796287536621094e-06,
      "step": 82175,
      "training_step_time": 8.346147298812866
    },
    {
      "epoch": 7.7972412109375e-06,
      "model_forward_time": 1.7779018878936768,
      "step": 82176
    },
    {
      "epoch": 7.7972412109375e-06,
      "step": 82176,
      "training_step_time": 7.717753171920776
    },
    {
      "epoch": 7.798194885253906e-06,
      "model_forward_time": 1.7883822917938232,
      "step": 82177
    },
    {
      "epoch": 7.798194885253906e-06,
      "step": 82177,
      "training_step_time": 6.7800445556640625
    },
    {
      "epoch": 7.799148559570313e-06,
      "model_forward_time": 1.8029954433441162,
      "step": 82178
    },
    {
      "epoch": 7.799148559570313e-06,
      "step": 82178,
      "training_step_time": 6.79547643661499
    },
    {
      "epoch": 7.800102233886719e-06,
      "model_forward_time": 1.7902731895446777,
      "step": 82179
    },
    {
      "epoch": 7.800102233886719e-06,
      "step": 82179,
      "training_step_time": 6.81304931640625
    },
    {
      "epoch": 7.801055908203125e-06,
      "grad_norm": 0.08375280350446701,
      "learning_rate": 8.43432928536288e-07,
      "loss": 0.0556,
      "step": 82180
    },
    {
      "epoch": 7.801055908203125e-06,
      "model_forward_time": 1.7838001251220703,
      "step": 82180
    },
    {
      "dynamics_loss_avg": 0.063118439540267,
      "epoch": 7.801055908203125e-06,
      "step": 82180
    },
    {
      "action_loss_avg": 0.0017931058100657538,
      "epoch": 7.801055908203125e-06,
      "step": 82180
    },
    {
      "epoch": 7.801055908203125e-06,
      "step": 82180,
      "training_step_time": 7.596991539001465
    },
    {
      "epoch": 7.80200958251953e-06,
      "model_forward_time": 1.7865638732910156,
      "step": 82181
    },
    {
      "epoch": 7.80200958251953e-06,
      "step": 82181,
      "training_step_time": 8.517513036727905
    },
    {
      "epoch": 7.802963256835938e-06,
      "model_forward_time": 1.7801988124847412,
      "step": 82182
    },
    {
      "epoch": 7.802963256835938e-06,
      "step": 82182,
      "training_step_time": 9.569924116134644
    },
    {
      "epoch": 7.803916931152344e-06,
      "model_forward_time": 1.888230562210083,
      "step": 82183
    },
    {
      "epoch": 7.803916931152344e-06,
      "step": 82183,
      "training_step_time": 8.340177774429321
    },
    {
      "epoch": 7.80487060546875e-06,
      "model_forward_time": 2.882277011871338,
      "step": 82184
    },
    {
      "epoch": 7.80487060546875e-06,
      "step": 82184,
      "training_step_time": 9.884614706039429
    },
    {
      "epoch": 7.805824279785157e-06,
      "model_forward_time": 1.7865803241729736,
      "step": 82185
    },
    {
      "epoch": 7.805824279785157e-06,
      "step": 82185,
      "training_step_time": 7.700411319732666
    },
    {
      "epoch": 7.806777954101562e-06,
      "model_forward_time": 1.7900471687316895,
      "step": 82186
    },
    {
      "epoch": 7.806777954101562e-06,
      "step": 82186,
      "training_step_time": 7.857056140899658
    },
    {
      "epoch": 7.807731628417968e-06,
      "model_forward_time": 1.7937331199645996,
      "step": 82187
    },
    {
      "epoch": 7.807731628417968e-06,
      "step": 82187,
      "training_step_time": 6.83403205871582
    },
    {
      "epoch": 7.808685302734376e-06,
      "model_forward_time": 1.7608826160430908,
      "step": 82188
    },
    {
      "epoch": 7.808685302734376e-06,
      "step": 82188,
      "training_step_time": 6.758682012557983
    },
    {
      "epoch": 7.809638977050781e-06,
      "model_forward_time": 1.7878484725952148,
      "step": 82189
    },
    {
      "epoch": 7.809638977050781e-06,
      "step": 82189,
      "training_step_time": 7.67547607421875
    },
    {
      "epoch": 7.810592651367187e-06,
      "grad_norm": 0.15566371381282806,
      "learning_rate": 8.425141508102608e-07,
      "loss": 0.0554,
      "step": 82190
    },
    {
      "epoch": 7.810592651367187e-06,
      "model_forward_time": 1.7890057563781738,
      "step": 82190
    },
    {
      "dynamics_loss_avg": 0.04958950085565448,
      "epoch": 7.810592651367187e-06,
      "step": 82190
    },
    {
      "action_loss_avg": 0.0013525436370400712,
      "epoch": 7.810592651367187e-06,
      "step": 82190
    },
    {
      "epoch": 7.810592651367187e-06,
      "step": 82190,
      "training_step_time": 7.437995433807373
    },
    {
      "epoch": 7.811546325683594e-06,
      "model_forward_time": 1.7800815105438232,
      "step": 82191
    },
    {
      "epoch": 7.811546325683594e-06,
      "step": 82191,
      "training_step_time": 8.43270468711853
    },
    {
      "epoch": 7.8125e-06,
      "model_forward_time": 1.8536198139190674,
      "step": 82192
    },
    {
      "epoch": 7.8125e-06,
      "step": 82192,
      "training_step_time": 10.329526901245117
    },
    {
      "epoch": 7.813453674316406e-06,
      "model_forward_time": 2.920262575149536,
      "step": 82193
    },
    {
      "epoch": 7.813453674316406e-06,
      "step": 82193,
      "training_step_time": 8.760082721710205
    },
    {
      "epoch": 7.814407348632813e-06,
      "model_forward_time": 1.7908122539520264,
      "step": 82194
    },
    {
      "epoch": 7.814407348632813e-06,
      "step": 82194,
      "training_step_time": 7.878091096878052
    },
    {
      "epoch": 7.815361022949219e-06,
      "model_forward_time": 1.7869350910186768,
      "step": 82195
    },
    {
      "epoch": 7.815361022949219e-06,
      "step": 82195,
      "training_step_time": 7.825050354003906
    },
    {
      "epoch": 7.816314697265625e-06,
      "model_forward_time": 1.7959668636322021,
      "step": 82196
    },
    {
      "epoch": 7.816314697265625e-06,
      "step": 82196,
      "training_step_time": 6.85717511177063
    },
    {
      "epoch": 7.81726837158203e-06,
      "model_forward_time": 1.793581247329712,
      "step": 82197
    },
    {
      "epoch": 7.81726837158203e-06,
      "step": 82197,
      "training_step_time": 6.858862400054932
    },
    {
      "epoch": 7.818222045898438e-06,
      "model_forward_time": 1.7934987545013428,
      "step": 82198
    },
    {
      "epoch": 7.818222045898438e-06,
      "step": 82198,
      "training_step_time": 7.469174146652222
    },
    {
      "epoch": 7.819175720214844e-06,
      "model_forward_time": 1.7906477451324463,
      "step": 82199
    },
    {
      "epoch": 7.819175720214844e-06,
      "step": 82199,
      "training_step_time": 7.390461444854736
    },
    {
      "epoch": 7.82012939453125e-06,
      "grad_norm": 0.10507165640592575,
      "learning_rate": 8.415958277406366e-07,
      "loss": 0.0552,
      "step": 82200
    },
    {
      "epoch": 7.82012939453125e-06,
      "model_forward_time": 1.7455992698669434,
      "step": 82200
    },
    {
      "dynamics_loss_avg": 0.05098732002079487,
      "epoch": 7.82012939453125e-06,
      "step": 82200
    },
    {
      "action_loss_avg": 0.0013315911724930628,
      "epoch": 7.82012939453125e-06,
      "step": 82200
    },
    {
      "epoch": 7.82012939453125e-06,
      "step": 82200,
      "training_step_time": 8.54975938796997
    },
    {
      "epoch": 7.821083068847657e-06,
      "model_forward_time": 1.8062467575073242,
      "step": 82201
    },
    {
      "epoch": 7.821083068847657e-06,
      "step": 82201,
      "training_step_time": 8.849583387374878
    },
    {
      "epoch": 7.822036743164062e-06,
      "model_forward_time": 2.906750202178955,
      "step": 82202
    },
    {
      "epoch": 7.822036743164062e-06,
      "step": 82202,
      "training_step_time": 8.71460509300232
    },
    {
      "epoch": 7.822990417480468e-06,
      "model_forward_time": 1.7997982501983643,
      "step": 82203
    },
    {
      "epoch": 7.822990417480468e-06,
      "step": 82203,
      "training_step_time": 7.832247495651245
    },
    {
      "epoch": 7.823944091796876e-06,
      "model_forward_time": 1.7923712730407715,
      "step": 82204
    },
    {
      "epoch": 7.823944091796876e-06,
      "step": 82204,
      "training_step_time": 7.719939470291138
    },
    {
      "epoch": 7.824897766113281e-06,
      "model_forward_time": 1.7924387454986572,
      "step": 82205
    },
    {
      "epoch": 7.824897766113281e-06,
      "step": 82205,
      "training_step_time": 6.841696500778198
    },
    {
      "epoch": 7.825851440429687e-06,
      "model_forward_time": 1.8138294219970703,
      "step": 82206
    },
    {
      "epoch": 7.825851440429687e-06,
      "step": 82206,
      "training_step_time": 6.871840715408325
    },
    {
      "epoch": 7.826805114746094e-06,
      "model_forward_time": 1.7953362464904785,
      "step": 82207
    },
    {
      "epoch": 7.826805114746094e-06,
      "step": 82207,
      "training_step_time": 7.733385324478149
    },
    {
      "epoch": 7.8277587890625e-06,
      "model_forward_time": 1.7824363708496094,
      "step": 82208
    },
    {
      "epoch": 7.8277587890625e-06,
      "step": 82208,
      "training_step_time": 7.538649559020996
    },
    {
      "epoch": 7.828712463378906e-06,
      "model_forward_time": 1.781947135925293,
      "step": 82209
    },
    {
      "epoch": 7.828712463378906e-06,
      "step": 82209,
      "training_step_time": 7.831407308578491
    },
    {
      "epoch": 7.829666137695313e-06,
      "grad_norm": 0.10741852968931198,
      "learning_rate": 8.406779594278391e-07,
      "loss": 0.0543,
      "step": 82210
    },
    {
      "epoch": 7.829666137695313e-06,
      "model_forward_time": 1.7870368957519531,
      "step": 82210
    },
    {
      "dynamics_loss_avg": 0.057481957413256166,
      "epoch": 7.829666137695313e-06,
      "step": 82210
    },
    {
      "action_loss_avg": 0.0017282010230701417,
      "epoch": 7.829666137695313e-06,
      "step": 82210
    },
    {
      "epoch": 7.829666137695313e-06,
      "step": 82210,
      "training_step_time": 8.399003028869629
    },
    {
      "epoch": 7.830619812011719e-06,
      "model_forward_time": 2.896836757659912,
      "step": 82211
    },
    {
      "epoch": 7.830619812011719e-06,
      "step": 82211,
      "training_step_time": 8.915429830551147
    },
    {
      "epoch": 7.831573486328125e-06,
      "model_forward_time": 1.7972164154052734,
      "step": 82212
    },
    {
      "epoch": 7.831573486328125e-06,
      "step": 82212,
      "training_step_time": 9.35411262512207
    },
    {
      "epoch": 7.83252716064453e-06,
      "model_forward_time": 1.791750431060791,
      "step": 82213
    },
    {
      "epoch": 7.83252716064453e-06,
      "step": 82213,
      "training_step_time": 7.6548802852630615
    },
    {
      "epoch": 7.833480834960938e-06,
      "model_forward_time": 1.7793214321136475,
      "step": 82214
    },
    {
      "epoch": 7.833480834960938e-06,
      "step": 82214,
      "training_step_time": 7.7648398876190186
    },
    {
      "epoch": 7.834434509277344e-06,
      "model_forward_time": 1.7891662120819092,
      "step": 82215
    },
    {
      "epoch": 7.834434509277344e-06,
      "step": 82215,
      "training_step_time": 6.803286790847778
    },
    {
      "epoch": 7.83538818359375e-06,
      "model_forward_time": 1.8094744682312012,
      "step": 82216
    },
    {
      "epoch": 7.83538818359375e-06,
      "step": 82216,
      "training_step_time": 7.629438638687134
    },
    {
      "epoch": 7.836341857910157e-06,
      "model_forward_time": 1.787532091140747,
      "step": 82217
    },
    {
      "epoch": 7.836341857910157e-06,
      "step": 82217,
      "training_step_time": 7.543793201446533
    },
    {
      "epoch": 7.837295532226562e-06,
      "model_forward_time": 1.7872092723846436,
      "step": 82218
    },
    {
      "epoch": 7.837295532226562e-06,
      "step": 82218,
      "training_step_time": 7.780351400375366
    },
    {
      "epoch": 7.838249206542968e-06,
      "model_forward_time": 1.7958104610443115,
      "step": 82219
    },
    {
      "epoch": 7.838249206542968e-06,
      "step": 82219,
      "training_step_time": 7.815714597702026
    },
    {
      "epoch": 7.839202880859376e-06,
      "grad_norm": 0.07756439596414566,
      "learning_rate": 8.397605459722469e-07,
      "loss": 0.054,
      "step": 82220
    },
    {
      "epoch": 7.839202880859376e-06,
      "model_forward_time": 2.9149296283721924,
      "step": 82220
    },
    {
      "dynamics_loss_avg": 0.05429653413593769,
      "epoch": 7.839202880859376e-06,
      "step": 82220
    },
    {
      "action_loss_avg": 0.0016954461578279733,
      "epoch": 7.839202880859376e-06,
      "step": 82220
    },
    {
      "epoch": 7.839202880859376e-06,
      "step": 82220,
      "training_step_time": 9.22983455657959
    },
    {
      "epoch": 7.840156555175781e-06,
      "model_forward_time": 1.7844300270080566,
      "step": 82221
    },
    {
      "epoch": 7.840156555175781e-06,
      "step": 82221,
      "training_step_time": 8.717522621154785
    },
    {
      "epoch": 7.841110229492187e-06,
      "model_forward_time": 1.7547979354858398,
      "step": 82222
    },
    {
      "epoch": 7.841110229492187e-06,
      "step": 82222,
      "training_step_time": 7.756871223449707
    },
    {
      "epoch": 7.842063903808594e-06,
      "model_forward_time": 1.7666034698486328,
      "step": 82223
    },
    {
      "epoch": 7.842063903808594e-06,
      "step": 82223,
      "training_step_time": 7.8481738567352295
    },
    {
      "epoch": 7.843017578125e-06,
      "model_forward_time": 1.7822537422180176,
      "step": 82224
    },
    {
      "epoch": 7.843017578125e-06,
      "step": 82224,
      "training_step_time": 6.812293529510498
    },
    {
      "epoch": 7.843971252441406e-06,
      "model_forward_time": 1.7654757499694824,
      "step": 82225
    },
    {
      "epoch": 7.843971252441406e-06,
      "step": 82225,
      "training_step_time": 7.617938995361328
    },
    {
      "epoch": 7.844924926757813e-06,
      "model_forward_time": 1.7688329219818115,
      "step": 82226
    },
    {
      "epoch": 7.844924926757813e-06,
      "step": 82226,
      "training_step_time": 7.550425291061401
    },
    {
      "epoch": 7.845878601074219e-06,
      "model_forward_time": 1.7577085494995117,
      "step": 82227
    },
    {
      "epoch": 7.845878601074219e-06,
      "step": 82227,
      "training_step_time": 8.570383071899414
    },
    {
      "epoch": 7.846832275390625e-06,
      "model_forward_time": 1.795985460281372,
      "step": 82228
    },
    {
      "epoch": 7.846832275390625e-06,
      "step": 82228,
      "training_step_time": 9.204487800598145
    },
    {
      "epoch": 7.84778594970703e-06,
      "model_forward_time": 1.8592774868011475,
      "step": 82229
    },
    {
      "epoch": 7.84778594970703e-06,
      "step": 82229,
      "training_step_time": 10.937248706817627
    },
    {
      "epoch": 7.848739624023438e-06,
      "grad_norm": 0.0980515331029892,
      "learning_rate": 8.388435874741863e-07,
      "loss": 0.054,
      "step": 82230
    },
    {
      "epoch": 7.848739624023438e-06,
      "model_forward_time": 1.7833268642425537,
      "step": 82230
    },
    {
      "dynamics_loss_avg": 0.054073984269052744,
      "epoch": 7.848739624023438e-06,
      "step": 82230
    },
    {
      "action_loss_avg": 0.001294990349560976,
      "epoch": 7.848739624023438e-06,
      "step": 82230
    },
    {
      "epoch": 7.848739624023438e-06,
      "step": 82230,
      "training_step_time": 9.43747591972351
    },
    {
      "epoch": 7.849693298339844e-06,
      "model_forward_time": 1.789968490600586,
      "step": 82231
    },
    {
      "epoch": 7.849693298339844e-06,
      "step": 82231,
      "training_step_time": 7.759407043457031
    },
    {
      "epoch": 7.85064697265625e-06,
      "model_forward_time": 1.7777738571166992,
      "step": 82232
    },
    {
      "epoch": 7.85064697265625e-06,
      "step": 82232,
      "training_step_time": 7.836187839508057
    },
    {
      "epoch": 7.851600646972657e-06,
      "model_forward_time": 1.7896173000335693,
      "step": 82233
    },
    {
      "epoch": 7.851600646972657e-06,
      "step": 82233,
      "training_step_time": 6.833952903747559
    },
    {
      "epoch": 7.852554321289062e-06,
      "model_forward_time": 1.4230036735534668,
      "step": 82234
    },
    {
      "epoch": 7.852554321289062e-06,
      "step": 82234,
      "training_step_time": 6.815450429916382
    },
    {
      "epoch": 7.853507995605468e-06,
      "model_forward_time": 1.7863819599151611,
      "step": 82235
    },
    {
      "epoch": 7.853507995605468e-06,
      "step": 82235,
      "training_step_time": 7.661130666732788
    },
    {
      "epoch": 7.854461669921876e-06,
      "model_forward_time": 1.7615675926208496,
      "step": 82236
    },
    {
      "epoch": 7.854461669921876e-06,
      "step": 82236,
      "training_step_time": 7.618346691131592
    },
    {
      "epoch": 7.855415344238281e-06,
      "model_forward_time": 1.7771806716918945,
      "step": 82237
    },
    {
      "epoch": 7.855415344238281e-06,
      "step": 82237,
      "training_step_time": 8.49910855293274
    },
    {
      "epoch": 7.856369018554687e-06,
      "model_forward_time": 1.8325755596160889,
      "step": 82238
    },
    {
      "epoch": 7.856369018554687e-06,
      "step": 82238,
      "training_step_time": 8.476457118988037
    },
    {
      "epoch": 7.857322692871094e-06,
      "model_forward_time": 2.9160425662994385,
      "step": 82239
    },
    {
      "epoch": 7.857322692871094e-06,
      "step": 82239,
      "training_step_time": 8.423085451126099
    },
    {
      "epoch": 7.8582763671875e-06,
      "grad_norm": 0.07959491014480591,
      "learning_rate": 8.37927084033936e-07,
      "loss": 0.0543,
      "step": 82240
    },
    {
      "epoch": 7.8582763671875e-06,
      "model_forward_time": 1.7937767505645752,
      "step": 82240
    },
    {
      "dynamics_loss_avg": 0.062008631974458696,
      "epoch": 7.8582763671875e-06,
      "step": 82240
    },
    {
      "action_loss_avg": 0.0012456199678126723,
      "epoch": 7.8582763671875e-06,
      "step": 82240
    },
    {
      "epoch": 7.8582763671875e-06,
      "step": 82240,
      "training_step_time": 9.350888967514038
    },
    {
      "epoch": 7.859230041503906e-06,
      "model_forward_time": 1.7923424243927002,
      "step": 82241
    },
    {
      "epoch": 7.859230041503906e-06,
      "step": 82241,
      "training_step_time": 7.800743818283081
    },
    {
      "epoch": 7.860183715820313e-06,
      "model_forward_time": 1.785844087600708,
      "step": 82242
    },
    {
      "epoch": 7.860183715820313e-06,
      "step": 82242,
      "training_step_time": 6.845258951187134
    },
    {
      "epoch": 7.861137390136719e-06,
      "model_forward_time": 1.7915759086608887,
      "step": 82243
    },
    {
      "epoch": 7.861137390136719e-06,
      "step": 82243,
      "training_step_time": 6.84977126121521
    },
    {
      "epoch": 7.862091064453125e-06,
      "model_forward_time": 1.7927677631378174,
      "step": 82244
    },
    {
      "epoch": 7.862091064453125e-06,
      "step": 82244,
      "training_step_time": 7.722765684127808
    },
    {
      "epoch": 7.86304473876953e-06,
      "model_forward_time": 1.7902727127075195,
      "step": 82245
    },
    {
      "epoch": 7.86304473876953e-06,
      "step": 82245,
      "training_step_time": 7.488867521286011
    },
    {
      "epoch": 7.863998413085938e-06,
      "model_forward_time": 1.771798849105835,
      "step": 82246
    },
    {
      "epoch": 7.863998413085938e-06,
      "step": 82246,
      "training_step_time": 9.620871543884277
    },
    {
      "epoch": 7.864952087402344e-06,
      "model_forward_time": 1.3882248401641846,
      "step": 82247
    },
    {
      "epoch": 7.864952087402344e-06,
      "step": 82247,
      "training_step_time": 9.291430473327637
    },
    {
      "epoch": 7.86590576171875e-06,
      "model_forward_time": 2.903893232345581,
      "step": 82248
    },
    {
      "epoch": 7.86590576171875e-06,
      "step": 82248,
      "training_step_time": 10.16528582572937
    },
    {
      "epoch": 7.866859436035157e-06,
      "model_forward_time": 1.771935224533081,
      "step": 82249
    },
    {
      "epoch": 7.866859436035157e-06,
      "step": 82249,
      "training_step_time": 8.725677967071533
    },
    {
      "epoch": 7.867813110351562e-06,
      "grad_norm": 0.09122338891029358,
      "learning_rate": 8.370110357517208e-07,
      "loss": 0.0544,
      "step": 82250
    },
    {
      "epoch": 7.867813110351562e-06,
      "model_forward_time": 1.784553050994873,
      "step": 82250
    },
    {
      "dynamics_loss_avg": 0.05002443511039019,
      "epoch": 7.867813110351562e-06,
      "step": 82250
    },
    {
      "action_loss_avg": 0.0008526820631232113,
      "epoch": 7.867813110351562e-06,
      "step": 82250
    },
    {
      "epoch": 7.867813110351562e-06,
      "step": 82250,
      "training_step_time": 7.952123641967773
    },
    {
      "epoch": 7.868766784667968e-06,
      "model_forward_time": 1.7780158519744873,
      "step": 82251
    },
    {
      "epoch": 7.868766784667968e-06,
      "step": 82251,
      "training_step_time": 7.472052097320557
    },
    {
      "epoch": 7.869720458984376e-06,
      "model_forward_time": 1.7715051174163818,
      "step": 82252
    },
    {
      "epoch": 7.869720458984376e-06,
      "step": 82252,
      "training_step_time": 6.878962278366089
    },
    {
      "epoch": 7.870674133300781e-06,
      "model_forward_time": 1.7729079723358154,
      "step": 82253
    },
    {
      "epoch": 7.870674133300781e-06,
      "step": 82253,
      "training_step_time": 7.784448862075806
    },
    {
      "epoch": 7.871627807617187e-06,
      "model_forward_time": 1.7726733684539795,
      "step": 82254
    },
    {
      "epoch": 7.871627807617187e-06,
      "step": 82254,
      "training_step_time": 7.625121593475342
    },
    {
      "epoch": 7.872581481933594e-06,
      "model_forward_time": 1.7673418521881104,
      "step": 82255
    },
    {
      "epoch": 7.872581481933594e-06,
      "step": 82255,
      "training_step_time": 9.534358501434326
    },
    {
      "epoch": 7.87353515625e-06,
      "model_forward_time": 1.7655980587005615,
      "step": 82256
    },
    {
      "epoch": 7.87353515625e-06,
      "step": 82256,
      "training_step_time": 10.672587871551514
    },
    {
      "epoch": 7.874488830566406e-06,
      "model_forward_time": 2.8468635082244873,
      "step": 82257
    },
    {
      "epoch": 7.874488830566406e-06,
      "step": 82257,
      "training_step_time": 9.870363712310791
    },
    {
      "epoch": 7.875442504882812e-06,
      "model_forward_time": 1.7694239616394043,
      "step": 82258
    },
    {
      "epoch": 7.875442504882812e-06,
      "step": 82258,
      "training_step_time": 9.914293766021729
    },
    {
      "epoch": 7.876396179199219e-06,
      "model_forward_time": 1.7639007568359375,
      "step": 82259
    },
    {
      "epoch": 7.876396179199219e-06,
      "step": 82259,
      "training_step_time": 7.795092344284058
    },
    {
      "epoch": 7.877349853515625e-06,
      "grad_norm": 0.07650619745254517,
      "learning_rate": 8.360954427277201e-07,
      "loss": 0.054,
      "step": 82260
    },
    {
      "epoch": 7.877349853515625e-06,
      "model_forward_time": 1.7766368389129639,
      "step": 82260
    },
    {
      "dynamics_loss_avg": 0.06490308661013841,
      "epoch": 7.877349853515625e-06,
      "step": 82260
    },
    {
      "action_loss_avg": 0.0014322094502858818,
      "epoch": 7.877349853515625e-06,
      "step": 82260
    },
    {
      "epoch": 7.877349853515625e-06,
      "step": 82260,
      "training_step_time": 7.385686635971069
    },
    {
      "epoch": 7.87830352783203e-06,
      "model_forward_time": 1.764960765838623,
      "step": 82261
    },
    {
      "epoch": 7.87830352783203e-06,
      "step": 82261,
      "training_step_time": 6.879705905914307
    },
    {
      "epoch": 7.879257202148438e-06,
      "model_forward_time": 1.7644007205963135,
      "step": 82262
    },
    {
      "epoch": 7.879257202148438e-06,
      "step": 82262,
      "training_step_time": 7.7184295654296875
    },
    {
      "epoch": 7.880210876464844e-06,
      "model_forward_time": 1.7650809288024902,
      "step": 82263
    },
    {
      "epoch": 7.880210876464844e-06,
      "step": 82263,
      "training_step_time": 7.585106611251831
    },
    {
      "epoch": 7.88116455078125e-06,
      "model_forward_time": 1.7921743392944336,
      "step": 82264
    },
    {
      "epoch": 7.88116455078125e-06,
      "step": 82264,
      "training_step_time": 7.73698353767395
    },
    {
      "epoch": 7.882118225097657e-06,
      "model_forward_time": 1.7662668228149414,
      "step": 82265
    },
    {
      "epoch": 7.882118225097657e-06,
      "step": 82265,
      "training_step_time": 8.861952781677246
    },
    {
      "epoch": 7.883071899414063e-06,
      "model_forward_time": 2.831847906112671,
      "step": 82266
    },
    {
      "epoch": 7.883071899414063e-06,
      "step": 82266,
      "training_step_time": 10.853443622589111
    },
    {
      "epoch": 7.884025573730468e-06,
      "model_forward_time": 1.7654123306274414,
      "step": 82267
    },
    {
      "epoch": 7.884025573730468e-06,
      "step": 82267,
      "training_step_time": 8.850990533828735
    },
    {
      "epoch": 7.884979248046876e-06,
      "model_forward_time": 1.7683818340301514,
      "step": 82268
    },
    {
      "epoch": 7.884979248046876e-06,
      "step": 82268,
      "training_step_time": 8.014932632446289
    },
    {
      "epoch": 7.885932922363281e-06,
      "model_forward_time": 1.7639801502227783,
      "step": 82269
    },
    {
      "epoch": 7.885932922363281e-06,
      "step": 82269,
      "training_step_time": 6.9635303020477295
    },
    {
      "epoch": 7.886886596679687e-06,
      "grad_norm": 0.07673163712024689,
      "learning_rate": 8.351803050620627e-07,
      "loss": 0.0551,
      "step": 82270
    },
    {
      "epoch": 7.886886596679687e-06,
      "model_forward_time": 1.7561986446380615,
      "step": 82270
    },
    {
      "dynamics_loss_avg": 0.04644677024334669,
      "epoch": 7.886886596679687e-06,
      "step": 82270
    },
    {
      "action_loss_avg": 0.0013308867579326033,
      "epoch": 7.886886596679687e-06,
      "step": 82270
    },
    {
      "epoch": 7.886886596679687e-06,
      "step": 82270,
      "training_step_time": 6.926387786865234
    },
    {
      "epoch": 7.887840270996095e-06,
      "model_forward_time": 1.7749309539794922,
      "step": 82271
    },
    {
      "epoch": 7.887840270996095e-06,
      "step": 82271,
      "training_step_time": 7.678032875061035
    },
    {
      "epoch": 7.8887939453125e-06,
      "model_forward_time": 1.756666660308838,
      "step": 82272
    },
    {
      "epoch": 7.8887939453125e-06,
      "step": 82272,
      "training_step_time": 7.5094122886657715
    },
    {
      "epoch": 7.889747619628906e-06,
      "model_forward_time": 1.800121545791626,
      "step": 82273
    },
    {
      "epoch": 7.889747619628906e-06,
      "step": 82273,
      "training_step_time": 7.7404656410217285
    },
    {
      "epoch": 7.890701293945312e-06,
      "model_forward_time": 1.7730011940002441,
      "step": 82274
    },
    {
      "epoch": 7.890701293945312e-06,
      "step": 82274,
      "training_step_time": 9.874431133270264
    },
    {
      "epoch": 7.891654968261719e-06,
      "model_forward_time": 1.7938618659973145,
      "step": 82275
    },
    {
      "epoch": 7.891654968261719e-06,
      "step": 82275,
      "training_step_time": 10.782214879989624
    },
    {
      "epoch": 7.892608642578125e-06,
      "model_forward_time": 2.8937337398529053,
      "step": 82276
    },
    {
      "epoch": 7.892608642578125e-06,
      "step": 82276,
      "training_step_time": 8.663010120391846
    },
    {
      "epoch": 7.89356231689453e-06,
      "model_forward_time": 1.7720143795013428,
      "step": 82277
    },
    {
      "epoch": 7.89356231689453e-06,
      "step": 82277,
      "training_step_time": 8.00136685371399
    },
    {
      "epoch": 7.894515991210938e-06,
      "model_forward_time": 1.4082586765289307,
      "step": 82278
    },
    {
      "epoch": 7.894515991210938e-06,
      "step": 82278,
      "training_step_time": 7.851342678070068
    },
    {
      "epoch": 7.895469665527344e-06,
      "model_forward_time": 1.7634944915771484,
      "step": 82279
    },
    {
      "epoch": 7.895469665527344e-06,
      "step": 82279,
      "training_step_time": 7.355481147766113
    },
    {
      "epoch": 7.89642333984375e-06,
      "grad_norm": 0.07732589542865753,
      "learning_rate": 8.342656228548241e-07,
      "loss": 0.0541,
      "step": 82280
    },
    {
      "epoch": 7.89642333984375e-06,
      "model_forward_time": 1.761063575744629,
      "step": 82280
    },
    {
      "dynamics_loss_avg": 0.061421144753694534,
      "epoch": 7.89642333984375e-06,
      "step": 82280
    },
    {
      "action_loss_avg": 0.0027045329770771787,
      "epoch": 7.89642333984375e-06,
      "step": 82280
    },
    {
      "epoch": 7.89642333984375e-06,
      "step": 82280,
      "training_step_time": 6.951303720474243
    },
    {
      "epoch": 7.897377014160157e-06,
      "model_forward_time": 1.7621822357177734,
      "step": 82281
    },
    {
      "epoch": 7.897377014160157e-06,
      "step": 82281,
      "training_step_time": 8.109798669815063
    },
    {
      "epoch": 7.898330688476563e-06,
      "model_forward_time": 1.7892279624938965,
      "step": 82282
    },
    {
      "epoch": 7.898330688476563e-06,
      "step": 82282,
      "training_step_time": 7.653551340103149
    },
    {
      "epoch": 7.899284362792968e-06,
      "model_forward_time": 1.762343168258667,
      "step": 82283
    },
    {
      "epoch": 7.899284362792968e-06,
      "step": 82283,
      "training_step_time": 8.251468420028687
    },
    {
      "epoch": 7.900238037109376e-06,
      "model_forward_time": 1.7649762630462646,
      "step": 82284
    },
    {
      "epoch": 7.900238037109376e-06,
      "step": 82284,
      "training_step_time": 12.330769777297974
    },
    {
      "epoch": 7.901191711425781e-06,
      "model_forward_time": 2.9370744228363037,
      "step": 82285
    },
    {
      "epoch": 7.901191711425781e-06,
      "step": 82285,
      "training_step_time": 7.874423027038574
    },
    {
      "epoch": 7.902145385742187e-06,
      "model_forward_time": 1.7762789726257324,
      "step": 82286
    },
    {
      "epoch": 7.902145385742187e-06,
      "step": 82286,
      "training_step_time": 8.396754264831543
    },
    {
      "epoch": 7.903099060058595e-06,
      "model_forward_time": 1.7656610012054443,
      "step": 82287
    },
    {
      "epoch": 7.903099060058595e-06,
      "step": 82287,
      "training_step_time": 7.801190376281738
    },
    {
      "epoch": 7.904052734375e-06,
      "model_forward_time": 1.7671542167663574,
      "step": 82288
    },
    {
      "epoch": 7.904052734375e-06,
      "step": 82288,
      "training_step_time": 7.349652528762817
    },
    {
      "epoch": 7.905006408691406e-06,
      "model_forward_time": 1.7724745273590088,
      "step": 82289
    },
    {
      "epoch": 7.905006408691406e-06,
      "step": 82289,
      "training_step_time": 6.911980152130127
    },
    {
      "epoch": 7.905960083007812e-06,
      "grad_norm": 0.08088735491037369,
      "learning_rate": 8.333513962060336e-07,
      "loss": 0.0551,
      "step": 82290
    },
    {
      "epoch": 7.905960083007812e-06,
      "model_forward_time": 1.758103847503662,
      "step": 82290
    },
    {
      "dynamics_loss_avg": 0.05430537303909659,
      "epoch": 7.905960083007812e-06,
      "step": 82290
    },
    {
      "action_loss_avg": 0.0013572576397564262,
      "epoch": 7.905960083007812e-06,
      "step": 82290
    },
    {
      "epoch": 7.905960083007812e-06,
      "step": 82290,
      "training_step_time": 7.532843351364136
    },
    {
      "epoch": 7.906913757324219e-06,
      "model_forward_time": 1.7908787727355957,
      "step": 82291
    },
    {
      "epoch": 7.906913757324219e-06,
      "step": 82291,
      "training_step_time": 7.474632740020752
    },
    {
      "epoch": 7.907867431640625e-06,
      "model_forward_time": 1.7662503719329834,
      "step": 82292
    },
    {
      "epoch": 7.907867431640625e-06,
      "step": 82292,
      "training_step_time": 7.773196697235107
    },
    {
      "epoch": 7.90882110595703e-06,
      "model_forward_time": 1.7706425189971924,
      "step": 82293
    },
    {
      "epoch": 7.90882110595703e-06,
      "step": 82293,
      "training_step_time": 9.983705282211304
    },
    {
      "epoch": 7.909774780273438e-06,
      "model_forward_time": 2.8341338634490967,
      "step": 82294
    },
    {
      "epoch": 7.909774780273438e-06,
      "step": 82294,
      "training_step_time": 8.674458742141724
    },
    {
      "epoch": 7.910728454589844e-06,
      "model_forward_time": 1.7631092071533203,
      "step": 82295
    },
    {
      "epoch": 7.910728454589844e-06,
      "step": 82295,
      "training_step_time": 8.711410999298096
    },
    {
      "epoch": 7.91168212890625e-06,
      "model_forward_time": 1.7635128498077393,
      "step": 82296
    },
    {
      "epoch": 7.91168212890625e-06,
      "step": 82296,
      "training_step_time": 7.942619323730469
    },
    {
      "epoch": 7.912635803222657e-06,
      "model_forward_time": 1.7808151245117188,
      "step": 82297
    },
    {
      "epoch": 7.912635803222657e-06,
      "step": 82297,
      "training_step_time": 6.964214086532593
    },
    {
      "epoch": 7.913589477539063e-06,
      "model_forward_time": 1.773740530014038,
      "step": 82298
    },
    {
      "epoch": 7.913589477539063e-06,
      "step": 82298,
      "training_step_time": 6.888351678848267
    },
    {
      "epoch": 7.914543151855468e-06,
      "model_forward_time": 1.7804856300354004,
      "step": 82299
    },
    {
      "epoch": 7.914543151855468e-06,
      "step": 82299,
      "training_step_time": 7.520343780517578
    },
    {
      "epoch": 7.915496826171876e-06,
      "grad_norm": 0.10569349676370621,
      "learning_rate": 8.324376252156702e-07,
      "loss": 0.057,
      "step": 82300
    },
    {
      "epoch": 7.915496826171876e-06,
      "model_forward_time": 1.7875428199768066,
      "step": 82300
    },
    {
      "dynamics_loss_avg": 0.042219805903732775,
      "epoch": 7.915496826171876e-06,
      "step": 82300
    },
    {
      "action_loss_avg": 0.0015537888335529714,
      "epoch": 7.915496826171876e-06,
      "step": 82300
    },
    {
      "epoch": 7.915496826171876e-06,
      "step": 82300,
      "training_step_time": 7.474422454833984
    },
    {
      "epoch": 7.916450500488281e-06,
      "model_forward_time": 1.7653272151947021,
      "step": 82301
    },
    {
      "epoch": 7.916450500488281e-06,
      "step": 82301,
      "training_step_time": 9.84010362625122
    },
    {
      "epoch": 7.917404174804687e-06,
      "model_forward_time": 1.7764294147491455,
      "step": 82302
    },
    {
      "epoch": 7.917404174804687e-06,
      "step": 82302,
      "training_step_time": 9.706963300704956
    },
    {
      "epoch": 7.918357849121095e-06,
      "model_forward_time": 2.8474044799804688,
      "step": 82303
    },
    {
      "epoch": 7.918357849121095e-06,
      "step": 82303,
      "training_step_time": 7.890535354614258
    },
    {
      "epoch": 7.9193115234375e-06,
      "model_forward_time": 1.7662055492401123,
      "step": 82304
    },
    {
      "epoch": 7.9193115234375e-06,
      "step": 82304,
      "training_step_time": 7.7892491817474365
    },
    {
      "epoch": 7.920265197753906e-06,
      "model_forward_time": 1.7733385562896729,
      "step": 82305
    },
    {
      "epoch": 7.920265197753906e-06,
      "step": 82305,
      "training_step_time": 8.776939392089844
    },
    {
      "epoch": 7.921218872070312e-06,
      "model_forward_time": 1.7706255912780762,
      "step": 82306
    },
    {
      "epoch": 7.921218872070312e-06,
      "step": 82306,
      "training_step_time": 7.8387274742126465
    },
    {
      "epoch": 7.922172546386719e-06,
      "model_forward_time": 1.7818999290466309,
      "step": 82307
    },
    {
      "epoch": 7.922172546386719e-06,
      "step": 82307,
      "training_step_time": 6.9065539836883545
    },
    {
      "epoch": 7.923126220703125e-06,
      "model_forward_time": 1.769068956375122,
      "step": 82308
    },
    {
      "epoch": 7.923126220703125e-06,
      "step": 82308,
      "training_step_time": 7.538111448287964
    },
    {
      "epoch": 7.92407989501953e-06,
      "model_forward_time": 1.7953238487243652,
      "step": 82309
    },
    {
      "epoch": 7.92407989501953e-06,
      "step": 82309,
      "training_step_time": 7.5111987590789795
    },
    {
      "epoch": 7.925033569335938e-06,
      "grad_norm": 0.15934324264526367,
      "learning_rate": 8.315243099836634e-07,
      "loss": 0.0575,
      "step": 82310
    },
    {
      "epoch": 7.925033569335938e-06,
      "model_forward_time": 1.7684438228607178,
      "step": 82310
    },
    {
      "dynamics_loss_avg": 0.05871176458895207,
      "epoch": 7.925033569335938e-06,
      "step": 82310
    },
    {
      "action_loss_avg": 0.0015353179071098566,
      "epoch": 7.925033569335938e-06,
      "step": 82310
    },
    {
      "epoch": 7.925033569335938e-06,
      "step": 82310,
      "training_step_time": 9.574795246124268
    },
    {
      "epoch": 7.925987243652344e-06,
      "model_forward_time": 1.783736228942871,
      "step": 82311
    },
    {
      "epoch": 7.925987243652344e-06,
      "step": 82311,
      "training_step_time": 7.7971577644348145
    },
    {
      "epoch": 7.92694091796875e-06,
      "model_forward_time": 2.845566511154175,
      "step": 82312
    },
    {
      "epoch": 7.92694091796875e-06,
      "step": 82312,
      "training_step_time": 7.881300210952759
    },
    {
      "epoch": 7.927894592285157e-06,
      "model_forward_time": 1.772214412689209,
      "step": 82313
    },
    {
      "epoch": 7.927894592285157e-06,
      "step": 82313,
      "training_step_time": 9.649258613586426
    },
    {
      "epoch": 7.928848266601563e-06,
      "model_forward_time": 1.7807648181915283,
      "step": 82314
    },
    {
      "epoch": 7.928848266601563e-06,
      "step": 82314,
      "training_step_time": 8.851197957992554
    },
    {
      "epoch": 7.929801940917968e-06,
      "model_forward_time": 1.7814230918884277,
      "step": 82315
    },
    {
      "epoch": 7.929801940917968e-06,
      "step": 82315,
      "training_step_time": 7.8523383140563965
    },
    {
      "epoch": 7.930755615234376e-06,
      "model_forward_time": 1.7759945392608643,
      "step": 82316
    },
    {
      "epoch": 7.930755615234376e-06,
      "step": 82316,
      "training_step_time": 6.891715049743652
    },
    {
      "epoch": 7.931709289550781e-06,
      "model_forward_time": 1.7688367366790771,
      "step": 82317
    },
    {
      "epoch": 7.931709289550781e-06,
      "step": 82317,
      "training_step_time": 8.169443130493164
    },
    {
      "epoch": 7.932662963867187e-06,
      "model_forward_time": 1.7675676345825195,
      "step": 82318
    },
    {
      "epoch": 7.932662963867187e-06,
      "step": 82318,
      "training_step_time": 7.46476411819458
    },
    {
      "epoch": 7.933616638183595e-06,
      "model_forward_time": 1.7581157684326172,
      "step": 82319
    },
    {
      "epoch": 7.933616638183595e-06,
      "step": 82319,
      "training_step_time": 7.762789964675903
    },
    {
      "epoch": 7.9345703125e-06,
      "grad_norm": 0.09973876923322678,
      "learning_rate": 8.3061145060989e-07,
      "loss": 0.0576,
      "step": 82320
    },
    {
      "epoch": 7.9345703125e-06,
      "model_forward_time": 1.7624163627624512,
      "step": 82320
    },
    {
      "dynamics_loss_avg": 0.04645832050591707,
      "epoch": 7.9345703125e-06,
      "step": 82320
    },
    {
      "action_loss_avg": 0.0019005735754035412,
      "epoch": 7.9345703125e-06,
      "step": 82320
    },
    {
      "epoch": 7.9345703125e-06,
      "step": 82320,
      "training_step_time": 7.880211353302002
    },
    {
      "epoch": 7.935523986816406e-06,
      "model_forward_time": 1.7629477977752686,
      "step": 82321
    },
    {
      "epoch": 7.935523986816406e-06,
      "step": 82321,
      "training_step_time": 9.357900142669678
    },
    {
      "epoch": 7.936477661132812e-06,
      "model_forward_time": 1.7647953033447266,
      "step": 82322
    },
    {
      "epoch": 7.936477661132812e-06,
      "step": 82322,
      "training_step_time": 9.233981609344482
    },
    {
      "epoch": 7.937431335449219e-06,
      "model_forward_time": 1.7596652507781982,
      "step": 82323
    },
    {
      "epoch": 7.937431335449219e-06,
      "step": 82323,
      "training_step_time": 8.533190727233887
    },
    {
      "epoch": 7.938385009765625e-06,
      "model_forward_time": 1.7685365676879883,
      "step": 82324
    },
    {
      "epoch": 7.938385009765625e-06,
      "step": 82324,
      "training_step_time": 7.792961359024048
    },
    {
      "epoch": 7.93933868408203e-06,
      "model_forward_time": 1.7621309757232666,
      "step": 82325
    },
    {
      "epoch": 7.93933868408203e-06,
      "step": 82325,
      "training_step_time": 7.46105694770813
    },
    {
      "epoch": 7.940292358398438e-06,
      "model_forward_time": 1.757366418838501,
      "step": 82326
    },
    {
      "epoch": 7.940292358398438e-06,
      "step": 82326,
      "training_step_time": 7.523169040679932
    },
    {
      "epoch": 7.941246032714844e-06,
      "model_forward_time": 1.7769157886505127,
      "step": 82327
    },
    {
      "epoch": 7.941246032714844e-06,
      "step": 82327,
      "training_step_time": 7.465804576873779
    },
    {
      "epoch": 7.94219970703125e-06,
      "model_forward_time": 1.763049602508545,
      "step": 82328
    },
    {
      "epoch": 7.94219970703125e-06,
      "step": 82328,
      "training_step_time": 7.738134145736694
    },
    {
      "epoch": 7.943153381347657e-06,
      "model_forward_time": 1.7671711444854736,
      "step": 82329
    },
    {
      "epoch": 7.943153381347657e-06,
      "step": 82329,
      "training_step_time": 9.400468587875366
    },
    {
      "epoch": 7.944107055664063e-06,
      "grad_norm": 0.1006280705332756,
      "learning_rate": 8.296990471941807e-07,
      "loss": 0.0589,
      "step": 82330
    },
    {
      "epoch": 7.944107055664063e-06,
      "model_forward_time": 1.7444303035736084,
      "step": 82330
    },
    {
      "dynamics_loss_avg": 0.05199633538722992,
      "epoch": 7.944107055664063e-06,
      "step": 82330
    },
    {
      "action_loss_avg": 0.0011081618315074593,
      "epoch": 7.944107055664063e-06,
      "step": 82330
    },
    {
      "epoch": 7.944107055664063e-06,
      "step": 82330,
      "training_step_time": 8.719949007034302
    },
    {
      "epoch": 7.945060729980468e-06,
      "model_forward_time": 2.854198455810547,
      "step": 82331
    },
    {
      "epoch": 7.945060729980468e-06,
      "step": 82331,
      "training_step_time": 9.858954668045044
    },
    {
      "epoch": 7.946014404296876e-06,
      "model_forward_time": 1.7561187744140625,
      "step": 82332
    },
    {
      "epoch": 7.946014404296876e-06,
      "step": 82332,
      "training_step_time": 7.934980869293213
    },
    {
      "epoch": 7.946968078613281e-06,
      "model_forward_time": 1.7544479370117188,
      "step": 82333
    },
    {
      "epoch": 7.946968078613281e-06,
      "step": 82333,
      "training_step_time": 7.802458047866821
    },
    {
      "epoch": 7.947921752929687e-06,
      "model_forward_time": 1.7580103874206543,
      "step": 82334
    },
    {
      "epoch": 7.947921752929687e-06,
      "step": 82334,
      "training_step_time": 7.756184101104736
    },
    {
      "epoch": 7.948875427246095e-06,
      "model_forward_time": 1.7680423259735107,
      "step": 82335
    },
    {
      "epoch": 7.948875427246095e-06,
      "step": 82335,
      "training_step_time": 7.614268064498901
    },
    {
      "epoch": 7.9498291015625e-06,
      "model_forward_time": 1.7632436752319336,
      "step": 82336
    },
    {
      "epoch": 7.9498291015625e-06,
      "step": 82336,
      "training_step_time": 7.517350196838379
    },
    {
      "epoch": 7.950782775878906e-06,
      "model_forward_time": 1.7898118495941162,
      "step": 82337
    },
    {
      "epoch": 7.950782775878906e-06,
      "step": 82337,
      "training_step_time": 8.388895750045776
    },
    {
      "epoch": 7.951736450195312e-06,
      "model_forward_time": 1.7658367156982422,
      "step": 82338
    },
    {
      "epoch": 7.951736450195312e-06,
      "step": 82338,
      "training_step_time": 10.466536521911621
    },
    {
      "epoch": 7.952690124511719e-06,
      "model_forward_time": 1.7645294666290283,
      "step": 82339
    },
    {
      "epoch": 7.952690124511719e-06,
      "step": 82339,
      "training_step_time": 9.895472049713135
    },
    {
      "epoch": 7.953643798828125e-06,
      "grad_norm": 0.07732287794351578,
      "learning_rate": 8.287870998363112e-07,
      "loss": 0.0579,
      "step": 82340
    },
    {
      "epoch": 7.953643798828125e-06,
      "model_forward_time": 2.867220401763916,
      "step": 82340
    },
    {
      "dynamics_loss_avg": 0.0622922582551837,
      "epoch": 7.953643798828125e-06,
      "step": 82340
    },
    {
      "action_loss_avg": 0.0014270695275627077,
      "epoch": 7.953643798828125e-06,
      "step": 82340
    },
    {
      "epoch": 7.953643798828125e-06,
      "step": 82340,
      "training_step_time": 9.293543100357056
    },
    {
      "epoch": 7.95459747314453e-06,
      "model_forward_time": 1.7474031448364258,
      "step": 82341
    },
    {
      "epoch": 7.95459747314453e-06,
      "step": 82341,
      "training_step_time": 8.402126550674438
    },
    {
      "epoch": 7.955551147460938e-06,
      "model_forward_time": 1.7634344100952148,
      "step": 82342
    },
    {
      "epoch": 7.955551147460938e-06,
      "step": 82342,
      "training_step_time": 7.904830455780029
    },
    {
      "epoch": 7.956504821777344e-06,
      "model_forward_time": 1.7544255256652832,
      "step": 82343
    },
    {
      "epoch": 7.956504821777344e-06,
      "step": 82343,
      "training_step_time": 7.799104928970337
    },
    {
      "epoch": 7.95745849609375e-06,
      "model_forward_time": 1.7591607570648193,
      "step": 82344
    },
    {
      "epoch": 7.95745849609375e-06,
      "step": 82344,
      "training_step_time": 7.552603483200073
    },
    {
      "epoch": 7.958412170410157e-06,
      "model_forward_time": 1.7651739120483398,
      "step": 82345
    },
    {
      "epoch": 7.958412170410157e-06,
      "step": 82345,
      "training_step_time": 7.44081974029541
    },
    {
      "epoch": 7.959365844726563e-06,
      "model_forward_time": 1.78568434715271,
      "step": 82346
    },
    {
      "epoch": 7.959365844726563e-06,
      "step": 82346,
      "training_step_time": 7.508594274520874
    },
    {
      "epoch": 7.960319519042968e-06,
      "model_forward_time": 1.7593886852264404,
      "step": 82347
    },
    {
      "epoch": 7.960319519042968e-06,
      "step": 82347,
      "training_step_time": 7.803410053253174
    },
    {
      "epoch": 7.961273193359376e-06,
      "model_forward_time": 1.7589094638824463,
      "step": 82348
    },
    {
      "epoch": 7.961273193359376e-06,
      "step": 82348,
      "training_step_time": 8.890641689300537
    },
    {
      "epoch": 7.962226867675781e-06,
      "model_forward_time": 2.8511271476745605,
      "step": 82349
    },
    {
      "epoch": 7.962226867675781e-06,
      "step": 82349,
      "training_step_time": 7.939854860305786
    },
    {
      "epoch": 7.963180541992187e-06,
      "grad_norm": 0.10677490383386612,
      "learning_rate": 8.278756086360157e-07,
      "loss": 0.0586,
      "step": 82350
    },
    {
      "epoch": 7.963180541992187e-06,
      "model_forward_time": 1.7557497024536133,
      "step": 82350
    },
    {
      "dynamics_loss_avg": 0.04117416152730584,
      "epoch": 7.963180541992187e-06,
      "step": 82350
    },
    {
      "action_loss_avg": 0.0007349315332248807,
      "epoch": 7.963180541992187e-06,
      "step": 82350
    },
    {
      "epoch": 7.963180541992187e-06,
      "step": 82350,
      "training_step_time": 8.401464462280273
    },
    {
      "epoch": 7.964134216308595e-06,
      "model_forward_time": 1.7683510780334473,
      "step": 82351
    },
    {
      "epoch": 7.964134216308595e-06,
      "step": 82351,
      "training_step_time": 8.004664897918701
    },
    {
      "epoch": 7.965087890625e-06,
      "model_forward_time": 1.7569801807403564,
      "step": 82352
    },
    {
      "epoch": 7.965087890625e-06,
      "step": 82352,
      "training_step_time": 7.746707201004028
    },
    {
      "epoch": 7.966041564941406e-06,
      "model_forward_time": 1.7604084014892578,
      "step": 82353
    },
    {
      "epoch": 7.966041564941406e-06,
      "step": 82353,
      "training_step_time": 7.562982559204102
    },
    {
      "epoch": 7.966995239257812e-06,
      "model_forward_time": 1.7553064823150635,
      "step": 82354
    },
    {
      "epoch": 7.966995239257812e-06,
      "step": 82354,
      "training_step_time": 7.352278232574463
    },
    {
      "epoch": 7.96794891357422e-06,
      "model_forward_time": 1.7881276607513428,
      "step": 82355
    },
    {
      "epoch": 7.96794891357422e-06,
      "step": 82355,
      "training_step_time": 7.417844772338867
    },
    {
      "epoch": 7.968902587890625e-06,
      "model_forward_time": 1.7599327564239502,
      "step": 82356
    },
    {
      "epoch": 7.968902587890625e-06,
      "step": 82356,
      "training_step_time": 7.804039478302002
    },
    {
      "epoch": 7.96985626220703e-06,
      "model_forward_time": 1.7644538879394531,
      "step": 82357
    },
    {
      "epoch": 7.96985626220703e-06,
      "step": 82357,
      "training_step_time": 9.000752687454224
    },
    {
      "epoch": 7.970809936523438e-06,
      "model_forward_time": 2.881066083908081,
      "step": 82358
    },
    {
      "epoch": 7.970809936523438e-06,
      "step": 82358,
      "training_step_time": 10.443792343139648
    },
    {
      "epoch": 7.971763610839844e-06,
      "model_forward_time": 1.7583870887756348,
      "step": 82359
    },
    {
      "epoch": 7.971763610839844e-06,
      "step": 82359,
      "training_step_time": 8.45427942276001
    },
    {
      "epoch": 7.97271728515625e-06,
      "grad_norm": 0.10330206900835037,
      "learning_rate": 8.269645736929693e-07,
      "loss": 0.0592,
      "step": 82360
    },
    {
      "epoch": 7.97271728515625e-06,
      "model_forward_time": 1.7577273845672607,
      "step": 82360
    },
    {
      "dynamics_loss_avg": 0.06029635202139616,
      "epoch": 7.97271728515625e-06,
      "step": 82360
    },
    {
      "action_loss_avg": 0.0018686905328650027,
      "epoch": 7.97271728515625e-06,
      "step": 82360
    },
    {
      "epoch": 7.97271728515625e-06,
      "step": 82360,
      "training_step_time": 8.62537169456482
    },
    {
      "epoch": 7.973670959472657e-06,
      "model_forward_time": 1.7619423866271973,
      "step": 82361
    },
    {
      "epoch": 7.973670959472657e-06,
      "step": 82361,
      "training_step_time": 7.7411158084869385
    },
    {
      "epoch": 7.974624633789063e-06,
      "model_forward_time": 1.7716715335845947,
      "step": 82362
    },
    {
      "epoch": 7.974624633789063e-06,
      "step": 82362,
      "training_step_time": 7.8620445728302
    },
    {
      "epoch": 7.975578308105468e-06,
      "model_forward_time": 1.7566611766815186,
      "step": 82363
    },
    {
      "epoch": 7.975578308105468e-06,
      "step": 82363,
      "training_step_time": 7.203185558319092
    },
    {
      "epoch": 7.976531982421876e-06,
      "model_forward_time": 1.7623870372772217,
      "step": 82364
    },
    {
      "epoch": 7.976531982421876e-06,
      "step": 82364,
      "training_step_time": 7.3705971240997314
    },
    {
      "epoch": 7.977485656738281e-06,
      "model_forward_time": 1.7655599117279053,
      "step": 82365
    },
    {
      "epoch": 7.977485656738281e-06,
      "step": 82365,
      "training_step_time": 8.765896320343018
    },
    {
      "epoch": 7.978439331054687e-06,
      "model_forward_time": 1.772775650024414,
      "step": 82366
    },
    {
      "epoch": 7.978439331054687e-06,
      "step": 82366,
      "training_step_time": 8.4674973487854
    },
    {
      "epoch": 7.979393005371095e-06,
      "model_forward_time": 2.826138734817505,
      "step": 82367
    },
    {
      "epoch": 7.979393005371095e-06,
      "step": 82367,
      "training_step_time": 10.108295917510986
    },
    {
      "epoch": 7.9803466796875e-06,
      "model_forward_time": 1.8026540279388428,
      "step": 82368
    },
    {
      "epoch": 7.9803466796875e-06,
      "step": 82368,
      "training_step_time": 9.752197980880737
    },
    {
      "epoch": 7.981300354003906e-06,
      "model_forward_time": 1.772247314453125,
      "step": 82369
    },
    {
      "epoch": 7.981300354003906e-06,
      "step": 82369,
      "training_step_time": 7.810478210449219
    },
    {
      "epoch": 7.982254028320312e-06,
      "grad_norm": 0.0907399132847786,
      "learning_rate": 8.260539951068047e-07,
      "loss": 0.0581,
      "step": 82370
    },
    {
      "epoch": 7.982254028320312e-06,
      "model_forward_time": 1.7659351825714111,
      "step": 82370
    },
    {
      "dynamics_loss_avg": 0.07073103226721286,
      "epoch": 7.982254028320312e-06,
      "step": 82370
    },
    {
      "action_loss_avg": 0.0016296339221298695,
      "epoch": 7.982254028320312e-06,
      "step": 82370
    },
    {
      "epoch": 7.982254028320312e-06,
      "step": 82370,
      "training_step_time": 7.995863914489746
    },
    {
      "epoch": 7.98320770263672e-06,
      "model_forward_time": 1.773075819015503,
      "step": 82371
    },
    {
      "epoch": 7.98320770263672e-06,
      "step": 82371,
      "training_step_time": 7.888753175735474
    },
    {
      "epoch": 7.984161376953125e-06,
      "model_forward_time": 1.759429693222046,
      "step": 82372
    },
    {
      "epoch": 7.984161376953125e-06,
      "step": 82372,
      "training_step_time": 7.181126594543457
    },
    {
      "epoch": 7.98511505126953e-06,
      "model_forward_time": 1.781599760055542,
      "step": 82373
    },
    {
      "epoch": 7.98511505126953e-06,
      "step": 82373,
      "training_step_time": 7.391430616378784
    },
    {
      "epoch": 7.986068725585938e-06,
      "model_forward_time": 1.7720625400543213,
      "step": 82374
    },
    {
      "epoch": 7.986068725585938e-06,
      "step": 82374,
      "training_step_time": 7.729779005050659
    },
    {
      "epoch": 7.987022399902344e-06,
      "model_forward_time": 1.7696421146392822,
      "step": 82375
    },
    {
      "epoch": 7.987022399902344e-06,
      "step": 82375,
      "training_step_time": 7.698749780654907
    },
    {
      "epoch": 7.98797607421875e-06,
      "model_forward_time": 1.7470676898956299,
      "step": 82376
    },
    {
      "epoch": 7.98797607421875e-06,
      "step": 82376,
      "training_step_time": 8.488457441329956
    },
    {
      "epoch": 7.988929748535157e-06,
      "model_forward_time": 2.8716659545898438,
      "step": 82377
    },
    {
      "epoch": 7.988929748535157e-06,
      "step": 82377,
      "training_step_time": 8.907680988311768
    },
    {
      "epoch": 7.989883422851563e-06,
      "model_forward_time": 1.0616674423217773,
      "step": 82378
    },
    {
      "epoch": 7.989883422851563e-06,
      "step": 82378,
      "training_step_time": 8.271590232849121
    },
    {
      "epoch": 7.990837097167968e-06,
      "model_forward_time": 1.7684569358825684,
      "step": 82379
    },
    {
      "epoch": 7.990837097167968e-06,
      "step": 82379,
      "training_step_time": 7.999850749969482
    },
    {
      "epoch": 7.991790771484376e-06,
      "grad_norm": 0.08838659524917603,
      "learning_rate": 8.251438729770983e-07,
      "loss": 0.0593,
      "step": 82380
    },
    {
      "epoch": 7.991790771484376e-06,
      "model_forward_time": 1.4015073776245117,
      "step": 82380
    },
    {
      "dynamics_loss_avg": 0.0712609339505434,
      "epoch": 7.991790771484376e-06,
      "step": 82380
    },
    {
      "action_loss_avg": 0.001610724232159555,
      "epoch": 7.991790771484376e-06,
      "step": 82380
    },
    {
      "epoch": 7.991790771484376e-06,
      "step": 82380,
      "training_step_time": 7.811384201049805
    },
    {
      "epoch": 7.992744445800781e-06,
      "model_forward_time": 1.7654635906219482,
      "step": 82381
    },
    {
      "epoch": 7.992744445800781e-06,
      "step": 82381,
      "training_step_time": 7.46607232093811
    },
    {
      "epoch": 7.993698120117187e-06,
      "model_forward_time": 1.7704474925994873,
      "step": 82382
    },
    {
      "epoch": 7.993698120117187e-06,
      "step": 82382,
      "training_step_time": 8.002966403961182
    },
    {
      "epoch": 7.994651794433593e-06,
      "model_forward_time": 1.3962178230285645,
      "step": 82383
    },
    {
      "epoch": 7.994651794433593e-06,
      "step": 82383,
      "training_step_time": 7.797628164291382
    },
    {
      "epoch": 7.99560546875e-06,
      "model_forward_time": 1.7732148170471191,
      "step": 82384
    },
    {
      "epoch": 7.99560546875e-06,
      "step": 82384,
      "training_step_time": 9.399368047714233
    },
    {
      "epoch": 7.996559143066406e-06,
      "model_forward_time": 1.7722337245941162,
      "step": 82385
    },
    {
      "epoch": 7.996559143066406e-06,
      "step": 82385,
      "training_step_time": 9.755295276641846
    },
    {
      "epoch": 7.997512817382812e-06,
      "model_forward_time": 2.8221075534820557,
      "step": 82386
    },
    {
      "epoch": 7.997512817382812e-06,
      "step": 82386,
      "training_step_time": 8.04067587852478
    },
    {
      "epoch": 7.99846649169922e-06,
      "model_forward_time": 1.7712697982788086,
      "step": 82387
    },
    {
      "epoch": 7.99846649169922e-06,
      "step": 82387,
      "training_step_time": 8.477687358856201
    },
    {
      "epoch": 7.999420166015625e-06,
      "model_forward_time": 1.397714614868164,
      "step": 82388
    },
    {
      "epoch": 7.999420166015625e-06,
      "step": 82388,
      "training_step_time": 7.834111928939819
    },
    {
      "epoch": 8.00037384033203e-06,
      "model_forward_time": 1.7555840015411377,
      "step": 82389
    },
    {
      "epoch": 8.00037384033203e-06,
      "step": 82389,
      "training_step_time": 7.60043478012085
    },
    {
      "epoch": 8.001327514648438e-06,
      "grad_norm": 0.07449569553136826,
      "learning_rate": 8.242342074033827e-07,
      "loss": 0.0603,
      "step": 82390
    },
    {
      "epoch": 8.001327514648438e-06,
      "model_forward_time": 1.768198013305664,
      "step": 82390
    },
    {
      "dynamics_loss_avg": 0.06258940417319536,
      "epoch": 8.001327514648438e-06,
      "step": 82390
    },
    {
      "action_loss_avg": 0.001428574594319798,
      "epoch": 8.001327514648438e-06,
      "step": 82390
    },
    {
      "epoch": 8.001327514648438e-06,
      "step": 82390,
      "training_step_time": 7.590057373046875
    },
    {
      "epoch": 8.002281188964844e-06,
      "model_forward_time": 1.0551645755767822,
      "step": 82391
    },
    {
      "epoch": 8.002281188964844e-06,
      "step": 82391,
      "training_step_time": 7.3778932094573975
    },
    {
      "epoch": 8.00323486328125e-06,
      "model_forward_time": 1.776038646697998,
      "step": 82392
    },
    {
      "epoch": 8.00323486328125e-06,
      "step": 82392,
      "training_step_time": 7.4690093994140625
    },
    {
      "epoch": 8.004188537597657e-06,
      "model_forward_time": 1.7687132358551025,
      "step": 82393
    },
    {
      "epoch": 8.004188537597657e-06,
      "step": 82393,
      "training_step_time": 7.805917024612427
    },
    {
      "epoch": 8.005142211914063e-06,
      "model_forward_time": 1.770216703414917,
      "step": 82394
    },
    {
      "epoch": 8.005142211914063e-06,
      "step": 82394,
      "training_step_time": 7.880019664764404
    },
    {
      "epoch": 8.006095886230468e-06,
      "model_forward_time": 1.7908825874328613,
      "step": 82395
    },
    {
      "epoch": 8.006095886230468e-06,
      "step": 82395,
      "training_step_time": 8.857106924057007
    },
    {
      "epoch": 8.007049560546876e-06,
      "model_forward_time": 2.870861768722534,
      "step": 82396
    },
    {
      "epoch": 8.007049560546876e-06,
      "step": 82396,
      "training_step_time": 8.628249168395996
    },
    {
      "epoch": 8.008003234863281e-06,
      "model_forward_time": 1.7700293064117432,
      "step": 82397
    },
    {
      "epoch": 8.008003234863281e-06,
      "step": 82397,
      "training_step_time": 7.805659770965576
    },
    {
      "epoch": 8.008956909179687e-06,
      "model_forward_time": 1.7592933177947998,
      "step": 82398
    },
    {
      "epoch": 8.008956909179687e-06,
      "step": 82398,
      "training_step_time": 7.940663576126099
    },
    {
      "epoch": 8.009910583496093e-06,
      "model_forward_time": 1.7756972312927246,
      "step": 82399
    },
    {
      "epoch": 8.009910583496093e-06,
      "step": 82399,
      "training_step_time": 7.75046181678772
    },
    {
      "epoch": 8.0108642578125e-06,
      "grad_norm": 0.1074344590306282,
      "learning_rate": 8.233249984851344e-07,
      "loss": 0.0585,
      "step": 82400
    },
    {
      "epoch": 8.0108642578125e-06,
      "model_forward_time": 1.771406650543213,
      "step": 82400
    },
    {
      "dynamics_loss_avg": 0.06929139010608196,
      "epoch": 8.0108642578125e-06,
      "step": 82400
    },
    {
      "action_loss_avg": 0.0014453958952799439,
      "epoch": 8.0108642578125e-06,
      "step": 82400
    },
    {
      "epoch": 8.0108642578125e-06,
      "step": 82400,
      "training_step_time": 7.285090208053589
    },
    {
      "epoch": 8.011817932128906e-06,
      "model_forward_time": 1.785658836364746,
      "step": 82401
    },
    {
      "epoch": 8.011817932128906e-06,
      "step": 82401,
      "training_step_time": 7.369596004486084
    },
    {
      "epoch": 8.012771606445312e-06,
      "model_forward_time": 1.7726614475250244,
      "step": 82402
    },
    {
      "epoch": 8.012771606445312e-06,
      "step": 82402,
      "training_step_time": 7.793562889099121
    },
    {
      "epoch": 8.01372528076172e-06,
      "model_forward_time": 1.7607996463775635,
      "step": 82403
    },
    {
      "epoch": 8.01372528076172e-06,
      "step": 82403,
      "training_step_time": 9.259267091751099
    },
    {
      "epoch": 8.014678955078125e-06,
      "model_forward_time": 1.7717502117156982,
      "step": 82404
    },
    {
      "epoch": 8.014678955078125e-06,
      "step": 82404,
      "training_step_time": 10.00317096710205
    },
    {
      "epoch": 8.01563262939453e-06,
      "model_forward_time": 2.8630032539367676,
      "step": 82405
    },
    {
      "epoch": 8.01563262939453e-06,
      "step": 82405,
      "training_step_time": 7.836859703063965
    },
    {
      "epoch": 8.016586303710938e-06,
      "model_forward_time": 1.7670540809631348,
      "step": 82406
    },
    {
      "epoch": 8.016586303710938e-06,
      "step": 82406,
      "training_step_time": 9.147045373916626
    },
    {
      "epoch": 8.017539978027344e-06,
      "model_forward_time": 1.7600080966949463,
      "step": 82407
    },
    {
      "epoch": 8.017539978027344e-06,
      "step": 82407,
      "training_step_time": 8.70589303970337
    },
    {
      "epoch": 8.01849365234375e-06,
      "model_forward_time": 1.768167495727539,
      "step": 82408
    },
    {
      "epoch": 8.01849365234375e-06,
      "step": 82408,
      "training_step_time": 7.61544132232666
    },
    {
      "epoch": 8.019447326660157e-06,
      "model_forward_time": 1.7556469440460205,
      "step": 82409
    },
    {
      "epoch": 8.019447326660157e-06,
      "step": 82409,
      "training_step_time": 7.387885093688965
    },
    {
      "epoch": 8.020401000976563e-06,
      "grad_norm": 0.09020417183637619,
      "learning_rate": 8.22416246321785e-07,
      "loss": 0.0585,
      "step": 82410
    },
    {
      "epoch": 8.020401000976563e-06,
      "model_forward_time": 1.7705910205841064,
      "step": 82410
    },
    {
      "dynamics_loss_avg": 0.0534073181450367,
      "epoch": 8.020401000976563e-06,
      "step": 82410
    },
    {
      "action_loss_avg": 0.0015754596213810145,
      "epoch": 8.020401000976563e-06,
      "step": 82410
    },
    {
      "epoch": 8.020401000976563e-06,
      "step": 82410,
      "training_step_time": 7.421850681304932
    },
    {
      "epoch": 8.021354675292968e-06,
      "model_forward_time": 1.7598397731781006,
      "step": 82411
    },
    {
      "epoch": 8.021354675292968e-06,
      "step": 82411,
      "training_step_time": 7.779010772705078
    },
    {
      "epoch": 8.022308349609376e-06,
      "model_forward_time": 1.7572722434997559,
      "step": 82412
    },
    {
      "epoch": 8.022308349609376e-06,
      "step": 82412,
      "training_step_time": 8.559317350387573
    },
    {
      "epoch": 8.023262023925781e-06,
      "model_forward_time": 1.7519676685333252,
      "step": 82413
    },
    {
      "epoch": 8.023262023925781e-06,
      "step": 82413,
      "training_step_time": 9.839516162872314
    },
    {
      "epoch": 8.024215698242187e-06,
      "model_forward_time": 1.7547016143798828,
      "step": 82414
    },
    {
      "epoch": 8.024215698242187e-06,
      "step": 82414,
      "training_step_time": 9.664308786392212
    },
    {
      "epoch": 8.025169372558593e-06,
      "model_forward_time": 1.751967430114746,
      "step": 82415
    },
    {
      "epoch": 8.025169372558593e-06,
      "step": 82415,
      "training_step_time": 7.8405845165252686
    },
    {
      "epoch": 8.026123046875e-06,
      "model_forward_time": 1.0670695304870605,
      "step": 82416
    },
    {
      "epoch": 8.026123046875e-06,
      "step": 82416,
      "training_step_time": 7.9661500453948975
    },
    {
      "epoch": 8.027076721191406e-06,
      "model_forward_time": 1.761423110961914,
      "step": 82417
    },
    {
      "epoch": 8.027076721191406e-06,
      "step": 82417,
      "training_step_time": 7.428800582885742
    },
    {
      "epoch": 8.028030395507812e-06,
      "model_forward_time": 1.7565715312957764,
      "step": 82418
    },
    {
      "epoch": 8.028030395507812e-06,
      "step": 82418,
      "training_step_time": 7.715563535690308
    },
    {
      "epoch": 8.02898406982422e-06,
      "model_forward_time": 1.766798734664917,
      "step": 82419
    },
    {
      "epoch": 8.02898406982422e-06,
      "step": 82419,
      "training_step_time": 7.364076137542725
    },
    {
      "epoch": 8.029937744140625e-06,
      "grad_norm": 0.10845977813005447,
      "learning_rate": 8.215079510127139e-07,
      "loss": 0.0581,
      "step": 82420
    },
    {
      "epoch": 8.029937744140625e-06,
      "model_forward_time": 1.7588191032409668,
      "step": 82420
    },
    {
      "dynamics_loss_avg": 0.05619228817522526,
      "epoch": 8.029937744140625e-06,
      "step": 82420
    },
    {
      "action_loss_avg": 0.0013771172380074859,
      "epoch": 8.029937744140625e-06,
      "step": 82420
    },
    {
      "epoch": 8.029937744140625e-06,
      "step": 82420,
      "training_step_time": 7.74522590637207
    },
    {
      "epoch": 8.03089141845703e-06,
      "model_forward_time": 1.7556579113006592,
      "step": 82421
    },
    {
      "epoch": 8.03089141845703e-06,
      "step": 82421,
      "training_step_time": 7.716000318527222
    },
    {
      "epoch": 8.031845092773438e-06,
      "model_forward_time": 1.771101474761963,
      "step": 82422
    },
    {
      "epoch": 8.031845092773438e-06,
      "step": 82422,
      "training_step_time": 8.410847187042236
    },
    {
      "epoch": 8.032798767089844e-06,
      "model_forward_time": 1.7883894443511963,
      "step": 82423
    },
    {
      "epoch": 8.032798767089844e-06,
      "step": 82423,
      "training_step_time": 7.794113636016846
    },
    {
      "epoch": 8.03375244140625e-06,
      "model_forward_time": 2.867746353149414,
      "step": 82424
    },
    {
      "epoch": 8.03375244140625e-06,
      "step": 82424,
      "training_step_time": 7.780789852142334
    },
    {
      "epoch": 8.034706115722657e-06,
      "model_forward_time": 1.764941930770874,
      "step": 82425
    },
    {
      "epoch": 8.034706115722657e-06,
      "step": 82425,
      "training_step_time": 8.854812622070312
    },
    {
      "epoch": 8.035659790039063e-06,
      "model_forward_time": 1.7633769512176514,
      "step": 82426
    },
    {
      "epoch": 8.035659790039063e-06,
      "step": 82426,
      "training_step_time": 8.25970983505249
    },
    {
      "epoch": 8.036613464355468e-06,
      "model_forward_time": 1.7692031860351562,
      "step": 82427
    },
    {
      "epoch": 8.036613464355468e-06,
      "step": 82427,
      "training_step_time": 7.332895278930664
    },
    {
      "epoch": 8.037567138671876e-06,
      "model_forward_time": 1.7730791568756104,
      "step": 82428
    },
    {
      "epoch": 8.037567138671876e-06,
      "step": 82428,
      "training_step_time": 7.721114873886108
    },
    {
      "epoch": 8.038520812988282e-06,
      "model_forward_time": 1.7732181549072266,
      "step": 82429
    },
    {
      "epoch": 8.038520812988282e-06,
      "step": 82429,
      "training_step_time": 8.385249376296997
    },
    {
      "epoch": 8.039474487304687e-06,
      "grad_norm": 0.09894471615552902,
      "learning_rate": 8.206001126572522e-07,
      "loss": 0.0582,
      "step": 82430
    },
    {
      "epoch": 8.039474487304687e-06,
      "model_forward_time": 1.7771825790405273,
      "step": 82430
    },
    {
      "dynamics_loss_avg": 0.0804071843624115,
      "epoch": 8.039474487304687e-06,
      "step": 82430
    },
    {
      "action_loss_avg": 0.002757975994609296,
      "epoch": 8.039474487304687e-06,
      "step": 82430
    },
    {
      "epoch": 8.039474487304687e-06,
      "step": 82430,
      "training_step_time": 8.357024908065796
    },
    {
      "epoch": 8.040428161621093e-06,
      "model_forward_time": 1.767590045928955,
      "step": 82431
    },
    {
      "epoch": 8.040428161621093e-06,
      "step": 82431,
      "training_step_time": 10.254173278808594
    },
    {
      "epoch": 8.0413818359375e-06,
      "model_forward_time": 1.8654911518096924,
      "step": 82432
    },
    {
      "epoch": 8.0413818359375e-06,
      "step": 82432,
      "training_step_time": 9.290576457977295
    },
    {
      "epoch": 8.042335510253906e-06,
      "model_forward_time": 2.843585729598999,
      "step": 82433
    },
    {
      "epoch": 8.042335510253906e-06,
      "step": 82433,
      "training_step_time": 7.84831166267395
    },
    {
      "epoch": 8.043289184570312e-06,
      "model_forward_time": 1.7735509872436523,
      "step": 82434
    },
    {
      "epoch": 8.043289184570312e-06,
      "step": 82434,
      "training_step_time": 8.513209104537964
    },
    {
      "epoch": 8.04424285888672e-06,
      "model_forward_time": 1.7684576511383057,
      "step": 82435
    },
    {
      "epoch": 8.04424285888672e-06,
      "step": 82435,
      "training_step_time": 8.00004243850708
    },
    {
      "epoch": 8.045196533203125e-06,
      "model_forward_time": 1.7686266899108887,
      "step": 82436
    },
    {
      "epoch": 8.045196533203125e-06,
      "step": 82436,
      "training_step_time": 7.460913419723511
    },
    {
      "epoch": 8.04615020751953e-06,
      "model_forward_time": 1.773223876953125,
      "step": 82437
    },
    {
      "epoch": 8.04615020751953e-06,
      "step": 82437,
      "training_step_time": 7.739337205886841
    },
    {
      "epoch": 8.047103881835938e-06,
      "model_forward_time": 1.7647762298583984,
      "step": 82438
    },
    {
      "epoch": 8.047103881835938e-06,
      "step": 82438,
      "training_step_time": 7.756460189819336
    },
    {
      "epoch": 8.048057556152344e-06,
      "model_forward_time": 1.776778221130371,
      "step": 82439
    },
    {
      "epoch": 8.048057556152344e-06,
      "step": 82439,
      "training_step_time": 8.720364093780518
    },
    {
      "epoch": 8.04901123046875e-06,
      "grad_norm": 0.09787889569997787,
      "learning_rate": 8.196927313546776e-07,
      "loss": 0.0588,
      "step": 82440
    },
    {
      "epoch": 8.04901123046875e-06,
      "model_forward_time": 1.7690250873565674,
      "step": 82440
    },
    {
      "dynamics_loss_avg": 0.05361948125064373,
      "epoch": 8.04901123046875e-06,
      "step": 82440
    },
    {
      "action_loss_avg": 0.002048288710648194,
      "epoch": 8.04901123046875e-06,
      "step": 82440
    },
    {
      "epoch": 8.04901123046875e-06,
      "step": 82440,
      "training_step_time": 8.802542448043823
    },
    {
      "epoch": 8.049964904785157e-06,
      "model_forward_time": 1.8288006782531738,
      "step": 82441
    },
    {
      "epoch": 8.049964904785157e-06,
      "step": 82441,
      "training_step_time": 8.88900089263916
    },
    {
      "epoch": 8.050918579101563e-06,
      "model_forward_time": 2.8513071537017822,
      "step": 82442
    },
    {
      "epoch": 8.050918579101563e-06,
      "step": 82442,
      "training_step_time": 7.8307044506073
    },
    {
      "epoch": 8.051872253417968e-06,
      "model_forward_time": 1.7697103023529053,
      "step": 82443
    },
    {
      "epoch": 8.051872253417968e-06,
      "step": 82443,
      "training_step_time": 8.353744506835938
    },
    {
      "epoch": 8.052825927734376e-06,
      "model_forward_time": 1.7720351219177246,
      "step": 82444
    },
    {
      "epoch": 8.052825927734376e-06,
      "step": 82444,
      "training_step_time": 7.925434112548828
    },
    {
      "epoch": 8.053779602050782e-06,
      "model_forward_time": 1.778353214263916,
      "step": 82445
    },
    {
      "epoch": 8.053779602050782e-06,
      "step": 82445,
      "training_step_time": 7.518317461013794
    },
    {
      "epoch": 8.054733276367187e-06,
      "model_forward_time": 1.772665023803711,
      "step": 82446
    },
    {
      "epoch": 8.054733276367187e-06,
      "step": 82446,
      "training_step_time": 8.444780826568604
    },
    {
      "epoch": 8.055686950683593e-06,
      "model_forward_time": 1.7610883712768555,
      "step": 82447
    },
    {
      "epoch": 8.055686950683593e-06,
      "step": 82447,
      "training_step_time": 7.786795139312744
    },
    {
      "epoch": 8.056640625e-06,
      "model_forward_time": 1.7765283584594727,
      "step": 82448
    },
    {
      "epoch": 8.056640625e-06,
      "step": 82448,
      "training_step_time": 7.6055192947387695
    },
    {
      "epoch": 8.057594299316406e-06,
      "model_forward_time": 1.758530855178833,
      "step": 82449
    },
    {
      "epoch": 8.057594299316406e-06,
      "step": 82449,
      "training_step_time": 8.879422426223755
    },
    {
      "epoch": 8.058547973632812e-06,
      "grad_norm": 0.09013542532920837,
      "learning_rate": 8.18785807204222e-07,
      "loss": 0.0576,
      "step": 82450
    },
    {
      "epoch": 8.058547973632812e-06,
      "model_forward_time": 1.8010549545288086,
      "step": 82450
    },
    {
      "dynamics_loss_avg": 0.06282563358545304,
      "epoch": 8.058547973632812e-06,
      "step": 82450
    },
    {
      "action_loss_avg": 0.0013217231025919318,
      "epoch": 8.058547973632812e-06,
      "step": 82450
    },
    {
      "epoch": 8.058547973632812e-06,
      "step": 82450,
      "training_step_time": 9.048139572143555
    },
    {
      "epoch": 8.05950164794922e-06,
      "model_forward_time": 0.7876536846160889,
      "step": 82451
    },
    {
      "epoch": 8.05950164794922e-06,
      "step": 82451,
      "training_step_time": 9.58184003829956
    },
    {
      "epoch": 8.060455322265625e-06,
      "model_forward_time": 2.8862979412078857,
      "step": 82452
    },
    {
      "epoch": 8.060455322265625e-06,
      "step": 82452,
      "training_step_time": 7.74491024017334
    },
    {
      "epoch": 8.06140899658203e-06,
      "model_forward_time": 1.7757213115692139,
      "step": 82453
    },
    {
      "epoch": 8.06140899658203e-06,
      "step": 82453,
      "training_step_time": 7.686701774597168
    },
    {
      "epoch": 8.062362670898438e-06,
      "model_forward_time": 1.776522159576416,
      "step": 82454
    },
    {
      "epoch": 8.062362670898438e-06,
      "step": 82454,
      "training_step_time": 7.955380916595459
    },
    {
      "epoch": 8.063316345214844e-06,
      "model_forward_time": 1.7651331424713135,
      "step": 82455
    },
    {
      "epoch": 8.063316345214844e-06,
      "step": 82455,
      "training_step_time": 7.139854431152344
    },
    {
      "epoch": 8.06427001953125e-06,
      "model_forward_time": 1.7703115940093994,
      "step": 82456
    },
    {
      "epoch": 8.06427001953125e-06,
      "step": 82456,
      "training_step_time": 7.835760593414307
    },
    {
      "epoch": 8.065223693847657e-06,
      "model_forward_time": 1.408292293548584,
      "step": 82457
    },
    {
      "epoch": 8.065223693847657e-06,
      "step": 82457,
      "training_step_time": 8.237796783447266
    },
    {
      "epoch": 8.066177368164063e-06,
      "model_forward_time": 1.7776904106140137,
      "step": 82458
    },
    {
      "epoch": 8.066177368164063e-06,
      "step": 82458,
      "training_step_time": 8.593303680419922
    },
    {
      "epoch": 8.067131042480468e-06,
      "model_forward_time": 1.7775564193725586,
      "step": 82459
    },
    {
      "epoch": 8.067131042480468e-06,
      "step": 82459,
      "training_step_time": 8.034846782684326
    },
    {
      "epoch": 8.068084716796876e-06,
      "grad_norm": 0.07387176156044006,
      "learning_rate": 8.178793403050628e-07,
      "loss": 0.0579,
      "step": 82460
    },
    {
      "epoch": 8.068084716796876e-06,
      "model_forward_time": 1.771493673324585,
      "step": 82460
    },
    {
      "dynamics_loss_avg": 0.05088352989405394,
      "epoch": 8.068084716796876e-06,
      "step": 82460
    },
    {
      "action_loss_avg": 0.0016967060044407844,
      "epoch": 8.068084716796876e-06,
      "step": 82460
    },
    {
      "epoch": 8.068084716796876e-06,
      "step": 82460,
      "training_step_time": 9.6056067943573
    },
    {
      "epoch": 8.069038391113282e-06,
      "model_forward_time": 2.8709490299224854,
      "step": 82461
    },
    {
      "epoch": 8.069038391113282e-06,
      "step": 82461,
      "training_step_time": 7.8735175132751465
    },
    {
      "epoch": 8.069992065429687e-06,
      "model_forward_time": 1.7787394523620605,
      "step": 82462
    },
    {
      "epoch": 8.069992065429687e-06,
      "step": 82462,
      "training_step_time": 7.750913858413696
    },
    {
      "epoch": 8.070945739746093e-06,
      "model_forward_time": 1.7794218063354492,
      "step": 82463
    },
    {
      "epoch": 8.070945739746093e-06,
      "step": 82463,
      "training_step_time": 7.560184955596924
    },
    {
      "epoch": 8.0718994140625e-06,
      "model_forward_time": 1.7712275981903076,
      "step": 82464
    },
    {
      "epoch": 8.0718994140625e-06,
      "step": 82464,
      "training_step_time": 7.151472806930542
    },
    {
      "epoch": 8.072853088378906e-06,
      "model_forward_time": 1.7716374397277832,
      "step": 82465
    },
    {
      "epoch": 8.072853088378906e-06,
      "step": 82465,
      "training_step_time": 7.882344722747803
    },
    {
      "epoch": 8.073806762695312e-06,
      "model_forward_time": 1.7852771282196045,
      "step": 82466
    },
    {
      "epoch": 8.073806762695312e-06,
      "step": 82466,
      "training_step_time": 7.707631587982178
    },
    {
      "epoch": 8.07476043701172e-06,
      "model_forward_time": 1.7748222351074219,
      "step": 82467
    },
    {
      "epoch": 8.07476043701172e-06,
      "step": 82467,
      "training_step_time": 7.802531003952026
    },
    {
      "epoch": 8.075714111328125e-06,
      "model_forward_time": 1.7751877307891846,
      "step": 82468
    },
    {
      "epoch": 8.075714111328125e-06,
      "step": 82468,
      "training_step_time": 7.967906475067139
    },
    {
      "epoch": 8.07666778564453e-06,
      "model_forward_time": 1.8911328315734863,
      "step": 82469
    },
    {
      "epoch": 8.07666778564453e-06,
      "step": 82469,
      "training_step_time": 7.787019968032837
    },
    {
      "epoch": 8.077621459960938e-06,
      "grad_norm": 0.11882327497005463,
      "learning_rate": 8.16973330756331e-07,
      "loss": 0.0569,
      "step": 82470
    },
    {
      "epoch": 8.077621459960938e-06,
      "model_forward_time": 1.7746548652648926,
      "step": 82470
    },
    {
      "dynamics_loss_avg": 0.06643265634775161,
      "epoch": 8.077621459960938e-06,
      "step": 82470
    },
    {
      "action_loss_avg": 0.001408328884281218,
      "epoch": 8.077621459960938e-06,
      "step": 82470
    },
    {
      "epoch": 8.077621459960938e-06,
      "step": 82470,
      "training_step_time": 10.5378737449646
    },
    {
      "epoch": 8.078575134277344e-06,
      "model_forward_time": 1.7653977870941162,
      "step": 82471
    },
    {
      "epoch": 8.078575134277344e-06,
      "step": 82471,
      "training_step_time": 7.726203441619873
    },
    {
      "epoch": 8.07952880859375e-06,
      "model_forward_time": 1.7775473594665527,
      "step": 82472
    },
    {
      "epoch": 8.07952880859375e-06,
      "step": 82472,
      "training_step_time": 7.694253921508789
    },
    {
      "epoch": 8.080482482910157e-06,
      "model_forward_time": 1.7719438076019287,
      "step": 82473
    },
    {
      "epoch": 8.080482482910157e-06,
      "step": 82473,
      "training_step_time": 7.937849998474121
    },
    {
      "epoch": 8.081436157226563e-06,
      "model_forward_time": 1.7750074863433838,
      "step": 82474
    },
    {
      "epoch": 8.081436157226563e-06,
      "step": 82474,
      "training_step_time": 7.825493335723877
    },
    {
      "epoch": 8.082389831542968e-06,
      "model_forward_time": 1.7733268737792969,
      "step": 82475
    },
    {
      "epoch": 8.082389831542968e-06,
      "step": 82475,
      "training_step_time": 7.7517313957214355
    },
    {
      "epoch": 8.083343505859376e-06,
      "model_forward_time": 1.7787649631500244,
      "step": 82476
    },
    {
      "epoch": 8.083343505859376e-06,
      "step": 82476,
      "training_step_time": 8.72634506225586
    },
    {
      "epoch": 8.084297180175782e-06,
      "model_forward_time": 1.7734272480010986,
      "step": 82477
    },
    {
      "epoch": 8.084297180175782e-06,
      "step": 82477,
      "training_step_time": 9.589100360870361
    },
    {
      "epoch": 8.085250854492187e-06,
      "model_forward_time": 1.8893241882324219,
      "step": 82478
    },
    {
      "epoch": 8.085250854492187e-06,
      "step": 82478,
      "training_step_time": 8.775059461593628
    },
    {
      "epoch": 8.086204528808593e-06,
      "model_forward_time": 1.7731702327728271,
      "step": 82479
    },
    {
      "epoch": 8.086204528808593e-06,
      "step": 82479,
      "training_step_time": 7.7558698654174805
    },
    {
      "epoch": 8.087158203125e-06,
      "grad_norm": 0.0786043331027031,
      "learning_rate": 8.16067778657107e-07,
      "loss": 0.0575,
      "step": 82480
    },
    {
      "epoch": 8.087158203125e-06,
      "model_forward_time": 2.8644556999206543,
      "step": 82480
    },
    {
      "dynamics_loss_avg": 0.04681741017848253,
      "epoch": 8.087158203125e-06,
      "step": 82480
    },
    {
      "action_loss_avg": 0.0014288259233580902,
      "epoch": 8.087158203125e-06,
      "step": 82480
    },
    {
      "epoch": 8.087158203125e-06,
      "step": 82480,
      "training_step_time": 8.912427425384521
    },
    {
      "epoch": 8.088111877441406e-06,
      "model_forward_time": 1.7841029167175293,
      "step": 82481
    },
    {
      "epoch": 8.088111877441406e-06,
      "step": 82481,
      "training_step_time": 7.697349309921265
    },
    {
      "epoch": 8.089065551757812e-06,
      "model_forward_time": 1.7780063152313232,
      "step": 82482
    },
    {
      "epoch": 8.089065551757812e-06,
      "step": 82482,
      "training_step_time": 7.863329172134399
    },
    {
      "epoch": 8.09001922607422e-06,
      "model_forward_time": 1.7698774337768555,
      "step": 82483
    },
    {
      "epoch": 8.09001922607422e-06,
      "step": 82483,
      "training_step_time": 8.420023679733276
    },
    {
      "epoch": 8.090972900390625e-06,
      "model_forward_time": 1.778698444366455,
      "step": 82484
    },
    {
      "epoch": 8.090972900390625e-06,
      "step": 82484,
      "training_step_time": 7.728219509124756
    },
    {
      "epoch": 8.09192657470703e-06,
      "model_forward_time": 1.7713613510131836,
      "step": 82485
    },
    {
      "epoch": 8.09192657470703e-06,
      "step": 82485,
      "training_step_time": 8.272035360336304
    },
    {
      "epoch": 8.092880249023438e-06,
      "model_forward_time": 1.772979974746704,
      "step": 82486
    },
    {
      "epoch": 8.092880249023438e-06,
      "step": 82486,
      "training_step_time": 10.075959205627441
    },
    {
      "epoch": 8.093833923339844e-06,
      "model_forward_time": 1.8554906845092773,
      "step": 82487
    },
    {
      "epoch": 8.093833923339844e-06,
      "step": 82487,
      "training_step_time": 8.0050048828125
    },
    {
      "epoch": 8.09478759765625e-06,
      "model_forward_time": 1.7816052436828613,
      "step": 82488
    },
    {
      "epoch": 8.09478759765625e-06,
      "step": 82488,
      "training_step_time": 7.80785059928894
    },
    {
      "epoch": 8.095741271972657e-06,
      "model_forward_time": 2.8906941413879395,
      "step": 82489
    },
    {
      "epoch": 8.095741271972657e-06,
      "step": 82489,
      "training_step_time": 7.821706533432007
    },
    {
      "epoch": 8.096694946289063e-06,
      "grad_norm": 0.08129388839006424,
      "learning_rate": 8.151626841064214e-07,
      "loss": 0.0581,
      "step": 82490
    },
    {
      "epoch": 8.096694946289063e-06,
      "model_forward_time": 1.7770249843597412,
      "step": 82490
    },
    {
      "dynamics_loss_avg": 0.058856606297194955,
      "epoch": 8.096694946289063e-06,
      "step": 82490
    },
    {
      "action_loss_avg": 0.001344551972579211,
      "epoch": 8.096694946289063e-06,
      "step": 82490
    },
    {
      "epoch": 8.096694946289063e-06,
      "step": 82490,
      "training_step_time": 7.730103969573975
    },
    {
      "epoch": 8.097648620605468e-06,
      "model_forward_time": 1.7707500457763672,
      "step": 82491
    },
    {
      "epoch": 8.097648620605468e-06,
      "step": 82491,
      "training_step_time": 7.9651947021484375
    },
    {
      "epoch": 8.098602294921876e-06,
      "model_forward_time": 1.778226375579834,
      "step": 82492
    },
    {
      "epoch": 8.098602294921876e-06,
      "step": 82492,
      "training_step_time": 7.495559930801392
    },
    {
      "epoch": 8.099555969238282e-06,
      "model_forward_time": 1.7667300701141357,
      "step": 82493
    },
    {
      "epoch": 8.099555969238282e-06,
      "step": 82493,
      "training_step_time": 7.844783306121826
    },
    {
      "epoch": 8.100509643554687e-06,
      "model_forward_time": 1.7767002582550049,
      "step": 82494
    },
    {
      "epoch": 8.100509643554687e-06,
      "step": 82494,
      "training_step_time": 8.824546813964844
    },
    {
      "epoch": 8.101463317871093e-06,
      "model_forward_time": 1.7710189819335938,
      "step": 82495
    },
    {
      "epoch": 8.101463317871093e-06,
      "step": 82495,
      "training_step_time": 8.768410205841064
    },
    {
      "epoch": 8.1024169921875e-06,
      "model_forward_time": 1.8243417739868164,
      "step": 82496
    },
    {
      "epoch": 8.1024169921875e-06,
      "step": 82496,
      "training_step_time": 10.522361040115356
    },
    {
      "epoch": 8.103370666503906e-06,
      "model_forward_time": 1.7747342586517334,
      "step": 82497
    },
    {
      "epoch": 8.103370666503906e-06,
      "step": 82497,
      "training_step_time": 8.730060577392578
    },
    {
      "epoch": 8.104324340820312e-06,
      "model_forward_time": 2.934400796890259,
      "step": 82498
    },
    {
      "epoch": 8.104324340820312e-06,
      "step": 82498,
      "training_step_time": 8.391240358352661
    },
    {
      "epoch": 8.10527801513672e-06,
      "model_forward_time": 1.7842323780059814,
      "step": 82499
    },
    {
      "epoch": 8.10527801513672e-06,
      "step": 82499,
      "training_step_time": 8.551348209381104
    },
    {
      "epoch": 8.106231689453125e-06,
      "grad_norm": 0.10274147242307663,
      "learning_rate": 8.142580472032519e-07,
      "loss": 0.058,
      "step": 82500
    },
    {
      "epoch": 8.106231689453125e-06,
      "model_forward_time": 1.75876784324646,
      "step": 82500
    },
    {
      "dynamics_loss_avg": 0.07199952993541955,
      "epoch": 8.106231689453125e-06,
      "step": 82500
    },
    {
      "action_loss_avg": 0.0024888910236768426,
      "epoch": 8.106231689453125e-06,
      "step": 82500
    },
    {
      "epoch": 8.106231689453125e-06,
      "step": 82500,
      "training_step_time": 7.632514715194702
    },
    {
      "epoch": 8.10718536376953e-06,
      "model_forward_time": 1.7799954414367676,
      "step": 82501
    },
    {
      "epoch": 8.10718536376953e-06,
      "step": 82501,
      "training_step_time": 8.077093362808228
    },
    {
      "epoch": 8.108139038085938e-06,
      "model_forward_time": 1.4133925437927246,
      "step": 82502
    },
    {
      "epoch": 8.108139038085938e-06,
      "step": 82502,
      "training_step_time": 7.831084966659546
    },
    {
      "epoch": 8.109092712402344e-06,
      "model_forward_time": 1.7567369937896729,
      "step": 82503
    },
    {
      "epoch": 8.109092712402344e-06,
      "step": 82503,
      "training_step_time": 8.317193508148193
    },
    {
      "epoch": 8.11004638671875e-06,
      "model_forward_time": 1.7812495231628418,
      "step": 82504
    },
    {
      "epoch": 8.11004638671875e-06,
      "step": 82504,
      "training_step_time": 8.668249607086182
    },
    {
      "epoch": 8.111000061035157e-06,
      "model_forward_time": 1.7942454814910889,
      "step": 82505
    },
    {
      "epoch": 8.111000061035157e-06,
      "step": 82505,
      "training_step_time": 8.101463556289673
    },
    {
      "epoch": 8.111953735351563e-06,
      "model_forward_time": 1.756929636001587,
      "step": 82506
    },
    {
      "epoch": 8.111953735351563e-06,
      "step": 82506,
      "training_step_time": 7.764781951904297
    },
    {
      "epoch": 8.112907409667968e-06,
      "model_forward_time": 2.856222629547119,
      "step": 82507
    },
    {
      "epoch": 8.112907409667968e-06,
      "step": 82507,
      "training_step_time": 8.706146240234375
    },
    {
      "epoch": 8.113861083984374e-06,
      "model_forward_time": 1.773939609527588,
      "step": 82508
    },
    {
      "epoch": 8.113861083984374e-06,
      "step": 82508,
      "training_step_time": 8.479394435882568
    },
    {
      "epoch": 8.114814758300782e-06,
      "model_forward_time": 1.07975435256958,
      "step": 82509
    },
    {
      "epoch": 8.114814758300782e-06,
      "step": 82509,
      "training_step_time": 7.953561305999756
    },
    {
      "epoch": 8.115768432617187e-06,
      "grad_norm": 0.10035846382379532,
      "learning_rate": 8.133538680465286e-07,
      "loss": 0.0569,
      "step": 82510
    },
    {
      "epoch": 8.115768432617187e-06,
      "model_forward_time": 1.7834973335266113,
      "step": 82510
    },
    {
      "dynamics_loss_avg": 0.059131179191172126,
      "epoch": 8.115768432617187e-06,
      "step": 82510
    },
    {
      "action_loss_avg": 0.0014853485394269229,
      "epoch": 8.115768432617187e-06,
      "step": 82510
    },
    {
      "epoch": 8.115768432617187e-06,
      "step": 82510,
      "training_step_time": 7.448713541030884
    },
    {
      "epoch": 8.116722106933593e-06,
      "model_forward_time": 1.7663793563842773,
      "step": 82511
    },
    {
      "epoch": 8.116722106933593e-06,
      "step": 82511,
      "training_step_time": 7.779950141906738
    },
    {
      "epoch": 8.11767578125e-06,
      "model_forward_time": 1.7754101753234863,
      "step": 82512
    },
    {
      "epoch": 8.11767578125e-06,
      "step": 82512,
      "training_step_time": 7.743615627288818
    },
    {
      "epoch": 8.118629455566406e-06,
      "model_forward_time": 1.7734708786010742,
      "step": 82513
    },
    {
      "epoch": 8.118629455566406e-06,
      "step": 82513,
      "training_step_time": 7.772764444351196
    },
    {
      "epoch": 8.119583129882812e-06,
      "model_forward_time": 1.7684950828552246,
      "step": 82514
    },
    {
      "epoch": 8.119583129882812e-06,
      "step": 82514,
      "training_step_time": 10.91836667060852
    },
    {
      "epoch": 8.12053680419922e-06,
      "model_forward_time": 1.7708630561828613,
      "step": 82515
    },
    {
      "epoch": 8.12053680419922e-06,
      "step": 82515,
      "training_step_time": 9.180392980575562
    },
    {
      "epoch": 8.121490478515625e-06,
      "model_forward_time": 1.7667937278747559,
      "step": 82516
    },
    {
      "epoch": 8.121490478515625e-06,
      "step": 82516,
      "training_step_time": 8.444048643112183
    },
    {
      "epoch": 8.12244415283203e-06,
      "model_forward_time": 2.899423122406006,
      "step": 82517
    },
    {
      "epoch": 8.12244415283203e-06,
      "step": 82517,
      "training_step_time": 7.805666923522949
    },
    {
      "epoch": 8.123397827148438e-06,
      "model_forward_time": 1.7722673416137695,
      "step": 82518
    },
    {
      "epoch": 8.123397827148438e-06,
      "step": 82518,
      "training_step_time": 7.969329595565796
    },
    {
      "epoch": 8.124351501464844e-06,
      "model_forward_time": 1.7742671966552734,
      "step": 82519
    },
    {
      "epoch": 8.124351501464844e-06,
      "step": 82519,
      "training_step_time": 7.493664264678955
    },
    {
      "epoch": 8.12530517578125e-06,
      "grad_norm": 0.10560210794210434,
      "learning_rate": 8.124501467351337e-07,
      "loss": 0.0565,
      "step": 82520
    },
    {
      "epoch": 8.12530517578125e-06,
      "model_forward_time": 1.777578353881836,
      "step": 82520
    },
    {
      "dynamics_loss_avg": 0.057700541242957114,
      "epoch": 8.12530517578125e-06,
      "step": 82520
    },
    {
      "action_loss_avg": 0.0016214586910791695,
      "epoch": 8.12530517578125e-06,
      "step": 82520
    },
    {
      "epoch": 8.12530517578125e-06,
      "step": 82520,
      "training_step_time": 7.721398115158081
    },
    {
      "epoch": 8.126258850097657e-06,
      "model_forward_time": 1.7717046737670898,
      "step": 82521
    },
    {
      "epoch": 8.126258850097657e-06,
      "step": 82521,
      "training_step_time": 9.26829195022583
    },
    {
      "epoch": 8.127212524414063e-06,
      "model_forward_time": 1.7779107093811035,
      "step": 82522
    },
    {
      "epoch": 8.127212524414063e-06,
      "step": 82522,
      "training_step_time": 8.800932168960571
    },
    {
      "epoch": 8.128166198730468e-06,
      "model_forward_time": 1.753519058227539,
      "step": 82523
    },
    {
      "epoch": 8.128166198730468e-06,
      "step": 82523,
      "training_step_time": 10.206082105636597
    },
    {
      "epoch": 8.129119873046874e-06,
      "model_forward_time": 1.8838191032409668,
      "step": 82524
    },
    {
      "epoch": 8.129119873046874e-06,
      "step": 82524,
      "training_step_time": 8.553659200668335
    },
    {
      "epoch": 8.130073547363282e-06,
      "model_forward_time": 1.7702465057373047,
      "step": 82525
    },
    {
      "epoch": 8.130073547363282e-06,
      "step": 82525,
      "training_step_time": 8.327691555023193
    },
    {
      "epoch": 8.131027221679687e-06,
      "model_forward_time": 2.9266929626464844,
      "step": 82526
    },
    {
      "epoch": 8.131027221679687e-06,
      "step": 82526,
      "training_step_time": 7.8991241455078125
    },
    {
      "epoch": 8.131980895996093e-06,
      "model_forward_time": 1.798896312713623,
      "step": 82527
    },
    {
      "epoch": 8.131980895996093e-06,
      "step": 82527,
      "training_step_time": 7.969172716140747
    },
    {
      "epoch": 8.1329345703125e-06,
      "model_forward_time": 1.7724509239196777,
      "step": 82528
    },
    {
      "epoch": 8.1329345703125e-06,
      "step": 82528,
      "training_step_time": 7.513394832611084
    },
    {
      "epoch": 8.133888244628906e-06,
      "model_forward_time": 1.7656025886535645,
      "step": 82529
    },
    {
      "epoch": 8.133888244628906e-06,
      "step": 82529,
      "training_step_time": 7.731700897216797
    },
    {
      "epoch": 8.134841918945312e-06,
      "grad_norm": 0.11595468968153,
      "learning_rate": 8.11546883367893e-07,
      "loss": 0.0566,
      "step": 82530
    },
    {
      "epoch": 8.134841918945312e-06,
      "model_forward_time": 1.7776525020599365,
      "step": 82530
    },
    {
      "dynamics_loss_avg": 0.06553062293678522,
      "epoch": 8.134841918945312e-06,
      "step": 82530
    },
    {
      "action_loss_avg": 0.0018534827831899747,
      "epoch": 8.134841918945312e-06,
      "step": 82530
    },
    {
      "epoch": 8.134841918945312e-06,
      "step": 82530,
      "training_step_time": 7.792956590652466
    },
    {
      "epoch": 8.13579559326172e-06,
      "model_forward_time": 1.7728071212768555,
      "step": 82531
    },
    {
      "epoch": 8.13579559326172e-06,
      "step": 82531,
      "training_step_time": 7.813793897628784
    },
    {
      "epoch": 8.136749267578125e-06,
      "model_forward_time": 1.7724854946136475,
      "step": 82532
    },
    {
      "epoch": 8.136749267578125e-06,
      "step": 82532,
      "training_step_time": 9.45901083946228
    },
    {
      "epoch": 8.13770294189453e-06,
      "model_forward_time": 1.8914470672607422,
      "step": 82533
    },
    {
      "epoch": 8.13770294189453e-06,
      "step": 82533,
      "training_step_time": 10.284889221191406
    },
    {
      "epoch": 8.138656616210938e-06,
      "model_forward_time": 1.780949354171753,
      "step": 82534
    },
    {
      "epoch": 8.138656616210938e-06,
      "step": 82534,
      "training_step_time": 8.460083484649658
    },
    {
      "epoch": 8.139610290527344e-06,
      "model_forward_time": 2.896709442138672,
      "step": 82535
    },
    {
      "epoch": 8.139610290527344e-06,
      "step": 82535,
      "training_step_time": 9.114205360412598
    },
    {
      "epoch": 8.14056396484375e-06,
      "model_forward_time": 1.7790331840515137,
      "step": 82536
    },
    {
      "epoch": 8.14056396484375e-06,
      "step": 82536,
      "training_step_time": 8.876765966415405
    },
    {
      "epoch": 8.141517639160157e-06,
      "model_forward_time": 1.7759037017822266,
      "step": 82537
    },
    {
      "epoch": 8.141517639160157e-06,
      "step": 82537,
      "training_step_time": 8.048319339752197
    },
    {
      "epoch": 8.142471313476563e-06,
      "model_forward_time": 1.7774324417114258,
      "step": 82538
    },
    {
      "epoch": 8.142471313476563e-06,
      "step": 82538,
      "training_step_time": 7.853197813034058
    },
    {
      "epoch": 8.143424987792968e-06,
      "model_forward_time": 1.7765536308288574,
      "step": 82539
    },
    {
      "epoch": 8.143424987792968e-06,
      "step": 82539,
      "training_step_time": 8.669915199279785
    },
    {
      "epoch": 8.144378662109374e-06,
      "grad_norm": 0.07011426985263824,
      "learning_rate": 8.106440780435881e-07,
      "loss": 0.0551,
      "step": 82540
    },
    {
      "epoch": 8.144378662109374e-06,
      "model_forward_time": 1.7761969566345215,
      "step": 82540
    },
    {
      "dynamics_loss_avg": 0.050475400127470496,
      "epoch": 8.144378662109374e-06,
      "step": 82540
    },
    {
      "action_loss_avg": 0.0010947307921014727,
      "epoch": 8.144378662109374e-06,
      "step": 82540
    },
    {
      "epoch": 8.144378662109374e-06,
      "step": 82540,
      "training_step_time": 9.371299028396606
    },
    {
      "epoch": 8.145332336425782e-06,
      "model_forward_time": 1.7824513912200928,
      "step": 82541
    },
    {
      "epoch": 8.145332336425782e-06,
      "step": 82541,
      "training_step_time": 8.455265760421753
    },
    {
      "epoch": 8.146286010742187e-06,
      "model_forward_time": 1.859480619430542,
      "step": 82542
    },
    {
      "epoch": 8.146286010742187e-06,
      "step": 82542,
      "training_step_time": 9.085067987442017
    },
    {
      "epoch": 8.147239685058593e-06,
      "model_forward_time": 1.781053066253662,
      "step": 82543
    },
    {
      "epoch": 8.147239685058593e-06,
      "step": 82543,
      "training_step_time": 8.709209203720093
    },
    {
      "epoch": 8.148193359375e-06,
      "model_forward_time": 2.8512580394744873,
      "step": 82544
    },
    {
      "epoch": 8.148193359375e-06,
      "step": 82544,
      "training_step_time": 8.84796929359436
    },
    {
      "epoch": 8.149147033691406e-06,
      "model_forward_time": 1.778437614440918,
      "step": 82545
    },
    {
      "epoch": 8.149147033691406e-06,
      "step": 82545,
      "training_step_time": 7.790908098220825
    },
    {
      "epoch": 8.150100708007812e-06,
      "model_forward_time": 1.7806532382965088,
      "step": 82546
    },
    {
      "epoch": 8.150100708007812e-06,
      "step": 82546,
      "training_step_time": 8.071058511734009
    },
    {
      "epoch": 8.15105438232422e-06,
      "model_forward_time": 1.7715730667114258,
      "step": 82547
    },
    {
      "epoch": 8.15105438232422e-06,
      "step": 82547,
      "training_step_time": 7.205406188964844
    },
    {
      "epoch": 8.152008056640625e-06,
      "model_forward_time": 1.7666864395141602,
      "step": 82548
    },
    {
      "epoch": 8.152008056640625e-06,
      "step": 82548,
      "training_step_time": 7.837791681289673
    },
    {
      "epoch": 8.15296173095703e-06,
      "model_forward_time": 1.784583330154419,
      "step": 82549
    },
    {
      "epoch": 8.15296173095703e-06,
      "step": 82549,
      "training_step_time": 7.904341697692871
    },
    {
      "epoch": 8.153915405273438e-06,
      "grad_norm": 0.07444692403078079,
      "learning_rate": 8.097417308609485e-07,
      "loss": 0.0549,
      "step": 82550
    },
    {
      "epoch": 8.153915405273438e-06,
      "model_forward_time": 1.7719619274139404,
      "step": 82550
    },
    {
      "dynamics_loss_avg": 0.05851665195077658,
      "epoch": 8.153915405273438e-06,
      "step": 82550
    },
    {
      "action_loss_avg": 0.0019764644326642155,
      "epoch": 8.153915405273438e-06,
      "step": 82550
    },
    {
      "epoch": 8.153915405273438e-06,
      "step": 82550,
      "training_step_time": 7.887195587158203
    },
    {
      "epoch": 8.154869079589844e-06,
      "model_forward_time": 1.8265881538391113,
      "step": 82551
    },
    {
      "epoch": 8.154869079589844e-06,
      "step": 82551,
      "training_step_time": 9.139633417129517
    },
    {
      "epoch": 8.15582275390625e-06,
      "model_forward_time": 1.7752189636230469,
      "step": 82552
    },
    {
      "epoch": 8.15582275390625e-06,
      "step": 82552,
      "training_step_time": 8.70042371749878
    },
    {
      "epoch": 8.156776428222657e-06,
      "model_forward_time": 1.7692296504974365,
      "step": 82553
    },
    {
      "epoch": 8.156776428222657e-06,
      "step": 82553,
      "training_step_time": 8.854713201522827
    },
    {
      "epoch": 8.157730102539063e-06,
      "model_forward_time": 1.0811481475830078,
      "step": 82554
    },
    {
      "epoch": 8.157730102539063e-06,
      "step": 82554,
      "training_step_time": 8.62432312965393
    },
    {
      "epoch": 8.158683776855468e-06,
      "model_forward_time": 1.7642102241516113,
      "step": 82555
    },
    {
      "epoch": 8.158683776855468e-06,
      "step": 82555,
      "training_step_time": 8.589231014251709
    },
    {
      "epoch": 8.159637451171874e-06,
      "model_forward_time": 1.7852437496185303,
      "step": 82556
    },
    {
      "epoch": 8.159637451171874e-06,
      "step": 82556,
      "training_step_time": 8.071169376373291
    },
    {
      "epoch": 8.160591125488282e-06,
      "model_forward_time": 1.4028875827789307,
      "step": 82557
    },
    {
      "epoch": 8.160591125488282e-06,
      "step": 82557,
      "training_step_time": 7.806694507598877
    },
    {
      "epoch": 8.161544799804687e-06,
      "model_forward_time": 1.7770652770996094,
      "step": 82558
    },
    {
      "epoch": 8.161544799804687e-06,
      "step": 82558,
      "training_step_time": 7.882282257080078
    },
    {
      "epoch": 8.162498474121093e-06,
      "model_forward_time": 1.7880775928497314,
      "step": 82559
    },
    {
      "epoch": 8.162498474121093e-06,
      "step": 82559,
      "training_step_time": 7.783663749694824
    },
    {
      "epoch": 8.1634521484375e-06,
      "grad_norm": 0.12786947190761566,
      "learning_rate": 8.088398419186544e-07,
      "loss": 0.054,
      "step": 82560
    },
    {
      "epoch": 8.1634521484375e-06,
      "model_forward_time": 1.7981796264648438,
      "step": 82560
    },
    {
      "dynamics_loss_avg": 0.05372625021263957,
      "epoch": 8.1634521484375e-06,
      "step": 82560
    },
    {
      "action_loss_avg": 0.0010998951649526135,
      "epoch": 8.1634521484375e-06,
      "step": 82560
    },
    {
      "epoch": 8.1634521484375e-06,
      "step": 82560,
      "training_step_time": 8.742350578308105
    },
    {
      "epoch": 8.164405822753906e-06,
      "model_forward_time": 1.7725021839141846,
      "step": 82561
    },
    {
      "epoch": 8.164405822753906e-06,
      "step": 82561,
      "training_step_time": 7.973686456680298
    },
    {
      "epoch": 8.165359497070312e-06,
      "model_forward_time": 1.4310197830200195,
      "step": 82562
    },
    {
      "epoch": 8.165359497070312e-06,
      "step": 82562,
      "training_step_time": 7.778709888458252
    },
    {
      "epoch": 8.16631317138672e-06,
      "model_forward_time": 2.8353748321533203,
      "step": 82563
    },
    {
      "epoch": 8.16631317138672e-06,
      "step": 82563,
      "training_step_time": 7.834165573120117
    },
    {
      "epoch": 8.167266845703125e-06,
      "model_forward_time": 1.7751212120056152,
      "step": 82564
    },
    {
      "epoch": 8.167266845703125e-06,
      "step": 82564,
      "training_step_time": 8.815273761749268
    },
    {
      "epoch": 8.16822052001953e-06,
      "model_forward_time": 1.7774403095245361,
      "step": 82565
    },
    {
      "epoch": 8.16822052001953e-06,
      "step": 82565,
      "training_step_time": 8.006138324737549
    },
    {
      "epoch": 8.169174194335938e-06,
      "model_forward_time": 1.761636734008789,
      "step": 82566
    },
    {
      "epoch": 8.169174194335938e-06,
      "step": 82566,
      "training_step_time": 7.776934385299683
    },
    {
      "epoch": 8.170127868652344e-06,
      "model_forward_time": 1.7628848552703857,
      "step": 82567
    },
    {
      "epoch": 8.170127868652344e-06,
      "step": 82567,
      "training_step_time": 7.850689888000488
    },
    {
      "epoch": 8.17108154296875e-06,
      "model_forward_time": 1.794633150100708,
      "step": 82568
    },
    {
      "epoch": 8.17108154296875e-06,
      "step": 82568,
      "training_step_time": 8.454918622970581
    },
    {
      "epoch": 8.172035217285157e-06,
      "model_forward_time": 1.7698783874511719,
      "step": 82569
    },
    {
      "epoch": 8.172035217285157e-06,
      "step": 82569,
      "training_step_time": 9.547682046890259
    },
    {
      "epoch": 8.172988891601563e-06,
      "grad_norm": 0.07255934923887253,
      "learning_rate": 8.079384113153321e-07,
      "loss": 0.0545,
      "step": 82570
    },
    {
      "epoch": 8.172988891601563e-06,
      "model_forward_time": 1.8754775524139404,
      "step": 82570
    },
    {
      "dynamics_loss_avg": 0.06807006672024726,
      "epoch": 8.172988891601563e-06,
      "step": 82570
    },
    {
      "action_loss_avg": 0.001997581584146246,
      "epoch": 8.172988891601563e-06,
      "step": 82570
    },
    {
      "epoch": 8.172988891601563e-06,
      "step": 82570,
      "training_step_time": 8.754210710525513
    },
    {
      "epoch": 8.173942565917968e-06,
      "model_forward_time": 1.7723937034606934,
      "step": 82571
    },
    {
      "epoch": 8.173942565917968e-06,
      "step": 82571,
      "training_step_time": 7.743457317352295
    },
    {
      "epoch": 8.174896240234374e-06,
      "model_forward_time": 1.7689788341522217,
      "step": 82572
    },
    {
      "epoch": 8.174896240234374e-06,
      "step": 82572,
      "training_step_time": 9.821116924285889
    },
    {
      "epoch": 8.175849914550782e-06,
      "model_forward_time": 1.7660274505615234,
      "step": 82573
    },
    {
      "epoch": 8.175849914550782e-06,
      "step": 82573,
      "training_step_time": 7.599442481994629
    },
    {
      "epoch": 8.176803588867187e-06,
      "model_forward_time": 1.7584013938903809,
      "step": 82574
    },
    {
      "epoch": 8.176803588867187e-06,
      "step": 82574,
      "training_step_time": 8.006853342056274
    },
    {
      "epoch": 8.177757263183593e-06,
      "model_forward_time": 1.7786307334899902,
      "step": 82575
    },
    {
      "epoch": 8.177757263183593e-06,
      "step": 82575,
      "training_step_time": 8.472108602523804
    },
    {
      "epoch": 8.1787109375e-06,
      "model_forward_time": 1.0712006092071533,
      "step": 82576
    },
    {
      "epoch": 8.1787109375e-06,
      "step": 82576,
      "training_step_time": 8.778784036636353
    },
    {
      "epoch": 8.179664611816406e-06,
      "model_forward_time": 1.7801926136016846,
      "step": 82577
    },
    {
      "epoch": 8.179664611816406e-06,
      "step": 82577,
      "training_step_time": 11.455224752426147
    },
    {
      "epoch": 8.180618286132812e-06,
      "model_forward_time": 1.7672491073608398,
      "step": 82578
    },
    {
      "epoch": 8.180618286132812e-06,
      "step": 82578,
      "training_step_time": 8.767574787139893
    },
    {
      "epoch": 8.18157196044922e-06,
      "model_forward_time": 1.9207632541656494,
      "step": 82579
    },
    {
      "epoch": 8.18157196044922e-06,
      "step": 82579,
      "training_step_time": 9.444523572921753
    },
    {
      "epoch": 8.182525634765625e-06,
      "grad_norm": 0.07853492349386215,
      "learning_rate": 8.070374391495645e-07,
      "loss": 0.0539,
      "step": 82580
    },
    {
      "epoch": 8.182525634765625e-06,
      "model_forward_time": 1.9462661743164062,
      "step": 82580
    },
    {
      "dynamics_loss_avg": 0.06709785424172879,
      "epoch": 8.182525634765625e-06,
      "step": 82580
    },
    {
      "action_loss_avg": 0.001263226178707555,
      "epoch": 8.182525634765625e-06,
      "step": 82580
    },
    {
      "epoch": 8.182525634765625e-06,
      "step": 82580,
      "training_step_time": 9.747853517532349
    },
    {
      "epoch": 8.18347930908203e-06,
      "model_forward_time": 1.9421985149383545,
      "step": 82581
    },
    {
      "epoch": 8.18347930908203e-06,
      "step": 82581,
      "training_step_time": 7.793494701385498
    },
    {
      "epoch": 8.184432983398438e-06,
      "model_forward_time": 2.9549617767333984,
      "step": 82582
    },
    {
      "epoch": 8.184432983398438e-06,
      "step": 82582,
      "training_step_time": 8.97904109954834
    },
    {
      "epoch": 8.185386657714844e-06,
      "model_forward_time": 1.9582488536834717,
      "step": 82583
    },
    {
      "epoch": 8.185386657714844e-06,
      "step": 82583,
      "training_step_time": 7.973209857940674
    },
    {
      "epoch": 8.18634033203125e-06,
      "model_forward_time": 1.936058521270752,
      "step": 82584
    },
    {
      "epoch": 8.18634033203125e-06,
      "step": 82584,
      "training_step_time": 7.353425025939941
    },
    {
      "epoch": 8.187294006347657e-06,
      "model_forward_time": 1.942694902420044,
      "step": 82585
    },
    {
      "epoch": 8.187294006347657e-06,
      "step": 82585,
      "training_step_time": 7.80698299407959
    },
    {
      "epoch": 8.188247680664063e-06,
      "model_forward_time": 1.9675536155700684,
      "step": 82586
    },
    {
      "epoch": 8.188247680664063e-06,
      "step": 82586,
      "training_step_time": 10.415791988372803
    },
    {
      "epoch": 8.189201354980469e-06,
      "model_forward_time": 1.9598398208618164,
      "step": 82587
    },
    {
      "epoch": 8.189201354980469e-06,
      "step": 82587,
      "training_step_time": 9.084349632263184
    },
    {
      "epoch": 8.190155029296874e-06,
      "model_forward_time": 1.9648826122283936,
      "step": 82588
    },
    {
      "epoch": 8.190155029296874e-06,
      "step": 82588,
      "training_step_time": 7.9951701164245605
    },
    {
      "epoch": 8.191108703613282e-06,
      "model_forward_time": 1.9381563663482666,
      "step": 82589
    },
    {
      "epoch": 8.191108703613282e-06,
      "step": 82589,
      "training_step_time": 8.651479005813599
    },
    {
      "epoch": 8.192062377929687e-06,
      "grad_norm": 0.06886675208806992,
      "learning_rate": 8.061369255198764e-07,
      "loss": 0.0539,
      "step": 82590
    },
    {
      "epoch": 8.192062377929687e-06,
      "model_forward_time": 1.952918291091919,
      "step": 82590
    },
    {
      "dynamics_loss_avg": 0.07016521133482456,
      "epoch": 8.192062377929687e-06,
      "step": 82590
    },
    {
      "action_loss_avg": 0.0011365671292878688,
      "epoch": 8.192062377929687e-06,
      "step": 82590
    },
    {
      "epoch": 8.192062377929687e-06,
      "step": 82590,
      "training_step_time": 7.953471660614014
    },
    {
      "epoch": 8.193016052246093e-06,
      "model_forward_time": 3.033125400543213,
      "step": 82591
    },
    {
      "epoch": 8.193016052246093e-06,
      "step": 82591,
      "training_step_time": 8.823776960372925
    },
    {
      "epoch": 8.1939697265625e-06,
      "model_forward_time": 1.9315333366394043,
      "step": 82592
    },
    {
      "epoch": 8.1939697265625e-06,
      "step": 82592,
      "training_step_time": 8.738482475280762
    },
    {
      "epoch": 8.194923400878906e-06,
      "model_forward_time": 1.9498145580291748,
      "step": 82593
    },
    {
      "epoch": 8.194923400878906e-06,
      "step": 82593,
      "training_step_time": 7.523311614990234
    },
    {
      "epoch": 8.195877075195312e-06,
      "model_forward_time": 1.9428701400756836,
      "step": 82594
    },
    {
      "epoch": 8.195877075195312e-06,
      "step": 82594,
      "training_step_time": 7.792585611343384
    },
    {
      "epoch": 8.19683074951172e-06,
      "model_forward_time": 1.923677682876587,
      "step": 82595
    },
    {
      "epoch": 8.19683074951172e-06,
      "step": 82595,
      "training_step_time": 7.970085382461548
    },
    {
      "epoch": 8.197784423828125e-06,
      "model_forward_time": 1.9394512176513672,
      "step": 82596
    },
    {
      "epoch": 8.197784423828125e-06,
      "step": 82596,
      "training_step_time": 7.809509038925171
    },
    {
      "epoch": 8.19873809814453e-06,
      "model_forward_time": 1.9097836017608643,
      "step": 82597
    },
    {
      "epoch": 8.19873809814453e-06,
      "step": 82597,
      "training_step_time": 7.9096574783325195
    },
    {
      "epoch": 8.199691772460938e-06,
      "model_forward_time": 1.944054365158081,
      "step": 82598
    },
    {
      "epoch": 8.199691772460938e-06,
      "step": 82598,
      "training_step_time": 8.601830244064331
    },
    {
      "epoch": 8.200645446777344e-06,
      "model_forward_time": 1.9385716915130615,
      "step": 82599
    },
    {
      "epoch": 8.200645446777344e-06,
      "step": 82599,
      "training_step_time": 8.898561716079712
    },
    {
      "epoch": 8.20159912109375e-06,
      "grad_norm": 0.08748704195022583,
      "learning_rate": 8.052368705247494e-07,
      "loss": 0.0533,
      "step": 82600
    },
    {
      "epoch": 8.20159912109375e-06,
      "model_forward_time": 2.9609344005584717,
      "step": 82600
    },
    {
      "dynamics_loss_avg": 0.046924649737775326,
      "epoch": 8.20159912109375e-06,
      "step": 82600
    },
    {
      "action_loss_avg": 0.0011400271265301853,
      "epoch": 8.20159912109375e-06,
      "step": 82600
    },
    {
      "epoch": 8.20159912109375e-06,
      "step": 82600,
      "training_step_time": 8.394976615905762
    },
    {
      "epoch": 8.202552795410157e-06,
      "model_forward_time": 1.937962293624878,
      "step": 82601
    },
    {
      "epoch": 8.202552795410157e-06,
      "step": 82601,
      "training_step_time": 8.563644170761108
    },
    {
      "epoch": 8.203506469726563e-06,
      "model_forward_time": 1.8974933624267578,
      "step": 82602
    },
    {
      "epoch": 8.203506469726563e-06,
      "step": 82602,
      "training_step_time": 7.473621845245361
    },
    {
      "epoch": 8.204460144042969e-06,
      "model_forward_time": 1.9319002628326416,
      "step": 82603
    },
    {
      "epoch": 8.204460144042969e-06,
      "step": 82603,
      "training_step_time": 7.705322504043579
    },
    {
      "epoch": 8.205413818359374e-06,
      "model_forward_time": 1.9258849620819092,
      "step": 82604
    },
    {
      "epoch": 8.205413818359374e-06,
      "step": 82604,
      "training_step_time": 9.463574647903442
    },
    {
      "epoch": 8.206367492675782e-06,
      "model_forward_time": 1.9482617378234863,
      "step": 82605
    },
    {
      "epoch": 8.206367492675782e-06,
      "step": 82605,
      "training_step_time": 9.948615550994873
    },
    {
      "epoch": 8.207321166992187e-06,
      "model_forward_time": 1.9490869045257568,
      "step": 82606
    },
    {
      "epoch": 8.207321166992187e-06,
      "step": 82606,
      "training_step_time": 8.919441938400269
    },
    {
      "epoch": 8.208274841308593e-06,
      "model_forward_time": 1.9320111274719238,
      "step": 82607
    },
    {
      "epoch": 8.208274841308593e-06,
      "step": 82607,
      "training_step_time": 8.043489217758179
    },
    {
      "epoch": 8.209228515625e-06,
      "model_forward_time": 1.9483661651611328,
      "step": 82608
    },
    {
      "epoch": 8.209228515625e-06,
      "step": 82608,
      "training_step_time": 7.828906536102295
    },
    {
      "epoch": 8.210182189941406e-06,
      "model_forward_time": 2.97701096534729,
      "step": 82609
    },
    {
      "epoch": 8.210182189941406e-06,
      "step": 82609,
      "training_step_time": 7.8946852684021
    },
    {
      "epoch": 8.211135864257812e-06,
      "grad_norm": 0.09353726357221603,
      "learning_rate": 8.043372742626115e-07,
      "loss": 0.0535,
      "step": 82610
    },
    {
      "epoch": 8.211135864257812e-06,
      "model_forward_time": 1.9303746223449707,
      "step": 82610
    },
    {
      "dynamics_loss_avg": 0.06279597878456115,
      "epoch": 8.211135864257812e-06,
      "step": 82610
    },
    {
      "action_loss_avg": 0.0017988760751904919,
      "epoch": 8.211135864257812e-06,
      "step": 82610
    },
    {
      "epoch": 8.211135864257812e-06,
      "step": 82610,
      "training_step_time": 7.770395994186401
    },
    {
      "epoch": 8.21208953857422e-06,
      "model_forward_time": 1.9388456344604492,
      "step": 82611
    },
    {
      "epoch": 8.21208953857422e-06,
      "step": 82611,
      "training_step_time": 8.000593900680542
    },
    {
      "epoch": 8.213043212890625e-06,
      "model_forward_time": 1.5914793014526367,
      "step": 82612
    },
    {
      "epoch": 8.213043212890625e-06,
      "step": 82612,
      "training_step_time": 7.677665948867798
    },
    {
      "epoch": 8.21399688720703e-06,
      "model_forward_time": 1.9431774616241455,
      "step": 82613
    },
    {
      "epoch": 8.21399688720703e-06,
      "step": 82613,
      "training_step_time": 7.748013496398926
    },
    {
      "epoch": 8.214950561523438e-06,
      "model_forward_time": 1.9353418350219727,
      "step": 82614
    },
    {
      "epoch": 8.214950561523438e-06,
      "step": 82614,
      "training_step_time": 11.426214933395386
    },
    {
      "epoch": 8.215904235839844e-06,
      "model_forward_time": 1.9257335662841797,
      "step": 82615
    },
    {
      "epoch": 8.215904235839844e-06,
      "step": 82615,
      "training_step_time": 7.805818319320679
    },
    {
      "epoch": 8.21685791015625e-06,
      "model_forward_time": 1.925459384918213,
      "step": 82616
    },
    {
      "epoch": 8.21685791015625e-06,
      "step": 82616,
      "training_step_time": 8.714434623718262
    },
    {
      "epoch": 8.217811584472657e-06,
      "model_forward_time": 1.9363715648651123,
      "step": 82617
    },
    {
      "epoch": 8.217811584472657e-06,
      "step": 82617,
      "training_step_time": 8.8633873462677
    },
    {
      "epoch": 8.218765258789063e-06,
      "model_forward_time": 1.9494028091430664,
      "step": 82618
    },
    {
      "epoch": 8.218765258789063e-06,
      "step": 82618,
      "training_step_time": 9.657114744186401
    },
    {
      "epoch": 8.219718933105469e-06,
      "model_forward_time": 2.988435983657837,
      "step": 82619
    },
    {
      "epoch": 8.219718933105469e-06,
      "step": 82619,
      "training_step_time": 7.945858955383301
    },
    {
      "epoch": 8.220672607421874e-06,
      "grad_norm": 0.08953063189983368,
      "learning_rate": 8.034381368318422e-07,
      "loss": 0.0538,
      "step": 82620
    },
    {
      "epoch": 8.220672607421874e-06,
      "model_forward_time": 1.952894926071167,
      "step": 82620
    },
    {
      "dynamics_loss_avg": 0.047177298180758956,
      "epoch": 8.220672607421874e-06,
      "step": 82620
    },
    {
      "action_loss_avg": 0.0008563119044993073,
      "epoch": 8.220672607421874e-06,
      "step": 82620
    },
    {
      "epoch": 8.220672607421874e-06,
      "step": 82620,
      "training_step_time": 8.84810209274292
    },
    {
      "epoch": 8.221626281738282e-06,
      "model_forward_time": 1.932396650314331,
      "step": 82621
    },
    {
      "epoch": 8.221626281738282e-06,
      "step": 82621,
      "training_step_time": 7.5634236335754395
    },
    {
      "epoch": 8.222579956054687e-06,
      "model_forward_time": 1.9334707260131836,
      "step": 82622
    },
    {
      "epoch": 8.222579956054687e-06,
      "step": 82622,
      "training_step_time": 7.752593517303467
    },
    {
      "epoch": 8.223533630371093e-06,
      "model_forward_time": 1.952455759048462,
      "step": 82623
    },
    {
      "epoch": 8.223533630371093e-06,
      "step": 82623,
      "training_step_time": 9.841145038604736
    },
    {
      "epoch": 8.2244873046875e-06,
      "model_forward_time": 1.9194386005401611,
      "step": 82624
    },
    {
      "epoch": 8.2244873046875e-06,
      "step": 82624,
      "training_step_time": 8.59554672241211
    },
    {
      "epoch": 8.225440979003906e-06,
      "model_forward_time": 1.9111549854278564,
      "step": 82625
    },
    {
      "epoch": 8.225440979003906e-06,
      "step": 82625,
      "training_step_time": 8.603605270385742
    },
    {
      "epoch": 8.226394653320312e-06,
      "model_forward_time": 1.9457759857177734,
      "step": 82626
    },
    {
      "epoch": 8.226394653320312e-06,
      "step": 82626,
      "training_step_time": 8.688601732254028
    },
    {
      "epoch": 8.22734832763672e-06,
      "model_forward_time": 1.9588944911956787,
      "step": 82627
    },
    {
      "epoch": 8.22734832763672e-06,
      "step": 82627,
      "training_step_time": 8.394771814346313
    },
    {
      "epoch": 8.228302001953125e-06,
      "model_forward_time": 2.9569666385650635,
      "step": 82628
    },
    {
      "epoch": 8.228302001953125e-06,
      "step": 82628,
      "training_step_time": 8.80593729019165
    },
    {
      "epoch": 8.22925567626953e-06,
      "model_forward_time": 1.9382166862487793,
      "step": 82629
    },
    {
      "epoch": 8.22925567626953e-06,
      "step": 82629,
      "training_step_time": 7.778764009475708
    },
    {
      "epoch": 8.230209350585938e-06,
      "grad_norm": 0.07455076277256012,
      "learning_rate": 8.025394583307678e-07,
      "loss": 0.0534,
      "step": 82630
    },
    {
      "epoch": 8.230209350585938e-06,
      "model_forward_time": 1.8918015956878662,
      "step": 82630
    },
    {
      "dynamics_loss_avg": 0.051549849659204484,
      "epoch": 8.230209350585938e-06,
      "step": 82630
    },
    {
      "action_loss_avg": 0.0011054813803639263,
      "epoch": 8.230209350585938e-06,
      "step": 82630
    },
    {
      "epoch": 8.230209350585938e-06,
      "step": 82630,
      "training_step_time": 8.023617029190063
    },
    {
      "epoch": 8.231163024902344e-06,
      "model_forward_time": 1.8898861408233643,
      "step": 82631
    },
    {
      "epoch": 8.231163024902344e-06,
      "step": 82631,
      "training_step_time": 7.810231447219849
    },
    {
      "epoch": 8.23211669921875e-06,
      "model_forward_time": 1.9482176303863525,
      "step": 82632
    },
    {
      "epoch": 8.23211669921875e-06,
      "step": 82632,
      "training_step_time": 8.803040266036987
    },
    {
      "epoch": 8.233070373535155e-06,
      "model_forward_time": 1.923635482788086,
      "step": 82633
    },
    {
      "epoch": 8.233070373535155e-06,
      "step": 82633,
      "training_step_time": 8.397496223449707
    },
    {
      "epoch": 8.234024047851563e-06,
      "model_forward_time": 1.9018309116363525,
      "step": 82634
    },
    {
      "epoch": 8.234024047851563e-06,
      "step": 82634,
      "training_step_time": 8.512573480606079
    },
    {
      "epoch": 8.234977722167969e-06,
      "model_forward_time": 1.8914470672607422,
      "step": 82635
    },
    {
      "epoch": 8.234977722167969e-06,
      "step": 82635,
      "training_step_time": 7.910682201385498
    },
    {
      "epoch": 8.235931396484374e-06,
      "model_forward_time": 1.9277503490447998,
      "step": 82636
    },
    {
      "epoch": 8.235931396484374e-06,
      "step": 82636,
      "training_step_time": 7.893832206726074
    },
    {
      "epoch": 8.236885070800782e-06,
      "model_forward_time": 3.0393950939178467,
      "step": 82637
    },
    {
      "epoch": 8.236885070800782e-06,
      "step": 82637,
      "training_step_time": 9.639336824417114
    },
    {
      "epoch": 8.237838745117187e-06,
      "model_forward_time": 1.9192044734954834,
      "step": 82638
    },
    {
      "epoch": 8.237838745117187e-06,
      "step": 82638,
      "training_step_time": 7.744055986404419
    },
    {
      "epoch": 8.238792419433593e-06,
      "model_forward_time": 1.932300329208374,
      "step": 82639
    },
    {
      "epoch": 8.238792419433593e-06,
      "step": 82639,
      "training_step_time": 8.157650470733643
    },
    {
      "epoch": 8.23974609375e-06,
      "grad_norm": 0.08078749477863312,
      "learning_rate": 8.016412388576684e-07,
      "loss": 0.0527,
      "step": 82640
    },
    {
      "epoch": 8.23974609375e-06,
      "model_forward_time": 1.9437134265899658,
      "step": 82640
    },
    {
      "dynamics_loss_avg": 0.05805468019098044,
      "epoch": 8.23974609375e-06,
      "step": 82640
    },
    {
      "action_loss_avg": 0.0011722604336682706,
      "epoch": 8.23974609375e-06,
      "step": 82640
    },
    {
      "epoch": 8.23974609375e-06,
      "step": 82640,
      "training_step_time": 7.854199171066284
    },
    {
      "epoch": 8.240699768066406e-06,
      "model_forward_time": 1.9465248584747314,
      "step": 82641
    },
    {
      "epoch": 8.240699768066406e-06,
      "step": 82641,
      "training_step_time": 9.390640020370483
    },
    {
      "epoch": 8.241653442382812e-06,
      "model_forward_time": 1.932037591934204,
      "step": 82642
    },
    {
      "epoch": 8.241653442382812e-06,
      "step": 82642,
      "training_step_time": 7.915113925933838
    },
    {
      "epoch": 8.24260711669922e-06,
      "model_forward_time": 1.9393131732940674,
      "step": 82643
    },
    {
      "epoch": 8.24260711669922e-06,
      "step": 82643,
      "training_step_time": 8.811766862869263
    },
    {
      "epoch": 8.243560791015625e-06,
      "model_forward_time": 1.9256138801574707,
      "step": 82644
    },
    {
      "epoch": 8.243560791015625e-06,
      "step": 82644,
      "training_step_time": 9.758216857910156
    },
    {
      "epoch": 8.24451446533203e-06,
      "model_forward_time": 1.938286542892456,
      "step": 82645
    },
    {
      "epoch": 8.24451446533203e-06,
      "step": 82645,
      "training_step_time": 7.829681396484375
    },
    {
      "epoch": 8.245468139648438e-06,
      "model_forward_time": 2.6307480335235596,
      "step": 82646
    },
    {
      "epoch": 8.245468139648438e-06,
      "step": 82646,
      "training_step_time": 9.21899127960205
    },
    {
      "epoch": 8.246421813964844e-06,
      "model_forward_time": 1.9418964385986328,
      "step": 82647
    },
    {
      "epoch": 8.246421813964844e-06,
      "step": 82647,
      "training_step_time": 7.743180990219116
    },
    {
      "epoch": 8.24737548828125e-06,
      "model_forward_time": 1.9077515602111816,
      "step": 82648
    },
    {
      "epoch": 8.24737548828125e-06,
      "step": 82648,
      "training_step_time": 7.978566646575928
    },
    {
      "epoch": 8.248329162597655e-06,
      "model_forward_time": 1.920189619064331,
      "step": 82649
    },
    {
      "epoch": 8.248329162597655e-06,
      "step": 82649,
      "training_step_time": 7.710463047027588
    },
    {
      "epoch": 8.249282836914063e-06,
      "grad_norm": 0.10488211363554001,
      "learning_rate": 8.007434785107698e-07,
      "loss": 0.0537,
      "step": 82650
    },
    {
      "epoch": 8.249282836914063e-06,
      "model_forward_time": 1.9228131771087646,
      "step": 82650
    },
    {
      "dynamics_loss_avg": 0.05206015631556511,
      "epoch": 8.249282836914063e-06,
      "step": 82650
    },
    {
      "action_loss_avg": 0.001786469729267992,
      "epoch": 8.249282836914063e-06,
      "step": 82650
    },
    {
      "epoch": 8.249282836914063e-06,
      "step": 82650,
      "training_step_time": 7.737627267837524
    },
    {
      "epoch": 8.250236511230469e-06,
      "model_forward_time": 1.932962417602539,
      "step": 82651
    },
    {
      "epoch": 8.250236511230469e-06,
      "step": 82651,
      "training_step_time": 9.334198951721191
    },
    {
      "epoch": 8.251190185546874e-06,
      "model_forward_time": 1.9430625438690186,
      "step": 82652
    },
    {
      "epoch": 8.251190185546874e-06,
      "step": 82652,
      "training_step_time": 8.402733564376831
    },
    {
      "epoch": 8.252143859863282e-06,
      "model_forward_time": 1.9549312591552734,
      "step": 82653
    },
    {
      "epoch": 8.252143859863282e-06,
      "step": 82653,
      "training_step_time": 8.785009145736694
    },
    {
      "epoch": 8.253097534179687e-06,
      "model_forward_time": 1.9196419715881348,
      "step": 82654
    },
    {
      "epoch": 8.253097534179687e-06,
      "step": 82654,
      "training_step_time": 8.51821255683899
    },
    {
      "epoch": 8.254051208496093e-06,
      "model_forward_time": 1.9324772357940674,
      "step": 82655
    },
    {
      "epoch": 8.254051208496093e-06,
      "step": 82655,
      "training_step_time": 9.673883438110352
    },
    {
      "epoch": 8.2550048828125e-06,
      "model_forward_time": 2.9779574871063232,
      "step": 82656
    },
    {
      "epoch": 8.2550048828125e-06,
      "step": 82656,
      "training_step_time": 7.944074869155884
    },
    {
      "epoch": 8.255958557128906e-06,
      "model_forward_time": 1.9372646808624268,
      "step": 82657
    },
    {
      "epoch": 8.255958557128906e-06,
      "step": 82657,
      "training_step_time": 8.793521404266357
    },
    {
      "epoch": 8.256912231445312e-06,
      "model_forward_time": 1.5796136856079102,
      "step": 82658
    },
    {
      "epoch": 8.256912231445312e-06,
      "step": 82658,
      "training_step_time": 7.770209312438965
    },
    {
      "epoch": 8.25786590576172e-06,
      "model_forward_time": 1.9500133991241455,
      "step": 82659
    },
    {
      "epoch": 8.25786590576172e-06,
      "step": 82659,
      "training_step_time": 9.653043031692505
    },
    {
      "epoch": 8.258819580078125e-06,
      "grad_norm": 0.08205782622098923,
      "learning_rate": 7.99846177388251e-07,
      "loss": 0.054,
      "step": 82660
    },
    {
      "epoch": 8.258819580078125e-06,
      "model_forward_time": 1.9405996799468994,
      "step": 82660
    },
    {
      "dynamics_loss_avg": 0.0585799366235733,
      "epoch": 8.258819580078125e-06,
      "step": 82660
    },
    {
      "action_loss_avg": 0.001145169703522697,
      "epoch": 8.258819580078125e-06,
      "step": 82660
    },
    {
      "epoch": 8.258819580078125e-06,
      "step": 82660,
      "training_step_time": 8.833456993103027
    },
    {
      "epoch": 8.259773254394531e-06,
      "model_forward_time": 1.9282548427581787,
      "step": 82661
    },
    {
      "epoch": 8.259773254394531e-06,
      "step": 82661,
      "training_step_time": 12.581738471984863
    },
    {
      "epoch": 8.260726928710938e-06,
      "model_forward_time": 1.9576327800750732,
      "step": 82662
    },
    {
      "epoch": 8.260726928710938e-06,
      "step": 82662,
      "training_step_time": 6.923497200012207
    },
    {
      "epoch": 8.261680603027344e-06,
      "model_forward_time": 1.9459285736083984,
      "step": 82663
    },
    {
      "epoch": 8.261680603027344e-06,
      "step": 82663,
      "training_step_time": 7.837761402130127
    },
    {
      "epoch": 8.26263427734375e-06,
      "model_forward_time": 1.906477451324463,
      "step": 82664
    },
    {
      "epoch": 8.26263427734375e-06,
      "step": 82664,
      "training_step_time": 8.439849853515625
    },
    {
      "epoch": 8.263587951660155e-06,
      "model_forward_time": 1.9102168083190918,
      "step": 82665
    },
    {
      "epoch": 8.263587951660155e-06,
      "step": 82665,
      "training_step_time": 8.34507966041565
    },
    {
      "epoch": 8.264541625976563e-06,
      "model_forward_time": 1.9189283847808838,
      "step": 82666
    },
    {
      "epoch": 8.264541625976563e-06,
      "step": 82666,
      "training_step_time": 6.9627509117126465
    },
    {
      "epoch": 8.265495300292969e-06,
      "model_forward_time": 1.9060144424438477,
      "step": 82667
    },
    {
      "epoch": 8.265495300292969e-06,
      "step": 82667,
      "training_step_time": 6.996612548828125
    },
    {
      "epoch": 8.266448974609374e-06,
      "model_forward_time": 1.904296875,
      "step": 82668
    },
    {
      "epoch": 8.266448974609374e-06,
      "step": 82668,
      "training_step_time": 6.949282169342041
    },
    {
      "epoch": 8.267402648925782e-06,
      "model_forward_time": 1.9127049446105957,
      "step": 82669
    },
    {
      "epoch": 8.267402648925782e-06,
      "step": 82669,
      "training_step_time": 6.962196350097656
    },
    {
      "epoch": 8.268356323242187e-06,
      "grad_norm": 0.09466961026191711,
      "learning_rate": 7.989493355882394e-07,
      "loss": 0.0531,
      "step": 82670
    },
    {
      "epoch": 8.268356323242187e-06,
      "model_forward_time": 1.9577605724334717,
      "step": 82670
    },
    {
      "dynamics_loss_avg": 0.04861561711877584,
      "epoch": 8.268356323242187e-06,
      "step": 82670
    },
    {
      "action_loss_avg": 0.0013849921873770655,
      "epoch": 8.268356323242187e-06,
      "step": 82670
    },
    {
      "epoch": 8.268356323242187e-06,
      "step": 82670,
      "training_step_time": 6.963486671447754
    },
    {
      "epoch": 8.269309997558593e-06,
      "model_forward_time": 2.965210437774658,
      "step": 82671
    },
    {
      "epoch": 8.269309997558593e-06,
      "step": 82671,
      "training_step_time": 7.981818914413452
    },
    {
      "epoch": 8.270263671875e-06,
      "model_forward_time": 1.9103355407714844,
      "step": 82672
    },
    {
      "epoch": 8.270263671875e-06,
      "step": 82672,
      "training_step_time": 6.866619825363159
    },
    {
      "epoch": 8.271217346191406e-06,
      "model_forward_time": 1.9060752391815186,
      "step": 82673
    },
    {
      "epoch": 8.271217346191406e-06,
      "step": 82673,
      "training_step_time": 7.65083384513855
    },
    {
      "epoch": 8.272171020507812e-06,
      "model_forward_time": 1.947723627090454,
      "step": 82674
    },
    {
      "epoch": 8.272171020507812e-06,
      "step": 82674,
      "training_step_time": 7.552389621734619
    },
    {
      "epoch": 8.27312469482422e-06,
      "model_forward_time": 1.9204230308532715,
      "step": 82675
    },
    {
      "epoch": 8.27312469482422e-06,
      "step": 82675,
      "training_step_time": 7.855177879333496
    },
    {
      "epoch": 8.274078369140625e-06,
      "model_forward_time": 1.9194080829620361,
      "step": 82676
    },
    {
      "epoch": 8.274078369140625e-06,
      "step": 82676,
      "training_step_time": 6.85322380065918
    },
    {
      "epoch": 8.275032043457031e-06,
      "model_forward_time": 1.9292070865631104,
      "step": 82677
    },
    {
      "epoch": 8.275032043457031e-06,
      "step": 82677,
      "training_step_time": 7.025883197784424
    },
    {
      "epoch": 8.275985717773438e-06,
      "model_forward_time": 1.935180902481079,
      "step": 82678
    },
    {
      "epoch": 8.275985717773438e-06,
      "step": 82678,
      "training_step_time": 6.944341421127319
    },
    {
      "epoch": 8.276939392089844e-06,
      "model_forward_time": 1.9355220794677734,
      "step": 82679
    },
    {
      "epoch": 8.276939392089844e-06,
      "step": 82679,
      "training_step_time": 6.938831329345703
    },
    {
      "epoch": 8.27789306640625e-06,
      "grad_norm": 0.07240515202283859,
      "learning_rate": 7.980529532088138e-07,
      "loss": 0.055,
      "step": 82680
    },
    {
      "epoch": 8.27789306640625e-06,
      "model_forward_time": 2.9504547119140625,
      "step": 82680
    },
    {
      "dynamics_loss_avg": 0.0456789443269372,
      "epoch": 8.27789306640625e-06,
      "step": 82680
    },
    {
      "action_loss_avg": 0.0007498448248952627,
      "epoch": 8.27789306640625e-06,
      "step": 82680
    },
    {
      "epoch": 8.27789306640625e-06,
      "step": 82680,
      "training_step_time": 8.025027751922607
    },
    {
      "epoch": 8.278846740722655e-06,
      "model_forward_time": 1.9172987937927246,
      "step": 82681
    },
    {
      "epoch": 8.278846740722655e-06,
      "step": 82681,
      "training_step_time": 6.9478161334991455
    },
    {
      "epoch": 8.279800415039063e-06,
      "model_forward_time": 1.9084932804107666,
      "step": 82682
    },
    {
      "epoch": 8.279800415039063e-06,
      "step": 82682,
      "training_step_time": 7.665640115737915
    },
    {
      "epoch": 8.280754089355469e-06,
      "model_forward_time": 1.9267916679382324,
      "step": 82683
    },
    {
      "epoch": 8.280754089355469e-06,
      "step": 82683,
      "training_step_time": 7.511958837509155
    },
    {
      "epoch": 8.281707763671874e-06,
      "model_forward_time": 1.9213411808013916,
      "step": 82684
    },
    {
      "epoch": 8.281707763671874e-06,
      "step": 82684,
      "training_step_time": 7.785478353500366
    },
    {
      "epoch": 8.282661437988282e-06,
      "model_forward_time": 1.907684326171875,
      "step": 82685
    },
    {
      "epoch": 8.282661437988282e-06,
      "step": 82685,
      "training_step_time": 6.918008327484131
    },
    {
      "epoch": 8.283615112304687e-06,
      "model_forward_time": 1.927076816558838,
      "step": 82686
    },
    {
      "epoch": 8.283615112304687e-06,
      "step": 82686,
      "training_step_time": 6.9058167934417725
    },
    {
      "epoch": 8.284568786621093e-06,
      "model_forward_time": 1.929823398590088,
      "step": 82687
    },
    {
      "epoch": 8.284568786621093e-06,
      "step": 82687,
      "training_step_time": 6.91177773475647
    },
    {
      "epoch": 8.2855224609375e-06,
      "model_forward_time": 1.9263780117034912,
      "step": 82688
    },
    {
      "epoch": 8.2855224609375e-06,
      "step": 82688,
      "training_step_time": 6.976821184158325
    },
    {
      "epoch": 8.286476135253906e-06,
      "model_forward_time": 2.9614415168762207,
      "step": 82689
    },
    {
      "epoch": 8.286476135253906e-06,
      "step": 82689,
      "training_step_time": 8.053165674209595
    },
    {
      "epoch": 8.287429809570312e-06,
      "grad_norm": 0.07143199443817139,
      "learning_rate": 7.97157030347998e-07,
      "loss": 0.056,
      "step": 82690
    },
    {
      "epoch": 8.287429809570312e-06,
      "model_forward_time": 1.919752597808838,
      "step": 82690
    },
    {
      "dynamics_loss_avg": 0.060840200446546075,
      "epoch": 8.287429809570312e-06,
      "step": 82690
    },
    {
      "action_loss_avg": 0.0015248964715283363,
      "epoch": 8.287429809570312e-06,
      "step": 82690
    },
    {
      "epoch": 8.287429809570312e-06,
      "step": 82690,
      "training_step_time": 7.686606168746948
    },
    {
      "epoch": 8.28838348388672e-06,
      "model_forward_time": 1.9122815132141113,
      "step": 82691
    },
    {
      "epoch": 8.28838348388672e-06,
      "step": 82691,
      "training_step_time": 7.603528022766113
    },
    {
      "epoch": 8.289337158203125e-06,
      "model_forward_time": 1.931908369064331,
      "step": 82692
    },
    {
      "epoch": 8.289337158203125e-06,
      "step": 82692,
      "training_step_time": 7.647707462310791
    },
    {
      "epoch": 8.290290832519531e-06,
      "model_forward_time": 1.9406547546386719,
      "step": 82693
    },
    {
      "epoch": 8.290290832519531e-06,
      "step": 82693,
      "training_step_time": 7.778785467147827
    },
    {
      "epoch": 8.291244506835938e-06,
      "model_forward_time": 1.931661605834961,
      "step": 82694
    },
    {
      "epoch": 8.291244506835938e-06,
      "step": 82694,
      "training_step_time": 6.897672414779663
    },
    {
      "epoch": 8.292198181152344e-06,
      "model_forward_time": 1.9236822128295898,
      "step": 82695
    },
    {
      "epoch": 8.292198181152344e-06,
      "step": 82695,
      "training_step_time": 6.933881759643555
    },
    {
      "epoch": 8.29315185546875e-06,
      "model_forward_time": 1.1945242881774902,
      "step": 82696
    },
    {
      "epoch": 8.29315185546875e-06,
      "step": 82696,
      "training_step_time": 6.947443246841431
    },
    {
      "epoch": 8.294105529785155e-06,
      "model_forward_time": 1.9260749816894531,
      "step": 82697
    },
    {
      "epoch": 8.294105529785155e-06,
      "step": 82697,
      "training_step_time": 6.936072826385498
    },
    {
      "epoch": 8.295059204101563e-06,
      "model_forward_time": 2.9331374168395996,
      "step": 82698
    },
    {
      "epoch": 8.295059204101563e-06,
      "step": 82698,
      "training_step_time": 10.378600597381592
    },
    {
      "epoch": 8.296012878417969e-06,
      "model_forward_time": 1.9173648357391357,
      "step": 82699
    },
    {
      "epoch": 8.296012878417969e-06,
      "step": 82699,
      "training_step_time": 8.03020453453064
    },
    {
      "epoch": 8.296966552734374e-06,
      "grad_norm": 0.0776967853307724,
      "learning_rate": 7.962615671037722e-07,
      "loss": 0.0553,
      "step": 82700
    },
    {
      "epoch": 8.296966552734374e-06,
      "model_forward_time": 1.9385828971862793,
      "step": 82700
    },
    {
      "dynamics_loss_avg": 0.0549012191593647,
      "epoch": 8.296966552734374e-06,
      "step": 82700
    },
    {
      "action_loss_avg": 0.0014783119026105852,
      "epoch": 8.296966552734374e-06,
      "step": 82700
    },
    {
      "epoch": 8.296966552734374e-06,
      "step": 82700,
      "training_step_time": 7.534573078155518
    },
    {
      "epoch": 8.297920227050782e-06,
      "model_forward_time": 1.9491069316864014,
      "step": 82701
    },
    {
      "epoch": 8.297920227050782e-06,
      "step": 82701,
      "training_step_time": 7.642691135406494
    },
    {
      "epoch": 8.298873901367187e-06,
      "model_forward_time": 1.9435062408447266,
      "step": 82702
    },
    {
      "epoch": 8.298873901367187e-06,
      "step": 82702,
      "training_step_time": 7.82266902923584
    },
    {
      "epoch": 8.299827575683593e-06,
      "model_forward_time": 1.9374334812164307,
      "step": 82703
    },
    {
      "epoch": 8.299827575683593e-06,
      "step": 82703,
      "training_step_time": 6.921948671340942
    },
    {
      "epoch": 8.30078125e-06,
      "model_forward_time": 1.9395320415496826,
      "step": 82704
    },
    {
      "epoch": 8.30078125e-06,
      "step": 82704,
      "training_step_time": 6.903756856918335
    },
    {
      "epoch": 8.301734924316406e-06,
      "model_forward_time": 1.9406094551086426,
      "step": 82705
    },
    {
      "epoch": 8.301734924316406e-06,
      "step": 82705,
      "training_step_time": 6.9296746253967285
    },
    {
      "epoch": 8.302688598632812e-06,
      "model_forward_time": 1.9020774364471436,
      "step": 82706
    },
    {
      "epoch": 8.302688598632812e-06,
      "step": 82706,
      "training_step_time": 6.923504829406738
    },
    {
      "epoch": 8.30364227294922e-06,
      "model_forward_time": 1.9267768859863281,
      "step": 82707
    },
    {
      "epoch": 8.30364227294922e-06,
      "step": 82707,
      "training_step_time": 10.654972553253174
    },
    {
      "epoch": 8.304595947265625e-06,
      "model_forward_time": 2.9651618003845215,
      "step": 82708
    },
    {
      "epoch": 8.304595947265625e-06,
      "step": 82708,
      "training_step_time": 8.051115274429321
    },
    {
      "epoch": 8.305549621582031e-06,
      "model_forward_time": 1.9020915031433105,
      "step": 82709
    },
    {
      "epoch": 8.305549621582031e-06,
      "step": 82709,
      "training_step_time": 6.958564758300781
    },
    {
      "epoch": 8.306503295898438e-06,
      "grad_norm": 0.0807076022028923,
      "learning_rate": 7.953665635740593e-07,
      "loss": 0.056,
      "step": 82710
    },
    {
      "epoch": 8.306503295898438e-06,
      "model_forward_time": 1.9357445240020752,
      "step": 82710
    },
    {
      "dynamics_loss_avg": 0.05969242043793201,
      "epoch": 8.306503295898438e-06,
      "step": 82710
    },
    {
      "action_loss_avg": 0.0013462587143294514,
      "epoch": 8.306503295898438e-06,
      "step": 82710
    },
    {
      "epoch": 8.306503295898438e-06,
      "step": 82710,
      "training_step_time": 8.610200881958008
    },
    {
      "epoch": 8.307456970214844e-06,
      "model_forward_time": 1.9529130458831787,
      "step": 82711
    },
    {
      "epoch": 8.307456970214844e-06,
      "step": 82711,
      "training_step_time": 7.713622093200684
    },
    {
      "epoch": 8.30841064453125e-06,
      "model_forward_time": 1.9230492115020752,
      "step": 82712
    },
    {
      "epoch": 8.30841064453125e-06,
      "step": 82712,
      "training_step_time": 7.845873832702637
    },
    {
      "epoch": 8.309364318847655e-06,
      "model_forward_time": 1.9387645721435547,
      "step": 82713
    },
    {
      "epoch": 8.309364318847655e-06,
      "step": 82713,
      "training_step_time": 7.109930038452148
    },
    {
      "epoch": 8.310317993164063e-06,
      "model_forward_time": 1.9307219982147217,
      "step": 82714
    },
    {
      "epoch": 8.310317993164063e-06,
      "step": 82714,
      "training_step_time": 6.976102590560913
    },
    {
      "epoch": 8.311271667480469e-06,
      "model_forward_time": 1.9528658390045166,
      "step": 82715
    },
    {
      "epoch": 8.311271667480469e-06,
      "step": 82715,
      "training_step_time": 6.94887638092041
    },
    {
      "epoch": 8.312225341796874e-06,
      "model_forward_time": 1.9476280212402344,
      "step": 82716
    },
    {
      "epoch": 8.312225341796874e-06,
      "step": 82716,
      "training_step_time": 10.462859392166138
    },
    {
      "epoch": 8.313179016113282e-06,
      "model_forward_time": 2.934508800506592,
      "step": 82717
    },
    {
      "epoch": 8.313179016113282e-06,
      "step": 82717,
      "training_step_time": 8.932937622070312
    },
    {
      "epoch": 8.314132690429687e-06,
      "model_forward_time": 1.9307050704956055,
      "step": 82718
    },
    {
      "epoch": 8.314132690429687e-06,
      "step": 82718,
      "training_step_time": 7.894827604293823
    },
    {
      "epoch": 8.315086364746093e-06,
      "model_forward_time": 1.9363677501678467,
      "step": 82719
    },
    {
      "epoch": 8.315086364746093e-06,
      "step": 82719,
      "training_step_time": 7.7104172706604
    },
    {
      "epoch": 8.3160400390625e-06,
      "grad_norm": 0.0936797559261322,
      "learning_rate": 7.944720198567379e-07,
      "loss": 0.0547,
      "step": 82720
    },
    {
      "epoch": 8.3160400390625e-06,
      "model_forward_time": 1.9407763481140137,
      "step": 82720
    },
    {
      "dynamics_loss_avg": 0.0565937552601099,
      "epoch": 8.3160400390625e-06,
      "step": 82720
    },
    {
      "action_loss_avg": 0.0016468670306494459,
      "epoch": 8.3160400390625e-06,
      "step": 82720
    },
    {
      "epoch": 8.3160400390625e-06,
      "step": 82720,
      "training_step_time": 7.55446982383728
    },
    {
      "epoch": 8.316993713378906e-06,
      "model_forward_time": 1.9387314319610596,
      "step": 82721
    },
    {
      "epoch": 8.316993713378906e-06,
      "step": 82721,
      "training_step_time": 7.7124104499816895
    },
    {
      "epoch": 8.317947387695312e-06,
      "model_forward_time": 1.9546630382537842,
      "step": 82722
    },
    {
      "epoch": 8.317947387695312e-06,
      "step": 82722,
      "training_step_time": 6.960034132003784
    },
    {
      "epoch": 8.31890106201172e-06,
      "model_forward_time": 1.9366416931152344,
      "step": 82723
    },
    {
      "epoch": 8.31890106201172e-06,
      "step": 82723,
      "training_step_time": 6.988150358200073
    },
    {
      "epoch": 8.319854736328125e-06,
      "model_forward_time": 1.9402616024017334,
      "step": 82724
    },
    {
      "epoch": 8.319854736328125e-06,
      "step": 82724,
      "training_step_time": 6.956101179122925
    },
    {
      "epoch": 8.320808410644531e-06,
      "model_forward_time": 1.9269332885742188,
      "step": 82725
    },
    {
      "epoch": 8.320808410644531e-06,
      "step": 82725,
      "training_step_time": 9.294689893722534
    },
    {
      "epoch": 8.321762084960938e-06,
      "model_forward_time": 2.935222625732422,
      "step": 82726
    },
    {
      "epoch": 8.321762084960938e-06,
      "step": 82726,
      "training_step_time": 8.550271034240723
    },
    {
      "epoch": 8.322715759277344e-06,
      "model_forward_time": 1.9475407600402832,
      "step": 82727
    },
    {
      "epoch": 8.322715759277344e-06,
      "step": 82727,
      "training_step_time": 7.828352451324463
    },
    {
      "epoch": 8.32366943359375e-06,
      "model_forward_time": 1.931131362915039,
      "step": 82728
    },
    {
      "epoch": 8.32366943359375e-06,
      "step": 82728,
      "training_step_time": 7.725298643112183
    },
    {
      "epoch": 8.324623107910155e-06,
      "model_forward_time": 1.9299182891845703,
      "step": 82729
    },
    {
      "epoch": 8.324623107910155e-06,
      "step": 82729,
      "training_step_time": 7.523770332336426
    },
    {
      "epoch": 8.325576782226563e-06,
      "grad_norm": 0.0703529566526413,
      "learning_rate": 7.935779360496337e-07,
      "loss": 0.0556,
      "step": 82730
    },
    {
      "epoch": 8.325576782226563e-06,
      "model_forward_time": 1.9443705081939697,
      "step": 82730
    },
    {
      "dynamics_loss_avg": 0.05148772383108735,
      "epoch": 8.325576782226563e-06,
      "step": 82730
    },
    {
      "action_loss_avg": 0.001252286319504492,
      "epoch": 8.325576782226563e-06,
      "step": 82730
    },
    {
      "epoch": 8.325576782226563e-06,
      "step": 82730,
      "training_step_time": 7.793571472167969
    },
    {
      "epoch": 8.326530456542969e-06,
      "model_forward_time": 1.9311165809631348,
      "step": 82731
    },
    {
      "epoch": 8.326530456542969e-06,
      "step": 82731,
      "training_step_time": 6.900117635726929
    },
    {
      "epoch": 8.327484130859374e-06,
      "model_forward_time": 1.9412806034088135,
      "step": 82732
    },
    {
      "epoch": 8.327484130859374e-06,
      "step": 82732,
      "training_step_time": 6.947683334350586
    },
    {
      "epoch": 8.328437805175782e-06,
      "model_forward_time": 1.9352753162384033,
      "step": 82733
    },
    {
      "epoch": 8.328437805175782e-06,
      "step": 82733,
      "training_step_time": 6.901579856872559
    },
    {
      "epoch": 8.329391479492187e-06,
      "model_forward_time": 1.9569244384765625,
      "step": 82734
    },
    {
      "epoch": 8.329391479492187e-06,
      "step": 82734,
      "training_step_time": 9.389109373092651
    },
    {
      "epoch": 8.330345153808593e-06,
      "model_forward_time": 1.2338497638702393,
      "step": 82735
    },
    {
      "epoch": 8.330345153808593e-06,
      "step": 82735,
      "training_step_time": 8.393663167953491
    },
    {
      "epoch": 8.331298828125e-06,
      "model_forward_time": 1.9361114501953125,
      "step": 82736
    },
    {
      "epoch": 8.331298828125e-06,
      "step": 82736,
      "training_step_time": 8.063120126724243
    },
    {
      "epoch": 8.332252502441406e-06,
      "model_forward_time": 1.9378314018249512,
      "step": 82737
    },
    {
      "epoch": 8.332252502441406e-06,
      "step": 82737,
      "training_step_time": 7.77443528175354
    },
    {
      "epoch": 8.333206176757812e-06,
      "model_forward_time": 1.898653268814087,
      "step": 82738
    },
    {
      "epoch": 8.333206176757812e-06,
      "step": 82738,
      "training_step_time": 7.6440489292144775
    },
    {
      "epoch": 8.33415985107422e-06,
      "model_forward_time": 1.9462573528289795,
      "step": 82739
    },
    {
      "epoch": 8.33415985107422e-06,
      "step": 82739,
      "training_step_time": 7.859004974365234
    },
    {
      "epoch": 8.335113525390625e-06,
      "grad_norm": 0.08354548364877701,
      "learning_rate": 7.926843122505229e-07,
      "loss": 0.0558,
      "step": 82740
    },
    {
      "epoch": 8.335113525390625e-06,
      "model_forward_time": 1.9450013637542725,
      "step": 82740
    },
    {
      "dynamics_loss_avg": 0.04796868078410625,
      "epoch": 8.335113525390625e-06,
      "step": 82740
    },
    {
      "action_loss_avg": 0.0019364906853297724,
      "epoch": 8.335113525390625e-06,
      "step": 82740
    },
    {
      "epoch": 8.335113525390625e-06,
      "step": 82740,
      "training_step_time": 6.908385515213013
    },
    {
      "epoch": 8.336067199707031e-06,
      "model_forward_time": 1.9130034446716309,
      "step": 82741
    },
    {
      "epoch": 8.336067199707031e-06,
      "step": 82741,
      "training_step_time": 6.899144649505615
    },
    {
      "epoch": 8.337020874023438e-06,
      "model_forward_time": 1.9295876026153564,
      "step": 82742
    },
    {
      "epoch": 8.337020874023438e-06,
      "step": 82742,
      "training_step_time": 6.93284273147583
    },
    {
      "epoch": 8.337974548339844e-06,
      "model_forward_time": 1.9296650886535645,
      "step": 82743
    },
    {
      "epoch": 8.337974548339844e-06,
      "step": 82743,
      "training_step_time": 9.927874565124512
    },
    {
      "epoch": 8.33892822265625e-06,
      "model_forward_time": 1.9289813041687012,
      "step": 82744
    },
    {
      "epoch": 8.33892822265625e-06,
      "step": 82744,
      "training_step_time": 8.476120233535767
    },
    {
      "epoch": 8.339881896972656e-06,
      "model_forward_time": 3.026247262954712,
      "step": 82745
    },
    {
      "epoch": 8.339881896972656e-06,
      "step": 82745,
      "training_step_time": 8.090968132019043
    },
    {
      "epoch": 8.340835571289063e-06,
      "model_forward_time": 1.9230806827545166,
      "step": 82746
    },
    {
      "epoch": 8.340835571289063e-06,
      "step": 82746,
      "training_step_time": 7.574537754058838
    },
    {
      "epoch": 8.341789245605469e-06,
      "model_forward_time": 1.9560260772705078,
      "step": 82747
    },
    {
      "epoch": 8.341789245605469e-06,
      "step": 82747,
      "training_step_time": 7.642985820770264
    },
    {
      "epoch": 8.342742919921874e-06,
      "model_forward_time": 1.9517409801483154,
      "step": 82748
    },
    {
      "epoch": 8.342742919921874e-06,
      "step": 82748,
      "training_step_time": 7.730242967605591
    },
    {
      "epoch": 8.343696594238282e-06,
      "model_forward_time": 1.9477574825286865,
      "step": 82749
    },
    {
      "epoch": 8.343696594238282e-06,
      "step": 82749,
      "training_step_time": 7.8905251026153564
    },
    {
      "epoch": 8.344650268554688e-06,
      "grad_norm": 0.09768829494714737,
      "learning_rate": 7.917911485571289e-07,
      "loss": 0.0557,
      "step": 82750
    },
    {
      "epoch": 8.344650268554688e-06,
      "model_forward_time": 1.9166104793548584,
      "step": 82750
    },
    {
      "dynamics_loss_avg": 0.05091212913393974,
      "epoch": 8.344650268554688e-06,
      "step": 82750
    },
    {
      "action_loss_avg": 0.0016873718006536365,
      "epoch": 8.344650268554688e-06,
      "step": 82750
    },
    {
      "epoch": 8.344650268554688e-06,
      "step": 82750,
      "training_step_time": 7.078755855560303
    },
    {
      "epoch": 8.345603942871093e-06,
      "model_forward_time": 1.92950439453125,
      "step": 82751
    },
    {
      "epoch": 8.345603942871093e-06,
      "step": 82751,
      "training_step_time": 6.919660329818726
    },
    {
      "epoch": 8.3465576171875e-06,
      "model_forward_time": 1.9575583934783936,
      "step": 82752
    },
    {
      "epoch": 8.3465576171875e-06,
      "step": 82752,
      "training_step_time": 10.25053071975708
    },
    {
      "epoch": 8.347511291503906e-06,
      "model_forward_time": 1.9482395648956299,
      "step": 82753
    },
    {
      "epoch": 8.347511291503906e-06,
      "step": 82753,
      "training_step_time": 9.525573968887329
    },
    {
      "epoch": 8.348464965820312e-06,
      "model_forward_time": 2.961094856262207,
      "step": 82754
    },
    {
      "epoch": 8.348464965820312e-06,
      "step": 82754,
      "training_step_time": 7.925424575805664
    },
    {
      "epoch": 8.34941864013672e-06,
      "model_forward_time": 1.220609426498413,
      "step": 82755
    },
    {
      "epoch": 8.34941864013672e-06,
      "step": 82755,
      "training_step_time": 8.628445863723755
    },
    {
      "epoch": 8.350372314453125e-06,
      "model_forward_time": 1.901923418045044,
      "step": 82756
    },
    {
      "epoch": 8.350372314453125e-06,
      "step": 82756,
      "training_step_time": 7.647063970565796
    },
    {
      "epoch": 8.351325988769531e-06,
      "model_forward_time": 1.9429218769073486,
      "step": 82757
    },
    {
      "epoch": 8.351325988769531e-06,
      "step": 82757,
      "training_step_time": 7.681987524032593
    },
    {
      "epoch": 8.352279663085937e-06,
      "model_forward_time": 1.9285833835601807,
      "step": 82758
    },
    {
      "epoch": 8.352279663085937e-06,
      "step": 82758,
      "training_step_time": 7.93390417098999
    },
    {
      "epoch": 8.353233337402344e-06,
      "model_forward_time": 1.950472354888916,
      "step": 82759
    },
    {
      "epoch": 8.353233337402344e-06,
      "step": 82759,
      "training_step_time": 6.8923728466033936
    },
    {
      "epoch": 8.35418701171875e-06,
      "grad_norm": 0.09169302135705948,
      "learning_rate": 7.908984450671281e-07,
      "loss": 0.0554,
      "step": 82760
    },
    {
      "epoch": 8.35418701171875e-06,
      "model_forward_time": 1.9262747764587402,
      "step": 82760
    },
    {
      "dynamics_loss_avg": 0.07008925918489695,
      "epoch": 8.35418701171875e-06,
      "step": 82760
    },
    {
      "action_loss_avg": 0.0010256603738525883,
      "epoch": 8.35418701171875e-06,
      "step": 82760
    },
    {
      "epoch": 8.35418701171875e-06,
      "step": 82760,
      "training_step_time": 6.931041717529297
    },
    {
      "epoch": 8.355140686035156e-06,
      "model_forward_time": 1.9470691680908203,
      "step": 82761
    },
    {
      "epoch": 8.355140686035156e-06,
      "step": 82761,
      "training_step_time": 8.581363916397095
    },
    {
      "epoch": 8.356094360351563e-06,
      "model_forward_time": 1.9434866905212402,
      "step": 82762
    },
    {
      "epoch": 8.356094360351563e-06,
      "step": 82762,
      "training_step_time": 7.950760364532471
    },
    {
      "epoch": 8.357048034667969e-06,
      "model_forward_time": 1.9212422370910645,
      "step": 82763
    },
    {
      "epoch": 8.357048034667969e-06,
      "step": 82763,
      "training_step_time": 10.543380498886108
    },
    {
      "epoch": 8.358001708984374e-06,
      "model_forward_time": 1.955040454864502,
      "step": 82764
    },
    {
      "epoch": 8.358001708984374e-06,
      "step": 82764,
      "training_step_time": 7.8074257373809814
    },
    {
      "epoch": 8.358955383300782e-06,
      "model_forward_time": 1.9542851448059082,
      "step": 82765
    },
    {
      "epoch": 8.358955383300782e-06,
      "step": 82765,
      "training_step_time": 7.6723411083221436
    },
    {
      "epoch": 8.359909057617188e-06,
      "model_forward_time": 1.957228183746338,
      "step": 82766
    },
    {
      "epoch": 8.359909057617188e-06,
      "step": 82766,
      "training_step_time": 7.718871593475342
    },
    {
      "epoch": 8.360862731933593e-06,
      "model_forward_time": 1.948801040649414,
      "step": 82767
    },
    {
      "epoch": 8.360862731933593e-06,
      "step": 82767,
      "training_step_time": 7.803196907043457
    },
    {
      "epoch": 8.36181640625e-06,
      "model_forward_time": 1.950838565826416,
      "step": 82768
    },
    {
      "epoch": 8.36181640625e-06,
      "step": 82768,
      "training_step_time": 6.893806457519531
    },
    {
      "epoch": 8.362770080566406e-06,
      "model_forward_time": 1.9465701580047607,
      "step": 82769
    },
    {
      "epoch": 8.362770080566406e-06,
      "step": 82769,
      "training_step_time": 6.947880029678345
    },
    {
      "epoch": 8.363723754882812e-06,
      "grad_norm": 0.10312370955944061,
      "learning_rate": 7.900062018781463e-07,
      "loss": 0.0565,
      "step": 82770
    },
    {
      "epoch": 8.363723754882812e-06,
      "model_forward_time": 1.9405102729797363,
      "step": 82770
    },
    {
      "dynamics_loss_avg": 0.05972636509686709,
      "epoch": 8.363723754882812e-06,
      "step": 82770
    },
    {
      "action_loss_avg": 0.001297156437067315,
      "epoch": 8.363723754882812e-06,
      "step": 82770
    },
    {
      "epoch": 8.363723754882812e-06,
      "step": 82770,
      "training_step_time": 7.8911731243133545
    },
    {
      "epoch": 8.36467742919922e-06,
      "model_forward_time": 1.9323525428771973,
      "step": 82771
    },
    {
      "epoch": 8.36467742919922e-06,
      "step": 82771,
      "training_step_time": 9.257299423217773
    },
    {
      "epoch": 8.365631103515625e-06,
      "model_forward_time": 1.9262089729309082,
      "step": 82772
    },
    {
      "epoch": 8.365631103515625e-06,
      "step": 82772,
      "training_step_time": 9.487972736358643
    },
    {
      "epoch": 8.366584777832031e-06,
      "model_forward_time": 2.968313455581665,
      "step": 82773
    },
    {
      "epoch": 8.366584777832031e-06,
      "step": 82773,
      "training_step_time": 9.135632991790771
    },
    {
      "epoch": 8.367538452148437e-06,
      "model_forward_time": 1.946664571762085,
      "step": 82774
    },
    {
      "epoch": 8.367538452148437e-06,
      "step": 82774,
      "training_step_time": 8.03040337562561
    },
    {
      "epoch": 8.368492126464844e-06,
      "model_forward_time": 1.954117774963379,
      "step": 82775
    },
    {
      "epoch": 8.368492126464844e-06,
      "step": 82775,
      "training_step_time": 7.6960108280181885
    },
    {
      "epoch": 8.36944580078125e-06,
      "model_forward_time": 1.9543805122375488,
      "step": 82776
    },
    {
      "epoch": 8.36944580078125e-06,
      "step": 82776,
      "training_step_time": 7.598305702209473
    },
    {
      "epoch": 8.370399475097656e-06,
      "model_forward_time": 1.9536669254302979,
      "step": 82777
    },
    {
      "epoch": 8.370399475097656e-06,
      "step": 82777,
      "training_step_time": 6.940498352050781
    },
    {
      "epoch": 8.371353149414063e-06,
      "model_forward_time": 1.955467939376831,
      "step": 82778
    },
    {
      "epoch": 8.371353149414063e-06,
      "step": 82778,
      "training_step_time": 6.936229705810547
    },
    {
      "epoch": 8.372306823730469e-06,
      "model_forward_time": 1.9515423774719238,
      "step": 82779
    },
    {
      "epoch": 8.372306823730469e-06,
      "step": 82779,
      "training_step_time": 8.623706340789795
    },
    {
      "epoch": 8.373260498046874e-06,
      "grad_norm": 0.09427713602781296,
      "learning_rate": 7.891144190877554e-07,
      "loss": 0.0554,
      "step": 82780
    },
    {
      "epoch": 8.373260498046874e-06,
      "model_forward_time": 1.957418441772461,
      "step": 82780
    },
    {
      "dynamics_loss_avg": 0.07877924777567387,
      "epoch": 8.373260498046874e-06,
      "step": 82780
    },
    {
      "action_loss_avg": 0.002709400653839111,
      "epoch": 8.373260498046874e-06,
      "step": 82780
    },
    {
      "epoch": 8.373260498046874e-06,
      "step": 82780,
      "training_step_time": 9.649140357971191
    },
    {
      "epoch": 8.374214172363282e-06,
      "model_forward_time": 1.963010311126709,
      "step": 82781
    },
    {
      "epoch": 8.374214172363282e-06,
      "step": 82781,
      "training_step_time": 10.712143898010254
    },
    {
      "epoch": 8.375167846679688e-06,
      "model_forward_time": 3.0578010082244873,
      "step": 82782
    },
    {
      "epoch": 8.375167846679688e-06,
      "step": 82782,
      "training_step_time": 8.02613377571106
    },
    {
      "epoch": 8.376121520996093e-06,
      "model_forward_time": 1.9324872493743896,
      "step": 82783
    },
    {
      "epoch": 8.376121520996093e-06,
      "step": 82783,
      "training_step_time": 7.821511268615723
    },
    {
      "epoch": 8.3770751953125e-06,
      "model_forward_time": 1.945047378540039,
      "step": 82784
    },
    {
      "epoch": 8.3770751953125e-06,
      "step": 82784,
      "training_step_time": 7.698984146118164
    },
    {
      "epoch": 8.378028869628906e-06,
      "model_forward_time": 1.947404384613037,
      "step": 82785
    },
    {
      "epoch": 8.378028869628906e-06,
      "step": 82785,
      "training_step_time": 7.453197956085205
    },
    {
      "epoch": 8.378982543945312e-06,
      "model_forward_time": 1.9512088298797607,
      "step": 82786
    },
    {
      "epoch": 8.378982543945312e-06,
      "step": 82786,
      "training_step_time": 7.9932990074157715
    },
    {
      "epoch": 8.37993621826172e-06,
      "model_forward_time": 1.9319145679473877,
      "step": 82787
    },
    {
      "epoch": 8.37993621826172e-06,
      "step": 82787,
      "training_step_time": 6.9001405239105225
    },
    {
      "epoch": 8.380889892578125e-06,
      "model_forward_time": 1.9571359157562256,
      "step": 82788
    },
    {
      "epoch": 8.380889892578125e-06,
      "step": 82788,
      "training_step_time": 7.777235269546509
    },
    {
      "epoch": 8.381843566894531e-06,
      "model_forward_time": 1.960256814956665,
      "step": 82789
    },
    {
      "epoch": 8.381843566894531e-06,
      "step": 82789,
      "training_step_time": 7.823272705078125
    },
    {
      "epoch": 8.382797241210937e-06,
      "grad_norm": 0.09136181324720383,
      "learning_rate": 7.882230967934806e-07,
      "loss": 0.0555,
      "step": 82790
    },
    {
      "epoch": 8.382797241210937e-06,
      "model_forward_time": 1.9236464500427246,
      "step": 82790
    },
    {
      "dynamics_loss_avg": 0.05486060562543571,
      "epoch": 8.382797241210937e-06,
      "step": 82790
    },
    {
      "action_loss_avg": 0.0010195014823693783,
      "epoch": 8.382797241210937e-06,
      "step": 82790
    },
    {
      "epoch": 8.382797241210937e-06,
      "step": 82790,
      "training_step_time": 8.979631423950195
    },
    {
      "epoch": 8.383750915527344e-06,
      "model_forward_time": 3.0008857250213623,
      "step": 82791
    },
    {
      "epoch": 8.383750915527344e-06,
      "step": 82791,
      "training_step_time": 7.912597179412842
    },
    {
      "epoch": 8.38470458984375e-06,
      "model_forward_time": 1.9222278594970703,
      "step": 82792
    },
    {
      "epoch": 8.38470458984375e-06,
      "step": 82792,
      "training_step_time": 7.867695331573486
    },
    {
      "epoch": 8.385658264160156e-06,
      "model_forward_time": 1.934892177581787,
      "step": 82793
    },
    {
      "epoch": 8.385658264160156e-06,
      "step": 82793,
      "training_step_time": 7.723446607589722
    },
    {
      "epoch": 8.386611938476563e-06,
      "model_forward_time": 1.9375114440917969,
      "step": 82794
    },
    {
      "epoch": 8.386611938476563e-06,
      "step": 82794,
      "training_step_time": 7.399255752563477
    },
    {
      "epoch": 8.387565612792969e-06,
      "model_forward_time": 1.952446699142456,
      "step": 82795
    },
    {
      "epoch": 8.387565612792969e-06,
      "step": 82795,
      "training_step_time": 7.729478120803833
    },
    {
      "epoch": 8.388519287109374e-06,
      "model_forward_time": 1.9550848007202148,
      "step": 82796
    },
    {
      "epoch": 8.388519287109374e-06,
      "step": 82796,
      "training_step_time": 6.906572341918945
    },
    {
      "epoch": 8.389472961425782e-06,
      "model_forward_time": 1.9423408508300781,
      "step": 82797
    },
    {
      "epoch": 8.389472961425782e-06,
      "step": 82797,
      "training_step_time": 7.750545978546143
    },
    {
      "epoch": 8.390426635742188e-06,
      "model_forward_time": 1.9243645668029785,
      "step": 82798
    },
    {
      "epoch": 8.390426635742188e-06,
      "step": 82798,
      "training_step_time": 8.526182413101196
    },
    {
      "epoch": 8.391380310058593e-06,
      "model_forward_time": 1.966393232345581,
      "step": 82799
    },
    {
      "epoch": 8.391380310058593e-06,
      "step": 82799,
      "training_step_time": 11.159006357192993
    },
    {
      "epoch": 8.392333984375e-06,
      "grad_norm": 0.10034924745559692,
      "learning_rate": 7.873322350927959e-07,
      "loss": 0.0556,
      "step": 82800
    },
    {
      "epoch": 8.392333984375e-06,
      "model_forward_time": 3.028999090194702,
      "step": 82800
    },
    {
      "dynamics_loss_avg": 0.046692154929041864,
      "epoch": 8.392333984375e-06,
      "step": 82800
    },
    {
      "action_loss_avg": 0.0013887305394746363,
      "epoch": 8.392333984375e-06,
      "step": 82800
    },
    {
      "epoch": 8.392333984375e-06,
      "step": 82800,
      "training_step_time": 10.386412620544434
    },
    {
      "epoch": 8.393287658691406e-06,
      "model_forward_time": 1.9357290267944336,
      "step": 82801
    },
    {
      "epoch": 8.393287658691406e-06,
      "step": 82801,
      "training_step_time": 7.8173675537109375
    },
    {
      "epoch": 8.394241333007812e-06,
      "model_forward_time": 1.573622465133667,
      "step": 82802
    },
    {
      "epoch": 8.394241333007812e-06,
      "step": 82802,
      "training_step_time": 7.715199947357178
    },
    {
      "epoch": 8.39519500732422e-06,
      "model_forward_time": 1.9393901824951172,
      "step": 82803
    },
    {
      "epoch": 8.39519500732422e-06,
      "step": 82803,
      "training_step_time": 7.197680711746216
    },
    {
      "epoch": 8.396148681640625e-06,
      "model_forward_time": 1.92753267288208,
      "step": 82804
    },
    {
      "epoch": 8.396148681640625e-06,
      "step": 82804,
      "training_step_time": 7.883230209350586
    },
    {
      "epoch": 8.397102355957031e-06,
      "model_forward_time": 1.9319696426391602,
      "step": 82805
    },
    {
      "epoch": 8.397102355957031e-06,
      "step": 82805,
      "training_step_time": 6.906367540359497
    },
    {
      "epoch": 8.398056030273437e-06,
      "model_forward_time": 1.934999704360962,
      "step": 82806
    },
    {
      "epoch": 8.398056030273437e-06,
      "step": 82806,
      "training_step_time": 7.721242666244507
    },
    {
      "epoch": 8.399009704589844e-06,
      "model_forward_time": 1.9358577728271484,
      "step": 82807
    },
    {
      "epoch": 8.399009704589844e-06,
      "step": 82807,
      "training_step_time": 7.822201251983643
    },
    {
      "epoch": 8.39996337890625e-06,
      "model_forward_time": 1.9499688148498535,
      "step": 82808
    },
    {
      "epoch": 8.39996337890625e-06,
      "step": 82808,
      "training_step_time": 9.330608367919922
    },
    {
      "epoch": 8.400917053222656e-06,
      "model_forward_time": 3.009390115737915,
      "step": 82809
    },
    {
      "epoch": 8.400917053222656e-06,
      "step": 82809,
      "training_step_time": 8.696716785430908
    },
    {
      "epoch": 8.401870727539063e-06,
      "grad_norm": 0.07372766733169556,
      "learning_rate": 7.864418340831254e-07,
      "loss": 0.057,
      "step": 82810
    },
    {
      "epoch": 8.401870727539063e-06,
      "model_forward_time": 1.9327993392944336,
      "step": 82810
    },
    {
      "dynamics_loss_avg": 0.054095225594937804,
      "epoch": 8.401870727539063e-06,
      "step": 82810
    },
    {
      "action_loss_avg": 0.0013781169604044408,
      "epoch": 8.401870727539063e-06,
      "step": 82810
    },
    {
      "epoch": 8.401870727539063e-06,
      "step": 82810,
      "training_step_time": 7.731894493103027
    },
    {
      "epoch": 8.402824401855469e-06,
      "model_forward_time": 1.9413418769836426,
      "step": 82811
    },
    {
      "epoch": 8.402824401855469e-06,
      "step": 82811,
      "training_step_time": 7.704822301864624
    },
    {
      "epoch": 8.403778076171874e-06,
      "model_forward_time": 1.9486231803894043,
      "step": 82812
    },
    {
      "epoch": 8.403778076171874e-06,
      "step": 82812,
      "training_step_time": 7.847866773605347
    },
    {
      "epoch": 8.404731750488282e-06,
      "model_forward_time": 1.9185009002685547,
      "step": 82813
    },
    {
      "epoch": 8.404731750488282e-06,
      "step": 82813,
      "training_step_time": 7.990915536880493
    },
    {
      "epoch": 8.405685424804688e-06,
      "model_forward_time": 1.9410362243652344,
      "step": 82814
    },
    {
      "epoch": 8.405685424804688e-06,
      "step": 82814,
      "training_step_time": 6.955160140991211
    },
    {
      "epoch": 8.406639099121093e-06,
      "model_forward_time": 1.902512550354004,
      "step": 82815
    },
    {
      "epoch": 8.406639099121093e-06,
      "step": 82815,
      "training_step_time": 8.57575535774231
    },
    {
      "epoch": 8.4075927734375e-06,
      "model_forward_time": 1.911081314086914,
      "step": 82816
    },
    {
      "epoch": 8.4075927734375e-06,
      "step": 82816,
      "training_step_time": 7.886899471282959
    },
    {
      "epoch": 8.408546447753906e-06,
      "model_forward_time": 1.9323482513427734,
      "step": 82817
    },
    {
      "epoch": 8.408546447753906e-06,
      "step": 82817,
      "training_step_time": 10.711831092834473
    },
    {
      "epoch": 8.409500122070312e-06,
      "model_forward_time": 2.9760942459106445,
      "step": 82818
    },
    {
      "epoch": 8.409500122070312e-06,
      "step": 82818,
      "training_step_time": 9.094897985458374
    },
    {
      "epoch": 8.41045379638672e-06,
      "model_forward_time": 1.9343478679656982,
      "step": 82819
    },
    {
      "epoch": 8.41045379638672e-06,
      "step": 82819,
      "training_step_time": 8.716787815093994
    },
    {
      "epoch": 8.411407470703125e-06,
      "grad_norm": 0.09317129850387573,
      "learning_rate": 7.855518938618395e-07,
      "loss": 0.0572,
      "step": 82820
    },
    {
      "epoch": 8.411407470703125e-06,
      "model_forward_time": 1.9531948566436768,
      "step": 82820
    },
    {
      "dynamics_loss_avg": 0.05924572553485632,
      "epoch": 8.411407470703125e-06,
      "step": 82820
    },
    {
      "action_loss_avg": 0.001350321518839337,
      "epoch": 8.411407470703125e-06,
      "step": 82820
    },
    {
      "epoch": 8.411407470703125e-06,
      "step": 82820,
      "training_step_time": 9.051219463348389
    },
    {
      "epoch": 8.412361145019531e-06,
      "model_forward_time": 1.9412696361541748,
      "step": 82821
    },
    {
      "epoch": 8.412361145019531e-06,
      "step": 82821,
      "training_step_time": 7.761067867279053
    },
    {
      "epoch": 8.413314819335937e-06,
      "model_forward_time": 1.9399347305297852,
      "step": 82822
    },
    {
      "epoch": 8.413314819335937e-06,
      "step": 82822,
      "training_step_time": 7.55125617980957
    },
    {
      "epoch": 8.414268493652344e-06,
      "model_forward_time": 1.915966510772705,
      "step": 82823
    },
    {
      "epoch": 8.414268493652344e-06,
      "step": 82823,
      "training_step_time": 8.068914413452148
    },
    {
      "epoch": 8.41522216796875e-06,
      "model_forward_time": 1.9276063442230225,
      "step": 82824
    },
    {
      "epoch": 8.41522216796875e-06,
      "step": 82824,
      "training_step_time": 7.8145036697387695
    },
    {
      "epoch": 8.416175842285156e-06,
      "model_forward_time": 1.927173137664795,
      "step": 82825
    },
    {
      "epoch": 8.416175842285156e-06,
      "step": 82825,
      "training_step_time": 7.834685325622559
    },
    {
      "epoch": 8.417129516601563e-06,
      "model_forward_time": 1.944572925567627,
      "step": 82826
    },
    {
      "epoch": 8.417129516601563e-06,
      "step": 82826,
      "training_step_time": 9.360318183898926
    },
    {
      "epoch": 8.418083190917969e-06,
      "model_forward_time": 1.9114739894866943,
      "step": 82827
    },
    {
      "epoch": 8.418083190917969e-06,
      "step": 82827,
      "training_step_time": 9.378708124160767
    },
    {
      "epoch": 8.419036865234374e-06,
      "model_forward_time": 2.961177349090576,
      "step": 82828
    },
    {
      "epoch": 8.419036865234374e-06,
      "step": 82828,
      "training_step_time": 7.913024663925171
    },
    {
      "epoch": 8.419990539550782e-06,
      "model_forward_time": 1.9062368869781494,
      "step": 82829
    },
    {
      "epoch": 8.419990539550782e-06,
      "step": 82829,
      "training_step_time": 8.03542685508728
    },
    {
      "epoch": 8.420944213867188e-06,
      "grad_norm": 0.08045908808708191,
      "learning_rate": 7.846624145262633e-07,
      "loss": 0.0554,
      "step": 82830
    },
    {
      "epoch": 8.420944213867188e-06,
      "model_forward_time": 1.965406894683838,
      "step": 82830
    },
    {
      "dynamics_loss_avg": 0.054328550770878795,
      "epoch": 8.420944213867188e-06,
      "step": 82830
    },
    {
      "action_loss_avg": 0.0013296023913426324,
      "epoch": 8.420944213867188e-06,
      "step": 82830
    },
    {
      "epoch": 8.420944213867188e-06,
      "step": 82830,
      "training_step_time": 7.768527984619141
    },
    {
      "epoch": 8.421897888183593e-06,
      "model_forward_time": 1.9522595405578613,
      "step": 82831
    },
    {
      "epoch": 8.421897888183593e-06,
      "step": 82831,
      "training_step_time": 7.461994171142578
    },
    {
      "epoch": 8.4228515625e-06,
      "model_forward_time": 1.9326794147491455,
      "step": 82832
    },
    {
      "epoch": 8.4228515625e-06,
      "step": 82832,
      "training_step_time": 8.114262580871582
    },
    {
      "epoch": 8.423805236816406e-06,
      "model_forward_time": 1.9150567054748535,
      "step": 82833
    },
    {
      "epoch": 8.423805236816406e-06,
      "step": 82833,
      "training_step_time": 8.756421327590942
    },
    {
      "epoch": 8.424758911132812e-06,
      "model_forward_time": 1.9468200206756592,
      "step": 82834
    },
    {
      "epoch": 8.424758911132812e-06,
      "step": 82834,
      "training_step_time": 7.795938968658447
    },
    {
      "epoch": 8.42571258544922e-06,
      "model_forward_time": 1.916745901107788,
      "step": 82835
    },
    {
      "epoch": 8.42571258544922e-06,
      "step": 82835,
      "training_step_time": 9.09445309638977
    },
    {
      "epoch": 8.426666259765625e-06,
      "model_forward_time": 1.9438223838806152,
      "step": 82836
    },
    {
      "epoch": 8.426666259765625e-06,
      "step": 82836,
      "training_step_time": 8.563902378082275
    },
    {
      "epoch": 8.427619934082031e-06,
      "model_forward_time": 2.997022867202759,
      "step": 82837
    },
    {
      "epoch": 8.427619934082031e-06,
      "step": 82837,
      "training_step_time": 7.928326845169067
    },
    {
      "epoch": 8.428573608398437e-06,
      "model_forward_time": 1.9251255989074707,
      "step": 82838
    },
    {
      "epoch": 8.428573608398437e-06,
      "step": 82838,
      "training_step_time": 8.0529146194458
    },
    {
      "epoch": 8.429527282714844e-06,
      "model_forward_time": 1.950646162033081,
      "step": 82839
    },
    {
      "epoch": 8.429527282714844e-06,
      "step": 82839,
      "training_step_time": 7.75988245010376
    },
    {
      "epoch": 8.43048095703125e-06,
      "grad_norm": 0.07309241592884064,
      "learning_rate": 7.837733961736671e-07,
      "loss": 0.0554,
      "step": 82840
    },
    {
      "epoch": 8.43048095703125e-06,
      "model_forward_time": 1.9250328540802002,
      "step": 82840
    },
    {
      "dynamics_loss_avg": 0.05686632730066776,
      "epoch": 8.43048095703125e-06,
      "step": 82840
    },
    {
      "action_loss_avg": 0.001009792843251489,
      "epoch": 8.43048095703125e-06,
      "step": 82840
    },
    {
      "epoch": 8.43048095703125e-06,
      "step": 82840,
      "training_step_time": 7.271753787994385
    },
    {
      "epoch": 8.431434631347656e-06,
      "model_forward_time": 1.9541311264038086,
      "step": 82841
    },
    {
      "epoch": 8.431434631347656e-06,
      "step": 82841,
      "training_step_time": 8.042567253112793
    },
    {
      "epoch": 8.432388305664063e-06,
      "model_forward_time": 1.5768210887908936,
      "step": 82842
    },
    {
      "epoch": 8.432388305664063e-06,
      "step": 82842,
      "training_step_time": 7.715933799743652
    },
    {
      "epoch": 8.433341979980469e-06,
      "model_forward_time": 1.935643196105957,
      "step": 82843
    },
    {
      "epoch": 8.433341979980469e-06,
      "step": 82843,
      "training_step_time": 7.810867071151733
    },
    {
      "epoch": 8.434295654296874e-06,
      "model_forward_time": 1.9234232902526855,
      "step": 82844
    },
    {
      "epoch": 8.434295654296874e-06,
      "step": 82844,
      "training_step_time": 8.548567533493042
    },
    {
      "epoch": 8.435249328613282e-06,
      "model_forward_time": 1.9346792697906494,
      "step": 82845
    },
    {
      "epoch": 8.435249328613282e-06,
      "step": 82845,
      "training_step_time": 11.064066648483276
    },
    {
      "epoch": 8.436203002929688e-06,
      "model_forward_time": 3.017106533050537,
      "step": 82846
    },
    {
      "epoch": 8.436203002929688e-06,
      "step": 82846,
      "training_step_time": 8.428781509399414
    },
    {
      "epoch": 8.437156677246093e-06,
      "model_forward_time": 1.94854736328125,
      "step": 82847
    },
    {
      "epoch": 8.437156677246093e-06,
      "step": 82847,
      "training_step_time": 8.882370710372925
    },
    {
      "epoch": 8.4381103515625e-06,
      "model_forward_time": 1.9515783786773682,
      "step": 82848
    },
    {
      "epoch": 8.4381103515625e-06,
      "step": 82848,
      "training_step_time": 7.7822465896606445
    },
    {
      "epoch": 8.439064025878906e-06,
      "model_forward_time": 0.886462926864624,
      "step": 82849
    },
    {
      "epoch": 8.439064025878906e-06,
      "step": 82849,
      "training_step_time": 7.209563493728638
    },
    {
      "epoch": 8.440017700195312e-06,
      "grad_norm": 0.12477702647447586,
      "learning_rate": 7.828848389012727e-07,
      "loss": 0.0558,
      "step": 82850
    },
    {
      "epoch": 8.440017700195312e-06,
      "model_forward_time": 1.9469904899597168,
      "step": 82850
    },
    {
      "dynamics_loss_avg": 0.0545501608401537,
      "epoch": 8.440017700195312e-06,
      "step": 82850
    },
    {
      "action_loss_avg": 0.0012583937263116241,
      "epoch": 8.440017700195312e-06,
      "step": 82850
    },
    {
      "epoch": 8.440017700195312e-06,
      "step": 82850,
      "training_step_time": 7.56616997718811
    },
    {
      "epoch": 8.44097137451172e-06,
      "model_forward_time": 1.9126274585723877,
      "step": 82851
    },
    {
      "epoch": 8.44097137451172e-06,
      "step": 82851,
      "training_step_time": 9.057413578033447
    },
    {
      "epoch": 8.441925048828125e-06,
      "model_forward_time": 1.9430625438690186,
      "step": 82852
    },
    {
      "epoch": 8.441925048828125e-06,
      "step": 82852,
      "training_step_time": 8.67637014389038
    },
    {
      "epoch": 8.442878723144531e-06,
      "model_forward_time": 1.9229755401611328,
      "step": 82853
    },
    {
      "epoch": 8.442878723144531e-06,
      "step": 82853,
      "training_step_time": 7.827871561050415
    },
    {
      "epoch": 8.443832397460937e-06,
      "model_forward_time": 1.945815086364746,
      "step": 82854
    },
    {
      "epoch": 8.443832397460937e-06,
      "step": 82854,
      "training_step_time": 8.975244522094727
    },
    {
      "epoch": 8.444786071777344e-06,
      "model_forward_time": 1.9211194515228271,
      "step": 82855
    },
    {
      "epoch": 8.444786071777344e-06,
      "step": 82855,
      "training_step_time": 7.895670175552368
    },
    {
      "epoch": 8.44573974609375e-06,
      "model_forward_time": 2.962747812271118,
      "step": 82856
    },
    {
      "epoch": 8.44573974609375e-06,
      "step": 82856,
      "training_step_time": 8.958431959152222
    },
    {
      "epoch": 8.446693420410156e-06,
      "model_forward_time": 1.909043550491333,
      "step": 82857
    },
    {
      "epoch": 8.446693420410156e-06,
      "step": 82857,
      "training_step_time": 7.873927116394043
    },
    {
      "epoch": 8.447647094726563e-06,
      "model_forward_time": 1.9337444305419922,
      "step": 82858
    },
    {
      "epoch": 8.447647094726563e-06,
      "step": 82858,
      "training_step_time": 7.782963752746582
    },
    {
      "epoch": 8.448600769042969e-06,
      "model_forward_time": 1.9314191341400146,
      "step": 82859
    },
    {
      "epoch": 8.448600769042969e-06,
      "step": 82859,
      "training_step_time": 7.5116894245147705
    },
    {
      "epoch": 8.449554443359374e-06,
      "grad_norm": 0.09183578938245773,
      "learning_rate": 7.819967428062525e-07,
      "loss": 0.0555,
      "step": 82860
    },
    {
      "epoch": 8.449554443359374e-06,
      "model_forward_time": 1.9417355060577393,
      "step": 82860
    },
    {
      "dynamics_loss_avg": 0.06435445975512266,
      "epoch": 8.449554443359374e-06,
      "step": 82860
    },
    {
      "action_loss_avg": 0.0017602962761884555,
      "epoch": 8.449554443359374e-06,
      "step": 82860
    },
    {
      "epoch": 8.449554443359374e-06,
      "step": 82860,
      "training_step_time": 8.064376592636108
    },
    {
      "epoch": 8.450508117675782e-06,
      "model_forward_time": 1.9168877601623535,
      "step": 82861
    },
    {
      "epoch": 8.450508117675782e-06,
      "step": 82861,
      "training_step_time": 7.739298343658447
    },
    {
      "epoch": 8.451461791992188e-06,
      "model_forward_time": 1.9181852340698242,
      "step": 82862
    },
    {
      "epoch": 8.451461791992188e-06,
      "step": 82862,
      "training_step_time": 7.769943475723267
    },
    {
      "epoch": 8.452415466308593e-06,
      "model_forward_time": 1.9295926094055176,
      "step": 82863
    },
    {
      "epoch": 8.452415466308593e-06,
      "step": 82863,
      "training_step_time": 10.134905099868774
    },
    {
      "epoch": 8.453369140625e-06,
      "model_forward_time": 1.9257750511169434,
      "step": 82864
    },
    {
      "epoch": 8.453369140625e-06,
      "step": 82864,
      "training_step_time": 8.72676134109497
    },
    {
      "epoch": 8.454322814941406e-06,
      "model_forward_time": 3.0173113346099854,
      "step": 82865
    },
    {
      "epoch": 8.454322814941406e-06,
      "step": 82865,
      "training_step_time": 7.941232919692993
    },
    {
      "epoch": 8.455276489257812e-06,
      "model_forward_time": 1.9400243759155273,
      "step": 82866
    },
    {
      "epoch": 8.455276489257812e-06,
      "step": 82866,
      "training_step_time": 8.03609561920166
    },
    {
      "epoch": 8.45623016357422e-06,
      "model_forward_time": 1.9296300411224365,
      "step": 82867
    },
    {
      "epoch": 8.45623016357422e-06,
      "step": 82867,
      "training_step_time": 7.739386320114136
    },
    {
      "epoch": 8.457183837890625e-06,
      "model_forward_time": 1.9465994834899902,
      "step": 82868
    },
    {
      "epoch": 8.457183837890625e-06,
      "step": 82868,
      "training_step_time": 7.602463245391846
    },
    {
      "epoch": 8.458137512207031e-06,
      "model_forward_time": 1.9479117393493652,
      "step": 82869
    },
    {
      "epoch": 8.458137512207031e-06,
      "step": 82869,
      "training_step_time": 8.086792469024658
    },
    {
      "epoch": 8.459091186523437e-06,
      "grad_norm": 0.07834415137767792,
      "learning_rate": 7.811091079857275e-07,
      "loss": 0.0575,
      "step": 82870
    },
    {
      "epoch": 8.459091186523437e-06,
      "model_forward_time": 1.9451701641082764,
      "step": 82870
    },
    {
      "dynamics_loss_avg": 0.06498840674757958,
      "epoch": 8.459091186523437e-06,
      "step": 82870
    },
    {
      "action_loss_avg": 0.0012217836803756655,
      "epoch": 8.459091186523437e-06,
      "step": 82870
    },
    {
      "epoch": 8.459091186523437e-06,
      "step": 82870,
      "training_step_time": 7.729448556900024
    },
    {
      "epoch": 8.460044860839844e-06,
      "model_forward_time": 1.9601893424987793,
      "step": 82871
    },
    {
      "epoch": 8.460044860839844e-06,
      "step": 82871,
      "training_step_time": 8.944001197814941
    },
    {
      "epoch": 8.46099853515625e-06,
      "model_forward_time": 1.9535987377166748,
      "step": 82872
    },
    {
      "epoch": 8.46099853515625e-06,
      "step": 82872,
      "training_step_time": 9.74509310722351
    },
    {
      "epoch": 8.461952209472656e-06,
      "model_forward_time": 1.9378292560577393,
      "step": 82873
    },
    {
      "epoch": 8.461952209472656e-06,
      "step": 82873,
      "training_step_time": 7.843091011047363
    },
    {
      "epoch": 8.462905883789063e-06,
      "model_forward_time": 2.9964497089385986,
      "step": 82874
    },
    {
      "epoch": 8.462905883789063e-06,
      "step": 82874,
      "training_step_time": 7.960249185562134
    },
    {
      "epoch": 8.463859558105469e-06,
      "model_forward_time": 1.9431571960449219,
      "step": 82875
    },
    {
      "epoch": 8.463859558105469e-06,
      "step": 82875,
      "training_step_time": 8.594078063964844
    },
    {
      "epoch": 8.464813232421874e-06,
      "model_forward_time": 1.9494757652282715,
      "step": 82876
    },
    {
      "epoch": 8.464813232421874e-06,
      "step": 82876,
      "training_step_time": 7.7823591232299805
    },
    {
      "epoch": 8.465766906738282e-06,
      "model_forward_time": 1.946568489074707,
      "step": 82877
    },
    {
      "epoch": 8.465766906738282e-06,
      "step": 82877,
      "training_step_time": 7.502299070358276
    },
    {
      "epoch": 8.466720581054688e-06,
      "model_forward_time": 1.9455842971801758,
      "step": 82878
    },
    {
      "epoch": 8.466720581054688e-06,
      "step": 82878,
      "training_step_time": 8.081474304199219
    },
    {
      "epoch": 8.467674255371093e-06,
      "model_forward_time": 1.9542028903961182,
      "step": 82879
    },
    {
      "epoch": 8.467674255371093e-06,
      "step": 82879,
      "training_step_time": 7.71741509437561
    },
    {
      "epoch": 8.4686279296875e-06,
      "grad_norm": 0.06454917043447495,
      "learning_rate": 7.802219345367668e-07,
      "loss": 0.0536,
      "step": 82880
    },
    {
      "epoch": 8.4686279296875e-06,
      "model_forward_time": 1.9453952312469482,
      "step": 82880
    },
    {
      "dynamics_loss_avg": 0.04432466309517622,
      "epoch": 8.4686279296875e-06,
      "step": 82880
    },
    {
      "action_loss_avg": 0.0015761643648147584,
      "epoch": 8.4686279296875e-06,
      "step": 82880
    },
    {
      "epoch": 8.4686279296875e-06,
      "step": 82880,
      "training_step_time": 8.419211387634277
    },
    {
      "epoch": 8.469581604003906e-06,
      "model_forward_time": 1.944929599761963,
      "step": 82881
    },
    {
      "epoch": 8.469581604003906e-06,
      "step": 82881,
      "training_step_time": 8.456625938415527
    },
    {
      "epoch": 8.470535278320312e-06,
      "model_forward_time": 1.9377357959747314,
      "step": 82882
    },
    {
      "epoch": 8.470535278320312e-06,
      "step": 82882,
      "training_step_time": 9.280780792236328
    },
    {
      "epoch": 8.471488952636718e-06,
      "model_forward_time": 2.9554996490478516,
      "step": 82883
    },
    {
      "epoch": 8.471488952636718e-06,
      "step": 82883,
      "training_step_time": 7.896513938903809
    },
    {
      "epoch": 8.472442626953125e-06,
      "model_forward_time": 1.9289917945861816,
      "step": 82884
    },
    {
      "epoch": 8.472442626953125e-06,
      "step": 82884,
      "training_step_time": 8.286736965179443
    },
    {
      "epoch": 8.473396301269531e-06,
      "model_forward_time": 1.9002423286437988,
      "step": 82885
    },
    {
      "epoch": 8.473396301269531e-06,
      "step": 82885,
      "training_step_time": 7.789738416671753
    },
    {
      "epoch": 8.474349975585937e-06,
      "model_forward_time": 1.9009339809417725,
      "step": 82886
    },
    {
      "epoch": 8.474349975585937e-06,
      "step": 82886,
      "training_step_time": 8.380545616149902
    },
    {
      "epoch": 8.475303649902344e-06,
      "model_forward_time": 1.9390285015106201,
      "step": 82887
    },
    {
      "epoch": 8.475303649902344e-06,
      "step": 82887,
      "training_step_time": 7.839949607849121
    },
    {
      "epoch": 8.47625732421875e-06,
      "model_forward_time": 1.9192030429840088,
      "step": 82888
    },
    {
      "epoch": 8.47625732421875e-06,
      "step": 82888,
      "training_step_time": 7.749967575073242
    },
    {
      "epoch": 8.477210998535156e-06,
      "model_forward_time": 1.923229455947876,
      "step": 82889
    },
    {
      "epoch": 8.477210998535156e-06,
      "step": 82889,
      "training_step_time": 8.60954213142395
    },
    {
      "epoch": 8.478164672851563e-06,
      "grad_norm": 0.11382584273815155,
      "learning_rate": 7.793352225563916e-07,
      "loss": 0.0535,
      "step": 82890
    },
    {
      "epoch": 8.478164672851563e-06,
      "model_forward_time": 1.932499885559082,
      "step": 82890
    },
    {
      "dynamics_loss_avg": 0.04225118961185217,
      "epoch": 8.478164672851563e-06,
      "step": 82890
    },
    {
      "action_loss_avg": 0.001004031050251797,
      "epoch": 8.478164672851563e-06,
      "step": 82890
    },
    {
      "epoch": 8.478164672851563e-06,
      "step": 82890,
      "training_step_time": 8.711533546447754
    },
    {
      "epoch": 8.479118347167969e-06,
      "model_forward_time": 1.8952882289886475,
      "step": 82891
    },
    {
      "epoch": 8.479118347167969e-06,
      "step": 82891,
      "training_step_time": 8.530863285064697
    },
    {
      "epoch": 8.480072021484374e-06,
      "model_forward_time": 1.92856764793396,
      "step": 82892
    },
    {
      "epoch": 8.480072021484374e-06,
      "step": 82892,
      "training_step_time": 10.613755941390991
    },
    {
      "epoch": 8.481025695800782e-06,
      "model_forward_time": 2.2146265506744385,
      "step": 82893
    },
    {
      "epoch": 8.481025695800782e-06,
      "step": 82893,
      "training_step_time": 7.515901803970337
    },
    {
      "epoch": 8.481979370117188e-06,
      "model_forward_time": 1.9312381744384766,
      "step": 82894
    },
    {
      "epoch": 8.481979370117188e-06,
      "step": 82894,
      "training_step_time": 8.029752016067505
    },
    {
      "epoch": 8.482933044433593e-06,
      "model_forward_time": 1.9381828308105469,
      "step": 82895
    },
    {
      "epoch": 8.482933044433593e-06,
      "step": 82895,
      "training_step_time": 7.821101665496826
    },
    {
      "epoch": 8.48388671875e-06,
      "model_forward_time": 1.5679986476898193,
      "step": 82896
    },
    {
      "epoch": 8.48388671875e-06,
      "step": 82896,
      "training_step_time": 7.222497940063477
    },
    {
      "epoch": 8.484840393066406e-06,
      "model_forward_time": 1.9199063777923584,
      "step": 82897
    },
    {
      "epoch": 8.484840393066406e-06,
      "step": 82897,
      "training_step_time": 7.638767957687378
    },
    {
      "epoch": 8.485794067382812e-06,
      "model_forward_time": 1.9349043369293213,
      "step": 82898
    },
    {
      "epoch": 8.485794067382812e-06,
      "step": 82898,
      "training_step_time": 7.710280895233154
    },
    {
      "epoch": 8.486747741699218e-06,
      "model_forward_time": 1.9284029006958008,
      "step": 82899
    },
    {
      "epoch": 8.486747741699218e-06,
      "step": 82899,
      "training_step_time": 7.9019129276275635
    },
    {
      "epoch": 8.487701416015625e-06,
      "grad_norm": 0.08932065218687057,
      "learning_rate": 7.7844897214157e-07,
      "loss": 0.0533,
      "step": 82900
    },
    {
      "epoch": 8.487701416015625e-06,
      "model_forward_time": 1.915839672088623,
      "step": 82900
    },
    {
      "dynamics_loss_avg": 0.05220204228535295,
      "epoch": 8.487701416015625e-06,
      "step": 82900
    },
    {
      "action_loss_avg": 0.001261687942314893,
      "epoch": 8.487701416015625e-06,
      "step": 82900
    },
    {
      "epoch": 8.487701416015625e-06,
      "step": 82900,
      "training_step_time": 8.109556436538696
    },
    {
      "epoch": 8.488655090332031e-06,
      "model_forward_time": 1.910445213317871,
      "step": 82901
    },
    {
      "epoch": 8.488655090332031e-06,
      "step": 82901,
      "training_step_time": 8.232000589370728
    },
    {
      "epoch": 8.489608764648437e-06,
      "model_forward_time": 2.959975481033325,
      "step": 82902
    },
    {
      "epoch": 8.489608764648437e-06,
      "step": 82902,
      "training_step_time": 7.916156530380249
    },
    {
      "epoch": 8.490562438964844e-06,
      "model_forward_time": 1.899280309677124,
      "step": 82903
    },
    {
      "epoch": 8.490562438964844e-06,
      "step": 82903,
      "training_step_time": 8.885610103607178
    },
    {
      "epoch": 8.49151611328125e-06,
      "model_forward_time": 1.9053499698638916,
      "step": 82904
    },
    {
      "epoch": 8.49151611328125e-06,
      "step": 82904,
      "training_step_time": 8.506211042404175
    },
    {
      "epoch": 8.492469787597656e-06,
      "model_forward_time": 1.9264178276062012,
      "step": 82905
    },
    {
      "epoch": 8.492469787597656e-06,
      "step": 82905,
      "training_step_time": 7.079830646514893
    },
    {
      "epoch": 8.493423461914063e-06,
      "model_forward_time": 1.906010627746582,
      "step": 82906
    },
    {
      "epoch": 8.493423461914063e-06,
      "step": 82906,
      "training_step_time": 7.744769334793091
    },
    {
      "epoch": 8.494377136230469e-06,
      "model_forward_time": 1.927565336227417,
      "step": 82907
    },
    {
      "epoch": 8.494377136230469e-06,
      "step": 82907,
      "training_step_time": 8.608193159103394
    },
    {
      "epoch": 8.495330810546874e-06,
      "model_forward_time": 1.9071249961853027,
      "step": 82908
    },
    {
      "epoch": 8.495330810546874e-06,
      "step": 82908,
      "training_step_time": 7.5560829639434814
    },
    {
      "epoch": 8.496284484863282e-06,
      "model_forward_time": 1.9075853824615479,
      "step": 82909
    },
    {
      "epoch": 8.496284484863282e-06,
      "step": 82909,
      "training_step_time": 9.696954011917114
    },
    {
      "epoch": 8.497238159179688e-06,
      "grad_norm": 0.08908224105834961,
      "learning_rate": 7.775631833892228e-07,
      "loss": 0.0535,
      "step": 82910
    },
    {
      "epoch": 8.497238159179688e-06,
      "model_forward_time": 1.9110016822814941,
      "step": 82910
    },
    {
      "dynamics_loss_avg": 0.06582087613642215,
      "epoch": 8.497238159179688e-06,
      "step": 82910
    },
    {
      "action_loss_avg": 0.001930140779586509,
      "epoch": 8.497238159179688e-06,
      "step": 82910
    },
    {
      "epoch": 8.497238159179688e-06,
      "step": 82910,
      "training_step_time": 8.200033187866211
    },
    {
      "epoch": 8.498191833496093e-06,
      "model_forward_time": 1.2192251682281494,
      "step": 82911
    },
    {
      "epoch": 8.498191833496093e-06,
      "step": 82911,
      "training_step_time": 8.502678632736206
    },
    {
      "epoch": 8.4991455078125e-06,
      "model_forward_time": 2.956202507019043,
      "step": 82912
    },
    {
      "epoch": 8.4991455078125e-06,
      "step": 82912,
      "training_step_time": 7.6690897941589355
    },
    {
      "epoch": 8.500099182128906e-06,
      "model_forward_time": 1.9090499877929688,
      "step": 82913
    },
    {
      "epoch": 8.500099182128906e-06,
      "step": 82913,
      "training_step_time": 7.782772064208984
    },
    {
      "epoch": 8.501052856445312e-06,
      "model_forward_time": 1.8965651988983154,
      "step": 82914
    },
    {
      "epoch": 8.501052856445312e-06,
      "step": 82914,
      "training_step_time": 7.32868766784668
    },
    {
      "epoch": 8.502006530761718e-06,
      "model_forward_time": 1.9223363399505615,
      "step": 82915
    },
    {
      "epoch": 8.502006530761718e-06,
      "step": 82915,
      "training_step_time": 7.764026403427124
    },
    {
      "epoch": 8.502960205078125e-06,
      "model_forward_time": 1.9429748058319092,
      "step": 82916
    },
    {
      "epoch": 8.502960205078125e-06,
      "step": 82916,
      "training_step_time": 8.600762128829956
    },
    {
      "epoch": 8.503913879394531e-06,
      "model_forward_time": 1.9376561641693115,
      "step": 82917
    },
    {
      "epoch": 8.503913879394531e-06,
      "step": 82917,
      "training_step_time": 7.53546929359436
    },
    {
      "epoch": 8.504867553710937e-06,
      "model_forward_time": 1.895484447479248,
      "step": 82918
    },
    {
      "epoch": 8.504867553710937e-06,
      "step": 82918,
      "training_step_time": 8.67335057258606
    },
    {
      "epoch": 8.505821228027344e-06,
      "model_forward_time": 1.9144201278686523,
      "step": 82919
    },
    {
      "epoch": 8.505821228027344e-06,
      "step": 82919,
      "training_step_time": 9.356114387512207
    },
    {
      "epoch": 8.50677490234375e-06,
      "grad_norm": 0.13203410804271698,
      "learning_rate": 7.766778563962152e-07,
      "loss": 0.0533,
      "step": 82920
    },
    {
      "epoch": 8.50677490234375e-06,
      "model_forward_time": 1.9281349182128906,
      "step": 82920
    },
    {
      "dynamics_loss_avg": 0.04893459416925907,
      "epoch": 8.50677490234375e-06,
      "step": 82920
    },
    {
      "action_loss_avg": 0.0013840639527188613,
      "epoch": 8.50677490234375e-06,
      "step": 82920
    },
    {
      "epoch": 8.50677490234375e-06,
      "step": 82920,
      "training_step_time": 8.51817512512207
    },
    {
      "epoch": 8.507728576660156e-06,
      "model_forward_time": 2.9691004753112793,
      "step": 82921
    },
    {
      "epoch": 8.507728576660156e-06,
      "step": 82921,
      "training_step_time": 7.707513809204102
    },
    {
      "epoch": 8.508682250976563e-06,
      "model_forward_time": 1.5778214931488037,
      "step": 82922
    },
    {
      "epoch": 8.508682250976563e-06,
      "step": 82922,
      "training_step_time": 7.892423152923584
    },
    {
      "epoch": 8.509635925292969e-06,
      "model_forward_time": 1.9231147766113281,
      "step": 82923
    },
    {
      "epoch": 8.509635925292969e-06,
      "step": 82923,
      "training_step_time": 7.3922505378723145
    },
    {
      "epoch": 8.510589599609375e-06,
      "model_forward_time": 1.8960261344909668,
      "step": 82924
    },
    {
      "epoch": 8.510589599609375e-06,
      "step": 82924,
      "training_step_time": 7.815532922744751
    },
    {
      "epoch": 8.511543273925782e-06,
      "model_forward_time": 1.9386446475982666,
      "step": 82925
    },
    {
      "epoch": 8.511543273925782e-06,
      "step": 82925,
      "training_step_time": 7.562439203262329
    },
    {
      "epoch": 8.512496948242188e-06,
      "model_forward_time": 1.9332830905914307,
      "step": 82926
    },
    {
      "epoch": 8.512496948242188e-06,
      "step": 82926,
      "training_step_time": 8.072440385818481
    },
    {
      "epoch": 8.513450622558593e-06,
      "model_forward_time": 1.9314100742340088,
      "step": 82927
    },
    {
      "epoch": 8.513450622558593e-06,
      "step": 82927,
      "training_step_time": 9.473870515823364
    },
    {
      "epoch": 8.514404296875e-06,
      "model_forward_time": 1.9089446067810059,
      "step": 82928
    },
    {
      "epoch": 8.514404296875e-06,
      "step": 82928,
      "training_step_time": 7.763951063156128
    },
    {
      "epoch": 8.515357971191407e-06,
      "model_forward_time": 1.9383878707885742,
      "step": 82929
    },
    {
      "epoch": 8.515357971191407e-06,
      "step": 82929,
      "training_step_time": 8.553818464279175
    },
    {
      "epoch": 8.516311645507812e-06,
      "grad_norm": 0.0743982195854187,
      "learning_rate": 7.757929912593698e-07,
      "loss": 0.0527,
      "step": 82930
    },
    {
      "epoch": 8.516311645507812e-06,
      "model_forward_time": 2.9714481830596924,
      "step": 82930
    },
    {
      "dynamics_loss_avg": 0.04649573490023613,
      "epoch": 8.516311645507812e-06,
      "step": 82930
    },
    {
      "action_loss_avg": 0.0013768420205451549,
      "epoch": 8.516311645507812e-06,
      "step": 82930
    },
    {
      "epoch": 8.516311645507812e-06,
      "step": 82930,
      "training_step_time": 7.687748432159424
    },
    {
      "epoch": 8.517265319824218e-06,
      "model_forward_time": 1.9304375648498535,
      "step": 82931
    },
    {
      "epoch": 8.517265319824218e-06,
      "step": 82931,
      "training_step_time": 8.614495038986206
    },
    {
      "epoch": 8.518218994140625e-06,
      "model_forward_time": 1.942500114440918,
      "step": 82932
    },
    {
      "epoch": 8.518218994140625e-06,
      "step": 82932,
      "training_step_time": 7.753063917160034
    },
    {
      "epoch": 8.519172668457031e-06,
      "model_forward_time": 1.9402821063995361,
      "step": 82933
    },
    {
      "epoch": 8.519172668457031e-06,
      "step": 82933,
      "training_step_time": 7.819047451019287
    },
    {
      "epoch": 8.520126342773437e-06,
      "model_forward_time": 1.9384560585021973,
      "step": 82934
    },
    {
      "epoch": 8.520126342773437e-06,
      "step": 82934,
      "training_step_time": 7.59719705581665
    },
    {
      "epoch": 8.521080017089844e-06,
      "model_forward_time": 1.9243500232696533,
      "step": 82935
    },
    {
      "epoch": 8.521080017089844e-06,
      "step": 82935,
      "training_step_time": 8.473656177520752
    },
    {
      "epoch": 8.52203369140625e-06,
      "model_forward_time": 1.929443597793579,
      "step": 82936
    },
    {
      "epoch": 8.52203369140625e-06,
      "step": 82936,
      "training_step_time": 7.640109062194824
    },
    {
      "epoch": 8.522987365722656e-06,
      "model_forward_time": 1.9237217903137207,
      "step": 82937
    },
    {
      "epoch": 8.522987365722656e-06,
      "step": 82937,
      "training_step_time": 8.539369106292725
    },
    {
      "epoch": 8.523941040039063e-06,
      "model_forward_time": 1.918461799621582,
      "step": 82938
    },
    {
      "epoch": 8.523941040039063e-06,
      "step": 82938,
      "training_step_time": 7.660349607467651
    },
    {
      "epoch": 8.524894714355469e-06,
      "model_forward_time": 1.9144420623779297,
      "step": 82939
    },
    {
      "epoch": 8.524894714355469e-06,
      "step": 82939,
      "training_step_time": 8.729321479797363
    },
    {
      "epoch": 8.525848388671875e-06,
      "grad_norm": 0.07710288465023041,
      "learning_rate": 7.749085880754508e-07,
      "loss": 0.0538,
      "step": 82940
    },
    {
      "epoch": 8.525848388671875e-06,
      "model_forward_time": 1.9311301708221436,
      "step": 82940
    },
    {
      "dynamics_loss_avg": 0.06124074626713991,
      "epoch": 8.525848388671875e-06,
      "step": 82940
    },
    {
      "action_loss_avg": 0.002026388351805508,
      "epoch": 8.525848388671875e-06,
      "step": 82940
    },
    {
      "epoch": 8.525848388671875e-06,
      "step": 82940,
      "training_step_time": 7.652151823043823
    },
    {
      "epoch": 8.526802062988282e-06,
      "model_forward_time": 1.923635482788086,
      "step": 82941
    },
    {
      "epoch": 8.526802062988282e-06,
      "step": 82941,
      "training_step_time": 7.899067401885986
    },
    {
      "epoch": 8.527755737304688e-06,
      "model_forward_time": 1.9276838302612305,
      "step": 82942
    },
    {
      "epoch": 8.527755737304688e-06,
      "step": 82942,
      "training_step_time": 7.774551630020142
    },
    {
      "epoch": 8.528709411621093e-06,
      "model_forward_time": 1.9176158905029297,
      "step": 82943
    },
    {
      "epoch": 8.528709411621093e-06,
      "step": 82943,
      "training_step_time": 7.477878093719482
    },
    {
      "epoch": 8.5296630859375e-06,
      "model_forward_time": 1.9445686340332031,
      "step": 82944
    },
    {
      "epoch": 8.5296630859375e-06,
      "step": 82944,
      "training_step_time": 8.52401876449585
    },
    {
      "epoch": 8.530616760253907e-06,
      "model_forward_time": 1.929269552230835,
      "step": 82945
    },
    {
      "epoch": 8.530616760253907e-06,
      "step": 82945,
      "training_step_time": 7.635250091552734
    },
    {
      "epoch": 8.531570434570312e-06,
      "model_forward_time": 1.9169633388519287,
      "step": 82946
    },
    {
      "epoch": 8.531570434570312e-06,
      "step": 82946,
      "training_step_time": 8.369022607803345
    },
    {
      "epoch": 8.532524108886718e-06,
      "model_forward_time": 1.9384377002716064,
      "step": 82947
    },
    {
      "epoch": 8.532524108886718e-06,
      "step": 82947,
      "training_step_time": 9.303551197052002
    },
    {
      "epoch": 8.533477783203125e-06,
      "model_forward_time": 1.9498512744903564,
      "step": 82948
    },
    {
      "epoch": 8.533477783203125e-06,
      "step": 82948,
      "training_step_time": 7.618504524230957
    },
    {
      "epoch": 8.534431457519531e-06,
      "model_forward_time": 2.984220504760742,
      "step": 82949
    },
    {
      "epoch": 8.534431457519531e-06,
      "step": 82949,
      "training_step_time": 7.717440605163574
    },
    {
      "epoch": 8.535385131835937e-06,
      "grad_norm": 0.09066279232501984,
      "learning_rate": 7.740246469411778e-07,
      "loss": 0.0536,
      "step": 82950
    },
    {
      "epoch": 8.535385131835937e-06,
      "model_forward_time": 1.9386351108551025,
      "step": 82950
    },
    {
      "dynamics_loss_avg": 0.05991466268897057,
      "epoch": 8.535385131835937e-06,
      "step": 82950
    },
    {
      "action_loss_avg": 0.0016396470018662512,
      "epoch": 8.535385131835937e-06,
      "step": 82950
    },
    {
      "epoch": 8.535385131835937e-06,
      "step": 82950,
      "training_step_time": 8.353179454803467
    },
    {
      "epoch": 8.536338806152344e-06,
      "model_forward_time": 1.9241085052490234,
      "step": 82951
    },
    {
      "epoch": 8.536338806152344e-06,
      "step": 82951,
      "training_step_time": 7.324382543563843
    },
    {
      "epoch": 8.53729248046875e-06,
      "model_forward_time": 1.9341363906860352,
      "step": 82952
    },
    {
      "epoch": 8.53729248046875e-06,
      "step": 82952,
      "training_step_time": 7.8734130859375
    },
    {
      "epoch": 8.538246154785156e-06,
      "model_forward_time": 1.948838472366333,
      "step": 82953
    },
    {
      "epoch": 8.538246154785156e-06,
      "step": 82953,
      "training_step_time": 8.574504852294922
    },
    {
      "epoch": 8.539199829101563e-06,
      "model_forward_time": 1.9344162940979004,
      "step": 82954
    },
    {
      "epoch": 8.539199829101563e-06,
      "step": 82954,
      "training_step_time": 7.738975763320923
    },
    {
      "epoch": 8.540153503417969e-06,
      "model_forward_time": 1.9082863330841064,
      "step": 82955
    },
    {
      "epoch": 8.540153503417969e-06,
      "step": 82955,
      "training_step_time": 10.527563333511353
    },
    {
      "epoch": 8.541107177734375e-06,
      "model_forward_time": 1.9201030731201172,
      "step": 82956
    },
    {
      "epoch": 8.541107177734375e-06,
      "step": 82956,
      "training_step_time": 7.849736928939819
    },
    {
      "epoch": 8.542060852050782e-06,
      "model_forward_time": 1.9314630031585693,
      "step": 82957
    },
    {
      "epoch": 8.542060852050782e-06,
      "step": 82957,
      "training_step_time": 7.795735597610474
    },
    {
      "epoch": 8.543014526367188e-06,
      "model_forward_time": 3.0338010787963867,
      "step": 82958
    },
    {
      "epoch": 8.543014526367188e-06,
      "step": 82958,
      "training_step_time": 7.901979207992554
    },
    {
      "epoch": 8.543968200683593e-06,
      "model_forward_time": 1.9379403591156006,
      "step": 82959
    },
    {
      "epoch": 8.543968200683593e-06,
      "step": 82959,
      "training_step_time": 9.246079444885254
    },
    {
      "epoch": 8.544921875e-06,
      "grad_norm": 0.08630191534757614,
      "learning_rate": 7.731411679532141e-07,
      "loss": 0.0511,
      "step": 82960
    },
    {
      "epoch": 8.544921875e-06,
      "model_forward_time": 1.9400625228881836,
      "step": 82960
    },
    {
      "dynamics_loss_avg": 0.037256432324647905,
      "epoch": 8.544921875e-06,
      "step": 82960
    },
    {
      "action_loss_avg": 0.0014182382554281503,
      "epoch": 8.544921875e-06,
      "step": 82960
    },
    {
      "epoch": 8.544921875e-06,
      "step": 82960,
      "training_step_time": 8.28488826751709
    },
    {
      "epoch": 8.545875549316407e-06,
      "model_forward_time": 1.9401321411132812,
      "step": 82961
    },
    {
      "epoch": 8.545875549316407e-06,
      "step": 82961,
      "training_step_time": 8.688371419906616
    },
    {
      "epoch": 8.546829223632812e-06,
      "model_forward_time": 1.2261059284210205,
      "step": 82962
    },
    {
      "epoch": 8.546829223632812e-06,
      "step": 82962,
      "training_step_time": 9.08254623413086
    },
    {
      "epoch": 8.547782897949218e-06,
      "model_forward_time": 1.9305217266082764,
      "step": 82963
    },
    {
      "epoch": 8.547782897949218e-06,
      "step": 82963,
      "training_step_time": 7.786781311035156
    },
    {
      "epoch": 8.548736572265625e-06,
      "model_forward_time": 1.932250738143921,
      "step": 82964
    },
    {
      "epoch": 8.548736572265625e-06,
      "step": 82964,
      "training_step_time": 8.863041639328003
    },
    {
      "epoch": 8.549690246582031e-06,
      "model_forward_time": 1.9297337532043457,
      "step": 82965
    },
    {
      "epoch": 8.549690246582031e-06,
      "step": 82965,
      "training_step_time": 9.584794998168945
    },
    {
      "epoch": 8.550643920898437e-06,
      "model_forward_time": 1.9287145137786865,
      "step": 82966
    },
    {
      "epoch": 8.550643920898437e-06,
      "step": 82966,
      "training_step_time": 7.74748420715332
    },
    {
      "epoch": 8.551597595214844e-06,
      "model_forward_time": 3.063408136367798,
      "step": 82967
    },
    {
      "epoch": 8.551597595214844e-06,
      "step": 82967,
      "training_step_time": 7.898728132247925
    },
    {
      "epoch": 8.55255126953125e-06,
      "model_forward_time": 1.9394872188568115,
      "step": 82968
    },
    {
      "epoch": 8.55255126953125e-06,
      "step": 82968,
      "training_step_time": 7.686723470687866
    },
    {
      "epoch": 8.553504943847656e-06,
      "model_forward_time": 1.9349472522735596,
      "step": 82969
    },
    {
      "epoch": 8.553504943847656e-06,
      "step": 82969,
      "training_step_time": 8.786193609237671
    },
    {
      "epoch": 8.554458618164063e-06,
      "grad_norm": 0.09736160933971405,
      "learning_rate": 7.722581512081779e-07,
      "loss": 0.0521,
      "step": 82970
    },
    {
      "epoch": 8.554458618164063e-06,
      "model_forward_time": 1.9399127960205078,
      "step": 82970
    },
    {
      "dynamics_loss_avg": 0.04394367812201381,
      "epoch": 8.554458618164063e-06,
      "step": 82970
    },
    {
      "action_loss_avg": 0.0014986858353950084,
      "epoch": 8.554458618164063e-06,
      "step": 82970
    },
    {
      "epoch": 8.554458618164063e-06,
      "step": 82970,
      "training_step_time": 7.684137344360352
    },
    {
      "epoch": 8.555412292480469e-06,
      "model_forward_time": 1.5705718994140625,
      "step": 82971
    },
    {
      "epoch": 8.555412292480469e-06,
      "step": 82971,
      "training_step_time": 7.659528732299805
    },
    {
      "epoch": 8.556365966796875e-06,
      "model_forward_time": 1.9540207386016846,
      "step": 82972
    },
    {
      "epoch": 8.556365966796875e-06,
      "step": 82972,
      "training_step_time": 9.637789964675903
    },
    {
      "epoch": 8.557319641113282e-06,
      "model_forward_time": 1.9401695728302002,
      "step": 82973
    },
    {
      "epoch": 8.557319641113282e-06,
      "step": 82973,
      "training_step_time": 7.749933481216431
    },
    {
      "epoch": 8.558273315429688e-06,
      "model_forward_time": 1.9294257164001465,
      "step": 82974
    },
    {
      "epoch": 8.558273315429688e-06,
      "step": 82974,
      "training_step_time": 10.275790214538574
    },
    {
      "epoch": 8.559226989746093e-06,
      "model_forward_time": 1.9317121505737305,
      "step": 82975
    },
    {
      "epoch": 8.559226989746093e-06,
      "step": 82975,
      "training_step_time": 7.735823631286621
    },
    {
      "epoch": 8.5601806640625e-06,
      "model_forward_time": 1.9151909351348877,
      "step": 82976
    },
    {
      "epoch": 8.5601806640625e-06,
      "step": 82976,
      "training_step_time": 7.643424034118652
    },
    {
      "epoch": 8.561134338378907e-06,
      "model_forward_time": 2.9793994426727295,
      "step": 82977
    },
    {
      "epoch": 8.561134338378907e-06,
      "step": 82977,
      "training_step_time": 8.569964170455933
    },
    {
      "epoch": 8.562088012695312e-06,
      "model_forward_time": 1.9410042762756348,
      "step": 82978
    },
    {
      "epoch": 8.562088012695312e-06,
      "step": 82978,
      "training_step_time": 7.89174485206604
    },
    {
      "epoch": 8.563041687011718e-06,
      "model_forward_time": 1.9305529594421387,
      "step": 82979
    },
    {
      "epoch": 8.563041687011718e-06,
      "step": 82979,
      "training_step_time": 7.650157451629639
    },
    {
      "epoch": 8.563995361328125e-06,
      "grad_norm": 0.09111843258142471,
      "learning_rate": 7.713755968026332e-07,
      "loss": 0.0525,
      "step": 82980
    },
    {
      "epoch": 8.563995361328125e-06,
      "model_forward_time": 1.9359314441680908,
      "step": 82980
    },
    {
      "dynamics_loss_avg": 0.049709057249128816,
      "epoch": 8.563995361328125e-06,
      "step": 82980
    },
    {
      "action_loss_avg": 0.001525482494616881,
      "epoch": 8.563995361328125e-06,
      "step": 82980
    },
    {
      "epoch": 8.563995361328125e-06,
      "step": 82980,
      "training_step_time": 7.686508655548096
    },
    {
      "epoch": 8.564949035644531e-06,
      "model_forward_time": 1.9380676746368408,
      "step": 82981
    },
    {
      "epoch": 8.564949035644531e-06,
      "step": 82981,
      "training_step_time": 9.312984704971313
    },
    {
      "epoch": 8.565902709960937e-06,
      "model_forward_time": 1.920954942703247,
      "step": 82982
    },
    {
      "epoch": 8.565902709960937e-06,
      "step": 82982,
      "training_step_time": 9.72844409942627
    },
    {
      "epoch": 8.566856384277344e-06,
      "model_forward_time": 1.9293122291564941,
      "step": 82983
    },
    {
      "epoch": 8.566856384277344e-06,
      "step": 82983,
      "training_step_time": 8.945375204086304
    },
    {
      "epoch": 8.56781005859375e-06,
      "model_forward_time": 1.9218485355377197,
      "step": 82984
    },
    {
      "epoch": 8.56781005859375e-06,
      "step": 82984,
      "training_step_time": 8.73353362083435
    },
    {
      "epoch": 8.568763732910156e-06,
      "model_forward_time": 1.9237723350524902,
      "step": 82985
    },
    {
      "epoch": 8.568763732910156e-06,
      "step": 82985,
      "training_step_time": 7.700867414474487
    },
    {
      "epoch": 8.569717407226563e-06,
      "model_forward_time": 2.9737887382507324,
      "step": 82986
    },
    {
      "epoch": 8.569717407226563e-06,
      "step": 82986,
      "training_step_time": 7.744131565093994
    },
    {
      "epoch": 8.570671081542969e-06,
      "model_forward_time": 1.9278106689453125,
      "step": 82987
    },
    {
      "epoch": 8.570671081542969e-06,
      "step": 82987,
      "training_step_time": 7.532864332199097
    },
    {
      "epoch": 8.571624755859375e-06,
      "model_forward_time": 1.9055519104003906,
      "step": 82988
    },
    {
      "epoch": 8.571624755859375e-06,
      "step": 82988,
      "training_step_time": 7.832026958465576
    },
    {
      "epoch": 8.572578430175782e-06,
      "model_forward_time": 1.9182872772216797,
      "step": 82989
    },
    {
      "epoch": 8.572578430175782e-06,
      "step": 82989,
      "training_step_time": 9.623478889465332
    },
    {
      "epoch": 8.573532104492188e-06,
      "grad_norm": 0.07514794170856476,
      "learning_rate": 7.704935048330969e-07,
      "loss": 0.0546,
      "step": 82990
    },
    {
      "epoch": 8.573532104492188e-06,
      "model_forward_time": 1.7073051929473877,
      "step": 82990
    },
    {
      "dynamics_loss_avg": 0.061343401297926904,
      "epoch": 8.573532104492188e-06,
      "step": 82990
    },
    {
      "action_loss_avg": 0.0017125942395068705,
      "epoch": 8.573532104492188e-06,
      "step": 82990
    },
    {
      "epoch": 8.573532104492188e-06,
      "step": 82990,
      "training_step_time": 8.460472106933594
    },
    {
      "epoch": 8.574485778808593e-06,
      "model_forward_time": 1.7189385890960693,
      "step": 82991
    },
    {
      "epoch": 8.574485778808593e-06,
      "step": 82991,
      "training_step_time": 9.403006792068481
    },
    {
      "epoch": 8.575439453125e-06,
      "model_forward_time": 1.7210025787353516,
      "step": 82992
    },
    {
      "epoch": 8.575439453125e-06,
      "step": 82992,
      "training_step_time": 8.828385591506958
    },
    {
      "epoch": 8.576393127441407e-06,
      "model_forward_time": 1.7470178604125977,
      "step": 82993
    },
    {
      "epoch": 8.576393127441407e-06,
      "step": 82993,
      "training_step_time": 7.793996095657349
    },
    {
      "epoch": 8.577346801757812e-06,
      "model_forward_time": 1.7273194789886475,
      "step": 82994
    },
    {
      "epoch": 8.577346801757812e-06,
      "step": 82994,
      "training_step_time": 7.765435457229614
    },
    {
      "epoch": 8.578300476074218e-06,
      "model_forward_time": 2.356797456741333,
      "step": 82995
    },
    {
      "epoch": 8.578300476074218e-06,
      "step": 82995,
      "training_step_time": 7.518602609634399
    },
    {
      "epoch": 8.579254150390625e-06,
      "model_forward_time": 1.7180969715118408,
      "step": 82996
    },
    {
      "epoch": 8.579254150390625e-06,
      "step": 82996,
      "training_step_time": 7.640516757965088
    },
    {
      "epoch": 8.580207824707031e-06,
      "model_forward_time": 1.7255167961120605,
      "step": 82997
    },
    {
      "epoch": 8.580207824707031e-06,
      "step": 82997,
      "training_step_time": 7.932757616043091
    },
    {
      "epoch": 8.581161499023437e-06,
      "model_forward_time": 1.7415928840637207,
      "step": 82998
    },
    {
      "epoch": 8.581161499023437e-06,
      "step": 82998,
      "training_step_time": 8.078563213348389
    },
    {
      "epoch": 8.582115173339844e-06,
      "model_forward_time": 1.8098013401031494,
      "step": 82999
    },
    {
      "epoch": 8.582115173339844e-06,
      "step": 82999,
      "training_step_time": 9.358582258224487
    },
    {
      "epoch": 8.58306884765625e-06,
      "grad_norm": 0.07794860750436783,
      "learning_rate": 7.696118753960313e-07,
      "loss": 0.0548,
      "step": 83000
    },
    {
      "epoch": 8.58306884765625e-06,
      "model_forward_time": 1.8642122745513916,
      "step": 83000
    },
    {
      "dynamics_loss_avg": 0.04857895821332932,
      "epoch": 8.58306884765625e-06,
      "step": 83000
    },
    {
      "action_loss_avg": 0.0017393600952345877,
      "epoch": 8.58306884765625e-06,
      "step": 83000
    },
    {
      "epoch": 8.58306884765625e-06,
      "step": 83000,
      "training_step_time": 9.02635645866394
    },
    {
      "epoch": 8.584022521972656e-06,
      "model_forward_time": 1.7886922359466553,
      "step": 83001
    },
    {
      "epoch": 8.584022521972656e-06,
      "step": 83001,
      "training_step_time": 9.53016710281372
    },
    {
      "epoch": 8.584976196289063e-06,
      "model_forward_time": 1.7737963199615479,
      "step": 83002
    },
    {
      "epoch": 8.584976196289063e-06,
      "step": 83002,
      "training_step_time": 9.828176021575928
    },
    {
      "epoch": 8.585929870605469e-06,
      "model_forward_time": 1.7811765670776367,
      "step": 83003
    },
    {
      "epoch": 8.585929870605469e-06,
      "step": 83003,
      "training_step_time": 7.770267724990845
    },
    {
      "epoch": 8.586883544921875e-06,
      "model_forward_time": 3.50024151802063,
      "step": 83004
    },
    {
      "epoch": 8.586883544921875e-06,
      "step": 83004,
      "training_step_time": 8.45187258720398
    },
    {
      "epoch": 8.587837219238282e-06,
      "model_forward_time": 1.8023138046264648,
      "step": 83005
    },
    {
      "epoch": 8.587837219238282e-06,
      "step": 83005,
      "training_step_time": 8.641059160232544
    },
    {
      "epoch": 8.588790893554688e-06,
      "model_forward_time": 1.7789781093597412,
      "step": 83006
    },
    {
      "epoch": 8.588790893554688e-06,
      "step": 83006,
      "training_step_time": 7.767273426055908
    },
    {
      "epoch": 8.589744567871093e-06,
      "model_forward_time": 1.7770209312438965,
      "step": 83007
    },
    {
      "epoch": 8.589744567871093e-06,
      "step": 83007,
      "training_step_time": 7.824061870574951
    },
    {
      "epoch": 8.5906982421875e-06,
      "model_forward_time": 1.425119161605835,
      "step": 83008
    },
    {
      "epoch": 8.5906982421875e-06,
      "step": 83008,
      "training_step_time": 9.22170352935791
    },
    {
      "epoch": 8.591651916503907e-06,
      "model_forward_time": 1.7773289680480957,
      "step": 83009
    },
    {
      "epoch": 8.591651916503907e-06,
      "step": 83009,
      "training_step_time": 8.353501319885254
    },
    {
      "epoch": 8.592605590820312e-06,
      "grad_norm": 0.10946785658597946,
      "learning_rate": 7.687307085878503e-07,
      "loss": 0.0542,
      "step": 83010
    },
    {
      "epoch": 8.592605590820312e-06,
      "model_forward_time": 1.7834327220916748,
      "step": 83010
    },
    {
      "dynamics_loss_avg": 0.060518310591578486,
      "epoch": 8.592605590820312e-06,
      "step": 83010
    },
    {
      "action_loss_avg": 0.0019048354413826018,
      "epoch": 8.592605590820312e-06,
      "step": 83010
    },
    {
      "epoch": 8.592605590820312e-06,
      "step": 83010,
      "training_step_time": 10.662153244018555
    },
    {
      "epoch": 8.593559265136718e-06,
      "model_forward_time": 1.4885215759277344,
      "step": 83011
    },
    {
      "epoch": 8.593559265136718e-06,
      "step": 83011,
      "training_step_time": 8.534972429275513
    },
    {
      "epoch": 8.594512939453125e-06,
      "model_forward_time": 1.7631394863128662,
      "step": 83012
    },
    {
      "epoch": 8.594512939453125e-06,
      "step": 83012,
      "training_step_time": 7.729184627532959
    },
    {
      "epoch": 8.595466613769531e-06,
      "model_forward_time": 2.8925349712371826,
      "step": 83013
    },
    {
      "epoch": 8.595466613769531e-06,
      "step": 83013,
      "training_step_time": 8.636281728744507
    },
    {
      "epoch": 8.596420288085937e-06,
      "model_forward_time": 1.7773044109344482,
      "step": 83014
    },
    {
      "epoch": 8.596420288085937e-06,
      "step": 83014,
      "training_step_time": 7.747657299041748
    },
    {
      "epoch": 8.597373962402344e-06,
      "model_forward_time": 1.7790887355804443,
      "step": 83015
    },
    {
      "epoch": 8.597373962402344e-06,
      "step": 83015,
      "training_step_time": 9.942509174346924
    },
    {
      "epoch": 8.59832763671875e-06,
      "model_forward_time": 1.7773430347442627,
      "step": 83016
    },
    {
      "epoch": 8.59832763671875e-06,
      "step": 83016,
      "training_step_time": 7.892165422439575
    },
    {
      "epoch": 8.599281311035156e-06,
      "model_forward_time": 1.7754430770874023,
      "step": 83017
    },
    {
      "epoch": 8.599281311035156e-06,
      "step": 83017,
      "training_step_time": 7.71595311164856
    },
    {
      "epoch": 8.600234985351563e-06,
      "model_forward_time": 1.7841246128082275,
      "step": 83018
    },
    {
      "epoch": 8.600234985351563e-06,
      "step": 83018,
      "training_step_time": 7.90701150894165
    },
    {
      "epoch": 8.601188659667969e-06,
      "model_forward_time": 1.779203176498413,
      "step": 83019
    },
    {
      "epoch": 8.601188659667969e-06,
      "step": 83019,
      "training_step_time": 9.190542221069336
    },
    {
      "epoch": 8.602142333984375e-06,
      "grad_norm": 0.09966089576482773,
      "learning_rate": 7.678500045049187e-07,
      "loss": 0.0544,
      "step": 83020
    },
    {
      "epoch": 8.602142333984375e-06,
      "model_forward_time": 1.8434727191925049,
      "step": 83020
    },
    {
      "dynamics_loss_avg": 0.050408387742936614,
      "epoch": 8.602142333984375e-06,
      "step": 83020
    },
    {
      "action_loss_avg": 0.0016794673749245704,
      "epoch": 8.602142333984375e-06,
      "step": 83020
    },
    {
      "epoch": 8.602142333984375e-06,
      "step": 83020,
      "training_step_time": 7.602130651473999
    },
    {
      "epoch": 8.603096008300782e-06,
      "model_forward_time": 1.781510829925537,
      "step": 83021
    },
    {
      "epoch": 8.603096008300782e-06,
      "step": 83021,
      "training_step_time": 7.80849289894104
    },
    {
      "epoch": 8.604049682617188e-06,
      "model_forward_time": 0.6807544231414795,
      "step": 83022
    },
    {
      "epoch": 8.604049682617188e-06,
      "step": 83022,
      "training_step_time": 8.330763339996338
    },
    {
      "epoch": 8.605003356933593e-06,
      "model_forward_time": 2.3630127906799316,
      "step": 83023
    },
    {
      "epoch": 8.605003356933593e-06,
      "step": 83023,
      "training_step_time": 7.75347638130188
    },
    {
      "epoch": 8.60595703125e-06,
      "model_forward_time": 1.7414655685424805,
      "step": 83024
    },
    {
      "epoch": 8.60595703125e-06,
      "step": 83024,
      "training_step_time": 7.7621729373931885
    },
    {
      "epoch": 8.606910705566407e-06,
      "model_forward_time": 1.7464394569396973,
      "step": 83025
    },
    {
      "epoch": 8.606910705566407e-06,
      "step": 83025,
      "training_step_time": 7.807544469833374
    },
    {
      "epoch": 8.607864379882812e-06,
      "model_forward_time": 1.7410998344421387,
      "step": 83026
    },
    {
      "epoch": 8.607864379882812e-06,
      "step": 83026,
      "training_step_time": 8.745638608932495
    },
    {
      "epoch": 8.608818054199218e-06,
      "model_forward_time": 1.7445945739746094,
      "step": 83027
    },
    {
      "epoch": 8.608818054199218e-06,
      "step": 83027,
      "training_step_time": 7.838952302932739
    },
    {
      "epoch": 8.609771728515625e-06,
      "model_forward_time": 1.3566088676452637,
      "step": 83028
    },
    {
      "epoch": 8.609771728515625e-06,
      "step": 83028,
      "training_step_time": 8.454000473022461
    },
    {
      "epoch": 8.610725402832031e-06,
      "model_forward_time": 1.7226412296295166,
      "step": 83029
    },
    {
      "epoch": 8.610725402832031e-06,
      "step": 83029,
      "training_step_time": 8.922529458999634
    },
    {
      "epoch": 8.611679077148437e-06,
      "grad_norm": 0.0838686153292656,
      "learning_rate": 7.669697632435464e-07,
      "loss": 0.0545,
      "step": 83030
    },
    {
      "epoch": 8.611679077148437e-06,
      "model_forward_time": 1.730165719985962,
      "step": 83030
    },
    {
      "dynamics_loss_avg": 0.0319071869365871,
      "epoch": 8.611679077148437e-06,
      "step": 83030
    },
    {
      "action_loss_avg": 0.0027604795934166758,
      "epoch": 8.611679077148437e-06,
      "step": 83030
    },
    {
      "epoch": 8.611679077148437e-06,
      "step": 83030,
      "training_step_time": 9.544452667236328
    },
    {
      "epoch": 8.612632751464844e-06,
      "model_forward_time": 1.7388591766357422,
      "step": 83031
    },
    {
      "epoch": 8.612632751464844e-06,
      "step": 83031,
      "training_step_time": 8.657167911529541
    },
    {
      "epoch": 8.61358642578125e-06,
      "model_forward_time": 1.611236810684204,
      "step": 83032
    },
    {
      "epoch": 8.61358642578125e-06,
      "step": 83032,
      "training_step_time": 7.680883884429932
    },
    {
      "epoch": 8.614540100097656e-06,
      "model_forward_time": 1.7391493320465088,
      "step": 83033
    },
    {
      "epoch": 8.614540100097656e-06,
      "step": 83033,
      "training_step_time": 7.780935525894165
    },
    {
      "epoch": 8.615493774414063e-06,
      "model_forward_time": 1.737964153289795,
      "step": 83034
    },
    {
      "epoch": 8.615493774414063e-06,
      "step": 83034,
      "training_step_time": 8.420168161392212
    },
    {
      "epoch": 8.616447448730469e-06,
      "model_forward_time": 1.726142406463623,
      "step": 83035
    },
    {
      "epoch": 8.616447448730469e-06,
      "step": 83035,
      "training_step_time": 7.673536777496338
    },
    {
      "epoch": 8.617401123046875e-06,
      "model_forward_time": 1.7592575550079346,
      "step": 83036
    },
    {
      "epoch": 8.617401123046875e-06,
      "step": 83036,
      "training_step_time": 9.203101396560669
    },
    {
      "epoch": 8.618354797363282e-06,
      "model_forward_time": 1.7395610809326172,
      "step": 83037
    },
    {
      "epoch": 8.618354797363282e-06,
      "step": 83037,
      "training_step_time": 8.946378707885742
    },
    {
      "epoch": 8.619308471679688e-06,
      "model_forward_time": 1.7411115169525146,
      "step": 83038
    },
    {
      "epoch": 8.619308471679688e-06,
      "step": 83038,
      "training_step_time": 9.272210359573364
    },
    {
      "epoch": 8.620262145996093e-06,
      "model_forward_time": 1.8673946857452393,
      "step": 83039
    },
    {
      "epoch": 8.620262145996093e-06,
      "step": 83039,
      "training_step_time": 10.264983415603638
    },
    {
      "epoch": 8.6212158203125e-06,
      "grad_norm": 0.11458484828472137,
      "learning_rate": 7.660899848999964e-07,
      "loss": 0.0542,
      "step": 83040
    },
    {
      "epoch": 8.6212158203125e-06,
      "model_forward_time": 1.7354302406311035,
      "step": 83040
    },
    {
      "dynamics_loss_avg": 0.04648043997585773,
      "epoch": 8.6212158203125e-06,
      "step": 83040
    },
    {
      "action_loss_avg": 0.0014408201619517057,
      "epoch": 8.6212158203125e-06,
      "step": 83040
    },
    {
      "epoch": 8.6212158203125e-06,
      "step": 83040,
      "training_step_time": 7.782602548599243
    },
    {
      "epoch": 8.622169494628907e-06,
      "model_forward_time": 2.3539910316467285,
      "step": 83041
    },
    {
      "epoch": 8.622169494628907e-06,
      "step": 83041,
      "training_step_time": 7.579366683959961
    },
    {
      "epoch": 8.623123168945312e-06,
      "model_forward_time": 1.7498281002044678,
      "step": 83042
    },
    {
      "epoch": 8.623123168945312e-06,
      "step": 83042,
      "training_step_time": 8.552440881729126
    },
    {
      "epoch": 8.624076843261718e-06,
      "model_forward_time": 1.0220551490783691,
      "step": 83043
    },
    {
      "epoch": 8.624076843261718e-06,
      "step": 83043,
      "training_step_time": 7.795306444168091
    },
    {
      "epoch": 8.625030517578125e-06,
      "model_forward_time": 1.7479829788208008,
      "step": 83044
    },
    {
      "epoch": 8.625030517578125e-06,
      "step": 83044,
      "training_step_time": 7.827225685119629
    },
    {
      "epoch": 8.625984191894531e-06,
      "model_forward_time": 1.3694005012512207,
      "step": 83045
    },
    {
      "epoch": 8.625984191894531e-06,
      "step": 83045,
      "training_step_time": 9.759032726287842
    },
    {
      "epoch": 8.626937866210937e-06,
      "model_forward_time": 1.7390084266662598,
      "step": 83046
    },
    {
      "epoch": 8.626937866210937e-06,
      "step": 83046,
      "training_step_time": 9.332367420196533
    },
    {
      "epoch": 8.627891540527344e-06,
      "model_forward_time": 1.7367737293243408,
      "step": 83047
    },
    {
      "epoch": 8.627891540527344e-06,
      "step": 83047,
      "training_step_time": 9.557691097259521
    },
    {
      "epoch": 8.62884521484375e-06,
      "model_forward_time": 1.7643659114837646,
      "step": 83048
    },
    {
      "epoch": 8.62884521484375e-06,
      "step": 83048,
      "training_step_time": 9.861780405044556
    },
    {
      "epoch": 8.629798889160156e-06,
      "model_forward_time": 1.7346644401550293,
      "step": 83049
    },
    {
      "epoch": 8.629798889160156e-06,
      "step": 83049,
      "training_step_time": 8.225349187850952
    },
    {
      "epoch": 8.630752563476563e-06,
      "grad_norm": 0.08115386217832565,
      "learning_rate": 7.652106695704803e-07,
      "loss": 0.0527,
      "step": 83050
    },
    {
      "epoch": 8.630752563476563e-06,
      "model_forward_time": 1.7395777702331543,
      "step": 83050
    },
    {
      "dynamics_loss_avg": 0.039347395207732914,
      "epoch": 8.630752563476563e-06,
      "step": 83050
    },
    {
      "action_loss_avg": 0.0017355867486912758,
      "epoch": 8.630752563476563e-06,
      "step": 83050
    },
    {
      "epoch": 8.630752563476563e-06,
      "step": 83050,
      "training_step_time": 7.517495155334473
    },
    {
      "epoch": 8.631706237792969e-06,
      "model_forward_time": 2.444601535797119,
      "step": 83051
    },
    {
      "epoch": 8.631706237792969e-06,
      "step": 83051,
      "training_step_time": 7.656193733215332
    },
    {
      "epoch": 8.632659912109375e-06,
      "model_forward_time": 1.358750820159912,
      "step": 83052
    },
    {
      "epoch": 8.632659912109375e-06,
      "step": 83052,
      "training_step_time": 7.748779535293579
    },
    {
      "epoch": 8.633613586425782e-06,
      "model_forward_time": 1.7314064502716064,
      "step": 83053
    },
    {
      "epoch": 8.633613586425782e-06,
      "step": 83053,
      "training_step_time": 7.71285343170166
    },
    {
      "epoch": 8.634567260742188e-06,
      "model_forward_time": 1.7404403686523438,
      "step": 83054
    },
    {
      "epoch": 8.634567260742188e-06,
      "step": 83054,
      "training_step_time": 7.694550514221191
    },
    {
      "epoch": 8.635520935058593e-06,
      "model_forward_time": 1.7421176433563232,
      "step": 83055
    },
    {
      "epoch": 8.635520935058593e-06,
      "step": 83055,
      "training_step_time": 7.699914932250977
    },
    {
      "epoch": 8.636474609375e-06,
      "model_forward_time": 1.7339708805084229,
      "step": 83056
    },
    {
      "epoch": 8.636474609375e-06,
      "step": 83056,
      "training_step_time": 8.672963619232178
    },
    {
      "epoch": 8.637428283691407e-06,
      "model_forward_time": 1.7520272731781006,
      "step": 83057
    },
    {
      "epoch": 8.637428283691407e-06,
      "step": 83057,
      "training_step_time": 8.726303577423096
    },
    {
      "epoch": 8.638381958007812e-06,
      "model_forward_time": 1.7425665855407715,
      "step": 83058
    },
    {
      "epoch": 8.638381958007812e-06,
      "step": 83058,
      "training_step_time": 7.737152338027954
    },
    {
      "epoch": 8.639335632324218e-06,
      "model_forward_time": 1.7313389778137207,
      "step": 83059
    },
    {
      "epoch": 8.639335632324218e-06,
      "step": 83059,
      "training_step_time": 7.695186138153076
    },
    {
      "epoch": 8.640289306640625e-06,
      "grad_norm": 0.12858463823795319,
      "learning_rate": 7.643318173511599e-07,
      "loss": 0.0542,
      "step": 83060
    },
    {
      "epoch": 8.640289306640625e-06,
      "model_forward_time": 2.357490301132202,
      "step": 83060
    },
    {
      "dynamics_loss_avg": 0.0645443320274353,
      "epoch": 8.640289306640625e-06,
      "step": 83060
    },
    {
      "action_loss_avg": 0.0015459273068699985,
      "epoch": 8.640289306640625e-06,
      "step": 83060
    },
    {
      "epoch": 8.640289306640625e-06,
      "step": 83060,
      "training_step_time": 7.683821439743042
    },
    {
      "epoch": 8.641242980957031e-06,
      "model_forward_time": 1.727898359298706,
      "step": 83061
    },
    {
      "epoch": 8.641242980957031e-06,
      "step": 83061,
      "training_step_time": 7.646456003189087
    },
    {
      "epoch": 8.642196655273437e-06,
      "model_forward_time": 1.739928960800171,
      "step": 83062
    },
    {
      "epoch": 8.642196655273437e-06,
      "step": 83062,
      "training_step_time": 7.679494857788086
    },
    {
      "epoch": 8.643150329589844e-06,
      "model_forward_time": 1.7367656230926514,
      "step": 83063
    },
    {
      "epoch": 8.643150329589844e-06,
      "step": 83063,
      "training_step_time": 7.636805057525635
    },
    {
      "epoch": 8.64410400390625e-06,
      "model_forward_time": 1.734316349029541,
      "step": 83064
    },
    {
      "epoch": 8.64410400390625e-06,
      "step": 83064,
      "training_step_time": 7.745905637741089
    },
    {
      "epoch": 8.645057678222656e-06,
      "model_forward_time": 1.7462975978851318,
      "step": 83065
    },
    {
      "epoch": 8.645057678222656e-06,
      "step": 83065,
      "training_step_time": 8.410500288009644
    },
    {
      "epoch": 8.646011352539063e-06,
      "model_forward_time": 1.7421491146087646,
      "step": 83066
    },
    {
      "epoch": 8.646011352539063e-06,
      "step": 83066,
      "training_step_time": 9.999955415725708
    },
    {
      "epoch": 8.646965026855469e-06,
      "model_forward_time": 1.7409653663635254,
      "step": 83067
    },
    {
      "epoch": 8.646965026855469e-06,
      "step": 83067,
      "training_step_time": 8.572940826416016
    },
    {
      "epoch": 8.647918701171875e-06,
      "model_forward_time": 1.7394638061523438,
      "step": 83068
    },
    {
      "epoch": 8.647918701171875e-06,
      "step": 83068,
      "training_step_time": 7.658517122268677
    },
    {
      "epoch": 8.648872375488282e-06,
      "model_forward_time": 1.7413525581359863,
      "step": 83069
    },
    {
      "epoch": 8.648872375488282e-06,
      "step": 83069,
      "training_step_time": 7.712738037109375
    },
    {
      "epoch": 8.649826049804688e-06,
      "grad_norm": 0.10305824130773544,
      "learning_rate": 7.634534283381422e-07,
      "loss": 0.0535,
      "step": 83070
    },
    {
      "epoch": 8.649826049804688e-06,
      "model_forward_time": 2.3901658058166504,
      "step": 83070
    },
    {
      "dynamics_loss_avg": 0.06144161224365234,
      "epoch": 8.649826049804688e-06,
      "step": 83070
    },
    {
      "action_loss_avg": 0.0021654722688253967,
      "epoch": 8.649826049804688e-06,
      "step": 83070
    },
    {
      "epoch": 8.649826049804688e-06,
      "step": 83070,
      "training_step_time": 7.3504533767700195
    },
    {
      "epoch": 8.650779724121093e-06,
      "model_forward_time": 1.7364928722381592,
      "step": 83071
    },
    {
      "epoch": 8.650779724121093e-06,
      "step": 83071,
      "training_step_time": 7.681701421737671
    },
    {
      "epoch": 8.6517333984375e-06,
      "model_forward_time": 1.3611748218536377,
      "step": 83072
    },
    {
      "epoch": 8.6517333984375e-06,
      "step": 83072,
      "training_step_time": 8.644941568374634
    },
    {
      "epoch": 8.652687072753907e-06,
      "model_forward_time": 1.7568795680999756,
      "step": 83073
    },
    {
      "epoch": 8.652687072753907e-06,
      "step": 83073,
      "training_step_time": 9.206088304519653
    },
    {
      "epoch": 8.653640747070312e-06,
      "model_forward_time": 1.7313754558563232,
      "step": 83074
    },
    {
      "epoch": 8.653640747070312e-06,
      "step": 83074,
      "training_step_time": 9.317932367324829
    },
    {
      "epoch": 8.654594421386718e-06,
      "model_forward_time": 1.7364630699157715,
      "step": 83075
    },
    {
      "epoch": 8.654594421386718e-06,
      "step": 83075,
      "training_step_time": 7.753410577774048
    },
    {
      "epoch": 8.655548095703125e-06,
      "model_forward_time": 1.7336628437042236,
      "step": 83076
    },
    {
      "epoch": 8.655548095703125e-06,
      "step": 83076,
      "training_step_time": 8.794029474258423
    },
    {
      "epoch": 8.656501770019531e-06,
      "model_forward_time": 1.7318570613861084,
      "step": 83077
    },
    {
      "epoch": 8.656501770019531e-06,
      "step": 83077,
      "training_step_time": 7.731867074966431
    },
    {
      "epoch": 8.657455444335937e-06,
      "model_forward_time": 1.7387487888336182,
      "step": 83078
    },
    {
      "epoch": 8.657455444335937e-06,
      "step": 83078,
      "training_step_time": 9.627315282821655
    },
    {
      "epoch": 8.658409118652344e-06,
      "model_forward_time": 2.44167160987854,
      "step": 83079
    },
    {
      "epoch": 8.658409118652344e-06,
      "step": 83079,
      "training_step_time": 7.386606216430664
    },
    {
      "epoch": 8.65936279296875e-06,
      "grad_norm": 0.10212188214063644,
      "learning_rate": 7.625755026274889e-07,
      "loss": 0.0543,
      "step": 83080
    },
    {
      "epoch": 8.65936279296875e-06,
      "model_forward_time": 1.7395257949829102,
      "step": 83080
    },
    {
      "dynamics_loss_avg": 0.07963175624608994,
      "epoch": 8.65936279296875e-06,
      "step": 83080
    },
    {
      "action_loss_avg": 0.002827687177341431,
      "epoch": 8.65936279296875e-06,
      "step": 83080
    },
    {
      "epoch": 8.65936279296875e-06,
      "step": 83080,
      "training_step_time": 7.661834716796875
    },
    {
      "epoch": 8.660316467285156e-06,
      "model_forward_time": 1.7205328941345215,
      "step": 83081
    },
    {
      "epoch": 8.660316467285156e-06,
      "step": 83081,
      "training_step_time": 9.38678765296936
    },
    {
      "epoch": 8.661270141601563e-06,
      "model_forward_time": 1.7494702339172363,
      "step": 83082
    },
    {
      "epoch": 8.661270141601563e-06,
      "step": 83082,
      "training_step_time": 7.7393248081207275
    },
    {
      "epoch": 8.662223815917969e-06,
      "model_forward_time": 1.7363348007202148,
      "step": 83083
    },
    {
      "epoch": 8.662223815917969e-06,
      "step": 83083,
      "training_step_time": 9.235228776931763
    },
    {
      "epoch": 8.663177490234375e-06,
      "model_forward_time": 1.7327513694763184,
      "step": 83084
    },
    {
      "epoch": 8.663177490234375e-06,
      "step": 83084,
      "training_step_time": 8.894052267074585
    },
    {
      "epoch": 8.664131164550782e-06,
      "model_forward_time": 1.7523009777069092,
      "step": 83085
    },
    {
      "epoch": 8.664131164550782e-06,
      "step": 83085,
      "training_step_time": 7.659619569778442
    },
    {
      "epoch": 8.665084838867188e-06,
      "model_forward_time": 1.7391996383666992,
      "step": 83086
    },
    {
      "epoch": 8.665084838867188e-06,
      "step": 83086,
      "training_step_time": 7.630060434341431
    },
    {
      "epoch": 8.666038513183594e-06,
      "model_forward_time": 1.7329394817352295,
      "step": 83087
    },
    {
      "epoch": 8.666038513183594e-06,
      "step": 83087,
      "training_step_time": 8.693970203399658
    },
    {
      "epoch": 8.6669921875e-06,
      "model_forward_time": 2.4164493083953857,
      "step": 83088
    },
    {
      "epoch": 8.6669921875e-06,
      "step": 83088,
      "training_step_time": 7.606253385543823
    },
    {
      "epoch": 8.667945861816407e-06,
      "model_forward_time": 1.7356970310211182,
      "step": 83089
    },
    {
      "epoch": 8.667945861816407e-06,
      "step": 83089,
      "training_step_time": 7.597898244857788
    },
    {
      "epoch": 8.668899536132812e-06,
      "grad_norm": 0.2472919523715973,
      "learning_rate": 7.616980403152069e-07,
      "loss": 0.0539,
      "step": 83090
    },
    {
      "epoch": 8.668899536132812e-06,
      "model_forward_time": 1.738976001739502,
      "step": 83090
    },
    {
      "dynamics_loss_avg": 0.05190372969955206,
      "epoch": 8.668899536132812e-06,
      "step": 83090
    },
    {
      "action_loss_avg": 0.0009805162146221847,
      "epoch": 8.668899536132812e-06,
      "step": 83090
    },
    {
      "epoch": 8.668899536132812e-06,
      "step": 83090,
      "training_step_time": 7.57705545425415
    },
    {
      "epoch": 8.669853210449218e-06,
      "model_forward_time": 1.7596275806427002,
      "step": 83091
    },
    {
      "epoch": 8.669853210449218e-06,
      "step": 83091,
      "training_step_time": 7.669242858886719
    },
    {
      "epoch": 8.670806884765626e-06,
      "model_forward_time": 1.7377374172210693,
      "step": 83092
    },
    {
      "epoch": 8.670806884765626e-06,
      "step": 83092,
      "training_step_time": 8.49896788597107
    },
    {
      "epoch": 8.671760559082031e-06,
      "model_forward_time": 1.7367687225341797,
      "step": 83093
    },
    {
      "epoch": 8.671760559082031e-06,
      "step": 83093,
      "training_step_time": 9.914231061935425
    },
    {
      "epoch": 8.672714233398437e-06,
      "model_forward_time": 1.758753776550293,
      "step": 83094
    },
    {
      "epoch": 8.672714233398437e-06,
      "step": 83094,
      "training_step_time": 9.625630617141724
    },
    {
      "epoch": 8.673667907714844e-06,
      "model_forward_time": 1.7443904876708984,
      "step": 83095
    },
    {
      "epoch": 8.673667907714844e-06,
      "step": 83095,
      "training_step_time": 7.643442869186401
    },
    {
      "epoch": 8.67462158203125e-06,
      "model_forward_time": 1.745649814605713,
      "step": 83096
    },
    {
      "epoch": 8.67462158203125e-06,
      "step": 83096,
      "training_step_time": 7.46695613861084
    },
    {
      "epoch": 8.675575256347656e-06,
      "model_forward_time": 2.3838605880737305,
      "step": 83097
    },
    {
      "epoch": 8.675575256347656e-06,
      "step": 83097,
      "training_step_time": 7.648518323898315
    },
    {
      "epoch": 8.676528930664063e-06,
      "model_forward_time": 1.7389729022979736,
      "step": 83098
    },
    {
      "epoch": 8.676528930664063e-06,
      "step": 83098,
      "training_step_time": 7.315864324569702
    },
    {
      "epoch": 8.677482604980469e-06,
      "model_forward_time": 1.7390022277832031,
      "step": 83099
    },
    {
      "epoch": 8.677482604980469e-06,
      "step": 83099,
      "training_step_time": 9.430134296417236
    },
    {
      "epoch": 8.678436279296875e-06,
      "grad_norm": 0.07962616533041,
      "learning_rate": 7.608210414972561e-07,
      "loss": 0.0539,
      "step": 83100
    },
    {
      "epoch": 8.678436279296875e-06,
      "model_forward_time": 1.7392706871032715,
      "step": 83100
    },
    {
      "dynamics_loss_avg": 0.05495703211054206,
      "epoch": 8.678436279296875e-06,
      "step": 83100
    },
    {
      "action_loss_avg": 0.0026397429319331422,
      "epoch": 8.678436279296875e-06,
      "step": 83100
    },
    {
      "epoch": 8.678436279296875e-06,
      "step": 83100,
      "training_step_time": 8.63306212425232
    },
    {
      "epoch": 8.679389953613282e-06,
      "model_forward_time": 1.7359447479248047,
      "step": 83101
    },
    {
      "epoch": 8.679389953613282e-06,
      "step": 83101,
      "training_step_time": 7.6727614402771
    },
    {
      "epoch": 8.680343627929688e-06,
      "model_forward_time": 1.7392785549163818,
      "step": 83102
    },
    {
      "epoch": 8.680343627929688e-06,
      "step": 83102,
      "training_step_time": 10.894065141677856
    },
    {
      "epoch": 8.681297302246094e-06,
      "model_forward_time": 1.353747844696045,
      "step": 83103
    },
    {
      "epoch": 8.681297302246094e-06,
      "step": 83103,
      "training_step_time": 7.6619768142700195
    },
    {
      "epoch": 8.6822509765625e-06,
      "model_forward_time": 1.741645336151123,
      "step": 83104
    },
    {
      "epoch": 8.6822509765625e-06,
      "step": 83104,
      "training_step_time": 7.638836145401001
    },
    {
      "epoch": 8.683204650878907e-06,
      "model_forward_time": 1.730381965637207,
      "step": 83105
    },
    {
      "epoch": 8.683204650878907e-06,
      "step": 83105,
      "training_step_time": 8.227978706359863
    },
    {
      "epoch": 8.684158325195312e-06,
      "model_forward_time": 1.737786054611206,
      "step": 83106
    },
    {
      "epoch": 8.684158325195312e-06,
      "step": 83106,
      "training_step_time": 7.597563743591309
    },
    {
      "epoch": 8.685111999511718e-06,
      "model_forward_time": 2.3745131492614746,
      "step": 83107
    },
    {
      "epoch": 8.685111999511718e-06,
      "step": 83107,
      "training_step_time": 7.381238222122192
    },
    {
      "epoch": 8.686065673828126e-06,
      "model_forward_time": 1.7381372451782227,
      "step": 83108
    },
    {
      "epoch": 8.686065673828126e-06,
      "step": 83108,
      "training_step_time": 7.633821964263916
    },
    {
      "epoch": 8.687019348144531e-06,
      "model_forward_time": 1.7384216785430908,
      "step": 83109
    },
    {
      "epoch": 8.687019348144531e-06,
      "step": 83109,
      "training_step_time": 8.682934999465942
    },
    {
      "epoch": 8.687973022460937e-06,
      "grad_norm": 0.08873039484024048,
      "learning_rate": 7.599445062695404e-07,
      "loss": 0.0539,
      "step": 83110
    },
    {
      "epoch": 8.687973022460937e-06,
      "model_forward_time": 1.733659267425537,
      "step": 83110
    },
    {
      "dynamics_loss_avg": 0.04476503748446703,
      "epoch": 8.687973022460937e-06,
      "step": 83110
    },
    {
      "action_loss_avg": 0.0009120320290094241,
      "epoch": 8.687973022460937e-06,
      "step": 83110
    },
    {
      "epoch": 8.687973022460937e-06,
      "step": 83110,
      "training_step_time": 7.7368316650390625
    },
    {
      "epoch": 8.688926696777344e-06,
      "model_forward_time": 1.7372002601623535,
      "step": 83111
    },
    {
      "epoch": 8.688926696777344e-06,
      "step": 83111,
      "training_step_time": 11.080006122589111
    },
    {
      "epoch": 8.68988037109375e-06,
      "model_forward_time": 1.7389931678771973,
      "step": 83112
    },
    {
      "epoch": 8.68988037109375e-06,
      "step": 83112,
      "training_step_time": 9.526808500289917
    },
    {
      "epoch": 8.690834045410156e-06,
      "model_forward_time": 1.7334864139556885,
      "step": 83113
    },
    {
      "epoch": 8.690834045410156e-06,
      "step": 83113,
      "training_step_time": 8.706521272659302
    },
    {
      "epoch": 8.691787719726563e-06,
      "model_forward_time": 1.7330543994903564,
      "step": 83114
    },
    {
      "epoch": 8.691787719726563e-06,
      "step": 83114,
      "training_step_time": 7.60652494430542
    },
    {
      "epoch": 8.692741394042969e-06,
      "model_forward_time": 1.735949993133545,
      "step": 83115
    },
    {
      "epoch": 8.692741394042969e-06,
      "step": 83115,
      "training_step_time": 7.630311727523804
    },
    {
      "epoch": 8.693695068359375e-06,
      "model_forward_time": 2.368640422821045,
      "step": 83116
    },
    {
      "epoch": 8.693695068359375e-06,
      "step": 83116,
      "training_step_time": 8.726016521453857
    },
    {
      "epoch": 8.694648742675782e-06,
      "model_forward_time": 1.731233835220337,
      "step": 83117
    },
    {
      "epoch": 8.694648742675782e-06,
      "step": 83117,
      "training_step_time": 7.634348154067993
    },
    {
      "epoch": 8.695602416992188e-06,
      "model_forward_time": 1.73946213722229,
      "step": 83118
    },
    {
      "epoch": 8.695602416992188e-06,
      "step": 83118,
      "training_step_time": 7.675321340560913
    },
    {
      "epoch": 8.696556091308594e-06,
      "model_forward_time": 1.7506968975067139,
      "step": 83119
    },
    {
      "epoch": 8.696556091308594e-06,
      "step": 83119,
      "training_step_time": 8.241454601287842
    },
    {
      "epoch": 8.697509765625e-06,
      "grad_norm": 0.09044148027896881,
      "learning_rate": 7.590684347279215e-07,
      "loss": 0.0545,
      "step": 83120
    },
    {
      "epoch": 8.697509765625e-06,
      "model_forward_time": 1.7354161739349365,
      "step": 83120
    },
    {
      "dynamics_loss_avg": 0.04760052096098662,
      "epoch": 8.697509765625e-06,
      "step": 83120
    },
    {
      "action_loss_avg": 0.002005613251822069,
      "epoch": 8.697509765625e-06,
      "step": 83120
    },
    {
      "epoch": 8.697509765625e-06,
      "step": 83120,
      "training_step_time": 8.517880201339722
    },
    {
      "epoch": 8.698463439941407e-06,
      "model_forward_time": 1.7317044734954834,
      "step": 83121
    },
    {
      "epoch": 8.698463439941407e-06,
      "step": 83121,
      "training_step_time": 9.540716409683228
    },
    {
      "epoch": 8.699417114257812e-06,
      "model_forward_time": 1.7411394119262695,
      "step": 83122
    },
    {
      "epoch": 8.699417114257812e-06,
      "step": 83122,
      "training_step_time": 7.700988292694092
    },
    {
      "epoch": 8.700370788574218e-06,
      "model_forward_time": 1.7372689247131348,
      "step": 83123
    },
    {
      "epoch": 8.700370788574218e-06,
      "step": 83123,
      "training_step_time": 7.695671796798706
    },
    {
      "epoch": 8.701324462890626e-06,
      "model_forward_time": 1.7362258434295654,
      "step": 83124
    },
    {
      "epoch": 8.701324462890626e-06,
      "step": 83124,
      "training_step_time": 7.47046971321106
    },
    {
      "epoch": 8.702278137207031e-06,
      "model_forward_time": 2.407499313354492,
      "step": 83125
    },
    {
      "epoch": 8.702278137207031e-06,
      "step": 83125,
      "training_step_time": 7.684880971908569
    },
    {
      "epoch": 8.703231811523437e-06,
      "model_forward_time": 1.7395496368408203,
      "step": 83126
    },
    {
      "epoch": 8.703231811523437e-06,
      "step": 83126,
      "training_step_time": 7.61451530456543
    },
    {
      "epoch": 8.704185485839844e-06,
      "model_forward_time": 1.738372564315796,
      "step": 83127
    },
    {
      "epoch": 8.704185485839844e-06,
      "step": 83127,
      "training_step_time": 7.712567090988159
    },
    {
      "epoch": 8.70513916015625e-06,
      "model_forward_time": 1.742875576019287,
      "step": 83128
    },
    {
      "epoch": 8.70513916015625e-06,
      "step": 83128,
      "training_step_time": 8.502877950668335
    },
    {
      "epoch": 8.706092834472656e-06,
      "model_forward_time": 1.7364811897277832,
      "step": 83129
    },
    {
      "epoch": 8.706092834472656e-06,
      "step": 83129,
      "training_step_time": 8.901215076446533
    },
    {
      "epoch": 8.707046508789063e-06,
      "grad_norm": 0.10281708836555481,
      "learning_rate": 7.581928269682004e-07,
      "loss": 0.0532,
      "step": 83130
    },
    {
      "epoch": 8.707046508789063e-06,
      "model_forward_time": 1.7402284145355225,
      "step": 83130
    },
    {
      "dynamics_loss_avg": 0.04515906199812889,
      "epoch": 8.707046508789063e-06,
      "step": 83130
    },
    {
      "action_loss_avg": 0.0025767229846678672,
      "epoch": 8.707046508789063e-06,
      "step": 83130
    },
    {
      "epoch": 8.707046508789063e-06,
      "step": 83130,
      "training_step_time": 8.397122621536255
    },
    {
      "epoch": 8.708000183105469e-06,
      "model_forward_time": 1.7986948490142822,
      "step": 83131
    },
    {
      "epoch": 8.708000183105469e-06,
      "step": 83131,
      "training_step_time": 8.266397953033447
    },
    {
      "epoch": 8.708953857421875e-06,
      "model_forward_time": 1.741969347000122,
      "step": 83132
    },
    {
      "epoch": 8.708953857421875e-06,
      "step": 83132,
      "training_step_time": 7.551415681838989
    },
    {
      "epoch": 8.70990753173828e-06,
      "model_forward_time": 1.7363314628601074,
      "step": 83133
    },
    {
      "epoch": 8.70990753173828e-06,
      "step": 83133,
      "training_step_time": 7.623254060745239
    },
    {
      "epoch": 8.710861206054688e-06,
      "model_forward_time": 2.4032437801361084,
      "step": 83134
    },
    {
      "epoch": 8.710861206054688e-06,
      "step": 83134,
      "training_step_time": 7.674649238586426
    },
    {
      "epoch": 8.711814880371094e-06,
      "model_forward_time": 1.743370771408081,
      "step": 83135
    },
    {
      "epoch": 8.711814880371094e-06,
      "step": 83135,
      "training_step_time": 7.609950065612793
    },
    {
      "epoch": 8.7127685546875e-06,
      "model_forward_time": 1.7426378726959229,
      "step": 83136
    },
    {
      "epoch": 8.7127685546875e-06,
      "step": 83136,
      "training_step_time": 7.632061719894409
    },
    {
      "epoch": 8.713722229003907e-06,
      "model_forward_time": 1.7515535354614258,
      "step": 83137
    },
    {
      "epoch": 8.713722229003907e-06,
      "step": 83137,
      "training_step_time": 8.235793590545654
    },
    {
      "epoch": 8.714675903320312e-06,
      "model_forward_time": 1.7404513359069824,
      "step": 83138
    },
    {
      "epoch": 8.714675903320312e-06,
      "step": 83138,
      "training_step_time": 8.648396015167236
    },
    {
      "epoch": 8.715629577636718e-06,
      "model_forward_time": 1.735443115234375,
      "step": 83139
    },
    {
      "epoch": 8.715629577636718e-06,
      "step": 83139,
      "training_step_time": 10.455079317092896
    },
    {
      "epoch": 8.716583251953126e-06,
      "grad_norm": 0.10732249170541763,
      "learning_rate": 7.573176830861362e-07,
      "loss": 0.0538,
      "step": 83140
    },
    {
      "epoch": 8.716583251953126e-06,
      "model_forward_time": 1.7453179359436035,
      "step": 83140
    },
    {
      "dynamics_loss_avg": 0.05571259949356318,
      "epoch": 8.716583251953126e-06,
      "step": 83140
    },
    {
      "action_loss_avg": 0.0014560168841853739,
      "epoch": 8.716583251953126e-06,
      "step": 83140
    },
    {
      "epoch": 8.716583251953126e-06,
      "step": 83140,
      "training_step_time": 9.202053308486938
    },
    {
      "epoch": 8.717536926269531e-06,
      "model_forward_time": 1.7286098003387451,
      "step": 83141
    },
    {
      "epoch": 8.717536926269531e-06,
      "step": 83141,
      "training_step_time": 8.538158655166626
    },
    {
      "epoch": 8.718490600585937e-06,
      "model_forward_time": 1.7325432300567627,
      "step": 83142
    },
    {
      "epoch": 8.718490600585937e-06,
      "step": 83142,
      "training_step_time": 8.583983421325684
    },
    {
      "epoch": 8.719444274902344e-06,
      "model_forward_time": 2.345919609069824,
      "step": 83143
    },
    {
      "epoch": 8.719444274902344e-06,
      "step": 83143,
      "training_step_time": 7.707193851470947
    },
    {
      "epoch": 8.72039794921875e-06,
      "model_forward_time": 1.7383053302764893,
      "step": 83144
    },
    {
      "epoch": 8.72039794921875e-06,
      "step": 83144,
      "training_step_time": 7.661953687667847
    },
    {
      "epoch": 8.721351623535156e-06,
      "model_forward_time": 1.733057975769043,
      "step": 83145
    },
    {
      "epoch": 8.721351623535156e-06,
      "step": 83145,
      "training_step_time": 8.754709243774414
    },
    {
      "epoch": 8.722305297851563e-06,
      "model_forward_time": 1.751749038696289,
      "step": 83146
    },
    {
      "epoch": 8.722305297851563e-06,
      "step": 83146,
      "training_step_time": 7.650169849395752
    },
    {
      "epoch": 8.723258972167969e-06,
      "model_forward_time": 1.735346794128418,
      "step": 83147
    },
    {
      "epoch": 8.723258972167969e-06,
      "step": 83147,
      "training_step_time": 9.54318380355835
    },
    {
      "epoch": 8.724212646484375e-06,
      "model_forward_time": 1.7346088886260986,
      "step": 83148
    },
    {
      "epoch": 8.724212646484375e-06,
      "step": 83148,
      "training_step_time": 7.817156076431274
    },
    {
      "epoch": 8.72516632080078e-06,
      "model_forward_time": 1.7418692111968994,
      "step": 83149
    },
    {
      "epoch": 8.72516632080078e-06,
      "step": 83149,
      "training_step_time": 7.714228391647339
    },
    {
      "epoch": 8.726119995117188e-06,
      "grad_norm": 0.07226061820983887,
      "learning_rate": 7.5644300317743e-07,
      "loss": 0.0534,
      "step": 83150
    },
    {
      "epoch": 8.726119995117188e-06,
      "model_forward_time": 1.7320466041564941,
      "step": 83150
    },
    {
      "dynamics_loss_avg": 0.04265642575919628,
      "epoch": 8.726119995117188e-06,
      "step": 83150
    },
    {
      "action_loss_avg": 0.0019193938438547774,
      "epoch": 8.726119995117188e-06,
      "step": 83150
    },
    {
      "epoch": 8.726119995117188e-06,
      "step": 83150,
      "training_step_time": 7.603593587875366
    },
    {
      "epoch": 8.727073669433594e-06,
      "model_forward_time": 1.7341082096099854,
      "step": 83151
    },
    {
      "epoch": 8.727073669433594e-06,
      "step": 83151,
      "training_step_time": 8.27510929107666
    },
    {
      "epoch": 8.72802734375e-06,
      "model_forward_time": 1.735318660736084,
      "step": 83152
    },
    {
      "epoch": 8.72802734375e-06,
      "step": 83152,
      "training_step_time": 7.53549337387085
    },
    {
      "epoch": 8.728981018066407e-06,
      "model_forward_time": 2.3843538761138916,
      "step": 83153
    },
    {
      "epoch": 8.728981018066407e-06,
      "step": 83153,
      "training_step_time": 8.634473085403442
    },
    {
      "epoch": 8.729934692382812e-06,
      "model_forward_time": 1.7379837036132812,
      "step": 83154
    },
    {
      "epoch": 8.729934692382812e-06,
      "step": 83154,
      "training_step_time": 7.693105459213257
    },
    {
      "epoch": 8.730888366699218e-06,
      "model_forward_time": 1.7393913269042969,
      "step": 83155
    },
    {
      "epoch": 8.730888366699218e-06,
      "step": 83155,
      "training_step_time": 7.644592761993408
    },
    {
      "epoch": 8.731842041015626e-06,
      "model_forward_time": 1.7313225269317627,
      "step": 83156
    },
    {
      "epoch": 8.731842041015626e-06,
      "step": 83156,
      "training_step_time": 8.287102699279785
    },
    {
      "epoch": 8.732795715332031e-06,
      "model_forward_time": 1.7340991497039795,
      "step": 83157
    },
    {
      "epoch": 8.732795715332031e-06,
      "step": 83157,
      "training_step_time": 9.295636415481567
    },
    {
      "epoch": 8.733749389648437e-06,
      "model_forward_time": 1.7346949577331543,
      "step": 83158
    },
    {
      "epoch": 8.733749389648437e-06,
      "step": 83158,
      "training_step_time": 9.457043409347534
    },
    {
      "epoch": 8.734703063964844e-06,
      "model_forward_time": 1.7289767265319824,
      "step": 83159
    },
    {
      "epoch": 8.734703063964844e-06,
      "step": 83159,
      "training_step_time": 8.708502054214478
    },
    {
      "epoch": 8.73565673828125e-06,
      "grad_norm": 0.08340173214673996,
      "learning_rate": 7.555687873377376e-07,
      "loss": 0.0536,
      "step": 83160
    },
    {
      "epoch": 8.73565673828125e-06,
      "model_forward_time": 1.7221174240112305,
      "step": 83160
    },
    {
      "dynamics_loss_avg": 0.060076946020126344,
      "epoch": 8.73565673828125e-06,
      "step": 83160
    },
    {
      "action_loss_avg": 0.002031224683014443,
      "epoch": 8.73565673828125e-06,
      "step": 83160
    },
    {
      "epoch": 8.73565673828125e-06,
      "step": 83160,
      "training_step_time": 8.066508769989014
    },
    {
      "epoch": 8.736610412597656e-06,
      "model_forward_time": 1.3502075672149658,
      "step": 83161
    },
    {
      "epoch": 8.736610412597656e-06,
      "step": 83161,
      "training_step_time": 7.4272096157073975
    },
    {
      "epoch": 8.737564086914063e-06,
      "model_forward_time": 2.3956868648529053,
      "step": 83162
    },
    {
      "epoch": 8.737564086914063e-06,
      "step": 83162,
      "training_step_time": 7.510666370391846
    },
    {
      "epoch": 8.738517761230469e-06,
      "model_forward_time": 1.7356772422790527,
      "step": 83163
    },
    {
      "epoch": 8.738517761230469e-06,
      "step": 83163,
      "training_step_time": 8.385971546173096
    },
    {
      "epoch": 8.739471435546875e-06,
      "model_forward_time": 1.733090877532959,
      "step": 83164
    },
    {
      "epoch": 8.739471435546875e-06,
      "step": 83164,
      "training_step_time": 7.620293617248535
    },
    {
      "epoch": 8.74042510986328e-06,
      "model_forward_time": 1.7537012100219727,
      "step": 83165
    },
    {
      "epoch": 8.74042510986328e-06,
      "step": 83165,
      "training_step_time": 8.697729349136353
    },
    {
      "epoch": 8.741378784179688e-06,
      "model_forward_time": 1.7220680713653564,
      "step": 83166
    },
    {
      "epoch": 8.741378784179688e-06,
      "step": 83166,
      "training_step_time": 9.104742288589478
    },
    {
      "epoch": 8.742332458496094e-06,
      "model_forward_time": 1.7363801002502441,
      "step": 83167
    },
    {
      "epoch": 8.742332458496094e-06,
      "step": 83167,
      "training_step_time": 8.525103330612183
    },
    {
      "epoch": 8.7432861328125e-06,
      "model_forward_time": 1.7531805038452148,
      "step": 83168
    },
    {
      "epoch": 8.7432861328125e-06,
      "step": 83168,
      "training_step_time": 7.63892674446106
    },
    {
      "epoch": 8.744239807128907e-06,
      "model_forward_time": 1.7346222400665283,
      "step": 83169
    },
    {
      "epoch": 8.744239807128907e-06,
      "step": 83169,
      "training_step_time": 7.6837427616119385
    },
    {
      "epoch": 8.745193481445312e-06,
      "grad_norm": 0.06208541616797447,
      "learning_rate": 7.546950356626592e-07,
      "loss": 0.0538,
      "step": 83170
    },
    {
      "epoch": 8.745193481445312e-06,
      "model_forward_time": 1.7307896614074707,
      "step": 83170
    },
    {
      "dynamics_loss_avg": 0.05949931684881449,
      "epoch": 8.745193481445312e-06,
      "step": 83170
    },
    {
      "action_loss_avg": 0.0020256553616491146,
      "epoch": 8.745193481445312e-06,
      "step": 83170
    },
    {
      "epoch": 8.745193481445312e-06,
      "step": 83170,
      "training_step_time": 7.5356221199035645
    },
    {
      "epoch": 8.746147155761718e-06,
      "model_forward_time": 2.3782615661621094,
      "step": 83171
    },
    {
      "epoch": 8.746147155761718e-06,
      "step": 83171,
      "training_step_time": 8.163991212844849
    },
    {
      "epoch": 8.747100830078126e-06,
      "model_forward_time": 1.7330517768859863,
      "step": 83172
    },
    {
      "epoch": 8.747100830078126e-06,
      "step": 83172,
      "training_step_time": 7.712842702865601
    },
    {
      "epoch": 8.748054504394531e-06,
      "model_forward_time": 1.7330243587493896,
      "step": 83173
    },
    {
      "epoch": 8.748054504394531e-06,
      "step": 83173,
      "training_step_time": 7.5900797843933105
    },
    {
      "epoch": 8.749008178710937e-06,
      "model_forward_time": 1.7465996742248535,
      "step": 83174
    },
    {
      "epoch": 8.749008178710937e-06,
      "step": 83174,
      "training_step_time": 8.374057054519653
    },
    {
      "epoch": 8.749961853027344e-06,
      "model_forward_time": 1.7276909351348877,
      "step": 83175
    },
    {
      "epoch": 8.749961853027344e-06,
      "step": 83175,
      "training_step_time": 7.674954652786255
    },
    {
      "epoch": 8.75091552734375e-06,
      "model_forward_time": 1.7310664653778076,
      "step": 83176
    },
    {
      "epoch": 8.75091552734375e-06,
      "step": 83176,
      "training_step_time": 8.533157348632812
    },
    {
      "epoch": 8.751869201660156e-06,
      "model_forward_time": 1.7614109516143799,
      "step": 83177
    },
    {
      "epoch": 8.751869201660156e-06,
      "step": 83177,
      "training_step_time": 8.633508205413818
    },
    {
      "epoch": 8.752822875976563e-06,
      "model_forward_time": 1.734302282333374,
      "step": 83178
    },
    {
      "epoch": 8.752822875976563e-06,
      "step": 83178,
      "training_step_time": 8.695024967193604
    },
    {
      "epoch": 8.753776550292969e-06,
      "model_forward_time": 1.0024540424346924,
      "step": 83179
    },
    {
      "epoch": 8.753776550292969e-06,
      "step": 83179,
      "training_step_time": 7.628223180770874
    },
    {
      "epoch": 8.754730224609375e-06,
      "grad_norm": 0.08278446644544601,
      "learning_rate": 7.538217482477516e-07,
      "loss": 0.054,
      "step": 83180
    },
    {
      "epoch": 8.754730224609375e-06,
      "model_forward_time": 2.336069107055664,
      "step": 83180
    },
    {
      "dynamics_loss_avg": 0.06099640037864447,
      "epoch": 8.754730224609375e-06,
      "step": 83180
    },
    {
      "action_loss_avg": 0.001621733684442006,
      "epoch": 8.754730224609375e-06,
      "step": 83180
    },
    {
      "epoch": 8.754730224609375e-06,
      "step": 83180,
      "training_step_time": 8.161405086517334
    },
    {
      "epoch": 8.75568389892578e-06,
      "model_forward_time": 1.7323641777038574,
      "step": 83181
    },
    {
      "epoch": 8.75568389892578e-06,
      "step": 83181,
      "training_step_time": 7.665424585342407
    },
    {
      "epoch": 8.756637573242188e-06,
      "model_forward_time": 1.734753131866455,
      "step": 83182
    },
    {
      "epoch": 8.756637573242188e-06,
      "step": 83182,
      "training_step_time": 7.5856547355651855
    },
    {
      "epoch": 8.757591247558594e-06,
      "model_forward_time": 1.7451777458190918,
      "step": 83183
    },
    {
      "epoch": 8.757591247558594e-06,
      "step": 83183,
      "training_step_time": 7.612542390823364
    },
    {
      "epoch": 8.758544921875e-06,
      "model_forward_time": 1.730154037475586,
      "step": 83184
    },
    {
      "epoch": 8.758544921875e-06,
      "step": 83184,
      "training_step_time": 8.192850828170776
    },
    {
      "epoch": 8.759498596191407e-06,
      "model_forward_time": 1.7334754467010498,
      "step": 83185
    },
    {
      "epoch": 8.759498596191407e-06,
      "step": 83185,
      "training_step_time": 8.601855039596558
    },
    {
      "epoch": 8.760452270507812e-06,
      "model_forward_time": 1.757495641708374,
      "step": 83186
    },
    {
      "epoch": 8.760452270507812e-06,
      "step": 83186,
      "training_step_time": 8.278351068496704
    },
    {
      "epoch": 8.761405944824218e-06,
      "model_forward_time": 1.7317171096801758,
      "step": 83187
    },
    {
      "epoch": 8.761405944824218e-06,
      "step": 83187,
      "training_step_time": 8.507908821105957
    },
    {
      "epoch": 8.762359619140626e-06,
      "model_forward_time": 1.732576608657837,
      "step": 83188
    },
    {
      "epoch": 8.762359619140626e-06,
      "step": 83188,
      "training_step_time": 7.542260646820068
    },
    {
      "epoch": 8.763313293457031e-06,
      "model_forward_time": 1.733689308166504,
      "step": 83189
    },
    {
      "epoch": 8.763313293457031e-06,
      "step": 83189,
      "training_step_time": 7.49274206161499
    },
    {
      "epoch": 8.764266967773437e-06,
      "grad_norm": 0.08579665422439575,
      "learning_rate": 7.529489251885119e-07,
      "loss": 0.0522,
      "step": 83190
    },
    {
      "epoch": 8.764266967773437e-06,
      "model_forward_time": 2.378358840942383,
      "step": 83190
    },
    {
      "dynamics_loss_avg": 0.06525275455787778,
      "epoch": 8.764266967773437e-06,
      "step": 83190
    },
    {
      "action_loss_avg": 0.0030189047916792332,
      "epoch": 8.764266967773437e-06,
      "step": 83190
    },
    {
      "epoch": 8.764266967773437e-06,
      "step": 83190,
      "training_step_time": 7.646186828613281
    },
    {
      "epoch": 8.765220642089844e-06,
      "model_forward_time": 1.7346007823944092,
      "step": 83191
    },
    {
      "epoch": 8.765220642089844e-06,
      "step": 83191,
      "training_step_time": 7.5229620933532715
    },
    {
      "epoch": 8.76617431640625e-06,
      "model_forward_time": 1.3524293899536133,
      "step": 83192
    },
    {
      "epoch": 8.76617431640625e-06,
      "step": 83192,
      "training_step_time": 7.545603036880493
    },
    {
      "epoch": 8.767127990722656e-06,
      "model_forward_time": 1.7390952110290527,
      "step": 83193
    },
    {
      "epoch": 8.767127990722656e-06,
      "step": 83193,
      "training_step_time": 7.669360160827637
    },
    {
      "epoch": 8.768081665039063e-06,
      "model_forward_time": 1.7311921119689941,
      "step": 83194
    },
    {
      "epoch": 8.768081665039063e-06,
      "step": 83194,
      "training_step_time": 9.943050861358643
    },
    {
      "epoch": 8.769035339355469e-06,
      "model_forward_time": 1.7608070373535156,
      "step": 83195
    },
    {
      "epoch": 8.769035339355469e-06,
      "step": 83195,
      "training_step_time": 9.304907321929932
    },
    {
      "epoch": 8.769989013671875e-06,
      "model_forward_time": 1.7371857166290283,
      "step": 83196
    },
    {
      "epoch": 8.769989013671875e-06,
      "step": 83196,
      "training_step_time": 7.639382362365723
    },
    {
      "epoch": 8.77094268798828e-06,
      "model_forward_time": 1.7338464260101318,
      "step": 83197
    },
    {
      "epoch": 8.77094268798828e-06,
      "step": 83197,
      "training_step_time": 7.507141351699829
    },
    {
      "epoch": 8.771896362304688e-06,
      "model_forward_time": 1.7377960681915283,
      "step": 83198
    },
    {
      "epoch": 8.771896362304688e-06,
      "step": 83198,
      "training_step_time": 7.353635311126709
    },
    {
      "epoch": 8.772850036621094e-06,
      "model_forward_time": 2.3996951580047607,
      "step": 83199
    },
    {
      "epoch": 8.772850036621094e-06,
      "step": 83199,
      "training_step_time": 9.097991228103638
    },
    {
      "epoch": 8.7738037109375e-06,
      "grad_norm": 0.08746524155139923,
      "learning_rate": 7.520765665803942e-07,
      "loss": 0.0542,
      "step": 83200
    },
    {
      "epoch": 8.7738037109375e-06,
      "model_forward_time": 1.7260446548461914,
      "step": 83200
    },
    {
      "dynamics_loss_avg": 0.05632096901535988,
      "epoch": 8.7738037109375e-06,
      "step": 83200
    },
    {
      "action_loss_avg": 0.001415574188285973,
      "epoch": 8.7738037109375e-06,
      "step": 83200
    },
    {
      "epoch": 8.7738037109375e-06,
      "step": 83200,
      "training_step_time": 7.2864906787872314
    },
    {
      "epoch": 8.774757385253907e-06,
      "model_forward_time": 1.7345049381256104,
      "step": 83201
    },
    {
      "epoch": 8.774757385253907e-06,
      "step": 83201,
      "training_step_time": 9.667918682098389
    },
    {
      "epoch": 8.775711059570312e-06,
      "model_forward_time": 1.7306742668151855,
      "step": 83202
    },
    {
      "epoch": 8.775711059570312e-06,
      "step": 83202,
      "training_step_time": 7.643054962158203
    },
    {
      "epoch": 8.776664733886718e-06,
      "model_forward_time": 1.7357456684112549,
      "step": 83203
    },
    {
      "epoch": 8.776664733886718e-06,
      "step": 83203,
      "training_step_time": 8.712384939193726
    },
    {
      "epoch": 8.777618408203126e-06,
      "model_forward_time": 1.7552311420440674,
      "step": 83204
    },
    {
      "epoch": 8.777618408203126e-06,
      "step": 83204,
      "training_step_time": 9.91968059539795
    },
    {
      "epoch": 8.778572082519531e-06,
      "model_forward_time": 1.7325541973114014,
      "step": 83205
    },
    {
      "epoch": 8.778572082519531e-06,
      "step": 83205,
      "training_step_time": 8.236770629882812
    },
    {
      "epoch": 8.779525756835937e-06,
      "model_forward_time": 1.7313907146453857,
      "step": 83206
    },
    {
      "epoch": 8.779525756835937e-06,
      "step": 83206,
      "training_step_time": 7.5762035846710205
    },
    {
      "epoch": 8.780479431152344e-06,
      "model_forward_time": 1.732508897781372,
      "step": 83207
    },
    {
      "epoch": 8.780479431152344e-06,
      "step": 83207,
      "training_step_time": 7.611471176147461
    },
    {
      "epoch": 8.78143310546875e-06,
      "model_forward_time": 2.416459798812866,
      "step": 83208
    },
    {
      "epoch": 8.78143310546875e-06,
      "step": 83208,
      "training_step_time": 8.605591058731079
    },
    {
      "epoch": 8.782386779785156e-06,
      "model_forward_time": 1.731745958328247,
      "step": 83209
    },
    {
      "epoch": 8.782386779785156e-06,
      "step": 83209,
      "training_step_time": 7.282804250717163
    },
    {
      "epoch": 8.783340454101563e-06,
      "grad_norm": 0.11966472864151001,
      "learning_rate": 7.512046725187949e-07,
      "loss": 0.0528,
      "step": 83210
    },
    {
      "epoch": 8.783340454101563e-06,
      "model_forward_time": 1.7370495796203613,
      "step": 83210
    },
    {
      "dynamics_loss_avg": 0.04758307198062539,
      "epoch": 8.783340454101563e-06,
      "step": 83210
    },
    {
      "action_loss_avg": 0.0012088721588952467,
      "epoch": 8.783340454101563e-06,
      "step": 83210
    },
    {
      "epoch": 8.783340454101563e-06,
      "step": 83210,
      "training_step_time": 9.138582706451416
    },
    {
      "epoch": 8.784294128417969e-06,
      "model_forward_time": 1.7492084503173828,
      "step": 83211
    },
    {
      "epoch": 8.784294128417969e-06,
      "step": 83211,
      "training_step_time": 7.6005730628967285
    },
    {
      "epoch": 8.785247802734375e-06,
      "model_forward_time": 1.7335495948791504,
      "step": 83212
    },
    {
      "epoch": 8.785247802734375e-06,
      "step": 83212,
      "training_step_time": 9.309142827987671
    },
    {
      "epoch": 8.78620147705078e-06,
      "model_forward_time": 1.752807378768921,
      "step": 83213
    },
    {
      "epoch": 8.78620147705078e-06,
      "step": 83213,
      "training_step_time": 10.213624238967896
    },
    {
      "epoch": 8.787155151367188e-06,
      "model_forward_time": 1.7309939861297607,
      "step": 83214
    },
    {
      "epoch": 8.787155151367188e-06,
      "step": 83214,
      "training_step_time": 7.671765565872192
    },
    {
      "epoch": 8.788108825683594e-06,
      "model_forward_time": 1.7308998107910156,
      "step": 83215
    },
    {
      "epoch": 8.788108825683594e-06,
      "step": 83215,
      "training_step_time": 7.640369176864624
    },
    {
      "epoch": 8.7890625e-06,
      "model_forward_time": 1.7342371940612793,
      "step": 83216
    },
    {
      "epoch": 8.7890625e-06,
      "step": 83216,
      "training_step_time": 7.595845937728882
    },
    {
      "epoch": 8.790016174316407e-06,
      "model_forward_time": 2.3790836334228516,
      "step": 83217
    },
    {
      "epoch": 8.790016174316407e-06,
      "step": 83217,
      "training_step_time": 10.335429906845093
    },
    {
      "epoch": 8.790969848632812e-06,
      "model_forward_time": 1.7244153022766113,
      "step": 83218
    },
    {
      "epoch": 8.790969848632812e-06,
      "step": 83218,
      "training_step_time": 8.477476358413696
    },
    {
      "epoch": 8.791923522949218e-06,
      "model_forward_time": 1.7353951930999756,
      "step": 83219
    },
    {
      "epoch": 8.791923522949218e-06,
      "step": 83219,
      "training_step_time": 8.621665239334106
    },
    {
      "epoch": 8.792877197265626e-06,
      "grad_norm": 0.07452342659235,
      "learning_rate": 7.503332430990646e-07,
      "loss": 0.0532,
      "step": 83220
    },
    {
      "epoch": 8.792877197265626e-06,
      "model_forward_time": 1.7422142028808594,
      "step": 83220
    },
    {
      "dynamics_loss_avg": 0.0487011045217514,
      "epoch": 8.792877197265626e-06,
      "step": 83220
    },
    {
      "action_loss_avg": 0.0024628988816402853,
      "epoch": 8.792877197265626e-06,
      "step": 83220
    },
    {
      "epoch": 8.792877197265626e-06,
      "step": 83220,
      "training_step_time": 8.110283851623535
    },
    {
      "epoch": 8.793830871582031e-06,
      "model_forward_time": 1.7306954860687256,
      "step": 83221
    },
    {
      "epoch": 8.793830871582031e-06,
      "step": 83221,
      "training_step_time": 8.12592625617981
    },
    {
      "epoch": 8.794784545898437e-06,
      "model_forward_time": 1.7542915344238281,
      "step": 83222
    },
    {
      "epoch": 8.794784545898437e-06,
      "step": 83222,
      "training_step_time": 8.46310567855835
    },
    {
      "epoch": 8.795738220214844e-06,
      "model_forward_time": 1.7412099838256836,
      "step": 83223
    },
    {
      "epoch": 8.795738220214844e-06,
      "step": 83223,
      "training_step_time": 10.254134893417358
    },
    {
      "epoch": 8.79669189453125e-06,
      "model_forward_time": 1.7275099754333496,
      "step": 83224
    },
    {
      "epoch": 8.79669189453125e-06,
      "step": 83224,
      "training_step_time": 7.622210264205933
    },
    {
      "epoch": 8.797645568847656e-06,
      "model_forward_time": 1.7372455596923828,
      "step": 83225
    },
    {
      "epoch": 8.797645568847656e-06,
      "step": 83225,
      "training_step_time": 8.43909478187561
    },
    {
      "epoch": 8.798599243164063e-06,
      "model_forward_time": 1.7288625240325928,
      "step": 83226
    },
    {
      "epoch": 8.798599243164063e-06,
      "step": 83226,
      "training_step_time": 9.003173828125
    },
    {
      "epoch": 8.799552917480469e-06,
      "model_forward_time": 1.7334458827972412,
      "step": 83227
    },
    {
      "epoch": 8.799552917480469e-06,
      "step": 83227,
      "training_step_time": 7.751508474349976
    },
    {
      "epoch": 8.800506591796875e-06,
      "model_forward_time": 1.71950101852417,
      "step": 83228
    },
    {
      "epoch": 8.800506591796875e-06,
      "step": 83228,
      "training_step_time": 7.574145317077637
    },
    {
      "epoch": 8.80146026611328e-06,
      "model_forward_time": 1.7422220706939697,
      "step": 83229
    },
    {
      "epoch": 8.80146026611328e-06,
      "step": 83229,
      "training_step_time": 7.587671518325806
    },
    {
      "epoch": 8.802413940429688e-06,
      "grad_norm": 0.09395677596330643,
      "learning_rate": 7.494622784165018e-07,
      "loss": 0.0533,
      "step": 83230
    },
    {
      "epoch": 8.802413940429688e-06,
      "model_forward_time": 1.7274556159973145,
      "step": 83230
    },
    {
      "dynamics_loss_avg": 0.06941328346729278,
      "epoch": 8.802413940429688e-06,
      "step": 83230
    },
    {
      "action_loss_avg": 0.0011460475507192314,
      "epoch": 8.802413940429688e-06,
      "step": 83230
    },
    {
      "epoch": 8.802413940429688e-06,
      "step": 83230,
      "training_step_time": 7.68585205078125
    },
    {
      "epoch": 8.803367614746094e-06,
      "model_forward_time": 1.7452924251556396,
      "step": 83231
    },
    {
      "epoch": 8.803367614746094e-06,
      "step": 83231,
      "training_step_time": 7.70673131942749
    },
    {
      "epoch": 8.8043212890625e-06,
      "model_forward_time": 1.7254161834716797,
      "step": 83232
    },
    {
      "epoch": 8.8043212890625e-06,
      "step": 83232,
      "training_step_time": 7.714833736419678
    },
    {
      "epoch": 8.805274963378907e-06,
      "model_forward_time": 1.7243313789367676,
      "step": 83233
    },
    {
      "epoch": 8.805274963378907e-06,
      "step": 83233,
      "training_step_time": 7.6415629386901855
    },
    {
      "epoch": 8.806228637695312e-06,
      "model_forward_time": 1.733557939529419,
      "step": 83234
    },
    {
      "epoch": 8.806228637695312e-06,
      "step": 83234,
      "training_step_time": 7.479648590087891
    },
    {
      "epoch": 8.807182312011718e-06,
      "model_forward_time": 1.727268934249878,
      "step": 83235
    },
    {
      "epoch": 8.807182312011718e-06,
      "step": 83235,
      "training_step_time": 7.6206347942352295
    },
    {
      "epoch": 8.808135986328126e-06,
      "model_forward_time": 2.3679792881011963,
      "step": 83236
    },
    {
      "epoch": 8.808135986328126e-06,
      "step": 83236,
      "training_step_time": 7.75483512878418
    },
    {
      "epoch": 8.809089660644531e-06,
      "model_forward_time": 1.7303996086120605,
      "step": 83237
    },
    {
      "epoch": 8.809089660644531e-06,
      "step": 83237,
      "training_step_time": 7.592696666717529
    },
    {
      "epoch": 8.810043334960937e-06,
      "model_forward_time": 1.7386956214904785,
      "step": 83238
    },
    {
      "epoch": 8.810043334960937e-06,
      "step": 83238,
      "training_step_time": 7.606176853179932
    },
    {
      "epoch": 8.810997009277344e-06,
      "model_forward_time": 1.7326860427856445,
      "step": 83239
    },
    {
      "epoch": 8.810997009277344e-06,
      "step": 83239,
      "training_step_time": 8.226197719573975
    },
    {
      "epoch": 8.81195068359375e-06,
      "grad_norm": 0.07621567696332932,
      "learning_rate": 7.485917785663543e-07,
      "loss": 0.054,
      "step": 83240
    },
    {
      "epoch": 8.81195068359375e-06,
      "model_forward_time": 1.753265142440796,
      "step": 83240
    },
    {
      "dynamics_loss_avg": 0.04472826067358256,
      "epoch": 8.81195068359375e-06,
      "step": 83240
    },
    {
      "action_loss_avg": 0.0021883527311729266,
      "epoch": 8.81195068359375e-06,
      "step": 83240
    },
    {
      "epoch": 8.81195068359375e-06,
      "step": 83240,
      "training_step_time": 7.697526216506958
    },
    {
      "epoch": 8.812904357910156e-06,
      "model_forward_time": 1.738422155380249,
      "step": 83241
    },
    {
      "epoch": 8.812904357910156e-06,
      "step": 83241,
      "training_step_time": 8.366331577301025
    },
    {
      "epoch": 8.813858032226563e-06,
      "model_forward_time": 1.7355964183807373,
      "step": 83242
    },
    {
      "epoch": 8.813858032226563e-06,
      "step": 83242,
      "training_step_time": 8.274649620056152
    },
    {
      "epoch": 8.814811706542969e-06,
      "model_forward_time": 1.7366650104522705,
      "step": 83243
    },
    {
      "epoch": 8.814811706542969e-06,
      "step": 83243,
      "training_step_time": 7.594056129455566
    },
    {
      "epoch": 8.815765380859375e-06,
      "model_forward_time": 1.737473964691162,
      "step": 83244
    },
    {
      "epoch": 8.815765380859375e-06,
      "step": 83244,
      "training_step_time": 8.655016422271729
    },
    {
      "epoch": 8.81671905517578e-06,
      "model_forward_time": 2.3874919414520264,
      "step": 83245
    },
    {
      "epoch": 8.81671905517578e-06,
      "step": 83245,
      "training_step_time": 7.586082458496094
    },
    {
      "epoch": 8.817672729492188e-06,
      "model_forward_time": 1.7326390743255615,
      "step": 83246
    },
    {
      "epoch": 8.817672729492188e-06,
      "step": 83246,
      "training_step_time": 7.637054204940796
    },
    {
      "epoch": 8.818626403808594e-06,
      "model_forward_time": 1.733849287033081,
      "step": 83247
    },
    {
      "epoch": 8.818626403808594e-06,
      "step": 83247,
      "training_step_time": 9.576764345169067
    },
    {
      "epoch": 8.819580078125e-06,
      "model_forward_time": 1.7355647087097168,
      "step": 83248
    },
    {
      "epoch": 8.819580078125e-06,
      "step": 83248,
      "training_step_time": 7.58726692199707
    },
    {
      "epoch": 8.820533752441407e-06,
      "model_forward_time": 1.740859031677246,
      "step": 83249
    },
    {
      "epoch": 8.820533752441407e-06,
      "step": 83249,
      "training_step_time": 8.544655561447144
    },
    {
      "epoch": 8.821487426757812e-06,
      "grad_norm": 0.07143743336200714,
      "learning_rate": 7.477217436438172e-07,
      "loss": 0.0532,
      "step": 83250
    },
    {
      "epoch": 8.821487426757812e-06,
      "model_forward_time": 1.766822099685669,
      "step": 83250
    },
    {
      "dynamics_loss_avg": 0.0518654752522707,
      "epoch": 8.821487426757812e-06,
      "step": 83250
    },
    {
      "action_loss_avg": 0.0017634980787988753,
      "epoch": 8.821487426757812e-06,
      "step": 83250
    },
    {
      "epoch": 8.821487426757812e-06,
      "step": 83250,
      "training_step_time": 9.168286800384521
    },
    {
      "epoch": 8.822441101074218e-06,
      "model_forward_time": 1.7306113243103027,
      "step": 83251
    },
    {
      "epoch": 8.822441101074218e-06,
      "step": 83251,
      "training_step_time": 7.731792211532593
    },
    {
      "epoch": 8.823394775390626e-06,
      "model_forward_time": 1.731257677078247,
      "step": 83252
    },
    {
      "epoch": 8.823394775390626e-06,
      "step": 83252,
      "training_step_time": 8.828224420547485
    },
    {
      "epoch": 8.824348449707031e-06,
      "model_forward_time": 1.731316328048706,
      "step": 83253
    },
    {
      "epoch": 8.824348449707031e-06,
      "step": 83253,
      "training_step_time": 7.261420249938965
    },
    {
      "epoch": 8.825302124023437e-06,
      "model_forward_time": 2.3894684314727783,
      "step": 83254
    },
    {
      "epoch": 8.825302124023437e-06,
      "step": 83254,
      "training_step_time": 7.590147972106934
    },
    {
      "epoch": 8.826255798339844e-06,
      "model_forward_time": 1.7384941577911377,
      "step": 83255
    },
    {
      "epoch": 8.826255798339844e-06,
      "step": 83255,
      "training_step_time": 7.665158748626709
    },
    {
      "epoch": 8.82720947265625e-06,
      "model_forward_time": 1.7182250022888184,
      "step": 83256
    },
    {
      "epoch": 8.82720947265625e-06,
      "step": 83256,
      "training_step_time": 7.6125030517578125
    },
    {
      "epoch": 8.828163146972656e-06,
      "model_forward_time": 1.749894380569458,
      "step": 83257
    },
    {
      "epoch": 8.828163146972656e-06,
      "step": 83257,
      "training_step_time": 8.627248048782349
    },
    {
      "epoch": 8.829116821289062e-06,
      "model_forward_time": 1.7271928787231445,
      "step": 83258
    },
    {
      "epoch": 8.829116821289062e-06,
      "step": 83258,
      "training_step_time": 8.716250896453857
    },
    {
      "epoch": 8.830070495605469e-06,
      "model_forward_time": 1.7482240200042725,
      "step": 83259
    },
    {
      "epoch": 8.830070495605469e-06,
      "step": 83259,
      "training_step_time": 8.792465925216675
    },
    {
      "epoch": 8.831024169921875e-06,
      "grad_norm": 0.11464453488588333,
      "learning_rate": 7.468521737440371e-07,
      "loss": 0.0536,
      "step": 83260
    },
    {
      "epoch": 8.831024169921875e-06,
      "model_forward_time": 1.725895643234253,
      "step": 83260
    },
    {
      "dynamics_loss_avg": 0.05374214667826891,
      "epoch": 8.831024169921875e-06,
      "step": 83260
    },
    {
      "action_loss_avg": 0.001181895993067883,
      "epoch": 8.831024169921875e-06,
      "step": 83260
    },
    {
      "epoch": 8.831024169921875e-06,
      "step": 83260,
      "training_step_time": 9.41073989868164
    },
    {
      "epoch": 8.83197784423828e-06,
      "model_forward_time": 1.7339727878570557,
      "step": 83261
    },
    {
      "epoch": 8.83197784423828e-06,
      "step": 83261,
      "training_step_time": 7.59909725189209
    },
    {
      "epoch": 8.832931518554688e-06,
      "model_forward_time": 1.7329959869384766,
      "step": 83262
    },
    {
      "epoch": 8.832931518554688e-06,
      "step": 83262,
      "training_step_time": 8.353304147720337
    },
    {
      "epoch": 8.833885192871094e-06,
      "model_forward_time": 2.412614345550537,
      "step": 83263
    },
    {
      "epoch": 8.833885192871094e-06,
      "step": 83263,
      "training_step_time": 7.625337600708008
    },
    {
      "epoch": 8.8348388671875e-06,
      "model_forward_time": 1.7298583984375,
      "step": 83264
    },
    {
      "epoch": 8.8348388671875e-06,
      "step": 83264,
      "training_step_time": 7.684220314025879
    },
    {
      "epoch": 8.835792541503907e-06,
      "model_forward_time": 1.7326245307922363,
      "step": 83265
    },
    {
      "epoch": 8.835792541503907e-06,
      "step": 83265,
      "training_step_time": 7.5787353515625
    },
    {
      "epoch": 8.836746215820313e-06,
      "model_forward_time": 1.741107702255249,
      "step": 83266
    },
    {
      "epoch": 8.836746215820313e-06,
      "step": 83266,
      "training_step_time": 7.494526386260986
    },
    {
      "epoch": 8.837699890136718e-06,
      "model_forward_time": 1.734931468963623,
      "step": 83267
    },
    {
      "epoch": 8.837699890136718e-06,
      "step": 83267,
      "training_step_time": 7.653048515319824
    },
    {
      "epoch": 8.838653564453126e-06,
      "model_forward_time": 1.7480359077453613,
      "step": 83268
    },
    {
      "epoch": 8.838653564453126e-06,
      "step": 83268,
      "training_step_time": 8.191724061965942
    },
    {
      "epoch": 8.839607238769531e-06,
      "model_forward_time": 1.7307891845703125,
      "step": 83269
    },
    {
      "epoch": 8.839607238769531e-06,
      "step": 83269,
      "training_step_time": 7.628942012786865
    },
    {
      "epoch": 8.840560913085937e-06,
      "grad_norm": 0.10870276391506195,
      "learning_rate": 7.4598306896211e-07,
      "loss": 0.0541,
      "step": 83270
    },
    {
      "epoch": 8.840560913085937e-06,
      "model_forward_time": 1.730043888092041,
      "step": 83270
    },
    {
      "dynamics_loss_avg": 0.056334435008466244,
      "epoch": 8.840560913085937e-06,
      "step": 83270
    },
    {
      "action_loss_avg": 0.0010619565029628575,
      "epoch": 8.840560913085937e-06,
      "step": 83270
    },
    {
      "epoch": 8.840560913085937e-06,
      "step": 83270,
      "training_step_time": 7.651552438735962
    },
    {
      "epoch": 8.841514587402345e-06,
      "model_forward_time": 1.732295036315918,
      "step": 83271
    },
    {
      "epoch": 8.841514587402345e-06,
      "step": 83271,
      "training_step_time": 7.336728811264038
    },
    {
      "epoch": 8.84246826171875e-06,
      "model_forward_time": 2.4065587520599365,
      "step": 83272
    },
    {
      "epoch": 8.84246826171875e-06,
      "step": 83272,
      "training_step_time": 7.608747243881226
    },
    {
      "epoch": 8.843421936035156e-06,
      "model_forward_time": 1.7364225387573242,
      "step": 83273
    },
    {
      "epoch": 8.843421936035156e-06,
      "step": 83273,
      "training_step_time": 7.671633005142212
    },
    {
      "epoch": 8.844375610351562e-06,
      "model_forward_time": 1.729201316833496,
      "step": 83274
    },
    {
      "epoch": 8.844375610351562e-06,
      "step": 83274,
      "training_step_time": 8.824697971343994
    },
    {
      "epoch": 8.845329284667969e-06,
      "model_forward_time": 1.7408106327056885,
      "step": 83275
    },
    {
      "epoch": 8.845329284667969e-06,
      "step": 83275,
      "training_step_time": 7.491088390350342
    },
    {
      "epoch": 8.846282958984375e-06,
      "model_forward_time": 1.7315287590026855,
      "step": 83276
    },
    {
      "epoch": 8.846282958984375e-06,
      "step": 83276,
      "training_step_time": 7.638758897781372
    },
    {
      "epoch": 8.84723663330078e-06,
      "model_forward_time": 1.741575002670288,
      "step": 83277
    },
    {
      "epoch": 8.84723663330078e-06,
      "step": 83277,
      "training_step_time": 7.687443017959595
    },
    {
      "epoch": 8.848190307617188e-06,
      "model_forward_time": 1.7324185371398926,
      "step": 83278
    },
    {
      "epoch": 8.848190307617188e-06,
      "step": 83278,
      "training_step_time": 7.76525616645813
    },
    {
      "epoch": 8.849143981933594e-06,
      "model_forward_time": 1.732158899307251,
      "step": 83279
    },
    {
      "epoch": 8.849143981933594e-06,
      "step": 83279,
      "training_step_time": 9.406801700592041
    },
    {
      "epoch": 8.85009765625e-06,
      "grad_norm": 0.08179837465286255,
      "learning_rate": 7.451144293930773e-07,
      "loss": 0.0547,
      "step": 83280
    },
    {
      "epoch": 8.85009765625e-06,
      "model_forward_time": 1.7140941619873047,
      "step": 83280
    },
    {
      "dynamics_loss_avg": 0.05518664848059416,
      "epoch": 8.85009765625e-06,
      "step": 83280
    },
    {
      "action_loss_avg": 0.0013570290509960614,
      "epoch": 8.85009765625e-06,
      "step": 83280
    },
    {
      "epoch": 8.85009765625e-06,
      "step": 83280,
      "training_step_time": 7.56951904296875
    },
    {
      "epoch": 8.851051330566407e-06,
      "model_forward_time": 1.7339808940887451,
      "step": 83281
    },
    {
      "epoch": 8.851051330566407e-06,
      "step": 83281,
      "training_step_time": 8.186416625976562
    },
    {
      "epoch": 8.852005004882813e-06,
      "model_forward_time": 1.7329461574554443,
      "step": 83282
    },
    {
      "epoch": 8.852005004882813e-06,
      "step": 83282,
      "training_step_time": 8.243588209152222
    },
    {
      "epoch": 8.852958679199218e-06,
      "model_forward_time": 1.7196738719940186,
      "step": 83283
    },
    {
      "epoch": 8.852958679199218e-06,
      "step": 83283,
      "training_step_time": 7.695418119430542
    },
    {
      "epoch": 8.853912353515626e-06,
      "model_forward_time": 1.7358298301696777,
      "step": 83284
    },
    {
      "epoch": 8.853912353515626e-06,
      "step": 83284,
      "training_step_time": 9.440442562103271
    },
    {
      "epoch": 8.854866027832031e-06,
      "model_forward_time": 1.7382087707519531,
      "step": 83285
    },
    {
      "epoch": 8.854866027832031e-06,
      "step": 83285,
      "training_step_time": 7.636855602264404
    },
    {
      "epoch": 8.855819702148437e-06,
      "model_forward_time": 1.7473454475402832,
      "step": 83286
    },
    {
      "epoch": 8.855819702148437e-06,
      "step": 83286,
      "training_step_time": 8.139061212539673
    },
    {
      "epoch": 8.856773376464845e-06,
      "model_forward_time": 1.7337946891784668,
      "step": 83287
    },
    {
      "epoch": 8.856773376464845e-06,
      "step": 83287,
      "training_step_time": 9.489428520202637
    },
    {
      "epoch": 8.85772705078125e-06,
      "model_forward_time": 1.7316243648529053,
      "step": 83288
    },
    {
      "epoch": 8.85772705078125e-06,
      "step": 83288,
      "training_step_time": 8.451392412185669
    },
    {
      "epoch": 8.858680725097656e-06,
      "model_forward_time": 1.7314808368682861,
      "step": 83289
    },
    {
      "epoch": 8.858680725097656e-06,
      "step": 83289,
      "training_step_time": 7.62877893447876
    },
    {
      "epoch": 8.859634399414062e-06,
      "grad_norm": 0.08696132153272629,
      "learning_rate": 7.442462551319346e-07,
      "loss": 0.0552,
      "step": 83290
    },
    {
      "epoch": 8.859634399414062e-06,
      "model_forward_time": 1.7220535278320312,
      "step": 83290
    },
    {
      "dynamics_loss_avg": 0.05220443233847618,
      "epoch": 8.859634399414062e-06,
      "step": 83290
    },
    {
      "action_loss_avg": 0.002088447884307243,
      "epoch": 8.859634399414062e-06,
      "step": 83290
    },
    {
      "epoch": 8.859634399414062e-06,
      "step": 83290,
      "training_step_time": 8.120144367218018
    },
    {
      "epoch": 8.860588073730469e-06,
      "model_forward_time": 2.3859307765960693,
      "step": 83291
    },
    {
      "epoch": 8.860588073730469e-06,
      "step": 83291,
      "training_step_time": 7.6632726192474365
    },
    {
      "epoch": 8.861541748046875e-06,
      "model_forward_time": 1.7271764278411865,
      "step": 83292
    },
    {
      "epoch": 8.861541748046875e-06,
      "step": 83292,
      "training_step_time": 7.84029746055603
    },
    {
      "epoch": 8.86249542236328e-06,
      "model_forward_time": 1.7361023426055908,
      "step": 83293
    },
    {
      "epoch": 8.86249542236328e-06,
      "step": 83293,
      "training_step_time": 7.6704325675964355
    },
    {
      "epoch": 8.863449096679688e-06,
      "model_forward_time": 1.7578349113464355,
      "step": 83294
    },
    {
      "epoch": 8.863449096679688e-06,
      "step": 83294,
      "training_step_time": 8.180241823196411
    },
    {
      "epoch": 8.864402770996094e-06,
      "model_forward_time": 1.7393310070037842,
      "step": 83295
    },
    {
      "epoch": 8.864402770996094e-06,
      "step": 83295,
      "training_step_time": 8.322072505950928
    },
    {
      "epoch": 8.8653564453125e-06,
      "model_forward_time": 1.7283077239990234,
      "step": 83296
    },
    {
      "epoch": 8.8653564453125e-06,
      "step": 83296,
      "training_step_time": 7.953373908996582
    },
    {
      "epoch": 8.866310119628907e-06,
      "model_forward_time": 1.7307534217834473,
      "step": 83297
    },
    {
      "epoch": 8.866310119628907e-06,
      "step": 83297,
      "training_step_time": 10.048652648925781
    },
    {
      "epoch": 8.867263793945313e-06,
      "model_forward_time": 1.7281849384307861,
      "step": 83298
    },
    {
      "epoch": 8.867263793945313e-06,
      "step": 83298,
      "training_step_time": 8.439396858215332
    },
    {
      "epoch": 8.868217468261718e-06,
      "model_forward_time": 1.7288720607757568,
      "step": 83299
    },
    {
      "epoch": 8.868217468261718e-06,
      "step": 83299,
      "training_step_time": 7.521446943283081
    },
    {
      "epoch": 8.869171142578126e-06,
      "grad_norm": 0.21790958940982819,
      "learning_rate": 7.433785462736209e-07,
      "loss": 0.0552,
      "step": 83300
    },
    {
      "epoch": 8.869171142578126e-06,
      "model_forward_time": 2.383904457092285,
      "step": 83300
    },
    {
      "dynamics_loss_avg": 0.048577731102705,
      "epoch": 8.869171142578126e-06,
      "step": 83300
    },
    {
      "action_loss_avg": 0.0015569216455332934,
      "epoch": 8.869171142578126e-06,
      "step": 83300
    },
    {
      "epoch": 8.869171142578126e-06,
      "step": 83300,
      "training_step_time": 7.5565245151519775
    },
    {
      "epoch": 8.870124816894531e-06,
      "model_forward_time": 1.7265615463256836,
      "step": 83301
    },
    {
      "epoch": 8.870124816894531e-06,
      "step": 83301,
      "training_step_time": 7.808144807815552
    },
    {
      "epoch": 8.871078491210937e-06,
      "model_forward_time": 1.7263224124908447,
      "step": 83302
    },
    {
      "epoch": 8.871078491210937e-06,
      "step": 83302,
      "training_step_time": 7.614005088806152
    },
    {
      "epoch": 8.872032165527345e-06,
      "model_forward_time": 1.745016098022461,
      "step": 83303
    },
    {
      "epoch": 8.872032165527345e-06,
      "step": 83303,
      "training_step_time": 12.006050825119019
    },
    {
      "epoch": 8.87298583984375e-06,
      "model_forward_time": 1.7337121963500977,
      "step": 83304
    },
    {
      "epoch": 8.87298583984375e-06,
      "step": 83304,
      "training_step_time": 6.823566436767578
    },
    {
      "epoch": 8.873939514160156e-06,
      "model_forward_time": 1.728252649307251,
      "step": 83305
    },
    {
      "epoch": 8.873939514160156e-06,
      "step": 83305,
      "training_step_time": 7.738757848739624
    },
    {
      "epoch": 8.874893188476562e-06,
      "model_forward_time": 1.734778642654419,
      "step": 83306
    },
    {
      "epoch": 8.874893188476562e-06,
      "step": 83306,
      "training_step_time": 8.17608118057251
    },
    {
      "epoch": 8.875846862792969e-06,
      "model_forward_time": 1.7278826236724854,
      "step": 83307
    },
    {
      "epoch": 8.875846862792969e-06,
      "step": 83307,
      "training_step_time": 7.809038400650024
    },
    {
      "epoch": 8.876800537109375e-06,
      "model_forward_time": 1.729081392288208,
      "step": 83308
    },
    {
      "epoch": 8.876800537109375e-06,
      "step": 83308,
      "training_step_time": 6.7887654304504395
    },
    {
      "epoch": 8.87775421142578e-06,
      "model_forward_time": 1.7303950786590576,
      "step": 83309
    },
    {
      "epoch": 8.87775421142578e-06,
      "step": 83309,
      "training_step_time": 6.742810010910034
    },
    {
      "epoch": 8.878707885742188e-06,
      "grad_norm": 0.08700846880674362,
      "learning_rate": 7.425113029130315e-07,
      "loss": 0.0553,
      "step": 83310
    },
    {
      "epoch": 8.878707885742188e-06,
      "model_forward_time": 1.7319488525390625,
      "step": 83310
    },
    {
      "dynamics_loss_avg": 0.04871355313807726,
      "epoch": 8.878707885742188e-06,
      "step": 83310
    },
    {
      "action_loss_avg": 0.0012551373452879488,
      "epoch": 8.878707885742188e-06,
      "step": 83310
    },
    {
      "epoch": 8.878707885742188e-06,
      "step": 83310,
      "training_step_time": 6.7631919384002686
    },
    {
      "epoch": 8.879661560058594e-06,
      "model_forward_time": 1.7262952327728271,
      "step": 83311
    },
    {
      "epoch": 8.879661560058594e-06,
      "step": 83311,
      "training_step_time": 6.717248916625977
    },
    {
      "epoch": 8.880615234375e-06,
      "model_forward_time": 1.7277486324310303,
      "step": 83312
    },
    {
      "epoch": 8.880615234375e-06,
      "step": 83312,
      "training_step_time": 6.784363031387329
    },
    {
      "epoch": 8.881568908691407e-06,
      "model_forward_time": 2.385653495788574,
      "step": 83313
    },
    {
      "epoch": 8.881568908691407e-06,
      "step": 83313,
      "training_step_time": 7.704546689987183
    },
    {
      "epoch": 8.882522583007813e-06,
      "model_forward_time": 1.7250094413757324,
      "step": 83314
    },
    {
      "epoch": 8.882522583007813e-06,
      "step": 83314,
      "training_step_time": 6.8003058433532715
    },
    {
      "epoch": 8.883476257324218e-06,
      "model_forward_time": 1.7337088584899902,
      "step": 83315
    },
    {
      "epoch": 8.883476257324218e-06,
      "step": 83315,
      "training_step_time": 7.558125257492065
    },
    {
      "epoch": 8.884429931640626e-06,
      "model_forward_time": 1.3409638404846191,
      "step": 83316
    },
    {
      "epoch": 8.884429931640626e-06,
      "step": 83316,
      "training_step_time": 7.2399537563323975
    },
    {
      "epoch": 8.885383605957031e-06,
      "model_forward_time": 1.7303006649017334,
      "step": 83317
    },
    {
      "epoch": 8.885383605957031e-06,
      "step": 83317,
      "training_step_time": 7.8070595264434814
    },
    {
      "epoch": 8.886337280273437e-06,
      "model_forward_time": 1.733572006225586,
      "step": 83318
    },
    {
      "epoch": 8.886337280273437e-06,
      "step": 83318,
      "training_step_time": 6.798169851303101
    },
    {
      "epoch": 8.887290954589845e-06,
      "model_forward_time": 1.7309074401855469,
      "step": 83319
    },
    {
      "epoch": 8.887290954589845e-06,
      "step": 83319,
      "training_step_time": 6.736618518829346
    },
    {
      "epoch": 8.88824462890625e-06,
      "grad_norm": 0.1263345181941986,
      "learning_rate": 7.416445251450044e-07,
      "loss": 0.0561,
      "step": 83320
    },
    {
      "epoch": 8.88824462890625e-06,
      "model_forward_time": 1.7266943454742432,
      "step": 83320
    },
    {
      "dynamics_loss_avg": 0.06974061289802194,
      "epoch": 8.88824462890625e-06,
      "step": 83320
    },
    {
      "action_loss_avg": 0.001277888420736417,
      "epoch": 8.88824462890625e-06,
      "step": 83320
    },
    {
      "epoch": 8.88824462890625e-06,
      "step": 83320,
      "training_step_time": 6.808157444000244
    },
    {
      "epoch": 8.889198303222656e-06,
      "model_forward_time": 1.7317817211151123,
      "step": 83321
    },
    {
      "epoch": 8.889198303222656e-06,
      "step": 83321,
      "training_step_time": 6.807393550872803
    },
    {
      "epoch": 8.890151977539062e-06,
      "model_forward_time": 2.3968405723571777,
      "step": 83322
    },
    {
      "epoch": 8.890151977539062e-06,
      "step": 83322,
      "training_step_time": 8.817150592803955
    },
    {
      "epoch": 8.891105651855469e-06,
      "model_forward_time": 1.7183029651641846,
      "step": 83323
    },
    {
      "epoch": 8.891105651855469e-06,
      "step": 83323,
      "training_step_time": 6.778764486312866
    },
    {
      "epoch": 8.892059326171875e-06,
      "model_forward_time": 1.7340610027313232,
      "step": 83324
    },
    {
      "epoch": 8.892059326171875e-06,
      "step": 83324,
      "training_step_time": 7.69569206237793
    },
    {
      "epoch": 8.89301300048828e-06,
      "model_forward_time": 1.7347774505615234,
      "step": 83325
    },
    {
      "epoch": 8.89301300048828e-06,
      "step": 83325,
      "training_step_time": 7.352635145187378
    },
    {
      "epoch": 8.893966674804688e-06,
      "model_forward_time": 1.7373180389404297,
      "step": 83326
    },
    {
      "epoch": 8.893966674804688e-06,
      "step": 83326,
      "training_step_time": 7.795943737030029
    },
    {
      "epoch": 8.894920349121094e-06,
      "model_forward_time": 1.7340147495269775,
      "step": 83327
    },
    {
      "epoch": 8.894920349121094e-06,
      "step": 83327,
      "training_step_time": 6.821695327758789
    },
    {
      "epoch": 8.8958740234375e-06,
      "model_forward_time": 1.7328245639801025,
      "step": 83328
    },
    {
      "epoch": 8.8958740234375e-06,
      "step": 83328,
      "training_step_time": 6.8738625049591064
    },
    {
      "epoch": 8.896827697753907e-06,
      "model_forward_time": 1.7318203449249268,
      "step": 83329
    },
    {
      "epoch": 8.896827697753907e-06,
      "step": 83329,
      "training_step_time": 6.898357391357422
    },
    {
      "epoch": 8.897781372070313e-06,
      "grad_norm": 0.12599225342273712,
      "learning_rate": 7.407782130643304e-07,
      "loss": 0.0559,
      "step": 83330
    },
    {
      "epoch": 8.897781372070313e-06,
      "model_forward_time": 1.7336757183074951,
      "step": 83330
    },
    {
      "dynamics_loss_avg": 0.05013997573405504,
      "epoch": 8.897781372070313e-06,
      "step": 83330
    },
    {
      "action_loss_avg": 0.001080763549543917,
      "epoch": 8.897781372070313e-06,
      "step": 83330
    },
    {
      "epoch": 8.897781372070313e-06,
      "step": 83330,
      "training_step_time": 6.809496164321899
    },
    {
      "epoch": 8.898735046386718e-06,
      "model_forward_time": 2.431818962097168,
      "step": 83331
    },
    {
      "epoch": 8.898735046386718e-06,
      "step": 83331,
      "training_step_time": 10.209185600280762
    },
    {
      "epoch": 8.899688720703126e-06,
      "model_forward_time": 1.7238633632659912,
      "step": 83332
    },
    {
      "epoch": 8.899688720703126e-06,
      "step": 83332,
      "training_step_time": 7.673383474349976
    },
    {
      "epoch": 8.900642395019531e-06,
      "model_forward_time": 1.7406229972839355,
      "step": 83333
    },
    {
      "epoch": 8.900642395019531e-06,
      "step": 83333,
      "training_step_time": 7.7260496616363525
    },
    {
      "epoch": 8.901596069335937e-06,
      "model_forward_time": 1.7314624786376953,
      "step": 83334
    },
    {
      "epoch": 8.901596069335937e-06,
      "step": 83334,
      "training_step_time": 7.33855128288269
    },
    {
      "epoch": 8.902549743652345e-06,
      "model_forward_time": 1.733750820159912,
      "step": 83335
    },
    {
      "epoch": 8.902549743652345e-06,
      "step": 83335,
      "training_step_time": 7.76919960975647
    },
    {
      "epoch": 8.90350341796875e-06,
      "model_forward_time": 1.7360866069793701,
      "step": 83336
    },
    {
      "epoch": 8.90350341796875e-06,
      "step": 83336,
      "training_step_time": 6.870037078857422
    },
    {
      "epoch": 8.904457092285156e-06,
      "model_forward_time": 1.7342429161071777,
      "step": 83337
    },
    {
      "epoch": 8.904457092285156e-06,
      "step": 83337,
      "training_step_time": 6.887649774551392
    },
    {
      "epoch": 8.905410766601562e-06,
      "model_forward_time": 1.7296292781829834,
      "step": 83338
    },
    {
      "epoch": 8.905410766601562e-06,
      "step": 83338,
      "training_step_time": 6.853213310241699
    },
    {
      "epoch": 8.906364440917969e-06,
      "model_forward_time": 1.7313041687011719,
      "step": 83339
    },
    {
      "epoch": 8.906364440917969e-06,
      "step": 83339,
      "training_step_time": 6.8402488231658936
    },
    {
      "epoch": 8.907318115234375e-06,
      "grad_norm": 0.1211136057972908,
      "learning_rate": 7.399123667657465e-07,
      "loss": 0.0578,
      "step": 83340
    },
    {
      "epoch": 8.907318115234375e-06,
      "model_forward_time": 2.419792652130127,
      "step": 83340
    },
    {
      "dynamics_loss_avg": 0.051347828656435016,
      "epoch": 8.907318115234375e-06,
      "step": 83340
    },
    {
      "action_loss_avg": 0.0021991423127474263,
      "epoch": 8.907318115234375e-06,
      "step": 83340
    },
    {
      "epoch": 8.907318115234375e-06,
      "step": 83340,
      "training_step_time": 11.571174621582031
    },
    {
      "epoch": 8.90827178955078e-06,
      "model_forward_time": 1.7315614223480225,
      "step": 83341
    },
    {
      "epoch": 8.90827178955078e-06,
      "step": 83341,
      "training_step_time": 7.752784490585327
    },
    {
      "epoch": 8.909225463867188e-06,
      "model_forward_time": 1.7271192073822021,
      "step": 83342
    },
    {
      "epoch": 8.909225463867188e-06,
      "step": 83342,
      "training_step_time": 8.351027011871338
    },
    {
      "epoch": 8.910179138183594e-06,
      "model_forward_time": 1.73518705368042,
      "step": 83343
    },
    {
      "epoch": 8.910179138183594e-06,
      "step": 83343,
      "training_step_time": 7.684109210968018
    },
    {
      "epoch": 8.9111328125e-06,
      "model_forward_time": 1.7269268035888672,
      "step": 83344
    },
    {
      "epoch": 8.9111328125e-06,
      "step": 83344,
      "training_step_time": 7.87623143196106
    },
    {
      "epoch": 8.912086486816407e-06,
      "model_forward_time": 1.731977939605713,
      "step": 83345
    },
    {
      "epoch": 8.912086486816407e-06,
      "step": 83345,
      "training_step_time": 6.8469438552856445
    },
    {
      "epoch": 8.913040161132813e-06,
      "model_forward_time": 1.7302284240722656,
      "step": 83346
    },
    {
      "epoch": 8.913040161132813e-06,
      "step": 83346,
      "training_step_time": 6.871655702590942
    },
    {
      "epoch": 8.913993835449218e-06,
      "model_forward_time": 1.7296180725097656,
      "step": 83347
    },
    {
      "epoch": 8.913993835449218e-06,
      "step": 83347,
      "training_step_time": 6.8629302978515625
    },
    {
      "epoch": 8.914947509765626e-06,
      "model_forward_time": 1.7319097518920898,
      "step": 83348
    },
    {
      "epoch": 8.914947509765626e-06,
      "step": 83348,
      "training_step_time": 6.819397211074829
    },
    {
      "epoch": 8.915901184082031e-06,
      "model_forward_time": 1.7284128665924072,
      "step": 83349
    },
    {
      "epoch": 8.915901184082031e-06,
      "step": 83349,
      "training_step_time": 10.084109544754028
    },
    {
      "epoch": 8.916854858398437e-06,
      "grad_norm": 0.0713077113032341,
      "learning_rate": 7.390469863439426e-07,
      "loss": 0.0564,
      "step": 83350
    },
    {
      "epoch": 8.916854858398437e-06,
      "model_forward_time": 1.7405328750610352,
      "step": 83350
    },
    {
      "dynamics_loss_avg": 0.055136737693101165,
      "epoch": 8.916854858398437e-06,
      "step": 83350
    },
    {
      "action_loss_avg": 0.001580509770428762,
      "epoch": 8.916854858398437e-06,
      "step": 83350
    },
    {
      "epoch": 8.916854858398437e-06,
      "step": 83350,
      "training_step_time": 7.729119300842285
    },
    {
      "epoch": 8.917808532714845e-06,
      "model_forward_time": 1.3491919040679932,
      "step": 83351
    },
    {
      "epoch": 8.917808532714845e-06,
      "step": 83351,
      "training_step_time": 7.908761978149414
    },
    {
      "epoch": 8.91876220703125e-06,
      "model_forward_time": 1.3470354080200195,
      "step": 83352
    },
    {
      "epoch": 8.91876220703125e-06,
      "step": 83352,
      "training_step_time": 7.6536877155303955
    },
    {
      "epoch": 8.919715881347656e-06,
      "model_forward_time": 1.7283990383148193,
      "step": 83353
    },
    {
      "epoch": 8.919715881347656e-06,
      "step": 83353,
      "training_step_time": 7.82783317565918
    },
    {
      "epoch": 8.920669555664062e-06,
      "model_forward_time": 1.731571912765503,
      "step": 83354
    },
    {
      "epoch": 8.920669555664062e-06,
      "step": 83354,
      "training_step_time": 7.258930206298828
    },
    {
      "epoch": 8.92162322998047e-06,
      "model_forward_time": 1.7416667938232422,
      "step": 83355
    },
    {
      "epoch": 8.92162322998047e-06,
      "step": 83355,
      "training_step_time": 6.855125904083252
    },
    {
      "epoch": 8.922576904296875e-06,
      "model_forward_time": 1.7257704734802246,
      "step": 83356
    },
    {
      "epoch": 8.922576904296875e-06,
      "step": 83356,
      "training_step_time": 6.814977169036865
    },
    {
      "epoch": 8.92353057861328e-06,
      "model_forward_time": 1.7269518375396729,
      "step": 83357
    },
    {
      "epoch": 8.92353057861328e-06,
      "step": 83357,
      "training_step_time": 6.935558319091797
    },
    {
      "epoch": 8.924484252929688e-06,
      "model_forward_time": 1.7359228134155273,
      "step": 83358
    },
    {
      "epoch": 8.924484252929688e-06,
      "step": 83358,
      "training_step_time": 8.704457521438599
    },
    {
      "epoch": 8.925437927246094e-06,
      "model_forward_time": 2.3783044815063477,
      "step": 83359
    },
    {
      "epoch": 8.925437927246094e-06,
      "step": 83359,
      "training_step_time": 8.780408143997192
    },
    {
      "epoch": 8.9263916015625e-06,
      "grad_norm": 0.08738099038600922,
      "learning_rate": 7.381820718935512e-07,
      "loss": 0.058,
      "step": 83360
    },
    {
      "epoch": 8.9263916015625e-06,
      "model_forward_time": 1.7340490818023682,
      "step": 83360
    },
    {
      "dynamics_loss_avg": 0.05492700524628162,
      "epoch": 8.9263916015625e-06,
      "step": 83360
    },
    {
      "action_loss_avg": 0.0009929364547133445,
      "epoch": 8.9263916015625e-06,
      "step": 83360
    },
    {
      "epoch": 8.9263916015625e-06,
      "step": 83360,
      "training_step_time": 7.488321304321289
    },
    {
      "epoch": 8.927345275878907e-06,
      "model_forward_time": 1.7337942123413086,
      "step": 83361
    },
    {
      "epoch": 8.927345275878907e-06,
      "step": 83361,
      "training_step_time": 7.632059812545776
    },
    {
      "epoch": 8.928298950195313e-06,
      "model_forward_time": 1.7286124229431152,
      "step": 83362
    },
    {
      "epoch": 8.928298950195313e-06,
      "step": 83362,
      "training_step_time": 7.333893060684204
    },
    {
      "epoch": 8.929252624511718e-06,
      "model_forward_time": 1.3456807136535645,
      "step": 83363
    },
    {
      "epoch": 8.929252624511718e-06,
      "step": 83363,
      "training_step_time": 7.830286741256714
    },
    {
      "epoch": 8.930206298828126e-06,
      "model_forward_time": 1.7349426746368408,
      "step": 83364
    },
    {
      "epoch": 8.930206298828126e-06,
      "step": 83364,
      "training_step_time": 6.72431755065918
    },
    {
      "epoch": 8.931159973144531e-06,
      "model_forward_time": 1.733203649520874,
      "step": 83365
    },
    {
      "epoch": 8.931159973144531e-06,
      "step": 83365,
      "training_step_time": 6.760875463485718
    },
    {
      "epoch": 8.932113647460937e-06,
      "model_forward_time": 1.731034517288208,
      "step": 83366
    },
    {
      "epoch": 8.932113647460937e-06,
      "step": 83366,
      "training_step_time": 6.92496919631958
    },
    {
      "epoch": 8.933067321777345e-06,
      "model_forward_time": 1.7145130634307861,
      "step": 83367
    },
    {
      "epoch": 8.933067321777345e-06,
      "step": 83367,
      "training_step_time": 9.43018627166748
    },
    {
      "epoch": 8.93402099609375e-06,
      "model_forward_time": 2.398712635040283,
      "step": 83368
    },
    {
      "epoch": 8.93402099609375e-06,
      "step": 83368,
      "training_step_time": 7.748723268508911
    },
    {
      "epoch": 8.934974670410156e-06,
      "model_forward_time": 1.7355608940124512,
      "step": 83369
    },
    {
      "epoch": 8.934974670410156e-06,
      "step": 83369,
      "training_step_time": 7.658157587051392
    },
    {
      "epoch": 8.935928344726562e-06,
      "grad_norm": 0.09417661279439926,
      "learning_rate": 7.373176235091634e-07,
      "loss": 0.0594,
      "step": 83370
    },
    {
      "epoch": 8.935928344726562e-06,
      "model_forward_time": 1.7317473888397217,
      "step": 83370
    },
    {
      "dynamics_loss_avg": 0.07423747889697552,
      "epoch": 8.935928344726562e-06,
      "step": 83370
    },
    {
      "action_loss_avg": 0.002154878090368584,
      "epoch": 8.935928344726562e-06,
      "step": 83370
    },
    {
      "epoch": 8.935928344726562e-06,
      "step": 83370,
      "training_step_time": 7.489861965179443
    },
    {
      "epoch": 8.93688201904297e-06,
      "model_forward_time": 1.7314844131469727,
      "step": 83371
    },
    {
      "epoch": 8.93688201904297e-06,
      "step": 83371,
      "training_step_time": 7.600292444229126
    },
    {
      "epoch": 8.937835693359375e-06,
      "model_forward_time": 1.732006311416626,
      "step": 83372
    },
    {
      "epoch": 8.937835693359375e-06,
      "step": 83372,
      "training_step_time": 8.637644052505493
    },
    {
      "epoch": 8.93878936767578e-06,
      "model_forward_time": 1.7114653587341309,
      "step": 83373
    },
    {
      "epoch": 8.93878936767578e-06,
      "step": 83373,
      "training_step_time": 6.825839281082153
    },
    {
      "epoch": 8.939743041992188e-06,
      "model_forward_time": 1.7224609851837158,
      "step": 83374
    },
    {
      "epoch": 8.939743041992188e-06,
      "step": 83374,
      "training_step_time": 6.806228876113892
    },
    {
      "epoch": 8.940696716308594e-06,
      "model_forward_time": 1.7364122867584229,
      "step": 83375
    },
    {
      "epoch": 8.940696716308594e-06,
      "step": 83375,
      "training_step_time": 6.86714768409729
    },
    {
      "epoch": 8.941650390625e-06,
      "model_forward_time": 1.7330973148345947,
      "step": 83376
    },
    {
      "epoch": 8.941650390625e-06,
      "step": 83376,
      "training_step_time": 9.212357997894287
    },
    {
      "epoch": 8.942604064941407e-06,
      "model_forward_time": 1.752145767211914,
      "step": 83377
    },
    {
      "epoch": 8.942604064941407e-06,
      "step": 83377,
      "training_step_time": 9.137281894683838
    },
    {
      "epoch": 8.943557739257813e-06,
      "model_forward_time": 1.7375056743621826,
      "step": 83378
    },
    {
      "epoch": 8.943557739257813e-06,
      "step": 83378,
      "training_step_time": 7.669978618621826
    },
    {
      "epoch": 8.944511413574218e-06,
      "model_forward_time": 1.732196569442749,
      "step": 83379
    },
    {
      "epoch": 8.944511413574218e-06,
      "step": 83379,
      "training_step_time": 7.516206979751587
    },
    {
      "epoch": 8.945465087890626e-06,
      "grad_norm": 0.08896432816982269,
      "learning_rate": 7.364536412853102e-07,
      "loss": 0.0578,
      "step": 83380
    },
    {
      "epoch": 8.945465087890626e-06,
      "model_forward_time": 1.7313587665557861,
      "step": 83380
    },
    {
      "dynamics_loss_avg": 0.051357113756239414,
      "epoch": 8.945465087890626e-06,
      "step": 83380
    },
    {
      "action_loss_avg": 0.0010012970422394573,
      "epoch": 8.945465087890626e-06,
      "step": 83380
    },
    {
      "epoch": 8.945465087890626e-06,
      "step": 83380,
      "training_step_time": 7.601419448852539
    },
    {
      "epoch": 8.946418762207031e-06,
      "model_forward_time": 1.3588364124298096,
      "step": 83381
    },
    {
      "epoch": 8.946418762207031e-06,
      "step": 83381,
      "training_step_time": 7.8524158000946045
    },
    {
      "epoch": 8.947372436523437e-06,
      "model_forward_time": 1.357123613357544,
      "step": 83382
    },
    {
      "epoch": 8.947372436523437e-06,
      "step": 83382,
      "training_step_time": 7.466890335083008
    },
    {
      "epoch": 8.948326110839843e-06,
      "model_forward_time": 1.7416434288024902,
      "step": 83383
    },
    {
      "epoch": 8.948326110839843e-06,
      "step": 83383,
      "training_step_time": 6.823275566101074
    },
    {
      "epoch": 8.94927978515625e-06,
      "model_forward_time": 1.7382633686065674,
      "step": 83384
    },
    {
      "epoch": 8.94927978515625e-06,
      "step": 83384,
      "training_step_time": 6.772892951965332
    },
    {
      "epoch": 8.950233459472656e-06,
      "model_forward_time": 1.7418041229248047,
      "step": 83385
    },
    {
      "epoch": 8.950233459472656e-06,
      "step": 83385,
      "training_step_time": 9.716645240783691
    },
    {
      "epoch": 8.951187133789062e-06,
      "model_forward_time": 1.7312610149383545,
      "step": 83386
    },
    {
      "epoch": 8.951187133789062e-06,
      "step": 83386,
      "training_step_time": 8.527135133743286
    },
    {
      "epoch": 8.95214080810547e-06,
      "model_forward_time": 2.4115169048309326,
      "step": 83387
    },
    {
      "epoch": 8.95214080810547e-06,
      "step": 83387,
      "training_step_time": 7.707056045532227
    },
    {
      "epoch": 8.953094482421875e-06,
      "model_forward_time": 1.740185260772705,
      "step": 83388
    },
    {
      "epoch": 8.953094482421875e-06,
      "step": 83388,
      "training_step_time": 7.560123682022095
    },
    {
      "epoch": 8.95404815673828e-06,
      "model_forward_time": 1.7330741882324219,
      "step": 83389
    },
    {
      "epoch": 8.95404815673828e-06,
      "step": 83389,
      "training_step_time": 7.584306716918945
    },
    {
      "epoch": 8.955001831054688e-06,
      "grad_norm": 0.13621054589748383,
      "learning_rate": 7.355901253164776e-07,
      "loss": 0.0582,
      "step": 83390
    },
    {
      "epoch": 8.955001831054688e-06,
      "model_forward_time": 1.7383403778076172,
      "step": 83390
    },
    {
      "dynamics_loss_avg": 0.05576325785368681,
      "epoch": 8.955001831054688e-06,
      "step": 83390
    },
    {
      "action_loss_avg": 0.0015943825506838038,
      "epoch": 8.955001831054688e-06,
      "step": 83390
    },
    {
      "epoch": 8.955001831054688e-06,
      "step": 83390,
      "training_step_time": 7.834527969360352
    },
    {
      "epoch": 8.955955505371094e-06,
      "model_forward_time": 1.7425434589385986,
      "step": 83391
    },
    {
      "epoch": 8.955955505371094e-06,
      "step": 83391,
      "training_step_time": 7.352231025695801
    },
    {
      "epoch": 8.9569091796875e-06,
      "model_forward_time": 1.742607831954956,
      "step": 83392
    },
    {
      "epoch": 8.9569091796875e-06,
      "step": 83392,
      "training_step_time": 6.737345457077026
    },
    {
      "epoch": 8.957862854003907e-06,
      "model_forward_time": 1.7414655685424805,
      "step": 83393
    },
    {
      "epoch": 8.957862854003907e-06,
      "step": 83393,
      "training_step_time": 6.799145698547363
    },
    {
      "epoch": 8.958816528320313e-06,
      "model_forward_time": 1.7414722442626953,
      "step": 83394
    },
    {
      "epoch": 8.958816528320313e-06,
      "step": 83394,
      "training_step_time": 7.617589712142944
    },
    {
      "epoch": 8.959770202636718e-06,
      "model_forward_time": 1.7385036945343018,
      "step": 83395
    },
    {
      "epoch": 8.959770202636718e-06,
      "step": 83395,
      "training_step_time": 9.052135467529297
    },
    {
      "epoch": 8.960723876953126e-06,
      "model_forward_time": 2.423444986343384,
      "step": 83396
    },
    {
      "epoch": 8.960723876953126e-06,
      "step": 83396,
      "training_step_time": 7.686984539031982
    },
    {
      "epoch": 8.961677551269531e-06,
      "model_forward_time": 1.7354981899261475,
      "step": 83397
    },
    {
      "epoch": 8.961677551269531e-06,
      "step": 83397,
      "training_step_time": 8.50680422782898
    },
    {
      "epoch": 8.962631225585937e-06,
      "model_forward_time": 1.7355411052703857,
      "step": 83398
    },
    {
      "epoch": 8.962631225585937e-06,
      "step": 83398,
      "training_step_time": 7.5491368770599365
    },
    {
      "epoch": 8.963584899902343e-06,
      "model_forward_time": 1.7347121238708496,
      "step": 83399
    },
    {
      "epoch": 8.963584899902343e-06,
      "step": 83399,
      "training_step_time": 7.79416561126709
    },
    {
      "epoch": 8.96453857421875e-06,
      "grad_norm": 0.08014660328626633,
      "learning_rate": 7.34727075697097e-07,
      "loss": 0.0583,
      "step": 83400
    },
    {
      "epoch": 8.96453857421875e-06,
      "model_forward_time": 1.7391133308410645,
      "step": 83400
    },
    {
      "dynamics_loss_avg": 0.04954763278365135,
      "epoch": 8.96453857421875e-06,
      "step": 83400
    },
    {
      "action_loss_avg": 0.0014183043880620972,
      "epoch": 8.96453857421875e-06,
      "step": 83400
    },
    {
      "epoch": 8.96453857421875e-06,
      "step": 83400,
      "training_step_time": 7.350841522216797
    },
    {
      "epoch": 8.965492248535156e-06,
      "model_forward_time": 1.735856294631958,
      "step": 83401
    },
    {
      "epoch": 8.965492248535156e-06,
      "step": 83401,
      "training_step_time": 6.811333417892456
    },
    {
      "epoch": 8.966445922851562e-06,
      "model_forward_time": 1.735384225845337,
      "step": 83402
    },
    {
      "epoch": 8.966445922851562e-06,
      "step": 83402,
      "training_step_time": 6.869310140609741
    },
    {
      "epoch": 8.96739959716797e-06,
      "model_forward_time": 1.737994909286499,
      "step": 83403
    },
    {
      "epoch": 8.96739959716797e-06,
      "step": 83403,
      "training_step_time": 7.679539918899536
    },
    {
      "epoch": 8.968353271484375e-06,
      "model_forward_time": 1.7387735843658447,
      "step": 83404
    },
    {
      "epoch": 8.968353271484375e-06,
      "step": 83404,
      "training_step_time": 8.501815557479858
    },
    {
      "epoch": 8.96930694580078e-06,
      "model_forward_time": 2.3917179107666016,
      "step": 83405
    },
    {
      "epoch": 8.96930694580078e-06,
      "step": 83405,
      "training_step_time": 7.8068249225616455
    },
    {
      "epoch": 8.970260620117188e-06,
      "model_forward_time": 1.7419404983520508,
      "step": 83406
    },
    {
      "epoch": 8.970260620117188e-06,
      "step": 83406,
      "training_step_time": 7.687220096588135
    },
    {
      "epoch": 8.971214294433594e-06,
      "model_forward_time": 1.735802412033081,
      "step": 83407
    },
    {
      "epoch": 8.971214294433594e-06,
      "step": 83407,
      "training_step_time": 7.619865894317627
    },
    {
      "epoch": 8.97216796875e-06,
      "model_forward_time": 1.7303407192230225,
      "step": 83408
    },
    {
      "epoch": 8.97216796875e-06,
      "step": 83408,
      "training_step_time": 7.672625303268433
    },
    {
      "epoch": 8.973121643066407e-06,
      "model_forward_time": 1.7379817962646484,
      "step": 83409
    },
    {
      "epoch": 8.973121643066407e-06,
      "step": 83409,
      "training_step_time": 8.70707392692566
    },
    {
      "epoch": 8.974075317382813e-06,
      "grad_norm": 0.0933358445763588,
      "learning_rate": 7.33864492521551e-07,
      "loss": 0.0586,
      "step": 83410
    },
    {
      "epoch": 8.974075317382813e-06,
      "model_forward_time": 1.7352871894836426,
      "step": 83410
    },
    {
      "dynamics_loss_avg": 0.05739982016384602,
      "epoch": 8.974075317382813e-06,
      "step": 83410
    },
    {
      "action_loss_avg": 0.0017114059475716204,
      "epoch": 8.974075317382813e-06,
      "step": 83410
    },
    {
      "epoch": 8.974075317382813e-06,
      "step": 83410,
      "training_step_time": 6.899889230728149
    },
    {
      "epoch": 8.975028991699218e-06,
      "model_forward_time": 1.3580315113067627,
      "step": 83411
    },
    {
      "epoch": 8.975028991699218e-06,
      "step": 83411,
      "training_step_time": 6.8416218757629395
    },
    {
      "epoch": 8.975982666015626e-06,
      "model_forward_time": 1.7369420528411865,
      "step": 83412
    },
    {
      "epoch": 8.975982666015626e-06,
      "step": 83412,
      "training_step_time": 7.780759811401367
    },
    {
      "epoch": 8.976936340332031e-06,
      "model_forward_time": 1.7459564208984375,
      "step": 83413
    },
    {
      "epoch": 8.976936340332031e-06,
      "step": 83413,
      "training_step_time": 7.843631267547607
    },
    {
      "epoch": 8.977890014648437e-06,
      "model_forward_time": 1.757622480392456,
      "step": 83414
    },
    {
      "epoch": 8.977890014648437e-06,
      "step": 83414,
      "training_step_time": 10.45558786392212
    },
    {
      "epoch": 8.978843688964843e-06,
      "model_forward_time": 2.3776261806488037,
      "step": 83415
    },
    {
      "epoch": 8.978843688964843e-06,
      "step": 83415,
      "training_step_time": 7.80777645111084
    },
    {
      "epoch": 8.97979736328125e-06,
      "model_forward_time": 1.7445504665374756,
      "step": 83416
    },
    {
      "epoch": 8.97979736328125e-06,
      "step": 83416,
      "training_step_time": 7.5486907958984375
    },
    {
      "epoch": 8.980751037597656e-06,
      "model_forward_time": 1.7397329807281494,
      "step": 83417
    },
    {
      "epoch": 8.980751037597656e-06,
      "step": 83417,
      "training_step_time": 7.685494661331177
    },
    {
      "epoch": 8.981704711914062e-06,
      "model_forward_time": 1.736997365951538,
      "step": 83418
    },
    {
      "epoch": 8.981704711914062e-06,
      "step": 83418,
      "training_step_time": 7.836670637130737
    },
    {
      "epoch": 8.98265838623047e-06,
      "model_forward_time": 1.7412266731262207,
      "step": 83419
    },
    {
      "epoch": 8.98265838623047e-06,
      "step": 83419,
      "training_step_time": 7.454261779785156
    },
    {
      "epoch": 8.983612060546875e-06,
      "grad_norm": 0.07529034465551376,
      "learning_rate": 7.330023758841693e-07,
      "loss": 0.0588,
      "step": 83420
    },
    {
      "epoch": 8.983612060546875e-06,
      "model_forward_time": 1.7392807006835938,
      "step": 83420
    },
    {
      "dynamics_loss_avg": 0.05115728490054607,
      "epoch": 8.983612060546875e-06,
      "step": 83420
    },
    {
      "action_loss_avg": 0.0010948188370093702,
      "epoch": 8.983612060546875e-06,
      "step": 83420
    },
    {
      "epoch": 8.983612060546875e-06,
      "step": 83420,
      "training_step_time": 6.889298439025879
    },
    {
      "epoch": 8.98456573486328e-06,
      "model_forward_time": 1.7378113269805908,
      "step": 83421
    },
    {
      "epoch": 8.98456573486328e-06,
      "step": 83421,
      "training_step_time": 7.705564498901367
    },
    {
      "epoch": 8.985519409179688e-06,
      "model_forward_time": 1.7394921779632568,
      "step": 83422
    },
    {
      "epoch": 8.985519409179688e-06,
      "step": 83422,
      "training_step_time": 8.249979257583618
    },
    {
      "epoch": 8.986473083496094e-06,
      "model_forward_time": 1.7492680549621582,
      "step": 83423
    },
    {
      "epoch": 8.986473083496094e-06,
      "step": 83423,
      "training_step_time": 10.855732202529907
    },
    {
      "epoch": 8.9874267578125e-06,
      "model_forward_time": 2.401651620864868,
      "step": 83424
    },
    {
      "epoch": 8.9874267578125e-06,
      "step": 83424,
      "training_step_time": 7.7375874519348145
    },
    {
      "epoch": 8.988380432128907e-06,
      "model_forward_time": 1.0258381366729736,
      "step": 83425
    },
    {
      "epoch": 8.988380432128907e-06,
      "step": 83425,
      "training_step_time": 7.646039962768555
    },
    {
      "epoch": 8.989334106445313e-06,
      "model_forward_time": 1.7409687042236328,
      "step": 83426
    },
    {
      "epoch": 8.989334106445313e-06,
      "step": 83426,
      "training_step_time": 7.443520784378052
    },
    {
      "epoch": 8.990287780761718e-06,
      "model_forward_time": 1.7434308528900146,
      "step": 83427
    },
    {
      "epoch": 8.990287780761718e-06,
      "step": 83427,
      "training_step_time": 7.758967876434326
    },
    {
      "epoch": 8.991241455078126e-06,
      "model_forward_time": 1.739605188369751,
      "step": 83428
    },
    {
      "epoch": 8.991241455078126e-06,
      "step": 83428,
      "training_step_time": 7.446118116378784
    },
    {
      "epoch": 8.992195129394532e-06,
      "model_forward_time": 1.7418098449707031,
      "step": 83429
    },
    {
      "epoch": 8.992195129394532e-06,
      "step": 83429,
      "training_step_time": 6.81840181350708
    },
    {
      "epoch": 8.993148803710937e-06,
      "grad_norm": 0.22468779981136322,
      "learning_rate": 7.321407258792323e-07,
      "loss": 0.0581,
      "step": 83430
    },
    {
      "epoch": 8.993148803710937e-06,
      "model_forward_time": 1.738837718963623,
      "step": 83430
    },
    {
      "dynamics_loss_avg": 0.054268221370875835,
      "epoch": 8.993148803710937e-06,
      "step": 83430
    },
    {
      "action_loss_avg": 0.0014845070196315645,
      "epoch": 8.993148803710937e-06,
      "step": 83430
    },
    {
      "epoch": 8.993148803710937e-06,
      "step": 83430,
      "training_step_time": 7.723168611526489
    },
    {
      "epoch": 8.994102478027343e-06,
      "model_forward_time": 1.741269588470459,
      "step": 83431
    },
    {
      "epoch": 8.994102478027343e-06,
      "step": 83431,
      "training_step_time": 7.96299147605896
    },
    {
      "epoch": 8.99505615234375e-06,
      "model_forward_time": 1.73829984664917,
      "step": 83432
    },
    {
      "epoch": 8.99505615234375e-06,
      "step": 83432,
      "training_step_time": 10.195799589157104
    },
    {
      "epoch": 8.996009826660156e-06,
      "model_forward_time": 2.3965842723846436,
      "step": 83433
    },
    {
      "epoch": 8.996009826660156e-06,
      "step": 83433,
      "training_step_time": 8.43195390701294
    },
    {
      "epoch": 8.996963500976562e-06,
      "model_forward_time": 1.7387423515319824,
      "step": 83434
    },
    {
      "epoch": 8.996963500976562e-06,
      "step": 83434,
      "training_step_time": 7.624763011932373
    },
    {
      "epoch": 8.99791717529297e-06,
      "model_forward_time": 1.742753505706787,
      "step": 83435
    },
    {
      "epoch": 8.99791717529297e-06,
      "step": 83435,
      "training_step_time": 8.994099140167236
    },
    {
      "epoch": 8.998870849609375e-06,
      "model_forward_time": 1.7405312061309814,
      "step": 83436
    },
    {
      "epoch": 8.998870849609375e-06,
      "step": 83436,
      "training_step_time": 7.742713928222656
    },
    {
      "epoch": 8.99982452392578e-06,
      "model_forward_time": 1.7405366897583008,
      "step": 83437
    },
    {
      "epoch": 8.99982452392578e-06,
      "step": 83437,
      "training_step_time": 7.52356743812561
    },
    {
      "epoch": 9.000778198242188e-06,
      "model_forward_time": 1.7397747039794922,
      "step": 83438
    },
    {
      "epoch": 9.000778198242188e-06,
      "step": 83438,
      "training_step_time": 6.908605337142944
    },
    {
      "epoch": 9.001731872558594e-06,
      "model_forward_time": 1.7385241985321045,
      "step": 83439
    },
    {
      "epoch": 9.001731872558594e-06,
      "step": 83439,
      "training_step_time": 8.489843845367432
    },
    {
      "epoch": 9.002685546875e-06,
      "grad_norm": 0.18106986582279205,
      "learning_rate": 7.312795426009695e-07,
      "loss": 0.0581,
      "step": 83440
    },
    {
      "epoch": 9.002685546875e-06,
      "model_forward_time": 1.7318115234375,
      "step": 83440
    },
    {
      "dynamics_loss_avg": 0.06191864851862192,
      "epoch": 9.002685546875e-06,
      "step": 83440
    },
    {
      "action_loss_avg": 0.0014301213319413363,
      "epoch": 9.002685546875e-06,
      "step": 83440
    },
    {
      "epoch": 9.002685546875e-06,
      "step": 83440,
      "training_step_time": 7.686085224151611
    },
    {
      "epoch": 9.003639221191407e-06,
      "model_forward_time": 1.7460596561431885,
      "step": 83441
    },
    {
      "epoch": 9.003639221191407e-06,
      "step": 83441,
      "training_step_time": 9.117557525634766
    },
    {
      "epoch": 9.004592895507813e-06,
      "model_forward_time": 1.80159592628479,
      "step": 83442
    },
    {
      "epoch": 9.004592895507813e-06,
      "step": 83442,
      "training_step_time": 8.809098482131958
    },
    {
      "epoch": 9.005546569824218e-06,
      "model_forward_time": 2.380693197250366,
      "step": 83443
    },
    {
      "epoch": 9.005546569824218e-06,
      "step": 83443,
      "training_step_time": 9.579566955566406
    },
    {
      "epoch": 9.006500244140626e-06,
      "model_forward_time": 1.7385222911834717,
      "step": 83444
    },
    {
      "epoch": 9.006500244140626e-06,
      "step": 83444,
      "training_step_time": 7.361103534698486
    },
    {
      "epoch": 9.007453918457032e-06,
      "model_forward_time": 1.7449054718017578,
      "step": 83445
    },
    {
      "epoch": 9.007453918457032e-06,
      "step": 83445,
      "training_step_time": 7.611447811126709
    },
    {
      "epoch": 9.008407592773437e-06,
      "model_forward_time": 1.7439277172088623,
      "step": 83446
    },
    {
      "epoch": 9.008407592773437e-06,
      "step": 83446,
      "training_step_time": 7.136398077011108
    },
    {
      "epoch": 9.009361267089843e-06,
      "model_forward_time": 1.7379474639892578,
      "step": 83447
    },
    {
      "epoch": 9.009361267089843e-06,
      "step": 83447,
      "training_step_time": 7.341159343719482
    },
    {
      "epoch": 9.01031494140625e-06,
      "model_forward_time": 1.749574899673462,
      "step": 83448
    },
    {
      "epoch": 9.01031494140625e-06,
      "step": 83448,
      "training_step_time": 7.595069169998169
    },
    {
      "epoch": 9.011268615722656e-06,
      "model_forward_time": 1.3778963088989258,
      "step": 83449
    },
    {
      "epoch": 9.011268615722656e-06,
      "step": 83449,
      "training_step_time": 7.670468091964722
    },
    {
      "epoch": 9.012222290039062e-06,
      "grad_norm": 0.09497056901454926,
      "learning_rate": 7.304188261435586e-07,
      "loss": 0.0588,
      "step": 83450
    },
    {
      "epoch": 9.012222290039062e-06,
      "model_forward_time": 1.7488739490509033,
      "step": 83450
    },
    {
      "dynamics_loss_avg": 0.08257592283189297,
      "epoch": 9.012222290039062e-06,
      "step": 83450
    },
    {
      "action_loss_avg": 0.0021378250035922973,
      "epoch": 9.012222290039062e-06,
      "step": 83450
    },
    {
      "epoch": 9.012222290039062e-06,
      "step": 83450,
      "training_step_time": 10.216809034347534
    },
    {
      "epoch": 9.01317596435547e-06,
      "model_forward_time": 1.7601594924926758,
      "step": 83451
    },
    {
      "epoch": 9.01317596435547e-06,
      "step": 83451,
      "training_step_time": 9.452962398529053
    },
    {
      "epoch": 9.014129638671875e-06,
      "model_forward_time": 2.410268545150757,
      "step": 83452
    },
    {
      "epoch": 9.014129638671875e-06,
      "step": 83452,
      "training_step_time": 7.6321423053741455
    },
    {
      "epoch": 9.01508331298828e-06,
      "model_forward_time": 1.7537710666656494,
      "step": 83453
    },
    {
      "epoch": 9.01508331298828e-06,
      "step": 83453,
      "training_step_time": 7.497663497924805
    },
    {
      "epoch": 9.016036987304688e-06,
      "model_forward_time": 1.7534668445587158,
      "step": 83454
    },
    {
      "epoch": 9.016036987304688e-06,
      "step": 83454,
      "training_step_time": 8.20519232749939
    },
    {
      "epoch": 9.016990661621094e-06,
      "model_forward_time": 1.7539575099945068,
      "step": 83455
    },
    {
      "epoch": 9.016990661621094e-06,
      "step": 83455,
      "training_step_time": 7.595254182815552
    },
    {
      "epoch": 9.0179443359375e-06,
      "model_forward_time": 1.7505557537078857,
      "step": 83456
    },
    {
      "epoch": 9.0179443359375e-06,
      "step": 83456,
      "training_step_time": 7.3569231033325195
    },
    {
      "epoch": 9.018898010253907e-06,
      "model_forward_time": 1.7493391036987305,
      "step": 83457
    },
    {
      "epoch": 9.018898010253907e-06,
      "step": 83457,
      "training_step_time": 8.407962322235107
    },
    {
      "epoch": 9.019851684570313e-06,
      "model_forward_time": 1.7591564655303955,
      "step": 83458
    },
    {
      "epoch": 9.019851684570313e-06,
      "step": 83458,
      "training_step_time": 7.6249470710754395
    },
    {
      "epoch": 9.020805358886718e-06,
      "model_forward_time": 1.7489416599273682,
      "step": 83459
    },
    {
      "epoch": 9.020805358886718e-06,
      "step": 83459,
      "training_step_time": 9.5792396068573
    },
    {
      "epoch": 9.021759033203126e-06,
      "grad_norm": 0.06746895611286163,
      "learning_rate": 7.295585766011243e-07,
      "loss": 0.0565,
      "step": 83460
    },
    {
      "epoch": 9.021759033203126e-06,
      "model_forward_time": 1.7460837364196777,
      "step": 83460
    },
    {
      "dynamics_loss_avg": 0.053172149881720546,
      "epoch": 9.021759033203126e-06,
      "step": 83460
    },
    {
      "action_loss_avg": 0.00163483776486828,
      "epoch": 9.021759033203126e-06,
      "step": 83460
    },
    {
      "epoch": 9.021759033203126e-06,
      "step": 83460,
      "training_step_time": 8.425554990768433
    },
    {
      "epoch": 9.022712707519532e-06,
      "model_forward_time": 2.4331130981445312,
      "step": 83461
    },
    {
      "epoch": 9.022712707519532e-06,
      "step": 83461,
      "training_step_time": 7.67594313621521
    },
    {
      "epoch": 9.023666381835937e-06,
      "model_forward_time": 1.7561004161834717,
      "step": 83462
    },
    {
      "epoch": 9.023666381835937e-06,
      "step": 83462,
      "training_step_time": 8.436273336410522
    },
    {
      "epoch": 9.024620056152343e-06,
      "model_forward_time": 1.7495741844177246,
      "step": 83463
    },
    {
      "epoch": 9.024620056152343e-06,
      "step": 83463,
      "training_step_time": 7.666805982589722
    },
    {
      "epoch": 9.02557373046875e-06,
      "model_forward_time": 1.7514984607696533,
      "step": 83464
    },
    {
      "epoch": 9.02557373046875e-06,
      "step": 83464,
      "training_step_time": 7.648355007171631
    },
    {
      "epoch": 9.026527404785156e-06,
      "model_forward_time": 1.7511718273162842,
      "step": 83465
    },
    {
      "epoch": 9.026527404785156e-06,
      "step": 83465,
      "training_step_time": 7.389797210693359
    },
    {
      "epoch": 9.027481079101562e-06,
      "model_forward_time": 1.7532405853271484,
      "step": 83466
    },
    {
      "epoch": 9.027481079101562e-06,
      "step": 83466,
      "training_step_time": 7.585828065872192
    },
    {
      "epoch": 9.02843475341797e-06,
      "model_forward_time": 1.7545289993286133,
      "step": 83467
    },
    {
      "epoch": 9.02843475341797e-06,
      "step": 83467,
      "training_step_time": 8.570100545883179
    },
    {
      "epoch": 9.029388427734375e-06,
      "model_forward_time": 1.7485837936401367,
      "step": 83468
    },
    {
      "epoch": 9.029388427734375e-06,
      "step": 83468,
      "training_step_time": 9.421007633209229
    },
    {
      "epoch": 9.03034210205078e-06,
      "model_forward_time": 1.7545011043548584,
      "step": 83469
    },
    {
      "epoch": 9.03034210205078e-06,
      "step": 83469,
      "training_step_time": 8.568317413330078
    },
    {
      "epoch": 9.031295776367188e-06,
      "grad_norm": 0.08970535546541214,
      "learning_rate": 7.286987940677437e-07,
      "loss": 0.0582,
      "step": 83470
    },
    {
      "epoch": 9.031295776367188e-06,
      "model_forward_time": 2.4318559169769287,
      "step": 83470
    },
    {
      "dynamics_loss_avg": 0.06053628493100405,
      "epoch": 9.031295776367188e-06,
      "step": 83470
    },
    {
      "action_loss_avg": 0.0020940219052135943,
      "epoch": 9.031295776367188e-06,
      "step": 83470
    },
    {
      "epoch": 9.031295776367188e-06,
      "step": 83470,
      "training_step_time": 7.701933145523071
    },
    {
      "epoch": 9.032249450683594e-06,
      "model_forward_time": 1.752303123474121,
      "step": 83471
    },
    {
      "epoch": 9.032249450683594e-06,
      "step": 83471,
      "training_step_time": 7.5595011711120605
    },
    {
      "epoch": 9.033203125e-06,
      "model_forward_time": 1.7517368793487549,
      "step": 83472
    },
    {
      "epoch": 9.033203125e-06,
      "step": 83472,
      "training_step_time": 7.708599328994751
    },
    {
      "epoch": 9.034156799316407e-06,
      "model_forward_time": 1.7499470710754395,
      "step": 83473
    },
    {
      "epoch": 9.034156799316407e-06,
      "step": 83473,
      "training_step_time": 7.607825517654419
    },
    {
      "epoch": 9.035110473632813e-06,
      "model_forward_time": 1.7501049041748047,
      "step": 83474
    },
    {
      "epoch": 9.035110473632813e-06,
      "step": 83474,
      "training_step_time": 7.3493945598602295
    },
    {
      "epoch": 9.036064147949218e-06,
      "model_forward_time": 1.7471561431884766,
      "step": 83475
    },
    {
      "epoch": 9.036064147949218e-06,
      "step": 83475,
      "training_step_time": 7.571995735168457
    },
    {
      "epoch": 9.037017822265626e-06,
      "model_forward_time": 1.7567920684814453,
      "step": 83476
    },
    {
      "epoch": 9.037017822265626e-06,
      "step": 83476,
      "training_step_time": 7.591130495071411
    },
    {
      "epoch": 9.037971496582032e-06,
      "model_forward_time": 1.7512054443359375,
      "step": 83477
    },
    {
      "epoch": 9.037971496582032e-06,
      "step": 83477,
      "training_step_time": 7.899221658706665
    },
    {
      "epoch": 9.038925170898437e-06,
      "model_forward_time": 1.7554278373718262,
      "step": 83478
    },
    {
      "epoch": 9.038925170898437e-06,
      "step": 83478,
      "training_step_time": 9.05874752998352
    },
    {
      "epoch": 9.039878845214843e-06,
      "model_forward_time": 1.7822232246398926,
      "step": 83479
    },
    {
      "epoch": 9.039878845214843e-06,
      "step": 83479,
      "training_step_time": 8.914179563522339
    },
    {
      "epoch": 9.04083251953125e-06,
      "grad_norm": 0.08809175342321396,
      "learning_rate": 7.278394786374427e-07,
      "loss": 0.0567,
      "step": 83480
    },
    {
      "epoch": 9.04083251953125e-06,
      "model_forward_time": 1.7466380596160889,
      "step": 83480
    },
    {
      "dynamics_loss_avg": 0.04942073505371809,
      "epoch": 9.04083251953125e-06,
      "step": 83480
    },
    {
      "action_loss_avg": 0.001233042374951765,
      "epoch": 9.04083251953125e-06,
      "step": 83480
    },
    {
      "epoch": 9.04083251953125e-06,
      "step": 83480,
      "training_step_time": 9.556045532226562
    },
    {
      "epoch": 9.041786193847656e-06,
      "model_forward_time": 1.745441198348999,
      "step": 83481
    },
    {
      "epoch": 9.041786193847656e-06,
      "step": 83481,
      "training_step_time": 7.751153945922852
    },
    {
      "epoch": 9.042739868164062e-06,
      "model_forward_time": 1.7489490509033203,
      "step": 83482
    },
    {
      "epoch": 9.042739868164062e-06,
      "step": 83482,
      "training_step_time": 7.535090684890747
    },
    {
      "epoch": 9.04369354248047e-06,
      "model_forward_time": 1.7546260356903076,
      "step": 83483
    },
    {
      "epoch": 9.04369354248047e-06,
      "step": 83483,
      "training_step_time": 7.549165487289429
    },
    {
      "epoch": 9.044647216796875e-06,
      "model_forward_time": 1.7295403480529785,
      "step": 83484
    },
    {
      "epoch": 9.044647216796875e-06,
      "step": 83484,
      "training_step_time": 7.702901840209961
    },
    {
      "epoch": 9.04560089111328e-06,
      "model_forward_time": 1.737257719039917,
      "step": 83485
    },
    {
      "epoch": 9.04560089111328e-06,
      "step": 83485,
      "training_step_time": 8.548450946807861
    },
    {
      "epoch": 9.046554565429688e-06,
      "model_forward_time": 1.367544412612915,
      "step": 83486
    },
    {
      "epoch": 9.046554565429688e-06,
      "step": 83486,
      "training_step_time": 7.783480167388916
    },
    {
      "epoch": 9.047508239746094e-06,
      "model_forward_time": 1.7502775192260742,
      "step": 83487
    },
    {
      "epoch": 9.047508239746094e-06,
      "step": 83487,
      "training_step_time": 9.3865647315979
    },
    {
      "epoch": 9.0484619140625e-06,
      "model_forward_time": 1.7588903903961182,
      "step": 83488
    },
    {
      "epoch": 9.0484619140625e-06,
      "step": 83488,
      "training_step_time": 7.797992944717407
    },
    {
      "epoch": 9.049415588378907e-06,
      "model_forward_time": 2.370926856994629,
      "step": 83489
    },
    {
      "epoch": 9.049415588378907e-06,
      "step": 83489,
      "training_step_time": 7.7092366218566895
    },
    {
      "epoch": 9.050369262695313e-06,
      "grad_norm": 0.08023130893707275,
      "learning_rate": 7.269806304041915e-07,
      "loss": 0.0556,
      "step": 83490
    },
    {
      "epoch": 9.050369262695313e-06,
      "model_forward_time": 1.7387218475341797,
      "step": 83490
    },
    {
      "dynamics_loss_avg": 0.061291616223752496,
      "epoch": 9.050369262695313e-06,
      "step": 83490
    },
    {
      "action_loss_avg": 0.0016013796703191474,
      "epoch": 9.050369262695313e-06,
      "step": 83490
    },
    {
      "epoch": 9.050369262695313e-06,
      "step": 83490,
      "training_step_time": 7.770200967788696
    },
    {
      "epoch": 9.051322937011718e-06,
      "model_forward_time": 1.7403628826141357,
      "step": 83491
    },
    {
      "epoch": 9.051322937011718e-06,
      "step": 83491,
      "training_step_time": 7.601197242736816
    },
    {
      "epoch": 9.052276611328126e-06,
      "model_forward_time": 1.746260166168213,
      "step": 83492
    },
    {
      "epoch": 9.052276611328126e-06,
      "step": 83492,
      "training_step_time": 8.545742511749268
    },
    {
      "epoch": 9.053230285644532e-06,
      "model_forward_time": 1.7432727813720703,
      "step": 83493
    },
    {
      "epoch": 9.053230285644532e-06,
      "step": 83493,
      "training_step_time": 7.6635191440582275
    },
    {
      "epoch": 9.054183959960937e-06,
      "model_forward_time": 1.7464306354522705,
      "step": 83494
    },
    {
      "epoch": 9.054183959960937e-06,
      "step": 83494,
      "training_step_time": 7.569345951080322
    },
    {
      "epoch": 9.055137634277343e-06,
      "model_forward_time": 1.740109920501709,
      "step": 83495
    },
    {
      "epoch": 9.055137634277343e-06,
      "step": 83495,
      "training_step_time": 7.719193696975708
    },
    {
      "epoch": 9.05609130859375e-06,
      "model_forward_time": 1.743232011795044,
      "step": 83496
    },
    {
      "epoch": 9.05609130859375e-06,
      "step": 83496,
      "training_step_time": 8.481034517288208
    },
    {
      "epoch": 9.057044982910156e-06,
      "model_forward_time": 1.7511215209960938,
      "step": 83497
    },
    {
      "epoch": 9.057044982910156e-06,
      "step": 83497,
      "training_step_time": 8.544461727142334
    },
    {
      "epoch": 9.057998657226562e-06,
      "model_forward_time": 2.3940038681030273,
      "step": 83498
    },
    {
      "epoch": 9.057998657226562e-06,
      "step": 83498,
      "training_step_time": 8.382982730865479
    },
    {
      "epoch": 9.05895233154297e-06,
      "model_forward_time": 1.7432267665863037,
      "step": 83499
    },
    {
      "epoch": 9.05895233154297e-06,
      "step": 83499,
      "training_step_time": 7.841106176376343
    },
    {
      "epoch": 9.059906005859375e-06,
      "grad_norm": 0.08365292102098465,
      "learning_rate": 7.261222494619146e-07,
      "loss": 0.0542,
      "step": 83500
    },
    {
      "epoch": 9.059906005859375e-06,
      "model_forward_time": 1.7378051280975342,
      "step": 83500
    },
    {
      "dynamics_loss_avg": 0.05977313239127398,
      "epoch": 9.059906005859375e-06,
      "step": 83500
    },
    {
      "action_loss_avg": 0.0020240555342752485,
      "epoch": 9.059906005859375e-06,
      "step": 83500
    },
    {
      "epoch": 9.059906005859375e-06,
      "step": 83500,
      "training_step_time": 8.236558675765991
    },
    {
      "epoch": 9.06085968017578e-06,
      "model_forward_time": 1.7317135334014893,
      "step": 83501
    },
    {
      "epoch": 9.06085968017578e-06,
      "step": 83501,
      "training_step_time": 7.610639572143555
    },
    {
      "epoch": 9.061813354492188e-06,
      "model_forward_time": 1.7367193698883057,
      "step": 83502
    },
    {
      "epoch": 9.061813354492188e-06,
      "step": 83502,
      "training_step_time": 7.585450887680054
    },
    {
      "epoch": 9.062767028808594e-06,
      "model_forward_time": 1.3709633350372314,
      "step": 83503
    },
    {
      "epoch": 9.062767028808594e-06,
      "step": 83503,
      "training_step_time": 7.615880012512207
    },
    {
      "epoch": 9.063720703125e-06,
      "model_forward_time": 1.7395052909851074,
      "step": 83504
    },
    {
      "epoch": 9.063720703125e-06,
      "step": 83504,
      "training_step_time": 7.709181785583496
    },
    {
      "epoch": 9.064674377441407e-06,
      "model_forward_time": 1.7366535663604736,
      "step": 83505
    },
    {
      "epoch": 9.064674377441407e-06,
      "step": 83505,
      "training_step_time": 9.75892949104309
    },
    {
      "epoch": 9.065628051757813e-06,
      "model_forward_time": 1.743990421295166,
      "step": 83506
    },
    {
      "epoch": 9.065628051757813e-06,
      "step": 83506,
      "training_step_time": 9.03231692314148
    },
    {
      "epoch": 9.066581726074218e-06,
      "model_forward_time": 2.3777027130126953,
      "step": 83507
    },
    {
      "epoch": 9.066581726074218e-06,
      "step": 83507,
      "training_step_time": 7.7135419845581055
    },
    {
      "epoch": 9.067535400390624e-06,
      "model_forward_time": 1.7368450164794922,
      "step": 83508
    },
    {
      "epoch": 9.067535400390624e-06,
      "step": 83508,
      "training_step_time": 7.77090311050415
    },
    {
      "epoch": 9.068489074707032e-06,
      "model_forward_time": 1.7590563297271729,
      "step": 83509
    },
    {
      "epoch": 9.068489074707032e-06,
      "step": 83509,
      "training_step_time": 7.469471454620361
    },
    {
      "epoch": 9.069442749023437e-06,
      "grad_norm": 0.08117760717868805,
      "learning_rate": 7.252643359044825e-07,
      "loss": 0.0553,
      "step": 83510
    },
    {
      "epoch": 9.069442749023437e-06,
      "model_forward_time": 1.7567284107208252,
      "step": 83510
    },
    {
      "dynamics_loss_avg": 0.06597336754202843,
      "epoch": 9.069442749023437e-06,
      "step": 83510
    },
    {
      "action_loss_avg": 0.0014695610298076645,
      "epoch": 9.069442749023437e-06,
      "step": 83510
    },
    {
      "epoch": 9.069442749023437e-06,
      "step": 83510,
      "training_step_time": 7.622028827667236
    },
    {
      "epoch": 9.070396423339843e-06,
      "model_forward_time": 1.7637951374053955,
      "step": 83511
    },
    {
      "epoch": 9.070396423339843e-06,
      "step": 83511,
      "training_step_time": 7.681953430175781
    },
    {
      "epoch": 9.07135009765625e-06,
      "model_forward_time": 1.7589426040649414,
      "step": 83512
    },
    {
      "epoch": 9.07135009765625e-06,
      "step": 83512,
      "training_step_time": 7.5974814891815186
    },
    {
      "epoch": 9.072303771972656e-06,
      "model_forward_time": 1.755171298980713,
      "step": 83513
    },
    {
      "epoch": 9.072303771972656e-06,
      "step": 83513,
      "training_step_time": 8.121659755706787
    },
    {
      "epoch": 9.073257446289062e-06,
      "model_forward_time": 1.7661385536193848,
      "step": 83514
    },
    {
      "epoch": 9.073257446289062e-06,
      "step": 83514,
      "training_step_time": 7.9048755168914795
    },
    {
      "epoch": 9.07421112060547e-06,
      "model_forward_time": 1.3859944343566895,
      "step": 83515
    },
    {
      "epoch": 9.07421112060547e-06,
      "step": 83515,
      "training_step_time": 7.812505006790161
    },
    {
      "epoch": 9.075164794921875e-06,
      "model_forward_time": 1.7854135036468506,
      "step": 83516
    },
    {
      "epoch": 9.075164794921875e-06,
      "step": 83516,
      "training_step_time": 8.88144040107727
    },
    {
      "epoch": 9.07611846923828e-06,
      "model_forward_time": 1.7341725826263428,
      "step": 83517
    },
    {
      "epoch": 9.07611846923828e-06,
      "step": 83517,
      "training_step_time": 7.847287893295288
    },
    {
      "epoch": 9.077072143554688e-06,
      "model_forward_time": 1.7501277923583984,
      "step": 83518
    },
    {
      "epoch": 9.077072143554688e-06,
      "step": 83518,
      "training_step_time": 7.503419399261475
    },
    {
      "epoch": 9.078025817871094e-06,
      "model_forward_time": 1.7485907077789307,
      "step": 83519
    },
    {
      "epoch": 9.078025817871094e-06,
      "step": 83519,
      "training_step_time": 7.658135414123535
    },
    {
      "epoch": 9.0789794921875e-06,
      "grad_norm": 0.07790816575288773,
      "learning_rate": 7.244068898257173e-07,
      "loss": 0.0546,
      "step": 83520
    },
    {
      "epoch": 9.0789794921875e-06,
      "model_forward_time": 1.7514252662658691,
      "step": 83520
    },
    {
      "dynamics_loss_avg": 0.056552615482360125,
      "epoch": 9.0789794921875e-06,
      "step": 83520
    },
    {
      "action_loss_avg": 0.003351545619079843,
      "epoch": 9.0789794921875e-06,
      "step": 83520
    },
    {
      "epoch": 9.0789794921875e-06,
      "step": 83520,
      "training_step_time": 7.736928701400757
    },
    {
      "epoch": 9.079933166503907e-06,
      "model_forward_time": 1.7514300346374512,
      "step": 83521
    },
    {
      "epoch": 9.079933166503907e-06,
      "step": 83521,
      "training_step_time": 7.600943565368652
    },
    {
      "epoch": 9.080886840820313e-06,
      "model_forward_time": 1.7600774765014648,
      "step": 83522
    },
    {
      "epoch": 9.080886840820313e-06,
      "step": 83522,
      "training_step_time": 8.185598611831665
    },
    {
      "epoch": 9.081840515136718e-06,
      "model_forward_time": 1.7953062057495117,
      "step": 83523
    },
    {
      "epoch": 9.081840515136718e-06,
      "step": 83523,
      "training_step_time": 9.57814073562622
    },
    {
      "epoch": 9.082794189453124e-06,
      "model_forward_time": 1.7510578632354736,
      "step": 83524
    },
    {
      "epoch": 9.082794189453124e-06,
      "step": 83524,
      "training_step_time": 9.615951776504517
    },
    {
      "epoch": 9.083747863769532e-06,
      "model_forward_time": 1.3738107681274414,
      "step": 83525
    },
    {
      "epoch": 9.083747863769532e-06,
      "step": 83525,
      "training_step_time": 9.62391972541809
    },
    {
      "epoch": 9.084701538085937e-06,
      "model_forward_time": 2.4418458938598633,
      "step": 83526
    },
    {
      "epoch": 9.084701538085937e-06,
      "step": 83526,
      "training_step_time": 7.758945941925049
    },
    {
      "epoch": 9.085655212402343e-06,
      "model_forward_time": 1.7715606689453125,
      "step": 83527
    },
    {
      "epoch": 9.085655212402343e-06,
      "step": 83527,
      "training_step_time": 7.461824893951416
    },
    {
      "epoch": 9.08660888671875e-06,
      "model_forward_time": 1.8246445655822754,
      "step": 83528
    },
    {
      "epoch": 9.08660888671875e-06,
      "step": 83528,
      "training_step_time": 7.570372581481934
    },
    {
      "epoch": 9.087562561035156e-06,
      "model_forward_time": 1.7735755443572998,
      "step": 83529
    },
    {
      "epoch": 9.087562561035156e-06,
      "step": 83529,
      "training_step_time": 7.6451146602630615
    },
    {
      "epoch": 9.088516235351562e-06,
      "grad_norm": 0.2544724941253662,
      "learning_rate": 7.23549911319385e-07,
      "loss": 0.0532,
      "step": 83530
    },
    {
      "epoch": 9.088516235351562e-06,
      "model_forward_time": 1.7669131755828857,
      "step": 83530
    },
    {
      "dynamics_loss_avg": 0.06111950110644102,
      "epoch": 9.088516235351562e-06,
      "step": 83530
    },
    {
      "action_loss_avg": 0.0012531333544757218,
      "epoch": 9.088516235351562e-06,
      "step": 83530
    },
    {
      "epoch": 9.088516235351562e-06,
      "step": 83530,
      "training_step_time": 7.2281458377838135
    },
    {
      "epoch": 9.08946990966797e-06,
      "model_forward_time": 1.7918260097503662,
      "step": 83531
    },
    {
      "epoch": 9.08946990966797e-06,
      "step": 83531,
      "training_step_time": 7.543680667877197
    },
    {
      "epoch": 9.090423583984375e-06,
      "model_forward_time": 1.7790734767913818,
      "step": 83532
    },
    {
      "epoch": 9.090423583984375e-06,
      "step": 83532,
      "training_step_time": 7.818977355957031
    },
    {
      "epoch": 9.09137725830078e-06,
      "model_forward_time": 1.790198564529419,
      "step": 83533
    },
    {
      "epoch": 9.09137725830078e-06,
      "step": 83533,
      "training_step_time": 9.343549013137817
    },
    {
      "epoch": 9.092330932617188e-06,
      "model_forward_time": 1.4049091339111328,
      "step": 83534
    },
    {
      "epoch": 9.092330932617188e-06,
      "step": 83534,
      "training_step_time": 7.797486305236816
    },
    {
      "epoch": 9.093284606933594e-06,
      "model_forward_time": 2.460731029510498,
      "step": 83535
    },
    {
      "epoch": 9.093284606933594e-06,
      "step": 83535,
      "training_step_time": 8.653613805770874
    },
    {
      "epoch": 9.09423828125e-06,
      "model_forward_time": 1.7780709266662598,
      "step": 83536
    },
    {
      "epoch": 9.09423828125e-06,
      "step": 83536,
      "training_step_time": 7.56458044052124
    },
    {
      "epoch": 9.095191955566407e-06,
      "model_forward_time": 1.7772185802459717,
      "step": 83537
    },
    {
      "epoch": 9.095191955566407e-06,
      "step": 83537,
      "training_step_time": 7.309810638427734
    },
    {
      "epoch": 9.096145629882813e-06,
      "model_forward_time": 1.7753262519836426,
      "step": 83538
    },
    {
      "epoch": 9.096145629882813e-06,
      "step": 83538,
      "training_step_time": 7.6275811195373535
    },
    {
      "epoch": 9.097099304199218e-06,
      "model_forward_time": 1.7714409828186035,
      "step": 83539
    },
    {
      "epoch": 9.097099304199218e-06,
      "step": 83539,
      "training_step_time": 7.341427564620972
    },
    {
      "epoch": 9.098052978515624e-06,
      "grad_norm": 0.09370886534452438,
      "learning_rate": 7.226934004792058e-07,
      "loss": 0.0538,
      "step": 83540
    },
    {
      "epoch": 9.098052978515624e-06,
      "model_forward_time": 1.7610931396484375,
      "step": 83540
    },
    {
      "dynamics_loss_avg": 0.0607266578823328,
      "epoch": 9.098052978515624e-06,
      "step": 83540
    },
    {
      "action_loss_avg": 0.0014018111047334968,
      "epoch": 9.098052978515624e-06,
      "step": 83540
    },
    {
      "epoch": 9.098052978515624e-06,
      "step": 83540,
      "training_step_time": 7.552289247512817
    },
    {
      "epoch": 9.099006652832032e-06,
      "model_forward_time": 1.7830049991607666,
      "step": 83541
    },
    {
      "epoch": 9.099006652832032e-06,
      "step": 83541,
      "training_step_time": 9.166328430175781
    },
    {
      "epoch": 9.099960327148437e-06,
      "model_forward_time": 1.7743067741394043,
      "step": 83542
    },
    {
      "epoch": 9.099960327148437e-06,
      "step": 83542,
      "training_step_time": 9.021135330200195
    },
    {
      "epoch": 9.100914001464843e-06,
      "model_forward_time": 1.065239429473877,
      "step": 83543
    },
    {
      "epoch": 9.100914001464843e-06,
      "step": 83543,
      "training_step_time": 8.544820547103882
    },
    {
      "epoch": 9.10186767578125e-06,
      "model_forward_time": 1.7909445762634277,
      "step": 83544
    },
    {
      "epoch": 9.10186767578125e-06,
      "step": 83544,
      "training_step_time": 8.580594539642334
    },
    {
      "epoch": 9.102821350097656e-06,
      "model_forward_time": 2.4745919704437256,
      "step": 83545
    },
    {
      "epoch": 9.102821350097656e-06,
      "step": 83545,
      "training_step_time": 7.691979885101318
    },
    {
      "epoch": 9.103775024414062e-06,
      "model_forward_time": 1.770838975906372,
      "step": 83546
    },
    {
      "epoch": 9.103775024414062e-06,
      "step": 83546,
      "training_step_time": 7.5478761196136475
    },
    {
      "epoch": 9.10472869873047e-06,
      "model_forward_time": 1.7828261852264404,
      "step": 83547
    },
    {
      "epoch": 9.10472869873047e-06,
      "step": 83547,
      "training_step_time": 7.848657131195068
    },
    {
      "epoch": 9.105682373046875e-06,
      "model_forward_time": 1.7706081867218018,
      "step": 83548
    },
    {
      "epoch": 9.105682373046875e-06,
      "step": 83548,
      "training_step_time": 7.32621431350708
    },
    {
      "epoch": 9.10663604736328e-06,
      "model_forward_time": 1.7878143787384033,
      "step": 83549
    },
    {
      "epoch": 9.10663604736328e-06,
      "step": 83549,
      "training_step_time": 7.64394211769104
    },
    {
      "epoch": 9.107589721679688e-06,
      "grad_norm": 0.0958809033036232,
      "learning_rate": 7.218373573988436e-07,
      "loss": 0.0547,
      "step": 83550
    },
    {
      "epoch": 9.107589721679688e-06,
      "model_forward_time": 1.7617104053497314,
      "step": 83550
    },
    {
      "dynamics_loss_avg": 0.04576642122119665,
      "epoch": 9.107589721679688e-06,
      "step": 83550
    },
    {
      "action_loss_avg": 0.0014381299348315223,
      "epoch": 9.107589721679688e-06,
      "step": 83550
    },
    {
      "epoch": 9.107589721679688e-06,
      "step": 83550,
      "training_step_time": 8.335474491119385
    },
    {
      "epoch": 9.108543395996094e-06,
      "model_forward_time": 1.7733871936798096,
      "step": 83551
    },
    {
      "epoch": 9.108543395996094e-06,
      "step": 83551,
      "training_step_time": 7.912179231643677
    },
    {
      "epoch": 9.1094970703125e-06,
      "model_forward_time": 1.7662925720214844,
      "step": 83552
    },
    {
      "epoch": 9.1094970703125e-06,
      "step": 83552,
      "training_step_time": 10.726637125015259
    },
    {
      "epoch": 9.110450744628907e-06,
      "model_forward_time": 1.779951572418213,
      "step": 83553
    },
    {
      "epoch": 9.110450744628907e-06,
      "step": 83553,
      "training_step_time": 7.848882675170898
    },
    {
      "epoch": 9.111404418945313e-06,
      "model_forward_time": 2.4752683639526367,
      "step": 83554
    },
    {
      "epoch": 9.111404418945313e-06,
      "step": 83554,
      "training_step_time": 7.798141956329346
    },
    {
      "epoch": 9.112358093261718e-06,
      "model_forward_time": 1.7814042568206787,
      "step": 83555
    },
    {
      "epoch": 9.112358093261718e-06,
      "step": 83555,
      "training_step_time": 7.613756418228149
    },
    {
      "epoch": 9.113311767578124e-06,
      "model_forward_time": 1.7758135795593262,
      "step": 83556
    },
    {
      "epoch": 9.113311767578124e-06,
      "step": 83556,
      "training_step_time": 7.802770614624023
    },
    {
      "epoch": 9.114265441894532e-06,
      "model_forward_time": 1.779778003692627,
      "step": 83557
    },
    {
      "epoch": 9.114265441894532e-06,
      "step": 83557,
      "training_step_time": 7.801582336425781
    },
    {
      "epoch": 9.115219116210937e-06,
      "model_forward_time": 1.7765367031097412,
      "step": 83558
    },
    {
      "epoch": 9.115219116210937e-06,
      "step": 83558,
      "training_step_time": 7.7322351932525635
    },
    {
      "epoch": 9.116172790527343e-06,
      "model_forward_time": 1.7754416465759277,
      "step": 83559
    },
    {
      "epoch": 9.116172790527343e-06,
      "step": 83559,
      "training_step_time": 9.693032503128052
    },
    {
      "epoch": 9.11712646484375e-06,
      "grad_norm": 0.07677439600229263,
      "learning_rate": 7.209817821719179e-07,
      "loss": 0.055,
      "step": 83560
    },
    {
      "epoch": 9.11712646484375e-06,
      "model_forward_time": 1.763899564743042,
      "step": 83560
    },
    {
      "dynamics_loss_avg": 0.06986998803913594,
      "epoch": 9.11712646484375e-06,
      "step": 83560
    },
    {
      "action_loss_avg": 0.0014359521097503602,
      "epoch": 9.11712646484375e-06,
      "step": 83560
    },
    {
      "epoch": 9.11712646484375e-06,
      "step": 83560,
      "training_step_time": 8.482872009277344
    },
    {
      "epoch": 9.118080139160156e-06,
      "model_forward_time": 1.7747056484222412,
      "step": 83561
    },
    {
      "epoch": 9.118080139160156e-06,
      "step": 83561,
      "training_step_time": 10.067427158355713
    },
    {
      "epoch": 9.119033813476562e-06,
      "model_forward_time": 1.7824394702911377,
      "step": 83562
    },
    {
      "epoch": 9.119033813476562e-06,
      "step": 83562,
      "training_step_time": 8.882856130599976
    },
    {
      "epoch": 9.11998748779297e-06,
      "model_forward_time": 2.470639705657959,
      "step": 83563
    },
    {
      "epoch": 9.11998748779297e-06,
      "step": 83563,
      "training_step_time": 7.795252561569214
    },
    {
      "epoch": 9.120941162109375e-06,
      "model_forward_time": 1.804426908493042,
      "step": 83564
    },
    {
      "epoch": 9.120941162109375e-06,
      "step": 83564,
      "training_step_time": 8.41353964805603
    },
    {
      "epoch": 9.12189483642578e-06,
      "model_forward_time": 1.7986419200897217,
      "step": 83565
    },
    {
      "epoch": 9.12189483642578e-06,
      "step": 83565,
      "training_step_time": 7.7411253452301025
    },
    {
      "epoch": 9.122848510742188e-06,
      "model_forward_time": 1.7762842178344727,
      "step": 83566
    },
    {
      "epoch": 9.122848510742188e-06,
      "step": 83566,
      "training_step_time": 7.567527532577515
    },
    {
      "epoch": 9.123802185058594e-06,
      "model_forward_time": 1.77420973777771,
      "step": 83567
    },
    {
      "epoch": 9.123802185058594e-06,
      "step": 83567,
      "training_step_time": 7.677739143371582
    },
    {
      "epoch": 9.124755859375e-06,
      "model_forward_time": 1.7785694599151611,
      "step": 83568
    },
    {
      "epoch": 9.124755859375e-06,
      "step": 83568,
      "training_step_time": 8.716108798980713
    },
    {
      "epoch": 9.125709533691407e-06,
      "model_forward_time": 1.772871494293213,
      "step": 83569
    },
    {
      "epoch": 9.125709533691407e-06,
      "step": 83569,
      "training_step_time": 7.786798477172852
    },
    {
      "epoch": 9.126663208007813e-06,
      "grad_norm": 0.09301574528217316,
      "learning_rate": 7.201266748919894e-07,
      "loss": 0.0547,
      "step": 83570
    },
    {
      "epoch": 9.126663208007813e-06,
      "model_forward_time": 1.7628941535949707,
      "step": 83570
    },
    {
      "dynamics_loss_avg": 0.05560033228248358,
      "epoch": 9.126663208007813e-06,
      "step": 83570
    },
    {
      "action_loss_avg": 0.0018919835158158094,
      "epoch": 9.126663208007813e-06,
      "step": 83570
    },
    {
      "epoch": 9.126663208007813e-06,
      "step": 83570,
      "training_step_time": 9.245930671691895
    },
    {
      "epoch": 9.127616882324218e-06,
      "model_forward_time": 1.7723560333251953,
      "step": 83571
    },
    {
      "epoch": 9.127616882324218e-06,
      "step": 83571,
      "training_step_time": 8.987460851669312
    },
    {
      "epoch": 9.128570556640624e-06,
      "model_forward_time": 2.5014753341674805,
      "step": 83572
    },
    {
      "epoch": 9.128570556640624e-06,
      "step": 83572,
      "training_step_time": 7.7348573207855225
    },
    {
      "epoch": 9.129524230957032e-06,
      "model_forward_time": 1.7659249305725098,
      "step": 83573
    },
    {
      "epoch": 9.129524230957032e-06,
      "step": 83573,
      "training_step_time": 7.605951547622681
    },
    {
      "epoch": 9.130477905273437e-06,
      "model_forward_time": 1.767812728881836,
      "step": 83574
    },
    {
      "epoch": 9.130477905273437e-06,
      "step": 83574,
      "training_step_time": 8.325160503387451
    },
    {
      "epoch": 9.131431579589843e-06,
      "model_forward_time": 1.7860205173492432,
      "step": 83575
    },
    {
      "epoch": 9.131431579589843e-06,
      "step": 83575,
      "training_step_time": 7.670679092407227
    },
    {
      "epoch": 9.13238525390625e-06,
      "model_forward_time": 1.7720890045166016,
      "step": 83576
    },
    {
      "epoch": 9.13238525390625e-06,
      "step": 83576,
      "training_step_time": 8.325708150863647
    },
    {
      "epoch": 9.133338928222656e-06,
      "model_forward_time": 1.7667474746704102,
      "step": 83577
    },
    {
      "epoch": 9.133338928222656e-06,
      "step": 83577,
      "training_step_time": 7.68257212638855
    },
    {
      "epoch": 9.134292602539062e-06,
      "model_forward_time": 1.770686149597168,
      "step": 83578
    },
    {
      "epoch": 9.134292602539062e-06,
      "step": 83578,
      "training_step_time": 8.6050124168396
    },
    {
      "epoch": 9.13524627685547e-06,
      "model_forward_time": 1.7883141040802002,
      "step": 83579
    },
    {
      "epoch": 9.13524627685547e-06,
      "step": 83579,
      "training_step_time": 8.955413341522217
    },
    {
      "epoch": 9.136199951171875e-06,
      "grad_norm": 0.11906744539737701,
      "learning_rate": 7.192720356525751e-07,
      "loss": 0.0553,
      "step": 83580
    },
    {
      "epoch": 9.136199951171875e-06,
      "model_forward_time": 1.7610206604003906,
      "step": 83580
    },
    {
      "dynamics_loss_avg": 0.06087557226419449,
      "epoch": 9.136199951171875e-06,
      "step": 83580
    },
    {
      "action_loss_avg": 0.0017488027515355498,
      "epoch": 9.136199951171875e-06,
      "step": 83580
    },
    {
      "epoch": 9.136199951171875e-06,
      "step": 83580,
      "training_step_time": 8.261166334152222
    },
    {
      "epoch": 9.13715362548828e-06,
      "model_forward_time": 2.3958218097686768,
      "step": 83581
    },
    {
      "epoch": 9.13715362548828e-06,
      "step": 83581,
      "training_step_time": 7.756006479263306
    },
    {
      "epoch": 9.138107299804688e-06,
      "model_forward_time": 1.8800361156463623,
      "step": 83582
    },
    {
      "epoch": 9.138107299804688e-06,
      "step": 83582,
      "training_step_time": 7.68062949180603
    },
    {
      "epoch": 9.139060974121094e-06,
      "model_forward_time": 1.771766185760498,
      "step": 83583
    },
    {
      "epoch": 9.139060974121094e-06,
      "step": 83583,
      "training_step_time": 7.65391731262207
    },
    {
      "epoch": 9.1400146484375e-06,
      "model_forward_time": 1.760817050933838,
      "step": 83584
    },
    {
      "epoch": 9.1400146484375e-06,
      "step": 83584,
      "training_step_time": 7.646937608718872
    },
    {
      "epoch": 9.140968322753907e-06,
      "model_forward_time": 1.7683098316192627,
      "step": 83585
    },
    {
      "epoch": 9.140968322753907e-06,
      "step": 83585,
      "training_step_time": 7.321627616882324
    },
    {
      "epoch": 9.141921997070313e-06,
      "model_forward_time": 1.7665565013885498,
      "step": 83586
    },
    {
      "epoch": 9.141921997070313e-06,
      "step": 83586,
      "training_step_time": 9.520094633102417
    },
    {
      "epoch": 9.142875671386719e-06,
      "model_forward_time": 1.7786893844604492,
      "step": 83587
    },
    {
      "epoch": 9.142875671386719e-06,
      "step": 83587,
      "training_step_time": 9.310163974761963
    },
    {
      "epoch": 9.143829345703124e-06,
      "model_forward_time": 1.0621416568756104,
      "step": 83588
    },
    {
      "epoch": 9.143829345703124e-06,
      "step": 83588,
      "training_step_time": 8.133499145507812
    },
    {
      "epoch": 9.144783020019532e-06,
      "model_forward_time": 1.7673301696777344,
      "step": 83589
    },
    {
      "epoch": 9.144783020019532e-06,
      "step": 83589,
      "training_step_time": 7.681792974472046
    },
    {
      "epoch": 9.145736694335937e-06,
      "grad_norm": 0.08158162981271744,
      "learning_rate": 7.184178645471335e-07,
      "loss": 0.0556,
      "step": 83590
    },
    {
      "epoch": 9.145736694335937e-06,
      "model_forward_time": 1.7645537853240967,
      "step": 83590
    },
    {
      "dynamics_loss_avg": 0.046372168511152265,
      "epoch": 9.145736694335937e-06,
      "step": 83590
    },
    {
      "action_loss_avg": 0.0009670592669863253,
      "epoch": 9.145736694335937e-06,
      "step": 83590
    },
    {
      "epoch": 9.145736694335937e-06,
      "step": 83590,
      "training_step_time": 7.624243497848511
    },
    {
      "epoch": 9.146690368652343e-06,
      "model_forward_time": 2.454061985015869,
      "step": 83591
    },
    {
      "epoch": 9.146690368652343e-06,
      "step": 83591,
      "training_step_time": 7.560938596725464
    },
    {
      "epoch": 9.14764404296875e-06,
      "model_forward_time": 1.7740304470062256,
      "step": 83592
    },
    {
      "epoch": 9.14764404296875e-06,
      "step": 83592,
      "training_step_time": 7.681007385253906
    },
    {
      "epoch": 9.148597717285156e-06,
      "model_forward_time": 1.7582600116729736,
      "step": 83593
    },
    {
      "epoch": 9.148597717285156e-06,
      "step": 83593,
      "training_step_time": 7.551601886749268
    },
    {
      "epoch": 9.149551391601562e-06,
      "model_forward_time": 1.7801356315612793,
      "step": 83594
    },
    {
      "epoch": 9.149551391601562e-06,
      "step": 83594,
      "training_step_time": 8.61427617073059
    },
    {
      "epoch": 9.15050506591797e-06,
      "model_forward_time": 1.763725996017456,
      "step": 83595
    },
    {
      "epoch": 9.15050506591797e-06,
      "step": 83595,
      "training_step_time": 7.585342884063721
    },
    {
      "epoch": 9.151458740234375e-06,
      "model_forward_time": 1.7751333713531494,
      "step": 83596
    },
    {
      "epoch": 9.151458740234375e-06,
      "step": 83596,
      "training_step_time": 9.092148065567017
    },
    {
      "epoch": 9.15241241455078e-06,
      "model_forward_time": 1.7699410915374756,
      "step": 83597
    },
    {
      "epoch": 9.15241241455078e-06,
      "step": 83597,
      "training_step_time": 9.3227379322052
    },
    {
      "epoch": 9.153366088867188e-06,
      "model_forward_time": 1.7543985843658447,
      "step": 83598
    },
    {
      "epoch": 9.153366088867188e-06,
      "step": 83598,
      "training_step_time": 8.66068434715271
    },
    {
      "epoch": 9.154319763183594e-06,
      "model_forward_time": 1.7685306072235107,
      "step": 83599
    },
    {
      "epoch": 9.154319763183594e-06,
      "step": 83599,
      "training_step_time": 8.794997453689575
    },
    {
      "epoch": 9.1552734375e-06,
      "grad_norm": 0.08998040854930878,
      "learning_rate": 7.175641616690776e-07,
      "loss": 0.0547,
      "step": 83600
    },
    {
      "epoch": 9.1552734375e-06,
      "model_forward_time": 2.4452064037323,
      "step": 83600
    },
    {
      "dynamics_loss_avg": 0.05922615639865399,
      "epoch": 9.1552734375e-06,
      "step": 83600
    },
    {
      "action_loss_avg": 0.0022091137449024244,
      "epoch": 9.1552734375e-06,
      "step": 83600
    },
    {
      "epoch": 9.1552734375e-06,
      "step": 83600,
      "training_step_time": 8.419779062271118
    },
    {
      "epoch": 9.156227111816407e-06,
      "model_forward_time": 1.7784998416900635,
      "step": 83601
    },
    {
      "epoch": 9.156227111816407e-06,
      "step": 83601,
      "training_step_time": 7.778461694717407
    },
    {
      "epoch": 9.157180786132813e-06,
      "model_forward_time": 1.7742846012115479,
      "step": 83602
    },
    {
      "epoch": 9.157180786132813e-06,
      "step": 83602,
      "training_step_time": 7.58552098274231
    },
    {
      "epoch": 9.158134460449219e-06,
      "model_forward_time": 1.7755391597747803,
      "step": 83603
    },
    {
      "epoch": 9.158134460449219e-06,
      "step": 83603,
      "training_step_time": 8.52722692489624
    },
    {
      "epoch": 9.159088134765624e-06,
      "model_forward_time": 1.7805495262145996,
      "step": 83604
    },
    {
      "epoch": 9.159088134765624e-06,
      "step": 83604,
      "training_step_time": 7.6730523109436035
    },
    {
      "epoch": 9.160041809082032e-06,
      "model_forward_time": 1.7706482410430908,
      "step": 83605
    },
    {
      "epoch": 9.160041809082032e-06,
      "step": 83605,
      "training_step_time": 8.495258569717407
    },
    {
      "epoch": 9.160995483398437e-06,
      "model_forward_time": 1.7704451084136963,
      "step": 83606
    },
    {
      "epoch": 9.160995483398437e-06,
      "step": 83606,
      "training_step_time": 9.413204908370972
    },
    {
      "epoch": 9.161949157714843e-06,
      "model_forward_time": 1.7738733291625977,
      "step": 83607
    },
    {
      "epoch": 9.161949157714843e-06,
      "step": 83607,
      "training_step_time": 9.073111772537231
    },
    {
      "epoch": 9.16290283203125e-06,
      "model_forward_time": 1.4094839096069336,
      "step": 83608
    },
    {
      "epoch": 9.16290283203125e-06,
      "step": 83608,
      "training_step_time": 7.670393943786621
    },
    {
      "epoch": 9.163856506347656e-06,
      "model_forward_time": 1.8004422187805176,
      "step": 83609
    },
    {
      "epoch": 9.163856506347656e-06,
      "step": 83609,
      "training_step_time": 9.195286273956299
    },
    {
      "epoch": 9.164810180664062e-06,
      "grad_norm": 0.09690828621387482,
      "learning_rate": 7.167109271117662e-07,
      "loss": 0.0565,
      "step": 83610
    },
    {
      "epoch": 9.164810180664062e-06,
      "model_forward_time": 2.3968307971954346,
      "step": 83610
    },
    {
      "dynamics_loss_avg": 0.057299868948757646,
      "epoch": 9.164810180664062e-06,
      "step": 83610
    },
    {
      "action_loss_avg": 0.0013019271777011455,
      "epoch": 9.164810180664062e-06,
      "step": 83610
    },
    {
      "epoch": 9.164810180664062e-06,
      "step": 83610,
      "training_step_time": 7.582183837890625
    },
    {
      "epoch": 9.16576385498047e-06,
      "model_forward_time": 1.7749297618865967,
      "step": 83611
    },
    {
      "epoch": 9.16576385498047e-06,
      "step": 83611,
      "training_step_time": 7.5570387840271
    },
    {
      "epoch": 9.166717529296875e-06,
      "model_forward_time": 1.773979663848877,
      "step": 83612
    },
    {
      "epoch": 9.166717529296875e-06,
      "step": 83612,
      "training_step_time": 7.603544235229492
    },
    {
      "epoch": 9.16767120361328e-06,
      "model_forward_time": 1.773447036743164,
      "step": 83613
    },
    {
      "epoch": 9.16767120361328e-06,
      "step": 83613,
      "training_step_time": 7.630420207977295
    },
    {
      "epoch": 9.168624877929688e-06,
      "model_forward_time": 1.7765345573425293,
      "step": 83614
    },
    {
      "epoch": 9.168624877929688e-06,
      "step": 83614,
      "training_step_time": 7.6033172607421875
    },
    {
      "epoch": 9.169578552246094e-06,
      "model_forward_time": 1.771747350692749,
      "step": 83615
    },
    {
      "epoch": 9.169578552246094e-06,
      "step": 83615,
      "training_step_time": 7.666606187820435
    },
    {
      "epoch": 9.1705322265625e-06,
      "model_forward_time": 1.7751574516296387,
      "step": 83616
    },
    {
      "epoch": 9.1705322265625e-06,
      "step": 83616,
      "training_step_time": 8.756864309310913
    },
    {
      "epoch": 9.171485900878907e-06,
      "model_forward_time": 1.7730028629302979,
      "step": 83617
    },
    {
      "epoch": 9.171485900878907e-06,
      "step": 83617,
      "training_step_time": 8.083253860473633
    },
    {
      "epoch": 9.172439575195313e-06,
      "model_forward_time": 1.773702621459961,
      "step": 83618
    },
    {
      "epoch": 9.172439575195313e-06,
      "step": 83618,
      "training_step_time": 8.667209148406982
    },
    {
      "epoch": 9.173393249511719e-06,
      "model_forward_time": 2.4089033603668213,
      "step": 83619
    },
    {
      "epoch": 9.173393249511719e-06,
      "step": 83619,
      "training_step_time": 7.608628034591675
    },
    {
      "epoch": 9.174346923828124e-06,
      "grad_norm": 0.0689261257648468,
      "learning_rate": 7.158581609685073e-07,
      "loss": 0.0558,
      "step": 83620
    },
    {
      "epoch": 9.174346923828124e-06,
      "model_forward_time": 1.763537883758545,
      "step": 83620
    },
    {
      "dynamics_loss_avg": 0.03958845268934965,
      "epoch": 9.174346923828124e-06,
      "step": 83620
    },
    {
      "action_loss_avg": 0.001506145900930278,
      "epoch": 9.174346923828124e-06,
      "step": 83620
    },
    {
      "epoch": 9.174346923828124e-06,
      "step": 83620,
      "training_step_time": 7.9103288650512695
    },
    {
      "epoch": 9.175300598144532e-06,
      "model_forward_time": 1.7690448760986328,
      "step": 83621
    },
    {
      "epoch": 9.175300598144532e-06,
      "step": 83621,
      "training_step_time": 7.432405948638916
    },
    {
      "epoch": 9.176254272460937e-06,
      "model_forward_time": 1.7883470058441162,
      "step": 83622
    },
    {
      "epoch": 9.176254272460937e-06,
      "step": 83622,
      "training_step_time": 8.572357416152954
    },
    {
      "epoch": 9.177207946777343e-06,
      "model_forward_time": 1.7682061195373535,
      "step": 83623
    },
    {
      "epoch": 9.177207946777343e-06,
      "step": 83623,
      "training_step_time": 7.561058759689331
    },
    {
      "epoch": 9.17816162109375e-06,
      "model_forward_time": 1.7677679061889648,
      "step": 83624
    },
    {
      "epoch": 9.17816162109375e-06,
      "step": 83624,
      "training_step_time": 7.710334539413452
    },
    {
      "epoch": 9.179115295410156e-06,
      "model_forward_time": 1.7772927284240723,
      "step": 83625
    },
    {
      "epoch": 9.179115295410156e-06,
      "step": 83625,
      "training_step_time": 9.059436321258545
    },
    {
      "epoch": 9.180068969726562e-06,
      "model_forward_time": 1.7717680931091309,
      "step": 83626
    },
    {
      "epoch": 9.180068969726562e-06,
      "step": 83626,
      "training_step_time": 8.251548290252686
    },
    {
      "epoch": 9.18102264404297e-06,
      "model_forward_time": 1.7689707279205322,
      "step": 83627
    },
    {
      "epoch": 9.18102264404297e-06,
      "step": 83627,
      "training_step_time": 7.667299509048462
    },
    {
      "epoch": 9.181976318359375e-06,
      "model_forward_time": 2.4851648807525635,
      "step": 83628
    },
    {
      "epoch": 9.181976318359375e-06,
      "step": 83628,
      "training_step_time": 7.704937219619751
    },
    {
      "epoch": 9.18292999267578e-06,
      "model_forward_time": 1.7703766822814941,
      "step": 83629
    },
    {
      "epoch": 9.18292999267578e-06,
      "step": 83629,
      "training_step_time": 7.413762807846069
    },
    {
      "epoch": 9.183883666992188e-06,
      "grad_norm": 0.08486534655094147,
      "learning_rate": 7.15005863332559e-07,
      "loss": 0.0551,
      "step": 83630
    },
    {
      "epoch": 9.183883666992188e-06,
      "model_forward_time": 1.7724101543426514,
      "step": 83630
    },
    {
      "dynamics_loss_avg": 0.054880250617861746,
      "epoch": 9.183883666992188e-06,
      "step": 83630
    },
    {
      "action_loss_avg": 0.001847056479891762,
      "epoch": 9.183883666992188e-06,
      "step": 83630
    },
    {
      "epoch": 9.183883666992188e-06,
      "step": 83630,
      "training_step_time": 7.614912509918213
    },
    {
      "epoch": 9.184837341308594e-06,
      "model_forward_time": 1.7710483074188232,
      "step": 83631
    },
    {
      "epoch": 9.184837341308594e-06,
      "step": 83631,
      "training_step_time": 7.66420316696167
    },
    {
      "epoch": 9.185791015625e-06,
      "model_forward_time": 1.7783901691436768,
      "step": 83632
    },
    {
      "epoch": 9.185791015625e-06,
      "step": 83632,
      "training_step_time": 8.477815866470337
    },
    {
      "epoch": 9.186744689941405e-06,
      "model_forward_time": 1.7685062885284424,
      "step": 83633
    },
    {
      "epoch": 9.186744689941405e-06,
      "step": 83633,
      "training_step_time": 8.343962907791138
    },
    {
      "epoch": 9.187698364257813e-06,
      "model_forward_time": 1.7672474384307861,
      "step": 83634
    },
    {
      "epoch": 9.187698364257813e-06,
      "step": 83634,
      "training_step_time": 8.399385213851929
    },
    {
      "epoch": 9.188652038574219e-06,
      "model_forward_time": 1.765547752380371,
      "step": 83635
    },
    {
      "epoch": 9.188652038574219e-06,
      "step": 83635,
      "training_step_time": 7.7833240032196045
    },
    {
      "epoch": 9.189605712890624e-06,
      "model_forward_time": 1.764587163925171,
      "step": 83636
    },
    {
      "epoch": 9.189605712890624e-06,
      "step": 83636,
      "training_step_time": 8.81411337852478
    },
    {
      "epoch": 9.190559387207032e-06,
      "model_forward_time": 2.4642906188964844,
      "step": 83637
    },
    {
      "epoch": 9.190559387207032e-06,
      "step": 83637,
      "training_step_time": 7.651890516281128
    },
    {
      "epoch": 9.191513061523437e-06,
      "model_forward_time": 1.7812681198120117,
      "step": 83638
    },
    {
      "epoch": 9.191513061523437e-06,
      "step": 83638,
      "training_step_time": 7.496851921081543
    },
    {
      "epoch": 9.192466735839843e-06,
      "model_forward_time": 1.7675585746765137,
      "step": 83639
    },
    {
      "epoch": 9.192466735839843e-06,
      "step": 83639,
      "training_step_time": 7.671488046646118
    },
    {
      "epoch": 9.19342041015625e-06,
      "grad_norm": 0.10235685110092163,
      "learning_rate": 7.141540342971287e-07,
      "loss": 0.0557,
      "step": 83640
    },
    {
      "epoch": 9.19342041015625e-06,
      "model_forward_time": 1.762176752090454,
      "step": 83640
    },
    {
      "dynamics_loss_avg": 0.05191161129623652,
      "epoch": 9.19342041015625e-06,
      "step": 83640
    },
    {
      "action_loss_avg": 0.001147149666212499,
      "epoch": 9.19342041015625e-06,
      "step": 83640
    },
    {
      "epoch": 9.19342041015625e-06,
      "step": 83640,
      "training_step_time": 8.624251127243042
    },
    {
      "epoch": 9.194374084472656e-06,
      "model_forward_time": 1.76851487159729,
      "step": 83641
    },
    {
      "epoch": 9.194374084472656e-06,
      "step": 83641,
      "training_step_time": 7.743441104888916
    },
    {
      "epoch": 9.195327758789062e-06,
      "model_forward_time": 1.7710967063903809,
      "step": 83642
    },
    {
      "epoch": 9.195327758789062e-06,
      "step": 83642,
      "training_step_time": 8.741264820098877
    },
    {
      "epoch": 9.19628143310547e-06,
      "model_forward_time": 1.7770330905914307,
      "step": 83643
    },
    {
      "epoch": 9.19628143310547e-06,
      "step": 83643,
      "training_step_time": 7.671344995498657
    },
    {
      "epoch": 9.197235107421875e-06,
      "model_forward_time": 1.7735671997070312,
      "step": 83644
    },
    {
      "epoch": 9.197235107421875e-06,
      "step": 83644,
      "training_step_time": 9.3416268825531
    },
    {
      "epoch": 9.19818878173828e-06,
      "model_forward_time": 1.7656528949737549,
      "step": 83645
    },
    {
      "epoch": 9.19818878173828e-06,
      "step": 83645,
      "training_step_time": 9.399092197418213
    },
    {
      "epoch": 9.199142456054688e-06,
      "model_forward_time": 2.4411609172821045,
      "step": 83646
    },
    {
      "epoch": 9.199142456054688e-06,
      "step": 83646,
      "training_step_time": 7.759517431259155
    },
    {
      "epoch": 9.200096130371094e-06,
      "model_forward_time": 1.7702112197875977,
      "step": 83647
    },
    {
      "epoch": 9.200096130371094e-06,
      "step": 83647,
      "training_step_time": 8.297719955444336
    },
    {
      "epoch": 9.2010498046875e-06,
      "model_forward_time": 1.7524821758270264,
      "step": 83648
    },
    {
      "epoch": 9.2010498046875e-06,
      "step": 83648,
      "training_step_time": 8.547905445098877
    },
    {
      "epoch": 9.202003479003905e-06,
      "model_forward_time": 1.7467126846313477,
      "step": 83649
    },
    {
      "epoch": 9.202003479003905e-06,
      "step": 83649,
      "training_step_time": 7.727254152297974
    },
    {
      "epoch": 9.202957153320313e-06,
      "grad_norm": 0.10575710982084274,
      "learning_rate": 7.133026739553678e-07,
      "loss": 0.0563,
      "step": 83650
    },
    {
      "epoch": 9.202957153320313e-06,
      "model_forward_time": 1.7676432132720947,
      "step": 83650
    },
    {
      "dynamics_loss_avg": 0.06819749735295773,
      "epoch": 9.202957153320313e-06,
      "step": 83650
    },
    {
      "action_loss_avg": 0.0012437232478987425,
      "epoch": 9.202957153320313e-06,
      "step": 83650
    },
    {
      "epoch": 9.202957153320313e-06,
      "step": 83650,
      "training_step_time": 7.705199241638184
    },
    {
      "epoch": 9.203910827636719e-06,
      "model_forward_time": 1.7507526874542236,
      "step": 83651
    },
    {
      "epoch": 9.203910827636719e-06,
      "step": 83651,
      "training_step_time": 7.644857883453369
    },
    {
      "epoch": 9.204864501953124e-06,
      "model_forward_time": 1.8425428867340088,
      "step": 83652
    },
    {
      "epoch": 9.204864501953124e-06,
      "step": 83652,
      "training_step_time": 7.731548070907593
    },
    {
      "epoch": 9.205818176269532e-06,
      "model_forward_time": 1.7596321105957031,
      "step": 83653
    },
    {
      "epoch": 9.205818176269532e-06,
      "step": 83653,
      "training_step_time": 10.285499334335327
    },
    {
      "epoch": 9.206771850585937e-06,
      "model_forward_time": 1.7705280780792236,
      "step": 83654
    },
    {
      "epoch": 9.206771850585937e-06,
      "step": 83654,
      "training_step_time": 8.230501651763916
    },
    {
      "epoch": 9.207725524902343e-06,
      "model_forward_time": 1.757340669631958,
      "step": 83655
    },
    {
      "epoch": 9.207725524902343e-06,
      "step": 83655,
      "training_step_time": 8.66746711730957
    },
    {
      "epoch": 9.20867919921875e-06,
      "model_forward_time": 2.8560357093811035,
      "step": 83656
    },
    {
      "epoch": 9.20867919921875e-06,
      "step": 83656,
      "training_step_time": 7.574845552444458
    },
    {
      "epoch": 9.209632873535156e-06,
      "model_forward_time": 1.7537589073181152,
      "step": 83657
    },
    {
      "epoch": 9.209632873535156e-06,
      "step": 83657,
      "training_step_time": 7.597918272018433
    },
    {
      "epoch": 9.210586547851562e-06,
      "model_forward_time": 1.7617487907409668,
      "step": 83658
    },
    {
      "epoch": 9.210586547851562e-06,
      "step": 83658,
      "training_step_time": 7.65504264831543
    },
    {
      "epoch": 9.21154022216797e-06,
      "model_forward_time": 1.7572059631347656,
      "step": 83659
    },
    {
      "epoch": 9.21154022216797e-06,
      "step": 83659,
      "training_step_time": 7.640213251113892
    },
    {
      "epoch": 9.212493896484375e-06,
      "grad_norm": 0.13098618388175964,
      "learning_rate": 7.124517824003829e-07,
      "loss": 0.0546,
      "step": 83660
    },
    {
      "epoch": 9.212493896484375e-06,
      "model_forward_time": 1.760803461074829,
      "step": 83660
    },
    {
      "dynamics_loss_avg": 0.049873193725943565,
      "epoch": 9.212493896484375e-06,
      "step": 83660
    },
    {
      "action_loss_avg": 0.0014207730841008016,
      "epoch": 9.212493896484375e-06,
      "step": 83660
    },
    {
      "epoch": 9.212493896484375e-06,
      "step": 83660,
      "training_step_time": 9.607292175292969
    },
    {
      "epoch": 9.213447570800781e-06,
      "model_forward_time": 1.823176383972168,
      "step": 83661
    },
    {
      "epoch": 9.213447570800781e-06,
      "step": 83661,
      "training_step_time": 8.831037998199463
    },
    {
      "epoch": 9.214401245117188e-06,
      "model_forward_time": 1.76652193069458,
      "step": 83662
    },
    {
      "epoch": 9.214401245117188e-06,
      "step": 83662,
      "training_step_time": 8.39563512802124
    },
    {
      "epoch": 9.215354919433594e-06,
      "model_forward_time": 1.7725675106048584,
      "step": 83663
    },
    {
      "epoch": 9.215354919433594e-06,
      "step": 83663,
      "training_step_time": 8.184572219848633
    },
    {
      "epoch": 9.21630859375e-06,
      "model_forward_time": 1.7620766162872314,
      "step": 83664
    },
    {
      "epoch": 9.21630859375e-06,
      "step": 83664,
      "training_step_time": 7.707848072052002
    },
    {
      "epoch": 9.217262268066405e-06,
      "model_forward_time": 2.8479766845703125,
      "step": 83665
    },
    {
      "epoch": 9.217262268066405e-06,
      "step": 83665,
      "training_step_time": 7.814800977706909
    },
    {
      "epoch": 9.218215942382813e-06,
      "model_forward_time": 1.7729556560516357,
      "step": 83666
    },
    {
      "epoch": 9.218215942382813e-06,
      "step": 83666,
      "training_step_time": 7.583562135696411
    },
    {
      "epoch": 9.219169616699219e-06,
      "model_forward_time": 1.7663359642028809,
      "step": 83667
    },
    {
      "epoch": 9.219169616699219e-06,
      "step": 83667,
      "training_step_time": 7.48248553276062
    },
    {
      "epoch": 9.220123291015624e-06,
      "model_forward_time": 1.763401985168457,
      "step": 83668
    },
    {
      "epoch": 9.220123291015624e-06,
      "step": 83668,
      "training_step_time": 8.518162250518799
    },
    {
      "epoch": 9.221076965332032e-06,
      "model_forward_time": 1.770505666732788,
      "step": 83669
    },
    {
      "epoch": 9.221076965332032e-06,
      "step": 83669,
      "training_step_time": 7.641010046005249
    },
    {
      "epoch": 9.222030639648437e-06,
      "grad_norm": 0.07978447526693344,
      "learning_rate": 7.116013597252236e-07,
      "loss": 0.0537,
      "step": 83670
    },
    {
      "epoch": 9.222030639648437e-06,
      "model_forward_time": 1.7989306449890137,
      "step": 83670
    },
    {
      "dynamics_loss_avg": 0.05781475063413381,
      "epoch": 9.222030639648437e-06,
      "step": 83670
    },
    {
      "action_loss_avg": 0.0017508075514342635,
      "epoch": 9.222030639648437e-06,
      "step": 83670
    },
    {
      "epoch": 9.222030639648437e-06,
      "step": 83670,
      "training_step_time": 8.797140121459961
    },
    {
      "epoch": 9.222984313964843e-06,
      "model_forward_time": 1.7625362873077393,
      "step": 83671
    },
    {
      "epoch": 9.222984313964843e-06,
      "step": 83671,
      "training_step_time": 8.418992757797241
    },
    {
      "epoch": 9.22393798828125e-06,
      "model_forward_time": 1.767503261566162,
      "step": 83672
    },
    {
      "epoch": 9.22393798828125e-06,
      "step": 83672,
      "training_step_time": 9.055496215820312
    },
    {
      "epoch": 9.224891662597656e-06,
      "model_forward_time": 1.7618277072906494,
      "step": 83673
    },
    {
      "epoch": 9.224891662597656e-06,
      "step": 83673,
      "training_step_time": 7.693576812744141
    },
    {
      "epoch": 9.225845336914062e-06,
      "model_forward_time": 2.837196111679077,
      "step": 83674
    },
    {
      "epoch": 9.225845336914062e-06,
      "step": 83674,
      "training_step_time": 7.740251541137695
    },
    {
      "epoch": 9.22679901123047e-06,
      "model_forward_time": 1.7663660049438477,
      "step": 83675
    },
    {
      "epoch": 9.22679901123047e-06,
      "step": 83675,
      "training_step_time": 7.615471124649048
    },
    {
      "epoch": 9.227752685546875e-06,
      "model_forward_time": 1.7626838684082031,
      "step": 83676
    },
    {
      "epoch": 9.227752685546875e-06,
      "step": 83676,
      "training_step_time": 7.546729326248169
    },
    {
      "epoch": 9.228706359863281e-06,
      "model_forward_time": 1.755615472793579,
      "step": 83677
    },
    {
      "epoch": 9.228706359863281e-06,
      "step": 83677,
      "training_step_time": 8.633527994155884
    },
    {
      "epoch": 9.229660034179688e-06,
      "model_forward_time": 1.7673649787902832,
      "step": 83678
    },
    {
      "epoch": 9.229660034179688e-06,
      "step": 83678,
      "training_step_time": 7.6406731605529785
    },
    {
      "epoch": 9.230613708496094e-06,
      "model_forward_time": 1.784611701965332,
      "step": 83679
    },
    {
      "epoch": 9.230613708496094e-06,
      "step": 83679,
      "training_step_time": 7.826866626739502
    },
    {
      "epoch": 9.2315673828125e-06,
      "grad_norm": 0.08569850772619247,
      "learning_rate": 7.107514060228921e-07,
      "loss": 0.0548,
      "step": 83680
    },
    {
      "epoch": 9.2315673828125e-06,
      "model_forward_time": 1.789036512374878,
      "step": 83680
    },
    {
      "dynamics_loss_avg": 0.07150869220495223,
      "epoch": 9.2315673828125e-06,
      "step": 83680
    },
    {
      "action_loss_avg": 0.0017615081567782908,
      "epoch": 9.2315673828125e-06,
      "step": 83680
    },
    {
      "epoch": 9.2315673828125e-06,
      "step": 83680,
      "training_step_time": 8.512603521347046
    },
    {
      "epoch": 9.232521057128905e-06,
      "model_forward_time": 1.7761099338531494,
      "step": 83681
    },
    {
      "epoch": 9.232521057128905e-06,
      "step": 83681,
      "training_step_time": 7.768579959869385
    },
    {
      "epoch": 9.233474731445313e-06,
      "model_forward_time": 1.766758918762207,
      "step": 83682
    },
    {
      "epoch": 9.233474731445313e-06,
      "step": 83682,
      "training_step_time": 7.717900514602661
    },
    {
      "epoch": 9.234428405761719e-06,
      "model_forward_time": 2.85654354095459,
      "step": 83683
    },
    {
      "epoch": 9.234428405761719e-06,
      "step": 83683,
      "training_step_time": 7.680737257003784
    },
    {
      "epoch": 9.235382080078124e-06,
      "model_forward_time": 1.7631473541259766,
      "step": 83684
    },
    {
      "epoch": 9.235382080078124e-06,
      "step": 83684,
      "training_step_time": 8.352980136871338
    },
    {
      "epoch": 9.236335754394532e-06,
      "model_forward_time": 1.765169382095337,
      "step": 83685
    },
    {
      "epoch": 9.236335754394532e-06,
      "step": 83685,
      "training_step_time": 8.504162788391113
    },
    {
      "epoch": 9.237289428710937e-06,
      "model_forward_time": 1.765758752822876,
      "step": 83686
    },
    {
      "epoch": 9.237289428710937e-06,
      "step": 83686,
      "training_step_time": 7.658125162124634
    },
    {
      "epoch": 9.238243103027343e-06,
      "model_forward_time": 1.7677500247955322,
      "step": 83687
    },
    {
      "epoch": 9.238243103027343e-06,
      "step": 83687,
      "training_step_time": 7.654708385467529
    },
    {
      "epoch": 9.23919677734375e-06,
      "model_forward_time": 1.7658498287200928,
      "step": 83688
    },
    {
      "epoch": 9.23919677734375e-06,
      "step": 83688,
      "training_step_time": 8.294815301895142
    },
    {
      "epoch": 9.240150451660156e-06,
      "model_forward_time": 1.8517591953277588,
      "step": 83689
    },
    {
      "epoch": 9.240150451660156e-06,
      "step": 83689,
      "training_step_time": 8.347736835479736
    },
    {
      "epoch": 9.241104125976562e-06,
      "grad_norm": 0.08939105272293091,
      "learning_rate": 7.099019213863378e-07,
      "loss": 0.0541,
      "step": 83690
    },
    {
      "epoch": 9.241104125976562e-06,
      "model_forward_time": 1.7792670726776123,
      "step": 83690
    },
    {
      "dynamics_loss_avg": 0.05430017411708832,
      "epoch": 9.241104125976562e-06,
      "step": 83690
    },
    {
      "action_loss_avg": 0.002111969323595986,
      "epoch": 9.241104125976562e-06,
      "step": 83690
    },
    {
      "epoch": 9.241104125976562e-06,
      "step": 83690,
      "training_step_time": 7.864443063735962
    },
    {
      "epoch": 9.24205780029297e-06,
      "model_forward_time": 1.7656683921813965,
      "step": 83691
    },
    {
      "epoch": 9.24205780029297e-06,
      "step": 83691,
      "training_step_time": 8.60456657409668
    },
    {
      "epoch": 9.243011474609375e-06,
      "model_forward_time": 2.8151307106018066,
      "step": 83692
    },
    {
      "epoch": 9.243011474609375e-06,
      "step": 83692,
      "training_step_time": 7.632461309432983
    },
    {
      "epoch": 9.243965148925781e-06,
      "model_forward_time": 1.7719168663024902,
      "step": 83693
    },
    {
      "epoch": 9.243965148925781e-06,
      "step": 83693,
      "training_step_time": 7.757330656051636
    },
    {
      "epoch": 9.244918823242188e-06,
      "model_forward_time": 1.7705650329589844,
      "step": 83694
    },
    {
      "epoch": 9.244918823242188e-06,
      "step": 83694,
      "training_step_time": 7.6392786502838135
    },
    {
      "epoch": 9.245872497558594e-06,
      "model_forward_time": 1.768563985824585,
      "step": 83695
    },
    {
      "epoch": 9.245872497558594e-06,
      "step": 83695,
      "training_step_time": 7.703617095947266
    },
    {
      "epoch": 9.246826171875e-06,
      "model_forward_time": 1.7834019660949707,
      "step": 83696
    },
    {
      "epoch": 9.246826171875e-06,
      "step": 83696,
      "training_step_time": 7.670565128326416
    },
    {
      "epoch": 9.247779846191405e-06,
      "model_forward_time": 1.7648344039916992,
      "step": 83697
    },
    {
      "epoch": 9.247779846191405e-06,
      "step": 83697,
      "training_step_time": 8.698787450790405
    },
    {
      "epoch": 9.248733520507813e-06,
      "model_forward_time": 1.8033394813537598,
      "step": 83698
    },
    {
      "epoch": 9.248733520507813e-06,
      "step": 83698,
      "training_step_time": 7.677682161331177
    },
    {
      "epoch": 9.249687194824219e-06,
      "model_forward_time": 1.7601141929626465,
      "step": 83699
    },
    {
      "epoch": 9.249687194824219e-06,
      "step": 83699,
      "training_step_time": 8.584535837173462
    },
    {
      "epoch": 9.250640869140624e-06,
      "grad_norm": 0.07326894998550415,
      "learning_rate": 7.090529059084611e-07,
      "loss": 0.0554,
      "step": 83700
    },
    {
      "epoch": 9.250640869140624e-06,
      "model_forward_time": 1.789635419845581,
      "step": 83700
    },
    {
      "dynamics_loss_avg": 0.04813825152814388,
      "epoch": 9.250640869140624e-06,
      "step": 83700
    },
    {
      "action_loss_avg": 0.000954291460220702,
      "epoch": 9.250640869140624e-06,
      "step": 83700
    },
    {
      "epoch": 9.250640869140624e-06,
      "step": 83700,
      "training_step_time": 7.692806959152222
    },
    {
      "epoch": 9.251594543457032e-06,
      "model_forward_time": 1.4282469749450684,
      "step": 83701
    },
    {
      "epoch": 9.251594543457032e-06,
      "step": 83701,
      "training_step_time": 8.33869194984436
    },
    {
      "epoch": 9.252548217773437e-06,
      "model_forward_time": 2.8836352825164795,
      "step": 83702
    },
    {
      "epoch": 9.252548217773437e-06,
      "step": 83702,
      "training_step_time": 8.457731246948242
    },
    {
      "epoch": 9.253501892089843e-06,
      "model_forward_time": 1.775974988937378,
      "step": 83703
    },
    {
      "epoch": 9.253501892089843e-06,
      "step": 83703,
      "training_step_time": 8.746095657348633
    },
    {
      "epoch": 9.25445556640625e-06,
      "model_forward_time": 1.832859754562378,
      "step": 83704
    },
    {
      "epoch": 9.25445556640625e-06,
      "step": 83704,
      "training_step_time": 7.268815755844116
    },
    {
      "epoch": 9.255409240722656e-06,
      "model_forward_time": 1.7785115242004395,
      "step": 83705
    },
    {
      "epoch": 9.255409240722656e-06,
      "step": 83705,
      "training_step_time": 9.461150407791138
    },
    {
      "epoch": 9.256362915039062e-06,
      "model_forward_time": 1.764371395111084,
      "step": 83706
    },
    {
      "epoch": 9.256362915039062e-06,
      "step": 83706,
      "training_step_time": 9.958044052124023
    },
    {
      "epoch": 9.25731658935547e-06,
      "model_forward_time": 1.769970417022705,
      "step": 83707
    },
    {
      "epoch": 9.25731658935547e-06,
      "step": 83707,
      "training_step_time": 8.76048755645752
    },
    {
      "epoch": 9.258270263671875e-06,
      "model_forward_time": 1.7655034065246582,
      "step": 83708
    },
    {
      "epoch": 9.258270263671875e-06,
      "step": 83708,
      "training_step_time": 8.974734544754028
    },
    {
      "epoch": 9.259223937988281e-06,
      "model_forward_time": 1.7592840194702148,
      "step": 83709
    },
    {
      "epoch": 9.259223937988281e-06,
      "step": 83709,
      "training_step_time": 9.720921039581299
    },
    {
      "epoch": 9.260177612304688e-06,
      "grad_norm": 0.1847761869430542,
      "learning_rate": 7.082043596821059e-07,
      "loss": 0.0553,
      "step": 83710
    },
    {
      "epoch": 9.260177612304688e-06,
      "model_forward_time": 1.7683160305023193,
      "step": 83710
    },
    {
      "dynamics_loss_avg": 0.06832711584866047,
      "epoch": 9.260177612304688e-06,
      "step": 83710
    },
    {
      "action_loss_avg": 0.0017776596127077938,
      "epoch": 9.260177612304688e-06,
      "step": 83710
    },
    {
      "epoch": 9.260177612304688e-06,
      "step": 83710,
      "training_step_time": 7.68477988243103
    },
    {
      "epoch": 9.261131286621094e-06,
      "model_forward_time": 2.397122621536255,
      "step": 83711
    },
    {
      "epoch": 9.261131286621094e-06,
      "step": 83711,
      "training_step_time": 8.66035771369934
    },
    {
      "epoch": 9.2620849609375e-06,
      "model_forward_time": 1.822293996810913,
      "step": 83712
    },
    {
      "epoch": 9.2620849609375e-06,
      "step": 83712,
      "training_step_time": 8.494489192962646
    },
    {
      "epoch": 9.263038635253905e-06,
      "model_forward_time": 1.7663569450378418,
      "step": 83713
    },
    {
      "epoch": 9.263038635253905e-06,
      "step": 83713,
      "training_step_time": 7.542835235595703
    },
    {
      "epoch": 9.263992309570313e-06,
      "model_forward_time": 1.7602460384368896,
      "step": 83714
    },
    {
      "epoch": 9.263992309570313e-06,
      "step": 83714,
      "training_step_time": 7.8112382888793945
    },
    {
      "epoch": 9.264945983886719e-06,
      "model_forward_time": 1.7573955059051514,
      "step": 83715
    },
    {
      "epoch": 9.264945983886719e-06,
      "step": 83715,
      "training_step_time": 7.766701698303223
    },
    {
      "epoch": 9.265899658203124e-06,
      "model_forward_time": 1.7597393989562988,
      "step": 83716
    },
    {
      "epoch": 9.265899658203124e-06,
      "step": 83716,
      "training_step_time": 9.422391414642334
    },
    {
      "epoch": 9.266853332519532e-06,
      "model_forward_time": 1.7585937976837158,
      "step": 83717
    },
    {
      "epoch": 9.266853332519532e-06,
      "step": 83717,
      "training_step_time": 8.897073984146118
    },
    {
      "epoch": 9.267807006835937e-06,
      "model_forward_time": 1.7504363059997559,
      "step": 83718
    },
    {
      "epoch": 9.267807006835937e-06,
      "step": 83718,
      "training_step_time": 8.364298105239868
    },
    {
      "epoch": 9.268760681152343e-06,
      "model_forward_time": 1.754225492477417,
      "step": 83719
    },
    {
      "epoch": 9.268760681152343e-06,
      "step": 83719,
      "training_step_time": 7.7932610511779785
    },
    {
      "epoch": 9.26971435546875e-06,
      "grad_norm": 0.1345939338207245,
      "learning_rate": 7.073562828000691e-07,
      "loss": 0.0538,
      "step": 83720
    },
    {
      "epoch": 9.26971435546875e-06,
      "model_forward_time": 2.3875479698181152,
      "step": 83720
    },
    {
      "dynamics_loss_avg": 0.04181386977434158,
      "epoch": 9.26971435546875e-06,
      "step": 83720
    },
    {
      "action_loss_avg": 0.0009622025943826884,
      "epoch": 9.26971435546875e-06,
      "step": 83720
    },
    {
      "epoch": 9.26971435546875e-06,
      "step": 83720,
      "training_step_time": 7.862716197967529
    },
    {
      "epoch": 9.270668029785156e-06,
      "model_forward_time": 1.800330400466919,
      "step": 83721
    },
    {
      "epoch": 9.270668029785156e-06,
      "step": 83721,
      "training_step_time": 7.76821231842041
    },
    {
      "epoch": 9.271621704101562e-06,
      "model_forward_time": 1.7610280513763428,
      "step": 83722
    },
    {
      "epoch": 9.271621704101562e-06,
      "step": 83722,
      "training_step_time": 8.414837121963501
    },
    {
      "epoch": 9.27257537841797e-06,
      "model_forward_time": 1.7606794834136963,
      "step": 83723
    },
    {
      "epoch": 9.27257537841797e-06,
      "step": 83723,
      "training_step_time": 9.754725933074951
    },
    {
      "epoch": 9.273529052734375e-06,
      "model_forward_time": 1.7525644302368164,
      "step": 83724
    },
    {
      "epoch": 9.273529052734375e-06,
      "step": 83724,
      "training_step_time": 7.689676523208618
    },
    {
      "epoch": 9.274482727050781e-06,
      "model_forward_time": 1.7544841766357422,
      "step": 83725
    },
    {
      "epoch": 9.274482727050781e-06,
      "step": 83725,
      "training_step_time": 7.827553987503052
    },
    {
      "epoch": 9.275436401367188e-06,
      "model_forward_time": 1.7446153163909912,
      "step": 83726
    },
    {
      "epoch": 9.275436401367188e-06,
      "step": 83726,
      "training_step_time": 9.86731243133545
    },
    {
      "epoch": 9.276390075683594e-06,
      "model_forward_time": 1.758202314376831,
      "step": 83727
    },
    {
      "epoch": 9.276390075683594e-06,
      "step": 83727,
      "training_step_time": 7.8059563636779785
    },
    {
      "epoch": 9.27734375e-06,
      "model_forward_time": 1.029242992401123,
      "step": 83728
    },
    {
      "epoch": 9.27734375e-06,
      "step": 83728,
      "training_step_time": 7.7347798347473145
    },
    {
      "epoch": 9.278297424316405e-06,
      "model_forward_time": 1.7694528102874756,
      "step": 83729
    },
    {
      "epoch": 9.278297424316405e-06,
      "step": 83729,
      "training_step_time": 8.10372257232666
    },
    {
      "epoch": 9.279251098632813e-06,
      "grad_norm": 0.09342369437217712,
      "learning_rate": 7.065086753550965e-07,
      "loss": 0.0536,
      "step": 83730
    },
    {
      "epoch": 9.279251098632813e-06,
      "model_forward_time": 1.762472152709961,
      "step": 83730
    },
    {
      "dynamics_loss_avg": 0.04918626062572003,
      "epoch": 9.279251098632813e-06,
      "step": 83730
    },
    {
      "action_loss_avg": 0.0012493722140789032,
      "epoch": 9.279251098632813e-06,
      "step": 83730
    },
    {
      "epoch": 9.279251098632813e-06,
      "step": 83730,
      "training_step_time": 7.849549770355225
    },
    {
      "epoch": 9.280204772949219e-06,
      "model_forward_time": 1.7515814304351807,
      "step": 83731
    },
    {
      "epoch": 9.280204772949219e-06,
      "step": 83731,
      "training_step_time": 7.630483627319336
    },
    {
      "epoch": 9.281158447265624e-06,
      "model_forward_time": 1.7577614784240723,
      "step": 83732
    },
    {
      "epoch": 9.281158447265624e-06,
      "step": 83732,
      "training_step_time": 7.630548000335693
    },
    {
      "epoch": 9.282112121582032e-06,
      "model_forward_time": 1.7603752613067627,
      "step": 83733
    },
    {
      "epoch": 9.282112121582032e-06,
      "step": 83733,
      "training_step_time": 7.695258855819702
    },
    {
      "epoch": 9.283065795898437e-06,
      "model_forward_time": 1.7593708038330078,
      "step": 83734
    },
    {
      "epoch": 9.283065795898437e-06,
      "step": 83734,
      "training_step_time": 9.389838457107544
    },
    {
      "epoch": 9.284019470214843e-06,
      "model_forward_time": 1.766057014465332,
      "step": 83735
    },
    {
      "epoch": 9.284019470214843e-06,
      "step": 83735,
      "training_step_time": 8.252686500549316
    },
    {
      "epoch": 9.28497314453125e-06,
      "model_forward_time": 1.7606637477874756,
      "step": 83736
    },
    {
      "epoch": 9.28497314453125e-06,
      "step": 83736,
      "training_step_time": 9.09910273551941
    },
    {
      "epoch": 9.285926818847656e-06,
      "model_forward_time": 1.7481398582458496,
      "step": 83737
    },
    {
      "epoch": 9.285926818847656e-06,
      "step": 83737,
      "training_step_time": 8.336590051651001
    },
    {
      "epoch": 9.286880493164062e-06,
      "model_forward_time": 1.7603435516357422,
      "step": 83738
    },
    {
      "epoch": 9.286880493164062e-06,
      "step": 83738,
      "training_step_time": 7.594581842422485
    },
    {
      "epoch": 9.28783416748047e-06,
      "model_forward_time": 2.3908817768096924,
      "step": 83739
    },
    {
      "epoch": 9.28783416748047e-06,
      "step": 83739,
      "training_step_time": 7.807328462600708
    },
    {
      "epoch": 9.288787841796875e-06,
      "grad_norm": 0.10702326893806458,
      "learning_rate": 7.056615374398789e-07,
      "loss": 0.0547,
      "step": 83740
    },
    {
      "epoch": 9.288787841796875e-06,
      "model_forward_time": 1.8597047328948975,
      "step": 83740
    },
    {
      "dynamics_loss_avg": 0.05846252366900444,
      "epoch": 9.288787841796875e-06,
      "step": 83740
    },
    {
      "action_loss_avg": 0.0019195589644368737,
      "epoch": 9.288787841796875e-06,
      "step": 83740
    },
    {
      "epoch": 9.288787841796875e-06,
      "step": 83740,
      "training_step_time": 7.637054920196533
    },
    {
      "epoch": 9.289741516113281e-06,
      "model_forward_time": 1.7730941772460938,
      "step": 83741
    },
    {
      "epoch": 9.289741516113281e-06,
      "step": 83741,
      "training_step_time": 7.595764875411987
    },
    {
      "epoch": 9.290695190429688e-06,
      "model_forward_time": 1.7623012065887451,
      "step": 83742
    },
    {
      "epoch": 9.290695190429688e-06,
      "step": 83742,
      "training_step_time": 7.715565204620361
    },
    {
      "epoch": 9.291648864746094e-06,
      "model_forward_time": 1.7527544498443604,
      "step": 83743
    },
    {
      "epoch": 9.291648864746094e-06,
      "step": 83743,
      "training_step_time": 7.812734842300415
    },
    {
      "epoch": 9.2926025390625e-06,
      "model_forward_time": 1.761465072631836,
      "step": 83744
    },
    {
      "epoch": 9.2926025390625e-06,
      "step": 83744,
      "training_step_time": 8.368110179901123
    },
    {
      "epoch": 9.293556213378906e-06,
      "model_forward_time": 1.7608129978179932,
      "step": 83745
    },
    {
      "epoch": 9.293556213378906e-06,
      "step": 83745,
      "training_step_time": 10.217798233032227
    },
    {
      "epoch": 9.294509887695313e-06,
      "model_forward_time": 1.769923210144043,
      "step": 83746
    },
    {
      "epoch": 9.294509887695313e-06,
      "step": 83746,
      "training_step_time": 8.327914714813232
    },
    {
      "epoch": 9.295463562011719e-06,
      "model_forward_time": 1.759756326675415,
      "step": 83747
    },
    {
      "epoch": 9.295463562011719e-06,
      "step": 83747,
      "training_step_time": 8.200337648391724
    },
    {
      "epoch": 9.296417236328124e-06,
      "model_forward_time": 2.394216537475586,
      "step": 83748
    },
    {
      "epoch": 9.296417236328124e-06,
      "step": 83748,
      "training_step_time": 7.843537330627441
    },
    {
      "epoch": 9.297370910644532e-06,
      "model_forward_time": 1.8892815113067627,
      "step": 83749
    },
    {
      "epoch": 9.297370910644532e-06,
      "step": 83749,
      "training_step_time": 7.7276129722595215
    },
    {
      "epoch": 9.298324584960938e-06,
      "grad_norm": 0.08128724247217178,
      "learning_rate": 7.048148691470591e-07,
      "loss": 0.0539,
      "step": 83750
    },
    {
      "epoch": 9.298324584960938e-06,
      "model_forward_time": 1.7498080730438232,
      "step": 83750
    },
    {
      "dynamics_loss_avg": 0.06156795136630535,
      "epoch": 9.298324584960938e-06,
      "step": 83750
    },
    {
      "action_loss_avg": 0.0015446556659298948,
      "epoch": 9.298324584960938e-06,
      "step": 83750
    },
    {
      "epoch": 9.298324584960938e-06,
      "step": 83750,
      "training_step_time": 7.600845813751221
    },
    {
      "epoch": 9.299278259277343e-06,
      "model_forward_time": 1.7573068141937256,
      "step": 83751
    },
    {
      "epoch": 9.299278259277343e-06,
      "step": 83751,
      "training_step_time": 7.748686075210571
    },
    {
      "epoch": 9.30023193359375e-06,
      "model_forward_time": 1.7565891742706299,
      "step": 83752
    },
    {
      "epoch": 9.30023193359375e-06,
      "step": 83752,
      "training_step_time": 8.74940800666809
    },
    {
      "epoch": 9.301185607910156e-06,
      "model_forward_time": 1.754725456237793,
      "step": 83753
    },
    {
      "epoch": 9.301185607910156e-06,
      "step": 83753,
      "training_step_time": 8.70423698425293
    },
    {
      "epoch": 9.302139282226562e-06,
      "model_forward_time": 1.76405930519104,
      "step": 83754
    },
    {
      "epoch": 9.302139282226562e-06,
      "step": 83754,
      "training_step_time": 9.364377737045288
    },
    {
      "epoch": 9.30309295654297e-06,
      "model_forward_time": 1.7599496841430664,
      "step": 83755
    },
    {
      "epoch": 9.30309295654297e-06,
      "step": 83755,
      "training_step_time": 7.7819504737854
    },
    {
      "epoch": 9.304046630859375e-06,
      "model_forward_time": 1.7577404975891113,
      "step": 83756
    },
    {
      "epoch": 9.304046630859375e-06,
      "step": 83756,
      "training_step_time": 8.731393098831177
    },
    {
      "epoch": 9.305000305175781e-06,
      "model_forward_time": 2.4094488620758057,
      "step": 83757
    },
    {
      "epoch": 9.305000305175781e-06,
      "step": 83757,
      "training_step_time": 7.713102340698242
    },
    {
      "epoch": 9.305953979492187e-06,
      "model_forward_time": 1.8527257442474365,
      "step": 83758
    },
    {
      "epoch": 9.305953979492187e-06,
      "step": 83758,
      "training_step_time": 7.726974725723267
    },
    {
      "epoch": 9.306907653808594e-06,
      "model_forward_time": 1.7316322326660156,
      "step": 83759
    },
    {
      "epoch": 9.306907653808594e-06,
      "step": 83759,
      "training_step_time": 7.643711805343628
    },
    {
      "epoch": 9.307861328125e-06,
      "grad_norm": 0.07769947499036789,
      "learning_rate": 7.039686705692272e-07,
      "loss": 0.0544,
      "step": 83760
    },
    {
      "epoch": 9.307861328125e-06,
      "model_forward_time": 1.7499158382415771,
      "step": 83760
    },
    {
      "dynamics_loss_avg": 0.07045597787946463,
      "epoch": 9.307861328125e-06,
      "step": 83760
    },
    {
      "action_loss_avg": 0.0013162436953280121,
      "epoch": 9.307861328125e-06,
      "step": 83760
    },
    {
      "epoch": 9.307861328125e-06,
      "step": 83760,
      "training_step_time": 7.80125093460083
    },
    {
      "epoch": 9.308815002441406e-06,
      "model_forward_time": 1.7707293033599854,
      "step": 83761
    },
    {
      "epoch": 9.308815002441406e-06,
      "step": 83761,
      "training_step_time": 7.747308969497681
    },
    {
      "epoch": 9.309768676757813e-06,
      "model_forward_time": 1.7591447830200195,
      "step": 83762
    },
    {
      "epoch": 9.309768676757813e-06,
      "step": 83762,
      "training_step_time": 7.791701793670654
    },
    {
      "epoch": 9.310722351074219e-06,
      "model_forward_time": 1.7552673816680908,
      "step": 83763
    },
    {
      "epoch": 9.310722351074219e-06,
      "step": 83763,
      "training_step_time": 8.380799055099487
    },
    {
      "epoch": 9.311676025390624e-06,
      "model_forward_time": 1.7497668266296387,
      "step": 83764
    },
    {
      "epoch": 9.311676025390624e-06,
      "step": 83764,
      "training_step_time": 8.22530746459961
    },
    {
      "epoch": 9.312629699707032e-06,
      "model_forward_time": 1.7591512203216553,
      "step": 83765
    },
    {
      "epoch": 9.312629699707032e-06,
      "step": 83765,
      "training_step_time": 8.307754755020142
    },
    {
      "epoch": 9.313583374023438e-06,
      "model_forward_time": 2.4259276390075684,
      "step": 83766
    },
    {
      "epoch": 9.313583374023438e-06,
      "step": 83766,
      "training_step_time": 7.568252325057983
    },
    {
      "epoch": 9.314537048339843e-06,
      "model_forward_time": 1.8157997131347656,
      "step": 83767
    },
    {
      "epoch": 9.314537048339843e-06,
      "step": 83767,
      "training_step_time": 7.671263694763184
    },
    {
      "epoch": 9.31549072265625e-06,
      "model_forward_time": 1.7563042640686035,
      "step": 83768
    },
    {
      "epoch": 9.31549072265625e-06,
      "step": 83768,
      "training_step_time": 7.6067893505096436
    },
    {
      "epoch": 9.316444396972656e-06,
      "model_forward_time": 1.7538011074066162,
      "step": 83769
    },
    {
      "epoch": 9.316444396972656e-06,
      "step": 83769,
      "training_step_time": 7.575923681259155
    },
    {
      "epoch": 9.317398071289062e-06,
      "grad_norm": 0.07424025237560272,
      "learning_rate": 7.031229417989244e-07,
      "loss": 0.0544,
      "step": 83770
    },
    {
      "epoch": 9.317398071289062e-06,
      "model_forward_time": 1.7574717998504639,
      "step": 83770
    },
    {
      "dynamics_loss_avg": 0.060808098316192626,
      "epoch": 9.317398071289062e-06,
      "step": 83770
    },
    {
      "action_loss_avg": 0.0019345743960002438,
      "epoch": 9.317398071289062e-06,
      "step": 83770
    },
    {
      "epoch": 9.317398071289062e-06,
      "step": 83770,
      "training_step_time": 8.149230718612671
    },
    {
      "epoch": 9.31835174560547e-06,
      "model_forward_time": 1.759629726409912,
      "step": 83771
    },
    {
      "epoch": 9.31835174560547e-06,
      "step": 83771,
      "training_step_time": 8.309094667434692
    },
    {
      "epoch": 9.319305419921875e-06,
      "model_forward_time": 1.7609210014343262,
      "step": 83772
    },
    {
      "epoch": 9.319305419921875e-06,
      "step": 83772,
      "training_step_time": 8.52634596824646
    },
    {
      "epoch": 9.320259094238281e-06,
      "model_forward_time": 1.7635271549224854,
      "step": 83773
    },
    {
      "epoch": 9.320259094238281e-06,
      "step": 83773,
      "training_step_time": 8.974082708358765
    },
    {
      "epoch": 9.321212768554687e-06,
      "model_forward_time": 1.7539384365081787,
      "step": 83774
    },
    {
      "epoch": 9.321212768554687e-06,
      "step": 83774,
      "training_step_time": 7.770455598831177
    },
    {
      "epoch": 9.322166442871094e-06,
      "model_forward_time": 2.447794198989868,
      "step": 83775
    },
    {
      "epoch": 9.322166442871094e-06,
      "step": 83775,
      "training_step_time": 8.664723873138428
    },
    {
      "epoch": 9.3231201171875e-06,
      "model_forward_time": 1.7934696674346924,
      "step": 83776
    },
    {
      "epoch": 9.3231201171875e-06,
      "step": 83776,
      "training_step_time": 7.832272052764893
    },
    {
      "epoch": 9.324073791503906e-06,
      "model_forward_time": 1.7593259811401367,
      "step": 83777
    },
    {
      "epoch": 9.324073791503906e-06,
      "step": 83777,
      "training_step_time": 7.628139495849609
    },
    {
      "epoch": 9.325027465820313e-06,
      "model_forward_time": 1.7638378143310547,
      "step": 83778
    },
    {
      "epoch": 9.325027465820313e-06,
      "step": 83778,
      "training_step_time": 8.537345170974731
    },
    {
      "epoch": 9.325981140136719e-06,
      "model_forward_time": 1.7579681873321533,
      "step": 83779
    },
    {
      "epoch": 9.325981140136719e-06,
      "step": 83779,
      "training_step_time": 7.706446886062622
    },
    {
      "epoch": 9.326934814453124e-06,
      "grad_norm": 0.07926202565431595,
      "learning_rate": 7.022776829286348e-07,
      "loss": 0.0553,
      "step": 83780
    },
    {
      "epoch": 9.326934814453124e-06,
      "model_forward_time": 1.7318687438964844,
      "step": 83780
    },
    {
      "dynamics_loss_avg": 0.06983815934509038,
      "epoch": 9.326934814453124e-06,
      "step": 83780
    },
    {
      "action_loss_avg": 0.001112591676064767,
      "epoch": 9.326934814453124e-06,
      "step": 83780
    },
    {
      "epoch": 9.326934814453124e-06,
      "step": 83780,
      "training_step_time": 8.505216598510742
    },
    {
      "epoch": 9.327888488769532e-06,
      "model_forward_time": 1.771009922027588,
      "step": 83781
    },
    {
      "epoch": 9.327888488769532e-06,
      "step": 83781,
      "training_step_time": 8.48667311668396
    },
    {
      "epoch": 9.328842163085938e-06,
      "model_forward_time": 1.7500710487365723,
      "step": 83782
    },
    {
      "epoch": 9.328842163085938e-06,
      "step": 83782,
      "training_step_time": 8.502575159072876
    },
    {
      "epoch": 9.329795837402343e-06,
      "model_forward_time": 1.750023603439331,
      "step": 83783
    },
    {
      "epoch": 9.329795837402343e-06,
      "step": 83783,
      "training_step_time": 8.644457578659058
    },
    {
      "epoch": 9.33074951171875e-06,
      "model_forward_time": 2.3727476596832275,
      "step": 83784
    },
    {
      "epoch": 9.33074951171875e-06,
      "step": 83784,
      "training_step_time": 7.674027442932129
    },
    {
      "epoch": 9.331703186035156e-06,
      "model_forward_time": 1.7550766468048096,
      "step": 83785
    },
    {
      "epoch": 9.331703186035156e-06,
      "step": 83785,
      "training_step_time": 7.833187818527222
    },
    {
      "epoch": 9.332656860351562e-06,
      "model_forward_time": 1.756852626800537,
      "step": 83786
    },
    {
      "epoch": 9.332656860351562e-06,
      "step": 83786,
      "training_step_time": 8.351595401763916
    },
    {
      "epoch": 9.33361053466797e-06,
      "model_forward_time": 1.7540316581726074,
      "step": 83787
    },
    {
      "epoch": 9.33361053466797e-06,
      "step": 83787,
      "training_step_time": 7.771238565444946
    },
    {
      "epoch": 9.334564208984375e-06,
      "model_forward_time": 1.7493970394134521,
      "step": 83788
    },
    {
      "epoch": 9.334564208984375e-06,
      "step": 83788,
      "training_step_time": 7.680269718170166
    },
    {
      "epoch": 9.335517883300781e-06,
      "model_forward_time": 1.7546484470367432,
      "step": 83789
    },
    {
      "epoch": 9.335517883300781e-06,
      "step": 83789,
      "training_step_time": 8.887691736221313
    },
    {
      "epoch": 9.336471557617187e-06,
      "grad_norm": 0.09187618643045425,
      "learning_rate": 7.014328940507975e-07,
      "loss": 0.0554,
      "step": 83790
    },
    {
      "epoch": 9.336471557617187e-06,
      "model_forward_time": 1.7477340698242188,
      "step": 83790
    },
    {
      "dynamics_loss_avg": 0.050419504009187224,
      "epoch": 9.336471557617187e-06,
      "step": 83790
    },
    {
      "action_loss_avg": 0.002238351374398917,
      "epoch": 9.336471557617187e-06,
      "step": 83790
    },
    {
      "epoch": 9.336471557617187e-06,
      "step": 83790,
      "training_step_time": 9.72837519645691
    },
    {
      "epoch": 9.337425231933594e-06,
      "model_forward_time": 1.758359432220459,
      "step": 83791
    },
    {
      "epoch": 9.337425231933594e-06,
      "step": 83791,
      "training_step_time": 9.668530941009521
    },
    {
      "epoch": 9.33837890625e-06,
      "model_forward_time": 1.7753653526306152,
      "step": 83792
    },
    {
      "epoch": 9.33837890625e-06,
      "step": 83792,
      "training_step_time": 9.208777666091919
    },
    {
      "epoch": 9.339332580566406e-06,
      "model_forward_time": 1.758793830871582,
      "step": 83793
    },
    {
      "epoch": 9.339332580566406e-06,
      "step": 83793,
      "training_step_time": 7.630106687545776
    },
    {
      "epoch": 9.340286254882813e-06,
      "model_forward_time": 2.516758680343628,
      "step": 83794
    },
    {
      "epoch": 9.340286254882813e-06,
      "step": 83794,
      "training_step_time": 9.542770862579346
    },
    {
      "epoch": 9.341239929199219e-06,
      "model_forward_time": 1.775080680847168,
      "step": 83795
    },
    {
      "epoch": 9.341239929199219e-06,
      "step": 83795,
      "training_step_time": 8.162163734436035
    },
    {
      "epoch": 9.342193603515624e-06,
      "model_forward_time": 1.7518525123596191,
      "step": 83796
    },
    {
      "epoch": 9.342193603515624e-06,
      "step": 83796,
      "training_step_time": 7.622474670410156
    },
    {
      "epoch": 9.343147277832032e-06,
      "model_forward_time": 1.7506403923034668,
      "step": 83797
    },
    {
      "epoch": 9.343147277832032e-06,
      "step": 83797,
      "training_step_time": 7.632887363433838
    },
    {
      "epoch": 9.344100952148438e-06,
      "model_forward_time": 1.8017199039459229,
      "step": 83798
    },
    {
      "epoch": 9.344100952148438e-06,
      "step": 83798,
      "training_step_time": 7.792755603790283
    },
    {
      "epoch": 9.345054626464843e-06,
      "model_forward_time": 1.7507421970367432,
      "step": 83799
    },
    {
      "epoch": 9.345054626464843e-06,
      "step": 83799,
      "training_step_time": 8.822420597076416
    },
    {
      "epoch": 9.34600830078125e-06,
      "grad_norm": 0.07756538689136505,
      "learning_rate": 7.005885752577946e-07,
      "loss": 0.0548,
      "step": 83800
    },
    {
      "epoch": 9.34600830078125e-06,
      "model_forward_time": 1.7512648105621338,
      "step": 83800
    },
    {
      "dynamics_loss_avg": 0.045611090026795864,
      "epoch": 9.34600830078125e-06,
      "step": 83800
    },
    {
      "action_loss_avg": 0.0012502576908445917,
      "epoch": 9.34600830078125e-06,
      "step": 83800
    },
    {
      "epoch": 9.34600830078125e-06,
      "step": 83800,
      "training_step_time": 7.759124994277954
    },
    {
      "epoch": 9.346961975097656e-06,
      "model_forward_time": 1.7513189315795898,
      "step": 83801
    },
    {
      "epoch": 9.346961975097656e-06,
      "step": 83801,
      "training_step_time": 9.144400119781494
    },
    {
      "epoch": 9.347915649414062e-06,
      "model_forward_time": 1.7580060958862305,
      "step": 83802
    },
    {
      "epoch": 9.347915649414062e-06,
      "step": 83802,
      "training_step_time": 8.546861410140991
    },
    {
      "epoch": 9.34886932373047e-06,
      "model_forward_time": 2.386706590652466,
      "step": 83803
    },
    {
      "epoch": 9.34886932373047e-06,
      "step": 83803,
      "training_step_time": 8.167709827423096
    },
    {
      "epoch": 9.349822998046875e-06,
      "model_forward_time": 1.754737138748169,
      "step": 83804
    },
    {
      "epoch": 9.349822998046875e-06,
      "step": 83804,
      "training_step_time": 7.706238031387329
    },
    {
      "epoch": 9.350776672363281e-06,
      "model_forward_time": 1.853170394897461,
      "step": 83805
    },
    {
      "epoch": 9.350776672363281e-06,
      "step": 83805,
      "training_step_time": 7.662023305892944
    },
    {
      "epoch": 9.351730346679687e-06,
      "model_forward_time": 1.023402452468872,
      "step": 83806
    },
    {
      "epoch": 9.351730346679687e-06,
      "step": 83806,
      "training_step_time": 7.556157350540161
    },
    {
      "epoch": 9.352684020996094e-06,
      "model_forward_time": 1.7597250938415527,
      "step": 83807
    },
    {
      "epoch": 9.352684020996094e-06,
      "step": 83807,
      "training_step_time": 8.271283149719238
    },
    {
      "epoch": 9.3536376953125e-06,
      "model_forward_time": 1.3739562034606934,
      "step": 83808
    },
    {
      "epoch": 9.3536376953125e-06,
      "step": 83808,
      "training_step_time": 7.645707845687866
    },
    {
      "epoch": 9.354591369628906e-06,
      "model_forward_time": 1.7465784549713135,
      "step": 83809
    },
    {
      "epoch": 9.354591369628906e-06,
      "step": 83809,
      "training_step_time": 7.666890382766724
    },
    {
      "epoch": 9.355545043945313e-06,
      "grad_norm": 0.11016078293323517,
      "learning_rate": 6.997447266419616e-07,
      "loss": 0.0548,
      "step": 83810
    },
    {
      "epoch": 9.355545043945313e-06,
      "model_forward_time": 1.7448177337646484,
      "step": 83810
    },
    {
      "dynamics_loss_avg": 0.058356015384197234,
      "epoch": 9.355545043945313e-06,
      "step": 83810
    },
    {
      "action_loss_avg": 0.0022644063748884944,
      "epoch": 9.355545043945313e-06,
      "step": 83810
    },
    {
      "epoch": 9.355545043945313e-06,
      "step": 83810,
      "training_step_time": 9.07158088684082
    },
    {
      "epoch": 9.356498718261719e-06,
      "model_forward_time": 1.0284833908081055,
      "step": 83811
    },
    {
      "epoch": 9.356498718261719e-06,
      "step": 83811,
      "training_step_time": 7.808310270309448
    },
    {
      "epoch": 9.357452392578124e-06,
      "model_forward_time": 1.760293960571289,
      "step": 83812
    },
    {
      "epoch": 9.357452392578124e-06,
      "step": 83812,
      "training_step_time": 7.632976293563843
    },
    {
      "epoch": 9.358406066894532e-06,
      "model_forward_time": 2.415712833404541,
      "step": 83813
    },
    {
      "epoch": 9.358406066894532e-06,
      "step": 83813,
      "training_step_time": 8.588758945465088
    },
    {
      "epoch": 9.359359741210938e-06,
      "model_forward_time": 1.8243176937103271,
      "step": 83814
    },
    {
      "epoch": 9.359359741210938e-06,
      "step": 83814,
      "training_step_time": 7.643215656280518
    },
    {
      "epoch": 9.360313415527343e-06,
      "model_forward_time": 1.7586309909820557,
      "step": 83815
    },
    {
      "epoch": 9.360313415527343e-06,
      "step": 83815,
      "training_step_time": 7.691101312637329
    },
    {
      "epoch": 9.36126708984375e-06,
      "model_forward_time": 1.75693678855896,
      "step": 83816
    },
    {
      "epoch": 9.36126708984375e-06,
      "step": 83816,
      "training_step_time": 8.60118317604065
    },
    {
      "epoch": 9.362220764160156e-06,
      "model_forward_time": 1.7496113777160645,
      "step": 83817
    },
    {
      "epoch": 9.362220764160156e-06,
      "step": 83817,
      "training_step_time": 8.318495035171509
    },
    {
      "epoch": 9.363174438476562e-06,
      "model_forward_time": 1.7423076629638672,
      "step": 83818
    },
    {
      "epoch": 9.363174438476562e-06,
      "step": 83818,
      "training_step_time": 7.720722198486328
    },
    {
      "epoch": 9.36412811279297e-06,
      "model_forward_time": 1.767812967300415,
      "step": 83819
    },
    {
      "epoch": 9.36412811279297e-06,
      "step": 83819,
      "training_step_time": 7.6483237743377686
    },
    {
      "epoch": 9.365081787109375e-06,
      "grad_norm": 0.079409159719944,
      "learning_rate": 6.989013482955798e-07,
      "loss": 0.0555,
      "step": 83820
    },
    {
      "epoch": 9.365081787109375e-06,
      "model_forward_time": 0.7102198600769043,
      "step": 83820
    },
    {
      "dynamics_loss_avg": 0.05705284969881177,
      "epoch": 9.365081787109375e-06,
      "step": 83820
    },
    {
      "action_loss_avg": 0.001623033607029356,
      "epoch": 9.365081787109375e-06,
      "step": 83820
    },
    {
      "epoch": 9.365081787109375e-06,
      "step": 83820,
      "training_step_time": 7.73047661781311
    },
    {
      "epoch": 9.366035461425781e-06,
      "model_forward_time": 1.7474982738494873,
      "step": 83821
    },
    {
      "epoch": 9.366035461425781e-06,
      "step": 83821,
      "training_step_time": 7.58045768737793
    },
    {
      "epoch": 9.366989135742187e-06,
      "model_forward_time": 2.415015697479248,
      "step": 83822
    },
    {
      "epoch": 9.366989135742187e-06,
      "step": 83822,
      "training_step_time": 8.398218154907227
    },
    {
      "epoch": 9.367942810058594e-06,
      "model_forward_time": 1.8074877262115479,
      "step": 83823
    },
    {
      "epoch": 9.367942810058594e-06,
      "step": 83823,
      "training_step_time": 9.363857746124268
    },
    {
      "epoch": 9.368896484375e-06,
      "model_forward_time": 1.7947063446044922,
      "step": 83824
    },
    {
      "epoch": 9.368896484375e-06,
      "step": 83824,
      "training_step_time": 7.708873510360718
    },
    {
      "epoch": 9.369850158691406e-06,
      "model_forward_time": 1.7491505146026611,
      "step": 83825
    },
    {
      "epoch": 9.369850158691406e-06,
      "step": 83825,
      "training_step_time": 7.687097787857056
    },
    {
      "epoch": 9.370803833007813e-06,
      "model_forward_time": 1.8918414115905762,
      "step": 83826
    },
    {
      "epoch": 9.370803833007813e-06,
      "step": 83826,
      "training_step_time": 7.7261927127838135
    },
    {
      "epoch": 9.371757507324219e-06,
      "model_forward_time": 1.8974735736846924,
      "step": 83827
    },
    {
      "epoch": 9.371757507324219e-06,
      "step": 83827,
      "training_step_time": 8.852509498596191
    },
    {
      "epoch": 9.372711181640624e-06,
      "model_forward_time": 1.9088287353515625,
      "step": 83828
    },
    {
      "epoch": 9.372711181640624e-06,
      "step": 83828,
      "training_step_time": 9.35966420173645
    },
    {
      "epoch": 9.373664855957032e-06,
      "model_forward_time": 1.8187596797943115,
      "step": 83829
    },
    {
      "epoch": 9.373664855957032e-06,
      "step": 83829,
      "training_step_time": 8.54803466796875
    },
    {
      "epoch": 9.374618530273438e-06,
      "grad_norm": 0.09290555119514465,
      "learning_rate": 6.980584403108808e-07,
      "loss": 0.0559,
      "step": 83830
    },
    {
      "epoch": 9.374618530273438e-06,
      "model_forward_time": 1.7539992332458496,
      "step": 83830
    },
    {
      "dynamics_loss_avg": 0.05967299584299326,
      "epoch": 9.374618530273438e-06,
      "step": 83830
    },
    {
      "action_loss_avg": 0.001430051785428077,
      "epoch": 9.374618530273438e-06,
      "step": 83830
    },
    {
      "epoch": 9.374618530273438e-06,
      "step": 83830,
      "training_step_time": 7.707009792327881
    },
    {
      "epoch": 9.375572204589843e-06,
      "model_forward_time": 2.3576018810272217,
      "step": 83831
    },
    {
      "epoch": 9.375572204589843e-06,
      "step": 83831,
      "training_step_time": 8.76814866065979
    },
    {
      "epoch": 9.37652587890625e-06,
      "model_forward_time": 1.7500157356262207,
      "step": 83832
    },
    {
      "epoch": 9.37652587890625e-06,
      "step": 83832,
      "training_step_time": 8.334687232971191
    },
    {
      "epoch": 9.377479553222656e-06,
      "model_forward_time": 1.8680546283721924,
      "step": 83833
    },
    {
      "epoch": 9.377479553222656e-06,
      "step": 83833,
      "training_step_time": 7.671614170074463
    },
    {
      "epoch": 9.378433227539062e-06,
      "model_forward_time": 1.758251667022705,
      "step": 83834
    },
    {
      "epoch": 9.378433227539062e-06,
      "step": 83834,
      "training_step_time": 8.073208570480347
    },
    {
      "epoch": 9.37938690185547e-06,
      "model_forward_time": 1.3731932640075684,
      "step": 83835
    },
    {
      "epoch": 9.37938690185547e-06,
      "step": 83835,
      "training_step_time": 7.769661903381348
    },
    {
      "epoch": 9.380340576171875e-06,
      "model_forward_time": 1.7343242168426514,
      "step": 83836
    },
    {
      "epoch": 9.380340576171875e-06,
      "step": 83836,
      "training_step_time": 8.399781227111816
    },
    {
      "epoch": 9.381294250488281e-06,
      "model_forward_time": 1.7377190589904785,
      "step": 83837
    },
    {
      "epoch": 9.381294250488281e-06,
      "step": 83837,
      "training_step_time": 7.72237229347229
    },
    {
      "epoch": 9.382247924804687e-06,
      "model_forward_time": 1.7367565631866455,
      "step": 83838
    },
    {
      "epoch": 9.382247924804687e-06,
      "step": 83838,
      "training_step_time": 8.724662780761719
    },
    {
      "epoch": 9.383201599121094e-06,
      "model_forward_time": 1.7283680438995361,
      "step": 83839
    },
    {
      "epoch": 9.383201599121094e-06,
      "step": 83839,
      "training_step_time": 8.27487301826477
    },
    {
      "epoch": 9.3841552734375e-06,
      "grad_norm": 0.08201014995574951,
      "learning_rate": 6.972160027800417e-07,
      "loss": 0.0543,
      "step": 83840
    },
    {
      "epoch": 9.3841552734375e-06,
      "model_forward_time": 1.7266497611999512,
      "step": 83840
    },
    {
      "dynamics_loss_avg": 0.06628909967839718,
      "epoch": 9.3841552734375e-06,
      "step": 83840
    },
    {
      "action_loss_avg": 0.0015580182836856693,
      "epoch": 9.3841552734375e-06,
      "step": 83840
    },
    {
      "epoch": 9.3841552734375e-06,
      "step": 83840,
      "training_step_time": 7.83466100692749
    },
    {
      "epoch": 9.385108947753906e-06,
      "model_forward_time": 2.417863130569458,
      "step": 83841
    },
    {
      "epoch": 9.385108947753906e-06,
      "step": 83841,
      "training_step_time": 8.468266725540161
    },
    {
      "epoch": 9.386062622070313e-06,
      "model_forward_time": 1.7566218376159668,
      "step": 83842
    },
    {
      "epoch": 9.386062622070313e-06,
      "step": 83842,
      "training_step_time": 8.625561714172363
    },
    {
      "epoch": 9.387016296386719e-06,
      "model_forward_time": 1.7375147342681885,
      "step": 83843
    },
    {
      "epoch": 9.387016296386719e-06,
      "step": 83843,
      "training_step_time": 7.6690239906311035
    },
    {
      "epoch": 9.387969970703124e-06,
      "model_forward_time": 1.7460861206054688,
      "step": 83844
    },
    {
      "epoch": 9.387969970703124e-06,
      "step": 83844,
      "training_step_time": 7.770425796508789
    },
    {
      "epoch": 9.388923645019532e-06,
      "model_forward_time": 1.7403461933135986,
      "step": 83845
    },
    {
      "epoch": 9.388923645019532e-06,
      "step": 83845,
      "training_step_time": 7.8111748695373535
    },
    {
      "epoch": 9.389877319335938e-06,
      "model_forward_time": 1.742509365081787,
      "step": 83846
    },
    {
      "epoch": 9.389877319335938e-06,
      "step": 83846,
      "training_step_time": 10.266761302947998
    },
    {
      "epoch": 9.390830993652343e-06,
      "model_forward_time": 1.7378320693969727,
      "step": 83847
    },
    {
      "epoch": 9.390830993652343e-06,
      "step": 83847,
      "training_step_time": 7.801272869110107
    },
    {
      "epoch": 9.39178466796875e-06,
      "model_forward_time": 1.7370045185089111,
      "step": 83848
    },
    {
      "epoch": 9.39178466796875e-06,
      "step": 83848,
      "training_step_time": 8.515228748321533
    },
    {
      "epoch": 9.392738342285156e-06,
      "model_forward_time": 1.736192226409912,
      "step": 83849
    },
    {
      "epoch": 9.392738342285156e-06,
      "step": 83849,
      "training_step_time": 8.631668329238892
    },
    {
      "epoch": 9.393692016601562e-06,
      "grad_norm": 0.06745428591966629,
      "learning_rate": 6.963740357951931e-07,
      "loss": 0.0552,
      "step": 83850
    },
    {
      "epoch": 9.393692016601562e-06,
      "model_forward_time": 2.378411054611206,
      "step": 83850
    },
    {
      "dynamics_loss_avg": 0.04385991171002388,
      "epoch": 9.393692016601562e-06,
      "step": 83850
    },
    {
      "action_loss_avg": 0.0009849145484622569,
      "epoch": 9.393692016601562e-06,
      "step": 83850
    },
    {
      "epoch": 9.393692016601562e-06,
      "step": 83850,
      "training_step_time": 7.745232582092285
    },
    {
      "epoch": 9.39464569091797e-06,
      "model_forward_time": 1.7325243949890137,
      "step": 83851
    },
    {
      "epoch": 9.39464569091797e-06,
      "step": 83851,
      "training_step_time": 9.470308065414429
    },
    {
      "epoch": 9.395599365234375e-06,
      "model_forward_time": 1.7550978660583496,
      "step": 83852
    },
    {
      "epoch": 9.395599365234375e-06,
      "step": 83852,
      "training_step_time": 7.843315601348877
    },
    {
      "epoch": 9.396553039550781e-06,
      "model_forward_time": 1.7388744354248047,
      "step": 83853
    },
    {
      "epoch": 9.396553039550781e-06,
      "step": 83853,
      "training_step_time": 8.427701711654663
    },
    {
      "epoch": 9.397506713867187e-06,
      "model_forward_time": 1.7676417827606201,
      "step": 83854
    },
    {
      "epoch": 9.397506713867187e-06,
      "step": 83854,
      "training_step_time": 7.841288805007935
    },
    {
      "epoch": 9.398460388183594e-06,
      "model_forward_time": 1.7377948760986328,
      "step": 83855
    },
    {
      "epoch": 9.398460388183594e-06,
      "step": 83855,
      "training_step_time": 8.346943378448486
    },
    {
      "epoch": 9.3994140625e-06,
      "model_forward_time": 1.7375602722167969,
      "step": 83856
    },
    {
      "epoch": 9.3994140625e-06,
      "step": 83856,
      "training_step_time": 7.848781585693359
    },
    {
      "epoch": 9.400367736816406e-06,
      "model_forward_time": 1.7377054691314697,
      "step": 83857
    },
    {
      "epoch": 9.400367736816406e-06,
      "step": 83857,
      "training_step_time": 7.835709810256958
    },
    {
      "epoch": 9.401321411132813e-06,
      "model_forward_time": 1.7379088401794434,
      "step": 83858
    },
    {
      "epoch": 9.401321411132813e-06,
      "step": 83858,
      "training_step_time": 7.7125184535980225
    },
    {
      "epoch": 9.402275085449219e-06,
      "model_forward_time": 2.417611837387085,
      "step": 83859
    },
    {
      "epoch": 9.402275085449219e-06,
      "step": 83859,
      "training_step_time": 7.835641384124756
    },
    {
      "epoch": 9.403228759765624e-06,
      "grad_norm": 0.10029547661542892,
      "learning_rate": 6.95532539448408e-07,
      "loss": 0.0538,
      "step": 83860
    },
    {
      "epoch": 9.403228759765624e-06,
      "model_forward_time": 1.7223296165466309,
      "step": 83860
    },
    {
      "dynamics_loss_avg": 0.062114561535418035,
      "epoch": 9.403228759765624e-06,
      "step": 83860
    },
    {
      "action_loss_avg": 0.000895955934538506,
      "epoch": 9.403228759765624e-06,
      "step": 83860
    },
    {
      "epoch": 9.403228759765624e-06,
      "step": 83860,
      "training_step_time": 9.124510765075684
    },
    {
      "epoch": 9.404182434082032e-06,
      "model_forward_time": 1.7656750679016113,
      "step": 83861
    },
    {
      "epoch": 9.404182434082032e-06,
      "step": 83861,
      "training_step_time": 7.822721719741821
    },
    {
      "epoch": 9.405136108398438e-06,
      "model_forward_time": 1.3596737384796143,
      "step": 83862
    },
    {
      "epoch": 9.405136108398438e-06,
      "step": 83862,
      "training_step_time": 7.819597959518433
    },
    {
      "epoch": 9.406089782714843e-06,
      "model_forward_time": 1.7668836116790771,
      "step": 83863
    },
    {
      "epoch": 9.406089782714843e-06,
      "step": 83863,
      "training_step_time": 7.832060813903809
    },
    {
      "epoch": 9.40704345703125e-06,
      "model_forward_time": 1.7473466396331787,
      "step": 83864
    },
    {
      "epoch": 9.40704345703125e-06,
      "step": 83864,
      "training_step_time": 7.816562652587891
    },
    {
      "epoch": 9.407997131347656e-06,
      "model_forward_time": 1.3671331405639648,
      "step": 83865
    },
    {
      "epoch": 9.407997131347656e-06,
      "step": 83865,
      "training_step_time": 7.846819877624512
    },
    {
      "epoch": 9.408950805664062e-06,
      "model_forward_time": 1.7492952346801758,
      "step": 83866
    },
    {
      "epoch": 9.408950805664062e-06,
      "step": 83866,
      "training_step_time": 7.720994234085083
    },
    {
      "epoch": 9.40990447998047e-06,
      "model_forward_time": 1.7479791641235352,
      "step": 83867
    },
    {
      "epoch": 9.40990447998047e-06,
      "step": 83867,
      "training_step_time": 8.785404682159424
    },
    {
      "epoch": 9.410858154296875e-06,
      "model_forward_time": 1.7461974620819092,
      "step": 83868
    },
    {
      "epoch": 9.410858154296875e-06,
      "step": 83868,
      "training_step_time": 7.784057378768921
    },
    {
      "epoch": 9.411811828613281e-06,
      "model_forward_time": 2.4018962383270264,
      "step": 83869
    },
    {
      "epoch": 9.411811828613281e-06,
      "step": 83869,
      "training_step_time": 9.095776081085205
    },
    {
      "epoch": 9.412765502929687e-06,
      "grad_norm": 0.07155238091945648,
      "learning_rate": 6.946915138317129e-07,
      "loss": 0.0545,
      "step": 83870
    },
    {
      "epoch": 9.412765502929687e-06,
      "model_forward_time": 1.7440898418426514,
      "step": 83870
    },
    {
      "dynamics_loss_avg": 0.07127052061259746,
      "epoch": 9.412765502929687e-06,
      "step": 83870
    },
    {
      "action_loss_avg": 0.0021218565118033437,
      "epoch": 9.412765502929687e-06,
      "step": 83870
    },
    {
      "epoch": 9.412765502929687e-06,
      "step": 83870,
      "training_step_time": 8.448241472244263
    },
    {
      "epoch": 9.413719177246094e-06,
      "model_forward_time": 1.7487316131591797,
      "step": 83871
    },
    {
      "epoch": 9.413719177246094e-06,
      "step": 83871,
      "training_step_time": 7.7651143074035645
    },
    {
      "epoch": 9.4146728515625e-06,
      "model_forward_time": 1.0299851894378662,
      "step": 83872
    },
    {
      "epoch": 9.4146728515625e-06,
      "step": 83872,
      "training_step_time": 7.741220474243164
    },
    {
      "epoch": 9.415626525878906e-06,
      "model_forward_time": 1.744739294052124,
      "step": 83873
    },
    {
      "epoch": 9.415626525878906e-06,
      "step": 83873,
      "training_step_time": 9.701071977615356
    },
    {
      "epoch": 9.416580200195313e-06,
      "model_forward_time": 1.7483813762664795,
      "step": 83874
    },
    {
      "epoch": 9.416580200195313e-06,
      "step": 83874,
      "training_step_time": 9.518221378326416
    },
    {
      "epoch": 9.417533874511719e-06,
      "model_forward_time": 1.748459815979004,
      "step": 83875
    },
    {
      "epoch": 9.417533874511719e-06,
      "step": 83875,
      "training_step_time": 7.766472101211548
    },
    {
      "epoch": 9.418487548828124e-06,
      "model_forward_time": 1.0278923511505127,
      "step": 83876
    },
    {
      "epoch": 9.418487548828124e-06,
      "step": 83876,
      "training_step_time": 8.491410493850708
    },
    {
      "epoch": 9.419441223144532e-06,
      "model_forward_time": 1.7457995414733887,
      "step": 83877
    },
    {
      "epoch": 9.419441223144532e-06,
      "step": 83877,
      "training_step_time": 9.53475546836853
    },
    {
      "epoch": 9.420394897460938e-06,
      "model_forward_time": 2.3640646934509277,
      "step": 83878
    },
    {
      "epoch": 9.420394897460938e-06,
      "step": 83878,
      "training_step_time": 8.781699419021606
    },
    {
      "epoch": 9.421348571777343e-06,
      "model_forward_time": 1.7594494819641113,
      "step": 83879
    },
    {
      "epoch": 9.421348571777343e-06,
      "step": 83879,
      "training_step_time": 7.745616912841797
    },
    {
      "epoch": 9.42230224609375e-06,
      "grad_norm": 0.08704668283462524,
      "learning_rate": 6.938509590370807e-07,
      "loss": 0.0556,
      "step": 83880
    },
    {
      "epoch": 9.42230224609375e-06,
      "model_forward_time": 1.739623785018921,
      "step": 83880
    },
    {
      "dynamics_loss_avg": 0.05973015557974577,
      "epoch": 9.42230224609375e-06,
      "step": 83880
    },
    {
      "action_loss_avg": 0.0017113599926233292,
      "epoch": 9.42230224609375e-06,
      "step": 83880
    },
    {
      "epoch": 9.42230224609375e-06,
      "step": 83880,
      "training_step_time": 8.562513589859009
    },
    {
      "epoch": 9.423255920410156e-06,
      "model_forward_time": 1.7615289688110352,
      "step": 83881
    },
    {
      "epoch": 9.423255920410156e-06,
      "step": 83881,
      "training_step_time": 7.72671914100647
    },
    {
      "epoch": 9.424209594726562e-06,
      "model_forward_time": 1.746872901916504,
      "step": 83882
    },
    {
      "epoch": 9.424209594726562e-06,
      "step": 83882,
      "training_step_time": 8.792702913284302
    },
    {
      "epoch": 9.425163269042968e-06,
      "model_forward_time": 1.7510650157928467,
      "step": 83883
    },
    {
      "epoch": 9.425163269042968e-06,
      "step": 83883,
      "training_step_time": 11.327839136123657
    },
    {
      "epoch": 9.426116943359375e-06,
      "model_forward_time": 1.7462100982666016,
      "step": 83884
    },
    {
      "epoch": 9.426116943359375e-06,
      "step": 83884,
      "training_step_time": 9.423691987991333
    },
    {
      "epoch": 9.427070617675781e-06,
      "model_forward_time": 1.7473621368408203,
      "step": 83885
    },
    {
      "epoch": 9.427070617675781e-06,
      "step": 83885,
      "training_step_time": 7.908412933349609
    },
    {
      "epoch": 9.428024291992187e-06,
      "model_forward_time": 1.7441086769104004,
      "step": 83886
    },
    {
      "epoch": 9.428024291992187e-06,
      "step": 83886,
      "training_step_time": 7.722051620483398
    },
    {
      "epoch": 9.428977966308594e-06,
      "model_forward_time": 1.7512285709381104,
      "step": 83887
    },
    {
      "epoch": 9.428977966308594e-06,
      "step": 83887,
      "training_step_time": 7.521092176437378
    },
    {
      "epoch": 9.429931640625e-06,
      "model_forward_time": 2.449108839035034,
      "step": 83888
    },
    {
      "epoch": 9.429931640625e-06,
      "step": 83888,
      "training_step_time": 8.545522212982178
    },
    {
      "epoch": 9.430885314941406e-06,
      "model_forward_time": 1.7436630725860596,
      "step": 83889
    },
    {
      "epoch": 9.430885314941406e-06,
      "step": 83889,
      "training_step_time": 7.53407096862793
    },
    {
      "epoch": 9.431838989257813e-06,
      "grad_norm": 0.07329729199409485,
      "learning_rate": 6.930108751564346e-07,
      "loss": 0.0553,
      "step": 83890
    },
    {
      "epoch": 9.431838989257813e-06,
      "model_forward_time": 1.735158920288086,
      "step": 83890
    },
    {
      "dynamics_loss_avg": 0.04599792957305908,
      "epoch": 9.431838989257813e-06,
      "step": 83890
    },
    {
      "action_loss_avg": 0.0010903695816523395,
      "epoch": 9.431838989257813e-06,
      "step": 83890
    },
    {
      "epoch": 9.431838989257813e-06,
      "step": 83890,
      "training_step_time": 8.951786518096924
    },
    {
      "epoch": 9.432792663574219e-06,
      "model_forward_time": 1.748643398284912,
      "step": 83891
    },
    {
      "epoch": 9.432792663574219e-06,
      "step": 83891,
      "training_step_time": 7.813985824584961
    },
    {
      "epoch": 9.433746337890624e-06,
      "model_forward_time": 1.748664379119873,
      "step": 83892
    },
    {
      "epoch": 9.433746337890624e-06,
      "step": 83892,
      "training_step_time": 8.400716781616211
    },
    {
      "epoch": 9.434700012207032e-06,
      "model_forward_time": 1.7558484077453613,
      "step": 83893
    },
    {
      "epoch": 9.434700012207032e-06,
      "step": 83893,
      "training_step_time": 7.8638482093811035
    },
    {
      "epoch": 9.435653686523438e-06,
      "model_forward_time": 1.7526164054870605,
      "step": 83894
    },
    {
      "epoch": 9.435653686523438e-06,
      "step": 83894,
      "training_step_time": 8.24178433418274
    },
    {
      "epoch": 9.436607360839843e-06,
      "model_forward_time": 1.7265849113464355,
      "step": 83895
    },
    {
      "epoch": 9.436607360839843e-06,
      "step": 83895,
      "training_step_time": 8.824103116989136
    },
    {
      "epoch": 9.43756103515625e-06,
      "model_forward_time": 1.7493560314178467,
      "step": 83896
    },
    {
      "epoch": 9.43756103515625e-06,
      "step": 83896,
      "training_step_time": 7.716301679611206
    },
    {
      "epoch": 9.438514709472656e-06,
      "model_forward_time": 1.9582273960113525,
      "step": 83897
    },
    {
      "epoch": 9.438514709472656e-06,
      "step": 83897,
      "training_step_time": 8.672976732254028
    },
    {
      "epoch": 9.439468383789062e-06,
      "model_forward_time": 1.7388405799865723,
      "step": 83898
    },
    {
      "epoch": 9.439468383789062e-06,
      "step": 83898,
      "training_step_time": 7.669891357421875
    },
    {
      "epoch": 9.440422058105468e-06,
      "model_forward_time": 1.7463300228118896,
      "step": 83899
    },
    {
      "epoch": 9.440422058105468e-06,
      "step": 83899,
      "training_step_time": 8.527515888214111
    },
    {
      "epoch": 9.441375732421875e-06,
      "grad_norm": 0.0849563330411911,
      "learning_rate": 6.921712622816429e-07,
      "loss": 0.0552,
      "step": 83900
    },
    {
      "epoch": 9.441375732421875e-06,
      "model_forward_time": 1.3800146579742432,
      "step": 83900
    },
    {
      "dynamics_loss_avg": 0.06435017362236976,
      "epoch": 9.441375732421875e-06,
      "step": 83900
    },
    {
      "action_loss_avg": 0.00201953629439231,
      "epoch": 9.441375732421875e-06,
      "step": 83900
    },
    {
      "epoch": 9.441375732421875e-06,
      "step": 83900,
      "training_step_time": 8.904635190963745
    },
    {
      "epoch": 9.442329406738281e-06,
      "model_forward_time": 1.7390408515930176,
      "step": 83901
    },
    {
      "epoch": 9.442329406738281e-06,
      "step": 83901,
      "training_step_time": 9.668893814086914
    },
    {
      "epoch": 9.443283081054687e-06,
      "model_forward_time": 1.733109474182129,
      "step": 83902
    },
    {
      "epoch": 9.443283081054687e-06,
      "step": 83902,
      "training_step_time": 8.871477127075195
    },
    {
      "epoch": 9.444236755371094e-06,
      "model_forward_time": 1.7456390857696533,
      "step": 83903
    },
    {
      "epoch": 9.444236755371094e-06,
      "step": 83903,
      "training_step_time": 7.777013063430786
    },
    {
      "epoch": 9.4451904296875e-06,
      "model_forward_time": 1.733170509338379,
      "step": 83904
    },
    {
      "epoch": 9.4451904296875e-06,
      "step": 83904,
      "training_step_time": 9.312363624572754
    },
    {
      "epoch": 9.446144104003906e-06,
      "model_forward_time": 1.740105390548706,
      "step": 83905
    },
    {
      "epoch": 9.446144104003906e-06,
      "step": 83905,
      "training_step_time": 8.42526912689209
    },
    {
      "epoch": 9.447097778320313e-06,
      "model_forward_time": 2.394577741622925,
      "step": 83906
    },
    {
      "epoch": 9.447097778320313e-06,
      "step": 83906,
      "training_step_time": 7.567497730255127
    },
    {
      "epoch": 9.448051452636719e-06,
      "model_forward_time": 1.7532000541687012,
      "step": 83907
    },
    {
      "epoch": 9.448051452636719e-06,
      "step": 83907,
      "training_step_time": 7.823285341262817
    },
    {
      "epoch": 9.449005126953124e-06,
      "model_forward_time": 1.735382318496704,
      "step": 83908
    },
    {
      "epoch": 9.449005126953124e-06,
      "step": 83908,
      "training_step_time": 7.685976982116699
    },
    {
      "epoch": 9.449958801269532e-06,
      "model_forward_time": 1.7663025856018066,
      "step": 83909
    },
    {
      "epoch": 9.449958801269532e-06,
      "step": 83909,
      "training_step_time": 8.793945550918579
    },
    {
      "epoch": 9.450912475585938e-06,
      "grad_norm": 0.08585842698812485,
      "learning_rate": 6.91332120504526e-07,
      "loss": 0.0545,
      "step": 83910
    },
    {
      "epoch": 9.450912475585938e-06,
      "model_forward_time": 1.728722333908081,
      "step": 83910
    },
    {
      "dynamics_loss_avg": 0.06404130924493075,
      "epoch": 9.450912475585938e-06,
      "step": 83910
    },
    {
      "action_loss_avg": 0.001582543650874868,
      "epoch": 9.450912475585938e-06,
      "step": 83910
    },
    {
      "epoch": 9.450912475585938e-06,
      "step": 83910,
      "training_step_time": 7.917627573013306
    },
    {
      "epoch": 9.451866149902343e-06,
      "model_forward_time": 1.7389352321624756,
      "step": 83911
    },
    {
      "epoch": 9.451866149902343e-06,
      "step": 83911,
      "training_step_time": 8.924295902252197
    },
    {
      "epoch": 9.45281982421875e-06,
      "model_forward_time": 1.7444658279418945,
      "step": 83912
    },
    {
      "epoch": 9.45281982421875e-06,
      "step": 83912,
      "training_step_time": 8.574640274047852
    },
    {
      "epoch": 9.453773498535156e-06,
      "model_forward_time": 1.734602451324463,
      "step": 83913
    },
    {
      "epoch": 9.453773498535156e-06,
      "step": 83913,
      "training_step_time": 10.041268348693848
    },
    {
      "epoch": 9.454727172851562e-06,
      "model_forward_time": 1.741748332977295,
      "step": 83914
    },
    {
      "epoch": 9.454727172851562e-06,
      "step": 83914,
      "training_step_time": 7.763395547866821
    },
    {
      "epoch": 9.455680847167968e-06,
      "model_forward_time": 1.7415595054626465,
      "step": 83915
    },
    {
      "epoch": 9.455680847167968e-06,
      "step": 83915,
      "training_step_time": 8.176759243011475
    },
    {
      "epoch": 9.456634521484375e-06,
      "model_forward_time": 1.7523472309112549,
      "step": 83916
    },
    {
      "epoch": 9.456634521484375e-06,
      "step": 83916,
      "training_step_time": 7.672309637069702
    },
    {
      "epoch": 9.457588195800781e-06,
      "model_forward_time": 1.7361316680908203,
      "step": 83917
    },
    {
      "epoch": 9.457588195800781e-06,
      "step": 83917,
      "training_step_time": 7.722837209701538
    },
    {
      "epoch": 9.458541870117187e-06,
      "model_forward_time": 1.765791893005371,
      "step": 83918
    },
    {
      "epoch": 9.458541870117187e-06,
      "step": 83918,
      "training_step_time": 7.7258360385894775
    },
    {
      "epoch": 9.459495544433594e-06,
      "model_forward_time": 1.7397077083587646,
      "step": 83919
    },
    {
      "epoch": 9.459495544433594e-06,
      "step": 83919,
      "training_step_time": 10.364191770553589
    },
    {
      "epoch": 9.46044921875e-06,
      "grad_norm": 0.07699897885322571,
      "learning_rate": 6.904934499168492e-07,
      "loss": 0.0557,
      "step": 83920
    },
    {
      "epoch": 9.46044921875e-06,
      "model_forward_time": 1.7327144145965576,
      "step": 83920
    },
    {
      "dynamics_loss_avg": 0.07271200288087129,
      "epoch": 9.46044921875e-06,
      "step": 83920
    },
    {
      "action_loss_avg": 0.0017765673488611355,
      "epoch": 9.46044921875e-06,
      "step": 83920
    },
    {
      "epoch": 9.46044921875e-06,
      "step": 83920,
      "training_step_time": 8.920244455337524
    },
    {
      "epoch": 9.461402893066406e-06,
      "model_forward_time": 1.7502920627593994,
      "step": 83921
    },
    {
      "epoch": 9.461402893066406e-06,
      "step": 83921,
      "training_step_time": 7.682219505310059
    },
    {
      "epoch": 9.462356567382813e-06,
      "model_forward_time": 1.7399773597717285,
      "step": 83922
    },
    {
      "epoch": 9.462356567382813e-06,
      "step": 83922,
      "training_step_time": 9.520501375198364
    },
    {
      "epoch": 9.463310241699219e-06,
      "model_forward_time": 1.748798131942749,
      "step": 83923
    },
    {
      "epoch": 9.463310241699219e-06,
      "step": 83923,
      "training_step_time": 7.791204214096069
    },
    {
      "epoch": 9.464263916015625e-06,
      "model_forward_time": 1.7419793605804443,
      "step": 83924
    },
    {
      "epoch": 9.464263916015625e-06,
      "step": 83924,
      "training_step_time": 7.467904329299927
    },
    {
      "epoch": 9.465217590332032e-06,
      "model_forward_time": 2.4264605045318604,
      "step": 83925
    },
    {
      "epoch": 9.465217590332032e-06,
      "step": 83925,
      "training_step_time": 9.249339580535889
    },
    {
      "epoch": 9.466171264648438e-06,
      "model_forward_time": 1.746558427810669,
      "step": 83926
    },
    {
      "epoch": 9.466171264648438e-06,
      "step": 83926,
      "training_step_time": 7.616846084594727
    },
    {
      "epoch": 9.467124938964843e-06,
      "model_forward_time": 1.7876596450805664,
      "step": 83927
    },
    {
      "epoch": 9.467124938964843e-06,
      "step": 83927,
      "training_step_time": 9.33969497680664
    },
    {
      "epoch": 9.46807861328125e-06,
      "model_forward_time": 1.798088550567627,
      "step": 83928
    },
    {
      "epoch": 9.46807861328125e-06,
      "step": 83928,
      "training_step_time": 7.8289954662323
    },
    {
      "epoch": 9.469032287597657e-06,
      "model_forward_time": 1.7966251373291016,
      "step": 83929
    },
    {
      "epoch": 9.469032287597657e-06,
      "step": 83929,
      "training_step_time": 8.725430488586426
    },
    {
      "epoch": 9.469985961914062e-06,
      "grad_norm": 0.08996591717004776,
      "learning_rate": 6.896552506103299e-07,
      "loss": 0.0521,
      "step": 83930
    },
    {
      "epoch": 9.469985961914062e-06,
      "model_forward_time": 1.8218657970428467,
      "step": 83930
    },
    {
      "dynamics_loss_avg": 0.05272536743432284,
      "epoch": 9.469985961914062e-06,
      "step": 83930
    },
    {
      "action_loss_avg": 0.0009184277696476784,
      "epoch": 9.469985961914062e-06,
      "step": 83930
    },
    {
      "epoch": 9.469985961914062e-06,
      "step": 83930,
      "training_step_time": 7.828596830368042
    },
    {
      "epoch": 9.470939636230468e-06,
      "model_forward_time": 1.8210484981536865,
      "step": 83931
    },
    {
      "epoch": 9.470939636230468e-06,
      "step": 83931,
      "training_step_time": 7.529856443405151
    },
    {
      "epoch": 9.471893310546875e-06,
      "model_forward_time": 1.8054537773132324,
      "step": 83932
    },
    {
      "epoch": 9.471893310546875e-06,
      "step": 83932,
      "training_step_time": 7.805621147155762
    },
    {
      "epoch": 9.472846984863281e-06,
      "model_forward_time": 1.8145506381988525,
      "step": 83933
    },
    {
      "epoch": 9.472846984863281e-06,
      "step": 83933,
      "training_step_time": 7.558769464492798
    },
    {
      "epoch": 9.473800659179687e-06,
      "model_forward_time": 2.8719165325164795,
      "step": 83934
    },
    {
      "epoch": 9.473800659179687e-06,
      "step": 83934,
      "training_step_time": 7.721410036087036
    },
    {
      "epoch": 9.474754333496094e-06,
      "model_forward_time": 1.8173787593841553,
      "step": 83935
    },
    {
      "epoch": 9.474754333496094e-06,
      "step": 83935,
      "training_step_time": 8.21188497543335
    },
    {
      "epoch": 9.4757080078125e-06,
      "model_forward_time": 1.7910423278808594,
      "step": 83936
    },
    {
      "epoch": 9.4757080078125e-06,
      "step": 83936,
      "training_step_time": 8.660370826721191
    },
    {
      "epoch": 9.476661682128906e-06,
      "model_forward_time": 1.916736364364624,
      "step": 83937
    },
    {
      "epoch": 9.476661682128906e-06,
      "step": 83937,
      "training_step_time": 8.771822690963745
    },
    {
      "epoch": 9.477615356445313e-06,
      "model_forward_time": 1.8067286014556885,
      "step": 83938
    },
    {
      "epoch": 9.477615356445313e-06,
      "step": 83938,
      "training_step_time": 7.783113479614258
    },
    {
      "epoch": 9.478569030761719e-06,
      "model_forward_time": 1.8132128715515137,
      "step": 83939
    },
    {
      "epoch": 9.478569030761719e-06,
      "step": 83939,
      "training_step_time": 9.213780164718628
    },
    {
      "epoch": 9.479522705078125e-06,
      "grad_norm": 0.10814021527767181,
      "learning_rate": 6.888175226766314e-07,
      "loss": 0.0536,
      "step": 83940
    },
    {
      "epoch": 9.479522705078125e-06,
      "model_forward_time": 1.799154281616211,
      "step": 83940
    },
    {
      "dynamics_loss_avg": 0.05128618367016315,
      "epoch": 9.479522705078125e-06,
      "step": 83940
    },
    {
      "action_loss_avg": 0.0013982431264594197,
      "epoch": 9.479522705078125e-06,
      "step": 83940
    },
    {
      "epoch": 9.479522705078125e-06,
      "step": 83940,
      "training_step_time": 7.608592748641968
    },
    {
      "epoch": 9.480476379394532e-06,
      "model_forward_time": 1.7985289096832275,
      "step": 83941
    },
    {
      "epoch": 9.480476379394532e-06,
      "step": 83941,
      "training_step_time": 8.697591304779053
    },
    {
      "epoch": 9.481430053710938e-06,
      "model_forward_time": 1.815439224243164,
      "step": 83942
    },
    {
      "epoch": 9.481430053710938e-06,
      "step": 83942,
      "training_step_time": 8.35673713684082
    },
    {
      "epoch": 9.482383728027343e-06,
      "model_forward_time": 2.8804304599761963,
      "step": 83943
    },
    {
      "epoch": 9.482383728027343e-06,
      "step": 83943,
      "training_step_time": 7.750780820846558
    },
    {
      "epoch": 9.48333740234375e-06,
      "model_forward_time": 1.8055872917175293,
      "step": 83944
    },
    {
      "epoch": 9.48333740234375e-06,
      "step": 83944,
      "training_step_time": 7.797490835189819
    },
    {
      "epoch": 9.484291076660157e-06,
      "model_forward_time": 1.7318871021270752,
      "step": 83945
    },
    {
      "epoch": 9.484291076660157e-06,
      "step": 83945,
      "training_step_time": 8.488854885101318
    },
    {
      "epoch": 9.485244750976562e-06,
      "model_forward_time": 1.7223236560821533,
      "step": 83946
    },
    {
      "epoch": 9.485244750976562e-06,
      "step": 83946,
      "training_step_time": 8.476378917694092
    },
    {
      "epoch": 9.486198425292968e-06,
      "model_forward_time": 1.7263360023498535,
      "step": 83947
    },
    {
      "epoch": 9.486198425292968e-06,
      "step": 83947,
      "training_step_time": 7.881542921066284
    },
    {
      "epoch": 9.487152099609375e-06,
      "model_forward_time": 1.7164840698242188,
      "step": 83948
    },
    {
      "epoch": 9.487152099609375e-06,
      "step": 83948,
      "training_step_time": 8.333192110061646
    },
    {
      "epoch": 9.488105773925781e-06,
      "model_forward_time": 1.7198491096496582,
      "step": 83949
    },
    {
      "epoch": 9.488105773925781e-06,
      "step": 83949,
      "training_step_time": 8.125171422958374
    },
    {
      "epoch": 9.489059448242187e-06,
      "grad_norm": 0.0853477418422699,
      "learning_rate": 6.879802662073687e-07,
      "loss": 0.0521,
      "step": 83950
    },
    {
      "epoch": 9.489059448242187e-06,
      "model_forward_time": 1.7199699878692627,
      "step": 83950
    },
    {
      "dynamics_loss_avg": 0.045171450823545456,
      "epoch": 9.489059448242187e-06,
      "step": 83950
    },
    {
      "action_loss_avg": 0.001582654018420726,
      "epoch": 9.489059448242187e-06,
      "step": 83950
    },
    {
      "epoch": 9.489059448242187e-06,
      "step": 83950,
      "training_step_time": 7.712264537811279
    },
    {
      "epoch": 9.490013122558594e-06,
      "model_forward_time": 1.7222914695739746,
      "step": 83951
    },
    {
      "epoch": 9.490013122558594e-06,
      "step": 83951,
      "training_step_time": 7.892799615859985
    },
    {
      "epoch": 9.490966796875e-06,
      "model_forward_time": 2.3606455326080322,
      "step": 83952
    },
    {
      "epoch": 9.490966796875e-06,
      "step": 83952,
      "training_step_time": 8.66352915763855
    },
    {
      "epoch": 9.491920471191406e-06,
      "model_forward_time": 1.7284824848175049,
      "step": 83953
    },
    {
      "epoch": 9.491920471191406e-06,
      "step": 83953,
      "training_step_time": 7.808304309844971
    },
    {
      "epoch": 9.492874145507813e-06,
      "model_forward_time": 1.7466390132904053,
      "step": 83954
    },
    {
      "epoch": 9.492874145507813e-06,
      "step": 83954,
      "training_step_time": 9.391072511672974
    },
    {
      "epoch": 9.493827819824219e-06,
      "model_forward_time": 1.7219929695129395,
      "step": 83955
    },
    {
      "epoch": 9.493827819824219e-06,
      "step": 83955,
      "training_step_time": 8.671459674835205
    },
    {
      "epoch": 9.494781494140625e-06,
      "model_forward_time": 1.7199201583862305,
      "step": 83956
    },
    {
      "epoch": 9.494781494140625e-06,
      "step": 83956,
      "training_step_time": 7.83982253074646
    },
    {
      "epoch": 9.495735168457032e-06,
      "model_forward_time": 1.7210183143615723,
      "step": 83957
    },
    {
      "epoch": 9.495735168457032e-06,
      "step": 83957,
      "training_step_time": 8.230680227279663
    },
    {
      "epoch": 9.496688842773438e-06,
      "model_forward_time": 1.002638339996338,
      "step": 83958
    },
    {
      "epoch": 9.496688842773438e-06,
      "step": 83958,
      "training_step_time": 9.31186294555664
    },
    {
      "epoch": 9.497642517089843e-06,
      "model_forward_time": 1.7254199981689453,
      "step": 83959
    },
    {
      "epoch": 9.497642517089843e-06,
      "step": 83959,
      "training_step_time": 8.353746891021729
    },
    {
      "epoch": 9.49859619140625e-06,
      "grad_norm": 0.07668524980545044,
      "learning_rate": 6.871434812940992e-07,
      "loss": 0.0531,
      "step": 83960
    },
    {
      "epoch": 9.49859619140625e-06,
      "model_forward_time": 1.7124390602111816,
      "step": 83960
    },
    {
      "dynamics_loss_avg": 0.06410715244710445,
      "epoch": 9.49859619140625e-06,
      "step": 83960
    },
    {
      "action_loss_avg": 0.0017827878560638054,
      "epoch": 9.49859619140625e-06,
      "step": 83960
    },
    {
      "epoch": 9.49859619140625e-06,
      "step": 83960,
      "training_step_time": 7.735357284545898
    },
    {
      "epoch": 9.499549865722657e-06,
      "model_forward_time": 2.356451988220215,
      "step": 83961
    },
    {
      "epoch": 9.499549865722657e-06,
      "step": 83961,
      "training_step_time": 7.55991268157959
    },
    {
      "epoch": 9.500503540039062e-06,
      "model_forward_time": 1.7320544719696045,
      "step": 83962
    },
    {
      "epoch": 9.500503540039062e-06,
      "step": 83962,
      "training_step_time": 7.577138662338257
    },
    {
      "epoch": 9.501457214355468e-06,
      "model_forward_time": 1.7496507167816162,
      "step": 83963
    },
    {
      "epoch": 9.501457214355468e-06,
      "step": 83963,
      "training_step_time": 7.618398904800415
    },
    {
      "epoch": 9.502410888671875e-06,
      "model_forward_time": 1.7216787338256836,
      "step": 83964
    },
    {
      "epoch": 9.502410888671875e-06,
      "step": 83964,
      "training_step_time": 7.655481338500977
    },
    {
      "epoch": 9.503364562988281e-06,
      "model_forward_time": 1.721409797668457,
      "step": 83965
    },
    {
      "epoch": 9.503364562988281e-06,
      "step": 83965,
      "training_step_time": 7.666927099227905
    },
    {
      "epoch": 9.504318237304687e-06,
      "model_forward_time": 1.7291834354400635,
      "step": 83966
    },
    {
      "epoch": 9.504318237304687e-06,
      "step": 83966,
      "training_step_time": 7.612639427185059
    },
    {
      "epoch": 9.505271911621094e-06,
      "model_forward_time": 1.7204711437225342,
      "step": 83967
    },
    {
      "epoch": 9.505271911621094e-06,
      "step": 83967,
      "training_step_time": 8.688251256942749
    },
    {
      "epoch": 9.5062255859375e-06,
      "model_forward_time": 1.7205283641815186,
      "step": 83968
    },
    {
      "epoch": 9.5062255859375e-06,
      "step": 83968,
      "training_step_time": 11.158532619476318
    },
    {
      "epoch": 9.507179260253906e-06,
      "model_forward_time": 1.716233253479004,
      "step": 83969
    },
    {
      "epoch": 9.507179260253906e-06,
      "step": 83969,
      "training_step_time": 8.22491717338562
    },
    {
      "epoch": 9.508132934570313e-06,
      "grad_norm": 0.1061074361205101,
      "learning_rate": 6.863071680283345e-07,
      "loss": 0.0531,
      "step": 83970
    },
    {
      "epoch": 9.508132934570313e-06,
      "model_forward_time": 1.7263462543487549,
      "step": 83970
    },
    {
      "dynamics_loss_avg": 0.04276337008923292,
      "epoch": 9.508132934570313e-06,
      "step": 83970
    },
    {
      "action_loss_avg": 0.0008563528914237395,
      "epoch": 9.508132934570313e-06,
      "step": 83970
    },
    {
      "epoch": 9.508132934570313e-06,
      "step": 83970,
      "training_step_time": 8.36271619796753
    },
    {
      "epoch": 9.509086608886719e-06,
      "model_forward_time": 1.0035016536712646,
      "step": 83971
    },
    {
      "epoch": 9.509086608886719e-06,
      "step": 83971,
      "training_step_time": 7.828475475311279
    },
    {
      "epoch": 9.510040283203125e-06,
      "model_forward_time": 1.7273497581481934,
      "step": 83972
    },
    {
      "epoch": 9.510040283203125e-06,
      "step": 83972,
      "training_step_time": 8.2494215965271
    },
    {
      "epoch": 9.510993957519532e-06,
      "model_forward_time": 1.7301931381225586,
      "step": 83973
    },
    {
      "epoch": 9.510993957519532e-06,
      "step": 83973,
      "training_step_time": 6.893068313598633
    },
    {
      "epoch": 9.511947631835938e-06,
      "model_forward_time": 1.7059783935546875,
      "step": 83974
    },
    {
      "epoch": 9.511947631835938e-06,
      "step": 83974,
      "training_step_time": 6.785974979400635
    },
    {
      "epoch": 9.512901306152343e-06,
      "model_forward_time": 1.7252061367034912,
      "step": 83975
    },
    {
      "epoch": 9.512901306152343e-06,
      "step": 83975,
      "training_step_time": 6.822983980178833
    },
    {
      "epoch": 9.51385498046875e-06,
      "model_forward_time": 1.343733787536621,
      "step": 83976
    },
    {
      "epoch": 9.51385498046875e-06,
      "step": 83976,
      "training_step_time": 6.8443474769592285
    },
    {
      "epoch": 9.514808654785157e-06,
      "model_forward_time": 1.7162866592407227,
      "step": 83977
    },
    {
      "epoch": 9.514808654785157e-06,
      "step": 83977,
      "training_step_time": 9.853455781936646
    },
    {
      "epoch": 9.515762329101562e-06,
      "model_forward_time": 2.349581241607666,
      "step": 83978
    },
    {
      "epoch": 9.515762329101562e-06,
      "step": 83978,
      "training_step_time": 7.90205717086792
    },
    {
      "epoch": 9.516716003417968e-06,
      "model_forward_time": 1.7175979614257812,
      "step": 83979
    },
    {
      "epoch": 9.516716003417968e-06,
      "step": 83979,
      "training_step_time": 6.852376937866211
    },
    {
      "epoch": 9.517669677734375e-06,
      "grad_norm": 0.08507037162780762,
      "learning_rate": 6.854713265015339e-07,
      "loss": 0.0534,
      "step": 83980
    },
    {
      "epoch": 9.517669677734375e-06,
      "model_forward_time": 1.7111990451812744,
      "step": 83980
    },
    {
      "dynamics_loss_avg": 0.04484348790720105,
      "epoch": 9.517669677734375e-06,
      "step": 83980
    },
    {
      "action_loss_avg": 0.0017750941216945649,
      "epoch": 9.517669677734375e-06,
      "step": 83980
    },
    {
      "epoch": 9.517669677734375e-06,
      "step": 83980,
      "training_step_time": 7.821289777755737
    },
    {
      "epoch": 9.518623352050781e-06,
      "model_forward_time": 1.7260594367980957,
      "step": 83981
    },
    {
      "epoch": 9.518623352050781e-06,
      "step": 83981,
      "training_step_time": 7.471457242965698
    },
    {
      "epoch": 9.519577026367187e-06,
      "model_forward_time": 1.7067899703979492,
      "step": 83982
    },
    {
      "epoch": 9.519577026367187e-06,
      "step": 83982,
      "training_step_time": 7.735543966293335
    },
    {
      "epoch": 9.520530700683594e-06,
      "model_forward_time": 1.7184770107269287,
      "step": 83983
    },
    {
      "epoch": 9.520530700683594e-06,
      "step": 83983,
      "training_step_time": 6.895385026931763
    },
    {
      "epoch": 9.521484375e-06,
      "model_forward_time": 1.7218596935272217,
      "step": 83984
    },
    {
      "epoch": 9.521484375e-06,
      "step": 83984,
      "training_step_time": 6.898376703262329
    },
    {
      "epoch": 9.522438049316406e-06,
      "model_forward_time": 1.7186791896820068,
      "step": 83985
    },
    {
      "epoch": 9.522438049316406e-06,
      "step": 83985,
      "training_step_time": 6.898403167724609
    },
    {
      "epoch": 9.523391723632813e-06,
      "model_forward_time": 1.7189750671386719,
      "step": 83986
    },
    {
      "epoch": 9.523391723632813e-06,
      "step": 83986,
      "training_step_time": 7.977521181106567
    },
    {
      "epoch": 9.524345397949219e-06,
      "model_forward_time": 2.353996515274048,
      "step": 83987
    },
    {
      "epoch": 9.524345397949219e-06,
      "step": 83987,
      "training_step_time": 7.880271673202515
    },
    {
      "epoch": 9.525299072265625e-06,
      "model_forward_time": 1.7176427841186523,
      "step": 83988
    },
    {
      "epoch": 9.525299072265625e-06,
      "step": 83988,
      "training_step_time": 6.846346378326416
    },
    {
      "epoch": 9.526252746582032e-06,
      "model_forward_time": 1.7153244018554688,
      "step": 83989
    },
    {
      "epoch": 9.526252746582032e-06,
      "step": 83989,
      "training_step_time": 7.783405542373657
    },
    {
      "epoch": 9.527206420898438e-06,
      "grad_norm": 0.08508898317813873,
      "learning_rate": 6.846359568051003e-07,
      "loss": 0.0537,
      "step": 83990
    },
    {
      "epoch": 9.527206420898438e-06,
      "model_forward_time": 1.7254736423492432,
      "step": 83990
    },
    {
      "dynamics_loss_avg": 0.07387884203344583,
      "epoch": 9.527206420898438e-06,
      "step": 83990
    },
    {
      "action_loss_avg": 0.0013985011712065898,
      "epoch": 9.527206420898438e-06,
      "step": 83990
    },
    {
      "epoch": 9.527206420898438e-06,
      "step": 83990,
      "training_step_time": 7.399288892745972
    },
    {
      "epoch": 9.528160095214843e-06,
      "model_forward_time": 1.7233061790466309,
      "step": 83991
    },
    {
      "epoch": 9.528160095214843e-06,
      "step": 83991,
      "training_step_time": 7.677629709243774
    },
    {
      "epoch": 9.52911376953125e-06,
      "model_forward_time": 1.7213373184204102,
      "step": 83992
    },
    {
      "epoch": 9.52911376953125e-06,
      "step": 83992,
      "training_step_time": 6.802216291427612
    },
    {
      "epoch": 9.530067443847657e-06,
      "model_forward_time": 1.7142763137817383,
      "step": 83993
    },
    {
      "epoch": 9.530067443847657e-06,
      "step": 83993,
      "training_step_time": 6.84800124168396
    },
    {
      "epoch": 9.531021118164062e-06,
      "model_forward_time": 1.7208356857299805,
      "step": 83994
    },
    {
      "epoch": 9.531021118164062e-06,
      "step": 83994,
      "training_step_time": 6.853967666625977
    },
    {
      "epoch": 9.531974792480468e-06,
      "model_forward_time": 1.7228899002075195,
      "step": 83995
    },
    {
      "epoch": 9.531974792480468e-06,
      "step": 83995,
      "training_step_time": 6.791877746582031
    },
    {
      "epoch": 9.532928466796875e-06,
      "model_forward_time": 1.7553668022155762,
      "step": 83996
    },
    {
      "epoch": 9.532928466796875e-06,
      "step": 83996,
      "training_step_time": 8.909638166427612
    },
    {
      "epoch": 9.533882141113281e-06,
      "model_forward_time": 1.7155048847198486,
      "step": 83997
    },
    {
      "epoch": 9.533882141113281e-06,
      "step": 83997,
      "training_step_time": 6.824672698974609
    },
    {
      "epoch": 9.534835815429687e-06,
      "model_forward_time": 1.7507820129394531,
      "step": 83998
    },
    {
      "epoch": 9.534835815429687e-06,
      "step": 83998,
      "training_step_time": 8.06761908531189
    },
    {
      "epoch": 9.535789489746094e-06,
      "model_forward_time": 1.4116442203521729,
      "step": 83999
    },
    {
      "epoch": 9.535789489746094e-06,
      "step": 83999,
      "training_step_time": 7.81382942199707
    },
    {
      "epoch": 9.5367431640625e-06,
      "grad_norm": 0.07985251396894455,
      "learning_rate": 6.838010590303912e-07,
      "loss": 0.052,
      "step": 84000
    },
    {
      "epoch": 9.5367431640625e-06,
      "model_forward_time": 1.7828569412231445,
      "step": 84000
    },
    {
      "dynamics_loss_avg": 0.05579463765025139,
      "epoch": 9.5367431640625e-06,
      "step": 84000
    },
    {
      "action_loss_avg": 0.0020795313408598303,
      "epoch": 9.5367431640625e-06,
      "step": 84000
    },
    {
      "epoch": 9.5367431640625e-06,
      "step": 84000,
      "training_step_time": 7.641490936279297
    },
    {
      "epoch": 9.537696838378906e-06,
      "model_forward_time": 1.8079774379730225,
      "step": 84001
    },
    {
      "epoch": 9.537696838378906e-06,
      "step": 84001,
      "training_step_time": 6.973540544509888
    },
    {
      "epoch": 9.538650512695313e-06,
      "model_forward_time": 1.8129301071166992,
      "step": 84002
    },
    {
      "epoch": 9.538650512695313e-06,
      "step": 84002,
      "training_step_time": 6.920844793319702
    },
    {
      "epoch": 9.539604187011719e-06,
      "model_forward_time": 1.8053152561187744,
      "step": 84003
    },
    {
      "epoch": 9.539604187011719e-06,
      "step": 84003,
      "training_step_time": 6.93498969078064
    },
    {
      "epoch": 9.540557861328125e-06,
      "model_forward_time": 1.8080031871795654,
      "step": 84004
    },
    {
      "epoch": 9.540557861328125e-06,
      "step": 84004,
      "training_step_time": 8.941359758377075
    },
    {
      "epoch": 9.541511535644532e-06,
      "model_forward_time": 3.4763619899749756,
      "step": 84005
    },
    {
      "epoch": 9.541511535644532e-06,
      "step": 84005,
      "training_step_time": 9.598652601242065
    },
    {
      "epoch": 9.542465209960938e-06,
      "model_forward_time": 1.865147590637207,
      "step": 84006
    },
    {
      "epoch": 9.542465209960938e-06,
      "step": 84006,
      "training_step_time": 7.9759767055511475
    },
    {
      "epoch": 9.543418884277343e-06,
      "model_forward_time": 1.8000679016113281,
      "step": 84007
    },
    {
      "epoch": 9.543418884277343e-06,
      "step": 84007,
      "training_step_time": 7.954670190811157
    },
    {
      "epoch": 9.54437255859375e-06,
      "model_forward_time": 1.796602487564087,
      "step": 84008
    },
    {
      "epoch": 9.54437255859375e-06,
      "step": 84008,
      "training_step_time": 7.793287754058838
    },
    {
      "epoch": 9.545326232910157e-06,
      "model_forward_time": 1.8138999938964844,
      "step": 84009
    },
    {
      "epoch": 9.545326232910157e-06,
      "step": 84009,
      "training_step_time": 7.7452967166900635
    },
    {
      "epoch": 9.546279907226562e-06,
      "grad_norm": 0.08211935311555862,
      "learning_rate": 6.829666332687084e-07,
      "loss": 0.0537,
      "step": 84010
    },
    {
      "epoch": 9.546279907226562e-06,
      "model_forward_time": 1.787247657775879,
      "step": 84010
    },
    {
      "dynamics_loss_avg": 0.05468851402401924,
      "epoch": 9.546279907226562e-06,
      "step": 84010
    },
    {
      "action_loss_avg": 0.0013878372046747245,
      "epoch": 9.546279907226562e-06,
      "step": 84010
    },
    {
      "epoch": 9.546279907226562e-06,
      "step": 84010,
      "training_step_time": 7.847450494766235
    },
    {
      "epoch": 9.547233581542968e-06,
      "model_forward_time": 1.802466630935669,
      "step": 84011
    },
    {
      "epoch": 9.547233581542968e-06,
      "step": 84011,
      "training_step_time": 6.974412441253662
    },
    {
      "epoch": 9.548187255859375e-06,
      "model_forward_time": 1.804957628250122,
      "step": 84012
    },
    {
      "epoch": 9.548187255859375e-06,
      "step": 84012,
      "training_step_time": 6.971729755401611
    },
    {
      "epoch": 9.549140930175781e-06,
      "model_forward_time": 1.791182041168213,
      "step": 84013
    },
    {
      "epoch": 9.549140930175781e-06,
      "step": 84013,
      "training_step_time": 6.9997687339782715
    },
    {
      "epoch": 9.550094604492187e-06,
      "model_forward_time": 1.8664872646331787,
      "step": 84014
    },
    {
      "epoch": 9.550094604492187e-06,
      "step": 84014,
      "training_step_time": 10.283905029296875
    },
    {
      "epoch": 9.551048278808594e-06,
      "model_forward_time": 1.805363655090332,
      "step": 84015
    },
    {
      "epoch": 9.551048278808594e-06,
      "step": 84015,
      "training_step_time": 8.1135573387146
    },
    {
      "epoch": 9.552001953125e-06,
      "model_forward_time": 1.8054890632629395,
      "step": 84016
    },
    {
      "epoch": 9.552001953125e-06,
      "step": 84016,
      "training_step_time": 7.571708679199219
    },
    {
      "epoch": 9.552955627441406e-06,
      "model_forward_time": 1.8030779361724854,
      "step": 84017
    },
    {
      "epoch": 9.552955627441406e-06,
      "step": 84017,
      "training_step_time": 7.872565031051636
    },
    {
      "epoch": 9.553909301757813e-06,
      "model_forward_time": 1.8000929355621338,
      "step": 84018
    },
    {
      "epoch": 9.553909301757813e-06,
      "step": 84018,
      "training_step_time": 7.450958728790283
    },
    {
      "epoch": 9.554862976074219e-06,
      "model_forward_time": 1.7986555099487305,
      "step": 84019
    },
    {
      "epoch": 9.554862976074219e-06,
      "step": 84019,
      "training_step_time": 7.827337741851807
    },
    {
      "epoch": 9.555816650390625e-06,
      "grad_norm": 0.07996562868356705,
      "learning_rate": 6.82132679611306e-07,
      "loss": 0.0535,
      "step": 84020
    },
    {
      "epoch": 9.555816650390625e-06,
      "model_forward_time": 1.801623821258545,
      "step": 84020
    },
    {
      "dynamics_loss_avg": 0.04574438109993935,
      "epoch": 9.555816650390625e-06,
      "step": 84020
    },
    {
      "action_loss_avg": 0.0014764374907827004,
      "epoch": 9.555816650390625e-06,
      "step": 84020
    },
    {
      "epoch": 9.555816650390625e-06,
      "step": 84020,
      "training_step_time": 7.025621652603149
    },
    {
      "epoch": 9.556770324707032e-06,
      "model_forward_time": 1.4452643394470215,
      "step": 84021
    },
    {
      "epoch": 9.556770324707032e-06,
      "step": 84021,
      "training_step_time": 6.980793476104736
    },
    {
      "epoch": 9.557723999023438e-06,
      "model_forward_time": 1.8075132369995117,
      "step": 84022
    },
    {
      "epoch": 9.557723999023438e-06,
      "step": 84022,
      "training_step_time": 7.037725448608398
    },
    {
      "epoch": 9.558677673339843e-06,
      "model_forward_time": 1.861804485321045,
      "step": 84023
    },
    {
      "epoch": 9.558677673339843e-06,
      "step": 84023,
      "training_step_time": 8.845388889312744
    },
    {
      "epoch": 9.55963134765625e-06,
      "model_forward_time": 2.55180025100708,
      "step": 84024
    },
    {
      "epoch": 9.55963134765625e-06,
      "step": 84024,
      "training_step_time": 8.091394662857056
    },
    {
      "epoch": 9.560585021972657e-06,
      "model_forward_time": 1.7978224754333496,
      "step": 84025
    },
    {
      "epoch": 9.560585021972657e-06,
      "step": 84025,
      "training_step_time": 7.777357339859009
    },
    {
      "epoch": 9.561538696289062e-06,
      "model_forward_time": 1.8073780536651611,
      "step": 84026
    },
    {
      "epoch": 9.561538696289062e-06,
      "step": 84026,
      "training_step_time": 7.984910726547241
    },
    {
      "epoch": 9.562492370605468e-06,
      "model_forward_time": 1.8094382286071777,
      "step": 84027
    },
    {
      "epoch": 9.562492370605468e-06,
      "step": 84027,
      "training_step_time": 7.609649658203125
    },
    {
      "epoch": 9.563446044921875e-06,
      "model_forward_time": 1.8032023906707764,
      "step": 84028
    },
    {
      "epoch": 9.563446044921875e-06,
      "step": 84028,
      "training_step_time": 7.817598819732666
    },
    {
      "epoch": 9.564399719238281e-06,
      "model_forward_time": 1.8083395957946777,
      "step": 84029
    },
    {
      "epoch": 9.564399719238281e-06,
      "step": 84029,
      "training_step_time": 6.968732833862305
    },
    {
      "epoch": 9.565353393554687e-06,
      "grad_norm": 0.11425140500068665,
      "learning_rate": 6.8129919814938e-07,
      "loss": 0.053,
      "step": 84030
    },
    {
      "epoch": 9.565353393554687e-06,
      "model_forward_time": 1.7981665134429932,
      "step": 84030
    },
    {
      "dynamics_loss_avg": 0.04002542290836573,
      "epoch": 9.565353393554687e-06,
      "step": 84030
    },
    {
      "action_loss_avg": 0.0017514260311145336,
      "epoch": 9.565353393554687e-06,
      "step": 84030
    },
    {
      "epoch": 9.565353393554687e-06,
      "step": 84030,
      "training_step_time": 6.951824188232422
    },
    {
      "epoch": 9.566307067871094e-06,
      "model_forward_time": 1.811614751815796,
      "step": 84031
    },
    {
      "epoch": 9.566307067871094e-06,
      "step": 84031,
      "training_step_time": 6.923962593078613
    },
    {
      "epoch": 9.5672607421875e-06,
      "model_forward_time": 1.861816167831421,
      "step": 84032
    },
    {
      "epoch": 9.5672607421875e-06,
      "step": 84032,
      "training_step_time": 8.954901933670044
    },
    {
      "epoch": 9.568214416503906e-06,
      "model_forward_time": 2.9129481315612793,
      "step": 84033
    },
    {
      "epoch": 9.568214416503906e-06,
      "step": 84033,
      "training_step_time": 8.99461579322815
    },
    {
      "epoch": 9.569168090820313e-06,
      "model_forward_time": 1.8173751831054688,
      "step": 84034
    },
    {
      "epoch": 9.569168090820313e-06,
      "step": 84034,
      "training_step_time": 7.335633754730225
    },
    {
      "epoch": 9.570121765136719e-06,
      "model_forward_time": 1.7879095077514648,
      "step": 84035
    },
    {
      "epoch": 9.570121765136719e-06,
      "step": 84035,
      "training_step_time": 7.879220962524414
    },
    {
      "epoch": 9.571075439453125e-06,
      "model_forward_time": 1.8024482727050781,
      "step": 84036
    },
    {
      "epoch": 9.571075439453125e-06,
      "step": 84036,
      "training_step_time": 7.9405341148376465
    },
    {
      "epoch": 9.572029113769532e-06,
      "model_forward_time": 1.8014647960662842,
      "step": 84037
    },
    {
      "epoch": 9.572029113769532e-06,
      "step": 84037,
      "training_step_time": 8.579489946365356
    },
    {
      "epoch": 9.572982788085938e-06,
      "model_forward_time": 1.810807466506958,
      "step": 84038
    },
    {
      "epoch": 9.572982788085938e-06,
      "step": 84038,
      "training_step_time": 7.012509346008301
    },
    {
      "epoch": 9.573936462402343e-06,
      "model_forward_time": 1.8023300170898438,
      "step": 84039
    },
    {
      "epoch": 9.573936462402343e-06,
      "step": 84039,
      "training_step_time": 7.0128254890441895
    },
    {
      "epoch": 9.57489013671875e-06,
      "grad_norm": 0.23210948705673218,
      "learning_rate": 6.804661889740821e-07,
      "loss": 0.0542,
      "step": 84040
    },
    {
      "epoch": 9.57489013671875e-06,
      "model_forward_time": 1.7918267250061035,
      "step": 84040
    },
    {
      "dynamics_loss_avg": 0.04277582410722971,
      "epoch": 9.57489013671875e-06,
      "step": 84040
    },
    {
      "action_loss_avg": 0.0011276657198322937,
      "epoch": 9.57489013671875e-06,
      "step": 84040
    },
    {
      "epoch": 9.57489013671875e-06,
      "step": 84040,
      "training_step_time": 7.0791027545928955
    },
    {
      "epoch": 9.575843811035157e-06,
      "model_forward_time": 1.8298099040985107,
      "step": 84041
    },
    {
      "epoch": 9.575843811035157e-06,
      "step": 84041,
      "training_step_time": 11.476221799850464
    },
    {
      "epoch": 9.576797485351562e-06,
      "model_forward_time": 2.8809051513671875,
      "step": 84042
    },
    {
      "epoch": 9.576797485351562e-06,
      "step": 84042,
      "training_step_time": 9.773180961608887
    },
    {
      "epoch": 9.577751159667968e-06,
      "model_forward_time": 1.7956368923187256,
      "step": 84043
    },
    {
      "epoch": 9.577751159667968e-06,
      "step": 84043,
      "training_step_time": 7.910955905914307
    },
    {
      "epoch": 9.578704833984375e-06,
      "model_forward_time": 1.8049795627593994,
      "step": 84044
    },
    {
      "epoch": 9.578704833984375e-06,
      "step": 84044,
      "training_step_time": 7.6163270473480225
    },
    {
      "epoch": 9.579658508300781e-06,
      "model_forward_time": 1.8104257583618164,
      "step": 84045
    },
    {
      "epoch": 9.579658508300781e-06,
      "step": 84045,
      "training_step_time": 8.800681829452515
    },
    {
      "epoch": 9.580612182617187e-06,
      "model_forward_time": 1.8019587993621826,
      "step": 84046
    },
    {
      "epoch": 9.580612182617187e-06,
      "step": 84046,
      "training_step_time": 7.847186088562012
    },
    {
      "epoch": 9.581565856933594e-06,
      "model_forward_time": 1.7967307567596436,
      "step": 84047
    },
    {
      "epoch": 9.581565856933594e-06,
      "step": 84047,
      "training_step_time": 7.948623418807983
    },
    {
      "epoch": 9.58251953125e-06,
      "model_forward_time": 1.8077754974365234,
      "step": 84048
    },
    {
      "epoch": 9.58251953125e-06,
      "step": 84048,
      "training_step_time": 7.010385751724243
    },
    {
      "epoch": 9.583473205566406e-06,
      "model_forward_time": 1.8171868324279785,
      "step": 84049
    },
    {
      "epoch": 9.583473205566406e-06,
      "step": 84049,
      "training_step_time": 7.024115800857544
    },
    {
      "epoch": 9.584426879882813e-06,
      "grad_norm": 0.07680581510066986,
      "learning_rate": 6.796336521765062e-07,
      "loss": 0.0541,
      "step": 84050
    },
    {
      "epoch": 9.584426879882813e-06,
      "model_forward_time": 1.8020458221435547,
      "step": 84050
    },
    {
      "dynamics_loss_avg": 0.05277015008032322,
      "epoch": 9.584426879882813e-06,
      "step": 84050
    },
    {
      "action_loss_avg": 0.0011677715403493494,
      "epoch": 9.584426879882813e-06,
      "step": 84050
    },
    {
      "epoch": 9.584426879882813e-06,
      "step": 84050,
      "training_step_time": 11.02880072593689
    },
    {
      "epoch": 9.585380554199219e-06,
      "model_forward_time": 1.7903525829315186,
      "step": 84051
    },
    {
      "epoch": 9.585380554199219e-06,
      "step": 84051,
      "training_step_time": 12.19212293624878
    },
    {
      "epoch": 9.586334228515625e-06,
      "model_forward_time": 1.793382167816162,
      "step": 84052
    },
    {
      "epoch": 9.586334228515625e-06,
      "step": 84052,
      "training_step_time": 7.980422258377075
    },
    {
      "epoch": 9.587287902832032e-06,
      "model_forward_time": 1.8098154067993164,
      "step": 84053
    },
    {
      "epoch": 9.587287902832032e-06,
      "step": 84053,
      "training_step_time": 7.610680103302002
    },
    {
      "epoch": 9.588241577148438e-06,
      "model_forward_time": 1.8182940483093262,
      "step": 84054
    },
    {
      "epoch": 9.588241577148438e-06,
      "step": 84054,
      "training_step_time": 8.012192010879517
    },
    {
      "epoch": 9.589195251464843e-06,
      "model_forward_time": 1.791874885559082,
      "step": 84055
    },
    {
      "epoch": 9.589195251464843e-06,
      "step": 84055,
      "training_step_time": 7.6937642097473145
    },
    {
      "epoch": 9.59014892578125e-06,
      "model_forward_time": 1.8108093738555908,
      "step": 84056
    },
    {
      "epoch": 9.59014892578125e-06,
      "step": 84056,
      "training_step_time": 7.5746307373046875
    },
    {
      "epoch": 9.591102600097657e-06,
      "model_forward_time": 1.8166229724884033,
      "step": 84057
    },
    {
      "epoch": 9.591102600097657e-06,
      "step": 84057,
      "training_step_time": 6.94355845451355
    },
    {
      "epoch": 9.592056274414062e-06,
      "model_forward_time": 1.8054730892181396,
      "step": 84058
    },
    {
      "epoch": 9.592056274414062e-06,
      "step": 84058,
      "training_step_time": 6.948092460632324
    },
    {
      "epoch": 9.593009948730468e-06,
      "model_forward_time": 1.8114259243011475,
      "step": 84059
    },
    {
      "epoch": 9.593009948730468e-06,
      "step": 84059,
      "training_step_time": 6.989909648895264
    },
    {
      "epoch": 9.593963623046875e-06,
      "grad_norm": 0.06691373139619827,
      "learning_rate": 6.788015878476983e-07,
      "loss": 0.0562,
      "step": 84060
    },
    {
      "epoch": 9.593963623046875e-06,
      "model_forward_time": 1.8721473217010498,
      "step": 84060
    },
    {
      "dynamics_loss_avg": 0.040616005659103394,
      "epoch": 9.593963623046875e-06,
      "step": 84060
    },
    {
      "action_loss_avg": 0.0018047387013211847,
      "epoch": 9.593963623046875e-06,
      "step": 84060
    },
    {
      "epoch": 9.593963623046875e-06,
      "step": 84060,
      "training_step_time": 11.193196773529053
    },
    {
      "epoch": 9.594917297363281e-06,
      "model_forward_time": 2.8939266204833984,
      "step": 84061
    },
    {
      "epoch": 9.594917297363281e-06,
      "step": 84061,
      "training_step_time": 8.0576331615448
    },
    {
      "epoch": 9.595870971679687e-06,
      "model_forward_time": 1.8019676208496094,
      "step": 84062
    },
    {
      "epoch": 9.595870971679687e-06,
      "step": 84062,
      "training_step_time": 7.756195306777954
    },
    {
      "epoch": 9.596824645996094e-06,
      "model_forward_time": 1.808018445968628,
      "step": 84063
    },
    {
      "epoch": 9.596824645996094e-06,
      "step": 84063,
      "training_step_time": 7.766955614089966
    },
    {
      "epoch": 9.5977783203125e-06,
      "model_forward_time": 1.8068768978118896,
      "step": 84064
    },
    {
      "epoch": 9.5977783203125e-06,
      "step": 84064,
      "training_step_time": 7.783643484115601
    },
    {
      "epoch": 9.598731994628906e-06,
      "model_forward_time": 1.8082516193389893,
      "step": 84065
    },
    {
      "epoch": 9.598731994628906e-06,
      "step": 84065,
      "training_step_time": 7.805783033370972
    },
    {
      "epoch": 9.599685668945313e-06,
      "model_forward_time": 1.8041954040527344,
      "step": 84066
    },
    {
      "epoch": 9.599685668945313e-06,
      "step": 84066,
      "training_step_time": 6.893219470977783
    },
    {
      "epoch": 9.600639343261719e-06,
      "model_forward_time": 1.8042495250701904,
      "step": 84067
    },
    {
      "epoch": 9.600639343261719e-06,
      "step": 84067,
      "training_step_time": 6.878718137741089
    },
    {
      "epoch": 9.601593017578125e-06,
      "model_forward_time": 1.804957628250122,
      "step": 84068
    },
    {
      "epoch": 9.601593017578125e-06,
      "step": 84068,
      "training_step_time": 6.920217275619507
    },
    {
      "epoch": 9.602546691894532e-06,
      "model_forward_time": 1.867844820022583,
      "step": 84069
    },
    {
      "epoch": 9.602546691894532e-06,
      "step": 84069,
      "training_step_time": 11.458604335784912
    },
    {
      "epoch": 9.603500366210938e-06,
      "grad_norm": 0.07909419387578964,
      "learning_rate": 6.779699960786523e-07,
      "loss": 0.0544,
      "step": 84070
    },
    {
      "epoch": 9.603500366210938e-06,
      "model_forward_time": 2.913168430328369,
      "step": 84070
    },
    {
      "dynamics_loss_avg": 0.04736609044484794,
      "epoch": 9.603500366210938e-06,
      "step": 84070
    },
    {
      "action_loss_avg": 0.0015236300896503963,
      "epoch": 9.603500366210938e-06,
      "step": 84070
    },
    {
      "epoch": 9.603500366210938e-06,
      "step": 84070,
      "training_step_time": 9.513633251190186
    },
    {
      "epoch": 9.604454040527343e-06,
      "model_forward_time": 1.7790021896362305,
      "step": 84071
    },
    {
      "epoch": 9.604454040527343e-06,
      "step": 84071,
      "training_step_time": 7.679965972900391
    },
    {
      "epoch": 9.60540771484375e-06,
      "model_forward_time": 1.823190450668335,
      "step": 84072
    },
    {
      "epoch": 9.60540771484375e-06,
      "step": 84072,
      "training_step_time": 7.635003089904785
    },
    {
      "epoch": 9.606361389160157e-06,
      "model_forward_time": 1.8030996322631836,
      "step": 84073
    },
    {
      "epoch": 9.606361389160157e-06,
      "step": 84073,
      "training_step_time": 8.362768173217773
    },
    {
      "epoch": 9.607315063476562e-06,
      "model_forward_time": 1.8019814491271973,
      "step": 84074
    },
    {
      "epoch": 9.607315063476562e-06,
      "step": 84074,
      "training_step_time": 7.771468639373779
    },
    {
      "epoch": 9.608268737792968e-06,
      "model_forward_time": 1.8037097454071045,
      "step": 84075
    },
    {
      "epoch": 9.608268737792968e-06,
      "step": 84075,
      "training_step_time": 7.736413955688477
    },
    {
      "epoch": 9.609222412109375e-06,
      "model_forward_time": 1.8143093585968018,
      "step": 84076
    },
    {
      "epoch": 9.609222412109375e-06,
      "step": 84076,
      "training_step_time": 6.883128643035889
    },
    {
      "epoch": 9.610176086425781e-06,
      "model_forward_time": 1.8016817569732666,
      "step": 84077
    },
    {
      "epoch": 9.610176086425781e-06,
      "step": 84077,
      "training_step_time": 6.913414716720581
    },
    {
      "epoch": 9.611129760742187e-06,
      "model_forward_time": 1.850745677947998,
      "step": 84078
    },
    {
      "epoch": 9.611129760742187e-06,
      "step": 84078,
      "training_step_time": 9.461403369903564
    },
    {
      "epoch": 9.612083435058594e-06,
      "model_forward_time": 2.90714168548584,
      "step": 84079
    },
    {
      "epoch": 9.612083435058594e-06,
      "step": 84079,
      "training_step_time": 10.026769638061523
    },
    {
      "epoch": 9.613037109375e-06,
      "grad_norm": 0.082986980676651,
      "learning_rate": 6.771388769603104e-07,
      "loss": 0.0566,
      "step": 84080
    },
    {
      "epoch": 9.613037109375e-06,
      "model_forward_time": 1.793138027191162,
      "step": 84080
    },
    {
      "dynamics_loss_avg": 0.04548197221010923,
      "epoch": 9.613037109375e-06,
      "step": 84080
    },
    {
      "action_loss_avg": 0.000930563293513842,
      "epoch": 9.613037109375e-06,
      "step": 84080
    },
    {
      "epoch": 9.613037109375e-06,
      "step": 84080,
      "training_step_time": 7.743128299713135
    },
    {
      "epoch": 9.613990783691406e-06,
      "model_forward_time": 1.8127295970916748,
      "step": 84081
    },
    {
      "epoch": 9.613990783691406e-06,
      "step": 84081,
      "training_step_time": 7.615687370300293
    },
    {
      "epoch": 9.614944458007813e-06,
      "model_forward_time": 1.8241560459136963,
      "step": 84082
    },
    {
      "epoch": 9.614944458007813e-06,
      "step": 84082,
      "training_step_time": 7.842538118362427
    },
    {
      "epoch": 9.615898132324219e-06,
      "model_forward_time": 1.7956962585449219,
      "step": 84083
    },
    {
      "epoch": 9.615898132324219e-06,
      "step": 84083,
      "training_step_time": 7.662416934967041
    },
    {
      "epoch": 9.616851806640625e-06,
      "model_forward_time": 1.7951514720916748,
      "step": 84084
    },
    {
      "epoch": 9.616851806640625e-06,
      "step": 84084,
      "training_step_time": 7.806907415390015
    },
    {
      "epoch": 9.617805480957032e-06,
      "model_forward_time": 1.8225502967834473,
      "step": 84085
    },
    {
      "epoch": 9.617805480957032e-06,
      "step": 84085,
      "training_step_time": 6.917747974395752
    },
    {
      "epoch": 9.618759155273438e-06,
      "model_forward_time": 1.8208119869232178,
      "step": 84086
    },
    {
      "epoch": 9.618759155273438e-06,
      "step": 84086,
      "training_step_time": 6.847233057022095
    },
    {
      "epoch": 9.619712829589844e-06,
      "model_forward_time": 1.8026297092437744,
      "step": 84087
    },
    {
      "epoch": 9.619712829589844e-06,
      "step": 84087,
      "training_step_time": 10.82517409324646
    },
    {
      "epoch": 9.62066650390625e-06,
      "model_forward_time": 2.8869924545288086,
      "step": 84088
    },
    {
      "epoch": 9.62066650390625e-06,
      "step": 84088,
      "training_step_time": 8.688507318496704
    },
    {
      "epoch": 9.621620178222657e-06,
      "model_forward_time": 1.7987873554229736,
      "step": 84089
    },
    {
      "epoch": 9.621620178222657e-06,
      "step": 84089,
      "training_step_time": 8.398014545440674
    },
    {
      "epoch": 9.622573852539062e-06,
      "grad_norm": 0.10446726530790329,
      "learning_rate": 6.763082305835606e-07,
      "loss": 0.0569,
      "step": 84090
    },
    {
      "epoch": 9.622573852539062e-06,
      "model_forward_time": 1.809438943862915,
      "step": 84090
    },
    {
      "dynamics_loss_avg": 0.056314104422926906,
      "epoch": 9.622573852539062e-06,
      "step": 84090
    },
    {
      "action_loss_avg": 0.0008722276659682393,
      "epoch": 9.622573852539062e-06,
      "step": 84090
    },
    {
      "epoch": 9.622573852539062e-06,
      "step": 84090,
      "training_step_time": 7.617853403091431
    },
    {
      "epoch": 9.623527526855468e-06,
      "model_forward_time": 1.8110127449035645,
      "step": 84091
    },
    {
      "epoch": 9.623527526855468e-06,
      "step": 84091,
      "training_step_time": 7.7895636558532715
    },
    {
      "epoch": 9.624481201171876e-06,
      "model_forward_time": 1.8187131881713867,
      "step": 84092
    },
    {
      "epoch": 9.624481201171876e-06,
      "step": 84092,
      "training_step_time": 7.6104512214660645
    },
    {
      "epoch": 9.625434875488281e-06,
      "model_forward_time": 1.821444034576416,
      "step": 84093
    },
    {
      "epoch": 9.625434875488281e-06,
      "step": 84093,
      "training_step_time": 7.739619970321655
    },
    {
      "epoch": 9.626388549804687e-06,
      "model_forward_time": 1.7093846797943115,
      "step": 84094
    },
    {
      "epoch": 9.626388549804687e-06,
      "step": 84094,
      "training_step_time": 6.8105244636535645
    },
    {
      "epoch": 9.627342224121094e-06,
      "model_forward_time": 1.7274436950683594,
      "step": 84095
    },
    {
      "epoch": 9.627342224121094e-06,
      "step": 84095,
      "training_step_time": 6.861003875732422
    },
    {
      "epoch": 9.6282958984375e-06,
      "model_forward_time": 1.740257740020752,
      "step": 84096
    },
    {
      "epoch": 9.6282958984375e-06,
      "step": 84096,
      "training_step_time": 7.826754093170166
    },
    {
      "epoch": 9.629249572753906e-06,
      "model_forward_time": 2.3337953090667725,
      "step": 84097
    },
    {
      "epoch": 9.629249572753906e-06,
      "step": 84097,
      "training_step_time": 9.450045347213745
    },
    {
      "epoch": 9.630203247070313e-06,
      "model_forward_time": 1.7193694114685059,
      "step": 84098
    },
    {
      "epoch": 9.630203247070313e-06,
      "step": 84098,
      "training_step_time": 7.920039653778076
    },
    {
      "epoch": 9.631156921386719e-06,
      "model_forward_time": 1.7271950244903564,
      "step": 84099
    },
    {
      "epoch": 9.631156921386719e-06,
      "step": 84099,
      "training_step_time": 7.7512078285217285
    },
    {
      "epoch": 9.632110595703125e-06,
      "grad_norm": 0.07290336489677429,
      "learning_rate": 6.754780570392433e-07,
      "loss": 0.0577,
      "step": 84100
    },
    {
      "epoch": 9.632110595703125e-06,
      "model_forward_time": 1.7206318378448486,
      "step": 84100
    },
    {
      "dynamics_loss_avg": 0.04941472318023443,
      "epoch": 9.632110595703125e-06,
      "step": 84100
    },
    {
      "action_loss_avg": 0.001292353423195891,
      "epoch": 9.632110595703125e-06,
      "step": 84100
    },
    {
      "epoch": 9.632110595703125e-06,
      "step": 84100,
      "training_step_time": 7.817591428756714
    },
    {
      "epoch": 9.633064270019532e-06,
      "model_forward_time": 1.7236225605010986,
      "step": 84101
    },
    {
      "epoch": 9.633064270019532e-06,
      "step": 84101,
      "training_step_time": 7.8643505573272705
    },
    {
      "epoch": 9.634017944335938e-06,
      "model_forward_time": 1.7340755462646484,
      "step": 84102
    },
    {
      "epoch": 9.634017944335938e-06,
      "step": 84102,
      "training_step_time": 7.853871822357178
    },
    {
      "epoch": 9.634971618652344e-06,
      "model_forward_time": 1.727034330368042,
      "step": 84103
    },
    {
      "epoch": 9.634971618652344e-06,
      "step": 84103,
      "training_step_time": 6.839695453643799
    },
    {
      "epoch": 9.63592529296875e-06,
      "model_forward_time": 1.72391676902771,
      "step": 84104
    },
    {
      "epoch": 9.63592529296875e-06,
      "step": 84104,
      "training_step_time": 6.840248107910156
    },
    {
      "epoch": 9.636878967285157e-06,
      "model_forward_time": 1.7260253429412842,
      "step": 84105
    },
    {
      "epoch": 9.636878967285157e-06,
      "step": 84105,
      "training_step_time": 9.316962718963623
    },
    {
      "epoch": 9.637832641601562e-06,
      "model_forward_time": 1.7577791213989258,
      "step": 84106
    },
    {
      "epoch": 9.637832641601562e-06,
      "step": 84106,
      "training_step_time": 7.880645513534546
    },
    {
      "epoch": 9.638786315917968e-06,
      "model_forward_time": 2.3475654125213623,
      "step": 84107
    },
    {
      "epoch": 9.638786315917968e-06,
      "step": 84107,
      "training_step_time": 8.029035568237305
    },
    {
      "epoch": 9.639739990234376e-06,
      "model_forward_time": 1.728071928024292,
      "step": 84108
    },
    {
      "epoch": 9.639739990234376e-06,
      "step": 84108,
      "training_step_time": 7.513317584991455
    },
    {
      "epoch": 9.640693664550781e-06,
      "model_forward_time": 1.7301950454711914,
      "step": 84109
    },
    {
      "epoch": 9.640693664550781e-06,
      "step": 84109,
      "training_step_time": 7.709293842315674
    },
    {
      "epoch": 9.641647338867187e-06,
      "grad_norm": 0.07902407646179199,
      "learning_rate": 6.746483564181422e-07,
      "loss": 0.0563,
      "step": 84110
    },
    {
      "epoch": 9.641647338867187e-06,
      "model_forward_time": 1.7138442993164062,
      "step": 84110
    },
    {
      "dynamics_loss_avg": 0.05255656568333507,
      "epoch": 9.641647338867187e-06,
      "step": 84110
    },
    {
      "action_loss_avg": 0.0012307968747336418,
      "epoch": 9.641647338867187e-06,
      "step": 84110
    },
    {
      "epoch": 9.641647338867187e-06,
      "step": 84110,
      "training_step_time": 8.315839290618896
    },
    {
      "epoch": 9.642601013183594e-06,
      "model_forward_time": 1.7770240306854248,
      "step": 84111
    },
    {
      "epoch": 9.642601013183594e-06,
      "step": 84111,
      "training_step_time": 7.856608152389526
    },
    {
      "epoch": 9.6435546875e-06,
      "model_forward_time": 1.7740859985351562,
      "step": 84112
    },
    {
      "epoch": 9.6435546875e-06,
      "step": 84112,
      "training_step_time": 6.8475494384765625
    },
    {
      "epoch": 9.644508361816406e-06,
      "model_forward_time": 1.712812900543213,
      "step": 84113
    },
    {
      "epoch": 9.644508361816406e-06,
      "step": 84113,
      "training_step_time": 6.902436971664429
    },
    {
      "epoch": 9.645462036132813e-06,
      "model_forward_time": 1.7668640613555908,
      "step": 84114
    },
    {
      "epoch": 9.645462036132813e-06,
      "step": 84114,
      "training_step_time": 8.118792057037354
    },
    {
      "epoch": 9.646415710449219e-06,
      "model_forward_time": 1.7942118644714355,
      "step": 84115
    },
    {
      "epoch": 9.646415710449219e-06,
      "step": 84115,
      "training_step_time": 8.536519765853882
    },
    {
      "epoch": 9.647369384765625e-06,
      "model_forward_time": 2.433255672454834,
      "step": 84116
    },
    {
      "epoch": 9.647369384765625e-06,
      "step": 84116,
      "training_step_time": 8.891339778900146
    },
    {
      "epoch": 9.648323059082032e-06,
      "model_forward_time": 1.7726233005523682,
      "step": 84117
    },
    {
      "epoch": 9.648323059082032e-06,
      "step": 84117,
      "training_step_time": 7.758869647979736
    },
    {
      "epoch": 9.649276733398438e-06,
      "model_forward_time": 1.778480052947998,
      "step": 84118
    },
    {
      "epoch": 9.649276733398438e-06,
      "step": 84118,
      "training_step_time": 7.6707541942596436
    },
    {
      "epoch": 9.650230407714844e-06,
      "model_forward_time": 1.777740478515625,
      "step": 84119
    },
    {
      "epoch": 9.650230407714844e-06,
      "step": 84119,
      "training_step_time": 7.8288798332214355
    },
    {
      "epoch": 9.65118408203125e-06,
      "grad_norm": 0.09485069662332535,
      "learning_rate": 6.738191288109952e-07,
      "loss": 0.0572,
      "step": 84120
    },
    {
      "epoch": 9.65118408203125e-06,
      "model_forward_time": 1.7729663848876953,
      "step": 84120
    },
    {
      "dynamics_loss_avg": 0.049855054914951326,
      "epoch": 9.65118408203125e-06,
      "step": 84120
    },
    {
      "action_loss_avg": 0.0015478339046239853,
      "epoch": 9.65118408203125e-06,
      "step": 84120
    },
    {
      "epoch": 9.65118408203125e-06,
      "step": 84120,
      "training_step_time": 8.368155717849731
    },
    {
      "epoch": 9.652137756347657e-06,
      "model_forward_time": 1.770073652267456,
      "step": 84121
    },
    {
      "epoch": 9.652137756347657e-06,
      "step": 84121,
      "training_step_time": 6.91121506690979
    },
    {
      "epoch": 9.653091430664062e-06,
      "model_forward_time": 1.7853288650512695,
      "step": 84122
    },
    {
      "epoch": 9.653091430664062e-06,
      "step": 84122,
      "training_step_time": 6.887613534927368
    },
    {
      "epoch": 9.654045104980468e-06,
      "model_forward_time": 1.7812905311584473,
      "step": 84123
    },
    {
      "epoch": 9.654045104980468e-06,
      "step": 84123,
      "training_step_time": 7.556593894958496
    },
    {
      "epoch": 9.654998779296876e-06,
      "model_forward_time": 1.783229112625122,
      "step": 84124
    },
    {
      "epoch": 9.654998779296876e-06,
      "step": 84124,
      "training_step_time": 7.803955078125
    },
    {
      "epoch": 9.655952453613281e-06,
      "model_forward_time": 2.439176321029663,
      "step": 84125
    },
    {
      "epoch": 9.655952453613281e-06,
      "step": 84125,
      "training_step_time": 9.611798763275146
    },
    {
      "epoch": 9.656906127929687e-06,
      "model_forward_time": 1.7772877216339111,
      "step": 84126
    },
    {
      "epoch": 9.656906127929687e-06,
      "step": 84126,
      "training_step_time": 7.765677452087402
    },
    {
      "epoch": 9.657859802246094e-06,
      "model_forward_time": 1.7710583209991455,
      "step": 84127
    },
    {
      "epoch": 9.657859802246094e-06,
      "step": 84127,
      "training_step_time": 7.4907612800598145
    },
    {
      "epoch": 9.6588134765625e-06,
      "model_forward_time": 1.782726526260376,
      "step": 84128
    },
    {
      "epoch": 9.6588134765625e-06,
      "step": 84128,
      "training_step_time": 7.8748695850372314
    },
    {
      "epoch": 9.659767150878906e-06,
      "model_forward_time": 1.7825253009796143,
      "step": 84129
    },
    {
      "epoch": 9.659767150878906e-06,
      "step": 84129,
      "training_step_time": 7.760819435119629
    },
    {
      "epoch": 9.660720825195313e-06,
      "grad_norm": 0.08954932540655136,
      "learning_rate": 6.729903743084821e-07,
      "loss": 0.0578,
      "step": 84130
    },
    {
      "epoch": 9.660720825195313e-06,
      "model_forward_time": 1.7820079326629639,
      "step": 84130
    },
    {
      "dynamics_loss_avg": 0.05589164867997169,
      "epoch": 9.660720825195313e-06,
      "step": 84130
    },
    {
      "action_loss_avg": 0.0011522418877575546,
      "epoch": 9.660720825195313e-06,
      "step": 84130
    },
    {
      "epoch": 9.660720825195313e-06,
      "step": 84130,
      "training_step_time": 7.839605331420898
    },
    {
      "epoch": 9.661674499511719e-06,
      "model_forward_time": 1.7696549892425537,
      "step": 84131
    },
    {
      "epoch": 9.661674499511719e-06,
      "step": 84131,
      "training_step_time": 6.8769447803497314
    },
    {
      "epoch": 9.662628173828125e-06,
      "model_forward_time": 1.7709031105041504,
      "step": 84132
    },
    {
      "epoch": 9.662628173828125e-06,
      "step": 84132,
      "training_step_time": 8.528608322143555
    },
    {
      "epoch": 9.66358184814453e-06,
      "model_forward_time": 1.803098201751709,
      "step": 84133
    },
    {
      "epoch": 9.66358184814453e-06,
      "step": 84133,
      "training_step_time": 8.789305448532104
    },
    {
      "epoch": 9.664535522460938e-06,
      "model_forward_time": 2.472994089126587,
      "step": 84134
    },
    {
      "epoch": 9.664535522460938e-06,
      "step": 84134,
      "training_step_time": 8.87444019317627
    },
    {
      "epoch": 9.665489196777344e-06,
      "model_forward_time": 1.7847566604614258,
      "step": 84135
    },
    {
      "epoch": 9.665489196777344e-06,
      "step": 84135,
      "training_step_time": 8.411015748977661
    },
    {
      "epoch": 9.66644287109375e-06,
      "model_forward_time": 1.774280071258545,
      "step": 84136
    },
    {
      "epoch": 9.66644287109375e-06,
      "step": 84136,
      "training_step_time": 7.455487012863159
    },
    {
      "epoch": 9.667396545410157e-06,
      "model_forward_time": 1.7842609882354736,
      "step": 84137
    },
    {
      "epoch": 9.667396545410157e-06,
      "step": 84137,
      "training_step_time": 7.741447925567627
    },
    {
      "epoch": 9.668350219726562e-06,
      "model_forward_time": 1.384082555770874,
      "step": 84138
    },
    {
      "epoch": 9.668350219726562e-06,
      "step": 84138,
      "training_step_time": 7.5980870723724365
    },
    {
      "epoch": 9.669303894042968e-06,
      "model_forward_time": 1.7732834815979004,
      "step": 84139
    },
    {
      "epoch": 9.669303894042968e-06,
      "step": 84139,
      "training_step_time": 7.928753614425659
    },
    {
      "epoch": 9.670257568359376e-06,
      "grad_norm": 0.12196467816829681,
      "learning_rate": 6.72162093001238e-07,
      "loss": 0.0567,
      "step": 84140
    },
    {
      "epoch": 9.670257568359376e-06,
      "model_forward_time": 1.7084178924560547,
      "step": 84140
    },
    {
      "dynamics_loss_avg": 0.04922131486237049,
      "epoch": 9.670257568359376e-06,
      "step": 84140
    },
    {
      "action_loss_avg": 0.0011101672454969958,
      "epoch": 9.670257568359376e-06,
      "step": 84140
    },
    {
      "epoch": 9.670257568359376e-06,
      "step": 84140,
      "training_step_time": 6.815954208374023
    },
    {
      "epoch": 9.671211242675781e-06,
      "model_forward_time": 1.7841315269470215,
      "step": 84141
    },
    {
      "epoch": 9.671211242675781e-06,
      "step": 84141,
      "training_step_time": 7.45012640953064
    },
    {
      "epoch": 9.672164916992187e-06,
      "model_forward_time": 1.7874114513397217,
      "step": 84142
    },
    {
      "epoch": 9.672164916992187e-06,
      "step": 84142,
      "training_step_time": 8.300446033477783
    },
    {
      "epoch": 9.673118591308594e-06,
      "model_forward_time": 2.397185802459717,
      "step": 84143
    },
    {
      "epoch": 9.673118591308594e-06,
      "step": 84143,
      "training_step_time": 10.96860671043396
    },
    {
      "epoch": 9.674072265625e-06,
      "model_forward_time": 1.7679462432861328,
      "step": 84144
    },
    {
      "epoch": 9.674072265625e-06,
      "step": 84144,
      "training_step_time": 8.842376947402954
    },
    {
      "epoch": 9.675025939941406e-06,
      "model_forward_time": 1.770705223083496,
      "step": 84145
    },
    {
      "epoch": 9.675025939941406e-06,
      "step": 84145,
      "training_step_time": 7.622548580169678
    },
    {
      "epoch": 9.675979614257813e-06,
      "model_forward_time": 1.783813714981079,
      "step": 84146
    },
    {
      "epoch": 9.675979614257813e-06,
      "step": 84146,
      "training_step_time": 7.617650747299194
    },
    {
      "epoch": 9.676933288574219e-06,
      "model_forward_time": 1.7657601833343506,
      "step": 84147
    },
    {
      "epoch": 9.676933288574219e-06,
      "step": 84147,
      "training_step_time": 7.577195644378662
    },
    {
      "epoch": 9.677886962890625e-06,
      "model_forward_time": 1.7739651203155518,
      "step": 84148
    },
    {
      "epoch": 9.677886962890625e-06,
      "step": 84148,
      "training_step_time": 7.954875469207764
    },
    {
      "epoch": 9.67884063720703e-06,
      "model_forward_time": 1.7843551635742188,
      "step": 84149
    },
    {
      "epoch": 9.67884063720703e-06,
      "step": 84149,
      "training_step_time": 6.836549758911133
    },
    {
      "epoch": 9.679794311523438e-06,
      "grad_norm": 0.07317794859409332,
      "learning_rate": 6.713342849798393e-07,
      "loss": 0.0574,
      "step": 84150
    },
    {
      "epoch": 9.679794311523438e-06,
      "model_forward_time": 1.7907617092132568,
      "step": 84150
    },
    {
      "dynamics_loss_avg": 0.05704620648175478,
      "epoch": 9.679794311523438e-06,
      "step": 84150
    },
    {
      "action_loss_avg": 0.0014093406454776414,
      "epoch": 9.679794311523438e-06,
      "step": 84150
    },
    {
      "epoch": 9.679794311523438e-06,
      "step": 84150,
      "training_step_time": 7.430349111557007
    },
    {
      "epoch": 9.680747985839844e-06,
      "model_forward_time": 1.7233328819274902,
      "step": 84151
    },
    {
      "epoch": 9.680747985839844e-06,
      "step": 84151,
      "training_step_time": 9.609297037124634
    },
    {
      "epoch": 9.68170166015625e-06,
      "model_forward_time": 1.755354642868042,
      "step": 84152
    },
    {
      "epoch": 9.68170166015625e-06,
      "step": 84152,
      "training_step_time": 9.682579278945923
    },
    {
      "epoch": 9.682655334472657e-06,
      "model_forward_time": 1.979609489440918,
      "step": 84153
    },
    {
      "epoch": 9.682655334472657e-06,
      "step": 84153,
      "training_step_time": 8.354089736938477
    },
    {
      "epoch": 9.683609008789062e-06,
      "model_forward_time": 1.7503762245178223,
      "step": 84154
    },
    {
      "epoch": 9.683609008789062e-06,
      "step": 84154,
      "training_step_time": 7.650103330612183
    },
    {
      "epoch": 9.684562683105468e-06,
      "model_forward_time": 1.7421298027038574,
      "step": 84155
    },
    {
      "epoch": 9.684562683105468e-06,
      "step": 84155,
      "training_step_time": 7.585330247879028
    },
    {
      "epoch": 9.685516357421876e-06,
      "model_forward_time": 1.7526302337646484,
      "step": 84156
    },
    {
      "epoch": 9.685516357421876e-06,
      "step": 84156,
      "training_step_time": 7.489513635635376
    },
    {
      "epoch": 9.686470031738281e-06,
      "model_forward_time": 1.7461802959442139,
      "step": 84157
    },
    {
      "epoch": 9.686470031738281e-06,
      "step": 84157,
      "training_step_time": 8.529375314712524
    },
    {
      "epoch": 9.687423706054687e-06,
      "model_forward_time": 1.7533962726593018,
      "step": 84158
    },
    {
      "epoch": 9.687423706054687e-06,
      "step": 84158,
      "training_step_time": 6.75773811340332
    },
    {
      "epoch": 9.688377380371094e-06,
      "model_forward_time": 1.7437753677368164,
      "step": 84159
    },
    {
      "epoch": 9.688377380371094e-06,
      "step": 84159,
      "training_step_time": 7.428468227386475
    },
    {
      "epoch": 9.6893310546875e-06,
      "grad_norm": 0.08981513231992722,
      "learning_rate": 6.705069503348167e-07,
      "loss": 0.0573,
      "step": 84160
    },
    {
      "epoch": 9.6893310546875e-06,
      "model_forward_time": 1.740044355392456,
      "step": 84160
    },
    {
      "dynamics_loss_avg": 0.04955050647258759,
      "epoch": 9.6893310546875e-06,
      "step": 84160
    },
    {
      "action_loss_avg": 0.001624016385176219,
      "epoch": 9.6893310546875e-06,
      "step": 84160
    },
    {
      "epoch": 9.6893310546875e-06,
      "step": 84160,
      "training_step_time": 7.562624931335449
    },
    {
      "epoch": 9.690284729003906e-06,
      "model_forward_time": 1.8394582271575928,
      "step": 84161
    },
    {
      "epoch": 9.690284729003906e-06,
      "step": 84161,
      "training_step_time": 9.057311296463013
    },
    {
      "epoch": 9.691238403320313e-06,
      "model_forward_time": 2.3827826976776123,
      "step": 84162
    },
    {
      "epoch": 9.691238403320313e-06,
      "step": 84162,
      "training_step_time": 8.445618867874146
    },
    {
      "epoch": 9.692192077636719e-06,
      "model_forward_time": 1.7457354068756104,
      "step": 84163
    },
    {
      "epoch": 9.692192077636719e-06,
      "step": 84163,
      "training_step_time": 8.237473011016846
    },
    {
      "epoch": 9.693145751953125e-06,
      "model_forward_time": 1.7510552406311035,
      "step": 84164
    },
    {
      "epoch": 9.693145751953125e-06,
      "step": 84164,
      "training_step_time": 7.627719879150391
    },
    {
      "epoch": 9.69409942626953e-06,
      "model_forward_time": 1.742105484008789,
      "step": 84165
    },
    {
      "epoch": 9.69409942626953e-06,
      "step": 84165,
      "training_step_time": 7.599037408828735
    },
    {
      "epoch": 9.695053100585938e-06,
      "model_forward_time": 1.7389709949493408,
      "step": 84166
    },
    {
      "epoch": 9.695053100585938e-06,
      "step": 84166,
      "training_step_time": 7.618919372558594
    },
    {
      "epoch": 9.696006774902344e-06,
      "model_forward_time": 1.745051622390747,
      "step": 84167
    },
    {
      "epoch": 9.696006774902344e-06,
      "step": 84167,
      "training_step_time": 7.748500108718872
    },
    {
      "epoch": 9.69696044921875e-06,
      "model_forward_time": 1.7548279762268066,
      "step": 84168
    },
    {
      "epoch": 9.69696044921875e-06,
      "step": 84168,
      "training_step_time": 7.401455402374268
    },
    {
      "epoch": 9.697914123535157e-06,
      "model_forward_time": 1.7489330768585205,
      "step": 84169
    },
    {
      "epoch": 9.697914123535157e-06,
      "step": 84169,
      "training_step_time": 7.707075834274292
    },
    {
      "epoch": 9.698867797851562e-06,
      "grad_norm": 0.07932110130786896,
      "learning_rate": 6.696800891566435e-07,
      "loss": 0.0575,
      "step": 84170
    },
    {
      "epoch": 9.698867797851562e-06,
      "model_forward_time": 1.8165538311004639,
      "step": 84170
    },
    {
      "dynamics_loss_avg": 0.0483336066827178,
      "epoch": 9.698867797851562e-06,
      "step": 84170
    },
    {
      "action_loss_avg": 0.0009974742424674332,
      "epoch": 9.698867797851562e-06,
      "step": 84170
    },
    {
      "epoch": 9.698867797851562e-06,
      "step": 84170,
      "training_step_time": 8.718369960784912
    },
    {
      "epoch": 9.699821472167968e-06,
      "model_forward_time": 2.3979597091674805,
      "step": 84171
    },
    {
      "epoch": 9.699821472167968e-06,
      "step": 84171,
      "training_step_time": 10.070327520370483
    },
    {
      "epoch": 9.700775146484376e-06,
      "model_forward_time": 1.7271292209625244,
      "step": 84172
    },
    {
      "epoch": 9.700775146484376e-06,
      "step": 84172,
      "training_step_time": 7.753748416900635
    },
    {
      "epoch": 9.701728820800781e-06,
      "model_forward_time": 1.3504841327667236,
      "step": 84173
    },
    {
      "epoch": 9.701728820800781e-06,
      "step": 84173,
      "training_step_time": 7.6767258644104
    },
    {
      "epoch": 9.702682495117187e-06,
      "model_forward_time": 1.7263357639312744,
      "step": 84174
    },
    {
      "epoch": 9.702682495117187e-06,
      "step": 84174,
      "training_step_time": 7.573918581008911
    },
    {
      "epoch": 9.703636169433594e-06,
      "model_forward_time": 1.7311460971832275,
      "step": 84175
    },
    {
      "epoch": 9.703636169433594e-06,
      "step": 84175,
      "training_step_time": 7.630631446838379
    },
    {
      "epoch": 9.70458984375e-06,
      "model_forward_time": 1.728698492050171,
      "step": 84176
    },
    {
      "epoch": 9.70458984375e-06,
      "step": 84176,
      "training_step_time": 7.822441339492798
    },
    {
      "epoch": 9.705543518066406e-06,
      "model_forward_time": 1.7283415794372559,
      "step": 84177
    },
    {
      "epoch": 9.705543518066406e-06,
      "step": 84177,
      "training_step_time": 7.436002492904663
    },
    {
      "epoch": 9.706497192382813e-06,
      "model_forward_time": 1.7270278930664062,
      "step": 84178
    },
    {
      "epoch": 9.706497192382813e-06,
      "step": 84178,
      "training_step_time": 7.707269668579102
    },
    {
      "epoch": 9.707450866699219e-06,
      "model_forward_time": 1.8112974166870117,
      "step": 84179
    },
    {
      "epoch": 9.707450866699219e-06,
      "step": 84179,
      "training_step_time": 8.80856990814209
    },
    {
      "epoch": 9.708404541015625e-06,
      "grad_norm": 0.07724035531282425,
      "learning_rate": 6.688537015357449e-07,
      "loss": 0.0583,
      "step": 84180
    },
    {
      "epoch": 9.708404541015625e-06,
      "model_forward_time": 1.7222206592559814,
      "step": 84180
    },
    {
      "dynamics_loss_avg": 0.05366406161338091,
      "epoch": 9.708404541015625e-06,
      "step": 84180
    },
    {
      "action_loss_avg": 0.0012217081588460133,
      "epoch": 9.708404541015625e-06,
      "step": 84180
    },
    {
      "epoch": 9.708404541015625e-06,
      "step": 84180,
      "training_step_time": 13.685652256011963
    },
    {
      "epoch": 9.70935821533203e-06,
      "model_forward_time": 1.7307820320129395,
      "step": 84181
    },
    {
      "epoch": 9.70935821533203e-06,
      "step": 84181,
      "training_step_time": 9.097017765045166
    },
    {
      "epoch": 9.710311889648438e-06,
      "model_forward_time": 1.7311060428619385,
      "step": 84182
    },
    {
      "epoch": 9.710311889648438e-06,
      "step": 84182,
      "training_step_time": 7.686987400054932
    },
    {
      "epoch": 9.711265563964844e-06,
      "model_forward_time": 1.7231497764587402,
      "step": 84183
    },
    {
      "epoch": 9.711265563964844e-06,
      "step": 84183,
      "training_step_time": 7.581827402114868
    },
    {
      "epoch": 9.71221923828125e-06,
      "model_forward_time": 1.7244842052459717,
      "step": 84184
    },
    {
      "epoch": 9.71221923828125e-06,
      "step": 84184,
      "training_step_time": 7.54987096786499
    },
    {
      "epoch": 9.713172912597657e-06,
      "model_forward_time": 1.7236144542694092,
      "step": 84185
    },
    {
      "epoch": 9.713172912597657e-06,
      "step": 84185,
      "training_step_time": 8.040103912353516
    },
    {
      "epoch": 9.714126586914062e-06,
      "model_forward_time": 1.3707287311553955,
      "step": 84186
    },
    {
      "epoch": 9.714126586914062e-06,
      "step": 84186,
      "training_step_time": 7.395061016082764
    },
    {
      "epoch": 9.715080261230468e-06,
      "model_forward_time": 1.7418580055236816,
      "step": 84187
    },
    {
      "epoch": 9.715080261230468e-06,
      "step": 84187,
      "training_step_time": 7.569235324859619
    },
    {
      "epoch": 9.716033935546876e-06,
      "model_forward_time": 1.8210077285766602,
      "step": 84188
    },
    {
      "epoch": 9.716033935546876e-06,
      "step": 84188,
      "training_step_time": 7.702881097793579
    },
    {
      "epoch": 9.716987609863281e-06,
      "model_forward_time": 1.7335927486419678,
      "step": 84189
    },
    {
      "epoch": 9.716987609863281e-06,
      "step": 84189,
      "training_step_time": 11.139638423919678
    },
    {
      "epoch": 9.717941284179687e-06,
      "grad_norm": 0.09629333019256592,
      "learning_rate": 6.68027787562494e-07,
      "loss": 0.0576,
      "step": 84190
    },
    {
      "epoch": 9.717941284179687e-06,
      "model_forward_time": 2.3772780895233154,
      "step": 84190
    },
    {
      "dynamics_loss_avg": 0.058634744584560396,
      "epoch": 9.717941284179687e-06,
      "step": 84190
    },
    {
      "action_loss_avg": 0.0015727302292361856,
      "epoch": 9.717941284179687e-06,
      "step": 84190
    },
    {
      "epoch": 9.717941284179687e-06,
      "step": 84190,
      "training_step_time": 9.208170652389526
    },
    {
      "epoch": 9.718894958496094e-06,
      "model_forward_time": 1.740328073501587,
      "step": 84191
    },
    {
      "epoch": 9.718894958496094e-06,
      "step": 84191,
      "training_step_time": 7.631035089492798
    },
    {
      "epoch": 9.7198486328125e-06,
      "model_forward_time": 1.7365493774414062,
      "step": 84192
    },
    {
      "epoch": 9.7198486328125e-06,
      "step": 84192,
      "training_step_time": 7.560929536819458
    },
    {
      "epoch": 9.720802307128906e-06,
      "model_forward_time": 1.737783432006836,
      "step": 84193
    },
    {
      "epoch": 9.720802307128906e-06,
      "step": 84193,
      "training_step_time": 7.532914400100708
    },
    {
      "epoch": 9.721755981445313e-06,
      "model_forward_time": 1.7426600456237793,
      "step": 84194
    },
    {
      "epoch": 9.721755981445313e-06,
      "step": 84194,
      "training_step_time": 7.3869311809539795
    },
    {
      "epoch": 9.722709655761719e-06,
      "model_forward_time": 1.737518548965454,
      "step": 84195
    },
    {
      "epoch": 9.722709655761719e-06,
      "step": 84195,
      "training_step_time": 7.387434482574463
    },
    {
      "epoch": 9.723663330078125e-06,
      "model_forward_time": 1.7395203113555908,
      "step": 84196
    },
    {
      "epoch": 9.723663330078125e-06,
      "step": 84196,
      "training_step_time": 8.258765935897827
    },
    {
      "epoch": 9.72461700439453e-06,
      "model_forward_time": 1.7806153297424316,
      "step": 84197
    },
    {
      "epoch": 9.72461700439453e-06,
      "step": 84197,
      "training_step_time": 9.807678461074829
    },
    {
      "epoch": 9.725570678710938e-06,
      "model_forward_time": 1.7338802814483643,
      "step": 84198
    },
    {
      "epoch": 9.725570678710938e-06,
      "step": 84198,
      "training_step_time": 10.05935287475586
    },
    {
      "epoch": 9.726524353027344e-06,
      "model_forward_time": 1.6157479286193848,
      "step": 84199
    },
    {
      "epoch": 9.726524353027344e-06,
      "step": 84199,
      "training_step_time": 8.518455266952515
    },
    {
      "epoch": 9.72747802734375e-06,
      "grad_norm": 0.09301591664552689,
      "learning_rate": 6.672023473272121e-07,
      "loss": 0.0561,
      "step": 84200
    },
    {
      "epoch": 9.72747802734375e-06,
      "model_forward_time": 1.7342958450317383,
      "step": 84200
    },
    {
      "dynamics_loss_avg": 0.04887449592351913,
      "epoch": 9.72747802734375e-06,
      "step": 84200
    },
    {
      "action_loss_avg": 0.0019818856788333507,
      "epoch": 9.72747802734375e-06,
      "step": 84200
    },
    {
      "epoch": 9.72747802734375e-06,
      "step": 84200,
      "training_step_time": 8.268886804580688
    },
    {
      "epoch": 9.728431701660157e-06,
      "model_forward_time": 1.7433440685272217,
      "step": 84201
    },
    {
      "epoch": 9.728431701660157e-06,
      "step": 84201,
      "training_step_time": 7.705451726913452
    },
    {
      "epoch": 9.729385375976562e-06,
      "model_forward_time": 1.7774322032928467,
      "step": 84202
    },
    {
      "epoch": 9.729385375976562e-06,
      "step": 84202,
      "training_step_time": 7.547096252441406
    },
    {
      "epoch": 9.730339050292968e-06,
      "model_forward_time": 1.7546074390411377,
      "step": 84203
    },
    {
      "epoch": 9.730339050292968e-06,
      "step": 84203,
      "training_step_time": 7.560626983642578
    },
    {
      "epoch": 9.731292724609376e-06,
      "model_forward_time": 1.7462611198425293,
      "step": 84204
    },
    {
      "epoch": 9.731292724609376e-06,
      "step": 84204,
      "training_step_time": 7.739191055297852
    },
    {
      "epoch": 9.732246398925781e-06,
      "model_forward_time": 1.7682030200958252,
      "step": 84205
    },
    {
      "epoch": 9.732246398925781e-06,
      "step": 84205,
      "training_step_time": 7.731236696243286
    },
    {
      "epoch": 9.733200073242187e-06,
      "model_forward_time": 1.8311307430267334,
      "step": 84206
    },
    {
      "epoch": 9.733200073242187e-06,
      "step": 84206,
      "training_step_time": 8.803063154220581
    },
    {
      "epoch": 9.734153747558594e-06,
      "model_forward_time": 1.7508327960968018,
      "step": 84207
    },
    {
      "epoch": 9.734153747558594e-06,
      "step": 84207,
      "training_step_time": 8.82164216041565
    },
    {
      "epoch": 9.735107421875e-06,
      "model_forward_time": 1.7507715225219727,
      "step": 84208
    },
    {
      "epoch": 9.735107421875e-06,
      "step": 84208,
      "training_step_time": 8.931606769561768
    },
    {
      "epoch": 9.736061096191406e-06,
      "model_forward_time": 1.7525625228881836,
      "step": 84209
    },
    {
      "epoch": 9.736061096191406e-06,
      "step": 84209,
      "training_step_time": 7.770991086959839
    },
    {
      "epoch": 9.737014770507813e-06,
      "grad_norm": 0.08290842175483704,
      "learning_rate": 6.663773809201668e-07,
      "loss": 0.0568,
      "step": 84210
    },
    {
      "epoch": 9.737014770507813e-06,
      "model_forward_time": 1.772050380706787,
      "step": 84210
    },
    {
      "dynamics_loss_avg": 0.04997676443308592,
      "epoch": 9.737014770507813e-06,
      "step": 84210
    },
    {
      "action_loss_avg": 0.0011978118694969454,
      "epoch": 9.737014770507813e-06,
      "step": 84210
    },
    {
      "epoch": 9.737014770507813e-06,
      "step": 84210,
      "training_step_time": 7.82297682762146
    },
    {
      "epoch": 9.737968444824219e-06,
      "model_forward_time": 1.75295090675354,
      "step": 84211
    },
    {
      "epoch": 9.737968444824219e-06,
      "step": 84211,
      "training_step_time": 7.7565598487854
    },
    {
      "epoch": 9.738922119140625e-06,
      "model_forward_time": 1.757863998413086,
      "step": 84212
    },
    {
      "epoch": 9.738922119140625e-06,
      "step": 84212,
      "training_step_time": 7.191351890563965
    },
    {
      "epoch": 9.73987579345703e-06,
      "model_forward_time": 1.7588956356048584,
      "step": 84213
    },
    {
      "epoch": 9.73987579345703e-06,
      "step": 84213,
      "training_step_time": 7.7030439376831055
    },
    {
      "epoch": 9.740829467773438e-06,
      "model_forward_time": 1.7495648860931396,
      "step": 84214
    },
    {
      "epoch": 9.740829467773438e-06,
      "step": 84214,
      "training_step_time": 7.731558561325073
    },
    {
      "epoch": 9.741783142089844e-06,
      "model_forward_time": 1.7753968238830566,
      "step": 84215
    },
    {
      "epoch": 9.741783142089844e-06,
      "step": 84215,
      "training_step_time": 7.776256084442139
    },
    {
      "epoch": 9.74273681640625e-06,
      "model_forward_time": 1.0043718814849854,
      "step": 84216
    },
    {
      "epoch": 9.74273681640625e-06,
      "step": 84216,
      "training_step_time": 8.61675763130188
    },
    {
      "epoch": 9.743690490722657e-06,
      "model_forward_time": 1.7468361854553223,
      "step": 84217
    },
    {
      "epoch": 9.743690490722657e-06,
      "step": 84217,
      "training_step_time": 10.714439630508423
    },
    {
      "epoch": 9.744644165039062e-06,
      "model_forward_time": 2.4066050052642822,
      "step": 84218
    },
    {
      "epoch": 9.744644165039062e-06,
      "step": 84218,
      "training_step_time": 7.765118598937988
    },
    {
      "epoch": 9.745597839355468e-06,
      "model_forward_time": 1.7157952785491943,
      "step": 84219
    },
    {
      "epoch": 9.745597839355468e-06,
      "step": 84219,
      "training_step_time": 7.680025815963745
    },
    {
      "epoch": 9.746551513671876e-06,
      "grad_norm": 0.08388319611549377,
      "learning_rate": 6.655528884315748e-07,
      "loss": 0.0545,
      "step": 84220
    },
    {
      "epoch": 9.746551513671876e-06,
      "model_forward_time": 1.7416768074035645,
      "step": 84220
    },
    {
      "dynamics_loss_avg": 0.04843354057520628,
      "epoch": 9.746551513671876e-06,
      "step": 84220
    },
    {
      "action_loss_avg": 0.0013946353254141286,
      "epoch": 9.746551513671876e-06,
      "step": 84220
    },
    {
      "epoch": 9.746551513671876e-06,
      "step": 84220,
      "training_step_time": 7.732249736785889
    },
    {
      "epoch": 9.747505187988281e-06,
      "model_forward_time": 1.7409353256225586,
      "step": 84221
    },
    {
      "epoch": 9.747505187988281e-06,
      "step": 84221,
      "training_step_time": 7.1346211433410645
    },
    {
      "epoch": 9.748458862304687e-06,
      "model_forward_time": 1.7517735958099365,
      "step": 84222
    },
    {
      "epoch": 9.748458862304687e-06,
      "step": 84222,
      "training_step_time": 7.702524900436401
    },
    {
      "epoch": 9.749412536621094e-06,
      "model_forward_time": 1.7607345581054688,
      "step": 84223
    },
    {
      "epoch": 9.749412536621094e-06,
      "step": 84223,
      "training_step_time": 7.779095888137817
    },
    {
      "epoch": 9.7503662109375e-06,
      "model_forward_time": 1.739746332168579,
      "step": 84224
    },
    {
      "epoch": 9.7503662109375e-06,
      "step": 84224,
      "training_step_time": 9.45231819152832
    },
    {
      "epoch": 9.751319885253906e-06,
      "model_forward_time": 1.7456157207489014,
      "step": 84225
    },
    {
      "epoch": 9.751319885253906e-06,
      "step": 84225,
      "training_step_time": 7.77158522605896
    },
    {
      "epoch": 9.752273559570313e-06,
      "model_forward_time": 1.7411432266235352,
      "step": 84226
    },
    {
      "epoch": 9.752273559570313e-06,
      "step": 84226,
      "training_step_time": 8.48021912574768
    },
    {
      "epoch": 9.753227233886719e-06,
      "model_forward_time": 2.3807766437530518,
      "step": 84227
    },
    {
      "epoch": 9.753227233886719e-06,
      "step": 84227,
      "training_step_time": 9.90241527557373
    },
    {
      "epoch": 9.754180908203125e-06,
      "model_forward_time": 1.74202299118042,
      "step": 84228
    },
    {
      "epoch": 9.754180908203125e-06,
      "step": 84228,
      "training_step_time": 7.789121627807617
    },
    {
      "epoch": 9.75513458251953e-06,
      "model_forward_time": 1.7386581897735596,
      "step": 84229
    },
    {
      "epoch": 9.75513458251953e-06,
      "step": 84229,
      "training_step_time": 7.862564563751221
    },
    {
      "epoch": 9.756088256835938e-06,
      "grad_norm": 0.07426337152719498,
      "learning_rate": 6.647288699516036e-07,
      "loss": 0.0547,
      "step": 84230
    },
    {
      "epoch": 9.756088256835938e-06,
      "model_forward_time": 1.751002550125122,
      "step": 84230
    },
    {
      "dynamics_loss_avg": 0.055001712031662466,
      "epoch": 9.756088256835938e-06,
      "step": 84230
    },
    {
      "action_loss_avg": 0.00174984309123829,
      "epoch": 9.756088256835938e-06,
      "step": 84230
    },
    {
      "epoch": 9.756088256835938e-06,
      "step": 84230,
      "training_step_time": 7.16536545753479
    },
    {
      "epoch": 9.757041931152344e-06,
      "model_forward_time": 1.7598934173583984,
      "step": 84231
    },
    {
      "epoch": 9.757041931152344e-06,
      "step": 84231,
      "training_step_time": 7.665184497833252
    },
    {
      "epoch": 9.75799560546875e-06,
      "model_forward_time": 1.3673007488250732,
      "step": 84232
    },
    {
      "epoch": 9.75799560546875e-06,
      "step": 84232,
      "training_step_time": 7.700910329818726
    },
    {
      "epoch": 9.758949279785157e-06,
      "model_forward_time": 1.741776704788208,
      "step": 84233
    },
    {
      "epoch": 9.758949279785157e-06,
      "step": 84233,
      "training_step_time": 7.700906276702881
    },
    {
      "epoch": 9.759902954101562e-06,
      "model_forward_time": 1.8314130306243896,
      "step": 84234
    },
    {
      "epoch": 9.759902954101562e-06,
      "step": 84234,
      "training_step_time": 7.7489378452301025
    },
    {
      "epoch": 9.760856628417968e-06,
      "model_forward_time": 1.74092698097229,
      "step": 84235
    },
    {
      "epoch": 9.760856628417968e-06,
      "step": 84235,
      "training_step_time": 9.317795753479004
    },
    {
      "epoch": 9.761810302734376e-06,
      "model_forward_time": 1.7402822971343994,
      "step": 84236
    },
    {
      "epoch": 9.761810302734376e-06,
      "step": 84236,
      "training_step_time": 7.858688831329346
    },
    {
      "epoch": 9.762763977050781e-06,
      "model_forward_time": 2.3522026538848877,
      "step": 84237
    },
    {
      "epoch": 9.762763977050781e-06,
      "step": 84237,
      "training_step_time": 7.76224422454834
    },
    {
      "epoch": 9.763717651367187e-06,
      "model_forward_time": 1.7461931705474854,
      "step": 84238
    },
    {
      "epoch": 9.763717651367187e-06,
      "step": 84238,
      "training_step_time": 7.7575178146362305
    },
    {
      "epoch": 9.764671325683594e-06,
      "model_forward_time": 1.7371814250946045,
      "step": 84239
    },
    {
      "epoch": 9.764671325683594e-06,
      "step": 84239,
      "training_step_time": 8.260544061660767
    },
    {
      "epoch": 9.765625e-06,
      "grad_norm": 0.11437669396400452,
      "learning_rate": 6.639053255703642e-07,
      "loss": 0.0544,
      "step": 84240
    },
    {
      "epoch": 9.765625e-06,
      "model_forward_time": 1.746894121170044,
      "step": 84240
    },
    {
      "dynamics_loss_avg": 0.04621211728081107,
      "epoch": 9.765625e-06,
      "step": 84240
    },
    {
      "action_loss_avg": 0.0011419666872825474,
      "epoch": 9.765625e-06,
      "step": 84240
    },
    {
      "epoch": 9.765625e-06,
      "step": 84240,
      "training_step_time": 8.079230308532715
    },
    {
      "epoch": 9.766578674316406e-06,
      "model_forward_time": 1.7408671379089355,
      "step": 84241
    },
    {
      "epoch": 9.766578674316406e-06,
      "step": 84241,
      "training_step_time": 7.550497055053711
    },
    {
      "epoch": 9.767532348632813e-06,
      "model_forward_time": 1.7397539615631104,
      "step": 84242
    },
    {
      "epoch": 9.767532348632813e-06,
      "step": 84242,
      "training_step_time": 7.379608869552612
    },
    {
      "epoch": 9.768486022949219e-06,
      "model_forward_time": 1.833641767501831,
      "step": 84243
    },
    {
      "epoch": 9.768486022949219e-06,
      "step": 84243,
      "training_step_time": 8.960952043533325
    },
    {
      "epoch": 9.769439697265625e-06,
      "model_forward_time": 1.740870475769043,
      "step": 84244
    },
    {
      "epoch": 9.769439697265625e-06,
      "step": 84244,
      "training_step_time": 8.407838106155396
    },
    {
      "epoch": 9.77039337158203e-06,
      "model_forward_time": 1.7432329654693604,
      "step": 84245
    },
    {
      "epoch": 9.77039337158203e-06,
      "step": 84245,
      "training_step_time": 8.890005350112915
    },
    {
      "epoch": 9.771347045898438e-06,
      "model_forward_time": 2.3563451766967773,
      "step": 84246
    },
    {
      "epoch": 9.771347045898438e-06,
      "step": 84246,
      "training_step_time": 8.346467018127441
    },
    {
      "epoch": 9.772300720214844e-06,
      "model_forward_time": 1.7356884479522705,
      "step": 84247
    },
    {
      "epoch": 9.772300720214844e-06,
      "step": 84247,
      "training_step_time": 7.668140649795532
    },
    {
      "epoch": 9.77325439453125e-06,
      "model_forward_time": 1.756631851196289,
      "step": 84248
    },
    {
      "epoch": 9.77325439453125e-06,
      "step": 84248,
      "training_step_time": 7.542909145355225
    },
    {
      "epoch": 9.774208068847657e-06,
      "model_forward_time": 1.7591707706451416,
      "step": 84249
    },
    {
      "epoch": 9.774208068847657e-06,
      "step": 84249,
      "training_step_time": 6.989546298980713
    },
    {
      "epoch": 9.775161743164062e-06,
      "grad_norm": 0.12171271443367004,
      "learning_rate": 6.630822553779193e-07,
      "loss": 0.0551,
      "step": 84250
    },
    {
      "epoch": 9.775161743164062e-06,
      "model_forward_time": 1.7539749145507812,
      "step": 84250
    },
    {
      "dynamics_loss_avg": 0.042728794179856776,
      "epoch": 9.775161743164062e-06,
      "step": 84250
    },
    {
      "action_loss_avg": 0.0007850393274566158,
      "epoch": 9.775161743164062e-06,
      "step": 84250
    },
    {
      "epoch": 9.775161743164062e-06,
      "step": 84250,
      "training_step_time": 7.535125494003296
    },
    {
      "epoch": 9.776115417480468e-06,
      "model_forward_time": 1.7500085830688477,
      "step": 84251
    },
    {
      "epoch": 9.776115417480468e-06,
      "step": 84251,
      "training_step_time": 7.422020196914673
    },
    {
      "epoch": 9.777069091796876e-06,
      "model_forward_time": 1.8120410442352295,
      "step": 84252
    },
    {
      "epoch": 9.777069091796876e-06,
      "step": 84252,
      "training_step_time": 7.640881299972534
    },
    {
      "epoch": 9.778022766113281e-06,
      "model_forward_time": 1.7489352226257324,
      "step": 84253
    },
    {
      "epoch": 9.778022766113281e-06,
      "step": 84253,
      "training_step_time": 8.451722383499146
    },
    {
      "epoch": 9.778976440429687e-06,
      "model_forward_time": 1.7459220886230469,
      "step": 84254
    },
    {
      "epoch": 9.778976440429687e-06,
      "step": 84254,
      "training_step_time": 8.474640369415283
    },
    {
      "epoch": 9.779930114746094e-06,
      "model_forward_time": 2.3976218700408936,
      "step": 84255
    },
    {
      "epoch": 9.779930114746094e-06,
      "step": 84255,
      "training_step_time": 9.319885730743408
    },
    {
      "epoch": 9.7808837890625e-06,
      "model_forward_time": 1.750363826751709,
      "step": 84256
    },
    {
      "epoch": 9.7808837890625e-06,
      "step": 84256,
      "training_step_time": 7.635291337966919
    },
    {
      "epoch": 9.781837463378906e-06,
      "model_forward_time": 1.748563289642334,
      "step": 84257
    },
    {
      "epoch": 9.781837463378906e-06,
      "step": 84257,
      "training_step_time": 7.579598903656006
    },
    {
      "epoch": 9.782791137695312e-06,
      "model_forward_time": 1.771639108657837,
      "step": 84258
    },
    {
      "epoch": 9.782791137695312e-06,
      "step": 84258,
      "training_step_time": 7.078994989395142
    },
    {
      "epoch": 9.783744812011719e-06,
      "model_forward_time": 1.754610538482666,
      "step": 84259
    },
    {
      "epoch": 9.783744812011719e-06,
      "step": 84259,
      "training_step_time": 7.585621118545532
    },
    {
      "epoch": 9.784698486328125e-06,
      "grad_norm": 0.08167939633131027,
      "learning_rate": 6.622596594642777e-07,
      "loss": 0.0563,
      "step": 84260
    },
    {
      "epoch": 9.784698486328125e-06,
      "model_forward_time": 1.7508244514465332,
      "step": 84260
    },
    {
      "dynamics_loss_avg": 0.06197841828688979,
      "epoch": 9.784698486328125e-06,
      "step": 84260
    },
    {
      "action_loss_avg": 0.0017396470939274877,
      "epoch": 9.784698486328125e-06,
      "step": 84260
    },
    {
      "epoch": 9.784698486328125e-06,
      "step": 84260,
      "training_step_time": 7.385671377182007
    },
    {
      "epoch": 9.78565216064453e-06,
      "model_forward_time": 1.7880651950836182,
      "step": 84261
    },
    {
      "epoch": 9.78565216064453e-06,
      "step": 84261,
      "training_step_time": 9.530282020568848
    },
    {
      "epoch": 9.786605834960938e-06,
      "model_forward_time": 1.7492499351501465,
      "step": 84262
    },
    {
      "epoch": 9.786605834960938e-06,
      "step": 84262,
      "training_step_time": 9.469670057296753
    },
    {
      "epoch": 9.787559509277344e-06,
      "model_forward_time": 1.760331630706787,
      "step": 84263
    },
    {
      "epoch": 9.787559509277344e-06,
      "step": 84263,
      "training_step_time": 8.279411554336548
    },
    {
      "epoch": 9.78851318359375e-06,
      "model_forward_time": 2.3844730854034424,
      "step": 84264
    },
    {
      "epoch": 9.78851318359375e-06,
      "step": 84264,
      "training_step_time": 7.6081321239471436
    },
    {
      "epoch": 9.789466857910157e-06,
      "model_forward_time": 1.7507336139678955,
      "step": 84265
    },
    {
      "epoch": 9.789466857910157e-06,
      "step": 84265,
      "training_step_time": 7.719496011734009
    },
    {
      "epoch": 9.790420532226563e-06,
      "model_forward_time": 1.7638657093048096,
      "step": 84266
    },
    {
      "epoch": 9.790420532226563e-06,
      "step": 84266,
      "training_step_time": 7.532073497772217
    },
    {
      "epoch": 9.791374206542968e-06,
      "model_forward_time": 1.7593955993652344,
      "step": 84267
    },
    {
      "epoch": 9.791374206542968e-06,
      "step": 84267,
      "training_step_time": 7.032044410705566
    },
    {
      "epoch": 9.792327880859376e-06,
      "model_forward_time": 1.7667396068572998,
      "step": 84268
    },
    {
      "epoch": 9.792327880859376e-06,
      "step": 84268,
      "training_step_time": 7.697141647338867
    },
    {
      "epoch": 9.793281555175781e-06,
      "model_forward_time": 1.751500129699707,
      "step": 84269
    },
    {
      "epoch": 9.793281555175781e-06,
      "step": 84269,
      "training_step_time": 7.5092854499816895
    },
    {
      "epoch": 9.794235229492187e-06,
      "grad_norm": 0.08943825215101242,
      "learning_rate": 6.614375379193999e-07,
      "loss": 0.0549,
      "step": 84270
    },
    {
      "epoch": 9.794235229492187e-06,
      "model_forward_time": 1.750425100326538,
      "step": 84270
    },
    {
      "dynamics_loss_avg": 0.05247639622539282,
      "epoch": 9.794235229492187e-06,
      "step": 84270
    },
    {
      "action_loss_avg": 0.0014422702603042125,
      "epoch": 9.794235229492187e-06,
      "step": 84270
    },
    {
      "epoch": 9.794235229492187e-06,
      "step": 84270,
      "training_step_time": 8.598111629486084
    },
    {
      "epoch": 9.795188903808595e-06,
      "model_forward_time": 1.7690110206604004,
      "step": 84271
    },
    {
      "epoch": 9.795188903808595e-06,
      "step": 84271,
      "training_step_time": 7.68228554725647
    },
    {
      "epoch": 9.796142578125e-06,
      "model_forward_time": 1.7463290691375732,
      "step": 84272
    },
    {
      "epoch": 9.796142578125e-06,
      "step": 84272,
      "training_step_time": 10.174286603927612
    },
    {
      "epoch": 9.797096252441406e-06,
      "model_forward_time": 2.3752176761627197,
      "step": 84273
    },
    {
      "epoch": 9.797096252441406e-06,
      "step": 84273,
      "training_step_time": 8.687628030776978
    },
    {
      "epoch": 9.798049926757812e-06,
      "model_forward_time": 1.7638936042785645,
      "step": 84274
    },
    {
      "epoch": 9.798049926757812e-06,
      "step": 84274,
      "training_step_time": 8.696436166763306
    },
    {
      "epoch": 9.799003601074219e-06,
      "model_forward_time": 1.7575147151947021,
      "step": 84275
    },
    {
      "epoch": 9.799003601074219e-06,
      "step": 84275,
      "training_step_time": 7.3435163497924805
    },
    {
      "epoch": 9.799957275390625e-06,
      "model_forward_time": 1.7560102939605713,
      "step": 84276
    },
    {
      "epoch": 9.799957275390625e-06,
      "step": 84276,
      "training_step_time": 7.551863193511963
    },
    {
      "epoch": 9.80091094970703e-06,
      "model_forward_time": 1.758631944656372,
      "step": 84277
    },
    {
      "epoch": 9.80091094970703e-06,
      "step": 84277,
      "training_step_time": 7.671662330627441
    },
    {
      "epoch": 9.801864624023438e-06,
      "model_forward_time": 1.7587530612945557,
      "step": 84278
    },
    {
      "epoch": 9.801864624023438e-06,
      "step": 84278,
      "training_step_time": 7.398439884185791
    },
    {
      "epoch": 9.802818298339844e-06,
      "model_forward_time": 1.7592175006866455,
      "step": 84279
    },
    {
      "epoch": 9.802818298339844e-06,
      "step": 84279,
      "training_step_time": 7.577927112579346
    },
    {
      "epoch": 9.80377197265625e-06,
      "grad_norm": 0.0903845801949501,
      "learning_rate": 6.606158908331888e-07,
      "loss": 0.055,
      "step": 84280
    },
    {
      "epoch": 9.80377197265625e-06,
      "model_forward_time": 1.8422093391418457,
      "step": 84280
    },
    {
      "dynamics_loss_avg": 0.06913121193647384,
      "epoch": 9.80377197265625e-06,
      "step": 84280
    },
    {
      "action_loss_avg": 0.0013932471425505354,
      "epoch": 9.80377197265625e-06,
      "step": 84280
    },
    {
      "epoch": 9.80377197265625e-06,
      "step": 84280,
      "training_step_time": 9.40711760520935
    },
    {
      "epoch": 9.804725646972657e-06,
      "model_forward_time": 1.7757346630096436,
      "step": 84281
    },
    {
      "epoch": 9.804725646972657e-06,
      "step": 84281,
      "training_step_time": 13.054147481918335
    },
    {
      "epoch": 9.805679321289063e-06,
      "model_forward_time": 2.880650281906128,
      "step": 84282
    },
    {
      "epoch": 9.805679321289063e-06,
      "step": 84282,
      "training_step_time": 8.507066488265991
    },
    {
      "epoch": 9.806632995605468e-06,
      "model_forward_time": 1.7940850257873535,
      "step": 84283
    },
    {
      "epoch": 9.806632995605468e-06,
      "step": 84283,
      "training_step_time": 9.662910461425781
    },
    {
      "epoch": 9.807586669921876e-06,
      "model_forward_time": 1.7986886501312256,
      "step": 84284
    },
    {
      "epoch": 9.807586669921876e-06,
      "step": 84284,
      "training_step_time": 8.56975245475769
    },
    {
      "epoch": 9.808540344238281e-06,
      "model_forward_time": 1.802292823791504,
      "step": 84285
    },
    {
      "epoch": 9.808540344238281e-06,
      "step": 84285,
      "training_step_time": 8.072412729263306
    },
    {
      "epoch": 9.809494018554687e-06,
      "model_forward_time": 1.8014729022979736,
      "step": 84286
    },
    {
      "epoch": 9.809494018554687e-06,
      "step": 84286,
      "training_step_time": 7.572855472564697
    },
    {
      "epoch": 9.810447692871095e-06,
      "model_forward_time": 1.798861026763916,
      "step": 84287
    },
    {
      "epoch": 9.810447692871095e-06,
      "step": 84287,
      "training_step_time": 7.5680084228515625
    },
    {
      "epoch": 9.8114013671875e-06,
      "model_forward_time": 1.8002488613128662,
      "step": 84288
    },
    {
      "epoch": 9.8114013671875e-06,
      "step": 84288,
      "training_step_time": 7.382460355758667
    },
    {
      "epoch": 9.812355041503906e-06,
      "model_forward_time": 1.8213920593261719,
      "step": 84289
    },
    {
      "epoch": 9.812355041503906e-06,
      "step": 84289,
      "training_step_time": 7.648578643798828
    },
    {
      "epoch": 9.813308715820312e-06,
      "grad_norm": 0.0968749150633812,
      "learning_rate": 6.597947182955006e-07,
      "loss": 0.0541,
      "step": 84290
    },
    {
      "epoch": 9.813308715820312e-06,
      "model_forward_time": 1.809474229812622,
      "step": 84290
    },
    {
      "dynamics_loss_avg": 0.041595191694796085,
      "epoch": 9.813308715820312e-06,
      "step": 84290
    },
    {
      "action_loss_avg": 0.0008998132208944299,
      "epoch": 9.813308715820312e-06,
      "step": 84290
    },
    {
      "epoch": 9.813308715820312e-06,
      "step": 84290,
      "training_step_time": 8.484060287475586
    },
    {
      "epoch": 9.814262390136719e-06,
      "model_forward_time": 2.857367992401123,
      "step": 84291
    },
    {
      "epoch": 9.814262390136719e-06,
      "step": 84291,
      "training_step_time": 8.310862064361572
    },
    {
      "epoch": 9.815216064453125e-06,
      "model_forward_time": 1.7951292991638184,
      "step": 84292
    },
    {
      "epoch": 9.815216064453125e-06,
      "step": 84292,
      "training_step_time": 9.226317644119263
    },
    {
      "epoch": 9.81616973876953e-06,
      "model_forward_time": 1.8019590377807617,
      "step": 84293
    },
    {
      "epoch": 9.81616973876953e-06,
      "step": 84293,
      "training_step_time": 7.565657377243042
    },
    {
      "epoch": 9.817123413085938e-06,
      "model_forward_time": 1.7946128845214844,
      "step": 84294
    },
    {
      "epoch": 9.817123413085938e-06,
      "step": 84294,
      "training_step_time": 7.558605432510376
    },
    {
      "epoch": 9.818077087402344e-06,
      "model_forward_time": 1.7990398406982422,
      "step": 84295
    },
    {
      "epoch": 9.818077087402344e-06,
      "step": 84295,
      "training_step_time": 7.562472105026245
    },
    {
      "epoch": 9.81903076171875e-06,
      "model_forward_time": 1.8039319515228271,
      "step": 84296
    },
    {
      "epoch": 9.81903076171875e-06,
      "step": 84296,
      "training_step_time": 7.569473028182983
    },
    {
      "epoch": 9.819984436035157e-06,
      "model_forward_time": 1.795253038406372,
      "step": 84297
    },
    {
      "epoch": 9.819984436035157e-06,
      "step": 84297,
      "training_step_time": 7.954287052154541
    },
    {
      "epoch": 9.820938110351563e-06,
      "model_forward_time": 1.7993807792663574,
      "step": 84298
    },
    {
      "epoch": 9.820938110351563e-06,
      "step": 84298,
      "training_step_time": 7.60210394859314
    },
    {
      "epoch": 9.821891784667968e-06,
      "model_forward_time": 1.7960569858551025,
      "step": 84299
    },
    {
      "epoch": 9.821891784667968e-06,
      "step": 84299,
      "training_step_time": 9.35216760635376
    },
    {
      "epoch": 9.822845458984376e-06,
      "grad_norm": 0.11888433992862701,
      "learning_rate": 6.589740203961359e-07,
      "loss": 0.0561,
      "step": 84300
    },
    {
      "epoch": 9.822845458984376e-06,
      "model_forward_time": 2.8481576442718506,
      "step": 84300
    },
    {
      "dynamics_loss_avg": 0.04250948205590248,
      "epoch": 9.822845458984376e-06,
      "step": 84300
    },
    {
      "action_loss_avg": 0.0010541433686739766,
      "epoch": 9.822845458984376e-06,
      "step": 84300
    },
    {
      "epoch": 9.822845458984376e-06,
      "step": 84300,
      "training_step_time": 8.484363794326782
    },
    {
      "epoch": 9.823799133300781e-06,
      "model_forward_time": 1.7940237522125244,
      "step": 84301
    },
    {
      "epoch": 9.823799133300781e-06,
      "step": 84301,
      "training_step_time": 9.699601411819458
    },
    {
      "epoch": 9.824752807617187e-06,
      "model_forward_time": 1.7975962162017822,
      "step": 84302
    },
    {
      "epoch": 9.824752807617187e-06,
      "step": 84302,
      "training_step_time": 7.6373291015625
    },
    {
      "epoch": 9.825706481933595e-06,
      "model_forward_time": 1.7976796627044678,
      "step": 84303
    },
    {
      "epoch": 9.825706481933595e-06,
      "step": 84303,
      "training_step_time": 7.353402137756348
    },
    {
      "epoch": 9.82666015625e-06,
      "model_forward_time": 1.7926502227783203,
      "step": 84304
    },
    {
      "epoch": 9.82666015625e-06,
      "step": 84304,
      "training_step_time": 7.517507076263428
    },
    {
      "epoch": 9.827613830566406e-06,
      "model_forward_time": 1.7892358303070068,
      "step": 84305
    },
    {
      "epoch": 9.827613830566406e-06,
      "step": 84305,
      "training_step_time": 7.591692209243774
    },
    {
      "epoch": 9.828567504882812e-06,
      "model_forward_time": 1.801175594329834,
      "step": 84306
    },
    {
      "epoch": 9.828567504882812e-06,
      "step": 84306,
      "training_step_time": 7.895724773406982
    },
    {
      "epoch": 9.829521179199219e-06,
      "model_forward_time": 1.7902402877807617,
      "step": 84307
    },
    {
      "epoch": 9.829521179199219e-06,
      "step": 84307,
      "training_step_time": 7.6060144901275635
    },
    {
      "epoch": 9.830474853515625e-06,
      "model_forward_time": 1.907914638519287,
      "step": 84308
    },
    {
      "epoch": 9.830474853515625e-06,
      "step": 84308,
      "training_step_time": 8.699828863143921
    },
    {
      "epoch": 9.83142852783203e-06,
      "model_forward_time": 2.8524022102355957,
      "step": 84309
    },
    {
      "epoch": 9.83142852783203e-06,
      "step": 84309,
      "training_step_time": 9.198082447052002
    },
    {
      "epoch": 9.832382202148438e-06,
      "grad_norm": 0.1001189574599266,
      "learning_rate": 6.581537972248464e-07,
      "loss": 0.0561,
      "step": 84310
    },
    {
      "epoch": 9.832382202148438e-06,
      "model_forward_time": 1.7924325466156006,
      "step": 84310
    },
    {
      "dynamics_loss_avg": 0.028064565360546113,
      "epoch": 9.832382202148438e-06,
      "step": 84310
    },
    {
      "action_loss_avg": 0.0009450338104215917,
      "epoch": 9.832382202148438e-06,
      "step": 84310
    },
    {
      "epoch": 9.832382202148438e-06,
      "step": 84310,
      "training_step_time": 7.622124195098877
    },
    {
      "epoch": 9.833335876464844e-06,
      "model_forward_time": 1.79819655418396,
      "step": 84311
    },
    {
      "epoch": 9.833335876464844e-06,
      "step": 84311,
      "training_step_time": 8.32379961013794
    },
    {
      "epoch": 9.83428955078125e-06,
      "model_forward_time": 1.8011584281921387,
      "step": 84312
    },
    {
      "epoch": 9.83428955078125e-06,
      "step": 84312,
      "training_step_time": 7.549983739852905
    },
    {
      "epoch": 9.835243225097657e-06,
      "model_forward_time": 1.7960331439971924,
      "step": 84313
    },
    {
      "epoch": 9.835243225097657e-06,
      "step": 84313,
      "training_step_time": 7.571044206619263
    },
    {
      "epoch": 9.836196899414063e-06,
      "model_forward_time": 1.7998912334442139,
      "step": 84314
    },
    {
      "epoch": 9.836196899414063e-06,
      "step": 84314,
      "training_step_time": 7.623752593994141
    },
    {
      "epoch": 9.837150573730468e-06,
      "model_forward_time": 1.7881662845611572,
      "step": 84315
    },
    {
      "epoch": 9.837150573730468e-06,
      "step": 84315,
      "training_step_time": 8.312821865081787
    },
    {
      "epoch": 9.838104248046876e-06,
      "model_forward_time": 1.787168025970459,
      "step": 84316
    },
    {
      "epoch": 9.838104248046876e-06,
      "step": 84316,
      "training_step_time": 8.479508399963379
    },
    {
      "epoch": 9.839057922363281e-06,
      "model_forward_time": 1.861332893371582,
      "step": 84317
    },
    {
      "epoch": 9.839057922363281e-06,
      "step": 84317,
      "training_step_time": 10.752091646194458
    },
    {
      "epoch": 9.840011596679687e-06,
      "model_forward_time": 2.866697072982788,
      "step": 84318
    },
    {
      "epoch": 9.840011596679687e-06,
      "step": 84318,
      "training_step_time": 9.503284215927124
    },
    {
      "epoch": 9.840965270996095e-06,
      "model_forward_time": 1.7683453559875488,
      "step": 84319
    },
    {
      "epoch": 9.840965270996095e-06,
      "step": 84319,
      "training_step_time": 7.706148862838745
    },
    {
      "epoch": 9.8419189453125e-06,
      "grad_norm": 0.07287720590829849,
      "learning_rate": 6.573340488713275e-07,
      "loss": 0.0553,
      "step": 84320
    },
    {
      "epoch": 9.8419189453125e-06,
      "model_forward_time": 1.4239881038665771,
      "step": 84320
    },
    {
      "dynamics_loss_avg": 0.04903717637062073,
      "epoch": 9.8419189453125e-06,
      "step": 84320
    },
    {
      "action_loss_avg": 0.0009263350657420233,
      "epoch": 9.8419189453125e-06,
      "step": 84320
    },
    {
      "epoch": 9.8419189453125e-06,
      "step": 84320,
      "training_step_time": 8.76860761642456
    },
    {
      "epoch": 9.842872619628906e-06,
      "model_forward_time": 1.7823512554168701,
      "step": 84321
    },
    {
      "epoch": 9.842872619628906e-06,
      "step": 84321,
      "training_step_time": 7.56897497177124
    },
    {
      "epoch": 9.843826293945312e-06,
      "model_forward_time": 1.792952299118042,
      "step": 84322
    },
    {
      "epoch": 9.843826293945312e-06,
      "step": 84322,
      "training_step_time": 7.667159795761108
    },
    {
      "epoch": 9.844779968261719e-06,
      "model_forward_time": 1.3975646495819092,
      "step": 84323
    },
    {
      "epoch": 9.844779968261719e-06,
      "step": 84323,
      "training_step_time": 7.523159742355347
    },
    {
      "epoch": 9.845733642578125e-06,
      "model_forward_time": 1.7822215557098389,
      "step": 84324
    },
    {
      "epoch": 9.845733642578125e-06,
      "step": 84324,
      "training_step_time": 7.828782320022583
    },
    {
      "epoch": 9.84668731689453e-06,
      "model_forward_time": 1.7924487590789795,
      "step": 84325
    },
    {
      "epoch": 9.84668731689453e-06,
      "step": 84325,
      "training_step_time": 7.594650030136108
    },
    {
      "epoch": 9.847640991210938e-06,
      "model_forward_time": 1.8347814083099365,
      "step": 84326
    },
    {
      "epoch": 9.847640991210938e-06,
      "step": 84326,
      "training_step_time": 8.130407333374023
    },
    {
      "epoch": 9.848594665527344e-06,
      "model_forward_time": 1.788994312286377,
      "step": 84327
    },
    {
      "epoch": 9.848594665527344e-06,
      "step": 84327,
      "training_step_time": 9.661823987960815
    },
    {
      "epoch": 9.84954833984375e-06,
      "model_forward_time": 1.7965900897979736,
      "step": 84328
    },
    {
      "epoch": 9.84954833984375e-06,
      "step": 84328,
      "training_step_time": 9.078693151473999
    },
    {
      "epoch": 9.850502014160157e-06,
      "model_forward_time": 1.7741522789001465,
      "step": 84329
    },
    {
      "epoch": 9.850502014160157e-06,
      "step": 84329,
      "training_step_time": 7.6167402267456055
    },
    {
      "epoch": 9.851455688476563e-06,
      "grad_norm": 0.07274330407381058,
      "learning_rate": 6.565147754252304e-07,
      "loss": 0.0553,
      "step": 84330
    },
    {
      "epoch": 9.851455688476563e-06,
      "model_forward_time": 1.7981340885162354,
      "step": 84330
    },
    {
      "dynamics_loss_avg": 0.06420378684997559,
      "epoch": 9.851455688476563e-06,
      "step": 84330
    },
    {
      "action_loss_avg": 0.001399083499563858,
      "epoch": 9.851455688476563e-06,
      "step": 84330
    },
    {
      "epoch": 9.851455688476563e-06,
      "step": 84330,
      "training_step_time": 8.587969064712524
    },
    {
      "epoch": 9.852409362792968e-06,
      "model_forward_time": 1.7869079113006592,
      "step": 84331
    },
    {
      "epoch": 9.852409362792968e-06,
      "step": 84331,
      "training_step_time": 7.660717010498047
    },
    {
      "epoch": 9.853363037109376e-06,
      "model_forward_time": 1.7807316780090332,
      "step": 84332
    },
    {
      "epoch": 9.853363037109376e-06,
      "step": 84332,
      "training_step_time": 8.019912481307983
    },
    {
      "epoch": 9.854316711425781e-06,
      "model_forward_time": 1.7877161502838135,
      "step": 84333
    },
    {
      "epoch": 9.854316711425781e-06,
      "step": 84333,
      "training_step_time": 7.0877296924591064
    },
    {
      "epoch": 9.855270385742187e-06,
      "model_forward_time": 1.7856636047363281,
      "step": 84334
    },
    {
      "epoch": 9.855270385742187e-06,
      "step": 84334,
      "training_step_time": 7.586606979370117
    },
    {
      "epoch": 9.856224060058595e-06,
      "model_forward_time": 1.8020298480987549,
      "step": 84335
    },
    {
      "epoch": 9.856224060058595e-06,
      "step": 84335,
      "training_step_time": 9.357201337814331
    },
    {
      "epoch": 9.857177734375e-06,
      "model_forward_time": 1.7952287197113037,
      "step": 84336
    },
    {
      "epoch": 9.857177734375e-06,
      "step": 84336,
      "training_step_time": 7.716012477874756
    },
    {
      "epoch": 9.858131408691406e-06,
      "model_forward_time": 2.832662582397461,
      "step": 84337
    },
    {
      "epoch": 9.858131408691406e-06,
      "step": 84337,
      "training_step_time": 8.256767511367798
    },
    {
      "epoch": 9.859085083007812e-06,
      "model_forward_time": 1.7836723327636719,
      "step": 84338
    },
    {
      "epoch": 9.859085083007812e-06,
      "step": 84338,
      "training_step_time": 8.425334930419922
    },
    {
      "epoch": 9.860038757324219e-06,
      "model_forward_time": 1.779604196548462,
      "step": 84339
    },
    {
      "epoch": 9.860038757324219e-06,
      "step": 84339,
      "training_step_time": 7.611615419387817
    },
    {
      "epoch": 9.860992431640625e-06,
      "grad_norm": 0.08550725132226944,
      "learning_rate": 6.556959769761462e-07,
      "loss": 0.0563,
      "step": 84340
    },
    {
      "epoch": 9.860992431640625e-06,
      "model_forward_time": 1.7834017276763916,
      "step": 84340
    },
    {
      "dynamics_loss_avg": 0.05420093443244696,
      "epoch": 9.860992431640625e-06,
      "step": 84340
    },
    {
      "action_loss_avg": 0.001238034086418338,
      "epoch": 9.860992431640625e-06,
      "step": 84340
    },
    {
      "epoch": 9.860992431640625e-06,
      "step": 84340,
      "training_step_time": 7.627458810806274
    },
    {
      "epoch": 9.86194610595703e-06,
      "model_forward_time": 1.7886121273040771,
      "step": 84341
    },
    {
      "epoch": 9.86194610595703e-06,
      "step": 84341,
      "training_step_time": 7.36480188369751
    },
    {
      "epoch": 9.862899780273438e-06,
      "model_forward_time": 1.4310617446899414,
      "step": 84342
    },
    {
      "epoch": 9.862899780273438e-06,
      "step": 84342,
      "training_step_time": 7.550339221954346
    },
    {
      "epoch": 9.863853454589844e-06,
      "model_forward_time": 1.7916584014892578,
      "step": 84343
    },
    {
      "epoch": 9.863853454589844e-06,
      "step": 84343,
      "training_step_time": 8.21847677230835
    },
    {
      "epoch": 9.86480712890625e-06,
      "model_forward_time": 1.785188913345337,
      "step": 84344
    },
    {
      "epoch": 9.86480712890625e-06,
      "step": 84344,
      "training_step_time": 8.263400793075562
    },
    {
      "epoch": 9.865760803222657e-06,
      "model_forward_time": 1.8872532844543457,
      "step": 84345
    },
    {
      "epoch": 9.865760803222657e-06,
      "step": 84345,
      "training_step_time": 9.569553852081299
    },
    {
      "epoch": 9.866714477539063e-06,
      "model_forward_time": 2.8225438594818115,
      "step": 84346
    },
    {
      "epoch": 9.866714477539063e-06,
      "step": 84346,
      "training_step_time": 7.610466957092285
    },
    {
      "epoch": 9.867668151855468e-06,
      "model_forward_time": 1.7867205142974854,
      "step": 84347
    },
    {
      "epoch": 9.867668151855468e-06,
      "step": 84347,
      "training_step_time": 8.644821643829346
    },
    {
      "epoch": 9.868621826171876e-06,
      "model_forward_time": 1.7871742248535156,
      "step": 84348
    },
    {
      "epoch": 9.868621826171876e-06,
      "step": 84348,
      "training_step_time": 7.584334373474121
    },
    {
      "epoch": 9.869575500488281e-06,
      "model_forward_time": 1.7886266708374023,
      "step": 84349
    },
    {
      "epoch": 9.869575500488281e-06,
      "step": 84349,
      "training_step_time": 7.576645851135254
    },
    {
      "epoch": 9.870529174804687e-06,
      "grad_norm": 0.07737398147583008,
      "learning_rate": 6.548776536136193e-07,
      "loss": 0.0563,
      "step": 84350
    },
    {
      "epoch": 9.870529174804687e-06,
      "model_forward_time": 1.771453619003296,
      "step": 84350
    },
    {
      "dynamics_loss_avg": 0.05070620831102133,
      "epoch": 9.870529174804687e-06,
      "step": 84350
    },
    {
      "action_loss_avg": 0.0012960641935933382,
      "epoch": 9.870529174804687e-06,
      "step": 84350
    },
    {
      "epoch": 9.870529174804687e-06,
      "step": 84350,
      "training_step_time": 7.433577060699463
    },
    {
      "epoch": 9.871482849121095e-06,
      "model_forward_time": 1.7692642211914062,
      "step": 84351
    },
    {
      "epoch": 9.871482849121095e-06,
      "step": 84351,
      "training_step_time": 7.648998975753784
    },
    {
      "epoch": 9.8724365234375e-06,
      "model_forward_time": 1.7889437675476074,
      "step": 84352
    },
    {
      "epoch": 9.8724365234375e-06,
      "step": 84352,
      "training_step_time": 7.526058197021484
    },
    {
      "epoch": 9.873390197753906e-06,
      "model_forward_time": 1.785038948059082,
      "step": 84353
    },
    {
      "epoch": 9.873390197753906e-06,
      "step": 84353,
      "training_step_time": 7.662081956863403
    },
    {
      "epoch": 9.874343872070312e-06,
      "model_forward_time": 1.8980281352996826,
      "step": 84354
    },
    {
      "epoch": 9.874343872070312e-06,
      "step": 84354,
      "training_step_time": 8.309754848480225
    },
    {
      "epoch": 9.87529754638672e-06,
      "model_forward_time": 2.8683950901031494,
      "step": 84355
    },
    {
      "epoch": 9.87529754638672e-06,
      "step": 84355,
      "training_step_time": 10.080369472503662
    },
    {
      "epoch": 9.876251220703125e-06,
      "model_forward_time": 1.7605490684509277,
      "step": 84356
    },
    {
      "epoch": 9.876251220703125e-06,
      "step": 84356,
      "training_step_time": 8.397319078445435
    },
    {
      "epoch": 9.87720489501953e-06,
      "model_forward_time": 1.7745490074157715,
      "step": 84357
    },
    {
      "epoch": 9.87720489501953e-06,
      "step": 84357,
      "training_step_time": 8.686647415161133
    },
    {
      "epoch": 9.878158569335938e-06,
      "model_forward_time": 1.7876710891723633,
      "step": 84358
    },
    {
      "epoch": 9.878158569335938e-06,
      "step": 84358,
      "training_step_time": 7.586486339569092
    },
    {
      "epoch": 9.879112243652344e-06,
      "model_forward_time": 1.7811410427093506,
      "step": 84359
    },
    {
      "epoch": 9.879112243652344e-06,
      "step": 84359,
      "training_step_time": 7.491814613342285
    },
    {
      "epoch": 9.88006591796875e-06,
      "grad_norm": 0.09456674009561539,
      "learning_rate": 6.54059805427138e-07,
      "loss": 0.056,
      "step": 84360
    },
    {
      "epoch": 9.88006591796875e-06,
      "model_forward_time": 1.7698981761932373,
      "step": 84360
    },
    {
      "dynamics_loss_avg": 0.05832398645579815,
      "epoch": 9.88006591796875e-06,
      "step": 84360
    },
    {
      "action_loss_avg": 0.0013854343807906844,
      "epoch": 9.88006591796875e-06,
      "step": 84360
    },
    {
      "epoch": 9.88006591796875e-06,
      "step": 84360,
      "training_step_time": 7.615598917007446
    },
    {
      "epoch": 9.881019592285157e-06,
      "model_forward_time": 1.7902112007141113,
      "step": 84361
    },
    {
      "epoch": 9.881019592285157e-06,
      "step": 84361,
      "training_step_time": 7.147879123687744
    },
    {
      "epoch": 9.881973266601563e-06,
      "model_forward_time": 1.781627893447876,
      "step": 84362
    },
    {
      "epoch": 9.881973266601563e-06,
      "step": 84362,
      "training_step_time": 8.188316106796265
    },
    {
      "epoch": 9.882926940917968e-06,
      "model_forward_time": 1.8468286991119385,
      "step": 84363
    },
    {
      "epoch": 9.882926940917968e-06,
      "step": 84363,
      "training_step_time": 8.553777933120728
    },
    {
      "epoch": 9.883880615234376e-06,
      "model_forward_time": 2.862758159637451,
      "step": 84364
    },
    {
      "epoch": 9.883880615234376e-06,
      "step": 84364,
      "training_step_time": 8.399997234344482
    },
    {
      "epoch": 9.884834289550781e-06,
      "model_forward_time": 1.8002071380615234,
      "step": 84365
    },
    {
      "epoch": 9.884834289550781e-06,
      "step": 84365,
      "training_step_time": 8.548489570617676
    },
    {
      "epoch": 9.885787963867187e-06,
      "model_forward_time": 1.8099560737609863,
      "step": 84366
    },
    {
      "epoch": 9.885787963867187e-06,
      "step": 84366,
      "training_step_time": 8.710840225219727
    },
    {
      "epoch": 9.886741638183595e-06,
      "model_forward_time": 1.8137106895446777,
      "step": 84367
    },
    {
      "epoch": 9.886741638183595e-06,
      "step": 84367,
      "training_step_time": 7.6499083042144775
    },
    {
      "epoch": 9.8876953125e-06,
      "model_forward_time": 1.8005733489990234,
      "step": 84368
    },
    {
      "epoch": 9.8876953125e-06,
      "step": 84368,
      "training_step_time": 7.608250617980957
    },
    {
      "epoch": 9.888648986816406e-06,
      "model_forward_time": 1.8152871131896973,
      "step": 84369
    },
    {
      "epoch": 9.888648986816406e-06,
      "step": 84369,
      "training_step_time": 7.638637542724609
    },
    {
      "epoch": 9.889602661132812e-06,
      "grad_norm": 0.07242032140493393,
      "learning_rate": 6.532424325061437e-07,
      "loss": 0.0555,
      "step": 84370
    },
    {
      "epoch": 9.889602661132812e-06,
      "model_forward_time": 1.8083837032318115,
      "step": 84370
    },
    {
      "dynamics_loss_avg": 0.05546165555715561,
      "epoch": 9.889602661132812e-06,
      "step": 84370
    },
    {
      "action_loss_avg": 0.0011104302597232164,
      "epoch": 9.889602661132812e-06,
      "step": 84370
    },
    {
      "epoch": 9.889602661132812e-06,
      "step": 84370,
      "training_step_time": 7.791288137435913
    },
    {
      "epoch": 9.89055633544922e-06,
      "model_forward_time": 1.80240797996521,
      "step": 84371
    },
    {
      "epoch": 9.89055633544922e-06,
      "step": 84371,
      "training_step_time": 7.615815162658691
    },
    {
      "epoch": 9.891510009765625e-06,
      "model_forward_time": 1.8517305850982666,
      "step": 84372
    },
    {
      "epoch": 9.891510009765625e-06,
      "step": 84372,
      "training_step_time": 9.189372062683105
    },
    {
      "epoch": 9.89246368408203e-06,
      "model_forward_time": 2.8894169330596924,
      "step": 84373
    },
    {
      "epoch": 9.89246368408203e-06,
      "step": 84373,
      "training_step_time": 8.854447364807129
    },
    {
      "epoch": 9.893417358398438e-06,
      "model_forward_time": 1.7964403629302979,
      "step": 84374
    },
    {
      "epoch": 9.893417358398438e-06,
      "step": 84374,
      "training_step_time": 9.909238576889038
    },
    {
      "epoch": 9.894371032714844e-06,
      "model_forward_time": 1.7977371215820312,
      "step": 84375
    },
    {
      "epoch": 9.894371032714844e-06,
      "step": 84375,
      "training_step_time": 7.677723407745361
    },
    {
      "epoch": 9.89532470703125e-06,
      "model_forward_time": 1.8030993938446045,
      "step": 84376
    },
    {
      "epoch": 9.89532470703125e-06,
      "step": 84376,
      "training_step_time": 7.731086015701294
    },
    {
      "epoch": 9.896278381347657e-06,
      "model_forward_time": 1.8056890964508057,
      "step": 84377
    },
    {
      "epoch": 9.896278381347657e-06,
      "step": 84377,
      "training_step_time": 7.461547374725342
    },
    {
      "epoch": 9.897232055664063e-06,
      "model_forward_time": 1.8013997077941895,
      "step": 84378
    },
    {
      "epoch": 9.897232055664063e-06,
      "step": 84378,
      "training_step_time": 7.565934419631958
    },
    {
      "epoch": 9.898185729980468e-06,
      "model_forward_time": 1.8114378452301025,
      "step": 84379
    },
    {
      "epoch": 9.898185729980468e-06,
      "step": 84379,
      "training_step_time": 7.657507419586182
    },
    {
      "epoch": 9.899139404296876e-06,
      "grad_norm": 0.09334353357553482,
      "learning_rate": 6.524255349400193e-07,
      "loss": 0.0563,
      "step": 84380
    },
    {
      "epoch": 9.899139404296876e-06,
      "model_forward_time": 1.7990143299102783,
      "step": 84380
    },
    {
      "dynamics_loss_avg": 0.05108438525348902,
      "epoch": 9.899139404296876e-06,
      "step": 84380
    },
    {
      "action_loss_avg": 0.0014178324570821132,
      "epoch": 9.899139404296876e-06,
      "step": 84380
    },
    {
      "epoch": 9.899139404296876e-06,
      "step": 84380,
      "training_step_time": 7.556196689605713
    },
    {
      "epoch": 9.900093078613281e-06,
      "model_forward_time": 1.8221771717071533,
      "step": 84381
    },
    {
      "epoch": 9.900093078613281e-06,
      "step": 84381,
      "training_step_time": 8.54393982887268
    },
    {
      "epoch": 9.901046752929687e-06,
      "model_forward_time": 1.8024909496307373,
      "step": 84382
    },
    {
      "epoch": 9.901046752929687e-06,
      "step": 84382,
      "training_step_time": 8.286683797836304
    },
    {
      "epoch": 9.902000427246093e-06,
      "model_forward_time": 1.8035497665405273,
      "step": 84383
    },
    {
      "epoch": 9.902000427246093e-06,
      "step": 84383,
      "training_step_time": 7.725666046142578
    },
    {
      "epoch": 9.9029541015625e-06,
      "model_forward_time": 1.8205296993255615,
      "step": 84384
    },
    {
      "epoch": 9.9029541015625e-06,
      "step": 84384,
      "training_step_time": 7.626632213592529
    },
    {
      "epoch": 9.903907775878906e-06,
      "model_forward_time": 1.8042747974395752,
      "step": 84385
    },
    {
      "epoch": 9.903907775878906e-06,
      "step": 84385,
      "training_step_time": 7.6996283531188965
    },
    {
      "epoch": 9.904861450195312e-06,
      "model_forward_time": 1.7975895404815674,
      "step": 84386
    },
    {
      "epoch": 9.904861450195312e-06,
      "step": 84386,
      "training_step_time": 8.089657545089722
    },
    {
      "epoch": 9.90581512451172e-06,
      "model_forward_time": 1.8114423751831055,
      "step": 84387
    },
    {
      "epoch": 9.90581512451172e-06,
      "step": 84387,
      "training_step_time": 7.601677656173706
    },
    {
      "epoch": 9.906768798828125e-06,
      "model_forward_time": 1.8111546039581299,
      "step": 84388
    },
    {
      "epoch": 9.906768798828125e-06,
      "step": 84388,
      "training_step_time": 7.643810033798218
    },
    {
      "epoch": 9.90772247314453e-06,
      "model_forward_time": 1.804441213607788,
      "step": 84389
    },
    {
      "epoch": 9.90772247314453e-06,
      "step": 84389,
      "training_step_time": 9.588301181793213
    },
    {
      "epoch": 9.908676147460938e-06,
      "grad_norm": 0.08876286447048187,
      "learning_rate": 6.516091128181046e-07,
      "loss": 0.0555,
      "step": 84390
    },
    {
      "epoch": 9.908676147460938e-06,
      "model_forward_time": 1.814948320388794,
      "step": 84390
    },
    {
      "dynamics_loss_avg": 0.06339948065578938,
      "epoch": 9.908676147460938e-06,
      "step": 84390
    },
    {
      "action_loss_avg": 0.0011853631207486615,
      "epoch": 9.908676147460938e-06,
      "step": 84390
    },
    {
      "epoch": 9.908676147460938e-06,
      "step": 84390,
      "training_step_time": 8.750894546508789
    },
    {
      "epoch": 9.909629821777344e-06,
      "model_forward_time": 1.8141553401947021,
      "step": 84391
    },
    {
      "epoch": 9.909629821777344e-06,
      "step": 84391,
      "training_step_time": 8.300185441970825
    },
    {
      "epoch": 9.91058349609375e-06,
      "model_forward_time": 2.8484716415405273,
      "step": 84392
    },
    {
      "epoch": 9.91058349609375e-06,
      "step": 84392,
      "training_step_time": 9.47372031211853
    },
    {
      "epoch": 9.911537170410157e-06,
      "model_forward_time": 1.8097481727600098,
      "step": 84393
    },
    {
      "epoch": 9.911537170410157e-06,
      "step": 84393,
      "training_step_time": 8.667285442352295
    },
    {
      "epoch": 9.912490844726563e-06,
      "model_forward_time": 1.805558443069458,
      "step": 84394
    },
    {
      "epoch": 9.912490844726563e-06,
      "step": 84394,
      "training_step_time": 7.5909788608551025
    },
    {
      "epoch": 9.913444519042968e-06,
      "model_forward_time": 1.8108491897583008,
      "step": 84395
    },
    {
      "epoch": 9.913444519042968e-06,
      "step": 84395,
      "training_step_time": 7.322723627090454
    },
    {
      "epoch": 9.914398193359376e-06,
      "model_forward_time": 1.4469187259674072,
      "step": 84396
    },
    {
      "epoch": 9.914398193359376e-06,
      "step": 84396,
      "training_step_time": 7.564168453216553
    },
    {
      "epoch": 9.915351867675781e-06,
      "model_forward_time": 1.7555301189422607,
      "step": 84397
    },
    {
      "epoch": 9.915351867675781e-06,
      "step": 84397,
      "training_step_time": 7.581189393997192
    },
    {
      "epoch": 9.916305541992187e-06,
      "model_forward_time": 1.7899861335754395,
      "step": 84398
    },
    {
      "epoch": 9.916305541992187e-06,
      "step": 84398,
      "training_step_time": 7.3987390995025635
    },
    {
      "epoch": 9.917259216308593e-06,
      "model_forward_time": 1.8115427494049072,
      "step": 84399
    },
    {
      "epoch": 9.917259216308593e-06,
      "step": 84399,
      "training_step_time": 8.330625295639038
    },
    {
      "epoch": 9.918212890625e-06,
      "grad_norm": 0.09479331225156784,
      "learning_rate": 6.507931662296785e-07,
      "loss": 0.0554,
      "step": 84400
    },
    {
      "epoch": 9.918212890625e-06,
      "model_forward_time": 1.9217286109924316,
      "step": 84400
    },
    {
      "dynamics_loss_avg": 0.05261856559664011,
      "epoch": 9.918212890625e-06,
      "step": 84400
    },
    {
      "action_loss_avg": 0.0014796168950852006,
      "epoch": 9.918212890625e-06,
      "step": 84400
    },
    {
      "epoch": 9.918212890625e-06,
      "step": 84400,
      "training_step_time": 8.385545492172241
    },
    {
      "epoch": 9.919166564941406e-06,
      "model_forward_time": 2.9219651222229004,
      "step": 84401
    },
    {
      "epoch": 9.919166564941406e-06,
      "step": 84401,
      "training_step_time": 8.774830341339111
    },
    {
      "epoch": 9.920120239257812e-06,
      "model_forward_time": 1.7942240238189697,
      "step": 84402
    },
    {
      "epoch": 9.920120239257812e-06,
      "step": 84402,
      "training_step_time": 7.623849391937256
    },
    {
      "epoch": 9.92107391357422e-06,
      "model_forward_time": 1.8072841167449951,
      "step": 84403
    },
    {
      "epoch": 9.92107391357422e-06,
      "step": 84403,
      "training_step_time": 7.646787881851196
    },
    {
      "epoch": 9.922027587890625e-06,
      "model_forward_time": 1.4580564498901367,
      "step": 84404
    },
    {
      "epoch": 9.922027587890625e-06,
      "step": 84404,
      "training_step_time": 7.628854036331177
    },
    {
      "epoch": 9.92298126220703e-06,
      "model_forward_time": 1.8150081634521484,
      "step": 84405
    },
    {
      "epoch": 9.92298126220703e-06,
      "step": 84405,
      "training_step_time": 7.435302019119263
    },
    {
      "epoch": 9.923934936523438e-06,
      "model_forward_time": 1.8079960346221924,
      "step": 84406
    },
    {
      "epoch": 9.923934936523438e-06,
      "step": 84406,
      "training_step_time": 7.518118858337402
    },
    {
      "epoch": 9.924888610839844e-06,
      "model_forward_time": 1.8054192066192627,
      "step": 84407
    },
    {
      "epoch": 9.924888610839844e-06,
      "step": 84407,
      "training_step_time": 7.332247972488403
    },
    {
      "epoch": 9.92584228515625e-06,
      "model_forward_time": 1.8160285949707031,
      "step": 84408
    },
    {
      "epoch": 9.92584228515625e-06,
      "step": 84408,
      "training_step_time": 8.432010173797607
    },
    {
      "epoch": 9.926795959472657e-06,
      "model_forward_time": 1.911181926727295,
      "step": 84409
    },
    {
      "epoch": 9.926795959472657e-06,
      "step": 84409,
      "training_step_time": 8.735038757324219
    },
    {
      "epoch": 9.927749633789063e-06,
      "grad_norm": 0.09824350476264954,
      "learning_rate": 6.499776952639736e-07,
      "loss": 0.0574,
      "step": 84410
    },
    {
      "epoch": 9.927749633789063e-06,
      "model_forward_time": 2.5285141468048096,
      "step": 84410
    },
    {
      "dynamics_loss_avg": 0.056819830741733315,
      "epoch": 9.927749633789063e-06,
      "step": 84410
    },
    {
      "action_loss_avg": 0.0014666703136754222,
      "epoch": 9.927749633789063e-06,
      "step": 84410
    },
    {
      "epoch": 9.927749633789063e-06,
      "step": 84410,
      "training_step_time": 8.61393690109253
    },
    {
      "epoch": 9.928703308105468e-06,
      "model_forward_time": 1.8011066913604736,
      "step": 84411
    },
    {
      "epoch": 9.928703308105468e-06,
      "step": 84411,
      "training_step_time": 9.095612287521362
    },
    {
      "epoch": 9.929656982421876e-06,
      "model_forward_time": 1.8147766590118408,
      "step": 84412
    },
    {
      "epoch": 9.929656982421876e-06,
      "step": 84412,
      "training_step_time": 8.36355209350586
    },
    {
      "epoch": 9.930610656738281e-06,
      "model_forward_time": 1.8279073238372803,
      "step": 84413
    },
    {
      "epoch": 9.930610656738281e-06,
      "step": 84413,
      "training_step_time": 7.655572891235352
    },
    {
      "epoch": 9.931564331054687e-06,
      "model_forward_time": 1.8118717670440674,
      "step": 84414
    },
    {
      "epoch": 9.931564331054687e-06,
      "step": 84414,
      "training_step_time": 7.381174802780151
    },
    {
      "epoch": 9.932518005371093e-06,
      "model_forward_time": 1.8077888488769531,
      "step": 84415
    },
    {
      "epoch": 9.932518005371093e-06,
      "step": 84415,
      "training_step_time": 7.529749870300293
    },
    {
      "epoch": 9.9334716796875e-06,
      "model_forward_time": 1.8134706020355225,
      "step": 84416
    },
    {
      "epoch": 9.9334716796875e-06,
      "step": 84416,
      "training_step_time": 7.29331111907959
    },
    {
      "epoch": 9.934425354003906e-06,
      "model_forward_time": 1.8074312210083008,
      "step": 84417
    },
    {
      "epoch": 9.934425354003906e-06,
      "step": 84417,
      "training_step_time": 8.459693908691406
    },
    {
      "epoch": 9.935379028320312e-06,
      "model_forward_time": 1.9012084007263184,
      "step": 84418
    },
    {
      "epoch": 9.935379028320312e-06,
      "step": 84418,
      "training_step_time": 8.411275625228882
    },
    {
      "epoch": 9.93633270263672e-06,
      "model_forward_time": 1.8016693592071533,
      "step": 84419
    },
    {
      "epoch": 9.93633270263672e-06,
      "step": 84419,
      "training_step_time": 9.040255308151245
    },
    {
      "epoch": 9.937286376953125e-06,
      "grad_norm": 0.09338043630123138,
      "learning_rate": 6.491627000101664e-07,
      "loss": 0.0549,
      "step": 84420
    },
    {
      "epoch": 9.937286376953125e-06,
      "model_forward_time": 1.7581322193145752,
      "step": 84420
    },
    {
      "dynamics_loss_avg": 0.05637202709913254,
      "epoch": 9.937286376953125e-06,
      "step": 84420
    },
    {
      "action_loss_avg": 0.001307277934392914,
      "epoch": 9.937286376953125e-06,
      "step": 84420
    },
    {
      "epoch": 9.937286376953125e-06,
      "step": 84420,
      "training_step_time": 8.735122442245483
    },
    {
      "epoch": 9.93824005126953e-06,
      "model_forward_time": 1.7895026206970215,
      "step": 84421
    },
    {
      "epoch": 9.93824005126953e-06,
      "step": 84421,
      "training_step_time": 8.576508283615112
    },
    {
      "epoch": 9.939193725585938e-06,
      "model_forward_time": 1.7807602882385254,
      "step": 84422
    },
    {
      "epoch": 9.939193725585938e-06,
      "step": 84422,
      "training_step_time": 7.674403667449951
    },
    {
      "epoch": 9.940147399902344e-06,
      "model_forward_time": 1.788320779800415,
      "step": 84423
    },
    {
      "epoch": 9.940147399902344e-06,
      "step": 84423,
      "training_step_time": 7.532233715057373
    },
    {
      "epoch": 9.94110107421875e-06,
      "model_forward_time": 1.7874913215637207,
      "step": 84424
    },
    {
      "epoch": 9.94110107421875e-06,
      "step": 84424,
      "training_step_time": 7.519856214523315
    },
    {
      "epoch": 9.942054748535157e-06,
      "model_forward_time": 1.7951102256774902,
      "step": 84425
    },
    {
      "epoch": 9.942054748535157e-06,
      "step": 84425,
      "training_step_time": 8.095645666122437
    },
    {
      "epoch": 9.943008422851563e-06,
      "model_forward_time": 1.7855124473571777,
      "step": 84426
    },
    {
      "epoch": 9.943008422851563e-06,
      "step": 84426,
      "training_step_time": 7.541895151138306
    },
    {
      "epoch": 9.943962097167968e-06,
      "model_forward_time": 1.844510793685913,
      "step": 84427
    },
    {
      "epoch": 9.943962097167968e-06,
      "step": 84427,
      "training_step_time": 9.261857271194458
    },
    {
      "epoch": 9.944915771484376e-06,
      "model_forward_time": 1.790254831314087,
      "step": 84428
    },
    {
      "epoch": 9.944915771484376e-06,
      "step": 84428,
      "training_step_time": 9.424547672271729
    },
    {
      "epoch": 9.945869445800782e-06,
      "model_forward_time": 2.821955442428589,
      "step": 84429
    },
    {
      "epoch": 9.945869445800782e-06,
      "step": 84429,
      "training_step_time": 9.514288425445557
    },
    {
      "epoch": 9.946823120117187e-06,
      "grad_norm": 0.08863590657711029,
      "learning_rate": 6.483481805573854e-07,
      "loss": 0.0559,
      "step": 84430
    },
    {
      "epoch": 9.946823120117187e-06,
      "model_forward_time": 1.777808427810669,
      "step": 84430
    },
    {
      "dynamics_loss_avg": 0.041673954017460345,
      "epoch": 9.946823120117187e-06,
      "step": 84430
    },
    {
      "action_loss_avg": 0.0011596205236855893,
      "epoch": 9.946823120117187e-06,
      "step": 84430
    },
    {
      "epoch": 9.946823120117187e-06,
      "step": 84430,
      "training_step_time": 9.39144492149353
    },
    {
      "epoch": 9.947776794433593e-06,
      "model_forward_time": 1.791213035583496,
      "step": 84431
    },
    {
      "epoch": 9.947776794433593e-06,
      "step": 84431,
      "training_step_time": 7.62987756729126
    },
    {
      "epoch": 9.94873046875e-06,
      "model_forward_time": 1.7919762134552002,
      "step": 84432
    },
    {
      "epoch": 9.94873046875e-06,
      "step": 84432,
      "training_step_time": 7.6156957149505615
    },
    {
      "epoch": 9.949684143066406e-06,
      "model_forward_time": 1.7816839218139648,
      "step": 84433
    },
    {
      "epoch": 9.949684143066406e-06,
      "step": 84433,
      "training_step_time": 7.5024638175964355
    },
    {
      "epoch": 9.950637817382812e-06,
      "model_forward_time": 1.7868893146514893,
      "step": 84434
    },
    {
      "epoch": 9.950637817382812e-06,
      "step": 84434,
      "training_step_time": 7.5960304737091064
    },
    {
      "epoch": 9.95159149169922e-06,
      "model_forward_time": 1.7842695713043213,
      "step": 84435
    },
    {
      "epoch": 9.95159149169922e-06,
      "step": 84435,
      "training_step_time": 8.499430656433105
    },
    {
      "epoch": 9.952545166015625e-06,
      "model_forward_time": 1.820458173751831,
      "step": 84436
    },
    {
      "epoch": 9.952545166015625e-06,
      "step": 84436,
      "training_step_time": 7.678648471832275
    },
    {
      "epoch": 9.95349884033203e-06,
      "model_forward_time": 1.7737724781036377,
      "step": 84437
    },
    {
      "epoch": 9.95349884033203e-06,
      "step": 84437,
      "training_step_time": 11.12360405921936
    },
    {
      "epoch": 9.954452514648438e-06,
      "model_forward_time": 2.8680155277252197,
      "step": 84438
    },
    {
      "epoch": 9.954452514648438e-06,
      "step": 84438,
      "training_step_time": 9.345826864242554
    },
    {
      "epoch": 9.955406188964844e-06,
      "model_forward_time": 1.7821464538574219,
      "step": 84439
    },
    {
      "epoch": 9.955406188964844e-06,
      "step": 84439,
      "training_step_time": 8.376461029052734
    },
    {
      "epoch": 9.95635986328125e-06,
      "grad_norm": 0.10387750715017319,
      "learning_rate": 6.475341369947047e-07,
      "loss": 0.0548,
      "step": 84440
    },
    {
      "epoch": 9.95635986328125e-06,
      "model_forward_time": 1.7990474700927734,
      "step": 84440
    },
    {
      "dynamics_loss_avg": 0.04634557459503412,
      "epoch": 9.95635986328125e-06,
      "step": 84440
    },
    {
      "action_loss_avg": 0.0010984614607878029,
      "epoch": 9.95635986328125e-06,
      "step": 84440
    },
    {
      "epoch": 9.95635986328125e-06,
      "step": 84440,
      "training_step_time": 7.627037763595581
    },
    {
      "epoch": 9.957313537597657e-06,
      "model_forward_time": 1.7641403675079346,
      "step": 84441
    },
    {
      "epoch": 9.957313537597657e-06,
      "step": 84441,
      "training_step_time": 7.630383253097534
    },
    {
      "epoch": 9.958267211914063e-06,
      "model_forward_time": 1.7941293716430664,
      "step": 84442
    },
    {
      "epoch": 9.958267211914063e-06,
      "step": 84442,
      "training_step_time": 7.3677544593811035
    },
    {
      "epoch": 9.959220886230468e-06,
      "model_forward_time": 1.7922687530517578,
      "step": 84443
    },
    {
      "epoch": 9.959220886230468e-06,
      "step": 84443,
      "training_step_time": 7.554489374160767
    },
    {
      "epoch": 9.960174560546876e-06,
      "model_forward_time": 1.7944247722625732,
      "step": 84444
    },
    {
      "epoch": 9.960174560546876e-06,
      "step": 84444,
      "training_step_time": 7.4919281005859375
    },
    {
      "epoch": 9.961128234863282e-06,
      "model_forward_time": 1.7976603507995605,
      "step": 84445
    },
    {
      "epoch": 9.961128234863282e-06,
      "step": 84445,
      "training_step_time": 7.647298097610474
    },
    {
      "epoch": 9.962081909179687e-06,
      "model_forward_time": 1.791670799255371,
      "step": 84446
    },
    {
      "epoch": 9.962081909179687e-06,
      "step": 84446,
      "training_step_time": 8.749330043792725
    },
    {
      "epoch": 9.963035583496093e-06,
      "model_forward_time": 2.864699363708496,
      "step": 84447
    },
    {
      "epoch": 9.963035583496093e-06,
      "step": 84447,
      "training_step_time": 9.864365816116333
    },
    {
      "epoch": 9.9639892578125e-06,
      "model_forward_time": 1.7866930961608887,
      "step": 84448
    },
    {
      "epoch": 9.9639892578125e-06,
      "step": 84448,
      "training_step_time": 8.5265052318573
    },
    {
      "epoch": 9.964942932128906e-06,
      "model_forward_time": 1.7955679893493652,
      "step": 84449
    },
    {
      "epoch": 9.964942932128906e-06,
      "step": 84449,
      "training_step_time": 8.485385179519653
    },
    {
      "epoch": 9.965896606445312e-06,
      "grad_norm": 0.09243134409189224,
      "learning_rate": 6.467205694111478e-07,
      "loss": 0.0558,
      "step": 84450
    },
    {
      "epoch": 9.965896606445312e-06,
      "model_forward_time": 1.7979271411895752,
      "step": 84450
    },
    {
      "dynamics_loss_avg": 0.05388295985758305,
      "epoch": 9.965896606445312e-06,
      "step": 84450
    },
    {
      "action_loss_avg": 0.0014508959924569354,
      "epoch": 9.965896606445312e-06,
      "step": 84450
    },
    {
      "epoch": 9.965896606445312e-06,
      "step": 84450,
      "training_step_time": 7.602965593338013
    },
    {
      "epoch": 9.96685028076172e-06,
      "model_forward_time": 1.7931041717529297,
      "step": 84451
    },
    {
      "epoch": 9.96685028076172e-06,
      "step": 84451,
      "training_step_time": 7.537977457046509
    },
    {
      "epoch": 9.967803955078125e-06,
      "model_forward_time": 1.7883286476135254,
      "step": 84452
    },
    {
      "epoch": 9.967803955078125e-06,
      "step": 84452,
      "training_step_time": 7.5507118701934814
    },
    {
      "epoch": 9.96875762939453e-06,
      "model_forward_time": 1.7715308666229248,
      "step": 84453
    },
    {
      "epoch": 9.96875762939453e-06,
      "step": 84453,
      "training_step_time": 7.469561576843262
    },
    {
      "epoch": 9.969711303710938e-06,
      "model_forward_time": 1.8005056381225586,
      "step": 84454
    },
    {
      "epoch": 9.969711303710938e-06,
      "step": 84454,
      "training_step_time": 8.556649446487427
    },
    {
      "epoch": 9.970664978027344e-06,
      "model_forward_time": 1.9000484943389893,
      "step": 84455
    },
    {
      "epoch": 9.970664978027344e-06,
      "step": 84455,
      "training_step_time": 8.209426641464233
    },
    {
      "epoch": 9.97161865234375e-06,
      "model_forward_time": 1.0867035388946533,
      "step": 84456
    },
    {
      "epoch": 9.97161865234375e-06,
      "step": 84456,
      "training_step_time": 8.368774175643921
    },
    {
      "epoch": 9.972572326660157e-06,
      "model_forward_time": 2.804377794265747,
      "step": 84457
    },
    {
      "epoch": 9.972572326660157e-06,
      "step": 84457,
      "training_step_time": 8.514404535293579
    },
    {
      "epoch": 9.973526000976563e-06,
      "model_forward_time": 1.7877707481384277,
      "step": 84458
    },
    {
      "epoch": 9.973526000976563e-06,
      "step": 84458,
      "training_step_time": 9.946237087249756
    },
    {
      "epoch": 9.974479675292968e-06,
      "model_forward_time": 1.7891383171081543,
      "step": 84459
    },
    {
      "epoch": 9.974479675292968e-06,
      "step": 84459,
      "training_step_time": 7.600134611129761
    },
    {
      "epoch": 9.975433349609376e-06,
      "grad_norm": 0.0791449248790741,
      "learning_rate": 6.459074778956836e-07,
      "loss": 0.0568,
      "step": 84460
    },
    {
      "epoch": 9.975433349609376e-06,
      "model_forward_time": 1.7764437198638916,
      "step": 84460
    },
    {
      "dynamics_loss_avg": 0.047882174141705036,
      "epoch": 9.975433349609376e-06,
      "step": 84460
    },
    {
      "action_loss_avg": 0.0008296461077407002,
      "epoch": 9.975433349609376e-06,
      "step": 84460
    },
    {
      "epoch": 9.975433349609376e-06,
      "step": 84460,
      "training_step_time": 7.450855016708374
    },
    {
      "epoch": 9.976387023925782e-06,
      "model_forward_time": 1.796562910079956,
      "step": 84461
    },
    {
      "epoch": 9.976387023925782e-06,
      "step": 84461,
      "training_step_time": 7.592940092086792
    },
    {
      "epoch": 9.977340698242187e-06,
      "model_forward_time": 1.7916615009307861,
      "step": 84462
    },
    {
      "epoch": 9.977340698242187e-06,
      "step": 84462,
      "training_step_time": 8.135707139968872
    },
    {
      "epoch": 9.978294372558593e-06,
      "model_forward_time": 1.7898766994476318,
      "step": 84463
    },
    {
      "epoch": 9.978294372558593e-06,
      "step": 84463,
      "training_step_time": 8.224459886550903
    },
    {
      "epoch": 9.979248046875e-06,
      "model_forward_time": 1.874100923538208,
      "step": 84464
    },
    {
      "epoch": 9.979248046875e-06,
      "step": 84464,
      "training_step_time": 7.69153356552124
    },
    {
      "epoch": 9.980201721191406e-06,
      "model_forward_time": 1.4280903339385986,
      "step": 84465
    },
    {
      "epoch": 9.980201721191406e-06,
      "step": 84465,
      "training_step_time": 9.998922109603882
    },
    {
      "epoch": 9.981155395507812e-06,
      "model_forward_time": 2.801828384399414,
      "step": 84466
    },
    {
      "epoch": 9.981155395507812e-06,
      "step": 84466,
      "training_step_time": 7.61209511756897
    },
    {
      "epoch": 9.98210906982422e-06,
      "model_forward_time": 1.7973737716674805,
      "step": 84467
    },
    {
      "epoch": 9.98210906982422e-06,
      "step": 84467,
      "training_step_time": 7.642495393753052
    },
    {
      "epoch": 9.983062744140625e-06,
      "model_forward_time": 1.79478120803833,
      "step": 84468
    },
    {
      "epoch": 9.983062744140625e-06,
      "step": 84468,
      "training_step_time": 10.225586414337158
    },
    {
      "epoch": 9.98401641845703e-06,
      "model_forward_time": 1.7951114177703857,
      "step": 84469
    },
    {
      "epoch": 9.98401641845703e-06,
      "step": 84469,
      "training_step_time": 7.585839509963989
    },
    {
      "epoch": 9.984970092773438e-06,
      "grad_norm": 0.11760049313306808,
      "learning_rate": 6.45094862537231e-07,
      "loss": 0.0563,
      "step": 84470
    },
    {
      "epoch": 9.984970092773438e-06,
      "model_forward_time": 1.7842273712158203,
      "step": 84470
    },
    {
      "dynamics_loss_avg": 0.0572218157351017,
      "epoch": 9.984970092773438e-06,
      "step": 84470
    },
    {
      "action_loss_avg": 0.0011904836079338565,
      "epoch": 9.984970092773438e-06,
      "step": 84470
    },
    {
      "epoch": 9.984970092773438e-06,
      "step": 84470,
      "training_step_time": 7.553800821304321
    },
    {
      "epoch": 9.985923767089844e-06,
      "model_forward_time": 1.7912652492523193,
      "step": 84471
    },
    {
      "epoch": 9.985923767089844e-06,
      "step": 84471,
      "training_step_time": 7.603042840957642
    },
    {
      "epoch": 9.98687744140625e-06,
      "model_forward_time": 1.793149709701538,
      "step": 84472
    },
    {
      "epoch": 9.98687744140625e-06,
      "step": 84472,
      "training_step_time": 7.613512992858887
    },
    {
      "epoch": 9.987831115722657e-06,
      "model_forward_time": 1.8544199466705322,
      "step": 84473
    },
    {
      "epoch": 9.987831115722657e-06,
      "step": 84473,
      "training_step_time": 9.142436981201172
    },
    {
      "epoch": 9.988784790039063e-06,
      "model_forward_time": 1.783203125,
      "step": 84474
    },
    {
      "epoch": 9.988784790039063e-06,
      "step": 84474,
      "training_step_time": 8.379170894622803
    },
    {
      "epoch": 9.989738464355468e-06,
      "model_forward_time": 2.8750460147857666,
      "step": 84475
    },
    {
      "epoch": 9.989738464355468e-06,
      "step": 84475,
      "training_step_time": 8.508408546447754
    },
    {
      "epoch": 9.990692138671876e-06,
      "model_forward_time": 1.777430534362793,
      "step": 84476
    },
    {
      "epoch": 9.990692138671876e-06,
      "step": 84476,
      "training_step_time": 8.779117345809937
    },
    {
      "epoch": 9.991645812988282e-06,
      "model_forward_time": 1.7810494899749756,
      "step": 84477
    },
    {
      "epoch": 9.991645812988282e-06,
      "step": 84477,
      "training_step_time": 8.174216508865356
    },
    {
      "epoch": 9.992599487304687e-06,
      "model_forward_time": 1.7905611991882324,
      "step": 84478
    },
    {
      "epoch": 9.992599487304687e-06,
      "step": 84478,
      "training_step_time": 7.639042615890503
    },
    {
      "epoch": 9.993553161621093e-06,
      "model_forward_time": 1.788003921508789,
      "step": 84479
    },
    {
      "epoch": 9.993553161621093e-06,
      "step": 84479,
      "training_step_time": 8.471290111541748
    },
    {
      "epoch": 9.9945068359375e-06,
      "grad_norm": 0.09471361339092255,
      "learning_rate": 6.442827234246585e-07,
      "loss": 0.0567,
      "step": 84480
    },
    {
      "epoch": 9.9945068359375e-06,
      "model_forward_time": 1.7756328582763672,
      "step": 84480
    },
    {
      "dynamics_loss_avg": 0.0564443638548255,
      "epoch": 9.9945068359375e-06,
      "step": 84480
    },
    {
      "action_loss_avg": 0.0014484853709291201,
      "epoch": 9.9945068359375e-06,
      "step": 84480
    },
    {
      "epoch": 9.9945068359375e-06,
      "step": 84480,
      "training_step_time": 7.6161298751831055
    },
    {
      "epoch": 9.995460510253906e-06,
      "model_forward_time": 1.788038969039917,
      "step": 84481
    },
    {
      "epoch": 9.995460510253906e-06,
      "step": 84481,
      "training_step_time": 7.600231170654297
    },
    {
      "epoch": 9.996414184570312e-06,
      "model_forward_time": 1.8254308700561523,
      "step": 84482
    },
    {
      "epoch": 9.996414184570312e-06,
      "step": 84482,
      "training_step_time": 8.216087579727173
    },
    {
      "epoch": 9.99736785888672e-06,
      "model_forward_time": 1.7911744117736816,
      "step": 84483
    },
    {
      "epoch": 9.99736785888672e-06,
      "step": 84483,
      "training_step_time": 11.141445875167847
    },
    {
      "epoch": 9.998321533203125e-06,
      "model_forward_time": 2.8760697841644287,
      "step": 84484
    },
    {
      "epoch": 9.998321533203125e-06,
      "step": 84484,
      "training_step_time": 9.395330429077148
    },
    {
      "epoch": 9.99927520751953e-06,
      "model_forward_time": 1.7884581089019775,
      "step": 84485
    },
    {
      "epoch": 9.99927520751953e-06,
      "step": 84485,
      "training_step_time": 7.8346734046936035
    },
    {
      "epoch": 1.0000228881835938e-05,
      "model_forward_time": 1.7842020988464355,
      "step": 84486
    },
    {
      "epoch": 1.0000228881835938e-05,
      "step": 84486,
      "training_step_time": 7.620272636413574
    },
    {
      "epoch": 1.0001182556152344e-05,
      "model_forward_time": 1.776672124862671,
      "step": 84487
    },
    {
      "epoch": 1.0001182556152344e-05,
      "step": 84487,
      "training_step_time": 7.667239427566528
    },
    {
      "epoch": 1.000213623046875e-05,
      "model_forward_time": 1.789111852645874,
      "step": 84488
    },
    {
      "epoch": 1.000213623046875e-05,
      "step": 84488,
      "training_step_time": 7.463144302368164
    },
    {
      "epoch": 1.0003089904785157e-05,
      "model_forward_time": 1.7979650497436523,
      "step": 84489
    },
    {
      "epoch": 1.0003089904785157e-05,
      "step": 84489,
      "training_step_time": 7.5552659034729
    },
    {
      "epoch": 1.0004043579101563e-05,
      "grad_norm": 0.1133919432759285,
      "learning_rate": 6.43471060646777e-07,
      "loss": 0.0552,
      "step": 84490
    },
    {
      "epoch": 1.0004043579101563e-05,
      "model_forward_time": 1.7808177471160889,
      "step": 84490
    },
    {
      "dynamics_loss_avg": 0.051998275145888326,
      "epoch": 1.0004043579101563e-05,
      "step": 84490
    },
    {
      "action_loss_avg": 0.001232994336169213,
      "epoch": 1.0004043579101563e-05,
      "step": 84490
    },
    {
      "epoch": 1.0004043579101563e-05,
      "step": 84490,
      "training_step_time": 7.471809387207031
    },
    {
      "epoch": 1.0004997253417968e-05,
      "model_forward_time": 1.8058195114135742,
      "step": 84491
    },
    {
      "epoch": 1.0004997253417968e-05,
      "step": 84491,
      "training_step_time": 7.68532133102417
    },
    {
      "epoch": 1.0005950927734376e-05,
      "model_forward_time": 1.7935495376586914,
      "step": 84492
    },
    {
      "epoch": 1.0005950927734376e-05,
      "step": 84492,
      "training_step_time": 10.171648263931274
    },
    {
      "epoch": 1.0006904602050782e-05,
      "model_forward_time": 2.8090806007385254,
      "step": 84493
    },
    {
      "epoch": 1.0006904602050782e-05,
      "step": 84493,
      "training_step_time": 8.522056341171265
    },
    {
      "epoch": 1.0007858276367187e-05,
      "model_forward_time": 1.7618350982666016,
      "step": 84494
    },
    {
      "epoch": 1.0007858276367187e-05,
      "step": 84494,
      "training_step_time": 8.416234731674194
    },
    {
      "epoch": 1.0008811950683593e-05,
      "model_forward_time": 1.7566263675689697,
      "step": 84495
    },
    {
      "epoch": 1.0008811950683593e-05,
      "step": 84495,
      "training_step_time": 7.6105687618255615
    },
    {
      "epoch": 1.0009765625e-05,
      "model_forward_time": 1.761728286743164,
      "step": 84496
    },
    {
      "epoch": 1.0009765625e-05,
      "step": 84496,
      "training_step_time": 7.614270448684692
    },
    {
      "epoch": 1.0010719299316406e-05,
      "model_forward_time": 1.7708666324615479,
      "step": 84497
    },
    {
      "epoch": 1.0010719299316406e-05,
      "step": 84497,
      "training_step_time": 7.40877103805542
    },
    {
      "epoch": 1.0011672973632812e-05,
      "model_forward_time": 1.7951099872589111,
      "step": 84498
    },
    {
      "epoch": 1.0011672973632812e-05,
      "step": 84498,
      "training_step_time": 7.519794940948486
    },
    {
      "epoch": 1.001262664794922e-05,
      "model_forward_time": 1.7863962650299072,
      "step": 84499
    },
    {
      "epoch": 1.001262664794922e-05,
      "step": 84499,
      "training_step_time": 8.197938680648804
    },
    {
      "epoch": 1.0013580322265625e-05,
      "grad_norm": 0.08904078602790833,
      "learning_rate": 6.426598742923512e-07,
      "loss": 0.0563,
      "step": 84500
    },
    {
      "epoch": 1.0013580322265625e-05,
      "model_forward_time": 1.7973730564117432,
      "step": 84500
    },
    {
      "dynamics_loss_avg": 0.04632256347686052,
      "epoch": 1.0013580322265625e-05,
      "step": 84500
    },
    {
      "action_loss_avg": 0.0008166308980435133,
      "epoch": 1.0013580322265625e-05,
      "step": 84500
    },
    {
      "epoch": 1.0013580322265625e-05,
      "step": 84500,
      "training_step_time": 9.405662536621094
    },
    {
      "epoch": 1.001453399658203e-05,
      "model_forward_time": 1.896803379058838,
      "step": 84501
    },
    {
      "epoch": 1.001453399658203e-05,
      "step": 84501,
      "training_step_time": 8.670789003372192
    },
    {
      "epoch": 1.0015487670898438e-05,
      "model_forward_time": 1.7962849140167236,
      "step": 84502
    },
    {
      "epoch": 1.0015487670898438e-05,
      "step": 84502,
      "training_step_time": 8.225204706192017
    },
    {
      "epoch": 1.0016441345214844e-05,
      "model_forward_time": 1.790926456451416,
      "step": 84503
    },
    {
      "epoch": 1.0016441345214844e-05,
      "step": 84503,
      "training_step_time": 8.745260953903198
    },
    {
      "epoch": 1.001739501953125e-05,
      "model_forward_time": 1.7972297668457031,
      "step": 84504
    },
    {
      "epoch": 1.001739501953125e-05,
      "step": 84504,
      "training_step_time": 8.794243335723877
    },
    {
      "epoch": 1.0018348693847657e-05,
      "model_forward_time": 1.788536548614502,
      "step": 84505
    },
    {
      "epoch": 1.0018348693847657e-05,
      "step": 84505,
      "training_step_time": 8.643007755279541
    },
    {
      "epoch": 1.0019302368164063e-05,
      "model_forward_time": 1.7911088466644287,
      "step": 84506
    },
    {
      "epoch": 1.0019302368164063e-05,
      "step": 84506,
      "training_step_time": 8.564236640930176
    },
    {
      "epoch": 1.0020256042480468e-05,
      "model_forward_time": 1.7919671535491943,
      "step": 84507
    },
    {
      "epoch": 1.0020256042480468e-05,
      "step": 84507,
      "training_step_time": 7.567785024642944
    },
    {
      "epoch": 1.0021209716796874e-05,
      "model_forward_time": 1.7918500900268555,
      "step": 84508
    },
    {
      "epoch": 1.0021209716796874e-05,
      "step": 84508,
      "training_step_time": 7.663366317749023
    },
    {
      "epoch": 1.0022163391113282e-05,
      "model_forward_time": 1.7885205745697021,
      "step": 84509
    },
    {
      "epoch": 1.0022163391113282e-05,
      "step": 84509,
      "training_step_time": 8.490952014923096
    },
    {
      "epoch": 1.0023117065429687e-05,
      "grad_norm": 0.07904614508152008,
      "learning_rate": 6.418491644500885e-07,
      "loss": 0.0546,
      "step": 84510
    },
    {
      "epoch": 1.0023117065429687e-05,
      "model_forward_time": 1.9026317596435547,
      "step": 84510
    },
    {
      "dynamics_loss_avg": 0.037432511895895006,
      "epoch": 1.0023117065429687e-05,
      "step": 84510
    },
    {
      "action_loss_avg": 0.001046362343186047,
      "epoch": 1.0023117065429687e-05,
      "step": 84510
    },
    {
      "epoch": 1.0023117065429687e-05,
      "step": 84510,
      "training_step_time": 9.707801342010498
    },
    {
      "epoch": 1.0024070739746093e-05,
      "model_forward_time": 1.777106761932373,
      "step": 84511
    },
    {
      "epoch": 1.0024070739746093e-05,
      "step": 84511,
      "training_step_time": 8.722792148590088
    },
    {
      "epoch": 1.00250244140625e-05,
      "model_forward_time": 2.893571138381958,
      "step": 84512
    },
    {
      "epoch": 1.00250244140625e-05,
      "step": 84512,
      "training_step_time": 8.396788835525513
    },
    {
      "epoch": 1.0025978088378906e-05,
      "model_forward_time": 1.7803215980529785,
      "step": 84513
    },
    {
      "epoch": 1.0025978088378906e-05,
      "step": 84513,
      "training_step_time": 8.677373886108398
    },
    {
      "epoch": 1.0026931762695312e-05,
      "model_forward_time": 1.7894346714019775,
      "step": 84514
    },
    {
      "epoch": 1.0026931762695312e-05,
      "step": 84514,
      "training_step_time": 7.7500159740448
    },
    {
      "epoch": 1.002788543701172e-05,
      "model_forward_time": 1.785038948059082,
      "step": 84515
    },
    {
      "epoch": 1.002788543701172e-05,
      "step": 84515,
      "training_step_time": 7.722850561141968
    },
    {
      "epoch": 1.0028839111328125e-05,
      "model_forward_time": 1.7880771160125732,
      "step": 84516
    },
    {
      "epoch": 1.0028839111328125e-05,
      "step": 84516,
      "training_step_time": 8.322314262390137
    },
    {
      "epoch": 1.002979278564453e-05,
      "model_forward_time": 1.7939062118530273,
      "step": 84517
    },
    {
      "epoch": 1.002979278564453e-05,
      "step": 84517,
      "training_step_time": 8.31990385055542
    },
    {
      "epoch": 1.0030746459960938e-05,
      "model_forward_time": 1.7954497337341309,
      "step": 84518
    },
    {
      "epoch": 1.0030746459960938e-05,
      "step": 84518,
      "training_step_time": 7.624980449676514
    },
    {
      "epoch": 1.0031700134277344e-05,
      "model_forward_time": 1.868539810180664,
      "step": 84519
    },
    {
      "epoch": 1.0031700134277344e-05,
      "step": 84519,
      "training_step_time": 8.341998100280762
    },
    {
      "epoch": 1.003265380859375e-05,
      "grad_norm": 0.08759395778179169,
      "learning_rate": 6.410389312086513e-07,
      "loss": 0.0561,
      "step": 84520
    },
    {
      "epoch": 1.003265380859375e-05,
      "model_forward_time": 1.7796740531921387,
      "step": 84520
    },
    {
      "dynamics_loss_avg": 0.041919337771832946,
      "epoch": 1.003265380859375e-05,
      "step": 84520
    },
    {
      "action_loss_avg": 0.0011404930031858385,
      "epoch": 1.003265380859375e-05,
      "step": 84520
    },
    {
      "epoch": 1.003265380859375e-05,
      "step": 84520,
      "training_step_time": 7.658661603927612
    },
    {
      "epoch": 1.0033607482910157e-05,
      "model_forward_time": 2.88684344291687,
      "step": 84521
    },
    {
      "epoch": 1.0033607482910157e-05,
      "step": 84521,
      "training_step_time": 8.481591701507568
    },
    {
      "epoch": 1.0034561157226563e-05,
      "model_forward_time": 1.7945637702941895,
      "step": 84522
    },
    {
      "epoch": 1.0034561157226563e-05,
      "step": 84522,
      "training_step_time": 7.6286609172821045
    },
    {
      "epoch": 1.0035514831542968e-05,
      "model_forward_time": 1.7893242835998535,
      "step": 84523
    },
    {
      "epoch": 1.0035514831542968e-05,
      "step": 84523,
      "training_step_time": 7.67141580581665
    },
    {
      "epoch": 1.0036468505859374e-05,
      "model_forward_time": 1.7930049896240234,
      "step": 84524
    },
    {
      "epoch": 1.0036468505859374e-05,
      "step": 84524,
      "training_step_time": 7.715019702911377
    },
    {
      "epoch": 1.0037422180175782e-05,
      "model_forward_time": 1.7913978099822998,
      "step": 84525
    },
    {
      "epoch": 1.0037422180175782e-05,
      "step": 84525,
      "training_step_time": 7.409090042114258
    },
    {
      "epoch": 1.0038375854492187e-05,
      "model_forward_time": 1.787858486175537,
      "step": 84526
    },
    {
      "epoch": 1.0038375854492187e-05,
      "step": 84526,
      "training_step_time": 7.5620551109313965
    },
    {
      "epoch": 1.0039329528808593e-05,
      "model_forward_time": 1.7912042140960693,
      "step": 84527
    },
    {
      "epoch": 1.0039329528808593e-05,
      "step": 84527,
      "training_step_time": 8.633897542953491
    },
    {
      "epoch": 1.0040283203125e-05,
      "model_forward_time": 1.8319814205169678,
      "step": 84528
    },
    {
      "epoch": 1.0040283203125e-05,
      "step": 84528,
      "training_step_time": 7.664463758468628
    },
    {
      "epoch": 1.0041236877441406e-05,
      "model_forward_time": 1.7900934219360352,
      "step": 84529
    },
    {
      "epoch": 1.0041236877441406e-05,
      "step": 84529,
      "training_step_time": 7.733839511871338
    },
    {
      "epoch": 1.0042190551757812e-05,
      "grad_norm": 0.07670804113149643,
      "learning_rate": 6.402291746566414e-07,
      "loss": 0.0547,
      "step": 84530
    },
    {
      "epoch": 1.0042190551757812e-05,
      "model_forward_time": 2.8310530185699463,
      "step": 84530
    },
    {
      "dynamics_loss_avg": 0.04724772665649653,
      "epoch": 1.0042190551757812e-05,
      "step": 84530
    },
    {
      "action_loss_avg": 0.001029826031299308,
      "epoch": 1.0042190551757812e-05,
      "step": 84530
    },
    {
      "epoch": 1.0042190551757812e-05,
      "step": 84530,
      "training_step_time": 8.199931383132935
    },
    {
      "epoch": 1.004314422607422e-05,
      "model_forward_time": 1.774193286895752,
      "step": 84531
    },
    {
      "epoch": 1.004314422607422e-05,
      "step": 84531,
      "training_step_time": 8.13055944442749
    },
    {
      "epoch": 1.0044097900390625e-05,
      "model_forward_time": 1.76788330078125,
      "step": 84532
    },
    {
      "epoch": 1.0044097900390625e-05,
      "step": 84532,
      "training_step_time": 7.699229955673218
    },
    {
      "epoch": 1.004505157470703e-05,
      "model_forward_time": 1.7665841579437256,
      "step": 84533
    },
    {
      "epoch": 1.004505157470703e-05,
      "step": 84533,
      "training_step_time": 8.17053508758545
    },
    {
      "epoch": 1.0046005249023438e-05,
      "model_forward_time": 1.7806353569030762,
      "step": 84534
    },
    {
      "epoch": 1.0046005249023438e-05,
      "step": 84534,
      "training_step_time": 7.36138653755188
    },
    {
      "epoch": 1.0046958923339844e-05,
      "model_forward_time": 1.779409408569336,
      "step": 84535
    },
    {
      "epoch": 1.0046958923339844e-05,
      "step": 84535,
      "training_step_time": 7.572314262390137
    },
    {
      "epoch": 1.004791259765625e-05,
      "model_forward_time": 1.7753112316131592,
      "step": 84536
    },
    {
      "epoch": 1.004791259765625e-05,
      "step": 84536,
      "training_step_time": 7.52036190032959
    },
    {
      "epoch": 1.0048866271972657e-05,
      "model_forward_time": 1.7525575160980225,
      "step": 84537
    },
    {
      "epoch": 1.0048866271972657e-05,
      "step": 84537,
      "training_step_time": 8.65980339050293
    },
    {
      "epoch": 1.0049819946289063e-05,
      "model_forward_time": 1.7521896362304688,
      "step": 84538
    },
    {
      "epoch": 1.0049819946289063e-05,
      "step": 84538,
      "training_step_time": 7.678344249725342
    },
    {
      "epoch": 1.0050773620605468e-05,
      "model_forward_time": 2.826632499694824,
      "step": 84539
    },
    {
      "epoch": 1.0050773620605468e-05,
      "step": 84539,
      "training_step_time": 9.089174509048462
    },
    {
      "epoch": 1.0051727294921874e-05,
      "grad_norm": 0.0775807574391365,
      "learning_rate": 6.394198948826158e-07,
      "loss": 0.0543,
      "step": 84540
    },
    {
      "epoch": 1.0051727294921874e-05,
      "model_forward_time": 1.7612924575805664,
      "step": 84540
    },
    {
      "dynamics_loss_avg": 0.052312766760587694,
      "epoch": 1.0051727294921874e-05,
      "step": 84540
    },
    {
      "action_loss_avg": 0.0023925847504870034,
      "epoch": 1.0051727294921874e-05,
      "step": 84540
    },
    {
      "epoch": 1.0051727294921874e-05,
      "step": 84540,
      "training_step_time": 8.60217833518982
    },
    {
      "epoch": 1.0052680969238282e-05,
      "model_forward_time": 1.760894775390625,
      "step": 84541
    },
    {
      "epoch": 1.0052680969238282e-05,
      "step": 84541,
      "training_step_time": 8.682984352111816
    },
    {
      "epoch": 1.0053634643554687e-05,
      "model_forward_time": 1.7619576454162598,
      "step": 84542
    },
    {
      "epoch": 1.0053634643554687e-05,
      "step": 84542,
      "training_step_time": 8.317013025283813
    },
    {
      "epoch": 1.0054588317871093e-05,
      "model_forward_time": 1.7726140022277832,
      "step": 84543
    },
    {
      "epoch": 1.0054588317871093e-05,
      "step": 84543,
      "training_step_time": 7.509186744689941
    },
    {
      "epoch": 1.00555419921875e-05,
      "model_forward_time": 1.7612330913543701,
      "step": 84544
    },
    {
      "epoch": 1.00555419921875e-05,
      "step": 84544,
      "training_step_time": 7.596487283706665
    },
    {
      "epoch": 1.0056495666503906e-05,
      "model_forward_time": 1.7568833827972412,
      "step": 84545
    },
    {
      "epoch": 1.0056495666503906e-05,
      "step": 84545,
      "training_step_time": 8.136083841323853
    },
    {
      "epoch": 1.0057449340820312e-05,
      "model_forward_time": 1.7765109539031982,
      "step": 84546
    },
    {
      "epoch": 1.0057449340820312e-05,
      "step": 84546,
      "training_step_time": 8.292418003082275
    },
    {
      "epoch": 1.005840301513672e-05,
      "model_forward_time": 1.8770966529846191,
      "step": 84547
    },
    {
      "epoch": 1.005840301513672e-05,
      "step": 84547,
      "training_step_time": 9.245332956314087
    },
    {
      "epoch": 1.0059356689453125e-05,
      "model_forward_time": 2.8031322956085205,
      "step": 84548
    },
    {
      "epoch": 1.0059356689453125e-05,
      "step": 84548,
      "training_step_time": 8.359322547912598
    },
    {
      "epoch": 1.006031036376953e-05,
      "model_forward_time": 1.7680680751800537,
      "step": 84549
    },
    {
      "epoch": 1.006031036376953e-05,
      "step": 84549,
      "training_step_time": 8.734097480773926
    },
    {
      "epoch": 1.0061264038085938e-05,
      "grad_norm": 0.08128970861434937,
      "learning_rate": 6.386110919750727e-07,
      "loss": 0.0551,
      "step": 84550
    },
    {
      "epoch": 1.0061264038085938e-05,
      "model_forward_time": 1.7633123397827148,
      "step": 84550
    },
    {
      "dynamics_loss_avg": 0.06268942765891553,
      "epoch": 1.0061264038085938e-05,
      "step": 84550
    },
    {
      "action_loss_avg": 0.002765522641129792,
      "epoch": 1.0061264038085938e-05,
      "step": 84550
    },
    {
      "epoch": 1.0061264038085938e-05,
      "step": 84550,
      "training_step_time": 8.542582750320435
    },
    {
      "epoch": 1.0062217712402344e-05,
      "model_forward_time": 1.771085262298584,
      "step": 84551
    },
    {
      "epoch": 1.0062217712402344e-05,
      "step": 84551,
      "training_step_time": 8.531810283660889
    },
    {
      "epoch": 1.006317138671875e-05,
      "model_forward_time": 1.771993637084961,
      "step": 84552
    },
    {
      "epoch": 1.006317138671875e-05,
      "step": 84552,
      "training_step_time": 7.7569801807403564
    },
    {
      "epoch": 1.0064125061035157e-05,
      "model_forward_time": 1.7680573463439941,
      "step": 84553
    },
    {
      "epoch": 1.0064125061035157e-05,
      "step": 84553,
      "training_step_time": 8.30501389503479
    },
    {
      "epoch": 1.0065078735351563e-05,
      "model_forward_time": 1.7635619640350342,
      "step": 84554
    },
    {
      "epoch": 1.0065078735351563e-05,
      "step": 84554,
      "training_step_time": 7.610246181488037
    },
    {
      "epoch": 1.0066032409667968e-05,
      "model_forward_time": 1.7657661437988281,
      "step": 84555
    },
    {
      "epoch": 1.0066032409667968e-05,
      "step": 84555,
      "training_step_time": 7.685283184051514
    },
    {
      "epoch": 1.0066986083984374e-05,
      "model_forward_time": 1.83284592628479,
      "step": 84556
    },
    {
      "epoch": 1.0066986083984374e-05,
      "step": 84556,
      "training_step_time": 7.83494758605957
    },
    {
      "epoch": 1.0067939758300782e-05,
      "model_forward_time": 1.7787511348724365,
      "step": 84557
    },
    {
      "epoch": 1.0067939758300782e-05,
      "step": 84557,
      "training_step_time": 9.447921752929688
    },
    {
      "epoch": 1.0068893432617187e-05,
      "model_forward_time": 2.8314685821533203,
      "step": 84558
    },
    {
      "epoch": 1.0068893432617187e-05,
      "step": 84558,
      "training_step_time": 9.645919799804688
    },
    {
      "epoch": 1.0069847106933593e-05,
      "model_forward_time": 1.7646064758300781,
      "step": 84559
    },
    {
      "epoch": 1.0069847106933593e-05,
      "step": 84559,
      "training_step_time": 8.847572326660156
    },
    {
      "epoch": 1.007080078125e-05,
      "grad_norm": 0.08514691144227982,
      "learning_rate": 6.37802766022465e-07,
      "loss": 0.0556,
      "step": 84560
    },
    {
      "epoch": 1.007080078125e-05,
      "model_forward_time": 1.7505428791046143,
      "step": 84560
    },
    {
      "dynamics_loss_avg": 0.05681615956127643,
      "epoch": 1.007080078125e-05,
      "step": 84560
    },
    {
      "action_loss_avg": 0.001263948112318758,
      "epoch": 1.007080078125e-05,
      "step": 84560
    },
    {
      "epoch": 1.007080078125e-05,
      "step": 84560,
      "training_step_time": 7.75246787071228
    },
    {
      "epoch": 1.0071754455566406e-05,
      "model_forward_time": 1.747920274734497,
      "step": 84561
    },
    {
      "epoch": 1.0071754455566406e-05,
      "step": 84561,
      "training_step_time": 7.890944242477417
    },
    {
      "epoch": 1.0072708129882812e-05,
      "model_forward_time": 1.7503297328948975,
      "step": 84562
    },
    {
      "epoch": 1.0072708129882812e-05,
      "step": 84562,
      "training_step_time": 7.6653265953063965
    },
    {
      "epoch": 1.007366180419922e-05,
      "model_forward_time": 1.7371551990509033,
      "step": 84563
    },
    {
      "epoch": 1.007366180419922e-05,
      "step": 84563,
      "training_step_time": 7.613009214401245
    },
    {
      "epoch": 1.0074615478515625e-05,
      "model_forward_time": 1.7504794597625732,
      "step": 84564
    },
    {
      "epoch": 1.0074615478515625e-05,
      "step": 84564,
      "training_step_time": 9.25906753540039
    },
    {
      "epoch": 1.007556915283203e-05,
      "model_forward_time": 1.7917382717132568,
      "step": 84565
    },
    {
      "epoch": 1.007556915283203e-05,
      "step": 84565,
      "training_step_time": 8.514167547225952
    },
    {
      "epoch": 1.0076522827148438e-05,
      "model_forward_time": 1.7434566020965576,
      "step": 84566
    },
    {
      "epoch": 1.0076522827148438e-05,
      "step": 84566,
      "training_step_time": 7.813631534576416
    },
    {
      "epoch": 1.0077476501464844e-05,
      "model_forward_time": 2.8427627086639404,
      "step": 84567
    },
    {
      "epoch": 1.0077476501464844e-05,
      "step": 84567,
      "training_step_time": 8.438110589981079
    },
    {
      "epoch": 1.007843017578125e-05,
      "model_forward_time": 1.754152536392212,
      "step": 84568
    },
    {
      "epoch": 1.007843017578125e-05,
      "step": 84568,
      "training_step_time": 7.751361608505249
    },
    {
      "epoch": 1.0079383850097657e-05,
      "model_forward_time": 1.745119571685791,
      "step": 84569
    },
    {
      "epoch": 1.0079383850097657e-05,
      "step": 84569,
      "training_step_time": 7.660903453826904
    },
    {
      "epoch": 1.0080337524414063e-05,
      "grad_norm": 0.12578582763671875,
      "learning_rate": 6.369949171131856e-07,
      "loss": 0.0539,
      "step": 84570
    },
    {
      "epoch": 1.0080337524414063e-05,
      "model_forward_time": 1.7531099319458008,
      "step": 84570
    },
    {
      "dynamics_loss_avg": 0.04039961434900761,
      "epoch": 1.0080337524414063e-05,
      "step": 84570
    },
    {
      "action_loss_avg": 0.001215190498623997,
      "epoch": 1.0080337524414063e-05,
      "step": 84570
    },
    {
      "epoch": 1.0080337524414063e-05,
      "step": 84570,
      "training_step_time": 7.714532375335693
    },
    {
      "epoch": 1.0081291198730468e-05,
      "model_forward_time": 1.7421929836273193,
      "step": 84571
    },
    {
      "epoch": 1.0081291198730468e-05,
      "step": 84571,
      "training_step_time": 7.554055213928223
    },
    {
      "epoch": 1.0082244873046874e-05,
      "model_forward_time": 1.7751688957214355,
      "step": 84572
    },
    {
      "epoch": 1.0082244873046874e-05,
      "step": 84572,
      "training_step_time": 7.578732252120972
    },
    {
      "epoch": 1.0083198547363282e-05,
      "model_forward_time": 1.7513384819030762,
      "step": 84573
    },
    {
      "epoch": 1.0083198547363282e-05,
      "step": 84573,
      "training_step_time": 7.668254375457764
    },
    {
      "epoch": 1.0084152221679687e-05,
      "model_forward_time": 1.7572081089019775,
      "step": 84574
    },
    {
      "epoch": 1.0084152221679687e-05,
      "step": 84574,
      "training_step_time": 8.260379791259766
    },
    {
      "epoch": 1.0085105895996093e-05,
      "model_forward_time": 1.7458081245422363,
      "step": 84575
    },
    {
      "epoch": 1.0085105895996093e-05,
      "step": 84575,
      "training_step_time": 8.591195344924927
    },
    {
      "epoch": 1.00860595703125e-05,
      "model_forward_time": 2.8556876182556152,
      "step": 84576
    },
    {
      "epoch": 1.00860595703125e-05,
      "step": 84576,
      "training_step_time": 7.6035988330841064
    },
    {
      "epoch": 1.0087013244628906e-05,
      "model_forward_time": 1.743222713470459,
      "step": 84577
    },
    {
      "epoch": 1.0087013244628906e-05,
      "step": 84577,
      "training_step_time": 7.670303106307983
    },
    {
      "epoch": 1.0087966918945312e-05,
      "model_forward_time": 1.7484512329101562,
      "step": 84578
    },
    {
      "epoch": 1.0087966918945312e-05,
      "step": 84578,
      "training_step_time": 8.634338140487671
    },
    {
      "epoch": 1.008892059326172e-05,
      "model_forward_time": 1.745589256286621,
      "step": 84579
    },
    {
      "epoch": 1.008892059326172e-05,
      "step": 84579,
      "training_step_time": 7.712450981140137
    },
    {
      "epoch": 1.0089874267578125e-05,
      "grad_norm": 0.09155435860157013,
      "learning_rate": 6.361875453355848e-07,
      "loss": 0.0546,
      "step": 84580
    },
    {
      "epoch": 1.0089874267578125e-05,
      "model_forward_time": 1.7564153671264648,
      "step": 84580
    },
    {
      "dynamics_loss_avg": 0.05906953848898411,
      "epoch": 1.0089874267578125e-05,
      "step": 84580
    },
    {
      "action_loss_avg": 0.0014878663088893518,
      "epoch": 1.0089874267578125e-05,
      "step": 84580
    },
    {
      "epoch": 1.0089874267578125e-05,
      "step": 84580,
      "training_step_time": 7.640642881393433
    },
    {
      "epoch": 1.009082794189453e-05,
      "model_forward_time": 1.3902268409729004,
      "step": 84581
    },
    {
      "epoch": 1.009082794189453e-05,
      "step": 84581,
      "training_step_time": 7.6020660400390625
    },
    {
      "epoch": 1.0091781616210938e-05,
      "model_forward_time": 1.7535974979400635,
      "step": 84582
    },
    {
      "epoch": 1.0091781616210938e-05,
      "step": 84582,
      "training_step_time": 9.318486452102661
    },
    {
      "epoch": 1.0092735290527344e-05,
      "model_forward_time": 1.7546005249023438,
      "step": 84583
    },
    {
      "epoch": 1.0092735290527344e-05,
      "step": 84583,
      "training_step_time": 7.755099773406982
    },
    {
      "epoch": 1.009368896484375e-05,
      "model_forward_time": 1.861703634262085,
      "step": 84584
    },
    {
      "epoch": 1.009368896484375e-05,
      "step": 84584,
      "training_step_time": 8.751446723937988
    },
    {
      "epoch": 1.0094642639160157e-05,
      "model_forward_time": 2.7999322414398193,
      "step": 84585
    },
    {
      "epoch": 1.0094642639160157e-05,
      "step": 84585,
      "training_step_time": 9.546297311782837
    },
    {
      "epoch": 1.0095596313476563e-05,
      "model_forward_time": 1.7469482421875,
      "step": 84586
    },
    {
      "epoch": 1.0095596313476563e-05,
      "step": 84586,
      "training_step_time": 9.580540895462036
    },
    {
      "epoch": 1.0096549987792969e-05,
      "model_forward_time": 1.7615752220153809,
      "step": 84587
    },
    {
      "epoch": 1.0096549987792969e-05,
      "step": 84587,
      "training_step_time": 7.685635328292847
    },
    {
      "epoch": 1.0097503662109374e-05,
      "model_forward_time": 1.749065637588501,
      "step": 84588
    },
    {
      "epoch": 1.0097503662109374e-05,
      "step": 84588,
      "training_step_time": 9.565646886825562
    },
    {
      "epoch": 1.0098457336425782e-05,
      "model_forward_time": 1.7653913497924805,
      "step": 84589
    },
    {
      "epoch": 1.0098457336425782e-05,
      "step": 84589,
      "training_step_time": 7.737151384353638
    },
    {
      "epoch": 1.0099411010742187e-05,
      "grad_norm": 0.0875454917550087,
      "learning_rate": 6.353806507779525e-07,
      "loss": 0.0553,
      "step": 84590
    },
    {
      "epoch": 1.0099411010742187e-05,
      "model_forward_time": 1.761972188949585,
      "step": 84590
    },
    {
      "dynamics_loss_avg": 0.061480862647295,
      "epoch": 1.0099411010742187e-05,
      "step": 84590
    },
    {
      "action_loss_avg": 0.0013278284430271014,
      "epoch": 1.0099411010742187e-05,
      "step": 84590
    },
    {
      "epoch": 1.0099411010742187e-05,
      "step": 84590,
      "training_step_time": 7.615643501281738
    },
    {
      "epoch": 1.0100364685058593e-05,
      "model_forward_time": 1.755779504776001,
      "step": 84591
    },
    {
      "epoch": 1.0100364685058593e-05,
      "step": 84591,
      "training_step_time": 7.669810771942139
    },
    {
      "epoch": 1.0101318359375e-05,
      "model_forward_time": 1.7690980434417725,
      "step": 84592
    },
    {
      "epoch": 1.0101318359375e-05,
      "step": 84592,
      "training_step_time": 7.66877555847168
    },
    {
      "epoch": 1.0102272033691406e-05,
      "model_forward_time": 1.8556041717529297,
      "step": 84593
    },
    {
      "epoch": 1.0102272033691406e-05,
      "step": 84593,
      "training_step_time": 9.669973134994507
    },
    {
      "epoch": 1.0103225708007812e-05,
      "model_forward_time": 1.7423009872436523,
      "step": 84594
    },
    {
      "epoch": 1.0103225708007812e-05,
      "step": 84594,
      "training_step_time": 8.993445873260498
    },
    {
      "epoch": 1.010417938232422e-05,
      "model_forward_time": 1.7808940410614014,
      "step": 84595
    },
    {
      "epoch": 1.010417938232422e-05,
      "step": 84595,
      "training_step_time": 7.698755741119385
    },
    {
      "epoch": 1.0105133056640625e-05,
      "model_forward_time": 1.3978781700134277,
      "step": 84596
    },
    {
      "epoch": 1.0105133056640625e-05,
      "step": 84596,
      "training_step_time": 7.640817642211914
    },
    {
      "epoch": 1.010608673095703e-05,
      "model_forward_time": 1.7560186386108398,
      "step": 84597
    },
    {
      "epoch": 1.010608673095703e-05,
      "step": 84597,
      "training_step_time": 7.487643003463745
    },
    {
      "epoch": 1.0107040405273438e-05,
      "model_forward_time": 1.760957956314087,
      "step": 84598
    },
    {
      "epoch": 1.0107040405273438e-05,
      "step": 84598,
      "training_step_time": 7.712839603424072
    },
    {
      "epoch": 1.0107994079589844e-05,
      "model_forward_time": 1.7616982460021973,
      "step": 84599
    },
    {
      "epoch": 1.0107994079589844e-05,
      "step": 84599,
      "training_step_time": 7.519704103469849
    },
    {
      "epoch": 1.010894775390625e-05,
      "grad_norm": 0.17740604281425476,
      "learning_rate": 6.345742335285304e-07,
      "loss": 0.0536,
      "step": 84600
    },
    {
      "epoch": 1.010894775390625e-05,
      "model_forward_time": 1.7619543075561523,
      "step": 84600
    },
    {
      "dynamics_loss_avg": 0.05071651376783848,
      "epoch": 1.010894775390625e-05,
      "step": 84600
    },
    {
      "action_loss_avg": 0.0009655477362684906,
      "epoch": 1.010894775390625e-05,
      "step": 84600
    },
    {
      "epoch": 1.010894775390625e-05,
      "step": 84600,
      "training_step_time": 7.669041633605957
    },
    {
      "epoch": 1.0109901428222657e-05,
      "model_forward_time": 1.7576234340667725,
      "step": 84601
    },
    {
      "epoch": 1.0109901428222657e-05,
      "step": 84601,
      "training_step_time": 8.081038475036621
    },
    {
      "epoch": 1.0110855102539063e-05,
      "model_forward_time": 1.8310213088989258,
      "step": 84602
    },
    {
      "epoch": 1.0110855102539063e-05,
      "step": 84602,
      "training_step_time": 9.256956100463867
    },
    {
      "epoch": 1.0111808776855469e-05,
      "model_forward_time": 1.770942211151123,
      "step": 84603
    },
    {
      "epoch": 1.0111808776855469e-05,
      "step": 84603,
      "training_step_time": 8.997585773468018
    },
    {
      "epoch": 1.0112762451171874e-05,
      "model_forward_time": 2.8930342197418213,
      "step": 84604
    },
    {
      "epoch": 1.0112762451171874e-05,
      "step": 84604,
      "training_step_time": 9.261662721633911
    },
    {
      "epoch": 1.0113716125488282e-05,
      "model_forward_time": 1.7572722434997559,
      "step": 84605
    },
    {
      "epoch": 1.0113716125488282e-05,
      "step": 84605,
      "training_step_time": 7.583113670349121
    },
    {
      "epoch": 1.0114669799804687e-05,
      "model_forward_time": 1.7850236892700195,
      "step": 84606
    },
    {
      "epoch": 1.0114669799804687e-05,
      "step": 84606,
      "training_step_time": 8.614789962768555
    },
    {
      "epoch": 1.0115623474121093e-05,
      "model_forward_time": 1.3947055339813232,
      "step": 84607
    },
    {
      "epoch": 1.0115623474121093e-05,
      "step": 84607,
      "training_step_time": 8.818323850631714
    },
    {
      "epoch": 1.01165771484375e-05,
      "model_forward_time": 1.7629354000091553,
      "step": 84608
    },
    {
      "epoch": 1.01165771484375e-05,
      "step": 84608,
      "training_step_time": 7.5771684646606445
    },
    {
      "epoch": 1.0117530822753906e-05,
      "model_forward_time": 1.7608749866485596,
      "step": 84609
    },
    {
      "epoch": 1.0117530822753906e-05,
      "step": 84609,
      "training_step_time": 7.608149766921997
    },
    {
      "epoch": 1.0118484497070312e-05,
      "grad_norm": 0.10428967326879501,
      "learning_rate": 6.337682936755063e-07,
      "loss": 0.0549,
      "step": 84610
    },
    {
      "epoch": 1.0118484497070312e-05,
      "model_forward_time": 1.7516403198242188,
      "step": 84610
    },
    {
      "dynamics_loss_avg": 0.0535118056461215,
      "epoch": 1.0118484497070312e-05,
      "step": 84610
    },
    {
      "action_loss_avg": 0.0015329127825680188,
      "epoch": 1.0118484497070312e-05,
      "step": 84610
    },
    {
      "epoch": 1.0118484497070312e-05,
      "step": 84610,
      "training_step_time": 7.615540981292725
    },
    {
      "epoch": 1.011943817138672e-05,
      "model_forward_time": 1.8062889575958252,
      "step": 84611
    },
    {
      "epoch": 1.011943817138672e-05,
      "step": 84611,
      "training_step_time": 7.778404474258423
    },
    {
      "epoch": 1.0120391845703125e-05,
      "model_forward_time": 1.7666120529174805,
      "step": 84612
    },
    {
      "epoch": 1.0120391845703125e-05,
      "step": 84612,
      "training_step_time": 8.391048669815063
    },
    {
      "epoch": 1.012134552001953e-05,
      "model_forward_time": 2.814617395401001,
      "step": 84613
    },
    {
      "epoch": 1.012134552001953e-05,
      "step": 84613,
      "training_step_time": 7.609167814254761
    },
    {
      "epoch": 1.0122299194335938e-05,
      "model_forward_time": 1.7531449794769287,
      "step": 84614
    },
    {
      "epoch": 1.0122299194335938e-05,
      "step": 84614,
      "training_step_time": 12.69453239440918
    },
    {
      "epoch": 1.0123252868652344e-05,
      "model_forward_time": 1.7444684505462646,
      "step": 84615
    },
    {
      "epoch": 1.0123252868652344e-05,
      "step": 84615,
      "training_step_time": 7.376183271408081
    },
    {
      "epoch": 1.012420654296875e-05,
      "model_forward_time": 1.7515106201171875,
      "step": 84616
    },
    {
      "epoch": 1.012420654296875e-05,
      "step": 84616,
      "training_step_time": 7.789700984954834
    },
    {
      "epoch": 1.0125160217285157e-05,
      "model_forward_time": 1.7473716735839844,
      "step": 84617
    },
    {
      "epoch": 1.0125160217285157e-05,
      "step": 84617,
      "training_step_time": 8.197712182998657
    },
    {
      "epoch": 1.0126113891601563e-05,
      "model_forward_time": 1.7471230030059814,
      "step": 84618
    },
    {
      "epoch": 1.0126113891601563e-05,
      "step": 84618,
      "training_step_time": 7.781570911407471
    },
    {
      "epoch": 1.0127067565917969e-05,
      "model_forward_time": 1.75199294090271,
      "step": 84619
    },
    {
      "epoch": 1.0127067565917969e-05,
      "step": 84619,
      "training_step_time": 6.790662527084351
    },
    {
      "epoch": 1.0128021240234374e-05,
      "grad_norm": 0.105812206864357,
      "learning_rate": 6.32962831307018e-07,
      "loss": 0.0551,
      "step": 84620
    },
    {
      "epoch": 1.0128021240234374e-05,
      "model_forward_time": 1.756361722946167,
      "step": 84620
    },
    {
      "dynamics_loss_avg": 0.04536665380001068,
      "epoch": 1.0128021240234374e-05,
      "step": 84620
    },
    {
      "action_loss_avg": 0.001324062838102691,
      "epoch": 1.0128021240234374e-05,
      "step": 84620
    },
    {
      "epoch": 1.0128021240234374e-05,
      "step": 84620,
      "training_step_time": 6.812567949295044
    },
    {
      "epoch": 1.0128974914550782e-05,
      "model_forward_time": 1.746812343597412,
      "step": 84621
    },
    {
      "epoch": 1.0128974914550782e-05,
      "step": 84621,
      "training_step_time": 6.750521659851074
    },
    {
      "epoch": 1.0129928588867187e-05,
      "model_forward_time": 1.7412903308868408,
      "step": 84622
    },
    {
      "epoch": 1.0129928588867187e-05,
      "step": 84622,
      "training_step_time": 6.921251535415649
    },
    {
      "epoch": 1.0130882263183593e-05,
      "model_forward_time": 1.74796724319458,
      "step": 84623
    },
    {
      "epoch": 1.0130882263183593e-05,
      "step": 84623,
      "training_step_time": 10.133806467056274
    },
    {
      "epoch": 1.01318359375e-05,
      "model_forward_time": 1.7453343868255615,
      "step": 84624
    },
    {
      "epoch": 1.01318359375e-05,
      "step": 84624,
      "training_step_time": 7.78367805480957
    },
    {
      "epoch": 1.0132789611816406e-05,
      "model_forward_time": 1.7525725364685059,
      "step": 84625
    },
    {
      "epoch": 1.0132789611816406e-05,
      "step": 84625,
      "training_step_time": 6.809173583984375
    },
    {
      "epoch": 1.0133743286132812e-05,
      "model_forward_time": 1.7712328433990479,
      "step": 84626
    },
    {
      "epoch": 1.0133743286132812e-05,
      "step": 84626,
      "training_step_time": 7.608108758926392
    },
    {
      "epoch": 1.013469696044922e-05,
      "model_forward_time": 1.7863562107086182,
      "step": 84627
    },
    {
      "epoch": 1.013469696044922e-05,
      "step": 84627,
      "training_step_time": 7.306822061538696
    },
    {
      "epoch": 1.0135650634765625e-05,
      "model_forward_time": 1.7597475051879883,
      "step": 84628
    },
    {
      "epoch": 1.0135650634765625e-05,
      "step": 84628,
      "training_step_time": 7.5732421875
    },
    {
      "epoch": 1.013660430908203e-05,
      "model_forward_time": 1.75486421585083,
      "step": 84629
    },
    {
      "epoch": 1.013660430908203e-05,
      "step": 84629,
      "training_step_time": 6.870913028717041
    },
    {
      "epoch": 1.0137557983398438e-05,
      "grad_norm": 0.12751363217830658,
      "learning_rate": 6.321578465111478e-07,
      "loss": 0.0553,
      "step": 84630
    },
    {
      "epoch": 1.0137557983398438e-05,
      "model_forward_time": 1.7510089874267578,
      "step": 84630
    },
    {
      "dynamics_loss_avg": 0.04172674175351858,
      "epoch": 1.0137557983398438e-05,
      "step": 84630
    },
    {
      "action_loss_avg": 0.00068380581215024,
      "epoch": 1.0137557983398438e-05,
      "step": 84630
    },
    {
      "epoch": 1.0137557983398438e-05,
      "step": 84630,
      "training_step_time": 6.85038685798645
    },
    {
      "epoch": 1.0138511657714844e-05,
      "model_forward_time": 1.74971342086792,
      "step": 84631
    },
    {
      "epoch": 1.0138511657714844e-05,
      "step": 84631,
      "training_step_time": 6.903201580047607
    },
    {
      "epoch": 1.013946533203125e-05,
      "model_forward_time": 1.7542469501495361,
      "step": 84632
    },
    {
      "epoch": 1.013946533203125e-05,
      "step": 84632,
      "training_step_time": 8.816930055618286
    },
    {
      "epoch": 1.0140419006347655e-05,
      "model_forward_time": 1.7694802284240723,
      "step": 84633
    },
    {
      "epoch": 1.0140419006347655e-05,
      "step": 84633,
      "training_step_time": 6.912990093231201
    },
    {
      "epoch": 1.0141372680664063e-05,
      "model_forward_time": 1.3907825946807861,
      "step": 84634
    },
    {
      "epoch": 1.0141372680664063e-05,
      "step": 84634,
      "training_step_time": 7.375279903411865
    },
    {
      "epoch": 1.0142326354980469e-05,
      "model_forward_time": 1.752263069152832,
      "step": 84635
    },
    {
      "epoch": 1.0142326354980469e-05,
      "step": 84635,
      "training_step_time": 7.713462591171265
    },
    {
      "epoch": 1.0143280029296874e-05,
      "model_forward_time": 1.7518911361694336,
      "step": 84636
    },
    {
      "epoch": 1.0143280029296874e-05,
      "step": 84636,
      "training_step_time": 7.371822118759155
    },
    {
      "epoch": 1.0144233703613282e-05,
      "model_forward_time": 1.748570442199707,
      "step": 84637
    },
    {
      "epoch": 1.0144233703613282e-05,
      "step": 84637,
      "training_step_time": 7.687904119491577
    },
    {
      "epoch": 1.0145187377929687e-05,
      "model_forward_time": 1.7513985633850098,
      "step": 84638
    },
    {
      "epoch": 1.0145187377929687e-05,
      "step": 84638,
      "training_step_time": 6.903138637542725
    },
    {
      "epoch": 1.0146141052246093e-05,
      "model_forward_time": 1.7490508556365967,
      "step": 84639
    },
    {
      "epoch": 1.0146141052246093e-05,
      "step": 84639,
      "training_step_time": 6.819510459899902
    },
    {
      "epoch": 1.01470947265625e-05,
      "grad_norm": 0.0929822102189064,
      "learning_rate": 6.31353339375928e-07,
      "loss": 0.0555,
      "step": 84640
    },
    {
      "epoch": 1.01470947265625e-05,
      "model_forward_time": 1.7540934085845947,
      "step": 84640
    },
    {
      "dynamics_loss_avg": 0.05261011496186256,
      "epoch": 1.01470947265625e-05,
      "step": 84640
    },
    {
      "action_loss_avg": 0.0012922533554956318,
      "epoch": 1.01470947265625e-05,
      "step": 84640
    },
    {
      "epoch": 1.01470947265625e-05,
      "step": 84640,
      "training_step_time": 6.80066442489624
    },
    {
      "epoch": 1.0148048400878906e-05,
      "model_forward_time": 1.7623708248138428,
      "step": 84641
    },
    {
      "epoch": 1.0148048400878906e-05,
      "step": 84641,
      "training_step_time": 7.867525100708008
    },
    {
      "epoch": 1.0149002075195312e-05,
      "model_forward_time": 2.8231875896453857,
      "step": 84642
    },
    {
      "epoch": 1.0149002075195312e-05,
      "step": 84642,
      "training_step_time": 7.802198648452759
    },
    {
      "epoch": 1.014995574951172e-05,
      "model_forward_time": 1.7517790794372559,
      "step": 84643
    },
    {
      "epoch": 1.014995574951172e-05,
      "step": 84643,
      "training_step_time": 7.762639760971069
    },
    {
      "epoch": 1.0150909423828125e-05,
      "model_forward_time": 1.7619800567626953,
      "step": 84644
    },
    {
      "epoch": 1.0150909423828125e-05,
      "step": 84644,
      "training_step_time": 7.682177543640137
    },
    {
      "epoch": 1.015186309814453e-05,
      "model_forward_time": 1.7457587718963623,
      "step": 84645
    },
    {
      "epoch": 1.015186309814453e-05,
      "step": 84645,
      "training_step_time": 7.383685827255249
    },
    {
      "epoch": 1.0152816772460938e-05,
      "model_forward_time": 1.7613883018493652,
      "step": 84646
    },
    {
      "epoch": 1.0152816772460938e-05,
      "step": 84646,
      "training_step_time": 7.838907718658447
    },
    {
      "epoch": 1.0153770446777344e-05,
      "model_forward_time": 1.7739951610565186,
      "step": 84647
    },
    {
      "epoch": 1.0153770446777344e-05,
      "step": 84647,
      "training_step_time": 6.810712099075317
    },
    {
      "epoch": 1.015472412109375e-05,
      "model_forward_time": 1.773939847946167,
      "step": 84648
    },
    {
      "epoch": 1.015472412109375e-05,
      "step": 84648,
      "training_step_time": 6.834496974945068
    },
    {
      "epoch": 1.0155677795410155e-05,
      "model_forward_time": 1.7659993171691895,
      "step": 84649
    },
    {
      "epoch": 1.0155677795410155e-05,
      "step": 84649,
      "training_step_time": 6.7859721183776855
    },
    {
      "epoch": 1.0156631469726563e-05,
      "grad_norm": 0.08590856194496155,
      "learning_rate": 6.305493099893395e-07,
      "loss": 0.0554,
      "step": 84650
    },
    {
      "epoch": 1.0156631469726563e-05,
      "model_forward_time": 1.7751076221466064,
      "step": 84650
    },
    {
      "dynamics_loss_avg": 0.04671863932162523,
      "epoch": 1.0156631469726563e-05,
      "step": 84650
    },
    {
      "action_loss_avg": 0.0006361391657264903,
      "epoch": 1.0156631469726563e-05,
      "step": 84650
    },
    {
      "epoch": 1.0156631469726563e-05,
      "step": 84650,
      "training_step_time": 7.816680431365967
    },
    {
      "epoch": 1.0157585144042969e-05,
      "model_forward_time": 2.8228025436401367,
      "step": 84651
    },
    {
      "epoch": 1.0157585144042969e-05,
      "step": 84651,
      "training_step_time": 7.732517242431641
    },
    {
      "epoch": 1.0158538818359374e-05,
      "model_forward_time": 1.7745370864868164,
      "step": 84652
    },
    {
      "epoch": 1.0158538818359374e-05,
      "step": 84652,
      "training_step_time": 7.741390943527222
    },
    {
      "epoch": 1.0159492492675782e-05,
      "model_forward_time": 1.7491135597229004,
      "step": 84653
    },
    {
      "epoch": 1.0159492492675782e-05,
      "step": 84653,
      "training_step_time": 7.644112825393677
    },
    {
      "epoch": 1.0160446166992187e-05,
      "model_forward_time": 1.771205186843872,
      "step": 84654
    },
    {
      "epoch": 1.0160446166992187e-05,
      "step": 84654,
      "training_step_time": 7.240815877914429
    },
    {
      "epoch": 1.0161399841308593e-05,
      "model_forward_time": 1.7678732872009277,
      "step": 84655
    },
    {
      "epoch": 1.0161399841308593e-05,
      "step": 84655,
      "training_step_time": 7.751686334609985
    },
    {
      "epoch": 1.0162353515625e-05,
      "model_forward_time": 1.7714660167694092,
      "step": 84656
    },
    {
      "epoch": 1.0162353515625e-05,
      "step": 84656,
      "training_step_time": 6.775984048843384
    },
    {
      "epoch": 1.0163307189941406e-05,
      "model_forward_time": 1.755584478378296,
      "step": 84657
    },
    {
      "epoch": 1.0163307189941406e-05,
      "step": 84657,
      "training_step_time": 6.747965097427368
    },
    {
      "epoch": 1.0164260864257812e-05,
      "model_forward_time": 1.7476098537445068,
      "step": 84658
    },
    {
      "epoch": 1.0164260864257812e-05,
      "step": 84658,
      "training_step_time": 6.723422050476074
    },
    {
      "epoch": 1.016521453857422e-05,
      "model_forward_time": 1.3928887844085693,
      "step": 84659
    },
    {
      "epoch": 1.016521453857422e-05,
      "step": 84659,
      "training_step_time": 6.818119764328003
    },
    {
      "epoch": 1.0166168212890625e-05,
      "grad_norm": 0.08677057176828384,
      "learning_rate": 6.297457584393097e-07,
      "loss": 0.0562,
      "step": 84660
    },
    {
      "epoch": 1.0166168212890625e-05,
      "model_forward_time": 2.8145477771759033,
      "step": 84660
    },
    {
      "dynamics_loss_avg": 0.05238033067435026,
      "epoch": 1.0166168212890625e-05,
      "step": 84660
    },
    {
      "action_loss_avg": 0.0009382943739183247,
      "epoch": 1.0166168212890625e-05,
      "step": 84660
    },
    {
      "epoch": 1.0166168212890625e-05,
      "step": 84660,
      "training_step_time": 10.151298999786377
    },
    {
      "epoch": 1.0167121887207031e-05,
      "model_forward_time": 1.7462611198425293,
      "step": 84661
    },
    {
      "epoch": 1.0167121887207031e-05,
      "step": 84661,
      "training_step_time": 7.671004295349121
    },
    {
      "epoch": 1.0168075561523438e-05,
      "model_forward_time": 1.7703685760498047,
      "step": 84662
    },
    {
      "epoch": 1.0168075561523438e-05,
      "step": 84662,
      "training_step_time": 8.473041772842407
    },
    {
      "epoch": 1.0169029235839844e-05,
      "model_forward_time": 1.7645776271820068,
      "step": 84663
    },
    {
      "epoch": 1.0169029235839844e-05,
      "step": 84663,
      "training_step_time": 7.331717252731323
    },
    {
      "epoch": 1.016998291015625e-05,
      "model_forward_time": 1.7666263580322266,
      "step": 84664
    },
    {
      "epoch": 1.016998291015625e-05,
      "step": 84664,
      "training_step_time": 7.79533839225769
    },
    {
      "epoch": 1.0170936584472655e-05,
      "model_forward_time": 1.7532081604003906,
      "step": 84665
    },
    {
      "epoch": 1.0170936584472655e-05,
      "step": 84665,
      "training_step_time": 6.7737767696380615
    },
    {
      "epoch": 1.0171890258789063e-05,
      "model_forward_time": 1.7622215747833252,
      "step": 84666
    },
    {
      "epoch": 1.0171890258789063e-05,
      "step": 84666,
      "training_step_time": 6.728543996810913
    },
    {
      "epoch": 1.0172843933105469e-05,
      "model_forward_time": 1.7637217044830322,
      "step": 84667
    },
    {
      "epoch": 1.0172843933105469e-05,
      "step": 84667,
      "training_step_time": 6.718310356140137
    },
    {
      "epoch": 1.0173797607421874e-05,
      "model_forward_time": 1.7610585689544678,
      "step": 84668
    },
    {
      "epoch": 1.0173797607421874e-05,
      "step": 84668,
      "training_step_time": 6.728050470352173
    },
    {
      "epoch": 1.0174751281738282e-05,
      "model_forward_time": 2.8344130516052246,
      "step": 84669
    },
    {
      "epoch": 1.0174751281738282e-05,
      "step": 84669,
      "training_step_time": 9.239563703536987
    },
    {
      "epoch": 1.0175704956054687e-05,
      "grad_norm": 0.08136022835969925,
      "learning_rate": 6.289426848137125e-07,
      "loss": 0.0563,
      "step": 84670
    },
    {
      "epoch": 1.0175704956054687e-05,
      "model_forward_time": 1.737633466720581,
      "step": 84670
    },
    {
      "dynamics_loss_avg": 0.04691687487065792,
      "epoch": 1.0175704956054687e-05,
      "step": 84670
    },
    {
      "action_loss_avg": 0.0011382004158804193,
      "epoch": 1.0175704956054687e-05,
      "step": 84670
    },
    {
      "epoch": 1.0175704956054687e-05,
      "step": 84670,
      "training_step_time": 7.784624099731445
    },
    {
      "epoch": 1.0176658630371093e-05,
      "model_forward_time": 1.7559995651245117,
      "step": 84671
    },
    {
      "epoch": 1.0176658630371093e-05,
      "step": 84671,
      "training_step_time": 7.636180639266968
    },
    {
      "epoch": 1.01776123046875e-05,
      "model_forward_time": 1.7839610576629639,
      "step": 84672
    },
    {
      "epoch": 1.01776123046875e-05,
      "step": 84672,
      "training_step_time": 8.438407182693481
    },
    {
      "epoch": 1.0178565979003906e-05,
      "model_forward_time": 1.7519183158874512,
      "step": 84673
    },
    {
      "epoch": 1.0178565979003906e-05,
      "step": 84673,
      "training_step_time": 8.130668640136719
    },
    {
      "epoch": 1.0179519653320312e-05,
      "model_forward_time": 1.7521607875823975,
      "step": 84674
    },
    {
      "epoch": 1.0179519653320312e-05,
      "step": 84674,
      "training_step_time": 7.683259963989258
    },
    {
      "epoch": 1.018047332763672e-05,
      "model_forward_time": 1.7516064643859863,
      "step": 84675
    },
    {
      "epoch": 1.018047332763672e-05,
      "step": 84675,
      "training_step_time": 6.704633712768555
    },
    {
      "epoch": 1.0181427001953125e-05,
      "model_forward_time": 1.7859232425689697,
      "step": 84676
    },
    {
      "epoch": 1.0181427001953125e-05,
      "step": 84676,
      "training_step_time": 6.750066757202148
    },
    {
      "epoch": 1.0182380676269531e-05,
      "model_forward_time": 1.749833106994629,
      "step": 84677
    },
    {
      "epoch": 1.0182380676269531e-05,
      "step": 84677,
      "training_step_time": 6.751105308532715
    },
    {
      "epoch": 1.0183334350585938e-05,
      "model_forward_time": 1.774116039276123,
      "step": 84678
    },
    {
      "epoch": 1.0183334350585938e-05,
      "step": 84678,
      "training_step_time": 9.24898910522461
    },
    {
      "epoch": 1.0184288024902344e-05,
      "model_forward_time": 1.7544784545898438,
      "step": 84679
    },
    {
      "epoch": 1.0184288024902344e-05,
      "step": 84679,
      "training_step_time": 7.88104248046875
    },
    {
      "epoch": 1.018524169921875e-05,
      "grad_norm": 0.07964161038398743,
      "learning_rate": 6.281400892003714e-07,
      "loss": 0.0572,
      "step": 84680
    },
    {
      "epoch": 1.018524169921875e-05,
      "model_forward_time": 1.749297857284546,
      "step": 84680
    },
    {
      "dynamics_loss_avg": 0.0509584279730916,
      "epoch": 1.018524169921875e-05,
      "step": 84680
    },
    {
      "action_loss_avg": 0.0012643335794564337,
      "epoch": 1.018524169921875e-05,
      "step": 84680
    },
    {
      "epoch": 1.018524169921875e-05,
      "step": 84680,
      "training_step_time": 7.9052886962890625
    },
    {
      "epoch": 1.0186195373535155e-05,
      "model_forward_time": 1.7581470012664795,
      "step": 84681
    },
    {
      "epoch": 1.0186195373535155e-05,
      "step": 84681,
      "training_step_time": 8.02775526046753
    },
    {
      "epoch": 1.0187149047851563e-05,
      "model_forward_time": 1.7652885913848877,
      "step": 84682
    },
    {
      "epoch": 1.0187149047851563e-05,
      "step": 84682,
      "training_step_time": 7.435697078704834
    },
    {
      "epoch": 1.0188102722167969e-05,
      "model_forward_time": 1.7483630180358887,
      "step": 84683
    },
    {
      "epoch": 1.0188102722167969e-05,
      "step": 84683,
      "training_step_time": 7.883511066436768
    },
    {
      "epoch": 1.0189056396484374e-05,
      "model_forward_time": 1.7532498836517334,
      "step": 84684
    },
    {
      "epoch": 1.0189056396484374e-05,
      "step": 84684,
      "training_step_time": 6.908790826797485
    },
    {
      "epoch": 1.0190010070800782e-05,
      "model_forward_time": 1.7481746673583984,
      "step": 84685
    },
    {
      "epoch": 1.0190010070800782e-05,
      "step": 84685,
      "training_step_time": 6.923781156539917
    },
    {
      "epoch": 1.0190963745117187e-05,
      "model_forward_time": 1.767338514328003,
      "step": 84686
    },
    {
      "epoch": 1.0190963745117187e-05,
      "step": 84686,
      "training_step_time": 6.8698835372924805
    },
    {
      "epoch": 1.0191917419433593e-05,
      "model_forward_time": 1.7658801078796387,
      "step": 84687
    },
    {
      "epoch": 1.0191917419433593e-05,
      "step": 84687,
      "training_step_time": 8.833024978637695
    },
    {
      "epoch": 1.019287109375e-05,
      "model_forward_time": 2.807108163833618,
      "step": 84688
    },
    {
      "epoch": 1.019287109375e-05,
      "step": 84688,
      "training_step_time": 9.93030595779419
    },
    {
      "epoch": 1.0193824768066406e-05,
      "model_forward_time": 1.7557685375213623,
      "step": 84689
    },
    {
      "epoch": 1.0193824768066406e-05,
      "step": 84689,
      "training_step_time": 7.850128173828125
    },
    {
      "epoch": 1.0194778442382812e-05,
      "grad_norm": 0.09145299345254898,
      "learning_rate": 6.273379716870581e-07,
      "loss": 0.0572,
      "step": 84690
    },
    {
      "epoch": 1.0194778442382812e-05,
      "model_forward_time": 1.750779628753662,
      "step": 84690
    },
    {
      "dynamics_loss_avg": 0.05389515813440084,
      "epoch": 1.0194778442382812e-05,
      "step": 84690
    },
    {
      "action_loss_avg": 0.0016983060049824417,
      "epoch": 1.0194778442382812e-05,
      "step": 84690
    },
    {
      "epoch": 1.0194778442382812e-05,
      "step": 84690,
      "training_step_time": 7.9514546394348145
    },
    {
      "epoch": 1.019573211669922e-05,
      "model_forward_time": 1.7535336017608643,
      "step": 84691
    },
    {
      "epoch": 1.019573211669922e-05,
      "step": 84691,
      "training_step_time": 7.397847652435303
    },
    {
      "epoch": 1.0196685791015625e-05,
      "model_forward_time": 1.7913076877593994,
      "step": 84692
    },
    {
      "epoch": 1.0196685791015625e-05,
      "step": 84692,
      "training_step_time": 7.837746620178223
    },
    {
      "epoch": 1.0197639465332031e-05,
      "model_forward_time": 1.7652652263641357,
      "step": 84693
    },
    {
      "epoch": 1.0197639465332031e-05,
      "step": 84693,
      "training_step_time": 6.860941410064697
    },
    {
      "epoch": 1.0198593139648438e-05,
      "model_forward_time": 1.7698638439178467,
      "step": 84694
    },
    {
      "epoch": 1.0198593139648438e-05,
      "step": 84694,
      "training_step_time": 6.913696765899658
    },
    {
      "epoch": 1.0199546813964844e-05,
      "model_forward_time": 1.769313097000122,
      "step": 84695
    },
    {
      "epoch": 1.0199546813964844e-05,
      "step": 84695,
      "training_step_time": 6.894047498703003
    },
    {
      "epoch": 1.020050048828125e-05,
      "model_forward_time": 1.7497658729553223,
      "step": 84696
    },
    {
      "epoch": 1.020050048828125e-05,
      "step": 84696,
      "training_step_time": 9.37167501449585
    },
    {
      "epoch": 1.0201454162597655e-05,
      "model_forward_time": 2.8397796154022217,
      "step": 84697
    },
    {
      "epoch": 1.0201454162597655e-05,
      "step": 84697,
      "training_step_time": 7.802795171737671
    },
    {
      "epoch": 1.0202407836914063e-05,
      "model_forward_time": 1.7565655708312988,
      "step": 84698
    },
    {
      "epoch": 1.0202407836914063e-05,
      "step": 84698,
      "training_step_time": 7.752458572387695
    },
    {
      "epoch": 1.0203361511230469e-05,
      "model_forward_time": 1.769892930984497,
      "step": 84699
    },
    {
      "epoch": 1.0203361511230469e-05,
      "step": 84699,
      "training_step_time": 7.928015947341919
    },
    {
      "epoch": 1.0204315185546874e-05,
      "grad_norm": 0.10258512943983078,
      "learning_rate": 6.265363323614887e-07,
      "loss": 0.0571,
      "step": 84700
    },
    {
      "epoch": 1.0204315185546874e-05,
      "model_forward_time": 1.7529633045196533,
      "step": 84700
    },
    {
      "dynamics_loss_avg": 0.060924664977937935,
      "epoch": 1.0204315185546874e-05,
      "step": 84700
    },
    {
      "action_loss_avg": 0.0021807458106195554,
      "epoch": 1.0204315185546874e-05,
      "step": 84700
    },
    {
      "epoch": 1.0204315185546874e-05,
      "step": 84700,
      "training_step_time": 7.3301682472229
    },
    {
      "epoch": 1.0205268859863282e-05,
      "model_forward_time": 1.7485136985778809,
      "step": 84701
    },
    {
      "epoch": 1.0205268859863282e-05,
      "step": 84701,
      "training_step_time": 7.625084400177002
    },
    {
      "epoch": 1.0206222534179687e-05,
      "model_forward_time": 1.754807710647583,
      "step": 84702
    },
    {
      "epoch": 1.0206222534179687e-05,
      "step": 84702,
      "training_step_time": 7.809069871902466
    },
    {
      "epoch": 1.0207176208496093e-05,
      "model_forward_time": 1.7567319869995117,
      "step": 84703
    },
    {
      "epoch": 1.0207176208496093e-05,
      "step": 84703,
      "training_step_time": 6.8761351108551025
    },
    {
      "epoch": 1.02081298828125e-05,
      "model_forward_time": 1.7769622802734375,
      "step": 84704
    },
    {
      "epoch": 1.02081298828125e-05,
      "step": 84704,
      "training_step_time": 6.894680500030518
    },
    {
      "epoch": 1.0209083557128906e-05,
      "model_forward_time": 1.7520017623901367,
      "step": 84705
    },
    {
      "epoch": 1.0209083557128906e-05,
      "step": 84705,
      "training_step_time": 7.5597991943359375
    },
    {
      "epoch": 1.0210037231445312e-05,
      "model_forward_time": 2.8315067291259766,
      "step": 84706
    },
    {
      "epoch": 1.0210037231445312e-05,
      "step": 84706,
      "training_step_time": 7.837918043136597
    },
    {
      "epoch": 1.021099090576172e-05,
      "model_forward_time": 1.7446963787078857,
      "step": 84707
    },
    {
      "epoch": 1.021099090576172e-05,
      "step": 84707,
      "training_step_time": 7.827708721160889
    },
    {
      "epoch": 1.0211944580078125e-05,
      "model_forward_time": 1.756277084350586,
      "step": 84708
    },
    {
      "epoch": 1.0211944580078125e-05,
      "step": 84708,
      "training_step_time": 9.843015432357788
    },
    {
      "epoch": 1.0212898254394531e-05,
      "model_forward_time": 1.7726125717163086,
      "step": 84709
    },
    {
      "epoch": 1.0212898254394531e-05,
      "step": 84709,
      "training_step_time": 7.386643886566162
    },
    {
      "epoch": 1.0213851928710938e-05,
      "grad_norm": 0.08083921670913696,
      "learning_rate": 6.257351713113308e-07,
      "loss": 0.057,
      "step": 84710
    },
    {
      "epoch": 1.0213851928710938e-05,
      "model_forward_time": 1.7542836666107178,
      "step": 84710
    },
    {
      "dynamics_loss_avg": 0.060959587106481194,
      "epoch": 1.0213851928710938e-05,
      "step": 84710
    },
    {
      "action_loss_avg": 0.001412373554194346,
      "epoch": 1.0213851928710938e-05,
      "step": 84710
    },
    {
      "epoch": 1.0213851928710938e-05,
      "step": 84710,
      "training_step_time": 7.70763897895813
    },
    {
      "epoch": 1.0214805603027344e-05,
      "model_forward_time": 1.7520906925201416,
      "step": 84711
    },
    {
      "epoch": 1.0214805603027344e-05,
      "step": 84711,
      "training_step_time": 7.942502021789551
    },
    {
      "epoch": 1.021575927734375e-05,
      "model_forward_time": 1.7545506954193115,
      "step": 84712
    },
    {
      "epoch": 1.021575927734375e-05,
      "step": 84712,
      "training_step_time": 6.944148778915405
    },
    {
      "epoch": 1.0216712951660155e-05,
      "model_forward_time": 1.7675704956054688,
      "step": 84713
    },
    {
      "epoch": 1.0216712951660155e-05,
      "step": 84713,
      "training_step_time": 6.871392011642456
    },
    {
      "epoch": 1.0217666625976563e-05,
      "model_forward_time": 1.7598867416381836,
      "step": 84714
    },
    {
      "epoch": 1.0217666625976563e-05,
      "step": 84714,
      "training_step_time": 7.378404378890991
    },
    {
      "epoch": 1.0218620300292969e-05,
      "model_forward_time": 2.837287425994873,
      "step": 84715
    },
    {
      "epoch": 1.0218620300292969e-05,
      "step": 84715,
      "training_step_time": 9.469445705413818
    },
    {
      "epoch": 1.0219573974609374e-05,
      "model_forward_time": 1.751009225845337,
      "step": 84716
    },
    {
      "epoch": 1.0219573974609374e-05,
      "step": 84716,
      "training_step_time": 9.240548133850098
    },
    {
      "epoch": 1.0220527648925782e-05,
      "model_forward_time": 1.7505087852478027,
      "step": 84717
    },
    {
      "epoch": 1.0220527648925782e-05,
      "step": 84717,
      "training_step_time": 7.730194330215454
    },
    {
      "epoch": 1.0221481323242187e-05,
      "model_forward_time": 1.7584083080291748,
      "step": 84718
    },
    {
      "epoch": 1.0221481323242187e-05,
      "step": 84718,
      "training_step_time": 7.337071418762207
    },
    {
      "epoch": 1.0222434997558593e-05,
      "model_forward_time": 1.751495361328125,
      "step": 84719
    },
    {
      "epoch": 1.0222434997558593e-05,
      "step": 84719,
      "training_step_time": 7.365447759628296
    },
    {
      "epoch": 1.0223388671875e-05,
      "grad_norm": 0.06949477642774582,
      "learning_rate": 6.249344886241982e-07,
      "loss": 0.0568,
      "step": 84720
    },
    {
      "epoch": 1.0223388671875e-05,
      "model_forward_time": 1.7591087818145752,
      "step": 84720
    },
    {
      "dynamics_loss_avg": 0.05421917550265789,
      "epoch": 1.0223388671875e-05,
      "step": 84720
    },
    {
      "action_loss_avg": 0.0008725953550310806,
      "epoch": 1.0223388671875e-05,
      "step": 84720
    },
    {
      "epoch": 1.0223388671875e-05,
      "step": 84720,
      "training_step_time": 7.815935134887695
    },
    {
      "epoch": 1.0224342346191406e-05,
      "model_forward_time": 1.752284288406372,
      "step": 84721
    },
    {
      "epoch": 1.0224342346191406e-05,
      "step": 84721,
      "training_step_time": 6.803496837615967
    },
    {
      "epoch": 1.0225296020507812e-05,
      "model_forward_time": 1.7660541534423828,
      "step": 84722
    },
    {
      "epoch": 1.0225296020507812e-05,
      "step": 84722,
      "training_step_time": 6.794211149215698
    },
    {
      "epoch": 1.022624969482422e-05,
      "model_forward_time": 1.7507789134979248,
      "step": 84723
    },
    {
      "epoch": 1.022624969482422e-05,
      "step": 84723,
      "training_step_time": 8.14688515663147
    },
    {
      "epoch": 1.0227203369140625e-05,
      "model_forward_time": 2.8569536209106445,
      "step": 84724
    },
    {
      "epoch": 1.0227203369140625e-05,
      "step": 84724,
      "training_step_time": 8.416943073272705
    },
    {
      "epoch": 1.0228157043457031e-05,
      "model_forward_time": 1.753422737121582,
      "step": 84725
    },
    {
      "epoch": 1.0228157043457031e-05,
      "step": 84725,
      "training_step_time": 7.735529184341431
    },
    {
      "epoch": 1.0229110717773438e-05,
      "model_forward_time": 1.7613449096679688,
      "step": 84726
    },
    {
      "epoch": 1.0229110717773438e-05,
      "step": 84726,
      "training_step_time": 7.837337255477905
    },
    {
      "epoch": 1.0230064392089844e-05,
      "model_forward_time": 1.7579801082611084,
      "step": 84727
    },
    {
      "epoch": 1.0230064392089844e-05,
      "step": 84727,
      "training_step_time": 7.696903705596924
    },
    {
      "epoch": 1.023101806640625e-05,
      "model_forward_time": 1.752025842666626,
      "step": 84728
    },
    {
      "epoch": 1.023101806640625e-05,
      "step": 84728,
      "training_step_time": 7.336233139038086
    },
    {
      "epoch": 1.0231971740722655e-05,
      "model_forward_time": 1.7481849193572998,
      "step": 84729
    },
    {
      "epoch": 1.0231971740722655e-05,
      "step": 84729,
      "training_step_time": 8.271929740905762
    },
    {
      "epoch": 1.0232925415039063e-05,
      "grad_norm": 0.09691063314676285,
      "learning_rate": 6.24134284387653e-07,
      "loss": 0.0574,
      "step": 84730
    },
    {
      "epoch": 1.0232925415039063e-05,
      "model_forward_time": 1.7520267963409424,
      "step": 84730
    },
    {
      "dynamics_loss_avg": 0.04949539508670568,
      "epoch": 1.0232925415039063e-05,
      "step": 84730
    },
    {
      "action_loss_avg": 0.0012424233005731366,
      "epoch": 1.0232925415039063e-05,
      "step": 84730
    },
    {
      "epoch": 1.0232925415039063e-05,
      "step": 84730,
      "training_step_time": 6.8231518268585205
    },
    {
      "epoch": 1.0233879089355469e-05,
      "model_forward_time": 1.7511835098266602,
      "step": 84731
    },
    {
      "epoch": 1.0233879089355469e-05,
      "step": 84731,
      "training_step_time": 6.7808451652526855
    },
    {
      "epoch": 1.0234832763671874e-05,
      "model_forward_time": 1.776392936706543,
      "step": 84732
    },
    {
      "epoch": 1.0234832763671874e-05,
      "step": 84732,
      "training_step_time": 7.580610036849976
    },
    {
      "epoch": 1.0235786437988282e-05,
      "model_forward_time": 1.7472271919250488,
      "step": 84733
    },
    {
      "epoch": 1.0235786437988282e-05,
      "step": 84733,
      "training_step_time": 8.991222143173218
    },
    {
      "epoch": 1.0236740112304687e-05,
      "model_forward_time": 1.8545622825622559,
      "step": 84734
    },
    {
      "epoch": 1.0236740112304687e-05,
      "step": 84734,
      "training_step_time": 7.8042638301849365
    },
    {
      "epoch": 1.0237693786621093e-05,
      "model_forward_time": 1.7644875049591064,
      "step": 84735
    },
    {
      "epoch": 1.0237693786621093e-05,
      "step": 84735,
      "training_step_time": 8.745955228805542
    },
    {
      "epoch": 1.02386474609375e-05,
      "model_forward_time": 1.396623134613037,
      "step": 84736
    },
    {
      "epoch": 1.02386474609375e-05,
      "step": 84736,
      "training_step_time": 7.720598936080933
    },
    {
      "epoch": 1.0239601135253906e-05,
      "model_forward_time": 1.757206916809082,
      "step": 84737
    },
    {
      "epoch": 1.0239601135253906e-05,
      "step": 84737,
      "training_step_time": 7.325812578201294
    },
    {
      "epoch": 1.0240554809570312e-05,
      "model_forward_time": 1.770721197128296,
      "step": 84738
    },
    {
      "epoch": 1.0240554809570312e-05,
      "step": 84738,
      "training_step_time": 7.620386123657227
    },
    {
      "epoch": 1.024150848388672e-05,
      "model_forward_time": 1.7647590637207031,
      "step": 84739
    },
    {
      "epoch": 1.024150848388672e-05,
      "step": 84739,
      "training_step_time": 7.791273832321167
    },
    {
      "epoch": 1.0242462158203125e-05,
      "grad_norm": 0.12790222465991974,
      "learning_rate": 6.233345586892026e-07,
      "loss": 0.0563,
      "step": 84740
    },
    {
      "epoch": 1.0242462158203125e-05,
      "model_forward_time": 1.7602624893188477,
      "step": 84740
    },
    {
      "dynamics_loss_avg": 0.04865201264619827,
      "epoch": 1.0242462158203125e-05,
      "step": 84740
    },
    {
      "action_loss_avg": 0.0013236305021564476,
      "epoch": 1.0242462158203125e-05,
      "step": 84740
    },
    {
      "epoch": 1.0242462158203125e-05,
      "step": 84740,
      "training_step_time": 6.77433443069458
    },
    {
      "epoch": 1.0243415832519531e-05,
      "model_forward_time": 1.773603916168213,
      "step": 84741
    },
    {
      "epoch": 1.0243415832519531e-05,
      "step": 84741,
      "training_step_time": 7.578969240188599
    },
    {
      "epoch": 1.0244369506835938e-05,
      "model_forward_time": 1.7762219905853271,
      "step": 84742
    },
    {
      "epoch": 1.0244369506835938e-05,
      "step": 84742,
      "training_step_time": 7.762625455856323
    },
    {
      "epoch": 1.0245323181152344e-05,
      "model_forward_time": 2.8512728214263916,
      "step": 84743
    },
    {
      "epoch": 1.0245323181152344e-05,
      "step": 84743,
      "training_step_time": 11.752186298370361
    },
    {
      "epoch": 1.024627685546875e-05,
      "model_forward_time": 1.7622334957122803,
      "step": 84744
    },
    {
      "epoch": 1.024627685546875e-05,
      "step": 84744,
      "training_step_time": 7.747057914733887
    },
    {
      "epoch": 1.0247230529785156e-05,
      "model_forward_time": 1.7643404006958008,
      "step": 84745
    },
    {
      "epoch": 1.0247230529785156e-05,
      "step": 84745,
      "training_step_time": 7.9310142993927
    },
    {
      "epoch": 1.0248184204101563e-05,
      "model_forward_time": 1.7667176723480225,
      "step": 84746
    },
    {
      "epoch": 1.0248184204101563e-05,
      "step": 84746,
      "training_step_time": 8.109013557434082
    },
    {
      "epoch": 1.0249137878417969e-05,
      "model_forward_time": 1.749692678451538,
      "step": 84747
    },
    {
      "epoch": 1.0249137878417969e-05,
      "step": 84747,
      "training_step_time": 7.730522394180298
    },
    {
      "epoch": 1.0250091552734374e-05,
      "model_forward_time": 1.7592031955718994,
      "step": 84748
    },
    {
      "epoch": 1.0250091552734374e-05,
      "step": 84748,
      "training_step_time": 7.987693786621094
    },
    {
      "epoch": 1.0251045227050782e-05,
      "model_forward_time": 1.7566876411437988,
      "step": 84749
    },
    {
      "epoch": 1.0251045227050782e-05,
      "step": 84749,
      "training_step_time": 6.897037982940674
    },
    {
      "epoch": 1.0251998901367188e-05,
      "grad_norm": 0.08338566869497299,
      "learning_rate": 6.225353116163052e-07,
      "loss": 0.055,
      "step": 84750
    },
    {
      "epoch": 1.0251998901367188e-05,
      "model_forward_time": 1.7620468139648438,
      "step": 84750
    },
    {
      "dynamics_loss_avg": 0.048592272028326985,
      "epoch": 1.0251998901367188e-05,
      "step": 84750
    },
    {
      "action_loss_avg": 0.0011284308246104046,
      "epoch": 1.0251998901367188e-05,
      "step": 84750
    },
    {
      "epoch": 1.0251998901367188e-05,
      "step": 84750,
      "training_step_time": 8.4732346534729
    },
    {
      "epoch": 1.0252952575683593e-05,
      "model_forward_time": 1.7538292407989502,
      "step": 84751
    },
    {
      "epoch": 1.0252952575683593e-05,
      "step": 84751,
      "training_step_time": 8.94918942451477
    },
    {
      "epoch": 1.025390625e-05,
      "model_forward_time": 2.8549909591674805,
      "step": 84752
    },
    {
      "epoch": 1.025390625e-05,
      "step": 84752,
      "training_step_time": 9.604015111923218
    },
    {
      "epoch": 1.0254859924316406e-05,
      "model_forward_time": 1.7450721263885498,
      "step": 84753
    },
    {
      "epoch": 1.0254859924316406e-05,
      "step": 84753,
      "training_step_time": 7.823831558227539
    },
    {
      "epoch": 1.0255813598632812e-05,
      "model_forward_time": 1.7613661289215088,
      "step": 84754
    },
    {
      "epoch": 1.0255813598632812e-05,
      "step": 84754,
      "training_step_time": 8.528693675994873
    },
    {
      "epoch": 1.025676727294922e-05,
      "model_forward_time": 1.7558116912841797,
      "step": 84755
    },
    {
      "epoch": 1.025676727294922e-05,
      "step": 84755,
      "training_step_time": 7.384114980697632
    },
    {
      "epoch": 1.0257720947265625e-05,
      "model_forward_time": 1.7481112480163574,
      "step": 84756
    },
    {
      "epoch": 1.0257720947265625e-05,
      "step": 84756,
      "training_step_time": 7.730136156082153
    },
    {
      "epoch": 1.0258674621582031e-05,
      "model_forward_time": 1.7794251441955566,
      "step": 84757
    },
    {
      "epoch": 1.0258674621582031e-05,
      "step": 84757,
      "training_step_time": 7.946422100067139
    },
    {
      "epoch": 1.0259628295898437e-05,
      "model_forward_time": 1.7624731063842773,
      "step": 84758
    },
    {
      "epoch": 1.0259628295898437e-05,
      "step": 84758,
      "training_step_time": 6.914093732833862
    },
    {
      "epoch": 1.0260581970214844e-05,
      "model_forward_time": 1.7612566947937012,
      "step": 84759
    },
    {
      "epoch": 1.0260581970214844e-05,
      "step": 84759,
      "training_step_time": 7.323241233825684
    },
    {
      "epoch": 1.026153564453125e-05,
      "grad_norm": 0.09702335298061371,
      "learning_rate": 6.217365432563627e-07,
      "loss": 0.0561,
      "step": 84760
    },
    {
      "epoch": 1.026153564453125e-05,
      "model_forward_time": 1.7510483264923096,
      "step": 84760
    },
    {
      "dynamics_loss_avg": 0.053520156629383565,
      "epoch": 1.026153564453125e-05,
      "step": 84760
    },
    {
      "action_loss_avg": 0.0010414397249405738,
      "epoch": 1.026153564453125e-05,
      "step": 84760
    },
    {
      "epoch": 1.026153564453125e-05,
      "step": 84760,
      "training_step_time": 7.750678539276123
    },
    {
      "epoch": 1.0262489318847656e-05,
      "model_forward_time": 2.918766736984253,
      "step": 84761
    },
    {
      "epoch": 1.0262489318847656e-05,
      "step": 84761,
      "training_step_time": 7.891860008239746
    },
    {
      "epoch": 1.0263442993164063e-05,
      "model_forward_time": 1.758681297302246,
      "step": 84762
    },
    {
      "epoch": 1.0263442993164063e-05,
      "step": 84762,
      "training_step_time": 9.351539850234985
    },
    {
      "epoch": 1.0264396667480469e-05,
      "model_forward_time": 1.7567012310028076,
      "step": 84763
    },
    {
      "epoch": 1.0264396667480469e-05,
      "step": 84763,
      "training_step_time": 7.798576831817627
    },
    {
      "epoch": 1.0265350341796874e-05,
      "model_forward_time": 1.7719476222991943,
      "step": 84764
    },
    {
      "epoch": 1.0265350341796874e-05,
      "step": 84764,
      "training_step_time": 7.283690690994263
    },
    {
      "epoch": 1.0266304016113282e-05,
      "model_forward_time": 1.761127233505249,
      "step": 84765
    },
    {
      "epoch": 1.0266304016113282e-05,
      "step": 84765,
      "training_step_time": 7.437947511672974
    },
    {
      "epoch": 1.0267257690429688e-05,
      "model_forward_time": 1.7588167190551758,
      "step": 84766
    },
    {
      "epoch": 1.0267257690429688e-05,
      "step": 84766,
      "training_step_time": 7.866079092025757
    },
    {
      "epoch": 1.0268211364746093e-05,
      "model_forward_time": 1.7644157409667969,
      "step": 84767
    },
    {
      "epoch": 1.0268211364746093e-05,
      "step": 84767,
      "training_step_time": 6.813840866088867
    },
    {
      "epoch": 1.02691650390625e-05,
      "model_forward_time": 1.7765514850616455,
      "step": 84768
    },
    {
      "epoch": 1.02691650390625e-05,
      "step": 84768,
      "training_step_time": 7.2789788246154785
    },
    {
      "epoch": 1.0270118713378906e-05,
      "model_forward_time": 1.771512508392334,
      "step": 84769
    },
    {
      "epoch": 1.0270118713378906e-05,
      "step": 84769,
      "training_step_time": 9.411408185958862
    },
    {
      "epoch": 1.0271072387695312e-05,
      "grad_norm": 0.06663443148136139,
      "learning_rate": 6.209382536967317e-07,
      "loss": 0.0556,
      "step": 84770
    },
    {
      "epoch": 1.0271072387695312e-05,
      "model_forward_time": 2.8806731700897217,
      "step": 84770
    },
    {
      "dynamics_loss_avg": 0.04006642922759056,
      "epoch": 1.0271072387695312e-05,
      "step": 84770
    },
    {
      "action_loss_avg": 0.0022153163270559164,
      "epoch": 1.0271072387695312e-05,
      "step": 84770
    },
    {
      "epoch": 1.0271072387695312e-05,
      "step": 84770,
      "training_step_time": 10.431794166564941
    },
    {
      "epoch": 1.027202606201172e-05,
      "model_forward_time": 1.748840093612671,
      "step": 84771
    },
    {
      "epoch": 1.027202606201172e-05,
      "step": 84771,
      "training_step_time": 8.790217399597168
    },
    {
      "epoch": 1.0272979736328125e-05,
      "model_forward_time": 1.77640700340271,
      "step": 84772
    },
    {
      "epoch": 1.0272979736328125e-05,
      "step": 84772,
      "training_step_time": 7.664996862411499
    },
    {
      "epoch": 1.0273933410644531e-05,
      "model_forward_time": 1.7763140201568604,
      "step": 84773
    },
    {
      "epoch": 1.0273933410644531e-05,
      "step": 84773,
      "training_step_time": 8.6434326171875
    },
    {
      "epoch": 1.0274887084960937e-05,
      "model_forward_time": 1.780660629272461,
      "step": 84774
    },
    {
      "epoch": 1.0274887084960937e-05,
      "step": 84774,
      "training_step_time": 7.3725786209106445
    },
    {
      "epoch": 1.0275840759277344e-05,
      "model_forward_time": 1.7752912044525146,
      "step": 84775
    },
    {
      "epoch": 1.0275840759277344e-05,
      "step": 84775,
      "training_step_time": 7.774177312850952
    },
    {
      "epoch": 1.027679443359375e-05,
      "model_forward_time": 1.7603509426116943,
      "step": 84776
    },
    {
      "epoch": 1.027679443359375e-05,
      "step": 84776,
      "training_step_time": 6.79022216796875
    },
    {
      "epoch": 1.0277748107910156e-05,
      "model_forward_time": 1.7624835968017578,
      "step": 84777
    },
    {
      "epoch": 1.0277748107910156e-05,
      "step": 84777,
      "training_step_time": 7.248655796051025
    },
    {
      "epoch": 1.0278701782226563e-05,
      "model_forward_time": 1.7598893642425537,
      "step": 84778
    },
    {
      "epoch": 1.0278701782226563e-05,
      "step": 84778,
      "training_step_time": 8.636632204055786
    },
    {
      "epoch": 1.0279655456542969e-05,
      "model_forward_time": 2.8267924785614014,
      "step": 84779
    },
    {
      "epoch": 1.0279655456542969e-05,
      "step": 84779,
      "training_step_time": 9.535043716430664
    },
    {
      "epoch": 1.0280609130859374e-05,
      "grad_norm": 0.0950528159737587,
      "learning_rate": 6.201404430247082e-07,
      "loss": 0.0561,
      "step": 84780
    },
    {
      "epoch": 1.0280609130859374e-05,
      "model_forward_time": 1.7435555458068848,
      "step": 84780
    },
    {
      "dynamics_loss_avg": 0.042733976244926454,
      "epoch": 1.0280609130859374e-05,
      "step": 84780
    },
    {
      "action_loss_avg": 0.001078988640801981,
      "epoch": 1.0280609130859374e-05,
      "step": 84780
    },
    {
      "epoch": 1.0280609130859374e-05,
      "step": 84780,
      "training_step_time": 9.02800726890564
    },
    {
      "epoch": 1.0281562805175782e-05,
      "model_forward_time": 1.7727653980255127,
      "step": 84781
    },
    {
      "epoch": 1.0281562805175782e-05,
      "step": 84781,
      "training_step_time": 8.227599382400513
    },
    {
      "epoch": 1.0282516479492188e-05,
      "model_forward_time": 1.767275094985962,
      "step": 84782
    },
    {
      "epoch": 1.0282516479492188e-05,
      "step": 84782,
      "training_step_time": 7.783865451812744
    },
    {
      "epoch": 1.0283470153808593e-05,
      "model_forward_time": 1.7719786167144775,
      "step": 84783
    },
    {
      "epoch": 1.0283470153808593e-05,
      "step": 84783,
      "training_step_time": 7.21085000038147
    },
    {
      "epoch": 1.0284423828125e-05,
      "model_forward_time": 1.7679245471954346,
      "step": 84784
    },
    {
      "epoch": 1.0284423828125e-05,
      "step": 84784,
      "training_step_time": 7.568442344665527
    },
    {
      "epoch": 1.0285377502441406e-05,
      "model_forward_time": 1.7506775856018066,
      "step": 84785
    },
    {
      "epoch": 1.0285377502441406e-05,
      "step": 84785,
      "training_step_time": 7.7087562084198
    },
    {
      "epoch": 1.0286331176757812e-05,
      "model_forward_time": 1.7682578563690186,
      "step": 84786
    },
    {
      "epoch": 1.0286331176757812e-05,
      "step": 84786,
      "training_step_time": 7.231088161468506
    },
    {
      "epoch": 1.028728485107422e-05,
      "model_forward_time": 1.764984130859375,
      "step": 84787
    },
    {
      "epoch": 1.028728485107422e-05,
      "step": 84787,
      "training_step_time": 8.470242738723755
    },
    {
      "epoch": 1.0288238525390625e-05,
      "model_forward_time": 2.8140065670013428,
      "step": 84788
    },
    {
      "epoch": 1.0288238525390625e-05,
      "step": 84788,
      "training_step_time": 9.352257013320923
    },
    {
      "epoch": 1.0289192199707031e-05,
      "model_forward_time": 1.8496694564819336,
      "step": 84789
    },
    {
      "epoch": 1.0289192199707031e-05,
      "step": 84789,
      "training_step_time": 9.318665981292725
    },
    {
      "epoch": 1.0290145874023437e-05,
      "grad_norm": 0.1048101931810379,
      "learning_rate": 6.193431113275417e-07,
      "loss": 0.055,
      "step": 84790
    },
    {
      "epoch": 1.0290145874023437e-05,
      "model_forward_time": 1.747957468032837,
      "step": 84790
    },
    {
      "dynamics_loss_avg": 0.05497403740882874,
      "epoch": 1.0290145874023437e-05,
      "step": 84790
    },
    {
      "action_loss_avg": 0.0017202220013132318,
      "epoch": 1.0290145874023437e-05,
      "step": 84790
    },
    {
      "epoch": 1.0290145874023437e-05,
      "step": 84790,
      "training_step_time": 8.53836703300476
    },
    {
      "epoch": 1.0291099548339844e-05,
      "model_forward_time": 1.748861312866211,
      "step": 84791
    },
    {
      "epoch": 1.0291099548339844e-05,
      "step": 84791,
      "training_step_time": 7.834685802459717
    },
    {
      "epoch": 1.029205322265625e-05,
      "model_forward_time": 1.7624802589416504,
      "step": 84792
    },
    {
      "epoch": 1.029205322265625e-05,
      "step": 84792,
      "training_step_time": 7.395764589309692
    },
    {
      "epoch": 1.0293006896972656e-05,
      "model_forward_time": 1.7619366645812988,
      "step": 84793
    },
    {
      "epoch": 1.0293006896972656e-05,
      "step": 84793,
      "training_step_time": 7.485163450241089
    },
    {
      "epoch": 1.0293960571289063e-05,
      "model_forward_time": 1.754577398300171,
      "step": 84794
    },
    {
      "epoch": 1.0293960571289063e-05,
      "step": 84794,
      "training_step_time": 7.862796306610107
    },
    {
      "epoch": 1.0294914245605469e-05,
      "model_forward_time": 1.7635350227355957,
      "step": 84795
    },
    {
      "epoch": 1.0294914245605469e-05,
      "step": 84795,
      "training_step_time": 7.616070985794067
    },
    {
      "epoch": 1.0295867919921874e-05,
      "model_forward_time": 1.7563722133636475,
      "step": 84796
    },
    {
      "epoch": 1.0295867919921874e-05,
      "step": 84796,
      "training_step_time": 7.715966701507568
    },
    {
      "epoch": 1.0296821594238282e-05,
      "model_forward_time": 2.8024892807006836,
      "step": 84797
    },
    {
      "epoch": 1.0296821594238282e-05,
      "step": 84797,
      "training_step_time": 10.461904525756836
    },
    {
      "epoch": 1.0297775268554688e-05,
      "model_forward_time": 1.8887548446655273,
      "step": 84798
    },
    {
      "epoch": 1.0297775268554688e-05,
      "step": 84798,
      "training_step_time": 7.771078109741211
    },
    {
      "epoch": 1.0298728942871093e-05,
      "model_forward_time": 1.757760763168335,
      "step": 84799
    },
    {
      "epoch": 1.0298728942871093e-05,
      "step": 84799,
      "training_step_time": 8.41662073135376
    },
    {
      "epoch": 1.02996826171875e-05,
      "grad_norm": 0.11395775526762009,
      "learning_rate": 6.185462586924257e-07,
      "loss": 0.0552,
      "step": 84800
    },
    {
      "epoch": 1.02996826171875e-05,
      "model_forward_time": 1.7542688846588135,
      "step": 84800
    },
    {
      "dynamics_loss_avg": 0.06408204473555087,
      "epoch": 1.02996826171875e-05,
      "step": 84800
    },
    {
      "action_loss_avg": 0.0010698599915485829,
      "epoch": 1.02996826171875e-05,
      "step": 84800
    },
    {
      "epoch": 1.02996826171875e-05,
      "step": 84800,
      "training_step_time": 7.960047960281372
    },
    {
      "epoch": 1.0300636291503906e-05,
      "model_forward_time": 1.7739009857177734,
      "step": 84801
    },
    {
      "epoch": 1.0300636291503906e-05,
      "step": 84801,
      "training_step_time": 7.7237043380737305
    },
    {
      "epoch": 1.0301589965820312e-05,
      "model_forward_time": 1.7642571926116943,
      "step": 84802
    },
    {
      "epoch": 1.0301589965820312e-05,
      "step": 84802,
      "training_step_time": 7.579752206802368
    },
    {
      "epoch": 1.030254364013672e-05,
      "model_forward_time": 1.782020092010498,
      "step": 84803
    },
    {
      "epoch": 1.030254364013672e-05,
      "step": 84803,
      "training_step_time": 7.8826282024383545
    },
    {
      "epoch": 1.0303497314453125e-05,
      "model_forward_time": 1.7751657962799072,
      "step": 84804
    },
    {
      "epoch": 1.0303497314453125e-05,
      "step": 84804,
      "training_step_time": 7.615463018417358
    },
    {
      "epoch": 1.0304450988769531e-05,
      "model_forward_time": 1.774965763092041,
      "step": 84805
    },
    {
      "epoch": 1.0304450988769531e-05,
      "step": 84805,
      "training_step_time": 7.680877685546875
    },
    {
      "epoch": 1.0305404663085937e-05,
      "model_forward_time": 2.840824842453003,
      "step": 84806
    },
    {
      "epoch": 1.0305404663085937e-05,
      "step": 84806,
      "training_step_time": 7.718141555786133
    },
    {
      "epoch": 1.0306358337402344e-05,
      "model_forward_time": 1.8762400150299072,
      "step": 84807
    },
    {
      "epoch": 1.0306358337402344e-05,
      "step": 84807,
      "training_step_time": 7.822828769683838
    },
    {
      "epoch": 1.030731201171875e-05,
      "model_forward_time": 1.756941318511963,
      "step": 84808
    },
    {
      "epoch": 1.030731201171875e-05,
      "step": 84808,
      "training_step_time": 8.681397438049316
    },
    {
      "epoch": 1.0308265686035156e-05,
      "model_forward_time": 1.7591049671173096,
      "step": 84809
    },
    {
      "epoch": 1.0308265686035156e-05,
      "step": 84809,
      "training_step_time": 7.986363410949707
    },
    {
      "epoch": 1.0309219360351563e-05,
      "grad_norm": 0.14325334131717682,
      "learning_rate": 6.177498852065039e-07,
      "loss": 0.0534,
      "step": 84810
    },
    {
      "epoch": 1.0309219360351563e-05,
      "model_forward_time": 1.7554030418395996,
      "step": 84810
    },
    {
      "dynamics_loss_avg": 0.061240654718130824,
      "epoch": 1.0309219360351563e-05,
      "step": 84810
    },
    {
      "action_loss_avg": 0.0010227099868643564,
      "epoch": 1.0309219360351563e-05,
      "step": 84810
    },
    {
      "epoch": 1.0309219360351563e-05,
      "step": 84810,
      "training_step_time": 7.2968316078186035
    },
    {
      "epoch": 1.0310173034667969e-05,
      "model_forward_time": 1.7744431495666504,
      "step": 84811
    },
    {
      "epoch": 1.0310173034667969e-05,
      "step": 84811,
      "training_step_time": 8.594277620315552
    },
    {
      "epoch": 1.0311126708984374e-05,
      "model_forward_time": 1.7799298763275146,
      "step": 84812
    },
    {
      "epoch": 1.0311126708984374e-05,
      "step": 84812,
      "training_step_time": 7.893311500549316
    },
    {
      "epoch": 1.0312080383300782e-05,
      "model_forward_time": 1.7595508098602295,
      "step": 84813
    },
    {
      "epoch": 1.0312080383300782e-05,
      "step": 84813,
      "training_step_time": 7.536604404449463
    },
    {
      "epoch": 1.0313034057617188e-05,
      "model_forward_time": 1.757819414138794,
      "step": 84814
    },
    {
      "epoch": 1.0313034057617188e-05,
      "step": 84814,
      "training_step_time": 7.707639217376709
    },
    {
      "epoch": 1.0313987731933593e-05,
      "model_forward_time": 2.7995543479919434,
      "step": 84815
    },
    {
      "epoch": 1.0313987731933593e-05,
      "step": 84815,
      "training_step_time": 8.430439710617065
    },
    {
      "epoch": 1.031494140625e-05,
      "model_forward_time": 1.8848400115966797,
      "step": 84816
    },
    {
      "epoch": 1.031494140625e-05,
      "step": 84816,
      "training_step_time": 8.678943395614624
    },
    {
      "epoch": 1.0315895080566406e-05,
      "model_forward_time": 1.7669684886932373,
      "step": 84817
    },
    {
      "epoch": 1.0315895080566406e-05,
      "step": 84817,
      "training_step_time": 9.393371105194092
    },
    {
      "epoch": 1.0316848754882812e-05,
      "model_forward_time": 1.7629332542419434,
      "step": 84818
    },
    {
      "epoch": 1.0316848754882812e-05,
      "step": 84818,
      "training_step_time": 8.407976150512695
    },
    {
      "epoch": 1.031780242919922e-05,
      "model_forward_time": 1.778141975402832,
      "step": 84819
    },
    {
      "epoch": 1.031780242919922e-05,
      "step": 84819,
      "training_step_time": 7.808335542678833
    },
    {
      "epoch": 1.0318756103515625e-05,
      "grad_norm": 0.2195844203233719,
      "learning_rate": 6.169539909568656e-07,
      "loss": 0.054,
      "step": 84820
    },
    {
      "epoch": 1.0318756103515625e-05,
      "model_forward_time": 1.7627556324005127,
      "step": 84820
    },
    {
      "dynamics_loss_avg": 0.04341751961037517,
      "epoch": 1.0318756103515625e-05,
      "step": 84820
    },
    {
      "action_loss_avg": 0.0010382900698459707,
      "epoch": 1.0318756103515625e-05,
      "step": 84820
    },
    {
      "epoch": 1.0318756103515625e-05,
      "step": 84820,
      "training_step_time": 7.742573022842407
    },
    {
      "epoch": 1.0319709777832031e-05,
      "model_forward_time": 1.7734816074371338,
      "step": 84821
    },
    {
      "epoch": 1.0319709777832031e-05,
      "step": 84821,
      "training_step_time": 7.9477410316467285
    },
    {
      "epoch": 1.0320663452148437e-05,
      "model_forward_time": 1.7598130702972412,
      "step": 84822
    },
    {
      "epoch": 1.0320663452148437e-05,
      "step": 84822,
      "training_step_time": 7.533679246902466
    },
    {
      "epoch": 1.0321617126464844e-05,
      "model_forward_time": 1.7841565608978271,
      "step": 84823
    },
    {
      "epoch": 1.0321617126464844e-05,
      "step": 84823,
      "training_step_time": 7.697127103805542
    },
    {
      "epoch": 1.032257080078125e-05,
      "model_forward_time": 2.8221981525421143,
      "step": 84824
    },
    {
      "epoch": 1.032257080078125e-05,
      "step": 84824,
      "training_step_time": 7.681208848953247
    },
    {
      "epoch": 1.0323524475097656e-05,
      "model_forward_time": 1.8735251426696777,
      "step": 84825
    },
    {
      "epoch": 1.0323524475097656e-05,
      "step": 84825,
      "training_step_time": 8.49941372871399
    },
    {
      "epoch": 1.0324478149414063e-05,
      "model_forward_time": 1.7585551738739014,
      "step": 84826
    },
    {
      "epoch": 1.0324478149414063e-05,
      "step": 84826,
      "training_step_time": 8.74933409690857
    },
    {
      "epoch": 1.0325431823730469e-05,
      "model_forward_time": 1.743001937866211,
      "step": 84827
    },
    {
      "epoch": 1.0325431823730469e-05,
      "step": 84827,
      "training_step_time": 7.684390068054199
    },
    {
      "epoch": 1.0326385498046874e-05,
      "model_forward_time": 1.780303955078125,
      "step": 84828
    },
    {
      "epoch": 1.0326385498046874e-05,
      "step": 84828,
      "training_step_time": 7.726765155792236
    },
    {
      "epoch": 1.0327339172363282e-05,
      "model_forward_time": 1.783571481704712,
      "step": 84829
    },
    {
      "epoch": 1.0327339172363282e-05,
      "step": 84829,
      "training_step_time": 7.537691116333008
    },
    {
      "epoch": 1.0328292846679688e-05,
      "grad_norm": 0.07991354167461395,
      "learning_rate": 6.161585760305489e-07,
      "loss": 0.0532,
      "step": 84830
    },
    {
      "epoch": 1.0328292846679688e-05,
      "model_forward_time": 1.7773094177246094,
      "step": 84830
    },
    {
      "dynamics_loss_avg": 0.054085152596235274,
      "epoch": 1.0328292846679688e-05,
      "step": 84830
    },
    {
      "action_loss_avg": 0.0015866072062635794,
      "epoch": 1.0328292846679688e-05,
      "step": 84830
    },
    {
      "epoch": 1.0328292846679688e-05,
      "step": 84830,
      "training_step_time": 8.455230236053467
    },
    {
      "epoch": 1.0329246520996093e-05,
      "model_forward_time": 1.7581441402435303,
      "step": 84831
    },
    {
      "epoch": 1.0329246520996093e-05,
      "step": 84831,
      "training_step_time": 7.497767686843872
    },
    {
      "epoch": 1.03302001953125e-05,
      "model_forward_time": 1.764822244644165,
      "step": 84832
    },
    {
      "epoch": 1.03302001953125e-05,
      "step": 84832,
      "training_step_time": 7.687346458435059
    },
    {
      "epoch": 1.0331153869628906e-05,
      "model_forward_time": 2.8141045570373535,
      "step": 84833
    },
    {
      "epoch": 1.0331153869628906e-05,
      "step": 84833,
      "training_step_time": 8.300432682037354
    },
    {
      "epoch": 1.0332107543945312e-05,
      "model_forward_time": 1.868359088897705,
      "step": 84834
    },
    {
      "epoch": 1.0332107543945312e-05,
      "step": 84834,
      "training_step_time": 8.468864440917969
    },
    {
      "epoch": 1.033306121826172e-05,
      "model_forward_time": 1.7693004608154297,
      "step": 84835
    },
    {
      "epoch": 1.033306121826172e-05,
      "step": 84835,
      "training_step_time": 9.272894382476807
    },
    {
      "epoch": 1.0334014892578125e-05,
      "model_forward_time": 1.7428529262542725,
      "step": 84836
    },
    {
      "epoch": 1.0334014892578125e-05,
      "step": 84836,
      "training_step_time": 9.757653713226318
    },
    {
      "epoch": 1.0334968566894531e-05,
      "model_forward_time": 1.759082555770874,
      "step": 84837
    },
    {
      "epoch": 1.0334968566894531e-05,
      "step": 84837,
      "training_step_time": 7.66526460647583
    },
    {
      "epoch": 1.0335922241210937e-05,
      "model_forward_time": 1.7571969032287598,
      "step": 84838
    },
    {
      "epoch": 1.0335922241210937e-05,
      "step": 84838,
      "training_step_time": 7.5172436237335205
    },
    {
      "epoch": 1.0336875915527344e-05,
      "model_forward_time": 1.7786633968353271,
      "step": 84839
    },
    {
      "epoch": 1.0336875915527344e-05,
      "step": 84839,
      "training_step_time": 7.488346099853516
    },
    {
      "epoch": 1.033782958984375e-05,
      "grad_norm": 0.15077954530715942,
      "learning_rate": 6.153636405145392e-07,
      "loss": 0.0525,
      "step": 84840
    },
    {
      "epoch": 1.033782958984375e-05,
      "model_forward_time": 1.7621259689331055,
      "step": 84840
    },
    {
      "dynamics_loss_avg": 0.04751852713525295,
      "epoch": 1.033782958984375e-05,
      "step": 84840
    },
    {
      "action_loss_avg": 0.0013258078863145784,
      "epoch": 1.033782958984375e-05,
      "step": 84840
    },
    {
      "epoch": 1.033782958984375e-05,
      "step": 84840,
      "training_step_time": 7.520657300949097
    },
    {
      "epoch": 1.0338783264160156e-05,
      "model_forward_time": 1.7715277671813965,
      "step": 84841
    },
    {
      "epoch": 1.0338783264160156e-05,
      "step": 84841,
      "training_step_time": 7.575465440750122
    },
    {
      "epoch": 1.0339736938476563e-05,
      "model_forward_time": 2.8245041370391846,
      "step": 84842
    },
    {
      "epoch": 1.0339736938476563e-05,
      "step": 84842,
      "training_step_time": 7.5307488441467285
    },
    {
      "epoch": 1.0340690612792969e-05,
      "model_forward_time": 1.892904281616211,
      "step": 84843
    },
    {
      "epoch": 1.0340690612792969e-05,
      "step": 84843,
      "training_step_time": 10.861752033233643
    },
    {
      "epoch": 1.0341644287109374e-05,
      "model_forward_time": 1.7466051578521729,
      "step": 84844
    },
    {
      "epoch": 1.0341644287109374e-05,
      "step": 84844,
      "training_step_time": 7.636006116867065
    },
    {
      "epoch": 1.0342597961425782e-05,
      "model_forward_time": 1.741943120956421,
      "step": 84845
    },
    {
      "epoch": 1.0342597961425782e-05,
      "step": 84845,
      "training_step_time": 7.561250925064087
    },
    {
      "epoch": 1.0343551635742188e-05,
      "model_forward_time": 1.7654354572296143,
      "step": 84846
    },
    {
      "epoch": 1.0343551635742188e-05,
      "step": 84846,
      "training_step_time": 8.639062881469727
    },
    {
      "epoch": 1.0344505310058593e-05,
      "model_forward_time": 1.7787270545959473,
      "step": 84847
    },
    {
      "epoch": 1.0344505310058593e-05,
      "step": 84847,
      "training_step_time": 8.596827983856201
    },
    {
      "epoch": 1.0345458984375e-05,
      "model_forward_time": 1.775240182876587,
      "step": 84848
    },
    {
      "epoch": 1.0345458984375e-05,
      "step": 84848,
      "training_step_time": 7.302155017852783
    },
    {
      "epoch": 1.0346412658691406e-05,
      "model_forward_time": 1.751884937286377,
      "step": 84849
    },
    {
      "epoch": 1.0346412658691406e-05,
      "step": 84849,
      "training_step_time": 7.376479864120483
    },
    {
      "epoch": 1.0347366333007812e-05,
      "grad_norm": 0.12262086570262909,
      "learning_rate": 6.145691844957702e-07,
      "loss": 0.0536,
      "step": 84850
    },
    {
      "epoch": 1.0347366333007812e-05,
      "model_forward_time": 1.7549309730529785,
      "step": 84850
    },
    {
      "dynamics_loss_avg": 0.04936151560395956,
      "epoch": 1.0347366333007812e-05,
      "step": 84850
    },
    {
      "action_loss_avg": 0.0008772597531788052,
      "epoch": 1.0347366333007812e-05,
      "step": 84850
    },
    {
      "epoch": 1.0347366333007812e-05,
      "step": 84850,
      "training_step_time": 7.595593214035034
    },
    {
      "epoch": 1.034832000732422e-05,
      "model_forward_time": 2.83677077293396,
      "step": 84851
    },
    {
      "epoch": 1.034832000732422e-05,
      "step": 84851,
      "training_step_time": 7.5564634799957275
    },
    {
      "epoch": 1.0349273681640625e-05,
      "model_forward_time": 1.8972272872924805,
      "step": 84852
    },
    {
      "epoch": 1.0349273681640625e-05,
      "step": 84852,
      "training_step_time": 9.414003372192383
    },
    {
      "epoch": 1.0350227355957031e-05,
      "model_forward_time": 1.7476222515106201,
      "step": 84853
    },
    {
      "epoch": 1.0350227355957031e-05,
      "step": 84853,
      "training_step_time": 7.563847541809082
    },
    {
      "epoch": 1.0351181030273437e-05,
      "model_forward_time": 1.7468714714050293,
      "step": 84854
    },
    {
      "epoch": 1.0351181030273437e-05,
      "step": 84854,
      "training_step_time": 9.200813293457031
    },
    {
      "epoch": 1.0352134704589844e-05,
      "model_forward_time": 1.7348179817199707,
      "step": 84855
    },
    {
      "epoch": 1.0352134704589844e-05,
      "step": 84855,
      "training_step_time": 8.396331787109375
    },
    {
      "epoch": 1.035308837890625e-05,
      "model_forward_time": 1.7550115585327148,
      "step": 84856
    },
    {
      "epoch": 1.035308837890625e-05,
      "step": 84856,
      "training_step_time": 7.591788053512573
    },
    {
      "epoch": 1.0354042053222656e-05,
      "model_forward_time": 1.3455684185028076,
      "step": 84857
    },
    {
      "epoch": 1.0354042053222656e-05,
      "step": 84857,
      "training_step_time": 8.149389028549194
    },
    {
      "epoch": 1.0354995727539063e-05,
      "model_forward_time": 1.7375397682189941,
      "step": 84858
    },
    {
      "epoch": 1.0354995727539063e-05,
      "step": 84858,
      "training_step_time": 7.432956218719482
    },
    {
      "epoch": 1.0355949401855469e-05,
      "model_forward_time": 1.745558261871338,
      "step": 84859
    },
    {
      "epoch": 1.0355949401855469e-05,
      "step": 84859,
      "training_step_time": 7.604739665985107
    },
    {
      "epoch": 1.0356903076171874e-05,
      "grad_norm": 0.5138217806816101,
      "learning_rate": 6.137752080611215e-07,
      "loss": 0.053,
      "step": 84860
    },
    {
      "epoch": 1.0356903076171874e-05,
      "model_forward_time": 2.3511569499969482,
      "step": 84860
    },
    {
      "dynamics_loss_avg": 0.04975330512970686,
      "epoch": 1.0356903076171874e-05,
      "step": 84860
    },
    {
      "action_loss_avg": 0.001441654414520599,
      "epoch": 1.0356903076171874e-05,
      "step": 84860
    },
    {
      "epoch": 1.0356903076171874e-05,
      "step": 84860,
      "training_step_time": 7.036371946334839
    },
    {
      "epoch": 1.0357856750488282e-05,
      "model_forward_time": 1.7427775859832764,
      "step": 84861
    },
    {
      "epoch": 1.0357856750488282e-05,
      "step": 84861,
      "training_step_time": 8.530639410018921
    },
    {
      "epoch": 1.0358810424804688e-05,
      "model_forward_time": 1.7540771961212158,
      "step": 84862
    },
    {
      "epoch": 1.0358810424804688e-05,
      "step": 84862,
      "training_step_time": 9.128252983093262
    },
    {
      "epoch": 1.0359764099121093e-05,
      "model_forward_time": 1.7714214324951172,
      "step": 84863
    },
    {
      "epoch": 1.0359764099121093e-05,
      "step": 84863,
      "training_step_time": 9.432287216186523
    },
    {
      "epoch": 1.03607177734375e-05,
      "model_forward_time": 1.7379837036132812,
      "step": 84864
    },
    {
      "epoch": 1.03607177734375e-05,
      "step": 84864,
      "training_step_time": 9.341060161590576
    },
    {
      "epoch": 1.0361671447753906e-05,
      "model_forward_time": 1.7466115951538086,
      "step": 84865
    },
    {
      "epoch": 1.0361671447753906e-05,
      "step": 84865,
      "training_step_time": 7.733028888702393
    },
    {
      "epoch": 1.0362625122070312e-05,
      "model_forward_time": 1.7432913780212402,
      "step": 84866
    },
    {
      "epoch": 1.0362625122070312e-05,
      "step": 84866,
      "training_step_time": 7.629005432128906
    },
    {
      "epoch": 1.036357879638672e-05,
      "model_forward_time": 1.7537710666656494,
      "step": 84867
    },
    {
      "epoch": 1.036357879638672e-05,
      "step": 84867,
      "training_step_time": 7.628370046615601
    },
    {
      "epoch": 1.0364532470703125e-05,
      "model_forward_time": 1.7345809936523438,
      "step": 84868
    },
    {
      "epoch": 1.0364532470703125e-05,
      "step": 84868,
      "training_step_time": 7.637171268463135
    },
    {
      "epoch": 1.0365486145019531e-05,
      "model_forward_time": 2.3727712631225586,
      "step": 84869
    },
    {
      "epoch": 1.0365486145019531e-05,
      "step": 84869,
      "training_step_time": 7.229568719863892
    },
    {
      "epoch": 1.0366439819335937e-05,
      "grad_norm": 0.08915745466947556,
      "learning_rate": 6.129817112974218e-07,
      "loss": 0.0531,
      "step": 84870
    },
    {
      "epoch": 1.0366439819335937e-05,
      "model_forward_time": 1.7369561195373535,
      "step": 84870
    },
    {
      "dynamics_loss_avg": 0.05882114470005036,
      "epoch": 1.0366439819335937e-05,
      "step": 84870
    },
    {
      "action_loss_avg": 0.0014803414116613567,
      "epoch": 1.0366439819335937e-05,
      "step": 84870
    },
    {
      "epoch": 1.0366439819335937e-05,
      "step": 84870,
      "training_step_time": 9.592666149139404
    },
    {
      "epoch": 1.0367393493652344e-05,
      "model_forward_time": 1.759490966796875,
      "step": 84871
    },
    {
      "epoch": 1.0367393493652344e-05,
      "step": 84871,
      "training_step_time": 8.358423471450806
    },
    {
      "epoch": 1.036834716796875e-05,
      "model_forward_time": 1.760824203491211,
      "step": 84872
    },
    {
      "epoch": 1.036834716796875e-05,
      "step": 84872,
      "training_step_time": 8.762397527694702
    },
    {
      "epoch": 1.0369300842285156e-05,
      "model_forward_time": 1.7443532943725586,
      "step": 84873
    },
    {
      "epoch": 1.0369300842285156e-05,
      "step": 84873,
      "training_step_time": 8.425745010375977
    },
    {
      "epoch": 1.0370254516601563e-05,
      "model_forward_time": 1.7299435138702393,
      "step": 84874
    },
    {
      "epoch": 1.0370254516601563e-05,
      "step": 84874,
      "training_step_time": 7.689961194992065
    },
    {
      "epoch": 1.0371208190917969e-05,
      "model_forward_time": 1.7323110103607178,
      "step": 84875
    },
    {
      "epoch": 1.0371208190917969e-05,
      "step": 84875,
      "training_step_time": 7.645133972167969
    },
    {
      "epoch": 1.0372161865234374e-05,
      "model_forward_time": 1.7340829372406006,
      "step": 84876
    },
    {
      "epoch": 1.0372161865234374e-05,
      "step": 84876,
      "training_step_time": 8.02921724319458
    },
    {
      "epoch": 1.0373115539550782e-05,
      "model_forward_time": 1.7430980205535889,
      "step": 84877
    },
    {
      "epoch": 1.0373115539550782e-05,
      "step": 84877,
      "training_step_time": 7.6404101848602295
    },
    {
      "epoch": 1.0374069213867188e-05,
      "model_forward_time": 2.391946315765381,
      "step": 84878
    },
    {
      "epoch": 1.0374069213867188e-05,
      "step": 84878,
      "training_step_time": 7.109018087387085
    },
    {
      "epoch": 1.0375022888183593e-05,
      "model_forward_time": 1.7358872890472412,
      "step": 84879
    },
    {
      "epoch": 1.0375022888183593e-05,
      "step": 84879,
      "training_step_time": 9.26468276977539
    },
    {
      "epoch": 1.03759765625e-05,
      "grad_norm": 0.06576285511255264,
      "learning_rate": 6.121886942914457e-07,
      "loss": 0.0523,
      "step": 84880
    },
    {
      "epoch": 1.03759765625e-05,
      "model_forward_time": 1.7417123317718506,
      "step": 84880
    },
    {
      "dynamics_loss_avg": 0.03584666680544615,
      "epoch": 1.03759765625e-05,
      "step": 84880
    },
    {
      "action_loss_avg": 0.0007254247524542734,
      "epoch": 1.03759765625e-05,
      "step": 84880
    },
    {
      "epoch": 1.03759765625e-05,
      "step": 84880,
      "training_step_time": 9.828734636306763
    },
    {
      "epoch": 1.0376930236816406e-05,
      "model_forward_time": 1.7600486278533936,
      "step": 84881
    },
    {
      "epoch": 1.0376930236816406e-05,
      "step": 84881,
      "training_step_time": 7.630508184432983
    },
    {
      "epoch": 1.0377883911132812e-05,
      "model_forward_time": 1.742163896560669,
      "step": 84882
    },
    {
      "epoch": 1.0377883911132812e-05,
      "step": 84882,
      "training_step_time": 8.469473123550415
    },
    {
      "epoch": 1.0378837585449218e-05,
      "model_forward_time": 1.73677396774292,
      "step": 84883
    },
    {
      "epoch": 1.0378837585449218e-05,
      "step": 84883,
      "training_step_time": 8.432515859603882
    },
    {
      "epoch": 1.0379791259765625e-05,
      "model_forward_time": 1.7363512516021729,
      "step": 84884
    },
    {
      "epoch": 1.0379791259765625e-05,
      "step": 84884,
      "training_step_time": 7.685718536376953
    },
    {
      "epoch": 1.0380744934082031e-05,
      "model_forward_time": 1.7274045944213867,
      "step": 84885
    },
    {
      "epoch": 1.0380744934082031e-05,
      "step": 84885,
      "training_step_time": 7.848164796829224
    },
    {
      "epoch": 1.0381698608398437e-05,
      "model_forward_time": 1.3496949672698975,
      "step": 84886
    },
    {
      "epoch": 1.0381698608398437e-05,
      "step": 84886,
      "training_step_time": 7.630698919296265
    },
    {
      "epoch": 1.0382652282714844e-05,
      "model_forward_time": 2.3716022968292236,
      "step": 84887
    },
    {
      "epoch": 1.0382652282714844e-05,
      "step": 84887,
      "training_step_time": 7.123026609420776
    },
    {
      "epoch": 1.038360595703125e-05,
      "model_forward_time": 1.7412242889404297,
      "step": 84888
    },
    {
      "epoch": 1.038360595703125e-05,
      "step": 84888,
      "training_step_time": 7.7247021198272705
    },
    {
      "epoch": 1.0384559631347656e-05,
      "model_forward_time": 1.7602684497833252,
      "step": 84889
    },
    {
      "epoch": 1.0384559631347656e-05,
      "step": 84889,
      "training_step_time": 8.333492994308472
    },
    {
      "epoch": 1.0385513305664063e-05,
      "grad_norm": 0.10070410370826721,
      "learning_rate": 6.113961571299171e-07,
      "loss": 0.0525,
      "step": 84890
    },
    {
      "epoch": 1.0385513305664063e-05,
      "model_forward_time": 1.7578649520874023,
      "step": 84890
    },
    {
      "dynamics_loss_avg": 0.04682887215167284,
      "epoch": 1.0385513305664063e-05,
      "step": 84890
    },
    {
      "action_loss_avg": 0.0018946830910863354,
      "epoch": 1.0385513305664063e-05,
      "step": 84890
    },
    {
      "epoch": 1.0385513305664063e-05,
      "step": 84890,
      "training_step_time": 10.071136236190796
    },
    {
      "epoch": 1.0386466979980469e-05,
      "model_forward_time": 1.7431433200836182,
      "step": 84891
    },
    {
      "epoch": 1.0386466979980469e-05,
      "step": 84891,
      "training_step_time": 9.779294967651367
    },
    {
      "epoch": 1.0387420654296874e-05,
      "model_forward_time": 1.7457664012908936,
      "step": 84892
    },
    {
      "epoch": 1.0387420654296874e-05,
      "step": 84892,
      "training_step_time": 7.637811183929443
    },
    {
      "epoch": 1.0388374328613282e-05,
      "model_forward_time": 1.7420661449432373,
      "step": 84893
    },
    {
      "epoch": 1.0388374328613282e-05,
      "step": 84893,
      "training_step_time": 7.6934521198272705
    },
    {
      "epoch": 1.0389328002929688e-05,
      "model_forward_time": 1.7527611255645752,
      "step": 84894
    },
    {
      "epoch": 1.0389328002929688e-05,
      "step": 84894,
      "training_step_time": 7.442044019699097
    },
    {
      "epoch": 1.0390281677246093e-05,
      "model_forward_time": 1.7560577392578125,
      "step": 84895
    },
    {
      "epoch": 1.0390281677246093e-05,
      "step": 84895,
      "training_step_time": 7.575931072235107
    },
    {
      "epoch": 1.03912353515625e-05,
      "model_forward_time": 1.7438042163848877,
      "step": 84896
    },
    {
      "epoch": 1.03912353515625e-05,
      "step": 84896,
      "training_step_time": 7.3536927700042725
    },
    {
      "epoch": 1.0392189025878906e-05,
      "model_forward_time": 2.3639304637908936,
      "step": 84897
    },
    {
      "epoch": 1.0392189025878906e-05,
      "step": 84897,
      "training_step_time": 7.674811601638794
    },
    {
      "epoch": 1.0393142700195312e-05,
      "model_forward_time": 1.7636618614196777,
      "step": 84898
    },
    {
      "epoch": 1.0393142700195312e-05,
      "step": 84898,
      "training_step_time": 7.640873432159424
    },
    {
      "epoch": 1.0394096374511718e-05,
      "model_forward_time": 1.774010419845581,
      "step": 84899
    },
    {
      "epoch": 1.0394096374511718e-05,
      "step": 84899,
      "training_step_time": 9.03471064567566
    },
    {
      "epoch": 1.0395050048828125e-05,
      "grad_norm": 0.10338436812162399,
      "learning_rate": 6.106040998995066e-07,
      "loss": 0.0539,
      "step": 84900
    },
    {
      "epoch": 1.0395050048828125e-05,
      "model_forward_time": 1.739755392074585,
      "step": 84900
    },
    {
      "dynamics_loss_avg": 0.045140193030238154,
      "epoch": 1.0395050048828125e-05,
      "step": 84900
    },
    {
      "action_loss_avg": 0.001004573644604534,
      "epoch": 1.0395050048828125e-05,
      "step": 84900
    },
    {
      "epoch": 1.0395050048828125e-05,
      "step": 84900,
      "training_step_time": 8.471683502197266
    },
    {
      "epoch": 1.0396003723144531e-05,
      "model_forward_time": 1.7411344051361084,
      "step": 84901
    },
    {
      "epoch": 1.0396003723144531e-05,
      "step": 84901,
      "training_step_time": 7.544390678405762
    },
    {
      "epoch": 1.0396957397460937e-05,
      "model_forward_time": 1.7447254657745361,
      "step": 84902
    },
    {
      "epoch": 1.0396957397460937e-05,
      "step": 84902,
      "training_step_time": 7.721880912780762
    },
    {
      "epoch": 1.0397911071777344e-05,
      "model_forward_time": 1.7393779754638672,
      "step": 84903
    },
    {
      "epoch": 1.0397911071777344e-05,
      "step": 84903,
      "training_step_time": 7.516099452972412
    },
    {
      "epoch": 1.039886474609375e-05,
      "model_forward_time": 1.747960090637207,
      "step": 84904
    },
    {
      "epoch": 1.039886474609375e-05,
      "step": 84904,
      "training_step_time": 8.200631856918335
    },
    {
      "epoch": 1.0399818420410156e-05,
      "model_forward_time": 1.7492661476135254,
      "step": 84905
    },
    {
      "epoch": 1.0399818420410156e-05,
      "step": 84905,
      "training_step_time": 7.494927167892456
    },
    {
      "epoch": 1.0400772094726563e-05,
      "model_forward_time": 2.3672187328338623,
      "step": 84906
    },
    {
      "epoch": 1.0400772094726563e-05,
      "step": 84906,
      "training_step_time": 7.749115943908691
    },
    {
      "epoch": 1.0401725769042969e-05,
      "model_forward_time": 1.7442553043365479,
      "step": 84907
    },
    {
      "epoch": 1.0401725769042969e-05,
      "step": 84907,
      "training_step_time": 7.723891019821167
    },
    {
      "epoch": 1.0402679443359374e-05,
      "model_forward_time": 1.8259670734405518,
      "step": 84908
    },
    {
      "epoch": 1.0402679443359374e-05,
      "step": 84908,
      "training_step_time": 8.83378291130066
    },
    {
      "epoch": 1.0403633117675782e-05,
      "model_forward_time": 1.748997688293457,
      "step": 84909
    },
    {
      "epoch": 1.0403633117675782e-05,
      "step": 84909,
      "training_step_time": 7.869678974151611
    },
    {
      "epoch": 1.0404586791992188e-05,
      "grad_norm": 0.08395273983478546,
      "learning_rate": 6.098125226868328e-07,
      "loss": 0.0545,
      "step": 84910
    },
    {
      "epoch": 1.0404586791992188e-05,
      "model_forward_time": 1.7400047779083252,
      "step": 84910
    },
    {
      "dynamics_loss_avg": 0.04802220519632101,
      "epoch": 1.0404586791992188e-05,
      "step": 84910
    },
    {
      "action_loss_avg": 0.0010427936795167625,
      "epoch": 1.0404586791992188e-05,
      "step": 84910
    },
    {
      "epoch": 1.0404586791992188e-05,
      "step": 84910,
      "training_step_time": 8.58278489112854
    },
    {
      "epoch": 1.0405540466308593e-05,
      "model_forward_time": 1.7422904968261719,
      "step": 84911
    },
    {
      "epoch": 1.0405540466308593e-05,
      "step": 84911,
      "training_step_time": 8.699120998382568
    },
    {
      "epoch": 1.0406494140625e-05,
      "model_forward_time": 1.7427051067352295,
      "step": 84912
    },
    {
      "epoch": 1.0406494140625e-05,
      "step": 84912,
      "training_step_time": 7.742008209228516
    },
    {
      "epoch": 1.0407447814941406e-05,
      "model_forward_time": 1.750788688659668,
      "step": 84913
    },
    {
      "epoch": 1.0407447814941406e-05,
      "step": 84913,
      "training_step_time": 8.114609241485596
    },
    {
      "epoch": 1.0408401489257812e-05,
      "model_forward_time": 1.7476415634155273,
      "step": 84914
    },
    {
      "epoch": 1.0408401489257812e-05,
      "step": 84914,
      "training_step_time": 7.693809747695923
    },
    {
      "epoch": 1.0409355163574218e-05,
      "model_forward_time": 2.3780319690704346,
      "step": 84915
    },
    {
      "epoch": 1.0409355163574218e-05,
      "step": 84915,
      "training_step_time": 7.740004777908325
    },
    {
      "epoch": 1.0410308837890625e-05,
      "model_forward_time": 1.7464540004730225,
      "step": 84916
    },
    {
      "epoch": 1.0410308837890625e-05,
      "step": 84916,
      "training_step_time": 9.507261991500854
    },
    {
      "epoch": 1.0411262512207031e-05,
      "model_forward_time": 1.778792381286621,
      "step": 84917
    },
    {
      "epoch": 1.0411262512207031e-05,
      "step": 84917,
      "training_step_time": 8.545165538787842
    },
    {
      "epoch": 1.0412216186523437e-05,
      "model_forward_time": 1.745985507965088,
      "step": 84918
    },
    {
      "epoch": 1.0412216186523437e-05,
      "step": 84918,
      "training_step_time": 8.616858005523682
    },
    {
      "epoch": 1.0413169860839844e-05,
      "model_forward_time": 1.7470195293426514,
      "step": 84919
    },
    {
      "epoch": 1.0413169860839844e-05,
      "step": 84919,
      "training_step_time": 8.36032509803772
    },
    {
      "epoch": 1.041412353515625e-05,
      "grad_norm": 0.11084972321987152,
      "learning_rate": 6.090214255784604e-07,
      "loss": 0.0534,
      "step": 84920
    },
    {
      "epoch": 1.041412353515625e-05,
      "model_forward_time": 1.7414398193359375,
      "step": 84920
    },
    {
      "dynamics_loss_avg": 0.03611928150057793,
      "epoch": 1.041412353515625e-05,
      "step": 84920
    },
    {
      "action_loss_avg": 0.001009699514543172,
      "epoch": 1.041412353515625e-05,
      "step": 84920
    },
    {
      "epoch": 1.041412353515625e-05,
      "step": 84920,
      "training_step_time": 7.733644962310791
    },
    {
      "epoch": 1.0415077209472656e-05,
      "model_forward_time": 1.7506012916564941,
      "step": 84921
    },
    {
      "epoch": 1.0415077209472656e-05,
      "step": 84921,
      "training_step_time": 7.752030849456787
    },
    {
      "epoch": 1.0416030883789063e-05,
      "model_forward_time": 1.7479443550109863,
      "step": 84922
    },
    {
      "epoch": 1.0416030883789063e-05,
      "step": 84922,
      "training_step_time": 7.502071857452393
    },
    {
      "epoch": 1.0416984558105469e-05,
      "model_forward_time": 1.7504143714904785,
      "step": 84923
    },
    {
      "epoch": 1.0416984558105469e-05,
      "step": 84923,
      "training_step_time": 7.728597164154053
    },
    {
      "epoch": 1.0417938232421875e-05,
      "model_forward_time": 2.4249215126037598,
      "step": 84924
    },
    {
      "epoch": 1.0417938232421875e-05,
      "step": 84924,
      "training_step_time": 7.64298415184021
    },
    {
      "epoch": 1.0418891906738282e-05,
      "model_forward_time": 1.7492337226867676,
      "step": 84925
    },
    {
      "epoch": 1.0418891906738282e-05,
      "step": 84925,
      "training_step_time": 8.740804672241211
    },
    {
      "epoch": 1.0419845581054688e-05,
      "model_forward_time": 1.7611196041107178,
      "step": 84926
    },
    {
      "epoch": 1.0419845581054688e-05,
      "step": 84926,
      "training_step_time": 9.359527826309204
    },
    {
      "epoch": 1.0420799255371093e-05,
      "model_forward_time": 1.7841718196868896,
      "step": 84927
    },
    {
      "epoch": 1.0420799255371093e-05,
      "step": 84927,
      "training_step_time": 9.537856578826904
    },
    {
      "epoch": 1.04217529296875e-05,
      "model_forward_time": 1.7505183219909668,
      "step": 84928
    },
    {
      "epoch": 1.04217529296875e-05,
      "step": 84928,
      "training_step_time": 9.099400997161865
    },
    {
      "epoch": 1.0422706604003907e-05,
      "model_forward_time": 1.7504515647888184,
      "step": 84929
    },
    {
      "epoch": 1.0422706604003907e-05,
      "step": 84929,
      "training_step_time": 8.399959564208984
    },
    {
      "epoch": 1.0423660278320312e-05,
      "grad_norm": 0.07086050510406494,
      "learning_rate": 6.082308086609035e-07,
      "loss": 0.0534,
      "step": 84930
    },
    {
      "epoch": 1.0423660278320312e-05,
      "model_forward_time": 1.738081693649292,
      "step": 84930
    },
    {
      "dynamics_loss_avg": 0.051755672320723534,
      "epoch": 1.0423660278320312e-05,
      "step": 84930
    },
    {
      "action_loss_avg": 0.0008069148403592408,
      "epoch": 1.0423660278320312e-05,
      "step": 84930
    },
    {
      "epoch": 1.0423660278320312e-05,
      "step": 84930,
      "training_step_time": 7.668523550033569
    },
    {
      "epoch": 1.0424613952636718e-05,
      "model_forward_time": 1.7485945224761963,
      "step": 84931
    },
    {
      "epoch": 1.0424613952636718e-05,
      "step": 84931,
      "training_step_time": 7.586277484893799
    },
    {
      "epoch": 1.0425567626953125e-05,
      "model_forward_time": 1.7493610382080078,
      "step": 84932
    },
    {
      "epoch": 1.0425567626953125e-05,
      "step": 84932,
      "training_step_time": 7.626404047012329
    },
    {
      "epoch": 1.0426521301269531e-05,
      "model_forward_time": 2.4038002490997314,
      "step": 84933
    },
    {
      "epoch": 1.0426521301269531e-05,
      "step": 84933,
      "training_step_time": 7.612328290939331
    },
    {
      "epoch": 1.0427474975585937e-05,
      "model_forward_time": 1.7479910850524902,
      "step": 84934
    },
    {
      "epoch": 1.0427474975585937e-05,
      "step": 84934,
      "training_step_time": 7.588107109069824
    },
    {
      "epoch": 1.0428428649902344e-05,
      "model_forward_time": 1.758897304534912,
      "step": 84935
    },
    {
      "epoch": 1.0428428649902344e-05,
      "step": 84935,
      "training_step_time": 8.798811674118042
    },
    {
      "epoch": 1.042938232421875e-05,
      "model_forward_time": 1.759416103363037,
      "step": 84936
    },
    {
      "epoch": 1.042938232421875e-05,
      "step": 84936,
      "training_step_time": 7.695013999938965
    },
    {
      "epoch": 1.0430335998535156e-05,
      "model_forward_time": 1.7489328384399414,
      "step": 84937
    },
    {
      "epoch": 1.0430335998535156e-05,
      "step": 84937,
      "training_step_time": 8.546136856079102
    },
    {
      "epoch": 1.0431289672851563e-05,
      "model_forward_time": 1.7458815574645996,
      "step": 84938
    },
    {
      "epoch": 1.0431289672851563e-05,
      "step": 84938,
      "training_step_time": 8.656197786331177
    },
    {
      "epoch": 1.0432243347167969e-05,
      "model_forward_time": 1.7427740097045898,
      "step": 84939
    },
    {
      "epoch": 1.0432243347167969e-05,
      "step": 84939,
      "training_step_time": 8.561495304107666
    },
    {
      "epoch": 1.0433197021484375e-05,
      "grad_norm": 0.09746836870908737,
      "learning_rate": 6.074406720206233e-07,
      "loss": 0.0542,
      "step": 84940
    },
    {
      "epoch": 1.0433197021484375e-05,
      "model_forward_time": 1.731879472732544,
      "step": 84940
    },
    {
      "dynamics_loss_avg": 0.05069941971451044,
      "epoch": 1.0433197021484375e-05,
      "step": 84940
    },
    {
      "action_loss_avg": 0.0014040874026250093,
      "epoch": 1.0433197021484375e-05,
      "step": 84940
    },
    {
      "epoch": 1.0433197021484375e-05,
      "step": 84940,
      "training_step_time": 7.627363681793213
    },
    {
      "epoch": 1.0434150695800782e-05,
      "model_forward_time": 1.7497992515563965,
      "step": 84941
    },
    {
      "epoch": 1.0434150695800782e-05,
      "step": 84941,
      "training_step_time": 7.716970682144165
    },
    {
      "epoch": 1.0435104370117188e-05,
      "model_forward_time": 1.7390751838684082,
      "step": 84942
    },
    {
      "epoch": 1.0435104370117188e-05,
      "step": 84942,
      "training_step_time": 8.166024684906006
    },
    {
      "epoch": 1.0436058044433593e-05,
      "model_forward_time": 1.7453291416168213,
      "step": 84943
    },
    {
      "epoch": 1.0436058044433593e-05,
      "step": 84943,
      "training_step_time": 7.588175058364868
    },
    {
      "epoch": 1.043701171875e-05,
      "model_forward_time": 1.7441129684448242,
      "step": 84944
    },
    {
      "epoch": 1.043701171875e-05,
      "step": 84944,
      "training_step_time": 8.595512866973877
    },
    {
      "epoch": 1.0437965393066407e-05,
      "model_forward_time": 1.767874002456665,
      "step": 84945
    },
    {
      "epoch": 1.0437965393066407e-05,
      "step": 84945,
      "training_step_time": 9.221066236495972
    },
    {
      "epoch": 1.0438919067382812e-05,
      "model_forward_time": 1.7399637699127197,
      "step": 84946
    },
    {
      "epoch": 1.0438919067382812e-05,
      "step": 84946,
      "training_step_time": 7.673197984695435
    },
    {
      "epoch": 1.0439872741699218e-05,
      "model_forward_time": 1.7344167232513428,
      "step": 84947
    },
    {
      "epoch": 1.0439872741699218e-05,
      "step": 84947,
      "training_step_time": 8.410625219345093
    },
    {
      "epoch": 1.0440826416015625e-05,
      "model_forward_time": 1.7460095882415771,
      "step": 84948
    },
    {
      "epoch": 1.0440826416015625e-05,
      "step": 84948,
      "training_step_time": 7.747772216796875
    },
    {
      "epoch": 1.0441780090332031e-05,
      "model_forward_time": 1.7460064888000488,
      "step": 84949
    },
    {
      "epoch": 1.0441780090332031e-05,
      "step": 84949,
      "training_step_time": 7.776462554931641
    },
    {
      "epoch": 1.0442733764648437e-05,
      "grad_norm": 0.5607650279998779,
      "learning_rate": 6.066510157440258e-07,
      "loss": 0.0548,
      "step": 84950
    },
    {
      "epoch": 1.0442733764648437e-05,
      "model_forward_time": 1.7288410663604736,
      "step": 84950
    },
    {
      "dynamics_loss_avg": 0.048764432221651076,
      "epoch": 1.0442733764648437e-05,
      "step": 84950
    },
    {
      "action_loss_avg": 0.0008771067688940093,
      "epoch": 1.0442733764648437e-05,
      "step": 84950
    },
    {
      "epoch": 1.0442733764648437e-05,
      "step": 84950,
      "training_step_time": 8.676970958709717
    },
    {
      "epoch": 1.0443687438964844e-05,
      "model_forward_time": 1.7339801788330078,
      "step": 84951
    },
    {
      "epoch": 1.0443687438964844e-05,
      "step": 84951,
      "training_step_time": 7.623527526855469
    },
    {
      "epoch": 1.044464111328125e-05,
      "model_forward_time": 2.366675615310669,
      "step": 84952
    },
    {
      "epoch": 1.044464111328125e-05,
      "step": 84952,
      "training_step_time": 8.624285221099854
    },
    {
      "epoch": 1.0445594787597656e-05,
      "model_forward_time": 1.7476015090942383,
      "step": 84953
    },
    {
      "epoch": 1.0445594787597656e-05,
      "step": 84953,
      "training_step_time": 8.569103956222534
    },
    {
      "epoch": 1.0446548461914063e-05,
      "model_forward_time": 1.8078649044036865,
      "step": 84954
    },
    {
      "epoch": 1.0446548461914063e-05,
      "step": 84954,
      "training_step_time": 7.714936017990112
    },
    {
      "epoch": 1.0447502136230469e-05,
      "model_forward_time": 1.7447669506072998,
      "step": 84955
    },
    {
      "epoch": 1.0447502136230469e-05,
      "step": 84955,
      "training_step_time": 7.753422975540161
    },
    {
      "epoch": 1.0448455810546875e-05,
      "model_forward_time": 1.7264249324798584,
      "step": 84956
    },
    {
      "epoch": 1.0448455810546875e-05,
      "step": 84956,
      "training_step_time": 7.63407564163208
    },
    {
      "epoch": 1.0449409484863282e-05,
      "model_forward_time": 1.734079122543335,
      "step": 84957
    },
    {
      "epoch": 1.0449409484863282e-05,
      "step": 84957,
      "training_step_time": 7.693036079406738
    },
    {
      "epoch": 1.0450363159179688e-05,
      "model_forward_time": 1.7319910526275635,
      "step": 84958
    },
    {
      "epoch": 1.0450363159179688e-05,
      "step": 84958,
      "training_step_time": 7.637460947036743
    },
    {
      "epoch": 1.0451316833496093e-05,
      "model_forward_time": 1.730882167816162,
      "step": 84959
    },
    {
      "epoch": 1.0451316833496093e-05,
      "step": 84959,
      "training_step_time": 7.625157117843628
    },
    {
      "epoch": 1.04522705078125e-05,
      "grad_norm": 0.09678174555301666,
      "learning_rate": 6.058618399174676e-07,
      "loss": 0.0536,
      "step": 84960
    },
    {
      "epoch": 1.04522705078125e-05,
      "model_forward_time": 1.7217347621917725,
      "step": 84960
    },
    {
      "dynamics_loss_avg": 0.04935088139027357,
      "epoch": 1.04522705078125e-05,
      "step": 84960
    },
    {
      "action_loss_avg": 0.0009661801115726121,
      "epoch": 1.04522705078125e-05,
      "step": 84960
    },
    {
      "epoch": 1.04522705078125e-05,
      "step": 84960,
      "training_step_time": 7.695285797119141
    },
    {
      "epoch": 1.0453224182128907e-05,
      "model_forward_time": 2.3605480194091797,
      "step": 84961
    },
    {
      "epoch": 1.0453224182128907e-05,
      "step": 84961,
      "training_step_time": 7.646354913711548
    },
    {
      "epoch": 1.0454177856445312e-05,
      "model_forward_time": 1.7356245517730713,
      "step": 84962
    },
    {
      "epoch": 1.0454177856445312e-05,
      "step": 84962,
      "training_step_time": 8.316415548324585
    },
    {
      "epoch": 1.0455131530761718e-05,
      "model_forward_time": 1.789764642715454,
      "step": 84963
    },
    {
      "epoch": 1.0455131530761718e-05,
      "step": 84963,
      "training_step_time": 9.675285577774048
    },
    {
      "epoch": 1.0456085205078125e-05,
      "model_forward_time": 1.7380568981170654,
      "step": 84964
    },
    {
      "epoch": 1.0456085205078125e-05,
      "step": 84964,
      "training_step_time": 7.695627927780151
    },
    {
      "epoch": 1.0457038879394531e-05,
      "model_forward_time": 1.7363204956054688,
      "step": 84965
    },
    {
      "epoch": 1.0457038879394531e-05,
      "step": 84965,
      "training_step_time": 7.745506048202515
    },
    {
      "epoch": 1.0457992553710937e-05,
      "model_forward_time": 1.7318148612976074,
      "step": 84966
    },
    {
      "epoch": 1.0457992553710937e-05,
      "step": 84966,
      "training_step_time": 8.296220064163208
    },
    {
      "epoch": 1.0458946228027344e-05,
      "model_forward_time": 1.7314176559448242,
      "step": 84967
    },
    {
      "epoch": 1.0458946228027344e-05,
      "step": 84967,
      "training_step_time": 7.6282665729522705
    },
    {
      "epoch": 1.045989990234375e-05,
      "model_forward_time": 1.7521591186523438,
      "step": 84968
    },
    {
      "epoch": 1.045989990234375e-05,
      "step": 84968,
      "training_step_time": 7.721238851547241
    },
    {
      "epoch": 1.0460853576660156e-05,
      "model_forward_time": 1.7305378913879395,
      "step": 84969
    },
    {
      "epoch": 1.0460853576660156e-05,
      "step": 84969,
      "training_step_time": 7.7482500076293945
    },
    {
      "epoch": 1.0461807250976563e-05,
      "grad_norm": 0.11186595261096954,
      "learning_rate": 6.050731446272529e-07,
      "loss": 0.0539,
      "step": 84970
    },
    {
      "epoch": 1.0461807250976563e-05,
      "model_forward_time": 2.381434202194214,
      "step": 84970
    },
    {
      "dynamics_loss_avg": 0.05005481839179993,
      "epoch": 1.0461807250976563e-05,
      "step": 84970
    },
    {
      "action_loss_avg": 0.0013222915993537753,
      "epoch": 1.0461807250976563e-05,
      "step": 84970
    },
    {
      "epoch": 1.0461807250976563e-05,
      "step": 84970,
      "training_step_time": 7.607316732406616
    },
    {
      "epoch": 1.0462760925292969e-05,
      "model_forward_time": 1.7283902168273926,
      "step": 84971
    },
    {
      "epoch": 1.0462760925292969e-05,
      "step": 84971,
      "training_step_time": 9.435425996780396
    },
    {
      "epoch": 1.0463714599609375e-05,
      "model_forward_time": 1.739445686340332,
      "step": 84972
    },
    {
      "epoch": 1.0463714599609375e-05,
      "step": 84972,
      "training_step_time": 7.594344139099121
    },
    {
      "epoch": 1.0464668273925782e-05,
      "model_forward_time": 1.7557806968688965,
      "step": 84973
    },
    {
      "epoch": 1.0464668273925782e-05,
      "step": 84973,
      "training_step_time": 8.828166007995605
    },
    {
      "epoch": 1.0465621948242188e-05,
      "model_forward_time": 1.7299559116363525,
      "step": 84974
    },
    {
      "epoch": 1.0465621948242188e-05,
      "step": 84974,
      "training_step_time": 8.527401447296143
    },
    {
      "epoch": 1.0466575622558593e-05,
      "model_forward_time": 1.7281708717346191,
      "step": 84975
    },
    {
      "epoch": 1.0466575622558593e-05,
      "step": 84975,
      "training_step_time": 8.616653203964233
    },
    {
      "epoch": 1.0467529296875e-05,
      "model_forward_time": 0.7006833553314209,
      "step": 84976
    },
    {
      "epoch": 1.0467529296875e-05,
      "step": 84976,
      "training_step_time": 7.663906574249268
    },
    {
      "epoch": 1.0468482971191407e-05,
      "model_forward_time": 1.7347605228424072,
      "step": 84977
    },
    {
      "epoch": 1.0468482971191407e-05,
      "step": 84977,
      "training_step_time": 7.731164455413818
    },
    {
      "epoch": 1.0469436645507812e-05,
      "model_forward_time": 1.7291481494903564,
      "step": 84978
    },
    {
      "epoch": 1.0469436645507812e-05,
      "step": 84978,
      "training_step_time": 7.715755462646484
    },
    {
      "epoch": 1.0470390319824218e-05,
      "model_forward_time": 1.7488768100738525,
      "step": 84979
    },
    {
      "epoch": 1.0470390319824218e-05,
      "step": 84979,
      "training_step_time": 7.6740641593933105
    },
    {
      "epoch": 1.0471343994140625e-05,
      "grad_norm": 0.10359297692775726,
      "learning_rate": 6.04284929959632e-07,
      "loss": 0.0536,
      "step": 84980
    },
    {
      "epoch": 1.0471343994140625e-05,
      "model_forward_time": 2.3738160133361816,
      "step": 84980
    },
    {
      "dynamics_loss_avg": 0.04918033257126808,
      "epoch": 1.0471343994140625e-05,
      "step": 84980
    },
    {
      "action_loss_avg": 0.0013018012250540778,
      "epoch": 1.0471343994140625e-05,
      "step": 84980
    },
    {
      "epoch": 1.0471343994140625e-05,
      "step": 84980,
      "training_step_time": 7.586049795150757
    },
    {
      "epoch": 1.0472297668457031e-05,
      "model_forward_time": 1.7447571754455566,
      "step": 84981
    },
    {
      "epoch": 1.0472297668457031e-05,
      "step": 84981,
      "training_step_time": 9.059694290161133
    },
    {
      "epoch": 1.0473251342773437e-05,
      "model_forward_time": 1.7445313930511475,
      "step": 84982
    },
    {
      "epoch": 1.0473251342773437e-05,
      "step": 84982,
      "training_step_time": 8.771959781646729
    },
    {
      "epoch": 1.0474205017089844e-05,
      "model_forward_time": 1.7469522953033447,
      "step": 84983
    },
    {
      "epoch": 1.0474205017089844e-05,
      "step": 84983,
      "training_step_time": 9.379111289978027
    },
    {
      "epoch": 1.047515869140625e-05,
      "model_forward_time": 1.7404296398162842,
      "step": 84984
    },
    {
      "epoch": 1.047515869140625e-05,
      "step": 84984,
      "training_step_time": 7.625182390213013
    },
    {
      "epoch": 1.0476112365722656e-05,
      "model_forward_time": 1.7421882152557373,
      "step": 84985
    },
    {
      "epoch": 1.0476112365722656e-05,
      "step": 84985,
      "training_step_time": 7.610781669616699
    },
    {
      "epoch": 1.0477066040039063e-05,
      "model_forward_time": 1.7492988109588623,
      "step": 84986
    },
    {
      "epoch": 1.0477066040039063e-05,
      "step": 84986,
      "training_step_time": 7.658631801605225
    },
    {
      "epoch": 1.0478019714355469e-05,
      "model_forward_time": 1.7380669116973877,
      "step": 84987
    },
    {
      "epoch": 1.0478019714355469e-05,
      "step": 84987,
      "training_step_time": 8.534311771392822
    },
    {
      "epoch": 1.0478973388671875e-05,
      "model_forward_time": 1.7486002445220947,
      "step": 84988
    },
    {
      "epoch": 1.0478973388671875e-05,
      "step": 84988,
      "training_step_time": 7.669534683227539
    },
    {
      "epoch": 1.0479927062988282e-05,
      "model_forward_time": 2.3630852699279785,
      "step": 84989
    },
    {
      "epoch": 1.0479927062988282e-05,
      "step": 84989,
      "training_step_time": 8.220398426055908
    },
    {
      "epoch": 1.0480880737304688e-05,
      "grad_norm": 0.1859450787305832,
      "learning_rate": 6.034971960008012e-07,
      "loss": 0.0545,
      "step": 84990
    },
    {
      "epoch": 1.0480880737304688e-05,
      "model_forward_time": 1.76646089553833,
      "step": 84990
    },
    {
      "dynamics_loss_avg": 0.048388564586639406,
      "epoch": 1.0480880737304688e-05,
      "step": 84990
    },
    {
      "action_loss_avg": 0.0009490962111158296,
      "epoch": 1.0480880737304688e-05,
      "step": 84990
    },
    {
      "epoch": 1.0480880737304688e-05,
      "step": 84990,
      "training_step_time": 8.243581533432007
    },
    {
      "epoch": 1.0481834411621093e-05,
      "model_forward_time": 1.8370399475097656,
      "step": 84991
    },
    {
      "epoch": 1.0481834411621093e-05,
      "step": 84991,
      "training_step_time": 8.340569257736206
    },
    {
      "epoch": 1.04827880859375e-05,
      "model_forward_time": 1.7556238174438477,
      "step": 84992
    },
    {
      "epoch": 1.04827880859375e-05,
      "step": 84992,
      "training_step_time": 9.52056074142456
    },
    {
      "epoch": 1.0483741760253907e-05,
      "model_forward_time": 1.759559154510498,
      "step": 84993
    },
    {
      "epoch": 1.0483741760253907e-05,
      "step": 84993,
      "training_step_time": 8.388817310333252
    },
    {
      "epoch": 1.0484695434570312e-05,
      "model_forward_time": 1.7490785121917725,
      "step": 84994
    },
    {
      "epoch": 1.0484695434570312e-05,
      "step": 84994,
      "training_step_time": 7.61495304107666
    },
    {
      "epoch": 1.0485649108886718e-05,
      "model_forward_time": 1.7655482292175293,
      "step": 84995
    },
    {
      "epoch": 1.0485649108886718e-05,
      "step": 84995,
      "training_step_time": 7.752107620239258
    },
    {
      "epoch": 1.0486602783203125e-05,
      "model_forward_time": 1.7585067749023438,
      "step": 84996
    },
    {
      "epoch": 1.0486602783203125e-05,
      "step": 84996,
      "training_step_time": 7.6495361328125
    },
    {
      "epoch": 1.0487556457519531e-05,
      "model_forward_time": 1.7741284370422363,
      "step": 84997
    },
    {
      "epoch": 1.0487556457519531e-05,
      "step": 84997,
      "training_step_time": 7.718079090118408
    },
    {
      "epoch": 1.0488510131835937e-05,
      "model_forward_time": 2.4785327911376953,
      "step": 84998
    },
    {
      "epoch": 1.0488510131835937e-05,
      "step": 84998,
      "training_step_time": 8.5974702835083
    },
    {
      "epoch": 1.0489463806152344e-05,
      "model_forward_time": 1.7800443172454834,
      "step": 84999
    },
    {
      "epoch": 1.0489463806152344e-05,
      "step": 84999,
      "training_step_time": 8.584607124328613
    },
    {
      "epoch": 1.049041748046875e-05,
      "grad_norm": 0.0975237712264061,
      "learning_rate": 6.027099428369082e-07,
      "loss": 0.0529,
      "step": 85000
    },
    {
      "epoch": 1.049041748046875e-05,
      "model_forward_time": 1.7653169631958008,
      "step": 85000
    },
    {
      "dynamics_loss_avg": 0.05830743201076984,
      "epoch": 1.049041748046875e-05,
      "step": 85000
    },
    {
      "action_loss_avg": 0.0013901167134463322,
      "epoch": 1.049041748046875e-05,
      "step": 85000
    },
    {
      "epoch": 1.049041748046875e-05,
      "step": 85000,
      "training_step_time": 9.075156450271606
    },
    {
      "epoch": 1.0491371154785156e-05,
      "model_forward_time": 1.7707273960113525,
      "step": 85001
    },
    {
      "epoch": 1.0491371154785156e-05,
      "step": 85001,
      "training_step_time": 9.272443294525146
    },
    {
      "epoch": 1.0492324829101563e-05,
      "model_forward_time": 1.7737607955932617,
      "step": 85002
    },
    {
      "epoch": 1.0492324829101563e-05,
      "step": 85002,
      "training_step_time": 9.31900668144226
    },
    {
      "epoch": 1.0493278503417969e-05,
      "model_forward_time": 1.777226209640503,
      "step": 85003
    },
    {
      "epoch": 1.0493278503417969e-05,
      "step": 85003,
      "training_step_time": 8.48599624633789
    },
    {
      "epoch": 1.0494232177734375e-05,
      "model_forward_time": 1.7725436687469482,
      "step": 85004
    },
    {
      "epoch": 1.0494232177734375e-05,
      "step": 85004,
      "training_step_time": 8.252559661865234
    },
    {
      "epoch": 1.0495185852050782e-05,
      "model_forward_time": 1.778564691543579,
      "step": 85005
    },
    {
      "epoch": 1.0495185852050782e-05,
      "step": 85005,
      "training_step_time": 7.750789165496826
    },
    {
      "epoch": 1.0496139526367188e-05,
      "model_forward_time": 1.7780284881591797,
      "step": 85006
    },
    {
      "epoch": 1.0496139526367188e-05,
      "step": 85006,
      "training_step_time": 7.719799041748047
    },
    {
      "epoch": 1.0497093200683593e-05,
      "model_forward_time": 3.041081666946411,
      "step": 85007
    },
    {
      "epoch": 1.0497093200683593e-05,
      "step": 85007,
      "training_step_time": 8.732365369796753
    },
    {
      "epoch": 1.0498046875e-05,
      "model_forward_time": 1.7658793926239014,
      "step": 85008
    },
    {
      "epoch": 1.0498046875e-05,
      "step": 85008,
      "training_step_time": 10.58338975906372
    },
    {
      "epoch": 1.0499000549316407e-05,
      "model_forward_time": 1.8404905796051025,
      "step": 85009
    },
    {
      "epoch": 1.0499000549316407e-05,
      "step": 85009,
      "training_step_time": 7.664705038070679
    },
    {
      "epoch": 1.0499954223632812e-05,
      "grad_norm": 0.0720079243183136,
      "learning_rate": 6.019231705540435e-07,
      "loss": 0.0531,
      "step": 85010
    },
    {
      "epoch": 1.0499954223632812e-05,
      "model_forward_time": 1.756438970565796,
      "step": 85010
    },
    {
      "dynamics_loss_avg": 0.04456541948020458,
      "epoch": 1.0499954223632812e-05,
      "step": 85010
    },
    {
      "action_loss_avg": 0.0008449772227322683,
      "epoch": 1.0499954223632812e-05,
      "step": 85010
    },
    {
      "epoch": 1.0499954223632812e-05,
      "step": 85010,
      "training_step_time": 9.33710789680481
    },
    {
      "epoch": 1.0500907897949218e-05,
      "model_forward_time": 1.7541522979736328,
      "step": 85011
    },
    {
      "epoch": 1.0500907897949218e-05,
      "step": 85011,
      "training_step_time": 9.441597938537598
    },
    {
      "epoch": 1.0501861572265625e-05,
      "model_forward_time": 1.769742727279663,
      "step": 85012
    },
    {
      "epoch": 1.0501861572265625e-05,
      "step": 85012,
      "training_step_time": 8.801034212112427
    },
    {
      "epoch": 1.0502815246582031e-05,
      "model_forward_time": 1.7597379684448242,
      "step": 85013
    },
    {
      "epoch": 1.0502815246582031e-05,
      "step": 85013,
      "training_step_time": 7.680126667022705
    },
    {
      "epoch": 1.0503768920898437e-05,
      "model_forward_time": 1.7658483982086182,
      "step": 85014
    },
    {
      "epoch": 1.0503768920898437e-05,
      "step": 85014,
      "training_step_time": 8.3478262424469
    },
    {
      "epoch": 1.0504722595214844e-05,
      "model_forward_time": 1.7642741203308105,
      "step": 85015
    },
    {
      "epoch": 1.0504722595214844e-05,
      "step": 85015,
      "training_step_time": 8.490344524383545
    },
    {
      "epoch": 1.050567626953125e-05,
      "model_forward_time": 2.3622798919677734,
      "step": 85016
    },
    {
      "epoch": 1.050567626953125e-05,
      "step": 85016,
      "training_step_time": 7.533350229263306
    },
    {
      "epoch": 1.0506629943847656e-05,
      "model_forward_time": 1.7495357990264893,
      "step": 85017
    },
    {
      "epoch": 1.0506629943847656e-05,
      "step": 85017,
      "training_step_time": 8.69910478591919
    },
    {
      "epoch": 1.0507583618164063e-05,
      "model_forward_time": 1.8073678016662598,
      "step": 85018
    },
    {
      "epoch": 1.0507583618164063e-05,
      "step": 85018,
      "training_step_time": 7.655100584030151
    },
    {
      "epoch": 1.0508537292480469e-05,
      "model_forward_time": 1.750852346420288,
      "step": 85019
    },
    {
      "epoch": 1.0508537292480469e-05,
      "step": 85019,
      "training_step_time": 9.604820013046265
    },
    {
      "epoch": 1.0509490966796875e-05,
      "grad_norm": 0.08579214662313461,
      "learning_rate": 6.011368792382488e-07,
      "loss": 0.0543,
      "step": 85020
    },
    {
      "epoch": 1.0509490966796875e-05,
      "model_forward_time": 1.7477822303771973,
      "step": 85020
    },
    {
      "dynamics_loss_avg": 0.0658532690256834,
      "epoch": 1.0509490966796875e-05,
      "step": 85020
    },
    {
      "action_loss_avg": 0.0021512462466489525,
      "epoch": 1.0509490966796875e-05,
      "step": 85020
    },
    {
      "epoch": 1.0509490966796875e-05,
      "step": 85020,
      "training_step_time": 7.853145122528076
    },
    {
      "epoch": 1.0510444641113282e-05,
      "model_forward_time": 1.75343656539917,
      "step": 85021
    },
    {
      "epoch": 1.0510444641113282e-05,
      "step": 85021,
      "training_step_time": 8.471758604049683
    },
    {
      "epoch": 1.0511398315429688e-05,
      "model_forward_time": 1.7525360584259033,
      "step": 85022
    },
    {
      "epoch": 1.0511398315429688e-05,
      "step": 85022,
      "training_step_time": 8.45896315574646
    },
    {
      "epoch": 1.0512351989746093e-05,
      "model_forward_time": 1.753286361694336,
      "step": 85023
    },
    {
      "epoch": 1.0512351989746093e-05,
      "step": 85023,
      "training_step_time": 7.681869268417358
    },
    {
      "epoch": 1.05133056640625e-05,
      "model_forward_time": 1.757167100906372,
      "step": 85024
    },
    {
      "epoch": 1.05133056640625e-05,
      "step": 85024,
      "training_step_time": 8.477214813232422
    },
    {
      "epoch": 1.0514259338378907e-05,
      "model_forward_time": 2.379556894302368,
      "step": 85025
    },
    {
      "epoch": 1.0514259338378907e-05,
      "step": 85025,
      "training_step_time": 7.560342073440552
    },
    {
      "epoch": 1.0515213012695312e-05,
      "model_forward_time": 1.755753517150879,
      "step": 85026
    },
    {
      "epoch": 1.0515213012695312e-05,
      "step": 85026,
      "training_step_time": 8.286189317703247
    },
    {
      "epoch": 1.0516166687011718e-05,
      "model_forward_time": 1.7731854915618896,
      "step": 85027
    },
    {
      "epoch": 1.0516166687011718e-05,
      "step": 85027,
      "training_step_time": 7.563355445861816
    },
    {
      "epoch": 1.0517120361328125e-05,
      "model_forward_time": 1.7662131786346436,
      "step": 85028
    },
    {
      "epoch": 1.0517120361328125e-05,
      "step": 85028,
      "training_step_time": 7.663393974304199
    },
    {
      "epoch": 1.0518074035644531e-05,
      "model_forward_time": 1.7466297149658203,
      "step": 85029
    },
    {
      "epoch": 1.0518074035644531e-05,
      "step": 85029,
      "training_step_time": 7.8945159912109375
    },
    {
      "epoch": 1.0519027709960937e-05,
      "grad_norm": 0.11274728178977966,
      "learning_rate": 6.00351068975511e-07,
      "loss": 0.0524,
      "step": 85030
    },
    {
      "epoch": 1.0519027709960937e-05,
      "model_forward_time": 1.7430477142333984,
      "step": 85030
    },
    {
      "dynamics_loss_avg": 0.048536129482090476,
      "epoch": 1.0519027709960937e-05,
      "step": 85030
    },
    {
      "action_loss_avg": 0.0012506713450420649,
      "epoch": 1.0519027709960937e-05,
      "step": 85030
    },
    {
      "epoch": 1.0519027709960937e-05,
      "step": 85030,
      "training_step_time": 7.600682735443115
    },
    {
      "epoch": 1.0519981384277344e-05,
      "model_forward_time": 1.7532563209533691,
      "step": 85031
    },
    {
      "epoch": 1.0519981384277344e-05,
      "step": 85031,
      "training_step_time": 8.526463747024536
    },
    {
      "epoch": 1.052093505859375e-05,
      "model_forward_time": 1.7519783973693848,
      "step": 85032
    },
    {
      "epoch": 1.052093505859375e-05,
      "step": 85032,
      "training_step_time": 7.640959978103638
    },
    {
      "epoch": 1.0521888732910156e-05,
      "model_forward_time": 1.7645986080169678,
      "step": 85033
    },
    {
      "epoch": 1.0521888732910156e-05,
      "step": 85033,
      "training_step_time": 7.582141160964966
    },
    {
      "epoch": 1.0522842407226563e-05,
      "model_forward_time": 2.4219017028808594,
      "step": 85034
    },
    {
      "epoch": 1.0522842407226563e-05,
      "step": 85034,
      "training_step_time": 7.526276350021362
    },
    {
      "epoch": 1.0523796081542969e-05,
      "model_forward_time": 1.7579455375671387,
      "step": 85035
    },
    {
      "epoch": 1.0523796081542969e-05,
      "step": 85035,
      "training_step_time": 8.819129228591919
    },
    {
      "epoch": 1.0524749755859375e-05,
      "model_forward_time": 1.7552216053009033,
      "step": 85036
    },
    {
      "epoch": 1.0524749755859375e-05,
      "step": 85036,
      "training_step_time": 7.674540996551514
    },
    {
      "epoch": 1.0525703430175782e-05,
      "model_forward_time": 1.7557339668273926,
      "step": 85037
    },
    {
      "epoch": 1.0525703430175782e-05,
      "step": 85037,
      "training_step_time": 8.981995105743408
    },
    {
      "epoch": 1.0526657104492188e-05,
      "model_forward_time": 1.7558212280273438,
      "step": 85038
    },
    {
      "epoch": 1.0526657104492188e-05,
      "step": 85038,
      "training_step_time": 9.645405292510986
    },
    {
      "epoch": 1.0527610778808593e-05,
      "model_forward_time": 1.7576322555541992,
      "step": 85039
    },
    {
      "epoch": 1.0527610778808593e-05,
      "step": 85039,
      "training_step_time": 8.452749490737915
    },
    {
      "epoch": 1.0528564453125e-05,
      "grad_norm": 0.08841616660356522,
      "learning_rate": 5.995657398517668e-07,
      "loss": 0.052,
      "step": 85040
    },
    {
      "epoch": 1.0528564453125e-05,
      "model_forward_time": 1.7548036575317383,
      "step": 85040
    },
    {
      "dynamics_loss_avg": 0.05655661188066006,
      "epoch": 1.0528564453125e-05,
      "step": 85040
    },
    {
      "action_loss_avg": 0.0016646338102873416,
      "epoch": 1.0528564453125e-05,
      "step": 85040
    },
    {
      "epoch": 1.0528564453125e-05,
      "step": 85040,
      "training_step_time": 8.42815089225769
    },
    {
      "epoch": 1.0529518127441407e-05,
      "model_forward_time": 1.7562370300292969,
      "step": 85041
    },
    {
      "epoch": 1.0529518127441407e-05,
      "step": 85041,
      "training_step_time": 8.715753316879272
    },
    {
      "epoch": 1.0530471801757812e-05,
      "model_forward_time": 1.7534289360046387,
      "step": 85042
    },
    {
      "epoch": 1.0530471801757812e-05,
      "step": 85042,
      "training_step_time": 7.795078754425049
    },
    {
      "epoch": 1.0531425476074218e-05,
      "model_forward_time": 2.3677592277526855,
      "step": 85043
    },
    {
      "epoch": 1.0531425476074218e-05,
      "step": 85043,
      "training_step_time": 7.640766143798828
    },
    {
      "epoch": 1.0532379150390625e-05,
      "model_forward_time": 1.755300760269165,
      "step": 85044
    },
    {
      "epoch": 1.0532379150390625e-05,
      "step": 85044,
      "training_step_time": 8.58789610862732
    },
    {
      "epoch": 1.0533332824707031e-05,
      "model_forward_time": 1.7575304508209229,
      "step": 85045
    },
    {
      "epoch": 1.0533332824707031e-05,
      "step": 85045,
      "training_step_time": 7.634385824203491
    },
    {
      "epoch": 1.0534286499023437e-05,
      "model_forward_time": 1.8455867767333984,
      "step": 85046
    },
    {
      "epoch": 1.0534286499023437e-05,
      "step": 85046,
      "training_step_time": 7.603451728820801
    },
    {
      "epoch": 1.0535240173339844e-05,
      "model_forward_time": 1.7502317428588867,
      "step": 85047
    },
    {
      "epoch": 1.0535240173339844e-05,
      "step": 85047,
      "training_step_time": 8.701923847198486
    },
    {
      "epoch": 1.053619384765625e-05,
      "model_forward_time": 1.754152536392212,
      "step": 85048
    },
    {
      "epoch": 1.053619384765625e-05,
      "step": 85048,
      "training_step_time": 7.786275625228882
    },
    {
      "epoch": 1.0537147521972656e-05,
      "model_forward_time": 1.7529213428497314,
      "step": 85049
    },
    {
      "epoch": 1.0537147521972656e-05,
      "step": 85049,
      "training_step_time": 8.28418493270874
    },
    {
      "epoch": 1.0538101196289063e-05,
      "grad_norm": 0.10903944075107574,
      "learning_rate": 5.987808919528959e-07,
      "loss": 0.0532,
      "step": 85050
    },
    {
      "epoch": 1.0538101196289063e-05,
      "model_forward_time": 1.757385015487671,
      "step": 85050
    },
    {
      "dynamics_loss_avg": 0.05939919147640467,
      "epoch": 1.0538101196289063e-05,
      "step": 85050
    },
    {
      "action_loss_avg": 0.0014084787049796431,
      "epoch": 1.0538101196289063e-05,
      "step": 85050
    },
    {
      "epoch": 1.0538101196289063e-05,
      "step": 85050,
      "training_step_time": 7.6392741203308105
    },
    {
      "epoch": 1.0539054870605469e-05,
      "model_forward_time": 1.7529470920562744,
      "step": 85051
    },
    {
      "epoch": 1.0539054870605469e-05,
      "step": 85051,
      "training_step_time": 7.734248876571655
    },
    {
      "epoch": 1.0540008544921875e-05,
      "model_forward_time": 1.7531228065490723,
      "step": 85052
    },
    {
      "epoch": 1.0540008544921875e-05,
      "step": 85052,
      "training_step_time": 8.990575790405273
    },
    {
      "epoch": 1.0540962219238282e-05,
      "model_forward_time": 1.7460711002349854,
      "step": 85053
    },
    {
      "epoch": 1.0540962219238282e-05,
      "step": 85053,
      "training_step_time": 7.60529637336731
    },
    {
      "epoch": 1.0541915893554688e-05,
      "model_forward_time": 1.7460801601409912,
      "step": 85054
    },
    {
      "epoch": 1.0541915893554688e-05,
      "step": 85054,
      "training_step_time": 7.625226259231567
    },
    {
      "epoch": 1.0542869567871093e-05,
      "model_forward_time": 1.7685952186584473,
      "step": 85055
    },
    {
      "epoch": 1.0542869567871093e-05,
      "step": 85055,
      "training_step_time": 9.524421691894531
    },
    {
      "epoch": 1.05438232421875e-05,
      "model_forward_time": 1.7855348587036133,
      "step": 85056
    },
    {
      "epoch": 1.05438232421875e-05,
      "step": 85056,
      "training_step_time": 7.663100719451904
    },
    {
      "epoch": 1.0544776916503907e-05,
      "model_forward_time": 1.7675817012786865,
      "step": 85057
    },
    {
      "epoch": 1.0544776916503907e-05,
      "step": 85057,
      "training_step_time": 7.844041585922241
    },
    {
      "epoch": 1.0545730590820312e-05,
      "model_forward_time": 1.760849952697754,
      "step": 85058
    },
    {
      "epoch": 1.0545730590820312e-05,
      "step": 85058,
      "training_step_time": 7.641363143920898
    },
    {
      "epoch": 1.0546684265136718e-05,
      "model_forward_time": 1.7720789909362793,
      "step": 85059
    },
    {
      "epoch": 1.0546684265136718e-05,
      "step": 85059,
      "training_step_time": 8.500062227249146
    },
    {
      "epoch": 1.0547637939453125e-05,
      "grad_norm": 0.10233255475759506,
      "learning_rate": 5.979965253647307e-07,
      "loss": 0.0532,
      "step": 85060
    },
    {
      "epoch": 1.0547637939453125e-05,
      "model_forward_time": 1.7985138893127441,
      "step": 85060
    },
    {
      "dynamics_loss_avg": 0.06860579811036586,
      "epoch": 1.0547637939453125e-05,
      "step": 85060
    },
    {
      "action_loss_avg": 0.0013979106064653025,
      "epoch": 1.0547637939453125e-05,
      "step": 85060
    },
    {
      "epoch": 1.0547637939453125e-05,
      "step": 85060,
      "training_step_time": 9.398251056671143
    },
    {
      "epoch": 1.0548591613769531e-05,
      "model_forward_time": 1.7559585571289062,
      "step": 85061
    },
    {
      "epoch": 1.0548591613769531e-05,
      "step": 85061,
      "training_step_time": 9.748348712921143
    },
    {
      "epoch": 1.0549545288085937e-05,
      "model_forward_time": 2.37412691116333,
      "step": 85062
    },
    {
      "epoch": 1.0549545288085937e-05,
      "step": 85062,
      "training_step_time": 8.46164608001709
    },
    {
      "epoch": 1.0550498962402344e-05,
      "model_forward_time": 1.7754788398742676,
      "step": 85063
    },
    {
      "epoch": 1.0550498962402344e-05,
      "step": 85063,
      "training_step_time": 8.329815864562988
    },
    {
      "epoch": 1.055145263671875e-05,
      "model_forward_time": 1.7663424015045166,
      "step": 85064
    },
    {
      "epoch": 1.055145263671875e-05,
      "step": 85064,
      "training_step_time": 7.72124981880188
    },
    {
      "epoch": 1.0552406311035156e-05,
      "model_forward_time": 1.7592744827270508,
      "step": 85065
    },
    {
      "epoch": 1.0552406311035156e-05,
      "step": 85065,
      "training_step_time": 7.7623207569122314
    },
    {
      "epoch": 1.0553359985351563e-05,
      "model_forward_time": 1.7440288066864014,
      "step": 85066
    },
    {
      "epoch": 1.0553359985351563e-05,
      "step": 85066,
      "training_step_time": 7.808829069137573
    },
    {
      "epoch": 1.0554313659667969e-05,
      "model_forward_time": 1.7611451148986816,
      "step": 85067
    },
    {
      "epoch": 1.0554313659667969e-05,
      "step": 85067,
      "training_step_time": 7.674431800842285
    },
    {
      "epoch": 1.0555267333984375e-05,
      "model_forward_time": 1.7716946601867676,
      "step": 85068
    },
    {
      "epoch": 1.0555267333984375e-05,
      "step": 85068,
      "training_step_time": 7.628742694854736
    },
    {
      "epoch": 1.0556221008300782e-05,
      "model_forward_time": 1.7814888954162598,
      "step": 85069
    },
    {
      "epoch": 1.0556221008300782e-05,
      "step": 85069,
      "training_step_time": 7.809022665023804
    },
    {
      "epoch": 1.0557174682617188e-05,
      "grad_norm": 0.0907430425286293,
      "learning_rate": 5.972126401730466e-07,
      "loss": 0.0524,
      "step": 85070
    },
    {
      "epoch": 1.0557174682617188e-05,
      "model_forward_time": 1.7519450187683105,
      "step": 85070
    },
    {
      "dynamics_loss_avg": 0.06768925078213214,
      "epoch": 1.0557174682617188e-05,
      "step": 85070
    },
    {
      "action_loss_avg": 0.001106124761281535,
      "epoch": 1.0557174682617188e-05,
      "step": 85070
    },
    {
      "epoch": 1.0557174682617188e-05,
      "step": 85070,
      "training_step_time": 7.59737753868103
    },
    {
      "epoch": 1.0558128356933593e-05,
      "model_forward_time": 2.383586883544922,
      "step": 85071
    },
    {
      "epoch": 1.0558128356933593e-05,
      "step": 85071,
      "training_step_time": 7.662167310714722
    },
    {
      "epoch": 1.055908203125e-05,
      "model_forward_time": 1.7747960090637207,
      "step": 85072
    },
    {
      "epoch": 1.055908203125e-05,
      "step": 85072,
      "training_step_time": 9.325479984283447
    },
    {
      "epoch": 1.0560035705566407e-05,
      "model_forward_time": 1.7542505264282227,
      "step": 85073
    },
    {
      "epoch": 1.0560035705566407e-05,
      "step": 85073,
      "training_step_time": 8.670029401779175
    },
    {
      "epoch": 1.0560989379882812e-05,
      "model_forward_time": 1.7606093883514404,
      "step": 85074
    },
    {
      "epoch": 1.0560989379882812e-05,
      "step": 85074,
      "training_step_time": 7.7576494216918945
    },
    {
      "epoch": 1.0561943054199218e-05,
      "model_forward_time": 1.7672553062438965,
      "step": 85075
    },
    {
      "epoch": 1.0561943054199218e-05,
      "step": 85075,
      "training_step_time": 7.762233018875122
    },
    {
      "epoch": 1.0562896728515625e-05,
      "model_forward_time": 1.7648437023162842,
      "step": 85076
    },
    {
      "epoch": 1.0562896728515625e-05,
      "step": 85076,
      "training_step_time": 7.6363537311553955
    },
    {
      "epoch": 1.0563850402832031e-05,
      "model_forward_time": 1.7637982368469238,
      "step": 85077
    },
    {
      "epoch": 1.0563850402832031e-05,
      "step": 85077,
      "training_step_time": 7.682854652404785
    },
    {
      "epoch": 1.0564804077148437e-05,
      "model_forward_time": 1.7643187046051025,
      "step": 85078
    },
    {
      "epoch": 1.0564804077148437e-05,
      "step": 85078,
      "training_step_time": 8.710556030273438
    },
    {
      "epoch": 1.0565757751464844e-05,
      "model_forward_time": 1.7590482234954834,
      "step": 85079
    },
    {
      "epoch": 1.0565757751464844e-05,
      "step": 85079,
      "training_step_time": 7.754168510437012
    },
    {
      "epoch": 1.056671142578125e-05,
      "grad_norm": 0.10592533648014069,
      "learning_rate": 5.964292364635682e-07,
      "loss": 0.0532,
      "step": 85080
    },
    {
      "epoch": 1.056671142578125e-05,
      "model_forward_time": 2.3918700218200684,
      "step": 85080
    },
    {
      "dynamics_loss_avg": 0.04563008053228259,
      "epoch": 1.056671142578125e-05,
      "step": 85080
    },
    {
      "action_loss_avg": 0.0012153734889579937,
      "epoch": 1.056671142578125e-05,
      "step": 85080
    },
    {
      "epoch": 1.056671142578125e-05,
      "step": 85080,
      "training_step_time": 7.63782000541687
    },
    {
      "epoch": 1.0567665100097656e-05,
      "model_forward_time": 1.7690074443817139,
      "step": 85081
    },
    {
      "epoch": 1.0567665100097656e-05,
      "step": 85081,
      "training_step_time": 8.401983976364136
    },
    {
      "epoch": 1.0568618774414063e-05,
      "model_forward_time": 1.7547929286956787,
      "step": 85082
    },
    {
      "epoch": 1.0568618774414063e-05,
      "step": 85082,
      "training_step_time": 9.758629560470581
    },
    {
      "epoch": 1.0569572448730469e-05,
      "model_forward_time": 1.758899211883545,
      "step": 85083
    },
    {
      "epoch": 1.0569572448730469e-05,
      "step": 85083,
      "training_step_time": 8.465760946273804
    },
    {
      "epoch": 1.0570526123046875e-05,
      "model_forward_time": 1.761538028717041,
      "step": 85084
    },
    {
      "epoch": 1.0570526123046875e-05,
      "step": 85084,
      "training_step_time": 9.108433961868286
    },
    {
      "epoch": 1.0571479797363282e-05,
      "model_forward_time": 1.7526965141296387,
      "step": 85085
    },
    {
      "epoch": 1.0571479797363282e-05,
      "step": 85085,
      "training_step_time": 7.861698389053345
    },
    {
      "epoch": 1.0572433471679688e-05,
      "model_forward_time": 1.7456531524658203,
      "step": 85086
    },
    {
      "epoch": 1.0572433471679688e-05,
      "step": 85086,
      "training_step_time": 7.673784017562866
    },
    {
      "epoch": 1.0573387145996094e-05,
      "model_forward_time": 1.7494096755981445,
      "step": 85087
    },
    {
      "epoch": 1.0573387145996094e-05,
      "step": 85087,
      "training_step_time": 7.669815540313721
    },
    {
      "epoch": 1.05743408203125e-05,
      "model_forward_time": 1.8005683422088623,
      "step": 85088
    },
    {
      "epoch": 1.05743408203125e-05,
      "step": 85088,
      "training_step_time": 7.693909406661987
    },
    {
      "epoch": 1.0575294494628907e-05,
      "model_forward_time": 2.3939244747161865,
      "step": 85089
    },
    {
      "epoch": 1.0575294494628907e-05,
      "step": 85089,
      "training_step_time": 7.641392707824707
    },
    {
      "epoch": 1.0576248168945312e-05,
      "grad_norm": 0.0756058469414711,
      "learning_rate": 5.956463143219682e-07,
      "loss": 0.0511,
      "step": 85090
    },
    {
      "epoch": 1.0576248168945312e-05,
      "model_forward_time": 1.7446396350860596,
      "step": 85090
    },
    {
      "dynamics_loss_avg": 0.050249783881008626,
      "epoch": 1.0576248168945312e-05,
      "step": 85090
    },
    {
      "action_loss_avg": 0.0011287604560493492,
      "epoch": 1.0576248168945312e-05,
      "step": 85090
    },
    {
      "epoch": 1.0576248168945312e-05,
      "step": 85090,
      "training_step_time": 8.315315961837769
    },
    {
      "epoch": 1.0577201843261718e-05,
      "model_forward_time": 1.741091012954712,
      "step": 85091
    },
    {
      "epoch": 1.0577201843261718e-05,
      "step": 85091,
      "training_step_time": 8.647080421447754
    },
    {
      "epoch": 1.0578155517578126e-05,
      "model_forward_time": 1.7619760036468506,
      "step": 85092
    },
    {
      "epoch": 1.0578155517578126e-05,
      "step": 85092,
      "training_step_time": 8.327611207962036
    },
    {
      "epoch": 1.0579109191894531e-05,
      "model_forward_time": 1.7393391132354736,
      "step": 85093
    },
    {
      "epoch": 1.0579109191894531e-05,
      "step": 85093,
      "training_step_time": 8.471497297286987
    },
    {
      "epoch": 1.0580062866210937e-05,
      "model_forward_time": 1.748164176940918,
      "step": 85094
    },
    {
      "epoch": 1.0580062866210937e-05,
      "step": 85094,
      "training_step_time": 7.842163562774658
    },
    {
      "epoch": 1.0581016540527344e-05,
      "model_forward_time": 1.74434494972229,
      "step": 85095
    },
    {
      "epoch": 1.0581016540527344e-05,
      "step": 85095,
      "training_step_time": 7.622052192687988
    },
    {
      "epoch": 1.058197021484375e-05,
      "model_forward_time": 1.7400422096252441,
      "step": 85096
    },
    {
      "epoch": 1.058197021484375e-05,
      "step": 85096,
      "training_step_time": 7.66985297203064
    },
    {
      "epoch": 1.0582923889160156e-05,
      "model_forward_time": 1.80021333694458,
      "step": 85097
    },
    {
      "epoch": 1.0582923889160156e-05,
      "step": 85097,
      "training_step_time": 8.321937561035156
    },
    {
      "epoch": 1.0583877563476563e-05,
      "model_forward_time": 2.349165201187134,
      "step": 85098
    },
    {
      "epoch": 1.0583877563476563e-05,
      "step": 85098,
      "training_step_time": 8.4286949634552
    },
    {
      "epoch": 1.0584831237792969e-05,
      "model_forward_time": 1.742724895477295,
      "step": 85099
    },
    {
      "epoch": 1.0584831237792969e-05,
      "step": 85099,
      "training_step_time": 8.251533508300781
    },
    {
      "epoch": 1.0585784912109375e-05,
      "grad_norm": 0.07834377884864807,
      "learning_rate": 5.948638738338664e-07,
      "loss": 0.0532,
      "step": 85100
    },
    {
      "epoch": 1.0585784912109375e-05,
      "model_forward_time": 1.7393226623535156,
      "step": 85100
    },
    {
      "dynamics_loss_avg": 0.06540744006633759,
      "epoch": 1.0585784912109375e-05,
      "step": 85100
    },
    {
      "action_loss_avg": 0.0015551007498288527,
      "epoch": 1.0585784912109375e-05,
      "step": 85100
    },
    {
      "epoch": 1.0585784912109375e-05,
      "step": 85100,
      "training_step_time": 9.475347518920898
    },
    {
      "epoch": 1.0586738586425782e-05,
      "model_forward_time": 1.8750660419464111,
      "step": 85101
    },
    {
      "epoch": 1.0586738586425782e-05,
      "step": 85101,
      "training_step_time": 8.73124647140503
    },
    {
      "epoch": 1.0587692260742188e-05,
      "model_forward_time": 1.8702845573425293,
      "step": 85102
    },
    {
      "epoch": 1.0587692260742188e-05,
      "step": 85102,
      "training_step_time": 8.6857430934906
    },
    {
      "epoch": 1.0588645935058594e-05,
      "model_forward_time": 1.8938777446746826,
      "step": 85103
    },
    {
      "epoch": 1.0588645935058594e-05,
      "step": 85103,
      "training_step_time": 7.825534105300903
    },
    {
      "epoch": 1.0589599609375e-05,
      "model_forward_time": 1.8858468532562256,
      "step": 85104
    },
    {
      "epoch": 1.0589599609375e-05,
      "step": 85104,
      "training_step_time": 8.534536123275757
    },
    {
      "epoch": 1.0590553283691407e-05,
      "model_forward_time": 1.8805859088897705,
      "step": 85105
    },
    {
      "epoch": 1.0590553283691407e-05,
      "step": 85105,
      "training_step_time": 7.7012717723846436
    },
    {
      "epoch": 1.0591506958007812e-05,
      "model_forward_time": 1.8793013095855713,
      "step": 85106
    },
    {
      "epoch": 1.0591506958007812e-05,
      "step": 85106,
      "training_step_time": 7.782702445983887
    },
    {
      "epoch": 1.0592460632324218e-05,
      "model_forward_time": 1.8957889080047607,
      "step": 85107
    },
    {
      "epoch": 1.0592460632324218e-05,
      "step": 85107,
      "training_step_time": 9.248454093933105
    },
    {
      "epoch": 1.0593414306640626e-05,
      "model_forward_time": 2.472935438156128,
      "step": 85108
    },
    {
      "epoch": 1.0593414306640626e-05,
      "step": 85108,
      "training_step_time": 7.661003828048706
    },
    {
      "epoch": 1.0594367980957031e-05,
      "model_forward_time": 1.8679249286651611,
      "step": 85109
    },
    {
      "epoch": 1.0594367980957031e-05,
      "step": 85109,
      "training_step_time": 9.672605037689209
    },
    {
      "epoch": 1.0595321655273437e-05,
      "grad_norm": 0.14060288667678833,
      "learning_rate": 5.940819150848271e-07,
      "loss": 0.0525,
      "step": 85110
    },
    {
      "epoch": 1.0595321655273437e-05,
      "model_forward_time": 1.908879280090332,
      "step": 85110
    },
    {
      "dynamics_loss_avg": 0.053287510946393016,
      "epoch": 1.0595321655273437e-05,
      "step": 85110
    },
    {
      "action_loss_avg": 0.0013581899576820434,
      "epoch": 1.0595321655273437e-05,
      "step": 85110
    },
    {
      "epoch": 1.0595321655273437e-05,
      "step": 85110,
      "training_step_time": 9.222484111785889
    },
    {
      "epoch": 1.0596275329589844e-05,
      "model_forward_time": 1.889875888824463,
      "step": 85111
    },
    {
      "epoch": 1.0596275329589844e-05,
      "step": 85111,
      "training_step_time": 7.732827186584473
    },
    {
      "epoch": 1.059722900390625e-05,
      "model_forward_time": 1.883619785308838,
      "step": 85112
    },
    {
      "epoch": 1.059722900390625e-05,
      "step": 85112,
      "training_step_time": 7.82686710357666
    },
    {
      "epoch": 1.0598182678222656e-05,
      "model_forward_time": 1.904209852218628,
      "step": 85113
    },
    {
      "epoch": 1.0598182678222656e-05,
      "step": 85113,
      "training_step_time": 7.633251428604126
    },
    {
      "epoch": 1.0599136352539063e-05,
      "model_forward_time": 1.890160083770752,
      "step": 85114
    },
    {
      "epoch": 1.0599136352539063e-05,
      "step": 85114,
      "training_step_time": 7.6793177127838135
    },
    {
      "epoch": 1.0600090026855469e-05,
      "model_forward_time": 1.8846774101257324,
      "step": 85115
    },
    {
      "epoch": 1.0600090026855469e-05,
      "step": 85115,
      "training_step_time": 7.755619764328003
    },
    {
      "epoch": 1.0601043701171875e-05,
      "model_forward_time": 1.8785526752471924,
      "step": 85116
    },
    {
      "epoch": 1.0601043701171875e-05,
      "step": 85116,
      "training_step_time": 9.292772769927979
    },
    {
      "epoch": 1.0601997375488282e-05,
      "model_forward_time": 2.4518914222717285,
      "step": 85117
    },
    {
      "epoch": 1.0601997375488282e-05,
      "step": 85117,
      "training_step_time": 8.692179203033447
    },
    {
      "epoch": 1.0602951049804688e-05,
      "model_forward_time": 1.8949916362762451,
      "step": 85118
    },
    {
      "epoch": 1.0602951049804688e-05,
      "step": 85118,
      "training_step_time": 9.490614175796509
    },
    {
      "epoch": 1.0603904724121094e-05,
      "model_forward_time": 1.8989489078521729,
      "step": 85119
    },
    {
      "epoch": 1.0603904724121094e-05,
      "step": 85119,
      "training_step_time": 11.013999223709106
    },
    {
      "epoch": 1.06048583984375e-05,
      "grad_norm": 0.08759338408708572,
      "learning_rate": 5.933004381603663e-07,
      "loss": 0.0525,
      "step": 85120
    },
    {
      "epoch": 1.06048583984375e-05,
      "model_forward_time": 1.8665344715118408,
      "step": 85120
    },
    {
      "dynamics_loss_avg": 0.05801528375595808,
      "epoch": 1.06048583984375e-05,
      "step": 85120
    },
    {
      "action_loss_avg": 0.0015745378535939381,
      "epoch": 1.06048583984375e-05,
      "step": 85120
    },
    {
      "epoch": 1.06048583984375e-05,
      "step": 85120,
      "training_step_time": 9.716942548751831
    },
    {
      "epoch": 1.0605812072753907e-05,
      "model_forward_time": 1.8948628902435303,
      "step": 85121
    },
    {
      "epoch": 1.0605812072753907e-05,
      "step": 85121,
      "training_step_time": 7.853471279144287
    },
    {
      "epoch": 1.0606765747070312e-05,
      "model_forward_time": 1.9040138721466064,
      "step": 85122
    },
    {
      "epoch": 1.0606765747070312e-05,
      "step": 85122,
      "training_step_time": 7.648773908615112
    },
    {
      "epoch": 1.0607719421386718e-05,
      "model_forward_time": 1.8991119861602783,
      "step": 85123
    },
    {
      "epoch": 1.0607719421386718e-05,
      "step": 85123,
      "training_step_time": 8.440166473388672
    },
    {
      "epoch": 1.0608673095703126e-05,
      "model_forward_time": 1.8928861618041992,
      "step": 85124
    },
    {
      "epoch": 1.0608673095703126e-05,
      "step": 85124,
      "training_step_time": 7.715685129165649
    },
    {
      "epoch": 1.0609626770019531e-05,
      "model_forward_time": 1.8846094608306885,
      "step": 85125
    },
    {
      "epoch": 1.0609626770019531e-05,
      "step": 85125,
      "training_step_time": 8.251876831054688
    },
    {
      "epoch": 1.0610580444335937e-05,
      "model_forward_time": 2.445906400680542,
      "step": 85126
    },
    {
      "epoch": 1.0610580444335937e-05,
      "step": 85126,
      "training_step_time": 7.69106912612915
    },
    {
      "epoch": 1.0611534118652344e-05,
      "model_forward_time": 1.8972234725952148,
      "step": 85127
    },
    {
      "epoch": 1.0611534118652344e-05,
      "step": 85127,
      "training_step_time": 8.257826328277588
    },
    {
      "epoch": 1.061248779296875e-05,
      "model_forward_time": 1.8853466510772705,
      "step": 85128
    },
    {
      "epoch": 1.061248779296875e-05,
      "step": 85128,
      "training_step_time": 8.720600843429565
    },
    {
      "epoch": 1.0613441467285156e-05,
      "model_forward_time": 1.8943629264831543,
      "step": 85129
    },
    {
      "epoch": 1.0613441467285156e-05,
      "step": 85129,
      "training_step_time": 8.233129024505615
    },
    {
      "epoch": 1.0614395141601563e-05,
      "grad_norm": 0.08441617339849472,
      "learning_rate": 5.92519443145943e-07,
      "loss": 0.0533,
      "step": 85130
    },
    {
      "epoch": 1.0614395141601563e-05,
      "model_forward_time": 1.8727285861968994,
      "step": 85130
    },
    {
      "dynamics_loss_avg": 0.051496139727532864,
      "epoch": 1.0614395141601563e-05,
      "step": 85130
    },
    {
      "action_loss_avg": 0.001462627318687737,
      "epoch": 1.0614395141601563e-05,
      "step": 85130
    },
    {
      "epoch": 1.0614395141601563e-05,
      "step": 85130,
      "training_step_time": 9.676830768585205
    },
    {
      "epoch": 1.0615348815917969e-05,
      "model_forward_time": 1.8891289234161377,
      "step": 85131
    },
    {
      "epoch": 1.0615348815917969e-05,
      "step": 85131,
      "training_step_time": 9.323164224624634
    },
    {
      "epoch": 1.0616302490234375e-05,
      "model_forward_time": 1.8844630718231201,
      "step": 85132
    },
    {
      "epoch": 1.0616302490234375e-05,
      "step": 85132,
      "training_step_time": 8.157896280288696
    },
    {
      "epoch": 1.061725616455078e-05,
      "model_forward_time": 1.8800413608551025,
      "step": 85133
    },
    {
      "epoch": 1.061725616455078e-05,
      "step": 85133,
      "training_step_time": 8.583848714828491
    },
    {
      "epoch": 1.0618209838867188e-05,
      "model_forward_time": 1.888620376586914,
      "step": 85134
    },
    {
      "epoch": 1.0618209838867188e-05,
      "step": 85134,
      "training_step_time": 9.527562618255615
    },
    {
      "epoch": 1.0619163513183594e-05,
      "model_forward_time": 2.4908552169799805,
      "step": 85135
    },
    {
      "epoch": 1.0619163513183594e-05,
      "step": 85135,
      "training_step_time": 7.792686462402344
    },
    {
      "epoch": 1.06201171875e-05,
      "model_forward_time": 1.865262746810913,
      "step": 85136
    },
    {
      "epoch": 1.06201171875e-05,
      "step": 85136,
      "training_step_time": 8.380800724029541
    },
    {
      "epoch": 1.0621070861816407e-05,
      "model_forward_time": 1.883375883102417,
      "step": 85137
    },
    {
      "epoch": 1.0621070861816407e-05,
      "step": 85137,
      "training_step_time": 9.329808950424194
    },
    {
      "epoch": 1.0622024536132812e-05,
      "model_forward_time": 1.882340669631958,
      "step": 85138
    },
    {
      "epoch": 1.0622024536132812e-05,
      "step": 85138,
      "training_step_time": 8.529016494750977
    },
    {
      "epoch": 1.0622978210449218e-05,
      "model_forward_time": 1.8972969055175781,
      "step": 85139
    },
    {
      "epoch": 1.0622978210449218e-05,
      "step": 85139,
      "training_step_time": 7.649634599685669
    },
    {
      "epoch": 1.0623931884765626e-05,
      "grad_norm": 0.0691220685839653,
      "learning_rate": 5.917389301269671e-07,
      "loss": 0.0528,
      "step": 85140
    },
    {
      "epoch": 1.0623931884765626e-05,
      "model_forward_time": 1.8915085792541504,
      "step": 85140
    },
    {
      "dynamics_loss_avg": 0.04107078863307834,
      "epoch": 1.0623931884765626e-05,
      "step": 85140
    },
    {
      "action_loss_avg": 0.0011585107415157836,
      "epoch": 1.0623931884765626e-05,
      "step": 85140
    },
    {
      "epoch": 1.0623931884765626e-05,
      "step": 85140,
      "training_step_time": 7.807996988296509
    },
    {
      "epoch": 1.0624885559082031e-05,
      "model_forward_time": 1.9004108905792236,
      "step": 85141
    },
    {
      "epoch": 1.0624885559082031e-05,
      "step": 85141,
      "training_step_time": 7.688780069351196
    },
    {
      "epoch": 1.0625839233398437e-05,
      "model_forward_time": 1.8836233615875244,
      "step": 85142
    },
    {
      "epoch": 1.0625839233398437e-05,
      "step": 85142,
      "training_step_time": 7.720535039901733
    },
    {
      "epoch": 1.0626792907714844e-05,
      "model_forward_time": 1.879969835281372,
      "step": 85143
    },
    {
      "epoch": 1.0626792907714844e-05,
      "step": 85143,
      "training_step_time": 8.262946844100952
    },
    {
      "epoch": 1.062774658203125e-05,
      "model_forward_time": 2.525451183319092,
      "step": 85144
    },
    {
      "epoch": 1.062774658203125e-05,
      "step": 85144,
      "training_step_time": 7.562635183334351
    },
    {
      "epoch": 1.0628700256347656e-05,
      "model_forward_time": 1.8901424407958984,
      "step": 85145
    },
    {
      "epoch": 1.0628700256347656e-05,
      "step": 85145,
      "training_step_time": 9.723625421524048
    },
    {
      "epoch": 1.0629653930664063e-05,
      "model_forward_time": 1.9018430709838867,
      "step": 85146
    },
    {
      "epoch": 1.0629653930664063e-05,
      "step": 85146,
      "training_step_time": 7.684823036193848
    },
    {
      "epoch": 1.0630607604980469e-05,
      "model_forward_time": 1.8954010009765625,
      "step": 85147
    },
    {
      "epoch": 1.0630607604980469e-05,
      "step": 85147,
      "training_step_time": 8.68065595626831
    },
    {
      "epoch": 1.0631561279296875e-05,
      "model_forward_time": 1.9065556526184082,
      "step": 85148
    },
    {
      "epoch": 1.0631561279296875e-05,
      "step": 85148,
      "training_step_time": 8.309218406677246
    },
    {
      "epoch": 1.063251495361328e-05,
      "model_forward_time": 1.8827235698699951,
      "step": 85149
    },
    {
      "epoch": 1.063251495361328e-05,
      "step": 85149,
      "training_step_time": 9.397360563278198
    },
    {
      "epoch": 1.0633468627929688e-05,
      "grad_norm": 0.09044025093317032,
      "learning_rate": 5.909588991887943e-07,
      "loss": 0.055,
      "step": 85150
    },
    {
      "epoch": 1.0633468627929688e-05,
      "model_forward_time": 1.8822448253631592,
      "step": 85150
    },
    {
      "dynamics_loss_avg": 0.05506545174866915,
      "epoch": 1.0633468627929688e-05,
      "step": 85150
    },
    {
      "action_loss_avg": 0.0013080389457172715,
      "epoch": 1.0633468627929688e-05,
      "step": 85150
    },
    {
      "epoch": 1.0633468627929688e-05,
      "step": 85150,
      "training_step_time": 7.618919134140015
    },
    {
      "epoch": 1.0634422302246094e-05,
      "model_forward_time": 1.8761863708496094,
      "step": 85151
    },
    {
      "epoch": 1.0634422302246094e-05,
      "step": 85151,
      "training_step_time": 7.698022127151489
    },
    {
      "epoch": 1.06353759765625e-05,
      "model_forward_time": 1.8910460472106934,
      "step": 85152
    },
    {
      "epoch": 1.06353759765625e-05,
      "step": 85152,
      "training_step_time": 7.776944160461426
    },
    {
      "epoch": 1.0636329650878907e-05,
      "model_forward_time": 1.903489351272583,
      "step": 85153
    },
    {
      "epoch": 1.0636329650878907e-05,
      "step": 85153,
      "training_step_time": 9.262827634811401
    },
    {
      "epoch": 1.0637283325195312e-05,
      "model_forward_time": 1.8619935512542725,
      "step": 85154
    },
    {
      "epoch": 1.0637283325195312e-05,
      "step": 85154,
      "training_step_time": 8.700477838516235
    },
    {
      "epoch": 1.0638236999511718e-05,
      "model_forward_time": 1.8764777183532715,
      "step": 85155
    },
    {
      "epoch": 1.0638236999511718e-05,
      "step": 85155,
      "training_step_time": 9.495669603347778
    },
    {
      "epoch": 1.0639190673828126e-05,
      "model_forward_time": 1.8724052906036377,
      "step": 85156
    },
    {
      "epoch": 1.0639190673828126e-05,
      "step": 85156,
      "training_step_time": 8.237677335739136
    },
    {
      "epoch": 1.0640144348144531e-05,
      "model_forward_time": 1.8897912502288818,
      "step": 85157
    },
    {
      "epoch": 1.0640144348144531e-05,
      "step": 85157,
      "training_step_time": 10.441858768463135
    },
    {
      "epoch": 1.0641098022460937e-05,
      "model_forward_time": 1.8788633346557617,
      "step": 85158
    },
    {
      "epoch": 1.0641098022460937e-05,
      "step": 85158,
      "training_step_time": 7.724608421325684
    },
    {
      "epoch": 1.0642051696777344e-05,
      "model_forward_time": 1.9025583267211914,
      "step": 85159
    },
    {
      "epoch": 1.0642051696777344e-05,
      "step": 85159,
      "training_step_time": 7.506946563720703
    },
    {
      "epoch": 1.064300537109375e-05,
      "grad_norm": 0.0788196325302124,
      "learning_rate": 5.901793504167286e-07,
      "loss": 0.0541,
      "step": 85160
    },
    {
      "epoch": 1.064300537109375e-05,
      "model_forward_time": 1.8869011402130127,
      "step": 85160
    },
    {
      "dynamics_loss_avg": 0.0757060750387609,
      "epoch": 1.064300537109375e-05,
      "step": 85160
    },
    {
      "action_loss_avg": 0.0011434225292759948,
      "epoch": 1.064300537109375e-05,
      "step": 85160
    },
    {
      "epoch": 1.064300537109375e-05,
      "step": 85160,
      "training_step_time": 7.665447235107422
    },
    {
      "epoch": 1.0643959045410156e-05,
      "model_forward_time": 1.897244930267334,
      "step": 85161
    },
    {
      "epoch": 1.0643959045410156e-05,
      "step": 85161,
      "training_step_time": 7.794007301330566
    },
    {
      "epoch": 1.0644912719726563e-05,
      "model_forward_time": 1.906557321548462,
      "step": 85162
    },
    {
      "epoch": 1.0644912719726563e-05,
      "step": 85162,
      "training_step_time": 7.711453437805176
    },
    {
      "epoch": 1.0645866394042969e-05,
      "model_forward_time": 2.513925790786743,
      "step": 85163
    },
    {
      "epoch": 1.0645866394042969e-05,
      "step": 85163,
      "training_step_time": 7.754935026168823
    },
    {
      "epoch": 1.0646820068359375e-05,
      "model_forward_time": 1.880073070526123,
      "step": 85164
    },
    {
      "epoch": 1.0646820068359375e-05,
      "step": 85164,
      "training_step_time": 8.445322036743164
    },
    {
      "epoch": 1.064777374267578e-05,
      "model_forward_time": 1.9089407920837402,
      "step": 85165
    },
    {
      "epoch": 1.064777374267578e-05,
      "step": 85165,
      "training_step_time": 9.363916397094727
    },
    {
      "epoch": 1.0648727416992188e-05,
      "model_forward_time": 1.879718542098999,
      "step": 85166
    },
    {
      "epoch": 1.0648727416992188e-05,
      "step": 85166,
      "training_step_time": 8.795460939407349
    },
    {
      "epoch": 1.0649681091308594e-05,
      "model_forward_time": 1.8815279006958008,
      "step": 85167
    },
    {
      "epoch": 1.0649681091308594e-05,
      "step": 85167,
      "training_step_time": 8.686014175415039
    },
    {
      "epoch": 1.0650634765625e-05,
      "model_forward_time": 1.8852050304412842,
      "step": 85168
    },
    {
      "epoch": 1.0650634765625e-05,
      "step": 85168,
      "training_step_time": 7.656299352645874
    },
    {
      "epoch": 1.0651588439941407e-05,
      "model_forward_time": 1.8828456401824951,
      "step": 85169
    },
    {
      "epoch": 1.0651588439941407e-05,
      "step": 85169,
      "training_step_time": 7.671423435211182
    },
    {
      "epoch": 1.0652542114257812e-05,
      "grad_norm": 0.10778583586215973,
      "learning_rate": 5.894002838960172e-07,
      "loss": 0.0539,
      "step": 85170
    },
    {
      "epoch": 1.0652542114257812e-05,
      "model_forward_time": 1.9040062427520752,
      "step": 85170
    },
    {
      "dynamics_loss_avg": 0.05497284159064293,
      "epoch": 1.0652542114257812e-05,
      "step": 85170
    },
    {
      "action_loss_avg": 0.0020223734885803425,
      "epoch": 1.0652542114257812e-05,
      "step": 85170
    },
    {
      "epoch": 1.0652542114257812e-05,
      "step": 85170,
      "training_step_time": 7.6965038776397705
    },
    {
      "epoch": 1.0653495788574218e-05,
      "model_forward_time": 1.899552345275879,
      "step": 85171
    },
    {
      "epoch": 1.0653495788574218e-05,
      "step": 85171,
      "training_step_time": 7.67092227935791
    },
    {
      "epoch": 1.0654449462890626e-05,
      "model_forward_time": 2.4832165241241455,
      "step": 85172
    },
    {
      "epoch": 1.0654449462890626e-05,
      "step": 85172,
      "training_step_time": 8.488790512084961
    },
    {
      "epoch": 1.0655403137207031e-05,
      "model_forward_time": 1.8949227333068848,
      "step": 85173
    },
    {
      "epoch": 1.0655403137207031e-05,
      "step": 85173,
      "training_step_time": 8.751071453094482
    },
    {
      "epoch": 1.0656356811523437e-05,
      "model_forward_time": 1.9062576293945312,
      "step": 85174
    },
    {
      "epoch": 1.0656356811523437e-05,
      "step": 85174,
      "training_step_time": 7.813449859619141
    },
    {
      "epoch": 1.0657310485839844e-05,
      "model_forward_time": 1.8934581279754639,
      "step": 85175
    },
    {
      "epoch": 1.0657310485839844e-05,
      "step": 85175,
      "training_step_time": 7.931218385696411
    },
    {
      "epoch": 1.065826416015625e-05,
      "model_forward_time": 1.906935214996338,
      "step": 85176
    },
    {
      "epoch": 1.065826416015625e-05,
      "step": 85176,
      "training_step_time": 8.526158332824707
    },
    {
      "epoch": 1.0659217834472656e-05,
      "model_forward_time": 1.908656120300293,
      "step": 85177
    },
    {
      "epoch": 1.0659217834472656e-05,
      "step": 85177,
      "training_step_time": 7.957682847976685
    },
    {
      "epoch": 1.0660171508789063e-05,
      "model_forward_time": 1.9065124988555908,
      "step": 85178
    },
    {
      "epoch": 1.0660171508789063e-05,
      "step": 85178,
      "training_step_time": 7.76299524307251
    },
    {
      "epoch": 1.0661125183105469e-05,
      "model_forward_time": 1.910247802734375,
      "step": 85179
    },
    {
      "epoch": 1.0661125183105469e-05,
      "step": 85179,
      "training_step_time": 7.800995826721191
    },
    {
      "epoch": 1.0662078857421875e-05,
      "grad_norm": 0.0955297127366066,
      "learning_rate": 5.886216997118605e-07,
      "loss": 0.0557,
      "step": 85180
    },
    {
      "epoch": 1.0662078857421875e-05,
      "model_forward_time": 1.8935234546661377,
      "step": 85180
    },
    {
      "dynamics_loss_avg": 0.05455976650118828,
      "epoch": 1.0662078857421875e-05,
      "step": 85180
    },
    {
      "action_loss_avg": 0.0010085006622830405,
      "epoch": 1.0662078857421875e-05,
      "step": 85180
    },
    {
      "epoch": 1.0662078857421875e-05,
      "step": 85180,
      "training_step_time": 7.838947772979736
    },
    {
      "epoch": 1.066303253173828e-05,
      "model_forward_time": 2.5135245323181152,
      "step": 85181
    },
    {
      "epoch": 1.066303253173828e-05,
      "step": 85181,
      "training_step_time": 7.845905780792236
    },
    {
      "epoch": 1.0663986206054688e-05,
      "model_forward_time": 1.8880398273468018,
      "step": 85182
    },
    {
      "epoch": 1.0663986206054688e-05,
      "step": 85182,
      "training_step_time": 7.757247447967529
    },
    {
      "epoch": 1.0664939880371094e-05,
      "model_forward_time": 1.9007153511047363,
      "step": 85183
    },
    {
      "epoch": 1.0664939880371094e-05,
      "step": 85183,
      "training_step_time": 7.622845888137817
    },
    {
      "epoch": 1.06658935546875e-05,
      "model_forward_time": 1.9066765308380127,
      "step": 85184
    },
    {
      "epoch": 1.06658935546875e-05,
      "step": 85184,
      "training_step_time": 7.86982536315918
    },
    {
      "epoch": 1.0666847229003907e-05,
      "model_forward_time": 1.9092586040496826,
      "step": 85185
    },
    {
      "epoch": 1.0666847229003907e-05,
      "step": 85185,
      "training_step_time": 8.496734142303467
    },
    {
      "epoch": 1.0667800903320312e-05,
      "model_forward_time": 1.9181149005889893,
      "step": 85186
    },
    {
      "epoch": 1.0667800903320312e-05,
      "step": 85186,
      "training_step_time": 7.923325777053833
    },
    {
      "epoch": 1.0668754577636718e-05,
      "model_forward_time": 1.8818323612213135,
      "step": 85187
    },
    {
      "epoch": 1.0668754577636718e-05,
      "step": 85187,
      "training_step_time": 7.756237268447876
    },
    {
      "epoch": 1.0669708251953126e-05,
      "model_forward_time": 1.9047939777374268,
      "step": 85188
    },
    {
      "epoch": 1.0669708251953126e-05,
      "step": 85188,
      "training_step_time": 10.286306381225586
    },
    {
      "epoch": 1.0670661926269531e-05,
      "model_forward_time": 1.8926939964294434,
      "step": 85189
    },
    {
      "epoch": 1.0670661926269531e-05,
      "step": 85189,
      "training_step_time": 8.584547519683838
    },
    {
      "epoch": 1.0671615600585937e-05,
      "grad_norm": 0.07776070386171341,
      "learning_rate": 5.878435979494024e-07,
      "loss": 0.0549,
      "step": 85190
    },
    {
      "epoch": 1.0671615600585937e-05,
      "model_forward_time": 2.5040078163146973,
      "step": 85190
    },
    {
      "dynamics_loss_avg": 0.05207425020635128,
      "epoch": 1.0671615600585937e-05,
      "step": 85190
    },
    {
      "action_loss_avg": 0.0016693144745659084,
      "epoch": 1.0671615600585937e-05,
      "step": 85190
    },
    {
      "epoch": 1.0671615600585937e-05,
      "step": 85190,
      "training_step_time": 7.897666692733765
    },
    {
      "epoch": 1.0672569274902344e-05,
      "model_forward_time": 1.8913967609405518,
      "step": 85191
    },
    {
      "epoch": 1.0672569274902344e-05,
      "step": 85191,
      "training_step_time": 8.59521222114563
    },
    {
      "epoch": 1.067352294921875e-05,
      "model_forward_time": 1.8798279762268066,
      "step": 85192
    },
    {
      "epoch": 1.067352294921875e-05,
      "step": 85192,
      "training_step_time": 8.586835622787476
    },
    {
      "epoch": 1.0674476623535156e-05,
      "model_forward_time": 1.9169731140136719,
      "step": 85193
    },
    {
      "epoch": 1.0674476623535156e-05,
      "step": 85193,
      "training_step_time": 8.373292922973633
    },
    {
      "epoch": 1.0675430297851563e-05,
      "model_forward_time": 1.9069876670837402,
      "step": 85194
    },
    {
      "epoch": 1.0675430297851563e-05,
      "step": 85194,
      "training_step_time": 8.850036144256592
    },
    {
      "epoch": 1.0676383972167969e-05,
      "model_forward_time": 1.8991899490356445,
      "step": 85195
    },
    {
      "epoch": 1.0676383972167969e-05,
      "step": 85195,
      "training_step_time": 8.997143983840942
    },
    {
      "epoch": 1.0677337646484375e-05,
      "model_forward_time": 1.8902544975280762,
      "step": 85196
    },
    {
      "epoch": 1.0677337646484375e-05,
      "step": 85196,
      "training_step_time": 7.704960107803345
    },
    {
      "epoch": 1.067829132080078e-05,
      "model_forward_time": 1.9020564556121826,
      "step": 85197
    },
    {
      "epoch": 1.067829132080078e-05,
      "step": 85197,
      "training_step_time": 7.854970932006836
    },
    {
      "epoch": 1.0679244995117188e-05,
      "model_forward_time": 1.9025120735168457,
      "step": 85198
    },
    {
      "epoch": 1.0679244995117188e-05,
      "step": 85198,
      "training_step_time": 7.8433873653411865
    },
    {
      "epoch": 1.0680198669433594e-05,
      "model_forward_time": 2.5097527503967285,
      "step": 85199
    },
    {
      "epoch": 1.0680198669433594e-05,
      "step": 85199,
      "training_step_time": 7.931041479110718
    },
    {
      "epoch": 1.068115234375e-05,
      "grad_norm": 0.09779386967420578,
      "learning_rate": 5.870659786937344e-07,
      "loss": 0.0556,
      "step": 85200
    },
    {
      "epoch": 1.068115234375e-05,
      "model_forward_time": 1.9005677700042725,
      "step": 85200
    },
    {
      "dynamics_loss_avg": 0.06756695304065943,
      "epoch": 1.068115234375e-05,
      "step": 85200
    },
    {
      "action_loss_avg": 0.001647254644194618,
      "epoch": 1.068115234375e-05,
      "step": 85200
    },
    {
      "epoch": 1.068115234375e-05,
      "step": 85200,
      "training_step_time": 8.469544887542725
    },
    {
      "epoch": 1.0682106018066407e-05,
      "model_forward_time": 1.8981783390045166,
      "step": 85201
    },
    {
      "epoch": 1.0682106018066407e-05,
      "step": 85201,
      "training_step_time": 8.706596612930298
    },
    {
      "epoch": 1.0683059692382812e-05,
      "model_forward_time": 1.8966724872589111,
      "step": 85202
    },
    {
      "epoch": 1.0683059692382812e-05,
      "step": 85202,
      "training_step_time": 7.807086944580078
    },
    {
      "epoch": 1.0684013366699218e-05,
      "model_forward_time": 1.889843225479126,
      "step": 85203
    },
    {
      "epoch": 1.0684013366699218e-05,
      "step": 85203,
      "training_step_time": 7.792262315750122
    },
    {
      "epoch": 1.0684967041015626e-05,
      "model_forward_time": 1.8822011947631836,
      "step": 85204
    },
    {
      "epoch": 1.0684967041015626e-05,
      "step": 85204,
      "training_step_time": 7.888631105422974
    },
    {
      "epoch": 1.0685920715332031e-05,
      "model_forward_time": 1.8969645500183105,
      "step": 85205
    },
    {
      "epoch": 1.0685920715332031e-05,
      "step": 85205,
      "training_step_time": 7.671724319458008
    },
    {
      "epoch": 1.0686874389648437e-05,
      "model_forward_time": 1.9019734859466553,
      "step": 85206
    },
    {
      "epoch": 1.0686874389648437e-05,
      "step": 85206,
      "training_step_time": 7.771982669830322
    },
    {
      "epoch": 1.0687828063964844e-05,
      "model_forward_time": 1.8991034030914307,
      "step": 85207
    },
    {
      "epoch": 1.0687828063964844e-05,
      "step": 85207,
      "training_step_time": 7.836897850036621
    },
    {
      "epoch": 1.068878173828125e-05,
      "model_forward_time": 2.4798014163970947,
      "step": 85208
    },
    {
      "epoch": 1.068878173828125e-05,
      "step": 85208,
      "training_step_time": 8.588656187057495
    },
    {
      "epoch": 1.0689735412597656e-05,
      "model_forward_time": 1.911712646484375,
      "step": 85209
    },
    {
      "epoch": 1.0689735412597656e-05,
      "step": 85209,
      "training_step_time": 7.892144203186035
    },
    {
      "epoch": 1.0690689086914063e-05,
      "grad_norm": 0.07110543549060822,
      "learning_rate": 5.862888420298962e-07,
      "loss": 0.0563,
      "step": 85210
    },
    {
      "epoch": 1.0690689086914063e-05,
      "model_forward_time": 1.8980724811553955,
      "step": 85210
    },
    {
      "dynamics_loss_avg": 0.0551527445204556,
      "epoch": 1.0690689086914063e-05,
      "step": 85210
    },
    {
      "action_loss_avg": 0.001568569999653846,
      "epoch": 1.0690689086914063e-05,
      "step": 85210
    },
    {
      "epoch": 1.0690689086914063e-05,
      "step": 85210,
      "training_step_time": 7.686078786849976
    },
    {
      "epoch": 1.0691642761230469e-05,
      "model_forward_time": 1.8982408046722412,
      "step": 85211
    },
    {
      "epoch": 1.0691642761230469e-05,
      "step": 85211,
      "training_step_time": 7.840087890625
    },
    {
      "epoch": 1.0692596435546875e-05,
      "model_forward_time": 1.9054381847381592,
      "step": 85212
    },
    {
      "epoch": 1.0692596435546875e-05,
      "step": 85212,
      "training_step_time": 8.383188962936401
    },
    {
      "epoch": 1.069355010986328e-05,
      "model_forward_time": 1.8888719081878662,
      "step": 85213
    },
    {
      "epoch": 1.069355010986328e-05,
      "step": 85213,
      "training_step_time": 8.927572011947632
    },
    {
      "epoch": 1.0694503784179688e-05,
      "model_forward_time": 1.8608925342559814,
      "step": 85214
    },
    {
      "epoch": 1.0694503784179688e-05,
      "step": 85214,
      "training_step_time": 7.840535879135132
    },
    {
      "epoch": 1.0695457458496094e-05,
      "model_forward_time": 1.903815507888794,
      "step": 85215
    },
    {
      "epoch": 1.0695457458496094e-05,
      "step": 85215,
      "training_step_time": 8.354853630065918
    },
    {
      "epoch": 1.06964111328125e-05,
      "model_forward_time": 1.8966922760009766,
      "step": 85216
    },
    {
      "epoch": 1.06964111328125e-05,
      "step": 85216,
      "training_step_time": 7.8550803661346436
    },
    {
      "epoch": 1.0697364807128907e-05,
      "model_forward_time": 2.492760181427002,
      "step": 85217
    },
    {
      "epoch": 1.0697364807128907e-05,
      "step": 85217,
      "training_step_time": 8.839518070220947
    },
    {
      "epoch": 1.0698318481445312e-05,
      "model_forward_time": 1.8719489574432373,
      "step": 85218
    },
    {
      "epoch": 1.0698318481445312e-05,
      "step": 85218,
      "training_step_time": 7.9036760330200195
    },
    {
      "epoch": 1.0699272155761718e-05,
      "model_forward_time": 1.887695074081421,
      "step": 85219
    },
    {
      "epoch": 1.0699272155761718e-05,
      "step": 85219,
      "training_step_time": 7.762434005737305
    },
    {
      "epoch": 1.0700225830078126e-05,
      "grad_norm": 0.07590020447969437,
      "learning_rate": 5.855121880428744e-07,
      "loss": 0.0558,
      "step": 85220
    },
    {
      "epoch": 1.0700225830078126e-05,
      "model_forward_time": 1.913238525390625,
      "step": 85220
    },
    {
      "dynamics_loss_avg": 0.06169134341180325,
      "epoch": 1.0700225830078126e-05,
      "step": 85220
    },
    {
      "action_loss_avg": 0.001486389574711211,
      "epoch": 1.0700225830078126e-05,
      "step": 85220
    },
    {
      "epoch": 1.0700225830078126e-05,
      "step": 85220,
      "training_step_time": 8.696537256240845
    },
    {
      "epoch": 1.0701179504394531e-05,
      "model_forward_time": 1.8941667079925537,
      "step": 85221
    },
    {
      "epoch": 1.0701179504394531e-05,
      "step": 85221,
      "training_step_time": 8.887526750564575
    },
    {
      "epoch": 1.0702133178710937e-05,
      "model_forward_time": 1.531991720199585,
      "step": 85222
    },
    {
      "epoch": 1.0702133178710937e-05,
      "step": 85222,
      "training_step_time": 7.800031423568726
    },
    {
      "epoch": 1.0703086853027344e-05,
      "model_forward_time": 1.893054723739624,
      "step": 85223
    },
    {
      "epoch": 1.0703086853027344e-05,
      "step": 85223,
      "training_step_time": 7.952735185623169
    },
    {
      "epoch": 1.070404052734375e-05,
      "model_forward_time": 1.8930108547210693,
      "step": 85224
    },
    {
      "epoch": 1.070404052734375e-05,
      "step": 85224,
      "training_step_time": 7.788067579269409
    },
    {
      "epoch": 1.0704994201660156e-05,
      "model_forward_time": 1.8780829906463623,
      "step": 85225
    },
    {
      "epoch": 1.0704994201660156e-05,
      "step": 85225,
      "training_step_time": 7.928036689758301
    },
    {
      "epoch": 1.0705947875976563e-05,
      "model_forward_time": 2.437255620956421,
      "step": 85226
    },
    {
      "epoch": 1.0705947875976563e-05,
      "step": 85226,
      "training_step_time": 8.387053489685059
    },
    {
      "epoch": 1.0706901550292969e-05,
      "model_forward_time": 1.8963987827301025,
      "step": 85227
    },
    {
      "epoch": 1.0706901550292969e-05,
      "step": 85227,
      "training_step_time": 8.928348302841187
    },
    {
      "epoch": 1.0707855224609375e-05,
      "model_forward_time": 1.8652050495147705,
      "step": 85228
    },
    {
      "epoch": 1.0707855224609375e-05,
      "step": 85228,
      "training_step_time": 7.829307556152344
    },
    {
      "epoch": 1.070880889892578e-05,
      "model_forward_time": 1.8891570568084717,
      "step": 85229
    },
    {
      "epoch": 1.070880889892578e-05,
      "step": 85229,
      "training_step_time": 9.865863561630249
    },
    {
      "epoch": 1.0709762573242188e-05,
      "grad_norm": 0.09128924459218979,
      "learning_rate": 5.847360168176031e-07,
      "loss": 0.055,
      "step": 85230
    },
    {
      "epoch": 1.0709762573242188e-05,
      "model_forward_time": 1.8902740478515625,
      "step": 85230
    },
    {
      "dynamics_loss_avg": 0.07173664961010218,
      "epoch": 1.0709762573242188e-05,
      "step": 85230
    },
    {
      "action_loss_avg": 0.0016346366464858875,
      "epoch": 1.0709762573242188e-05,
      "step": 85230
    },
    {
      "epoch": 1.0709762573242188e-05,
      "step": 85230,
      "training_step_time": 9.747661590576172
    },
    {
      "epoch": 1.0710716247558594e-05,
      "model_forward_time": 1.8680686950683594,
      "step": 85231
    },
    {
      "epoch": 1.0710716247558594e-05,
      "step": 85231,
      "training_step_time": 9.258848905563354
    },
    {
      "epoch": 1.0711669921875e-05,
      "model_forward_time": 1.8869094848632812,
      "step": 85232
    },
    {
      "epoch": 1.0711669921875e-05,
      "step": 85232,
      "training_step_time": 7.980926036834717
    },
    {
      "epoch": 1.0712623596191407e-05,
      "model_forward_time": 1.8933422565460205,
      "step": 85233
    },
    {
      "epoch": 1.0712623596191407e-05,
      "step": 85233,
      "training_step_time": 7.775288105010986
    },
    {
      "epoch": 1.0713577270507812e-05,
      "model_forward_time": 1.908574104309082,
      "step": 85234
    },
    {
      "epoch": 1.0713577270507812e-05,
      "step": 85234,
      "training_step_time": 7.842334508895874
    },
    {
      "epoch": 1.0714530944824218e-05,
      "model_forward_time": 2.4635961055755615,
      "step": 85235
    },
    {
      "epoch": 1.0714530944824218e-05,
      "step": 85235,
      "training_step_time": 7.8990442752838135
    },
    {
      "epoch": 1.0715484619140626e-05,
      "model_forward_time": 1.8857479095458984,
      "step": 85236
    },
    {
      "epoch": 1.0715484619140626e-05,
      "step": 85236,
      "training_step_time": 7.888216733932495
    },
    {
      "epoch": 1.0716438293457031e-05,
      "model_forward_time": 1.9060397148132324,
      "step": 85237
    },
    {
      "epoch": 1.0716438293457031e-05,
      "step": 85237,
      "training_step_time": 8.718511819839478
    },
    {
      "epoch": 1.0717391967773437e-05,
      "model_forward_time": 1.9054384231567383,
      "step": 85238
    },
    {
      "epoch": 1.0717391967773437e-05,
      "step": 85238,
      "training_step_time": 7.8212890625
    },
    {
      "epoch": 1.0718345642089844e-05,
      "model_forward_time": 1.9132378101348877,
      "step": 85239
    },
    {
      "epoch": 1.0718345642089844e-05,
      "step": 85239,
      "training_step_time": 9.569910049438477
    },
    {
      "epoch": 1.071929931640625e-05,
      "grad_norm": 0.078120656311512,
      "learning_rate": 5.839603284389622e-07,
      "loss": 0.0552,
      "step": 85240
    },
    {
      "epoch": 1.071929931640625e-05,
      "model_forward_time": 1.9073247909545898,
      "step": 85240
    },
    {
      "dynamics_loss_avg": 0.06907815560698509,
      "epoch": 1.071929931640625e-05,
      "step": 85240
    },
    {
      "action_loss_avg": 0.0014898324094247072,
      "epoch": 1.071929931640625e-05,
      "step": 85240
    },
    {
      "epoch": 1.071929931640625e-05,
      "step": 85240,
      "training_step_time": 8.921576023101807
    },
    {
      "epoch": 1.0720252990722656e-05,
      "model_forward_time": 1.8976647853851318,
      "step": 85241
    },
    {
      "epoch": 1.0720252990722656e-05,
      "step": 85241,
      "training_step_time": 9.674439907073975
    },
    {
      "epoch": 1.0721206665039063e-05,
      "model_forward_time": 1.9129376411437988,
      "step": 85242
    },
    {
      "epoch": 1.0721206665039063e-05,
      "step": 85242,
      "training_step_time": 7.847655534744263
    },
    {
      "epoch": 1.0722160339355469e-05,
      "model_forward_time": 1.9020404815673828,
      "step": 85243
    },
    {
      "epoch": 1.0722160339355469e-05,
      "step": 85243,
      "training_step_time": 7.7812464237213135
    },
    {
      "epoch": 1.0723114013671875e-05,
      "model_forward_time": 2.4683220386505127,
      "step": 85244
    },
    {
      "epoch": 1.0723114013671875e-05,
      "step": 85244,
      "training_step_time": 7.922487020492554
    },
    {
      "epoch": 1.072406768798828e-05,
      "model_forward_time": 1.8962321281433105,
      "step": 85245
    },
    {
      "epoch": 1.072406768798828e-05,
      "step": 85245,
      "training_step_time": 7.846492767333984
    },
    {
      "epoch": 1.0725021362304688e-05,
      "model_forward_time": 1.8966999053955078,
      "step": 85246
    },
    {
      "epoch": 1.0725021362304688e-05,
      "step": 85246,
      "training_step_time": 8.929387092590332
    },
    {
      "epoch": 1.0725975036621094e-05,
      "model_forward_time": 1.8937819004058838,
      "step": 85247
    },
    {
      "epoch": 1.0725975036621094e-05,
      "step": 85247,
      "training_step_time": 8.539103269577026
    },
    {
      "epoch": 1.07269287109375e-05,
      "model_forward_time": 1.9006102085113525,
      "step": 85248
    },
    {
      "epoch": 1.07269287109375e-05,
      "step": 85248,
      "training_step_time": 9.180931568145752
    },
    {
      "epoch": 1.0727882385253907e-05,
      "model_forward_time": 1.8822402954101562,
      "step": 85249
    },
    {
      "epoch": 1.0727882385253907e-05,
      "step": 85249,
      "training_step_time": 7.870323419570923
    },
    {
      "epoch": 1.0728836059570312e-05,
      "grad_norm": 0.07701253145933151,
      "learning_rate": 5.831851229917813e-07,
      "loss": 0.053,
      "step": 85250
    },
    {
      "epoch": 1.0728836059570312e-05,
      "model_forward_time": 1.895075798034668,
      "step": 85250
    },
    {
      "dynamics_loss_avg": 0.06829128302633762,
      "epoch": 1.0728836059570312e-05,
      "step": 85250
    },
    {
      "action_loss_avg": 0.0012688543298281728,
      "epoch": 1.0728836059570312e-05,
      "step": 85250
    },
    {
      "epoch": 1.0728836059570312e-05,
      "step": 85250,
      "training_step_time": 8.81765627861023
    },
    {
      "epoch": 1.0729789733886718e-05,
      "model_forward_time": 1.9071919918060303,
      "step": 85251
    },
    {
      "epoch": 1.0729789733886718e-05,
      "step": 85251,
      "training_step_time": 8.845800399780273
    },
    {
      "epoch": 1.0730743408203126e-05,
      "model_forward_time": 1.8722848892211914,
      "step": 85252
    },
    {
      "epoch": 1.0730743408203126e-05,
      "step": 85252,
      "training_step_time": 7.791524410247803
    },
    {
      "epoch": 1.0731697082519531e-05,
      "model_forward_time": 2.524517059326172,
      "step": 85253
    },
    {
      "epoch": 1.0731697082519531e-05,
      "step": 85253,
      "training_step_time": 8.437214136123657
    },
    {
      "epoch": 1.0732650756835937e-05,
      "model_forward_time": 1.8932316303253174,
      "step": 85254
    },
    {
      "epoch": 1.0732650756835937e-05,
      "step": 85254,
      "training_step_time": 7.9076642990112305
    },
    {
      "epoch": 1.0733604431152344e-05,
      "model_forward_time": 1.8956694602966309,
      "step": 85255
    },
    {
      "epoch": 1.0733604431152344e-05,
      "step": 85255,
      "training_step_time": 7.866690397262573
    },
    {
      "epoch": 1.073455810546875e-05,
      "model_forward_time": 1.8886635303497314,
      "step": 85256
    },
    {
      "epoch": 1.073455810546875e-05,
      "step": 85256,
      "training_step_time": 7.75294828414917
    },
    {
      "epoch": 1.0735511779785156e-05,
      "model_forward_time": 1.9058754444122314,
      "step": 85257
    },
    {
      "epoch": 1.0735511779785156e-05,
      "step": 85257,
      "training_step_time": 8.75801420211792
    },
    {
      "epoch": 1.0736465454101562e-05,
      "model_forward_time": 1.9120008945465088,
      "step": 85258
    },
    {
      "epoch": 1.0736465454101562e-05,
      "step": 85258,
      "training_step_time": 7.926444053649902
    },
    {
      "epoch": 1.0737419128417969e-05,
      "model_forward_time": 1.8971741199493408,
      "step": 85259
    },
    {
      "epoch": 1.0737419128417969e-05,
      "step": 85259,
      "training_step_time": 7.961257457733154
    },
    {
      "epoch": 1.0738372802734375e-05,
      "grad_norm": 0.0847989171743393,
      "learning_rate": 5.824104005608339e-07,
      "loss": 0.0541,
      "step": 85260
    },
    {
      "epoch": 1.0738372802734375e-05,
      "model_forward_time": 1.8979125022888184,
      "step": 85260
    },
    {
      "dynamics_loss_avg": 0.050473408214747904,
      "epoch": 1.0738372802734375e-05,
      "step": 85260
    },
    {
      "action_loss_avg": 0.0010316869331290946,
      "epoch": 1.0738372802734375e-05,
      "step": 85260
    },
    {
      "epoch": 1.0738372802734375e-05,
      "step": 85260,
      "training_step_time": 11.639632225036621
    },
    {
      "epoch": 1.073932647705078e-05,
      "model_forward_time": 1.9032320976257324,
      "step": 85261
    },
    {
      "epoch": 1.073932647705078e-05,
      "step": 85261,
      "training_step_time": 7.659709930419922
    },
    {
      "epoch": 1.0740280151367188e-05,
      "model_forward_time": 1.8960902690887451,
      "step": 85262
    },
    {
      "epoch": 1.0740280151367188e-05,
      "step": 85262,
      "training_step_time": 8.965131521224976
    },
    {
      "epoch": 1.0741233825683594e-05,
      "model_forward_time": 1.8981826305389404,
      "step": 85263
    },
    {
      "epoch": 1.0741233825683594e-05,
      "step": 85263,
      "training_step_time": 8.433186054229736
    },
    {
      "epoch": 1.07421875e-05,
      "model_forward_time": 1.8975346088409424,
      "step": 85264
    },
    {
      "epoch": 1.07421875e-05,
      "step": 85264,
      "training_step_time": 8.799571514129639
    },
    {
      "epoch": 1.0743141174316407e-05,
      "model_forward_time": 1.8973305225372314,
      "step": 85265
    },
    {
      "epoch": 1.0743141174316407e-05,
      "step": 85265,
      "training_step_time": 6.951501369476318
    },
    {
      "epoch": 1.0744094848632813e-05,
      "model_forward_time": 1.9126181602478027,
      "step": 85266
    },
    {
      "epoch": 1.0744094848632813e-05,
      "step": 85266,
      "training_step_time": 6.909665107727051
    },
    {
      "epoch": 1.0745048522949218e-05,
      "model_forward_time": 1.8910624980926514,
      "step": 85267
    },
    {
      "epoch": 1.0745048522949218e-05,
      "step": 85267,
      "training_step_time": 6.8842997550964355
    },
    {
      "epoch": 1.0746002197265626e-05,
      "model_forward_time": 1.8650283813476562,
      "step": 85268
    },
    {
      "epoch": 1.0746002197265626e-05,
      "step": 85268,
      "training_step_time": 6.955891132354736
    },
    {
      "epoch": 1.0746955871582031e-05,
      "model_forward_time": 1.8912534713745117,
      "step": 85269
    },
    {
      "epoch": 1.0746955871582031e-05,
      "step": 85269,
      "training_step_time": 10.06174087524414
    },
    {
      "epoch": 1.0747909545898437e-05,
      "grad_norm": 0.08436063677072525,
      "learning_rate": 5.81636161230843e-07,
      "loss": 0.0533,
      "step": 85270
    },
    {
      "epoch": 1.0747909545898437e-05,
      "model_forward_time": 2.509517192840576,
      "step": 85270
    },
    {
      "dynamics_loss_avg": 0.05889360383152962,
      "epoch": 1.0747909545898437e-05,
      "step": 85270
    },
    {
      "action_loss_avg": 0.0024326830229256303,
      "epoch": 1.0747909545898437e-05,
      "step": 85270
    },
    {
      "epoch": 1.0747909545898437e-05,
      "step": 85270,
      "training_step_time": 7.947643280029297
    },
    {
      "epoch": 1.0748863220214845e-05,
      "model_forward_time": 1.8952991962432861,
      "step": 85271
    },
    {
      "epoch": 1.0748863220214845e-05,
      "step": 85271,
      "training_step_time": 6.908262491226196
    },
    {
      "epoch": 1.074981689453125e-05,
      "model_forward_time": 1.9033796787261963,
      "step": 85272
    },
    {
      "epoch": 1.074981689453125e-05,
      "step": 85272,
      "training_step_time": 7.823997259140015
    },
    {
      "epoch": 1.0750770568847656e-05,
      "model_forward_time": 1.8921592235565186,
      "step": 85273
    },
    {
      "epoch": 1.0750770568847656e-05,
      "step": 85273,
      "training_step_time": 7.862441301345825
    },
    {
      "epoch": 1.0751724243164062e-05,
      "model_forward_time": 1.8883757591247559,
      "step": 85274
    },
    {
      "epoch": 1.0751724243164062e-05,
      "step": 85274,
      "training_step_time": 7.606861114501953
    },
    {
      "epoch": 1.0752677917480469e-05,
      "model_forward_time": 1.892777919769287,
      "step": 85275
    },
    {
      "epoch": 1.0752677917480469e-05,
      "step": 85275,
      "training_step_time": 6.918058633804321
    },
    {
      "epoch": 1.0753631591796875e-05,
      "model_forward_time": 1.8935978412628174,
      "step": 85276
    },
    {
      "epoch": 1.0753631591796875e-05,
      "step": 85276,
      "training_step_time": 6.911121129989624
    },
    {
      "epoch": 1.075458526611328e-05,
      "model_forward_time": 1.889983892440796,
      "step": 85277
    },
    {
      "epoch": 1.075458526611328e-05,
      "step": 85277,
      "training_step_time": 6.955676555633545
    },
    {
      "epoch": 1.0755538940429688e-05,
      "model_forward_time": 1.897775411605835,
      "step": 85278
    },
    {
      "epoch": 1.0755538940429688e-05,
      "step": 85278,
      "training_step_time": 8.719491004943848
    },
    {
      "epoch": 1.0756492614746094e-05,
      "model_forward_time": 2.52481746673584,
      "step": 85279
    },
    {
      "epoch": 1.0756492614746094e-05,
      "step": 85279,
      "training_step_time": 7.930785894393921
    },
    {
      "epoch": 1.07574462890625e-05,
      "grad_norm": 0.09332010895013809,
      "learning_rate": 5.808624050864792e-07,
      "loss": 0.0523,
      "step": 85280
    },
    {
      "epoch": 1.07574462890625e-05,
      "model_forward_time": 1.8869454860687256,
      "step": 85280
    },
    {
      "dynamics_loss_avg": 0.05002420302480459,
      "epoch": 1.07574462890625e-05,
      "step": 85280
    },
    {
      "action_loss_avg": 0.0014406735834199935,
      "epoch": 1.07574462890625e-05,
      "step": 85280
    },
    {
      "epoch": 1.07574462890625e-05,
      "step": 85280,
      "training_step_time": 6.881259918212891
    },
    {
      "epoch": 1.0758399963378907e-05,
      "model_forward_time": 1.8872606754302979,
      "step": 85281
    },
    {
      "epoch": 1.0758399963378907e-05,
      "step": 85281,
      "training_step_time": 7.952636241912842
    },
    {
      "epoch": 1.0759353637695313e-05,
      "model_forward_time": 1.8724923133850098,
      "step": 85282
    },
    {
      "epoch": 1.0759353637695313e-05,
      "step": 85282,
      "training_step_time": 7.763880491256714
    },
    {
      "epoch": 1.0760307312011718e-05,
      "model_forward_time": 1.8794329166412354,
      "step": 85283
    },
    {
      "epoch": 1.0760307312011718e-05,
      "step": 85283,
      "training_step_time": 7.7793049812316895
    },
    {
      "epoch": 1.0761260986328126e-05,
      "model_forward_time": 1.901092767715454,
      "step": 85284
    },
    {
      "epoch": 1.0761260986328126e-05,
      "step": 85284,
      "training_step_time": 6.913712024688721
    },
    {
      "epoch": 1.0762214660644531e-05,
      "model_forward_time": 1.8962123394012451,
      "step": 85285
    },
    {
      "epoch": 1.0762214660644531e-05,
      "step": 85285,
      "training_step_time": 6.894618988037109
    },
    {
      "epoch": 1.0763168334960937e-05,
      "model_forward_time": 1.8927061557769775,
      "step": 85286
    },
    {
      "epoch": 1.0763168334960937e-05,
      "step": 85286,
      "training_step_time": 6.902755975723267
    },
    {
      "epoch": 1.0764122009277345e-05,
      "model_forward_time": 1.5417156219482422,
      "step": 85287
    },
    {
      "epoch": 1.0764122009277345e-05,
      "step": 85287,
      "training_step_time": 7.952254295349121
    },
    {
      "epoch": 1.076507568359375e-05,
      "model_forward_time": 2.4895100593566895,
      "step": 85288
    },
    {
      "epoch": 1.076507568359375e-05,
      "step": 85288,
      "training_step_time": 9.07201361656189
    },
    {
      "epoch": 1.0766029357910156e-05,
      "model_forward_time": 1.894181251525879,
      "step": 85289
    },
    {
      "epoch": 1.0766029357910156e-05,
      "step": 85289,
      "training_step_time": 7.1137917041778564
    },
    {
      "epoch": 1.0766983032226562e-05,
      "grad_norm": 0.09429445117712021,
      "learning_rate": 5.800891322123592e-07,
      "loss": 0.0525,
      "step": 85290
    },
    {
      "epoch": 1.0766983032226562e-05,
      "model_forward_time": 1.9003527164459229,
      "step": 85290
    },
    {
      "dynamics_loss_avg": 0.0650758508592844,
      "epoch": 1.0766983032226562e-05,
      "step": 85290
    },
    {
      "action_loss_avg": 0.0010717103010392747,
      "epoch": 1.0766983032226562e-05,
      "step": 85290
    },
    {
      "epoch": 1.0766983032226562e-05,
      "step": 85290,
      "training_step_time": 7.808379173278809
    },
    {
      "epoch": 1.0767936706542969e-05,
      "model_forward_time": 1.897695541381836,
      "step": 85291
    },
    {
      "epoch": 1.0767936706542969e-05,
      "step": 85291,
      "training_step_time": 7.673253297805786
    },
    {
      "epoch": 1.0768890380859375e-05,
      "model_forward_time": 1.896406650543213,
      "step": 85292
    },
    {
      "epoch": 1.0768890380859375e-05,
      "step": 85292,
      "training_step_time": 7.8129777908325195
    },
    {
      "epoch": 1.076984405517578e-05,
      "model_forward_time": 1.8911926746368408,
      "step": 85293
    },
    {
      "epoch": 1.076984405517578e-05,
      "step": 85293,
      "training_step_time": 6.865173101425171
    },
    {
      "epoch": 1.0770797729492188e-05,
      "model_forward_time": 1.875197410583496,
      "step": 85294
    },
    {
      "epoch": 1.0770797729492188e-05,
      "step": 85294,
      "training_step_time": 6.924018859863281
    },
    {
      "epoch": 1.0771751403808594e-05,
      "model_forward_time": 1.8905766010284424,
      "step": 85295
    },
    {
      "epoch": 1.0771751403808594e-05,
      "step": 85295,
      "training_step_time": 6.880216360092163
    },
    {
      "epoch": 1.0772705078125e-05,
      "model_forward_time": 1.9028735160827637,
      "step": 85296
    },
    {
      "epoch": 1.0772705078125e-05,
      "step": 85296,
      "training_step_time": 7.930389881134033
    },
    {
      "epoch": 1.0773658752441407e-05,
      "model_forward_time": 2.497953414916992,
      "step": 85297
    },
    {
      "epoch": 1.0773658752441407e-05,
      "step": 85297,
      "training_step_time": 8.97822380065918
    },
    {
      "epoch": 1.0774612426757813e-05,
      "model_forward_time": 1.8837318420410156,
      "step": 85298
    },
    {
      "epoch": 1.0774612426757813e-05,
      "step": 85298,
      "training_step_time": 7.91001033782959
    },
    {
      "epoch": 1.0775566101074218e-05,
      "model_forward_time": 1.8979418277740479,
      "step": 85299
    },
    {
      "epoch": 1.0775566101074218e-05,
      "step": 85299,
      "training_step_time": 7.7929463386535645
    },
    {
      "epoch": 1.0776519775390626e-05,
      "grad_norm": 0.0895141139626503,
      "learning_rate": 5.793163426930454e-07,
      "loss": 0.0531,
      "step": 85300
    },
    {
      "epoch": 1.0776519775390626e-05,
      "model_forward_time": 1.8933777809143066,
      "step": 85300
    },
    {
      "dynamics_loss_avg": 0.090318463742733,
      "epoch": 1.0776519775390626e-05,
      "step": 85300
    },
    {
      "action_loss_avg": 0.0023205111909192057,
      "epoch": 1.0776519775390626e-05,
      "step": 85300
    },
    {
      "epoch": 1.0776519775390626e-05,
      "step": 85300,
      "training_step_time": 7.695938587188721
    },
    {
      "epoch": 1.0777473449707031e-05,
      "model_forward_time": 1.8947367668151855,
      "step": 85301
    },
    {
      "epoch": 1.0777473449707031e-05,
      "step": 85301,
      "training_step_time": 7.841444253921509
    },
    {
      "epoch": 1.0778427124023437e-05,
      "model_forward_time": 1.8673741817474365,
      "step": 85302
    },
    {
      "epoch": 1.0778427124023437e-05,
      "step": 85302,
      "training_step_time": 6.863871812820435
    },
    {
      "epoch": 1.0779380798339845e-05,
      "model_forward_time": 1.889538049697876,
      "step": 85303
    },
    {
      "epoch": 1.0779380798339845e-05,
      "step": 85303,
      "training_step_time": 6.847904682159424
    },
    {
      "epoch": 1.078033447265625e-05,
      "model_forward_time": 1.8848786354064941,
      "step": 85304
    },
    {
      "epoch": 1.078033447265625e-05,
      "step": 85304,
      "training_step_time": 6.854554891586304
    },
    {
      "epoch": 1.0781288146972656e-05,
      "model_forward_time": 1.8888943195343018,
      "step": 85305
    },
    {
      "epoch": 1.0781288146972656e-05,
      "step": 85305,
      "training_step_time": 7.9511878490448
    },
    {
      "epoch": 1.0782241821289062e-05,
      "model_forward_time": 2.489577293395996,
      "step": 85306
    },
    {
      "epoch": 1.0782241821289062e-05,
      "step": 85306,
      "training_step_time": 10.181458234786987
    },
    {
      "epoch": 1.0783195495605469e-05,
      "model_forward_time": 1.8986587524414062,
      "step": 85307
    },
    {
      "epoch": 1.0783195495605469e-05,
      "step": 85307,
      "training_step_time": 7.843464612960815
    },
    {
      "epoch": 1.0784149169921875e-05,
      "model_forward_time": 1.8915331363677979,
      "step": 85308
    },
    {
      "epoch": 1.0784149169921875e-05,
      "step": 85308,
      "training_step_time": 7.749160528182983
    },
    {
      "epoch": 1.078510284423828e-05,
      "model_forward_time": 1.9015846252441406,
      "step": 85309
    },
    {
      "epoch": 1.078510284423828e-05,
      "step": 85309,
      "training_step_time": 7.313846588134766
    },
    {
      "epoch": 1.0786056518554688e-05,
      "grad_norm": 0.09981803596019745,
      "learning_rate": 5.785440366130512e-07,
      "loss": 0.0528,
      "step": 85310
    },
    {
      "epoch": 1.0786056518554688e-05,
      "model_forward_time": 1.8777501583099365,
      "step": 85310
    },
    {
      "dynamics_loss_avg": 0.051928423903882505,
      "epoch": 1.0786056518554688e-05,
      "step": 85310
    },
    {
      "action_loss_avg": 0.0011754691455280407,
      "epoch": 1.0786056518554688e-05,
      "step": 85310
    },
    {
      "epoch": 1.0786056518554688e-05,
      "step": 85310,
      "training_step_time": 7.669833421707153
    },
    {
      "epoch": 1.0787010192871094e-05,
      "model_forward_time": 1.512972116470337,
      "step": 85311
    },
    {
      "epoch": 1.0787010192871094e-05,
      "step": 85311,
      "training_step_time": 7.491586923599243
    },
    {
      "epoch": 1.07879638671875e-05,
      "model_forward_time": 1.883378505706787,
      "step": 85312
    },
    {
      "epoch": 1.07879638671875e-05,
      "step": 85312,
      "training_step_time": 6.896064519882202
    },
    {
      "epoch": 1.0788917541503907e-05,
      "model_forward_time": 1.9085543155670166,
      "step": 85313
    },
    {
      "epoch": 1.0788917541503907e-05,
      "step": 85313,
      "training_step_time": 6.871407985687256
    },
    {
      "epoch": 1.0789871215820313e-05,
      "model_forward_time": 1.8776028156280518,
      "step": 85314
    },
    {
      "epoch": 1.0789871215820313e-05,
      "step": 85314,
      "training_step_time": 7.9387829303741455
    },
    {
      "epoch": 1.0790824890136718e-05,
      "model_forward_time": 1.893465518951416,
      "step": 85315
    },
    {
      "epoch": 1.0790824890136718e-05,
      "step": 85315,
      "training_step_time": 11.143026113510132
    },
    {
      "epoch": 1.0791778564453126e-05,
      "model_forward_time": 2.5046308040618896,
      "step": 85316
    },
    {
      "epoch": 1.0791778564453126e-05,
      "step": 85316,
      "training_step_time": 7.839738368988037
    },
    {
      "epoch": 1.0792732238769531e-05,
      "model_forward_time": 1.8986232280731201,
      "step": 85317
    },
    {
      "epoch": 1.0792732238769531e-05,
      "step": 85317,
      "training_step_time": 6.8757483959198
    },
    {
      "epoch": 1.0793685913085937e-05,
      "model_forward_time": 1.8949377536773682,
      "step": 85318
    },
    {
      "epoch": 1.0793685913085937e-05,
      "step": 85318,
      "training_step_time": 7.434791326522827
    },
    {
      "epoch": 1.0794639587402345e-05,
      "model_forward_time": 1.893038034439087,
      "step": 85319
    },
    {
      "epoch": 1.0794639587402345e-05,
      "step": 85319,
      "training_step_time": 7.680685043334961
    },
    {
      "epoch": 1.079559326171875e-05,
      "grad_norm": 0.1186203807592392,
      "learning_rate": 5.777722140568326e-07,
      "loss": 0.0503,
      "step": 85320
    },
    {
      "epoch": 1.079559326171875e-05,
      "model_forward_time": 1.8964159488677979,
      "step": 85320
    },
    {
      "dynamics_loss_avg": 0.061715902760624886,
      "epoch": 1.079559326171875e-05,
      "step": 85320
    },
    {
      "action_loss_avg": 0.0018915262859081849,
      "epoch": 1.079559326171875e-05,
      "step": 85320
    },
    {
      "epoch": 1.079559326171875e-05,
      "step": 85320,
      "training_step_time": 7.484034538269043
    },
    {
      "epoch": 1.0796546936035156e-05,
      "model_forward_time": 1.8963088989257812,
      "step": 85321
    },
    {
      "epoch": 1.0796546936035156e-05,
      "step": 85321,
      "training_step_time": 6.879471302032471
    },
    {
      "epoch": 1.0797500610351562e-05,
      "model_forward_time": 1.892857313156128,
      "step": 85322
    },
    {
      "epoch": 1.0797500610351562e-05,
      "step": 85322,
      "training_step_time": 6.836752653121948
    },
    {
      "epoch": 1.0798454284667969e-05,
      "model_forward_time": 1.901564359664917,
      "step": 85323
    },
    {
      "epoch": 1.0798454284667969e-05,
      "step": 85323,
      "training_step_time": 6.804126977920532
    },
    {
      "epoch": 1.0799407958984375e-05,
      "model_forward_time": 1.870873212814331,
      "step": 85324
    },
    {
      "epoch": 1.0799407958984375e-05,
      "step": 85324,
      "training_step_time": 9.474627017974854
    },
    {
      "epoch": 1.080036163330078e-05,
      "model_forward_time": 2.4865829944610596,
      "step": 85325
    },
    {
      "epoch": 1.080036163330078e-05,
      "step": 85325,
      "training_step_time": 9.046007871627808
    },
    {
      "epoch": 1.0801315307617188e-05,
      "model_forward_time": 1.9022395610809326,
      "step": 85326
    },
    {
      "epoch": 1.0801315307617188e-05,
      "step": 85326,
      "training_step_time": 7.664509534835815
    },
    {
      "epoch": 1.0802268981933594e-05,
      "model_forward_time": 1.9013903141021729,
      "step": 85327
    },
    {
      "epoch": 1.0802268981933594e-05,
      "step": 85327,
      "training_step_time": 7.7350122928619385
    },
    {
      "epoch": 1.080322265625e-05,
      "model_forward_time": 1.8448824882507324,
      "step": 85328
    },
    {
      "epoch": 1.080322265625e-05,
      "step": 85328,
      "training_step_time": 7.593583345413208
    },
    {
      "epoch": 1.0804176330566407e-05,
      "model_forward_time": 1.8600969314575195,
      "step": 85329
    },
    {
      "epoch": 1.0804176330566407e-05,
      "step": 85329,
      "training_step_time": 7.42500114440918
    },
    {
      "epoch": 1.0805130004882813e-05,
      "grad_norm": 0.1018836721777916,
      "learning_rate": 5.770008751087969e-07,
      "loss": 0.0518,
      "step": 85330
    },
    {
      "epoch": 1.0805130004882813e-05,
      "model_forward_time": 1.8954975605010986,
      "step": 85330
    },
    {
      "dynamics_loss_avg": 0.05053529497236013,
      "epoch": 1.0805130004882813e-05,
      "step": 85330
    },
    {
      "action_loss_avg": 0.0010472915833815931,
      "epoch": 1.0805130004882813e-05,
      "step": 85330
    },
    {
      "epoch": 1.0805130004882813e-05,
      "step": 85330,
      "training_step_time": 6.8359246253967285
    },
    {
      "epoch": 1.0806083679199218e-05,
      "model_forward_time": 1.896878957748413,
      "step": 85331
    },
    {
      "epoch": 1.0806083679199218e-05,
      "step": 85331,
      "training_step_time": 6.856187105178833
    },
    {
      "epoch": 1.0807037353515626e-05,
      "model_forward_time": 1.8849143981933594,
      "step": 85332
    },
    {
      "epoch": 1.0807037353515626e-05,
      "step": 85332,
      "training_step_time": 7.824080467224121
    },
    {
      "epoch": 1.0807991027832031e-05,
      "model_forward_time": 1.9004969596862793,
      "step": 85333
    },
    {
      "epoch": 1.0807991027832031e-05,
      "step": 85333,
      "training_step_time": 8.824076175689697
    },
    {
      "epoch": 1.0808944702148437e-05,
      "model_forward_time": 2.5059478282928467,
      "step": 85334
    },
    {
      "epoch": 1.0808944702148437e-05,
      "step": 85334,
      "training_step_time": 8.802700519561768
    },
    {
      "epoch": 1.0809898376464845e-05,
      "model_forward_time": 1.8697762489318848,
      "step": 85335
    },
    {
      "epoch": 1.0809898376464845e-05,
      "step": 85335,
      "training_step_time": 7.847602605819702
    },
    {
      "epoch": 1.081085205078125e-05,
      "model_forward_time": 1.8774442672729492,
      "step": 85336
    },
    {
      "epoch": 1.081085205078125e-05,
      "step": 85336,
      "training_step_time": 7.686039924621582
    },
    {
      "epoch": 1.0811805725097656e-05,
      "model_forward_time": 1.8794455528259277,
      "step": 85337
    },
    {
      "epoch": 1.0811805725097656e-05,
      "step": 85337,
      "training_step_time": 8.235551834106445
    },
    {
      "epoch": 1.0812759399414062e-05,
      "model_forward_time": 1.8897674083709717,
      "step": 85338
    },
    {
      "epoch": 1.0812759399414062e-05,
      "step": 85338,
      "training_step_time": 7.363178014755249
    },
    {
      "epoch": 1.0813713073730469e-05,
      "model_forward_time": 1.869088888168335,
      "step": 85339
    },
    {
      "epoch": 1.0813713073730469e-05,
      "step": 85339,
      "training_step_time": 6.843973875045776
    },
    {
      "epoch": 1.0814666748046875e-05,
      "grad_norm": 0.0787847712635994,
      "learning_rate": 5.762300198532933e-07,
      "loss": 0.0522,
      "step": 85340
    },
    {
      "epoch": 1.0814666748046875e-05,
      "model_forward_time": 1.911525011062622,
      "step": 85340
    },
    {
      "dynamics_loss_avg": 0.04660614226013422,
      "epoch": 1.0814666748046875e-05,
      "step": 85340
    },
    {
      "action_loss_avg": 0.0016266337188426405,
      "epoch": 1.0814666748046875e-05,
      "step": 85340
    },
    {
      "epoch": 1.0814666748046875e-05,
      "step": 85340,
      "training_step_time": 6.806968927383423
    },
    {
      "epoch": 1.081562042236328e-05,
      "model_forward_time": 1.8777203559875488,
      "step": 85341
    },
    {
      "epoch": 1.081562042236328e-05,
      "step": 85341,
      "training_step_time": 6.812159776687622
    },
    {
      "epoch": 1.0816574096679688e-05,
      "model_forward_time": 1.914825677871704,
      "step": 85342
    },
    {
      "epoch": 1.0816574096679688e-05,
      "step": 85342,
      "training_step_time": 8.560189008712769
    },
    {
      "epoch": 1.0817527770996094e-05,
      "model_forward_time": 2.531033515930176,
      "step": 85343
    },
    {
      "epoch": 1.0817527770996094e-05,
      "step": 85343,
      "training_step_time": 8.59811782836914
    },
    {
      "epoch": 1.08184814453125e-05,
      "model_forward_time": 1.9248847961425781,
      "step": 85344
    },
    {
      "epoch": 1.08184814453125e-05,
      "step": 85344,
      "training_step_time": 7.807851552963257
    },
    {
      "epoch": 1.0819435119628907e-05,
      "model_forward_time": 1.9105732440948486,
      "step": 85345
    },
    {
      "epoch": 1.0819435119628907e-05,
      "step": 85345,
      "training_step_time": 7.388951301574707
    },
    {
      "epoch": 1.0820388793945313e-05,
      "model_forward_time": 1.9008162021636963,
      "step": 85346
    },
    {
      "epoch": 1.0820388793945313e-05,
      "step": 85346,
      "training_step_time": 9.405717611312866
    },
    {
      "epoch": 1.0821342468261718e-05,
      "model_forward_time": 1.8608241081237793,
      "step": 85347
    },
    {
      "epoch": 1.0821342468261718e-05,
      "step": 85347,
      "training_step_time": 7.164198398590088
    },
    {
      "epoch": 1.0822296142578126e-05,
      "model_forward_time": 1.8973534107208252,
      "step": 85348
    },
    {
      "epoch": 1.0822296142578126e-05,
      "step": 85348,
      "training_step_time": 7.413830280303955
    },
    {
      "epoch": 1.0823249816894531e-05,
      "model_forward_time": 1.8700144290924072,
      "step": 85349
    },
    {
      "epoch": 1.0823249816894531e-05,
      "step": 85349,
      "training_step_time": 6.799613952636719
    },
    {
      "epoch": 1.0824203491210937e-05,
      "grad_norm": 0.09773602336645126,
      "learning_rate": 5.754596483746261e-07,
      "loss": 0.0521,
      "step": 85350
    },
    {
      "epoch": 1.0824203491210937e-05,
      "model_forward_time": 1.8901560306549072,
      "step": 85350
    },
    {
      "dynamics_loss_avg": 0.05629931911826134,
      "epoch": 1.0824203491210937e-05,
      "step": 85350
    },
    {
      "action_loss_avg": 0.0021854927705135196,
      "epoch": 1.0824203491210937e-05,
      "step": 85350
    },
    {
      "epoch": 1.0824203491210937e-05,
      "step": 85350,
      "training_step_time": 6.799470901489258
    },
    {
      "epoch": 1.0825157165527345e-05,
      "model_forward_time": 1.8821008205413818,
      "step": 85351
    },
    {
      "epoch": 1.0825157165527345e-05,
      "step": 85351,
      "training_step_time": 7.833194017410278
    },
    {
      "epoch": 1.082611083984375e-05,
      "model_forward_time": 2.4955806732177734,
      "step": 85352
    },
    {
      "epoch": 1.082611083984375e-05,
      "step": 85352,
      "training_step_time": 8.722218751907349
    },
    {
      "epoch": 1.0827064514160156e-05,
      "model_forward_time": 1.8727130889892578,
      "step": 85353
    },
    {
      "epoch": 1.0827064514160156e-05,
      "step": 85353,
      "training_step_time": 8.287605285644531
    },
    {
      "epoch": 1.0828018188476562e-05,
      "model_forward_time": 1.8895649909973145,
      "step": 85354
    },
    {
      "epoch": 1.0828018188476562e-05,
      "step": 85354,
      "training_step_time": 7.533447742462158
    },
    {
      "epoch": 1.082897186279297e-05,
      "model_forward_time": 1.8779447078704834,
      "step": 85355
    },
    {
      "epoch": 1.082897186279297e-05,
      "step": 85355,
      "training_step_time": 7.796937942504883
    },
    {
      "epoch": 1.0829925537109375e-05,
      "model_forward_time": 1.8876774311065674,
      "step": 85356
    },
    {
      "epoch": 1.0829925537109375e-05,
      "step": 85356,
      "training_step_time": 7.663356304168701
    },
    {
      "epoch": 1.083087921142578e-05,
      "model_forward_time": 1.892195701599121,
      "step": 85357
    },
    {
      "epoch": 1.083087921142578e-05,
      "step": 85357,
      "training_step_time": 7.512487888336182
    },
    {
      "epoch": 1.0831832885742188e-05,
      "model_forward_time": 1.8984737396240234,
      "step": 85358
    },
    {
      "epoch": 1.0831832885742188e-05,
      "step": 85358,
      "training_step_time": 6.8205811977386475
    },
    {
      "epoch": 1.0832786560058594e-05,
      "model_forward_time": 1.9040241241455078,
      "step": 85359
    },
    {
      "epoch": 1.0832786560058594e-05,
      "step": 85359,
      "training_step_time": 6.937408208847046
    },
    {
      "epoch": 1.0833740234375e-05,
      "grad_norm": 0.08124645054340363,
      "learning_rate": 5.746897607570384e-07,
      "loss": 0.0518,
      "step": 85360
    },
    {
      "epoch": 1.0833740234375e-05,
      "model_forward_time": 1.9011955261230469,
      "step": 85360
    },
    {
      "dynamics_loss_avg": 0.06892489939928055,
      "epoch": 1.0833740234375e-05,
      "step": 85360
    },
    {
      "action_loss_avg": 0.0011842153733596205,
      "epoch": 1.0833740234375e-05,
      "step": 85360
    },
    {
      "epoch": 1.0833740234375e-05,
      "step": 85360,
      "training_step_time": 8.323554039001465
    },
    {
      "epoch": 1.0834693908691407e-05,
      "model_forward_time": 1.8842577934265137,
      "step": 85361
    },
    {
      "epoch": 1.0834693908691407e-05,
      "step": 85361,
      "training_step_time": 12.011856079101562
    },
    {
      "epoch": 1.0835647583007813e-05,
      "model_forward_time": 2.5555365085601807,
      "step": 85362
    },
    {
      "epoch": 1.0835647583007813e-05,
      "step": 85362,
      "training_step_time": 7.914257526397705
    },
    {
      "epoch": 1.0836601257324218e-05,
      "model_forward_time": 1.9125618934631348,
      "step": 85363
    },
    {
      "epoch": 1.0836601257324218e-05,
      "step": 85363,
      "training_step_time": 7.940203428268433
    },
    {
      "epoch": 1.0837554931640626e-05,
      "model_forward_time": 1.920590877532959,
      "step": 85364
    },
    {
      "epoch": 1.0837554931640626e-05,
      "step": 85364,
      "training_step_time": 7.7202842235565186
    },
    {
      "epoch": 1.0838508605957031e-05,
      "model_forward_time": 1.9003491401672363,
      "step": 85365
    },
    {
      "epoch": 1.0838508605957031e-05,
      "step": 85365,
      "training_step_time": 7.718923807144165
    },
    {
      "epoch": 1.0839462280273437e-05,
      "model_forward_time": 1.909722089767456,
      "step": 85366
    },
    {
      "epoch": 1.0839462280273437e-05,
      "step": 85366,
      "training_step_time": 7.470953941345215
    },
    {
      "epoch": 1.0840415954589845e-05,
      "model_forward_time": 1.9104726314544678,
      "step": 85367
    },
    {
      "epoch": 1.0840415954589845e-05,
      "step": 85367,
      "training_step_time": 6.873135805130005
    },
    {
      "epoch": 1.084136962890625e-05,
      "model_forward_time": 1.8736746311187744,
      "step": 85368
    },
    {
      "epoch": 1.084136962890625e-05,
      "step": 85368,
      "training_step_time": 6.816247224807739
    },
    {
      "epoch": 1.0842323303222656e-05,
      "model_forward_time": 1.900221824645996,
      "step": 85369
    },
    {
      "epoch": 1.0842323303222656e-05,
      "step": 85369,
      "training_step_time": 7.721250295639038
    },
    {
      "epoch": 1.0843276977539062e-05,
      "grad_norm": 0.09121789038181305,
      "learning_rate": 5.739203570847262e-07,
      "loss": 0.0515,
      "step": 85370
    },
    {
      "epoch": 1.0843276977539062e-05,
      "model_forward_time": 1.884002685546875,
      "step": 85370
    },
    {
      "dynamics_loss_avg": 0.05564265362918377,
      "epoch": 1.0843276977539062e-05,
      "step": 85370
    },
    {
      "action_loss_avg": 0.0010696556943003087,
      "epoch": 1.0843276977539062e-05,
      "step": 85370
    },
    {
      "epoch": 1.0843276977539062e-05,
      "step": 85370,
      "training_step_time": 8.541823863983154
    },
    {
      "epoch": 1.084423065185547e-05,
      "model_forward_time": 2.5324549674987793,
      "step": 85371
    },
    {
      "epoch": 1.084423065185547e-05,
      "step": 85371,
      "training_step_time": 8.56007719039917
    },
    {
      "epoch": 1.0845184326171875e-05,
      "model_forward_time": 1.9054663181304932,
      "step": 85372
    },
    {
      "epoch": 1.0845184326171875e-05,
      "step": 85372,
      "training_step_time": 7.798568248748779
    },
    {
      "epoch": 1.084613800048828e-05,
      "model_forward_time": 1.9033489227294922,
      "step": 85373
    },
    {
      "epoch": 1.084613800048828e-05,
      "step": 85373,
      "training_step_time": 7.6975929737091064
    },
    {
      "epoch": 1.0847091674804688e-05,
      "model_forward_time": 1.8895065784454346,
      "step": 85374
    },
    {
      "epoch": 1.0847091674804688e-05,
      "step": 85374,
      "training_step_time": 9.539228200912476
    },
    {
      "epoch": 1.0848045349121094e-05,
      "model_forward_time": 1.9059345722198486,
      "step": 85375
    },
    {
      "epoch": 1.0848045349121094e-05,
      "step": 85375,
      "training_step_time": 7.403291940689087
    },
    {
      "epoch": 1.08489990234375e-05,
      "model_forward_time": 1.899630069732666,
      "step": 85376
    },
    {
      "epoch": 1.08489990234375e-05,
      "step": 85376,
      "training_step_time": 6.705096006393433
    },
    {
      "epoch": 1.0849952697753907e-05,
      "model_forward_time": 1.908909559249878,
      "step": 85377
    },
    {
      "epoch": 1.0849952697753907e-05,
      "step": 85377,
      "training_step_time": 6.723639488220215
    },
    {
      "epoch": 1.0850906372070313e-05,
      "model_forward_time": 1.8830499649047852,
      "step": 85378
    },
    {
      "epoch": 1.0850906372070313e-05,
      "step": 85378,
      "training_step_time": 10.360363006591797
    },
    {
      "epoch": 1.0851860046386718e-05,
      "model_forward_time": 1.9014272689819336,
      "step": 85379
    },
    {
      "epoch": 1.0851860046386718e-05,
      "step": 85379,
      "training_step_time": 9.604442119598389
    },
    {
      "epoch": 1.0852813720703126e-05,
      "grad_norm": 0.09792351722717285,
      "learning_rate": 5.731514374418278e-07,
      "loss": 0.0527,
      "step": 85380
    },
    {
      "epoch": 1.0852813720703126e-05,
      "model_forward_time": 2.5353198051452637,
      "step": 85380
    },
    {
      "dynamics_loss_avg": 0.055317381769418715,
      "epoch": 1.0852813720703126e-05,
      "step": 85380
    },
    {
      "action_loss_avg": 0.000626113757607527,
      "epoch": 1.0852813720703126e-05,
      "step": 85380
    },
    {
      "epoch": 1.0852813720703126e-05,
      "step": 85380,
      "training_step_time": 9.246645212173462
    },
    {
      "epoch": 1.0853767395019531e-05,
      "model_forward_time": 1.9068784713745117,
      "step": 85381
    },
    {
      "epoch": 1.0853767395019531e-05,
      "step": 85381,
      "training_step_time": 7.750967264175415
    },
    {
      "epoch": 1.0854721069335937e-05,
      "model_forward_time": 1.8761699199676514,
      "step": 85382
    },
    {
      "epoch": 1.0854721069335937e-05,
      "step": 85382,
      "training_step_time": 8.386940956115723
    },
    {
      "epoch": 1.0855674743652343e-05,
      "model_forward_time": 1.8868837356567383,
      "step": 85383
    },
    {
      "epoch": 1.0855674743652343e-05,
      "step": 85383,
      "training_step_time": 7.559754133224487
    },
    {
      "epoch": 1.085662841796875e-05,
      "model_forward_time": 1.8949863910675049,
      "step": 85384
    },
    {
      "epoch": 1.085662841796875e-05,
      "step": 85384,
      "training_step_time": 7.558727979660034
    },
    {
      "epoch": 1.0857582092285156e-05,
      "model_forward_time": 1.9096250534057617,
      "step": 85385
    },
    {
      "epoch": 1.0857582092285156e-05,
      "step": 85385,
      "training_step_time": 6.680584669113159
    },
    {
      "epoch": 1.0858535766601562e-05,
      "model_forward_time": 1.9083616733551025,
      "step": 85386
    },
    {
      "epoch": 1.0858535766601562e-05,
      "step": 85386,
      "training_step_time": 6.696107387542725
    },
    {
      "epoch": 1.085948944091797e-05,
      "model_forward_time": 1.9082374572753906,
      "step": 85387
    },
    {
      "epoch": 1.085948944091797e-05,
      "step": 85387,
      "training_step_time": 8.337053537368774
    },
    {
      "epoch": 1.0860443115234375e-05,
      "model_forward_time": 1.923140525817871,
      "step": 85388
    },
    {
      "epoch": 1.0860443115234375e-05,
      "step": 85388,
      "training_step_time": 10.460412979125977
    },
    {
      "epoch": 1.086139678955078e-05,
      "model_forward_time": 2.540513277053833,
      "step": 85389
    },
    {
      "epoch": 1.086139678955078e-05,
      "step": 85389,
      "training_step_time": 9.465856552124023
    },
    {
      "epoch": 1.0862350463867188e-05,
      "grad_norm": 0.07999476790428162,
      "learning_rate": 5.723830019124332e-07,
      "loss": 0.0529,
      "step": 85390
    },
    {
      "epoch": 1.0862350463867188e-05,
      "model_forward_time": 1.8917818069458008,
      "step": 85390
    },
    {
      "dynamics_loss_avg": 0.05585363730788231,
      "epoch": 1.0862350463867188e-05,
      "step": 85390
    },
    {
      "action_loss_avg": 0.0012714054057141766,
      "epoch": 1.0862350463867188e-05,
      "step": 85390
    },
    {
      "epoch": 1.0862350463867188e-05,
      "step": 85390,
      "training_step_time": 7.736964464187622
    },
    {
      "epoch": 1.0863304138183594e-05,
      "model_forward_time": 1.9139721393585205,
      "step": 85391
    },
    {
      "epoch": 1.0863304138183594e-05,
      "step": 85391,
      "training_step_time": 7.606614351272583
    },
    {
      "epoch": 1.08642578125e-05,
      "model_forward_time": 1.9020819664001465,
      "step": 85392
    },
    {
      "epoch": 1.08642578125e-05,
      "step": 85392,
      "training_step_time": 7.8309221267700195
    },
    {
      "epoch": 1.0865211486816407e-05,
      "model_forward_time": 1.9027328491210938,
      "step": 85393
    },
    {
      "epoch": 1.0865211486816407e-05,
      "step": 85393,
      "training_step_time": 7.7443132400512695
    },
    {
      "epoch": 1.0866165161132813e-05,
      "model_forward_time": 1.9213917255401611,
      "step": 85394
    },
    {
      "epoch": 1.0866165161132813e-05,
      "step": 85394,
      "training_step_time": 6.842514276504517
    },
    {
      "epoch": 1.0867118835449218e-05,
      "model_forward_time": 1.909611701965332,
      "step": 85395
    },
    {
      "epoch": 1.0867118835449218e-05,
      "step": 85395,
      "training_step_time": 6.844801664352417
    },
    {
      "epoch": 1.0868072509765626e-05,
      "model_forward_time": 1.9119184017181396,
      "step": 85396
    },
    {
      "epoch": 1.0868072509765626e-05,
      "step": 85396,
      "training_step_time": 7.6833672523498535
    },
    {
      "epoch": 1.0869026184082031e-05,
      "model_forward_time": 1.9107084274291992,
      "step": 85397
    },
    {
      "epoch": 1.0869026184082031e-05,
      "step": 85397,
      "training_step_time": 11.155818700790405
    },
    {
      "epoch": 1.0869979858398437e-05,
      "model_forward_time": 2.5172688961029053,
      "step": 85398
    },
    {
      "epoch": 1.0869979858398437e-05,
      "step": 85398,
      "training_step_time": 9.279972076416016
    },
    {
      "epoch": 1.0870933532714843e-05,
      "model_forward_time": 1.9154078960418701,
      "step": 85399
    },
    {
      "epoch": 1.0870933532714843e-05,
      "step": 85399,
      "training_step_time": 7.75322699546814
    },
    {
      "epoch": 1.087188720703125e-05,
      "grad_norm": 0.08232590556144714,
      "learning_rate": 5.716150505805762e-07,
      "loss": 0.0537,
      "step": 85400
    },
    {
      "epoch": 1.087188720703125e-05,
      "model_forward_time": 1.928678274154663,
      "step": 85400
    },
    {
      "dynamics_loss_avg": 0.04390157703310251,
      "epoch": 1.087188720703125e-05,
      "step": 85400
    },
    {
      "action_loss_avg": 0.0009983238531276583,
      "epoch": 1.087188720703125e-05,
      "step": 85400
    },
    {
      "epoch": 1.087188720703125e-05,
      "step": 85400,
      "training_step_time": 7.546931266784668
    },
    {
      "epoch": 1.0872840881347656e-05,
      "model_forward_time": 1.9187476634979248,
      "step": 85401
    },
    {
      "epoch": 1.0872840881347656e-05,
      "step": 85401,
      "training_step_time": 8.59620213508606
    },
    {
      "epoch": 1.0873794555664062e-05,
      "model_forward_time": 1.9014534950256348,
      "step": 85402
    },
    {
      "epoch": 1.0873794555664062e-05,
      "step": 85402,
      "training_step_time": 7.578048467636108
    },
    {
      "epoch": 1.087474822998047e-05,
      "model_forward_time": 1.9016363620758057,
      "step": 85403
    },
    {
      "epoch": 1.087474822998047e-05,
      "step": 85403,
      "training_step_time": 7.360499858856201
    },
    {
      "epoch": 1.0875701904296875e-05,
      "model_forward_time": 1.9155051708221436,
      "step": 85404
    },
    {
      "epoch": 1.0875701904296875e-05,
      "step": 85404,
      "training_step_time": 6.723290920257568
    },
    {
      "epoch": 1.087665557861328e-05,
      "model_forward_time": 1.922269344329834,
      "step": 85405
    },
    {
      "epoch": 1.087665557861328e-05,
      "step": 85405,
      "training_step_time": 7.57999062538147
    },
    {
      "epoch": 1.0877609252929688e-05,
      "model_forward_time": 1.8606536388397217,
      "step": 85406
    },
    {
      "epoch": 1.0877609252929688e-05,
      "step": 85406,
      "training_step_time": 8.710281610488892
    },
    {
      "epoch": 1.0878562927246094e-05,
      "model_forward_time": 1.897742509841919,
      "step": 85407
    },
    {
      "epoch": 1.0878562927246094e-05,
      "step": 85407,
      "training_step_time": 9.352523565292358
    },
    {
      "epoch": 1.08795166015625e-05,
      "model_forward_time": 1.903510570526123,
      "step": 85408
    },
    {
      "epoch": 1.08795166015625e-05,
      "step": 85408,
      "training_step_time": 7.681937217712402
    },
    {
      "epoch": 1.0880470275878907e-05,
      "model_forward_time": 1.9024631977081299,
      "step": 85409
    },
    {
      "epoch": 1.0880470275878907e-05,
      "step": 85409,
      "training_step_time": 7.78047513961792
    },
    {
      "epoch": 1.0881423950195313e-05,
      "grad_norm": 0.07551554590463638,
      "learning_rate": 5.708475835302407e-07,
      "loss": 0.0539,
      "step": 85410
    },
    {
      "epoch": 1.0881423950195313e-05,
      "model_forward_time": 1.923954963684082,
      "step": 85410
    },
    {
      "dynamics_loss_avg": 0.06617597322911024,
      "epoch": 1.0881423950195313e-05,
      "step": 85410
    },
    {
      "action_loss_avg": 0.0025869638891890647,
      "epoch": 1.0881423950195313e-05,
      "step": 85410
    },
    {
      "epoch": 1.0881423950195313e-05,
      "step": 85410,
      "training_step_time": 8.658753633499146
    },
    {
      "epoch": 1.0882377624511718e-05,
      "model_forward_time": 1.9233293533325195,
      "step": 85411
    },
    {
      "epoch": 1.0882377624511718e-05,
      "step": 85411,
      "training_step_time": 7.606062650680542
    },
    {
      "epoch": 1.0883331298828126e-05,
      "model_forward_time": 1.9217846393585205,
      "step": 85412
    },
    {
      "epoch": 1.0883331298828126e-05,
      "step": 85412,
      "training_step_time": 7.339003801345825
    },
    {
      "epoch": 1.0884284973144531e-05,
      "model_forward_time": 1.913508415222168,
      "step": 85413
    },
    {
      "epoch": 1.0884284973144531e-05,
      "step": 85413,
      "training_step_time": 6.724310398101807
    },
    {
      "epoch": 1.0885238647460937e-05,
      "model_forward_time": 1.9281563758850098,
      "step": 85414
    },
    {
      "epoch": 1.0885238647460937e-05,
      "step": 85414,
      "training_step_time": 7.568524122238159
    },
    {
      "epoch": 1.0886192321777343e-05,
      "model_forward_time": 1.9211769104003906,
      "step": 85415
    },
    {
      "epoch": 1.0886192321777343e-05,
      "step": 85415,
      "training_step_time": 9.981956243515015
    },
    {
      "epoch": 1.088714599609375e-05,
      "model_forward_time": 1.9109749794006348,
      "step": 85416
    },
    {
      "epoch": 1.088714599609375e-05,
      "step": 85416,
      "training_step_time": 10.252057552337646
    },
    {
      "epoch": 1.0888099670410156e-05,
      "model_forward_time": 2.5500268936157227,
      "step": 85417
    },
    {
      "epoch": 1.0888099670410156e-05,
      "step": 85417,
      "training_step_time": 8.520844459533691
    },
    {
      "epoch": 1.0889053344726562e-05,
      "model_forward_time": 1.8994066715240479,
      "step": 85418
    },
    {
      "epoch": 1.0889053344726562e-05,
      "step": 85418,
      "training_step_time": 8.421736001968384
    },
    {
      "epoch": 1.089000701904297e-05,
      "model_forward_time": 1.9027888774871826,
      "step": 85419
    },
    {
      "epoch": 1.089000701904297e-05,
      "step": 85419,
      "training_step_time": 7.682284832000732
    },
    {
      "epoch": 1.0890960693359375e-05,
      "grad_norm": 0.09593123197555542,
      "learning_rate": 5.70080600845353e-07,
      "loss": 0.0541,
      "step": 85420
    },
    {
      "epoch": 1.0890960693359375e-05,
      "model_forward_time": 1.9018244743347168,
      "step": 85420
    },
    {
      "dynamics_loss_avg": 0.05138801708817482,
      "epoch": 1.0890960693359375e-05,
      "step": 85420
    },
    {
      "action_loss_avg": 0.0013840156374499203,
      "epoch": 1.0890960693359375e-05,
      "step": 85420
    },
    {
      "epoch": 1.0890960693359375e-05,
      "step": 85420,
      "training_step_time": 7.571475267410278
    },
    {
      "epoch": 1.089191436767578e-05,
      "model_forward_time": 1.9261271953582764,
      "step": 85421
    },
    {
      "epoch": 1.089191436767578e-05,
      "step": 85421,
      "training_step_time": 7.359436273574829
    },
    {
      "epoch": 1.0892868041992188e-05,
      "model_forward_time": 1.919304609298706,
      "step": 85422
    },
    {
      "epoch": 1.0892868041992188e-05,
      "step": 85422,
      "training_step_time": 6.694995164871216
    },
    {
      "epoch": 1.0893821716308594e-05,
      "model_forward_time": 1.9168837070465088,
      "step": 85423
    },
    {
      "epoch": 1.0893821716308594e-05,
      "step": 85423,
      "training_step_time": 7.927070140838623
    },
    {
      "epoch": 1.0894775390625e-05,
      "model_forward_time": 1.9209401607513428,
      "step": 85424
    },
    {
      "epoch": 1.0894775390625e-05,
      "step": 85424,
      "training_step_time": 8.325170755386353
    },
    {
      "epoch": 1.0895729064941407e-05,
      "model_forward_time": 1.9103882312774658,
      "step": 85425
    },
    {
      "epoch": 1.0895729064941407e-05,
      "step": 85425,
      "training_step_time": 9.104533195495605
    },
    {
      "epoch": 1.0896682739257813e-05,
      "model_forward_time": 2.551433563232422,
      "step": 85426
    },
    {
      "epoch": 1.0896682739257813e-05,
      "step": 85426,
      "training_step_time": 7.819510221481323
    },
    {
      "epoch": 1.0897636413574218e-05,
      "model_forward_time": 1.919600486755371,
      "step": 85427
    },
    {
      "epoch": 1.0897636413574218e-05,
      "step": 85427,
      "training_step_time": 8.469769477844238
    },
    {
      "epoch": 1.0898590087890626e-05,
      "model_forward_time": 1.9263081550598145,
      "step": 85428
    },
    {
      "epoch": 1.0898590087890626e-05,
      "step": 85428,
      "training_step_time": 7.6914756298065186
    },
    {
      "epoch": 1.0899543762207032e-05,
      "model_forward_time": 1.9144744873046875,
      "step": 85429
    },
    {
      "epoch": 1.0899543762207032e-05,
      "step": 85429,
      "training_step_time": 7.655510663986206
    },
    {
      "epoch": 1.0900497436523437e-05,
      "grad_norm": 0.11858724802732468,
      "learning_rate": 5.693141026097909e-07,
      "loss": 0.0528,
      "step": 85430
    },
    {
      "epoch": 1.0900497436523437e-05,
      "model_forward_time": 1.9139039516448975,
      "step": 85430
    },
    {
      "dynamics_loss_avg": 0.05704075191169977,
      "epoch": 1.0900497436523437e-05,
      "step": 85430
    },
    {
      "action_loss_avg": 0.0011534525692695752,
      "epoch": 1.0900497436523437e-05,
      "step": 85430
    },
    {
      "epoch": 1.0900497436523437e-05,
      "step": 85430,
      "training_step_time": 7.422897577285767
    },
    {
      "epoch": 1.0901451110839843e-05,
      "model_forward_time": 1.9195082187652588,
      "step": 85431
    },
    {
      "epoch": 1.0901451110839843e-05,
      "step": 85431,
      "training_step_time": 6.7796454429626465
    },
    {
      "epoch": 1.090240478515625e-05,
      "model_forward_time": 1.9115445613861084,
      "step": 85432
    },
    {
      "epoch": 1.090240478515625e-05,
      "step": 85432,
      "training_step_time": 7.675920009613037
    },
    {
      "epoch": 1.0903358459472656e-05,
      "model_forward_time": 1.9305739402770996,
      "step": 85433
    },
    {
      "epoch": 1.0903358459472656e-05,
      "step": 85433,
      "training_step_time": 8.69551944732666
    },
    {
      "epoch": 1.0904312133789062e-05,
      "model_forward_time": 1.8750040531158447,
      "step": 85434
    },
    {
      "epoch": 1.0904312133789062e-05,
      "step": 85434,
      "training_step_time": 7.801571369171143
    },
    {
      "epoch": 1.090526580810547e-05,
      "model_forward_time": 2.4825522899627686,
      "step": 85435
    },
    {
      "epoch": 1.090526580810547e-05,
      "step": 85435,
      "training_step_time": 8.827596187591553
    },
    {
      "epoch": 1.0906219482421875e-05,
      "model_forward_time": 1.910555124282837,
      "step": 85436
    },
    {
      "epoch": 1.0906219482421875e-05,
      "step": 85436,
      "training_step_time": 9.788145065307617
    },
    {
      "epoch": 1.090717315673828e-05,
      "model_forward_time": 1.9123191833496094,
      "step": 85437
    },
    {
      "epoch": 1.090717315673828e-05,
      "step": 85437,
      "training_step_time": 8.517207384109497
    },
    {
      "epoch": 1.0908126831054688e-05,
      "model_forward_time": 1.9078714847564697,
      "step": 85438
    },
    {
      "epoch": 1.0908126831054688e-05,
      "step": 85438,
      "training_step_time": 7.55188775062561
    },
    {
      "epoch": 1.0909080505371094e-05,
      "model_forward_time": 1.9130592346191406,
      "step": 85439
    },
    {
      "epoch": 1.0909080505371094e-05,
      "step": 85439,
      "training_step_time": 7.403038024902344
    },
    {
      "epoch": 1.09100341796875e-05,
      "grad_norm": 0.09659409523010254,
      "learning_rate": 5.685480889073775e-07,
      "loss": 0.054,
      "step": 85440
    },
    {
      "epoch": 1.09100341796875e-05,
      "model_forward_time": 1.900336503982544,
      "step": 85440
    },
    {
      "dynamics_loss_avg": 0.07166942674666643,
      "epoch": 1.09100341796875e-05,
      "step": 85440
    },
    {
      "action_loss_avg": 0.0011195169150596484,
      "epoch": 1.09100341796875e-05,
      "step": 85440
    },
    {
      "epoch": 1.09100341796875e-05,
      "step": 85440,
      "training_step_time": 6.777801275253296
    },
    {
      "epoch": 1.0910987854003907e-05,
      "model_forward_time": 1.885538101196289,
      "step": 85441
    },
    {
      "epoch": 1.0910987854003907e-05,
      "step": 85441,
      "training_step_time": 7.563359975814819
    },
    {
      "epoch": 1.0911941528320313e-05,
      "model_forward_time": 1.8971357345581055,
      "step": 85442
    },
    {
      "epoch": 1.0911941528320313e-05,
      "step": 85442,
      "training_step_time": 7.587149620056152
    },
    {
      "epoch": 1.0912895202636718e-05,
      "model_forward_time": 1.9172978401184082,
      "step": 85443
    },
    {
      "epoch": 1.0912895202636718e-05,
      "step": 85443,
      "training_step_time": 9.94165587425232
    },
    {
      "epoch": 1.0913848876953126e-05,
      "model_forward_time": 2.5105960369110107,
      "step": 85444
    },
    {
      "epoch": 1.0913848876953126e-05,
      "step": 85444,
      "training_step_time": 9.209516763687134
    },
    {
      "epoch": 1.0914802551269532e-05,
      "model_forward_time": 1.906031608581543,
      "step": 85445
    },
    {
      "epoch": 1.0914802551269532e-05,
      "step": 85445,
      "training_step_time": 7.824591159820557
    },
    {
      "epoch": 1.0915756225585937e-05,
      "model_forward_time": 1.889857530593872,
      "step": 85446
    },
    {
      "epoch": 1.0915756225585937e-05,
      "step": 85446,
      "training_step_time": 7.804452180862427
    },
    {
      "epoch": 1.0916709899902343e-05,
      "model_forward_time": 1.893341302871704,
      "step": 85447
    },
    {
      "epoch": 1.0916709899902343e-05,
      "step": 85447,
      "training_step_time": 8.320363521575928
    },
    {
      "epoch": 1.091766357421875e-05,
      "model_forward_time": 1.9045116901397705,
      "step": 85448
    },
    {
      "epoch": 1.091766357421875e-05,
      "step": 85448,
      "training_step_time": 7.374415636062622
    },
    {
      "epoch": 1.0918617248535156e-05,
      "model_forward_time": 1.9059031009674072,
      "step": 85449
    },
    {
      "epoch": 1.0918617248535156e-05,
      "step": 85449,
      "training_step_time": 7.45126748085022
    },
    {
      "epoch": 1.0919570922851562e-05,
      "grad_norm": 0.08286392688751221,
      "learning_rate": 5.677825598218822e-07,
      "loss": 0.0544,
      "step": 85450
    },
    {
      "epoch": 1.0919570922851562e-05,
      "model_forward_time": 1.8850135803222656,
      "step": 85450
    },
    {
      "dynamics_loss_avg": 0.06356622613966464,
      "epoch": 1.0919570922851562e-05,
      "step": 85450
    },
    {
      "action_loss_avg": 0.0012102252861950547,
      "epoch": 1.0919570922851562e-05,
      "step": 85450
    },
    {
      "epoch": 1.0919570922851562e-05,
      "step": 85450,
      "training_step_time": 7.694790840148926
    },
    {
      "epoch": 1.092052459716797e-05,
      "model_forward_time": 1.8930845260620117,
      "step": 85451
    },
    {
      "epoch": 1.092052459716797e-05,
      "step": 85451,
      "training_step_time": 7.739474534988403
    },
    {
      "epoch": 1.0921478271484375e-05,
      "model_forward_time": 1.8926599025726318,
      "step": 85452
    },
    {
      "epoch": 1.0921478271484375e-05,
      "step": 85452,
      "training_step_time": 7.690490961074829
    },
    {
      "epoch": 1.092243194580078e-05,
      "model_forward_time": 2.5165798664093018,
      "step": 85453
    },
    {
      "epoch": 1.092243194580078e-05,
      "step": 85453,
      "training_step_time": 9.404393196105957
    },
    {
      "epoch": 1.0923385620117188e-05,
      "model_forward_time": 1.8897480964660645,
      "step": 85454
    },
    {
      "epoch": 1.0923385620117188e-05,
      "step": 85454,
      "training_step_time": 7.815253973007202
    },
    {
      "epoch": 1.0924339294433594e-05,
      "model_forward_time": 1.9119393825531006,
      "step": 85455
    },
    {
      "epoch": 1.0924339294433594e-05,
      "step": 85455,
      "training_step_time": 8.635272026062012
    },
    {
      "epoch": 1.092529296875e-05,
      "model_forward_time": 1.9016900062561035,
      "step": 85456
    },
    {
      "epoch": 1.092529296875e-05,
      "step": 85456,
      "training_step_time": 7.6992881298065186
    },
    {
      "epoch": 1.0926246643066407e-05,
      "model_forward_time": 1.8979458808898926,
      "step": 85457
    },
    {
      "epoch": 1.0926246643066407e-05,
      "step": 85457,
      "training_step_time": 7.504194259643555
    },
    {
      "epoch": 1.0927200317382813e-05,
      "model_forward_time": 1.8999214172363281,
      "step": 85458
    },
    {
      "epoch": 1.0927200317382813e-05,
      "step": 85458,
      "training_step_time": 7.3286309242248535
    },
    {
      "epoch": 1.0928153991699218e-05,
      "model_forward_time": 1.8806517124176025,
      "step": 85459
    },
    {
      "epoch": 1.0928153991699218e-05,
      "step": 85459,
      "training_step_time": 7.540624618530273
    },
    {
      "epoch": 1.0929107666015626e-05,
      "grad_norm": 0.5550617575645447,
      "learning_rate": 5.670175154370216e-07,
      "loss": 0.0529,
      "step": 85460
    },
    {
      "epoch": 1.0929107666015626e-05,
      "model_forward_time": 1.9061188697814941,
      "step": 85460
    },
    {
      "dynamics_loss_avg": 0.06662319209426641,
      "epoch": 1.0929107666015626e-05,
      "step": 85460
    },
    {
      "action_loss_avg": 0.0019973037473391742,
      "epoch": 1.0929107666015626e-05,
      "step": 85460
    },
    {
      "epoch": 1.0929107666015626e-05,
      "step": 85460,
      "training_step_time": 8.204984188079834
    },
    {
      "epoch": 1.0930061340332032e-05,
      "model_forward_time": 1.9080650806427002,
      "step": 85461
    },
    {
      "epoch": 1.0930061340332032e-05,
      "step": 85461,
      "training_step_time": 7.573554277420044
    },
    {
      "epoch": 1.0931015014648437e-05,
      "model_forward_time": 1.9033467769622803,
      "step": 85462
    },
    {
      "epoch": 1.0931015014648437e-05,
      "step": 85462,
      "training_step_time": 10.192925214767456
    },
    {
      "epoch": 1.0931968688964843e-05,
      "model_forward_time": 1.9201741218566895,
      "step": 85463
    },
    {
      "epoch": 1.0931968688964843e-05,
      "step": 85463,
      "training_step_time": 8.739013195037842
    },
    {
      "epoch": 1.093292236328125e-05,
      "model_forward_time": 1.9051084518432617,
      "step": 85464
    },
    {
      "epoch": 1.093292236328125e-05,
      "step": 85464,
      "training_step_time": 7.8867645263671875
    },
    {
      "epoch": 1.0933876037597656e-05,
      "model_forward_time": 1.9222939014434814,
      "step": 85465
    },
    {
      "epoch": 1.0933876037597656e-05,
      "step": 85465,
      "training_step_time": 7.720456600189209
    },
    {
      "epoch": 1.0934829711914062e-05,
      "model_forward_time": 1.8905940055847168,
      "step": 85466
    },
    {
      "epoch": 1.0934829711914062e-05,
      "step": 85466,
      "training_step_time": 7.625272274017334
    },
    {
      "epoch": 1.093578338623047e-05,
      "model_forward_time": 1.9198534488677979,
      "step": 85467
    },
    {
      "epoch": 1.093578338623047e-05,
      "step": 85467,
      "training_step_time": 7.466118335723877
    },
    {
      "epoch": 1.0936737060546875e-05,
      "model_forward_time": 1.9002950191497803,
      "step": 85468
    },
    {
      "epoch": 1.0936737060546875e-05,
      "step": 85468,
      "training_step_time": 7.203987121582031
    },
    {
      "epoch": 1.093769073486328e-05,
      "model_forward_time": 1.9041945934295654,
      "step": 85469
    },
    {
      "epoch": 1.093769073486328e-05,
      "step": 85469,
      "training_step_time": 8.568309307098389
    },
    {
      "epoch": 1.0938644409179688e-05,
      "grad_norm": 0.08832938969135284,
      "learning_rate": 5.662529558364616e-07,
      "loss": 0.0532,
      "step": 85470
    },
    {
      "epoch": 1.0938644409179688e-05,
      "model_forward_time": 1.8974080085754395,
      "step": 85470
    },
    {
      "dynamics_loss_avg": 0.05868503451347351,
      "epoch": 1.0938644409179688e-05,
      "step": 85470
    },
    {
      "action_loss_avg": 0.0016941415262408555,
      "epoch": 1.0938644409179688e-05,
      "step": 85470
    },
    {
      "epoch": 1.0938644409179688e-05,
      "step": 85470,
      "training_step_time": 7.610191822052002
    },
    {
      "epoch": 1.0939598083496094e-05,
      "model_forward_time": 1.887150764465332,
      "step": 85471
    },
    {
      "epoch": 1.0939598083496094e-05,
      "step": 85471,
      "training_step_time": 8.392859935760498
    },
    {
      "epoch": 1.09405517578125e-05,
      "model_forward_time": 2.525076150894165,
      "step": 85472
    },
    {
      "epoch": 1.09405517578125e-05,
      "step": 85472,
      "training_step_time": 10.3238046169281
    },
    {
      "epoch": 1.0941505432128907e-05,
      "model_forward_time": 1.552182912826538,
      "step": 85473
    },
    {
      "epoch": 1.0941505432128907e-05,
      "step": 85473,
      "training_step_time": 7.869511604309082
    },
    {
      "epoch": 1.0942459106445313e-05,
      "model_forward_time": 1.9239706993103027,
      "step": 85474
    },
    {
      "epoch": 1.0942459106445313e-05,
      "step": 85474,
      "training_step_time": 7.677172899246216
    },
    {
      "epoch": 1.0943412780761718e-05,
      "model_forward_time": 1.9251675605773926,
      "step": 85475
    },
    {
      "epoch": 1.0943412780761718e-05,
      "step": 85475,
      "training_step_time": 7.865344524383545
    },
    {
      "epoch": 1.0944366455078126e-05,
      "model_forward_time": 1.9303526878356934,
      "step": 85476
    },
    {
      "epoch": 1.0944366455078126e-05,
      "step": 85476,
      "training_step_time": 7.5207037925720215
    },
    {
      "epoch": 1.0945320129394532e-05,
      "model_forward_time": 1.9337692260742188,
      "step": 85477
    },
    {
      "epoch": 1.0945320129394532e-05,
      "step": 85477,
      "training_step_time": 6.900094032287598
    },
    {
      "epoch": 1.0946273803710937e-05,
      "model_forward_time": 1.9266657829284668,
      "step": 85478
    },
    {
      "epoch": 1.0946273803710937e-05,
      "step": 85478,
      "training_step_time": 7.737391710281372
    },
    {
      "epoch": 1.0947227478027343e-05,
      "model_forward_time": 1.9270102977752686,
      "step": 85479
    },
    {
      "epoch": 1.0947227478027343e-05,
      "step": 85479,
      "training_step_time": 9.950767993927002
    },
    {
      "epoch": 1.094818115234375e-05,
      "grad_norm": 0.08383256942033768,
      "learning_rate": 5.654888811038128e-07,
      "loss": 0.0531,
      "step": 85480
    },
    {
      "epoch": 1.094818115234375e-05,
      "model_forward_time": 1.9051027297973633,
      "step": 85480
    },
    {
      "dynamics_loss_avg": 0.05399320442229509,
      "epoch": 1.094818115234375e-05,
      "step": 85480
    },
    {
      "action_loss_avg": 0.0010594682244118303,
      "epoch": 1.094818115234375e-05,
      "step": 85480
    },
    {
      "epoch": 1.094818115234375e-05,
      "step": 85480,
      "training_step_time": 8.562992334365845
    },
    {
      "epoch": 1.0949134826660156e-05,
      "model_forward_time": 2.5176961421966553,
      "step": 85481
    },
    {
      "epoch": 1.0949134826660156e-05,
      "step": 85481,
      "training_step_time": 9.39497709274292
    },
    {
      "epoch": 1.0950088500976562e-05,
      "model_forward_time": 1.927582025527954,
      "step": 85482
    },
    {
      "epoch": 1.0950088500976562e-05,
      "step": 85482,
      "training_step_time": 8.710664749145508
    },
    {
      "epoch": 1.095104217529297e-05,
      "model_forward_time": 1.9243485927581787,
      "step": 85483
    },
    {
      "epoch": 1.095104217529297e-05,
      "step": 85483,
      "training_step_time": 8.274171590805054
    },
    {
      "epoch": 1.0951995849609375e-05,
      "model_forward_time": 1.8998537063598633,
      "step": 85484
    },
    {
      "epoch": 1.0951995849609375e-05,
      "step": 85484,
      "training_step_time": 7.707256555557251
    },
    {
      "epoch": 1.095294952392578e-05,
      "model_forward_time": 1.9193496704101562,
      "step": 85485
    },
    {
      "epoch": 1.095294952392578e-05,
      "step": 85485,
      "training_step_time": 7.379396200180054
    },
    {
      "epoch": 1.0953903198242188e-05,
      "model_forward_time": 1.8996095657348633,
      "step": 85486
    },
    {
      "epoch": 1.0953903198242188e-05,
      "step": 85486,
      "training_step_time": 6.741832733154297
    },
    {
      "epoch": 1.0954856872558594e-05,
      "model_forward_time": 1.918017864227295,
      "step": 85487
    },
    {
      "epoch": 1.0954856872558594e-05,
      "step": 85487,
      "training_step_time": 7.657715082168579
    },
    {
      "epoch": 1.0955810546875e-05,
      "model_forward_time": 1.9309086799621582,
      "step": 85488
    },
    {
      "epoch": 1.0955810546875e-05,
      "step": 85488,
      "training_step_time": 7.5609259605407715
    },
    {
      "epoch": 1.0956764221191407e-05,
      "model_forward_time": 1.9270024299621582,
      "step": 85489
    },
    {
      "epoch": 1.0956764221191407e-05,
      "step": 85489,
      "training_step_time": 8.285833358764648
    },
    {
      "epoch": 1.0957717895507813e-05,
      "grad_norm": 0.08175131678581238,
      "learning_rate": 5.647252913226315e-07,
      "loss": 0.0538,
      "step": 85490
    },
    {
      "epoch": 1.0957717895507813e-05,
      "model_forward_time": 2.5715079307556152,
      "step": 85490
    },
    {
      "dynamics_loss_avg": 0.0675959998741746,
      "epoch": 1.0957717895507813e-05,
      "step": 85490
    },
    {
      "action_loss_avg": 0.0020014571899082513,
      "epoch": 1.0957717895507813e-05,
      "step": 85490
    },
    {
      "epoch": 1.0957717895507813e-05,
      "step": 85490,
      "training_step_time": 9.253788232803345
    },
    {
      "epoch": 1.0958671569824218e-05,
      "model_forward_time": 1.9302027225494385,
      "step": 85491
    },
    {
      "epoch": 1.0958671569824218e-05,
      "step": 85491,
      "training_step_time": 8.409547328948975
    },
    {
      "epoch": 1.0959625244140626e-05,
      "model_forward_time": 1.9176692962646484,
      "step": 85492
    },
    {
      "epoch": 1.0959625244140626e-05,
      "step": 85492,
      "training_step_time": 7.674653053283691
    },
    {
      "epoch": 1.0960578918457032e-05,
      "model_forward_time": 1.902207851409912,
      "step": 85493
    },
    {
      "epoch": 1.0960578918457032e-05,
      "step": 85493,
      "training_step_time": 7.681448221206665
    },
    {
      "epoch": 1.0961532592773437e-05,
      "model_forward_time": 1.9234094619750977,
      "step": 85494
    },
    {
      "epoch": 1.0961532592773437e-05,
      "step": 85494,
      "training_step_time": 7.350227355957031
    },
    {
      "epoch": 1.0962486267089843e-05,
      "model_forward_time": 1.919628620147705,
      "step": 85495
    },
    {
      "epoch": 1.0962486267089843e-05,
      "step": 85495,
      "training_step_time": 6.71911883354187
    },
    {
      "epoch": 1.096343994140625e-05,
      "model_forward_time": 1.9141242504119873,
      "step": 85496
    },
    {
      "epoch": 1.096343994140625e-05,
      "step": 85496,
      "training_step_time": 7.6275036334991455
    },
    {
      "epoch": 1.0964393615722656e-05,
      "model_forward_time": 1.9268453121185303,
      "step": 85497
    },
    {
      "epoch": 1.0964393615722656e-05,
      "step": 85497,
      "training_step_time": 7.572962045669556
    },
    {
      "epoch": 1.0965347290039062e-05,
      "model_forward_time": 1.9111850261688232,
      "step": 85498
    },
    {
      "epoch": 1.0965347290039062e-05,
      "step": 85498,
      "training_step_time": 8.66539454460144
    },
    {
      "epoch": 1.096630096435547e-05,
      "model_forward_time": 2.5598769187927246,
      "step": 85499
    },
    {
      "epoch": 1.096630096435547e-05,
      "step": 85499,
      "training_step_time": 9.866562366485596
    },
    {
      "epoch": 1.0967254638671875e-05,
      "grad_norm": 0.0822644978761673,
      "learning_rate": 5.639621865764256e-07,
      "loss": 0.0537,
      "step": 85500
    },
    {
      "epoch": 1.0967254638671875e-05,
      "model_forward_time": 1.9341506958007812,
      "step": 85500
    },
    {
      "dynamics_loss_avg": 0.059554281644523144,
      "epoch": 1.0967254638671875e-05,
      "step": 85500
    },
    {
      "action_loss_avg": 0.001687497747479938,
      "epoch": 1.0967254638671875e-05,
      "step": 85500
    },
    {
      "epoch": 1.0967254638671875e-05,
      "step": 85500,
      "training_step_time": 9.672409057617188
    },
    {
      "epoch": 1.096820831298828e-05,
      "model_forward_time": 1.932654857635498,
      "step": 85501
    },
    {
      "epoch": 1.096820831298828e-05,
      "step": 85501,
      "training_step_time": 8.507922410964966
    },
    {
      "epoch": 1.0969161987304688e-05,
      "model_forward_time": 1.9203848838806152,
      "step": 85502
    },
    {
      "epoch": 1.0969161987304688e-05,
      "step": 85502,
      "training_step_time": 8.700778007507324
    },
    {
      "epoch": 1.0970115661621094e-05,
      "model_forward_time": 1.9266905784606934,
      "step": 85503
    },
    {
      "epoch": 1.0970115661621094e-05,
      "step": 85503,
      "training_step_time": 7.9287145137786865
    },
    {
      "epoch": 1.09710693359375e-05,
      "model_forward_time": 1.9217777252197266,
      "step": 85504
    },
    {
      "epoch": 1.09710693359375e-05,
      "step": 85504,
      "training_step_time": 7.258047342300415
    },
    {
      "epoch": 1.0972023010253907e-05,
      "model_forward_time": 1.9194676876068115,
      "step": 85505
    },
    {
      "epoch": 1.0972023010253907e-05,
      "step": 85505,
      "training_step_time": 7.615285634994507
    },
    {
      "epoch": 1.0972976684570313e-05,
      "model_forward_time": 1.9311397075653076,
      "step": 85506
    },
    {
      "epoch": 1.0972976684570313e-05,
      "step": 85506,
      "training_step_time": 7.96076226234436
    },
    {
      "epoch": 1.0973930358886718e-05,
      "model_forward_time": 1.9253077507019043,
      "step": 85507
    },
    {
      "epoch": 1.0973930358886718e-05,
      "step": 85507,
      "training_step_time": 8.381795167922974
    },
    {
      "epoch": 1.0974884033203124e-05,
      "model_forward_time": 1.9122459888458252,
      "step": 85508
    },
    {
      "epoch": 1.0974884033203124e-05,
      "step": 85508,
      "training_step_time": 10.50975751876831
    },
    {
      "epoch": 1.0975837707519532e-05,
      "model_forward_time": 2.5593111515045166,
      "step": 85509
    },
    {
      "epoch": 1.0975837707519532e-05,
      "step": 85509,
      "training_step_time": 8.664458274841309
    },
    {
      "epoch": 1.0976791381835937e-05,
      "grad_norm": 0.07032298296689987,
      "learning_rate": 5.63199566948644e-07,
      "loss": 0.0546,
      "step": 85510
    },
    {
      "epoch": 1.0976791381835937e-05,
      "model_forward_time": 1.917236566543579,
      "step": 85510
    },
    {
      "dynamics_loss_avg": 0.05816456638276577,
      "epoch": 1.0976791381835937e-05,
      "step": 85510
    },
    {
      "action_loss_avg": 0.0019814424798823894,
      "epoch": 1.0976791381835937e-05,
      "step": 85510
    },
    {
      "epoch": 1.0976791381835937e-05,
      "step": 85510,
      "training_step_time": 7.774935722351074
    },
    {
      "epoch": 1.0977745056152343e-05,
      "model_forward_time": 1.9083070755004883,
      "step": 85511
    },
    {
      "epoch": 1.0977745056152343e-05,
      "step": 85511,
      "training_step_time": 7.677459001541138
    },
    {
      "epoch": 1.097869873046875e-05,
      "model_forward_time": 1.923903465270996,
      "step": 85512
    },
    {
      "epoch": 1.097869873046875e-05,
      "step": 85512,
      "training_step_time": 7.441656112670898
    },
    {
      "epoch": 1.0979652404785156e-05,
      "model_forward_time": 1.9179039001464844,
      "step": 85513
    },
    {
      "epoch": 1.0979652404785156e-05,
      "step": 85513,
      "training_step_time": 7.322230815887451
    },
    {
      "epoch": 1.0980606079101562e-05,
      "model_forward_time": 1.9175593852996826,
      "step": 85514
    },
    {
      "epoch": 1.0980606079101562e-05,
      "step": 85514,
      "training_step_time": 8.368197202682495
    },
    {
      "epoch": 1.098155975341797e-05,
      "model_forward_time": 1.9212212562561035,
      "step": 85515
    },
    {
      "epoch": 1.098155975341797e-05,
      "step": 85515,
      "training_step_time": 8.477592706680298
    },
    {
      "epoch": 1.0982513427734375e-05,
      "model_forward_time": 1.9286210536956787,
      "step": 85516
    },
    {
      "epoch": 1.0982513427734375e-05,
      "step": 85516,
      "training_step_time": 8.184724569320679
    },
    {
      "epoch": 1.098346710205078e-05,
      "model_forward_time": 1.9190332889556885,
      "step": 85517
    },
    {
      "epoch": 1.098346710205078e-05,
      "step": 85517,
      "training_step_time": 9.331724643707275
    },
    {
      "epoch": 1.0984420776367188e-05,
      "model_forward_time": 2.5402841567993164,
      "step": 85518
    },
    {
      "epoch": 1.0984420776367188e-05,
      "step": 85518,
      "training_step_time": 8.656845092773438
    },
    {
      "epoch": 1.0985374450683594e-05,
      "model_forward_time": 1.9064652919769287,
      "step": 85519
    },
    {
      "epoch": 1.0985374450683594e-05,
      "step": 85519,
      "training_step_time": 7.7728612422943115
    },
    {
      "epoch": 1.0986328125e-05,
      "grad_norm": 0.10001389682292938,
      "learning_rate": 5.624374325226883e-07,
      "loss": 0.0557,
      "step": 85520
    },
    {
      "epoch": 1.0986328125e-05,
      "model_forward_time": 1.9174530506134033,
      "step": 85520
    },
    {
      "dynamics_loss_avg": 0.06285743582993746,
      "epoch": 1.0986328125e-05,
      "step": 85520
    },
    {
      "action_loss_avg": 0.0013616197742521764,
      "epoch": 1.0986328125e-05,
      "step": 85520
    },
    {
      "epoch": 1.0986328125e-05,
      "step": 85520,
      "training_step_time": 8.405281066894531
    },
    {
      "epoch": 1.0987281799316407e-05,
      "model_forward_time": 1.9287807941436768,
      "step": 85521
    },
    {
      "epoch": 1.0987281799316407e-05,
      "step": 85521,
      "training_step_time": 7.326935291290283
    },
    {
      "epoch": 1.0988235473632813e-05,
      "model_forward_time": 1.9000368118286133,
      "step": 85522
    },
    {
      "epoch": 1.0988235473632813e-05,
      "step": 85522,
      "training_step_time": 7.339157581329346
    },
    {
      "epoch": 1.0989189147949218e-05,
      "model_forward_time": 1.924546480178833,
      "step": 85523
    },
    {
      "epoch": 1.0989189147949218e-05,
      "step": 85523,
      "training_step_time": 7.7251434326171875
    },
    {
      "epoch": 1.0990142822265624e-05,
      "model_forward_time": 1.9246742725372314,
      "step": 85524
    },
    {
      "epoch": 1.0990142822265624e-05,
      "step": 85524,
      "training_step_time": 7.427531719207764
    },
    {
      "epoch": 1.0991096496582032e-05,
      "model_forward_time": 1.920161247253418,
      "step": 85525
    },
    {
      "epoch": 1.0991096496582032e-05,
      "step": 85525,
      "training_step_time": 7.4325315952301025
    },
    {
      "epoch": 1.0992050170898437e-05,
      "model_forward_time": 1.9117259979248047,
      "step": 85526
    },
    {
      "epoch": 1.0992050170898437e-05,
      "step": 85526,
      "training_step_time": 7.777982234954834
    },
    {
      "epoch": 1.0993003845214843e-05,
      "model_forward_time": 2.467301845550537,
      "step": 85527
    },
    {
      "epoch": 1.0993003845214843e-05,
      "step": 85527,
      "training_step_time": 8.573653221130371
    },
    {
      "epoch": 1.099395751953125e-05,
      "model_forward_time": 1.9357941150665283,
      "step": 85528
    },
    {
      "epoch": 1.099395751953125e-05,
      "step": 85528,
      "training_step_time": 7.805958271026611
    },
    {
      "epoch": 1.0994911193847656e-05,
      "model_forward_time": 1.9299020767211914,
      "step": 85529
    },
    {
      "epoch": 1.0994911193847656e-05,
      "step": 85529,
      "training_step_time": 7.77742862701416
    },
    {
      "epoch": 1.0995864868164062e-05,
      "grad_norm": 0.11741868406534195,
      "learning_rate": 5.616757833819009e-07,
      "loss": 0.0557,
      "step": 85530
    },
    {
      "epoch": 1.0995864868164062e-05,
      "model_forward_time": 1.931910514831543,
      "step": 85530
    },
    {
      "dynamics_loss_avg": 0.06264272183179856,
      "epoch": 1.0995864868164062e-05,
      "step": 85530
    },
    {
      "action_loss_avg": 0.0013263569737318904,
      "epoch": 1.0995864868164062e-05,
      "step": 85530
    },
    {
      "epoch": 1.0995864868164062e-05,
      "step": 85530,
      "training_step_time": 7.738718748092651
    },
    {
      "epoch": 1.099681854248047e-05,
      "model_forward_time": 1.9284625053405762,
      "step": 85531
    },
    {
      "epoch": 1.099681854248047e-05,
      "step": 85531,
      "training_step_time": 8.379919528961182
    },
    {
      "epoch": 1.0997772216796875e-05,
      "model_forward_time": 1.9224555492401123,
      "step": 85532
    },
    {
      "epoch": 1.0997772216796875e-05,
      "step": 85532,
      "training_step_time": 7.706707954406738
    },
    {
      "epoch": 1.099872589111328e-05,
      "model_forward_time": 1.9085469245910645,
      "step": 85533
    },
    {
      "epoch": 1.099872589111328e-05,
      "step": 85533,
      "training_step_time": 7.417295217514038
    },
    {
      "epoch": 1.0999679565429688e-05,
      "model_forward_time": 1.9076037406921387,
      "step": 85534
    },
    {
      "epoch": 1.0999679565429688e-05,
      "step": 85534,
      "training_step_time": 7.659420728683472
    },
    {
      "epoch": 1.1000633239746094e-05,
      "model_forward_time": 1.921440839767456,
      "step": 85535
    },
    {
      "epoch": 1.1000633239746094e-05,
      "step": 85535,
      "training_step_time": 9.255898237228394
    },
    {
      "epoch": 1.10015869140625e-05,
      "model_forward_time": 2.5703115463256836,
      "step": 85536
    },
    {
      "epoch": 1.10015869140625e-05,
      "step": 85536,
      "training_step_time": 8.578895330429077
    },
    {
      "epoch": 1.1002540588378907e-05,
      "model_forward_time": 1.931549310684204,
      "step": 85537
    },
    {
      "epoch": 1.1002540588378907e-05,
      "step": 85537,
      "training_step_time": 10.302755117416382
    },
    {
      "epoch": 1.1003494262695313e-05,
      "model_forward_time": 1.919224500656128,
      "step": 85538
    },
    {
      "epoch": 1.1003494262695313e-05,
      "step": 85538,
      "training_step_time": 8.592101573944092
    },
    {
      "epoch": 1.1004447937011718e-05,
      "model_forward_time": 1.9102482795715332,
      "step": 85539
    },
    {
      "epoch": 1.1004447937011718e-05,
      "step": 85539,
      "training_step_time": 7.697784423828125
    },
    {
      "epoch": 1.1005401611328124e-05,
      "grad_norm": 0.10645788908004761,
      "learning_rate": 5.609146196095794e-07,
      "loss": 0.0542,
      "step": 85540
    },
    {
      "epoch": 1.1005401611328124e-05,
      "model_forward_time": 1.7706832885742188,
      "step": 85540
    },
    {
      "dynamics_loss_avg": 0.055574226193130016,
      "epoch": 1.1005401611328124e-05,
      "step": 85540
    },
    {
      "action_loss_avg": 0.0015421856602188199,
      "epoch": 1.1005401611328124e-05,
      "step": 85540
    },
    {
      "epoch": 1.1005401611328124e-05,
      "step": 85540,
      "training_step_time": 7.255218029022217
    },
    {
      "epoch": 1.1006355285644532e-05,
      "model_forward_time": 1.7696428298950195,
      "step": 85541
    },
    {
      "epoch": 1.1006355285644532e-05,
      "step": 85541,
      "training_step_time": 7.653628587722778
    },
    {
      "epoch": 1.1007308959960937e-05,
      "model_forward_time": 1.762157678604126,
      "step": 85542
    },
    {
      "epoch": 1.1007308959960937e-05,
      "step": 85542,
      "training_step_time": 7.367555618286133
    },
    {
      "epoch": 1.1008262634277343e-05,
      "model_forward_time": 1.7718143463134766,
      "step": 85543
    },
    {
      "epoch": 1.1008262634277343e-05,
      "step": 85543,
      "training_step_time": 7.620176315307617
    },
    {
      "epoch": 1.100921630859375e-05,
      "model_forward_time": 1.7669305801391602,
      "step": 85544
    },
    {
      "epoch": 1.100921630859375e-05,
      "step": 85544,
      "training_step_time": 7.652771472930908
    },
    {
      "epoch": 1.1010169982910156e-05,
      "model_forward_time": 2.4238617420196533,
      "step": 85545
    },
    {
      "epoch": 1.1010169982910156e-05,
      "step": 85545,
      "training_step_time": 9.013251066207886
    },
    {
      "epoch": 1.1011123657226562e-05,
      "model_forward_time": 1.7579498291015625,
      "step": 85546
    },
    {
      "epoch": 1.1011123657226562e-05,
      "step": 85546,
      "training_step_time": 7.637441635131836
    },
    {
      "epoch": 1.101207733154297e-05,
      "model_forward_time": 1.7651095390319824,
      "step": 85547
    },
    {
      "epoch": 1.101207733154297e-05,
      "step": 85547,
      "training_step_time": 9.340500593185425
    },
    {
      "epoch": 1.1013031005859375e-05,
      "model_forward_time": 1.767937183380127,
      "step": 85548
    },
    {
      "epoch": 1.1013031005859375e-05,
      "step": 85548,
      "training_step_time": 7.475496530532837
    },
    {
      "epoch": 1.101398468017578e-05,
      "model_forward_time": 1.7625737190246582,
      "step": 85549
    },
    {
      "epoch": 1.101398468017578e-05,
      "step": 85549,
      "training_step_time": 7.295352458953857
    },
    {
      "epoch": 1.1014938354492188e-05,
      "grad_norm": 0.10427336394786835,
      "learning_rate": 5.6015394128896e-07,
      "loss": 0.055,
      "step": 85550
    },
    {
      "epoch": 1.1014938354492188e-05,
      "model_forward_time": 1.7687079906463623,
      "step": 85550
    },
    {
      "dynamics_loss_avg": 0.05516853928565979,
      "epoch": 1.1014938354492188e-05,
      "step": 85550
    },
    {
      "action_loss_avg": 0.0008652951830299572,
      "epoch": 1.1014938354492188e-05,
      "step": 85550
    },
    {
      "epoch": 1.1014938354492188e-05,
      "step": 85550,
      "training_step_time": 7.601722240447998
    },
    {
      "epoch": 1.1015892028808594e-05,
      "model_forward_time": 1.7703711986541748,
      "step": 85551
    },
    {
      "epoch": 1.1015892028808594e-05,
      "step": 85551,
      "training_step_time": 7.368772983551025
    },
    {
      "epoch": 1.1016845703125e-05,
      "model_forward_time": 1.7714695930480957,
      "step": 85552
    },
    {
      "epoch": 1.1016845703125e-05,
      "step": 85552,
      "training_step_time": 7.502286672592163
    },
    {
      "epoch": 1.1017799377441407e-05,
      "model_forward_time": 1.7589194774627686,
      "step": 85553
    },
    {
      "epoch": 1.1017799377441407e-05,
      "step": 85553,
      "training_step_time": 9.637587547302246
    },
    {
      "epoch": 1.1018753051757813e-05,
      "model_forward_time": 1.7656373977661133,
      "step": 85554
    },
    {
      "epoch": 1.1018753051757813e-05,
      "step": 85554,
      "training_step_time": 8.945889711380005
    },
    {
      "epoch": 1.1019706726074218e-05,
      "model_forward_time": 1.7806804180145264,
      "step": 85555
    },
    {
      "epoch": 1.1019706726074218e-05,
      "step": 85555,
      "training_step_time": 8.51528000831604
    },
    {
      "epoch": 1.1020660400390624e-05,
      "model_forward_time": 1.7621588706970215,
      "step": 85556
    },
    {
      "epoch": 1.1020660400390624e-05,
      "step": 85556,
      "training_step_time": 8.658798933029175
    },
    {
      "epoch": 1.1021614074707032e-05,
      "model_forward_time": 1.7775371074676514,
      "step": 85557
    },
    {
      "epoch": 1.1021614074707032e-05,
      "step": 85557,
      "training_step_time": 7.7059924602508545
    },
    {
      "epoch": 1.1022567749023437e-05,
      "model_forward_time": 1.7929329872131348,
      "step": 85558
    },
    {
      "epoch": 1.1022567749023437e-05,
      "step": 85558,
      "training_step_time": 7.378973484039307
    },
    {
      "epoch": 1.1023521423339843e-05,
      "model_forward_time": 1.7794218063354492,
      "step": 85559
    },
    {
      "epoch": 1.1023521423339843e-05,
      "step": 85559,
      "training_step_time": 7.3492043018341064
    },
    {
      "epoch": 1.102447509765625e-05,
      "grad_norm": 0.08183462917804718,
      "learning_rate": 5.593937485032308e-07,
      "loss": 0.0558,
      "step": 85560
    },
    {
      "epoch": 1.102447509765625e-05,
      "model_forward_time": 1.776317834854126,
      "step": 85560
    },
    {
      "dynamics_loss_avg": 0.05555870048701763,
      "epoch": 1.102447509765625e-05,
      "step": 85560
    },
    {
      "action_loss_avg": 0.001533323444891721,
      "epoch": 1.102447509765625e-05,
      "step": 85560
    },
    {
      "epoch": 1.102447509765625e-05,
      "step": 85560,
      "training_step_time": 7.413999319076538
    },
    {
      "epoch": 1.1025428771972656e-05,
      "model_forward_time": 1.7806990146636963,
      "step": 85561
    },
    {
      "epoch": 1.1025428771972656e-05,
      "step": 85561,
      "training_step_time": 6.982357740402222
    },
    {
      "epoch": 1.1026382446289062e-05,
      "model_forward_time": 1.8009593486785889,
      "step": 85562
    },
    {
      "epoch": 1.1026382446289062e-05,
      "step": 85562,
      "training_step_time": 8.335252285003662
    },
    {
      "epoch": 1.102733612060547e-05,
      "model_forward_time": 1.7876501083374023,
      "step": 85563
    },
    {
      "epoch": 1.102733612060547e-05,
      "step": 85563,
      "training_step_time": 8.967746257781982
    },
    {
      "epoch": 1.1028289794921875e-05,
      "model_forward_time": 2.503995656967163,
      "step": 85564
    },
    {
      "epoch": 1.1028289794921875e-05,
      "step": 85564,
      "training_step_time": 9.407865047454834
    },
    {
      "epoch": 1.102924346923828e-05,
      "model_forward_time": 1.7805383205413818,
      "step": 85565
    },
    {
      "epoch": 1.102924346923828e-05,
      "step": 85565,
      "training_step_time": 8.375674724578857
    },
    {
      "epoch": 1.1030197143554688e-05,
      "model_forward_time": 1.7888827323913574,
      "step": 85566
    },
    {
      "epoch": 1.1030197143554688e-05,
      "step": 85566,
      "training_step_time": 7.624116897583008
    },
    {
      "epoch": 1.1031150817871094e-05,
      "model_forward_time": 1.7833888530731201,
      "step": 85567
    },
    {
      "epoch": 1.1031150817871094e-05,
      "step": 85567,
      "training_step_time": 7.4499828815460205
    },
    {
      "epoch": 1.10321044921875e-05,
      "model_forward_time": 1.778956413269043,
      "step": 85568
    },
    {
      "epoch": 1.10321044921875e-05,
      "step": 85568,
      "training_step_time": 7.497427225112915
    },
    {
      "epoch": 1.1033058166503907e-05,
      "model_forward_time": 1.7630374431610107,
      "step": 85569
    },
    {
      "epoch": 1.1033058166503907e-05,
      "step": 85569,
      "training_step_time": 7.357592344284058
    },
    {
      "epoch": 1.1034011840820313e-05,
      "grad_norm": 0.09638990461826324,
      "learning_rate": 5.586340413355245e-07,
      "loss": 0.0558,
      "step": 85570
    },
    {
      "epoch": 1.1034011840820313e-05,
      "model_forward_time": 1.7716064453125,
      "step": 85570
    },
    {
      "dynamics_loss_avg": 0.05197828309610486,
      "epoch": 1.1034011840820313e-05,
      "step": 85570
    },
    {
      "action_loss_avg": 0.0015976098307874053,
      "epoch": 1.1034011840820313e-05,
      "step": 85570
    },
    {
      "epoch": 1.1034011840820313e-05,
      "step": 85570,
      "training_step_time": 6.9691386222839355
    },
    {
      "epoch": 1.1034965515136718e-05,
      "model_forward_time": 1.7859728336334229,
      "step": 85571
    },
    {
      "epoch": 1.1034965515136718e-05,
      "step": 85571,
      "training_step_time": 7.641186475753784
    },
    {
      "epoch": 1.1035919189453124e-05,
      "model_forward_time": 1.7853643894195557,
      "step": 85572
    },
    {
      "epoch": 1.1035919189453124e-05,
      "step": 85572,
      "training_step_time": 8.65244460105896
    },
    {
      "epoch": 1.1036872863769532e-05,
      "model_forward_time": 2.484282970428467,
      "step": 85573
    },
    {
      "epoch": 1.1036872863769532e-05,
      "step": 85573,
      "training_step_time": 7.730880260467529
    },
    {
      "epoch": 1.1037826538085937e-05,
      "model_forward_time": 1.7830302715301514,
      "step": 85574
    },
    {
      "epoch": 1.1037826538085937e-05,
      "step": 85574,
      "training_step_time": 8.355324506759644
    },
    {
      "epoch": 1.1038780212402343e-05,
      "model_forward_time": 1.7859904766082764,
      "step": 85575
    },
    {
      "epoch": 1.1038780212402343e-05,
      "step": 85575,
      "training_step_time": 8.506550073623657
    },
    {
      "epoch": 1.103973388671875e-05,
      "model_forward_time": 1.7854032516479492,
      "step": 85576
    },
    {
      "epoch": 1.103973388671875e-05,
      "step": 85576,
      "training_step_time": 7.641598701477051
    },
    {
      "epoch": 1.1040687561035156e-05,
      "model_forward_time": 1.7834007740020752,
      "step": 85577
    },
    {
      "epoch": 1.1040687561035156e-05,
      "step": 85577,
      "training_step_time": 7.6341392993927
    },
    {
      "epoch": 1.1041641235351562e-05,
      "model_forward_time": 1.7864818572998047,
      "step": 85578
    },
    {
      "epoch": 1.1041641235351562e-05,
      "step": 85578,
      "training_step_time": 8.13556432723999
    },
    {
      "epoch": 1.104259490966797e-05,
      "model_forward_time": 1.7813646793365479,
      "step": 85579
    },
    {
      "epoch": 1.104259490966797e-05,
      "step": 85579,
      "training_step_time": 7.309326171875
    },
    {
      "epoch": 1.1043548583984375e-05,
      "grad_norm": 0.09491577744483948,
      "learning_rate": 5.578748198689226e-07,
      "loss": 0.0544,
      "step": 85580
    },
    {
      "epoch": 1.1043548583984375e-05,
      "model_forward_time": 1.7762808799743652,
      "step": 85580
    },
    {
      "dynamics_loss_avg": 0.04605762008577585,
      "epoch": 1.1043548583984375e-05,
      "step": 85580
    },
    {
      "action_loss_avg": 0.001197307778056711,
      "epoch": 1.1043548583984375e-05,
      "step": 85580
    },
    {
      "epoch": 1.1043548583984375e-05,
      "step": 85580,
      "training_step_time": 7.5570456981658936
    },
    {
      "epoch": 1.104450225830078e-05,
      "model_forward_time": 1.7926833629608154,
      "step": 85581
    },
    {
      "epoch": 1.104450225830078e-05,
      "step": 85581,
      "training_step_time": 7.630060911178589
    },
    {
      "epoch": 1.1045455932617188e-05,
      "model_forward_time": 2.470233917236328,
      "step": 85582
    },
    {
      "epoch": 1.1045455932617188e-05,
      "step": 85582,
      "training_step_time": 8.803919076919556
    },
    {
      "epoch": 1.1046409606933594e-05,
      "model_forward_time": 1.7813663482666016,
      "step": 85583
    },
    {
      "epoch": 1.1046409606933594e-05,
      "step": 85583,
      "training_step_time": 10.329562187194824
    },
    {
      "epoch": 1.104736328125e-05,
      "model_forward_time": 1.7911155223846436,
      "step": 85584
    },
    {
      "epoch": 1.104736328125e-05,
      "step": 85584,
      "training_step_time": 9.321827411651611
    },
    {
      "epoch": 1.1048316955566407e-05,
      "model_forward_time": 1.7819254398345947,
      "step": 85585
    },
    {
      "epoch": 1.1048316955566407e-05,
      "step": 85585,
      "training_step_time": 7.707035064697266
    },
    {
      "epoch": 1.1049270629882813e-05,
      "model_forward_time": 1.7876651287078857,
      "step": 85586
    },
    {
      "epoch": 1.1049270629882813e-05,
      "step": 85586,
      "training_step_time": 7.226354598999023
    },
    {
      "epoch": 1.1050224304199219e-05,
      "model_forward_time": 1.7807233333587646,
      "step": 85587
    },
    {
      "epoch": 1.1050224304199219e-05,
      "step": 85587,
      "training_step_time": 7.5562403202056885
    },
    {
      "epoch": 1.1051177978515624e-05,
      "model_forward_time": 1.8030831813812256,
      "step": 85588
    },
    {
      "epoch": 1.1051177978515624e-05,
      "step": 85588,
      "training_step_time": 7.331123352050781
    },
    {
      "epoch": 1.1052131652832032e-05,
      "model_forward_time": 1.7832419872283936,
      "step": 85589
    },
    {
      "epoch": 1.1052131652832032e-05,
      "step": 85589,
      "training_step_time": 7.506325960159302
    },
    {
      "epoch": 1.1053085327148437e-05,
      "grad_norm": 0.08116275072097778,
      "learning_rate": 5.571160841864503e-07,
      "loss": 0.055,
      "step": 85590
    },
    {
      "epoch": 1.1053085327148437e-05,
      "model_forward_time": 1.7737526893615723,
      "step": 85590
    },
    {
      "dynamics_loss_avg": 0.05383908879011869,
      "epoch": 1.1053085327148437e-05,
      "step": 85590
    },
    {
      "action_loss_avg": 0.001394826389150694,
      "epoch": 1.1053085327148437e-05,
      "step": 85590
    },
    {
      "epoch": 1.1053085327148437e-05,
      "step": 85590,
      "training_step_time": 8.534810304641724
    },
    {
      "epoch": 1.1054039001464843e-05,
      "model_forward_time": 2.441138982772827,
      "step": 85591
    },
    {
      "epoch": 1.1054039001464843e-05,
      "step": 85591,
      "training_step_time": 9.749160289764404
    },
    {
      "epoch": 1.105499267578125e-05,
      "model_forward_time": 1.7804303169250488,
      "step": 85592
    },
    {
      "epoch": 1.105499267578125e-05,
      "step": 85592,
      "training_step_time": 9.030060052871704
    },
    {
      "epoch": 1.1055946350097656e-05,
      "model_forward_time": 1.776299238204956,
      "step": 85593
    },
    {
      "epoch": 1.1055946350097656e-05,
      "step": 85593,
      "training_step_time": 7.773558855056763
    },
    {
      "epoch": 1.1056900024414062e-05,
      "model_forward_time": 1.781797170639038,
      "step": 85594
    },
    {
      "epoch": 1.1056900024414062e-05,
      "step": 85594,
      "training_step_time": 8.686645984649658
    },
    {
      "epoch": 1.105785369873047e-05,
      "model_forward_time": 1.776012897491455,
      "step": 85595
    },
    {
      "epoch": 1.105785369873047e-05,
      "step": 85595,
      "training_step_time": 7.275661945343018
    },
    {
      "epoch": 1.1058807373046875e-05,
      "model_forward_time": 1.7526917457580566,
      "step": 85596
    },
    {
      "epoch": 1.1058807373046875e-05,
      "step": 85596,
      "training_step_time": 7.590240478515625
    },
    {
      "epoch": 1.105976104736328e-05,
      "model_forward_time": 1.774094820022583,
      "step": 85597
    },
    {
      "epoch": 1.105976104736328e-05,
      "step": 85597,
      "training_step_time": 7.379862308502197
    },
    {
      "epoch": 1.1060714721679688e-05,
      "model_forward_time": 1.7739861011505127,
      "step": 85598
    },
    {
      "epoch": 1.1060714721679688e-05,
      "step": 85598,
      "training_step_time": 7.761068820953369
    },
    {
      "epoch": 1.1061668395996094e-05,
      "model_forward_time": 1.7878837585449219,
      "step": 85599
    },
    {
      "epoch": 1.1061668395996094e-05,
      "step": 85599,
      "training_step_time": 8.790624618530273
    },
    {
      "epoch": 1.10626220703125e-05,
      "grad_norm": 0.09088580310344696,
      "learning_rate": 5.563578343710846e-07,
      "loss": 0.0537,
      "step": 85600
    },
    {
      "epoch": 1.10626220703125e-05,
      "model_forward_time": 2.409600019454956,
      "step": 85600
    },
    {
      "dynamics_loss_avg": 0.0654434559866786,
      "epoch": 1.10626220703125e-05,
      "step": 85600
    },
    {
      "action_loss_avg": 0.0017529870325233786,
      "epoch": 1.10626220703125e-05,
      "step": 85600
    },
    {
      "epoch": 1.10626220703125e-05,
      "step": 85600,
      "training_step_time": 9.406221628189087
    },
    {
      "epoch": 1.1063575744628907e-05,
      "model_forward_time": 1.7868821620941162,
      "step": 85601
    },
    {
      "epoch": 1.1063575744628907e-05,
      "step": 85601,
      "training_step_time": 9.91506552696228
    },
    {
      "epoch": 1.1064529418945313e-05,
      "model_forward_time": 1.783982515335083,
      "step": 85602
    },
    {
      "epoch": 1.1064529418945313e-05,
      "step": 85602,
      "training_step_time": 8.549359798431396
    },
    {
      "epoch": 1.1065483093261719e-05,
      "model_forward_time": 1.7851803302764893,
      "step": 85603
    },
    {
      "epoch": 1.1065483093261719e-05,
      "step": 85603,
      "training_step_time": 7.626217603683472
    },
    {
      "epoch": 1.1066436767578124e-05,
      "model_forward_time": 1.772447109222412,
      "step": 85604
    },
    {
      "epoch": 1.1066436767578124e-05,
      "step": 85604,
      "training_step_time": 7.484029769897461
    },
    {
      "epoch": 1.1067390441894532e-05,
      "model_forward_time": 1.771514654159546,
      "step": 85605
    },
    {
      "epoch": 1.1067390441894532e-05,
      "step": 85605,
      "training_step_time": 8.683972120285034
    },
    {
      "epoch": 1.1068344116210937e-05,
      "model_forward_time": 1.7944748401641846,
      "step": 85606
    },
    {
      "epoch": 1.1068344116210937e-05,
      "step": 85606,
      "training_step_time": 7.36444616317749
    },
    {
      "epoch": 1.1069297790527343e-05,
      "model_forward_time": 1.775777816772461,
      "step": 85607
    },
    {
      "epoch": 1.1069297790527343e-05,
      "step": 85607,
      "training_step_time": 7.150139331817627
    },
    {
      "epoch": 1.107025146484375e-05,
      "model_forward_time": 1.783996820449829,
      "step": 85608
    },
    {
      "epoch": 1.107025146484375e-05,
      "step": 85608,
      "training_step_time": 8.66537857055664
    },
    {
      "epoch": 1.1071205139160156e-05,
      "model_forward_time": 2.404102325439453,
      "step": 85609
    },
    {
      "epoch": 1.1071205139160156e-05,
      "step": 85609,
      "training_step_time": 8.498028755187988
    },
    {
      "epoch": 1.1072158813476562e-05,
      "grad_norm": 0.11857947707176208,
      "learning_rate": 5.556000705057446e-07,
      "loss": 0.0547,
      "step": 85610
    },
    {
      "epoch": 1.1072158813476562e-05,
      "model_forward_time": 1.7625799179077148,
      "step": 85610
    },
    {
      "dynamics_loss_avg": 0.041029560193419454,
      "epoch": 1.1072158813476562e-05,
      "step": 85610
    },
    {
      "action_loss_avg": 0.0010797961353091524,
      "epoch": 1.1072158813476562e-05,
      "step": 85610
    },
    {
      "epoch": 1.1072158813476562e-05,
      "step": 85610,
      "training_step_time": 7.672205924987793
    },
    {
      "epoch": 1.107311248779297e-05,
      "model_forward_time": 1.7811331748962402,
      "step": 85611
    },
    {
      "epoch": 1.107311248779297e-05,
      "step": 85611,
      "training_step_time": 8.721927404403687
    },
    {
      "epoch": 1.1074066162109375e-05,
      "model_forward_time": 1.789815902709961,
      "step": 85612
    },
    {
      "epoch": 1.1074066162109375e-05,
      "step": 85612,
      "training_step_time": 10.090595483779907
    },
    {
      "epoch": 1.107501983642578e-05,
      "model_forward_time": 1.7832844257354736,
      "step": 85613
    },
    {
      "epoch": 1.107501983642578e-05,
      "step": 85613,
      "training_step_time": 7.62113356590271
    },
    {
      "epoch": 1.1075973510742188e-05,
      "model_forward_time": 1.7893695831298828,
      "step": 85614
    },
    {
      "epoch": 1.1075973510742188e-05,
      "step": 85614,
      "training_step_time": 7.6687233448028564
    },
    {
      "epoch": 1.1076927185058594e-05,
      "model_forward_time": 1.7792770862579346,
      "step": 85615
    },
    {
      "epoch": 1.1076927185058594e-05,
      "step": 85615,
      "training_step_time": 7.45741081237793
    },
    {
      "epoch": 1.1077880859375e-05,
      "model_forward_time": 1.7722289562225342,
      "step": 85616
    },
    {
      "epoch": 1.1077880859375e-05,
      "step": 85616,
      "training_step_time": 7.154608488082886
    },
    {
      "epoch": 1.1078834533691407e-05,
      "model_forward_time": 1.7809429168701172,
      "step": 85617
    },
    {
      "epoch": 1.1078834533691407e-05,
      "step": 85617,
      "training_step_time": 8.30717420578003
    },
    {
      "epoch": 1.1079788208007813e-05,
      "model_forward_time": 1.7771098613739014,
      "step": 85618
    },
    {
      "epoch": 1.1079788208007813e-05,
      "step": 85618,
      "training_step_time": 9.29085922241211
    },
    {
      "epoch": 1.1080741882324219e-05,
      "model_forward_time": 2.4962027072906494,
      "step": 85619
    },
    {
      "epoch": 1.1080741882324219e-05,
      "step": 85619,
      "training_step_time": 7.777990818023682
    },
    {
      "epoch": 1.1081695556640624e-05,
      "grad_norm": 0.10198947787284851,
      "learning_rate": 5.548427926732997e-07,
      "loss": 0.0538,
      "step": 85620
    },
    {
      "epoch": 1.1081695556640624e-05,
      "model_forward_time": 1.77128267288208,
      "step": 85620
    },
    {
      "dynamics_loss_avg": 0.06582112610340118,
      "epoch": 1.1081695556640624e-05,
      "step": 85620
    },
    {
      "action_loss_avg": 0.001265676345792599,
      "epoch": 1.1081695556640624e-05,
      "step": 85620
    },
    {
      "epoch": 1.1081695556640624e-05,
      "step": 85620,
      "training_step_time": 8.974176168441772
    },
    {
      "epoch": 1.1082649230957032e-05,
      "model_forward_time": 1.7735657691955566,
      "step": 85621
    },
    {
      "epoch": 1.1082649230957032e-05,
      "step": 85621,
      "training_step_time": 7.644393682479858
    },
    {
      "epoch": 1.1083602905273437e-05,
      "model_forward_time": 1.7859830856323242,
      "step": 85622
    },
    {
      "epoch": 1.1083602905273437e-05,
      "step": 85622,
      "training_step_time": 7.686830043792725
    },
    {
      "epoch": 1.1084556579589843e-05,
      "model_forward_time": 1.7793309688568115,
      "step": 85623
    },
    {
      "epoch": 1.1084556579589843e-05,
      "step": 85623,
      "training_step_time": 7.186097860336304
    },
    {
      "epoch": 1.108551025390625e-05,
      "model_forward_time": 1.7914307117462158,
      "step": 85624
    },
    {
      "epoch": 1.108551025390625e-05,
      "step": 85624,
      "training_step_time": 7.464128494262695
    },
    {
      "epoch": 1.1086463928222656e-05,
      "model_forward_time": 1.7710940837860107,
      "step": 85625
    },
    {
      "epoch": 1.1086463928222656e-05,
      "step": 85625,
      "training_step_time": 7.2402307987213135
    },
    {
      "epoch": 1.1087417602539062e-05,
      "model_forward_time": 1.7908964157104492,
      "step": 85626
    },
    {
      "epoch": 1.1087417602539062e-05,
      "step": 85626,
      "training_step_time": 8.320977449417114
    },
    {
      "epoch": 1.108837127685547e-05,
      "model_forward_time": 1.7801306247711182,
      "step": 85627
    },
    {
      "epoch": 1.108837127685547e-05,
      "step": 85627,
      "training_step_time": 7.678681135177612
    },
    {
      "epoch": 1.1089324951171875e-05,
      "model_forward_time": 2.4600062370300293,
      "step": 85628
    },
    {
      "epoch": 1.1089324951171875e-05,
      "step": 85628,
      "training_step_time": 8.899715423583984
    },
    {
      "epoch": 1.109027862548828e-05,
      "model_forward_time": 1.778926134109497,
      "step": 85629
    },
    {
      "epoch": 1.109027862548828e-05,
      "step": 85629,
      "training_step_time": 8.700046300888062
    },
    {
      "epoch": 1.1091232299804688e-05,
      "grad_norm": 0.21987369656562805,
      "learning_rate": 5.540860009565629e-07,
      "loss": 0.0536,
      "step": 85630
    },
    {
      "epoch": 1.1091232299804688e-05,
      "model_forward_time": 1.7699930667877197,
      "step": 85630
    },
    {
      "dynamics_loss_avg": 0.05685499329119921,
      "epoch": 1.1091232299804688e-05,
      "step": 85630
    },
    {
      "action_loss_avg": 0.001092694327235222,
      "epoch": 1.1091232299804688e-05,
      "step": 85630
    },
    {
      "epoch": 1.1091232299804688e-05,
      "step": 85630,
      "training_step_time": 7.850590467453003
    },
    {
      "epoch": 1.1092185974121094e-05,
      "model_forward_time": 1.779916763305664,
      "step": 85631
    },
    {
      "epoch": 1.1092185974121094e-05,
      "step": 85631,
      "training_step_time": 7.693815469741821
    },
    {
      "epoch": 1.10931396484375e-05,
      "model_forward_time": 1.78426194190979,
      "step": 85632
    },
    {
      "epoch": 1.10931396484375e-05,
      "step": 85632,
      "training_step_time": 7.567904472351074
    },
    {
      "epoch": 1.1094093322753905e-05,
      "model_forward_time": 1.786790132522583,
      "step": 85633
    },
    {
      "epoch": 1.1094093322753905e-05,
      "step": 85633,
      "training_step_time": 7.413947105407715
    },
    {
      "epoch": 1.1095046997070313e-05,
      "model_forward_time": 1.78190016746521,
      "step": 85634
    },
    {
      "epoch": 1.1095046997070313e-05,
      "step": 85634,
      "training_step_time": 7.483689785003662
    },
    {
      "epoch": 1.1096000671386719e-05,
      "model_forward_time": 1.7783575057983398,
      "step": 85635
    },
    {
      "epoch": 1.1096000671386719e-05,
      "step": 85635,
      "training_step_time": 7.581507205963135
    },
    {
      "epoch": 1.1096954345703124e-05,
      "model_forward_time": 1.7726798057556152,
      "step": 85636
    },
    {
      "epoch": 1.1096954345703124e-05,
      "step": 85636,
      "training_step_time": 7.679141998291016
    },
    {
      "epoch": 1.1097908020019532e-05,
      "model_forward_time": 2.3937625885009766,
      "step": 85637
    },
    {
      "epoch": 1.1097908020019532e-05,
      "step": 85637,
      "training_step_time": 8.923372983932495
    },
    {
      "epoch": 1.1098861694335937e-05,
      "model_forward_time": 1.7858738899230957,
      "step": 85638
    },
    {
      "epoch": 1.1098861694335937e-05,
      "step": 85638,
      "training_step_time": 9.76838493347168
    },
    {
      "epoch": 1.1099815368652343e-05,
      "model_forward_time": 1.7768287658691406,
      "step": 85639
    },
    {
      "epoch": 1.1099815368652343e-05,
      "step": 85639,
      "training_step_time": 7.899980306625366
    },
    {
      "epoch": 1.110076904296875e-05,
      "grad_norm": 0.10050306469202042,
      "learning_rate": 5.533296954382967e-07,
      "loss": 0.0549,
      "step": 85640
    },
    {
      "epoch": 1.110076904296875e-05,
      "model_forward_time": 1.773749828338623,
      "step": 85640
    },
    {
      "dynamics_loss_avg": 0.05886835679411888,
      "epoch": 1.110076904296875e-05,
      "step": 85640
    },
    {
      "action_loss_avg": 0.0014095779217313976,
      "epoch": 1.110076904296875e-05,
      "step": 85640
    },
    {
      "epoch": 1.110076904296875e-05,
      "step": 85640,
      "training_step_time": 8.453888893127441
    },
    {
      "epoch": 1.1101722717285156e-05,
      "model_forward_time": 1.7775769233703613,
      "step": 85641
    },
    {
      "epoch": 1.1101722717285156e-05,
      "step": 85641,
      "training_step_time": 7.633244276046753
    },
    {
      "epoch": 1.1102676391601562e-05,
      "model_forward_time": 1.7874376773834229,
      "step": 85642
    },
    {
      "epoch": 1.1102676391601562e-05,
      "step": 85642,
      "training_step_time": 7.432936191558838
    },
    {
      "epoch": 1.110363006591797e-05,
      "model_forward_time": 1.7740366458892822,
      "step": 85643
    },
    {
      "epoch": 1.110363006591797e-05,
      "step": 85643,
      "training_step_time": 8.458783864974976
    },
    {
      "epoch": 1.1104583740234375e-05,
      "model_forward_time": 1.7820026874542236,
      "step": 85644
    },
    {
      "epoch": 1.1104583740234375e-05,
      "step": 85644,
      "training_step_time": 7.498931407928467
    },
    {
      "epoch": 1.110553741455078e-05,
      "model_forward_time": 1.4053454399108887,
      "step": 85645
    },
    {
      "epoch": 1.110553741455078e-05,
      "step": 85645,
      "training_step_time": 7.691391706466675
    },
    {
      "epoch": 1.1106491088867188e-05,
      "model_forward_time": 2.4176766872406006,
      "step": 85646
    },
    {
      "epoch": 1.1106491088867188e-05,
      "step": 85646,
      "training_step_time": 8.445698022842407
    },
    {
      "epoch": 1.1107444763183594e-05,
      "model_forward_time": 1.7772152423858643,
      "step": 85647
    },
    {
      "epoch": 1.1107444763183594e-05,
      "step": 85647,
      "training_step_time": 8.497859239578247
    },
    {
      "epoch": 1.11083984375e-05,
      "model_forward_time": 1.778937816619873,
      "step": 85648
    },
    {
      "epoch": 1.11083984375e-05,
      "step": 85648,
      "training_step_time": 8.542912006378174
    },
    {
      "epoch": 1.1109352111816405e-05,
      "model_forward_time": 1.782078504562378,
      "step": 85649
    },
    {
      "epoch": 1.1109352111816405e-05,
      "step": 85649,
      "training_step_time": 8.495698690414429
    },
    {
      "epoch": 1.1110305786132813e-05,
      "grad_norm": 0.09995196014642715,
      "learning_rate": 5.525738762012095e-07,
      "loss": 0.0544,
      "step": 85650
    },
    {
      "epoch": 1.1110305786132813e-05,
      "model_forward_time": 1.7607975006103516,
      "step": 85650
    },
    {
      "dynamics_loss_avg": 0.06507909819483756,
      "epoch": 1.1110305786132813e-05,
      "step": 85650
    },
    {
      "action_loss_avg": 0.0013809151569148526,
      "epoch": 1.1110305786132813e-05,
      "step": 85650
    },
    {
      "epoch": 1.1110305786132813e-05,
      "step": 85650,
      "training_step_time": 7.407763481140137
    },
    {
      "epoch": 1.1111259460449219e-05,
      "model_forward_time": 1.7735912799835205,
      "step": 85651
    },
    {
      "epoch": 1.1111259460449219e-05,
      "step": 85651,
      "training_step_time": 7.327804803848267
    },
    {
      "epoch": 1.1112213134765624e-05,
      "model_forward_time": 1.7914173603057861,
      "step": 85652
    },
    {
      "epoch": 1.1112213134765624e-05,
      "step": 85652,
      "training_step_time": 7.463972806930542
    },
    {
      "epoch": 1.1113166809082032e-05,
      "model_forward_time": 1.7787995338439941,
      "step": 85653
    },
    {
      "epoch": 1.1113166809082032e-05,
      "step": 85653,
      "training_step_time": 7.530146360397339
    },
    {
      "epoch": 1.1114120483398437e-05,
      "model_forward_time": 1.778799057006836,
      "step": 85654
    },
    {
      "epoch": 1.1114120483398437e-05,
      "step": 85654,
      "training_step_time": 8.141642332077026
    },
    {
      "epoch": 1.1115074157714843e-05,
      "model_forward_time": 2.409806489944458,
      "step": 85655
    },
    {
      "epoch": 1.1115074157714843e-05,
      "step": 85655,
      "training_step_time": 8.84768009185791
    },
    {
      "epoch": 1.111602783203125e-05,
      "model_forward_time": 1.9061787128448486,
      "step": 85656
    },
    {
      "epoch": 1.111602783203125e-05,
      "step": 85656,
      "training_step_time": 8.909420490264893
    },
    {
      "epoch": 1.1116981506347656e-05,
      "model_forward_time": 1.9225103855133057,
      "step": 85657
    },
    {
      "epoch": 1.1116981506347656e-05,
      "step": 85657,
      "training_step_time": 8.327367782592773
    },
    {
      "epoch": 1.1117935180664062e-05,
      "model_forward_time": 1.9275801181793213,
      "step": 85658
    },
    {
      "epoch": 1.1117935180664062e-05,
      "step": 85658,
      "training_step_time": 8.467264652252197
    },
    {
      "epoch": 1.111888885498047e-05,
      "model_forward_time": 1.9202370643615723,
      "step": 85659
    },
    {
      "epoch": 1.111888885498047e-05,
      "step": 85659,
      "training_step_time": 7.4533257484436035
    },
    {
      "epoch": 1.1119842529296875e-05,
      "grad_norm": 0.07556469738483429,
      "learning_rate": 5.518185433279577e-07,
      "loss": 0.0519,
      "step": 85660
    },
    {
      "epoch": 1.1119842529296875e-05,
      "model_forward_time": 1.899582862854004,
      "step": 85660
    },
    {
      "dynamics_loss_avg": 0.057727962359786035,
      "epoch": 1.1119842529296875e-05,
      "step": 85660
    },
    {
      "action_loss_avg": 0.0009806175075937063,
      "epoch": 1.1119842529296875e-05,
      "step": 85660
    },
    {
      "epoch": 1.1119842529296875e-05,
      "step": 85660,
      "training_step_time": 7.392764568328857
    },
    {
      "epoch": 1.1120796203613281e-05,
      "model_forward_time": 1.9152727127075195,
      "step": 85661
    },
    {
      "epoch": 1.1120796203613281e-05,
      "step": 85661,
      "training_step_time": 7.3569395542144775
    },
    {
      "epoch": 1.1121749877929688e-05,
      "model_forward_time": 1.9076473712921143,
      "step": 85662
    },
    {
      "epoch": 1.1121749877929688e-05,
      "step": 85662,
      "training_step_time": 7.131293535232544
    },
    {
      "epoch": 1.1122703552246094e-05,
      "model_forward_time": 1.9289448261260986,
      "step": 85663
    },
    {
      "epoch": 1.1122703552246094e-05,
      "step": 85663,
      "training_step_time": 9.326836585998535
    },
    {
      "epoch": 1.11236572265625e-05,
      "model_forward_time": 1.9182391166687012,
      "step": 85664
    },
    {
      "epoch": 1.11236572265625e-05,
      "step": 85664,
      "training_step_time": 7.8279709815979
    },
    {
      "epoch": 1.1124610900878905e-05,
      "model_forward_time": 2.4654245376586914,
      "step": 85665
    },
    {
      "epoch": 1.1124610900878905e-05,
      "step": 85665,
      "training_step_time": 9.568519353866577
    },
    {
      "epoch": 1.1125564575195313e-05,
      "model_forward_time": 1.767653226852417,
      "step": 85666
    },
    {
      "epoch": 1.1125564575195313e-05,
      "step": 85666,
      "training_step_time": 7.811882257461548
    },
    {
      "epoch": 1.1126518249511719e-05,
      "model_forward_time": 1.7694718837738037,
      "step": 85667
    },
    {
      "epoch": 1.1126518249511719e-05,
      "step": 85667,
      "training_step_time": 7.704161643981934
    },
    {
      "epoch": 1.1127471923828124e-05,
      "model_forward_time": 1.779932975769043,
      "step": 85668
    },
    {
      "epoch": 1.1127471923828124e-05,
      "step": 85668,
      "training_step_time": 8.240280866622925
    },
    {
      "epoch": 1.1128425598144532e-05,
      "model_forward_time": 1.7697160243988037,
      "step": 85669
    },
    {
      "epoch": 1.1128425598144532e-05,
      "step": 85669,
      "training_step_time": 7.536559820175171
    },
    {
      "epoch": 1.1129379272460937e-05,
      "grad_norm": 0.08145619183778763,
      "learning_rate": 5.510636969011407e-07,
      "loss": 0.0532,
      "step": 85670
    },
    {
      "epoch": 1.1129379272460937e-05,
      "model_forward_time": 1.772730827331543,
      "step": 85670
    },
    {
      "dynamics_loss_avg": 0.06050585359334946,
      "epoch": 1.1129379272460937e-05,
      "step": 85670
    },
    {
      "action_loss_avg": 0.0016175118274986745,
      "epoch": 1.1129379272460937e-05,
      "step": 85670
    },
    {
      "epoch": 1.1129379272460937e-05,
      "step": 85670,
      "training_step_time": 8.104865789413452
    },
    {
      "epoch": 1.1130332946777343e-05,
      "model_forward_time": 1.7927818298339844,
      "step": 85671
    },
    {
      "epoch": 1.1130332946777343e-05,
      "step": 85671,
      "training_step_time": 7.589062452316284
    },
    {
      "epoch": 1.113128662109375e-05,
      "model_forward_time": 1.774580478668213,
      "step": 85672
    },
    {
      "epoch": 1.113128662109375e-05,
      "step": 85672,
      "training_step_time": 7.705592155456543
    },
    {
      "epoch": 1.1132240295410156e-05,
      "model_forward_time": 1.77079439163208,
      "step": 85673
    },
    {
      "epoch": 1.1132240295410156e-05,
      "step": 85673,
      "training_step_time": 8.640172719955444
    },
    {
      "epoch": 1.1133193969726562e-05,
      "model_forward_time": 2.4084770679473877,
      "step": 85674
    },
    {
      "epoch": 1.1133193969726562e-05,
      "step": 85674,
      "training_step_time": 8.512198209762573
    },
    {
      "epoch": 1.113414764404297e-05,
      "model_forward_time": 1.7799317836761475,
      "step": 85675
    },
    {
      "epoch": 1.113414764404297e-05,
      "step": 85675,
      "training_step_time": 7.781203985214233
    },
    {
      "epoch": 1.1135101318359375e-05,
      "model_forward_time": 1.7789323329925537,
      "step": 85676
    },
    {
      "epoch": 1.1135101318359375e-05,
      "step": 85676,
      "training_step_time": 8.571346044540405
    },
    {
      "epoch": 1.1136054992675781e-05,
      "model_forward_time": 1.7775192260742188,
      "step": 85677
    },
    {
      "epoch": 1.1136054992675781e-05,
      "step": 85677,
      "training_step_time": 8.673879623413086
    },
    {
      "epoch": 1.1137008666992188e-05,
      "model_forward_time": 1.785616159439087,
      "step": 85678
    },
    {
      "epoch": 1.1137008666992188e-05,
      "step": 85678,
      "training_step_time": 8.165522575378418
    },
    {
      "epoch": 1.1137962341308594e-05,
      "model_forward_time": 1.7928080558776855,
      "step": 85679
    },
    {
      "epoch": 1.1137962341308594e-05,
      "step": 85679,
      "training_step_time": 7.48149299621582
    },
    {
      "epoch": 1.1138916015625e-05,
      "grad_norm": 0.08300767838954926,
      "learning_rate": 5.503093370033097e-07,
      "loss": 0.0524,
      "step": 85680
    },
    {
      "epoch": 1.1138916015625e-05,
      "model_forward_time": 1.7662758827209473,
      "step": 85680
    },
    {
      "dynamics_loss_avg": 0.061067001335322856,
      "epoch": 1.1138916015625e-05,
      "step": 85680
    },
    {
      "action_loss_avg": 0.0017412948131095619,
      "epoch": 1.1138916015625e-05,
      "step": 85680
    },
    {
      "epoch": 1.1138916015625e-05,
      "step": 85680,
      "training_step_time": 7.621609210968018
    },
    {
      "epoch": 1.1139869689941405e-05,
      "model_forward_time": 1.7785146236419678,
      "step": 85681
    },
    {
      "epoch": 1.1139869689941405e-05,
      "step": 85681,
      "training_step_time": 7.725152254104614
    },
    {
      "epoch": 1.1140823364257813e-05,
      "model_forward_time": 1.7955021858215332,
      "step": 85682
    },
    {
      "epoch": 1.1140823364257813e-05,
      "step": 85682,
      "training_step_time": 8.523362874984741
    },
    {
      "epoch": 1.1141777038574219e-05,
      "model_forward_time": 2.411038398742676,
      "step": 85683
    },
    {
      "epoch": 1.1141777038574219e-05,
      "step": 85683,
      "training_step_time": 9.599722623825073
    },
    {
      "epoch": 1.1142730712890624e-05,
      "model_forward_time": 1.7813630104064941,
      "step": 85684
    },
    {
      "epoch": 1.1142730712890624e-05,
      "step": 85684,
      "training_step_time": 7.801668643951416
    },
    {
      "epoch": 1.1143684387207032e-05,
      "model_forward_time": 1.775810956954956,
      "step": 85685
    },
    {
      "epoch": 1.1143684387207032e-05,
      "step": 85685,
      "training_step_time": 7.705592393875122
    },
    {
      "epoch": 1.1144638061523437e-05,
      "model_forward_time": 1.793790578842163,
      "step": 85686
    },
    {
      "epoch": 1.1144638061523437e-05,
      "step": 85686,
      "training_step_time": 7.640234708786011
    },
    {
      "epoch": 1.1145591735839843e-05,
      "model_forward_time": 1.7965421676635742,
      "step": 85687
    },
    {
      "epoch": 1.1145591735839843e-05,
      "step": 85687,
      "training_step_time": 7.567046165466309
    },
    {
      "epoch": 1.114654541015625e-05,
      "model_forward_time": 1.7732884883880615,
      "step": 85688
    },
    {
      "epoch": 1.114654541015625e-05,
      "step": 85688,
      "training_step_time": 7.63071870803833
    },
    {
      "epoch": 1.1147499084472656e-05,
      "model_forward_time": 1.8000714778900146,
      "step": 85689
    },
    {
      "epoch": 1.1147499084472656e-05,
      "step": 85689,
      "training_step_time": 7.560680866241455
    },
    {
      "epoch": 1.1148452758789062e-05,
      "grad_norm": 0.07502235472202301,
      "learning_rate": 5.495554637169604e-07,
      "loss": 0.0531,
      "step": 85690
    },
    {
      "epoch": 1.1148452758789062e-05,
      "model_forward_time": 1.7747786045074463,
      "step": 85690
    },
    {
      "dynamics_loss_avg": 0.05489903707057238,
      "epoch": 1.1148452758789062e-05,
      "step": 85690
    },
    {
      "action_loss_avg": 0.001828967742039822,
      "epoch": 1.1148452758789062e-05,
      "step": 85690
    },
    {
      "epoch": 1.1148452758789062e-05,
      "step": 85690,
      "training_step_time": 8.19142770767212
    },
    {
      "epoch": 1.114940643310547e-05,
      "model_forward_time": 1.7845866680145264,
      "step": 85691
    },
    {
      "epoch": 1.114940643310547e-05,
      "step": 85691,
      "training_step_time": 7.67096734046936
    },
    {
      "epoch": 1.1150360107421875e-05,
      "model_forward_time": 2.4689972400665283,
      "step": 85692
    },
    {
      "epoch": 1.1150360107421875e-05,
      "step": 85692,
      "training_step_time": 9.257649660110474
    },
    {
      "epoch": 1.1151313781738281e-05,
      "model_forward_time": 1.7817597389221191,
      "step": 85693
    },
    {
      "epoch": 1.1151313781738281e-05,
      "step": 85693,
      "training_step_time": 7.903493642807007
    },
    {
      "epoch": 1.1152267456054688e-05,
      "model_forward_time": 1.8011665344238281,
      "step": 85694
    },
    {
      "epoch": 1.1152267456054688e-05,
      "step": 85694,
      "training_step_time": 9.397861242294312
    },
    {
      "epoch": 1.1153221130371094e-05,
      "model_forward_time": 1.7826967239379883,
      "step": 85695
    },
    {
      "epoch": 1.1153221130371094e-05,
      "step": 85695,
      "training_step_time": 8.732554197311401
    },
    {
      "epoch": 1.11541748046875e-05,
      "model_forward_time": 1.7714838981628418,
      "step": 85696
    },
    {
      "epoch": 1.11541748046875e-05,
      "step": 85696,
      "training_step_time": 7.508485555648804
    },
    {
      "epoch": 1.1155128479003905e-05,
      "model_forward_time": 1.773834228515625,
      "step": 85697
    },
    {
      "epoch": 1.1155128479003905e-05,
      "step": 85697,
      "training_step_time": 8.64773440361023
    },
    {
      "epoch": 1.1156082153320313e-05,
      "model_forward_time": 1.7957968711853027,
      "step": 85698
    },
    {
      "epoch": 1.1156082153320313e-05,
      "step": 85698,
      "training_step_time": 7.5327887535095215
    },
    {
      "epoch": 1.1157035827636719e-05,
      "model_forward_time": 1.756864070892334,
      "step": 85699
    },
    {
      "epoch": 1.1157035827636719e-05,
      "step": 85699,
      "training_step_time": 7.150866985321045
    },
    {
      "epoch": 1.1157989501953124e-05,
      "grad_norm": 0.10737323760986328,
      "learning_rate": 5.488020771245329e-07,
      "loss": 0.0537,
      "step": 85700
    },
    {
      "epoch": 1.1157989501953124e-05,
      "model_forward_time": 1.787062406539917,
      "step": 85700
    },
    {
      "dynamics_loss_avg": 0.07219922225922346,
      "epoch": 1.1157989501953124e-05,
      "step": 85700
    },
    {
      "action_loss_avg": 0.0014259835064876825,
      "epoch": 1.1157989501953124e-05,
      "step": 85700
    },
    {
      "epoch": 1.1157989501953124e-05,
      "step": 85700,
      "training_step_time": 7.726403713226318
    },
    {
      "epoch": 1.1158943176269532e-05,
      "model_forward_time": 2.4045746326446533,
      "step": 85701
    },
    {
      "epoch": 1.1158943176269532e-05,
      "step": 85701,
      "training_step_time": 10.182828903198242
    },
    {
      "epoch": 1.1159896850585937e-05,
      "model_forward_time": 1.8912875652313232,
      "step": 85702
    },
    {
      "epoch": 1.1159896850585937e-05,
      "step": 85702,
      "training_step_time": 8.767780065536499
    },
    {
      "epoch": 1.1160850524902343e-05,
      "model_forward_time": 1.778019905090332,
      "step": 85703
    },
    {
      "epoch": 1.1160850524902343e-05,
      "step": 85703,
      "training_step_time": 8.317015647888184
    },
    {
      "epoch": 1.116180419921875e-05,
      "model_forward_time": 1.7868103981018066,
      "step": 85704
    },
    {
      "epoch": 1.116180419921875e-05,
      "step": 85704,
      "training_step_time": 7.741596221923828
    },
    {
      "epoch": 1.1162757873535156e-05,
      "model_forward_time": 1.7811546325683594,
      "step": 85705
    },
    {
      "epoch": 1.1162757873535156e-05,
      "step": 85705,
      "training_step_time": 7.715254306793213
    },
    {
      "epoch": 1.1163711547851562e-05,
      "model_forward_time": 1.7904791831970215,
      "step": 85706
    },
    {
      "epoch": 1.1163711547851562e-05,
      "step": 85706,
      "training_step_time": 7.722728967666626
    },
    {
      "epoch": 1.116466522216797e-05,
      "model_forward_time": 1.7955331802368164,
      "step": 85707
    },
    {
      "epoch": 1.116466522216797e-05,
      "step": 85707,
      "training_step_time": 7.500053644180298
    },
    {
      "epoch": 1.1165618896484375e-05,
      "model_forward_time": 1.788635015487671,
      "step": 85708
    },
    {
      "epoch": 1.1165618896484375e-05,
      "step": 85708,
      "training_step_time": 7.63249945640564
    },
    {
      "epoch": 1.1166572570800781e-05,
      "model_forward_time": 1.7797493934631348,
      "step": 85709
    },
    {
      "epoch": 1.1166572570800781e-05,
      "step": 85709,
      "training_step_time": 7.698307752609253
    },
    {
      "epoch": 1.1167526245117188e-05,
      "grad_norm": 0.0857081189751625,
      "learning_rate": 5.480491773084196e-07,
      "loss": 0.0542,
      "step": 85710
    },
    {
      "epoch": 1.1167526245117188e-05,
      "model_forward_time": 1.7708594799041748,
      "step": 85710
    },
    {
      "dynamics_loss_avg": 0.07362288646399975,
      "epoch": 1.1167526245117188e-05,
      "step": 85710
    },
    {
      "action_loss_avg": 0.001981842378154397,
      "epoch": 1.1167526245117188e-05,
      "step": 85710
    },
    {
      "epoch": 1.1167526245117188e-05,
      "step": 85710,
      "training_step_time": 7.7524168491363525
    },
    {
      "epoch": 1.1168479919433594e-05,
      "model_forward_time": 2.5013720989227295,
      "step": 85711
    },
    {
      "epoch": 1.1168479919433594e-05,
      "step": 85711,
      "training_step_time": 8.821719646453857
    },
    {
      "epoch": 1.116943359375e-05,
      "model_forward_time": 1.7802937030792236,
      "step": 85712
    },
    {
      "epoch": 1.116943359375e-05,
      "step": 85712,
      "training_step_time": 7.827482223510742
    },
    {
      "epoch": 1.1170387268066405e-05,
      "model_forward_time": 1.782785415649414,
      "step": 85713
    },
    {
      "epoch": 1.1170387268066405e-05,
      "step": 85713,
      "training_step_time": 7.812228679656982
    },
    {
      "epoch": 1.1171340942382813e-05,
      "model_forward_time": 1.7834393978118896,
      "step": 85714
    },
    {
      "epoch": 1.1171340942382813e-05,
      "step": 85714,
      "training_step_time": 7.738630056381226
    },
    {
      "epoch": 1.1172294616699219e-05,
      "model_forward_time": 1.791353702545166,
      "step": 85715
    },
    {
      "epoch": 1.1172294616699219e-05,
      "step": 85715,
      "training_step_time": 9.310168266296387
    },
    {
      "epoch": 1.1173248291015624e-05,
      "model_forward_time": 1.7789239883422852,
      "step": 85716
    },
    {
      "epoch": 1.1173248291015624e-05,
      "step": 85716,
      "training_step_time": 7.532575845718384
    },
    {
      "epoch": 1.1174201965332032e-05,
      "model_forward_time": 1.7731125354766846,
      "step": 85717
    },
    {
      "epoch": 1.1174201965332032e-05,
      "step": 85717,
      "training_step_time": 8.149253129959106
    },
    {
      "epoch": 1.1175155639648437e-05,
      "model_forward_time": 1.7819914817810059,
      "step": 85718
    },
    {
      "epoch": 1.1175155639648437e-05,
      "step": 85718,
      "training_step_time": 8.55936312675476
    },
    {
      "epoch": 1.1176109313964843e-05,
      "model_forward_time": 1.7883448600769043,
      "step": 85719
    },
    {
      "epoch": 1.1176109313964843e-05,
      "step": 85719,
      "training_step_time": 9.652724027633667
    },
    {
      "epoch": 1.117706298828125e-05,
      "grad_norm": 0.09313742071390152,
      "learning_rate": 5.472967643509525e-07,
      "loss": 0.0538,
      "step": 85720
    },
    {
      "epoch": 1.117706298828125e-05,
      "model_forward_time": 2.4592337608337402,
      "step": 85720
    },
    {
      "dynamics_loss_avg": 0.057931799255311486,
      "epoch": 1.117706298828125e-05,
      "step": 85720
    },
    {
      "action_loss_avg": 0.0016233303293120116,
      "epoch": 1.117706298828125e-05,
      "step": 85720
    },
    {
      "epoch": 1.117706298828125e-05,
      "step": 85720,
      "training_step_time": 8.645809412002563
    },
    {
      "epoch": 1.1178016662597656e-05,
      "model_forward_time": 1.7814648151397705,
      "step": 85721
    },
    {
      "epoch": 1.1178016662597656e-05,
      "step": 85721,
      "training_step_time": 8.391420841217041
    },
    {
      "epoch": 1.1178970336914062e-05,
      "model_forward_time": 1.7839999198913574,
      "step": 85722
    },
    {
      "epoch": 1.1178970336914062e-05,
      "step": 85722,
      "training_step_time": 7.748301267623901
    },
    {
      "epoch": 1.117992401123047e-05,
      "model_forward_time": 1.7844016551971436,
      "step": 85723
    },
    {
      "epoch": 1.117992401123047e-05,
      "step": 85723,
      "training_step_time": 7.7143394947052
    },
    {
      "epoch": 1.1180877685546875e-05,
      "model_forward_time": 1.784984827041626,
      "step": 85724
    },
    {
      "epoch": 1.1180877685546875e-05,
      "step": 85724,
      "training_step_time": 7.533518552780151
    },
    {
      "epoch": 1.1181831359863281e-05,
      "model_forward_time": 1.7702534198760986,
      "step": 85725
    },
    {
      "epoch": 1.1181831359863281e-05,
      "step": 85725,
      "training_step_time": 7.646967172622681
    },
    {
      "epoch": 1.1182785034179688e-05,
      "model_forward_time": 1.7837588787078857,
      "step": 85726
    },
    {
      "epoch": 1.1182785034179688e-05,
      "step": 85726,
      "training_step_time": 7.603822231292725
    },
    {
      "epoch": 1.1183738708496094e-05,
      "model_forward_time": 1.782351016998291,
      "step": 85727
    },
    {
      "epoch": 1.1183738708496094e-05,
      "step": 85727,
      "training_step_time": 7.581019401550293
    },
    {
      "epoch": 1.11846923828125e-05,
      "model_forward_time": 1.7851512432098389,
      "step": 85728
    },
    {
      "epoch": 1.11846923828125e-05,
      "step": 85728,
      "training_step_time": 8.46748161315918
    },
    {
      "epoch": 1.1185646057128905e-05,
      "model_forward_time": 2.516385555267334,
      "step": 85729
    },
    {
      "epoch": 1.1185646057128905e-05,
      "step": 85729,
      "training_step_time": 9.002456903457642
    },
    {
      "epoch": 1.1186599731445313e-05,
      "grad_norm": 0.10360376536846161,
      "learning_rate": 5.465448383344191e-07,
      "loss": 0.0533,
      "step": 85730
    },
    {
      "epoch": 1.1186599731445313e-05,
      "model_forward_time": 1.7758631706237793,
      "step": 85730
    },
    {
      "dynamics_loss_avg": 0.05378238447010517,
      "epoch": 1.1186599731445313e-05,
      "step": 85730
    },
    {
      "action_loss_avg": 0.0012555847875773906,
      "epoch": 1.1186599731445313e-05,
      "step": 85730
    },
    {
      "epoch": 1.1186599731445313e-05,
      "step": 85730,
      "training_step_time": 7.7892045974731445
    },
    {
      "epoch": 1.1187553405761719e-05,
      "model_forward_time": 1.759889841079712,
      "step": 85731
    },
    {
      "epoch": 1.1187553405761719e-05,
      "step": 85731,
      "training_step_time": 7.805736780166626
    },
    {
      "epoch": 1.1188507080078124e-05,
      "model_forward_time": 1.765178918838501,
      "step": 85732
    },
    {
      "epoch": 1.1188507080078124e-05,
      "step": 85732,
      "training_step_time": 8.82305908203125
    },
    {
      "epoch": 1.1189460754394532e-05,
      "model_forward_time": 1.7584352493286133,
      "step": 85733
    },
    {
      "epoch": 1.1189460754394532e-05,
      "step": 85733,
      "training_step_time": 8.145281076431274
    },
    {
      "epoch": 1.1190414428710937e-05,
      "model_forward_time": 1.7619240283966064,
      "step": 85734
    },
    {
      "epoch": 1.1190414428710937e-05,
      "step": 85734,
      "training_step_time": 7.719124794006348
    },
    {
      "epoch": 1.1191368103027343e-05,
      "model_forward_time": 1.7885470390319824,
      "step": 85735
    },
    {
      "epoch": 1.1191368103027343e-05,
      "step": 85735,
      "training_step_time": 7.664714336395264
    },
    {
      "epoch": 1.119232177734375e-05,
      "model_forward_time": 1.742131233215332,
      "step": 85736
    },
    {
      "epoch": 1.119232177734375e-05,
      "step": 85736,
      "training_step_time": 7.505928993225098
    },
    {
      "epoch": 1.1193275451660156e-05,
      "model_forward_time": 1.762542963027954,
      "step": 85737
    },
    {
      "epoch": 1.1193275451660156e-05,
      "step": 85737,
      "training_step_time": 9.67361569404602
    },
    {
      "epoch": 1.1194229125976562e-05,
      "model_forward_time": 2.8164899349212646,
      "step": 85738
    },
    {
      "epoch": 1.1194229125976562e-05,
      "step": 85738,
      "training_step_time": 8.722317457199097
    },
    {
      "epoch": 1.119518280029297e-05,
      "model_forward_time": 1.8842244148254395,
      "step": 85739
    },
    {
      "epoch": 1.119518280029297e-05,
      "step": 85739,
      "training_step_time": 9.19906735420227
    },
    {
      "epoch": 1.1196136474609375e-05,
      "grad_norm": 0.0842590257525444,
      "learning_rate": 5.457933993410452e-07,
      "loss": 0.054,
      "step": 85740
    },
    {
      "epoch": 1.1196136474609375e-05,
      "model_forward_time": 1.7544686794281006,
      "step": 85740
    },
    {
      "dynamics_loss_avg": 0.0678125717677176,
      "epoch": 1.1196136474609375e-05,
      "step": 85740
    },
    {
      "action_loss_avg": 0.0011678089533234016,
      "epoch": 1.1196136474609375e-05,
      "step": 85740
    },
    {
      "epoch": 1.1196136474609375e-05,
      "step": 85740,
      "training_step_time": 8.668759107589722
    },
    {
      "epoch": 1.1197090148925781e-05,
      "model_forward_time": 1.759697675704956,
      "step": 85741
    },
    {
      "epoch": 1.1197090148925781e-05,
      "step": 85741,
      "training_step_time": 7.670004367828369
    },
    {
      "epoch": 1.1198043823242188e-05,
      "model_forward_time": 1.7652974128723145,
      "step": 85742
    },
    {
      "epoch": 1.1198043823242188e-05,
      "step": 85742,
      "training_step_time": 7.559157609939575
    },
    {
      "epoch": 1.1198997497558594e-05,
      "model_forward_time": 1.762239694595337,
      "step": 85743
    },
    {
      "epoch": 1.1198997497558594e-05,
      "step": 85743,
      "training_step_time": 7.564350366592407
    },
    {
      "epoch": 1.1199951171875e-05,
      "model_forward_time": 1.7706224918365479,
      "step": 85744
    },
    {
      "epoch": 1.1199951171875e-05,
      "step": 85744,
      "training_step_time": 7.562572240829468
    },
    {
      "epoch": 1.1200904846191406e-05,
      "model_forward_time": 1.7546396255493164,
      "step": 85745
    },
    {
      "epoch": 1.1200904846191406e-05,
      "step": 85745,
      "training_step_time": 8.123273134231567
    },
    {
      "epoch": 1.1201858520507813e-05,
      "model_forward_time": 1.77069091796875,
      "step": 85746
    },
    {
      "epoch": 1.1201858520507813e-05,
      "step": 85746,
      "training_step_time": 8.7313232421875
    },
    {
      "epoch": 1.1202812194824219e-05,
      "model_forward_time": 2.8087151050567627,
      "step": 85747
    },
    {
      "epoch": 1.1202812194824219e-05,
      "step": 85747,
      "training_step_time": 8.496754169464111
    },
    {
      "epoch": 1.1203765869140624e-05,
      "model_forward_time": 1.8833320140838623,
      "step": 85748
    },
    {
      "epoch": 1.1203765869140624e-05,
      "step": 85748,
      "training_step_time": 8.40721607208252
    },
    {
      "epoch": 1.1204719543457032e-05,
      "model_forward_time": 1.7535936832427979,
      "step": 85749
    },
    {
      "epoch": 1.1204719543457032e-05,
      "step": 85749,
      "training_step_time": 9.570881843566895
    },
    {
      "epoch": 1.1205673217773438e-05,
      "grad_norm": 0.08704402297735214,
      "learning_rate": 5.450424474530092e-07,
      "loss": 0.0543,
      "step": 85750
    },
    {
      "epoch": 1.1205673217773438e-05,
      "model_forward_time": 1.7559316158294678,
      "step": 85750
    },
    {
      "dynamics_loss_avg": 0.05431021563708782,
      "epoch": 1.1205673217773438e-05,
      "step": 85750
    },
    {
      "action_loss_avg": 0.0013688571110833437,
      "epoch": 1.1205673217773438e-05,
      "step": 85750
    },
    {
      "epoch": 1.1205673217773438e-05,
      "step": 85750,
      "training_step_time": 8.245364427566528
    },
    {
      "epoch": 1.1206626892089843e-05,
      "model_forward_time": 1.761551856994629,
      "step": 85751
    },
    {
      "epoch": 1.1206626892089843e-05,
      "step": 85751,
      "training_step_time": 7.647931814193726
    },
    {
      "epoch": 1.120758056640625e-05,
      "model_forward_time": 1.7583775520324707,
      "step": 85752
    },
    {
      "epoch": 1.120758056640625e-05,
      "step": 85752,
      "training_step_time": 7.484404563903809
    },
    {
      "epoch": 1.1208534240722656e-05,
      "model_forward_time": 1.767087697982788,
      "step": 85753
    },
    {
      "epoch": 1.1208534240722656e-05,
      "step": 85753,
      "training_step_time": 7.516289234161377
    },
    {
      "epoch": 1.1209487915039062e-05,
      "model_forward_time": 1.7598340511322021,
      "step": 85754
    },
    {
      "epoch": 1.1209487915039062e-05,
      "step": 85754,
      "training_step_time": 7.521904945373535
    },
    {
      "epoch": 1.121044158935547e-05,
      "model_forward_time": 1.7766368389129639,
      "step": 85755
    },
    {
      "epoch": 1.121044158935547e-05,
      "step": 85755,
      "training_step_time": 9.732109785079956
    },
    {
      "epoch": 1.1211395263671875e-05,
      "model_forward_time": 1.7812373638153076,
      "step": 85756
    },
    {
      "epoch": 1.1211395263671875e-05,
      "step": 85756,
      "training_step_time": 9.314457178115845
    },
    {
      "epoch": 1.1212348937988281e-05,
      "model_forward_time": 1.8771474361419678,
      "step": 85757
    },
    {
      "epoch": 1.1212348937988281e-05,
      "step": 85757,
      "training_step_time": 8.71253776550293
    },
    {
      "epoch": 1.1213302612304687e-05,
      "model_forward_time": 1.7857482433319092,
      "step": 85758
    },
    {
      "epoch": 1.1213302612304687e-05,
      "step": 85758,
      "training_step_time": 7.587268352508545
    },
    {
      "epoch": 1.1214256286621094e-05,
      "model_forward_time": 1.7672016620635986,
      "step": 85759
    },
    {
      "epoch": 1.1214256286621094e-05,
      "step": 85759,
      "training_step_time": 7.661056756973267
    },
    {
      "epoch": 1.12152099609375e-05,
      "grad_norm": 0.11909973621368408,
      "learning_rate": 5.442919827524323e-07,
      "loss": 0.0538,
      "step": 85760
    },
    {
      "epoch": 1.12152099609375e-05,
      "model_forward_time": 1.772468090057373,
      "step": 85760
    },
    {
      "dynamics_loss_avg": 0.04809151887893677,
      "epoch": 1.12152099609375e-05,
      "step": 85760
    },
    {
      "action_loss_avg": 0.00174287193804048,
      "epoch": 1.12152099609375e-05,
      "step": 85760
    },
    {
      "epoch": 1.12152099609375e-05,
      "step": 85760,
      "training_step_time": 7.577944993972778
    },
    {
      "epoch": 1.1216163635253906e-05,
      "model_forward_time": 1.7741177082061768,
      "step": 85761
    },
    {
      "epoch": 1.1216163635253906e-05,
      "step": 85761,
      "training_step_time": 7.464004039764404
    },
    {
      "epoch": 1.1217117309570313e-05,
      "model_forward_time": 1.7726452350616455,
      "step": 85762
    },
    {
      "epoch": 1.1217117309570313e-05,
      "step": 85762,
      "training_step_time": 7.438585042953491
    },
    {
      "epoch": 1.1218070983886719e-05,
      "model_forward_time": 1.7719669342041016,
      "step": 85763
    },
    {
      "epoch": 1.1218070983886719e-05,
      "step": 85763,
      "training_step_time": 8.207082748413086
    },
    {
      "epoch": 1.1219024658203124e-05,
      "model_forward_time": 1.766204595565796,
      "step": 85764
    },
    {
      "epoch": 1.1219024658203124e-05,
      "step": 85764,
      "training_step_time": 8.657492876052856
    },
    {
      "epoch": 1.1219978332519532e-05,
      "model_forward_time": 1.7819859981536865,
      "step": 85765
    },
    {
      "epoch": 1.1219978332519532e-05,
      "step": 85765,
      "training_step_time": 7.609713792800903
    },
    {
      "epoch": 1.1220932006835938e-05,
      "model_forward_time": 2.8364756107330322,
      "step": 85766
    },
    {
      "epoch": 1.1220932006835938e-05,
      "step": 85766,
      "training_step_time": 7.833898067474365
    },
    {
      "epoch": 1.1221885681152343e-05,
      "model_forward_time": 1.7641410827636719,
      "step": 85767
    },
    {
      "epoch": 1.1221885681152343e-05,
      "step": 85767,
      "training_step_time": 9.910430669784546
    },
    {
      "epoch": 1.122283935546875e-05,
      "model_forward_time": 1.7575554847717285,
      "step": 85768
    },
    {
      "epoch": 1.122283935546875e-05,
      "step": 85768,
      "training_step_time": 7.734307050704956
    },
    {
      "epoch": 1.1223793029785156e-05,
      "model_forward_time": 1.7588675022125244,
      "step": 85769
    },
    {
      "epoch": 1.1223793029785156e-05,
      "step": 85769,
      "training_step_time": 7.679279327392578
    },
    {
      "epoch": 1.1224746704101562e-05,
      "grad_norm": 0.07181522250175476,
      "learning_rate": 5.435420053213863e-07,
      "loss": 0.0542,
      "step": 85770
    },
    {
      "epoch": 1.1224746704101562e-05,
      "model_forward_time": 1.762901782989502,
      "step": 85770
    },
    {
      "dynamics_loss_avg": 0.05396885629743338,
      "epoch": 1.1224746704101562e-05,
      "step": 85770
    },
    {
      "action_loss_avg": 0.0015059739293064923,
      "epoch": 1.1224746704101562e-05,
      "step": 85770
    },
    {
      "epoch": 1.1224746704101562e-05,
      "step": 85770,
      "training_step_time": 8.430665493011475
    },
    {
      "epoch": 1.122570037841797e-05,
      "model_forward_time": 1.772022008895874,
      "step": 85771
    },
    {
      "epoch": 1.122570037841797e-05,
      "step": 85771,
      "training_step_time": 7.570619106292725
    },
    {
      "epoch": 1.1226654052734375e-05,
      "model_forward_time": 1.7692859172821045,
      "step": 85772
    },
    {
      "epoch": 1.1226654052734375e-05,
      "step": 85772,
      "training_step_time": 7.5836710929870605
    },
    {
      "epoch": 1.1227607727050781e-05,
      "model_forward_time": 1.7591993808746338,
      "step": 85773
    },
    {
      "epoch": 1.1227607727050781e-05,
      "step": 85773,
      "training_step_time": 7.618818998336792
    },
    {
      "epoch": 1.1228561401367187e-05,
      "model_forward_time": 1.7713892459869385,
      "step": 85774
    },
    {
      "epoch": 1.1228561401367187e-05,
      "step": 85774,
      "training_step_time": 7.615219354629517
    },
    {
      "epoch": 1.1229515075683594e-05,
      "model_forward_time": 2.832688570022583,
      "step": 85775
    },
    {
      "epoch": 1.1229515075683594e-05,
      "step": 85775,
      "training_step_time": 8.715102195739746
    },
    {
      "epoch": 1.123046875e-05,
      "model_forward_time": 1.7672522068023682,
      "step": 85776
    },
    {
      "epoch": 1.123046875e-05,
      "step": 85776,
      "training_step_time": 8.471095085144043
    },
    {
      "epoch": 1.1231422424316406e-05,
      "model_forward_time": 1.7708852291107178,
      "step": 85777
    },
    {
      "epoch": 1.1231422424316406e-05,
      "step": 85777,
      "training_step_time": 8.68973994255066
    },
    {
      "epoch": 1.1232376098632813e-05,
      "model_forward_time": 1.7623722553253174,
      "step": 85778
    },
    {
      "epoch": 1.1232376098632813e-05,
      "step": 85778,
      "training_step_time": 9.560155868530273
    },
    {
      "epoch": 1.1233329772949219e-05,
      "model_forward_time": 1.771528959274292,
      "step": 85779
    },
    {
      "epoch": 1.1233329772949219e-05,
      "step": 85779,
      "training_step_time": 9.246851921081543
    },
    {
      "epoch": 1.1234283447265624e-05,
      "grad_norm": 0.1035877987742424,
      "learning_rate": 5.427925152418845e-07,
      "loss": 0.0551,
      "step": 85780
    },
    {
      "epoch": 1.1234283447265624e-05,
      "model_forward_time": 1.7539875507354736,
      "step": 85780
    },
    {
      "dynamics_loss_avg": 0.07335978224873543,
      "epoch": 1.1234283447265624e-05,
      "step": 85780
    },
    {
      "action_loss_avg": 0.0010425210712128319,
      "epoch": 1.1234283447265624e-05,
      "step": 85780
    },
    {
      "epoch": 1.1234283447265624e-05,
      "step": 85780,
      "training_step_time": 8.129260063171387
    },
    {
      "epoch": 1.1235237121582032e-05,
      "model_forward_time": 1.7680201530456543,
      "step": 85781
    },
    {
      "epoch": 1.1235237121582032e-05,
      "step": 85781,
      "training_step_time": 7.483686923980713
    },
    {
      "epoch": 1.1236190795898438e-05,
      "model_forward_time": 1.7677721977233887,
      "step": 85782
    },
    {
      "epoch": 1.1236190795898438e-05,
      "step": 85782,
      "training_step_time": 7.677685260772705
    },
    {
      "epoch": 1.1237144470214843e-05,
      "model_forward_time": 1.7682175636291504,
      "step": 85783
    },
    {
      "epoch": 1.1237144470214843e-05,
      "step": 85783,
      "training_step_time": 8.111315488815308
    },
    {
      "epoch": 1.123809814453125e-05,
      "model_forward_time": 2.8308522701263428,
      "step": 85784
    },
    {
      "epoch": 1.123809814453125e-05,
      "step": 85784,
      "training_step_time": 7.742849826812744
    },
    {
      "epoch": 1.1239051818847656e-05,
      "model_forward_time": 1.771505355834961,
      "step": 85785
    },
    {
      "epoch": 1.1239051818847656e-05,
      "step": 85785,
      "training_step_time": 8.414217948913574
    },
    {
      "epoch": 1.1240005493164062e-05,
      "model_forward_time": 1.7605459690093994,
      "step": 85786
    },
    {
      "epoch": 1.1240005493164062e-05,
      "step": 85786,
      "training_step_time": 8.656340837478638
    },
    {
      "epoch": 1.124095916748047e-05,
      "model_forward_time": 1.7684752941131592,
      "step": 85787
    },
    {
      "epoch": 1.124095916748047e-05,
      "step": 85787,
      "training_step_time": 7.752167463302612
    },
    {
      "epoch": 1.1241912841796875e-05,
      "model_forward_time": 1.7635509967803955,
      "step": 85788
    },
    {
      "epoch": 1.1241912841796875e-05,
      "step": 85788,
      "training_step_time": 8.396416187286377
    },
    {
      "epoch": 1.1242866516113281e-05,
      "model_forward_time": 1.7761836051940918,
      "step": 85789
    },
    {
      "epoch": 1.1242866516113281e-05,
      "step": 85789,
      "training_step_time": 7.437073707580566
    },
    {
      "epoch": 1.1243820190429687e-05,
      "grad_norm": 0.08933977782726288,
      "learning_rate": 5.420435125958934e-07,
      "loss": 0.0551,
      "step": 85790
    },
    {
      "epoch": 1.1243820190429687e-05,
      "model_forward_time": 1.755847454071045,
      "step": 85790
    },
    {
      "dynamics_loss_avg": 0.04168168306350708,
      "epoch": 1.1243820190429687e-05,
      "step": 85790
    },
    {
      "action_loss_avg": 0.0006885809496452566,
      "epoch": 1.1243820190429687e-05,
      "step": 85790
    },
    {
      "epoch": 1.1243820190429687e-05,
      "step": 85790,
      "training_step_time": 7.517379999160767
    },
    {
      "epoch": 1.1244773864746094e-05,
      "model_forward_time": 1.74503493309021,
      "step": 85791
    },
    {
      "epoch": 1.1244773864746094e-05,
      "step": 85791,
      "training_step_time": 8.35245156288147
    },
    {
      "epoch": 1.12457275390625e-05,
      "model_forward_time": 1.7613213062286377,
      "step": 85792
    },
    {
      "epoch": 1.12457275390625e-05,
      "step": 85792,
      "training_step_time": 8.58489441871643
    },
    {
      "epoch": 1.1246681213378906e-05,
      "model_forward_time": 2.8346705436706543,
      "step": 85793
    },
    {
      "epoch": 1.1246681213378906e-05,
      "step": 85793,
      "training_step_time": 9.446039199829102
    },
    {
      "epoch": 1.1247634887695313e-05,
      "model_forward_time": 1.7588613033294678,
      "step": 85794
    },
    {
      "epoch": 1.1247634887695313e-05,
      "step": 85794,
      "training_step_time": 8.478824138641357
    },
    {
      "epoch": 1.1248588562011719e-05,
      "model_forward_time": 1.7465872764587402,
      "step": 85795
    },
    {
      "epoch": 1.1248588562011719e-05,
      "step": 85795,
      "training_step_time": 8.219890117645264
    },
    {
      "epoch": 1.1249542236328124e-05,
      "model_forward_time": 1.7668211460113525,
      "step": 85796
    },
    {
      "epoch": 1.1249542236328124e-05,
      "step": 85796,
      "training_step_time": 7.734505891799927
    },
    {
      "epoch": 1.1250495910644532e-05,
      "model_forward_time": 1.7626514434814453,
      "step": 85797
    },
    {
      "epoch": 1.1250495910644532e-05,
      "step": 85797,
      "training_step_time": 7.637770652770996
    },
    {
      "epoch": 1.1251449584960938e-05,
      "model_forward_time": 1.7691340446472168,
      "step": 85798
    },
    {
      "epoch": 1.1251449584960938e-05,
      "step": 85798,
      "training_step_time": 7.557955741882324
    },
    {
      "epoch": 1.1252403259277343e-05,
      "model_forward_time": 1.7697677612304688,
      "step": 85799
    },
    {
      "epoch": 1.1252403259277343e-05,
      "step": 85799,
      "training_step_time": 8.534780740737915
    },
    {
      "epoch": 1.125335693359375e-05,
      "grad_norm": 0.09270112961530685,
      "learning_rate": 5.412949974653209e-07,
      "loss": 0.055,
      "step": 85800
    },
    {
      "epoch": 1.125335693359375e-05,
      "model_forward_time": 1.7675468921661377,
      "step": 85800
    },
    {
      "dynamics_loss_avg": 0.05734664909541607,
      "epoch": 1.125335693359375e-05,
      "step": 85800
    },
    {
      "action_loss_avg": 0.001506976381642744,
      "epoch": 1.125335693359375e-05,
      "step": 85800
    },
    {
      "epoch": 1.125335693359375e-05,
      "step": 85800,
      "training_step_time": 7.679305076599121
    },
    {
      "epoch": 1.1254310607910156e-05,
      "model_forward_time": 1.784350872039795,
      "step": 85801
    },
    {
      "epoch": 1.1254310607910156e-05,
      "step": 85801,
      "training_step_time": 7.814961671829224
    },
    {
      "epoch": 1.1255264282226562e-05,
      "model_forward_time": 2.847202777862549,
      "step": 85802
    },
    {
      "epoch": 1.1255264282226562e-05,
      "step": 85802,
      "training_step_time": 8.588407516479492
    },
    {
      "epoch": 1.125621795654297e-05,
      "model_forward_time": 1.7551319599151611,
      "step": 85803
    },
    {
      "epoch": 1.125621795654297e-05,
      "step": 85803,
      "training_step_time": 7.634779691696167
    },
    {
      "epoch": 1.1257171630859375e-05,
      "model_forward_time": 1.7964439392089844,
      "step": 85804
    },
    {
      "epoch": 1.1257171630859375e-05,
      "step": 85804,
      "training_step_time": 7.661830186843872
    },
    {
      "epoch": 1.1258125305175781e-05,
      "model_forward_time": 1.7708594799041748,
      "step": 85805
    },
    {
      "epoch": 1.1258125305175781e-05,
      "step": 85805,
      "training_step_time": 8.414640188217163
    },
    {
      "epoch": 1.1259078979492187e-05,
      "model_forward_time": 1.7573962211608887,
      "step": 85806
    },
    {
      "epoch": 1.1259078979492187e-05,
      "step": 85806,
      "training_step_time": 7.660446405410767
    },
    {
      "epoch": 1.1260032653808594e-05,
      "model_forward_time": 1.785017728805542,
      "step": 85807
    },
    {
      "epoch": 1.1260032653808594e-05,
      "step": 85807,
      "training_step_time": 7.566905736923218
    },
    {
      "epoch": 1.1260986328125e-05,
      "model_forward_time": 1.7634153366088867,
      "step": 85808
    },
    {
      "epoch": 1.1260986328125e-05,
      "step": 85808,
      "training_step_time": 7.544925928115845
    },
    {
      "epoch": 1.1261940002441406e-05,
      "model_forward_time": 1.762415885925293,
      "step": 85809
    },
    {
      "epoch": 1.1261940002441406e-05,
      "step": 85809,
      "training_step_time": 8.866913318634033
    },
    {
      "epoch": 1.1262893676757813e-05,
      "grad_norm": 0.07796412706375122,
      "learning_rate": 5.405469699320248e-07,
      "loss": 0.0551,
      "step": 85810
    },
    {
      "epoch": 1.1262893676757813e-05,
      "model_forward_time": 1.7672550678253174,
      "step": 85810
    },
    {
      "dynamics_loss_avg": 0.04661338273435831,
      "epoch": 1.1262893676757813e-05,
      "step": 85810
    },
    {
      "action_loss_avg": 0.0011739642795873806,
      "epoch": 1.1262893676757813e-05,
      "step": 85810
    },
    {
      "epoch": 1.1262893676757813e-05,
      "step": 85810,
      "training_step_time": 7.866647005081177
    },
    {
      "epoch": 1.1263847351074219e-05,
      "model_forward_time": 2.8925561904907227,
      "step": 85811
    },
    {
      "epoch": 1.1263847351074219e-05,
      "step": 85811,
      "training_step_time": 8.51237440109253
    },
    {
      "epoch": 1.1264801025390624e-05,
      "model_forward_time": 1.7582385540008545,
      "step": 85812
    },
    {
      "epoch": 1.1264801025390624e-05,
      "step": 85812,
      "training_step_time": 7.690725326538086
    },
    {
      "epoch": 1.1265754699707032e-05,
      "model_forward_time": 1.7801251411437988,
      "step": 85813
    },
    {
      "epoch": 1.1265754699707032e-05,
      "step": 85813,
      "training_step_time": 9.183432340621948
    },
    {
      "epoch": 1.1266708374023438e-05,
      "model_forward_time": 1.766695261001587,
      "step": 85814
    },
    {
      "epoch": 1.1266708374023438e-05,
      "step": 85814,
      "training_step_time": 8.699554204940796
    },
    {
      "epoch": 1.1267662048339843e-05,
      "model_forward_time": 1.7721374034881592,
      "step": 85815
    },
    {
      "epoch": 1.1267662048339843e-05,
      "step": 85815,
      "training_step_time": 7.65495491027832
    },
    {
      "epoch": 1.126861572265625e-05,
      "model_forward_time": 1.7662134170532227,
      "step": 85816
    },
    {
      "epoch": 1.126861572265625e-05,
      "step": 85816,
      "training_step_time": 7.568004131317139
    },
    {
      "epoch": 1.1269569396972656e-05,
      "model_forward_time": 1.7555656433105469,
      "step": 85817
    },
    {
      "epoch": 1.1269569396972656e-05,
      "step": 85817,
      "training_step_time": 7.553457260131836
    },
    {
      "epoch": 1.1270523071289062e-05,
      "model_forward_time": 1.7727305889129639,
      "step": 85818
    },
    {
      "epoch": 1.1270523071289062e-05,
      "step": 85818,
      "training_step_time": 7.467292547225952
    },
    {
      "epoch": 1.127147674560547e-05,
      "model_forward_time": 1.7786204814910889,
      "step": 85819
    },
    {
      "epoch": 1.127147674560547e-05,
      "step": 85819,
      "training_step_time": 7.776993751525879
    },
    {
      "epoch": 1.1272430419921875e-05,
      "grad_norm": 0.14114531874656677,
      "learning_rate": 5.397994300778065e-07,
      "loss": 0.0549,
      "step": 85820
    },
    {
      "epoch": 1.1272430419921875e-05,
      "model_forward_time": 2.8929667472839355,
      "step": 85820
    },
    {
      "dynamics_loss_avg": 0.05815063454210758,
      "epoch": 1.1272430419921875e-05,
      "step": 85820
    },
    {
      "action_loss_avg": 0.0013205716386437416,
      "epoch": 1.1272430419921875e-05,
      "step": 85820
    },
    {
      "epoch": 1.1272430419921875e-05,
      "step": 85820,
      "training_step_time": 8.49618411064148
    },
    {
      "epoch": 1.1273384094238281e-05,
      "model_forward_time": 1.7644188404083252,
      "step": 85821
    },
    {
      "epoch": 1.1273384094238281e-05,
      "step": 85821,
      "training_step_time": 8.259886264801025
    },
    {
      "epoch": 1.1274337768554687e-05,
      "model_forward_time": 1.7773840427398682,
      "step": 85822
    },
    {
      "epoch": 1.1274337768554687e-05,
      "step": 85822,
      "training_step_time": 9.711983919143677
    },
    {
      "epoch": 1.1275291442871094e-05,
      "model_forward_time": 1.769697904586792,
      "step": 85823
    },
    {
      "epoch": 1.1275291442871094e-05,
      "step": 85823,
      "training_step_time": 7.700982570648193
    },
    {
      "epoch": 1.12762451171875e-05,
      "model_forward_time": 1.777038812637329,
      "step": 85824
    },
    {
      "epoch": 1.12762451171875e-05,
      "step": 85824,
      "training_step_time": 7.686444282531738
    },
    {
      "epoch": 1.1277198791503906e-05,
      "model_forward_time": 1.7720448970794678,
      "step": 85825
    },
    {
      "epoch": 1.1277198791503906e-05,
      "step": 85825,
      "training_step_time": 8.376255989074707
    },
    {
      "epoch": 1.1278152465820313e-05,
      "model_forward_time": 1.7877528667449951,
      "step": 85826
    },
    {
      "epoch": 1.1278152465820313e-05,
      "step": 85826,
      "training_step_time": 8.186656951904297
    },
    {
      "epoch": 1.1279106140136719e-05,
      "model_forward_time": 1.7812857627868652,
      "step": 85827
    },
    {
      "epoch": 1.1279106140136719e-05,
      "step": 85827,
      "training_step_time": 9.305541038513184
    },
    {
      "epoch": 1.1280059814453124e-05,
      "model_forward_time": 1.7584877014160156,
      "step": 85828
    },
    {
      "epoch": 1.1280059814453124e-05,
      "step": 85828,
      "training_step_time": 7.778832674026489
    },
    {
      "epoch": 1.1281013488769532e-05,
      "model_forward_time": 2.908797025680542,
      "step": 85829
    },
    {
      "epoch": 1.1281013488769532e-05,
      "step": 85829,
      "training_step_time": 8.729246616363525
    },
    {
      "epoch": 1.1281967163085938e-05,
      "grad_norm": 0.07303820550441742,
      "learning_rate": 5.39052377984417e-07,
      "loss": 0.0547,
      "step": 85830
    },
    {
      "epoch": 1.1281967163085938e-05,
      "model_forward_time": 1.7617626190185547,
      "step": 85830
    },
    {
      "dynamics_loss_avg": 0.06301014814525843,
      "epoch": 1.1281967163085938e-05,
      "step": 85830
    },
    {
      "action_loss_avg": 0.0015336485666921362,
      "epoch": 1.1281967163085938e-05,
      "step": 85830
    },
    {
      "epoch": 1.1281967163085938e-05,
      "step": 85830,
      "training_step_time": 10.40518569946289
    },
    {
      "epoch": 1.1282920837402343e-05,
      "model_forward_time": 1.7778112888336182,
      "step": 85831
    },
    {
      "epoch": 1.1282920837402343e-05,
      "step": 85831,
      "training_step_time": 9.03423523902893
    },
    {
      "epoch": 1.128387451171875e-05,
      "model_forward_time": 1.7608397006988525,
      "step": 85832
    },
    {
      "epoch": 1.128387451171875e-05,
      "step": 85832,
      "training_step_time": 8.134630918502808
    },
    {
      "epoch": 1.1284828186035156e-05,
      "model_forward_time": 1.7731871604919434,
      "step": 85833
    },
    {
      "epoch": 1.1284828186035156e-05,
      "step": 85833,
      "training_step_time": 7.706122875213623
    },
    {
      "epoch": 1.1285781860351562e-05,
      "model_forward_time": 1.7565064430236816,
      "step": 85834
    },
    {
      "epoch": 1.1285781860351562e-05,
      "step": 85834,
      "training_step_time": 7.523592710494995
    },
    {
      "epoch": 1.128673553466797e-05,
      "model_forward_time": 1.7653110027313232,
      "step": 85835
    },
    {
      "epoch": 1.128673553466797e-05,
      "step": 85835,
      "training_step_time": 8.2649245262146
    },
    {
      "epoch": 1.1287689208984375e-05,
      "model_forward_time": 1.7686681747436523,
      "step": 85836
    },
    {
      "epoch": 1.1287689208984375e-05,
      "step": 85836,
      "training_step_time": 7.557771444320679
    },
    {
      "epoch": 1.1288642883300781e-05,
      "model_forward_time": 1.7739715576171875,
      "step": 85837
    },
    {
      "epoch": 1.1288642883300781e-05,
      "step": 85837,
      "training_step_time": 7.690828800201416
    },
    {
      "epoch": 1.1289596557617187e-05,
      "model_forward_time": 2.829803228378296,
      "step": 85838
    },
    {
      "epoch": 1.1289596557617187e-05,
      "step": 85838,
      "training_step_time": 8.219746828079224
    },
    {
      "epoch": 1.1290550231933594e-05,
      "model_forward_time": 1.8882384300231934,
      "step": 85839
    },
    {
      "epoch": 1.1290550231933594e-05,
      "step": 85839,
      "training_step_time": 7.680303573608398
    },
    {
      "epoch": 1.129150390625e-05,
      "grad_norm": 0.11962438374757767,
      "learning_rate": 5.383058137335523e-07,
      "loss": 0.0549,
      "step": 85840
    },
    {
      "epoch": 1.129150390625e-05,
      "model_forward_time": 1.7610790729522705,
      "step": 85840
    },
    {
      "dynamics_loss_avg": 0.0683592788875103,
      "epoch": 1.129150390625e-05,
      "step": 85840
    },
    {
      "action_loss_avg": 0.0020276127237593756,
      "epoch": 1.129150390625e-05,
      "step": 85840
    },
    {
      "epoch": 1.129150390625e-05,
      "step": 85840,
      "training_step_time": 7.761484384536743
    },
    {
      "epoch": 1.1292457580566406e-05,
      "model_forward_time": 1.7690362930297852,
      "step": 85841
    },
    {
      "epoch": 1.1292457580566406e-05,
      "step": 85841,
      "training_step_time": 7.703399896621704
    },
    {
      "epoch": 1.1293411254882813e-05,
      "model_forward_time": 1.757619857788086,
      "step": 85842
    },
    {
      "epoch": 1.1293411254882813e-05,
      "step": 85842,
      "training_step_time": 9.656532764434814
    },
    {
      "epoch": 1.1294364929199219e-05,
      "model_forward_time": 1.7683000564575195,
      "step": 85843
    },
    {
      "epoch": 1.1294364929199219e-05,
      "step": 85843,
      "training_step_time": 7.640699863433838
    },
    {
      "epoch": 1.1295318603515624e-05,
      "model_forward_time": 1.7589976787567139,
      "step": 85844
    },
    {
      "epoch": 1.1295318603515624e-05,
      "step": 85844,
      "training_step_time": 8.366439580917358
    },
    {
      "epoch": 1.1296272277832032e-05,
      "model_forward_time": 1.7666873931884766,
      "step": 85845
    },
    {
      "epoch": 1.1296272277832032e-05,
      "step": 85845,
      "training_step_time": 7.583894968032837
    },
    {
      "epoch": 1.1297225952148438e-05,
      "model_forward_time": 1.7766685485839844,
      "step": 85846
    },
    {
      "epoch": 1.1297225952148438e-05,
      "step": 85846,
      "training_step_time": 7.637271881103516
    },
    {
      "epoch": 1.1298179626464843e-05,
      "model_forward_time": 2.856083869934082,
      "step": 85847
    },
    {
      "epoch": 1.1298179626464843e-05,
      "step": 85847,
      "training_step_time": 9.980738162994385
    },
    {
      "epoch": 1.129913330078125e-05,
      "model_forward_time": 1.8959124088287354,
      "step": 85848
    },
    {
      "epoch": 1.129913330078125e-05,
      "step": 85848,
      "training_step_time": 9.763758659362793
    },
    {
      "epoch": 1.1300086975097656e-05,
      "model_forward_time": 1.7699720859527588,
      "step": 85849
    },
    {
      "epoch": 1.1300086975097656e-05,
      "step": 85849,
      "training_step_time": 7.677444696426392
    },
    {
      "epoch": 1.1301040649414062e-05,
      "grad_norm": 0.09660337120294571,
      "learning_rate": 5.375597374068553e-07,
      "loss": 0.0541,
      "step": 85850
    },
    {
      "epoch": 1.1301040649414062e-05,
      "model_forward_time": 1.7658054828643799,
      "step": 85850
    },
    {
      "dynamics_loss_avg": 0.06137386951595545,
      "epoch": 1.1301040649414062e-05,
      "step": 85850
    },
    {
      "action_loss_avg": 0.001622177881654352,
      "epoch": 1.1301040649414062e-05,
      "step": 85850
    },
    {
      "epoch": 1.1301040649414062e-05,
      "step": 85850,
      "training_step_time": 7.66489052772522
    },
    {
      "epoch": 1.130199432373047e-05,
      "model_forward_time": 1.7694454193115234,
      "step": 85851
    },
    {
      "epoch": 1.130199432373047e-05,
      "step": 85851,
      "training_step_time": 7.751844167709351
    },
    {
      "epoch": 1.1302947998046875e-05,
      "model_forward_time": 1.7611629962921143,
      "step": 85852
    },
    {
      "epoch": 1.1302947998046875e-05,
      "step": 85852,
      "training_step_time": 7.615590333938599
    },
    {
      "epoch": 1.1303901672363281e-05,
      "model_forward_time": 1.7713608741760254,
      "step": 85853
    },
    {
      "epoch": 1.1303901672363281e-05,
      "step": 85853,
      "training_step_time": 7.390411376953125
    },
    {
      "epoch": 1.1304855346679687e-05,
      "model_forward_time": 1.7512338161468506,
      "step": 85854
    },
    {
      "epoch": 1.1304855346679687e-05,
      "step": 85854,
      "training_step_time": 8.061593770980835
    },
    {
      "epoch": 1.1305809020996094e-05,
      "model_forward_time": 1.769495964050293,
      "step": 85855
    },
    {
      "epoch": 1.1305809020996094e-05,
      "step": 85855,
      "training_step_time": 7.455276250839233
    },
    {
      "epoch": 1.13067626953125e-05,
      "model_forward_time": 2.889838933944702,
      "step": 85856
    },
    {
      "epoch": 1.13067626953125e-05,
      "step": 85856,
      "training_step_time": 8.791793584823608
    },
    {
      "epoch": 1.1307716369628906e-05,
      "model_forward_time": 1.9001355171203613,
      "step": 85857
    },
    {
      "epoch": 1.1307716369628906e-05,
      "step": 85857,
      "training_step_time": 8.151243448257446
    },
    {
      "epoch": 1.1308670043945313e-05,
      "model_forward_time": 1.7646706104278564,
      "step": 85858
    },
    {
      "epoch": 1.1308670043945313e-05,
      "step": 85858,
      "training_step_time": 7.702232122421265
    },
    {
      "epoch": 1.1309623718261719e-05,
      "model_forward_time": 1.7627224922180176,
      "step": 85859
    },
    {
      "epoch": 1.1309623718261719e-05,
      "step": 85859,
      "training_step_time": 8.08283543586731
    },
    {
      "epoch": 1.1310577392578124e-05,
      "grad_norm": 0.07545031607151031,
      "learning_rate": 5.368141490859163e-07,
      "loss": 0.0538,
      "step": 85860
    },
    {
      "epoch": 1.1310577392578124e-05,
      "model_forward_time": 1.7926342487335205,
      "step": 85860
    },
    {
      "dynamics_loss_avg": 0.05427788868546486,
      "epoch": 1.1310577392578124e-05,
      "step": 85860
    },
    {
      "action_loss_avg": 0.0017985543294344098,
      "epoch": 1.1310577392578124e-05,
      "step": 85860
    },
    {
      "epoch": 1.1310577392578124e-05,
      "step": 85860,
      "training_step_time": 8.912770748138428
    },
    {
      "epoch": 1.1311531066894532e-05,
      "model_forward_time": 1.7661261558532715,
      "step": 85861
    },
    {
      "epoch": 1.1311531066894532e-05,
      "step": 85861,
      "training_step_time": 7.609495162963867
    },
    {
      "epoch": 1.1312484741210938e-05,
      "model_forward_time": 1.7739737033843994,
      "step": 85862
    },
    {
      "epoch": 1.1312484741210938e-05,
      "step": 85862,
      "training_step_time": 7.3811726570129395
    },
    {
      "epoch": 1.1313438415527343e-05,
      "model_forward_time": 1.7797303199768066,
      "step": 85863
    },
    {
      "epoch": 1.1313438415527343e-05,
      "step": 85863,
      "training_step_time": 7.535919666290283
    },
    {
      "epoch": 1.131439208984375e-05,
      "model_forward_time": 1.7575016021728516,
      "step": 85864
    },
    {
      "epoch": 1.131439208984375e-05,
      "step": 85864,
      "training_step_time": 7.43736720085144
    },
    {
      "epoch": 1.1315345764160156e-05,
      "model_forward_time": 2.8434464931488037,
      "step": 85865
    },
    {
      "epoch": 1.1315345764160156e-05,
      "step": 85865,
      "training_step_time": 7.816480875015259
    },
    {
      "epoch": 1.1316299438476562e-05,
      "model_forward_time": 1.858083724975586,
      "step": 85866
    },
    {
      "epoch": 1.1316299438476562e-05,
      "step": 85866,
      "training_step_time": 7.638256549835205
    },
    {
      "epoch": 1.131725311279297e-05,
      "model_forward_time": 1.774939775466919,
      "step": 85867
    },
    {
      "epoch": 1.131725311279297e-05,
      "step": 85867,
      "training_step_time": 9.739267349243164
    },
    {
      "epoch": 1.1318206787109375e-05,
      "model_forward_time": 1.7693719863891602,
      "step": 85868
    },
    {
      "epoch": 1.1318206787109375e-05,
      "step": 85868,
      "training_step_time": 7.982226133346558
    },
    {
      "epoch": 1.1319160461425781e-05,
      "model_forward_time": 1.7737901210784912,
      "step": 85869
    },
    {
      "epoch": 1.1319160461425781e-05,
      "step": 85869,
      "training_step_time": 9.057219982147217
    },
    {
      "epoch": 1.1320114135742187e-05,
      "grad_norm": 0.08252210170030594,
      "learning_rate": 5.36069048852273e-07,
      "loss": 0.0537,
      "step": 85870
    },
    {
      "epoch": 1.1320114135742187e-05,
      "model_forward_time": 1.7479352951049805,
      "step": 85870
    },
    {
      "dynamics_loss_avg": 0.06617965158075094,
      "epoch": 1.1320114135742187e-05,
      "step": 85870
    },
    {
      "action_loss_avg": 0.0013085879327263683,
      "epoch": 1.1320114135742187e-05,
      "step": 85870
    },
    {
      "epoch": 1.1320114135742187e-05,
      "step": 85870,
      "training_step_time": 9.536418437957764
    },
    {
      "epoch": 1.1321067810058594e-05,
      "model_forward_time": 1.7720396518707275,
      "step": 85871
    },
    {
      "epoch": 1.1321067810058594e-05,
      "step": 85871,
      "training_step_time": 7.362334966659546
    },
    {
      "epoch": 1.1322021484375e-05,
      "model_forward_time": 1.776202917098999,
      "step": 85872
    },
    {
      "epoch": 1.1322021484375e-05,
      "step": 85872,
      "training_step_time": 7.536706924438477
    },
    {
      "epoch": 1.1322975158691406e-05,
      "model_forward_time": 1.7764873504638672,
      "step": 85873
    },
    {
      "epoch": 1.1322975158691406e-05,
      "step": 85873,
      "training_step_time": 8.20372223854065
    },
    {
      "epoch": 1.1323928833007813e-05,
      "model_forward_time": 2.8722691535949707,
      "step": 85874
    },
    {
      "epoch": 1.1323928833007813e-05,
      "step": 85874,
      "training_step_time": 7.837397575378418
    },
    {
      "epoch": 1.1324882507324219e-05,
      "model_forward_time": 1.826723575592041,
      "step": 85875
    },
    {
      "epoch": 1.1324882507324219e-05,
      "step": 85875,
      "training_step_time": 8.662036895751953
    },
    {
      "epoch": 1.1325836181640624e-05,
      "model_forward_time": 1.7649567127227783,
      "step": 85876
    },
    {
      "epoch": 1.1325836181640624e-05,
      "step": 85876,
      "training_step_time": 7.736079931259155
    },
    {
      "epoch": 1.1326789855957032e-05,
      "model_forward_time": 1.7833681106567383,
      "step": 85877
    },
    {
      "epoch": 1.1326789855957032e-05,
      "step": 85877,
      "training_step_time": 7.628120422363281
    },
    {
      "epoch": 1.1327743530273438e-05,
      "model_forward_time": 1.7739739418029785,
      "step": 85878
    },
    {
      "epoch": 1.1327743530273438e-05,
      "step": 85878,
      "training_step_time": 8.664406299591064
    },
    {
      "epoch": 1.1328697204589843e-05,
      "model_forward_time": 1.768212080001831,
      "step": 85879
    },
    {
      "epoch": 1.1328697204589843e-05,
      "step": 85879,
      "training_step_time": 8.73763370513916
    },
    {
      "epoch": 1.132965087890625e-05,
      "grad_norm": 0.09100866317749023,
      "learning_rate": 5.353244367874055e-07,
      "loss": 0.0549,
      "step": 85880
    },
    {
      "epoch": 1.132965087890625e-05,
      "model_forward_time": 1.7711613178253174,
      "step": 85880
    },
    {
      "dynamics_loss_avg": 0.055974262021481994,
      "epoch": 1.132965087890625e-05,
      "step": 85880
    },
    {
      "action_loss_avg": 0.0012766640633344651,
      "epoch": 1.132965087890625e-05,
      "step": 85880
    },
    {
      "epoch": 1.132965087890625e-05,
      "step": 85880,
      "training_step_time": 7.524012088775635
    },
    {
      "epoch": 1.1330604553222656e-05,
      "model_forward_time": 1.7740890979766846,
      "step": 85881
    },
    {
      "epoch": 1.1330604553222656e-05,
      "step": 85881,
      "training_step_time": 9.460004568099976
    },
    {
      "epoch": 1.1331558227539062e-05,
      "model_forward_time": 1.7872798442840576,
      "step": 85882
    },
    {
      "epoch": 1.1331558227539062e-05,
      "step": 85882,
      "training_step_time": 7.397323131561279
    },
    {
      "epoch": 1.1332511901855468e-05,
      "model_forward_time": 2.8486533164978027,
      "step": 85883
    },
    {
      "epoch": 1.1332511901855468e-05,
      "step": 85883,
      "training_step_time": 7.592423677444458
    },
    {
      "epoch": 1.1333465576171875e-05,
      "model_forward_time": 1.7952752113342285,
      "step": 85884
    },
    {
      "epoch": 1.1333465576171875e-05,
      "step": 85884,
      "training_step_time": 7.651348352432251
    },
    {
      "epoch": 1.1334419250488281e-05,
      "model_forward_time": 1.7574498653411865,
      "step": 85885
    },
    {
      "epoch": 1.1334419250488281e-05,
      "step": 85885,
      "training_step_time": 7.707088947296143
    },
    {
      "epoch": 1.1335372924804687e-05,
      "model_forward_time": 1.7518494129180908,
      "step": 85886
    },
    {
      "epoch": 1.1335372924804687e-05,
      "step": 85886,
      "training_step_time": 8.53866457939148
    },
    {
      "epoch": 1.1336326599121094e-05,
      "model_forward_time": 1.7573881149291992,
      "step": 85887
    },
    {
      "epoch": 1.1336326599121094e-05,
      "step": 85887,
      "training_step_time": 10.106664657592773
    },
    {
      "epoch": 1.13372802734375e-05,
      "model_forward_time": 1.7513806819915771,
      "step": 85888
    },
    {
      "epoch": 1.13372802734375e-05,
      "step": 85888,
      "training_step_time": 7.631763935089111
    },
    {
      "epoch": 1.1338233947753906e-05,
      "model_forward_time": 1.7757372856140137,
      "step": 85889
    },
    {
      "epoch": 1.1338233947753906e-05,
      "step": 85889,
      "training_step_time": 7.652822494506836
    },
    {
      "epoch": 1.1339187622070313e-05,
      "grad_norm": 0.08853675425052643,
      "learning_rate": 5.345803129727456e-07,
      "loss": 0.0528,
      "step": 85890
    },
    {
      "epoch": 1.1339187622070313e-05,
      "model_forward_time": 1.7512168884277344,
      "step": 85890
    },
    {
      "dynamics_loss_avg": 0.0627356419339776,
      "epoch": 1.1339187622070313e-05,
      "step": 85890
    },
    {
      "action_loss_avg": 0.001615533820586279,
      "epoch": 1.1339187622070313e-05,
      "step": 85890
    },
    {
      "epoch": 1.1339187622070313e-05,
      "step": 85890,
      "training_step_time": 7.594575643539429
    },
    {
      "epoch": 1.1340141296386719e-05,
      "model_forward_time": 1.7688539028167725,
      "step": 85891
    },
    {
      "epoch": 1.1340141296386719e-05,
      "step": 85891,
      "training_step_time": 7.3953118324279785
    },
    {
      "epoch": 1.1341094970703124e-05,
      "model_forward_time": 2.860126256942749,
      "step": 85892
    },
    {
      "epoch": 1.1341094970703124e-05,
      "step": 85892,
      "training_step_time": 8.253260374069214
    },
    {
      "epoch": 1.1342048645019532e-05,
      "model_forward_time": 1.749419927597046,
      "step": 85893
    },
    {
      "epoch": 1.1342048645019532e-05,
      "step": 85893,
      "training_step_time": 9.786509275436401
    },
    {
      "epoch": 1.1343002319335938e-05,
      "model_forward_time": 1.7529940605163574,
      "step": 85894
    },
    {
      "epoch": 1.1343002319335938e-05,
      "step": 85894,
      "training_step_time": 7.658042669296265
    },
    {
      "epoch": 1.1343955993652343e-05,
      "model_forward_time": 1.7441089153289795,
      "step": 85895
    },
    {
      "epoch": 1.1343955993652343e-05,
      "step": 85895,
      "training_step_time": 7.70029616355896
    },
    {
      "epoch": 1.134490966796875e-05,
      "model_forward_time": 1.7455127239227295,
      "step": 85896
    },
    {
      "epoch": 1.134490966796875e-05,
      "step": 85896,
      "training_step_time": 7.595512390136719
    },
    {
      "epoch": 1.1345863342285156e-05,
      "model_forward_time": 1.7527284622192383,
      "step": 85897
    },
    {
      "epoch": 1.1345863342285156e-05,
      "step": 85897,
      "training_step_time": 7.662507057189941
    },
    {
      "epoch": 1.1346817016601562e-05,
      "model_forward_time": 1.7761495113372803,
      "step": 85898
    },
    {
      "epoch": 1.1346817016601562e-05,
      "step": 85898,
      "training_step_time": 7.739809036254883
    },
    {
      "epoch": 1.1347770690917968e-05,
      "model_forward_time": 1.7535510063171387,
      "step": 85899
    },
    {
      "epoch": 1.1347770690917968e-05,
      "step": 85899,
      "training_step_time": 8.354432106018066
    },
    {
      "epoch": 1.1348724365234375e-05,
      "grad_norm": 0.11222411692142487,
      "learning_rate": 5.338366774896697e-07,
      "loss": 0.0554,
      "step": 85900
    },
    {
      "epoch": 1.1348724365234375e-05,
      "model_forward_time": 1.7525522708892822,
      "step": 85900
    },
    {
      "dynamics_loss_avg": 0.06487373746931553,
      "epoch": 1.1348724365234375e-05,
      "step": 85900
    },
    {
      "action_loss_avg": 0.0015455392189323903,
      "epoch": 1.1348724365234375e-05,
      "step": 85900
    },
    {
      "epoch": 1.1348724365234375e-05,
      "step": 85900,
      "training_step_time": 7.478087425231934
    },
    {
      "epoch": 1.1349678039550781e-05,
      "model_forward_time": 2.8492977619171143,
      "step": 85901
    },
    {
      "epoch": 1.1349678039550781e-05,
      "step": 85901,
      "training_step_time": 8.313233137130737
    },
    {
      "epoch": 1.1350631713867187e-05,
      "model_forward_time": 1.7527086734771729,
      "step": 85902
    },
    {
      "epoch": 1.1350631713867187e-05,
      "step": 85902,
      "training_step_time": 7.72182297706604
    },
    {
      "epoch": 1.1351585388183594e-05,
      "model_forward_time": 1.8274927139282227,
      "step": 85903
    },
    {
      "epoch": 1.1351585388183594e-05,
      "step": 85903,
      "training_step_time": 7.736422300338745
    },
    {
      "epoch": 1.13525390625e-05,
      "model_forward_time": 1.753385066986084,
      "step": 85904
    },
    {
      "epoch": 1.13525390625e-05,
      "step": 85904,
      "training_step_time": 8.668171167373657
    },
    {
      "epoch": 1.1353492736816406e-05,
      "model_forward_time": 1.757542371749878,
      "step": 85905
    },
    {
      "epoch": 1.1353492736816406e-05,
      "step": 85905,
      "training_step_time": 7.604092359542847
    },
    {
      "epoch": 1.1354446411132813e-05,
      "model_forward_time": 1.7677788734436035,
      "step": 85906
    },
    {
      "epoch": 1.1354446411132813e-05,
      "step": 85906,
      "training_step_time": 7.662402391433716
    },
    {
      "epoch": 1.1355400085449219e-05,
      "model_forward_time": 1.7690813541412354,
      "step": 85907
    },
    {
      "epoch": 1.1355400085449219e-05,
      "step": 85907,
      "training_step_time": 8.651912927627563
    },
    {
      "epoch": 1.1356353759765624e-05,
      "model_forward_time": 1.7404468059539795,
      "step": 85908
    },
    {
      "epoch": 1.1356353759765624e-05,
      "step": 85908,
      "training_step_time": 7.54273247718811
    },
    {
      "epoch": 1.1357307434082032e-05,
      "model_forward_time": 1.764770746231079,
      "step": 85909
    },
    {
      "epoch": 1.1357307434082032e-05,
      "step": 85909,
      "training_step_time": 7.476211309432983
    },
    {
      "epoch": 1.1358261108398438e-05,
      "grad_norm": 0.10285450518131256,
      "learning_rate": 5.330935304194995e-07,
      "loss": 0.054,
      "step": 85910
    },
    {
      "epoch": 1.1358261108398438e-05,
      "model_forward_time": 2.829091787338257,
      "step": 85910
    },
    {
      "dynamics_loss_avg": 0.04908850528299809,
      "epoch": 1.1358261108398438e-05,
      "step": 85910
    },
    {
      "action_loss_avg": 0.001804931933293119,
      "epoch": 1.1358261108398438e-05,
      "step": 85910
    },
    {
      "epoch": 1.1358261108398438e-05,
      "step": 85910,
      "training_step_time": 7.557424068450928
    },
    {
      "epoch": 1.1359214782714843e-05,
      "model_forward_time": 1.745345115661621,
      "step": 85911
    },
    {
      "epoch": 1.1359214782714843e-05,
      "step": 85911,
      "training_step_time": 7.865937232971191
    },
    {
      "epoch": 1.136016845703125e-05,
      "model_forward_time": 1.7540605068206787,
      "step": 85912
    },
    {
      "epoch": 1.136016845703125e-05,
      "step": 85912,
      "training_step_time": 8.701979398727417
    },
    {
      "epoch": 1.1361122131347656e-05,
      "model_forward_time": 1.7841796875,
      "step": 85913
    },
    {
      "epoch": 1.1361122131347656e-05,
      "step": 85913,
      "training_step_time": 8.711035966873169
    },
    {
      "epoch": 1.1362075805664062e-05,
      "model_forward_time": 1.7731914520263672,
      "step": 85914
    },
    {
      "epoch": 1.1362075805664062e-05,
      "step": 85914,
      "training_step_time": 9.189568281173706
    },
    {
      "epoch": 1.1363029479980468e-05,
      "model_forward_time": 1.7708454132080078,
      "step": 85915
    },
    {
      "epoch": 1.1363029479980468e-05,
      "step": 85915,
      "training_step_time": 7.613402366638184
    },
    {
      "epoch": 1.1363983154296875e-05,
      "model_forward_time": 1.7651615142822266,
      "step": 85916
    },
    {
      "epoch": 1.1363983154296875e-05,
      "step": 85916,
      "training_step_time": 8.573898553848267
    },
    {
      "epoch": 1.1364936828613281e-05,
      "model_forward_time": 1.7752933502197266,
      "step": 85917
    },
    {
      "epoch": 1.1364936828613281e-05,
      "step": 85917,
      "training_step_time": 7.485023021697998
    },
    {
      "epoch": 1.1365890502929687e-05,
      "model_forward_time": 1.7672572135925293,
      "step": 85918
    },
    {
      "epoch": 1.1365890502929687e-05,
      "step": 85918,
      "training_step_time": 7.503381252288818
    },
    {
      "epoch": 1.1366844177246094e-05,
      "model_forward_time": 2.83223819732666,
      "step": 85919
    },
    {
      "epoch": 1.1366844177246094e-05,
      "step": 85919,
      "training_step_time": 7.549750089645386
    },
    {
      "epoch": 1.13677978515625e-05,
      "grad_norm": 0.06905041635036469,
      "learning_rate": 5.323508718435044e-07,
      "loss": 0.0549,
      "step": 85920
    },
    {
      "epoch": 1.13677978515625e-05,
      "model_forward_time": 1.7776517868041992,
      "step": 85920
    },
    {
      "dynamics_loss_avg": 0.06191060356795788,
      "epoch": 1.13677978515625e-05,
      "step": 85920
    },
    {
      "action_loss_avg": 0.001011833650409244,
      "epoch": 1.13677978515625e-05,
      "step": 85920
    },
    {
      "epoch": 1.13677978515625e-05,
      "step": 85920,
      "training_step_time": 8.171162843704224
    },
    {
      "epoch": 1.1368751525878906e-05,
      "model_forward_time": 1.8030791282653809,
      "step": 85921
    },
    {
      "epoch": 1.1368751525878906e-05,
      "step": 85921,
      "training_step_time": 8.634726285934448
    },
    {
      "epoch": 1.1369705200195313e-05,
      "model_forward_time": 1.767146348953247,
      "step": 85922
    },
    {
      "epoch": 1.1369705200195313e-05,
      "step": 85922,
      "training_step_time": 8.408522844314575
    },
    {
      "epoch": 1.1370658874511719e-05,
      "model_forward_time": 1.7590434551239014,
      "step": 85923
    },
    {
      "epoch": 1.1370658874511719e-05,
      "step": 85923,
      "training_step_time": 8.189443826675415
    },
    {
      "epoch": 1.1371612548828125e-05,
      "model_forward_time": 1.743321418762207,
      "step": 85924
    },
    {
      "epoch": 1.1371612548828125e-05,
      "step": 85924,
      "training_step_time": 7.632273435592651
    },
    {
      "epoch": 1.1372566223144532e-05,
      "model_forward_time": 1.748387336730957,
      "step": 85925
    },
    {
      "epoch": 1.1372566223144532e-05,
      "step": 85925,
      "training_step_time": 9.575271844863892
    },
    {
      "epoch": 1.1373519897460938e-05,
      "model_forward_time": 1.7485485076904297,
      "step": 85926
    },
    {
      "epoch": 1.1373519897460938e-05,
      "step": 85926,
      "training_step_time": 8.490920543670654
    },
    {
      "epoch": 1.1374473571777343e-05,
      "model_forward_time": 1.8262848854064941,
      "step": 85927
    },
    {
      "epoch": 1.1374473571777343e-05,
      "step": 85927,
      "training_step_time": 7.527737140655518
    },
    {
      "epoch": 1.137542724609375e-05,
      "model_forward_time": 2.8471827507019043,
      "step": 85928
    },
    {
      "epoch": 1.137542724609375e-05,
      "step": 85928,
      "training_step_time": 7.592575788497925
    },
    {
      "epoch": 1.1376380920410157e-05,
      "model_forward_time": 2.410796880722046,
      "step": 85929
    },
    {
      "epoch": 1.1376380920410157e-05,
      "step": 85929,
      "training_step_time": 9.151405572891235
    },
    {
      "epoch": 1.1377334594726562e-05,
      "grad_norm": 0.08435521274805069,
      "learning_rate": 5.316087018429012e-07,
      "loss": 0.0541,
      "step": 85930
    },
    {
      "epoch": 1.1377334594726562e-05,
      "model_forward_time": 1.7593810558319092,
      "step": 85930
    },
    {
      "dynamics_loss_avg": 0.07568088844418526,
      "epoch": 1.1377334594726562e-05,
      "step": 85930
    },
    {
      "action_loss_avg": 0.001804399397224188,
      "epoch": 1.1377334594726562e-05,
      "step": 85930
    },
    {
      "epoch": 1.1377334594726562e-05,
      "step": 85930,
      "training_step_time": 7.905508995056152
    },
    {
      "epoch": 1.1378288269042968e-05,
      "model_forward_time": 1.7665300369262695,
      "step": 85931
    },
    {
      "epoch": 1.1378288269042968e-05,
      "step": 85931,
      "training_step_time": 7.401515245437622
    },
    {
      "epoch": 1.1379241943359375e-05,
      "model_forward_time": 1.7676644325256348,
      "step": 85932
    },
    {
      "epoch": 1.1379241943359375e-05,
      "step": 85932,
      "training_step_time": 7.4903645515441895
    },
    {
      "epoch": 1.1380195617675781e-05,
      "model_forward_time": 1.764103889465332,
      "step": 85933
    },
    {
      "epoch": 1.1380195617675781e-05,
      "step": 85933,
      "training_step_time": 7.16858983039856
    },
    {
      "epoch": 1.1381149291992187e-05,
      "model_forward_time": 1.7624807357788086,
      "step": 85934
    },
    {
      "epoch": 1.1381149291992187e-05,
      "step": 85934,
      "training_step_time": 6.7344183921813965
    },
    {
      "epoch": 1.1382102966308594e-05,
      "model_forward_time": 1.761368989944458,
      "step": 85935
    },
    {
      "epoch": 1.1382102966308594e-05,
      "step": 85935,
      "training_step_time": 6.786930084228516
    },
    {
      "epoch": 1.1383056640625e-05,
      "model_forward_time": 1.7649941444396973,
      "step": 85936
    },
    {
      "epoch": 1.1383056640625e-05,
      "step": 85936,
      "training_step_time": 6.85370397567749
    },
    {
      "epoch": 1.1384010314941406e-05,
      "model_forward_time": 1.7735521793365479,
      "step": 85937
    },
    {
      "epoch": 1.1384010314941406e-05,
      "step": 85937,
      "training_step_time": 8.336862325668335
    },
    {
      "epoch": 1.1384963989257813e-05,
      "model_forward_time": 1.760504961013794,
      "step": 85938
    },
    {
      "epoch": 1.1384963989257813e-05,
      "step": 85938,
      "training_step_time": 8.281590223312378
    },
    {
      "epoch": 1.1385917663574219e-05,
      "model_forward_time": 1.7654247283935547,
      "step": 85939
    },
    {
      "epoch": 1.1385917663574219e-05,
      "step": 85939,
      "training_step_time": 8.6504967212677
    },
    {
      "epoch": 1.1386871337890625e-05,
      "grad_norm": 0.08458903431892395,
      "learning_rate": 5.30867020498853e-07,
      "loss": 0.0551,
      "step": 85940
    },
    {
      "epoch": 1.1386871337890625e-05,
      "model_forward_time": 1.7638709545135498,
      "step": 85940
    },
    {
      "dynamics_loss_avg": 0.08727462589740753,
      "epoch": 1.1386871337890625e-05,
      "step": 85940
    },
    {
      "action_loss_avg": 0.0011754315055441111,
      "epoch": 1.1386871337890625e-05,
      "step": 85940
    },
    {
      "epoch": 1.1386871337890625e-05,
      "step": 85940,
      "training_step_time": 8.590095520019531
    },
    {
      "epoch": 1.1387825012207032e-05,
      "model_forward_time": 1.7609202861785889,
      "step": 85941
    },
    {
      "epoch": 1.1387825012207032e-05,
      "step": 85941,
      "training_step_time": 8.525398969650269
    },
    {
      "epoch": 1.1388778686523438e-05,
      "model_forward_time": 1.7947666645050049,
      "step": 85942
    },
    {
      "epoch": 1.1388778686523438e-05,
      "step": 85942,
      "training_step_time": 6.7291576862335205
    },
    {
      "epoch": 1.1389732360839843e-05,
      "model_forward_time": 1.3923718929290771,
      "step": 85943
    },
    {
      "epoch": 1.1389732360839843e-05,
      "step": 85943,
      "training_step_time": 6.693652868270874
    },
    {
      "epoch": 1.139068603515625e-05,
      "model_forward_time": 1.7620222568511963,
      "step": 85944
    },
    {
      "epoch": 1.139068603515625e-05,
      "step": 85944,
      "training_step_time": 6.777128458023071
    },
    {
      "epoch": 1.1391639709472657e-05,
      "model_forward_time": 1.7671663761138916,
      "step": 85945
    },
    {
      "epoch": 1.1391639709472657e-05,
      "step": 85945,
      "training_step_time": 6.78644323348999
    },
    {
      "epoch": 1.1392593383789062e-05,
      "model_forward_time": 1.7521765232086182,
      "step": 85946
    },
    {
      "epoch": 1.1392593383789062e-05,
      "step": 85946,
      "training_step_time": 6.753529071807861
    },
    {
      "epoch": 1.1393547058105468e-05,
      "model_forward_time": 2.8432202339172363,
      "step": 85947
    },
    {
      "epoch": 1.1393547058105468e-05,
      "step": 85947,
      "training_step_time": 9.541235208511353
    },
    {
      "epoch": 1.1394500732421875e-05,
      "model_forward_time": 1.7327139377593994,
      "step": 85948
    },
    {
      "epoch": 1.1394500732421875e-05,
      "step": 85948,
      "training_step_time": 7.758338212966919
    },
    {
      "epoch": 1.1395454406738281e-05,
      "model_forward_time": 1.7566142082214355,
      "step": 85949
    },
    {
      "epoch": 1.1395454406738281e-05,
      "step": 85949,
      "training_step_time": 7.552233934402466
    },
    {
      "epoch": 1.1396408081054687e-05,
      "grad_norm": 0.07874424010515213,
      "learning_rate": 5.301258278924671e-07,
      "loss": 0.055,
      "step": 85950
    },
    {
      "epoch": 1.1396408081054687e-05,
      "model_forward_time": 1.755145788192749,
      "step": 85950
    },
    {
      "dynamics_loss_avg": 0.0455800938885659,
      "epoch": 1.1396408081054687e-05,
      "step": 85950
    },
    {
      "action_loss_avg": 0.0013537340564653277,
      "epoch": 1.1396408081054687e-05,
      "step": 85950
    },
    {
      "epoch": 1.1396408081054687e-05,
      "step": 85950,
      "training_step_time": 7.533834457397461
    },
    {
      "epoch": 1.1397361755371094e-05,
      "model_forward_time": 1.7681517601013184,
      "step": 85951
    },
    {
      "epoch": 1.1397361755371094e-05,
      "step": 85951,
      "training_step_time": 7.204439640045166
    },
    {
      "epoch": 1.13983154296875e-05,
      "model_forward_time": 1.7497644424438477,
      "step": 85952
    },
    {
      "epoch": 1.13983154296875e-05,
      "step": 85952,
      "training_step_time": 6.722597122192383
    },
    {
      "epoch": 1.1399269104003906e-05,
      "model_forward_time": 1.7614758014678955,
      "step": 85953
    },
    {
      "epoch": 1.1399269104003906e-05,
      "step": 85953,
      "training_step_time": 6.708598613739014
    },
    {
      "epoch": 1.1400222778320313e-05,
      "model_forward_time": 1.7403264045715332,
      "step": 85954
    },
    {
      "epoch": 1.1400222778320313e-05,
      "step": 85954,
      "training_step_time": 6.73091197013855
    },
    {
      "epoch": 1.1401176452636719e-05,
      "model_forward_time": 1.7446563243865967,
      "step": 85955
    },
    {
      "epoch": 1.1401176452636719e-05,
      "step": 85955,
      "training_step_time": 6.725435018539429
    },
    {
      "epoch": 1.1402130126953125e-05,
      "model_forward_time": 2.8619375228881836,
      "step": 85956
    },
    {
      "epoch": 1.1402130126953125e-05,
      "step": 85956,
      "training_step_time": 7.6571619510650635
    },
    {
      "epoch": 1.1403083801269532e-05,
      "model_forward_time": 1.7728359699249268,
      "step": 85957
    },
    {
      "epoch": 1.1403083801269532e-05,
      "step": 85957,
      "training_step_time": 10.542658567428589
    },
    {
      "epoch": 1.1404037475585938e-05,
      "model_forward_time": 1.7676444053649902,
      "step": 85958
    },
    {
      "epoch": 1.1404037475585938e-05,
      "step": 85958,
      "training_step_time": 7.547057628631592
    },
    {
      "epoch": 1.1404991149902343e-05,
      "model_forward_time": 1.767512559890747,
      "step": 85959
    },
    {
      "epoch": 1.1404991149902343e-05,
      "step": 85959,
      "training_step_time": 7.517402410507202
    },
    {
      "epoch": 1.140594482421875e-05,
      "grad_norm": 0.07426999509334564,
      "learning_rate": 5.293851241048015e-07,
      "loss": 0.0537,
      "step": 85960
    },
    {
      "epoch": 1.140594482421875e-05,
      "model_forward_time": 1.7643263339996338,
      "step": 85960
    },
    {
      "dynamics_loss_avg": 0.053232628665864465,
      "epoch": 1.140594482421875e-05,
      "step": 85960
    },
    {
      "action_loss_avg": 0.0013259165280032903,
      "epoch": 1.140594482421875e-05,
      "step": 85960
    },
    {
      "epoch": 1.140594482421875e-05,
      "step": 85960,
      "training_step_time": 7.163607597351074
    },
    {
      "epoch": 1.1406898498535157e-05,
      "model_forward_time": 1.7644639015197754,
      "step": 85961
    },
    {
      "epoch": 1.1406898498535157e-05,
      "step": 85961,
      "training_step_time": 6.743401050567627
    },
    {
      "epoch": 1.1407852172851562e-05,
      "model_forward_time": 1.7574210166931152,
      "step": 85962
    },
    {
      "epoch": 1.1407852172851562e-05,
      "step": 85962,
      "training_step_time": 6.709518671035767
    },
    {
      "epoch": 1.1408805847167968e-05,
      "model_forward_time": 1.766249418258667,
      "step": 85963
    },
    {
      "epoch": 1.1408805847167968e-05,
      "step": 85963,
      "training_step_time": 6.723378896713257
    },
    {
      "epoch": 1.1409759521484375e-05,
      "model_forward_time": 1.786630630493164,
      "step": 85964
    },
    {
      "epoch": 1.1409759521484375e-05,
      "step": 85964,
      "training_step_time": 6.753373622894287
    },
    {
      "epoch": 1.1410713195800781e-05,
      "model_forward_time": 2.9037420749664307,
      "step": 85965
    },
    {
      "epoch": 1.1410713195800781e-05,
      "step": 85965,
      "training_step_time": 9.394839763641357
    },
    {
      "epoch": 1.1411666870117187e-05,
      "model_forward_time": 1.7664718627929688,
      "step": 85966
    },
    {
      "epoch": 1.1411666870117187e-05,
      "step": 85966,
      "training_step_time": 9.8645658493042
    },
    {
      "epoch": 1.1412620544433594e-05,
      "model_forward_time": 1.7608239650726318,
      "step": 85967
    },
    {
      "epoch": 1.1412620544433594e-05,
      "step": 85967,
      "training_step_time": 7.764919757843018
    },
    {
      "epoch": 1.141357421875e-05,
      "model_forward_time": 1.7764086723327637,
      "step": 85968
    },
    {
      "epoch": 1.141357421875e-05,
      "step": 85968,
      "training_step_time": 7.635167837142944
    },
    {
      "epoch": 1.1414527893066406e-05,
      "model_forward_time": 1.7645585536956787,
      "step": 85969
    },
    {
      "epoch": 1.1414527893066406e-05,
      "step": 85969,
      "training_step_time": 7.447757005691528
    },
    {
      "epoch": 1.1415481567382813e-05,
      "grad_norm": 0.10757552087306976,
      "learning_rate": 5.286449092168555e-07,
      "loss": 0.0542,
      "step": 85970
    },
    {
      "epoch": 1.1415481567382813e-05,
      "model_forward_time": 1.7774326801300049,
      "step": 85970
    },
    {
      "dynamics_loss_avg": 0.06500651836395263,
      "epoch": 1.1415481567382813e-05,
      "step": 85970
    },
    {
      "action_loss_avg": 0.001639836686081253,
      "epoch": 1.1415481567382813e-05,
      "step": 85970
    },
    {
      "epoch": 1.1415481567382813e-05,
      "step": 85970,
      "training_step_time": 6.897339105606079
    },
    {
      "epoch": 1.1416435241699219e-05,
      "model_forward_time": 1.7679154872894287,
      "step": 85971
    },
    {
      "epoch": 1.1416435241699219e-05,
      "step": 85971,
      "training_step_time": 6.826077222824097
    },
    {
      "epoch": 1.1417388916015625e-05,
      "model_forward_time": 1.7740304470062256,
      "step": 85972
    },
    {
      "epoch": 1.1417388916015625e-05,
      "step": 85972,
      "training_step_time": 6.755465984344482
    },
    {
      "epoch": 1.1418342590332032e-05,
      "model_forward_time": 1.7696397304534912,
      "step": 85973
    },
    {
      "epoch": 1.1418342590332032e-05,
      "step": 85973,
      "training_step_time": 6.862507581710815
    },
    {
      "epoch": 1.1419296264648438e-05,
      "model_forward_time": 2.8267548084259033,
      "step": 85974
    },
    {
      "epoch": 1.1419296264648438e-05,
      "step": 85974,
      "training_step_time": 8.79582166671753
    },
    {
      "epoch": 1.1420249938964843e-05,
      "model_forward_time": 1.758941888809204,
      "step": 85975
    },
    {
      "epoch": 1.1420249938964843e-05,
      "step": 85975,
      "training_step_time": 8.698781967163086
    },
    {
      "epoch": 1.142120361328125e-05,
      "model_forward_time": 1.7548127174377441,
      "step": 85976
    },
    {
      "epoch": 1.142120361328125e-05,
      "step": 85976,
      "training_step_time": 7.809741258621216
    },
    {
      "epoch": 1.1422157287597657e-05,
      "model_forward_time": 1.7459449768066406,
      "step": 85977
    },
    {
      "epoch": 1.1422157287597657e-05,
      "step": 85977,
      "training_step_time": 7.388460397720337
    },
    {
      "epoch": 1.1423110961914062e-05,
      "model_forward_time": 1.751976728439331,
      "step": 85978
    },
    {
      "epoch": 1.1423110961914062e-05,
      "step": 85978,
      "training_step_time": 8.100945711135864
    },
    {
      "epoch": 1.1424064636230468e-05,
      "model_forward_time": 1.7429583072662354,
      "step": 85979
    },
    {
      "epoch": 1.1424064636230468e-05,
      "step": 85979,
      "training_step_time": 7.15163516998291
    },
    {
      "epoch": 1.1425018310546875e-05,
      "grad_norm": 0.08500238507986069,
      "learning_rate": 5.27905183309581e-07,
      "loss": 0.0553,
      "step": 85980
    },
    {
      "epoch": 1.1425018310546875e-05,
      "model_forward_time": 1.7443690299987793,
      "step": 85980
    },
    {
      "dynamics_loss_avg": 0.05399370566010475,
      "epoch": 1.1425018310546875e-05,
      "step": 85980
    },
    {
      "action_loss_avg": 0.0015373633912531659,
      "epoch": 1.1425018310546875e-05,
      "step": 85980
    },
    {
      "epoch": 1.1425018310546875e-05,
      "step": 85980,
      "training_step_time": 6.779492616653442
    },
    {
      "epoch": 1.1425971984863281e-05,
      "model_forward_time": 1.746694803237915,
      "step": 85981
    },
    {
      "epoch": 1.1425971984863281e-05,
      "step": 85981,
      "training_step_time": 6.819982051849365
    },
    {
      "epoch": 1.1426925659179687e-05,
      "model_forward_time": 1.7554097175598145,
      "step": 85982
    },
    {
      "epoch": 1.1426925659179687e-05,
      "step": 85982,
      "training_step_time": 6.810763597488403
    },
    {
      "epoch": 1.1427879333496094e-05,
      "model_forward_time": 2.832775354385376,
      "step": 85983
    },
    {
      "epoch": 1.1427879333496094e-05,
      "step": 85983,
      "training_step_time": 7.617194652557373
    },
    {
      "epoch": 1.14288330078125e-05,
      "model_forward_time": 1.7549915313720703,
      "step": 85984
    },
    {
      "epoch": 1.14288330078125e-05,
      "step": 85984,
      "training_step_time": 7.776885032653809
    },
    {
      "epoch": 1.1429786682128906e-05,
      "model_forward_time": 1.7563254833221436,
      "step": 85985
    },
    {
      "epoch": 1.1429786682128906e-05,
      "step": 85985,
      "training_step_time": 10.202424049377441
    },
    {
      "epoch": 1.1430740356445313e-05,
      "model_forward_time": 1.7468492984771729,
      "step": 85986
    },
    {
      "epoch": 1.1430740356445313e-05,
      "step": 85986,
      "training_step_time": 7.4778831005096436
    },
    {
      "epoch": 1.1431694030761719e-05,
      "model_forward_time": 1.750619649887085,
      "step": 85987
    },
    {
      "epoch": 1.1431694030761719e-05,
      "step": 85987,
      "training_step_time": 7.6353960037231445
    },
    {
      "epoch": 1.1432647705078125e-05,
      "model_forward_time": 1.7507691383361816,
      "step": 85988
    },
    {
      "epoch": 1.1432647705078125e-05,
      "step": 85988,
      "training_step_time": 7.195403099060059
    },
    {
      "epoch": 1.1433601379394532e-05,
      "model_forward_time": 1.7474653720855713,
      "step": 85989
    },
    {
      "epoch": 1.1433601379394532e-05,
      "step": 85989,
      "training_step_time": 6.771746635437012
    },
    {
      "epoch": 1.1434555053710938e-05,
      "grad_norm": 0.06802637875080109,
      "learning_rate": 5.271659464638718e-07,
      "loss": 0.054,
      "step": 85990
    },
    {
      "epoch": 1.1434555053710938e-05,
      "model_forward_time": 1.3750081062316895,
      "step": 85990
    },
    {
      "dynamics_loss_avg": 0.07091498747467995,
      "epoch": 1.1434555053710938e-05,
      "step": 85990
    },
    {
      "action_loss_avg": 0.0015525722614256666,
      "epoch": 1.1434555053710938e-05,
      "step": 85990
    },
    {
      "epoch": 1.1434555053710938e-05,
      "step": 85990,
      "training_step_time": 6.828657865524292
    },
    {
      "epoch": 1.1435508728027343e-05,
      "model_forward_time": 1.7470040321350098,
      "step": 85991
    },
    {
      "epoch": 1.1435508728027343e-05,
      "step": 85991,
      "training_step_time": 6.689487934112549
    },
    {
      "epoch": 1.143646240234375e-05,
      "model_forward_time": 1.7370624542236328,
      "step": 85992
    },
    {
      "epoch": 1.143646240234375e-05,
      "step": 85992,
      "training_step_time": 7.382665157318115
    },
    {
      "epoch": 1.1437416076660157e-05,
      "model_forward_time": 2.86732816696167,
      "step": 85993
    },
    {
      "epoch": 1.1437416076660157e-05,
      "step": 85993,
      "training_step_time": 8.824281692504883
    },
    {
      "epoch": 1.1438369750976562e-05,
      "model_forward_time": 1.7400765419006348,
      "step": 85994
    },
    {
      "epoch": 1.1438369750976562e-05,
      "step": 85994,
      "training_step_time": 9.1301908493042
    },
    {
      "epoch": 1.1439323425292968e-05,
      "model_forward_time": 1.7484707832336426,
      "step": 85995
    },
    {
      "epoch": 1.1439323425292968e-05,
      "step": 85995,
      "training_step_time": 7.65185284614563
    },
    {
      "epoch": 1.1440277099609375e-05,
      "model_forward_time": 1.7407505512237549,
      "step": 85996
    },
    {
      "epoch": 1.1440277099609375e-05,
      "step": 85996,
      "training_step_time": 8.499904870986938
    },
    {
      "epoch": 1.1441230773925781e-05,
      "model_forward_time": 1.7307345867156982,
      "step": 85997
    },
    {
      "epoch": 1.1441230773925781e-05,
      "step": 85997,
      "training_step_time": 7.235708236694336
    },
    {
      "epoch": 1.1442184448242187e-05,
      "model_forward_time": 1.7939248085021973,
      "step": 85998
    },
    {
      "epoch": 1.1442184448242187e-05,
      "step": 85998,
      "training_step_time": 6.889623641967773
    },
    {
      "epoch": 1.1443138122558594e-05,
      "model_forward_time": 1.7892746925354004,
      "step": 85999
    },
    {
      "epoch": 1.1443138122558594e-05,
      "step": 85999,
      "training_step_time": 6.9355034828186035
    },
    {
      "epoch": 1.1444091796875e-05,
      "grad_norm": 0.08440960198640823,
      "learning_rate": 5.264271987605701e-07,
      "loss": 0.053,
      "step": 86000
    },
    {
      "epoch": 1.1444091796875e-05,
      "model_forward_time": 1.8198957443237305,
      "step": 86000
    },
    {
      "dynamics_loss_avg": 0.06363827958703042,
      "epoch": 1.1444091796875e-05,
      "step": 86000
    },
    {
      "action_loss_avg": 0.0015103574609383942,
      "epoch": 1.1444091796875e-05,
      "step": 86000
    },
    {
      "epoch": 1.1444091796875e-05,
      "step": 86000,
      "training_step_time": 6.623924255371094
    },
    {
      "epoch": 1.1445045471191406e-05,
      "model_forward_time": 3.055708169937134,
      "step": 86001
    },
    {
      "epoch": 1.1445045471191406e-05,
      "step": 86001,
      "training_step_time": 7.894626617431641
    },
    {
      "epoch": 1.1445999145507813e-05,
      "model_forward_time": 1.8463847637176514,
      "step": 86002
    },
    {
      "epoch": 1.1445999145507813e-05,
      "step": 86002,
      "training_step_time": 9.113247156143188
    },
    {
      "epoch": 1.1446952819824219e-05,
      "model_forward_time": 1.7754945755004883,
      "step": 86003
    },
    {
      "epoch": 1.1446952819824219e-05,
      "step": 86003,
      "training_step_time": 10.904693126678467
    },
    {
      "epoch": 1.1447906494140625e-05,
      "model_forward_time": 1.7511379718780518,
      "step": 86004
    },
    {
      "epoch": 1.1447906494140625e-05,
      "step": 86004,
      "training_step_time": 7.697040557861328
    },
    {
      "epoch": 1.1448860168457032e-05,
      "model_forward_time": 1.766371726989746,
      "step": 86005
    },
    {
      "epoch": 1.1448860168457032e-05,
      "step": 86005,
      "training_step_time": 8.131069898605347
    },
    {
      "epoch": 1.1449813842773438e-05,
      "model_forward_time": 1.764441728591919,
      "step": 86006
    },
    {
      "epoch": 1.1449813842773438e-05,
      "step": 86006,
      "training_step_time": 7.658443450927734
    },
    {
      "epoch": 1.1450767517089843e-05,
      "model_forward_time": 1.7714488506317139,
      "step": 86007
    },
    {
      "epoch": 1.1450767517089843e-05,
      "step": 86007,
      "training_step_time": 6.724670171737671
    },
    {
      "epoch": 1.145172119140625e-05,
      "model_forward_time": 1.7753841876983643,
      "step": 86008
    },
    {
      "epoch": 1.145172119140625e-05,
      "step": 86008,
      "training_step_time": 6.749251842498779
    },
    {
      "epoch": 1.1452674865722657e-05,
      "model_forward_time": 1.7525596618652344,
      "step": 86009
    },
    {
      "epoch": 1.1452674865722657e-05,
      "step": 86009,
      "training_step_time": 6.723288059234619
    },
    {
      "epoch": 1.1453628540039062e-05,
      "grad_norm": 0.09451708197593689,
      "learning_rate": 5.256889402804632e-07,
      "loss": 0.0529,
      "step": 86010
    },
    {
      "epoch": 1.1453628540039062e-05,
      "model_forward_time": 1.7632665634155273,
      "step": 86010
    },
    {
      "dynamics_loss_avg": 0.05070262793451548,
      "epoch": 1.1453628540039062e-05,
      "step": 86010
    },
    {
      "action_loss_avg": 0.0011161821690620854,
      "epoch": 1.1453628540039062e-05,
      "step": 86010
    },
    {
      "epoch": 1.1453628540039062e-05,
      "step": 86010,
      "training_step_time": 7.321779727935791
    },
    {
      "epoch": 1.1454582214355468e-05,
      "model_forward_time": 1.8489680290222168,
      "step": 86011
    },
    {
      "epoch": 1.1454582214355468e-05,
      "step": 86011,
      "training_step_time": 7.783941030502319
    },
    {
      "epoch": 1.1455535888671875e-05,
      "model_forward_time": 1.760110855102539,
      "step": 86012
    },
    {
      "epoch": 1.1455535888671875e-05,
      "step": 86012,
      "training_step_time": 11.132951974868774
    },
    {
      "epoch": 1.1456489562988281e-05,
      "model_forward_time": 1.0680265426635742,
      "step": 86013
    },
    {
      "epoch": 1.1456489562988281e-05,
      "step": 86013,
      "training_step_time": 8.46409559249878
    },
    {
      "epoch": 1.1457443237304687e-05,
      "model_forward_time": 1.7650256156921387,
      "step": 86014
    },
    {
      "epoch": 1.1457443237304687e-05,
      "step": 86014,
      "training_step_time": 7.262390375137329
    },
    {
      "epoch": 1.1458396911621094e-05,
      "model_forward_time": 1.776153802871704,
      "step": 86015
    },
    {
      "epoch": 1.1458396911621094e-05,
      "step": 86015,
      "training_step_time": 7.682714462280273
    },
    {
      "epoch": 1.14593505859375e-05,
      "model_forward_time": 1.7734034061431885,
      "step": 86016
    },
    {
      "epoch": 1.14593505859375e-05,
      "step": 86016,
      "training_step_time": 6.755796432495117
    },
    {
      "epoch": 1.1460304260253906e-05,
      "model_forward_time": 1.7619926929473877,
      "step": 86017
    },
    {
      "epoch": 1.1460304260253906e-05,
      "step": 86017,
      "training_step_time": 6.769017934799194
    },
    {
      "epoch": 1.1461257934570313e-05,
      "model_forward_time": 1.7817704677581787,
      "step": 86018
    },
    {
      "epoch": 1.1461257934570313e-05,
      "step": 86018,
      "training_step_time": 6.7271270751953125
    },
    {
      "epoch": 1.1462211608886719e-05,
      "model_forward_time": 1.772564172744751,
      "step": 86019
    },
    {
      "epoch": 1.1462211608886719e-05,
      "step": 86019,
      "training_step_time": 6.748316287994385
    },
    {
      "epoch": 1.1463165283203125e-05,
      "grad_norm": 0.14813120663166046,
      "learning_rate": 5.249511711042881e-07,
      "loss": 0.0528,
      "step": 86020
    },
    {
      "epoch": 1.1463165283203125e-05,
      "model_forward_time": 2.421466827392578,
      "step": 86020
    },
    {
      "dynamics_loss_avg": 0.05523719973862171,
      "epoch": 1.1463165283203125e-05,
      "step": 86020
    },
    {
      "action_loss_avg": 0.0011105311859864742,
      "epoch": 1.1463165283203125e-05,
      "step": 86020
    },
    {
      "epoch": 1.1463165283203125e-05,
      "step": 86020,
      "training_step_time": 7.669466018676758
    },
    {
      "epoch": 1.1464118957519532e-05,
      "model_forward_time": 1.769690990447998,
      "step": 86021
    },
    {
      "epoch": 1.1464118957519532e-05,
      "step": 86021,
      "training_step_time": 10.509875774383545
    },
    {
      "epoch": 1.1465072631835938e-05,
      "model_forward_time": 1.7656786441802979,
      "step": 86022
    },
    {
      "epoch": 1.1465072631835938e-05,
      "step": 86022,
      "training_step_time": 10.511222839355469
    },
    {
      "epoch": 1.1466026306152343e-05,
      "model_forward_time": 1.7776660919189453,
      "step": 86023
    },
    {
      "epoch": 1.1466026306152343e-05,
      "step": 86023,
      "training_step_time": 7.473263263702393
    },
    {
      "epoch": 1.146697998046875e-05,
      "model_forward_time": 1.7726061344146729,
      "step": 86024
    },
    {
      "epoch": 1.146697998046875e-05,
      "step": 86024,
      "training_step_time": 7.606154680252075
    },
    {
      "epoch": 1.1467933654785157e-05,
      "model_forward_time": 1.7789433002471924,
      "step": 86025
    },
    {
      "epoch": 1.1467933654785157e-05,
      "step": 86025,
      "training_step_time": 6.749646186828613
    },
    {
      "epoch": 1.1468887329101562e-05,
      "model_forward_time": 1.7760651111602783,
      "step": 86026
    },
    {
      "epoch": 1.1468887329101562e-05,
      "step": 86026,
      "training_step_time": 6.755805253982544
    },
    {
      "epoch": 1.1469841003417968e-05,
      "model_forward_time": 1.7698144912719727,
      "step": 86027
    },
    {
      "epoch": 1.1469841003417968e-05,
      "step": 86027,
      "training_step_time": 6.731928825378418
    },
    {
      "epoch": 1.1470794677734375e-05,
      "model_forward_time": 1.7659218311309814,
      "step": 86028
    },
    {
      "epoch": 1.1470794677734375e-05,
      "step": 86028,
      "training_step_time": 6.7108824253082275
    },
    {
      "epoch": 1.1471748352050781e-05,
      "model_forward_time": 2.433138370513916,
      "step": 86029
    },
    {
      "epoch": 1.1471748352050781e-05,
      "step": 86029,
      "training_step_time": 7.685010671615601
    },
    {
      "epoch": 1.1472702026367187e-05,
      "grad_norm": 0.07119202613830566,
      "learning_rate": 5.242138913127237e-07,
      "loss": 0.0524,
      "step": 86030
    },
    {
      "epoch": 1.1472702026367187e-05,
      "model_forward_time": 1.7504973411560059,
      "step": 86030
    },
    {
      "dynamics_loss_avg": 0.04554017819464207,
      "epoch": 1.1472702026367187e-05,
      "step": 86030
    },
    {
      "action_loss_avg": 0.0009317747681052424,
      "epoch": 1.1472702026367187e-05,
      "step": 86030
    },
    {
      "epoch": 1.1472702026367187e-05,
      "step": 86030,
      "training_step_time": 10.981358051300049
    },
    {
      "epoch": 1.1473655700683594e-05,
      "model_forward_time": 1.758624792098999,
      "step": 86031
    },
    {
      "epoch": 1.1473655700683594e-05,
      "step": 86031,
      "training_step_time": 8.485992908477783
    },
    {
      "epoch": 1.1474609375e-05,
      "model_forward_time": 1.7787408828735352,
      "step": 86032
    },
    {
      "epoch": 1.1474609375e-05,
      "step": 86032,
      "training_step_time": 7.713895797729492
    },
    {
      "epoch": 1.1475563049316406e-05,
      "model_forward_time": 1.7630469799041748,
      "step": 86033
    },
    {
      "epoch": 1.1475563049316406e-05,
      "step": 86033,
      "training_step_time": 7.393118858337402
    },
    {
      "epoch": 1.1476516723632813e-05,
      "model_forward_time": 1.7689018249511719,
      "step": 86034
    },
    {
      "epoch": 1.1476516723632813e-05,
      "step": 86034,
      "training_step_time": 7.739548921585083
    },
    {
      "epoch": 1.1477470397949219e-05,
      "model_forward_time": 1.7608551979064941,
      "step": 86035
    },
    {
      "epoch": 1.1477470397949219e-05,
      "step": 86035,
      "training_step_time": 6.793567180633545
    },
    {
      "epoch": 1.1478424072265625e-05,
      "model_forward_time": 1.7785699367523193,
      "step": 86036
    },
    {
      "epoch": 1.1478424072265625e-05,
      "step": 86036,
      "training_step_time": 6.854204177856445
    },
    {
      "epoch": 1.1479377746582032e-05,
      "model_forward_time": 1.780970811843872,
      "step": 86037
    },
    {
      "epoch": 1.1479377746582032e-05,
      "step": 86037,
      "training_step_time": 6.874234914779663
    },
    {
      "epoch": 1.1480331420898438e-05,
      "model_forward_time": 2.4345602989196777,
      "step": 86038
    },
    {
      "epoch": 1.1480331420898438e-05,
      "step": 86038,
      "training_step_time": 7.768247127532959
    },
    {
      "epoch": 1.1481285095214843e-05,
      "model_forward_time": 1.7623941898345947,
      "step": 86039
    },
    {
      "epoch": 1.1481285095214843e-05,
      "step": 86039,
      "training_step_time": 7.811110258102417
    },
    {
      "epoch": 1.148223876953125e-05,
      "grad_norm": 0.09310485422611237,
      "learning_rate": 5.234771009863987e-07,
      "loss": 0.0532,
      "step": 86040
    },
    {
      "epoch": 1.148223876953125e-05,
      "model_forward_time": 1.7544543743133545,
      "step": 86040
    },
    {
      "dynamics_loss_avg": 0.04380970466881991,
      "epoch": 1.148223876953125e-05,
      "step": 86040
    },
    {
      "action_loss_avg": 0.0010162762278923764,
      "epoch": 1.148223876953125e-05,
      "step": 86040
    },
    {
      "epoch": 1.148223876953125e-05,
      "step": 86040,
      "training_step_time": 8.335479497909546
    },
    {
      "epoch": 1.1483192443847657e-05,
      "model_forward_time": 1.8833675384521484,
      "step": 86041
    },
    {
      "epoch": 1.1483192443847657e-05,
      "step": 86041,
      "training_step_time": 7.741079807281494
    },
    {
      "epoch": 1.1484146118164062e-05,
      "model_forward_time": 1.5456981658935547,
      "step": 86042
    },
    {
      "epoch": 1.1484146118164062e-05,
      "step": 86042,
      "training_step_time": 8.05486798286438
    },
    {
      "epoch": 1.1485099792480468e-05,
      "model_forward_time": 1.922886848449707,
      "step": 86043
    },
    {
      "epoch": 1.1485099792480468e-05,
      "step": 86043,
      "training_step_time": 7.671249866485596
    },
    {
      "epoch": 1.1486053466796875e-05,
      "model_forward_time": 1.9211955070495605,
      "step": 86044
    },
    {
      "epoch": 1.1486053466796875e-05,
      "step": 86044,
      "training_step_time": 6.811687707901001
    },
    {
      "epoch": 1.1487007141113281e-05,
      "model_forward_time": 1.9140067100524902,
      "step": 86045
    },
    {
      "epoch": 1.1487007141113281e-05,
      "step": 86045,
      "training_step_time": 6.818457126617432
    },
    {
      "epoch": 1.1487960815429687e-05,
      "model_forward_time": 1.9202508926391602,
      "step": 86046
    },
    {
      "epoch": 1.1487960815429687e-05,
      "step": 86046,
      "training_step_time": 6.860278844833374
    },
    {
      "epoch": 1.1488914489746094e-05,
      "model_forward_time": 1.9216694831848145,
      "step": 86047
    },
    {
      "epoch": 1.1488914489746094e-05,
      "step": 86047,
      "training_step_time": 7.626994848251343
    },
    {
      "epoch": 1.14898681640625e-05,
      "model_forward_time": 2.520207643508911,
      "step": 86048
    },
    {
      "epoch": 1.14898681640625e-05,
      "step": 86048,
      "training_step_time": 9.141540288925171
    },
    {
      "epoch": 1.1490821838378906e-05,
      "model_forward_time": 1.9213008880615234,
      "step": 86049
    },
    {
      "epoch": 1.1490821838378906e-05,
      "step": 86049,
      "training_step_time": 9.66989278793335
    },
    {
      "epoch": 1.1491775512695313e-05,
      "grad_norm": 0.07646805793046951,
      "learning_rate": 5.227408002058882e-07,
      "loss": 0.0526,
      "step": 86050
    },
    {
      "epoch": 1.1491775512695313e-05,
      "model_forward_time": 1.9031744003295898,
      "step": 86050
    },
    {
      "dynamics_loss_avg": 0.06281369049102067,
      "epoch": 1.1491775512695313e-05,
      "step": 86050
    },
    {
      "action_loss_avg": 0.0011766706185881049,
      "epoch": 1.1491775512695313e-05,
      "step": 86050
    },
    {
      "epoch": 1.1491775512695313e-05,
      "step": 86050,
      "training_step_time": 8.46443510055542
    },
    {
      "epoch": 1.1492729187011719e-05,
      "model_forward_time": 1.8725183010101318,
      "step": 86051
    },
    {
      "epoch": 1.1492729187011719e-05,
      "step": 86051,
      "training_step_time": 7.287554979324341
    },
    {
      "epoch": 1.1493682861328125e-05,
      "model_forward_time": 1.9248933792114258,
      "step": 86052
    },
    {
      "epoch": 1.1493682861328125e-05,
      "step": 86052,
      "training_step_time": 7.760373115539551
    },
    {
      "epoch": 1.1494636535644532e-05,
      "model_forward_time": 1.9011428356170654,
      "step": 86053
    },
    {
      "epoch": 1.1494636535644532e-05,
      "step": 86053,
      "training_step_time": 6.835677623748779
    },
    {
      "epoch": 1.1495590209960938e-05,
      "model_forward_time": 1.9141850471496582,
      "step": 86054
    },
    {
      "epoch": 1.1495590209960938e-05,
      "step": 86054,
      "training_step_time": 6.8433191776275635
    },
    {
      "epoch": 1.1496543884277343e-05,
      "model_forward_time": 1.9213347434997559,
      "step": 86055
    },
    {
      "epoch": 1.1496543884277343e-05,
      "step": 86055,
      "training_step_time": 6.905949354171753
    },
    {
      "epoch": 1.149749755859375e-05,
      "model_forward_time": 1.9000403881072998,
      "step": 86056
    },
    {
      "epoch": 1.149749755859375e-05,
      "step": 86056,
      "training_step_time": 6.822183609008789
    },
    {
      "epoch": 1.1498451232910157e-05,
      "model_forward_time": 2.5456645488739014,
      "step": 86057
    },
    {
      "epoch": 1.1498451232910157e-05,
      "step": 86057,
      "training_step_time": 7.829615116119385
    },
    {
      "epoch": 1.1499404907226562e-05,
      "model_forward_time": 1.9202539920806885,
      "step": 86058
    },
    {
      "epoch": 1.1499404907226562e-05,
      "step": 86058,
      "training_step_time": 10.54737901687622
    },
    {
      "epoch": 1.1500358581542968e-05,
      "model_forward_time": 1.9176852703094482,
      "step": 86059
    },
    {
      "epoch": 1.1500358581542968e-05,
      "step": 86059,
      "training_step_time": 9.410091161727905
    },
    {
      "epoch": 1.1501312255859375e-05,
      "grad_norm": 0.08347907662391663,
      "learning_rate": 5.220049890517126e-07,
      "loss": 0.0531,
      "step": 86060
    },
    {
      "epoch": 1.1501312255859375e-05,
      "model_forward_time": 1.9290878772735596,
      "step": 86060
    },
    {
      "dynamics_loss_avg": 0.056356780044734475,
      "epoch": 1.1501312255859375e-05,
      "step": 86060
    },
    {
      "action_loss_avg": 0.0011254013908910566,
      "epoch": 1.1501312255859375e-05,
      "step": 86060
    },
    {
      "epoch": 1.1501312255859375e-05,
      "step": 86060,
      "training_step_time": 7.698753833770752
    },
    {
      "epoch": 1.1502265930175781e-05,
      "model_forward_time": 1.9155457019805908,
      "step": 86061
    },
    {
      "epoch": 1.1502265930175781e-05,
      "step": 86061,
      "training_step_time": 7.745823621749878
    },
    {
      "epoch": 1.1503219604492187e-05,
      "model_forward_time": 1.915621042251587,
      "step": 86062
    },
    {
      "epoch": 1.1503219604492187e-05,
      "step": 86062,
      "training_step_time": 7.246878623962402
    },
    {
      "epoch": 1.1504173278808594e-05,
      "model_forward_time": 1.9244697093963623,
      "step": 86063
    },
    {
      "epoch": 1.1504173278808594e-05,
      "step": 86063,
      "training_step_time": 6.8974549770355225
    },
    {
      "epoch": 1.1505126953125e-05,
      "model_forward_time": 1.9107930660247803,
      "step": 86064
    },
    {
      "epoch": 1.1505126953125e-05,
      "step": 86064,
      "training_step_time": 6.947525978088379
    },
    {
      "epoch": 1.1506080627441406e-05,
      "model_forward_time": 1.9419505596160889,
      "step": 86065
    },
    {
      "epoch": 1.1506080627441406e-05,
      "step": 86065,
      "training_step_time": 6.901368856430054
    },
    {
      "epoch": 1.1507034301757813e-05,
      "model_forward_time": 2.144235134124756,
      "step": 86066
    },
    {
      "epoch": 1.1507034301757813e-05,
      "step": 86066,
      "training_step_time": 8.3624906539917
    },
    {
      "epoch": 1.1507987976074219e-05,
      "model_forward_time": 1.9116170406341553,
      "step": 86067
    },
    {
      "epoch": 1.1507987976074219e-05,
      "step": 86067,
      "training_step_time": 10.196890354156494
    },
    {
      "epoch": 1.1508941650390625e-05,
      "model_forward_time": 1.9202942848205566,
      "step": 86068
    },
    {
      "epoch": 1.1508941650390625e-05,
      "step": 86068,
      "training_step_time": 8.716184139251709
    },
    {
      "epoch": 1.1509895324707032e-05,
      "model_forward_time": 1.9108834266662598,
      "step": 86069
    },
    {
      "epoch": 1.1509895324707032e-05,
      "step": 86069,
      "training_step_time": 7.699876308441162
    },
    {
      "epoch": 1.1510848999023438e-05,
      "grad_norm": 0.10184947401285172,
      "learning_rate": 5.212696676043378e-07,
      "loss": 0.0526,
      "step": 86070
    },
    {
      "epoch": 1.1510848999023438e-05,
      "model_forward_time": 1.9102840423583984,
      "step": 86070
    },
    {
      "dynamics_loss_avg": 0.05356593173928559,
      "epoch": 1.1510848999023438e-05,
      "step": 86070
    },
    {
      "action_loss_avg": 0.000997768295928836,
      "epoch": 1.1510848999023438e-05,
      "step": 86070
    },
    {
      "epoch": 1.1510848999023438e-05,
      "step": 86070,
      "training_step_time": 7.729454278945923
    },
    {
      "epoch": 1.1511802673339843e-05,
      "model_forward_time": 1.9255175590515137,
      "step": 86071
    },
    {
      "epoch": 1.1511802673339843e-05,
      "step": 86071,
      "training_step_time": 7.246954679489136
    },
    {
      "epoch": 1.151275634765625e-05,
      "model_forward_time": 1.9068751335144043,
      "step": 86072
    },
    {
      "epoch": 1.151275634765625e-05,
      "step": 86072,
      "training_step_time": 6.875763893127441
    },
    {
      "epoch": 1.1513710021972657e-05,
      "model_forward_time": 1.9019145965576172,
      "step": 86073
    },
    {
      "epoch": 1.1513710021972657e-05,
      "step": 86073,
      "training_step_time": 6.85958456993103
    },
    {
      "epoch": 1.1514663696289062e-05,
      "model_forward_time": 1.9018735885620117,
      "step": 86074
    },
    {
      "epoch": 1.1514663696289062e-05,
      "step": 86074,
      "training_step_time": 6.887784719467163
    },
    {
      "epoch": 1.1515617370605468e-05,
      "model_forward_time": 2.5198471546173096,
      "step": 86075
    },
    {
      "epoch": 1.1515617370605468e-05,
      "step": 86075,
      "training_step_time": 7.826782941818237
    },
    {
      "epoch": 1.1516571044921875e-05,
      "model_forward_time": 1.5395793914794922,
      "step": 86076
    },
    {
      "epoch": 1.1516571044921875e-05,
      "step": 86076,
      "training_step_time": 11.269972801208496
    },
    {
      "epoch": 1.1517524719238281e-05,
      "model_forward_time": 1.9192516803741455,
      "step": 86077
    },
    {
      "epoch": 1.1517524719238281e-05,
      "step": 86077,
      "training_step_time": 8.966455459594727
    },
    {
      "epoch": 1.1518478393554687e-05,
      "model_forward_time": 1.9192466735839844,
      "step": 86078
    },
    {
      "epoch": 1.1518478393554687e-05,
      "step": 86078,
      "training_step_time": 7.569398403167725
    },
    {
      "epoch": 1.1519432067871094e-05,
      "model_forward_time": 1.9218575954437256,
      "step": 86079
    },
    {
      "epoch": 1.1519432067871094e-05,
      "step": 86079,
      "training_step_time": 8.837041139602661
    },
    {
      "epoch": 1.15203857421875e-05,
      "grad_norm": 0.08548449724912643,
      "learning_rate": 5.205348359441798e-07,
      "loss": 0.0533,
      "step": 86080
    },
    {
      "epoch": 1.15203857421875e-05,
      "model_forward_time": 1.9040615558624268,
      "step": 86080
    },
    {
      "dynamics_loss_avg": 0.04830972161144018,
      "epoch": 1.15203857421875e-05,
      "step": 86080
    },
    {
      "action_loss_avg": 0.00135819535353221,
      "epoch": 1.15203857421875e-05,
      "step": 86080
    },
    {
      "epoch": 1.15203857421875e-05,
      "step": 86080,
      "training_step_time": 7.264892816543579
    },
    {
      "epoch": 1.1521339416503906e-05,
      "model_forward_time": 1.8993923664093018,
      "step": 86081
    },
    {
      "epoch": 1.1521339416503906e-05,
      "step": 86081,
      "training_step_time": 6.89015007019043
    },
    {
      "epoch": 1.1522293090820313e-05,
      "model_forward_time": 1.9243314266204834,
      "step": 86082
    },
    {
      "epoch": 1.1522293090820313e-05,
      "step": 86082,
      "training_step_time": 6.862457513809204
    },
    {
      "epoch": 1.1523246765136719e-05,
      "model_forward_time": 1.931713342666626,
      "step": 86083
    },
    {
      "epoch": 1.1523246765136719e-05,
      "step": 86083,
      "training_step_time": 6.870786666870117
    },
    {
      "epoch": 1.1524200439453125e-05,
      "model_forward_time": 1.9177758693695068,
      "step": 86084
    },
    {
      "epoch": 1.1524200439453125e-05,
      "step": 86084,
      "training_step_time": 8.665591955184937
    },
    {
      "epoch": 1.1525154113769532e-05,
      "model_forward_time": 2.5475103855133057,
      "step": 86085
    },
    {
      "epoch": 1.1525154113769532e-05,
      "step": 86085,
      "training_step_time": 9.563724994659424
    },
    {
      "epoch": 1.1526107788085938e-05,
      "model_forward_time": 1.9113836288452148,
      "step": 86086
    },
    {
      "epoch": 1.1526107788085938e-05,
      "step": 86086,
      "training_step_time": 9.096704006195068
    },
    {
      "epoch": 1.1527061462402344e-05,
      "model_forward_time": 1.8821570873260498,
      "step": 86087
    },
    {
      "epoch": 1.1527061462402344e-05,
      "step": 86087,
      "training_step_time": 8.866913557052612
    },
    {
      "epoch": 1.152801513671875e-05,
      "model_forward_time": 1.902054786682129,
      "step": 86088
    },
    {
      "epoch": 1.152801513671875e-05,
      "step": 86088,
      "training_step_time": 7.6499528884887695
    },
    {
      "epoch": 1.1528968811035157e-05,
      "model_forward_time": 1.918950080871582,
      "step": 86089
    },
    {
      "epoch": 1.1528968811035157e-05,
      "step": 86089,
      "training_step_time": 7.827601432800293
    },
    {
      "epoch": 1.1529922485351562e-05,
      "grad_norm": 0.08402954787015915,
      "learning_rate": 5.198004941515966e-07,
      "loss": 0.0544,
      "step": 86090
    },
    {
      "epoch": 1.1529922485351562e-05,
      "model_forward_time": 1.8916358947753906,
      "step": 86090
    },
    {
      "dynamics_loss_avg": 0.0402378624305129,
      "epoch": 1.1529922485351562e-05,
      "step": 86090
    },
    {
      "action_loss_avg": 0.001206982904113829,
      "epoch": 1.1529922485351562e-05,
      "step": 86090
    },
    {
      "epoch": 1.1529922485351562e-05,
      "step": 86090,
      "training_step_time": 6.963995695114136
    },
    {
      "epoch": 1.1530876159667968e-05,
      "model_forward_time": 1.9125092029571533,
      "step": 86091
    },
    {
      "epoch": 1.1530876159667968e-05,
      "step": 86091,
      "training_step_time": 6.967221260070801
    },
    {
      "epoch": 1.1531829833984376e-05,
      "model_forward_time": 1.9129891395568848,
      "step": 86092
    },
    {
      "epoch": 1.1531829833984376e-05,
      "step": 86092,
      "training_step_time": 6.933310270309448
    },
    {
      "epoch": 1.1532783508300781e-05,
      "model_forward_time": 1.9078354835510254,
      "step": 86093
    },
    {
      "epoch": 1.1532783508300781e-05,
      "step": 86093,
      "training_step_time": 7.74364972114563
    },
    {
      "epoch": 1.1533737182617187e-05,
      "model_forward_time": 2.5343713760375977,
      "step": 86094
    },
    {
      "epoch": 1.1533737182617187e-05,
      "step": 86094,
      "training_step_time": 9.615869522094727
    },
    {
      "epoch": 1.1534690856933594e-05,
      "model_forward_time": 1.9167962074279785,
      "step": 86095
    },
    {
      "epoch": 1.1534690856933594e-05,
      "step": 86095,
      "training_step_time": 11.488866329193115
    },
    {
      "epoch": 1.153564453125e-05,
      "model_forward_time": 1.906785249710083,
      "step": 86096
    },
    {
      "epoch": 1.153564453125e-05,
      "step": 86096,
      "training_step_time": 10.23285174369812
    },
    {
      "epoch": 1.1536598205566406e-05,
      "model_forward_time": 1.9189221858978271,
      "step": 86097
    },
    {
      "epoch": 1.1536598205566406e-05,
      "step": 86097,
      "training_step_time": 7.671161651611328
    },
    {
      "epoch": 1.1537551879882813e-05,
      "model_forward_time": 1.916698694229126,
      "step": 86098
    },
    {
      "epoch": 1.1537551879882813e-05,
      "step": 86098,
      "training_step_time": 7.900082111358643
    },
    {
      "epoch": 1.1538505554199219e-05,
      "model_forward_time": 1.917309284210205,
      "step": 86099
    },
    {
      "epoch": 1.1538505554199219e-05,
      "step": 86099,
      "training_step_time": 6.846694469451904
    },
    {
      "epoch": 1.1539459228515625e-05,
      "grad_norm": 0.18830472230911255,
      "learning_rate": 5.190666423068952e-07,
      "loss": 0.0556,
      "step": 86100
    },
    {
      "epoch": 1.1539459228515625e-05,
      "model_forward_time": 1.9205355644226074,
      "step": 86100
    },
    {
      "dynamics_loss_avg": 0.04269611369818449,
      "epoch": 1.1539459228515625e-05,
      "step": 86100
    },
    {
      "action_loss_avg": 0.0010708625137340277,
      "epoch": 1.1539459228515625e-05,
      "step": 86100
    },
    {
      "epoch": 1.1539459228515625e-05,
      "step": 86100,
      "training_step_time": 6.814691781997681
    },
    {
      "epoch": 1.1540412902832032e-05,
      "model_forward_time": 1.9061253070831299,
      "step": 86101
    },
    {
      "epoch": 1.1540412902832032e-05,
      "step": 86101,
      "training_step_time": 6.814474582672119
    },
    {
      "epoch": 1.1541366577148438e-05,
      "model_forward_time": 1.9081759452819824,
      "step": 86102
    },
    {
      "epoch": 1.1541366577148438e-05,
      "step": 86102,
      "training_step_time": 6.813515901565552
    },
    {
      "epoch": 1.1542320251464844e-05,
      "model_forward_time": 2.537997245788574,
      "step": 86103
    },
    {
      "epoch": 1.1542320251464844e-05,
      "step": 86103,
      "training_step_time": 8.546127557754517
    },
    {
      "epoch": 1.154327392578125e-05,
      "model_forward_time": 1.9208953380584717,
      "step": 86104
    },
    {
      "epoch": 1.154327392578125e-05,
      "step": 86104,
      "training_step_time": 9.97356367111206
    },
    {
      "epoch": 1.1544227600097657e-05,
      "model_forward_time": 1.9231266975402832,
      "step": 86105
    },
    {
      "epoch": 1.1544227600097657e-05,
      "step": 86105,
      "training_step_time": 8.606697082519531
    },
    {
      "epoch": 1.1545181274414062e-05,
      "model_forward_time": 1.9267113208770752,
      "step": 86106
    },
    {
      "epoch": 1.1545181274414062e-05,
      "step": 86106,
      "training_step_time": 8.571306228637695
    },
    {
      "epoch": 1.1546134948730468e-05,
      "model_forward_time": 1.9235570430755615,
      "step": 86107
    },
    {
      "epoch": 1.1546134948730468e-05,
      "step": 86107,
      "training_step_time": 7.4091997146606445
    },
    {
      "epoch": 1.1547088623046876e-05,
      "model_forward_time": 1.9208338260650635,
      "step": 86108
    },
    {
      "epoch": 1.1547088623046876e-05,
      "step": 86108,
      "training_step_time": 7.829010963439941
    },
    {
      "epoch": 1.1548042297363281e-05,
      "model_forward_time": 1.9187355041503906,
      "step": 86109
    },
    {
      "epoch": 1.1548042297363281e-05,
      "step": 86109,
      "training_step_time": 6.925672769546509
    },
    {
      "epoch": 1.1548995971679687e-05,
      "grad_norm": 0.1115846261382103,
      "learning_rate": 5.183332804903285e-07,
      "loss": 0.054,
      "step": 86110
    },
    {
      "epoch": 1.1548995971679687e-05,
      "model_forward_time": 1.938206672668457,
      "step": 86110
    },
    {
      "dynamics_loss_avg": 0.04491452304646373,
      "epoch": 1.1548995971679687e-05,
      "step": 86110
    },
    {
      "action_loss_avg": 0.0009736337058711797,
      "epoch": 1.1548995971679687e-05,
      "step": 86110
    },
    {
      "epoch": 1.1548995971679687e-05,
      "step": 86110,
      "training_step_time": 6.902125835418701
    },
    {
      "epoch": 1.1549949645996094e-05,
      "model_forward_time": 1.9266130924224854,
      "step": 86111
    },
    {
      "epoch": 1.1549949645996094e-05,
      "step": 86111,
      "training_step_time": 6.979382038116455
    },
    {
      "epoch": 1.15509033203125e-05,
      "model_forward_time": 2.53823184967041,
      "step": 86112
    },
    {
      "epoch": 1.15509033203125e-05,
      "step": 86112,
      "training_step_time": 9.819185256958008
    },
    {
      "epoch": 1.1551856994628906e-05,
      "model_forward_time": 1.8886032104492188,
      "step": 86113
    },
    {
      "epoch": 1.1551856994628906e-05,
      "step": 86113,
      "training_step_time": 7.909173965454102
    },
    {
      "epoch": 1.1552810668945313e-05,
      "model_forward_time": 1.9141051769256592,
      "step": 86114
    },
    {
      "epoch": 1.1552810668945313e-05,
      "step": 86114,
      "training_step_time": 9.965914964675903
    },
    {
      "epoch": 1.1553764343261719e-05,
      "model_forward_time": 1.9127929210662842,
      "step": 86115
    },
    {
      "epoch": 1.1553764343261719e-05,
      "step": 86115,
      "training_step_time": 7.700679302215576
    },
    {
      "epoch": 1.1554718017578125e-05,
      "model_forward_time": 1.9202187061309814,
      "step": 86116
    },
    {
      "epoch": 1.1554718017578125e-05,
      "step": 86116,
      "training_step_time": 7.546308755874634
    },
    {
      "epoch": 1.1555671691894532e-05,
      "model_forward_time": 1.7606885433197021,
      "step": 86117
    },
    {
      "epoch": 1.1555671691894532e-05,
      "step": 86117,
      "training_step_time": 7.547071695327759
    },
    {
      "epoch": 1.1556625366210938e-05,
      "model_forward_time": 1.7720720767974854,
      "step": 86118
    },
    {
      "epoch": 1.1556625366210938e-05,
      "step": 86118,
      "training_step_time": 6.75465202331543
    },
    {
      "epoch": 1.1557579040527344e-05,
      "model_forward_time": 1.7824008464813232,
      "step": 86119
    },
    {
      "epoch": 1.1557579040527344e-05,
      "step": 86119,
      "training_step_time": 6.805041313171387
    },
    {
      "epoch": 1.155853271484375e-05,
      "grad_norm": 0.12117083370685577,
      "learning_rate": 5.176004087820979e-07,
      "loss": 0.0552,
      "step": 86120
    },
    {
      "epoch": 1.155853271484375e-05,
      "model_forward_time": 1.7672650814056396,
      "step": 86120
    },
    {
      "dynamics_loss_avg": 0.05067800022661686,
      "epoch": 1.155853271484375e-05,
      "step": 86120
    },
    {
      "action_loss_avg": 0.0014424130640691146,
      "epoch": 1.155853271484375e-05,
      "step": 86120
    },
    {
      "epoch": 1.155853271484375e-05,
      "step": 86120,
      "training_step_time": 6.763183832168579
    },
    {
      "epoch": 1.1559486389160157e-05,
      "model_forward_time": 1.7857789993286133,
      "step": 86121
    },
    {
      "epoch": 1.1559486389160157e-05,
      "step": 86121,
      "training_step_time": 8.726327896118164
    },
    {
      "epoch": 1.1560440063476562e-05,
      "model_forward_time": 1.763714075088501,
      "step": 86122
    },
    {
      "epoch": 1.1560440063476562e-05,
      "step": 86122,
      "training_step_time": 8.414786100387573
    },
    {
      "epoch": 1.1561393737792968e-05,
      "model_forward_time": 1.757922887802124,
      "step": 86123
    },
    {
      "epoch": 1.1561393737792968e-05,
      "step": 86123,
      "training_step_time": 10.154279470443726
    },
    {
      "epoch": 1.1562347412109376e-05,
      "model_forward_time": 1.777420997619629,
      "step": 86124
    },
    {
      "epoch": 1.1562347412109376e-05,
      "step": 86124,
      "training_step_time": 7.67763090133667
    },
    {
      "epoch": 1.1563301086425781e-05,
      "model_forward_time": 1.825502634048462,
      "step": 86125
    },
    {
      "epoch": 1.1563301086425781e-05,
      "step": 86125,
      "training_step_time": 7.517888069152832
    },
    {
      "epoch": 1.1564254760742187e-05,
      "model_forward_time": 1.7817859649658203,
      "step": 86126
    },
    {
      "epoch": 1.1564254760742187e-05,
      "step": 86126,
      "training_step_time": 7.623952150344849
    },
    {
      "epoch": 1.1565208435058594e-05,
      "model_forward_time": 1.7748289108276367,
      "step": 86127
    },
    {
      "epoch": 1.1565208435058594e-05,
      "step": 86127,
      "training_step_time": 6.785319805145264
    },
    {
      "epoch": 1.1566162109375e-05,
      "model_forward_time": 1.7802417278289795,
      "step": 86128
    },
    {
      "epoch": 1.1566162109375e-05,
      "step": 86128,
      "training_step_time": 6.834654092788696
    },
    {
      "epoch": 1.1567115783691406e-05,
      "model_forward_time": 1.7665669918060303,
      "step": 86129
    },
    {
      "epoch": 1.1567115783691406e-05,
      "step": 86129,
      "training_step_time": 6.799847364425659
    },
    {
      "epoch": 1.1568069458007813e-05,
      "grad_norm": 0.07472189515829086,
      "learning_rate": 5.168680272623455e-07,
      "loss": 0.0534,
      "step": 86130
    },
    {
      "epoch": 1.1568069458007813e-05,
      "model_forward_time": 1.7698240280151367,
      "step": 86130
    },
    {
      "dynamics_loss_avg": 0.04879280235618353,
      "epoch": 1.1568069458007813e-05,
      "step": 86130
    },
    {
      "action_loss_avg": 0.0009402887691976503,
      "epoch": 1.1568069458007813e-05,
      "step": 86130
    },
    {
      "epoch": 1.1568069458007813e-05,
      "step": 86130,
      "training_step_time": 7.596083641052246
    },
    {
      "epoch": 1.1569023132324219e-05,
      "model_forward_time": 2.5264699459075928,
      "step": 86131
    },
    {
      "epoch": 1.1569023132324219e-05,
      "step": 86131,
      "training_step_time": 9.333305358886719
    },
    {
      "epoch": 1.1569976806640625e-05,
      "model_forward_time": 1.7691187858581543,
      "step": 86132
    },
    {
      "epoch": 1.1569976806640625e-05,
      "step": 86132,
      "training_step_time": 8.583794593811035
    },
    {
      "epoch": 1.157093048095703e-05,
      "model_forward_time": 1.7733266353607178,
      "step": 86133
    },
    {
      "epoch": 1.157093048095703e-05,
      "step": 86133,
      "training_step_time": 7.791579484939575
    },
    {
      "epoch": 1.1571884155273438e-05,
      "model_forward_time": 1.7639834880828857,
      "step": 86134
    },
    {
      "epoch": 1.1571884155273438e-05,
      "step": 86134,
      "training_step_time": 8.449339866638184
    },
    {
      "epoch": 1.1572837829589844e-05,
      "model_forward_time": 1.7770261764526367,
      "step": 86135
    },
    {
      "epoch": 1.1572837829589844e-05,
      "step": 86135,
      "training_step_time": 7.59466552734375
    },
    {
      "epoch": 1.157379150390625e-05,
      "model_forward_time": 1.7720897197723389,
      "step": 86136
    },
    {
      "epoch": 1.157379150390625e-05,
      "step": 86136,
      "training_step_time": 6.755275011062622
    },
    {
      "epoch": 1.1574745178222657e-05,
      "model_forward_time": 1.784315586090088,
      "step": 86137
    },
    {
      "epoch": 1.1574745178222657e-05,
      "step": 86137,
      "training_step_time": 6.769015312194824
    },
    {
      "epoch": 1.1575698852539062e-05,
      "model_forward_time": 1.7721245288848877,
      "step": 86138
    },
    {
      "epoch": 1.1575698852539062e-05,
      "step": 86138,
      "training_step_time": 6.804823160171509
    },
    {
      "epoch": 1.1576652526855468e-05,
      "model_forward_time": 1.7750763893127441,
      "step": 86139
    },
    {
      "epoch": 1.1576652526855468e-05,
      "step": 86139,
      "training_step_time": 7.511810541152954
    },
    {
      "epoch": 1.1577606201171876e-05,
      "grad_norm": 0.07690086215734482,
      "learning_rate": 5.16136136011166e-07,
      "loss": 0.0549,
      "step": 86140
    },
    {
      "epoch": 1.1577606201171876e-05,
      "model_forward_time": 2.510685443878174,
      "step": 86140
    },
    {
      "dynamics_loss_avg": 0.04254697617143392,
      "epoch": 1.1577606201171876e-05,
      "step": 86140
    },
    {
      "action_loss_avg": 0.0010413325595436617,
      "epoch": 1.1577606201171876e-05,
      "step": 86140
    },
    {
      "epoch": 1.1577606201171876e-05,
      "step": 86140,
      "training_step_time": 8.249636888504028
    },
    {
      "epoch": 1.1578559875488281e-05,
      "model_forward_time": 1.3789680004119873,
      "step": 86141
    },
    {
      "epoch": 1.1578559875488281e-05,
      "step": 86141,
      "training_step_time": 10.429018497467041
    },
    {
      "epoch": 1.1579513549804687e-05,
      "model_forward_time": 1.7780380249023438,
      "step": 86142
    },
    {
      "epoch": 1.1579513549804687e-05,
      "step": 86142,
      "training_step_time": 8.72632360458374
    },
    {
      "epoch": 1.1580467224121094e-05,
      "model_forward_time": 1.7899248600006104,
      "step": 86143
    },
    {
      "epoch": 1.1580467224121094e-05,
      "step": 86143,
      "training_step_time": 7.63264274597168
    },
    {
      "epoch": 1.15814208984375e-05,
      "model_forward_time": 1.7713391780853271,
      "step": 86144
    },
    {
      "epoch": 1.15814208984375e-05,
      "step": 86144,
      "training_step_time": 7.558711767196655
    },
    {
      "epoch": 1.1582374572753906e-05,
      "model_forward_time": 1.7895255088806152,
      "step": 86145
    },
    {
      "epoch": 1.1582374572753906e-05,
      "step": 86145,
      "training_step_time": 7.636005878448486
    },
    {
      "epoch": 1.1583328247070313e-05,
      "model_forward_time": 1.7717413902282715,
      "step": 86146
    },
    {
      "epoch": 1.1583328247070313e-05,
      "step": 86146,
      "training_step_time": 6.78578519821167
    },
    {
      "epoch": 1.1584281921386719e-05,
      "model_forward_time": 1.777350664138794,
      "step": 86147
    },
    {
      "epoch": 1.1584281921386719e-05,
      "step": 86147,
      "training_step_time": 6.731336832046509
    },
    {
      "epoch": 1.1585235595703125e-05,
      "model_forward_time": 1.7918341159820557,
      "step": 86148
    },
    {
      "epoch": 1.1585235595703125e-05,
      "step": 86148,
      "training_step_time": 8.168073415756226
    },
    {
      "epoch": 1.158618927001953e-05,
      "model_forward_time": 2.447204828262329,
      "step": 86149
    },
    {
      "epoch": 1.158618927001953e-05,
      "step": 86149,
      "training_step_time": 8.673367738723755
    },
    {
      "epoch": 1.1587142944335938e-05,
      "grad_norm": 0.10895171016454697,
      "learning_rate": 5.154047351085983e-07,
      "loss": 0.055,
      "step": 86150
    },
    {
      "epoch": 1.1587142944335938e-05,
      "model_forward_time": 1.7602031230926514,
      "step": 86150
    },
    {
      "dynamics_loss_avg": 0.05625428073108196,
      "epoch": 1.1587142944335938e-05,
      "step": 86150
    },
    {
      "action_loss_avg": 0.0010683712549507617,
      "epoch": 1.1587142944335938e-05,
      "step": 86150
    },
    {
      "epoch": 1.1587142944335938e-05,
      "step": 86150,
      "training_step_time": 8.67650842666626
    },
    {
      "epoch": 1.1588096618652344e-05,
      "model_forward_time": 1.777193307876587,
      "step": 86151
    },
    {
      "epoch": 1.1588096618652344e-05,
      "step": 86151,
      "training_step_time": 8.679471969604492
    },
    {
      "epoch": 1.158905029296875e-05,
      "model_forward_time": 1.7621917724609375,
      "step": 86152
    },
    {
      "epoch": 1.158905029296875e-05,
      "step": 86152,
      "training_step_time": 7.593151092529297
    },
    {
      "epoch": 1.1590003967285157e-05,
      "model_forward_time": 1.7719359397888184,
      "step": 86153
    },
    {
      "epoch": 1.1590003967285157e-05,
      "step": 86153,
      "training_step_time": 7.488677024841309
    },
    {
      "epoch": 1.1590957641601562e-05,
      "model_forward_time": 1.7826855182647705,
      "step": 86154
    },
    {
      "epoch": 1.1590957641601562e-05,
      "step": 86154,
      "training_step_time": 7.593911170959473
    },
    {
      "epoch": 1.1591911315917968e-05,
      "model_forward_time": 1.7702953815460205,
      "step": 86155
    },
    {
      "epoch": 1.1591911315917968e-05,
      "step": 86155,
      "training_step_time": 6.7632386684417725
    },
    {
      "epoch": 1.1592864990234376e-05,
      "model_forward_time": 1.783050537109375,
      "step": 86156
    },
    {
      "epoch": 1.1592864990234376e-05,
      "step": 86156,
      "training_step_time": 6.695875406265259
    },
    {
      "epoch": 1.1593818664550781e-05,
      "model_forward_time": 1.7676801681518555,
      "step": 86157
    },
    {
      "epoch": 1.1593818664550781e-05,
      "step": 86157,
      "training_step_time": 8.109131813049316
    },
    {
      "epoch": 1.1594772338867187e-05,
      "model_forward_time": 2.398311138153076,
      "step": 86158
    },
    {
      "epoch": 1.1594772338867187e-05,
      "step": 86158,
      "training_step_time": 8.63667607307434
    },
    {
      "epoch": 1.1595726013183594e-05,
      "model_forward_time": 1.87382173538208,
      "step": 86159
    },
    {
      "epoch": 1.1595726013183594e-05,
      "step": 86159,
      "training_step_time": 9.119385719299316
    },
    {
      "epoch": 1.15966796875e-05,
      "grad_norm": 0.0727689117193222,
      "learning_rate": 5.146738246346239e-07,
      "loss": 0.0551,
      "step": 86160
    },
    {
      "epoch": 1.15966796875e-05,
      "model_forward_time": 1.7720613479614258,
      "step": 86160
    },
    {
      "dynamics_loss_avg": 0.06155268149450421,
      "epoch": 1.15966796875e-05,
      "step": 86160
    },
    {
      "action_loss_avg": 0.0014213598740752786,
      "epoch": 1.15966796875e-05,
      "step": 86160
    },
    {
      "epoch": 1.15966796875e-05,
      "step": 86160,
      "training_step_time": 8.547205924987793
    },
    {
      "epoch": 1.1597633361816406e-05,
      "model_forward_time": 1.7557921409606934,
      "step": 86161
    },
    {
      "epoch": 1.1597633361816406e-05,
      "step": 86161,
      "training_step_time": 7.632306098937988
    },
    {
      "epoch": 1.1598587036132813e-05,
      "model_forward_time": 1.7709808349609375,
      "step": 86162
    },
    {
      "epoch": 1.1598587036132813e-05,
      "step": 86162,
      "training_step_time": 9.177329301834106
    },
    {
      "epoch": 1.1599540710449219e-05,
      "model_forward_time": 1.4234957695007324,
      "step": 86163
    },
    {
      "epoch": 1.1599540710449219e-05,
      "step": 86163,
      "training_step_time": 7.455651521682739
    },
    {
      "epoch": 1.1600494384765625e-05,
      "model_forward_time": 1.7756028175354004,
      "step": 86164
    },
    {
      "epoch": 1.1600494384765625e-05,
      "step": 86164,
      "training_step_time": 6.731942176818848
    },
    {
      "epoch": 1.160144805908203e-05,
      "model_forward_time": 1.758929967880249,
      "step": 86165
    },
    {
      "epoch": 1.160144805908203e-05,
      "step": 86165,
      "training_step_time": 6.6931984424591064
    },
    {
      "epoch": 1.1602401733398438e-05,
      "model_forward_time": 1.7695841789245605,
      "step": 86166
    },
    {
      "epoch": 1.1602401733398438e-05,
      "step": 86166,
      "training_step_time": 7.439093589782715
    },
    {
      "epoch": 1.1603355407714844e-05,
      "model_forward_time": 1.7704949378967285,
      "step": 86167
    },
    {
      "epoch": 1.1603355407714844e-05,
      "step": 86167,
      "training_step_time": 7.616286754608154
    },
    {
      "epoch": 1.160430908203125e-05,
      "model_forward_time": 2.5170347690582275,
      "step": 86168
    },
    {
      "epoch": 1.160430908203125e-05,
      "step": 86168,
      "training_step_time": 8.298393726348877
    },
    {
      "epoch": 1.1605262756347657e-05,
      "model_forward_time": 1.7692539691925049,
      "step": 86169
    },
    {
      "epoch": 1.1605262756347657e-05,
      "step": 86169,
      "training_step_time": 7.691946744918823
    },
    {
      "epoch": 1.1606216430664062e-05,
      "grad_norm": 0.10403347760438919,
      "learning_rate": 5.139434046691788e-07,
      "loss": 0.0557,
      "step": 86170
    },
    {
      "epoch": 1.1606216430664062e-05,
      "model_forward_time": 1.3964695930480957,
      "step": 86170
    },
    {
      "dynamics_loss_avg": 0.06711228173226118,
      "epoch": 1.1606216430664062e-05,
      "step": 86170
    },
    {
      "action_loss_avg": 0.0016770320042269304,
      "epoch": 1.1606216430664062e-05,
      "step": 86170
    },
    {
      "epoch": 1.1606216430664062e-05,
      "step": 86170,
      "training_step_time": 7.655282974243164
    },
    {
      "epoch": 1.1607170104980468e-05,
      "model_forward_time": 1.766791582107544,
      "step": 86171
    },
    {
      "epoch": 1.1607170104980468e-05,
      "step": 86171,
      "training_step_time": 7.321104526519775
    },
    {
      "epoch": 1.1608123779296876e-05,
      "model_forward_time": 1.7694647312164307,
      "step": 86172
    },
    {
      "epoch": 1.1608123779296876e-05,
      "step": 86172,
      "training_step_time": 7.6243579387664795
    },
    {
      "epoch": 1.1609077453613281e-05,
      "model_forward_time": 1.762016773223877,
      "step": 86173
    },
    {
      "epoch": 1.1609077453613281e-05,
      "step": 86173,
      "training_step_time": 6.730939865112305
    },
    {
      "epoch": 1.1610031127929687e-05,
      "model_forward_time": 1.7960395812988281,
      "step": 86174
    },
    {
      "epoch": 1.1610031127929687e-05,
      "step": 86174,
      "training_step_time": 6.7841901779174805
    },
    {
      "epoch": 1.1610984802246094e-05,
      "model_forward_time": 1.7800304889678955,
      "step": 86175
    },
    {
      "epoch": 1.1610984802246094e-05,
      "step": 86175,
      "training_step_time": 6.8035993576049805
    },
    {
      "epoch": 1.16119384765625e-05,
      "model_forward_time": 1.8005120754241943,
      "step": 86176
    },
    {
      "epoch": 1.16119384765625e-05,
      "step": 86176,
      "training_step_time": 8.4377601146698
    },
    {
      "epoch": 1.1612892150878906e-05,
      "model_forward_time": 2.4239423274993896,
      "step": 86177
    },
    {
      "epoch": 1.1612892150878906e-05,
      "step": 86177,
      "training_step_time": 8.829100131988525
    },
    {
      "epoch": 1.1613845825195313e-05,
      "model_forward_time": 1.7717292308807373,
      "step": 86178
    },
    {
      "epoch": 1.1613845825195313e-05,
      "step": 86178,
      "training_step_time": 9.891264915466309
    },
    {
      "epoch": 1.1614799499511719e-05,
      "model_forward_time": 1.7778525352478027,
      "step": 86179
    },
    {
      "epoch": 1.1614799499511719e-05,
      "step": 86179,
      "training_step_time": 8.554712057113647
    },
    {
      "epoch": 1.1615753173828125e-05,
      "grad_norm": 0.1303047090768814,
      "learning_rate": 5.13213475292137e-07,
      "loss": 0.0557,
      "step": 86180
    },
    {
      "epoch": 1.1615753173828125e-05,
      "model_forward_time": 1.7595534324645996,
      "step": 86180
    },
    {
      "dynamics_loss_avg": 0.06668155323714017,
      "epoch": 1.1615753173828125e-05,
      "step": 86180
    },
    {
      "action_loss_avg": 0.001973624818492681,
      "epoch": 1.1615753173828125e-05,
      "step": 86180
    },
    {
      "epoch": 1.1615753173828125e-05,
      "step": 86180,
      "training_step_time": 7.660543918609619
    },
    {
      "epoch": 1.161670684814453e-05,
      "model_forward_time": 1.7874033451080322,
      "step": 86181
    },
    {
      "epoch": 1.161670684814453e-05,
      "step": 86181,
      "training_step_time": 7.598896503448486
    },
    {
      "epoch": 1.1617660522460938e-05,
      "model_forward_time": 1.7773113250732422,
      "step": 86182
    },
    {
      "epoch": 1.1617660522460938e-05,
      "step": 86182,
      "training_step_time": 6.768473863601685
    },
    {
      "epoch": 1.1618614196777344e-05,
      "model_forward_time": 1.7655084133148193,
      "step": 86183
    },
    {
      "epoch": 1.1618614196777344e-05,
      "step": 86183,
      "training_step_time": 6.802731990814209
    },
    {
      "epoch": 1.161956787109375e-05,
      "model_forward_time": 1.764462947845459,
      "step": 86184
    },
    {
      "epoch": 1.161956787109375e-05,
      "step": 86184,
      "training_step_time": 6.7840986251831055
    },
    {
      "epoch": 1.1620521545410157e-05,
      "model_forward_time": 1.7750985622406006,
      "step": 86185
    },
    {
      "epoch": 1.1620521545410157e-05,
      "step": 86185,
      "training_step_time": 7.741838216781616
    },
    {
      "epoch": 1.1621475219726562e-05,
      "model_forward_time": 2.4336113929748535,
      "step": 86186
    },
    {
      "epoch": 1.1621475219726562e-05,
      "step": 86186,
      "training_step_time": 10.543571710586548
    },
    {
      "epoch": 1.1622428894042968e-05,
      "model_forward_time": 1.8738269805908203,
      "step": 86187
    },
    {
      "epoch": 1.1622428894042968e-05,
      "step": 86187,
      "training_step_time": 9.156771183013916
    },
    {
      "epoch": 1.1623382568359376e-05,
      "model_forward_time": 1.7627203464508057,
      "step": 86188
    },
    {
      "epoch": 1.1623382568359376e-05,
      "step": 86188,
      "training_step_time": 9.46934986114502
    },
    {
      "epoch": 1.1624336242675781e-05,
      "model_forward_time": 1.7593028545379639,
      "step": 86189
    },
    {
      "epoch": 1.1624336242675781e-05,
      "step": 86189,
      "training_step_time": 7.61431622505188
    },
    {
      "epoch": 1.1625289916992187e-05,
      "grad_norm": 0.11272542178630829,
      "learning_rate": 5.12484036583325e-07,
      "loss": 0.0537,
      "step": 86190
    },
    {
      "epoch": 1.1625289916992187e-05,
      "model_forward_time": 1.7613754272460938,
      "step": 86190
    },
    {
      "dynamics_loss_avg": 0.049166402965784076,
      "epoch": 1.1625289916992187e-05,
      "step": 86190
    },
    {
      "action_loss_avg": 0.0010429231071611867,
      "epoch": 1.1625289916992187e-05,
      "step": 86190
    },
    {
      "epoch": 1.1625289916992187e-05,
      "step": 86190,
      "training_step_time": 8.150858640670776
    },
    {
      "epoch": 1.1626243591308594e-05,
      "model_forward_time": 1.7922697067260742,
      "step": 86191
    },
    {
      "epoch": 1.1626243591308594e-05,
      "step": 86191,
      "training_step_time": 6.859790325164795
    },
    {
      "epoch": 1.1627197265625e-05,
      "model_forward_time": 1.7844488620758057,
      "step": 86192
    },
    {
      "epoch": 1.1627197265625e-05,
      "step": 86192,
      "training_step_time": 6.746055364608765
    },
    {
      "epoch": 1.1628150939941406e-05,
      "model_forward_time": 1.7665815353393555,
      "step": 86193
    },
    {
      "epoch": 1.1628150939941406e-05,
      "step": 86193,
      "training_step_time": 6.80596137046814
    },
    {
      "epoch": 1.1629104614257813e-05,
      "model_forward_time": 1.764904260635376,
      "step": 86194
    },
    {
      "epoch": 1.1629104614257813e-05,
      "step": 86194,
      "training_step_time": 7.656318664550781
    },
    {
      "epoch": 1.1630058288574219e-05,
      "model_forward_time": 2.3992421627044678,
      "step": 86195
    },
    {
      "epoch": 1.1630058288574219e-05,
      "step": 86195,
      "training_step_time": 7.675186634063721
    },
    {
      "epoch": 1.1631011962890625e-05,
      "model_forward_time": 1.4745700359344482,
      "step": 86196
    },
    {
      "epoch": 1.1631011962890625e-05,
      "step": 86196,
      "training_step_time": 8.443562984466553
    },
    {
      "epoch": 1.163196563720703e-05,
      "model_forward_time": 1.766636848449707,
      "step": 86197
    },
    {
      "epoch": 1.163196563720703e-05,
      "step": 86197,
      "training_step_time": 9.373574256896973
    },
    {
      "epoch": 1.1632919311523438e-05,
      "model_forward_time": 1.7559096813201904,
      "step": 86198
    },
    {
      "epoch": 1.1632919311523438e-05,
      "step": 86198,
      "training_step_time": 7.680848598480225
    },
    {
      "epoch": 1.1633872985839844e-05,
      "model_forward_time": 1.7726356983184814,
      "step": 86199
    },
    {
      "epoch": 1.1633872985839844e-05,
      "step": 86199,
      "training_step_time": 7.268517732620239
    },
    {
      "epoch": 1.163482666015625e-05,
      "grad_norm": 0.0829814076423645,
      "learning_rate": 5.117550886225109e-07,
      "loss": 0.0544,
      "step": 86200
    },
    {
      "epoch": 1.163482666015625e-05,
      "model_forward_time": 1.7558519840240479,
      "step": 86200
    },
    {
      "dynamics_loss_avg": 0.06060017310082912,
      "epoch": 1.163482666015625e-05,
      "step": 86200
    },
    {
      "action_loss_avg": 0.001784380810568109,
      "epoch": 1.163482666015625e-05,
      "step": 86200
    },
    {
      "epoch": 1.163482666015625e-05,
      "step": 86200,
      "training_step_time": 7.62710165977478
    },
    {
      "epoch": 1.1635780334472657e-05,
      "model_forward_time": 1.7760298252105713,
      "step": 86201
    },
    {
      "epoch": 1.1635780334472657e-05,
      "step": 86201,
      "training_step_time": 6.754920721054077
    },
    {
      "epoch": 1.1636734008789062e-05,
      "model_forward_time": 1.0698978900909424,
      "step": 86202
    },
    {
      "epoch": 1.1636734008789062e-05,
      "step": 86202,
      "training_step_time": 6.715783596038818
    },
    {
      "epoch": 1.1637687683105468e-05,
      "model_forward_time": 1.763930320739746,
      "step": 86203
    },
    {
      "epoch": 1.1637687683105468e-05,
      "step": 86203,
      "training_step_time": 8.688176155090332
    },
    {
      "epoch": 1.1638641357421876e-05,
      "model_forward_time": 1.7640249729156494,
      "step": 86204
    },
    {
      "epoch": 1.1638641357421876e-05,
      "step": 86204,
      "training_step_time": 8.80763578414917
    },
    {
      "epoch": 1.1639595031738281e-05,
      "model_forward_time": 2.434130907058716,
      "step": 86205
    },
    {
      "epoch": 1.1639595031738281e-05,
      "step": 86205,
      "training_step_time": 8.438244342803955
    },
    {
      "epoch": 1.1640548706054687e-05,
      "model_forward_time": 1.7631380558013916,
      "step": 86206
    },
    {
      "epoch": 1.1640548706054687e-05,
      "step": 86206,
      "training_step_time": 8.40101957321167
    },
    {
      "epoch": 1.1641502380371094e-05,
      "model_forward_time": 1.7600085735321045,
      "step": 86207
    },
    {
      "epoch": 1.1641502380371094e-05,
      "step": 86207,
      "training_step_time": 8.429178953170776
    },
    {
      "epoch": 1.16424560546875e-05,
      "model_forward_time": 1.770179033279419,
      "step": 86208
    },
    {
      "epoch": 1.16424560546875e-05,
      "step": 86208,
      "training_step_time": 8.292264223098755
    },
    {
      "epoch": 1.1643409729003906e-05,
      "model_forward_time": 1.7627170085906982,
      "step": 86209
    },
    {
      "epoch": 1.1643409729003906e-05,
      "step": 86209,
      "training_step_time": 7.71424674987793
    },
    {
      "epoch": 1.1644363403320313e-05,
      "grad_norm": 0.09722251445055008,
      "learning_rate": 5.11026631489413e-07,
      "loss": 0.0545,
      "step": 86210
    },
    {
      "epoch": 1.1644363403320313e-05,
      "model_forward_time": 1.7638728618621826,
      "step": 86210
    },
    {
      "dynamics_loss_avg": 0.03999135382473469,
      "epoch": 1.1644363403320313e-05,
      "step": 86210
    },
    {
      "action_loss_avg": 0.0008926711918320507,
      "epoch": 1.1644363403320313e-05,
      "step": 86210
    },
    {
      "epoch": 1.1644363403320313e-05,
      "step": 86210,
      "training_step_time": 6.8368918895721436
    },
    {
      "epoch": 1.1645317077636719e-05,
      "model_forward_time": 1.7895455360412598,
      "step": 86211
    },
    {
      "epoch": 1.1645317077636719e-05,
      "step": 86211,
      "training_step_time": 6.767951011657715
    },
    {
      "epoch": 1.1646270751953125e-05,
      "model_forward_time": 1.7783153057098389,
      "step": 86212
    },
    {
      "epoch": 1.1646270751953125e-05,
      "step": 86212,
      "training_step_time": 8.34677529335022
    },
    {
      "epoch": 1.164722442626953e-05,
      "model_forward_time": 1.7905011177062988,
      "step": 86213
    },
    {
      "epoch": 1.164722442626953e-05,
      "step": 86213,
      "training_step_time": 8.608757257461548
    },
    {
      "epoch": 1.1648178100585938e-05,
      "model_forward_time": 1.7833077907562256,
      "step": 86214
    },
    {
      "epoch": 1.1648178100585938e-05,
      "step": 86214,
      "training_step_time": 9.770750045776367
    },
    {
      "epoch": 1.1649131774902344e-05,
      "model_forward_time": 1.770871639251709,
      "step": 86215
    },
    {
      "epoch": 1.1649131774902344e-05,
      "step": 86215,
      "training_step_time": 9.511325120925903
    },
    {
      "epoch": 1.165008544921875e-05,
      "model_forward_time": 1.758777141571045,
      "step": 86216
    },
    {
      "epoch": 1.165008544921875e-05,
      "step": 86216,
      "training_step_time": 8.363338232040405
    },
    {
      "epoch": 1.1651039123535157e-05,
      "model_forward_time": 1.7769370079040527,
      "step": 86217
    },
    {
      "epoch": 1.1651039123535157e-05,
      "step": 86217,
      "training_step_time": 7.696880102157593
    },
    {
      "epoch": 1.1651992797851562e-05,
      "model_forward_time": 1.8408174514770508,
      "step": 86218
    },
    {
      "epoch": 1.1651992797851562e-05,
      "step": 86218,
      "training_step_time": 7.662355661392212
    },
    {
      "epoch": 1.1652946472167968e-05,
      "model_forward_time": 1.7606008052825928,
      "step": 86219
    },
    {
      "epoch": 1.1652946472167968e-05,
      "step": 86219,
      "training_step_time": 6.765965938568115
    },
    {
      "epoch": 1.1653900146484376e-05,
      "grad_norm": 0.07952996343374252,
      "learning_rate": 5.102986652636926e-07,
      "loss": 0.055,
      "step": 86220
    },
    {
      "epoch": 1.1653900146484376e-05,
      "model_forward_time": 1.7586402893066406,
      "step": 86220
    },
    {
      "dynamics_loss_avg": 0.055791429989039895,
      "epoch": 1.1653900146484376e-05,
      "step": 86220
    },
    {
      "action_loss_avg": 0.0017390209279255942,
      "epoch": 1.1653900146484376e-05,
      "step": 86220
    },
    {
      "epoch": 1.1653900146484376e-05,
      "step": 86220,
      "training_step_time": 6.7921061515808105
    },
    {
      "epoch": 1.1654853820800781e-05,
      "model_forward_time": 1.7695708274841309,
      "step": 86221
    },
    {
      "epoch": 1.1654853820800781e-05,
      "step": 86221,
      "training_step_time": 7.64065957069397
    },
    {
      "epoch": 1.1655807495117187e-05,
      "model_forward_time": 1.768355369567871,
      "step": 86222
    },
    {
      "epoch": 1.1655807495117187e-05,
      "step": 86222,
      "training_step_time": 8.857563018798828
    },
    {
      "epoch": 1.1656761169433594e-05,
      "model_forward_time": 1.7745838165283203,
      "step": 86223
    },
    {
      "epoch": 1.1656761169433594e-05,
      "step": 86223,
      "training_step_time": 9.21419882774353
    },
    {
      "epoch": 1.165771484375e-05,
      "model_forward_time": 2.502436637878418,
      "step": 86224
    },
    {
      "epoch": 1.165771484375e-05,
      "step": 86224,
      "training_step_time": 8.666081666946411
    },
    {
      "epoch": 1.1658668518066406e-05,
      "model_forward_time": 1.7657413482666016,
      "step": 86225
    },
    {
      "epoch": 1.1658668518066406e-05,
      "step": 86225,
      "training_step_time": 7.613885402679443
    },
    {
      "epoch": 1.1659622192382813e-05,
      "model_forward_time": 1.7760205268859863,
      "step": 86226
    },
    {
      "epoch": 1.1659622192382813e-05,
      "step": 86226,
      "training_step_time": 7.681700229644775
    },
    {
      "epoch": 1.1660575866699219e-05,
      "model_forward_time": 1.7977440357208252,
      "step": 86227
    },
    {
      "epoch": 1.1660575866699219e-05,
      "step": 86227,
      "training_step_time": 7.653202056884766
    },
    {
      "epoch": 1.1661529541015625e-05,
      "model_forward_time": 1.7864527702331543,
      "step": 86228
    },
    {
      "epoch": 1.1661529541015625e-05,
      "step": 86228,
      "training_step_time": 6.791069984436035
    },
    {
      "epoch": 1.166248321533203e-05,
      "model_forward_time": 1.7762033939361572,
      "step": 86229
    },
    {
      "epoch": 1.166248321533203e-05,
      "step": 86229,
      "training_step_time": 6.783736228942871
    },
    {
      "epoch": 1.1663436889648438e-05,
      "grad_norm": 0.08912340551614761,
      "learning_rate": 5.095711900249605e-07,
      "loss": 0.0533,
      "step": 86230
    },
    {
      "epoch": 1.1663436889648438e-05,
      "model_forward_time": 1.7684407234191895,
      "step": 86230
    },
    {
      "dynamics_loss_avg": 0.05029327040538192,
      "epoch": 1.1663436889648438e-05,
      "step": 86230
    },
    {
      "action_loss_avg": 0.0008319905115058646,
      "epoch": 1.1663436889648438e-05,
      "step": 86230
    },
    {
      "epoch": 1.1663436889648438e-05,
      "step": 86230,
      "training_step_time": 7.637153387069702
    },
    {
      "epoch": 1.1664390563964844e-05,
      "model_forward_time": 1.769219160079956,
      "step": 86231
    },
    {
      "epoch": 1.1664390563964844e-05,
      "step": 86231,
      "training_step_time": 7.711500883102417
    },
    {
      "epoch": 1.166534423828125e-05,
      "model_forward_time": 1.769681453704834,
      "step": 86232
    },
    {
      "epoch": 1.166534423828125e-05,
      "step": 86232,
      "training_step_time": 8.74323558807373
    },
    {
      "epoch": 1.1666297912597657e-05,
      "model_forward_time": 2.4482460021972656,
      "step": 86233
    },
    {
      "epoch": 1.1666297912597657e-05,
      "step": 86233,
      "training_step_time": 8.637003421783447
    },
    {
      "epoch": 1.1667251586914062e-05,
      "model_forward_time": 1.7766938209533691,
      "step": 86234
    },
    {
      "epoch": 1.1667251586914062e-05,
      "step": 86234,
      "training_step_time": 8.680603265762329
    },
    {
      "epoch": 1.1668205261230468e-05,
      "model_forward_time": 1.7691433429718018,
      "step": 86235
    },
    {
      "epoch": 1.1668205261230468e-05,
      "step": 86235,
      "training_step_time": 8.53518795967102
    },
    {
      "epoch": 1.1669158935546876e-05,
      "model_forward_time": 1.7832748889923096,
      "step": 86236
    },
    {
      "epoch": 1.1669158935546876e-05,
      "step": 86236,
      "training_step_time": 7.678788900375366
    },
    {
      "epoch": 1.1670112609863281e-05,
      "model_forward_time": 1.7697029113769531,
      "step": 86237
    },
    {
      "epoch": 1.1670112609863281e-05,
      "step": 86237,
      "training_step_time": 7.177642583847046
    },
    {
      "epoch": 1.1671066284179687e-05,
      "model_forward_time": 1.7919814586639404,
      "step": 86238
    },
    {
      "epoch": 1.1671066284179687e-05,
      "step": 86238,
      "training_step_time": 6.771131992340088
    },
    {
      "epoch": 1.1672019958496094e-05,
      "model_forward_time": 1.7686381340026855,
      "step": 86239
    },
    {
      "epoch": 1.1672019958496094e-05,
      "step": 86239,
      "training_step_time": 7.644150018692017
    },
    {
      "epoch": 1.16729736328125e-05,
      "grad_norm": 0.09110178053379059,
      "learning_rate": 5.088442058527715e-07,
      "loss": 0.0546,
      "step": 86240
    },
    {
      "epoch": 1.16729736328125e-05,
      "model_forward_time": 1.7843332290649414,
      "step": 86240
    },
    {
      "dynamics_loss_avg": 0.05471490193158388,
      "epoch": 1.16729736328125e-05,
      "step": 86240
    },
    {
      "action_loss_avg": 0.0017439483606722205,
      "epoch": 1.16729736328125e-05,
      "step": 86240
    },
    {
      "epoch": 1.16729736328125e-05,
      "step": 86240,
      "training_step_time": 7.604156494140625
    },
    {
      "epoch": 1.1673927307128906e-05,
      "model_forward_time": 1.7687032222747803,
      "step": 86241
    },
    {
      "epoch": 1.1673927307128906e-05,
      "step": 86241,
      "training_step_time": 9.371658325195312
    },
    {
      "epoch": 1.1674880981445313e-05,
      "model_forward_time": 2.445667266845703,
      "step": 86242
    },
    {
      "epoch": 1.1674880981445313e-05,
      "step": 86242,
      "training_step_time": 7.764658689498901
    },
    {
      "epoch": 1.1675834655761719e-05,
      "model_forward_time": 1.7718865871429443,
      "step": 86243
    },
    {
      "epoch": 1.1675834655761719e-05,
      "step": 86243,
      "training_step_time": 9.141418933868408
    },
    {
      "epoch": 1.1676788330078125e-05,
      "model_forward_time": 1.7758231163024902,
      "step": 86244
    },
    {
      "epoch": 1.1676788330078125e-05,
      "step": 86244,
      "training_step_time": 8.230790615081787
    },
    {
      "epoch": 1.167774200439453e-05,
      "model_forward_time": 1.7645907402038574,
      "step": 86245
    },
    {
      "epoch": 1.167774200439453e-05,
      "step": 86245,
      "training_step_time": 7.619744062423706
    },
    {
      "epoch": 1.1678695678710938e-05,
      "model_forward_time": 1.7819535732269287,
      "step": 86246
    },
    {
      "epoch": 1.1678695678710938e-05,
      "step": 86246,
      "training_step_time": 7.525963306427002
    },
    {
      "epoch": 1.1679649353027344e-05,
      "model_forward_time": 1.790431022644043,
      "step": 86247
    },
    {
      "epoch": 1.1679649353027344e-05,
      "step": 86247,
      "training_step_time": 6.769301891326904
    },
    {
      "epoch": 1.168060302734375e-05,
      "model_forward_time": 1.758751392364502,
      "step": 86248
    },
    {
      "epoch": 1.168060302734375e-05,
      "step": 86248,
      "training_step_time": 7.624262809753418
    },
    {
      "epoch": 1.1681556701660157e-05,
      "model_forward_time": 1.777367115020752,
      "step": 86249
    },
    {
      "epoch": 1.1681556701660157e-05,
      "step": 86249,
      "training_step_time": 7.618553400039673
    },
    {
      "epoch": 1.1682510375976562e-05,
      "grad_norm": 0.0899568721652031,
      "learning_rate": 5.08117712826629e-07,
      "loss": 0.0548,
      "step": 86250
    },
    {
      "epoch": 1.1682510375976562e-05,
      "model_forward_time": 1.764007806777954,
      "step": 86250
    },
    {
      "dynamics_loss_avg": 0.05743585657328367,
      "epoch": 1.1682510375976562e-05,
      "step": 86250
    },
    {
      "action_loss_avg": 0.001532736126682721,
      "epoch": 1.1682510375976562e-05,
      "step": 86250
    },
    {
      "epoch": 1.1682510375976562e-05,
      "step": 86250,
      "training_step_time": 9.578841924667358
    },
    {
      "epoch": 1.1683464050292968e-05,
      "model_forward_time": 2.42397403717041,
      "step": 86251
    },
    {
      "epoch": 1.1683464050292968e-05,
      "step": 86251,
      "training_step_time": 8.680022716522217
    },
    {
      "epoch": 1.1684417724609376e-05,
      "model_forward_time": 1.854499101638794,
      "step": 86252
    },
    {
      "epoch": 1.1684417724609376e-05,
      "step": 86252,
      "training_step_time": 9.252046823501587
    },
    {
      "epoch": 1.1685371398925781e-05,
      "model_forward_time": 1.7655842304229736,
      "step": 86253
    },
    {
      "epoch": 1.1685371398925781e-05,
      "step": 86253,
      "training_step_time": 8.177931308746338
    },
    {
      "epoch": 1.1686325073242187e-05,
      "model_forward_time": 1.767620325088501,
      "step": 86254
    },
    {
      "epoch": 1.1686325073242187e-05,
      "step": 86254,
      "training_step_time": 8.562437534332275
    },
    {
      "epoch": 1.1687278747558594e-05,
      "model_forward_time": 1.7783770561218262,
      "step": 86255
    },
    {
      "epoch": 1.1687278747558594e-05,
      "step": 86255,
      "training_step_time": 7.517088174819946
    },
    {
      "epoch": 1.1688232421875e-05,
      "model_forward_time": 1.7812161445617676,
      "step": 86256
    },
    {
      "epoch": 1.1688232421875e-05,
      "step": 86256,
      "training_step_time": 6.76394248008728
    },
    {
      "epoch": 1.1689186096191406e-05,
      "model_forward_time": 1.7771141529083252,
      "step": 86257
    },
    {
      "epoch": 1.1689186096191406e-05,
      "step": 86257,
      "training_step_time": 7.426669120788574
    },
    {
      "epoch": 1.1690139770507812e-05,
      "model_forward_time": 1.7658884525299072,
      "step": 86258
    },
    {
      "epoch": 1.1690139770507812e-05,
      "step": 86258,
      "training_step_time": 7.663612127304077
    },
    {
      "epoch": 1.1691093444824219e-05,
      "model_forward_time": 1.761564016342163,
      "step": 86259
    },
    {
      "epoch": 1.1691093444824219e-05,
      "step": 86259,
      "training_step_time": 8.437033414840698
    },
    {
      "epoch": 1.1692047119140625e-05,
      "grad_norm": 0.0724848210811615,
      "learning_rate": 5.073917110259785e-07,
      "loss": 0.0535,
      "step": 86260
    },
    {
      "epoch": 1.1692047119140625e-05,
      "model_forward_time": 1.0677568912506104,
      "step": 86260
    },
    {
      "dynamics_loss_avg": 0.047170573472976686,
      "epoch": 1.1692047119140625e-05,
      "step": 86260
    },
    {
      "action_loss_avg": 0.0018693313526455314,
      "epoch": 1.1692047119140625e-05,
      "step": 86260
    },
    {
      "epoch": 1.1692047119140625e-05,
      "step": 86260,
      "training_step_time": 7.653677940368652
    },
    {
      "epoch": 1.169300079345703e-05,
      "model_forward_time": 2.493755340576172,
      "step": 86261
    },
    {
      "epoch": 1.169300079345703e-05,
      "step": 86261,
      "training_step_time": 10.517393827438354
    },
    {
      "epoch": 1.1693954467773438e-05,
      "model_forward_time": 1.9316518306732178,
      "step": 86262
    },
    {
      "epoch": 1.1693954467773438e-05,
      "step": 86262,
      "training_step_time": 8.445440530776978
    },
    {
      "epoch": 1.1694908142089844e-05,
      "model_forward_time": 1.9041054248809814,
      "step": 86263
    },
    {
      "epoch": 1.1694908142089844e-05,
      "step": 86263,
      "training_step_time": 7.7380406856536865
    },
    {
      "epoch": 1.169586181640625e-05,
      "model_forward_time": 1.9186697006225586,
      "step": 86264
    },
    {
      "epoch": 1.169586181640625e-05,
      "step": 86264,
      "training_step_time": 8.422723531723022
    },
    {
      "epoch": 1.1696815490722657e-05,
      "model_forward_time": 1.541978359222412,
      "step": 86265
    },
    {
      "epoch": 1.1696815490722657e-05,
      "step": 86265,
      "training_step_time": 6.880162477493286
    },
    {
      "epoch": 1.1697769165039063e-05,
      "model_forward_time": 1.9090325832366943,
      "step": 86266
    },
    {
      "epoch": 1.1697769165039063e-05,
      "step": 86266,
      "training_step_time": 7.5446860790252686
    },
    {
      "epoch": 1.1698722839355468e-05,
      "model_forward_time": 1.9188544750213623,
      "step": 86267
    },
    {
      "epoch": 1.1698722839355468e-05,
      "step": 86267,
      "training_step_time": 7.759596347808838
    },
    {
      "epoch": 1.1699676513671876e-05,
      "model_forward_time": 1.9046449661254883,
      "step": 86268
    },
    {
      "epoch": 1.1699676513671876e-05,
      "step": 86268,
      "training_step_time": 9.748696327209473
    },
    {
      "epoch": 1.1700630187988281e-05,
      "model_forward_time": 1.9104716777801514,
      "step": 86269
    },
    {
      "epoch": 1.1700630187988281e-05,
      "step": 86269,
      "training_step_time": 8.501997709274292
    },
    {
      "epoch": 1.1701583862304687e-05,
      "grad_norm": 0.12244728952646255,
      "learning_rate": 5.066662005302175e-07,
      "loss": 0.0548,
      "step": 86270
    },
    {
      "epoch": 1.1701583862304687e-05,
      "model_forward_time": 2.55249285697937,
      "step": 86270
    },
    {
      "dynamics_loss_avg": 0.046293018665164706,
      "epoch": 1.1701583862304687e-05,
      "step": 86270
    },
    {
      "action_loss_avg": 0.0007003086066106334,
      "epoch": 1.1701583862304687e-05,
      "step": 86270
    },
    {
      "epoch": 1.1701583862304687e-05,
      "step": 86270,
      "training_step_time": 10.252092361450195
    },
    {
      "epoch": 1.1702537536621095e-05,
      "model_forward_time": 1.9108161926269531,
      "step": 86271
    },
    {
      "epoch": 1.1702537536621095e-05,
      "step": 86271,
      "training_step_time": 8.364112377166748
    },
    {
      "epoch": 1.17034912109375e-05,
      "model_forward_time": 1.546464204788208,
      "step": 86272
    },
    {
      "epoch": 1.17034912109375e-05,
      "step": 86272,
      "training_step_time": 7.665374994277954
    },
    {
      "epoch": 1.1704444885253906e-05,
      "model_forward_time": 1.9107160568237305,
      "step": 86273
    },
    {
      "epoch": 1.1704444885253906e-05,
      "step": 86273,
      "training_step_time": 7.646079063415527
    },
    {
      "epoch": 1.1705398559570312e-05,
      "model_forward_time": 1.9149861335754395,
      "step": 86274
    },
    {
      "epoch": 1.1705398559570312e-05,
      "step": 86274,
      "training_step_time": 7.626629829406738
    },
    {
      "epoch": 1.1706352233886719e-05,
      "model_forward_time": 1.8971199989318848,
      "step": 86275
    },
    {
      "epoch": 1.1706352233886719e-05,
      "step": 86275,
      "training_step_time": 7.525968074798584
    },
    {
      "epoch": 1.1707305908203125e-05,
      "model_forward_time": 1.9265601634979248,
      "step": 86276
    },
    {
      "epoch": 1.1707305908203125e-05,
      "step": 86276,
      "training_step_time": 7.827760219573975
    },
    {
      "epoch": 1.170825958251953e-05,
      "model_forward_time": 1.9076199531555176,
      "step": 86277
    },
    {
      "epoch": 1.170825958251953e-05,
      "step": 86277,
      "training_step_time": 7.892470836639404
    },
    {
      "epoch": 1.1709213256835938e-05,
      "model_forward_time": 1.9139821529388428,
      "step": 86278
    },
    {
      "epoch": 1.1709213256835938e-05,
      "step": 86278,
      "training_step_time": 7.669888734817505
    },
    {
      "epoch": 1.1710166931152344e-05,
      "model_forward_time": 2.5441701412200928,
      "step": 86279
    },
    {
      "epoch": 1.1710166931152344e-05,
      "step": 86279,
      "training_step_time": 8.288938760757446
    },
    {
      "epoch": 1.171112060546875e-05,
      "grad_norm": 0.0901380181312561,
      "learning_rate": 5.059411814186837e-07,
      "loss": 0.0536,
      "step": 86280
    },
    {
      "epoch": 1.171112060546875e-05,
      "model_forward_time": 1.9078483581542969,
      "step": 86280
    },
    {
      "dynamics_loss_avg": 0.05008405186235905,
      "epoch": 1.171112060546875e-05,
      "step": 86280
    },
    {
      "action_loss_avg": 0.0009126581426244229,
      "epoch": 1.171112060546875e-05,
      "step": 86280
    },
    {
      "epoch": 1.171112060546875e-05,
      "step": 86280,
      "training_step_time": 8.560321569442749
    },
    {
      "epoch": 1.1712074279785157e-05,
      "model_forward_time": 1.908292531967163,
      "step": 86281
    },
    {
      "epoch": 1.1712074279785157e-05,
      "step": 86281,
      "training_step_time": 8.229182481765747
    },
    {
      "epoch": 1.1713027954101563e-05,
      "model_forward_time": 1.9280664920806885,
      "step": 86282
    },
    {
      "epoch": 1.1713027954101563e-05,
      "step": 86282,
      "training_step_time": 7.660923719406128
    },
    {
      "epoch": 1.1713981628417968e-05,
      "model_forward_time": 1.918508529663086,
      "step": 86283
    },
    {
      "epoch": 1.1713981628417968e-05,
      "step": 86283,
      "training_step_time": 7.560274839401245
    },
    {
      "epoch": 1.1714935302734376e-05,
      "model_forward_time": 1.1537408828735352,
      "step": 86284
    },
    {
      "epoch": 1.1714935302734376e-05,
      "step": 86284,
      "training_step_time": 7.478683710098267
    },
    {
      "epoch": 1.1715888977050781e-05,
      "model_forward_time": 1.921022891998291,
      "step": 86285
    },
    {
      "epoch": 1.1715888977050781e-05,
      "step": 86285,
      "training_step_time": 8.30275297164917
    },
    {
      "epoch": 1.1716842651367187e-05,
      "model_forward_time": 1.9167165756225586,
      "step": 86286
    },
    {
      "epoch": 1.1716842651367187e-05,
      "step": 86286,
      "training_step_time": 7.927783250808716
    },
    {
      "epoch": 1.1717796325683595e-05,
      "model_forward_time": 1.9108917713165283,
      "step": 86287
    },
    {
      "epoch": 1.1717796325683595e-05,
      "step": 86287,
      "training_step_time": 9.731504201889038
    },
    {
      "epoch": 1.171875e-05,
      "model_forward_time": 1.9225528240203857,
      "step": 86288
    },
    {
      "epoch": 1.171875e-05,
      "step": 86288,
      "training_step_time": 8.4225332736969
    },
    {
      "epoch": 1.1719703674316406e-05,
      "model_forward_time": 2.48313307762146,
      "step": 86289
    },
    {
      "epoch": 1.1719703674316406e-05,
      "step": 86289,
      "training_step_time": 9.732848405838013
    },
    {
      "epoch": 1.1720657348632812e-05,
      "grad_norm": 0.09593115746974945,
      "learning_rate": 5.052166537706659e-07,
      "loss": 0.0546,
      "step": 86290
    },
    {
      "epoch": 1.1720657348632812e-05,
      "model_forward_time": 1.9011428356170654,
      "step": 86290
    },
    {
      "dynamics_loss_avg": 0.06281924303621053,
      "epoch": 1.1720657348632812e-05,
      "step": 86290
    },
    {
      "action_loss_avg": 0.0010661701409844682,
      "epoch": 1.1720657348632812e-05,
      "step": 86290
    },
    {
      "epoch": 1.1720657348632812e-05,
      "step": 86290,
      "training_step_time": 8.749229192733765
    },
    {
      "epoch": 1.1721611022949219e-05,
      "model_forward_time": 1.9195044040679932,
      "step": 86291
    },
    {
      "epoch": 1.1721611022949219e-05,
      "step": 86291,
      "training_step_time": 8.72005033493042
    },
    {
      "epoch": 1.1722564697265625e-05,
      "model_forward_time": 1.9194443225860596,
      "step": 86292
    },
    {
      "epoch": 1.1722564697265625e-05,
      "step": 86292,
      "training_step_time": 7.56856369972229
    },
    {
      "epoch": 1.172351837158203e-05,
      "model_forward_time": 1.9072229862213135,
      "step": 86293
    },
    {
      "epoch": 1.172351837158203e-05,
      "step": 86293,
      "training_step_time": 7.500887870788574
    },
    {
      "epoch": 1.1724472045898438e-05,
      "model_forward_time": 1.911867618560791,
      "step": 86294
    },
    {
      "epoch": 1.1724472045898438e-05,
      "step": 86294,
      "training_step_time": 7.637646198272705
    },
    {
      "epoch": 1.1725425720214844e-05,
      "model_forward_time": 1.8938336372375488,
      "step": 86295
    },
    {
      "epoch": 1.1725425720214844e-05,
      "step": 86295,
      "training_step_time": 9.639026880264282
    },
    {
      "epoch": 1.172637939453125e-05,
      "model_forward_time": 1.903585433959961,
      "step": 86296
    },
    {
      "epoch": 1.172637939453125e-05,
      "step": 86296,
      "training_step_time": 9.689242601394653
    },
    {
      "epoch": 1.1727333068847657e-05,
      "model_forward_time": 1.9251484870910645,
      "step": 86297
    },
    {
      "epoch": 1.1727333068847657e-05,
      "step": 86297,
      "training_step_time": 9.220449447631836
    },
    {
      "epoch": 1.1728286743164063e-05,
      "model_forward_time": 2.5687224864959717,
      "step": 86298
    },
    {
      "epoch": 1.1728286743164063e-05,
      "step": 86298,
      "training_step_time": 7.907925605773926
    },
    {
      "epoch": 1.1729240417480468e-05,
      "model_forward_time": 1.910461664199829,
      "step": 86299
    },
    {
      "epoch": 1.1729240417480468e-05,
      "step": 86299,
      "training_step_time": 8.559149026870728
    },
    {
      "epoch": 1.1730194091796876e-05,
      "grad_norm": 0.09595313668251038,
      "learning_rate": 5.044926176653969e-07,
      "loss": 0.055,
      "step": 86300
    },
    {
      "epoch": 1.1730194091796876e-05,
      "model_forward_time": 1.5149693489074707,
      "step": 86300
    },
    {
      "dynamics_loss_avg": 0.0440678327344358,
      "epoch": 1.1730194091796876e-05,
      "step": 86300
    },
    {
      "action_loss_avg": 0.0009489296498941258,
      "epoch": 1.1730194091796876e-05,
      "step": 86300
    },
    {
      "epoch": 1.1730194091796876e-05,
      "step": 86300,
      "training_step_time": 8.285819292068481
    },
    {
      "epoch": 1.1731147766113281e-05,
      "model_forward_time": 1.9065425395965576,
      "step": 86301
    },
    {
      "epoch": 1.1731147766113281e-05,
      "step": 86301,
      "training_step_time": 8.71792721748352
    },
    {
      "epoch": 1.1732101440429687e-05,
      "model_forward_time": 1.9107787609100342,
      "step": 86302
    },
    {
      "epoch": 1.1732101440429687e-05,
      "step": 86302,
      "training_step_time": 7.435635805130005
    },
    {
      "epoch": 1.1733055114746095e-05,
      "model_forward_time": 1.9119024276733398,
      "step": 86303
    },
    {
      "epoch": 1.1733055114746095e-05,
      "step": 86303,
      "training_step_time": 7.726176500320435
    },
    {
      "epoch": 1.17340087890625e-05,
      "model_forward_time": 1.9008288383483887,
      "step": 86304
    },
    {
      "epoch": 1.17340087890625e-05,
      "step": 86304,
      "training_step_time": 7.626938581466675
    },
    {
      "epoch": 1.1734962463378906e-05,
      "model_forward_time": 1.8943631649017334,
      "step": 86305
    },
    {
      "epoch": 1.1734962463378906e-05,
      "step": 86305,
      "training_step_time": 9.71780776977539
    },
    {
      "epoch": 1.1735916137695312e-05,
      "model_forward_time": 1.8988590240478516,
      "step": 86306
    },
    {
      "epoch": 1.1735916137695312e-05,
      "step": 86306,
      "training_step_time": 8.677049398422241
    },
    {
      "epoch": 1.1736869812011719e-05,
      "model_forward_time": 2.5469791889190674,
      "step": 86307
    },
    {
      "epoch": 1.1736869812011719e-05,
      "step": 86307,
      "training_step_time": 8.282471895217896
    },
    {
      "epoch": 1.1737823486328125e-05,
      "model_forward_time": 1.9062087535858154,
      "step": 86308
    },
    {
      "epoch": 1.1737823486328125e-05,
      "step": 86308,
      "training_step_time": 8.470166444778442
    },
    {
      "epoch": 1.173877716064453e-05,
      "model_forward_time": 1.9225318431854248,
      "step": 86309
    },
    {
      "epoch": 1.173877716064453e-05,
      "step": 86309,
      "training_step_time": 8.669887065887451
    },
    {
      "epoch": 1.1739730834960938e-05,
      "grad_norm": 0.08254028111696243,
      "learning_rate": 5.037690731820571e-07,
      "loss": 0.0549,
      "step": 86310
    },
    {
      "epoch": 1.1739730834960938e-05,
      "model_forward_time": 1.9102561473846436,
      "step": 86310
    },
    {
      "dynamics_loss_avg": 0.06312639228999614,
      "epoch": 1.1739730834960938e-05,
      "step": 86310
    },
    {
      "action_loss_avg": 0.001153684689779766,
      "epoch": 1.1739730834960938e-05,
      "step": 86310
    },
    {
      "epoch": 1.1739730834960938e-05,
      "step": 86310,
      "training_step_time": 8.002886772155762
    },
    {
      "epoch": 1.1740684509277344e-05,
      "model_forward_time": 1.8981311321258545,
      "step": 86311
    },
    {
      "epoch": 1.1740684509277344e-05,
      "step": 86311,
      "training_step_time": 7.557189702987671
    },
    {
      "epoch": 1.174163818359375e-05,
      "model_forward_time": 1.913560152053833,
      "step": 86312
    },
    {
      "epoch": 1.174163818359375e-05,
      "step": 86312,
      "training_step_time": 7.655284881591797
    },
    {
      "epoch": 1.1742591857910157e-05,
      "model_forward_time": 1.9049527645111084,
      "step": 86313
    },
    {
      "epoch": 1.1742591857910157e-05,
      "step": 86313,
      "training_step_time": 7.607663631439209
    },
    {
      "epoch": 1.1743545532226563e-05,
      "model_forward_time": 1.9077837467193604,
      "step": 86314
    },
    {
      "epoch": 1.1743545532226563e-05,
      "step": 86314,
      "training_step_time": 7.87212061882019
    },
    {
      "epoch": 1.1744499206542968e-05,
      "model_forward_time": 1.9017350673675537,
      "step": 86315
    },
    {
      "epoch": 1.1744499206542968e-05,
      "step": 86315,
      "training_step_time": 9.749956369400024
    },
    {
      "epoch": 1.1745452880859376e-05,
      "model_forward_time": 1.5358836650848389,
      "step": 86316
    },
    {
      "epoch": 1.1745452880859376e-05,
      "step": 86316,
      "training_step_time": 7.836156606674194
    },
    {
      "epoch": 1.1746406555175781e-05,
      "model_forward_time": 2.4929845333099365,
      "step": 86317
    },
    {
      "epoch": 1.1746406555175781e-05,
      "step": 86317,
      "training_step_time": 7.781715393066406
    },
    {
      "epoch": 1.1747360229492187e-05,
      "model_forward_time": 1.9162979125976562,
      "step": 86318
    },
    {
      "epoch": 1.1747360229492187e-05,
      "step": 86318,
      "training_step_time": 9.550851345062256
    },
    {
      "epoch": 1.1748313903808595e-05,
      "model_forward_time": 1.9087131023406982,
      "step": 86319
    },
    {
      "epoch": 1.1748313903808595e-05,
      "step": 86319,
      "training_step_time": 7.56325626373291
    },
    {
      "epoch": 1.1749267578125e-05,
      "grad_norm": 0.08782930672168732,
      "learning_rate": 5.030460203997711e-07,
      "loss": 0.055,
      "step": 86320
    },
    {
      "epoch": 1.1749267578125e-05,
      "model_forward_time": 1.9071433544158936,
      "step": 86320
    },
    {
      "dynamics_loss_avg": 0.05976786557585001,
      "epoch": 1.1749267578125e-05,
      "step": 86320
    },
    {
      "action_loss_avg": 0.0012823488505091518,
      "epoch": 1.1749267578125e-05,
      "step": 86320
    },
    {
      "epoch": 1.1749267578125e-05,
      "step": 86320,
      "training_step_time": 7.615126848220825
    },
    {
      "epoch": 1.1750221252441406e-05,
      "model_forward_time": 1.8921422958374023,
      "step": 86321
    },
    {
      "epoch": 1.1750221252441406e-05,
      "step": 86321,
      "training_step_time": 7.694340944290161
    },
    {
      "epoch": 1.1751174926757812e-05,
      "model_forward_time": 1.8978500366210938,
      "step": 86322
    },
    {
      "epoch": 1.1751174926757812e-05,
      "step": 86322,
      "training_step_time": 7.61789083480835
    },
    {
      "epoch": 1.1752128601074219e-05,
      "model_forward_time": 1.9100561141967773,
      "step": 86323
    },
    {
      "epoch": 1.1752128601074219e-05,
      "step": 86323,
      "training_step_time": 8.694988250732422
    },
    {
      "epoch": 1.1753082275390625e-05,
      "model_forward_time": 1.9058983325958252,
      "step": 86324
    },
    {
      "epoch": 1.1753082275390625e-05,
      "step": 86324,
      "training_step_time": 8.784260511398315
    },
    {
      "epoch": 1.175403594970703e-05,
      "model_forward_time": 1.9017696380615234,
      "step": 86325
    },
    {
      "epoch": 1.175403594970703e-05,
      "step": 86325,
      "training_step_time": 8.381134986877441
    },
    {
      "epoch": 1.1754989624023438e-05,
      "model_forward_time": 2.509324312210083,
      "step": 86326
    },
    {
      "epoch": 1.1754989624023438e-05,
      "step": 86326,
      "training_step_time": 8.784188032150269
    },
    {
      "epoch": 1.1755943298339844e-05,
      "model_forward_time": 1.9288103580474854,
      "step": 86327
    },
    {
      "epoch": 1.1755943298339844e-05,
      "step": 86327,
      "training_step_time": 8.139904260635376
    },
    {
      "epoch": 1.175689697265625e-05,
      "model_forward_time": 1.1736760139465332,
      "step": 86328
    },
    {
      "epoch": 1.175689697265625e-05,
      "step": 86328,
      "training_step_time": 8.112573623657227
    },
    {
      "epoch": 1.1757850646972657e-05,
      "model_forward_time": 1.897374153137207,
      "step": 86329
    },
    {
      "epoch": 1.1757850646972657e-05,
      "step": 86329,
      "training_step_time": 7.689207315444946
    },
    {
      "epoch": 1.1758804321289063e-05,
      "grad_norm": 0.06960449367761612,
      "learning_rate": 5.023234593976118e-07,
      "loss": 0.0553,
      "step": 86330
    },
    {
      "epoch": 1.1758804321289063e-05,
      "model_forward_time": 1.906369686126709,
      "step": 86330
    },
    {
      "dynamics_loss_avg": 0.03800949808210134,
      "epoch": 1.1758804321289063e-05,
      "step": 86330
    },
    {
      "action_loss_avg": 0.0009931853623129428,
      "epoch": 1.1758804321289063e-05,
      "step": 86330
    },
    {
      "epoch": 1.1758804321289063e-05,
      "step": 86330,
      "training_step_time": 7.706944942474365
    },
    {
      "epoch": 1.1759757995605468e-05,
      "model_forward_time": 1.9047415256500244,
      "step": 86331
    },
    {
      "epoch": 1.1759757995605468e-05,
      "step": 86331,
      "training_step_time": 7.694826364517212
    },
    {
      "epoch": 1.1760711669921876e-05,
      "model_forward_time": 1.9016706943511963,
      "step": 86332
    },
    {
      "epoch": 1.1760711669921876e-05,
      "step": 86332,
      "training_step_time": 8.534799337387085
    },
    {
      "epoch": 1.1761665344238281e-05,
      "model_forward_time": 1.9070165157318115,
      "step": 86333
    },
    {
      "epoch": 1.1761665344238281e-05,
      "step": 86333,
      "training_step_time": 7.690276861190796
    },
    {
      "epoch": 1.1762619018554687e-05,
      "model_forward_time": 1.9216411113739014,
      "step": 86334
    },
    {
      "epoch": 1.1762619018554687e-05,
      "step": 86334,
      "training_step_time": 9.224300622940063
    },
    {
      "epoch": 1.1763572692871095e-05,
      "model_forward_time": 2.5447421073913574,
      "step": 86335
    },
    {
      "epoch": 1.1763572692871095e-05,
      "step": 86335,
      "training_step_time": 7.905652761459351
    },
    {
      "epoch": 1.17645263671875e-05,
      "model_forward_time": 1.8805596828460693,
      "step": 86336
    },
    {
      "epoch": 1.17645263671875e-05,
      "step": 86336,
      "training_step_time": 7.677537202835083
    },
    {
      "epoch": 1.1765480041503906e-05,
      "model_forward_time": 1.8926105499267578,
      "step": 86337
    },
    {
      "epoch": 1.1765480041503906e-05,
      "step": 86337,
      "training_step_time": 10.39484453201294
    },
    {
      "epoch": 1.1766433715820312e-05,
      "model_forward_time": 1.9264698028564453,
      "step": 86338
    },
    {
      "epoch": 1.1766433715820312e-05,
      "step": 86338,
      "training_step_time": 7.682476758956909
    },
    {
      "epoch": 1.1767387390136719e-05,
      "model_forward_time": 1.9020984172821045,
      "step": 86339
    },
    {
      "epoch": 1.1767387390136719e-05,
      "step": 86339,
      "training_step_time": 7.666251182556152
    },
    {
      "epoch": 1.1768341064453125e-05,
      "grad_norm": 0.08201900124549866,
      "learning_rate": 5.016013902545957e-07,
      "loss": 0.0556,
      "step": 86340
    },
    {
      "epoch": 1.1768341064453125e-05,
      "model_forward_time": 1.8940036296844482,
      "step": 86340
    },
    {
      "dynamics_loss_avg": 0.04667940475046635,
      "epoch": 1.1768341064453125e-05,
      "step": 86340
    },
    {
      "action_loss_avg": 0.0008180203411029651,
      "epoch": 1.1768341064453125e-05,
      "step": 86340
    },
    {
      "epoch": 1.1768341064453125e-05,
      "step": 86340,
      "training_step_time": 7.620413541793823
    },
    {
      "epoch": 1.176929473876953e-05,
      "model_forward_time": 1.9018633365631104,
      "step": 86341
    },
    {
      "epoch": 1.176929473876953e-05,
      "step": 86341,
      "training_step_time": 7.858182191848755
    },
    {
      "epoch": 1.1770248413085938e-05,
      "model_forward_time": 1.9052340984344482,
      "step": 86342
    },
    {
      "epoch": 1.1770248413085938e-05,
      "step": 86342,
      "training_step_time": 7.620810270309448
    },
    {
      "epoch": 1.1771202087402344e-05,
      "model_forward_time": 1.8914411067962646,
      "step": 86343
    },
    {
      "epoch": 1.1771202087402344e-05,
      "step": 86343,
      "training_step_time": 8.734254598617554
    },
    {
      "epoch": 1.177215576171875e-05,
      "model_forward_time": 1.8973908424377441,
      "step": 86344
    },
    {
      "epoch": 1.177215576171875e-05,
      "step": 86344,
      "training_step_time": 7.824785232543945
    },
    {
      "epoch": 1.1773109436035157e-05,
      "model_forward_time": 2.5046913623809814,
      "step": 86345
    },
    {
      "epoch": 1.1773109436035157e-05,
      "step": 86345,
      "training_step_time": 7.6111979484558105
    },
    {
      "epoch": 1.1774063110351563e-05,
      "model_forward_time": 1.9067473411560059,
      "step": 86346
    },
    {
      "epoch": 1.1774063110351563e-05,
      "step": 86346,
      "training_step_time": 7.6072564125061035
    },
    {
      "epoch": 1.1775016784667968e-05,
      "model_forward_time": 1.9006822109222412,
      "step": 86347
    },
    {
      "epoch": 1.1775016784667968e-05,
      "step": 86347,
      "training_step_time": 7.661915063858032
    },
    {
      "epoch": 1.1775970458984376e-05,
      "model_forward_time": 1.9026150703430176,
      "step": 86348
    },
    {
      "epoch": 1.1775970458984376e-05,
      "step": 86348,
      "training_step_time": 7.6654462814331055
    },
    {
      "epoch": 1.1776924133300781e-05,
      "model_forward_time": 1.8959078788757324,
      "step": 86349
    },
    {
      "epoch": 1.1776924133300781e-05,
      "step": 86349,
      "training_step_time": 7.634352445602417
    },
    {
      "epoch": 1.1777877807617187e-05,
      "grad_norm": 0.08430106192827225,
      "learning_rate": 5.008798130496889e-07,
      "loss": 0.0561,
      "step": 86350
    },
    {
      "epoch": 1.1777877807617187e-05,
      "model_forward_time": 1.8786187171936035,
      "step": 86350
    },
    {
      "dynamics_loss_avg": 0.04951551649719477,
      "epoch": 1.1777877807617187e-05,
      "step": 86350
    },
    {
      "action_loss_avg": 0.0009855627693468704,
      "epoch": 1.1777877807617187e-05,
      "step": 86350
    },
    {
      "epoch": 1.1777877807617187e-05,
      "step": 86350,
      "training_step_time": 8.047969102859497
    },
    {
      "epoch": 1.1778831481933595e-05,
      "model_forward_time": 1.9118409156799316,
      "step": 86351
    },
    {
      "epoch": 1.1778831481933595e-05,
      "step": 86351,
      "training_step_time": 7.881880044937134
    },
    {
      "epoch": 1.177978515625e-05,
      "model_forward_time": 1.9193344116210938,
      "step": 86352
    },
    {
      "epoch": 1.177978515625e-05,
      "step": 86352,
      "training_step_time": 9.741597175598145
    },
    {
      "epoch": 1.1780738830566406e-05,
      "model_forward_time": 1.8852770328521729,
      "step": 86353
    },
    {
      "epoch": 1.1780738830566406e-05,
      "step": 86353,
      "training_step_time": 9.335758686065674
    },
    {
      "epoch": 1.1781692504882812e-05,
      "model_forward_time": 2.498565912246704,
      "step": 86354
    },
    {
      "epoch": 1.1781692504882812e-05,
      "step": 86354,
      "training_step_time": 8.849088907241821
    },
    {
      "epoch": 1.178264617919922e-05,
      "model_forward_time": 1.9073758125305176,
      "step": 86355
    },
    {
      "epoch": 1.178264617919922e-05,
      "step": 86355,
      "training_step_time": 8.441288471221924
    },
    {
      "epoch": 1.1783599853515625e-05,
      "model_forward_time": 1.920057773590088,
      "step": 86356
    },
    {
      "epoch": 1.1783599853515625e-05,
      "step": 86356,
      "training_step_time": 7.6041953563690186
    },
    {
      "epoch": 1.178455352783203e-05,
      "model_forward_time": 1.9065592288970947,
      "step": 86357
    },
    {
      "epoch": 1.178455352783203e-05,
      "step": 86357,
      "training_step_time": 7.685961484909058
    },
    {
      "epoch": 1.1785507202148438e-05,
      "model_forward_time": 1.8927059173583984,
      "step": 86358
    },
    {
      "epoch": 1.1785507202148438e-05,
      "step": 86358,
      "training_step_time": 7.588896036148071
    },
    {
      "epoch": 1.1786460876464844e-05,
      "model_forward_time": 1.9076921939849854,
      "step": 86359
    },
    {
      "epoch": 1.1786460876464844e-05,
      "step": 86359,
      "training_step_time": 7.544835567474365
    },
    {
      "epoch": 1.178741455078125e-05,
      "grad_norm": 0.09026920795440674,
      "learning_rate": 5.001587278618008e-07,
      "loss": 0.0555,
      "step": 86360
    },
    {
      "epoch": 1.178741455078125e-05,
      "model_forward_time": 1.8936922550201416,
      "step": 86360
    },
    {
      "dynamics_loss_avg": 0.049577430821955204,
      "epoch": 1.178741455078125e-05,
      "step": 86360
    },
    {
      "action_loss_avg": 0.0007838356570573524,
      "epoch": 1.178741455078125e-05,
      "step": 86360
    },
    {
      "epoch": 1.178741455078125e-05,
      "step": 86360,
      "training_step_time": 8.677196502685547
    },
    {
      "epoch": 1.1788368225097657e-05,
      "model_forward_time": 1.9094560146331787,
      "step": 86361
    },
    {
      "epoch": 1.1788368225097657e-05,
      "step": 86361,
      "training_step_time": 8.576645374298096
    },
    {
      "epoch": 1.1789321899414063e-05,
      "model_forward_time": 1.915633201599121,
      "step": 86362
    },
    {
      "epoch": 1.1789321899414063e-05,
      "step": 86362,
      "training_step_time": 9.679767370223999
    },
    {
      "epoch": 1.1790275573730468e-05,
      "model_forward_time": 2.4992384910583496,
      "step": 86363
    },
    {
      "epoch": 1.1790275573730468e-05,
      "step": 86363,
      "training_step_time": 7.7149083614349365
    },
    {
      "epoch": 1.1791229248046876e-05,
      "model_forward_time": 1.8922028541564941,
      "step": 86364
    },
    {
      "epoch": 1.1791229248046876e-05,
      "step": 86364,
      "training_step_time": 7.643341541290283
    },
    {
      "epoch": 1.1792182922363281e-05,
      "model_forward_time": 1.8886165618896484,
      "step": 86365
    },
    {
      "epoch": 1.1792182922363281e-05,
      "step": 86365,
      "training_step_time": 7.617286920547485
    },
    {
      "epoch": 1.1793136596679687e-05,
      "model_forward_time": 1.8972527980804443,
      "step": 86366
    },
    {
      "epoch": 1.1793136596679687e-05,
      "step": 86366,
      "training_step_time": 7.8405680656433105
    },
    {
      "epoch": 1.1794090270996095e-05,
      "model_forward_time": 1.9026565551757812,
      "step": 86367
    },
    {
      "epoch": 1.1794090270996095e-05,
      "step": 86367,
      "training_step_time": 7.760420560836792
    },
    {
      "epoch": 1.17950439453125e-05,
      "model_forward_time": 1.9172632694244385,
      "step": 86368
    },
    {
      "epoch": 1.17950439453125e-05,
      "step": 86368,
      "training_step_time": 8.479843378067017
    },
    {
      "epoch": 1.1795997619628906e-05,
      "model_forward_time": 1.8838286399841309,
      "step": 86369
    },
    {
      "epoch": 1.1795997619628906e-05,
      "step": 86369,
      "training_step_time": 8.07809066772461
    },
    {
      "epoch": 1.1796951293945312e-05,
      "grad_norm": 0.09014135599136353,
      "learning_rate": 4.99438134769789e-07,
      "loss": 0.0541,
      "step": 86370
    },
    {
      "epoch": 1.1796951293945312e-05,
      "model_forward_time": 1.9077754020690918,
      "step": 86370
    },
    {
      "dynamics_loss_avg": 0.05245915297418833,
      "epoch": 1.1796951293945312e-05,
      "step": 86370
    },
    {
      "action_loss_avg": 0.0009972077532438561,
      "epoch": 1.1796951293945312e-05,
      "step": 86370
    },
    {
      "epoch": 1.1796951293945312e-05,
      "step": 86370,
      "training_step_time": 7.8656346797943115
    },
    {
      "epoch": 1.179790496826172e-05,
      "model_forward_time": 1.9129984378814697,
      "step": 86371
    },
    {
      "epoch": 1.179790496826172e-05,
      "step": 86371,
      "training_step_time": 9.203051328659058
    },
    {
      "epoch": 1.1798858642578125e-05,
      "model_forward_time": 2.5149474143981934,
      "step": 86372
    },
    {
      "epoch": 1.1798858642578125e-05,
      "step": 86372,
      "training_step_time": 8.437620162963867
    },
    {
      "epoch": 1.179981231689453e-05,
      "model_forward_time": 1.9202709197998047,
      "step": 86373
    },
    {
      "epoch": 1.179981231689453e-05,
      "step": 86373,
      "training_step_time": 7.824774980545044
    },
    {
      "epoch": 1.1800765991210938e-05,
      "model_forward_time": 1.933330774307251,
      "step": 86374
    },
    {
      "epoch": 1.1800765991210938e-05,
      "step": 86374,
      "training_step_time": 7.623892307281494
    },
    {
      "epoch": 1.1801719665527344e-05,
      "model_forward_time": 1.9244153499603271,
      "step": 86375
    },
    {
      "epoch": 1.1801719665527344e-05,
      "step": 86375,
      "training_step_time": 7.940407991409302
    },
    {
      "epoch": 1.180267333984375e-05,
      "model_forward_time": 1.9010820388793945,
      "step": 86376
    },
    {
      "epoch": 1.180267333984375e-05,
      "step": 86376,
      "training_step_time": 7.773481130599976
    },
    {
      "epoch": 1.1803627014160157e-05,
      "model_forward_time": 1.9200873374938965,
      "step": 86377
    },
    {
      "epoch": 1.1803627014160157e-05,
      "step": 86377,
      "training_step_time": 7.5851452350616455
    },
    {
      "epoch": 1.1804580688476563e-05,
      "model_forward_time": 1.1766698360443115,
      "step": 86378
    },
    {
      "epoch": 1.1804580688476563e-05,
      "step": 86378,
      "training_step_time": 7.983432769775391
    },
    {
      "epoch": 1.1805534362792968e-05,
      "model_forward_time": 1.9178404808044434,
      "step": 86379
    },
    {
      "epoch": 1.1805534362792968e-05,
      "step": 86379,
      "training_step_time": 9.702829599380493
    },
    {
      "epoch": 1.1806488037109376e-05,
      "grad_norm": 0.30158790946006775,
      "learning_rate": 4.987180338524572e-07,
      "loss": 0.0552,
      "step": 86380
    },
    {
      "epoch": 1.1806488037109376e-05,
      "model_forward_time": 1.9036521911621094,
      "step": 86380
    },
    {
      "dynamics_loss_avg": 0.0442870382219553,
      "epoch": 1.1806488037109376e-05,
      "step": 86380
    },
    {
      "action_loss_avg": 0.0008885100192856044,
      "epoch": 1.1806488037109376e-05,
      "step": 86380
    },
    {
      "epoch": 1.1806488037109376e-05,
      "step": 86380,
      "training_step_time": 7.862192630767822
    },
    {
      "epoch": 1.1807441711425781e-05,
      "model_forward_time": 1.9368329048156738,
      "step": 86381
    },
    {
      "epoch": 1.1807441711425781e-05,
      "step": 86381,
      "training_step_time": 7.994607448577881
    },
    {
      "epoch": 1.1808395385742187e-05,
      "model_forward_time": 1.7982838153839111,
      "step": 86382
    },
    {
      "epoch": 1.1808395385742187e-05,
      "step": 86382,
      "training_step_time": 8.70736813545227
    },
    {
      "epoch": 1.1809349060058593e-05,
      "model_forward_time": 1.916947364807129,
      "step": 86383
    },
    {
      "epoch": 1.1809349060058593e-05,
      "step": 86383,
      "training_step_time": 8.604607820510864
    },
    {
      "epoch": 1.1810302734375e-05,
      "model_forward_time": 1.9154329299926758,
      "step": 86384
    },
    {
      "epoch": 1.1810302734375e-05,
      "step": 86384,
      "training_step_time": 7.903958082199097
    },
    {
      "epoch": 1.1811256408691406e-05,
      "model_forward_time": 1.928067684173584,
      "step": 86385
    },
    {
      "epoch": 1.1811256408691406e-05,
      "step": 86385,
      "training_step_time": 7.779482841491699
    },
    {
      "epoch": 1.1812210083007812e-05,
      "model_forward_time": 1.9235265254974365,
      "step": 86386
    },
    {
      "epoch": 1.1812210083007812e-05,
      "step": 86386,
      "training_step_time": 8.42577862739563
    },
    {
      "epoch": 1.181316375732422e-05,
      "model_forward_time": 1.9212090969085693,
      "step": 86387
    },
    {
      "epoch": 1.181316375732422e-05,
      "step": 86387,
      "training_step_time": 7.832289218902588
    },
    {
      "epoch": 1.1814117431640625e-05,
      "model_forward_time": 1.926668405532837,
      "step": 86388
    },
    {
      "epoch": 1.1814117431640625e-05,
      "step": 86388,
      "training_step_time": 8.672261953353882
    },
    {
      "epoch": 1.181507110595703e-05,
      "model_forward_time": 1.912858486175537,
      "step": 86389
    },
    {
      "epoch": 1.181507110595703e-05,
      "step": 86389,
      "training_step_time": 9.282019853591919
    },
    {
      "epoch": 1.1816024780273438e-05,
      "grad_norm": 0.08311126381158829,
      "learning_rate": 4.979984251885528e-07,
      "loss": 0.0547,
      "step": 86390
    },
    {
      "epoch": 1.1816024780273438e-05,
      "model_forward_time": 1.9286811351776123,
      "step": 86390
    },
    {
      "dynamics_loss_avg": 0.0430537985637784,
      "epoch": 1.1816024780273438e-05,
      "step": 86390
    },
    {
      "action_loss_avg": 0.0013202911068219692,
      "epoch": 1.1816024780273438e-05,
      "step": 86390
    },
    {
      "epoch": 1.1816024780273438e-05,
      "step": 86390,
      "training_step_time": 9.240567922592163
    },
    {
      "epoch": 1.1816978454589844e-05,
      "model_forward_time": 2.567506790161133,
      "step": 86391
    },
    {
      "epoch": 1.1816978454589844e-05,
      "step": 86391,
      "training_step_time": 7.507932424545288
    },
    {
      "epoch": 1.181793212890625e-05,
      "model_forward_time": 1.9125514030456543,
      "step": 86392
    },
    {
      "epoch": 1.181793212890625e-05,
      "step": 86392,
      "training_step_time": 8.301510095596313
    },
    {
      "epoch": 1.1818885803222657e-05,
      "model_forward_time": 1.931074857711792,
      "step": 86393
    },
    {
      "epoch": 1.1818885803222657e-05,
      "step": 86393,
      "training_step_time": 7.683716773986816
    },
    {
      "epoch": 1.1819839477539063e-05,
      "model_forward_time": 1.9293689727783203,
      "step": 86394
    },
    {
      "epoch": 1.1819839477539063e-05,
      "step": 86394,
      "training_step_time": 7.677458047866821
    },
    {
      "epoch": 1.1820793151855468e-05,
      "model_forward_time": 1.9258904457092285,
      "step": 86395
    },
    {
      "epoch": 1.1820793151855468e-05,
      "step": 86395,
      "training_step_time": 7.3864524364471436
    },
    {
      "epoch": 1.1821746826171876e-05,
      "model_forward_time": 1.9304227828979492,
      "step": 86396
    },
    {
      "epoch": 1.1821746826171876e-05,
      "step": 86396,
      "training_step_time": 7.5450286865234375
    },
    {
      "epoch": 1.1822700500488281e-05,
      "model_forward_time": 1.9243872165679932,
      "step": 86397
    },
    {
      "epoch": 1.1822700500488281e-05,
      "step": 86397,
      "training_step_time": 8.63148832321167
    },
    {
      "epoch": 1.1823654174804687e-05,
      "model_forward_time": 1.8860769271850586,
      "step": 86398
    },
    {
      "epoch": 1.1823654174804687e-05,
      "step": 86398,
      "training_step_time": 8.677372455596924
    },
    {
      "epoch": 1.1824607849121093e-05,
      "model_forward_time": 1.910961627960205,
      "step": 86399
    },
    {
      "epoch": 1.1824607849121093e-05,
      "step": 86399,
      "training_step_time": 9.386045694351196
    },
    {
      "epoch": 1.18255615234375e-05,
      "grad_norm": 0.0843530148267746,
      "learning_rate": 4.972793088567729e-07,
      "loss": 0.0543,
      "step": 86400
    },
    {
      "epoch": 1.18255615234375e-05,
      "model_forward_time": 2.5281155109405518,
      "step": 86400
    },
    {
      "dynamics_loss_avg": 0.05087875947356224,
      "epoch": 1.18255615234375e-05,
      "step": 86400
    },
    {
      "action_loss_avg": 0.0011950016778428108,
      "epoch": 1.18255615234375e-05,
      "step": 86400
    },
    {
      "epoch": 1.18255615234375e-05,
      "step": 86400,
      "training_step_time": 7.504423379898071
    },
    {
      "epoch": 1.1826515197753906e-05,
      "model_forward_time": 1.9356133937835693,
      "step": 86401
    },
    {
      "epoch": 1.1826515197753906e-05,
      "step": 86401,
      "training_step_time": 7.711435317993164
    },
    {
      "epoch": 1.1827468872070312e-05,
      "model_forward_time": 1.92183256149292,
      "step": 86402
    },
    {
      "epoch": 1.1827468872070312e-05,
      "step": 86402,
      "training_step_time": 7.795945882797241
    },
    {
      "epoch": 1.182842254638672e-05,
      "model_forward_time": 1.908228874206543,
      "step": 86403
    },
    {
      "epoch": 1.182842254638672e-05,
      "step": 86403,
      "training_step_time": 7.686633825302124
    },
    {
      "epoch": 1.1829376220703125e-05,
      "model_forward_time": 1.9140007495880127,
      "step": 86404
    },
    {
      "epoch": 1.1829376220703125e-05,
      "step": 86404,
      "training_step_time": 7.38623571395874
    },
    {
      "epoch": 1.183032989501953e-05,
      "model_forward_time": 1.9214732646942139,
      "step": 86405
    },
    {
      "epoch": 1.183032989501953e-05,
      "step": 86405,
      "training_step_time": 8.083258867263794
    },
    {
      "epoch": 1.1831283569335938e-05,
      "model_forward_time": 1.915271520614624,
      "step": 86406
    },
    {
      "epoch": 1.1831283569335938e-05,
      "step": 86406,
      "training_step_time": 8.974483489990234
    },
    {
      "epoch": 1.1832237243652344e-05,
      "model_forward_time": 1.9146955013275146,
      "step": 86407
    },
    {
      "epoch": 1.1832237243652344e-05,
      "step": 86407,
      "training_step_time": 7.734657526016235
    },
    {
      "epoch": 1.183319091796875e-05,
      "model_forward_time": 1.9216234683990479,
      "step": 86408
    },
    {
      "epoch": 1.183319091796875e-05,
      "step": 86408,
      "training_step_time": 8.616594552993774
    },
    {
      "epoch": 1.1834144592285157e-05,
      "model_forward_time": 1.9072458744049072,
      "step": 86409
    },
    {
      "epoch": 1.1834144592285157e-05,
      "step": 86409,
      "training_step_time": 8.38839340209961
    },
    {
      "epoch": 1.1835098266601563e-05,
      "grad_norm": 0.14343608915805817,
      "learning_rate": 4.965606849357574e-07,
      "loss": 0.0528,
      "step": 86410
    },
    {
      "epoch": 1.1835098266601563e-05,
      "model_forward_time": 1.9129984378814697,
      "step": 86410
    },
    {
      "dynamics_loss_avg": 0.05817738808691501,
      "epoch": 1.1835098266601563e-05,
      "step": 86410
    },
    {
      "action_loss_avg": 0.0010475706571014599,
      "epoch": 1.1835098266601563e-05,
      "step": 86410
    },
    {
      "epoch": 1.1835098266601563e-05,
      "step": 86410,
      "training_step_time": 8.7293062210083
    },
    {
      "epoch": 1.1836051940917968e-05,
      "model_forward_time": 1.9133191108703613,
      "step": 86411
    },
    {
      "epoch": 1.1836051940917968e-05,
      "step": 86411,
      "training_step_time": 8.316325664520264
    },
    {
      "epoch": 1.1837005615234376e-05,
      "model_forward_time": 1.9193992614746094,
      "step": 86412
    },
    {
      "epoch": 1.1837005615234376e-05,
      "step": 86412,
      "training_step_time": 7.6705286502838135
    },
    {
      "epoch": 1.1837959289550781e-05,
      "model_forward_time": 1.9177424907684326,
      "step": 86413
    },
    {
      "epoch": 1.1837959289550781e-05,
      "step": 86413,
      "training_step_time": 7.3940606117248535
    },
    {
      "epoch": 1.1838912963867187e-05,
      "model_forward_time": 1.928607702255249,
      "step": 86414
    },
    {
      "epoch": 1.1838912963867187e-05,
      "step": 86414,
      "training_step_time": 7.550487518310547
    },
    {
      "epoch": 1.1839866638183593e-05,
      "model_forward_time": 1.89774751663208,
      "step": 86415
    },
    {
      "epoch": 1.1839866638183593e-05,
      "step": 86415,
      "training_step_time": 8.515164136886597
    },
    {
      "epoch": 1.18408203125e-05,
      "model_forward_time": 1.9168822765350342,
      "step": 86416
    },
    {
      "epoch": 1.18408203125e-05,
      "step": 86416,
      "training_step_time": 8.3034029006958
    },
    {
      "epoch": 1.1841773986816406e-05,
      "model_forward_time": 1.9133031368255615,
      "step": 86417
    },
    {
      "epoch": 1.1841773986816406e-05,
      "step": 86417,
      "training_step_time": 10.13537883758545
    },
    {
      "epoch": 1.1842727661132812e-05,
      "model_forward_time": 1.7451586723327637,
      "step": 86418
    },
    {
      "epoch": 1.1842727661132812e-05,
      "step": 86418,
      "training_step_time": 7.774235010147095
    },
    {
      "epoch": 1.184368133544922e-05,
      "model_forward_time": 2.4011871814727783,
      "step": 86419
    },
    {
      "epoch": 1.184368133544922e-05,
      "step": 86419,
      "training_step_time": 7.653461217880249
    },
    {
      "epoch": 1.1844635009765625e-05,
      "grad_norm": 0.09556422382593155,
      "learning_rate": 4.958425535040945e-07,
      "loss": 0.0531,
      "step": 86420
    },
    {
      "epoch": 1.1844635009765625e-05,
      "model_forward_time": 1.7491586208343506,
      "step": 86420
    },
    {
      "dynamics_loss_avg": 0.06967045608907937,
      "epoch": 1.1844635009765625e-05,
      "step": 86420
    },
    {
      "action_loss_avg": 0.0018593676271848381,
      "epoch": 1.1844635009765625e-05,
      "step": 86420
    },
    {
      "epoch": 1.1844635009765625e-05,
      "step": 86420,
      "training_step_time": 7.6929097175598145
    },
    {
      "epoch": 1.184558868408203e-05,
      "model_forward_time": 1.750190019607544,
      "step": 86421
    },
    {
      "epoch": 1.184558868408203e-05,
      "step": 86421,
      "training_step_time": 7.690736532211304
    },
    {
      "epoch": 1.1846542358398438e-05,
      "model_forward_time": 1.7584805488586426,
      "step": 86422
    },
    {
      "epoch": 1.1846542358398438e-05,
      "step": 86422,
      "training_step_time": 7.565635442733765
    },
    {
      "epoch": 1.1847496032714844e-05,
      "model_forward_time": 1.7408223152160645,
      "step": 86423
    },
    {
      "epoch": 1.1847496032714844e-05,
      "step": 86423,
      "training_step_time": 8.422438621520996
    },
    {
      "epoch": 1.184844970703125e-05,
      "model_forward_time": 1.7521467208862305,
      "step": 86424
    },
    {
      "epoch": 1.184844970703125e-05,
      "step": 86424,
      "training_step_time": 7.9201130867004395
    },
    {
      "epoch": 1.1849403381347657e-05,
      "model_forward_time": 1.7537128925323486,
      "step": 86425
    },
    {
      "epoch": 1.1849403381347657e-05,
      "step": 86425,
      "training_step_time": 7.82854151725769
    },
    {
      "epoch": 1.1850357055664063e-05,
      "model_forward_time": 1.8691024780273438,
      "step": 86426
    },
    {
      "epoch": 1.1850357055664063e-05,
      "step": 86426,
      "training_step_time": 9.051391363143921
    },
    {
      "epoch": 1.1851310729980468e-05,
      "model_forward_time": 1.747934341430664,
      "step": 86427
    },
    {
      "epoch": 1.1851310729980468e-05,
      "step": 86427,
      "training_step_time": 9.734073877334595
    },
    {
      "epoch": 1.1852264404296876e-05,
      "model_forward_time": 2.370307445526123,
      "step": 86428
    },
    {
      "epoch": 1.1852264404296876e-05,
      "step": 86428,
      "training_step_time": 8.745593547821045
    },
    {
      "epoch": 1.1853218078613282e-05,
      "model_forward_time": 1.7486727237701416,
      "step": 86429
    },
    {
      "epoch": 1.1853218078613282e-05,
      "step": 86429,
      "training_step_time": 7.705167055130005
    },
    {
      "epoch": 1.1854171752929687e-05,
      "grad_norm": 0.09479761123657227,
      "learning_rate": 4.951249146403175e-07,
      "loss": 0.0534,
      "step": 86430
    },
    {
      "epoch": 1.1854171752929687e-05,
      "model_forward_time": 1.757676362991333,
      "step": 86430
    },
    {
      "dynamics_loss_avg": 0.062322180531919,
      "epoch": 1.1854171752929687e-05,
      "step": 86430
    },
    {
      "action_loss_avg": 0.000979552519856952,
      "epoch": 1.1854171752929687e-05,
      "step": 86430
    },
    {
      "epoch": 1.1854171752929687e-05,
      "step": 86430,
      "training_step_time": 7.62188458442688
    },
    {
      "epoch": 1.1855125427246093e-05,
      "model_forward_time": 1.7533674240112305,
      "step": 86431
    },
    {
      "epoch": 1.1855125427246093e-05,
      "step": 86431,
      "training_step_time": 7.579059839248657
    },
    {
      "epoch": 1.18560791015625e-05,
      "model_forward_time": 1.748622179031372,
      "step": 86432
    },
    {
      "epoch": 1.18560791015625e-05,
      "step": 86432,
      "training_step_time": 8.317471742630005
    },
    {
      "epoch": 1.1857032775878906e-05,
      "model_forward_time": 1.7599828243255615,
      "step": 86433
    },
    {
      "epoch": 1.1857032775878906e-05,
      "step": 86433,
      "training_step_time": 8.54691481590271
    },
    {
      "epoch": 1.1857986450195312e-05,
      "model_forward_time": 1.7533042430877686,
      "step": 86434
    },
    {
      "epoch": 1.1857986450195312e-05,
      "step": 86434,
      "training_step_time": 8.279524803161621
    },
    {
      "epoch": 1.185894012451172e-05,
      "model_forward_time": 1.8614177703857422,
      "step": 86435
    },
    {
      "epoch": 1.185894012451172e-05,
      "step": 86435,
      "training_step_time": 7.750931978225708
    },
    {
      "epoch": 1.1859893798828125e-05,
      "model_forward_time": 1.7510473728179932,
      "step": 86436
    },
    {
      "epoch": 1.1859893798828125e-05,
      "step": 86436,
      "training_step_time": 8.828040838241577
    },
    {
      "epoch": 1.186084747314453e-05,
      "model_forward_time": 2.394487142562866,
      "step": 86437
    },
    {
      "epoch": 1.186084747314453e-05,
      "step": 86437,
      "training_step_time": 8.844910860061646
    },
    {
      "epoch": 1.1861801147460938e-05,
      "model_forward_time": 1.7493770122528076,
      "step": 86438
    },
    {
      "epoch": 1.1861801147460938e-05,
      "step": 86438,
      "training_step_time": 8.551238059997559
    },
    {
      "epoch": 1.1862754821777344e-05,
      "model_forward_time": 1.7483067512512207,
      "step": 86439
    },
    {
      "epoch": 1.1862754821777344e-05,
      "step": 86439,
      "training_step_time": 8.824106454849243
    },
    {
      "epoch": 1.186370849609375e-05,
      "grad_norm": 0.22045330703258514,
      "learning_rate": 4.944077684229082e-07,
      "loss": 0.0516,
      "step": 86440
    },
    {
      "epoch": 1.186370849609375e-05,
      "model_forward_time": 1.7426295280456543,
      "step": 86440
    },
    {
      "dynamics_loss_avg": 0.055689406022429465,
      "epoch": 1.186370849609375e-05,
      "step": 86440
    },
    {
      "action_loss_avg": 0.001269932452123612,
      "epoch": 1.186370849609375e-05,
      "step": 86440
    },
    {
      "epoch": 1.186370849609375e-05,
      "step": 86440,
      "training_step_time": 7.619162082672119
    },
    {
      "epoch": 1.1864662170410157e-05,
      "model_forward_time": 1.7526922225952148,
      "step": 86441
    },
    {
      "epoch": 1.1864662170410157e-05,
      "step": 86441,
      "training_step_time": 7.392655849456787
    },
    {
      "epoch": 1.1865615844726563e-05,
      "model_forward_time": 1.7442123889923096,
      "step": 86442
    },
    {
      "epoch": 1.1865615844726563e-05,
      "step": 86442,
      "training_step_time": 7.594678640365601
    },
    {
      "epoch": 1.1866569519042968e-05,
      "model_forward_time": 1.7463877201080322,
      "step": 86443
    },
    {
      "epoch": 1.1866569519042968e-05,
      "step": 86443,
      "training_step_time": 8.451796531677246
    },
    {
      "epoch": 1.1867523193359376e-05,
      "model_forward_time": 1.8506200313568115,
      "step": 86444
    },
    {
      "epoch": 1.1867523193359376e-05,
      "step": 86444,
      "training_step_time": 8.188130140304565
    },
    {
      "epoch": 1.1868476867675782e-05,
      "model_forward_time": 1.7442564964294434,
      "step": 86445
    },
    {
      "epoch": 1.1868476867675782e-05,
      "step": 86445,
      "training_step_time": 8.55253267288208
    },
    {
      "epoch": 1.1869430541992187e-05,
      "model_forward_time": 2.403949499130249,
      "step": 86446
    },
    {
      "epoch": 1.1869430541992187e-05,
      "step": 86446,
      "training_step_time": 8.597071170806885
    },
    {
      "epoch": 1.1870384216308593e-05,
      "model_forward_time": 1.7485952377319336,
      "step": 86447
    },
    {
      "epoch": 1.1870384216308593e-05,
      "step": 86447,
      "training_step_time": 8.534196853637695
    },
    {
      "epoch": 1.1871337890625e-05,
      "model_forward_time": 1.7538683414459229,
      "step": 86448
    },
    {
      "epoch": 1.1871337890625e-05,
      "step": 86448,
      "training_step_time": 7.505934715270996
    },
    {
      "epoch": 1.1872291564941406e-05,
      "model_forward_time": 1.7535405158996582,
      "step": 86449
    },
    {
      "epoch": 1.1872291564941406e-05,
      "step": 86449,
      "training_step_time": 7.567035436630249
    },
    {
      "epoch": 1.1873245239257812e-05,
      "grad_norm": 0.11380917578935623,
      "learning_rate": 4.936911149302903e-07,
      "loss": 0.0515,
      "step": 86450
    },
    {
      "epoch": 1.1873245239257812e-05,
      "model_forward_time": 1.729020118713379,
      "step": 86450
    },
    {
      "dynamics_loss_avg": 0.052388711785897615,
      "epoch": 1.1873245239257812e-05,
      "step": 86450
    },
    {
      "action_loss_avg": 0.0009165258801658638,
      "epoch": 1.1873245239257812e-05,
      "step": 86450
    },
    {
      "epoch": 1.1873245239257812e-05,
      "step": 86450,
      "training_step_time": 7.3305442333221436
    },
    {
      "epoch": 1.187419891357422e-05,
      "model_forward_time": 1.7455635070800781,
      "step": 86451
    },
    {
      "epoch": 1.187419891357422e-05,
      "step": 86451,
      "training_step_time": 8.41569185256958
    },
    {
      "epoch": 1.1875152587890625e-05,
      "model_forward_time": 1.7466835975646973,
      "step": 86452
    },
    {
      "epoch": 1.1875152587890625e-05,
      "step": 86452,
      "training_step_time": 9.341039180755615
    },
    {
      "epoch": 1.187610626220703e-05,
      "model_forward_time": 1.8636670112609863,
      "step": 86453
    },
    {
      "epoch": 1.187610626220703e-05,
      "step": 86453,
      "training_step_time": 8.606180667877197
    },
    {
      "epoch": 1.1877059936523438e-05,
      "model_forward_time": 1.7448155879974365,
      "step": 86454
    },
    {
      "epoch": 1.1877059936523438e-05,
      "step": 86454,
      "training_step_time": 7.653867483139038
    },
    {
      "epoch": 1.1878013610839844e-05,
      "model_forward_time": 2.3974647521972656,
      "step": 86455
    },
    {
      "epoch": 1.1878013610839844e-05,
      "step": 86455,
      "training_step_time": 7.7994160652160645
    },
    {
      "epoch": 1.187896728515625e-05,
      "model_forward_time": 1.7398443222045898,
      "step": 86456
    },
    {
      "epoch": 1.187896728515625e-05,
      "step": 86456,
      "training_step_time": 7.659395217895508
    },
    {
      "epoch": 1.1879920959472657e-05,
      "model_forward_time": 1.744910717010498,
      "step": 86457
    },
    {
      "epoch": 1.1879920959472657e-05,
      "step": 86457,
      "training_step_time": 8.39475965499878
    },
    {
      "epoch": 1.1880874633789063e-05,
      "model_forward_time": 1.7490975856781006,
      "step": 86458
    },
    {
      "epoch": 1.1880874633789063e-05,
      "step": 86458,
      "training_step_time": 7.570512056350708
    },
    {
      "epoch": 1.1881828308105468e-05,
      "model_forward_time": 1.7383615970611572,
      "step": 86459
    },
    {
      "epoch": 1.1881828308105468e-05,
      "step": 86459,
      "training_step_time": 7.373847961425781
    },
    {
      "epoch": 1.1882781982421876e-05,
      "grad_norm": 0.08013661205768585,
      "learning_rate": 4.92974954240838e-07,
      "loss": 0.0524,
      "step": 86460
    },
    {
      "epoch": 1.1882781982421876e-05,
      "model_forward_time": 1.7434489727020264,
      "step": 86460
    },
    {
      "dynamics_loss_avg": 0.045087302755564454,
      "epoch": 1.1882781982421876e-05,
      "step": 86460
    },
    {
      "action_loss_avg": 0.0008878242457285524,
      "epoch": 1.1882781982421876e-05,
      "step": 86460
    },
    {
      "epoch": 1.1882781982421876e-05,
      "step": 86460,
      "training_step_time": 7.546924352645874
    },
    {
      "epoch": 1.1883735656738282e-05,
      "model_forward_time": 1.7429537773132324,
      "step": 86461
    },
    {
      "epoch": 1.1883735656738282e-05,
      "step": 86461,
      "training_step_time": 8.566948175430298
    },
    {
      "epoch": 1.1884689331054687e-05,
      "model_forward_time": 1.8605949878692627,
      "step": 86462
    },
    {
      "epoch": 1.1884689331054687e-05,
      "step": 86462,
      "training_step_time": 8.448815822601318
    },
    {
      "epoch": 1.1885643005371093e-05,
      "model_forward_time": 1.7490594387054443,
      "step": 86463
    },
    {
      "epoch": 1.1885643005371093e-05,
      "step": 86463,
      "training_step_time": 9.96102786064148
    },
    {
      "epoch": 1.18865966796875e-05,
      "model_forward_time": 1.7404749393463135,
      "step": 86464
    },
    {
      "epoch": 1.18865966796875e-05,
      "step": 86464,
      "training_step_time": 8.220399856567383
    },
    {
      "epoch": 1.1887550354003906e-05,
      "model_forward_time": 1.7432372570037842,
      "step": 86465
    },
    {
      "epoch": 1.1887550354003906e-05,
      "step": 86465,
      "training_step_time": 8.272401332855225
    },
    {
      "epoch": 1.1888504028320312e-05,
      "model_forward_time": 1.7473506927490234,
      "step": 86466
    },
    {
      "epoch": 1.1888504028320312e-05,
      "step": 86466,
      "training_step_time": 8.26360559463501
    },
    {
      "epoch": 1.188945770263672e-05,
      "model_forward_time": 1.7345614433288574,
      "step": 86467
    },
    {
      "epoch": 1.188945770263672e-05,
      "step": 86467,
      "training_step_time": 9.360121726989746
    },
    {
      "epoch": 1.1890411376953125e-05,
      "model_forward_time": 1.741499900817871,
      "step": 86468
    },
    {
      "epoch": 1.1890411376953125e-05,
      "step": 86468,
      "training_step_time": 7.60802435874939
    },
    {
      "epoch": 1.189136505126953e-05,
      "model_forward_time": 1.7674322128295898,
      "step": 86469
    },
    {
      "epoch": 1.189136505126953e-05,
      "step": 86469,
      "training_step_time": 8.546405792236328
    },
    {
      "epoch": 1.1892318725585938e-05,
      "grad_norm": 0.08946394175291061,
      "learning_rate": 4.922592864328673e-07,
      "loss": 0.0537,
      "step": 86470
    },
    {
      "epoch": 1.1892318725585938e-05,
      "model_forward_time": 1.7347173690795898,
      "step": 86470
    },
    {
      "dynamics_loss_avg": 0.06438715923577547,
      "epoch": 1.1892318725585938e-05,
      "step": 86470
    },
    {
      "action_loss_avg": 0.0015480969945201651,
      "epoch": 1.1892318725585938e-05,
      "step": 86470
    },
    {
      "epoch": 1.1892318725585938e-05,
      "step": 86470,
      "training_step_time": 8.157789945602417
    },
    {
      "epoch": 1.1893272399902344e-05,
      "model_forward_time": 1.0308911800384521,
      "step": 86471
    },
    {
      "epoch": 1.1893272399902344e-05,
      "step": 86471,
      "training_step_time": 7.714667797088623
    },
    {
      "epoch": 1.189422607421875e-05,
      "model_forward_time": 1.7232801914215088,
      "step": 86472
    },
    {
      "epoch": 1.189422607421875e-05,
      "step": 86472,
      "training_step_time": 10.206804752349854
    },
    {
      "epoch": 1.1895179748535157e-05,
      "model_forward_time": 1.7357378005981445,
      "step": 86473
    },
    {
      "epoch": 1.1895179748535157e-05,
      "step": 86473,
      "training_step_time": 7.750934839248657
    },
    {
      "epoch": 1.1896133422851563e-05,
      "model_forward_time": 2.3846311569213867,
      "step": 86474
    },
    {
      "epoch": 1.1896133422851563e-05,
      "step": 86474,
      "training_step_time": 7.65430760383606
    },
    {
      "epoch": 1.1897087097167968e-05,
      "model_forward_time": 1.7474124431610107,
      "step": 86475
    },
    {
      "epoch": 1.1897087097167968e-05,
      "step": 86475,
      "training_step_time": 7.585768461227417
    },
    {
      "epoch": 1.1898040771484376e-05,
      "model_forward_time": 1.742931604385376,
      "step": 86476
    },
    {
      "epoch": 1.1898040771484376e-05,
      "step": 86476,
      "training_step_time": 7.628369569778442
    },
    {
      "epoch": 1.1898994445800782e-05,
      "model_forward_time": 1.7520639896392822,
      "step": 86477
    },
    {
      "epoch": 1.1898994445800782e-05,
      "step": 86477,
      "training_step_time": 7.646602153778076
    },
    {
      "epoch": 1.1899948120117187e-05,
      "model_forward_time": 1.7412474155426025,
      "step": 86478
    },
    {
      "epoch": 1.1899948120117187e-05,
      "step": 86478,
      "training_step_time": 7.592707872390747
    },
    {
      "epoch": 1.1900901794433593e-05,
      "model_forward_time": 1.740502119064331,
      "step": 86479
    },
    {
      "epoch": 1.1900901794433593e-05,
      "step": 86479,
      "training_step_time": 7.531017780303955
    },
    {
      "epoch": 1.190185546875e-05,
      "grad_norm": 0.06773798912763596,
      "learning_rate": 4.915441115846436e-07,
      "loss": 0.0523,
      "step": 86480
    },
    {
      "epoch": 1.190185546875e-05,
      "model_forward_time": 1.806044578552246,
      "step": 86480
    },
    {
      "dynamics_loss_avg": 0.0532541056163609,
      "epoch": 1.190185546875e-05,
      "step": 86480
    },
    {
      "action_loss_avg": 0.001300334164989181,
      "epoch": 1.190185546875e-05,
      "step": 86480
    },
    {
      "epoch": 1.190185546875e-05,
      "step": 86480,
      "training_step_time": 8.19972276687622
    },
    {
      "epoch": 1.1902809143066406e-05,
      "model_forward_time": 1.7447595596313477,
      "step": 86481
    },
    {
      "epoch": 1.1902809143066406e-05,
      "step": 86481,
      "training_step_time": 9.72002649307251
    },
    {
      "epoch": 1.1903762817382812e-05,
      "model_forward_time": 1.7463915348052979,
      "step": 86482
    },
    {
      "epoch": 1.1903762817382812e-05,
      "step": 86482,
      "training_step_time": 8.804327487945557
    },
    {
      "epoch": 1.190471649169922e-05,
      "model_forward_time": 2.403691291809082,
      "step": 86483
    },
    {
      "epoch": 1.190471649169922e-05,
      "step": 86483,
      "training_step_time": 8.694054365158081
    },
    {
      "epoch": 1.1905670166015625e-05,
      "model_forward_time": 1.7354447841644287,
      "step": 86484
    },
    {
      "epoch": 1.1905670166015625e-05,
      "step": 86484,
      "training_step_time": 8.470820903778076
    },
    {
      "epoch": 1.190662384033203e-05,
      "model_forward_time": 1.7448105812072754,
      "step": 86485
    },
    {
      "epoch": 1.190662384033203e-05,
      "step": 86485,
      "training_step_time": 7.558733940124512
    },
    {
      "epoch": 1.1907577514648438e-05,
      "model_forward_time": 1.749204158782959,
      "step": 86486
    },
    {
      "epoch": 1.1907577514648438e-05,
      "step": 86486,
      "training_step_time": 7.587780952453613
    },
    {
      "epoch": 1.1908531188964844e-05,
      "model_forward_time": 1.746610403060913,
      "step": 86487
    },
    {
      "epoch": 1.1908531188964844e-05,
      "step": 86487,
      "training_step_time": 7.58822774887085
    },
    {
      "epoch": 1.190948486328125e-05,
      "model_forward_time": 1.7378284931182861,
      "step": 86488
    },
    {
      "epoch": 1.190948486328125e-05,
      "step": 86488,
      "training_step_time": 7.534512996673584
    },
    {
      "epoch": 1.1910438537597657e-05,
      "model_forward_time": 1.7783102989196777,
      "step": 86489
    },
    {
      "epoch": 1.1910438537597657e-05,
      "step": 86489,
      "training_step_time": 9.537558555603027
    },
    {
      "epoch": 1.1911392211914063e-05,
      "grad_norm": 0.07705073803663254,
      "learning_rate": 4.908294297743782e-07,
      "loss": 0.0524,
      "step": 86490
    },
    {
      "epoch": 1.1911392211914063e-05,
      "model_forward_time": 1.743177890777588,
      "step": 86490
    },
    {
      "dynamics_loss_avg": 0.049575235694646835,
      "epoch": 1.1911392211914063e-05,
      "step": 86490
    },
    {
      "action_loss_avg": 0.0015991912980098278,
      "epoch": 1.1911392211914063e-05,
      "step": 86490
    },
    {
      "epoch": 1.1911392211914063e-05,
      "step": 86490,
      "training_step_time": 9.463140964508057
    },
    {
      "epoch": 1.1912345886230468e-05,
      "model_forward_time": 1.7379841804504395,
      "step": 86491
    },
    {
      "epoch": 1.1912345886230468e-05,
      "step": 86491,
      "training_step_time": 7.718116998672485
    },
    {
      "epoch": 1.1913299560546876e-05,
      "model_forward_time": 2.3697633743286133,
      "step": 86492
    },
    {
      "epoch": 1.1913299560546876e-05,
      "step": 86492,
      "training_step_time": 9.55801773071289
    },
    {
      "epoch": 1.1914253234863282e-05,
      "model_forward_time": 1.7449371814727783,
      "step": 86493
    },
    {
      "epoch": 1.1914253234863282e-05,
      "step": 86493,
      "training_step_time": 8.473179340362549
    },
    {
      "epoch": 1.1915206909179687e-05,
      "model_forward_time": 1.7537758350372314,
      "step": 86494
    },
    {
      "epoch": 1.1915206909179687e-05,
      "step": 86494,
      "training_step_time": 7.54594612121582
    },
    {
      "epoch": 1.1916160583496093e-05,
      "model_forward_time": 1.762434720993042,
      "step": 86495
    },
    {
      "epoch": 1.1916160583496093e-05,
      "step": 86495,
      "training_step_time": 7.605926275253296
    },
    {
      "epoch": 1.19171142578125e-05,
      "model_forward_time": 1.7577595710754395,
      "step": 86496
    },
    {
      "epoch": 1.19171142578125e-05,
      "step": 86496,
      "training_step_time": 7.572556018829346
    },
    {
      "epoch": 1.1918067932128906e-05,
      "model_forward_time": 1.7493865489959717,
      "step": 86497
    },
    {
      "epoch": 1.1918067932128906e-05,
      "step": 86497,
      "training_step_time": 8.22595500946045
    },
    {
      "epoch": 1.1919021606445312e-05,
      "model_forward_time": 1.7392160892486572,
      "step": 86498
    },
    {
      "epoch": 1.1919021606445312e-05,
      "step": 86498,
      "training_step_time": 10.730405807495117
    },
    {
      "epoch": 1.191997528076172e-05,
      "model_forward_time": 1.745805025100708,
      "step": 86499
    },
    {
      "epoch": 1.191997528076172e-05,
      "step": 86499,
      "training_step_time": 7.624001502990723
    },
    {
      "epoch": 1.1920928955078125e-05,
      "grad_norm": 0.08847672492265701,
      "learning_rate": 4.901152410802274e-07,
      "loss": 0.0539,
      "step": 86500
    },
    {
      "epoch": 1.1920928955078125e-05,
      "model_forward_time": 1.7384839057922363,
      "step": 86500
    },
    {
      "dynamics_loss_avg": 0.04773893468081951,
      "epoch": 1.1920928955078125e-05,
      "step": 86500
    },
    {
      "action_loss_avg": 0.001099840204551583,
      "epoch": 1.1920928955078125e-05,
      "step": 86500
    },
    {
      "epoch": 1.1920928955078125e-05,
      "step": 86500,
      "training_step_time": 8.4951012134552
    },
    {
      "epoch": 1.192188262939453e-05,
      "model_forward_time": 1.7411091327667236,
      "step": 86501
    },
    {
      "epoch": 1.192188262939453e-05,
      "step": 86501,
      "training_step_time": 9.411134958267212
    },
    {
      "epoch": 1.1922836303710938e-05,
      "model_forward_time": 2.3819611072540283,
      "step": 86502
    },
    {
      "epoch": 1.1922836303710938e-05,
      "step": 86502,
      "training_step_time": 9.016097068786621
    },
    {
      "epoch": 1.1923789978027344e-05,
      "model_forward_time": 1.7447001934051514,
      "step": 86503
    },
    {
      "epoch": 1.1923789978027344e-05,
      "step": 86503,
      "training_step_time": 8.191539525985718
    },
    {
      "epoch": 1.192474365234375e-05,
      "model_forward_time": 1.7360024452209473,
      "step": 86504
    },
    {
      "epoch": 1.192474365234375e-05,
      "step": 86504,
      "training_step_time": 7.687469244003296
    },
    {
      "epoch": 1.1925697326660157e-05,
      "model_forward_time": 1.738339900970459,
      "step": 86505
    },
    {
      "epoch": 1.1925697326660157e-05,
      "step": 86505,
      "training_step_time": 7.631181716918945
    },
    {
      "epoch": 1.1926651000976563e-05,
      "model_forward_time": 1.7494606971740723,
      "step": 86506
    },
    {
      "epoch": 1.1926651000976563e-05,
      "step": 86506,
      "training_step_time": 7.506871223449707
    },
    {
      "epoch": 1.1927604675292968e-05,
      "model_forward_time": 1.7467639446258545,
      "step": 86507
    },
    {
      "epoch": 1.1927604675292968e-05,
      "step": 86507,
      "training_step_time": 8.333117723464966
    },
    {
      "epoch": 1.1928558349609374e-05,
      "model_forward_time": 1.8643369674682617,
      "step": 86508
    },
    {
      "epoch": 1.1928558349609374e-05,
      "step": 86508,
      "training_step_time": 9.093897104263306
    },
    {
      "epoch": 1.1929512023925782e-05,
      "model_forward_time": 1.7439911365509033,
      "step": 86509
    },
    {
      "epoch": 1.1929512023925782e-05,
      "step": 86509,
      "training_step_time": 8.612165689468384
    },
    {
      "epoch": 1.1930465698242187e-05,
      "grad_norm": 0.10226213932037354,
      "learning_rate": 4.894015455802919e-07,
      "loss": 0.0545,
      "step": 86510
    },
    {
      "epoch": 1.1930465698242187e-05,
      "model_forward_time": 1.7408013343811035,
      "step": 86510
    },
    {
      "dynamics_loss_avg": 0.04879443235695362,
      "epoch": 1.1930465698242187e-05,
      "step": 86510
    },
    {
      "action_loss_avg": 0.001034681935561821,
      "epoch": 1.1930465698242187e-05,
      "step": 86510
    },
    {
      "epoch": 1.1930465698242187e-05,
      "step": 86510,
      "training_step_time": 7.685268402099609
    },
    {
      "epoch": 1.1931419372558593e-05,
      "model_forward_time": 2.414083480834961,
      "step": 86511
    },
    {
      "epoch": 1.1931419372558593e-05,
      "step": 86511,
      "training_step_time": 9.048030138015747
    },
    {
      "epoch": 1.1932373046875e-05,
      "model_forward_time": 1.7473289966583252,
      "step": 86512
    },
    {
      "epoch": 1.1932373046875e-05,
      "step": 86512,
      "training_step_time": 7.631486892700195
    },
    {
      "epoch": 1.1933326721191406e-05,
      "model_forward_time": 1.7427380084991455,
      "step": 86513
    },
    {
      "epoch": 1.1933326721191406e-05,
      "step": 86513,
      "training_step_time": 7.653592824935913
    },
    {
      "epoch": 1.1934280395507812e-05,
      "model_forward_time": 1.753235101699829,
      "step": 86514
    },
    {
      "epoch": 1.1934280395507812e-05,
      "step": 86514,
      "training_step_time": 7.748083114624023
    },
    {
      "epoch": 1.193523406982422e-05,
      "model_forward_time": 1.7666380405426025,
      "step": 86515
    },
    {
      "epoch": 1.193523406982422e-05,
      "step": 86515,
      "training_step_time": 8.371689081192017
    },
    {
      "epoch": 1.1936187744140625e-05,
      "model_forward_time": 1.7467527389526367,
      "step": 86516
    },
    {
      "epoch": 1.1936187744140625e-05,
      "step": 86516,
      "training_step_time": 8.048246622085571
    },
    {
      "epoch": 1.193714141845703e-05,
      "model_forward_time": 1.8635015487670898,
      "step": 86517
    },
    {
      "epoch": 1.193714141845703e-05,
      "step": 86517,
      "training_step_time": 10.43223786354065
    },
    {
      "epoch": 1.1938095092773438e-05,
      "model_forward_time": 1.7529137134552002,
      "step": 86518
    },
    {
      "epoch": 1.1938095092773438e-05,
      "step": 86518,
      "training_step_time": 7.626894950866699
    },
    {
      "epoch": 1.1939048767089844e-05,
      "model_forward_time": 1.7494397163391113,
      "step": 86519
    },
    {
      "epoch": 1.1939048767089844e-05,
      "step": 86519,
      "training_step_time": 8.671687126159668
    },
    {
      "epoch": 1.194000244140625e-05,
      "grad_norm": 0.07694787532091141,
      "learning_rate": 4.886883433526235e-07,
      "loss": 0.0556,
      "step": 86520
    },
    {
      "epoch": 1.194000244140625e-05,
      "model_forward_time": 2.38141131401062,
      "step": 86520
    },
    {
      "dynamics_loss_avg": 0.04600999355316162,
      "epoch": 1.194000244140625e-05,
      "step": 86520
    },
    {
      "action_loss_avg": 0.0008959711529314518,
      "epoch": 1.194000244140625e-05,
      "step": 86520
    },
    {
      "epoch": 1.194000244140625e-05,
      "step": 86520,
      "training_step_time": 7.6631410121917725
    },
    {
      "epoch": 1.1940956115722657e-05,
      "model_forward_time": 1.759617567062378,
      "step": 86521
    },
    {
      "epoch": 1.1940956115722657e-05,
      "step": 86521,
      "training_step_time": 8.684929370880127
    },
    {
      "epoch": 1.1941909790039063e-05,
      "model_forward_time": 1.3921794891357422,
      "step": 86522
    },
    {
      "epoch": 1.1941909790039063e-05,
      "step": 86522,
      "training_step_time": 9.323683977127075
    },
    {
      "epoch": 1.1942863464355468e-05,
      "model_forward_time": 1.749769687652588,
      "step": 86523
    },
    {
      "epoch": 1.1942863464355468e-05,
      "step": 86523,
      "training_step_time": 7.581110954284668
    },
    {
      "epoch": 1.1943817138671874e-05,
      "model_forward_time": 1.7493436336517334,
      "step": 86524
    },
    {
      "epoch": 1.1943817138671874e-05,
      "step": 86524,
      "training_step_time": 7.6833655834198
    },
    {
      "epoch": 1.1944770812988282e-05,
      "model_forward_time": 1.7507970333099365,
      "step": 86525
    },
    {
      "epoch": 1.1944770812988282e-05,
      "step": 86525,
      "training_step_time": 8.314745903015137
    },
    {
      "epoch": 1.1945724487304687e-05,
      "model_forward_time": 1.8668994903564453,
      "step": 86526
    },
    {
      "epoch": 1.1945724487304687e-05,
      "step": 86526,
      "training_step_time": 8.360033512115479
    },
    {
      "epoch": 1.1946678161621093e-05,
      "model_forward_time": 1.7567296028137207,
      "step": 86527
    },
    {
      "epoch": 1.1946678161621093e-05,
      "step": 86527,
      "training_step_time": 8.194085121154785
    },
    {
      "epoch": 1.19476318359375e-05,
      "model_forward_time": 1.747868299484253,
      "step": 86528
    },
    {
      "epoch": 1.19476318359375e-05,
      "step": 86528,
      "training_step_time": 8.176934003829956
    },
    {
      "epoch": 1.1948585510253906e-05,
      "model_forward_time": 2.4102253913879395,
      "step": 86529
    },
    {
      "epoch": 1.1948585510253906e-05,
      "step": 86529,
      "training_step_time": 8.39340591430664
    },
    {
      "epoch": 1.1949539184570312e-05,
      "grad_norm": 0.08189380913972855,
      "learning_rate": 4.87975634475214e-07,
      "loss": 0.0548,
      "step": 86530
    },
    {
      "epoch": 1.1949539184570312e-05,
      "model_forward_time": 1.7507617473602295,
      "step": 86530
    },
    {
      "dynamics_loss_avg": 0.034463748708367346,
      "epoch": 1.1949539184570312e-05,
      "step": 86530
    },
    {
      "action_loss_avg": 0.0010280746355419978,
      "epoch": 1.1949539184570312e-05,
      "step": 86530
    },
    {
      "epoch": 1.1949539184570312e-05,
      "step": 86530,
      "training_step_time": 7.649508237838745
    },
    {
      "epoch": 1.195049285888672e-05,
      "model_forward_time": 1.7456681728363037,
      "step": 86531
    },
    {
      "epoch": 1.195049285888672e-05,
      "step": 86531,
      "training_step_time": 7.587156295776367
    },
    {
      "epoch": 1.1951446533203125e-05,
      "model_forward_time": 1.7438406944274902,
      "step": 86532
    },
    {
      "epoch": 1.1951446533203125e-05,
      "step": 86532,
      "training_step_time": 7.5443947315216064
    },
    {
      "epoch": 1.195240020751953e-05,
      "model_forward_time": 1.7501404285430908,
      "step": 86533
    },
    {
      "epoch": 1.195240020751953e-05,
      "step": 86533,
      "training_step_time": 7.688625812530518
    },
    {
      "epoch": 1.1953353881835938e-05,
      "model_forward_time": 1.7530007362365723,
      "step": 86534
    },
    {
      "epoch": 1.1953353881835938e-05,
      "step": 86534,
      "training_step_time": 7.459605693817139
    },
    {
      "epoch": 1.1954307556152344e-05,
      "model_forward_time": 1.8563244342803955,
      "step": 86535
    },
    {
      "epoch": 1.1954307556152344e-05,
      "step": 86535,
      "training_step_time": 9.191656827926636
    },
    {
      "epoch": 1.195526123046875e-05,
      "model_forward_time": 1.7422692775726318,
      "step": 86536
    },
    {
      "epoch": 1.195526123046875e-05,
      "step": 86536,
      "training_step_time": 8.84090256690979
    },
    {
      "epoch": 1.1956214904785157e-05,
      "model_forward_time": 1.74593186378479,
      "step": 86537
    },
    {
      "epoch": 1.1956214904785157e-05,
      "step": 86537,
      "training_step_time": 7.689340829849243
    },
    {
      "epoch": 1.1957168579101563e-05,
      "model_forward_time": 1.3756773471832275,
      "step": 86538
    },
    {
      "epoch": 1.1957168579101563e-05,
      "step": 86538,
      "training_step_time": 8.365393161773682
    },
    {
      "epoch": 1.1958122253417968e-05,
      "model_forward_time": 2.397597074508667,
      "step": 86539
    },
    {
      "epoch": 1.1958122253417968e-05,
      "step": 86539,
      "training_step_time": 7.6232006549835205
    },
    {
      "epoch": 1.1959075927734374e-05,
      "grad_norm": 0.07322739064693451,
      "learning_rate": 4.872634190260062e-07,
      "loss": 0.0554,
      "step": 86540
    },
    {
      "epoch": 1.1959075927734374e-05,
      "model_forward_time": 1.744708776473999,
      "step": 86540
    },
    {
      "dynamics_loss_avg": 0.051417671609669924,
      "epoch": 1.1959075927734374e-05,
      "step": 86540
    },
    {
      "action_loss_avg": 0.0007733221922535449,
      "epoch": 1.1959075927734374e-05,
      "step": 86540
    },
    {
      "epoch": 1.1959075927734374e-05,
      "step": 86540,
      "training_step_time": 7.619826316833496
    },
    {
      "epoch": 1.1960029602050782e-05,
      "model_forward_time": 1.7492389678955078,
      "step": 86541
    },
    {
      "epoch": 1.1960029602050782e-05,
      "step": 86541,
      "training_step_time": 7.584983825683594
    },
    {
      "epoch": 1.1960983276367187e-05,
      "model_forward_time": 1.7508690357208252,
      "step": 86542
    },
    {
      "epoch": 1.1960983276367187e-05,
      "step": 86542,
      "training_step_time": 7.748340368270874
    },
    {
      "epoch": 1.1961936950683593e-05,
      "model_forward_time": 1.7685003280639648,
      "step": 86543
    },
    {
      "epoch": 1.1961936950683593e-05,
      "step": 86543,
      "training_step_time": 7.508557081222534
    },
    {
      "epoch": 1.1962890625e-05,
      "model_forward_time": 1.8539996147155762,
      "step": 86544
    },
    {
      "epoch": 1.1962890625e-05,
      "step": 86544,
      "training_step_time": 8.463267803192139
    },
    {
      "epoch": 1.1963844299316406e-05,
      "model_forward_time": 1.7495718002319336,
      "step": 86545
    },
    {
      "epoch": 1.1963844299316406e-05,
      "step": 86545,
      "training_step_time": 8.56535029411316
    },
    {
      "epoch": 1.1964797973632812e-05,
      "model_forward_time": 1.7438240051269531,
      "step": 86546
    },
    {
      "epoch": 1.1964797973632812e-05,
      "step": 86546,
      "training_step_time": 8.347896814346313
    },
    {
      "epoch": 1.196575164794922e-05,
      "model_forward_time": 1.7444090843200684,
      "step": 86547
    },
    {
      "epoch": 1.196575164794922e-05,
      "step": 86547,
      "training_step_time": 8.794212341308594
    },
    {
      "epoch": 1.1966705322265625e-05,
      "model_forward_time": 1.9940221309661865,
      "step": 86548
    },
    {
      "epoch": 1.1966705322265625e-05,
      "step": 86548,
      "training_step_time": 8.70332407951355
    },
    {
      "epoch": 1.196765899658203e-05,
      "model_forward_time": 1.744811773300171,
      "step": 86549
    },
    {
      "epoch": 1.196765899658203e-05,
      "step": 86549,
      "training_step_time": 7.750535726547241
    },
    {
      "epoch": 1.1968612670898438e-05,
      "grad_norm": 0.09246626496315002,
      "learning_rate": 4.865516970828843e-07,
      "loss": 0.0549,
      "step": 86550
    },
    {
      "epoch": 1.1968612670898438e-05,
      "model_forward_time": 1.7427740097045898,
      "step": 86550
    },
    {
      "dynamics_loss_avg": 0.04816649630665779,
      "epoch": 1.1968612670898438e-05,
      "step": 86550
    },
    {
      "action_loss_avg": 0.001436154416296631,
      "epoch": 1.1968612670898438e-05,
      "step": 86550
    },
    {
      "epoch": 1.1968612670898438e-05,
      "step": 86550,
      "training_step_time": 8.470937728881836
    },
    {
      "epoch": 1.1969566345214844e-05,
      "model_forward_time": 1.7747879028320312,
      "step": 86551
    },
    {
      "epoch": 1.1969566345214844e-05,
      "step": 86551,
      "training_step_time": 7.749785900115967
    },
    {
      "epoch": 1.197052001953125e-05,
      "model_forward_time": 1.8049869537353516,
      "step": 86552
    },
    {
      "epoch": 1.197052001953125e-05,
      "step": 86552,
      "training_step_time": 7.52366042137146
    },
    {
      "epoch": 1.1971473693847657e-05,
      "model_forward_time": 1.7814390659332275,
      "step": 86553
    },
    {
      "epoch": 1.1971473693847657e-05,
      "step": 86553,
      "training_step_time": 8.909423828125
    },
    {
      "epoch": 1.1972427368164063e-05,
      "model_forward_time": 1.7511286735534668,
      "step": 86554
    },
    {
      "epoch": 1.1972427368164063e-05,
      "step": 86554,
      "training_step_time": 7.647017478942871
    },
    {
      "epoch": 1.1973381042480468e-05,
      "model_forward_time": 1.7476167678833008,
      "step": 86555
    },
    {
      "epoch": 1.1973381042480468e-05,
      "step": 86555,
      "training_step_time": 8.17768406867981
    },
    {
      "epoch": 1.1974334716796874e-05,
      "model_forward_time": 1.7478604316711426,
      "step": 86556
    },
    {
      "epoch": 1.1974334716796874e-05,
      "step": 86556,
      "training_step_time": 7.662051439285278
    },
    {
      "epoch": 1.1975288391113282e-05,
      "model_forward_time": 2.3899307250976562,
      "step": 86557
    },
    {
      "epoch": 1.1975288391113282e-05,
      "step": 86557,
      "training_step_time": 7.638423681259155
    },
    {
      "epoch": 1.1976242065429687e-05,
      "model_forward_time": 1.7692244052886963,
      "step": 86558
    },
    {
      "epoch": 1.1976242065429687e-05,
      "step": 86558,
      "training_step_time": 7.706162214279175
    },
    {
      "epoch": 1.1977195739746093e-05,
      "model_forward_time": 1.7465922832489014,
      "step": 86559
    },
    {
      "epoch": 1.1977195739746093e-05,
      "step": 86559,
      "training_step_time": 7.6169397830963135
    },
    {
      "epoch": 1.19781494140625e-05,
      "grad_norm": 0.12201522290706635,
      "learning_rate": 4.858404687236851e-07,
      "loss": 0.0537,
      "step": 86560
    },
    {
      "epoch": 1.19781494140625e-05,
      "model_forward_time": 1.0347461700439453,
      "step": 86560
    },
    {
      "dynamics_loss_avg": 0.05686724130064249,
      "epoch": 1.19781494140625e-05,
      "step": 86560
    },
    {
      "action_loss_avg": 0.0014303371717687696,
      "epoch": 1.19781494140625e-05,
      "step": 86560
    },
    {
      "epoch": 1.19781494140625e-05,
      "step": 86560,
      "training_step_time": 7.65985894203186
    },
    {
      "epoch": 1.1979103088378906e-05,
      "model_forward_time": 1.7547075748443604,
      "step": 86561
    },
    {
      "epoch": 1.1979103088378906e-05,
      "step": 86561,
      "training_step_time": 7.708590030670166
    },
    {
      "epoch": 1.1980056762695312e-05,
      "model_forward_time": 1.7526588439941406,
      "step": 86562
    },
    {
      "epoch": 1.1980056762695312e-05,
      "step": 86562,
      "training_step_time": 7.597519874572754
    },
    {
      "epoch": 1.198101043701172e-05,
      "model_forward_time": 1.7438628673553467,
      "step": 86563
    },
    {
      "epoch": 1.198101043701172e-05,
      "step": 86563,
      "training_step_time": 7.706640720367432
    },
    {
      "epoch": 1.1981964111328125e-05,
      "model_forward_time": 1.7550098896026611,
      "step": 86564
    },
    {
      "epoch": 1.1981964111328125e-05,
      "step": 86564,
      "training_step_time": 7.7446019649505615
    },
    {
      "epoch": 1.198291778564453e-05,
      "model_forward_time": 1.0492448806762695,
      "step": 86565
    },
    {
      "epoch": 1.198291778564453e-05,
      "step": 86565,
      "training_step_time": 9.128500938415527
    },
    {
      "epoch": 1.1983871459960938e-05,
      "model_forward_time": 1.7465662956237793,
      "step": 86566
    },
    {
      "epoch": 1.1983871459960938e-05,
      "step": 86566,
      "training_step_time": 7.643813848495483
    },
    {
      "epoch": 1.1984825134277344e-05,
      "model_forward_time": 2.3804879188537598,
      "step": 86567
    },
    {
      "epoch": 1.1984825134277344e-05,
      "step": 86567,
      "training_step_time": 7.673323631286621
    },
    {
      "epoch": 1.198577880859375e-05,
      "model_forward_time": 1.7449491024017334,
      "step": 86568
    },
    {
      "epoch": 1.198577880859375e-05,
      "step": 86568,
      "training_step_time": 7.635220527648926
    },
    {
      "epoch": 1.1986732482910157e-05,
      "model_forward_time": 1.7514369487762451,
      "step": 86569
    },
    {
      "epoch": 1.1986732482910157e-05,
      "step": 86569,
      "training_step_time": 7.678627252578735
    },
    {
      "epoch": 1.1987686157226563e-05,
      "grad_norm": 0.10281930863857269,
      "learning_rate": 4.85129734026184e-07,
      "loss": 0.0539,
      "step": 86570
    },
    {
      "epoch": 1.1987686157226563e-05,
      "model_forward_time": 1.7519409656524658,
      "step": 86570
    },
    {
      "dynamics_loss_avg": 0.048917172476649284,
      "epoch": 1.1987686157226563e-05,
      "step": 86570
    },
    {
      "action_loss_avg": 0.0008288509416161105,
      "epoch": 1.1987686157226563e-05,
      "step": 86570
    },
    {
      "epoch": 1.1987686157226563e-05,
      "step": 86570,
      "training_step_time": 7.733482599258423
    },
    {
      "epoch": 1.1988639831542968e-05,
      "model_forward_time": 1.742175817489624,
      "step": 86571
    },
    {
      "epoch": 1.1988639831542968e-05,
      "step": 86571,
      "training_step_time": 8.134254455566406
    },
    {
      "epoch": 1.1989593505859374e-05,
      "model_forward_time": 1.862330675125122,
      "step": 86572
    },
    {
      "epoch": 1.1989593505859374e-05,
      "step": 86572,
      "training_step_time": 7.6680004596710205
    },
    {
      "epoch": 1.1990547180175782e-05,
      "model_forward_time": 1.748011827468872,
      "step": 86573
    },
    {
      "epoch": 1.1990547180175782e-05,
      "step": 86573,
      "training_step_time": 7.711514472961426
    },
    {
      "epoch": 1.1991500854492187e-05,
      "model_forward_time": 1.7500898838043213,
      "step": 86574
    },
    {
      "epoch": 1.1991500854492187e-05,
      "step": 86574,
      "training_step_time": 7.783986330032349
    },
    {
      "epoch": 1.1992454528808593e-05,
      "model_forward_time": 1.7439677715301514,
      "step": 86575
    },
    {
      "epoch": 1.1992454528808593e-05,
      "step": 86575,
      "training_step_time": 7.689731597900391
    },
    {
      "epoch": 1.1993408203125e-05,
      "model_forward_time": 1.6414570808410645,
      "step": 86576
    },
    {
      "epoch": 1.1993408203125e-05,
      "step": 86576,
      "training_step_time": 9.422341585159302
    },
    {
      "epoch": 1.1994361877441406e-05,
      "model_forward_time": 1.760352611541748,
      "step": 86577
    },
    {
      "epoch": 1.1994361877441406e-05,
      "step": 86577,
      "training_step_time": 8.701010465621948
    },
    {
      "epoch": 1.1995315551757812e-05,
      "model_forward_time": 1.744431495666504,
      "step": 86578
    },
    {
      "epoch": 1.1995315551757812e-05,
      "step": 86578,
      "training_step_time": 8.396928548812866
    },
    {
      "epoch": 1.199626922607422e-05,
      "model_forward_time": 1.7527117729187012,
      "step": 86579
    },
    {
      "epoch": 1.199626922607422e-05,
      "step": 86579,
      "training_step_time": 7.657934188842773
    },
    {
      "epoch": 1.1997222900390625e-05,
      "grad_norm": 0.0828729122877121,
      "learning_rate": 4.844194930681073e-07,
      "loss": 0.0543,
      "step": 86580
    },
    {
      "epoch": 1.1997222900390625e-05,
      "model_forward_time": 1.7382748126983643,
      "step": 86580
    },
    {
      "dynamics_loss_avg": 0.050490851327776906,
      "epoch": 1.1997222900390625e-05,
      "step": 86580
    },
    {
      "action_loss_avg": 0.0014004208147525786,
      "epoch": 1.1997222900390625e-05,
      "step": 86580
    },
    {
      "epoch": 1.1997222900390625e-05,
      "step": 86580,
      "training_step_time": 7.533313512802124
    },
    {
      "epoch": 1.199817657470703e-05,
      "model_forward_time": 1.839785099029541,
      "step": 86581
    },
    {
      "epoch": 1.199817657470703e-05,
      "step": 86581,
      "training_step_time": 8.308891773223877
    },
    {
      "epoch": 1.1999130249023438e-05,
      "model_forward_time": 2.4564366340637207,
      "step": 86582
    },
    {
      "epoch": 1.1999130249023438e-05,
      "step": 86582,
      "training_step_time": 9.451379299163818
    },
    {
      "epoch": 1.2000083923339844e-05,
      "model_forward_time": 1.7683906555175781,
      "step": 86583
    },
    {
      "epoch": 1.2000083923339844e-05,
      "step": 86583,
      "training_step_time": 8.25858449935913
    },
    {
      "epoch": 1.200103759765625e-05,
      "model_forward_time": 1.8058454990386963,
      "step": 86584
    },
    {
      "epoch": 1.200103759765625e-05,
      "step": 86584,
      "training_step_time": 7.361641883850098
    },
    {
      "epoch": 1.2001991271972657e-05,
      "model_forward_time": 1.777057409286499,
      "step": 86585
    },
    {
      "epoch": 1.2001991271972657e-05,
      "step": 86585,
      "training_step_time": 8.588348865509033
    },
    {
      "epoch": 1.2002944946289063e-05,
      "model_forward_time": 1.7787985801696777,
      "step": 86586
    },
    {
      "epoch": 1.2002944946289063e-05,
      "step": 86586,
      "training_step_time": 6.831088542938232
    },
    {
      "epoch": 1.2003898620605469e-05,
      "model_forward_time": 1.774177074432373,
      "step": 86587
    },
    {
      "epoch": 1.2003898620605469e-05,
      "step": 86587,
      "training_step_time": 6.845062732696533
    },
    {
      "epoch": 1.2004852294921874e-05,
      "model_forward_time": 1.774381160736084,
      "step": 86588
    },
    {
      "epoch": 1.2004852294921874e-05,
      "step": 86588,
      "training_step_time": 6.910362720489502
    },
    {
      "epoch": 1.2005805969238282e-05,
      "model_forward_time": 1.787285327911377,
      "step": 86589
    },
    {
      "epoch": 1.2005805969238282e-05,
      "step": 86589,
      "training_step_time": 6.800266981124878
    },
    {
      "epoch": 1.2006759643554687e-05,
      "grad_norm": 0.10173232853412628,
      "learning_rate": 4.837097459271267e-07,
      "loss": 0.0536,
      "step": 86590
    },
    {
      "epoch": 1.2006759643554687e-05,
      "model_forward_time": 1.7611169815063477,
      "step": 86590
    },
    {
      "dynamics_loss_avg": 0.055044588167220354,
      "epoch": 1.2006759643554687e-05,
      "step": 86590
    },
    {
      "action_loss_avg": 0.0007684504264034331,
      "epoch": 1.2006759643554687e-05,
      "step": 86590
    },
    {
      "epoch": 1.2006759643554687e-05,
      "step": 86590,
      "training_step_time": 6.794752836227417
    },
    {
      "epoch": 1.2007713317871093e-05,
      "model_forward_time": 2.4658701419830322,
      "step": 86591
    },
    {
      "epoch": 1.2007713317871093e-05,
      "step": 86591,
      "training_step_time": 8.930671691894531
    },
    {
      "epoch": 1.20086669921875e-05,
      "model_forward_time": 1.7782683372497559,
      "step": 86592
    },
    {
      "epoch": 1.20086669921875e-05,
      "step": 86592,
      "training_step_time": 8.659752368927002
    },
    {
      "epoch": 1.2009620666503906e-05,
      "model_forward_time": 1.7588660717010498,
      "step": 86593
    },
    {
      "epoch": 1.2009620666503906e-05,
      "step": 86593,
      "training_step_time": 8.310773134231567
    },
    {
      "epoch": 1.2010574340820312e-05,
      "model_forward_time": 1.7762432098388672,
      "step": 86594
    },
    {
      "epoch": 1.2010574340820312e-05,
      "step": 86594,
      "training_step_time": 8.581907510757446
    },
    {
      "epoch": 1.201152801513672e-05,
      "model_forward_time": 1.789741039276123,
      "step": 86595
    },
    {
      "epoch": 1.201152801513672e-05,
      "step": 86595,
      "training_step_time": 7.1023948192596436
    },
    {
      "epoch": 1.2012481689453125e-05,
      "model_forward_time": 1.7751915454864502,
      "step": 86596
    },
    {
      "epoch": 1.2012481689453125e-05,
      "step": 86596,
      "training_step_time": 6.7225542068481445
    },
    {
      "epoch": 1.201343536376953e-05,
      "model_forward_time": 1.7683429718017578,
      "step": 86597
    },
    {
      "epoch": 1.201343536376953e-05,
      "step": 86597,
      "training_step_time": 6.863417148590088
    },
    {
      "epoch": 1.2014389038085938e-05,
      "model_forward_time": 1.7710578441619873,
      "step": 86598
    },
    {
      "epoch": 1.2014389038085938e-05,
      "step": 86598,
      "training_step_time": 6.81833028793335
    },
    {
      "epoch": 1.2015342712402344e-05,
      "model_forward_time": 1.7968356609344482,
      "step": 86599
    },
    {
      "epoch": 1.2015342712402344e-05,
      "step": 86599,
      "training_step_time": 6.805302619934082
    },
    {
      "epoch": 1.201629638671875e-05,
      "grad_norm": 0.1142909973859787,
      "learning_rate": 4.830004926808573e-07,
      "loss": 0.0521,
      "step": 86600
    },
    {
      "epoch": 1.201629638671875e-05,
      "model_forward_time": 2.4756546020507812,
      "step": 86600
    },
    {
      "dynamics_loss_avg": 0.045660842582583426,
      "epoch": 1.201629638671875e-05,
      "step": 86600
    },
    {
      "action_loss_avg": 0.0009608008811483159,
      "epoch": 1.201629638671875e-05,
      "step": 86600
    },
    {
      "epoch": 1.201629638671875e-05,
      "step": 86600,
      "training_step_time": 7.904031753540039
    },
    {
      "epoch": 1.2017250061035157e-05,
      "model_forward_time": 1.7710113525390625,
      "step": 86601
    },
    {
      "epoch": 1.2017250061035157e-05,
      "step": 86601,
      "training_step_time": 8.60646367073059
    },
    {
      "epoch": 1.2018203735351563e-05,
      "model_forward_time": 1.7735347747802734,
      "step": 86602
    },
    {
      "epoch": 1.2018203735351563e-05,
      "step": 86602,
      "training_step_time": 7.74550986289978
    },
    {
      "epoch": 1.2019157409667969e-05,
      "model_forward_time": 1.7752785682678223,
      "step": 86603
    },
    {
      "epoch": 1.2019157409667969e-05,
      "step": 86603,
      "training_step_time": 7.727556467056274
    },
    {
      "epoch": 1.2020111083984374e-05,
      "model_forward_time": 1.7841529846191406,
      "step": 86604
    },
    {
      "epoch": 1.2020111083984374e-05,
      "step": 86604,
      "training_step_time": 7.266834497451782
    },
    {
      "epoch": 1.2021064758300782e-05,
      "model_forward_time": 1.7790415287017822,
      "step": 86605
    },
    {
      "epoch": 1.2021064758300782e-05,
      "step": 86605,
      "training_step_time": 6.7846174240112305
    },
    {
      "epoch": 1.2022018432617187e-05,
      "model_forward_time": 1.7764875888824463,
      "step": 86606
    },
    {
      "epoch": 1.2022018432617187e-05,
      "step": 86606,
      "training_step_time": 6.8041722774505615
    },
    {
      "epoch": 1.2022972106933593e-05,
      "model_forward_time": 1.7727375030517578,
      "step": 86607
    },
    {
      "epoch": 1.2022972106933593e-05,
      "step": 86607,
      "training_step_time": 6.7739012241363525
    },
    {
      "epoch": 1.202392578125e-05,
      "model_forward_time": 1.7697560787200928,
      "step": 86608
    },
    {
      "epoch": 1.202392578125e-05,
      "step": 86608,
      "training_step_time": 6.805149555206299
    },
    {
      "epoch": 1.2024879455566406e-05,
      "model_forward_time": 2.4400417804718018,
      "step": 86609
    },
    {
      "epoch": 1.2024879455566406e-05,
      "step": 86609,
      "training_step_time": 7.899776220321655
    },
    {
      "epoch": 1.2025833129882812e-05,
      "grad_norm": 0.08546363562345505,
      "learning_rate": 4.822917334068627e-07,
      "loss": 0.0534,
      "step": 86610
    },
    {
      "epoch": 1.2025833129882812e-05,
      "model_forward_time": 1.8962233066558838,
      "step": 86610
    },
    {
      "dynamics_loss_avg": 0.07863290272653103,
      "epoch": 1.2025833129882812e-05,
      "step": 86610
    },
    {
      "action_loss_avg": 0.00125595880090259,
      "epoch": 1.2025833129882812e-05,
      "step": 86610
    },
    {
      "epoch": 1.2025833129882812e-05,
      "step": 86610,
      "training_step_time": 11.185684204101562
    },
    {
      "epoch": 1.202678680419922e-05,
      "model_forward_time": 1.7835729122161865,
      "step": 86611
    },
    {
      "epoch": 1.202678680419922e-05,
      "step": 86611,
      "training_step_time": 7.707008123397827
    },
    {
      "epoch": 1.2027740478515625e-05,
      "model_forward_time": 1.7972004413604736,
      "step": 86612
    },
    {
      "epoch": 1.2027740478515625e-05,
      "step": 86612,
      "training_step_time": 8.526950120925903
    },
    {
      "epoch": 1.202869415283203e-05,
      "model_forward_time": 1.7826104164123535,
      "step": 86613
    },
    {
      "epoch": 1.202869415283203e-05,
      "step": 86613,
      "training_step_time": 7.234880685806274
    },
    {
      "epoch": 1.2029647827148438e-05,
      "model_forward_time": 1.7765841484069824,
      "step": 86614
    },
    {
      "epoch": 1.2029647827148438e-05,
      "step": 86614,
      "training_step_time": 6.791741609573364
    },
    {
      "epoch": 1.2030601501464844e-05,
      "model_forward_time": 1.7830376625061035,
      "step": 86615
    },
    {
      "epoch": 1.2030601501464844e-05,
      "step": 86615,
      "training_step_time": 6.907295227050781
    },
    {
      "epoch": 1.203155517578125e-05,
      "model_forward_time": 1.7880971431732178,
      "step": 86616
    },
    {
      "epoch": 1.203155517578125e-05,
      "step": 86616,
      "training_step_time": 6.77484393119812
    },
    {
      "epoch": 1.2032508850097657e-05,
      "model_forward_time": 1.7851710319519043,
      "step": 86617
    },
    {
      "epoch": 1.2032508850097657e-05,
      "step": 86617,
      "training_step_time": 6.816172122955322
    },
    {
      "epoch": 1.2033462524414063e-05,
      "model_forward_time": 2.439185380935669,
      "step": 86618
    },
    {
      "epoch": 1.2033462524414063e-05,
      "step": 86618,
      "training_step_time": 8.803215742111206
    },
    {
      "epoch": 1.2034416198730469e-05,
      "model_forward_time": 1.8316006660461426,
      "step": 86619
    },
    {
      "epoch": 1.2034416198730469e-05,
      "step": 86619,
      "training_step_time": 11.823395729064941
    },
    {
      "epoch": 1.2035369873046874e-05,
      "grad_norm": 0.12319029122591019,
      "learning_rate": 4.81583468182652e-07,
      "loss": 0.0526,
      "step": 86620
    },
    {
      "epoch": 1.2035369873046874e-05,
      "model_forward_time": 1.7642674446105957,
      "step": 86620
    },
    {
      "dynamics_loss_avg": 0.053107965551316735,
      "epoch": 1.2035369873046874e-05,
      "step": 86620
    },
    {
      "action_loss_avg": 0.0011186993564479054,
      "epoch": 1.2035369873046874e-05,
      "step": 86620
    },
    {
      "epoch": 1.2035369873046874e-05,
      "step": 86620,
      "training_step_time": 7.785538911819458
    },
    {
      "epoch": 1.2036323547363282e-05,
      "model_forward_time": 1.7898976802825928,
      "step": 86621
    },
    {
      "epoch": 1.2036323547363282e-05,
      "step": 86621,
      "training_step_time": 7.6749067306518555
    },
    {
      "epoch": 1.2037277221679687e-05,
      "model_forward_time": 1.7685585021972656,
      "step": 86622
    },
    {
      "epoch": 1.2037277221679687e-05,
      "step": 86622,
      "training_step_time": 7.837701320648193
    },
    {
      "epoch": 1.2038230895996093e-05,
      "model_forward_time": 1.7767064571380615,
      "step": 86623
    },
    {
      "epoch": 1.2038230895996093e-05,
      "step": 86623,
      "training_step_time": 6.847205877304077
    },
    {
      "epoch": 1.20391845703125e-05,
      "model_forward_time": 1.779193639755249,
      "step": 86624
    },
    {
      "epoch": 1.20391845703125e-05,
      "step": 86624,
      "training_step_time": 6.829942226409912
    },
    {
      "epoch": 1.2040138244628906e-05,
      "model_forward_time": 1.78153395652771,
      "step": 86625
    },
    {
      "epoch": 1.2040138244628906e-05,
      "step": 86625,
      "training_step_time": 6.826514720916748
    },
    {
      "epoch": 1.2041091918945312e-05,
      "model_forward_time": 1.758436918258667,
      "step": 86626
    },
    {
      "epoch": 1.2041091918945312e-05,
      "step": 86626,
      "training_step_time": 6.811553001403809
    },
    {
      "epoch": 1.204204559326172e-05,
      "model_forward_time": 2.3643481731414795,
      "step": 86627
    },
    {
      "epoch": 1.204204559326172e-05,
      "step": 86627,
      "training_step_time": 7.694907188415527
    },
    {
      "epoch": 1.2042999267578125e-05,
      "model_forward_time": 1.7658641338348389,
      "step": 86628
    },
    {
      "epoch": 1.2042999267578125e-05,
      "step": 86628,
      "training_step_time": 7.846972465515137
    },
    {
      "epoch": 1.204395294189453e-05,
      "model_forward_time": 1.753467321395874,
      "step": 86629
    },
    {
      "epoch": 1.204395294189453e-05,
      "step": 86629,
      "training_step_time": 7.828708648681641
    },
    {
      "epoch": 1.2044906616210938e-05,
      "grad_norm": 0.08059408515691757,
      "learning_rate": 4.808756970856809e-07,
      "loss": 0.0524,
      "step": 86630
    },
    {
      "epoch": 1.2044906616210938e-05,
      "model_forward_time": 1.7548351287841797,
      "step": 86630
    },
    {
      "dynamics_loss_avg": 0.06265333518385888,
      "epoch": 1.2044906616210938e-05,
      "step": 86630
    },
    {
      "action_loss_avg": 0.0020435131969861685,
      "epoch": 1.2044906616210938e-05,
      "step": 86630
    },
    {
      "epoch": 1.2044906616210938e-05,
      "step": 86630,
      "training_step_time": 7.652083158493042
    },
    {
      "epoch": 1.2045860290527344e-05,
      "model_forward_time": 1.7517294883728027,
      "step": 86631
    },
    {
      "epoch": 1.2045860290527344e-05,
      "step": 86631,
      "training_step_time": 7.798661947250366
    },
    {
      "epoch": 1.204681396484375e-05,
      "model_forward_time": 1.7553057670593262,
      "step": 86632
    },
    {
      "epoch": 1.204681396484375e-05,
      "step": 86632,
      "training_step_time": 6.784695148468018
    },
    {
      "epoch": 1.2047767639160155e-05,
      "model_forward_time": 1.774864912033081,
      "step": 86633
    },
    {
      "epoch": 1.2047767639160155e-05,
      "step": 86633,
      "training_step_time": 6.814542293548584
    },
    {
      "epoch": 1.2048721313476563e-05,
      "model_forward_time": 1.7727282047271729,
      "step": 86634
    },
    {
      "epoch": 1.2048721313476563e-05,
      "step": 86634,
      "training_step_time": 6.804861307144165
    },
    {
      "epoch": 1.2049674987792969e-05,
      "model_forward_time": 1.7893710136413574,
      "step": 86635
    },
    {
      "epoch": 1.2049674987792969e-05,
      "step": 86635,
      "training_step_time": 6.791755437850952
    },
    {
      "epoch": 1.2050628662109374e-05,
      "model_forward_time": 2.385267972946167,
      "step": 86636
    },
    {
      "epoch": 1.2050628662109374e-05,
      "step": 86636,
      "training_step_time": 7.741574287414551
    },
    {
      "epoch": 1.2051582336425782e-05,
      "model_forward_time": 1.7730324268341064,
      "step": 86637
    },
    {
      "epoch": 1.2051582336425782e-05,
      "step": 86637,
      "training_step_time": 9.355015516281128
    },
    {
      "epoch": 1.2052536010742187e-05,
      "model_forward_time": 1.8960018157958984,
      "step": 86638
    },
    {
      "epoch": 1.2052536010742187e-05,
      "step": 86638,
      "training_step_time": 7.790922164916992
    },
    {
      "epoch": 1.2053489685058593e-05,
      "model_forward_time": 1.7787835597991943,
      "step": 86639
    },
    {
      "epoch": 1.2053489685058593e-05,
      "step": 86639,
      "training_step_time": 7.637378454208374
    },
    {
      "epoch": 1.2054443359375e-05,
      "grad_norm": 0.08393678069114685,
      "learning_rate": 4.801684201933476e-07,
      "loss": 0.0525,
      "step": 86640
    },
    {
      "epoch": 1.2054443359375e-05,
      "model_forward_time": 1.776606798171997,
      "step": 86640
    },
    {
      "dynamics_loss_avg": 0.03457672791555524,
      "epoch": 1.2054443359375e-05,
      "step": 86640
    },
    {
      "action_loss_avg": 0.0008642243570648134,
      "epoch": 1.2054443359375e-05,
      "step": 86640
    },
    {
      "epoch": 1.2054443359375e-05,
      "step": 86640,
      "training_step_time": 8.31329607963562
    },
    {
      "epoch": 1.2055397033691406e-05,
      "model_forward_time": 1.7872326374053955,
      "step": 86641
    },
    {
      "epoch": 1.2055397033691406e-05,
      "step": 86641,
      "training_step_time": 6.777625322341919
    },
    {
      "epoch": 1.2056350708007812e-05,
      "model_forward_time": 1.786484956741333,
      "step": 86642
    },
    {
      "epoch": 1.2056350708007812e-05,
      "step": 86642,
      "training_step_time": 6.7965168952941895
    },
    {
      "epoch": 1.205730438232422e-05,
      "model_forward_time": 1.7975873947143555,
      "step": 86643
    },
    {
      "epoch": 1.205730438232422e-05,
      "step": 86643,
      "training_step_time": 6.7874743938446045
    },
    {
      "epoch": 1.2058258056640625e-05,
      "model_forward_time": 1.7813866138458252,
      "step": 86644
    },
    {
      "epoch": 1.2058258056640625e-05,
      "step": 86644,
      "training_step_time": 6.7861573696136475
    },
    {
      "epoch": 1.205921173095703e-05,
      "model_forward_time": 1.7769534587860107,
      "step": 86645
    },
    {
      "epoch": 1.205921173095703e-05,
      "step": 86645,
      "training_step_time": 7.71151328086853
    },
    {
      "epoch": 1.2060165405273438e-05,
      "model_forward_time": 2.4045286178588867,
      "step": 86646
    },
    {
      "epoch": 1.2060165405273438e-05,
      "step": 86646,
      "training_step_time": 8.72734522819519
    },
    {
      "epoch": 1.2061119079589844e-05,
      "model_forward_time": 1.8318257331848145,
      "step": 86647
    },
    {
      "epoch": 1.2061119079589844e-05,
      "step": 86647,
      "training_step_time": 7.818111181259155
    },
    {
      "epoch": 1.206207275390625e-05,
      "model_forward_time": 1.755434274673462,
      "step": 86648
    },
    {
      "epoch": 1.206207275390625e-05,
      "step": 86648,
      "training_step_time": 8.645240068435669
    },
    {
      "epoch": 1.2063026428222655e-05,
      "model_forward_time": 1.7637920379638672,
      "step": 86649
    },
    {
      "epoch": 1.2063026428222655e-05,
      "step": 86649,
      "training_step_time": 8.306398630142212
    },
    {
      "epoch": 1.2063980102539063e-05,
      "grad_norm": 0.10705030709505081,
      "learning_rate": 4.794616375830019e-07,
      "loss": 0.0539,
      "step": 86650
    },
    {
      "epoch": 1.2063980102539063e-05,
      "model_forward_time": 1.7483439445495605,
      "step": 86650
    },
    {
      "dynamics_loss_avg": 0.06091763861477375,
      "epoch": 1.2063980102539063e-05,
      "step": 86650
    },
    {
      "action_loss_avg": 0.0014435443881666288,
      "epoch": 1.2063980102539063e-05,
      "step": 86650
    },
    {
      "epoch": 1.2063980102539063e-05,
      "step": 86650,
      "training_step_time": 7.303914308547974
    },
    {
      "epoch": 1.2064933776855469e-05,
      "model_forward_time": 1.758396863937378,
      "step": 86651
    },
    {
      "epoch": 1.2064933776855469e-05,
      "step": 86651,
      "training_step_time": 6.774914741516113
    },
    {
      "epoch": 1.2065887451171874e-05,
      "model_forward_time": 1.7599225044250488,
      "step": 86652
    },
    {
      "epoch": 1.2065887451171874e-05,
      "step": 86652,
      "training_step_time": 6.723860502243042
    },
    {
      "epoch": 1.2066841125488282e-05,
      "model_forward_time": 1.7519912719726562,
      "step": 86653
    },
    {
      "epoch": 1.2066841125488282e-05,
      "step": 86653,
      "training_step_time": 6.772217750549316
    },
    {
      "epoch": 1.2067794799804687e-05,
      "model_forward_time": 1.7548675537109375,
      "step": 86654
    },
    {
      "epoch": 1.2067794799804687e-05,
      "step": 86654,
      "training_step_time": 7.652319669723511
    },
    {
      "epoch": 1.2068748474121093e-05,
      "model_forward_time": 2.406902551651001,
      "step": 86655
    },
    {
      "epoch": 1.2068748474121093e-05,
      "step": 86655,
      "training_step_time": 7.709611177444458
    },
    {
      "epoch": 1.20697021484375e-05,
      "model_forward_time": 1.8474152088165283,
      "step": 86656
    },
    {
      "epoch": 1.20697021484375e-05,
      "step": 86656,
      "training_step_time": 8.314650535583496
    },
    {
      "epoch": 1.2070655822753906e-05,
      "model_forward_time": 1.779918909072876,
      "step": 86657
    },
    {
      "epoch": 1.2070655822753906e-05,
      "step": 86657,
      "training_step_time": 8.807857990264893
    },
    {
      "epoch": 1.2071609497070312e-05,
      "model_forward_time": 1.748349666595459,
      "step": 86658
    },
    {
      "epoch": 1.2071609497070312e-05,
      "step": 86658,
      "training_step_time": 8.19694185256958
    },
    {
      "epoch": 1.207256317138672e-05,
      "model_forward_time": 1.754023790359497,
      "step": 86659
    },
    {
      "epoch": 1.207256317138672e-05,
      "step": 86659,
      "training_step_time": 7.790038824081421
    },
    {
      "epoch": 1.2073516845703125e-05,
      "grad_norm": 0.1718498021364212,
      "learning_rate": 4.787553493319341e-07,
      "loss": 0.0537,
      "step": 86660
    },
    {
      "epoch": 1.2073516845703125e-05,
      "model_forward_time": 1.755338430404663,
      "step": 86660
    },
    {
      "dynamics_loss_avg": 0.04965944699943066,
      "epoch": 1.2073516845703125e-05,
      "step": 86660
    },
    {
      "action_loss_avg": 0.001182468023034744,
      "epoch": 1.2073516845703125e-05,
      "step": 86660
    },
    {
      "epoch": 1.2073516845703125e-05,
      "step": 86660,
      "training_step_time": 6.811872482299805
    },
    {
      "epoch": 1.2074470520019531e-05,
      "model_forward_time": 1.759347915649414,
      "step": 86661
    },
    {
      "epoch": 1.2074470520019531e-05,
      "step": 86661,
      "training_step_time": 6.788858413696289
    },
    {
      "epoch": 1.2075424194335938e-05,
      "model_forward_time": 1.7573034763336182,
      "step": 86662
    },
    {
      "epoch": 1.2075424194335938e-05,
      "step": 86662,
      "training_step_time": 6.802971601486206
    },
    {
      "epoch": 1.2076377868652344e-05,
      "model_forward_time": 1.7461705207824707,
      "step": 86663
    },
    {
      "epoch": 1.2076377868652344e-05,
      "step": 86663,
      "training_step_time": 6.786036491394043
    },
    {
      "epoch": 1.207733154296875e-05,
      "model_forward_time": 2.392972707748413,
      "step": 86664
    },
    {
      "epoch": 1.207733154296875e-05,
      "step": 86664,
      "training_step_time": 7.700189590454102
    },
    {
      "epoch": 1.2078285217285155e-05,
      "model_forward_time": 1.8101179599761963,
      "step": 86665
    },
    {
      "epoch": 1.2078285217285155e-05,
      "step": 86665,
      "training_step_time": 9.703308820724487
    },
    {
      "epoch": 1.2079238891601563e-05,
      "model_forward_time": 1.7735834121704102,
      "step": 86666
    },
    {
      "epoch": 1.2079238891601563e-05,
      "step": 86666,
      "training_step_time": 7.821915864944458
    },
    {
      "epoch": 1.2080192565917969e-05,
      "model_forward_time": 1.380573034286499,
      "step": 86667
    },
    {
      "epoch": 1.2080192565917969e-05,
      "step": 86667,
      "training_step_time": 7.739104270935059
    },
    {
      "epoch": 1.2081146240234374e-05,
      "model_forward_time": 1.780078649520874,
      "step": 86668
    },
    {
      "epoch": 1.2081146240234374e-05,
      "step": 86668,
      "training_step_time": 7.8462300300598145
    },
    {
      "epoch": 1.2082099914550782e-05,
      "model_forward_time": 1.778630256652832,
      "step": 86669
    },
    {
      "epoch": 1.2082099914550782e-05,
      "step": 86669,
      "training_step_time": 7.185498476028442
    },
    {
      "epoch": 1.2083053588867187e-05,
      "grad_norm": 0.0701252892613411,
      "learning_rate": 4.780495555173836e-07,
      "loss": 0.055,
      "step": 86670
    },
    {
      "epoch": 1.2083053588867187e-05,
      "model_forward_time": 1.797926902770996,
      "step": 86670
    },
    {
      "dynamics_loss_avg": 0.04850603714585304,
      "epoch": 1.2083053588867187e-05,
      "step": 86670
    },
    {
      "action_loss_avg": 0.0009160897956462577,
      "epoch": 1.2083053588867187e-05,
      "step": 86670
    },
    {
      "epoch": 1.2083053588867187e-05,
      "step": 86670,
      "training_step_time": 6.723026752471924
    },
    {
      "epoch": 1.2084007263183593e-05,
      "model_forward_time": 1.761664628982544,
      "step": 86671
    },
    {
      "epoch": 1.2084007263183593e-05,
      "step": 86671,
      "training_step_time": 6.814397573471069
    },
    {
      "epoch": 1.20849609375e-05,
      "model_forward_time": 1.7739791870117188,
      "step": 86672
    },
    {
      "epoch": 1.20849609375e-05,
      "step": 86672,
      "training_step_time": 6.844366073608398
    },
    {
      "epoch": 1.2085914611816406e-05,
      "model_forward_time": 2.4070355892181396,
      "step": 86673
    },
    {
      "epoch": 1.2085914611816406e-05,
      "step": 86673,
      "training_step_time": 8.794756412506104
    },
    {
      "epoch": 1.2086868286132812e-05,
      "model_forward_time": 1.781648874282837,
      "step": 86674
    },
    {
      "epoch": 1.2086868286132812e-05,
      "step": 86674,
      "training_step_time": 8.901957988739014
    },
    {
      "epoch": 1.208782196044922e-05,
      "model_forward_time": 1.7703287601470947,
      "step": 86675
    },
    {
      "epoch": 1.208782196044922e-05,
      "step": 86675,
      "training_step_time": 8.524939060211182
    },
    {
      "epoch": 1.2088775634765625e-05,
      "model_forward_time": 1.780461072921753,
      "step": 86676
    },
    {
      "epoch": 1.2088775634765625e-05,
      "step": 86676,
      "training_step_time": 7.672202110290527
    },
    {
      "epoch": 1.2089729309082031e-05,
      "model_forward_time": 1.7727458477020264,
      "step": 86677
    },
    {
      "epoch": 1.2089729309082031e-05,
      "step": 86677,
      "training_step_time": 7.83266282081604
    },
    {
      "epoch": 1.2090682983398438e-05,
      "model_forward_time": 1.7631068229675293,
      "step": 86678
    },
    {
      "epoch": 1.2090682983398438e-05,
      "step": 86678,
      "training_step_time": 7.204802751541138
    },
    {
      "epoch": 1.2091636657714844e-05,
      "model_forward_time": 1.791067123413086,
      "step": 86679
    },
    {
      "epoch": 1.2091636657714844e-05,
      "step": 86679,
      "training_step_time": 6.961384296417236
    },
    {
      "epoch": 1.209259033203125e-05,
      "grad_norm": 0.09247579425573349,
      "learning_rate": 4.773442562165348e-07,
      "loss": 0.0553,
      "step": 86680
    },
    {
      "epoch": 1.209259033203125e-05,
      "model_forward_time": 1.7682723999023438,
      "step": 86680
    },
    {
      "dynamics_loss_avg": 0.05731053967028856,
      "epoch": 1.209259033203125e-05,
      "step": 86680
    },
    {
      "action_loss_avg": 0.0013452309125568719,
      "epoch": 1.209259033203125e-05,
      "step": 86680
    },
    {
      "epoch": 1.209259033203125e-05,
      "step": 86680,
      "training_step_time": 6.821025609970093
    },
    {
      "epoch": 1.2093544006347655e-05,
      "model_forward_time": 1.7824184894561768,
      "step": 86681
    },
    {
      "epoch": 1.2093544006347655e-05,
      "step": 86681,
      "training_step_time": 6.823429107666016
    },
    {
      "epoch": 1.2094497680664063e-05,
      "model_forward_time": 1.778397798538208,
      "step": 86682
    },
    {
      "epoch": 1.2094497680664063e-05,
      "step": 86682,
      "training_step_time": 8.335175037384033
    },
    {
      "epoch": 1.2095451354980469e-05,
      "model_forward_time": 1.3908360004425049,
      "step": 86683
    },
    {
      "epoch": 1.2095451354980469e-05,
      "step": 86683,
      "training_step_time": 8.925163745880127
    },
    {
      "epoch": 1.2096405029296874e-05,
      "model_forward_time": 1.8883657455444336,
      "step": 86684
    },
    {
      "epoch": 1.2096405029296874e-05,
      "step": 86684,
      "training_step_time": 9.734315633773804
    },
    {
      "epoch": 1.2097358703613282e-05,
      "model_forward_time": 1.7670211791992188,
      "step": 86685
    },
    {
      "epoch": 1.2097358703613282e-05,
      "step": 86685,
      "training_step_time": 7.734125852584839
    },
    {
      "epoch": 1.2098312377929687e-05,
      "model_forward_time": 1.7822887897491455,
      "step": 86686
    },
    {
      "epoch": 1.2098312377929687e-05,
      "step": 86686,
      "training_step_time": 8.240024089813232
    },
    {
      "epoch": 1.2099266052246093e-05,
      "model_forward_time": 1.7789716720581055,
      "step": 86687
    },
    {
      "epoch": 1.2099266052246093e-05,
      "step": 86687,
      "training_step_time": 7.256967306137085
    },
    {
      "epoch": 1.21002197265625e-05,
      "model_forward_time": 1.7770717144012451,
      "step": 86688
    },
    {
      "epoch": 1.21002197265625e-05,
      "step": 86688,
      "training_step_time": 6.857385635375977
    },
    {
      "epoch": 1.2101173400878906e-05,
      "model_forward_time": 1.7709128856658936,
      "step": 86689
    },
    {
      "epoch": 1.2101173400878906e-05,
      "step": 86689,
      "training_step_time": 6.996448993682861
    },
    {
      "epoch": 1.2102127075195312e-05,
      "grad_norm": 0.0890517607331276,
      "learning_rate": 4.766394515065198e-07,
      "loss": 0.0542,
      "step": 86690
    },
    {
      "epoch": 1.2102127075195312e-05,
      "model_forward_time": 1.749405860900879,
      "step": 86690
    },
    {
      "dynamics_loss_avg": 0.04082452617585659,
      "epoch": 1.2102127075195312e-05,
      "step": 86690
    },
    {
      "action_loss_avg": 0.0009484239213634283,
      "epoch": 1.2102127075195312e-05,
      "step": 86690
    },
    {
      "epoch": 1.2102127075195312e-05,
      "step": 86690,
      "training_step_time": 6.866403818130493
    },
    {
      "epoch": 1.210308074951172e-05,
      "model_forward_time": 1.7729847431182861,
      "step": 86691
    },
    {
      "epoch": 1.210308074951172e-05,
      "step": 86691,
      "training_step_time": 7.704615116119385
    },
    {
      "epoch": 1.2104034423828125e-05,
      "model_forward_time": 2.3954837322235107,
      "step": 86692
    },
    {
      "epoch": 1.2104034423828125e-05,
      "step": 86692,
      "training_step_time": 8.46408987045288
    },
    {
      "epoch": 1.2104988098144531e-05,
      "model_forward_time": 1.4362008571624756,
      "step": 86693
    },
    {
      "epoch": 1.2104988098144531e-05,
      "step": 86693,
      "training_step_time": 10.684032201766968
    },
    {
      "epoch": 1.2105941772460938e-05,
      "model_forward_time": 1.3713860511779785,
      "step": 86694
    },
    {
      "epoch": 1.2105941772460938e-05,
      "step": 86694,
      "training_step_time": 8.220493793487549
    },
    {
      "epoch": 1.2106895446777344e-05,
      "model_forward_time": 1.748783826828003,
      "step": 86695
    },
    {
      "epoch": 1.2106895446777344e-05,
      "step": 86695,
      "training_step_time": 7.6345319747924805
    },
    {
      "epoch": 1.210784912109375e-05,
      "model_forward_time": 1.3836979866027832,
      "step": 86696
    },
    {
      "epoch": 1.210784912109375e-05,
      "step": 86696,
      "training_step_time": 7.694255352020264
    },
    {
      "epoch": 1.2108802795410155e-05,
      "model_forward_time": 1.377762794494629,
      "step": 86697
    },
    {
      "epoch": 1.2108802795410155e-05,
      "step": 86697,
      "training_step_time": 7.155430793762207
    },
    {
      "epoch": 1.2109756469726563e-05,
      "model_forward_time": 1.7602763175964355,
      "step": 86698
    },
    {
      "epoch": 1.2109756469726563e-05,
      "step": 86698,
      "training_step_time": 6.76958155632019
    },
    {
      "epoch": 1.2110710144042969e-05,
      "model_forward_time": 1.7516698837280273,
      "step": 86699
    },
    {
      "epoch": 1.2110710144042969e-05,
      "step": 86699,
      "training_step_time": 6.740605592727661
    },
    {
      "epoch": 1.2111663818359374e-05,
      "grad_norm": 0.10591622442007065,
      "learning_rate": 4.75935141464412e-07,
      "loss": 0.0573,
      "step": 86700
    },
    {
      "epoch": 1.2111663818359374e-05,
      "model_forward_time": 1.7476706504821777,
      "step": 86700
    },
    {
      "dynamics_loss_avg": 0.05490069463849068,
      "epoch": 1.2111663818359374e-05,
      "step": 86700
    },
    {
      "action_loss_avg": 0.0014969307143473999,
      "epoch": 1.2111663818359374e-05,
      "step": 86700
    },
    {
      "epoch": 1.2111663818359374e-05,
      "step": 86700,
      "training_step_time": 7.406758785247803
    },
    {
      "epoch": 1.2112617492675782e-05,
      "model_forward_time": 2.402869939804077,
      "step": 86701
    },
    {
      "epoch": 1.2112617492675782e-05,
      "step": 86701,
      "training_step_time": 9.738882780075073
    },
    {
      "epoch": 1.2113571166992187e-05,
      "model_forward_time": 1.8151123523712158,
      "step": 86702
    },
    {
      "epoch": 1.2113571166992187e-05,
      "step": 86702,
      "training_step_time": 10.26951551437378
    },
    {
      "epoch": 1.2114524841308593e-05,
      "model_forward_time": 1.7557213306427002,
      "step": 86703
    },
    {
      "epoch": 1.2114524841308593e-05,
      "step": 86703,
      "training_step_time": 9.574584007263184
    },
    {
      "epoch": 1.2115478515625e-05,
      "model_forward_time": 1.7657244205474854,
      "step": 86704
    },
    {
      "epoch": 1.2115478515625e-05,
      "step": 86704,
      "training_step_time": 7.7467265129089355
    },
    {
      "epoch": 1.2116432189941406e-05,
      "model_forward_time": 1.7498359680175781,
      "step": 86705
    },
    {
      "epoch": 1.2116432189941406e-05,
      "step": 86705,
      "training_step_time": 7.881159543991089
    },
    {
      "epoch": 1.2117385864257812e-05,
      "model_forward_time": 1.7518346309661865,
      "step": 86706
    },
    {
      "epoch": 1.2117385864257812e-05,
      "step": 86706,
      "training_step_time": 7.239154100418091
    },
    {
      "epoch": 1.211833953857422e-05,
      "model_forward_time": 1.7575654983520508,
      "step": 86707
    },
    {
      "epoch": 1.211833953857422e-05,
      "step": 86707,
      "training_step_time": 6.920393466949463
    },
    {
      "epoch": 1.2119293212890625e-05,
      "model_forward_time": 1.7578353881835938,
      "step": 86708
    },
    {
      "epoch": 1.2119293212890625e-05,
      "step": 86708,
      "training_step_time": 6.8773417472839355
    },
    {
      "epoch": 1.2120246887207031e-05,
      "model_forward_time": 1.7593588829040527,
      "step": 86709
    },
    {
      "epoch": 1.2120246887207031e-05,
      "step": 86709,
      "training_step_time": 7.624090194702148
    },
    {
      "epoch": 1.2121200561523438e-05,
      "grad_norm": 0.08383476734161377,
      "learning_rate": 4.75231326167237e-07,
      "loss": 0.0551,
      "step": 86710
    },
    {
      "epoch": 1.2121200561523438e-05,
      "model_forward_time": 1.7463033199310303,
      "step": 86710
    },
    {
      "dynamics_loss_avg": 0.04197352733463049,
      "epoch": 1.2121200561523438e-05,
      "step": 86710
    },
    {
      "action_loss_avg": 0.000930187405901961,
      "epoch": 1.2121200561523438e-05,
      "step": 86710
    },
    {
      "epoch": 1.2121200561523438e-05,
      "step": 86710,
      "training_step_time": 9.170422077178955
    },
    {
      "epoch": 1.2122154235839844e-05,
      "model_forward_time": 1.7860727310180664,
      "step": 86711
    },
    {
      "epoch": 1.2122154235839844e-05,
      "step": 86711,
      "training_step_time": 10.142898559570312
    },
    {
      "epoch": 1.212310791015625e-05,
      "model_forward_time": 1.759873628616333,
      "step": 86712
    },
    {
      "epoch": 1.212310791015625e-05,
      "step": 86712,
      "training_step_time": 10.331228733062744
    },
    {
      "epoch": 1.2124061584472655e-05,
      "model_forward_time": 1.7437894344329834,
      "step": 86713
    },
    {
      "epoch": 1.2124061584472655e-05,
      "step": 86713,
      "training_step_time": 8.627419233322144
    },
    {
      "epoch": 1.2125015258789063e-05,
      "model_forward_time": 1.7510437965393066,
      "step": 86714
    },
    {
      "epoch": 1.2125015258789063e-05,
      "step": 86714,
      "training_step_time": 7.881359338760376
    },
    {
      "epoch": 1.2125968933105469e-05,
      "model_forward_time": 1.7580904960632324,
      "step": 86715
    },
    {
      "epoch": 1.2125968933105469e-05,
      "step": 86715,
      "training_step_time": 7.265087366104126
    },
    {
      "epoch": 1.2126922607421874e-05,
      "model_forward_time": 1.7480950355529785,
      "step": 86716
    },
    {
      "epoch": 1.2126922607421874e-05,
      "step": 86716,
      "training_step_time": 6.905347585678101
    },
    {
      "epoch": 1.2127876281738282e-05,
      "model_forward_time": 1.7579073905944824,
      "step": 86717
    },
    {
      "epoch": 1.2127876281738282e-05,
      "step": 86717,
      "training_step_time": 6.90477442741394
    },
    {
      "epoch": 1.2128829956054687e-05,
      "model_forward_time": 1.7668538093566895,
      "step": 86718
    },
    {
      "epoch": 1.2128829956054687e-05,
      "step": 86718,
      "training_step_time": 7.617184400558472
    },
    {
      "epoch": 1.2129783630371093e-05,
      "model_forward_time": 1.76204252243042,
      "step": 86719
    },
    {
      "epoch": 1.2129783630371093e-05,
      "step": 86719,
      "training_step_time": 8.515598058700562
    },
    {
      "epoch": 1.21307373046875e-05,
      "grad_norm": 0.09881909936666489,
      "learning_rate": 4.7452800569195987e-07,
      "loss": 0.0549,
      "step": 86720
    },
    {
      "epoch": 1.21307373046875e-05,
      "model_forward_time": 2.4198663234710693,
      "step": 86720
    },
    {
      "dynamics_loss_avg": 0.05216264259070158,
      "epoch": 1.21307373046875e-05,
      "step": 86720
    },
    {
      "action_loss_avg": 0.0010892014426644892,
      "epoch": 1.21307373046875e-05,
      "step": 86720
    },
    {
      "epoch": 1.21307373046875e-05,
      "step": 86720,
      "training_step_time": 8.91715693473816
    },
    {
      "epoch": 1.2131690979003906e-05,
      "model_forward_time": 1.754103660583496,
      "step": 86721
    },
    {
      "epoch": 1.2131690979003906e-05,
      "step": 86721,
      "training_step_time": 9.316557168960571
    },
    {
      "epoch": 1.2132644653320312e-05,
      "model_forward_time": 1.7685389518737793,
      "step": 86722
    },
    {
      "epoch": 1.2132644653320312e-05,
      "step": 86722,
      "training_step_time": 7.778595209121704
    },
    {
      "epoch": 1.213359832763672e-05,
      "model_forward_time": 1.7508225440979004,
      "step": 86723
    },
    {
      "epoch": 1.213359832763672e-05,
      "step": 86723,
      "training_step_time": 7.423357725143433
    },
    {
      "epoch": 1.2134552001953125e-05,
      "model_forward_time": 1.7497241497039795,
      "step": 86724
    },
    {
      "epoch": 1.2134552001953125e-05,
      "step": 86724,
      "training_step_time": 7.919131755828857
    },
    {
      "epoch": 1.2135505676269531e-05,
      "model_forward_time": 1.7675645351409912,
      "step": 86725
    },
    {
      "epoch": 1.2135505676269531e-05,
      "step": 86725,
      "training_step_time": 6.927490472793579
    },
    {
      "epoch": 1.2136459350585938e-05,
      "model_forward_time": 1.7773306369781494,
      "step": 86726
    },
    {
      "epoch": 1.2136459350585938e-05,
      "step": 86726,
      "training_step_time": 6.925567150115967
    },
    {
      "epoch": 1.2137413024902344e-05,
      "model_forward_time": 1.7538166046142578,
      "step": 86727
    },
    {
      "epoch": 1.2137413024902344e-05,
      "step": 86727,
      "training_step_time": 7.608956575393677
    },
    {
      "epoch": 1.213836669921875e-05,
      "model_forward_time": 1.7527563571929932,
      "step": 86728
    },
    {
      "epoch": 1.213836669921875e-05,
      "step": 86728,
      "training_step_time": 7.828117609024048
    },
    {
      "epoch": 1.2139320373535155e-05,
      "model_forward_time": 2.4180939197540283,
      "step": 86729
    },
    {
      "epoch": 1.2139320373535155e-05,
      "step": 86729,
      "training_step_time": 9.999594688415527
    },
    {
      "epoch": 1.2140274047851563e-05,
      "grad_norm": 0.15019366145133972,
      "learning_rate": 4.738251801154975e-07,
      "loss": 0.0554,
      "step": 86730
    },
    {
      "epoch": 1.2140274047851563e-05,
      "model_forward_time": 1.850442886352539,
      "step": 86730
    },
    {
      "dynamics_loss_avg": 0.053412769176065925,
      "epoch": 1.2140274047851563e-05,
      "step": 86730
    },
    {
      "action_loss_avg": 0.0011714703432517126,
      "epoch": 1.2140274047851563e-05,
      "step": 86730
    },
    {
      "epoch": 1.2140274047851563e-05,
      "step": 86730,
      "training_step_time": 9.431828022003174
    },
    {
      "epoch": 1.2141227722167969e-05,
      "model_forward_time": 1.751234531402588,
      "step": 86731
    },
    {
      "epoch": 1.2141227722167969e-05,
      "step": 86731,
      "training_step_time": 7.817740440368652
    },
    {
      "epoch": 1.2142181396484374e-05,
      "model_forward_time": 1.7428536415100098,
      "step": 86732
    },
    {
      "epoch": 1.2142181396484374e-05,
      "step": 86732,
      "training_step_time": 7.5811028480529785
    },
    {
      "epoch": 1.2143135070800782e-05,
      "model_forward_time": 1.747312307357788,
      "step": 86733
    },
    {
      "epoch": 1.2143135070800782e-05,
      "step": 86733,
      "training_step_time": 8.496716022491455
    },
    {
      "epoch": 1.2144088745117187e-05,
      "model_forward_time": 1.7513539791107178,
      "step": 86734
    },
    {
      "epoch": 1.2144088745117187e-05,
      "step": 86734,
      "training_step_time": 6.9156060218811035
    },
    {
      "epoch": 1.2145042419433593e-05,
      "model_forward_time": 1.7574172019958496,
      "step": 86735
    },
    {
      "epoch": 1.2145042419433593e-05,
      "step": 86735,
      "training_step_time": 6.926634073257446
    },
    {
      "epoch": 1.214599609375e-05,
      "model_forward_time": 1.7482271194458008,
      "step": 86736
    },
    {
      "epoch": 1.214599609375e-05,
      "step": 86736,
      "training_step_time": 7.587721586227417
    },
    {
      "epoch": 1.2146949768066406e-05,
      "model_forward_time": 1.7470731735229492,
      "step": 86737
    },
    {
      "epoch": 1.2146949768066406e-05,
      "step": 86737,
      "training_step_time": 8.83713984489441
    },
    {
      "epoch": 1.2147903442382812e-05,
      "model_forward_time": 2.4120242595672607,
      "step": 86738
    },
    {
      "epoch": 1.2147903442382812e-05,
      "step": 86738,
      "training_step_time": 8.487261772155762
    },
    {
      "epoch": 1.214885711669922e-05,
      "model_forward_time": 1.8195993900299072,
      "step": 86739
    },
    {
      "epoch": 1.214885711669922e-05,
      "step": 86739,
      "training_step_time": 8.866783857345581
    },
    {
      "epoch": 1.2149810791015625e-05,
      "grad_norm": 0.10359812527894974,
      "learning_rate": 4.731228495147061e-07,
      "loss": 0.056,
      "step": 86740
    },
    {
      "epoch": 1.2149810791015625e-05,
      "model_forward_time": 1.7550528049468994,
      "step": 86740
    },
    {
      "dynamics_loss_avg": 0.05784555394202471,
      "epoch": 1.2149810791015625e-05,
      "step": 86740
    },
    {
      "action_loss_avg": 0.0015688658168073744,
      "epoch": 1.2149810791015625e-05,
      "step": 86740
    },
    {
      "epoch": 1.2149810791015625e-05,
      "step": 86740,
      "training_step_time": 7.821319341659546
    },
    {
      "epoch": 1.2150764465332031e-05,
      "model_forward_time": 1.7500364780426025,
      "step": 86741
    },
    {
      "epoch": 1.2150764465332031e-05,
      "step": 86741,
      "training_step_time": 8.389336347579956
    },
    {
      "epoch": 1.2151718139648438e-05,
      "model_forward_time": 1.751323938369751,
      "step": 86742
    },
    {
      "epoch": 1.2151718139648438e-05,
      "step": 86742,
      "training_step_time": 7.906245470046997
    },
    {
      "epoch": 1.2152671813964844e-05,
      "model_forward_time": 1.7501940727233887,
      "step": 86743
    },
    {
      "epoch": 1.2152671813964844e-05,
      "step": 86743,
      "training_step_time": 6.875740051269531
    },
    {
      "epoch": 1.215362548828125e-05,
      "model_forward_time": 1.7513916492462158,
      "step": 86744
    },
    {
      "epoch": 1.215362548828125e-05,
      "step": 86744,
      "training_step_time": 6.948180675506592
    },
    {
      "epoch": 1.2154579162597656e-05,
      "model_forward_time": 1.7497949600219727,
      "step": 86745
    },
    {
      "epoch": 1.2154579162597656e-05,
      "step": 86745,
      "training_step_time": 6.989932060241699
    },
    {
      "epoch": 1.2155532836914063e-05,
      "model_forward_time": 1.7474253177642822,
      "step": 86746
    },
    {
      "epoch": 1.2155532836914063e-05,
      "step": 86746,
      "training_step_time": 8.652218580245972
    },
    {
      "epoch": 1.2156486511230469e-05,
      "model_forward_time": 2.4332494735717773,
      "step": 86747
    },
    {
      "epoch": 1.2156486511230469e-05,
      "step": 86747,
      "training_step_time": 7.889200687408447
    },
    {
      "epoch": 1.2157440185546874e-05,
      "model_forward_time": 1.8059437274932861,
      "step": 86748
    },
    {
      "epoch": 1.2157440185546874e-05,
      "step": 86748,
      "training_step_time": 9.453631401062012
    },
    {
      "epoch": 1.2158393859863282e-05,
      "model_forward_time": 1.749527931213379,
      "step": 86749
    },
    {
      "epoch": 1.2158393859863282e-05,
      "step": 86749,
      "training_step_time": 8.824095487594604
    },
    {
      "epoch": 1.2159347534179688e-05,
      "grad_norm": 0.16695274412631989,
      "learning_rate": 4.7242101396639594e-07,
      "loss": 0.0544,
      "step": 86750
    },
    {
      "epoch": 1.2159347534179688e-05,
      "model_forward_time": 1.9014787673950195,
      "step": 86750
    },
    {
      "dynamics_loss_avg": 0.07127459496259689,
      "epoch": 1.2159347534179688e-05,
      "step": 86750
    },
    {
      "action_loss_avg": 0.0018275017326232045,
      "epoch": 1.2159347534179688e-05,
      "step": 86750
    },
    {
      "epoch": 1.2159347534179688e-05,
      "step": 86750,
      "training_step_time": 7.906822681427002
    },
    {
      "epoch": 1.2160301208496093e-05,
      "model_forward_time": 1.8947975635528564,
      "step": 86751
    },
    {
      "epoch": 1.2160301208496093e-05,
      "step": 86751,
      "training_step_time": 7.969620227813721
    },
    {
      "epoch": 1.21612548828125e-05,
      "model_forward_time": 1.8921189308166504,
      "step": 86752
    },
    {
      "epoch": 1.21612548828125e-05,
      "step": 86752,
      "training_step_time": 7.304016590118408
    },
    {
      "epoch": 1.2162208557128906e-05,
      "model_forward_time": 1.9157428741455078,
      "step": 86753
    },
    {
      "epoch": 1.2162208557128906e-05,
      "step": 86753,
      "training_step_time": 6.987385511398315
    },
    {
      "epoch": 1.2163162231445312e-05,
      "model_forward_time": 1.9003567695617676,
      "step": 86754
    },
    {
      "epoch": 1.2163162231445312e-05,
      "step": 86754,
      "training_step_time": 6.992040395736694
    },
    {
      "epoch": 1.216411590576172e-05,
      "model_forward_time": 1.9014256000518799,
      "step": 86755
    },
    {
      "epoch": 1.216411590576172e-05,
      "step": 86755,
      "training_step_time": 7.648485898971558
    },
    {
      "epoch": 1.2165069580078125e-05,
      "model_forward_time": 2.492466688156128,
      "step": 86756
    },
    {
      "epoch": 1.2165069580078125e-05,
      "step": 86756,
      "training_step_time": 9.753751516342163
    },
    {
      "epoch": 1.2166023254394531e-05,
      "model_forward_time": 1.533118724822998,
      "step": 86757
    },
    {
      "epoch": 1.2166023254394531e-05,
      "step": 86757,
      "training_step_time": 10.89360761642456
    },
    {
      "epoch": 1.2166976928710937e-05,
      "model_forward_time": 1.8865954875946045,
      "step": 86758
    },
    {
      "epoch": 1.2166976928710937e-05,
      "step": 86758,
      "training_step_time": 12.140825748443604
    },
    {
      "epoch": 1.2167930603027344e-05,
      "model_forward_time": 1.9196689128875732,
      "step": 86759
    },
    {
      "epoch": 1.2167930603027344e-05,
      "step": 86759,
      "training_step_time": 8.927164554595947
    },
    {
      "epoch": 1.216888427734375e-05,
      "grad_norm": 0.08820735663175583,
      "learning_rate": 4.7171967354731507e-07,
      "loss": 0.0545,
      "step": 86760
    },
    {
      "epoch": 1.216888427734375e-05,
      "model_forward_time": 1.8788704872131348,
      "step": 86760
    },
    {
      "dynamics_loss_avg": 0.04712684769183397,
      "epoch": 1.216888427734375e-05,
      "step": 86760
    },
    {
      "action_loss_avg": 0.0017933958501089365,
      "epoch": 1.216888427734375e-05,
      "step": 86760
    },
    {
      "epoch": 1.216888427734375e-05,
      "step": 86760,
      "training_step_time": 7.612746715545654
    },
    {
      "epoch": 1.2169837951660156e-05,
      "model_forward_time": 1.8957977294921875,
      "step": 86761
    },
    {
      "epoch": 1.2169837951660156e-05,
      "step": 86761,
      "training_step_time": 7.923403263092041
    },
    {
      "epoch": 1.2170791625976563e-05,
      "model_forward_time": 1.9188134670257568,
      "step": 86762
    },
    {
      "epoch": 1.2170791625976563e-05,
      "step": 86762,
      "training_step_time": 6.965195655822754
    },
    {
      "epoch": 1.2171745300292969e-05,
      "model_forward_time": 1.9178874492645264,
      "step": 86763
    },
    {
      "epoch": 1.2171745300292969e-05,
      "step": 86763,
      "training_step_time": 6.992293357849121
    },
    {
      "epoch": 1.2172698974609374e-05,
      "model_forward_time": 1.90102219581604,
      "step": 86764
    },
    {
      "epoch": 1.2172698974609374e-05,
      "step": 86764,
      "training_step_time": 7.597118139266968
    },
    {
      "epoch": 1.2173652648925782e-05,
      "model_forward_time": 1.8988134860992432,
      "step": 86765
    },
    {
      "epoch": 1.2173652648925782e-05,
      "step": 86765,
      "training_step_time": 7.861243724822998
    },
    {
      "epoch": 1.2174606323242188e-05,
      "model_forward_time": 2.493896722793579,
      "step": 86766
    },
    {
      "epoch": 1.2174606323242188e-05,
      "step": 86766,
      "training_step_time": 8.705920457839966
    },
    {
      "epoch": 1.2175559997558593e-05,
      "model_forward_time": 1.8962228298187256,
      "step": 86767
    },
    {
      "epoch": 1.2175559997558593e-05,
      "step": 86767,
      "training_step_time": 9.668812036514282
    },
    {
      "epoch": 1.2176513671875e-05,
      "model_forward_time": 1.9080979824066162,
      "step": 86768
    },
    {
      "epoch": 1.2176513671875e-05,
      "step": 86768,
      "training_step_time": 8.412579774856567
    },
    {
      "epoch": 1.2177467346191406e-05,
      "model_forward_time": 1.9093797206878662,
      "step": 86769
    },
    {
      "epoch": 1.2177467346191406e-05,
      "step": 86769,
      "training_step_time": 7.618193864822388
    },
    {
      "epoch": 1.2178421020507812e-05,
      "grad_norm": 0.06336487084627151,
      "learning_rate": 4.710188283341643e-07,
      "loss": 0.055,
      "step": 86770
    },
    {
      "epoch": 1.2178421020507812e-05,
      "model_forward_time": 1.911825180053711,
      "step": 86770
    },
    {
      "dynamics_loss_avg": 0.06841626903042197,
      "epoch": 1.2178421020507812e-05,
      "step": 86770
    },
    {
      "action_loss_avg": 0.0014366331764904317,
      "epoch": 1.2178421020507812e-05,
      "step": 86770
    },
    {
      "epoch": 1.2178421020507812e-05,
      "step": 86770,
      "training_step_time": 7.877535581588745
    },
    {
      "epoch": 1.217937469482422e-05,
      "model_forward_time": 1.9050025939941406,
      "step": 86771
    },
    {
      "epoch": 1.217937469482422e-05,
      "step": 86771,
      "training_step_time": 7.3032684326171875
    },
    {
      "epoch": 1.2180328369140625e-05,
      "model_forward_time": 1.9111247062683105,
      "step": 86772
    },
    {
      "epoch": 1.2180328369140625e-05,
      "step": 86772,
      "training_step_time": 6.935547828674316
    },
    {
      "epoch": 1.2181282043457031e-05,
      "model_forward_time": 1.879333734512329,
      "step": 86773
    },
    {
      "epoch": 1.2181282043457031e-05,
      "step": 86773,
      "training_step_time": 8.371748208999634
    },
    {
      "epoch": 1.2182235717773437e-05,
      "model_forward_time": 1.8951690196990967,
      "step": 86774
    },
    {
      "epoch": 1.2182235717773437e-05,
      "step": 86774,
      "training_step_time": 8.622880220413208
    },
    {
      "epoch": 1.2183189392089844e-05,
      "model_forward_time": 2.528507709503174,
      "step": 86775
    },
    {
      "epoch": 1.2183189392089844e-05,
      "step": 86775,
      "training_step_time": 8.965882539749146
    },
    {
      "epoch": 1.218414306640625e-05,
      "model_forward_time": 1.902513027191162,
      "step": 86776
    },
    {
      "epoch": 1.218414306640625e-05,
      "step": 86776,
      "training_step_time": 8.913254499435425
    },
    {
      "epoch": 1.2185096740722656e-05,
      "model_forward_time": 1.9037370681762695,
      "step": 86777
    },
    {
      "epoch": 1.2185096740722656e-05,
      "step": 86777,
      "training_step_time": 8.650736331939697
    },
    {
      "epoch": 1.2186050415039063e-05,
      "model_forward_time": 1.898676872253418,
      "step": 86778
    },
    {
      "epoch": 1.2186050415039063e-05,
      "step": 86778,
      "training_step_time": 8.697732925415039
    },
    {
      "epoch": 1.2187004089355469e-05,
      "model_forward_time": 1.9151525497436523,
      "step": 86779
    },
    {
      "epoch": 1.2187004089355469e-05,
      "step": 86779,
      "training_step_time": 8.010464906692505
    },
    {
      "epoch": 1.2187957763671874e-05,
      "grad_norm": 0.08902952820062637,
      "learning_rate": 4.703184784035836e-07,
      "loss": 0.0537,
      "step": 86780
    },
    {
      "epoch": 1.2187957763671874e-05,
      "model_forward_time": 1.9088921546936035,
      "step": 86780
    },
    {
      "dynamics_loss_avg": 0.06369336321949959,
      "epoch": 1.2187957763671874e-05,
      "step": 86780
    },
    {
      "action_loss_avg": 0.0013873127638362349,
      "epoch": 1.2187957763671874e-05,
      "step": 86780
    },
    {
      "epoch": 1.2187957763671874e-05,
      "step": 86780,
      "training_step_time": 7.307644605636597
    },
    {
      "epoch": 1.2188911437988282e-05,
      "model_forward_time": 1.8847081661224365,
      "step": 86781
    },
    {
      "epoch": 1.2188911437988282e-05,
      "step": 86781,
      "training_step_time": 7.007365465164185
    },
    {
      "epoch": 1.2189865112304688e-05,
      "model_forward_time": 1.910320520401001,
      "step": 86782
    },
    {
      "epoch": 1.2189865112304688e-05,
      "step": 86782,
      "training_step_time": 7.516196012496948
    },
    {
      "epoch": 1.2190818786621093e-05,
      "model_forward_time": 1.9204809665679932,
      "step": 86783
    },
    {
      "epoch": 1.2190818786621093e-05,
      "step": 86783,
      "training_step_time": 7.8592848777771
    },
    {
      "epoch": 1.21917724609375e-05,
      "model_forward_time": 2.521801710128784,
      "step": 86784
    },
    {
      "epoch": 1.21917724609375e-05,
      "step": 86784,
      "training_step_time": 9.464842081069946
    },
    {
      "epoch": 1.2192726135253906e-05,
      "model_forward_time": 1.9087939262390137,
      "step": 86785
    },
    {
      "epoch": 1.2192726135253906e-05,
      "step": 86785,
      "training_step_time": 8.64845895767212
    },
    {
      "epoch": 1.2193679809570312e-05,
      "model_forward_time": 1.846705436706543,
      "step": 86786
    },
    {
      "epoch": 1.2193679809570312e-05,
      "step": 86786,
      "training_step_time": 9.843112707138062
    },
    {
      "epoch": 1.219463348388672e-05,
      "model_forward_time": 1.7592756748199463,
      "step": 86787
    },
    {
      "epoch": 1.219463348388672e-05,
      "step": 86787,
      "training_step_time": 7.885684251785278
    },
    {
      "epoch": 1.2195587158203125e-05,
      "model_forward_time": 1.8066325187683105,
      "step": 86788
    },
    {
      "epoch": 1.2195587158203125e-05,
      "step": 86788,
      "training_step_time": 7.889725208282471
    },
    {
      "epoch": 1.2196540832519531e-05,
      "model_forward_time": 1.7459256649017334,
      "step": 86789
    },
    {
      "epoch": 1.2196540832519531e-05,
      "step": 86789,
      "training_step_time": 7.309921741485596
    },
    {
      "epoch": 1.2197494506835937e-05,
      "grad_norm": 0.07558703422546387,
      "learning_rate": 4.6961862383216426e-07,
      "loss": 0.0529,
      "step": 86790
    },
    {
      "epoch": 1.2197494506835937e-05,
      "model_forward_time": 1.7486164569854736,
      "step": 86790
    },
    {
      "dynamics_loss_avg": 0.044097531214356424,
      "epoch": 1.2197494506835937e-05,
      "step": 86790
    },
    {
      "action_loss_avg": 0.0007792986332788132,
      "epoch": 1.2197494506835937e-05,
      "step": 86790
    },
    {
      "epoch": 1.2197494506835937e-05,
      "step": 86790,
      "training_step_time": 7.027857303619385
    },
    {
      "epoch": 1.2198448181152344e-05,
      "model_forward_time": 1.7572879791259766,
      "step": 86791
    },
    {
      "epoch": 1.2198448181152344e-05,
      "step": 86791,
      "training_step_time": 8.305108547210693
    },
    {
      "epoch": 1.219940185546875e-05,
      "model_forward_time": 1.7513508796691895,
      "step": 86792
    },
    {
      "epoch": 1.219940185546875e-05,
      "step": 86792,
      "training_step_time": 7.818981647491455
    },
    {
      "epoch": 1.2200355529785156e-05,
      "model_forward_time": 2.411332368850708,
      "step": 86793
    },
    {
      "epoch": 1.2200355529785156e-05,
      "step": 86793,
      "training_step_time": 8.914393424987793
    },
    {
      "epoch": 1.2201309204101563e-05,
      "model_forward_time": 1.747758150100708,
      "step": 86794
    },
    {
      "epoch": 1.2201309204101563e-05,
      "step": 86794,
      "training_step_time": 8.569906234741211
    },
    {
      "epoch": 1.2202262878417969e-05,
      "model_forward_time": 1.752912998199463,
      "step": 86795
    },
    {
      "epoch": 1.2202262878417969e-05,
      "step": 86795,
      "training_step_time": 9.522611618041992
    },
    {
      "epoch": 1.2203216552734374e-05,
      "model_forward_time": 1.7607769966125488,
      "step": 86796
    },
    {
      "epoch": 1.2203216552734374e-05,
      "step": 86796,
      "training_step_time": 7.8934326171875
    },
    {
      "epoch": 1.2204170227050782e-05,
      "model_forward_time": 1.7592971324920654,
      "step": 86797
    },
    {
      "epoch": 1.2204170227050782e-05,
      "step": 86797,
      "training_step_time": 7.902470111846924
    },
    {
      "epoch": 1.2205123901367188e-05,
      "model_forward_time": 1.7469885349273682,
      "step": 86798
    },
    {
      "epoch": 1.2205123901367188e-05,
      "step": 86798,
      "training_step_time": 6.946616172790527
    },
    {
      "epoch": 1.2206077575683593e-05,
      "model_forward_time": 1.7569553852081299,
      "step": 86799
    },
    {
      "epoch": 1.2206077575683593e-05,
      "step": 86799,
      "training_step_time": 7.248530864715576
    },
    {
      "epoch": 1.220703125e-05,
      "grad_norm": 0.10220404714345932,
      "learning_rate": 4.689192646964408e-07,
      "loss": 0.0531,
      "step": 86800
    },
    {
      "epoch": 1.220703125e-05,
      "model_forward_time": 1.7635881900787354,
      "step": 86800
    },
    {
      "dynamics_loss_avg": 0.04808528861030936,
      "epoch": 1.220703125e-05,
      "step": 86800
    },
    {
      "action_loss_avg": 0.0009937488939613104,
      "epoch": 1.220703125e-05,
      "step": 86800
    },
    {
      "epoch": 1.220703125e-05,
      "step": 86800,
      "training_step_time": 7.483499526977539
    },
    {
      "epoch": 1.2207984924316406e-05,
      "model_forward_time": 1.7515408992767334,
      "step": 86801
    },
    {
      "epoch": 1.2207984924316406e-05,
      "step": 86801,
      "training_step_time": 7.707781791687012
    },
    {
      "epoch": 1.2208938598632812e-05,
      "model_forward_time": 1.7554810047149658,
      "step": 86802
    },
    {
      "epoch": 1.2208938598632812e-05,
      "step": 86802,
      "training_step_time": 8.404847621917725
    },
    {
      "epoch": 1.220989227294922e-05,
      "model_forward_time": 1.7571635246276855,
      "step": 86803
    },
    {
      "epoch": 1.220989227294922e-05,
      "step": 86803,
      "training_step_time": 8.530226707458496
    },
    {
      "epoch": 1.2210845947265625e-05,
      "model_forward_time": 1.8743188381195068,
      "step": 86804
    },
    {
      "epoch": 1.2210845947265625e-05,
      "step": 86804,
      "training_step_time": 10.13918161392212
    },
    {
      "epoch": 1.2211799621582031e-05,
      "model_forward_time": 1.7586743831634521,
      "step": 86805
    },
    {
      "epoch": 1.2211799621582031e-05,
      "step": 86805,
      "training_step_time": 7.908090829849243
    },
    {
      "epoch": 1.2212753295898437e-05,
      "model_forward_time": 1.7543940544128418,
      "step": 86806
    },
    {
      "epoch": 1.2212753295898437e-05,
      "step": 86806,
      "training_step_time": 7.654702663421631
    },
    {
      "epoch": 1.2213706970214844e-05,
      "model_forward_time": 1.7583255767822266,
      "step": 86807
    },
    {
      "epoch": 1.2213706970214844e-05,
      "step": 86807,
      "training_step_time": 7.954757928848267
    },
    {
      "epoch": 1.221466064453125e-05,
      "model_forward_time": 1.7589738368988037,
      "step": 86808
    },
    {
      "epoch": 1.221466064453125e-05,
      "step": 86808,
      "training_step_time": 7.3024444580078125
    },
    {
      "epoch": 1.2215614318847656e-05,
      "model_forward_time": 1.7569005489349365,
      "step": 86809
    },
    {
      "epoch": 1.2215614318847656e-05,
      "step": 86809,
      "training_step_time": 6.92227840423584
    },
    {
      "epoch": 1.2216567993164063e-05,
      "grad_norm": 0.2587986886501312,
      "learning_rate": 4.6822040107289525e-07,
      "loss": 0.0531,
      "step": 86810
    },
    {
      "epoch": 1.2216567993164063e-05,
      "model_forward_time": 1.7526555061340332,
      "step": 86810
    },
    {
      "dynamics_loss_avg": 0.05493737235665321,
      "epoch": 1.2216567993164063e-05,
      "step": 86810
    },
    {
      "action_loss_avg": 0.0008035606297198683,
      "epoch": 1.2216567993164063e-05,
      "step": 86810
    },
    {
      "epoch": 1.2216567993164063e-05,
      "step": 86810,
      "training_step_time": 8.620140552520752
    },
    {
      "epoch": 1.2217521667480469e-05,
      "model_forward_time": 1.7645089626312256,
      "step": 86811
    },
    {
      "epoch": 1.2217521667480469e-05,
      "step": 86811,
      "training_step_time": 8.887623310089111
    },
    {
      "epoch": 1.2218475341796874e-05,
      "model_forward_time": 2.3856325149536133,
      "step": 86812
    },
    {
      "epoch": 1.2218475341796874e-05,
      "step": 86812,
      "training_step_time": 7.962658882141113
    },
    {
      "epoch": 1.2219429016113282e-05,
      "model_forward_time": 1.8433074951171875,
      "step": 86813
    },
    {
      "epoch": 1.2219429016113282e-05,
      "step": 86813,
      "training_step_time": 9.464995622634888
    },
    {
      "epoch": 1.2220382690429688e-05,
      "model_forward_time": 1.750722885131836,
      "step": 86814
    },
    {
      "epoch": 1.2220382690429688e-05,
      "step": 86814,
      "training_step_time": 8.633854389190674
    },
    {
      "epoch": 1.2221336364746093e-05,
      "model_forward_time": 1.75649094581604,
      "step": 86815
    },
    {
      "epoch": 1.2221336364746093e-05,
      "step": 86815,
      "training_step_time": 7.861642122268677
    },
    {
      "epoch": 1.22222900390625e-05,
      "model_forward_time": 1.7618634700775146,
      "step": 86816
    },
    {
      "epoch": 1.22222900390625e-05,
      "step": 86816,
      "training_step_time": 7.859913349151611
    },
    {
      "epoch": 1.2223243713378906e-05,
      "model_forward_time": 1.761385440826416,
      "step": 86817
    },
    {
      "epoch": 1.2223243713378906e-05,
      "step": 86817,
      "training_step_time": 7.826215028762817
    },
    {
      "epoch": 1.2224197387695312e-05,
      "model_forward_time": 1.7481555938720703,
      "step": 86818
    },
    {
      "epoch": 1.2224197387695312e-05,
      "step": 86818,
      "training_step_time": 7.5082926750183105
    },
    {
      "epoch": 1.222515106201172e-05,
      "model_forward_time": 1.7465956211090088,
      "step": 86819
    },
    {
      "epoch": 1.222515106201172e-05,
      "step": 86819,
      "training_step_time": 8.239333152770996
    },
    {
      "epoch": 1.2226104736328125e-05,
      "grad_norm": 0.08680661767721176,
      "learning_rate": 4.6752203303795206e-07,
      "loss": 0.0529,
      "step": 86820
    },
    {
      "epoch": 1.2226104736328125e-05,
      "model_forward_time": 1.747664451599121,
      "step": 86820
    },
    {
      "dynamics_loss_avg": 0.053491972386837006,
      "epoch": 1.2226104736328125e-05,
      "step": 86820
    },
    {
      "action_loss_avg": 0.001347043481655419,
      "epoch": 1.2226104736328125e-05,
      "step": 86820
    },
    {
      "epoch": 1.2226104736328125e-05,
      "step": 86820,
      "training_step_time": 7.770142316818237
    },
    {
      "epoch": 1.2227058410644531e-05,
      "model_forward_time": 2.3940377235412598,
      "step": 86821
    },
    {
      "epoch": 1.2227058410644531e-05,
      "step": 86821,
      "training_step_time": 9.439953088760376
    },
    {
      "epoch": 1.2228012084960937e-05,
      "model_forward_time": 1.8297929763793945,
      "step": 86822
    },
    {
      "epoch": 1.2228012084960937e-05,
      "step": 86822,
      "training_step_time": 10.10502028465271
    },
    {
      "epoch": 1.2228965759277344e-05,
      "model_forward_time": 1.765099048614502,
      "step": 86823
    },
    {
      "epoch": 1.2228965759277344e-05,
      "step": 86823,
      "training_step_time": 10.607730865478516
    },
    {
      "epoch": 1.222991943359375e-05,
      "model_forward_time": 1.769223928451538,
      "step": 86824
    },
    {
      "epoch": 1.222991943359375e-05,
      "step": 86824,
      "training_step_time": 7.900980710983276
    },
    {
      "epoch": 1.2230873107910156e-05,
      "model_forward_time": 1.7679922580718994,
      "step": 86825
    },
    {
      "epoch": 1.2230873107910156e-05,
      "step": 86825,
      "training_step_time": 7.583706617355347
    },
    {
      "epoch": 1.2231826782226563e-05,
      "model_forward_time": 1.7794065475463867,
      "step": 86826
    },
    {
      "epoch": 1.2231826782226563e-05,
      "step": 86826,
      "training_step_time": 8.36039400100708
    },
    {
      "epoch": 1.2232780456542969e-05,
      "model_forward_time": 1.7545843124389648,
      "step": 86827
    },
    {
      "epoch": 1.2232780456542969e-05,
      "step": 86827,
      "training_step_time": 7.459599018096924
    },
    {
      "epoch": 1.2233734130859374e-05,
      "model_forward_time": 1.762082576751709,
      "step": 86828
    },
    {
      "epoch": 1.2233734130859374e-05,
      "step": 86828,
      "training_step_time": 7.522446870803833
    },
    {
      "epoch": 1.2234687805175782e-05,
      "model_forward_time": 1.7591168880462646,
      "step": 86829
    },
    {
      "epoch": 1.2234687805175782e-05,
      "step": 86829,
      "training_step_time": 7.766327619552612
    },
    {
      "epoch": 1.2235641479492188e-05,
      "grad_norm": 0.21998313069343567,
      "learning_rate": 4.6682416066798463e-07,
      "loss": 0.0523,
      "step": 86830
    },
    {
      "epoch": 1.2235641479492188e-05,
      "model_forward_time": 2.3927876949310303,
      "step": 86830
    },
    {
      "dynamics_loss_avg": 0.05692866155877709,
      "epoch": 1.2235641479492188e-05,
      "step": 86830
    },
    {
      "action_loss_avg": 0.0013935620605479925,
      "epoch": 1.2235641479492188e-05,
      "step": 86830
    },
    {
      "epoch": 1.2235641479492188e-05,
      "step": 86830,
      "training_step_time": 7.9962615966796875
    },
    {
      "epoch": 1.2236595153808593e-05,
      "model_forward_time": 1.7963054180145264,
      "step": 86831
    },
    {
      "epoch": 1.2236595153808593e-05,
      "step": 86831,
      "training_step_time": 8.522192001342773
    },
    {
      "epoch": 1.2237548828125e-05,
      "model_forward_time": 1.7721521854400635,
      "step": 86832
    },
    {
      "epoch": 1.2237548828125e-05,
      "step": 86832,
      "training_step_time": 7.990744113922119
    },
    {
      "epoch": 1.2238502502441406e-05,
      "model_forward_time": 1.7581253051757812,
      "step": 86833
    },
    {
      "epoch": 1.2238502502441406e-05,
      "step": 86833,
      "training_step_time": 9.345411777496338
    },
    {
      "epoch": 1.2239456176757812e-05,
      "model_forward_time": 1.7580540180206299,
      "step": 86834
    },
    {
      "epoch": 1.2239456176757812e-05,
      "step": 86834,
      "training_step_time": 7.562745094299316
    },
    {
      "epoch": 1.224040985107422e-05,
      "model_forward_time": 1.76255202293396,
      "step": 86835
    },
    {
      "epoch": 1.224040985107422e-05,
      "step": 86835,
      "training_step_time": 7.811464548110962
    },
    {
      "epoch": 1.2241363525390625e-05,
      "model_forward_time": 1.7542438507080078,
      "step": 86836
    },
    {
      "epoch": 1.2241363525390625e-05,
      "step": 86836,
      "training_step_time": 7.999783515930176
    },
    {
      "epoch": 1.2242317199707031e-05,
      "model_forward_time": 1.767899513244629,
      "step": 86837
    },
    {
      "epoch": 1.2242317199707031e-05,
      "step": 86837,
      "training_step_time": 7.5244457721710205
    },
    {
      "epoch": 1.2243270874023437e-05,
      "model_forward_time": 1.7747821807861328,
      "step": 86838
    },
    {
      "epoch": 1.2243270874023437e-05,
      "step": 86838,
      "training_step_time": 8.59997272491455
    },
    {
      "epoch": 1.2244224548339844e-05,
      "model_forward_time": 2.4410603046417236,
      "step": 86839
    },
    {
      "epoch": 1.2244224548339844e-05,
      "step": 86839,
      "training_step_time": 8.03104019165039
    },
    {
      "epoch": 1.224517822265625e-05,
      "grad_norm": 0.0986349806189537,
      "learning_rate": 4.66126784039313e-07,
      "loss": 0.0535,
      "step": 86840
    },
    {
      "epoch": 1.224517822265625e-05,
      "model_forward_time": 1.775177001953125,
      "step": 86840
    },
    {
      "dynamics_loss_avg": 0.06231113709509373,
      "epoch": 1.224517822265625e-05,
      "step": 86840
    },
    {
      "action_loss_avg": 0.0017546611343277618,
      "epoch": 1.224517822265625e-05,
      "step": 86840
    },
    {
      "epoch": 1.224517822265625e-05,
      "step": 86840,
      "training_step_time": 9.411033630371094
    },
    {
      "epoch": 1.2246131896972656e-05,
      "model_forward_time": 1.7656755447387695,
      "step": 86841
    },
    {
      "epoch": 1.2246131896972656e-05,
      "step": 86841,
      "training_step_time": 8.01449990272522
    },
    {
      "epoch": 1.2247085571289063e-05,
      "model_forward_time": 1.7590880393981934,
      "step": 86842
    },
    {
      "epoch": 1.2247085571289063e-05,
      "step": 86842,
      "training_step_time": 8.689564228057861
    },
    {
      "epoch": 1.2248039245605469e-05,
      "model_forward_time": 1.759312629699707,
      "step": 86843
    },
    {
      "epoch": 1.2248039245605469e-05,
      "step": 86843,
      "training_step_time": 7.832731485366821
    },
    {
      "epoch": 1.2248992919921874e-05,
      "model_forward_time": 1.7410881519317627,
      "step": 86844
    },
    {
      "epoch": 1.2248992919921874e-05,
      "step": 86844,
      "training_step_time": 7.893804311752319
    },
    {
      "epoch": 1.2249946594238282e-05,
      "model_forward_time": 1.7561261653900146,
      "step": 86845
    },
    {
      "epoch": 1.2249946594238282e-05,
      "step": 86845,
      "training_step_time": 7.468717336654663
    },
    {
      "epoch": 1.2250900268554688e-05,
      "model_forward_time": 1.7710745334625244,
      "step": 86846
    },
    {
      "epoch": 1.2250900268554688e-05,
      "step": 86846,
      "training_step_time": 7.452482223510742
    },
    {
      "epoch": 1.2251853942871093e-05,
      "model_forward_time": 1.3812904357910156,
      "step": 86847
    },
    {
      "epoch": 1.2251853942871093e-05,
      "step": 86847,
      "training_step_time": 8.727316856384277
    },
    {
      "epoch": 1.22528076171875e-05,
      "model_forward_time": 1.7508013248443604,
      "step": 86848
    },
    {
      "epoch": 1.22528076171875e-05,
      "step": 86848,
      "training_step_time": 8.472682237625122
    },
    {
      "epoch": 1.2253761291503906e-05,
      "model_forward_time": 1.401426076889038,
      "step": 86849
    },
    {
      "epoch": 1.2253761291503906e-05,
      "step": 86849,
      "training_step_time": 9.710621118545532
    },
    {
      "epoch": 1.2254714965820312e-05,
      "grad_norm": 0.07775204628705978,
      "learning_rate": 4.654299032281984e-07,
      "loss": 0.0527,
      "step": 86850
    },
    {
      "epoch": 1.2254714965820312e-05,
      "model_forward_time": 1.7493700981140137,
      "step": 86850
    },
    {
      "dynamics_loss_avg": 0.043686136975884436,
      "epoch": 1.2254714965820312e-05,
      "step": 86850
    },
    {
      "action_loss_avg": 0.0009707481076475233,
      "epoch": 1.2254714965820312e-05,
      "step": 86850
    },
    {
      "epoch": 1.2254714965820312e-05,
      "step": 86850,
      "training_step_time": 9.856218814849854
    },
    {
      "epoch": 1.225566864013672e-05,
      "model_forward_time": 1.7596421241760254,
      "step": 86851
    },
    {
      "epoch": 1.225566864013672e-05,
      "step": 86851,
      "training_step_time": 7.780847787857056
    },
    {
      "epoch": 1.2256622314453125e-05,
      "model_forward_time": 1.7614245414733887,
      "step": 86852
    },
    {
      "epoch": 1.2256622314453125e-05,
      "step": 86852,
      "training_step_time": 7.8392884731292725
    },
    {
      "epoch": 1.2257575988769531e-05,
      "model_forward_time": 1.7695574760437012,
      "step": 86853
    },
    {
      "epoch": 1.2257575988769531e-05,
      "step": 86853,
      "training_step_time": 7.877326965332031
    },
    {
      "epoch": 1.2258529663085937e-05,
      "model_forward_time": 1.7710597515106201,
      "step": 86854
    },
    {
      "epoch": 1.2258529663085937e-05,
      "step": 86854,
      "training_step_time": 6.939826011657715
    },
    {
      "epoch": 1.2259483337402344e-05,
      "model_forward_time": 1.76119065284729,
      "step": 86855
    },
    {
      "epoch": 1.2259483337402344e-05,
      "step": 86855,
      "training_step_time": 7.846649169921875
    },
    {
      "epoch": 1.226043701171875e-05,
      "model_forward_time": 1.7625503540039062,
      "step": 86856
    },
    {
      "epoch": 1.226043701171875e-05,
      "step": 86856,
      "training_step_time": 7.5661070346832275
    },
    {
      "epoch": 1.2261390686035156e-05,
      "model_forward_time": 1.760519027709961,
      "step": 86857
    },
    {
      "epoch": 1.2261390686035156e-05,
      "step": 86857,
      "training_step_time": 9.327958345413208
    },
    {
      "epoch": 1.2262344360351563e-05,
      "model_forward_time": 2.002471923828125,
      "step": 86858
    },
    {
      "epoch": 1.2262344360351563e-05,
      "step": 86858,
      "training_step_time": 9.88604736328125
    },
    {
      "epoch": 1.2263298034667969e-05,
      "model_forward_time": 1.8845205307006836,
      "step": 86859
    },
    {
      "epoch": 1.2263298034667969e-05,
      "step": 86859,
      "training_step_time": 10.20932650566101
    },
    {
      "epoch": 1.2264251708984374e-05,
      "grad_norm": 0.10134689509868622,
      "learning_rate": 4.64733518310852e-07,
      "loss": 0.0527,
      "step": 86860
    },
    {
      "epoch": 1.2264251708984374e-05,
      "model_forward_time": 1.7468419075012207,
      "step": 86860
    },
    {
      "dynamics_loss_avg": 0.0476496959105134,
      "epoch": 1.2264251708984374e-05,
      "step": 86860
    },
    {
      "action_loss_avg": 0.0011491750192362815,
      "epoch": 1.2264251708984374e-05,
      "step": 86860
    },
    {
      "epoch": 1.2264251708984374e-05,
      "step": 86860,
      "training_step_time": 7.977279186248779
    },
    {
      "epoch": 1.2265205383300782e-05,
      "model_forward_time": 1.773505449295044,
      "step": 86861
    },
    {
      "epoch": 1.2265205383300782e-05,
      "step": 86861,
      "training_step_time": 7.795705795288086
    },
    {
      "epoch": 1.2266159057617188e-05,
      "model_forward_time": 1.7572097778320312,
      "step": 86862
    },
    {
      "epoch": 1.2266159057617188e-05,
      "step": 86862,
      "training_step_time": 7.802523851394653
    },
    {
      "epoch": 1.2267112731933593e-05,
      "model_forward_time": 1.7607777118682861,
      "step": 86863
    },
    {
      "epoch": 1.2267112731933593e-05,
      "step": 86863,
      "training_step_time": 7.436475038528442
    },
    {
      "epoch": 1.226806640625e-05,
      "model_forward_time": 1.7624454498291016,
      "step": 86864
    },
    {
      "epoch": 1.226806640625e-05,
      "step": 86864,
      "training_step_time": 7.266457796096802
    },
    {
      "epoch": 1.2269020080566406e-05,
      "model_forward_time": 1.766810655593872,
      "step": 86865
    },
    {
      "epoch": 1.2269020080566406e-05,
      "step": 86865,
      "training_step_time": 7.417283535003662
    },
    {
      "epoch": 1.2269973754882812e-05,
      "model_forward_time": 1.7594141960144043,
      "step": 86866
    },
    {
      "epoch": 1.2269973754882812e-05,
      "step": 86866,
      "training_step_time": 7.725720643997192
    },
    {
      "epoch": 1.227092742919922e-05,
      "model_forward_time": 2.408830165863037,
      "step": 86867
    },
    {
      "epoch": 1.227092742919922e-05,
      "step": 86867,
      "training_step_time": 7.992002725601196
    },
    {
      "epoch": 1.2271881103515625e-05,
      "model_forward_time": 1.8809030055999756,
      "step": 86868
    },
    {
      "epoch": 1.2271881103515625e-05,
      "step": 86868,
      "training_step_time": 10.349485635757446
    },
    {
      "epoch": 1.2272834777832031e-05,
      "model_forward_time": 1.7482366561889648,
      "step": 86869
    },
    {
      "epoch": 1.2272834777832031e-05,
      "step": 86869,
      "training_step_time": 9.294743537902832
    },
    {
      "epoch": 1.2273788452148437e-05,
      "grad_norm": 0.07302612066268921,
      "learning_rate": 4.640376293634291e-07,
      "loss": 0.0538,
      "step": 86870
    },
    {
      "epoch": 1.2273788452148437e-05,
      "model_forward_time": 1.7604341506958008,
      "step": 86870
    },
    {
      "dynamics_loss_avg": 0.04290615813806653,
      "epoch": 1.2273788452148437e-05,
      "step": 86870
    },
    {
      "action_loss_avg": 0.001589674418210052,
      "epoch": 1.2273788452148437e-05,
      "step": 86870
    },
    {
      "epoch": 1.2273788452148437e-05,
      "step": 86870,
      "training_step_time": 7.804869890213013
    },
    {
      "epoch": 1.2274742126464844e-05,
      "model_forward_time": 1.3925485610961914,
      "step": 86871
    },
    {
      "epoch": 1.2274742126464844e-05,
      "step": 86871,
      "training_step_time": 7.558178186416626
    },
    {
      "epoch": 1.227569580078125e-05,
      "model_forward_time": 1.7730472087860107,
      "step": 86872
    },
    {
      "epoch": 1.227569580078125e-05,
      "step": 86872,
      "training_step_time": 7.696551561355591
    },
    {
      "epoch": 1.2276649475097656e-05,
      "model_forward_time": 1.7732927799224854,
      "step": 86873
    },
    {
      "epoch": 1.2276649475097656e-05,
      "step": 86873,
      "training_step_time": 7.854807376861572
    },
    {
      "epoch": 1.2277603149414063e-05,
      "model_forward_time": 1.747851848602295,
      "step": 86874
    },
    {
      "epoch": 1.2277603149414063e-05,
      "step": 86874,
      "training_step_time": 7.422957897186279
    },
    {
      "epoch": 1.2278556823730469e-05,
      "model_forward_time": 1.7623882293701172,
      "step": 86875
    },
    {
      "epoch": 1.2278556823730469e-05,
      "step": 86875,
      "training_step_time": 8.859777450561523
    },
    {
      "epoch": 1.2279510498046874e-05,
      "model_forward_time": 2.4118001461029053,
      "step": 86876
    },
    {
      "epoch": 1.2279510498046874e-05,
      "step": 86876,
      "training_step_time": 9.849618911743164
    },
    {
      "epoch": 1.2280464172363282e-05,
      "model_forward_time": 1.8795769214630127,
      "step": 86877
    },
    {
      "epoch": 1.2280464172363282e-05,
      "step": 86877,
      "training_step_time": 8.324512481689453
    },
    {
      "epoch": 1.2281417846679688e-05,
      "model_forward_time": 1.7606925964355469,
      "step": 86878
    },
    {
      "epoch": 1.2281417846679688e-05,
      "step": 86878,
      "training_step_time": 9.591857433319092
    },
    {
      "epoch": 1.2282371520996093e-05,
      "model_forward_time": 1.7601814270019531,
      "step": 86879
    },
    {
      "epoch": 1.2282371520996093e-05,
      "step": 86879,
      "training_step_time": 8.40196442604065
    },
    {
      "epoch": 1.22833251953125e-05,
      "grad_norm": 0.08147993683815002,
      "learning_rate": 4.633422364620327e-07,
      "loss": 0.0528,
      "step": 86880
    },
    {
      "epoch": 1.22833251953125e-05,
      "model_forward_time": 1.748769998550415,
      "step": 86880
    },
    {
      "dynamics_loss_avg": 0.06373708667233587,
      "epoch": 1.22833251953125e-05,
      "step": 86880
    },
    {
      "action_loss_avg": 0.0009121374539972748,
      "epoch": 1.22833251953125e-05,
      "step": 86880
    },
    {
      "epoch": 1.22833251953125e-05,
      "step": 86880,
      "training_step_time": 7.8086442947387695
    },
    {
      "epoch": 1.2284278869628906e-05,
      "model_forward_time": 1.7849841117858887,
      "step": 86881
    },
    {
      "epoch": 1.2284278869628906e-05,
      "step": 86881,
      "training_step_time": 7.7242937088012695
    },
    {
      "epoch": 1.2285232543945312e-05,
      "model_forward_time": 1.7580323219299316,
      "step": 86882
    },
    {
      "epoch": 1.2285232543945312e-05,
      "step": 86882,
      "training_step_time": 7.937601804733276
    },
    {
      "epoch": 1.2286186218261718e-05,
      "model_forward_time": 1.7812063694000244,
      "step": 86883
    },
    {
      "epoch": 1.2286186218261718e-05,
      "step": 86883,
      "training_step_time": 7.397470712661743
    },
    {
      "epoch": 1.2287139892578125e-05,
      "model_forward_time": 1.7618215084075928,
      "step": 86884
    },
    {
      "epoch": 1.2287139892578125e-05,
      "step": 86884,
      "training_step_time": 7.780562877655029
    },
    {
      "epoch": 1.2288093566894531e-05,
      "model_forward_time": 1.761516809463501,
      "step": 86885
    },
    {
      "epoch": 1.2288093566894531e-05,
      "step": 86885,
      "training_step_time": 9.503944873809814
    },
    {
      "epoch": 1.2289047241210937e-05,
      "model_forward_time": 1.8722739219665527,
      "step": 86886
    },
    {
      "epoch": 1.2289047241210937e-05,
      "step": 86886,
      "training_step_time": 10.989850282669067
    },
    {
      "epoch": 1.2290000915527344e-05,
      "model_forward_time": 1.7648770809173584,
      "step": 86887
    },
    {
      "epoch": 1.2290000915527344e-05,
      "step": 86887,
      "training_step_time": 9.317070960998535
    },
    {
      "epoch": 1.229095458984375e-05,
      "model_forward_time": 1.7415173053741455,
      "step": 86888
    },
    {
      "epoch": 1.229095458984375e-05,
      "step": 86888,
      "training_step_time": 7.958559513092041
    },
    {
      "epoch": 1.2291908264160156e-05,
      "model_forward_time": 1.7576310634613037,
      "step": 86889
    },
    {
      "epoch": 1.2291908264160156e-05,
      "step": 86889,
      "training_step_time": 7.738040208816528
    },
    {
      "epoch": 1.2292861938476563e-05,
      "grad_norm": 0.16381755471229553,
      "learning_rate": 4.6264733968270746e-07,
      "loss": 0.0541,
      "step": 86890
    },
    {
      "epoch": 1.2292861938476563e-05,
      "model_forward_time": 1.393542766571045,
      "step": 86890
    },
    {
      "dynamics_loss_avg": 0.061573652736842635,
      "epoch": 1.2292861938476563e-05,
      "step": 86890
    },
    {
      "action_loss_avg": 0.0016961516317678615,
      "epoch": 1.2292861938476563e-05,
      "step": 86890
    },
    {
      "epoch": 1.2292861938476563e-05,
      "step": 86890,
      "training_step_time": 7.734621047973633
    },
    {
      "epoch": 1.2293815612792969e-05,
      "model_forward_time": 1.7660748958587646,
      "step": 86891
    },
    {
      "epoch": 1.2293815612792969e-05,
      "step": 86891,
      "training_step_time": 7.801922798156738
    },
    {
      "epoch": 1.2294769287109374e-05,
      "model_forward_time": 1.7734878063201904,
      "step": 86892
    },
    {
      "epoch": 1.2294769287109374e-05,
      "step": 86892,
      "training_step_time": 7.408017635345459
    },
    {
      "epoch": 1.2295722961425782e-05,
      "model_forward_time": 1.7774333953857422,
      "step": 86893
    },
    {
      "epoch": 1.2295722961425782e-05,
      "step": 86893,
      "training_step_time": 8.747506856918335
    },
    {
      "epoch": 1.2296676635742188e-05,
      "model_forward_time": 1.7851014137268066,
      "step": 86894
    },
    {
      "epoch": 1.2296676635742188e-05,
      "step": 86894,
      "training_step_time": 9.570694923400879
    },
    {
      "epoch": 1.2297630310058593e-05,
      "model_forward_time": 2.466623067855835,
      "step": 86895
    },
    {
      "epoch": 1.2297630310058593e-05,
      "step": 86895,
      "training_step_time": 8.612180709838867
    },
    {
      "epoch": 1.2298583984375e-05,
      "model_forward_time": 1.7696912288665771,
      "step": 86896
    },
    {
      "epoch": 1.2298583984375e-05,
      "step": 86896,
      "training_step_time": 8.606179475784302
    },
    {
      "epoch": 1.2299537658691406e-05,
      "model_forward_time": 1.7710206508636475,
      "step": 86897
    },
    {
      "epoch": 1.2299537658691406e-05,
      "step": 86897,
      "training_step_time": 8.88587212562561
    },
    {
      "epoch": 1.2300491333007812e-05,
      "model_forward_time": 1.7660887241363525,
      "step": 86898
    },
    {
      "epoch": 1.2300491333007812e-05,
      "step": 86898,
      "training_step_time": 7.736072063446045
    },
    {
      "epoch": 1.2301445007324218e-05,
      "model_forward_time": 1.7709629535675049,
      "step": 86899
    },
    {
      "epoch": 1.2301445007324218e-05,
      "step": 86899,
      "training_step_time": 8.584968328475952
    },
    {
      "epoch": 1.2302398681640625e-05,
      "grad_norm": 0.08861267566680908,
      "learning_rate": 4.6195293910144823e-07,
      "loss": 0.0528,
      "step": 86900
    },
    {
      "epoch": 1.2302398681640625e-05,
      "model_forward_time": 1.755704402923584,
      "step": 86900
    },
    {
      "dynamics_loss_avg": 0.051176943071186545,
      "epoch": 1.2302398681640625e-05,
      "step": 86900
    },
    {
      "action_loss_avg": 0.0010165215702727438,
      "epoch": 1.2302398681640625e-05,
      "step": 86900
    },
    {
      "epoch": 1.2302398681640625e-05,
      "step": 86900,
      "training_step_time": 7.799287796020508
    },
    {
      "epoch": 1.2303352355957031e-05,
      "model_forward_time": 1.768320083618164,
      "step": 86901
    },
    {
      "epoch": 1.2303352355957031e-05,
      "step": 86901,
      "training_step_time": 7.378971338272095
    },
    {
      "epoch": 1.2304306030273437e-05,
      "model_forward_time": 1.7697646617889404,
      "step": 86902
    },
    {
      "epoch": 1.2304306030273437e-05,
      "step": 86902,
      "training_step_time": 7.7755677700042725
    },
    {
      "epoch": 1.2305259704589844e-05,
      "model_forward_time": 1.773693561553955,
      "step": 86903
    },
    {
      "epoch": 1.2305259704589844e-05,
      "step": 86903,
      "training_step_time": 9.611709833145142
    },
    {
      "epoch": 1.230621337890625e-05,
      "model_forward_time": 2.482661008834839,
      "step": 86904
    },
    {
      "epoch": 1.230621337890625e-05,
      "step": 86904,
      "training_step_time": 7.94041633605957
    },
    {
      "epoch": 1.2307167053222656e-05,
      "model_forward_time": 1.7652392387390137,
      "step": 86905
    },
    {
      "epoch": 1.2307167053222656e-05,
      "step": 86905,
      "training_step_time": 7.872544050216675
    },
    {
      "epoch": 1.2308120727539063e-05,
      "model_forward_time": 1.7591145038604736,
      "step": 86906
    },
    {
      "epoch": 1.2308120727539063e-05,
      "step": 86906,
      "training_step_time": 8.03738784790039
    },
    {
      "epoch": 1.2309074401855469e-05,
      "model_forward_time": 1.7707884311676025,
      "step": 86907
    },
    {
      "epoch": 1.2309074401855469e-05,
      "step": 86907,
      "training_step_time": 10.318670749664307
    },
    {
      "epoch": 1.2310028076171874e-05,
      "model_forward_time": 1.7682528495788574,
      "step": 86908
    },
    {
      "epoch": 1.2310028076171874e-05,
      "step": 86908,
      "training_step_time": 7.7376368045806885
    },
    {
      "epoch": 1.2310981750488282e-05,
      "model_forward_time": 1.7620649337768555,
      "step": 86909
    },
    {
      "epoch": 1.2310981750488282e-05,
      "step": 86909,
      "training_step_time": 7.715600490570068
    },
    {
      "epoch": 1.2311935424804688e-05,
      "grad_norm": 0.09377170354127884,
      "learning_rate": 4.61259034794192e-07,
      "loss": 0.0551,
      "step": 86910
    },
    {
      "epoch": 1.2311935424804688e-05,
      "model_forward_time": 1.3768713474273682,
      "step": 86910
    },
    {
      "dynamics_loss_avg": 0.04807044062763453,
      "epoch": 1.2311935424804688e-05,
      "step": 86910
    },
    {
      "action_loss_avg": 0.001212752060382627,
      "epoch": 1.2311935424804688e-05,
      "step": 86910
    },
    {
      "epoch": 1.2311935424804688e-05,
      "step": 86910,
      "training_step_time": 7.778070688247681
    },
    {
      "epoch": 1.2312889099121093e-05,
      "model_forward_time": 1.7609131336212158,
      "step": 86911
    },
    {
      "epoch": 1.2312889099121093e-05,
      "step": 86911,
      "training_step_time": 7.78294563293457
    },
    {
      "epoch": 1.23138427734375e-05,
      "model_forward_time": 1.761657476425171,
      "step": 86912
    },
    {
      "epoch": 1.23138427734375e-05,
      "step": 86912,
      "training_step_time": 9.325565576553345
    },
    {
      "epoch": 1.2314796447753906e-05,
      "model_forward_time": 2.441624879837036,
      "step": 86913
    },
    {
      "epoch": 1.2314796447753906e-05,
      "step": 86913,
      "training_step_time": 10.125432252883911
    },
    {
      "epoch": 1.2315750122070312e-05,
      "model_forward_time": 1.7545413970947266,
      "step": 86914
    },
    {
      "epoch": 1.2315750122070312e-05,
      "step": 86914,
      "training_step_time": 7.83099365234375
    },
    {
      "epoch": 1.2316703796386718e-05,
      "model_forward_time": 1.751152515411377,
      "step": 86915
    },
    {
      "epoch": 1.2316703796386718e-05,
      "step": 86915,
      "training_step_time": 8.914966583251953
    },
    {
      "epoch": 1.2317657470703125e-05,
      "model_forward_time": 1.7545907497406006,
      "step": 86916
    },
    {
      "epoch": 1.2317657470703125e-05,
      "step": 86916,
      "training_step_time": 8.560044288635254
    },
    {
      "epoch": 1.2318611145019531e-05,
      "model_forward_time": 1.7587933540344238,
      "step": 86917
    },
    {
      "epoch": 1.2318611145019531e-05,
      "step": 86917,
      "training_step_time": 7.8545427322387695
    },
    {
      "epoch": 1.2319564819335937e-05,
      "model_forward_time": 1.7798821926116943,
      "step": 86918
    },
    {
      "epoch": 1.2319564819335937e-05,
      "step": 86918,
      "training_step_time": 7.812315464019775
    },
    {
      "epoch": 1.2320518493652344e-05,
      "model_forward_time": 1.754166603088379,
      "step": 86919
    },
    {
      "epoch": 1.2320518493652344e-05,
      "step": 86919,
      "training_step_time": 7.794945001602173
    },
    {
      "epoch": 1.232147216796875e-05,
      "grad_norm": 0.08498440682888031,
      "learning_rate": 4.6056562683682473e-07,
      "loss": 0.0539,
      "step": 86920
    },
    {
      "epoch": 1.232147216796875e-05,
      "model_forward_time": 1.7486646175384521,
      "step": 86920
    },
    {
      "dynamics_loss_avg": 0.05482002757489681,
      "epoch": 1.232147216796875e-05,
      "step": 86920
    },
    {
      "action_loss_avg": 0.0007938213064335286,
      "epoch": 1.232147216796875e-05,
      "step": 86920
    },
    {
      "epoch": 1.232147216796875e-05,
      "step": 86920,
      "training_step_time": 7.8274829387664795
    },
    {
      "epoch": 1.2322425842285156e-05,
      "model_forward_time": 1.7604024410247803,
      "step": 86921
    },
    {
      "epoch": 1.2322425842285156e-05,
      "step": 86921,
      "training_step_time": 8.936973094940186
    },
    {
      "epoch": 1.2323379516601563e-05,
      "model_forward_time": 1.759211540222168,
      "step": 86922
    },
    {
      "epoch": 1.2323379516601563e-05,
      "step": 86922,
      "training_step_time": 9.258870840072632
    },
    {
      "epoch": 1.2324333190917969e-05,
      "model_forward_time": 2.4705047607421875,
      "step": 86923
    },
    {
      "epoch": 1.2324333190917969e-05,
      "step": 86923,
      "training_step_time": 8.88273811340332
    },
    {
      "epoch": 1.2325286865234375e-05,
      "model_forward_time": 1.7523775100708008,
      "step": 86924
    },
    {
      "epoch": 1.2325286865234375e-05,
      "step": 86924,
      "training_step_time": 8.569100379943848
    },
    {
      "epoch": 1.2326240539550782e-05,
      "model_forward_time": 1.7549409866333008,
      "step": 86925
    },
    {
      "epoch": 1.2326240539550782e-05,
      "step": 86925,
      "training_step_time": 8.564781904220581
    },
    {
      "epoch": 1.2327194213867188e-05,
      "model_forward_time": 1.7552390098571777,
      "step": 86926
    },
    {
      "epoch": 1.2327194213867188e-05,
      "step": 86926,
      "training_step_time": 7.843980550765991
    },
    {
      "epoch": 1.2328147888183593e-05,
      "model_forward_time": 1.75714111328125,
      "step": 86927
    },
    {
      "epoch": 1.2328147888183593e-05,
      "step": 86927,
      "training_step_time": 7.814594030380249
    },
    {
      "epoch": 1.23291015625e-05,
      "model_forward_time": 1.7606523036956787,
      "step": 86928
    },
    {
      "epoch": 1.23291015625e-05,
      "step": 86928,
      "training_step_time": 7.002306699752808
    },
    {
      "epoch": 1.2330055236816407e-05,
      "model_forward_time": 1.763012409210205,
      "step": 86929
    },
    {
      "epoch": 1.2330055236816407e-05,
      "step": 86929,
      "training_step_time": 8.898512363433838
    },
    {
      "epoch": 1.2331008911132812e-05,
      "grad_norm": 0.09359730035066605,
      "learning_rate": 4.5987271530517353e-07,
      "loss": 0.0547,
      "step": 86930
    },
    {
      "epoch": 1.2331008911132812e-05,
      "model_forward_time": 1.745232105255127,
      "step": 86930
    },
    {
      "dynamics_loss_avg": 0.04585817139595747,
      "epoch": 1.2331008911132812e-05,
      "step": 86930
    },
    {
      "action_loss_avg": 0.001175742840860039,
      "epoch": 1.2331008911132812e-05,
      "step": 86930
    },
    {
      "epoch": 1.2331008911132812e-05,
      "step": 86930,
      "training_step_time": 7.847033262252808
    },
    {
      "epoch": 1.2331962585449218e-05,
      "model_forward_time": 1.7537519931793213,
      "step": 86931
    },
    {
      "epoch": 1.2331962585449218e-05,
      "step": 86931,
      "training_step_time": 8.51929259300232
    },
    {
      "epoch": 1.2332916259765625e-05,
      "model_forward_time": 2.3875608444213867,
      "step": 86932
    },
    {
      "epoch": 1.2332916259765625e-05,
      "step": 86932,
      "training_step_time": 8.013493299484253
    },
    {
      "epoch": 1.2333869934082031e-05,
      "model_forward_time": 1.761155128479004,
      "step": 86933
    },
    {
      "epoch": 1.2333869934082031e-05,
      "step": 86933,
      "training_step_time": 8.911382675170898
    },
    {
      "epoch": 1.2334823608398437e-05,
      "model_forward_time": 1.7670924663543701,
      "step": 86934
    },
    {
      "epoch": 1.2334823608398437e-05,
      "step": 86934,
      "training_step_time": 7.845626354217529
    },
    {
      "epoch": 1.2335777282714844e-05,
      "model_forward_time": 1.744258165359497,
      "step": 86935
    },
    {
      "epoch": 1.2335777282714844e-05,
      "step": 86935,
      "training_step_time": 7.737043380737305
    },
    {
      "epoch": 1.233673095703125e-05,
      "model_forward_time": 1.7579216957092285,
      "step": 86936
    },
    {
      "epoch": 1.233673095703125e-05,
      "step": 86936,
      "training_step_time": 7.717619895935059
    },
    {
      "epoch": 1.2337684631347656e-05,
      "model_forward_time": 1.764314889907837,
      "step": 86937
    },
    {
      "epoch": 1.2337684631347656e-05,
      "step": 86937,
      "training_step_time": 6.929027557373047
    },
    {
      "epoch": 1.2338638305664063e-05,
      "model_forward_time": 1.7774033546447754,
      "step": 86938
    },
    {
      "epoch": 1.2338638305664063e-05,
      "step": 86938,
      "training_step_time": 7.93112325668335
    },
    {
      "epoch": 1.2339591979980469e-05,
      "model_forward_time": 1.754150390625,
      "step": 86939
    },
    {
      "epoch": 1.2339591979980469e-05,
      "step": 86939,
      "training_step_time": 7.875471830368042
    },
    {
      "epoch": 1.2340545654296875e-05,
      "grad_norm": 0.08956456184387207,
      "learning_rate": 4.591803002750178e-07,
      "loss": 0.0543,
      "step": 86940
    },
    {
      "epoch": 1.2340545654296875e-05,
      "model_forward_time": 1.7522454261779785,
      "step": 86940
    },
    {
      "dynamics_loss_avg": 0.05049100928008556,
      "epoch": 1.2340545654296875e-05,
      "step": 86940
    },
    {
      "action_loss_avg": 0.0008312205696711317,
      "epoch": 1.2340545654296875e-05,
      "step": 86940
    },
    {
      "epoch": 1.2340545654296875e-05,
      "step": 86940,
      "training_step_time": 9.896961450576782
    },
    {
      "epoch": 1.2341499328613282e-05,
      "model_forward_time": 1.699028730392456,
      "step": 86941
    },
    {
      "epoch": 1.2341499328613282e-05,
      "step": 86941,
      "training_step_time": 8.012014627456665
    },
    {
      "epoch": 1.2342453002929688e-05,
      "model_forward_time": 1.3841688632965088,
      "step": 86942
    },
    {
      "epoch": 1.2342453002929688e-05,
      "step": 86942,
      "training_step_time": 7.894705057144165
    },
    {
      "epoch": 1.2343406677246093e-05,
      "model_forward_time": 1.76698899269104,
      "step": 86943
    },
    {
      "epoch": 1.2343406677246093e-05,
      "step": 86943,
      "training_step_time": 7.920236825942993
    },
    {
      "epoch": 1.23443603515625e-05,
      "model_forward_time": 1.7715978622436523,
      "step": 86944
    },
    {
      "epoch": 1.23443603515625e-05,
      "step": 86944,
      "training_step_time": 8.59527850151062
    },
    {
      "epoch": 1.2345314025878907e-05,
      "model_forward_time": 1.3864212036132812,
      "step": 86945
    },
    {
      "epoch": 1.2345314025878907e-05,
      "step": 86945,
      "training_step_time": 8.57182240486145
    },
    {
      "epoch": 1.2346267700195312e-05,
      "model_forward_time": 1.7640094757080078,
      "step": 86946
    },
    {
      "epoch": 1.2346267700195312e-05,
      "step": 86946,
      "training_step_time": 7.284400463104248
    },
    {
      "epoch": 1.2347221374511718e-05,
      "model_forward_time": 1.7576115131378174,
      "step": 86947
    },
    {
      "epoch": 1.2347221374511718e-05,
      "step": 86947,
      "training_step_time": 7.841727018356323
    },
    {
      "epoch": 1.2348175048828125e-05,
      "model_forward_time": 1.757432222366333,
      "step": 86948
    },
    {
      "epoch": 1.2348175048828125e-05,
      "step": 86948,
      "training_step_time": 7.910537481307983
    },
    {
      "epoch": 1.2349128723144531e-05,
      "model_forward_time": 1.7712857723236084,
      "step": 86949
    },
    {
      "epoch": 1.2349128723144531e-05,
      "step": 86949,
      "training_step_time": 9.74913239479065
    },
    {
      "epoch": 1.2350082397460937e-05,
      "grad_norm": 0.08613163232803345,
      "learning_rate": 4.58488381822077e-07,
      "loss": 0.054,
      "step": 86950
    },
    {
      "epoch": 1.2350082397460937e-05,
      "model_forward_time": 2.427546262741089,
      "step": 86950
    },
    {
      "dynamics_loss_avg": 0.04807297270745039,
      "epoch": 1.2350082397460937e-05,
      "step": 86950
    },
    {
      "action_loss_avg": 0.001187266077613458,
      "epoch": 1.2350082397460937e-05,
      "step": 86950
    },
    {
      "epoch": 1.2350082397460937e-05,
      "step": 86950,
      "training_step_time": 7.948499917984009
    },
    {
      "epoch": 1.2351036071777344e-05,
      "model_forward_time": 1.045341968536377,
      "step": 86951
    },
    {
      "epoch": 1.2351036071777344e-05,
      "step": 86951,
      "training_step_time": 7.895118474960327
    },
    {
      "epoch": 1.235198974609375e-05,
      "model_forward_time": 1.7582080364227295,
      "step": 86952
    },
    {
      "epoch": 1.235198974609375e-05,
      "step": 86952,
      "training_step_time": 8.037775039672852
    },
    {
      "epoch": 1.2352943420410156e-05,
      "model_forward_time": 1.7466020584106445,
      "step": 86953
    },
    {
      "epoch": 1.2352943420410156e-05,
      "step": 86953,
      "training_step_time": 9.297579288482666
    },
    {
      "epoch": 1.2353897094726563e-05,
      "model_forward_time": 1.7683336734771729,
      "step": 86954
    },
    {
      "epoch": 1.2353897094726563e-05,
      "step": 86954,
      "training_step_time": 7.7251386642456055
    },
    {
      "epoch": 1.2354850769042969e-05,
      "model_forward_time": 1.7606232166290283,
      "step": 86955
    },
    {
      "epoch": 1.2354850769042969e-05,
      "step": 86955,
      "training_step_time": 7.724268436431885
    },
    {
      "epoch": 1.2355804443359375e-05,
      "model_forward_time": 1.7593183517456055,
      "step": 86956
    },
    {
      "epoch": 1.2355804443359375e-05,
      "step": 86956,
      "training_step_time": 7.792831897735596
    },
    {
      "epoch": 1.2356758117675782e-05,
      "model_forward_time": 1.758558988571167,
      "step": 86957
    },
    {
      "epoch": 1.2356758117675782e-05,
      "step": 86957,
      "training_step_time": 7.793678045272827
    },
    {
      "epoch": 1.2357711791992188e-05,
      "model_forward_time": 1.751060962677002,
      "step": 86958
    },
    {
      "epoch": 1.2357711791992188e-05,
      "step": 86958,
      "training_step_time": 9.68199372291565
    },
    {
      "epoch": 1.2358665466308593e-05,
      "model_forward_time": 1.781306505203247,
      "step": 86959
    },
    {
      "epoch": 1.2358665466308593e-05,
      "step": 86959,
      "training_step_time": 9.624795198440552
    },
    {
      "epoch": 1.2359619140625e-05,
      "grad_norm": 0.08320356905460358,
      "learning_rate": 4.5779696002201944e-07,
      "loss": 0.0531,
      "step": 86960
    },
    {
      "epoch": 1.2359619140625e-05,
      "model_forward_time": 2.41162109375,
      "step": 86960
    },
    {
      "dynamics_loss_avg": 0.06097728107124567,
      "epoch": 1.2359619140625e-05,
      "step": 86960
    },
    {
      "action_loss_avg": 0.001549373849411495,
      "epoch": 1.2359619140625e-05,
      "step": 86960
    },
    {
      "epoch": 1.2359619140625e-05,
      "step": 86960,
      "training_step_time": 9.601601600646973
    },
    {
      "epoch": 1.2360572814941407e-05,
      "model_forward_time": 1.752697229385376,
      "step": 86961
    },
    {
      "epoch": 1.2360572814941407e-05,
      "step": 86961,
      "training_step_time": 8.513919591903687
    },
    {
      "epoch": 1.2361526489257812e-05,
      "model_forward_time": 1.7620110511779785,
      "step": 86962
    },
    {
      "epoch": 1.2361526489257812e-05,
      "step": 86962,
      "training_step_time": 8.387312650680542
    },
    {
      "epoch": 1.2362480163574218e-05,
      "model_forward_time": 1.767160177230835,
      "step": 86963
    },
    {
      "epoch": 1.2362480163574218e-05,
      "step": 86963,
      "training_step_time": 7.771251201629639
    },
    {
      "epoch": 1.2363433837890625e-05,
      "model_forward_time": 1.7663040161132812,
      "step": 86964
    },
    {
      "epoch": 1.2363433837890625e-05,
      "step": 86964,
      "training_step_time": 7.819144248962402
    },
    {
      "epoch": 1.2364387512207031e-05,
      "model_forward_time": 1.7584877014160156,
      "step": 86965
    },
    {
      "epoch": 1.2364387512207031e-05,
      "step": 86965,
      "training_step_time": 7.777010917663574
    },
    {
      "epoch": 1.2365341186523437e-05,
      "model_forward_time": 1.756568193435669,
      "step": 86966
    },
    {
      "epoch": 1.2365341186523437e-05,
      "step": 86966,
      "training_step_time": 7.8023295402526855
    },
    {
      "epoch": 1.2366294860839844e-05,
      "model_forward_time": 1.394282579421997,
      "step": 86967
    },
    {
      "epoch": 1.2366294860839844e-05,
      "step": 86967,
      "training_step_time": 7.789865732192993
    },
    {
      "epoch": 1.236724853515625e-05,
      "model_forward_time": 1.7603387832641602,
      "step": 86968
    },
    {
      "epoch": 1.236724853515625e-05,
      "step": 86968,
      "training_step_time": 9.722485780715942
    },
    {
      "epoch": 1.2368202209472656e-05,
      "model_forward_time": 2.389943838119507,
      "step": 86969
    },
    {
      "epoch": 1.2368202209472656e-05,
      "step": 86969,
      "training_step_time": 8.918297529220581
    },
    {
      "epoch": 1.2369155883789063e-05,
      "grad_norm": 0.09612308442592621,
      "learning_rate": 4.5710603495045637e-07,
      "loss": 0.0545,
      "step": 86970
    },
    {
      "epoch": 1.2369155883789063e-05,
      "model_forward_time": 1.8213632106781006,
      "step": 86970
    },
    {
      "dynamics_loss_avg": 0.05411764532327652,
      "epoch": 1.2369155883789063e-05,
      "step": 86970
    },
    {
      "action_loss_avg": 0.0011174624029081314,
      "epoch": 1.2369155883789063e-05,
      "step": 86970
    },
    {
      "epoch": 1.2369155883789063e-05,
      "step": 86970,
      "training_step_time": 9.021850109100342
    },
    {
      "epoch": 1.2370109558105469e-05,
      "model_forward_time": 1.7505569458007812,
      "step": 86971
    },
    {
      "epoch": 1.2370109558105469e-05,
      "step": 86971,
      "training_step_time": 8.523244857788086
    },
    {
      "epoch": 1.2371063232421875e-05,
      "model_forward_time": 1.7464396953582764,
      "step": 86972
    },
    {
      "epoch": 1.2371063232421875e-05,
      "step": 86972,
      "training_step_time": 7.694849014282227
    },
    {
      "epoch": 1.2372016906738282e-05,
      "model_forward_time": 1.746021032333374,
      "step": 86973
    },
    {
      "epoch": 1.2372016906738282e-05,
      "step": 86973,
      "training_step_time": 7.7423646450042725
    },
    {
      "epoch": 1.2372970581054688e-05,
      "model_forward_time": 1.7579751014709473,
      "step": 86974
    },
    {
      "epoch": 1.2372970581054688e-05,
      "step": 86974,
      "training_step_time": 7.127053499221802
    },
    {
      "epoch": 1.2373924255371093e-05,
      "model_forward_time": 1.7685372829437256,
      "step": 86975
    },
    {
      "epoch": 1.2373924255371093e-05,
      "step": 86975,
      "training_step_time": 7.792333602905273
    },
    {
      "epoch": 1.23748779296875e-05,
      "model_forward_time": 1.7641627788543701,
      "step": 86976
    },
    {
      "epoch": 1.23748779296875e-05,
      "step": 86976,
      "training_step_time": 9.389407873153687
    },
    {
      "epoch": 1.2375831604003907e-05,
      "model_forward_time": 1.765099287033081,
      "step": 86977
    },
    {
      "epoch": 1.2375831604003907e-05,
      "step": 86977,
      "training_step_time": 8.63520359992981
    },
    {
      "epoch": 1.2376785278320312e-05,
      "model_forward_time": 1.845332145690918,
      "step": 86978
    },
    {
      "epoch": 1.2376785278320312e-05,
      "step": 86978,
      "training_step_time": 8.700506210327148
    },
    {
      "epoch": 1.2377738952636718e-05,
      "model_forward_time": 2.422020435333252,
      "step": 86979
    },
    {
      "epoch": 1.2377738952636718e-05,
      "step": 86979,
      "training_step_time": 9.49109435081482
    },
    {
      "epoch": 1.2378692626953125e-05,
      "grad_norm": 0.08581197261810303,
      "learning_rate": 4.5641560668294784e-07,
      "loss": 0.0546,
      "step": 86980
    },
    {
      "epoch": 1.2378692626953125e-05,
      "model_forward_time": 1.7523961067199707,
      "step": 86980
    },
    {
      "dynamics_loss_avg": 0.04452903158962727,
      "epoch": 1.2378692626953125e-05,
      "step": 86980
    },
    {
      "action_loss_avg": 0.0007435633451677859,
      "epoch": 1.2378692626953125e-05,
      "step": 86980
    },
    {
      "epoch": 1.2378692626953125e-05,
      "step": 86980,
      "training_step_time": 8.801390171051025
    },
    {
      "epoch": 1.2379646301269531e-05,
      "model_forward_time": 1.7587776184082031,
      "step": 86981
    },
    {
      "epoch": 1.2379646301269531e-05,
      "step": 86981,
      "training_step_time": 8.765788555145264
    },
    {
      "epoch": 1.2380599975585937e-05,
      "model_forward_time": 1.7725505828857422,
      "step": 86982
    },
    {
      "epoch": 1.2380599975585937e-05,
      "step": 86982,
      "training_step_time": 7.6969077587127686
    },
    {
      "epoch": 1.2381553649902344e-05,
      "model_forward_time": 1.752201795578003,
      "step": 86983
    },
    {
      "epoch": 1.2381553649902344e-05,
      "step": 86983,
      "training_step_time": 7.808342933654785
    },
    {
      "epoch": 1.238250732421875e-05,
      "model_forward_time": 1.7533800601959229,
      "step": 86984
    },
    {
      "epoch": 1.238250732421875e-05,
      "step": 86984,
      "training_step_time": 7.84414005279541
    },
    {
      "epoch": 1.2383460998535156e-05,
      "model_forward_time": 1.769881248474121,
      "step": 86985
    },
    {
      "epoch": 1.2383460998535156e-05,
      "step": 86985,
      "training_step_time": 8.170763969421387
    },
    {
      "epoch": 1.2384414672851563e-05,
      "model_forward_time": 1.3934924602508545,
      "step": 86986
    },
    {
      "epoch": 1.2384414672851563e-05,
      "step": 86986,
      "training_step_time": 7.789520740509033
    },
    {
      "epoch": 1.2385368347167969e-05,
      "model_forward_time": 1.810504674911499,
      "step": 86987
    },
    {
      "epoch": 1.2385368347167969e-05,
      "step": 86987,
      "training_step_time": 9.001065492630005
    },
    {
      "epoch": 1.2386322021484375e-05,
      "model_forward_time": 2.4060590267181396,
      "step": 86988
    },
    {
      "epoch": 1.2386322021484375e-05,
      "step": 86988,
      "training_step_time": 9.060703992843628
    },
    {
      "epoch": 1.2387275695800782e-05,
      "model_forward_time": 1.7518267631530762,
      "step": 86989
    },
    {
      "epoch": 1.2387275695800782e-05,
      "step": 86989,
      "training_step_time": 7.76530122756958
    },
    {
      "epoch": 1.2388229370117188e-05,
      "grad_norm": 0.09359951317310333,
      "learning_rate": 4.557256752949957e-07,
      "loss": 0.0547,
      "step": 86990
    },
    {
      "epoch": 1.2388229370117188e-05,
      "model_forward_time": 1.7586705684661865,
      "step": 86990
    },
    {
      "dynamics_loss_avg": 0.0505703923292458,
      "epoch": 1.2388229370117188e-05,
      "step": 86990
    },
    {
      "action_loss_avg": 0.0011889331100974232,
      "epoch": 1.2388229370117188e-05,
      "step": 86990
    },
    {
      "epoch": 1.2388229370117188e-05,
      "step": 86990,
      "training_step_time": 7.969123125076294
    },
    {
      "epoch": 1.2389183044433593e-05,
      "model_forward_time": 1.7735791206359863,
      "step": 86991
    },
    {
      "epoch": 1.2389183044433593e-05,
      "step": 86991,
      "training_step_time": 8.66485333442688
    },
    {
      "epoch": 1.239013671875e-05,
      "model_forward_time": 1.772491455078125,
      "step": 86992
    },
    {
      "epoch": 1.239013671875e-05,
      "step": 86992,
      "training_step_time": 7.878118515014648
    },
    {
      "epoch": 1.2391090393066407e-05,
      "model_forward_time": 1.7822234630584717,
      "step": 86993
    },
    {
      "epoch": 1.2391090393066407e-05,
      "step": 86993,
      "training_step_time": 7.709108352661133
    },
    {
      "epoch": 1.2392044067382812e-05,
      "model_forward_time": 1.7702622413635254,
      "step": 86994
    },
    {
      "epoch": 1.2392044067382812e-05,
      "step": 86994,
      "training_step_time": 8.494114875793457
    },
    {
      "epoch": 1.2392997741699218e-05,
      "model_forward_time": 1.7698440551757812,
      "step": 86995
    },
    {
      "epoch": 1.2392997741699218e-05,
      "step": 86995,
      "training_step_time": 7.818485498428345
    },
    {
      "epoch": 1.2393951416015625e-05,
      "model_forward_time": 1.8238835334777832,
      "step": 86996
    },
    {
      "epoch": 1.2393951416015625e-05,
      "step": 86996,
      "training_step_time": 9.518430948257446
    },
    {
      "epoch": 1.2394905090332031e-05,
      "model_forward_time": 2.424579381942749,
      "step": 86997
    },
    {
      "epoch": 1.2394905090332031e-05,
      "step": 86997,
      "training_step_time": 8.83327579498291
    },
    {
      "epoch": 1.2395858764648437e-05,
      "model_forward_time": 1.8204731941223145,
      "step": 86998
    },
    {
      "epoch": 1.2395858764648437e-05,
      "step": 86998,
      "training_step_time": 8.07878065109253
    },
    {
      "epoch": 1.2396812438964844e-05,
      "model_forward_time": 1.816422939300537,
      "step": 86999
    },
    {
      "epoch": 1.2396812438964844e-05,
      "step": 86999,
      "training_step_time": 8.469234704971313
    },
    {
      "epoch": 1.239776611328125e-05,
      "grad_norm": 0.09589570760726929,
      "learning_rate": 4.5503624086205343e-07,
      "loss": 0.053,
      "step": 87000
    },
    {
      "epoch": 1.239776611328125e-05,
      "model_forward_time": 1.8506765365600586,
      "step": 87000
    },
    {
      "dynamics_loss_avg": 0.04375040028244257,
      "epoch": 1.239776611328125e-05,
      "step": 87000
    },
    {
      "action_loss_avg": 0.0010448335116961971,
      "epoch": 1.239776611328125e-05,
      "step": 87000
    },
    {
      "epoch": 1.239776611328125e-05,
      "step": 87000,
      "training_step_time": 6.745980501174927
    },
    {
      "epoch": 1.2398719787597656e-05,
      "model_forward_time": 1.7627511024475098,
      "step": 87001
    },
    {
      "epoch": 1.2398719787597656e-05,
      "step": 87001,
      "training_step_time": 7.814985752105713
    },
    {
      "epoch": 1.2399673461914063e-05,
      "model_forward_time": 1.7613842487335205,
      "step": 87002
    },
    {
      "epoch": 1.2399673461914063e-05,
      "step": 87002,
      "training_step_time": 7.4154744148254395
    },
    {
      "epoch": 1.2400627136230469e-05,
      "model_forward_time": 1.7610814571380615,
      "step": 87003
    },
    {
      "epoch": 1.2400627136230469e-05,
      "step": 87003,
      "training_step_time": 8.422628402709961
    },
    {
      "epoch": 1.2401580810546875e-05,
      "model_forward_time": 1.7710344791412354,
      "step": 87004
    },
    {
      "epoch": 1.2401580810546875e-05,
      "step": 87004,
      "training_step_time": 8.570250034332275
    },
    {
      "epoch": 1.2402534484863282e-05,
      "model_forward_time": 1.7572977542877197,
      "step": 87005
    },
    {
      "epoch": 1.2402534484863282e-05,
      "step": 87005,
      "training_step_time": 12.34579348564148
    },
    {
      "epoch": 1.2403488159179688e-05,
      "model_forward_time": 1.751619577407837,
      "step": 87006
    },
    {
      "epoch": 1.2403488159179688e-05,
      "step": 87006,
      "training_step_time": 9.186429500579834
    },
    {
      "epoch": 1.2404441833496093e-05,
      "model_forward_time": 1.8106870651245117,
      "step": 87007
    },
    {
      "epoch": 1.2404441833496093e-05,
      "step": 87007,
      "training_step_time": 8.652405738830566
    },
    {
      "epoch": 1.24053955078125e-05,
      "model_forward_time": 1.7607603073120117,
      "step": 87008
    },
    {
      "epoch": 1.24053955078125e-05,
      "step": 87008,
      "training_step_time": 9.373347043991089
    },
    {
      "epoch": 1.2406349182128907e-05,
      "model_forward_time": 1.4088339805603027,
      "step": 87009
    },
    {
      "epoch": 1.2406349182128907e-05,
      "step": 87009,
      "training_step_time": 7.491244077682495
    },
    {
      "epoch": 1.2407302856445312e-05,
      "grad_norm": 0.0836491733789444,
      "learning_rate": 4.543473034595136e-07,
      "loss": 0.0546,
      "step": 87010
    },
    {
      "epoch": 1.2407302856445312e-05,
      "model_forward_time": 1.7578222751617432,
      "step": 87010
    },
    {
      "dynamics_loss_avg": 0.04772292654961348,
      "epoch": 1.2407302856445312e-05,
      "step": 87010
    },
    {
      "action_loss_avg": 0.0009698183712316677,
      "epoch": 1.2407302856445312e-05,
      "step": 87010
    },
    {
      "epoch": 1.2407302856445312e-05,
      "step": 87010,
      "training_step_time": 7.802248001098633
    },
    {
      "epoch": 1.2408256530761718e-05,
      "model_forward_time": 1.7709026336669922,
      "step": 87011
    },
    {
      "epoch": 1.2408256530761718e-05,
      "step": 87011,
      "training_step_time": 7.186267852783203
    },
    {
      "epoch": 1.2409210205078125e-05,
      "model_forward_time": 1.7665412425994873,
      "step": 87012
    },
    {
      "epoch": 1.2409210205078125e-05,
      "step": 87012,
      "training_step_time": 7.760019779205322
    },
    {
      "epoch": 1.2410163879394531e-05,
      "model_forward_time": 1.7783498764038086,
      "step": 87013
    },
    {
      "epoch": 1.2410163879394531e-05,
      "step": 87013,
      "training_step_time": 7.824195146560669
    },
    {
      "epoch": 1.2411117553710937e-05,
      "model_forward_time": 1.7659838199615479,
      "step": 87014
    },
    {
      "epoch": 1.2411117553710937e-05,
      "step": 87014,
      "training_step_time": 7.898637056350708
    },
    {
      "epoch": 1.2412071228027344e-05,
      "model_forward_time": 2.470449924468994,
      "step": 87015
    },
    {
      "epoch": 1.2412071228027344e-05,
      "step": 87015,
      "training_step_time": 7.883121013641357
    },
    {
      "epoch": 1.241302490234375e-05,
      "model_forward_time": 1.765594720840454,
      "step": 87016
    },
    {
      "epoch": 1.241302490234375e-05,
      "step": 87016,
      "training_step_time": 10.397207498550415
    },
    {
      "epoch": 1.2413978576660156e-05,
      "model_forward_time": 1.764909029006958,
      "step": 87017
    },
    {
      "epoch": 1.2413978576660156e-05,
      "step": 87017,
      "training_step_time": 8.765488147735596
    },
    {
      "epoch": 1.2414932250976563e-05,
      "model_forward_time": 1.7635600566864014,
      "step": 87018
    },
    {
      "epoch": 1.2414932250976563e-05,
      "step": 87018,
      "training_step_time": 7.696749925613403
    },
    {
      "epoch": 1.2415885925292969e-05,
      "model_forward_time": 1.7631189823150635,
      "step": 87019
    },
    {
      "epoch": 1.2415885925292969e-05,
      "step": 87019,
      "training_step_time": 7.724191188812256
    },
    {
      "epoch": 1.2416839599609375e-05,
      "grad_norm": 0.07863782346248627,
      "learning_rate": 4.536588631627192e-07,
      "loss": 0.0541,
      "step": 87020
    },
    {
      "epoch": 1.2416839599609375e-05,
      "model_forward_time": 1.760697364807129,
      "step": 87020
    },
    {
      "dynamics_loss_avg": 0.055647002346813676,
      "epoch": 1.2416839599609375e-05,
      "step": 87020
    },
    {
      "action_loss_avg": 0.0022244674764806405,
      "epoch": 1.2416839599609375e-05,
      "step": 87020
    },
    {
      "epoch": 1.2416839599609375e-05,
      "step": 87020,
      "training_step_time": 7.173642158508301
    },
    {
      "epoch": 1.2417793273925782e-05,
      "model_forward_time": 1.7632834911346436,
      "step": 87021
    },
    {
      "epoch": 1.2417793273925782e-05,
      "step": 87021,
      "training_step_time": 7.764834403991699
    },
    {
      "epoch": 1.2418746948242188e-05,
      "model_forward_time": 1.7695024013519287,
      "step": 87022
    },
    {
      "epoch": 1.2418746948242188e-05,
      "step": 87022,
      "training_step_time": 7.777971267700195
    },
    {
      "epoch": 1.2419700622558593e-05,
      "model_forward_time": 1.7753477096557617,
      "step": 87023
    },
    {
      "epoch": 1.2419700622558593e-05,
      "step": 87023,
      "training_step_time": 9.711671113967896
    },
    {
      "epoch": 1.2420654296875e-05,
      "model_forward_time": 2.5194594860076904,
      "step": 87024
    },
    {
      "epoch": 1.2420654296875e-05,
      "step": 87024,
      "training_step_time": 7.890727281570435
    },
    {
      "epoch": 1.2421607971191407e-05,
      "model_forward_time": 1.7678322792053223,
      "step": 87025
    },
    {
      "epoch": 1.2421607971191407e-05,
      "step": 87025,
      "training_step_time": 8.816467761993408
    },
    {
      "epoch": 1.2422561645507812e-05,
      "model_forward_time": 1.7586970329284668,
      "step": 87026
    },
    {
      "epoch": 1.2422561645507812e-05,
      "step": 87026,
      "training_step_time": 7.718561887741089
    },
    {
      "epoch": 1.2423515319824218e-05,
      "model_forward_time": 1.7647974491119385,
      "step": 87027
    },
    {
      "epoch": 1.2423515319824218e-05,
      "step": 87027,
      "training_step_time": 7.905547857284546
    },
    {
      "epoch": 1.2424468994140625e-05,
      "model_forward_time": 1.763427734375,
      "step": 87028
    },
    {
      "epoch": 1.2424468994140625e-05,
      "step": 87028,
      "training_step_time": 8.47886347770691
    },
    {
      "epoch": 1.2425422668457031e-05,
      "model_forward_time": 1.7349367141723633,
      "step": 87029
    },
    {
      "epoch": 1.2425422668457031e-05,
      "step": 87029,
      "training_step_time": 7.1960389614105225
    },
    {
      "epoch": 1.2426376342773437e-05,
      "grad_norm": 0.0746460109949112,
      "learning_rate": 4.5297092004695566e-07,
      "loss": 0.0539,
      "step": 87030
    },
    {
      "epoch": 1.2426376342773437e-05,
      "model_forward_time": 1.75766921043396,
      "step": 87030
    },
    {
      "dynamics_loss_avg": 0.04403497353196144,
      "epoch": 1.2426376342773437e-05,
      "step": 87030
    },
    {
      "action_loss_avg": 0.000845866950112395,
      "epoch": 1.2426376342773437e-05,
      "step": 87030
    },
    {
      "epoch": 1.2426376342773437e-05,
      "step": 87030,
      "training_step_time": 7.723036766052246
    },
    {
      "epoch": 1.2427330017089844e-05,
      "model_forward_time": 1.759049415588379,
      "step": 87031
    },
    {
      "epoch": 1.2427330017089844e-05,
      "step": 87031,
      "training_step_time": 7.80835747718811
    },
    {
      "epoch": 1.242828369140625e-05,
      "model_forward_time": 1.7614636421203613,
      "step": 87032
    },
    {
      "epoch": 1.242828369140625e-05,
      "step": 87032,
      "training_step_time": 9.415008068084717
    },
    {
      "epoch": 1.2429237365722656e-05,
      "model_forward_time": 2.4802849292755127,
      "step": 87033
    },
    {
      "epoch": 1.2429237365722656e-05,
      "step": 87033,
      "training_step_time": 7.892452955245972
    },
    {
      "epoch": 1.2430191040039063e-05,
      "model_forward_time": 1.7492411136627197,
      "step": 87034
    },
    {
      "epoch": 1.2430191040039063e-05,
      "step": 87034,
      "training_step_time": 8.881418466567993
    },
    {
      "epoch": 1.2431144714355469e-05,
      "model_forward_time": 1.7656702995300293,
      "step": 87035
    },
    {
      "epoch": 1.2431144714355469e-05,
      "step": 87035,
      "training_step_time": 9.404374122619629
    },
    {
      "epoch": 1.2432098388671875e-05,
      "model_forward_time": 1.7497804164886475,
      "step": 87036
    },
    {
      "epoch": 1.2432098388671875e-05,
      "step": 87036,
      "training_step_time": 8.719612836837769
    },
    {
      "epoch": 1.2433052062988282e-05,
      "model_forward_time": 1.7641324996948242,
      "step": 87037
    },
    {
      "epoch": 1.2433052062988282e-05,
      "step": 87037,
      "training_step_time": 7.709916830062866
    },
    {
      "epoch": 1.2434005737304688e-05,
      "model_forward_time": 1.7585949897766113,
      "step": 87038
    },
    {
      "epoch": 1.2434005737304688e-05,
      "step": 87038,
      "training_step_time": 7.782073020935059
    },
    {
      "epoch": 1.2434959411621093e-05,
      "model_forward_time": 1.7512505054473877,
      "step": 87039
    },
    {
      "epoch": 1.2434959411621093e-05,
      "step": 87039,
      "training_step_time": 8.775988340377808
    },
    {
      "epoch": 1.24359130859375e-05,
      "grad_norm": 0.09115981310606003,
      "learning_rate": 4.5228347418745546e-07,
      "loss": 0.0552,
      "step": 87040
    },
    {
      "epoch": 1.24359130859375e-05,
      "model_forward_time": 1.7503526210784912,
      "step": 87040
    },
    {
      "dynamics_loss_avg": 0.03621985614299774,
      "epoch": 1.24359130859375e-05,
      "step": 87040
    },
    {
      "action_loss_avg": 0.0008621691871667281,
      "epoch": 1.24359130859375e-05,
      "step": 87040
    },
    {
      "epoch": 1.24359130859375e-05,
      "step": 87040,
      "training_step_time": 7.767965078353882
    },
    {
      "epoch": 1.2436866760253907e-05,
      "model_forward_time": 1.7480571269989014,
      "step": 87041
    },
    {
      "epoch": 1.2436866760253907e-05,
      "step": 87041,
      "training_step_time": 8.007982015609741
    },
    {
      "epoch": 1.2437820434570312e-05,
      "model_forward_time": 2.3671681880950928,
      "step": 87042
    },
    {
      "epoch": 1.2437820434570312e-05,
      "step": 87042,
      "training_step_time": 8.653387546539307
    },
    {
      "epoch": 1.2438774108886718e-05,
      "model_forward_time": 1.5034451484680176,
      "step": 87043
    },
    {
      "epoch": 1.2438774108886718e-05,
      "step": 87043,
      "training_step_time": 8.017330169677734
    },
    {
      "epoch": 1.2439727783203125e-05,
      "model_forward_time": 1.7653191089630127,
      "step": 87044
    },
    {
      "epoch": 1.2439727783203125e-05,
      "step": 87044,
      "training_step_time": 7.844085693359375
    },
    {
      "epoch": 1.2440681457519531e-05,
      "model_forward_time": 1.7602870464324951,
      "step": 87045
    },
    {
      "epoch": 1.2440681457519531e-05,
      "step": 87045,
      "training_step_time": 8.65037727355957
    },
    {
      "epoch": 1.2441635131835937e-05,
      "model_forward_time": 1.7607593536376953,
      "step": 87046
    },
    {
      "epoch": 1.2441635131835937e-05,
      "step": 87046,
      "training_step_time": 7.571325063705444
    },
    {
      "epoch": 1.2442588806152344e-05,
      "model_forward_time": 1.7445850372314453,
      "step": 87047
    },
    {
      "epoch": 1.2442588806152344e-05,
      "step": 87047,
      "training_step_time": 8.893720626831055
    },
    {
      "epoch": 1.244354248046875e-05,
      "model_forward_time": 1.762037992477417,
      "step": 87048
    },
    {
      "epoch": 1.244354248046875e-05,
      "step": 87048,
      "training_step_time": 7.80888819694519
    },
    {
      "epoch": 1.2444496154785156e-05,
      "model_forward_time": 1.7576820850372314,
      "step": 87049
    },
    {
      "epoch": 1.2444496154785156e-05,
      "step": 87049,
      "training_step_time": 8.318287372589111
    },
    {
      "epoch": 1.2445449829101563e-05,
      "grad_norm": 0.08217199891805649,
      "learning_rate": 4.515965256593974e-07,
      "loss": 0.0557,
      "step": 87050
    },
    {
      "epoch": 1.2445449829101563e-05,
      "model_forward_time": 1.7432076930999756,
      "step": 87050
    },
    {
      "dynamics_loss_avg": 0.04714173302054405,
      "epoch": 1.2445449829101563e-05,
      "step": 87050
    },
    {
      "action_loss_avg": 0.001600272025098093,
      "epoch": 1.2445449829101563e-05,
      "step": 87050
    },
    {
      "epoch": 1.2445449829101563e-05,
      "step": 87050,
      "training_step_time": 7.846925258636475
    },
    {
      "epoch": 1.2446403503417969e-05,
      "model_forward_time": 1.761155605316162,
      "step": 87051
    },
    {
      "epoch": 1.2446403503417969e-05,
      "step": 87051,
      "training_step_time": 9.347416400909424
    },
    {
      "epoch": 1.2447357177734375e-05,
      "model_forward_time": 2.4524943828582764,
      "step": 87052
    },
    {
      "epoch": 1.2447357177734375e-05,
      "step": 87052,
      "training_step_time": 8.591379642486572
    },
    {
      "epoch": 1.2448310852050782e-05,
      "model_forward_time": 1.760340690612793,
      "step": 87053
    },
    {
      "epoch": 1.2448310852050782e-05,
      "step": 87053,
      "training_step_time": 7.784890651702881
    },
    {
      "epoch": 1.2449264526367188e-05,
      "model_forward_time": 1.7891900539398193,
      "step": 87054
    },
    {
      "epoch": 1.2449264526367188e-05,
      "step": 87054,
      "training_step_time": 7.660275220870972
    },
    {
      "epoch": 1.2450218200683593e-05,
      "model_forward_time": 1.7513189315795898,
      "step": 87055
    },
    {
      "epoch": 1.2450218200683593e-05,
      "step": 87055,
      "training_step_time": 7.855146408081055
    },
    {
      "epoch": 1.2451171875e-05,
      "model_forward_time": 1.7528445720672607,
      "step": 87056
    },
    {
      "epoch": 1.2451171875e-05,
      "step": 87056,
      "training_step_time": 7.816365718841553
    },
    {
      "epoch": 1.2452125549316407e-05,
      "model_forward_time": 1.7898237705230713,
      "step": 87057
    },
    {
      "epoch": 1.2452125549316407e-05,
      "step": 87057,
      "training_step_time": 7.798461198806763
    },
    {
      "epoch": 1.2453079223632812e-05,
      "model_forward_time": 1.7836318016052246,
      "step": 87058
    },
    {
      "epoch": 1.2453079223632812e-05,
      "step": 87058,
      "training_step_time": 8.537972688674927
    },
    {
      "epoch": 1.2454032897949218e-05,
      "model_forward_time": 1.7555174827575684,
      "step": 87059
    },
    {
      "epoch": 1.2454032897949218e-05,
      "step": 87059,
      "training_step_time": 7.875829219818115
    },
    {
      "epoch": 1.2454986572265625e-05,
      "grad_norm": 0.0803372859954834,
      "learning_rate": 4.509100745379058e-07,
      "loss": 0.0556,
      "step": 87060
    },
    {
      "epoch": 1.2454986572265625e-05,
      "model_forward_time": 1.750481128692627,
      "step": 87060
    },
    {
      "dynamics_loss_avg": 0.05530550982803106,
      "epoch": 1.2454986572265625e-05,
      "step": 87060
    },
    {
      "action_loss_avg": 0.000794976280303672,
      "epoch": 1.2454986572265625e-05,
      "step": 87060
    },
    {
      "epoch": 1.2454986572265625e-05,
      "step": 87060,
      "training_step_time": 9.508709907531738
    },
    {
      "epoch": 1.2455940246582031e-05,
      "model_forward_time": 2.4799962043762207,
      "step": 87061
    },
    {
      "epoch": 1.2455940246582031e-05,
      "step": 87061,
      "training_step_time": 7.917723655700684
    },
    {
      "epoch": 1.2456893920898437e-05,
      "model_forward_time": 1.7699565887451172,
      "step": 87062
    },
    {
      "epoch": 1.2456893920898437e-05,
      "step": 87062,
      "training_step_time": 7.8836829662323
    },
    {
      "epoch": 1.2457847595214844e-05,
      "model_forward_time": 1.7779853343963623,
      "step": 87063
    },
    {
      "epoch": 1.2457847595214844e-05,
      "step": 87063,
      "training_step_time": 8.270002126693726
    },
    {
      "epoch": 1.245880126953125e-05,
      "model_forward_time": 1.7576103210449219,
      "step": 87064
    },
    {
      "epoch": 1.245880126953125e-05,
      "step": 87064,
      "training_step_time": 7.899594306945801
    },
    {
      "epoch": 1.2459754943847656e-05,
      "model_forward_time": 1.7637534141540527,
      "step": 87065
    },
    {
      "epoch": 1.2459754943847656e-05,
      "step": 87065,
      "training_step_time": 7.840741872787476
    },
    {
      "epoch": 1.2460708618164063e-05,
      "model_forward_time": 1.7532780170440674,
      "step": 87066
    },
    {
      "epoch": 1.2460708618164063e-05,
      "step": 87066,
      "training_step_time": 7.817042589187622
    },
    {
      "epoch": 1.2461662292480469e-05,
      "model_forward_time": 1.3970251083374023,
      "step": 87067
    },
    {
      "epoch": 1.2461662292480469e-05,
      "step": 87067,
      "training_step_time": 7.83347487449646
    },
    {
      "epoch": 1.2462615966796875e-05,
      "model_forward_time": 1.7569046020507812,
      "step": 87068
    },
    {
      "epoch": 1.2462615966796875e-05,
      "step": 87068,
      "training_step_time": 7.864384412765503
    },
    {
      "epoch": 1.2463569641113282e-05,
      "model_forward_time": 1.7583692073822021,
      "step": 87069
    },
    {
      "epoch": 1.2463569641113282e-05,
      "step": 87069,
      "training_step_time": 7.869426488876343
    },
    {
      "epoch": 1.2464523315429688e-05,
      "grad_norm": 0.1230103075504303,
      "learning_rate": 4.502241208980479e-07,
      "loss": 0.056,
      "step": 87070
    },
    {
      "epoch": 1.2464523315429688e-05,
      "model_forward_time": 2.4470832347869873,
      "step": 87070
    },
    {
      "dynamics_loss_avg": 0.043387998919934036,
      "epoch": 1.2464523315429688e-05,
      "step": 87070
    },
    {
      "action_loss_avg": 0.0008868418168276549,
      "epoch": 1.2464523315429688e-05,
      "step": 87070
    },
    {
      "epoch": 1.2464523315429688e-05,
      "step": 87070,
      "training_step_time": 7.869682788848877
    },
    {
      "epoch": 1.2465476989746093e-05,
      "model_forward_time": 1.7583470344543457,
      "step": 87071
    },
    {
      "epoch": 1.2465476989746093e-05,
      "step": 87071,
      "training_step_time": 8.572413682937622
    },
    {
      "epoch": 1.24664306640625e-05,
      "model_forward_time": 1.7570345401763916,
      "step": 87072
    },
    {
      "epoch": 1.24664306640625e-05,
      "step": 87072,
      "training_step_time": 7.8120808601379395
    },
    {
      "epoch": 1.2467384338378907e-05,
      "model_forward_time": 1.7609801292419434,
      "step": 87073
    },
    {
      "epoch": 1.2467384338378907e-05,
      "step": 87073,
      "training_step_time": 7.729318380355835
    },
    {
      "epoch": 1.2468338012695312e-05,
      "model_forward_time": 1.76304292678833,
      "step": 87074
    },
    {
      "epoch": 1.2468338012695312e-05,
      "step": 87074,
      "training_step_time": 7.817490100860596
    },
    {
      "epoch": 1.2469291687011718e-05,
      "model_forward_time": 1.7632677555084229,
      "step": 87075
    },
    {
      "epoch": 1.2469291687011718e-05,
      "step": 87075,
      "training_step_time": 8.88113522529602
    },
    {
      "epoch": 1.2470245361328125e-05,
      "model_forward_time": 1.7546379566192627,
      "step": 87076
    },
    {
      "epoch": 1.2470245361328125e-05,
      "step": 87076,
      "training_step_time": 7.718588829040527
    },
    {
      "epoch": 1.2471199035644531e-05,
      "model_forward_time": 1.757645845413208,
      "step": 87077
    },
    {
      "epoch": 1.2471199035644531e-05,
      "step": 87077,
      "training_step_time": 7.847111225128174
    },
    {
      "epoch": 1.2472152709960937e-05,
      "model_forward_time": 1.4016144275665283,
      "step": 87078
    },
    {
      "epoch": 1.2472152709960937e-05,
      "step": 87078,
      "training_step_time": 7.838950872421265
    },
    {
      "epoch": 1.2473106384277344e-05,
      "model_forward_time": 2.3701906204223633,
      "step": 87079
    },
    {
      "epoch": 1.2473106384277344e-05,
      "step": 87079,
      "training_step_time": 9.582143783569336
    },
    {
      "epoch": 1.247406005859375e-05,
      "grad_norm": 0.08004788309335709,
      "learning_rate": 4.495386648148398e-07,
      "loss": 0.056,
      "step": 87080
    },
    {
      "epoch": 1.247406005859375e-05,
      "model_forward_time": 1.7618696689605713,
      "step": 87080
    },
    {
      "dynamics_loss_avg": 0.06647053863853217,
      "epoch": 1.247406005859375e-05,
      "step": 87080
    },
    {
      "action_loss_avg": 0.001813346597191412,
      "epoch": 1.247406005859375e-05,
      "step": 87080
    },
    {
      "epoch": 1.247406005859375e-05,
      "step": 87080,
      "training_step_time": 9.770859479904175
    },
    {
      "epoch": 1.2475013732910156e-05,
      "model_forward_time": 1.759108304977417,
      "step": 87081
    },
    {
      "epoch": 1.2475013732910156e-05,
      "step": 87081,
      "training_step_time": 7.808070182800293
    },
    {
      "epoch": 1.2475967407226563e-05,
      "model_forward_time": 1.7621750831604004,
      "step": 87082
    },
    {
      "epoch": 1.2475967407226563e-05,
      "step": 87082,
      "training_step_time": 7.965059757232666
    },
    {
      "epoch": 1.2476921081542969e-05,
      "model_forward_time": 1.7598822116851807,
      "step": 87083
    },
    {
      "epoch": 1.2476921081542969e-05,
      "step": 87083,
      "training_step_time": 7.7848756313323975
    },
    {
      "epoch": 1.2477874755859375e-05,
      "model_forward_time": 1.7589969635009766,
      "step": 87084
    },
    {
      "epoch": 1.2477874755859375e-05,
      "step": 87084,
      "training_step_time": 7.84671425819397
    },
    {
      "epoch": 1.2478828430175782e-05,
      "model_forward_time": 1.7597980499267578,
      "step": 87085
    },
    {
      "epoch": 1.2478828430175782e-05,
      "step": 87085,
      "training_step_time": 7.756739616394043
    },
    {
      "epoch": 1.2479782104492188e-05,
      "model_forward_time": 1.7661006450653076,
      "step": 87086
    },
    {
      "epoch": 1.2479782104492188e-05,
      "step": 87086,
      "training_step_time": 9.079021453857422
    },
    {
      "epoch": 1.2480735778808594e-05,
      "model_forward_time": 1.7586963176727295,
      "step": 87087
    },
    {
      "epoch": 1.2480735778808594e-05,
      "step": 87087,
      "training_step_time": 8.657414197921753
    },
    {
      "epoch": 1.2481689453125e-05,
      "model_forward_time": 1.7340998649597168,
      "step": 87088
    },
    {
      "epoch": 1.2481689453125e-05,
      "step": 87088,
      "training_step_time": 9.144006729125977
    },
    {
      "epoch": 1.2482643127441407e-05,
      "model_forward_time": 2.4483470916748047,
      "step": 87089
    },
    {
      "epoch": 1.2482643127441407e-05,
      "step": 87089,
      "training_step_time": 9.26593279838562
    },
    {
      "epoch": 1.2483596801757812e-05,
      "grad_norm": 0.09416769444942474,
      "learning_rate": 4.488537063632431e-07,
      "loss": 0.0566,
      "step": 87090
    },
    {
      "epoch": 1.2483596801757812e-05,
      "model_forward_time": 1.742720365524292,
      "step": 87090
    },
    {
      "dynamics_loss_avg": 0.04936448857188225,
      "epoch": 1.2483596801757812e-05,
      "step": 87090
    },
    {
      "action_loss_avg": 0.000962241915112827,
      "epoch": 1.2483596801757812e-05,
      "step": 87090
    },
    {
      "epoch": 1.2483596801757812e-05,
      "step": 87090,
      "training_step_time": 7.789330959320068
    },
    {
      "epoch": 1.2484550476074218e-05,
      "model_forward_time": 1.7482709884643555,
      "step": 87091
    },
    {
      "epoch": 1.2484550476074218e-05,
      "step": 87091,
      "training_step_time": 7.905947208404541
    },
    {
      "epoch": 1.2485504150390626e-05,
      "model_forward_time": 1.7525455951690674,
      "step": 87092
    },
    {
      "epoch": 1.2485504150390626e-05,
      "step": 87092,
      "training_step_time": 7.542853832244873
    },
    {
      "epoch": 1.2486457824707031e-05,
      "model_forward_time": 1.762683391571045,
      "step": 87093
    },
    {
      "epoch": 1.2486457824707031e-05,
      "step": 87093,
      "training_step_time": 7.842264413833618
    },
    {
      "epoch": 1.2487411499023437e-05,
      "model_forward_time": 1.7464971542358398,
      "step": 87094
    },
    {
      "epoch": 1.2487411499023437e-05,
      "step": 87094,
      "training_step_time": 7.868178129196167
    },
    {
      "epoch": 1.2488365173339844e-05,
      "model_forward_time": 1.7543792724609375,
      "step": 87095
    },
    {
      "epoch": 1.2488365173339844e-05,
      "step": 87095,
      "training_step_time": 8.464407682418823
    },
    {
      "epoch": 1.248931884765625e-05,
      "model_forward_time": 1.7570440769195557,
      "step": 87096
    },
    {
      "epoch": 1.248931884765625e-05,
      "step": 87096,
      "training_step_time": 9.66699504852295
    },
    {
      "epoch": 1.2490272521972656e-05,
      "model_forward_time": 1.7480978965759277,
      "step": 87097
    },
    {
      "epoch": 1.2490272521972656e-05,
      "step": 87097,
      "training_step_time": 8.954036712646484
    },
    {
      "epoch": 1.2491226196289063e-05,
      "model_forward_time": 2.4469830989837646,
      "step": 87098
    },
    {
      "epoch": 1.2491226196289063e-05,
      "step": 87098,
      "training_step_time": 8.438857555389404
    },
    {
      "epoch": 1.2492179870605469e-05,
      "model_forward_time": 1.7766258716583252,
      "step": 87099
    },
    {
      "epoch": 1.2492179870605469e-05,
      "step": 87099,
      "training_step_time": 7.901010513305664
    },
    {
      "epoch": 1.2493133544921875e-05,
      "grad_norm": 0.10533600300550461,
      "learning_rate": 4.4816924561816076e-07,
      "loss": 0.0564,
      "step": 87100
    },
    {
      "epoch": 1.2493133544921875e-05,
      "model_forward_time": 1.7665143013000488,
      "step": 87100
    },
    {
      "dynamics_loss_avg": 0.04720217026770115,
      "epoch": 1.2493133544921875e-05,
      "step": 87100
    },
    {
      "action_loss_avg": 0.0013356428316910751,
      "epoch": 1.2493133544921875e-05,
      "step": 87100
    },
    {
      "epoch": 1.2493133544921875e-05,
      "step": 87100,
      "training_step_time": 7.714380264282227
    },
    {
      "epoch": 1.2494087219238282e-05,
      "model_forward_time": 1.7590274810791016,
      "step": 87101
    },
    {
      "epoch": 1.2494087219238282e-05,
      "step": 87101,
      "training_step_time": 7.908920764923096
    },
    {
      "epoch": 1.2495040893554688e-05,
      "model_forward_time": 1.7595195770263672,
      "step": 87102
    },
    {
      "epoch": 1.2495040893554688e-05,
      "step": 87102,
      "training_step_time": 7.835200548171997
    },
    {
      "epoch": 1.2495994567871094e-05,
      "model_forward_time": 1.7650554180145264,
      "step": 87103
    },
    {
      "epoch": 1.2495994567871094e-05,
      "step": 87103,
      "training_step_time": 7.750808954238892
    },
    {
      "epoch": 1.24969482421875e-05,
      "model_forward_time": 1.7821686267852783,
      "step": 87104
    },
    {
      "epoch": 1.24969482421875e-05,
      "step": 87104,
      "training_step_time": 8.69613766670227
    },
    {
      "epoch": 1.2497901916503907e-05,
      "model_forward_time": 1.3912122249603271,
      "step": 87105
    },
    {
      "epoch": 1.2497901916503907e-05,
      "step": 87105,
      "training_step_time": 8.4379301071167
    },
    {
      "epoch": 1.2498855590820312e-05,
      "model_forward_time": 1.7740538120269775,
      "step": 87106
    },
    {
      "epoch": 1.2498855590820312e-05,
      "step": 87106,
      "training_step_time": 7.949230194091797
    },
    {
      "epoch": 1.2499809265136718e-05,
      "model_forward_time": 2.4302632808685303,
      "step": 87107
    },
    {
      "epoch": 1.2499809265136718e-05,
      "step": 87107,
      "training_step_time": 8.867613315582275
    },
    {
      "epoch": 1.2500762939453126e-05,
      "model_forward_time": 1.7618818283081055,
      "step": 87108
    },
    {
      "epoch": 1.2500762939453126e-05,
      "step": 87108,
      "training_step_time": 7.881596326828003
    },
    {
      "epoch": 1.2501716613769531e-05,
      "model_forward_time": 1.7707164287567139,
      "step": 87109
    },
    {
      "epoch": 1.2501716613769531e-05,
      "step": 87109,
      "training_step_time": 8.252421379089355
    },
    {
      "epoch": 1.2502670288085937e-05,
      "grad_norm": 0.24425633251667023,
      "learning_rate": 4.474852826544479e-07,
      "loss": 0.0563,
      "step": 87110
    },
    {
      "epoch": 1.2502670288085937e-05,
      "model_forward_time": 1.7590148448944092,
      "step": 87110
    },
    {
      "dynamics_loss_avg": 0.051142340153455736,
      "epoch": 1.2502670288085937e-05,
      "step": 87110
    },
    {
      "action_loss_avg": 0.001092167664319277,
      "epoch": 1.2502670288085937e-05,
      "step": 87110
    },
    {
      "epoch": 1.2502670288085937e-05,
      "step": 87110,
      "training_step_time": 7.691319942474365
    },
    {
      "epoch": 1.2503623962402344e-05,
      "model_forward_time": 1.7610256671905518,
      "step": 87111
    },
    {
      "epoch": 1.2503623962402344e-05,
      "step": 87111,
      "training_step_time": 7.84377646446228
    },
    {
      "epoch": 1.250457763671875e-05,
      "model_forward_time": 1.773524522781372,
      "step": 87112
    },
    {
      "epoch": 1.250457763671875e-05,
      "step": 87112,
      "training_step_time": 7.867482900619507
    },
    {
      "epoch": 1.2505531311035156e-05,
      "model_forward_time": 1.766449213027954,
      "step": 87113
    },
    {
      "epoch": 1.2505531311035156e-05,
      "step": 87113,
      "training_step_time": 7.859038352966309
    },
    {
      "epoch": 1.2506484985351563e-05,
      "model_forward_time": 1.7659709453582764,
      "step": 87114
    },
    {
      "epoch": 1.2506484985351563e-05,
      "step": 87114,
      "training_step_time": 7.816332101821899
    },
    {
      "epoch": 1.2507438659667969e-05,
      "model_forward_time": 1.8123414516448975,
      "step": 87115
    },
    {
      "epoch": 1.2507438659667969e-05,
      "step": 87115,
      "training_step_time": 7.968345880508423
    },
    {
      "epoch": 1.2508392333984375e-05,
      "model_forward_time": 2.4375720024108887,
      "step": 87116
    },
    {
      "epoch": 1.2508392333984375e-05,
      "step": 87116,
      "training_step_time": 8.553465127944946
    },
    {
      "epoch": 1.2509346008300782e-05,
      "model_forward_time": 1.7708022594451904,
      "step": 87117
    },
    {
      "epoch": 1.2509346008300782e-05,
      "step": 87117,
      "training_step_time": 8.830137252807617
    },
    {
      "epoch": 1.2510299682617188e-05,
      "model_forward_time": 1.7735261917114258,
      "step": 87118
    },
    {
      "epoch": 1.2510299682617188e-05,
      "step": 87118,
      "training_step_time": 8.599878549575806
    },
    {
      "epoch": 1.2511253356933594e-05,
      "model_forward_time": 1.7660717964172363,
      "step": 87119
    },
    {
      "epoch": 1.2511253356933594e-05,
      "step": 87119,
      "training_step_time": 7.919332265853882
    },
    {
      "epoch": 1.251220703125e-05,
      "grad_norm": 0.08325225114822388,
      "learning_rate": 4.4680181754689744e-07,
      "loss": 0.0564,
      "step": 87120
    },
    {
      "epoch": 1.251220703125e-05,
      "model_forward_time": 1.805936336517334,
      "step": 87120
    },
    {
      "dynamics_loss_avg": 0.058464914187788965,
      "epoch": 1.251220703125e-05,
      "step": 87120
    },
    {
      "action_loss_avg": 0.0014987771777668969,
      "epoch": 1.251220703125e-05,
      "step": 87120
    },
    {
      "epoch": 1.251220703125e-05,
      "step": 87120,
      "training_step_time": 7.907384872436523
    },
    {
      "epoch": 1.2513160705566407e-05,
      "model_forward_time": 1.7775840759277344,
      "step": 87121
    },
    {
      "epoch": 1.2513160705566407e-05,
      "step": 87121,
      "training_step_time": 7.869359493255615
    },
    {
      "epoch": 1.2514114379882812e-05,
      "model_forward_time": 1.7681186199188232,
      "step": 87122
    },
    {
      "epoch": 1.2514114379882812e-05,
      "step": 87122,
      "training_step_time": 8.674554586410522
    },
    {
      "epoch": 1.2515068054199218e-05,
      "model_forward_time": 1.7745704650878906,
      "step": 87123
    },
    {
      "epoch": 1.2515068054199218e-05,
      "step": 87123,
      "training_step_time": 7.9033591747283936
    },
    {
      "epoch": 1.2516021728515626e-05,
      "model_forward_time": 1.767052173614502,
      "step": 87124
    },
    {
      "epoch": 1.2516021728515626e-05,
      "step": 87124,
      "training_step_time": 8.793575525283813
    },
    {
      "epoch": 1.2516975402832031e-05,
      "model_forward_time": 1.8095390796661377,
      "step": 87125
    },
    {
      "epoch": 1.2516975402832031e-05,
      "step": 87125,
      "training_step_time": 8.482686519622803
    },
    {
      "epoch": 1.2517929077148437e-05,
      "model_forward_time": 1.4035840034484863,
      "step": 87126
    },
    {
      "epoch": 1.2517929077148437e-05,
      "step": 87126,
      "training_step_time": 8.489176988601685
    },
    {
      "epoch": 1.2518882751464844e-05,
      "model_forward_time": 1.7746305465698242,
      "step": 87127
    },
    {
      "epoch": 1.2518882751464844e-05,
      "step": 87127,
      "training_step_time": 7.732738256454468
    },
    {
      "epoch": 1.251983642578125e-05,
      "model_forward_time": 1.7643868923187256,
      "step": 87128
    },
    {
      "epoch": 1.251983642578125e-05,
      "step": 87128,
      "training_step_time": 7.6969170570373535
    },
    {
      "epoch": 1.2520790100097656e-05,
      "model_forward_time": 1.7890925407409668,
      "step": 87129
    },
    {
      "epoch": 1.2520790100097656e-05,
      "step": 87129,
      "training_step_time": 7.775589942932129
    },
    {
      "epoch": 1.2521743774414063e-05,
      "grad_norm": 0.11602023243904114,
      "learning_rate": 4.461188503702568e-07,
      "loss": 0.0563,
      "step": 87130
    },
    {
      "epoch": 1.2521743774414063e-05,
      "model_forward_time": 1.7550642490386963,
      "step": 87130
    },
    {
      "dynamics_loss_avg": 0.04702461482957006,
      "epoch": 1.2521743774414063e-05,
      "step": 87130
    },
    {
      "action_loss_avg": 0.0010181716963415966,
      "epoch": 1.2521743774414063e-05,
      "step": 87130
    },
    {
      "epoch": 1.2521743774414063e-05,
      "step": 87130,
      "training_step_time": 7.850761651992798
    },
    {
      "epoch": 1.2522697448730469e-05,
      "model_forward_time": 1.766890525817871,
      "step": 87131
    },
    {
      "epoch": 1.2522697448730469e-05,
      "step": 87131,
      "training_step_time": 7.708908319473267
    },
    {
      "epoch": 1.2523651123046875e-05,
      "model_forward_time": 1.7593438625335693,
      "step": 87132
    },
    {
      "epoch": 1.2523651123046875e-05,
      "step": 87132,
      "training_step_time": 7.787765979766846
    },
    {
      "epoch": 1.252460479736328e-05,
      "model_forward_time": 1.7609984874725342,
      "step": 87133
    },
    {
      "epoch": 1.252460479736328e-05,
      "step": 87133,
      "training_step_time": 7.859255075454712
    },
    {
      "epoch": 1.2525558471679688e-05,
      "model_forward_time": 1.7729616165161133,
      "step": 87134
    },
    {
      "epoch": 1.2525558471679688e-05,
      "step": 87134,
      "training_step_time": 8.80890703201294
    },
    {
      "epoch": 1.2526512145996094e-05,
      "model_forward_time": 2.408339500427246,
      "step": 87135
    },
    {
      "epoch": 1.2526512145996094e-05,
      "step": 87135,
      "training_step_time": 8.575778484344482
    },
    {
      "epoch": 1.25274658203125e-05,
      "model_forward_time": 1.757479190826416,
      "step": 87136
    },
    {
      "epoch": 1.25274658203125e-05,
      "step": 87136,
      "training_step_time": 7.7082648277282715
    },
    {
      "epoch": 1.2528419494628907e-05,
      "model_forward_time": 1.773223876953125,
      "step": 87137
    },
    {
      "epoch": 1.2528419494628907e-05,
      "step": 87137,
      "training_step_time": 8.379651308059692
    },
    {
      "epoch": 1.2529373168945312e-05,
      "model_forward_time": 1.3934779167175293,
      "step": 87138
    },
    {
      "epoch": 1.2529373168945312e-05,
      "step": 87138,
      "training_step_time": 7.796041011810303
    },
    {
      "epoch": 1.2530326843261718e-05,
      "model_forward_time": 1.7601072788238525,
      "step": 87139
    },
    {
      "epoch": 1.2530326843261718e-05,
      "step": 87139,
      "training_step_time": 7.832104921340942
    },
    {
      "epoch": 1.2531280517578126e-05,
      "grad_norm": 0.09447767585515976,
      "learning_rate": 4.454363811992107e-07,
      "loss": 0.0577,
      "step": 87140
    },
    {
      "epoch": 1.2531280517578126e-05,
      "model_forward_time": 1.7531561851501465,
      "step": 87140
    },
    {
      "dynamics_loss_avg": 0.05288205463439226,
      "epoch": 1.2531280517578126e-05,
      "step": 87140
    },
    {
      "action_loss_avg": 0.0011155432483064942,
      "epoch": 1.2531280517578126e-05,
      "step": 87140
    },
    {
      "epoch": 1.2531280517578126e-05,
      "step": 87140,
      "training_step_time": 8.592897653579712
    },
    {
      "epoch": 1.2532234191894531e-05,
      "model_forward_time": 1.3948569297790527,
      "step": 87141
    },
    {
      "epoch": 1.2532234191894531e-05,
      "step": 87141,
      "training_step_time": 8.287992715835571
    },
    {
      "epoch": 1.2533187866210937e-05,
      "model_forward_time": 1.7670583724975586,
      "step": 87142
    },
    {
      "epoch": 1.2533187866210937e-05,
      "step": 87142,
      "training_step_time": 8.804580926895142
    },
    {
      "epoch": 1.2534141540527344e-05,
      "model_forward_time": 1.7762961387634277,
      "step": 87143
    },
    {
      "epoch": 1.2534141540527344e-05,
      "step": 87143,
      "training_step_time": 10.491293668746948
    },
    {
      "epoch": 1.253509521484375e-05,
      "model_forward_time": 2.1099276542663574,
      "step": 87144
    },
    {
      "epoch": 1.253509521484375e-05,
      "step": 87144,
      "training_step_time": 8.352691173553467
    },
    {
      "epoch": 1.2536048889160156e-05,
      "model_forward_time": 1.7640690803527832,
      "step": 87145
    },
    {
      "epoch": 1.2536048889160156e-05,
      "step": 87145,
      "training_step_time": 7.842206954956055
    },
    {
      "epoch": 1.2537002563476563e-05,
      "model_forward_time": 1.7610900402069092,
      "step": 87146
    },
    {
      "epoch": 1.2537002563476563e-05,
      "step": 87146,
      "training_step_time": 7.664637565612793
    },
    {
      "epoch": 1.2537956237792969e-05,
      "model_forward_time": 1.7608115673065186,
      "step": 87147
    },
    {
      "epoch": 1.2537956237792969e-05,
      "step": 87147,
      "training_step_time": 8.552032232284546
    },
    {
      "epoch": 1.2538909912109375e-05,
      "model_forward_time": 1.7582542896270752,
      "step": 87148
    },
    {
      "epoch": 1.2538909912109375e-05,
      "step": 87148,
      "training_step_time": 7.754359245300293
    },
    {
      "epoch": 1.253986358642578e-05,
      "model_forward_time": 1.7662529945373535,
      "step": 87149
    },
    {
      "epoch": 1.253986358642578e-05,
      "step": 87149,
      "training_step_time": 7.72393798828125
    },
    {
      "epoch": 1.2540817260742188e-05,
      "grad_norm": 0.08592061698436737,
      "learning_rate": 4.447544101083956e-07,
      "loss": 0.0577,
      "step": 87150
    },
    {
      "epoch": 1.2540817260742188e-05,
      "model_forward_time": 1.7492296695709229,
      "step": 87150
    },
    {
      "dynamics_loss_avg": 0.05024875737726688,
      "epoch": 1.2540817260742188e-05,
      "step": 87150
    },
    {
      "action_loss_avg": 0.0012979097780771553,
      "epoch": 1.2540817260742188e-05,
      "step": 87150
    },
    {
      "epoch": 1.2540817260742188e-05,
      "step": 87150,
      "training_step_time": 8.823435068130493
    },
    {
      "epoch": 1.2541770935058594e-05,
      "model_forward_time": 1.7595205307006836,
      "step": 87151
    },
    {
      "epoch": 1.2541770935058594e-05,
      "step": 87151,
      "training_step_time": 8.636223077774048
    },
    {
      "epoch": 1.2542724609375e-05,
      "model_forward_time": 1.7605712413787842,
      "step": 87152
    },
    {
      "epoch": 1.2542724609375e-05,
      "step": 87152,
      "training_step_time": 7.8520612716674805
    },
    {
      "epoch": 1.2543678283691407e-05,
      "model_forward_time": 2.40779709815979,
      "step": 87153
    },
    {
      "epoch": 1.2543678283691407e-05,
      "step": 87153,
      "training_step_time": 7.851401329040527
    },
    {
      "epoch": 1.2544631958007812e-05,
      "model_forward_time": 1.7468595504760742,
      "step": 87154
    },
    {
      "epoch": 1.2544631958007812e-05,
      "step": 87154,
      "training_step_time": 8.58845567703247
    },
    {
      "epoch": 1.2545585632324218e-05,
      "model_forward_time": 1.763671875,
      "step": 87155
    },
    {
      "epoch": 1.2545585632324218e-05,
      "step": 87155,
      "training_step_time": 8.515174865722656
    },
    {
      "epoch": 1.2546539306640626e-05,
      "model_forward_time": 1.7615885734558105,
      "step": 87156
    },
    {
      "epoch": 1.2546539306640626e-05,
      "step": 87156,
      "training_step_time": 7.792660474777222
    },
    {
      "epoch": 1.2547492980957031e-05,
      "model_forward_time": 1.7535772323608398,
      "step": 87157
    },
    {
      "epoch": 1.2547492980957031e-05,
      "step": 87157,
      "training_step_time": 7.7998692989349365
    },
    {
      "epoch": 1.2548446655273437e-05,
      "model_forward_time": 1.7565655708312988,
      "step": 87158
    },
    {
      "epoch": 1.2548446655273437e-05,
      "step": 87158,
      "training_step_time": 7.7858970165252686
    },
    {
      "epoch": 1.2549400329589844e-05,
      "model_forward_time": 1.7650105953216553,
      "step": 87159
    },
    {
      "epoch": 1.2549400329589844e-05,
      "step": 87159,
      "training_step_time": 8.62199330329895
    },
    {
      "epoch": 1.255035400390625e-05,
      "grad_norm": 0.07532837986946106,
      "learning_rate": 4.440729371723884e-07,
      "loss": 0.0578,
      "step": 87160
    },
    {
      "epoch": 1.255035400390625e-05,
      "model_forward_time": 1.744028091430664,
      "step": 87160
    },
    {
      "dynamics_loss_avg": 0.06861204011365771,
      "epoch": 1.255035400390625e-05,
      "step": 87160
    },
    {
      "action_loss_avg": 0.002703379618469626,
      "epoch": 1.255035400390625e-05,
      "step": 87160
    },
    {
      "epoch": 1.255035400390625e-05,
      "step": 87160,
      "training_step_time": 7.787523508071899
    },
    {
      "epoch": 1.2551307678222656e-05,
      "model_forward_time": 1.7496163845062256,
      "step": 87161
    },
    {
      "epoch": 1.2551307678222656e-05,
      "step": 87161,
      "training_step_time": 7.924766778945923
    },
    {
      "epoch": 1.2552261352539063e-05,
      "model_forward_time": 2.441519260406494,
      "step": 87162
    },
    {
      "epoch": 1.2552261352539063e-05,
      "step": 87162,
      "training_step_time": 8.904061794281006
    },
    {
      "epoch": 1.2553215026855469e-05,
      "model_forward_time": 1.7525289058685303,
      "step": 87163
    },
    {
      "epoch": 1.2553215026855469e-05,
      "step": 87163,
      "training_step_time": 7.880245685577393
    },
    {
      "epoch": 1.2554168701171875e-05,
      "model_forward_time": 1.755732774734497,
      "step": 87164
    },
    {
      "epoch": 1.2554168701171875e-05,
      "step": 87164,
      "training_step_time": 8.200761795043945
    },
    {
      "epoch": 1.255512237548828e-05,
      "model_forward_time": 1.3791754245758057,
      "step": 87165
    },
    {
      "epoch": 1.255512237548828e-05,
      "step": 87165,
      "training_step_time": 7.719099283218384
    },
    {
      "epoch": 1.2556076049804688e-05,
      "model_forward_time": 1.7553579807281494,
      "step": 87166
    },
    {
      "epoch": 1.2556076049804688e-05,
      "step": 87166,
      "training_step_time": 7.757930040359497
    },
    {
      "epoch": 1.2557029724121094e-05,
      "model_forward_time": 1.7537715435028076,
      "step": 87167
    },
    {
      "epoch": 1.2557029724121094e-05,
      "step": 87167,
      "training_step_time": 7.837511777877808
    },
    {
      "epoch": 1.25579833984375e-05,
      "model_forward_time": 1.7587292194366455,
      "step": 87168
    },
    {
      "epoch": 1.25579833984375e-05,
      "step": 87168,
      "training_step_time": 8.464078187942505
    },
    {
      "epoch": 1.2558937072753907e-05,
      "model_forward_time": 1.75205659866333,
      "step": 87169
    },
    {
      "epoch": 1.2558937072753907e-05,
      "step": 87169,
      "training_step_time": 7.895646095275879
    },
    {
      "epoch": 1.2559890747070312e-05,
      "grad_norm": 0.09031679481267929,
      "learning_rate": 4.4339196246571626e-07,
      "loss": 0.0562,
      "step": 87170
    },
    {
      "epoch": 1.2559890747070312e-05,
      "model_forward_time": 1.7801356315612793,
      "step": 87170
    },
    {
      "dynamics_loss_avg": 0.07836928013712167,
      "epoch": 1.2559890747070312e-05,
      "step": 87170
    },
    {
      "action_loss_avg": 0.0014315618755063042,
      "epoch": 1.2559890747070312e-05,
      "step": 87170
    },
    {
      "epoch": 1.2559890747070312e-05,
      "step": 87170,
      "training_step_time": 8.325716257095337
    },
    {
      "epoch": 1.2560844421386718e-05,
      "model_forward_time": 1.7706272602081299,
      "step": 87171
    },
    {
      "epoch": 1.2560844421386718e-05,
      "step": 87171,
      "training_step_time": 7.829060077667236
    },
    {
      "epoch": 1.2561798095703126e-05,
      "model_forward_time": 2.388212203979492,
      "step": 87172
    },
    {
      "epoch": 1.2561798095703126e-05,
      "step": 87172,
      "training_step_time": 7.943235158920288
    },
    {
      "epoch": 1.2562751770019531e-05,
      "model_forward_time": 1.7540416717529297,
      "step": 87173
    },
    {
      "epoch": 1.2562751770019531e-05,
      "step": 87173,
      "training_step_time": 8.498167514801025
    },
    {
      "epoch": 1.2563705444335937e-05,
      "model_forward_time": 1.758218765258789,
      "step": 87174
    },
    {
      "epoch": 1.2563705444335937e-05,
      "step": 87174,
      "training_step_time": 7.652418851852417
    },
    {
      "epoch": 1.2564659118652344e-05,
      "model_forward_time": 1.76182222366333,
      "step": 87175
    },
    {
      "epoch": 1.2564659118652344e-05,
      "step": 87175,
      "training_step_time": 8.357449054718018
    },
    {
      "epoch": 1.256561279296875e-05,
      "model_forward_time": 1.399993658065796,
      "step": 87176
    },
    {
      "epoch": 1.256561279296875e-05,
      "step": 87176,
      "training_step_time": 7.9283061027526855
    },
    {
      "epoch": 1.2566566467285156e-05,
      "model_forward_time": 1.7623436450958252,
      "step": 87177
    },
    {
      "epoch": 1.2566566467285156e-05,
      "step": 87177,
      "training_step_time": 7.473442077636719
    },
    {
      "epoch": 1.2567520141601563e-05,
      "model_forward_time": 1.787062644958496,
      "step": 87178
    },
    {
      "epoch": 1.2567520141601563e-05,
      "step": 87178,
      "training_step_time": 9.07224988937378
    },
    {
      "epoch": 1.2568473815917969e-05,
      "model_forward_time": 1.7987923622131348,
      "step": 87179
    },
    {
      "epoch": 1.2568473815917969e-05,
      "step": 87179,
      "training_step_time": 7.837138414382935
    },
    {
      "epoch": 1.2569427490234375e-05,
      "grad_norm": 0.12114138156175613,
      "learning_rate": 4.427114860628462e-07,
      "loss": 0.0557,
      "step": 87180
    },
    {
      "epoch": 1.2569427490234375e-05,
      "model_forward_time": 1.7560765743255615,
      "step": 87180
    },
    {
      "dynamics_loss_avg": 0.05459717195481062,
      "epoch": 1.2569427490234375e-05,
      "step": 87180
    },
    {
      "action_loss_avg": 0.0005661391245666891,
      "epoch": 1.2569427490234375e-05,
      "step": 87180
    },
    {
      "epoch": 1.2569427490234375e-05,
      "step": 87180,
      "training_step_time": 8.828302383422852
    },
    {
      "epoch": 1.257038116455078e-05,
      "model_forward_time": 2.4301810264587402,
      "step": 87181
    },
    {
      "epoch": 1.257038116455078e-05,
      "step": 87181,
      "training_step_time": 7.8236706256866455
    },
    {
      "epoch": 1.2571334838867188e-05,
      "model_forward_time": 1.77290678024292,
      "step": 87182
    },
    {
      "epoch": 1.2571334838867188e-05,
      "step": 87182,
      "training_step_time": 9.532158136367798
    },
    {
      "epoch": 1.2572288513183594e-05,
      "model_forward_time": 1.7609128952026367,
      "step": 87183
    },
    {
      "epoch": 1.2572288513183594e-05,
      "step": 87183,
      "training_step_time": 8.429004430770874
    },
    {
      "epoch": 1.25732421875e-05,
      "model_forward_time": 1.7678163051605225,
      "step": 87184
    },
    {
      "epoch": 1.25732421875e-05,
      "step": 87184,
      "training_step_time": 7.841485261917114
    },
    {
      "epoch": 1.2574195861816407e-05,
      "model_forward_time": 1.7624294757843018,
      "step": 87185
    },
    {
      "epoch": 1.2574195861816407e-05,
      "step": 87185,
      "training_step_time": 7.787004232406616
    },
    {
      "epoch": 1.2575149536132812e-05,
      "model_forward_time": 1.7676169872283936,
      "step": 87186
    },
    {
      "epoch": 1.2575149536132812e-05,
      "step": 87186,
      "training_step_time": 7.419828653335571
    },
    {
      "epoch": 1.2576103210449218e-05,
      "model_forward_time": 1.7990264892578125,
      "step": 87187
    },
    {
      "epoch": 1.2576103210449218e-05,
      "step": 87187,
      "training_step_time": 7.83302903175354
    },
    {
      "epoch": 1.2577056884765626e-05,
      "model_forward_time": 1.758368968963623,
      "step": 87188
    },
    {
      "epoch": 1.2577056884765626e-05,
      "step": 87188,
      "training_step_time": 8.954568147659302
    },
    {
      "epoch": 1.2578010559082031e-05,
      "model_forward_time": 1.7602016925811768,
      "step": 87189
    },
    {
      "epoch": 1.2578010559082031e-05,
      "step": 87189,
      "training_step_time": 10.166830778121948
    },
    {
      "epoch": 1.2578964233398437e-05,
      "grad_norm": 0.0851266086101532,
      "learning_rate": 4.420315080381982e-07,
      "loss": 0.0556,
      "step": 87190
    },
    {
      "epoch": 1.2578964233398437e-05,
      "model_forward_time": 2.4166574478149414,
      "step": 87190
    },
    {
      "dynamics_loss_avg": 0.05866862554103136,
      "epoch": 1.2578964233398437e-05,
      "step": 87190
    },
    {
      "action_loss_avg": 0.001440894631377887,
      "epoch": 1.2578964233398437e-05,
      "step": 87190
    },
    {
      "epoch": 1.2578964233398437e-05,
      "step": 87190,
      "training_step_time": 8.662446975708008
    },
    {
      "epoch": 1.2579917907714844e-05,
      "model_forward_time": 1.7579541206359863,
      "step": 87191
    },
    {
      "epoch": 1.2579917907714844e-05,
      "step": 87191,
      "training_step_time": 9.751865148544312
    },
    {
      "epoch": 1.258087158203125e-05,
      "model_forward_time": 1.7601561546325684,
      "step": 87192
    },
    {
      "epoch": 1.258087158203125e-05,
      "step": 87192,
      "training_step_time": 8.560258388519287
    },
    {
      "epoch": 1.2581825256347656e-05,
      "model_forward_time": 1.3967208862304688,
      "step": 87193
    },
    {
      "epoch": 1.2581825256347656e-05,
      "step": 87193,
      "training_step_time": 7.783057451248169
    },
    {
      "epoch": 1.2582778930664063e-05,
      "model_forward_time": 1.7836108207702637,
      "step": 87194
    },
    {
      "epoch": 1.2582778930664063e-05,
      "step": 87194,
      "training_step_time": 7.81688380241394
    },
    {
      "epoch": 1.2583732604980469e-05,
      "model_forward_time": 1.7598633766174316,
      "step": 87195
    },
    {
      "epoch": 1.2583732604980469e-05,
      "step": 87195,
      "training_step_time": 8.471946716308594
    },
    {
      "epoch": 1.2584686279296875e-05,
      "model_forward_time": 1.7631268501281738,
      "step": 87196
    },
    {
      "epoch": 1.2584686279296875e-05,
      "step": 87196,
      "training_step_time": 8.438318729400635
    },
    {
      "epoch": 1.258563995361328e-05,
      "model_forward_time": 1.760744333267212,
      "step": 87197
    },
    {
      "epoch": 1.258563995361328e-05,
      "step": 87197,
      "training_step_time": 8.627781391143799
    },
    {
      "epoch": 1.2586593627929688e-05,
      "model_forward_time": 1.7548139095306396,
      "step": 87198
    },
    {
      "epoch": 1.2586593627929688e-05,
      "step": 87198,
      "training_step_time": 8.694923400878906
    },
    {
      "epoch": 1.2587547302246094e-05,
      "model_forward_time": 1.7655375003814697,
      "step": 87199
    },
    {
      "epoch": 1.2587547302246094e-05,
      "step": 87199,
      "training_step_time": 9.285284996032715
    },
    {
      "epoch": 1.25885009765625e-05,
      "grad_norm": 0.08786769211292267,
      "learning_rate": 4.4135202846613045e-07,
      "loss": 0.0564,
      "step": 87200
    },
    {
      "epoch": 1.25885009765625e-05,
      "model_forward_time": 1.8695883750915527,
      "step": 87200
    },
    {
      "dynamics_loss_avg": 0.05811056308448315,
      "epoch": 1.25885009765625e-05,
      "step": 87200
    },
    {
      "action_loss_avg": 0.0015012450487120077,
      "epoch": 1.25885009765625e-05,
      "step": 87200
    },
    {
      "epoch": 1.25885009765625e-05,
      "step": 87200,
      "training_step_time": 8.413423538208008
    },
    {
      "epoch": 1.2589454650878907e-05,
      "model_forward_time": 1.7601637840270996,
      "step": 87201
    },
    {
      "epoch": 1.2589454650878907e-05,
      "step": 87201,
      "training_step_time": 7.783546209335327
    },
    {
      "epoch": 1.2590408325195312e-05,
      "model_forward_time": 1.7457966804504395,
      "step": 87202
    },
    {
      "epoch": 1.2590408325195312e-05,
      "step": 87202,
      "training_step_time": 8.724020004272461
    },
    {
      "epoch": 1.2591361999511718e-05,
      "model_forward_time": 1.767547845840454,
      "step": 87203
    },
    {
      "epoch": 1.2591361999511718e-05,
      "step": 87203,
      "training_step_time": 7.864089488983154
    },
    {
      "epoch": 1.2592315673828126e-05,
      "model_forward_time": 1.7602524757385254,
      "step": 87204
    },
    {
      "epoch": 1.2592315673828126e-05,
      "step": 87204,
      "training_step_time": 7.867769956588745
    },
    {
      "epoch": 1.2593269348144531e-05,
      "model_forward_time": 1.763451099395752,
      "step": 87205
    },
    {
      "epoch": 1.2593269348144531e-05,
      "step": 87205,
      "training_step_time": 7.8424437046051025
    },
    {
      "epoch": 1.2594223022460937e-05,
      "model_forward_time": 1.7553296089172363,
      "step": 87206
    },
    {
      "epoch": 1.2594223022460937e-05,
      "step": 87206,
      "training_step_time": 8.71152114868164
    },
    {
      "epoch": 1.2595176696777344e-05,
      "model_forward_time": 1.764672040939331,
      "step": 87207
    },
    {
      "epoch": 1.2595176696777344e-05,
      "step": 87207,
      "training_step_time": 8.971760749816895
    },
    {
      "epoch": 1.259613037109375e-05,
      "model_forward_time": 1.761526107788086,
      "step": 87208
    },
    {
      "epoch": 1.259613037109375e-05,
      "step": 87208,
      "training_step_time": 7.8141045570373535
    },
    {
      "epoch": 1.2597084045410156e-05,
      "model_forward_time": 2.4849870204925537,
      "step": 87209
    },
    {
      "epoch": 1.2597084045410156e-05,
      "step": 87209,
      "training_step_time": 9.047900438308716
    },
    {
      "epoch": 1.2598037719726563e-05,
      "grad_norm": 0.08533846586942673,
      "learning_rate": 4.4067304742095253e-07,
      "loss": 0.0561,
      "step": 87210
    },
    {
      "epoch": 1.2598037719726563e-05,
      "model_forward_time": 1.746976613998413,
      "step": 87210
    },
    {
      "dynamics_loss_avg": 0.05015568705275655,
      "epoch": 1.2598037719726563e-05,
      "step": 87210
    },
    {
      "action_loss_avg": 0.0011291562113910913,
      "epoch": 1.2598037719726563e-05,
      "step": 87210
    },
    {
      "epoch": 1.2598037719726563e-05,
      "step": 87210,
      "training_step_time": 8.45586371421814
    },
    {
      "epoch": 1.2598991394042969e-05,
      "model_forward_time": 1.7723534107208252,
      "step": 87211
    },
    {
      "epoch": 1.2598991394042969e-05,
      "step": 87211,
      "training_step_time": 8.657151460647583
    },
    {
      "epoch": 1.2599945068359375e-05,
      "model_forward_time": 1.7594473361968994,
      "step": 87212
    },
    {
      "epoch": 1.2599945068359375e-05,
      "step": 87212,
      "training_step_time": 8.885533809661865
    },
    {
      "epoch": 1.260089874267578e-05,
      "model_forward_time": 1.7756564617156982,
      "step": 87213
    },
    {
      "epoch": 1.260089874267578e-05,
      "step": 87213,
      "training_step_time": 7.548643112182617
    },
    {
      "epoch": 1.2601852416992188e-05,
      "model_forward_time": 1.7731118202209473,
      "step": 87214
    },
    {
      "epoch": 1.2601852416992188e-05,
      "step": 87214,
      "training_step_time": 8.708093166351318
    },
    {
      "epoch": 1.2602806091308594e-05,
      "model_forward_time": 1.7724182605743408,
      "step": 87215
    },
    {
      "epoch": 1.2602806091308594e-05,
      "step": 87215,
      "training_step_time": 7.808551549911499
    },
    {
      "epoch": 1.2603759765625e-05,
      "model_forward_time": 1.7562956809997559,
      "step": 87216
    },
    {
      "epoch": 1.2603759765625e-05,
      "step": 87216,
      "training_step_time": 8.824029207229614
    },
    {
      "epoch": 1.2604713439941407e-05,
      "model_forward_time": 1.7563774585723877,
      "step": 87217
    },
    {
      "epoch": 1.2604713439941407e-05,
      "step": 87217,
      "training_step_time": 7.807077646255493
    },
    {
      "epoch": 1.2605667114257812e-05,
      "model_forward_time": 2.4823508262634277,
      "step": 87218
    },
    {
      "epoch": 1.2605667114257812e-05,
      "step": 87218,
      "training_step_time": 7.895280599594116
    },
    {
      "epoch": 1.2606620788574218e-05,
      "model_forward_time": 1.773796558380127,
      "step": 87219
    },
    {
      "epoch": 1.2606620788574218e-05,
      "step": 87219,
      "training_step_time": 8.68876028060913
    },
    {
      "epoch": 1.2607574462890626e-05,
      "grad_norm": 0.6546337604522705,
      "learning_rate": 4.3999456497691385e-07,
      "loss": 0.0556,
      "step": 87220
    },
    {
      "epoch": 1.2607574462890626e-05,
      "model_forward_time": 1.7584824562072754,
      "step": 87220
    },
    {
      "dynamics_loss_avg": 0.049975478136911985,
      "epoch": 1.2607574462890626e-05,
      "step": 87220
    },
    {
      "action_loss_avg": 0.0007829895330360159,
      "epoch": 1.2607574462890626e-05,
      "step": 87220
    },
    {
      "epoch": 1.2607574462890626e-05,
      "step": 87220,
      "training_step_time": 7.708369970321655
    },
    {
      "epoch": 1.2608528137207031e-05,
      "model_forward_time": 1.7619845867156982,
      "step": 87221
    },
    {
      "epoch": 1.2608528137207031e-05,
      "step": 87221,
      "training_step_time": 7.9832236766815186
    },
    {
      "epoch": 1.2609481811523437e-05,
      "model_forward_time": 1.7666029930114746,
      "step": 87222
    },
    {
      "epoch": 1.2609481811523437e-05,
      "step": 87222,
      "training_step_time": 7.946772575378418
    },
    {
      "epoch": 1.2610435485839844e-05,
      "model_forward_time": 1.766066074371338,
      "step": 87223
    },
    {
      "epoch": 1.2610435485839844e-05,
      "step": 87223,
      "training_step_time": 8.803951740264893
    },
    {
      "epoch": 1.261138916015625e-05,
      "model_forward_time": 1.758984088897705,
      "step": 87224
    },
    {
      "epoch": 1.261138916015625e-05,
      "step": 87224,
      "training_step_time": 12.248481750488281
    },
    {
      "epoch": 1.2612342834472656e-05,
      "model_forward_time": 1.7646923065185547,
      "step": 87225
    },
    {
      "epoch": 1.2612342834472656e-05,
      "step": 87225,
      "training_step_time": 6.904869318008423
    },
    {
      "epoch": 1.2613296508789063e-05,
      "model_forward_time": 1.7640080451965332,
      "step": 87226
    },
    {
      "epoch": 1.2613296508789063e-05,
      "step": 87226,
      "training_step_time": 7.5011961460113525
    },
    {
      "epoch": 1.2614250183105469e-05,
      "model_forward_time": 1.778916835784912,
      "step": 87227
    },
    {
      "epoch": 1.2614250183105469e-05,
      "step": 87227,
      "training_step_time": 8.19035792350769
    },
    {
      "epoch": 1.2615203857421875e-05,
      "model_forward_time": 1.7579567432403564,
      "step": 87228
    },
    {
      "epoch": 1.2615203857421875e-05,
      "step": 87228,
      "training_step_time": 7.988481760025024
    },
    {
      "epoch": 1.261615753173828e-05,
      "model_forward_time": 1.7603962421417236,
      "step": 87229
    },
    {
      "epoch": 1.261615753173828e-05,
      "step": 87229,
      "training_step_time": 6.872220277786255
    },
    {
      "epoch": 1.2617111206054688e-05,
      "grad_norm": 0.09453118592500687,
      "learning_rate": 4.393165812082151e-07,
      "loss": 0.055,
      "step": 87230
    },
    {
      "epoch": 1.2617111206054688e-05,
      "model_forward_time": 1.7862756252288818,
      "step": 87230
    },
    {
      "dynamics_loss_avg": 0.05590478200465441,
      "epoch": 1.2617111206054688e-05,
      "step": 87230
    },
    {
      "action_loss_avg": 0.0010813554137712345,
      "epoch": 1.2617111206054688e-05,
      "step": 87230
    },
    {
      "epoch": 1.2617111206054688e-05,
      "step": 87230,
      "training_step_time": 6.889686584472656
    },
    {
      "epoch": 1.2618064880371094e-05,
      "model_forward_time": 1.7648417949676514,
      "step": 87231
    },
    {
      "epoch": 1.2618064880371094e-05,
      "step": 87231,
      "training_step_time": 6.9753992557525635
    },
    {
      "epoch": 1.26190185546875e-05,
      "model_forward_time": 1.3883631229400635,
      "step": 87232
    },
    {
      "epoch": 1.26190185546875e-05,
      "step": 87232,
      "training_step_time": 6.957540512084961
    },
    {
      "epoch": 1.2619972229003907e-05,
      "model_forward_time": 1.762035846710205,
      "step": 87233
    },
    {
      "epoch": 1.2619972229003907e-05,
      "step": 87233,
      "training_step_time": 6.902314901351929
    },
    {
      "epoch": 1.2620925903320312e-05,
      "model_forward_time": 2.4644224643707275,
      "step": 87234
    },
    {
      "epoch": 1.2620925903320312e-05,
      "step": 87234,
      "training_step_time": 7.8859100341796875
    },
    {
      "epoch": 1.2621879577636718e-05,
      "model_forward_time": 1.7739062309265137,
      "step": 87235
    },
    {
      "epoch": 1.2621879577636718e-05,
      "step": 87235,
      "training_step_time": 6.963484764099121
    },
    {
      "epoch": 1.2622833251953126e-05,
      "model_forward_time": 1.7637732028961182,
      "step": 87236
    },
    {
      "epoch": 1.2622833251953126e-05,
      "step": 87236,
      "training_step_time": 7.826235294342041
    },
    {
      "epoch": 1.2623786926269531e-05,
      "model_forward_time": 1.7609972953796387,
      "step": 87237
    },
    {
      "epoch": 1.2623786926269531e-05,
      "step": 87237,
      "training_step_time": 7.79522180557251
    },
    {
      "epoch": 1.2624740600585937e-05,
      "model_forward_time": 1.757789134979248,
      "step": 87238
    },
    {
      "epoch": 1.2624740600585937e-05,
      "step": 87238,
      "training_step_time": 7.797299146652222
    },
    {
      "epoch": 1.2625694274902344e-05,
      "model_forward_time": 1.7804126739501953,
      "step": 87239
    },
    {
      "epoch": 1.2625694274902344e-05,
      "step": 87239,
      "training_step_time": 6.921651840209961
    },
    {
      "epoch": 1.262664794921875e-05,
      "grad_norm": 0.10471798479557037,
      "learning_rate": 4.386390961889969e-07,
      "loss": 0.0542,
      "step": 87240
    },
    {
      "epoch": 1.262664794921875e-05,
      "model_forward_time": 1.7496669292449951,
      "step": 87240
    },
    {
      "dynamics_loss_avg": 0.05290745720267296,
      "epoch": 1.262664794921875e-05,
      "step": 87240
    },
    {
      "action_loss_avg": 0.0011771457764552906,
      "epoch": 1.262664794921875e-05,
      "step": 87240
    },
    {
      "epoch": 1.262664794921875e-05,
      "step": 87240,
      "training_step_time": 6.813951730728149
    },
    {
      "epoch": 1.2627601623535156e-05,
      "model_forward_time": 1.7701151371002197,
      "step": 87241
    },
    {
      "epoch": 1.2627601623535156e-05,
      "step": 87241,
      "training_step_time": 6.882505178451538
    },
    {
      "epoch": 1.2628555297851563e-05,
      "model_forward_time": 1.7720577716827393,
      "step": 87242
    },
    {
      "epoch": 1.2628555297851563e-05,
      "step": 87242,
      "training_step_time": 6.8759002685546875
    },
    {
      "epoch": 1.2629508972167969e-05,
      "model_forward_time": 2.4794442653656006,
      "step": 87243
    },
    {
      "epoch": 1.2629508972167969e-05,
      "step": 87243,
      "training_step_time": 8.014116048812866
    },
    {
      "epoch": 1.2630462646484375e-05,
      "model_forward_time": 1.7731537818908691,
      "step": 87244
    },
    {
      "epoch": 1.2630462646484375e-05,
      "step": 87244,
      "training_step_time": 6.940298557281494
    },
    {
      "epoch": 1.263141632080078e-05,
      "model_forward_time": 1.7696824073791504,
      "step": 87245
    },
    {
      "epoch": 1.263141632080078e-05,
      "step": 87245,
      "training_step_time": 7.649535417556763
    },
    {
      "epoch": 1.2632369995117188e-05,
      "model_forward_time": 1.7727510929107666,
      "step": 87246
    },
    {
      "epoch": 1.2632369995117188e-05,
      "step": 87246,
      "training_step_time": 7.6109466552734375
    },
    {
      "epoch": 1.2633323669433594e-05,
      "model_forward_time": 1.3888187408447266,
      "step": 87247
    },
    {
      "epoch": 1.2633323669433594e-05,
      "step": 87247,
      "training_step_time": 8.441834449768066
    },
    {
      "epoch": 1.263427734375e-05,
      "model_forward_time": 1.7860422134399414,
      "step": 87248
    },
    {
      "epoch": 1.263427734375e-05,
      "step": 87248,
      "training_step_time": 7.625415086746216
    },
    {
      "epoch": 1.2635231018066407e-05,
      "model_forward_time": 1.7738275527954102,
      "step": 87249
    },
    {
      "epoch": 1.2635231018066407e-05,
      "step": 87249,
      "training_step_time": 6.850123405456543
    },
    {
      "epoch": 1.2636184692382812e-05,
      "grad_norm": 0.08554989099502563,
      "learning_rate": 4.37962109993349e-07,
      "loss": 0.054,
      "step": 87250
    },
    {
      "epoch": 1.2636184692382812e-05,
      "model_forward_time": 1.762765884399414,
      "step": 87250
    },
    {
      "dynamics_loss_avg": 0.06131305806338787,
      "epoch": 1.2636184692382812e-05,
      "step": 87250
    },
    {
      "action_loss_avg": 0.0017561563872732222,
      "epoch": 1.2636184692382812e-05,
      "step": 87250
    },
    {
      "epoch": 1.2636184692382812e-05,
      "step": 87250,
      "training_step_time": 6.880108118057251
    },
    {
      "epoch": 1.2637138366699218e-05,
      "model_forward_time": 1.7678921222686768,
      "step": 87251
    },
    {
      "epoch": 1.2637138366699218e-05,
      "step": 87251,
      "training_step_time": 6.834866046905518
    },
    {
      "epoch": 1.2638092041015626e-05,
      "model_forward_time": 2.4671778678894043,
      "step": 87252
    },
    {
      "epoch": 1.2638092041015626e-05,
      "step": 87252,
      "training_step_time": 9.738619327545166
    },
    {
      "epoch": 1.2639045715332031e-05,
      "model_forward_time": 1.7745137214660645,
      "step": 87253
    },
    {
      "epoch": 1.2639045715332031e-05,
      "step": 87253,
      "training_step_time": 7.77264142036438
    },
    {
      "epoch": 1.2639999389648437e-05,
      "model_forward_time": 1.770698070526123,
      "step": 87254
    },
    {
      "epoch": 1.2639999389648437e-05,
      "step": 87254,
      "training_step_time": 7.787656545639038
    },
    {
      "epoch": 1.2640953063964844e-05,
      "model_forward_time": 1.780644416809082,
      "step": 87255
    },
    {
      "epoch": 1.2640953063964844e-05,
      "step": 87255,
      "training_step_time": 7.841244459152222
    },
    {
      "epoch": 1.264190673828125e-05,
      "model_forward_time": 1.746640920639038,
      "step": 87256
    },
    {
      "epoch": 1.264190673828125e-05,
      "step": 87256,
      "training_step_time": 7.788875341415405
    },
    {
      "epoch": 1.2642860412597656e-05,
      "model_forward_time": 1.753422498703003,
      "step": 87257
    },
    {
      "epoch": 1.2642860412597656e-05,
      "step": 87257,
      "training_step_time": 7.544909238815308
    },
    {
      "epoch": 1.2643814086914062e-05,
      "model_forward_time": 1.766176462173462,
      "step": 87258
    },
    {
      "epoch": 1.2643814086914062e-05,
      "step": 87258,
      "training_step_time": 6.910930156707764
    },
    {
      "epoch": 1.2644767761230469e-05,
      "model_forward_time": 1.7674989700317383,
      "step": 87259
    },
    {
      "epoch": 1.2644767761230469e-05,
      "step": 87259,
      "training_step_time": 6.8548078536987305
    },
    {
      "epoch": 1.2645721435546875e-05,
      "grad_norm": 0.07989587634801865,
      "learning_rate": 4.372856226953065e-07,
      "loss": 0.054,
      "step": 87260
    },
    {
      "epoch": 1.2645721435546875e-05,
      "model_forward_time": 1.750276803970337,
      "step": 87260
    },
    {
      "dynamics_loss_avg": 0.05746212434023619,
      "epoch": 1.2645721435546875e-05,
      "step": 87260
    },
    {
      "action_loss_avg": 0.0015217683161608876,
      "epoch": 1.2645721435546875e-05,
      "step": 87260
    },
    {
      "epoch": 1.2645721435546875e-05,
      "step": 87260,
      "training_step_time": 6.883559942245483
    },
    {
      "epoch": 1.264667510986328e-05,
      "model_forward_time": 2.394838809967041,
      "step": 87261
    },
    {
      "epoch": 1.264667510986328e-05,
      "step": 87261,
      "training_step_time": 9.174888372421265
    },
    {
      "epoch": 1.2647628784179688e-05,
      "model_forward_time": 1.755260944366455,
      "step": 87262
    },
    {
      "epoch": 1.2647628784179688e-05,
      "step": 87262,
      "training_step_time": 7.83404541015625
    },
    {
      "epoch": 1.2648582458496094e-05,
      "model_forward_time": 1.7593531608581543,
      "step": 87263
    },
    {
      "epoch": 1.2648582458496094e-05,
      "step": 87263,
      "training_step_time": 7.767206192016602
    },
    {
      "epoch": 1.26495361328125e-05,
      "model_forward_time": 1.7603554725646973,
      "step": 87264
    },
    {
      "epoch": 1.26495361328125e-05,
      "step": 87264,
      "training_step_time": 7.78772759437561
    },
    {
      "epoch": 1.2650489807128907e-05,
      "model_forward_time": 1.7515687942504883,
      "step": 87265
    },
    {
      "epoch": 1.2650489807128907e-05,
      "step": 87265,
      "training_step_time": 7.808258533477783
    },
    {
      "epoch": 1.2651443481445313e-05,
      "model_forward_time": 1.7578155994415283,
      "step": 87266
    },
    {
      "epoch": 1.2651443481445313e-05,
      "step": 87266,
      "training_step_time": 7.587773323059082
    },
    {
      "epoch": 1.2652397155761718e-05,
      "model_forward_time": 1.7611281871795654,
      "step": 87267
    },
    {
      "epoch": 1.2652397155761718e-05,
      "step": 87267,
      "training_step_time": 6.884579658508301
    },
    {
      "epoch": 1.2653350830078126e-05,
      "model_forward_time": 1.7576048374176025,
      "step": 87268
    },
    {
      "epoch": 1.2653350830078126e-05,
      "step": 87268,
      "training_step_time": 6.924230337142944
    },
    {
      "epoch": 1.2654304504394531e-05,
      "model_forward_time": 1.7593591213226318,
      "step": 87269
    },
    {
      "epoch": 1.2654304504394531e-05,
      "step": 87269,
      "training_step_time": 6.901899814605713
    },
    {
      "epoch": 1.2655258178710937e-05,
      "grad_norm": 0.07040630280971527,
      "learning_rate": 4.3660963436884864e-07,
      "loss": 0.0514,
      "step": 87270
    },
    {
      "epoch": 1.2655258178710937e-05,
      "model_forward_time": 1.7728095054626465,
      "step": 87270
    },
    {
      "dynamics_loss_avg": 0.04469774402678013,
      "epoch": 1.2655258178710937e-05,
      "step": 87270
    },
    {
      "action_loss_avg": 0.0011433604973717592,
      "epoch": 1.2655258178710937e-05,
      "step": 87270
    },
    {
      "epoch": 1.2655258178710937e-05,
      "step": 87270,
      "training_step_time": 12.390713691711426
    },
    {
      "epoch": 1.2656211853027345e-05,
      "model_forward_time": 2.468196153640747,
      "step": 87271
    },
    {
      "epoch": 1.2656211853027345e-05,
      "step": 87271,
      "training_step_time": 7.8902294635772705
    },
    {
      "epoch": 1.265716552734375e-05,
      "model_forward_time": 1.7563064098358154,
      "step": 87272
    },
    {
      "epoch": 1.265716552734375e-05,
      "step": 87272,
      "training_step_time": 7.971277713775635
    },
    {
      "epoch": 1.2658119201660156e-05,
      "model_forward_time": 1.7569053173065186,
      "step": 87273
    },
    {
      "epoch": 1.2658119201660156e-05,
      "step": 87273,
      "training_step_time": 8.670779943466187
    },
    {
      "epoch": 1.2659072875976562e-05,
      "model_forward_time": 1.3795139789581299,
      "step": 87274
    },
    {
      "epoch": 1.2659072875976562e-05,
      "step": 87274,
      "training_step_time": 7.961672306060791
    },
    {
      "epoch": 1.2660026550292969e-05,
      "model_forward_time": 1.7660603523254395,
      "step": 87275
    },
    {
      "epoch": 1.2660026550292969e-05,
      "step": 87275,
      "training_step_time": 7.535529613494873
    },
    {
      "epoch": 1.2660980224609375e-05,
      "model_forward_time": 1.7569220066070557,
      "step": 87276
    },
    {
      "epoch": 1.2660980224609375e-05,
      "step": 87276,
      "training_step_time": 6.8944408893585205
    },
    {
      "epoch": 1.266193389892578e-05,
      "model_forward_time": 1.7583744525909424,
      "step": 87277
    },
    {
      "epoch": 1.266193389892578e-05,
      "step": 87277,
      "training_step_time": 6.894216060638428
    },
    {
      "epoch": 1.2662887573242188e-05,
      "model_forward_time": 1.7549035549163818,
      "step": 87278
    },
    {
      "epoch": 1.2662887573242188e-05,
      "step": 87278,
      "training_step_time": 6.927525997161865
    },
    {
      "epoch": 1.2663841247558594e-05,
      "model_forward_time": 1.7544565200805664,
      "step": 87279
    },
    {
      "epoch": 1.2663841247558594e-05,
      "step": 87279,
      "training_step_time": 10.863537549972534
    },
    {
      "epoch": 1.2664794921875e-05,
      "grad_norm": 0.07280105352401733,
      "learning_rate": 4.3593414508789953e-07,
      "loss": 0.0526,
      "step": 87280
    },
    {
      "epoch": 1.2664794921875e-05,
      "model_forward_time": 2.400237560272217,
      "step": 87280
    },
    {
      "dynamics_loss_avg": 0.04528635963797569,
      "epoch": 1.2664794921875e-05,
      "step": 87280
    },
    {
      "action_loss_avg": 0.0006332206845399924,
      "epoch": 1.2664794921875e-05,
      "step": 87280
    },
    {
      "epoch": 1.2664794921875e-05,
      "step": 87280,
      "training_step_time": 8.58703088760376
    },
    {
      "epoch": 1.2665748596191407e-05,
      "model_forward_time": 1.7637217044830322,
      "step": 87281
    },
    {
      "epoch": 1.2665748596191407e-05,
      "step": 87281,
      "training_step_time": 7.874521493911743
    },
    {
      "epoch": 1.2666702270507813e-05,
      "model_forward_time": 1.758641242980957,
      "step": 87282
    },
    {
      "epoch": 1.2666702270507813e-05,
      "step": 87282,
      "training_step_time": 8.013715505599976
    },
    {
      "epoch": 1.2667655944824218e-05,
      "model_forward_time": 1.7572417259216309,
      "step": 87283
    },
    {
      "epoch": 1.2667655944824218e-05,
      "step": 87283,
      "training_step_time": 7.8217856884002686
    },
    {
      "epoch": 1.2668609619140626e-05,
      "model_forward_time": 1.7703800201416016,
      "step": 87284
    },
    {
      "epoch": 1.2668609619140626e-05,
      "step": 87284,
      "training_step_time": 7.577601909637451
    },
    {
      "epoch": 1.2669563293457031e-05,
      "model_forward_time": 1.7581560611724854,
      "step": 87285
    },
    {
      "epoch": 1.2669563293457031e-05,
      "step": 87285,
      "training_step_time": 7.554945468902588
    },
    {
      "epoch": 1.2670516967773437e-05,
      "model_forward_time": 1.7523643970489502,
      "step": 87286
    },
    {
      "epoch": 1.2670516967773437e-05,
      "step": 87286,
      "training_step_time": 6.952964782714844
    },
    {
      "epoch": 1.2671470642089845e-05,
      "model_forward_time": 1.7518277168273926,
      "step": 87287
    },
    {
      "epoch": 1.2671470642089845e-05,
      "step": 87287,
      "training_step_time": 6.957581281661987
    },
    {
      "epoch": 1.267242431640625e-05,
      "model_forward_time": 1.7532012462615967,
      "step": 87288
    },
    {
      "epoch": 1.267242431640625e-05,
      "step": 87288,
      "training_step_time": 8.290666103363037
    },
    {
      "epoch": 1.2673377990722656e-05,
      "model_forward_time": 2.447504997253418,
      "step": 87289
    },
    {
      "epoch": 1.2673377990722656e-05,
      "step": 87289,
      "training_step_time": 8.647092580795288
    },
    {
      "epoch": 1.2674331665039062e-05,
      "grad_norm": 0.06938739866018295,
      "learning_rate": 4.352591549263302e-07,
      "loss": 0.0529,
      "step": 87290
    },
    {
      "epoch": 1.2674331665039062e-05,
      "model_forward_time": 1.7557501792907715,
      "step": 87290
    },
    {
      "dynamics_loss_avg": 0.055441072583198546,
      "epoch": 1.2674331665039062e-05,
      "step": 87290
    },
    {
      "action_loss_avg": 0.00070052943774499,
      "epoch": 1.2674331665039062e-05,
      "step": 87290
    },
    {
      "epoch": 1.2674331665039062e-05,
      "step": 87290,
      "training_step_time": 7.852853298187256
    },
    {
      "epoch": 1.2675285339355469e-05,
      "model_forward_time": 1.753849744796753,
      "step": 87291
    },
    {
      "epoch": 1.2675285339355469e-05,
      "step": 87291,
      "training_step_time": 7.988848924636841
    },
    {
      "epoch": 1.2676239013671875e-05,
      "model_forward_time": 1.759547233581543,
      "step": 87292
    },
    {
      "epoch": 1.2676239013671875e-05,
      "step": 87292,
      "training_step_time": 7.739561319351196
    },
    {
      "epoch": 1.267719268798828e-05,
      "model_forward_time": 1.7662279605865479,
      "step": 87293
    },
    {
      "epoch": 1.267719268798828e-05,
      "step": 87293,
      "training_step_time": 7.788369178771973
    },
    {
      "epoch": 1.2678146362304688e-05,
      "model_forward_time": 1.7570703029632568,
      "step": 87294
    },
    {
      "epoch": 1.2678146362304688e-05,
      "step": 87294,
      "training_step_time": 7.5234198570251465
    },
    {
      "epoch": 1.2679100036621094e-05,
      "model_forward_time": 1.7605528831481934,
      "step": 87295
    },
    {
      "epoch": 1.2679100036621094e-05,
      "step": 87295,
      "training_step_time": 6.890681505203247
    },
    {
      "epoch": 1.26800537109375e-05,
      "model_forward_time": 1.7599258422851562,
      "step": 87296
    },
    {
      "epoch": 1.26800537109375e-05,
      "step": 87296,
      "training_step_time": 6.911983251571655
    },
    {
      "epoch": 1.2681007385253907e-05,
      "model_forward_time": 1.7583086490631104,
      "step": 87297
    },
    {
      "epoch": 1.2681007385253907e-05,
      "step": 87297,
      "training_step_time": 8.473305940628052
    },
    {
      "epoch": 1.2681961059570313e-05,
      "model_forward_time": 2.4257867336273193,
      "step": 87298
    },
    {
      "epoch": 1.2681961059570313e-05,
      "step": 87298,
      "training_step_time": 10.631208896636963
    },
    {
      "epoch": 1.2682914733886718e-05,
      "model_forward_time": 1.7584104537963867,
      "step": 87299
    },
    {
      "epoch": 1.2682914733886718e-05,
      "step": 87299,
      "training_step_time": 7.892538785934448
    },
    {
      "epoch": 1.2683868408203126e-05,
      "grad_norm": 0.12796710431575775,
      "learning_rate": 4.3458466395795815e-07,
      "loss": 0.0515,
      "step": 87300
    },
    {
      "epoch": 1.2683868408203126e-05,
      "model_forward_time": 1.7555336952209473,
      "step": 87300
    },
    {
      "dynamics_loss_avg": 0.04837501402944326,
      "epoch": 1.2683868408203126e-05,
      "step": 87300
    },
    {
      "action_loss_avg": 0.0015139154420467094,
      "epoch": 1.2683868408203126e-05,
      "step": 87300
    },
    {
      "epoch": 1.2683868408203126e-05,
      "step": 87300,
      "training_step_time": 7.926645755767822
    },
    {
      "epoch": 1.2684822082519531e-05,
      "model_forward_time": 1.7579481601715088,
      "step": 87301
    },
    {
      "epoch": 1.2684822082519531e-05,
      "step": 87301,
      "training_step_time": 7.79266881942749
    },
    {
      "epoch": 1.2685775756835937e-05,
      "model_forward_time": 1.7525832653045654,
      "step": 87302
    },
    {
      "epoch": 1.2685775756835937e-05,
      "step": 87302,
      "training_step_time": 7.764718055725098
    },
    {
      "epoch": 1.2686729431152345e-05,
      "model_forward_time": 1.7647104263305664,
      "step": 87303
    },
    {
      "epoch": 1.2686729431152345e-05,
      "step": 87303,
      "training_step_time": 7.36040735244751
    },
    {
      "epoch": 1.268768310546875e-05,
      "model_forward_time": 1.7561118602752686,
      "step": 87304
    },
    {
      "epoch": 1.268768310546875e-05,
      "step": 87304,
      "training_step_time": 7.5665366649627686
    },
    {
      "epoch": 1.2688636779785156e-05,
      "model_forward_time": 1.7600862979888916,
      "step": 87305
    },
    {
      "epoch": 1.2688636779785156e-05,
      "step": 87305,
      "training_step_time": 6.945901393890381
    },
    {
      "epoch": 1.2689590454101562e-05,
      "model_forward_time": 1.7770214080810547,
      "step": 87306
    },
    {
      "epoch": 1.2689590454101562e-05,
      "step": 87306,
      "training_step_time": 7.348133087158203
    },
    {
      "epoch": 1.2690544128417969e-05,
      "model_forward_time": 1.7740452289581299,
      "step": 87307
    },
    {
      "epoch": 1.2690544128417969e-05,
      "step": 87307,
      "training_step_time": 11.722051620483398
    },
    {
      "epoch": 1.2691497802734375e-05,
      "model_forward_time": 1.872960090637207,
      "step": 87308
    },
    {
      "epoch": 1.2691497802734375e-05,
      "step": 87308,
      "training_step_time": 7.880929231643677
    },
    {
      "epoch": 1.269245147705078e-05,
      "model_forward_time": 1.755483627319336,
      "step": 87309
    },
    {
      "epoch": 1.269245147705078e-05,
      "step": 87309,
      "training_step_time": 7.818607568740845
    },
    {
      "epoch": 1.2693405151367188e-05,
      "grad_norm": 0.09322632849216461,
      "learning_rate": 4.339106722565417e-07,
      "loss": 0.0523,
      "step": 87310
    },
    {
      "epoch": 1.2693405151367188e-05,
      "model_forward_time": 1.7481050491333008,
      "step": 87310
    },
    {
      "dynamics_loss_avg": 0.041319580469280484,
      "epoch": 1.2693405151367188e-05,
      "step": 87310
    },
    {
      "action_loss_avg": 0.0011203013113117778,
      "epoch": 1.2693405151367188e-05,
      "step": 87310
    },
    {
      "epoch": 1.2693405151367188e-05,
      "step": 87310,
      "training_step_time": 7.843421220779419
    },
    {
      "epoch": 1.2694358825683594e-05,
      "model_forward_time": 1.7585420608520508,
      "step": 87311
    },
    {
      "epoch": 1.2694358825683594e-05,
      "step": 87311,
      "training_step_time": 7.830796480178833
    },
    {
      "epoch": 1.26953125e-05,
      "model_forward_time": 1.7614622116088867,
      "step": 87312
    },
    {
      "epoch": 1.26953125e-05,
      "step": 87312,
      "training_step_time": 7.281747817993164
    },
    {
      "epoch": 1.2696266174316407e-05,
      "model_forward_time": 1.7707369327545166,
      "step": 87313
    },
    {
      "epoch": 1.2696266174316407e-05,
      "step": 87313,
      "training_step_time": 7.507768392562866
    },
    {
      "epoch": 1.2697219848632813e-05,
      "model_forward_time": 1.7577872276306152,
      "step": 87314
    },
    {
      "epoch": 1.2697219848632813e-05,
      "step": 87314,
      "training_step_time": 6.74330735206604
    },
    {
      "epoch": 1.2698173522949218e-05,
      "model_forward_time": 1.7599108219146729,
      "step": 87315
    },
    {
      "epoch": 1.2698173522949218e-05,
      "step": 87315,
      "training_step_time": 8.059170961380005
    },
    {
      "epoch": 1.2699127197265626e-05,
      "model_forward_time": 1.7485501766204834,
      "step": 87316
    },
    {
      "epoch": 1.2699127197265626e-05,
      "step": 87316,
      "training_step_time": 9.005239725112915
    },
    {
      "epoch": 1.2700080871582031e-05,
      "model_forward_time": 2.450469493865967,
      "step": 87317
    },
    {
      "epoch": 1.2700080871582031e-05,
      "step": 87317,
      "training_step_time": 7.953766345977783
    },
    {
      "epoch": 1.2701034545898437e-05,
      "model_forward_time": 1.7556977272033691,
      "step": 87318
    },
    {
      "epoch": 1.2701034545898437e-05,
      "step": 87318,
      "training_step_time": 7.919127464294434
    },
    {
      "epoch": 1.2701988220214845e-05,
      "model_forward_time": 1.7555599212646484,
      "step": 87319
    },
    {
      "epoch": 1.2701988220214845e-05,
      "step": 87319,
      "training_step_time": 7.850343465805054
    },
    {
      "epoch": 1.270294189453125e-05,
      "grad_norm": 0.1501195877790451,
      "learning_rate": 4.332371798957896e-07,
      "loss": 0.0519,
      "step": 87320
    },
    {
      "epoch": 1.270294189453125e-05,
      "model_forward_time": 1.7544732093811035,
      "step": 87320
    },
    {
      "dynamics_loss_avg": 0.054824969917535785,
      "epoch": 1.270294189453125e-05,
      "step": 87320
    },
    {
      "action_loss_avg": 0.0012843177595641465,
      "epoch": 1.270294189453125e-05,
      "step": 87320
    },
    {
      "epoch": 1.270294189453125e-05,
      "step": 87320,
      "training_step_time": 7.7830727100372314
    },
    {
      "epoch": 1.2703895568847656e-05,
      "model_forward_time": 1.774658441543579,
      "step": 87321
    },
    {
      "epoch": 1.2703895568847656e-05,
      "step": 87321,
      "training_step_time": 7.34956431388855
    },
    {
      "epoch": 1.2704849243164062e-05,
      "model_forward_time": 1.7608554363250732,
      "step": 87322
    },
    {
      "epoch": 1.2704849243164062e-05,
      "step": 87322,
      "training_step_time": 7.559638738632202
    },
    {
      "epoch": 1.2705802917480469e-05,
      "model_forward_time": 1.7583019733428955,
      "step": 87323
    },
    {
      "epoch": 1.2705802917480469e-05,
      "step": 87323,
      "training_step_time": 6.9055798053741455
    },
    {
      "epoch": 1.2706756591796875e-05,
      "model_forward_time": 1.765439748764038,
      "step": 87324
    },
    {
      "epoch": 1.2706756591796875e-05,
      "step": 87324,
      "training_step_time": 7.404245376586914
    },
    {
      "epoch": 1.270771026611328e-05,
      "model_forward_time": 1.7608215808868408,
      "step": 87325
    },
    {
      "epoch": 1.270771026611328e-05,
      "step": 87325,
      "training_step_time": 7.773585081100464
    },
    {
      "epoch": 1.2708663940429688e-05,
      "model_forward_time": 2.46110463142395,
      "step": 87326
    },
    {
      "epoch": 1.2708663940429688e-05,
      "step": 87326,
      "training_step_time": 9.448350191116333
    },
    {
      "epoch": 1.2709617614746094e-05,
      "model_forward_time": 1.767892599105835,
      "step": 87327
    },
    {
      "epoch": 1.2709617614746094e-05,
      "step": 87327,
      "training_step_time": 7.8219897747039795
    },
    {
      "epoch": 1.27105712890625e-05,
      "model_forward_time": 1.7590584754943848,
      "step": 87328
    },
    {
      "epoch": 1.27105712890625e-05,
      "step": 87328,
      "training_step_time": 7.8396759033203125
    },
    {
      "epoch": 1.2711524963378907e-05,
      "model_forward_time": 1.7579588890075684,
      "step": 87329
    },
    {
      "epoch": 1.2711524963378907e-05,
      "step": 87329,
      "training_step_time": 7.7358856201171875
    },
    {
      "epoch": 1.2712478637695313e-05,
      "grad_norm": 0.11578915268182755,
      "learning_rate": 4.32564186949353e-07,
      "loss": 0.0516,
      "step": 87330
    },
    {
      "epoch": 1.2712478637695313e-05,
      "model_forward_time": 1.745124101638794,
      "step": 87330
    },
    {
      "dynamics_loss_avg": 0.046909451670944693,
      "epoch": 1.2712478637695313e-05,
      "step": 87330
    },
    {
      "action_loss_avg": 0.0010539123904891312,
      "epoch": 1.2712478637695313e-05,
      "step": 87330
    },
    {
      "epoch": 1.2712478637695313e-05,
      "step": 87330,
      "training_step_time": 6.876420974731445
    },
    {
      "epoch": 1.2713432312011718e-05,
      "model_forward_time": 1.793574333190918,
      "step": 87331
    },
    {
      "epoch": 1.2713432312011718e-05,
      "step": 87331,
      "training_step_time": 8.159385442733765
    },
    {
      "epoch": 1.2714385986328126e-05,
      "model_forward_time": 1.7638919353485107,
      "step": 87332
    },
    {
      "epoch": 1.2714385986328126e-05,
      "step": 87332,
      "training_step_time": 6.818658113479614
    },
    {
      "epoch": 1.2715339660644531e-05,
      "model_forward_time": 1.3888037204742432,
      "step": 87333
    },
    {
      "epoch": 1.2715339660644531e-05,
      "step": 87333,
      "training_step_time": 7.313175439834595
    },
    {
      "epoch": 1.2716293334960937e-05,
      "model_forward_time": 1.7528133392333984,
      "step": 87334
    },
    {
      "epoch": 1.2716293334960937e-05,
      "step": 87334,
      "training_step_time": 8.363610029220581
    },
    {
      "epoch": 1.2717247009277345e-05,
      "model_forward_time": 2.4938204288482666,
      "step": 87335
    },
    {
      "epoch": 1.2717247009277345e-05,
      "step": 87335,
      "training_step_time": 10.413045406341553
    },
    {
      "epoch": 1.271820068359375e-05,
      "model_forward_time": 1.7607462406158447,
      "step": 87336
    },
    {
      "epoch": 1.271820068359375e-05,
      "step": 87336,
      "training_step_time": 8.770564794540405
    },
    {
      "epoch": 1.2719154357910156e-05,
      "model_forward_time": 1.7789573669433594,
      "step": 87337
    },
    {
      "epoch": 1.2719154357910156e-05,
      "step": 87337,
      "training_step_time": 7.751279830932617
    },
    {
      "epoch": 1.2720108032226562e-05,
      "model_forward_time": 1.7622838020324707,
      "step": 87338
    },
    {
      "epoch": 1.2720108032226562e-05,
      "step": 87338,
      "training_step_time": 7.6945576667785645
    },
    {
      "epoch": 1.2721061706542969e-05,
      "model_forward_time": 1.7632617950439453,
      "step": 87339
    },
    {
      "epoch": 1.2721061706542969e-05,
      "step": 87339,
      "training_step_time": 6.728990316390991
    },
    {
      "epoch": 1.2722015380859375e-05,
      "grad_norm": 0.10003557056188583,
      "learning_rate": 4.318916934908313e-07,
      "loss": 0.0518,
      "step": 87340
    },
    {
      "epoch": 1.2722015380859375e-05,
      "model_forward_time": 1.7613978385925293,
      "step": 87340
    },
    {
      "dynamics_loss_avg": 0.06270563639700413,
      "epoch": 1.2722015380859375e-05,
      "step": 87340
    },
    {
      "action_loss_avg": 0.001841484781471081,
      "epoch": 1.2722015380859375e-05,
      "step": 87340
    },
    {
      "epoch": 1.2722015380859375e-05,
      "step": 87340,
      "training_step_time": 7.160798072814941
    },
    {
      "epoch": 1.272296905517578e-05,
      "model_forward_time": 1.7644784450531006,
      "step": 87341
    },
    {
      "epoch": 1.272296905517578e-05,
      "step": 87341,
      "training_step_time": 7.39509129524231
    },
    {
      "epoch": 1.2723922729492188e-05,
      "model_forward_time": 1.7617888450622559,
      "step": 87342
    },
    {
      "epoch": 1.2723922729492188e-05,
      "step": 87342,
      "training_step_time": 7.323883771896362
    },
    {
      "epoch": 1.2724876403808594e-05,
      "model_forward_time": 1.759150505065918,
      "step": 87343
    },
    {
      "epoch": 1.2724876403808594e-05,
      "step": 87343,
      "training_step_time": 7.687626123428345
    },
    {
      "epoch": 1.2725830078125e-05,
      "model_forward_time": 2.4205760955810547,
      "step": 87344
    },
    {
      "epoch": 1.2725830078125e-05,
      "step": 87344,
      "training_step_time": 9.678364038467407
    },
    {
      "epoch": 1.2726783752441407e-05,
      "model_forward_time": 1.7585437297821045,
      "step": 87345
    },
    {
      "epoch": 1.2726783752441407e-05,
      "step": 87345,
      "training_step_time": 7.78814697265625
    },
    {
      "epoch": 1.2727737426757813e-05,
      "model_forward_time": 1.7557008266448975,
      "step": 87346
    },
    {
      "epoch": 1.2727737426757813e-05,
      "step": 87346,
      "training_step_time": 7.759109973907471
    },
    {
      "epoch": 1.2728691101074218e-05,
      "model_forward_time": 1.7545158863067627,
      "step": 87347
    },
    {
      "epoch": 1.2728691101074218e-05,
      "step": 87347,
      "training_step_time": 7.8162736892700195
    },
    {
      "epoch": 1.2729644775390626e-05,
      "model_forward_time": 1.760345220565796,
      "step": 87348
    },
    {
      "epoch": 1.2729644775390626e-05,
      "step": 87348,
      "training_step_time": 6.863447427749634
    },
    {
      "epoch": 1.2730598449707031e-05,
      "model_forward_time": 1.7574007511138916,
      "step": 87349
    },
    {
      "epoch": 1.2730598449707031e-05,
      "step": 87349,
      "training_step_time": 7.20425820350647
    },
    {
      "epoch": 1.2731552124023437e-05,
      "grad_norm": 0.08831564337015152,
      "learning_rate": 4.312196995937645e-07,
      "loss": 0.0511,
      "step": 87350
    },
    {
      "epoch": 1.2731552124023437e-05,
      "model_forward_time": 1.7486462593078613,
      "step": 87350
    },
    {
      "dynamics_loss_avg": 0.04279324226081371,
      "epoch": 1.2731552124023437e-05,
      "step": 87350
    },
    {
      "action_loss_avg": 0.0009714965170132927,
      "epoch": 1.2731552124023437e-05,
      "step": 87350
    },
    {
      "epoch": 1.2731552124023437e-05,
      "step": 87350,
      "training_step_time": 7.432842493057251
    },
    {
      "epoch": 1.2732505798339845e-05,
      "model_forward_time": 1.7787926197052002,
      "step": 87351
    },
    {
      "epoch": 1.2732505798339845e-05,
      "step": 87351,
      "training_step_time": 7.861262559890747
    },
    {
      "epoch": 1.273345947265625e-05,
      "model_forward_time": 1.7618153095245361,
      "step": 87352
    },
    {
      "epoch": 1.273345947265625e-05,
      "step": 87352,
      "training_step_time": 7.6899354457855225
    },
    {
      "epoch": 1.2734413146972656e-05,
      "model_forward_time": 1.7518141269683838,
      "step": 87353
    },
    {
      "epoch": 1.2734413146972656e-05,
      "step": 87353,
      "training_step_time": 11.249846458435059
    },
    {
      "epoch": 1.2735366821289062e-05,
      "model_forward_time": 1.8742082118988037,
      "step": 87354
    },
    {
      "epoch": 1.2735366821289062e-05,
      "step": 87354,
      "training_step_time": 8.635247230529785
    },
    {
      "epoch": 1.273632049560547e-05,
      "model_forward_time": 1.7575457096099854,
      "step": 87355
    },
    {
      "epoch": 1.273632049560547e-05,
      "step": 87355,
      "training_step_time": 8.236318826675415
    },
    {
      "epoch": 1.2737274169921875e-05,
      "model_forward_time": 1.7679190635681152,
      "step": 87356
    },
    {
      "epoch": 1.2737274169921875e-05,
      "step": 87356,
      "training_step_time": 8.672003030776978
    },
    {
      "epoch": 1.273822784423828e-05,
      "model_forward_time": 1.7727351188659668,
      "step": 87357
    },
    {
      "epoch": 1.273822784423828e-05,
      "step": 87357,
      "training_step_time": 7.431083917617798
    },
    {
      "epoch": 1.2739181518554688e-05,
      "model_forward_time": 1.7505271434783936,
      "step": 87358
    },
    {
      "epoch": 1.2739181518554688e-05,
      "step": 87358,
      "training_step_time": 6.7828850746154785
    },
    {
      "epoch": 1.2740135192871094e-05,
      "model_forward_time": 1.7525103092193604,
      "step": 87359
    },
    {
      "epoch": 1.2740135192871094e-05,
      "step": 87359,
      "training_step_time": 7.413167953491211
    },
    {
      "epoch": 1.27410888671875e-05,
      "grad_norm": 0.08112883567810059,
      "learning_rate": 4.305482053316434e-07,
      "loss": 0.0535,
      "step": 87360
    },
    {
      "epoch": 1.27410888671875e-05,
      "model_forward_time": 1.7530226707458496,
      "step": 87360
    },
    {
      "dynamics_loss_avg": 0.04206860363483429,
      "epoch": 1.27410888671875e-05,
      "step": 87360
    },
    {
      "action_loss_avg": 0.0006987703294726088,
      "epoch": 1.27410888671875e-05,
      "step": 87360
    },
    {
      "epoch": 1.27410888671875e-05,
      "step": 87360,
      "training_step_time": 7.033892393112183
    },
    {
      "epoch": 1.2742042541503907e-05,
      "model_forward_time": 1.7452130317687988,
      "step": 87361
    },
    {
      "epoch": 1.2742042541503907e-05,
      "step": 87361,
      "training_step_time": 8.40493655204773
    },
    {
      "epoch": 1.2742996215820313e-05,
      "model_forward_time": 1.7458512783050537,
      "step": 87362
    },
    {
      "epoch": 1.2742996215820313e-05,
      "step": 87362,
      "training_step_time": 10.876154661178589
    },
    {
      "epoch": 1.2743949890136718e-05,
      "model_forward_time": 2.4882125854492188,
      "step": 87363
    },
    {
      "epoch": 1.2743949890136718e-05,
      "step": 87363,
      "training_step_time": 8.426796913146973
    },
    {
      "epoch": 1.2744903564453126e-05,
      "model_forward_time": 1.7786874771118164,
      "step": 87364
    },
    {
      "epoch": 1.2744903564453126e-05,
      "step": 87364,
      "training_step_time": 7.722242832183838
    },
    {
      "epoch": 1.2745857238769531e-05,
      "model_forward_time": 1.7710623741149902,
      "step": 87365
    },
    {
      "epoch": 1.2745857238769531e-05,
      "step": 87365,
      "training_step_time": 7.773087739944458
    },
    {
      "epoch": 1.2746810913085937e-05,
      "model_forward_time": 1.749037265777588,
      "step": 87366
    },
    {
      "epoch": 1.2746810913085937e-05,
      "step": 87366,
      "training_step_time": 7.554621934890747
    },
    {
      "epoch": 1.2747764587402345e-05,
      "model_forward_time": 1.7452738285064697,
      "step": 87367
    },
    {
      "epoch": 1.2747764587402345e-05,
      "step": 87367,
      "training_step_time": 6.840203046798706
    },
    {
      "epoch": 1.274871826171875e-05,
      "model_forward_time": 1.7499735355377197,
      "step": 87368
    },
    {
      "epoch": 1.274871826171875e-05,
      "step": 87368,
      "training_step_time": 7.472324371337891
    },
    {
      "epoch": 1.2749671936035156e-05,
      "model_forward_time": 1.7504777908325195,
      "step": 87369
    },
    {
      "epoch": 1.2749671936035156e-05,
      "step": 87369,
      "training_step_time": 7.16493034362793
    },
    {
      "epoch": 1.2750625610351562e-05,
      "grad_norm": 0.08029159903526306,
      "learning_rate": 4.2987721077789926e-07,
      "loss": 0.0532,
      "step": 87370
    },
    {
      "epoch": 1.2750625610351562e-05,
      "model_forward_time": 1.7401230335235596,
      "step": 87370
    },
    {
      "dynamics_loss_avg": 0.03909700438380241,
      "epoch": 1.2750625610351562e-05,
      "step": 87370
    },
    {
      "action_loss_avg": 0.0009832450159592554,
      "epoch": 1.2750625610351562e-05,
      "step": 87370
    },
    {
      "epoch": 1.2750625610351562e-05,
      "step": 87370,
      "training_step_time": 7.653264760971069
    },
    {
      "epoch": 1.275157928466797e-05,
      "model_forward_time": 1.7358193397521973,
      "step": 87371
    },
    {
      "epoch": 1.275157928466797e-05,
      "step": 87371,
      "training_step_time": 9.104115009307861
    },
    {
      "epoch": 1.2752532958984375e-05,
      "model_forward_time": 2.41483473777771,
      "step": 87372
    },
    {
      "epoch": 1.2752532958984375e-05,
      "step": 87372,
      "training_step_time": 9.890016555786133
    },
    {
      "epoch": 1.275348663330078e-05,
      "model_forward_time": 1.740943431854248,
      "step": 87373
    },
    {
      "epoch": 1.275348663330078e-05,
      "step": 87373,
      "training_step_time": 7.752732038497925
    },
    {
      "epoch": 1.2754440307617188e-05,
      "model_forward_time": 1.754133939743042,
      "step": 87374
    },
    {
      "epoch": 1.2754440307617188e-05,
      "step": 87374,
      "training_step_time": 7.796594619750977
    },
    {
      "epoch": 1.2755393981933594e-05,
      "model_forward_time": 1.7437353134155273,
      "step": 87375
    },
    {
      "epoch": 1.2755393981933594e-05,
      "step": 87375,
      "training_step_time": 7.577223539352417
    },
    {
      "epoch": 1.275634765625e-05,
      "model_forward_time": 1.7495956420898438,
      "step": 87376
    },
    {
      "epoch": 1.275634765625e-05,
      "step": 87376,
      "training_step_time": 6.890196800231934
    },
    {
      "epoch": 1.2757301330566407e-05,
      "model_forward_time": 1.740065574645996,
      "step": 87377
    },
    {
      "epoch": 1.2757301330566407e-05,
      "step": 87377,
      "training_step_time": 7.520137786865234
    },
    {
      "epoch": 1.2758255004882813e-05,
      "model_forward_time": 1.757685899734497,
      "step": 87378
    },
    {
      "epoch": 1.2758255004882813e-05,
      "step": 87378,
      "training_step_time": 7.211588621139526
    },
    {
      "epoch": 1.2759208679199218e-05,
      "model_forward_time": 1.7517657279968262,
      "step": 87379
    },
    {
      "epoch": 1.2759208679199218e-05,
      "step": 87379,
      "training_step_time": 7.3418872356414795
    },
    {
      "epoch": 1.2760162353515626e-05,
      "grad_norm": 0.0739961490035057,
      "learning_rate": 4.2920671600591326e-07,
      "loss": 0.0532,
      "step": 87380
    },
    {
      "epoch": 1.2760162353515626e-05,
      "model_forward_time": 1.7460508346557617,
      "step": 87380
    },
    {
      "dynamics_loss_avg": 0.05092813242226839,
      "epoch": 1.2760162353515626e-05,
      "step": 87380
    },
    {
      "action_loss_avg": 0.0009161829366348684,
      "epoch": 1.2760162353515626e-05,
      "step": 87380
    },
    {
      "epoch": 1.2760162353515626e-05,
      "step": 87380,
      "training_step_time": 10.167532444000244
    },
    {
      "epoch": 1.2761116027832031e-05,
      "model_forward_time": 2.35967755317688,
      "step": 87381
    },
    {
      "epoch": 1.2761116027832031e-05,
      "step": 87381,
      "training_step_time": 7.903047800064087
    },
    {
      "epoch": 1.2762069702148437e-05,
      "model_forward_time": 1.862393856048584,
      "step": 87382
    },
    {
      "epoch": 1.2762069702148437e-05,
      "step": 87382,
      "training_step_time": 7.850252151489258
    },
    {
      "epoch": 1.2763023376464843e-05,
      "model_forward_time": 1.7472550868988037,
      "step": 87383
    },
    {
      "epoch": 1.2763023376464843e-05,
      "step": 87383,
      "training_step_time": 8.309349775314331
    },
    {
      "epoch": 1.276397705078125e-05,
      "model_forward_time": 1.7495570182800293,
      "step": 87384
    },
    {
      "epoch": 1.276397705078125e-05,
      "step": 87384,
      "training_step_time": 7.667769193649292
    },
    {
      "epoch": 1.2764930725097656e-05,
      "model_forward_time": 1.7476933002471924,
      "step": 87385
    },
    {
      "epoch": 1.2764930725097656e-05,
      "step": 87385,
      "training_step_time": 7.2225282192230225
    },
    {
      "epoch": 1.2765884399414062e-05,
      "model_forward_time": 1.74247407913208,
      "step": 87386
    },
    {
      "epoch": 1.2765884399414062e-05,
      "step": 87386,
      "training_step_time": 7.485017538070679
    },
    {
      "epoch": 1.276683807373047e-05,
      "model_forward_time": 1.749035120010376,
      "step": 87387
    },
    {
      "epoch": 1.276683807373047e-05,
      "step": 87387,
      "training_step_time": 7.189655065536499
    },
    {
      "epoch": 1.2767791748046875e-05,
      "model_forward_time": 1.7531826496124268,
      "step": 87388
    },
    {
      "epoch": 1.2767791748046875e-05,
      "step": 87388,
      "training_step_time": 7.357745885848999
    },
    {
      "epoch": 1.276874542236328e-05,
      "model_forward_time": 1.7500841617584229,
      "step": 87389
    },
    {
      "epoch": 1.276874542236328e-05,
      "step": 87389,
      "training_step_time": 9.868579149246216
    },
    {
      "epoch": 1.2769699096679688e-05,
      "grad_norm": 0.08467397838830948,
      "learning_rate": 4.2853672108900846e-07,
      "loss": 0.0542,
      "step": 87390
    },
    {
      "epoch": 1.2769699096679688e-05,
      "model_forward_time": 2.40781831741333,
      "step": 87390
    },
    {
      "dynamics_loss_avg": 0.04657344594597816,
      "epoch": 1.2769699096679688e-05,
      "step": 87390
    },
    {
      "action_loss_avg": 0.0010591262573143468,
      "epoch": 1.2769699096679688e-05,
      "step": 87390
    },
    {
      "epoch": 1.2769699096679688e-05,
      "step": 87390,
      "training_step_time": 10.68751835823059
    },
    {
      "epoch": 1.2770652770996094e-05,
      "model_forward_time": 1.8732092380523682,
      "step": 87391
    },
    {
      "epoch": 1.2770652770996094e-05,
      "step": 87391,
      "training_step_time": 8.393426418304443
    },
    {
      "epoch": 1.27716064453125e-05,
      "model_forward_time": 1.7845008373260498,
      "step": 87392
    },
    {
      "epoch": 1.27716064453125e-05,
      "step": 87392,
      "training_step_time": 7.671477317810059
    },
    {
      "epoch": 1.2772560119628907e-05,
      "model_forward_time": 1.7415874004364014,
      "step": 87393
    },
    {
      "epoch": 1.2772560119628907e-05,
      "step": 87393,
      "training_step_time": 8.443104028701782
    },
    {
      "epoch": 1.2773513793945313e-05,
      "model_forward_time": 1.742006540298462,
      "step": 87394
    },
    {
      "epoch": 1.2773513793945313e-05,
      "step": 87394,
      "training_step_time": 7.620914697647095
    },
    {
      "epoch": 1.2774467468261718e-05,
      "model_forward_time": 1.7565512657165527,
      "step": 87395
    },
    {
      "epoch": 1.2774467468261718e-05,
      "step": 87395,
      "training_step_time": 7.174501657485962
    },
    {
      "epoch": 1.2775421142578126e-05,
      "model_forward_time": 1.74237060546875,
      "step": 87396
    },
    {
      "epoch": 1.2775421142578126e-05,
      "step": 87396,
      "training_step_time": 7.342053174972534
    },
    {
      "epoch": 1.2776374816894531e-05,
      "model_forward_time": 1.7487928867340088,
      "step": 87397
    },
    {
      "epoch": 1.2776374816894531e-05,
      "step": 87397,
      "training_step_time": 7.293151140213013
    },
    {
      "epoch": 1.2777328491210937e-05,
      "model_forward_time": 1.7387354373931885,
      "step": 87398
    },
    {
      "epoch": 1.2777328491210937e-05,
      "step": 87398,
      "training_step_time": 7.6957573890686035
    },
    {
      "epoch": 1.2778282165527343e-05,
      "model_forward_time": 2.3752365112304688,
      "step": 87399
    },
    {
      "epoch": 1.2778282165527343e-05,
      "step": 87399,
      "training_step_time": 8.846272230148315
    },
    {
      "epoch": 1.277923583984375e-05,
      "grad_norm": 0.09636576473712921,
      "learning_rate": 4.2786722610045517e-07,
      "loss": 0.0561,
      "step": 87400
    },
    {
      "epoch": 1.277923583984375e-05,
      "model_forward_time": 1.8440189361572266,
      "step": 87400
    },
    {
      "dynamics_loss_avg": 0.04866181761026382,
      "epoch": 1.277923583984375e-05,
      "step": 87400
    },
    {
      "action_loss_avg": 0.0011541778629180044,
      "epoch": 1.277923583984375e-05,
      "step": 87400
    },
    {
      "epoch": 1.277923583984375e-05,
      "step": 87400,
      "training_step_time": 9.321110248565674
    },
    {
      "epoch": 1.2780189514160156e-05,
      "model_forward_time": 1.7548937797546387,
      "step": 87401
    },
    {
      "epoch": 1.2780189514160156e-05,
      "step": 87401,
      "training_step_time": 7.669837474822998
    },
    {
      "epoch": 1.2781143188476562e-05,
      "model_forward_time": 1.7472577095031738,
      "step": 87402
    },
    {
      "epoch": 1.2781143188476562e-05,
      "step": 87402,
      "training_step_time": 7.74841570854187
    },
    {
      "epoch": 1.278209686279297e-05,
      "model_forward_time": 1.7473671436309814,
      "step": 87403
    },
    {
      "epoch": 1.278209686279297e-05,
      "step": 87403,
      "training_step_time": 7.565924882888794
    },
    {
      "epoch": 1.2783050537109375e-05,
      "model_forward_time": 1.743276596069336,
      "step": 87404
    },
    {
      "epoch": 1.2783050537109375e-05,
      "step": 87404,
      "training_step_time": 7.154327630996704
    },
    {
      "epoch": 1.278400421142578e-05,
      "model_forward_time": 1.747277021408081,
      "step": 87405
    },
    {
      "epoch": 1.278400421142578e-05,
      "step": 87405,
      "training_step_time": 7.433607339859009
    },
    {
      "epoch": 1.2784957885742188e-05,
      "model_forward_time": 1.7278046607971191,
      "step": 87406
    },
    {
      "epoch": 1.2784957885742188e-05,
      "step": 87406,
      "training_step_time": 6.721206188201904
    },
    {
      "epoch": 1.2785911560058594e-05,
      "model_forward_time": 1.7500193119049072,
      "step": 87407
    },
    {
      "epoch": 1.2785911560058594e-05,
      "step": 87407,
      "training_step_time": 7.616127252578735
    },
    {
      "epoch": 1.2786865234375e-05,
      "model_forward_time": 1.7390704154968262,
      "step": 87408
    },
    {
      "epoch": 1.2786865234375e-05,
      "step": 87408,
      "training_step_time": 8.568187713623047
    },
    {
      "epoch": 1.2787818908691407e-05,
      "model_forward_time": 2.508436441421509,
      "step": 87409
    },
    {
      "epoch": 1.2787818908691407e-05,
      "step": 87409,
      "training_step_time": 8.611288785934448
    },
    {
      "epoch": 1.2788772583007813e-05,
      "grad_norm": 0.08404773473739624,
      "learning_rate": 4.2719823111346705e-07,
      "loss": 0.0547,
      "step": 87410
    },
    {
      "epoch": 1.2788772583007813e-05,
      "model_forward_time": 1.7411856651306152,
      "step": 87410
    },
    {
      "dynamics_loss_avg": 0.044443299621343614,
      "epoch": 1.2788772583007813e-05,
      "step": 87410
    },
    {
      "action_loss_avg": 0.0005703186994651333,
      "epoch": 1.2788772583007813e-05,
      "step": 87410
    },
    {
      "epoch": 1.2788772583007813e-05,
      "step": 87410,
      "training_step_time": 7.759472846984863
    },
    {
      "epoch": 1.2789726257324218e-05,
      "model_forward_time": 1.741640567779541,
      "step": 87411
    },
    {
      "epoch": 1.2789726257324218e-05,
      "step": 87411,
      "training_step_time": 7.749761343002319
    },
    {
      "epoch": 1.2790679931640626e-05,
      "model_forward_time": 1.7551543712615967,
      "step": 87412
    },
    {
      "epoch": 1.2790679931640626e-05,
      "step": 87412,
      "training_step_time": 7.567120552062988
    },
    {
      "epoch": 1.2791633605957031e-05,
      "model_forward_time": 1.7507009506225586,
      "step": 87413
    },
    {
      "epoch": 1.2791633605957031e-05,
      "step": 87413,
      "training_step_time": 7.221389293670654
    },
    {
      "epoch": 1.2792587280273437e-05,
      "model_forward_time": 1.7429251670837402,
      "step": 87414
    },
    {
      "epoch": 1.2792587280273437e-05,
      "step": 87414,
      "training_step_time": 7.457374095916748
    },
    {
      "epoch": 1.2793540954589843e-05,
      "model_forward_time": 1.7373344898223877,
      "step": 87415
    },
    {
      "epoch": 1.2793540954589843e-05,
      "step": 87415,
      "training_step_time": 7.284587383270264
    },
    {
      "epoch": 1.279449462890625e-05,
      "model_forward_time": 1.7492992877960205,
      "step": 87416
    },
    {
      "epoch": 1.279449462890625e-05,
      "step": 87416,
      "training_step_time": 7.616426706314087
    },
    {
      "epoch": 1.2795448303222656e-05,
      "model_forward_time": 1.7443201541900635,
      "step": 87417
    },
    {
      "epoch": 1.2795448303222656e-05,
      "step": 87417,
      "training_step_time": 12.345720052719116
    },
    {
      "epoch": 1.2796401977539062e-05,
      "model_forward_time": 2.414267063140869,
      "step": 87418
    },
    {
      "epoch": 1.2796401977539062e-05,
      "step": 87418,
      "training_step_time": 8.94764518737793
    },
    {
      "epoch": 1.279735565185547e-05,
      "model_forward_time": 1.7266490459442139,
      "step": 87419
    },
    {
      "epoch": 1.279735565185547e-05,
      "step": 87419,
      "training_step_time": 8.614556789398193
    },
    {
      "epoch": 1.2798309326171875e-05,
      "grad_norm": 0.12443374842405319,
      "learning_rate": 4.265297362012061e-07,
      "loss": 0.055,
      "step": 87420
    },
    {
      "epoch": 1.2798309326171875e-05,
      "model_forward_time": 1.730919599533081,
      "step": 87420
    },
    {
      "dynamics_loss_avg": 0.05135936513543129,
      "epoch": 1.2798309326171875e-05,
      "step": 87420
    },
    {
      "action_loss_avg": 0.0010925526556093245,
      "epoch": 1.2798309326171875e-05,
      "step": 87420
    },
    {
      "epoch": 1.2798309326171875e-05,
      "step": 87420,
      "training_step_time": 7.659907102584839
    },
    {
      "epoch": 1.279926300048828e-05,
      "model_forward_time": 1.7258055210113525,
      "step": 87421
    },
    {
      "epoch": 1.279926300048828e-05,
      "step": 87421,
      "training_step_time": 7.5156569480896
    },
    {
      "epoch": 1.2800216674804688e-05,
      "model_forward_time": 1.7317421436309814,
      "step": 87422
    },
    {
      "epoch": 1.2800216674804688e-05,
      "step": 87422,
      "training_step_time": 7.604135036468506
    },
    {
      "epoch": 1.2801170349121094e-05,
      "model_forward_time": 1.7405261993408203,
      "step": 87423
    },
    {
      "epoch": 1.2801170349121094e-05,
      "step": 87423,
      "training_step_time": 7.421641826629639
    },
    {
      "epoch": 1.28021240234375e-05,
      "model_forward_time": 1.733431100845337,
      "step": 87424
    },
    {
      "epoch": 1.28021240234375e-05,
      "step": 87424,
      "training_step_time": 7.291937351226807
    },
    {
      "epoch": 1.2803077697753907e-05,
      "model_forward_time": 1.7329339981079102,
      "step": 87425
    },
    {
      "epoch": 1.2803077697753907e-05,
      "step": 87425,
      "training_step_time": 7.633300304412842
    },
    {
      "epoch": 1.2804031372070313e-05,
      "model_forward_time": 1.7294855117797852,
      "step": 87426
    },
    {
      "epoch": 1.2804031372070313e-05,
      "step": 87426,
      "training_step_time": 9.24873161315918
    },
    {
      "epoch": 1.2804985046386718e-05,
      "model_forward_time": 2.3915553092956543,
      "step": 87427
    },
    {
      "epoch": 1.2804985046386718e-05,
      "step": 87427,
      "training_step_time": 9.34921669960022
    },
    {
      "epoch": 1.2805938720703126e-05,
      "model_forward_time": 1.7271232604980469,
      "step": 87428
    },
    {
      "epoch": 1.2805938720703126e-05,
      "step": 87428,
      "training_step_time": 7.869298458099365
    },
    {
      "epoch": 1.2806892395019532e-05,
      "model_forward_time": 1.7338662147521973,
      "step": 87429
    },
    {
      "epoch": 1.2806892395019532e-05,
      "step": 87429,
      "training_step_time": 7.711588382720947
    },
    {
      "epoch": 1.2807846069335937e-05,
      "grad_norm": 0.0815286636352539,
      "learning_rate": 4.258617414367766e-07,
      "loss": 0.0569,
      "step": 87430
    },
    {
      "epoch": 1.2807846069335937e-05,
      "model_forward_time": 1.726233959197998,
      "step": 87430
    },
    {
      "dynamics_loss_avg": 0.04158441945910454,
      "epoch": 1.2807846069335937e-05,
      "step": 87430
    },
    {
      "action_loss_avg": 0.0009793806020752527,
      "epoch": 1.2807846069335937e-05,
      "step": 87430
    },
    {
      "epoch": 1.2807846069335937e-05,
      "step": 87430,
      "training_step_time": 7.707208156585693
    },
    {
      "epoch": 1.2808799743652343e-05,
      "model_forward_time": 1.3508038520812988,
      "step": 87431
    },
    {
      "epoch": 1.2808799743652343e-05,
      "step": 87431,
      "training_step_time": 7.588833332061768
    },
    {
      "epoch": 1.280975341796875e-05,
      "model_forward_time": 1.729677677154541,
      "step": 87432
    },
    {
      "epoch": 1.280975341796875e-05,
      "step": 87432,
      "training_step_time": 7.41842246055603
    },
    {
      "epoch": 1.2810707092285156e-05,
      "model_forward_time": 1.735239028930664,
      "step": 87433
    },
    {
      "epoch": 1.2810707092285156e-05,
      "step": 87433,
      "training_step_time": 7.2378926277160645
    },
    {
      "epoch": 1.2811660766601562e-05,
      "model_forward_time": 1.354496717453003,
      "step": 87434
    },
    {
      "epoch": 1.2811660766601562e-05,
      "step": 87434,
      "training_step_time": 7.6446874141693115
    },
    {
      "epoch": 1.281261444091797e-05,
      "model_forward_time": 1.734623670578003,
      "step": 87435
    },
    {
      "epoch": 1.281261444091797e-05,
      "step": 87435,
      "training_step_time": 8.588688373565674
    },
    {
      "epoch": 1.2813568115234375e-05,
      "model_forward_time": 2.3500173091888428,
      "step": 87436
    },
    {
      "epoch": 1.2813568115234375e-05,
      "step": 87436,
      "training_step_time": 9.212912559509277
    },
    {
      "epoch": 1.281452178955078e-05,
      "model_forward_time": 1.8143301010131836,
      "step": 87437
    },
    {
      "epoch": 1.281452178955078e-05,
      "step": 87437,
      "training_step_time": 8.545574426651001
    },
    {
      "epoch": 1.2815475463867188e-05,
      "model_forward_time": 1.718550205230713,
      "step": 87438
    },
    {
      "epoch": 1.2815475463867188e-05,
      "step": 87438,
      "training_step_time": 7.6991167068481445
    },
    {
      "epoch": 1.2816429138183594e-05,
      "model_forward_time": 1.7304270267486572,
      "step": 87439
    },
    {
      "epoch": 1.2816429138183594e-05,
      "step": 87439,
      "training_step_time": 7.745485782623291
    },
    {
      "epoch": 1.28173828125e-05,
      "grad_norm": 0.09049444645643234,
      "learning_rate": 4.2519424689322966e-07,
      "loss": 0.0574,
      "step": 87440
    },
    {
      "epoch": 1.28173828125e-05,
      "model_forward_time": 1.7269878387451172,
      "step": 87440
    },
    {
      "dynamics_loss_avg": 0.04239746741950512,
      "epoch": 1.28173828125e-05,
      "step": 87440
    },
    {
      "action_loss_avg": 0.001355808311200235,
      "epoch": 1.28173828125e-05,
      "step": 87440
    },
    {
      "epoch": 1.28173828125e-05,
      "step": 87440,
      "training_step_time": 7.44523811340332
    },
    {
      "epoch": 1.2818336486816407e-05,
      "model_forward_time": 1.7279210090637207,
      "step": 87441
    },
    {
      "epoch": 1.2818336486816407e-05,
      "step": 87441,
      "training_step_time": 7.121812105178833
    },
    {
      "epoch": 1.2819290161132813e-05,
      "model_forward_time": 1.734833002090454,
      "step": 87442
    },
    {
      "epoch": 1.2819290161132813e-05,
      "step": 87442,
      "training_step_time": 7.438835859298706
    },
    {
      "epoch": 1.2820243835449218e-05,
      "model_forward_time": 1.7428231239318848,
      "step": 87443
    },
    {
      "epoch": 1.2820243835449218e-05,
      "step": 87443,
      "training_step_time": 7.5310845375061035
    },
    {
      "epoch": 1.2821197509765626e-05,
      "model_forward_time": 1.7397964000701904,
      "step": 87444
    },
    {
      "epoch": 1.2821197509765626e-05,
      "step": 87444,
      "training_step_time": 8.251185417175293
    },
    {
      "epoch": 1.2822151184082032e-05,
      "model_forward_time": 1.7369837760925293,
      "step": 87445
    },
    {
      "epoch": 1.2822151184082032e-05,
      "step": 87445,
      "training_step_time": 11.101523399353027
    },
    {
      "epoch": 1.2823104858398437e-05,
      "model_forward_time": 2.3823647499084473,
      "step": 87446
    },
    {
      "epoch": 1.2823104858398437e-05,
      "step": 87446,
      "training_step_time": 7.828618764877319
    },
    {
      "epoch": 1.2824058532714843e-05,
      "model_forward_time": 1.7630512714385986,
      "step": 87447
    },
    {
      "epoch": 1.2824058532714843e-05,
      "step": 87447,
      "training_step_time": 7.759019374847412
    },
    {
      "epoch": 1.282501220703125e-05,
      "model_forward_time": 1.744602918624878,
      "step": 87448
    },
    {
      "epoch": 1.282501220703125e-05,
      "step": 87448,
      "training_step_time": 7.835156440734863
    },
    {
      "epoch": 1.2825965881347656e-05,
      "model_forward_time": 1.7441885471343994,
      "step": 87449
    },
    {
      "epoch": 1.2825965881347656e-05,
      "step": 87449,
      "training_step_time": 7.45173454284668
    },
    {
      "epoch": 1.2826919555664062e-05,
      "grad_norm": 0.08062499016523361,
      "learning_rate": 4.2452725264356175e-07,
      "loss": 0.0576,
      "step": 87450
    },
    {
      "epoch": 1.2826919555664062e-05,
      "model_forward_time": 1.7373297214508057,
      "step": 87450
    },
    {
      "dynamics_loss_avg": 0.04686534982174635,
      "epoch": 1.2826919555664062e-05,
      "step": 87450
    },
    {
      "action_loss_avg": 0.0009168720003799536,
      "epoch": 1.2826919555664062e-05,
      "step": 87450
    },
    {
      "epoch": 1.2826919555664062e-05,
      "step": 87450,
      "training_step_time": 7.606757164001465
    },
    {
      "epoch": 1.282787322998047e-05,
      "model_forward_time": 1.7551796436309814,
      "step": 87451
    },
    {
      "epoch": 1.282787322998047e-05,
      "step": 87451,
      "training_step_time": 7.325516939163208
    },
    {
      "epoch": 1.2828826904296875e-05,
      "model_forward_time": 1.7421538829803467,
      "step": 87452
    },
    {
      "epoch": 1.2828826904296875e-05,
      "step": 87452,
      "training_step_time": 7.460036993026733
    },
    {
      "epoch": 1.282978057861328e-05,
      "model_forward_time": 1.7427496910095215,
      "step": 87453
    },
    {
      "epoch": 1.282978057861328e-05,
      "step": 87453,
      "training_step_time": 7.772204160690308
    },
    {
      "epoch": 1.2830734252929688e-05,
      "model_forward_time": 1.776360273361206,
      "step": 87454
    },
    {
      "epoch": 1.2830734252929688e-05,
      "step": 87454,
      "training_step_time": 10.459024429321289
    },
    {
      "epoch": 1.2831687927246094e-05,
      "model_forward_time": 2.42521071434021,
      "step": 87455
    },
    {
      "epoch": 1.2831687927246094e-05,
      "step": 87455,
      "training_step_time": 9.409366846084595
    },
    {
      "epoch": 1.28326416015625e-05,
      "model_forward_time": 1.7647535800933838,
      "step": 87456
    },
    {
      "epoch": 1.28326416015625e-05,
      "step": 87456,
      "training_step_time": 7.716146945953369
    },
    {
      "epoch": 1.2833595275878907e-05,
      "model_forward_time": 1.7532949447631836,
      "step": 87457
    },
    {
      "epoch": 1.2833595275878907e-05,
      "step": 87457,
      "training_step_time": 7.697671890258789
    },
    {
      "epoch": 1.2834548950195313e-05,
      "model_forward_time": 1.7576379776000977,
      "step": 87458
    },
    {
      "epoch": 1.2834548950195313e-05,
      "step": 87458,
      "training_step_time": 7.184619903564453
    },
    {
      "epoch": 1.2835502624511718e-05,
      "model_forward_time": 1.7671866416931152,
      "step": 87459
    },
    {
      "epoch": 1.2835502624511718e-05,
      "step": 87459,
      "training_step_time": 7.571493148803711
    },
    {
      "epoch": 1.2836456298828126e-05,
      "grad_norm": 0.08558191359043121,
      "learning_rate": 4.2386075876071577e-07,
      "loss": 0.0557,
      "step": 87460
    },
    {
      "epoch": 1.2836456298828126e-05,
      "model_forward_time": 1.7603943347930908,
      "step": 87460
    },
    {
      "dynamics_loss_avg": 0.05456469962373376,
      "epoch": 1.2836456298828126e-05,
      "step": 87460
    },
    {
      "action_loss_avg": 0.0013452401384711266,
      "epoch": 1.2836456298828126e-05,
      "step": 87460
    },
    {
      "epoch": 1.2836456298828126e-05,
      "step": 87460,
      "training_step_time": 7.436139106750488
    },
    {
      "epoch": 1.2837409973144532e-05,
      "model_forward_time": 1.7645847797393799,
      "step": 87461
    },
    {
      "epoch": 1.2837409973144532e-05,
      "step": 87461,
      "training_step_time": 7.4554970264434814
    },
    {
      "epoch": 1.2838363647460937e-05,
      "model_forward_time": 1.757047176361084,
      "step": 87462
    },
    {
      "epoch": 1.2838363647460937e-05,
      "step": 87462,
      "training_step_time": 8.79144835472107
    },
    {
      "epoch": 1.2839317321777343e-05,
      "model_forward_time": 1.764331579208374,
      "step": 87463
    },
    {
      "epoch": 1.2839317321777343e-05,
      "step": 87463,
      "training_step_time": 8.522550582885742
    },
    {
      "epoch": 1.284027099609375e-05,
      "model_forward_time": 2.4312660694122314,
      "step": 87464
    },
    {
      "epoch": 1.284027099609375e-05,
      "step": 87464,
      "training_step_time": 9.209211826324463
    },
    {
      "epoch": 1.2841224670410156e-05,
      "model_forward_time": 1.761080026626587,
      "step": 87465
    },
    {
      "epoch": 1.2841224670410156e-05,
      "step": 87465,
      "training_step_time": 8.974152326583862
    },
    {
      "epoch": 1.2842178344726562e-05,
      "model_forward_time": 1.7590413093566895,
      "step": 87466
    },
    {
      "epoch": 1.2842178344726562e-05,
      "step": 87466,
      "training_step_time": 7.694939851760864
    },
    {
      "epoch": 1.284313201904297e-05,
      "model_forward_time": 1.3804845809936523,
      "step": 87467
    },
    {
      "epoch": 1.284313201904297e-05,
      "step": 87467,
      "training_step_time": 7.325703382492065
    },
    {
      "epoch": 1.2844085693359375e-05,
      "model_forward_time": 1.7616002559661865,
      "step": 87468
    },
    {
      "epoch": 1.2844085693359375e-05,
      "step": 87468,
      "training_step_time": 7.522470712661743
    },
    {
      "epoch": 1.284503936767578e-05,
      "model_forward_time": 1.7615256309509277,
      "step": 87469
    },
    {
      "epoch": 1.284503936767578e-05,
      "step": 87469,
      "training_step_time": 7.366096496582031
    },
    {
      "epoch": 1.2845993041992188e-05,
      "grad_norm": 0.09268061071634293,
      "learning_rate": 4.231947653175755e-07,
      "loss": 0.056,
      "step": 87470
    },
    {
      "epoch": 1.2845993041992188e-05,
      "model_forward_time": 1.7521765232086182,
      "step": 87470
    },
    {
      "dynamics_loss_avg": 0.05283596143126488,
      "epoch": 1.2845993041992188e-05,
      "step": 87470
    },
    {
      "action_loss_avg": 0.0018225144915049896,
      "epoch": 1.2845993041992188e-05,
      "step": 87470
    },
    {
      "epoch": 1.2845993041992188e-05,
      "step": 87470,
      "training_step_time": 7.439842462539673
    },
    {
      "epoch": 1.2846946716308594e-05,
      "model_forward_time": 1.7633137702941895,
      "step": 87471
    },
    {
      "epoch": 1.2846946716308594e-05,
      "step": 87471,
      "training_step_time": 7.693948745727539
    },
    {
      "epoch": 1.2847900390625e-05,
      "model_forward_time": 1.3930127620697021,
      "step": 87472
    },
    {
      "epoch": 1.2847900390625e-05,
      "step": 87472,
      "training_step_time": 10.359512329101562
    },
    {
      "epoch": 1.2848854064941407e-05,
      "model_forward_time": 2.3693478107452393,
      "step": 87473
    },
    {
      "epoch": 1.2848854064941407e-05,
      "step": 87473,
      "training_step_time": 11.430575370788574
    },
    {
      "epoch": 1.2849807739257813e-05,
      "model_forward_time": 1.7542846202850342,
      "step": 87474
    },
    {
      "epoch": 1.2849807739257813e-05,
      "step": 87474,
      "training_step_time": 8.921695709228516
    },
    {
      "epoch": 1.2850761413574218e-05,
      "model_forward_time": 1.7438774108886719,
      "step": 87475
    },
    {
      "epoch": 1.2850761413574218e-05,
      "step": 87475,
      "training_step_time": 8.694204807281494
    },
    {
      "epoch": 1.2851715087890626e-05,
      "model_forward_time": 1.743987798690796,
      "step": 87476
    },
    {
      "epoch": 1.2851715087890626e-05,
      "step": 87476,
      "training_step_time": 7.7247092723846436
    },
    {
      "epoch": 1.2852668762207032e-05,
      "model_forward_time": 1.7443671226501465,
      "step": 87477
    },
    {
      "epoch": 1.2852668762207032e-05,
      "step": 87477,
      "training_step_time": 7.40335488319397
    },
    {
      "epoch": 1.2853622436523437e-05,
      "model_forward_time": 1.7456979751586914,
      "step": 87478
    },
    {
      "epoch": 1.2853622436523437e-05,
      "step": 87478,
      "training_step_time": 7.609808921813965
    },
    {
      "epoch": 1.2854576110839843e-05,
      "model_forward_time": 1.741837501525879,
      "step": 87479
    },
    {
      "epoch": 1.2854576110839843e-05,
      "step": 87479,
      "training_step_time": 7.4349143505096436
    },
    {
      "epoch": 1.285552978515625e-05,
      "grad_norm": 0.16974857449531555,
      "learning_rate": 4.225292723869762e-07,
      "loss": 0.0558,
      "step": 87480
    },
    {
      "epoch": 1.285552978515625e-05,
      "model_forward_time": 1.7433526515960693,
      "step": 87480
    },
    {
      "dynamics_loss_avg": 0.04233089238405228,
      "epoch": 1.285552978515625e-05,
      "step": 87480
    },
    {
      "action_loss_avg": 0.0007486655318643898,
      "epoch": 1.285552978515625e-05,
      "step": 87480
    },
    {
      "epoch": 1.285552978515625e-05,
      "step": 87480,
      "training_step_time": 7.701012134552002
    },
    {
      "epoch": 1.2856483459472656e-05,
      "model_forward_time": 1.7461602687835693,
      "step": 87481
    },
    {
      "epoch": 1.2856483459472656e-05,
      "step": 87481,
      "training_step_time": 7.711992502212524
    },
    {
      "epoch": 1.2857437133789062e-05,
      "model_forward_time": 1.7414429187774658,
      "step": 87482
    },
    {
      "epoch": 1.2857437133789062e-05,
      "step": 87482,
      "training_step_time": 9.632214307785034
    },
    {
      "epoch": 1.285839080810547e-05,
      "model_forward_time": 2.3891518115997314,
      "step": 87483
    },
    {
      "epoch": 1.285839080810547e-05,
      "step": 87483,
      "training_step_time": 8.596072673797607
    },
    {
      "epoch": 1.2859344482421875e-05,
      "model_forward_time": 1.7470955848693848,
      "step": 87484
    },
    {
      "epoch": 1.2859344482421875e-05,
      "step": 87484,
      "training_step_time": 7.651485443115234
    },
    {
      "epoch": 1.286029815673828e-05,
      "model_forward_time": 1.735886573791504,
      "step": 87485
    },
    {
      "epoch": 1.286029815673828e-05,
      "step": 87485,
      "training_step_time": 7.837034463882446
    },
    {
      "epoch": 1.2861251831054688e-05,
      "model_forward_time": 1.7482142448425293,
      "step": 87486
    },
    {
      "epoch": 1.2861251831054688e-05,
      "step": 87486,
      "training_step_time": 7.39687705039978
    },
    {
      "epoch": 1.2862205505371094e-05,
      "model_forward_time": 1.7421684265136719,
      "step": 87487
    },
    {
      "epoch": 1.2862205505371094e-05,
      "step": 87487,
      "training_step_time": 7.50512170791626
    },
    {
      "epoch": 1.28631591796875e-05,
      "model_forward_time": 1.7527687549591064,
      "step": 87488
    },
    {
      "epoch": 1.28631591796875e-05,
      "step": 87488,
      "training_step_time": 7.381253004074097
    },
    {
      "epoch": 1.2864112854003907e-05,
      "model_forward_time": 1.7408714294433594,
      "step": 87489
    },
    {
      "epoch": 1.2864112854003907e-05,
      "step": 87489,
      "training_step_time": 7.721829175949097
    },
    {
      "epoch": 1.2865066528320313e-05,
      "grad_norm": 0.103627510368824,
      "learning_rate": 4.2186428004169166e-07,
      "loss": 0.0566,
      "step": 87490
    },
    {
      "epoch": 1.2865066528320313e-05,
      "model_forward_time": 1.733309268951416,
      "step": 87490
    },
    {
      "dynamics_loss_avg": 0.05891118161380291,
      "epoch": 1.2865066528320313e-05,
      "step": 87490
    },
    {
      "action_loss_avg": 0.0014986937720095738,
      "epoch": 1.2865066528320313e-05,
      "step": 87490
    },
    {
      "epoch": 1.2865066528320313e-05,
      "step": 87490,
      "training_step_time": 7.650735139846802
    },
    {
      "epoch": 1.2866020202636718e-05,
      "model_forward_time": 1.750110149383545,
      "step": 87491
    },
    {
      "epoch": 1.2866020202636718e-05,
      "step": 87491,
      "training_step_time": 10.290930986404419
    },
    {
      "epoch": 1.2866973876953126e-05,
      "model_forward_time": 2.0007715225219727,
      "step": 87492
    },
    {
      "epoch": 1.2866973876953126e-05,
      "step": 87492,
      "training_step_time": 8.613583326339722
    },
    {
      "epoch": 1.2867927551269532e-05,
      "model_forward_time": 1.7475719451904297,
      "step": 87493
    },
    {
      "epoch": 1.2867927551269532e-05,
      "step": 87493,
      "training_step_time": 7.629485607147217
    },
    {
      "epoch": 1.2868881225585937e-05,
      "model_forward_time": 1.7447295188903809,
      "step": 87494
    },
    {
      "epoch": 1.2868881225585937e-05,
      "step": 87494,
      "training_step_time": 8.673277139663696
    },
    {
      "epoch": 1.2869834899902343e-05,
      "model_forward_time": 1.744184970855713,
      "step": 87495
    },
    {
      "epoch": 1.2869834899902343e-05,
      "step": 87495,
      "training_step_time": 7.282870769500732
    },
    {
      "epoch": 1.287078857421875e-05,
      "model_forward_time": 1.7492749691009521,
      "step": 87496
    },
    {
      "epoch": 1.287078857421875e-05,
      "step": 87496,
      "training_step_time": 7.405139207839966
    },
    {
      "epoch": 1.2871742248535156e-05,
      "model_forward_time": 1.745518684387207,
      "step": 87497
    },
    {
      "epoch": 1.2871742248535156e-05,
      "step": 87497,
      "training_step_time": 7.3475048542022705
    },
    {
      "epoch": 1.2872695922851562e-05,
      "model_forward_time": 1.7712912559509277,
      "step": 87498
    },
    {
      "epoch": 1.2872695922851562e-05,
      "step": 87498,
      "training_step_time": 8.886483192443848
    },
    {
      "epoch": 1.287364959716797e-05,
      "model_forward_time": 1.7474112510681152,
      "step": 87499
    },
    {
      "epoch": 1.287364959716797e-05,
      "step": 87499,
      "training_step_time": 7.635884523391724
    },
    {
      "epoch": 1.2874603271484375e-05,
      "grad_norm": 0.09004400670528412,
      "learning_rate": 4.211997883544472e-07,
      "loss": 0.0548,
      "step": 87500
    },
    {
      "epoch": 1.2874603271484375e-05,
      "model_forward_time": 1.7323172092437744,
      "step": 87500
    },
    {
      "dynamics_loss_avg": 0.05081669483333826,
      "epoch": 1.2874603271484375e-05,
      "step": 87500
    },
    {
      "action_loss_avg": 0.0016102979119750672,
      "epoch": 1.2874603271484375e-05,
      "step": 87500
    },
    {
      "epoch": 1.2874603271484375e-05,
      "step": 87500,
      "training_step_time": 8.201112508773804
    },
    {
      "epoch": 1.287555694580078e-05,
      "model_forward_time": 2.414210557937622,
      "step": 87501
    },
    {
      "epoch": 1.287555694580078e-05,
      "step": 87501,
      "training_step_time": 8.370759963989258
    },
    {
      "epoch": 1.2876510620117188e-05,
      "model_forward_time": 1.7466933727264404,
      "step": 87502
    },
    {
      "epoch": 1.2876510620117188e-05,
      "step": 87502,
      "training_step_time": 7.781020164489746
    },
    {
      "epoch": 1.2877464294433594e-05,
      "model_forward_time": 1.7431814670562744,
      "step": 87503
    },
    {
      "epoch": 1.2877464294433594e-05,
      "step": 87503,
      "training_step_time": 7.635179281234741
    },
    {
      "epoch": 1.287841796875e-05,
      "model_forward_time": 1.7487201690673828,
      "step": 87504
    },
    {
      "epoch": 1.287841796875e-05,
      "step": 87504,
      "training_step_time": 8.49311375617981
    },
    {
      "epoch": 1.2879371643066407e-05,
      "model_forward_time": 1.7567975521087646,
      "step": 87505
    },
    {
      "epoch": 1.2879371643066407e-05,
      "step": 87505,
      "training_step_time": 7.3682661056518555
    },
    {
      "epoch": 1.2880325317382813e-05,
      "model_forward_time": 1.7575547695159912,
      "step": 87506
    },
    {
      "epoch": 1.2880325317382813e-05,
      "step": 87506,
      "training_step_time": 7.4904398918151855
    },
    {
      "epoch": 1.2881278991699218e-05,
      "model_forward_time": 1.7372617721557617,
      "step": 87507
    },
    {
      "epoch": 1.2881278991699218e-05,
      "step": 87507,
      "training_step_time": 7.4268529415130615
    },
    {
      "epoch": 1.2882232666015624e-05,
      "model_forward_time": 1.7425308227539062,
      "step": 87508
    },
    {
      "epoch": 1.2882232666015624e-05,
      "step": 87508,
      "training_step_time": 7.653489589691162
    },
    {
      "epoch": 1.2883186340332032e-05,
      "model_forward_time": 1.7561593055725098,
      "step": 87509
    },
    {
      "epoch": 1.2883186340332032e-05,
      "step": 87509,
      "training_step_time": 8.558628559112549
    },
    {
      "epoch": 1.2884140014648437e-05,
      "grad_norm": 0.09540453553199768,
      "learning_rate": 4.2053579739790964e-07,
      "loss": 0.0553,
      "step": 87510
    },
    {
      "epoch": 1.2884140014648437e-05,
      "model_forward_time": 2.382819414138794,
      "step": 87510
    },
    {
      "dynamics_loss_avg": 0.06174561195075512,
      "epoch": 1.2884140014648437e-05,
      "step": 87510
    },
    {
      "action_loss_avg": 0.001930286327842623,
      "epoch": 1.2884140014648437e-05,
      "step": 87510
    },
    {
      "epoch": 1.2884140014648437e-05,
      "step": 87510,
      "training_step_time": 9.197407007217407
    },
    {
      "epoch": 1.2885093688964843e-05,
      "model_forward_time": 1.7532665729522705,
      "step": 87511
    },
    {
      "epoch": 1.2885093688964843e-05,
      "step": 87511,
      "training_step_time": 7.773087739944458
    },
    {
      "epoch": 1.288604736328125e-05,
      "model_forward_time": 1.745969533920288,
      "step": 87512
    },
    {
      "epoch": 1.288604736328125e-05,
      "step": 87512,
      "training_step_time": 8.689390659332275
    },
    {
      "epoch": 1.2887001037597656e-05,
      "model_forward_time": 1.7375140190124512,
      "step": 87513
    },
    {
      "epoch": 1.2887001037597656e-05,
      "step": 87513,
      "training_step_time": 7.664207935333252
    },
    {
      "epoch": 1.2887954711914062e-05,
      "model_forward_time": 1.7496318817138672,
      "step": 87514
    },
    {
      "epoch": 1.2887954711914062e-05,
      "step": 87514,
      "training_step_time": 7.3906309604644775
    },
    {
      "epoch": 1.288890838623047e-05,
      "model_forward_time": 1.7477004528045654,
      "step": 87515
    },
    {
      "epoch": 1.288890838623047e-05,
      "step": 87515,
      "training_step_time": 7.576140642166138
    },
    {
      "epoch": 1.2889862060546875e-05,
      "model_forward_time": 1.7482893466949463,
      "step": 87516
    },
    {
      "epoch": 1.2889862060546875e-05,
      "step": 87516,
      "training_step_time": 8.105565309524536
    },
    {
      "epoch": 1.289081573486328e-05,
      "model_forward_time": 1.3643758296966553,
      "step": 87517
    },
    {
      "epoch": 1.289081573486328e-05,
      "step": 87517,
      "training_step_time": 7.623786926269531
    },
    {
      "epoch": 1.2891769409179688e-05,
      "model_forward_time": 1.7461423873901367,
      "step": 87518
    },
    {
      "epoch": 1.2891769409179688e-05,
      "step": 87518,
      "training_step_time": 8.957927227020264
    },
    {
      "epoch": 1.2892723083496094e-05,
      "model_forward_time": 1.8082239627838135,
      "step": 87519
    },
    {
      "epoch": 1.2892723083496094e-05,
      "step": 87519,
      "training_step_time": 10.606187582015991
    },
    {
      "epoch": 1.28936767578125e-05,
      "grad_norm": 0.09584637731313705,
      "learning_rate": 4.1987230724469254e-07,
      "loss": 0.0561,
      "step": 87520
    },
    {
      "epoch": 1.28936767578125e-05,
      "model_forward_time": 2.4173195362091064,
      "step": 87520
    },
    {
      "dynamics_loss_avg": 0.06712779253721238,
      "epoch": 1.28936767578125e-05,
      "step": 87520
    },
    {
      "action_loss_avg": 0.0012203379243146627,
      "epoch": 1.28936767578125e-05,
      "step": 87520
    },
    {
      "epoch": 1.28936767578125e-05,
      "step": 87520,
      "training_step_time": 7.8069257736206055
    },
    {
      "epoch": 1.2894630432128907e-05,
      "model_forward_time": 1.7397470474243164,
      "step": 87521
    },
    {
      "epoch": 1.2894630432128907e-05,
      "step": 87521,
      "training_step_time": 8.22026777267456
    },
    {
      "epoch": 1.2895584106445313e-05,
      "model_forward_time": 1.752826452255249,
      "step": 87522
    },
    {
      "epoch": 1.2895584106445313e-05,
      "step": 87522,
      "training_step_time": 7.604241847991943
    },
    {
      "epoch": 1.2896537780761718e-05,
      "model_forward_time": 1.7438557147979736,
      "step": 87523
    },
    {
      "epoch": 1.2896537780761718e-05,
      "step": 87523,
      "training_step_time": 7.339749574661255
    },
    {
      "epoch": 1.2897491455078124e-05,
      "model_forward_time": 1.7402942180633545,
      "step": 87524
    },
    {
      "epoch": 1.2897491455078124e-05,
      "step": 87524,
      "training_step_time": 7.5378639698028564
    },
    {
      "epoch": 1.2898445129394532e-05,
      "model_forward_time": 1.7490179538726807,
      "step": 87525
    },
    {
      "epoch": 1.2898445129394532e-05,
      "step": 87525,
      "training_step_time": 7.466811180114746
    },
    {
      "epoch": 1.2899398803710937e-05,
      "model_forward_time": 1.7631573677062988,
      "step": 87526
    },
    {
      "epoch": 1.2899398803710937e-05,
      "step": 87526,
      "training_step_time": 7.584669351577759
    },
    {
      "epoch": 1.2900352478027343e-05,
      "model_forward_time": 1.7458794116973877,
      "step": 87527
    },
    {
      "epoch": 1.2900352478027343e-05,
      "step": 87527,
      "training_step_time": 9.300477981567383
    },
    {
      "epoch": 1.290130615234375e-05,
      "model_forward_time": 1.7501132488250732,
      "step": 87528
    },
    {
      "epoch": 1.290130615234375e-05,
      "step": 87528,
      "training_step_time": 9.571073770523071
    },
    {
      "epoch": 1.2902259826660156e-05,
      "model_forward_time": 2.3537087440490723,
      "step": 87529
    },
    {
      "epoch": 1.2902259826660156e-05,
      "step": 87529,
      "training_step_time": 8.328031301498413
    },
    {
      "epoch": 1.2903213500976562e-05,
      "grad_norm": 0.10463777929544449,
      "learning_rate": 4.192093179673534e-07,
      "loss": 0.0553,
      "step": 87530
    },
    {
      "epoch": 1.2903213500976562e-05,
      "model_forward_time": 1.734325885772705,
      "step": 87530
    },
    {
      "dynamics_loss_avg": 0.05628424547612667,
      "epoch": 1.2903213500976562e-05,
      "step": 87530
    },
    {
      "action_loss_avg": 0.0008326531475177035,
      "epoch": 1.2903213500976562e-05,
      "step": 87530
    },
    {
      "epoch": 1.2903213500976562e-05,
      "step": 87530,
      "training_step_time": 8.164056777954102
    },
    {
      "epoch": 1.290416717529297e-05,
      "model_forward_time": 1.7816636562347412,
      "step": 87531
    },
    {
      "epoch": 1.290416717529297e-05,
      "step": 87531,
      "training_step_time": 7.549904823303223
    },
    {
      "epoch": 1.2905120849609375e-05,
      "model_forward_time": 1.7375075817108154,
      "step": 87532
    },
    {
      "epoch": 1.2905120849609375e-05,
      "step": 87532,
      "training_step_time": 7.33815598487854
    },
    {
      "epoch": 1.290607452392578e-05,
      "model_forward_time": 1.7431538105010986,
      "step": 87533
    },
    {
      "epoch": 1.290607452392578e-05,
      "step": 87533,
      "training_step_time": 8.070643901824951
    },
    {
      "epoch": 1.2907028198242188e-05,
      "model_forward_time": 1.7468349933624268,
      "step": 87534
    },
    {
      "epoch": 1.2907028198242188e-05,
      "step": 87534,
      "training_step_time": 6.645618915557861
    },
    {
      "epoch": 1.2907981872558594e-05,
      "model_forward_time": 1.7452247142791748,
      "step": 87535
    },
    {
      "epoch": 1.2907981872558594e-05,
      "step": 87535,
      "training_step_time": 8.120599269866943
    },
    {
      "epoch": 1.2908935546875e-05,
      "model_forward_time": 1.7432036399841309,
      "step": 87536
    },
    {
      "epoch": 1.2908935546875e-05,
      "step": 87536,
      "training_step_time": 8.622323989868164
    },
    {
      "epoch": 1.2909889221191407e-05,
      "model_forward_time": 1.7469534873962402,
      "step": 87537
    },
    {
      "epoch": 1.2909889221191407e-05,
      "step": 87537,
      "training_step_time": 10.424875497817993
    },
    {
      "epoch": 1.2910842895507813e-05,
      "model_forward_time": 2.3675601482391357,
      "step": 87538
    },
    {
      "epoch": 1.2910842895507813e-05,
      "step": 87538,
      "training_step_time": 7.758312463760376
    },
    {
      "epoch": 1.2911796569824218e-05,
      "model_forward_time": 1.7421934604644775,
      "step": 87539
    },
    {
      "epoch": 1.2911796569824218e-05,
      "step": 87539,
      "training_step_time": 7.765054941177368
    },
    {
      "epoch": 1.2912750244140624e-05,
      "grad_norm": 0.09941708296537399,
      "learning_rate": 4.18546829638396e-07,
      "loss": 0.0545,
      "step": 87540
    },
    {
      "epoch": 1.2912750244140624e-05,
      "model_forward_time": 1.7369122505187988,
      "step": 87540
    },
    {
      "dynamics_loss_avg": 0.05037437919527292,
      "epoch": 1.2912750244140624e-05,
      "step": 87540
    },
    {
      "action_loss_avg": 0.0014635726416599936,
      "epoch": 1.2912750244140624e-05,
      "step": 87540
    },
    {
      "epoch": 1.2912750244140624e-05,
      "step": 87540,
      "training_step_time": 7.658026695251465
    },
    {
      "epoch": 1.2913703918457032e-05,
      "model_forward_time": 1.3660452365875244,
      "step": 87541
    },
    {
      "epoch": 1.2913703918457032e-05,
      "step": 87541,
      "training_step_time": 7.408557415008545
    },
    {
      "epoch": 1.2914657592773437e-05,
      "model_forward_time": 1.7433674335479736,
      "step": 87542
    },
    {
      "epoch": 1.2914657592773437e-05,
      "step": 87542,
      "training_step_time": 7.105215549468994
    },
    {
      "epoch": 1.2915611267089843e-05,
      "model_forward_time": 1.7547965049743652,
      "step": 87543
    },
    {
      "epoch": 1.2915611267089843e-05,
      "step": 87543,
      "training_step_time": 7.624613523483276
    },
    {
      "epoch": 1.291656494140625e-05,
      "model_forward_time": 1.7434792518615723,
      "step": 87544
    },
    {
      "epoch": 1.291656494140625e-05,
      "step": 87544,
      "training_step_time": 7.685837984085083
    },
    {
      "epoch": 1.2917518615722656e-05,
      "model_forward_time": 1.744905710220337,
      "step": 87545
    },
    {
      "epoch": 1.2917518615722656e-05,
      "step": 87545,
      "training_step_time": 8.469402551651001
    },
    {
      "epoch": 1.2918472290039062e-05,
      "model_forward_time": 1.7464475631713867,
      "step": 87546
    },
    {
      "epoch": 1.2918472290039062e-05,
      "step": 87546,
      "training_step_time": 9.324423551559448
    },
    {
      "epoch": 1.291942596435547e-05,
      "model_forward_time": 2.3720834255218506,
      "step": 87547
    },
    {
      "epoch": 1.291942596435547e-05,
      "step": 87547,
      "training_step_time": 8.596058368682861
    },
    {
      "epoch": 1.2920379638671875e-05,
      "model_forward_time": 1.7458412647247314,
      "step": 87548
    },
    {
      "epoch": 1.2920379638671875e-05,
      "step": 87548,
      "training_step_time": 7.753963232040405
    },
    {
      "epoch": 1.292133331298828e-05,
      "model_forward_time": 1.7440106868743896,
      "step": 87549
    },
    {
      "epoch": 1.292133331298828e-05,
      "step": 87549,
      "training_step_time": 9.090556621551514
    },
    {
      "epoch": 1.2922286987304688e-05,
      "grad_norm": 0.08689446747303009,
      "learning_rate": 4.1788484233027e-07,
      "loss": 0.0537,
      "step": 87550
    },
    {
      "epoch": 1.2922286987304688e-05,
      "model_forward_time": 1.7374646663665771,
      "step": 87550
    },
    {
      "dynamics_loss_avg": 0.04436974860727787,
      "epoch": 1.2922286987304688e-05,
      "step": 87550
    },
    {
      "action_loss_avg": 0.001182290492579341,
      "epoch": 1.2922286987304688e-05,
      "step": 87550
    },
    {
      "epoch": 1.2922286987304688e-05,
      "step": 87550,
      "training_step_time": 7.343359708786011
    },
    {
      "epoch": 1.2923240661621094e-05,
      "model_forward_time": 1.7401680946350098,
      "step": 87551
    },
    {
      "epoch": 1.2923240661621094e-05,
      "step": 87551,
      "training_step_time": 7.310473680496216
    },
    {
      "epoch": 1.29241943359375e-05,
      "model_forward_time": 1.7432162761688232,
      "step": 87552
    },
    {
      "epoch": 1.29241943359375e-05,
      "step": 87552,
      "training_step_time": 7.569499731063843
    },
    {
      "epoch": 1.2925148010253907e-05,
      "model_forward_time": 1.7378885746002197,
      "step": 87553
    },
    {
      "epoch": 1.2925148010253907e-05,
      "step": 87553,
      "training_step_time": 8.307162046432495
    },
    {
      "epoch": 1.2926101684570313e-05,
      "model_forward_time": 1.7439956665039062,
      "step": 87554
    },
    {
      "epoch": 1.2926101684570313e-05,
      "step": 87554,
      "training_step_time": 7.658424615859985
    },
    {
      "epoch": 1.2927055358886718e-05,
      "model_forward_time": 1.7489051818847656,
      "step": 87555
    },
    {
      "epoch": 1.2927055358886718e-05,
      "step": 87555,
      "training_step_time": 8.638934850692749
    },
    {
      "epoch": 1.2928009033203124e-05,
      "model_forward_time": 1.7580995559692383,
      "step": 87556
    },
    {
      "epoch": 1.2928009033203124e-05,
      "step": 87556,
      "training_step_time": 9.654677152633667
    },
    {
      "epoch": 1.2928962707519532e-05,
      "model_forward_time": 1.7482008934020996,
      "step": 87557
    },
    {
      "epoch": 1.2928962707519532e-05,
      "step": 87557,
      "training_step_time": 10.77919054031372
    },
    {
      "epoch": 1.2929916381835937e-05,
      "model_forward_time": 1.7331397533416748,
      "step": 87558
    },
    {
      "epoch": 1.2929916381835937e-05,
      "step": 87558,
      "training_step_time": 7.513876676559448
    },
    {
      "epoch": 1.2930870056152343e-05,
      "model_forward_time": 1.7532358169555664,
      "step": 87559
    },
    {
      "epoch": 1.2930870056152343e-05,
      "step": 87559,
      "training_step_time": 7.74378228187561
    },
    {
      "epoch": 1.293182373046875e-05,
      "grad_norm": 0.08780635893344879,
      "learning_rate": 4.1722335611536766e-07,
      "loss": 0.0546,
      "step": 87560
    },
    {
      "epoch": 1.293182373046875e-05,
      "model_forward_time": 1.7405571937561035,
      "step": 87560
    },
    {
      "dynamics_loss_avg": 0.04821602487936616,
      "epoch": 1.293182373046875e-05,
      "step": 87560
    },
    {
      "action_loss_avg": 0.0014582549527403898,
      "epoch": 1.293182373046875e-05,
      "step": 87560
    },
    {
      "epoch": 1.293182373046875e-05,
      "step": 87560,
      "training_step_time": 7.361407995223999
    },
    {
      "epoch": 1.2932777404785156e-05,
      "model_forward_time": 1.742445468902588,
      "step": 87561
    },
    {
      "epoch": 1.2932777404785156e-05,
      "step": 87561,
      "training_step_time": 7.556803464889526
    },
    {
      "epoch": 1.2933731079101562e-05,
      "model_forward_time": 1.736802339553833,
      "step": 87562
    },
    {
      "epoch": 1.2933731079101562e-05,
      "step": 87562,
      "training_step_time": 8.433624982833862
    },
    {
      "epoch": 1.293468475341797e-05,
      "model_forward_time": 1.758699893951416,
      "step": 87563
    },
    {
      "epoch": 1.293468475341797e-05,
      "step": 87563,
      "training_step_time": 8.268137454986572
    },
    {
      "epoch": 1.2935638427734375e-05,
      "model_forward_time": 1.775972604751587,
      "step": 87564
    },
    {
      "epoch": 1.2935638427734375e-05,
      "step": 87564,
      "training_step_time": 7.724721908569336
    },
    {
      "epoch": 1.293659210205078e-05,
      "model_forward_time": 1.7374107837677002,
      "step": 87565
    },
    {
      "epoch": 1.293659210205078e-05,
      "step": 87565,
      "training_step_time": 8.15821886062622
    },
    {
      "epoch": 1.2937545776367188e-05,
      "model_forward_time": 2.3797128200531006,
      "step": 87566
    },
    {
      "epoch": 1.2937545776367188e-05,
      "step": 87566,
      "training_step_time": 8.247127294540405
    },
    {
      "epoch": 1.2938499450683594e-05,
      "model_forward_time": 1.7467083930969238,
      "step": 87567
    },
    {
      "epoch": 1.2938499450683594e-05,
      "step": 87567,
      "training_step_time": 8.418270587921143
    },
    {
      "epoch": 1.2939453125e-05,
      "model_forward_time": 1.741943597793579,
      "step": 87568
    },
    {
      "epoch": 1.2939453125e-05,
      "step": 87568,
      "training_step_time": 8.64065957069397
    },
    {
      "epoch": 1.2940406799316407e-05,
      "model_forward_time": 1.7467341423034668,
      "step": 87569
    },
    {
      "epoch": 1.2940406799316407e-05,
      "step": 87569,
      "training_step_time": 7.394177675247192
    },
    {
      "epoch": 1.2941360473632813e-05,
      "grad_norm": 0.08139817416667938,
      "learning_rate": 4.165623710660288e-07,
      "loss": 0.0543,
      "step": 87570
    },
    {
      "epoch": 1.2941360473632813e-05,
      "model_forward_time": 1.7359182834625244,
      "step": 87570
    },
    {
      "dynamics_loss_avg": 0.052998534962534906,
      "epoch": 1.2941360473632813e-05,
      "step": 87570
    },
    {
      "action_loss_avg": 0.0015677568793762476,
      "epoch": 1.2941360473632813e-05,
      "step": 87570
    },
    {
      "epoch": 1.2941360473632813e-05,
      "step": 87570,
      "training_step_time": 6.6790266036987305
    },
    {
      "epoch": 1.2942314147949218e-05,
      "model_forward_time": 1.738016128540039,
      "step": 87571
    },
    {
      "epoch": 1.2942314147949218e-05,
      "step": 87571,
      "training_step_time": 7.373438358306885
    },
    {
      "epoch": 1.2943267822265624e-05,
      "model_forward_time": 1.747014045715332,
      "step": 87572
    },
    {
      "epoch": 1.2943267822265624e-05,
      "step": 87572,
      "training_step_time": 7.753223896026611
    },
    {
      "epoch": 1.2944221496582032e-05,
      "model_forward_time": 1.7356336116790771,
      "step": 87573
    },
    {
      "epoch": 1.2944221496582032e-05,
      "step": 87573,
      "training_step_time": 8.433377742767334
    },
    {
      "epoch": 1.2945175170898437e-05,
      "model_forward_time": 1.740105152130127,
      "step": 87574
    },
    {
      "epoch": 1.2945175170898437e-05,
      "step": 87574,
      "training_step_time": 8.6549391746521
    },
    {
      "epoch": 1.2946128845214843e-05,
      "model_forward_time": 2.3868253231048584,
      "step": 87575
    },
    {
      "epoch": 1.2946128845214843e-05,
      "step": 87575,
      "training_step_time": 8.29172945022583
    },
    {
      "epoch": 1.294708251953125e-05,
      "model_forward_time": 1.7391576766967773,
      "step": 87576
    },
    {
      "epoch": 1.294708251953125e-05,
      "step": 87576,
      "training_step_time": 7.6767966747283936
    },
    {
      "epoch": 1.2948036193847656e-05,
      "model_forward_time": 1.743420124053955,
      "step": 87577
    },
    {
      "epoch": 1.2948036193847656e-05,
      "step": 87577,
      "training_step_time": 7.658725261688232
    },
    {
      "epoch": 1.2948989868164062e-05,
      "model_forward_time": 1.7485666275024414,
      "step": 87578
    },
    {
      "epoch": 1.2948989868164062e-05,
      "step": 87578,
      "training_step_time": 7.347615957260132
    },
    {
      "epoch": 1.294994354248047e-05,
      "model_forward_time": 1.7408971786499023,
      "step": 87579
    },
    {
      "epoch": 1.294994354248047e-05,
      "step": 87579,
      "training_step_time": 6.730223655700684
    },
    {
      "epoch": 1.2950897216796875e-05,
      "grad_norm": 0.12435605376958847,
      "learning_rate": 4.159018872545384e-07,
      "loss": 0.0557,
      "step": 87580
    },
    {
      "epoch": 1.2950897216796875e-05,
      "model_forward_time": 1.733055830001831,
      "step": 87580
    },
    {
      "dynamics_loss_avg": 0.050388149358332156,
      "epoch": 1.2950897216796875e-05,
      "step": 87580
    },
    {
      "action_loss_avg": 0.0008126662083668635,
      "epoch": 1.2950897216796875e-05,
      "step": 87580
    },
    {
      "epoch": 1.2950897216796875e-05,
      "step": 87580,
      "training_step_time": 7.469186544418335
    },
    {
      "epoch": 1.295185089111328e-05,
      "model_forward_time": 1.7369093894958496,
      "step": 87581
    },
    {
      "epoch": 1.295185089111328e-05,
      "step": 87581,
      "training_step_time": 7.701873302459717
    },
    {
      "epoch": 1.2952804565429688e-05,
      "model_forward_time": 1.7794408798217773,
      "step": 87582
    },
    {
      "epoch": 1.2952804565429688e-05,
      "step": 87582,
      "training_step_time": 8.235728979110718
    },
    {
      "epoch": 1.2953758239746094e-05,
      "model_forward_time": 1.7433645725250244,
      "step": 87583
    },
    {
      "epoch": 1.2953758239746094e-05,
      "step": 87583,
      "training_step_time": 7.668756723403931
    },
    {
      "epoch": 1.29547119140625e-05,
      "model_forward_time": 2.400451183319092,
      "step": 87584
    },
    {
      "epoch": 1.29547119140625e-05,
      "step": 87584,
      "training_step_time": 7.732976198196411
    },
    {
      "epoch": 1.2955665588378907e-05,
      "model_forward_time": 1.771486520767212,
      "step": 87585
    },
    {
      "epoch": 1.2955665588378907e-05,
      "step": 87585,
      "training_step_time": 8.584976434707642
    },
    {
      "epoch": 1.2956619262695313e-05,
      "model_forward_time": 1.7618308067321777,
      "step": 87586
    },
    {
      "epoch": 1.2956619262695313e-05,
      "step": 87586,
      "training_step_time": 7.722261667251587
    },
    {
      "epoch": 1.2957572937011719e-05,
      "model_forward_time": 1.746321201324463,
      "step": 87587
    },
    {
      "epoch": 1.2957572937011719e-05,
      "step": 87587,
      "training_step_time": 8.67587924003601
    },
    {
      "epoch": 1.2958526611328124e-05,
      "model_forward_time": 1.7416818141937256,
      "step": 87588
    },
    {
      "epoch": 1.2958526611328124e-05,
      "step": 87588,
      "training_step_time": 7.444483041763306
    },
    {
      "epoch": 1.2959480285644532e-05,
      "model_forward_time": 1.7666621208190918,
      "step": 87589
    },
    {
      "epoch": 1.2959480285644532e-05,
      "step": 87589,
      "training_step_time": 7.565930128097534
    },
    {
      "epoch": 1.2960433959960937e-05,
      "grad_norm": 0.09740112721920013,
      "learning_rate": 4.152419047531259e-07,
      "loss": 0.056,
      "step": 87590
    },
    {
      "epoch": 1.2960433959960937e-05,
      "model_forward_time": 1.7354578971862793,
      "step": 87590
    },
    {
      "dynamics_loss_avg": 0.06044836919754744,
      "epoch": 1.2960433959960937e-05,
      "step": 87590
    },
    {
      "action_loss_avg": 0.0015310746239265427,
      "epoch": 1.2960433959960937e-05,
      "step": 87590
    },
    {
      "epoch": 1.2960433959960937e-05,
      "step": 87590,
      "training_step_time": 7.69482684135437
    },
    {
      "epoch": 1.2961387634277343e-05,
      "model_forward_time": 1.7427935600280762,
      "step": 87591
    },
    {
      "epoch": 1.2961387634277343e-05,
      "step": 87591,
      "training_step_time": 8.661257982254028
    },
    {
      "epoch": 1.296234130859375e-05,
      "model_forward_time": 1.7432551383972168,
      "step": 87592
    },
    {
      "epoch": 1.296234130859375e-05,
      "step": 87592,
      "training_step_time": 7.705830335617065
    },
    {
      "epoch": 1.2963294982910156e-05,
      "model_forward_time": 2.3759942054748535,
      "step": 87593
    },
    {
      "epoch": 1.2963294982910156e-05,
      "step": 87593,
      "training_step_time": 8.574903726577759
    },
    {
      "epoch": 1.2964248657226562e-05,
      "model_forward_time": 1.8189435005187988,
      "step": 87594
    },
    {
      "epoch": 1.2964248657226562e-05,
      "step": 87594,
      "training_step_time": 8.526856184005737
    },
    {
      "epoch": 1.296520233154297e-05,
      "model_forward_time": 1.7323696613311768,
      "step": 87595
    },
    {
      "epoch": 1.296520233154297e-05,
      "step": 87595,
      "training_step_time": 7.579408168792725
    },
    {
      "epoch": 1.2966156005859375e-05,
      "model_forward_time": 1.7736408710479736,
      "step": 87596
    },
    {
      "epoch": 1.2966156005859375e-05,
      "step": 87596,
      "training_step_time": 7.512174129486084
    },
    {
      "epoch": 1.296710968017578e-05,
      "model_forward_time": 1.0228896141052246,
      "step": 87597
    },
    {
      "epoch": 1.296710968017578e-05,
      "step": 87597,
      "training_step_time": 7.340209722518921
    },
    {
      "epoch": 1.2968063354492188e-05,
      "model_forward_time": 1.7516322135925293,
      "step": 87598
    },
    {
      "epoch": 1.2968063354492188e-05,
      "step": 87598,
      "training_step_time": 8.24944806098938
    },
    {
      "epoch": 1.2969017028808594e-05,
      "model_forward_time": 1.7388389110565186,
      "step": 87599
    },
    {
      "epoch": 1.2969017028808594e-05,
      "step": 87599,
      "training_step_time": 7.558935880661011
    },
    {
      "epoch": 1.2969970703125e-05,
      "grad_norm": 0.06739125400781631,
      "learning_rate": 4.1458242363396514e-07,
      "loss": 0.0544,
      "step": 87600
    },
    {
      "epoch": 1.2969970703125e-05,
      "model_forward_time": 1.7352852821350098,
      "step": 87600
    },
    {
      "dynamics_loss_avg": 0.04266163911670447,
      "epoch": 1.2969970703125e-05,
      "step": 87600
    },
    {
      "action_loss_avg": 0.0010154019721085206,
      "epoch": 1.2969970703125e-05,
      "step": 87600
    },
    {
      "epoch": 1.2969970703125e-05,
      "step": 87600,
      "training_step_time": 7.530133485794067
    },
    {
      "epoch": 1.2970924377441407e-05,
      "model_forward_time": 1.7469596862792969,
      "step": 87601
    },
    {
      "epoch": 1.2970924377441407e-05,
      "step": 87601,
      "training_step_time": 7.62732458114624
    },
    {
      "epoch": 1.2971878051757813e-05,
      "model_forward_time": 1.7417891025543213,
      "step": 87602
    },
    {
      "epoch": 1.2971878051757813e-05,
      "step": 87602,
      "training_step_time": 7.671818494796753
    },
    {
      "epoch": 1.2972831726074219e-05,
      "model_forward_time": 2.3663930892944336,
      "step": 87603
    },
    {
      "epoch": 1.2972831726074219e-05,
      "step": 87603,
      "training_step_time": 9.437291145324707
    },
    {
      "epoch": 1.2973785400390624e-05,
      "model_forward_time": 1.752476692199707,
      "step": 87604
    },
    {
      "epoch": 1.2973785400390624e-05,
      "step": 87604,
      "training_step_time": 9.012449979782104
    },
    {
      "epoch": 1.2974739074707032e-05,
      "model_forward_time": 1.7381165027618408,
      "step": 87605
    },
    {
      "epoch": 1.2974739074707032e-05,
      "step": 87605,
      "training_step_time": 7.621184349060059
    },
    {
      "epoch": 1.2975692749023437e-05,
      "model_forward_time": 1.73612380027771,
      "step": 87606
    },
    {
      "epoch": 1.2975692749023437e-05,
      "step": 87606,
      "training_step_time": 7.387862682342529
    },
    {
      "epoch": 1.2976646423339843e-05,
      "model_forward_time": 1.7524363994598389,
      "step": 87607
    },
    {
      "epoch": 1.2976646423339843e-05,
      "step": 87607,
      "training_step_time": 7.37633490562439
    },
    {
      "epoch": 1.297760009765625e-05,
      "model_forward_time": 1.7300128936767578,
      "step": 87608
    },
    {
      "epoch": 1.297760009765625e-05,
      "step": 87608,
      "training_step_time": 8.476582527160645
    },
    {
      "epoch": 1.2978553771972656e-05,
      "model_forward_time": 1.725257396697998,
      "step": 87609
    },
    {
      "epoch": 1.2978553771972656e-05,
      "step": 87609,
      "training_step_time": 7.61669135093689
    },
    {
      "epoch": 1.2979507446289062e-05,
      "grad_norm": 0.07435213029384613,
      "learning_rate": 4.1392344396917694e-07,
      "loss": 0.0544,
      "step": 87610
    },
    {
      "epoch": 1.2979507446289062e-05,
      "model_forward_time": 1.3478999137878418,
      "step": 87610
    },
    {
      "dynamics_loss_avg": 0.06418296620249749,
      "epoch": 1.2979507446289062e-05,
      "step": 87610
    },
    {
      "action_loss_avg": 0.0017418176343198867,
      "epoch": 1.2979507446289062e-05,
      "step": 87610
    },
    {
      "epoch": 1.2979507446289062e-05,
      "step": 87610,
      "training_step_time": 9.460326671600342
    },
    {
      "epoch": 1.298046112060547e-05,
      "model_forward_time": 1.7333755493164062,
      "step": 87611
    },
    {
      "epoch": 1.298046112060547e-05,
      "step": 87611,
      "training_step_time": 8.790602684020996
    },
    {
      "epoch": 1.2981414794921875e-05,
      "model_forward_time": 1.680812120437622,
      "step": 87612
    },
    {
      "epoch": 1.2981414794921875e-05,
      "step": 87612,
      "training_step_time": 8.816370010375977
    },
    {
      "epoch": 1.298236846923828e-05,
      "model_forward_time": 1.7378294467926025,
      "step": 87613
    },
    {
      "epoch": 1.298236846923828e-05,
      "step": 87613,
      "training_step_time": 7.781969308853149
    },
    {
      "epoch": 1.2983322143554688e-05,
      "model_forward_time": 1.7280755043029785,
      "step": 87614
    },
    {
      "epoch": 1.2983322143554688e-05,
      "step": 87614,
      "training_step_time": 7.6947832107543945
    },
    {
      "epoch": 1.2984275817871094e-05,
      "model_forward_time": 1.7358038425445557,
      "step": 87615
    },
    {
      "epoch": 1.2984275817871094e-05,
      "step": 87615,
      "training_step_time": 7.151628255844116
    },
    {
      "epoch": 1.29852294921875e-05,
      "model_forward_time": 1.7331757545471191,
      "step": 87616
    },
    {
      "epoch": 1.29852294921875e-05,
      "step": 87616,
      "training_step_time": 7.478113412857056
    },
    {
      "epoch": 1.2986183166503907e-05,
      "model_forward_time": 1.729511022567749,
      "step": 87617
    },
    {
      "epoch": 1.2986183166503907e-05,
      "step": 87617,
      "training_step_time": 8.94697117805481
    },
    {
      "epoch": 1.2987136840820313e-05,
      "model_forward_time": 1.737278938293457,
      "step": 87618
    },
    {
      "epoch": 1.2987136840820313e-05,
      "step": 87618,
      "training_step_time": 7.710984706878662
    },
    {
      "epoch": 1.2988090515136719e-05,
      "model_forward_time": 1.7310938835144043,
      "step": 87619
    },
    {
      "epoch": 1.2988090515136719e-05,
      "step": 87619,
      "training_step_time": 8.752037525177002
    },
    {
      "epoch": 1.2989044189453124e-05,
      "grad_norm": 0.08014565706253052,
      "learning_rate": 4.1326496583082523e-07,
      "loss": 0.0562,
      "step": 87620
    },
    {
      "epoch": 1.2989044189453124e-05,
      "model_forward_time": 1.7249493598937988,
      "step": 87620
    },
    {
      "dynamics_loss_avg": 0.04585851016454399,
      "epoch": 1.2989044189453124e-05,
      "step": 87620
    },
    {
      "action_loss_avg": 0.0009597751661203802,
      "epoch": 1.2989044189453124e-05,
      "step": 87620
    },
    {
      "epoch": 1.2989044189453124e-05,
      "step": 87620,
      "training_step_time": 9.645797491073608
    },
    {
      "epoch": 1.2989997863769532e-05,
      "model_forward_time": 1.7932395935058594,
      "step": 87621
    },
    {
      "epoch": 1.2989997863769532e-05,
      "step": 87621,
      "training_step_time": 8.739418983459473
    },
    {
      "epoch": 1.2990951538085937e-05,
      "model_forward_time": 2.3866381645202637,
      "step": 87622
    },
    {
      "epoch": 1.2990951538085937e-05,
      "step": 87622,
      "training_step_time": 7.944231748580933
    },
    {
      "epoch": 1.2991905212402343e-05,
      "model_forward_time": 1.7382075786590576,
      "step": 87623
    },
    {
      "epoch": 1.2991905212402343e-05,
      "step": 87623,
      "training_step_time": 7.840654611587524
    },
    {
      "epoch": 1.299285888671875e-05,
      "model_forward_time": 1.7371876239776611,
      "step": 87624
    },
    {
      "epoch": 1.299285888671875e-05,
      "step": 87624,
      "training_step_time": 7.199181795120239
    },
    {
      "epoch": 1.2993812561035156e-05,
      "model_forward_time": 1.731461524963379,
      "step": 87625
    },
    {
      "epoch": 1.2993812561035156e-05,
      "step": 87625,
      "training_step_time": 7.5285422801971436
    },
    {
      "epoch": 1.2994766235351562e-05,
      "model_forward_time": 1.7285308837890625,
      "step": 87626
    },
    {
      "epoch": 1.2994766235351562e-05,
      "step": 87626,
      "training_step_time": 7.578583240509033
    },
    {
      "epoch": 1.299571990966797e-05,
      "model_forward_time": 1.740583896636963,
      "step": 87627
    },
    {
      "epoch": 1.299571990966797e-05,
      "step": 87627,
      "training_step_time": 8.626556158065796
    },
    {
      "epoch": 1.2996673583984375e-05,
      "model_forward_time": 1.7346687316894531,
      "step": 87628
    },
    {
      "epoch": 1.2996673583984375e-05,
      "step": 87628,
      "training_step_time": 8.552055358886719
    },
    {
      "epoch": 1.299762725830078e-05,
      "model_forward_time": 1.7429008483886719,
      "step": 87629
    },
    {
      "epoch": 1.299762725830078e-05,
      "step": 87629,
      "training_step_time": 8.507308006286621
    },
    {
      "epoch": 1.2998580932617188e-05,
      "grad_norm": 0.11115994304418564,
      "learning_rate": 4.1260698929092127e-07,
      "loss": 0.0571,
      "step": 87630
    },
    {
      "epoch": 1.2998580932617188e-05,
      "model_forward_time": 1.7600994110107422,
      "step": 87630
    },
    {
      "dynamics_loss_avg": 0.05584104508161545,
      "epoch": 1.2998580932617188e-05,
      "step": 87630
    },
    {
      "action_loss_avg": 0.0011777073916164227,
      "epoch": 1.2998580932617188e-05,
      "step": 87630
    },
    {
      "epoch": 1.2998580932617188e-05,
      "step": 87630,
      "training_step_time": 8.501938819885254
    },
    {
      "epoch": 1.2999534606933594e-05,
      "model_forward_time": 2.382888078689575,
      "step": 87631
    },
    {
      "epoch": 1.2999534606933594e-05,
      "step": 87631,
      "training_step_time": 9.169106483459473
    },
    {
      "epoch": 1.300048828125e-05,
      "model_forward_time": 1.7549488544464111,
      "step": 87632
    },
    {
      "epoch": 1.300048828125e-05,
      "step": 87632,
      "training_step_time": 7.606288433074951
    },
    {
      "epoch": 1.3001441955566405e-05,
      "model_forward_time": 1.7324416637420654,
      "step": 87633
    },
    {
      "epoch": 1.3001441955566405e-05,
      "step": 87633,
      "training_step_time": 7.183212757110596
    },
    {
      "epoch": 1.3002395629882813e-05,
      "model_forward_time": 1.7346107959747314,
      "step": 87634
    },
    {
      "epoch": 1.3002395629882813e-05,
      "step": 87634,
      "training_step_time": 7.300861120223999
    },
    {
      "epoch": 1.3003349304199219e-05,
      "model_forward_time": 1.7380425930023193,
      "step": 87635
    },
    {
      "epoch": 1.3003349304199219e-05,
      "step": 87635,
      "training_step_time": 7.372323274612427
    },
    {
      "epoch": 1.3004302978515624e-05,
      "model_forward_time": 1.741987943649292,
      "step": 87636
    },
    {
      "epoch": 1.3004302978515624e-05,
      "step": 87636,
      "training_step_time": 7.645007848739624
    },
    {
      "epoch": 1.3005256652832032e-05,
      "model_forward_time": 1.730102300643921,
      "step": 87637
    },
    {
      "epoch": 1.3005256652832032e-05,
      "step": 87637,
      "training_step_time": 7.608435153961182
    },
    {
      "epoch": 1.3006210327148437e-05,
      "model_forward_time": 1.7322444915771484,
      "step": 87638
    },
    {
      "epoch": 1.3006210327148437e-05,
      "step": 87638,
      "training_step_time": 8.505847454071045
    },
    {
      "epoch": 1.3007164001464843e-05,
      "model_forward_time": 1.7575104236602783,
      "step": 87639
    },
    {
      "epoch": 1.3007164001464843e-05,
      "step": 87639,
      "training_step_time": 10.589892387390137
    },
    {
      "epoch": 1.300811767578125e-05,
      "grad_norm": 0.0788630023598671,
      "learning_rate": 4.119495144214192e-07,
      "loss": 0.054,
      "step": 87640
    },
    {
      "epoch": 1.300811767578125e-05,
      "model_forward_time": 2.4149467945098877,
      "step": 87640
    },
    {
      "dynamics_loss_avg": 0.04743014201521874,
      "epoch": 1.300811767578125e-05,
      "step": 87640
    },
    {
      "action_loss_avg": 0.0017256354913115501,
      "epoch": 1.300811767578125e-05,
      "step": 87640
    },
    {
      "epoch": 1.300811767578125e-05,
      "step": 87640,
      "training_step_time": 7.660192012786865
    },
    {
      "epoch": 1.3009071350097656e-05,
      "model_forward_time": 1.746582269668579,
      "step": 87641
    },
    {
      "epoch": 1.3009071350097656e-05,
      "step": 87641,
      "training_step_time": 7.596128940582275
    },
    {
      "epoch": 1.3010025024414062e-05,
      "model_forward_time": 1.7435390949249268,
      "step": 87642
    },
    {
      "epoch": 1.3010025024414062e-05,
      "step": 87642,
      "training_step_time": 7.426692008972168
    },
    {
      "epoch": 1.301097869873047e-05,
      "model_forward_time": 1.728461503982544,
      "step": 87643
    },
    {
      "epoch": 1.301097869873047e-05,
      "step": 87643,
      "training_step_time": 7.313821315765381
    },
    {
      "epoch": 1.3011932373046875e-05,
      "model_forward_time": 1.7407934665679932,
      "step": 87644
    },
    {
      "epoch": 1.3011932373046875e-05,
      "step": 87644,
      "training_step_time": 7.506806135177612
    },
    {
      "epoch": 1.301288604736328e-05,
      "model_forward_time": 1.7477853298187256,
      "step": 87645
    },
    {
      "epoch": 1.301288604736328e-05,
      "step": 87645,
      "training_step_time": 8.844761848449707
    },
    {
      "epoch": 1.3013839721679688e-05,
      "model_forward_time": 1.7466652393341064,
      "step": 87646
    },
    {
      "epoch": 1.3013839721679688e-05,
      "step": 87646,
      "training_step_time": 7.752455234527588
    },
    {
      "epoch": 1.3014793395996094e-05,
      "model_forward_time": 1.7459876537322998,
      "step": 87647
    },
    {
      "epoch": 1.3014793395996094e-05,
      "step": 87647,
      "training_step_time": 9.588059425354004
    },
    {
      "epoch": 1.30157470703125e-05,
      "model_forward_time": 1.7407474517822266,
      "step": 87648
    },
    {
      "epoch": 1.30157470703125e-05,
      "step": 87648,
      "training_step_time": 8.442954301834106
    },
    {
      "epoch": 1.3016700744628905e-05,
      "model_forward_time": 2.4120960235595703,
      "step": 87649
    },
    {
      "epoch": 1.3016700744628905e-05,
      "step": 87649,
      "training_step_time": 7.78903603553772
    },
    {
      "epoch": 1.3017654418945313e-05,
      "grad_norm": 0.08125066012144089,
      "learning_rate": 4.112925412942215e-07,
      "loss": 0.0551,
      "step": 87650
    },
    {
      "epoch": 1.3017654418945313e-05,
      "model_forward_time": 1.7821309566497803,
      "step": 87650
    },
    {
      "dynamics_loss_avg": 0.04101746389642358,
      "epoch": 1.3017654418945313e-05,
      "step": 87650
    },
    {
      "action_loss_avg": 0.0009008438180899247,
      "epoch": 1.3017654418945313e-05,
      "step": 87650
    },
    {
      "epoch": 1.3017654418945313e-05,
      "step": 87650,
      "training_step_time": 7.856796503067017
    },
    {
      "epoch": 1.3018608093261719e-05,
      "model_forward_time": 1.7341840267181396,
      "step": 87651
    },
    {
      "epoch": 1.3018608093261719e-05,
      "step": 87651,
      "training_step_time": 7.692213773727417
    },
    {
      "epoch": 1.3019561767578124e-05,
      "model_forward_time": 1.7399377822875977,
      "step": 87652
    },
    {
      "epoch": 1.3019561767578124e-05,
      "step": 87652,
      "training_step_time": 7.288114070892334
    },
    {
      "epoch": 1.3020515441894532e-05,
      "model_forward_time": 1.7444441318511963,
      "step": 87653
    },
    {
      "epoch": 1.3020515441894532e-05,
      "step": 87653,
      "training_step_time": 7.504324913024902
    },
    {
      "epoch": 1.3021469116210937e-05,
      "model_forward_time": 1.7265312671661377,
      "step": 87654
    },
    {
      "epoch": 1.3021469116210937e-05,
      "step": 87654,
      "training_step_time": 8.466823816299438
    },
    {
      "epoch": 1.3022422790527343e-05,
      "model_forward_time": 1.7257933616638184,
      "step": 87655
    },
    {
      "epoch": 1.3022422790527343e-05,
      "step": 87655,
      "training_step_time": 7.718465566635132
    },
    {
      "epoch": 1.302337646484375e-05,
      "model_forward_time": 1.7515428066253662,
      "step": 87656
    },
    {
      "epoch": 1.302337646484375e-05,
      "step": 87656,
      "training_step_time": 9.718772411346436
    },
    {
      "epoch": 1.3024330139160156e-05,
      "model_forward_time": 1.7356312274932861,
      "step": 87657
    },
    {
      "epoch": 1.3024330139160156e-05,
      "step": 87657,
      "training_step_time": 8.30487847328186
    },
    {
      "epoch": 1.3025283813476562e-05,
      "model_forward_time": 2.3617331981658936,
      "step": 87658
    },
    {
      "epoch": 1.3025283813476562e-05,
      "step": 87658,
      "training_step_time": 9.456382513046265
    },
    {
      "epoch": 1.302623748779297e-05,
      "model_forward_time": 1.7328212261199951,
      "step": 87659
    },
    {
      "epoch": 1.302623748779297e-05,
      "step": 87659,
      "training_step_time": 7.841490268707275
    },
    {
      "epoch": 1.3027191162109375e-05,
      "grad_norm": 0.08596568554639816,
      "learning_rate": 4.106360699811712e-07,
      "loss": 0.0544,
      "step": 87660
    },
    {
      "epoch": 1.3027191162109375e-05,
      "model_forward_time": 1.721224069595337,
      "step": 87660
    },
    {
      "dynamics_loss_avg": 0.04662205092608929,
      "epoch": 1.3027191162109375e-05,
      "step": 87660
    },
    {
      "action_loss_avg": 0.0010294193183653989,
      "epoch": 1.3027191162109375e-05,
      "step": 87660
    },
    {
      "epoch": 1.3027191162109375e-05,
      "step": 87660,
      "training_step_time": 7.7212135791778564
    },
    {
      "epoch": 1.3028144836425781e-05,
      "model_forward_time": 1.7353055477142334,
      "step": 87661
    },
    {
      "epoch": 1.3028144836425781e-05,
      "step": 87661,
      "training_step_time": 7.706620216369629
    },
    {
      "epoch": 1.3029098510742188e-05,
      "model_forward_time": 1.7227282524108887,
      "step": 87662
    },
    {
      "epoch": 1.3029098510742188e-05,
      "step": 87662,
      "training_step_time": 7.506015062332153
    },
    {
      "epoch": 1.3030052185058594e-05,
      "model_forward_time": 1.7335786819458008,
      "step": 87663
    },
    {
      "epoch": 1.3030052185058594e-05,
      "step": 87663,
      "training_step_time": 8.243464708328247
    },
    {
      "epoch": 1.3031005859375e-05,
      "model_forward_time": 1.7389862537384033,
      "step": 87664
    },
    {
      "epoch": 1.3031005859375e-05,
      "step": 87664,
      "training_step_time": 7.793392896652222
    },
    {
      "epoch": 1.3031959533691405e-05,
      "model_forward_time": 1.7416207790374756,
      "step": 87665
    },
    {
      "epoch": 1.3031959533691405e-05,
      "step": 87665,
      "training_step_time": 9.435513973236084
    },
    {
      "epoch": 1.3032913208007813e-05,
      "model_forward_time": 1.7456374168395996,
      "step": 87666
    },
    {
      "epoch": 1.3032913208007813e-05,
      "step": 87666,
      "training_step_time": 8.690983772277832
    },
    {
      "epoch": 1.3033866882324219e-05,
      "model_forward_time": 1.7442944049835205,
      "step": 87667
    },
    {
      "epoch": 1.3033866882324219e-05,
      "step": 87667,
      "training_step_time": 10.050642490386963
    },
    {
      "epoch": 1.3034820556640624e-05,
      "model_forward_time": 1.7330036163330078,
      "step": 87668
    },
    {
      "epoch": 1.3034820556640624e-05,
      "step": 87668,
      "training_step_time": 7.867166757583618
    },
    {
      "epoch": 1.3035774230957032e-05,
      "model_forward_time": 1.7338948249816895,
      "step": 87669
    },
    {
      "epoch": 1.3035774230957032e-05,
      "step": 87669,
      "training_step_time": 9.587490797042847
    },
    {
      "epoch": 1.3036727905273437e-05,
      "grad_norm": 0.07200463861227036,
      "learning_rate": 4.099801005540616e-07,
      "loss": 0.0539,
      "step": 87670
    },
    {
      "epoch": 1.3036727905273437e-05,
      "model_forward_time": 1.7266089916229248,
      "step": 87670
    },
    {
      "dynamics_loss_avg": 0.05322272395715118,
      "epoch": 1.3036727905273437e-05,
      "step": 87670
    },
    {
      "action_loss_avg": 0.0014494169619865716,
      "epoch": 1.3036727905273437e-05,
      "step": 87670
    },
    {
      "epoch": 1.3036727905273437e-05,
      "step": 87670,
      "training_step_time": 7.24034309387207
    },
    {
      "epoch": 1.3037681579589843e-05,
      "model_forward_time": 1.7377080917358398,
      "step": 87671
    },
    {
      "epoch": 1.3037681579589843e-05,
      "step": 87671,
      "training_step_time": 7.581246376037598
    },
    {
      "epoch": 1.303863525390625e-05,
      "model_forward_time": 1.7291948795318604,
      "step": 87672
    },
    {
      "epoch": 1.303863525390625e-05,
      "step": 87672,
      "training_step_time": 8.651636838912964
    },
    {
      "epoch": 1.3039588928222656e-05,
      "model_forward_time": 1.7232427597045898,
      "step": 87673
    },
    {
      "epoch": 1.3039588928222656e-05,
      "step": 87673,
      "training_step_time": 8.718802213668823
    },
    {
      "epoch": 1.3040542602539062e-05,
      "model_forward_time": 1.7377266883850098,
      "step": 87674
    },
    {
      "epoch": 1.3040542602539062e-05,
      "step": 87674,
      "training_step_time": 7.8038434982299805
    },
    {
      "epoch": 1.304149627685547e-05,
      "model_forward_time": 1.7335293292999268,
      "step": 87675
    },
    {
      "epoch": 1.304149627685547e-05,
      "step": 87675,
      "training_step_time": 10.672175645828247
    },
    {
      "epoch": 1.3042449951171875e-05,
      "model_forward_time": 1.73490571975708,
      "step": 87676
    },
    {
      "epoch": 1.3042449951171875e-05,
      "step": 87676,
      "training_step_time": 9.561272859573364
    },
    {
      "epoch": 1.3043403625488281e-05,
      "model_forward_time": 2.351247549057007,
      "step": 87677
    },
    {
      "epoch": 1.3043403625488281e-05,
      "step": 87677,
      "training_step_time": 8.899118900299072
    },
    {
      "epoch": 1.3044357299804688e-05,
      "model_forward_time": 1.7293527126312256,
      "step": 87678
    },
    {
      "epoch": 1.3044357299804688e-05,
      "step": 87678,
      "training_step_time": 7.844770193099976
    },
    {
      "epoch": 1.3045310974121094e-05,
      "model_forward_time": 1.731088638305664,
      "step": 87679
    },
    {
      "epoch": 1.3045310974121094e-05,
      "step": 87679,
      "training_step_time": 7.8027966022491455
    },
    {
      "epoch": 1.30462646484375e-05,
      "grad_norm": 0.08352740854024887,
      "learning_rate": 4.0932463308462567e-07,
      "loss": 0.0534,
      "step": 87680
    },
    {
      "epoch": 1.30462646484375e-05,
      "model_forward_time": 1.718247890472412,
      "step": 87680
    },
    {
      "dynamics_loss_avg": 0.060367292165756224,
      "epoch": 1.30462646484375e-05,
      "step": 87680
    },
    {
      "action_loss_avg": 0.001875470398226753,
      "epoch": 1.30462646484375e-05,
      "step": 87680
    },
    {
      "epoch": 1.30462646484375e-05,
      "step": 87680,
      "training_step_time": 7.552168846130371
    },
    {
      "epoch": 1.3047218322753905e-05,
      "model_forward_time": 1.735365867614746,
      "step": 87681
    },
    {
      "epoch": 1.3047218322753905e-05,
      "step": 87681,
      "training_step_time": 7.6320202350616455
    },
    {
      "epoch": 1.3048171997070313e-05,
      "model_forward_time": 1.728419303894043,
      "step": 87682
    },
    {
      "epoch": 1.3048171997070313e-05,
      "step": 87682,
      "training_step_time": 7.920198440551758
    },
    {
      "epoch": 1.3049125671386719e-05,
      "model_forward_time": 1.7299573421478271,
      "step": 87683
    },
    {
      "epoch": 1.3049125671386719e-05,
      "step": 87683,
      "training_step_time": 7.792525053024292
    },
    {
      "epoch": 1.3050079345703124e-05,
      "model_forward_time": 1.7473053932189941,
      "step": 87684
    },
    {
      "epoch": 1.3050079345703124e-05,
      "step": 87684,
      "training_step_time": 8.81563663482666
    },
    {
      "epoch": 1.3051033020019532e-05,
      "model_forward_time": 1.7237868309020996,
      "step": 87685
    },
    {
      "epoch": 1.3051033020019532e-05,
      "step": 87685,
      "training_step_time": 7.780513763427734
    },
    {
      "epoch": 1.3051986694335937e-05,
      "model_forward_time": 2.330810785293579,
      "step": 87686
    },
    {
      "epoch": 1.3051986694335937e-05,
      "step": 87686,
      "training_step_time": 7.784746885299683
    },
    {
      "epoch": 1.3052940368652343e-05,
      "model_forward_time": 1.7316226959228516,
      "step": 87687
    },
    {
      "epoch": 1.3052940368652343e-05,
      "step": 87687,
      "training_step_time": 7.862624883651733
    },
    {
      "epoch": 1.305389404296875e-05,
      "model_forward_time": 1.7190282344818115,
      "step": 87688
    },
    {
      "epoch": 1.305389404296875e-05,
      "step": 87688,
      "training_step_time": 7.879631280899048
    },
    {
      "epoch": 1.3054847717285156e-05,
      "model_forward_time": 1.344127893447876,
      "step": 87689
    },
    {
      "epoch": 1.3054847717285156e-05,
      "step": 87689,
      "training_step_time": 7.478101491928101
    },
    {
      "epoch": 1.3055801391601562e-05,
      "grad_norm": 0.09205618500709534,
      "learning_rate": 4.0866966764454673e-07,
      "loss": 0.0538,
      "step": 87690
    },
    {
      "epoch": 1.3055801391601562e-05,
      "model_forward_time": 1.725372076034546,
      "step": 87690
    },
    {
      "dynamics_loss_avg": 0.06568478718400002,
      "epoch": 1.3055801391601562e-05,
      "step": 87690
    },
    {
      "action_loss_avg": 0.001210388954496011,
      "epoch": 1.3055801391601562e-05,
      "step": 87690
    },
    {
      "epoch": 1.3055801391601562e-05,
      "step": 87690,
      "training_step_time": 7.540229797363281
    },
    {
      "epoch": 1.305675506591797e-05,
      "model_forward_time": 1.7322449684143066,
      "step": 87691
    },
    {
      "epoch": 1.305675506591797e-05,
      "step": 87691,
      "training_step_time": 8.686675071716309
    },
    {
      "epoch": 1.3057708740234375e-05,
      "model_forward_time": 1.7232270240783691,
      "step": 87692
    },
    {
      "epoch": 1.3057708740234375e-05,
      "step": 87692,
      "training_step_time": 7.6150407791137695
    },
    {
      "epoch": 1.3058662414550781e-05,
      "model_forward_time": 1.741483211517334,
      "step": 87693
    },
    {
      "epoch": 1.3058662414550781e-05,
      "step": 87693,
      "training_step_time": 10.213780641555786
    },
    {
      "epoch": 1.3059616088867188e-05,
      "model_forward_time": 1.7294836044311523,
      "step": 87694
    },
    {
      "epoch": 1.3059616088867188e-05,
      "step": 87694,
      "training_step_time": 7.692704916000366
    },
    {
      "epoch": 1.3060569763183594e-05,
      "model_forward_time": 2.3430747985839844,
      "step": 87695
    },
    {
      "epoch": 1.3060569763183594e-05,
      "step": 87695,
      "training_step_time": 8.709510087966919
    },
    {
      "epoch": 1.30615234375e-05,
      "model_forward_time": 1.7276744842529297,
      "step": 87696
    },
    {
      "epoch": 1.30615234375e-05,
      "step": 87696,
      "training_step_time": 7.651023864746094
    },
    {
      "epoch": 1.3062477111816405e-05,
      "model_forward_time": 1.736029863357544,
      "step": 87697
    },
    {
      "epoch": 1.3062477111816405e-05,
      "step": 87697,
      "training_step_time": 7.71964693069458
    },
    {
      "epoch": 1.3063430786132813e-05,
      "model_forward_time": 1.7311677932739258,
      "step": 87698
    },
    {
      "epoch": 1.3063430786132813e-05,
      "step": 87698,
      "training_step_time": 8.007399320602417
    },
    {
      "epoch": 1.3064384460449219e-05,
      "model_forward_time": 1.7412865161895752,
      "step": 87699
    },
    {
      "epoch": 1.3064384460449219e-05,
      "step": 87699,
      "training_step_time": 7.36393666267395
    },
    {
      "epoch": 1.3065338134765624e-05,
      "grad_norm": 0.08832624554634094,
      "learning_rate": 4.080152043054492e-07,
      "loss": 0.0543,
      "step": 87700
    },
    {
      "epoch": 1.3065338134765624e-05,
      "model_forward_time": 1.724025011062622,
      "step": 87700
    },
    {
      "dynamics_loss_avg": 0.05422387300059199,
      "epoch": 1.3065338134765624e-05,
      "step": 87700
    },
    {
      "action_loss_avg": 0.0018308710248675197,
      "epoch": 1.3065338134765624e-05,
      "step": 87700
    },
    {
      "epoch": 1.3065338134765624e-05,
      "step": 87700,
      "training_step_time": 7.641748905181885
    },
    {
      "epoch": 1.3066291809082032e-05,
      "model_forward_time": 1.7271521091461182,
      "step": 87701
    },
    {
      "epoch": 1.3066291809082032e-05,
      "step": 87701,
      "training_step_time": 7.655462026596069
    },
    {
      "epoch": 1.3067245483398437e-05,
      "model_forward_time": 1.7405407428741455,
      "step": 87702
    },
    {
      "epoch": 1.3067245483398437e-05,
      "step": 87702,
      "training_step_time": 7.697764158248901
    },
    {
      "epoch": 1.3068199157714843e-05,
      "model_forward_time": 1.7346761226654053,
      "step": 87703
    },
    {
      "epoch": 1.3068199157714843e-05,
      "step": 87703,
      "training_step_time": 9.23535704612732
    },
    {
      "epoch": 1.306915283203125e-05,
      "model_forward_time": 2.324765682220459,
      "step": 87704
    },
    {
      "epoch": 1.306915283203125e-05,
      "step": 87704,
      "training_step_time": 10.813238859176636
    },
    {
      "epoch": 1.3070106506347656e-05,
      "model_forward_time": 1.728264331817627,
      "step": 87705
    },
    {
      "epoch": 1.3070106506347656e-05,
      "step": 87705,
      "training_step_time": 7.6243226528167725
    },
    {
      "epoch": 1.3071060180664062e-05,
      "model_forward_time": 1.3468689918518066,
      "step": 87706
    },
    {
      "epoch": 1.3071060180664062e-05,
      "step": 87706,
      "training_step_time": 7.51416540145874
    },
    {
      "epoch": 1.307201385498047e-05,
      "model_forward_time": 1.7331812381744385,
      "step": 87707
    },
    {
      "epoch": 1.307201385498047e-05,
      "step": 87707,
      "training_step_time": 8.588847160339355
    },
    {
      "epoch": 1.3072967529296875e-05,
      "model_forward_time": 1.737231731414795,
      "step": 87708
    },
    {
      "epoch": 1.3072967529296875e-05,
      "step": 87708,
      "training_step_time": 7.338600158691406
    },
    {
      "epoch": 1.3073921203613281e-05,
      "model_forward_time": 1.7459547519683838,
      "step": 87709
    },
    {
      "epoch": 1.3073921203613281e-05,
      "step": 87709,
      "training_step_time": 7.580691337585449
    },
    {
      "epoch": 1.3074874877929688e-05,
      "grad_norm": 0.08678903430700302,
      "learning_rate": 4.0736124313890623e-07,
      "loss": 0.0536,
      "step": 87710
    },
    {
      "epoch": 1.3074874877929688e-05,
      "model_forward_time": 1.7476096153259277,
      "step": 87710
    },
    {
      "dynamics_loss_avg": 0.05573319960385561,
      "epoch": 1.3074874877929688e-05,
      "step": 87710
    },
    {
      "action_loss_avg": 0.0018223595281597227,
      "epoch": 1.3074874877929688e-05,
      "step": 87710
    },
    {
      "epoch": 1.3074874877929688e-05,
      "step": 87710,
      "training_step_time": 7.629682540893555
    },
    {
      "epoch": 1.3075828552246094e-05,
      "model_forward_time": 1.7274448871612549,
      "step": 87711
    },
    {
      "epoch": 1.3075828552246094e-05,
      "step": 87711,
      "training_step_time": 10.826973915100098
    },
    {
      "epoch": 1.30767822265625e-05,
      "model_forward_time": 1.7485532760620117,
      "step": 87712
    },
    {
      "epoch": 1.30767822265625e-05,
      "step": 87712,
      "training_step_time": 7.750245571136475
    },
    {
      "epoch": 1.3077735900878905e-05,
      "model_forward_time": 1.7465932369232178,
      "step": 87713
    },
    {
      "epoch": 1.3077735900878905e-05,
      "step": 87713,
      "training_step_time": 7.6656107902526855
    },
    {
      "epoch": 1.3078689575195313e-05,
      "model_forward_time": 2.385918378829956,
      "step": 87714
    },
    {
      "epoch": 1.3078689575195313e-05,
      "step": 87714,
      "training_step_time": 8.511925220489502
    },
    {
      "epoch": 1.3079643249511719e-05,
      "model_forward_time": 1.3708300590515137,
      "step": 87715
    },
    {
      "epoch": 1.3079643249511719e-05,
      "step": 87715,
      "training_step_time": 7.685640811920166
    },
    {
      "epoch": 1.3080596923828124e-05,
      "model_forward_time": 1.7530426979064941,
      "step": 87716
    },
    {
      "epoch": 1.3080596923828124e-05,
      "step": 87716,
      "training_step_time": 7.95322585105896
    },
    {
      "epoch": 1.3081550598144532e-05,
      "model_forward_time": 1.746985912322998,
      "step": 87717
    },
    {
      "epoch": 1.3081550598144532e-05,
      "step": 87717,
      "training_step_time": 7.524239778518677
    },
    {
      "epoch": 1.3082504272460937e-05,
      "model_forward_time": 1.7414653301239014,
      "step": 87718
    },
    {
      "epoch": 1.3082504272460937e-05,
      "step": 87718,
      "training_step_time": 8.051984071731567
    },
    {
      "epoch": 1.3083457946777343e-05,
      "model_forward_time": 1.7487244606018066,
      "step": 87719
    },
    {
      "epoch": 1.3083457946777343e-05,
      "step": 87719,
      "training_step_time": 9.576550483703613
    },
    {
      "epoch": 1.308441162109375e-05,
      "grad_norm": 0.0919608324766159,
      "learning_rate": 4.067077842164319e-07,
      "loss": 0.0538,
      "step": 87720
    },
    {
      "epoch": 1.308441162109375e-05,
      "model_forward_time": 1.7351603507995605,
      "step": 87720
    },
    {
      "dynamics_loss_avg": 0.041101332660764454,
      "epoch": 1.308441162109375e-05,
      "step": 87720
    },
    {
      "action_loss_avg": 0.0011467528456705622,
      "epoch": 1.308441162109375e-05,
      "step": 87720
    },
    {
      "epoch": 1.308441162109375e-05,
      "step": 87720,
      "training_step_time": 7.70876407623291
    },
    {
      "epoch": 1.3085365295410156e-05,
      "model_forward_time": 1.7729458808898926,
      "step": 87721
    },
    {
      "epoch": 1.3085365295410156e-05,
      "step": 87721,
      "training_step_time": 9.608546018600464
    },
    {
      "epoch": 1.3086318969726562e-05,
      "model_forward_time": 1.7519667148590088,
      "step": 87722
    },
    {
      "epoch": 1.3086318969726562e-05,
      "step": 87722,
      "training_step_time": 8.740405082702637
    },
    {
      "epoch": 1.308727264404297e-05,
      "model_forward_time": 2.4192159175872803,
      "step": 87723
    },
    {
      "epoch": 1.308727264404297e-05,
      "step": 87723,
      "training_step_time": 8.77484941482544
    },
    {
      "epoch": 1.3088226318359375e-05,
      "model_forward_time": 1.747450590133667,
      "step": 87724
    },
    {
      "epoch": 1.3088226318359375e-05,
      "step": 87724,
      "training_step_time": 8.644289255142212
    },
    {
      "epoch": 1.3089179992675781e-05,
      "model_forward_time": 1.7428996562957764,
      "step": 87725
    },
    {
      "epoch": 1.3089179992675781e-05,
      "step": 87725,
      "training_step_time": 7.765289545059204
    },
    {
      "epoch": 1.3090133666992188e-05,
      "model_forward_time": 1.7490060329437256,
      "step": 87726
    },
    {
      "epoch": 1.3090133666992188e-05,
      "step": 87726,
      "training_step_time": 7.374359130859375
    },
    {
      "epoch": 1.3091087341308594e-05,
      "model_forward_time": 1.7423291206359863,
      "step": 87727
    },
    {
      "epoch": 1.3091087341308594e-05,
      "step": 87727,
      "training_step_time": 7.495718002319336
    },
    {
      "epoch": 1.3092041015625e-05,
      "model_forward_time": 1.746950387954712,
      "step": 87728
    },
    {
      "epoch": 1.3092041015625e-05,
      "step": 87728,
      "training_step_time": 8.563209295272827
    },
    {
      "epoch": 1.3092994689941405e-05,
      "model_forward_time": 1.7273085117340088,
      "step": 87729
    },
    {
      "epoch": 1.3092994689941405e-05,
      "step": 87729,
      "training_step_time": 8.603944540023804
    },
    {
      "epoch": 1.3093948364257813e-05,
      "grad_norm": 0.09756175428628922,
      "learning_rate": 4.0605482760948944e-07,
      "loss": 0.0535,
      "step": 87730
    },
    {
      "epoch": 1.3093948364257813e-05,
      "model_forward_time": 1.7492315769195557,
      "step": 87730
    },
    {
      "dynamics_loss_avg": 0.052909152768552306,
      "epoch": 1.3093948364257813e-05,
      "step": 87730
    },
    {
      "action_loss_avg": 0.0013747353397775442,
      "epoch": 1.3093948364257813e-05,
      "step": 87730
    },
    {
      "epoch": 1.3093948364257813e-05,
      "step": 87730,
      "training_step_time": 7.648457765579224
    },
    {
      "epoch": 1.3094902038574219e-05,
      "model_forward_time": 1.7493486404418945,
      "step": 87731
    },
    {
      "epoch": 1.3094902038574219e-05,
      "step": 87731,
      "training_step_time": 8.56237244606018
    },
    {
      "epoch": 1.3095855712890624e-05,
      "model_forward_time": 2.3786637783050537,
      "step": 87732
    },
    {
      "epoch": 1.3095855712890624e-05,
      "step": 87732,
      "training_step_time": 9.510682821273804
    },
    {
      "epoch": 1.3096809387207032e-05,
      "model_forward_time": 1.7435824871063232,
      "step": 87733
    },
    {
      "epoch": 1.3096809387207032e-05,
      "step": 87733,
      "training_step_time": 7.782861232757568
    },
    {
      "epoch": 1.3097763061523437e-05,
      "model_forward_time": 1.7454674243927002,
      "step": 87734
    },
    {
      "epoch": 1.3097763061523437e-05,
      "step": 87734,
      "training_step_time": 7.946872234344482
    },
    {
      "epoch": 1.3098716735839843e-05,
      "model_forward_time": 1.7414093017578125,
      "step": 87735
    },
    {
      "epoch": 1.3098716735839843e-05,
      "step": 87735,
      "training_step_time": 7.431613445281982
    },
    {
      "epoch": 1.309967041015625e-05,
      "model_forward_time": 1.741744041442871,
      "step": 87736
    },
    {
      "epoch": 1.309967041015625e-05,
      "step": 87736,
      "training_step_time": 7.708234071731567
    },
    {
      "epoch": 1.3100624084472656e-05,
      "model_forward_time": 1.7395100593566895,
      "step": 87737
    },
    {
      "epoch": 1.3100624084472656e-05,
      "step": 87737,
      "training_step_time": 9.673975229263306
    },
    {
      "epoch": 1.3101577758789062e-05,
      "model_forward_time": 1.738328456878662,
      "step": 87738
    },
    {
      "epoch": 1.3101577758789062e-05,
      "step": 87738,
      "training_step_time": 7.714720249176025
    },
    {
      "epoch": 1.310253143310547e-05,
      "model_forward_time": 1.7492756843566895,
      "step": 87739
    },
    {
      "epoch": 1.310253143310547e-05,
      "step": 87739,
      "training_step_time": 7.690523386001587
    },
    {
      "epoch": 1.3103485107421875e-05,
      "grad_norm": 0.0804993212223053,
      "learning_rate": 4.0540237338948295e-07,
      "loss": 0.0542,
      "step": 87740
    },
    {
      "epoch": 1.3103485107421875e-05,
      "model_forward_time": 1.7383921146392822,
      "step": 87740
    },
    {
      "dynamics_loss_avg": 0.0636544380337,
      "epoch": 1.3103485107421875e-05,
      "step": 87740
    },
    {
      "action_loss_avg": 0.0007671946121263317,
      "epoch": 1.3103485107421875e-05,
      "step": 87740
    },
    {
      "epoch": 1.3103485107421875e-05,
      "step": 87740,
      "training_step_time": 8.879704713821411
    },
    {
      "epoch": 1.3104438781738281e-05,
      "model_forward_time": 2.417712688446045,
      "step": 87741
    },
    {
      "epoch": 1.3104438781738281e-05,
      "step": 87741,
      "training_step_time": 8.770690679550171
    },
    {
      "epoch": 1.3105392456054688e-05,
      "model_forward_time": 1.7491741180419922,
      "step": 87742
    },
    {
      "epoch": 1.3105392456054688e-05,
      "step": 87742,
      "training_step_time": 7.744795083999634
    },
    {
      "epoch": 1.3106346130371094e-05,
      "model_forward_time": 1.754303216934204,
      "step": 87743
    },
    {
      "epoch": 1.3106346130371094e-05,
      "step": 87743,
      "training_step_time": 8.599801063537598
    },
    {
      "epoch": 1.31072998046875e-05,
      "model_forward_time": 1.7458300590515137,
      "step": 87744
    },
    {
      "epoch": 1.31072998046875e-05,
      "step": 87744,
      "training_step_time": 7.3325722217559814
    },
    {
      "epoch": 1.3108253479003906e-05,
      "model_forward_time": 1.7461299896240234,
      "step": 87745
    },
    {
      "epoch": 1.3108253479003906e-05,
      "step": 87745,
      "training_step_time": 7.6304612159729
    },
    {
      "epoch": 1.3109207153320313e-05,
      "model_forward_time": 1.7468578815460205,
      "step": 87746
    },
    {
      "epoch": 1.3109207153320313e-05,
      "step": 87746,
      "training_step_time": 8.798250198364258
    },
    {
      "epoch": 1.3110160827636719e-05,
      "model_forward_time": 1.740990400314331,
      "step": 87747
    },
    {
      "epoch": 1.3110160827636719e-05,
      "step": 87747,
      "training_step_time": 8.723903894424438
    },
    {
      "epoch": 1.3111114501953124e-05,
      "model_forward_time": 1.7483444213867188,
      "step": 87748
    },
    {
      "epoch": 1.3111114501953124e-05,
      "step": 87748,
      "training_step_time": 7.887702226638794
    },
    {
      "epoch": 1.3112068176269532e-05,
      "model_forward_time": 1.763798713684082,
      "step": 87749
    },
    {
      "epoch": 1.3112068176269532e-05,
      "step": 87749,
      "training_step_time": 7.870309114456177
    },
    {
      "epoch": 1.3113021850585938e-05,
      "grad_norm": 0.08457490056753159,
      "learning_rate": 4.0475042162776535e-07,
      "loss": 0.0546,
      "step": 87750
    },
    {
      "epoch": 1.3113021850585938e-05,
      "model_forward_time": 1.7530698776245117,
      "step": 87750
    },
    {
      "dynamics_loss_avg": 0.055958321504294875,
      "epoch": 1.3113021850585938e-05,
      "step": 87750
    },
    {
      "action_loss_avg": 0.0011212143261218443,
      "epoch": 1.3113021850585938e-05,
      "step": 87750
    },
    {
      "epoch": 1.3113021850585938e-05,
      "step": 87750,
      "training_step_time": 8.476123809814453
    },
    {
      "epoch": 1.3113975524902343e-05,
      "model_forward_time": 1.7539870738983154,
      "step": 87751
    },
    {
      "epoch": 1.3113975524902343e-05,
      "step": 87751,
      "training_step_time": 8.512725353240967
    },
    {
      "epoch": 1.311492919921875e-05,
      "model_forward_time": 1.7476603984832764,
      "step": 87752
    },
    {
      "epoch": 1.311492919921875e-05,
      "step": 87752,
      "training_step_time": 8.023091316223145
    },
    {
      "epoch": 1.3115882873535156e-05,
      "model_forward_time": 1.7460849285125732,
      "step": 87753
    },
    {
      "epoch": 1.3115882873535156e-05,
      "step": 87753,
      "training_step_time": 7.883061408996582
    },
    {
      "epoch": 1.3116836547851562e-05,
      "model_forward_time": 1.7548375129699707,
      "step": 87754
    },
    {
      "epoch": 1.3116836547851562e-05,
      "step": 87754,
      "training_step_time": 7.5536415576934814
    },
    {
      "epoch": 1.311779022216797e-05,
      "model_forward_time": 1.7487564086914062,
      "step": 87755
    },
    {
      "epoch": 1.311779022216797e-05,
      "step": 87755,
      "training_step_time": 8.349166870117188
    },
    {
      "epoch": 1.3118743896484375e-05,
      "model_forward_time": 1.7505052089691162,
      "step": 87756
    },
    {
      "epoch": 1.3118743896484375e-05,
      "step": 87756,
      "training_step_time": 8.632766008377075
    },
    {
      "epoch": 1.3119697570800781e-05,
      "model_forward_time": 1.7466773986816406,
      "step": 87757
    },
    {
      "epoch": 1.3119697570800781e-05,
      "step": 87757,
      "training_step_time": 8.459446907043457
    },
    {
      "epoch": 1.3120651245117187e-05,
      "model_forward_time": 1.7755317687988281,
      "step": 87758
    },
    {
      "epoch": 1.3120651245117187e-05,
      "step": 87758,
      "training_step_time": 7.83755350112915
    },
    {
      "epoch": 1.3121604919433594e-05,
      "model_forward_time": 1.7607238292694092,
      "step": 87759
    },
    {
      "epoch": 1.3121604919433594e-05,
      "step": 87759,
      "training_step_time": 9.034969568252563
    },
    {
      "epoch": 1.312255859375e-05,
      "grad_norm": 0.10891485959291458,
      "learning_rate": 4.0409897239563236e-07,
      "loss": 0.0532,
      "step": 87760
    },
    {
      "epoch": 1.312255859375e-05,
      "model_forward_time": 2.405660390853882,
      "step": 87760
    },
    {
      "dynamics_loss_avg": 0.04835708253085613,
      "epoch": 1.312255859375e-05,
      "step": 87760
    },
    {
      "action_loss_avg": 0.0011106699297670276,
      "epoch": 1.312255859375e-05,
      "step": 87760
    },
    {
      "epoch": 1.312255859375e-05,
      "step": 87760,
      "training_step_time": 7.831295967102051
    },
    {
      "epoch": 1.3123512268066406e-05,
      "model_forward_time": 1.750589370727539,
      "step": 87761
    },
    {
      "epoch": 1.3123512268066406e-05,
      "step": 87761,
      "training_step_time": 8.407586336135864
    },
    {
      "epoch": 1.3124465942382813e-05,
      "model_forward_time": 1.7554895877838135,
      "step": 87762
    },
    {
      "epoch": 1.3124465942382813e-05,
      "step": 87762,
      "training_step_time": 7.76862096786499
    },
    {
      "epoch": 1.3125419616699219e-05,
      "model_forward_time": 1.7483298778533936,
      "step": 87763
    },
    {
      "epoch": 1.3125419616699219e-05,
      "step": 87763,
      "training_step_time": 7.502522706985474
    },
    {
      "epoch": 1.3126373291015624e-05,
      "model_forward_time": 1.7426724433898926,
      "step": 87764
    },
    {
      "epoch": 1.3126373291015624e-05,
      "step": 87764,
      "training_step_time": 7.661188364028931
    },
    {
      "epoch": 1.3127326965332032e-05,
      "model_forward_time": 1.7552449703216553,
      "step": 87765
    },
    {
      "epoch": 1.3127326965332032e-05,
      "step": 87765,
      "training_step_time": 7.760391473770142
    },
    {
      "epoch": 1.3128280639648438e-05,
      "model_forward_time": 1.7534770965576172,
      "step": 87766
    },
    {
      "epoch": 1.3128280639648438e-05,
      "step": 87766,
      "training_step_time": 9.459404468536377
    },
    {
      "epoch": 1.3129234313964843e-05,
      "model_forward_time": 1.7804770469665527,
      "step": 87767
    },
    {
      "epoch": 1.3129234313964843e-05,
      "step": 87767,
      "training_step_time": 8.750056266784668
    },
    {
      "epoch": 1.313018798828125e-05,
      "model_forward_time": 1.7471506595611572,
      "step": 87768
    },
    {
      "epoch": 1.313018798828125e-05,
      "step": 87768,
      "training_step_time": 9.738888502120972
    },
    {
      "epoch": 1.3131141662597656e-05,
      "model_forward_time": 2.425884962081909,
      "step": 87769
    },
    {
      "epoch": 1.3131141662597656e-05,
      "step": 87769,
      "training_step_time": 7.839492321014404
    },
    {
      "epoch": 1.3132095336914062e-05,
      "grad_norm": 0.08161750435829163,
      "learning_rate": 4.0344802576432697e-07,
      "loss": 0.0536,
      "step": 87770
    },
    {
      "epoch": 1.3132095336914062e-05,
      "model_forward_time": 1.740553855895996,
      "step": 87770
    },
    {
      "dynamics_loss_avg": 0.03634024197235704,
      "epoch": 1.3132095336914062e-05,
      "step": 87770
    },
    {
      "action_loss_avg": 0.0011302174170850776,
      "epoch": 1.3132095336914062e-05,
      "step": 87770
    },
    {
      "epoch": 1.3132095336914062e-05,
      "step": 87770,
      "training_step_time": 9.282642841339111
    },
    {
      "epoch": 1.313304901123047e-05,
      "model_forward_time": 1.7456285953521729,
      "step": 87771
    },
    {
      "epoch": 1.313304901123047e-05,
      "step": 87771,
      "training_step_time": 8.882551431655884
    },
    {
      "epoch": 1.3134002685546875e-05,
      "model_forward_time": 1.7507054805755615,
      "step": 87772
    },
    {
      "epoch": 1.3134002685546875e-05,
      "step": 87772,
      "training_step_time": 8.232502937316895
    },
    {
      "epoch": 1.3134956359863281e-05,
      "model_forward_time": 1.7494797706604004,
      "step": 87773
    },
    {
      "epoch": 1.3134956359863281e-05,
      "step": 87773,
      "training_step_time": 9.462470769882202
    },
    {
      "epoch": 1.3135910034179687e-05,
      "model_forward_time": 1.7512531280517578,
      "step": 87774
    },
    {
      "epoch": 1.3135910034179687e-05,
      "step": 87774,
      "training_step_time": 8.738462448120117
    },
    {
      "epoch": 1.3136863708496094e-05,
      "model_forward_time": 1.7451109886169434,
      "step": 87775
    },
    {
      "epoch": 1.3136863708496094e-05,
      "step": 87775,
      "training_step_time": 7.694354057312012
    },
    {
      "epoch": 1.31378173828125e-05,
      "model_forward_time": 1.7606415748596191,
      "step": 87776
    },
    {
      "epoch": 1.31378173828125e-05,
      "step": 87776,
      "training_step_time": 9.575376272201538
    },
    {
      "epoch": 1.3138771057128906e-05,
      "model_forward_time": 1.7466304302215576,
      "step": 87777
    },
    {
      "epoch": 1.3138771057128906e-05,
      "step": 87777,
      "training_step_time": 9.312333583831787
    },
    {
      "epoch": 1.3139724731445313e-05,
      "model_forward_time": 2.388947010040283,
      "step": 87778
    },
    {
      "epoch": 1.3139724731445313e-05,
      "step": 87778,
      "training_step_time": 7.836645841598511
    },
    {
      "epoch": 1.3140678405761719e-05,
      "model_forward_time": 1.7404515743255615,
      "step": 87779
    },
    {
      "epoch": 1.3140678405761719e-05,
      "step": 87779,
      "training_step_time": 7.790421724319458
    },
    {
      "epoch": 1.3141632080078124e-05,
      "grad_norm": 0.08360087871551514,
      "learning_rate": 4.027975818050339e-07,
      "loss": 0.054,
      "step": 87780
    },
    {
      "epoch": 1.3141632080078124e-05,
      "model_forward_time": 1.7445650100708008,
      "step": 87780
    },
    {
      "dynamics_loss_avg": 0.05754418820142746,
      "epoch": 1.3141632080078124e-05,
      "step": 87780
    },
    {
      "action_loss_avg": 0.0014698664541356265,
      "epoch": 1.3141632080078124e-05,
      "step": 87780
    },
    {
      "epoch": 1.3141632080078124e-05,
      "step": 87780,
      "training_step_time": 7.769159555435181
    },
    {
      "epoch": 1.3142585754394532e-05,
      "model_forward_time": 1.7494206428527832,
      "step": 87781
    },
    {
      "epoch": 1.3142585754394532e-05,
      "step": 87781,
      "training_step_time": 7.743514060974121
    },
    {
      "epoch": 1.3143539428710938e-05,
      "model_forward_time": 1.751729965209961,
      "step": 87782
    },
    {
      "epoch": 1.3143539428710938e-05,
      "step": 87782,
      "training_step_time": 7.694059610366821
    },
    {
      "epoch": 1.3144493103027343e-05,
      "model_forward_time": 1.7463152408599854,
      "step": 87783
    },
    {
      "epoch": 1.3144493103027343e-05,
      "step": 87783,
      "training_step_time": 8.776578664779663
    },
    {
      "epoch": 1.314544677734375e-05,
      "model_forward_time": 1.743678092956543,
      "step": 87784
    },
    {
      "epoch": 1.314544677734375e-05,
      "step": 87784,
      "training_step_time": 8.18867015838623
    },
    {
      "epoch": 1.3146400451660156e-05,
      "model_forward_time": 1.7742035388946533,
      "step": 87785
    },
    {
      "epoch": 1.3146400451660156e-05,
      "step": 87785,
      "training_step_time": 8.762471675872803
    },
    {
      "epoch": 1.3147354125976562e-05,
      "model_forward_time": 1.7398455142974854,
      "step": 87786
    },
    {
      "epoch": 1.3147354125976562e-05,
      "step": 87786,
      "training_step_time": 8.36956000328064
    },
    {
      "epoch": 1.314830780029297e-05,
      "model_forward_time": 2.377681255340576,
      "step": 87787
    },
    {
      "epoch": 1.314830780029297e-05,
      "step": 87787,
      "training_step_time": 7.859344959259033
    },
    {
      "epoch": 1.3149261474609375e-05,
      "model_forward_time": 1.7530205249786377,
      "step": 87788
    },
    {
      "epoch": 1.3149261474609375e-05,
      "step": 87788,
      "training_step_time": 7.763399839401245
    },
    {
      "epoch": 1.3150215148925781e-05,
      "model_forward_time": 1.7421014308929443,
      "step": 87789
    },
    {
      "epoch": 1.3150215148925781e-05,
      "step": 87789,
      "training_step_time": 7.824431419372559
    },
    {
      "epoch": 1.3151168823242187e-05,
      "grad_norm": 0.08380712568759918,
      "learning_rate": 4.021476405888858e-07,
      "loss": 0.0537,
      "step": 87790
    },
    {
      "epoch": 1.3151168823242187e-05,
      "model_forward_time": 1.7516875267028809,
      "step": 87790
    },
    {
      "dynamics_loss_avg": 0.044713287614285946,
      "epoch": 1.3151168823242187e-05,
      "step": 87790
    },
    {
      "action_loss_avg": 0.0007748083298793063,
      "epoch": 1.3151168823242187e-05,
      "step": 87790
    },
    {
      "epoch": 1.3151168823242187e-05,
      "step": 87790,
      "training_step_time": 7.707209825515747
    },
    {
      "epoch": 1.3152122497558594e-05,
      "model_forward_time": 1.3645777702331543,
      "step": 87791
    },
    {
      "epoch": 1.3152122497558594e-05,
      "step": 87791,
      "training_step_time": 7.686924695968628
    },
    {
      "epoch": 1.3153076171875e-05,
      "model_forward_time": 1.752615213394165,
      "step": 87792
    },
    {
      "epoch": 1.3153076171875e-05,
      "step": 87792,
      "training_step_time": 8.260720252990723
    },
    {
      "epoch": 1.3154029846191406e-05,
      "model_forward_time": 1.7534613609313965,
      "step": 87793
    },
    {
      "epoch": 1.3154029846191406e-05,
      "step": 87793,
      "training_step_time": 7.750115394592285
    },
    {
      "epoch": 1.3154983520507813e-05,
      "model_forward_time": 1.7581720352172852,
      "step": 87794
    },
    {
      "epoch": 1.3154983520507813e-05,
      "step": 87794,
      "training_step_time": 9.626930952072144
    },
    {
      "epoch": 1.3155937194824219e-05,
      "model_forward_time": 1.7368218898773193,
      "step": 87795
    },
    {
      "epoch": 1.3155937194824219e-05,
      "step": 87795,
      "training_step_time": 9.711121559143066
    },
    {
      "epoch": 1.3156890869140624e-05,
      "model_forward_time": 1.7539141178131104,
      "step": 87796
    },
    {
      "epoch": 1.3156890869140624e-05,
      "step": 87796,
      "training_step_time": 7.892503261566162
    },
    {
      "epoch": 1.3157844543457032e-05,
      "model_forward_time": 2.001655101776123,
      "step": 87797
    },
    {
      "epoch": 1.3157844543457032e-05,
      "step": 87797,
      "training_step_time": 9.475497961044312
    },
    {
      "epoch": 1.3158798217773438e-05,
      "model_forward_time": 1.7446324825286865,
      "step": 87798
    },
    {
      "epoch": 1.3158798217773438e-05,
      "step": 87798,
      "training_step_time": 7.886080980300903
    },
    {
      "epoch": 1.3159751892089843e-05,
      "model_forward_time": 1.7476494312286377,
      "step": 87799
    },
    {
      "epoch": 1.3159751892089843e-05,
      "step": 87799,
      "training_step_time": 7.801831483840942
    },
    {
      "epoch": 1.316070556640625e-05,
      "grad_norm": 0.15683802962303162,
      "learning_rate": 4.01498202186959e-07,
      "loss": 0.0538,
      "step": 87800
    },
    {
      "epoch": 1.316070556640625e-05,
      "model_forward_time": 1.3739662170410156,
      "step": 87800
    },
    {
      "dynamics_loss_avg": 0.0460063349455595,
      "epoch": 1.316070556640625e-05,
      "step": 87800
    },
    {
      "action_loss_avg": 0.0015371334942756222,
      "epoch": 1.316070556640625e-05,
      "step": 87800
    },
    {
      "epoch": 1.316070556640625e-05,
      "step": 87800,
      "training_step_time": 7.6424314975738525
    },
    {
      "epoch": 1.3161659240722656e-05,
      "model_forward_time": 1.7516450881958008,
      "step": 87801
    },
    {
      "epoch": 1.3161659240722656e-05,
      "step": 87801,
      "training_step_time": 8.786028623580933
    },
    {
      "epoch": 1.3162612915039062e-05,
      "model_forward_time": 1.7439627647399902,
      "step": 87802
    },
    {
      "epoch": 1.3162612915039062e-05,
      "step": 87802,
      "training_step_time": 8.646946907043457
    },
    {
      "epoch": 1.316356658935547e-05,
      "model_forward_time": 1.7575483322143555,
      "step": 87803
    },
    {
      "epoch": 1.316356658935547e-05,
      "step": 87803,
      "training_step_time": 9.409437894821167
    },
    {
      "epoch": 1.3164520263671875e-05,
      "model_forward_time": 1.747511863708496,
      "step": 87804
    },
    {
      "epoch": 1.3164520263671875e-05,
      "step": 87804,
      "training_step_time": 9.88255000114441
    },
    {
      "epoch": 1.3165473937988281e-05,
      "model_forward_time": 1.7460360527038574,
      "step": 87805
    },
    {
      "epoch": 1.3165473937988281e-05,
      "step": 87805,
      "training_step_time": 9.359517812728882
    },
    {
      "epoch": 1.3166427612304687e-05,
      "model_forward_time": 2.434253454208374,
      "step": 87806
    },
    {
      "epoch": 1.3166427612304687e-05,
      "step": 87806,
      "training_step_time": 8.850123882293701
    },
    {
      "epoch": 1.3167381286621094e-05,
      "model_forward_time": 1.7487945556640625,
      "step": 87807
    },
    {
      "epoch": 1.3167381286621094e-05,
      "step": 87807,
      "training_step_time": 7.750196695327759
    },
    {
      "epoch": 1.31683349609375e-05,
      "model_forward_time": 1.7453365325927734,
      "step": 87808
    },
    {
      "epoch": 1.31683349609375e-05,
      "step": 87808,
      "training_step_time": 9.671751022338867
    },
    {
      "epoch": 1.3169288635253906e-05,
      "model_forward_time": 1.7621879577636719,
      "step": 87809
    },
    {
      "epoch": 1.3169288635253906e-05,
      "step": 87809,
      "training_step_time": 7.636100769042969
    },
    {
      "epoch": 1.3170242309570313e-05,
      "grad_norm": 0.09216969460248947,
      "learning_rate": 4.0084926667027515e-07,
      "loss": 0.0528,
      "step": 87810
    },
    {
      "epoch": 1.3170242309570313e-05,
      "model_forward_time": 1.74778151512146,
      "step": 87810
    },
    {
      "dynamics_loss_avg": 0.04512356463819742,
      "epoch": 1.3170242309570313e-05,
      "step": 87810
    },
    {
      "action_loss_avg": 0.0010470554683706724,
      "epoch": 1.3170242309570313e-05,
      "step": 87810
    },
    {
      "epoch": 1.3170242309570313e-05,
      "step": 87810,
      "training_step_time": 7.752317667007446
    },
    {
      "epoch": 1.3171195983886719e-05,
      "model_forward_time": 1.7682368755340576,
      "step": 87811
    },
    {
      "epoch": 1.3171195983886719e-05,
      "step": 87811,
      "training_step_time": 9.457446098327637
    },
    {
      "epoch": 1.3172149658203124e-05,
      "model_forward_time": 1.765479326248169,
      "step": 87812
    },
    {
      "epoch": 1.3172149658203124e-05,
      "step": 87812,
      "training_step_time": 8.824046850204468
    },
    {
      "epoch": 1.3173103332519532e-05,
      "model_forward_time": 1.8892674446105957,
      "step": 87813
    },
    {
      "epoch": 1.3173103332519532e-05,
      "step": 87813,
      "training_step_time": 8.53690505027771
    },
    {
      "epoch": 1.3174057006835938e-05,
      "model_forward_time": 1.745431661605835,
      "step": 87814
    },
    {
      "epoch": 1.3174057006835938e-05,
      "step": 87814,
      "training_step_time": 7.910870790481567
    },
    {
      "epoch": 1.3175010681152343e-05,
      "model_forward_time": 2.4166154861450195,
      "step": 87815
    },
    {
      "epoch": 1.3175010681152343e-05,
      "step": 87815,
      "training_step_time": 8.723469972610474
    },
    {
      "epoch": 1.317596435546875e-05,
      "model_forward_time": 1.7712514400482178,
      "step": 87816
    },
    {
      "epoch": 1.317596435546875e-05,
      "step": 87816,
      "training_step_time": 7.952311992645264
    },
    {
      "epoch": 1.3176918029785156e-05,
      "model_forward_time": 1.760577917098999,
      "step": 87817
    },
    {
      "epoch": 1.3176918029785156e-05,
      "step": 87817,
      "training_step_time": 7.78413987159729
    },
    {
      "epoch": 1.3177871704101562e-05,
      "model_forward_time": 1.7643563747406006,
      "step": 87818
    },
    {
      "epoch": 1.3177871704101562e-05,
      "step": 87818,
      "training_step_time": 7.666771173477173
    },
    {
      "epoch": 1.317882537841797e-05,
      "model_forward_time": 1.7629246711730957,
      "step": 87819
    },
    {
      "epoch": 1.317882537841797e-05,
      "step": 87819,
      "training_step_time": 8.317801713943481
    },
    {
      "epoch": 1.3179779052734375e-05,
      "grad_norm": 0.07230570167303085,
      "learning_rate": 4.0020083410980014e-07,
      "loss": 0.0533,
      "step": 87820
    },
    {
      "epoch": 1.3179779052734375e-05,
      "model_forward_time": 1.7599232196807861,
      "step": 87820
    },
    {
      "dynamics_loss_avg": 0.040817652456462385,
      "epoch": 1.3179779052734375e-05,
      "step": 87820
    },
    {
      "action_loss_avg": 0.0014077873958740384,
      "epoch": 1.3179779052734375e-05,
      "step": 87820
    },
    {
      "epoch": 1.3179779052734375e-05,
      "step": 87820,
      "training_step_time": 7.872297525405884
    },
    {
      "epoch": 1.3180732727050781e-05,
      "model_forward_time": 1.7644121646881104,
      "step": 87821
    },
    {
      "epoch": 1.3180732727050781e-05,
      "step": 87821,
      "training_step_time": 7.718760967254639
    },
    {
      "epoch": 1.3181686401367187e-05,
      "model_forward_time": 1.8614835739135742,
      "step": 87822
    },
    {
      "epoch": 1.3181686401367187e-05,
      "step": 87822,
      "training_step_time": 7.813588380813599
    },
    {
      "epoch": 1.3182640075683594e-05,
      "model_forward_time": 1.7415649890899658,
      "step": 87823
    },
    {
      "epoch": 1.3182640075683594e-05,
      "step": 87823,
      "training_step_time": 8.989046812057495
    },
    {
      "epoch": 1.318359375e-05,
      "model_forward_time": 2.4020678997039795,
      "step": 87824
    },
    {
      "epoch": 1.318359375e-05,
      "step": 87824,
      "training_step_time": 9.787381410598755
    },
    {
      "epoch": 1.3184547424316406e-05,
      "model_forward_time": 1.3847301006317139,
      "step": 87825
    },
    {
      "epoch": 1.3184547424316406e-05,
      "step": 87825,
      "training_step_time": 9.4943528175354
    },
    {
      "epoch": 1.3185501098632813e-05,
      "model_forward_time": 1.769392967224121,
      "step": 87826
    },
    {
      "epoch": 1.3185501098632813e-05,
      "step": 87826,
      "training_step_time": 7.772437810897827
    },
    {
      "epoch": 1.3186454772949219e-05,
      "model_forward_time": 1.7771470546722412,
      "step": 87827
    },
    {
      "epoch": 1.3186454772949219e-05,
      "step": 87827,
      "training_step_time": 7.611368179321289
    },
    {
      "epoch": 1.3187408447265624e-05,
      "model_forward_time": 1.7574024200439453,
      "step": 87828
    },
    {
      "epoch": 1.3187408447265624e-05,
      "step": 87828,
      "training_step_time": 7.68796968460083
    },
    {
      "epoch": 1.3188362121582032e-05,
      "model_forward_time": 1.7673265933990479,
      "step": 87829
    },
    {
      "epoch": 1.3188362121582032e-05,
      "step": 87829,
      "training_step_time": 8.478378057479858
    },
    {
      "epoch": 1.3189315795898438e-05,
      "grad_norm": 0.08721339702606201,
      "learning_rate": 3.9955290457644626e-07,
      "loss": 0.0517,
      "step": 87830
    },
    {
      "epoch": 1.3189315795898438e-05,
      "model_forward_time": 1.7715673446655273,
      "step": 87830
    },
    {
      "dynamics_loss_avg": 0.04731124583631754,
      "epoch": 1.3189315795898438e-05,
      "step": 87830
    },
    {
      "action_loss_avg": 0.00097753896843642,
      "epoch": 1.3189315795898438e-05,
      "step": 87830
    },
    {
      "epoch": 1.3189315795898438e-05,
      "step": 87830,
      "training_step_time": 10.407877445220947
    },
    {
      "epoch": 1.3190269470214843e-05,
      "model_forward_time": 1.8310496807098389,
      "step": 87831
    },
    {
      "epoch": 1.3190269470214843e-05,
      "step": 87831,
      "training_step_time": 7.808751583099365
    },
    {
      "epoch": 1.319122314453125e-05,
      "model_forward_time": 1.7630157470703125,
      "step": 87832
    },
    {
      "epoch": 1.319122314453125e-05,
      "step": 87832,
      "training_step_time": 7.732100248336792
    },
    {
      "epoch": 1.3192176818847656e-05,
      "model_forward_time": 1.764662504196167,
      "step": 87833
    },
    {
      "epoch": 1.3192176818847656e-05,
      "step": 87833,
      "training_step_time": 7.838341474533081
    },
    {
      "epoch": 1.3193130493164062e-05,
      "model_forward_time": 2.3946149349212646,
      "step": 87834
    },
    {
      "epoch": 1.3193130493164062e-05,
      "step": 87834,
      "training_step_time": 9.757960796356201
    },
    {
      "epoch": 1.319408416748047e-05,
      "model_forward_time": 1.7637488842010498,
      "step": 87835
    },
    {
      "epoch": 1.319408416748047e-05,
      "step": 87835,
      "training_step_time": 7.764413118362427
    },
    {
      "epoch": 1.3195037841796875e-05,
      "model_forward_time": 1.7588133811950684,
      "step": 87836
    },
    {
      "epoch": 1.3195037841796875e-05,
      "step": 87836,
      "training_step_time": 8.023694515228271
    },
    {
      "epoch": 1.3195991516113281e-05,
      "model_forward_time": 1.778425693511963,
      "step": 87837
    },
    {
      "epoch": 1.3195991516113281e-05,
      "step": 87837,
      "training_step_time": 8.518445491790771
    },
    {
      "epoch": 1.3196945190429687e-05,
      "model_forward_time": 1.7658512592315674,
      "step": 87838
    },
    {
      "epoch": 1.3196945190429687e-05,
      "step": 87838,
      "training_step_time": 7.740386247634888
    },
    {
      "epoch": 1.3197898864746094e-05,
      "model_forward_time": 1.7855875492095947,
      "step": 87839
    },
    {
      "epoch": 1.3197898864746094e-05,
      "step": 87839,
      "training_step_time": 7.720029354095459
    },
    {
      "epoch": 1.31988525390625e-05,
      "grad_norm": 0.07959356904029846,
      "learning_rate": 3.989054781410712e-07,
      "loss": 0.0529,
      "step": 87840
    },
    {
      "epoch": 1.31988525390625e-05,
      "model_forward_time": 1.7655324935913086,
      "step": 87840
    },
    {
      "dynamics_loss_avg": 0.05419928263872862,
      "epoch": 1.31988525390625e-05,
      "step": 87840
    },
    {
      "action_loss_avg": 0.0007459577667759731,
      "epoch": 1.31988525390625e-05,
      "step": 87840
    },
    {
      "epoch": 1.31988525390625e-05,
      "step": 87840,
      "training_step_time": 7.794158697128296
    },
    {
      "epoch": 1.3199806213378906e-05,
      "model_forward_time": 1.7759482860565186,
      "step": 87841
    },
    {
      "epoch": 1.3199806213378906e-05,
      "step": 87841,
      "training_step_time": 7.7782886028289795
    },
    {
      "epoch": 1.3200759887695313e-05,
      "model_forward_time": 1.773650884628296,
      "step": 87842
    },
    {
      "epoch": 1.3200759887695313e-05,
      "step": 87842,
      "training_step_time": 9.399614810943604
    },
    {
      "epoch": 1.3201713562011719e-05,
      "model_forward_time": 2.4169914722442627,
      "step": 87843
    },
    {
      "epoch": 1.3201713562011719e-05,
      "step": 87843,
      "training_step_time": 8.688348054885864
    },
    {
      "epoch": 1.3202667236328124e-05,
      "model_forward_time": 1.7556726932525635,
      "step": 87844
    },
    {
      "epoch": 1.3202667236328124e-05,
      "step": 87844,
      "training_step_time": 7.6818015575408936
    },
    {
      "epoch": 1.3203620910644532e-05,
      "model_forward_time": 1.7633392810821533,
      "step": 87845
    },
    {
      "epoch": 1.3203620910644532e-05,
      "step": 87845,
      "training_step_time": 7.788499355316162
    },
    {
      "epoch": 1.3204574584960938e-05,
      "model_forward_time": 1.7636523246765137,
      "step": 87846
    },
    {
      "epoch": 1.3204574584960938e-05,
      "step": 87846,
      "training_step_time": 7.7106616497039795
    },
    {
      "epoch": 1.3205528259277343e-05,
      "model_forward_time": 1.754384994506836,
      "step": 87847
    },
    {
      "epoch": 1.3205528259277343e-05,
      "step": 87847,
      "training_step_time": 8.266679525375366
    },
    {
      "epoch": 1.320648193359375e-05,
      "model_forward_time": 1.7594413757324219,
      "step": 87848
    },
    {
      "epoch": 1.320648193359375e-05,
      "step": 87848,
      "training_step_time": 7.786885976791382
    },
    {
      "epoch": 1.3207435607910156e-05,
      "model_forward_time": 1.7524182796478271,
      "step": 87849
    },
    {
      "epoch": 1.3207435607910156e-05,
      "step": 87849,
      "training_step_time": 8.82091736793518
    },
    {
      "epoch": 1.3208389282226562e-05,
      "grad_norm": 0.17809537053108215,
      "learning_rate": 3.982585548744744e-07,
      "loss": 0.0535,
      "step": 87850
    },
    {
      "epoch": 1.3208389282226562e-05,
      "model_forward_time": 1.86625337600708,
      "step": 87850
    },
    {
      "dynamics_loss_avg": 0.054981141723692416,
      "epoch": 1.3208389282226562e-05,
      "step": 87850
    },
    {
      "action_loss_avg": 0.0006607682444155216,
      "epoch": 1.3208389282226562e-05,
      "step": 87850
    },
    {
      "epoch": 1.3208389282226562e-05,
      "step": 87850,
      "training_step_time": 7.813685417175293
    },
    {
      "epoch": 1.320934295654297e-05,
      "model_forward_time": 1.3818144798278809,
      "step": 87851
    },
    {
      "epoch": 1.320934295654297e-05,
      "step": 87851,
      "training_step_time": 8.643666744232178
    },
    {
      "epoch": 1.3210296630859375e-05,
      "model_forward_time": 2.404446601867676,
      "step": 87852
    },
    {
      "epoch": 1.3210296630859375e-05,
      "step": 87852,
      "training_step_time": 9.591854333877563
    },
    {
      "epoch": 1.3211250305175781e-05,
      "model_forward_time": 1.751023769378662,
      "step": 87853
    },
    {
      "epoch": 1.3211250305175781e-05,
      "step": 87853,
      "training_step_time": 8.586457967758179
    },
    {
      "epoch": 1.3212203979492187e-05,
      "model_forward_time": 1.7546916007995605,
      "step": 87854
    },
    {
      "epoch": 1.3212203979492187e-05,
      "step": 87854,
      "training_step_time": 7.786203384399414
    },
    {
      "epoch": 1.3213157653808594e-05,
      "model_forward_time": 1.7627012729644775,
      "step": 87855
    },
    {
      "epoch": 1.3213157653808594e-05,
      "step": 87855,
      "training_step_time": 7.799870014190674
    },
    {
      "epoch": 1.3214111328125e-05,
      "model_forward_time": 1.7581892013549805,
      "step": 87856
    },
    {
      "epoch": 1.3214111328125e-05,
      "step": 87856,
      "training_step_time": 7.786519289016724
    },
    {
      "epoch": 1.3215065002441406e-05,
      "model_forward_time": 1.753833293914795,
      "step": 87857
    },
    {
      "epoch": 1.3215065002441406e-05,
      "step": 87857,
      "training_step_time": 8.696166515350342
    },
    {
      "epoch": 1.3216018676757813e-05,
      "model_forward_time": 1.758434772491455,
      "step": 87858
    },
    {
      "epoch": 1.3216018676757813e-05,
      "step": 87858,
      "training_step_time": 9.210124015808105
    },
    {
      "epoch": 1.3216972351074219e-05,
      "model_forward_time": 1.8686161041259766,
      "step": 87859
    },
    {
      "epoch": 1.3216972351074219e-05,
      "step": 87859,
      "training_step_time": 8.567391395568848
    },
    {
      "epoch": 1.3217926025390624e-05,
      "grad_norm": 0.11188847571611404,
      "learning_rate": 3.9761213484740435e-07,
      "loss": 0.0525,
      "step": 87860
    },
    {
      "epoch": 1.3217926025390624e-05,
      "model_forward_time": 1.7446551322937012,
      "step": 87860
    },
    {
      "dynamics_loss_avg": 0.05436368603259325,
      "epoch": 1.3217926025390624e-05,
      "step": 87860
    },
    {
      "action_loss_avg": 0.000997851276770234,
      "epoch": 1.3217926025390624e-05,
      "step": 87860
    },
    {
      "epoch": 1.3217926025390624e-05,
      "step": 87860,
      "training_step_time": 9.42716360092163
    },
    {
      "epoch": 1.3218879699707032e-05,
      "model_forward_time": 2.4072928428649902,
      "step": 87861
    },
    {
      "epoch": 1.3218879699707032e-05,
      "step": 87861,
      "training_step_time": 8.701728343963623
    },
    {
      "epoch": 1.3219833374023438e-05,
      "model_forward_time": 1.7562878131866455,
      "step": 87862
    },
    {
      "epoch": 1.3219833374023438e-05,
      "step": 87862,
      "training_step_time": 7.86704421043396
    },
    {
      "epoch": 1.3220787048339843e-05,
      "model_forward_time": 1.7562768459320068,
      "step": 87863
    },
    {
      "epoch": 1.3220787048339843e-05,
      "step": 87863,
      "training_step_time": 8.562499523162842
    },
    {
      "epoch": 1.322174072265625e-05,
      "model_forward_time": 1.7609615325927734,
      "step": 87864
    },
    {
      "epoch": 1.322174072265625e-05,
      "step": 87864,
      "training_step_time": 7.781712055206299
    },
    {
      "epoch": 1.3222694396972656e-05,
      "model_forward_time": 1.3778488636016846,
      "step": 87865
    },
    {
      "epoch": 1.3222694396972656e-05,
      "step": 87865,
      "training_step_time": 7.820313453674316
    },
    {
      "epoch": 1.3223648071289062e-05,
      "model_forward_time": 1.7365655899047852,
      "step": 87866
    },
    {
      "epoch": 1.3223648071289062e-05,
      "step": 87866,
      "training_step_time": 8.380776166915894
    },
    {
      "epoch": 1.322460174560547e-05,
      "model_forward_time": 1.738168716430664,
      "step": 87867
    },
    {
      "epoch": 1.322460174560547e-05,
      "step": 87867,
      "training_step_time": 9.570374011993408
    },
    {
      "epoch": 1.3225555419921875e-05,
      "model_forward_time": 1.7524774074554443,
      "step": 87868
    },
    {
      "epoch": 1.3225555419921875e-05,
      "step": 87868,
      "training_step_time": 7.8149168491363525
    },
    {
      "epoch": 1.3226509094238281e-05,
      "model_forward_time": 1.741617202758789,
      "step": 87869
    },
    {
      "epoch": 1.3226509094238281e-05,
      "step": 87869,
      "training_step_time": 7.570264101028442
    },
    {
      "epoch": 1.3227462768554687e-05,
      "grad_norm": 0.11561154574155807,
      "learning_rate": 3.9696621813055057e-07,
      "loss": 0.0516,
      "step": 87870
    },
    {
      "epoch": 1.3227462768554687e-05,
      "model_forward_time": 1.7511093616485596,
      "step": 87870
    },
    {
      "dynamics_loss_avg": 0.045226181391626596,
      "epoch": 1.3227462768554687e-05,
      "step": 87870
    },
    {
      "action_loss_avg": 0.0013444648953736759,
      "epoch": 1.3227462768554687e-05,
      "step": 87870
    },
    {
      "epoch": 1.3227462768554687e-05,
      "step": 87870,
      "training_step_time": 8.6177237033844
    },
    {
      "epoch": 1.3228416442871094e-05,
      "model_forward_time": 2.3939409255981445,
      "step": 87871
    },
    {
      "epoch": 1.3228416442871094e-05,
      "step": 87871,
      "training_step_time": 7.817445993423462
    },
    {
      "epoch": 1.32293701171875e-05,
      "model_forward_time": 1.735917568206787,
      "step": 87872
    },
    {
      "epoch": 1.32293701171875e-05,
      "step": 87872,
      "training_step_time": 8.906108856201172
    },
    {
      "epoch": 1.3230323791503906e-05,
      "model_forward_time": 1.7439661026000977,
      "step": 87873
    },
    {
      "epoch": 1.3230323791503906e-05,
      "step": 87873,
      "training_step_time": 8.664759635925293
    },
    {
      "epoch": 1.3231277465820313e-05,
      "model_forward_time": 1.7431650161743164,
      "step": 87874
    },
    {
      "epoch": 1.3231277465820313e-05,
      "step": 87874,
      "training_step_time": 7.718470335006714
    },
    {
      "epoch": 1.3232231140136719e-05,
      "model_forward_time": 1.7479276657104492,
      "step": 87875
    },
    {
      "epoch": 1.3232231140136719e-05,
      "step": 87875,
      "training_step_time": 8.76163649559021
    },
    {
      "epoch": 1.3233184814453124e-05,
      "model_forward_time": 1.7444794178009033,
      "step": 87876
    },
    {
      "epoch": 1.3233184814453124e-05,
      "step": 87876,
      "training_step_time": 8.81034541130066
    },
    {
      "epoch": 1.3234138488769532e-05,
      "model_forward_time": 1.7450072765350342,
      "step": 87877
    },
    {
      "epoch": 1.3234138488769532e-05,
      "step": 87877,
      "training_step_time": 7.788561105728149
    },
    {
      "epoch": 1.3235092163085938e-05,
      "model_forward_time": 1.734766960144043,
      "step": 87878
    },
    {
      "epoch": 1.3235092163085938e-05,
      "step": 87878,
      "training_step_time": 8.597033739089966
    },
    {
      "epoch": 1.3236045837402343e-05,
      "model_forward_time": 1.750600814819336,
      "step": 87879
    },
    {
      "epoch": 1.3236045837402343e-05,
      "step": 87879,
      "training_step_time": 9.353132247924805
    },
    {
      "epoch": 1.323699951171875e-05,
      "grad_norm": 0.06816329807043076,
      "learning_rate": 3.9632080479455046e-07,
      "loss": 0.0523,
      "step": 87880
    },
    {
      "epoch": 1.323699951171875e-05,
      "model_forward_time": 2.3732657432556152,
      "step": 87880
    },
    {
      "dynamics_loss_avg": 0.05743486881256103,
      "epoch": 1.323699951171875e-05,
      "step": 87880
    },
    {
      "action_loss_avg": 0.0011626813327893614,
      "epoch": 1.323699951171875e-05,
      "step": 87880
    },
    {
      "epoch": 1.323699951171875e-05,
      "step": 87880,
      "training_step_time": 7.789433002471924
    },
    {
      "epoch": 1.3237953186035156e-05,
      "model_forward_time": 1.7421305179595947,
      "step": 87881
    },
    {
      "epoch": 1.3237953186035156e-05,
      "step": 87881,
      "training_step_time": 7.8656487464904785
    },
    {
      "epoch": 1.3238906860351562e-05,
      "model_forward_time": 1.72536039352417,
      "step": 87882
    },
    {
      "epoch": 1.3238906860351562e-05,
      "step": 87882,
      "training_step_time": 7.695967435836792
    },
    {
      "epoch": 1.3239860534667968e-05,
      "model_forward_time": 1.7338593006134033,
      "step": 87883
    },
    {
      "epoch": 1.3239860534667968e-05,
      "step": 87883,
      "training_step_time": 7.7335898876190186
    },
    {
      "epoch": 1.3240814208984375e-05,
      "model_forward_time": 1.3613674640655518,
      "step": 87884
    },
    {
      "epoch": 1.3240814208984375e-05,
      "step": 87884,
      "training_step_time": 8.259569883346558
    },
    {
      "epoch": 1.3241767883300781e-05,
      "model_forward_time": 1.7507517337799072,
      "step": 87885
    },
    {
      "epoch": 1.3241767883300781e-05,
      "step": 87885,
      "training_step_time": 8.49320387840271
    },
    {
      "epoch": 1.3242721557617187e-05,
      "model_forward_time": 1.7378592491149902,
      "step": 87886
    },
    {
      "epoch": 1.3242721557617187e-05,
      "step": 87886,
      "training_step_time": 7.838291168212891
    },
    {
      "epoch": 1.3243675231933594e-05,
      "model_forward_time": 1.745877742767334,
      "step": 87887
    },
    {
      "epoch": 1.3243675231933594e-05,
      "step": 87887,
      "training_step_time": 7.665121793746948
    },
    {
      "epoch": 1.324462890625e-05,
      "model_forward_time": 1.7529802322387695,
      "step": 87888
    },
    {
      "epoch": 1.324462890625e-05,
      "step": 87888,
      "training_step_time": 7.8802995681762695
    },
    {
      "epoch": 1.3245582580566406e-05,
      "model_forward_time": 2.3814680576324463,
      "step": 87889
    },
    {
      "epoch": 1.3245582580566406e-05,
      "step": 87889,
      "training_step_time": 11.59920859336853
    },
    {
      "epoch": 1.3246536254882813e-05,
      "grad_norm": 0.07909691333770752,
      "learning_rate": 3.956758949099859e-07,
      "loss": 0.0536,
      "step": 87890
    },
    {
      "epoch": 1.3246536254882813e-05,
      "model_forward_time": 1.739257574081421,
      "step": 87890
    },
    {
      "dynamics_loss_avg": 0.03888375787064433,
      "epoch": 1.3246536254882813e-05,
      "step": 87890
    },
    {
      "action_loss_avg": 0.0011587620101636276,
      "epoch": 1.3246536254882813e-05,
      "step": 87890
    },
    {
      "epoch": 1.3246536254882813e-05,
      "step": 87890,
      "training_step_time": 8.688562393188477
    },
    {
      "epoch": 1.3247489929199219e-05,
      "model_forward_time": 1.7448220252990723,
      "step": 87891
    },
    {
      "epoch": 1.3247489929199219e-05,
      "step": 87891,
      "training_step_time": 7.475688934326172
    },
    {
      "epoch": 1.3248443603515624e-05,
      "model_forward_time": 1.7626829147338867,
      "step": 87892
    },
    {
      "epoch": 1.3248443603515624e-05,
      "step": 87892,
      "training_step_time": 8.598926305770874
    },
    {
      "epoch": 1.3249397277832032e-05,
      "model_forward_time": 1.7437312602996826,
      "step": 87893
    },
    {
      "epoch": 1.3249397277832032e-05,
      "step": 87893,
      "training_step_time": 6.868958950042725
    },
    {
      "epoch": 1.3250350952148438e-05,
      "model_forward_time": 1.751943588256836,
      "step": 87894
    },
    {
      "epoch": 1.3250350952148438e-05,
      "step": 87894,
      "training_step_time": 6.8211283683776855
    },
    {
      "epoch": 1.3251304626464843e-05,
      "model_forward_time": 1.7523994445800781,
      "step": 87895
    },
    {
      "epoch": 1.3251304626464843e-05,
      "step": 87895,
      "training_step_time": 6.8208723068237305
    },
    {
      "epoch": 1.325225830078125e-05,
      "model_forward_time": 1.3623855113983154,
      "step": 87896
    },
    {
      "epoch": 1.325225830078125e-05,
      "step": 87896,
      "training_step_time": 6.827252388000488
    },
    {
      "epoch": 1.3253211975097656e-05,
      "model_forward_time": 1.752676248550415,
      "step": 87897
    },
    {
      "epoch": 1.3253211975097656e-05,
      "step": 87897,
      "training_step_time": 6.836045503616333
    },
    {
      "epoch": 1.3254165649414062e-05,
      "model_forward_time": 1.7490828037261963,
      "step": 87898
    },
    {
      "epoch": 1.3254165649414062e-05,
      "step": 87898,
      "training_step_time": 6.873028039932251
    },
    {
      "epoch": 1.3255119323730468e-05,
      "model_forward_time": 2.4556496143341064,
      "step": 87899
    },
    {
      "epoch": 1.3255119323730468e-05,
      "step": 87899,
      "training_step_time": 8.72626519203186
    },
    {
      "epoch": 1.3256072998046875e-05,
      "grad_norm": 0.08781913667917252,
      "learning_rate": 3.9503148854738383e-07,
      "loss": 0.0533,
      "step": 87900
    },
    {
      "epoch": 1.3256072998046875e-05,
      "model_forward_time": 1.7580063343048096,
      "step": 87900
    },
    {
      "dynamics_loss_avg": 0.05670438352972269,
      "epoch": 1.3256072998046875e-05,
      "step": 87900
    },
    {
      "action_loss_avg": 0.0013125212368322535,
      "epoch": 1.3256072998046875e-05,
      "step": 87900
    },
    {
      "epoch": 1.3256072998046875e-05,
      "step": 87900,
      "training_step_time": 8.661490440368652
    },
    {
      "epoch": 1.3257026672363281e-05,
      "model_forward_time": 1.7578258514404297,
      "step": 87901
    },
    {
      "epoch": 1.3257026672363281e-05,
      "step": 87901,
      "training_step_time": 8.578973770141602
    },
    {
      "epoch": 1.3257980346679687e-05,
      "model_forward_time": 1.7645158767700195,
      "step": 87902
    },
    {
      "epoch": 1.3257980346679687e-05,
      "step": 87902,
      "training_step_time": 6.823246479034424
    },
    {
      "epoch": 1.3258934020996094e-05,
      "model_forward_time": 1.762087106704712,
      "step": 87903
    },
    {
      "epoch": 1.3258934020996094e-05,
      "step": 87903,
      "training_step_time": 6.82342529296875
    },
    {
      "epoch": 1.32598876953125e-05,
      "model_forward_time": 1.7637546062469482,
      "step": 87904
    },
    {
      "epoch": 1.32598876953125e-05,
      "step": 87904,
      "training_step_time": 6.844176530838013
    },
    {
      "epoch": 1.3260841369628906e-05,
      "model_forward_time": 1.7652277946472168,
      "step": 87905
    },
    {
      "epoch": 1.3260841369628906e-05,
      "step": 87905,
      "training_step_time": 6.8661463260650635
    },
    {
      "epoch": 1.3261795043945313e-05,
      "model_forward_time": 1.763204574584961,
      "step": 87906
    },
    {
      "epoch": 1.3261795043945313e-05,
      "step": 87906,
      "training_step_time": 6.824401140213013
    },
    {
      "epoch": 1.3262748718261719e-05,
      "model_forward_time": 1.7551801204681396,
      "step": 87907
    },
    {
      "epoch": 1.3262748718261719e-05,
      "step": 87907,
      "training_step_time": 6.823206663131714
    },
    {
      "epoch": 1.3263702392578124e-05,
      "model_forward_time": 2.4604809284210205,
      "step": 87908
    },
    {
      "epoch": 1.3263702392578124e-05,
      "step": 87908,
      "training_step_time": 7.914279460906982
    },
    {
      "epoch": 1.3264656066894532e-05,
      "model_forward_time": 1.7505698204040527,
      "step": 87909
    },
    {
      "epoch": 1.3264656066894532e-05,
      "step": 87909,
      "training_step_time": 7.755814790725708
    },
    {
      "epoch": 1.3265609741210938e-05,
      "grad_norm": 0.11531244218349457,
      "learning_rate": 3.9438758577721393e-07,
      "loss": 0.054,
      "step": 87910
    },
    {
      "epoch": 1.3265609741210938e-05,
      "model_forward_time": 1.7532355785369873,
      "step": 87910
    },
    {
      "dynamics_loss_avg": 0.04725222741253674,
      "epoch": 1.3265609741210938e-05,
      "step": 87910
    },
    {
      "action_loss_avg": 0.0020593773806467652,
      "epoch": 1.3265609741210938e-05,
      "step": 87910
    },
    {
      "epoch": 1.3265609741210938e-05,
      "step": 87910,
      "training_step_time": 7.4377830028533936
    },
    {
      "epoch": 1.3266563415527343e-05,
      "model_forward_time": 1.7683956623077393,
      "step": 87911
    },
    {
      "epoch": 1.3266563415527343e-05,
      "step": 87911,
      "training_step_time": 7.651827812194824
    },
    {
      "epoch": 1.326751708984375e-05,
      "model_forward_time": 1.7704956531524658,
      "step": 87912
    },
    {
      "epoch": 1.326751708984375e-05,
      "step": 87912,
      "training_step_time": 6.8934714794158936
    },
    {
      "epoch": 1.3268470764160156e-05,
      "model_forward_time": 1.763381004333496,
      "step": 87913
    },
    {
      "epoch": 1.3268470764160156e-05,
      "step": 87913,
      "training_step_time": 6.817209959030151
    },
    {
      "epoch": 1.3269424438476562e-05,
      "model_forward_time": 1.7705764770507812,
      "step": 87914
    },
    {
      "epoch": 1.3269424438476562e-05,
      "step": 87914,
      "training_step_time": 6.82880163192749
    },
    {
      "epoch": 1.3270378112792968e-05,
      "model_forward_time": 1.8163819313049316,
      "step": 87915
    },
    {
      "epoch": 1.3270378112792968e-05,
      "step": 87915,
      "training_step_time": 6.887531518936157
    },
    {
      "epoch": 1.3271331787109375e-05,
      "model_forward_time": 1.771207332611084,
      "step": 87916
    },
    {
      "epoch": 1.3271331787109375e-05,
      "step": 87916,
      "training_step_time": 6.874496698379517
    },
    {
      "epoch": 1.3272285461425781e-05,
      "model_forward_time": 2.475473165512085,
      "step": 87917
    },
    {
      "epoch": 1.3272285461425781e-05,
      "step": 87917,
      "training_step_time": 8.381524801254272
    },
    {
      "epoch": 1.3273239135742187e-05,
      "model_forward_time": 1.7689392566680908,
      "step": 87918
    },
    {
      "epoch": 1.3273239135742187e-05,
      "step": 87918,
      "training_step_time": 7.661046981811523
    },
    {
      "epoch": 1.3274192810058594e-05,
      "model_forward_time": 1.773939847946167,
      "step": 87919
    },
    {
      "epoch": 1.3274192810058594e-05,
      "step": 87919,
      "training_step_time": 7.43552827835083
    },
    {
      "epoch": 1.3275146484375e-05,
      "grad_norm": 0.101329006254673,
      "learning_rate": 3.9374418666989376e-07,
      "loss": 0.0542,
      "step": 87920
    },
    {
      "epoch": 1.3275146484375e-05,
      "model_forward_time": 1.0731117725372314,
      "step": 87920
    },
    {
      "dynamics_loss_avg": 0.03665674524381757,
      "epoch": 1.3275146484375e-05,
      "step": 87920
    },
    {
      "action_loss_avg": 0.0012649892538320272,
      "epoch": 1.3275146484375e-05,
      "step": 87920
    },
    {
      "epoch": 1.3275146484375e-05,
      "step": 87920,
      "training_step_time": 7.65315580368042
    },
    {
      "epoch": 1.3276100158691406e-05,
      "model_forward_time": 1.7693727016448975,
      "step": 87921
    },
    {
      "epoch": 1.3276100158691406e-05,
      "step": 87921,
      "training_step_time": 6.811147212982178
    },
    {
      "epoch": 1.3277053833007813e-05,
      "model_forward_time": 1.7698543071746826,
      "step": 87922
    },
    {
      "epoch": 1.3277053833007813e-05,
      "step": 87922,
      "training_step_time": 6.813176870346069
    },
    {
      "epoch": 1.3278007507324219e-05,
      "model_forward_time": 1.806333065032959,
      "step": 87923
    },
    {
      "epoch": 1.3278007507324219e-05,
      "step": 87923,
      "training_step_time": 6.886368989944458
    },
    {
      "epoch": 1.3278961181640625e-05,
      "model_forward_time": 1.7687873840332031,
      "step": 87924
    },
    {
      "epoch": 1.3278961181640625e-05,
      "step": 87924,
      "training_step_time": 6.7891294956207275
    },
    {
      "epoch": 1.3279914855957032e-05,
      "model_forward_time": 1.7738182544708252,
      "step": 87925
    },
    {
      "epoch": 1.3279914855957032e-05,
      "step": 87925,
      "training_step_time": 6.831863164901733
    },
    {
      "epoch": 1.3280868530273438e-05,
      "model_forward_time": 1.7731208801269531,
      "step": 87926
    },
    {
      "epoch": 1.3280868530273438e-05,
      "step": 87926,
      "training_step_time": 9.987565755844116
    },
    {
      "epoch": 1.3281822204589843e-05,
      "model_forward_time": 2.430954933166504,
      "step": 87927
    },
    {
      "epoch": 1.3281822204589843e-05,
      "step": 87927,
      "training_step_time": 7.752803325653076
    },
    {
      "epoch": 1.328277587890625e-05,
      "model_forward_time": 1.4169800281524658,
      "step": 87928
    },
    {
      "epoch": 1.328277587890625e-05,
      "step": 87928,
      "training_step_time": 7.521724700927734
    },
    {
      "epoch": 1.3283729553222657e-05,
      "model_forward_time": 1.7923450469970703,
      "step": 87929
    },
    {
      "epoch": 1.3283729553222657e-05,
      "step": 87929,
      "training_step_time": 7.67028021812439
    },
    {
      "epoch": 1.3284683227539062e-05,
      "grad_norm": 0.19252599775791168,
      "learning_rate": 3.9310129129578376e-07,
      "loss": 0.0547,
      "step": 87930
    },
    {
      "epoch": 1.3284683227539062e-05,
      "model_forward_time": 1.7625036239624023,
      "step": 87930
    },
    {
      "dynamics_loss_avg": 0.06448982451111078,
      "epoch": 1.3284683227539062e-05,
      "step": 87930
    },
    {
      "action_loss_avg": 0.0013962415672722273,
      "epoch": 1.3284683227539062e-05,
      "step": 87930
    },
    {
      "epoch": 1.3284683227539062e-05,
      "step": 87930,
      "training_step_time": 6.818303823471069
    },
    {
      "epoch": 1.3285636901855468e-05,
      "model_forward_time": 1.7611713409423828,
      "step": 87931
    },
    {
      "epoch": 1.3285636901855468e-05,
      "step": 87931,
      "training_step_time": 6.873066425323486
    },
    {
      "epoch": 1.3286590576171875e-05,
      "model_forward_time": 1.7734720706939697,
      "step": 87932
    },
    {
      "epoch": 1.3286590576171875e-05,
      "step": 87932,
      "training_step_time": 6.746778249740601
    },
    {
      "epoch": 1.3287544250488281e-05,
      "model_forward_time": 1.7837779521942139,
      "step": 87933
    },
    {
      "epoch": 1.3287544250488281e-05,
      "step": 87933,
      "training_step_time": 6.708439111709595
    },
    {
      "epoch": 1.3288497924804687e-05,
      "model_forward_time": 1.4096169471740723,
      "step": 87934
    },
    {
      "epoch": 1.3288497924804687e-05,
      "step": 87934,
      "training_step_time": 6.764012813568115
    },
    {
      "epoch": 1.3289451599121094e-05,
      "model_forward_time": 1.7964248657226562,
      "step": 87935
    },
    {
      "epoch": 1.3289451599121094e-05,
      "step": 87935,
      "training_step_time": 8.80588412284851
    },
    {
      "epoch": 1.32904052734375e-05,
      "model_forward_time": 2.4039480686187744,
      "step": 87936
    },
    {
      "epoch": 1.32904052734375e-05,
      "step": 87936,
      "training_step_time": 7.894818544387817
    },
    {
      "epoch": 1.3291358947753906e-05,
      "model_forward_time": 1.7796177864074707,
      "step": 87937
    },
    {
      "epoch": 1.3291358947753906e-05,
      "step": 87937,
      "training_step_time": 7.743449687957764
    },
    {
      "epoch": 1.3292312622070313e-05,
      "model_forward_time": 1.7715308666229248,
      "step": 87938
    },
    {
      "epoch": 1.3292312622070313e-05,
      "step": 87938,
      "training_step_time": 7.7133026123046875
    },
    {
      "epoch": 1.3293266296386719e-05,
      "model_forward_time": 1.774796485900879,
      "step": 87939
    },
    {
      "epoch": 1.3293266296386719e-05,
      "step": 87939,
      "training_step_time": 7.222416400909424
    },
    {
      "epoch": 1.3294219970703125e-05,
      "grad_norm": 0.07570034265518188,
      "learning_rate": 3.9245889972519047e-07,
      "loss": 0.0553,
      "step": 87940
    },
    {
      "epoch": 1.3294219970703125e-05,
      "model_forward_time": 1.7628817558288574,
      "step": 87940
    },
    {
      "dynamics_loss_avg": 0.06283188797533512,
      "epoch": 1.3294219970703125e-05,
      "step": 87940
    },
    {
      "action_loss_avg": 0.0013758023254922592,
      "epoch": 1.3294219970703125e-05,
      "step": 87940
    },
    {
      "epoch": 1.3294219970703125e-05,
      "step": 87940,
      "training_step_time": 6.8105950355529785
    },
    {
      "epoch": 1.3295173645019532e-05,
      "model_forward_time": 1.7761242389678955,
      "step": 87941
    },
    {
      "epoch": 1.3295173645019532e-05,
      "step": 87941,
      "training_step_time": 6.793553352355957
    },
    {
      "epoch": 1.3296127319335938e-05,
      "model_forward_time": 1.7781040668487549,
      "step": 87942
    },
    {
      "epoch": 1.3296127319335938e-05,
      "step": 87942,
      "training_step_time": 6.838738679885864
    },
    {
      "epoch": 1.3297080993652343e-05,
      "model_forward_time": 1.7561585903167725,
      "step": 87943
    },
    {
      "epoch": 1.3297080993652343e-05,
      "step": 87943,
      "training_step_time": 6.805439710617065
    },
    {
      "epoch": 1.329803466796875e-05,
      "model_forward_time": 1.769331455230713,
      "step": 87944
    },
    {
      "epoch": 1.329803466796875e-05,
      "step": 87944,
      "training_step_time": 9.51683783531189
    },
    {
      "epoch": 1.3298988342285157e-05,
      "model_forward_time": 2.4273698329925537,
      "step": 87945
    },
    {
      "epoch": 1.3298988342285157e-05,
      "step": 87945,
      "training_step_time": 7.95250678062439
    },
    {
      "epoch": 1.3299942016601562e-05,
      "model_forward_time": 1.769287109375,
      "step": 87946
    },
    {
      "epoch": 1.3299942016601562e-05,
      "step": 87946,
      "training_step_time": 7.757659912109375
    },
    {
      "epoch": 1.3300895690917968e-05,
      "model_forward_time": 1.7720351219177246,
      "step": 87947
    },
    {
      "epoch": 1.3300895690917968e-05,
      "step": 87947,
      "training_step_time": 7.698463678359985
    },
    {
      "epoch": 1.3301849365234375e-05,
      "model_forward_time": 1.772935152053833,
      "step": 87948
    },
    {
      "epoch": 1.3301849365234375e-05,
      "step": 87948,
      "training_step_time": 7.586520671844482
    },
    {
      "epoch": 1.3302803039550781e-05,
      "model_forward_time": 1.7745561599731445,
      "step": 87949
    },
    {
      "epoch": 1.3302803039550781e-05,
      "step": 87949,
      "training_step_time": 6.864612102508545
    },
    {
      "epoch": 1.3303756713867187e-05,
      "grad_norm": 0.08234116435050964,
      "learning_rate": 3.918170120283626e-07,
      "loss": 0.0552,
      "step": 87950
    },
    {
      "epoch": 1.3303756713867187e-05,
      "model_forward_time": 1.7715873718261719,
      "step": 87950
    },
    {
      "dynamics_loss_avg": 0.041552369482815266,
      "epoch": 1.3303756713867187e-05,
      "step": 87950
    },
    {
      "action_loss_avg": 0.001177209391607903,
      "epoch": 1.3303756713867187e-05,
      "step": 87950
    },
    {
      "epoch": 1.3303756713867187e-05,
      "step": 87950,
      "training_step_time": 6.8008973598480225
    },
    {
      "epoch": 1.3304710388183594e-05,
      "model_forward_time": 1.7746446132659912,
      "step": 87951
    },
    {
      "epoch": 1.3304710388183594e-05,
      "step": 87951,
      "training_step_time": 6.845823049545288
    },
    {
      "epoch": 1.33056640625e-05,
      "model_forward_time": 1.767019510269165,
      "step": 87952
    },
    {
      "epoch": 1.33056640625e-05,
      "step": 87952,
      "training_step_time": 6.843486070632935
    },
    {
      "epoch": 1.3306617736816406e-05,
      "model_forward_time": 1.7760839462280273,
      "step": 87953
    },
    {
      "epoch": 1.3306617736816406e-05,
      "step": 87953,
      "training_step_time": 8.693005800247192
    },
    {
      "epoch": 1.3307571411132813e-05,
      "model_forward_time": 2.0648372173309326,
      "step": 87954
    },
    {
      "epoch": 1.3307571411132813e-05,
      "step": 87954,
      "training_step_time": 8.76688265800476
    },
    {
      "epoch": 1.3308525085449219e-05,
      "model_forward_time": 1.7724628448486328,
      "step": 87955
    },
    {
      "epoch": 1.3308525085449219e-05,
      "step": 87955,
      "training_step_time": 8.859830617904663
    },
    {
      "epoch": 1.3309478759765625e-05,
      "model_forward_time": 1.766897439956665,
      "step": 87956
    },
    {
      "epoch": 1.3309478759765625e-05,
      "step": 87956,
      "training_step_time": 7.6018290519714355
    },
    {
      "epoch": 1.3310432434082032e-05,
      "model_forward_time": 1.8010454177856445,
      "step": 87957
    },
    {
      "epoch": 1.3310432434082032e-05,
      "step": 87957,
      "training_step_time": 7.717860221862793
    },
    {
      "epoch": 1.3311386108398438e-05,
      "model_forward_time": 1.7638399600982666,
      "step": 87958
    },
    {
      "epoch": 1.3311386108398438e-05,
      "step": 87958,
      "training_step_time": 6.813900709152222
    },
    {
      "epoch": 1.3312339782714843e-05,
      "model_forward_time": 1.7655720710754395,
      "step": 87959
    },
    {
      "epoch": 1.3312339782714843e-05,
      "step": 87959,
      "training_step_time": 6.8347344398498535
    },
    {
      "epoch": 1.331329345703125e-05,
      "grad_norm": 0.11327899247407913,
      "learning_rate": 3.911756282755003e-07,
      "loss": 0.0564,
      "step": 87960
    },
    {
      "epoch": 1.331329345703125e-05,
      "model_forward_time": 1.7603123188018799,
      "step": 87960
    },
    {
      "dynamics_loss_avg": 0.04540935549885035,
      "epoch": 1.331329345703125e-05,
      "step": 87960
    },
    {
      "action_loss_avg": 0.0009735102168633603,
      "epoch": 1.331329345703125e-05,
      "step": 87960
    },
    {
      "epoch": 1.331329345703125e-05,
      "step": 87960,
      "training_step_time": 6.833637475967407
    },
    {
      "epoch": 1.3314247131347657e-05,
      "model_forward_time": 1.7665596008300781,
      "step": 87961
    },
    {
      "epoch": 1.3314247131347657e-05,
      "step": 87961,
      "training_step_time": 6.853790283203125
    },
    {
      "epoch": 1.3315200805664062e-05,
      "model_forward_time": 1.7644405364990234,
      "step": 87962
    },
    {
      "epoch": 1.3315200805664062e-05,
      "step": 87962,
      "training_step_time": 8.301050662994385
    },
    {
      "epoch": 1.3316154479980468e-05,
      "model_forward_time": 1.7663171291351318,
      "step": 87963
    },
    {
      "epoch": 1.3316154479980468e-05,
      "step": 87963,
      "training_step_time": 10.48883581161499
    },
    {
      "epoch": 1.3317108154296875e-05,
      "model_forward_time": 2.395552158355713,
      "step": 87964
    },
    {
      "epoch": 1.3317108154296875e-05,
      "step": 87964,
      "training_step_time": 7.842033624649048
    },
    {
      "epoch": 1.3318061828613281e-05,
      "model_forward_time": 1.7674014568328857,
      "step": 87965
    },
    {
      "epoch": 1.3318061828613281e-05,
      "step": 87965,
      "training_step_time": 7.651595830917358
    },
    {
      "epoch": 1.3319015502929687e-05,
      "model_forward_time": 1.7675111293792725,
      "step": 87966
    },
    {
      "epoch": 1.3319015502929687e-05,
      "step": 87966,
      "training_step_time": 7.764773845672607
    },
    {
      "epoch": 1.3319969177246094e-05,
      "model_forward_time": 1.7713277339935303,
      "step": 87967
    },
    {
      "epoch": 1.3319969177246094e-05,
      "step": 87967,
      "training_step_time": 6.853550434112549
    },
    {
      "epoch": 1.33209228515625e-05,
      "model_forward_time": 1.7712490558624268,
      "step": 87968
    },
    {
      "epoch": 1.33209228515625e-05,
      "step": 87968,
      "training_step_time": 6.827371835708618
    },
    {
      "epoch": 1.3321876525878906e-05,
      "model_forward_time": 1.8031353950500488,
      "step": 87969
    },
    {
      "epoch": 1.3321876525878906e-05,
      "step": 87969,
      "training_step_time": 6.842658281326294
    },
    {
      "epoch": 1.3322830200195313e-05,
      "grad_norm": 0.08317159861326218,
      "learning_rate": 3.905347485367411e-07,
      "loss": 0.0573,
      "step": 87970
    },
    {
      "epoch": 1.3322830200195313e-05,
      "model_forward_time": 1.75374436378479,
      "step": 87970
    },
    {
      "dynamics_loss_avg": 0.046994384936988355,
      "epoch": 1.3322830200195313e-05,
      "step": 87970
    },
    {
      "action_loss_avg": 0.0011118965936475433,
      "epoch": 1.3322830200195313e-05,
      "step": 87970
    },
    {
      "epoch": 1.3322830200195313e-05,
      "step": 87970,
      "training_step_time": 6.794232130050659
    },
    {
      "epoch": 1.3323783874511719e-05,
      "model_forward_time": 1.7648417949676514,
      "step": 87971
    },
    {
      "epoch": 1.3323783874511719e-05,
      "step": 87971,
      "training_step_time": 8.698815107345581
    },
    {
      "epoch": 1.3324737548828125e-05,
      "model_forward_time": 1.4071619510650635,
      "step": 87972
    },
    {
      "epoch": 1.3324737548828125e-05,
      "step": 87972,
      "training_step_time": 9.865082263946533
    },
    {
      "epoch": 1.3325691223144532e-05,
      "model_forward_time": 2.4118504524230957,
      "step": 87973
    },
    {
      "epoch": 1.3325691223144532e-05,
      "step": 87973,
      "training_step_time": 7.847260236740112
    },
    {
      "epoch": 1.3326644897460938e-05,
      "model_forward_time": 1.7813496589660645,
      "step": 87974
    },
    {
      "epoch": 1.3326644897460938e-05,
      "step": 87974,
      "training_step_time": 7.663980960845947
    },
    {
      "epoch": 1.3327598571777343e-05,
      "model_forward_time": 1.7649846076965332,
      "step": 87975
    },
    {
      "epoch": 1.3327598571777343e-05,
      "step": 87975,
      "training_step_time": 7.703819751739502
    },
    {
      "epoch": 1.332855224609375e-05,
      "model_forward_time": 1.7652230262756348,
      "step": 87976
    },
    {
      "epoch": 1.332855224609375e-05,
      "step": 87976,
      "training_step_time": 6.840730667114258
    },
    {
      "epoch": 1.3329505920410157e-05,
      "model_forward_time": 1.7583935260772705,
      "step": 87977
    },
    {
      "epoch": 1.3329505920410157e-05,
      "step": 87977,
      "training_step_time": 6.895727157592773
    },
    {
      "epoch": 1.3330459594726562e-05,
      "model_forward_time": 1.7536842823028564,
      "step": 87978
    },
    {
      "epoch": 1.3330459594726562e-05,
      "step": 87978,
      "training_step_time": 6.810265064239502
    },
    {
      "epoch": 1.3331413269042968e-05,
      "model_forward_time": 1.777453899383545,
      "step": 87979
    },
    {
      "epoch": 1.3331413269042968e-05,
      "step": 87979,
      "training_step_time": 6.798168659210205
    },
    {
      "epoch": 1.3332366943359375e-05,
      "grad_norm": 0.20546464622020721,
      "learning_rate": 3.8989437288217246e-07,
      "loss": 0.0564,
      "step": 87980
    },
    {
      "epoch": 1.3332366943359375e-05,
      "model_forward_time": 1.7600929737091064,
      "step": 87980
    },
    {
      "dynamics_loss_avg": 0.04517735214903951,
      "epoch": 1.3332366943359375e-05,
      "step": 87980
    },
    {
      "action_loss_avg": 0.0016508774715475738,
      "epoch": 1.3332366943359375e-05,
      "step": 87980
    },
    {
      "epoch": 1.3332366943359375e-05,
      "step": 87980,
      "training_step_time": 8.548250436782837
    },
    {
      "epoch": 1.3333320617675781e-05,
      "model_forward_time": 1.7675137519836426,
      "step": 87981
    },
    {
      "epoch": 1.3333320617675781e-05,
      "step": 87981,
      "training_step_time": 11.497942447662354
    },
    {
      "epoch": 1.3334274291992187e-05,
      "model_forward_time": 2.4219627380371094,
      "step": 87982
    },
    {
      "epoch": 1.3334274291992187e-05,
      "step": 87982,
      "training_step_time": 10.073742866516113
    },
    {
      "epoch": 1.3335227966308594e-05,
      "model_forward_time": 1.775125503540039,
      "step": 87983
    },
    {
      "epoch": 1.3335227966308594e-05,
      "step": 87983,
      "training_step_time": 7.704771041870117
    },
    {
      "epoch": 1.3336181640625e-05,
      "model_forward_time": 1.7799861431121826,
      "step": 87984
    },
    {
      "epoch": 1.3336181640625e-05,
      "step": 87984,
      "training_step_time": 8.08759331703186
    },
    {
      "epoch": 1.3337135314941406e-05,
      "model_forward_time": 1.7633044719696045,
      "step": 87985
    },
    {
      "epoch": 1.3337135314941406e-05,
      "step": 87985,
      "training_step_time": 7.456082582473755
    },
    {
      "epoch": 1.3338088989257813e-05,
      "model_forward_time": 1.779862880706787,
      "step": 87986
    },
    {
      "epoch": 1.3338088989257813e-05,
      "step": 87986,
      "training_step_time": 6.739441394805908
    },
    {
      "epoch": 1.3339042663574219e-05,
      "model_forward_time": 1.4107365608215332,
      "step": 87987
    },
    {
      "epoch": 1.3339042663574219e-05,
      "step": 87987,
      "training_step_time": 6.798240661621094
    },
    {
      "epoch": 1.3339996337890625e-05,
      "model_forward_time": 1.7733840942382812,
      "step": 87988
    },
    {
      "epoch": 1.3339996337890625e-05,
      "step": 87988,
      "training_step_time": 6.734524726867676
    },
    {
      "epoch": 1.3340950012207032e-05,
      "model_forward_time": 1.7655189037322998,
      "step": 87989
    },
    {
      "epoch": 1.3340950012207032e-05,
      "step": 87989,
      "training_step_time": 8.789817571640015
    },
    {
      "epoch": 1.3341903686523438e-05,
      "grad_norm": 0.23676955699920654,
      "learning_rate": 3.8925450138182275e-07,
      "loss": 0.0567,
      "step": 87990
    },
    {
      "epoch": 1.3341903686523438e-05,
      "model_forward_time": 1.7953810691833496,
      "step": 87990
    },
    {
      "dynamics_loss_avg": 0.06093231216073036,
      "epoch": 1.3341903686523438e-05,
      "step": 87990
    },
    {
      "action_loss_avg": 0.0012319298286456615,
      "epoch": 1.3341903686523438e-05,
      "step": 87990
    },
    {
      "epoch": 1.3341903686523438e-05,
      "step": 87990,
      "training_step_time": 10.591189622879028
    },
    {
      "epoch": 1.3342857360839843e-05,
      "model_forward_time": 2.4122345447540283,
      "step": 87991
    },
    {
      "epoch": 1.3342857360839843e-05,
      "step": 87991,
      "training_step_time": 10.819751977920532
    },
    {
      "epoch": 1.334381103515625e-05,
      "model_forward_time": 1.7619514465332031,
      "step": 87992
    },
    {
      "epoch": 1.334381103515625e-05,
      "step": 87992,
      "training_step_time": 7.818360805511475
    },
    {
      "epoch": 1.3344764709472657e-05,
      "model_forward_time": 1.7754158973693848,
      "step": 87993
    },
    {
      "epoch": 1.3344764709472657e-05,
      "step": 87993,
      "training_step_time": 7.571242094039917
    },
    {
      "epoch": 1.3345718383789062e-05,
      "model_forward_time": 1.7745740413665771,
      "step": 87994
    },
    {
      "epoch": 1.3345718383789062e-05,
      "step": 87994,
      "training_step_time": 7.669897079467773
    },
    {
      "epoch": 1.3346672058105468e-05,
      "model_forward_time": 1.7952766418457031,
      "step": 87995
    },
    {
      "epoch": 1.3346672058105468e-05,
      "step": 87995,
      "training_step_time": 6.749232292175293
    },
    {
      "epoch": 1.3347625732421875e-05,
      "model_forward_time": 1.775735855102539,
      "step": 87996
    },
    {
      "epoch": 1.3347625732421875e-05,
      "step": 87996,
      "training_step_time": 6.783702850341797
    },
    {
      "epoch": 1.3348579406738281e-05,
      "model_forward_time": 1.7766010761260986,
      "step": 87997
    },
    {
      "epoch": 1.3348579406738281e-05,
      "step": 87997,
      "training_step_time": 6.7740256786346436
    },
    {
      "epoch": 1.3349533081054687e-05,
      "model_forward_time": 1.446354627609253,
      "step": 87998
    },
    {
      "epoch": 1.3349533081054687e-05,
      "step": 87998,
      "training_step_time": 7.987070560455322
    },
    {
      "epoch": 1.3350486755371094e-05,
      "model_forward_time": 1.7847752571105957,
      "step": 87999
    },
    {
      "epoch": 1.3350486755371094e-05,
      "step": 87999,
      "training_step_time": 10.922460079193115
    },
    {
      "epoch": 1.33514404296875e-05,
      "grad_norm": 0.09814023971557617,
      "learning_rate": 3.8861513410566876e-07,
      "loss": 0.0559,
      "step": 88000
    },
    {
      "epoch": 1.33514404296875e-05,
      "model_forward_time": 3.0811877250671387,
      "step": 88000
    },
    {
      "dynamics_loss_avg": 0.05774493273347616,
      "epoch": 1.33514404296875e-05,
      "step": 88000
    },
    {
      "action_loss_avg": 0.0013187265896704048,
      "epoch": 1.33514404296875e-05,
      "step": 88000
    },
    {
      "epoch": 1.33514404296875e-05,
      "step": 88000,
      "training_step_time": 12.455727338790894
    },
    {
      "epoch": 1.3352394104003906e-05,
      "model_forward_time": 1.8299541473388672,
      "step": 88001
    },
    {
      "epoch": 1.3352394104003906e-05,
      "step": 88001,
      "training_step_time": 7.907696962356567
    },
    {
      "epoch": 1.3353347778320313e-05,
      "model_forward_time": 1.7675366401672363,
      "step": 88002
    },
    {
      "epoch": 1.3353347778320313e-05,
      "step": 88002,
      "training_step_time": 7.624067783355713
    },
    {
      "epoch": 1.3354301452636719e-05,
      "model_forward_time": 1.3889424800872803,
      "step": 88003
    },
    {
      "epoch": 1.3354301452636719e-05,
      "step": 88003,
      "training_step_time": 7.798841238021851
    },
    {
      "epoch": 1.3355255126953125e-05,
      "model_forward_time": 1.7625391483306885,
      "step": 88004
    },
    {
      "epoch": 1.3355255126953125e-05,
      "step": 88004,
      "training_step_time": 6.947449445724487
    },
    {
      "epoch": 1.3356208801269532e-05,
      "model_forward_time": 1.759453296661377,
      "step": 88005
    },
    {
      "epoch": 1.3356208801269532e-05,
      "step": 88005,
      "training_step_time": 6.864591360092163
    },
    {
      "epoch": 1.3357162475585938e-05,
      "model_forward_time": 1.759354591369629,
      "step": 88006
    },
    {
      "epoch": 1.3357162475585938e-05,
      "step": 88006,
      "training_step_time": 7.716845750808716
    },
    {
      "epoch": 1.3358116149902343e-05,
      "model_forward_time": 1.7704253196716309,
      "step": 88007
    },
    {
      "epoch": 1.3358116149902343e-05,
      "step": 88007,
      "training_step_time": 8.587945461273193
    },
    {
      "epoch": 1.335906982421875e-05,
      "model_forward_time": 1.7586207389831543,
      "step": 88008
    },
    {
      "epoch": 1.335906982421875e-05,
      "step": 88008,
      "training_step_time": 12.381344079971313
    },
    {
      "epoch": 1.3360023498535157e-05,
      "model_forward_time": 2.3956470489501953,
      "step": 88009
    },
    {
      "epoch": 1.3360023498535157e-05,
      "step": 88009,
      "training_step_time": 9.784253120422363
    },
    {
      "epoch": 1.3360977172851562e-05,
      "grad_norm": 0.07295972853899002,
      "learning_rate": 3.879762711236307e-07,
      "loss": 0.0558,
      "step": 88010
    },
    {
      "epoch": 1.3360977172851562e-05,
      "model_forward_time": 1.8223774433135986,
      "step": 88010
    },
    {
      "dynamics_loss_avg": 0.05278668031096458,
      "epoch": 1.3360977172851562e-05,
      "step": 88010
    },
    {
      "action_loss_avg": 0.0014903759642038495,
      "epoch": 1.3360977172851562e-05,
      "step": 88010
    },
    {
      "epoch": 1.3360977172851562e-05,
      "step": 88010,
      "training_step_time": 7.979184865951538
    },
    {
      "epoch": 1.3361930847167968e-05,
      "model_forward_time": 1.7712571620941162,
      "step": 88011
    },
    {
      "epoch": 1.3361930847167968e-05,
      "step": 88011,
      "training_step_time": 7.7750279903411865
    },
    {
      "epoch": 1.3362884521484375e-05,
      "model_forward_time": 1.7655689716339111,
      "step": 88012
    },
    {
      "epoch": 1.3362884521484375e-05,
      "step": 88012,
      "training_step_time": 7.723210334777832
    },
    {
      "epoch": 1.3363838195800781e-05,
      "model_forward_time": 1.7761852741241455,
      "step": 88013
    },
    {
      "epoch": 1.3363838195800781e-05,
      "step": 88013,
      "training_step_time": 6.897168397903442
    },
    {
      "epoch": 1.3364791870117187e-05,
      "model_forward_time": 1.770266056060791,
      "step": 88014
    },
    {
      "epoch": 1.3364791870117187e-05,
      "step": 88014,
      "training_step_time": 6.874318361282349
    },
    {
      "epoch": 1.3365745544433594e-05,
      "model_forward_time": 1.7714195251464844,
      "step": 88015
    },
    {
      "epoch": 1.3365745544433594e-05,
      "step": 88015,
      "training_step_time": 6.88308048248291
    },
    {
      "epoch": 1.336669921875e-05,
      "model_forward_time": 1.7691428661346436,
      "step": 88016
    },
    {
      "epoch": 1.336669921875e-05,
      "step": 88016,
      "training_step_time": 7.8636040687561035
    },
    {
      "epoch": 1.3367652893066406e-05,
      "model_forward_time": 1.7745306491851807,
      "step": 88017
    },
    {
      "epoch": 1.3367652893066406e-05,
      "step": 88017,
      "training_step_time": 9.785716533660889
    },
    {
      "epoch": 1.3368606567382813e-05,
      "model_forward_time": 1.7688570022583008,
      "step": 88018
    },
    {
      "epoch": 1.3368606567382813e-05,
      "step": 88018,
      "training_step_time": 11.929949283599854
    },
    {
      "epoch": 1.3369560241699219e-05,
      "model_forward_time": 1.8301966190338135,
      "step": 88019
    },
    {
      "epoch": 1.3369560241699219e-05,
      "step": 88019,
      "training_step_time": 10.553938627243042
    },
    {
      "epoch": 1.3370513916015625e-05,
      "grad_norm": 0.11771958321332932,
      "learning_rate": 3.873379125055743e-07,
      "loss": 0.0559,
      "step": 88020
    },
    {
      "epoch": 1.3370513916015625e-05,
      "model_forward_time": 1.7732882499694824,
      "step": 88020
    },
    {
      "dynamics_loss_avg": 0.04716151673346758,
      "epoch": 1.3370513916015625e-05,
      "step": 88020
    },
    {
      "action_loss_avg": 0.0018461653700796888,
      "epoch": 1.3370513916015625e-05,
      "step": 88020
    },
    {
      "epoch": 1.3370513916015625e-05,
      "step": 88020,
      "training_step_time": 7.649788856506348
    },
    {
      "epoch": 1.3371467590332032e-05,
      "model_forward_time": 1.7719085216522217,
      "step": 88021
    },
    {
      "epoch": 1.3371467590332032e-05,
      "step": 88021,
      "training_step_time": 7.775177240371704
    },
    {
      "epoch": 1.3372421264648438e-05,
      "model_forward_time": 1.7729907035827637,
      "step": 88022
    },
    {
      "epoch": 1.3372421264648438e-05,
      "step": 88022,
      "training_step_time": 6.84774112701416
    },
    {
      "epoch": 1.3373374938964843e-05,
      "model_forward_time": 1.7732369899749756,
      "step": 88023
    },
    {
      "epoch": 1.3373374938964843e-05,
      "step": 88023,
      "training_step_time": 6.828319549560547
    },
    {
      "epoch": 1.337432861328125e-05,
      "model_forward_time": 1.7800359725952148,
      "step": 88024
    },
    {
      "epoch": 1.337432861328125e-05,
      "step": 88024,
      "training_step_time": 6.828180551528931
    },
    {
      "epoch": 1.3375282287597657e-05,
      "model_forward_time": 1.7626957893371582,
      "step": 88025
    },
    {
      "epoch": 1.3375282287597657e-05,
      "step": 88025,
      "training_step_time": 8.303401231765747
    },
    {
      "epoch": 1.3376235961914062e-05,
      "model_forward_time": 1.7761421203613281,
      "step": 88026
    },
    {
      "epoch": 1.3376235961914062e-05,
      "step": 88026,
      "training_step_time": 7.795276403427124
    },
    {
      "epoch": 1.3377189636230468e-05,
      "model_forward_time": 1.7778286933898926,
      "step": 88027
    },
    {
      "epoch": 1.3377189636230468e-05,
      "step": 88027,
      "training_step_time": 7.877837181091309
    },
    {
      "epoch": 1.3378143310546875e-05,
      "model_forward_time": 2.446904420852661,
      "step": 88028
    },
    {
      "epoch": 1.3378143310546875e-05,
      "step": 88028,
      "training_step_time": 9.279218673706055
    },
    {
      "epoch": 1.3379096984863281e-05,
      "model_forward_time": 1.7849855422973633,
      "step": 88029
    },
    {
      "epoch": 1.3379096984863281e-05,
      "step": 88029,
      "training_step_time": 7.7318525314331055
    },
    {
      "epoch": 1.3380050659179687e-05,
      "grad_norm": 0.17170333862304688,
      "learning_rate": 3.8670005832130817e-07,
      "loss": 0.056,
      "step": 88030
    },
    {
      "epoch": 1.3380050659179687e-05,
      "model_forward_time": 1.7735886573791504,
      "step": 88030
    },
    {
      "dynamics_loss_avg": 0.05627476144582033,
      "epoch": 1.3380050659179687e-05,
      "step": 88030
    },
    {
      "action_loss_avg": 0.0015082320955116303,
      "epoch": 1.3380050659179687e-05,
      "step": 88030
    },
    {
      "epoch": 1.3380050659179687e-05,
      "step": 88030,
      "training_step_time": 7.696703910827637
    },
    {
      "epoch": 1.3381004333496094e-05,
      "model_forward_time": 1.766096830368042,
      "step": 88031
    },
    {
      "epoch": 1.3381004333496094e-05,
      "step": 88031,
      "training_step_time": 7.361619472503662
    },
    {
      "epoch": 1.33819580078125e-05,
      "model_forward_time": 1.7671024799346924,
      "step": 88032
    },
    {
      "epoch": 1.33819580078125e-05,
      "step": 88032,
      "training_step_time": 6.886216878890991
    },
    {
      "epoch": 1.3382911682128906e-05,
      "model_forward_time": 1.7688236236572266,
      "step": 88033
    },
    {
      "epoch": 1.3382911682128906e-05,
      "step": 88033,
      "training_step_time": 6.8864076137542725
    },
    {
      "epoch": 1.3383865356445313e-05,
      "model_forward_time": 1.7696118354797363,
      "step": 88034
    },
    {
      "epoch": 1.3383865356445313e-05,
      "step": 88034,
      "training_step_time": 7.766472101211548
    },
    {
      "epoch": 1.3384819030761719e-05,
      "model_forward_time": 1.7794084548950195,
      "step": 88035
    },
    {
      "epoch": 1.3384819030761719e-05,
      "step": 88035,
      "training_step_time": 9.402015924453735
    },
    {
      "epoch": 1.3385772705078125e-05,
      "model_forward_time": 1.7738070487976074,
      "step": 88036
    },
    {
      "epoch": 1.3385772705078125e-05,
      "step": 88036,
      "training_step_time": 8.89751124382019
    },
    {
      "epoch": 1.3386726379394532e-05,
      "model_forward_time": 2.4290432929992676,
      "step": 88037
    },
    {
      "epoch": 1.3386726379394532e-05,
      "step": 88037,
      "training_step_time": 8.965064764022827
    },
    {
      "epoch": 1.3387680053710938e-05,
      "model_forward_time": 1.7680647373199463,
      "step": 88038
    },
    {
      "epoch": 1.3387680053710938e-05,
      "step": 88038,
      "training_step_time": 7.794466018676758
    },
    {
      "epoch": 1.3388633728027343e-05,
      "model_forward_time": 1.7764272689819336,
      "step": 88039
    },
    {
      "epoch": 1.3388633728027343e-05,
      "step": 88039,
      "training_step_time": 7.711641311645508
    },
    {
      "epoch": 1.338958740234375e-05,
      "grad_norm": 0.10578498244285583,
      "learning_rate": 3.860627086405871e-07,
      "loss": 0.0557,
      "step": 88040
    },
    {
      "epoch": 1.338958740234375e-05,
      "model_forward_time": 1.7578246593475342,
      "step": 88040
    },
    {
      "dynamics_loss_avg": 0.061668135784566404,
      "epoch": 1.338958740234375e-05,
      "step": 88040
    },
    {
      "action_loss_avg": 0.0021178041759412737,
      "epoch": 1.338958740234375e-05,
      "step": 88040
    },
    {
      "epoch": 1.338958740234375e-05,
      "step": 88040,
      "training_step_time": 7.5177507400512695
    },
    {
      "epoch": 1.3390541076660157e-05,
      "model_forward_time": 1.7748563289642334,
      "step": 88041
    },
    {
      "epoch": 1.3390541076660157e-05,
      "step": 88041,
      "training_step_time": 6.8980653285980225
    },
    {
      "epoch": 1.3391494750976562e-05,
      "model_forward_time": 1.7717959880828857,
      "step": 88042
    },
    {
      "epoch": 1.3391494750976562e-05,
      "step": 88042,
      "training_step_time": 6.8772783279418945
    },
    {
      "epoch": 1.3392448425292968e-05,
      "model_forward_time": 1.7589550018310547,
      "step": 88043
    },
    {
      "epoch": 1.3392448425292968e-05,
      "step": 88043,
      "training_step_time": 7.747282981872559
    },
    {
      "epoch": 1.3393402099609375e-05,
      "model_forward_time": 1.764235496520996,
      "step": 88044
    },
    {
      "epoch": 1.3393402099609375e-05,
      "step": 88044,
      "training_step_time": 9.614092111587524
    },
    {
      "epoch": 1.3394355773925781e-05,
      "model_forward_time": 1.7725298404693604,
      "step": 88045
    },
    {
      "epoch": 1.3394355773925781e-05,
      "step": 88045,
      "training_step_time": 9.738328695297241
    },
    {
      "epoch": 1.3395309448242187e-05,
      "model_forward_time": 2.391284227371216,
      "step": 88046
    },
    {
      "epoch": 1.3395309448242187e-05,
      "step": 88046,
      "training_step_time": 10.629605054855347
    },
    {
      "epoch": 1.3396263122558594e-05,
      "model_forward_time": 1.8374316692352295,
      "step": 88047
    },
    {
      "epoch": 1.3396263122558594e-05,
      "step": 88047,
      "training_step_time": 9.468730926513672
    },
    {
      "epoch": 1.3397216796875e-05,
      "model_forward_time": 1.7697153091430664,
      "step": 88048
    },
    {
      "epoch": 1.3397216796875e-05,
      "step": 88048,
      "training_step_time": 7.681976079940796
    },
    {
      "epoch": 1.3398170471191406e-05,
      "model_forward_time": 1.7660949230194092,
      "step": 88049
    },
    {
      "epoch": 1.3398170471191406e-05,
      "step": 88049,
      "training_step_time": 7.697717905044556
    },
    {
      "epoch": 1.3399124145507813e-05,
      "grad_norm": 0.07394002377986908,
      "learning_rate": 3.8542586353311264e-07,
      "loss": 0.0573,
      "step": 88050
    },
    {
      "epoch": 1.3399124145507813e-05,
      "model_forward_time": 1.7590887546539307,
      "step": 88050
    },
    {
      "dynamics_loss_avg": 0.05497793741524219,
      "epoch": 1.3399124145507813e-05,
      "step": 88050
    },
    {
      "action_loss_avg": 0.0013772772043012082,
      "epoch": 1.3399124145507813e-05,
      "step": 88050
    },
    {
      "epoch": 1.3399124145507813e-05,
      "step": 88050,
      "training_step_time": 6.816260814666748
    },
    {
      "epoch": 1.3400077819824219e-05,
      "model_forward_time": 1.7613472938537598,
      "step": 88051
    },
    {
      "epoch": 1.3400077819824219e-05,
      "step": 88051,
      "training_step_time": 6.850572347640991
    },
    {
      "epoch": 1.3401031494140625e-05,
      "model_forward_time": 1.7654106616973877,
      "step": 88052
    },
    {
      "epoch": 1.3401031494140625e-05,
      "step": 88052,
      "training_step_time": 8.536420106887817
    },
    {
      "epoch": 1.3401985168457032e-05,
      "model_forward_time": 1.7620549201965332,
      "step": 88053
    },
    {
      "epoch": 1.3401985168457032e-05,
      "step": 88053,
      "training_step_time": 8.129371881484985
    },
    {
      "epoch": 1.3402938842773438e-05,
      "model_forward_time": 1.7531960010528564,
      "step": 88054
    },
    {
      "epoch": 1.3402938842773438e-05,
      "step": 88054,
      "training_step_time": 9.354247808456421
    },
    {
      "epoch": 1.3403892517089843e-05,
      "model_forward_time": 2.4123547077178955,
      "step": 88055
    },
    {
      "epoch": 1.3403892517089843e-05,
      "step": 88055,
      "training_step_time": 9.773735046386719
    },
    {
      "epoch": 1.340484619140625e-05,
      "model_forward_time": 1.8328642845153809,
      "step": 88056
    },
    {
      "epoch": 1.340484619140625e-05,
      "step": 88056,
      "training_step_time": 8.350596189498901
    },
    {
      "epoch": 1.3405799865722657e-05,
      "model_forward_time": 1.7704126834869385,
      "step": 88057
    },
    {
      "epoch": 1.3405799865722657e-05,
      "step": 88057,
      "training_step_time": 7.691599607467651
    },
    {
      "epoch": 1.3406753540039062e-05,
      "model_forward_time": 1.758683681488037,
      "step": 88058
    },
    {
      "epoch": 1.3406753540039062e-05,
      "step": 88058,
      "training_step_time": 7.6691319942474365
    },
    {
      "epoch": 1.3407707214355468e-05,
      "model_forward_time": 1.768472671508789,
      "step": 88059
    },
    {
      "epoch": 1.3407707214355468e-05,
      "step": 88059,
      "training_step_time": 7.5378265380859375
    },
    {
      "epoch": 1.3408660888671875e-05,
      "grad_norm": 0.07106813788414001,
      "learning_rate": 3.8478952306852667e-07,
      "loss": 0.0568,
      "step": 88060
    },
    {
      "epoch": 1.3408660888671875e-05,
      "model_forward_time": 1.7583820819854736,
      "step": 88060
    },
    {
      "dynamics_loss_avg": 0.07516051176935434,
      "epoch": 1.3408660888671875e-05,
      "step": 88060
    },
    {
      "action_loss_avg": 0.0023236061388161033,
      "epoch": 1.3408660888671875e-05,
      "step": 88060
    },
    {
      "epoch": 1.3408660888671875e-05,
      "step": 88060,
      "training_step_time": 6.890286684036255
    },
    {
      "epoch": 1.3409614562988281e-05,
      "model_forward_time": 1.7529363632202148,
      "step": 88061
    },
    {
      "epoch": 1.3409614562988281e-05,
      "step": 88061,
      "training_step_time": 7.754798889160156
    },
    {
      "epoch": 1.3410568237304687e-05,
      "model_forward_time": 1.7595112323760986,
      "step": 88062
    },
    {
      "epoch": 1.3410568237304687e-05,
      "step": 88062,
      "training_step_time": 8.5650315284729
    },
    {
      "epoch": 1.3411521911621094e-05,
      "model_forward_time": 1.7524394989013672,
      "step": 88063
    },
    {
      "epoch": 1.3411521911621094e-05,
      "step": 88063,
      "training_step_time": 8.833195686340332
    },
    {
      "epoch": 1.34124755859375e-05,
      "model_forward_time": 2.4375908374786377,
      "step": 88064
    },
    {
      "epoch": 1.34124755859375e-05,
      "step": 88064,
      "training_step_time": 8.848638772964478
    },
    {
      "epoch": 1.3413429260253906e-05,
      "model_forward_time": 1.7837307453155518,
      "step": 88065
    },
    {
      "epoch": 1.3413429260253906e-05,
      "step": 88065,
      "training_step_time": 9.014111042022705
    },
    {
      "epoch": 1.3414382934570313e-05,
      "model_forward_time": 1.7707099914550781,
      "step": 88066
    },
    {
      "epoch": 1.3414382934570313e-05,
      "step": 88066,
      "training_step_time": 7.716459274291992
    },
    {
      "epoch": 1.3415336608886719e-05,
      "model_forward_time": 1.7716197967529297,
      "step": 88067
    },
    {
      "epoch": 1.3415336608886719e-05,
      "step": 88067,
      "training_step_time": 7.526310443878174
    },
    {
      "epoch": 1.3416290283203125e-05,
      "model_forward_time": 1.7663373947143555,
      "step": 88068
    },
    {
      "epoch": 1.3416290283203125e-05,
      "step": 88068,
      "training_step_time": 7.726862668991089
    },
    {
      "epoch": 1.3417243957519532e-05,
      "model_forward_time": 1.767683982849121,
      "step": 88069
    },
    {
      "epoch": 1.3417243957519532e-05,
      "step": 88069,
      "training_step_time": 6.943706512451172
    },
    {
      "epoch": 1.3418197631835938e-05,
      "grad_norm": 0.0942101702094078,
      "learning_rate": 3.841536873164198e-07,
      "loss": 0.0554,
      "step": 88070
    },
    {
      "epoch": 1.3418197631835938e-05,
      "model_forward_time": 1.7711806297302246,
      "step": 88070
    },
    {
      "dynamics_loss_avg": 0.05406220844015479,
      "epoch": 1.3418197631835938e-05,
      "step": 88070
    },
    {
      "action_loss_avg": 0.0013930245710071175,
      "epoch": 1.3418197631835938e-05,
      "step": 88070
    },
    {
      "epoch": 1.3418197631835938e-05,
      "step": 88070,
      "training_step_time": 6.9391562938690186
    },
    {
      "epoch": 1.3419151306152343e-05,
      "model_forward_time": 1.768784523010254,
      "step": 88071
    },
    {
      "epoch": 1.3419151306152343e-05,
      "step": 88071,
      "training_step_time": 7.8581178188323975
    },
    {
      "epoch": 1.342010498046875e-05,
      "model_forward_time": 1.7701060771942139,
      "step": 88072
    },
    {
      "epoch": 1.342010498046875e-05,
      "step": 88072,
      "training_step_time": 9.173316478729248
    },
    {
      "epoch": 1.3421058654785157e-05,
      "model_forward_time": 2.8449482917785645,
      "step": 88073
    },
    {
      "epoch": 1.3421058654785157e-05,
      "step": 88073,
      "training_step_time": 9.607592821121216
    },
    {
      "epoch": 1.3422012329101562e-05,
      "model_forward_time": 1.7924323081970215,
      "step": 88074
    },
    {
      "epoch": 1.3422012329101562e-05,
      "step": 88074,
      "training_step_time": 9.45056700706482
    },
    {
      "epoch": 1.3422966003417968e-05,
      "model_forward_time": 1.7751121520996094,
      "step": 88075
    },
    {
      "epoch": 1.3422966003417968e-05,
      "step": 88075,
      "training_step_time": 7.77886438369751
    },
    {
      "epoch": 1.3423919677734375e-05,
      "model_forward_time": 1.762037992477417,
      "step": 88076
    },
    {
      "epoch": 1.3423919677734375e-05,
      "step": 88076,
      "training_step_time": 7.616286277770996
    },
    {
      "epoch": 1.3424873352050781e-05,
      "model_forward_time": 1.7667522430419922,
      "step": 88077
    },
    {
      "epoch": 1.3424873352050781e-05,
      "step": 88077,
      "training_step_time": 7.732923269271851
    },
    {
      "epoch": 1.3425827026367187e-05,
      "model_forward_time": 1.7617413997650146,
      "step": 88078
    },
    {
      "epoch": 1.3425827026367187e-05,
      "step": 88078,
      "training_step_time": 6.9258246421813965
    },
    {
      "epoch": 1.3426780700683594e-05,
      "model_forward_time": 1.3951494693756104,
      "step": 88079
    },
    {
      "epoch": 1.3426780700683594e-05,
      "step": 88079,
      "training_step_time": 6.947739839553833
    },
    {
      "epoch": 1.3427734375e-05,
      "grad_norm": 0.08533749729394913,
      "learning_rate": 3.8351835634632626e-07,
      "loss": 0.0564,
      "step": 88080
    },
    {
      "epoch": 1.3427734375e-05,
      "model_forward_time": 1.7747104167938232,
      "step": 88080
    },
    {
      "dynamics_loss_avg": 0.040727516263723375,
      "epoch": 1.3427734375e-05,
      "step": 88080
    },
    {
      "action_loss_avg": 0.001778835168806836,
      "epoch": 1.3427734375e-05,
      "step": 88080
    },
    {
      "epoch": 1.3427734375e-05,
      "step": 88080,
      "training_step_time": 8.754176616668701
    },
    {
      "epoch": 1.3428688049316406e-05,
      "model_forward_time": 1.7372677326202393,
      "step": 88081
    },
    {
      "epoch": 1.3428688049316406e-05,
      "step": 88081,
      "training_step_time": 8.82359266281128
    },
    {
      "epoch": 1.3429641723632813e-05,
      "model_forward_time": 1.7643280029296875,
      "step": 88082
    },
    {
      "epoch": 1.3429641723632813e-05,
      "step": 88082,
      "training_step_time": 9.72636342048645
    },
    {
      "epoch": 1.3430595397949219e-05,
      "model_forward_time": 2.860664129257202,
      "step": 88083
    },
    {
      "epoch": 1.3430595397949219e-05,
      "step": 88083,
      "training_step_time": 9.64819073677063
    },
    {
      "epoch": 1.3431549072265625e-05,
      "model_forward_time": 1.764127254486084,
      "step": 88084
    },
    {
      "epoch": 1.3431549072265625e-05,
      "step": 88084,
      "training_step_time": 7.754641532897949
    },
    {
      "epoch": 1.3432502746582032e-05,
      "model_forward_time": 1.7492177486419678,
      "step": 88085
    },
    {
      "epoch": 1.3432502746582032e-05,
      "step": 88085,
      "training_step_time": 8.619962215423584
    },
    {
      "epoch": 1.3433456420898438e-05,
      "model_forward_time": 1.7553200721740723,
      "step": 88086
    },
    {
      "epoch": 1.3433456420898438e-05,
      "step": 88086,
      "training_step_time": 7.635887622833252
    },
    {
      "epoch": 1.3434410095214844e-05,
      "model_forward_time": 1.7575078010559082,
      "step": 88087
    },
    {
      "epoch": 1.3434410095214844e-05,
      "step": 88087,
      "training_step_time": 6.842270374298096
    },
    {
      "epoch": 1.343536376953125e-05,
      "model_forward_time": 1.7593488693237305,
      "step": 88088
    },
    {
      "epoch": 1.343536376953125e-05,
      "step": 88088,
      "training_step_time": 6.846176385879517
    },
    {
      "epoch": 1.3436317443847657e-05,
      "model_forward_time": 1.382338047027588,
      "step": 88089
    },
    {
      "epoch": 1.3436317443847657e-05,
      "step": 88089,
      "training_step_time": 8.571788549423218
    },
    {
      "epoch": 1.3437271118164062e-05,
      "grad_norm": 0.06586135178804398,
      "learning_rate": 3.8288353022772453e-07,
      "loss": 0.0563,
      "step": 88090
    },
    {
      "epoch": 1.3437271118164062e-05,
      "model_forward_time": 1.7504734992980957,
      "step": 88090
    },
    {
      "dynamics_loss_avg": 0.06089777387678623,
      "epoch": 1.3437271118164062e-05,
      "step": 88090
    },
    {
      "action_loss_avg": 0.0012100373016437516,
      "epoch": 1.3437271118164062e-05,
      "step": 88090
    },
    {
      "epoch": 1.3437271118164062e-05,
      "step": 88090,
      "training_step_time": 7.7038469314575195
    },
    {
      "epoch": 1.3438224792480468e-05,
      "model_forward_time": 1.7506933212280273,
      "step": 88091
    },
    {
      "epoch": 1.3438224792480468e-05,
      "step": 88091,
      "training_step_time": 10.728116035461426
    },
    {
      "epoch": 1.3439178466796876e-05,
      "model_forward_time": 2.849489450454712,
      "step": 88092
    },
    {
      "epoch": 1.3439178466796876e-05,
      "step": 88092,
      "training_step_time": 8.581228494644165
    },
    {
      "epoch": 1.3440132141113281e-05,
      "model_forward_time": 1.7753024101257324,
      "step": 88093
    },
    {
      "epoch": 1.3440132141113281e-05,
      "step": 88093,
      "training_step_time": 8.8193359375
    },
    {
      "epoch": 1.3441085815429687e-05,
      "model_forward_time": 1.7555291652679443,
      "step": 88094
    },
    {
      "epoch": 1.3441085815429687e-05,
      "step": 88094,
      "training_step_time": 7.668545246124268
    },
    {
      "epoch": 1.3442039489746094e-05,
      "model_forward_time": 1.768540620803833,
      "step": 88095
    },
    {
      "epoch": 1.3442039489746094e-05,
      "step": 88095,
      "training_step_time": 9.026484251022339
    },
    {
      "epoch": 1.34429931640625e-05,
      "model_forward_time": 1.3884332180023193,
      "step": 88096
    },
    {
      "epoch": 1.34429931640625e-05,
      "step": 88096,
      "training_step_time": 6.830580949783325
    },
    {
      "epoch": 1.3443946838378906e-05,
      "model_forward_time": 1.758484125137329,
      "step": 88097
    },
    {
      "epoch": 1.3443946838378906e-05,
      "step": 88097,
      "training_step_time": 6.84347128868103
    },
    {
      "epoch": 1.3444900512695313e-05,
      "model_forward_time": 1.756021499633789,
      "step": 88098
    },
    {
      "epoch": 1.3444900512695313e-05,
      "step": 88098,
      "training_step_time": 7.694231986999512
    },
    {
      "epoch": 1.3445854187011719e-05,
      "model_forward_time": 1.3993592262268066,
      "step": 88099
    },
    {
      "epoch": 1.3445854187011719e-05,
      "step": 88099,
      "training_step_time": 7.752924680709839
    },
    {
      "epoch": 1.3446807861328125e-05,
      "grad_norm": 0.09166412800550461,
      "learning_rate": 3.822492090300378e-07,
      "loss": 0.0555,
      "step": 88100
    },
    {
      "epoch": 1.3446807861328125e-05,
      "model_forward_time": 1.758774757385254,
      "step": 88100
    },
    {
      "dynamics_loss_avg": 0.05893558189272881,
      "epoch": 1.3446807861328125e-05,
      "step": 88100
    },
    {
      "action_loss_avg": 0.002397240107529797,
      "epoch": 1.3446807861328125e-05,
      "step": 88100
    },
    {
      "epoch": 1.3446807861328125e-05,
      "step": 88100,
      "training_step_time": 8.521388053894043
    },
    {
      "epoch": 1.3447761535644532e-05,
      "model_forward_time": 1.7590842247009277,
      "step": 88101
    },
    {
      "epoch": 1.3447761535644532e-05,
      "step": 88101,
      "training_step_time": 9.679969310760498
    },
    {
      "epoch": 1.3448715209960938e-05,
      "model_forward_time": 1.8169119358062744,
      "step": 88102
    },
    {
      "epoch": 1.3448715209960938e-05,
      "step": 88102,
      "training_step_time": 6.824839115142822
    },
    {
      "epoch": 1.3449668884277344e-05,
      "model_forward_time": 1.763916015625,
      "step": 88103
    },
    {
      "epoch": 1.3449668884277344e-05,
      "step": 88103,
      "training_step_time": 7.643903017044067
    },
    {
      "epoch": 1.345062255859375e-05,
      "model_forward_time": 1.7748663425445557,
      "step": 88104
    },
    {
      "epoch": 1.345062255859375e-05,
      "step": 88104,
      "training_step_time": 7.180279970169067
    },
    {
      "epoch": 1.3451576232910157e-05,
      "model_forward_time": 1.7881395816802979,
      "step": 88105
    },
    {
      "epoch": 1.3451576232910157e-05,
      "step": 88105,
      "training_step_time": 7.673452138900757
    },
    {
      "epoch": 1.3452529907226562e-05,
      "model_forward_time": 1.7682795524597168,
      "step": 88106
    },
    {
      "epoch": 1.3452529907226562e-05,
      "step": 88106,
      "training_step_time": 6.822524070739746
    },
    {
      "epoch": 1.3453483581542968e-05,
      "model_forward_time": 1.746901035308838,
      "step": 88107
    },
    {
      "epoch": 1.3453483581542968e-05,
      "step": 88107,
      "training_step_time": 7.517158269882202
    },
    {
      "epoch": 1.3454437255859376e-05,
      "model_forward_time": 1.7650296688079834,
      "step": 88108
    },
    {
      "epoch": 1.3454437255859376e-05,
      "step": 88108,
      "training_step_time": 7.749663591384888
    },
    {
      "epoch": 1.3455390930175781e-05,
      "model_forward_time": 1.7620160579681396,
      "step": 88109
    },
    {
      "epoch": 1.3455390930175781e-05,
      "step": 88109,
      "training_step_time": 8.793411254882812
    },
    {
      "epoch": 1.3456344604492187e-05,
      "grad_norm": 0.08918532729148865,
      "learning_rate": 3.816153928226357e-07,
      "loss": 0.0551,
      "step": 88110
    },
    {
      "epoch": 1.3456344604492187e-05,
      "model_forward_time": 1.765455722808838,
      "step": 88110
    },
    {
      "dynamics_loss_avg": 0.06103394562378526,
      "epoch": 1.3456344604492187e-05,
      "step": 88110
    },
    {
      "action_loss_avg": 0.0017709483625367283,
      "epoch": 1.3456344604492187e-05,
      "step": 88110
    },
    {
      "epoch": 1.3456344604492187e-05,
      "step": 88110,
      "training_step_time": 9.297351837158203
    },
    {
      "epoch": 1.3457298278808594e-05,
      "model_forward_time": 1.7907273769378662,
      "step": 88111
    },
    {
      "epoch": 1.3457298278808594e-05,
      "step": 88111,
      "training_step_time": 7.826632022857666
    },
    {
      "epoch": 1.3458251953125e-05,
      "model_forward_time": 1.7545139789581299,
      "step": 88112
    },
    {
      "epoch": 1.3458251953125e-05,
      "step": 88112,
      "training_step_time": 8.759786605834961
    },
    {
      "epoch": 1.3459205627441406e-05,
      "model_forward_time": 1.7601165771484375,
      "step": 88113
    },
    {
      "epoch": 1.3459205627441406e-05,
      "step": 88113,
      "training_step_time": 7.348628044128418
    },
    {
      "epoch": 1.3460159301757813e-05,
      "model_forward_time": 1.7575981616973877,
      "step": 88114
    },
    {
      "epoch": 1.3460159301757813e-05,
      "step": 88114,
      "training_step_time": 7.734963417053223
    },
    {
      "epoch": 1.3461112976074219e-05,
      "model_forward_time": 1.750363826751709,
      "step": 88115
    },
    {
      "epoch": 1.3461112976074219e-05,
      "step": 88115,
      "training_step_time": 6.890131235122681
    },
    {
      "epoch": 1.3462066650390625e-05,
      "model_forward_time": 1.7516772747039795,
      "step": 88116
    },
    {
      "epoch": 1.3462066650390625e-05,
      "step": 88116,
      "training_step_time": 8.334564208984375
    },
    {
      "epoch": 1.3463020324707032e-05,
      "model_forward_time": 1.7608089447021484,
      "step": 88117
    },
    {
      "epoch": 1.3463020324707032e-05,
      "step": 88117,
      "training_step_time": 7.8249266147613525
    },
    {
      "epoch": 1.3463973999023438e-05,
      "model_forward_time": 1.7662546634674072,
      "step": 88118
    },
    {
      "epoch": 1.3463973999023438e-05,
      "step": 88118,
      "training_step_time": 7.786144256591797
    },
    {
      "epoch": 1.3464927673339844e-05,
      "model_forward_time": 1.7617273330688477,
      "step": 88119
    },
    {
      "epoch": 1.3464927673339844e-05,
      "step": 88119,
      "training_step_time": 9.510159730911255
    },
    {
      "epoch": 1.346588134765625e-05,
      "grad_norm": 0.09342481940984726,
      "learning_rate": 3.8098208167483e-07,
      "loss": 0.0552,
      "step": 88120
    },
    {
      "epoch": 1.346588134765625e-05,
      "model_forward_time": 2.923753023147583,
      "step": 88120
    },
    {
      "dynamics_loss_avg": 0.06032089181244373,
      "epoch": 1.346588134765625e-05,
      "step": 88120
    },
    {
      "action_loss_avg": 0.0018121070868801325,
      "epoch": 1.346588134765625e-05,
      "step": 88120
    },
    {
      "epoch": 1.346588134765625e-05,
      "step": 88120,
      "training_step_time": 8.96109676361084
    },
    {
      "epoch": 1.3466835021972657e-05,
      "model_forward_time": 1.7554421424865723,
      "step": 88121
    },
    {
      "epoch": 1.3466835021972657e-05,
      "step": 88121,
      "training_step_time": 8.660165548324585
    },
    {
      "epoch": 1.3467788696289062e-05,
      "model_forward_time": 1.7614755630493164,
      "step": 88122
    },
    {
      "epoch": 1.3467788696289062e-05,
      "step": 88122,
      "training_step_time": 8.713348150253296
    },
    {
      "epoch": 1.3468742370605468e-05,
      "model_forward_time": 1.7623910903930664,
      "step": 88123
    },
    {
      "epoch": 1.3468742370605468e-05,
      "step": 88123,
      "training_step_time": 7.786017656326294
    },
    {
      "epoch": 1.3469696044921876e-05,
      "model_forward_time": 1.7568933963775635,
      "step": 88124
    },
    {
      "epoch": 1.3469696044921876e-05,
      "step": 88124,
      "training_step_time": 6.884053945541382
    },
    {
      "epoch": 1.3470649719238281e-05,
      "model_forward_time": 1.7552955150604248,
      "step": 88125
    },
    {
      "epoch": 1.3470649719238281e-05,
      "step": 88125,
      "training_step_time": 7.571109771728516
    },
    {
      "epoch": 1.3471603393554687e-05,
      "model_forward_time": 1.7534406185150146,
      "step": 88126
    },
    {
      "epoch": 1.3471603393554687e-05,
      "step": 88126,
      "training_step_time": 7.792182207107544
    },
    {
      "epoch": 1.3472557067871094e-05,
      "model_forward_time": 1.3722245693206787,
      "step": 88127
    },
    {
      "epoch": 1.3472557067871094e-05,
      "step": 88127,
      "training_step_time": 8.747834205627441
    },
    {
      "epoch": 1.34735107421875e-05,
      "model_forward_time": 1.7508292198181152,
      "step": 88128
    },
    {
      "epoch": 1.34735107421875e-05,
      "step": 88128,
      "training_step_time": 8.384302616119385
    },
    {
      "epoch": 1.3474464416503906e-05,
      "model_forward_time": 2.926802635192871,
      "step": 88129
    },
    {
      "epoch": 1.3474464416503906e-05,
      "step": 88129,
      "training_step_time": 7.872798681259155
    },
    {
      "epoch": 1.3475418090820313e-05,
      "grad_norm": 0.09936801344156265,
      "learning_rate": 3.8034927565587964e-07,
      "loss": 0.055,
      "step": 88130
    },
    {
      "epoch": 1.3475418090820313e-05,
      "model_forward_time": 1.7678942680358887,
      "step": 88130
    },
    {
      "dynamics_loss_avg": 0.06025337893515825,
      "epoch": 1.3475418090820313e-05,
      "step": 88130
    },
    {
      "action_loss_avg": 0.0015330462949350477,
      "epoch": 1.3475418090820313e-05,
      "step": 88130
    },
    {
      "epoch": 1.3475418090820313e-05,
      "step": 88130,
      "training_step_time": 7.863612174987793
    },
    {
      "epoch": 1.3476371765136719e-05,
      "model_forward_time": 1.7535591125488281,
      "step": 88131
    },
    {
      "epoch": 1.3476371765136719e-05,
      "step": 88131,
      "training_step_time": 7.719355344772339
    },
    {
      "epoch": 1.3477325439453125e-05,
      "model_forward_time": 1.753864049911499,
      "step": 88132
    },
    {
      "epoch": 1.3477325439453125e-05,
      "step": 88132,
      "training_step_time": 7.685811758041382
    },
    {
      "epoch": 1.347827911376953e-05,
      "model_forward_time": 1.754607915878296,
      "step": 88133
    },
    {
      "epoch": 1.347827911376953e-05,
      "step": 88133,
      "training_step_time": 7.211172103881836
    },
    {
      "epoch": 1.3479232788085938e-05,
      "model_forward_time": 1.7560303211212158,
      "step": 88134
    },
    {
      "epoch": 1.3479232788085938e-05,
      "step": 88134,
      "training_step_time": 7.550876140594482
    },
    {
      "epoch": 1.3480186462402344e-05,
      "model_forward_time": 1.7571783065795898,
      "step": 88135
    },
    {
      "epoch": 1.3480186462402344e-05,
      "step": 88135,
      "training_step_time": 7.756737947463989
    },
    {
      "epoch": 1.348114013671875e-05,
      "model_forward_time": 1.3877441883087158,
      "step": 88136
    },
    {
      "epoch": 1.348114013671875e-05,
      "step": 88136,
      "training_step_time": 7.764660835266113
    },
    {
      "epoch": 1.3482093811035157e-05,
      "model_forward_time": 1.7621281147003174,
      "step": 88137
    },
    {
      "epoch": 1.3482093811035157e-05,
      "step": 88137,
      "training_step_time": 9.741297006607056
    },
    {
      "epoch": 1.3483047485351562e-05,
      "model_forward_time": 2.881589412689209,
      "step": 88138
    },
    {
      "epoch": 1.3483047485351562e-05,
      "step": 88138,
      "training_step_time": 8.62227988243103
    },
    {
      "epoch": 1.3484001159667968e-05,
      "model_forward_time": 1.7731132507324219,
      "step": 88139
    },
    {
      "epoch": 1.3484001159667968e-05,
      "step": 88139,
      "training_step_time": 8.676204442977905
    },
    {
      "epoch": 1.3484954833984376e-05,
      "grad_norm": 0.1314987689256668,
      "learning_rate": 3.797169748349855e-07,
      "loss": 0.0566,
      "step": 88140
    },
    {
      "epoch": 1.3484954833984376e-05,
      "model_forward_time": 1.7398910522460938,
      "step": 88140
    },
    {
      "dynamics_loss_avg": 0.056274589896202085,
      "epoch": 1.3484954833984376e-05,
      "step": 88140
    },
    {
      "action_loss_avg": 0.0017939030251000077,
      "epoch": 1.3484954833984376e-05,
      "step": 88140
    },
    {
      "epoch": 1.3484954833984376e-05,
      "step": 88140,
      "training_step_time": 7.828345060348511
    },
    {
      "epoch": 1.3485908508300781e-05,
      "model_forward_time": 1.7446250915527344,
      "step": 88141
    },
    {
      "epoch": 1.3485908508300781e-05,
      "step": 88141,
      "training_step_time": 7.652742147445679
    },
    {
      "epoch": 1.3486862182617187e-05,
      "model_forward_time": 1.7486062049865723,
      "step": 88142
    },
    {
      "epoch": 1.3486862182617187e-05,
      "step": 88142,
      "training_step_time": 7.188983917236328
    },
    {
      "epoch": 1.3487815856933594e-05,
      "model_forward_time": 1.7431566715240479,
      "step": 88143
    },
    {
      "epoch": 1.3487815856933594e-05,
      "step": 88143,
      "training_step_time": 7.559696674346924
    },
    {
      "epoch": 1.348876953125e-05,
      "model_forward_time": 1.73183012008667,
      "step": 88144
    },
    {
      "epoch": 1.348876953125e-05,
      "step": 88144,
      "training_step_time": 8.24690318107605
    },
    {
      "epoch": 1.3489723205566406e-05,
      "model_forward_time": 1.737861156463623,
      "step": 88145
    },
    {
      "epoch": 1.3489723205566406e-05,
      "step": 88145,
      "training_step_time": 8.73421573638916
    },
    {
      "epoch": 1.3490676879882813e-05,
      "model_forward_time": 1.7400972843170166,
      "step": 88146
    },
    {
      "epoch": 1.3490676879882813e-05,
      "step": 88146,
      "training_step_time": 8.31389832496643
    },
    {
      "epoch": 1.3491630554199219e-05,
      "model_forward_time": 2.7978835105895996,
      "step": 88147
    },
    {
      "epoch": 1.3491630554199219e-05,
      "step": 88147,
      "training_step_time": 8.628427028656006
    },
    {
      "epoch": 1.3492584228515625e-05,
      "model_forward_time": 1.777561902999878,
      "step": 88148
    },
    {
      "epoch": 1.3492584228515625e-05,
      "step": 88148,
      "training_step_time": 7.737647294998169
    },
    {
      "epoch": 1.349353790283203e-05,
      "model_forward_time": 1.742292881011963,
      "step": 88149
    },
    {
      "epoch": 1.349353790283203e-05,
      "step": 88149,
      "training_step_time": 8.617681980133057
    },
    {
      "epoch": 1.3494491577148438e-05,
      "grad_norm": 0.11620022356510162,
      "learning_rate": 3.790851792812977e-07,
      "loss": 0.0548,
      "step": 88150
    },
    {
      "epoch": 1.3494491577148438e-05,
      "model_forward_time": 1.735264539718628,
      "step": 88150
    },
    {
      "dynamics_loss_avg": 0.06497414484620094,
      "epoch": 1.3494491577148438e-05,
      "step": 88150
    },
    {
      "action_loss_avg": 0.0020873055676929654,
      "epoch": 1.3494491577148438e-05,
      "step": 88150
    },
    {
      "epoch": 1.3494491577148438e-05,
      "step": 88150,
      "training_step_time": 7.70581579208374
    },
    {
      "epoch": 1.3495445251464844e-05,
      "model_forward_time": 1.7395763397216797,
      "step": 88151
    },
    {
      "epoch": 1.3495445251464844e-05,
      "step": 88151,
      "training_step_time": 7.314924001693726
    },
    {
      "epoch": 1.349639892578125e-05,
      "model_forward_time": 1.744842767715454,
      "step": 88152
    },
    {
      "epoch": 1.349639892578125e-05,
      "step": 88152,
      "training_step_time": 7.53792405128479
    },
    {
      "epoch": 1.3497352600097657e-05,
      "model_forward_time": 1.3749644756317139,
      "step": 88153
    },
    {
      "epoch": 1.3497352600097657e-05,
      "step": 88153,
      "training_step_time": 7.4217729568481445
    },
    {
      "epoch": 1.3498306274414062e-05,
      "model_forward_time": 1.7352135181427002,
      "step": 88154
    },
    {
      "epoch": 1.3498306274414062e-05,
      "step": 88154,
      "training_step_time": 8.396286487579346
    },
    {
      "epoch": 1.3499259948730468e-05,
      "model_forward_time": 1.7461800575256348,
      "step": 88155
    },
    {
      "epoch": 1.3499259948730468e-05,
      "step": 88155,
      "training_step_time": 9.007082462310791
    },
    {
      "epoch": 1.3500213623046876e-05,
      "model_forward_time": 1.7523202896118164,
      "step": 88156
    },
    {
      "epoch": 1.3500213623046876e-05,
      "step": 88156,
      "training_step_time": 7.945459365844727
    },
    {
      "epoch": 1.3501167297363281e-05,
      "model_forward_time": 2.8931872844696045,
      "step": 88157
    },
    {
      "epoch": 1.3501167297363281e-05,
      "step": 88157,
      "training_step_time": 8.8346688747406
    },
    {
      "epoch": 1.3502120971679687e-05,
      "model_forward_time": 1.7397830486297607,
      "step": 88158
    },
    {
      "epoch": 1.3502120971679687e-05,
      "step": 88158,
      "training_step_time": 7.757822513580322
    },
    {
      "epoch": 1.3503074645996094e-05,
      "model_forward_time": 1.7599728107452393,
      "step": 88159
    },
    {
      "epoch": 1.3503074645996094e-05,
      "step": 88159,
      "training_step_time": 8.653367042541504
    },
    {
      "epoch": 1.35040283203125e-05,
      "grad_norm": 0.08031857013702393,
      "learning_rate": 3.7845388906390724e-07,
      "loss": 0.0555,
      "step": 88160
    },
    {
      "epoch": 1.35040283203125e-05,
      "model_forward_time": 1.7415246963500977,
      "step": 88160
    },
    {
      "dynamics_loss_avg": 0.06909694205969572,
      "epoch": 1.35040283203125e-05,
      "step": 88160
    },
    {
      "action_loss_avg": 0.001484423060901463,
      "epoch": 1.35040283203125e-05,
      "step": 88160
    },
    {
      "epoch": 1.35040283203125e-05,
      "step": 88160,
      "training_step_time": 7.665173768997192
    },
    {
      "epoch": 1.3504981994628906e-05,
      "model_forward_time": 1.7524328231811523,
      "step": 88161
    },
    {
      "epoch": 1.3504981994628906e-05,
      "step": 88161,
      "training_step_time": 7.517950773239136
    },
    {
      "epoch": 1.3505935668945313e-05,
      "model_forward_time": 1.7445068359375,
      "step": 88162
    },
    {
      "epoch": 1.3505935668945313e-05,
      "step": 88162,
      "training_step_time": 7.361472129821777
    },
    {
      "epoch": 1.3506889343261719e-05,
      "model_forward_time": 1.7545504570007324,
      "step": 88163
    },
    {
      "epoch": 1.3506889343261719e-05,
      "step": 88163,
      "training_step_time": 9.077552318572998
    },
    {
      "epoch": 1.3507843017578125e-05,
      "model_forward_time": 1.7454330921173096,
      "step": 88164
    },
    {
      "epoch": 1.3507843017578125e-05,
      "step": 88164,
      "training_step_time": 7.7493555545806885
    },
    {
      "epoch": 1.350879669189453e-05,
      "model_forward_time": 1.7400684356689453,
      "step": 88165
    },
    {
      "epoch": 1.350879669189453e-05,
      "step": 88165,
      "training_step_time": 7.912067413330078
    },
    {
      "epoch": 1.3509750366210938e-05,
      "model_forward_time": 2.847870349884033,
      "step": 88166
    },
    {
      "epoch": 1.3509750366210938e-05,
      "step": 88166,
      "training_step_time": 9.544857263565063
    },
    {
      "epoch": 1.3510704040527344e-05,
      "model_forward_time": 1.8245625495910645,
      "step": 88167
    },
    {
      "epoch": 1.3510704040527344e-05,
      "step": 88167,
      "training_step_time": 8.573668956756592
    },
    {
      "epoch": 1.351165771484375e-05,
      "model_forward_time": 1.7501118183135986,
      "step": 88168
    },
    {
      "epoch": 1.351165771484375e-05,
      "step": 88168,
      "training_step_time": 7.651836156845093
    },
    {
      "epoch": 1.3512611389160157e-05,
      "model_forward_time": 1.7438092231750488,
      "step": 88169
    },
    {
      "epoch": 1.3512611389160157e-05,
      "step": 88169,
      "training_step_time": 7.225658893585205
    },
    {
      "epoch": 1.3513565063476562e-05,
      "grad_norm": 0.08481750637292862,
      "learning_rate": 3.7782310425185155e-07,
      "loss": 0.0548,
      "step": 88170
    },
    {
      "epoch": 1.3513565063476562e-05,
      "model_forward_time": 1.7419605255126953,
      "step": 88170
    },
    {
      "dynamics_loss_avg": 0.06274990625679493,
      "epoch": 1.3513565063476562e-05,
      "step": 88170
    },
    {
      "action_loss_avg": 0.002031860381248407,
      "epoch": 1.3513565063476562e-05,
      "step": 88170
    },
    {
      "epoch": 1.3513565063476562e-05,
      "step": 88170,
      "training_step_time": 7.532659292221069
    },
    {
      "epoch": 1.3514518737792968e-05,
      "model_forward_time": 1.7441089153289795,
      "step": 88171
    },
    {
      "epoch": 1.3514518737792968e-05,
      "step": 88171,
      "training_step_time": 7.287447690963745
    },
    {
      "epoch": 1.3515472412109376e-05,
      "model_forward_time": 1.7436017990112305,
      "step": 88172
    },
    {
      "epoch": 1.3515472412109376e-05,
      "step": 88172,
      "training_step_time": 7.604297876358032
    },
    {
      "epoch": 1.3516426086425781e-05,
      "model_forward_time": 1.7394073009490967,
      "step": 88173
    },
    {
      "epoch": 1.3516426086425781e-05,
      "step": 88173,
      "training_step_time": 7.794431209564209
    },
    {
      "epoch": 1.3517379760742187e-05,
      "model_forward_time": 1.7394609451293945,
      "step": 88174
    },
    {
      "epoch": 1.3517379760742187e-05,
      "step": 88174,
      "training_step_time": 10.141541957855225
    },
    {
      "epoch": 1.3518333435058594e-05,
      "model_forward_time": 2.818046808242798,
      "step": 88175
    },
    {
      "epoch": 1.3518333435058594e-05,
      "step": 88175,
      "training_step_time": 8.409495830535889
    },
    {
      "epoch": 1.3519287109375e-05,
      "model_forward_time": 1.8122189044952393,
      "step": 88176
    },
    {
      "epoch": 1.3519287109375e-05,
      "step": 88176,
      "training_step_time": 8.626875877380371
    },
    {
      "epoch": 1.3520240783691406e-05,
      "model_forward_time": 1.7479352951049805,
      "step": 88177
    },
    {
      "epoch": 1.3520240783691406e-05,
      "step": 88177,
      "training_step_time": 7.673884868621826
    },
    {
      "epoch": 1.3521194458007813e-05,
      "model_forward_time": 1.7513668537139893,
      "step": 88178
    },
    {
      "epoch": 1.3521194458007813e-05,
      "step": 88178,
      "training_step_time": 7.614719867706299
    },
    {
      "epoch": 1.3522148132324219e-05,
      "model_forward_time": 1.7506628036499023,
      "step": 88179
    },
    {
      "epoch": 1.3522148132324219e-05,
      "step": 88179,
      "training_step_time": 8.123363018035889
    },
    {
      "epoch": 1.3523101806640625e-05,
      "grad_norm": 0.07681597769260406,
      "learning_rate": 3.77192824914111e-07,
      "loss": 0.0557,
      "step": 88180
    },
    {
      "epoch": 1.3523101806640625e-05,
      "model_forward_time": 1.7415997982025146,
      "step": 88180
    },
    {
      "dynamics_loss_avg": 0.061556576378643514,
      "epoch": 1.3523101806640625e-05,
      "step": 88180
    },
    {
      "action_loss_avg": 0.002445484831696376,
      "epoch": 1.3523101806640625e-05,
      "step": 88180
    },
    {
      "epoch": 1.3523101806640625e-05,
      "step": 88180,
      "training_step_time": 6.81720757484436
    },
    {
      "epoch": 1.352405548095703e-05,
      "model_forward_time": 1.7442247867584229,
      "step": 88181
    },
    {
      "epoch": 1.352405548095703e-05,
      "step": 88181,
      "training_step_time": 7.735759735107422
    },
    {
      "epoch": 1.3525009155273438e-05,
      "model_forward_time": 1.7420017719268799,
      "step": 88182
    },
    {
      "epoch": 1.3525009155273438e-05,
      "step": 88182,
      "training_step_time": 8.624555349349976
    },
    {
      "epoch": 1.3525962829589844e-05,
      "model_forward_time": 1.741410493850708,
      "step": 88183
    },
    {
      "epoch": 1.3525962829589844e-05,
      "step": 88183,
      "training_step_time": 7.786266565322876
    },
    {
      "epoch": 1.352691650390625e-05,
      "model_forward_time": 2.8148629665374756,
      "step": 88184
    },
    {
      "epoch": 1.352691650390625e-05,
      "step": 88184,
      "training_step_time": 9.209187746047974
    },
    {
      "epoch": 1.3527870178222657e-05,
      "model_forward_time": 1.8209211826324463,
      "step": 88185
    },
    {
      "epoch": 1.3527870178222657e-05,
      "step": 88185,
      "training_step_time": 9.805940389633179
    },
    {
      "epoch": 1.3528823852539062e-05,
      "model_forward_time": 1.7401213645935059,
      "step": 88186
    },
    {
      "epoch": 1.3528823852539062e-05,
      "step": 88186,
      "training_step_time": 7.565141201019287
    },
    {
      "epoch": 1.3529777526855468e-05,
      "model_forward_time": 1.7460265159606934,
      "step": 88187
    },
    {
      "epoch": 1.3529777526855468e-05,
      "step": 88187,
      "training_step_time": 7.613498210906982
    },
    {
      "epoch": 1.3530731201171876e-05,
      "model_forward_time": 1.7407844066619873,
      "step": 88188
    },
    {
      "epoch": 1.3530731201171876e-05,
      "step": 88188,
      "training_step_time": 7.455637693405151
    },
    {
      "epoch": 1.3531684875488281e-05,
      "model_forward_time": 1.7494373321533203,
      "step": 88189
    },
    {
      "epoch": 1.3531684875488281e-05,
      "step": 88189,
      "training_step_time": 7.497329235076904
    },
    {
      "epoch": 1.3532638549804687e-05,
      "grad_norm": 0.10353397578001022,
      "learning_rate": 3.765630511196133e-07,
      "loss": 0.0563,
      "step": 88190
    },
    {
      "epoch": 1.3532638549804687e-05,
      "model_forward_time": 1.7413649559020996,
      "step": 88190
    },
    {
      "dynamics_loss_avg": 0.06372246742248536,
      "epoch": 1.3532638549804687e-05,
      "step": 88190
    },
    {
      "action_loss_avg": 0.002036550367483869,
      "epoch": 1.3532638549804687e-05,
      "step": 88190
    },
    {
      "epoch": 1.3532638549804687e-05,
      "step": 88190,
      "training_step_time": 7.722118854522705
    },
    {
      "epoch": 1.3533592224121094e-05,
      "model_forward_time": 1.7479887008666992,
      "step": 88191
    },
    {
      "epoch": 1.3533592224121094e-05,
      "step": 88191,
      "training_step_time": 7.727093935012817
    },
    {
      "epoch": 1.35345458984375e-05,
      "model_forward_time": 1.7413837909698486,
      "step": 88192
    },
    {
      "epoch": 1.35345458984375e-05,
      "step": 88192,
      "training_step_time": 7.753441572189331
    },
    {
      "epoch": 1.3535499572753906e-05,
      "model_forward_time": 2.7840476036071777,
      "step": 88193
    },
    {
      "epoch": 1.3535499572753906e-05,
      "step": 88193,
      "training_step_time": 9.973782300949097
    },
    {
      "epoch": 1.3536453247070313e-05,
      "model_forward_time": 1.7483301162719727,
      "step": 88194
    },
    {
      "epoch": 1.3536453247070313e-05,
      "step": 88194,
      "training_step_time": 9.43856167793274
    },
    {
      "epoch": 1.3537406921386719e-05,
      "model_forward_time": 1.7376627922058105,
      "step": 88195
    },
    {
      "epoch": 1.3537406921386719e-05,
      "step": 88195,
      "training_step_time": 7.702868700027466
    },
    {
      "epoch": 1.3538360595703125e-05,
      "model_forward_time": 1.7446188926696777,
      "step": 88196
    },
    {
      "epoch": 1.3538360595703125e-05,
      "step": 88196,
      "training_step_time": 7.660294771194458
    },
    {
      "epoch": 1.353931427001953e-05,
      "model_forward_time": 1.0445258617401123,
      "step": 88197
    },
    {
      "epoch": 1.353931427001953e-05,
      "step": 88197,
      "training_step_time": 7.459566116333008
    },
    {
      "epoch": 1.3540267944335938e-05,
      "model_forward_time": 1.7445838451385498,
      "step": 88198
    },
    {
      "epoch": 1.3540267944335938e-05,
      "step": 88198,
      "training_step_time": 7.489092826843262
    },
    {
      "epoch": 1.3541221618652344e-05,
      "model_forward_time": 1.7412528991699219,
      "step": 88199
    },
    {
      "epoch": 1.3541221618652344e-05,
      "step": 88199,
      "training_step_time": 7.693690061569214
    },
    {
      "epoch": 1.354217529296875e-05,
      "grad_norm": 0.13591966032981873,
      "learning_rate": 3.7593378293722783e-07,
      "loss": 0.0563,
      "step": 88200
    },
    {
      "epoch": 1.354217529296875e-05,
      "model_forward_time": 1.7463676929473877,
      "step": 88200
    },
    {
      "dynamics_loss_avg": 0.07143034804612398,
      "epoch": 1.354217529296875e-05,
      "step": 88200
    },
    {
      "action_loss_avg": 0.0021028647024650127,
      "epoch": 1.354217529296875e-05,
      "step": 88200
    },
    {
      "epoch": 1.354217529296875e-05,
      "step": 88200,
      "training_step_time": 8.271467208862305
    },
    {
      "epoch": 1.3543128967285157e-05,
      "model_forward_time": 1.7529950141906738,
      "step": 88201
    },
    {
      "epoch": 1.3543128967285157e-05,
      "step": 88201,
      "training_step_time": 8.966152667999268
    },
    {
      "epoch": 1.3544082641601562e-05,
      "model_forward_time": 1.7449519634246826,
      "step": 88202
    },
    {
      "epoch": 1.3544082641601562e-05,
      "step": 88202,
      "training_step_time": 8.496687650680542
    },
    {
      "epoch": 1.3545036315917968e-05,
      "model_forward_time": 2.843196153640747,
      "step": 88203
    },
    {
      "epoch": 1.3545036315917968e-05,
      "step": 88203,
      "training_step_time": 9.085339069366455
    },
    {
      "epoch": 1.3545989990234376e-05,
      "model_forward_time": 1.769374132156372,
      "step": 88204
    },
    {
      "epoch": 1.3545989990234376e-05,
      "step": 88204,
      "training_step_time": 8.467985153198242
    },
    {
      "epoch": 1.3546943664550781e-05,
      "model_forward_time": 1.7462129592895508,
      "step": 88205
    },
    {
      "epoch": 1.3546943664550781e-05,
      "step": 88205,
      "training_step_time": 7.632210969924927
    },
    {
      "epoch": 1.3547897338867187e-05,
      "model_forward_time": 1.741901159286499,
      "step": 88206
    },
    {
      "epoch": 1.3547897338867187e-05,
      "step": 88206,
      "training_step_time": 7.600757837295532
    },
    {
      "epoch": 1.3548851013183594e-05,
      "model_forward_time": 1.742905616760254,
      "step": 88207
    },
    {
      "epoch": 1.3548851013183594e-05,
      "step": 88207,
      "training_step_time": 7.51487135887146
    },
    {
      "epoch": 1.35498046875e-05,
      "model_forward_time": 1.7352008819580078,
      "step": 88208
    },
    {
      "epoch": 1.35498046875e-05,
      "step": 88208,
      "training_step_time": 7.476032495498657
    },
    {
      "epoch": 1.3550758361816406e-05,
      "model_forward_time": 1.7415311336517334,
      "step": 88209
    },
    {
      "epoch": 1.3550758361816406e-05,
      "step": 88209,
      "training_step_time": 8.737935066223145
    },
    {
      "epoch": 1.3551712036132813e-05,
      "grad_norm": 0.07998327910900116,
      "learning_rate": 3.7530502043577334e-07,
      "loss": 0.0563,
      "step": 88210
    },
    {
      "epoch": 1.3551712036132813e-05,
      "model_forward_time": 1.7414405345916748,
      "step": 88210
    },
    {
      "dynamics_loss_avg": 0.05831682235002518,
      "epoch": 1.3551712036132813e-05,
      "step": 88210
    },
    {
      "action_loss_avg": 0.0015887155022937804,
      "epoch": 1.3551712036132813e-05,
      "step": 88210
    },
    {
      "epoch": 1.3551712036132813e-05,
      "step": 88210,
      "training_step_time": 8.726952075958252
    },
    {
      "epoch": 1.3552665710449219e-05,
      "model_forward_time": 1.7388191223144531,
      "step": 88211
    },
    {
      "epoch": 1.3552665710449219e-05,
      "step": 88211,
      "training_step_time": 7.993072032928467
    },
    {
      "epoch": 1.3553619384765625e-05,
      "model_forward_time": 2.820258617401123,
      "step": 88212
    },
    {
      "epoch": 1.3553619384765625e-05,
      "step": 88212,
      "training_step_time": 8.42476224899292
    },
    {
      "epoch": 1.355457305908203e-05,
      "model_forward_time": 1.7973699569702148,
      "step": 88213
    },
    {
      "epoch": 1.355457305908203e-05,
      "step": 88213,
      "training_step_time": 8.693397045135498
    },
    {
      "epoch": 1.3555526733398438e-05,
      "model_forward_time": 1.7528424263000488,
      "step": 88214
    },
    {
      "epoch": 1.3555526733398438e-05,
      "step": 88214,
      "training_step_time": 7.652745962142944
    },
    {
      "epoch": 1.3556480407714844e-05,
      "model_forward_time": 1.7481591701507568,
      "step": 88215
    },
    {
      "epoch": 1.3556480407714844e-05,
      "step": 88215,
      "training_step_time": 7.782445907592773
    },
    {
      "epoch": 1.355743408203125e-05,
      "model_forward_time": 1.7489829063415527,
      "step": 88216
    },
    {
      "epoch": 1.355743408203125e-05,
      "step": 88216,
      "training_step_time": 7.56523060798645
    },
    {
      "epoch": 1.3558387756347657e-05,
      "model_forward_time": 1.7369213104248047,
      "step": 88217
    },
    {
      "epoch": 1.3558387756347657e-05,
      "step": 88217,
      "training_step_time": 7.3395676612854
    },
    {
      "epoch": 1.3559341430664062e-05,
      "model_forward_time": 1.744616985321045,
      "step": 88218
    },
    {
      "epoch": 1.3559341430664062e-05,
      "step": 88218,
      "training_step_time": 8.306445837020874
    },
    {
      "epoch": 1.3560295104980468e-05,
      "model_forward_time": 1.7392418384552002,
      "step": 88219
    },
    {
      "epoch": 1.3560295104980468e-05,
      "step": 88219,
      "training_step_time": 9.494763612747192
    },
    {
      "epoch": 1.3561248779296876e-05,
      "grad_norm": 0.08715399354696274,
      "learning_rate": 3.746767636840076e-07,
      "loss": 0.0561,
      "step": 88220
    },
    {
      "epoch": 1.3561248779296876e-05,
      "model_forward_time": 1.3731269836425781,
      "step": 88220
    },
    {
      "dynamics_loss_avg": 0.05668447520583868,
      "epoch": 1.3561248779296876e-05,
      "step": 88220
    },
    {
      "action_loss_avg": 0.0018568763975054027,
      "epoch": 1.3561248779296876e-05,
      "step": 88220
    },
    {
      "epoch": 1.3561248779296876e-05,
      "step": 88220,
      "training_step_time": 9.871685028076172
    },
    {
      "epoch": 1.3562202453613281e-05,
      "model_forward_time": 1.0502049922943115,
      "step": 88221
    },
    {
      "epoch": 1.3562202453613281e-05,
      "step": 88221,
      "training_step_time": 9.238947629928589
    },
    {
      "epoch": 1.3563156127929687e-05,
      "model_forward_time": 2.893440008163452,
      "step": 88222
    },
    {
      "epoch": 1.3563156127929687e-05,
      "step": 88222,
      "training_step_time": 7.784662485122681
    },
    {
      "epoch": 1.3564109802246094e-05,
      "model_forward_time": 1.3756344318389893,
      "step": 88223
    },
    {
      "epoch": 1.3564109802246094e-05,
      "step": 88223,
      "training_step_time": 7.6194775104522705
    },
    {
      "epoch": 1.35650634765625e-05,
      "model_forward_time": 1.741513967514038,
      "step": 88224
    },
    {
      "epoch": 1.35650634765625e-05,
      "step": 88224,
      "training_step_time": 7.801284074783325
    },
    {
      "epoch": 1.3566017150878906e-05,
      "model_forward_time": 1.7405369281768799,
      "step": 88225
    },
    {
      "epoch": 1.3566017150878906e-05,
      "step": 88225,
      "training_step_time": 7.390614032745361
    },
    {
      "epoch": 1.3566970825195313e-05,
      "model_forward_time": 1.7399816513061523,
      "step": 88226
    },
    {
      "epoch": 1.3566970825195313e-05,
      "step": 88226,
      "training_step_time": 7.570230960845947
    },
    {
      "epoch": 1.3567924499511719e-05,
      "model_forward_time": 1.7466764450073242,
      "step": 88227
    },
    {
      "epoch": 1.3567924499511719e-05,
      "step": 88227,
      "training_step_time": 7.79742169380188
    },
    {
      "epoch": 1.3568878173828125e-05,
      "model_forward_time": 1.7345914840698242,
      "step": 88228
    },
    {
      "epoch": 1.3568878173828125e-05,
      "step": 88228,
      "training_step_time": 7.801121711730957
    },
    {
      "epoch": 1.356983184814453e-05,
      "model_forward_time": 1.731292486190796,
      "step": 88229
    },
    {
      "epoch": 1.356983184814453e-05,
      "step": 88229,
      "training_step_time": 8.492652654647827
    },
    {
      "epoch": 1.3570785522460938e-05,
      "grad_norm": 0.0997094064950943,
      "learning_rate": 3.74049012750638e-07,
      "loss": 0.0552,
      "step": 88230
    },
    {
      "epoch": 1.3570785522460938e-05,
      "model_forward_time": 1.748368740081787,
      "step": 88230
    },
    {
      "dynamics_loss_avg": 0.06752520874142647,
      "epoch": 1.3570785522460938e-05,
      "step": 88230
    },
    {
      "action_loss_avg": 0.0010633224854245781,
      "epoch": 1.3570785522460938e-05,
      "step": 88230
    },
    {
      "epoch": 1.3570785522460938e-05,
      "step": 88230,
      "training_step_time": 9.651709794998169
    },
    {
      "epoch": 1.3571739196777344e-05,
      "model_forward_time": 2.8220107555389404,
      "step": 88231
    },
    {
      "epoch": 1.3571739196777344e-05,
      "step": 88231,
      "training_step_time": 8.89668321609497
    },
    {
      "epoch": 1.357269287109375e-05,
      "model_forward_time": 1.7482831478118896,
      "step": 88232
    },
    {
      "epoch": 1.357269287109375e-05,
      "step": 88232,
      "training_step_time": 9.328854322433472
    },
    {
      "epoch": 1.3573646545410157e-05,
      "model_forward_time": 1.7323029041290283,
      "step": 88233
    },
    {
      "epoch": 1.3573646545410157e-05,
      "step": 88233,
      "training_step_time": 7.788816452026367
    },
    {
      "epoch": 1.3574600219726562e-05,
      "model_forward_time": 1.7432019710540771,
      "step": 88234
    },
    {
      "epoch": 1.3574600219726562e-05,
      "step": 88234,
      "training_step_time": 7.389832258224487
    },
    {
      "epoch": 1.3575553894042968e-05,
      "model_forward_time": 1.7438468933105469,
      "step": 88235
    },
    {
      "epoch": 1.3575553894042968e-05,
      "step": 88235,
      "training_step_time": 8.569562435150146
    },
    {
      "epoch": 1.3576507568359376e-05,
      "model_forward_time": 1.746825933456421,
      "step": 88236
    },
    {
      "epoch": 1.3576507568359376e-05,
      "step": 88236,
      "training_step_time": 7.802558422088623
    },
    {
      "epoch": 1.3577461242675781e-05,
      "model_forward_time": 1.7345027923583984,
      "step": 88237
    },
    {
      "epoch": 1.3577461242675781e-05,
      "step": 88237,
      "training_step_time": 7.849388837814331
    },
    {
      "epoch": 1.3578414916992187e-05,
      "model_forward_time": 1.732642412185669,
      "step": 88238
    },
    {
      "epoch": 1.3578414916992187e-05,
      "step": 88238,
      "training_step_time": 7.740305662155151
    },
    {
      "epoch": 1.3579368591308594e-05,
      "model_forward_time": 1.7510724067687988,
      "step": 88239
    },
    {
      "epoch": 1.3579368591308594e-05,
      "step": 88239,
      "training_step_time": 8.803374290466309
    },
    {
      "epoch": 1.3580322265625e-05,
      "grad_norm": 0.0855301097035408,
      "learning_rate": 3.7342176770431284e-07,
      "loss": 0.0573,
      "step": 88240
    },
    {
      "epoch": 1.3580322265625e-05,
      "model_forward_time": 2.830557346343994,
      "step": 88240
    },
    {
      "dynamics_loss_avg": 0.061472314037382604,
      "epoch": 1.3580322265625e-05,
      "step": 88240
    },
    {
      "action_loss_avg": 0.003243130980990827,
      "epoch": 1.3580322265625e-05,
      "step": 88240
    },
    {
      "epoch": 1.3580322265625e-05,
      "step": 88240,
      "training_step_time": 7.7863545417785645
    },
    {
      "epoch": 1.3581275939941406e-05,
      "model_forward_time": 1.7518579959869385,
      "step": 88241
    },
    {
      "epoch": 1.3581275939941406e-05,
      "step": 88241,
      "training_step_time": 8.222190380096436
    },
    {
      "epoch": 1.3582229614257813e-05,
      "model_forward_time": 1.73750638961792,
      "step": 88242
    },
    {
      "epoch": 1.3582229614257813e-05,
      "step": 88242,
      "training_step_time": 8.270730018615723
    },
    {
      "epoch": 1.3583183288574219e-05,
      "model_forward_time": 1.734908103942871,
      "step": 88243
    },
    {
      "epoch": 1.3583183288574219e-05,
      "step": 88243,
      "training_step_time": 7.705796957015991
    },
    {
      "epoch": 1.3584136962890625e-05,
      "model_forward_time": 1.736266851425171,
      "step": 88244
    },
    {
      "epoch": 1.3584136962890625e-05,
      "step": 88244,
      "training_step_time": 7.567778587341309
    },
    {
      "epoch": 1.358509063720703e-05,
      "model_forward_time": 1.746886968612671,
      "step": 88245
    },
    {
      "epoch": 1.358509063720703e-05,
      "step": 88245,
      "training_step_time": 7.7068190574646
    },
    {
      "epoch": 1.3586044311523438e-05,
      "model_forward_time": 1.7387192249298096,
      "step": 88246
    },
    {
      "epoch": 1.3586044311523438e-05,
      "step": 88246,
      "training_step_time": 9.56598424911499
    },
    {
      "epoch": 1.3586997985839844e-05,
      "model_forward_time": 1.738053560256958,
      "step": 88247
    },
    {
      "epoch": 1.3586997985839844e-05,
      "step": 88247,
      "training_step_time": 7.8240580558776855
    },
    {
      "epoch": 1.358795166015625e-05,
      "model_forward_time": 1.745218276977539,
      "step": 88248
    },
    {
      "epoch": 1.358795166015625e-05,
      "step": 88248,
      "training_step_time": 8.84967041015625
    },
    {
      "epoch": 1.3588905334472657e-05,
      "model_forward_time": 2.7937276363372803,
      "step": 88249
    },
    {
      "epoch": 1.3588905334472657e-05,
      "step": 88249,
      "training_step_time": 9.29935097694397
    },
    {
      "epoch": 1.3589859008789062e-05,
      "grad_norm": 0.08374223113059998,
      "learning_rate": 3.727950286136267e-07,
      "loss": 0.0563,
      "step": 88250
    },
    {
      "epoch": 1.3589859008789062e-05,
      "model_forward_time": 1.8163752555847168,
      "step": 88250
    },
    {
      "dynamics_loss_avg": 0.06418930310755969,
      "epoch": 1.3589859008789062e-05,
      "step": 88250
    },
    {
      "action_loss_avg": 0.0031336659740190953,
      "epoch": 1.3589859008789062e-05,
      "step": 88250
    },
    {
      "epoch": 1.3589859008789062e-05,
      "step": 88250,
      "training_step_time": 7.691194534301758
    },
    {
      "epoch": 1.3590812683105468e-05,
      "model_forward_time": 1.7482044696807861,
      "step": 88251
    },
    {
      "epoch": 1.3590812683105468e-05,
      "step": 88251,
      "training_step_time": 7.689377784729004
    },
    {
      "epoch": 1.3591766357421876e-05,
      "model_forward_time": 1.7418627738952637,
      "step": 88252
    },
    {
      "epoch": 1.3591766357421876e-05,
      "step": 88252,
      "training_step_time": 7.722105264663696
    },
    {
      "epoch": 1.3592720031738281e-05,
      "model_forward_time": 1.7497966289520264,
      "step": 88253
    },
    {
      "epoch": 1.3592720031738281e-05,
      "step": 88253,
      "training_step_time": 7.556396007537842
    },
    {
      "epoch": 1.3593673706054687e-05,
      "model_forward_time": 1.7431683540344238,
      "step": 88254
    },
    {
      "epoch": 1.3593673706054687e-05,
      "step": 88254,
      "training_step_time": 7.6663947105407715
    },
    {
      "epoch": 1.3594627380371094e-05,
      "model_forward_time": 1.75075101852417,
      "step": 88255
    },
    {
      "epoch": 1.3594627380371094e-05,
      "step": 88255,
      "training_step_time": 7.83487343788147
    },
    {
      "epoch": 1.35955810546875e-05,
      "model_forward_time": 1.7293047904968262,
      "step": 88256
    },
    {
      "epoch": 1.35955810546875e-05,
      "step": 88256,
      "training_step_time": 9.740576267242432
    },
    {
      "epoch": 1.3596534729003906e-05,
      "model_forward_time": 1.745987892150879,
      "step": 88257
    },
    {
      "epoch": 1.3596534729003906e-05,
      "step": 88257,
      "training_step_time": 8.444975137710571
    },
    {
      "epoch": 1.3597488403320312e-05,
      "model_forward_time": 1.7488598823547363,
      "step": 88258
    },
    {
      "epoch": 1.3597488403320312e-05,
      "step": 88258,
      "training_step_time": 8.462453365325928
    },
    {
      "epoch": 1.3598442077636719e-05,
      "model_forward_time": 2.8673794269561768,
      "step": 88259
    },
    {
      "epoch": 1.3598442077636719e-05,
      "step": 88259,
      "training_step_time": 7.765172004699707
    },
    {
      "epoch": 1.3599395751953125e-05,
      "grad_norm": 0.07843280583620071,
      "learning_rate": 3.721687955471193e-07,
      "loss": 0.0563,
      "step": 88260
    },
    {
      "epoch": 1.3599395751953125e-05,
      "model_forward_time": 1.7388384342193604,
      "step": 88260
    },
    {
      "dynamics_loss_avg": 0.07492915857583285,
      "epoch": 1.3599395751953125e-05,
      "step": 88260
    },
    {
      "action_loss_avg": 0.002439273247728124,
      "epoch": 1.3599395751953125e-05,
      "step": 88260
    },
    {
      "epoch": 1.3599395751953125e-05,
      "step": 88260,
      "training_step_time": 7.690986156463623
    },
    {
      "epoch": 1.360034942626953e-05,
      "model_forward_time": 1.7422921657562256,
      "step": 88261
    },
    {
      "epoch": 1.360034942626953e-05,
      "step": 88261,
      "training_step_time": 7.7247865200042725
    },
    {
      "epoch": 1.3601303100585938e-05,
      "model_forward_time": 1.742049217224121,
      "step": 88262
    },
    {
      "epoch": 1.3601303100585938e-05,
      "step": 88262,
      "training_step_time": 7.59476375579834
    },
    {
      "epoch": 1.3602256774902344e-05,
      "model_forward_time": 1.7439887523651123,
      "step": 88263
    },
    {
      "epoch": 1.3602256774902344e-05,
      "step": 88263,
      "training_step_time": 7.671318531036377
    },
    {
      "epoch": 1.360321044921875e-05,
      "model_forward_time": 1.74021577835083,
      "step": 88264
    },
    {
      "epoch": 1.360321044921875e-05,
      "step": 88264,
      "training_step_time": 7.874169588088989
    },
    {
      "epoch": 1.3604164123535157e-05,
      "model_forward_time": 1.7458980083465576,
      "step": 88265
    },
    {
      "epoch": 1.3604164123535157e-05,
      "step": 88265,
      "training_step_time": 8.789756059646606
    },
    {
      "epoch": 1.3605117797851563e-05,
      "model_forward_time": 1.7507827281951904,
      "step": 88266
    },
    {
      "epoch": 1.3605117797851563e-05,
      "step": 88266,
      "training_step_time": 8.503077507019043
    },
    {
      "epoch": 1.3606071472167968e-05,
      "model_forward_time": 1.7501087188720703,
      "step": 88267
    },
    {
      "epoch": 1.3606071472167968e-05,
      "step": 88267,
      "training_step_time": 8.406416654586792
    },
    {
      "epoch": 1.3607025146484376e-05,
      "model_forward_time": 2.9067323207855225,
      "step": 88268
    },
    {
      "epoch": 1.3607025146484376e-05,
      "step": 88268,
      "training_step_time": 7.80874228477478
    },
    {
      "epoch": 1.3607978820800781e-05,
      "model_forward_time": 1.766726016998291,
      "step": 88269
    },
    {
      "epoch": 1.3607978820800781e-05,
      "step": 88269,
      "training_step_time": 8.621206760406494
    },
    {
      "epoch": 1.3608932495117187e-05,
      "grad_norm": 0.07932000607252121,
      "learning_rate": 3.715430685732757e-07,
      "loss": 0.0561,
      "step": 88270
    },
    {
      "epoch": 1.3608932495117187e-05,
      "model_forward_time": 1.7546429634094238,
      "step": 88270
    },
    {
      "dynamics_loss_avg": 0.05737673845142126,
      "epoch": 1.3608932495117187e-05,
      "step": 88270
    },
    {
      "action_loss_avg": 0.001768953533610329,
      "epoch": 1.3608932495117187e-05,
      "step": 88270
    },
    {
      "epoch": 1.3608932495117187e-05,
      "step": 88270,
      "training_step_time": 7.7947211265563965
    },
    {
      "epoch": 1.3609886169433595e-05,
      "model_forward_time": 1.7660655975341797,
      "step": 88271
    },
    {
      "epoch": 1.3609886169433595e-05,
      "step": 88271,
      "training_step_time": 7.5117223262786865
    },
    {
      "epoch": 1.361083984375e-05,
      "model_forward_time": 1.766247272491455,
      "step": 88272
    },
    {
      "epoch": 1.361083984375e-05,
      "step": 88272,
      "training_step_time": 7.681769132614136
    },
    {
      "epoch": 1.3611793518066406e-05,
      "model_forward_time": 1.7611525058746338,
      "step": 88273
    },
    {
      "epoch": 1.3611793518066406e-05,
      "step": 88273,
      "training_step_time": 7.875434637069702
    },
    {
      "epoch": 1.3612747192382812e-05,
      "model_forward_time": 1.7609355449676514,
      "step": 88274
    },
    {
      "epoch": 1.3612747192382812e-05,
      "step": 88274,
      "training_step_time": 7.840656995773315
    },
    {
      "epoch": 1.3613700866699219e-05,
      "model_forward_time": 1.765082597732544,
      "step": 88275
    },
    {
      "epoch": 1.3613700866699219e-05,
      "step": 88275,
      "training_step_time": 9.46543002128601
    },
    {
      "epoch": 1.3614654541015625e-05,
      "model_forward_time": 1.757944107055664,
      "step": 88276
    },
    {
      "epoch": 1.3614654541015625e-05,
      "step": 88276,
      "training_step_time": 10.154206275939941
    },
    {
      "epoch": 1.361560821533203e-05,
      "model_forward_time": 2.938145875930786,
      "step": 88277
    },
    {
      "epoch": 1.361560821533203e-05,
      "step": 88277,
      "training_step_time": 7.849310874938965
    },
    {
      "epoch": 1.3616561889648438e-05,
      "model_forward_time": 1.7640438079833984,
      "step": 88278
    },
    {
      "epoch": 1.3616561889648438e-05,
      "step": 88278,
      "training_step_time": 7.729389905929565
    },
    {
      "epoch": 1.3617515563964844e-05,
      "model_forward_time": 1.7569844722747803,
      "step": 88279
    },
    {
      "epoch": 1.3617515563964844e-05,
      "step": 88279,
      "training_step_time": 8.241117477416992
    },
    {
      "epoch": 1.361846923828125e-05,
      "grad_norm": 0.07863099128007889,
      "learning_rate": 3.709178477605224e-07,
      "loss": 0.0559,
      "step": 88280
    },
    {
      "epoch": 1.361846923828125e-05,
      "model_forward_time": 1.729504108428955,
      "step": 88280
    },
    {
      "dynamics_loss_avg": 0.05525715071707964,
      "epoch": 1.361846923828125e-05,
      "step": 88280
    },
    {
      "action_loss_avg": 0.001430940962745808,
      "epoch": 1.361846923828125e-05,
      "step": 88280
    },
    {
      "epoch": 1.361846923828125e-05,
      "step": 88280,
      "training_step_time": 7.5985682010650635
    },
    {
      "epoch": 1.3619422912597657e-05,
      "model_forward_time": 1.7443838119506836,
      "step": 88281
    },
    {
      "epoch": 1.3619422912597657e-05,
      "step": 88281,
      "training_step_time": 7.709636449813843
    },
    {
      "epoch": 1.3620376586914063e-05,
      "model_forward_time": 1.752037525177002,
      "step": 88282
    },
    {
      "epoch": 1.3620376586914063e-05,
      "step": 88282,
      "training_step_time": 7.821814775466919
    },
    {
      "epoch": 1.3621330261230468e-05,
      "model_forward_time": 1.7459731101989746,
      "step": 88283
    },
    {
      "epoch": 1.3621330261230468e-05,
      "step": 88283,
      "training_step_time": 7.834810256958008
    },
    {
      "epoch": 1.3622283935546876e-05,
      "model_forward_time": 1.7512023448944092,
      "step": 88284
    },
    {
      "epoch": 1.3622283935546876e-05,
      "step": 88284,
      "training_step_time": 7.796483993530273
    },
    {
      "epoch": 1.3623237609863281e-05,
      "model_forward_time": 1.7439701557159424,
      "step": 88285
    },
    {
      "epoch": 1.3623237609863281e-05,
      "step": 88285,
      "training_step_time": 9.569586753845215
    },
    {
      "epoch": 1.3624191284179687e-05,
      "model_forward_time": 2.817875385284424,
      "step": 88286
    },
    {
      "epoch": 1.3624191284179687e-05,
      "step": 88286,
      "training_step_time": 8.656235933303833
    },
    {
      "epoch": 1.3625144958496095e-05,
      "model_forward_time": 1.8025321960449219,
      "step": 88287
    },
    {
      "epoch": 1.3625144958496095e-05,
      "step": 88287,
      "training_step_time": 8.582671403884888
    },
    {
      "epoch": 1.36260986328125e-05,
      "model_forward_time": 1.747575283050537,
      "step": 88288
    },
    {
      "epoch": 1.36260986328125e-05,
      "step": 88288,
      "training_step_time": 7.698502779006958
    },
    {
      "epoch": 1.3627052307128906e-05,
      "model_forward_time": 1.7450273036956787,
      "step": 88289
    },
    {
      "epoch": 1.3627052307128906e-05,
      "step": 88289,
      "training_step_time": 7.771749496459961
    },
    {
      "epoch": 1.3628005981445312e-05,
      "grad_norm": 0.08129063993692398,
      "learning_rate": 3.702931331772336e-07,
      "loss": 0.056,
      "step": 88290
    },
    {
      "epoch": 1.3628005981445312e-05,
      "model_forward_time": 1.7408366203308105,
      "step": 88290
    },
    {
      "dynamics_loss_avg": 0.06154838381335139,
      "epoch": 1.3628005981445312e-05,
      "step": 88290
    },
    {
      "action_loss_avg": 0.002406428020913154,
      "epoch": 1.3628005981445312e-05,
      "step": 88290
    },
    {
      "epoch": 1.3628005981445312e-05,
      "step": 88290,
      "training_step_time": 8.712327003479004
    },
    {
      "epoch": 1.3628959655761719e-05,
      "model_forward_time": 1.7564265727996826,
      "step": 88291
    },
    {
      "epoch": 1.3628959655761719e-05,
      "step": 88291,
      "training_step_time": 7.506528854370117
    },
    {
      "epoch": 1.3629913330078125e-05,
      "model_forward_time": 1.7472646236419678,
      "step": 88292
    },
    {
      "epoch": 1.3629913330078125e-05,
      "step": 88292,
      "training_step_time": 7.884297609329224
    },
    {
      "epoch": 1.363086700439453e-05,
      "model_forward_time": 1.7433910369873047,
      "step": 88293
    },
    {
      "epoch": 1.363086700439453e-05,
      "step": 88293,
      "training_step_time": 8.450010538101196
    },
    {
      "epoch": 1.3631820678710938e-05,
      "model_forward_time": 1.3759326934814453,
      "step": 88294
    },
    {
      "epoch": 1.3631820678710938e-05,
      "step": 88294,
      "training_step_time": 9.724934816360474
    },
    {
      "epoch": 1.3632774353027344e-05,
      "model_forward_time": 2.846184015274048,
      "step": 88295
    },
    {
      "epoch": 1.3632774353027344e-05,
      "step": 88295,
      "training_step_time": 8.48898458480835
    },
    {
      "epoch": 1.363372802734375e-05,
      "model_forward_time": 1.7892303466796875,
      "step": 88296
    },
    {
      "epoch": 1.363372802734375e-05,
      "step": 88296,
      "training_step_time": 7.709608316421509
    },
    {
      "epoch": 1.3634681701660157e-05,
      "model_forward_time": 1.7463111877441406,
      "step": 88297
    },
    {
      "epoch": 1.3634681701660157e-05,
      "step": 88297,
      "training_step_time": 7.659191131591797
    },
    {
      "epoch": 1.3635635375976563e-05,
      "model_forward_time": 1.0436515808105469,
      "step": 88298
    },
    {
      "epoch": 1.3635635375976563e-05,
      "step": 88298,
      "training_step_time": 7.654680252075195
    },
    {
      "epoch": 1.3636589050292968e-05,
      "model_forward_time": 1.740572452545166,
      "step": 88299
    },
    {
      "epoch": 1.3636589050292968e-05,
      "step": 88299,
      "training_step_time": 7.690532922744751
    },
    {
      "epoch": 1.3637542724609376e-05,
      "grad_norm": 0.08841944485902786,
      "learning_rate": 3.6966892489172787e-07,
      "loss": 0.0566,
      "step": 88300
    },
    {
      "epoch": 1.3637542724609376e-05,
      "model_forward_time": 1.7856101989746094,
      "step": 88300
    },
    {
      "dynamics_loss_avg": 0.05319279972463846,
      "epoch": 1.3637542724609376e-05,
      "step": 88300
    },
    {
      "action_loss_avg": 0.0009099934948608279,
      "epoch": 1.3637542724609376e-05,
      "step": 88300
    },
    {
      "epoch": 1.3637542724609376e-05,
      "step": 88300,
      "training_step_time": 7.515009641647339
    },
    {
      "epoch": 1.3638496398925781e-05,
      "model_forward_time": 1.7635388374328613,
      "step": 88301
    },
    {
      "epoch": 1.3638496398925781e-05,
      "step": 88301,
      "training_step_time": 7.854352951049805
    },
    {
      "epoch": 1.3639450073242187e-05,
      "model_forward_time": 1.7643263339996338,
      "step": 88302
    },
    {
      "epoch": 1.3639450073242187e-05,
      "step": 88302,
      "training_step_time": 8.66459608078003
    },
    {
      "epoch": 1.3640403747558595e-05,
      "model_forward_time": 1.7404799461364746,
      "step": 88303
    },
    {
      "epoch": 1.3640403747558595e-05,
      "step": 88303,
      "training_step_time": 7.806364297866821
    },
    {
      "epoch": 1.3641357421875e-05,
      "model_forward_time": 1.7610464096069336,
      "step": 88304
    },
    {
      "epoch": 1.3641357421875e-05,
      "step": 88304,
      "training_step_time": 8.942787170410156
    },
    {
      "epoch": 1.3642311096191406e-05,
      "model_forward_time": 2.4614980220794678,
      "step": 88305
    },
    {
      "epoch": 1.3642311096191406e-05,
      "step": 88305,
      "training_step_time": 7.746152639389038
    },
    {
      "epoch": 1.3643264770507812e-05,
      "model_forward_time": 1.8006207942962646,
      "step": 88306
    },
    {
      "epoch": 1.3643264770507812e-05,
      "step": 88306,
      "training_step_time": 7.714782238006592
    },
    {
      "epoch": 1.3644218444824219e-05,
      "model_forward_time": 1.7590112686157227,
      "step": 88307
    },
    {
      "epoch": 1.3644218444824219e-05,
      "step": 88307,
      "training_step_time": 7.683882713317871
    },
    {
      "epoch": 1.3645172119140625e-05,
      "model_forward_time": 1.753664493560791,
      "step": 88308
    },
    {
      "epoch": 1.3645172119140625e-05,
      "step": 88308,
      "training_step_time": 8.587719678878784
    },
    {
      "epoch": 1.364612579345703e-05,
      "model_forward_time": 1.7738499641418457,
      "step": 88309
    },
    {
      "epoch": 1.364612579345703e-05,
      "step": 88309,
      "training_step_time": 7.5462892055511475
    },
    {
      "epoch": 1.3647079467773438e-05,
      "grad_norm": 0.10596897453069687,
      "learning_rate": 3.6904522297226573e-07,
      "loss": 0.056,
      "step": 88310
    },
    {
      "epoch": 1.3647079467773438e-05,
      "model_forward_time": 1.747924566268921,
      "step": 88310
    },
    {
      "dynamics_loss_avg": 0.0729274194687605,
      "epoch": 1.3647079467773438e-05,
      "step": 88310
    },
    {
      "action_loss_avg": 0.0012042314046993852,
      "epoch": 1.3647079467773438e-05,
      "step": 88310
    },
    {
      "epoch": 1.3647079467773438e-05,
      "step": 88310,
      "training_step_time": 9.696933269500732
    },
    {
      "epoch": 1.3648033142089844e-05,
      "model_forward_time": 1.7660465240478516,
      "step": 88311
    },
    {
      "epoch": 1.3648033142089844e-05,
      "step": 88311,
      "training_step_time": 8.751077651977539
    },
    {
      "epoch": 1.364898681640625e-05,
      "model_forward_time": 1.7555274963378906,
      "step": 88312
    },
    {
      "epoch": 1.364898681640625e-05,
      "step": 88312,
      "training_step_time": 7.80229115486145
    },
    {
      "epoch": 1.3649940490722657e-05,
      "model_forward_time": 1.7562713623046875,
      "step": 88313
    },
    {
      "epoch": 1.3649940490722657e-05,
      "step": 88313,
      "training_step_time": 9.13623309135437
    },
    {
      "epoch": 1.3650894165039063e-05,
      "model_forward_time": 2.4040677547454834,
      "step": 88314
    },
    {
      "epoch": 1.3650894165039063e-05,
      "step": 88314,
      "training_step_time": 8.520222425460815
    },
    {
      "epoch": 1.3651847839355468e-05,
      "model_forward_time": 1.7537128925323486,
      "step": 88315
    },
    {
      "epoch": 1.3651847839355468e-05,
      "step": 88315,
      "training_step_time": 7.66960597038269
    },
    {
      "epoch": 1.3652801513671876e-05,
      "model_forward_time": 1.7694628238677979,
      "step": 88316
    },
    {
      "epoch": 1.3652801513671876e-05,
      "step": 88316,
      "training_step_time": 7.717773199081421
    },
    {
      "epoch": 1.3653755187988281e-05,
      "model_forward_time": 1.7586429119110107,
      "step": 88317
    },
    {
      "epoch": 1.3653755187988281e-05,
      "step": 88317,
      "training_step_time": 6.947699069976807
    },
    {
      "epoch": 1.3654708862304687e-05,
      "model_forward_time": 1.7698593139648438,
      "step": 88318
    },
    {
      "epoch": 1.3654708862304687e-05,
      "step": 88318,
      "training_step_time": 7.670422554016113
    },
    {
      "epoch": 1.3655662536621095e-05,
      "model_forward_time": 1.7579894065856934,
      "step": 88319
    },
    {
      "epoch": 1.3655662536621095e-05,
      "step": 88319,
      "training_step_time": 8.818053722381592
    },
    {
      "epoch": 1.36566162109375e-05,
      "grad_norm": 0.14889416098594666,
      "learning_rate": 3.6842202748705645e-07,
      "loss": 0.0579,
      "step": 88320
    },
    {
      "epoch": 1.36566162109375e-05,
      "model_forward_time": 1.7566275596618652,
      "step": 88320
    },
    {
      "dynamics_loss_avg": 0.06450377330183983,
      "epoch": 1.36566162109375e-05,
      "step": 88320
    },
    {
      "action_loss_avg": 0.0022594626003410666,
      "epoch": 1.36566162109375e-05,
      "step": 88320
    },
    {
      "epoch": 1.36566162109375e-05,
      "step": 88320,
      "training_step_time": 8.550607442855835
    },
    {
      "epoch": 1.3657569885253906e-05,
      "model_forward_time": 1.765127420425415,
      "step": 88321
    },
    {
      "epoch": 1.3657569885253906e-05,
      "step": 88321,
      "training_step_time": 8.583182096481323
    },
    {
      "epoch": 1.3658523559570312e-05,
      "model_forward_time": 1.771838665008545,
      "step": 88322
    },
    {
      "epoch": 1.3658523559570312e-05,
      "step": 88322,
      "training_step_time": 7.967009544372559
    },
    {
      "epoch": 1.3659477233886719e-05,
      "model_forward_time": 1.758265495300293,
      "step": 88323
    },
    {
      "epoch": 1.3659477233886719e-05,
      "step": 88323,
      "training_step_time": 9.402729511260986
    },
    {
      "epoch": 1.3660430908203125e-05,
      "model_forward_time": 1.7647547721862793,
      "step": 88324
    },
    {
      "epoch": 1.3660430908203125e-05,
      "step": 88324,
      "training_step_time": 8.359395027160645
    },
    {
      "epoch": 1.366138458251953e-05,
      "model_forward_time": 1.828193187713623,
      "step": 88325
    },
    {
      "epoch": 1.366138458251953e-05,
      "step": 88325,
      "training_step_time": 7.74531626701355
    },
    {
      "epoch": 1.3662338256835938e-05,
      "model_forward_time": 1.7655200958251953,
      "step": 88326
    },
    {
      "epoch": 1.3662338256835938e-05,
      "step": 88326,
      "training_step_time": 7.237810134887695
    },
    {
      "epoch": 1.3663291931152344e-05,
      "model_forward_time": 1.7609021663665771,
      "step": 88327
    },
    {
      "epoch": 1.3663291931152344e-05,
      "step": 88327,
      "training_step_time": 8.229532718658447
    },
    {
      "epoch": 1.366424560546875e-05,
      "model_forward_time": 1.7599148750305176,
      "step": 88328
    },
    {
      "epoch": 1.366424560546875e-05,
      "step": 88328,
      "training_step_time": 7.617122650146484
    },
    {
      "epoch": 1.3665199279785157e-05,
      "model_forward_time": 1.767657995223999,
      "step": 88329
    },
    {
      "epoch": 1.3665199279785157e-05,
      "step": 88329,
      "training_step_time": 7.86431360244751
    },
    {
      "epoch": 1.3666152954101563e-05,
      "grad_norm": 0.0765586644411087,
      "learning_rate": 3.6779933850424944e-07,
      "loss": 0.0555,
      "step": 88330
    },
    {
      "epoch": 1.3666152954101563e-05,
      "model_forward_time": 1.754126787185669,
      "step": 88330
    },
    {
      "dynamics_loss_avg": 0.056799104996025564,
      "epoch": 1.3666152954101563e-05,
      "step": 88330
    },
    {
      "action_loss_avg": 0.0032095131173264235,
      "epoch": 1.3666152954101563e-05,
      "step": 88330
    },
    {
      "epoch": 1.3666152954101563e-05,
      "step": 88330,
      "training_step_time": 7.734870672225952
    },
    {
      "epoch": 1.3667106628417968e-05,
      "model_forward_time": 1.774090051651001,
      "step": 88331
    },
    {
      "epoch": 1.3667106628417968e-05,
      "step": 88331,
      "training_step_time": 8.922959327697754
    },
    {
      "epoch": 1.3668060302734376e-05,
      "model_forward_time": 1.7585456371307373,
      "step": 88332
    },
    {
      "epoch": 1.3668060302734376e-05,
      "step": 88332,
      "training_step_time": 10.041137456893921
    },
    {
      "epoch": 1.3669013977050781e-05,
      "model_forward_time": 2.3865673542022705,
      "step": 88333
    },
    {
      "epoch": 1.3669013977050781e-05,
      "step": 88333,
      "training_step_time": 7.745946168899536
    },
    {
      "epoch": 1.3669967651367187e-05,
      "model_forward_time": 1.8313336372375488,
      "step": 88334
    },
    {
      "epoch": 1.3669967651367187e-05,
      "step": 88334,
      "training_step_time": 7.712317943572998
    },
    {
      "epoch": 1.3670921325683595e-05,
      "model_forward_time": 1.7615442276000977,
      "step": 88335
    },
    {
      "epoch": 1.3670921325683595e-05,
      "step": 88335,
      "training_step_time": 7.821141242980957
    },
    {
      "epoch": 1.3671875e-05,
      "model_forward_time": 1.762437343597412,
      "step": 88336
    },
    {
      "epoch": 1.3671875e-05,
      "step": 88336,
      "training_step_time": 7.650764226913452
    },
    {
      "epoch": 1.3672828674316406e-05,
      "model_forward_time": 1.7612197399139404,
      "step": 88337
    },
    {
      "epoch": 1.3672828674316406e-05,
      "step": 88337,
      "training_step_time": 7.735388278961182
    },
    {
      "epoch": 1.3673782348632812e-05,
      "model_forward_time": 1.759988784790039,
      "step": 88338
    },
    {
      "epoch": 1.3673782348632812e-05,
      "step": 88338,
      "training_step_time": 8.33139157295227
    },
    {
      "epoch": 1.3674736022949219e-05,
      "model_forward_time": 1.7624099254608154,
      "step": 88339
    },
    {
      "epoch": 1.3674736022949219e-05,
      "step": 88339,
      "training_step_time": 8.787816286087036
    },
    {
      "epoch": 1.3675689697265625e-05,
      "grad_norm": 0.08213174343109131,
      "learning_rate": 3.6717715609194404e-07,
      "loss": 0.0569,
      "step": 88340
    },
    {
      "epoch": 1.3675689697265625e-05,
      "model_forward_time": 1.7577109336853027,
      "step": 88340
    },
    {
      "dynamics_loss_avg": 0.04746732152998447,
      "epoch": 1.3675689697265625e-05,
      "step": 88340
    },
    {
      "action_loss_avg": 0.0012474257440771908,
      "epoch": 1.3675689697265625e-05,
      "step": 88340
    },
    {
      "epoch": 1.3675689697265625e-05,
      "step": 88340,
      "training_step_time": 9.323344707489014
    },
    {
      "epoch": 1.367664337158203e-05,
      "model_forward_time": 1.757802963256836,
      "step": 88341
    },
    {
      "epoch": 1.367664337158203e-05,
      "step": 88341,
      "training_step_time": 8.460394859313965
    },
    {
      "epoch": 1.3677597045898438e-05,
      "model_forward_time": 2.386370897293091,
      "step": 88342
    },
    {
      "epoch": 1.3677597045898438e-05,
      "step": 88342,
      "training_step_time": 9.242017984390259
    },
    {
      "epoch": 1.3678550720214844e-05,
      "model_forward_time": 1.8303852081298828,
      "step": 88343
    },
    {
      "epoch": 1.3678550720214844e-05,
      "step": 88343,
      "training_step_time": 7.788799524307251
    },
    {
      "epoch": 1.367950439453125e-05,
      "model_forward_time": 1.7684226036071777,
      "step": 88344
    },
    {
      "epoch": 1.367950439453125e-05,
      "step": 88344,
      "training_step_time": 7.746695518493652
    },
    {
      "epoch": 1.3680458068847657e-05,
      "model_forward_time": 1.7909958362579346,
      "step": 88345
    },
    {
      "epoch": 1.3680458068847657e-05,
      "step": 88345,
      "training_step_time": 7.718406438827515
    },
    {
      "epoch": 1.3681411743164063e-05,
      "model_forward_time": 1.7566914558410645,
      "step": 88346
    },
    {
      "epoch": 1.3681411743164063e-05,
      "step": 88346,
      "training_step_time": 7.824254274368286
    },
    {
      "epoch": 1.3682365417480468e-05,
      "model_forward_time": 1.7701680660247803,
      "step": 88347
    },
    {
      "epoch": 1.3682365417480468e-05,
      "step": 88347,
      "training_step_time": 8.552905559539795
    },
    {
      "epoch": 1.3683319091796876e-05,
      "model_forward_time": 1.7606925964355469,
      "step": 88348
    },
    {
      "epoch": 1.3683319091796876e-05,
      "step": 88348,
      "training_step_time": 7.971887111663818
    },
    {
      "epoch": 1.3684272766113281e-05,
      "model_forward_time": 1.7573246955871582,
      "step": 88349
    },
    {
      "epoch": 1.3684272766113281e-05,
      "step": 88349,
      "training_step_time": 8.839824199676514
    },
    {
      "epoch": 1.3685226440429687e-05,
      "grad_norm": 0.11461184918880463,
      "learning_rate": 3.6655548031817865e-07,
      "loss": 0.057,
      "step": 88350
    },
    {
      "epoch": 1.3685226440429687e-05,
      "model_forward_time": 1.738584280014038,
      "step": 88350
    },
    {
      "dynamics_loss_avg": 0.05335235111415386,
      "epoch": 1.3685226440429687e-05,
      "step": 88350
    },
    {
      "action_loss_avg": 0.0013705887831747532,
      "epoch": 1.3685226440429687e-05,
      "step": 88350
    },
    {
      "epoch": 1.3685226440429687e-05,
      "step": 88350,
      "training_step_time": 8.95139455795288
    },
    {
      "epoch": 1.3686180114746095e-05,
      "model_forward_time": 1.9746432304382324,
      "step": 88351
    },
    {
      "epoch": 1.3686180114746095e-05,
      "step": 88351,
      "training_step_time": 8.660621881484985
    },
    {
      "epoch": 1.36871337890625e-05,
      "model_forward_time": 1.759896993637085,
      "step": 88352
    },
    {
      "epoch": 1.36871337890625e-05,
      "step": 88352,
      "training_step_time": 7.760405540466309
    },
    {
      "epoch": 1.3688087463378906e-05,
      "model_forward_time": 1.7476892471313477,
      "step": 88353
    },
    {
      "epoch": 1.3688087463378906e-05,
      "step": 88353,
      "training_step_time": 7.804797410964966
    },
    {
      "epoch": 1.3689041137695312e-05,
      "model_forward_time": 1.7412691116333008,
      "step": 88354
    },
    {
      "epoch": 1.3689041137695312e-05,
      "step": 88354,
      "training_step_time": 8.632111310958862
    },
    {
      "epoch": 1.368999481201172e-05,
      "model_forward_time": 1.746495246887207,
      "step": 88355
    },
    {
      "epoch": 1.368999481201172e-05,
      "step": 88355,
      "training_step_time": 7.875366449356079
    },
    {
      "epoch": 1.3690948486328125e-05,
      "model_forward_time": 1.750335931777954,
      "step": 88356
    },
    {
      "epoch": 1.3690948486328125e-05,
      "step": 88356,
      "training_step_time": 7.958786487579346
    },
    {
      "epoch": 1.369190216064453e-05,
      "model_forward_time": 1.382101058959961,
      "step": 88357
    },
    {
      "epoch": 1.369190216064453e-05,
      "step": 88357,
      "training_step_time": 9.092679738998413
    },
    {
      "epoch": 1.3692855834960938e-05,
      "model_forward_time": 1.7767047882080078,
      "step": 88358
    },
    {
      "epoch": 1.3692855834960938e-05,
      "step": 88358,
      "training_step_time": 8.09541940689087
    },
    {
      "epoch": 1.3693809509277344e-05,
      "model_forward_time": 1.7475414276123047,
      "step": 88359
    },
    {
      "epoch": 1.3693809509277344e-05,
      "step": 88359,
      "training_step_time": 10.480826377868652
    },
    {
      "epoch": 1.369476318359375e-05,
      "grad_norm": 0.08063487708568573,
      "learning_rate": 3.6593431125094004e-07,
      "loss": 0.0579,
      "step": 88360
    },
    {
      "epoch": 1.369476318359375e-05,
      "model_forward_time": 1.9804444313049316,
      "step": 88360
    },
    {
      "dynamics_loss_avg": 0.06579291187226773,
      "epoch": 1.369476318359375e-05,
      "step": 88360
    },
    {
      "action_loss_avg": 0.002693702405667864,
      "epoch": 1.369476318359375e-05,
      "step": 88360
    },
    {
      "epoch": 1.369476318359375e-05,
      "step": 88360,
      "training_step_time": 8.452305555343628
    },
    {
      "epoch": 1.3695716857910157e-05,
      "model_forward_time": 1.7452776432037354,
      "step": 88361
    },
    {
      "epoch": 1.3695716857910157e-05,
      "step": 88361,
      "training_step_time": 7.781217813491821
    },
    {
      "epoch": 1.3696670532226563e-05,
      "model_forward_time": 1.4647772312164307,
      "step": 88362
    },
    {
      "epoch": 1.3696670532226563e-05,
      "step": 88362,
      "training_step_time": 7.945472717285156
    },
    {
      "epoch": 1.3697624206542968e-05,
      "model_forward_time": 1.7428827285766602,
      "step": 88363
    },
    {
      "epoch": 1.3697624206542968e-05,
      "step": 88363,
      "training_step_time": 7.612300395965576
    },
    {
      "epoch": 1.3698577880859376e-05,
      "model_forward_time": 1.7429249286651611,
      "step": 88364
    },
    {
      "epoch": 1.3698577880859376e-05,
      "step": 88364,
      "training_step_time": 7.812200307846069
    },
    {
      "epoch": 1.3699531555175781e-05,
      "model_forward_time": 1.7548573017120361,
      "step": 88365
    },
    {
      "epoch": 1.3699531555175781e-05,
      "step": 88365,
      "training_step_time": 8.88071084022522
    },
    {
      "epoch": 1.3700485229492187e-05,
      "model_forward_time": 1.3814175128936768,
      "step": 88366
    },
    {
      "epoch": 1.3700485229492187e-05,
      "step": 88366,
      "training_step_time": 7.951821327209473
    },
    {
      "epoch": 1.3701438903808595e-05,
      "model_forward_time": 1.744715929031372,
      "step": 88367
    },
    {
      "epoch": 1.3701438903808595e-05,
      "step": 88367,
      "training_step_time": 10.088941812515259
    },
    {
      "epoch": 1.3702392578125e-05,
      "model_forward_time": 1.7461745738983154,
      "step": 88368
    },
    {
      "epoch": 1.3702392578125e-05,
      "step": 88368,
      "training_step_time": 7.982587814331055
    },
    {
      "epoch": 1.3703346252441406e-05,
      "model_forward_time": 2.3528003692626953,
      "step": 88369
    },
    {
      "epoch": 1.3703346252441406e-05,
      "step": 88369,
      "training_step_time": 7.843136787414551
    },
    {
      "epoch": 1.3704299926757812e-05,
      "grad_norm": 0.08757264912128448,
      "learning_rate": 3.6531364895815767e-07,
      "loss": 0.0573,
      "step": 88370
    },
    {
      "epoch": 1.3704299926757812e-05,
      "model_forward_time": 1.746164083480835,
      "step": 88370
    },
    {
      "dynamics_loss_avg": 0.07569979596883059,
      "epoch": 1.3704299926757812e-05,
      "step": 88370
    },
    {
      "action_loss_avg": 0.001782534667290747,
      "epoch": 1.3704299926757812e-05,
      "step": 88370
    },
    {
      "epoch": 1.3704299926757812e-05,
      "step": 88370,
      "training_step_time": 7.757424592971802
    },
    {
      "epoch": 1.370525360107422e-05,
      "model_forward_time": 1.8074846267700195,
      "step": 88371
    },
    {
      "epoch": 1.370525360107422e-05,
      "step": 88371,
      "training_step_time": 8.66442084312439
    },
    {
      "epoch": 1.3706207275390625e-05,
      "model_forward_time": 1.7456727027893066,
      "step": 88372
    },
    {
      "epoch": 1.3706207275390625e-05,
      "step": 88372,
      "training_step_time": 7.338624715805054
    },
    {
      "epoch": 1.370716094970703e-05,
      "model_forward_time": 1.746337652206421,
      "step": 88373
    },
    {
      "epoch": 1.370716094970703e-05,
      "step": 88373,
      "training_step_time": 7.712893009185791
    },
    {
      "epoch": 1.3708114624023438e-05,
      "model_forward_time": 1.7443761825561523,
      "step": 88374
    },
    {
      "epoch": 1.3708114624023438e-05,
      "step": 88374,
      "training_step_time": 7.74634861946106
    },
    {
      "epoch": 1.3709068298339844e-05,
      "model_forward_time": 1.749091386795044,
      "step": 88375
    },
    {
      "epoch": 1.3709068298339844e-05,
      "step": 88375,
      "training_step_time": 7.849661350250244
    },
    {
      "epoch": 1.371002197265625e-05,
      "model_forward_time": 1.7464022636413574,
      "step": 88376
    },
    {
      "epoch": 1.371002197265625e-05,
      "step": 88376,
      "training_step_time": 7.80238938331604
    },
    {
      "epoch": 1.3710975646972657e-05,
      "model_forward_time": 1.7471923828125,
      "step": 88377
    },
    {
      "epoch": 1.3710975646972657e-05,
      "step": 88377,
      "training_step_time": 7.9609761238098145
    },
    {
      "epoch": 1.3711929321289063e-05,
      "model_forward_time": 1.745851755142212,
      "step": 88378
    },
    {
      "epoch": 1.3711929321289063e-05,
      "step": 88378,
      "training_step_time": 9.370493173599243
    },
    {
      "epoch": 1.3712882995605468e-05,
      "model_forward_time": 2.394315719604492,
      "step": 88379
    },
    {
      "epoch": 1.3712882995605468e-05,
      "step": 88379,
      "training_step_time": 7.590125322341919
    },
    {
      "epoch": 1.3713836669921876e-05,
      "grad_norm": 0.12077131122350693,
      "learning_rate": 3.6469349350770677e-07,
      "loss": 0.057,
      "step": 88380
    },
    {
      "epoch": 1.3713836669921876e-05,
      "model_forward_time": 1.8015484809875488,
      "step": 88380
    },
    {
      "dynamics_loss_avg": 0.04889711663126946,
      "epoch": 1.3713836669921876e-05,
      "step": 88380
    },
    {
      "action_loss_avg": 0.0013636701973155142,
      "epoch": 1.3713836669921876e-05,
      "step": 88380
    },
    {
      "epoch": 1.3713836669921876e-05,
      "step": 88380,
      "training_step_time": 7.713694334030151
    },
    {
      "epoch": 1.3714790344238281e-05,
      "model_forward_time": 1.7676584720611572,
      "step": 88381
    },
    {
      "epoch": 1.3714790344238281e-05,
      "step": 88381,
      "training_step_time": 7.684642314910889
    },
    {
      "epoch": 1.3715744018554687e-05,
      "model_forward_time": 1.7696194648742676,
      "step": 88382
    },
    {
      "epoch": 1.3715744018554687e-05,
      "step": 88382,
      "training_step_time": 7.746418237686157
    },
    {
      "epoch": 1.3716697692871093e-05,
      "model_forward_time": 1.7710628509521484,
      "step": 88383
    },
    {
      "epoch": 1.3716697692871093e-05,
      "step": 88383,
      "training_step_time": 8.311464786529541
    },
    {
      "epoch": 1.37176513671875e-05,
      "model_forward_time": 1.7707188129425049,
      "step": 88384
    },
    {
      "epoch": 1.37176513671875e-05,
      "step": 88384,
      "training_step_time": 7.8208112716674805
    },
    {
      "epoch": 1.3718605041503906e-05,
      "model_forward_time": 1.7625575065612793,
      "step": 88385
    },
    {
      "epoch": 1.3718605041503906e-05,
      "step": 88385,
      "training_step_time": 7.777788162231445
    },
    {
      "epoch": 1.3719558715820312e-05,
      "model_forward_time": 1.7687504291534424,
      "step": 88386
    },
    {
      "epoch": 1.3719558715820312e-05,
      "step": 88386,
      "training_step_time": 8.895787000656128
    },
    {
      "epoch": 1.372051239013672e-05,
      "model_forward_time": 1.7664759159088135,
      "step": 88387
    },
    {
      "epoch": 1.372051239013672e-05,
      "step": 88387,
      "training_step_time": 7.785812854766846
    },
    {
      "epoch": 1.3721466064453125e-05,
      "model_forward_time": 2.398162841796875,
      "step": 88388
    },
    {
      "epoch": 1.3721466064453125e-05,
      "step": 88388,
      "training_step_time": 9.455939292907715
    },
    {
      "epoch": 1.372241973876953e-05,
      "model_forward_time": 1.7818169593811035,
      "step": 88389
    },
    {
      "epoch": 1.372241973876953e-05,
      "step": 88389,
      "training_step_time": 9.54322862625122
    },
    {
      "epoch": 1.3723373413085938e-05,
      "grad_norm": 0.08856078237295151,
      "learning_rate": 3.640738449674053e-07,
      "loss": 0.0581,
      "step": 88390
    },
    {
      "epoch": 1.3723373413085938e-05,
      "model_forward_time": 1.7509088516235352,
      "step": 88390
    },
    {
      "dynamics_loss_avg": 0.07588072828948497,
      "epoch": 1.3723373413085938e-05,
      "step": 88390
    },
    {
      "action_loss_avg": 0.00274494070908986,
      "epoch": 1.3723373413085938e-05,
      "step": 88390
    },
    {
      "epoch": 1.3723373413085938e-05,
      "step": 88390,
      "training_step_time": 7.6816112995147705
    },
    {
      "epoch": 1.3724327087402344e-05,
      "model_forward_time": 1.766420602798462,
      "step": 88391
    },
    {
      "epoch": 1.3724327087402344e-05,
      "step": 88391,
      "training_step_time": 7.689703464508057
    },
    {
      "epoch": 1.372528076171875e-05,
      "model_forward_time": 1.7468993663787842,
      "step": 88392
    },
    {
      "epoch": 1.372528076171875e-05,
      "step": 88392,
      "training_step_time": 7.773505449295044
    },
    {
      "epoch": 1.3726234436035157e-05,
      "model_forward_time": 1.7503604888916016,
      "step": 88393
    },
    {
      "epoch": 1.3726234436035157e-05,
      "step": 88393,
      "training_step_time": 7.757596969604492
    },
    {
      "epoch": 1.3727188110351563e-05,
      "model_forward_time": 1.746506690979004,
      "step": 88394
    },
    {
      "epoch": 1.3727188110351563e-05,
      "step": 88394,
      "training_step_time": 8.918245077133179
    },
    {
      "epoch": 1.3728141784667968e-05,
      "model_forward_time": 1.7497801780700684,
      "step": 88395
    },
    {
      "epoch": 1.3728141784667968e-05,
      "step": 88395,
      "training_step_time": 8.799322128295898
    },
    {
      "epoch": 1.3729095458984376e-05,
      "model_forward_time": 1.7584455013275146,
      "step": 88396
    },
    {
      "epoch": 1.3729095458984376e-05,
      "step": 88396,
      "training_step_time": 8.686485528945923
    },
    {
      "epoch": 1.3730049133300781e-05,
      "model_forward_time": 2.3565852642059326,
      "step": 88397
    },
    {
      "epoch": 1.3730049133300781e-05,
      "step": 88397,
      "training_step_time": 7.790153741836548
    },
    {
      "epoch": 1.3731002807617187e-05,
      "model_forward_time": 1.7426974773406982,
      "step": 88398
    },
    {
      "epoch": 1.3731002807617187e-05,
      "step": 88398,
      "training_step_time": 7.778560161590576
    },
    {
      "epoch": 1.3731956481933593e-05,
      "model_forward_time": 1.7584261894226074,
      "step": 88399
    },
    {
      "epoch": 1.3731956481933593e-05,
      "step": 88399,
      "training_step_time": 7.631677150726318
    },
    {
      "epoch": 1.373291015625e-05,
      "grad_norm": 0.1363878697156906,
      "learning_rate": 3.6345470340501955e-07,
      "loss": 0.0572,
      "step": 88400
    },
    {
      "epoch": 1.373291015625e-05,
      "model_forward_time": 1.74849534034729,
      "step": 88400
    },
    {
      "dynamics_loss_avg": 0.0686068594455719,
      "epoch": 1.373291015625e-05,
      "step": 88400
    },
    {
      "action_loss_avg": 0.001740582485217601,
      "epoch": 1.373291015625e-05,
      "step": 88400
    },
    {
      "epoch": 1.373291015625e-05,
      "step": 88400,
      "training_step_time": 7.691793441772461
    },
    {
      "epoch": 1.3733863830566406e-05,
      "model_forward_time": 1.7512233257293701,
      "step": 88401
    },
    {
      "epoch": 1.3733863830566406e-05,
      "step": 88401,
      "training_step_time": 7.73852276802063
    },
    {
      "epoch": 1.3734817504882812e-05,
      "model_forward_time": 1.7466416358947754,
      "step": 88402
    },
    {
      "epoch": 1.3734817504882812e-05,
      "step": 88402,
      "training_step_time": 8.275939226150513
    },
    {
      "epoch": 1.373577117919922e-05,
      "model_forward_time": 1.7539777755737305,
      "step": 88403
    },
    {
      "epoch": 1.373577117919922e-05,
      "step": 88403,
      "training_step_time": 8.732499361038208
    },
    {
      "epoch": 1.3736724853515625e-05,
      "model_forward_time": 1.3923625946044922,
      "step": 88404
    },
    {
      "epoch": 1.3736724853515625e-05,
      "step": 88404,
      "training_step_time": 7.7898547649383545
    },
    {
      "epoch": 1.373767852783203e-05,
      "model_forward_time": 1.7547683715820312,
      "step": 88405
    },
    {
      "epoch": 1.373767852783203e-05,
      "step": 88405,
      "training_step_time": 7.922640800476074
    },
    {
      "epoch": 1.3738632202148438e-05,
      "model_forward_time": 2.3956291675567627,
      "step": 88406
    },
    {
      "epoch": 1.3738632202148438e-05,
      "step": 88406,
      "training_step_time": 7.828331470489502
    },
    {
      "epoch": 1.3739585876464844e-05,
      "model_forward_time": 1.748469352722168,
      "step": 88407
    },
    {
      "epoch": 1.3739585876464844e-05,
      "step": 88407,
      "training_step_time": 8.625834226608276
    },
    {
      "epoch": 1.374053955078125e-05,
      "model_forward_time": 1.8190281391143799,
      "step": 88408
    },
    {
      "epoch": 1.374053955078125e-05,
      "step": 88408,
      "training_step_time": 7.79542875289917
    },
    {
      "epoch": 1.3741493225097657e-05,
      "model_forward_time": 1.7554619312286377,
      "step": 88409
    },
    {
      "epoch": 1.3741493225097657e-05,
      "step": 88409,
      "training_step_time": 7.6893250942230225
    },
    {
      "epoch": 1.3742446899414063e-05,
      "grad_norm": 0.07960859686136246,
      "learning_rate": 3.628360688882554e-07,
      "loss": 0.0551,
      "step": 88410
    },
    {
      "epoch": 1.3742446899414063e-05,
      "model_forward_time": 1.7388794422149658,
      "step": 88410
    },
    {
      "dynamics_loss_avg": 0.06024878062307835,
      "epoch": 1.3742446899414063e-05,
      "step": 88410
    },
    {
      "action_loss_avg": 0.0015280306979548186,
      "epoch": 1.3742446899414063e-05,
      "step": 88410
    },
    {
      "epoch": 1.3742446899414063e-05,
      "step": 88410,
      "training_step_time": 8.154165506362915
    },
    {
      "epoch": 1.3743400573730468e-05,
      "model_forward_time": 1.7510225772857666,
      "step": 88411
    },
    {
      "epoch": 1.3743400573730468e-05,
      "step": 88411,
      "training_step_time": 7.784358263015747
    },
    {
      "epoch": 1.3744354248046876e-05,
      "model_forward_time": 1.7440402507781982,
      "step": 88412
    },
    {
      "epoch": 1.3744354248046876e-05,
      "step": 88412,
      "training_step_time": 9.433161497116089
    },
    {
      "epoch": 1.3745307922363281e-05,
      "model_forward_time": 1.7647311687469482,
      "step": 88413
    },
    {
      "epoch": 1.3745307922363281e-05,
      "step": 88413,
      "training_step_time": 8.340266227722168
    },
    {
      "epoch": 1.3746261596679687e-05,
      "model_forward_time": 1.7423174381256104,
      "step": 88414
    },
    {
      "epoch": 1.3746261596679687e-05,
      "step": 88414,
      "training_step_time": 8.976452589035034
    },
    {
      "epoch": 1.3747215270996093e-05,
      "model_forward_time": 1.748033046722412,
      "step": 88415
    },
    {
      "epoch": 1.3747215270996093e-05,
      "step": 88415,
      "training_step_time": 8.390016317367554
    },
    {
      "epoch": 1.37481689453125e-05,
      "model_forward_time": 1.7412407398223877,
      "step": 88416
    },
    {
      "epoch": 1.37481689453125e-05,
      "step": 88416,
      "training_step_time": 8.976670742034912
    },
    {
      "epoch": 1.3749122619628906e-05,
      "model_forward_time": 1.8143079280853271,
      "step": 88417
    },
    {
      "epoch": 1.3749122619628906e-05,
      "step": 88417,
      "training_step_time": 7.822004318237305
    },
    {
      "epoch": 1.3750076293945312e-05,
      "model_forward_time": 1.7590696811676025,
      "step": 88418
    },
    {
      "epoch": 1.3750076293945312e-05,
      "step": 88418,
      "training_step_time": 7.754420757293701
    },
    {
      "epoch": 1.375102996826172e-05,
      "model_forward_time": 1.7446520328521729,
      "step": 88419
    },
    {
      "epoch": 1.375102996826172e-05,
      "step": 88419,
      "training_step_time": 8.277225732803345
    },
    {
      "epoch": 1.3751983642578125e-05,
      "grad_norm": 0.07681296765804291,
      "learning_rate": 3.622179414847676e-07,
      "loss": 0.0572,
      "step": 88420
    },
    {
      "epoch": 1.3751983642578125e-05,
      "model_forward_time": 1.738480806350708,
      "step": 88420
    },
    {
      "dynamics_loss_avg": 0.04970635361969471,
      "epoch": 1.3751983642578125e-05,
      "step": 88420
    },
    {
      "action_loss_avg": 0.0016506795451277866,
      "epoch": 1.3751983642578125e-05,
      "step": 88420
    },
    {
      "epoch": 1.3751983642578125e-05,
      "step": 88420,
      "training_step_time": 7.864544868469238
    },
    {
      "epoch": 1.375293731689453e-05,
      "model_forward_time": 1.7648215293884277,
      "step": 88421
    },
    {
      "epoch": 1.375293731689453e-05,
      "step": 88421,
      "training_step_time": 8.421991348266602
    },
    {
      "epoch": 1.3753890991210938e-05,
      "model_forward_time": 1.7572605609893799,
      "step": 88422
    },
    {
      "epoch": 1.3753890991210938e-05,
      "step": 88422,
      "training_step_time": 9.371442556381226
    },
    {
      "epoch": 1.3754844665527344e-05,
      "model_forward_time": 1.7390367984771729,
      "step": 88423
    },
    {
      "epoch": 1.3754844665527344e-05,
      "step": 88423,
      "training_step_time": 9.878240823745728
    },
    {
      "epoch": 1.375579833984375e-05,
      "model_forward_time": 1.7573521137237549,
      "step": 88424
    },
    {
      "epoch": 1.375579833984375e-05,
      "step": 88424,
      "training_step_time": 9.06210708618164
    },
    {
      "epoch": 1.3756752014160157e-05,
      "model_forward_time": 1.6660566329956055,
      "step": 88425
    },
    {
      "epoch": 1.3756752014160157e-05,
      "step": 88425,
      "training_step_time": 7.54665732383728
    },
    {
      "epoch": 1.3757705688476563e-05,
      "model_forward_time": 1.8128392696380615,
      "step": 88426
    },
    {
      "epoch": 1.3757705688476563e-05,
      "step": 88426,
      "training_step_time": 7.7996485233306885
    },
    {
      "epoch": 1.3758659362792968e-05,
      "model_forward_time": 1.7580077648162842,
      "step": 88427
    },
    {
      "epoch": 1.3758659362792968e-05,
      "step": 88427,
      "training_step_time": 7.677165985107422
    },
    {
      "epoch": 1.3759613037109376e-05,
      "model_forward_time": 1.7574501037597656,
      "step": 88428
    },
    {
      "epoch": 1.3759613037109376e-05,
      "step": 88428,
      "training_step_time": 7.680307865142822
    },
    {
      "epoch": 1.3760566711425782e-05,
      "model_forward_time": 1.7532541751861572,
      "step": 88429
    },
    {
      "epoch": 1.3760566711425782e-05,
      "step": 88429,
      "training_step_time": 7.915448904037476
    },
    {
      "epoch": 1.3761520385742187e-05,
      "grad_norm": 0.08314904570579529,
      "learning_rate": 3.6160032126215274e-07,
      "loss": 0.0581,
      "step": 88430
    },
    {
      "epoch": 1.3761520385742187e-05,
      "model_forward_time": 1.7414491176605225,
      "step": 88430
    },
    {
      "dynamics_loss_avg": 0.06815171763300895,
      "epoch": 1.3761520385742187e-05,
      "step": 88430
    },
    {
      "action_loss_avg": 0.0018821961130015552,
      "epoch": 1.3761520385742187e-05,
      "step": 88430
    },
    {
      "epoch": 1.3761520385742187e-05,
      "step": 88430,
      "training_step_time": 8.701835870742798
    },
    {
      "epoch": 1.3762474060058593e-05,
      "model_forward_time": 1.754899501800537,
      "step": 88431
    },
    {
      "epoch": 1.3762474060058593e-05,
      "step": 88431,
      "training_step_time": 8.676642894744873
    },
    {
      "epoch": 1.3763427734375e-05,
      "model_forward_time": 1.7497262954711914,
      "step": 88432
    },
    {
      "epoch": 1.3763427734375e-05,
      "step": 88432,
      "training_step_time": 8.570280313491821
    },
    {
      "epoch": 1.3764381408691406e-05,
      "model_forward_time": 1.7471439838409424,
      "step": 88433
    },
    {
      "epoch": 1.3764381408691406e-05,
      "step": 88433,
      "training_step_time": 8.810057878494263
    },
    {
      "epoch": 1.3765335083007812e-05,
      "model_forward_time": 2.409158706665039,
      "step": 88434
    },
    {
      "epoch": 1.3765335083007812e-05,
      "step": 88434,
      "training_step_time": 8.3242027759552
    },
    {
      "epoch": 1.376628875732422e-05,
      "model_forward_time": 1.807265281677246,
      "step": 88435
    },
    {
      "epoch": 1.376628875732422e-05,
      "step": 88435,
      "training_step_time": 9.22627067565918
    },
    {
      "epoch": 1.3767242431640625e-05,
      "model_forward_time": 1.7552165985107422,
      "step": 88436
    },
    {
      "epoch": 1.3767242431640625e-05,
      "step": 88436,
      "training_step_time": 7.650945425033569
    },
    {
      "epoch": 1.376819610595703e-05,
      "model_forward_time": 1.7501401901245117,
      "step": 88437
    },
    {
      "epoch": 1.376819610595703e-05,
      "step": 88437,
      "training_step_time": 7.664708137512207
    },
    {
      "epoch": 1.3769149780273438e-05,
      "model_forward_time": 1.7563786506652832,
      "step": 88438
    },
    {
      "epoch": 1.3769149780273438e-05,
      "step": 88438,
      "training_step_time": 7.856356620788574
    },
    {
      "epoch": 1.3770103454589844e-05,
      "model_forward_time": 1.7598295211791992,
      "step": 88439
    },
    {
      "epoch": 1.3770103454589844e-05,
      "step": 88439,
      "training_step_time": 8.836587190628052
    },
    {
      "epoch": 1.377105712890625e-05,
      "grad_norm": 0.08341778069734573,
      "learning_rate": 3.609832082879533e-07,
      "loss": 0.0578,
      "step": 88440
    },
    {
      "epoch": 1.377105712890625e-05,
      "model_forward_time": 1.750758171081543,
      "step": 88440
    },
    {
      "dynamics_loss_avg": 0.06467995997518301,
      "epoch": 1.377105712890625e-05,
      "step": 88440
    },
    {
      "action_loss_avg": 0.002399149409029633,
      "epoch": 1.377105712890625e-05,
      "step": 88440
    },
    {
      "epoch": 1.377105712890625e-05,
      "step": 88440,
      "training_step_time": 9.689602851867676
    },
    {
      "epoch": 1.3772010803222657e-05,
      "model_forward_time": 1.7704699039459229,
      "step": 88441
    },
    {
      "epoch": 1.3772010803222657e-05,
      "step": 88441,
      "training_step_time": 9.597471237182617
    },
    {
      "epoch": 1.3772964477539063e-05,
      "model_forward_time": 1.9100592136383057,
      "step": 88442
    },
    {
      "epoch": 1.3772964477539063e-05,
      "step": 88442,
      "training_step_time": 9.350371599197388
    },
    {
      "epoch": 1.3773918151855468e-05,
      "model_forward_time": 1.517427682876587,
      "step": 88443
    },
    {
      "epoch": 1.3773918151855468e-05,
      "step": 88443,
      "training_step_time": 8.356988430023193
    },
    {
      "epoch": 1.3774871826171876e-05,
      "model_forward_time": 1.9009222984313965,
      "step": 88444
    },
    {
      "epoch": 1.3774871826171876e-05,
      "step": 88444,
      "training_step_time": 8.422782897949219
    },
    {
      "epoch": 1.3775825500488282e-05,
      "model_forward_time": 1.9087164402008057,
      "step": 88445
    },
    {
      "epoch": 1.3775825500488282e-05,
      "step": 88445,
      "training_step_time": 7.783605337142944
    },
    {
      "epoch": 1.3776779174804687e-05,
      "model_forward_time": 1.912156581878662,
      "step": 88446
    },
    {
      "epoch": 1.3776779174804687e-05,
      "step": 88446,
      "training_step_time": 7.711736679077148
    },
    {
      "epoch": 1.3777732849121093e-05,
      "model_forward_time": 1.9006049633026123,
      "step": 88447
    },
    {
      "epoch": 1.3777732849121093e-05,
      "step": 88447,
      "training_step_time": 7.5480241775512695
    },
    {
      "epoch": 1.37786865234375e-05,
      "model_forward_time": 1.9031379222869873,
      "step": 88448
    },
    {
      "epoch": 1.37786865234375e-05,
      "step": 88448,
      "training_step_time": 7.819363117218018
    },
    {
      "epoch": 1.3779640197753906e-05,
      "model_forward_time": 1.9123120307922363,
      "step": 88449
    },
    {
      "epoch": 1.3779640197753906e-05,
      "step": 88449,
      "training_step_time": 8.536325216293335
    },
    {
      "epoch": 1.3780593872070312e-05,
      "grad_norm": 0.12632204592227936,
      "learning_rate": 3.6036660262965486e-07,
      "loss": 0.0577,
      "step": 88450
    },
    {
      "epoch": 1.3780593872070312e-05,
      "model_forward_time": 1.9037940502166748,
      "step": 88450
    },
    {
      "dynamics_loss_avg": 0.06846850216388703,
      "epoch": 1.3780593872070312e-05,
      "step": 88450
    },
    {
      "action_loss_avg": 0.001861410099081695,
      "epoch": 1.3780593872070312e-05,
      "step": 88450
    },
    {
      "epoch": 1.3780593872070312e-05,
      "step": 88450,
      "training_step_time": 8.699815034866333
    },
    {
      "epoch": 1.378154754638672e-05,
      "model_forward_time": 1.9107048511505127,
      "step": 88451
    },
    {
      "epoch": 1.378154754638672e-05,
      "step": 88451,
      "training_step_time": 9.356184005737305
    },
    {
      "epoch": 1.3782501220703125e-05,
      "model_forward_time": 1.9202535152435303,
      "step": 88452
    },
    {
      "epoch": 1.3782501220703125e-05,
      "step": 88452,
      "training_step_time": 7.773560285568237
    },
    {
      "epoch": 1.378345489501953e-05,
      "model_forward_time": 2.5250964164733887,
      "step": 88453
    },
    {
      "epoch": 1.378345489501953e-05,
      "step": 88453,
      "training_step_time": 7.6295530796051025
    },
    {
      "epoch": 1.3784408569335938e-05,
      "model_forward_time": 1.8910267353057861,
      "step": 88454
    },
    {
      "epoch": 1.3784408569335938e-05,
      "step": 88454,
      "training_step_time": 8.744717597961426
    },
    {
      "epoch": 1.3785362243652344e-05,
      "model_forward_time": 1.9011991024017334,
      "step": 88455
    },
    {
      "epoch": 1.3785362243652344e-05,
      "step": 88455,
      "training_step_time": 8.534186124801636
    },
    {
      "epoch": 1.378631591796875e-05,
      "model_forward_time": 1.908691644668579,
      "step": 88456
    },
    {
      "epoch": 1.378631591796875e-05,
      "step": 88456,
      "training_step_time": 7.5772926807403564
    },
    {
      "epoch": 1.3787269592285157e-05,
      "model_forward_time": 1.9039597511291504,
      "step": 88457
    },
    {
      "epoch": 1.3787269592285157e-05,
      "step": 88457,
      "training_step_time": 8.350018978118896
    },
    {
      "epoch": 1.3788223266601563e-05,
      "model_forward_time": 1.9006803035736084,
      "step": 88458
    },
    {
      "epoch": 1.3788223266601563e-05,
      "step": 88458,
      "training_step_time": 8.695318222045898
    },
    {
      "epoch": 1.3789176940917968e-05,
      "model_forward_time": 1.9088575839996338,
      "step": 88459
    },
    {
      "epoch": 1.3789176940917968e-05,
      "step": 88459,
      "training_step_time": 7.732552528381348
    },
    {
      "epoch": 1.3790130615234376e-05,
      "grad_norm": 0.06756329536437988,
      "learning_rate": 3.597505043546895e-07,
      "loss": 0.0572,
      "step": 88460
    },
    {
      "epoch": 1.3790130615234376e-05,
      "model_forward_time": 1.8967969417572021,
      "step": 88460
    },
    {
      "dynamics_loss_avg": 0.06465526334941388,
      "epoch": 1.3790130615234376e-05,
      "step": 88460
    },
    {
      "action_loss_avg": 0.0016093676968012005,
      "epoch": 1.3790130615234376e-05,
      "step": 88460
    },
    {
      "epoch": 1.3790130615234376e-05,
      "step": 88460,
      "training_step_time": 7.792707443237305
    },
    {
      "epoch": 1.3791084289550782e-05,
      "model_forward_time": 1.9050297737121582,
      "step": 88461
    },
    {
      "epoch": 1.3791084289550782e-05,
      "step": 88461,
      "training_step_time": 8.405213832855225
    },
    {
      "epoch": 1.3792037963867187e-05,
      "model_forward_time": 2.510610342025757,
      "step": 88462
    },
    {
      "epoch": 1.3792037963867187e-05,
      "step": 88462,
      "training_step_time": 7.796749114990234
    },
    {
      "epoch": 1.3792991638183593e-05,
      "model_forward_time": 1.8929002285003662,
      "step": 88463
    },
    {
      "epoch": 1.3792991638183593e-05,
      "step": 88463,
      "training_step_time": 8.538861513137817
    },
    {
      "epoch": 1.37939453125e-05,
      "model_forward_time": 1.8744113445281982,
      "step": 88464
    },
    {
      "epoch": 1.37939453125e-05,
      "step": 88464,
      "training_step_time": 7.753741502761841
    },
    {
      "epoch": 1.3794898986816406e-05,
      "model_forward_time": 1.8737123012542725,
      "step": 88465
    },
    {
      "epoch": 1.3794898986816406e-05,
      "step": 88465,
      "training_step_time": 7.54514479637146
    },
    {
      "epoch": 1.3795852661132812e-05,
      "model_forward_time": 1.9103059768676758,
      "step": 88466
    },
    {
      "epoch": 1.3795852661132812e-05,
      "step": 88466,
      "training_step_time": 7.929076910018921
    },
    {
      "epoch": 1.379680633544922e-05,
      "model_forward_time": 1.896240472793579,
      "step": 88467
    },
    {
      "epoch": 1.379680633544922e-05,
      "step": 88467,
      "training_step_time": 8.827266454696655
    },
    {
      "epoch": 1.3797760009765625e-05,
      "model_forward_time": 1.9044897556304932,
      "step": 88468
    },
    {
      "epoch": 1.3797760009765625e-05,
      "step": 88468,
      "training_step_time": 7.753685235977173
    },
    {
      "epoch": 1.379871368408203e-05,
      "model_forward_time": 1.9077484607696533,
      "step": 88469
    },
    {
      "epoch": 1.379871368408203e-05,
      "step": 88469,
      "training_step_time": 7.773051500320435
    },
    {
      "epoch": 1.3799667358398438e-05,
      "grad_norm": 0.20215001702308655,
      "learning_rate": 3.591349135304323e-07,
      "loss": 0.0561,
      "step": 88470
    },
    {
      "epoch": 1.3799667358398438e-05,
      "model_forward_time": 1.9019460678100586,
      "step": 88470
    },
    {
      "dynamics_loss_avg": 0.07070228382945061,
      "epoch": 1.3799667358398438e-05,
      "step": 88470
    },
    {
      "action_loss_avg": 0.002189507125876844,
      "epoch": 1.3799667358398438e-05,
      "step": 88470
    },
    {
      "epoch": 1.3799667358398438e-05,
      "step": 88470,
      "training_step_time": 7.778227806091309
    },
    {
      "epoch": 1.3800621032714844e-05,
      "model_forward_time": 2.4886317253112793,
      "step": 88471
    },
    {
      "epoch": 1.3800621032714844e-05,
      "step": 88471,
      "training_step_time": 7.791846990585327
    },
    {
      "epoch": 1.380157470703125e-05,
      "model_forward_time": 1.8938426971435547,
      "step": 88472
    },
    {
      "epoch": 1.380157470703125e-05,
      "step": 88472,
      "training_step_time": 7.866318225860596
    },
    {
      "epoch": 1.3802528381347657e-05,
      "model_forward_time": 1.5288875102996826,
      "step": 88473
    },
    {
      "epoch": 1.3802528381347657e-05,
      "step": 88473,
      "training_step_time": 8.558084964752197
    },
    {
      "epoch": 1.3803482055664063e-05,
      "model_forward_time": 1.9092340469360352,
      "step": 88474
    },
    {
      "epoch": 1.3803482055664063e-05,
      "step": 88474,
      "training_step_time": 7.6234235763549805
    },
    {
      "epoch": 1.3804435729980468e-05,
      "model_forward_time": 1.9021117687225342,
      "step": 88475
    },
    {
      "epoch": 1.3804435729980468e-05,
      "step": 88475,
      "training_step_time": 8.926233530044556
    },
    {
      "epoch": 1.3805389404296876e-05,
      "model_forward_time": 1.8999888896942139,
      "step": 88476
    },
    {
      "epoch": 1.3805389404296876e-05,
      "step": 88476,
      "training_step_time": 8.881518125534058
    },
    {
      "epoch": 1.3806343078613282e-05,
      "model_forward_time": 1.9177265167236328,
      "step": 88477
    },
    {
      "epoch": 1.3806343078613282e-05,
      "step": 88477,
      "training_step_time": 7.863829612731934
    },
    {
      "epoch": 1.3807296752929687e-05,
      "model_forward_time": 1.9024419784545898,
      "step": 88478
    },
    {
      "epoch": 1.3807296752929687e-05,
      "step": 88478,
      "training_step_time": 9.6650550365448
    },
    {
      "epoch": 1.3808250427246093e-05,
      "model_forward_time": 1.8872113227844238,
      "step": 88479
    },
    {
      "epoch": 1.3808250427246093e-05,
      "step": 88479,
      "training_step_time": 9.634980916976929
    },
    {
      "epoch": 1.38092041015625e-05,
      "grad_norm": 0.06731635332107544,
      "learning_rate": 3.585198302242038e-07,
      "loss": 0.0565,
      "step": 88480
    },
    {
      "epoch": 1.38092041015625e-05,
      "model_forward_time": 2.5362298488616943,
      "step": 88480
    },
    {
      "dynamics_loss_avg": 0.04992442186921835,
      "epoch": 1.38092041015625e-05,
      "step": 88480
    },
    {
      "action_loss_avg": 0.0016232756432145834,
      "epoch": 1.38092041015625e-05,
      "step": 88480
    },
    {
      "epoch": 1.38092041015625e-05,
      "step": 88480,
      "training_step_time": 7.747178554534912
    },
    {
      "epoch": 1.3810157775878906e-05,
      "model_forward_time": 1.9093852043151855,
      "step": 88481
    },
    {
      "epoch": 1.3810157775878906e-05,
      "step": 88481,
      "training_step_time": 8.58665680885315
    },
    {
      "epoch": 1.3811111450195312e-05,
      "model_forward_time": 1.8937962055206299,
      "step": 88482
    },
    {
      "epoch": 1.3811111450195312e-05,
      "step": 88482,
      "training_step_time": 7.808437824249268
    },
    {
      "epoch": 1.381206512451172e-05,
      "model_forward_time": 1.8906149864196777,
      "step": 88483
    },
    {
      "epoch": 1.381206512451172e-05,
      "step": 88483,
      "training_step_time": 7.740686655044556
    },
    {
      "epoch": 1.3813018798828125e-05,
      "model_forward_time": 1.91300368309021,
      "step": 88484
    },
    {
      "epoch": 1.3813018798828125e-05,
      "step": 88484,
      "training_step_time": 7.753008842468262
    },
    {
      "epoch": 1.381397247314453e-05,
      "model_forward_time": 1.173417568206787,
      "step": 88485
    },
    {
      "epoch": 1.381397247314453e-05,
      "step": 88485,
      "training_step_time": 7.805187463760376
    },
    {
      "epoch": 1.3814926147460938e-05,
      "model_forward_time": 1.8883230686187744,
      "step": 88486
    },
    {
      "epoch": 1.3814926147460938e-05,
      "step": 88486,
      "training_step_time": 8.585144758224487
    },
    {
      "epoch": 1.3815879821777344e-05,
      "model_forward_time": 1.8840360641479492,
      "step": 88487
    },
    {
      "epoch": 1.3815879821777344e-05,
      "step": 88487,
      "training_step_time": 7.888867616653442
    },
    {
      "epoch": 1.381683349609375e-05,
      "model_forward_time": 1.901458740234375,
      "step": 88488
    },
    {
      "epoch": 1.381683349609375e-05,
      "step": 88488,
      "training_step_time": 8.789170026779175
    },
    {
      "epoch": 1.3817787170410157e-05,
      "model_forward_time": 1.9027085304260254,
      "step": 88489
    },
    {
      "epoch": 1.3817787170410157e-05,
      "step": 88489,
      "training_step_time": 7.745193243026733
    },
    {
      "epoch": 1.3818740844726563e-05,
      "grad_norm": 0.07464545965194702,
      "learning_rate": 3.5790525450326796e-07,
      "loss": 0.0558,
      "step": 88490
    },
    {
      "epoch": 1.3818740844726563e-05,
      "model_forward_time": 2.5035791397094727,
      "step": 88490
    },
    {
      "dynamics_loss_avg": 0.05063045285642147,
      "epoch": 1.3818740844726563e-05,
      "step": 88490
    },
    {
      "action_loss_avg": 0.001584322887356393,
      "epoch": 1.3818740844726563e-05,
      "step": 88490
    },
    {
      "epoch": 1.3818740844726563e-05,
      "step": 88490,
      "training_step_time": 7.557661771774292
    },
    {
      "epoch": 1.3819694519042968e-05,
      "model_forward_time": 1.8907811641693115,
      "step": 88491
    },
    {
      "epoch": 1.3819694519042968e-05,
      "step": 88491,
      "training_step_time": 8.780373573303223
    },
    {
      "epoch": 1.3820648193359376e-05,
      "model_forward_time": 1.896519422531128,
      "step": 88492
    },
    {
      "epoch": 1.3820648193359376e-05,
      "step": 88492,
      "training_step_time": 7.744016885757446
    },
    {
      "epoch": 1.3821601867675782e-05,
      "model_forward_time": 1.8894908428192139,
      "step": 88493
    },
    {
      "epoch": 1.3821601867675782e-05,
      "step": 88493,
      "training_step_time": 7.736837148666382
    },
    {
      "epoch": 1.3822555541992187e-05,
      "model_forward_time": 1.9023878574371338,
      "step": 88494
    },
    {
      "epoch": 1.3822555541992187e-05,
      "step": 88494,
      "training_step_time": 7.826577663421631
    },
    {
      "epoch": 1.3823509216308593e-05,
      "model_forward_time": 1.898418664932251,
      "step": 88495
    },
    {
      "epoch": 1.3823509216308593e-05,
      "step": 88495,
      "training_step_time": 7.867824554443359
    },
    {
      "epoch": 1.3824462890625e-05,
      "model_forward_time": 1.9051513671875,
      "step": 88496
    },
    {
      "epoch": 1.3824462890625e-05,
      "step": 88496,
      "training_step_time": 7.9531638622283936
    },
    {
      "epoch": 1.3825416564941406e-05,
      "model_forward_time": 1.9094727039337158,
      "step": 88497
    },
    {
      "epoch": 1.3825416564941406e-05,
      "step": 88497,
      "training_step_time": 9.672630310058594
    },
    {
      "epoch": 1.3826370239257812e-05,
      "model_forward_time": 1.9064204692840576,
      "step": 88498
    },
    {
      "epoch": 1.3826370239257812e-05,
      "step": 88498,
      "training_step_time": 7.709853887557983
    },
    {
      "epoch": 1.382732391357422e-05,
      "model_forward_time": 2.4931211471557617,
      "step": 88499
    },
    {
      "epoch": 1.382732391357422e-05,
      "step": 88499,
      "training_step_time": 8.250661849975586
    },
    {
      "epoch": 1.3828277587890625e-05,
      "grad_norm": 0.0988178700208664,
      "learning_rate": 3.5729118643483443e-07,
      "loss": 0.0559,
      "step": 88500
    },
    {
      "epoch": 1.3828277587890625e-05,
      "model_forward_time": 1.8866448402404785,
      "step": 88500
    },
    {
      "dynamics_loss_avg": 0.0779528982937336,
      "epoch": 1.3828277587890625e-05,
      "step": 88500
    },
    {
      "action_loss_avg": 0.0029159643920138477,
      "epoch": 1.3828277587890625e-05,
      "step": 88500
    },
    {
      "epoch": 1.3828277587890625e-05,
      "step": 88500,
      "training_step_time": 7.702613353729248
    },
    {
      "epoch": 1.382923126220703e-05,
      "model_forward_time": 1.900486946105957,
      "step": 88501
    },
    {
      "epoch": 1.382923126220703e-05,
      "step": 88501,
      "training_step_time": 7.692890405654907
    },
    {
      "epoch": 1.3830184936523438e-05,
      "model_forward_time": 1.875976800918579,
      "step": 88502
    },
    {
      "epoch": 1.3830184936523438e-05,
      "step": 88502,
      "training_step_time": 8.742747068405151
    },
    {
      "epoch": 1.3831138610839844e-05,
      "model_forward_time": 1.5350549221038818,
      "step": 88503
    },
    {
      "epoch": 1.3831138610839844e-05,
      "step": 88503,
      "training_step_time": 7.775594234466553
    },
    {
      "epoch": 1.383209228515625e-05,
      "model_forward_time": 1.9039356708526611,
      "step": 88504
    },
    {
      "epoch": 1.383209228515625e-05,
      "step": 88504,
      "training_step_time": 7.764825344085693
    },
    {
      "epoch": 1.3833045959472657e-05,
      "model_forward_time": 1.9041707515716553,
      "step": 88505
    },
    {
      "epoch": 1.3833045959472657e-05,
      "step": 88505,
      "training_step_time": 9.72582721710205
    },
    {
      "epoch": 1.3833999633789063e-05,
      "model_forward_time": 1.9198133945465088,
      "step": 88506
    },
    {
      "epoch": 1.3833999633789063e-05,
      "step": 88506,
      "training_step_time": 9.781399965286255
    },
    {
      "epoch": 1.3834953308105468e-05,
      "model_forward_time": 1.9130427837371826,
      "step": 88507
    },
    {
      "epoch": 1.3834953308105468e-05,
      "step": 88507,
      "training_step_time": 8.54990267753601
    },
    {
      "epoch": 1.3835906982421874e-05,
      "model_forward_time": 2.549708127975464,
      "step": 88508
    },
    {
      "epoch": 1.3835906982421874e-05,
      "step": 88508,
      "training_step_time": 7.494380474090576
    },
    {
      "epoch": 1.3836860656738282e-05,
      "model_forward_time": 1.892266035079956,
      "step": 88509
    },
    {
      "epoch": 1.3836860656738282e-05,
      "step": 88509,
      "training_step_time": 8.488211154937744
    },
    {
      "epoch": 1.3837814331054687e-05,
      "grad_norm": 0.32920441031455994,
      "learning_rate": 3.566776260860572e-07,
      "loss": 0.0571,
      "step": 88510
    },
    {
      "epoch": 1.3837814331054687e-05,
      "model_forward_time": 1.908088207244873,
      "step": 88510
    },
    {
      "dynamics_loss_avg": 0.049956911616027354,
      "epoch": 1.3837814331054687e-05,
      "step": 88510
    },
    {
      "action_loss_avg": 0.0017442894459236413,
      "epoch": 1.3837814331054687e-05,
      "step": 88510
    },
    {
      "epoch": 1.3837814331054687e-05,
      "step": 88510,
      "training_step_time": 8.280105352401733
    },
    {
      "epoch": 1.3838768005371093e-05,
      "model_forward_time": 1.9106671810150146,
      "step": 88511
    },
    {
      "epoch": 1.3838768005371093e-05,
      "step": 88511,
      "training_step_time": 7.711501121520996
    },
    {
      "epoch": 1.38397216796875e-05,
      "model_forward_time": 1.9208323955535889,
      "step": 88512
    },
    {
      "epoch": 1.38397216796875e-05,
      "step": 88512,
      "training_step_time": 7.692332029342651
    },
    {
      "epoch": 1.3840675354003906e-05,
      "model_forward_time": 1.9011931419372559,
      "step": 88513
    },
    {
      "epoch": 1.3840675354003906e-05,
      "step": 88513,
      "training_step_time": 7.723526954650879
    },
    {
      "epoch": 1.3841629028320312e-05,
      "model_forward_time": 1.9029293060302734,
      "step": 88514
    },
    {
      "epoch": 1.3841629028320312e-05,
      "step": 88514,
      "training_step_time": 8.791089296340942
    },
    {
      "epoch": 1.384258270263672e-05,
      "model_forward_time": 1.8796124458312988,
      "step": 88515
    },
    {
      "epoch": 1.384258270263672e-05,
      "step": 88515,
      "training_step_time": 9.125698328018188
    },
    {
      "epoch": 1.3843536376953125e-05,
      "model_forward_time": 1.905796766281128,
      "step": 88516
    },
    {
      "epoch": 1.3843536376953125e-05,
      "step": 88516,
      "training_step_time": 7.698936223983765
    },
    {
      "epoch": 1.384449005126953e-05,
      "model_forward_time": 1.9101505279541016,
      "step": 88517
    },
    {
      "epoch": 1.384449005126953e-05,
      "step": 88517,
      "training_step_time": 8.353360652923584
    },
    {
      "epoch": 1.3845443725585938e-05,
      "model_forward_time": 1.907689094543457,
      "step": 88518
    },
    {
      "epoch": 1.3845443725585938e-05,
      "step": 88518,
      "training_step_time": 7.700451374053955
    },
    {
      "epoch": 1.3846397399902344e-05,
      "model_forward_time": 1.9021711349487305,
      "step": 88519
    },
    {
      "epoch": 1.3846397399902344e-05,
      "step": 88519,
      "training_step_time": 9.536906719207764
    },
    {
      "epoch": 1.384735107421875e-05,
      "grad_norm": 0.08343426883220673,
      "learning_rate": 3.5606457352403254e-07,
      "loss": 0.0537,
      "step": 88520
    },
    {
      "epoch": 1.384735107421875e-05,
      "model_forward_time": 1.89479398727417,
      "step": 88520
    },
    {
      "dynamics_loss_avg": 0.057874524407088755,
      "epoch": 1.384735107421875e-05,
      "step": 88520
    },
    {
      "action_loss_avg": 0.0010760056757135317,
      "epoch": 1.384735107421875e-05,
      "step": 88520
    },
    {
      "epoch": 1.384735107421875e-05,
      "step": 88520,
      "training_step_time": 7.717564105987549
    },
    {
      "epoch": 1.3848304748535157e-05,
      "model_forward_time": 1.9114484786987305,
      "step": 88521
    },
    {
      "epoch": 1.3848304748535157e-05,
      "step": 88521,
      "training_step_time": 7.64541220664978
    },
    {
      "epoch": 1.3849258422851563e-05,
      "model_forward_time": 1.909257173538208,
      "step": 88522
    },
    {
      "epoch": 1.3849258422851563e-05,
      "step": 88522,
      "training_step_time": 7.763121604919434
    },
    {
      "epoch": 1.3850212097167968e-05,
      "model_forward_time": 1.9203288555145264,
      "step": 88523
    },
    {
      "epoch": 1.3850212097167968e-05,
      "step": 88523,
      "training_step_time": 9.19119644165039
    },
    {
      "epoch": 1.3851165771484374e-05,
      "model_forward_time": 1.926534652709961,
      "step": 88524
    },
    {
      "epoch": 1.3851165771484374e-05,
      "step": 88524,
      "training_step_time": 7.883425951004028
    },
    {
      "epoch": 1.3852119445800782e-05,
      "model_forward_time": 1.9118578433990479,
      "step": 88525
    },
    {
      "epoch": 1.3852119445800782e-05,
      "step": 88525,
      "training_step_time": 8.721360683441162
    },
    {
      "epoch": 1.3853073120117187e-05,
      "model_forward_time": 1.8800594806671143,
      "step": 88526
    },
    {
      "epoch": 1.3853073120117187e-05,
      "step": 88526,
      "training_step_time": 7.632473468780518
    },
    {
      "epoch": 1.3854026794433593e-05,
      "model_forward_time": 2.510131359100342,
      "step": 88527
    },
    {
      "epoch": 1.3854026794433593e-05,
      "step": 88527,
      "training_step_time": 8.185851812362671
    },
    {
      "epoch": 1.385498046875e-05,
      "model_forward_time": 1.9185492992401123,
      "step": 88528
    },
    {
      "epoch": 1.385498046875e-05,
      "step": 88528,
      "training_step_time": 7.669635534286499
    },
    {
      "epoch": 1.3855934143066406e-05,
      "model_forward_time": 1.9264609813690186,
      "step": 88529
    },
    {
      "epoch": 1.3855934143066406e-05,
      "step": 88529,
      "training_step_time": 9.319563388824463
    },
    {
      "epoch": 1.3856887817382812e-05,
      "grad_norm": 0.08111438900232315,
      "learning_rate": 3.554520288158042e-07,
      "loss": 0.0563,
      "step": 88530
    },
    {
      "epoch": 1.3856887817382812e-05,
      "model_forward_time": 1.9079139232635498,
      "step": 88530
    },
    {
      "dynamics_loss_avg": 0.05253584990277886,
      "epoch": 1.3856887817382812e-05,
      "step": 88530
    },
    {
      "action_loss_avg": 0.0014883169715176336,
      "epoch": 1.3856887817382812e-05,
      "step": 88530
    },
    {
      "epoch": 1.3856887817382812e-05,
      "step": 88530,
      "training_step_time": 7.632761478424072
    },
    {
      "epoch": 1.385784149169922e-05,
      "model_forward_time": 1.903928279876709,
      "step": 88531
    },
    {
      "epoch": 1.385784149169922e-05,
      "step": 88531,
      "training_step_time": 8.299945831298828
    },
    {
      "epoch": 1.3858795166015625e-05,
      "model_forward_time": 1.903808355331421,
      "step": 88532
    },
    {
      "epoch": 1.3858795166015625e-05,
      "step": 88532,
      "training_step_time": 7.7333714962005615
    },
    {
      "epoch": 1.385974884033203e-05,
      "model_forward_time": 1.9107086658477783,
      "step": 88533
    },
    {
      "epoch": 1.385974884033203e-05,
      "step": 88533,
      "training_step_time": 8.654251098632812
    },
    {
      "epoch": 1.3860702514648438e-05,
      "model_forward_time": 1.9210128784179688,
      "step": 88534
    },
    {
      "epoch": 1.3860702514648438e-05,
      "step": 88534,
      "training_step_time": 8.470831394195557
    },
    {
      "epoch": 1.3861656188964844e-05,
      "model_forward_time": 1.9024841785430908,
      "step": 88535
    },
    {
      "epoch": 1.3861656188964844e-05,
      "step": 88535,
      "training_step_time": 11.622541904449463
    },
    {
      "epoch": 1.386260986328125e-05,
      "model_forward_time": 1.9047489166259766,
      "step": 88536
    },
    {
      "epoch": 1.386260986328125e-05,
      "step": 88536,
      "training_step_time": 7.717591047286987
    },
    {
      "epoch": 1.3863563537597657e-05,
      "model_forward_time": 1.9081227779388428,
      "step": 88537
    },
    {
      "epoch": 1.3863563537597657e-05,
      "step": 88537,
      "training_step_time": 8.544950723648071
    },
    {
      "epoch": 1.3864517211914063e-05,
      "model_forward_time": 1.9064552783966064,
      "step": 88538
    },
    {
      "epoch": 1.3864517211914063e-05,
      "step": 88538,
      "training_step_time": 8.517115116119385
    },
    {
      "epoch": 1.3865470886230468e-05,
      "model_forward_time": 1.9135570526123047,
      "step": 88539
    },
    {
      "epoch": 1.3865470886230468e-05,
      "step": 88539,
      "training_step_time": 7.568952798843384
    },
    {
      "epoch": 1.3866424560546874e-05,
      "grad_norm": 0.1057141125202179,
      "learning_rate": 3.548399920283591e-07,
      "loss": 0.0553,
      "step": 88540
    },
    {
      "epoch": 1.3866424560546874e-05,
      "model_forward_time": 1.9019029140472412,
      "step": 88540
    },
    {
      "dynamics_loss_avg": 0.06465741246938705,
      "epoch": 1.3866424560546874e-05,
      "step": 88540
    },
    {
      "action_loss_avg": 0.00211588560778182,
      "epoch": 1.3866424560546874e-05,
      "step": 88540
    },
    {
      "epoch": 1.3866424560546874e-05,
      "step": 88540,
      "training_step_time": 6.822182655334473
    },
    {
      "epoch": 1.3867378234863282e-05,
      "model_forward_time": 1.9127488136291504,
      "step": 88541
    },
    {
      "epoch": 1.3867378234863282e-05,
      "step": 88541,
      "training_step_time": 6.838006258010864
    },
    {
      "epoch": 1.3868331909179687e-05,
      "model_forward_time": 1.8987960815429688,
      "step": 88542
    },
    {
      "epoch": 1.3868331909179687e-05,
      "step": 88542,
      "training_step_time": 6.792536020278931
    },
    {
      "epoch": 1.3869285583496093e-05,
      "model_forward_time": 1.9035637378692627,
      "step": 88543
    },
    {
      "epoch": 1.3869285583496093e-05,
      "step": 88543,
      "training_step_time": 6.835110664367676
    },
    {
      "epoch": 1.38702392578125e-05,
      "model_forward_time": 1.9027771949768066,
      "step": 88544
    },
    {
      "epoch": 1.38702392578125e-05,
      "step": 88544,
      "training_step_time": 6.837554931640625
    },
    {
      "epoch": 1.3871192932128906e-05,
      "model_forward_time": 2.5291659832000732,
      "step": 88545
    },
    {
      "epoch": 1.3871192932128906e-05,
      "step": 88545,
      "training_step_time": 7.793382406234741
    },
    {
      "epoch": 1.3872146606445312e-05,
      "model_forward_time": 1.899073839187622,
      "step": 88546
    },
    {
      "epoch": 1.3872146606445312e-05,
      "step": 88546,
      "training_step_time": 7.458786964416504
    },
    {
      "epoch": 1.387310028076172e-05,
      "model_forward_time": 1.5464086532592773,
      "step": 88547
    },
    {
      "epoch": 1.387310028076172e-05,
      "step": 88547,
      "training_step_time": 8.498485088348389
    },
    {
      "epoch": 1.3874053955078125e-05,
      "model_forward_time": 1.9032402038574219,
      "step": 88548
    },
    {
      "epoch": 1.3874053955078125e-05,
      "step": 88548,
      "training_step_time": 7.430856466293335
    },
    {
      "epoch": 1.387500762939453e-05,
      "model_forward_time": 1.8892815113067627,
      "step": 88549
    },
    {
      "epoch": 1.387500762939453e-05,
      "step": 88549,
      "training_step_time": 6.7274861335754395
    },
    {
      "epoch": 1.3875961303710938e-05,
      "grad_norm": 0.2829737961292267,
      "learning_rate": 3.542284632286291e-07,
      "loss": 0.0559,
      "step": 88550
    },
    {
      "epoch": 1.3875961303710938e-05,
      "model_forward_time": 1.8998792171478271,
      "step": 88550
    },
    {
      "dynamics_loss_avg": 0.05786850787699223,
      "epoch": 1.3875961303710938e-05,
      "step": 88550
    },
    {
      "action_loss_avg": 0.0017131729051470756,
      "epoch": 1.3875961303710938e-05,
      "step": 88550
    },
    {
      "epoch": 1.3875961303710938e-05,
      "step": 88550,
      "training_step_time": 6.720763683319092
    },
    {
      "epoch": 1.3876914978027344e-05,
      "model_forward_time": 1.8956754207611084,
      "step": 88551
    },
    {
      "epoch": 1.3876914978027344e-05,
      "step": 88551,
      "training_step_time": 6.6986083984375
    },
    {
      "epoch": 1.387786865234375e-05,
      "model_forward_time": 1.9155521392822266,
      "step": 88552
    },
    {
      "epoch": 1.387786865234375e-05,
      "step": 88552,
      "training_step_time": 6.7067365646362305
    },
    {
      "epoch": 1.3878822326660157e-05,
      "model_forward_time": 1.8783040046691895,
      "step": 88553
    },
    {
      "epoch": 1.3878822326660157e-05,
      "step": 88553,
      "training_step_time": 6.790730237960815
    },
    {
      "epoch": 1.3879776000976563e-05,
      "model_forward_time": 2.5437819957733154,
      "step": 88554
    },
    {
      "epoch": 1.3879776000976563e-05,
      "step": 88554,
      "training_step_time": 7.824796676635742
    },
    {
      "epoch": 1.3880729675292968e-05,
      "model_forward_time": 1.8978760242462158,
      "step": 88555
    },
    {
      "epoch": 1.3880729675292968e-05,
      "step": 88555,
      "training_step_time": 7.604602098464966
    },
    {
      "epoch": 1.3881683349609374e-05,
      "model_forward_time": 1.9249305725097656,
      "step": 88556
    },
    {
      "epoch": 1.3881683349609374e-05,
      "step": 88556,
      "training_step_time": 8.827218055725098
    },
    {
      "epoch": 1.3882637023925782e-05,
      "model_forward_time": 1.8966217041015625,
      "step": 88557
    },
    {
      "epoch": 1.3882637023925782e-05,
      "step": 88557,
      "training_step_time": 7.575059175491333
    },
    {
      "epoch": 1.3883590698242187e-05,
      "model_forward_time": 1.8972282409667969,
      "step": 88558
    },
    {
      "epoch": 1.3883590698242187e-05,
      "step": 88558,
      "training_step_time": 6.772397994995117
    },
    {
      "epoch": 1.3884544372558593e-05,
      "model_forward_time": 1.9120728969573975,
      "step": 88559
    },
    {
      "epoch": 1.3884544372558593e-05,
      "step": 88559,
      "training_step_time": 6.773681640625
    },
    {
      "epoch": 1.3885498046875e-05,
      "grad_norm": 0.08620153367519379,
      "learning_rate": 3.5361744248348926e-07,
      "loss": 0.056,
      "step": 88560
    },
    {
      "epoch": 1.3885498046875e-05,
      "model_forward_time": 1.925243854522705,
      "step": 88560
    },
    {
      "dynamics_loss_avg": 0.06689400002360343,
      "epoch": 1.3885498046875e-05,
      "step": 88560
    },
    {
      "action_loss_avg": 0.0014978855208028108,
      "epoch": 1.3885498046875e-05,
      "step": 88560
    },
    {
      "epoch": 1.3885498046875e-05,
      "step": 88560,
      "training_step_time": 6.794233798980713
    },
    {
      "epoch": 1.3886451721191406e-05,
      "model_forward_time": 1.9088397026062012,
      "step": 88561
    },
    {
      "epoch": 1.3886451721191406e-05,
      "step": 88561,
      "training_step_time": 6.773804187774658
    },
    {
      "epoch": 1.3887405395507812e-05,
      "model_forward_time": 1.5424339771270752,
      "step": 88562
    },
    {
      "epoch": 1.3887405395507812e-05,
      "step": 88562,
      "training_step_time": 6.774566650390625
    },
    {
      "epoch": 1.388835906982422e-05,
      "model_forward_time": 2.1159374713897705,
      "step": 88563
    },
    {
      "epoch": 1.388835906982422e-05,
      "step": 88563,
      "training_step_time": 8.840075016021729
    },
    {
      "epoch": 1.3889312744140625e-05,
      "model_forward_time": 1.909799337387085,
      "step": 88564
    },
    {
      "epoch": 1.3889312744140625e-05,
      "step": 88564,
      "training_step_time": 7.695848703384399
    },
    {
      "epoch": 1.389026641845703e-05,
      "model_forward_time": 1.9141972064971924,
      "step": 88565
    },
    {
      "epoch": 1.389026641845703e-05,
      "step": 88565,
      "training_step_time": 7.593268156051636
    },
    {
      "epoch": 1.3891220092773438e-05,
      "model_forward_time": 1.9084038734436035,
      "step": 88566
    },
    {
      "epoch": 1.3891220092773438e-05,
      "step": 88566,
      "training_step_time": 7.682683944702148
    },
    {
      "epoch": 1.3892173767089844e-05,
      "model_forward_time": 1.9187264442443848,
      "step": 88567
    },
    {
      "epoch": 1.3892173767089844e-05,
      "step": 88567,
      "training_step_time": 7.3078086376190186
    },
    {
      "epoch": 1.389312744140625e-05,
      "model_forward_time": 1.928487777709961,
      "step": 88568
    },
    {
      "epoch": 1.389312744140625e-05,
      "step": 88568,
      "training_step_time": 6.824448823928833
    },
    {
      "epoch": 1.3894081115722657e-05,
      "model_forward_time": 1.9238321781158447,
      "step": 88569
    },
    {
      "epoch": 1.3894081115722657e-05,
      "step": 88569,
      "training_step_time": 6.811062335968018
    },
    {
      "epoch": 1.3895034790039063e-05,
      "grad_norm": 0.10606715083122253,
      "learning_rate": 3.530069298597605e-07,
      "loss": 0.0561,
      "step": 88570
    },
    {
      "epoch": 1.3895034790039063e-05,
      "model_forward_time": 1.930553913116455,
      "step": 88570
    },
    {
      "dynamics_loss_avg": 0.052262688614428045,
      "epoch": 1.3895034790039063e-05,
      "step": 88570
    },
    {
      "action_loss_avg": 0.0015153015381656587,
      "epoch": 1.3895034790039063e-05,
      "step": 88570
    },
    {
      "epoch": 1.3895034790039063e-05,
      "step": 88570,
      "training_step_time": 6.79712176322937
    },
    {
      "epoch": 1.3895988464355468e-05,
      "model_forward_time": 1.9323327541351318,
      "step": 88571
    },
    {
      "epoch": 1.3895988464355468e-05,
      "step": 88571,
      "training_step_time": 6.803269624710083
    },
    {
      "epoch": 1.3896942138671874e-05,
      "model_forward_time": 2.531829357147217,
      "step": 88572
    },
    {
      "epoch": 1.3896942138671874e-05,
      "step": 88572,
      "training_step_time": 9.856703758239746
    },
    {
      "epoch": 1.3897895812988282e-05,
      "model_forward_time": 1.9112849235534668,
      "step": 88573
    },
    {
      "epoch": 1.3897895812988282e-05,
      "step": 88573,
      "training_step_time": 7.710496187210083
    },
    {
      "epoch": 1.3898849487304687e-05,
      "model_forward_time": 1.9152371883392334,
      "step": 88574
    },
    {
      "epoch": 1.3898849487304687e-05,
      "step": 88574,
      "training_step_time": 7.609396934509277
    },
    {
      "epoch": 1.3899803161621093e-05,
      "model_forward_time": 1.1772217750549316,
      "step": 88575
    },
    {
      "epoch": 1.3899803161621093e-05,
      "step": 88575,
      "training_step_time": 7.677858114242554
    },
    {
      "epoch": 1.39007568359375e-05,
      "model_forward_time": 1.9284119606018066,
      "step": 88576
    },
    {
      "epoch": 1.39007568359375e-05,
      "step": 88576,
      "training_step_time": 7.664193391799927
    },
    {
      "epoch": 1.3901710510253906e-05,
      "model_forward_time": 1.9213857650756836,
      "step": 88577
    },
    {
      "epoch": 1.3901710510253906e-05,
      "step": 88577,
      "training_step_time": 6.791180610656738
    },
    {
      "epoch": 1.3902664184570312e-05,
      "model_forward_time": 1.9013755321502686,
      "step": 88578
    },
    {
      "epoch": 1.3902664184570312e-05,
      "step": 88578,
      "training_step_time": 6.774172782897949
    },
    {
      "epoch": 1.390361785888672e-05,
      "model_forward_time": 1.9262754917144775,
      "step": 88579
    },
    {
      "epoch": 1.390361785888672e-05,
      "step": 88579,
      "training_step_time": 6.721719741821289
    },
    {
      "epoch": 1.3904571533203125e-05,
      "grad_norm": 0.07052665203809738,
      "learning_rate": 3.52396925424206e-07,
      "loss": 0.0555,
      "step": 88580
    },
    {
      "epoch": 1.3904571533203125e-05,
      "model_forward_time": 1.917588710784912,
      "step": 88580
    },
    {
      "dynamics_loss_avg": 0.06360816415399313,
      "epoch": 1.3904571533203125e-05,
      "step": 88580
    },
    {
      "action_loss_avg": 0.0016260250180494041,
      "epoch": 1.3904571533203125e-05,
      "step": 88580
    },
    {
      "epoch": 1.3904571533203125e-05,
      "step": 88580,
      "training_step_time": 6.735106468200684
    },
    {
      "epoch": 1.390552520751953e-05,
      "model_forward_time": 1.905970573425293,
      "step": 88581
    },
    {
      "epoch": 1.390552520751953e-05,
      "step": 88581,
      "training_step_time": 10.266834735870361
    },
    {
      "epoch": 1.3906478881835938e-05,
      "model_forward_time": 2.5425872802734375,
      "step": 88582
    },
    {
      "epoch": 1.3906478881835938e-05,
      "step": 88582,
      "training_step_time": 7.629478693008423
    },
    {
      "epoch": 1.3907432556152344e-05,
      "model_forward_time": 1.910757064819336,
      "step": 88583
    },
    {
      "epoch": 1.3907432556152344e-05,
      "step": 88583,
      "training_step_time": 7.525161981582642
    },
    {
      "epoch": 1.390838623046875e-05,
      "model_forward_time": 1.5407638549804688,
      "step": 88584
    },
    {
      "epoch": 1.390838623046875e-05,
      "step": 88584,
      "training_step_time": 7.51147723197937
    },
    {
      "epoch": 1.3909339904785157e-05,
      "model_forward_time": 1.9065423011779785,
      "step": 88585
    },
    {
      "epoch": 1.3909339904785157e-05,
      "step": 88585,
      "training_step_time": 7.549142599105835
    },
    {
      "epoch": 1.3910293579101563e-05,
      "model_forward_time": 1.9002573490142822,
      "step": 88586
    },
    {
      "epoch": 1.3910293579101563e-05,
      "step": 88586,
      "training_step_time": 7.339201927185059
    },
    {
      "epoch": 1.3911247253417969e-05,
      "model_forward_time": 1.9053490161895752,
      "step": 88587
    },
    {
      "epoch": 1.3911247253417969e-05,
      "step": 88587,
      "training_step_time": 6.714616298675537
    },
    {
      "epoch": 1.3912200927734374e-05,
      "model_forward_time": 1.8963572978973389,
      "step": 88588
    },
    {
      "epoch": 1.3912200927734374e-05,
      "step": 88588,
      "training_step_time": 6.7420525550842285
    },
    {
      "epoch": 1.3913154602050782e-05,
      "model_forward_time": 1.9012043476104736,
      "step": 88589
    },
    {
      "epoch": 1.3913154602050782e-05,
      "step": 88589,
      "training_step_time": 6.7244579792022705
    },
    {
      "epoch": 1.3914108276367187e-05,
      "grad_norm": 0.07640209048986435,
      "learning_rate": 3.517874292435375e-07,
      "loss": 0.0572,
      "step": 88590
    },
    {
      "epoch": 1.3914108276367187e-05,
      "model_forward_time": 1.909564733505249,
      "step": 88590
    },
    {
      "dynamics_loss_avg": 0.06040395796298981,
      "epoch": 1.3914108276367187e-05,
      "step": 88590
    },
    {
      "action_loss_avg": 0.0015346761880209669,
      "epoch": 1.3914108276367187e-05,
      "step": 88590
    },
    {
      "epoch": 1.3914108276367187e-05,
      "step": 88590,
      "training_step_time": 9.453216552734375
    },
    {
      "epoch": 1.3915061950683593e-05,
      "model_forward_time": 2.5377357006073,
      "step": 88591
    },
    {
      "epoch": 1.3915061950683593e-05,
      "step": 88591,
      "training_step_time": 7.73011589050293
    },
    {
      "epoch": 1.3916015625e-05,
      "model_forward_time": 1.9033002853393555,
      "step": 88592
    },
    {
      "epoch": 1.3916015625e-05,
      "step": 88592,
      "training_step_time": 7.526244878768921
    },
    {
      "epoch": 1.3916969299316406e-05,
      "model_forward_time": 1.9097788333892822,
      "step": 88593
    },
    {
      "epoch": 1.3916969299316406e-05,
      "step": 88593,
      "training_step_time": 7.539170503616333
    },
    {
      "epoch": 1.3917922973632812e-05,
      "model_forward_time": 1.9043920040130615,
      "step": 88594
    },
    {
      "epoch": 1.3917922973632812e-05,
      "step": 88594,
      "training_step_time": 7.566077947616577
    },
    {
      "epoch": 1.391887664794922e-05,
      "model_forward_time": 1.911494493484497,
      "step": 88595
    },
    {
      "epoch": 1.391887664794922e-05,
      "step": 88595,
      "training_step_time": 7.218127965927124
    },
    {
      "epoch": 1.3919830322265625e-05,
      "model_forward_time": 1.9022114276885986,
      "step": 88596
    },
    {
      "epoch": 1.3919830322265625e-05,
      "step": 88596,
      "training_step_time": 6.734737396240234
    },
    {
      "epoch": 1.392078399658203e-05,
      "model_forward_time": 1.8926782608032227,
      "step": 88597
    },
    {
      "epoch": 1.392078399658203e-05,
      "step": 88597,
      "training_step_time": 6.719625473022461
    },
    {
      "epoch": 1.3921737670898438e-05,
      "model_forward_time": 1.9026012420654297,
      "step": 88598
    },
    {
      "epoch": 1.3921737670898438e-05,
      "step": 88598,
      "training_step_time": 6.686154127120972
    },
    {
      "epoch": 1.3922691345214844e-05,
      "model_forward_time": 1.9123027324676514,
      "step": 88599
    },
    {
      "epoch": 1.3922691345214844e-05,
      "step": 88599,
      "training_step_time": 9.704338312149048
    },
    {
      "epoch": 1.392364501953125e-05,
      "grad_norm": 0.09797049313783646,
      "learning_rate": 3.511784413844066e-07,
      "loss": 0.0564,
      "step": 88600
    },
    {
      "epoch": 1.392364501953125e-05,
      "model_forward_time": 2.5020291805267334,
      "step": 88600
    },
    {
      "dynamics_loss_avg": 0.04844240667298436,
      "epoch": 1.392364501953125e-05,
      "step": 88600
    },
    {
      "action_loss_avg": 0.0017057776276487857,
      "epoch": 1.392364501953125e-05,
      "step": 88600
    },
    {
      "epoch": 1.392364501953125e-05,
      "step": 88600,
      "training_step_time": 8.329968690872192
    },
    {
      "epoch": 1.3924598693847657e-05,
      "model_forward_time": 1.9061594009399414,
      "step": 88601
    },
    {
      "epoch": 1.3924598693847657e-05,
      "step": 88601,
      "training_step_time": 8.40648341178894
    },
    {
      "epoch": 1.3925552368164063e-05,
      "model_forward_time": 1.9070687294006348,
      "step": 88602
    },
    {
      "epoch": 1.3925552368164063e-05,
      "step": 88602,
      "training_step_time": 7.554367542266846
    },
    {
      "epoch": 1.3926506042480469e-05,
      "model_forward_time": 1.9132492542266846,
      "step": 88603
    },
    {
      "epoch": 1.3926506042480469e-05,
      "step": 88603,
      "training_step_time": 7.6344146728515625
    },
    {
      "epoch": 1.3927459716796874e-05,
      "model_forward_time": 1.9043426513671875,
      "step": 88604
    },
    {
      "epoch": 1.3927459716796874e-05,
      "step": 88604,
      "training_step_time": 6.716925621032715
    },
    {
      "epoch": 1.3928413391113282e-05,
      "model_forward_time": 1.908799648284912,
      "step": 88605
    },
    {
      "epoch": 1.3928413391113282e-05,
      "step": 88605,
      "training_step_time": 7.178086757659912
    },
    {
      "epoch": 1.3929367065429687e-05,
      "model_forward_time": 1.8867402076721191,
      "step": 88606
    },
    {
      "epoch": 1.3929367065429687e-05,
      "step": 88606,
      "training_step_time": 6.675552129745483
    },
    {
      "epoch": 1.3930320739746093e-05,
      "model_forward_time": 1.9023356437683105,
      "step": 88607
    },
    {
      "epoch": 1.3930320739746093e-05,
      "step": 88607,
      "training_step_time": 6.755089998245239
    },
    {
      "epoch": 1.39312744140625e-05,
      "model_forward_time": 1.5263886451721191,
      "step": 88608
    },
    {
      "epoch": 1.39312744140625e-05,
      "step": 88608,
      "training_step_time": 7.754469633102417
    },
    {
      "epoch": 1.3932228088378906e-05,
      "model_forward_time": 1.8405015468597412,
      "step": 88609
    },
    {
      "epoch": 1.3932228088378906e-05,
      "step": 88609,
      "training_step_time": 10.893617630004883
    },
    {
      "epoch": 1.3933181762695312e-05,
      "grad_norm": 0.09388547390699387,
      "learning_rate": 3.5056996191341287e-07,
      "loss": 0.0567,
      "step": 88610
    },
    {
      "epoch": 1.3933181762695312e-05,
      "model_forward_time": 2.5081276893615723,
      "step": 88610
    },
    {
      "dynamics_loss_avg": 0.05486369766294956,
      "epoch": 1.3933181762695312e-05,
      "step": 88610
    },
    {
      "action_loss_avg": 0.001995264185825363,
      "epoch": 1.3933181762695312e-05,
      "step": 88610
    },
    {
      "epoch": 1.3933181762695312e-05,
      "step": 88610,
      "training_step_time": 7.662307977676392
    },
    {
      "epoch": 1.393413543701172e-05,
      "model_forward_time": 1.901097297668457,
      "step": 88611
    },
    {
      "epoch": 1.393413543701172e-05,
      "step": 88611,
      "training_step_time": 8.582314014434814
    },
    {
      "epoch": 1.3935089111328125e-05,
      "model_forward_time": 1.9055888652801514,
      "step": 88612
    },
    {
      "epoch": 1.3935089111328125e-05,
      "step": 88612,
      "training_step_time": 8.44931411743164
    },
    {
      "epoch": 1.393604278564453e-05,
      "model_forward_time": 1.8979485034942627,
      "step": 88613
    },
    {
      "epoch": 1.393604278564453e-05,
      "step": 88613,
      "training_step_time": 6.680703639984131
    },
    {
      "epoch": 1.3936996459960938e-05,
      "model_forward_time": 1.9052073955535889,
      "step": 88614
    },
    {
      "epoch": 1.3936996459960938e-05,
      "step": 88614,
      "training_step_time": 7.224510431289673
    },
    {
      "epoch": 1.3937950134277344e-05,
      "model_forward_time": 1.914914608001709,
      "step": 88615
    },
    {
      "epoch": 1.3937950134277344e-05,
      "step": 88615,
      "training_step_time": 6.705679893493652
    },
    {
      "epoch": 1.393890380859375e-05,
      "model_forward_time": 1.8639168739318848,
      "step": 88616
    },
    {
      "epoch": 1.393890380859375e-05,
      "step": 88616,
      "training_step_time": 6.559777021408081
    },
    {
      "epoch": 1.3939857482910157e-05,
      "model_forward_time": 1.9074065685272217,
      "step": 88617
    },
    {
      "epoch": 1.3939857482910157e-05,
      "step": 88617,
      "training_step_time": 6.732553005218506
    },
    {
      "epoch": 1.3940811157226563e-05,
      "model_forward_time": 1.8972623348236084,
      "step": 88618
    },
    {
      "epoch": 1.3940811157226563e-05,
      "step": 88618,
      "training_step_time": 7.667080402374268
    },
    {
      "epoch": 1.3941764831542969e-05,
      "model_forward_time": 2.498527765274048,
      "step": 88619
    },
    {
      "epoch": 1.3941764831542969e-05,
      "step": 88619,
      "training_step_time": 7.643190145492554
    },
    {
      "epoch": 1.3942718505859374e-05,
      "grad_norm": 0.06603586673736572,
      "learning_rate": 3.4996199089709695e-07,
      "loss": 0.0572,
      "step": 88620
    },
    {
      "epoch": 1.3942718505859374e-05,
      "model_forward_time": 1.910008430480957,
      "step": 88620
    },
    {
      "dynamics_loss_avg": 0.059524689335376024,
      "epoch": 1.3942718505859374e-05,
      "step": 88620
    },
    {
      "action_loss_avg": 0.002081898896722123,
      "epoch": 1.3942718505859374e-05,
      "step": 88620
    },
    {
      "epoch": 1.3942718505859374e-05,
      "step": 88620,
      "training_step_time": 7.543506860733032
    },
    {
      "epoch": 1.3943672180175782e-05,
      "model_forward_time": 1.8741793632507324,
      "step": 88621
    },
    {
      "epoch": 1.3943672180175782e-05,
      "step": 88621,
      "training_step_time": 8.080771446228027
    },
    {
      "epoch": 1.3944625854492187e-05,
      "model_forward_time": 1.9069929122924805,
      "step": 88622
    },
    {
      "epoch": 1.3944625854492187e-05,
      "step": 88622,
      "training_step_time": 6.991305351257324
    },
    {
      "epoch": 1.3945579528808593e-05,
      "model_forward_time": 1.9061977863311768,
      "step": 88623
    },
    {
      "epoch": 1.3945579528808593e-05,
      "step": 88623,
      "training_step_time": 7.157348871231079
    },
    {
      "epoch": 1.3946533203125e-05,
      "model_forward_time": 1.9015555381774902,
      "step": 88624
    },
    {
      "epoch": 1.3946533203125e-05,
      "step": 88624,
      "training_step_time": 6.684095144271851
    },
    {
      "epoch": 1.3947486877441406e-05,
      "model_forward_time": 1.914649486541748,
      "step": 88625
    },
    {
      "epoch": 1.3947486877441406e-05,
      "step": 88625,
      "training_step_time": 6.70136570930481
    },
    {
      "epoch": 1.3948440551757812e-05,
      "model_forward_time": 1.90885329246521,
      "step": 88626
    },
    {
      "epoch": 1.3948440551757812e-05,
      "step": 88626,
      "training_step_time": 6.693013906478882
    },
    {
      "epoch": 1.394939422607422e-05,
      "model_forward_time": 1.8905348777770996,
      "step": 88627
    },
    {
      "epoch": 1.394939422607422e-05,
      "step": 88627,
      "training_step_time": 8.355157375335693
    },
    {
      "epoch": 1.3950347900390625e-05,
      "model_forward_time": 2.5303585529327393,
      "step": 88628
    },
    {
      "epoch": 1.3950347900390625e-05,
      "step": 88628,
      "training_step_time": 8.22258996963501
    },
    {
      "epoch": 1.395130157470703e-05,
      "model_forward_time": 1.5425264835357666,
      "step": 88629
    },
    {
      "epoch": 1.395130157470703e-05,
      "step": 88629,
      "training_step_time": 7.696338891983032
    },
    {
      "epoch": 1.3952255249023438e-05,
      "grad_norm": 0.10076267272233963,
      "learning_rate": 3.4935452840194726e-07,
      "loss": 0.057,
      "step": 88630
    },
    {
      "epoch": 1.3952255249023438e-05,
      "model_forward_time": 1.9059150218963623,
      "step": 88630
    },
    {
      "dynamics_loss_avg": 0.05023493506014347,
      "epoch": 1.3952255249023438e-05,
      "step": 88630
    },
    {
      "action_loss_avg": 0.0015589779053698295,
      "epoch": 1.3952255249023438e-05,
      "step": 88630
    },
    {
      "epoch": 1.3952255249023438e-05,
      "step": 88630,
      "training_step_time": 8.079962253570557
    },
    {
      "epoch": 1.3953208923339844e-05,
      "model_forward_time": 1.9097139835357666,
      "step": 88631
    },
    {
      "epoch": 1.3953208923339844e-05,
      "step": 88631,
      "training_step_time": 7.128666400909424
    },
    {
      "epoch": 1.395416259765625e-05,
      "model_forward_time": 1.9051294326782227,
      "step": 88632
    },
    {
      "epoch": 1.395416259765625e-05,
      "step": 88632,
      "training_step_time": 7.238696575164795
    },
    {
      "epoch": 1.3955116271972655e-05,
      "model_forward_time": 1.9183621406555176,
      "step": 88633
    },
    {
      "epoch": 1.3955116271972655e-05,
      "step": 88633,
      "training_step_time": 6.788152456283569
    },
    {
      "epoch": 1.3956069946289063e-05,
      "model_forward_time": 1.905289888381958,
      "step": 88634
    },
    {
      "epoch": 1.3956069946289063e-05,
      "step": 88634,
      "training_step_time": 6.848941087722778
    },
    {
      "epoch": 1.3957023620605469e-05,
      "model_forward_time": 1.906860113143921,
      "step": 88635
    },
    {
      "epoch": 1.3957023620605469e-05,
      "step": 88635,
      "training_step_time": 6.768797397613525
    },
    {
      "epoch": 1.3957977294921874e-05,
      "model_forward_time": 1.9052770137786865,
      "step": 88636
    },
    {
      "epoch": 1.3957977294921874e-05,
      "step": 88636,
      "training_step_time": 8.420559883117676
    },
    {
      "epoch": 1.3958930969238282e-05,
      "model_forward_time": 2.495079278945923,
      "step": 88637
    },
    {
      "epoch": 1.3958930969238282e-05,
      "step": 88637,
      "training_step_time": 7.810810089111328
    },
    {
      "epoch": 1.3959884643554687e-05,
      "model_forward_time": 1.8953351974487305,
      "step": 88638
    },
    {
      "epoch": 1.3959884643554687e-05,
      "step": 88638,
      "training_step_time": 8.571821689605713
    },
    {
      "epoch": 1.3960838317871093e-05,
      "model_forward_time": 1.901461124420166,
      "step": 88639
    },
    {
      "epoch": 1.3960838317871093e-05,
      "step": 88639,
      "training_step_time": 7.739352703094482
    },
    {
      "epoch": 1.39617919921875e-05,
      "grad_norm": 0.09885025769472122,
      "learning_rate": 3.48747574494393e-07,
      "loss": 0.0573,
      "step": 88640
    },
    {
      "epoch": 1.39617919921875e-05,
      "model_forward_time": 1.8915412425994873,
      "step": 88640
    },
    {
      "dynamics_loss_avg": 0.09090296551585197,
      "epoch": 1.39617919921875e-05,
      "step": 88640
    },
    {
      "action_loss_avg": 0.0024028886982705444,
      "epoch": 1.39617919921875e-05,
      "step": 88640
    },
    {
      "epoch": 1.39617919921875e-05,
      "step": 88640,
      "training_step_time": 7.776423692703247
    },
    {
      "epoch": 1.3962745666503906e-05,
      "model_forward_time": 1.905550479888916,
      "step": 88641
    },
    {
      "epoch": 1.3962745666503906e-05,
      "step": 88641,
      "training_step_time": 7.633302211761475
    },
    {
      "epoch": 1.3963699340820312e-05,
      "model_forward_time": 1.8797998428344727,
      "step": 88642
    },
    {
      "epoch": 1.3963699340820312e-05,
      "step": 88642,
      "training_step_time": 6.859053134918213
    },
    {
      "epoch": 1.396465301513672e-05,
      "model_forward_time": 1.906846523284912,
      "step": 88643
    },
    {
      "epoch": 1.396465301513672e-05,
      "step": 88643,
      "training_step_time": 6.955683708190918
    },
    {
      "epoch": 1.3965606689453125e-05,
      "model_forward_time": 1.1689040660858154,
      "step": 88644
    },
    {
      "epoch": 1.3965606689453125e-05,
      "step": 88644,
      "training_step_time": 6.934782266616821
    },
    {
      "epoch": 1.396656036376953e-05,
      "model_forward_time": 1.8976695537567139,
      "step": 88645
    },
    {
      "epoch": 1.396656036376953e-05,
      "step": 88645,
      "training_step_time": 8.878876209259033
    },
    {
      "epoch": 1.3967514038085938e-05,
      "model_forward_time": 1.8735108375549316,
      "step": 88646
    },
    {
      "epoch": 1.3967514038085938e-05,
      "step": 88646,
      "training_step_time": 11.407114744186401
    },
    {
      "epoch": 1.3968467712402344e-05,
      "model_forward_time": 2.5154783725738525,
      "step": 88647
    },
    {
      "epoch": 1.3968467712402344e-05,
      "step": 88647,
      "training_step_time": 8.416242361068726
    },
    {
      "epoch": 1.396942138671875e-05,
      "model_forward_time": 1.8909900188446045,
      "step": 88648
    },
    {
      "epoch": 1.396942138671875e-05,
      "step": 88648,
      "training_step_time": 7.751796722412109
    },
    {
      "epoch": 1.3970375061035155e-05,
      "model_forward_time": 1.9103128910064697,
      "step": 88649
    },
    {
      "epoch": 1.3970375061035155e-05,
      "step": 88649,
      "training_step_time": 7.847663879394531
    },
    {
      "epoch": 1.3971328735351563e-05,
      "grad_norm": 0.08178328722715378,
      "learning_rate": 3.481411292408116e-07,
      "loss": 0.0575,
      "step": 88650
    },
    {
      "epoch": 1.3971328735351563e-05,
      "model_forward_time": 1.9089696407318115,
      "step": 88650
    },
    {
      "dynamics_loss_avg": 0.0679162947461009,
      "epoch": 1.3971328735351563e-05,
      "step": 88650
    },
    {
      "action_loss_avg": 0.002417797953239642,
      "epoch": 1.3971328735351563e-05,
      "step": 88650
    },
    {
      "epoch": 1.3971328735351563e-05,
      "step": 88650,
      "training_step_time": 7.603236436843872
    },
    {
      "epoch": 1.3972282409667969e-05,
      "model_forward_time": 1.899686574935913,
      "step": 88651
    },
    {
      "epoch": 1.3972282409667969e-05,
      "step": 88651,
      "training_step_time": 6.8863489627838135
    },
    {
      "epoch": 1.3973236083984374e-05,
      "model_forward_time": 1.8993208408355713,
      "step": 88652
    },
    {
      "epoch": 1.3973236083984374e-05,
      "step": 88652,
      "training_step_time": 6.85372257232666
    },
    {
      "epoch": 1.3974189758300782e-05,
      "model_forward_time": 1.8975729942321777,
      "step": 88653
    },
    {
      "epoch": 1.3974189758300782e-05,
      "step": 88653,
      "training_step_time": 6.8841307163238525
    },
    {
      "epoch": 1.3975143432617187e-05,
      "model_forward_time": 1.8990478515625,
      "step": 88654
    },
    {
      "epoch": 1.3975143432617187e-05,
      "step": 88654,
      "training_step_time": 7.5479655265808105
    },
    {
      "epoch": 1.3976097106933593e-05,
      "model_forward_time": 1.9014956951141357,
      "step": 88655
    },
    {
      "epoch": 1.3976097106933593e-05,
      "step": 88655,
      "training_step_time": 11.316534519195557
    },
    {
      "epoch": 1.397705078125e-05,
      "model_forward_time": 2.557774543762207,
      "step": 88656
    },
    {
      "epoch": 1.397705078125e-05,
      "step": 88656,
      "training_step_time": 8.468036413192749
    },
    {
      "epoch": 1.3978004455566406e-05,
      "model_forward_time": 1.9008722305297852,
      "step": 88657
    },
    {
      "epoch": 1.3978004455566406e-05,
      "step": 88657,
      "training_step_time": 7.679261922836304
    },
    {
      "epoch": 1.3978958129882812e-05,
      "model_forward_time": 1.9004347324371338,
      "step": 88658
    },
    {
      "epoch": 1.3978958129882812e-05,
      "step": 88658,
      "training_step_time": 7.717104434967041
    },
    {
      "epoch": 1.397991180419922e-05,
      "model_forward_time": 1.899467945098877,
      "step": 88659
    },
    {
      "epoch": 1.397991180419922e-05,
      "step": 88659,
      "training_step_time": 7.544097185134888
    },
    {
      "epoch": 1.3980865478515625e-05,
      "grad_norm": 0.07048020511865616,
      "learning_rate": 3.475351927075221e-07,
      "loss": 0.0568,
      "step": 88660
    },
    {
      "epoch": 1.3980865478515625e-05,
      "model_forward_time": 1.9080021381378174,
      "step": 88660
    },
    {
      "dynamics_loss_avg": 0.05984117947518826,
      "epoch": 1.3980865478515625e-05,
      "step": 88660
    },
    {
      "action_loss_avg": 0.0022753765108063815,
      "epoch": 1.3980865478515625e-05,
      "step": 88660
    },
    {
      "epoch": 1.3980865478515625e-05,
      "step": 88660,
      "training_step_time": 6.798739433288574
    },
    {
      "epoch": 1.3981819152832031e-05,
      "model_forward_time": 1.8941736221313477,
      "step": 88661
    },
    {
      "epoch": 1.3981819152832031e-05,
      "step": 88661,
      "training_step_time": 6.846277713775635
    },
    {
      "epoch": 1.3982772827148438e-05,
      "model_forward_time": 1.9040679931640625,
      "step": 88662
    },
    {
      "epoch": 1.3982772827148438e-05,
      "step": 88662,
      "training_step_time": 6.855929136276245
    },
    {
      "epoch": 1.3983726501464844e-05,
      "model_forward_time": 1.9057328701019287,
      "step": 88663
    },
    {
      "epoch": 1.3983726501464844e-05,
      "step": 88663,
      "training_step_time": 7.524867057800293
    },
    {
      "epoch": 1.398468017578125e-05,
      "model_forward_time": 1.9124696254730225,
      "step": 88664
    },
    {
      "epoch": 1.398468017578125e-05,
      "step": 88664,
      "training_step_time": 10.869678735733032
    },
    {
      "epoch": 1.3985633850097655e-05,
      "model_forward_time": 2.4857239723205566,
      "step": 88665
    },
    {
      "epoch": 1.3985633850097655e-05,
      "step": 88665,
      "training_step_time": 7.815718650817871
    },
    {
      "epoch": 1.3986587524414063e-05,
      "model_forward_time": 1.890944480895996,
      "step": 88666
    },
    {
      "epoch": 1.3986587524414063e-05,
      "step": 88666,
      "training_step_time": 8.246053457260132
    },
    {
      "epoch": 1.3987541198730469e-05,
      "model_forward_time": 1.8957760334014893,
      "step": 88667
    },
    {
      "epoch": 1.3987541198730469e-05,
      "step": 88667,
      "training_step_time": 8.297213792800903
    },
    {
      "epoch": 1.3988494873046874e-05,
      "model_forward_time": 1.8998112678527832,
      "step": 88668
    },
    {
      "epoch": 1.3988494873046874e-05,
      "step": 88668,
      "training_step_time": 7.564466953277588
    },
    {
      "epoch": 1.3989448547363282e-05,
      "model_forward_time": 1.9107751846313477,
      "step": 88669
    },
    {
      "epoch": 1.3989448547363282e-05,
      "step": 88669,
      "training_step_time": 6.811747312545776
    },
    {
      "epoch": 1.3990402221679687e-05,
      "grad_norm": 1.6611040830612183,
      "learning_rate": 3.4692976496078946e-07,
      "loss": 0.0562,
      "step": 88670
    },
    {
      "epoch": 1.3990402221679687e-05,
      "model_forward_time": 1.906848669052124,
      "step": 88670
    },
    {
      "dynamics_loss_avg": 0.06643000040203333,
      "epoch": 1.3990402221679687e-05,
      "step": 88670
    },
    {
      "action_loss_avg": 0.0023787357553374023,
      "epoch": 1.3990402221679687e-05,
      "step": 88670
    },
    {
      "epoch": 1.3990402221679687e-05,
      "step": 88670,
      "training_step_time": 6.7714760303497314
    },
    {
      "epoch": 1.3991355895996093e-05,
      "model_forward_time": 1.9023282527923584,
      "step": 88671
    },
    {
      "epoch": 1.3991355895996093e-05,
      "step": 88671,
      "training_step_time": 6.7968909740448
    },
    {
      "epoch": 1.39923095703125e-05,
      "model_forward_time": 1.909442663192749,
      "step": 88672
    },
    {
      "epoch": 1.39923095703125e-05,
      "step": 88672,
      "training_step_time": 7.564900159835815
    },
    {
      "epoch": 1.3993263244628906e-05,
      "model_forward_time": 1.8954002857208252,
      "step": 88673
    },
    {
      "epoch": 1.3993263244628906e-05,
      "step": 88673,
      "training_step_time": 11.402839660644531
    },
    {
      "epoch": 1.3994216918945312e-05,
      "model_forward_time": 2.486199140548706,
      "step": 88674
    },
    {
      "epoch": 1.3994216918945312e-05,
      "step": 88674,
      "training_step_time": 9.829394102096558
    },
    {
      "epoch": 1.399517059326172e-05,
      "model_forward_time": 1.9105048179626465,
      "step": 88675
    },
    {
      "epoch": 1.399517059326172e-05,
      "step": 88675,
      "training_step_time": 7.775200843811035
    },
    {
      "epoch": 1.3996124267578125e-05,
      "model_forward_time": 1.9027197360992432,
      "step": 88676
    },
    {
      "epoch": 1.3996124267578125e-05,
      "step": 88676,
      "training_step_time": 7.737960577011108
    },
    {
      "epoch": 1.3997077941894531e-05,
      "model_forward_time": 1.9070706367492676,
      "step": 88677
    },
    {
      "epoch": 1.3997077941894531e-05,
      "step": 88677,
      "training_step_time": 8.760708093643188
    },
    {
      "epoch": 1.3998031616210938e-05,
      "model_forward_time": 1.8718855381011963,
      "step": 88678
    },
    {
      "epoch": 1.3998031616210938e-05,
      "step": 88678,
      "training_step_time": 6.889257431030273
    },
    {
      "epoch": 1.3998985290527344e-05,
      "model_forward_time": 1.8884952068328857,
      "step": 88679
    },
    {
      "epoch": 1.3998985290527344e-05,
      "step": 88679,
      "training_step_time": 6.887155055999756
    },
    {
      "epoch": 1.399993896484375e-05,
      "grad_norm": 0.08542018383741379,
      "learning_rate": 3.463248460668217e-07,
      "loss": 0.0573,
      "step": 88680
    },
    {
      "epoch": 1.399993896484375e-05,
      "model_forward_time": 1.9069101810455322,
      "step": 88680
    },
    {
      "dynamics_loss_avg": 0.05341755812987685,
      "epoch": 1.399993896484375e-05,
      "step": 88680
    },
    {
      "action_loss_avg": 0.0018005319929216058,
      "epoch": 1.399993896484375e-05,
      "step": 88680
    },
    {
      "epoch": 1.399993896484375e-05,
      "step": 88680,
      "training_step_time": 6.888693571090698
    },
    {
      "epoch": 1.4000892639160155e-05,
      "model_forward_time": 1.90167236328125,
      "step": 88681
    },
    {
      "epoch": 1.4000892639160155e-05,
      "step": 88681,
      "training_step_time": 7.626419305801392
    },
    {
      "epoch": 1.4001846313476563e-05,
      "model_forward_time": 1.8960661888122559,
      "step": 88682
    },
    {
      "epoch": 1.4001846313476563e-05,
      "step": 88682,
      "training_step_time": 11.347476243972778
    },
    {
      "epoch": 1.4002799987792969e-05,
      "model_forward_time": 1.8753836154937744,
      "step": 88683
    },
    {
      "epoch": 1.4002799987792969e-05,
      "step": 88683,
      "training_step_time": 9.267007827758789
    },
    {
      "epoch": 1.4003753662109374e-05,
      "model_forward_time": 1.8794450759887695,
      "step": 88684
    },
    {
      "epoch": 1.4003753662109374e-05,
      "step": 88684,
      "training_step_time": 7.79048228263855
    },
    {
      "epoch": 1.4004707336425782e-05,
      "model_forward_time": 1.8830358982086182,
      "step": 88685
    },
    {
      "epoch": 1.4004707336425782e-05,
      "step": 88685,
      "training_step_time": 8.2561194896698
    },
    {
      "epoch": 1.4005661010742187e-05,
      "model_forward_time": 1.9072329998016357,
      "step": 88686
    },
    {
      "epoch": 1.4005661010742187e-05,
      "step": 88686,
      "training_step_time": 7.749834299087524
    },
    {
      "epoch": 1.4006614685058593e-05,
      "model_forward_time": 1.898270606994629,
      "step": 88687
    },
    {
      "epoch": 1.4006614685058593e-05,
      "step": 88687,
      "training_step_time": 7.816906213760376
    },
    {
      "epoch": 1.4007568359375e-05,
      "model_forward_time": 1.903130054473877,
      "step": 88688
    },
    {
      "epoch": 1.4007568359375e-05,
      "step": 88688,
      "training_step_time": 6.890591859817505
    },
    {
      "epoch": 1.4008522033691406e-05,
      "model_forward_time": 1.8912358283996582,
      "step": 88689
    },
    {
      "epoch": 1.4008522033691406e-05,
      "step": 88689,
      "training_step_time": 6.957866430282593
    },
    {
      "epoch": 1.4009475708007812e-05,
      "grad_norm": 0.07782680541276932,
      "learning_rate": 3.4572043609177263e-07,
      "loss": 0.0572,
      "step": 88690
    },
    {
      "epoch": 1.4009475708007812e-05,
      "model_forward_time": 1.8846912384033203,
      "step": 88690
    },
    {
      "dynamics_loss_avg": 0.06905390694737434,
      "epoch": 1.4009475708007812e-05,
      "step": 88690
    },
    {
      "action_loss_avg": 0.002926942118210718,
      "epoch": 1.4009475708007812e-05,
      "step": 88690
    },
    {
      "epoch": 1.4009475708007812e-05,
      "step": 88690,
      "training_step_time": 7.29386830329895
    },
    {
      "epoch": 1.401042938232422e-05,
      "model_forward_time": 1.890282392501831,
      "step": 88691
    },
    {
      "epoch": 1.401042938232422e-05,
      "step": 88691,
      "training_step_time": 10.008916139602661
    },
    {
      "epoch": 1.4011383056640625e-05,
      "model_forward_time": 1.8586013317108154,
      "step": 88692
    },
    {
      "epoch": 1.4011383056640625e-05,
      "step": 88692,
      "training_step_time": 10.857182502746582
    },
    {
      "epoch": 1.4012336730957031e-05,
      "model_forward_time": 2.427981376647949,
      "step": 88693
    },
    {
      "epoch": 1.4012336730957031e-05,
      "step": 88693,
      "training_step_time": 7.751319885253906
    },
    {
      "epoch": 1.4013290405273438e-05,
      "model_forward_time": 1.7505793571472168,
      "step": 88694
    },
    {
      "epoch": 1.4013290405273438e-05,
      "step": 88694,
      "training_step_time": 7.6743738651275635
    },
    {
      "epoch": 1.4014244079589844e-05,
      "model_forward_time": 1.7360584735870361,
      "step": 88695
    },
    {
      "epoch": 1.4014244079589844e-05,
      "step": 88695,
      "training_step_time": 7.697690725326538
    },
    {
      "epoch": 1.401519775390625e-05,
      "model_forward_time": 1.7435717582702637,
      "step": 88696
    },
    {
      "epoch": 1.401519775390625e-05,
      "step": 88696,
      "training_step_time": 7.560739755630493
    },
    {
      "epoch": 1.4016151428222655e-05,
      "model_forward_time": 1.7356209754943848,
      "step": 88697
    },
    {
      "epoch": 1.4016151428222655e-05,
      "step": 88697,
      "training_step_time": 7.39060378074646
    },
    {
      "epoch": 1.4017105102539063e-05,
      "model_forward_time": 1.7403316497802734,
      "step": 88698
    },
    {
      "epoch": 1.4017105102539063e-05,
      "step": 88698,
      "training_step_time": 6.968185901641846
    },
    {
      "epoch": 1.4018058776855469e-05,
      "model_forward_time": 1.7430708408355713,
      "step": 88699
    },
    {
      "epoch": 1.4018058776855469e-05,
      "step": 88699,
      "training_step_time": 7.25360894203186
    },
    {
      "epoch": 1.4019012451171874e-05,
      "grad_norm": 0.18275503814220428,
      "learning_rate": 3.451165351017383e-07,
      "loss": 0.0579,
      "step": 88700
    },
    {
      "epoch": 1.4019012451171874e-05,
      "model_forward_time": 1.733616590499878,
      "step": 88700
    },
    {
      "dynamics_loss_avg": 0.06227795034646988,
      "epoch": 1.4019012451171874e-05,
      "step": 88700
    },
    {
      "action_loss_avg": 0.0024717803375096993,
      "epoch": 1.4019012451171874e-05,
      "step": 88700
    },
    {
      "epoch": 1.4019012451171874e-05,
      "step": 88700,
      "training_step_time": 7.861985206604004
    },
    {
      "epoch": 1.4019966125488282e-05,
      "model_forward_time": 1.7396187782287598,
      "step": 88701
    },
    {
      "epoch": 1.4019966125488282e-05,
      "step": 88701,
      "training_step_time": 10.369565725326538
    },
    {
      "epoch": 1.4020919799804687e-05,
      "model_forward_time": 2.436718225479126,
      "step": 88702
    },
    {
      "epoch": 1.4020919799804687e-05,
      "step": 88702,
      "training_step_time": 8.633306503295898
    },
    {
      "epoch": 1.4021873474121093e-05,
      "model_forward_time": 1.7415015697479248,
      "step": 88703
    },
    {
      "epoch": 1.4021873474121093e-05,
      "step": 88703,
      "training_step_time": 7.924611330032349
    },
    {
      "epoch": 1.40228271484375e-05,
      "model_forward_time": 1.747312307357788,
      "step": 88704
    },
    {
      "epoch": 1.40228271484375e-05,
      "step": 88704,
      "training_step_time": 7.757561922073364
    },
    {
      "epoch": 1.4023780822753906e-05,
      "model_forward_time": 1.026857614517212,
      "step": 88705
    },
    {
      "epoch": 1.4023780822753906e-05,
      "step": 88705,
      "training_step_time": 7.430457592010498
    },
    {
      "epoch": 1.4024734497070312e-05,
      "model_forward_time": 1.7327418327331543,
      "step": 88706
    },
    {
      "epoch": 1.4024734497070312e-05,
      "step": 88706,
      "training_step_time": 7.381831169128418
    },
    {
      "epoch": 1.402568817138672e-05,
      "model_forward_time": 1.7381350994110107,
      "step": 88707
    },
    {
      "epoch": 1.402568817138672e-05,
      "step": 88707,
      "training_step_time": 6.883519887924194
    },
    {
      "epoch": 1.4026641845703125e-05,
      "model_forward_time": 1.9021995067596436,
      "step": 88708
    },
    {
      "epoch": 1.4026641845703125e-05,
      "step": 88708,
      "training_step_time": 6.901190996170044
    },
    {
      "epoch": 1.4027595520019531e-05,
      "model_forward_time": 1.8889219760894775,
      "step": 88709
    },
    {
      "epoch": 1.4027595520019531e-05,
      "step": 88709,
      "training_step_time": 7.825225591659546
    },
    {
      "epoch": 1.4028549194335938e-05,
      "grad_norm": 0.09463292360305786,
      "learning_rate": 3.4451314316276074e-07,
      "loss": 0.0568,
      "step": 88710
    },
    {
      "epoch": 1.4028549194335938e-05,
      "model_forward_time": 1.7785975933074951,
      "step": 88710
    },
    {
      "dynamics_loss_avg": 0.034597691241651775,
      "epoch": 1.4028549194335938e-05,
      "step": 88710
    },
    {
      "action_loss_avg": 0.0015599683189066126,
      "epoch": 1.4028549194335938e-05,
      "step": 88710
    },
    {
      "epoch": 1.4028549194335938e-05,
      "step": 88710,
      "training_step_time": 8.842589855194092
    },
    {
      "epoch": 1.4029502868652344e-05,
      "model_forward_time": 2.4453580379486084,
      "step": 88711
    },
    {
      "epoch": 1.4029502868652344e-05,
      "step": 88711,
      "training_step_time": 8.565427780151367
    },
    {
      "epoch": 1.403045654296875e-05,
      "model_forward_time": 1.7324516773223877,
      "step": 88712
    },
    {
      "epoch": 1.403045654296875e-05,
      "step": 88712,
      "training_step_time": 7.851909399032593
    },
    {
      "epoch": 1.4031410217285155e-05,
      "model_forward_time": 1.73768949508667,
      "step": 88713
    },
    {
      "epoch": 1.4031410217285155e-05,
      "step": 88713,
      "training_step_time": 7.727156162261963
    },
    {
      "epoch": 1.4032363891601563e-05,
      "model_forward_time": 1.7441163063049316,
      "step": 88714
    },
    {
      "epoch": 1.4032363891601563e-05,
      "step": 88714,
      "training_step_time": 7.8282270431518555
    },
    {
      "epoch": 1.4033317565917969e-05,
      "model_forward_time": 1.738290786743164,
      "step": 88715
    },
    {
      "epoch": 1.4033317565917969e-05,
      "step": 88715,
      "training_step_time": 7.402771234512329
    },
    {
      "epoch": 1.4034271240234374e-05,
      "model_forward_time": 1.7342462539672852,
      "step": 88716
    },
    {
      "epoch": 1.4034271240234374e-05,
      "step": 88716,
      "training_step_time": 6.93265700340271
    },
    {
      "epoch": 1.4035224914550782e-05,
      "model_forward_time": 1.7345497608184814,
      "step": 88717
    },
    {
      "epoch": 1.4035224914550782e-05,
      "step": 88717,
      "training_step_time": 7.218026161193848
    },
    {
      "epoch": 1.4036178588867187e-05,
      "model_forward_time": 1.7654423713684082,
      "step": 88718
    },
    {
      "epoch": 1.4036178588867187e-05,
      "step": 88718,
      "training_step_time": 7.807821035385132
    },
    {
      "epoch": 1.4037132263183593e-05,
      "model_forward_time": 1.736565113067627,
      "step": 88719
    },
    {
      "epoch": 1.4037132263183593e-05,
      "step": 88719,
      "training_step_time": 10.925307273864746
    },
    {
      "epoch": 1.40380859375e-05,
      "grad_norm": 0.08913595974445343,
      "learning_rate": 3.4391026034082674e-07,
      "loss": 0.0583,
      "step": 88720
    },
    {
      "epoch": 1.40380859375e-05,
      "model_forward_time": 2.439547061920166,
      "step": 88720
    },
    {
      "dynamics_loss_avg": 0.08206995688378811,
      "epoch": 1.40380859375e-05,
      "step": 88720
    },
    {
      "action_loss_avg": 0.0023724561295239257,
      "epoch": 1.40380859375e-05,
      "step": 88720
    },
    {
      "epoch": 1.40380859375e-05,
      "step": 88720,
      "training_step_time": 8.445370435714722
    },
    {
      "epoch": 1.4039039611816406e-05,
      "model_forward_time": 1.7987253665924072,
      "step": 88721
    },
    {
      "epoch": 1.4039039611816406e-05,
      "step": 88721,
      "training_step_time": 7.803468704223633
    },
    {
      "epoch": 1.4039993286132812e-05,
      "model_forward_time": 1.7423439025878906,
      "step": 88722
    },
    {
      "epoch": 1.4039993286132812e-05,
      "step": 88722,
      "training_step_time": 8.151006698608398
    },
    {
      "epoch": 1.404094696044922e-05,
      "model_forward_time": 1.7425110340118408,
      "step": 88723
    },
    {
      "epoch": 1.404094696044922e-05,
      "step": 88723,
      "training_step_time": 7.821789741516113
    },
    {
      "epoch": 1.4041900634765625e-05,
      "model_forward_time": 1.7667794227600098,
      "step": 88724
    },
    {
      "epoch": 1.4041900634765625e-05,
      "step": 88724,
      "training_step_time": 7.337324380874634
    },
    {
      "epoch": 1.4042854309082031e-05,
      "model_forward_time": 1.7307038307189941,
      "step": 88725
    },
    {
      "epoch": 1.4042854309082031e-05,
      "step": 88725,
      "training_step_time": 6.920450448989868
    },
    {
      "epoch": 1.4043807983398438e-05,
      "model_forward_time": 0.712195873260498,
      "step": 88726
    },
    {
      "epoch": 1.4043807983398438e-05,
      "step": 88726,
      "training_step_time": 7.197458267211914
    },
    {
      "epoch": 1.4044761657714844e-05,
      "model_forward_time": 1.7415847778320312,
      "step": 88727
    },
    {
      "epoch": 1.4044761657714844e-05,
      "step": 88727,
      "training_step_time": 7.677109003067017
    },
    {
      "epoch": 1.404571533203125e-05,
      "model_forward_time": 1.7390990257263184,
      "step": 88728
    },
    {
      "epoch": 1.404571533203125e-05,
      "step": 88728,
      "training_step_time": 10.296695947647095
    },
    {
      "epoch": 1.4046669006347655e-05,
      "model_forward_time": 1.7333903312683105,
      "step": 88729
    },
    {
      "epoch": 1.4046669006347655e-05,
      "step": 88729,
      "training_step_time": 9.037057638168335
    },
    {
      "epoch": 1.4047622680664063e-05,
      "grad_norm": 0.12160755693912506,
      "learning_rate": 3.433078867018674e-07,
      "loss": 0.0577,
      "step": 88730
    },
    {
      "epoch": 1.4047622680664063e-05,
      "model_forward_time": 2.3978912830352783,
      "step": 88730
    },
    {
      "dynamics_loss_avg": 0.052512104995548724,
      "epoch": 1.4047622680664063e-05,
      "step": 88730
    },
    {
      "action_loss_avg": 0.0013102845899993555,
      "epoch": 1.4047622680664063e-05,
      "step": 88730
    },
    {
      "epoch": 1.4047622680664063e-05,
      "step": 88730,
      "training_step_time": 7.841417551040649
    },
    {
      "epoch": 1.4048576354980469e-05,
      "model_forward_time": 1.7363247871398926,
      "step": 88731
    },
    {
      "epoch": 1.4048576354980469e-05,
      "step": 88731,
      "training_step_time": 8.206048011779785
    },
    {
      "epoch": 1.4049530029296874e-05,
      "model_forward_time": 1.7353112697601318,
      "step": 88732
    },
    {
      "epoch": 1.4049530029296874e-05,
      "step": 88732,
      "training_step_time": 7.775105714797974
    },
    {
      "epoch": 1.4050483703613282e-05,
      "model_forward_time": 1.727240800857544,
      "step": 88733
    },
    {
      "epoch": 1.4050483703613282e-05,
      "step": 88733,
      "training_step_time": 7.60277795791626
    },
    {
      "epoch": 1.4051437377929687e-05,
      "model_forward_time": 1.7294561862945557,
      "step": 88734
    },
    {
      "epoch": 1.4051437377929687e-05,
      "step": 88734,
      "training_step_time": 7.3805251121521
    },
    {
      "epoch": 1.4052391052246093e-05,
      "model_forward_time": 1.7313485145568848,
      "step": 88735
    },
    {
      "epoch": 1.4052391052246093e-05,
      "step": 88735,
      "training_step_time": 7.239761829376221
    },
    {
      "epoch": 1.40533447265625e-05,
      "model_forward_time": 1.7353713512420654,
      "step": 88736
    },
    {
      "epoch": 1.40533447265625e-05,
      "step": 88736,
      "training_step_time": 8.414230346679688
    },
    {
      "epoch": 1.4054298400878906e-05,
      "model_forward_time": 1.7338886260986328,
      "step": 88737
    },
    {
      "epoch": 1.4054298400878906e-05,
      "step": 88737,
      "training_step_time": 10.99597430229187
    },
    {
      "epoch": 1.4055252075195312e-05,
      "model_forward_time": 1.7328603267669678,
      "step": 88738
    },
    {
      "epoch": 1.4055252075195312e-05,
      "step": 88738,
      "training_step_time": 9.771403789520264
    },
    {
      "epoch": 1.405620574951172e-05,
      "model_forward_time": 2.426136016845703,
      "step": 88739
    },
    {
      "epoch": 1.405620574951172e-05,
      "step": 88739,
      "training_step_time": 8.379762172698975
    },
    {
      "epoch": 1.4057159423828125e-05,
      "grad_norm": 0.07249867171049118,
      "learning_rate": 3.427060223117551e-07,
      "loss": 0.0576,
      "step": 88740
    },
    {
      "epoch": 1.4057159423828125e-05,
      "model_forward_time": 1.73244047164917,
      "step": 88740
    },
    {
      "dynamics_loss_avg": 0.060046642646193506,
      "epoch": 1.4057159423828125e-05,
      "step": 88740
    },
    {
      "action_loss_avg": 0.0024766476708464324,
      "epoch": 1.4057159423828125e-05,
      "step": 88740
    },
    {
      "epoch": 1.4057159423828125e-05,
      "step": 88740,
      "training_step_time": 8.34124755859375
    },
    {
      "epoch": 1.4058113098144531e-05,
      "model_forward_time": 1.7317924499511719,
      "step": 88741
    },
    {
      "epoch": 1.4058113098144531e-05,
      "step": 88741,
      "training_step_time": 7.725097417831421
    },
    {
      "epoch": 1.4059066772460938e-05,
      "model_forward_time": 1.7288384437561035,
      "step": 88742
    },
    {
      "epoch": 1.4059066772460938e-05,
      "step": 88742,
      "training_step_time": 7.610366582870483
    },
    {
      "epoch": 1.4060020446777344e-05,
      "model_forward_time": 1.7350246906280518,
      "step": 88743
    },
    {
      "epoch": 1.4060020446777344e-05,
      "step": 88743,
      "training_step_time": 7.373086452484131
    },
    {
      "epoch": 1.406097412109375e-05,
      "model_forward_time": 1.729752779006958,
      "step": 88744
    },
    {
      "epoch": 1.406097412109375e-05,
      "step": 88744,
      "training_step_time": 7.458858013153076
    },
    {
      "epoch": 1.4061927795410156e-05,
      "model_forward_time": 1.7280328273773193,
      "step": 88745
    },
    {
      "epoch": 1.4061927795410156e-05,
      "step": 88745,
      "training_step_time": 7.938663482666016
    },
    {
      "epoch": 1.4062881469726563e-05,
      "model_forward_time": 1.741882562637329,
      "step": 88746
    },
    {
      "epoch": 1.4062881469726563e-05,
      "step": 88746,
      "training_step_time": 8.791558742523193
    },
    {
      "epoch": 1.4063835144042969e-05,
      "model_forward_time": 1.7274627685546875,
      "step": 88747
    },
    {
      "epoch": 1.4063835144042969e-05,
      "step": 88747,
      "training_step_time": 9.067536115646362
    },
    {
      "epoch": 1.4064788818359374e-05,
      "model_forward_time": 2.4463582038879395,
      "step": 88748
    },
    {
      "epoch": 1.4064788818359374e-05,
      "step": 88748,
      "training_step_time": 7.942744731903076
    },
    {
      "epoch": 1.4065742492675782e-05,
      "model_forward_time": 1.4488251209259033,
      "step": 88749
    },
    {
      "epoch": 1.4065742492675782e-05,
      "step": 88749,
      "training_step_time": 8.278802633285522
    },
    {
      "epoch": 1.4066696166992188e-05,
      "grad_norm": 0.09448880702257156,
      "learning_rate": 3.421046672363104e-07,
      "loss": 0.0557,
      "step": 88750
    },
    {
      "epoch": 1.4066696166992188e-05,
      "model_forward_time": 1.7278823852539062,
      "step": 88750
    },
    {
      "dynamics_loss_avg": 0.04324082955718041,
      "epoch": 1.4066696166992188e-05,
      "step": 88750
    },
    {
      "action_loss_avg": 0.001099138197605498,
      "epoch": 1.4066696166992188e-05,
      "step": 88750
    },
    {
      "epoch": 1.4066696166992188e-05,
      "step": 88750,
      "training_step_time": 7.6491358280181885
    },
    {
      "epoch": 1.4067649841308593e-05,
      "model_forward_time": 1.7349207401275635,
      "step": 88751
    },
    {
      "epoch": 1.4067649841308593e-05,
      "step": 88751,
      "training_step_time": 7.676784515380859
    },
    {
      "epoch": 1.4068603515625e-05,
      "model_forward_time": 1.732975721359253,
      "step": 88752
    },
    {
      "epoch": 1.4068603515625e-05,
      "step": 88752,
      "training_step_time": 7.312216520309448
    },
    {
      "epoch": 1.4069557189941406e-05,
      "model_forward_time": 1.7308859825134277,
      "step": 88753
    },
    {
      "epoch": 1.4069557189941406e-05,
      "step": 88753,
      "training_step_time": 6.816633939743042
    },
    {
      "epoch": 1.4070510864257812e-05,
      "model_forward_time": 1.7287604808807373,
      "step": 88754
    },
    {
      "epoch": 1.4070510864257812e-05,
      "step": 88754,
      "training_step_time": 7.396091461181641
    },
    {
      "epoch": 1.407146453857422e-05,
      "model_forward_time": 1.7290785312652588,
      "step": 88755
    },
    {
      "epoch": 1.407146453857422e-05,
      "step": 88755,
      "training_step_time": 9.124318361282349
    },
    {
      "epoch": 1.4072418212890625e-05,
      "model_forward_time": 1.7150211334228516,
      "step": 88756
    },
    {
      "epoch": 1.4072418212890625e-05,
      "step": 88756,
      "training_step_time": 10.076457500457764
    },
    {
      "epoch": 1.4073371887207031e-05,
      "model_forward_time": 1.0061266422271729,
      "step": 88757
    },
    {
      "epoch": 1.4073371887207031e-05,
      "step": 88757,
      "training_step_time": 8.966858386993408
    },
    {
      "epoch": 1.4074325561523437e-05,
      "model_forward_time": 2.351806402206421,
      "step": 88758
    },
    {
      "epoch": 1.4074325561523437e-05,
      "step": 88758,
      "training_step_time": 7.846078634262085
    },
    {
      "epoch": 1.4075279235839844e-05,
      "model_forward_time": 1.353496789932251,
      "step": 88759
    },
    {
      "epoch": 1.4075279235839844e-05,
      "step": 88759,
      "training_step_time": 7.498304843902588
    },
    {
      "epoch": 1.407623291015625e-05,
      "grad_norm": 0.09777285158634186,
      "learning_rate": 3.415038215412969e-07,
      "loss": 0.056,
      "step": 88760
    },
    {
      "epoch": 1.407623291015625e-05,
      "model_forward_time": 1.758986234664917,
      "step": 88760
    },
    {
      "dynamics_loss_avg": 0.0546029475517571,
      "epoch": 1.407623291015625e-05,
      "step": 88760
    },
    {
      "action_loss_avg": 0.001604777219472453,
      "epoch": 1.407623291015625e-05,
      "step": 88760
    },
    {
      "epoch": 1.407623291015625e-05,
      "step": 88760,
      "training_step_time": 7.743981838226318
    },
    {
      "epoch": 1.4077186584472656e-05,
      "model_forward_time": 1.7339296340942383,
      "step": 88761
    },
    {
      "epoch": 1.4077186584472656e-05,
      "step": 88761,
      "training_step_time": 7.688510894775391
    },
    {
      "epoch": 1.4078140258789063e-05,
      "model_forward_time": 1.7279627323150635,
      "step": 88762
    },
    {
      "epoch": 1.4078140258789063e-05,
      "step": 88762,
      "training_step_time": 6.927893400192261
    },
    {
      "epoch": 1.4079093933105469e-05,
      "model_forward_time": 1.7248566150665283,
      "step": 88763
    },
    {
      "epoch": 1.4079093933105469e-05,
      "step": 88763,
      "training_step_time": 7.4015913009643555
    },
    {
      "epoch": 1.4080047607421874e-05,
      "model_forward_time": 1.7305350303649902,
      "step": 88764
    },
    {
      "epoch": 1.4080047607421874e-05,
      "step": 88764,
      "training_step_time": 8.720088481903076
    },
    {
      "epoch": 1.4081001281738282e-05,
      "model_forward_time": 1.7288637161254883,
      "step": 88765
    },
    {
      "epoch": 1.4081001281738282e-05,
      "step": 88765,
      "training_step_time": 7.754142761230469
    },
    {
      "epoch": 1.4081954956054688e-05,
      "model_forward_time": 1.7481105327606201,
      "step": 88766
    },
    {
      "epoch": 1.4081954956054688e-05,
      "step": 88766,
      "training_step_time": 7.986421823501587
    },
    {
      "epoch": 1.4082908630371093e-05,
      "model_forward_time": 2.371919870376587,
      "step": 88767
    },
    {
      "epoch": 1.4082908630371093e-05,
      "step": 88767,
      "training_step_time": 7.804392099380493
    },
    {
      "epoch": 1.40838623046875e-05,
      "model_forward_time": 1.7103967666625977,
      "step": 88768
    },
    {
      "epoch": 1.40838623046875e-05,
      "step": 88768,
      "training_step_time": 7.7328572273254395
    },
    {
      "epoch": 1.4084815979003906e-05,
      "model_forward_time": 1.7243592739105225,
      "step": 88769
    },
    {
      "epoch": 1.4084815979003906e-05,
      "step": 88769,
      "training_step_time": 7.511854648590088
    },
    {
      "epoch": 1.4085769653320312e-05,
      "grad_norm": 0.0856751874089241,
      "learning_rate": 3.409034852924209e-07,
      "loss": 0.0555,
      "step": 88770
    },
    {
      "epoch": 1.4085769653320312e-05,
      "model_forward_time": 1.713440179824829,
      "step": 88770
    },
    {
      "dynamics_loss_avg": 0.05285679958760738,
      "epoch": 1.4085769653320312e-05,
      "step": 88770
    },
    {
      "action_loss_avg": 0.001827769135707058,
      "epoch": 1.4085769653320312e-05,
      "step": 88770
    },
    {
      "epoch": 1.4085769653320312e-05,
      "step": 88770,
      "training_step_time": 7.711100816726685
    },
    {
      "epoch": 1.408672332763672e-05,
      "model_forward_time": 1.731278896331787,
      "step": 88771
    },
    {
      "epoch": 1.408672332763672e-05,
      "step": 88771,
      "training_step_time": 7.327730894088745
    },
    {
      "epoch": 1.4087677001953125e-05,
      "model_forward_time": 1.726414442062378,
      "step": 88772
    },
    {
      "epoch": 1.4087677001953125e-05,
      "step": 88772,
      "training_step_time": 7.216332197189331
    },
    {
      "epoch": 1.4088630676269531e-05,
      "model_forward_time": 1.7222349643707275,
      "step": 88773
    },
    {
      "epoch": 1.4088630676269531e-05,
      "step": 88773,
      "training_step_time": 8.758709192276001
    },
    {
      "epoch": 1.4089584350585937e-05,
      "model_forward_time": 1.7236249446868896,
      "step": 88774
    },
    {
      "epoch": 1.4089584350585937e-05,
      "step": 88774,
      "training_step_time": 10.909279346466064
    },
    {
      "epoch": 1.4090538024902344e-05,
      "model_forward_time": 1.71645188331604,
      "step": 88775
    },
    {
      "epoch": 1.4090538024902344e-05,
      "step": 88775,
      "training_step_time": 8.495023488998413
    },
    {
      "epoch": 1.409149169921875e-05,
      "model_forward_time": 2.3936586380004883,
      "step": 88776
    },
    {
      "epoch": 1.409149169921875e-05,
      "step": 88776,
      "training_step_time": 8.88190221786499
    },
    {
      "epoch": 1.4092445373535156e-05,
      "model_forward_time": 1.7272734642028809,
      "step": 88777
    },
    {
      "epoch": 1.4092445373535156e-05,
      "step": 88777,
      "training_step_time": 7.7620627880096436
    },
    {
      "epoch": 1.4093399047851563e-05,
      "model_forward_time": 1.7275831699371338,
      "step": 88778
    },
    {
      "epoch": 1.4093399047851563e-05,
      "step": 88778,
      "training_step_time": 7.540273666381836
    },
    {
      "epoch": 1.4094352722167969e-05,
      "model_forward_time": 1.7303650379180908,
      "step": 88779
    },
    {
      "epoch": 1.4094352722167969e-05,
      "step": 88779,
      "training_step_time": 7.709423303604126
    },
    {
      "epoch": 1.4095306396484374e-05,
      "grad_norm": 0.08294695615768433,
      "learning_rate": 3.403036585553349e-07,
      "loss": 0.0551,
      "step": 88780
    },
    {
      "epoch": 1.4095306396484374e-05,
      "model_forward_time": 1.7221271991729736,
      "step": 88780
    },
    {
      "dynamics_loss_avg": 0.0730820819735527,
      "epoch": 1.4095306396484374e-05,
      "step": 88780
    },
    {
      "action_loss_avg": 0.0025954598560929297,
      "epoch": 1.4095306396484374e-05,
      "step": 88780
    },
    {
      "epoch": 1.4095306396484374e-05,
      "step": 88780,
      "training_step_time": 7.313191175460815
    },
    {
      "epoch": 1.4096260070800782e-05,
      "model_forward_time": 1.3498954772949219,
      "step": 88781
    },
    {
      "epoch": 1.4096260070800782e-05,
      "step": 88781,
      "training_step_time": 7.195751667022705
    },
    {
      "epoch": 1.4097213745117188e-05,
      "model_forward_time": 1.7203125953674316,
      "step": 88782
    },
    {
      "epoch": 1.4097213745117188e-05,
      "step": 88782,
      "training_step_time": 8.240585565567017
    },
    {
      "epoch": 1.4098167419433593e-05,
      "model_forward_time": 1.7109313011169434,
      "step": 88783
    },
    {
      "epoch": 1.4098167419433593e-05,
      "step": 88783,
      "training_step_time": 7.763208389282227
    },
    {
      "epoch": 1.409912109375e-05,
      "model_forward_time": 1.7284486293792725,
      "step": 88784
    },
    {
      "epoch": 1.409912109375e-05,
      "step": 88784,
      "training_step_time": 8.972236633300781
    },
    {
      "epoch": 1.4100074768066406e-05,
      "model_forward_time": 1.7376115322113037,
      "step": 88785
    },
    {
      "epoch": 1.4100074768066406e-05,
      "step": 88785,
      "training_step_time": 7.912532329559326
    },
    {
      "epoch": 1.4101028442382812e-05,
      "model_forward_time": 2.3565237522125244,
      "step": 88786
    },
    {
      "epoch": 1.4101028442382812e-05,
      "step": 88786,
      "training_step_time": 8.698352336883545
    },
    {
      "epoch": 1.410198211669922e-05,
      "model_forward_time": 1.7334873676300049,
      "step": 88787
    },
    {
      "epoch": 1.410198211669922e-05,
      "step": 88787,
      "training_step_time": 7.93949031829834
    },
    {
      "epoch": 1.4102935791015625e-05,
      "model_forward_time": 1.7375235557556152,
      "step": 88788
    },
    {
      "epoch": 1.4102935791015625e-05,
      "step": 88788,
      "training_step_time": 7.712858200073242
    },
    {
      "epoch": 1.4103889465332031e-05,
      "model_forward_time": 1.7266128063201904,
      "step": 88789
    },
    {
      "epoch": 1.4103889465332031e-05,
      "step": 88789,
      "training_step_time": 7.3518571853637695
    },
    {
      "epoch": 1.4104843139648437e-05,
      "grad_norm": 0.13214923441410065,
      "learning_rate": 3.3970434139563467e-07,
      "loss": 0.0552,
      "step": 88790
    },
    {
      "epoch": 1.4104843139648437e-05,
      "model_forward_time": 1.7253351211547852,
      "step": 88790
    },
    {
      "dynamics_loss_avg": 0.05741118099540472,
      "epoch": 1.4104843139648437e-05,
      "step": 88790
    },
    {
      "action_loss_avg": 0.002002640676801093,
      "epoch": 1.4104843139648437e-05,
      "step": 88790
    },
    {
      "epoch": 1.4104843139648437e-05,
      "step": 88790,
      "training_step_time": 7.282524585723877
    },
    {
      "epoch": 1.4105796813964844e-05,
      "model_forward_time": 1.7390296459197998,
      "step": 88791
    },
    {
      "epoch": 1.4105796813964844e-05,
      "step": 88791,
      "training_step_time": 7.787915229797363
    },
    {
      "epoch": 1.410675048828125e-05,
      "model_forward_time": 1.739884614944458,
      "step": 88792
    },
    {
      "epoch": 1.410675048828125e-05,
      "step": 88792,
      "training_step_time": 8.407378196716309
    },
    {
      "epoch": 1.4107704162597656e-05,
      "model_forward_time": 1.7422544956207275,
      "step": 88793
    },
    {
      "epoch": 1.4107704162597656e-05,
      "step": 88793,
      "training_step_time": 8.363303184509277
    },
    {
      "epoch": 1.4108657836914063e-05,
      "model_forward_time": 1.8658421039581299,
      "step": 88794
    },
    {
      "epoch": 1.4108657836914063e-05,
      "step": 88794,
      "training_step_time": 8.793843507766724
    },
    {
      "epoch": 1.4109611511230469e-05,
      "model_forward_time": 2.387216091156006,
      "step": 88795
    },
    {
      "epoch": 1.4109611511230469e-05,
      "step": 88795,
      "training_step_time": 8.540099620819092
    },
    {
      "epoch": 1.4110565185546874e-05,
      "model_forward_time": 1.749187707901001,
      "step": 88796
    },
    {
      "epoch": 1.4110565185546874e-05,
      "step": 88796,
      "training_step_time": 8.81206727027893
    },
    {
      "epoch": 1.4111518859863282e-05,
      "model_forward_time": 1.7448627948760986,
      "step": 88797
    },
    {
      "epoch": 1.4111518859863282e-05,
      "step": 88797,
      "training_step_time": 7.66275954246521
    },
    {
      "epoch": 1.4112472534179688e-05,
      "model_forward_time": 1.7502927780151367,
      "step": 88798
    },
    {
      "epoch": 1.4112472534179688e-05,
      "step": 88798,
      "training_step_time": 7.415841341018677
    },
    {
      "epoch": 1.4113426208496093e-05,
      "model_forward_time": 1.750993013381958,
      "step": 88799
    },
    {
      "epoch": 1.4113426208496093e-05,
      "step": 88799,
      "training_step_time": 7.29610538482666
    },
    {
      "epoch": 1.41143798828125e-05,
      "grad_norm": 0.08163369446992874,
      "learning_rate": 3.3910553387886235e-07,
      "loss": 0.055,
      "step": 88800
    },
    {
      "epoch": 1.41143798828125e-05,
      "model_forward_time": 1.742260456085205,
      "step": 88800
    },
    {
      "dynamics_loss_avg": 0.06188090872019529,
      "epoch": 1.41143798828125e-05,
      "step": 88800
    },
    {
      "action_loss_avg": 0.001386774267302826,
      "epoch": 1.41143798828125e-05,
      "step": 88800
    },
    {
      "epoch": 1.41143798828125e-05,
      "step": 88800,
      "training_step_time": 9.407552242279053
    },
    {
      "epoch": 1.4115333557128906e-05,
      "model_forward_time": 1.7382709980010986,
      "step": 88801
    },
    {
      "epoch": 1.4115333557128906e-05,
      "step": 88801,
      "training_step_time": 8.724864721298218
    },
    {
      "epoch": 1.4116287231445312e-05,
      "model_forward_time": 1.742619276046753,
      "step": 88802
    },
    {
      "epoch": 1.4116287231445312e-05,
      "step": 88802,
      "training_step_time": 9.44139575958252
    },
    {
      "epoch": 1.411724090576172e-05,
      "model_forward_time": 1.8460240364074707,
      "step": 88803
    },
    {
      "epoch": 1.411724090576172e-05,
      "step": 88803,
      "training_step_time": 11.467066049575806
    },
    {
      "epoch": 1.4118194580078125e-05,
      "model_forward_time": 2.362595558166504,
      "step": 88804
    },
    {
      "epoch": 1.4118194580078125e-05,
      "step": 88804,
      "training_step_time": 7.829975843429565
    },
    {
      "epoch": 1.4119148254394531e-05,
      "model_forward_time": 1.7440085411071777,
      "step": 88805
    },
    {
      "epoch": 1.4119148254394531e-05,
      "step": 88805,
      "training_step_time": 7.888492107391357
    },
    {
      "epoch": 1.4120101928710937e-05,
      "model_forward_time": 1.743544578552246,
      "step": 88806
    },
    {
      "epoch": 1.4120101928710937e-05,
      "step": 88806,
      "training_step_time": 7.780066251754761
    },
    {
      "epoch": 1.4121055603027344e-05,
      "model_forward_time": 1.7417428493499756,
      "step": 88807
    },
    {
      "epoch": 1.4121055603027344e-05,
      "step": 88807,
      "training_step_time": 7.448251485824585
    },
    {
      "epoch": 1.412200927734375e-05,
      "model_forward_time": 1.7473206520080566,
      "step": 88808
    },
    {
      "epoch": 1.412200927734375e-05,
      "step": 88808,
      "training_step_time": 7.3225486278533936
    },
    {
      "epoch": 1.4122962951660156e-05,
      "model_forward_time": 1.7446937561035156,
      "step": 88809
    },
    {
      "epoch": 1.4122962951660156e-05,
      "step": 88809,
      "training_step_time": 7.844769239425659
    },
    {
      "epoch": 1.4123916625976563e-05,
      "grad_norm": 0.07288112491369247,
      "learning_rate": 3.3850723607049994e-07,
      "loss": 0.0548,
      "step": 88810
    },
    {
      "epoch": 1.4123916625976563e-05,
      "model_forward_time": 1.744067907333374,
      "step": 88810
    },
    {
      "dynamics_loss_avg": 0.05080899633467197,
      "epoch": 1.4123916625976563e-05,
      "step": 88810
    },
    {
      "action_loss_avg": 0.0014854805747745558,
      "epoch": 1.4123916625976563e-05,
      "step": 88810
    },
    {
      "epoch": 1.4123916625976563e-05,
      "step": 88810,
      "training_step_time": 8.400970697402954
    },
    {
      "epoch": 1.4124870300292969e-05,
      "model_forward_time": 1.7325806617736816,
      "step": 88811
    },
    {
      "epoch": 1.4124870300292969e-05,
      "step": 88811,
      "training_step_time": 10.46785306930542
    },
    {
      "epoch": 1.4125823974609374e-05,
      "model_forward_time": 1.835270881652832,
      "step": 88812
    },
    {
      "epoch": 1.4125823974609374e-05,
      "step": 88812,
      "training_step_time": 8.595377206802368
    },
    {
      "epoch": 1.4126777648925782e-05,
      "model_forward_time": 2.399616003036499,
      "step": 88813
    },
    {
      "epoch": 1.4126777648925782e-05,
      "step": 88813,
      "training_step_time": 7.791683673858643
    },
    {
      "epoch": 1.4127731323242188e-05,
      "model_forward_time": 1.7587072849273682,
      "step": 88814
    },
    {
      "epoch": 1.4127731323242188e-05,
      "step": 88814,
      "training_step_time": 7.887309551239014
    },
    {
      "epoch": 1.4128684997558593e-05,
      "model_forward_time": 1.7458341121673584,
      "step": 88815
    },
    {
      "epoch": 1.4128684997558593e-05,
      "step": 88815,
      "training_step_time": 7.712594270706177
    },
    {
      "epoch": 1.4129638671875e-05,
      "model_forward_time": 1.7341594696044922,
      "step": 88816
    },
    {
      "epoch": 1.4129638671875e-05,
      "step": 88816,
      "training_step_time": 7.829128980636597
    },
    {
      "epoch": 1.4130592346191406e-05,
      "model_forward_time": 1.762251615524292,
      "step": 88817
    },
    {
      "epoch": 1.4130592346191406e-05,
      "step": 88817,
      "training_step_time": 7.447365045547485
    },
    {
      "epoch": 1.4131546020507812e-05,
      "model_forward_time": 1.7528810501098633,
      "step": 88818
    },
    {
      "epoch": 1.4131546020507812e-05,
      "step": 88818,
      "training_step_time": 7.779797554016113
    },
    {
      "epoch": 1.413249969482422e-05,
      "model_forward_time": 1.7359364032745361,
      "step": 88819
    },
    {
      "epoch": 1.413249969482422e-05,
      "step": 88819,
      "training_step_time": 8.670661211013794
    },
    {
      "epoch": 1.4133453369140625e-05,
      "grad_norm": 0.09468504041433334,
      "learning_rate": 3.3790944803597904e-07,
      "loss": 0.0543,
      "step": 88820
    },
    {
      "epoch": 1.4133453369140625e-05,
      "model_forward_time": 1.729814052581787,
      "step": 88820
    },
    {
      "dynamics_loss_avg": 0.06583019532263279,
      "epoch": 1.4133453369140625e-05,
      "step": 88820
    },
    {
      "action_loss_avg": 0.0017221926129423082,
      "epoch": 1.4133453369140625e-05,
      "step": 88820
    },
    {
      "epoch": 1.4133453369140625e-05,
      "step": 88820,
      "training_step_time": 7.838381290435791
    },
    {
      "epoch": 1.4134407043457031e-05,
      "model_forward_time": 1.7741215229034424,
      "step": 88821
    },
    {
      "epoch": 1.4134407043457031e-05,
      "step": 88821,
      "training_step_time": 10.514930248260498
    },
    {
      "epoch": 1.4135360717773437e-05,
      "model_forward_time": 2.394413709640503,
      "step": 88822
    },
    {
      "epoch": 1.4135360717773437e-05,
      "step": 88822,
      "training_step_time": 8.830802202224731
    },
    {
      "epoch": 1.4136314392089844e-05,
      "model_forward_time": 1.3536179065704346,
      "step": 88823
    },
    {
      "epoch": 1.4136314392089844e-05,
      "step": 88823,
      "training_step_time": 7.872253894805908
    },
    {
      "epoch": 1.413726806640625e-05,
      "model_forward_time": 1.3710172176361084,
      "step": 88824
    },
    {
      "epoch": 1.413726806640625e-05,
      "step": 88824,
      "training_step_time": 7.67848801612854
    },
    {
      "epoch": 1.4138221740722656e-05,
      "model_forward_time": 1.3639566898345947,
      "step": 88825
    },
    {
      "epoch": 1.4138221740722656e-05,
      "step": 88825,
      "training_step_time": 7.898991107940674
    },
    {
      "epoch": 1.4139175415039063e-05,
      "model_forward_time": 1.713930368423462,
      "step": 88826
    },
    {
      "epoch": 1.4139175415039063e-05,
      "step": 88826,
      "training_step_time": 7.3322672843933105
    },
    {
      "epoch": 1.4140129089355469e-05,
      "model_forward_time": 1.7339444160461426,
      "step": 88827
    },
    {
      "epoch": 1.4140129089355469e-05,
      "step": 88827,
      "training_step_time": 7.810826301574707
    },
    {
      "epoch": 1.4141082763671874e-05,
      "model_forward_time": 1.734935998916626,
      "step": 88828
    },
    {
      "epoch": 1.4141082763671874e-05,
      "step": 88828,
      "training_step_time": 8.756739616394043
    },
    {
      "epoch": 1.4142036437988282e-05,
      "model_forward_time": 1.7450261116027832,
      "step": 88829
    },
    {
      "epoch": 1.4142036437988282e-05,
      "step": 88829,
      "training_step_time": 8.80875015258789
    },
    {
      "epoch": 1.4142990112304688e-05,
      "grad_norm": 0.12795625627040863,
      "learning_rate": 3.373121698406706e-07,
      "loss": 0.0541,
      "step": 88830
    },
    {
      "epoch": 1.4142990112304688e-05,
      "model_forward_time": 1.7381536960601807,
      "step": 88830
    },
    {
      "dynamics_loss_avg": 0.0721192111261189,
      "epoch": 1.4142990112304688e-05,
      "step": 88830
    },
    {
      "action_loss_avg": 0.0018277373106684535,
      "epoch": 1.4142990112304688e-05,
      "step": 88830
    },
    {
      "epoch": 1.4142990112304688e-05,
      "step": 88830,
      "training_step_time": 8.660297155380249
    },
    {
      "epoch": 1.4143943786621093e-05,
      "model_forward_time": 1.7433199882507324,
      "step": 88831
    },
    {
      "epoch": 1.4143943786621093e-05,
      "step": 88831,
      "training_step_time": 7.807112216949463
    },
    {
      "epoch": 1.41448974609375e-05,
      "model_forward_time": 2.3598837852478027,
      "step": 88832
    },
    {
      "epoch": 1.41448974609375e-05,
      "step": 88832,
      "training_step_time": 7.963942050933838
    },
    {
      "epoch": 1.4145851135253906e-05,
      "model_forward_time": 1.7340431213378906,
      "step": 88833
    },
    {
      "epoch": 1.4145851135253906e-05,
      "step": 88833,
      "training_step_time": 7.691048860549927
    },
    {
      "epoch": 1.4146804809570312e-05,
      "model_forward_time": 1.743048906326294,
      "step": 88834
    },
    {
      "epoch": 1.4146804809570312e-05,
      "step": 88834,
      "training_step_time": 7.852170705795288
    },
    {
      "epoch": 1.414775848388672e-05,
      "model_forward_time": 1.7494843006134033,
      "step": 88835
    },
    {
      "epoch": 1.414775848388672e-05,
      "step": 88835,
      "training_step_time": 7.934390068054199
    },
    {
      "epoch": 1.4148712158203125e-05,
      "model_forward_time": 1.7471508979797363,
      "step": 88836
    },
    {
      "epoch": 1.4148712158203125e-05,
      "step": 88836,
      "training_step_time": 7.7840046882629395
    },
    {
      "epoch": 1.4149665832519531e-05,
      "model_forward_time": 1.7322990894317627,
      "step": 88837
    },
    {
      "epoch": 1.4149665832519531e-05,
      "step": 88837,
      "training_step_time": 7.842458009719849
    },
    {
      "epoch": 1.4150619506835937e-05,
      "model_forward_time": 1.7478559017181396,
      "step": 88838
    },
    {
      "epoch": 1.4150619506835937e-05,
      "step": 88838,
      "training_step_time": 8.346582651138306
    },
    {
      "epoch": 1.4151573181152344e-05,
      "model_forward_time": 1.7413654327392578,
      "step": 88839
    },
    {
      "epoch": 1.4151573181152344e-05,
      "step": 88839,
      "training_step_time": 10.114150762557983
    },
    {
      "epoch": 1.415252685546875e-05,
      "grad_norm": 0.08623448014259338,
      "learning_rate": 3.36715401549893e-07,
      "loss": 0.055,
      "step": 88840
    },
    {
      "epoch": 1.415252685546875e-05,
      "model_forward_time": 1.8252835273742676,
      "step": 88840
    },
    {
      "dynamics_loss_avg": 0.05601401105523109,
      "epoch": 1.415252685546875e-05,
      "step": 88840
    },
    {
      "action_loss_avg": 0.0012277920061023906,
      "epoch": 1.415252685546875e-05,
      "step": 88840
    },
    {
      "epoch": 1.415252685546875e-05,
      "step": 88840,
      "training_step_time": 9.36133074760437
    },
    {
      "epoch": 1.4153480529785156e-05,
      "model_forward_time": 2.376779317855835,
      "step": 88841
    },
    {
      "epoch": 1.4153480529785156e-05,
      "step": 88841,
      "training_step_time": 8.462928056716919
    },
    {
      "epoch": 1.4154434204101563e-05,
      "model_forward_time": 1.7371406555175781,
      "step": 88842
    },
    {
      "epoch": 1.4154434204101563e-05,
      "step": 88842,
      "training_step_time": 8.708058834075928
    },
    {
      "epoch": 1.4155387878417969e-05,
      "model_forward_time": 1.7665786743164062,
      "step": 88843
    },
    {
      "epoch": 1.4155387878417969e-05,
      "step": 88843,
      "training_step_time": 7.699802875518799
    },
    {
      "epoch": 1.4156341552734374e-05,
      "model_forward_time": 1.7560601234436035,
      "step": 88844
    },
    {
      "epoch": 1.4156341552734374e-05,
      "step": 88844,
      "training_step_time": 8.475735902786255
    },
    {
      "epoch": 1.4157295227050782e-05,
      "model_forward_time": 1.7588801383972168,
      "step": 88845
    },
    {
      "epoch": 1.4157295227050782e-05,
      "step": 88845,
      "training_step_time": 7.728063583374023
    },
    {
      "epoch": 1.4158248901367188e-05,
      "model_forward_time": 1.7610371112823486,
      "step": 88846
    },
    {
      "epoch": 1.4158248901367188e-05,
      "step": 88846,
      "training_step_time": 9.047497987747192
    },
    {
      "epoch": 1.4159202575683593e-05,
      "model_forward_time": 1.7599925994873047,
      "step": 88847
    },
    {
      "epoch": 1.4159202575683593e-05,
      "step": 88847,
      "training_step_time": 8.566013813018799
    },
    {
      "epoch": 1.416015625e-05,
      "model_forward_time": 1.756460189819336,
      "step": 88848
    },
    {
      "epoch": 1.416015625e-05,
      "step": 88848,
      "training_step_time": 8.889845371246338
    },
    {
      "epoch": 1.4161109924316406e-05,
      "model_forward_time": 1.8481383323669434,
      "step": 88849
    },
    {
      "epoch": 1.4161109924316406e-05,
      "step": 88849,
      "training_step_time": 9.399825096130371
    },
    {
      "epoch": 1.4162063598632812e-05,
      "grad_norm": 0.08358262479305267,
      "learning_rate": 3.3611914322890794e-07,
      "loss": 0.0546,
      "step": 88850
    },
    {
      "epoch": 1.4162063598632812e-05,
      "model_forward_time": 2.428388833999634,
      "step": 88850
    },
    {
      "dynamics_loss_avg": 0.050971776619553565,
      "epoch": 1.4162063598632812e-05,
      "step": 88850
    },
    {
      "action_loss_avg": 0.002501480205683038,
      "epoch": 1.4162063598632812e-05,
      "step": 88850
    },
    {
      "epoch": 1.4162063598632812e-05,
      "step": 88850,
      "training_step_time": 9.394908666610718
    },
    {
      "epoch": 1.416301727294922e-05,
      "model_forward_time": 1.762416124343872,
      "step": 88851
    },
    {
      "epoch": 1.416301727294922e-05,
      "step": 88851,
      "training_step_time": 7.712184906005859
    },
    {
      "epoch": 1.4163970947265625e-05,
      "model_forward_time": 1.7507987022399902,
      "step": 88852
    },
    {
      "epoch": 1.4163970947265625e-05,
      "step": 88852,
      "training_step_time": 7.499873399734497
    },
    {
      "epoch": 1.4164924621582031e-05,
      "model_forward_time": 1.7510077953338623,
      "step": 88853
    },
    {
      "epoch": 1.4164924621582031e-05,
      "step": 88853,
      "training_step_time": 7.730176687240601
    },
    {
      "epoch": 1.4165878295898437e-05,
      "model_forward_time": 1.7450816631317139,
      "step": 88854
    },
    {
      "epoch": 1.4165878295898437e-05,
      "step": 88854,
      "training_step_time": 7.75261664390564
    },
    {
      "epoch": 1.4166831970214844e-05,
      "model_forward_time": 1.7450127601623535,
      "step": 88855
    },
    {
      "epoch": 1.4166831970214844e-05,
      "step": 88855,
      "training_step_time": 8.417400121688843
    },
    {
      "epoch": 1.416778564453125e-05,
      "model_forward_time": 1.7604360580444336,
      "step": 88856
    },
    {
      "epoch": 1.416778564453125e-05,
      "step": 88856,
      "training_step_time": 8.604725360870361
    },
    {
      "epoch": 1.4168739318847656e-05,
      "model_forward_time": 1.7533979415893555,
      "step": 88857
    },
    {
      "epoch": 1.4168739318847656e-05,
      "step": 88857,
      "training_step_time": 9.226607322692871
    },
    {
      "epoch": 1.4169692993164063e-05,
      "model_forward_time": 1.8674259185791016,
      "step": 88858
    },
    {
      "epoch": 1.4169692993164063e-05,
      "step": 88858,
      "training_step_time": 7.74123215675354
    },
    {
      "epoch": 1.4170646667480469e-05,
      "model_forward_time": 2.417253255844116,
      "step": 88859
    },
    {
      "epoch": 1.4170646667480469e-05,
      "step": 88859,
      "training_step_time": 8.745736598968506
    },
    {
      "epoch": 1.4171600341796874e-05,
      "grad_norm": 0.08362404257059097,
      "learning_rate": 3.3552339494292206e-07,
      "loss": 0.0555,
      "step": 88860
    },
    {
      "epoch": 1.4171600341796874e-05,
      "model_forward_time": 1.7438297271728516,
      "step": 88860
    },
    {
      "dynamics_loss_avg": 0.04898234959691763,
      "epoch": 1.4171600341796874e-05,
      "step": 88860
    },
    {
      "action_loss_avg": 0.0011125902325147762,
      "epoch": 1.4171600341796874e-05,
      "step": 88860
    },
    {
      "epoch": 1.4171600341796874e-05,
      "step": 88860,
      "training_step_time": 7.6543731689453125
    },
    {
      "epoch": 1.4172554016113282e-05,
      "model_forward_time": 1.7630696296691895,
      "step": 88861
    },
    {
      "epoch": 1.4172554016113282e-05,
      "step": 88861,
      "training_step_time": 7.495260000228882
    },
    {
      "epoch": 1.4173507690429688e-05,
      "model_forward_time": 1.7600090503692627,
      "step": 88862
    },
    {
      "epoch": 1.4173507690429688e-05,
      "step": 88862,
      "training_step_time": 7.7738120555877686
    },
    {
      "epoch": 1.4174461364746093e-05,
      "model_forward_time": 1.736924409866333,
      "step": 88863
    },
    {
      "epoch": 1.4174461364746093e-05,
      "step": 88863,
      "training_step_time": 7.25447940826416
    },
    {
      "epoch": 1.41754150390625e-05,
      "model_forward_time": 1.759969711303711,
      "step": 88864
    },
    {
      "epoch": 1.41754150390625e-05,
      "step": 88864,
      "training_step_time": 7.76114821434021
    },
    {
      "epoch": 1.4176368713378906e-05,
      "model_forward_time": 1.7813913822174072,
      "step": 88865
    },
    {
      "epoch": 1.4176368713378906e-05,
      "step": 88865,
      "training_step_time": 9.617658853530884
    },
    {
      "epoch": 1.4177322387695312e-05,
      "model_forward_time": 1.7508933544158936,
      "step": 88866
    },
    {
      "epoch": 1.4177322387695312e-05,
      "step": 88866,
      "training_step_time": 9.498180389404297
    },
    {
      "epoch": 1.417827606201172e-05,
      "model_forward_time": 1.8570804595947266,
      "step": 88867
    },
    {
      "epoch": 1.417827606201172e-05,
      "step": 88867,
      "training_step_time": 7.734728574752808
    },
    {
      "epoch": 1.4179229736328125e-05,
      "model_forward_time": 1.7348253726959229,
      "step": 88868
    },
    {
      "epoch": 1.4179229736328125e-05,
      "step": 88868,
      "training_step_time": 10.429226160049438
    },
    {
      "epoch": 1.4180183410644531e-05,
      "model_forward_time": 1.7481184005737305,
      "step": 88869
    },
    {
      "epoch": 1.4180183410644531e-05,
      "step": 88869,
      "training_step_time": 8.853497505187988
    },
    {
      "epoch": 1.4181137084960937e-05,
      "grad_norm": 0.08729525655508041,
      "learning_rate": 3.349281567570839e-07,
      "loss": 0.0561,
      "step": 88870
    },
    {
      "epoch": 1.4181137084960937e-05,
      "model_forward_time": 1.7503674030303955,
      "step": 88870
    },
    {
      "dynamics_loss_avg": 0.07162986360490323,
      "epoch": 1.4181137084960937e-05,
      "step": 88870
    },
    {
      "action_loss_avg": 0.003085539228050038,
      "epoch": 1.4181137084960937e-05,
      "step": 88870
    },
    {
      "epoch": 1.4181137084960937e-05,
      "step": 88870,
      "training_step_time": 7.767049312591553
    },
    {
      "epoch": 1.4182090759277344e-05,
      "model_forward_time": 1.753591537475586,
      "step": 88871
    },
    {
      "epoch": 1.4182090759277344e-05,
      "step": 88871,
      "training_step_time": 7.79528021812439
    },
    {
      "epoch": 1.418304443359375e-05,
      "model_forward_time": 1.748570442199707,
      "step": 88872
    },
    {
      "epoch": 1.418304443359375e-05,
      "step": 88872,
      "training_step_time": 7.732414722442627
    },
    {
      "epoch": 1.4183998107910156e-05,
      "model_forward_time": 1.7499830722808838,
      "step": 88873
    },
    {
      "epoch": 1.4183998107910156e-05,
      "step": 88873,
      "training_step_time": 7.8787219524383545
    },
    {
      "epoch": 1.4184951782226563e-05,
      "model_forward_time": 1.7540013790130615,
      "step": 88874
    },
    {
      "epoch": 1.4184951782226563e-05,
      "step": 88874,
      "training_step_time": 7.860946416854858
    },
    {
      "epoch": 1.4185905456542969e-05,
      "model_forward_time": 1.7488694190979004,
      "step": 88875
    },
    {
      "epoch": 1.4185905456542969e-05,
      "step": 88875,
      "training_step_time": 9.88701581954956
    },
    {
      "epoch": 1.4186859130859374e-05,
      "model_forward_time": 1.8122987747192383,
      "step": 88876
    },
    {
      "epoch": 1.4186859130859374e-05,
      "step": 88876,
      "training_step_time": 9.496514081954956
    },
    {
      "epoch": 1.4187812805175782e-05,
      "model_forward_time": 1.745370626449585,
      "step": 88877
    },
    {
      "epoch": 1.4187812805175782e-05,
      "step": 88877,
      "training_step_time": 12.071942329406738
    },
    {
      "epoch": 1.4188766479492188e-05,
      "model_forward_time": 2.4051313400268555,
      "step": 88878
    },
    {
      "epoch": 1.4188766479492188e-05,
      "step": 88878,
      "training_step_time": 7.855417013168335
    },
    {
      "epoch": 1.4189720153808593e-05,
      "model_forward_time": 1.742516279220581,
      "step": 88879
    },
    {
      "epoch": 1.4189720153808593e-05,
      "step": 88879,
      "training_step_time": 7.660760879516602
    },
    {
      "epoch": 1.4190673828125e-05,
      "grad_norm": 0.07132678478956223,
      "learning_rate": 3.3433342873648964e-07,
      "loss": 0.0549,
      "step": 88880
    },
    {
      "epoch": 1.4190673828125e-05,
      "model_forward_time": 1.7448065280914307,
      "step": 88880
    },
    {
      "dynamics_loss_avg": 0.04539793338626623,
      "epoch": 1.4190673828125e-05,
      "step": 88880
    },
    {
      "action_loss_avg": 0.0013945363141829148,
      "epoch": 1.4190673828125e-05,
      "step": 88880
    },
    {
      "epoch": 1.4190673828125e-05,
      "step": 88880,
      "training_step_time": 7.637727975845337
    },
    {
      "epoch": 1.4191627502441406e-05,
      "model_forward_time": 1.7504334449768066,
      "step": 88881
    },
    {
      "epoch": 1.4191627502441406e-05,
      "step": 88881,
      "training_step_time": 8.361929893493652
    },
    {
      "epoch": 1.4192581176757812e-05,
      "model_forward_time": 1.7512142658233643,
      "step": 88882
    },
    {
      "epoch": 1.4192581176757812e-05,
      "step": 88882,
      "training_step_time": 7.847170352935791
    },
    {
      "epoch": 1.4193534851074218e-05,
      "model_forward_time": 1.7478399276733398,
      "step": 88883
    },
    {
      "epoch": 1.4193534851074218e-05,
      "step": 88883,
      "training_step_time": 7.643548250198364
    },
    {
      "epoch": 1.4194488525390625e-05,
      "model_forward_time": 1.7437551021575928,
      "step": 88884
    },
    {
      "epoch": 1.4194488525390625e-05,
      "step": 88884,
      "training_step_time": 7.663525342941284
    },
    {
      "epoch": 1.4195442199707031e-05,
      "model_forward_time": 1.7703301906585693,
      "step": 88885
    },
    {
      "epoch": 1.4195442199707031e-05,
      "step": 88885,
      "training_step_time": 8.472172737121582
    },
    {
      "epoch": 1.4196395874023437e-05,
      "model_forward_time": 1.9074106216430664,
      "step": 88886
    },
    {
      "epoch": 1.4196395874023437e-05,
      "step": 88886,
      "training_step_time": 8.592458724975586
    },
    {
      "epoch": 1.4197349548339844e-05,
      "model_forward_time": 2.4537453651428223,
      "step": 88887
    },
    {
      "epoch": 1.4197349548339844e-05,
      "step": 88887,
      "training_step_time": 8.987330436706543
    },
    {
      "epoch": 1.419830322265625e-05,
      "model_forward_time": 1.749596118927002,
      "step": 88888
    },
    {
      "epoch": 1.419830322265625e-05,
      "step": 88888,
      "training_step_time": 8.640055894851685
    },
    {
      "epoch": 1.4199256896972656e-05,
      "model_forward_time": 1.7464401721954346,
      "step": 88889
    },
    {
      "epoch": 1.4199256896972656e-05,
      "step": 88889,
      "training_step_time": 7.511984348297119
    },
    {
      "epoch": 1.4200210571289063e-05,
      "grad_norm": 0.08860083669424057,
      "learning_rate": 3.337392109461762e-07,
      "loss": 0.0562,
      "step": 88890
    },
    {
      "epoch": 1.4200210571289063e-05,
      "model_forward_time": 1.7393772602081299,
      "step": 88890
    },
    {
      "dynamics_loss_avg": 0.049510041158646344,
      "epoch": 1.4200210571289063e-05,
      "step": 88890
    },
    {
      "action_loss_avg": 0.0011050582572352142,
      "epoch": 1.4200210571289063e-05,
      "step": 88890
    },
    {
      "epoch": 1.4200210571289063e-05,
      "step": 88890,
      "training_step_time": 7.724807024002075
    },
    {
      "epoch": 1.4201164245605469e-05,
      "model_forward_time": 1.7504377365112305,
      "step": 88891
    },
    {
      "epoch": 1.4201164245605469e-05,
      "step": 88891,
      "training_step_time": 7.680700302124023
    },
    {
      "epoch": 1.4202117919921874e-05,
      "model_forward_time": 0.6908626556396484,
      "step": 88892
    },
    {
      "epoch": 1.4202117919921874e-05,
      "step": 88892,
      "training_step_time": 7.663860082626343
    },
    {
      "epoch": 1.4203071594238282e-05,
      "model_forward_time": 1.7501044273376465,
      "step": 88893
    },
    {
      "epoch": 1.4203071594238282e-05,
      "step": 88893,
      "training_step_time": 9.563388586044312
    },
    {
      "epoch": 1.4204025268554688e-05,
      "model_forward_time": 1.7442705631256104,
      "step": 88894
    },
    {
      "epoch": 1.4204025268554688e-05,
      "step": 88894,
      "training_step_time": 10.42262578010559
    },
    {
      "epoch": 1.4204978942871093e-05,
      "model_forward_time": 1.7441964149475098,
      "step": 88895
    },
    {
      "epoch": 1.4204978942871093e-05,
      "step": 88895,
      "training_step_time": 9.828500270843506
    },
    {
      "epoch": 1.42059326171875e-05,
      "model_forward_time": 1.7464649677276611,
      "step": 88896
    },
    {
      "epoch": 1.42059326171875e-05,
      "step": 88896,
      "training_step_time": 9.265717506408691
    },
    {
      "epoch": 1.4206886291503906e-05,
      "model_forward_time": 1.7534501552581787,
      "step": 88897
    },
    {
      "epoch": 1.4206886291503906e-05,
      "step": 88897,
      "training_step_time": 7.870830535888672
    },
    {
      "epoch": 1.4207839965820312e-05,
      "model_forward_time": 1.7548832893371582,
      "step": 88898
    },
    {
      "epoch": 1.4207839965820312e-05,
      "step": 88898,
      "training_step_time": 7.289715766906738
    },
    {
      "epoch": 1.4208793640136718e-05,
      "model_forward_time": 1.7555537223815918,
      "step": 88899
    },
    {
      "epoch": 1.4208793640136718e-05,
      "step": 88899,
      "training_step_time": 7.751342058181763
    },
    {
      "epoch": 1.4209747314453125e-05,
      "grad_norm": 0.07758411020040512,
      "learning_rate": 3.331455034511266e-07,
      "loss": 0.0565,
      "step": 88900
    },
    {
      "epoch": 1.4209747314453125e-05,
      "model_forward_time": 1.7490930557250977,
      "step": 88900
    },
    {
      "dynamics_loss_avg": 0.05685963090509176,
      "epoch": 1.4209747314453125e-05,
      "step": 88900
    },
    {
      "action_loss_avg": 0.0022568598127691076,
      "epoch": 1.4209747314453125e-05,
      "step": 88900
    },
    {
      "epoch": 1.4209747314453125e-05,
      "step": 88900,
      "training_step_time": 7.742740631103516
    },
    {
      "epoch": 1.4210700988769531e-05,
      "model_forward_time": 1.7450881004333496,
      "step": 88901
    },
    {
      "epoch": 1.4210700988769531e-05,
      "step": 88901,
      "training_step_time": 7.359086275100708
    },
    {
      "epoch": 1.4211654663085937e-05,
      "model_forward_time": 1.7566728591918945,
      "step": 88902
    },
    {
      "epoch": 1.4211654663085937e-05,
      "step": 88902,
      "training_step_time": 8.779958963394165
    },
    {
      "epoch": 1.4212608337402344e-05,
      "model_forward_time": 1.7492458820343018,
      "step": 88903
    },
    {
      "epoch": 1.4212608337402344e-05,
      "step": 88903,
      "training_step_time": 8.774658441543579
    },
    {
      "epoch": 1.421356201171875e-05,
      "model_forward_time": 1.8807580471038818,
      "step": 88904
    },
    {
      "epoch": 1.421356201171875e-05,
      "step": 88904,
      "training_step_time": 8.339765548706055
    },
    {
      "epoch": 1.4214515686035156e-05,
      "model_forward_time": 1.762460470199585,
      "step": 88905
    },
    {
      "epoch": 1.4214515686035156e-05,
      "step": 88905,
      "training_step_time": 9.170291185379028
    },
    {
      "epoch": 1.4215469360351563e-05,
      "model_forward_time": 2.371882915496826,
      "step": 88906
    },
    {
      "epoch": 1.4215469360351563e-05,
      "step": 88906,
      "training_step_time": 7.786067008972168
    },
    {
      "epoch": 1.4216423034667969e-05,
      "model_forward_time": 1.734337329864502,
      "step": 88907
    },
    {
      "epoch": 1.4216423034667969e-05,
      "step": 88907,
      "training_step_time": 7.670711040496826
    },
    {
      "epoch": 1.4217376708984374e-05,
      "model_forward_time": 1.7435402870178223,
      "step": 88908
    },
    {
      "epoch": 1.4217376708984374e-05,
      "step": 88908,
      "training_step_time": 7.468531370162964
    },
    {
      "epoch": 1.4218330383300782e-05,
      "model_forward_time": 1.7387170791625977,
      "step": 88909
    },
    {
      "epoch": 1.4218330383300782e-05,
      "step": 88909,
      "training_step_time": 8.67102599143982
    },
    {
      "epoch": 1.4219284057617188e-05,
      "grad_norm": 0.09082327038049698,
      "learning_rate": 3.325523063162689e-07,
      "loss": 0.0563,
      "step": 88910
    },
    {
      "epoch": 1.4219284057617188e-05,
      "model_forward_time": 1.7421762943267822,
      "step": 88910
    },
    {
      "dynamics_loss_avg": 0.06690104883164168,
      "epoch": 1.4219284057617188e-05,
      "step": 88910
    },
    {
      "action_loss_avg": 0.0030304920219350606,
      "epoch": 1.4219284057617188e-05,
      "step": 88910
    },
    {
      "epoch": 1.4219284057617188e-05,
      "step": 88910,
      "training_step_time": 7.339073181152344
    },
    {
      "epoch": 1.4220237731933593e-05,
      "model_forward_time": 1.7355029582977295,
      "step": 88911
    },
    {
      "epoch": 1.4220237731933593e-05,
      "step": 88911,
      "training_step_time": 7.696913003921509
    },
    {
      "epoch": 1.422119140625e-05,
      "model_forward_time": 1.740037441253662,
      "step": 88912
    },
    {
      "epoch": 1.422119140625e-05,
      "step": 88912,
      "training_step_time": 8.757026195526123
    },
    {
      "epoch": 1.4222145080566406e-05,
      "model_forward_time": 1.7669711112976074,
      "step": 88913
    },
    {
      "epoch": 1.4222145080566406e-05,
      "step": 88913,
      "training_step_time": 8.863081216812134
    },
    {
      "epoch": 1.4223098754882812e-05,
      "model_forward_time": 1.7399189472198486,
      "step": 88914
    },
    {
      "epoch": 1.4223098754882812e-05,
      "step": 88914,
      "training_step_time": 8.602097511291504
    },
    {
      "epoch": 1.4224052429199218e-05,
      "model_forward_time": 2.3654983043670654,
      "step": 88915
    },
    {
      "epoch": 1.4224052429199218e-05,
      "step": 88915,
      "training_step_time": 7.714850425720215
    },
    {
      "epoch": 1.4225006103515625e-05,
      "model_forward_time": 1.7365148067474365,
      "step": 88916
    },
    {
      "epoch": 1.4225006103515625e-05,
      "step": 88916,
      "training_step_time": 7.617007732391357
    },
    {
      "epoch": 1.4225959777832031e-05,
      "model_forward_time": 1.7459673881530762,
      "step": 88917
    },
    {
      "epoch": 1.4225959777832031e-05,
      "step": 88917,
      "training_step_time": 7.4586403369903564
    },
    {
      "epoch": 1.4226913452148437e-05,
      "model_forward_time": 1.746725082397461,
      "step": 88918
    },
    {
      "epoch": 1.4226913452148437e-05,
      "step": 88918,
      "training_step_time": 7.667687892913818
    },
    {
      "epoch": 1.4227867126464844e-05,
      "model_forward_time": 1.7489454746246338,
      "step": 88919
    },
    {
      "epoch": 1.4227867126464844e-05,
      "step": 88919,
      "training_step_time": 7.668259620666504
    },
    {
      "epoch": 1.422882080078125e-05,
      "grad_norm": 0.08142714202404022,
      "learning_rate": 3.3195961960647396e-07,
      "loss": 0.0568,
      "step": 88920
    },
    {
      "epoch": 1.422882080078125e-05,
      "model_forward_time": 1.7387757301330566,
      "step": 88920
    },
    {
      "dynamics_loss_avg": 0.047618408035486934,
      "epoch": 1.422882080078125e-05,
      "step": 88920
    },
    {
      "action_loss_avg": 0.0019925844419049097,
      "epoch": 1.422882080078125e-05,
      "step": 88920
    },
    {
      "epoch": 1.422882080078125e-05,
      "step": 88920,
      "training_step_time": 9.377875566482544
    },
    {
      "epoch": 1.4229774475097656e-05,
      "model_forward_time": 1.748187780380249,
      "step": 88921
    },
    {
      "epoch": 1.4229774475097656e-05,
      "step": 88921,
      "training_step_time": 7.6688947677612305
    },
    {
      "epoch": 1.4230728149414063e-05,
      "model_forward_time": 1.7596333026885986,
      "step": 88922
    },
    {
      "epoch": 1.4230728149414063e-05,
      "step": 88922,
      "training_step_time": 8.589208602905273
    },
    {
      "epoch": 1.4231681823730469e-05,
      "model_forward_time": 1.7501475811004639,
      "step": 88923
    },
    {
      "epoch": 1.4231681823730469e-05,
      "step": 88923,
      "training_step_time": 7.770982265472412
    },
    {
      "epoch": 1.4232635498046875e-05,
      "model_forward_time": 2.0168378353118896,
      "step": 88924
    },
    {
      "epoch": 1.4232635498046875e-05,
      "step": 88924,
      "training_step_time": 8.643510103225708
    },
    {
      "epoch": 1.4233589172363282e-05,
      "model_forward_time": 1.7463665008544922,
      "step": 88925
    },
    {
      "epoch": 1.4233589172363282e-05,
      "step": 88925,
      "training_step_time": 7.733980655670166
    },
    {
      "epoch": 1.4234542846679688e-05,
      "model_forward_time": 1.7428059577941895,
      "step": 88926
    },
    {
      "epoch": 1.4234542846679688e-05,
      "step": 88926,
      "training_step_time": 7.511587858200073
    },
    {
      "epoch": 1.4235496520996093e-05,
      "model_forward_time": 1.7495853900909424,
      "step": 88927
    },
    {
      "epoch": 1.4235496520996093e-05,
      "step": 88927,
      "training_step_time": 7.680758237838745
    },
    {
      "epoch": 1.42364501953125e-05,
      "model_forward_time": 1.745537281036377,
      "step": 88928
    },
    {
      "epoch": 1.42364501953125e-05,
      "step": 88928,
      "training_step_time": 8.255887031555176
    },
    {
      "epoch": 1.4237403869628907e-05,
      "model_forward_time": 1.7440452575683594,
      "step": 88929
    },
    {
      "epoch": 1.4237403869628907e-05,
      "step": 88929,
      "training_step_time": 7.801387786865234
    },
    {
      "epoch": 1.4238357543945312e-05,
      "grad_norm": 0.08417174220085144,
      "learning_rate": 3.31367443386556e-07,
      "loss": 0.057,
      "step": 88930
    },
    {
      "epoch": 1.4238357543945312e-05,
      "model_forward_time": 1.7377145290374756,
      "step": 88930
    },
    {
      "dynamics_loss_avg": 0.04790671691298485,
      "epoch": 1.4238357543945312e-05,
      "step": 88930
    },
    {
      "action_loss_avg": 0.0026585346815409137,
      "epoch": 1.4238357543945312e-05,
      "step": 88930
    },
    {
      "epoch": 1.4238357543945312e-05,
      "step": 88930,
      "training_step_time": 8.65860366821289
    },
    {
      "epoch": 1.4239311218261718e-05,
      "model_forward_time": 1.7818090915679932,
      "step": 88931
    },
    {
      "epoch": 1.4239311218261718e-05,
      "step": 88931,
      "training_step_time": 10.168391942977905
    },
    {
      "epoch": 1.4240264892578125e-05,
      "model_forward_time": 1.761408805847168,
      "step": 88932
    },
    {
      "epoch": 1.4240264892578125e-05,
      "step": 88932,
      "training_step_time": 9.620586395263672
    },
    {
      "epoch": 1.4241218566894531e-05,
      "model_forward_time": 2.416539430618286,
      "step": 88933
    },
    {
      "epoch": 1.4241218566894531e-05,
      "step": 88933,
      "training_step_time": 8.778811931610107
    },
    {
      "epoch": 1.4242172241210937e-05,
      "model_forward_time": 1.7728140354156494,
      "step": 88934
    },
    {
      "epoch": 1.4242172241210937e-05,
      "step": 88934,
      "training_step_time": 7.777066946029663
    },
    {
      "epoch": 1.4243125915527344e-05,
      "model_forward_time": 1.3960175514221191,
      "step": 88935
    },
    {
      "epoch": 1.4243125915527344e-05,
      "step": 88935,
      "training_step_time": 7.676554203033447
    },
    {
      "epoch": 1.424407958984375e-05,
      "model_forward_time": 1.7645304203033447,
      "step": 88936
    },
    {
      "epoch": 1.424407958984375e-05,
      "step": 88936,
      "training_step_time": 7.755273818969727
    },
    {
      "epoch": 1.4245033264160156e-05,
      "model_forward_time": 1.7695820331573486,
      "step": 88937
    },
    {
      "epoch": 1.4245033264160156e-05,
      "step": 88937,
      "training_step_time": 7.748104810714722
    },
    {
      "epoch": 1.4245986938476563e-05,
      "model_forward_time": 1.774193525314331,
      "step": 88938
    },
    {
      "epoch": 1.4245986938476563e-05,
      "step": 88938,
      "training_step_time": 7.79149866104126
    },
    {
      "epoch": 1.4246940612792969e-05,
      "model_forward_time": 1.766646385192871,
      "step": 88939
    },
    {
      "epoch": 1.4246940612792969e-05,
      "step": 88939,
      "training_step_time": 7.734102487564087
    },
    {
      "epoch": 1.4247894287109375e-05,
      "grad_norm": 0.07343023270368576,
      "learning_rate": 3.3077577772127554e-07,
      "loss": 0.056,
      "step": 88940
    },
    {
      "epoch": 1.4247894287109375e-05,
      "model_forward_time": 1.7659130096435547,
      "step": 88940
    },
    {
      "dynamics_loss_avg": 0.04838931392878294,
      "epoch": 1.4247894287109375e-05,
      "step": 88940
    },
    {
      "action_loss_avg": 0.0010258925263769924,
      "epoch": 1.4247894287109375e-05,
      "step": 88940
    },
    {
      "epoch": 1.4247894287109375e-05,
      "step": 88940,
      "training_step_time": 10.171556949615479
    },
    {
      "epoch": 1.4248847961425782e-05,
      "model_forward_time": 1.7752156257629395,
      "step": 88941
    },
    {
      "epoch": 1.4248847961425782e-05,
      "step": 88941,
      "training_step_time": 8.624072313308716
    },
    {
      "epoch": 1.4249801635742188e-05,
      "model_forward_time": 1.779531717300415,
      "step": 88942
    },
    {
      "epoch": 1.4249801635742188e-05,
      "step": 88942,
      "training_step_time": 7.854955434799194
    },
    {
      "epoch": 1.4250755310058593e-05,
      "model_forward_time": 2.4112613201141357,
      "step": 88943
    },
    {
      "epoch": 1.4250755310058593e-05,
      "step": 88943,
      "training_step_time": 8.61147665977478
    },
    {
      "epoch": 1.4251708984375e-05,
      "model_forward_time": 1.7744338512420654,
      "step": 88944
    },
    {
      "epoch": 1.4251708984375e-05,
      "step": 88944,
      "training_step_time": 8.158249616622925
    },
    {
      "epoch": 1.4252662658691407e-05,
      "model_forward_time": 1.7712724208831787,
      "step": 88945
    },
    {
      "epoch": 1.4252662658691407e-05,
      "step": 88945,
      "training_step_time": 7.75593638420105
    },
    {
      "epoch": 1.4253616333007812e-05,
      "model_forward_time": 1.768155574798584,
      "step": 88946
    },
    {
      "epoch": 1.4253616333007812e-05,
      "step": 88946,
      "training_step_time": 7.789384126663208
    },
    {
      "epoch": 1.4254570007324218e-05,
      "model_forward_time": 1.8170983791351318,
      "step": 88947
    },
    {
      "epoch": 1.4254570007324218e-05,
      "step": 88947,
      "training_step_time": 8.231356382369995
    },
    {
      "epoch": 1.4255523681640625e-05,
      "model_forward_time": 1.7680282592773438,
      "step": 88948
    },
    {
      "epoch": 1.4255523681640625e-05,
      "step": 88948,
      "training_step_time": 8.501616716384888
    },
    {
      "epoch": 1.4256477355957031e-05,
      "model_forward_time": 1.7545123100280762,
      "step": 88949
    },
    {
      "epoch": 1.4256477355957031e-05,
      "step": 88949,
      "training_step_time": 8.832106590270996
    },
    {
      "epoch": 1.4257431030273437e-05,
      "grad_norm": 0.08037854731082916,
      "learning_rate": 3.301846226753358e-07,
      "loss": 0.0567,
      "step": 88950
    },
    {
      "epoch": 1.4257431030273437e-05,
      "model_forward_time": 1.7649965286254883,
      "step": 88950
    },
    {
      "dynamics_loss_avg": 0.08911220356822014,
      "epoch": 1.4257431030273437e-05,
      "step": 88950
    },
    {
      "action_loss_avg": 0.0026544908352661876,
      "epoch": 1.4257431030273437e-05,
      "step": 88950
    },
    {
      "epoch": 1.4257431030273437e-05,
      "step": 88950,
      "training_step_time": 8.36087679862976
    },
    {
      "epoch": 1.4258384704589844e-05,
      "model_forward_time": 1.79164457321167,
      "step": 88951
    },
    {
      "epoch": 1.4258384704589844e-05,
      "step": 88951,
      "training_step_time": 7.772045612335205
    },
    {
      "epoch": 1.425933837890625e-05,
      "model_forward_time": 2.4134745597839355,
      "step": 88952
    },
    {
      "epoch": 1.425933837890625e-05,
      "step": 88952,
      "training_step_time": 7.795233488082886
    },
    {
      "epoch": 1.4260292053222656e-05,
      "model_forward_time": 1.76847505569458,
      "step": 88953
    },
    {
      "epoch": 1.4260292053222656e-05,
      "step": 88953,
      "training_step_time": 7.54583477973938
    },
    {
      "epoch": 1.4261245727539063e-05,
      "model_forward_time": 1.7598481178283691,
      "step": 88954
    },
    {
      "epoch": 1.4261245727539063e-05,
      "step": 88954,
      "training_step_time": 7.601870536804199
    },
    {
      "epoch": 1.4262199401855469e-05,
      "model_forward_time": 1.755176067352295,
      "step": 88955
    },
    {
      "epoch": 1.4262199401855469e-05,
      "step": 88955,
      "training_step_time": 7.779684543609619
    },
    {
      "epoch": 1.4263153076171875e-05,
      "model_forward_time": 1.7670249938964844,
      "step": 88956
    },
    {
      "epoch": 1.4263153076171875e-05,
      "step": 88956,
      "training_step_time": 8.26832103729248
    },
    {
      "epoch": 1.4264106750488282e-05,
      "model_forward_time": 1.77506685256958,
      "step": 88957
    },
    {
      "epoch": 1.4264106750488282e-05,
      "step": 88957,
      "training_step_time": 8.28039813041687
    },
    {
      "epoch": 1.4265060424804688e-05,
      "model_forward_time": 1.7685043811798096,
      "step": 88958
    },
    {
      "epoch": 1.4265060424804688e-05,
      "step": 88958,
      "training_step_time": 7.757768630981445
    },
    {
      "epoch": 1.4266014099121093e-05,
      "model_forward_time": 1.7578418254852295,
      "step": 88959
    },
    {
      "epoch": 1.4266014099121093e-05,
      "step": 88959,
      "training_step_time": 9.536789417266846
    },
    {
      "epoch": 1.42669677734375e-05,
      "grad_norm": 0.09343063831329346,
      "learning_rate": 3.2959397831338446e-07,
      "loss": 0.0559,
      "step": 88960
    },
    {
      "epoch": 1.42669677734375e-05,
      "model_forward_time": 1.7557144165039062,
      "step": 88960
    },
    {
      "dynamics_loss_avg": 0.05530112069100142,
      "epoch": 1.42669677734375e-05,
      "step": 88960
    },
    {
      "action_loss_avg": 0.0019486897333990782,
      "epoch": 1.42669677734375e-05,
      "step": 88960
    },
    {
      "epoch": 1.42669677734375e-05,
      "step": 88960,
      "training_step_time": 7.781182765960693
    },
    {
      "epoch": 1.4267921447753907e-05,
      "model_forward_time": 2.4305331707000732,
      "step": 88961
    },
    {
      "epoch": 1.4267921447753907e-05,
      "step": 88961,
      "training_step_time": 9.88562822341919
    },
    {
      "epoch": 1.4268875122070312e-05,
      "model_forward_time": 1.758035659790039,
      "step": 88962
    },
    {
      "epoch": 1.4268875122070312e-05,
      "step": 88962,
      "training_step_time": 7.796313762664795
    },
    {
      "epoch": 1.4269828796386718e-05,
      "model_forward_time": 1.7629363536834717,
      "step": 88963
    },
    {
      "epoch": 1.4269828796386718e-05,
      "step": 88963,
      "training_step_time": 8.075013637542725
    },
    {
      "epoch": 1.4270782470703125e-05,
      "model_forward_time": 1.7671339511871338,
      "step": 88964
    },
    {
      "epoch": 1.4270782470703125e-05,
      "step": 88964,
      "training_step_time": 7.770623683929443
    },
    {
      "epoch": 1.4271736145019531e-05,
      "model_forward_time": 1.7718122005462646,
      "step": 88965
    },
    {
      "epoch": 1.4271736145019531e-05,
      "step": 88965,
      "training_step_time": 7.538398504257202
    },
    {
      "epoch": 1.4272689819335937e-05,
      "model_forward_time": 1.7643866539001465,
      "step": 88966
    },
    {
      "epoch": 1.4272689819335937e-05,
      "step": 88966,
      "training_step_time": 7.811935186386108
    },
    {
      "epoch": 1.4273643493652344e-05,
      "model_forward_time": 1.7669341564178467,
      "step": 88967
    },
    {
      "epoch": 1.4273643493652344e-05,
      "step": 88967,
      "training_step_time": 8.856825351715088
    },
    {
      "epoch": 1.427459716796875e-05,
      "model_forward_time": 1.764831304550171,
      "step": 88968
    },
    {
      "epoch": 1.427459716796875e-05,
      "step": 88968,
      "training_step_time": 7.8798394203186035
    },
    {
      "epoch": 1.4275550842285156e-05,
      "model_forward_time": 1.8398973941802979,
      "step": 88969
    },
    {
      "epoch": 1.4275550842285156e-05,
      "step": 88969,
      "training_step_time": 9.452951908111572
    },
    {
      "epoch": 1.4276504516601563e-05,
      "grad_norm": 0.08172484487295151,
      "learning_rate": 3.2900384470001435e-07,
      "loss": 0.0555,
      "step": 88970
    },
    {
      "epoch": 1.4276504516601563e-05,
      "model_forward_time": 2.4611270427703857,
      "step": 88970
    },
    {
      "dynamics_loss_avg": 0.05775791890919209,
      "epoch": 1.4276504516601563e-05,
      "step": 88970
    },
    {
      "action_loss_avg": 0.0016620200418401509,
      "epoch": 1.4276504516601563e-05,
      "step": 88970
    },
    {
      "epoch": 1.4276504516601563e-05,
      "step": 88970,
      "training_step_time": 9.513084411621094
    },
    {
      "epoch": 1.4277458190917969e-05,
      "model_forward_time": 1.7587883472442627,
      "step": 88971
    },
    {
      "epoch": 1.4277458190917969e-05,
      "step": 88971,
      "training_step_time": 7.793539524078369
    },
    {
      "epoch": 1.4278411865234375e-05,
      "model_forward_time": 1.7629446983337402,
      "step": 88972
    },
    {
      "epoch": 1.4278411865234375e-05,
      "step": 88972,
      "training_step_time": 8.596885681152344
    },
    {
      "epoch": 1.4279365539550782e-05,
      "model_forward_time": 1.7672581672668457,
      "step": 88973
    },
    {
      "epoch": 1.4279365539550782e-05,
      "step": 88973,
      "training_step_time": 7.744678735733032
    },
    {
      "epoch": 1.4280319213867188e-05,
      "model_forward_time": 1.7698779106140137,
      "step": 88974
    },
    {
      "epoch": 1.4280319213867188e-05,
      "step": 88974,
      "training_step_time": 7.264228582382202
    },
    {
      "epoch": 1.4281272888183593e-05,
      "model_forward_time": 1.7687370777130127,
      "step": 88975
    },
    {
      "epoch": 1.4281272888183593e-05,
      "step": 88975,
      "training_step_time": 8.319297075271606
    },
    {
      "epoch": 1.42822265625e-05,
      "model_forward_time": 1.7843270301818848,
      "step": 88976
    },
    {
      "epoch": 1.42822265625e-05,
      "step": 88976,
      "training_step_time": 8.86280107498169
    },
    {
      "epoch": 1.4283180236816407e-05,
      "model_forward_time": 1.7818455696105957,
      "step": 88977
    },
    {
      "epoch": 1.4283180236816407e-05,
      "step": 88977,
      "training_step_time": 10.458140850067139
    },
    {
      "epoch": 1.4284133911132812e-05,
      "model_forward_time": 1.7901990413665771,
      "step": 88978
    },
    {
      "epoch": 1.4284133911132812e-05,
      "step": 88978,
      "training_step_time": 8.50481390953064
    },
    {
      "epoch": 1.4285087585449218e-05,
      "model_forward_time": 1.8624391555786133,
      "step": 88979
    },
    {
      "epoch": 1.4285087585449218e-05,
      "step": 88979,
      "training_step_time": 9.373766899108887
    },
    {
      "epoch": 1.4286041259765625e-05,
      "grad_norm": 0.07599131762981415,
      "learning_rate": 3.2841422189976104e-07,
      "loss": 0.0563,
      "step": 88980
    },
    {
      "epoch": 1.4286041259765625e-05,
      "model_forward_time": 1.7835109233856201,
      "step": 88980
    },
    {
      "dynamics_loss_avg": 0.04889251235872507,
      "epoch": 1.4286041259765625e-05,
      "step": 88980
    },
    {
      "action_loss_avg": 0.0019318467704579235,
      "epoch": 1.4286041259765625e-05,
      "step": 88980
    },
    {
      "epoch": 1.4286041259765625e-05,
      "step": 88980,
      "training_step_time": 7.822112083435059
    },
    {
      "epoch": 1.4286994934082031e-05,
      "model_forward_time": 1.7892963886260986,
      "step": 88981
    },
    {
      "epoch": 1.4286994934082031e-05,
      "step": 88981,
      "training_step_time": 7.645851135253906
    },
    {
      "epoch": 1.4287948608398437e-05,
      "model_forward_time": 1.7977643013000488,
      "step": 88982
    },
    {
      "epoch": 1.4287948608398437e-05,
      "step": 88982,
      "training_step_time": 7.797409772872925
    },
    {
      "epoch": 1.4288902282714844e-05,
      "model_forward_time": 1.7919337749481201,
      "step": 88983
    },
    {
      "epoch": 1.4288902282714844e-05,
      "step": 88983,
      "training_step_time": 7.859605550765991
    },
    {
      "epoch": 1.428985595703125e-05,
      "model_forward_time": 1.794538974761963,
      "step": 88984
    },
    {
      "epoch": 1.428985595703125e-05,
      "step": 88984,
      "training_step_time": 7.738559007644653
    },
    {
      "epoch": 1.4290809631347656e-05,
      "model_forward_time": 1.7996306419372559,
      "step": 88985
    },
    {
      "epoch": 1.4290809631347656e-05,
      "step": 88985,
      "training_step_time": 9.410194158554077
    },
    {
      "epoch": 1.4291763305664063e-05,
      "model_forward_time": 1.7866520881652832,
      "step": 88986
    },
    {
      "epoch": 1.4291763305664063e-05,
      "step": 88986,
      "training_step_time": 11.195891618728638
    },
    {
      "epoch": 1.4292716979980469e-05,
      "model_forward_time": 1.79518461227417,
      "step": 88987
    },
    {
      "epoch": 1.4292716979980469e-05,
      "step": 88987,
      "training_step_time": 9.7082359790802
    },
    {
      "epoch": 1.4293670654296875e-05,
      "model_forward_time": 1.8636870384216309,
      "step": 88988
    },
    {
      "epoch": 1.4293670654296875e-05,
      "step": 88988,
      "training_step_time": 7.945525407791138
    },
    {
      "epoch": 1.4294624328613282e-05,
      "model_forward_time": 2.8687827587127686,
      "step": 88989
    },
    {
      "epoch": 1.4294624328613282e-05,
      "step": 88989,
      "training_step_time": 8.805468082427979
    },
    {
      "epoch": 1.4295578002929688e-05,
      "grad_norm": 0.08154313266277313,
      "learning_rate": 3.278251099771046e-07,
      "loss": 0.0548,
      "step": 88990
    },
    {
      "epoch": 1.4295578002929688e-05,
      "model_forward_time": 1.799363136291504,
      "step": 88990
    },
    {
      "dynamics_loss_avg": 0.0593626881018281,
      "epoch": 1.4295578002929688e-05,
      "step": 88990
    },
    {
      "action_loss_avg": 0.002500467753270641,
      "epoch": 1.4295578002929688e-05,
      "step": 88990
    },
    {
      "epoch": 1.4295578002929688e-05,
      "step": 88990,
      "training_step_time": 8.307687282562256
    },
    {
      "epoch": 1.4296531677246093e-05,
      "model_forward_time": 1.7841238975524902,
      "step": 88991
    },
    {
      "epoch": 1.4296531677246093e-05,
      "step": 88991,
      "training_step_time": 7.767684459686279
    },
    {
      "epoch": 1.42974853515625e-05,
      "model_forward_time": 1.7927193641662598,
      "step": 88992
    },
    {
      "epoch": 1.42974853515625e-05,
      "step": 88992,
      "training_step_time": 7.661165714263916
    },
    {
      "epoch": 1.4298439025878907e-05,
      "model_forward_time": 1.7958240509033203,
      "step": 88993
    },
    {
      "epoch": 1.4298439025878907e-05,
      "step": 88993,
      "training_step_time": 9.101147413253784
    },
    {
      "epoch": 1.4299392700195312e-05,
      "model_forward_time": 1.7907359600067139,
      "step": 88994
    },
    {
      "epoch": 1.4299392700195312e-05,
      "step": 88994,
      "training_step_time": 8.60730266571045
    },
    {
      "epoch": 1.4300346374511718e-05,
      "model_forward_time": 1.790069341659546,
      "step": 88995
    },
    {
      "epoch": 1.4300346374511718e-05,
      "step": 88995,
      "training_step_time": 8.8666672706604
    },
    {
      "epoch": 1.4301300048828125e-05,
      "model_forward_time": 1.794084072113037,
      "step": 88996
    },
    {
      "epoch": 1.4301300048828125e-05,
      "step": 88996,
      "training_step_time": 8.289987564086914
    },
    {
      "epoch": 1.4302253723144531e-05,
      "model_forward_time": 1.8535149097442627,
      "step": 88997
    },
    {
      "epoch": 1.4302253723144531e-05,
      "step": 88997,
      "training_step_time": 7.674445152282715
    },
    {
      "epoch": 1.4303207397460937e-05,
      "model_forward_time": 3.019078493118286,
      "step": 88998
    },
    {
      "epoch": 1.4303207397460937e-05,
      "step": 88998,
      "training_step_time": 9.138719320297241
    },
    {
      "epoch": 1.4304161071777344e-05,
      "model_forward_time": 1.7951245307922363,
      "step": 88999
    },
    {
      "epoch": 1.4304161071777344e-05,
      "step": 88999,
      "training_step_time": 8.477811813354492
    },
    {
      "epoch": 1.430511474609375e-05,
      "grad_norm": 0.07822861522436142,
      "learning_rate": 3.272365089964691e-07,
      "loss": 0.0554,
      "step": 89000
    },
    {
      "epoch": 1.430511474609375e-05,
      "model_forward_time": 1.7878649234771729,
      "step": 89000
    },
    {
      "dynamics_loss_avg": 0.06164215030148625,
      "epoch": 1.430511474609375e-05,
      "step": 89000
    },
    {
      "action_loss_avg": 0.0009326794155640527,
      "epoch": 1.430511474609375e-05,
      "step": 89000
    },
    {
      "epoch": 1.430511474609375e-05,
      "step": 89000,
      "training_step_time": 6.703962087631226
    },
    {
      "epoch": 1.4306068420410156e-05,
      "model_forward_time": 1.8036444187164307,
      "step": 89001
    },
    {
      "epoch": 1.4306068420410156e-05,
      "step": 89001,
      "training_step_time": 8.790320634841919
    },
    {
      "epoch": 1.4307022094726563e-05,
      "model_forward_time": 1.807189702987671,
      "step": 89002
    },
    {
      "epoch": 1.4307022094726563e-05,
      "step": 89002,
      "training_step_time": 8.428186416625977
    },
    {
      "epoch": 1.4307975769042969e-05,
      "model_forward_time": 1.807985782623291,
      "step": 89003
    },
    {
      "epoch": 1.4307975769042969e-05,
      "step": 89003,
      "training_step_time": 10.52461552619934
    },
    {
      "epoch": 1.4308929443359375e-05,
      "model_forward_time": 1.813711166381836,
      "step": 89004
    },
    {
      "epoch": 1.4308929443359375e-05,
      "step": 89004,
      "training_step_time": 9.102695941925049
    },
    {
      "epoch": 1.4309883117675782e-05,
      "model_forward_time": 1.8064024448394775,
      "step": 89005
    },
    {
      "epoch": 1.4309883117675782e-05,
      "step": 89005,
      "training_step_time": 8.704826354980469
    },
    {
      "epoch": 1.4310836791992188e-05,
      "model_forward_time": 1.8871335983276367,
      "step": 89006
    },
    {
      "epoch": 1.4310836791992188e-05,
      "step": 89006,
      "training_step_time": 10.006661415100098
    },
    {
      "epoch": 1.4311790466308593e-05,
      "model_forward_time": 1.7971515655517578,
      "step": 89007
    },
    {
      "epoch": 1.4311790466308593e-05,
      "step": 89007,
      "training_step_time": 9.480053424835205
    },
    {
      "epoch": 1.4312744140625e-05,
      "model_forward_time": 1.8076624870300293,
      "step": 89008
    },
    {
      "epoch": 1.4312744140625e-05,
      "step": 89008,
      "training_step_time": 9.504430532455444
    },
    {
      "epoch": 1.4313697814941407e-05,
      "model_forward_time": 1.7608425617218018,
      "step": 89009
    },
    {
      "epoch": 1.4313697814941407e-05,
      "step": 89009,
      "training_step_time": 7.675694942474365
    },
    {
      "epoch": 1.4314651489257812e-05,
      "grad_norm": 0.1176551952958107,
      "learning_rate": 3.266484190222247e-07,
      "loss": 0.0561,
      "step": 89010
    },
    {
      "epoch": 1.4314651489257812e-05,
      "model_forward_time": 1.7960882186889648,
      "step": 89010
    },
    {
      "dynamics_loss_avg": 0.06054887799546123,
      "epoch": 1.4314651489257812e-05,
      "step": 89010
    },
    {
      "action_loss_avg": 0.0014816898299613967,
      "epoch": 1.4314651489257812e-05,
      "step": 89010
    },
    {
      "epoch": 1.4314651489257812e-05,
      "step": 89010,
      "training_step_time": 7.695531368255615
    },
    {
      "epoch": 1.4315605163574218e-05,
      "model_forward_time": 1.7889885902404785,
      "step": 89011
    },
    {
      "epoch": 1.4315605163574218e-05,
      "step": 89011,
      "training_step_time": 7.698755264282227
    },
    {
      "epoch": 1.4316558837890625e-05,
      "model_forward_time": 1.7884292602539062,
      "step": 89012
    },
    {
      "epoch": 1.4316558837890625e-05,
      "step": 89012,
      "training_step_time": 7.7501959800720215
    },
    {
      "epoch": 1.4317512512207031e-05,
      "model_forward_time": 1.82171630859375,
      "step": 89013
    },
    {
      "epoch": 1.4317512512207031e-05,
      "step": 89013,
      "training_step_time": 8.384735107421875
    },
    {
      "epoch": 1.4318466186523437e-05,
      "model_forward_time": 1.7861244678497314,
      "step": 89014
    },
    {
      "epoch": 1.4318466186523437e-05,
      "step": 89014,
      "training_step_time": 9.180998802185059
    },
    {
      "epoch": 1.4319419860839844e-05,
      "model_forward_time": 1.8543174266815186,
      "step": 89015
    },
    {
      "epoch": 1.4319419860839844e-05,
      "step": 89015,
      "training_step_time": 8.713175535202026
    },
    {
      "epoch": 1.432037353515625e-05,
      "model_forward_time": 2.873497247695923,
      "step": 89016
    },
    {
      "epoch": 1.432037353515625e-05,
      "step": 89016,
      "training_step_time": 8.354022026062012
    },
    {
      "epoch": 1.4321327209472656e-05,
      "model_forward_time": 1.786705493927002,
      "step": 89017
    },
    {
      "epoch": 1.4321327209472656e-05,
      "step": 89017,
      "training_step_time": 7.815002679824829
    },
    {
      "epoch": 1.4322280883789063e-05,
      "model_forward_time": 1.7879626750946045,
      "step": 89018
    },
    {
      "epoch": 1.4322280883789063e-05,
      "step": 89018,
      "training_step_time": 7.806544065475464
    },
    {
      "epoch": 1.4323234558105469e-05,
      "model_forward_time": 1.7945220470428467,
      "step": 89019
    },
    {
      "epoch": 1.4323234558105469e-05,
      "step": 89019,
      "training_step_time": 7.534383773803711
    },
    {
      "epoch": 1.4324188232421875e-05,
      "grad_norm": 0.07398297637701035,
      "learning_rate": 3.260608401186821e-07,
      "loss": 0.0546,
      "step": 89020
    },
    {
      "epoch": 1.4324188232421875e-05,
      "model_forward_time": 1.7893083095550537,
      "step": 89020
    },
    {
      "dynamics_loss_avg": 0.06634179204702377,
      "epoch": 1.4324188232421875e-05,
      "step": 89020
    },
    {
      "action_loss_avg": 0.0012237386545166373,
      "epoch": 1.4324188232421875e-05,
      "step": 89020
    },
    {
      "epoch": 1.4324188232421875e-05,
      "step": 89020,
      "training_step_time": 7.8645734786987305
    },
    {
      "epoch": 1.4325141906738282e-05,
      "model_forward_time": 1.7801330089569092,
      "step": 89021
    },
    {
      "epoch": 1.4325141906738282e-05,
      "step": 89021,
      "training_step_time": 8.809517860412598
    },
    {
      "epoch": 1.4326095581054688e-05,
      "model_forward_time": 1.7985992431640625,
      "step": 89022
    },
    {
      "epoch": 1.4326095581054688e-05,
      "step": 89022,
      "training_step_time": 7.795552968978882
    },
    {
      "epoch": 1.4327049255371093e-05,
      "model_forward_time": 1.7795865535736084,
      "step": 89023
    },
    {
      "epoch": 1.4327049255371093e-05,
      "step": 89023,
      "training_step_time": 7.898485898971558
    },
    {
      "epoch": 1.43280029296875e-05,
      "model_forward_time": 1.850069522857666,
      "step": 89024
    },
    {
      "epoch": 1.43280029296875e-05,
      "step": 89024,
      "training_step_time": 7.832842111587524
    },
    {
      "epoch": 1.4328956604003907e-05,
      "model_forward_time": 2.888256788253784,
      "step": 89025
    },
    {
      "epoch": 1.4328956604003907e-05,
      "step": 89025,
      "training_step_time": 10.399451732635498
    },
    {
      "epoch": 1.4329910278320312e-05,
      "model_forward_time": 1.776214838027954,
      "step": 89026
    },
    {
      "epoch": 1.4329910278320312e-05,
      "step": 89026,
      "training_step_time": 7.833843469619751
    },
    {
      "epoch": 1.4330863952636718e-05,
      "model_forward_time": 1.7856175899505615,
      "step": 89027
    },
    {
      "epoch": 1.4330863952636718e-05,
      "step": 89027,
      "training_step_time": 8.679455757141113
    },
    {
      "epoch": 1.4331817626953125e-05,
      "model_forward_time": 1.8008079528808594,
      "step": 89028
    },
    {
      "epoch": 1.4331817626953125e-05,
      "step": 89028,
      "training_step_time": 7.5059263706207275
    },
    {
      "epoch": 1.4332771301269531e-05,
      "model_forward_time": 1.7838478088378906,
      "step": 89029
    },
    {
      "epoch": 1.4332771301269531e-05,
      "step": 89029,
      "training_step_time": 7.754430532455444
    },
    {
      "epoch": 1.4333724975585937e-05,
      "grad_norm": 0.1329788714647293,
      "learning_rate": 3.2547377235009893e-07,
      "loss": 0.055,
      "step": 89030
    },
    {
      "epoch": 1.4333724975585937e-05,
      "model_forward_time": 1.793346643447876,
      "step": 89030
    },
    {
      "dynamics_loss_avg": 0.052498935163021086,
      "epoch": 1.4333724975585937e-05,
      "step": 89030
    },
    {
      "action_loss_avg": 0.0012282691430300474,
      "epoch": 1.4333724975585937e-05,
      "step": 89030
    },
    {
      "epoch": 1.4333724975585937e-05,
      "step": 89030,
      "training_step_time": 7.790834426879883
    },
    {
      "epoch": 1.4334678649902344e-05,
      "model_forward_time": 1.7889349460601807,
      "step": 89031
    },
    {
      "epoch": 1.4334678649902344e-05,
      "step": 89031,
      "training_step_time": 8.317527294158936
    },
    {
      "epoch": 1.433563232421875e-05,
      "model_forward_time": 1.7762532234191895,
      "step": 89032
    },
    {
      "epoch": 1.433563232421875e-05,
      "step": 89032,
      "training_step_time": 8.604506015777588
    },
    {
      "epoch": 1.4336585998535156e-05,
      "model_forward_time": 1.8351106643676758,
      "step": 89033
    },
    {
      "epoch": 1.4336585998535156e-05,
      "step": 89033,
      "training_step_time": 9.637623310089111
    },
    {
      "epoch": 1.4337539672851563e-05,
      "model_forward_time": 2.841015100479126,
      "step": 89034
    },
    {
      "epoch": 1.4337539672851563e-05,
      "step": 89034,
      "training_step_time": 7.674211740493774
    },
    {
      "epoch": 1.4338493347167969e-05,
      "model_forward_time": 1.8013341426849365,
      "step": 89035
    },
    {
      "epoch": 1.4338493347167969e-05,
      "step": 89035,
      "training_step_time": 7.777599811553955
    },
    {
      "epoch": 1.4339447021484375e-05,
      "model_forward_time": 1.790330410003662,
      "step": 89036
    },
    {
      "epoch": 1.4339447021484375e-05,
      "step": 89036,
      "training_step_time": 8.435561895370483
    },
    {
      "epoch": 1.4340400695800782e-05,
      "model_forward_time": 1.7862606048583984,
      "step": 89037
    },
    {
      "epoch": 1.4340400695800782e-05,
      "step": 89037,
      "training_step_time": 7.648571252822876
    },
    {
      "epoch": 1.4341354370117188e-05,
      "model_forward_time": 1.7991101741790771,
      "step": 89038
    },
    {
      "epoch": 1.4341354370117188e-05,
      "step": 89038,
      "training_step_time": 7.778701543807983
    },
    {
      "epoch": 1.4342308044433593e-05,
      "model_forward_time": 1.7949457168579102,
      "step": 89039
    },
    {
      "epoch": 1.4342308044433593e-05,
      "step": 89039,
      "training_step_time": 9.321727752685547
    },
    {
      "epoch": 1.434326171875e-05,
      "grad_norm": 0.08296577632427216,
      "learning_rate": 3.2488721578067596e-07,
      "loss": 0.0564,
      "step": 89040
    },
    {
      "epoch": 1.434326171875e-05,
      "model_forward_time": 1.8141145706176758,
      "step": 89040
    },
    {
      "dynamics_loss_avg": 0.05095609985291958,
      "epoch": 1.434326171875e-05,
      "step": 89040
    },
    {
      "action_loss_avg": 0.0018985791713930667,
      "epoch": 1.434326171875e-05,
      "step": 89040
    },
    {
      "epoch": 1.434326171875e-05,
      "step": 89040,
      "training_step_time": 9.52573823928833
    },
    {
      "epoch": 1.4344215393066407e-05,
      "model_forward_time": 1.8130130767822266,
      "step": 89041
    },
    {
      "epoch": 1.4344215393066407e-05,
      "step": 89041,
      "training_step_time": 7.756394386291504
    },
    {
      "epoch": 1.4345169067382812e-05,
      "model_forward_time": 1.8149938583374023,
      "step": 89042
    },
    {
      "epoch": 1.4345169067382812e-05,
      "step": 89042,
      "training_step_time": 8.381405591964722
    },
    {
      "epoch": 1.4346122741699218e-05,
      "model_forward_time": 2.9197192192077637,
      "step": 89043
    },
    {
      "epoch": 1.4346122741699218e-05,
      "step": 89043,
      "training_step_time": 7.737457275390625
    },
    {
      "epoch": 1.4347076416015625e-05,
      "model_forward_time": 1.8052427768707275,
      "step": 89044
    },
    {
      "epoch": 1.4347076416015625e-05,
      "step": 89044,
      "training_step_time": 7.7101287841796875
    },
    {
      "epoch": 1.4348030090332031e-05,
      "model_forward_time": 1.8098649978637695,
      "step": 89045
    },
    {
      "epoch": 1.4348030090332031e-05,
      "step": 89045,
      "training_step_time": 7.572752475738525
    },
    {
      "epoch": 1.4348983764648437e-05,
      "model_forward_time": 1.8124353885650635,
      "step": 89046
    },
    {
      "epoch": 1.4348983764648437e-05,
      "step": 89046,
      "training_step_time": 7.713327884674072
    },
    {
      "epoch": 1.4349937438964844e-05,
      "model_forward_time": 1.8115191459655762,
      "step": 89047
    },
    {
      "epoch": 1.4349937438964844e-05,
      "step": 89047,
      "training_step_time": 8.623499631881714
    },
    {
      "epoch": 1.435089111328125e-05,
      "model_forward_time": 1.8014769554138184,
      "step": 89048
    },
    {
      "epoch": 1.435089111328125e-05,
      "step": 89048,
      "training_step_time": 8.366912841796875
    },
    {
      "epoch": 1.4351844787597656e-05,
      "model_forward_time": 1.8080649375915527,
      "step": 89049
    },
    {
      "epoch": 1.4351844787597656e-05,
      "step": 89049,
      "training_step_time": 8.503427267074585
    },
    {
      "epoch": 1.4352798461914063e-05,
      "grad_norm": 0.0907038003206253,
      "learning_rate": 3.243011704745591e-07,
      "loss": 0.0557,
      "step": 89050
    },
    {
      "epoch": 1.4352798461914063e-05,
      "model_forward_time": 1.7862977981567383,
      "step": 89050
    },
    {
      "dynamics_loss_avg": 0.06486786492168903,
      "epoch": 1.4352798461914063e-05,
      "step": 89050
    },
    {
      "action_loss_avg": 0.0019687850552145393,
      "epoch": 1.4352798461914063e-05,
      "step": 89050
    },
    {
      "epoch": 1.4352798461914063e-05,
      "step": 89050,
      "training_step_time": 8.516548156738281
    },
    {
      "epoch": 1.4353752136230469e-05,
      "model_forward_time": 1.7932219505310059,
      "step": 89051
    },
    {
      "epoch": 1.4353752136230469e-05,
      "step": 89051,
      "training_step_time": 8.88520359992981
    },
    {
      "epoch": 1.4354705810546875e-05,
      "model_forward_time": 2.8923399448394775,
      "step": 89052
    },
    {
      "epoch": 1.4354705810546875e-05,
      "step": 89052,
      "training_step_time": 8.72626280784607
    },
    {
      "epoch": 1.4355659484863282e-05,
      "model_forward_time": 1.7922110557556152,
      "step": 89053
    },
    {
      "epoch": 1.4355659484863282e-05,
      "step": 89053,
      "training_step_time": 9.32150912284851
    },
    {
      "epoch": 1.4356613159179688e-05,
      "model_forward_time": 1.790053129196167,
      "step": 89054
    },
    {
      "epoch": 1.4356613159179688e-05,
      "step": 89054,
      "training_step_time": 7.573539972305298
    },
    {
      "epoch": 1.4357566833496093e-05,
      "model_forward_time": 1.7946882247924805,
      "step": 89055
    },
    {
      "epoch": 1.4357566833496093e-05,
      "step": 89055,
      "training_step_time": 8.38310170173645
    },
    {
      "epoch": 1.43585205078125e-05,
      "model_forward_time": 1.789320468902588,
      "step": 89056
    },
    {
      "epoch": 1.43585205078125e-05,
      "step": 89056,
      "training_step_time": 6.775475740432739
    },
    {
      "epoch": 1.4359474182128907e-05,
      "model_forward_time": 1.7928807735443115,
      "step": 89057
    },
    {
      "epoch": 1.4359474182128907e-05,
      "step": 89057,
      "training_step_time": 7.617765665054321
    },
    {
      "epoch": 1.4360427856445312e-05,
      "model_forward_time": 1.7807691097259521,
      "step": 89058
    },
    {
      "epoch": 1.4360427856445312e-05,
      "step": 89058,
      "training_step_time": 8.586474895477295
    },
    {
      "epoch": 1.4361381530761718e-05,
      "model_forward_time": 1.7972567081451416,
      "step": 89059
    },
    {
      "epoch": 1.4361381530761718e-05,
      "step": 89059,
      "training_step_time": 8.355013370513916
    },
    {
      "epoch": 1.4362335205078125e-05,
      "grad_norm": 0.0842151865363121,
      "learning_rate": 3.237156364958355e-07,
      "loss": 0.0558,
      "step": 89060
    },
    {
      "epoch": 1.4362335205078125e-05,
      "model_forward_time": 1.7843365669250488,
      "step": 89060
    },
    {
      "dynamics_loss_avg": 0.058141322992742064,
      "epoch": 1.4362335205078125e-05,
      "step": 89060
    },
    {
      "action_loss_avg": 0.0013328281696885825,
      "epoch": 1.4362335205078125e-05,
      "step": 89060
    },
    {
      "epoch": 1.4362335205078125e-05,
      "step": 89060,
      "training_step_time": 7.943138360977173
    },
    {
      "epoch": 1.4363288879394531e-05,
      "model_forward_time": 1.8782026767730713,
      "step": 89061
    },
    {
      "epoch": 1.4363288879394531e-05,
      "step": 89061,
      "training_step_time": 8.835617065429688
    },
    {
      "epoch": 1.4364242553710937e-05,
      "model_forward_time": 2.9019644260406494,
      "step": 89062
    },
    {
      "epoch": 1.4364242553710937e-05,
      "step": 89062,
      "training_step_time": 8.633125066757202
    },
    {
      "epoch": 1.4365196228027344e-05,
      "model_forward_time": 1.7898740768432617,
      "step": 89063
    },
    {
      "epoch": 1.4365196228027344e-05,
      "step": 89063,
      "training_step_time": 7.882870435714722
    },
    {
      "epoch": 1.436614990234375e-05,
      "model_forward_time": 1.809554100036621,
      "step": 89064
    },
    {
      "epoch": 1.436614990234375e-05,
      "step": 89064,
      "training_step_time": 7.81760573387146
    },
    {
      "epoch": 1.4367103576660156e-05,
      "model_forward_time": 1.7886779308319092,
      "step": 89065
    },
    {
      "epoch": 1.4367103576660156e-05,
      "step": 89065,
      "training_step_time": 7.557228326797485
    },
    {
      "epoch": 1.4368057250976563e-05,
      "model_forward_time": 1.8020696640014648,
      "step": 89066
    },
    {
      "epoch": 1.4368057250976563e-05,
      "step": 89066,
      "training_step_time": 7.914752244949341
    },
    {
      "epoch": 1.4369010925292969e-05,
      "model_forward_time": 1.7978293895721436,
      "step": 89067
    },
    {
      "epoch": 1.4369010925292969e-05,
      "step": 89067,
      "training_step_time": 8.324327945709229
    },
    {
      "epoch": 1.4369964599609375e-05,
      "model_forward_time": 1.7990343570709229,
      "step": 89068
    },
    {
      "epoch": 1.4369964599609375e-05,
      "step": 89068,
      "training_step_time": 7.908526420593262
    },
    {
      "epoch": 1.4370918273925782e-05,
      "model_forward_time": 1.7838878631591797,
      "step": 89069
    },
    {
      "epoch": 1.4370918273925782e-05,
      "step": 89069,
      "training_step_time": 8.795548915863037
    },
    {
      "epoch": 1.4371871948242188e-05,
      "grad_norm": 0.1887986958026886,
      "learning_rate": 3.231306139085394e-07,
      "loss": 0.0553,
      "step": 89070
    },
    {
      "epoch": 1.4371871948242188e-05,
      "model_forward_time": 1.8676705360412598,
      "step": 89070
    },
    {
      "dynamics_loss_avg": 0.046730687655508515,
      "epoch": 1.4371871948242188e-05,
      "step": 89070
    },
    {
      "action_loss_avg": 0.0012527066573966295,
      "epoch": 1.4371871948242188e-05,
      "step": 89070
    },
    {
      "epoch": 1.4371871948242188e-05,
      "step": 89070,
      "training_step_time": 7.794952869415283
    },
    {
      "epoch": 1.4372825622558593e-05,
      "model_forward_time": 2.9062986373901367,
      "step": 89071
    },
    {
      "epoch": 1.4372825622558593e-05,
      "step": 89071,
      "training_step_time": 7.840206146240234
    },
    {
      "epoch": 1.4373779296875e-05,
      "model_forward_time": 1.8065409660339355,
      "step": 89072
    },
    {
      "epoch": 1.4373779296875e-05,
      "step": 89072,
      "training_step_time": 7.867583513259888
    },
    {
      "epoch": 1.4374732971191407e-05,
      "model_forward_time": 1.8133454322814941,
      "step": 89073
    },
    {
      "epoch": 1.4374732971191407e-05,
      "step": 89073,
      "training_step_time": 7.717769384384155
    },
    {
      "epoch": 1.4375686645507812e-05,
      "model_forward_time": 1.792262077331543,
      "step": 89074
    },
    {
      "epoch": 1.4375686645507812e-05,
      "step": 89074,
      "training_step_time": 7.462510585784912
    },
    {
      "epoch": 1.4376640319824218e-05,
      "model_forward_time": 1.802600622177124,
      "step": 89075
    },
    {
      "epoch": 1.4376640319824218e-05,
      "step": 89075,
      "training_step_time": 7.792210102081299
    },
    {
      "epoch": 1.4377593994140625e-05,
      "model_forward_time": 1.8010075092315674,
      "step": 89076
    },
    {
      "epoch": 1.4377593994140625e-05,
      "step": 89076,
      "training_step_time": 7.723322629928589
    },
    {
      "epoch": 1.4378547668457031e-05,
      "model_forward_time": 1.7968189716339111,
      "step": 89077
    },
    {
      "epoch": 1.4378547668457031e-05,
      "step": 89077,
      "training_step_time": 8.880001544952393
    },
    {
      "epoch": 1.4379501342773437e-05,
      "model_forward_time": 1.8017292022705078,
      "step": 89078
    },
    {
      "epoch": 1.4379501342773437e-05,
      "step": 89078,
      "training_step_time": 7.916994094848633
    },
    {
      "epoch": 1.4380455017089844e-05,
      "model_forward_time": 1.8522248268127441,
      "step": 89079
    },
    {
      "epoch": 1.4380455017089844e-05,
      "step": 89079,
      "training_step_time": 8.548668622970581
    },
    {
      "epoch": 1.438140869140625e-05,
      "grad_norm": 0.07382038235664368,
      "learning_rate": 3.22546102776648e-07,
      "loss": 0.0559,
      "step": 89080
    },
    {
      "epoch": 1.438140869140625e-05,
      "model_forward_time": 2.897508382797241,
      "step": 89080
    },
    {
      "dynamics_loss_avg": 0.07167639303952456,
      "epoch": 1.438140869140625e-05,
      "step": 89080
    },
    {
      "action_loss_avg": 0.0031623699935153125,
      "epoch": 1.438140869140625e-05,
      "step": 89080
    },
    {
      "epoch": 1.438140869140625e-05,
      "step": 89080,
      "training_step_time": 7.808536529541016
    },
    {
      "epoch": 1.4382362365722656e-05,
      "model_forward_time": 1.7811219692230225,
      "step": 89081
    },
    {
      "epoch": 1.4382362365722656e-05,
      "step": 89081,
      "training_step_time": 7.817188262939453
    },
    {
      "epoch": 1.4383316040039063e-05,
      "model_forward_time": 1.8083999156951904,
      "step": 89082
    },
    {
      "epoch": 1.4383316040039063e-05,
      "step": 89082,
      "training_step_time": 7.674892902374268
    },
    {
      "epoch": 1.4384269714355469e-05,
      "model_forward_time": 1.8277907371520996,
      "step": 89083
    },
    {
      "epoch": 1.4384269714355469e-05,
      "step": 89083,
      "training_step_time": 7.53752326965332
    },
    {
      "epoch": 1.4385223388671875e-05,
      "model_forward_time": 1.7991876602172852,
      "step": 89084
    },
    {
      "epoch": 1.4385223388671875e-05,
      "step": 89084,
      "training_step_time": 8.375218391418457
    },
    {
      "epoch": 1.4386177062988282e-05,
      "model_forward_time": 1.807267189025879,
      "step": 89085
    },
    {
      "epoch": 1.4386177062988282e-05,
      "step": 89085,
      "training_step_time": 7.712969779968262
    },
    {
      "epoch": 1.4387130737304688e-05,
      "model_forward_time": 1.7961947917938232,
      "step": 89086
    },
    {
      "epoch": 1.4387130737304688e-05,
      "step": 89086,
      "training_step_time": 8.483043193817139
    },
    {
      "epoch": 1.4388084411621094e-05,
      "model_forward_time": 1.8029966354370117,
      "step": 89087
    },
    {
      "epoch": 1.4388084411621094e-05,
      "step": 89087,
      "training_step_time": 8.439380645751953
    },
    {
      "epoch": 1.43890380859375e-05,
      "model_forward_time": 1.8147358894348145,
      "step": 89088
    },
    {
      "epoch": 1.43890380859375e-05,
      "step": 89088,
      "training_step_time": 9.203550100326538
    },
    {
      "epoch": 1.4389991760253907e-05,
      "model_forward_time": 2.8685410022735596,
      "step": 89089
    },
    {
      "epoch": 1.4389991760253907e-05,
      "step": 89089,
      "training_step_time": 7.676341772079468
    },
    {
      "epoch": 1.4390945434570312e-05,
      "grad_norm": 0.08461082726716995,
      "learning_rate": 3.2196210316408183e-07,
      "loss": 0.0548,
      "step": 89090
    },
    {
      "epoch": 1.4390945434570312e-05,
      "model_forward_time": 1.8008687496185303,
      "step": 89090
    },
    {
      "dynamics_loss_avg": 0.06484734732657671,
      "epoch": 1.4390945434570312e-05,
      "step": 89090
    },
    {
      "action_loss_avg": 0.0021074604883324355,
      "epoch": 1.4390945434570312e-05,
      "step": 89090
    },
    {
      "epoch": 1.4390945434570312e-05,
      "step": 89090,
      "training_step_time": 9.069899797439575
    },
    {
      "epoch": 1.4391899108886718e-05,
      "model_forward_time": 1.8025243282318115,
      "step": 89091
    },
    {
      "epoch": 1.4391899108886718e-05,
      "step": 89091,
      "training_step_time": 8.551219701766968
    },
    {
      "epoch": 1.4392852783203126e-05,
      "model_forward_time": 1.807892084121704,
      "step": 89092
    },
    {
      "epoch": 1.4392852783203126e-05,
      "step": 89092,
      "training_step_time": 7.684561252593994
    },
    {
      "epoch": 1.4393806457519531e-05,
      "model_forward_time": 1.802170753479004,
      "step": 89093
    },
    {
      "epoch": 1.4393806457519531e-05,
      "step": 89093,
      "training_step_time": 7.5908801555633545
    },
    {
      "epoch": 1.4394760131835937e-05,
      "model_forward_time": 1.8081560134887695,
      "step": 89094
    },
    {
      "epoch": 1.4394760131835937e-05,
      "step": 89094,
      "training_step_time": 7.690639972686768
    },
    {
      "epoch": 1.4395713806152344e-05,
      "model_forward_time": 1.7828850746154785,
      "step": 89095
    },
    {
      "epoch": 1.4395713806152344e-05,
      "step": 89095,
      "training_step_time": 7.702329874038696
    },
    {
      "epoch": 1.439666748046875e-05,
      "model_forward_time": 1.8027725219726562,
      "step": 89096
    },
    {
      "epoch": 1.439666748046875e-05,
      "step": 89096,
      "training_step_time": 8.794142961502075
    },
    {
      "epoch": 1.4397621154785156e-05,
      "model_forward_time": 1.7875521183013916,
      "step": 89097
    },
    {
      "epoch": 1.4397621154785156e-05,
      "step": 89097,
      "training_step_time": 8.464861154556274
    },
    {
      "epoch": 1.4398574829101563e-05,
      "model_forward_time": 2.945183515548706,
      "step": 89098
    },
    {
      "epoch": 1.4398574829101563e-05,
      "step": 89098,
      "training_step_time": 7.730756044387817
    },
    {
      "epoch": 1.4399528503417969e-05,
      "model_forward_time": 1.7954955101013184,
      "step": 89099
    },
    {
      "epoch": 1.4399528503417969e-05,
      "step": 89099,
      "training_step_time": 7.783128261566162
    },
    {
      "epoch": 1.4400482177734375e-05,
      "grad_norm": 0.07608415931463242,
      "learning_rate": 3.213786151347065e-07,
      "loss": 0.0551,
      "step": 89100
    },
    {
      "epoch": 1.4400482177734375e-05,
      "model_forward_time": 1.8133528232574463,
      "step": 89100
    },
    {
      "dynamics_loss_avg": 0.049764485098421574,
      "epoch": 1.4400482177734375e-05,
      "step": 89100
    },
    {
      "action_loss_avg": 0.0009672878324636259,
      "epoch": 1.4400482177734375e-05,
      "step": 89100
    },
    {
      "epoch": 1.4400482177734375e-05,
      "step": 89100,
      "training_step_time": 8.643512487411499
    },
    {
      "epoch": 1.4401435852050782e-05,
      "model_forward_time": 1.80643630027771,
      "step": 89101
    },
    {
      "epoch": 1.4401435852050782e-05,
      "step": 89101,
      "training_step_time": 7.650259017944336
    },
    {
      "epoch": 1.4402389526367188e-05,
      "model_forward_time": 1.8077256679534912,
      "step": 89102
    },
    {
      "epoch": 1.4402389526367188e-05,
      "step": 89102,
      "training_step_time": 7.386260509490967
    },
    {
      "epoch": 1.4403343200683594e-05,
      "model_forward_time": 1.8084287643432617,
      "step": 89103
    },
    {
      "epoch": 1.4403343200683594e-05,
      "step": 89103,
      "training_step_time": 7.72434663772583
    },
    {
      "epoch": 1.4404296875e-05,
      "model_forward_time": 1.8062539100646973,
      "step": 89104
    },
    {
      "epoch": 1.4404296875e-05,
      "step": 89104,
      "training_step_time": 7.717810392379761
    },
    {
      "epoch": 1.4405250549316407e-05,
      "model_forward_time": 1.8001999855041504,
      "step": 89105
    },
    {
      "epoch": 1.4405250549316407e-05,
      "step": 89105,
      "training_step_time": 8.23572587966919
    },
    {
      "epoch": 1.4406204223632812e-05,
      "model_forward_time": 1.799039363861084,
      "step": 89106
    },
    {
      "epoch": 1.4406204223632812e-05,
      "step": 89106,
      "training_step_time": 7.844405651092529
    },
    {
      "epoch": 1.4407157897949218e-05,
      "model_forward_time": 1.8674290180206299,
      "step": 89107
    },
    {
      "epoch": 1.4407157897949218e-05,
      "step": 89107,
      "training_step_time": 9.761165380477905
    },
    {
      "epoch": 1.4408111572265626e-05,
      "model_forward_time": 2.896172523498535,
      "step": 89108
    },
    {
      "epoch": 1.4408111572265626e-05,
      "step": 89108,
      "training_step_time": 7.696218729019165
    },
    {
      "epoch": 1.4409065246582031e-05,
      "model_forward_time": 1.7874934673309326,
      "step": 89109
    },
    {
      "epoch": 1.4409065246582031e-05,
      "step": 89109,
      "training_step_time": 7.849993944168091
    },
    {
      "epoch": 1.4410018920898437e-05,
      "grad_norm": 0.08663419634103775,
      "learning_rate": 3.2079563875233257e-07,
      "loss": 0.0553,
      "step": 89110
    },
    {
      "epoch": 1.4410018920898437e-05,
      "model_forward_time": 1.788940191268921,
      "step": 89110
    },
    {
      "dynamics_loss_avg": 0.05660497862845659,
      "epoch": 1.4410018920898437e-05,
      "step": 89110
    },
    {
      "action_loss_avg": 0.0012859585141995922,
      "epoch": 1.4410018920898437e-05,
      "step": 89110
    },
    {
      "epoch": 1.4410018920898437e-05,
      "step": 89110,
      "training_step_time": 7.802995681762695
    },
    {
      "epoch": 1.4410972595214844e-05,
      "model_forward_time": 1.8137176036834717,
      "step": 89111
    },
    {
      "epoch": 1.4410972595214844e-05,
      "step": 89111,
      "training_step_time": 7.535213232040405
    },
    {
      "epoch": 1.441192626953125e-05,
      "model_forward_time": 1.8029780387878418,
      "step": 89112
    },
    {
      "epoch": 1.441192626953125e-05,
      "step": 89112,
      "training_step_time": 7.908770561218262
    },
    {
      "epoch": 1.4412879943847656e-05,
      "model_forward_time": 1.8119969367980957,
      "step": 89113
    },
    {
      "epoch": 1.4412879943847656e-05,
      "step": 89113,
      "training_step_time": 7.711395025253296
    },
    {
      "epoch": 1.4413833618164063e-05,
      "model_forward_time": 1.797093391418457,
      "step": 89114
    },
    {
      "epoch": 1.4413833618164063e-05,
      "step": 89114,
      "training_step_time": 8.488248348236084
    },
    {
      "epoch": 1.4414787292480469e-05,
      "model_forward_time": 1.7925176620483398,
      "step": 89115
    },
    {
      "epoch": 1.4414787292480469e-05,
      "step": 89115,
      "training_step_time": 8.467426538467407
    },
    {
      "epoch": 1.4415740966796875e-05,
      "model_forward_time": 1.8774819374084473,
      "step": 89116
    },
    {
      "epoch": 1.4415740966796875e-05,
      "step": 89116,
      "training_step_time": 7.655646085739136
    },
    {
      "epoch": 1.4416694641113282e-05,
      "model_forward_time": 2.859651565551758,
      "step": 89117
    },
    {
      "epoch": 1.4416694641113282e-05,
      "step": 89117,
      "training_step_time": 8.3899827003479
    },
    {
      "epoch": 1.4417648315429688e-05,
      "model_forward_time": 1.7958769798278809,
      "step": 89118
    },
    {
      "epoch": 1.4417648315429688e-05,
      "step": 89118,
      "training_step_time": 7.73239541053772
    },
    {
      "epoch": 1.4418601989746094e-05,
      "model_forward_time": 1.781334638595581,
      "step": 89119
    },
    {
      "epoch": 1.4418601989746094e-05,
      "step": 89119,
      "training_step_time": 7.62311863899231
    },
    {
      "epoch": 1.44195556640625e-05,
      "grad_norm": 0.08759571611881256,
      "learning_rate": 3.202131740807118e-07,
      "loss": 0.0553,
      "step": 89120
    },
    {
      "epoch": 1.44195556640625e-05,
      "model_forward_time": 1.7791898250579834,
      "step": 89120
    },
    {
      "dynamics_loss_avg": 0.06651630755513907,
      "epoch": 1.44195556640625e-05,
      "step": 89120
    },
    {
      "action_loss_avg": 0.0013026207190705464,
      "epoch": 1.44195556640625e-05,
      "step": 89120
    },
    {
      "epoch": 1.44195556640625e-05,
      "step": 89120,
      "training_step_time": 7.703853368759155
    },
    {
      "epoch": 1.4420509338378907e-05,
      "model_forward_time": 1.7835118770599365,
      "step": 89121
    },
    {
      "epoch": 1.4420509338378907e-05,
      "step": 89121,
      "training_step_time": 7.772762775421143
    },
    {
      "epoch": 1.4421463012695312e-05,
      "model_forward_time": 1.7785956859588623,
      "step": 89122
    },
    {
      "epoch": 1.4421463012695312e-05,
      "step": 89122,
      "training_step_time": 7.6401941776275635
    },
    {
      "epoch": 1.4422416687011718e-05,
      "model_forward_time": 1.770463228225708,
      "step": 89123
    },
    {
      "epoch": 1.4422416687011718e-05,
      "step": 89123,
      "training_step_time": 8.379027843475342
    },
    {
      "epoch": 1.4423370361328126e-05,
      "model_forward_time": 1.79380464553833,
      "step": 89124
    },
    {
      "epoch": 1.4423370361328126e-05,
      "step": 89124,
      "training_step_time": 8.318405628204346
    },
    {
      "epoch": 1.4424324035644531e-05,
      "model_forward_time": 1.8110661506652832,
      "step": 89125
    },
    {
      "epoch": 1.4424324035644531e-05,
      "step": 89125,
      "training_step_time": 7.665980100631714
    },
    {
      "epoch": 1.4425277709960937e-05,
      "model_forward_time": 2.8756039142608643,
      "step": 89126
    },
    {
      "epoch": 1.4425277709960937e-05,
      "step": 89126,
      "training_step_time": 9.601362466812134
    },
    {
      "epoch": 1.4426231384277344e-05,
      "model_forward_time": 1.7812669277191162,
      "step": 89127
    },
    {
      "epoch": 1.4426231384277344e-05,
      "step": 89127,
      "training_step_time": 8.292343378067017
    },
    {
      "epoch": 1.442718505859375e-05,
      "model_forward_time": 1.774076223373413,
      "step": 89128
    },
    {
      "epoch": 1.442718505859375e-05,
      "step": 89128,
      "training_step_time": 7.722546100616455
    },
    {
      "epoch": 1.4428138732910156e-05,
      "model_forward_time": 1.791712760925293,
      "step": 89129
    },
    {
      "epoch": 1.4428138732910156e-05,
      "step": 89129,
      "training_step_time": 7.68132209777832
    },
    {
      "epoch": 1.4429092407226563e-05,
      "grad_norm": 0.12827453017234802,
      "learning_rate": 3.196312211835434e-07,
      "loss": 0.056,
      "step": 89130
    },
    {
      "epoch": 1.4429092407226563e-05,
      "model_forward_time": 1.8007266521453857,
      "step": 89130
    },
    {
      "dynamics_loss_avg": 0.0675455903634429,
      "epoch": 1.4429092407226563e-05,
      "step": 89130
    },
    {
      "action_loss_avg": 0.0013415764580713585,
      "epoch": 1.4429092407226563e-05,
      "step": 89130
    },
    {
      "epoch": 1.4429092407226563e-05,
      "step": 89130,
      "training_step_time": 7.707610130310059
    },
    {
      "epoch": 1.4430046081542969e-05,
      "model_forward_time": 1.780364990234375,
      "step": 89131
    },
    {
      "epoch": 1.4430046081542969e-05,
      "step": 89131,
      "training_step_time": 8.480200052261353
    },
    {
      "epoch": 1.4430999755859375e-05,
      "model_forward_time": 1.7669076919555664,
      "step": 89132
    },
    {
      "epoch": 1.4430999755859375e-05,
      "step": 89132,
      "training_step_time": 8.454566717147827
    },
    {
      "epoch": 1.443195343017578e-05,
      "model_forward_time": 1.7879259586334229,
      "step": 89133
    },
    {
      "epoch": 1.443195343017578e-05,
      "step": 89133,
      "training_step_time": 10.05648946762085
    },
    {
      "epoch": 1.4432907104492188e-05,
      "model_forward_time": 1.8093714714050293,
      "step": 89134
    },
    {
      "epoch": 1.4432907104492188e-05,
      "step": 89134,
      "training_step_time": 8.246187448501587
    },
    {
      "epoch": 1.4433860778808594e-05,
      "model_forward_time": 2.8646399974823,
      "step": 89135
    },
    {
      "epoch": 1.4433860778808594e-05,
      "step": 89135,
      "training_step_time": 8.759002685546875
    },
    {
      "epoch": 1.4434814453125e-05,
      "model_forward_time": 1.7810416221618652,
      "step": 89136
    },
    {
      "epoch": 1.4434814453125e-05,
      "step": 89136,
      "training_step_time": 8.32648777961731
    },
    {
      "epoch": 1.4435768127441407e-05,
      "model_forward_time": 1.7740461826324463,
      "step": 89137
    },
    {
      "epoch": 1.4435768127441407e-05,
      "step": 89137,
      "training_step_time": 7.761939525604248
    },
    {
      "epoch": 1.4436721801757812e-05,
      "model_forward_time": 1.7787630558013916,
      "step": 89138
    },
    {
      "epoch": 1.4436721801757812e-05,
      "step": 89138,
      "training_step_time": 8.410327911376953
    },
    {
      "epoch": 1.4437675476074218e-05,
      "model_forward_time": 1.7655041217803955,
      "step": 89139
    },
    {
      "epoch": 1.4437675476074218e-05,
      "step": 89139,
      "training_step_time": 8.599547147750854
    },
    {
      "epoch": 1.4438629150390626e-05,
      "grad_norm": 0.0832926332950592,
      "learning_rate": 3.1904978012446687e-07,
      "loss": 0.0552,
      "step": 89140
    },
    {
      "epoch": 1.4438629150390626e-05,
      "model_forward_time": 1.7797760963439941,
      "step": 89140
    },
    {
      "dynamics_loss_avg": 0.06505452077835798,
      "epoch": 1.4438629150390626e-05,
      "step": 89140
    },
    {
      "action_loss_avg": 0.001958297920646146,
      "epoch": 1.4438629150390626e-05,
      "step": 89140
    },
    {
      "epoch": 1.4438629150390626e-05,
      "step": 89140,
      "training_step_time": 7.684311151504517
    },
    {
      "epoch": 1.4439582824707031e-05,
      "model_forward_time": 1.7945170402526855,
      "step": 89141
    },
    {
      "epoch": 1.4439582824707031e-05,
      "step": 89141,
      "training_step_time": 8.60156512260437
    },
    {
      "epoch": 1.4440536499023437e-05,
      "model_forward_time": 1.7784740924835205,
      "step": 89142
    },
    {
      "epoch": 1.4440536499023437e-05,
      "step": 89142,
      "training_step_time": 8.623286247253418
    },
    {
      "epoch": 1.4441490173339844e-05,
      "model_forward_time": 1.7692370414733887,
      "step": 89143
    },
    {
      "epoch": 1.4441490173339844e-05,
      "step": 89143,
      "training_step_time": 11.102589845657349
    },
    {
      "epoch": 1.444244384765625e-05,
      "model_forward_time": 2.901174306869507,
      "step": 89144
    },
    {
      "epoch": 1.444244384765625e-05,
      "step": 89144,
      "training_step_time": 8.687599420547485
    },
    {
      "epoch": 1.4443397521972656e-05,
      "model_forward_time": 1.809964656829834,
      "step": 89145
    },
    {
      "epoch": 1.4443397521972656e-05,
      "step": 89145,
      "training_step_time": 9.33187222480774
    },
    {
      "epoch": 1.4444351196289063e-05,
      "model_forward_time": 1.8004696369171143,
      "step": 89146
    },
    {
      "epoch": 1.4444351196289063e-05,
      "step": 89146,
      "training_step_time": 7.689167737960815
    },
    {
      "epoch": 1.4445304870605469e-05,
      "model_forward_time": 1.8090994358062744,
      "step": 89147
    },
    {
      "epoch": 1.4445304870605469e-05,
      "step": 89147,
      "training_step_time": 8.498200178146362
    },
    {
      "epoch": 1.4446258544921875e-05,
      "model_forward_time": 1.8106238842010498,
      "step": 89148
    },
    {
      "epoch": 1.4446258544921875e-05,
      "step": 89148,
      "training_step_time": 8.197654724121094
    },
    {
      "epoch": 1.444721221923828e-05,
      "model_forward_time": 1.8038926124572754,
      "step": 89149
    },
    {
      "epoch": 1.444721221923828e-05,
      "step": 89149,
      "training_step_time": 8.411235809326172
    },
    {
      "epoch": 1.4448165893554688e-05,
      "grad_norm": 0.08066218346357346,
      "learning_rate": 3.1846885096706914e-07,
      "loss": 0.0541,
      "step": 89150
    },
    {
      "epoch": 1.4448165893554688e-05,
      "model_forward_time": 1.8112561702728271,
      "step": 89150
    },
    {
      "dynamics_loss_avg": 0.06094182673841715,
      "epoch": 1.4448165893554688e-05,
      "step": 89150
    },
    {
      "action_loss_avg": 0.001984623383032158,
      "epoch": 1.4448165893554688e-05,
      "step": 89150
    },
    {
      "epoch": 1.4448165893554688e-05,
      "step": 89150,
      "training_step_time": 7.604322910308838
    },
    {
      "epoch": 1.4449119567871094e-05,
      "model_forward_time": 1.8125112056732178,
      "step": 89151
    },
    {
      "epoch": 1.4449119567871094e-05,
      "step": 89151,
      "training_step_time": 8.669110774993896
    },
    {
      "epoch": 1.44500732421875e-05,
      "model_forward_time": 1.7954864501953125,
      "step": 89152
    },
    {
      "epoch": 1.44500732421875e-05,
      "step": 89152,
      "training_step_time": 8.528059959411621
    },
    {
      "epoch": 1.4451026916503907e-05,
      "model_forward_time": 1.8447439670562744,
      "step": 89153
    },
    {
      "epoch": 1.4451026916503907e-05,
      "step": 89153,
      "training_step_time": 8.663256406784058
    },
    {
      "epoch": 1.4451980590820312e-05,
      "model_forward_time": 1.7873382568359375,
      "step": 89154
    },
    {
      "epoch": 1.4451980590820312e-05,
      "step": 89154,
      "training_step_time": 7.642159461975098
    },
    {
      "epoch": 1.4452934265136718e-05,
      "model_forward_time": 1.7893126010894775,
      "step": 89155
    },
    {
      "epoch": 1.4452934265136718e-05,
      "step": 89155,
      "training_step_time": 7.633609294891357
    },
    {
      "epoch": 1.4453887939453126e-05,
      "model_forward_time": 1.8032331466674805,
      "step": 89156
    },
    {
      "epoch": 1.4453887939453126e-05,
      "step": 89156,
      "training_step_time": 7.582431316375732
    },
    {
      "epoch": 1.4454841613769531e-05,
      "model_forward_time": 1.7956466674804688,
      "step": 89157
    },
    {
      "epoch": 1.4454841613769531e-05,
      "step": 89157,
      "training_step_time": 7.674468755722046
    },
    {
      "epoch": 1.4455795288085937e-05,
      "model_forward_time": 1.7899129390716553,
      "step": 89158
    },
    {
      "epoch": 1.4455795288085937e-05,
      "step": 89158,
      "training_step_time": 8.339163541793823
    },
    {
      "epoch": 1.4456748962402344e-05,
      "model_forward_time": 1.8001751899719238,
      "step": 89159
    },
    {
      "epoch": 1.4456748962402344e-05,
      "step": 89159,
      "training_step_time": 7.621542930603027
    },
    {
      "epoch": 1.445770263671875e-05,
      "grad_norm": 0.10562226921319962,
      "learning_rate": 3.178884337748783e-07,
      "loss": 0.0566,
      "step": 89160
    },
    {
      "epoch": 1.445770263671875e-05,
      "model_forward_time": 1.8080334663391113,
      "step": 89160
    },
    {
      "dynamics_loss_avg": 0.06332375817000865,
      "epoch": 1.445770263671875e-05,
      "step": 89160
    },
    {
      "action_loss_avg": 0.0008913013676647097,
      "epoch": 1.445770263671875e-05,
      "step": 89160
    },
    {
      "epoch": 1.445770263671875e-05,
      "step": 89160,
      "training_step_time": 7.675410270690918
    },
    {
      "epoch": 1.4458656311035156e-05,
      "model_forward_time": 1.7868216037750244,
      "step": 89161
    },
    {
      "epoch": 1.4458656311035156e-05,
      "step": 89161,
      "training_step_time": 9.082759618759155
    },
    {
      "epoch": 1.4459609985351563e-05,
      "model_forward_time": 1.8647685050964355,
      "step": 89162
    },
    {
      "epoch": 1.4459609985351563e-05,
      "step": 89162,
      "training_step_time": 8.407622337341309
    },
    {
      "epoch": 1.4460563659667969e-05,
      "model_forward_time": 2.903684377670288,
      "step": 89163
    },
    {
      "epoch": 1.4460563659667969e-05,
      "step": 89163,
      "training_step_time": 8.514242887496948
    },
    {
      "epoch": 1.4461517333984375e-05,
      "model_forward_time": 1.8126089572906494,
      "step": 89164
    },
    {
      "epoch": 1.4461517333984375e-05,
      "step": 89164,
      "training_step_time": 7.640573978424072
    },
    {
      "epoch": 1.446247100830078e-05,
      "model_forward_time": 1.7965409755706787,
      "step": 89165
    },
    {
      "epoch": 1.446247100830078e-05,
      "step": 89165,
      "training_step_time": 7.6357245445251465
    },
    {
      "epoch": 1.4463424682617188e-05,
      "model_forward_time": 1.8011393547058105,
      "step": 89166
    },
    {
      "epoch": 1.4463424682617188e-05,
      "step": 89166,
      "training_step_time": 7.630218267440796
    },
    {
      "epoch": 1.4464378356933594e-05,
      "model_forward_time": 1.7901990413665771,
      "step": 89167
    },
    {
      "epoch": 1.4464378356933594e-05,
      "step": 89167,
      "training_step_time": 7.807087182998657
    },
    {
      "epoch": 1.446533203125e-05,
      "model_forward_time": 1.800440788269043,
      "step": 89168
    },
    {
      "epoch": 1.446533203125e-05,
      "step": 89168,
      "training_step_time": 7.640485763549805
    },
    {
      "epoch": 1.4466285705566407e-05,
      "model_forward_time": 1.8066682815551758,
      "step": 89169
    },
    {
      "epoch": 1.4466285705566407e-05,
      "step": 89169,
      "training_step_time": 7.69268798828125
    },
    {
      "epoch": 1.4467239379882812e-05,
      "grad_norm": 0.11155155301094055,
      "learning_rate": 3.1730852861136973e-07,
      "loss": 0.0565,
      "step": 89170
    },
    {
      "epoch": 1.4467239379882812e-05,
      "model_forward_time": 1.792478084564209,
      "step": 89170
    },
    {
      "dynamics_loss_avg": 0.05078509319573641,
      "epoch": 1.4467239379882812e-05,
      "step": 89170
    },
    {
      "action_loss_avg": 0.0009096732828766108,
      "epoch": 1.4467239379882812e-05,
      "step": 89170
    },
    {
      "epoch": 1.4467239379882812e-05,
      "step": 89170,
      "training_step_time": 8.827799081802368
    },
    {
      "epoch": 1.4468193054199218e-05,
      "model_forward_time": 1.8720951080322266,
      "step": 89171
    },
    {
      "epoch": 1.4468193054199218e-05,
      "step": 89171,
      "training_step_time": 9.703476190567017
    },
    {
      "epoch": 1.4469146728515626e-05,
      "model_forward_time": 2.9177656173706055,
      "step": 89172
    },
    {
      "epoch": 1.4469146728515626e-05,
      "step": 89172,
      "training_step_time": 8.743774175643921
    },
    {
      "epoch": 1.4470100402832031e-05,
      "model_forward_time": 1.8087561130523682,
      "step": 89173
    },
    {
      "epoch": 1.4470100402832031e-05,
      "step": 89173,
      "training_step_time": 7.618954420089722
    },
    {
      "epoch": 1.4471054077148437e-05,
      "model_forward_time": 1.802288293838501,
      "step": 89174
    },
    {
      "epoch": 1.4471054077148437e-05,
      "step": 89174,
      "training_step_time": 7.64460015296936
    },
    {
      "epoch": 1.4472007751464844e-05,
      "model_forward_time": 1.788841724395752,
      "step": 89175
    },
    {
      "epoch": 1.4472007751464844e-05,
      "step": 89175,
      "training_step_time": 8.223479986190796
    },
    {
      "epoch": 1.447296142578125e-05,
      "model_forward_time": 1.8092610836029053,
      "step": 89176
    },
    {
      "epoch": 1.447296142578125e-05,
      "step": 89176,
      "training_step_time": 8.7514488697052
    },
    {
      "epoch": 1.4473915100097656e-05,
      "model_forward_time": 1.8092389106750488,
      "step": 89177
    },
    {
      "epoch": 1.4473915100097656e-05,
      "step": 89177,
      "training_step_time": 7.6606669425964355
    },
    {
      "epoch": 1.4474868774414063e-05,
      "model_forward_time": 1.789564847946167,
      "step": 89178
    },
    {
      "epoch": 1.4474868774414063e-05,
      "step": 89178,
      "training_step_time": 9.23298168182373
    },
    {
      "epoch": 1.4475822448730469e-05,
      "model_forward_time": 1.8016314506530762,
      "step": 89179
    },
    {
      "epoch": 1.4475822448730469e-05,
      "step": 89179,
      "training_step_time": 8.55746865272522
    },
    {
      "epoch": 1.4476776123046875e-05,
      "grad_norm": 0.07600294053554535,
      "learning_rate": 3.167291355399599e-07,
      "loss": 0.0546,
      "step": 89180
    },
    {
      "epoch": 1.4476776123046875e-05,
      "model_forward_time": 1.8559367656707764,
      "step": 89180
    },
    {
      "dynamics_loss_avg": 0.055927845742553474,
      "epoch": 1.4476776123046875e-05,
      "step": 89180
    },
    {
      "action_loss_avg": 0.0010197611525654792,
      "epoch": 1.4476776123046875e-05,
      "step": 89180
    },
    {
      "epoch": 1.4476776123046875e-05,
      "step": 89180,
      "training_step_time": 10.26379942893982
    },
    {
      "epoch": 1.447772979736328e-05,
      "model_forward_time": 1.805708885192871,
      "step": 89181
    },
    {
      "epoch": 1.447772979736328e-05,
      "step": 89181,
      "training_step_time": 7.494735240936279
    },
    {
      "epoch": 1.4478683471679688e-05,
      "model_forward_time": 1.8035576343536377,
      "step": 89182
    },
    {
      "epoch": 1.4478683471679688e-05,
      "step": 89182,
      "training_step_time": 8.694833517074585
    },
    {
      "epoch": 1.4479637145996094e-05,
      "model_forward_time": 1.4506111145019531,
      "step": 89183
    },
    {
      "epoch": 1.4479637145996094e-05,
      "step": 89183,
      "training_step_time": 6.793389320373535
    },
    {
      "epoch": 1.44805908203125e-05,
      "model_forward_time": 1.8051903247833252,
      "step": 89184
    },
    {
      "epoch": 1.44805908203125e-05,
      "step": 89184,
      "training_step_time": 8.805368423461914
    },
    {
      "epoch": 1.4481544494628907e-05,
      "model_forward_time": 1.8067514896392822,
      "step": 89185
    },
    {
      "epoch": 1.4481544494628907e-05,
      "step": 89185,
      "training_step_time": 6.798518180847168
    },
    {
      "epoch": 1.4482498168945312e-05,
      "model_forward_time": 1.802347183227539,
      "step": 89186
    },
    {
      "epoch": 1.4482498168945312e-05,
      "step": 89186,
      "training_step_time": 6.784120559692383
    },
    {
      "epoch": 1.4483451843261718e-05,
      "model_forward_time": 1.8066685199737549,
      "step": 89187
    },
    {
      "epoch": 1.4483451843261718e-05,
      "step": 89187,
      "training_step_time": 6.789220809936523
    },
    {
      "epoch": 1.4484405517578126e-05,
      "model_forward_time": 1.8042809963226318,
      "step": 89188
    },
    {
      "epoch": 1.4484405517578126e-05,
      "step": 89188,
      "training_step_time": 6.774017810821533
    },
    {
      "epoch": 1.4485359191894531e-05,
      "model_forward_time": 1.8182694911956787,
      "step": 89189
    },
    {
      "epoch": 1.4485359191894531e-05,
      "step": 89189,
      "training_step_time": 7.866841554641724
    },
    {
      "epoch": 1.4486312866210937e-05,
      "grad_norm": 0.07085513323545456,
      "learning_rate": 3.161502546240114e-07,
      "loss": 0.0541,
      "step": 89190
    },
    {
      "epoch": 1.4486312866210937e-05,
      "model_forward_time": 2.911149740219116,
      "step": 89190
    },
    {
      "dynamics_loss_avg": 0.06329836901277304,
      "epoch": 1.4486312866210937e-05,
      "step": 89190
    },
    {
      "action_loss_avg": 0.001119673578068614,
      "epoch": 1.4486312866210937e-05,
      "step": 89190
    },
    {
      "epoch": 1.4486312866210937e-05,
      "step": 89190,
      "training_step_time": 8.530304908752441
    },
    {
      "epoch": 1.4487266540527344e-05,
      "model_forward_time": 1.807722806930542,
      "step": 89191
    },
    {
      "epoch": 1.4487266540527344e-05,
      "step": 89191,
      "training_step_time": 8.561057090759277
    },
    {
      "epoch": 1.448822021484375e-05,
      "model_forward_time": 1.8100173473358154,
      "step": 89192
    },
    {
      "epoch": 1.448822021484375e-05,
      "step": 89192,
      "training_step_time": 8.25555419921875
    },
    {
      "epoch": 1.4489173889160156e-05,
      "model_forward_time": 1.811077356338501,
      "step": 89193
    },
    {
      "epoch": 1.4489173889160156e-05,
      "step": 89193,
      "training_step_time": 7.662583112716675
    },
    {
      "epoch": 1.4490127563476563e-05,
      "model_forward_time": 1.8101801872253418,
      "step": 89194
    },
    {
      "epoch": 1.4490127563476563e-05,
      "step": 89194,
      "training_step_time": 7.221680641174316
    },
    {
      "epoch": 1.4491081237792969e-05,
      "model_forward_time": 1.8105299472808838,
      "step": 89195
    },
    {
      "epoch": 1.4491081237792969e-05,
      "step": 89195,
      "training_step_time": 6.754173278808594
    },
    {
      "epoch": 1.4492034912109375e-05,
      "model_forward_time": 1.8076953887939453,
      "step": 89196
    },
    {
      "epoch": 1.4492034912109375e-05,
      "step": 89196,
      "training_step_time": 6.793207883834839
    },
    {
      "epoch": 1.449298858642578e-05,
      "model_forward_time": 1.8051960468292236,
      "step": 89197
    },
    {
      "epoch": 1.449298858642578e-05,
      "step": 89197,
      "training_step_time": 6.8301873207092285
    },
    {
      "epoch": 1.4493942260742188e-05,
      "model_forward_time": 1.7984421253204346,
      "step": 89198
    },
    {
      "epoch": 1.4493942260742188e-05,
      "step": 89198,
      "training_step_time": 6.7747650146484375
    },
    {
      "epoch": 1.4494895935058594e-05,
      "model_forward_time": 2.870615005493164,
      "step": 89199
    },
    {
      "epoch": 1.4494895935058594e-05,
      "step": 89199,
      "training_step_time": 9.325313329696655
    },
    {
      "epoch": 1.4495849609375e-05,
      "grad_norm": 0.09295754879713058,
      "learning_rate": 3.155718859268281e-07,
      "loss": 0.0551,
      "step": 89200
    },
    {
      "epoch": 1.4495849609375e-05,
      "model_forward_time": 1.8091833591461182,
      "step": 89200
    },
    {
      "dynamics_loss_avg": 0.04439828731119633,
      "epoch": 1.4495849609375e-05,
      "step": 89200
    },
    {
      "action_loss_avg": 0.0012028108758386225,
      "epoch": 1.4495849609375e-05,
      "step": 89200
    },
    {
      "epoch": 1.4495849609375e-05,
      "step": 89200,
      "training_step_time": 7.835493326187134
    },
    {
      "epoch": 1.4496803283691407e-05,
      "model_forward_time": 1.804816484451294,
      "step": 89201
    },
    {
      "epoch": 1.4496803283691407e-05,
      "step": 89201,
      "training_step_time": 7.382066249847412
    },
    {
      "epoch": 1.4497756958007812e-05,
      "model_forward_time": 1.7980268001556396,
      "step": 89202
    },
    {
      "epoch": 1.4497756958007812e-05,
      "step": 89202,
      "training_step_time": 7.700231552124023
    },
    {
      "epoch": 1.4498710632324218e-05,
      "model_forward_time": 1.8050291538238525,
      "step": 89203
    },
    {
      "epoch": 1.4498710632324218e-05,
      "step": 89203,
      "training_step_time": 7.621971607208252
    },
    {
      "epoch": 1.4499664306640626e-05,
      "model_forward_time": 1.804901361465454,
      "step": 89204
    },
    {
      "epoch": 1.4499664306640626e-05,
      "step": 89204,
      "training_step_time": 6.802820682525635
    },
    {
      "epoch": 1.4500617980957031e-05,
      "model_forward_time": 1.7989606857299805,
      "step": 89205
    },
    {
      "epoch": 1.4500617980957031e-05,
      "step": 89205,
      "training_step_time": 6.814865589141846
    },
    {
      "epoch": 1.4501571655273437e-05,
      "model_forward_time": 1.7960765361785889,
      "step": 89206
    },
    {
      "epoch": 1.4501571655273437e-05,
      "step": 89206,
      "training_step_time": 6.805278062820435
    },
    {
      "epoch": 1.4502525329589844e-05,
      "model_forward_time": 1.8095932006835938,
      "step": 89207
    },
    {
      "epoch": 1.4502525329589844e-05,
      "step": 89207,
      "training_step_time": 6.795325756072998
    },
    {
      "epoch": 1.450347900390625e-05,
      "model_forward_time": 2.887349843978882,
      "step": 89208
    },
    {
      "epoch": 1.450347900390625e-05,
      "step": 89208,
      "training_step_time": 9.958022117614746
    },
    {
      "epoch": 1.4504432678222656e-05,
      "model_forward_time": 1.8023221492767334,
      "step": 89209
    },
    {
      "epoch": 1.4504432678222656e-05,
      "step": 89209,
      "training_step_time": 7.80153226852417
    },
    {
      "epoch": 1.4505386352539063e-05,
      "grad_norm": 0.07594255357980728,
      "learning_rate": 3.1499402951166046e-07,
      "loss": 0.0534,
      "step": 89210
    },
    {
      "epoch": 1.4505386352539063e-05,
      "model_forward_time": 1.8319737911224365,
      "step": 89210
    },
    {
      "dynamics_loss_avg": 0.05741440560668707,
      "epoch": 1.4505386352539063e-05,
      "step": 89210
    },
    {
      "action_loss_avg": 0.001506380579667166,
      "epoch": 1.4505386352539063e-05,
      "step": 89210
    },
    {
      "epoch": 1.4505386352539063e-05,
      "step": 89210,
      "training_step_time": 7.535275459289551
    },
    {
      "epoch": 1.4506340026855469e-05,
      "model_forward_time": 1.8028011322021484,
      "step": 89211
    },
    {
      "epoch": 1.4506340026855469e-05,
      "step": 89211,
      "training_step_time": 7.691304683685303
    },
    {
      "epoch": 1.4507293701171875e-05,
      "model_forward_time": 1.8055822849273682,
      "step": 89212
    },
    {
      "epoch": 1.4507293701171875e-05,
      "step": 89212,
      "training_step_time": 7.6095874309539795
    },
    {
      "epoch": 1.450824737548828e-05,
      "model_forward_time": 1.8003830909729004,
      "step": 89213
    },
    {
      "epoch": 1.450824737548828e-05,
      "step": 89213,
      "training_step_time": 6.74436616897583
    },
    {
      "epoch": 1.4509201049804688e-05,
      "model_forward_time": 1.7996060848236084,
      "step": 89214
    },
    {
      "epoch": 1.4509201049804688e-05,
      "step": 89214,
      "training_step_time": 6.768133163452148
    },
    {
      "epoch": 1.4510154724121094e-05,
      "model_forward_time": 1.7989003658294678,
      "step": 89215
    },
    {
      "epoch": 1.4510154724121094e-05,
      "step": 89215,
      "training_step_time": 6.76771879196167
    },
    {
      "epoch": 1.45111083984375e-05,
      "model_forward_time": 1.8033339977264404,
      "step": 89216
    },
    {
      "epoch": 1.45111083984375e-05,
      "step": 89216,
      "training_step_time": 6.746321439743042
    },
    {
      "epoch": 1.4512062072753907e-05,
      "model_forward_time": 2.9032912254333496,
      "step": 89217
    },
    {
      "epoch": 1.4512062072753907e-05,
      "step": 89217,
      "training_step_time": 8.54815411567688
    },
    {
      "epoch": 1.4513015747070312e-05,
      "model_forward_time": 1.8013427257537842,
      "step": 89218
    },
    {
      "epoch": 1.4513015747070312e-05,
      "step": 89218,
      "training_step_time": 9.837239265441895
    },
    {
      "epoch": 1.4513969421386718e-05,
      "model_forward_time": 1.7956736087799072,
      "step": 89219
    },
    {
      "epoch": 1.4513969421386718e-05,
      "step": 89219,
      "training_step_time": 7.677417755126953
    },
    {
      "epoch": 1.4514923095703126e-05,
      "grad_norm": 0.08445718884468079,
      "learning_rate": 3.1441668544170187e-07,
      "loss": 0.0548,
      "step": 89220
    },
    {
      "epoch": 1.4514923095703126e-05,
      "model_forward_time": 1.799302577972412,
      "step": 89220
    },
    {
      "dynamics_loss_avg": 0.0454550102353096,
      "epoch": 1.4514923095703126e-05,
      "step": 89220
    },
    {
      "action_loss_avg": 0.001332830503815785,
      "epoch": 1.4514923095703126e-05,
      "step": 89220
    },
    {
      "epoch": 1.4514923095703126e-05,
      "step": 89220,
      "training_step_time": 7.681190490722656
    },
    {
      "epoch": 1.4515876770019531e-05,
      "model_forward_time": 1.800520420074463,
      "step": 89221
    },
    {
      "epoch": 1.4515876770019531e-05,
      "step": 89221,
      "training_step_time": 7.65970778465271
    },
    {
      "epoch": 1.4516830444335937e-05,
      "model_forward_time": 1.8015697002410889,
      "step": 89222
    },
    {
      "epoch": 1.4516830444335937e-05,
      "step": 89222,
      "training_step_time": 6.7187659740448
    },
    {
      "epoch": 1.4517784118652344e-05,
      "model_forward_time": 1.8040473461151123,
      "step": 89223
    },
    {
      "epoch": 1.4517784118652344e-05,
      "step": 89223,
      "training_step_time": 6.747023344039917
    },
    {
      "epoch": 1.451873779296875e-05,
      "model_forward_time": 1.7964375019073486,
      "step": 89224
    },
    {
      "epoch": 1.451873779296875e-05,
      "step": 89224,
      "training_step_time": 6.745461702346802
    },
    {
      "epoch": 1.4519691467285156e-05,
      "model_forward_time": 1.8010156154632568,
      "step": 89225
    },
    {
      "epoch": 1.4519691467285156e-05,
      "step": 89225,
      "training_step_time": 6.774803876876831
    },
    {
      "epoch": 1.4520645141601563e-05,
      "model_forward_time": 2.916767120361328,
      "step": 89226
    },
    {
      "epoch": 1.4520645141601563e-05,
      "step": 89226,
      "training_step_time": 8.577176570892334
    },
    {
      "epoch": 1.4521598815917969e-05,
      "model_forward_time": 1.7970993518829346,
      "step": 89227
    },
    {
      "epoch": 1.4521598815917969e-05,
      "step": 89227,
      "training_step_time": 10.75408387184143
    },
    {
      "epoch": 1.4522552490234375e-05,
      "model_forward_time": 1.7986173629760742,
      "step": 89228
    },
    {
      "epoch": 1.4522552490234375e-05,
      "step": 89228,
      "training_step_time": 8.313111066818237
    },
    {
      "epoch": 1.452350616455078e-05,
      "model_forward_time": 1.803102970123291,
      "step": 89229
    },
    {
      "epoch": 1.452350616455078e-05,
      "step": 89229,
      "training_step_time": 7.708603143692017
    },
    {
      "epoch": 1.4524459838867188e-05,
      "grad_norm": 0.0790606439113617,
      "learning_rate": 3.138398537800907e-07,
      "loss": 0.0539,
      "step": 89230
    },
    {
      "epoch": 1.4524459838867188e-05,
      "model_forward_time": 1.8019564151763916,
      "step": 89230
    },
    {
      "dynamics_loss_avg": 0.044049928709864614,
      "epoch": 1.4524459838867188e-05,
      "step": 89230
    },
    {
      "action_loss_avg": 0.0007379814516752959,
      "epoch": 1.4524459838867188e-05,
      "step": 89230
    },
    {
      "epoch": 1.4524459838867188e-05,
      "step": 89230,
      "training_step_time": 7.64568305015564
    },
    {
      "epoch": 1.4525413513183594e-05,
      "model_forward_time": 1.8014061450958252,
      "step": 89231
    },
    {
      "epoch": 1.4525413513183594e-05,
      "step": 89231,
      "training_step_time": 6.79181981086731
    },
    {
      "epoch": 1.45263671875e-05,
      "model_forward_time": 1.7996764183044434,
      "step": 89232
    },
    {
      "epoch": 1.45263671875e-05,
      "step": 89232,
      "training_step_time": 6.741785287857056
    },
    {
      "epoch": 1.4527320861816407e-05,
      "model_forward_time": 1.8019373416900635,
      "step": 89233
    },
    {
      "epoch": 1.4527320861816407e-05,
      "step": 89233,
      "training_step_time": 6.73226261138916
    },
    {
      "epoch": 1.4528274536132812e-05,
      "model_forward_time": 1.7972543239593506,
      "step": 89234
    },
    {
      "epoch": 1.4528274536132812e-05,
      "step": 89234,
      "training_step_time": 6.725749731063843
    },
    {
      "epoch": 1.4529228210449218e-05,
      "model_forward_time": 1.8003263473510742,
      "step": 89235
    },
    {
      "epoch": 1.4529228210449218e-05,
      "step": 89235,
      "training_step_time": 9.608073234558105
    },
    {
      "epoch": 1.4530181884765626e-05,
      "model_forward_time": 1.7840313911437988,
      "step": 89236
    },
    {
      "epoch": 1.4530181884765626e-05,
      "step": 89236,
      "training_step_time": 9.563144445419312
    },
    {
      "epoch": 1.4531135559082031e-05,
      "model_forward_time": 1.7792444229125977,
      "step": 89237
    },
    {
      "epoch": 1.4531135559082031e-05,
      "step": 89237,
      "training_step_time": 9.19025993347168
    },
    {
      "epoch": 1.4532089233398437e-05,
      "model_forward_time": 1.7965679168701172,
      "step": 89238
    },
    {
      "epoch": 1.4532089233398437e-05,
      "step": 89238,
      "training_step_time": 7.741211652755737
    },
    {
      "epoch": 1.4533042907714844e-05,
      "model_forward_time": 1.78532075881958,
      "step": 89239
    },
    {
      "epoch": 1.4533042907714844e-05,
      "step": 89239,
      "training_step_time": 7.677561283111572
    },
    {
      "epoch": 1.453399658203125e-05,
      "grad_norm": 0.07507779449224472,
      "learning_rate": 3.1326353458990656e-07,
      "loss": 0.054,
      "step": 89240
    },
    {
      "epoch": 1.453399658203125e-05,
      "model_forward_time": 1.7966406345367432,
      "step": 89240
    },
    {
      "dynamics_loss_avg": 0.059036328084766866,
      "epoch": 1.453399658203125e-05,
      "step": 89240
    },
    {
      "action_loss_avg": 0.0017857952159829437,
      "epoch": 1.453399658203125e-05,
      "step": 89240
    },
    {
      "epoch": 1.453399658203125e-05,
      "step": 89240,
      "training_step_time": 6.77237868309021
    },
    {
      "epoch": 1.4534950256347656e-05,
      "model_forward_time": 1.8216044902801514,
      "step": 89241
    },
    {
      "epoch": 1.4534950256347656e-05,
      "step": 89241,
      "training_step_time": 6.757916450500488
    },
    {
      "epoch": 1.4535903930664063e-05,
      "model_forward_time": 1.7916793823242188,
      "step": 89242
    },
    {
      "epoch": 1.4535903930664063e-05,
      "step": 89242,
      "training_step_time": 6.787210464477539
    },
    {
      "epoch": 1.4536857604980469e-05,
      "model_forward_time": 1.7942817211151123,
      "step": 89243
    },
    {
      "epoch": 1.4536857604980469e-05,
      "step": 89243,
      "training_step_time": 6.7713470458984375
    },
    {
      "epoch": 1.4537811279296875e-05,
      "model_forward_time": 1.7995378971099854,
      "step": 89244
    },
    {
      "epoch": 1.4537811279296875e-05,
      "step": 89244,
      "training_step_time": 7.4540674686431885
    },
    {
      "epoch": 1.453876495361328e-05,
      "model_forward_time": 2.881106376647949,
      "step": 89245
    },
    {
      "epoch": 1.453876495361328e-05,
      "step": 89245,
      "training_step_time": 9.871154546737671
    },
    {
      "epoch": 1.4539718627929688e-05,
      "model_forward_time": 1.7975566387176514,
      "step": 89246
    },
    {
      "epoch": 1.4539718627929688e-05,
      "step": 89246,
      "training_step_time": 7.843964338302612
    },
    {
      "epoch": 1.4540672302246094e-05,
      "model_forward_time": 1.804044485092163,
      "step": 89247
    },
    {
      "epoch": 1.4540672302246094e-05,
      "step": 89247,
      "training_step_time": 8.503066778182983
    },
    {
      "epoch": 1.45416259765625e-05,
      "model_forward_time": 1.8006646633148193,
      "step": 89248
    },
    {
      "epoch": 1.45416259765625e-05,
      "step": 89248,
      "training_step_time": 7.707061529159546
    },
    {
      "epoch": 1.4542579650878907e-05,
      "model_forward_time": 1.7816414833068848,
      "step": 89249
    },
    {
      "epoch": 1.4542579650878907e-05,
      "step": 89249,
      "training_step_time": 7.286309003829956
    },
    {
      "epoch": 1.4543533325195312e-05,
      "grad_norm": 0.15527015924453735,
      "learning_rate": 3.126877279341761e-07,
      "loss": 0.0541,
      "step": 89250
    },
    {
      "epoch": 1.4543533325195312e-05,
      "model_forward_time": 1.8023927211761475,
      "step": 89250
    },
    {
      "dynamics_loss_avg": 0.05810775849968195,
      "epoch": 1.4543533325195312e-05,
      "step": 89250
    },
    {
      "action_loss_avg": 0.0011372877866961061,
      "epoch": 1.4543533325195312e-05,
      "step": 89250
    },
    {
      "epoch": 1.4543533325195312e-05,
      "step": 89250,
      "training_step_time": 6.779094934463501
    },
    {
      "epoch": 1.4544486999511718e-05,
      "model_forward_time": 1.8003058433532715,
      "step": 89251
    },
    {
      "epoch": 1.4544486999511718e-05,
      "step": 89251,
      "training_step_time": 6.761451482772827
    },
    {
      "epoch": 1.4545440673828126e-05,
      "model_forward_time": 1.801518201828003,
      "step": 89252
    },
    {
      "epoch": 1.4545440673828126e-05,
      "step": 89252,
      "training_step_time": 6.780421495437622
    },
    {
      "epoch": 1.4546394348144531e-05,
      "model_forward_time": 1.8062729835510254,
      "step": 89253
    },
    {
      "epoch": 1.4546394348144531e-05,
      "step": 89253,
      "training_step_time": 7.461420297622681
    },
    {
      "epoch": 1.4547348022460937e-05,
      "model_forward_time": 2.86759090423584,
      "step": 89254
    },
    {
      "epoch": 1.4547348022460937e-05,
      "step": 89254,
      "training_step_time": 9.461446046829224
    },
    {
      "epoch": 1.4548301696777344e-05,
      "model_forward_time": 1.8032591342926025,
      "step": 89255
    },
    {
      "epoch": 1.4548301696777344e-05,
      "step": 89255,
      "training_step_time": 7.817171573638916
    },
    {
      "epoch": 1.454925537109375e-05,
      "model_forward_time": 1.803966999053955,
      "step": 89256
    },
    {
      "epoch": 1.454925537109375e-05,
      "step": 89256,
      "training_step_time": 8.75806713104248
    },
    {
      "epoch": 1.4550209045410156e-05,
      "model_forward_time": 1.7999982833862305,
      "step": 89257
    },
    {
      "epoch": 1.4550209045410156e-05,
      "step": 89257,
      "training_step_time": 7.711215496063232
    },
    {
      "epoch": 1.4551162719726562e-05,
      "model_forward_time": 1.7988109588623047,
      "step": 89258
    },
    {
      "epoch": 1.4551162719726562e-05,
      "step": 89258,
      "training_step_time": 7.330630540847778
    },
    {
      "epoch": 1.4552116394042969e-05,
      "model_forward_time": 1.7897536754608154,
      "step": 89259
    },
    {
      "epoch": 1.4552116394042969e-05,
      "step": 89259,
      "training_step_time": 6.816909313201904
    },
    {
      "epoch": 1.4553070068359375e-05,
      "grad_norm": 0.07622645795345306,
      "learning_rate": 3.1211243387586965e-07,
      "loss": 0.0529,
      "step": 89260
    },
    {
      "epoch": 1.4553070068359375e-05,
      "model_forward_time": 1.7922286987304688,
      "step": 89260
    },
    {
      "dynamics_loss_avg": 0.04912113659083843,
      "epoch": 1.4553070068359375e-05,
      "step": 89260
    },
    {
      "action_loss_avg": 0.0013083876052405684,
      "epoch": 1.4553070068359375e-05,
      "step": 89260
    },
    {
      "epoch": 1.4553070068359375e-05,
      "step": 89260,
      "training_step_time": 6.808775901794434
    },
    {
      "epoch": 1.455402374267578e-05,
      "model_forward_time": 1.7836201190948486,
      "step": 89261
    },
    {
      "epoch": 1.455402374267578e-05,
      "step": 89261,
      "training_step_time": 6.808465957641602
    },
    {
      "epoch": 1.4554977416992188e-05,
      "model_forward_time": 1.7927944660186768,
      "step": 89262
    },
    {
      "epoch": 1.4554977416992188e-05,
      "step": 89262,
      "training_step_time": 7.506664276123047
    },
    {
      "epoch": 1.4555931091308594e-05,
      "model_forward_time": 2.902372360229492,
      "step": 89263
    },
    {
      "epoch": 1.4555931091308594e-05,
      "step": 89263,
      "training_step_time": 8.286458730697632
    },
    {
      "epoch": 1.4556884765625e-05,
      "model_forward_time": 1.803687334060669,
      "step": 89264
    },
    {
      "epoch": 1.4556884765625e-05,
      "step": 89264,
      "training_step_time": 9.358890295028687
    },
    {
      "epoch": 1.4557838439941407e-05,
      "model_forward_time": 1.8027715682983398,
      "step": 89265
    },
    {
      "epoch": 1.4557838439941407e-05,
      "step": 89265,
      "training_step_time": 7.853845119476318
    },
    {
      "epoch": 1.4558792114257813e-05,
      "model_forward_time": 1.807621717453003,
      "step": 89266
    },
    {
      "epoch": 1.4558792114257813e-05,
      "step": 89266,
      "training_step_time": 7.764695882797241
    },
    {
      "epoch": 1.4559745788574218e-05,
      "model_forward_time": 1.8017988204956055,
      "step": 89267
    },
    {
      "epoch": 1.4559745788574218e-05,
      "step": 89267,
      "training_step_time": 8.15661883354187
    },
    {
      "epoch": 1.4560699462890626e-05,
      "model_forward_time": 1.796926736831665,
      "step": 89268
    },
    {
      "epoch": 1.4560699462890626e-05,
      "step": 89268,
      "training_step_time": 6.809654235839844
    },
    {
      "epoch": 1.4561653137207031e-05,
      "model_forward_time": 1.77947998046875,
      "step": 89269
    },
    {
      "epoch": 1.4561653137207031e-05,
      "step": 89269,
      "training_step_time": 6.81585431098938
    },
    {
      "epoch": 1.4562606811523437e-05,
      "grad_norm": 0.07282254844903946,
      "learning_rate": 3.115376524778979e-07,
      "loss": 0.0535,
      "step": 89270
    },
    {
      "epoch": 1.4562606811523437e-05,
      "model_forward_time": 1.791205883026123,
      "step": 89270
    },
    {
      "dynamics_loss_avg": 0.0532350892201066,
      "epoch": 1.4562606811523437e-05,
      "step": 89270
    },
    {
      "action_loss_avg": 0.0017469635495217517,
      "epoch": 1.4562606811523437e-05,
      "step": 89270
    },
    {
      "epoch": 1.4562606811523437e-05,
      "step": 89270,
      "training_step_time": 6.80120062828064
    },
    {
      "epoch": 1.4563560485839845e-05,
      "model_forward_time": 1.7783713340759277,
      "step": 89271
    },
    {
      "epoch": 1.4563560485839845e-05,
      "step": 89271,
      "training_step_time": 7.40390157699585
    },
    {
      "epoch": 1.456451416015625e-05,
      "model_forward_time": 2.8612258434295654,
      "step": 89272
    },
    {
      "epoch": 1.456451416015625e-05,
      "step": 89272,
      "training_step_time": 8.734613418579102
    },
    {
      "epoch": 1.4565467834472656e-05,
      "model_forward_time": 1.7824139595031738,
      "step": 89273
    },
    {
      "epoch": 1.4565467834472656e-05,
      "step": 89273,
      "training_step_time": 11.946171998977661
    },
    {
      "epoch": 1.4566421508789062e-05,
      "model_forward_time": 1.7737371921539307,
      "step": 89274
    },
    {
      "epoch": 1.4566421508789062e-05,
      "step": 89274,
      "training_step_time": 7.765357732772827
    },
    {
      "epoch": 1.4567375183105469e-05,
      "model_forward_time": 1.7784125804901123,
      "step": 89275
    },
    {
      "epoch": 1.4567375183105469e-05,
      "step": 89275,
      "training_step_time": 8.541095495223999
    },
    {
      "epoch": 1.4568328857421875e-05,
      "model_forward_time": 1.7891173362731934,
      "step": 89276
    },
    {
      "epoch": 1.4568328857421875e-05,
      "step": 89276,
      "training_step_time": 7.623599529266357
    },
    {
      "epoch": 1.456928253173828e-05,
      "model_forward_time": 1.778754472732544,
      "step": 89277
    },
    {
      "epoch": 1.456928253173828e-05,
      "step": 89277,
      "training_step_time": 7.266010522842407
    },
    {
      "epoch": 1.4570236206054688e-05,
      "model_forward_time": 1.783850908279419,
      "step": 89278
    },
    {
      "epoch": 1.4570236206054688e-05,
      "step": 89278,
      "training_step_time": 6.737679481506348
    },
    {
      "epoch": 1.4571189880371094e-05,
      "model_forward_time": 1.78568696975708,
      "step": 89279
    },
    {
      "epoch": 1.4571189880371094e-05,
      "step": 89279,
      "training_step_time": 6.810966968536377
    },
    {
      "epoch": 1.45721435546875e-05,
      "grad_norm": 0.1067010909318924,
      "learning_rate": 3.1096338380311955e-07,
      "loss": 0.0531,
      "step": 89280
    },
    {
      "epoch": 1.45721435546875e-05,
      "model_forward_time": 1.7808444499969482,
      "step": 89280
    },
    {
      "dynamics_loss_avg": 0.06488321162760258,
      "epoch": 1.45721435546875e-05,
      "step": 89280
    },
    {
      "action_loss_avg": 0.0018556394788902253,
      "epoch": 1.45721435546875e-05,
      "step": 89280
    },
    {
      "epoch": 1.45721435546875e-05,
      "step": 89280,
      "training_step_time": 7.466670751571655
    },
    {
      "epoch": 1.4573097229003907e-05,
      "model_forward_time": 2.8391799926757812,
      "step": 89281
    },
    {
      "epoch": 1.4573097229003907e-05,
      "step": 89281,
      "training_step_time": 9.696855545043945
    },
    {
      "epoch": 1.4574050903320313e-05,
      "model_forward_time": 1.7802822589874268,
      "step": 89282
    },
    {
      "epoch": 1.4574050903320313e-05,
      "step": 89282,
      "training_step_time": 12.296774864196777
    },
    {
      "epoch": 1.4575004577636718e-05,
      "model_forward_time": 1.7793958187103271,
      "step": 89283
    },
    {
      "epoch": 1.4575004577636718e-05,
      "step": 89283,
      "training_step_time": 7.824744939804077
    },
    {
      "epoch": 1.4575958251953126e-05,
      "model_forward_time": 1.7840814590454102,
      "step": 89284
    },
    {
      "epoch": 1.4575958251953126e-05,
      "step": 89284,
      "training_step_time": 7.720677137374878
    },
    {
      "epoch": 1.4576911926269531e-05,
      "model_forward_time": 1.7737491130828857,
      "step": 89285
    },
    {
      "epoch": 1.4576911926269531e-05,
      "step": 89285,
      "training_step_time": 7.632690906524658
    },
    {
      "epoch": 1.4577865600585937e-05,
      "model_forward_time": 1.7662131786346436,
      "step": 89286
    },
    {
      "epoch": 1.4577865600585937e-05,
      "step": 89286,
      "training_step_time": 6.705821990966797
    },
    {
      "epoch": 1.4578819274902345e-05,
      "model_forward_time": 1.7808239459991455,
      "step": 89287
    },
    {
      "epoch": 1.4578819274902345e-05,
      "step": 89287,
      "training_step_time": 6.787081956863403
    },
    {
      "epoch": 1.457977294921875e-05,
      "model_forward_time": 1.7808959484100342,
      "step": 89288
    },
    {
      "epoch": 1.457977294921875e-05,
      "step": 89288,
      "training_step_time": 6.791090965270996
    },
    {
      "epoch": 1.4580726623535156e-05,
      "model_forward_time": 1.8123860359191895,
      "step": 89289
    },
    {
      "epoch": 1.4580726623535156e-05,
      "step": 89289,
      "training_step_time": 7.442831993103027
    },
    {
      "epoch": 1.4581680297851562e-05,
      "grad_norm": 0.09185993671417236,
      "learning_rate": 3.103896279143359e-07,
      "loss": 0.0517,
      "step": 89290
    },
    {
      "epoch": 1.4581680297851562e-05,
      "model_forward_time": 1.7721199989318848,
      "step": 89290
    },
    {
      "dynamics_loss_avg": 0.0554419482126832,
      "epoch": 1.4581680297851562e-05,
      "step": 89290
    },
    {
      "action_loss_avg": 0.0015456670837011189,
      "epoch": 1.4581680297851562e-05,
      "step": 89290
    },
    {
      "epoch": 1.4581680297851562e-05,
      "step": 89290,
      "training_step_time": 8.786593198776245
    },
    {
      "epoch": 1.4582633972167969e-05,
      "model_forward_time": 2.8612353801727295,
      "step": 89291
    },
    {
      "epoch": 1.4582633972167969e-05,
      "step": 89291,
      "training_step_time": 10.94287395477295
    },
    {
      "epoch": 1.4583587646484375e-05,
      "model_forward_time": 1.7828595638275146,
      "step": 89292
    },
    {
      "epoch": 1.4583587646484375e-05,
      "step": 89292,
      "training_step_time": 9.257327318191528
    },
    {
      "epoch": 1.458454132080078e-05,
      "model_forward_time": 1.7960634231567383,
      "step": 89293
    },
    {
      "epoch": 1.458454132080078e-05,
      "step": 89293,
      "training_step_time": 7.6365368366241455
    },
    {
      "epoch": 1.4585494995117188e-05,
      "model_forward_time": 1.78226637840271,
      "step": 89294
    },
    {
      "epoch": 1.4585494995117188e-05,
      "step": 89294,
      "training_step_time": 7.617884397506714
    },
    {
      "epoch": 1.4586448669433594e-05,
      "model_forward_time": 1.7998285293579102,
      "step": 89295
    },
    {
      "epoch": 1.4586448669433594e-05,
      "step": 89295,
      "training_step_time": 7.71955132484436
    },
    {
      "epoch": 1.458740234375e-05,
      "model_forward_time": 1.801417589187622,
      "step": 89296
    },
    {
      "epoch": 1.458740234375e-05,
      "step": 89296,
      "training_step_time": 6.735503196716309
    },
    {
      "epoch": 1.4588356018066407e-05,
      "model_forward_time": 1.8014154434204102,
      "step": 89297
    },
    {
      "epoch": 1.4588356018066407e-05,
      "step": 89297,
      "training_step_time": 6.747006893157959
    },
    {
      "epoch": 1.4589309692382813e-05,
      "model_forward_time": 1.8050949573516846,
      "step": 89298
    },
    {
      "epoch": 1.4589309692382813e-05,
      "step": 89298,
      "training_step_time": 7.399420261383057
    },
    {
      "epoch": 1.4590263366699218e-05,
      "model_forward_time": 1.7862703800201416,
      "step": 89299
    },
    {
      "epoch": 1.4590263366699218e-05,
      "step": 89299,
      "training_step_time": 7.749878644943237
    },
    {
      "epoch": 1.4591217041015626e-05,
      "grad_norm": 0.08029009401798248,
      "learning_rate": 3.0981638487429244e-07,
      "loss": 0.0533,
      "step": 89300
    },
    {
      "epoch": 1.4591217041015626e-05,
      "model_forward_time": 2.9140429496765137,
      "step": 89300
    },
    {
      "dynamics_loss_avg": 0.04942044876515865,
      "epoch": 1.4591217041015626e-05,
      "step": 89300
    },
    {
      "action_loss_avg": 0.0014058147557079793,
      "epoch": 1.4591217041015626e-05,
      "step": 89300
    },
    {
      "epoch": 1.4591217041015626e-05,
      "step": 89300,
      "training_step_time": 8.309553384780884
    },
    {
      "epoch": 1.4592170715332031e-05,
      "model_forward_time": 1.8024945259094238,
      "step": 89301
    },
    {
      "epoch": 1.4592170715332031e-05,
      "step": 89301,
      "training_step_time": 9.543056011199951
    },
    {
      "epoch": 1.4593124389648437e-05,
      "model_forward_time": 1.8061254024505615,
      "step": 89302
    },
    {
      "epoch": 1.4593124389648437e-05,
      "step": 89302,
      "training_step_time": 7.682585000991821
    },
    {
      "epoch": 1.4594078063964845e-05,
      "model_forward_time": 1.8029217720031738,
      "step": 89303
    },
    {
      "epoch": 1.4594078063964845e-05,
      "step": 89303,
      "training_step_time": 7.644488334655762
    },
    {
      "epoch": 1.459503173828125e-05,
      "model_forward_time": 1.7987771034240723,
      "step": 89304
    },
    {
      "epoch": 1.459503173828125e-05,
      "step": 89304,
      "training_step_time": 7.256011486053467
    },
    {
      "epoch": 1.4595985412597656e-05,
      "model_forward_time": 1.7979755401611328,
      "step": 89305
    },
    {
      "epoch": 1.4595985412597656e-05,
      "step": 89305,
      "training_step_time": 7.287140607833862
    },
    {
      "epoch": 1.4596939086914062e-05,
      "model_forward_time": 1.7946906089782715,
      "step": 89306
    },
    {
      "epoch": 1.4596939086914062e-05,
      "step": 89306,
      "training_step_time": 6.7810139656066895
    },
    {
      "epoch": 1.4597892761230469e-05,
      "model_forward_time": 1.8042356967926025,
      "step": 89307
    },
    {
      "epoch": 1.4597892761230469e-05,
      "step": 89307,
      "training_step_time": 7.37912654876709
    },
    {
      "epoch": 1.4598846435546875e-05,
      "model_forward_time": 1.798426866531372,
      "step": 89308
    },
    {
      "epoch": 1.4598846435546875e-05,
      "step": 89308,
      "training_step_time": 7.688125848770142
    },
    {
      "epoch": 1.459980010986328e-05,
      "model_forward_time": 2.9034979343414307,
      "step": 89309
    },
    {
      "epoch": 1.459980010986328e-05,
      "step": 89309,
      "training_step_time": 9.396348714828491
    },
    {
      "epoch": 1.4600753784179688e-05,
      "grad_norm": 0.08917088061571121,
      "learning_rate": 3.092436547456767e-07,
      "loss": 0.054,
      "step": 89310
    },
    {
      "epoch": 1.4600753784179688e-05,
      "model_forward_time": 1.8056581020355225,
      "step": 89310
    },
    {
      "dynamics_loss_avg": 0.05995600465685129,
      "epoch": 1.4600753784179688e-05,
      "step": 89310
    },
    {
      "action_loss_avg": 0.0008357125392649323,
      "epoch": 1.4600753784179688e-05,
      "step": 89310
    },
    {
      "epoch": 1.4600753784179688e-05,
      "step": 89310,
      "training_step_time": 9.410017490386963
    },
    {
      "epoch": 1.4601707458496094e-05,
      "model_forward_time": 1.8122859001159668,
      "step": 89311
    },
    {
      "epoch": 1.4601707458496094e-05,
      "step": 89311,
      "training_step_time": 7.820540428161621
    },
    {
      "epoch": 1.46026611328125e-05,
      "model_forward_time": 1.7981886863708496,
      "step": 89312
    },
    {
      "epoch": 1.46026611328125e-05,
      "step": 89312,
      "training_step_time": 7.694480657577515
    },
    {
      "epoch": 1.4603614807128907e-05,
      "model_forward_time": 1.785484790802002,
      "step": 89313
    },
    {
      "epoch": 1.4603614807128907e-05,
      "step": 89313,
      "training_step_time": 7.331648588180542
    },
    {
      "epoch": 1.4604568481445313e-05,
      "model_forward_time": 1.8021454811096191,
      "step": 89314
    },
    {
      "epoch": 1.4604568481445313e-05,
      "step": 89314,
      "training_step_time": 7.3451457023620605
    },
    {
      "epoch": 1.4605522155761718e-05,
      "model_forward_time": 1.8058826923370361,
      "step": 89315
    },
    {
      "epoch": 1.4605522155761718e-05,
      "step": 89315,
      "training_step_time": 6.8164708614349365
    },
    {
      "epoch": 1.4606475830078126e-05,
      "model_forward_time": 1.8031864166259766,
      "step": 89316
    },
    {
      "epoch": 1.4606475830078126e-05,
      "step": 89316,
      "training_step_time": 7.531975030899048
    },
    {
      "epoch": 1.4607429504394531e-05,
      "model_forward_time": 1.8022897243499756,
      "step": 89317
    },
    {
      "epoch": 1.4607429504394531e-05,
      "step": 89317,
      "training_step_time": 7.572012662887573
    },
    {
      "epoch": 1.4608383178710937e-05,
      "model_forward_time": 2.925280809402466,
      "step": 89318
    },
    {
      "epoch": 1.4608383178710937e-05,
      "step": 89318,
      "training_step_time": 8.58427357673645
    },
    {
      "epoch": 1.4609336853027345e-05,
      "model_forward_time": 1.8042011260986328,
      "step": 89319
    },
    {
      "epoch": 1.4609336853027345e-05,
      "step": 89319,
      "training_step_time": 9.237542152404785
    },
    {
      "epoch": 1.461029052734375e-05,
      "grad_norm": 0.09460192173719406,
      "learning_rate": 3.0867143759112375e-07,
      "loss": 0.0545,
      "step": 89320
    },
    {
      "epoch": 1.461029052734375e-05,
      "model_forward_time": 1.7980608940124512,
      "step": 89320
    },
    {
      "dynamics_loss_avg": 0.06258787550032138,
      "epoch": 1.461029052734375e-05,
      "step": 89320
    },
    {
      "action_loss_avg": 0.0011446390708442777,
      "epoch": 1.461029052734375e-05,
      "step": 89320
    },
    {
      "epoch": 1.461029052734375e-05,
      "step": 89320,
      "training_step_time": 7.691481828689575
    },
    {
      "epoch": 1.4611244201660156e-05,
      "model_forward_time": 1.8013269901275635,
      "step": 89321
    },
    {
      "epoch": 1.4611244201660156e-05,
      "step": 89321,
      "training_step_time": 7.622211456298828
    },
    {
      "epoch": 1.4612197875976562e-05,
      "model_forward_time": 1.8009672164916992,
      "step": 89322
    },
    {
      "epoch": 1.4612197875976562e-05,
      "step": 89322,
      "training_step_time": 7.049177169799805
    },
    {
      "epoch": 1.4613151550292969e-05,
      "model_forward_time": 1.8107788562774658,
      "step": 89323
    },
    {
      "epoch": 1.4613151550292969e-05,
      "step": 89323,
      "training_step_time": 7.2920005321502686
    },
    {
      "epoch": 1.4614105224609375e-05,
      "model_forward_time": 1.8039562702178955,
      "step": 89324
    },
    {
      "epoch": 1.4614105224609375e-05,
      "step": 89324,
      "training_step_time": 6.731848239898682
    },
    {
      "epoch": 1.461505889892578e-05,
      "model_forward_time": 1.8036890029907227,
      "step": 89325
    },
    {
      "epoch": 1.461505889892578e-05,
      "step": 89325,
      "training_step_time": 7.406147003173828
    },
    {
      "epoch": 1.4616012573242188e-05,
      "model_forward_time": 1.7945728302001953,
      "step": 89326
    },
    {
      "epoch": 1.4616012573242188e-05,
      "step": 89326,
      "training_step_time": 7.528178453445435
    },
    {
      "epoch": 1.4616966247558594e-05,
      "model_forward_time": 2.9211626052856445,
      "step": 89327
    },
    {
      "epoch": 1.4616966247558594e-05,
      "step": 89327,
      "training_step_time": 7.672433853149414
    },
    {
      "epoch": 1.4617919921875e-05,
      "model_forward_time": 1.784278154373169,
      "step": 89328
    },
    {
      "epoch": 1.4617919921875e-05,
      "step": 89328,
      "training_step_time": 10.07864260673523
    },
    {
      "epoch": 1.4618873596191407e-05,
      "model_forward_time": 1.8044450283050537,
      "step": 89329
    },
    {
      "epoch": 1.4618873596191407e-05,
      "step": 89329,
      "training_step_time": 7.757049798965454
    },
    {
      "epoch": 1.4619827270507813e-05,
      "grad_norm": 0.08812956511974335,
      "learning_rate": 3.0809973347320776e-07,
      "loss": 0.0566,
      "step": 89330
    },
    {
      "epoch": 1.4619827270507813e-05,
      "model_forward_time": 1.7610034942626953,
      "step": 89330
    },
    {
      "dynamics_loss_avg": 0.04353702459484339,
      "epoch": 1.4619827270507813e-05,
      "step": 89330
    },
    {
      "action_loss_avg": 0.0014898933179210872,
      "epoch": 1.4619827270507813e-05,
      "step": 89330
    },
    {
      "epoch": 1.4619827270507813e-05,
      "step": 89330,
      "training_step_time": 8.727001667022705
    },
    {
      "epoch": 1.4620780944824218e-05,
      "model_forward_time": 1.7893178462982178,
      "step": 89331
    },
    {
      "epoch": 1.4620780944824218e-05,
      "step": 89331,
      "training_step_time": 7.597445964813232
    },
    {
      "epoch": 1.4621734619140626e-05,
      "model_forward_time": 1.7682151794433594,
      "step": 89332
    },
    {
      "epoch": 1.4621734619140626e-05,
      "step": 89332,
      "training_step_time": 7.319141626358032
    },
    {
      "epoch": 1.4622688293457031e-05,
      "model_forward_time": 1.780519962310791,
      "step": 89333
    },
    {
      "epoch": 1.4622688293457031e-05,
      "step": 89333,
      "training_step_time": 6.751632452011108
    },
    {
      "epoch": 1.4623641967773437e-05,
      "model_forward_time": 1.7844386100769043,
      "step": 89334
    },
    {
      "epoch": 1.4623641967773437e-05,
      "step": 89334,
      "training_step_time": 7.171903848648071
    },
    {
      "epoch": 1.4624595642089845e-05,
      "model_forward_time": 1.7725045680999756,
      "step": 89335
    },
    {
      "epoch": 1.4624595642089845e-05,
      "step": 89335,
      "training_step_time": 7.149585008621216
    },
    {
      "epoch": 1.462554931640625e-05,
      "model_forward_time": 1.764512300491333,
      "step": 89336
    },
    {
      "epoch": 1.462554931640625e-05,
      "step": 89336,
      "training_step_time": 10.456955671310425
    },
    {
      "epoch": 1.4626502990722656e-05,
      "model_forward_time": 1.7801098823547363,
      "step": 89337
    },
    {
      "epoch": 1.4626502990722656e-05,
      "step": 89337,
      "training_step_time": 10.431769132614136
    },
    {
      "epoch": 1.4627456665039062e-05,
      "model_forward_time": 1.7894611358642578,
      "step": 89338
    },
    {
      "epoch": 1.4627456665039062e-05,
      "step": 89338,
      "training_step_time": 7.772403001785278
    },
    {
      "epoch": 1.4628410339355469e-05,
      "model_forward_time": 1.7889747619628906,
      "step": 89339
    },
    {
      "epoch": 1.4628410339355469e-05,
      "step": 89339,
      "training_step_time": 7.762120246887207
    },
    {
      "epoch": 1.4629364013671875e-05,
      "grad_norm": 0.08624117076396942,
      "learning_rate": 3.075285424544516e-07,
      "loss": 0.0552,
      "step": 89340
    },
    {
      "epoch": 1.4629364013671875e-05,
      "model_forward_time": 1.7778346538543701,
      "step": 89340
    },
    {
      "dynamics_loss_avg": 0.05537236891686916,
      "epoch": 1.4629364013671875e-05,
      "step": 89340
    },
    {
      "action_loss_avg": 0.001822382176760584,
      "epoch": 1.4629364013671875e-05,
      "step": 89340
    },
    {
      "epoch": 1.4629364013671875e-05,
      "step": 89340,
      "training_step_time": 7.635134220123291
    },
    {
      "epoch": 1.463031768798828e-05,
      "model_forward_time": 1.7768440246582031,
      "step": 89341
    },
    {
      "epoch": 1.463031768798828e-05,
      "step": 89341,
      "training_step_time": 7.197077751159668
    },
    {
      "epoch": 1.4631271362304688e-05,
      "model_forward_time": 1.7883236408233643,
      "step": 89342
    },
    {
      "epoch": 1.4631271362304688e-05,
      "step": 89342,
      "training_step_time": 6.750152349472046
    },
    {
      "epoch": 1.4632225036621094e-05,
      "model_forward_time": 1.7894408702850342,
      "step": 89343
    },
    {
      "epoch": 1.4632225036621094e-05,
      "step": 89343,
      "training_step_time": 7.533729553222656
    },
    {
      "epoch": 1.46331787109375e-05,
      "model_forward_time": 1.7734463214874268,
      "step": 89344
    },
    {
      "epoch": 1.46331787109375e-05,
      "step": 89344,
      "training_step_time": 7.2403504848480225
    },
    {
      "epoch": 1.4634132385253907e-05,
      "model_forward_time": 1.7597510814666748,
      "step": 89345
    },
    {
      "epoch": 1.4634132385253907e-05,
      "step": 89345,
      "training_step_time": 7.6692795753479
    },
    {
      "epoch": 1.4635086059570313e-05,
      "model_forward_time": 2.8674652576446533,
      "step": 89346
    },
    {
      "epoch": 1.4635086059570313e-05,
      "step": 89346,
      "training_step_time": 10.726943016052246
    },
    {
      "epoch": 1.4636039733886718e-05,
      "model_forward_time": 1.779249668121338,
      "step": 89347
    },
    {
      "epoch": 1.4636039733886718e-05,
      "step": 89347,
      "training_step_time": 9.184020280838013
    },
    {
      "epoch": 1.4636993408203126e-05,
      "model_forward_time": 1.7868452072143555,
      "step": 89348
    },
    {
      "epoch": 1.4636993408203126e-05,
      "step": 89348,
      "training_step_time": 7.713907480239868
    },
    {
      "epoch": 1.4637947082519531e-05,
      "model_forward_time": 1.7813410758972168,
      "step": 89349
    },
    {
      "epoch": 1.4637947082519531e-05,
      "step": 89349,
      "training_step_time": 8.597708225250244
    },
    {
      "epoch": 1.4638900756835937e-05,
      "grad_norm": 0.08217203617095947,
      "learning_rate": 3.069578645973176e-07,
      "loss": 0.0571,
      "step": 89350
    },
    {
      "epoch": 1.4638900756835937e-05,
      "model_forward_time": 1.777486801147461,
      "step": 89350
    },
    {
      "dynamics_loss_avg": 0.07014935296028853,
      "epoch": 1.4638900756835937e-05,
      "step": 89350
    },
    {
      "action_loss_avg": 0.0012108658935176208,
      "epoch": 1.4638900756835937e-05,
      "step": 89350
    },
    {
      "epoch": 1.4638900756835937e-05,
      "step": 89350,
      "training_step_time": 7.3091535568237305
    },
    {
      "epoch": 1.4639854431152345e-05,
      "model_forward_time": 1.771430492401123,
      "step": 89351
    },
    {
      "epoch": 1.4639854431152345e-05,
      "step": 89351,
      "training_step_time": 7.32780385017395
    },
    {
      "epoch": 1.464080810546875e-05,
      "model_forward_time": 1.7917368412017822,
      "step": 89352
    },
    {
      "epoch": 1.464080810546875e-05,
      "step": 89352,
      "training_step_time": 7.581803798675537
    },
    {
      "epoch": 1.4641761779785156e-05,
      "model_forward_time": 1.7684211730957031,
      "step": 89353
    },
    {
      "epoch": 1.4641761779785156e-05,
      "step": 89353,
      "training_step_time": 7.271646022796631
    },
    {
      "epoch": 1.4642715454101562e-05,
      "model_forward_time": 1.7758910655975342,
      "step": 89354
    },
    {
      "epoch": 1.4642715454101562e-05,
      "step": 89354,
      "training_step_time": 8.715273380279541
    },
    {
      "epoch": 1.464366912841797e-05,
      "model_forward_time": 2.964374542236328,
      "step": 89355
    },
    {
      "epoch": 1.464366912841797e-05,
      "step": 89355,
      "training_step_time": 9.729133367538452
    },
    {
      "epoch": 1.4644622802734375e-05,
      "model_forward_time": 1.7976043224334717,
      "step": 89356
    },
    {
      "epoch": 1.4644622802734375e-05,
      "step": 89356,
      "training_step_time": 8.784591913223267
    },
    {
      "epoch": 1.464557647705078e-05,
      "model_forward_time": 1.7957749366760254,
      "step": 89357
    },
    {
      "epoch": 1.464557647705078e-05,
      "step": 89357,
      "training_step_time": 7.702854156494141
    },
    {
      "epoch": 1.4646530151367188e-05,
      "model_forward_time": 1.7860097885131836,
      "step": 89358
    },
    {
      "epoch": 1.4646530151367188e-05,
      "step": 89358,
      "training_step_time": 8.6835298538208
    },
    {
      "epoch": 1.4647483825683594e-05,
      "model_forward_time": 1.7625198364257812,
      "step": 89359
    },
    {
      "epoch": 1.4647483825683594e-05,
      "step": 89359,
      "training_step_time": 7.553169250488281
    },
    {
      "epoch": 1.46484375e-05,
      "grad_norm": 0.07639970630407333,
      "learning_rate": 3.0638769996421736e-07,
      "loss": 0.0571,
      "step": 89360
    },
    {
      "epoch": 1.46484375e-05,
      "model_forward_time": 1.7875850200653076,
      "step": 89360
    },
    {
      "dynamics_loss_avg": 0.06480159685015678,
      "epoch": 1.46484375e-05,
      "step": 89360
    },
    {
      "action_loss_avg": 0.0024114762840326876,
      "epoch": 1.46484375e-05,
      "step": 89360
    },
    {
      "epoch": 1.46484375e-05,
      "step": 89360,
      "training_step_time": 7.242355108261108
    },
    {
      "epoch": 1.4649391174316407e-05,
      "model_forward_time": 1.770462989807129,
      "step": 89361
    },
    {
      "epoch": 1.4649391174316407e-05,
      "step": 89361,
      "training_step_time": 7.610321044921875
    },
    {
      "epoch": 1.4650344848632813e-05,
      "model_forward_time": 1.7776587009429932,
      "step": 89362
    },
    {
      "epoch": 1.4650344848632813e-05,
      "step": 89362,
      "training_step_time": 7.725334405899048
    },
    {
      "epoch": 1.4651298522949218e-05,
      "model_forward_time": 1.8183977603912354,
      "step": 89363
    },
    {
      "epoch": 1.4651298522949218e-05,
      "step": 89363,
      "training_step_time": 7.639575719833374
    },
    {
      "epoch": 1.4652252197265626e-05,
      "model_forward_time": 2.9163382053375244,
      "step": 89364
    },
    {
      "epoch": 1.4652252197265626e-05,
      "step": 89364,
      "training_step_time": 9.826260328292847
    },
    {
      "epoch": 1.4653205871582031e-05,
      "model_forward_time": 1.792208194732666,
      "step": 89365
    },
    {
      "epoch": 1.4653205871582031e-05,
      "step": 89365,
      "training_step_time": 8.642799615859985
    },
    {
      "epoch": 1.4654159545898437e-05,
      "model_forward_time": 1.7286148071289062,
      "step": 89366
    },
    {
      "epoch": 1.4654159545898437e-05,
      "step": 89366,
      "training_step_time": 8.746142148971558
    },
    {
      "epoch": 1.4655113220214845e-05,
      "model_forward_time": 1.7351047992706299,
      "step": 89367
    },
    {
      "epoch": 1.4655113220214845e-05,
      "step": 89367,
      "training_step_time": 7.70561671257019
    },
    {
      "epoch": 1.465606689453125e-05,
      "model_forward_time": 1.733135461807251,
      "step": 89368
    },
    {
      "epoch": 1.465606689453125e-05,
      "step": 89368,
      "training_step_time": 8.528123140335083
    },
    {
      "epoch": 1.4657020568847656e-05,
      "model_forward_time": 1.7264478206634521,
      "step": 89369
    },
    {
      "epoch": 1.4657020568847656e-05,
      "step": 89369,
      "training_step_time": 7.3721394538879395
    },
    {
      "epoch": 1.4657974243164062e-05,
      "grad_norm": 0.07910550385713577,
      "learning_rate": 3.0581804861750055e-07,
      "loss": 0.0586,
      "step": 89370
    },
    {
      "epoch": 1.4657974243164062e-05,
      "model_forward_time": 1.7326605319976807,
      "step": 89370
    },
    {
      "dynamics_loss_avg": 0.0731479473412037,
      "epoch": 1.4657974243164062e-05,
      "step": 89370
    },
    {
      "action_loss_avg": 0.001533208356704563,
      "epoch": 1.4657974243164062e-05,
      "step": 89370
    },
    {
      "epoch": 1.4657974243164062e-05,
      "step": 89370,
      "training_step_time": 7.649914026260376
    },
    {
      "epoch": 1.465892791748047e-05,
      "model_forward_time": 1.7375984191894531,
      "step": 89371
    },
    {
      "epoch": 1.465892791748047e-05,
      "step": 89371,
      "training_step_time": 6.751443147659302
    },
    {
      "epoch": 1.4659881591796875e-05,
      "model_forward_time": 1.7349598407745361,
      "step": 89372
    },
    {
      "epoch": 1.4659881591796875e-05,
      "step": 89372,
      "training_step_time": 8.347305297851562
    },
    {
      "epoch": 1.466083526611328e-05,
      "model_forward_time": 2.3548388481140137,
      "step": 89373
    },
    {
      "epoch": 1.466083526611328e-05,
      "step": 89373,
      "training_step_time": 8.42423391342163
    },
    {
      "epoch": 1.4661788940429688e-05,
      "model_forward_time": 1.7443451881408691,
      "step": 89374
    },
    {
      "epoch": 1.4661788940429688e-05,
      "step": 89374,
      "training_step_time": 10.323875427246094
    },
    {
      "epoch": 1.4662742614746094e-05,
      "model_forward_time": 1.7275278568267822,
      "step": 89375
    },
    {
      "epoch": 1.4662742614746094e-05,
      "step": 89375,
      "training_step_time": 8.445679664611816
    },
    {
      "epoch": 1.46636962890625e-05,
      "model_forward_time": 1.7302544116973877,
      "step": 89376
    },
    {
      "epoch": 1.46636962890625e-05,
      "step": 89376,
      "training_step_time": 7.804543495178223
    },
    {
      "epoch": 1.4664649963378907e-05,
      "model_forward_time": 1.7216565608978271,
      "step": 89377
    },
    {
      "epoch": 1.4664649963378907e-05,
      "step": 89377,
      "training_step_time": 8.663019180297852
    },
    {
      "epoch": 1.4665603637695313e-05,
      "model_forward_time": 1.7457952499389648,
      "step": 89378
    },
    {
      "epoch": 1.4665603637695313e-05,
      "step": 89378,
      "training_step_time": 7.322112798690796
    },
    {
      "epoch": 1.4666557312011718e-05,
      "model_forward_time": 1.3691790103912354,
      "step": 89379
    },
    {
      "epoch": 1.4666557312011718e-05,
      "step": 89379,
      "training_step_time": 7.545920133590698
    },
    {
      "epoch": 1.4667510986328126e-05,
      "grad_norm": 0.0934050977230072,
      "learning_rate": 3.052489106194645e-07,
      "loss": 0.0584,
      "step": 89380
    },
    {
      "epoch": 1.4667510986328126e-05,
      "model_forward_time": 1.7282519340515137,
      "step": 89380
    },
    {
      "dynamics_loss_avg": 0.04673772845417261,
      "epoch": 1.4667510986328126e-05,
      "step": 89380
    },
    {
      "action_loss_avg": 0.001762825778860133,
      "epoch": 1.4667510986328126e-05,
      "step": 89380
    },
    {
      "epoch": 1.4667510986328126e-05,
      "step": 89380,
      "training_step_time": 6.733670234680176
    },
    {
      "epoch": 1.4668464660644531e-05,
      "model_forward_time": 1.7306840419769287,
      "step": 89381
    },
    {
      "epoch": 1.4668464660644531e-05,
      "step": 89381,
      "training_step_time": 7.2186784744262695
    },
    {
      "epoch": 1.4669418334960937e-05,
      "model_forward_time": 2.3685057163238525,
      "step": 89382
    },
    {
      "epoch": 1.4669418334960937e-05,
      "step": 89382,
      "training_step_time": 7.739596128463745
    },
    {
      "epoch": 1.4670372009277343e-05,
      "model_forward_time": 1.7479486465454102,
      "step": 89383
    },
    {
      "epoch": 1.4670372009277343e-05,
      "step": 89383,
      "training_step_time": 7.762639999389648
    },
    {
      "epoch": 1.467132568359375e-05,
      "model_forward_time": 1.734764575958252,
      "step": 89384
    },
    {
      "epoch": 1.467132568359375e-05,
      "step": 89384,
      "training_step_time": 8.761569023132324
    },
    {
      "epoch": 1.4672279357910156e-05,
      "model_forward_time": 1.736163854598999,
      "step": 89385
    },
    {
      "epoch": 1.4672279357910156e-05,
      "step": 89385,
      "training_step_time": 7.749192476272583
    },
    {
      "epoch": 1.4673233032226562e-05,
      "model_forward_time": 1.739501953125,
      "step": 89386
    },
    {
      "epoch": 1.4673233032226562e-05,
      "step": 89386,
      "training_step_time": 7.679556846618652
    },
    {
      "epoch": 1.467418670654297e-05,
      "model_forward_time": 1.7364182472229004,
      "step": 89387
    },
    {
      "epoch": 1.467418670654297e-05,
      "step": 89387,
      "training_step_time": 7.530251502990723
    },
    {
      "epoch": 1.4675140380859375e-05,
      "model_forward_time": 1.7313549518585205,
      "step": 89388
    },
    {
      "epoch": 1.4675140380859375e-05,
      "step": 89388,
      "training_step_time": 7.584627628326416
    },
    {
      "epoch": 1.467609405517578e-05,
      "model_forward_time": 1.7300455570220947,
      "step": 89389
    },
    {
      "epoch": 1.467609405517578e-05,
      "step": 89389,
      "training_step_time": 6.781839370727539
    },
    {
      "epoch": 1.4677047729492188e-05,
      "grad_norm": 0.12705953419208527,
      "learning_rate": 3.0468028603234944e-07,
      "loss": 0.0582,
      "step": 89390
    },
    {
      "epoch": 1.4677047729492188e-05,
      "model_forward_time": 1.7382073402404785,
      "step": 89390
    },
    {
      "dynamics_loss_avg": 0.03996898168697953,
      "epoch": 1.4677047729492188e-05,
      "step": 89390
    },
    {
      "action_loss_avg": 0.0008932126424042508,
      "epoch": 1.4677047729492188e-05,
      "step": 89390
    },
    {
      "epoch": 1.4677047729492188e-05,
      "step": 89390,
      "training_step_time": 7.260649919509888
    },
    {
      "epoch": 1.4678001403808594e-05,
      "model_forward_time": 2.3909542560577393,
      "step": 89391
    },
    {
      "epoch": 1.4678001403808594e-05,
      "step": 89391,
      "training_step_time": 8.548110246658325
    },
    {
      "epoch": 1.4678955078125e-05,
      "model_forward_time": 1.741936445236206,
      "step": 89392
    },
    {
      "epoch": 1.4678955078125e-05,
      "step": 89392,
      "training_step_time": 8.735836505889893
    },
    {
      "epoch": 1.4679908752441407e-05,
      "model_forward_time": 1.7377266883850098,
      "step": 89393
    },
    {
      "epoch": 1.4679908752441407e-05,
      "step": 89393,
      "training_step_time": 8.94961166381836
    },
    {
      "epoch": 1.4680862426757813e-05,
      "model_forward_time": 1.7316291332244873,
      "step": 89394
    },
    {
      "epoch": 1.4680862426757813e-05,
      "step": 89394,
      "training_step_time": 7.797042608261108
    },
    {
      "epoch": 1.4681816101074218e-05,
      "model_forward_time": 1.7324151992797852,
      "step": 89395
    },
    {
      "epoch": 1.4681816101074218e-05,
      "step": 89395,
      "training_step_time": 7.739722967147827
    },
    {
      "epoch": 1.4682769775390626e-05,
      "model_forward_time": 1.7418289184570312,
      "step": 89396
    },
    {
      "epoch": 1.4682769775390626e-05,
      "step": 89396,
      "training_step_time": 7.49120569229126
    },
    {
      "epoch": 1.4683723449707031e-05,
      "model_forward_time": 1.7371914386749268,
      "step": 89397
    },
    {
      "epoch": 1.4683723449707031e-05,
      "step": 89397,
      "training_step_time": 7.662758111953735
    },
    {
      "epoch": 1.4684677124023437e-05,
      "model_forward_time": 1.7317142486572266,
      "step": 89398
    },
    {
      "epoch": 1.4684677124023437e-05,
      "step": 89398,
      "training_step_time": 6.773916006088257
    },
    {
      "epoch": 1.4685630798339843e-05,
      "model_forward_time": 1.7304117679595947,
      "step": 89399
    },
    {
      "epoch": 1.4685630798339843e-05,
      "step": 89399,
      "training_step_time": 7.300015926361084
    },
    {
      "epoch": 1.468658447265625e-05,
      "grad_norm": 0.14002196490764618,
      "learning_rate": 3.0411217491833897e-07,
      "loss": 0.0569,
      "step": 89400
    },
    {
      "epoch": 1.468658447265625e-05,
      "model_forward_time": 2.4369914531707764,
      "step": 89400
    },
    {
      "dynamics_loss_avg": 0.05219164006412029,
      "epoch": 1.468658447265625e-05,
      "step": 89400
    },
    {
      "action_loss_avg": 0.0014674057063530199,
      "epoch": 1.468658447265625e-05,
      "step": 89400
    },
    {
      "epoch": 1.468658447265625e-05,
      "step": 89400,
      "training_step_time": 7.673338890075684
    },
    {
      "epoch": 1.4687538146972656e-05,
      "model_forward_time": 1.7287323474884033,
      "step": 89401
    },
    {
      "epoch": 1.4687538146972656e-05,
      "step": 89401,
      "training_step_time": 9.204717636108398
    },
    {
      "epoch": 1.4688491821289062e-05,
      "model_forward_time": 1.7450740337371826,
      "step": 89402
    },
    {
      "epoch": 1.4688491821289062e-05,
      "step": 89402,
      "training_step_time": 9.621399164199829
    },
    {
      "epoch": 1.468944549560547e-05,
      "model_forward_time": 1.7333667278289795,
      "step": 89403
    },
    {
      "epoch": 1.468944549560547e-05,
      "step": 89403,
      "training_step_time": 8.708413362503052
    },
    {
      "epoch": 1.4690399169921875e-05,
      "model_forward_time": 1.7321336269378662,
      "step": 89404
    },
    {
      "epoch": 1.4690399169921875e-05,
      "step": 89404,
      "training_step_time": 7.692523956298828
    },
    {
      "epoch": 1.469135284423828e-05,
      "model_forward_time": 1.738396167755127,
      "step": 89405
    },
    {
      "epoch": 1.469135284423828e-05,
      "step": 89405,
      "training_step_time": 7.6337971687316895
    },
    {
      "epoch": 1.4692306518554688e-05,
      "model_forward_time": 1.7375879287719727,
      "step": 89406
    },
    {
      "epoch": 1.4692306518554688e-05,
      "step": 89406,
      "training_step_time": 7.61918044090271
    },
    {
      "epoch": 1.4693260192871094e-05,
      "model_forward_time": 1.7324621677398682,
      "step": 89407
    },
    {
      "epoch": 1.4693260192871094e-05,
      "step": 89407,
      "training_step_time": 6.782861948013306
    },
    {
      "epoch": 1.46942138671875e-05,
      "model_forward_time": 1.730677604675293,
      "step": 89408
    },
    {
      "epoch": 1.46942138671875e-05,
      "step": 89408,
      "training_step_time": 7.3584911823272705
    },
    {
      "epoch": 1.4695167541503907e-05,
      "model_forward_time": 2.424567937850952,
      "step": 89409
    },
    {
      "epoch": 1.4695167541503907e-05,
      "step": 89409,
      "training_step_time": 7.697100639343262
    },
    {
      "epoch": 1.4696121215820313e-05,
      "grad_norm": 0.09437508881092072,
      "learning_rate": 3.0354457733955953e-07,
      "loss": 0.0586,
      "step": 89410
    },
    {
      "epoch": 1.4696121215820313e-05,
      "model_forward_time": 1.7429027557373047,
      "step": 89410
    },
    {
      "dynamics_loss_avg": 0.04959602812305093,
      "epoch": 1.4696121215820313e-05,
      "step": 89410
    },
    {
      "action_loss_avg": 0.001477202249225229,
      "epoch": 1.4696121215820313e-05,
      "step": 89410
    },
    {
      "epoch": 1.4696121215820313e-05,
      "step": 89410,
      "training_step_time": 8.80812954902649
    },
    {
      "epoch": 1.4697074890136718e-05,
      "model_forward_time": 1.7466068267822266,
      "step": 89411
    },
    {
      "epoch": 1.4697074890136718e-05,
      "step": 89411,
      "training_step_time": 10.237897396087646
    },
    {
      "epoch": 1.4698028564453126e-05,
      "model_forward_time": 1.7402591705322266,
      "step": 89412
    },
    {
      "epoch": 1.4698028564453126e-05,
      "step": 89412,
      "training_step_time": 7.7774341106414795
    },
    {
      "epoch": 1.4698982238769531e-05,
      "model_forward_time": 1.7334885597229004,
      "step": 89413
    },
    {
      "epoch": 1.4698982238769531e-05,
      "step": 89413,
      "training_step_time": 8.835292100906372
    },
    {
      "epoch": 1.4699935913085937e-05,
      "model_forward_time": 1.731942892074585,
      "step": 89414
    },
    {
      "epoch": 1.4699935913085937e-05,
      "step": 89414,
      "training_step_time": 7.623591184616089
    },
    {
      "epoch": 1.4700889587402343e-05,
      "model_forward_time": 1.7331061363220215,
      "step": 89415
    },
    {
      "epoch": 1.4700889587402343e-05,
      "step": 89415,
      "training_step_time": 7.575918912887573
    },
    {
      "epoch": 1.470184326171875e-05,
      "model_forward_time": 1.7355177402496338,
      "step": 89416
    },
    {
      "epoch": 1.470184326171875e-05,
      "step": 89416,
      "training_step_time": 6.788603782653809
    },
    {
      "epoch": 1.4702796936035156e-05,
      "model_forward_time": 1.7352774143218994,
      "step": 89417
    },
    {
      "epoch": 1.4702796936035156e-05,
      "step": 89417,
      "training_step_time": 6.77262544631958
    },
    {
      "epoch": 1.4703750610351562e-05,
      "model_forward_time": 2.3826406002044678,
      "step": 89418
    },
    {
      "epoch": 1.4703750610351562e-05,
      "step": 89418,
      "training_step_time": 7.6265058517456055
    },
    {
      "epoch": 1.470470428466797e-05,
      "model_forward_time": 1.7328062057495117,
      "step": 89419
    },
    {
      "epoch": 1.470470428466797e-05,
      "step": 89419,
      "training_step_time": 7.665355682373047
    },
    {
      "epoch": 1.4705657958984375e-05,
      "grad_norm": 0.08977357298135757,
      "learning_rate": 3.029774933580859e-07,
      "loss": 0.0585,
      "step": 89420
    },
    {
      "epoch": 1.4705657958984375e-05,
      "model_forward_time": 1.7425031661987305,
      "step": 89420
    },
    {
      "dynamics_loss_avg": 0.050400462746620175,
      "epoch": 1.4705657958984375e-05,
      "step": 89420
    },
    {
      "action_loss_avg": 0.0015584329725243152,
      "epoch": 1.4705657958984375e-05,
      "step": 89420
    },
    {
      "epoch": 1.4705657958984375e-05,
      "step": 89420,
      "training_step_time": 10.252296209335327
    },
    {
      "epoch": 1.470661163330078e-05,
      "model_forward_time": 1.7428944110870361,
      "step": 89421
    },
    {
      "epoch": 1.470661163330078e-05,
      "step": 89421,
      "training_step_time": 9.31258249282837
    },
    {
      "epoch": 1.4707565307617188e-05,
      "model_forward_time": 1.7350146770477295,
      "step": 89422
    },
    {
      "epoch": 1.4707565307617188e-05,
      "step": 89422,
      "training_step_time": 9.377167224884033
    },
    {
      "epoch": 1.4708518981933594e-05,
      "model_forward_time": 1.730093002319336,
      "step": 89423
    },
    {
      "epoch": 1.4708518981933594e-05,
      "step": 89423,
      "training_step_time": 7.762482643127441
    },
    {
      "epoch": 1.470947265625e-05,
      "model_forward_time": 1.7402124404907227,
      "step": 89424
    },
    {
      "epoch": 1.470947265625e-05,
      "step": 89424,
      "training_step_time": 7.486032485961914
    },
    {
      "epoch": 1.4710426330566407e-05,
      "model_forward_time": 1.7180652618408203,
      "step": 89425
    },
    {
      "epoch": 1.4710426330566407e-05,
      "step": 89425,
      "training_step_time": 7.328535556793213
    },
    {
      "epoch": 1.4711380004882813e-05,
      "model_forward_time": 1.734266996383667,
      "step": 89426
    },
    {
      "epoch": 1.4711380004882813e-05,
      "step": 89426,
      "training_step_time": 6.765375137329102
    },
    {
      "epoch": 1.4712333679199218e-05,
      "model_forward_time": 1.7373778820037842,
      "step": 89427
    },
    {
      "epoch": 1.4712333679199218e-05,
      "step": 89427,
      "training_step_time": 7.2923901081085205
    },
    {
      "epoch": 1.4713287353515626e-05,
      "model_forward_time": 2.3724634647369385,
      "step": 89428
    },
    {
      "epoch": 1.4713287353515626e-05,
      "step": 89428,
      "training_step_time": 9.577672243118286
    },
    {
      "epoch": 1.4714241027832032e-05,
      "model_forward_time": 1.7278919219970703,
      "step": 89429
    },
    {
      "epoch": 1.4714241027832032e-05,
      "step": 89429,
      "training_step_time": 10.55745005607605
    },
    {
      "epoch": 1.4715194702148437e-05,
      "grad_norm": 0.11101022362709045,
      "learning_rate": 3.024109230359307e-07,
      "loss": 0.0586,
      "step": 89430
    },
    {
      "epoch": 1.4715194702148437e-05,
      "model_forward_time": 1.733435869216919,
      "step": 89430
    },
    {
      "dynamics_loss_avg": 0.07506582513451576,
      "epoch": 1.4715194702148437e-05,
      "step": 89430
    },
    {
      "action_loss_avg": 0.0013130957551766187,
      "epoch": 1.4715194702148437e-05,
      "step": 89430
    },
    {
      "epoch": 1.4715194702148437e-05,
      "step": 89430,
      "training_step_time": 7.786257982254028
    },
    {
      "epoch": 1.4716148376464843e-05,
      "model_forward_time": 1.729431390762329,
      "step": 89431
    },
    {
      "epoch": 1.4716148376464843e-05,
      "step": 89431,
      "training_step_time": 7.739901542663574
    },
    {
      "epoch": 1.471710205078125e-05,
      "model_forward_time": 1.7346503734588623,
      "step": 89432
    },
    {
      "epoch": 1.471710205078125e-05,
      "step": 89432,
      "training_step_time": 7.712613821029663
    },
    {
      "epoch": 1.4718055725097656e-05,
      "model_forward_time": 1.7302830219268799,
      "step": 89433
    },
    {
      "epoch": 1.4718055725097656e-05,
      "step": 89433,
      "training_step_time": 8.283774375915527
    },
    {
      "epoch": 1.4719009399414062e-05,
      "model_forward_time": 1.7328884601593018,
      "step": 89434
    },
    {
      "epoch": 1.4719009399414062e-05,
      "step": 89434,
      "training_step_time": 7.3423542976379395
    },
    {
      "epoch": 1.471996307373047e-05,
      "model_forward_time": 1.728437900543213,
      "step": 89435
    },
    {
      "epoch": 1.471996307373047e-05,
      "step": 89435,
      "training_step_time": 6.853756904602051
    },
    {
      "epoch": 1.4720916748046875e-05,
      "model_forward_time": 1.73048734664917,
      "step": 89436
    },
    {
      "epoch": 1.4720916748046875e-05,
      "step": 89436,
      "training_step_time": 7.302743434906006
    },
    {
      "epoch": 1.472187042236328e-05,
      "model_forward_time": 2.386734962463379,
      "step": 89437
    },
    {
      "epoch": 1.472187042236328e-05,
      "step": 89437,
      "training_step_time": 8.66572618484497
    },
    {
      "epoch": 1.4722824096679688e-05,
      "model_forward_time": 1.7296698093414307,
      "step": 89438
    },
    {
      "epoch": 1.4722824096679688e-05,
      "step": 89438,
      "training_step_time": 9.528285026550293
    },
    {
      "epoch": 1.4723777770996094e-05,
      "model_forward_time": 1.7484900951385498,
      "step": 89439
    },
    {
      "epoch": 1.4723777770996094e-05,
      "step": 89439,
      "training_step_time": 7.76872992515564
    },
    {
      "epoch": 1.47247314453125e-05,
      "grad_norm": 0.09089349210262299,
      "learning_rate": 3.0184486643505537e-07,
      "loss": 0.0565,
      "step": 89440
    },
    {
      "epoch": 1.47247314453125e-05,
      "model_forward_time": 1.7320270538330078,
      "step": 89440
    },
    {
      "dynamics_loss_avg": 0.07102890387177467,
      "epoch": 1.47247314453125e-05,
      "step": 89440
    },
    {
      "action_loss_avg": 0.002331612689886242,
      "epoch": 1.47247314453125e-05,
      "step": 89440
    },
    {
      "epoch": 1.47247314453125e-05,
      "step": 89440,
      "training_step_time": 8.21237325668335
    },
    {
      "epoch": 1.4725685119628907e-05,
      "model_forward_time": 1.7354581356048584,
      "step": 89441
    },
    {
      "epoch": 1.4725685119628907e-05,
      "step": 89441,
      "training_step_time": 7.711288690567017
    },
    {
      "epoch": 1.4726638793945313e-05,
      "model_forward_time": 1.724311351776123,
      "step": 89442
    },
    {
      "epoch": 1.4726638793945313e-05,
      "step": 89442,
      "training_step_time": 8.453043222427368
    },
    {
      "epoch": 1.4727592468261718e-05,
      "model_forward_time": 1.7303831577301025,
      "step": 89443
    },
    {
      "epoch": 1.4727592468261718e-05,
      "step": 89443,
      "training_step_time": 7.467532157897949
    },
    {
      "epoch": 1.4728546142578126e-05,
      "model_forward_time": 1.3652338981628418,
      "step": 89444
    },
    {
      "epoch": 1.4728546142578126e-05,
      "step": 89444,
      "training_step_time": 6.7760303020477295
    },
    {
      "epoch": 1.4729499816894532e-05,
      "model_forward_time": 1.7232604026794434,
      "step": 89445
    },
    {
      "epoch": 1.4729499816894532e-05,
      "step": 89445,
      "training_step_time": 7.1818482875823975
    },
    {
      "epoch": 1.4730453491210937e-05,
      "model_forward_time": 2.4003446102142334,
      "step": 89446
    },
    {
      "epoch": 1.4730453491210937e-05,
      "step": 89446,
      "training_step_time": 8.315898418426514
    },
    {
      "epoch": 1.4731407165527343e-05,
      "model_forward_time": 1.728679895401001,
      "step": 89447
    },
    {
      "epoch": 1.4731407165527343e-05,
      "step": 89447,
      "training_step_time": 10.206721067428589
    },
    {
      "epoch": 1.473236083984375e-05,
      "model_forward_time": 1.7481844425201416,
      "step": 89448
    },
    {
      "epoch": 1.473236083984375e-05,
      "step": 89448,
      "training_step_time": 8.74381136894226
    },
    {
      "epoch": 1.4733314514160156e-05,
      "model_forward_time": 1.7384006977081299,
      "step": 89449
    },
    {
      "epoch": 1.4733314514160156e-05,
      "step": 89449,
      "training_step_time": 8.830526113510132
    },
    {
      "epoch": 1.4734268188476562e-05,
      "grad_norm": 0.09735918790102005,
      "learning_rate": 3.012793236173606e-07,
      "loss": 0.0568,
      "step": 89450
    },
    {
      "epoch": 1.4734268188476562e-05,
      "model_forward_time": 1.7275290489196777,
      "step": 89450
    },
    {
      "dynamics_loss_avg": 0.06705688796937466,
      "epoch": 1.4734268188476562e-05,
      "step": 89450
    },
    {
      "action_loss_avg": 0.001199678104603663,
      "epoch": 1.4734268188476562e-05,
      "step": 89450
    },
    {
      "epoch": 1.4734268188476562e-05,
      "step": 89450,
      "training_step_time": 8.517000913619995
    },
    {
      "epoch": 1.473522186279297e-05,
      "model_forward_time": 1.7322676181793213,
      "step": 89451
    },
    {
      "epoch": 1.473522186279297e-05,
      "step": 89451,
      "training_step_time": 7.58155632019043
    },
    {
      "epoch": 1.4736175537109375e-05,
      "model_forward_time": 1.7355594635009766,
      "step": 89452
    },
    {
      "epoch": 1.4736175537109375e-05,
      "step": 89452,
      "training_step_time": 7.519381284713745
    },
    {
      "epoch": 1.473712921142578e-05,
      "model_forward_time": 1.7305727005004883,
      "step": 89453
    },
    {
      "epoch": 1.473712921142578e-05,
      "step": 89453,
      "training_step_time": 6.710602760314941
    },
    {
      "epoch": 1.4738082885742188e-05,
      "model_forward_time": 1.7337291240692139,
      "step": 89454
    },
    {
      "epoch": 1.4738082885742188e-05,
      "step": 89454,
      "training_step_time": 7.238931894302368
    },
    {
      "epoch": 1.4739036560058594e-05,
      "model_forward_time": 2.4064748287200928,
      "step": 89455
    },
    {
      "epoch": 1.4739036560058594e-05,
      "step": 89455,
      "training_step_time": 7.599767446517944
    },
    {
      "epoch": 1.4739990234375e-05,
      "model_forward_time": 1.7301416397094727,
      "step": 89456
    },
    {
      "epoch": 1.4739990234375e-05,
      "step": 89456,
      "training_step_time": 7.767489910125732
    },
    {
      "epoch": 1.4740943908691407e-05,
      "model_forward_time": 1.7481679916381836,
      "step": 89457
    },
    {
      "epoch": 1.4740943908691407e-05,
      "step": 89457,
      "training_step_time": 10.321945190429688
    },
    {
      "epoch": 1.4741897583007813e-05,
      "model_forward_time": 1.7291662693023682,
      "step": 89458
    },
    {
      "epoch": 1.4741897583007813e-05,
      "step": 89458,
      "training_step_time": 8.767635822296143
    },
    {
      "epoch": 1.4742851257324218e-05,
      "model_forward_time": 1.7277095317840576,
      "step": 89459
    },
    {
      "epoch": 1.4742851257324218e-05,
      "step": 89459,
      "training_step_time": 9.205658912658691
    },
    {
      "epoch": 1.4743804931640626e-05,
      "grad_norm": 0.11574956774711609,
      "learning_rate": 3.0071429464469505e-07,
      "loss": 0.057,
      "step": 89460
    },
    {
      "epoch": 1.4743804931640626e-05,
      "model_forward_time": 1.732125997543335,
      "step": 89460
    },
    {
      "dynamics_loss_avg": 0.05636420045047998,
      "epoch": 1.4743804931640626e-05,
      "step": 89460
    },
    {
      "action_loss_avg": 0.001406455977121368,
      "epoch": 1.4743804931640626e-05,
      "step": 89460
    },
    {
      "epoch": 1.4743804931640626e-05,
      "step": 89460,
      "training_step_time": 7.586934804916382
    },
    {
      "epoch": 1.4744758605957032e-05,
      "model_forward_time": 1.7359442710876465,
      "step": 89461
    },
    {
      "epoch": 1.4744758605957032e-05,
      "step": 89461,
      "training_step_time": 7.5409088134765625
    },
    {
      "epoch": 1.4745712280273437e-05,
      "model_forward_time": 1.7291898727416992,
      "step": 89462
    },
    {
      "epoch": 1.4745712280273437e-05,
      "step": 89462,
      "training_step_time": 6.6607160568237305
    },
    {
      "epoch": 1.4746665954589843e-05,
      "model_forward_time": 1.7280209064483643,
      "step": 89463
    },
    {
      "epoch": 1.4746665954589843e-05,
      "step": 89463,
      "training_step_time": 7.177682876586914
    },
    {
      "epoch": 1.474761962890625e-05,
      "model_forward_time": 1.7296676635742188,
      "step": 89464
    },
    {
      "epoch": 1.474761962890625e-05,
      "step": 89464,
      "training_step_time": 8.528234243392944
    },
    {
      "epoch": 1.4748573303222656e-05,
      "model_forward_time": 1.7360823154449463,
      "step": 89465
    },
    {
      "epoch": 1.4748573303222656e-05,
      "step": 89465,
      "training_step_time": 9.514657735824585
    },
    {
      "epoch": 1.4749526977539062e-05,
      "model_forward_time": 1.745309829711914,
      "step": 89466
    },
    {
      "epoch": 1.4749526977539062e-05,
      "step": 89466,
      "training_step_time": 9.132410764694214
    },
    {
      "epoch": 1.475048065185547e-05,
      "model_forward_time": 1.7300620079040527,
      "step": 89467
    },
    {
      "epoch": 1.475048065185547e-05,
      "step": 89467,
      "training_step_time": 7.824644088745117
    },
    {
      "epoch": 1.4751434326171875e-05,
      "model_forward_time": 1.7322227954864502,
      "step": 89468
    },
    {
      "epoch": 1.4751434326171875e-05,
      "step": 89468,
      "training_step_time": 7.783636093139648
    },
    {
      "epoch": 1.475238800048828e-05,
      "model_forward_time": 1.7415456771850586,
      "step": 89469
    },
    {
      "epoch": 1.475238800048828e-05,
      "step": 89469,
      "training_step_time": 8.632425785064697
    },
    {
      "epoch": 1.4753341674804688e-05,
      "grad_norm": 0.09934641420841217,
      "learning_rate": 3.001497795788483e-07,
      "loss": 0.0558,
      "step": 89470
    },
    {
      "epoch": 1.4753341674804688e-05,
      "model_forward_time": 1.7295033931732178,
      "step": 89470
    },
    {
      "dynamics_loss_avg": 0.04799934364855289,
      "epoch": 1.4753341674804688e-05,
      "step": 89470
    },
    {
      "action_loss_avg": 0.0008671626390423625,
      "epoch": 1.4753341674804688e-05,
      "step": 89470
    },
    {
      "epoch": 1.4753341674804688e-05,
      "step": 89470,
      "training_step_time": 7.6718339920043945
    },
    {
      "epoch": 1.4754295349121094e-05,
      "model_forward_time": 1.7299413681030273,
      "step": 89471
    },
    {
      "epoch": 1.4754295349121094e-05,
      "step": 89471,
      "training_step_time": 7.40917444229126
    },
    {
      "epoch": 1.47552490234375e-05,
      "model_forward_time": 1.7297558784484863,
      "step": 89472
    },
    {
      "epoch": 1.47552490234375e-05,
      "step": 89472,
      "training_step_time": 6.865411043167114
    },
    {
      "epoch": 1.4756202697753907e-05,
      "model_forward_time": 1.7274055480957031,
      "step": 89473
    },
    {
      "epoch": 1.4756202697753907e-05,
      "step": 89473,
      "training_step_time": 8.127166032791138
    },
    {
      "epoch": 1.4757156372070313e-05,
      "model_forward_time": 2.381263256072998,
      "step": 89474
    },
    {
      "epoch": 1.4757156372070313e-05,
      "step": 89474,
      "training_step_time": 7.781338214874268
    },
    {
      "epoch": 1.4758110046386718e-05,
      "model_forward_time": 1.7393321990966797,
      "step": 89475
    },
    {
      "epoch": 1.4758110046386718e-05,
      "step": 89475,
      "training_step_time": 8.40464735031128
    },
    {
      "epoch": 1.4759063720703126e-05,
      "model_forward_time": 1.7255380153656006,
      "step": 89476
    },
    {
      "epoch": 1.4759063720703126e-05,
      "step": 89476,
      "training_step_time": 7.751093864440918
    },
    {
      "epoch": 1.4760017395019532e-05,
      "model_forward_time": 1.7296419143676758,
      "step": 89477
    },
    {
      "epoch": 1.4760017395019532e-05,
      "step": 89477,
      "training_step_time": 7.755905389785767
    },
    {
      "epoch": 1.4760971069335937e-05,
      "model_forward_time": 1.725564956665039,
      "step": 89478
    },
    {
      "epoch": 1.4760971069335937e-05,
      "step": 89478,
      "training_step_time": 7.843677997589111
    },
    {
      "epoch": 1.4761924743652343e-05,
      "model_forward_time": 1.7253270149230957,
      "step": 89479
    },
    {
      "epoch": 1.4761924743652343e-05,
      "step": 89479,
      "training_step_time": 7.732233047485352
    },
    {
      "epoch": 1.476287841796875e-05,
      "grad_norm": 0.12407039105892181,
      "learning_rate": 2.99585778481557e-07,
      "loss": 0.057,
      "step": 89480
    },
    {
      "epoch": 1.476287841796875e-05,
      "model_forward_time": 1.7329437732696533,
      "step": 89480
    },
    {
      "dynamics_loss_avg": 0.06810310296714306,
      "epoch": 1.476287841796875e-05,
      "step": 89480
    },
    {
      "action_loss_avg": 0.0014889044221490621,
      "epoch": 1.476287841796875e-05,
      "step": 89480
    },
    {
      "epoch": 1.476287841796875e-05,
      "step": 89480,
      "training_step_time": 7.682661771774292
    },
    {
      "epoch": 1.4763832092285156e-05,
      "model_forward_time": 1.7263250350952148,
      "step": 89481
    },
    {
      "epoch": 1.4763832092285156e-05,
      "step": 89481,
      "training_step_time": 6.8204505443573
    },
    {
      "epoch": 1.4764785766601562e-05,
      "model_forward_time": 1.7292470932006836,
      "step": 89482
    },
    {
      "epoch": 1.4764785766601562e-05,
      "step": 89482,
      "training_step_time": 7.556215047836304
    },
    {
      "epoch": 1.476573944091797e-05,
      "model_forward_time": 2.3971595764160156,
      "step": 89483
    },
    {
      "epoch": 1.476573944091797e-05,
      "step": 89483,
      "training_step_time": 8.603240013122559
    },
    {
      "epoch": 1.4766693115234375e-05,
      "model_forward_time": 1.74491548538208,
      "step": 89484
    },
    {
      "epoch": 1.4766693115234375e-05,
      "step": 89484,
      "training_step_time": 10.026931285858154
    },
    {
      "epoch": 1.476764678955078e-05,
      "model_forward_time": 1.7295613288879395,
      "step": 89485
    },
    {
      "epoch": 1.476764678955078e-05,
      "step": 89485,
      "training_step_time": 9.254949808120728
    },
    {
      "epoch": 1.4768600463867188e-05,
      "model_forward_time": 1.7315332889556885,
      "step": 89486
    },
    {
      "epoch": 1.4768600463867188e-05,
      "step": 89486,
      "training_step_time": 7.831388235092163
    },
    {
      "epoch": 1.4769554138183594e-05,
      "model_forward_time": 1.7283012866973877,
      "step": 89487
    },
    {
      "epoch": 1.4769554138183594e-05,
      "step": 89487,
      "training_step_time": 7.903144598007202
    },
    {
      "epoch": 1.47705078125e-05,
      "model_forward_time": 1.7291405200958252,
      "step": 89488
    },
    {
      "epoch": 1.47705078125e-05,
      "step": 89488,
      "training_step_time": 8.580964803695679
    },
    {
      "epoch": 1.4771461486816407e-05,
      "model_forward_time": 1.7279295921325684,
      "step": 89489
    },
    {
      "epoch": 1.4771461486816407e-05,
      "step": 89489,
      "training_step_time": 7.68220853805542
    },
    {
      "epoch": 1.4772415161132813e-05,
      "grad_norm": 0.08242017775774002,
      "learning_rate": 2.990222914144969e-07,
      "loss": 0.0564,
      "step": 89490
    },
    {
      "epoch": 1.4772415161132813e-05,
      "model_forward_time": 1.7273049354553223,
      "step": 89490
    },
    {
      "dynamics_loss_avg": 0.07117823157459498,
      "epoch": 1.4772415161132813e-05,
      "step": 89490
    },
    {
      "action_loss_avg": 0.00140378508658614,
      "epoch": 1.4772415161132813e-05,
      "step": 89490
    },
    {
      "epoch": 1.4772415161132813e-05,
      "step": 89490,
      "training_step_time": 7.326005220413208
    },
    {
      "epoch": 1.4773368835449218e-05,
      "model_forward_time": 1.72347092628479,
      "step": 89491
    },
    {
      "epoch": 1.4773368835449218e-05,
      "step": 89491,
      "training_step_time": 7.447260618209839
    },
    {
      "epoch": 1.4774322509765626e-05,
      "model_forward_time": 2.417452573776245,
      "step": 89492
    },
    {
      "epoch": 1.4774322509765626e-05,
      "step": 89492,
      "training_step_time": 7.698177814483643
    },
    {
      "epoch": 1.4775276184082032e-05,
      "model_forward_time": 1.7430453300476074,
      "step": 89493
    },
    {
      "epoch": 1.4775276184082032e-05,
      "step": 89493,
      "training_step_time": 10.017446279525757
    },
    {
      "epoch": 1.4776229858398437e-05,
      "model_forward_time": 1.726837396621704,
      "step": 89494
    },
    {
      "epoch": 1.4776229858398437e-05,
      "step": 89494,
      "training_step_time": 7.64260196685791
    },
    {
      "epoch": 1.4777183532714843e-05,
      "model_forward_time": 1.7323217391967773,
      "step": 89495
    },
    {
      "epoch": 1.4777183532714843e-05,
      "step": 89495,
      "training_step_time": 7.771777153015137
    },
    {
      "epoch": 1.477813720703125e-05,
      "model_forward_time": 1.7305264472961426,
      "step": 89496
    },
    {
      "epoch": 1.477813720703125e-05,
      "step": 89496,
      "training_step_time": 7.723841667175293
    },
    {
      "epoch": 1.4779090881347656e-05,
      "model_forward_time": 1.7265548706054688,
      "step": 89497
    },
    {
      "epoch": 1.4779090881347656e-05,
      "step": 89497,
      "training_step_time": 7.650662899017334
    },
    {
      "epoch": 1.4780044555664062e-05,
      "model_forward_time": 1.7328944206237793,
      "step": 89498
    },
    {
      "epoch": 1.4780044555664062e-05,
      "step": 89498,
      "training_step_time": 7.622282981872559
    },
    {
      "epoch": 1.478099822998047e-05,
      "model_forward_time": 1.7344434261322021,
      "step": 89499
    },
    {
      "epoch": 1.478099822998047e-05,
      "step": 89499,
      "training_step_time": 7.307524919509888
    },
    {
      "epoch": 1.4781951904296875e-05,
      "grad_norm": 0.09063918143510818,
      "learning_rate": 2.9845931843929146e-07,
      "loss": 0.0567,
      "step": 89500
    },
    {
      "epoch": 1.4781951904296875e-05,
      "model_forward_time": 1.7281367778778076,
      "step": 89500
    },
    {
      "dynamics_loss_avg": 0.05002054553478956,
      "epoch": 1.4781951904296875e-05,
      "step": 89500
    },
    {
      "action_loss_avg": 0.0019097537558991462,
      "epoch": 1.4781951904296875e-05,
      "step": 89500
    },
    {
      "epoch": 1.4781951904296875e-05,
      "step": 89500,
      "training_step_time": 7.441088676452637
    },
    {
      "epoch": 1.478290557861328e-05,
      "model_forward_time": 2.403651237487793,
      "step": 89501
    },
    {
      "epoch": 1.478290557861328e-05,
      "step": 89501,
      "training_step_time": 7.64155387878418
    },
    {
      "epoch": 1.4783859252929688e-05,
      "model_forward_time": 1.7460224628448486,
      "step": 89502
    },
    {
      "epoch": 1.4783859252929688e-05,
      "step": 89502,
      "training_step_time": 7.776869535446167
    },
    {
      "epoch": 1.4784812927246094e-05,
      "model_forward_time": 1.727302074432373,
      "step": 89503
    },
    {
      "epoch": 1.4784812927246094e-05,
      "step": 89503,
      "training_step_time": 9.655231952667236
    },
    {
      "epoch": 1.47857666015625e-05,
      "model_forward_time": 1.738316535949707,
      "step": 89504
    },
    {
      "epoch": 1.47857666015625e-05,
      "step": 89504,
      "training_step_time": 8.396198272705078
    },
    {
      "epoch": 1.4786720275878907e-05,
      "model_forward_time": 1.731370210647583,
      "step": 89505
    },
    {
      "epoch": 1.4786720275878907e-05,
      "step": 89505,
      "training_step_time": 8.212046384811401
    },
    {
      "epoch": 1.4787673950195313e-05,
      "model_forward_time": 1.7322988510131836,
      "step": 89506
    },
    {
      "epoch": 1.4787673950195313e-05,
      "step": 89506,
      "training_step_time": 7.801364421844482
    },
    {
      "epoch": 1.4788627624511718e-05,
      "model_forward_time": 1.756786584854126,
      "step": 89507
    },
    {
      "epoch": 1.4788627624511718e-05,
      "step": 89507,
      "training_step_time": 7.625125885009766
    },
    {
      "epoch": 1.4789581298828124e-05,
      "model_forward_time": 1.8235657215118408,
      "step": 89508
    },
    {
      "epoch": 1.4789581298828124e-05,
      "step": 89508,
      "training_step_time": 7.25723671913147
    },
    {
      "epoch": 1.4790534973144532e-05,
      "model_forward_time": 1.7908363342285156,
      "step": 89509
    },
    {
      "epoch": 1.4790534973144532e-05,
      "step": 89509,
      "training_step_time": 7.416072130203247
    },
    {
      "epoch": 1.4791488647460937e-05,
      "grad_norm": 0.09041465073823929,
      "learning_rate": 2.978968596175069e-07,
      "loss": 0.0569,
      "step": 89510
    },
    {
      "epoch": 1.4791488647460937e-05,
      "model_forward_time": 2.857663154602051,
      "step": 89510
    },
    {
      "dynamics_loss_avg": 0.09149032440036535,
      "epoch": 1.4791488647460937e-05,
      "step": 89510
    },
    {
      "action_loss_avg": 0.0014243349491152913,
      "epoch": 1.4791488647460937e-05,
      "step": 89510
    },
    {
      "epoch": 1.4791488647460937e-05,
      "step": 89510,
      "training_step_time": 7.607977628707886
    },
    {
      "epoch": 1.4792442321777343e-05,
      "model_forward_time": 1.8525936603546143,
      "step": 89511
    },
    {
      "epoch": 1.4792442321777343e-05,
      "step": 89511,
      "training_step_time": 9.971052885055542
    },
    {
      "epoch": 1.479339599609375e-05,
      "model_forward_time": 1.7897522449493408,
      "step": 89512
    },
    {
      "epoch": 1.479339599609375e-05,
      "step": 89512,
      "training_step_time": 9.01890254020691
    },
    {
      "epoch": 1.4794349670410156e-05,
      "model_forward_time": 1.793992042541504,
      "step": 89513
    },
    {
      "epoch": 1.4794349670410156e-05,
      "step": 89513,
      "training_step_time": 8.73420238494873
    },
    {
      "epoch": 1.4795303344726562e-05,
      "model_forward_time": 1.792332410812378,
      "step": 89514
    },
    {
      "epoch": 1.4795303344726562e-05,
      "step": 89514,
      "training_step_time": 7.713860034942627
    },
    {
      "epoch": 1.479625701904297e-05,
      "model_forward_time": 1.7894032001495361,
      "step": 89515
    },
    {
      "epoch": 1.479625701904297e-05,
      "step": 89515,
      "training_step_time": 7.793773889541626
    },
    {
      "epoch": 1.4797210693359375e-05,
      "model_forward_time": 1.7960288524627686,
      "step": 89516
    },
    {
      "epoch": 1.4797210693359375e-05,
      "step": 89516,
      "training_step_time": 7.668066501617432
    },
    {
      "epoch": 1.479816436767578e-05,
      "model_forward_time": 1.788536548614502,
      "step": 89517
    },
    {
      "epoch": 1.479816436767578e-05,
      "step": 89517,
      "training_step_time": 6.76699161529541
    },
    {
      "epoch": 1.4799118041992188e-05,
      "model_forward_time": 1.7898473739624023,
      "step": 89518
    },
    {
      "epoch": 1.4799118041992188e-05,
      "step": 89518,
      "training_step_time": 7.482512474060059
    },
    {
      "epoch": 1.4800071716308594e-05,
      "model_forward_time": 1.7917516231536865,
      "step": 89519
    },
    {
      "epoch": 1.4800071716308594e-05,
      "step": 89519,
      "training_step_time": 7.637292146682739
    },
    {
      "epoch": 1.4801025390625e-05,
      "grad_norm": 0.09549590200185776,
      "learning_rate": 2.9733491501065135e-07,
      "loss": 0.0548,
      "step": 89520
    },
    {
      "epoch": 1.4801025390625e-05,
      "model_forward_time": 2.8551740646362305,
      "step": 89520
    },
    {
      "dynamics_loss_avg": 0.0640472199767828,
      "epoch": 1.4801025390625e-05,
      "step": 89520
    },
    {
      "action_loss_avg": 0.0017732565145706757,
      "epoch": 1.4801025390625e-05,
      "step": 89520
    },
    {
      "epoch": 1.4801025390625e-05,
      "step": 89520,
      "training_step_time": 7.757567405700684
    },
    {
      "epoch": 1.4801979064941407e-05,
      "model_forward_time": 1.8122880458831787,
      "step": 89521
    },
    {
      "epoch": 1.4801979064941407e-05,
      "step": 89521,
      "training_step_time": 9.03343653678894
    },
    {
      "epoch": 1.4802932739257813e-05,
      "model_forward_time": 1.802318811416626,
      "step": 89522
    },
    {
      "epoch": 1.4802932739257813e-05,
      "step": 89522,
      "training_step_time": 9.299835443496704
    },
    {
      "epoch": 1.4803886413574218e-05,
      "model_forward_time": 1.808340311050415,
      "step": 89523
    },
    {
      "epoch": 1.4803886413574218e-05,
      "step": 89523,
      "training_step_time": 7.754248142242432
    },
    {
      "epoch": 1.4804840087890624e-05,
      "model_forward_time": 1.7844939231872559,
      "step": 89524
    },
    {
      "epoch": 1.4804840087890624e-05,
      "step": 89524,
      "training_step_time": 7.833767652511597
    },
    {
      "epoch": 1.4805793762207032e-05,
      "model_forward_time": 1.7715861797332764,
      "step": 89525
    },
    {
      "epoch": 1.4805793762207032e-05,
      "step": 89525,
      "training_step_time": 7.698683023452759
    },
    {
      "epoch": 1.4806747436523437e-05,
      "model_forward_time": 1.7949535846710205,
      "step": 89526
    },
    {
      "epoch": 1.4806747436523437e-05,
      "step": 89526,
      "training_step_time": 7.601416349411011
    },
    {
      "epoch": 1.4807701110839843e-05,
      "model_forward_time": 1.7724661827087402,
      "step": 89527
    },
    {
      "epoch": 1.4807701110839843e-05,
      "step": 89527,
      "training_step_time": 7.469238519668579
    },
    {
      "epoch": 1.480865478515625e-05,
      "model_forward_time": 1.7917253971099854,
      "step": 89528
    },
    {
      "epoch": 1.480865478515625e-05,
      "step": 89528,
      "training_step_time": 8.465619087219238
    },
    {
      "epoch": 1.4809608459472656e-05,
      "model_forward_time": 2.8779280185699463,
      "step": 89529
    },
    {
      "epoch": 1.4809608459472656e-05,
      "step": 89529,
      "training_step_time": 10.211300373077393
    },
    {
      "epoch": 1.4810562133789062e-05,
      "grad_norm": 0.11511868238449097,
      "learning_rate": 2.9677348468018007e-07,
      "loss": 0.0556,
      "step": 89530
    },
    {
      "epoch": 1.4810562133789062e-05,
      "model_forward_time": 1.7742342948913574,
      "step": 89530
    },
    {
      "dynamics_loss_avg": 0.048448539525270465,
      "epoch": 1.4810562133789062e-05,
      "step": 89530
    },
    {
      "action_loss_avg": 0.001475164043949917,
      "epoch": 1.4810562133789062e-05,
      "step": 89530
    },
    {
      "epoch": 1.4810562133789062e-05,
      "step": 89530,
      "training_step_time": 8.6555917263031
    },
    {
      "epoch": 1.481151580810547e-05,
      "model_forward_time": 1.7936677932739258,
      "step": 89531
    },
    {
      "epoch": 1.481151580810547e-05,
      "step": 89531,
      "training_step_time": 9.272103786468506
    },
    {
      "epoch": 1.4812469482421875e-05,
      "model_forward_time": 1.78495454788208,
      "step": 89532
    },
    {
      "epoch": 1.4812469482421875e-05,
      "step": 89532,
      "training_step_time": 8.275336027145386
    },
    {
      "epoch": 1.481342315673828e-05,
      "model_forward_time": 1.7872025966644287,
      "step": 89533
    },
    {
      "epoch": 1.481342315673828e-05,
      "step": 89533,
      "training_step_time": 8.74813437461853
    },
    {
      "epoch": 1.4814376831054688e-05,
      "model_forward_time": 1.7791945934295654,
      "step": 89534
    },
    {
      "epoch": 1.4814376831054688e-05,
      "step": 89534,
      "training_step_time": 7.672790765762329
    },
    {
      "epoch": 1.4815330505371094e-05,
      "model_forward_time": 1.7834622859954834,
      "step": 89535
    },
    {
      "epoch": 1.4815330505371094e-05,
      "step": 89535,
      "training_step_time": 7.609375238418579
    },
    {
      "epoch": 1.48162841796875e-05,
      "model_forward_time": 1.7890098094940186,
      "step": 89536
    },
    {
      "epoch": 1.48162841796875e-05,
      "step": 89536,
      "training_step_time": 7.4279022216796875
    },
    {
      "epoch": 1.4817237854003907e-05,
      "model_forward_time": 1.7761812210083008,
      "step": 89537
    },
    {
      "epoch": 1.4817237854003907e-05,
      "step": 89537,
      "training_step_time": 7.59691596031189
    },
    {
      "epoch": 1.4818191528320313e-05,
      "model_forward_time": 2.916398525238037,
      "step": 89538
    },
    {
      "epoch": 1.4818191528320313e-05,
      "step": 89538,
      "training_step_time": 8.767336130142212
    },
    {
      "epoch": 1.4819145202636718e-05,
      "model_forward_time": 1.7792563438415527,
      "step": 89539
    },
    {
      "epoch": 1.4819145202636718e-05,
      "step": 89539,
      "training_step_time": 7.820091247558594
    },
    {
      "epoch": 1.4820098876953124e-05,
      "grad_norm": 0.0848187506198883,
      "learning_rate": 2.962125686874873e-07,
      "loss": 0.0554,
      "step": 89540
    },
    {
      "epoch": 1.4820098876953124e-05,
      "model_forward_time": 1.790302038192749,
      "step": 89540
    },
    {
      "dynamics_loss_avg": 0.05269750282168388,
      "epoch": 1.4820098876953124e-05,
      "step": 89540
    },
    {
      "action_loss_avg": 0.0013965177698992194,
      "epoch": 1.4820098876953124e-05,
      "step": 89540
    },
    {
      "epoch": 1.4820098876953124e-05,
      "step": 89540,
      "training_step_time": 10.484813690185547
    },
    {
      "epoch": 1.4821052551269532e-05,
      "model_forward_time": 1.790574073791504,
      "step": 89541
    },
    {
      "epoch": 1.4821052551269532e-05,
      "step": 89541,
      "training_step_time": 7.852717638015747
    },
    {
      "epoch": 1.4822006225585937e-05,
      "model_forward_time": 1.8014962673187256,
      "step": 89542
    },
    {
      "epoch": 1.4822006225585937e-05,
      "step": 89542,
      "training_step_time": 7.792909622192383
    },
    {
      "epoch": 1.4822959899902343e-05,
      "model_forward_time": 1.7903389930725098,
      "step": 89543
    },
    {
      "epoch": 1.4822959899902343e-05,
      "step": 89543,
      "training_step_time": 8.84871530532837
    },
    {
      "epoch": 1.482391357421875e-05,
      "model_forward_time": 1.7801852226257324,
      "step": 89544
    },
    {
      "epoch": 1.482391357421875e-05,
      "step": 89544,
      "training_step_time": 7.723258018493652
    },
    {
      "epoch": 1.4824867248535156e-05,
      "model_forward_time": 1.7798371315002441,
      "step": 89545
    },
    {
      "epoch": 1.4824867248535156e-05,
      "step": 89545,
      "training_step_time": 7.522831916809082
    },
    {
      "epoch": 1.4825820922851562e-05,
      "model_forward_time": 1.7956600189208984,
      "step": 89546
    },
    {
      "epoch": 1.4825820922851562e-05,
      "step": 89546,
      "training_step_time": 7.662982225418091
    },
    {
      "epoch": 1.482677459716797e-05,
      "model_forward_time": 2.977019786834717,
      "step": 89547
    },
    {
      "epoch": 1.482677459716797e-05,
      "step": 89547,
      "training_step_time": 8.451092720031738
    },
    {
      "epoch": 1.4827728271484375e-05,
      "model_forward_time": 1.813452959060669,
      "step": 89548
    },
    {
      "epoch": 1.4827728271484375e-05,
      "step": 89548,
      "training_step_time": 7.80983304977417
    },
    {
      "epoch": 1.482868194580078e-05,
      "model_forward_time": 1.7898004055023193,
      "step": 89549
    },
    {
      "epoch": 1.482868194580078e-05,
      "step": 89549,
      "training_step_time": 9.211702108383179
    },
    {
      "epoch": 1.4829635620117188e-05,
      "grad_norm": 0.10317745804786682,
      "learning_rate": 2.9565216709391833e-07,
      "loss": 0.0552,
      "step": 89550
    },
    {
      "epoch": 1.4829635620117188e-05,
      "model_forward_time": 1.7930946350097656,
      "step": 89550
    },
    {
      "dynamics_loss_avg": 0.05657772580161691,
      "epoch": 1.4829635620117188e-05,
      "step": 89550
    },
    {
      "action_loss_avg": 0.001384659954055678,
      "epoch": 1.4829635620117188e-05,
      "step": 89550
    },
    {
      "epoch": 1.4829635620117188e-05,
      "step": 89550,
      "training_step_time": 8.660449266433716
    },
    {
      "epoch": 1.4830589294433594e-05,
      "model_forward_time": 1.7815730571746826,
      "step": 89551
    },
    {
      "epoch": 1.4830589294433594e-05,
      "step": 89551,
      "training_step_time": 7.751341819763184
    },
    {
      "epoch": 1.483154296875e-05,
      "model_forward_time": 1.788172721862793,
      "step": 89552
    },
    {
      "epoch": 1.483154296875e-05,
      "step": 89552,
      "training_step_time": 8.19883108139038
    },
    {
      "epoch": 1.4832496643066407e-05,
      "model_forward_time": 1.811326503753662,
      "step": 89553
    },
    {
      "epoch": 1.4832496643066407e-05,
      "step": 89553,
      "training_step_time": 7.545530080795288
    },
    {
      "epoch": 1.4833450317382813e-05,
      "model_forward_time": 1.787907361984253,
      "step": 89554
    },
    {
      "epoch": 1.4833450317382813e-05,
      "step": 89554,
      "training_step_time": 7.623650312423706
    },
    {
      "epoch": 1.4834403991699218e-05,
      "model_forward_time": 1.7937042713165283,
      "step": 89555
    },
    {
      "epoch": 1.4834403991699218e-05,
      "step": 89555,
      "training_step_time": 7.640496492385864
    },
    {
      "epoch": 1.4835357666015624e-05,
      "model_forward_time": 2.5158958435058594,
      "step": 89556
    },
    {
      "epoch": 1.4835357666015624e-05,
      "step": 89556,
      "training_step_time": 7.754039764404297
    },
    {
      "epoch": 1.4836311340332032e-05,
      "model_forward_time": 1.8077173233032227,
      "step": 89557
    },
    {
      "epoch": 1.4836311340332032e-05,
      "step": 89557,
      "training_step_time": 8.81210708618164
    },
    {
      "epoch": 1.4837265014648437e-05,
      "model_forward_time": 1.7836754322052002,
      "step": 89558
    },
    {
      "epoch": 1.4837265014648437e-05,
      "step": 89558,
      "training_step_time": 9.321120500564575
    },
    {
      "epoch": 1.4838218688964843e-05,
      "model_forward_time": 1.8066351413726807,
      "step": 89559
    },
    {
      "epoch": 1.4838218688964843e-05,
      "step": 89559,
      "training_step_time": 9.251133680343628
    },
    {
      "epoch": 1.483917236328125e-05,
      "grad_norm": 0.09544921666383743,
      "learning_rate": 2.950922799607542e-07,
      "loss": 0.056,
      "step": 89560
    },
    {
      "epoch": 1.483917236328125e-05,
      "model_forward_time": 1.7906334400177002,
      "step": 89560
    },
    {
      "dynamics_loss_avg": 0.04923243951052427,
      "epoch": 1.483917236328125e-05,
      "step": 89560
    },
    {
      "action_loss_avg": 0.001774177694460377,
      "epoch": 1.483917236328125e-05,
      "step": 89560
    },
    {
      "epoch": 1.483917236328125e-05,
      "step": 89560,
      "training_step_time": 7.745725393295288
    },
    {
      "epoch": 1.4840126037597656e-05,
      "model_forward_time": 1.8001205921173096,
      "step": 89561
    },
    {
      "epoch": 1.4840126037597656e-05,
      "step": 89561,
      "training_step_time": 7.70875096321106
    },
    {
      "epoch": 1.4841079711914062e-05,
      "model_forward_time": 1.7914464473724365,
      "step": 89562
    },
    {
      "epoch": 1.4841079711914062e-05,
      "step": 89562,
      "training_step_time": 7.683833360671997
    },
    {
      "epoch": 1.484203338623047e-05,
      "model_forward_time": 1.7957100868225098,
      "step": 89563
    },
    {
      "epoch": 1.484203338623047e-05,
      "step": 89563,
      "training_step_time": 7.647414207458496
    },
    {
      "epoch": 1.4842987060546875e-05,
      "model_forward_time": 1.8059864044189453,
      "step": 89564
    },
    {
      "epoch": 1.4842987060546875e-05,
      "step": 89564,
      "training_step_time": 8.656054973602295
    },
    {
      "epoch": 1.484394073486328e-05,
      "model_forward_time": 1.8585340976715088,
      "step": 89565
    },
    {
      "epoch": 1.484394073486328e-05,
      "step": 89565,
      "training_step_time": 7.749459266662598
    },
    {
      "epoch": 1.4844894409179688e-05,
      "model_forward_time": 2.888046979904175,
      "step": 89566
    },
    {
      "epoch": 1.4844894409179688e-05,
      "step": 89566,
      "training_step_time": 8.577589273452759
    },
    {
      "epoch": 1.4845848083496094e-05,
      "model_forward_time": 1.8068342208862305,
      "step": 89567
    },
    {
      "epoch": 1.4845848083496094e-05,
      "step": 89567,
      "training_step_time": 7.784152984619141
    },
    {
      "epoch": 1.48468017578125e-05,
      "model_forward_time": 1.808485984802246,
      "step": 89568
    },
    {
      "epoch": 1.48468017578125e-05,
      "step": 89568,
      "training_step_time": 9.678112506866455
    },
    {
      "epoch": 1.4847755432128907e-05,
      "model_forward_time": 1.7950456142425537,
      "step": 89569
    },
    {
      "epoch": 1.4847755432128907e-05,
      "step": 89569,
      "training_step_time": 8.472867250442505
    },
    {
      "epoch": 1.4848709106445313e-05,
      "grad_norm": 0.09105518460273743,
      "learning_rate": 2.9453290734922537e-07,
      "loss": 0.0559,
      "step": 89570
    },
    {
      "epoch": 1.4848709106445313e-05,
      "model_forward_time": 1.7992160320281982,
      "step": 89570
    },
    {
      "dynamics_loss_avg": 0.047570263780653474,
      "epoch": 1.4848709106445313e-05,
      "step": 89570
    },
    {
      "action_loss_avg": 0.002651978068752214,
      "epoch": 1.4848709106445313e-05,
      "step": 89570
    },
    {
      "epoch": 1.4848709106445313e-05,
      "step": 89570,
      "training_step_time": 7.756595134735107
    },
    {
      "epoch": 1.4849662780761718e-05,
      "model_forward_time": 1.786609411239624,
      "step": 89571
    },
    {
      "epoch": 1.4849662780761718e-05,
      "step": 89571,
      "training_step_time": 7.724633455276489
    },
    {
      "epoch": 1.4850616455078124e-05,
      "model_forward_time": 1.8023614883422852,
      "step": 89572
    },
    {
      "epoch": 1.4850616455078124e-05,
      "step": 89572,
      "training_step_time": 7.640707015991211
    },
    {
      "epoch": 1.4851570129394532e-05,
      "model_forward_time": 1.79591703414917,
      "step": 89573
    },
    {
      "epoch": 1.4851570129394532e-05,
      "step": 89573,
      "training_step_time": 7.660175323486328
    },
    {
      "epoch": 1.4852523803710937e-05,
      "model_forward_time": 1.841242790222168,
      "step": 89574
    },
    {
      "epoch": 1.4852523803710937e-05,
      "step": 89574,
      "training_step_time": 7.778662443161011
    },
    {
      "epoch": 1.4853477478027343e-05,
      "model_forward_time": 2.921755790710449,
      "step": 89575
    },
    {
      "epoch": 1.4853477478027343e-05,
      "step": 89575,
      "training_step_time": 7.808378458023071
    },
    {
      "epoch": 1.485443115234375e-05,
      "model_forward_time": 1.7960519790649414,
      "step": 89576
    },
    {
      "epoch": 1.485443115234375e-05,
      "step": 89576,
      "training_step_time": 8.948887586593628
    },
    {
      "epoch": 1.4855384826660156e-05,
      "model_forward_time": 1.7884526252746582,
      "step": 89577
    },
    {
      "epoch": 1.4855384826660156e-05,
      "step": 89577,
      "training_step_time": 10.240501165390015
    },
    {
      "epoch": 1.4856338500976562e-05,
      "model_forward_time": 1.7729237079620361,
      "step": 89578
    },
    {
      "epoch": 1.4856338500976562e-05,
      "step": 89578,
      "training_step_time": 7.722351789474487
    },
    {
      "epoch": 1.485729217529297e-05,
      "model_forward_time": 1.7942512035369873,
      "step": 89579
    },
    {
      "epoch": 1.485729217529297e-05,
      "step": 89579,
      "training_step_time": 8.288862943649292
    },
    {
      "epoch": 1.4858245849609375e-05,
      "grad_norm": 0.306554913520813,
      "learning_rate": 2.9397404932050287e-07,
      "loss": 0.0554,
      "step": 89580
    },
    {
      "epoch": 1.4858245849609375e-05,
      "model_forward_time": 1.7930045127868652,
      "step": 89580
    },
    {
      "dynamics_loss_avg": 0.054492474906146525,
      "epoch": 1.4858245849609375e-05,
      "step": 89580
    },
    {
      "action_loss_avg": 0.0015022450039396062,
      "epoch": 1.4858245849609375e-05,
      "step": 89580
    },
    {
      "epoch": 1.4858245849609375e-05,
      "step": 89580,
      "training_step_time": 8.414090394973755
    },
    {
      "epoch": 1.485919952392578e-05,
      "model_forward_time": 1.7925546169281006,
      "step": 89581
    },
    {
      "epoch": 1.485919952392578e-05,
      "step": 89581,
      "training_step_time": 7.642563581466675
    },
    {
      "epoch": 1.4860153198242188e-05,
      "model_forward_time": 1.784377098083496,
      "step": 89582
    },
    {
      "epoch": 1.4860153198242188e-05,
      "step": 89582,
      "training_step_time": 7.3791046142578125
    },
    {
      "epoch": 1.4861106872558594e-05,
      "model_forward_time": 1.850109577178955,
      "step": 89583
    },
    {
      "epoch": 1.4861106872558594e-05,
      "step": 89583,
      "training_step_time": 7.751321792602539
    },
    {
      "epoch": 1.4862060546875e-05,
      "model_forward_time": 2.914497137069702,
      "step": 89584
    },
    {
      "epoch": 1.4862060546875e-05,
      "step": 89584,
      "training_step_time": 7.727192640304565
    },
    {
      "epoch": 1.4863014221191407e-05,
      "model_forward_time": 1.7884926795959473,
      "step": 89585
    },
    {
      "epoch": 1.4863014221191407e-05,
      "step": 89585,
      "training_step_time": 7.857755661010742
    },
    {
      "epoch": 1.4863967895507813e-05,
      "model_forward_time": 1.8058803081512451,
      "step": 89586
    },
    {
      "epoch": 1.4863967895507813e-05,
      "step": 89586,
      "training_step_time": 8.343440771102905
    },
    {
      "epoch": 1.4864921569824219e-05,
      "model_forward_time": 1.8051395416259766,
      "step": 89587
    },
    {
      "epoch": 1.4864921569824219e-05,
      "step": 89587,
      "training_step_time": 7.7991485595703125
    },
    {
      "epoch": 1.4865875244140624e-05,
      "model_forward_time": 1.7837200164794922,
      "step": 89588
    },
    {
      "epoch": 1.4865875244140624e-05,
      "step": 89588,
      "training_step_time": 7.752679347991943
    },
    {
      "epoch": 1.4866828918457032e-05,
      "model_forward_time": 1.7888062000274658,
      "step": 89589
    },
    {
      "epoch": 1.4866828918457032e-05,
      "step": 89589,
      "training_step_time": 7.846598148345947
    },
    {
      "epoch": 1.4867782592773437e-05,
      "grad_norm": 0.0864701047539711,
      "learning_rate": 2.9341570593570345e-07,
      "loss": 0.0546,
      "step": 89590
    },
    {
      "epoch": 1.4867782592773437e-05,
      "model_forward_time": 1.8166587352752686,
      "step": 89590
    },
    {
      "dynamics_loss_avg": 0.06375215966254473,
      "epoch": 1.4867782592773437e-05,
      "step": 89590
    },
    {
      "action_loss_avg": 0.0015851808013394475,
      "epoch": 1.4867782592773437e-05,
      "step": 89590
    },
    {
      "epoch": 1.4867782592773437e-05,
      "step": 89590,
      "training_step_time": 7.588141679763794
    },
    {
      "epoch": 1.4868736267089843e-05,
      "model_forward_time": 1.7979042530059814,
      "step": 89591
    },
    {
      "epoch": 1.4868736267089843e-05,
      "step": 89591,
      "training_step_time": 7.650958776473999
    },
    {
      "epoch": 1.486968994140625e-05,
      "model_forward_time": 1.8331286907196045,
      "step": 89592
    },
    {
      "epoch": 1.486968994140625e-05,
      "step": 89592,
      "training_step_time": 7.794914722442627
    },
    {
      "epoch": 1.4870643615722656e-05,
      "model_forward_time": 2.8822438716888428,
      "step": 89593
    },
    {
      "epoch": 1.4870643615722656e-05,
      "step": 89593,
      "training_step_time": 8.726991653442383
    },
    {
      "epoch": 1.4871597290039062e-05,
      "model_forward_time": 1.7828011512756348,
      "step": 89594
    },
    {
      "epoch": 1.4871597290039062e-05,
      "step": 89594,
      "training_step_time": 8.86318039894104
    },
    {
      "epoch": 1.487255096435547e-05,
      "model_forward_time": 1.7839624881744385,
      "step": 89595
    },
    {
      "epoch": 1.487255096435547e-05,
      "step": 89595,
      "training_step_time": 7.811740398406982
    },
    {
      "epoch": 1.4873504638671875e-05,
      "model_forward_time": 1.8032031059265137,
      "step": 89596
    },
    {
      "epoch": 1.4873504638671875e-05,
      "step": 89596,
      "training_step_time": 7.841553211212158
    },
    {
      "epoch": 1.487445831298828e-05,
      "model_forward_time": 1.7824864387512207,
      "step": 89597
    },
    {
      "epoch": 1.487445831298828e-05,
      "step": 89597,
      "training_step_time": 7.811038970947266
    },
    {
      "epoch": 1.4875411987304688e-05,
      "model_forward_time": 1.781412124633789,
      "step": 89598
    },
    {
      "epoch": 1.4875411987304688e-05,
      "step": 89598,
      "training_step_time": 7.755922794342041
    },
    {
      "epoch": 1.4876365661621094e-05,
      "model_forward_time": 1.797607660293579,
      "step": 89599
    },
    {
      "epoch": 1.4876365661621094e-05,
      "step": 89599,
      "training_step_time": 7.762925386428833
    },
    {
      "epoch": 1.48773193359375e-05,
      "grad_norm": 0.0799604132771492,
      "learning_rate": 2.9285787725588433e-07,
      "loss": 0.055,
      "step": 89600
    },
    {
      "epoch": 1.48773193359375e-05,
      "model_forward_time": 1.7876579761505127,
      "step": 89600
    },
    {
      "dynamics_loss_avg": 0.057292846590280534,
      "epoch": 1.48773193359375e-05,
      "step": 89600
    },
    {
      "action_loss_avg": 0.001750590059964452,
      "epoch": 1.48773193359375e-05,
      "step": 89600
    },
    {
      "epoch": 1.48773193359375e-05,
      "step": 89600,
      "training_step_time": 8.633202075958252
    },
    {
      "epoch": 1.4878273010253907e-05,
      "model_forward_time": 1.8519973754882812,
      "step": 89601
    },
    {
      "epoch": 1.4878273010253907e-05,
      "step": 89601,
      "training_step_time": 7.772047519683838
    },
    {
      "epoch": 1.4879226684570313e-05,
      "model_forward_time": 2.8261771202087402,
      "step": 89602
    },
    {
      "epoch": 1.4879226684570313e-05,
      "step": 89602,
      "training_step_time": 7.808480501174927
    },
    {
      "epoch": 1.4880180358886719e-05,
      "model_forward_time": 1.781700849533081,
      "step": 89603
    },
    {
      "epoch": 1.4880180358886719e-05,
      "step": 89603,
      "training_step_time": 7.795848846435547
    },
    {
      "epoch": 1.4881134033203124e-05,
      "model_forward_time": 1.7882733345031738,
      "step": 89604
    },
    {
      "epoch": 1.4881134033203124e-05,
      "step": 89604,
      "training_step_time": 8.343379735946655
    },
    {
      "epoch": 1.4882087707519532e-05,
      "model_forward_time": 1.7994697093963623,
      "step": 89605
    },
    {
      "epoch": 1.4882087707519532e-05,
      "step": 89605,
      "training_step_time": 7.841716527938843
    },
    {
      "epoch": 1.4883041381835937e-05,
      "model_forward_time": 1.7937140464782715,
      "step": 89606
    },
    {
      "epoch": 1.4883041381835937e-05,
      "step": 89606,
      "training_step_time": 8.4416663646698
    },
    {
      "epoch": 1.4883995056152343e-05,
      "model_forward_time": 1.7846143245697021,
      "step": 89607
    },
    {
      "epoch": 1.4883995056152343e-05,
      "step": 89607,
      "training_step_time": 7.758193254470825
    },
    {
      "epoch": 1.488494873046875e-05,
      "model_forward_time": 1.7869219779968262,
      "step": 89608
    },
    {
      "epoch": 1.488494873046875e-05,
      "step": 89608,
      "training_step_time": 7.772193908691406
    },
    {
      "epoch": 1.4885902404785156e-05,
      "model_forward_time": 1.8043954372406006,
      "step": 89609
    },
    {
      "epoch": 1.4885902404785156e-05,
      "step": 89609,
      "training_step_time": 7.657500505447388
    },
    {
      "epoch": 1.4886856079101562e-05,
      "grad_norm": 0.08644786477088928,
      "learning_rate": 2.9230056334205284e-07,
      "loss": 0.0553,
      "step": 89610
    },
    {
      "epoch": 1.4886856079101562e-05,
      "model_forward_time": 1.854274034500122,
      "step": 89610
    },
    {
      "dynamics_loss_avg": 0.05634073503315449,
      "epoch": 1.4886856079101562e-05,
      "step": 89610
    },
    {
      "action_loss_avg": 0.000913584273075685,
      "epoch": 1.4886856079101562e-05,
      "step": 89610
    },
    {
      "epoch": 1.4886856079101562e-05,
      "step": 89610,
      "training_step_time": 7.682477235794067
    },
    {
      "epoch": 1.488780975341797e-05,
      "model_forward_time": 1.8094587326049805,
      "step": 89611
    },
    {
      "epoch": 1.488780975341797e-05,
      "step": 89611,
      "training_step_time": 7.780952215194702
    },
    {
      "epoch": 1.4888763427734375e-05,
      "model_forward_time": 2.891624927520752,
      "step": 89612
    },
    {
      "epoch": 1.4888763427734375e-05,
      "step": 89612,
      "training_step_time": 7.739492893218994
    },
    {
      "epoch": 1.488971710205078e-05,
      "model_forward_time": 1.7917909622192383,
      "step": 89613
    },
    {
      "epoch": 1.488971710205078e-05,
      "step": 89613,
      "training_step_time": 7.772587537765503
    },
    {
      "epoch": 1.4890670776367188e-05,
      "model_forward_time": 1.8001365661621094,
      "step": 89614
    },
    {
      "epoch": 1.4890670776367188e-05,
      "step": 89614,
      "training_step_time": 8.516840934753418
    },
    {
      "epoch": 1.4891624450683594e-05,
      "model_forward_time": 1.7990875244140625,
      "step": 89615
    },
    {
      "epoch": 1.4891624450683594e-05,
      "step": 89615,
      "training_step_time": 8.781835317611694
    },
    {
      "epoch": 1.4892578125e-05,
      "model_forward_time": 1.7929704189300537,
      "step": 89616
    },
    {
      "epoch": 1.4892578125e-05,
      "step": 89616,
      "training_step_time": 8.325226545333862
    },
    {
      "epoch": 1.4893531799316407e-05,
      "model_forward_time": 1.8064863681793213,
      "step": 89617
    },
    {
      "epoch": 1.4893531799316407e-05,
      "step": 89617,
      "training_step_time": 7.82463002204895
    },
    {
      "epoch": 1.4894485473632813e-05,
      "model_forward_time": 1.7846546173095703,
      "step": 89618
    },
    {
      "epoch": 1.4894485473632813e-05,
      "step": 89618,
      "training_step_time": 7.646504640579224
    },
    {
      "epoch": 1.4895439147949219e-05,
      "model_forward_time": 1.8538193702697754,
      "step": 89619
    },
    {
      "epoch": 1.4895439147949219e-05,
      "step": 89619,
      "training_step_time": 7.727399826049805
    },
    {
      "epoch": 1.4896392822265624e-05,
      "grad_norm": 0.09398285299539566,
      "learning_rate": 2.917437642551524e-07,
      "loss": 0.0546,
      "step": 89620
    },
    {
      "epoch": 1.4896392822265624e-05,
      "model_forward_time": 1.7999520301818848,
      "step": 89620
    },
    {
      "dynamics_loss_avg": 0.061303218081593516,
      "epoch": 1.4896392822265624e-05,
      "step": 89620
    },
    {
      "action_loss_avg": 0.0012889264966361226,
      "epoch": 1.4896392822265624e-05,
      "step": 89620
    },
    {
      "epoch": 1.4896392822265624e-05,
      "step": 89620,
      "training_step_time": 7.765232563018799
    },
    {
      "epoch": 1.4897346496582032e-05,
      "model_forward_time": 2.9119083881378174,
      "step": 89621
    },
    {
      "epoch": 1.4897346496582032e-05,
      "step": 89621,
      "training_step_time": 7.759490251541138
    },
    {
      "epoch": 1.4898300170898437e-05,
      "model_forward_time": 1.8025662899017334,
      "step": 89622
    },
    {
      "epoch": 1.4898300170898437e-05,
      "step": 89622,
      "training_step_time": 7.691303253173828
    },
    {
      "epoch": 1.4899253845214843e-05,
      "model_forward_time": 1.7977955341339111,
      "step": 89623
    },
    {
      "epoch": 1.4899253845214843e-05,
      "step": 89623,
      "training_step_time": 8.821120500564575
    },
    {
      "epoch": 1.490020751953125e-05,
      "model_forward_time": 1.8133234977722168,
      "step": 89624
    },
    {
      "epoch": 1.490020751953125e-05,
      "step": 89624,
      "training_step_time": 7.786733865737915
    },
    {
      "epoch": 1.4901161193847656e-05,
      "model_forward_time": 1.8013718128204346,
      "step": 89625
    },
    {
      "epoch": 1.4901161193847656e-05,
      "step": 89625,
      "training_step_time": 7.7462780475616455
    },
    {
      "epoch": 1.4902114868164062e-05,
      "model_forward_time": 1.7998967170715332,
      "step": 89626
    },
    {
      "epoch": 1.4902114868164062e-05,
      "step": 89626,
      "training_step_time": 7.751573085784912
    },
    {
      "epoch": 1.490306854248047e-05,
      "model_forward_time": 1.8181178569793701,
      "step": 89627
    },
    {
      "epoch": 1.490306854248047e-05,
      "step": 89627,
      "training_step_time": 7.596747636795044
    },
    {
      "epoch": 1.4904022216796875e-05,
      "model_forward_time": 1.8539958000183105,
      "step": 89628
    },
    {
      "epoch": 1.4904022216796875e-05,
      "step": 89628,
      "training_step_time": 8.194334506988525
    },
    {
      "epoch": 1.490497589111328e-05,
      "model_forward_time": 1.8009490966796875,
      "step": 89629
    },
    {
      "epoch": 1.490497589111328e-05,
      "step": 89629,
      "training_step_time": 9.18472957611084
    },
    {
      "epoch": 1.4905929565429688e-05,
      "grad_norm": 0.07497530430555344,
      "learning_rate": 2.91187480056076e-07,
      "loss": 0.0549,
      "step": 89630
    },
    {
      "epoch": 1.4905929565429688e-05,
      "model_forward_time": 2.8759896755218506,
      "step": 89630
    },
    {
      "dynamics_loss_avg": 0.05979431532323361,
      "epoch": 1.4905929565429688e-05,
      "step": 89630
    },
    {
      "action_loss_avg": 0.0014895054890075698,
      "epoch": 1.4905929565429688e-05,
      "step": 89630
    },
    {
      "epoch": 1.4905929565429688e-05,
      "step": 89630,
      "training_step_time": 9.373758554458618
    },
    {
      "epoch": 1.4906883239746094e-05,
      "model_forward_time": 1.7981553077697754,
      "step": 89631
    },
    {
      "epoch": 1.4906883239746094e-05,
      "step": 89631,
      "training_step_time": 8.60381007194519
    },
    {
      "epoch": 1.49078369140625e-05,
      "model_forward_time": 1.7944908142089844,
      "step": 89632
    },
    {
      "epoch": 1.49078369140625e-05,
      "step": 89632,
      "training_step_time": 9.415836572647095
    },
    {
      "epoch": 1.4908790588378905e-05,
      "model_forward_time": 1.786604642868042,
      "step": 89633
    },
    {
      "epoch": 1.4908790588378905e-05,
      "step": 89633,
      "training_step_time": 8.764228105545044
    },
    {
      "epoch": 1.4909744262695313e-05,
      "model_forward_time": 1.7924399375915527,
      "step": 89634
    },
    {
      "epoch": 1.4909744262695313e-05,
      "step": 89634,
      "training_step_time": 8.335943937301636
    },
    {
      "epoch": 1.4910697937011719e-05,
      "model_forward_time": 1.7996997833251953,
      "step": 89635
    },
    {
      "epoch": 1.4910697937011719e-05,
      "step": 89635,
      "training_step_time": 7.788404703140259
    },
    {
      "epoch": 1.4911651611328124e-05,
      "model_forward_time": 1.7940378189086914,
      "step": 89636
    },
    {
      "epoch": 1.4911651611328124e-05,
      "step": 89636,
      "training_step_time": 9.023448705673218
    },
    {
      "epoch": 1.4912605285644532e-05,
      "model_forward_time": 1.8626885414123535,
      "step": 89637
    },
    {
      "epoch": 1.4912605285644532e-05,
      "step": 89637,
      "training_step_time": 7.685516119003296
    },
    {
      "epoch": 1.4913558959960937e-05,
      "model_forward_time": 1.7883410453796387,
      "step": 89638
    },
    {
      "epoch": 1.4913558959960937e-05,
      "step": 89638,
      "training_step_time": 7.660501718521118
    },
    {
      "epoch": 1.4914512634277343e-05,
      "model_forward_time": 2.871044397354126,
      "step": 89639
    },
    {
      "epoch": 1.4914512634277343e-05,
      "step": 89639,
      "training_step_time": 7.7266223430633545
    },
    {
      "epoch": 1.491546630859375e-05,
      "grad_norm": 0.08019542694091797,
      "learning_rate": 2.9063171080565667e-07,
      "loss": 0.0529,
      "step": 89640
    },
    {
      "epoch": 1.491546630859375e-05,
      "model_forward_time": 1.7827589511871338,
      "step": 89640
    },
    {
      "dynamics_loss_avg": 0.059243308007717134,
      "epoch": 1.491546630859375e-05,
      "step": 89640
    },
    {
      "action_loss_avg": 0.0017155618988908827,
      "epoch": 1.491546630859375e-05,
      "step": 89640
    },
    {
      "epoch": 1.491546630859375e-05,
      "step": 89640,
      "training_step_time": 8.752426385879517
    },
    {
      "epoch": 1.4916419982910156e-05,
      "model_forward_time": 1.8012356758117676,
      "step": 89641
    },
    {
      "epoch": 1.4916419982910156e-05,
      "step": 89641,
      "training_step_time": 8.257577419281006
    },
    {
      "epoch": 1.4917373657226562e-05,
      "model_forward_time": 1.791384220123291,
      "step": 89642
    },
    {
      "epoch": 1.4917373657226562e-05,
      "step": 89642,
      "training_step_time": 8.63555121421814
    },
    {
      "epoch": 1.491832733154297e-05,
      "model_forward_time": 1.7961304187774658,
      "step": 89643
    },
    {
      "epoch": 1.491832733154297e-05,
      "step": 89643,
      "training_step_time": 7.677266359329224
    },
    {
      "epoch": 1.4919281005859375e-05,
      "model_forward_time": 1.800680160522461,
      "step": 89644
    },
    {
      "epoch": 1.4919281005859375e-05,
      "step": 89644,
      "training_step_time": 7.715606927871704
    },
    {
      "epoch": 1.492023468017578e-05,
      "model_forward_time": 1.8003451824188232,
      "step": 89645
    },
    {
      "epoch": 1.492023468017578e-05,
      "step": 89645,
      "training_step_time": 7.730914354324341
    },
    {
      "epoch": 1.4921188354492188e-05,
      "model_forward_time": 1.8589272499084473,
      "step": 89646
    },
    {
      "epoch": 1.4921188354492188e-05,
      "step": 89646,
      "training_step_time": 7.682458877563477
    },
    {
      "epoch": 1.4922142028808594e-05,
      "model_forward_time": 1.7947216033935547,
      "step": 89647
    },
    {
      "epoch": 1.4922142028808594e-05,
      "step": 89647,
      "training_step_time": 7.652705192565918
    },
    {
      "epoch": 1.4923095703125e-05,
      "model_forward_time": 2.88437819480896,
      "step": 89648
    },
    {
      "epoch": 1.4923095703125e-05,
      "step": 89648,
      "training_step_time": 7.727498769760132
    },
    {
      "epoch": 1.4924049377441405e-05,
      "model_forward_time": 1.7850332260131836,
      "step": 89649
    },
    {
      "epoch": 1.4924049377441405e-05,
      "step": 89649,
      "training_step_time": 8.734701871871948
    },
    {
      "epoch": 1.4925003051757813e-05,
      "grad_norm": 0.08779331296682358,
      "learning_rate": 2.900764565646741e-07,
      "loss": 0.0536,
      "step": 89650
    },
    {
      "epoch": 1.4925003051757813e-05,
      "model_forward_time": 1.8009898662567139,
      "step": 89650
    },
    {
      "dynamics_loss_avg": 0.07036691922694445,
      "epoch": 1.4925003051757813e-05,
      "step": 89650
    },
    {
      "action_loss_avg": 0.0014566925645340234,
      "epoch": 1.4925003051757813e-05,
      "step": 89650
    },
    {
      "epoch": 1.4925003051757813e-05,
      "step": 89650,
      "training_step_time": 9.36153268814087
    },
    {
      "epoch": 1.4925956726074219e-05,
      "model_forward_time": 1.8002920150756836,
      "step": 89651
    },
    {
      "epoch": 1.4925956726074219e-05,
      "step": 89651,
      "training_step_time": 8.590412855148315
    },
    {
      "epoch": 1.4926910400390624e-05,
      "model_forward_time": 1.807246446609497,
      "step": 89652
    },
    {
      "epoch": 1.4926910400390624e-05,
      "step": 89652,
      "training_step_time": 7.688229322433472
    },
    {
      "epoch": 1.4927864074707032e-05,
      "model_forward_time": 1.815237045288086,
      "step": 89653
    },
    {
      "epoch": 1.4927864074707032e-05,
      "step": 89653,
      "training_step_time": 8.233057260513306
    },
    {
      "epoch": 1.4928817749023437e-05,
      "model_forward_time": 1.8020482063293457,
      "step": 89654
    },
    {
      "epoch": 1.4928817749023437e-05,
      "step": 89654,
      "training_step_time": 7.783133029937744
    },
    {
      "epoch": 1.4929771423339843e-05,
      "model_forward_time": 1.8649699687957764,
      "step": 89655
    },
    {
      "epoch": 1.4929771423339843e-05,
      "step": 89655,
      "training_step_time": 9.099455833435059
    },
    {
      "epoch": 1.493072509765625e-05,
      "model_forward_time": 1.8060603141784668,
      "step": 89656
    },
    {
      "epoch": 1.493072509765625e-05,
      "step": 89656,
      "training_step_time": 7.654357194900513
    },
    {
      "epoch": 1.4931678771972656e-05,
      "model_forward_time": 1.7979106903076172,
      "step": 89657
    },
    {
      "epoch": 1.4931678771972656e-05,
      "step": 89657,
      "training_step_time": 8.578123331069946
    },
    {
      "epoch": 1.4932632446289062e-05,
      "model_forward_time": 1.8002698421478271,
      "step": 89658
    },
    {
      "epoch": 1.4932632446289062e-05,
      "step": 89658,
      "training_step_time": 8.627996444702148
    },
    {
      "epoch": 1.493358612060547e-05,
      "model_forward_time": 1.7988922595977783,
      "step": 89659
    },
    {
      "epoch": 1.493358612060547e-05,
      "step": 89659,
      "training_step_time": 7.753269195556641
    },
    {
      "epoch": 1.4934539794921875e-05,
      "grad_norm": 0.12868951261043549,
      "learning_rate": 2.895217173938475e-07,
      "loss": 0.0532,
      "step": 89660
    },
    {
      "epoch": 1.4934539794921875e-05,
      "model_forward_time": 1.8051371574401855,
      "step": 89660
    },
    {
      "dynamics_loss_avg": 0.043551383540034294,
      "epoch": 1.4934539794921875e-05,
      "step": 89660
    },
    {
      "action_loss_avg": 0.0010026756208389997,
      "epoch": 1.4934539794921875e-05,
      "step": 89660
    },
    {
      "epoch": 1.4934539794921875e-05,
      "step": 89660,
      "training_step_time": 8.801815509796143
    },
    {
      "epoch": 1.4935493469238281e-05,
      "model_forward_time": 1.8046793937683105,
      "step": 89661
    },
    {
      "epoch": 1.4935493469238281e-05,
      "step": 89661,
      "training_step_time": 7.684022903442383
    },
    {
      "epoch": 1.4936447143554688e-05,
      "model_forward_time": 1.801077127456665,
      "step": 89662
    },
    {
      "epoch": 1.4936447143554688e-05,
      "step": 89662,
      "training_step_time": 7.730313062667847
    },
    {
      "epoch": 1.4937400817871094e-05,
      "model_forward_time": 1.8093693256378174,
      "step": 89663
    },
    {
      "epoch": 1.4937400817871094e-05,
      "step": 89663,
      "training_step_time": 7.738116264343262
    },
    {
      "epoch": 1.49383544921875e-05,
      "model_forward_time": 1.879775047302246,
      "step": 89664
    },
    {
      "epoch": 1.49383544921875e-05,
      "step": 89664,
      "training_step_time": 8.45486068725586
    },
    {
      "epoch": 1.4939308166503905e-05,
      "model_forward_time": 1.793285846710205,
      "step": 89665
    },
    {
      "epoch": 1.4939308166503905e-05,
      "step": 89665,
      "training_step_time": 8.659817457199097
    },
    {
      "epoch": 1.4940261840820313e-05,
      "model_forward_time": 1.8065240383148193,
      "step": 89666
    },
    {
      "epoch": 1.4940261840820313e-05,
      "step": 89666,
      "training_step_time": 8.53092336654663
    },
    {
      "epoch": 1.4941215515136719e-05,
      "model_forward_time": 2.855570077896118,
      "step": 89667
    },
    {
      "epoch": 1.4941215515136719e-05,
      "step": 89667,
      "training_step_time": 7.736034631729126
    },
    {
      "epoch": 1.4942169189453124e-05,
      "model_forward_time": 1.8156275749206543,
      "step": 89668
    },
    {
      "epoch": 1.4942169189453124e-05,
      "step": 89668,
      "training_step_time": 9.848301649093628
    },
    {
      "epoch": 1.4943122863769532e-05,
      "model_forward_time": 1.8048961162567139,
      "step": 89669
    },
    {
      "epoch": 1.4943122863769532e-05,
      "step": 89669,
      "training_step_time": 9.457537412643433
    },
    {
      "epoch": 1.4944076538085937e-05,
      "grad_norm": 0.06840533018112183,
      "learning_rate": 2.8896749335384445e-07,
      "loss": 0.0529,
      "step": 89670
    },
    {
      "epoch": 1.4944076538085937e-05,
      "model_forward_time": 1.7890615463256836,
      "step": 89670
    },
    {
      "dynamics_loss_avg": 0.056380238756537436,
      "epoch": 1.4944076538085937e-05,
      "step": 89670
    },
    {
      "action_loss_avg": 0.0015636667027138173,
      "epoch": 1.4944076538085937e-05,
      "step": 89670
    },
    {
      "epoch": 1.4944076538085937e-05,
      "step": 89670,
      "training_step_time": 9.142162322998047
    },
    {
      "epoch": 1.4945030212402343e-05,
      "model_forward_time": 1.7963886260986328,
      "step": 89671
    },
    {
      "epoch": 1.4945030212402343e-05,
      "step": 89671,
      "training_step_time": 7.661571025848389
    },
    {
      "epoch": 1.494598388671875e-05,
      "model_forward_time": 1.781822681427002,
      "step": 89672
    },
    {
      "epoch": 1.494598388671875e-05,
      "step": 89672,
      "training_step_time": 7.7380406856536865
    },
    {
      "epoch": 1.4946937561035156e-05,
      "model_forward_time": 1.8780534267425537,
      "step": 89673
    },
    {
      "epoch": 1.4946937561035156e-05,
      "step": 89673,
      "training_step_time": 7.421764850616455
    },
    {
      "epoch": 1.4947891235351562e-05,
      "model_forward_time": 1.7691209316253662,
      "step": 89674
    },
    {
      "epoch": 1.4947891235351562e-05,
      "step": 89674,
      "training_step_time": 8.37692928314209
    },
    {
      "epoch": 1.494884490966797e-05,
      "model_forward_time": 1.8014168739318848,
      "step": 89675
    },
    {
      "epoch": 1.494884490966797e-05,
      "step": 89675,
      "training_step_time": 7.566222906112671
    },
    {
      "epoch": 1.4949798583984375e-05,
      "model_forward_time": 2.898589611053467,
      "step": 89676
    },
    {
      "epoch": 1.4949798583984375e-05,
      "step": 89676,
      "training_step_time": 7.721306085586548
    },
    {
      "epoch": 1.4950752258300781e-05,
      "model_forward_time": 1.804802656173706,
      "step": 89677
    },
    {
      "epoch": 1.4950752258300781e-05,
      "step": 89677,
      "training_step_time": 7.729130029678345
    },
    {
      "epoch": 1.4951705932617188e-05,
      "model_forward_time": 1.802988052368164,
      "step": 89678
    },
    {
      "epoch": 1.4951705932617188e-05,
      "step": 89678,
      "training_step_time": 9.362907648086548
    },
    {
      "epoch": 1.4952659606933594e-05,
      "model_forward_time": 1.8161001205444336,
      "step": 89679
    },
    {
      "epoch": 1.4952659606933594e-05,
      "step": 89679,
      "training_step_time": 9.442861795425415
    },
    {
      "epoch": 1.495361328125e-05,
      "grad_norm": 0.08099132031202316,
      "learning_rate": 2.8841378450527314e-07,
      "loss": 0.0519,
      "step": 89680
    },
    {
      "epoch": 1.495361328125e-05,
      "model_forward_time": 1.792539358139038,
      "step": 89680
    },
    {
      "dynamics_loss_avg": 0.04764416515827179,
      "epoch": 1.495361328125e-05,
      "step": 89680
    },
    {
      "action_loss_avg": 0.0014475117524852977,
      "epoch": 1.495361328125e-05,
      "step": 89680
    },
    {
      "epoch": 1.495361328125e-05,
      "step": 89680,
      "training_step_time": 8.536385774612427
    },
    {
      "epoch": 1.4954566955566405e-05,
      "model_forward_time": 1.8182792663574219,
      "step": 89681
    },
    {
      "epoch": 1.4954566955566405e-05,
      "step": 89681,
      "training_step_time": 7.813320159912109
    },
    {
      "epoch": 1.4955520629882813e-05,
      "model_forward_time": 1.8615553379058838,
      "step": 89682
    },
    {
      "epoch": 1.4955520629882813e-05,
      "step": 89682,
      "training_step_time": 7.70823860168457
    },
    {
      "epoch": 1.4956474304199219e-05,
      "model_forward_time": 1.7945895195007324,
      "step": 89683
    },
    {
      "epoch": 1.4956474304199219e-05,
      "step": 89683,
      "training_step_time": 7.715687990188599
    },
    {
      "epoch": 1.4957427978515624e-05,
      "model_forward_time": 1.8089075088500977,
      "step": 89684
    },
    {
      "epoch": 1.4957427978515624e-05,
      "step": 89684,
      "training_step_time": 7.606782913208008
    },
    {
      "epoch": 1.4958381652832032e-05,
      "model_forward_time": 2.8778321743011475,
      "step": 89685
    },
    {
      "epoch": 1.4958381652832032e-05,
      "step": 89685,
      "training_step_time": 8.427897214889526
    },
    {
      "epoch": 1.4959335327148437e-05,
      "model_forward_time": 1.8039610385894775,
      "step": 89686
    },
    {
      "epoch": 1.4959335327148437e-05,
      "step": 89686,
      "training_step_time": 9.124403953552246
    },
    {
      "epoch": 1.4960289001464843e-05,
      "model_forward_time": 1.8091230392456055,
      "step": 89687
    },
    {
      "epoch": 1.4960289001464843e-05,
      "step": 89687,
      "training_step_time": 8.24747109413147
    },
    {
      "epoch": 1.496124267578125e-05,
      "model_forward_time": 1.8126935958862305,
      "step": 89688
    },
    {
      "epoch": 1.496124267578125e-05,
      "step": 89688,
      "training_step_time": 7.77787971496582
    },
    {
      "epoch": 1.4962196350097656e-05,
      "model_forward_time": 1.8046398162841797,
      "step": 89689
    },
    {
      "epoch": 1.4962196350097656e-05,
      "step": 89689,
      "training_step_time": 8.621612548828125
    },
    {
      "epoch": 1.4963150024414062e-05,
      "grad_norm": 0.10803291201591492,
      "learning_rate": 2.878605909086873e-07,
      "loss": 0.0525,
      "step": 89690
    },
    {
      "epoch": 1.4963150024414062e-05,
      "model_forward_time": 1.7990550994873047,
      "step": 89690
    },
    {
      "dynamics_loss_avg": 0.05110274814069271,
      "epoch": 1.4963150024414062e-05,
      "step": 89690
    },
    {
      "action_loss_avg": 0.0009313192073022947,
      "epoch": 1.4963150024414062e-05,
      "step": 89690
    },
    {
      "epoch": 1.4963150024414062e-05,
      "step": 89690,
      "training_step_time": 7.700187921524048
    },
    {
      "epoch": 1.496410369873047e-05,
      "model_forward_time": 1.8615314960479736,
      "step": 89691
    },
    {
      "epoch": 1.496410369873047e-05,
      "step": 89691,
      "training_step_time": 7.692992210388184
    },
    {
      "epoch": 1.4965057373046875e-05,
      "model_forward_time": 1.8015596866607666,
      "step": 89692
    },
    {
      "epoch": 1.4965057373046875e-05,
      "step": 89692,
      "training_step_time": 7.627769708633423
    },
    {
      "epoch": 1.4966011047363281e-05,
      "model_forward_time": 1.8159990310668945,
      "step": 89693
    },
    {
      "epoch": 1.4966011047363281e-05,
      "step": 89693,
      "training_step_time": 7.61559796333313
    },
    {
      "epoch": 1.4966964721679688e-05,
      "model_forward_time": 2.9090495109558105,
      "step": 89694
    },
    {
      "epoch": 1.4966964721679688e-05,
      "step": 89694,
      "training_step_time": 8.789097785949707
    },
    {
      "epoch": 1.4967918395996094e-05,
      "model_forward_time": 1.791172981262207,
      "step": 89695
    },
    {
      "epoch": 1.4967918395996094e-05,
      "step": 89695,
      "training_step_time": 8.837467908859253
    },
    {
      "epoch": 1.49688720703125e-05,
      "model_forward_time": 1.7947156429290771,
      "step": 89696
    },
    {
      "epoch": 1.49688720703125e-05,
      "step": 89696,
      "training_step_time": 9.090922594070435
    },
    {
      "epoch": 1.4969825744628905e-05,
      "model_forward_time": 1.8117339611053467,
      "step": 89697
    },
    {
      "epoch": 1.4969825744628905e-05,
      "step": 89697,
      "training_step_time": 8.843743562698364
    },
    {
      "epoch": 1.4970779418945313e-05,
      "model_forward_time": 1.7911372184753418,
      "step": 89698
    },
    {
      "epoch": 1.4970779418945313e-05,
      "step": 89698,
      "training_step_time": 8.209739446640015
    },
    {
      "epoch": 1.4971733093261719e-05,
      "model_forward_time": 1.8112776279449463,
      "step": 89699
    },
    {
      "epoch": 1.4971733093261719e-05,
      "step": 89699,
      "training_step_time": 7.6900413036346436
    },
    {
      "epoch": 1.4972686767578124e-05,
      "grad_norm": 0.07853073626756668,
      "learning_rate": 2.87307912624582e-07,
      "loss": 0.0523,
      "step": 89700
    },
    {
      "epoch": 1.4972686767578124e-05,
      "model_forward_time": 1.872267484664917,
      "step": 89700
    },
    {
      "dynamics_loss_avg": 0.06703566862270235,
      "epoch": 1.4972686767578124e-05,
      "step": 89700
    },
    {
      "action_loss_avg": 0.0012708456604741513,
      "epoch": 1.4972686767578124e-05,
      "step": 89700
    },
    {
      "epoch": 1.4972686767578124e-05,
      "step": 89700,
      "training_step_time": 7.732455492019653
    },
    {
      "epoch": 1.4973640441894532e-05,
      "model_forward_time": 1.8137235641479492,
      "step": 89701
    },
    {
      "epoch": 1.4973640441894532e-05,
      "step": 89701,
      "training_step_time": 9.155659437179565
    },
    {
      "epoch": 1.4974594116210937e-05,
      "model_forward_time": 1.8030667304992676,
      "step": 89702
    },
    {
      "epoch": 1.4974594116210937e-05,
      "step": 89702,
      "training_step_time": 7.743042707443237
    },
    {
      "epoch": 1.4975547790527343e-05,
      "model_forward_time": 2.9744863510131836,
      "step": 89703
    },
    {
      "epoch": 1.4975547790527343e-05,
      "step": 89703,
      "training_step_time": 9.569459199905396
    },
    {
      "epoch": 1.497650146484375e-05,
      "model_forward_time": 1.8002617359161377,
      "step": 89704
    },
    {
      "epoch": 1.497650146484375e-05,
      "step": 89704,
      "training_step_time": 8.327330112457275
    },
    {
      "epoch": 1.4977455139160156e-05,
      "model_forward_time": 1.81642746925354,
      "step": 89705
    },
    {
      "epoch": 1.4977455139160156e-05,
      "step": 89705,
      "training_step_time": 8.34108018875122
    },
    {
      "epoch": 1.4978408813476562e-05,
      "model_forward_time": 1.809788465499878,
      "step": 89706
    },
    {
      "epoch": 1.4978408813476562e-05,
      "step": 89706,
      "training_step_time": 8.802945852279663
    },
    {
      "epoch": 1.497936248779297e-05,
      "model_forward_time": 1.8119585514068604,
      "step": 89707
    },
    {
      "epoch": 1.497936248779297e-05,
      "step": 89707,
      "training_step_time": 7.779309034347534
    },
    {
      "epoch": 1.4980316162109375e-05,
      "model_forward_time": 1.8153269290924072,
      "step": 89708
    },
    {
      "epoch": 1.4980316162109375e-05,
      "step": 89708,
      "training_step_time": 8.633427143096924
    },
    {
      "epoch": 1.4981269836425781e-05,
      "model_forward_time": 1.8697521686553955,
      "step": 89709
    },
    {
      "epoch": 1.4981269836425781e-05,
      "step": 89709,
      "training_step_time": 7.746236085891724
    },
    {
      "epoch": 1.4982223510742188e-05,
      "grad_norm": 0.09216336905956268,
      "learning_rate": 2.867557497133977e-07,
      "loss": 0.0533,
      "step": 89710
    },
    {
      "epoch": 1.4982223510742188e-05,
      "model_forward_time": 1.811107873916626,
      "step": 89710
    },
    {
      "dynamics_loss_avg": 0.055138074047863486,
      "epoch": 1.4982223510742188e-05,
      "step": 89710
    },
    {
      "action_loss_avg": 0.0010628724470734597,
      "epoch": 1.4982223510742188e-05,
      "step": 89710
    },
    {
      "epoch": 1.4982223510742188e-05,
      "step": 89710,
      "training_step_time": 7.565633058547974
    },
    {
      "epoch": 1.4983177185058594e-05,
      "model_forward_time": 1.801438808441162,
      "step": 89711
    },
    {
      "epoch": 1.4983177185058594e-05,
      "step": 89711,
      "training_step_time": 7.686283111572266
    },
    {
      "epoch": 1.4984130859375e-05,
      "model_forward_time": 2.8885979652404785,
      "step": 89712
    },
    {
      "epoch": 1.4984130859375e-05,
      "step": 89712,
      "training_step_time": 7.730898141860962
    },
    {
      "epoch": 1.4985084533691405e-05,
      "model_forward_time": 1.7947094440460205,
      "step": 89713
    },
    {
      "epoch": 1.4985084533691405e-05,
      "step": 89713,
      "training_step_time": 7.758858919143677
    },
    {
      "epoch": 1.4986038208007813e-05,
      "model_forward_time": 1.81015944480896,
      "step": 89714
    },
    {
      "epoch": 1.4986038208007813e-05,
      "step": 89714,
      "training_step_time": 8.335359811782837
    },
    {
      "epoch": 1.4986991882324219e-05,
      "model_forward_time": 1.8129174709320068,
      "step": 89715
    },
    {
      "epoch": 1.4986991882324219e-05,
      "step": 89715,
      "training_step_time": 9.283183097839355
    },
    {
      "epoch": 1.4987945556640624e-05,
      "model_forward_time": 1.8145809173583984,
      "step": 89716
    },
    {
      "epoch": 1.4987945556640624e-05,
      "step": 89716,
      "training_step_time": 8.612786769866943
    },
    {
      "epoch": 1.4988899230957032e-05,
      "model_forward_time": 1.7742722034454346,
      "step": 89717
    },
    {
      "epoch": 1.4988899230957032e-05,
      "step": 89717,
      "training_step_time": 8.31315302848816
    },
    {
      "epoch": 1.4989852905273437e-05,
      "model_forward_time": 1.854557991027832,
      "step": 89718
    },
    {
      "epoch": 1.4989852905273437e-05,
      "step": 89718,
      "training_step_time": 7.757458686828613
    },
    {
      "epoch": 1.4990806579589843e-05,
      "model_forward_time": 1.8116159439086914,
      "step": 89719
    },
    {
      "epoch": 1.4990806579589843e-05,
      "step": 89719,
      "training_step_time": 8.52545976638794
    },
    {
      "epoch": 1.499176025390625e-05,
      "grad_norm": 0.09184584766626358,
      "learning_rate": 2.8620410223551943e-07,
      "loss": 0.0531,
      "step": 89720
    },
    {
      "epoch": 1.499176025390625e-05,
      "model_forward_time": 1.7818772792816162,
      "step": 89720
    },
    {
      "dynamics_loss_avg": 0.058416043408215046,
      "epoch": 1.499176025390625e-05,
      "step": 89720
    },
    {
      "action_loss_avg": 0.0012805324913642835,
      "epoch": 1.499176025390625e-05,
      "step": 89720
    },
    {
      "epoch": 1.499176025390625e-05,
      "step": 89720,
      "training_step_time": 7.688664436340332
    },
    {
      "epoch": 1.4992713928222656e-05,
      "model_forward_time": 2.9000654220581055,
      "step": 89721
    },
    {
      "epoch": 1.4992713928222656e-05,
      "step": 89721,
      "training_step_time": 8.56783938407898
    },
    {
      "epoch": 1.4993667602539062e-05,
      "model_forward_time": 1.7697479724884033,
      "step": 89722
    },
    {
      "epoch": 1.4993667602539062e-05,
      "step": 89722,
      "training_step_time": 8.724955320358276
    },
    {
      "epoch": 1.499462127685547e-05,
      "model_forward_time": 1.8038647174835205,
      "step": 89723
    },
    {
      "epoch": 1.499462127685547e-05,
      "step": 89723,
      "training_step_time": 8.780885457992554
    },
    {
      "epoch": 1.4995574951171875e-05,
      "model_forward_time": 1.8091351985931396,
      "step": 89724
    },
    {
      "epoch": 1.4995574951171875e-05,
      "step": 89724,
      "training_step_time": 8.816762208938599
    },
    {
      "epoch": 1.4996528625488281e-05,
      "model_forward_time": 1.807971477508545,
      "step": 89725
    },
    {
      "epoch": 1.4996528625488281e-05,
      "step": 89725,
      "training_step_time": 8.49719786643982
    },
    {
      "epoch": 1.4997482299804688e-05,
      "model_forward_time": 1.801820993423462,
      "step": 89726
    },
    {
      "epoch": 1.4997482299804688e-05,
      "step": 89726,
      "training_step_time": 7.7558441162109375
    },
    {
      "epoch": 1.4998435974121094e-05,
      "model_forward_time": 1.8665239810943604,
      "step": 89727
    },
    {
      "epoch": 1.4998435974121094e-05,
      "step": 89727,
      "training_step_time": 7.71042799949646
    },
    {
      "epoch": 1.49993896484375e-05,
      "model_forward_time": 1.8040180206298828,
      "step": 89728
    },
    {
      "epoch": 1.49993896484375e-05,
      "step": 89728,
      "training_step_time": 7.748248100280762
    },
    {
      "epoch": 1.5000343322753905e-05,
      "model_forward_time": 1.7988388538360596,
      "step": 89729
    },
    {
      "epoch": 1.5000343322753905e-05,
      "step": 89729,
      "training_step_time": 8.668898582458496
    },
    {
      "epoch": 1.5001296997070313e-05,
      "grad_norm": 0.0944659560918808,
      "learning_rate": 2.8565297025127226e-07,
      "loss": 0.0524,
      "step": 89730
    },
    {
      "epoch": 1.5001296997070313e-05,
      "model_forward_time": 2.921617031097412,
      "step": 89730
    },
    {
      "dynamics_loss_avg": 0.04655268294736743,
      "epoch": 1.5001296997070313e-05,
      "step": 89730
    },
    {
      "action_loss_avg": 0.0012842409661971033,
      "epoch": 1.5001296997070313e-05,
      "step": 89730
    },
    {
      "epoch": 1.5001296997070313e-05,
      "step": 89730,
      "training_step_time": 7.760788679122925
    },
    {
      "epoch": 1.5002250671386719e-05,
      "model_forward_time": 1.8367648124694824,
      "step": 89731
    },
    {
      "epoch": 1.5002250671386719e-05,
      "step": 89731,
      "training_step_time": 7.763005971908569
    },
    {
      "epoch": 1.5003204345703124e-05,
      "model_forward_time": 1.8377718925476074,
      "step": 89732
    },
    {
      "epoch": 1.5003204345703124e-05,
      "step": 89732,
      "training_step_time": 7.712178707122803
    },
    {
      "epoch": 1.5004158020019532e-05,
      "model_forward_time": 1.8380272388458252,
      "step": 89733
    },
    {
      "epoch": 1.5004158020019532e-05,
      "step": 89733,
      "training_step_time": 7.684229612350464
    },
    {
      "epoch": 1.5005111694335937e-05,
      "model_forward_time": 1.851755142211914,
      "step": 89734
    },
    {
      "epoch": 1.5005111694335937e-05,
      "step": 89734,
      "training_step_time": 8.584258794784546
    },
    {
      "epoch": 1.5006065368652343e-05,
      "model_forward_time": 1.8296465873718262,
      "step": 89735
    },
    {
      "epoch": 1.5006065368652343e-05,
      "step": 89735,
      "training_step_time": 8.589914798736572
    },
    {
      "epoch": 1.500701904296875e-05,
      "model_forward_time": 1.7398836612701416,
      "step": 89736
    },
    {
      "epoch": 1.500701904296875e-05,
      "step": 89736,
      "training_step_time": 8.285809993743896
    },
    {
      "epoch": 1.5007972717285156e-05,
      "model_forward_time": 1.7497615814208984,
      "step": 89737
    },
    {
      "epoch": 1.5007972717285156e-05,
      "step": 89737,
      "training_step_time": 7.665910482406616
    },
    {
      "epoch": 1.5008926391601562e-05,
      "model_forward_time": 1.7383668422698975,
      "step": 89738
    },
    {
      "epoch": 1.5008926391601562e-05,
      "step": 89738,
      "training_step_time": 7.667331695556641
    },
    {
      "epoch": 1.500988006591797e-05,
      "model_forward_time": 2.387777090072632,
      "step": 89739
    },
    {
      "epoch": 1.500988006591797e-05,
      "step": 89739,
      "training_step_time": 7.636256694793701
    },
    {
      "epoch": 1.5010833740234375e-05,
      "grad_norm": 0.09885258972644806,
      "learning_rate": 2.8510235382092855e-07,
      "loss": 0.0538,
      "step": 89740
    },
    {
      "epoch": 1.5010833740234375e-05,
      "model_forward_time": 1.7574453353881836,
      "step": 89740
    },
    {
      "dynamics_loss_avg": 0.04695401955395937,
      "epoch": 1.5010833740234375e-05,
      "step": 89740
    },
    {
      "action_loss_avg": 0.0013394648965913802,
      "epoch": 1.5010833740234375e-05,
      "step": 89740
    },
    {
      "epoch": 1.5010833740234375e-05,
      "step": 89740,
      "training_step_time": 7.793270111083984
    },
    {
      "epoch": 1.5011787414550781e-05,
      "model_forward_time": 1.7395579814910889,
      "step": 89741
    },
    {
      "epoch": 1.5011787414550781e-05,
      "step": 89741,
      "training_step_time": 7.727998971939087
    },
    {
      "epoch": 1.5012741088867188e-05,
      "model_forward_time": 1.7387738227844238,
      "step": 89742
    },
    {
      "epoch": 1.5012741088867188e-05,
      "step": 89742,
      "training_step_time": 9.444284439086914
    },
    {
      "epoch": 1.5013694763183594e-05,
      "model_forward_time": 1.730557918548584,
      "step": 89743
    },
    {
      "epoch": 1.5013694763183594e-05,
      "step": 89743,
      "training_step_time": 8.840470790863037
    },
    {
      "epoch": 1.50146484375e-05,
      "model_forward_time": 1.7378630638122559,
      "step": 89744
    },
    {
      "epoch": 1.50146484375e-05,
      "step": 89744,
      "training_step_time": 8.693279266357422
    },
    {
      "epoch": 1.5015602111816406e-05,
      "model_forward_time": 1.7346785068511963,
      "step": 89745
    },
    {
      "epoch": 1.5015602111816406e-05,
      "step": 89745,
      "training_step_time": 8.549478769302368
    },
    {
      "epoch": 1.5016555786132813e-05,
      "model_forward_time": 1.7343671321868896,
      "step": 89746
    },
    {
      "epoch": 1.5016555786132813e-05,
      "step": 89746,
      "training_step_time": 7.732313871383667
    },
    {
      "epoch": 1.5017509460449219e-05,
      "model_forward_time": 1.7343926429748535,
      "step": 89747
    },
    {
      "epoch": 1.5017509460449219e-05,
      "step": 89747,
      "training_step_time": 9.833563327789307
    },
    {
      "epoch": 1.5018463134765624e-05,
      "model_forward_time": 2.363985300064087,
      "step": 89748
    },
    {
      "epoch": 1.5018463134765624e-05,
      "step": 89748,
      "training_step_time": 7.673704147338867
    },
    {
      "epoch": 1.5019416809082032e-05,
      "model_forward_time": 1.7531843185424805,
      "step": 89749
    },
    {
      "epoch": 1.5019416809082032e-05,
      "step": 89749,
      "training_step_time": 8.79756498336792
    },
    {
      "epoch": 1.5020370483398438e-05,
      "grad_norm": 0.11429889500141144,
      "learning_rate": 2.845522530047029e-07,
      "loss": 0.053,
      "step": 89750
    },
    {
      "epoch": 1.5020370483398438e-05,
      "model_forward_time": 1.735083818435669,
      "step": 89750
    },
    {
      "dynamics_loss_avg": 0.04842870514839888,
      "epoch": 1.5020370483398438e-05,
      "step": 89750
    },
    {
      "action_loss_avg": 0.001517225877614692,
      "epoch": 1.5020370483398438e-05,
      "step": 89750
    },
    {
      "epoch": 1.5020370483398438e-05,
      "step": 89750,
      "training_step_time": 8.580538272857666
    },
    {
      "epoch": 1.5021324157714843e-05,
      "model_forward_time": 1.7386696338653564,
      "step": 89751
    },
    {
      "epoch": 1.5021324157714843e-05,
      "step": 89751,
      "training_step_time": 10.306183576583862
    },
    {
      "epoch": 1.502227783203125e-05,
      "model_forward_time": 1.7403759956359863,
      "step": 89752
    },
    {
      "epoch": 1.502227783203125e-05,
      "step": 89752,
      "training_step_time": 7.827569007873535
    },
    {
      "epoch": 1.5023231506347656e-05,
      "model_forward_time": 1.7355148792266846,
      "step": 89753
    },
    {
      "epoch": 1.5023231506347656e-05,
      "step": 89753,
      "training_step_time": 8.662792205810547
    },
    {
      "epoch": 1.5024185180664062e-05,
      "model_forward_time": 1.73130464553833,
      "step": 89754
    },
    {
      "epoch": 1.5024185180664062e-05,
      "step": 89754,
      "training_step_time": 8.667479038238525
    },
    {
      "epoch": 1.502513885498047e-05,
      "model_forward_time": 1.7329390048980713,
      "step": 89755
    },
    {
      "epoch": 1.502513885498047e-05,
      "step": 89755,
      "training_step_time": 7.7614426612854
    },
    {
      "epoch": 1.5026092529296875e-05,
      "model_forward_time": 1.741563320159912,
      "step": 89756
    },
    {
      "epoch": 1.5026092529296875e-05,
      "step": 89756,
      "training_step_time": 7.685833215713501
    },
    {
      "epoch": 1.5027046203613281e-05,
      "model_forward_time": 2.3899641036987305,
      "step": 89757
    },
    {
      "epoch": 1.5027046203613281e-05,
      "step": 89757,
      "training_step_time": 7.622682094573975
    },
    {
      "epoch": 1.5027999877929687e-05,
      "model_forward_time": 1.7576208114624023,
      "step": 89758
    },
    {
      "epoch": 1.5027999877929687e-05,
      "step": 89758,
      "training_step_time": 8.631670951843262
    },
    {
      "epoch": 1.5028953552246094e-05,
      "model_forward_time": 1.7400562763214111,
      "step": 89759
    },
    {
      "epoch": 1.5028953552246094e-05,
      "step": 89759,
      "training_step_time": 8.4162757396698
    },
    {
      "epoch": 1.50299072265625e-05,
      "grad_norm": 0.09668074548244476,
      "learning_rate": 2.8400266786275387e-07,
      "loss": 0.0558,
      "step": 89760
    },
    {
      "epoch": 1.50299072265625e-05,
      "model_forward_time": 1.7236530780792236,
      "step": 89760
    },
    {
      "dynamics_loss_avg": 0.05065826307982206,
      "epoch": 1.50299072265625e-05,
      "step": 89760
    },
    {
      "action_loss_avg": 0.0010555104090599343,
      "epoch": 1.50299072265625e-05,
      "step": 89760
    },
    {
      "epoch": 1.50299072265625e-05,
      "step": 89760,
      "training_step_time": 7.763235807418823
    },
    {
      "epoch": 1.5030860900878906e-05,
      "model_forward_time": 1.7390944957733154,
      "step": 89761
    },
    {
      "epoch": 1.5030860900878906e-05,
      "step": 89761,
      "training_step_time": 8.271076440811157
    },
    {
      "epoch": 1.5031814575195313e-05,
      "model_forward_time": 1.7263917922973633,
      "step": 89762
    },
    {
      "epoch": 1.5031814575195313e-05,
      "step": 89762,
      "training_step_time": 7.735379934310913
    },
    {
      "epoch": 1.5032768249511719e-05,
      "model_forward_time": 1.748723030090332,
      "step": 89763
    },
    {
      "epoch": 1.5032768249511719e-05,
      "step": 89763,
      "training_step_time": 8.702242851257324
    },
    {
      "epoch": 1.5033721923828124e-05,
      "model_forward_time": 1.7498247623443604,
      "step": 89764
    },
    {
      "epoch": 1.5033721923828124e-05,
      "step": 89764,
      "training_step_time": 7.689094305038452
    },
    {
      "epoch": 1.5034675598144532e-05,
      "model_forward_time": 1.7451677322387695,
      "step": 89765
    },
    {
      "epoch": 1.5034675598144532e-05,
      "step": 89765,
      "training_step_time": 7.693445682525635
    },
    {
      "epoch": 1.5035629272460938e-05,
      "model_forward_time": 2.4235846996307373,
      "step": 89766
    },
    {
      "epoch": 1.5035629272460938e-05,
      "step": 89766,
      "training_step_time": 7.649132013320923
    },
    {
      "epoch": 1.5036582946777343e-05,
      "model_forward_time": 1.7713689804077148,
      "step": 89767
    },
    {
      "epoch": 1.5036582946777343e-05,
      "step": 89767,
      "training_step_time": 7.636624813079834
    },
    {
      "epoch": 1.503753662109375e-05,
      "model_forward_time": 1.7564857006072998,
      "step": 89768
    },
    {
      "epoch": 1.503753662109375e-05,
      "step": 89768,
      "training_step_time": 8.214724063873291
    },
    {
      "epoch": 1.5038490295410156e-05,
      "model_forward_time": 1.7459580898284912,
      "step": 89769
    },
    {
      "epoch": 1.5038490295410156e-05,
      "step": 89769,
      "training_step_time": 9.34694218635559
    },
    {
      "epoch": 1.5039443969726562e-05,
      "grad_norm": 0.09528882801532745,
      "learning_rate": 2.8345359845518173e-07,
      "loss": 0.0539,
      "step": 89770
    },
    {
      "epoch": 1.5039443969726562e-05,
      "model_forward_time": 1.744389295578003,
      "step": 89770
    },
    {
      "dynamics_loss_avg": 0.05750331766903401,
      "epoch": 1.5039443969726562e-05,
      "step": 89770
    },
    {
      "action_loss_avg": 0.0014102871806244365,
      "epoch": 1.5039443969726562e-05,
      "step": 89770
    },
    {
      "epoch": 1.5039443969726562e-05,
      "step": 89770,
      "training_step_time": 9.098096370697021
    },
    {
      "epoch": 1.504039764404297e-05,
      "model_forward_time": 1.7431797981262207,
      "step": 89771
    },
    {
      "epoch": 1.504039764404297e-05,
      "step": 89771,
      "training_step_time": 8.776458740234375
    },
    {
      "epoch": 1.5041351318359375e-05,
      "model_forward_time": 1.7485733032226562,
      "step": 89772
    },
    {
      "epoch": 1.5041351318359375e-05,
      "step": 89772,
      "training_step_time": 8.569230318069458
    },
    {
      "epoch": 1.5042304992675781e-05,
      "model_forward_time": 1.747002363204956,
      "step": 89773
    },
    {
      "epoch": 1.5042304992675781e-05,
      "step": 89773,
      "training_step_time": 7.773740530014038
    },
    {
      "epoch": 1.5043258666992187e-05,
      "model_forward_time": 1.7489509582519531,
      "step": 89774
    },
    {
      "epoch": 1.5043258666992187e-05,
      "step": 89774,
      "training_step_time": 7.742435693740845
    },
    {
      "epoch": 1.5044212341308594e-05,
      "model_forward_time": 2.4345271587371826,
      "step": 89775
    },
    {
      "epoch": 1.5044212341308594e-05,
      "step": 89775,
      "training_step_time": 7.597843885421753
    },
    {
      "epoch": 1.5045166015625e-05,
      "model_forward_time": 1.757439374923706,
      "step": 89776
    },
    {
      "epoch": 1.5045166015625e-05,
      "step": 89776,
      "training_step_time": 7.512133359909058
    },
    {
      "epoch": 1.5046119689941406e-05,
      "model_forward_time": 1.7476725578308105,
      "step": 89777
    },
    {
      "epoch": 1.5046119689941406e-05,
      "step": 89777,
      "training_step_time": 8.460899829864502
    },
    {
      "epoch": 1.5047073364257813e-05,
      "model_forward_time": 1.747138261795044,
      "step": 89778
    },
    {
      "epoch": 1.5047073364257813e-05,
      "step": 89778,
      "training_step_time": 7.624366044998169
    },
    {
      "epoch": 1.5048027038574219e-05,
      "model_forward_time": 1.7504842281341553,
      "step": 89779
    },
    {
      "epoch": 1.5048027038574219e-05,
      "step": 89779,
      "training_step_time": 7.63038969039917
    },
    {
      "epoch": 1.5048980712890624e-05,
      "grad_norm": 0.06915074586868286,
      "learning_rate": 2.82905044842034e-07,
      "loss": 0.0554,
      "step": 89780
    },
    {
      "epoch": 1.5048980712890624e-05,
      "model_forward_time": 1.747359037399292,
      "step": 89780
    },
    {
      "dynamics_loss_avg": 0.055195845477283,
      "epoch": 1.5048980712890624e-05,
      "step": 89780
    },
    {
      "action_loss_avg": 0.0009478500433033332,
      "epoch": 1.5048980712890624e-05,
      "step": 89780
    },
    {
      "epoch": 1.5048980712890624e-05,
      "step": 89780,
      "training_step_time": 7.689703941345215
    },
    {
      "epoch": 1.5049934387207032e-05,
      "model_forward_time": 1.7444391250610352,
      "step": 89781
    },
    {
      "epoch": 1.5049934387207032e-05,
      "step": 89781,
      "training_step_time": 7.636807441711426
    },
    {
      "epoch": 1.5050888061523438e-05,
      "model_forward_time": 1.7496566772460938,
      "step": 89782
    },
    {
      "epoch": 1.5050888061523438e-05,
      "step": 89782,
      "training_step_time": 7.636437892913818
    },
    {
      "epoch": 1.5051841735839843e-05,
      "model_forward_time": 1.7262516021728516,
      "step": 89783
    },
    {
      "epoch": 1.5051841735839843e-05,
      "step": 89783,
      "training_step_time": 7.540199041366577
    },
    {
      "epoch": 1.505279541015625e-05,
      "model_forward_time": 2.4194910526275635,
      "step": 89784
    },
    {
      "epoch": 1.505279541015625e-05,
      "step": 89784,
      "training_step_time": 7.622275114059448
    },
    {
      "epoch": 1.5053749084472656e-05,
      "model_forward_time": 1.7531709671020508,
      "step": 89785
    },
    {
      "epoch": 1.5053749084472656e-05,
      "step": 89785,
      "training_step_time": 7.662250995635986
    },
    {
      "epoch": 1.5054702758789062e-05,
      "model_forward_time": 1.7466356754302979,
      "step": 89786
    },
    {
      "epoch": 1.5054702758789062e-05,
      "step": 89786,
      "training_step_time": 7.735897541046143
    },
    {
      "epoch": 1.505565643310547e-05,
      "model_forward_time": 1.7410588264465332,
      "step": 89787
    },
    {
      "epoch": 1.505565643310547e-05,
      "step": 89787,
      "training_step_time": 7.703439235687256
    },
    {
      "epoch": 1.5056610107421875e-05,
      "model_forward_time": 1.743797779083252,
      "step": 89788
    },
    {
      "epoch": 1.5056610107421875e-05,
      "step": 89788,
      "training_step_time": 9.3008394241333
    },
    {
      "epoch": 1.5057563781738281e-05,
      "model_forward_time": 1.7448248863220215,
      "step": 89789
    },
    {
      "epoch": 1.5057563781738281e-05,
      "step": 89789,
      "training_step_time": 8.311723470687866
    },
    {
      "epoch": 1.5058517456054687e-05,
      "grad_norm": 0.06529811769723892,
      "learning_rate": 2.8235700708329705e-07,
      "loss": 0.0563,
      "step": 89790
    },
    {
      "epoch": 1.5058517456054687e-05,
      "model_forward_time": 1.7363309860229492,
      "step": 89790
    },
    {
      "dynamics_loss_avg": 0.05271726530045271,
      "epoch": 1.5058517456054687e-05,
      "step": 89790
    },
    {
      "action_loss_avg": 0.0012454714422347024,
      "epoch": 1.5058517456054687e-05,
      "step": 89790
    },
    {
      "epoch": 1.5058517456054687e-05,
      "step": 89790,
      "training_step_time": 7.6935155391693115
    },
    {
      "epoch": 1.5059471130371094e-05,
      "model_forward_time": 1.7476482391357422,
      "step": 89791
    },
    {
      "epoch": 1.5059471130371094e-05,
      "step": 89791,
      "training_step_time": 7.705103635787964
    },
    {
      "epoch": 1.50604248046875e-05,
      "model_forward_time": 1.7432081699371338,
      "step": 89792
    },
    {
      "epoch": 1.50604248046875e-05,
      "step": 89792,
      "training_step_time": 8.428425550460815
    },
    {
      "epoch": 1.5061378479003906e-05,
      "model_forward_time": 2.3848652839660645,
      "step": 89793
    },
    {
      "epoch": 1.5061378479003906e-05,
      "step": 89793,
      "training_step_time": 9.234208106994629
    },
    {
      "epoch": 1.5062332153320313e-05,
      "model_forward_time": 1.8305044174194336,
      "step": 89794
    },
    {
      "epoch": 1.5062332153320313e-05,
      "step": 89794,
      "training_step_time": 9.323963642120361
    },
    {
      "epoch": 1.5063285827636719e-05,
      "model_forward_time": 1.73995041847229,
      "step": 89795
    },
    {
      "epoch": 1.5063285827636719e-05,
      "step": 89795,
      "training_step_time": 7.704974889755249
    },
    {
      "epoch": 1.5064239501953124e-05,
      "model_forward_time": 1.740880012512207,
      "step": 89796
    },
    {
      "epoch": 1.5064239501953124e-05,
      "step": 89796,
      "training_step_time": 8.373616218566895
    },
    {
      "epoch": 1.5065193176269532e-05,
      "model_forward_time": 1.7472364902496338,
      "step": 89797
    },
    {
      "epoch": 1.5065193176269532e-05,
      "step": 89797,
      "training_step_time": 7.777278184890747
    },
    {
      "epoch": 1.5066146850585938e-05,
      "model_forward_time": 1.7396607398986816,
      "step": 89798
    },
    {
      "epoch": 1.5066146850585938e-05,
      "step": 89798,
      "training_step_time": 8.605286598205566
    },
    {
      "epoch": 1.5067100524902343e-05,
      "model_forward_time": 1.7443246841430664,
      "step": 89799
    },
    {
      "epoch": 1.5067100524902343e-05,
      "step": 89799,
      "training_step_time": 8.702603340148926
    },
    {
      "epoch": 1.506805419921875e-05,
      "grad_norm": 0.12777890264987946,
      "learning_rate": 2.8180948523890585e-07,
      "loss": 0.0554,
      "step": 89800
    },
    {
      "epoch": 1.506805419921875e-05,
      "model_forward_time": 1.741619348526001,
      "step": 89800
    },
    {
      "dynamics_loss_avg": 0.05173227926716208,
      "epoch": 1.506805419921875e-05,
      "step": 89800
    },
    {
      "action_loss_avg": 0.0012030583107843994,
      "epoch": 1.506805419921875e-05,
      "step": 89800
    },
    {
      "epoch": 1.506805419921875e-05,
      "step": 89800,
      "training_step_time": 7.729926586151123
    },
    {
      "epoch": 1.5069007873535156e-05,
      "model_forward_time": 1.7424325942993164,
      "step": 89801
    },
    {
      "epoch": 1.5069007873535156e-05,
      "step": 89801,
      "training_step_time": 7.739865779876709
    },
    {
      "epoch": 1.5069961547851562e-05,
      "model_forward_time": 2.389895439147949,
      "step": 89802
    },
    {
      "epoch": 1.5069961547851562e-05,
      "step": 89802,
      "training_step_time": 8.454667806625366
    },
    {
      "epoch": 1.507091522216797e-05,
      "model_forward_time": 1.8358705043792725,
      "step": 89803
    },
    {
      "epoch": 1.507091522216797e-05,
      "step": 89803,
      "training_step_time": 7.630600929260254
    },
    {
      "epoch": 1.5071868896484375e-05,
      "model_forward_time": 1.738743782043457,
      "step": 89804
    },
    {
      "epoch": 1.5071868896484375e-05,
      "step": 89804,
      "training_step_time": 7.592774391174316
    },
    {
      "epoch": 1.5072822570800781e-05,
      "model_forward_time": 1.7419934272766113,
      "step": 89805
    },
    {
      "epoch": 1.5072822570800781e-05,
      "step": 89805,
      "training_step_time": 7.686738014221191
    },
    {
      "epoch": 1.5073776245117187e-05,
      "model_forward_time": 1.749267578125,
      "step": 89806
    },
    {
      "epoch": 1.5073776245117187e-05,
      "step": 89806,
      "training_step_time": 8.397371768951416
    },
    {
      "epoch": 1.5074729919433594e-05,
      "model_forward_time": 1.7707343101501465,
      "step": 89807
    },
    {
      "epoch": 1.5074729919433594e-05,
      "step": 89807,
      "training_step_time": 8.766071081161499
    },
    {
      "epoch": 1.507568359375e-05,
      "model_forward_time": 1.743769645690918,
      "step": 89808
    },
    {
      "epoch": 1.507568359375e-05,
      "step": 89808,
      "training_step_time": 9.01710820198059
    },
    {
      "epoch": 1.5076637268066406e-05,
      "model_forward_time": 1.7386305332183838,
      "step": 89809
    },
    {
      "epoch": 1.5076637268066406e-05,
      "step": 89809,
      "training_step_time": 7.746166467666626
    },
    {
      "epoch": 1.5077590942382813e-05,
      "grad_norm": 0.0855003297328949,
      "learning_rate": 2.812624793687352e-07,
      "loss": 0.0546,
      "step": 89810
    },
    {
      "epoch": 1.5077590942382813e-05,
      "model_forward_time": 1.7322044372558594,
      "step": 89810
    },
    {
      "dynamics_loss_avg": 0.05760093089193106,
      "epoch": 1.5077590942382813e-05,
      "step": 89810
    },
    {
      "action_loss_avg": 0.0013683152967132628,
      "epoch": 1.5077590942382813e-05,
      "step": 89810
    },
    {
      "epoch": 1.5077590942382813e-05,
      "step": 89810,
      "training_step_time": 7.759483575820923
    },
    {
      "epoch": 1.5078544616699219e-05,
      "model_forward_time": 2.4189751148223877,
      "step": 89811
    },
    {
      "epoch": 1.5078544616699219e-05,
      "step": 89811,
      "training_step_time": 7.65804123878479
    },
    {
      "epoch": 1.5079498291015624e-05,
      "model_forward_time": 1.7997026443481445,
      "step": 89812
    },
    {
      "epoch": 1.5079498291015624e-05,
      "step": 89812,
      "training_step_time": 7.645210266113281
    },
    {
      "epoch": 1.5080451965332032e-05,
      "model_forward_time": 1.7480227947235107,
      "step": 89813
    },
    {
      "epoch": 1.5080451965332032e-05,
      "step": 89813,
      "training_step_time": 7.803987741470337
    },
    {
      "epoch": 1.5081405639648438e-05,
      "model_forward_time": 1.740006923675537,
      "step": 89814
    },
    {
      "epoch": 1.5081405639648438e-05,
      "step": 89814,
      "training_step_time": 9.471675157546997
    },
    {
      "epoch": 1.5082359313964843e-05,
      "model_forward_time": 1.7404084205627441,
      "step": 89815
    },
    {
      "epoch": 1.5082359313964843e-05,
      "step": 89815,
      "training_step_time": 8.727622032165527
    },
    {
      "epoch": 1.508331298828125e-05,
      "model_forward_time": 1.7183818817138672,
      "step": 89816
    },
    {
      "epoch": 1.508331298828125e-05,
      "step": 89816,
      "training_step_time": 9.407745122909546
    },
    {
      "epoch": 1.5084266662597656e-05,
      "model_forward_time": 1.7246217727661133,
      "step": 89817
    },
    {
      "epoch": 1.5084266662597656e-05,
      "step": 89817,
      "training_step_time": 8.393481731414795
    },
    {
      "epoch": 1.5085220336914062e-05,
      "model_forward_time": 1.722017765045166,
      "step": 89818
    },
    {
      "epoch": 1.5085220336914062e-05,
      "step": 89818,
      "training_step_time": 9.0326566696167
    },
    {
      "epoch": 1.508617401123047e-05,
      "model_forward_time": 1.7358219623565674,
      "step": 89819
    },
    {
      "epoch": 1.508617401123047e-05,
      "step": 89819,
      "training_step_time": 7.585965871810913
    },
    {
      "epoch": 1.5087127685546875e-05,
      "grad_norm": 0.08008912950754166,
      "learning_rate": 2.8071598953260616e-07,
      "loss": 0.0551,
      "step": 89820
    },
    {
      "epoch": 1.5087127685546875e-05,
      "model_forward_time": 2.3635287284851074,
      "step": 89820
    },
    {
      "dynamics_loss_avg": 0.041094516217708585,
      "epoch": 1.5087127685546875e-05,
      "step": 89820
    },
    {
      "action_loss_avg": 0.0015028060443000868,
      "epoch": 1.5087127685546875e-05,
      "step": 89820
    },
    {
      "epoch": 1.5087127685546875e-05,
      "step": 89820,
      "training_step_time": 7.575418949127197
    },
    {
      "epoch": 1.5088081359863281e-05,
      "model_forward_time": 1.771772861480713,
      "step": 89821
    },
    {
      "epoch": 1.5088081359863281e-05,
      "step": 89821,
      "training_step_time": 8.571414709091187
    },
    {
      "epoch": 1.5089035034179687e-05,
      "model_forward_time": 1.7493953704833984,
      "step": 89822
    },
    {
      "epoch": 1.5089035034179687e-05,
      "step": 89822,
      "training_step_time": 7.587249279022217
    },
    {
      "epoch": 1.5089988708496094e-05,
      "model_forward_time": 1.750812292098999,
      "step": 89823
    },
    {
      "epoch": 1.5089988708496094e-05,
      "step": 89823,
      "training_step_time": 7.6103901863098145
    },
    {
      "epoch": 1.50909423828125e-05,
      "model_forward_time": 1.7378606796264648,
      "step": 89824
    },
    {
      "epoch": 1.50909423828125e-05,
      "step": 89824,
      "training_step_time": 7.654895782470703
    },
    {
      "epoch": 1.5091896057128906e-05,
      "model_forward_time": 1.746995210647583,
      "step": 89825
    },
    {
      "epoch": 1.5091896057128906e-05,
      "step": 89825,
      "training_step_time": 7.656794309616089
    },
    {
      "epoch": 1.5092849731445313e-05,
      "model_forward_time": 1.7433111667633057,
      "step": 89826
    },
    {
      "epoch": 1.5092849731445313e-05,
      "step": 89826,
      "training_step_time": 7.615028619766235
    },
    {
      "epoch": 1.5093803405761719e-05,
      "model_forward_time": 1.73738431930542,
      "step": 89827
    },
    {
      "epoch": 1.5093803405761719e-05,
      "step": 89827,
      "training_step_time": 9.162206411361694
    },
    {
      "epoch": 1.5094757080078124e-05,
      "model_forward_time": 1.745845079421997,
      "step": 89828
    },
    {
      "epoch": 1.5094757080078124e-05,
      "step": 89828,
      "training_step_time": 8.300017356872559
    },
    {
      "epoch": 1.5095710754394532e-05,
      "model_forward_time": 2.38069486618042,
      "step": 89829
    },
    {
      "epoch": 1.5095710754394532e-05,
      "step": 89829,
      "training_step_time": 7.696873664855957
    },
    {
      "epoch": 1.5096664428710938e-05,
      "grad_norm": 0.07748987525701523,
      "learning_rate": 2.8017001579028036e-07,
      "loss": 0.0528,
      "step": 89830
    },
    {
      "epoch": 1.5096664428710938e-05,
      "model_forward_time": 1.753019094467163,
      "step": 89830
    },
    {
      "dynamics_loss_avg": 0.06355236060917377,
      "epoch": 1.5096664428710938e-05,
      "step": 89830
    },
    {
      "action_loss_avg": 0.0023935219855047763,
      "epoch": 1.5096664428710938e-05,
      "step": 89830
    },
    {
      "epoch": 1.5096664428710938e-05,
      "step": 89830,
      "training_step_time": 7.613598823547363
    },
    {
      "epoch": 1.5097618103027343e-05,
      "model_forward_time": 1.750441312789917,
      "step": 89831
    },
    {
      "epoch": 1.5097618103027343e-05,
      "step": 89831,
      "training_step_time": 7.706177234649658
    },
    {
      "epoch": 1.509857177734375e-05,
      "model_forward_time": 1.7437705993652344,
      "step": 89832
    },
    {
      "epoch": 1.509857177734375e-05,
      "step": 89832,
      "training_step_time": 7.6728835105896
    },
    {
      "epoch": 1.5099525451660156e-05,
      "model_forward_time": 1.7417824268341064,
      "step": 89833
    },
    {
      "epoch": 1.5099525451660156e-05,
      "step": 89833,
      "training_step_time": 8.229966402053833
    },
    {
      "epoch": 1.5100479125976562e-05,
      "model_forward_time": 1.7394962310791016,
      "step": 89834
    },
    {
      "epoch": 1.5100479125976562e-05,
      "step": 89834,
      "training_step_time": 7.73993992805481
    },
    {
      "epoch": 1.510143280029297e-05,
      "model_forward_time": 1.7498939037322998,
      "step": 89835
    },
    {
      "epoch": 1.510143280029297e-05,
      "step": 89835,
      "training_step_time": 8.083843231201172
    },
    {
      "epoch": 1.5102386474609375e-05,
      "model_forward_time": 1.7361299991607666,
      "step": 89836
    },
    {
      "epoch": 1.5102386474609375e-05,
      "step": 89836,
      "training_step_time": 7.711882829666138
    },
    {
      "epoch": 1.5103340148925781e-05,
      "model_forward_time": 1.7420592308044434,
      "step": 89837
    },
    {
      "epoch": 1.5103340148925781e-05,
      "step": 89837,
      "training_step_time": 7.79535436630249
    },
    {
      "epoch": 1.5104293823242187e-05,
      "model_forward_time": 2.3905105590820312,
      "step": 89838
    },
    {
      "epoch": 1.5104293823242187e-05,
      "step": 89838,
      "training_step_time": 7.787734508514404
    },
    {
      "epoch": 1.5105247497558594e-05,
      "model_forward_time": 1.743736743927002,
      "step": 89839
    },
    {
      "epoch": 1.5105247497558594e-05,
      "step": 89839,
      "training_step_time": 7.690050840377808
    },
    {
      "epoch": 1.5106201171875e-05,
      "grad_norm": 0.16516146063804626,
      "learning_rate": 2.796245582014662e-07,
      "loss": 0.0548,
      "step": 89840
    },
    {
      "epoch": 1.5106201171875e-05,
      "model_forward_time": 1.4568147659301758,
      "step": 89840
    },
    {
      "dynamics_loss_avg": 0.05472032148391008,
      "epoch": 1.5106201171875e-05,
      "step": 89840
    },
    {
      "action_loss_avg": 0.001679008424980566,
      "epoch": 1.5106201171875e-05,
      "step": 89840
    },
    {
      "epoch": 1.5106201171875e-05,
      "step": 89840,
      "training_step_time": 7.67177152633667
    },
    {
      "epoch": 1.5107154846191406e-05,
      "model_forward_time": 1.7477352619171143,
      "step": 89841
    },
    {
      "epoch": 1.5107154846191406e-05,
      "step": 89841,
      "training_step_time": 8.724861145019531
    },
    {
      "epoch": 1.5108108520507813e-05,
      "model_forward_time": 1.7364935874938965,
      "step": 89842
    },
    {
      "epoch": 1.5108108520507813e-05,
      "step": 89842,
      "training_step_time": 7.697426080703735
    },
    {
      "epoch": 1.5109062194824219e-05,
      "model_forward_time": 1.7417848110198975,
      "step": 89843
    },
    {
      "epoch": 1.5109062194824219e-05,
      "step": 89843,
      "training_step_time": 7.75855565071106
    },
    {
      "epoch": 1.5110015869140624e-05,
      "model_forward_time": 1.7464714050292969,
      "step": 89844
    },
    {
      "epoch": 1.5110015869140624e-05,
      "step": 89844,
      "training_step_time": 9.139271020889282
    },
    {
      "epoch": 1.5110969543457032e-05,
      "model_forward_time": 1.7368273735046387,
      "step": 89845
    },
    {
      "epoch": 1.5110969543457032e-05,
      "step": 89845,
      "training_step_time": 8.787842750549316
    },
    {
      "epoch": 1.5111923217773438e-05,
      "model_forward_time": 1.744567632675171,
      "step": 89846
    },
    {
      "epoch": 1.5111923217773438e-05,
      "step": 89846,
      "training_step_time": 7.680662393569946
    },
    {
      "epoch": 1.5112876892089843e-05,
      "model_forward_time": 2.3913986682891846,
      "step": 89847
    },
    {
      "epoch": 1.5112876892089843e-05,
      "step": 89847,
      "training_step_time": 7.730947971343994
    },
    {
      "epoch": 1.511383056640625e-05,
      "model_forward_time": 1.7395946979522705,
      "step": 89848
    },
    {
      "epoch": 1.511383056640625e-05,
      "step": 89848,
      "training_step_time": 7.784499168395996
    },
    {
      "epoch": 1.5114784240722656e-05,
      "model_forward_time": 1.8198649883270264,
      "step": 89849
    },
    {
      "epoch": 1.5114784240722656e-05,
      "step": 89849,
      "training_step_time": 7.748682260513306
    },
    {
      "epoch": 1.5115737915039062e-05,
      "grad_norm": 0.07963020354509354,
      "learning_rate": 2.790796168258125e-07,
      "loss": 0.0546,
      "step": 89850
    },
    {
      "epoch": 1.5115737915039062e-05,
      "model_forward_time": 2.804840326309204,
      "step": 89850
    },
    {
      "dynamics_loss_avg": 0.05022202273830771,
      "epoch": 1.5115737915039062e-05,
      "step": 89850
    },
    {
      "action_loss_avg": 0.0009103144853725098,
      "epoch": 1.5115737915039062e-05,
      "step": 89850
    },
    {
      "epoch": 1.5115737915039062e-05,
      "step": 89850,
      "training_step_time": 10.008070945739746
    },
    {
      "epoch": 1.511669158935547e-05,
      "model_forward_time": 1.714329481124878,
      "step": 89851
    },
    {
      "epoch": 1.511669158935547e-05,
      "step": 89851,
      "training_step_time": 6.84164834022522
    },
    {
      "epoch": 1.5117645263671875e-05,
      "model_forward_time": 1.7295165061950684,
      "step": 89852
    },
    {
      "epoch": 1.5117645263671875e-05,
      "step": 89852,
      "training_step_time": 7.5754523277282715
    },
    {
      "epoch": 1.5118598937988281e-05,
      "model_forward_time": 1.730649471282959,
      "step": 89853
    },
    {
      "epoch": 1.5118598937988281e-05,
      "step": 89853,
      "training_step_time": 8.27992844581604
    },
    {
      "epoch": 1.5119552612304687e-05,
      "model_forward_time": 1.7210216522216797,
      "step": 89854
    },
    {
      "epoch": 1.5119552612304687e-05,
      "step": 89854,
      "training_step_time": 7.64946985244751
    },
    {
      "epoch": 1.5120506286621094e-05,
      "model_forward_time": 1.7234587669372559,
      "step": 89855
    },
    {
      "epoch": 1.5120506286621094e-05,
      "step": 89855,
      "training_step_time": 6.8235626220703125
    },
    {
      "epoch": 1.51214599609375e-05,
      "model_forward_time": 1.7283945083618164,
      "step": 89856
    },
    {
      "epoch": 1.51214599609375e-05,
      "step": 89856,
      "training_step_time": 6.782557487487793
    },
    {
      "epoch": 1.5122413635253906e-05,
      "model_forward_time": 1.7348451614379883,
      "step": 89857
    },
    {
      "epoch": 1.5122413635253906e-05,
      "step": 89857,
      "training_step_time": 6.784220933914185
    },
    {
      "epoch": 1.5123367309570313e-05,
      "model_forward_time": 1.7332172393798828,
      "step": 89858
    },
    {
      "epoch": 1.5123367309570313e-05,
      "step": 89858,
      "training_step_time": 6.857000827789307
    },
    {
      "epoch": 1.5124320983886719e-05,
      "model_forward_time": 2.3641247749328613,
      "step": 89859
    },
    {
      "epoch": 1.5124320983886719e-05,
      "step": 89859,
      "training_step_time": 8.641318082809448
    },
    {
      "epoch": 1.5125274658203124e-05,
      "grad_norm": 0.10080168396234512,
      "learning_rate": 2.7853519172291454e-07,
      "loss": 0.055,
      "step": 89860
    },
    {
      "epoch": 1.5125274658203124e-05,
      "model_forward_time": 1.7180895805358887,
      "step": 89860
    },
    {
      "dynamics_loss_avg": 0.06654860880225896,
      "epoch": 1.5125274658203124e-05,
      "step": 89860
    },
    {
      "action_loss_avg": 0.0020526218810118734,
      "epoch": 1.5125274658203124e-05,
      "step": 89860
    },
    {
      "epoch": 1.5125274658203124e-05,
      "step": 89860,
      "training_step_time": 8.62122917175293
    },
    {
      "epoch": 1.5126228332519532e-05,
      "model_forward_time": 1.7278692722320557,
      "step": 89861
    },
    {
      "epoch": 1.5126228332519532e-05,
      "step": 89861,
      "training_step_time": 6.773267984390259
    },
    {
      "epoch": 1.5127182006835938e-05,
      "model_forward_time": 1.7199361324310303,
      "step": 89862
    },
    {
      "epoch": 1.5127182006835938e-05,
      "step": 89862,
      "training_step_time": 8.591581583023071
    },
    {
      "epoch": 1.5128135681152343e-05,
      "model_forward_time": 1.724823236465454,
      "step": 89863
    },
    {
      "epoch": 1.5128135681152343e-05,
      "step": 89863,
      "training_step_time": 7.145142555236816
    },
    {
      "epoch": 1.512908935546875e-05,
      "model_forward_time": 1.7286264896392822,
      "step": 89864
    },
    {
      "epoch": 1.512908935546875e-05,
      "step": 89864,
      "training_step_time": 6.92668080329895
    },
    {
      "epoch": 1.5130043029785156e-05,
      "model_forward_time": 1.7331154346466064,
      "step": 89865
    },
    {
      "epoch": 1.5130043029785156e-05,
      "step": 89865,
      "training_step_time": 6.88478684425354
    },
    {
      "epoch": 1.5130996704101562e-05,
      "model_forward_time": 1.7280564308166504,
      "step": 89866
    },
    {
      "epoch": 1.5130996704101562e-05,
      "step": 89866,
      "training_step_time": 6.8496620655059814
    },
    {
      "epoch": 1.513195037841797e-05,
      "model_forward_time": 1.7241780757904053,
      "step": 89867
    },
    {
      "epoch": 1.513195037841797e-05,
      "step": 89867,
      "training_step_time": 6.853514194488525
    },
    {
      "epoch": 1.5132904052734375e-05,
      "model_forward_time": 1.7279706001281738,
      "step": 89868
    },
    {
      "epoch": 1.5132904052734375e-05,
      "step": 89868,
      "training_step_time": 7.353652477264404
    },
    {
      "epoch": 1.5133857727050781e-05,
      "model_forward_time": 2.3795392513275146,
      "step": 89869
    },
    {
      "epoch": 1.5133857727050781e-05,
      "step": 89869,
      "training_step_time": 8.713911294937134
    },
    {
      "epoch": 1.5134811401367187e-05,
      "grad_norm": 0.10294566303491592,
      "learning_rate": 2.7799128295230904e-07,
      "loss": 0.0534,
      "step": 89870
    },
    {
      "epoch": 1.5134811401367187e-05,
      "model_forward_time": 1.7267732620239258,
      "step": 89870
    },
    {
      "dynamics_loss_avg": 0.058545629307627675,
      "epoch": 1.5134811401367187e-05,
      "step": 89870
    },
    {
      "action_loss_avg": 0.0015827499330043793,
      "epoch": 1.5134811401367187e-05,
      "step": 89870
    },
    {
      "epoch": 1.5134811401367187e-05,
      "step": 89870,
      "training_step_time": 7.600226163864136
    },
    {
      "epoch": 1.5135765075683594e-05,
      "model_forward_time": 1.7232229709625244,
      "step": 89871
    },
    {
      "epoch": 1.5135765075683594e-05,
      "step": 89871,
      "training_step_time": 8.028769969940186
    },
    {
      "epoch": 1.513671875e-05,
      "model_forward_time": 1.7221052646636963,
      "step": 89872
    },
    {
      "epoch": 1.513671875e-05,
      "step": 89872,
      "training_step_time": 7.655698537826538
    },
    {
      "epoch": 1.5137672424316406e-05,
      "model_forward_time": 1.7301194667816162,
      "step": 89873
    },
    {
      "epoch": 1.5137672424316406e-05,
      "step": 89873,
      "training_step_time": 6.812536954879761
    },
    {
      "epoch": 1.5138626098632813e-05,
      "model_forward_time": 1.7253270149230957,
      "step": 89874
    },
    {
      "epoch": 1.5138626098632813e-05,
      "step": 89874,
      "training_step_time": 6.755571603775024
    },
    {
      "epoch": 1.5139579772949219e-05,
      "model_forward_time": 1.729247808456421,
      "step": 89875
    },
    {
      "epoch": 1.5139579772949219e-05,
      "step": 89875,
      "training_step_time": 6.831956148147583
    },
    {
      "epoch": 1.5140533447265624e-05,
      "model_forward_time": 1.722557544708252,
      "step": 89876
    },
    {
      "epoch": 1.5140533447265624e-05,
      "step": 89876,
      "training_step_time": 6.798937559127808
    },
    {
      "epoch": 1.5141487121582032e-05,
      "model_forward_time": 1.721630334854126,
      "step": 89877
    },
    {
      "epoch": 1.5141487121582032e-05,
      "step": 89877,
      "training_step_time": 6.729660749435425
    },
    {
      "epoch": 1.5142440795898438e-05,
      "model_forward_time": 2.4018187522888184,
      "step": 89878
    },
    {
      "epoch": 1.5142440795898438e-05,
      "step": 89878,
      "training_step_time": 8.748722553253174
    },
    {
      "epoch": 1.5143394470214843e-05,
      "model_forward_time": 1.7279160022735596,
      "step": 89879
    },
    {
      "epoch": 1.5143394470214843e-05,
      "step": 89879,
      "training_step_time": 7.729767084121704
    },
    {
      "epoch": 1.514434814453125e-05,
      "grad_norm": 0.09048579633235931,
      "learning_rate": 2.774478905734784e-07,
      "loss": 0.0526,
      "step": 89880
    },
    {
      "epoch": 1.514434814453125e-05,
      "model_forward_time": 1.7184221744537354,
      "step": 89880
    },
    {
      "dynamics_loss_avg": 0.05979106929153204,
      "epoch": 1.514434814453125e-05,
      "step": 89880
    },
    {
      "action_loss_avg": 0.0009739373112097382,
      "epoch": 1.514434814453125e-05,
      "step": 89880
    },
    {
      "epoch": 1.514434814453125e-05,
      "step": 89880,
      "training_step_time": 7.450388431549072
    },
    {
      "epoch": 1.5145301818847656e-05,
      "model_forward_time": 1.7244219779968262,
      "step": 89881
    },
    {
      "epoch": 1.5145301818847656e-05,
      "step": 89881,
      "training_step_time": 7.59746789932251
    },
    {
      "epoch": 1.5146255493164062e-05,
      "model_forward_time": 1.7292110919952393,
      "step": 89882
    },
    {
      "epoch": 1.5146255493164062e-05,
      "step": 89882,
      "training_step_time": 6.850894451141357
    },
    {
      "epoch": 1.5147209167480468e-05,
      "model_forward_time": 1.7293000221252441,
      "step": 89883
    },
    {
      "epoch": 1.5147209167480468e-05,
      "step": 89883,
      "training_step_time": 6.77618932723999
    },
    {
      "epoch": 1.5148162841796875e-05,
      "model_forward_time": 1.7372050285339355,
      "step": 89884
    },
    {
      "epoch": 1.5148162841796875e-05,
      "step": 89884,
      "training_step_time": 6.844820737838745
    },
    {
      "epoch": 1.5149116516113281e-05,
      "model_forward_time": 1.3377156257629395,
      "step": 89885
    },
    {
      "epoch": 1.5149116516113281e-05,
      "step": 89885,
      "training_step_time": 6.84203839302063
    },
    {
      "epoch": 1.5150070190429687e-05,
      "model_forward_time": 1.7275691032409668,
      "step": 89886
    },
    {
      "epoch": 1.5150070190429687e-05,
      "step": 89886,
      "training_step_time": 6.869509696960449
    },
    {
      "epoch": 1.5151023864746094e-05,
      "model_forward_time": 2.397779941558838,
      "step": 89887
    },
    {
      "epoch": 1.5151023864746094e-05,
      "step": 89887,
      "training_step_time": 9.983573198318481
    },
    {
      "epoch": 1.51519775390625e-05,
      "model_forward_time": 1.7277414798736572,
      "step": 89888
    },
    {
      "epoch": 1.51519775390625e-05,
      "step": 89888,
      "training_step_time": 7.833671808242798
    },
    {
      "epoch": 1.5152931213378906e-05,
      "model_forward_time": 1.7227470874786377,
      "step": 89889
    },
    {
      "epoch": 1.5152931213378906e-05,
      "step": 89889,
      "training_step_time": 7.439208269119263
    },
    {
      "epoch": 1.5153884887695313e-05,
      "grad_norm": 0.07329369336366653,
      "learning_rate": 2.769050146458452e-07,
      "loss": 0.052,
      "step": 89890
    },
    {
      "epoch": 1.5153884887695313e-05,
      "model_forward_time": 1.7177278995513916,
      "step": 89890
    },
    {
      "dynamics_loss_avg": 0.055025983788073066,
      "epoch": 1.5153884887695313e-05,
      "step": 89890
    },
    {
      "action_loss_avg": 0.0012590923230163752,
      "epoch": 1.5153884887695313e-05,
      "step": 89890
    },
    {
      "epoch": 1.5153884887695313e-05,
      "step": 89890,
      "training_step_time": 7.732468605041504
    },
    {
      "epoch": 1.5154838562011719e-05,
      "model_forward_time": 1.7302522659301758,
      "step": 89891
    },
    {
      "epoch": 1.5154838562011719e-05,
      "step": 89891,
      "training_step_time": 6.801077604293823
    },
    {
      "epoch": 1.5155792236328124e-05,
      "model_forward_time": 1.7282772064208984,
      "step": 89892
    },
    {
      "epoch": 1.5155792236328124e-05,
      "step": 89892,
      "training_step_time": 6.773067951202393
    },
    {
      "epoch": 1.5156745910644532e-05,
      "model_forward_time": 1.721526861190796,
      "step": 89893
    },
    {
      "epoch": 1.5156745910644532e-05,
      "step": 89893,
      "training_step_time": 6.76642632484436
    },
    {
      "epoch": 1.5157699584960938e-05,
      "model_forward_time": 1.71982741355896,
      "step": 89894
    },
    {
      "epoch": 1.5157699584960938e-05,
      "step": 89894,
      "training_step_time": 6.813763618469238
    },
    {
      "epoch": 1.5158653259277343e-05,
      "model_forward_time": 1.7259433269500732,
      "step": 89895
    },
    {
      "epoch": 1.5158653259277343e-05,
      "step": 89895,
      "training_step_time": 6.832637786865234
    },
    {
      "epoch": 1.515960693359375e-05,
      "model_forward_time": 2.0073065757751465,
      "step": 89896
    },
    {
      "epoch": 1.515960693359375e-05,
      "step": 89896,
      "training_step_time": 7.806610822677612
    },
    {
      "epoch": 1.5160560607910156e-05,
      "model_forward_time": 1.7325403690338135,
      "step": 89897
    },
    {
      "epoch": 1.5160560607910156e-05,
      "step": 89897,
      "training_step_time": 8.661303043365479
    },
    {
      "epoch": 1.5161514282226562e-05,
      "model_forward_time": 1.7251677513122559,
      "step": 89898
    },
    {
      "epoch": 1.5161514282226562e-05,
      "step": 89898,
      "training_step_time": 7.454391241073608
    },
    {
      "epoch": 1.5162467956542968e-05,
      "model_forward_time": 1.7404260635375977,
      "step": 89899
    },
    {
      "epoch": 1.5162467956542968e-05,
      "step": 89899,
      "training_step_time": 7.629742383956909
    },
    {
      "epoch": 1.5163421630859375e-05,
      "grad_norm": 0.07635075598955154,
      "learning_rate": 2.763626552287796e-07,
      "loss": 0.0544,
      "step": 89900
    },
    {
      "epoch": 1.5163421630859375e-05,
      "model_forward_time": 1.7278051376342773,
      "step": 89900
    },
    {
      "dynamics_loss_avg": 0.04963477570563555,
      "epoch": 1.5163421630859375e-05,
      "step": 89900
    },
    {
      "action_loss_avg": 0.0008522570336936041,
      "epoch": 1.5163421630859375e-05,
      "step": 89900
    },
    {
      "epoch": 1.5163421630859375e-05,
      "step": 89900,
      "training_step_time": 7.271578073501587
    },
    {
      "epoch": 1.5164375305175781e-05,
      "model_forward_time": 1.7286465167999268,
      "step": 89901
    },
    {
      "epoch": 1.5164375305175781e-05,
      "step": 89901,
      "training_step_time": 6.805586099624634
    },
    {
      "epoch": 1.5165328979492187e-05,
      "model_forward_time": 1.726297378540039,
      "step": 89902
    },
    {
      "epoch": 1.5165328979492187e-05,
      "step": 89902,
      "training_step_time": 6.8109071254730225
    },
    {
      "epoch": 1.5166282653808594e-05,
      "model_forward_time": 1.7234361171722412,
      "step": 89903
    },
    {
      "epoch": 1.5166282653808594e-05,
      "step": 89903,
      "training_step_time": 6.764490604400635
    },
    {
      "epoch": 1.5167236328125e-05,
      "model_forward_time": 1.7213835716247559,
      "step": 89904
    },
    {
      "epoch": 1.5167236328125e-05,
      "step": 89904,
      "training_step_time": 6.810859441757202
    },
    {
      "epoch": 1.5168190002441406e-05,
      "model_forward_time": 2.3992605209350586,
      "step": 89905
    },
    {
      "epoch": 1.5168190002441406e-05,
      "step": 89905,
      "training_step_time": 7.799493074417114
    },
    {
      "epoch": 1.5169143676757813e-05,
      "model_forward_time": 1.7283174991607666,
      "step": 89906
    },
    {
      "epoch": 1.5169143676757813e-05,
      "step": 89906,
      "training_step_time": 8.856076955795288
    },
    {
      "epoch": 1.5170097351074219e-05,
      "model_forward_time": 1.727299451828003,
      "step": 89907
    },
    {
      "epoch": 1.5170097351074219e-05,
      "step": 89907,
      "training_step_time": 7.612260103225708
    },
    {
      "epoch": 1.5171051025390624e-05,
      "model_forward_time": 1.7286274433135986,
      "step": 89908
    },
    {
      "epoch": 1.5171051025390624e-05,
      "step": 89908,
      "training_step_time": 7.683646202087402
    },
    {
      "epoch": 1.5172004699707032e-05,
      "model_forward_time": 1.7322769165039062,
      "step": 89909
    },
    {
      "epoch": 1.5172004699707032e-05,
      "step": 89909,
      "training_step_time": 7.323389053344727
    },
    {
      "epoch": 1.5172958374023438e-05,
      "grad_norm": 0.07998621463775635,
      "learning_rate": 2.7582081238159086e-07,
      "loss": 0.0528,
      "step": 89910
    },
    {
      "epoch": 1.5172958374023438e-05,
      "model_forward_time": 1.7236952781677246,
      "step": 89910
    },
    {
      "dynamics_loss_avg": 0.055875512212514876,
      "epoch": 1.5172958374023438e-05,
      "step": 89910
    },
    {
      "action_loss_avg": 0.0020435061363969,
      "epoch": 1.5172958374023438e-05,
      "step": 89910
    },
    {
      "epoch": 1.5172958374023438e-05,
      "step": 89910,
      "training_step_time": 6.756283760070801
    },
    {
      "epoch": 1.5173912048339843e-05,
      "model_forward_time": 1.7182509899139404,
      "step": 89911
    },
    {
      "epoch": 1.5173912048339843e-05,
      "step": 89911,
      "training_step_time": 6.7945568561553955
    },
    {
      "epoch": 1.517486572265625e-05,
      "model_forward_time": 1.7203471660614014,
      "step": 89912
    },
    {
      "epoch": 1.517486572265625e-05,
      "step": 89912,
      "training_step_time": 6.801812648773193
    },
    {
      "epoch": 1.5175819396972656e-05,
      "model_forward_time": 1.7236685752868652,
      "step": 89913
    },
    {
      "epoch": 1.5175819396972656e-05,
      "step": 89913,
      "training_step_time": 6.767064571380615
    },
    {
      "epoch": 1.5176773071289062e-05,
      "model_forward_time": 1.7278366088867188,
      "step": 89914
    },
    {
      "epoch": 1.5176773071289062e-05,
      "step": 89914,
      "training_step_time": 9.29422903060913
    },
    {
      "epoch": 1.5177726745605468e-05,
      "model_forward_time": 1.729292631149292,
      "step": 89915
    },
    {
      "epoch": 1.5177726745605468e-05,
      "step": 89915,
      "training_step_time": 9.297016143798828
    },
    {
      "epoch": 1.5178680419921875e-05,
      "model_forward_time": 1.7255322933197021,
      "step": 89916
    },
    {
      "epoch": 1.5178680419921875e-05,
      "step": 89916,
      "training_step_time": 7.631205320358276
    },
    {
      "epoch": 1.5179634094238281e-05,
      "model_forward_time": 1.7319657802581787,
      "step": 89917
    },
    {
      "epoch": 1.5179634094238281e-05,
      "step": 89917,
      "training_step_time": 8.03697156906128
    },
    {
      "epoch": 1.5180587768554687e-05,
      "model_forward_time": 1.7260115146636963,
      "step": 89918
    },
    {
      "epoch": 1.5180587768554687e-05,
      "step": 89918,
      "training_step_time": 7.785341024398804
    },
    {
      "epoch": 1.5181541442871094e-05,
      "model_forward_time": 1.7246489524841309,
      "step": 89919
    },
    {
      "epoch": 1.5181541442871094e-05,
      "step": 89919,
      "training_step_time": 6.768344402313232
    },
    {
      "epoch": 1.51824951171875e-05,
      "grad_norm": 0.11491798609495163,
      "learning_rate": 2.752794861635355e-07,
      "loss": 0.0518,
      "step": 89920
    },
    {
      "epoch": 1.51824951171875e-05,
      "model_forward_time": 1.7245080471038818,
      "step": 89920
    },
    {
      "dynamics_loss_avg": 0.054649888165295125,
      "epoch": 1.51824951171875e-05,
      "step": 89920
    },
    {
      "action_loss_avg": 0.0010039972316008062,
      "epoch": 1.51824951171875e-05,
      "step": 89920
    },
    {
      "epoch": 1.51824951171875e-05,
      "step": 89920,
      "training_step_time": 6.777231693267822
    },
    {
      "epoch": 1.5183448791503906e-05,
      "model_forward_time": 1.728367805480957,
      "step": 89921
    },
    {
      "epoch": 1.5183448791503906e-05,
      "step": 89921,
      "training_step_time": 6.748273849487305
    },
    {
      "epoch": 1.5184402465820313e-05,
      "model_forward_time": 1.7310352325439453,
      "step": 89922
    },
    {
      "epoch": 1.5184402465820313e-05,
      "step": 89922,
      "training_step_time": 6.738634824752808
    },
    {
      "epoch": 1.5185356140136719e-05,
      "model_forward_time": 1.7274277210235596,
      "step": 89923
    },
    {
      "epoch": 1.5185356140136719e-05,
      "step": 89923,
      "training_step_time": 6.745704412460327
    },
    {
      "epoch": 1.5186309814453125e-05,
      "model_forward_time": 2.372302532196045,
      "step": 89924
    },
    {
      "epoch": 1.5186309814453125e-05,
      "step": 89924,
      "training_step_time": 10.670043230056763
    },
    {
      "epoch": 1.5187263488769532e-05,
      "model_forward_time": 1.7317347526550293,
      "step": 89925
    },
    {
      "epoch": 1.5187263488769532e-05,
      "step": 89925,
      "training_step_time": 9.105538129806519
    },
    {
      "epoch": 1.5188217163085938e-05,
      "model_forward_time": 1.7269761562347412,
      "step": 89926
    },
    {
      "epoch": 1.5188217163085938e-05,
      "step": 89926,
      "training_step_time": 7.690308332443237
    },
    {
      "epoch": 1.5189170837402343e-05,
      "model_forward_time": 1.725055456161499,
      "step": 89927
    },
    {
      "epoch": 1.5189170837402343e-05,
      "step": 89927,
      "training_step_time": 7.7067999839782715
    },
    {
      "epoch": 1.519012451171875e-05,
      "model_forward_time": 1.7275304794311523,
      "step": 89928
    },
    {
      "epoch": 1.519012451171875e-05,
      "step": 89928,
      "training_step_time": 6.759028196334839
    },
    {
      "epoch": 1.5191078186035157e-05,
      "model_forward_time": 1.7249438762664795,
      "step": 89929
    },
    {
      "epoch": 1.5191078186035157e-05,
      "step": 89929,
      "training_step_time": 6.74460768699646
    },
    {
      "epoch": 1.5192031860351562e-05,
      "grad_norm": 0.13929115235805511,
      "learning_rate": 2.7473867663381225e-07,
      "loss": 0.0538,
      "step": 89930
    },
    {
      "epoch": 1.5192031860351562e-05,
      "model_forward_time": 1.7232298851013184,
      "step": 89930
    },
    {
      "dynamics_loss_avg": 0.051965808030217886,
      "epoch": 1.5192031860351562e-05,
      "step": 89930
    },
    {
      "action_loss_avg": 0.0013739044676185585,
      "epoch": 1.5192031860351562e-05,
      "step": 89930
    },
    {
      "epoch": 1.5192031860351562e-05,
      "step": 89930,
      "training_step_time": 6.811912775039673
    },
    {
      "epoch": 1.5192985534667968e-05,
      "model_forward_time": 1.7207322120666504,
      "step": 89931
    },
    {
      "epoch": 1.5192985534667968e-05,
      "step": 89931,
      "training_step_time": 6.8140785694122314
    },
    {
      "epoch": 1.5193939208984375e-05,
      "model_forward_time": 1.7229232788085938,
      "step": 89932
    },
    {
      "epoch": 1.5193939208984375e-05,
      "step": 89932,
      "training_step_time": 6.824210166931152
    },
    {
      "epoch": 1.5194892883300781e-05,
      "model_forward_time": 2.4415981769561768,
      "step": 89933
    },
    {
      "epoch": 1.5194892883300781e-05,
      "step": 89933,
      "training_step_time": 11.646705865859985
    },
    {
      "epoch": 1.5195846557617187e-05,
      "model_forward_time": 1.7221386432647705,
      "step": 89934
    },
    {
      "epoch": 1.5195846557617187e-05,
      "step": 89934,
      "training_step_time": 9.690239667892456
    },
    {
      "epoch": 1.5196800231933594e-05,
      "model_forward_time": 1.7185890674591064,
      "step": 89935
    },
    {
      "epoch": 1.5196800231933594e-05,
      "step": 89935,
      "training_step_time": 7.616177082061768
    },
    {
      "epoch": 1.519775390625e-05,
      "model_forward_time": 1.7118213176727295,
      "step": 89936
    },
    {
      "epoch": 1.519775390625e-05,
      "step": 89936,
      "training_step_time": 7.657110214233398
    },
    {
      "epoch": 1.5198707580566406e-05,
      "model_forward_time": 1.7211353778839111,
      "step": 89937
    },
    {
      "epoch": 1.5198707580566406e-05,
      "step": 89937,
      "training_step_time": 7.2286622524261475
    },
    {
      "epoch": 1.5199661254882813e-05,
      "model_forward_time": 1.730391263961792,
      "step": 89938
    },
    {
      "epoch": 1.5199661254882813e-05,
      "step": 89938,
      "training_step_time": 6.7526280879974365
    },
    {
      "epoch": 1.5200614929199219e-05,
      "model_forward_time": 1.7237470149993896,
      "step": 89939
    },
    {
      "epoch": 1.5200614929199219e-05,
      "step": 89939,
      "training_step_time": 6.767984390258789
    },
    {
      "epoch": 1.5201568603515625e-05,
      "grad_norm": 0.07357333600521088,
      "learning_rate": 2.7419838385156385e-07,
      "loss": 0.0524,
      "step": 89940
    },
    {
      "epoch": 1.5201568603515625e-05,
      "model_forward_time": 1.7189676761627197,
      "step": 89940
    },
    {
      "dynamics_loss_avg": 0.05551249384880066,
      "epoch": 1.5201568603515625e-05,
      "step": 89940
    },
    {
      "action_loss_avg": 0.0014370777294971048,
      "epoch": 1.5201568603515625e-05,
      "step": 89940
    },
    {
      "epoch": 1.5201568603515625e-05,
      "step": 89940,
      "training_step_time": 6.748429536819458
    },
    {
      "epoch": 1.5202522277832032e-05,
      "model_forward_time": 1.7212212085723877,
      "step": 89941
    },
    {
      "epoch": 1.5202522277832032e-05,
      "step": 89941,
      "training_step_time": 6.743438959121704
    },
    {
      "epoch": 1.5203475952148438e-05,
      "model_forward_time": 2.3981826305389404,
      "step": 89942
    },
    {
      "epoch": 1.5203475952148438e-05,
      "step": 89942,
      "training_step_time": 9.802635669708252
    },
    {
      "epoch": 1.5204429626464843e-05,
      "model_forward_time": 1.7200713157653809,
      "step": 89943
    },
    {
      "epoch": 1.5204429626464843e-05,
      "step": 89943,
      "training_step_time": 10.72401213645935
    },
    {
      "epoch": 1.520538330078125e-05,
      "model_forward_time": 1.7277557849884033,
      "step": 89944
    },
    {
      "epoch": 1.520538330078125e-05,
      "step": 89944,
      "training_step_time": 7.613734483718872
    },
    {
      "epoch": 1.5206336975097657e-05,
      "model_forward_time": 1.7219338417053223,
      "step": 89945
    },
    {
      "epoch": 1.5206336975097657e-05,
      "step": 89945,
      "training_step_time": 7.6529834270477295
    },
    {
      "epoch": 1.5207290649414062e-05,
      "model_forward_time": 1.716902494430542,
      "step": 89946
    },
    {
      "epoch": 1.5207290649414062e-05,
      "step": 89946,
      "training_step_time": 7.250869989395142
    },
    {
      "epoch": 1.5208244323730468e-05,
      "model_forward_time": 1.72337007522583,
      "step": 89947
    },
    {
      "epoch": 1.5208244323730468e-05,
      "step": 89947,
      "training_step_time": 6.7427144050598145
    },
    {
      "epoch": 1.5209197998046875e-05,
      "model_forward_time": 1.7269549369812012,
      "step": 89948
    },
    {
      "epoch": 1.5209197998046875e-05,
      "step": 89948,
      "training_step_time": 6.7517547607421875
    },
    {
      "epoch": 1.5210151672363281e-05,
      "model_forward_time": 1.7284162044525146,
      "step": 89949
    },
    {
      "epoch": 1.5210151672363281e-05,
      "step": 89949,
      "training_step_time": 6.747260808944702
    },
    {
      "epoch": 1.5211105346679687e-05,
      "grad_norm": 0.06763433665037155,
      "learning_rate": 2.7365860787587405e-07,
      "loss": 0.0549,
      "step": 89950
    },
    {
      "epoch": 1.5211105346679687e-05,
      "model_forward_time": 1.7272427082061768,
      "step": 89950
    },
    {
      "dynamics_loss_avg": 0.0581714142113924,
      "epoch": 1.5211105346679687e-05,
      "step": 89950
    },
    {
      "action_loss_avg": 0.0012890652986243367,
      "epoch": 1.5211105346679687e-05,
      "step": 89950
    },
    {
      "epoch": 1.5211105346679687e-05,
      "step": 89950,
      "training_step_time": 6.747906923294067
    },
    {
      "epoch": 1.5212059020996094e-05,
      "model_forward_time": 2.418154001235962,
      "step": 89951
    },
    {
      "epoch": 1.5212059020996094e-05,
      "step": 89951,
      "training_step_time": 8.431253671646118
    },
    {
      "epoch": 1.52130126953125e-05,
      "model_forward_time": 1.7454094886779785,
      "step": 89952
    },
    {
      "epoch": 1.52130126953125e-05,
      "step": 89952,
      "training_step_time": 10.89038896560669
    },
    {
      "epoch": 1.5213966369628906e-05,
      "model_forward_time": 1.7252674102783203,
      "step": 89953
    },
    {
      "epoch": 1.5213966369628906e-05,
      "step": 89953,
      "training_step_time": 7.617652893066406
    },
    {
      "epoch": 1.5214920043945313e-05,
      "model_forward_time": 1.7214324474334717,
      "step": 89954
    },
    {
      "epoch": 1.5214920043945313e-05,
      "step": 89954,
      "training_step_time": 8.145384550094604
    },
    {
      "epoch": 1.5215873718261719e-05,
      "model_forward_time": 1.7292652130126953,
      "step": 89955
    },
    {
      "epoch": 1.5215873718261719e-05,
      "step": 89955,
      "training_step_time": 7.518004894256592
    },
    {
      "epoch": 1.5216827392578125e-05,
      "model_forward_time": 1.7237086296081543,
      "step": 89956
    },
    {
      "epoch": 1.5216827392578125e-05,
      "step": 89956,
      "training_step_time": 6.6986403465271
    },
    {
      "epoch": 1.5217781066894532e-05,
      "model_forward_time": 1.7264621257781982,
      "step": 89957
    },
    {
      "epoch": 1.5217781066894532e-05,
      "step": 89957,
      "training_step_time": 6.7129130363464355
    },
    {
      "epoch": 1.5218734741210938e-05,
      "model_forward_time": 1.720404863357544,
      "step": 89958
    },
    {
      "epoch": 1.5218734741210938e-05,
      "step": 89958,
      "training_step_time": 6.742300987243652
    },
    {
      "epoch": 1.5219688415527343e-05,
      "model_forward_time": 1.722320795059204,
      "step": 89959
    },
    {
      "epoch": 1.5219688415527343e-05,
      "step": 89959,
      "training_step_time": 6.704897880554199
    },
    {
      "epoch": 1.522064208984375e-05,
      "grad_norm": 0.08619727939367294,
      "learning_rate": 2.7311934876577296e-07,
      "loss": 0.0525,
      "step": 89960
    },
    {
      "epoch": 1.522064208984375e-05,
      "model_forward_time": 2.3464627265930176,
      "step": 89960
    },
    {
      "dynamics_loss_avg": 0.047371876798570155,
      "epoch": 1.522064208984375e-05,
      "step": 89960
    },
    {
      "action_loss_avg": 0.0014120910433121025,
      "epoch": 1.522064208984375e-05,
      "step": 89960
    },
    {
      "epoch": 1.522064208984375e-05,
      "step": 89960,
      "training_step_time": 8.627374649047852
    },
    {
      "epoch": 1.5221595764160157e-05,
      "model_forward_time": 1.7999367713928223,
      "step": 89961
    },
    {
      "epoch": 1.5221595764160157e-05,
      "step": 89961,
      "training_step_time": 8.480096340179443
    },
    {
      "epoch": 1.5222549438476562e-05,
      "model_forward_time": 1.7180328369140625,
      "step": 89962
    },
    {
      "epoch": 1.5222549438476562e-05,
      "step": 89962,
      "training_step_time": 8.476152896881104
    },
    {
      "epoch": 1.5223503112792968e-05,
      "model_forward_time": 1.7160773277282715,
      "step": 89963
    },
    {
      "epoch": 1.5223503112792968e-05,
      "step": 89963,
      "training_step_time": 7.671311855316162
    },
    {
      "epoch": 1.5224456787109375e-05,
      "model_forward_time": 1.7182669639587402,
      "step": 89964
    },
    {
      "epoch": 1.5224456787109375e-05,
      "step": 89964,
      "training_step_time": 7.612752676010132
    },
    {
      "epoch": 1.5225410461425781e-05,
      "model_forward_time": 1.7149591445922852,
      "step": 89965
    },
    {
      "epoch": 1.5225410461425781e-05,
      "step": 89965,
      "training_step_time": 6.750983715057373
    },
    {
      "epoch": 1.5226364135742187e-05,
      "model_forward_time": 1.7127470970153809,
      "step": 89966
    },
    {
      "epoch": 1.5226364135742187e-05,
      "step": 89966,
      "training_step_time": 6.699381113052368
    },
    {
      "epoch": 1.5227317810058594e-05,
      "model_forward_time": 1.7204790115356445,
      "step": 89967
    },
    {
      "epoch": 1.5227317810058594e-05,
      "step": 89967,
      "training_step_time": 6.699103832244873
    },
    {
      "epoch": 1.5228271484375e-05,
      "model_forward_time": 1.7177984714508057,
      "step": 89968
    },
    {
      "epoch": 1.5228271484375e-05,
      "step": 89968,
      "training_step_time": 6.71322226524353
    },
    {
      "epoch": 1.5229225158691406e-05,
      "model_forward_time": 1.7251083850860596,
      "step": 89969
    },
    {
      "epoch": 1.5229225158691406e-05,
      "step": 89969,
      "training_step_time": 10.907222509384155
    },
    {
      "epoch": 1.5230178833007813e-05,
      "grad_norm": 0.10874594748020172,
      "learning_rate": 2.7258060658023387e-07,
      "loss": 0.0535,
      "step": 89970
    },
    {
      "epoch": 1.5230178833007813e-05,
      "model_forward_time": 1.7830779552459717,
      "step": 89970
    },
    {
      "dynamics_loss_avg": 0.04819328524172306,
      "epoch": 1.5230178833007813e-05,
      "step": 89970
    },
    {
      "action_loss_avg": 0.001116127529530786,
      "epoch": 1.5230178833007813e-05,
      "step": 89970
    },
    {
      "epoch": 1.5230178833007813e-05,
      "step": 89970,
      "training_step_time": 10.631331205368042
    },
    {
      "epoch": 1.5231132507324219e-05,
      "model_forward_time": 1.7207834720611572,
      "step": 89971
    },
    {
      "epoch": 1.5231132507324219e-05,
      "step": 89971,
      "training_step_time": 7.734921932220459
    },
    {
      "epoch": 1.5232086181640625e-05,
      "model_forward_time": 1.711158275604248,
      "step": 89972
    },
    {
      "epoch": 1.5232086181640625e-05,
      "step": 89972,
      "training_step_time": 7.56333589553833
    },
    {
      "epoch": 1.5233039855957032e-05,
      "model_forward_time": 1.7118322849273682,
      "step": 89973
    },
    {
      "epoch": 1.5233039855957032e-05,
      "step": 89973,
      "training_step_time": 7.553524494171143
    },
    {
      "epoch": 1.5233993530273438e-05,
      "model_forward_time": 1.7224791049957275,
      "step": 89974
    },
    {
      "epoch": 1.5233993530273438e-05,
      "step": 89974,
      "training_step_time": 7.207728385925293
    },
    {
      "epoch": 1.5234947204589843e-05,
      "model_forward_time": 1.7156100273132324,
      "step": 89975
    },
    {
      "epoch": 1.5234947204589843e-05,
      "step": 89975,
      "training_step_time": 6.711628437042236
    },
    {
      "epoch": 1.523590087890625e-05,
      "model_forward_time": 1.7128398418426514,
      "step": 89976
    },
    {
      "epoch": 1.523590087890625e-05,
      "step": 89976,
      "training_step_time": 6.703688383102417
    },
    {
      "epoch": 1.5236854553222657e-05,
      "model_forward_time": 1.716973066329956,
      "step": 89977
    },
    {
      "epoch": 1.5236854553222657e-05,
      "step": 89977,
      "training_step_time": 6.722493886947632
    },
    {
      "epoch": 1.5237808227539062e-05,
      "model_forward_time": 1.7132244110107422,
      "step": 89978
    },
    {
      "epoch": 1.5237808227539062e-05,
      "step": 89978,
      "training_step_time": 8.54967737197876
    },
    {
      "epoch": 1.5238761901855468e-05,
      "model_forward_time": 2.4577813148498535,
      "step": 89979
    },
    {
      "epoch": 1.5238761901855468e-05,
      "step": 89979,
      "training_step_time": 9.162145376205444
    },
    {
      "epoch": 1.5239715576171875e-05,
      "grad_norm": 0.17045482993125916,
      "learning_rate": 2.720423813781714e-07,
      "loss": 0.0529,
      "step": 89980
    },
    {
      "epoch": 1.5239715576171875e-05,
      "model_forward_time": 1.7121126651763916,
      "step": 89980
    },
    {
      "dynamics_loss_avg": 0.05303252497687936,
      "epoch": 1.5239715576171875e-05,
      "step": 89980
    },
    {
      "action_loss_avg": 0.00103362409863621,
      "epoch": 1.5239715576171875e-05,
      "step": 89980
    },
    {
      "epoch": 1.5239715576171875e-05,
      "step": 89980,
      "training_step_time": 10.168388366699219
    },
    {
      "epoch": 1.5240669250488281e-05,
      "model_forward_time": 1.7150499820709229,
      "step": 89981
    },
    {
      "epoch": 1.5240669250488281e-05,
      "step": 89981,
      "training_step_time": 7.461940050125122
    },
    {
      "epoch": 1.5241622924804687e-05,
      "model_forward_time": 1.7135534286499023,
      "step": 89982
    },
    {
      "epoch": 1.5241622924804687e-05,
      "step": 89982,
      "training_step_time": 7.63540244102478
    },
    {
      "epoch": 1.5242576599121094e-05,
      "model_forward_time": 1.7216525077819824,
      "step": 89983
    },
    {
      "epoch": 1.5242576599121094e-05,
      "step": 89983,
      "training_step_time": 7.695955276489258
    },
    {
      "epoch": 1.52435302734375e-05,
      "model_forward_time": 1.7156708240509033,
      "step": 89984
    },
    {
      "epoch": 1.52435302734375e-05,
      "step": 89984,
      "training_step_time": 6.730226516723633
    },
    {
      "epoch": 1.5244483947753906e-05,
      "model_forward_time": 1.7150301933288574,
      "step": 89985
    },
    {
      "epoch": 1.5244483947753906e-05,
      "step": 89985,
      "training_step_time": 6.768089771270752
    },
    {
      "epoch": 1.5245437622070313e-05,
      "model_forward_time": 1.7160367965698242,
      "step": 89986
    },
    {
      "epoch": 1.5245437622070313e-05,
      "step": 89986,
      "training_step_time": 6.785440444946289
    },
    {
      "epoch": 1.5246391296386719e-05,
      "model_forward_time": 1.7140848636627197,
      "step": 89987
    },
    {
      "epoch": 1.5246391296386719e-05,
      "step": 89987,
      "training_step_time": 7.6620869636535645
    },
    {
      "epoch": 1.5247344970703125e-05,
      "model_forward_time": 2.403517007827759,
      "step": 89988
    },
    {
      "epoch": 1.5247344970703125e-05,
      "step": 89988,
      "training_step_time": 10.42146921157837
    },
    {
      "epoch": 1.5248298645019532e-05,
      "model_forward_time": 1.7142269611358643,
      "step": 89989
    },
    {
      "epoch": 1.5248298645019532e-05,
      "step": 89989,
      "training_step_time": 9.814374446868896
    },
    {
      "epoch": 1.5249252319335938e-05,
      "grad_norm": 0.22659799456596375,
      "learning_rate": 2.715046732184462e-07,
      "loss": 0.0536,
      "step": 89990
    },
    {
      "epoch": 1.5249252319335938e-05,
      "model_forward_time": 1.7113656997680664,
      "step": 89990
    },
    {
      "dynamics_loss_avg": 0.04712126236408949,
      "epoch": 1.5249252319335938e-05,
      "step": 89990
    },
    {
      "action_loss_avg": 0.0012400456878822296,
      "epoch": 1.5249252319335938e-05,
      "step": 89990
    },
    {
      "epoch": 1.5249252319335938e-05,
      "step": 89990,
      "training_step_time": 7.603761196136475
    },
    {
      "epoch": 1.5250205993652343e-05,
      "model_forward_time": 1.7139713764190674,
      "step": 89991
    },
    {
      "epoch": 1.5250205993652343e-05,
      "step": 89991,
      "training_step_time": 7.65869665145874
    },
    {
      "epoch": 1.525115966796875e-05,
      "model_forward_time": 1.714686632156372,
      "step": 89992
    },
    {
      "epoch": 1.525115966796875e-05,
      "step": 89992,
      "training_step_time": 7.664093971252441
    },
    {
      "epoch": 1.5252113342285157e-05,
      "model_forward_time": 1.7228574752807617,
      "step": 89993
    },
    {
      "epoch": 1.5252113342285157e-05,
      "step": 89993,
      "training_step_time": 6.757734537124634
    },
    {
      "epoch": 1.5253067016601562e-05,
      "model_forward_time": 1.7078886032104492,
      "step": 89994
    },
    {
      "epoch": 1.5253067016601562e-05,
      "step": 89994,
      "training_step_time": 6.647217512130737
    },
    {
      "epoch": 1.5254020690917968e-05,
      "model_forward_time": 1.717149257659912,
      "step": 89995
    },
    {
      "epoch": 1.5254020690917968e-05,
      "step": 89995,
      "training_step_time": 6.71227240562439
    },
    {
      "epoch": 1.5254974365234375e-05,
      "model_forward_time": 1.714024305343628,
      "step": 89996
    },
    {
      "epoch": 1.5254974365234375e-05,
      "step": 89996,
      "training_step_time": 7.63964056968689
    },
    {
      "epoch": 1.5255928039550781e-05,
      "model_forward_time": 2.383415937423706,
      "step": 89997
    },
    {
      "epoch": 1.5255928039550781e-05,
      "step": 89997,
      "training_step_time": 9.20321011543274
    },
    {
      "epoch": 1.5256881713867187e-05,
      "model_forward_time": 1.7872369289398193,
      "step": 89998
    },
    {
      "epoch": 1.5256881713867187e-05,
      "step": 89998,
      "training_step_time": 8.992263555526733
    },
    {
      "epoch": 1.5257835388183594e-05,
      "model_forward_time": 1.7133557796478271,
      "step": 89999
    },
    {
      "epoch": 1.5257835388183594e-05,
      "step": 89999,
      "training_step_time": 7.83573579788208
    },
    {
      "epoch": 1.52587890625e-05,
      "grad_norm": 0.11359456181526184,
      "learning_rate": 2.709674821598601e-07,
      "loss": 0.054,
      "step": 90000
    },
    {
      "epoch": 1.52587890625e-05,
      "model_forward_time": 1.76798415184021,
      "step": 90000
    },
    {
      "dynamics_loss_avg": 0.06254962179809809,
      "epoch": 1.52587890625e-05,
      "step": 90000
    },
    {
      "action_loss_avg": 0.0016014172055292875,
      "epoch": 1.52587890625e-05,
      "step": 90000
    },
    {
      "epoch": 1.52587890625e-05,
      "step": 90000,
      "training_step_time": 7.1931633949279785
    },
    {
      "epoch": 1.5259742736816407e-05,
      "model_forward_time": 1.7385783195495605,
      "step": 90001
    },
    {
      "epoch": 1.5259742736816407e-05,
      "step": 90001,
      "training_step_time": 7.8207848072052
    },
    {
      "epoch": 1.526069641113281e-05,
      "model_forward_time": 1.7327938079833984,
      "step": 90002
    },
    {
      "epoch": 1.526069641113281e-05,
      "step": 90002,
      "training_step_time": 6.900796890258789
    },
    {
      "epoch": 1.526165008544922e-05,
      "model_forward_time": 1.734515905380249,
      "step": 90003
    },
    {
      "epoch": 1.526165008544922e-05,
      "step": 90003,
      "training_step_time": 6.852376937866211
    },
    {
      "epoch": 1.5262603759765626e-05,
      "model_forward_time": 1.7361171245574951,
      "step": 90004
    },
    {
      "epoch": 1.5262603759765626e-05,
      "step": 90004,
      "training_step_time": 7.849644422531128
    },
    {
      "epoch": 1.526355743408203e-05,
      "model_forward_time": 1.740004062652588,
      "step": 90005
    },
    {
      "epoch": 1.526355743408203e-05,
      "step": 90005,
      "training_step_time": 9.918732166290283
    },
    {
      "epoch": 1.5264511108398438e-05,
      "model_forward_time": 1.7303900718688965,
      "step": 90006
    },
    {
      "epoch": 1.5264511108398438e-05,
      "step": 90006,
      "training_step_time": 10.362212181091309
    },
    {
      "epoch": 1.5265464782714845e-05,
      "model_forward_time": 1.8535876274108887,
      "step": 90007
    },
    {
      "epoch": 1.5265464782714845e-05,
      "step": 90007,
      "training_step_time": 11.063026428222656
    },
    {
      "epoch": 1.526641845703125e-05,
      "model_forward_time": 1.7271060943603516,
      "step": 90008
    },
    {
      "epoch": 1.526641845703125e-05,
      "step": 90008,
      "training_step_time": 8.44810938835144
    },
    {
      "epoch": 1.5267372131347657e-05,
      "model_forward_time": 1.732041835784912,
      "step": 90009
    },
    {
      "epoch": 1.5267372131347657e-05,
      "step": 90009,
      "training_step_time": 7.730334997177124
    },
    {
      "epoch": 1.5268325805664064e-05,
      "grad_norm": 0.09437859803438187,
      "learning_rate": 2.704308082611612e-07,
      "loss": 0.0554,
      "step": 90010
    },
    {
      "epoch": 1.5268325805664064e-05,
      "model_forward_time": 1.345527172088623,
      "step": 90010
    },
    {
      "dynamics_loss_avg": 0.04888607133179903,
      "epoch": 1.5268325805664064e-05,
      "step": 90010
    },
    {
      "action_loss_avg": 0.0014655124366981908,
      "epoch": 1.5268325805664064e-05,
      "step": 90010
    },
    {
      "epoch": 1.5268325805664064e-05,
      "step": 90010,
      "training_step_time": 7.75865912437439
    },
    {
      "epoch": 1.5269279479980468e-05,
      "model_forward_time": 1.7275736331939697,
      "step": 90011
    },
    {
      "epoch": 1.5269279479980468e-05,
      "step": 90011,
      "training_step_time": 7.384608745574951
    },
    {
      "epoch": 1.5270233154296875e-05,
      "model_forward_time": 1.7313873767852783,
      "step": 90012
    },
    {
      "epoch": 1.5270233154296875e-05,
      "step": 90012,
      "training_step_time": 6.866546630859375
    },
    {
      "epoch": 1.5271186828613283e-05,
      "model_forward_time": 1.7342267036437988,
      "step": 90013
    },
    {
      "epoch": 1.5271186828613283e-05,
      "step": 90013,
      "training_step_time": 6.7944300174713135
    },
    {
      "epoch": 1.5272140502929687e-05,
      "model_forward_time": 1.7266051769256592,
      "step": 90014
    },
    {
      "epoch": 1.5272140502929687e-05,
      "step": 90014,
      "training_step_time": 7.747941970825195
    },
    {
      "epoch": 1.5273094177246094e-05,
      "model_forward_time": 2.3751537799835205,
      "step": 90015
    },
    {
      "epoch": 1.5273094177246094e-05,
      "step": 90015,
      "training_step_time": 9.618685722351074
    },
    {
      "epoch": 1.52740478515625e-05,
      "model_forward_time": 1.8384182453155518,
      "step": 90016
    },
    {
      "epoch": 1.52740478515625e-05,
      "step": 90016,
      "training_step_time": 9.945910215377808
    },
    {
      "epoch": 1.5275001525878906e-05,
      "model_forward_time": 1.728987216949463,
      "step": 90017
    },
    {
      "epoch": 1.5275001525878906e-05,
      "step": 90017,
      "training_step_time": 9.660264730453491
    },
    {
      "epoch": 1.5275955200195313e-05,
      "model_forward_time": 1.7277989387512207,
      "step": 90018
    },
    {
      "epoch": 1.5275955200195313e-05,
      "step": 90018,
      "training_step_time": 7.637300491333008
    },
    {
      "epoch": 1.5276908874511717e-05,
      "model_forward_time": 1.7408854961395264,
      "step": 90019
    },
    {
      "epoch": 1.5276908874511717e-05,
      "step": 90019,
      "training_step_time": 7.783584356307983
    },
    {
      "epoch": 1.5277862548828125e-05,
      "grad_norm": 0.09086547046899796,
      "learning_rate": 2.6989465158103747e-07,
      "loss": 0.055,
      "step": 90020
    },
    {
      "epoch": 1.5277862548828125e-05,
      "model_forward_time": 1.7228758335113525,
      "step": 90020
    },
    {
      "dynamics_loss_avg": 0.06863239407539368,
      "epoch": 1.5277862548828125e-05,
      "step": 90020
    },
    {
      "action_loss_avg": 0.0023482184624299407,
      "epoch": 1.5277862548828125e-05,
      "step": 90020
    },
    {
      "epoch": 1.5277862548828125e-05,
      "step": 90020,
      "training_step_time": 7.215090036392212
    },
    {
      "epoch": 1.5278816223144532e-05,
      "model_forward_time": 1.7292771339416504,
      "step": 90021
    },
    {
      "epoch": 1.5278816223144532e-05,
      "step": 90021,
      "training_step_time": 7.269606351852417
    },
    {
      "epoch": 1.5279769897460936e-05,
      "model_forward_time": 1.7399063110351562,
      "step": 90022
    },
    {
      "epoch": 1.5279769897460936e-05,
      "step": 90022,
      "training_step_time": 6.751145839691162
    },
    {
      "epoch": 1.5280723571777343e-05,
      "model_forward_time": 1.736924409866333,
      "step": 90023
    },
    {
      "epoch": 1.5280723571777343e-05,
      "step": 90023,
      "training_step_time": 7.676227331161499
    },
    {
      "epoch": 1.528167724609375e-05,
      "model_forward_time": 2.3638103008270264,
      "step": 90024
    },
    {
      "epoch": 1.528167724609375e-05,
      "step": 90024,
      "training_step_time": 7.6764304637908936
    },
    {
      "epoch": 1.5282630920410155e-05,
      "model_forward_time": 1.7931323051452637,
      "step": 90025
    },
    {
      "epoch": 1.5282630920410155e-05,
      "step": 90025,
      "training_step_time": 9.479028463363647
    },
    {
      "epoch": 1.5283584594726562e-05,
      "model_forward_time": 1.7095391750335693,
      "step": 90026
    },
    {
      "epoch": 1.5283584594726562e-05,
      "step": 90026,
      "training_step_time": 8.794832229614258
    },
    {
      "epoch": 1.528453826904297e-05,
      "model_forward_time": 1.7179734706878662,
      "step": 90027
    },
    {
      "epoch": 1.528453826904297e-05,
      "step": 90027,
      "training_step_time": 7.597187757492065
    },
    {
      "epoch": 1.5285491943359374e-05,
      "model_forward_time": 1.7197182178497314,
      "step": 90028
    },
    {
      "epoch": 1.5285491943359374e-05,
      "step": 90028,
      "training_step_time": 7.774367809295654
    },
    {
      "epoch": 1.528644561767578e-05,
      "model_forward_time": 1.7217354774475098,
      "step": 90029
    },
    {
      "epoch": 1.528644561767578e-05,
      "step": 90029,
      "training_step_time": 7.177465200424194
    },
    {
      "epoch": 1.528739929199219e-05,
      "grad_norm": 0.106731116771698,
      "learning_rate": 2.6935901217812366e-07,
      "loss": 0.0552,
      "step": 90030
    },
    {
      "epoch": 1.528739929199219e-05,
      "model_forward_time": 1.7139811515808105,
      "step": 90030
    },
    {
      "dynamics_loss_avg": 0.040824894607067105,
      "epoch": 1.528739929199219e-05,
      "step": 90030
    },
    {
      "action_loss_avg": 0.0013904708044719881,
      "epoch": 1.528739929199219e-05,
      "step": 90030
    },
    {
      "epoch": 1.528739929199219e-05,
      "step": 90030,
      "training_step_time": 7.295938491821289
    },
    {
      "epoch": 1.5288352966308593e-05,
      "model_forward_time": 1.7223830223083496,
      "step": 90031
    },
    {
      "epoch": 1.5288352966308593e-05,
      "step": 90031,
      "training_step_time": 6.709864854812622
    },
    {
      "epoch": 1.5289306640625e-05,
      "model_forward_time": 1.7205102443695068,
      "step": 90032
    },
    {
      "epoch": 1.5289306640625e-05,
      "step": 90032,
      "training_step_time": 7.673475503921509
    },
    {
      "epoch": 1.5290260314941407e-05,
      "model_forward_time": 2.384397506713867,
      "step": 90033
    },
    {
      "epoch": 1.5290260314941407e-05,
      "step": 90033,
      "training_step_time": 8.10487675666809
    },
    {
      "epoch": 1.529121398925781e-05,
      "model_forward_time": 1.7554199695587158,
      "step": 90034
    },
    {
      "epoch": 1.529121398925781e-05,
      "step": 90034,
      "training_step_time": 8.417199611663818
    },
    {
      "epoch": 1.529216766357422e-05,
      "model_forward_time": 1.7205910682678223,
      "step": 90035
    },
    {
      "epoch": 1.529216766357422e-05,
      "step": 90035,
      "training_step_time": 8.698253393173218
    },
    {
      "epoch": 1.5293121337890626e-05,
      "model_forward_time": 1.7267029285430908,
      "step": 90036
    },
    {
      "epoch": 1.5293121337890626e-05,
      "step": 90036,
      "training_step_time": 9.240295171737671
    },
    {
      "epoch": 1.529407501220703e-05,
      "model_forward_time": 1.7223865985870361,
      "step": 90037
    },
    {
      "epoch": 1.529407501220703e-05,
      "step": 90037,
      "training_step_time": 8.66089677810669
    },
    {
      "epoch": 1.5295028686523438e-05,
      "model_forward_time": 1.7203669548034668,
      "step": 90038
    },
    {
      "epoch": 1.5295028686523438e-05,
      "step": 90038,
      "training_step_time": 7.623050689697266
    },
    {
      "epoch": 1.5295982360839845e-05,
      "model_forward_time": 1.72705078125,
      "step": 90039
    },
    {
      "epoch": 1.5295982360839845e-05,
      "step": 90039,
      "training_step_time": 7.281232833862305
    },
    {
      "epoch": 1.529693603515625e-05,
      "grad_norm": 0.09436559677124023,
      "learning_rate": 2.6882389011099517e-07,
      "loss": 0.0563,
      "step": 90040
    },
    {
      "epoch": 1.529693603515625e-05,
      "model_forward_time": 1.7148444652557373,
      "step": 90040
    },
    {
      "dynamics_loss_avg": 0.054937990568578246,
      "epoch": 1.529693603515625e-05,
      "step": 90040
    },
    {
      "action_loss_avg": 0.0024265261075925083,
      "epoch": 1.529693603515625e-05,
      "step": 90040
    },
    {
      "epoch": 1.529693603515625e-05,
      "step": 90040,
      "training_step_time": 6.768433094024658
    },
    {
      "epoch": 1.5297889709472657e-05,
      "model_forward_time": 1.7155957221984863,
      "step": 90041
    },
    {
      "epoch": 1.5297889709472657e-05,
      "step": 90041,
      "training_step_time": 7.702794790267944
    },
    {
      "epoch": 1.5298843383789064e-05,
      "model_forward_time": 2.356374740600586,
      "step": 90042
    },
    {
      "epoch": 1.5298843383789064e-05,
      "step": 90042,
      "training_step_time": 8.261974811553955
    },
    {
      "epoch": 1.5299797058105468e-05,
      "model_forward_time": 1.7261443138122559,
      "step": 90043
    },
    {
      "epoch": 1.5299797058105468e-05,
      "step": 90043,
      "training_step_time": 10.100852966308594
    },
    {
      "epoch": 1.5300750732421875e-05,
      "model_forward_time": 1.7265725135803223,
      "step": 90044
    },
    {
      "epoch": 1.5300750732421875e-05,
      "step": 90044,
      "training_step_time": 7.760051250457764
    },
    {
      "epoch": 1.5301704406738283e-05,
      "model_forward_time": 1.7213106155395508,
      "step": 90045
    },
    {
      "epoch": 1.5301704406738283e-05,
      "step": 90045,
      "training_step_time": 8.353336334228516
    },
    {
      "epoch": 1.5302658081054687e-05,
      "model_forward_time": 1.7214863300323486,
      "step": 90046
    },
    {
      "epoch": 1.5302658081054687e-05,
      "step": 90046,
      "training_step_time": 8.2705557346344
    },
    {
      "epoch": 1.5303611755371094e-05,
      "model_forward_time": 1.7262184619903564,
      "step": 90047
    },
    {
      "epoch": 1.5303611755371094e-05,
      "step": 90047,
      "training_step_time": 7.723318576812744
    },
    {
      "epoch": 1.53045654296875e-05,
      "model_forward_time": 1.7210023403167725,
      "step": 90048
    },
    {
      "epoch": 1.53045654296875e-05,
      "step": 90048,
      "training_step_time": 7.244704484939575
    },
    {
      "epoch": 1.5305519104003906e-05,
      "model_forward_time": 1.7200891971588135,
      "step": 90049
    },
    {
      "epoch": 1.5305519104003906e-05,
      "step": 90049,
      "training_step_time": 6.872848033905029
    },
    {
      "epoch": 1.5306472778320313e-05,
      "grad_norm": 0.1271255910396576,
      "learning_rate": 2.6828928543817303e-07,
      "loss": 0.0561,
      "step": 90050
    },
    {
      "epoch": 1.5306472778320313e-05,
      "model_forward_time": 1.7173199653625488,
      "step": 90050
    },
    {
      "dynamics_loss_avg": 0.048467309586703776,
      "epoch": 1.5306472778320313e-05,
      "step": 90050
    },
    {
      "action_loss_avg": 0.0011233597761020064,
      "epoch": 1.5306472778320313e-05,
      "step": 90050
    },
    {
      "epoch": 1.5306472778320313e-05,
      "step": 90050,
      "training_step_time": 8.192126750946045
    },
    {
      "epoch": 1.5307426452636717e-05,
      "model_forward_time": 1.7175498008728027,
      "step": 90051
    },
    {
      "epoch": 1.5307426452636717e-05,
      "step": 90051,
      "training_step_time": 9.844779253005981
    },
    {
      "epoch": 1.5308380126953125e-05,
      "model_forward_time": 1.7102746963500977,
      "step": 90052
    },
    {
      "epoch": 1.5308380126953125e-05,
      "step": 90052,
      "training_step_time": 9.462332248687744
    },
    {
      "epoch": 1.5309333801269532e-05,
      "model_forward_time": 1.7435643672943115,
      "step": 90053
    },
    {
      "epoch": 1.5309333801269532e-05,
      "step": 90053,
      "training_step_time": 7.86142897605896
    },
    {
      "epoch": 1.5310287475585936e-05,
      "model_forward_time": 1.7412433624267578,
      "step": 90054
    },
    {
      "epoch": 1.5310287475585936e-05,
      "step": 90054,
      "training_step_time": 7.719426393508911
    },
    {
      "epoch": 1.5311241149902343e-05,
      "model_forward_time": 1.7298271656036377,
      "step": 90055
    },
    {
      "epoch": 1.5311241149902343e-05,
      "step": 90055,
      "training_step_time": 7.616034984588623
    },
    {
      "epoch": 1.531219482421875e-05,
      "model_forward_time": 1.7177760601043701,
      "step": 90056
    },
    {
      "epoch": 1.531219482421875e-05,
      "step": 90056,
      "training_step_time": 7.7555694580078125
    },
    {
      "epoch": 1.5313148498535155e-05,
      "model_forward_time": 1.724644660949707,
      "step": 90057
    },
    {
      "epoch": 1.5313148498535155e-05,
      "step": 90057,
      "training_step_time": 6.753688335418701
    },
    {
      "epoch": 1.5314102172851562e-05,
      "model_forward_time": 1.723419427871704,
      "step": 90058
    },
    {
      "epoch": 1.5314102172851562e-05,
      "step": 90058,
      "training_step_time": 7.271527290344238
    },
    {
      "epoch": 1.531505584716797e-05,
      "model_forward_time": 1.7202045917510986,
      "step": 90059
    },
    {
      "epoch": 1.531505584716797e-05,
      "step": 90059,
      "training_step_time": 6.785120248794556
    },
    {
      "epoch": 1.5316009521484374e-05,
      "grad_norm": 0.08894151449203491,
      "learning_rate": 2.6775519821812036e-07,
      "loss": 0.0568,
      "step": 90060
    },
    {
      "epoch": 1.5316009521484374e-05,
      "model_forward_time": 1.703031301498413,
      "step": 90060
    },
    {
      "dynamics_loss_avg": 0.04365634061396122,
      "epoch": 1.5316009521484374e-05,
      "step": 90060
    },
    {
      "action_loss_avg": 0.0009516972524579615,
      "epoch": 1.5316009521484374e-05,
      "step": 90060
    },
    {
      "epoch": 1.5316009521484374e-05,
      "step": 90060,
      "training_step_time": 8.458427906036377
    },
    {
      "epoch": 1.531696319580078e-05,
      "model_forward_time": 2.3513054847717285,
      "step": 90061
    },
    {
      "epoch": 1.531696319580078e-05,
      "step": 90061,
      "training_step_time": 9.220332145690918
    },
    {
      "epoch": 1.531791687011719e-05,
      "model_forward_time": 1.786691427230835,
      "step": 90062
    },
    {
      "epoch": 1.531791687011719e-05,
      "step": 90062,
      "training_step_time": 9.27989411354065
    },
    {
      "epoch": 1.5318870544433593e-05,
      "model_forward_time": 1.7182660102844238,
      "step": 90063
    },
    {
      "epoch": 1.5318870544433593e-05,
      "step": 90063,
      "training_step_time": 7.776660680770874
    },
    {
      "epoch": 1.531982421875e-05,
      "model_forward_time": 1.3364641666412354,
      "step": 90064
    },
    {
      "epoch": 1.531982421875e-05,
      "step": 90064,
      "training_step_time": 7.570559024810791
    },
    {
      "epoch": 1.5320777893066407e-05,
      "model_forward_time": 1.7220993041992188,
      "step": 90065
    },
    {
      "epoch": 1.5320777893066407e-05,
      "step": 90065,
      "training_step_time": 7.664980888366699
    },
    {
      "epoch": 1.532173156738281e-05,
      "model_forward_time": 1.7189383506774902,
      "step": 90066
    },
    {
      "epoch": 1.532173156738281e-05,
      "step": 90066,
      "training_step_time": 7.200697422027588
    },
    {
      "epoch": 1.532268524169922e-05,
      "model_forward_time": 1.7222545146942139,
      "step": 90067
    },
    {
      "epoch": 1.532268524169922e-05,
      "step": 90067,
      "training_step_time": 7.23987078666687
    },
    {
      "epoch": 1.5323638916015626e-05,
      "model_forward_time": 1.7235934734344482,
      "step": 90068
    },
    {
      "epoch": 1.5323638916015626e-05,
      "step": 90068,
      "training_step_time": 6.718850374221802
    },
    {
      "epoch": 1.532459259033203e-05,
      "model_forward_time": 1.7205803394317627,
      "step": 90069
    },
    {
      "epoch": 1.532459259033203e-05,
      "step": 90069,
      "training_step_time": 7.672647476196289
    },
    {
      "epoch": 1.5325546264648438e-05,
      "grad_norm": 0.0732230618596077,
      "learning_rate": 2.6722162850924495e-07,
      "loss": 0.0556,
      "step": 90070
    },
    {
      "epoch": 1.5325546264648438e-05,
      "model_forward_time": 2.3729965686798096,
      "step": 90070
    },
    {
      "dynamics_loss_avg": 0.05880025019869208,
      "epoch": 1.5325546264648438e-05,
      "step": 90070
    },
    {
      "action_loss_avg": 0.0010548746155109256,
      "epoch": 1.5325546264648438e-05,
      "step": 90070
    },
    {
      "epoch": 1.5325546264648438e-05,
      "step": 90070,
      "training_step_time": 9.508238077163696
    },
    {
      "epoch": 1.5326499938964845e-05,
      "model_forward_time": 1.733476161956787,
      "step": 90071
    },
    {
      "epoch": 1.5326499938964845e-05,
      "step": 90071,
      "training_step_time": 9.344124794006348
    },
    {
      "epoch": 1.532745361328125e-05,
      "model_forward_time": 1.7228953838348389,
      "step": 90072
    },
    {
      "epoch": 1.532745361328125e-05,
      "step": 90072,
      "training_step_time": 9.233514070510864
    },
    {
      "epoch": 1.5328407287597657e-05,
      "model_forward_time": 1.7268376350402832,
      "step": 90073
    },
    {
      "epoch": 1.5328407287597657e-05,
      "step": 90073,
      "training_step_time": 7.56110405921936
    },
    {
      "epoch": 1.5329360961914064e-05,
      "model_forward_time": 1.3618850708007812,
      "step": 90074
    },
    {
      "epoch": 1.5329360961914064e-05,
      "step": 90074,
      "training_step_time": 7.693552255630493
    },
    {
      "epoch": 1.5330314636230468e-05,
      "model_forward_time": 1.7208223342895508,
      "step": 90075
    },
    {
      "epoch": 1.5330314636230468e-05,
      "step": 90075,
      "training_step_time": 7.434257745742798
    },
    {
      "epoch": 1.5331268310546875e-05,
      "model_forward_time": 1.7199783325195312,
      "step": 90076
    },
    {
      "epoch": 1.5331268310546875e-05,
      "step": 90076,
      "training_step_time": 7.203312397003174
    },
    {
      "epoch": 1.5332221984863283e-05,
      "model_forward_time": 1.721022367477417,
      "step": 90077
    },
    {
      "epoch": 1.5332221984863283e-05,
      "step": 90077,
      "training_step_time": 6.727528810501099
    },
    {
      "epoch": 1.5333175659179687e-05,
      "model_forward_time": 1.725386381149292,
      "step": 90078
    },
    {
      "epoch": 1.5333175659179687e-05,
      "step": 90078,
      "training_step_time": 7.083682298660278
    },
    {
      "epoch": 1.5334129333496094e-05,
      "model_forward_time": 2.8539421558380127,
      "step": 90079
    },
    {
      "epoch": 1.5334129333496094e-05,
      "step": 90079,
      "training_step_time": 7.6368186473846436
    },
    {
      "epoch": 1.53350830078125e-05,
      "grad_norm": 0.11831226944923401,
      "learning_rate": 2.666885763698962e-07,
      "loss": 0.0568,
      "step": 90080
    },
    {
      "epoch": 1.53350830078125e-05,
      "model_forward_time": 1.7159900665283203,
      "step": 90080
    },
    {
      "dynamics_loss_avg": 0.05433731712400913,
      "epoch": 1.53350830078125e-05,
      "step": 90080
    },
    {
      "action_loss_avg": 0.0023478026385419072,
      "epoch": 1.53350830078125e-05,
      "step": 90080
    },
    {
      "epoch": 1.53350830078125e-05,
      "step": 90080,
      "training_step_time": 9.226909875869751
    },
    {
      "epoch": 1.5336036682128906e-05,
      "model_forward_time": 1.3590278625488281,
      "step": 90081
    },
    {
      "epoch": 1.5336036682128906e-05,
      "step": 90081,
      "training_step_time": 8.684340715408325
    },
    {
      "epoch": 1.5336990356445313e-05,
      "model_forward_time": 1.716123104095459,
      "step": 90082
    },
    {
      "epoch": 1.5336990356445313e-05,
      "step": 90082,
      "training_step_time": 7.750574350357056
    },
    {
      "epoch": 1.5337944030761717e-05,
      "model_forward_time": 1.7218017578125,
      "step": 90083
    },
    {
      "epoch": 1.5337944030761717e-05,
      "step": 90083,
      "training_step_time": 9.737240552902222
    },
    {
      "epoch": 1.5338897705078125e-05,
      "model_forward_time": 1.719848394393921,
      "step": 90084
    },
    {
      "epoch": 1.5338897705078125e-05,
      "step": 90084,
      "training_step_time": 7.430667400360107
    },
    {
      "epoch": 1.5339851379394532e-05,
      "model_forward_time": 1.7173285484313965,
      "step": 90085
    },
    {
      "epoch": 1.5339851379394532e-05,
      "step": 90085,
      "training_step_time": 6.792995929718018
    },
    {
      "epoch": 1.5340805053710936e-05,
      "model_forward_time": 1.714968204498291,
      "step": 90086
    },
    {
      "epoch": 1.5340805053710936e-05,
      "step": 90086,
      "training_step_time": 7.240707159042358
    },
    {
      "epoch": 1.5341758728027344e-05,
      "model_forward_time": 1.723592758178711,
      "step": 90087
    },
    {
      "epoch": 1.5341758728027344e-05,
      "step": 90087,
      "training_step_time": 7.15459680557251
    },
    {
      "epoch": 1.534271240234375e-05,
      "model_forward_time": 1.7133982181549072,
      "step": 90088
    },
    {
      "epoch": 1.534271240234375e-05,
      "step": 90088,
      "training_step_time": 7.672590017318726
    },
    {
      "epoch": 1.5343666076660155e-05,
      "model_forward_time": 2.405416965484619,
      "step": 90089
    },
    {
      "epoch": 1.5343666076660155e-05,
      "step": 90089,
      "training_step_time": 10.264495849609375
    },
    {
      "epoch": 1.5344619750976562e-05,
      "grad_norm": 0.09244344383478165,
      "learning_rate": 2.661560418583692e-07,
      "loss": 0.0557,
      "step": 90090
    },
    {
      "epoch": 1.5344619750976562e-05,
      "model_forward_time": 1.743607997894287,
      "step": 90090
    },
    {
      "dynamics_loss_avg": 0.0423766965046525,
      "epoch": 1.5344619750976562e-05,
      "step": 90090
    },
    {
      "action_loss_avg": 0.0011486527044326067,
      "epoch": 1.5344619750976562e-05,
      "step": 90090
    },
    {
      "epoch": 1.5344619750976562e-05,
      "step": 90090,
      "training_step_time": 8.990067720413208
    },
    {
      "epoch": 1.534557342529297e-05,
      "model_forward_time": 1.7244672775268555,
      "step": 90091
    },
    {
      "epoch": 1.534557342529297e-05,
      "step": 90091,
      "training_step_time": 8.76966905593872
    },
    {
      "epoch": 1.5346527099609374e-05,
      "model_forward_time": 1.7279162406921387,
      "step": 90092
    },
    {
      "epoch": 1.5346527099609374e-05,
      "step": 90092,
      "training_step_time": 7.7571165561676025
    },
    {
      "epoch": 1.534748077392578e-05,
      "model_forward_time": 1.7356443405151367,
      "step": 90093
    },
    {
      "epoch": 1.534748077392578e-05,
      "step": 90093,
      "training_step_time": 7.618149042129517
    },
    {
      "epoch": 1.534843444824219e-05,
      "model_forward_time": 1.7390098571777344,
      "step": 90094
    },
    {
      "epoch": 1.534843444824219e-05,
      "step": 90094,
      "training_step_time": 7.404935598373413
    },
    {
      "epoch": 1.5349388122558593e-05,
      "model_forward_time": 1.7394134998321533,
      "step": 90095
    },
    {
      "epoch": 1.5349388122558593e-05,
      "step": 90095,
      "training_step_time": 7.232947587966919
    },
    {
      "epoch": 1.5350341796875e-05,
      "model_forward_time": 1.75117826461792,
      "step": 90096
    },
    {
      "epoch": 1.5350341796875e-05,
      "step": 90096,
      "training_step_time": 7.04809832572937
    },
    {
      "epoch": 1.5351295471191407e-05,
      "model_forward_time": 1.7203500270843506,
      "step": 90097
    },
    {
      "epoch": 1.5351295471191407e-05,
      "step": 90097,
      "training_step_time": 7.461005210876465
    },
    {
      "epoch": 1.535224914550781e-05,
      "model_forward_time": 2.4312477111816406,
      "step": 90098
    },
    {
      "epoch": 1.535224914550781e-05,
      "step": 90098,
      "training_step_time": 9.018676519393921
    },
    {
      "epoch": 1.535320281982422e-05,
      "model_forward_time": 1.7421002388000488,
      "step": 90099
    },
    {
      "epoch": 1.535320281982422e-05,
      "step": 90099,
      "training_step_time": 8.775545120239258
    },
    {
      "epoch": 1.5354156494140626e-05,
      "grad_norm": 0.08815886825323105,
      "learning_rate": 2.656240250328995e-07,
      "loss": 0.056,
      "step": 90100
    },
    {
      "epoch": 1.5354156494140626e-05,
      "model_forward_time": 1.72981858253479,
      "step": 90100
    },
    {
      "dynamics_loss_avg": 0.06856474913656711,
      "epoch": 1.5354156494140626e-05,
      "step": 90100
    },
    {
      "action_loss_avg": 0.0016153921955265106,
      "epoch": 1.5354156494140626e-05,
      "step": 90100
    },
    {
      "epoch": 1.5354156494140626e-05,
      "step": 90100,
      "training_step_time": 8.561522960662842
    },
    {
      "epoch": 1.535511016845703e-05,
      "model_forward_time": 1.727848768234253,
      "step": 90101
    },
    {
      "epoch": 1.535511016845703e-05,
      "step": 90101,
      "training_step_time": 7.6755523681640625
    },
    {
      "epoch": 1.5356063842773438e-05,
      "model_forward_time": 1.7414965629577637,
      "step": 90102
    },
    {
      "epoch": 1.5356063842773438e-05,
      "step": 90102,
      "training_step_time": 7.775213003158569
    },
    {
      "epoch": 1.5357017517089845e-05,
      "model_forward_time": 1.733529806137085,
      "step": 90103
    },
    {
      "epoch": 1.5357017517089845e-05,
      "step": 90103,
      "training_step_time": 7.458130836486816
    },
    {
      "epoch": 1.535797119140625e-05,
      "model_forward_time": 1.726283311843872,
      "step": 90104
    },
    {
      "epoch": 1.535797119140625e-05,
      "step": 90104,
      "training_step_time": 7.281000137329102
    },
    {
      "epoch": 1.5358924865722657e-05,
      "model_forward_time": 1.7381365299224854,
      "step": 90105
    },
    {
      "epoch": 1.5358924865722657e-05,
      "step": 90105,
      "training_step_time": 6.808466196060181
    },
    {
      "epoch": 1.5359878540039064e-05,
      "model_forward_time": 1.7304823398590088,
      "step": 90106
    },
    {
      "epoch": 1.5359878540039064e-05,
      "step": 90106,
      "training_step_time": 7.6177756786346436
    },
    {
      "epoch": 1.5360832214355468e-05,
      "model_forward_time": 2.3919527530670166,
      "step": 90107
    },
    {
      "epoch": 1.5360832214355468e-05,
      "step": 90107,
      "training_step_time": 9.087207317352295
    },
    {
      "epoch": 1.5361785888671876e-05,
      "model_forward_time": 1.7501380443572998,
      "step": 90108
    },
    {
      "epoch": 1.5361785888671876e-05,
      "step": 90108,
      "training_step_time": 9.611348628997803
    },
    {
      "epoch": 1.5362739562988283e-05,
      "model_forward_time": 1.725994348526001,
      "step": 90109
    },
    {
      "epoch": 1.5362739562988283e-05,
      "step": 90109,
      "training_step_time": 8.792528629302979
    },
    {
      "epoch": 1.5363693237304687e-05,
      "grad_norm": 0.0750800371170044,
      "learning_rate": 2.650925259516679e-07,
      "loss": 0.055,
      "step": 90110
    },
    {
      "epoch": 1.5363693237304687e-05,
      "model_forward_time": 1.7336838245391846,
      "step": 90110
    },
    {
      "dynamics_loss_avg": 0.07420700900256634,
      "epoch": 1.5363693237304687e-05,
      "step": 90110
    },
    {
      "action_loss_avg": 0.002036141883581877,
      "epoch": 1.5363693237304687e-05,
      "step": 90110
    },
    {
      "epoch": 1.5363693237304687e-05,
      "step": 90110,
      "training_step_time": 7.684528350830078
    },
    {
      "epoch": 1.5364646911621094e-05,
      "model_forward_time": 1.7298331260681152,
      "step": 90111
    },
    {
      "epoch": 1.5364646911621094e-05,
      "step": 90111,
      "training_step_time": 7.748398065567017
    },
    {
      "epoch": 1.53656005859375e-05,
      "model_forward_time": 1.7284996509552002,
      "step": 90112
    },
    {
      "epoch": 1.53656005859375e-05,
      "step": 90112,
      "training_step_time": 7.4771575927734375
    },
    {
      "epoch": 1.5366554260253906e-05,
      "model_forward_time": 1.7281277179718018,
      "step": 90113
    },
    {
      "epoch": 1.5366554260253906e-05,
      "step": 90113,
      "training_step_time": 7.315783977508545
    },
    {
      "epoch": 1.5367507934570313e-05,
      "model_forward_time": 1.738515853881836,
      "step": 90114
    },
    {
      "epoch": 1.5367507934570313e-05,
      "step": 90114,
      "training_step_time": 6.850153684616089
    },
    {
      "epoch": 1.5368461608886717e-05,
      "model_forward_time": 1.7551112174987793,
      "step": 90115
    },
    {
      "epoch": 1.5368461608886717e-05,
      "step": 90115,
      "training_step_time": 7.560735464096069
    },
    {
      "epoch": 1.5369415283203125e-05,
      "model_forward_time": 2.3791184425354004,
      "step": 90116
    },
    {
      "epoch": 1.5369415283203125e-05,
      "step": 90116,
      "training_step_time": 8.669809341430664
    },
    {
      "epoch": 1.5370368957519532e-05,
      "model_forward_time": 1.8434886932373047,
      "step": 90117
    },
    {
      "epoch": 1.5370368957519532e-05,
      "step": 90117,
      "training_step_time": 10.769221544265747
    },
    {
      "epoch": 1.5371322631835936e-05,
      "model_forward_time": 1.7401044368743896,
      "step": 90118
    },
    {
      "epoch": 1.5371322631835936e-05,
      "step": 90118,
      "training_step_time": 8.49838662147522
    },
    {
      "epoch": 1.5372276306152344e-05,
      "model_forward_time": 1.7548222541809082,
      "step": 90119
    },
    {
      "epoch": 1.5372276306152344e-05,
      "step": 90119,
      "training_step_time": 7.7610814571380615
    },
    {
      "epoch": 1.537322998046875e-05,
      "grad_norm": 0.07184359431266785,
      "learning_rate": 2.645615446727995e-07,
      "loss": 0.0549,
      "step": 90120
    },
    {
      "epoch": 1.537322998046875e-05,
      "model_forward_time": 1.74554443359375,
      "step": 90120
    },
    {
      "dynamics_loss_avg": 0.07325396034866571,
      "epoch": 1.537322998046875e-05,
      "step": 90120
    },
    {
      "action_loss_avg": 0.0020175115147139876,
      "epoch": 1.537322998046875e-05,
      "step": 90120
    },
    {
      "epoch": 1.537322998046875e-05,
      "step": 90120,
      "training_step_time": 7.760262966156006
    },
    {
      "epoch": 1.5374183654785155e-05,
      "model_forward_time": 1.7468132972717285,
      "step": 90121
    },
    {
      "epoch": 1.5374183654785155e-05,
      "step": 90121,
      "training_step_time": 7.437211275100708
    },
    {
      "epoch": 1.5375137329101562e-05,
      "model_forward_time": 1.7431352138519287,
      "step": 90122
    },
    {
      "epoch": 1.5375137329101562e-05,
      "step": 90122,
      "training_step_time": 7.2514424324035645
    },
    {
      "epoch": 1.537609100341797e-05,
      "model_forward_time": 1.7568442821502686,
      "step": 90123
    },
    {
      "epoch": 1.537609100341797e-05,
      "step": 90123,
      "training_step_time": 6.777622699737549
    },
    {
      "epoch": 1.5377044677734374e-05,
      "model_forward_time": 1.7452480792999268,
      "step": 90124
    },
    {
      "epoch": 1.5377044677734374e-05,
      "step": 90124,
      "training_step_time": 7.505935192108154
    },
    {
      "epoch": 1.537799835205078e-05,
      "model_forward_time": 2.367403268814087,
      "step": 90125
    },
    {
      "epoch": 1.537799835205078e-05,
      "step": 90125,
      "training_step_time": 7.842389345169067
    },
    {
      "epoch": 1.537895202636719e-05,
      "model_forward_time": 1.8168442249298096,
      "step": 90126
    },
    {
      "epoch": 1.537895202636719e-05,
      "step": 90126,
      "training_step_time": 8.619100570678711
    },
    {
      "epoch": 1.5379905700683593e-05,
      "model_forward_time": 1.7463009357452393,
      "step": 90127
    },
    {
      "epoch": 1.5379905700683593e-05,
      "step": 90127,
      "training_step_time": 9.094848871231079
    },
    {
      "epoch": 1.5380859375e-05,
      "model_forward_time": 1.7416186332702637,
      "step": 90128
    },
    {
      "epoch": 1.5380859375e-05,
      "step": 90128,
      "training_step_time": 8.320444345474243
    },
    {
      "epoch": 1.5381813049316408e-05,
      "model_forward_time": 1.7574107646942139,
      "step": 90129
    },
    {
      "epoch": 1.5381813049316408e-05,
      "step": 90129,
      "training_step_time": 7.765332460403442
    },
    {
      "epoch": 1.538276672363281e-05,
      "grad_norm": 0.09457777440547943,
      "learning_rate": 2.6403108125436115e-07,
      "loss": 0.0547,
      "step": 90130
    },
    {
      "epoch": 1.538276672363281e-05,
      "model_forward_time": 1.7433390617370605,
      "step": 90130
    },
    {
      "dynamics_loss_avg": 0.06064941976219416,
      "epoch": 1.538276672363281e-05,
      "step": 90130
    },
    {
      "action_loss_avg": 0.0014210219669621438,
      "epoch": 1.538276672363281e-05,
      "step": 90130
    },
    {
      "epoch": 1.538276672363281e-05,
      "step": 90130,
      "training_step_time": 7.776700973510742
    },
    {
      "epoch": 1.538372039794922e-05,
      "model_forward_time": 1.7469301223754883,
      "step": 90131
    },
    {
      "epoch": 1.538372039794922e-05,
      "step": 90131,
      "training_step_time": 7.504810094833374
    },
    {
      "epoch": 1.5384674072265626e-05,
      "model_forward_time": 1.7472248077392578,
      "step": 90132
    },
    {
      "epoch": 1.5384674072265626e-05,
      "step": 90132,
      "training_step_time": 7.309481143951416
    },
    {
      "epoch": 1.538562774658203e-05,
      "model_forward_time": 1.769911289215088,
      "step": 90133
    },
    {
      "epoch": 1.538562774658203e-05,
      "step": 90133,
      "training_step_time": 7.532933235168457
    },
    {
      "epoch": 1.5386581420898438e-05,
      "model_forward_time": 1.7441442012786865,
      "step": 90134
    },
    {
      "epoch": 1.5386581420898438e-05,
      "step": 90134,
      "training_step_time": 8.626656293869019
    },
    {
      "epoch": 1.5387535095214845e-05,
      "model_forward_time": 2.4642882347106934,
      "step": 90135
    },
    {
      "epoch": 1.5387535095214845e-05,
      "step": 90135,
      "training_step_time": 10.128115177154541
    },
    {
      "epoch": 1.538848876953125e-05,
      "model_forward_time": 1.7516758441925049,
      "step": 90136
    },
    {
      "epoch": 1.538848876953125e-05,
      "step": 90136,
      "training_step_time": 9.852395057678223
    },
    {
      "epoch": 1.5389442443847657e-05,
      "model_forward_time": 1.7499475479125977,
      "step": 90137
    },
    {
      "epoch": 1.5389442443847657e-05,
      "step": 90137,
      "training_step_time": 7.708407402038574
    },
    {
      "epoch": 1.5390396118164064e-05,
      "model_forward_time": 1.7435364723205566,
      "step": 90138
    },
    {
      "epoch": 1.5390396118164064e-05,
      "step": 90138,
      "training_step_time": 7.733190298080444
    },
    {
      "epoch": 1.5391349792480468e-05,
      "model_forward_time": 1.7515158653259277,
      "step": 90139
    },
    {
      "epoch": 1.5391349792480468e-05,
      "step": 90139,
      "training_step_time": 7.642322540283203
    },
    {
      "epoch": 1.5392303466796876e-05,
      "grad_norm": 0.09260361641645432,
      "learning_rate": 2.6350113575436266e-07,
      "loss": 0.0548,
      "step": 90140
    },
    {
      "epoch": 1.5392303466796876e-05,
      "model_forward_time": 1.7442033290863037,
      "step": 90140
    },
    {
      "dynamics_loss_avg": 0.06179754175245762,
      "epoch": 1.5392303466796876e-05,
      "step": 90140
    },
    {
      "action_loss_avg": 0.0016013561267754993,
      "epoch": 1.5392303466796876e-05,
      "step": 90140
    },
    {
      "epoch": 1.5392303466796876e-05,
      "step": 90140,
      "training_step_time": 7.490785598754883
    },
    {
      "epoch": 1.539325714111328e-05,
      "model_forward_time": 1.7564630508422852,
      "step": 90141
    },
    {
      "epoch": 1.539325714111328e-05,
      "step": 90141,
      "training_step_time": 7.328397989273071
    },
    {
      "epoch": 1.5394210815429687e-05,
      "model_forward_time": 1.7474236488342285,
      "step": 90142
    },
    {
      "epoch": 1.5394210815429687e-05,
      "step": 90142,
      "training_step_time": 7.139577388763428
    },
    {
      "epoch": 1.5395164489746094e-05,
      "model_forward_time": 1.7429821491241455,
      "step": 90143
    },
    {
      "epoch": 1.5395164489746094e-05,
      "step": 90143,
      "training_step_time": 7.8162970542907715
    },
    {
      "epoch": 1.53961181640625e-05,
      "model_forward_time": 2.440730333328247,
      "step": 90144
    },
    {
      "epoch": 1.53961181640625e-05,
      "step": 90144,
      "training_step_time": 9.530403852462769
    },
    {
      "epoch": 1.5397071838378906e-05,
      "model_forward_time": 1.7477378845214844,
      "step": 90145
    },
    {
      "epoch": 1.5397071838378906e-05,
      "step": 90145,
      "training_step_time": 9.703309774398804
    },
    {
      "epoch": 1.5398025512695313e-05,
      "model_forward_time": 1.750497579574585,
      "step": 90146
    },
    {
      "epoch": 1.5398025512695313e-05,
      "step": 90146,
      "training_step_time": 8.883065462112427
    },
    {
      "epoch": 1.5398979187011717e-05,
      "model_forward_time": 1.7432844638824463,
      "step": 90147
    },
    {
      "epoch": 1.5398979187011717e-05,
      "step": 90147,
      "training_step_time": 7.8315606117248535
    },
    {
      "epoch": 1.5399932861328125e-05,
      "model_forward_time": 1.7793009281158447,
      "step": 90148
    },
    {
      "epoch": 1.5399932861328125e-05,
      "step": 90148,
      "training_step_time": 7.703577756881714
    },
    {
      "epoch": 1.5400886535644532e-05,
      "model_forward_time": 1.7377889156341553,
      "step": 90149
    },
    {
      "epoch": 1.5400886535644532e-05,
      "step": 90149,
      "training_step_time": 7.501251220703125
    },
    {
      "epoch": 1.5401840209960936e-05,
      "grad_norm": 0.06966027617454529,
      "learning_rate": 2.629717082307598e-07,
      "loss": 0.0538,
      "step": 90150
    },
    {
      "epoch": 1.5401840209960936e-05,
      "model_forward_time": 1.735252857208252,
      "step": 90150
    },
    {
      "dynamics_loss_avg": 0.06679910998791457,
      "epoch": 1.5401840209960936e-05,
      "step": 90150
    },
    {
      "action_loss_avg": 0.001498395626549609,
      "epoch": 1.5401840209960936e-05,
      "step": 90150
    },
    {
      "epoch": 1.5401840209960936e-05,
      "step": 90150,
      "training_step_time": 7.334827184677124
    },
    {
      "epoch": 1.5402793884277344e-05,
      "model_forward_time": 1.745575189590454,
      "step": 90151
    },
    {
      "epoch": 1.5402793884277344e-05,
      "step": 90151,
      "training_step_time": 6.806236982345581
    },
    {
      "epoch": 1.540374755859375e-05,
      "model_forward_time": 1.7751052379608154,
      "step": 90152
    },
    {
      "epoch": 1.540374755859375e-05,
      "step": 90152,
      "training_step_time": 9.313028335571289
    },
    {
      "epoch": 1.5404701232910155e-05,
      "model_forward_time": 2.415771961212158,
      "step": 90153
    },
    {
      "epoch": 1.5404701232910155e-05,
      "step": 90153,
      "training_step_time": 7.647661209106445
    },
    {
      "epoch": 1.5405654907226562e-05,
      "model_forward_time": 1.8889906406402588,
      "step": 90154
    },
    {
      "epoch": 1.5405654907226562e-05,
      "step": 90154,
      "training_step_time": 7.774280071258545
    },
    {
      "epoch": 1.540660858154297e-05,
      "model_forward_time": 1.7388942241668701,
      "step": 90155
    },
    {
      "epoch": 1.540660858154297e-05,
      "step": 90155,
      "training_step_time": 9.798112869262695
    },
    {
      "epoch": 1.5407562255859374e-05,
      "model_forward_time": 1.7732017040252686,
      "step": 90156
    },
    {
      "epoch": 1.5407562255859374e-05,
      "step": 90156,
      "training_step_time": 7.750950574874878
    },
    {
      "epoch": 1.540851593017578e-05,
      "model_forward_time": 1.7592203617095947,
      "step": 90157
    },
    {
      "epoch": 1.540851593017578e-05,
      "step": 90157,
      "training_step_time": 7.723630428314209
    },
    {
      "epoch": 1.540946960449219e-05,
      "model_forward_time": 1.7887771129608154,
      "step": 90158
    },
    {
      "epoch": 1.540946960449219e-05,
      "step": 90158,
      "training_step_time": 7.705200910568237
    },
    {
      "epoch": 1.5410423278808593e-05,
      "model_forward_time": 1.7747797966003418,
      "step": 90159
    },
    {
      "epoch": 1.5410423278808593e-05,
      "step": 90159,
      "training_step_time": 7.175528049468994
    },
    {
      "epoch": 1.5411376953125e-05,
      "grad_norm": 0.08689256012439728,
      "learning_rate": 2.6244279874144796e-07,
      "loss": 0.0539,
      "step": 90160
    },
    {
      "epoch": 1.5411376953125e-05,
      "model_forward_time": 1.7613470554351807,
      "step": 90160
    },
    {
      "dynamics_loss_avg": 0.04805840440094471,
      "epoch": 1.5411376953125e-05,
      "step": 90160
    },
    {
      "action_loss_avg": 0.0017687743733404205,
      "epoch": 1.5411376953125e-05,
      "step": 90160
    },
    {
      "epoch": 1.5411376953125e-05,
      "step": 90160,
      "training_step_time": 7.2553088665008545
    },
    {
      "epoch": 1.5412330627441408e-05,
      "model_forward_time": 1.7766988277435303,
      "step": 90161
    },
    {
      "epoch": 1.5412330627441408e-05,
      "step": 90161,
      "training_step_time": 8.112605810165405
    },
    {
      "epoch": 1.541328430175781e-05,
      "model_forward_time": 2.4637138843536377,
      "step": 90162
    },
    {
      "epoch": 1.541328430175781e-05,
      "step": 90162,
      "training_step_time": 8.129053115844727
    },
    {
      "epoch": 1.541423797607422e-05,
      "model_forward_time": 1.8816945552825928,
      "step": 90163
    },
    {
      "epoch": 1.541423797607422e-05,
      "step": 90163,
      "training_step_time": 9.789580821990967
    },
    {
      "epoch": 1.5415191650390626e-05,
      "model_forward_time": 1.7733356952667236,
      "step": 90164
    },
    {
      "epoch": 1.5415191650390626e-05,
      "step": 90164,
      "training_step_time": 8.765494585037231
    },
    {
      "epoch": 1.541614532470703e-05,
      "model_forward_time": 1.7706501483917236,
      "step": 90165
    },
    {
      "epoch": 1.541614532470703e-05,
      "step": 90165,
      "training_step_time": 8.407701969146729
    },
    {
      "epoch": 1.5417098999023438e-05,
      "model_forward_time": 1.0583386421203613,
      "step": 90166
    },
    {
      "epoch": 1.5417098999023438e-05,
      "step": 90166,
      "training_step_time": 7.736016511917114
    },
    {
      "epoch": 1.5418052673339845e-05,
      "model_forward_time": 1.7767024040222168,
      "step": 90167
    },
    {
      "epoch": 1.5418052673339845e-05,
      "step": 90167,
      "training_step_time": 7.727804660797119
    },
    {
      "epoch": 1.541900634765625e-05,
      "model_forward_time": 1.7756612300872803,
      "step": 90168
    },
    {
      "epoch": 1.541900634765625e-05,
      "step": 90168,
      "training_step_time": 7.523112535476685
    },
    {
      "epoch": 1.5419960021972657e-05,
      "model_forward_time": 1.8003463745117188,
      "step": 90169
    },
    {
      "epoch": 1.5419960021972657e-05,
      "step": 90169,
      "training_step_time": 7.331143140792847
    },
    {
      "epoch": 1.5420913696289064e-05,
      "grad_norm": 0.09143029898405075,
      "learning_rate": 2.6191440734426865e-07,
      "loss": 0.0524,
      "step": 90170
    },
    {
      "epoch": 1.5420913696289064e-05,
      "model_forward_time": 1.769639492034912,
      "step": 90170
    },
    {
      "dynamics_loss_avg": 0.06418232452124358,
      "epoch": 1.5420913696289064e-05,
      "step": 90170
    },
    {
      "action_loss_avg": 0.001006991055328399,
      "epoch": 1.5420913696289064e-05,
      "step": 90170
    },
    {
      "epoch": 1.5420913696289064e-05,
      "step": 90170,
      "training_step_time": 7.496445655822754
    },
    {
      "epoch": 1.5421867370605468e-05,
      "model_forward_time": 1.7744731903076172,
      "step": 90171
    },
    {
      "epoch": 1.5421867370605468e-05,
      "step": 90171,
      "training_step_time": 8.657332420349121
    },
    {
      "epoch": 1.5422821044921876e-05,
      "model_forward_time": 2.5043091773986816,
      "step": 90172
    },
    {
      "epoch": 1.5422821044921876e-05,
      "step": 90172,
      "training_step_time": 8.852525234222412
    },
    {
      "epoch": 1.542377471923828e-05,
      "model_forward_time": 1.7786974906921387,
      "step": 90173
    },
    {
      "epoch": 1.542377471923828e-05,
      "step": 90173,
      "training_step_time": 11.224486827850342
    },
    {
      "epoch": 1.5424728393554687e-05,
      "model_forward_time": 1.782609462738037,
      "step": 90174
    },
    {
      "epoch": 1.5424728393554687e-05,
      "step": 90174,
      "training_step_time": 8.507667541503906
    },
    {
      "epoch": 1.5425682067871094e-05,
      "model_forward_time": 1.7738885879516602,
      "step": 90175
    },
    {
      "epoch": 1.5425682067871094e-05,
      "step": 90175,
      "training_step_time": 7.7909932136535645
    },
    {
      "epoch": 1.54266357421875e-05,
      "model_forward_time": 1.7726705074310303,
      "step": 90176
    },
    {
      "epoch": 1.54266357421875e-05,
      "step": 90176,
      "training_step_time": 7.726316928863525
    },
    {
      "epoch": 1.5427589416503906e-05,
      "model_forward_time": 1.7729713916778564,
      "step": 90177
    },
    {
      "epoch": 1.5427589416503906e-05,
      "step": 90177,
      "training_step_time": 7.515482425689697
    },
    {
      "epoch": 1.5428543090820313e-05,
      "model_forward_time": 1.7638394832611084,
      "step": 90178
    },
    {
      "epoch": 1.5428543090820313e-05,
      "step": 90178,
      "training_step_time": 7.339522123336792
    },
    {
      "epoch": 1.5429496765136717e-05,
      "model_forward_time": 1.7779152393341064,
      "step": 90179
    },
    {
      "epoch": 1.5429496765136717e-05,
      "step": 90179,
      "training_step_time": 8.078640937805176
    },
    {
      "epoch": 1.5430450439453125e-05,
      "grad_norm": 0.07641912996768951,
      "learning_rate": 2.613865340970062e-07,
      "loss": 0.052,
      "step": 90180
    },
    {
      "epoch": 1.5430450439453125e-05,
      "model_forward_time": 1.7679803371429443,
      "step": 90180
    },
    {
      "dynamics_loss_avg": 0.050460370071232316,
      "epoch": 1.5430450439453125e-05,
      "step": 90180
    },
    {
      "action_loss_avg": 0.001176985146594234,
      "epoch": 1.5430450439453125e-05,
      "step": 90180
    },
    {
      "epoch": 1.5430450439453125e-05,
      "step": 90180,
      "training_step_time": 7.807657480239868
    },
    {
      "epoch": 1.5431404113769532e-05,
      "model_forward_time": 2.495121479034424,
      "step": 90181
    },
    {
      "epoch": 1.5431404113769532e-05,
      "step": 90181,
      "training_step_time": 9.130589962005615
    },
    {
      "epoch": 1.5432357788085936e-05,
      "model_forward_time": 1.7741987705230713,
      "step": 90182
    },
    {
      "epoch": 1.5432357788085936e-05,
      "step": 90182,
      "training_step_time": 10.382995843887329
    },
    {
      "epoch": 1.5433311462402344e-05,
      "model_forward_time": 1.77762770652771,
      "step": 90183
    },
    {
      "epoch": 1.5433311462402344e-05,
      "step": 90183,
      "training_step_time": 7.802436828613281
    },
    {
      "epoch": 1.543426513671875e-05,
      "model_forward_time": 1.7832629680633545,
      "step": 90184
    },
    {
      "epoch": 1.543426513671875e-05,
      "step": 90184,
      "training_step_time": 7.6762847900390625
    },
    {
      "epoch": 1.5435218811035155e-05,
      "model_forward_time": 1.7687098979949951,
      "step": 90185
    },
    {
      "epoch": 1.5435218811035155e-05,
      "step": 90185,
      "training_step_time": 7.7189507484436035
    },
    {
      "epoch": 1.5436172485351562e-05,
      "model_forward_time": 1.7837262153625488,
      "step": 90186
    },
    {
      "epoch": 1.5436172485351562e-05,
      "step": 90186,
      "training_step_time": 7.475577116012573
    },
    {
      "epoch": 1.543712615966797e-05,
      "model_forward_time": 1.7743451595306396,
      "step": 90187
    },
    {
      "epoch": 1.543712615966797e-05,
      "step": 90187,
      "training_step_time": 7.276294946670532
    },
    {
      "epoch": 1.5438079833984374e-05,
      "model_forward_time": 1.7813115119934082,
      "step": 90188
    },
    {
      "epoch": 1.5438079833984374e-05,
      "step": 90188,
      "training_step_time": 7.3667755126953125
    },
    {
      "epoch": 1.543903350830078e-05,
      "model_forward_time": 1.7763419151306152,
      "step": 90189
    },
    {
      "epoch": 1.543903350830078e-05,
      "step": 90189,
      "training_step_time": 8.610839366912842
    },
    {
      "epoch": 1.543998718261719e-05,
      "grad_norm": 0.07052921503782272,
      "learning_rate": 2.6085917905738887e-07,
      "loss": 0.0532,
      "step": 90190
    },
    {
      "epoch": 1.543998718261719e-05,
      "model_forward_time": 2.4328527450561523,
      "step": 90190
    },
    {
      "dynamics_loss_avg": 0.05361379943788051,
      "epoch": 1.543998718261719e-05,
      "step": 90190
    },
    {
      "action_loss_avg": 0.0013804117043036968,
      "epoch": 1.543998718261719e-05,
      "step": 90190
    },
    {
      "epoch": 1.543998718261719e-05,
      "step": 90190,
      "training_step_time": 7.7147157192230225
    },
    {
      "epoch": 1.5440940856933593e-05,
      "model_forward_time": 1.769493818283081,
      "step": 90191
    },
    {
      "epoch": 1.5440940856933593e-05,
      "step": 90191,
      "training_step_time": 7.796082258224487
    },
    {
      "epoch": 1.544189453125e-05,
      "model_forward_time": 1.779860019683838,
      "step": 90192
    },
    {
      "epoch": 1.544189453125e-05,
      "step": 90192,
      "training_step_time": 8.80715012550354
    },
    {
      "epoch": 1.5442848205566408e-05,
      "model_forward_time": 1.7764220237731934,
      "step": 90193
    },
    {
      "epoch": 1.5442848205566408e-05,
      "step": 90193,
      "training_step_time": 7.702872037887573
    },
    {
      "epoch": 1.544380187988281e-05,
      "model_forward_time": 1.7741713523864746,
      "step": 90194
    },
    {
      "epoch": 1.544380187988281e-05,
      "step": 90194,
      "training_step_time": 8.715271949768066
    },
    {
      "epoch": 1.544475555419922e-05,
      "model_forward_time": 1.7884831428527832,
      "step": 90195
    },
    {
      "epoch": 1.544475555419922e-05,
      "step": 90195,
      "training_step_time": 7.6769938468933105
    },
    {
      "epoch": 1.5445709228515626e-05,
      "model_forward_time": 1.7795166969299316,
      "step": 90196
    },
    {
      "epoch": 1.5445709228515626e-05,
      "step": 90196,
      "training_step_time": 7.294132709503174
    },
    {
      "epoch": 1.544666290283203e-05,
      "model_forward_time": 1.7707195281982422,
      "step": 90197
    },
    {
      "epoch": 1.544666290283203e-05,
      "step": 90197,
      "training_step_time": 7.386396884918213
    },
    {
      "epoch": 1.5447616577148438e-05,
      "model_forward_time": 1.778294563293457,
      "step": 90198
    },
    {
      "epoch": 1.5447616577148438e-05,
      "step": 90198,
      "training_step_time": 7.7956929206848145
    },
    {
      "epoch": 1.5448570251464845e-05,
      "model_forward_time": 2.4904417991638184,
      "step": 90199
    },
    {
      "epoch": 1.5448570251464845e-05,
      "step": 90199,
      "training_step_time": 8.43392825126648
    },
    {
      "epoch": 1.544952392578125e-05,
      "grad_norm": 0.08219436556100845,
      "learning_rate": 2.603323422830861e-07,
      "loss": 0.0534,
      "step": 90200
    },
    {
      "epoch": 1.544952392578125e-05,
      "model_forward_time": 1.8719496726989746,
      "step": 90200
    },
    {
      "dynamics_loss_avg": 0.04402809459716082,
      "epoch": 1.544952392578125e-05,
      "step": 90200
    },
    {
      "action_loss_avg": 0.0010597151762340219,
      "epoch": 1.544952392578125e-05,
      "step": 90200
    },
    {
      "epoch": 1.544952392578125e-05,
      "step": 90200,
      "training_step_time": 9.764926671981812
    },
    {
      "epoch": 1.5450477600097657e-05,
      "model_forward_time": 1.7792408466339111,
      "step": 90201
    },
    {
      "epoch": 1.5450477600097657e-05,
      "step": 90201,
      "training_step_time": 8.689059734344482
    },
    {
      "epoch": 1.5451431274414064e-05,
      "model_forward_time": 1.7726244926452637,
      "step": 90202
    },
    {
      "epoch": 1.5451431274414064e-05,
      "step": 90202,
      "training_step_time": 7.810307502746582
    },
    {
      "epoch": 1.5452384948730468e-05,
      "model_forward_time": 1.7691636085510254,
      "step": 90203
    },
    {
      "epoch": 1.5452384948730468e-05,
      "step": 90203,
      "training_step_time": 7.794529914855957
    },
    {
      "epoch": 1.5453338623046876e-05,
      "model_forward_time": 1.7770414352416992,
      "step": 90204
    },
    {
      "epoch": 1.5453338623046876e-05,
      "step": 90204,
      "training_step_time": 7.69731879234314
    },
    {
      "epoch": 1.545429229736328e-05,
      "model_forward_time": 1.7747471332550049,
      "step": 90205
    },
    {
      "epoch": 1.545429229736328e-05,
      "step": 90205,
      "training_step_time": 7.463223218917847
    },
    {
      "epoch": 1.5455245971679687e-05,
      "model_forward_time": 1.764308214187622,
      "step": 90206
    },
    {
      "epoch": 1.5455245971679687e-05,
      "step": 90206,
      "training_step_time": 7.351999044418335
    },
    {
      "epoch": 1.5456199645996094e-05,
      "model_forward_time": 1.7701330184936523,
      "step": 90207
    },
    {
      "epoch": 1.5456199645996094e-05,
      "step": 90207,
      "training_step_time": 8.349568843841553
    },
    {
      "epoch": 1.54571533203125e-05,
      "model_forward_time": 2.452458620071411,
      "step": 90208
    },
    {
      "epoch": 1.54571533203125e-05,
      "step": 90208,
      "training_step_time": 9.310898542404175
    },
    {
      "epoch": 1.5458106994628906e-05,
      "model_forward_time": 1.8779783248901367,
      "step": 90209
    },
    {
      "epoch": 1.5458106994628906e-05,
      "step": 90209,
      "training_step_time": 11.25276255607605
    },
    {
      "epoch": 1.5459060668945313e-05,
      "grad_norm": 0.6311371326446533,
      "learning_rate": 2.5980602383171184e-07,
      "loss": 0.0525,
      "step": 90210
    },
    {
      "epoch": 1.5459060668945313e-05,
      "model_forward_time": 1.7399308681488037,
      "step": 90210
    },
    {
      "dynamics_loss_avg": 0.05632718820124864,
      "epoch": 1.5459060668945313e-05,
      "step": 90210
    },
    {
      "action_loss_avg": 0.0015418960654642432,
      "epoch": 1.5459060668945313e-05,
      "step": 90210
    },
    {
      "epoch": 1.5459060668945313e-05,
      "step": 90210,
      "training_step_time": 8.80232310295105
    },
    {
      "epoch": 1.5460014343261717e-05,
      "model_forward_time": 1.764103651046753,
      "step": 90211
    },
    {
      "epoch": 1.5460014343261717e-05,
      "step": 90211,
      "training_step_time": 7.778172731399536
    },
    {
      "epoch": 1.5460968017578125e-05,
      "model_forward_time": 1.724909782409668,
      "step": 90212
    },
    {
      "epoch": 1.5460968017578125e-05,
      "step": 90212,
      "training_step_time": 7.796064376831055
    },
    {
      "epoch": 1.5461921691894532e-05,
      "model_forward_time": 1.7453360557556152,
      "step": 90213
    },
    {
      "epoch": 1.5461921691894532e-05,
      "step": 90213,
      "training_step_time": 7.758312463760376
    },
    {
      "epoch": 1.5462875366210936e-05,
      "model_forward_time": 1.7480947971343994,
      "step": 90214
    },
    {
      "epoch": 1.5462875366210936e-05,
      "step": 90214,
      "training_step_time": 7.472116470336914
    },
    {
      "epoch": 1.5463829040527344e-05,
      "model_forward_time": 1.7577390670776367,
      "step": 90215
    },
    {
      "epoch": 1.5463829040527344e-05,
      "step": 90215,
      "training_step_time": 7.377023458480835
    },
    {
      "epoch": 1.546478271484375e-05,
      "model_forward_time": 1.7449781894683838,
      "step": 90216
    },
    {
      "epoch": 1.546478271484375e-05,
      "step": 90216,
      "training_step_time": 7.681167840957642
    },
    {
      "epoch": 1.5465736389160155e-05,
      "model_forward_time": 2.3690686225891113,
      "step": 90217
    },
    {
      "epoch": 1.5465736389160155e-05,
      "step": 90217,
      "training_step_time": 9.307435274124146
    },
    {
      "epoch": 1.5466690063476562e-05,
      "model_forward_time": 1.8331406116485596,
      "step": 90218
    },
    {
      "epoch": 1.5466690063476562e-05,
      "step": 90218,
      "training_step_time": 7.838321208953857
    },
    {
      "epoch": 1.546764373779297e-05,
      "model_forward_time": 1.7400476932525635,
      "step": 90219
    },
    {
      "epoch": 1.546764373779297e-05,
      "step": 90219,
      "training_step_time": 8.320710182189941
    },
    {
      "epoch": 1.5468597412109374e-05,
      "grad_norm": 0.09253111481666565,
      "learning_rate": 2.5928022376082494e-07,
      "loss": 0.0546,
      "step": 90220
    },
    {
      "epoch": 1.5468597412109374e-05,
      "model_forward_time": 1.7229909896850586,
      "step": 90220
    },
    {
      "dynamics_loss_avg": 0.059274287149310115,
      "epoch": 1.5468597412109374e-05,
      "step": 90220
    },
    {
      "action_loss_avg": 0.0015652978501748294,
      "epoch": 1.5468597412109374e-05,
      "step": 90220
    },
    {
      "epoch": 1.5468597412109374e-05,
      "step": 90220,
      "training_step_time": 7.749898195266724
    },
    {
      "epoch": 1.546955108642578e-05,
      "model_forward_time": 1.7422921657562256,
      "step": 90221
    },
    {
      "epoch": 1.546955108642578e-05,
      "step": 90221,
      "training_step_time": 7.865005254745483
    },
    {
      "epoch": 1.547050476074219e-05,
      "model_forward_time": 1.7342095375061035,
      "step": 90222
    },
    {
      "epoch": 1.547050476074219e-05,
      "step": 90222,
      "training_step_time": 7.818876504898071
    },
    {
      "epoch": 1.5471458435058593e-05,
      "model_forward_time": 1.7304770946502686,
      "step": 90223
    },
    {
      "epoch": 1.5471458435058593e-05,
      "step": 90223,
      "training_step_time": 7.543072938919067
    },
    {
      "epoch": 1.5472412109375e-05,
      "model_forward_time": 1.7375473976135254,
      "step": 90224
    },
    {
      "epoch": 1.5472412109375e-05,
      "step": 90224,
      "training_step_time": 8.227640151977539
    },
    {
      "epoch": 1.5473365783691408e-05,
      "model_forward_time": 1.729968547821045,
      "step": 90225
    },
    {
      "epoch": 1.5473365783691408e-05,
      "step": 90225,
      "training_step_time": 7.688902139663696
    },
    {
      "epoch": 1.547431945800781e-05,
      "model_forward_time": 1.7279706001281738,
      "step": 90226
    },
    {
      "epoch": 1.547431945800781e-05,
      "step": 90226,
      "training_step_time": 7.888717412948608
    },
    {
      "epoch": 1.547527313232422e-05,
      "model_forward_time": 2.4323692321777344,
      "step": 90227
    },
    {
      "epoch": 1.547527313232422e-05,
      "step": 90227,
      "training_step_time": 10.68624496459961
    },
    {
      "epoch": 1.5476226806640626e-05,
      "model_forward_time": 1.7311499118804932,
      "step": 90228
    },
    {
      "epoch": 1.5476226806640626e-05,
      "step": 90228,
      "training_step_time": 9.754917621612549
    },
    {
      "epoch": 1.547718048095703e-05,
      "model_forward_time": 1.7349095344543457,
      "step": 90229
    },
    {
      "epoch": 1.547718048095703e-05,
      "step": 90229,
      "training_step_time": 8.33821725845337
    },
    {
      "epoch": 1.5478134155273438e-05,
      "grad_norm": 0.10957727581262589,
      "learning_rate": 2.5875494212792505e-07,
      "loss": 0.0529,
      "step": 90230
    },
    {
      "epoch": 1.5478134155273438e-05,
      "model_forward_time": 1.7285451889038086,
      "step": 90230
    },
    {
      "dynamics_loss_avg": 0.04415131155401468,
      "epoch": 1.5478134155273438e-05,
      "step": 90230
    },
    {
      "action_loss_avg": 0.001396711870620493,
      "epoch": 1.5478134155273438e-05,
      "step": 90230
    },
    {
      "epoch": 1.5478134155273438e-05,
      "step": 90230,
      "training_step_time": 7.693724632263184
    },
    {
      "epoch": 1.5479087829589845e-05,
      "model_forward_time": 1.7467811107635498,
      "step": 90231
    },
    {
      "epoch": 1.5479087829589845e-05,
      "step": 90231,
      "training_step_time": 7.764727592468262
    },
    {
      "epoch": 1.548004150390625e-05,
      "model_forward_time": 1.730264663696289,
      "step": 90232
    },
    {
      "epoch": 1.548004150390625e-05,
      "step": 90232,
      "training_step_time": 7.739332437515259
    },
    {
      "epoch": 1.5480995178222657e-05,
      "model_forward_time": 1.7371826171875,
      "step": 90233
    },
    {
      "epoch": 1.5480995178222657e-05,
      "step": 90233,
      "training_step_time": 7.358534812927246
    },
    {
      "epoch": 1.5481948852539064e-05,
      "model_forward_time": 1.725951910018921,
      "step": 90234
    },
    {
      "epoch": 1.5481948852539064e-05,
      "step": 90234,
      "training_step_time": 7.729128360748291
    },
    {
      "epoch": 1.5482902526855468e-05,
      "model_forward_time": 1.743727684020996,
      "step": 90235
    },
    {
      "epoch": 1.5482902526855468e-05,
      "step": 90235,
      "training_step_time": 7.753944158554077
    },
    {
      "epoch": 1.5483856201171876e-05,
      "model_forward_time": 2.439955472946167,
      "step": 90236
    },
    {
      "epoch": 1.5483856201171876e-05,
      "step": 90236,
      "training_step_time": 7.7801902294158936
    },
    {
      "epoch": 1.548480987548828e-05,
      "model_forward_time": 1.7443273067474365,
      "step": 90237
    },
    {
      "epoch": 1.548480987548828e-05,
      "step": 90237,
      "training_step_time": 9.20238733291626
    },
    {
      "epoch": 1.5485763549804687e-05,
      "model_forward_time": 1.3674190044403076,
      "step": 90238
    },
    {
      "epoch": 1.5485763549804687e-05,
      "step": 90238,
      "training_step_time": 7.77384090423584
    },
    {
      "epoch": 1.5486717224121094e-05,
      "model_forward_time": 1.7341094017028809,
      "step": 90239
    },
    {
      "epoch": 1.5486717224121094e-05,
      "step": 90239,
      "training_step_time": 8.496750593185425
    },
    {
      "epoch": 1.54876708984375e-05,
      "grad_norm": 0.12854233384132385,
      "learning_rate": 2.5823017899045565e-07,
      "loss": 0.054,
      "step": 90240
    },
    {
      "epoch": 1.54876708984375e-05,
      "model_forward_time": 1.7331538200378418,
      "step": 90240
    },
    {
      "dynamics_loss_avg": 0.04595056492835283,
      "epoch": 1.54876708984375e-05,
      "step": 90240
    },
    {
      "action_loss_avg": 0.0014542517135851085,
      "epoch": 1.54876708984375e-05,
      "step": 90240
    },
    {
      "epoch": 1.54876708984375e-05,
      "step": 90240,
      "training_step_time": 7.607927560806274
    },
    {
      "epoch": 1.5488624572753906e-05,
      "model_forward_time": 1.7399325370788574,
      "step": 90241
    },
    {
      "epoch": 1.5488624572753906e-05,
      "step": 90241,
      "training_step_time": 7.7332375049591064
    },
    {
      "epoch": 1.5489578247070313e-05,
      "model_forward_time": 1.7342193126678467,
      "step": 90242
    },
    {
      "epoch": 1.5489578247070313e-05,
      "step": 90242,
      "training_step_time": 7.260295152664185
    },
    {
      "epoch": 1.5490531921386717e-05,
      "model_forward_time": 1.733954668045044,
      "step": 90243
    },
    {
      "epoch": 1.5490531921386717e-05,
      "step": 90243,
      "training_step_time": 8.362976551055908
    },
    {
      "epoch": 1.5491485595703125e-05,
      "model_forward_time": 1.7425861358642578,
      "step": 90244
    },
    {
      "epoch": 1.5491485595703125e-05,
      "step": 90244,
      "training_step_time": 7.704572677612305
    },
    {
      "epoch": 1.5492439270019532e-05,
      "model_forward_time": 2.447594165802002,
      "step": 90245
    },
    {
      "epoch": 1.5492439270019532e-05,
      "step": 90245,
      "training_step_time": 7.706385135650635
    },
    {
      "epoch": 1.5493392944335936e-05,
      "model_forward_time": 1.7570257186889648,
      "step": 90246
    },
    {
      "epoch": 1.5493392944335936e-05,
      "step": 90246,
      "training_step_time": 9.312004089355469
    },
    {
      "epoch": 1.5494346618652344e-05,
      "model_forward_time": 1.7621829509735107,
      "step": 90247
    },
    {
      "epoch": 1.5494346618652344e-05,
      "step": 90247,
      "training_step_time": 7.727002859115601
    },
    {
      "epoch": 1.549530029296875e-05,
      "model_forward_time": 1.7413330078125,
      "step": 90248
    },
    {
      "epoch": 1.549530029296875e-05,
      "step": 90248,
      "training_step_time": 7.7809226512908936
    },
    {
      "epoch": 1.5496253967285155e-05,
      "model_forward_time": 1.7339043617248535,
      "step": 90249
    },
    {
      "epoch": 1.5496253967285155e-05,
      "step": 90249,
      "training_step_time": 7.707484722137451
    },
    {
      "epoch": 1.5497207641601562e-05,
      "grad_norm": 0.0848613828420639,
      "learning_rate": 2.577059344058047e-07,
      "loss": 0.0544,
      "step": 90250
    },
    {
      "epoch": 1.5497207641601562e-05,
      "model_forward_time": 1.7458317279815674,
      "step": 90250
    },
    {
      "dynamics_loss_avg": 0.0700118524953723,
      "epoch": 1.5497207641601562e-05,
      "step": 90250
    },
    {
      "action_loss_avg": 0.0017580211628228427,
      "epoch": 1.5497207641601562e-05,
      "step": 90250
    },
    {
      "epoch": 1.5497207641601562e-05,
      "step": 90250,
      "training_step_time": 7.733417987823486
    },
    {
      "epoch": 1.549816131591797e-05,
      "model_forward_time": 1.7314846515655518,
      "step": 90251
    },
    {
      "epoch": 1.549816131591797e-05,
      "step": 90251,
      "training_step_time": 7.725039958953857
    },
    {
      "epoch": 1.5499114990234374e-05,
      "model_forward_time": 1.7387328147888184,
      "step": 90252
    },
    {
      "epoch": 1.5499114990234374e-05,
      "step": 90252,
      "training_step_time": 7.581552743911743
    },
    {
      "epoch": 1.550006866455078e-05,
      "model_forward_time": 1.7362773418426514,
      "step": 90253
    },
    {
      "epoch": 1.550006866455078e-05,
      "step": 90253,
      "training_step_time": 7.67957329750061
    },
    {
      "epoch": 1.550102233886719e-05,
      "model_forward_time": 2.413182497024536,
      "step": 90254
    },
    {
      "epoch": 1.550102233886719e-05,
      "step": 90254,
      "training_step_time": 9.862756967544556
    },
    {
      "epoch": 1.5501976013183593e-05,
      "model_forward_time": 1.7645716667175293,
      "step": 90255
    },
    {
      "epoch": 1.5501976013183593e-05,
      "step": 90255,
      "training_step_time": 8.588566064834595
    },
    {
      "epoch": 1.55029296875e-05,
      "model_forward_time": 1.733128547668457,
      "step": 90256
    },
    {
      "epoch": 1.55029296875e-05,
      "step": 90256,
      "training_step_time": 10.187938451766968
    },
    {
      "epoch": 1.5503883361816408e-05,
      "model_forward_time": 1.735363245010376,
      "step": 90257
    },
    {
      "epoch": 1.5503883361816408e-05,
      "step": 90257,
      "training_step_time": 7.76042103767395
    },
    {
      "epoch": 1.550483703613281e-05,
      "model_forward_time": 1.7386987209320068,
      "step": 90258
    },
    {
      "epoch": 1.550483703613281e-05,
      "step": 90258,
      "training_step_time": 7.652549982070923
    },
    {
      "epoch": 1.550579071044922e-05,
      "model_forward_time": 1.7405579090118408,
      "step": 90259
    },
    {
      "epoch": 1.550579071044922e-05,
      "step": 90259,
      "training_step_time": 7.657470226287842
    },
    {
      "epoch": 1.5506744384765626e-05,
      "grad_norm": 0.0858374685049057,
      "learning_rate": 2.5718220843130414e-07,
      "loss": 0.0556,
      "step": 90260
    },
    {
      "epoch": 1.5506744384765626e-05,
      "model_forward_time": 1.7306792736053467,
      "step": 90260
    },
    {
      "dynamics_loss_avg": 0.06104681007564068,
      "epoch": 1.5506744384765626e-05,
      "step": 90260
    },
    {
      "action_loss_avg": 0.002025885810144246,
      "epoch": 1.5506744384765626e-05,
      "step": 90260
    },
    {
      "epoch": 1.5506744384765626e-05,
      "step": 90260,
      "training_step_time": 8.549912214279175
    },
    {
      "epoch": 1.550769805908203e-05,
      "model_forward_time": 1.744114875793457,
      "step": 90261
    },
    {
      "epoch": 1.550769805908203e-05,
      "step": 90261,
      "training_step_time": 7.5755980014801025
    },
    {
      "epoch": 1.5508651733398438e-05,
      "model_forward_time": 1.738551139831543,
      "step": 90262
    },
    {
      "epoch": 1.5508651733398438e-05,
      "step": 90262,
      "training_step_time": 7.691092252731323
    },
    {
      "epoch": 1.5509605407714845e-05,
      "model_forward_time": 1.7394366264343262,
      "step": 90263
    },
    {
      "epoch": 1.5509605407714845e-05,
      "step": 90263,
      "training_step_time": 8.45755672454834
    },
    {
      "epoch": 1.551055908203125e-05,
      "model_forward_time": 1.7720615863800049,
      "step": 90264
    },
    {
      "epoch": 1.551055908203125e-05,
      "step": 90264,
      "training_step_time": 8.743214845657349
    },
    {
      "epoch": 1.5511512756347657e-05,
      "model_forward_time": 1.7487070560455322,
      "step": 90265
    },
    {
      "epoch": 1.5511512756347657e-05,
      "step": 90265,
      "training_step_time": 9.72181248664856
    },
    {
      "epoch": 1.551246643066406e-05,
      "model_forward_time": 1.7479472160339355,
      "step": 90266
    },
    {
      "epoch": 1.551246643066406e-05,
      "step": 90266,
      "training_step_time": 8.33891773223877
    },
    {
      "epoch": 1.5513420104980468e-05,
      "model_forward_time": 1.7375266551971436,
      "step": 90267
    },
    {
      "epoch": 1.5513420104980468e-05,
      "step": 90267,
      "training_step_time": 7.703770875930786
    },
    {
      "epoch": 1.5514373779296876e-05,
      "model_forward_time": 1.7466771602630615,
      "step": 90268
    },
    {
      "epoch": 1.5514373779296876e-05,
      "step": 90268,
      "training_step_time": 7.66999888420105
    },
    {
      "epoch": 1.551532745361328e-05,
      "model_forward_time": 1.7500433921813965,
      "step": 90269
    },
    {
      "epoch": 1.551532745361328e-05,
      "step": 90269,
      "training_step_time": 8.575648784637451
    },
    {
      "epoch": 1.5516281127929687e-05,
      "grad_norm": 0.19556312263011932,
      "learning_rate": 2.566590011242254e-07,
      "loss": 0.0548,
      "step": 90270
    },
    {
      "epoch": 1.5516281127929687e-05,
      "model_forward_time": 1.7357358932495117,
      "step": 90270
    },
    {
      "dynamics_loss_avg": 0.05351187214255333,
      "epoch": 1.5516281127929687e-05,
      "step": 90270
    },
    {
      "action_loss_avg": 0.0012296308152144775,
      "epoch": 1.5516281127929687e-05,
      "step": 90270
    },
    {
      "epoch": 1.5516281127929687e-05,
      "step": 90270,
      "training_step_time": 8.305111646652222
    },
    {
      "epoch": 1.5517234802246095e-05,
      "model_forward_time": 1.7467150688171387,
      "step": 90271
    },
    {
      "epoch": 1.5517234802246095e-05,
      "step": 90271,
      "training_step_time": 8.429443120956421
    },
    {
      "epoch": 1.55181884765625e-05,
      "model_forward_time": 1.7384071350097656,
      "step": 90272
    },
    {
      "epoch": 1.55181884765625e-05,
      "step": 90272,
      "training_step_time": 7.709444284439087
    },
    {
      "epoch": 1.5519142150878906e-05,
      "model_forward_time": 2.366518497467041,
      "step": 90273
    },
    {
      "epoch": 1.5519142150878906e-05,
      "step": 90273,
      "training_step_time": 9.481122255325317
    },
    {
      "epoch": 1.5520095825195313e-05,
      "model_forward_time": 1.750960350036621,
      "step": 90274
    },
    {
      "epoch": 1.5520095825195313e-05,
      "step": 90274,
      "training_step_time": 8.387136459350586
    },
    {
      "epoch": 1.5521049499511717e-05,
      "model_forward_time": 1.7398405075073242,
      "step": 90275
    },
    {
      "epoch": 1.5521049499511717e-05,
      "step": 90275,
      "training_step_time": 8.435841083526611
    },
    {
      "epoch": 1.5522003173828125e-05,
      "model_forward_time": 1.7409253120422363,
      "step": 90276
    },
    {
      "epoch": 1.5522003173828125e-05,
      "step": 90276,
      "training_step_time": 7.735081434249878
    },
    {
      "epoch": 1.5522956848144532e-05,
      "model_forward_time": 1.736783742904663,
      "step": 90277
    },
    {
      "epoch": 1.5522956848144532e-05,
      "step": 90277,
      "training_step_time": 7.620077848434448
    },
    {
      "epoch": 1.5523910522460936e-05,
      "model_forward_time": 1.7364320755004883,
      "step": 90278
    },
    {
      "epoch": 1.5523910522460936e-05,
      "step": 90278,
      "training_step_time": 7.7433576583862305
    },
    {
      "epoch": 1.5524864196777344e-05,
      "model_forward_time": 1.734933853149414,
      "step": 90279
    },
    {
      "epoch": 1.5524864196777344e-05,
      "step": 90279,
      "training_step_time": 7.680658340454102
    },
    {
      "epoch": 1.552581787109375e-05,
      "grad_norm": 0.07556094229221344,
      "learning_rate": 2.561363125417876e-07,
      "loss": 0.0547,
      "step": 90280
    },
    {
      "epoch": 1.552581787109375e-05,
      "model_forward_time": 1.7372691631317139,
      "step": 90280
    },
    {
      "dynamics_loss_avg": 0.052141948416829106,
      "epoch": 1.552581787109375e-05,
      "step": 90280
    },
    {
      "action_loss_avg": 0.0012728406640235336,
      "epoch": 1.552581787109375e-05,
      "step": 90280
    },
    {
      "epoch": 1.552581787109375e-05,
      "step": 90280,
      "training_step_time": 7.6375367641448975
    },
    {
      "epoch": 1.5526771545410155e-05,
      "model_forward_time": 1.7377703189849854,
      "step": 90281
    },
    {
      "epoch": 1.5526771545410155e-05,
      "step": 90281,
      "training_step_time": 7.749725818634033
    },
    {
      "epoch": 1.5527725219726563e-05,
      "model_forward_time": 2.3662190437316895,
      "step": 90282
    },
    {
      "epoch": 1.5527725219726563e-05,
      "step": 90282,
      "training_step_time": 8.953469038009644
    },
    {
      "epoch": 1.552867889404297e-05,
      "model_forward_time": 1.7372922897338867,
      "step": 90283
    },
    {
      "epoch": 1.552867889404297e-05,
      "step": 90283,
      "training_step_time": 10.112017393112183
    },
    {
      "epoch": 1.5529632568359374e-05,
      "model_forward_time": 1.755788803100586,
      "step": 90284
    },
    {
      "epoch": 1.5529632568359374e-05,
      "step": 90284,
      "training_step_time": 8.67171835899353
    },
    {
      "epoch": 1.553058624267578e-05,
      "model_forward_time": 1.7380638122558594,
      "step": 90285
    },
    {
      "epoch": 1.553058624267578e-05,
      "step": 90285,
      "training_step_time": 7.827286958694458
    },
    {
      "epoch": 1.553153991699219e-05,
      "model_forward_time": 1.7358582019805908,
      "step": 90286
    },
    {
      "epoch": 1.553153991699219e-05,
      "step": 90286,
      "training_step_time": 7.716145277023315
    },
    {
      "epoch": 1.5532493591308593e-05,
      "model_forward_time": 1.736379623413086,
      "step": 90287
    },
    {
      "epoch": 1.5532493591308593e-05,
      "step": 90287,
      "training_step_time": 7.766494989395142
    },
    {
      "epoch": 1.5533447265625e-05,
      "model_forward_time": 1.7444782257080078,
      "step": 90288
    },
    {
      "epoch": 1.5533447265625e-05,
      "step": 90288,
      "training_step_time": 7.709392786026001
    },
    {
      "epoch": 1.5534400939941408e-05,
      "model_forward_time": 1.7343552112579346,
      "step": 90289
    },
    {
      "epoch": 1.5534400939941408e-05,
      "step": 90289,
      "training_step_time": 8.691044092178345
    },
    {
      "epoch": 1.553535461425781e-05,
      "grad_norm": 0.07474185526371002,
      "learning_rate": 2.556141427411496e-07,
      "loss": 0.0558,
      "step": 90290
    },
    {
      "epoch": 1.553535461425781e-05,
      "model_forward_time": 1.7357048988342285,
      "step": 90290
    },
    {
      "dynamics_loss_avg": 0.05741429161280394,
      "epoch": 1.553535461425781e-05,
      "step": 90290
    },
    {
      "action_loss_avg": 0.001073954376624897,
      "epoch": 1.553535461425781e-05,
      "step": 90290
    },
    {
      "epoch": 1.553535461425781e-05,
      "step": 90290,
      "training_step_time": 9.9942626953125
    },
    {
      "epoch": 1.553630828857422e-05,
      "model_forward_time": 2.370183229446411,
      "step": 90291
    },
    {
      "epoch": 1.553630828857422e-05,
      "step": 90291,
      "training_step_time": 9.079253196716309
    },
    {
      "epoch": 1.5537261962890627e-05,
      "model_forward_time": 1.7384908199310303,
      "step": 90292
    },
    {
      "epoch": 1.5537261962890627e-05,
      "step": 90292,
      "training_step_time": 10.165838718414307
    },
    {
      "epoch": 1.553821563720703e-05,
      "model_forward_time": 1.7518067359924316,
      "step": 90293
    },
    {
      "epoch": 1.553821563720703e-05,
      "step": 90293,
      "training_step_time": 8.827276229858398
    },
    {
      "epoch": 1.5539169311523438e-05,
      "model_forward_time": 1.7389190196990967,
      "step": 90294
    },
    {
      "epoch": 1.5539169311523438e-05,
      "step": 90294,
      "training_step_time": 7.820047855377197
    },
    {
      "epoch": 1.5540122985839845e-05,
      "model_forward_time": 1.7325966358184814,
      "step": 90295
    },
    {
      "epoch": 1.5540122985839845e-05,
      "step": 90295,
      "training_step_time": 7.778307199478149
    },
    {
      "epoch": 1.554107666015625e-05,
      "model_forward_time": 1.740401029586792,
      "step": 90296
    },
    {
      "epoch": 1.554107666015625e-05,
      "step": 90296,
      "training_step_time": 7.759899139404297
    },
    {
      "epoch": 1.5542030334472657e-05,
      "model_forward_time": 1.7372522354125977,
      "step": 90297
    },
    {
      "epoch": 1.5542030334472657e-05,
      "step": 90297,
      "training_step_time": 7.683758497238159
    },
    {
      "epoch": 1.554298400878906e-05,
      "model_forward_time": 1.7411506175994873,
      "step": 90298
    },
    {
      "epoch": 1.554298400878906e-05,
      "step": 90298,
      "training_step_time": 7.622382402420044
    },
    {
      "epoch": 1.5543937683105468e-05,
      "model_forward_time": 1.7359950542449951,
      "step": 90299
    },
    {
      "epoch": 1.5543937683105468e-05,
      "step": 90299,
      "training_step_time": 9.346495151519775
    },
    {
      "epoch": 1.5544891357421876e-05,
      "grad_norm": 0.07927580177783966,
      "learning_rate": 2.5509249177941566e-07,
      "loss": 0.0545,
      "step": 90300
    },
    {
      "epoch": 1.5544891357421876e-05,
      "model_forward_time": 2.3688924312591553,
      "step": 90300
    },
    {
      "dynamics_loss_avg": 0.04917690511792898,
      "epoch": 1.5544891357421876e-05,
      "step": 90300
    },
    {
      "action_loss_avg": 0.0012557111156638711,
      "epoch": 1.5544891357421876e-05,
      "step": 90300
    },
    {
      "epoch": 1.5544891357421876e-05,
      "step": 90300,
      "training_step_time": 8.517249584197998
    },
    {
      "epoch": 1.554584503173828e-05,
      "model_forward_time": 1.7436299324035645,
      "step": 90301
    },
    {
      "epoch": 1.554584503173828e-05,
      "step": 90301,
      "training_step_time": 8.294123888015747
    },
    {
      "epoch": 1.5546798706054687e-05,
      "model_forward_time": 1.7552831172943115,
      "step": 90302
    },
    {
      "epoch": 1.5546798706054687e-05,
      "step": 90302,
      "training_step_time": 9.289734125137329
    },
    {
      "epoch": 1.5547752380371095e-05,
      "model_forward_time": 1.7364418506622314,
      "step": 90303
    },
    {
      "epoch": 1.5547752380371095e-05,
      "step": 90303,
      "training_step_time": 8.567214250564575
    },
    {
      "epoch": 1.55487060546875e-05,
      "model_forward_time": 1.7442798614501953,
      "step": 90304
    },
    {
      "epoch": 1.55487060546875e-05,
      "step": 90304,
      "training_step_time": 7.676939010620117
    },
    {
      "epoch": 1.5549659729003906e-05,
      "model_forward_time": 1.7244510650634766,
      "step": 90305
    },
    {
      "epoch": 1.5549659729003906e-05,
      "step": 90305,
      "training_step_time": 7.687339544296265
    },
    {
      "epoch": 1.5550613403320313e-05,
      "model_forward_time": 1.7256615161895752,
      "step": 90306
    },
    {
      "epoch": 1.5550613403320313e-05,
      "step": 90306,
      "training_step_time": 6.818986654281616
    },
    {
      "epoch": 1.5551567077636717e-05,
      "model_forward_time": 1.7367274761199951,
      "step": 90307
    },
    {
      "epoch": 1.5551567077636717e-05,
      "step": 90307,
      "training_step_time": 7.6598851680755615
    },
    {
      "epoch": 1.5552520751953125e-05,
      "model_forward_time": 1.7350530624389648,
      "step": 90308
    },
    {
      "epoch": 1.5552520751953125e-05,
      "step": 90308,
      "training_step_time": 8.726011276245117
    },
    {
      "epoch": 1.5553474426269532e-05,
      "model_forward_time": 2.3641269207000732,
      "step": 90309
    },
    {
      "epoch": 1.5553474426269532e-05,
      "step": 90309,
      "training_step_time": 10.161761999130249
    },
    {
      "epoch": 1.5554428100585936e-05,
      "grad_norm": 0.0812167152762413,
      "learning_rate": 2.5457135971363287e-07,
      "loss": 0.0564,
      "step": 90310
    },
    {
      "epoch": 1.5554428100585936e-05,
      "model_forward_time": 1.3594839572906494,
      "step": 90310
    },
    {
      "dynamics_loss_avg": 0.052139928378164765,
      "epoch": 1.5554428100585936e-05,
      "step": 90310
    },
    {
      "action_loss_avg": 0.0010343798305257224,
      "epoch": 1.5554428100585936e-05,
      "step": 90310
    },
    {
      "epoch": 1.5554428100585936e-05,
      "step": 90310,
      "training_step_time": 8.45499873161316
    },
    {
      "epoch": 1.5555381774902344e-05,
      "model_forward_time": 1.7421941757202148,
      "step": 90311
    },
    {
      "epoch": 1.5555381774902344e-05,
      "step": 90311,
      "training_step_time": 7.636111259460449
    },
    {
      "epoch": 1.555633544921875e-05,
      "model_forward_time": 1.7353029251098633,
      "step": 90312
    },
    {
      "epoch": 1.555633544921875e-05,
      "step": 90312,
      "training_step_time": 7.755703687667847
    },
    {
      "epoch": 1.5557289123535155e-05,
      "model_forward_time": 1.7455317974090576,
      "step": 90313
    },
    {
      "epoch": 1.5557289123535155e-05,
      "step": 90313,
      "training_step_time": 7.7685387134552
    },
    {
      "epoch": 1.5558242797851563e-05,
      "model_forward_time": 1.731046438217163,
      "step": 90314
    },
    {
      "epoch": 1.5558242797851563e-05,
      "step": 90314,
      "training_step_time": 7.576293468475342
    },
    {
      "epoch": 1.555919647216797e-05,
      "model_forward_time": 1.731672763824463,
      "step": 90315
    },
    {
      "epoch": 1.555919647216797e-05,
      "step": 90315,
      "training_step_time": 7.740791082382202
    },
    {
      "epoch": 1.5560150146484374e-05,
      "model_forward_time": 1.7396917343139648,
      "step": 90316
    },
    {
      "epoch": 1.5560150146484374e-05,
      "step": 90316,
      "training_step_time": 7.665349245071411
    },
    {
      "epoch": 1.556110382080078e-05,
      "model_forward_time": 1.7416892051696777,
      "step": 90317
    },
    {
      "epoch": 1.556110382080078e-05,
      "step": 90317,
      "training_step_time": 8.221410512924194
    },
    {
      "epoch": 1.556205749511719e-05,
      "model_forward_time": 1.76106595993042,
      "step": 90318
    },
    {
      "epoch": 1.556205749511719e-05,
      "step": 90318,
      "training_step_time": 8.326813220977783
    },
    {
      "epoch": 1.5563011169433593e-05,
      "model_forward_time": 2.4203920364379883,
      "step": 90319
    },
    {
      "epoch": 1.5563011169433593e-05,
      "step": 90319,
      "training_step_time": 7.773822784423828
    },
    {
      "epoch": 1.556396484375e-05,
      "grad_norm": 0.12086079269647598,
      "learning_rate": 2.54050746600793e-07,
      "loss": 0.0552,
      "step": 90320
    },
    {
      "epoch": 1.556396484375e-05,
      "model_forward_time": 1.7332029342651367,
      "step": 90320
    },
    {
      "dynamics_loss_avg": 0.05467467345297337,
      "epoch": 1.556396484375e-05,
      "step": 90320
    },
    {
      "action_loss_avg": 0.0016348784702131525,
      "epoch": 1.556396484375e-05,
      "step": 90320
    },
    {
      "epoch": 1.556396484375e-05,
      "step": 90320,
      "training_step_time": 9.67967414855957
    },
    {
      "epoch": 1.5564918518066408e-05,
      "model_forward_time": 1.7617838382720947,
      "step": 90321
    },
    {
      "epoch": 1.5564918518066408e-05,
      "step": 90321,
      "training_step_time": 9.249422073364258
    },
    {
      "epoch": 1.5565872192382812e-05,
      "model_forward_time": 1.7363061904907227,
      "step": 90322
    },
    {
      "epoch": 1.5565872192382812e-05,
      "step": 90322,
      "training_step_time": 7.754007816314697
    },
    {
      "epoch": 1.556682586669922e-05,
      "model_forward_time": 1.7349491119384766,
      "step": 90323
    },
    {
      "epoch": 1.556682586669922e-05,
      "step": 90323,
      "training_step_time": 7.798319578170776
    },
    {
      "epoch": 1.5567779541015627e-05,
      "model_forward_time": 1.7595877647399902,
      "step": 90324
    },
    {
      "epoch": 1.5567779541015627e-05,
      "step": 90324,
      "training_step_time": 7.733146905899048
    },
    {
      "epoch": 1.556873321533203e-05,
      "model_forward_time": 1.8368360996246338,
      "step": 90325
    },
    {
      "epoch": 1.556873321533203e-05,
      "step": 90325,
      "training_step_time": 7.483293771743774
    },
    {
      "epoch": 1.5569686889648438e-05,
      "model_forward_time": 1.8540618419647217,
      "step": 90326
    },
    {
      "epoch": 1.5569686889648438e-05,
      "step": 90326,
      "training_step_time": 7.679681301116943
    },
    {
      "epoch": 1.5570640563964845e-05,
      "model_forward_time": 1.833486557006836,
      "step": 90327
    },
    {
      "epoch": 1.5570640563964845e-05,
      "step": 90327,
      "training_step_time": 7.725498199462891
    },
    {
      "epoch": 1.557159423828125e-05,
      "model_forward_time": 2.006636619567871,
      "step": 90328
    },
    {
      "epoch": 1.557159423828125e-05,
      "step": 90328,
      "training_step_time": 8.219620704650879
    },
    {
      "epoch": 1.5572547912597657e-05,
      "model_forward_time": 1.741774082183838,
      "step": 90329
    },
    {
      "epoch": 1.5572547912597657e-05,
      "step": 90329,
      "training_step_time": 8.544921875
    },
    {
      "epoch": 1.557350158691406e-05,
      "grad_norm": 0.09486740082502365,
      "learning_rate": 2.5353065249782647e-07,
      "loss": 0.0551,
      "step": 90330
    },
    {
      "epoch": 1.557350158691406e-05,
      "model_forward_time": 1.7509024143218994,
      "step": 90330
    },
    {
      "dynamics_loss_avg": 0.07220531515777111,
      "epoch": 1.557350158691406e-05,
      "step": 90330
    },
    {
      "action_loss_avg": 0.0014098344021476806,
      "epoch": 1.557350158691406e-05,
      "step": 90330
    },
    {
      "epoch": 1.557350158691406e-05,
      "step": 90330,
      "training_step_time": 9.52939224243164
    },
    {
      "epoch": 1.5574455261230468e-05,
      "model_forward_time": 1.7272963523864746,
      "step": 90331
    },
    {
      "epoch": 1.5574455261230468e-05,
      "step": 90331,
      "training_step_time": 7.772928237915039
    },
    {
      "epoch": 1.5575408935546876e-05,
      "model_forward_time": 1.73667573928833,
      "step": 90332
    },
    {
      "epoch": 1.5575408935546876e-05,
      "step": 90332,
      "training_step_time": 8.84168815612793
    },
    {
      "epoch": 1.557636260986328e-05,
      "model_forward_time": 1.7418315410614014,
      "step": 90333
    },
    {
      "epoch": 1.557636260986328e-05,
      "step": 90333,
      "training_step_time": 7.7035768032073975
    },
    {
      "epoch": 1.5577316284179687e-05,
      "model_forward_time": 1.738663673400879,
      "step": 90334
    },
    {
      "epoch": 1.5577316284179687e-05,
      "step": 90334,
      "training_step_time": 7.3941330909729
    },
    {
      "epoch": 1.5578269958496095e-05,
      "model_forward_time": 1.741044282913208,
      "step": 90335
    },
    {
      "epoch": 1.5578269958496095e-05,
      "step": 90335,
      "training_step_time": 7.59480357170105
    },
    {
      "epoch": 1.55792236328125e-05,
      "model_forward_time": 1.7676825523376465,
      "step": 90336
    },
    {
      "epoch": 1.55792236328125e-05,
      "step": 90336,
      "training_step_time": 8.581169605255127
    },
    {
      "epoch": 1.5580177307128906e-05,
      "model_forward_time": 2.390047311782837,
      "step": 90337
    },
    {
      "epoch": 1.5580177307128906e-05,
      "step": 90337,
      "training_step_time": 8.452086448669434
    },
    {
      "epoch": 1.5581130981445313e-05,
      "model_forward_time": 1.7357642650604248,
      "step": 90338
    },
    {
      "epoch": 1.5581130981445313e-05,
      "step": 90338,
      "training_step_time": 8.505261182785034
    },
    {
      "epoch": 1.5582084655761717e-05,
      "model_forward_time": 1.7561628818511963,
      "step": 90339
    },
    {
      "epoch": 1.5582084655761717e-05,
      "step": 90339,
      "training_step_time": 8.782111406326294
    },
    {
      "epoch": 1.5583038330078125e-05,
      "grad_norm": 0.0805957242846489,
      "learning_rate": 2.530110774616129e-07,
      "loss": 0.0556,
      "step": 90340
    },
    {
      "epoch": 1.5583038330078125e-05,
      "model_forward_time": 1.7316679954528809,
      "step": 90340
    },
    {
      "dynamics_loss_avg": 0.07136127091944218,
      "epoch": 1.5583038330078125e-05,
      "step": 90340
    },
    {
      "action_loss_avg": 0.0014396824582945556,
      "epoch": 1.5583038330078125e-05,
      "step": 90340
    },
    {
      "epoch": 1.5583038330078125e-05,
      "step": 90340,
      "training_step_time": 8.477203607559204
    },
    {
      "epoch": 1.5583992004394532e-05,
      "model_forward_time": 1.7429461479187012,
      "step": 90341
    },
    {
      "epoch": 1.5583992004394532e-05,
      "step": 90341,
      "training_step_time": 8.556962490081787
    },
    {
      "epoch": 1.5584945678710936e-05,
      "model_forward_time": 1.737288475036621,
      "step": 90342
    },
    {
      "epoch": 1.5584945678710936e-05,
      "step": 90342,
      "training_step_time": 7.6789727210998535
    },
    {
      "epoch": 1.5585899353027344e-05,
      "model_forward_time": 1.7402589321136475,
      "step": 90343
    },
    {
      "epoch": 1.5585899353027344e-05,
      "step": 90343,
      "training_step_time": 7.483213424682617
    },
    {
      "epoch": 1.558685302734375e-05,
      "model_forward_time": 1.745006799697876,
      "step": 90344
    },
    {
      "epoch": 1.558685302734375e-05,
      "step": 90344,
      "training_step_time": 7.662217378616333
    },
    {
      "epoch": 1.5587806701660155e-05,
      "model_forward_time": 1.756545066833496,
      "step": 90345
    },
    {
      "epoch": 1.5587806701660155e-05,
      "step": 90345,
      "training_step_time": 9.166090965270996
    },
    {
      "epoch": 1.5588760375976563e-05,
      "model_forward_time": 1.7392299175262451,
      "step": 90346
    },
    {
      "epoch": 1.5588760375976563e-05,
      "step": 90346,
      "training_step_time": 9.486068964004517
    },
    {
      "epoch": 1.558971405029297e-05,
      "model_forward_time": 1.7370383739471436,
      "step": 90347
    },
    {
      "epoch": 1.558971405029297e-05,
      "step": 90347,
      "training_step_time": 7.741786956787109
    },
    {
      "epoch": 1.5590667724609374e-05,
      "model_forward_time": 1.7351794242858887,
      "step": 90348
    },
    {
      "epoch": 1.5590667724609374e-05,
      "step": 90348,
      "training_step_time": 7.824721574783325
    },
    {
      "epoch": 1.559162139892578e-05,
      "model_forward_time": 1.7482385635375977,
      "step": 90349
    },
    {
      "epoch": 1.559162139892578e-05,
      "step": 90349,
      "training_step_time": 7.798431158065796
    },
    {
      "epoch": 1.559257507324219e-05,
      "grad_norm": 0.08430087566375732,
      "learning_rate": 2.5249202154897013e-07,
      "loss": 0.0538,
      "step": 90350
    },
    {
      "epoch": 1.559257507324219e-05,
      "model_forward_time": 1.7283363342285156,
      "step": 90350
    },
    {
      "dynamics_loss_avg": 0.06346650142222643,
      "epoch": 1.559257507324219e-05,
      "step": 90350
    },
    {
      "action_loss_avg": 0.0016547001534490847,
      "epoch": 1.559257507324219e-05,
      "step": 90350
    },
    {
      "epoch": 1.559257507324219e-05,
      "step": 90350,
      "training_step_time": 7.641843557357788
    },
    {
      "epoch": 1.5593528747558593e-05,
      "model_forward_time": 1.7342486381530762,
      "step": 90351
    },
    {
      "epoch": 1.5593528747558593e-05,
      "step": 90351,
      "training_step_time": 9.251331567764282
    },
    {
      "epoch": 1.5594482421875e-05,
      "model_forward_time": 1.7466447353363037,
      "step": 90352
    },
    {
      "epoch": 1.5594482421875e-05,
      "step": 90352,
      "training_step_time": 7.458504676818848
    },
    {
      "epoch": 1.5595436096191408e-05,
      "model_forward_time": 1.7372477054595947,
      "step": 90353
    },
    {
      "epoch": 1.5595436096191408e-05,
      "step": 90353,
      "training_step_time": 7.666494607925415
    },
    {
      "epoch": 1.5596389770507812e-05,
      "model_forward_time": 1.760143518447876,
      "step": 90354
    },
    {
      "epoch": 1.5596389770507812e-05,
      "step": 90354,
      "training_step_time": 7.736219167709351
    },
    {
      "epoch": 1.559734344482422e-05,
      "model_forward_time": 1.7433640956878662,
      "step": 90355
    },
    {
      "epoch": 1.559734344482422e-05,
      "step": 90355,
      "training_step_time": 8.816666603088379
    },
    {
      "epoch": 1.5598297119140627e-05,
      "model_forward_time": 2.40736722946167,
      "step": 90356
    },
    {
      "epoch": 1.5598297119140627e-05,
      "step": 90356,
      "training_step_time": 7.780027389526367
    },
    {
      "epoch": 1.559925079345703e-05,
      "model_forward_time": 1.734025239944458,
      "step": 90357
    },
    {
      "epoch": 1.559925079345703e-05,
      "step": 90357,
      "training_step_time": 7.754400253295898
    },
    {
      "epoch": 1.5600204467773438e-05,
      "model_forward_time": 1.760258674621582,
      "step": 90358
    },
    {
      "epoch": 1.5600204467773438e-05,
      "step": 90358,
      "training_step_time": 9.463558912277222
    },
    {
      "epoch": 1.5601158142089845e-05,
      "model_forward_time": 1.736788034439087,
      "step": 90359
    },
    {
      "epoch": 1.5601158142089845e-05,
      "step": 90359,
      "training_step_time": 7.693999290466309
    },
    {
      "epoch": 1.560211181640625e-05,
      "grad_norm": 0.07809704542160034,
      "learning_rate": 2.519734848166627e-07,
      "loss": 0.0546,
      "step": 90360
    },
    {
      "epoch": 1.560211181640625e-05,
      "model_forward_time": 1.7327697277069092,
      "step": 90360
    },
    {
      "dynamics_loss_avg": 0.06868164874613285,
      "epoch": 1.560211181640625e-05,
      "step": 90360
    },
    {
      "action_loss_avg": 0.0020735313330078496,
      "epoch": 1.560211181640625e-05,
      "step": 90360
    },
    {
      "epoch": 1.560211181640625e-05,
      "step": 90360,
      "training_step_time": 7.660550832748413
    },
    {
      "epoch": 1.5603065490722657e-05,
      "model_forward_time": 1.7401788234710693,
      "step": 90361
    },
    {
      "epoch": 1.5603065490722657e-05,
      "step": 90361,
      "training_step_time": 7.6993489265441895
    },
    {
      "epoch": 1.560401916503906e-05,
      "model_forward_time": 1.731189489364624,
      "step": 90362
    },
    {
      "epoch": 1.560401916503906e-05,
      "step": 90362,
      "training_step_time": 8.871752262115479
    },
    {
      "epoch": 1.560497283935547e-05,
      "model_forward_time": 1.7612838745117188,
      "step": 90363
    },
    {
      "epoch": 1.560497283935547e-05,
      "step": 90363,
      "training_step_time": 8.97490930557251
    },
    {
      "epoch": 1.5605926513671876e-05,
      "model_forward_time": 1.7372546195983887,
      "step": 90364
    },
    {
      "epoch": 1.5605926513671876e-05,
      "step": 90364,
      "training_step_time": 7.66857385635376
    },
    {
      "epoch": 1.560688018798828e-05,
      "model_forward_time": 2.403196334838867,
      "step": 90365
    },
    {
      "epoch": 1.560688018798828e-05,
      "step": 90365,
      "training_step_time": 7.759670734405518
    },
    {
      "epoch": 1.5607833862304687e-05,
      "model_forward_time": 1.7356269359588623,
      "step": 90366
    },
    {
      "epoch": 1.5607833862304687e-05,
      "step": 90366,
      "training_step_time": 8.493398904800415
    },
    {
      "epoch": 1.5608787536621095e-05,
      "model_forward_time": 1.754856824874878,
      "step": 90367
    },
    {
      "epoch": 1.5608787536621095e-05,
      "step": 90367,
      "training_step_time": 8.637401342391968
    },
    {
      "epoch": 1.56097412109375e-05,
      "model_forward_time": 1.7331092357635498,
      "step": 90368
    },
    {
      "epoch": 1.56097412109375e-05,
      "step": 90368,
      "training_step_time": 7.734044551849365
    },
    {
      "epoch": 1.5610694885253906e-05,
      "model_forward_time": 1.7453501224517822,
      "step": 90369
    },
    {
      "epoch": 1.5610694885253906e-05,
      "step": 90369,
      "training_step_time": 8.831214189529419
    },
    {
      "epoch": 1.5611648559570313e-05,
      "grad_norm": 0.061068687587976456,
      "learning_rate": 2.514554673213948e-07,
      "loss": 0.0531,
      "step": 90370
    },
    {
      "epoch": 1.5611648559570313e-05,
      "model_forward_time": 1.731996774673462,
      "step": 90370
    },
    {
      "dynamics_loss_avg": 0.04271697690710426,
      "epoch": 1.5611648559570313e-05,
      "step": 90370
    },
    {
      "action_loss_avg": 0.0011369095998816192,
      "epoch": 1.5611648559570313e-05,
      "step": 90370
    },
    {
      "epoch": 1.5611648559570313e-05,
      "step": 90370,
      "training_step_time": 7.629056453704834
    },
    {
      "epoch": 1.5612602233886717e-05,
      "model_forward_time": 1.7243220806121826,
      "step": 90371
    },
    {
      "epoch": 1.5612602233886717e-05,
      "step": 90371,
      "training_step_time": 8.38133192062378
    },
    {
      "epoch": 1.5613555908203125e-05,
      "model_forward_time": 1.7638425827026367,
      "step": 90372
    },
    {
      "epoch": 1.5613555908203125e-05,
      "step": 90372,
      "training_step_time": 8.575331926345825
    },
    {
      "epoch": 1.5614509582519532e-05,
      "model_forward_time": 1.7376248836517334,
      "step": 90373
    },
    {
      "epoch": 1.5614509582519532e-05,
      "step": 90373,
      "training_step_time": 7.665825605392456
    },
    {
      "epoch": 1.5615463256835936e-05,
      "model_forward_time": 2.377686023712158,
      "step": 90374
    },
    {
      "epoch": 1.5615463256835936e-05,
      "step": 90374,
      "training_step_time": 8.21393895149231
    },
    {
      "epoch": 1.5616416931152344e-05,
      "model_forward_time": 1.7294621467590332,
      "step": 90375
    },
    {
      "epoch": 1.5616416931152344e-05,
      "step": 90375,
      "training_step_time": 7.667925834655762
    },
    {
      "epoch": 1.561737060546875e-05,
      "model_forward_time": 1.7474470138549805,
      "step": 90376
    },
    {
      "epoch": 1.561737060546875e-05,
      "step": 90376,
      "training_step_time": 8.384774446487427
    },
    {
      "epoch": 1.5618324279785155e-05,
      "model_forward_time": 1.7319011688232422,
      "step": 90377
    },
    {
      "epoch": 1.5618324279785155e-05,
      "step": 90377,
      "training_step_time": 7.785891532897949
    },
    {
      "epoch": 1.5619277954101563e-05,
      "model_forward_time": 1.7178926467895508,
      "step": 90378
    },
    {
      "epoch": 1.5619277954101563e-05,
      "step": 90378,
      "training_step_time": 7.7176854610443115
    },
    {
      "epoch": 1.562023162841797e-05,
      "model_forward_time": 1.7386589050292969,
      "step": 90379
    },
    {
      "epoch": 1.562023162841797e-05,
      "step": 90379,
      "training_step_time": 7.465421199798584
    },
    {
      "epoch": 1.5621185302734374e-05,
      "grad_norm": 0.07469060271978378,
      "learning_rate": 2.5093796911981936e-07,
      "loss": 0.0547,
      "step": 90380
    },
    {
      "epoch": 1.5621185302734374e-05,
      "model_forward_time": 1.7285373210906982,
      "step": 90380
    },
    {
      "dynamics_loss_avg": 0.07406840361654758,
      "epoch": 1.5621185302734374e-05,
      "step": 90380
    },
    {
      "action_loss_avg": 0.00154494111193344,
      "epoch": 1.5621185302734374e-05,
      "step": 90380
    },
    {
      "epoch": 1.5621185302734374e-05,
      "step": 90380,
      "training_step_time": 7.670618534088135
    },
    {
      "epoch": 1.562213897705078e-05,
      "model_forward_time": 1.7704827785491943,
      "step": 90381
    },
    {
      "epoch": 1.562213897705078e-05,
      "step": 90381,
      "training_step_time": 7.789102077484131
    },
    {
      "epoch": 1.562309265136719e-05,
      "model_forward_time": 1.733996868133545,
      "step": 90382
    },
    {
      "epoch": 1.562309265136719e-05,
      "step": 90382,
      "training_step_time": 7.767537593841553
    },
    {
      "epoch": 1.5624046325683593e-05,
      "model_forward_time": 2.409492015838623,
      "step": 90383
    },
    {
      "epoch": 1.5624046325683593e-05,
      "step": 90383,
      "training_step_time": 8.554982900619507
    },
    {
      "epoch": 1.5625e-05,
      "model_forward_time": 1.7429702281951904,
      "step": 90384
    },
    {
      "epoch": 1.5625e-05,
      "step": 90384,
      "training_step_time": 8.410172700881958
    },
    {
      "epoch": 1.5625953674316408e-05,
      "model_forward_time": 1.7396740913391113,
      "step": 90385
    },
    {
      "epoch": 1.5625953674316408e-05,
      "step": 90385,
      "training_step_time": 8.78628659248352
    },
    {
      "epoch": 1.5626907348632812e-05,
      "model_forward_time": 1.7326881885528564,
      "step": 90386
    },
    {
      "epoch": 1.5626907348632812e-05,
      "step": 90386,
      "training_step_time": 7.769511461257935
    },
    {
      "epoch": 1.562786102294922e-05,
      "model_forward_time": 1.7350051403045654,
      "step": 90387
    },
    {
      "epoch": 1.562786102294922e-05,
      "step": 90387,
      "training_step_time": 7.778076887130737
    },
    {
      "epoch": 1.5628814697265627e-05,
      "model_forward_time": 1.7356069087982178,
      "step": 90388
    },
    {
      "epoch": 1.5628814697265627e-05,
      "step": 90388,
      "training_step_time": 7.728976011276245
    },
    {
      "epoch": 1.562976837158203e-05,
      "model_forward_time": 1.8067524433135986,
      "step": 90389
    },
    {
      "epoch": 1.562976837158203e-05,
      "step": 90389,
      "training_step_time": 7.795031785964966
    },
    {
      "epoch": 1.5630722045898438e-05,
      "grad_norm": 0.15025493502616882,
      "learning_rate": 2.5042099026852674e-07,
      "loss": 0.0542,
      "step": 90390
    },
    {
      "epoch": 1.5630722045898438e-05,
      "model_forward_time": 1.8048996925354004,
      "step": 90390
    },
    {
      "dynamics_loss_avg": 0.06824018079787493,
      "epoch": 1.5630722045898438e-05,
      "step": 90390
    },
    {
      "action_loss_avg": 0.0013838453392963856,
      "epoch": 1.5630722045898438e-05,
      "step": 90390
    },
    {
      "epoch": 1.5630722045898438e-05,
      "step": 90390,
      "training_step_time": 7.750720500946045
    },
    {
      "epoch": 1.5631675720214842e-05,
      "model_forward_time": 1.8089275360107422,
      "step": 90391
    },
    {
      "epoch": 1.5631675720214842e-05,
      "step": 90391,
      "training_step_time": 8.5324387550354
    },
    {
      "epoch": 1.563262939453125e-05,
      "model_forward_time": 2.897756338119507,
      "step": 90392
    },
    {
      "epoch": 1.563262939453125e-05,
      "step": 90392,
      "training_step_time": 10.343549013137817
    },
    {
      "epoch": 1.5633583068847657e-05,
      "model_forward_time": 1.8380765914916992,
      "step": 90393
    },
    {
      "epoch": 1.5633583068847657e-05,
      "step": 90393,
      "training_step_time": 8.283080577850342
    },
    {
      "epoch": 1.563453674316406e-05,
      "model_forward_time": 1.836064338684082,
      "step": 90394
    },
    {
      "epoch": 1.563453674316406e-05,
      "step": 90394,
      "training_step_time": 7.762817144393921
    },
    {
      "epoch": 1.563549041748047e-05,
      "model_forward_time": 1.7969563007354736,
      "step": 90395
    },
    {
      "epoch": 1.563549041748047e-05,
      "step": 90395,
      "training_step_time": 8.494458436965942
    },
    {
      "epoch": 1.5636444091796876e-05,
      "model_forward_time": 1.7992839813232422,
      "step": 90396
    },
    {
      "epoch": 1.5636444091796876e-05,
      "step": 90396,
      "training_step_time": 7.762061834335327
    },
    {
      "epoch": 1.563739776611328e-05,
      "model_forward_time": 1.8319532871246338,
      "step": 90397
    },
    {
      "epoch": 1.563739776611328e-05,
      "step": 90397,
      "training_step_time": 7.707802057266235
    },
    {
      "epoch": 1.5638351440429687e-05,
      "model_forward_time": 1.8299758434295654,
      "step": 90398
    },
    {
      "epoch": 1.5638351440429687e-05,
      "step": 90398,
      "training_step_time": 7.710668563842773
    },
    {
      "epoch": 1.5639305114746095e-05,
      "model_forward_time": 1.8009815216064453,
      "step": 90399
    },
    {
      "epoch": 1.5639305114746095e-05,
      "step": 90399,
      "training_step_time": 7.708069086074829
    },
    {
      "epoch": 1.56402587890625e-05,
      "grad_norm": 0.08750978112220764,
      "learning_rate": 2.4990453082405443e-07,
      "loss": 0.0528,
      "step": 90400
    },
    {
      "epoch": 1.56402587890625e-05,
      "model_forward_time": 1.8409266471862793,
      "step": 90400
    },
    {
      "dynamics_loss_avg": 0.056063847802579406,
      "epoch": 1.56402587890625e-05,
      "step": 90400
    },
    {
      "action_loss_avg": 0.0010552459949394687,
      "epoch": 1.56402587890625e-05,
      "step": 90400
    },
    {
      "epoch": 1.56402587890625e-05,
      "step": 90400,
      "training_step_time": 7.723648309707642
    },
    {
      "epoch": 1.5641212463378906e-05,
      "model_forward_time": 1.8294639587402344,
      "step": 90401
    },
    {
      "epoch": 1.5641212463378906e-05,
      "step": 90401,
      "training_step_time": 9.24215817451477
    },
    {
      "epoch": 1.5642166137695313e-05,
      "model_forward_time": 1.8748993873596191,
      "step": 90402
    },
    {
      "epoch": 1.5642166137695313e-05,
      "step": 90402,
      "training_step_time": 7.763389825820923
    },
    {
      "epoch": 1.5643119812011718e-05,
      "model_forward_time": 1.8137469291687012,
      "step": 90403
    },
    {
      "epoch": 1.5643119812011718e-05,
      "step": 90403,
      "training_step_time": 8.624918222427368
    },
    {
      "epoch": 1.5644073486328125e-05,
      "model_forward_time": 1.7766811847686768,
      "step": 90404
    },
    {
      "epoch": 1.5644073486328125e-05,
      "step": 90404,
      "training_step_time": 8.332833528518677
    },
    {
      "epoch": 1.5645027160644532e-05,
      "model_forward_time": 1.455289363861084,
      "step": 90405
    },
    {
      "epoch": 1.5645027160644532e-05,
      "step": 90405,
      "training_step_time": 7.832884073257446
    },
    {
      "epoch": 1.5645980834960936e-05,
      "model_forward_time": 1.8093149662017822,
      "step": 90406
    },
    {
      "epoch": 1.5645980834960936e-05,
      "step": 90406,
      "training_step_time": 7.81427264213562
    },
    {
      "epoch": 1.5646934509277344e-05,
      "model_forward_time": 1.8172409534454346,
      "step": 90407
    },
    {
      "epoch": 1.5646934509277344e-05,
      "step": 90407,
      "training_step_time": 7.766986608505249
    },
    {
      "epoch": 1.564788818359375e-05,
      "model_forward_time": 1.8103368282318115,
      "step": 90408
    },
    {
      "epoch": 1.564788818359375e-05,
      "step": 90408,
      "training_step_time": 7.794471263885498
    },
    {
      "epoch": 1.5648841857910155e-05,
      "model_forward_time": 1.8185248374938965,
      "step": 90409
    },
    {
      "epoch": 1.5648841857910155e-05,
      "step": 90409,
      "training_step_time": 7.763117790222168
    },
    {
      "epoch": 1.5649795532226563e-05,
      "grad_norm": 0.08910168707370758,
      "learning_rate": 2.4938859084288e-07,
      "loss": 0.0533,
      "step": 90410
    },
    {
      "epoch": 1.5649795532226563e-05,
      "model_forward_time": 1.7835826873779297,
      "step": 90410
    },
    {
      "dynamics_loss_avg": 0.0602365618571639,
      "epoch": 1.5649795532226563e-05,
      "step": 90410
    },
    {
      "action_loss_avg": 0.0019767309713643044,
      "epoch": 1.5649795532226563e-05,
      "step": 90410
    },
    {
      "epoch": 1.5649795532226563e-05,
      "step": 90410,
      "training_step_time": 8.269158601760864
    },
    {
      "epoch": 1.565074920654297e-05,
      "model_forward_time": 2.9058849811553955,
      "step": 90411
    },
    {
      "epoch": 1.565074920654297e-05,
      "step": 90411,
      "training_step_time": 9.465319871902466
    },
    {
      "epoch": 1.5651702880859374e-05,
      "model_forward_time": 1.79280424118042,
      "step": 90412
    },
    {
      "epoch": 1.5651702880859374e-05,
      "step": 90412,
      "training_step_time": 10.2965567111969
    },
    {
      "epoch": 1.565265655517578e-05,
      "model_forward_time": 1.8023483753204346,
      "step": 90413
    },
    {
      "epoch": 1.565265655517578e-05,
      "step": 90413,
      "training_step_time": 8.365610837936401
    },
    {
      "epoch": 1.565361022949219e-05,
      "model_forward_time": 1.7944154739379883,
      "step": 90414
    },
    {
      "epoch": 1.565361022949219e-05,
      "step": 90414,
      "training_step_time": 7.812962055206299
    },
    {
      "epoch": 1.5654563903808593e-05,
      "model_forward_time": 1.8358933925628662,
      "step": 90415
    },
    {
      "epoch": 1.5654563903808593e-05,
      "step": 90415,
      "training_step_time": 7.8363142013549805
    },
    {
      "epoch": 1.5655517578125e-05,
      "model_forward_time": 1.8186795711517334,
      "step": 90416
    },
    {
      "epoch": 1.5655517578125e-05,
      "step": 90416,
      "training_step_time": 7.690929174423218
    },
    {
      "epoch": 1.5656471252441408e-05,
      "model_forward_time": 1.786468505859375,
      "step": 90417
    },
    {
      "epoch": 1.5656471252441408e-05,
      "step": 90417,
      "training_step_time": 8.868087768554688
    },
    {
      "epoch": 1.5657424926757812e-05,
      "model_forward_time": 1.836540699005127,
      "step": 90418
    },
    {
      "epoch": 1.5657424926757812e-05,
      "step": 90418,
      "training_step_time": 7.68534255027771
    },
    {
      "epoch": 1.565837860107422e-05,
      "model_forward_time": 1.8189921379089355,
      "step": 90419
    },
    {
      "epoch": 1.565837860107422e-05,
      "step": 90419,
      "training_step_time": 7.72301459312439
    },
    {
      "epoch": 1.5659332275390627e-05,
      "grad_norm": 0.09466470777988434,
      "learning_rate": 2.4887317038142734e-07,
      "loss": 0.0538,
      "step": 90420
    },
    {
      "epoch": 1.5659332275390627e-05,
      "model_forward_time": 2.9354114532470703,
      "step": 90420
    },
    {
      "dynamics_loss_avg": 0.05019154977053404,
      "epoch": 1.5659332275390627e-05,
      "step": 90420
    },
    {
      "action_loss_avg": 0.0013715760200284422,
      "epoch": 1.5659332275390627e-05,
      "step": 90420
    },
    {
      "epoch": 1.5659332275390627e-05,
      "step": 90420,
      "training_step_time": 7.803857326507568
    },
    {
      "epoch": 1.566028594970703e-05,
      "model_forward_time": 1.8359529972076416,
      "step": 90421
    },
    {
      "epoch": 1.566028594970703e-05,
      "step": 90421,
      "training_step_time": 7.686111211776733
    },
    {
      "epoch": 1.5661239624023438e-05,
      "model_forward_time": 1.803039789199829,
      "step": 90422
    },
    {
      "epoch": 1.5661239624023438e-05,
      "step": 90422,
      "training_step_time": 8.3112154006958
    },
    {
      "epoch": 1.5662193298339842e-05,
      "model_forward_time": 1.8004379272460938,
      "step": 90423
    },
    {
      "epoch": 1.5662193298339842e-05,
      "step": 90423,
      "training_step_time": 8.998966932296753
    },
    {
      "epoch": 1.566314697265625e-05,
      "model_forward_time": 1.8371052742004395,
      "step": 90424
    },
    {
      "epoch": 1.566314697265625e-05,
      "step": 90424,
      "training_step_time": 8.783540725708008
    },
    {
      "epoch": 1.5664100646972657e-05,
      "model_forward_time": 1.8173635005950928,
      "step": 90425
    },
    {
      "epoch": 1.5664100646972657e-05,
      "step": 90425,
      "training_step_time": 7.439017295837402
    },
    {
      "epoch": 1.566505432128906e-05,
      "model_forward_time": 1.793626070022583,
      "step": 90426
    },
    {
      "epoch": 1.566505432128906e-05,
      "step": 90426,
      "training_step_time": 7.699187517166138
    },
    {
      "epoch": 1.566600799560547e-05,
      "model_forward_time": 1.81656813621521,
      "step": 90427
    },
    {
      "epoch": 1.566600799560547e-05,
      "step": 90427,
      "training_step_time": 7.651329755783081
    },
    {
      "epoch": 1.5666961669921876e-05,
      "model_forward_time": 1.8264002799987793,
      "step": 90428
    },
    {
      "epoch": 1.5666961669921876e-05,
      "step": 90428,
      "training_step_time": 7.787554025650024
    },
    {
      "epoch": 1.566791534423828e-05,
      "model_forward_time": 2.8579790592193604,
      "step": 90429
    },
    {
      "epoch": 1.566791534423828e-05,
      "step": 90429,
      "training_step_time": 8.950047254562378
    },
    {
      "epoch": 1.5668869018554687e-05,
      "grad_norm": 0.1393132209777832,
      "learning_rate": 2.4835826949606125e-07,
      "loss": 0.0556,
      "step": 90430
    },
    {
      "epoch": 1.5668869018554687e-05,
      "model_forward_time": 1.8072965145111084,
      "step": 90430
    },
    {
      "dynamics_loss_avg": 0.06326689589768648,
      "epoch": 1.5668869018554687e-05,
      "step": 90430
    },
    {
      "action_loss_avg": 0.0022600221447646616,
      "epoch": 1.5668869018554687e-05,
      "step": 90430
    },
    {
      "epoch": 1.5668869018554687e-05,
      "step": 90430,
      "training_step_time": 7.677074909210205
    },
    {
      "epoch": 1.5669822692871095e-05,
      "model_forward_time": 1.8066084384918213,
      "step": 90431
    },
    {
      "epoch": 1.5669822692871095e-05,
      "step": 90431,
      "training_step_time": 8.791726350784302
    },
    {
      "epoch": 1.56707763671875e-05,
      "model_forward_time": 1.821237564086914,
      "step": 90432
    },
    {
      "epoch": 1.56707763671875e-05,
      "step": 90432,
      "training_step_time": 7.778396368026733
    },
    {
      "epoch": 1.5671730041503906e-05,
      "model_forward_time": 1.8287272453308105,
      "step": 90433
    },
    {
      "epoch": 1.5671730041503906e-05,
      "step": 90433,
      "training_step_time": 7.794198751449585
    },
    {
      "epoch": 1.5672683715820314e-05,
      "model_forward_time": 1.8123812675476074,
      "step": 90434
    },
    {
      "epoch": 1.5672683715820314e-05,
      "step": 90434,
      "training_step_time": 7.7574732303619385
    },
    {
      "epoch": 1.5673637390136718e-05,
      "model_forward_time": 1.7984533309936523,
      "step": 90435
    },
    {
      "epoch": 1.5673637390136718e-05,
      "step": 90435,
      "training_step_time": 8.288322448730469
    },
    {
      "epoch": 1.5674591064453125e-05,
      "model_forward_time": 1.8399975299835205,
      "step": 90436
    },
    {
      "epoch": 1.5674591064453125e-05,
      "step": 90436,
      "training_step_time": 8.172165870666504
    },
    {
      "epoch": 1.5675544738769532e-05,
      "model_forward_time": 1.8098981380462646,
      "step": 90437
    },
    {
      "epoch": 1.5675544738769532e-05,
      "step": 90437,
      "training_step_time": 9.2598135471344
    },
    {
      "epoch": 1.5676498413085936e-05,
      "model_forward_time": 2.869986057281494,
      "step": 90438
    },
    {
      "epoch": 1.5676498413085936e-05,
      "step": 90438,
      "training_step_time": 8.460124731063843
    },
    {
      "epoch": 1.5677452087402344e-05,
      "model_forward_time": 1.8891756534576416,
      "step": 90439
    },
    {
      "epoch": 1.5677452087402344e-05,
      "step": 90439,
      "training_step_time": 7.667641639709473
    },
    {
      "epoch": 1.567840576171875e-05,
      "grad_norm": 0.07858763635158539,
      "learning_rate": 2.478438882430917e-07,
      "loss": 0.0543,
      "step": 90440
    },
    {
      "epoch": 1.567840576171875e-05,
      "model_forward_time": 1.8097407817840576,
      "step": 90440
    },
    {
      "dynamics_loss_avg": 0.04939244724810123,
      "epoch": 1.567840576171875e-05,
      "step": 90440
    },
    {
      "action_loss_avg": 0.0014245372614823282,
      "epoch": 1.567840576171875e-05,
      "step": 90440
    },
    {
      "epoch": 1.567840576171875e-05,
      "step": 90440,
      "training_step_time": 10.171658754348755
    },
    {
      "epoch": 1.5679359436035155e-05,
      "model_forward_time": 1.8201723098754883,
      "step": 90441
    },
    {
      "epoch": 1.5679359436035155e-05,
      "step": 90441,
      "training_step_time": 8.357625007629395
    },
    {
      "epoch": 1.5680313110351563e-05,
      "model_forward_time": 1.789524793624878,
      "step": 90442
    },
    {
      "epoch": 1.5680313110351563e-05,
      "step": 90442,
      "training_step_time": 7.783108949661255
    },
    {
      "epoch": 1.568126678466797e-05,
      "model_forward_time": 1.8111693859100342,
      "step": 90443
    },
    {
      "epoch": 1.568126678466797e-05,
      "step": 90443,
      "training_step_time": 8.734290599822998
    },
    {
      "epoch": 1.5682220458984374e-05,
      "model_forward_time": 1.8213145732879639,
      "step": 90444
    },
    {
      "epoch": 1.5682220458984374e-05,
      "step": 90444,
      "training_step_time": 7.445988416671753
    },
    {
      "epoch": 1.568317413330078e-05,
      "model_forward_time": 1.7771022319793701,
      "step": 90445
    },
    {
      "epoch": 1.568317413330078e-05,
      "step": 90445,
      "training_step_time": 8.766011238098145
    },
    {
      "epoch": 1.568412780761719e-05,
      "model_forward_time": 1.826897144317627,
      "step": 90446
    },
    {
      "epoch": 1.568412780761719e-05,
      "step": 90446,
      "training_step_time": 7.682762384414673
    },
    {
      "epoch": 1.5685081481933593e-05,
      "model_forward_time": 1.817094087600708,
      "step": 90447
    },
    {
      "epoch": 1.5685081481933593e-05,
      "step": 90447,
      "training_step_time": 8.433871746063232
    },
    {
      "epoch": 1.568603515625e-05,
      "model_forward_time": 1.8806233406066895,
      "step": 90448
    },
    {
      "epoch": 1.568603515625e-05,
      "step": 90448,
      "training_step_time": 9.849657773971558
    },
    {
      "epoch": 1.5686988830566408e-05,
      "model_forward_time": 1.8106334209442139,
      "step": 90449
    },
    {
      "epoch": 1.5686988830566408e-05,
      "step": 90449,
      "training_step_time": 9.304134607315063
    },
    {
      "epoch": 1.5687942504882812e-05,
      "grad_norm": 0.17618179321289062,
      "learning_rate": 2.473300266787687e-07,
      "loss": 0.0548,
      "step": 90450
    },
    {
      "epoch": 1.5687942504882812e-05,
      "model_forward_time": 1.8411505222320557,
      "step": 90450
    },
    {
      "dynamics_loss_avg": 0.05265172328799963,
      "epoch": 1.5687942504882812e-05,
      "step": 90450
    },
    {
      "action_loss_avg": 0.0008791634434601292,
      "epoch": 1.5687942504882812e-05,
      "step": 90450
    },
    {
      "epoch": 1.5687942504882812e-05,
      "step": 90450,
      "training_step_time": 8.6807701587677
    },
    {
      "epoch": 1.568889617919922e-05,
      "model_forward_time": 1.8194701671600342,
      "step": 90451
    },
    {
      "epoch": 1.568889617919922e-05,
      "step": 90451,
      "training_step_time": 8.266622066497803
    },
    {
      "epoch": 1.5689849853515627e-05,
      "model_forward_time": 1.8280491828918457,
      "step": 90452
    },
    {
      "epoch": 1.5689849853515627e-05,
      "step": 90452,
      "training_step_time": 7.711427927017212
    },
    {
      "epoch": 1.569080352783203e-05,
      "model_forward_time": 1.8352115154266357,
      "step": 90453
    },
    {
      "epoch": 1.569080352783203e-05,
      "step": 90453,
      "training_step_time": 7.426974534988403
    },
    {
      "epoch": 1.5691757202148438e-05,
      "model_forward_time": 1.8362300395965576,
      "step": 90454
    },
    {
      "epoch": 1.5691757202148438e-05,
      "step": 90454,
      "training_step_time": 7.764207601547241
    },
    {
      "epoch": 1.5692710876464842e-05,
      "model_forward_time": 1.7949140071868896,
      "step": 90455
    },
    {
      "epoch": 1.5692710876464842e-05,
      "step": 90455,
      "training_step_time": 7.613750457763672
    },
    {
      "epoch": 1.569366455078125e-05,
      "model_forward_time": 1.8310673236846924,
      "step": 90456
    },
    {
      "epoch": 1.569366455078125e-05,
      "step": 90456,
      "training_step_time": 7.810999870300293
    },
    {
      "epoch": 1.5694618225097657e-05,
      "model_forward_time": 2.9343936443328857,
      "step": 90457
    },
    {
      "epoch": 1.5694618225097657e-05,
      "step": 90457,
      "training_step_time": 7.822726488113403
    },
    {
      "epoch": 1.569557189941406e-05,
      "model_forward_time": 1.7906229496002197,
      "step": 90458
    },
    {
      "epoch": 1.569557189941406e-05,
      "step": 90458,
      "training_step_time": 8.358084201812744
    },
    {
      "epoch": 1.569652557373047e-05,
      "model_forward_time": 1.8436641693115234,
      "step": 90459
    },
    {
      "epoch": 1.569652557373047e-05,
      "step": 90459,
      "training_step_time": 8.349981307983398
    },
    {
      "epoch": 1.5697479248046876e-05,
      "grad_norm": 0.07737426459789276,
      "learning_rate": 2.46816684859289e-07,
      "loss": 0.0554,
      "step": 90460
    },
    {
      "epoch": 1.5697479248046876e-05,
      "model_forward_time": 1.8415701389312744,
      "step": 90460
    },
    {
      "dynamics_loss_avg": 0.060306557919830084,
      "epoch": 1.5697479248046876e-05,
      "step": 90460
    },
    {
      "action_loss_avg": 0.0012326834723353386,
      "epoch": 1.5697479248046876e-05,
      "step": 90460
    },
    {
      "epoch": 1.5697479248046876e-05,
      "step": 90460,
      "training_step_time": 7.755475044250488
    },
    {
      "epoch": 1.569843292236328e-05,
      "model_forward_time": 1.788520097732544,
      "step": 90461
    },
    {
      "epoch": 1.569843292236328e-05,
      "step": 90461,
      "training_step_time": 8.82983136177063
    },
    {
      "epoch": 1.5699386596679687e-05,
      "model_forward_time": 1.845757246017456,
      "step": 90462
    },
    {
      "epoch": 1.5699386596679687e-05,
      "step": 90462,
      "training_step_time": 7.744069576263428
    },
    {
      "epoch": 1.5700340270996095e-05,
      "model_forward_time": 1.8165576457977295,
      "step": 90463
    },
    {
      "epoch": 1.5700340270996095e-05,
      "step": 90463,
      "training_step_time": 7.728543043136597
    },
    {
      "epoch": 1.57012939453125e-05,
      "model_forward_time": 1.8015575408935547,
      "step": 90464
    },
    {
      "epoch": 1.57012939453125e-05,
      "step": 90464,
      "training_step_time": 7.67177152633667
    },
    {
      "epoch": 1.5702247619628906e-05,
      "model_forward_time": 1.855193853378296,
      "step": 90465
    },
    {
      "epoch": 1.5702247619628906e-05,
      "step": 90465,
      "training_step_time": 8.325336933135986
    },
    {
      "epoch": 1.5703201293945314e-05,
      "model_forward_time": 2.9601852893829346,
      "step": 90466
    },
    {
      "epoch": 1.5703201293945314e-05,
      "step": 90466,
      "training_step_time": 7.839010238647461
    },
    {
      "epoch": 1.5704154968261718e-05,
      "model_forward_time": 1.7894773483276367,
      "step": 90467
    },
    {
      "epoch": 1.5704154968261718e-05,
      "step": 90467,
      "training_step_time": 7.702733278274536
    },
    {
      "epoch": 1.5705108642578125e-05,
      "model_forward_time": 1.8555068969726562,
      "step": 90468
    },
    {
      "epoch": 1.5705108642578125e-05,
      "step": 90468,
      "training_step_time": 8.766617774963379
    },
    {
      "epoch": 1.5706062316894532e-05,
      "model_forward_time": 1.8436741828918457,
      "step": 90469
    },
    {
      "epoch": 1.5706062316894532e-05,
      "step": 90469,
      "training_step_time": 7.797569990158081
    },
    {
      "epoch": 1.5707015991210936e-05,
      "grad_norm": 0.11833741515874863,
      "learning_rate": 2.4630386284079043e-07,
      "loss": 0.0565,
      "step": 90470
    },
    {
      "epoch": 1.5707015991210936e-05,
      "model_forward_time": 1.8124377727508545,
      "step": 90470
    },
    {
      "dynamics_loss_avg": 0.06150047145783901,
      "epoch": 1.5707015991210936e-05,
      "step": 90470
    },
    {
      "action_loss_avg": 0.0014281219278927892,
      "epoch": 1.5707015991210936e-05,
      "step": 90470
    },
    {
      "epoch": 1.5707015991210936e-05,
      "step": 90470,
      "training_step_time": 7.759368896484375
    },
    {
      "epoch": 1.5707969665527344e-05,
      "model_forward_time": 1.856490135192871,
      "step": 90471
    },
    {
      "epoch": 1.5707969665527344e-05,
      "step": 90471,
      "training_step_time": 7.76188063621521
    },
    {
      "epoch": 1.570892333984375e-05,
      "model_forward_time": 1.8394997119903564,
      "step": 90472
    },
    {
      "epoch": 1.570892333984375e-05,
      "step": 90472,
      "training_step_time": 7.774933338165283
    },
    {
      "epoch": 1.5709877014160155e-05,
      "model_forward_time": 1.8028450012207031,
      "step": 90473
    },
    {
      "epoch": 1.5709877014160155e-05,
      "step": 90473,
      "training_step_time": 7.690117359161377
    },
    {
      "epoch": 1.5710830688476563e-05,
      "model_forward_time": 1.8424053192138672,
      "step": 90474
    },
    {
      "epoch": 1.5710830688476563e-05,
      "step": 90474,
      "training_step_time": 7.780155897140503
    },
    {
      "epoch": 1.571178436279297e-05,
      "model_forward_time": 2.9551005363464355,
      "step": 90475
    },
    {
      "epoch": 1.571178436279297e-05,
      "step": 90475,
      "training_step_time": 8.54195237159729
    },
    {
      "epoch": 1.5712738037109374e-05,
      "model_forward_time": 1.794895887374878,
      "step": 90476
    },
    {
      "epoch": 1.5712738037109374e-05,
      "step": 90476,
      "training_step_time": 8.80973505973816
    },
    {
      "epoch": 1.571369171142578e-05,
      "model_forward_time": 1.8345918655395508,
      "step": 90477
    },
    {
      "epoch": 1.571369171142578e-05,
      "step": 90477,
      "training_step_time": 10.740702629089355
    },
    {
      "epoch": 1.571464538574219e-05,
      "model_forward_time": 1.8121247291564941,
      "step": 90478
    },
    {
      "epoch": 1.571464538574219e-05,
      "step": 90478,
      "training_step_time": 7.846224784851074
    },
    {
      "epoch": 1.5715599060058593e-05,
      "model_forward_time": 1.934480905532837,
      "step": 90479
    },
    {
      "epoch": 1.5715599060058593e-05,
      "step": 90479,
      "training_step_time": 7.741246223449707
    },
    {
      "epoch": 1.5716552734375e-05,
      "grad_norm": 0.07994405180215836,
      "learning_rate": 2.457915606793543e-07,
      "loss": 0.0565,
      "step": 90480
    },
    {
      "epoch": 1.5716552734375e-05,
      "model_forward_time": 1.9246251583099365,
      "step": 90480
    },
    {
      "dynamics_loss_avg": 0.05247527835890651,
      "epoch": 1.5716552734375e-05,
      "step": 90480
    },
    {
      "action_loss_avg": 0.0017161200987175107,
      "epoch": 1.5716552734375e-05,
      "step": 90480
    },
    {
      "epoch": 1.5716552734375e-05,
      "step": 90480,
      "training_step_time": 8.67420506477356
    },
    {
      "epoch": 1.5717506408691408e-05,
      "model_forward_time": 1.9769678115844727,
      "step": 90481
    },
    {
      "epoch": 1.5717506408691408e-05,
      "step": 90481,
      "training_step_time": 8.25466275215149
    },
    {
      "epoch": 1.5718460083007812e-05,
      "model_forward_time": 1.9722585678100586,
      "step": 90482
    },
    {
      "epoch": 1.5718460083007812e-05,
      "step": 90482,
      "training_step_time": 7.66552209854126
    },
    {
      "epoch": 1.571941375732422e-05,
      "model_forward_time": 1.9569709300994873,
      "step": 90483
    },
    {
      "epoch": 1.571941375732422e-05,
      "step": 90483,
      "training_step_time": 8.384551286697388
    },
    {
      "epoch": 1.5720367431640627e-05,
      "model_forward_time": 3.0062873363494873,
      "step": 90484
    },
    {
      "epoch": 1.5720367431640627e-05,
      "step": 90484,
      "training_step_time": 9.540283918380737
    },
    {
      "epoch": 1.572132110595703e-05,
      "model_forward_time": 1.9512934684753418,
      "step": 90485
    },
    {
      "epoch": 1.572132110595703e-05,
      "step": 90485,
      "training_step_time": 8.76847791671753
    },
    {
      "epoch": 1.5722274780273438e-05,
      "model_forward_time": 1.9455459117889404,
      "step": 90486
    },
    {
      "epoch": 1.5722274780273438e-05,
      "step": 90486,
      "training_step_time": 8.26610255241394
    },
    {
      "epoch": 1.5723228454589842e-05,
      "model_forward_time": 1.9749865531921387,
      "step": 90487
    },
    {
      "epoch": 1.5723228454589842e-05,
      "step": 90487,
      "training_step_time": 7.7438859939575195
    },
    {
      "epoch": 1.572418212890625e-05,
      "model_forward_time": 1.9570870399475098,
      "step": 90488
    },
    {
      "epoch": 1.572418212890625e-05,
      "step": 90488,
      "training_step_time": 7.706669569015503
    },
    {
      "epoch": 1.5725135803222657e-05,
      "model_forward_time": 1.9554314613342285,
      "step": 90489
    },
    {
      "epoch": 1.5725135803222657e-05,
      "step": 90489,
      "training_step_time": 7.689422607421875
    },
    {
      "epoch": 1.572608947753906e-05,
      "grad_norm": 0.06989840418100357,
      "learning_rate": 2.452797784310046e-07,
      "loss": 0.0536,
      "step": 90490
    },
    {
      "epoch": 1.572608947753906e-05,
      "model_forward_time": 1.9703333377838135,
      "step": 90490
    },
    {
      "dynamics_loss_avg": 0.05793348848819733,
      "epoch": 1.572608947753906e-05,
      "step": 90490
    },
    {
      "action_loss_avg": 0.0013659912976436317,
      "epoch": 1.572608947753906e-05,
      "step": 90490
    },
    {
      "epoch": 1.572608947753906e-05,
      "step": 90490,
      "training_step_time": 7.739194631576538
    },
    {
      "epoch": 1.572704315185547e-05,
      "model_forward_time": 1.9506385326385498,
      "step": 90491
    },
    {
      "epoch": 1.572704315185547e-05,
      "step": 90491,
      "training_step_time": 8.272944211959839
    },
    {
      "epoch": 1.5727996826171876e-05,
      "model_forward_time": 1.9724619388580322,
      "step": 90492
    },
    {
      "epoch": 1.5727996826171876e-05,
      "step": 90492,
      "training_step_time": 7.612226247787476
    },
    {
      "epoch": 1.572895050048828e-05,
      "model_forward_time": 3.030012607574463,
      "step": 90493
    },
    {
      "epoch": 1.572895050048828e-05,
      "step": 90493,
      "training_step_time": 7.8741066455841064
    },
    {
      "epoch": 1.5729904174804687e-05,
      "model_forward_time": 1.9253709316253662,
      "step": 90494
    },
    {
      "epoch": 1.5729904174804687e-05,
      "step": 90494,
      "training_step_time": 8.699617147445679
    },
    {
      "epoch": 1.5730857849121095e-05,
      "model_forward_time": 1.9223480224609375,
      "step": 90495
    },
    {
      "epoch": 1.5730857849121095e-05,
      "step": 90495,
      "training_step_time": 9.317813634872437
    },
    {
      "epoch": 1.57318115234375e-05,
      "model_forward_time": 1.9227962493896484,
      "step": 90496
    },
    {
      "epoch": 1.57318115234375e-05,
      "step": 90496,
      "training_step_time": 8.263824701309204
    },
    {
      "epoch": 1.5732765197753906e-05,
      "model_forward_time": 1.930542230606079,
      "step": 90497
    },
    {
      "epoch": 1.5732765197753906e-05,
      "step": 90497,
      "training_step_time": 8.241299867630005
    },
    {
      "epoch": 1.5733718872070314e-05,
      "model_forward_time": 1.9421210289001465,
      "step": 90498
    },
    {
      "epoch": 1.5733718872070314e-05,
      "step": 90498,
      "training_step_time": 8.819068431854248
    },
    {
      "epoch": 1.5734672546386718e-05,
      "model_forward_time": 1.93304443359375,
      "step": 90499
    },
    {
      "epoch": 1.5734672546386718e-05,
      "step": 90499,
      "training_step_time": 7.762449264526367
    },
    {
      "epoch": 1.5735626220703125e-05,
      "grad_norm": 0.07264266163110733,
      "learning_rate": 2.447685161517105e-07,
      "loss": 0.0535,
      "step": 90500
    },
    {
      "epoch": 1.5735626220703125e-05,
      "model_forward_time": 1.9143402576446533,
      "step": 90500
    },
    {
      "dynamics_loss_avg": 0.05199610237032175,
      "epoch": 1.5735626220703125e-05,
      "step": 90500
    },
    {
      "action_loss_avg": 0.0010517260103370062,
      "epoch": 1.5735626220703125e-05,
      "step": 90500
    },
    {
      "epoch": 1.5735626220703125e-05,
      "step": 90500,
      "training_step_time": 7.760636329650879
    },
    {
      "epoch": 1.5736579895019532e-05,
      "model_forward_time": 1.9093220233917236,
      "step": 90501
    },
    {
      "epoch": 1.5736579895019532e-05,
      "step": 90501,
      "training_step_time": 7.667065143585205
    },
    {
      "epoch": 1.5737533569335936e-05,
      "model_forward_time": 1.9487085342407227,
      "step": 90502
    },
    {
      "epoch": 1.5737533569335936e-05,
      "step": 90502,
      "training_step_time": 11.58078908920288
    },
    {
      "epoch": 1.5738487243652344e-05,
      "model_forward_time": 1.894787073135376,
      "step": 90503
    },
    {
      "epoch": 1.5738487243652344e-05,
      "step": 90503,
      "training_step_time": 8.088919639587402
    },
    {
      "epoch": 1.573944091796875e-05,
      "model_forward_time": 1.931164264678955,
      "step": 90504
    },
    {
      "epoch": 1.573944091796875e-05,
      "step": 90504,
      "training_step_time": 6.95249605178833
    },
    {
      "epoch": 1.5740394592285155e-05,
      "model_forward_time": 1.8971912860870361,
      "step": 90505
    },
    {
      "epoch": 1.5740394592285155e-05,
      "step": 90505,
      "training_step_time": 7.692951679229736
    },
    {
      "epoch": 1.5741348266601563e-05,
      "model_forward_time": 1.9025249481201172,
      "step": 90506
    },
    {
      "epoch": 1.5741348266601563e-05,
      "step": 90506,
      "training_step_time": 7.599196672439575
    },
    {
      "epoch": 1.574230194091797e-05,
      "model_forward_time": 1.9091770648956299,
      "step": 90507
    },
    {
      "epoch": 1.574230194091797e-05,
      "step": 90507,
      "training_step_time": 6.865477561950684
    },
    {
      "epoch": 1.5743255615234374e-05,
      "model_forward_time": 1.9340193271636963,
      "step": 90508
    },
    {
      "epoch": 1.5743255615234374e-05,
      "step": 90508,
      "training_step_time": 6.868130683898926
    },
    {
      "epoch": 1.574420928955078e-05,
      "model_forward_time": 1.9177486896514893,
      "step": 90509
    },
    {
      "epoch": 1.574420928955078e-05,
      "step": 90509,
      "training_step_time": 6.864063024520874
    },
    {
      "epoch": 1.574516296386719e-05,
      "grad_norm": 0.07598744332790375,
      "learning_rate": 2.4425777389738224e-07,
      "loss": 0.0548,
      "step": 90510
    },
    {
      "epoch": 1.574516296386719e-05,
      "model_forward_time": 1.915571689605713,
      "step": 90510
    },
    {
      "dynamics_loss_avg": 0.05299749029800296,
      "epoch": 1.574516296386719e-05,
      "step": 90510
    },
    {
      "action_loss_avg": 0.001497290818952024,
      "epoch": 1.574516296386719e-05,
      "step": 90510
    },
    {
      "epoch": 1.574516296386719e-05,
      "step": 90510,
      "training_step_time": 6.869520902633667
    },
    {
      "epoch": 1.5746116638183593e-05,
      "model_forward_time": 1.566115140914917,
      "step": 90511
    },
    {
      "epoch": 1.5746116638183593e-05,
      "step": 90511,
      "training_step_time": 6.872538089752197
    },
    {
      "epoch": 1.57470703125e-05,
      "model_forward_time": 2.9593124389648438,
      "step": 90512
    },
    {
      "epoch": 1.57470703125e-05,
      "step": 90512,
      "training_step_time": 8.774606466293335
    },
    {
      "epoch": 1.5748023986816408e-05,
      "model_forward_time": 1.5868444442749023,
      "step": 90513
    },
    {
      "epoch": 1.5748023986816408e-05,
      "step": 90513,
      "training_step_time": 7.409081220626831
    },
    {
      "epoch": 1.5748977661132812e-05,
      "model_forward_time": 1.9396157264709473,
      "step": 90514
    },
    {
      "epoch": 1.5748977661132812e-05,
      "step": 90514,
      "training_step_time": 8.646109104156494
    },
    {
      "epoch": 1.574993133544922e-05,
      "model_forward_time": 1.906681776046753,
      "step": 90515
    },
    {
      "epoch": 1.574993133544922e-05,
      "step": 90515,
      "training_step_time": 6.885039329528809
    },
    {
      "epoch": 1.5750885009765623e-05,
      "model_forward_time": 1.957812786102295,
      "step": 90516
    },
    {
      "epoch": 1.5750885009765623e-05,
      "step": 90516,
      "training_step_time": 6.857389688491821
    },
    {
      "epoch": 1.575183868408203e-05,
      "model_forward_time": 1.889061689376831,
      "step": 90517
    },
    {
      "epoch": 1.575183868408203e-05,
      "step": 90517,
      "training_step_time": 6.8481361865997314
    },
    {
      "epoch": 1.5752792358398438e-05,
      "model_forward_time": 1.7812843322753906,
      "step": 90518
    },
    {
      "epoch": 1.5752792358398438e-05,
      "step": 90518,
      "training_step_time": 6.842606544494629
    },
    {
      "epoch": 1.5753746032714842e-05,
      "model_forward_time": 1.7879643440246582,
      "step": 90519
    },
    {
      "epoch": 1.5753746032714842e-05,
      "step": 90519,
      "training_step_time": 6.847145080566406
    },
    {
      "epoch": 1.575469970703125e-05,
      "grad_norm": 0.06890509277582169,
      "learning_rate": 2.43747551723873e-07,
      "loss": 0.0552,
      "step": 90520
    },
    {
      "epoch": 1.575469970703125e-05,
      "model_forward_time": 1.790557861328125,
      "step": 90520
    },
    {
      "dynamics_loss_avg": 0.0653012627735734,
      "epoch": 1.575469970703125e-05,
      "step": 90520
    },
    {
      "action_loss_avg": 0.001139307813718915,
      "epoch": 1.575469970703125e-05,
      "step": 90520
    },
    {
      "epoch": 1.575469970703125e-05,
      "step": 90520,
      "training_step_time": 6.821519374847412
    },
    {
      "epoch": 1.5755653381347657e-05,
      "model_forward_time": 2.8869049549102783,
      "step": 90521
    },
    {
      "epoch": 1.5755653381347657e-05,
      "step": 90521,
      "training_step_time": 9.248367071151733
    },
    {
      "epoch": 1.575660705566406e-05,
      "model_forward_time": 1.7862777709960938,
      "step": 90522
    },
    {
      "epoch": 1.575660705566406e-05,
      "step": 90522,
      "training_step_time": 7.356556415557861
    },
    {
      "epoch": 1.575756072998047e-05,
      "model_forward_time": 1.7862603664398193,
      "step": 90523
    },
    {
      "epoch": 1.575756072998047e-05,
      "step": 90523,
      "training_step_time": 7.6766180992126465
    },
    {
      "epoch": 1.5758514404296876e-05,
      "model_forward_time": 1.797422170639038,
      "step": 90524
    },
    {
      "epoch": 1.5758514404296876e-05,
      "step": 90524,
      "training_step_time": 7.666558742523193
    },
    {
      "epoch": 1.575946807861328e-05,
      "model_forward_time": 1.7955055236816406,
      "step": 90525
    },
    {
      "epoch": 1.575946807861328e-05,
      "step": 90525,
      "training_step_time": 6.819294691085815
    },
    {
      "epoch": 1.5760421752929687e-05,
      "model_forward_time": 1.795877456665039,
      "step": 90526
    },
    {
      "epoch": 1.5760421752929687e-05,
      "step": 90526,
      "training_step_time": 6.818490982055664
    },
    {
      "epoch": 1.5761375427246095e-05,
      "model_forward_time": 1.787292242050171,
      "step": 90527
    },
    {
      "epoch": 1.5761375427246095e-05,
      "step": 90527,
      "training_step_time": 6.820754528045654
    },
    {
      "epoch": 1.57623291015625e-05,
      "model_forward_time": 1.7918016910552979,
      "step": 90528
    },
    {
      "epoch": 1.57623291015625e-05,
      "step": 90528,
      "training_step_time": 6.827540636062622
    },
    {
      "epoch": 1.5763282775878906e-05,
      "model_forward_time": 1.7987456321716309,
      "step": 90529
    },
    {
      "epoch": 1.5763282775878906e-05,
      "step": 90529,
      "training_step_time": 6.785143136978149
    },
    {
      "epoch": 1.5764236450195314e-05,
      "grad_norm": 0.0954630970954895,
      "learning_rate": 2.432378496869814e-07,
      "loss": 0.0524,
      "step": 90530
    },
    {
      "epoch": 1.5764236450195314e-05,
      "model_forward_time": 2.915421724319458,
      "step": 90530
    },
    {
      "dynamics_loss_avg": 0.07559650018811226,
      "epoch": 1.5764236450195314e-05,
      "step": 90530
    },
    {
      "action_loss_avg": 0.0012613979284651577,
      "epoch": 1.5764236450195314e-05,
      "step": 90530
    },
    {
      "epoch": 1.5764236450195314e-05,
      "step": 90530,
      "training_step_time": 9.88395357131958
    },
    {
      "epoch": 1.5765190124511718e-05,
      "model_forward_time": 1.7920973300933838,
      "step": 90531
    },
    {
      "epoch": 1.5765190124511718e-05,
      "step": 90531,
      "training_step_time": 7.580960035324097
    },
    {
      "epoch": 1.5766143798828125e-05,
      "model_forward_time": 1.7967441082000732,
      "step": 90532
    },
    {
      "epoch": 1.5766143798828125e-05,
      "step": 90532,
      "training_step_time": 7.635274410247803
    },
    {
      "epoch": 1.5767097473144532e-05,
      "model_forward_time": 1.800058126449585,
      "step": 90533
    },
    {
      "epoch": 1.5767097473144532e-05,
      "step": 90533,
      "training_step_time": 7.608165502548218
    },
    {
      "epoch": 1.5768051147460936e-05,
      "model_forward_time": 1.8030221462249756,
      "step": 90534
    },
    {
      "epoch": 1.5768051147460936e-05,
      "step": 90534,
      "training_step_time": 6.7740631103515625
    },
    {
      "epoch": 1.5769004821777344e-05,
      "model_forward_time": 1.7662274837493896,
      "step": 90535
    },
    {
      "epoch": 1.5769004821777344e-05,
      "step": 90535,
      "training_step_time": 6.809154510498047
    },
    {
      "epoch": 1.576995849609375e-05,
      "model_forward_time": 1.777564287185669,
      "step": 90536
    },
    {
      "epoch": 1.576995849609375e-05,
      "step": 90536,
      "training_step_time": 6.794244050979614
    },
    {
      "epoch": 1.5770912170410155e-05,
      "model_forward_time": 1.7807233333587646,
      "step": 90537
    },
    {
      "epoch": 1.5770912170410155e-05,
      "step": 90537,
      "training_step_time": 6.799609661102295
    },
    {
      "epoch": 1.5771865844726563e-05,
      "model_forward_time": 1.7748134136199951,
      "step": 90538
    },
    {
      "epoch": 1.5771865844726563e-05,
      "step": 90538,
      "training_step_time": 6.801746129989624
    },
    {
      "epoch": 1.577281951904297e-05,
      "model_forward_time": 1.8185358047485352,
      "step": 90539
    },
    {
      "epoch": 1.577281951904297e-05,
      "step": 90539,
      "training_step_time": 10.813992738723755
    },
    {
      "epoch": 1.5773773193359374e-05,
      "grad_norm": 0.0731036439538002,
      "learning_rate": 2.4272866784244567e-07,
      "loss": 0.0534,
      "step": 90540
    },
    {
      "epoch": 1.5773773193359374e-05,
      "model_forward_time": 1.7692289352416992,
      "step": 90540
    },
    {
      "dynamics_loss_avg": 0.06128743849694729,
      "epoch": 1.5773773193359374e-05,
      "step": 90540
    },
    {
      "action_loss_avg": 0.0016153591044712813,
      "epoch": 1.5773773193359374e-05,
      "step": 90540
    },
    {
      "epoch": 1.5773773193359374e-05,
      "step": 90540,
      "training_step_time": 7.7236104011535645
    },
    {
      "epoch": 1.577472686767578e-05,
      "model_forward_time": 1.7753946781158447,
      "step": 90541
    },
    {
      "epoch": 1.577472686767578e-05,
      "step": 90541,
      "training_step_time": 7.627732038497925
    },
    {
      "epoch": 1.577568054199219e-05,
      "model_forward_time": 1.7957987785339355,
      "step": 90542
    },
    {
      "epoch": 1.577568054199219e-05,
      "step": 90542,
      "training_step_time": 7.6482367515563965
    },
    {
      "epoch": 1.5776634216308593e-05,
      "model_forward_time": 1.7901380062103271,
      "step": 90543
    },
    {
      "epoch": 1.5776634216308593e-05,
      "step": 90543,
      "training_step_time": 6.788325786590576
    },
    {
      "epoch": 1.5777587890625e-05,
      "model_forward_time": 1.7885076999664307,
      "step": 90544
    },
    {
      "epoch": 1.5777587890625e-05,
      "step": 90544,
      "training_step_time": 6.8171021938323975
    },
    {
      "epoch": 1.5778541564941408e-05,
      "model_forward_time": 1.789823055267334,
      "step": 90545
    },
    {
      "epoch": 1.5778541564941408e-05,
      "step": 90545,
      "training_step_time": 6.800657033920288
    },
    {
      "epoch": 1.5779495239257812e-05,
      "model_forward_time": 1.7927167415618896,
      "step": 90546
    },
    {
      "epoch": 1.5779495239257812e-05,
      "step": 90546,
      "training_step_time": 6.808656215667725
    },
    {
      "epoch": 1.578044891357422e-05,
      "model_forward_time": 1.8062012195587158,
      "step": 90547
    },
    {
      "epoch": 1.578044891357422e-05,
      "step": 90547,
      "training_step_time": 6.768941879272461
    },
    {
      "epoch": 1.5781402587890623e-05,
      "model_forward_time": 1.8033277988433838,
      "step": 90548
    },
    {
      "epoch": 1.5781402587890623e-05,
      "step": 90548,
      "training_step_time": 9.398961782455444
    },
    {
      "epoch": 1.578235626220703e-05,
      "model_forward_time": 2.8824799060821533,
      "step": 90549
    },
    {
      "epoch": 1.578235626220703e-05,
      "step": 90549,
      "training_step_time": 7.741886854171753
    },
    {
      "epoch": 1.5783309936523438e-05,
      "grad_norm": 0.0804198831319809,
      "learning_rate": 2.4222000624595177e-07,
      "loss": 0.0543,
      "step": 90550
    },
    {
      "epoch": 1.5783309936523438e-05,
      "model_forward_time": 1.7905890941619873,
      "step": 90550
    },
    {
      "dynamics_loss_avg": 0.04748344421386719,
      "epoch": 1.5783309936523438e-05,
      "step": 90550
    },
    {
      "action_loss_avg": 0.0011053582013119013,
      "epoch": 1.5783309936523438e-05,
      "step": 90550
    },
    {
      "epoch": 1.5783309936523438e-05,
      "step": 90550,
      "training_step_time": 7.625128507614136
    },
    {
      "epoch": 1.5784263610839842e-05,
      "model_forward_time": 1.7903802394866943,
      "step": 90551
    },
    {
      "epoch": 1.5784263610839842e-05,
      "step": 90551,
      "training_step_time": 8.153069734573364
    },
    {
      "epoch": 1.578521728515625e-05,
      "model_forward_time": 1.792750597000122,
      "step": 90552
    },
    {
      "epoch": 1.578521728515625e-05,
      "step": 90552,
      "training_step_time": 6.82434868812561
    },
    {
      "epoch": 1.5786170959472657e-05,
      "model_forward_time": 1.7944822311401367,
      "step": 90553
    },
    {
      "epoch": 1.5786170959472657e-05,
      "step": 90553,
      "training_step_time": 6.835509777069092
    },
    {
      "epoch": 1.578712463378906e-05,
      "model_forward_time": 1.7954256534576416,
      "step": 90554
    },
    {
      "epoch": 1.578712463378906e-05,
      "step": 90554,
      "training_step_time": 6.811570405960083
    },
    {
      "epoch": 1.578807830810547e-05,
      "model_forward_time": 1.7950801849365234,
      "step": 90555
    },
    {
      "epoch": 1.578807830810547e-05,
      "step": 90555,
      "training_step_time": 6.794777154922485
    },
    {
      "epoch": 1.5789031982421876e-05,
      "model_forward_time": 1.7973005771636963,
      "step": 90556
    },
    {
      "epoch": 1.5789031982421876e-05,
      "step": 90556,
      "training_step_time": 6.797427415847778
    },
    {
      "epoch": 1.578998565673828e-05,
      "model_forward_time": 1.7856831550598145,
      "step": 90557
    },
    {
      "epoch": 1.578998565673828e-05,
      "step": 90557,
      "training_step_time": 8.752374410629272
    },
    {
      "epoch": 1.5790939331054687e-05,
      "model_forward_time": 2.8475654125213623,
      "step": 90558
    },
    {
      "epoch": 1.5790939331054687e-05,
      "step": 90558,
      "training_step_time": 8.623705863952637
    },
    {
      "epoch": 1.5791893005371095e-05,
      "model_forward_time": 1.7837517261505127,
      "step": 90559
    },
    {
      "epoch": 1.5791893005371095e-05,
      "step": 90559,
      "training_step_time": 7.654028654098511
    },
    {
      "epoch": 1.57928466796875e-05,
      "grad_norm": 0.11971078813076019,
      "learning_rate": 2.4171186495312303e-07,
      "loss": 0.0532,
      "step": 90560
    },
    {
      "epoch": 1.57928466796875e-05,
      "model_forward_time": 1.7946805953979492,
      "step": 90560
    },
    {
      "dynamics_loss_avg": 0.056431294232606885,
      "epoch": 1.57928466796875e-05,
      "step": 90560
    },
    {
      "action_loss_avg": 0.0017116995935793967,
      "epoch": 1.57928466796875e-05,
      "step": 90560
    },
    {
      "epoch": 1.57928466796875e-05,
      "step": 90560,
      "training_step_time": 8.296501398086548
    },
    {
      "epoch": 1.5793800354003906e-05,
      "model_forward_time": 1.7955150604248047,
      "step": 90561
    },
    {
      "epoch": 1.5793800354003906e-05,
      "step": 90561,
      "training_step_time": 7.693448066711426
    },
    {
      "epoch": 1.5794754028320314e-05,
      "model_forward_time": 1.785846471786499,
      "step": 90562
    },
    {
      "epoch": 1.5794754028320314e-05,
      "step": 90562,
      "training_step_time": 6.847473382949829
    },
    {
      "epoch": 1.5795707702636718e-05,
      "model_forward_time": 1.7803163528442383,
      "step": 90563
    },
    {
      "epoch": 1.5795707702636718e-05,
      "step": 90563,
      "training_step_time": 6.844295501708984
    },
    {
      "epoch": 1.5796661376953125e-05,
      "model_forward_time": 1.7861177921295166,
      "step": 90564
    },
    {
      "epoch": 1.5796661376953125e-05,
      "step": 90564,
      "training_step_time": 6.865185976028442
    },
    {
      "epoch": 1.5797615051269532e-05,
      "model_forward_time": 1.7879137992858887,
      "step": 90565
    },
    {
      "epoch": 1.5797615051269532e-05,
      "step": 90565,
      "training_step_time": 6.854538917541504
    },
    {
      "epoch": 1.5798568725585936e-05,
      "model_forward_time": 1.7894866466522217,
      "step": 90566
    },
    {
      "epoch": 1.5798568725585936e-05,
      "step": 90566,
      "training_step_time": 7.7203099727630615
    },
    {
      "epoch": 1.5799522399902344e-05,
      "model_forward_time": 2.874873638153076,
      "step": 90567
    },
    {
      "epoch": 1.5799522399902344e-05,
      "step": 90567,
      "training_step_time": 9.20929479598999
    },
    {
      "epoch": 1.580047607421875e-05,
      "model_forward_time": 1.7873947620391846,
      "step": 90568
    },
    {
      "epoch": 1.580047607421875e-05,
      "step": 90568,
      "training_step_time": 7.749279737472534
    },
    {
      "epoch": 1.5801429748535155e-05,
      "model_forward_time": 1.7916045188903809,
      "step": 90569
    },
    {
      "epoch": 1.5801429748535155e-05,
      "step": 90569,
      "training_step_time": 8.270601749420166
    },
    {
      "epoch": 1.5802383422851563e-05,
      "grad_norm": 0.12541091442108154,
      "learning_rate": 2.412042440195322e-07,
      "loss": 0.0525,
      "step": 90570
    },
    {
      "epoch": 1.5802383422851563e-05,
      "model_forward_time": 1.7930090427398682,
      "step": 90570
    },
    {
      "dynamics_loss_avg": 0.0629251567646861,
      "epoch": 1.5802383422851563e-05,
      "step": 90570
    },
    {
      "action_loss_avg": 0.0012812777684303,
      "epoch": 1.5802383422851563e-05,
      "step": 90570
    },
    {
      "epoch": 1.5802383422851563e-05,
      "step": 90570,
      "training_step_time": 7.670883655548096
    },
    {
      "epoch": 1.580333709716797e-05,
      "model_forward_time": 1.788041114807129,
      "step": 90571
    },
    {
      "epoch": 1.580333709716797e-05,
      "step": 90571,
      "training_step_time": 6.823009490966797
    },
    {
      "epoch": 1.5804290771484374e-05,
      "model_forward_time": 1.8027839660644531,
      "step": 90572
    },
    {
      "epoch": 1.5804290771484374e-05,
      "step": 90572,
      "training_step_time": 6.802405118942261
    },
    {
      "epoch": 1.580524444580078e-05,
      "model_forward_time": 1.7935521602630615,
      "step": 90573
    },
    {
      "epoch": 1.580524444580078e-05,
      "step": 90573,
      "training_step_time": 6.836813688278198
    },
    {
      "epoch": 1.580619812011719e-05,
      "model_forward_time": 1.7977347373962402,
      "step": 90574
    },
    {
      "epoch": 1.580619812011719e-05,
      "step": 90574,
      "training_step_time": 6.943943977355957
    },
    {
      "epoch": 1.5807151794433593e-05,
      "model_forward_time": 1.7844669818878174,
      "step": 90575
    },
    {
      "epoch": 1.5807151794433593e-05,
      "step": 90575,
      "training_step_time": 7.68906307220459
    },
    {
      "epoch": 1.580810546875e-05,
      "model_forward_time": 2.8982105255126953,
      "step": 90576
    },
    {
      "epoch": 1.580810546875e-05,
      "step": 90576,
      "training_step_time": 10.188043594360352
    },
    {
      "epoch": 1.5809059143066408e-05,
      "model_forward_time": 1.7915523052215576,
      "step": 90577
    },
    {
      "epoch": 1.5809059143066408e-05,
      "step": 90577,
      "training_step_time": 7.81851601600647
    },
    {
      "epoch": 1.5810012817382812e-05,
      "model_forward_time": 1.7814431190490723,
      "step": 90578
    },
    {
      "epoch": 1.5810012817382812e-05,
      "step": 90578,
      "training_step_time": 7.413967847824097
    },
    {
      "epoch": 1.581096649169922e-05,
      "model_forward_time": 1.7954134941101074,
      "step": 90579
    },
    {
      "epoch": 1.581096649169922e-05,
      "step": 90579,
      "training_step_time": 7.720763921737671
    },
    {
      "epoch": 1.5811920166015623e-05,
      "grad_norm": 0.11926374584436417,
      "learning_rate": 2.4069714350068995e-07,
      "loss": 0.0542,
      "step": 90580
    },
    {
      "epoch": 1.5811920166015623e-05,
      "model_forward_time": 1.7879002094268799,
      "step": 90580
    },
    {
      "dynamics_loss_avg": 0.0513960674405098,
      "epoch": 1.5811920166015623e-05,
      "step": 90580
    },
    {
      "action_loss_avg": 0.001346559025114402,
      "epoch": 1.5811920166015623e-05,
      "step": 90580
    },
    {
      "epoch": 1.5811920166015623e-05,
      "step": 90580,
      "training_step_time": 6.947392702102661
    },
    {
      "epoch": 1.581287384033203e-05,
      "model_forward_time": 1.7931780815124512,
      "step": 90581
    },
    {
      "epoch": 1.581287384033203e-05,
      "step": 90581,
      "training_step_time": 6.9285359382629395
    },
    {
      "epoch": 1.5813827514648438e-05,
      "model_forward_time": 1.7860023975372314,
      "step": 90582
    },
    {
      "epoch": 1.5813827514648438e-05,
      "step": 90582,
      "training_step_time": 6.8301637172698975
    },
    {
      "epoch": 1.5814781188964842e-05,
      "model_forward_time": 1.7879283428192139,
      "step": 90583
    },
    {
      "epoch": 1.5814781188964842e-05,
      "step": 90583,
      "training_step_time": 6.850025653839111
    },
    {
      "epoch": 1.581573486328125e-05,
      "model_forward_time": 1.7935876846313477,
      "step": 90584
    },
    {
      "epoch": 1.581573486328125e-05,
      "step": 90584,
      "training_step_time": 7.571230173110962
    },
    {
      "epoch": 1.5816688537597657e-05,
      "model_forward_time": 2.9040720462799072,
      "step": 90585
    },
    {
      "epoch": 1.5816688537597657e-05,
      "step": 90585,
      "training_step_time": 12.629048585891724
    },
    {
      "epoch": 1.581764221191406e-05,
      "model_forward_time": 1.7664737701416016,
      "step": 90586
    },
    {
      "epoch": 1.581764221191406e-05,
      "step": 90586,
      "training_step_time": 7.729057788848877
    },
    {
      "epoch": 1.581859588623047e-05,
      "model_forward_time": 1.776686191558838,
      "step": 90587
    },
    {
      "epoch": 1.581859588623047e-05,
      "step": 90587,
      "training_step_time": 7.353850603103638
    },
    {
      "epoch": 1.5819549560546876e-05,
      "model_forward_time": 1.7642569541931152,
      "step": 90588
    },
    {
      "epoch": 1.5819549560546876e-05,
      "step": 90588,
      "training_step_time": 7.596224069595337
    },
    {
      "epoch": 1.582050323486328e-05,
      "model_forward_time": 1.7691080570220947,
      "step": 90589
    },
    {
      "epoch": 1.582050323486328e-05,
      "step": 90589,
      "training_step_time": 6.860994577407837
    },
    {
      "epoch": 1.5821456909179687e-05,
      "grad_norm": 0.0821254774928093,
      "learning_rate": 2.401905634520529e-07,
      "loss": 0.0526,
      "step": 90590
    },
    {
      "epoch": 1.5821456909179687e-05,
      "model_forward_time": 1.7581291198730469,
      "step": 90590
    },
    {
      "dynamics_loss_avg": 0.048662088252604006,
      "epoch": 1.5821456909179687e-05,
      "step": 90590
    },
    {
      "action_loss_avg": 0.0015980071795638651,
      "epoch": 1.5821456909179687e-05,
      "step": 90590
    },
    {
      "epoch": 1.5821456909179687e-05,
      "step": 90590,
      "training_step_time": 6.884205102920532
    },
    {
      "epoch": 1.5822410583496095e-05,
      "model_forward_time": 1.758725643157959,
      "step": 90591
    },
    {
      "epoch": 1.5822410583496095e-05,
      "step": 90591,
      "training_step_time": 6.855935573577881
    },
    {
      "epoch": 1.58233642578125e-05,
      "model_forward_time": 1.7407541275024414,
      "step": 90592
    },
    {
      "epoch": 1.58233642578125e-05,
      "step": 90592,
      "training_step_time": 6.883075475692749
    },
    {
      "epoch": 1.5824317932128906e-05,
      "model_forward_time": 1.7604000568389893,
      "step": 90593
    },
    {
      "epoch": 1.5824317932128906e-05,
      "step": 90593,
      "training_step_time": 7.757282257080078
    },
    {
      "epoch": 1.5825271606445314e-05,
      "model_forward_time": 1.3845350742340088,
      "step": 90594
    },
    {
      "epoch": 1.5825271606445314e-05,
      "step": 90594,
      "training_step_time": 11.842204570770264
    },
    {
      "epoch": 1.5826225280761718e-05,
      "model_forward_time": 2.8597300052642822,
      "step": 90595
    },
    {
      "epoch": 1.5826225280761718e-05,
      "step": 90595,
      "training_step_time": 8.496288299560547
    },
    {
      "epoch": 1.5827178955078125e-05,
      "model_forward_time": 1.7535429000854492,
      "step": 90596
    },
    {
      "epoch": 1.5827178955078125e-05,
      "step": 90596,
      "training_step_time": 7.686070919036865
    },
    {
      "epoch": 1.5828132629394532e-05,
      "model_forward_time": 1.7623896598815918,
      "step": 90597
    },
    {
      "epoch": 1.5828132629394532e-05,
      "step": 90597,
      "training_step_time": 7.593825340270996
    },
    {
      "epoch": 1.5829086303710937e-05,
      "model_forward_time": 1.7469327449798584,
      "step": 90598
    },
    {
      "epoch": 1.5829086303710937e-05,
      "step": 90598,
      "training_step_time": 7.6654980182647705
    },
    {
      "epoch": 1.5830039978027344e-05,
      "model_forward_time": 1.752586841583252,
      "step": 90599
    },
    {
      "epoch": 1.5830039978027344e-05,
      "step": 90599,
      "training_step_time": 6.85340690612793
    },
    {
      "epoch": 1.583099365234375e-05,
      "grad_norm": 0.09631750732660294,
      "learning_rate": 2.396845039290191e-07,
      "loss": 0.0535,
      "step": 90600
    },
    {
      "epoch": 1.583099365234375e-05,
      "model_forward_time": 1.7588977813720703,
      "step": 90600
    },
    {
      "dynamics_loss_avg": 0.05102997366338968,
      "epoch": 1.583099365234375e-05,
      "step": 90600
    },
    {
      "action_loss_avg": 0.0010093997814692556,
      "epoch": 1.583099365234375e-05,
      "step": 90600
    },
    {
      "epoch": 1.583099365234375e-05,
      "step": 90600,
      "training_step_time": 6.885826587677002
    },
    {
      "epoch": 1.5831947326660155e-05,
      "model_forward_time": 1.7512168884277344,
      "step": 90601
    },
    {
      "epoch": 1.5831947326660155e-05,
      "step": 90601,
      "training_step_time": 6.883291482925415
    },
    {
      "epoch": 1.5832901000976563e-05,
      "model_forward_time": 1.7570738792419434,
      "step": 90602
    },
    {
      "epoch": 1.5832901000976563e-05,
      "step": 90602,
      "training_step_time": 7.091402530670166
    },
    {
      "epoch": 1.583385467529297e-05,
      "model_forward_time": 1.7646682262420654,
      "step": 90603
    },
    {
      "epoch": 1.583385467529297e-05,
      "step": 90603,
      "training_step_time": 10.807910442352295
    },
    {
      "epoch": 1.5834808349609374e-05,
      "model_forward_time": 2.862727165222168,
      "step": 90604
    },
    {
      "epoch": 1.5834808349609374e-05,
      "step": 90604,
      "training_step_time": 8.815515041351318
    },
    {
      "epoch": 1.583576202392578e-05,
      "model_forward_time": 1.7592623233795166,
      "step": 90605
    },
    {
      "epoch": 1.583576202392578e-05,
      "step": 90605,
      "training_step_time": 7.732857942581177
    },
    {
      "epoch": 1.583671569824219e-05,
      "model_forward_time": 1.7464821338653564,
      "step": 90606
    },
    {
      "epoch": 1.583671569824219e-05,
      "step": 90606,
      "training_step_time": 7.6065967082977295
    },
    {
      "epoch": 1.5837669372558593e-05,
      "model_forward_time": 1.7618703842163086,
      "step": 90607
    },
    {
      "epoch": 1.5837669372558593e-05,
      "step": 90607,
      "training_step_time": 7.631600618362427
    },
    {
      "epoch": 1.5838623046875e-05,
      "model_forward_time": 1.7528941631317139,
      "step": 90608
    },
    {
      "epoch": 1.5838623046875e-05,
      "step": 90608,
      "training_step_time": 6.881775617599487
    },
    {
      "epoch": 1.5839576721191408e-05,
      "model_forward_time": 1.766427755355835,
      "step": 90609
    },
    {
      "epoch": 1.5839576721191408e-05,
      "step": 90609,
      "training_step_time": 6.868288516998291
    },
    {
      "epoch": 1.5840530395507812e-05,
      "grad_norm": 0.12943239510059357,
      "learning_rate": 2.3917896498693193e-07,
      "loss": 0.0528,
      "step": 90610
    },
    {
      "epoch": 1.5840530395507812e-05,
      "model_forward_time": 1.7592809200286865,
      "step": 90610
    },
    {
      "dynamics_loss_avg": 0.05813702419400215,
      "epoch": 1.5840530395507812e-05,
      "step": 90610
    },
    {
      "action_loss_avg": 0.001475463734823279,
      "epoch": 1.5840530395507812e-05,
      "step": 90610
    },
    {
      "epoch": 1.5840530395507812e-05,
      "step": 90610,
      "training_step_time": 6.876915454864502
    },
    {
      "epoch": 1.584148406982422e-05,
      "model_forward_time": 1.7614216804504395,
      "step": 90611
    },
    {
      "epoch": 1.584148406982422e-05,
      "step": 90611,
      "training_step_time": 7.123520851135254
    },
    {
      "epoch": 1.5842437744140623e-05,
      "model_forward_time": 1.7605516910552979,
      "step": 90612
    },
    {
      "epoch": 1.5842437744140623e-05,
      "step": 90612,
      "training_step_time": 8.20439887046814
    },
    {
      "epoch": 1.584339141845703e-05,
      "model_forward_time": 2.874544858932495,
      "step": 90613
    },
    {
      "epoch": 1.584339141845703e-05,
      "step": 90613,
      "training_step_time": 10.128422975540161
    },
    {
      "epoch": 1.5844345092773438e-05,
      "model_forward_time": 1.7530121803283691,
      "step": 90614
    },
    {
      "epoch": 1.5844345092773438e-05,
      "step": 90614,
      "training_step_time": 7.725864410400391
    },
    {
      "epoch": 1.5845298767089842e-05,
      "model_forward_time": 1.7516541481018066,
      "step": 90615
    },
    {
      "epoch": 1.5845298767089842e-05,
      "step": 90615,
      "training_step_time": 7.695426940917969
    },
    {
      "epoch": 1.584625244140625e-05,
      "model_forward_time": 1.7590546607971191,
      "step": 90616
    },
    {
      "epoch": 1.584625244140625e-05,
      "step": 90616,
      "training_step_time": 7.674243688583374
    },
    {
      "epoch": 1.5847206115722657e-05,
      "model_forward_time": 1.757016897201538,
      "step": 90617
    },
    {
      "epoch": 1.5847206115722657e-05,
      "step": 90617,
      "training_step_time": 6.900075912475586
    },
    {
      "epoch": 1.584815979003906e-05,
      "model_forward_time": 1.7609734535217285,
      "step": 90618
    },
    {
      "epoch": 1.584815979003906e-05,
      "step": 90618,
      "training_step_time": 6.887625217437744
    },
    {
      "epoch": 1.584911346435547e-05,
      "model_forward_time": 1.7519426345825195,
      "step": 90619
    },
    {
      "epoch": 1.584911346435547e-05,
      "step": 90619,
      "training_step_time": 6.8810789585113525
    },
    {
      "epoch": 1.5850067138671876e-05,
      "grad_norm": 0.08672775328159332,
      "learning_rate": 2.3867394668107333e-07,
      "loss": 0.0539,
      "step": 90620
    },
    {
      "epoch": 1.5850067138671876e-05,
      "model_forward_time": 1.7382817268371582,
      "step": 90620
    },
    {
      "dynamics_loss_avg": 0.045191450510174036,
      "epoch": 1.5850067138671876e-05,
      "step": 90620
    },
    {
      "action_loss_avg": 0.0012471063528209925,
      "epoch": 1.5850067138671876e-05,
      "step": 90620
    },
    {
      "epoch": 1.5850067138671876e-05,
      "step": 90620,
      "training_step_time": 7.0797271728515625
    },
    {
      "epoch": 1.585102081298828e-05,
      "model_forward_time": 1.7611241340637207,
      "step": 90621
    },
    {
      "epoch": 1.585102081298828e-05,
      "step": 90621,
      "training_step_time": 7.721720218658447
    },
    {
      "epoch": 1.5851974487304687e-05,
      "model_forward_time": 2.8833837509155273,
      "step": 90622
    },
    {
      "epoch": 1.5851974487304687e-05,
      "step": 90622,
      "training_step_time": 10.192543268203735
    },
    {
      "epoch": 1.5852928161621095e-05,
      "model_forward_time": 1.7346856594085693,
      "step": 90623
    },
    {
      "epoch": 1.5852928161621095e-05,
      "step": 90623,
      "training_step_time": 8.496841430664062
    },
    {
      "epoch": 1.58538818359375e-05,
      "model_forward_time": 1.7582066059112549,
      "step": 90624
    },
    {
      "epoch": 1.58538818359375e-05,
      "step": 90624,
      "training_step_time": 7.362234354019165
    },
    {
      "epoch": 1.5854835510253906e-05,
      "model_forward_time": 1.0630905628204346,
      "step": 90625
    },
    {
      "epoch": 1.5854835510253906e-05,
      "step": 90625,
      "training_step_time": 7.608995676040649
    },
    {
      "epoch": 1.5855789184570314e-05,
      "model_forward_time": 1.768427848815918,
      "step": 90626
    },
    {
      "epoch": 1.5855789184570314e-05,
      "step": 90626,
      "training_step_time": 6.932046175003052
    },
    {
      "epoch": 1.5856742858886718e-05,
      "model_forward_time": 1.7692725658416748,
      "step": 90627
    },
    {
      "epoch": 1.5856742858886718e-05,
      "step": 90627,
      "training_step_time": 6.871227741241455
    },
    {
      "epoch": 1.5857696533203125e-05,
      "model_forward_time": 1.7699980735778809,
      "step": 90628
    },
    {
      "epoch": 1.5857696533203125e-05,
      "step": 90628,
      "training_step_time": 6.85582160949707
    },
    {
      "epoch": 1.5858650207519533e-05,
      "model_forward_time": 1.7653493881225586,
      "step": 90629
    },
    {
      "epoch": 1.5858650207519533e-05,
      "step": 90629,
      "training_step_time": 6.900432586669922
    },
    {
      "epoch": 1.5859603881835937e-05,
      "grad_norm": 0.09950126707553864,
      "learning_rate": 2.3816944906667526e-07,
      "loss": 0.0539,
      "step": 90630
    },
    {
      "epoch": 1.5859603881835937e-05,
      "model_forward_time": 1.766183853149414,
      "step": 90630
    },
    {
      "dynamics_loss_avg": 0.05405763108283281,
      "epoch": 1.5859603881835937e-05,
      "step": 90630
    },
    {
      "action_loss_avg": 0.0012798429001122714,
      "epoch": 1.5859603881835937e-05,
      "step": 90630
    },
    {
      "epoch": 1.5859603881835937e-05,
      "step": 90630,
      "training_step_time": 7.521238565444946
    },
    {
      "epoch": 1.5860557556152344e-05,
      "model_forward_time": 1.8219938278198242,
      "step": 90631
    },
    {
      "epoch": 1.5860557556152344e-05,
      "step": 90631,
      "training_step_time": 8.794830560684204
    },
    {
      "epoch": 1.586151123046875e-05,
      "model_forward_time": 2.876152515411377,
      "step": 90632
    },
    {
      "epoch": 1.586151123046875e-05,
      "step": 90632,
      "training_step_time": 8.399330139160156
    },
    {
      "epoch": 1.5862464904785155e-05,
      "model_forward_time": 1.775991439819336,
      "step": 90633
    },
    {
      "epoch": 1.5862464904785155e-05,
      "step": 90633,
      "training_step_time": 7.621252775192261
    },
    {
      "epoch": 1.5863418579101563e-05,
      "model_forward_time": 1.7670047283172607,
      "step": 90634
    },
    {
      "epoch": 1.5863418579101563e-05,
      "step": 90634,
      "training_step_time": 7.655508756637573
    },
    {
      "epoch": 1.586437225341797e-05,
      "model_forward_time": 1.771568775177002,
      "step": 90635
    },
    {
      "epoch": 1.586437225341797e-05,
      "step": 90635,
      "training_step_time": 7.652604103088379
    },
    {
      "epoch": 1.5865325927734374e-05,
      "model_forward_time": 1.7643544673919678,
      "step": 90636
    },
    {
      "epoch": 1.5865325927734374e-05,
      "step": 90636,
      "training_step_time": 6.862310886383057
    },
    {
      "epoch": 1.586627960205078e-05,
      "model_forward_time": 1.7590069770812988,
      "step": 90637
    },
    {
      "epoch": 1.586627960205078e-05,
      "step": 90637,
      "training_step_time": 6.9105987548828125
    },
    {
      "epoch": 1.586723327636719e-05,
      "model_forward_time": 1.7743289470672607,
      "step": 90638
    },
    {
      "epoch": 1.586723327636719e-05,
      "step": 90638,
      "training_step_time": 6.906047105789185
    },
    {
      "epoch": 1.5868186950683593e-05,
      "model_forward_time": 1.7612264156341553,
      "step": 90639
    },
    {
      "epoch": 1.5868186950683593e-05,
      "step": 90639,
      "training_step_time": 7.45208215713501
    },
    {
      "epoch": 1.5869140625e-05,
      "grad_norm": 0.10932693630456924,
      "learning_rate": 2.3766547219890633e-07,
      "loss": 0.0535,
      "step": 90640
    },
    {
      "epoch": 1.5869140625e-05,
      "model_forward_time": 1.8383216857910156,
      "step": 90640
    },
    {
      "dynamics_loss_avg": 0.06931959465146065,
      "epoch": 1.5869140625e-05,
      "step": 90640
    },
    {
      "action_loss_avg": 0.0011499417858431115,
      "epoch": 1.5869140625e-05,
      "step": 90640
    },
    {
      "epoch": 1.5869140625e-05,
      "step": 90640,
      "training_step_time": 8.499255657196045
    },
    {
      "epoch": 1.5870094299316405e-05,
      "model_forward_time": 2.863679885864258,
      "step": 90641
    },
    {
      "epoch": 1.5870094299316405e-05,
      "step": 90641,
      "training_step_time": 9.446537017822266
    },
    {
      "epoch": 1.5871047973632812e-05,
      "model_forward_time": 1.7533044815063477,
      "step": 90642
    },
    {
      "epoch": 1.5871047973632812e-05,
      "step": 90642,
      "training_step_time": 7.686386823654175
    },
    {
      "epoch": 1.587200164794922e-05,
      "model_forward_time": 1.7714941501617432,
      "step": 90643
    },
    {
      "epoch": 1.587200164794922e-05,
      "step": 90643,
      "training_step_time": 8.46004033088684
    },
    {
      "epoch": 1.5872955322265623e-05,
      "model_forward_time": 1.7875967025756836,
      "step": 90644
    },
    {
      "epoch": 1.5872955322265623e-05,
      "step": 90644,
      "training_step_time": 7.650137662887573
    },
    {
      "epoch": 1.587390899658203e-05,
      "model_forward_time": 1.7674827575683594,
      "step": 90645
    },
    {
      "epoch": 1.587390899658203e-05,
      "step": 90645,
      "training_step_time": 6.93065071105957
    },
    {
      "epoch": 1.5874862670898438e-05,
      "model_forward_time": 1.7653188705444336,
      "step": 90646
    },
    {
      "epoch": 1.5874862670898438e-05,
      "step": 90646,
      "training_step_time": 6.873300313949585
    },
    {
      "epoch": 1.5875816345214842e-05,
      "model_forward_time": 1.7551496028900146,
      "step": 90647
    },
    {
      "epoch": 1.5875816345214842e-05,
      "step": 90647,
      "training_step_time": 6.913669109344482
    },
    {
      "epoch": 1.587677001953125e-05,
      "model_forward_time": 1.7743804454803467,
      "step": 90648
    },
    {
      "epoch": 1.587677001953125e-05,
      "step": 90648,
      "training_step_time": 7.988420248031616
    },
    {
      "epoch": 1.5877723693847657e-05,
      "model_forward_time": 1.8217124938964844,
      "step": 90649
    },
    {
      "epoch": 1.5877723693847657e-05,
      "step": 90649,
      "training_step_time": 9.187265872955322
    },
    {
      "epoch": 1.587867736816406e-05,
      "grad_norm": 0.08397579938173294,
      "learning_rate": 2.3716201613288191e-07,
      "loss": 0.0547,
      "step": 90650
    },
    {
      "epoch": 1.587867736816406e-05,
      "model_forward_time": 2.831007480621338,
      "step": 90650
    },
    {
      "dynamics_loss_avg": 0.05209286408498883,
      "epoch": 1.587867736816406e-05,
      "step": 90650
    },
    {
      "action_loss_avg": 0.0011725396907422692,
      "epoch": 1.587867736816406e-05,
      "step": 90650
    },
    {
      "epoch": 1.587867736816406e-05,
      "step": 90650,
      "training_step_time": 9.558859825134277
    },
    {
      "epoch": 1.587963104248047e-05,
      "model_forward_time": 1.7690205574035645,
      "step": 90651
    },
    {
      "epoch": 1.587963104248047e-05,
      "step": 90651,
      "training_step_time": 8.542034149169922
    },
    {
      "epoch": 1.5880584716796876e-05,
      "model_forward_time": 1.758847951889038,
      "step": 90652
    },
    {
      "epoch": 1.5880584716796876e-05,
      "step": 90652,
      "training_step_time": 7.941994905471802
    },
    {
      "epoch": 1.588153839111328e-05,
      "model_forward_time": 1.7784037590026855,
      "step": 90653
    },
    {
      "epoch": 1.588153839111328e-05,
      "step": 90653,
      "training_step_time": 7.6511664390563965
    },
    {
      "epoch": 1.5882492065429687e-05,
      "model_forward_time": 1.770829200744629,
      "step": 90654
    },
    {
      "epoch": 1.5882492065429687e-05,
      "step": 90654,
      "training_step_time": 6.865366220474243
    },
    {
      "epoch": 1.5883445739746095e-05,
      "model_forward_time": 1.7689392566680908,
      "step": 90655
    },
    {
      "epoch": 1.5883445739746095e-05,
      "step": 90655,
      "training_step_time": 6.830999374389648
    },
    {
      "epoch": 1.58843994140625e-05,
      "model_forward_time": 1.769031286239624,
      "step": 90656
    },
    {
      "epoch": 1.58843994140625e-05,
      "step": 90656,
      "training_step_time": 6.841834545135498
    },
    {
      "epoch": 1.5885353088378906e-05,
      "model_forward_time": 1.7711913585662842,
      "step": 90657
    },
    {
      "epoch": 1.5885353088378906e-05,
      "step": 90657,
      "training_step_time": 7.2414281368255615
    },
    {
      "epoch": 1.5886306762695314e-05,
      "model_forward_time": 1.830207109451294,
      "step": 90658
    },
    {
      "epoch": 1.5886306762695314e-05,
      "step": 90658,
      "training_step_time": 9.250113725662231
    },
    {
      "epoch": 1.5887260437011718e-05,
      "model_forward_time": 2.8288097381591797,
      "step": 90659
    },
    {
      "epoch": 1.5887260437011718e-05,
      "step": 90659,
      "training_step_time": 12.872709512710571
    },
    {
      "epoch": 1.5888214111328125e-05,
      "grad_norm": 0.07458467036485672,
      "learning_rate": 2.366590809236574e-07,
      "loss": 0.0545,
      "step": 90660
    },
    {
      "epoch": 1.5888214111328125e-05,
      "model_forward_time": 1.7686433792114258,
      "step": 90660
    },
    {
      "dynamics_loss_avg": 0.04249253142625094,
      "epoch": 1.5888214111328125e-05,
      "step": 90660
    },
    {
      "action_loss_avg": 0.0015562820073682815,
      "epoch": 1.5888214111328125e-05,
      "step": 90660
    },
    {
      "epoch": 1.5888214111328125e-05,
      "step": 90660,
      "training_step_time": 7.774362564086914
    },
    {
      "epoch": 1.5889167785644533e-05,
      "model_forward_time": 1.7702596187591553,
      "step": 90661
    },
    {
      "epoch": 1.5889167785644533e-05,
      "step": 90661,
      "training_step_time": 7.645979404449463
    },
    {
      "epoch": 1.5890121459960937e-05,
      "model_forward_time": 1.7651042938232422,
      "step": 90662
    },
    {
      "epoch": 1.5890121459960937e-05,
      "step": 90662,
      "training_step_time": 7.655911445617676
    },
    {
      "epoch": 1.5891075134277344e-05,
      "model_forward_time": 1.7632575035095215,
      "step": 90663
    },
    {
      "epoch": 1.5891075134277344e-05,
      "step": 90663,
      "training_step_time": 6.877466917037964
    },
    {
      "epoch": 1.589202880859375e-05,
      "model_forward_time": 1.7772853374481201,
      "step": 90664
    },
    {
      "epoch": 1.589202880859375e-05,
      "step": 90664,
      "training_step_time": 6.851823329925537
    },
    {
      "epoch": 1.5892982482910155e-05,
      "model_forward_time": 1.7677087783813477,
      "step": 90665
    },
    {
      "epoch": 1.5892982482910155e-05,
      "step": 90665,
      "training_step_time": 6.889951229095459
    },
    {
      "epoch": 1.5893936157226563e-05,
      "model_forward_time": 1.773468017578125,
      "step": 90666
    },
    {
      "epoch": 1.5893936157226563e-05,
      "step": 90666,
      "training_step_time": 7.222578763961792
    },
    {
      "epoch": 1.589488983154297e-05,
      "model_forward_time": 1.829101324081421,
      "step": 90667
    },
    {
      "epoch": 1.589488983154297e-05,
      "step": 90667,
      "training_step_time": 7.592707872390747
    },
    {
      "epoch": 1.5895843505859374e-05,
      "model_forward_time": 2.839893102645874,
      "step": 90668
    },
    {
      "epoch": 1.5895843505859374e-05,
      "step": 90668,
      "training_step_time": 10.693193435668945
    },
    {
      "epoch": 1.5896797180175782e-05,
      "model_forward_time": 1.760138988494873,
      "step": 90669
    },
    {
      "epoch": 1.5896797180175782e-05,
      "step": 90669,
      "training_step_time": 8.316205978393555
    },
    {
      "epoch": 1.589775085449219e-05,
      "grad_norm": 0.10260424017906189,
      "learning_rate": 2.3615666662623493e-07,
      "loss": 0.0538,
      "step": 90670
    },
    {
      "epoch": 1.589775085449219e-05,
      "model_forward_time": 1.7614655494689941,
      "step": 90670
    },
    {
      "dynamics_loss_avg": 0.047720663249492645,
      "epoch": 1.589775085449219e-05,
      "step": 90670
    },
    {
      "action_loss_avg": 0.0017958170370548031,
      "epoch": 1.589775085449219e-05,
      "step": 90670
    },
    {
      "epoch": 1.589775085449219e-05,
      "step": 90670,
      "training_step_time": 7.615704298019409
    },
    {
      "epoch": 1.5898704528808593e-05,
      "model_forward_time": 1.7716264724731445,
      "step": 90671
    },
    {
      "epoch": 1.5898704528808593e-05,
      "step": 90671,
      "training_step_time": 7.675398111343384
    },
    {
      "epoch": 1.5899658203125e-05,
      "model_forward_time": 1.7607381343841553,
      "step": 90672
    },
    {
      "epoch": 1.5899658203125e-05,
      "step": 90672,
      "training_step_time": 6.89326286315918
    },
    {
      "epoch": 1.5900611877441405e-05,
      "model_forward_time": 1.7606964111328125,
      "step": 90673
    },
    {
      "epoch": 1.5900611877441405e-05,
      "step": 90673,
      "training_step_time": 6.870855331420898
    },
    {
      "epoch": 1.5901565551757812e-05,
      "model_forward_time": 1.7579896450042725,
      "step": 90674
    },
    {
      "epoch": 1.5901565551757812e-05,
      "step": 90674,
      "training_step_time": 6.876723051071167
    },
    {
      "epoch": 1.590251922607422e-05,
      "model_forward_time": 1.7685561180114746,
      "step": 90675
    },
    {
      "epoch": 1.590251922607422e-05,
      "step": 90675,
      "training_step_time": 8.210781812667847
    },
    {
      "epoch": 1.5903472900390623e-05,
      "model_forward_time": 1.8364317417144775,
      "step": 90676
    },
    {
      "epoch": 1.5903472900390623e-05,
      "step": 90676,
      "training_step_time": 7.686561346054077
    },
    {
      "epoch": 1.590442657470703e-05,
      "model_forward_time": 1.7685704231262207,
      "step": 90677
    },
    {
      "epoch": 1.590442657470703e-05,
      "step": 90677,
      "training_step_time": 11.377130031585693
    },
    {
      "epoch": 1.5905380249023438e-05,
      "model_forward_time": 2.7910516262054443,
      "step": 90678
    },
    {
      "epoch": 1.5905380249023438e-05,
      "step": 90678,
      "training_step_time": 7.883969306945801
    },
    {
      "epoch": 1.5906333923339842e-05,
      "model_forward_time": 1.7386126518249512,
      "step": 90679
    },
    {
      "epoch": 1.5906333923339842e-05,
      "step": 90679,
      "training_step_time": 7.71256947517395
    },
    {
      "epoch": 1.590728759765625e-05,
      "grad_norm": 0.08593371510505676,
      "learning_rate": 2.3565477329555663e-07,
      "loss": 0.0541,
      "step": 90680
    },
    {
      "epoch": 1.590728759765625e-05,
      "model_forward_time": 1.7437424659729004,
      "step": 90680
    },
    {
      "dynamics_loss_avg": 0.04965461473912001,
      "epoch": 1.590728759765625e-05,
      "step": 90680
    },
    {
      "action_loss_avg": 0.0010130392212886364,
      "epoch": 1.590728759765625e-05,
      "step": 90680
    },
    {
      "epoch": 1.590728759765625e-05,
      "step": 90680,
      "training_step_time": 7.66983437538147
    },
    {
      "epoch": 1.5908241271972657e-05,
      "model_forward_time": 1.7745285034179688,
      "step": 90681
    },
    {
      "epoch": 1.5908241271972657e-05,
      "step": 90681,
      "training_step_time": 7.659902811050415
    },
    {
      "epoch": 1.590919494628906e-05,
      "model_forward_time": 1.76603364944458,
      "step": 90682
    },
    {
      "epoch": 1.590919494628906e-05,
      "step": 90682,
      "training_step_time": 6.936994314193726
    },
    {
      "epoch": 1.591014862060547e-05,
      "model_forward_time": 1.7622888088226318,
      "step": 90683
    },
    {
      "epoch": 1.591014862060547e-05,
      "step": 90683,
      "training_step_time": 6.877430438995361
    },
    {
      "epoch": 1.5911102294921876e-05,
      "model_forward_time": 1.7630083560943604,
      "step": 90684
    },
    {
      "epoch": 1.5911102294921876e-05,
      "step": 90684,
      "training_step_time": 7.2557759284973145
    },
    {
      "epoch": 1.591205596923828e-05,
      "model_forward_time": 1.8309898376464844,
      "step": 90685
    },
    {
      "epoch": 1.591205596923828e-05,
      "step": 90685,
      "training_step_time": 7.562025547027588
    },
    {
      "epoch": 1.5913009643554687e-05,
      "model_forward_time": 1.7767972946166992,
      "step": 90686
    },
    {
      "epoch": 1.5913009643554687e-05,
      "step": 90686,
      "training_step_time": 11.281409502029419
    },
    {
      "epoch": 1.5913963317871095e-05,
      "model_forward_time": 2.920658588409424,
      "step": 90687
    },
    {
      "epoch": 1.5913963317871095e-05,
      "step": 90687,
      "training_step_time": 10.885273933410645
    },
    {
      "epoch": 1.59149169921875e-05,
      "model_forward_time": 1.7618012428283691,
      "step": 90688
    },
    {
      "epoch": 1.59149169921875e-05,
      "step": 90688,
      "training_step_time": 7.669005870819092
    },
    {
      "epoch": 1.5915870666503906e-05,
      "model_forward_time": 1.7797338962554932,
      "step": 90689
    },
    {
      "epoch": 1.5915870666503906e-05,
      "step": 90689,
      "training_step_time": 7.667659044265747
    },
    {
      "epoch": 1.5916824340820314e-05,
      "grad_norm": 0.08953101187944412,
      "learning_rate": 2.3515340098650974e-07,
      "loss": 0.0542,
      "step": 90690
    },
    {
      "epoch": 1.5916824340820314e-05,
      "model_forward_time": 1.7682921886444092,
      "step": 90690
    },
    {
      "dynamics_loss_avg": 0.04689604043960571,
      "epoch": 1.5916824340820314e-05,
      "step": 90690
    },
    {
      "action_loss_avg": 0.0013840181840350852,
      "epoch": 1.5916824340820314e-05,
      "step": 90690
    },
    {
      "epoch": 1.5916824340820314e-05,
      "step": 90690,
      "training_step_time": 7.690240144729614
    },
    {
      "epoch": 1.5917778015136718e-05,
      "model_forward_time": 1.763777494430542,
      "step": 90691
    },
    {
      "epoch": 1.5917778015136718e-05,
      "step": 90691,
      "training_step_time": 6.892414093017578
    },
    {
      "epoch": 1.5918731689453125e-05,
      "model_forward_time": 1.7633397579193115,
      "step": 90692
    },
    {
      "epoch": 1.5918731689453125e-05,
      "step": 90692,
      "training_step_time": 6.896074295043945
    },
    {
      "epoch": 1.5919685363769533e-05,
      "model_forward_time": 1.7815122604370117,
      "step": 90693
    },
    {
      "epoch": 1.5919685363769533e-05,
      "step": 90693,
      "training_step_time": 6.92696738243103
    },
    {
      "epoch": 1.5920639038085937e-05,
      "model_forward_time": 1.8416578769683838,
      "step": 90694
    },
    {
      "epoch": 1.5920639038085937e-05,
      "step": 90694,
      "training_step_time": 7.5361151695251465
    },
    {
      "epoch": 1.5921592712402344e-05,
      "model_forward_time": 1.765669584274292,
      "step": 90695
    },
    {
      "epoch": 1.5921592712402344e-05,
      "step": 90695,
      "training_step_time": 8.712379693984985
    },
    {
      "epoch": 1.592254638671875e-05,
      "model_forward_time": 2.874107599258423,
      "step": 90696
    },
    {
      "epoch": 1.592254638671875e-05,
      "step": 90696,
      "training_step_time": 8.66586184501648
    },
    {
      "epoch": 1.5923500061035155e-05,
      "model_forward_time": 1.771625280380249,
      "step": 90697
    },
    {
      "epoch": 1.5923500061035155e-05,
      "step": 90697,
      "training_step_time": 7.740614175796509
    },
    {
      "epoch": 1.5924453735351563e-05,
      "model_forward_time": 1.7645347118377686,
      "step": 90698
    },
    {
      "epoch": 1.5924453735351563e-05,
      "step": 90698,
      "training_step_time": 8.433008432388306
    },
    {
      "epoch": 1.592540740966797e-05,
      "model_forward_time": 1.7784366607666016,
      "step": 90699
    },
    {
      "epoch": 1.592540740966797e-05,
      "step": 90699,
      "training_step_time": 7.639405012130737
    },
    {
      "epoch": 1.5926361083984374e-05,
      "grad_norm": 0.10013709217309952,
      "learning_rate": 2.346525497539226e-07,
      "loss": 0.055,
      "step": 90700
    },
    {
      "epoch": 1.5926361083984374e-05,
      "model_forward_time": 1.7671644687652588,
      "step": 90700
    },
    {
      "dynamics_loss_avg": 0.07667808048427105,
      "epoch": 1.5926361083984374e-05,
      "step": 90700
    },
    {
      "action_loss_avg": 0.0016953347949311137,
      "epoch": 1.5926361083984374e-05,
      "step": 90700
    },
    {
      "epoch": 1.5926361083984374e-05,
      "step": 90700,
      "training_step_time": 6.842303514480591
    },
    {
      "epoch": 1.5927314758300782e-05,
      "model_forward_time": 1.7698895931243896,
      "step": 90701
    },
    {
      "epoch": 1.5927314758300782e-05,
      "step": 90701,
      "training_step_time": 6.869213104248047
    },
    {
      "epoch": 1.592826843261719e-05,
      "model_forward_time": 1.7650773525238037,
      "step": 90702
    },
    {
      "epoch": 1.592826843261719e-05,
      "step": 90702,
      "training_step_time": 6.882257699966431
    },
    {
      "epoch": 1.5929222106933593e-05,
      "model_forward_time": 1.8275299072265625,
      "step": 90703
    },
    {
      "epoch": 1.5929222106933593e-05,
      "step": 90703,
      "training_step_time": 7.429463863372803
    },
    {
      "epoch": 1.593017578125e-05,
      "model_forward_time": 1.7656610012054443,
      "step": 90704
    },
    {
      "epoch": 1.593017578125e-05,
      "step": 90704,
      "training_step_time": 10.551904439926147
    },
    {
      "epoch": 1.5931129455566405e-05,
      "model_forward_time": 2.8613595962524414,
      "step": 90705
    },
    {
      "epoch": 1.5931129455566405e-05,
      "step": 90705,
      "training_step_time": 9.864020347595215
    },
    {
      "epoch": 1.5932083129882812e-05,
      "model_forward_time": 1.7611401081085205,
      "step": 90706
    },
    {
      "epoch": 1.5932083129882812e-05,
      "step": 90706,
      "training_step_time": 8.518539428710938
    },
    {
      "epoch": 1.593303680419922e-05,
      "model_forward_time": 1.7620103359222412,
      "step": 90707
    },
    {
      "epoch": 1.593303680419922e-05,
      "step": 90707,
      "training_step_time": 8.790279865264893
    },
    {
      "epoch": 1.5933990478515623e-05,
      "model_forward_time": 1.7714731693267822,
      "step": 90708
    },
    {
      "epoch": 1.5933990478515623e-05,
      "step": 90708,
      "training_step_time": 7.593931436538696
    },
    {
      "epoch": 1.593494415283203e-05,
      "model_forward_time": 1.7651379108428955,
      "step": 90709
    },
    {
      "epoch": 1.593494415283203e-05,
      "step": 90709,
      "training_step_time": 6.847779989242554
    },
    {
      "epoch": 1.593589782714844e-05,
      "grad_norm": 0.09465009719133377,
      "learning_rate": 2.3415221965256807e-07,
      "loss": 0.0562,
      "step": 90710
    },
    {
      "epoch": 1.593589782714844e-05,
      "model_forward_time": 1.7638587951660156,
      "step": 90710
    },
    {
      "dynamics_loss_avg": 0.04073939099907875,
      "epoch": 1.593589782714844e-05,
      "step": 90710
    },
    {
      "action_loss_avg": 0.0014571338309906424,
      "epoch": 1.593589782714844e-05,
      "step": 90710
    },
    {
      "epoch": 1.593589782714844e-05,
      "step": 90710,
      "training_step_time": 6.843174695968628
    },
    {
      "epoch": 1.5936851501464842e-05,
      "model_forward_time": 1.755258560180664,
      "step": 90711
    },
    {
      "epoch": 1.5936851501464842e-05,
      "step": 90711,
      "training_step_time": 6.838054895401001
    },
    {
      "epoch": 1.593780517578125e-05,
      "model_forward_time": 1.8368229866027832,
      "step": 90712
    },
    {
      "epoch": 1.593780517578125e-05,
      "step": 90712,
      "training_step_time": 7.477971076965332
    },
    {
      "epoch": 1.5938758850097657e-05,
      "model_forward_time": 1.7768852710723877,
      "step": 90713
    },
    {
      "epoch": 1.5938758850097657e-05,
      "step": 90713,
      "training_step_time": 8.12121295928955
    },
    {
      "epoch": 1.593971252441406e-05,
      "model_forward_time": 2.839048147201538,
      "step": 90714
    },
    {
      "epoch": 1.593971252441406e-05,
      "step": 90714,
      "training_step_time": 9.771140336990356
    },
    {
      "epoch": 1.594066619873047e-05,
      "model_forward_time": 1.7746555805206299,
      "step": 90715
    },
    {
      "epoch": 1.594066619873047e-05,
      "step": 90715,
      "training_step_time": 7.6853132247924805
    },
    {
      "epoch": 1.5941619873046876e-05,
      "model_forward_time": 1.7648272514343262,
      "step": 90716
    },
    {
      "epoch": 1.5941619873046876e-05,
      "step": 90716,
      "training_step_time": 7.56310248374939
    },
    {
      "epoch": 1.594257354736328e-05,
      "model_forward_time": 1.7627463340759277,
      "step": 90717
    },
    {
      "epoch": 1.594257354736328e-05,
      "step": 90717,
      "training_step_time": 7.682870149612427
    },
    {
      "epoch": 1.5943527221679687e-05,
      "model_forward_time": 1.7699854373931885,
      "step": 90718
    },
    {
      "epoch": 1.5943527221679687e-05,
      "step": 90718,
      "training_step_time": 6.792247533798218
    },
    {
      "epoch": 1.5944480895996095e-05,
      "model_forward_time": 1.7827250957489014,
      "step": 90719
    },
    {
      "epoch": 1.5944480895996095e-05,
      "step": 90719,
      "training_step_time": 6.872447490692139
    },
    {
      "epoch": 1.59454345703125e-05,
      "grad_norm": 0.0812244638800621,
      "learning_rate": 2.3365241073716238e-07,
      "loss": 0.0551,
      "step": 90720
    },
    {
      "epoch": 1.59454345703125e-05,
      "model_forward_time": 1.7588772773742676,
      "step": 90720
    },
    {
      "dynamics_loss_avg": 0.04538122545927763,
      "epoch": 1.59454345703125e-05,
      "step": 90720
    },
    {
      "action_loss_avg": 0.0015469871432287619,
      "epoch": 1.59454345703125e-05,
      "step": 90720
    },
    {
      "epoch": 1.59454345703125e-05,
      "step": 90720,
      "training_step_time": 6.920251846313477
    },
    {
      "epoch": 1.5946388244628906e-05,
      "model_forward_time": 1.8355929851531982,
      "step": 90721
    },
    {
      "epoch": 1.5946388244628906e-05,
      "step": 90721,
      "training_step_time": 7.615867853164673
    },
    {
      "epoch": 1.5947341918945314e-05,
      "model_forward_time": 1.7745535373687744,
      "step": 90722
    },
    {
      "epoch": 1.5947341918945314e-05,
      "step": 90722,
      "training_step_time": 8.120361328125
    },
    {
      "epoch": 1.5948295593261718e-05,
      "model_forward_time": 1.7622780799865723,
      "step": 90723
    },
    {
      "epoch": 1.5948295593261718e-05,
      "step": 90723,
      "training_step_time": 11.537868976593018
    },
    {
      "epoch": 1.5949249267578125e-05,
      "model_forward_time": 1.7722108364105225,
      "step": 90724
    },
    {
      "epoch": 1.5949249267578125e-05,
      "step": 90724,
      "training_step_time": 10.494724035263062
    },
    {
      "epoch": 1.5950202941894533e-05,
      "model_forward_time": 1.775970697402954,
      "step": 90725
    },
    {
      "epoch": 1.5950202941894533e-05,
      "step": 90725,
      "training_step_time": 8.734602689743042
    },
    {
      "epoch": 1.5951156616210937e-05,
      "model_forward_time": 1.7823762893676758,
      "step": 90726
    },
    {
      "epoch": 1.5951156616210937e-05,
      "step": 90726,
      "training_step_time": 7.649685859680176
    },
    {
      "epoch": 1.5952110290527344e-05,
      "model_forward_time": 1.7610688209533691,
      "step": 90727
    },
    {
      "epoch": 1.5952110290527344e-05,
      "step": 90727,
      "training_step_time": 6.881572723388672
    },
    {
      "epoch": 1.595306396484375e-05,
      "model_forward_time": 1.769134521484375,
      "step": 90728
    },
    {
      "epoch": 1.595306396484375e-05,
      "step": 90728,
      "training_step_time": 6.888690948486328
    },
    {
      "epoch": 1.5954017639160155e-05,
      "model_forward_time": 1.7685511112213135,
      "step": 90729
    },
    {
      "epoch": 1.5954017639160155e-05,
      "step": 90729,
      "training_step_time": 6.884446620941162
    },
    {
      "epoch": 1.5954971313476563e-05,
      "grad_norm": 0.2688559889793396,
      "learning_rate": 2.3315312306236181e-07,
      "loss": 0.0537,
      "step": 90730
    },
    {
      "epoch": 1.5954971313476563e-05,
      "model_forward_time": 1.840597152709961,
      "step": 90730
    },
    {
      "dynamics_loss_avg": 0.05452221836894751,
      "epoch": 1.5954971313476563e-05,
      "step": 90730
    },
    {
      "action_loss_avg": 0.002017561113461852,
      "epoch": 1.5954971313476563e-05,
      "step": 90730
    },
    {
      "epoch": 1.5954971313476563e-05,
      "step": 90730,
      "training_step_time": 8.137637376785278
    },
    {
      "epoch": 1.595592498779297e-05,
      "model_forward_time": 1.7773187160491943,
      "step": 90731
    },
    {
      "epoch": 1.595592498779297e-05,
      "step": 90731,
      "training_step_time": 8.608614683151245
    },
    {
      "epoch": 1.5956878662109374e-05,
      "model_forward_time": 1.7726473808288574,
      "step": 90732
    },
    {
      "epoch": 1.5956878662109374e-05,
      "step": 90732,
      "training_step_time": 10.656280517578125
    },
    {
      "epoch": 1.5957832336425782e-05,
      "model_forward_time": 2.811176300048828,
      "step": 90733
    },
    {
      "epoch": 1.5957832336425782e-05,
      "step": 90733,
      "training_step_time": 9.23669171333313
    },
    {
      "epoch": 1.595878601074219e-05,
      "model_forward_time": 1.7751188278198242,
      "step": 90734
    },
    {
      "epoch": 1.595878601074219e-05,
      "step": 90734,
      "training_step_time": 7.875167369842529
    },
    {
      "epoch": 1.5959739685058593e-05,
      "model_forward_time": 1.772517204284668,
      "step": 90735
    },
    {
      "epoch": 1.5959739685058593e-05,
      "step": 90735,
      "training_step_time": 7.70967435836792
    },
    {
      "epoch": 1.5960693359375e-05,
      "model_forward_time": 1.7694659233093262,
      "step": 90736
    },
    {
      "epoch": 1.5960693359375e-05,
      "step": 90736,
      "training_step_time": 6.914423227310181
    },
    {
      "epoch": 1.5961647033691405e-05,
      "model_forward_time": 1.7723791599273682,
      "step": 90737
    },
    {
      "epoch": 1.5961647033691405e-05,
      "step": 90737,
      "training_step_time": 6.877521276473999
    },
    {
      "epoch": 1.5962600708007812e-05,
      "model_forward_time": 1.7744543552398682,
      "step": 90738
    },
    {
      "epoch": 1.5962600708007812e-05,
      "step": 90738,
      "training_step_time": 6.863393068313599
    },
    {
      "epoch": 1.596355438232422e-05,
      "model_forward_time": 1.8436534404754639,
      "step": 90739
    },
    {
      "epoch": 1.596355438232422e-05,
      "step": 90739,
      "training_step_time": 7.591706037521362
    },
    {
      "epoch": 1.5964508056640624e-05,
      "grad_norm": 0.11212926357984543,
      "learning_rate": 2.3265435668276992e-07,
      "loss": 0.0536,
      "step": 90740
    },
    {
      "epoch": 1.5964508056640624e-05,
      "model_forward_time": 1.7743101119995117,
      "step": 90740
    },
    {
      "dynamics_loss_avg": 0.06158677116036415,
      "epoch": 1.5964508056640624e-05,
      "step": 90740
    },
    {
      "action_loss_avg": 0.0010868685028981417,
      "epoch": 1.5964508056640624e-05,
      "step": 90740
    },
    {
      "epoch": 1.5964508056640624e-05,
      "step": 90740,
      "training_step_time": 8.383389234542847
    },
    {
      "epoch": 1.596546173095703e-05,
      "model_forward_time": 1.7672288417816162,
      "step": 90741
    },
    {
      "epoch": 1.596546173095703e-05,
      "step": 90741,
      "training_step_time": 10.681455135345459
    },
    {
      "epoch": 1.596641540527344e-05,
      "model_forward_time": 2.8685762882232666,
      "step": 90742
    },
    {
      "epoch": 1.596641540527344e-05,
      "step": 90742,
      "training_step_time": 8.630446434020996
    },
    {
      "epoch": 1.5967369079589842e-05,
      "model_forward_time": 1.7626621723175049,
      "step": 90743
    },
    {
      "epoch": 1.5967369079589842e-05,
      "step": 90743,
      "training_step_time": 8.5715172290802
    },
    {
      "epoch": 1.596832275390625e-05,
      "model_forward_time": 1.7847628593444824,
      "step": 90744
    },
    {
      "epoch": 1.596832275390625e-05,
      "step": 90744,
      "training_step_time": 7.643221139907837
    },
    {
      "epoch": 1.5969276428222657e-05,
      "model_forward_time": 1.785524845123291,
      "step": 90745
    },
    {
      "epoch": 1.5969276428222657e-05,
      "step": 90745,
      "training_step_time": 7.6790382862091064
    },
    {
      "epoch": 1.597023010253906e-05,
      "model_forward_time": 1.7772035598754883,
      "step": 90746
    },
    {
      "epoch": 1.597023010253906e-05,
      "step": 90746,
      "training_step_time": 6.866220235824585
    },
    {
      "epoch": 1.597118377685547e-05,
      "model_forward_time": 1.773576259613037,
      "step": 90747
    },
    {
      "epoch": 1.597118377685547e-05,
      "step": 90747,
      "training_step_time": 6.707391023635864
    },
    {
      "epoch": 1.5972137451171876e-05,
      "model_forward_time": 1.8372414112091064,
      "step": 90748
    },
    {
      "epoch": 1.5972137451171876e-05,
      "step": 90748,
      "training_step_time": 7.546850681304932
    },
    {
      "epoch": 1.597309112548828e-05,
      "model_forward_time": 1.7683982849121094,
      "step": 90749
    },
    {
      "epoch": 1.597309112548828e-05,
      "step": 90749,
      "training_step_time": 8.272510290145874
    },
    {
      "epoch": 1.5974044799804688e-05,
      "grad_norm": 0.07524780929088593,
      "learning_rate": 2.321561116529286e-07,
      "loss": 0.0541,
      "step": 90750
    },
    {
      "epoch": 1.5974044799804688e-05,
      "model_forward_time": 1.772409200668335,
      "step": 90750
    },
    {
      "dynamics_loss_avg": 0.06363156139850616,
      "epoch": 1.5974044799804688e-05,
      "step": 90750
    },
    {
      "action_loss_avg": 0.001813544292235747,
      "epoch": 1.5974044799804688e-05,
      "step": 90750
    },
    {
      "epoch": 1.5974044799804688e-05,
      "step": 90750,
      "training_step_time": 9.190674543380737
    },
    {
      "epoch": 1.5974998474121095e-05,
      "model_forward_time": 2.877939224243164,
      "step": 90751
    },
    {
      "epoch": 1.5974998474121095e-05,
      "step": 90751,
      "training_step_time": 9.197362184524536
    },
    {
      "epoch": 1.59759521484375e-05,
      "model_forward_time": 1.7687933444976807,
      "step": 90752
    },
    {
      "epoch": 1.59759521484375e-05,
      "step": 90752,
      "training_step_time": 7.9446821212768555
    },
    {
      "epoch": 1.5976905822753906e-05,
      "model_forward_time": 1.756723165512085,
      "step": 90753
    },
    {
      "epoch": 1.5976905822753906e-05,
      "step": 90753,
      "training_step_time": 8.192396879196167
    },
    {
      "epoch": 1.5977859497070314e-05,
      "model_forward_time": 1.7662956714630127,
      "step": 90754
    },
    {
      "epoch": 1.5977859497070314e-05,
      "step": 90754,
      "training_step_time": 7.661696434020996
    },
    {
      "epoch": 1.5978813171386718e-05,
      "model_forward_time": 1.7688167095184326,
      "step": 90755
    },
    {
      "epoch": 1.5978813171386718e-05,
      "step": 90755,
      "training_step_time": 6.802381753921509
    },
    {
      "epoch": 1.5979766845703125e-05,
      "model_forward_time": 1.7699916362762451,
      "step": 90756
    },
    {
      "epoch": 1.5979766845703125e-05,
      "step": 90756,
      "training_step_time": 6.838684558868408
    },
    {
      "epoch": 1.5980720520019533e-05,
      "model_forward_time": 1.8559589385986328,
      "step": 90757
    },
    {
      "epoch": 1.5980720520019533e-05,
      "step": 90757,
      "training_step_time": 7.487879514694214
    },
    {
      "epoch": 1.5981674194335937e-05,
      "model_forward_time": 1.7784991264343262,
      "step": 90758
    },
    {
      "epoch": 1.5981674194335937e-05,
      "step": 90758,
      "training_step_time": 7.588232040405273
    },
    {
      "epoch": 1.5982627868652344e-05,
      "model_forward_time": 1.7706537246704102,
      "step": 90759
    },
    {
      "epoch": 1.5982627868652344e-05,
      "step": 90759,
      "training_step_time": 9.267658233642578
    },
    {
      "epoch": 1.598358154296875e-05,
      "grad_norm": 0.09582531452178955,
      "learning_rate": 2.3165838802732764e-07,
      "loss": 0.0538,
      "step": 90760
    },
    {
      "epoch": 1.598358154296875e-05,
      "model_forward_time": 2.88088321685791,
      "step": 90760
    },
    {
      "dynamics_loss_avg": 0.05150846280157566,
      "epoch": 1.598358154296875e-05,
      "step": 90760
    },
    {
      "action_loss_avg": 0.001097929588286206,
      "epoch": 1.598358154296875e-05,
      "step": 90760
    },
    {
      "epoch": 1.598358154296875e-05,
      "step": 90760,
      "training_step_time": 9.338849782943726
    },
    {
      "epoch": 1.5984535217285156e-05,
      "model_forward_time": 1.7666690349578857,
      "step": 90761
    },
    {
      "epoch": 1.5984535217285156e-05,
      "step": 90761,
      "training_step_time": 8.771827459335327
    },
    {
      "epoch": 1.5985488891601563e-05,
      "model_forward_time": 1.7907390594482422,
      "step": 90762
    },
    {
      "epoch": 1.5985488891601563e-05,
      "step": 90762,
      "training_step_time": 7.887683868408203
    },
    {
      "epoch": 1.598644256591797e-05,
      "model_forward_time": 1.771040678024292,
      "step": 90763
    },
    {
      "epoch": 1.598644256591797e-05,
      "step": 90763,
      "training_step_time": 7.660287618637085
    },
    {
      "epoch": 1.5987396240234374e-05,
      "model_forward_time": 1.773923397064209,
      "step": 90764
    },
    {
      "epoch": 1.5987396240234374e-05,
      "step": 90764,
      "training_step_time": 6.893273830413818
    },
    {
      "epoch": 1.5988349914550782e-05,
      "model_forward_time": 1.773183822631836,
      "step": 90765
    },
    {
      "epoch": 1.5988349914550782e-05,
      "step": 90765,
      "training_step_time": 6.890031814575195
    },
    {
      "epoch": 1.5989303588867186e-05,
      "model_forward_time": 1.8204889297485352,
      "step": 90766
    },
    {
      "epoch": 1.5989303588867186e-05,
      "step": 90766,
      "training_step_time": 7.539588928222656
    },
    {
      "epoch": 1.5990257263183593e-05,
      "model_forward_time": 1.7835595607757568,
      "step": 90767
    },
    {
      "epoch": 1.5990257263183593e-05,
      "step": 90767,
      "training_step_time": 7.643894195556641
    },
    {
      "epoch": 1.59912109375e-05,
      "model_forward_time": 1.7752795219421387,
      "step": 90768
    },
    {
      "epoch": 1.59912109375e-05,
      "step": 90768,
      "training_step_time": 8.57910966873169
    },
    {
      "epoch": 1.5992164611816405e-05,
      "model_forward_time": 2.865493059158325,
      "step": 90769
    },
    {
      "epoch": 1.5992164611816405e-05,
      "step": 90769,
      "training_step_time": 9.609642744064331
    },
    {
      "epoch": 1.5993118286132812e-05,
      "grad_norm": 0.06984519213438034,
      "learning_rate": 2.311611858603957e-07,
      "loss": 0.0536,
      "step": 90770
    },
    {
      "epoch": 1.5993118286132812e-05,
      "model_forward_time": 1.768073558807373,
      "step": 90770
    },
    {
      "dynamics_loss_avg": 0.06301676984876395,
      "epoch": 1.5993118286132812e-05,
      "step": 90770
    },
    {
      "action_loss_avg": 0.0015009258582722395,
      "epoch": 1.5993118286132812e-05,
      "step": 90770
    },
    {
      "epoch": 1.5993118286132812e-05,
      "step": 90770,
      "training_step_time": 8.697142601013184
    },
    {
      "epoch": 1.599407196044922e-05,
      "model_forward_time": 1.7849159240722656,
      "step": 90771
    },
    {
      "epoch": 1.599407196044922e-05,
      "step": 90771,
      "training_step_time": 8.498094081878662
    },
    {
      "epoch": 1.5995025634765624e-05,
      "model_forward_time": 1.775576114654541,
      "step": 90772
    },
    {
      "epoch": 1.5995025634765624e-05,
      "step": 90772,
      "training_step_time": 8.608339548110962
    },
    {
      "epoch": 1.599597930908203e-05,
      "model_forward_time": 1.7721896171569824,
      "step": 90773
    },
    {
      "epoch": 1.599597930908203e-05,
      "step": 90773,
      "training_step_time": 6.921201467514038
    },
    {
      "epoch": 1.599693298339844e-05,
      "model_forward_time": 1.783179759979248,
      "step": 90774
    },
    {
      "epoch": 1.599693298339844e-05,
      "step": 90774,
      "training_step_time": 6.923335075378418
    },
    {
      "epoch": 1.5997886657714842e-05,
      "model_forward_time": 1.7906837463378906,
      "step": 90775
    },
    {
      "epoch": 1.5997886657714842e-05,
      "step": 90775,
      "training_step_time": 7.527928352355957
    },
    {
      "epoch": 1.599884033203125e-05,
      "model_forward_time": 1.7736353874206543,
      "step": 90776
    },
    {
      "epoch": 1.599884033203125e-05,
      "step": 90776,
      "training_step_time": 7.602086067199707
    },
    {
      "epoch": 1.5999794006347657e-05,
      "model_forward_time": 1.7730233669281006,
      "step": 90777
    },
    {
      "epoch": 1.5999794006347657e-05,
      "step": 90777,
      "training_step_time": 8.243985652923584
    },
    {
      "epoch": 1.600074768066406e-05,
      "model_forward_time": 1.7770497798919678,
      "step": 90778
    },
    {
      "epoch": 1.600074768066406e-05,
      "step": 90778,
      "training_step_time": 9.455274820327759
    },
    {
      "epoch": 1.600170135498047e-05,
      "model_forward_time": 1.7711701393127441,
      "step": 90779
    },
    {
      "epoch": 1.600170135498047e-05,
      "step": 90779,
      "training_step_time": 10.150454759597778
    },
    {
      "epoch": 1.6002655029296876e-05,
      "grad_norm": 0.07740712910890579,
      "learning_rate": 2.3066450520650762e-07,
      "loss": 0.0543,
      "step": 90780
    },
    {
      "epoch": 1.6002655029296876e-05,
      "model_forward_time": 1.4041857719421387,
      "step": 90780
    },
    {
      "dynamics_loss_avg": 0.0675804603844881,
      "epoch": 1.6002655029296876e-05,
      "step": 90780
    },
    {
      "action_loss_avg": 0.001348969218088314,
      "epoch": 1.6002655029296876e-05,
      "step": 90780
    },
    {
      "epoch": 1.6002655029296876e-05,
      "step": 90780,
      "training_step_time": 8.526665687561035
    },
    {
      "epoch": 1.600360870361328e-05,
      "model_forward_time": 1.779752254486084,
      "step": 90781
    },
    {
      "epoch": 1.600360870361328e-05,
      "step": 90781,
      "training_step_time": 7.694855213165283
    },
    {
      "epoch": 1.6004562377929688e-05,
      "model_forward_time": 1.7804131507873535,
      "step": 90782
    },
    {
      "epoch": 1.6004562377929688e-05,
      "step": 90782,
      "training_step_time": 7.577805519104004
    },
    {
      "epoch": 1.6005516052246095e-05,
      "model_forward_time": 1.763230562210083,
      "step": 90783
    },
    {
      "epoch": 1.6005516052246095e-05,
      "step": 90783,
      "training_step_time": 6.856605291366577
    },
    {
      "epoch": 1.60064697265625e-05,
      "model_forward_time": 1.7728290557861328,
      "step": 90784
    },
    {
      "epoch": 1.60064697265625e-05,
      "step": 90784,
      "training_step_time": 7.329260349273682
    },
    {
      "epoch": 1.6007423400878906e-05,
      "model_forward_time": 1.8366332054138184,
      "step": 90785
    },
    {
      "epoch": 1.6007423400878906e-05,
      "step": 90785,
      "training_step_time": 8.131789445877075
    },
    {
      "epoch": 1.6008377075195314e-05,
      "model_forward_time": 1.7764484882354736,
      "step": 90786
    },
    {
      "epoch": 1.6008377075195314e-05,
      "step": 90786,
      "training_step_time": 7.668595790863037
    },
    {
      "epoch": 1.6009330749511718e-05,
      "model_forward_time": 1.7690463066101074,
      "step": 90787
    },
    {
      "epoch": 1.6009330749511718e-05,
      "step": 90787,
      "training_step_time": 8.399509906768799
    },
    {
      "epoch": 1.6010284423828125e-05,
      "model_forward_time": 2.904886245727539,
      "step": 90788
    },
    {
      "epoch": 1.6010284423828125e-05,
      "step": 90788,
      "training_step_time": 8.419402837753296
    },
    {
      "epoch": 1.6011238098144533e-05,
      "model_forward_time": 1.7792272567749023,
      "step": 90789
    },
    {
      "epoch": 1.6011238098144533e-05,
      "step": 90789,
      "training_step_time": 7.739176034927368
    },
    {
      "epoch": 1.6012191772460937e-05,
      "grad_norm": 0.09012946486473083,
      "learning_rate": 2.3016834611997773e-07,
      "loss": 0.0535,
      "step": 90790
    },
    {
      "epoch": 1.6012191772460937e-05,
      "model_forward_time": 1.7805719375610352,
      "step": 90790
    },
    {
      "dynamics_loss_avg": 0.05393243040889502,
      "epoch": 1.6012191772460937e-05,
      "step": 90790
    },
    {
      "action_loss_avg": 0.001297488302225247,
      "epoch": 1.6012191772460937e-05,
      "step": 90790
    },
    {
      "epoch": 1.6012191772460937e-05,
      "step": 90790,
      "training_step_time": 7.969284296035767
    },
    {
      "epoch": 1.6013145446777344e-05,
      "model_forward_time": 1.7703847885131836,
      "step": 90791
    },
    {
      "epoch": 1.6013145446777344e-05,
      "step": 90791,
      "training_step_time": 7.491146087646484
    },
    {
      "epoch": 1.601409912109375e-05,
      "model_forward_time": 1.7802042961120605,
      "step": 90792
    },
    {
      "epoch": 1.601409912109375e-05,
      "step": 90792,
      "training_step_time": 6.8906378746032715
    },
    {
      "epoch": 1.6015052795410156e-05,
      "model_forward_time": 1.7789156436920166,
      "step": 90793
    },
    {
      "epoch": 1.6015052795410156e-05,
      "step": 90793,
      "training_step_time": 7.0965235233306885
    },
    {
      "epoch": 1.6016006469726563e-05,
      "model_forward_time": 1.842940330505371,
      "step": 90794
    },
    {
      "epoch": 1.6016006469726563e-05,
      "step": 90794,
      "training_step_time": 7.571969032287598
    },
    {
      "epoch": 1.601696014404297e-05,
      "model_forward_time": 1.7690563201904297,
      "step": 90795
    },
    {
      "epoch": 1.601696014404297e-05,
      "step": 90795,
      "training_step_time": 8.169172286987305
    },
    {
      "epoch": 1.6017913818359374e-05,
      "model_forward_time": 1.7813224792480469,
      "step": 90796
    },
    {
      "epoch": 1.6017913818359374e-05,
      "step": 90796,
      "training_step_time": 7.666267156600952
    },
    {
      "epoch": 1.6018867492675782e-05,
      "model_forward_time": 2.8237295150756836,
      "step": 90797
    },
    {
      "epoch": 1.6018867492675782e-05,
      "step": 90797,
      "training_step_time": 9.962892770767212
    },
    {
      "epoch": 1.6019821166992186e-05,
      "model_forward_time": 1.747999906539917,
      "step": 90798
    },
    {
      "epoch": 1.6019821166992186e-05,
      "step": 90798,
      "training_step_time": 9.242615461349487
    },
    {
      "epoch": 1.6020774841308593e-05,
      "model_forward_time": 1.7485580444335938,
      "step": 90799
    },
    {
      "epoch": 1.6020774841308593e-05,
      "step": 90799,
      "training_step_time": 8.58299970626831
    },
    {
      "epoch": 1.6021728515625e-05,
      "grad_norm": 0.09181012213230133,
      "learning_rate": 2.2967270865506653e-07,
      "loss": 0.0538,
      "step": 90800
    },
    {
      "epoch": 1.6021728515625e-05,
      "model_forward_time": 1.7443132400512695,
      "step": 90800
    },
    {
      "dynamics_loss_avg": 0.04548176601529121,
      "epoch": 1.6021728515625e-05,
      "step": 90800
    },
    {
      "action_loss_avg": 0.001210357988020405,
      "epoch": 1.6021728515625e-05,
      "step": 90800
    },
    {
      "epoch": 1.6021728515625e-05,
      "step": 90800,
      "training_step_time": 7.43071985244751
    },
    {
      "epoch": 1.6022682189941405e-05,
      "model_forward_time": 1.751354694366455,
      "step": 90801
    },
    {
      "epoch": 1.6022682189941405e-05,
      "step": 90801,
      "training_step_time": 6.865819692611694
    },
    {
      "epoch": 1.6023635864257812e-05,
      "model_forward_time": 1.7492234706878662,
      "step": 90802
    },
    {
      "epoch": 1.6023635864257812e-05,
      "step": 90802,
      "training_step_time": 6.892827033996582
    },
    {
      "epoch": 1.602458953857422e-05,
      "model_forward_time": 1.8122916221618652,
      "step": 90803
    },
    {
      "epoch": 1.602458953857422e-05,
      "step": 90803,
      "training_step_time": 7.581343650817871
    },
    {
      "epoch": 1.6025543212890624e-05,
      "model_forward_time": 1.7591471672058105,
      "step": 90804
    },
    {
      "epoch": 1.6025543212890624e-05,
      "step": 90804,
      "training_step_time": 7.581308126449585
    },
    {
      "epoch": 1.602649688720703e-05,
      "model_forward_time": 1.7724609375,
      "step": 90805
    },
    {
      "epoch": 1.602649688720703e-05,
      "step": 90805,
      "training_step_time": 7.720226287841797
    },
    {
      "epoch": 1.602745056152344e-05,
      "model_forward_time": 2.852647066116333,
      "step": 90806
    },
    {
      "epoch": 1.602745056152344e-05,
      "step": 90806,
      "training_step_time": 9.527451038360596
    },
    {
      "epoch": 1.6028404235839842e-05,
      "model_forward_time": 1.7690536975860596,
      "step": 90807
    },
    {
      "epoch": 1.6028404235839842e-05,
      "step": 90807,
      "training_step_time": 7.703402757644653
    },
    {
      "epoch": 1.602935791015625e-05,
      "model_forward_time": 1.782195806503296,
      "step": 90808
    },
    {
      "epoch": 1.602935791015625e-05,
      "step": 90808,
      "training_step_time": 7.869316101074219
    },
    {
      "epoch": 1.6030311584472657e-05,
      "model_forward_time": 1.76979398727417,
      "step": 90809
    },
    {
      "epoch": 1.6030311584472657e-05,
      "step": 90809,
      "training_step_time": 7.637208938598633
    },
    {
      "epoch": 1.603126525878906e-05,
      "grad_norm": 0.06250456720590591,
      "learning_rate": 2.2917759286597508e-07,
      "loss": 0.0531,
      "step": 90810
    },
    {
      "epoch": 1.603126525878906e-05,
      "model_forward_time": 1.7660200595855713,
      "step": 90810
    },
    {
      "dynamics_loss_avg": 0.05126053178682923,
      "epoch": 1.603126525878906e-05,
      "step": 90810
    },
    {
      "action_loss_avg": 0.0016105073067592458,
      "epoch": 1.603126525878906e-05,
      "step": 90810
    },
    {
      "epoch": 1.603126525878906e-05,
      "step": 90810,
      "training_step_time": 7.566859722137451
    },
    {
      "epoch": 1.603221893310547e-05,
      "model_forward_time": 1.7756068706512451,
      "step": 90811
    },
    {
      "epoch": 1.603221893310547e-05,
      "step": 90811,
      "training_step_time": 6.865543365478516
    },
    {
      "epoch": 1.6033172607421876e-05,
      "model_forward_time": 1.8351528644561768,
      "step": 90812
    },
    {
      "epoch": 1.6033172607421876e-05,
      "step": 90812,
      "training_step_time": 7.553165912628174
    },
    {
      "epoch": 1.603412628173828e-05,
      "model_forward_time": 1.7729051113128662,
      "step": 90813
    },
    {
      "epoch": 1.603412628173828e-05,
      "step": 90813,
      "training_step_time": 7.560264587402344
    },
    {
      "epoch": 1.6035079956054688e-05,
      "model_forward_time": 1.7692692279815674,
      "step": 90814
    },
    {
      "epoch": 1.6035079956054688e-05,
      "step": 90814,
      "training_step_time": 7.672448396682739
    },
    {
      "epoch": 1.6036033630371095e-05,
      "model_forward_time": 2.854837656021118,
      "step": 90815
    },
    {
      "epoch": 1.6036033630371095e-05,
      "step": 90815,
      "training_step_time": 9.650704145431519
    },
    {
      "epoch": 1.60369873046875e-05,
      "model_forward_time": 1.7661025524139404,
      "step": 90816
    },
    {
      "epoch": 1.60369873046875e-05,
      "step": 90816,
      "training_step_time": 9.488833665847778
    },
    {
      "epoch": 1.6037940979003906e-05,
      "model_forward_time": 1.7772912979125977,
      "step": 90817
    },
    {
      "epoch": 1.6037940979003906e-05,
      "step": 90817,
      "training_step_time": 8.517971992492676
    },
    {
      "epoch": 1.6038894653320314e-05,
      "model_forward_time": 1.7697057723999023,
      "step": 90818
    },
    {
      "epoch": 1.6038894653320314e-05,
      "step": 90818,
      "training_step_time": 7.865244150161743
    },
    {
      "epoch": 1.6039848327636718e-05,
      "model_forward_time": 1.774170160293579,
      "step": 90819
    },
    {
      "epoch": 1.6039848327636718e-05,
      "step": 90819,
      "training_step_time": 7.516058444976807
    },
    {
      "epoch": 1.6040802001953125e-05,
      "grad_norm": 5.268599033355713,
      "learning_rate": 2.2868299880685008e-07,
      "loss": 0.0553,
      "step": 90820
    },
    {
      "epoch": 1.6040802001953125e-05,
      "model_forward_time": 1.7889626026153564,
      "step": 90820
    },
    {
      "dynamics_loss_avg": 0.05510715954005718,
      "epoch": 1.6040802001953125e-05,
      "step": 90820
    },
    {
      "action_loss_avg": 0.0011332625872455537,
      "epoch": 1.6040802001953125e-05,
      "step": 90820
    },
    {
      "epoch": 1.6040802001953125e-05,
      "step": 90820,
      "training_step_time": 6.8687591552734375
    },
    {
      "epoch": 1.6041755676269533e-05,
      "model_forward_time": 1.800541877746582,
      "step": 90821
    },
    {
      "epoch": 1.6041755676269533e-05,
      "step": 90821,
      "training_step_time": 7.753528356552124
    },
    {
      "epoch": 1.6042709350585937e-05,
      "model_forward_time": 1.7695472240447998,
      "step": 90822
    },
    {
      "epoch": 1.6042709350585937e-05,
      "step": 90822,
      "training_step_time": 8.66185212135315
    },
    {
      "epoch": 1.6043663024902344e-05,
      "model_forward_time": 1.7706263065338135,
      "step": 90823
    },
    {
      "epoch": 1.6043663024902344e-05,
      "step": 90823,
      "training_step_time": 7.778661489486694
    },
    {
      "epoch": 1.604461669921875e-05,
      "model_forward_time": 1.7636973857879639,
      "step": 90824
    },
    {
      "epoch": 1.604461669921875e-05,
      "step": 90824,
      "training_step_time": 11.114487171173096
    },
    {
      "epoch": 1.6045570373535156e-05,
      "model_forward_time": 2.845858573913574,
      "step": 90825
    },
    {
      "epoch": 1.6045570373535156e-05,
      "step": 90825,
      "training_step_time": 8.37438416481018
    },
    {
      "epoch": 1.6046524047851563e-05,
      "model_forward_time": 1.7372355461120605,
      "step": 90826
    },
    {
      "epoch": 1.6046524047851563e-05,
      "step": 90826,
      "training_step_time": 8.326174020767212
    },
    {
      "epoch": 1.604747772216797e-05,
      "model_forward_time": 1.7457492351531982,
      "step": 90827
    },
    {
      "epoch": 1.604747772216797e-05,
      "step": 90827,
      "training_step_time": 7.951913356781006
    },
    {
      "epoch": 1.6048431396484374e-05,
      "model_forward_time": 1.7285633087158203,
      "step": 90828
    },
    {
      "epoch": 1.6048431396484374e-05,
      "step": 90828,
      "training_step_time": 7.474620342254639
    },
    {
      "epoch": 1.6049385070800782e-05,
      "model_forward_time": 1.7543258666992188,
      "step": 90829
    },
    {
      "epoch": 1.6049385070800782e-05,
      "step": 90829,
      "training_step_time": 6.91757345199585
    },
    {
      "epoch": 1.6050338745117186e-05,
      "grad_norm": 0.07020352780818939,
      "learning_rate": 2.2818892653177827e-07,
      "loss": 0.055,
      "step": 90830
    },
    {
      "epoch": 1.6050338745117186e-05,
      "model_forward_time": 1.7486042976379395,
      "step": 90830
    },
    {
      "dynamics_loss_avg": 0.06155639924108982,
      "epoch": 1.6050338745117186e-05,
      "step": 90830
    },
    {
      "action_loss_avg": 0.0014694802172016352,
      "epoch": 1.6050338745117186e-05,
      "step": 90830
    },
    {
      "epoch": 1.6050338745117186e-05,
      "step": 90830,
      "training_step_time": 7.667619705200195
    },
    {
      "epoch": 1.6051292419433593e-05,
      "model_forward_time": 1.8140995502471924,
      "step": 90831
    },
    {
      "epoch": 1.6051292419433593e-05,
      "step": 90831,
      "training_step_time": 8.225300312042236
    },
    {
      "epoch": 1.605224609375e-05,
      "model_forward_time": 1.755910873413086,
      "step": 90832
    },
    {
      "epoch": 1.605224609375e-05,
      "step": 90832,
      "training_step_time": 8.728030443191528
    },
    {
      "epoch": 1.6053199768066405e-05,
      "model_forward_time": 1.7416844367980957,
      "step": 90833
    },
    {
      "epoch": 1.6053199768066405e-05,
      "step": 90833,
      "training_step_time": 9.626935482025146
    },
    {
      "epoch": 1.6054153442382812e-05,
      "model_forward_time": 2.8537821769714355,
      "step": 90834
    },
    {
      "epoch": 1.6054153442382812e-05,
      "step": 90834,
      "training_step_time": 8.748497486114502
    },
    {
      "epoch": 1.605510711669922e-05,
      "model_forward_time": 1.744582176208496,
      "step": 90835
    },
    {
      "epoch": 1.605510711669922e-05,
      "step": 90835,
      "training_step_time": 7.931521892547607
    },
    {
      "epoch": 1.6056060791015624e-05,
      "model_forward_time": 1.7484607696533203,
      "step": 90836
    },
    {
      "epoch": 1.6056060791015624e-05,
      "step": 90836,
      "training_step_time": 9.037003755569458
    },
    {
      "epoch": 1.605701446533203e-05,
      "model_forward_time": 1.7487523555755615,
      "step": 90837
    },
    {
      "epoch": 1.605701446533203e-05,
      "step": 90837,
      "training_step_time": 8.517917156219482
    },
    {
      "epoch": 1.605796813964844e-05,
      "model_forward_time": 1.7866997718811035,
      "step": 90838
    },
    {
      "epoch": 1.605796813964844e-05,
      "step": 90838,
      "training_step_time": 7.54876184463501
    },
    {
      "epoch": 1.6058921813964842e-05,
      "model_forward_time": 1.7488892078399658,
      "step": 90839
    },
    {
      "epoch": 1.6058921813964842e-05,
      "step": 90839,
      "training_step_time": 7.436192512512207
    },
    {
      "epoch": 1.605987548828125e-05,
      "grad_norm": 0.10332049429416656,
      "learning_rate": 2.2769537609479198e-07,
      "loss": 0.0555,
      "step": 90840
    },
    {
      "epoch": 1.605987548828125e-05,
      "model_forward_time": 1.8166964054107666,
      "step": 90840
    },
    {
      "dynamics_loss_avg": 0.04950599111616612,
      "epoch": 1.605987548828125e-05,
      "step": 90840
    },
    {
      "action_loss_avg": 0.001106325932778418,
      "epoch": 1.605987548828125e-05,
      "step": 90840
    },
    {
      "epoch": 1.605987548828125e-05,
      "step": 90840,
      "training_step_time": 8.628138303756714
    },
    {
      "epoch": 1.6060829162597657e-05,
      "model_forward_time": 1.757927656173706,
      "step": 90841
    },
    {
      "epoch": 1.6060829162597657e-05,
      "step": 90841,
      "training_step_time": 7.689347982406616
    },
    {
      "epoch": 1.606178283691406e-05,
      "model_forward_time": 1.3991296291351318,
      "step": 90842
    },
    {
      "epoch": 1.606178283691406e-05,
      "step": 90842,
      "training_step_time": 10.006197214126587
    },
    {
      "epoch": 1.606273651123047e-05,
      "model_forward_time": 2.899503707885742,
      "step": 90843
    },
    {
      "epoch": 1.606273651123047e-05,
      "step": 90843,
      "training_step_time": 8.291670083999634
    },
    {
      "epoch": 1.6063690185546876e-05,
      "model_forward_time": 1.761483907699585,
      "step": 90844
    },
    {
      "epoch": 1.6063690185546876e-05,
      "step": 90844,
      "training_step_time": 8.684221744537354
    },
    {
      "epoch": 1.606464385986328e-05,
      "model_forward_time": 1.7630531787872314,
      "step": 90845
    },
    {
      "epoch": 1.606464385986328e-05,
      "step": 90845,
      "training_step_time": 8.77272653579712
    },
    {
      "epoch": 1.6065597534179688e-05,
      "model_forward_time": 1.771500825881958,
      "step": 90846
    },
    {
      "epoch": 1.6065597534179688e-05,
      "step": 90846,
      "training_step_time": 7.683865785598755
    },
    {
      "epoch": 1.6066551208496095e-05,
      "model_forward_time": 1.7673366069793701,
      "step": 90847
    },
    {
      "epoch": 1.6066551208496095e-05,
      "step": 90847,
      "training_step_time": 7.466945648193359
    },
    {
      "epoch": 1.60675048828125e-05,
      "model_forward_time": 1.761305570602417,
      "step": 90848
    },
    {
      "epoch": 1.60675048828125e-05,
      "step": 90848,
      "training_step_time": 7.247452020645142
    },
    {
      "epoch": 1.6068458557128906e-05,
      "model_forward_time": 1.8358559608459473,
      "step": 90849
    },
    {
      "epoch": 1.6068458557128906e-05,
      "step": 90849,
      "training_step_time": 7.549901723861694
    },
    {
      "epoch": 1.6069412231445314e-05,
      "grad_norm": 0.0890587866306305,
      "learning_rate": 2.272023475498636e-07,
      "loss": 0.0571,
      "step": 90850
    },
    {
      "epoch": 1.6069412231445314e-05,
      "model_forward_time": 1.7656216621398926,
      "step": 90850
    },
    {
      "dynamics_loss_avg": 0.05369393564760685,
      "epoch": 1.6069412231445314e-05,
      "step": 90850
    },
    {
      "action_loss_avg": 0.0013931103225331754,
      "epoch": 1.6069412231445314e-05,
      "step": 90850
    },
    {
      "epoch": 1.6069412231445314e-05,
      "step": 90850,
      "training_step_time": 8.721621751785278
    },
    {
      "epoch": 1.6070365905761718e-05,
      "model_forward_time": 1.7649047374725342,
      "step": 90851
    },
    {
      "epoch": 1.6070365905761718e-05,
      "step": 90851,
      "training_step_time": 8.665965557098389
    },
    {
      "epoch": 1.6071319580078125e-05,
      "model_forward_time": 2.859046459197998,
      "step": 90852
    },
    {
      "epoch": 1.6071319580078125e-05,
      "step": 90852,
      "training_step_time": 8.8865807056427
    },
    {
      "epoch": 1.6072273254394533e-05,
      "model_forward_time": 1.7577006816864014,
      "step": 90853
    },
    {
      "epoch": 1.6072273254394533e-05,
      "step": 90853,
      "training_step_time": 7.705597162246704
    },
    {
      "epoch": 1.6073226928710937e-05,
      "model_forward_time": 1.7700879573822021,
      "step": 90854
    },
    {
      "epoch": 1.6073226928710937e-05,
      "step": 90854,
      "training_step_time": 9.467081308364868
    },
    {
      "epoch": 1.6074180603027344e-05,
      "model_forward_time": 1.7652168273925781,
      "step": 90855
    },
    {
      "epoch": 1.6074180603027344e-05,
      "step": 90855,
      "training_step_time": 7.673530340194702
    },
    {
      "epoch": 1.607513427734375e-05,
      "model_forward_time": 1.763195276260376,
      "step": 90856
    },
    {
      "epoch": 1.607513427734375e-05,
      "step": 90856,
      "training_step_time": 7.5282182693481445
    },
    {
      "epoch": 1.6076087951660156e-05,
      "model_forward_time": 1.7457311153411865,
      "step": 90857
    },
    {
      "epoch": 1.6076087951660156e-05,
      "step": 90857,
      "training_step_time": 7.257214307785034
    },
    {
      "epoch": 1.6077041625976563e-05,
      "model_forward_time": 1.814732313156128,
      "step": 90858
    },
    {
      "epoch": 1.6077041625976563e-05,
      "step": 90858,
      "training_step_time": 8.01951813697815
    },
    {
      "epoch": 1.607799530029297e-05,
      "model_forward_time": 1.7415120601654053,
      "step": 90859
    },
    {
      "epoch": 1.607799530029297e-05,
      "step": 90859,
      "training_step_time": 8.507199764251709
    },
    {
      "epoch": 1.6078948974609374e-05,
      "grad_norm": 0.08342166990041733,
      "learning_rate": 2.2670984095091108e-07,
      "loss": 0.0556,
      "step": 90860
    },
    {
      "epoch": 1.6078948974609374e-05,
      "model_forward_time": 1.761394739151001,
      "step": 90860
    },
    {
      "dynamics_loss_avg": 0.04970994703471661,
      "epoch": 1.6078948974609374e-05,
      "step": 90860
    },
    {
      "action_loss_avg": 0.001664536316820886,
      "epoch": 1.6078948974609374e-05,
      "step": 90860
    },
    {
      "epoch": 1.6078948974609374e-05,
      "step": 90860,
      "training_step_time": 9.50595474243164
    },
    {
      "epoch": 1.6079902648925782e-05,
      "model_forward_time": 2.8068323135375977,
      "step": 90861
    },
    {
      "epoch": 1.6079902648925782e-05,
      "step": 90861,
      "training_step_time": 9.83114767074585
    },
    {
      "epoch": 1.6080856323242186e-05,
      "model_forward_time": 1.7503838539123535,
      "step": 90862
    },
    {
      "epoch": 1.6080856323242186e-05,
      "step": 90862,
      "training_step_time": 8.26559042930603
    },
    {
      "epoch": 1.6081809997558593e-05,
      "model_forward_time": 1.7431859970092773,
      "step": 90863
    },
    {
      "epoch": 1.6081809997558593e-05,
      "step": 90863,
      "training_step_time": 8.278976440429688
    },
    {
      "epoch": 1.6082763671875e-05,
      "model_forward_time": 1.7475249767303467,
      "step": 90864
    },
    {
      "epoch": 1.6082763671875e-05,
      "step": 90864,
      "training_step_time": 7.753910779953003
    },
    {
      "epoch": 1.6083717346191405e-05,
      "model_forward_time": 1.7464697360992432,
      "step": 90865
    },
    {
      "epoch": 1.6083717346191405e-05,
      "step": 90865,
      "training_step_time": 7.561795949935913
    },
    {
      "epoch": 1.6084671020507812e-05,
      "model_forward_time": 1.3793649673461914,
      "step": 90866
    },
    {
      "epoch": 1.6084671020507812e-05,
      "step": 90866,
      "training_step_time": 7.291253566741943
    },
    {
      "epoch": 1.608562469482422e-05,
      "model_forward_time": 1.7844717502593994,
      "step": 90867
    },
    {
      "epoch": 1.608562469482422e-05,
      "step": 90867,
      "training_step_time": 7.381662368774414
    },
    {
      "epoch": 1.6086578369140624e-05,
      "model_forward_time": 1.769775152206421,
      "step": 90868
    },
    {
      "epoch": 1.6086578369140624e-05,
      "step": 90868,
      "training_step_time": 7.662769317626953
    },
    {
      "epoch": 1.608753204345703e-05,
      "model_forward_time": 1.761672019958496,
      "step": 90869
    },
    {
      "epoch": 1.608753204345703e-05,
      "step": 90869,
      "training_step_time": 7.653719902038574
    },
    {
      "epoch": 1.608848571777344e-05,
      "grad_norm": 0.13126879930496216,
      "learning_rate": 2.2621785635179362e-07,
      "loss": 0.0563,
      "step": 90870
    },
    {
      "epoch": 1.608848571777344e-05,
      "model_forward_time": 1.7605712413787842,
      "step": 90870
    },
    {
      "dynamics_loss_avg": 0.049787578731775285,
      "epoch": 1.608848571777344e-05,
      "step": 90870
    },
    {
      "action_loss_avg": 0.0012665945163462312,
      "epoch": 1.608848571777344e-05,
      "step": 90870
    },
    {
      "epoch": 1.608848571777344e-05,
      "step": 90870,
      "training_step_time": 8.92114806175232
    },
    {
      "epoch": 1.6089439392089842e-05,
      "model_forward_time": 2.8391802310943604,
      "step": 90871
    },
    {
      "epoch": 1.6089439392089842e-05,
      "step": 90871,
      "training_step_time": 7.7424914836883545
    },
    {
      "epoch": 1.609039306640625e-05,
      "model_forward_time": 1.7607533931732178,
      "step": 90872
    },
    {
      "epoch": 1.609039306640625e-05,
      "step": 90872,
      "training_step_time": 8.218870162963867
    },
    {
      "epoch": 1.6091346740722657e-05,
      "model_forward_time": 1.7632508277893066,
      "step": 90873
    },
    {
      "epoch": 1.6091346740722657e-05,
      "step": 90873,
      "training_step_time": 8.992108345031738
    },
    {
      "epoch": 1.609230041503906e-05,
      "model_forward_time": 1.7605836391448975,
      "step": 90874
    },
    {
      "epoch": 1.609230041503906e-05,
      "step": 90874,
      "training_step_time": 7.691798686981201
    },
    {
      "epoch": 1.609325408935547e-05,
      "model_forward_time": 1.764665126800537,
      "step": 90875
    },
    {
      "epoch": 1.609325408935547e-05,
      "step": 90875,
      "training_step_time": 7.505671739578247
    },
    {
      "epoch": 1.6094207763671876e-05,
      "model_forward_time": 1.7616941928863525,
      "step": 90876
    },
    {
      "epoch": 1.6094207763671876e-05,
      "step": 90876,
      "training_step_time": 7.894864320755005
    },
    {
      "epoch": 1.609516143798828e-05,
      "model_forward_time": 1.771817684173584,
      "step": 90877
    },
    {
      "epoch": 1.609516143798828e-05,
      "step": 90877,
      "training_step_time": 8.689743757247925
    },
    {
      "epoch": 1.6096115112304688e-05,
      "model_forward_time": 1.769970417022705,
      "step": 90878
    },
    {
      "epoch": 1.6096115112304688e-05,
      "step": 90878,
      "training_step_time": 7.629992246627808
    },
    {
      "epoch": 1.6097068786621095e-05,
      "model_forward_time": 1.757890224456787,
      "step": 90879
    },
    {
      "epoch": 1.6097068786621095e-05,
      "step": 90879,
      "training_step_time": 8.425190687179565
    },
    {
      "epoch": 1.60980224609375e-05,
      "grad_norm": 0.09305834770202637,
      "learning_rate": 2.2572639380631368e-07,
      "loss": 0.0579,
      "step": 90880
    },
    {
      "epoch": 1.60980224609375e-05,
      "model_forward_time": 2.9179513454437256,
      "step": 90880
    },
    {
      "dynamics_loss_avg": 0.067104790546,
      "epoch": 1.60980224609375e-05,
      "step": 90880
    },
    {
      "action_loss_avg": 0.001770785340340808,
      "epoch": 1.60980224609375e-05,
      "step": 90880
    },
    {
      "epoch": 1.60980224609375e-05,
      "step": 90880,
      "training_step_time": 8.423432111740112
    },
    {
      "epoch": 1.6098976135253906e-05,
      "model_forward_time": 1.7620649337768555,
      "step": 90881
    },
    {
      "epoch": 1.6098976135253906e-05,
      "step": 90881,
      "training_step_time": 7.726327419281006
    },
    {
      "epoch": 1.6099929809570314e-05,
      "model_forward_time": 1.7725105285644531,
      "step": 90882
    },
    {
      "epoch": 1.6099929809570314e-05,
      "step": 90882,
      "training_step_time": 7.6799705028533936
    },
    {
      "epoch": 1.6100883483886718e-05,
      "model_forward_time": 1.7654211521148682,
      "step": 90883
    },
    {
      "epoch": 1.6100883483886718e-05,
      "step": 90883,
      "training_step_time": 8.325481414794922
    },
    {
      "epoch": 1.6101837158203125e-05,
      "model_forward_time": 1.755019187927246,
      "step": 90884
    },
    {
      "epoch": 1.6101837158203125e-05,
      "step": 90884,
      "training_step_time": 7.495404243469238
    },
    {
      "epoch": 1.6102790832519533e-05,
      "model_forward_time": 1.7648491859436035,
      "step": 90885
    },
    {
      "epoch": 1.6102790832519533e-05,
      "step": 90885,
      "training_step_time": 7.210036277770996
    },
    {
      "epoch": 1.6103744506835937e-05,
      "model_forward_time": 1.8350272178649902,
      "step": 90886
    },
    {
      "epoch": 1.6103744506835937e-05,
      "step": 90886,
      "training_step_time": 7.642628192901611
    },
    {
      "epoch": 1.6104698181152344e-05,
      "model_forward_time": 1.757368564605713,
      "step": 90887
    },
    {
      "epoch": 1.6104698181152344e-05,
      "step": 90887,
      "training_step_time": 8.180985927581787
    },
    {
      "epoch": 1.610565185546875e-05,
      "model_forward_time": 1.7612245082855225,
      "step": 90888
    },
    {
      "epoch": 1.610565185546875e-05,
      "step": 90888,
      "training_step_time": 7.75001072883606
    },
    {
      "epoch": 1.6106605529785156e-05,
      "model_forward_time": 2.8652071952819824,
      "step": 90889
    },
    {
      "epoch": 1.6106605529785156e-05,
      "step": 90889,
      "training_step_time": 8.280007123947144
    },
    {
      "epoch": 1.6107559204101563e-05,
      "grad_norm": 0.097123883664608,
      "learning_rate": 2.2523545336821718e-07,
      "loss": 0.0576,
      "step": 90890
    },
    {
      "epoch": 1.6107559204101563e-05,
      "model_forward_time": 1.7601191997528076,
      "step": 90890
    },
    {
      "dynamics_loss_avg": 0.06643290594220161,
      "epoch": 1.6107559204101563e-05,
      "step": 90890
    },
    {
      "action_loss_avg": 0.0017561704851686954,
      "epoch": 1.6107559204101563e-05,
      "step": 90890
    },
    {
      "epoch": 1.6107559204101563e-05,
      "step": 90890,
      "training_step_time": 9.271995782852173
    },
    {
      "epoch": 1.6108512878417967e-05,
      "model_forward_time": 1.7726600170135498,
      "step": 90891
    },
    {
      "epoch": 1.6108512878417967e-05,
      "step": 90891,
      "training_step_time": 8.233606338500977
    },
    {
      "epoch": 1.6109466552734374e-05,
      "model_forward_time": 1.7628719806671143,
      "step": 90892
    },
    {
      "epoch": 1.6109466552734374e-05,
      "step": 90892,
      "training_step_time": 8.111316442489624
    },
    {
      "epoch": 1.6110420227050782e-05,
      "model_forward_time": 1.7612862586975098,
      "step": 90893
    },
    {
      "epoch": 1.6110420227050782e-05,
      "step": 90893,
      "training_step_time": 7.5357983112335205
    },
    {
      "epoch": 1.6111373901367186e-05,
      "model_forward_time": 1.767885446548462,
      "step": 90894
    },
    {
      "epoch": 1.6111373901367186e-05,
      "step": 90894,
      "training_step_time": 7.204983472824097
    },
    {
      "epoch": 1.6112327575683593e-05,
      "model_forward_time": 1.833268642425537,
      "step": 90895
    },
    {
      "epoch": 1.6112327575683593e-05,
      "step": 90895,
      "training_step_time": 8.373705625534058
    },
    {
      "epoch": 1.611328125e-05,
      "model_forward_time": 1.7662122249603271,
      "step": 90896
    },
    {
      "epoch": 1.611328125e-05,
      "step": 90896,
      "training_step_time": 7.658025026321411
    },
    {
      "epoch": 1.6114234924316405e-05,
      "model_forward_time": 1.758786916732788,
      "step": 90897
    },
    {
      "epoch": 1.6114234924316405e-05,
      "step": 90897,
      "training_step_time": 8.71257734298706
    },
    {
      "epoch": 1.6115188598632812e-05,
      "model_forward_time": 2.8670148849487305,
      "step": 90898
    },
    {
      "epoch": 1.6115188598632812e-05,
      "step": 90898,
      "training_step_time": 8.216437339782715
    },
    {
      "epoch": 1.611614227294922e-05,
      "model_forward_time": 1.7561943531036377,
      "step": 90899
    },
    {
      "epoch": 1.611614227294922e-05,
      "step": 90899,
      "training_step_time": 9.404966115951538
    },
    {
      "epoch": 1.6117095947265624e-05,
      "grad_norm": 0.08411497622728348,
      "learning_rate": 2.2474503509119394e-07,
      "loss": 0.0563,
      "step": 90900
    },
    {
      "epoch": 1.6117095947265624e-05,
      "model_forward_time": 1.741929531097412,
      "step": 90900
    },
    {
      "dynamics_loss_avg": 0.05847771801054478,
      "epoch": 1.6117095947265624e-05,
      "step": 90900
    },
    {
      "action_loss_avg": 0.0012354913516901434,
      "epoch": 1.6117095947265624e-05,
      "step": 90900
    },
    {
      "epoch": 1.6117095947265624e-05,
      "step": 90900,
      "training_step_time": 8.144112825393677
    },
    {
      "epoch": 1.611804962158203e-05,
      "model_forward_time": 1.765108346939087,
      "step": 90901
    },
    {
      "epoch": 1.611804962158203e-05,
      "step": 90901,
      "training_step_time": 7.849442720413208
    },
    {
      "epoch": 1.611900329589844e-05,
      "model_forward_time": 1.7604386806488037,
      "step": 90902
    },
    {
      "epoch": 1.611900329589844e-05,
      "step": 90902,
      "training_step_time": 7.652123689651489
    },
    {
      "epoch": 1.6119956970214843e-05,
      "model_forward_time": 1.770982027053833,
      "step": 90903
    },
    {
      "epoch": 1.6119956970214843e-05,
      "step": 90903,
      "training_step_time": 7.509622097015381
    },
    {
      "epoch": 1.612091064453125e-05,
      "model_forward_time": 1.0712237358093262,
      "step": 90904
    },
    {
      "epoch": 1.612091064453125e-05,
      "step": 90904,
      "training_step_time": 7.727679014205933
    },
    {
      "epoch": 1.6121864318847657e-05,
      "model_forward_time": 1.759350061416626,
      "step": 90905
    },
    {
      "epoch": 1.6121864318847657e-05,
      "step": 90905,
      "training_step_time": 7.67899489402771
    },
    {
      "epoch": 1.612281799316406e-05,
      "model_forward_time": 1.7881169319152832,
      "step": 90906
    },
    {
      "epoch": 1.612281799316406e-05,
      "step": 90906,
      "training_step_time": 8.824108600616455
    },
    {
      "epoch": 1.612377166748047e-05,
      "model_forward_time": 1.7711093425750732,
      "step": 90907
    },
    {
      "epoch": 1.612377166748047e-05,
      "step": 90907,
      "training_step_time": 8.483316898345947
    },
    {
      "epoch": 1.6124725341796876e-05,
      "model_forward_time": 2.9286208152770996,
      "step": 90908
    },
    {
      "epoch": 1.6124725341796876e-05,
      "step": 90908,
      "training_step_time": 9.371307611465454
    },
    {
      "epoch": 1.612567901611328e-05,
      "model_forward_time": 1.7303712368011475,
      "step": 90909
    },
    {
      "epoch": 1.612567901611328e-05,
      "step": 90909,
      "training_step_time": 8.269131898880005
    },
    {
      "epoch": 1.6126632690429688e-05,
      "grad_norm": 0.09194562584161758,
      "learning_rate": 2.242551390288733e-07,
      "loss": 0.0553,
      "step": 90910
    },
    {
      "epoch": 1.6126632690429688e-05,
      "model_forward_time": 1.763828992843628,
      "step": 90910
    },
    {
      "dynamics_loss_avg": 0.05205898880958557,
      "epoch": 1.6126632690429688e-05,
      "step": 90910
    },
    {
      "action_loss_avg": 0.0010862509661819786,
      "epoch": 1.6126632690429688e-05,
      "step": 90910
    },
    {
      "epoch": 1.6126632690429688e-05,
      "step": 90910,
      "training_step_time": 7.941424369812012
    },
    {
      "epoch": 1.6127586364746095e-05,
      "model_forward_time": 1.386469841003418,
      "step": 90911
    },
    {
      "epoch": 1.6127586364746095e-05,
      "step": 90911,
      "training_step_time": 7.585169792175293
    },
    {
      "epoch": 1.61285400390625e-05,
      "model_forward_time": 1.7648236751556396,
      "step": 90912
    },
    {
      "epoch": 1.61285400390625e-05,
      "step": 90912,
      "training_step_time": 7.514371633529663
    },
    {
      "epoch": 1.6129493713378906e-05,
      "model_forward_time": 1.7857344150543213,
      "step": 90913
    },
    {
      "epoch": 1.6129493713378906e-05,
      "step": 90913,
      "training_step_time": 8.60792088508606
    },
    {
      "epoch": 1.6130447387695314e-05,
      "model_forward_time": 1.758396863937378,
      "step": 90914
    },
    {
      "epoch": 1.6130447387695314e-05,
      "step": 90914,
      "training_step_time": 7.561981439590454
    },
    {
      "epoch": 1.6131401062011718e-05,
      "model_forward_time": 1.7541141510009766,
      "step": 90915
    },
    {
      "epoch": 1.6131401062011718e-05,
      "step": 90915,
      "training_step_time": 9.274622201919556
    },
    {
      "epoch": 1.6132354736328125e-05,
      "model_forward_time": 1.7748370170593262,
      "step": 90916
    },
    {
      "epoch": 1.6132354736328125e-05,
      "step": 90916,
      "training_step_time": 9.684157609939575
    },
    {
      "epoch": 1.6133308410644533e-05,
      "model_forward_time": 2.857034206390381,
      "step": 90917
    },
    {
      "epoch": 1.6133308410644533e-05,
      "step": 90917,
      "training_step_time": 7.717518329620361
    },
    {
      "epoch": 1.6134262084960937e-05,
      "model_forward_time": 1.7550420761108398,
      "step": 90918
    },
    {
      "epoch": 1.6134262084960937e-05,
      "step": 90918,
      "training_step_time": 9.212899446487427
    },
    {
      "epoch": 1.6135215759277344e-05,
      "model_forward_time": 1.7568490505218506,
      "step": 90919
    },
    {
      "epoch": 1.6135215759277344e-05,
      "step": 90919,
      "training_step_time": 8.85734248161316
    },
    {
      "epoch": 1.613616943359375e-05,
      "grad_norm": 0.10414619743824005,
      "learning_rate": 2.2376576523483018e-07,
      "loss": 0.0561,
      "step": 90920
    },
    {
      "epoch": 1.613616943359375e-05,
      "model_forward_time": 1.7587289810180664,
      "step": 90920
    },
    {
      "dynamics_loss_avg": 0.051899578981101514,
      "epoch": 1.613616943359375e-05,
      "step": 90920
    },
    {
      "action_loss_avg": 0.0011872523493366316,
      "epoch": 1.613616943359375e-05,
      "step": 90920
    },
    {
      "epoch": 1.613616943359375e-05,
      "step": 90920,
      "training_step_time": 7.649506330490112
    },
    {
      "epoch": 1.6137123107910156e-05,
      "model_forward_time": 1.7608470916748047,
      "step": 90921
    },
    {
      "epoch": 1.6137123107910156e-05,
      "step": 90921,
      "training_step_time": 7.465101003646851
    },
    {
      "epoch": 1.6138076782226563e-05,
      "model_forward_time": 1.771280288696289,
      "step": 90922
    },
    {
      "epoch": 1.6138076782226563e-05,
      "step": 90922,
      "training_step_time": 7.681919813156128
    },
    {
      "epoch": 1.6139030456542967e-05,
      "model_forward_time": 1.7565515041351318,
      "step": 90923
    },
    {
      "epoch": 1.6139030456542967e-05,
      "step": 90923,
      "training_step_time": 8.620909929275513
    },
    {
      "epoch": 1.6139984130859375e-05,
      "model_forward_time": 1.7672648429870605,
      "step": 90924
    },
    {
      "epoch": 1.6139984130859375e-05,
      "step": 90924,
      "training_step_time": 7.6242334842681885
    },
    {
      "epoch": 1.6140937805175782e-05,
      "model_forward_time": 1.7458298206329346,
      "step": 90925
    },
    {
      "epoch": 1.6140937805175782e-05,
      "step": 90925,
      "training_step_time": 8.469749450683594
    },
    {
      "epoch": 1.6141891479492186e-05,
      "model_forward_time": 2.8710927963256836,
      "step": 90926
    },
    {
      "epoch": 1.6141891479492186e-05,
      "step": 90926,
      "training_step_time": 9.413260698318481
    },
    {
      "epoch": 1.6142845153808593e-05,
      "model_forward_time": 1.7400717735290527,
      "step": 90927
    },
    {
      "epoch": 1.6142845153808593e-05,
      "step": 90927,
      "training_step_time": 8.545124053955078
    },
    {
      "epoch": 1.6143798828125e-05,
      "model_forward_time": 1.753767728805542,
      "step": 90928
    },
    {
      "epoch": 1.6143798828125e-05,
      "step": 90928,
      "training_step_time": 8.521384716033936
    },
    {
      "epoch": 1.6144752502441405e-05,
      "model_forward_time": 1.751464605331421,
      "step": 90929
    },
    {
      "epoch": 1.6144752502441405e-05,
      "step": 90929,
      "training_step_time": 7.615333318710327
    },
    {
      "epoch": 1.6145706176757812e-05,
      "grad_norm": 0.07646859437227249,
      "learning_rate": 2.2327691376258286e-07,
      "loss": 0.0569,
      "step": 90930
    },
    {
      "epoch": 1.6145706176757812e-05,
      "model_forward_time": 1.74580979347229,
      "step": 90930
    },
    {
      "dynamics_loss_avg": 0.056958920508623126,
      "epoch": 1.6145706176757812e-05,
      "step": 90930
    },
    {
      "action_loss_avg": 0.001560015557333827,
      "epoch": 1.6145706176757812e-05,
      "step": 90930
    },
    {
      "epoch": 1.6145706176757812e-05,
      "step": 90930,
      "training_step_time": 7.552921295166016
    },
    {
      "epoch": 1.614665985107422e-05,
      "model_forward_time": 1.7478795051574707,
      "step": 90931
    },
    {
      "epoch": 1.614665985107422e-05,
      "step": 90931,
      "training_step_time": 7.287548542022705
    },
    {
      "epoch": 1.6147613525390624e-05,
      "model_forward_time": 1.81496262550354,
      "step": 90932
    },
    {
      "epoch": 1.6147613525390624e-05,
      "step": 90932,
      "training_step_time": 7.701605319976807
    },
    {
      "epoch": 1.614856719970703e-05,
      "model_forward_time": 1.7431538105010986,
      "step": 90933
    },
    {
      "epoch": 1.614856719970703e-05,
      "step": 90933,
      "training_step_time": 8.639000415802002
    },
    {
      "epoch": 1.614952087402344e-05,
      "model_forward_time": 1.7394077777862549,
      "step": 90934
    },
    {
      "epoch": 1.614952087402344e-05,
      "step": 90934,
      "training_step_time": 8.319303512573242
    },
    {
      "epoch": 1.6150474548339843e-05,
      "model_forward_time": 2.8110511302948,
      "step": 90935
    },
    {
      "epoch": 1.6150474548339843e-05,
      "step": 90935,
      "training_step_time": 8.72439455986023
    },
    {
      "epoch": 1.615142822265625e-05,
      "model_forward_time": 1.7697489261627197,
      "step": 90936
    },
    {
      "epoch": 1.615142822265625e-05,
      "step": 90936,
      "training_step_time": 9.547731399536133
    },
    {
      "epoch": 1.6152381896972657e-05,
      "model_forward_time": 1.7554101943969727,
      "step": 90937
    },
    {
      "epoch": 1.6152381896972657e-05,
      "step": 90937,
      "training_step_time": 7.931866645812988
    },
    {
      "epoch": 1.615333557128906e-05,
      "model_forward_time": 1.7598402500152588,
      "step": 90938
    },
    {
      "epoch": 1.615333557128906e-05,
      "step": 90938,
      "training_step_time": 7.656183958053589
    },
    {
      "epoch": 1.615428924560547e-05,
      "model_forward_time": 1.752152442932129,
      "step": 90939
    },
    {
      "epoch": 1.615428924560547e-05,
      "step": 90939,
      "training_step_time": 8.452057123184204
    },
    {
      "epoch": 1.6155242919921876e-05,
      "grad_norm": 0.16025449335575104,
      "learning_rate": 2.2278858466559028e-07,
      "loss": 0.0558,
      "step": 90940
    },
    {
      "epoch": 1.6155242919921876e-05,
      "model_forward_time": 1.751030445098877,
      "step": 90940
    },
    {
      "dynamics_loss_avg": 0.054787956178188324,
      "epoch": 1.6155242919921876e-05,
      "step": 90940
    },
    {
      "action_loss_avg": 0.0011415920191211626,
      "epoch": 1.6155242919921876e-05,
      "step": 90940
    },
    {
      "epoch": 1.6155242919921876e-05,
      "step": 90940,
      "training_step_time": 7.161794662475586
    },
    {
      "epoch": 1.615619659423828e-05,
      "model_forward_time": 1.8133010864257812,
      "step": 90941
    },
    {
      "epoch": 1.615619659423828e-05,
      "step": 90941,
      "training_step_time": 8.656492233276367
    },
    {
      "epoch": 1.6157150268554688e-05,
      "model_forward_time": 1.7486443519592285,
      "step": 90942
    },
    {
      "epoch": 1.6157150268554688e-05,
      "step": 90942,
      "training_step_time": 8.299746036529541
    },
    {
      "epoch": 1.6158103942871095e-05,
      "model_forward_time": 1.7641828060150146,
      "step": 90943
    },
    {
      "epoch": 1.6158103942871095e-05,
      "step": 90943,
      "training_step_time": 8.92520546913147
    },
    {
      "epoch": 1.61590576171875e-05,
      "model_forward_time": 1.755021572113037,
      "step": 90944
    },
    {
      "epoch": 1.61590576171875e-05,
      "step": 90944,
      "training_step_time": 8.273249864578247
    },
    {
      "epoch": 1.6160011291503907e-05,
      "model_forward_time": 1.7547330856323242,
      "step": 90945
    },
    {
      "epoch": 1.6160011291503907e-05,
      "step": 90945,
      "training_step_time": 9.078989505767822
    },
    {
      "epoch": 1.6160964965820314e-05,
      "model_forward_time": 1.7375247478485107,
      "step": 90946
    },
    {
      "epoch": 1.6160964965820314e-05,
      "step": 90946,
      "training_step_time": 9.407556295394897
    },
    {
      "epoch": 1.6161918640136718e-05,
      "model_forward_time": 1.7555668354034424,
      "step": 90947
    },
    {
      "epoch": 1.6161918640136718e-05,
      "step": 90947,
      "training_step_time": 7.898912668228149
    },
    {
      "epoch": 1.6162872314453125e-05,
      "model_forward_time": 1.3823637962341309,
      "step": 90948
    },
    {
      "epoch": 1.6162872314453125e-05,
      "step": 90948,
      "training_step_time": 7.5687644481658936
    },
    {
      "epoch": 1.6163825988769533e-05,
      "model_forward_time": 1.748420238494873,
      "step": 90949
    },
    {
      "epoch": 1.6163825988769533e-05,
      "step": 90949,
      "training_step_time": 7.521435022354126
    },
    {
      "epoch": 1.6164779663085937e-05,
      "grad_norm": 0.07468672096729279,
      "learning_rate": 2.2230077799725526e-07,
      "loss": 0.0556,
      "step": 90950
    },
    {
      "epoch": 1.6164779663085937e-05,
      "model_forward_time": 1.8335974216461182,
      "step": 90950
    },
    {
      "dynamics_loss_avg": 0.04652345050126314,
      "epoch": 1.6164779663085937e-05,
      "step": 90950
    },
    {
      "action_loss_avg": 0.0012011164799332619,
      "epoch": 1.6164779663085937e-05,
      "step": 90950
    },
    {
      "epoch": 1.6164779663085937e-05,
      "step": 90950,
      "training_step_time": 8.254923820495605
    },
    {
      "epoch": 1.6165733337402344e-05,
      "model_forward_time": 1.7457084655761719,
      "step": 90951
    },
    {
      "epoch": 1.6165733337402344e-05,
      "step": 90951,
      "training_step_time": 7.585669040679932
    },
    {
      "epoch": 1.616668701171875e-05,
      "model_forward_time": 1.7509562969207764,
      "step": 90952
    },
    {
      "epoch": 1.616668701171875e-05,
      "step": 90952,
      "training_step_time": 7.694809436798096
    },
    {
      "epoch": 1.6167640686035156e-05,
      "model_forward_time": 1.761042833328247,
      "step": 90953
    },
    {
      "epoch": 1.6167640686035156e-05,
      "step": 90953,
      "training_step_time": 9.881900548934937
    },
    {
      "epoch": 1.6168594360351563e-05,
      "model_forward_time": 2.8257968425750732,
      "step": 90954
    },
    {
      "epoch": 1.6168594360351563e-05,
      "step": 90954,
      "training_step_time": 9.246694803237915
    },
    {
      "epoch": 1.6169548034667967e-05,
      "model_forward_time": 1.7532963752746582,
      "step": 90955
    },
    {
      "epoch": 1.6169548034667967e-05,
      "step": 90955,
      "training_step_time": 7.713151454925537
    },
    {
      "epoch": 1.6170501708984375e-05,
      "model_forward_time": 1.3821966648101807,
      "step": 90956
    },
    {
      "epoch": 1.6170501708984375e-05,
      "step": 90956,
      "training_step_time": 7.718865633010864
    },
    {
      "epoch": 1.6171455383300782e-05,
      "model_forward_time": 1.7452032566070557,
      "step": 90957
    },
    {
      "epoch": 1.6171455383300782e-05,
      "step": 90957,
      "training_step_time": 7.6050333976745605
    },
    {
      "epoch": 1.6172409057617186e-05,
      "model_forward_time": 1.7624309062957764,
      "step": 90958
    },
    {
      "epoch": 1.6172409057617186e-05,
      "step": 90958,
      "training_step_time": 7.464434385299683
    },
    {
      "epoch": 1.6173362731933593e-05,
      "model_forward_time": 1.7906320095062256,
      "step": 90959
    },
    {
      "epoch": 1.6173362731933593e-05,
      "step": 90959,
      "training_step_time": 7.658604145050049
    },
    {
      "epoch": 1.617431640625e-05,
      "grad_norm": 0.1231004148721695,
      "learning_rate": 2.2181349381092398e-07,
      "loss": 0.0568,
      "step": 90960
    },
    {
      "epoch": 1.617431640625e-05,
      "model_forward_time": 1.73964262008667,
      "step": 90960
    },
    {
      "dynamics_loss_avg": 0.04646523352712393,
      "epoch": 1.617431640625e-05,
      "step": 90960
    },
    {
      "action_loss_avg": 0.001001173199620098,
      "epoch": 1.617431640625e-05,
      "step": 90960
    },
    {
      "epoch": 1.617431640625e-05,
      "step": 90960,
      "training_step_time": 7.5759217739105225
    },
    {
      "epoch": 1.6175270080566405e-05,
      "model_forward_time": 1.733414888381958,
      "step": 90961
    },
    {
      "epoch": 1.6175270080566405e-05,
      "step": 90961,
      "training_step_time": 7.756274700164795
    },
    {
      "epoch": 1.6176223754882812e-05,
      "model_forward_time": 1.759326457977295,
      "step": 90962
    },
    {
      "epoch": 1.6176223754882812e-05,
      "step": 90962,
      "training_step_time": 8.325346946716309
    },
    {
      "epoch": 1.617717742919922e-05,
      "model_forward_time": 2.8521745204925537,
      "step": 90963
    },
    {
      "epoch": 1.617717742919922e-05,
      "step": 90963,
      "training_step_time": 10.03133511543274
    },
    {
      "epoch": 1.6178131103515624e-05,
      "model_forward_time": 1.7481293678283691,
      "step": 90964
    },
    {
      "epoch": 1.6178131103515624e-05,
      "step": 90964,
      "training_step_time": 8.37438440322876
    },
    {
      "epoch": 1.617908477783203e-05,
      "model_forward_time": 1.7519214153289795,
      "step": 90965
    },
    {
      "epoch": 1.617908477783203e-05,
      "step": 90965,
      "training_step_time": 7.933031797409058
    },
    {
      "epoch": 1.618003845214844e-05,
      "model_forward_time": 1.7487988471984863,
      "step": 90966
    },
    {
      "epoch": 1.618003845214844e-05,
      "step": 90966,
      "training_step_time": 7.584422588348389
    },
    {
      "epoch": 1.6180992126464843e-05,
      "model_forward_time": 1.7517955303192139,
      "step": 90967
    },
    {
      "epoch": 1.6180992126464843e-05,
      "step": 90967,
      "training_step_time": 7.417819976806641
    },
    {
      "epoch": 1.618194580078125e-05,
      "model_forward_time": 1.7783737182617188,
      "step": 90968
    },
    {
      "epoch": 1.618194580078125e-05,
      "step": 90968,
      "training_step_time": 7.688807249069214
    },
    {
      "epoch": 1.6182899475097657e-05,
      "model_forward_time": 1.7522644996643066,
      "step": 90969
    },
    {
      "epoch": 1.6182899475097657e-05,
      "step": 90969,
      "training_step_time": 8.577250480651855
    },
    {
      "epoch": 1.618385314941406e-05,
      "grad_norm": 0.117511086165905,
      "learning_rate": 2.2132673215988553e-07,
      "loss": 0.0546,
      "step": 90970
    },
    {
      "epoch": 1.618385314941406e-05,
      "model_forward_time": 1.7545998096466064,
      "step": 90970
    },
    {
      "dynamics_loss_avg": 0.05377452746033669,
      "epoch": 1.618385314941406e-05,
      "step": 90970
    },
    {
      "action_loss_avg": 0.0009940830117557199,
      "epoch": 1.618385314941406e-05,
      "step": 90970
    },
    {
      "epoch": 1.618385314941406e-05,
      "step": 90970,
      "training_step_time": 7.6899800300598145
    },
    {
      "epoch": 1.618480682373047e-05,
      "model_forward_time": 1.754389762878418,
      "step": 90971
    },
    {
      "epoch": 1.618480682373047e-05,
      "step": 90971,
      "training_step_time": 9.325453519821167
    },
    {
      "epoch": 1.6185760498046876e-05,
      "model_forward_time": 2.828162670135498,
      "step": 90972
    },
    {
      "epoch": 1.6185760498046876e-05,
      "step": 90972,
      "training_step_time": 8.578052282333374
    },
    {
      "epoch": 1.618671417236328e-05,
      "model_forward_time": 1.7229807376861572,
      "step": 90973
    },
    {
      "epoch": 1.618671417236328e-05,
      "step": 90973,
      "training_step_time": 7.651909589767456
    },
    {
      "epoch": 1.6187667846679688e-05,
      "model_forward_time": 1.7538788318634033,
      "step": 90974
    },
    {
      "epoch": 1.6187667846679688e-05,
      "step": 90974,
      "training_step_time": 8.589294195175171
    },
    {
      "epoch": 1.6188621520996095e-05,
      "model_forward_time": 1.7589473724365234,
      "step": 90975
    },
    {
      "epoch": 1.6188621520996095e-05,
      "step": 90975,
      "training_step_time": 8.317786693572998
    },
    {
      "epoch": 1.61895751953125e-05,
      "model_forward_time": 1.7489612102508545,
      "step": 90976
    },
    {
      "epoch": 1.61895751953125e-05,
      "step": 90976,
      "training_step_time": 7.591212034225464
    },
    {
      "epoch": 1.6190528869628907e-05,
      "model_forward_time": 1.7593638896942139,
      "step": 90977
    },
    {
      "epoch": 1.6190528869628907e-05,
      "step": 90977,
      "training_step_time": 8.039359331130981
    },
    {
      "epoch": 1.6191482543945314e-05,
      "model_forward_time": 1.7494657039642334,
      "step": 90978
    },
    {
      "epoch": 1.6191482543945314e-05,
      "step": 90978,
      "training_step_time": 7.679492712020874
    },
    {
      "epoch": 1.6192436218261718e-05,
      "model_forward_time": 1.7548656463623047,
      "step": 90979
    },
    {
      "epoch": 1.6192436218261718e-05,
      "step": 90979,
      "training_step_time": 7.690400123596191
    },
    {
      "epoch": 1.6193389892578125e-05,
      "grad_norm": 0.08976147323846817,
      "learning_rate": 2.2084049309737065e-07,
      "loss": 0.0579,
      "step": 90980
    },
    {
      "epoch": 1.6193389892578125e-05,
      "model_forward_time": 1.7629327774047852,
      "step": 90980
    },
    {
      "dynamics_loss_avg": 0.0641448251903057,
      "epoch": 1.6193389892578125e-05,
      "step": 90980
    },
    {
      "action_loss_avg": 0.0020521432743407786,
      "epoch": 1.6193389892578125e-05,
      "step": 90980
    },
    {
      "epoch": 1.6193389892578125e-05,
      "step": 90980,
      "training_step_time": 9.058578252792358
    },
    {
      "epoch": 1.6194343566894533e-05,
      "model_forward_time": 2.832904815673828,
      "step": 90981
    },
    {
      "epoch": 1.6194343566894533e-05,
      "step": 90981,
      "training_step_time": 8.404445886611938
    },
    {
      "epoch": 1.6195297241210937e-05,
      "model_forward_time": 1.749582290649414,
      "step": 90982
    },
    {
      "epoch": 1.6195297241210937e-05,
      "step": 90982,
      "training_step_time": 7.757056951522827
    },
    {
      "epoch": 1.6196250915527344e-05,
      "model_forward_time": 1.7515385150909424,
      "step": 90983
    },
    {
      "epoch": 1.6196250915527344e-05,
      "step": 90983,
      "training_step_time": 7.569288969039917
    },
    {
      "epoch": 1.619720458984375e-05,
      "model_forward_time": 1.7513659000396729,
      "step": 90984
    },
    {
      "epoch": 1.619720458984375e-05,
      "step": 90984,
      "training_step_time": 7.977421045303345
    },
    {
      "epoch": 1.6198158264160156e-05,
      "model_forward_time": 1.7455408573150635,
      "step": 90985
    },
    {
      "epoch": 1.6198158264160156e-05,
      "step": 90985,
      "training_step_time": 7.604063987731934
    },
    {
      "epoch": 1.6199111938476563e-05,
      "model_forward_time": 1.7529997825622559,
      "step": 90986
    },
    {
      "epoch": 1.6199111938476563e-05,
      "step": 90986,
      "training_step_time": 7.494430780410767
    },
    {
      "epoch": 1.6200065612792967e-05,
      "model_forward_time": 1.8178071975708008,
      "step": 90987
    },
    {
      "epoch": 1.6200065612792967e-05,
      "step": 90987,
      "training_step_time": 8.45823359489441
    },
    {
      "epoch": 1.6201019287109375e-05,
      "model_forward_time": 1.7549774646759033,
      "step": 90988
    },
    {
      "epoch": 1.6201019287109375e-05,
      "step": 90988,
      "training_step_time": 7.625406265258789
    },
    {
      "epoch": 1.6201972961425782e-05,
      "model_forward_time": 1.7477021217346191,
      "step": 90989
    },
    {
      "epoch": 1.6201972961425782e-05,
      "step": 90989,
      "training_step_time": 8.289192914962769
    },
    {
      "epoch": 1.6202926635742186e-05,
      "grad_norm": 1.0265917778015137,
      "learning_rate": 2.2035477667655515e-07,
      "loss": 0.0555,
      "step": 90990
    },
    {
      "epoch": 1.6202926635742186e-05,
      "model_forward_time": 2.7912635803222656,
      "step": 90990
    },
    {
      "dynamics_loss_avg": 0.0674909356981516,
      "epoch": 1.6202926635742186e-05,
      "step": 90990
    },
    {
      "action_loss_avg": 0.0019346747765666805,
      "epoch": 1.6202926635742186e-05,
      "step": 90990
    },
    {
      "epoch": 1.6202926635742186e-05,
      "step": 90990,
      "training_step_time": 8.456622838973999
    },
    {
      "epoch": 1.6203880310058593e-05,
      "model_forward_time": 1.7488224506378174,
      "step": 90991
    },
    {
      "epoch": 1.6203880310058593e-05,
      "step": 90991,
      "training_step_time": 7.761375904083252
    },
    {
      "epoch": 1.6204833984375e-05,
      "model_forward_time": 1.749892234802246,
      "step": 90992
    },
    {
      "epoch": 1.6204833984375e-05,
      "step": 90992,
      "training_step_time": 8.745365619659424
    },
    {
      "epoch": 1.6205787658691405e-05,
      "model_forward_time": 1.7523126602172852,
      "step": 90993
    },
    {
      "epoch": 1.6205787658691405e-05,
      "step": 90993,
      "training_step_time": 7.950361967086792
    },
    {
      "epoch": 1.6206741333007812e-05,
      "model_forward_time": 1.7460262775421143,
      "step": 90994
    },
    {
      "epoch": 1.6206741333007812e-05,
      "step": 90994,
      "training_step_time": 7.626737117767334
    },
    {
      "epoch": 1.620769500732422e-05,
      "model_forward_time": 1.7475190162658691,
      "step": 90995
    },
    {
      "epoch": 1.620769500732422e-05,
      "step": 90995,
      "training_step_time": 7.470520496368408
    },
    {
      "epoch": 1.6208648681640624e-05,
      "model_forward_time": 1.8187909126281738,
      "step": 90996
    },
    {
      "epoch": 1.6208648681640624e-05,
      "step": 90996,
      "training_step_time": 8.614250421524048
    },
    {
      "epoch": 1.620960235595703e-05,
      "model_forward_time": 1.39143705368042,
      "step": 90997
    },
    {
      "epoch": 1.620960235595703e-05,
      "step": 90997,
      "training_step_time": 7.641970634460449
    },
    {
      "epoch": 1.621055603027344e-05,
      "model_forward_time": 1.7770965099334717,
      "step": 90998
    },
    {
      "epoch": 1.621055603027344e-05,
      "step": 90998,
      "training_step_time": 7.81689453125
    },
    {
      "epoch": 1.6211509704589843e-05,
      "model_forward_time": 1.7995109558105469,
      "step": 90999
    },
    {
      "epoch": 1.6211509704589843e-05,
      "step": 90999,
      "training_step_time": 7.760368824005127
    },
    {
      "epoch": 1.621246337890625e-05,
      "grad_norm": 0.06410578638315201,
      "learning_rate": 2.1986958295055317e-07,
      "loss": 0.057,
      "step": 91000
    },
    {
      "epoch": 1.621246337890625e-05,
      "model_forward_time": 1.7858293056488037,
      "step": 91000
    },
    {
      "dynamics_loss_avg": 0.05008132960647345,
      "epoch": 1.621246337890625e-05,
      "step": 91000
    },
    {
      "action_loss_avg": 0.0010724502761149779,
      "epoch": 1.621246337890625e-05,
      "step": 91000
    },
    {
      "epoch": 1.621246337890625e-05,
      "step": 91000,
      "training_step_time": 10.919909477233887
    },
    {
      "epoch": 1.6213417053222657e-05,
      "model_forward_time": 1.7424910068511963,
      "step": 91001
    },
    {
      "epoch": 1.6213417053222657e-05,
      "step": 91001,
      "training_step_time": 8.560411930084229
    },
    {
      "epoch": 1.621437072753906e-05,
      "model_forward_time": 1.7670307159423828,
      "step": 91002
    },
    {
      "epoch": 1.621437072753906e-05,
      "step": 91002,
      "training_step_time": 7.839498043060303
    },
    {
      "epoch": 1.621532440185547e-05,
      "model_forward_time": 1.7660043239593506,
      "step": 91003
    },
    {
      "epoch": 1.621532440185547e-05,
      "step": 91003,
      "training_step_time": 8.469491481781006
    },
    {
      "epoch": 1.6216278076171876e-05,
      "model_forward_time": 1.766611099243164,
      "step": 91004
    },
    {
      "epoch": 1.6216278076171876e-05,
      "step": 91004,
      "training_step_time": 7.455414056777954
    },
    {
      "epoch": 1.621723175048828e-05,
      "model_forward_time": 1.833329677581787,
      "step": 91005
    },
    {
      "epoch": 1.621723175048828e-05,
      "step": 91005,
      "training_step_time": 11.142075538635254
    },
    {
      "epoch": 1.6218185424804688e-05,
      "model_forward_time": 1.7699699401855469,
      "step": 91006
    },
    {
      "epoch": 1.6218185424804688e-05,
      "step": 91006,
      "training_step_time": 9.685529232025146
    },
    {
      "epoch": 1.6219139099121095e-05,
      "model_forward_time": 1.7604095935821533,
      "step": 91007
    },
    {
      "epoch": 1.6219139099121095e-05,
      "step": 91007,
      "training_step_time": 7.613851070404053
    },
    {
      "epoch": 1.62200927734375e-05,
      "model_forward_time": 1.7640855312347412,
      "step": 91008
    },
    {
      "epoch": 1.62200927734375e-05,
      "step": 91008,
      "training_step_time": 8.600810766220093
    },
    {
      "epoch": 1.6221046447753907e-05,
      "model_forward_time": 1.8596010208129883,
      "step": 91009
    },
    {
      "epoch": 1.6221046447753907e-05,
      "step": 91009,
      "training_step_time": 8.830998659133911
    },
    {
      "epoch": 1.6222000122070314e-05,
      "grad_norm": 0.08985470980405807,
      "learning_rate": 2.193849119724284e-07,
      "loss": 0.0557,
      "step": 91010
    },
    {
      "epoch": 1.6222000122070314e-05,
      "model_forward_time": 1.764664888381958,
      "step": 91010
    },
    {
      "dynamics_loss_avg": 0.0512138026766479,
      "epoch": 1.6222000122070314e-05,
      "step": 91010
    },
    {
      "action_loss_avg": 0.002060062209784519,
      "epoch": 1.6222000122070314e-05,
      "step": 91010
    },
    {
      "epoch": 1.6222000122070314e-05,
      "step": 91010,
      "training_step_time": 8.257843017578125
    },
    {
      "epoch": 1.6222953796386718e-05,
      "model_forward_time": 1.760606050491333,
      "step": 91011
    },
    {
      "epoch": 1.6222953796386718e-05,
      "step": 91011,
      "training_step_time": 7.967920780181885
    },
    {
      "epoch": 1.6223907470703125e-05,
      "model_forward_time": 1.7642998695373535,
      "step": 91012
    },
    {
      "epoch": 1.6223907470703125e-05,
      "step": 91012,
      "training_step_time": 8.431302547454834
    },
    {
      "epoch": 1.6224861145019533e-05,
      "model_forward_time": 1.7677834033966064,
      "step": 91013
    },
    {
      "epoch": 1.6224861145019533e-05,
      "step": 91013,
      "training_step_time": 7.616016387939453
    },
    {
      "epoch": 1.6225814819335937e-05,
      "model_forward_time": 1.826258659362793,
      "step": 91014
    },
    {
      "epoch": 1.6225814819335937e-05,
      "step": 91014,
      "training_step_time": 9.009109020233154
    },
    {
      "epoch": 1.6226768493652344e-05,
      "model_forward_time": 1.7601373195648193,
      "step": 91015
    },
    {
      "epoch": 1.6226768493652344e-05,
      "step": 91015,
      "training_step_time": 8.652504920959473
    },
    {
      "epoch": 1.622772216796875e-05,
      "model_forward_time": 1.766284465789795,
      "step": 91016
    },
    {
      "epoch": 1.622772216796875e-05,
      "step": 91016,
      "training_step_time": 8.743598699569702
    },
    {
      "epoch": 1.6228675842285156e-05,
      "model_forward_time": 1.7668917179107666,
      "step": 91017
    },
    {
      "epoch": 1.6228675842285156e-05,
      "step": 91017,
      "training_step_time": 8.456786632537842
    },
    {
      "epoch": 1.6229629516601563e-05,
      "model_forward_time": 2.8374316692352295,
      "step": 91018
    },
    {
      "epoch": 1.6229629516601563e-05,
      "step": 91018,
      "training_step_time": 8.674257278442383
    },
    {
      "epoch": 1.6230583190917967e-05,
      "model_forward_time": 1.7599542140960693,
      "step": 91019
    },
    {
      "epoch": 1.6230583190917967e-05,
      "step": 91019,
      "training_step_time": 7.740301132202148
    },
    {
      "epoch": 1.6231536865234375e-05,
      "grad_norm": 0.28585267066955566,
      "learning_rate": 2.1890076379518177e-07,
      "loss": 0.0554,
      "step": 91020
    },
    {
      "epoch": 1.6231536865234375e-05,
      "model_forward_time": 1.7793807983398438,
      "step": 91020
    },
    {
      "dynamics_loss_avg": 0.05900451522320509,
      "epoch": 1.6231536865234375e-05,
      "step": 91020
    },
    {
      "action_loss_avg": 0.0023290383454877884,
      "epoch": 1.6231536865234375e-05,
      "step": 91020
    },
    {
      "epoch": 1.6231536865234375e-05,
      "step": 91020,
      "training_step_time": 7.954024076461792
    },
    {
      "epoch": 1.6232490539550782e-05,
      "model_forward_time": 1.783538579940796,
      "step": 91021
    },
    {
      "epoch": 1.6232490539550782e-05,
      "step": 91021,
      "training_step_time": 7.635264873504639
    },
    {
      "epoch": 1.6233444213867186e-05,
      "model_forward_time": 1.7666780948638916,
      "step": 91022
    },
    {
      "epoch": 1.6233444213867186e-05,
      "step": 91022,
      "training_step_time": 7.562725067138672
    },
    {
      "epoch": 1.6234397888183593e-05,
      "model_forward_time": 1.811777114868164,
      "step": 91023
    },
    {
      "epoch": 1.6234397888183593e-05,
      "step": 91023,
      "training_step_time": 8.184906005859375
    },
    {
      "epoch": 1.62353515625e-05,
      "model_forward_time": 1.7843756675720215,
      "step": 91024
    },
    {
      "epoch": 1.62353515625e-05,
      "step": 91024,
      "training_step_time": 7.566502094268799
    },
    {
      "epoch": 1.6236305236816405e-05,
      "model_forward_time": 1.7621150016784668,
      "step": 91025
    },
    {
      "epoch": 1.6236305236816405e-05,
      "step": 91025,
      "training_step_time": 8.732630968093872
    },
    {
      "epoch": 1.6237258911132812e-05,
      "model_forward_time": 1.7618796825408936,
      "step": 91026
    },
    {
      "epoch": 1.6237258911132812e-05,
      "step": 91026,
      "training_step_time": 7.718510627746582
    },
    {
      "epoch": 1.623821258544922e-05,
      "model_forward_time": 2.8590476512908936,
      "step": 91027
    },
    {
      "epoch": 1.623821258544922e-05,
      "step": 91027,
      "training_step_time": 7.686516523361206
    },
    {
      "epoch": 1.6239166259765624e-05,
      "model_forward_time": 1.7550702095031738,
      "step": 91028
    },
    {
      "epoch": 1.6239166259765624e-05,
      "step": 91028,
      "training_step_time": 9.523534059524536
    },
    {
      "epoch": 1.624011993408203e-05,
      "model_forward_time": 1.761415958404541,
      "step": 91029
    },
    {
      "epoch": 1.624011993408203e-05,
      "step": 91029,
      "training_step_time": 7.772735834121704
    },
    {
      "epoch": 1.624107360839844e-05,
      "grad_norm": 0.08787979185581207,
      "learning_rate": 2.1841713847175928e-07,
      "loss": 0.0576,
      "step": 91030
    },
    {
      "epoch": 1.624107360839844e-05,
      "model_forward_time": 1.7685844898223877,
      "step": 91030
    },
    {
      "dynamics_loss_avg": 0.05269201509654522,
      "epoch": 1.624107360839844e-05,
      "step": 91030
    },
    {
      "action_loss_avg": 0.0012069961027009412,
      "epoch": 1.624107360839844e-05,
      "step": 91030
    },
    {
      "epoch": 1.624107360839844e-05,
      "step": 91030,
      "training_step_time": 8.010327100753784
    },
    {
      "epoch": 1.6242027282714843e-05,
      "model_forward_time": 1.7649691104888916,
      "step": 91031
    },
    {
      "epoch": 1.6242027282714843e-05,
      "step": 91031,
      "training_step_time": 8.435319423675537
    },
    {
      "epoch": 1.624298095703125e-05,
      "model_forward_time": 1.763523817062378,
      "step": 91032
    },
    {
      "epoch": 1.624298095703125e-05,
      "step": 91032,
      "training_step_time": 9.00972318649292
    },
    {
      "epoch": 1.6243934631347657e-05,
      "model_forward_time": 1.8351876735687256,
      "step": 91033
    },
    {
      "epoch": 1.6243934631347657e-05,
      "step": 91033,
      "training_step_time": 7.7132627964019775
    },
    {
      "epoch": 1.624488830566406e-05,
      "model_forward_time": 1.766261339187622,
      "step": 91034
    },
    {
      "epoch": 1.624488830566406e-05,
      "step": 91034,
      "training_step_time": 8.189379453659058
    },
    {
      "epoch": 1.624584197998047e-05,
      "model_forward_time": 1.7562484741210938,
      "step": 91035
    },
    {
      "epoch": 1.624584197998047e-05,
      "step": 91035,
      "training_step_time": 7.695628643035889
    },
    {
      "epoch": 1.6246795654296876e-05,
      "model_forward_time": 2.8141610622406006,
      "step": 91036
    },
    {
      "epoch": 1.6246795654296876e-05,
      "step": 91036,
      "training_step_time": 9.31728720664978
    },
    {
      "epoch": 1.624774932861328e-05,
      "model_forward_time": 1.7652602195739746,
      "step": 91037
    },
    {
      "epoch": 1.624774932861328e-05,
      "step": 91037,
      "training_step_time": 8.767343521118164
    },
    {
      "epoch": 1.6248703002929688e-05,
      "model_forward_time": 1.7641527652740479,
      "step": 91038
    },
    {
      "epoch": 1.6248703002929688e-05,
      "step": 91038,
      "training_step_time": 8.345335960388184
    },
    {
      "epoch": 1.6249656677246095e-05,
      "model_forward_time": 1.7664849758148193,
      "step": 91039
    },
    {
      "epoch": 1.6249656677246095e-05,
      "step": 91039,
      "training_step_time": 7.916429758071899
    },
    {
      "epoch": 1.62506103515625e-05,
      "grad_norm": 0.06810242682695389,
      "learning_rate": 2.179340360550497e-07,
      "loss": 0.0548,
      "step": 91040
    },
    {
      "epoch": 1.62506103515625e-05,
      "model_forward_time": 1.7689476013183594,
      "step": 91040
    },
    {
      "dynamics_loss_avg": 0.054969932325184344,
      "epoch": 1.62506103515625e-05,
      "step": 91040
    },
    {
      "action_loss_avg": 0.0033794973482144997,
      "epoch": 1.62506103515625e-05,
      "step": 91040
    },
    {
      "epoch": 1.62506103515625e-05,
      "step": 91040,
      "training_step_time": 7.5669896602630615
    },
    {
      "epoch": 1.6251564025878907e-05,
      "model_forward_time": 1.761089563369751,
      "step": 91041
    },
    {
      "epoch": 1.6251564025878907e-05,
      "step": 91041,
      "training_step_time": 8.114703178405762
    },
    {
      "epoch": 1.6252517700195314e-05,
      "model_forward_time": 1.8416311740875244,
      "step": 91042
    },
    {
      "epoch": 1.6252517700195314e-05,
      "step": 91042,
      "training_step_time": 7.676647901535034
    },
    {
      "epoch": 1.6253471374511718e-05,
      "model_forward_time": 1.759486436843872,
      "step": 91043
    },
    {
      "epoch": 1.6253471374511718e-05,
      "step": 91043,
      "training_step_time": 7.656316757202148
    },
    {
      "epoch": 1.6254425048828125e-05,
      "model_forward_time": 1.7672717571258545,
      "step": 91044
    },
    {
      "epoch": 1.6254425048828125e-05,
      "step": 91044,
      "training_step_time": 8.787110328674316
    },
    {
      "epoch": 1.6255378723144533e-05,
      "model_forward_time": 2.828484058380127,
      "step": 91045
    },
    {
      "epoch": 1.6255378723144533e-05,
      "step": 91045,
      "training_step_time": 8.806629657745361
    },
    {
      "epoch": 1.6256332397460937e-05,
      "model_forward_time": 1.7644522190093994,
      "step": 91046
    },
    {
      "epoch": 1.6256332397460937e-05,
      "step": 91046,
      "training_step_time": 9.455710887908936
    },
    {
      "epoch": 1.6257286071777344e-05,
      "model_forward_time": 1.7698249816894531,
      "step": 91047
    },
    {
      "epoch": 1.6257286071777344e-05,
      "step": 91047,
      "training_step_time": 9.957626581192017
    },
    {
      "epoch": 1.625823974609375e-05,
      "model_forward_time": 1.765566110610962,
      "step": 91048
    },
    {
      "epoch": 1.625823974609375e-05,
      "step": 91048,
      "training_step_time": 8.071875810623169
    },
    {
      "epoch": 1.6259193420410156e-05,
      "model_forward_time": 1.7728745937347412,
      "step": 91049
    },
    {
      "epoch": 1.6259193420410156e-05,
      "step": 91049,
      "training_step_time": 7.626160621643066
    },
    {
      "epoch": 1.6260147094726563e-05,
      "grad_norm": 0.08929785341024399,
      "learning_rate": 2.1745145659788413e-07,
      "loss": 0.0567,
      "step": 91050
    },
    {
      "epoch": 1.6260147094726563e-05,
      "model_forward_time": 1.7595224380493164,
      "step": 91050
    },
    {
      "dynamics_loss_avg": 0.05297528514638543,
      "epoch": 1.6260147094726563e-05,
      "step": 91050
    },
    {
      "action_loss_avg": 0.003514393090154044,
      "epoch": 1.6260147094726563e-05,
      "step": 91050
    },
    {
      "epoch": 1.6260147094726563e-05,
      "step": 91050,
      "training_step_time": 7.454993486404419
    },
    {
      "epoch": 1.6261100769042967e-05,
      "model_forward_time": 1.8242998123168945,
      "step": 91051
    },
    {
      "epoch": 1.6261100769042967e-05,
      "step": 91051,
      "training_step_time": 7.702052354812622
    },
    {
      "epoch": 1.6262054443359375e-05,
      "model_forward_time": 1.7638492584228516,
      "step": 91052
    },
    {
      "epoch": 1.6262054443359375e-05,
      "step": 91052,
      "training_step_time": 7.744595527648926
    },
    {
      "epoch": 1.6263008117675782e-05,
      "model_forward_time": 1.7703235149383545,
      "step": 91053
    },
    {
      "epoch": 1.6263008117675782e-05,
      "step": 91053,
      "training_step_time": 9.38623857498169
    },
    {
      "epoch": 1.6263961791992186e-05,
      "model_forward_time": 1.7662560939788818,
      "step": 91054
    },
    {
      "epoch": 1.6263961791992186e-05,
      "step": 91054,
      "training_step_time": 8.551185369491577
    },
    {
      "epoch": 1.6264915466308593e-05,
      "model_forward_time": 1.7614703178405762,
      "step": 91055
    },
    {
      "epoch": 1.6264915466308593e-05,
      "step": 91055,
      "training_step_time": 7.671369314193726
    },
    {
      "epoch": 1.6265869140625e-05,
      "model_forward_time": 1.3997056484222412,
      "step": 91056
    },
    {
      "epoch": 1.6265869140625e-05,
      "step": 91056,
      "training_step_time": 8.545001029968262
    },
    {
      "epoch": 1.6266822814941405e-05,
      "model_forward_time": 1.7641925811767578,
      "step": 91057
    },
    {
      "epoch": 1.6266822814941405e-05,
      "step": 91057,
      "training_step_time": 7.942214012145996
    },
    {
      "epoch": 1.6267776489257812e-05,
      "model_forward_time": 1.7666780948638916,
      "step": 91058
    },
    {
      "epoch": 1.6267776489257812e-05,
      "step": 91058,
      "training_step_time": 7.633542776107788
    },
    {
      "epoch": 1.626873016357422e-05,
      "model_forward_time": 1.7474441528320312,
      "step": 91059
    },
    {
      "epoch": 1.626873016357422e-05,
      "step": 91059,
      "training_step_time": 7.4960668087005615
    },
    {
      "epoch": 1.6269683837890624e-05,
      "grad_norm": 0.1006932258605957,
      "learning_rate": 2.1696940015303647e-07,
      "loss": 0.056,
      "step": 91060
    },
    {
      "epoch": 1.6269683837890624e-05,
      "model_forward_time": 1.8083624839782715,
      "step": 91060
    },
    {
      "dynamics_loss_avg": 0.04227740652859211,
      "epoch": 1.6269683837890624e-05,
      "step": 91060
    },
    {
      "action_loss_avg": 0.0015850549447350205,
      "epoch": 1.6269683837890624e-05,
      "step": 91060
    },
    {
      "epoch": 1.6269683837890624e-05,
      "step": 91060,
      "training_step_time": 7.702958583831787
    },
    {
      "epoch": 1.627063751220703e-05,
      "model_forward_time": 1.7638635635375977,
      "step": 91061
    },
    {
      "epoch": 1.627063751220703e-05,
      "step": 91061,
      "training_step_time": 7.6548967361450195
    },
    {
      "epoch": 1.627159118652344e-05,
      "model_forward_time": 1.761169195175171,
      "step": 91062
    },
    {
      "epoch": 1.627159118652344e-05,
      "step": 91062,
      "training_step_time": 7.683759689331055
    },
    {
      "epoch": 1.6272544860839843e-05,
      "model_forward_time": 1.7626817226409912,
      "step": 91063
    },
    {
      "epoch": 1.6272544860839843e-05,
      "step": 91063,
      "training_step_time": 8.327628374099731
    },
    {
      "epoch": 1.627349853515625e-05,
      "model_forward_time": 2.8411645889282227,
      "step": 91064
    },
    {
      "epoch": 1.627349853515625e-05,
      "step": 91064,
      "training_step_time": 8.728949546813965
    },
    {
      "epoch": 1.6274452209472657e-05,
      "model_forward_time": 1.7561097145080566,
      "step": 91065
    },
    {
      "epoch": 1.6274452209472657e-05,
      "step": 91065,
      "training_step_time": 9.298916578292847
    },
    {
      "epoch": 1.627540588378906e-05,
      "model_forward_time": 1.788097858428955,
      "step": 91066
    },
    {
      "epoch": 1.627540588378906e-05,
      "step": 91066,
      "training_step_time": 7.820024490356445
    },
    {
      "epoch": 1.627635955810547e-05,
      "model_forward_time": 1.7753674983978271,
      "step": 91067
    },
    {
      "epoch": 1.627635955810547e-05,
      "step": 91067,
      "training_step_time": 8.167731046676636
    },
    {
      "epoch": 1.6277313232421876e-05,
      "model_forward_time": 1.7603583335876465,
      "step": 91068
    },
    {
      "epoch": 1.6277313232421876e-05,
      "step": 91068,
      "training_step_time": 7.592237234115601
    },
    {
      "epoch": 1.627826690673828e-05,
      "model_forward_time": 1.7583608627319336,
      "step": 91069
    },
    {
      "epoch": 1.627826690673828e-05,
      "step": 91069,
      "training_step_time": 7.7125208377838135
    },
    {
      "epoch": 1.6279220581054688e-05,
      "grad_norm": 0.07813571393489838,
      "learning_rate": 2.1648786677322397e-07,
      "loss": 0.0565,
      "step": 91070
    },
    {
      "epoch": 1.6279220581054688e-05,
      "model_forward_time": 1.831667423248291,
      "step": 91070
    },
    {
      "dynamics_loss_avg": 0.06189250610768795,
      "epoch": 1.6279220581054688e-05,
      "step": 91070
    },
    {
      "action_loss_avg": 0.0030502884503221138,
      "epoch": 1.6279220581054688e-05,
      "step": 91070
    },
    {
      "epoch": 1.6279220581054688e-05,
      "step": 91070,
      "training_step_time": 7.63841438293457
    },
    {
      "epoch": 1.6280174255371095e-05,
      "model_forward_time": 1.7597906589508057,
      "step": 91071
    },
    {
      "epoch": 1.6280174255371095e-05,
      "step": 91071,
      "training_step_time": 8.330284595489502
    },
    {
      "epoch": 1.62811279296875e-05,
      "model_forward_time": 1.7638804912567139,
      "step": 91072
    },
    {
      "epoch": 1.62811279296875e-05,
      "step": 91072,
      "training_step_time": 7.702620983123779
    },
    {
      "epoch": 1.6282081604003907e-05,
      "model_forward_time": 2.823657274246216,
      "step": 91073
    },
    {
      "epoch": 1.6282081604003907e-05,
      "step": 91073,
      "training_step_time": 9.967138051986694
    },
    {
      "epoch": 1.6283035278320314e-05,
      "model_forward_time": 1.7881779670715332,
      "step": 91074
    },
    {
      "epoch": 1.6283035278320314e-05,
      "step": 91074,
      "training_step_time": 7.7077248096466064
    },
    {
      "epoch": 1.6283988952636718e-05,
      "model_forward_time": 1.7643470764160156,
      "step": 91075
    },
    {
      "epoch": 1.6283988952636718e-05,
      "step": 91075,
      "training_step_time": 8.332319736480713
    },
    {
      "epoch": 1.6284942626953125e-05,
      "model_forward_time": 1.747929334640503,
      "step": 91076
    },
    {
      "epoch": 1.6284942626953125e-05,
      "step": 91076,
      "training_step_time": 7.9130659103393555
    },
    {
      "epoch": 1.6285896301269533e-05,
      "model_forward_time": 1.7571797370910645,
      "step": 91077
    },
    {
      "epoch": 1.6285896301269533e-05,
      "step": 91077,
      "training_step_time": 7.5466930866241455
    },
    {
      "epoch": 1.6286849975585937e-05,
      "model_forward_time": 1.7799737453460693,
      "step": 91078
    },
    {
      "epoch": 1.6286849975585937e-05,
      "step": 91078,
      "training_step_time": 7.684768438339233
    },
    {
      "epoch": 1.6287803649902344e-05,
      "model_forward_time": 1.8297502994537354,
      "step": 91079
    },
    {
      "epoch": 1.6287803649902344e-05,
      "step": 91079,
      "training_step_time": 8.73384165763855
    },
    {
      "epoch": 1.628875732421875e-05,
      "grad_norm": 0.08197598159313202,
      "learning_rate": 2.1600685651110676e-07,
      "loss": 0.0565,
      "step": 91080
    },
    {
      "epoch": 1.628875732421875e-05,
      "model_forward_time": 1.7597787380218506,
      "step": 91080
    },
    {
      "dynamics_loss_avg": 0.06501922775059939,
      "epoch": 1.628875732421875e-05,
      "step": 91080
    },
    {
      "action_loss_avg": 0.0024074283981462942,
      "epoch": 1.628875732421875e-05,
      "step": 91080
    },
    {
      "epoch": 1.628875732421875e-05,
      "step": 91080,
      "training_step_time": 7.609931468963623
    },
    {
      "epoch": 1.6289710998535156e-05,
      "model_forward_time": 1.7559788227081299,
      "step": 91081
    },
    {
      "epoch": 1.6289710998535156e-05,
      "step": 91081,
      "training_step_time": 9.566139936447144
    },
    {
      "epoch": 1.6290664672851563e-05,
      "model_forward_time": 2.82822847366333,
      "step": 91082
    },
    {
      "epoch": 1.6290664672851563e-05,
      "step": 91082,
      "training_step_time": 8.82055377960205
    },
    {
      "epoch": 1.6291618347167967e-05,
      "model_forward_time": 1.7550361156463623,
      "step": 91083
    },
    {
      "epoch": 1.6291618347167967e-05,
      "step": 91083,
      "training_step_time": 9.7130765914917
    },
    {
      "epoch": 1.6292572021484375e-05,
      "model_forward_time": 1.7580111026763916,
      "step": 91084
    },
    {
      "epoch": 1.6292572021484375e-05,
      "step": 91084,
      "training_step_time": 7.571447849273682
    },
    {
      "epoch": 1.6293525695800782e-05,
      "model_forward_time": 1.7626028060913086,
      "step": 91085
    },
    {
      "epoch": 1.6293525695800782e-05,
      "step": 91085,
      "training_step_time": 7.7859787940979
    },
    {
      "epoch": 1.6294479370117186e-05,
      "model_forward_time": 1.7651867866516113,
      "step": 91086
    },
    {
      "epoch": 1.6294479370117186e-05,
      "step": 91086,
      "training_step_time": 7.51103138923645
    },
    {
      "epoch": 1.6295433044433594e-05,
      "model_forward_time": 1.755810260772705,
      "step": 91087
    },
    {
      "epoch": 1.6295433044433594e-05,
      "step": 91087,
      "training_step_time": 7.687539577484131
    },
    {
      "epoch": 1.629638671875e-05,
      "model_forward_time": 1.8167078495025635,
      "step": 91088
    },
    {
      "epoch": 1.629638671875e-05,
      "step": 91088,
      "training_step_time": 7.54636025428772
    },
    {
      "epoch": 1.6297340393066405e-05,
      "model_forward_time": 1.7574894428253174,
      "step": 91089
    },
    {
      "epoch": 1.6297340393066405e-05,
      "step": 91089,
      "training_step_time": 8.602736234664917
    },
    {
      "epoch": 1.6298294067382812e-05,
      "grad_norm": 0.10625681281089783,
      "learning_rate": 2.1552636941928717e-07,
      "loss": 0.0558,
      "step": 91090
    },
    {
      "epoch": 1.6298294067382812e-05,
      "model_forward_time": 1.752396583557129,
      "step": 91090
    },
    {
      "dynamics_loss_avg": 0.05536841861903667,
      "epoch": 1.6298294067382812e-05,
      "step": 91090
    },
    {
      "action_loss_avg": 0.0014394431433174758,
      "epoch": 1.6298294067382812e-05,
      "step": 91090
    },
    {
      "epoch": 1.6298294067382812e-05,
      "step": 91090,
      "training_step_time": 9.386412620544434
    },
    {
      "epoch": 1.629924774169922e-05,
      "model_forward_time": 2.801147222518921,
      "step": 91091
    },
    {
      "epoch": 1.629924774169922e-05,
      "step": 91091,
      "training_step_time": 9.670893430709839
    },
    {
      "epoch": 1.6300201416015624e-05,
      "model_forward_time": 1.7602431774139404,
      "step": 91092
    },
    {
      "epoch": 1.6300201416015624e-05,
      "step": 91092,
      "training_step_time": 7.74853515625
    },
    {
      "epoch": 1.630115509033203e-05,
      "model_forward_time": 1.7749371528625488,
      "step": 91093
    },
    {
      "epoch": 1.630115509033203e-05,
      "step": 91093,
      "training_step_time": 7.741055488586426
    },
    {
      "epoch": 1.630210876464844e-05,
      "model_forward_time": 1.7704308032989502,
      "step": 91094
    },
    {
      "epoch": 1.630210876464844e-05,
      "step": 91094,
      "training_step_time": 7.738303184509277
    },
    {
      "epoch": 1.6303062438964843e-05,
      "model_forward_time": 1.7711732387542725,
      "step": 91095
    },
    {
      "epoch": 1.6303062438964843e-05,
      "step": 91095,
      "training_step_time": 8.201014757156372
    },
    {
      "epoch": 1.630401611328125e-05,
      "model_forward_time": 1.764704942703247,
      "step": 91096
    },
    {
      "epoch": 1.630401611328125e-05,
      "step": 91096,
      "training_step_time": 7.697195291519165
    },
    {
      "epoch": 1.6304969787597657e-05,
      "model_forward_time": 1.7883086204528809,
      "step": 91097
    },
    {
      "epoch": 1.6304969787597657e-05,
      "step": 91097,
      "training_step_time": 8.587902545928955
    },
    {
      "epoch": 1.630592346191406e-05,
      "model_forward_time": 1.7638952732086182,
      "step": 91098
    },
    {
      "epoch": 1.630592346191406e-05,
      "step": 91098,
      "training_step_time": 9.53090214729309
    },
    {
      "epoch": 1.630687713623047e-05,
      "model_forward_time": 1.7713878154754639,
      "step": 91099
    },
    {
      "epoch": 1.630687713623047e-05,
      "step": 91099,
      "training_step_time": 8.441096067428589
    },
    {
      "epoch": 1.6307830810546876e-05,
      "grad_norm": 0.12891562283039093,
      "learning_rate": 2.1504640555031042e-07,
      "loss": 0.0553,
      "step": 91100
    },
    {
      "epoch": 1.6307830810546876e-05,
      "model_forward_time": 1.7529873847961426,
      "step": 91100
    },
    {
      "dynamics_loss_avg": 0.05879182033240795,
      "epoch": 1.6307830810546876e-05,
      "step": 91100
    },
    {
      "action_loss_avg": 0.003571265217033215,
      "epoch": 1.6307830810546876e-05,
      "step": 91100
    },
    {
      "epoch": 1.6307830810546876e-05,
      "step": 91100,
      "training_step_time": 8.955754280090332
    },
    {
      "epoch": 1.630878448486328e-05,
      "model_forward_time": 1.7373342514038086,
      "step": 91101
    },
    {
      "epoch": 1.630878448486328e-05,
      "step": 91101,
      "training_step_time": 7.646151304244995
    },
    {
      "epoch": 1.6309738159179688e-05,
      "model_forward_time": 1.7576379776000977,
      "step": 91102
    },
    {
      "epoch": 1.6309738159179688e-05,
      "step": 91102,
      "training_step_time": 7.693292617797852
    },
    {
      "epoch": 1.6310691833496095e-05,
      "model_forward_time": 1.7672786712646484,
      "step": 91103
    },
    {
      "epoch": 1.6310691833496095e-05,
      "step": 91103,
      "training_step_time": 7.711779356002808
    },
    {
      "epoch": 1.63116455078125e-05,
      "model_forward_time": 1.7704150676727295,
      "step": 91104
    },
    {
      "epoch": 1.63116455078125e-05,
      "step": 91104,
      "training_step_time": 7.613785743713379
    },
    {
      "epoch": 1.6312599182128907e-05,
      "model_forward_time": 1.7670199871063232,
      "step": 91105
    },
    {
      "epoch": 1.6312599182128907e-05,
      "step": 91105,
      "training_step_time": 7.656437635421753
    },
    {
      "epoch": 1.6313552856445314e-05,
      "model_forward_time": 1.766657829284668,
      "step": 91106
    },
    {
      "epoch": 1.6313552856445314e-05,
      "step": 91106,
      "training_step_time": 8.158978700637817
    },
    {
      "epoch": 1.6314506530761718e-05,
      "model_forward_time": 1.7667670249938965,
      "step": 91107
    },
    {
      "epoch": 1.6314506530761718e-05,
      "step": 91107,
      "training_step_time": 7.601374626159668
    },
    {
      "epoch": 1.6315460205078126e-05,
      "model_forward_time": 1.764763593673706,
      "step": 91108
    },
    {
      "epoch": 1.6315460205078126e-05,
      "step": 91108,
      "training_step_time": 8.109352350234985
    },
    {
      "epoch": 1.6316413879394533e-05,
      "model_forward_time": 1.7369651794433594,
      "step": 91109
    },
    {
      "epoch": 1.6316413879394533e-05,
      "step": 91109,
      "training_step_time": 7.637476921081543
    },
    {
      "epoch": 1.6317367553710937e-05,
      "grad_norm": 0.17114907503128052,
      "learning_rate": 2.1456696495666508e-07,
      "loss": 0.0555,
      "step": 91110
    },
    {
      "epoch": 1.6317367553710937e-05,
      "model_forward_time": 2.8263466358184814,
      "step": 91110
    },
    {
      "dynamics_loss_avg": 0.05927961934357882,
      "epoch": 1.6317367553710937e-05,
      "step": 91110
    },
    {
      "action_loss_avg": 0.0021333761455025525,
      "epoch": 1.6317367553710937e-05,
      "step": 91110
    },
    {
      "epoch": 1.6317367553710937e-05,
      "step": 91110,
      "training_step_time": 8.807031154632568
    },
    {
      "epoch": 1.6318321228027344e-05,
      "model_forward_time": 1.7512273788452148,
      "step": 91111
    },
    {
      "epoch": 1.6318321228027344e-05,
      "step": 91111,
      "training_step_time": 7.7250566482543945
    },
    {
      "epoch": 1.631927490234375e-05,
      "model_forward_time": 1.7656590938568115,
      "step": 91112
    },
    {
      "epoch": 1.631927490234375e-05,
      "step": 91112,
      "training_step_time": 8.822464942932129
    },
    {
      "epoch": 1.6320228576660156e-05,
      "model_forward_time": 1.7541851997375488,
      "step": 91113
    },
    {
      "epoch": 1.6320228576660156e-05,
      "step": 91113,
      "training_step_time": 7.624453783035278
    },
    {
      "epoch": 1.6321182250976563e-05,
      "model_forward_time": 1.75270676612854,
      "step": 91114
    },
    {
      "epoch": 1.6321182250976563e-05,
      "step": 91114,
      "training_step_time": 7.766453266143799
    },
    {
      "epoch": 1.6322135925292967e-05,
      "model_forward_time": 1.7581489086151123,
      "step": 91115
    },
    {
      "epoch": 1.6322135925292967e-05,
      "step": 91115,
      "training_step_time": 7.786350250244141
    },
    {
      "epoch": 1.6323089599609375e-05,
      "model_forward_time": 1.8245141506195068,
      "step": 91116
    },
    {
      "epoch": 1.6323089599609375e-05,
      "step": 91116,
      "training_step_time": 7.673730850219727
    },
    {
      "epoch": 1.6324043273925782e-05,
      "model_forward_time": 1.7521953582763672,
      "step": 91117
    },
    {
      "epoch": 1.6324043273925782e-05,
      "step": 91117,
      "training_step_time": 7.545022964477539
    },
    {
      "epoch": 1.6324996948242186e-05,
      "model_forward_time": 1.7539827823638916,
      "step": 91118
    },
    {
      "epoch": 1.6324996948242186e-05,
      "step": 91118,
      "training_step_time": 7.77461838722229
    },
    {
      "epoch": 1.6325950622558594e-05,
      "model_forward_time": 2.8300628662109375,
      "step": 91119
    },
    {
      "epoch": 1.6325950622558594e-05,
      "step": 91119,
      "training_step_time": 9.848419427871704
    },
    {
      "epoch": 1.6326904296875e-05,
      "grad_norm": 0.07515369355678558,
      "learning_rate": 2.1408804769078085e-07,
      "loss": 0.055,
      "step": 91120
    },
    {
      "epoch": 1.6326904296875e-05,
      "model_forward_time": 1.7539782524108887,
      "step": 91120
    },
    {
      "dynamics_loss_avg": 0.06219916790723801,
      "epoch": 1.6326904296875e-05,
      "step": 91120
    },
    {
      "action_loss_avg": 0.00401669353741454,
      "epoch": 1.6326904296875e-05,
      "step": 91120
    },
    {
      "epoch": 1.6326904296875e-05,
      "step": 91120,
      "training_step_time": 8.822311401367188
    },
    {
      "epoch": 1.6327857971191405e-05,
      "model_forward_time": 1.754103660583496,
      "step": 91121
    },
    {
      "epoch": 1.6327857971191405e-05,
      "step": 91121,
      "training_step_time": 9.3325035572052
    },
    {
      "epoch": 1.6328811645507812e-05,
      "model_forward_time": 1.7586750984191895,
      "step": 91122
    },
    {
      "epoch": 1.6328811645507812e-05,
      "step": 91122,
      "training_step_time": 7.715672492980957
    },
    {
      "epoch": 1.632976531982422e-05,
      "model_forward_time": 1.7512316703796387,
      "step": 91123
    },
    {
      "epoch": 1.632976531982422e-05,
      "step": 91123,
      "training_step_time": 7.831421613693237
    },
    {
      "epoch": 1.6330718994140624e-05,
      "model_forward_time": 1.7487585544586182,
      "step": 91124
    },
    {
      "epoch": 1.6330718994140624e-05,
      "step": 91124,
      "training_step_time": 7.699756622314453
    },
    {
      "epoch": 1.633167266845703e-05,
      "model_forward_time": 1.8227412700653076,
      "step": 91125
    },
    {
      "epoch": 1.633167266845703e-05,
      "step": 91125,
      "training_step_time": 8.625927925109863
    },
    {
      "epoch": 1.633262634277344e-05,
      "model_forward_time": 1.7600538730621338,
      "step": 91126
    },
    {
      "epoch": 1.633262634277344e-05,
      "step": 91126,
      "training_step_time": 7.686772346496582
    },
    {
      "epoch": 1.6333580017089843e-05,
      "model_forward_time": 1.7498183250427246,
      "step": 91127
    },
    {
      "epoch": 1.6333580017089843e-05,
      "step": 91127,
      "training_step_time": 8.839291334152222
    },
    {
      "epoch": 1.633453369140625e-05,
      "model_forward_time": 2.825949192047119,
      "step": 91128
    },
    {
      "epoch": 1.633453369140625e-05,
      "step": 91128,
      "training_step_time": 8.79440712928772
    },
    {
      "epoch": 1.6335487365722658e-05,
      "model_forward_time": 1.7527995109558105,
      "step": 91129
    },
    {
      "epoch": 1.6335487365722658e-05,
      "step": 91129,
      "training_step_time": 10.186896562576294
    },
    {
      "epoch": 1.633644104003906e-05,
      "grad_norm": 0.08477509021759033,
      "learning_rate": 2.136096538050325e-07,
      "loss": 0.0554,
      "step": 91130
    },
    {
      "epoch": 1.633644104003906e-05,
      "model_forward_time": 1.7749261856079102,
      "step": 91130
    },
    {
      "dynamics_loss_avg": 0.053730764985084535,
      "epoch": 1.633644104003906e-05,
      "step": 91130
    },
    {
      "action_loss_avg": 0.0011814381403382867,
      "epoch": 1.633644104003906e-05,
      "step": 91130
    },
    {
      "epoch": 1.633644104003906e-05,
      "step": 91130,
      "training_step_time": 9.353825569152832
    },
    {
      "epoch": 1.633739471435547e-05,
      "model_forward_time": 1.7532742023468018,
      "step": 91131
    },
    {
      "epoch": 1.633739471435547e-05,
      "step": 91131,
      "training_step_time": 7.7336509227752686
    },
    {
      "epoch": 1.6338348388671876e-05,
      "model_forward_time": 1.7541234493255615,
      "step": 91132
    },
    {
      "epoch": 1.6338348388671876e-05,
      "step": 91132,
      "training_step_time": 7.689104080200195
    },
    {
      "epoch": 1.633930206298828e-05,
      "model_forward_time": 1.7524163722991943,
      "step": 91133
    },
    {
      "epoch": 1.633930206298828e-05,
      "step": 91133,
      "training_step_time": 7.568115234375
    },
    {
      "epoch": 1.6340255737304688e-05,
      "model_forward_time": 1.8182525634765625,
      "step": 91134
    },
    {
      "epoch": 1.6340255737304688e-05,
      "step": 91134,
      "training_step_time": 8.457915544509888
    },
    {
      "epoch": 1.6341209411621095e-05,
      "model_forward_time": 1.7504513263702393,
      "step": 91135
    },
    {
      "epoch": 1.6341209411621095e-05,
      "step": 91135,
      "training_step_time": 7.565362215042114
    },
    {
      "epoch": 1.63421630859375e-05,
      "model_forward_time": 1.7569565773010254,
      "step": 91136
    },
    {
      "epoch": 1.63421630859375e-05,
      "step": 91136,
      "training_step_time": 8.935083389282227
    },
    {
      "epoch": 1.6343116760253907e-05,
      "model_forward_time": 2.853372097015381,
      "step": 91137
    },
    {
      "epoch": 1.6343116760253907e-05,
      "step": 91137,
      "training_step_time": 9.637008428573608
    },
    {
      "epoch": 1.6344070434570314e-05,
      "model_forward_time": 1.7462704181671143,
      "step": 91138
    },
    {
      "epoch": 1.6344070434570314e-05,
      "step": 91138,
      "training_step_time": 8.437535762786865
    },
    {
      "epoch": 1.6345024108886718e-05,
      "model_forward_time": 1.7413089275360107,
      "step": 91139
    },
    {
      "epoch": 1.6345024108886718e-05,
      "step": 91139,
      "training_step_time": 8.28455114364624
    },
    {
      "epoch": 1.6345977783203126e-05,
      "grad_norm": 0.08336672186851501,
      "learning_rate": 2.13131783351736e-07,
      "loss": 0.0548,
      "step": 91140
    },
    {
      "epoch": 1.6345977783203126e-05,
      "model_forward_time": 1.7476835250854492,
      "step": 91140
    },
    {
      "dynamics_loss_avg": 0.057718781568109986,
      "epoch": 1.6345977783203126e-05,
      "step": 91140
    },
    {
      "action_loss_avg": 0.00312587707885541,
      "epoch": 1.6345977783203126e-05,
      "step": 91140
    },
    {
      "epoch": 1.6345977783203126e-05,
      "step": 91140,
      "training_step_time": 7.683960676193237
    },
    {
      "epoch": 1.634693145751953e-05,
      "model_forward_time": 1.7449886798858643,
      "step": 91141
    },
    {
      "epoch": 1.634693145751953e-05,
      "step": 91141,
      "training_step_time": 7.683671236038208
    },
    {
      "epoch": 1.6347885131835937e-05,
      "model_forward_time": 1.7483534812927246,
      "step": 91142
    },
    {
      "epoch": 1.6347885131835937e-05,
      "step": 91142,
      "training_step_time": 7.569879531860352
    },
    {
      "epoch": 1.6348838806152344e-05,
      "model_forward_time": 1.7661972045898438,
      "step": 91143
    },
    {
      "epoch": 1.6348838806152344e-05,
      "step": 91143,
      "training_step_time": 7.613110303878784
    },
    {
      "epoch": 1.634979248046875e-05,
      "model_forward_time": 1.7483112812042236,
      "step": 91144
    },
    {
      "epoch": 1.634979248046875e-05,
      "step": 91144,
      "training_step_time": 7.5362465381622314
    },
    {
      "epoch": 1.6350746154785156e-05,
      "model_forward_time": 1.7439661026000977,
      "step": 91145
    },
    {
      "epoch": 1.6350746154785156e-05,
      "step": 91145,
      "training_step_time": 8.561245918273926
    },
    {
      "epoch": 1.6351699829101563e-05,
      "model_forward_time": 2.8342859745025635,
      "step": 91146
    },
    {
      "epoch": 1.6351699829101563e-05,
      "step": 91146,
      "training_step_time": 9.20481562614441
    },
    {
      "epoch": 1.6352653503417967e-05,
      "model_forward_time": 1.7458574771881104,
      "step": 91147
    },
    {
      "epoch": 1.6352653503417967e-05,
      "step": 91147,
      "training_step_time": 9.058183908462524
    },
    {
      "epoch": 1.6353607177734375e-05,
      "model_forward_time": 1.747049331665039,
      "step": 91148
    },
    {
      "epoch": 1.6353607177734375e-05,
      "step": 91148,
      "training_step_time": 8.464494705200195
    },
    {
      "epoch": 1.6354560852050782e-05,
      "model_forward_time": 1.7786481380462646,
      "step": 91149
    },
    {
      "epoch": 1.6354560852050782e-05,
      "step": 91149,
      "training_step_time": 7.626749753952026
    },
    {
      "epoch": 1.6355514526367186e-05,
      "grad_norm": 0.13279452919960022,
      "learning_rate": 2.1265443638315118e-07,
      "loss": 0.0558,
      "step": 91150
    },
    {
      "epoch": 1.6355514526367186e-05,
      "model_forward_time": 1.7333693504333496,
      "step": 91150
    },
    {
      "dynamics_loss_avg": 0.06145446337759495,
      "epoch": 1.6355514526367186e-05,
      "step": 91150
    },
    {
      "action_loss_avg": 0.00191823918139562,
      "epoch": 1.6355514526367186e-05,
      "step": 91150
    },
    {
      "epoch": 1.6355514526367186e-05,
      "step": 91150,
      "training_step_time": 9.094611644744873
    },
    {
      "epoch": 1.6356468200683594e-05,
      "model_forward_time": 1.3927948474884033,
      "step": 91151
    },
    {
      "epoch": 1.6356468200683594e-05,
      "step": 91151,
      "training_step_time": 7.576486825942993
    },
    {
      "epoch": 1.6357421875e-05,
      "model_forward_time": 1.7489135265350342,
      "step": 91152
    },
    {
      "epoch": 1.6357421875e-05,
      "step": 91152,
      "training_step_time": 7.5989415645599365
    },
    {
      "epoch": 1.6358375549316405e-05,
      "model_forward_time": 1.8224084377288818,
      "step": 91153
    },
    {
      "epoch": 1.6358375549316405e-05,
      "step": 91153,
      "training_step_time": 11.161497592926025
    },
    {
      "epoch": 1.6359329223632812e-05,
      "model_forward_time": 1.7714440822601318,
      "step": 91154
    },
    {
      "epoch": 1.6359329223632812e-05,
      "step": 91154,
      "training_step_time": 7.5754289627075195
    },
    {
      "epoch": 1.636028289794922e-05,
      "model_forward_time": 1.7488534450531006,
      "step": 91155
    },
    {
      "epoch": 1.636028289794922e-05,
      "step": 91155,
      "training_step_time": 7.648219347000122
    },
    {
      "epoch": 1.6361236572265624e-05,
      "model_forward_time": 1.7444186210632324,
      "step": 91156
    },
    {
      "epoch": 1.6361236572265624e-05,
      "step": 91156,
      "training_step_time": 8.189534187316895
    },
    {
      "epoch": 1.636219024658203e-05,
      "model_forward_time": 1.7534360885620117,
      "step": 91157
    },
    {
      "epoch": 1.636219024658203e-05,
      "step": 91157,
      "training_step_time": 7.6176917552948
    },
    {
      "epoch": 1.636314392089844e-05,
      "model_forward_time": 1.7725622653961182,
      "step": 91158
    },
    {
      "epoch": 1.636314392089844e-05,
      "step": 91158,
      "training_step_time": 6.907692909240723
    },
    {
      "epoch": 1.6364097595214843e-05,
      "model_forward_time": 1.7514708042144775,
      "step": 91159
    },
    {
      "epoch": 1.6364097595214843e-05,
      "step": 91159,
      "training_step_time": 6.853070497512817
    },
    {
      "epoch": 1.636505126953125e-05,
      "grad_norm": 0.10119039565324783,
      "learning_rate": 2.121776129514791e-07,
      "loss": 0.0545,
      "step": 91160
    },
    {
      "epoch": 1.636505126953125e-05,
      "model_forward_time": 1.7557408809661865,
      "step": 91160
    },
    {
      "dynamics_loss_avg": 0.05365287698805332,
      "epoch": 1.636505126953125e-05,
      "step": 91160
    },
    {
      "action_loss_avg": 0.001705951668554917,
      "epoch": 1.636505126953125e-05,
      "step": 91160
    },
    {
      "epoch": 1.636505126953125e-05,
      "step": 91160,
      "training_step_time": 6.848256349563599
    },
    {
      "epoch": 1.6366004943847658e-05,
      "model_forward_time": 1.7492692470550537,
      "step": 91161
    },
    {
      "epoch": 1.6366004943847658e-05,
      "step": 91161,
      "training_step_time": 6.860532283782959
    },
    {
      "epoch": 1.636695861816406e-05,
      "model_forward_time": 1.7787535190582275,
      "step": 91162
    },
    {
      "epoch": 1.636695861816406e-05,
      "step": 91162,
      "training_step_time": 6.837575674057007
    },
    {
      "epoch": 1.636791229248047e-05,
      "model_forward_time": 2.1679859161376953,
      "step": 91163
    },
    {
      "epoch": 1.636791229248047e-05,
      "step": 91163,
      "training_step_time": 7.731974363327026
    },
    {
      "epoch": 1.6368865966796876e-05,
      "model_forward_time": 1.752068042755127,
      "step": 91164
    },
    {
      "epoch": 1.6368865966796876e-05,
      "step": 91164,
      "training_step_time": 7.51183009147644
    },
    {
      "epoch": 1.636981964111328e-05,
      "model_forward_time": 1.751546859741211,
      "step": 91165
    },
    {
      "epoch": 1.636981964111328e-05,
      "step": 91165,
      "training_step_time": 7.625267505645752
    },
    {
      "epoch": 1.6370773315429688e-05,
      "model_forward_time": 1.7815067768096924,
      "step": 91166
    },
    {
      "epoch": 1.6370773315429688e-05,
      "step": 91166,
      "training_step_time": 9.273783206939697
    },
    {
      "epoch": 1.6371726989746095e-05,
      "model_forward_time": 1.7487359046936035,
      "step": 91167
    },
    {
      "epoch": 1.6371726989746095e-05,
      "step": 91167,
      "training_step_time": 6.82724666595459
    },
    {
      "epoch": 1.63726806640625e-05,
      "model_forward_time": 1.7510359287261963,
      "step": 91168
    },
    {
      "epoch": 1.63726806640625e-05,
      "step": 91168,
      "training_step_time": 6.852904558181763
    },
    {
      "epoch": 1.6373634338378907e-05,
      "model_forward_time": 1.7703559398651123,
      "step": 91169
    },
    {
      "epoch": 1.6373634338378907e-05,
      "step": 91169,
      "training_step_time": 6.813294172286987
    },
    {
      "epoch": 1.6374588012695314e-05,
      "grad_norm": 0.07133488357067108,
      "learning_rate": 2.1170131310886522e-07,
      "loss": 0.0546,
      "step": 91170
    },
    {
      "epoch": 1.6374588012695314e-05,
      "model_forward_time": 1.7680537700653076,
      "step": 91170
    },
    {
      "dynamics_loss_avg": 0.06091786473989487,
      "epoch": 1.6374588012695314e-05,
      "step": 91170
    },
    {
      "action_loss_avg": 0.0028382818098179998,
      "epoch": 1.6374588012695314e-05,
      "step": 91170
    },
    {
      "epoch": 1.6374588012695314e-05,
      "step": 91170,
      "training_step_time": 6.803411245346069
    },
    {
      "epoch": 1.6375541687011718e-05,
      "model_forward_time": 1.7534093856811523,
      "step": 91171
    },
    {
      "epoch": 1.6375541687011718e-05,
      "step": 91171,
      "training_step_time": 6.820393323898315
    },
    {
      "epoch": 1.6376495361328126e-05,
      "model_forward_time": 2.928102731704712,
      "step": 91172
    },
    {
      "epoch": 1.6376495361328126e-05,
      "step": 91172,
      "training_step_time": 8.6293785572052
    },
    {
      "epoch": 1.637744903564453e-05,
      "model_forward_time": 1.7658891677856445,
      "step": 91173
    },
    {
      "epoch": 1.637744903564453e-05,
      "step": 91173,
      "training_step_time": 7.592684030532837
    },
    {
      "epoch": 1.6378402709960937e-05,
      "model_forward_time": 1.7733509540557861,
      "step": 91174
    },
    {
      "epoch": 1.6378402709960937e-05,
      "step": 91174,
      "training_step_time": 7.669369220733643
    },
    {
      "epoch": 1.6379356384277344e-05,
      "model_forward_time": 1.7649908065795898,
      "step": 91175
    },
    {
      "epoch": 1.6379356384277344e-05,
      "step": 91175,
      "training_step_time": 7.89456582069397
    },
    {
      "epoch": 1.638031005859375e-05,
      "model_forward_time": 1.7692599296569824,
      "step": 91176
    },
    {
      "epoch": 1.638031005859375e-05,
      "step": 91176,
      "training_step_time": 7.718366384506226
    },
    {
      "epoch": 1.6381263732910156e-05,
      "model_forward_time": 1.7670462131500244,
      "step": 91177
    },
    {
      "epoch": 1.6381263732910156e-05,
      "step": 91177,
      "training_step_time": 6.828572034835815
    },
    {
      "epoch": 1.6382217407226563e-05,
      "model_forward_time": 1.778149127960205,
      "step": 91178
    },
    {
      "epoch": 1.6382217407226563e-05,
      "step": 91178,
      "training_step_time": 6.8208582401275635
    },
    {
      "epoch": 1.6383171081542967e-05,
      "model_forward_time": 1.7745790481567383,
      "step": 91179
    },
    {
      "epoch": 1.6383171081542967e-05,
      "step": 91179,
      "training_step_time": 6.825724363327026
    },
    {
      "epoch": 1.6384124755859375e-05,
      "grad_norm": 0.12927590310573578,
      "learning_rate": 2.1122553690739678e-07,
      "loss": 0.0543,
      "step": 91180
    },
    {
      "epoch": 1.6384124755859375e-05,
      "model_forward_time": 1.7728180885314941,
      "step": 91180
    },
    {
      "dynamics_loss_avg": 0.056041302252560855,
      "epoch": 1.6384124755859375e-05,
      "step": 91180
    },
    {
      "action_loss_avg": 0.0013857662794180214,
      "epoch": 1.6384124755859375e-05,
      "step": 91180
    },
    {
      "epoch": 1.6384124755859375e-05,
      "step": 91180,
      "training_step_time": 6.817262411117554
    },
    {
      "epoch": 1.6385078430175782e-05,
      "model_forward_time": 2.8781471252441406,
      "step": 91181
    },
    {
      "epoch": 1.6385078430175782e-05,
      "step": 91181,
      "training_step_time": 7.789426565170288
    },
    {
      "epoch": 1.6386032104492186e-05,
      "model_forward_time": 1.835541009902954,
      "step": 91182
    },
    {
      "epoch": 1.6386032104492186e-05,
      "step": 91182,
      "training_step_time": 7.862810850143433
    },
    {
      "epoch": 1.6386985778808594e-05,
      "model_forward_time": 1.7746374607086182,
      "step": 91183
    },
    {
      "epoch": 1.6386985778808594e-05,
      "step": 91183,
      "training_step_time": 7.103857040405273
    },
    {
      "epoch": 1.6387939453125e-05,
      "model_forward_time": 1.793457269668579,
      "step": 91184
    },
    {
      "epoch": 1.6387939453125e-05,
      "step": 91184,
      "training_step_time": 8.513615369796753
    },
    {
      "epoch": 1.6388893127441405e-05,
      "model_forward_time": 1.765038251876831,
      "step": 91185
    },
    {
      "epoch": 1.6388893127441405e-05,
      "step": 91185,
      "training_step_time": 7.796408176422119
    },
    {
      "epoch": 1.6389846801757812e-05,
      "model_forward_time": 1.7749834060668945,
      "step": 91186
    },
    {
      "epoch": 1.6389846801757812e-05,
      "step": 91186,
      "training_step_time": 6.776541471481323
    },
    {
      "epoch": 1.639080047607422e-05,
      "model_forward_time": 1.7681944370269775,
      "step": 91187
    },
    {
      "epoch": 1.639080047607422e-05,
      "step": 91187,
      "training_step_time": 6.793461561203003
    },
    {
      "epoch": 1.6391754150390624e-05,
      "model_forward_time": 1.7686243057250977,
      "step": 91188
    },
    {
      "epoch": 1.6391754150390624e-05,
      "step": 91188,
      "training_step_time": 6.763975620269775
    },
    {
      "epoch": 1.639270782470703e-05,
      "model_forward_time": 1.7705514430999756,
      "step": 91189
    },
    {
      "epoch": 1.639270782470703e-05,
      "step": 91189,
      "training_step_time": 6.817826509475708
    },
    {
      "epoch": 1.639366149902344e-05,
      "grad_norm": 0.07408276200294495,
      "learning_rate": 2.1075028439910327e-07,
      "loss": 0.0542,
      "step": 91190
    },
    {
      "epoch": 1.639366149902344e-05,
      "model_forward_time": 1.7643780708312988,
      "step": 91190
    },
    {
      "dynamics_loss_avg": 0.047187444753944874,
      "epoch": 1.639366149902344e-05,
      "step": 91190
    },
    {
      "action_loss_avg": 0.001936640715575777,
      "epoch": 1.639366149902344e-05,
      "step": 91190
    },
    {
      "epoch": 1.639366149902344e-05,
      "step": 91190,
      "training_step_time": 10.534675598144531
    },
    {
      "epoch": 1.6394615173339843e-05,
      "model_forward_time": 1.8287153244018555,
      "step": 91191
    },
    {
      "epoch": 1.6394615173339843e-05,
      "step": 91191,
      "training_step_time": 7.6977458000183105
    },
    {
      "epoch": 1.639556884765625e-05,
      "model_forward_time": 1.7662043571472168,
      "step": 91192
    },
    {
      "epoch": 1.639556884765625e-05,
      "step": 91192,
      "training_step_time": 8.086374998092651
    },
    {
      "epoch": 1.6396522521972658e-05,
      "model_forward_time": 1.7659218311309814,
      "step": 91193
    },
    {
      "epoch": 1.6396522521972658e-05,
      "step": 91193,
      "training_step_time": 7.566072225570679
    },
    {
      "epoch": 1.639747619628906e-05,
      "model_forward_time": 1.7688887119293213,
      "step": 91194
    },
    {
      "epoch": 1.639747619628906e-05,
      "step": 91194,
      "training_step_time": 9.445739269256592
    },
    {
      "epoch": 1.639842987060547e-05,
      "model_forward_time": 1.7676451206207275,
      "step": 91195
    },
    {
      "epoch": 1.639842987060547e-05,
      "step": 91195,
      "training_step_time": 6.796215534210205
    },
    {
      "epoch": 1.6399383544921876e-05,
      "model_forward_time": 1.7652485370635986,
      "step": 91196
    },
    {
      "epoch": 1.6399383544921876e-05,
      "step": 91196,
      "training_step_time": 6.796412467956543
    },
    {
      "epoch": 1.640033721923828e-05,
      "model_forward_time": 1.767054557800293,
      "step": 91197
    },
    {
      "epoch": 1.640033721923828e-05,
      "step": 91197,
      "training_step_time": 6.800450325012207
    },
    {
      "epoch": 1.6401290893554688e-05,
      "model_forward_time": 1.7681481838226318,
      "step": 91198
    },
    {
      "epoch": 1.6401290893554688e-05,
      "step": 91198,
      "training_step_time": 6.837141275405884
    },
    {
      "epoch": 1.6402244567871095e-05,
      "model_forward_time": 1.7717821598052979,
      "step": 91199
    },
    {
      "epoch": 1.6402244567871095e-05,
      "step": 91199,
      "training_step_time": 8.711110353469849
    },
    {
      "epoch": 1.64031982421875e-05,
      "grad_norm": 0.06334280222654343,
      "learning_rate": 2.1027555563595813e-07,
      "loss": 0.055,
      "step": 91200
    },
    {
      "epoch": 1.64031982421875e-05,
      "model_forward_time": 2.932295799255371,
      "step": 91200
    },
    {
      "dynamics_loss_avg": 0.05709922332316637,
      "epoch": 1.64031982421875e-05,
      "step": 91200
    },
    {
      "action_loss_avg": 0.002291075960965827,
      "epoch": 1.64031982421875e-05,
      "step": 91200
    },
    {
      "epoch": 1.64031982421875e-05,
      "step": 91200,
      "training_step_time": 7.739370584487915
    },
    {
      "epoch": 1.6404151916503907e-05,
      "model_forward_time": 1.7705004215240479,
      "step": 91201
    },
    {
      "epoch": 1.6404151916503907e-05,
      "step": 91201,
      "training_step_time": 7.516876935958862
    },
    {
      "epoch": 1.6405105590820314e-05,
      "model_forward_time": 1.768195629119873,
      "step": 91202
    },
    {
      "epoch": 1.6405105590820314e-05,
      "step": 91202,
      "training_step_time": 7.680257558822632
    },
    {
      "epoch": 1.6406059265136718e-05,
      "model_forward_time": 1.7670648097991943,
      "step": 91203
    },
    {
      "epoch": 1.6406059265136718e-05,
      "step": 91203,
      "training_step_time": 7.778262138366699
    },
    {
      "epoch": 1.6407012939453126e-05,
      "model_forward_time": 1.7773351669311523,
      "step": 91204
    },
    {
      "epoch": 1.6407012939453126e-05,
      "step": 91204,
      "training_step_time": 7.37194299697876
    },
    {
      "epoch": 1.640796661376953e-05,
      "model_forward_time": 1.8266081809997559,
      "step": 91205
    },
    {
      "epoch": 1.640796661376953e-05,
      "step": 91205,
      "training_step_time": 6.821274518966675
    },
    {
      "epoch": 1.6408920288085937e-05,
      "model_forward_time": 1.7696235179901123,
      "step": 91206
    },
    {
      "epoch": 1.6408920288085937e-05,
      "step": 91206,
      "training_step_time": 6.813142538070679
    },
    {
      "epoch": 1.6409873962402344e-05,
      "model_forward_time": 1.7715189456939697,
      "step": 91207
    },
    {
      "epoch": 1.6409873962402344e-05,
      "step": 91207,
      "training_step_time": 6.8322834968566895
    },
    {
      "epoch": 1.641082763671875e-05,
      "model_forward_time": 1.7631168365478516,
      "step": 91208
    },
    {
      "epoch": 1.641082763671875e-05,
      "step": 91208,
      "training_step_time": 8.177322149276733
    },
    {
      "epoch": 1.6411781311035156e-05,
      "model_forward_time": 2.9022257328033447,
      "step": 91209
    },
    {
      "epoch": 1.6411781311035156e-05,
      "step": 91209,
      "training_step_time": 7.702986717224121
    },
    {
      "epoch": 1.6412734985351563e-05,
      "grad_norm": 0.0792468786239624,
      "learning_rate": 2.0980135066987705e-07,
      "loss": 0.0548,
      "step": 91210
    },
    {
      "epoch": 1.6412734985351563e-05,
      "model_forward_time": 1.8590741157531738,
      "step": 91210
    },
    {
      "dynamics_loss_avg": 0.06983792912214995,
      "epoch": 1.6412734985351563e-05,
      "step": 91210
    },
    {
      "action_loss_avg": 0.0012428125832229854,
      "epoch": 1.6412734985351563e-05,
      "step": 91210
    },
    {
      "epoch": 1.6412734985351563e-05,
      "step": 91210,
      "training_step_time": 7.7163450717926025
    },
    {
      "epoch": 1.6413688659667967e-05,
      "model_forward_time": 1.7683498859405518,
      "step": 91211
    },
    {
      "epoch": 1.6413688659667967e-05,
      "step": 91211,
      "training_step_time": 7.533903360366821
    },
    {
      "epoch": 1.6414642333984375e-05,
      "model_forward_time": 1.7611961364746094,
      "step": 91212
    },
    {
      "epoch": 1.6414642333984375e-05,
      "step": 91212,
      "training_step_time": 7.768752574920654
    },
    {
      "epoch": 1.6415596008300782e-05,
      "model_forward_time": 1.7704224586486816,
      "step": 91213
    },
    {
      "epoch": 1.6415596008300782e-05,
      "step": 91213,
      "training_step_time": 7.393056869506836
    },
    {
      "epoch": 1.6416549682617186e-05,
      "model_forward_time": 1.7650730609893799,
      "step": 91214
    },
    {
      "epoch": 1.6416549682617186e-05,
      "step": 91214,
      "training_step_time": 6.716511011123657
    },
    {
      "epoch": 1.6417503356933594e-05,
      "model_forward_time": 1.7704780101776123,
      "step": 91215
    },
    {
      "epoch": 1.6417503356933594e-05,
      "step": 91215,
      "training_step_time": 6.798978567123413
    },
    {
      "epoch": 1.641845703125e-05,
      "model_forward_time": 1.7661170959472656,
      "step": 91216
    },
    {
      "epoch": 1.641845703125e-05,
      "step": 91216,
      "training_step_time": 6.826929330825806
    },
    {
      "epoch": 1.6419410705566405e-05,
      "model_forward_time": 1.7689557075500488,
      "step": 91217
    },
    {
      "epoch": 1.6419410705566405e-05,
      "step": 91217,
      "training_step_time": 8.365149736404419
    },
    {
      "epoch": 1.6420364379882812e-05,
      "model_forward_time": 2.874427556991577,
      "step": 91218
    },
    {
      "epoch": 1.6420364379882812e-05,
      "step": 91218,
      "training_step_time": 8.384875774383545
    },
    {
      "epoch": 1.642131805419922e-05,
      "model_forward_time": 1.838348627090454,
      "step": 91219
    },
    {
      "epoch": 1.642131805419922e-05,
      "step": 91219,
      "training_step_time": 7.668884038925171
    },
    {
      "epoch": 1.6422271728515624e-05,
      "grad_norm": 0.0829731673002243,
      "learning_rate": 2.093276695527191e-07,
      "loss": 0.0547,
      "step": 91220
    },
    {
      "epoch": 1.6422271728515624e-05,
      "model_forward_time": 1.7714288234710693,
      "step": 91220
    },
    {
      "dynamics_loss_avg": 0.05040847007185221,
      "epoch": 1.6422271728515624e-05,
      "step": 91220
    },
    {
      "action_loss_avg": 0.0016793911388958804,
      "epoch": 1.6422271728515624e-05,
      "step": 91220
    },
    {
      "epoch": 1.6422271728515624e-05,
      "step": 91220,
      "training_step_time": 7.3379225730896
    },
    {
      "epoch": 1.642322540283203e-05,
      "model_forward_time": 1.7694928646087646,
      "step": 91221
    },
    {
      "epoch": 1.642322540283203e-05,
      "step": 91221,
      "training_step_time": 8.56617283821106
    },
    {
      "epoch": 1.642417907714844e-05,
      "model_forward_time": 1.7628490924835205,
      "step": 91222
    },
    {
      "epoch": 1.642417907714844e-05,
      "step": 91222,
      "training_step_time": 7.63905143737793
    },
    {
      "epoch": 1.6425132751464843e-05,
      "model_forward_time": 1.7894611358642578,
      "step": 91223
    },
    {
      "epoch": 1.6425132751464843e-05,
      "step": 91223,
      "training_step_time": 6.800043344497681
    },
    {
      "epoch": 1.642608642578125e-05,
      "model_forward_time": 1.4015727043151855,
      "step": 91224
    },
    {
      "epoch": 1.642608642578125e-05,
      "step": 91224,
      "training_step_time": 6.791326999664307
    },
    {
      "epoch": 1.6427040100097658e-05,
      "model_forward_time": 1.769784688949585,
      "step": 91225
    },
    {
      "epoch": 1.6427040100097658e-05,
      "step": 91225,
      "training_step_time": 6.806257486343384
    },
    {
      "epoch": 1.642799377441406e-05,
      "model_forward_time": 1.765193223953247,
      "step": 91226
    },
    {
      "epoch": 1.642799377441406e-05,
      "step": 91226,
      "training_step_time": 8.395613670349121
    },
    {
      "epoch": 1.642894744873047e-05,
      "model_forward_time": 2.842193603515625,
      "step": 91227
    },
    {
      "epoch": 1.642894744873047e-05,
      "step": 91227,
      "training_step_time": 9.84451675415039
    },
    {
      "epoch": 1.6429901123046876e-05,
      "model_forward_time": 1.8249876499176025,
      "step": 91228
    },
    {
      "epoch": 1.6429901123046876e-05,
      "step": 91228,
      "training_step_time": 7.774608373641968
    },
    {
      "epoch": 1.643085479736328e-05,
      "model_forward_time": 1.7669825553894043,
      "step": 91229
    },
    {
      "epoch": 1.643085479736328e-05,
      "step": 91229,
      "training_step_time": 7.404206275939941
    },
    {
      "epoch": 1.6431808471679688e-05,
      "grad_norm": 0.07880080491304398,
      "learning_rate": 2.0885451233628394e-07,
      "loss": 0.057,
      "step": 91230
    },
    {
      "epoch": 1.6431808471679688e-05,
      "model_forward_time": 1.773559331893921,
      "step": 91230
    },
    {
      "dynamics_loss_avg": 0.06606221124529839,
      "epoch": 1.6431808471679688e-05,
      "step": 91230
    },
    {
      "action_loss_avg": 0.0026230724179185927,
      "epoch": 1.6431808471679688e-05,
      "step": 91230
    },
    {
      "epoch": 1.6431808471679688e-05,
      "step": 91230,
      "training_step_time": 7.633531808853149
    },
    {
      "epoch": 1.6432762145996095e-05,
      "model_forward_time": 1.7709779739379883,
      "step": 91231
    },
    {
      "epoch": 1.6432762145996095e-05,
      "step": 91231,
      "training_step_time": 7.7737486362457275
    },
    {
      "epoch": 1.64337158203125e-05,
      "model_forward_time": 1.776566982269287,
      "step": 91232
    },
    {
      "epoch": 1.64337158203125e-05,
      "step": 91232,
      "training_step_time": 6.82706618309021
    },
    {
      "epoch": 1.6434669494628907e-05,
      "model_forward_time": 1.7716917991638184,
      "step": 91233
    },
    {
      "epoch": 1.6434669494628907e-05,
      "step": 91233,
      "training_step_time": 6.79224967956543
    },
    {
      "epoch": 1.6435623168945314e-05,
      "model_forward_time": 1.7636191844940186,
      "step": 91234
    },
    {
      "epoch": 1.6435623168945314e-05,
      "step": 91234,
      "training_step_time": 6.804492473602295
    },
    {
      "epoch": 1.6436576843261718e-05,
      "model_forward_time": 1.7619669437408447,
      "step": 91235
    },
    {
      "epoch": 1.6436576843261718e-05,
      "step": 91235,
      "training_step_time": 7.660297870635986
    },
    {
      "epoch": 1.6437530517578126e-05,
      "model_forward_time": 1.7927169799804688,
      "step": 91236
    },
    {
      "epoch": 1.6437530517578126e-05,
      "step": 91236,
      "training_step_time": 10.752567052841187
    },
    {
      "epoch": 1.643848419189453e-05,
      "model_forward_time": 1.7742509841918945,
      "step": 91237
    },
    {
      "epoch": 1.643848419189453e-05,
      "step": 91237,
      "training_step_time": 7.6760454177856445
    },
    {
      "epoch": 1.6439437866210937e-05,
      "model_forward_time": 1.8470938205718994,
      "step": 91238
    },
    {
      "epoch": 1.6439437866210937e-05,
      "step": 91238,
      "training_step_time": 7.317711353302002
    },
    {
      "epoch": 1.6440391540527344e-05,
      "model_forward_time": 1.783898115158081,
      "step": 91239
    },
    {
      "epoch": 1.6440391540527344e-05,
      "step": 91239,
      "training_step_time": 7.561586618423462
    },
    {
      "epoch": 1.644134521484375e-05,
      "grad_norm": 0.08280853927135468,
      "learning_rate": 2.0838187907231634e-07,
      "loss": 0.0547,
      "step": 91240
    },
    {
      "epoch": 1.644134521484375e-05,
      "model_forward_time": 1.7672560214996338,
      "step": 91240
    },
    {
      "dynamics_loss_avg": 0.061936442740261555,
      "epoch": 1.644134521484375e-05,
      "step": 91240
    },
    {
      "action_loss_avg": 0.0019771552877500655,
      "epoch": 1.644134521484375e-05,
      "step": 91240
    },
    {
      "epoch": 1.644134521484375e-05,
      "step": 91240,
      "training_step_time": 7.756091833114624
    },
    {
      "epoch": 1.6442298889160156e-05,
      "model_forward_time": 1.776764154434204,
      "step": 91241
    },
    {
      "epoch": 1.6442298889160156e-05,
      "step": 91241,
      "training_step_time": 7.353335618972778
    },
    {
      "epoch": 1.6443252563476563e-05,
      "model_forward_time": 1.769721269607544,
      "step": 91242
    },
    {
      "epoch": 1.6443252563476563e-05,
      "step": 91242,
      "training_step_time": 6.744623422622681
    },
    {
      "epoch": 1.6444206237792967e-05,
      "model_forward_time": 1.774764060974121,
      "step": 91243
    },
    {
      "epoch": 1.6444206237792967e-05,
      "step": 91243,
      "training_step_time": 6.739749193191528
    },
    {
      "epoch": 1.6445159912109375e-05,
      "model_forward_time": 1.7787349224090576,
      "step": 91244
    },
    {
      "epoch": 1.6445159912109375e-05,
      "step": 91244,
      "training_step_time": 7.739701271057129
    },
    {
      "epoch": 1.6446113586425782e-05,
      "model_forward_time": 1.7811081409454346,
      "step": 91245
    },
    {
      "epoch": 1.6446113586425782e-05,
      "step": 91245,
      "training_step_time": 9.557454347610474
    },
    {
      "epoch": 1.6447067260742186e-05,
      "model_forward_time": 2.9120829105377197,
      "step": 91246
    },
    {
      "epoch": 1.6447067260742186e-05,
      "step": 91246,
      "training_step_time": 7.696761131286621
    },
    {
      "epoch": 1.6448020935058594e-05,
      "model_forward_time": 1.8354263305664062,
      "step": 91247
    },
    {
      "epoch": 1.6448020935058594e-05,
      "step": 91247,
      "training_step_time": 7.622093439102173
    },
    {
      "epoch": 1.6448974609375e-05,
      "model_forward_time": 1.776843547821045,
      "step": 91248
    },
    {
      "epoch": 1.6448974609375e-05,
      "step": 91248,
      "training_step_time": 8.437456369400024
    },
    {
      "epoch": 1.6449928283691405e-05,
      "model_forward_time": 1.7724287509918213,
      "step": 91249
    },
    {
      "epoch": 1.6449928283691405e-05,
      "step": 91249,
      "training_step_time": 8.208608627319336
    },
    {
      "epoch": 1.6450881958007812e-05,
      "grad_norm": 0.09010818600654602,
      "learning_rate": 2.0790976981250155e-07,
      "loss": 0.0536,
      "step": 91250
    },
    {
      "epoch": 1.6450881958007812e-05,
      "model_forward_time": 1.7703759670257568,
      "step": 91250
    },
    {
      "dynamics_loss_avg": 0.03144221352413297,
      "epoch": 1.6450881958007812e-05,
      "step": 91250
    },
    {
      "action_loss_avg": 0.0010554990323726088,
      "epoch": 1.6450881958007812e-05,
      "step": 91250
    },
    {
      "epoch": 1.6450881958007812e-05,
      "step": 91250,
      "training_step_time": 7.386387348175049
    },
    {
      "epoch": 1.645183563232422e-05,
      "model_forward_time": 1.7763681411743164,
      "step": 91251
    },
    {
      "epoch": 1.645183563232422e-05,
      "step": 91251,
      "training_step_time": 6.781686067581177
    },
    {
      "epoch": 1.6452789306640624e-05,
      "model_forward_time": 1.7840063571929932,
      "step": 91252
    },
    {
      "epoch": 1.6452789306640624e-05,
      "step": 91252,
      "training_step_time": 6.769628286361694
    },
    {
      "epoch": 1.645374298095703e-05,
      "model_forward_time": 1.7755300998687744,
      "step": 91253
    },
    {
      "epoch": 1.645374298095703e-05,
      "step": 91253,
      "training_step_time": 8.443087339401245
    },
    {
      "epoch": 1.645469665527344e-05,
      "model_forward_time": 1.7790629863739014,
      "step": 91254
    },
    {
      "epoch": 1.645469665527344e-05,
      "step": 91254,
      "training_step_time": 8.483474731445312
    },
    {
      "epoch": 1.6455650329589843e-05,
      "model_forward_time": 2.9291281700134277,
      "step": 91255
    },
    {
      "epoch": 1.6455650329589843e-05,
      "step": 91255,
      "training_step_time": 7.670670032501221
    },
    {
      "epoch": 1.645660400390625e-05,
      "model_forward_time": 1.845688819885254,
      "step": 91256
    },
    {
      "epoch": 1.645660400390625e-05,
      "step": 91256,
      "training_step_time": 8.565507888793945
    },
    {
      "epoch": 1.6457557678222658e-05,
      "model_forward_time": 1.7713031768798828,
      "step": 91257
    },
    {
      "epoch": 1.6457557678222658e-05,
      "step": 91257,
      "training_step_time": 7.538121223449707
    },
    {
      "epoch": 1.645851135253906e-05,
      "model_forward_time": 1.773479700088501,
      "step": 91258
    },
    {
      "epoch": 1.645851135253906e-05,
      "step": 91258,
      "training_step_time": 7.581226110458374
    },
    {
      "epoch": 1.645946502685547e-05,
      "model_forward_time": 1.7742936611175537,
      "step": 91259
    },
    {
      "epoch": 1.645946502685547e-05,
      "step": 91259,
      "training_step_time": 7.714528322219849
    },
    {
      "epoch": 1.6460418701171876e-05,
      "grad_norm": 0.0834713727235794,
      "learning_rate": 2.0743818460847e-07,
      "loss": 0.0548,
      "step": 91260
    },
    {
      "epoch": 1.6460418701171876e-05,
      "model_forward_time": 1.7760980129241943,
      "step": 91260
    },
    {
      "dynamics_loss_avg": 0.0677269421517849,
      "epoch": 1.6460418701171876e-05,
      "step": 91260
    },
    {
      "action_loss_avg": 0.0018333891639485955,
      "epoch": 1.6460418701171876e-05,
      "step": 91260
    },
    {
      "epoch": 1.6460418701171876e-05,
      "step": 91260,
      "training_step_time": 6.7866668701171875
    },
    {
      "epoch": 1.646137237548828e-05,
      "model_forward_time": 1.7697069644927979,
      "step": 91261
    },
    {
      "epoch": 1.646137237548828e-05,
      "step": 91261,
      "training_step_time": 6.731385231018066
    },
    {
      "epoch": 1.6462326049804688e-05,
      "model_forward_time": 1.769101858139038,
      "step": 91262
    },
    {
      "epoch": 1.6462326049804688e-05,
      "step": 91262,
      "training_step_time": 8.655843496322632
    },
    {
      "epoch": 1.6463279724121095e-05,
      "model_forward_time": 1.7781267166137695,
      "step": 91263
    },
    {
      "epoch": 1.6463279724121095e-05,
      "step": 91263,
      "training_step_time": 9.931998252868652
    },
    {
      "epoch": 1.64642333984375e-05,
      "model_forward_time": 2.906388282775879,
      "step": 91264
    },
    {
      "epoch": 1.64642333984375e-05,
      "step": 91264,
      "training_step_time": 9.21310305595398
    },
    {
      "epoch": 1.6465187072753907e-05,
      "model_forward_time": 1.8419864177703857,
      "step": 91265
    },
    {
      "epoch": 1.6465187072753907e-05,
      "step": 91265,
      "training_step_time": 9.250215768814087
    },
    {
      "epoch": 1.646614074707031e-05,
      "model_forward_time": 1.7774131298065186,
      "step": 91266
    },
    {
      "epoch": 1.646614074707031e-05,
      "step": 91266,
      "training_step_time": 7.336359024047852
    },
    {
      "epoch": 1.6467094421386718e-05,
      "model_forward_time": 1.7756447792053223,
      "step": 91267
    },
    {
      "epoch": 1.6467094421386718e-05,
      "step": 91267,
      "training_step_time": 7.674235582351685
    },
    {
      "epoch": 1.6468048095703126e-05,
      "model_forward_time": 1.7722210884094238,
      "step": 91268
    },
    {
      "epoch": 1.6468048095703126e-05,
      "step": 91268,
      "training_step_time": 7.7066051959991455
    },
    {
      "epoch": 1.646900177001953e-05,
      "model_forward_time": 1.7669422626495361,
      "step": 91269
    },
    {
      "epoch": 1.646900177001953e-05,
      "step": 91269,
      "training_step_time": 6.779912948608398
    },
    {
      "epoch": 1.6469955444335937e-05,
      "grad_norm": 0.0926256775856018,
      "learning_rate": 2.069671235117926e-07,
      "loss": 0.054,
      "step": 91270
    },
    {
      "epoch": 1.6469955444335937e-05,
      "model_forward_time": 1.7791671752929688,
      "step": 91270
    },
    {
      "dynamics_loss_avg": 0.056096904538571836,
      "epoch": 1.6469955444335937e-05,
      "step": 91270
    },
    {
      "action_loss_avg": 0.0019015124824363738,
      "epoch": 1.6469955444335937e-05,
      "step": 91270
    },
    {
      "epoch": 1.6469955444335937e-05,
      "step": 91270,
      "training_step_time": 6.783635854721069
    },
    {
      "epoch": 1.6470909118652345e-05,
      "model_forward_time": 1.7729415893554688,
      "step": 91271
    },
    {
      "epoch": 1.6470909118652345e-05,
      "step": 91271,
      "training_step_time": 8.362922668457031
    },
    {
      "epoch": 1.647186279296875e-05,
      "model_forward_time": 1.7698607444763184,
      "step": 91272
    },
    {
      "epoch": 1.647186279296875e-05,
      "step": 91272,
      "training_step_time": 8.319731950759888
    },
    {
      "epoch": 1.6472816467285156e-05,
      "model_forward_time": 2.882232666015625,
      "step": 91273
    },
    {
      "epoch": 1.6472816467285156e-05,
      "step": 91273,
      "training_step_time": 9.198782205581665
    },
    {
      "epoch": 1.6473770141601563e-05,
      "model_forward_time": 1.806469440460205,
      "step": 91274
    },
    {
      "epoch": 1.6473770141601563e-05,
      "step": 91274,
      "training_step_time": 7.7074806690216064
    },
    {
      "epoch": 1.6474723815917967e-05,
      "model_forward_time": 1.7753186225891113,
      "step": 91275
    },
    {
      "epoch": 1.6474723815917967e-05,
      "step": 91275,
      "training_step_time": 7.638810634613037
    },
    {
      "epoch": 1.6475677490234375e-05,
      "model_forward_time": 1.7716619968414307,
      "step": 91276
    },
    {
      "epoch": 1.6475677490234375e-05,
      "step": 91276,
      "training_step_time": 7.601040363311768
    },
    {
      "epoch": 1.6476631164550782e-05,
      "model_forward_time": 1.7789487838745117,
      "step": 91277
    },
    {
      "epoch": 1.6476631164550782e-05,
      "step": 91277,
      "training_step_time": 8.60783076286316
    },
    {
      "epoch": 1.6477584838867186e-05,
      "model_forward_time": 1.7819671630859375,
      "step": 91278
    },
    {
      "epoch": 1.6477584838867186e-05,
      "step": 91278,
      "training_step_time": 6.7649946212768555
    },
    {
      "epoch": 1.6478538513183594e-05,
      "model_forward_time": 1.7554683685302734,
      "step": 91279
    },
    {
      "epoch": 1.6478538513183594e-05,
      "step": 91279,
      "training_step_time": 6.778827905654907
    },
    {
      "epoch": 1.64794921875e-05,
      "grad_norm": 0.11656374484300613,
      "learning_rate": 2.0649658657398487e-07,
      "loss": 0.0539,
      "step": 91280
    },
    {
      "epoch": 1.64794921875e-05,
      "model_forward_time": 1.7559857368469238,
      "step": 91280
    },
    {
      "dynamics_loss_avg": 0.04760850295424461,
      "epoch": 1.64794921875e-05,
      "step": 91280
    },
    {
      "action_loss_avg": 0.0022206772584468125,
      "epoch": 1.64794921875e-05,
      "step": 91280
    },
    {
      "epoch": 1.64794921875e-05,
      "step": 91280,
      "training_step_time": 8.148202419281006
    },
    {
      "epoch": 1.6480445861816405e-05,
      "model_forward_time": 1.7599716186523438,
      "step": 91281
    },
    {
      "epoch": 1.6480445861816405e-05,
      "step": 91281,
      "training_step_time": 8.53079891204834
    },
    {
      "epoch": 1.6481399536132813e-05,
      "model_forward_time": 2.839877128601074,
      "step": 91282
    },
    {
      "epoch": 1.6481399536132813e-05,
      "step": 91282,
      "training_step_time": 10.221745729446411
    },
    {
      "epoch": 1.648235321044922e-05,
      "model_forward_time": 1.7824292182922363,
      "step": 91283
    },
    {
      "epoch": 1.648235321044922e-05,
      "step": 91283,
      "training_step_time": 8.20762014389038
    },
    {
      "epoch": 1.6483306884765624e-05,
      "model_forward_time": 1.7467126846313477,
      "step": 91284
    },
    {
      "epoch": 1.6483306884765624e-05,
      "step": 91284,
      "training_step_time": 7.594149112701416
    },
    {
      "epoch": 1.648426055908203e-05,
      "model_forward_time": 1.7503628730773926,
      "step": 91285
    },
    {
      "epoch": 1.648426055908203e-05,
      "step": 91285,
      "training_step_time": 7.616584777832031
    },
    {
      "epoch": 1.648521423339844e-05,
      "model_forward_time": 1.7464714050292969,
      "step": 91286
    },
    {
      "epoch": 1.648521423339844e-05,
      "step": 91286,
      "training_step_time": 7.809788465499878
    },
    {
      "epoch": 1.6486167907714843e-05,
      "model_forward_time": 1.7503981590270996,
      "step": 91287
    },
    {
      "epoch": 1.6486167907714843e-05,
      "step": 91287,
      "training_step_time": 7.290931224822998
    },
    {
      "epoch": 1.648712158203125e-05,
      "model_forward_time": 1.7393734455108643,
      "step": 91288
    },
    {
      "epoch": 1.648712158203125e-05,
      "step": 91288,
      "training_step_time": 6.808862209320068
    },
    {
      "epoch": 1.6488075256347658e-05,
      "model_forward_time": 1.7459838390350342,
      "step": 91289
    },
    {
      "epoch": 1.6488075256347658e-05,
      "step": 91289,
      "training_step_time": 7.722297430038452
    },
    {
      "epoch": 1.648902893066406e-05,
      "grad_norm": 0.08413030207157135,
      "learning_rate": 2.0602657384650338e-07,
      "loss": 0.0544,
      "step": 91290
    },
    {
      "epoch": 1.648902893066406e-05,
      "model_forward_time": 1.7422466278076172,
      "step": 91290
    },
    {
      "dynamics_loss_avg": 0.05467502772808075,
      "epoch": 1.648902893066406e-05,
      "step": 91290
    },
    {
      "action_loss_avg": 0.0029730298963841053,
      "epoch": 1.648902893066406e-05,
      "step": 91290
    },
    {
      "epoch": 1.648902893066406e-05,
      "step": 91290,
      "training_step_time": 7.666104555130005
    },
    {
      "epoch": 1.648998260498047e-05,
      "model_forward_time": 2.826779842376709,
      "step": 91291
    },
    {
      "epoch": 1.648998260498047e-05,
      "step": 91291,
      "training_step_time": 9.290449380874634
    },
    {
      "epoch": 1.6490936279296877e-05,
      "model_forward_time": 1.785257339477539,
      "step": 91292
    },
    {
      "epoch": 1.6490936279296877e-05,
      "step": 91292,
      "training_step_time": 7.609342575073242
    },
    {
      "epoch": 1.649188995361328e-05,
      "model_forward_time": 1.7785375118255615,
      "step": 91293
    },
    {
      "epoch": 1.649188995361328e-05,
      "step": 91293,
      "training_step_time": 7.822526454925537
    },
    {
      "epoch": 1.6492843627929688e-05,
      "model_forward_time": 1.7675838470458984,
      "step": 91294
    },
    {
      "epoch": 1.6492843627929688e-05,
      "step": 91294,
      "training_step_time": 7.560495615005493
    },
    {
      "epoch": 1.6493797302246095e-05,
      "model_forward_time": 1.7372910976409912,
      "step": 91295
    },
    {
      "epoch": 1.6493797302246095e-05,
      "step": 91295,
      "training_step_time": 7.562984943389893
    },
    {
      "epoch": 1.64947509765625e-05,
      "model_forward_time": 1.7585642337799072,
      "step": 91296
    },
    {
      "epoch": 1.64947509765625e-05,
      "step": 91296,
      "training_step_time": 7.771919012069702
    },
    {
      "epoch": 1.6495704650878907e-05,
      "model_forward_time": 1.757112741470337,
      "step": 91297
    },
    {
      "epoch": 1.6495704650878907e-05,
      "step": 91297,
      "training_step_time": 6.821697950363159
    },
    {
      "epoch": 1.649665832519531e-05,
      "model_forward_time": 1.7593741416931152,
      "step": 91298
    },
    {
      "epoch": 1.649665832519531e-05,
      "step": 91298,
      "training_step_time": 7.721916913986206
    },
    {
      "epoch": 1.6497611999511718e-05,
      "model_forward_time": 1.755547046661377,
      "step": 91299
    },
    {
      "epoch": 1.6497611999511718e-05,
      "step": 91299,
      "training_step_time": 7.635846138000488
    },
    {
      "epoch": 1.6498565673828126e-05,
      "grad_norm": 0.09213073551654816,
      "learning_rate": 2.055570853807487e-07,
      "loss": 0.0529,
      "step": 91300
    },
    {
      "epoch": 1.6498565673828126e-05,
      "model_forward_time": 1.7539262771606445,
      "step": 91300
    },
    {
      "dynamics_loss_avg": 0.054529355466365816,
      "epoch": 1.6498565673828126e-05,
      "step": 91300
    },
    {
      "action_loss_avg": 0.0014707176640513353,
      "epoch": 1.6498565673828126e-05,
      "step": 91300
    },
    {
      "epoch": 1.6498565673828126e-05,
      "step": 91300,
      "training_step_time": 9.18003249168396
    },
    {
      "epoch": 1.649951934814453e-05,
      "model_forward_time": 2.917985677719116,
      "step": 91301
    },
    {
      "epoch": 1.649951934814453e-05,
      "step": 91301,
      "training_step_time": 9.400605916976929
    },
    {
      "epoch": 1.6500473022460937e-05,
      "model_forward_time": 1.7741584777832031,
      "step": 91302
    },
    {
      "epoch": 1.6500473022460937e-05,
      "step": 91302,
      "training_step_time": 7.822139739990234
    },
    {
      "epoch": 1.6501426696777345e-05,
      "model_forward_time": 1.762629747390747,
      "step": 91303
    },
    {
      "epoch": 1.6501426696777345e-05,
      "step": 91303,
      "training_step_time": 7.598229169845581
    },
    {
      "epoch": 1.650238037109375e-05,
      "model_forward_time": 1.7612059116363525,
      "step": 91304
    },
    {
      "epoch": 1.650238037109375e-05,
      "step": 91304,
      "training_step_time": 7.981793642044067
    },
    {
      "epoch": 1.6503334045410156e-05,
      "model_forward_time": 1.761780023574829,
      "step": 91305
    },
    {
      "epoch": 1.6503334045410156e-05,
      "step": 91305,
      "training_step_time": 7.791826248168945
    },
    {
      "epoch": 1.6504287719726563e-05,
      "model_forward_time": 1.7625513076782227,
      "step": 91306
    },
    {
      "epoch": 1.6504287719726563e-05,
      "step": 91306,
      "training_step_time": 6.806368827819824
    },
    {
      "epoch": 1.6505241394042967e-05,
      "model_forward_time": 1.7620832920074463,
      "step": 91307
    },
    {
      "epoch": 1.6505241394042967e-05,
      "step": 91307,
      "training_step_time": 7.6478588581085205
    },
    {
      "epoch": 1.6506195068359375e-05,
      "model_forward_time": 1.7639570236206055,
      "step": 91308
    },
    {
      "epoch": 1.6506195068359375e-05,
      "step": 91308,
      "training_step_time": 7.609163284301758
    },
    {
      "epoch": 1.6507148742675782e-05,
      "model_forward_time": 1.7595491409301758,
      "step": 91309
    },
    {
      "epoch": 1.6507148742675782e-05,
      "step": 91309,
      "training_step_time": 8.374151945114136
    },
    {
      "epoch": 1.6508102416992186e-05,
      "grad_norm": 0.08900661766529083,
      "learning_rate": 2.0508812122806199e-07,
      "loss": 0.0542,
      "step": 91310
    },
    {
      "epoch": 1.6508102416992186e-05,
      "model_forward_time": 2.9107706546783447,
      "step": 91310
    },
    {
      "dynamics_loss_avg": 0.05863788537681103,
      "epoch": 1.6508102416992186e-05,
      "step": 91310
    },
    {
      "action_loss_avg": 0.003373046999331564,
      "epoch": 1.6508102416992186e-05,
      "step": 91310
    },
    {
      "epoch": 1.6508102416992186e-05,
      "step": 91310,
      "training_step_time": 8.349476337432861
    },
    {
      "epoch": 1.6509056091308594e-05,
      "model_forward_time": 1.7692937850952148,
      "step": 91311
    },
    {
      "epoch": 1.6509056091308594e-05,
      "step": 91311,
      "training_step_time": 9.604912281036377
    },
    {
      "epoch": 1.6510009765625e-05,
      "model_forward_time": 1.762389898300171,
      "step": 91312
    },
    {
      "epoch": 1.6510009765625e-05,
      "step": 91312,
      "training_step_time": 7.697813510894775
    },
    {
      "epoch": 1.6510963439941405e-05,
      "model_forward_time": 1.7707633972167969,
      "step": 91313
    },
    {
      "epoch": 1.6510963439941405e-05,
      "step": 91313,
      "training_step_time": 7.745053768157959
    },
    {
      "epoch": 1.6511917114257813e-05,
      "model_forward_time": 1.763559103012085,
      "step": 91314
    },
    {
      "epoch": 1.6511917114257813e-05,
      "step": 91314,
      "training_step_time": 7.791805028915405
    },
    {
      "epoch": 1.651287078857422e-05,
      "model_forward_time": 1.7633569240570068,
      "step": 91315
    },
    {
      "epoch": 1.651287078857422e-05,
      "step": 91315,
      "training_step_time": 6.854519605636597
    },
    {
      "epoch": 1.6513824462890624e-05,
      "model_forward_time": 1.7628414630889893,
      "step": 91316
    },
    {
      "epoch": 1.6513824462890624e-05,
      "step": 91316,
      "training_step_time": 7.675805330276489
    },
    {
      "epoch": 1.651477813720703e-05,
      "model_forward_time": 1.7650234699249268,
      "step": 91317
    },
    {
      "epoch": 1.651477813720703e-05,
      "step": 91317,
      "training_step_time": 8.376376867294312
    },
    {
      "epoch": 1.651573181152344e-05,
      "model_forward_time": 1.7606303691864014,
      "step": 91318
    },
    {
      "epoch": 1.651573181152344e-05,
      "step": 91318,
      "training_step_time": 9.073265075683594
    },
    {
      "epoch": 1.6516685485839843e-05,
      "model_forward_time": 2.932638645172119,
      "step": 91319
    },
    {
      "epoch": 1.6516685485839843e-05,
      "step": 91319,
      "training_step_time": 8.38506031036377
    },
    {
      "epoch": 1.651763916015625e-05,
      "grad_norm": 0.09823480993509293,
      "learning_rate": 2.0461968143972942e-07,
      "loss": 0.0527,
      "step": 91320
    },
    {
      "epoch": 1.651763916015625e-05,
      "model_forward_time": 1.7685487270355225,
      "step": 91320
    },
    {
      "dynamics_loss_avg": 0.05114958137273788,
      "epoch": 1.651763916015625e-05,
      "step": 91320
    },
    {
      "action_loss_avg": 0.001536346651846543,
      "epoch": 1.651763916015625e-05,
      "step": 91320
    },
    {
      "epoch": 1.651763916015625e-05,
      "step": 91320,
      "training_step_time": 7.712897300720215
    },
    {
      "epoch": 1.6518592834472658e-05,
      "model_forward_time": 1.7669427394866943,
      "step": 91321
    },
    {
      "epoch": 1.6518592834472658e-05,
      "step": 91321,
      "training_step_time": 7.727115154266357
    },
    {
      "epoch": 1.6519546508789062e-05,
      "model_forward_time": 1.7681999206542969,
      "step": 91322
    },
    {
      "epoch": 1.6519546508789062e-05,
      "step": 91322,
      "training_step_time": 7.636423349380493
    },
    {
      "epoch": 1.652050018310547e-05,
      "model_forward_time": 1.7643029689788818,
      "step": 91323
    },
    {
      "epoch": 1.652050018310547e-05,
      "step": 91323,
      "training_step_time": 9.64865756034851
    },
    {
      "epoch": 1.6521453857421877e-05,
      "model_forward_time": 1.7589256763458252,
      "step": 91324
    },
    {
      "epoch": 1.6521453857421877e-05,
      "step": 91324,
      "training_step_time": 6.764280796051025
    },
    {
      "epoch": 1.652240753173828e-05,
      "model_forward_time": 1.7671496868133545,
      "step": 91325
    },
    {
      "epoch": 1.652240753173828e-05,
      "step": 91325,
      "training_step_time": 7.634598016738892
    },
    {
      "epoch": 1.6523361206054688e-05,
      "model_forward_time": 1.7651276588439941,
      "step": 91326
    },
    {
      "epoch": 1.6523361206054688e-05,
      "step": 91326,
      "training_step_time": 7.565065622329712
    },
    {
      "epoch": 1.6524314880371095e-05,
      "model_forward_time": 1.7645695209503174,
      "step": 91327
    },
    {
      "epoch": 1.6524314880371095e-05,
      "step": 91327,
      "training_step_time": 8.464646339416504
    },
    {
      "epoch": 1.65252685546875e-05,
      "model_forward_time": 2.8679049015045166,
      "step": 91328
    },
    {
      "epoch": 1.65252685546875e-05,
      "step": 91328,
      "training_step_time": 8.638394594192505
    },
    {
      "epoch": 1.6526222229003907e-05,
      "model_forward_time": 1.785231351852417,
      "step": 91329
    },
    {
      "epoch": 1.6526222229003907e-05,
      "step": 91329,
      "training_step_time": 7.769598007202148
    },
    {
      "epoch": 1.652717590332031e-05,
      "grad_norm": 0.08708450198173523,
      "learning_rate": 2.0415176606697896e-07,
      "loss": 0.0523,
      "step": 91330
    },
    {
      "epoch": 1.652717590332031e-05,
      "model_forward_time": 1.783385992050171,
      "step": 91330
    },
    {
      "dynamics_loss_avg": 0.05471040941774845,
      "epoch": 1.652717590332031e-05,
      "step": 91330
    },
    {
      "action_loss_avg": 0.0018798402918037028,
      "epoch": 1.652717590332031e-05,
      "step": 91330
    },
    {
      "epoch": 1.652717590332031e-05,
      "step": 91330,
      "training_step_time": 7.785686016082764
    },
    {
      "epoch": 1.6528129577636718e-05,
      "model_forward_time": 1.7687058448791504,
      "step": 91331
    },
    {
      "epoch": 1.6528129577636718e-05,
      "step": 91331,
      "training_step_time": 8.083984851837158
    },
    {
      "epoch": 1.6529083251953126e-05,
      "model_forward_time": 1.7586853504180908,
      "step": 91332
    },
    {
      "epoch": 1.6529083251953126e-05,
      "step": 91332,
      "training_step_time": 8.312583684921265
    },
    {
      "epoch": 1.653003692626953e-05,
      "model_forward_time": 1.7580413818359375,
      "step": 91333
    },
    {
      "epoch": 1.653003692626953e-05,
      "step": 91333,
      "training_step_time": 7.4046430587768555
    },
    {
      "epoch": 1.6530990600585937e-05,
      "model_forward_time": 1.7644338607788086,
      "step": 91334
    },
    {
      "epoch": 1.6530990600585937e-05,
      "step": 91334,
      "training_step_time": 7.310941934585571
    },
    {
      "epoch": 1.6531944274902345e-05,
      "model_forward_time": 1.7685747146606445,
      "step": 91335
    },
    {
      "epoch": 1.6531944274902345e-05,
      "step": 91335,
      "training_step_time": 7.7927539348602295
    },
    {
      "epoch": 1.653289794921875e-05,
      "model_forward_time": 1.760702133178711,
      "step": 91336
    },
    {
      "epoch": 1.653289794921875e-05,
      "step": 91336,
      "training_step_time": 7.701205015182495
    },
    {
      "epoch": 1.6533851623535156e-05,
      "model_forward_time": 2.83044695854187,
      "step": 91337
    },
    {
      "epoch": 1.6533851623535156e-05,
      "step": 91337,
      "training_step_time": 9.460816144943237
    },
    {
      "epoch": 1.6534805297851563e-05,
      "model_forward_time": 1.7860970497131348,
      "step": 91338
    },
    {
      "epoch": 1.6534805297851563e-05,
      "step": 91338,
      "training_step_time": 8.58651614189148
    },
    {
      "epoch": 1.6535758972167967e-05,
      "model_forward_time": 1.7748439311981201,
      "step": 91339
    },
    {
      "epoch": 1.6535758972167967e-05,
      "step": 91339,
      "training_step_time": 8.631917238235474
    },
    {
      "epoch": 1.6536712646484375e-05,
      "grad_norm": 0.10793551802635193,
      "learning_rate": 2.0368437516098128e-07,
      "loss": 0.0531,
      "step": 91340
    },
    {
      "epoch": 1.6536712646484375e-05,
      "model_forward_time": 1.7598564624786377,
      "step": 91340
    },
    {
      "dynamics_loss_avg": 0.054207855649292466,
      "epoch": 1.6536712646484375e-05,
      "step": 91340
    },
    {
      "action_loss_avg": 0.0019300680520245805,
      "epoch": 1.6536712646484375e-05,
      "step": 91340
    },
    {
      "epoch": 1.6536712646484375e-05,
      "step": 91340,
      "training_step_time": 7.554238796234131
    },
    {
      "epoch": 1.6537666320800782e-05,
      "model_forward_time": 1.767730951309204,
      "step": 91341
    },
    {
      "epoch": 1.6537666320800782e-05,
      "step": 91341,
      "training_step_time": 7.667686939239502
    },
    {
      "epoch": 1.6538619995117186e-05,
      "model_forward_time": 1.7635254859924316,
      "step": 91342
    },
    {
      "epoch": 1.6538619995117186e-05,
      "step": 91342,
      "training_step_time": 7.828489780426025
    },
    {
      "epoch": 1.6539573669433594e-05,
      "model_forward_time": 1.7576851844787598,
      "step": 91343
    },
    {
      "epoch": 1.6539573669433594e-05,
      "step": 91343,
      "training_step_time": 7.601041555404663
    },
    {
      "epoch": 1.654052734375e-05,
      "model_forward_time": 1.7632036209106445,
      "step": 91344
    },
    {
      "epoch": 1.654052734375e-05,
      "step": 91344,
      "training_step_time": 7.617279291152954
    },
    {
      "epoch": 1.6541481018066405e-05,
      "model_forward_time": 1.7654297351837158,
      "step": 91345
    },
    {
      "epoch": 1.6541481018066405e-05,
      "step": 91345,
      "training_step_time": 7.638367414474487
    },
    {
      "epoch": 1.6542434692382813e-05,
      "model_forward_time": 1.761523962020874,
      "step": 91346
    },
    {
      "epoch": 1.6542434692382813e-05,
      "step": 91346,
      "training_step_time": 9.645476341247559
    },
    {
      "epoch": 1.654338836669922e-05,
      "model_forward_time": 1.788351058959961,
      "step": 91347
    },
    {
      "epoch": 1.654338836669922e-05,
      "step": 91347,
      "training_step_time": 9.846108436584473
    },
    {
      "epoch": 1.6544342041015624e-05,
      "model_forward_time": 1.7648005485534668,
      "step": 91348
    },
    {
      "epoch": 1.6544342041015624e-05,
      "step": 91348,
      "training_step_time": 7.649763107299805
    },
    {
      "epoch": 1.654529571533203e-05,
      "model_forward_time": 1.780881643295288,
      "step": 91349
    },
    {
      "epoch": 1.654529571533203e-05,
      "step": 91349,
      "training_step_time": 8.516422033309937
    },
    {
      "epoch": 1.654624938964844e-05,
      "grad_norm": 0.11597482115030289,
      "learning_rate": 2.032175087728483e-07,
      "loss": 0.0539,
      "step": 91350
    },
    {
      "epoch": 1.654624938964844e-05,
      "model_forward_time": 1.7612543106079102,
      "step": 91350
    },
    {
      "dynamics_loss_avg": 0.043923153169453145,
      "epoch": 1.654624938964844e-05,
      "step": 91350
    },
    {
      "action_loss_avg": 0.0015983427845640108,
      "epoch": 1.654624938964844e-05,
      "step": 91350
    },
    {
      "epoch": 1.654624938964844e-05,
      "step": 91350,
      "training_step_time": 7.678690433502197
    },
    {
      "epoch": 1.6547203063964843e-05,
      "model_forward_time": 1.7683234214782715,
      "step": 91351
    },
    {
      "epoch": 1.6547203063964843e-05,
      "step": 91351,
      "training_step_time": 7.783101320266724
    },
    {
      "epoch": 1.654815673828125e-05,
      "model_forward_time": 1.7671430110931396,
      "step": 91352
    },
    {
      "epoch": 1.654815673828125e-05,
      "step": 91352,
      "training_step_time": 7.685711622238159
    },
    {
      "epoch": 1.6549110412597658e-05,
      "model_forward_time": 1.7642006874084473,
      "step": 91353
    },
    {
      "epoch": 1.6549110412597658e-05,
      "step": 91353,
      "training_step_time": 7.702751874923706
    },
    {
      "epoch": 1.6550064086914062e-05,
      "model_forward_time": 1.7721071243286133,
      "step": 91354
    },
    {
      "epoch": 1.6550064086914062e-05,
      "step": 91354,
      "training_step_time": 7.645532608032227
    },
    {
      "epoch": 1.655101776123047e-05,
      "model_forward_time": 1.7652692794799805,
      "step": 91355
    },
    {
      "epoch": 1.655101776123047e-05,
      "step": 91355,
      "training_step_time": 8.28381061553955
    },
    {
      "epoch": 1.6551971435546877e-05,
      "model_forward_time": 2.9492905139923096,
      "step": 91356
    },
    {
      "epoch": 1.6551971435546877e-05,
      "step": 91356,
      "training_step_time": 10.043450593948364
    },
    {
      "epoch": 1.655292510986328e-05,
      "model_forward_time": 1.7518188953399658,
      "step": 91357
    },
    {
      "epoch": 1.655292510986328e-05,
      "step": 91357,
      "training_step_time": 8.690290212631226
    },
    {
      "epoch": 1.6553878784179688e-05,
      "model_forward_time": 1.787588357925415,
      "step": 91358
    },
    {
      "epoch": 1.6553878784179688e-05,
      "step": 91358,
      "training_step_time": 7.839848756790161
    },
    {
      "epoch": 1.6554832458496095e-05,
      "model_forward_time": 1.765561580657959,
      "step": 91359
    },
    {
      "epoch": 1.6554832458496095e-05,
      "step": 91359,
      "training_step_time": 8.528440237045288
    },
    {
      "epoch": 1.65557861328125e-05,
      "grad_norm": 0.08996402472257614,
      "learning_rate": 2.027511669536375e-07,
      "loss": 0.0537,
      "step": 91360
    },
    {
      "epoch": 1.65557861328125e-05,
      "model_forward_time": 1.7608075141906738,
      "step": 91360
    },
    {
      "dynamics_loss_avg": 0.04779098061844707,
      "epoch": 1.65557861328125e-05,
      "step": 91360
    },
    {
      "action_loss_avg": 0.0014903377676091622,
      "epoch": 1.65557861328125e-05,
      "step": 91360
    },
    {
      "epoch": 1.65557861328125e-05,
      "step": 91360,
      "training_step_time": 7.896054267883301
    },
    {
      "epoch": 1.6556739807128907e-05,
      "model_forward_time": 1.7626078128814697,
      "step": 91361
    },
    {
      "epoch": 1.6556739807128907e-05,
      "step": 91361,
      "training_step_time": 7.644753694534302
    },
    {
      "epoch": 1.655769348144531e-05,
      "model_forward_time": 1.7609305381774902,
      "step": 91362
    },
    {
      "epoch": 1.655769348144531e-05,
      "step": 91362,
      "training_step_time": 7.678103446960449
    },
    {
      "epoch": 1.655864715576172e-05,
      "model_forward_time": 1.732734203338623,
      "step": 91363
    },
    {
      "epoch": 1.655864715576172e-05,
      "step": 91363,
      "training_step_time": 7.509793043136597
    },
    {
      "epoch": 1.6559600830078126e-05,
      "model_forward_time": 1.383223295211792,
      "step": 91364
    },
    {
      "epoch": 1.6559600830078126e-05,
      "step": 91364,
      "training_step_time": 10.544947624206543
    },
    {
      "epoch": 1.656055450439453e-05,
      "model_forward_time": 2.922455072402954,
      "step": 91365
    },
    {
      "epoch": 1.656055450439453e-05,
      "step": 91365,
      "training_step_time": 7.75520396232605
    },
    {
      "epoch": 1.6561508178710937e-05,
      "model_forward_time": 1.7490932941436768,
      "step": 91366
    },
    {
      "epoch": 1.6561508178710937e-05,
      "step": 91366,
      "training_step_time": 9.180763959884644
    },
    {
      "epoch": 1.6562461853027345e-05,
      "model_forward_time": 1.7846665382385254,
      "step": 91367
    },
    {
      "epoch": 1.6562461853027345e-05,
      "step": 91367,
      "training_step_time": 7.733329772949219
    },
    {
      "epoch": 1.656341552734375e-05,
      "model_forward_time": 1.7475287914276123,
      "step": 91368
    },
    {
      "epoch": 1.656341552734375e-05,
      "step": 91368,
      "training_step_time": 7.932241439819336
    },
    {
      "epoch": 1.6564369201660156e-05,
      "model_forward_time": 1.7550387382507324,
      "step": 91369
    },
    {
      "epoch": 1.6564369201660156e-05,
      "step": 91369,
      "training_step_time": 9.102341413497925
    },
    {
      "epoch": 1.6565322875976563e-05,
      "grad_norm": 0.09019052982330322,
      "learning_rate": 2.0228534975434643e-07,
      "loss": 0.0534,
      "step": 91370
    },
    {
      "epoch": 1.6565322875976563e-05,
      "model_forward_time": 1.7543601989746094,
      "step": 91370
    },
    {
      "dynamics_loss_avg": 0.06036136858165264,
      "epoch": 1.6565322875976563e-05,
      "step": 91370
    },
    {
      "action_loss_avg": 0.0017343508399790153,
      "epoch": 1.6565322875976563e-05,
      "step": 91370
    },
    {
      "epoch": 1.6565322875976563e-05,
      "step": 91370,
      "training_step_time": 7.2761828899383545
    },
    {
      "epoch": 1.6566276550292967e-05,
      "model_forward_time": 1.7508158683776855,
      "step": 91371
    },
    {
      "epoch": 1.6566276550292967e-05,
      "step": 91371,
      "training_step_time": 7.6151673793792725
    },
    {
      "epoch": 1.6567230224609375e-05,
      "model_forward_time": 1.7489268779754639,
      "step": 91372
    },
    {
      "epoch": 1.6567230224609375e-05,
      "step": 91372,
      "training_step_time": 7.40976095199585
    },
    {
      "epoch": 1.6568183898925782e-05,
      "model_forward_time": 1.7537205219268799,
      "step": 91373
    },
    {
      "epoch": 1.6568183898925782e-05,
      "step": 91373,
      "training_step_time": 9.021313428878784
    },
    {
      "epoch": 1.6569137573242186e-05,
      "model_forward_time": 2.858142137527466,
      "step": 91374
    },
    {
      "epoch": 1.6569137573242186e-05,
      "step": 91374,
      "training_step_time": 10.242730379104614
    },
    {
      "epoch": 1.6570091247558594e-05,
      "model_forward_time": 1.7846794128417969,
      "step": 91375
    },
    {
      "epoch": 1.6570091247558594e-05,
      "step": 91375,
      "training_step_time": 9.383928298950195
    },
    {
      "epoch": 1.6571044921875e-05,
      "model_forward_time": 1.7676756381988525,
      "step": 91376
    },
    {
      "epoch": 1.6571044921875e-05,
      "step": 91376,
      "training_step_time": 7.763810634613037
    },
    {
      "epoch": 1.6571998596191405e-05,
      "model_forward_time": 1.7558989524841309,
      "step": 91377
    },
    {
      "epoch": 1.6571998596191405e-05,
      "step": 91377,
      "training_step_time": 7.648439407348633
    },
    {
      "epoch": 1.6572952270507813e-05,
      "model_forward_time": 1.7512180805206299,
      "step": 91378
    },
    {
      "epoch": 1.6572952270507813e-05,
      "step": 91378,
      "training_step_time": 8.20301365852356
    },
    {
      "epoch": 1.657390594482422e-05,
      "model_forward_time": 1.749769926071167,
      "step": 91379
    },
    {
      "epoch": 1.657390594482422e-05,
      "step": 91379,
      "training_step_time": 7.54502534866333
    },
    {
      "epoch": 1.6574859619140624e-05,
      "grad_norm": 0.08467242121696472,
      "learning_rate": 2.0182005722591548e-07,
      "loss": 0.0535,
      "step": 91380
    },
    {
      "epoch": 1.6574859619140624e-05,
      "model_forward_time": 1.756044626235962,
      "step": 91380
    },
    {
      "dynamics_loss_avg": 0.04687714884057641,
      "epoch": 1.6574859619140624e-05,
      "step": 91380
    },
    {
      "action_loss_avg": 0.0015664535982068627,
      "epoch": 1.6574859619140624e-05,
      "step": 91380
    },
    {
      "epoch": 1.6574859619140624e-05,
      "step": 91380,
      "training_step_time": 7.669921398162842
    },
    {
      "epoch": 1.657581329345703e-05,
      "model_forward_time": 1.7475295066833496,
      "step": 91381
    },
    {
      "epoch": 1.657581329345703e-05,
      "step": 91381,
      "training_step_time": 7.6173224449157715
    },
    {
      "epoch": 1.657676696777344e-05,
      "model_forward_time": 1.7316794395446777,
      "step": 91382
    },
    {
      "epoch": 1.657676696777344e-05,
      "step": 91382,
      "training_step_time": 8.408576488494873
    },
    {
      "epoch": 1.6577720642089843e-05,
      "model_forward_time": 2.402695894241333,
      "step": 91383
    },
    {
      "epoch": 1.6577720642089843e-05,
      "step": 91383,
      "training_step_time": 9.76911473274231
    },
    {
      "epoch": 1.657867431640625e-05,
      "model_forward_time": 1.752706527709961,
      "step": 91384
    },
    {
      "epoch": 1.657867431640625e-05,
      "step": 91384,
      "training_step_time": 7.737243175506592
    },
    {
      "epoch": 1.6579627990722658e-05,
      "model_forward_time": 1.7376623153686523,
      "step": 91385
    },
    {
      "epoch": 1.6579627990722658e-05,
      "step": 91385,
      "training_step_time": 8.578030347824097
    },
    {
      "epoch": 1.6580581665039062e-05,
      "model_forward_time": 1.7722651958465576,
      "step": 91386
    },
    {
      "epoch": 1.6580581665039062e-05,
      "step": 91386,
      "training_step_time": 7.701842546463013
    },
    {
      "epoch": 1.658153533935547e-05,
      "model_forward_time": 1.7337713241577148,
      "step": 91387
    },
    {
      "epoch": 1.658153533935547e-05,
      "step": 91387,
      "training_step_time": 7.732544660568237
    },
    {
      "epoch": 1.6582489013671877e-05,
      "model_forward_time": 1.7395029067993164,
      "step": 91388
    },
    {
      "epoch": 1.6582489013671877e-05,
      "step": 91388,
      "training_step_time": 7.665298700332642
    },
    {
      "epoch": 1.658344268798828e-05,
      "model_forward_time": 1.7395079135894775,
      "step": 91389
    },
    {
      "epoch": 1.658344268798828e-05,
      "step": 91389,
      "training_step_time": 7.657348155975342
    },
    {
      "epoch": 1.6584396362304688e-05,
      "grad_norm": 0.06795281171798706,
      "learning_rate": 2.0135528941922944e-07,
      "loss": 0.0541,
      "step": 91390
    },
    {
      "epoch": 1.6584396362304688e-05,
      "model_forward_time": 1.7421493530273438,
      "step": 91390
    },
    {
      "dynamics_loss_avg": 0.05358431152999401,
      "epoch": 1.6584396362304688e-05,
      "step": 91390
    },
    {
      "action_loss_avg": 0.0019216915097786114,
      "epoch": 1.6584396362304688e-05,
      "step": 91390
    },
    {
      "epoch": 1.6584396362304688e-05,
      "step": 91390,
      "training_step_time": 7.7211315631866455
    },
    {
      "epoch": 1.6585350036621092e-05,
      "model_forward_time": 1.7372965812683105,
      "step": 91391
    },
    {
      "epoch": 1.6585350036621092e-05,
      "step": 91391,
      "training_step_time": 9.234171628952026
    },
    {
      "epoch": 1.65863037109375e-05,
      "model_forward_time": 1.7428383827209473,
      "step": 91392
    },
    {
      "epoch": 1.65863037109375e-05,
      "step": 91392,
      "training_step_time": 10.833877325057983
    },
    {
      "epoch": 1.6587257385253907e-05,
      "model_forward_time": 1.7491917610168457,
      "step": 91393
    },
    {
      "epoch": 1.6587257385253907e-05,
      "step": 91393,
      "training_step_time": 8.600258827209473
    },
    {
      "epoch": 1.658821105957031e-05,
      "model_forward_time": 1.7449219226837158,
      "step": 91394
    },
    {
      "epoch": 1.658821105957031e-05,
      "step": 91394,
      "training_step_time": 7.737245798110962
    },
    {
      "epoch": 1.658916473388672e-05,
      "model_forward_time": 1.7568714618682861,
      "step": 91395
    },
    {
      "epoch": 1.658916473388672e-05,
      "step": 91395,
      "training_step_time": 7.678647041320801
    },
    {
      "epoch": 1.6590118408203126e-05,
      "model_forward_time": 1.739112377166748,
      "step": 91396
    },
    {
      "epoch": 1.6590118408203126e-05,
      "step": 91396,
      "training_step_time": 7.673068523406982
    },
    {
      "epoch": 1.659107208251953e-05,
      "model_forward_time": 1.7480783462524414,
      "step": 91397
    },
    {
      "epoch": 1.659107208251953e-05,
      "step": 91397,
      "training_step_time": 7.638978958129883
    },
    {
      "epoch": 1.6592025756835937e-05,
      "model_forward_time": 1.7399640083312988,
      "step": 91398
    },
    {
      "epoch": 1.6592025756835937e-05,
      "step": 91398,
      "training_step_time": 7.65291166305542
    },
    {
      "epoch": 1.6592979431152345e-05,
      "model_forward_time": 1.7412028312683105,
      "step": 91399
    },
    {
      "epoch": 1.6592979431152345e-05,
      "step": 91399,
      "training_step_time": 7.752545118331909
    },
    {
      "epoch": 1.659393310546875e-05,
      "grad_norm": 0.09890499711036682,
      "learning_rate": 2.0089104638511491e-07,
      "loss": 0.054,
      "step": 91400
    },
    {
      "epoch": 1.659393310546875e-05,
      "model_forward_time": 1.7393174171447754,
      "step": 91400
    },
    {
      "dynamics_loss_avg": 0.05094309486448765,
      "epoch": 1.659393310546875e-05,
      "step": 91400
    },
    {
      "action_loss_avg": 0.0014877128356602042,
      "epoch": 1.659393310546875e-05,
      "step": 91400
    },
    {
      "epoch": 1.659393310546875e-05,
      "step": 91400,
      "training_step_time": 8.738751888275146
    },
    {
      "epoch": 1.6594886779785156e-05,
      "model_forward_time": 1.7381036281585693,
      "step": 91401
    },
    {
      "epoch": 1.6594886779785156e-05,
      "step": 91401,
      "training_step_time": 9.52148699760437
    },
    {
      "epoch": 1.6595840454101563e-05,
      "model_forward_time": 1.7755703926086426,
      "step": 91402
    },
    {
      "epoch": 1.6595840454101563e-05,
      "step": 91402,
      "training_step_time": 10.369096755981445
    },
    {
      "epoch": 1.6596794128417968e-05,
      "model_forward_time": 1.7382831573486328,
      "step": 91403
    },
    {
      "epoch": 1.6596794128417968e-05,
      "step": 91403,
      "training_step_time": 7.8003249168396
    },
    {
      "epoch": 1.6597747802734375e-05,
      "model_forward_time": 1.7631323337554932,
      "step": 91404
    },
    {
      "epoch": 1.6597747802734375e-05,
      "step": 91404,
      "training_step_time": 7.686722755432129
    },
    {
      "epoch": 1.6598701477050782e-05,
      "model_forward_time": 1.7498927116394043,
      "step": 91405
    },
    {
      "epoch": 1.6598701477050782e-05,
      "step": 91405,
      "training_step_time": 9.04890251159668
    },
    {
      "epoch": 1.6599655151367186e-05,
      "model_forward_time": 1.7416889667510986,
      "step": 91406
    },
    {
      "epoch": 1.6599655151367186e-05,
      "step": 91406,
      "training_step_time": 7.775591850280762
    },
    {
      "epoch": 1.6600608825683594e-05,
      "model_forward_time": 1.73874831199646,
      "step": 91407
    },
    {
      "epoch": 1.6600608825683594e-05,
      "step": 91407,
      "training_step_time": 7.625506401062012
    },
    {
      "epoch": 1.66015625e-05,
      "model_forward_time": 1.7503952980041504,
      "step": 91408
    },
    {
      "epoch": 1.66015625e-05,
      "step": 91408,
      "training_step_time": 7.707123517990112
    },
    {
      "epoch": 1.6602516174316405e-05,
      "model_forward_time": 1.7371106147766113,
      "step": 91409
    },
    {
      "epoch": 1.6602516174316405e-05,
      "step": 91409,
      "training_step_time": 7.653770923614502
    },
    {
      "epoch": 1.6603469848632813e-05,
      "grad_norm": 0.08926279097795486,
      "learning_rate": 2.0042732817433963e-07,
      "loss": 0.0549,
      "step": 91410
    },
    {
      "epoch": 1.6603469848632813e-05,
      "model_forward_time": 1.7311255931854248,
      "step": 91410
    },
    {
      "dynamics_loss_avg": 0.06943614333868027,
      "epoch": 1.6603469848632813e-05,
      "step": 91410
    },
    {
      "action_loss_avg": 0.0021650889364536853,
      "epoch": 1.6603469848632813e-05,
      "step": 91410
    },
    {
      "epoch": 1.6603469848632813e-05,
      "step": 91410,
      "training_step_time": 8.289191484451294
    },
    {
      "epoch": 1.660442352294922e-05,
      "model_forward_time": 1.744788646697998,
      "step": 91411
    },
    {
      "epoch": 1.660442352294922e-05,
      "step": 91411,
      "training_step_time": 8.19028377532959
    },
    {
      "epoch": 1.6605377197265624e-05,
      "model_forward_time": 1.7363145351409912,
      "step": 91412
    },
    {
      "epoch": 1.6605377197265624e-05,
      "step": 91412,
      "training_step_time": 7.714867830276489
    },
    {
      "epoch": 1.660633087158203e-05,
      "model_forward_time": 1.726170301437378,
      "step": 91413
    },
    {
      "epoch": 1.660633087158203e-05,
      "step": 91413,
      "training_step_time": 7.718526601791382
    },
    {
      "epoch": 1.660728454589844e-05,
      "model_forward_time": 1.745455026626587,
      "step": 91414
    },
    {
      "epoch": 1.660728454589844e-05,
      "step": 91414,
      "training_step_time": 8.523847341537476
    },
    {
      "epoch": 1.6608238220214843e-05,
      "model_forward_time": 1.7378900051116943,
      "step": 91415
    },
    {
      "epoch": 1.6608238220214843e-05,
      "step": 91415,
      "training_step_time": 8.513550996780396
    },
    {
      "epoch": 1.660919189453125e-05,
      "model_forward_time": 1.7347793579101562,
      "step": 91416
    },
    {
      "epoch": 1.660919189453125e-05,
      "step": 91416,
      "training_step_time": 7.594342947006226
    },
    {
      "epoch": 1.6610145568847658e-05,
      "model_forward_time": 1.7394335269927979,
      "step": 91417
    },
    {
      "epoch": 1.6610145568847658e-05,
      "step": 91417,
      "training_step_time": 7.6224143505096436
    },
    {
      "epoch": 1.6611099243164062e-05,
      "model_forward_time": 1.7413272857666016,
      "step": 91418
    },
    {
      "epoch": 1.6611099243164062e-05,
      "step": 91418,
      "training_step_time": 7.5330140590667725
    },
    {
      "epoch": 1.661205291748047e-05,
      "model_forward_time": 1.7467398643493652,
      "step": 91419
    },
    {
      "epoch": 1.661205291748047e-05,
      "step": 91419,
      "training_step_time": 8.666685342788696
    },
    {
      "epoch": 1.6613006591796877e-05,
      "grad_norm": 0.13655376434326172,
      "learning_rate": 1.9996413483761578e-07,
      "loss": 0.0538,
      "step": 91420
    },
    {
      "epoch": 1.6613006591796877e-05,
      "model_forward_time": 2.8575997352600098,
      "step": 91420
    },
    {
      "dynamics_loss_avg": 0.0646475687623024,
      "epoch": 1.6613006591796877e-05,
      "step": 91420
    },
    {
      "action_loss_avg": 0.0029272108426084743,
      "epoch": 1.6613006591796877e-05,
      "step": 91420
    },
    {
      "epoch": 1.6613006591796877e-05,
      "step": 91420,
      "training_step_time": 8.196672677993774
    },
    {
      "epoch": 1.661396026611328e-05,
      "model_forward_time": 1.7519443035125732,
      "step": 91421
    },
    {
      "epoch": 1.661396026611328e-05,
      "step": 91421,
      "training_step_time": 8.212827205657959
    },
    {
      "epoch": 1.6614913940429688e-05,
      "model_forward_time": 1.7364501953125,
      "step": 91422
    },
    {
      "epoch": 1.6614913940429688e-05,
      "step": 91422,
      "training_step_time": 9.543277263641357
    },
    {
      "epoch": 1.6615867614746092e-05,
      "model_forward_time": 1.7574398517608643,
      "step": 91423
    },
    {
      "epoch": 1.6615867614746092e-05,
      "step": 91423,
      "training_step_time": 7.583251953125
    },
    {
      "epoch": 1.66168212890625e-05,
      "model_forward_time": 1.716343641281128,
      "step": 91424
    },
    {
      "epoch": 1.66168212890625e-05,
      "step": 91424,
      "training_step_time": 7.614140748977661
    },
    {
      "epoch": 1.6617774963378907e-05,
      "model_forward_time": 1.7411205768585205,
      "step": 91425
    },
    {
      "epoch": 1.6617774963378907e-05,
      "step": 91425,
      "training_step_time": 7.658094167709351
    },
    {
      "epoch": 1.661872863769531e-05,
      "model_forward_time": 1.7380633354187012,
      "step": 91426
    },
    {
      "epoch": 1.661872863769531e-05,
      "step": 91426,
      "training_step_time": 7.53106689453125
    },
    {
      "epoch": 1.661968231201172e-05,
      "model_forward_time": 1.7283515930175781,
      "step": 91427
    },
    {
      "epoch": 1.661968231201172e-05,
      "step": 91427,
      "training_step_time": 7.589478969573975
    },
    {
      "epoch": 1.6620635986328126e-05,
      "model_forward_time": 1.738718032836914,
      "step": 91428
    },
    {
      "epoch": 1.6620635986328126e-05,
      "step": 91428,
      "training_step_time": 7.669434070587158
    },
    {
      "epoch": 1.662158966064453e-05,
      "model_forward_time": 2.8852477073669434,
      "step": 91429
    },
    {
      "epoch": 1.662158966064453e-05,
      "step": 91429,
      "training_step_time": 10.215965509414673
    },
    {
      "epoch": 1.6622543334960937e-05,
      "grad_norm": 0.08622284978628159,
      "learning_rate": 1.9950146642559788e-07,
      "loss": 0.0521,
      "step": 91430
    },
    {
      "epoch": 1.6622543334960937e-05,
      "model_forward_time": 1.747173547744751,
      "step": 91430
    },
    {
      "dynamics_loss_avg": 0.03953409362584352,
      "epoch": 1.6622543334960937e-05,
      "step": 91430
    },
    {
      "action_loss_avg": 0.0018775234668282792,
      "epoch": 1.6622543334960937e-05,
      "step": 91430
    },
    {
      "epoch": 1.6622543334960937e-05,
      "step": 91430,
      "training_step_time": 8.575225830078125
    },
    {
      "epoch": 1.6623497009277345e-05,
      "model_forward_time": 1.7340104579925537,
      "step": 91431
    },
    {
      "epoch": 1.6623497009277345e-05,
      "step": 91431,
      "training_step_time": 8.549934387207031
    },
    {
      "epoch": 1.662445068359375e-05,
      "model_forward_time": 1.7524404525756836,
      "step": 91432
    },
    {
      "epoch": 1.662445068359375e-05,
      "step": 91432,
      "training_step_time": 7.579908132553101
    },
    {
      "epoch": 1.6625404357910156e-05,
      "model_forward_time": 1.738520860671997,
      "step": 91433
    },
    {
      "epoch": 1.6625404357910156e-05,
      "step": 91433,
      "training_step_time": 8.20835256576538
    },
    {
      "epoch": 1.6626358032226564e-05,
      "model_forward_time": 1.7327113151550293,
      "step": 91434
    },
    {
      "epoch": 1.6626358032226564e-05,
      "step": 91434,
      "training_step_time": 7.59201192855835
    },
    {
      "epoch": 1.6627311706542968e-05,
      "model_forward_time": 1.7338144779205322,
      "step": 91435
    },
    {
      "epoch": 1.6627311706542968e-05,
      "step": 91435,
      "training_step_time": 7.539200305938721
    },
    {
      "epoch": 1.6628265380859375e-05,
      "model_forward_time": 1.7368841171264648,
      "step": 91436
    },
    {
      "epoch": 1.6628265380859375e-05,
      "step": 91436,
      "training_step_time": 7.629549980163574
    },
    {
      "epoch": 1.6629219055175782e-05,
      "model_forward_time": 1.7443151473999023,
      "step": 91437
    },
    {
      "epoch": 1.6629219055175782e-05,
      "step": 91437,
      "training_step_time": 7.63334584236145
    },
    {
      "epoch": 1.6630172729492186e-05,
      "model_forward_time": 2.826549768447876,
      "step": 91438
    },
    {
      "epoch": 1.6630172729492186e-05,
      "step": 91438,
      "training_step_time": 8.574549198150635
    },
    {
      "epoch": 1.6631126403808594e-05,
      "model_forward_time": 1.7436332702636719,
      "step": 91439
    },
    {
      "epoch": 1.6631126403808594e-05,
      "step": 91439,
      "training_step_time": 8.780942440032959
    },
    {
      "epoch": 1.6632080078125e-05,
      "grad_norm": 0.0904448926448822,
      "learning_rate": 1.9903932298888152e-07,
      "loss": 0.053,
      "step": 91440
    },
    {
      "epoch": 1.6632080078125e-05,
      "model_forward_time": 1.7436511516571045,
      "step": 91440
    },
    {
      "dynamics_loss_avg": 0.05424372870475054,
      "epoch": 1.6632080078125e-05,
      "step": 91440
    },
    {
      "action_loss_avg": 0.0016001440060790628,
      "epoch": 1.6632080078125e-05,
      "step": 91440
    },
    {
      "epoch": 1.6632080078125e-05,
      "step": 91440,
      "training_step_time": 7.624161958694458
    },
    {
      "epoch": 1.6633033752441405e-05,
      "model_forward_time": 1.757828950881958,
      "step": 91441
    },
    {
      "epoch": 1.6633033752441405e-05,
      "step": 91441,
      "training_step_time": 7.656907081604004
    },
    {
      "epoch": 1.6633987426757813e-05,
      "model_forward_time": 1.736060380935669,
      "step": 91442
    },
    {
      "epoch": 1.6633987426757813e-05,
      "step": 91442,
      "training_step_time": 7.624498128890991
    },
    {
      "epoch": 1.663494110107422e-05,
      "model_forward_time": 1.7361443042755127,
      "step": 91443
    },
    {
      "epoch": 1.663494110107422e-05,
      "step": 91443,
      "training_step_time": 8.461377143859863
    },
    {
      "epoch": 1.6635894775390624e-05,
      "model_forward_time": 1.730738878250122,
      "step": 91444
    },
    {
      "epoch": 1.6635894775390624e-05,
      "step": 91444,
      "training_step_time": 7.545724153518677
    },
    {
      "epoch": 1.663684844970703e-05,
      "model_forward_time": 1.7355213165283203,
      "step": 91445
    },
    {
      "epoch": 1.663684844970703e-05,
      "step": 91445,
      "training_step_time": 7.573939561843872
    },
    {
      "epoch": 1.663780212402344e-05,
      "model_forward_time": 1.738152027130127,
      "step": 91446
    },
    {
      "epoch": 1.663780212402344e-05,
      "step": 91446,
      "training_step_time": 7.6132447719573975
    },
    {
      "epoch": 1.6638755798339843e-05,
      "model_forward_time": 2.819559097290039,
      "step": 91447
    },
    {
      "epoch": 1.6638755798339843e-05,
      "step": 91447,
      "training_step_time": 8.65360164642334
    },
    {
      "epoch": 1.663970947265625e-05,
      "model_forward_time": 1.7486059665679932,
      "step": 91448
    },
    {
      "epoch": 1.663970947265625e-05,
      "step": 91448,
      "training_step_time": 8.677648782730103
    },
    {
      "epoch": 1.6640663146972658e-05,
      "model_forward_time": 1.76190185546875,
      "step": 91449
    },
    {
      "epoch": 1.6640663146972658e-05,
      "step": 91449,
      "training_step_time": 9.172534465789795
    },
    {
      "epoch": 1.6641616821289062e-05,
      "grad_norm": 0.10137207061052322,
      "learning_rate": 1.9857770457800684e-07,
      "loss": 0.0526,
      "step": 91450
    },
    {
      "epoch": 1.6641616821289062e-05,
      "model_forward_time": 1.7350826263427734,
      "step": 91450
    },
    {
      "dynamics_loss_avg": 0.06237030513584614,
      "epoch": 1.6641616821289062e-05,
      "step": 91450
    },
    {
      "action_loss_avg": 0.0018900363123975695,
      "epoch": 1.6641616821289062e-05,
      "step": 91450
    },
    {
      "epoch": 1.6641616821289062e-05,
      "step": 91450,
      "training_step_time": 7.696402549743652
    },
    {
      "epoch": 1.664257049560547e-05,
      "model_forward_time": 1.749011516571045,
      "step": 91451
    },
    {
      "epoch": 1.664257049560547e-05,
      "step": 91451,
      "training_step_time": 8.147833347320557
    },
    {
      "epoch": 1.6643524169921877e-05,
      "model_forward_time": 1.7452316284179688,
      "step": 91452
    },
    {
      "epoch": 1.6643524169921877e-05,
      "step": 91452,
      "training_step_time": 7.694628000259399
    },
    {
      "epoch": 1.664447784423828e-05,
      "model_forward_time": 1.7475683689117432,
      "step": 91453
    },
    {
      "epoch": 1.664447784423828e-05,
      "step": 91453,
      "training_step_time": 7.4980878829956055
    },
    {
      "epoch": 1.6645431518554688e-05,
      "model_forward_time": 1.737290859222412,
      "step": 91454
    },
    {
      "epoch": 1.6645431518554688e-05,
      "step": 91454,
      "training_step_time": 8.336058616638184
    },
    {
      "epoch": 1.6646385192871092e-05,
      "model_forward_time": 1.74871826171875,
      "step": 91455
    },
    {
      "epoch": 1.6646385192871092e-05,
      "step": 91455,
      "training_step_time": 7.612519025802612
    },
    {
      "epoch": 1.66473388671875e-05,
      "model_forward_time": 2.4035277366638184,
      "step": 91456
    },
    {
      "epoch": 1.66473388671875e-05,
      "step": 91456,
      "training_step_time": 8.24104380607605
    },
    {
      "epoch": 1.6648292541503907e-05,
      "model_forward_time": 1.7381982803344727,
      "step": 91457
    },
    {
      "epoch": 1.6648292541503907e-05,
      "step": 91457,
      "training_step_time": 8.97653603553772
    },
    {
      "epoch": 1.664924621582031e-05,
      "model_forward_time": 1.7593433856964111,
      "step": 91458
    },
    {
      "epoch": 1.664924621582031e-05,
      "step": 91458,
      "training_step_time": 7.60507869720459
    },
    {
      "epoch": 1.665019989013672e-05,
      "model_forward_time": 1.7434918880462646,
      "step": 91459
    },
    {
      "epoch": 1.665019989013672e-05,
      "step": 91459,
      "training_step_time": 9.079100370407104
    },
    {
      "epoch": 1.6651153564453126e-05,
      "grad_norm": 0.12323389947414398,
      "learning_rate": 1.9811661124345515e-07,
      "loss": 0.0516,
      "step": 91460
    },
    {
      "epoch": 1.6651153564453126e-05,
      "model_forward_time": 1.7756342887878418,
      "step": 91460
    },
    {
      "dynamics_loss_avg": 0.05047841984778643,
      "epoch": 1.6651153564453126e-05,
      "step": 91460
    },
    {
      "action_loss_avg": 0.0018726389156654478,
      "epoch": 1.6651153564453126e-05,
      "step": 91460
    },
    {
      "epoch": 1.6651153564453126e-05,
      "step": 91460,
      "training_step_time": 7.6624767780303955
    },
    {
      "epoch": 1.665210723876953e-05,
      "model_forward_time": 1.738163709640503,
      "step": 91461
    },
    {
      "epoch": 1.665210723876953e-05,
      "step": 91461,
      "training_step_time": 8.140538930892944
    },
    {
      "epoch": 1.6653060913085937e-05,
      "model_forward_time": 1.7425284385681152,
      "step": 91462
    },
    {
      "epoch": 1.6653060913085937e-05,
      "step": 91462,
      "training_step_time": 7.46157693862915
    },
    {
      "epoch": 1.6654014587402345e-05,
      "model_forward_time": 1.751359462738037,
      "step": 91463
    },
    {
      "epoch": 1.6654014587402345e-05,
      "step": 91463,
      "training_step_time": 7.672660827636719
    },
    {
      "epoch": 1.665496826171875e-05,
      "model_forward_time": 1.7403154373168945,
      "step": 91464
    },
    {
      "epoch": 1.665496826171875e-05,
      "step": 91464,
      "training_step_time": 7.572307348251343
    },
    {
      "epoch": 1.6655921936035156e-05,
      "model_forward_time": 1.7410016059875488,
      "step": 91465
    },
    {
      "epoch": 1.6655921936035156e-05,
      "step": 91465,
      "training_step_time": 7.722891807556152
    },
    {
      "epoch": 1.6656875610351564e-05,
      "model_forward_time": 2.872276782989502,
      "step": 91466
    },
    {
      "epoch": 1.6656875610351564e-05,
      "step": 91466,
      "training_step_time": 7.714700698852539
    },
    {
      "epoch": 1.6657829284667968e-05,
      "model_forward_time": 1.7486169338226318,
      "step": 91467
    },
    {
      "epoch": 1.6657829284667968e-05,
      "step": 91467,
      "training_step_time": 9.655332326889038
    },
    {
      "epoch": 1.6658782958984375e-05,
      "model_forward_time": 1.7436466217041016,
      "step": 91468
    },
    {
      "epoch": 1.6658782958984375e-05,
      "step": 91468,
      "training_step_time": 8.354336261749268
    },
    {
      "epoch": 1.6659736633300782e-05,
      "model_forward_time": 1.7765462398529053,
      "step": 91469
    },
    {
      "epoch": 1.6659736633300782e-05,
      "step": 91469,
      "training_step_time": 7.672237157821655
    },
    {
      "epoch": 1.6660690307617186e-05,
      "grad_norm": 0.0852387472987175,
      "learning_rate": 1.9765604303565223e-07,
      "loss": 0.0523,
      "step": 91470
    },
    {
      "epoch": 1.6660690307617186e-05,
      "model_forward_time": 1.736600637435913,
      "step": 91470
    },
    {
      "dynamics_loss_avg": 0.07184650301933289,
      "epoch": 1.6660690307617186e-05,
      "step": 91470
    },
    {
      "action_loss_avg": 0.004563308833166957,
      "epoch": 1.6660690307617186e-05,
      "step": 91470
    },
    {
      "epoch": 1.6660690307617186e-05,
      "step": 91470,
      "training_step_time": 7.590450286865234
    },
    {
      "epoch": 1.6661643981933594e-05,
      "model_forward_time": 1.741342544555664,
      "step": 91471
    },
    {
      "epoch": 1.6661643981933594e-05,
      "step": 91471,
      "training_step_time": 7.610577821731567
    },
    {
      "epoch": 1.666259765625e-05,
      "model_forward_time": 1.7485647201538086,
      "step": 91472
    },
    {
      "epoch": 1.666259765625e-05,
      "step": 91472,
      "training_step_time": 7.584392309188843
    },
    {
      "epoch": 1.6663551330566405e-05,
      "model_forward_time": 1.750406265258789,
      "step": 91473
    },
    {
      "epoch": 1.6663551330566405e-05,
      "step": 91473,
      "training_step_time": 7.624298334121704
    },
    {
      "epoch": 1.6664505004882813e-05,
      "model_forward_time": 1.7365577220916748,
      "step": 91474
    },
    {
      "epoch": 1.6664505004882813e-05,
      "step": 91474,
      "training_step_time": 8.341970920562744
    },
    {
      "epoch": 1.666545867919922e-05,
      "model_forward_time": 2.839744806289673,
      "step": 91475
    },
    {
      "epoch": 1.666545867919922e-05,
      "step": 91475,
      "training_step_time": 9.01867389678955
    },
    {
      "epoch": 1.6666412353515624e-05,
      "model_forward_time": 1.7454445362091064,
      "step": 91476
    },
    {
      "epoch": 1.6666412353515624e-05,
      "step": 91476,
      "training_step_time": 9.654240608215332
    },
    {
      "epoch": 1.666736602783203e-05,
      "model_forward_time": 1.7704999446868896,
      "step": 91477
    },
    {
      "epoch": 1.666736602783203e-05,
      "step": 91477,
      "training_step_time": 7.658313274383545
    },
    {
      "epoch": 1.666831970214844e-05,
      "model_forward_time": 1.7725410461425781,
      "step": 91478
    },
    {
      "epoch": 1.666831970214844e-05,
      "step": 91478,
      "training_step_time": 8.598087549209595
    },
    {
      "epoch": 1.6669273376464843e-05,
      "model_forward_time": 1.7477855682373047,
      "step": 91479
    },
    {
      "epoch": 1.6669273376464843e-05,
      "step": 91479,
      "training_step_time": 8.17626404762268
    },
    {
      "epoch": 1.667022705078125e-05,
      "grad_norm": 0.15367241203784943,
      "learning_rate": 1.9719600000496387e-07,
      "loss": 0.0514,
      "step": 91480
    },
    {
      "epoch": 1.667022705078125e-05,
      "model_forward_time": 1.7449688911437988,
      "step": 91480
    },
    {
      "dynamics_loss_avg": 0.04574285577982664,
      "epoch": 1.667022705078125e-05,
      "step": 91480
    },
    {
      "action_loss_avg": 0.001789940771413967,
      "epoch": 1.667022705078125e-05,
      "step": 91480
    },
    {
      "epoch": 1.667022705078125e-05,
      "step": 91480,
      "training_step_time": 7.573113679885864
    },
    {
      "epoch": 1.6671180725097658e-05,
      "model_forward_time": 1.7577025890350342,
      "step": 91481
    },
    {
      "epoch": 1.6671180725097658e-05,
      "step": 91481,
      "training_step_time": 7.539449453353882
    },
    {
      "epoch": 1.6672134399414062e-05,
      "model_forward_time": 1.7433433532714844,
      "step": 91482
    },
    {
      "epoch": 1.6672134399414062e-05,
      "step": 91482,
      "training_step_time": 7.643562316894531
    },
    {
      "epoch": 1.667308807373047e-05,
      "model_forward_time": 1.751657485961914,
      "step": 91483
    },
    {
      "epoch": 1.667308807373047e-05,
      "step": 91483,
      "training_step_time": 7.690743684768677
    },
    {
      "epoch": 1.6674041748046877e-05,
      "model_forward_time": 2.8068463802337646,
      "step": 91484
    },
    {
      "epoch": 1.6674041748046877e-05,
      "step": 91484,
      "training_step_time": 10.447450160980225
    },
    {
      "epoch": 1.667499542236328e-05,
      "model_forward_time": 1.7440485954284668,
      "step": 91485
    },
    {
      "epoch": 1.667499542236328e-05,
      "step": 91485,
      "training_step_time": 9.657620191574097
    },
    {
      "epoch": 1.6675949096679688e-05,
      "model_forward_time": 1.750462293624878,
      "step": 91486
    },
    {
      "epoch": 1.6675949096679688e-05,
      "step": 91486,
      "training_step_time": 8.501176357269287
    },
    {
      "epoch": 1.6676902770996092e-05,
      "model_forward_time": 1.7603123188018799,
      "step": 91487
    },
    {
      "epoch": 1.6676902770996092e-05,
      "step": 91487,
      "training_step_time": 7.614020824432373
    },
    {
      "epoch": 1.66778564453125e-05,
      "model_forward_time": 1.7468180656433105,
      "step": 91488
    },
    {
      "epoch": 1.66778564453125e-05,
      "step": 91488,
      "training_step_time": 7.565780878067017
    },
    {
      "epoch": 1.6678810119628907e-05,
      "model_forward_time": 1.7434360980987549,
      "step": 91489
    },
    {
      "epoch": 1.6678810119628907e-05,
      "step": 91489,
      "training_step_time": 7.688145875930786
    },
    {
      "epoch": 1.667976379394531e-05,
      "grad_norm": 0.09338575601577759,
      "learning_rate": 1.9673648220170038e-07,
      "loss": 0.0532,
      "step": 91490
    },
    {
      "epoch": 1.667976379394531e-05,
      "model_forward_time": 1.7399792671203613,
      "step": 91490
    },
    {
      "dynamics_loss_avg": 0.040622046869248155,
      "epoch": 1.667976379394531e-05,
      "step": 91490
    },
    {
      "action_loss_avg": 0.002491023630136624,
      "epoch": 1.667976379394531e-05,
      "step": 91490
    },
    {
      "epoch": 1.667976379394531e-05,
      "step": 91490,
      "training_step_time": 7.635923862457275
    },
    {
      "epoch": 1.668071746826172e-05,
      "model_forward_time": 1.739649772644043,
      "step": 91491
    },
    {
      "epoch": 1.668071746826172e-05,
      "step": 91491,
      "training_step_time": 8.072988986968994
    },
    {
      "epoch": 1.6681671142578126e-05,
      "model_forward_time": 1.7459008693695068,
      "step": 91492
    },
    {
      "epoch": 1.6681671142578126e-05,
      "step": 91492,
      "training_step_time": 8.587706089019775
    },
    {
      "epoch": 1.668262481689453e-05,
      "model_forward_time": 2.8172082901000977,
      "step": 91493
    },
    {
      "epoch": 1.668262481689453e-05,
      "step": 91493,
      "training_step_time": 9.425126314163208
    },
    {
      "epoch": 1.6683578491210937e-05,
      "model_forward_time": 1.7476611137390137,
      "step": 91494
    },
    {
      "epoch": 1.6683578491210937e-05,
      "step": 91494,
      "training_step_time": 8.855541944503784
    },
    {
      "epoch": 1.6684532165527345e-05,
      "model_forward_time": 1.7598354816436768,
      "step": 91495
    },
    {
      "epoch": 1.6684532165527345e-05,
      "step": 91495,
      "training_step_time": 8.469257593154907
    },
    {
      "epoch": 1.668548583984375e-05,
      "model_forward_time": 1.7513716220855713,
      "step": 91496
    },
    {
      "epoch": 1.668548583984375e-05,
      "step": 91496,
      "training_step_time": 7.674203395843506
    },
    {
      "epoch": 1.6686439514160156e-05,
      "model_forward_time": 1.7405588626861572,
      "step": 91497
    },
    {
      "epoch": 1.6686439514160156e-05,
      "step": 91497,
      "training_step_time": 8.655609369277954
    },
    {
      "epoch": 1.6687393188476564e-05,
      "model_forward_time": 1.7420375347137451,
      "step": 91498
    },
    {
      "epoch": 1.6687393188476564e-05,
      "step": 91498,
      "training_step_time": 8.385496139526367
    },
    {
      "epoch": 1.6688346862792968e-05,
      "model_forward_time": 1.7443492412567139,
      "step": 91499
    },
    {
      "epoch": 1.6688346862792968e-05,
      "step": 91499,
      "training_step_time": 7.604318380355835
    },
    {
      "epoch": 1.6689300537109375e-05,
      "grad_norm": 0.08160792291164398,
      "learning_rate": 1.9627748967611271e-07,
      "loss": 0.0524,
      "step": 91500
    },
    {
      "epoch": 1.6689300537109375e-05,
      "model_forward_time": 1.7395610809326172,
      "step": 91500
    },
    {
      "dynamics_loss_avg": 0.0379077410325408,
      "epoch": 1.6689300537109375e-05,
      "step": 91500
    },
    {
      "action_loss_avg": 0.0016988736097118817,
      "epoch": 1.6689300537109375e-05,
      "step": 91500
    },
    {
      "epoch": 1.6689300537109375e-05,
      "step": 91500,
      "training_step_time": 7.564188718795776
    },
    {
      "epoch": 1.6690254211425782e-05,
      "model_forward_time": 1.746608018875122,
      "step": 91501
    },
    {
      "epoch": 1.6690254211425782e-05,
      "step": 91501,
      "training_step_time": 9.057329177856445
    },
    {
      "epoch": 1.6691207885742186e-05,
      "model_forward_time": 2.781930446624756,
      "step": 91502
    },
    {
      "epoch": 1.6691207885742186e-05,
      "step": 91502,
      "training_step_time": 7.587133169174194
    },
    {
      "epoch": 1.6692161560058594e-05,
      "model_forward_time": 1.7336695194244385,
      "step": 91503
    },
    {
      "epoch": 1.6692161560058594e-05,
      "step": 91503,
      "training_step_time": 9.075222253799438
    },
    {
      "epoch": 1.6693115234375e-05,
      "model_forward_time": 1.7624337673187256,
      "step": 91504
    },
    {
      "epoch": 1.6693115234375e-05,
      "step": 91504,
      "training_step_time": 7.818223476409912
    },
    {
      "epoch": 1.6694068908691405e-05,
      "model_forward_time": 1.7418136596679688,
      "step": 91505
    },
    {
      "epoch": 1.6694068908691405e-05,
      "step": 91505,
      "training_step_time": 8.653518438339233
    },
    {
      "epoch": 1.6695022583007813e-05,
      "model_forward_time": 1.7615039348602295,
      "step": 91506
    },
    {
      "epoch": 1.6695022583007813e-05,
      "step": 91506,
      "training_step_time": 7.612205982208252
    },
    {
      "epoch": 1.669597625732422e-05,
      "model_forward_time": 1.7433605194091797,
      "step": 91507
    },
    {
      "epoch": 1.669597625732422e-05,
      "step": 91507,
      "training_step_time": 7.583884000778198
    },
    {
      "epoch": 1.6696929931640624e-05,
      "model_forward_time": 1.7463271617889404,
      "step": 91508
    },
    {
      "epoch": 1.6696929931640624e-05,
      "step": 91508,
      "training_step_time": 7.591003656387329
    },
    {
      "epoch": 1.669788360595703e-05,
      "model_forward_time": 1.7383410930633545,
      "step": 91509
    },
    {
      "epoch": 1.669788360595703e-05,
      "step": 91509,
      "training_step_time": 8.307288408279419
    },
    {
      "epoch": 1.669883728027344e-05,
      "grad_norm": 0.0852910503745079,
      "learning_rate": 1.9581902247839679e-07,
      "loss": 0.0546,
      "step": 91510
    },
    {
      "epoch": 1.669883728027344e-05,
      "model_forward_time": 1.7457497119903564,
      "step": 91510
    },
    {
      "dynamics_loss_avg": 0.04903561621904373,
      "epoch": 1.669883728027344e-05,
      "step": 91510
    },
    {
      "action_loss_avg": 0.001731038439902477,
      "epoch": 1.669883728027344e-05,
      "step": 91510
    },
    {
      "epoch": 1.669883728027344e-05,
      "step": 91510,
      "training_step_time": 7.506608247756958
    },
    {
      "epoch": 1.6699790954589843e-05,
      "model_forward_time": 1.7441010475158691,
      "step": 91511
    },
    {
      "epoch": 1.6699790954589843e-05,
      "step": 91511,
      "training_step_time": 9.98546051979065
    },
    {
      "epoch": 1.670074462890625e-05,
      "model_forward_time": 2.835129499435425,
      "step": 91512
    },
    {
      "epoch": 1.670074462890625e-05,
      "step": 91512,
      "training_step_time": 8.204591274261475
    },
    {
      "epoch": 1.6701698303222658e-05,
      "model_forward_time": 1.747403621673584,
      "step": 91513
    },
    {
      "epoch": 1.6701698303222658e-05,
      "step": 91513,
      "training_step_time": 7.671652317047119
    },
    {
      "epoch": 1.6702651977539062e-05,
      "model_forward_time": 1.7537078857421875,
      "step": 91514
    },
    {
      "epoch": 1.6702651977539062e-05,
      "step": 91514,
      "training_step_time": 8.417733907699585
    },
    {
      "epoch": 1.670360565185547e-05,
      "model_forward_time": 1.7688839435577393,
      "step": 91515
    },
    {
      "epoch": 1.670360565185547e-05,
      "step": 91515,
      "training_step_time": 7.637614011764526
    },
    {
      "epoch": 1.6704559326171873e-05,
      "model_forward_time": 1.7455787658691406,
      "step": 91516
    },
    {
      "epoch": 1.6704559326171873e-05,
      "step": 91516,
      "training_step_time": 7.591487169265747
    },
    {
      "epoch": 1.670551300048828e-05,
      "model_forward_time": 1.7496604919433594,
      "step": 91517
    },
    {
      "epoch": 1.670551300048828e-05,
      "step": 91517,
      "training_step_time": 7.675467252731323
    },
    {
      "epoch": 1.6706466674804688e-05,
      "model_forward_time": 1.7461388111114502,
      "step": 91518
    },
    {
      "epoch": 1.6706466674804688e-05,
      "step": 91518,
      "training_step_time": 8.601783037185669
    },
    {
      "epoch": 1.6707420349121092e-05,
      "model_forward_time": 1.751704454421997,
      "step": 91519
    },
    {
      "epoch": 1.6707420349121092e-05,
      "step": 91519,
      "training_step_time": 7.623076438903809
    },
    {
      "epoch": 1.67083740234375e-05,
      "grad_norm": 0.0824313759803772,
      "learning_rate": 1.9536108065868974e-07,
      "loss": 0.0538,
      "step": 91520
    },
    {
      "epoch": 1.67083740234375e-05,
      "model_forward_time": 1.7304260730743408,
      "step": 91520
    },
    {
      "dynamics_loss_avg": 0.05778492363169789,
      "epoch": 1.67083740234375e-05,
      "step": 91520
    },
    {
      "action_loss_avg": 0.0041013121663127095,
      "epoch": 1.67083740234375e-05,
      "step": 91520
    },
    {
      "epoch": 1.67083740234375e-05,
      "step": 91520,
      "training_step_time": 10.467778444290161
    },
    {
      "epoch": 1.6709327697753907e-05,
      "model_forward_time": 2.861875534057617,
      "step": 91521
    },
    {
      "epoch": 1.6709327697753907e-05,
      "step": 91521,
      "training_step_time": 9.110122680664062
    },
    {
      "epoch": 1.671028137207031e-05,
      "model_forward_time": 1.7516710758209229,
      "step": 91522
    },
    {
      "epoch": 1.671028137207031e-05,
      "step": 91522,
      "training_step_time": 8.35891079902649
    },
    {
      "epoch": 1.671123504638672e-05,
      "model_forward_time": 1.7684283256530762,
      "step": 91523
    },
    {
      "epoch": 1.671123504638672e-05,
      "step": 91523,
      "training_step_time": 8.577810049057007
    },
    {
      "epoch": 1.6712188720703126e-05,
      "model_forward_time": 1.7762935161590576,
      "step": 91524
    },
    {
      "epoch": 1.6712188720703126e-05,
      "step": 91524,
      "training_step_time": 7.7798027992248535
    },
    {
      "epoch": 1.671314239501953e-05,
      "model_forward_time": 1.7580976486206055,
      "step": 91525
    },
    {
      "epoch": 1.671314239501953e-05,
      "step": 91525,
      "training_step_time": 7.671810150146484
    },
    {
      "epoch": 1.6714096069335937e-05,
      "model_forward_time": 1.7531797885894775,
      "step": 91526
    },
    {
      "epoch": 1.6714096069335937e-05,
      "step": 91526,
      "training_step_time": 7.6093573570251465
    },
    {
      "epoch": 1.6715049743652345e-05,
      "model_forward_time": 1.7516484260559082,
      "step": 91527
    },
    {
      "epoch": 1.6715049743652345e-05,
      "step": 91527,
      "training_step_time": 7.660479784011841
    },
    {
      "epoch": 1.671600341796875e-05,
      "model_forward_time": 1.7530086040496826,
      "step": 91528
    },
    {
      "epoch": 1.671600341796875e-05,
      "step": 91528,
      "training_step_time": 7.581039667129517
    },
    {
      "epoch": 1.6716957092285156e-05,
      "model_forward_time": 1.7522382736206055,
      "step": 91529
    },
    {
      "epoch": 1.6716957092285156e-05,
      "step": 91529,
      "training_step_time": 9.197553157806396
    },
    {
      "epoch": 1.6717910766601564e-05,
      "grad_norm": 0.09060601890087128,
      "learning_rate": 1.9490366426707153e-07,
      "loss": 0.0556,
      "step": 91530
    },
    {
      "epoch": 1.6717910766601564e-05,
      "model_forward_time": 2.8613500595092773,
      "step": 91530
    },
    {
      "dynamics_loss_avg": 0.062167270481586455,
      "epoch": 1.6717910766601564e-05,
      "step": 91530
    },
    {
      "action_loss_avg": 0.002153626193467062,
      "epoch": 1.6717910766601564e-05,
      "step": 91530
    },
    {
      "epoch": 1.6717910766601564e-05,
      "step": 91530,
      "training_step_time": 7.607428550720215
    },
    {
      "epoch": 1.6718864440917968e-05,
      "model_forward_time": 1.7554898262023926,
      "step": 91531
    },
    {
      "epoch": 1.6718864440917968e-05,
      "step": 91531,
      "training_step_time": 9.113389730453491
    },
    {
      "epoch": 1.6719818115234375e-05,
      "model_forward_time": 1.7602174282073975,
      "step": 91532
    },
    {
      "epoch": 1.6719818115234375e-05,
      "step": 91532,
      "training_step_time": 8.274919033050537
    },
    {
      "epoch": 1.6720771789550782e-05,
      "model_forward_time": 1.7759082317352295,
      "step": 91533
    },
    {
      "epoch": 1.6720771789550782e-05,
      "step": 91533,
      "training_step_time": 8.402251243591309
    },
    {
      "epoch": 1.6721725463867186e-05,
      "model_forward_time": 1.7664895057678223,
      "step": 91534
    },
    {
      "epoch": 1.6721725463867186e-05,
      "step": 91534,
      "training_step_time": 8.231114149093628
    },
    {
      "epoch": 1.6722679138183594e-05,
      "model_forward_time": 1.7527754306793213,
      "step": 91535
    },
    {
      "epoch": 1.6722679138183594e-05,
      "step": 91535,
      "training_step_time": 7.696599960327148
    },
    {
      "epoch": 1.67236328125e-05,
      "model_forward_time": 1.7582130432128906,
      "step": 91536
    },
    {
      "epoch": 1.67236328125e-05,
      "step": 91536,
      "training_step_time": 7.6796863079071045
    },
    {
      "epoch": 1.6724586486816405e-05,
      "model_forward_time": 1.7576425075531006,
      "step": 91537
    },
    {
      "epoch": 1.6724586486816405e-05,
      "step": 91537,
      "training_step_time": 7.562147617340088
    },
    {
      "epoch": 1.6725540161132813e-05,
      "model_forward_time": 1.7586250305175781,
      "step": 91538
    },
    {
      "epoch": 1.6725540161132813e-05,
      "step": 91538,
      "training_step_time": 8.059316873550415
    },
    {
      "epoch": 1.672649383544922e-05,
      "model_forward_time": 2.877880811691284,
      "step": 91539
    },
    {
      "epoch": 1.672649383544922e-05,
      "step": 91539,
      "training_step_time": 10.097196578979492
    },
    {
      "epoch": 1.6727447509765624e-05,
      "grad_norm": 0.10229107737541199,
      "learning_rate": 1.9444677335356377e-07,
      "loss": 0.056,
      "step": 91540
    },
    {
      "epoch": 1.6727447509765624e-05,
      "model_forward_time": 1.752309799194336,
      "step": 91540
    },
    {
      "dynamics_loss_avg": 0.05530144907534122,
      "epoch": 1.6727447509765624e-05,
      "step": 91540
    },
    {
      "action_loss_avg": 0.001452060736482963,
      "epoch": 1.6727447509765624e-05,
      "step": 91540
    },
    {
      "epoch": 1.6727447509765624e-05,
      "step": 91540,
      "training_step_time": 8.299201488494873
    },
    {
      "epoch": 1.672840118408203e-05,
      "model_forward_time": 1.7720999717712402,
      "step": 91541
    },
    {
      "epoch": 1.672840118408203e-05,
      "step": 91541,
      "training_step_time": 7.568912029266357
    },
    {
      "epoch": 1.672935485839844e-05,
      "model_forward_time": 1.7617299556732178,
      "step": 91542
    },
    {
      "epoch": 1.672935485839844e-05,
      "step": 91542,
      "training_step_time": 7.640763282775879
    },
    {
      "epoch": 1.6730308532714843e-05,
      "model_forward_time": 1.7571849822998047,
      "step": 91543
    },
    {
      "epoch": 1.6730308532714843e-05,
      "step": 91543,
      "training_step_time": 7.615172386169434
    },
    {
      "epoch": 1.673126220703125e-05,
      "model_forward_time": 1.7546818256378174,
      "step": 91544
    },
    {
      "epoch": 1.673126220703125e-05,
      "step": 91544,
      "training_step_time": 7.643698453903198
    },
    {
      "epoch": 1.6732215881347658e-05,
      "model_forward_time": 1.7562284469604492,
      "step": 91545
    },
    {
      "epoch": 1.6732215881347658e-05,
      "step": 91545,
      "training_step_time": 7.639459848403931
    },
    {
      "epoch": 1.6733169555664062e-05,
      "model_forward_time": 1.7412045001983643,
      "step": 91546
    },
    {
      "epoch": 1.6733169555664062e-05,
      "step": 91546,
      "training_step_time": 7.555982828140259
    },
    {
      "epoch": 1.673412322998047e-05,
      "model_forward_time": 1.753157615661621,
      "step": 91547
    },
    {
      "epoch": 1.673412322998047e-05,
      "step": 91547,
      "training_step_time": 7.6218743324279785
    },
    {
      "epoch": 1.6735076904296873e-05,
      "model_forward_time": 2.8489837646484375,
      "step": 91548
    },
    {
      "epoch": 1.6735076904296873e-05,
      "step": 91548,
      "training_step_time": 7.736513137817383
    },
    {
      "epoch": 1.673603057861328e-05,
      "model_forward_time": 1.7578098773956299,
      "step": 91549
    },
    {
      "epoch": 1.673603057861328e-05,
      "step": 91549,
      "training_step_time": 9.39031434059143
    },
    {
      "epoch": 1.6736984252929688e-05,
      "grad_norm": 0.0798538327217102,
      "learning_rate": 1.939904079681326e-07,
      "loss": 0.0554,
      "step": 91550
    },
    {
      "epoch": 1.6736984252929688e-05,
      "model_forward_time": 1.7760820388793945,
      "step": 91550
    },
    {
      "dynamics_loss_avg": 0.05840094834566116,
      "epoch": 1.6736984252929688e-05,
      "step": 91550
    },
    {
      "action_loss_avg": 0.001397683852701448,
      "epoch": 1.6736984252929688e-05,
      "step": 91550
    },
    {
      "epoch": 1.6736984252929688e-05,
      "step": 91550,
      "training_step_time": 8.228608846664429
    },
    {
      "epoch": 1.6737937927246092e-05,
      "model_forward_time": 1.7538785934448242,
      "step": 91551
    },
    {
      "epoch": 1.6737937927246092e-05,
      "step": 91551,
      "training_step_time": 9.592168807983398
    },
    {
      "epoch": 1.67388916015625e-05,
      "model_forward_time": 1.778913974761963,
      "step": 91552
    },
    {
      "epoch": 1.67388916015625e-05,
      "step": 91552,
      "training_step_time": 7.6322922706604
    },
    {
      "epoch": 1.6739845275878907e-05,
      "model_forward_time": 1.7485144138336182,
      "step": 91553
    },
    {
      "epoch": 1.6739845275878907e-05,
      "step": 91553,
      "training_step_time": 8.11867880821228
    },
    {
      "epoch": 1.674079895019531e-05,
      "model_forward_time": 1.7606768608093262,
      "step": 91554
    },
    {
      "epoch": 1.674079895019531e-05,
      "step": 91554,
      "training_step_time": 7.653024673461914
    },
    {
      "epoch": 1.674175262451172e-05,
      "model_forward_time": 1.7491703033447266,
      "step": 91555
    },
    {
      "epoch": 1.674175262451172e-05,
      "step": 91555,
      "training_step_time": 7.524693727493286
    },
    {
      "epoch": 1.6742706298828126e-05,
      "model_forward_time": 1.7486844062805176,
      "step": 91556
    },
    {
      "epoch": 1.6742706298828126e-05,
      "step": 91556,
      "training_step_time": 9.458552837371826
    },
    {
      "epoch": 1.674365997314453e-05,
      "model_forward_time": 2.8151803016662598,
      "step": 91557
    },
    {
      "epoch": 1.674365997314453e-05,
      "step": 91557,
      "training_step_time": 8.66312551498413
    },
    {
      "epoch": 1.6744613647460937e-05,
      "model_forward_time": 1.7567596435546875,
      "step": 91558
    },
    {
      "epoch": 1.6744613647460937e-05,
      "step": 91558,
      "training_step_time": 8.386867046356201
    },
    {
      "epoch": 1.6745567321777345e-05,
      "model_forward_time": 1.7825095653533936,
      "step": 91559
    },
    {
      "epoch": 1.6745567321777345e-05,
      "step": 91559,
      "training_step_time": 8.571541786193848
    },
    {
      "epoch": 1.674652099609375e-05,
      "grad_norm": 0.1068776473402977,
      "learning_rate": 1.9353456816068373e-07,
      "loss": 0.0559,
      "step": 91560
    },
    {
      "epoch": 1.674652099609375e-05,
      "model_forward_time": 1.7513887882232666,
      "step": 91560
    },
    {
      "dynamics_loss_avg": 0.0628572003915906,
      "epoch": 1.674652099609375e-05,
      "step": 91560
    },
    {
      "action_loss_avg": 0.0032408449857030063,
      "epoch": 1.674652099609375e-05,
      "step": 91560
    },
    {
      "epoch": 1.674652099609375e-05,
      "step": 91560,
      "training_step_time": 7.63806676864624
    },
    {
      "epoch": 1.6747474670410156e-05,
      "model_forward_time": 1.775653600692749,
      "step": 91561
    },
    {
      "epoch": 1.6747474670410156e-05,
      "step": 91561,
      "training_step_time": 7.653628349304199
    },
    {
      "epoch": 1.6748428344726564e-05,
      "model_forward_time": 1.758225917816162,
      "step": 91562
    },
    {
      "epoch": 1.6748428344726564e-05,
      "step": 91562,
      "training_step_time": 7.464760780334473
    },
    {
      "epoch": 1.6749382019042968e-05,
      "model_forward_time": 1.761002540588379,
      "step": 91563
    },
    {
      "epoch": 1.6749382019042968e-05,
      "step": 91563,
      "training_step_time": 7.655264139175415
    },
    {
      "epoch": 1.6750335693359375e-05,
      "model_forward_time": 1.7593986988067627,
      "step": 91564
    },
    {
      "epoch": 1.6750335693359375e-05,
      "step": 91564,
      "training_step_time": 7.552491903305054
    },
    {
      "epoch": 1.6751289367675782e-05,
      "model_forward_time": 1.753389596939087,
      "step": 91565
    },
    {
      "epoch": 1.6751289367675782e-05,
      "step": 91565,
      "training_step_time": 7.553867816925049
    },
    {
      "epoch": 1.6752243041992186e-05,
      "model_forward_time": 1.750579833984375,
      "step": 91566
    },
    {
      "epoch": 1.6752243041992186e-05,
      "step": 91566,
      "training_step_time": 8.194742918014526
    },
    {
      "epoch": 1.6753196716308594e-05,
      "model_forward_time": 1.757596492767334,
      "step": 91567
    },
    {
      "epoch": 1.6753196716308594e-05,
      "step": 91567,
      "training_step_time": 8.46800708770752
    },
    {
      "epoch": 1.6754150390625e-05,
      "model_forward_time": 1.7722702026367188,
      "step": 91568
    },
    {
      "epoch": 1.6754150390625e-05,
      "step": 91568,
      "training_step_time": 9.143279075622559
    },
    {
      "epoch": 1.6755104064941405e-05,
      "model_forward_time": 1.7510395050048828,
      "step": 91569
    },
    {
      "epoch": 1.6755104064941405e-05,
      "step": 91569,
      "training_step_time": 7.68485426902771
    },
    {
      "epoch": 1.6756057739257813e-05,
      "grad_norm": 0.07251109182834625,
      "learning_rate": 1.9307925398106942e-07,
      "loss": 0.057,
      "step": 91570
    },
    {
      "epoch": 1.6756057739257813e-05,
      "model_forward_time": 1.768427848815918,
      "step": 91570
    },
    {
      "dynamics_loss_avg": 0.04959529154002666,
      "epoch": 1.6756057739257813e-05,
      "step": 91570
    },
    {
      "action_loss_avg": 0.0027105445478809997,
      "epoch": 1.6756057739257813e-05,
      "step": 91570
    },
    {
      "epoch": 1.6756057739257813e-05,
      "step": 91570,
      "training_step_time": 8.257629871368408
    },
    {
      "epoch": 1.675701141357422e-05,
      "model_forward_time": 1.7615313529968262,
      "step": 91571
    },
    {
      "epoch": 1.675701141357422e-05,
      "step": 91571,
      "training_step_time": 7.546529054641724
    },
    {
      "epoch": 1.6757965087890624e-05,
      "model_forward_time": 1.761169672012329,
      "step": 91572
    },
    {
      "epoch": 1.6757965087890624e-05,
      "step": 91572,
      "training_step_time": 7.680248975753784
    },
    {
      "epoch": 1.675891876220703e-05,
      "model_forward_time": 1.7530157566070557,
      "step": 91573
    },
    {
      "epoch": 1.675891876220703e-05,
      "step": 91573,
      "training_step_time": 7.551690340042114
    },
    {
      "epoch": 1.675987243652344e-05,
      "model_forward_time": 1.7411417961120605,
      "step": 91574
    },
    {
      "epoch": 1.675987243652344e-05,
      "step": 91574,
      "training_step_time": 7.544375658035278
    },
    {
      "epoch": 1.6760826110839843e-05,
      "model_forward_time": 1.7561769485473633,
      "step": 91575
    },
    {
      "epoch": 1.6760826110839843e-05,
      "step": 91575,
      "training_step_time": 9.959192276000977
    },
    {
      "epoch": 1.676177978515625e-05,
      "model_forward_time": 2.48891282081604,
      "step": 91576
    },
    {
      "epoch": 1.676177978515625e-05,
      "step": 91576,
      "training_step_time": 7.588385343551636
    },
    {
      "epoch": 1.6762733459472658e-05,
      "model_forward_time": 1.7948088645935059,
      "step": 91577
    },
    {
      "epoch": 1.6762733459472658e-05,
      "step": 91577,
      "training_step_time": 8.669996738433838
    },
    {
      "epoch": 1.6763687133789062e-05,
      "model_forward_time": 1.7588984966278076,
      "step": 91578
    },
    {
      "epoch": 1.6763687133789062e-05,
      "step": 91578,
      "training_step_time": 7.6727893352508545
    },
    {
      "epoch": 1.676464080810547e-05,
      "model_forward_time": 1.7583692073822021,
      "step": 91579
    },
    {
      "epoch": 1.676464080810547e-05,
      "step": 91579,
      "training_step_time": 8.226659059524536
    },
    {
      "epoch": 1.6765594482421873e-05,
      "grad_norm": 0.08376773446798325,
      "learning_rate": 1.9262446547907875e-07,
      "loss": 0.0564,
      "step": 91580
    },
    {
      "epoch": 1.6765594482421873e-05,
      "model_forward_time": 1.7618958950042725,
      "step": 91580
    },
    {
      "dynamics_loss_avg": 0.04937223326414823,
      "epoch": 1.6765594482421873e-05,
      "step": 91580
    },
    {
      "action_loss_avg": 0.0021151726105017588,
      "epoch": 1.6765594482421873e-05,
      "step": 91580
    },
    {
      "epoch": 1.6765594482421873e-05,
      "step": 91580,
      "training_step_time": 7.6864540576934814
    },
    {
      "epoch": 1.676654815673828e-05,
      "model_forward_time": 1.7561862468719482,
      "step": 91581
    },
    {
      "epoch": 1.676654815673828e-05,
      "step": 91581,
      "training_step_time": 7.595199108123779
    },
    {
      "epoch": 1.6767501831054688e-05,
      "model_forward_time": 1.7635281085968018,
      "step": 91582
    },
    {
      "epoch": 1.6767501831054688e-05,
      "step": 91582,
      "training_step_time": 7.693132638931274
    },
    {
      "epoch": 1.6768455505371092e-05,
      "model_forward_time": 1.7618286609649658,
      "step": 91583
    },
    {
      "epoch": 1.6768455505371092e-05,
      "step": 91583,
      "training_step_time": 7.592403888702393
    },
    {
      "epoch": 1.67694091796875e-05,
      "model_forward_time": 1.7639365196228027,
      "step": 91584
    },
    {
      "epoch": 1.67694091796875e-05,
      "step": 91584,
      "training_step_time": 9.412199258804321
    },
    {
      "epoch": 1.6770362854003907e-05,
      "model_forward_time": 2.8966751098632812,
      "step": 91585
    },
    {
      "epoch": 1.6770362854003907e-05,
      "step": 91585,
      "training_step_time": 7.609263896942139
    },
    {
      "epoch": 1.677131652832031e-05,
      "model_forward_time": 1.7585382461547852,
      "step": 91586
    },
    {
      "epoch": 1.677131652832031e-05,
      "step": 91586,
      "training_step_time": 8.5569007396698
    },
    {
      "epoch": 1.677227020263672e-05,
      "model_forward_time": 1.7797002792358398,
      "step": 91587
    },
    {
      "epoch": 1.677227020263672e-05,
      "step": 91587,
      "training_step_time": 8.719300031661987
    },
    {
      "epoch": 1.6773223876953126e-05,
      "model_forward_time": 1.7495105266571045,
      "step": 91588
    },
    {
      "epoch": 1.6773223876953126e-05,
      "step": 91588,
      "training_step_time": 7.6555094718933105
    },
    {
      "epoch": 1.677417755126953e-05,
      "model_forward_time": 1.7844359874725342,
      "step": 91589
    },
    {
      "epoch": 1.677417755126953e-05,
      "step": 91589,
      "training_step_time": 9.214423418045044
    },
    {
      "epoch": 1.6775131225585937e-05,
      "grad_norm": 0.08529584854841232,
      "learning_rate": 1.921702027044503e-07,
      "loss": 0.0544,
      "step": 91590
    },
    {
      "epoch": 1.6775131225585937e-05,
      "model_forward_time": 1.7697904109954834,
      "step": 91590
    },
    {
      "dynamics_loss_avg": 0.0559354018419981,
      "epoch": 1.6775131225585937e-05,
      "step": 91590
    },
    {
      "action_loss_avg": 0.002311911166179925,
      "epoch": 1.6775131225585937e-05,
      "step": 91590
    },
    {
      "epoch": 1.6775131225585937e-05,
      "step": 91590,
      "training_step_time": 7.685263156890869
    },
    {
      "epoch": 1.6776084899902345e-05,
      "model_forward_time": 1.7626807689666748,
      "step": 91591
    },
    {
      "epoch": 1.6776084899902345e-05,
      "step": 91591,
      "training_step_time": 7.739670515060425
    },
    {
      "epoch": 1.677703857421875e-05,
      "model_forward_time": 1.764714002609253,
      "step": 91592
    },
    {
      "epoch": 1.677703857421875e-05,
      "step": 91592,
      "training_step_time": 7.718002557754517
    },
    {
      "epoch": 1.6777992248535156e-05,
      "model_forward_time": 1.7642645835876465,
      "step": 91593
    },
    {
      "epoch": 1.6777992248535156e-05,
      "step": 91593,
      "training_step_time": 8.571268558502197
    },
    {
      "epoch": 1.6778945922851564e-05,
      "model_forward_time": 2.8693268299102783,
      "step": 91594
    },
    {
      "epoch": 1.6778945922851564e-05,
      "step": 91594,
      "training_step_time": 7.712323904037476
    },
    {
      "epoch": 1.6779899597167968e-05,
      "model_forward_time": 1.7746741771697998,
      "step": 91595
    },
    {
      "epoch": 1.6779899597167968e-05,
      "step": 91595,
      "training_step_time": 8.309220314025879
    },
    {
      "epoch": 1.6780853271484375e-05,
      "model_forward_time": 1.7832331657409668,
      "step": 91596
    },
    {
      "epoch": 1.6780853271484375e-05,
      "step": 91596,
      "training_step_time": 8.63949465751648
    },
    {
      "epoch": 1.6781806945800782e-05,
      "model_forward_time": 1.7530395984649658,
      "step": 91597
    },
    {
      "epoch": 1.6781806945800782e-05,
      "step": 91597,
      "training_step_time": 7.7949676513671875
    },
    {
      "epoch": 1.6782760620117187e-05,
      "model_forward_time": 1.7796926498413086,
      "step": 91598
    },
    {
      "epoch": 1.6782760620117187e-05,
      "step": 91598,
      "training_step_time": 7.757525444030762
    },
    {
      "epoch": 1.6783714294433594e-05,
      "model_forward_time": 1.7660400867462158,
      "step": 91599
    },
    {
      "epoch": 1.6783714294433594e-05,
      "step": 91599,
      "training_step_time": 7.640681505203247
    },
    {
      "epoch": 1.678466796875e-05,
      "grad_norm": 0.0934675857424736,
      "learning_rate": 1.917164657068593e-07,
      "loss": 0.0554,
      "step": 91600
    },
    {
      "epoch": 1.678466796875e-05,
      "model_forward_time": 1.754716157913208,
      "step": 91600
    },
    {
      "dynamics_loss_avg": 0.06781111843883991,
      "epoch": 1.678466796875e-05,
      "step": 91600
    },
    {
      "action_loss_avg": 0.003258768568048254,
      "epoch": 1.678466796875e-05,
      "step": 91600
    },
    {
      "epoch": 1.678466796875e-05,
      "step": 91600,
      "training_step_time": 7.709681510925293
    },
    {
      "epoch": 1.6785621643066405e-05,
      "model_forward_time": 1.762974739074707,
      "step": 91601
    },
    {
      "epoch": 1.6785621643066405e-05,
      "step": 91601,
      "training_step_time": 7.743159055709839
    },
    {
      "epoch": 1.6786575317382813e-05,
      "model_forward_time": 1.7586843967437744,
      "step": 91602
    },
    {
      "epoch": 1.6786575317382813e-05,
      "step": 91602,
      "training_step_time": 8.54387879371643
    },
    {
      "epoch": 1.678752899169922e-05,
      "model_forward_time": 2.8124489784240723,
      "step": 91603
    },
    {
      "epoch": 1.678752899169922e-05,
      "step": 91603,
      "training_step_time": 7.659736633300781
    },
    {
      "epoch": 1.6788482666015624e-05,
      "model_forward_time": 1.7678439617156982,
      "step": 91604
    },
    {
      "epoch": 1.6788482666015624e-05,
      "step": 91604,
      "training_step_time": 8.586145877838135
    },
    {
      "epoch": 1.678943634033203e-05,
      "model_forward_time": 1.7860662937164307,
      "step": 91605
    },
    {
      "epoch": 1.678943634033203e-05,
      "step": 91605,
      "training_step_time": 8.818219900131226
    },
    {
      "epoch": 1.679039001464844e-05,
      "model_forward_time": 1.7486495971679688,
      "step": 91606
    },
    {
      "epoch": 1.679039001464844e-05,
      "step": 91606,
      "training_step_time": 9.519341945648193
    },
    {
      "epoch": 1.6791343688964843e-05,
      "model_forward_time": 1.784635066986084,
      "step": 91607
    },
    {
      "epoch": 1.6791343688964843e-05,
      "step": 91607,
      "training_step_time": 8.515851497650146
    },
    {
      "epoch": 1.679229736328125e-05,
      "model_forward_time": 1.7600665092468262,
      "step": 91608
    },
    {
      "epoch": 1.679229736328125e-05,
      "step": 91608,
      "training_step_time": 8.7270827293396
    },
    {
      "epoch": 1.6793251037597658e-05,
      "model_forward_time": 1.7570765018463135,
      "step": 91609
    },
    {
      "epoch": 1.6793251037597658e-05,
      "step": 91609,
      "training_step_time": 7.657219171524048
    },
    {
      "epoch": 1.6794204711914062e-05,
      "grad_norm": 0.5420230031013489,
      "learning_rate": 1.912632545359272e-07,
      "loss": 0.0554,
      "step": 91610
    },
    {
      "epoch": 1.6794204711914062e-05,
      "model_forward_time": 1.7577979564666748,
      "step": 91610
    },
    {
      "dynamics_loss_avg": 0.06055493168532848,
      "epoch": 1.6794204711914062e-05,
      "step": 91610
    },
    {
      "action_loss_avg": 0.0017515115265268832,
      "epoch": 1.6794204711914062e-05,
      "step": 91610
    },
    {
      "epoch": 1.6794204711914062e-05,
      "step": 91610,
      "training_step_time": 7.69314980506897
    },
    {
      "epoch": 1.679515838623047e-05,
      "model_forward_time": 1.765554428100586,
      "step": 91611
    },
    {
      "epoch": 1.679515838623047e-05,
      "step": 91611,
      "training_step_time": 8.67016315460205
    },
    {
      "epoch": 1.6796112060546873e-05,
      "model_forward_time": 2.8006489276885986,
      "step": 91612
    },
    {
      "epoch": 1.6796112060546873e-05,
      "step": 91612,
      "training_step_time": 7.77812123298645
    },
    {
      "epoch": 1.679706573486328e-05,
      "model_forward_time": 1.7596590518951416,
      "step": 91613
    },
    {
      "epoch": 1.679706573486328e-05,
      "step": 91613,
      "training_step_time": 7.7419819831848145
    },
    {
      "epoch": 1.6798019409179688e-05,
      "model_forward_time": 1.781238317489624,
      "step": 91614
    },
    {
      "epoch": 1.6798019409179688e-05,
      "step": 91614,
      "training_step_time": 7.669504880905151
    },
    {
      "epoch": 1.6798973083496092e-05,
      "model_forward_time": 1.7585818767547607,
      "step": 91615
    },
    {
      "epoch": 1.6798973083496092e-05,
      "step": 91615,
      "training_step_time": 7.720348119735718
    },
    {
      "epoch": 1.67999267578125e-05,
      "model_forward_time": 1.7789161205291748,
      "step": 91616
    },
    {
      "epoch": 1.67999267578125e-05,
      "step": 91616,
      "training_step_time": 8.424309015274048
    },
    {
      "epoch": 1.6800880432128907e-05,
      "model_forward_time": 1.764878749847412,
      "step": 91617
    },
    {
      "epoch": 1.6800880432128907e-05,
      "step": 91617,
      "training_step_time": 7.7452311515808105
    },
    {
      "epoch": 1.680183410644531e-05,
      "model_forward_time": 1.7609474658966064,
      "step": 91618
    },
    {
      "epoch": 1.680183410644531e-05,
      "step": 91618,
      "training_step_time": 8.287290573120117
    },
    {
      "epoch": 1.680278778076172e-05,
      "model_forward_time": 1.762075662612915,
      "step": 91619
    },
    {
      "epoch": 1.680278778076172e-05,
      "step": 91619,
      "training_step_time": 8.593704462051392
    },
    {
      "epoch": 1.6803741455078126e-05,
      "grad_norm": 0.07156281918287277,
      "learning_rate": 1.908105692412149e-07,
      "loss": 0.0558,
      "step": 91620
    },
    {
      "epoch": 1.6803741455078126e-05,
      "model_forward_time": 1.7601232528686523,
      "step": 91620
    },
    {
      "dynamics_loss_avg": 0.056378275714814664,
      "epoch": 1.6803741455078126e-05,
      "step": 91620
    },
    {
      "action_loss_avg": 0.003611287218518555,
      "epoch": 1.6803741455078126e-05,
      "step": 91620
    },
    {
      "epoch": 1.6803741455078126e-05,
      "step": 91620,
      "training_step_time": 8.676347970962524
    },
    {
      "epoch": 1.680469512939453e-05,
      "model_forward_time": 1.7616701126098633,
      "step": 91621
    },
    {
      "epoch": 1.680469512939453e-05,
      "step": 91621,
      "training_step_time": 7.737938642501831
    },
    {
      "epoch": 1.6805648803710937e-05,
      "model_forward_time": 2.8924973011016846,
      "step": 91622
    },
    {
      "epoch": 1.6805648803710937e-05,
      "step": 91622,
      "training_step_time": 8.448498249053955
    },
    {
      "epoch": 1.6806602478027345e-05,
      "model_forward_time": 1.7839932441711426,
      "step": 91623
    },
    {
      "epoch": 1.6806602478027345e-05,
      "step": 91623,
      "training_step_time": 9.900811910629272
    },
    {
      "epoch": 1.680755615234375e-05,
      "model_forward_time": 1.7625055313110352,
      "step": 91624
    },
    {
      "epoch": 1.680755615234375e-05,
      "step": 91624,
      "training_step_time": 8.489318132400513
    },
    {
      "epoch": 1.6808509826660156e-05,
      "model_forward_time": 1.7797985076904297,
      "step": 91625
    },
    {
      "epoch": 1.6808509826660156e-05,
      "step": 91625,
      "training_step_time": 8.948354244232178
    },
    {
      "epoch": 1.6809463500976564e-05,
      "model_forward_time": 1.7595136165618896,
      "step": 91626
    },
    {
      "epoch": 1.6809463500976564e-05,
      "step": 91626,
      "training_step_time": 7.857199192047119
    },
    {
      "epoch": 1.6810417175292968e-05,
      "model_forward_time": 1.7686433792114258,
      "step": 91627
    },
    {
      "epoch": 1.6810417175292968e-05,
      "step": 91627,
      "training_step_time": 8.293216466903687
    },
    {
      "epoch": 1.6811370849609375e-05,
      "model_forward_time": 1.7590222358703613,
      "step": 91628
    },
    {
      "epoch": 1.6811370849609375e-05,
      "step": 91628,
      "training_step_time": 7.794304847717285
    },
    {
      "epoch": 1.6812324523925783e-05,
      "model_forward_time": 1.7640101909637451,
      "step": 91629
    },
    {
      "epoch": 1.6812324523925783e-05,
      "step": 91629,
      "training_step_time": 7.697148323059082
    },
    {
      "epoch": 1.6813278198242187e-05,
      "grad_norm": 0.07868928462266922,
      "learning_rate": 1.9035840987222886e-07,
      "loss": 0.0565,
      "step": 91630
    },
    {
      "epoch": 1.6813278198242187e-05,
      "model_forward_time": 1.7579472064971924,
      "step": 91630
    },
    {
      "dynamics_loss_avg": 0.058530278876423834,
      "epoch": 1.6813278198242187e-05,
      "step": 91630
    },
    {
      "action_loss_avg": 0.0037889208761043847,
      "epoch": 1.6813278198242187e-05,
      "step": 91630
    },
    {
      "epoch": 1.6813278198242187e-05,
      "step": 91630,
      "training_step_time": 7.803835391998291
    },
    {
      "epoch": 1.6814231872558594e-05,
      "model_forward_time": 2.856983184814453,
      "step": 91631
    },
    {
      "epoch": 1.6814231872558594e-05,
      "step": 91631,
      "training_step_time": 7.69566011428833
    },
    {
      "epoch": 1.6815185546875e-05,
      "model_forward_time": 1.7674925327301025,
      "step": 91632
    },
    {
      "epoch": 1.6815185546875e-05,
      "step": 91632,
      "training_step_time": 7.760430335998535
    },
    {
      "epoch": 1.6816139221191405e-05,
      "model_forward_time": 1.7663633823394775,
      "step": 91633
    },
    {
      "epoch": 1.6816139221191405e-05,
      "step": 91633,
      "training_step_time": 9.31904125213623
    },
    {
      "epoch": 1.6817092895507813e-05,
      "model_forward_time": 1.760740041732788,
      "step": 91634
    },
    {
      "epoch": 1.6817092895507813e-05,
      "step": 91634,
      "training_step_time": 8.172386646270752
    },
    {
      "epoch": 1.681804656982422e-05,
      "model_forward_time": 1.76389479637146,
      "step": 91635
    },
    {
      "epoch": 1.681804656982422e-05,
      "step": 91635,
      "training_step_time": 9.605275869369507
    },
    {
      "epoch": 1.6819000244140624e-05,
      "model_forward_time": 1.7673344612121582,
      "step": 91636
    },
    {
      "epoch": 1.6819000244140624e-05,
      "step": 91636,
      "training_step_time": 8.18515157699585
    },
    {
      "epoch": 1.681995391845703e-05,
      "model_forward_time": 1.7450063228607178,
      "step": 91637
    },
    {
      "epoch": 1.681995391845703e-05,
      "step": 91637,
      "training_step_time": 8.558687210083008
    },
    {
      "epoch": 1.682090759277344e-05,
      "model_forward_time": 1.7501509189605713,
      "step": 91638
    },
    {
      "epoch": 1.682090759277344e-05,
      "step": 91638,
      "training_step_time": 7.671328783035278
    },
    {
      "epoch": 1.6821861267089843e-05,
      "model_forward_time": 1.7630300521850586,
      "step": 91639
    },
    {
      "epoch": 1.6821861267089843e-05,
      "step": 91639,
      "training_step_time": 8.881479024887085
    },
    {
      "epoch": 1.682281494140625e-05,
      "grad_norm": 0.0899977758526802,
      "learning_rate": 1.8990677647841516e-07,
      "loss": 0.0567,
      "step": 91640
    },
    {
      "epoch": 1.682281494140625e-05,
      "model_forward_time": 2.8458988666534424,
      "step": 91640
    },
    {
      "dynamics_loss_avg": 0.06016010269522667,
      "epoch": 1.682281494140625e-05,
      "step": 91640
    },
    {
      "action_loss_avg": 0.0025386001158040016,
      "epoch": 1.682281494140625e-05,
      "step": 91640
    },
    {
      "epoch": 1.682281494140625e-05,
      "step": 91640,
      "training_step_time": 7.736017465591431
    },
    {
      "epoch": 1.6823768615722655e-05,
      "model_forward_time": 1.7594935894012451,
      "step": 91641
    },
    {
      "epoch": 1.6823768615722655e-05,
      "step": 91641,
      "training_step_time": 10.695850610733032
    },
    {
      "epoch": 1.6824722290039062e-05,
      "model_forward_time": 1.7617793083190918,
      "step": 91642
    },
    {
      "epoch": 1.6824722290039062e-05,
      "step": 91642,
      "training_step_time": 7.726799011230469
    },
    {
      "epoch": 1.682567596435547e-05,
      "model_forward_time": 1.750396728515625,
      "step": 91643
    },
    {
      "epoch": 1.682567596435547e-05,
      "step": 91643,
      "training_step_time": 7.76350736618042
    },
    {
      "epoch": 1.6826629638671873e-05,
      "model_forward_time": 1.7832343578338623,
      "step": 91644
    },
    {
      "epoch": 1.6826629638671873e-05,
      "step": 91644,
      "training_step_time": 7.756498575210571
    },
    {
      "epoch": 1.682758331298828e-05,
      "model_forward_time": 1.7637014389038086,
      "step": 91645
    },
    {
      "epoch": 1.682758331298828e-05,
      "step": 91645,
      "training_step_time": 7.776340007781982
    },
    {
      "epoch": 1.6828536987304688e-05,
      "model_forward_time": 1.756892204284668,
      "step": 91646
    },
    {
      "epoch": 1.6828536987304688e-05,
      "step": 91646,
      "training_step_time": 7.860074520111084
    },
    {
      "epoch": 1.6829490661621092e-05,
      "model_forward_time": 1.757885217666626,
      "step": 91647
    },
    {
      "epoch": 1.6829490661621092e-05,
      "step": 91647,
      "training_step_time": 7.862421989440918
    },
    {
      "epoch": 1.68304443359375e-05,
      "model_forward_time": 1.763329267501831,
      "step": 91648
    },
    {
      "epoch": 1.68304443359375e-05,
      "step": 91648,
      "training_step_time": 8.280216455459595
    },
    {
      "epoch": 1.6831398010253907e-05,
      "model_forward_time": 2.860273838043213,
      "step": 91649
    },
    {
      "epoch": 1.6831398010253907e-05,
      "step": 91649,
      "training_step_time": 8.455723524093628
    },
    {
      "epoch": 1.683235168457031e-05,
      "grad_norm": 0.06538496911525726,
      "learning_rate": 1.8945566910916536e-07,
      "loss": 0.0562,
      "step": 91650
    },
    {
      "epoch": 1.683235168457031e-05,
      "model_forward_time": 1.746567726135254,
      "step": 91650
    },
    {
      "dynamics_loss_avg": 0.041362355649471286,
      "epoch": 1.683235168457031e-05,
      "step": 91650
    },
    {
      "action_loss_avg": 0.0012879872039775365,
      "epoch": 1.683235168457031e-05,
      "step": 91650
    },
    {
      "epoch": 1.683235168457031e-05,
      "step": 91650,
      "training_step_time": 10.515173435211182
    },
    {
      "epoch": 1.683330535888672e-05,
      "model_forward_time": 1.7788009643554688,
      "step": 91651
    },
    {
      "epoch": 1.683330535888672e-05,
      "step": 91651,
      "training_step_time": 7.689584255218506
    },
    {
      "epoch": 1.6834259033203126e-05,
      "model_forward_time": 1.377129077911377,
      "step": 91652
    },
    {
      "epoch": 1.6834259033203126e-05,
      "step": 91652,
      "training_step_time": 9.522179365158081
    },
    {
      "epoch": 1.683521270751953e-05,
      "model_forward_time": 1.7855582237243652,
      "step": 91653
    },
    {
      "epoch": 1.683521270751953e-05,
      "step": 91653,
      "training_step_time": 7.8187255859375
    },
    {
      "epoch": 1.6836166381835937e-05,
      "model_forward_time": 1.7536005973815918,
      "step": 91654
    },
    {
      "epoch": 1.6836166381835937e-05,
      "step": 91654,
      "training_step_time": 7.778395891189575
    },
    {
      "epoch": 1.6837120056152345e-05,
      "model_forward_time": 1.7648990154266357,
      "step": 91655
    },
    {
      "epoch": 1.6837120056152345e-05,
      "step": 91655,
      "training_step_time": 8.622028112411499
    },
    {
      "epoch": 1.683807373046875e-05,
      "model_forward_time": 1.7514867782592773,
      "step": 91656
    },
    {
      "epoch": 1.683807373046875e-05,
      "step": 91656,
      "training_step_time": 8.613458395004272
    },
    {
      "epoch": 1.6839027404785156e-05,
      "model_forward_time": 1.750483751296997,
      "step": 91657
    },
    {
      "epoch": 1.6839027404785156e-05,
      "step": 91657,
      "training_step_time": 9.355158805847168
    },
    {
      "epoch": 1.6839981079101564e-05,
      "model_forward_time": 2.7936348915100098,
      "step": 91658
    },
    {
      "epoch": 1.6839981079101564e-05,
      "step": 91658,
      "training_step_time": 7.678835391998291
    },
    {
      "epoch": 1.6840934753417968e-05,
      "model_forward_time": 1.7589061260223389,
      "step": 91659
    },
    {
      "epoch": 1.6840934753417968e-05,
      "step": 91659,
      "training_step_time": 8.749647855758667
    },
    {
      "epoch": 1.6841888427734375e-05,
      "grad_norm": 0.10082127898931503,
      "learning_rate": 1.8900508781381056e-07,
      "loss": 0.0562,
      "step": 91660
    },
    {
      "epoch": 1.6841888427734375e-05,
      "model_forward_time": 1.7721507549285889,
      "step": 91660
    },
    {
      "dynamics_loss_avg": 0.04954295493662357,
      "epoch": 1.6841888427734375e-05,
      "step": 91660
    },
    {
      "action_loss_avg": 0.00191389144747518,
      "epoch": 1.6841888427734375e-05,
      "step": 91660
    },
    {
      "epoch": 1.6841888427734375e-05,
      "step": 91660,
      "training_step_time": 7.615478992462158
    },
    {
      "epoch": 1.6842842102050783e-05,
      "model_forward_time": 1.7568771839141846,
      "step": 91661
    },
    {
      "epoch": 1.6842842102050783e-05,
      "step": 91661,
      "training_step_time": 7.708960771560669
    },
    {
      "epoch": 1.6843795776367187e-05,
      "model_forward_time": 1.7594871520996094,
      "step": 91662
    },
    {
      "epoch": 1.6843795776367187e-05,
      "step": 91662,
      "training_step_time": 7.652396202087402
    },
    {
      "epoch": 1.6844749450683594e-05,
      "model_forward_time": 1.7535889148712158,
      "step": 91663
    },
    {
      "epoch": 1.6844749450683594e-05,
      "step": 91663,
      "training_step_time": 7.706782817840576
    },
    {
      "epoch": 1.6845703125e-05,
      "model_forward_time": 1.7507567405700684,
      "step": 91664
    },
    {
      "epoch": 1.6845703125e-05,
      "step": 91664,
      "training_step_time": 7.642037391662598
    },
    {
      "epoch": 1.6846656799316405e-05,
      "model_forward_time": 1.756976842880249,
      "step": 91665
    },
    {
      "epoch": 1.6846656799316405e-05,
      "step": 91665,
      "training_step_time": 8.67720103263855
    },
    {
      "epoch": 1.6847610473632813e-05,
      "model_forward_time": 1.7527737617492676,
      "step": 91666
    },
    {
      "epoch": 1.6847610473632813e-05,
      "step": 91666,
      "training_step_time": 7.796545028686523
    },
    {
      "epoch": 1.684856414794922e-05,
      "model_forward_time": 1.7390577793121338,
      "step": 91667
    },
    {
      "epoch": 1.684856414794922e-05,
      "step": 91667,
      "training_step_time": 7.669376850128174
    },
    {
      "epoch": 1.6849517822265624e-05,
      "model_forward_time": 2.833686590194702,
      "step": 91668
    },
    {
      "epoch": 1.6849517822265624e-05,
      "step": 91668,
      "training_step_time": 9.291912078857422
    },
    {
      "epoch": 1.6850471496582032e-05,
      "model_forward_time": 1.7683393955230713,
      "step": 91669
    },
    {
      "epoch": 1.6850471496582032e-05,
      "step": 91669,
      "training_step_time": 7.732699394226074
    },
    {
      "epoch": 1.685142517089844e-05,
      "grad_norm": 0.08898873627185822,
      "learning_rate": 1.8855503264162634e-07,
      "loss": 0.0557,
      "step": 91670
    },
    {
      "epoch": 1.685142517089844e-05,
      "model_forward_time": 1.752413034439087,
      "step": 91670
    },
    {
      "dynamics_loss_avg": 0.05213724449276924,
      "epoch": 1.685142517089844e-05,
      "step": 91670
    },
    {
      "action_loss_avg": 0.0020990168413845822,
      "epoch": 1.685142517089844e-05,
      "step": 91670
    },
    {
      "epoch": 1.685142517089844e-05,
      "step": 91670,
      "training_step_time": 9.193711042404175
    },
    {
      "epoch": 1.6852378845214843e-05,
      "model_forward_time": 1.758573055267334,
      "step": 91671
    },
    {
      "epoch": 1.6852378845214843e-05,
      "step": 91671,
      "training_step_time": 8.170474767684937
    },
    {
      "epoch": 1.685333251953125e-05,
      "model_forward_time": 1.7546136379241943,
      "step": 91672
    },
    {
      "epoch": 1.685333251953125e-05,
      "step": 91672,
      "training_step_time": 8.705815315246582
    },
    {
      "epoch": 1.6854286193847655e-05,
      "model_forward_time": 1.7591824531555176,
      "step": 91673
    },
    {
      "epoch": 1.6854286193847655e-05,
      "step": 91673,
      "training_step_time": 7.590469121932983
    },
    {
      "epoch": 1.6855239868164062e-05,
      "model_forward_time": 1.7537546157836914,
      "step": 91674
    },
    {
      "epoch": 1.6855239868164062e-05,
      "step": 91674,
      "training_step_time": 7.697501182556152
    },
    {
      "epoch": 1.685619354248047e-05,
      "model_forward_time": 1.7588105201721191,
      "step": 91675
    },
    {
      "epoch": 1.685619354248047e-05,
      "step": 91675,
      "training_step_time": 8.676664590835571
    },
    {
      "epoch": 1.6857147216796873e-05,
      "model_forward_time": 1.761113166809082,
      "step": 91676
    },
    {
      "epoch": 1.6857147216796873e-05,
      "step": 91676,
      "training_step_time": 8.197059869766235
    },
    {
      "epoch": 1.685810089111328e-05,
      "model_forward_time": 2.879873752593994,
      "step": 91677
    },
    {
      "epoch": 1.685810089111328e-05,
      "step": 91677,
      "training_step_time": 9.166493892669678
    },
    {
      "epoch": 1.6859054565429688e-05,
      "model_forward_time": 1.7693066596984863,
      "step": 91678
    },
    {
      "epoch": 1.6859054565429688e-05,
      "step": 91678,
      "training_step_time": 7.66755223274231
    },
    {
      "epoch": 1.6860008239746092e-05,
      "model_forward_time": 1.7467238903045654,
      "step": 91679
    },
    {
      "epoch": 1.6860008239746092e-05,
      "step": 91679,
      "training_step_time": 9.457630634307861
    },
    {
      "epoch": 1.68609619140625e-05,
      "grad_norm": 0.08562979102134705,
      "learning_rate": 1.8810550364183e-07,
      "loss": 0.0568,
      "step": 91680
    },
    {
      "epoch": 1.68609619140625e-05,
      "model_forward_time": 1.7454884052276611,
      "step": 91680
    },
    {
      "dynamics_loss_avg": 0.05794761907309294,
      "epoch": 1.68609619140625e-05,
      "step": 91680
    },
    {
      "action_loss_avg": 0.002878394100116566,
      "epoch": 1.68609619140625e-05,
      "step": 91680
    },
    {
      "epoch": 1.68609619140625e-05,
      "step": 91680,
      "training_step_time": 7.627131223678589
    },
    {
      "epoch": 1.6861915588378907e-05,
      "model_forward_time": 1.777420997619629,
      "step": 91681
    },
    {
      "epoch": 1.6861915588378907e-05,
      "step": 91681,
      "training_step_time": 7.632286787033081
    },
    {
      "epoch": 1.686286926269531e-05,
      "model_forward_time": 1.7556850910186768,
      "step": 91682
    },
    {
      "epoch": 1.686286926269531e-05,
      "step": 91682,
      "training_step_time": 8.306212663650513
    },
    {
      "epoch": 1.686382293701172e-05,
      "model_forward_time": 1.7496271133422852,
      "step": 91683
    },
    {
      "epoch": 1.686382293701172e-05,
      "step": 91683,
      "training_step_time": 7.8226377964019775
    },
    {
      "epoch": 1.6864776611328126e-05,
      "model_forward_time": 1.7606136798858643,
      "step": 91684
    },
    {
      "epoch": 1.6864776611328126e-05,
      "step": 91684,
      "training_step_time": 8.596110820770264
    },
    {
      "epoch": 1.686573028564453e-05,
      "model_forward_time": 1.753389596939087,
      "step": 91685
    },
    {
      "epoch": 1.686573028564453e-05,
      "step": 91685,
      "training_step_time": 7.683605670928955
    },
    {
      "epoch": 1.6866683959960937e-05,
      "model_forward_time": 2.8345799446105957,
      "step": 91686
    },
    {
      "epoch": 1.6866683959960937e-05,
      "step": 91686,
      "training_step_time": 7.681681394577026
    },
    {
      "epoch": 1.6867637634277345e-05,
      "model_forward_time": 1.764719009399414,
      "step": 91687
    },
    {
      "epoch": 1.6867637634277345e-05,
      "step": 91687,
      "training_step_time": 7.63598895072937
    },
    {
      "epoch": 1.686859130859375e-05,
      "model_forward_time": 1.7785372734069824,
      "step": 91688
    },
    {
      "epoch": 1.686859130859375e-05,
      "step": 91688,
      "training_step_time": 8.414310932159424
    },
    {
      "epoch": 1.6869544982910156e-05,
      "model_forward_time": 1.7476344108581543,
      "step": 91689
    },
    {
      "epoch": 1.6869544982910156e-05,
      "step": 91689,
      "training_step_time": 8.421094417572021
    },
    {
      "epoch": 1.6870498657226564e-05,
      "grad_norm": 0.08164769411087036,
      "learning_rate": 1.8765650086358112e-07,
      "loss": 0.0552,
      "step": 91690
    },
    {
      "epoch": 1.6870498657226564e-05,
      "model_forward_time": 1.7687172889709473,
      "step": 91690
    },
    {
      "dynamics_loss_avg": 0.05738959610462189,
      "epoch": 1.6870498657226564e-05,
      "step": 91690
    },
    {
      "action_loss_avg": 0.0017093355301767588,
      "epoch": 1.6870498657226564e-05,
      "step": 91690
    },
    {
      "epoch": 1.6870498657226564e-05,
      "step": 91690,
      "training_step_time": 8.742962837219238
    },
    {
      "epoch": 1.6871452331542968e-05,
      "model_forward_time": 1.7538666725158691,
      "step": 91691
    },
    {
      "epoch": 1.6871452331542968e-05,
      "step": 91691,
      "training_step_time": 8.306348323822021
    },
    {
      "epoch": 1.6872406005859375e-05,
      "model_forward_time": 1.7575602531433105,
      "step": 91692
    },
    {
      "epoch": 1.6872406005859375e-05,
      "step": 91692,
      "training_step_time": 7.751829624176025
    },
    {
      "epoch": 1.6873359680175783e-05,
      "model_forward_time": 1.7386703491210938,
      "step": 91693
    },
    {
      "epoch": 1.6873359680175783e-05,
      "step": 91693,
      "training_step_time": 8.720616579055786
    },
    {
      "epoch": 1.6874313354492187e-05,
      "model_forward_time": 1.7542719841003418,
      "step": 91694
    },
    {
      "epoch": 1.6874313354492187e-05,
      "step": 91694,
      "training_step_time": 9.214735269546509
    },
    {
      "epoch": 1.6875267028808594e-05,
      "model_forward_time": 2.8147401809692383,
      "step": 91695
    },
    {
      "epoch": 1.6875267028808594e-05,
      "step": 91695,
      "training_step_time": 8.92132019996643
    },
    {
      "epoch": 1.6876220703125e-05,
      "model_forward_time": 1.7569794654846191,
      "step": 91696
    },
    {
      "epoch": 1.6876220703125e-05,
      "step": 91696,
      "training_step_time": 8.276917934417725
    },
    {
      "epoch": 1.6877174377441405e-05,
      "model_forward_time": 1.7738730907440186,
      "step": 91697
    },
    {
      "epoch": 1.6877174377441405e-05,
      "step": 91697,
      "training_step_time": 8.782787322998047
    },
    {
      "epoch": 1.6878128051757813e-05,
      "model_forward_time": 1.7369916439056396,
      "step": 91698
    },
    {
      "epoch": 1.6878128051757813e-05,
      "step": 91698,
      "training_step_time": 9.363007307052612
    },
    {
      "epoch": 1.687908172607422e-05,
      "model_forward_time": 1.7702150344848633,
      "step": 91699
    },
    {
      "epoch": 1.687908172607422e-05,
      "step": 91699,
      "training_step_time": 9.158436298370361
    },
    {
      "epoch": 1.6880035400390624e-05,
      "grad_norm": 0.07795275002717972,
      "learning_rate": 1.872080243559815e-07,
      "loss": 0.0571,
      "step": 91700
    },
    {
      "epoch": 1.6880035400390624e-05,
      "model_forward_time": 1.7517321109771729,
      "step": 91700
    },
    {
      "dynamics_loss_avg": 0.0533774078823626,
      "epoch": 1.6880035400390624e-05,
      "step": 91700
    },
    {
      "action_loss_avg": 0.002201669219357427,
      "epoch": 1.6880035400390624e-05,
      "step": 91700
    },
    {
      "epoch": 1.6880035400390624e-05,
      "step": 91700,
      "training_step_time": 7.690867185592651
    },
    {
      "epoch": 1.6880989074707032e-05,
      "model_forward_time": 1.7551429271697998,
      "step": 91701
    },
    {
      "epoch": 1.6880989074707032e-05,
      "step": 91701,
      "training_step_time": 9.571289777755737
    },
    {
      "epoch": 1.688194274902344e-05,
      "model_forward_time": 1.750640630722046,
      "step": 91702
    },
    {
      "epoch": 1.688194274902344e-05,
      "step": 91702,
      "training_step_time": 8.463698625564575
    },
    {
      "epoch": 1.6882896423339843e-05,
      "model_forward_time": 1.7538304328918457,
      "step": 91703
    },
    {
      "epoch": 1.6882896423339843e-05,
      "step": 91703,
      "training_step_time": 8.250375032424927
    },
    {
      "epoch": 1.688385009765625e-05,
      "model_forward_time": 2.8212833404541016,
      "step": 91704
    },
    {
      "epoch": 1.688385009765625e-05,
      "step": 91704,
      "training_step_time": 7.666132926940918
    },
    {
      "epoch": 1.6884803771972655e-05,
      "model_forward_time": 1.7525603771209717,
      "step": 91705
    },
    {
      "epoch": 1.6884803771972655e-05,
      "step": 91705,
      "training_step_time": 9.384247541427612
    },
    {
      "epoch": 1.6885757446289062e-05,
      "model_forward_time": 1.7790181636810303,
      "step": 91706
    },
    {
      "epoch": 1.6885757446289062e-05,
      "step": 91706,
      "training_step_time": 7.6967549324035645
    },
    {
      "epoch": 1.688671112060547e-05,
      "model_forward_time": 1.75239896774292,
      "step": 91707
    },
    {
      "epoch": 1.688671112060547e-05,
      "step": 91707,
      "training_step_time": 7.667648553848267
    },
    {
      "epoch": 1.6887664794921873e-05,
      "model_forward_time": 1.7726740837097168,
      "step": 91708
    },
    {
      "epoch": 1.6887664794921873e-05,
      "step": 91708,
      "training_step_time": 7.6671693325042725
    },
    {
      "epoch": 1.688861846923828e-05,
      "model_forward_time": 1.7461771965026855,
      "step": 91709
    },
    {
      "epoch": 1.688861846923828e-05,
      "step": 91709,
      "training_step_time": 7.670520305633545
    },
    {
      "epoch": 1.688957214355469e-05,
      "grad_norm": 0.10945640504360199,
      "learning_rate": 1.8676007416807695e-07,
      "loss": 0.0566,
      "step": 91710
    },
    {
      "epoch": 1.688957214355469e-05,
      "model_forward_time": 1.7404115200042725,
      "step": 91710
    },
    {
      "dynamics_loss_avg": 0.05589661262929439,
      "epoch": 1.688957214355469e-05,
      "step": 91710
    },
    {
      "action_loss_avg": 0.0033435462508350612,
      "epoch": 1.688957214355469e-05,
      "step": 91710
    },
    {
      "epoch": 1.688957214355469e-05,
      "step": 91710,
      "training_step_time": 7.602392911911011
    },
    {
      "epoch": 1.6890525817871092e-05,
      "model_forward_time": 1.7466275691986084,
      "step": 91711
    },
    {
      "epoch": 1.6890525817871092e-05,
      "step": 91711,
      "training_step_time": 7.6852498054504395
    },
    {
      "epoch": 1.68914794921875e-05,
      "model_forward_time": 1.7620751857757568,
      "step": 91712
    },
    {
      "epoch": 1.68914794921875e-05,
      "step": 91712,
      "training_step_time": 8.614747285842896
    },
    {
      "epoch": 1.6892433166503907e-05,
      "model_forward_time": 1.7600336074829102,
      "step": 91713
    },
    {
      "epoch": 1.6892433166503907e-05,
      "step": 91713,
      "training_step_time": 7.670220613479614
    },
    {
      "epoch": 1.689338684082031e-05,
      "model_forward_time": 2.8649911880493164,
      "step": 91714
    },
    {
      "epoch": 1.689338684082031e-05,
      "step": 91714,
      "training_step_time": 8.792406558990479
    },
    {
      "epoch": 1.689434051513672e-05,
      "model_forward_time": 1.7534308433532715,
      "step": 91715
    },
    {
      "epoch": 1.689434051513672e-05,
      "step": 91715,
      "training_step_time": 9.025403261184692
    },
    {
      "epoch": 1.6895294189453126e-05,
      "model_forward_time": 1.7505605220794678,
      "step": 91716
    },
    {
      "epoch": 1.6895294189453126e-05,
      "step": 91716,
      "training_step_time": 7.6137049198150635
    },
    {
      "epoch": 1.689624786376953e-05,
      "model_forward_time": 1.739614486694336,
      "step": 91717
    },
    {
      "epoch": 1.689624786376953e-05,
      "step": 91717,
      "training_step_time": 7.649993896484375
    },
    {
      "epoch": 1.6897201538085937e-05,
      "model_forward_time": 1.7549631595611572,
      "step": 91718
    },
    {
      "epoch": 1.6897201538085937e-05,
      "step": 91718,
      "training_step_time": 7.676809072494507
    },
    {
      "epoch": 1.6898155212402345e-05,
      "model_forward_time": 1.7508141994476318,
      "step": 91719
    },
    {
      "epoch": 1.6898155212402345e-05,
      "step": 91719,
      "training_step_time": 7.709259033203125
    },
    {
      "epoch": 1.689910888671875e-05,
      "grad_norm": 0.06715715676546097,
      "learning_rate": 1.8631265034885437e-07,
      "loss": 0.0551,
      "step": 91720
    },
    {
      "epoch": 1.689910888671875e-05,
      "model_forward_time": 1.7386021614074707,
      "step": 91720
    },
    {
      "dynamics_loss_avg": 0.05435052644461393,
      "epoch": 1.689910888671875e-05,
      "step": 91720
    },
    {
      "action_loss_avg": 0.0026592782989609985,
      "epoch": 1.689910888671875e-05,
      "step": 91720
    },
    {
      "epoch": 1.689910888671875e-05,
      "step": 91720,
      "training_step_time": 7.781721115112305
    },
    {
      "epoch": 1.6900062561035156e-05,
      "model_forward_time": 1.753126621246338,
      "step": 91721
    },
    {
      "epoch": 1.6900062561035156e-05,
      "step": 91721,
      "training_step_time": 8.465136051177979
    },
    {
      "epoch": 1.6901016235351564e-05,
      "model_forward_time": 1.7460575103759766,
      "step": 91722
    },
    {
      "epoch": 1.6901016235351564e-05,
      "step": 91722,
      "training_step_time": 7.637225866317749
    },
    {
      "epoch": 1.6901969909667968e-05,
      "model_forward_time": 2.8882627487182617,
      "step": 91723
    },
    {
      "epoch": 1.6901969909667968e-05,
      "step": 91723,
      "training_step_time": 7.693835258483887
    },
    {
      "epoch": 1.6902923583984375e-05,
      "model_forward_time": 1.7484214305877686,
      "step": 91724
    },
    {
      "epoch": 1.6902923583984375e-05,
      "step": 91724,
      "training_step_time": 9.643990516662598
    },
    {
      "epoch": 1.6903877258300783e-05,
      "model_forward_time": 1.7705104351043701,
      "step": 91725
    },
    {
      "epoch": 1.6903877258300783e-05,
      "step": 91725,
      "training_step_time": 7.80632758140564
    },
    {
      "epoch": 1.6904830932617187e-05,
      "model_forward_time": 1.733903408050537,
      "step": 91726
    },
    {
      "epoch": 1.6904830932617187e-05,
      "step": 91726,
      "training_step_time": 8.242912769317627
    },
    {
      "epoch": 1.6905784606933594e-05,
      "model_forward_time": 1.7838525772094727,
      "step": 91727
    },
    {
      "epoch": 1.6905784606933594e-05,
      "step": 91727,
      "training_step_time": 8.904696464538574
    },
    {
      "epoch": 1.690673828125e-05,
      "model_forward_time": 1.753730297088623,
      "step": 91728
    },
    {
      "epoch": 1.690673828125e-05,
      "step": 91728,
      "training_step_time": 8.295584917068481
    },
    {
      "epoch": 1.6907691955566405e-05,
      "model_forward_time": 1.7586076259613037,
      "step": 91729
    },
    {
      "epoch": 1.6907691955566405e-05,
      "step": 91729,
      "training_step_time": 8.25732159614563
    },
    {
      "epoch": 1.6908645629882813e-05,
      "grad_norm": 0.10011705756187439,
      "learning_rate": 1.858657529472424e-07,
      "loss": 0.0561,
      "step": 91730
    },
    {
      "epoch": 1.6908645629882813e-05,
      "model_forward_time": 1.7722456455230713,
      "step": 91730
    },
    {
      "dynamics_loss_avg": 0.07136694192886353,
      "epoch": 1.6908645629882813e-05,
      "step": 91730
    },
    {
      "action_loss_avg": 0.0029525830294005575,
      "epoch": 1.6908645629882813e-05,
      "step": 91730
    },
    {
      "epoch": 1.6908645629882813e-05,
      "step": 91730,
      "training_step_time": 8.623717784881592
    },
    {
      "epoch": 1.690959930419922e-05,
      "model_forward_time": 1.7662386894226074,
      "step": 91731
    },
    {
      "epoch": 1.690959930419922e-05,
      "step": 91731,
      "training_step_time": 7.75104546546936
    },
    {
      "epoch": 1.6910552978515624e-05,
      "model_forward_time": 2.4092013835906982,
      "step": 91732
    },
    {
      "epoch": 1.6910552978515624e-05,
      "step": 91732,
      "training_step_time": 7.6568427085876465
    },
    {
      "epoch": 1.6911506652832032e-05,
      "model_forward_time": 1.7697691917419434,
      "step": 91733
    },
    {
      "epoch": 1.6911506652832032e-05,
      "step": 91733,
      "training_step_time": 8.60224723815918
    },
    {
      "epoch": 1.691246032714844e-05,
      "model_forward_time": 1.7656340599060059,
      "step": 91734
    },
    {
      "epoch": 1.691246032714844e-05,
      "step": 91734,
      "training_step_time": 7.8361382484436035
    },
    {
      "epoch": 1.6913414001464843e-05,
      "model_forward_time": 1.846083402633667,
      "step": 91735
    },
    {
      "epoch": 1.6913414001464843e-05,
      "step": 91735,
      "training_step_time": 8.457475423812866
    },
    {
      "epoch": 1.691436767578125e-05,
      "model_forward_time": 1.7632012367248535,
      "step": 91736
    },
    {
      "epoch": 1.691436767578125e-05,
      "step": 91736,
      "training_step_time": 7.737859725952148
    },
    {
      "epoch": 1.6915321350097655e-05,
      "model_forward_time": 1.766284704208374,
      "step": 91737
    },
    {
      "epoch": 1.6915321350097655e-05,
      "step": 91737,
      "training_step_time": 7.764885902404785
    },
    {
      "epoch": 1.6916275024414062e-05,
      "model_forward_time": 1.7828714847564697,
      "step": 91738
    },
    {
      "epoch": 1.6916275024414062e-05,
      "step": 91738,
      "training_step_time": 8.781415462493896
    },
    {
      "epoch": 1.691722869873047e-05,
      "model_forward_time": 1.7732491493225098,
      "step": 91739
    },
    {
      "epoch": 1.691722869873047e-05,
      "step": 91739,
      "training_step_time": 7.801892518997192
    },
    {
      "epoch": 1.6918182373046874e-05,
      "grad_norm": 0.0826353058218956,
      "learning_rate": 1.8541938201211473e-07,
      "loss": 0.0535,
      "step": 91740
    },
    {
      "epoch": 1.6918182373046874e-05,
      "model_forward_time": 1.7678072452545166,
      "step": 91740
    },
    {
      "dynamics_loss_avg": 0.07398668620735407,
      "epoch": 1.6918182373046874e-05,
      "step": 91740
    },
    {
      "action_loss_avg": 0.004391385888447985,
      "epoch": 1.6918182373046874e-05,
      "step": 91740
    },
    {
      "epoch": 1.6918182373046874e-05,
      "step": 91740,
      "training_step_time": 7.715287446975708
    },
    {
      "epoch": 1.691913604736328e-05,
      "model_forward_time": 2.438220500946045,
      "step": 91741
    },
    {
      "epoch": 1.691913604736328e-05,
      "step": 91741,
      "training_step_time": 8.319838523864746
    },
    {
      "epoch": 1.692008972167969e-05,
      "model_forward_time": 1.7684245109558105,
      "step": 91742
    },
    {
      "epoch": 1.692008972167969e-05,
      "step": 91742,
      "training_step_time": 7.6341753005981445
    },
    {
      "epoch": 1.6921043395996092e-05,
      "model_forward_time": 1.766066551208496,
      "step": 91743
    },
    {
      "epoch": 1.6921043395996092e-05,
      "step": 91743,
      "training_step_time": 8.40087366104126
    },
    {
      "epoch": 1.69219970703125e-05,
      "model_forward_time": 1.810793161392212,
      "step": 91744
    },
    {
      "epoch": 1.69219970703125e-05,
      "step": 91744,
      "training_step_time": 7.848509788513184
    },
    {
      "epoch": 1.6922950744628907e-05,
      "model_forward_time": 1.7540805339813232,
      "step": 91745
    },
    {
      "epoch": 1.6922950744628907e-05,
      "step": 91745,
      "training_step_time": 7.645781755447388
    },
    {
      "epoch": 1.692390441894531e-05,
      "model_forward_time": 1.763282060623169,
      "step": 91746
    },
    {
      "epoch": 1.692390441894531e-05,
      "step": 91746,
      "training_step_time": 7.705555438995361
    },
    {
      "epoch": 1.692485809326172e-05,
      "model_forward_time": 1.8032803535461426,
      "step": 91747
    },
    {
      "epoch": 1.692485809326172e-05,
      "step": 91747,
      "training_step_time": 7.748333930969238
    },
    {
      "epoch": 1.6925811767578126e-05,
      "model_forward_time": 1.7665846347808838,
      "step": 91748
    },
    {
      "epoch": 1.6925811767578126e-05,
      "step": 91748,
      "training_step_time": 7.795510530471802
    },
    {
      "epoch": 1.692676544189453e-05,
      "model_forward_time": 1.7659316062927246,
      "step": 91749
    },
    {
      "epoch": 1.692676544189453e-05,
      "step": 91749,
      "training_step_time": 7.761547088623047
    },
    {
      "epoch": 1.6927719116210938e-05,
      "grad_norm": 0.09119907766580582,
      "learning_rate": 1.84973537592284e-07,
      "loss": 0.0542,
      "step": 91750
    },
    {
      "epoch": 1.6927719116210938e-05,
      "model_forward_time": 2.410684585571289,
      "step": 91750
    },
    {
      "dynamics_loss_avg": 0.041019643843173983,
      "epoch": 1.6927719116210938e-05,
      "step": 91750
    },
    {
      "action_loss_avg": 0.0016553727793507278,
      "epoch": 1.6927719116210938e-05,
      "step": 91750
    },
    {
      "epoch": 1.6927719116210938e-05,
      "step": 91750,
      "training_step_time": 7.717234134674072
    },
    {
      "epoch": 1.6928672790527345e-05,
      "model_forward_time": 1.7693397998809814,
      "step": 91751
    },
    {
      "epoch": 1.6928672790527345e-05,
      "step": 91751,
      "training_step_time": 8.360246896743774
    },
    {
      "epoch": 1.692962646484375e-05,
      "model_forward_time": 1.3942041397094727,
      "step": 91752
    },
    {
      "epoch": 1.692962646484375e-05,
      "step": 91752,
      "training_step_time": 9.23895001411438
    },
    {
      "epoch": 1.6930580139160156e-05,
      "model_forward_time": 1.7687172889709473,
      "step": 91753
    },
    {
      "epoch": 1.6930580139160156e-05,
      "step": 91753,
      "training_step_time": 7.777033805847168
    },
    {
      "epoch": 1.6931533813476564e-05,
      "model_forward_time": 1.835230827331543,
      "step": 91754
    },
    {
      "epoch": 1.6931533813476564e-05,
      "step": 91754,
      "training_step_time": 7.736205577850342
    },
    {
      "epoch": 1.6932487487792968e-05,
      "model_forward_time": 1.760632038116455,
      "step": 91755
    },
    {
      "epoch": 1.6932487487792968e-05,
      "step": 91755,
      "training_step_time": 7.790967226028442
    },
    {
      "epoch": 1.6933441162109375e-05,
      "model_forward_time": 1.770899772644043,
      "step": 91756
    },
    {
      "epoch": 1.6933441162109375e-05,
      "step": 91756,
      "training_step_time": 7.777278661727905
    },
    {
      "epoch": 1.6934394836425783e-05,
      "model_forward_time": 1.764918327331543,
      "step": 91757
    },
    {
      "epoch": 1.6934394836425783e-05,
      "step": 91757,
      "training_step_time": 7.735677719116211
    },
    {
      "epoch": 1.6935348510742187e-05,
      "model_forward_time": 1.7590205669403076,
      "step": 91758
    },
    {
      "epoch": 1.6935348510742187e-05,
      "step": 91758,
      "training_step_time": 7.773677349090576
    },
    {
      "epoch": 1.6936302185058594e-05,
      "model_forward_time": 1.7681288719177246,
      "step": 91759
    },
    {
      "epoch": 1.6936302185058594e-05,
      "step": 91759,
      "training_step_time": 7.692774057388306
    },
    {
      "epoch": 1.6937255859375e-05,
      "grad_norm": 0.17369408905506134,
      "learning_rate": 1.8452821973650835e-07,
      "loss": 0.0541,
      "step": 91760
    },
    {
      "epoch": 1.6937255859375e-05,
      "model_forward_time": 2.4207873344421387,
      "step": 91760
    },
    {
      "dynamics_loss_avg": 0.057425770256668326,
      "epoch": 1.6937255859375e-05,
      "step": 91760
    },
    {
      "action_loss_avg": 0.0025862353431875817,
      "epoch": 1.6937255859375e-05,
      "step": 91760
    },
    {
      "epoch": 1.6937255859375e-05,
      "step": 91760,
      "training_step_time": 7.664605379104614
    },
    {
      "epoch": 1.6938209533691406e-05,
      "model_forward_time": 1.774160385131836,
      "step": 91761
    },
    {
      "epoch": 1.6938209533691406e-05,
      "step": 91761,
      "training_step_time": 9.201224565505981
    },
    {
      "epoch": 1.6939163208007813e-05,
      "model_forward_time": 1.7890233993530273,
      "step": 91762
    },
    {
      "epoch": 1.6939163208007813e-05,
      "step": 91762,
      "training_step_time": 8.829421758651733
    },
    {
      "epoch": 1.694011688232422e-05,
      "model_forward_time": 1.824488878250122,
      "step": 91763
    },
    {
      "epoch": 1.694011688232422e-05,
      "step": 91763,
      "training_step_time": 7.680947780609131
    },
    {
      "epoch": 1.6941070556640624e-05,
      "model_forward_time": 1.7642991542816162,
      "step": 91764
    },
    {
      "epoch": 1.6941070556640624e-05,
      "step": 91764,
      "training_step_time": 7.678358554840088
    },
    {
      "epoch": 1.6942024230957032e-05,
      "model_forward_time": 1.7733666896820068,
      "step": 91765
    },
    {
      "epoch": 1.6942024230957032e-05,
      "step": 91765,
      "training_step_time": 7.773271322250366
    },
    {
      "epoch": 1.6942977905273436e-05,
      "model_forward_time": 1.7703840732574463,
      "step": 91766
    },
    {
      "epoch": 1.6942977905273436e-05,
      "step": 91766,
      "training_step_time": 7.7655298709869385
    },
    {
      "epoch": 1.6943931579589843e-05,
      "model_forward_time": 1.7696812152862549,
      "step": 91767
    },
    {
      "epoch": 1.6943931579589843e-05,
      "step": 91767,
      "training_step_time": 7.817159414291382
    },
    {
      "epoch": 1.694488525390625e-05,
      "model_forward_time": 1.7676022052764893,
      "step": 91768
    },
    {
      "epoch": 1.694488525390625e-05,
      "step": 91768,
      "training_step_time": 7.649505853652954
    },
    {
      "epoch": 1.6945838928222655e-05,
      "model_forward_time": 2.0166401863098145,
      "step": 91769
    },
    {
      "epoch": 1.6945838928222655e-05,
      "step": 91769,
      "training_step_time": 9.276003360748291
    },
    {
      "epoch": 1.6946792602539062e-05,
      "grad_norm": 0.09207528829574585,
      "learning_rate": 1.840834284934856e-07,
      "loss": 0.0551,
      "step": 91770
    },
    {
      "epoch": 1.6946792602539062e-05,
      "model_forward_time": 1.7692821025848389,
      "step": 91770
    },
    {
      "dynamics_loss_avg": 0.06523902006447316,
      "epoch": 1.6946792602539062e-05,
      "step": 91770
    },
    {
      "action_loss_avg": 0.0022203570726560427,
      "epoch": 1.6946792602539062e-05,
      "step": 91770
    },
    {
      "epoch": 1.6946792602539062e-05,
      "step": 91770,
      "training_step_time": 8.382838249206543
    },
    {
      "epoch": 1.694774627685547e-05,
      "model_forward_time": 1.7771470546722412,
      "step": 91771
    },
    {
      "epoch": 1.694774627685547e-05,
      "step": 91771,
      "training_step_time": 7.6908860206604
    },
    {
      "epoch": 1.6948699951171874e-05,
      "model_forward_time": 1.840508222579956,
      "step": 91772
    },
    {
      "epoch": 1.6948699951171874e-05,
      "step": 91772,
      "training_step_time": 7.641409635543823
    },
    {
      "epoch": 1.694965362548828e-05,
      "model_forward_time": 1.7635018825531006,
      "step": 91773
    },
    {
      "epoch": 1.694965362548828e-05,
      "step": 91773,
      "training_step_time": 8.73486328125
    },
    {
      "epoch": 1.695060729980469e-05,
      "model_forward_time": 1.780045747756958,
      "step": 91774
    },
    {
      "epoch": 1.695060729980469e-05,
      "step": 91774,
      "training_step_time": 7.657781362533569
    },
    {
      "epoch": 1.6951560974121092e-05,
      "model_forward_time": 1.7624561786651611,
      "step": 91775
    },
    {
      "epoch": 1.6951560974121092e-05,
      "step": 91775,
      "training_step_time": 8.599264144897461
    },
    {
      "epoch": 1.69525146484375e-05,
      "model_forward_time": 1.7733354568481445,
      "step": 91776
    },
    {
      "epoch": 1.69525146484375e-05,
      "step": 91776,
      "training_step_time": 8.83305549621582
    },
    {
      "epoch": 1.6953468322753907e-05,
      "model_forward_time": 1.818751573562622,
      "step": 91777
    },
    {
      "epoch": 1.6953468322753907e-05,
      "step": 91777,
      "training_step_time": 7.576124668121338
    },
    {
      "epoch": 1.695442199707031e-05,
      "model_forward_time": 2.428191661834717,
      "step": 91778
    },
    {
      "epoch": 1.695442199707031e-05,
      "step": 91778,
      "training_step_time": 8.231882572174072
    },
    {
      "epoch": 1.695537567138672e-05,
      "model_forward_time": 1.7683441638946533,
      "step": 91779
    },
    {
      "epoch": 1.695537567138672e-05,
      "step": 91779,
      "training_step_time": 8.189911127090454
    },
    {
      "epoch": 1.6956329345703126e-05,
      "grad_norm": 0.09462049603462219,
      "learning_rate": 1.8363916391185953e-07,
      "loss": 0.0557,
      "step": 91780
    },
    {
      "epoch": 1.6956329345703126e-05,
      "model_forward_time": 1.7691676616668701,
      "step": 91780
    },
    {
      "dynamics_loss_avg": 0.06662469133734703,
      "epoch": 1.6956329345703126e-05,
      "step": 91780
    },
    {
      "action_loss_avg": 0.001623617095174268,
      "epoch": 1.6956329345703126e-05,
      "step": 91780
    },
    {
      "epoch": 1.6956329345703126e-05,
      "step": 91780,
      "training_step_time": 9.083250761032104
    },
    {
      "epoch": 1.695728302001953e-05,
      "model_forward_time": 1.7681305408477783,
      "step": 91781
    },
    {
      "epoch": 1.695728302001953e-05,
      "step": 91781,
      "training_step_time": 9.787106037139893
    },
    {
      "epoch": 1.6958236694335938e-05,
      "model_forward_time": 1.8284356594085693,
      "step": 91782
    },
    {
      "epoch": 1.6958236694335938e-05,
      "step": 91782,
      "training_step_time": 8.402805089950562
    },
    {
      "epoch": 1.6959190368652345e-05,
      "model_forward_time": 1.7689552307128906,
      "step": 91783
    },
    {
      "epoch": 1.6959190368652345e-05,
      "step": 91783,
      "training_step_time": 7.804591655731201
    },
    {
      "epoch": 1.696014404296875e-05,
      "model_forward_time": 1.8029944896697998,
      "step": 91784
    },
    {
      "epoch": 1.696014404296875e-05,
      "step": 91784,
      "training_step_time": 7.8230602741241455
    },
    {
      "epoch": 1.6961097717285156e-05,
      "model_forward_time": 1.7660436630249023,
      "step": 91785
    },
    {
      "epoch": 1.6961097717285156e-05,
      "step": 91785,
      "training_step_time": 9.018452405929565
    },
    {
      "epoch": 1.6962051391601564e-05,
      "model_forward_time": 1.76479172706604,
      "step": 91786
    },
    {
      "epoch": 1.6962051391601564e-05,
      "step": 91786,
      "training_step_time": 8.269474029541016
    },
    {
      "epoch": 1.6963005065917968e-05,
      "model_forward_time": 2.3978464603424072,
      "step": 91787
    },
    {
      "epoch": 1.6963005065917968e-05,
      "step": 91787,
      "training_step_time": 8.9419264793396
    },
    {
      "epoch": 1.6963958740234375e-05,
      "model_forward_time": 1.7652778625488281,
      "step": 91788
    },
    {
      "epoch": 1.6963958740234375e-05,
      "step": 91788,
      "training_step_time": 7.720513343811035
    },
    {
      "epoch": 1.6964912414550783e-05,
      "model_forward_time": 1.8075363636016846,
      "step": 91789
    },
    {
      "epoch": 1.6964912414550783e-05,
      "step": 91789,
      "training_step_time": 8.63858699798584
    },
    {
      "epoch": 1.6965866088867187e-05,
      "grad_norm": 0.12671294808387756,
      "learning_rate": 1.8319542604021244e-07,
      "loss": 0.0548,
      "step": 91790
    },
    {
      "epoch": 1.6965866088867187e-05,
      "model_forward_time": 1.76462721824646,
      "step": 91790
    },
    {
      "dynamics_loss_avg": 0.06412254888564348,
      "epoch": 1.6965866088867187e-05,
      "step": 91790
    },
    {
      "action_loss_avg": 0.002092707098927349,
      "epoch": 1.6965866088867187e-05,
      "step": 91790
    },
    {
      "epoch": 1.6965866088867187e-05,
      "step": 91790,
      "training_step_time": 9.106386184692383
    },
    {
      "epoch": 1.6966819763183594e-05,
      "model_forward_time": 1.8270440101623535,
      "step": 91791
    },
    {
      "epoch": 1.6966819763183594e-05,
      "step": 91791,
      "training_step_time": 8.622957229614258
    },
    {
      "epoch": 1.69677734375e-05,
      "model_forward_time": 1.7691078186035156,
      "step": 91792
    },
    {
      "epoch": 1.69677734375e-05,
      "step": 91792,
      "training_step_time": 7.727287530899048
    },
    {
      "epoch": 1.6968727111816406e-05,
      "model_forward_time": 1.7728221416473389,
      "step": 91793
    },
    {
      "epoch": 1.6968727111816406e-05,
      "step": 91793,
      "training_step_time": 8.462772130966187
    },
    {
      "epoch": 1.6969680786132813e-05,
      "model_forward_time": 1.7855782508850098,
      "step": 91794
    },
    {
      "epoch": 1.6969680786132813e-05,
      "step": 91794,
      "training_step_time": 9.106053352355957
    },
    {
      "epoch": 1.697063446044922e-05,
      "model_forward_time": 1.7695910930633545,
      "step": 91795
    },
    {
      "epoch": 1.697063446044922e-05,
      "step": 91795,
      "training_step_time": 7.833184480667114
    },
    {
      "epoch": 1.6971588134765624e-05,
      "model_forward_time": 1.7774791717529297,
      "step": 91796
    },
    {
      "epoch": 1.6971588134765624e-05,
      "step": 91796,
      "training_step_time": 8.7636079788208
    },
    {
      "epoch": 1.6972541809082032e-05,
      "model_forward_time": 1.7694146633148193,
      "step": 91797
    },
    {
      "epoch": 1.6972541809082032e-05,
      "step": 91797,
      "training_step_time": 9.378840208053589
    },
    {
      "epoch": 1.6973495483398436e-05,
      "model_forward_time": 1.7754340171813965,
      "step": 91798
    },
    {
      "epoch": 1.6973495483398436e-05,
      "step": 91798,
      "training_step_time": 7.786117792129517
    },
    {
      "epoch": 1.6974449157714843e-05,
      "model_forward_time": 1.793010950088501,
      "step": 91799
    },
    {
      "epoch": 1.6974449157714843e-05,
      "step": 91799,
      "training_step_time": 7.7983598709106445
    },
    {
      "epoch": 1.697540283203125e-05,
      "grad_norm": 0.07493198662996292,
      "learning_rate": 1.8275221492707275e-07,
      "loss": 0.0546,
      "step": 91800
    },
    {
      "epoch": 1.697540283203125e-05,
      "model_forward_time": 1.8254690170288086,
      "step": 91800
    },
    {
      "dynamics_loss_avg": 0.04674750529229641,
      "epoch": 1.697540283203125e-05,
      "step": 91800
    },
    {
      "action_loss_avg": 0.0016119563937536442,
      "epoch": 1.697540283203125e-05,
      "step": 91800
    },
    {
      "epoch": 1.697540283203125e-05,
      "step": 91800,
      "training_step_time": 7.688073396682739
    },
    {
      "epoch": 1.6976356506347655e-05,
      "model_forward_time": 1.7648999691009521,
      "step": 91801
    },
    {
      "epoch": 1.6976356506347655e-05,
      "step": 91801,
      "training_step_time": 7.677811861038208
    },
    {
      "epoch": 1.6977310180664062e-05,
      "model_forward_time": 1.7743408679962158,
      "step": 91802
    },
    {
      "epoch": 1.6977310180664062e-05,
      "step": 91802,
      "training_step_time": 7.808820724487305
    },
    {
      "epoch": 1.697826385498047e-05,
      "model_forward_time": 1.7897615432739258,
      "step": 91803
    },
    {
      "epoch": 1.697826385498047e-05,
      "step": 91803,
      "training_step_time": 9.062649250030518
    },
    {
      "epoch": 1.6979217529296874e-05,
      "model_forward_time": 1.7852699756622314,
      "step": 91804
    },
    {
      "epoch": 1.6979217529296874e-05,
      "step": 91804,
      "training_step_time": 7.760054588317871
    },
    {
      "epoch": 1.698017120361328e-05,
      "model_forward_time": 1.7856230735778809,
      "step": 91805
    },
    {
      "epoch": 1.698017120361328e-05,
      "step": 91805,
      "training_step_time": 7.675907135009766
    },
    {
      "epoch": 1.698112487792969e-05,
      "model_forward_time": 2.41156005859375,
      "step": 91806
    },
    {
      "epoch": 1.698112487792969e-05,
      "step": 91806,
      "training_step_time": 7.600437641143799
    },
    {
      "epoch": 1.6982078552246092e-05,
      "model_forward_time": 1.8117902278900146,
      "step": 91807
    },
    {
      "epoch": 1.6982078552246092e-05,
      "step": 91807,
      "training_step_time": 8.535916090011597
    },
    {
      "epoch": 1.69830322265625e-05,
      "model_forward_time": 1.7819271087646484,
      "step": 91808
    },
    {
      "epoch": 1.69830322265625e-05,
      "step": 91808,
      "training_step_time": 9.37366008758545
    },
    {
      "epoch": 1.6983985900878907e-05,
      "model_forward_time": 1.7767987251281738,
      "step": 91809
    },
    {
      "epoch": 1.6983985900878907e-05,
      "step": 91809,
      "training_step_time": 7.740063190460205
    },
    {
      "epoch": 1.698493957519531e-05,
      "grad_norm": 0.07831898331642151,
      "learning_rate": 1.8230953062090716e-07,
      "loss": 0.0544,
      "step": 91810
    },
    {
      "epoch": 1.698493957519531e-05,
      "model_forward_time": 1.846161127090454,
      "step": 91810
    },
    {
      "dynamics_loss_avg": 0.05883848052471876,
      "epoch": 1.698493957519531e-05,
      "step": 91810
    },
    {
      "action_loss_avg": 0.002574662584811449,
      "epoch": 1.698493957519531e-05,
      "step": 91810
    },
    {
      "epoch": 1.698493957519531e-05,
      "step": 91810,
      "training_step_time": 8.26069188117981
    },
    {
      "epoch": 1.698589324951172e-05,
      "model_forward_time": 1.787550687789917,
      "step": 91811
    },
    {
      "epoch": 1.698589324951172e-05,
      "step": 91811,
      "training_step_time": 7.821536302566528
    },
    {
      "epoch": 1.6986846923828126e-05,
      "model_forward_time": 1.7775788307189941,
      "step": 91812
    },
    {
      "epoch": 1.6986846923828126e-05,
      "step": 91812,
      "training_step_time": 10.229772329330444
    },
    {
      "epoch": 1.698780059814453e-05,
      "model_forward_time": 1.7751877307891846,
      "step": 91813
    },
    {
      "epoch": 1.698780059814453e-05,
      "step": 91813,
      "training_step_time": 7.820125341415405
    },
    {
      "epoch": 1.6988754272460938e-05,
      "model_forward_time": 1.7809019088745117,
      "step": 91814
    },
    {
      "epoch": 1.6988754272460938e-05,
      "step": 91814,
      "training_step_time": 7.7657470703125
    },
    {
      "epoch": 1.6989707946777345e-05,
      "model_forward_time": 2.429826498031616,
      "step": 91815
    },
    {
      "epoch": 1.6989707946777345e-05,
      "step": 91815,
      "training_step_time": 7.67124605178833
    },
    {
      "epoch": 1.699066162109375e-05,
      "model_forward_time": 1.7767646312713623,
      "step": 91816
    },
    {
      "epoch": 1.699066162109375e-05,
      "step": 91816,
      "training_step_time": 7.6311962604522705
    },
    {
      "epoch": 1.6991615295410156e-05,
      "model_forward_time": 1.7812318801879883,
      "step": 91817
    },
    {
      "epoch": 1.6991615295410156e-05,
      "step": 91817,
      "training_step_time": 8.774897336959839
    },
    {
      "epoch": 1.6992568969726564e-05,
      "model_forward_time": 1.778259038925171,
      "step": 91818
    },
    {
      "epoch": 1.6992568969726564e-05,
      "step": 91818,
      "training_step_time": 7.722812652587891
    },
    {
      "epoch": 1.6993522644042968e-05,
      "model_forward_time": 1.8586070537567139,
      "step": 91819
    },
    {
      "epoch": 1.6993522644042968e-05,
      "step": 91819,
      "training_step_time": 11.693525314331055
    },
    {
      "epoch": 1.6994476318359375e-05,
      "grad_norm": 0.06685797870159149,
      "learning_rate": 1.8186737317012927e-07,
      "loss": 0.0563,
      "step": 91820
    },
    {
      "epoch": 1.6994476318359375e-05,
      "model_forward_time": 1.7875306606292725,
      "step": 91820
    },
    {
      "dynamics_loss_avg": 0.0578162370249629,
      "epoch": 1.6994476318359375e-05,
      "step": 91820
    },
    {
      "action_loss_avg": 0.002657632960472256,
      "epoch": 1.6994476318359375e-05,
      "step": 91820
    },
    {
      "epoch": 1.6994476318359375e-05,
      "step": 91820,
      "training_step_time": 7.321964263916016
    },
    {
      "epoch": 1.6995429992675783e-05,
      "model_forward_time": 1.775956153869629,
      "step": 91821
    },
    {
      "epoch": 1.6995429992675783e-05,
      "step": 91821,
      "training_step_time": 7.931799650192261
    },
    {
      "epoch": 1.6996383666992187e-05,
      "model_forward_time": 1.7756178379058838,
      "step": 91822
    },
    {
      "epoch": 1.6996383666992187e-05,
      "step": 91822,
      "training_step_time": 8.527336120605469
    },
    {
      "epoch": 1.6997337341308594e-05,
      "model_forward_time": 1.8303816318511963,
      "step": 91823
    },
    {
      "epoch": 1.6997337341308594e-05,
      "step": 91823,
      "training_step_time": 7.923377513885498
    },
    {
      "epoch": 1.6998291015625e-05,
      "model_forward_time": 1.7858967781066895,
      "step": 91824
    },
    {
      "epoch": 1.6998291015625e-05,
      "step": 91824,
      "training_step_time": 6.870226860046387
    },
    {
      "epoch": 1.6999244689941406e-05,
      "model_forward_time": 1.7847471237182617,
      "step": 91825
    },
    {
      "epoch": 1.6999244689941406e-05,
      "step": 91825,
      "training_step_time": 6.809004306793213
    },
    {
      "epoch": 1.7000198364257813e-05,
      "model_forward_time": 1.7820372581481934,
      "step": 91826
    },
    {
      "epoch": 1.7000198364257813e-05,
      "step": 91826,
      "training_step_time": 6.834198474884033
    },
    {
      "epoch": 1.700115203857422e-05,
      "model_forward_time": 1.7789409160614014,
      "step": 91827
    },
    {
      "epoch": 1.700115203857422e-05,
      "step": 91827,
      "training_step_time": 6.928467273712158
    },
    {
      "epoch": 1.7002105712890624e-05,
      "model_forward_time": 1.7882790565490723,
      "step": 91828
    },
    {
      "epoch": 1.7002105712890624e-05,
      "step": 91828,
      "training_step_time": 6.8690924644470215
    },
    {
      "epoch": 1.7003059387207032e-05,
      "model_forward_time": 2.455188751220703,
      "step": 91829
    },
    {
      "epoch": 1.7003059387207032e-05,
      "step": 91829,
      "training_step_time": 7.828411102294922
    },
    {
      "epoch": 1.7004013061523436e-05,
      "grad_norm": 0.08964008837938309,
      "learning_rate": 1.8142574262308976e-07,
      "loss": 0.0557,
      "step": 91830
    },
    {
      "epoch": 1.7004013061523436e-05,
      "model_forward_time": 1.7760546207427979,
      "step": 91830
    },
    {
      "dynamics_loss_avg": 0.06941037923097611,
      "epoch": 1.7004013061523436e-05,
      "step": 91830
    },
    {
      "action_loss_avg": 0.0022559234959771855,
      "epoch": 1.7004013061523436e-05,
      "step": 91830
    },
    {
      "epoch": 1.7004013061523436e-05,
      "step": 91830,
      "training_step_time": 6.840503215789795
    },
    {
      "epoch": 1.7004966735839843e-05,
      "model_forward_time": 1.776268482208252,
      "step": 91831
    },
    {
      "epoch": 1.7004966735839843e-05,
      "step": 91831,
      "training_step_time": 7.424952030181885
    },
    {
      "epoch": 1.700592041015625e-05,
      "model_forward_time": 1.78177809715271,
      "step": 91832
    },
    {
      "epoch": 1.700592041015625e-05,
      "step": 91832,
      "training_step_time": 7.504277944564819
    },
    {
      "epoch": 1.7006874084472655e-05,
      "model_forward_time": 1.7877731323242188,
      "step": 91833
    },
    {
      "epoch": 1.7006874084472655e-05,
      "step": 91833,
      "training_step_time": 7.604700565338135
    },
    {
      "epoch": 1.7007827758789062e-05,
      "model_forward_time": 1.7792410850524902,
      "step": 91834
    },
    {
      "epoch": 1.7007827758789062e-05,
      "step": 91834,
      "training_step_time": 6.842702150344849
    },
    {
      "epoch": 1.700878143310547e-05,
      "model_forward_time": 1.7859163284301758,
      "step": 91835
    },
    {
      "epoch": 1.700878143310547e-05,
      "step": 91835,
      "training_step_time": 6.882041931152344
    },
    {
      "epoch": 1.7009735107421874e-05,
      "model_forward_time": 1.7776014804840088,
      "step": 91836
    },
    {
      "epoch": 1.7009735107421874e-05,
      "step": 91836,
      "training_step_time": 6.898283958435059
    },
    {
      "epoch": 1.701068878173828e-05,
      "model_forward_time": 1.7682278156280518,
      "step": 91837
    },
    {
      "epoch": 1.701068878173828e-05,
      "step": 91837,
      "training_step_time": 6.8115599155426025
    },
    {
      "epoch": 1.701164245605469e-05,
      "model_forward_time": 2.4408366680145264,
      "step": 91838
    },
    {
      "epoch": 1.701164245605469e-05,
      "step": 91838,
      "training_step_time": 7.824823617935181
    },
    {
      "epoch": 1.7012596130371092e-05,
      "model_forward_time": 1.7857303619384766,
      "step": 91839
    },
    {
      "epoch": 1.7012596130371092e-05,
      "step": 91839,
      "training_step_time": 6.813812732696533
    },
    {
      "epoch": 1.70135498046875e-05,
      "grad_norm": 0.10341200232505798,
      "learning_rate": 1.8098463902808728e-07,
      "loss": 0.0543,
      "step": 91840
    },
    {
      "epoch": 1.70135498046875e-05,
      "model_forward_time": 1.7752130031585693,
      "step": 91840
    },
    {
      "dynamics_loss_avg": 0.06595235280692577,
      "epoch": 1.70135498046875e-05,
      "step": 91840
    },
    {
      "action_loss_avg": 0.0016341541660949588,
      "epoch": 1.70135498046875e-05,
      "step": 91840
    },
    {
      "epoch": 1.70135498046875e-05,
      "step": 91840,
      "training_step_time": 7.590310573577881
    },
    {
      "epoch": 1.7014503479003907e-05,
      "model_forward_time": 1.779200553894043,
      "step": 91841
    },
    {
      "epoch": 1.7014503479003907e-05,
      "step": 91841,
      "training_step_time": 7.5333521366119385
    },
    {
      "epoch": 1.701545715332031e-05,
      "model_forward_time": 1.7786023616790771,
      "step": 91842
    },
    {
      "epoch": 1.701545715332031e-05,
      "step": 91842,
      "training_step_time": 7.59974217414856
    },
    {
      "epoch": 1.701641082763672e-05,
      "model_forward_time": 1.7931947708129883,
      "step": 91843
    },
    {
      "epoch": 1.701641082763672e-05,
      "step": 91843,
      "training_step_time": 6.853237867355347
    },
    {
      "epoch": 1.7017364501953126e-05,
      "model_forward_time": 1.7779738903045654,
      "step": 91844
    },
    {
      "epoch": 1.7017364501953126e-05,
      "step": 91844,
      "training_step_time": 6.855534315109253
    },
    {
      "epoch": 1.701831817626953e-05,
      "model_forward_time": 1.7659263610839844,
      "step": 91845
    },
    {
      "epoch": 1.701831817626953e-05,
      "step": 91845,
      "training_step_time": 6.820363998413086
    },
    {
      "epoch": 1.7019271850585938e-05,
      "model_forward_time": 1.779505968093872,
      "step": 91846
    },
    {
      "epoch": 1.7019271850585938e-05,
      "step": 91846,
      "training_step_time": 6.820742607116699
    },
    {
      "epoch": 1.7020225524902345e-05,
      "model_forward_time": 2.4530205726623535,
      "step": 91847
    },
    {
      "epoch": 1.7020225524902345e-05,
      "step": 91847,
      "training_step_time": 8.546215057373047
    },
    {
      "epoch": 1.702117919921875e-05,
      "model_forward_time": 1.7701938152313232,
      "step": 91848
    },
    {
      "epoch": 1.702117919921875e-05,
      "step": 91848,
      "training_step_time": 7.733680963516235
    },
    {
      "epoch": 1.7022132873535156e-05,
      "model_forward_time": 1.8074190616607666,
      "step": 91849
    },
    {
      "epoch": 1.7022132873535156e-05,
      "step": 91849,
      "training_step_time": 7.312228441238403
    },
    {
      "epoch": 1.7023086547851564e-05,
      "grad_norm": 0.09417738020420074,
      "learning_rate": 1.805440624333593e-07,
      "loss": 0.0542,
      "step": 91850
    },
    {
      "epoch": 1.7023086547851564e-05,
      "model_forward_time": 1.7757999897003174,
      "step": 91850
    },
    {
      "dynamics_loss_avg": 0.05457851598039269,
      "epoch": 1.7023086547851564e-05,
      "step": 91850
    },
    {
      "action_loss_avg": 0.001440407763584517,
      "epoch": 1.7023086547851564e-05,
      "step": 91850
    },
    {
      "epoch": 1.7023086547851564e-05,
      "step": 91850,
      "training_step_time": 7.592871189117432
    },
    {
      "epoch": 1.7024040222167968e-05,
      "model_forward_time": 1.778198003768921,
      "step": 91851
    },
    {
      "epoch": 1.7024040222167968e-05,
      "step": 91851,
      "training_step_time": 7.544830799102783
    },
    {
      "epoch": 1.7024993896484375e-05,
      "model_forward_time": 1.782679796218872,
      "step": 91852
    },
    {
      "epoch": 1.7024993896484375e-05,
      "step": 91852,
      "training_step_time": 6.808557033538818
    },
    {
      "epoch": 1.7025947570800783e-05,
      "model_forward_time": 1.7636010646820068,
      "step": 91853
    },
    {
      "epoch": 1.7025947570800783e-05,
      "step": 91853,
      "training_step_time": 6.8015220165252686
    },
    {
      "epoch": 1.7026901245117187e-05,
      "model_forward_time": 1.7821667194366455,
      "step": 91854
    },
    {
      "epoch": 1.7026901245117187e-05,
      "step": 91854,
      "training_step_time": 6.842280149459839
    },
    {
      "epoch": 1.7027854919433594e-05,
      "model_forward_time": 1.781026840209961,
      "step": 91855
    },
    {
      "epoch": 1.7027854919433594e-05,
      "step": 91855,
      "training_step_time": 6.8610429763793945
    },
    {
      "epoch": 1.702880859375e-05,
      "model_forward_time": 2.4898414611816406,
      "step": 91856
    },
    {
      "epoch": 1.702880859375e-05,
      "step": 91856,
      "training_step_time": 12.013399362564087
    },
    {
      "epoch": 1.7029762268066406e-05,
      "model_forward_time": 1.7641911506652832,
      "step": 91857
    },
    {
      "epoch": 1.7029762268066406e-05,
      "step": 91857,
      "training_step_time": 7.7180094718933105
    },
    {
      "epoch": 1.7030715942382813e-05,
      "model_forward_time": 1.7868072986602783,
      "step": 91858
    },
    {
      "epoch": 1.7030715942382813e-05,
      "step": 91858,
      "training_step_time": 7.366631746292114
    },
    {
      "epoch": 1.703166961669922e-05,
      "model_forward_time": 1.7743268013000488,
      "step": 91859
    },
    {
      "epoch": 1.703166961669922e-05,
      "step": 91859,
      "training_step_time": 8.7929048538208
    },
    {
      "epoch": 1.7032623291015624e-05,
      "grad_norm": 0.08581564575433731,
      "learning_rate": 1.8010401288708734e-07,
      "loss": 0.0532,
      "step": 91860
    },
    {
      "epoch": 1.7032623291015624e-05,
      "model_forward_time": 1.78023099899292,
      "step": 91860
    },
    {
      "dynamics_loss_avg": 0.04477307349443436,
      "epoch": 1.7032623291015624e-05,
      "step": 91860
    },
    {
      "action_loss_avg": 0.0012718609199509957,
      "epoch": 1.7032623291015624e-05,
      "step": 91860
    },
    {
      "epoch": 1.7032623291015624e-05,
      "step": 91860,
      "training_step_time": 8.225403308868408
    },
    {
      "epoch": 1.7033576965332032e-05,
      "model_forward_time": 1.8291945457458496,
      "step": 91861
    },
    {
      "epoch": 1.7033576965332032e-05,
      "step": 91861,
      "training_step_time": 7.191425085067749
    },
    {
      "epoch": 1.7034530639648436e-05,
      "model_forward_time": 1.7818713188171387,
      "step": 91862
    },
    {
      "epoch": 1.7034530639648436e-05,
      "step": 91862,
      "training_step_time": 6.774734735488892
    },
    {
      "epoch": 1.7035484313964843e-05,
      "model_forward_time": 1.779578685760498,
      "step": 91863
    },
    {
      "epoch": 1.7035484313964843e-05,
      "step": 91863,
      "training_step_time": 6.777272462844849
    },
    {
      "epoch": 1.703643798828125e-05,
      "model_forward_time": 1.7766180038452148,
      "step": 91864
    },
    {
      "epoch": 1.703643798828125e-05,
      "step": 91864,
      "training_step_time": 6.778643608093262
    },
    {
      "epoch": 1.7037391662597655e-05,
      "model_forward_time": 2.430786371231079,
      "step": 91865
    },
    {
      "epoch": 1.7037391662597655e-05,
      "step": 91865,
      "training_step_time": 9.91283369064331
    },
    {
      "epoch": 1.7038345336914062e-05,
      "model_forward_time": 1.8450071811676025,
      "step": 91866
    },
    {
      "epoch": 1.7038345336914062e-05,
      "step": 91866,
      "training_step_time": 7.7274463176727295
    },
    {
      "epoch": 1.703929901123047e-05,
      "model_forward_time": 1.779707670211792,
      "step": 91867
    },
    {
      "epoch": 1.703929901123047e-05,
      "step": 91867,
      "training_step_time": 7.767542362213135
    },
    {
      "epoch": 1.7040252685546874e-05,
      "model_forward_time": 1.7932465076446533,
      "step": 91868
    },
    {
      "epoch": 1.7040252685546874e-05,
      "step": 91868,
      "training_step_time": 7.39749550819397
    },
    {
      "epoch": 1.704120635986328e-05,
      "model_forward_time": 1.7922711372375488,
      "step": 91869
    },
    {
      "epoch": 1.704120635986328e-05,
      "step": 91869,
      "training_step_time": 7.379444360733032
    },
    {
      "epoch": 1.704216003417969e-05,
      "grad_norm": 0.15652789175510406,
      "learning_rate": 1.7966449043739343e-07,
      "loss": 0.0537,
      "step": 91870
    },
    {
      "epoch": 1.704216003417969e-05,
      "model_forward_time": 1.7795088291168213,
      "step": 91870
    },
    {
      "dynamics_loss_avg": 0.052778776921331885,
      "epoch": 1.704216003417969e-05,
      "step": 91870
    },
    {
      "action_loss_avg": 0.002210967260180041,
      "epoch": 1.704216003417969e-05,
      "step": 91870
    },
    {
      "epoch": 1.704216003417969e-05,
      "step": 91870,
      "training_step_time": 7.524686098098755
    },
    {
      "epoch": 1.7043113708496092e-05,
      "model_forward_time": 1.7779855728149414,
      "step": 91871
    },
    {
      "epoch": 1.7043113708496092e-05,
      "step": 91871,
      "training_step_time": 6.827486276626587
    },
    {
      "epoch": 1.70440673828125e-05,
      "model_forward_time": 1.7828178405761719,
      "step": 91872
    },
    {
      "epoch": 1.70440673828125e-05,
      "step": 91872,
      "training_step_time": 6.770742416381836
    },
    {
      "epoch": 1.7045021057128907e-05,
      "model_forward_time": 1.778104543685913,
      "step": 91873
    },
    {
      "epoch": 1.7045021057128907e-05,
      "step": 91873,
      "training_step_time": 6.774811506271362
    },
    {
      "epoch": 1.704597473144531e-05,
      "model_forward_time": 1.7885022163391113,
      "step": 91874
    },
    {
      "epoch": 1.704597473144531e-05,
      "step": 91874,
      "training_step_time": 10.664588928222656
    },
    {
      "epoch": 1.704692840576172e-05,
      "model_forward_time": 1.7906007766723633,
      "step": 91875
    },
    {
      "epoch": 1.704692840576172e-05,
      "step": 91875,
      "training_step_time": 7.7392964363098145
    },
    {
      "epoch": 1.7047882080078126e-05,
      "model_forward_time": 1.789935827255249,
      "step": 91876
    },
    {
      "epoch": 1.7047882080078126e-05,
      "step": 91876,
      "training_step_time": 7.7317280769348145
    },
    {
      "epoch": 1.704883575439453e-05,
      "model_forward_time": 1.7852427959442139,
      "step": 91877
    },
    {
      "epoch": 1.704883575439453e-05,
      "step": 91877,
      "training_step_time": 7.27006721496582
    },
    {
      "epoch": 1.7049789428710938e-05,
      "model_forward_time": 1.7781462669372559,
      "step": 91878
    },
    {
      "epoch": 1.7049789428710938e-05,
      "step": 91878,
      "training_step_time": 7.453144788742065
    },
    {
      "epoch": 1.7050743103027345e-05,
      "model_forward_time": 1.7832846641540527,
      "step": 91879
    },
    {
      "epoch": 1.7050743103027345e-05,
      "step": 91879,
      "training_step_time": 7.52151894569397
    },
    {
      "epoch": 1.705169677734375e-05,
      "grad_norm": 0.07552336901426315,
      "learning_rate": 1.79225495132343e-07,
      "loss": 0.0551,
      "step": 91880
    },
    {
      "epoch": 1.705169677734375e-05,
      "model_forward_time": 1.794870376586914,
      "step": 91880
    },
    {
      "dynamics_loss_avg": 0.05391145702451468,
      "epoch": 1.705169677734375e-05,
      "step": 91880
    },
    {
      "action_loss_avg": 0.0007987029472133145,
      "epoch": 1.705169677734375e-05,
      "step": 91880
    },
    {
      "epoch": 1.705169677734375e-05,
      "step": 91880,
      "training_step_time": 6.798424005508423
    },
    {
      "epoch": 1.7052650451660156e-05,
      "model_forward_time": 1.7882189750671387,
      "step": 91881
    },
    {
      "epoch": 1.7052650451660156e-05,
      "step": 91881,
      "training_step_time": 6.792882204055786
    },
    {
      "epoch": 1.7053604125976564e-05,
      "model_forward_time": 1.735550880432129,
      "step": 91882
    },
    {
      "epoch": 1.7053604125976564e-05,
      "step": 91882,
      "training_step_time": 6.7986509799957275
    },
    {
      "epoch": 1.7054557800292968e-05,
      "model_forward_time": 1.7625298500061035,
      "step": 91883
    },
    {
      "epoch": 1.7054557800292968e-05,
      "step": 91883,
      "training_step_time": 8.646559953689575
    },
    {
      "epoch": 1.7055511474609375e-05,
      "model_forward_time": 2.3837575912475586,
      "step": 91884
    },
    {
      "epoch": 1.7055511474609375e-05,
      "step": 91884,
      "training_step_time": 9.037774562835693
    },
    {
      "epoch": 1.7056465148925783e-05,
      "model_forward_time": 1.7630290985107422,
      "step": 91885
    },
    {
      "epoch": 1.7056465148925783e-05,
      "step": 91885,
      "training_step_time": 7.82772421836853
    },
    {
      "epoch": 1.7057418823242187e-05,
      "model_forward_time": 1.765505075454712,
      "step": 91886
    },
    {
      "epoch": 1.7057418823242187e-05,
      "step": 91886,
      "training_step_time": 7.425134658813477
    },
    {
      "epoch": 1.7058372497558594e-05,
      "model_forward_time": 1.7539184093475342,
      "step": 91887
    },
    {
      "epoch": 1.7058372497558594e-05,
      "step": 91887,
      "training_step_time": 7.534149646759033
    },
    {
      "epoch": 1.7059326171875e-05,
      "model_forward_time": 1.760854959487915,
      "step": 91888
    },
    {
      "epoch": 1.7059326171875e-05,
      "step": 91888,
      "training_step_time": 7.588785409927368
    },
    {
      "epoch": 1.7060279846191406e-05,
      "model_forward_time": 1.760411024093628,
      "step": 91889
    },
    {
      "epoch": 1.7060279846191406e-05,
      "step": 91889,
      "training_step_time": 6.838195562362671
    },
    {
      "epoch": 1.7061233520507813e-05,
      "grad_norm": 0.07456547021865845,
      "learning_rate": 1.787870270199449e-07,
      "loss": 0.0554,
      "step": 91890
    },
    {
      "epoch": 1.7061233520507813e-05,
      "model_forward_time": 1.7535731792449951,
      "step": 91890
    },
    {
      "dynamics_loss_avg": 0.06431926656514406,
      "epoch": 1.7061233520507813e-05,
      "step": 91890
    },
    {
      "action_loss_avg": 0.0018091149206156843,
      "epoch": 1.7061233520507813e-05,
      "step": 91890
    },
    {
      "epoch": 1.7061233520507813e-05,
      "step": 91890,
      "training_step_time": 6.852242946624756
    },
    {
      "epoch": 1.7062187194824217e-05,
      "model_forward_time": 1.7550897598266602,
      "step": 91891
    },
    {
      "epoch": 1.7062187194824217e-05,
      "step": 91891,
      "training_step_time": 6.837806463241577
    },
    {
      "epoch": 1.7063140869140624e-05,
      "model_forward_time": 1.7586510181427002,
      "step": 91892
    },
    {
      "epoch": 1.7063140869140624e-05,
      "step": 91892,
      "training_step_time": 8.663724660873413
    },
    {
      "epoch": 1.7064094543457032e-05,
      "model_forward_time": 2.3805644512176514,
      "step": 91893
    },
    {
      "epoch": 1.7064094543457032e-05,
      "step": 91893,
      "training_step_time": 10.317433834075928
    },
    {
      "epoch": 1.7065048217773436e-05,
      "model_forward_time": 1.7620677947998047,
      "step": 91894
    },
    {
      "epoch": 1.7065048217773436e-05,
      "step": 91894,
      "training_step_time": 7.754072666168213
    },
    {
      "epoch": 1.7066001892089843e-05,
      "model_forward_time": 1.7720482349395752,
      "step": 91895
    },
    {
      "epoch": 1.7066001892089843e-05,
      "step": 91895,
      "training_step_time": 7.4341089725494385
    },
    {
      "epoch": 1.706695556640625e-05,
      "model_forward_time": 1.758652925491333,
      "step": 91896
    },
    {
      "epoch": 1.706695556640625e-05,
      "step": 91896,
      "training_step_time": 7.5297112464904785
    },
    {
      "epoch": 1.7067909240722655e-05,
      "model_forward_time": 1.4226479530334473,
      "step": 91897
    },
    {
      "epoch": 1.7067909240722655e-05,
      "step": 91897,
      "training_step_time": 7.6011903285980225
    },
    {
      "epoch": 1.7068862915039062e-05,
      "model_forward_time": 1.7912683486938477,
      "step": 91898
    },
    {
      "epoch": 1.7068862915039062e-05,
      "step": 91898,
      "training_step_time": 6.900375127792358
    },
    {
      "epoch": 1.706981658935547e-05,
      "model_forward_time": 1.7327642440795898,
      "step": 91899
    },
    {
      "epoch": 1.706981658935547e-05,
      "step": 91899,
      "training_step_time": 6.895213842391968
    },
    {
      "epoch": 1.7070770263671874e-05,
      "grad_norm": 0.06603258848190308,
      "learning_rate": 1.783490861481496e-07,
      "loss": 0.0548,
      "step": 91900
    },
    {
      "epoch": 1.7070770263671874e-05,
      "model_forward_time": 1.726506233215332,
      "step": 91900
    },
    {
      "dynamics_loss_avg": 0.06372142285108566,
      "epoch": 1.7070770263671874e-05,
      "step": 91900
    },
    {
      "action_loss_avg": 0.002271303115412593,
      "epoch": 1.7070770263671874e-05,
      "step": 91900
    },
    {
      "epoch": 1.7070770263671874e-05,
      "step": 91900,
      "training_step_time": 6.829740285873413
    },
    {
      "epoch": 1.707172393798828e-05,
      "model_forward_time": 1.7345566749572754,
      "step": 91901
    },
    {
      "epoch": 1.707172393798828e-05,
      "step": 91901,
      "training_step_time": 10.01798939704895
    },
    {
      "epoch": 1.707267761230469e-05,
      "model_forward_time": 2.85732102394104,
      "step": 91902
    },
    {
      "epoch": 1.707267761230469e-05,
      "step": 91902,
      "training_step_time": 10.491568803787231
    },
    {
      "epoch": 1.7073631286621093e-05,
      "model_forward_time": 1.8196587562561035,
      "step": 91903
    },
    {
      "epoch": 1.7073631286621093e-05,
      "step": 91903,
      "training_step_time": 7.8647544384002686
    },
    {
      "epoch": 1.70745849609375e-05,
      "model_forward_time": 1.7399110794067383,
      "step": 91904
    },
    {
      "epoch": 1.70745849609375e-05,
      "step": 91904,
      "training_step_time": 7.5977864265441895
    },
    {
      "epoch": 1.7075538635253907e-05,
      "model_forward_time": 1.7355763912200928,
      "step": 91905
    },
    {
      "epoch": 1.7075538635253907e-05,
      "step": 91905,
      "training_step_time": 7.466815948486328
    },
    {
      "epoch": 1.707649230957031e-05,
      "model_forward_time": 1.7395391464233398,
      "step": 91906
    },
    {
      "epoch": 1.707649230957031e-05,
      "step": 91906,
      "training_step_time": 7.529070615768433
    },
    {
      "epoch": 1.707744598388672e-05,
      "model_forward_time": 1.7504150867462158,
      "step": 91907
    },
    {
      "epoch": 1.707744598388672e-05,
      "step": 91907,
      "training_step_time": 7.173708438873291
    },
    {
      "epoch": 1.7078399658203126e-05,
      "model_forward_time": 1.7488622665405273,
      "step": 91908
    },
    {
      "epoch": 1.7078399658203126e-05,
      "step": 91908,
      "training_step_time": 6.86338472366333
    },
    {
      "epoch": 1.707935333251953e-05,
      "model_forward_time": 1.7339346408843994,
      "step": 91909
    },
    {
      "epoch": 1.707935333251953e-05,
      "step": 91909,
      "training_step_time": 6.818957328796387
    },
    {
      "epoch": 1.7080307006835938e-05,
      "grad_norm": 0.0863582119345665,
      "learning_rate": 1.779116725648483e-07,
      "loss": 0.0552,
      "step": 91910
    },
    {
      "epoch": 1.7080307006835938e-05,
      "model_forward_time": 1.737030267715454,
      "step": 91910
    },
    {
      "dynamics_loss_avg": 0.05740591185167432,
      "epoch": 1.7080307006835938e-05,
      "step": 91910
    },
    {
      "action_loss_avg": 0.001964445528574288,
      "epoch": 1.7080307006835938e-05,
      "step": 91910
    },
    {
      "epoch": 1.7080307006835938e-05,
      "step": 91910,
      "training_step_time": 7.802851676940918
    },
    {
      "epoch": 1.7081260681152345e-05,
      "model_forward_time": 2.8451759815216064,
      "step": 91911
    },
    {
      "epoch": 1.7081260681152345e-05,
      "step": 91911,
      "training_step_time": 10.516358137130737
    },
    {
      "epoch": 1.708221435546875e-05,
      "model_forward_time": 1.7840728759765625,
      "step": 91912
    },
    {
      "epoch": 1.708221435546875e-05,
      "step": 91912,
      "training_step_time": 7.740368127822876
    },
    {
      "epoch": 1.7083168029785156e-05,
      "model_forward_time": 1.737023115158081,
      "step": 91913
    },
    {
      "epoch": 1.7083168029785156e-05,
      "step": 91913,
      "training_step_time": 7.725059986114502
    },
    {
      "epoch": 1.7084121704101564e-05,
      "model_forward_time": 1.7373218536376953,
      "step": 91914
    },
    {
      "epoch": 1.7084121704101564e-05,
      "step": 91914,
      "training_step_time": 7.492823839187622
    },
    {
      "epoch": 1.7085075378417968e-05,
      "model_forward_time": 1.7384295463562012,
      "step": 91915
    },
    {
      "epoch": 1.7085075378417968e-05,
      "step": 91915,
      "training_step_time": 8.203611135482788
    },
    {
      "epoch": 1.7086029052734375e-05,
      "model_forward_time": 1.7365808486938477,
      "step": 91916
    },
    {
      "epoch": 1.7086029052734375e-05,
      "step": 91916,
      "training_step_time": 7.496191501617432
    },
    {
      "epoch": 1.7086982727050783e-05,
      "model_forward_time": 1.7354378700256348,
      "step": 91917
    },
    {
      "epoch": 1.7086982727050783e-05,
      "step": 91917,
      "training_step_time": 6.83459997177124
    },
    {
      "epoch": 1.7087936401367187e-05,
      "model_forward_time": 1.7371609210968018,
      "step": 91918
    },
    {
      "epoch": 1.7087936401367187e-05,
      "step": 91918,
      "training_step_time": 6.83374285697937
    },
    {
      "epoch": 1.7088890075683594e-05,
      "model_forward_time": 1.362816333770752,
      "step": 91919
    },
    {
      "epoch": 1.7088890075683594e-05,
      "step": 91919,
      "training_step_time": 7.738061189651489
    },
    {
      "epoch": 1.708984375e-05,
      "grad_norm": 0.09850923717021942,
      "learning_rate": 1.7747478631787663e-07,
      "loss": 0.054,
      "step": 91920
    },
    {
      "epoch": 1.708984375e-05,
      "model_forward_time": 2.773317575454712,
      "step": 91920
    },
    {
      "dynamics_loss_avg": 0.05871880333870649,
      "epoch": 1.708984375e-05,
      "step": 91920
    },
    {
      "action_loss_avg": 0.0024356477486435326,
      "epoch": 1.708984375e-05,
      "step": 91920
    },
    {
      "epoch": 1.708984375e-05,
      "step": 91920,
      "training_step_time": 7.813497543334961
    },
    {
      "epoch": 1.7090797424316406e-05,
      "model_forward_time": 1.825212001800537,
      "step": 91921
    },
    {
      "epoch": 1.7090797424316406e-05,
      "step": 91921,
      "training_step_time": 8.570195436477661
    },
    {
      "epoch": 1.7091751098632813e-05,
      "model_forward_time": 1.7666051387786865,
      "step": 91922
    },
    {
      "epoch": 1.7091751098632813e-05,
      "step": 91922,
      "training_step_time": 7.725666761398315
    },
    {
      "epoch": 1.7092704772949217e-05,
      "model_forward_time": 1.7647647857666016,
      "step": 91923
    },
    {
      "epoch": 1.7092704772949217e-05,
      "step": 91923,
      "training_step_time": 7.450238227844238
    },
    {
      "epoch": 1.7093658447265625e-05,
      "model_forward_time": 1.763200283050537,
      "step": 91924
    },
    {
      "epoch": 1.7093658447265625e-05,
      "step": 91924,
      "training_step_time": 7.497433662414551
    },
    {
      "epoch": 1.7094612121582032e-05,
      "model_forward_time": 1.7637712955474854,
      "step": 91925
    },
    {
      "epoch": 1.7094612121582032e-05,
      "step": 91925,
      "training_step_time": 7.482189416885376
    },
    {
      "epoch": 1.7095565795898436e-05,
      "model_forward_time": 1.786879062652588,
      "step": 91926
    },
    {
      "epoch": 1.7095565795898436e-05,
      "step": 91926,
      "training_step_time": 6.891478538513184
    },
    {
      "epoch": 1.7096519470214843e-05,
      "model_forward_time": 1.7622418403625488,
      "step": 91927
    },
    {
      "epoch": 1.7096519470214843e-05,
      "step": 91927,
      "training_step_time": 6.821896553039551
    },
    {
      "epoch": 1.709747314453125e-05,
      "model_forward_time": 1.770493745803833,
      "step": 91928
    },
    {
      "epoch": 1.709747314453125e-05,
      "step": 91928,
      "training_step_time": 7.776424169540405
    },
    {
      "epoch": 1.7098426818847655e-05,
      "model_forward_time": 1.7641682624816895,
      "step": 91929
    },
    {
      "epoch": 1.7098426818847655e-05,
      "step": 91929,
      "training_step_time": 10.788760423660278
    },
    {
      "epoch": 1.7099380493164062e-05,
      "grad_norm": 0.08755768835544586,
      "learning_rate": 1.7703842745501187e-07,
      "loss": 0.0547,
      "step": 91930
    },
    {
      "epoch": 1.7099380493164062e-05,
      "model_forward_time": 2.413738965988159,
      "step": 91930
    },
    {
      "dynamics_loss_avg": 0.0609465854242444,
      "epoch": 1.7099380493164062e-05,
      "step": 91930
    },
    {
      "action_loss_avg": 0.002754124079365283,
      "epoch": 1.7099380493164062e-05,
      "step": 91930
    },
    {
      "epoch": 1.7099380493164062e-05,
      "step": 91930,
      "training_step_time": 7.674063205718994
    },
    {
      "epoch": 1.710033416748047e-05,
      "model_forward_time": 1.7640018463134766,
      "step": 91931
    },
    {
      "epoch": 1.710033416748047e-05,
      "step": 91931,
      "training_step_time": 7.795771598815918
    },
    {
      "epoch": 1.7101287841796874e-05,
      "model_forward_time": 1.7636752128601074,
      "step": 91932
    },
    {
      "epoch": 1.7101287841796874e-05,
      "step": 91932,
      "training_step_time": 7.258048057556152
    },
    {
      "epoch": 1.710224151611328e-05,
      "model_forward_time": 1.759263515472412,
      "step": 91933
    },
    {
      "epoch": 1.710224151611328e-05,
      "step": 91933,
      "training_step_time": 7.6073267459869385
    },
    {
      "epoch": 1.710319519042969e-05,
      "model_forward_time": 1.7810494899749756,
      "step": 91934
    },
    {
      "epoch": 1.710319519042969e-05,
      "step": 91934,
      "training_step_time": 7.492775917053223
    },
    {
      "epoch": 1.7104148864746093e-05,
      "model_forward_time": 1.7649202346801758,
      "step": 91935
    },
    {
      "epoch": 1.7104148864746093e-05,
      "step": 91935,
      "training_step_time": 6.83726954460144
    },
    {
      "epoch": 1.71051025390625e-05,
      "model_forward_time": 1.7667124271392822,
      "step": 91936
    },
    {
      "epoch": 1.71051025390625e-05,
      "step": 91936,
      "training_step_time": 6.840771436691284
    },
    {
      "epoch": 1.7106056213378907e-05,
      "model_forward_time": 1.7718243598937988,
      "step": 91937
    },
    {
      "epoch": 1.7106056213378907e-05,
      "step": 91937,
      "training_step_time": 7.781856536865234
    },
    {
      "epoch": 1.710700988769531e-05,
      "model_forward_time": 1.760007381439209,
      "step": 91938
    },
    {
      "epoch": 1.710700988769531e-05,
      "step": 91938,
      "training_step_time": 8.389319658279419
    },
    {
      "epoch": 1.710796356201172e-05,
      "model_forward_time": 2.3859426975250244,
      "step": 91939
    },
    {
      "epoch": 1.710796356201172e-05,
      "step": 91939,
      "training_step_time": 8.224723815917969
    },
    {
      "epoch": 1.7108917236328126e-05,
      "grad_norm": 0.09273699671030045,
      "learning_rate": 1.7660259602397368e-07,
      "loss": 0.0555,
      "step": 91940
    },
    {
      "epoch": 1.7108917236328126e-05,
      "model_forward_time": 1.8224470615386963,
      "step": 91940
    },
    {
      "dynamics_loss_avg": 0.04588355030864477,
      "epoch": 1.7108917236328126e-05,
      "step": 91940
    },
    {
      "action_loss_avg": 0.0015798920387169347,
      "epoch": 1.7108917236328126e-05,
      "step": 91940
    },
    {
      "epoch": 1.7108917236328126e-05,
      "step": 91940,
      "training_step_time": 8.277837038040161
    },
    {
      "epoch": 1.710987091064453e-05,
      "model_forward_time": 1.7568330764770508,
      "step": 91941
    },
    {
      "epoch": 1.710987091064453e-05,
      "step": 91941,
      "training_step_time": 7.055067777633667
    },
    {
      "epoch": 1.7110824584960938e-05,
      "model_forward_time": 1.7623271942138672,
      "step": 91942
    },
    {
      "epoch": 1.7110824584960938e-05,
      "step": 91942,
      "training_step_time": 7.509723663330078
    },
    {
      "epoch": 1.7111778259277345e-05,
      "model_forward_time": 1.7698287963867188,
      "step": 91943
    },
    {
      "epoch": 1.7111778259277345e-05,
      "step": 91943,
      "training_step_time": 7.552441835403442
    },
    {
      "epoch": 1.711273193359375e-05,
      "model_forward_time": 1.7735137939453125,
      "step": 91944
    },
    {
      "epoch": 1.711273193359375e-05,
      "step": 91944,
      "training_step_time": 6.809283018112183
    },
    {
      "epoch": 1.7113685607910157e-05,
      "model_forward_time": 1.759685754776001,
      "step": 91945
    },
    {
      "epoch": 1.7113685607910157e-05,
      "step": 91945,
      "training_step_time": 6.8251447677612305
    },
    {
      "epoch": 1.7114639282226564e-05,
      "model_forward_time": 1.7652263641357422,
      "step": 91946
    },
    {
      "epoch": 1.7114639282226564e-05,
      "step": 91946,
      "training_step_time": 7.813078165054321
    },
    {
      "epoch": 1.7115592956542968e-05,
      "model_forward_time": 1.7638742923736572,
      "step": 91947
    },
    {
      "epoch": 1.7115592956542968e-05,
      "step": 91947,
      "training_step_time": 9.569880247116089
    },
    {
      "epoch": 1.7116546630859375e-05,
      "model_forward_time": 2.402355194091797,
      "step": 91948
    },
    {
      "epoch": 1.7116546630859375e-05,
      "step": 91948,
      "training_step_time": 9.168629884719849
    },
    {
      "epoch": 1.7117500305175783e-05,
      "model_forward_time": 1.82623291015625,
      "step": 91949
    },
    {
      "epoch": 1.7117500305175783e-05,
      "step": 91949,
      "training_step_time": 9.76806354522705
    },
    {
      "epoch": 1.7118453979492187e-05,
      "grad_norm": 0.0846223533153534,
      "learning_rate": 1.7616729207242388e-07,
      "loss": 0.0549,
      "step": 91950
    },
    {
      "epoch": 1.7118453979492187e-05,
      "model_forward_time": 1.7557337284088135,
      "step": 91950
    },
    {
      "dynamics_loss_avg": 0.049950499646365644,
      "epoch": 1.7118453979492187e-05,
      "step": 91950
    },
    {
      "action_loss_avg": 0.002025292592588812,
      "epoch": 1.7118453979492187e-05,
      "step": 91950
    },
    {
      "epoch": 1.7118453979492187e-05,
      "step": 91950,
      "training_step_time": 7.6273698806762695
    },
    {
      "epoch": 1.7119407653808594e-05,
      "model_forward_time": 1.7658498287200928,
      "step": 91951
    },
    {
      "epoch": 1.7119407653808594e-05,
      "step": 91951,
      "training_step_time": 8.465866327285767
    },
    {
      "epoch": 1.7120361328125e-05,
      "model_forward_time": 1.7664122581481934,
      "step": 91952
    },
    {
      "epoch": 1.7120361328125e-05,
      "step": 91952,
      "training_step_time": 8.574534177780151
    },
    {
      "epoch": 1.7121315002441406e-05,
      "model_forward_time": 1.757521152496338,
      "step": 91953
    },
    {
      "epoch": 1.7121315002441406e-05,
      "step": 91953,
      "training_step_time": 6.950714588165283
    },
    {
      "epoch": 1.7122268676757813e-05,
      "model_forward_time": 1.7616848945617676,
      "step": 91954
    },
    {
      "epoch": 1.7122268676757813e-05,
      "step": 91954,
      "training_step_time": 6.928114652633667
    },
    {
      "epoch": 1.7123222351074217e-05,
      "model_forward_time": 1.7644240856170654,
      "step": 91955
    },
    {
      "epoch": 1.7123222351074217e-05,
      "step": 91955,
      "training_step_time": 8.90060305595398
    },
    {
      "epoch": 1.7124176025390625e-05,
      "model_forward_time": 1.760646104812622,
      "step": 91956
    },
    {
      "epoch": 1.7124176025390625e-05,
      "step": 91956,
      "training_step_time": 7.737666845321655
    },
    {
      "epoch": 1.7125129699707032e-05,
      "model_forward_time": 2.4347615242004395,
      "step": 91957
    },
    {
      "epoch": 1.7125129699707032e-05,
      "step": 91957,
      "training_step_time": 9.497223615646362
    },
    {
      "epoch": 1.7126083374023436e-05,
      "model_forward_time": 1.8149116039276123,
      "step": 91958
    },
    {
      "epoch": 1.7126083374023436e-05,
      "step": 91958,
      "training_step_time": 7.642618417739868
    },
    {
      "epoch": 1.7127037048339843e-05,
      "model_forward_time": 1.753230333328247,
      "step": 91959
    },
    {
      "epoch": 1.7127037048339843e-05,
      "step": 91959,
      "training_step_time": 7.755843162536621
    },
    {
      "epoch": 1.712799072265625e-05,
      "grad_norm": 0.08958582580089569,
      "learning_rate": 1.7573251564796556e-07,
      "loss": 0.0544,
      "step": 91960
    },
    {
      "epoch": 1.712799072265625e-05,
      "model_forward_time": 1.7666189670562744,
      "step": 91960
    },
    {
      "dynamics_loss_avg": 0.04446820393204689,
      "epoch": 1.712799072265625e-05,
      "step": 91960
    },
    {
      "action_loss_avg": 0.001715908099868102,
      "epoch": 1.712799072265625e-05,
      "step": 91960
    },
    {
      "epoch": 1.712799072265625e-05,
      "step": 91960,
      "training_step_time": 7.274209260940552
    },
    {
      "epoch": 1.7128944396972655e-05,
      "model_forward_time": 1.765751600265503,
      "step": 91961
    },
    {
      "epoch": 1.7128944396972655e-05,
      "step": 91961,
      "training_step_time": 8.216146469116211
    },
    {
      "epoch": 1.7129898071289062e-05,
      "model_forward_time": 1.75636887550354,
      "step": 91962
    },
    {
      "epoch": 1.7129898071289062e-05,
      "step": 91962,
      "training_step_time": 7.4545135498046875
    },
    {
      "epoch": 1.713085174560547e-05,
      "model_forward_time": 1.7573821544647217,
      "step": 91963
    },
    {
      "epoch": 1.713085174560547e-05,
      "step": 91963,
      "training_step_time": 6.812366724014282
    },
    {
      "epoch": 1.7131805419921874e-05,
      "model_forward_time": 1.7599680423736572,
      "step": 91964
    },
    {
      "epoch": 1.7131805419921874e-05,
      "step": 91964,
      "training_step_time": 8.546947717666626
    },
    {
      "epoch": 1.713275909423828e-05,
      "model_forward_time": 1.7631285190582275,
      "step": 91965
    },
    {
      "epoch": 1.713275909423828e-05,
      "step": 91965,
      "training_step_time": 7.734674692153931
    },
    {
      "epoch": 1.713371276855469e-05,
      "model_forward_time": 2.395259380340576,
      "step": 91966
    },
    {
      "epoch": 1.713371276855469e-05,
      "step": 91966,
      "training_step_time": 8.3924560546875
    },
    {
      "epoch": 1.7134666442871093e-05,
      "model_forward_time": 1.832080364227295,
      "step": 91967
    },
    {
      "epoch": 1.7134666442871093e-05,
      "step": 91967,
      "training_step_time": 8.932718515396118
    },
    {
      "epoch": 1.71356201171875e-05,
      "model_forward_time": 1.7551679611206055,
      "step": 91968
    },
    {
      "epoch": 1.71356201171875e-05,
      "step": 91968,
      "training_step_time": 7.737170696258545
    },
    {
      "epoch": 1.7136573791503907e-05,
      "model_forward_time": 1.7406141757965088,
      "step": 91969
    },
    {
      "epoch": 1.7136573791503907e-05,
      "step": 91969,
      "training_step_time": 7.626728534698486
    },
    {
      "epoch": 1.713752746582031e-05,
      "grad_norm": 0.0949021652340889,
      "learning_rate": 1.7529826679814676e-07,
      "loss": 0.0544,
      "step": 91970
    },
    {
      "epoch": 1.713752746582031e-05,
      "model_forward_time": 1.7893822193145752,
      "step": 91970
    },
    {
      "dynamics_loss_avg": 0.06357462964951992,
      "epoch": 1.713752746582031e-05,
      "step": 91970
    },
    {
      "action_loss_avg": 0.0016796694486401974,
      "epoch": 1.713752746582031e-05,
      "step": 91970
    },
    {
      "epoch": 1.713752746582031e-05,
      "step": 91970,
      "training_step_time": 8.975992679595947
    },
    {
      "epoch": 1.713848114013672e-05,
      "model_forward_time": 1.7691056728363037,
      "step": 91971
    },
    {
      "epoch": 1.713848114013672e-05,
      "step": 91971,
      "training_step_time": 7.457810163497925
    },
    {
      "epoch": 1.7139434814453126e-05,
      "model_forward_time": 1.7589530944824219,
      "step": 91972
    },
    {
      "epoch": 1.7139434814453126e-05,
      "step": 91972,
      "training_step_time": 6.778520822525024
    },
    {
      "epoch": 1.714038848876953e-05,
      "model_forward_time": 1.7656328678131104,
      "step": 91973
    },
    {
      "epoch": 1.714038848876953e-05,
      "step": 91973,
      "training_step_time": 7.782410383224487
    },
    {
      "epoch": 1.7141342163085938e-05,
      "model_forward_time": 1.7755730152130127,
      "step": 91974
    },
    {
      "epoch": 1.7141342163085938e-05,
      "step": 91974,
      "training_step_time": 8.75340485572815
    },
    {
      "epoch": 1.7142295837402345e-05,
      "model_forward_time": 1.7661192417144775,
      "step": 91975
    },
    {
      "epoch": 1.7142295837402345e-05,
      "step": 91975,
      "training_step_time": 10.639172554016113
    },
    {
      "epoch": 1.714324951171875e-05,
      "model_forward_time": 1.772341012954712,
      "step": 91976
    },
    {
      "epoch": 1.714324951171875e-05,
      "step": 91976,
      "training_step_time": 8.28987979888916
    },
    {
      "epoch": 1.7144203186035157e-05,
      "model_forward_time": 1.7610998153686523,
      "step": 91977
    },
    {
      "epoch": 1.7144203186035157e-05,
      "step": 91977,
      "training_step_time": 7.688308477401733
    },
    {
      "epoch": 1.7145156860351564e-05,
      "model_forward_time": 1.7639379501342773,
      "step": 91978
    },
    {
      "epoch": 1.7145156860351564e-05,
      "step": 91978,
      "training_step_time": 7.715815544128418
    },
    {
      "epoch": 1.7146110534667968e-05,
      "model_forward_time": 1.7727627754211426,
      "step": 91979
    },
    {
      "epoch": 1.7146110534667968e-05,
      "step": 91979,
      "training_step_time": 7.649052858352661
    },
    {
      "epoch": 1.7147064208984375e-05,
      "grad_norm": 0.0977027490735054,
      "learning_rate": 1.7486454557045563e-07,
      "loss": 0.054,
      "step": 91980
    },
    {
      "epoch": 1.7147064208984375e-05,
      "model_forward_time": 1.7639305591583252,
      "step": 91980
    },
    {
      "dynamics_loss_avg": 0.052420977503061295,
      "epoch": 1.7147064208984375e-05,
      "step": 91980
    },
    {
      "action_loss_avg": 0.0019657435157569127,
      "epoch": 1.7147064208984375e-05,
      "step": 91980
    },
    {
      "epoch": 1.7147064208984375e-05,
      "step": 91980,
      "training_step_time": 7.624613285064697
    },
    {
      "epoch": 1.7148017883300783e-05,
      "model_forward_time": 1.7671043872833252,
      "step": 91981
    },
    {
      "epoch": 1.7148017883300783e-05,
      "step": 91981,
      "training_step_time": 6.757733345031738
    },
    {
      "epoch": 1.7148971557617187e-05,
      "model_forward_time": 1.760676383972168,
      "step": 91982
    },
    {
      "epoch": 1.7148971557617187e-05,
      "step": 91982,
      "training_step_time": 7.700547456741333
    },
    {
      "epoch": 1.7149925231933594e-05,
      "model_forward_time": 1.7715849876403809,
      "step": 91983
    },
    {
      "epoch": 1.7149925231933594e-05,
      "step": 91983,
      "training_step_time": 8.512186050415039
    },
    {
      "epoch": 1.715087890625e-05,
      "model_forward_time": 1.7652699947357178,
      "step": 91984
    },
    {
      "epoch": 1.715087890625e-05,
      "step": 91984,
      "training_step_time": 7.752671480178833
    },
    {
      "epoch": 1.7151832580566406e-05,
      "model_forward_time": 2.3821747303009033,
      "step": 91985
    },
    {
      "epoch": 1.7151832580566406e-05,
      "step": 91985,
      "training_step_time": 9.197246313095093
    },
    {
      "epoch": 1.7152786254882813e-05,
      "model_forward_time": 1.8284580707550049,
      "step": 91986
    },
    {
      "epoch": 1.7152786254882813e-05,
      "step": 91986,
      "training_step_time": 7.7852911949157715
    },
    {
      "epoch": 1.7153739929199217e-05,
      "model_forward_time": 1.768080472946167,
      "step": 91987
    },
    {
      "epoch": 1.7153739929199217e-05,
      "step": 91987,
      "training_step_time": 7.704527139663696
    },
    {
      "epoch": 1.7154693603515625e-05,
      "model_forward_time": 1.772322654724121,
      "step": 91988
    },
    {
      "epoch": 1.7154693603515625e-05,
      "step": 91988,
      "training_step_time": 7.680845022201538
    },
    {
      "epoch": 1.7155647277832032e-05,
      "model_forward_time": 1.7682545185089111,
      "step": 91989
    },
    {
      "epoch": 1.7155647277832032e-05,
      "step": 91989,
      "training_step_time": 7.754765033721924
    },
    {
      "epoch": 1.7156600952148436e-05,
      "grad_norm": 0.0862816572189331,
      "learning_rate": 1.7443135201232363e-07,
      "loss": 0.0546,
      "step": 91990
    },
    {
      "epoch": 1.7156600952148436e-05,
      "model_forward_time": 1.7689094543457031,
      "step": 91990
    },
    {
      "dynamics_loss_avg": 0.04248085776343942,
      "epoch": 1.7156600952148436e-05,
      "step": 91990
    },
    {
      "action_loss_avg": 0.0009175196595606394,
      "epoch": 1.7156600952148436e-05,
      "step": 91990
    },
    {
      "epoch": 1.7156600952148436e-05,
      "step": 91990,
      "training_step_time": 7.556790351867676
    },
    {
      "epoch": 1.7157554626464843e-05,
      "model_forward_time": 1.7677245140075684,
      "step": 91991
    },
    {
      "epoch": 1.7157554626464843e-05,
      "step": 91991,
      "training_step_time": 8.505246639251709
    },
    {
      "epoch": 1.715850830078125e-05,
      "model_forward_time": 1.7679357528686523,
      "step": 91992
    },
    {
      "epoch": 1.715850830078125e-05,
      "step": 91992,
      "training_step_time": 7.751640796661377
    },
    {
      "epoch": 1.7159461975097655e-05,
      "model_forward_time": 1.7721612453460693,
      "step": 91993
    },
    {
      "epoch": 1.7159461975097655e-05,
      "step": 91993,
      "training_step_time": 7.8479392528533936
    },
    {
      "epoch": 1.7160415649414062e-05,
      "model_forward_time": 2.408531427383423,
      "step": 91994
    },
    {
      "epoch": 1.7160415649414062e-05,
      "step": 91994,
      "training_step_time": 9.998196363449097
    },
    {
      "epoch": 1.716136932373047e-05,
      "model_forward_time": 1.8288450241088867,
      "step": 91995
    },
    {
      "epoch": 1.716136932373047e-05,
      "step": 91995,
      "training_step_time": 7.8721277713775635
    },
    {
      "epoch": 1.7162322998046874e-05,
      "model_forward_time": 1.7691764831542969,
      "step": 91996
    },
    {
      "epoch": 1.7162322998046874e-05,
      "step": 91996,
      "training_step_time": 7.8135457038879395
    },
    {
      "epoch": 1.716327667236328e-05,
      "model_forward_time": 1.7689120769500732,
      "step": 91997
    },
    {
      "epoch": 1.716327667236328e-05,
      "step": 91997,
      "training_step_time": 7.788794994354248
    },
    {
      "epoch": 1.716423034667969e-05,
      "model_forward_time": 1.7780396938323975,
      "step": 91998
    },
    {
      "epoch": 1.716423034667969e-05,
      "step": 91998,
      "training_step_time": 7.96478271484375
    },
    {
      "epoch": 1.7165184020996093e-05,
      "model_forward_time": 1.8047151565551758,
      "step": 91999
    },
    {
      "epoch": 1.7165184020996093e-05,
      "step": 91999,
      "training_step_time": 7.5760719776153564
    },
    {
      "epoch": 1.71661376953125e-05,
      "grad_norm": 0.08044355362653732,
      "learning_rate": 1.7399868617112347e-07,
      "loss": 0.0548,
      "step": 92000
    },
    {
      "epoch": 1.71661376953125e-05,
      "model_forward_time": 1.7559340000152588,
      "step": 92000
    },
    {
      "dynamics_loss_avg": 0.0724118398502469,
      "epoch": 1.71661376953125e-05,
      "step": 92000
    },
    {
      "action_loss_avg": 0.002353328625031281,
      "epoch": 1.71661376953125e-05,
      "step": 92000
    },
    {
      "epoch": 1.71661376953125e-05,
      "step": 92000,
      "training_step_time": 7.430719375610352
    },
    {
      "epoch": 1.7167091369628907e-05,
      "model_forward_time": 1.747079610824585,
      "step": 92001
    },
    {
      "epoch": 1.7167091369628907e-05,
      "step": 92001,
      "training_step_time": 11.42327094078064
    },
    {
      "epoch": 1.716804504394531e-05,
      "model_forward_time": 1.7506842613220215,
      "step": 92002
    },
    {
      "epoch": 1.716804504394531e-05,
      "step": 92002,
      "training_step_time": 11.306430101394653
    },
    {
      "epoch": 1.716899871826172e-05,
      "model_forward_time": 1.761629581451416,
      "step": 92003
    },
    {
      "epoch": 1.716899871826172e-05,
      "step": 92003,
      "training_step_time": 12.334357500076294
    },
    {
      "epoch": 1.7169952392578126e-05,
      "model_forward_time": 1.751042366027832,
      "step": 92004
    },
    {
      "epoch": 1.7169952392578126e-05,
      "step": 92004,
      "training_step_time": 9.772658824920654
    },
    {
      "epoch": 1.717090606689453e-05,
      "model_forward_time": 1.7673606872558594,
      "step": 92005
    },
    {
      "epoch": 1.717090606689453e-05,
      "step": 92005,
      "training_step_time": 8.37652850151062
    },
    {
      "epoch": 1.7171859741210938e-05,
      "model_forward_time": 1.7501599788665771,
      "step": 92006
    },
    {
      "epoch": 1.7171859741210938e-05,
      "step": 92006,
      "training_step_time": 9.538825750350952
    },
    {
      "epoch": 1.7172813415527345e-05,
      "model_forward_time": 1.7332062721252441,
      "step": 92007
    },
    {
      "epoch": 1.7172813415527345e-05,
      "step": 92007,
      "training_step_time": 7.703688859939575
    },
    {
      "epoch": 1.717376708984375e-05,
      "model_forward_time": 1.7361197471618652,
      "step": 92008
    },
    {
      "epoch": 1.717376708984375e-05,
      "step": 92008,
      "training_step_time": 7.504939317703247
    },
    {
      "epoch": 1.7174720764160157e-05,
      "model_forward_time": 1.747715950012207,
      "step": 92009
    },
    {
      "epoch": 1.7174720764160157e-05,
      "step": 92009,
      "training_step_time": 7.853954076766968
    },
    {
      "epoch": 1.7175674438476564e-05,
      "grad_norm": 0.10392536967992783,
      "learning_rate": 1.7356654809417173e-07,
      "loss": 0.0555,
      "step": 92010
    },
    {
      "epoch": 1.7175674438476564e-05,
      "model_forward_time": 1.736515760421753,
      "step": 92010
    },
    {
      "dynamics_loss_avg": 0.06677059344947338,
      "epoch": 1.7175674438476564e-05,
      "step": 92010
    },
    {
      "action_loss_avg": 0.001271240203641355,
      "epoch": 1.7175674438476564e-05,
      "step": 92010
    },
    {
      "epoch": 1.7175674438476564e-05,
      "step": 92010,
      "training_step_time": 8.685461044311523
    },
    {
      "epoch": 1.7176628112792968e-05,
      "model_forward_time": 1.7350335121154785,
      "step": 92011
    },
    {
      "epoch": 1.7176628112792968e-05,
      "step": 92011,
      "training_step_time": 8.595017910003662
    },
    {
      "epoch": 1.7177581787109375e-05,
      "model_forward_time": 2.8392045497894287,
      "step": 92012
    },
    {
      "epoch": 1.7177581787109375e-05,
      "step": 92012,
      "training_step_time": 10.4241361618042
    },
    {
      "epoch": 1.7178535461425783e-05,
      "model_forward_time": 1.7637217044830322,
      "step": 92013
    },
    {
      "epoch": 1.7178535461425783e-05,
      "step": 92013,
      "training_step_time": 8.966673374176025
    },
    {
      "epoch": 1.7179489135742187e-05,
      "model_forward_time": 1.738469123840332,
      "step": 92014
    },
    {
      "epoch": 1.7179489135742187e-05,
      "step": 92014,
      "training_step_time": 7.8031005859375
    },
    {
      "epoch": 1.7180442810058594e-05,
      "model_forward_time": 1.7350351810455322,
      "step": 92015
    },
    {
      "epoch": 1.7180442810058594e-05,
      "step": 92015,
      "training_step_time": 7.5314781665802
    },
    {
      "epoch": 1.7181396484375e-05,
      "model_forward_time": 1.7495965957641602,
      "step": 92016
    },
    {
      "epoch": 1.7181396484375e-05,
      "step": 92016,
      "training_step_time": 8.289137363433838
    },
    {
      "epoch": 1.7182350158691406e-05,
      "model_forward_time": 1.7395646572113037,
      "step": 92017
    },
    {
      "epoch": 1.7182350158691406e-05,
      "step": 92017,
      "training_step_time": 7.5182764530181885
    },
    {
      "epoch": 1.7183303833007813e-05,
      "model_forward_time": 1.7376165390014648,
      "step": 92018
    },
    {
      "epoch": 1.7183303833007813e-05,
      "step": 92018,
      "training_step_time": 7.778497219085693
    },
    {
      "epoch": 1.7184257507324217e-05,
      "model_forward_time": 1.719970703125,
      "step": 92019
    },
    {
      "epoch": 1.7184257507324217e-05,
      "step": 92019,
      "training_step_time": 9.377776384353638
    },
    {
      "epoch": 1.7185211181640625e-05,
      "grad_norm": 0.07242089509963989,
      "learning_rate": 1.731349378287245e-07,
      "loss": 0.0554,
      "step": 92020
    },
    {
      "epoch": 1.7185211181640625e-05,
      "model_forward_time": 1.7268781661987305,
      "step": 92020
    },
    {
      "dynamics_loss_avg": 0.056579865887761115,
      "epoch": 1.7185211181640625e-05,
      "step": 92020
    },
    {
      "action_loss_avg": 0.0013264936103951186,
      "epoch": 1.7185211181640625e-05,
      "step": 92020
    },
    {
      "epoch": 1.7185211181640625e-05,
      "step": 92020,
      "training_step_time": 7.746480226516724
    },
    {
      "epoch": 1.7186164855957032e-05,
      "model_forward_time": 2.8852033615112305,
      "step": 92021
    },
    {
      "epoch": 1.7186164855957032e-05,
      "step": 92021,
      "training_step_time": 9.665708780288696
    },
    {
      "epoch": 1.7187118530273436e-05,
      "model_forward_time": 1.7453582286834717,
      "step": 92022
    },
    {
      "epoch": 1.7187118530273436e-05,
      "step": 92022,
      "training_step_time": 8.520546197891235
    },
    {
      "epoch": 1.7188072204589843e-05,
      "model_forward_time": 1.755028247833252,
      "step": 92023
    },
    {
      "epoch": 1.7188072204589843e-05,
      "step": 92023,
      "training_step_time": 8.846624851226807
    },
    {
      "epoch": 1.718902587890625e-05,
      "model_forward_time": 1.7316725254058838,
      "step": 92024
    },
    {
      "epoch": 1.718902587890625e-05,
      "step": 92024,
      "training_step_time": 8.472700595855713
    },
    {
      "epoch": 1.7189979553222655e-05,
      "model_forward_time": 1.7294175624847412,
      "step": 92025
    },
    {
      "epoch": 1.7189979553222655e-05,
      "step": 92025,
      "training_step_time": 7.704516410827637
    },
    {
      "epoch": 1.7190933227539062e-05,
      "model_forward_time": 1.7451419830322266,
      "step": 92026
    },
    {
      "epoch": 1.7190933227539062e-05,
      "step": 92026,
      "training_step_time": 7.664891004562378
    },
    {
      "epoch": 1.719188690185547e-05,
      "model_forward_time": 1.739149570465088,
      "step": 92027
    },
    {
      "epoch": 1.719188690185547e-05,
      "step": 92027,
      "training_step_time": 8.33766770362854
    },
    {
      "epoch": 1.7192840576171874e-05,
      "model_forward_time": 1.730823278427124,
      "step": 92028
    },
    {
      "epoch": 1.7192840576171874e-05,
      "step": 92028,
      "training_step_time": 8.622986555099487
    },
    {
      "epoch": 1.719379425048828e-05,
      "model_forward_time": 1.737992286682129,
      "step": 92029
    },
    {
      "epoch": 1.719379425048828e-05,
      "step": 92029,
      "training_step_time": 7.803868770599365
    },
    {
      "epoch": 1.719474792480469e-05,
      "grad_norm": 0.09763398766517639,
      "learning_rate": 1.7270385542198452e-07,
      "loss": 0.0564,
      "step": 92030
    },
    {
      "epoch": 1.719474792480469e-05,
      "model_forward_time": 2.8417840003967285,
      "step": 92030
    },
    {
      "dynamics_loss_avg": 0.05140380915254354,
      "epoch": 1.719474792480469e-05,
      "step": 92030
    },
    {
      "action_loss_avg": 0.0015250115480739622,
      "epoch": 1.719474792480469e-05,
      "step": 92030
    },
    {
      "epoch": 1.719474792480469e-05,
      "step": 92030,
      "training_step_time": 9.523308277130127
    },
    {
      "epoch": 1.7195701599121093e-05,
      "model_forward_time": 1.7465565204620361,
      "step": 92031
    },
    {
      "epoch": 1.7195701599121093e-05,
      "step": 92031,
      "training_step_time": 8.862621545791626
    },
    {
      "epoch": 1.71966552734375e-05,
      "model_forward_time": 1.7512409687042236,
      "step": 92032
    },
    {
      "epoch": 1.71966552734375e-05,
      "step": 92032,
      "training_step_time": 7.817507982254028
    },
    {
      "epoch": 1.7197608947753907e-05,
      "model_forward_time": 1.7348895072937012,
      "step": 92033
    },
    {
      "epoch": 1.7197608947753907e-05,
      "step": 92033,
      "training_step_time": 8.313185214996338
    },
    {
      "epoch": 1.719856262207031e-05,
      "model_forward_time": 1.729231834411621,
      "step": 92034
    },
    {
      "epoch": 1.719856262207031e-05,
      "step": 92034,
      "training_step_time": 7.54733419418335
    },
    {
      "epoch": 1.719951629638672e-05,
      "model_forward_time": 1.7334160804748535,
      "step": 92035
    },
    {
      "epoch": 1.719951629638672e-05,
      "step": 92035,
      "training_step_time": 7.6744630336761475
    },
    {
      "epoch": 1.7200469970703126e-05,
      "model_forward_time": 1.7362706661224365,
      "step": 92036
    },
    {
      "epoch": 1.7200469970703126e-05,
      "step": 92036,
      "training_step_time": 7.887593746185303
    },
    {
      "epoch": 1.720142364501953e-05,
      "model_forward_time": 1.7418386936187744,
      "step": 92037
    },
    {
      "epoch": 1.720142364501953e-05,
      "step": 92037,
      "training_step_time": 8.190514087677002
    },
    {
      "epoch": 1.7202377319335938e-05,
      "model_forward_time": 1.7339274883270264,
      "step": 92038
    },
    {
      "epoch": 1.7202377319335938e-05,
      "step": 92038,
      "training_step_time": 9.534737348556519
    },
    {
      "epoch": 1.7203330993652345e-05,
      "model_forward_time": 2.8272414207458496,
      "step": 92039
    },
    {
      "epoch": 1.7203330993652345e-05,
      "step": 92039,
      "training_step_time": 8.460207462310791
    },
    {
      "epoch": 1.720428466796875e-05,
      "grad_norm": 0.08374661207199097,
      "learning_rate": 1.7227330092109306e-07,
      "loss": 0.0562,
      "step": 92040
    },
    {
      "epoch": 1.720428466796875e-05,
      "model_forward_time": 1.7295176982879639,
      "step": 92040
    },
    {
      "dynamics_loss_avg": 0.0360534843057394,
      "epoch": 1.720428466796875e-05,
      "step": 92040
    },
    {
      "action_loss_avg": 0.0011023122089682147,
      "epoch": 1.720428466796875e-05,
      "step": 92040
    },
    {
      "epoch": 1.720428466796875e-05,
      "step": 92040,
      "training_step_time": 7.634241580963135
    },
    {
      "epoch": 1.7205238342285157e-05,
      "model_forward_time": 1.7530031204223633,
      "step": 92041
    },
    {
      "epoch": 1.7205238342285157e-05,
      "step": 92041,
      "training_step_time": 7.615915775299072
    },
    {
      "epoch": 1.7206192016601564e-05,
      "model_forward_time": 1.742431879043579,
      "step": 92042
    },
    {
      "epoch": 1.7206192016601564e-05,
      "step": 92042,
      "training_step_time": 7.65156102180481
    },
    {
      "epoch": 1.7207145690917968e-05,
      "model_forward_time": 1.7223918437957764,
      "step": 92043
    },
    {
      "epoch": 1.7207145690917968e-05,
      "step": 92043,
      "training_step_time": 7.359595775604248
    },
    {
      "epoch": 1.7208099365234375e-05,
      "model_forward_time": 1.737947702407837,
      "step": 92044
    },
    {
      "epoch": 1.7208099365234375e-05,
      "step": 92044,
      "training_step_time": 7.4933130741119385
    },
    {
      "epoch": 1.7209053039550783e-05,
      "model_forward_time": 1.7444202899932861,
      "step": 92045
    },
    {
      "epoch": 1.7209053039550783e-05,
      "step": 92045,
      "training_step_time": 8.281795978546143
    },
    {
      "epoch": 1.7210006713867187e-05,
      "model_forward_time": 1.7267673015594482,
      "step": 92046
    },
    {
      "epoch": 1.7210006713867187e-05,
      "step": 92046,
      "training_step_time": 7.51911187171936
    },
    {
      "epoch": 1.7210960388183594e-05,
      "model_forward_time": 1.7381131649017334,
      "step": 92047
    },
    {
      "epoch": 1.7210960388183594e-05,
      "step": 92047,
      "training_step_time": 8.273551225662231
    },
    {
      "epoch": 1.72119140625e-05,
      "model_forward_time": 2.8557231426239014,
      "step": 92048
    },
    {
      "epoch": 1.72119140625e-05,
      "step": 92048,
      "training_step_time": 7.6238014698028564
    },
    {
      "epoch": 1.7212867736816406e-05,
      "model_forward_time": 1.7401814460754395,
      "step": 92049
    },
    {
      "epoch": 1.7212867736816406e-05,
      "step": 92049,
      "training_step_time": 9.690762758255005
    },
    {
      "epoch": 1.7213821411132813e-05,
      "grad_norm": 0.07455504685640335,
      "learning_rate": 1.718432743731352e-07,
      "loss": 0.0584,
      "step": 92050
    },
    {
      "epoch": 1.7213821411132813e-05,
      "model_forward_time": 1.7611489295959473,
      "step": 92050
    },
    {
      "dynamics_loss_avg": 0.05066665280610323,
      "epoch": 1.7213821411132813e-05,
      "step": 92050
    },
    {
      "action_loss_avg": 0.0013329053268535062,
      "epoch": 1.7213821411132813e-05,
      "step": 92050
    },
    {
      "epoch": 1.7213821411132813e-05,
      "step": 92050,
      "training_step_time": 8.30134916305542
    },
    {
      "epoch": 1.7214775085449217e-05,
      "model_forward_time": 1.7733244895935059,
      "step": 92051
    },
    {
      "epoch": 1.7214775085449217e-05,
      "step": 92051,
      "training_step_time": 7.668428421020508
    },
    {
      "epoch": 1.7215728759765625e-05,
      "model_forward_time": 1.7436339855194092,
      "step": 92052
    },
    {
      "epoch": 1.7215728759765625e-05,
      "step": 92052,
      "training_step_time": 7.377256870269775
    },
    {
      "epoch": 1.7216682434082032e-05,
      "model_forward_time": 1.7372791767120361,
      "step": 92053
    },
    {
      "epoch": 1.7216682434082032e-05,
      "step": 92053,
      "training_step_time": 7.638925790786743
    },
    {
      "epoch": 1.7217636108398436e-05,
      "model_forward_time": 1.7480988502502441,
      "step": 92054
    },
    {
      "epoch": 1.7217636108398436e-05,
      "step": 92054,
      "training_step_time": 7.395589590072632
    },
    {
      "epoch": 1.7218589782714843e-05,
      "model_forward_time": 1.7558655738830566,
      "step": 92055
    },
    {
      "epoch": 1.7218589782714843e-05,
      "step": 92055,
      "training_step_time": 7.754150390625
    },
    {
      "epoch": 1.721954345703125e-05,
      "model_forward_time": 1.7451896667480469,
      "step": 92056
    },
    {
      "epoch": 1.721954345703125e-05,
      "step": 92056,
      "training_step_time": 9.648431062698364
    },
    {
      "epoch": 1.7220497131347655e-05,
      "model_forward_time": 1.7581543922424316,
      "step": 92057
    },
    {
      "epoch": 1.7220497131347655e-05,
      "step": 92057,
      "training_step_time": 8.28618311882019
    },
    {
      "epoch": 1.7221450805664062e-05,
      "model_forward_time": 1.712510108947754,
      "step": 92058
    },
    {
      "epoch": 1.7221450805664062e-05,
      "step": 92058,
      "training_step_time": 7.593816757202148
    },
    {
      "epoch": 1.722240447998047e-05,
      "model_forward_time": 1.748091220855713,
      "step": 92059
    },
    {
      "epoch": 1.722240447998047e-05,
      "step": 92059,
      "training_step_time": 7.594757556915283
    },
    {
      "epoch": 1.7223358154296874e-05,
      "grad_norm": 0.07148992270231247,
      "learning_rate": 1.7141377582513718e-07,
      "loss": 0.056,
      "step": 92060
    },
    {
      "epoch": 1.7223358154296874e-05,
      "model_forward_time": 1.7641422748565674,
      "step": 92060
    },
    {
      "dynamics_loss_avg": 0.05614806795492768,
      "epoch": 1.7223358154296874e-05,
      "step": 92060
    },
    {
      "action_loss_avg": 0.0014161936807795428,
      "epoch": 1.7223358154296874e-05,
      "step": 92060
    },
    {
      "epoch": 1.7223358154296874e-05,
      "step": 92060,
      "training_step_time": 7.606842756271362
    },
    {
      "epoch": 1.722431182861328e-05,
      "model_forward_time": 1.7400920391082764,
      "step": 92061
    },
    {
      "epoch": 1.722431182861328e-05,
      "step": 92061,
      "training_step_time": 7.6105124950408936
    },
    {
      "epoch": 1.722526550292969e-05,
      "model_forward_time": 1.7318215370178223,
      "step": 92062
    },
    {
      "epoch": 1.722526550292969e-05,
      "step": 92062,
      "training_step_time": 7.585389137268066
    },
    {
      "epoch": 1.7226219177246093e-05,
      "model_forward_time": 1.7390546798706055,
      "step": 92063
    },
    {
      "epoch": 1.7226219177246093e-05,
      "step": 92063,
      "training_step_time": 7.098487615585327
    },
    {
      "epoch": 1.72271728515625e-05,
      "model_forward_time": 1.7465898990631104,
      "step": 92064
    },
    {
      "epoch": 1.72271728515625e-05,
      "step": 92064,
      "training_step_time": 7.756191968917847
    },
    {
      "epoch": 1.7228126525878907e-05,
      "model_forward_time": 1.7449183464050293,
      "step": 92065
    },
    {
      "epoch": 1.7228126525878907e-05,
      "step": 92065,
      "training_step_time": 8.178832769393921
    },
    {
      "epoch": 1.722908020019531e-05,
      "model_forward_time": 1.7380931377410889,
      "step": 92066
    },
    {
      "epoch": 1.722908020019531e-05,
      "step": 92066,
      "training_step_time": 8.47941255569458
    },
    {
      "epoch": 1.723003387451172e-05,
      "model_forward_time": 2.5077030658721924,
      "step": 92067
    },
    {
      "epoch": 1.723003387451172e-05,
      "step": 92067,
      "training_step_time": 9.217221736907959
    },
    {
      "epoch": 1.7230987548828126e-05,
      "model_forward_time": 1.7427318096160889,
      "step": 92068
    },
    {
      "epoch": 1.7230987548828126e-05,
      "step": 92068,
      "training_step_time": 7.777658700942993
    },
    {
      "epoch": 1.723194122314453e-05,
      "model_forward_time": 1.7579948902130127,
      "step": 92069
    },
    {
      "epoch": 1.723194122314453e-05,
      "step": 92069,
      "training_step_time": 8.664353132247925
    },
    {
      "epoch": 1.7232894897460938e-05,
      "grad_norm": 0.08053518086671829,
      "learning_rate": 1.709848053240698e-07,
      "loss": 0.0568,
      "step": 92070
    },
    {
      "epoch": 1.7232894897460938e-05,
      "model_forward_time": 1.7301337718963623,
      "step": 92070
    },
    {
      "dynamics_loss_avg": 0.05192309226840734,
      "epoch": 1.7232894897460938e-05,
      "step": 92070
    },
    {
      "action_loss_avg": 0.0011730705504305662,
      "epoch": 1.7232894897460938e-05,
      "step": 92070
    },
    {
      "epoch": 1.7232894897460938e-05,
      "step": 92070,
      "training_step_time": 7.696017503738403
    },
    {
      "epoch": 1.7233848571777345e-05,
      "model_forward_time": 1.7310936450958252,
      "step": 92071
    },
    {
      "epoch": 1.7233848571777345e-05,
      "step": 92071,
      "training_step_time": 7.434895753860474
    },
    {
      "epoch": 1.723480224609375e-05,
      "model_forward_time": 1.7370641231536865,
      "step": 92072
    },
    {
      "epoch": 1.723480224609375e-05,
      "step": 92072,
      "training_step_time": 7.60248589515686
    },
    {
      "epoch": 1.7235755920410157e-05,
      "model_forward_time": 1.745924472808838,
      "step": 92073
    },
    {
      "epoch": 1.7235755920410157e-05,
      "step": 92073,
      "training_step_time": 7.394012689590454
    },
    {
      "epoch": 1.7236709594726564e-05,
      "model_forward_time": 1.7367241382598877,
      "step": 92074
    },
    {
      "epoch": 1.7236709594726564e-05,
      "step": 92074,
      "training_step_time": 7.61047625541687
    },
    {
      "epoch": 1.7237663269042968e-05,
      "model_forward_time": 1.7361345291137695,
      "step": 92075
    },
    {
      "epoch": 1.7237663269042968e-05,
      "step": 92075,
      "training_step_time": 7.72711968421936
    },
    {
      "epoch": 1.7238616943359375e-05,
      "model_forward_time": 2.517530918121338,
      "step": 92076
    },
    {
      "epoch": 1.7238616943359375e-05,
      "step": 92076,
      "training_step_time": 8.490498065948486
    },
    {
      "epoch": 1.7239570617675783e-05,
      "model_forward_time": 1.024247169494629,
      "step": 92077
    },
    {
      "epoch": 1.7239570617675783e-05,
      "step": 92077,
      "training_step_time": 8.65876579284668
    },
    {
      "epoch": 1.7240524291992187e-05,
      "model_forward_time": 1.739424228668213,
      "step": 92078
    },
    {
      "epoch": 1.7240524291992187e-05,
      "step": 92078,
      "training_step_time": 8.377233505249023
    },
    {
      "epoch": 1.7241477966308594e-05,
      "model_forward_time": 1.7530171871185303,
      "step": 92079
    },
    {
      "epoch": 1.7241477966308594e-05,
      "step": 92079,
      "training_step_time": 7.692452430725098
    },
    {
      "epoch": 1.7242431640625e-05,
      "grad_norm": 0.23367851972579956,
      "learning_rate": 1.705563629168433e-07,
      "loss": 0.0576,
      "step": 92080
    },
    {
      "epoch": 1.7242431640625e-05,
      "model_forward_time": 1.7354066371917725,
      "step": 92080
    },
    {
      "dynamics_loss_avg": 0.060472548753023145,
      "epoch": 1.7242431640625e-05,
      "step": 92080
    },
    {
      "action_loss_avg": 0.0015449661761522294,
      "epoch": 1.7242431640625e-05,
      "step": 92080
    },
    {
      "epoch": 1.7242431640625e-05,
      "step": 92080,
      "training_step_time": 7.42755389213562
    },
    {
      "epoch": 1.7243385314941406e-05,
      "model_forward_time": 1.7309095859527588,
      "step": 92081
    },
    {
      "epoch": 1.7243385314941406e-05,
      "step": 92081,
      "training_step_time": 7.757233381271362
    },
    {
      "epoch": 1.7244338989257813e-05,
      "model_forward_time": 1.7392644882202148,
      "step": 92082
    },
    {
      "epoch": 1.7244338989257813e-05,
      "step": 92082,
      "training_step_time": 7.519289255142212
    },
    {
      "epoch": 1.7245292663574217e-05,
      "model_forward_time": 1.7271595001220703,
      "step": 92083
    },
    {
      "epoch": 1.7245292663574217e-05,
      "step": 92083,
      "training_step_time": 7.752026557922363
    },
    {
      "epoch": 1.7246246337890625e-05,
      "model_forward_time": 1.3617513179779053,
      "step": 92084
    },
    {
      "epoch": 1.7246246337890625e-05,
      "step": 92084,
      "training_step_time": 7.864624738693237
    },
    {
      "epoch": 1.7247200012207032e-05,
      "model_forward_time": 1.738161563873291,
      "step": 92085
    },
    {
      "epoch": 1.7247200012207032e-05,
      "step": 92085,
      "training_step_time": 9.909032106399536
    },
    {
      "epoch": 1.7248153686523436e-05,
      "model_forward_time": 1.7348508834838867,
      "step": 92086
    },
    {
      "epoch": 1.7248153686523436e-05,
      "step": 92086,
      "training_step_time": 8.823594808578491
    },
    {
      "epoch": 1.7249107360839844e-05,
      "model_forward_time": 1.7789127826690674,
      "step": 92087
    },
    {
      "epoch": 1.7249107360839844e-05,
      "step": 92087,
      "training_step_time": 9.594884395599365
    },
    {
      "epoch": 1.725006103515625e-05,
      "model_forward_time": 1.7331418991088867,
      "step": 92088
    },
    {
      "epoch": 1.725006103515625e-05,
      "step": 92088,
      "training_step_time": 7.782326936721802
    },
    {
      "epoch": 1.7251014709472655e-05,
      "model_forward_time": 1.7316985130310059,
      "step": 92089
    },
    {
      "epoch": 1.7251014709472655e-05,
      "step": 92089,
      "training_step_time": 7.758189916610718
    },
    {
      "epoch": 1.7251968383789062e-05,
      "grad_norm": 0.08816824853420258,
      "learning_rate": 1.7012844865031186e-07,
      "loss": 0.0551,
      "step": 92090
    },
    {
      "epoch": 1.7251968383789062e-05,
      "model_forward_time": 1.7406048774719238,
      "step": 92090
    },
    {
      "dynamics_loss_avg": 0.07370326295495033,
      "epoch": 1.7251968383789062e-05,
      "step": 92090
    },
    {
      "action_loss_avg": 0.0012977806443814188,
      "epoch": 1.7251968383789062e-05,
      "step": 92090
    },
    {
      "epoch": 1.7251968383789062e-05,
      "step": 92090,
      "training_step_time": 8.282147645950317
    },
    {
      "epoch": 1.725292205810547e-05,
      "model_forward_time": 1.7365679740905762,
      "step": 92091
    },
    {
      "epoch": 1.725292205810547e-05,
      "step": 92091,
      "training_step_time": 7.512053728103638
    },
    {
      "epoch": 1.7253875732421874e-05,
      "model_forward_time": 1.7376930713653564,
      "step": 92092
    },
    {
      "epoch": 1.7253875732421874e-05,
      "step": 92092,
      "training_step_time": 7.72121787071228
    },
    {
      "epoch": 1.725482940673828e-05,
      "model_forward_time": 1.7395131587982178,
      "step": 92093
    },
    {
      "epoch": 1.725482940673828e-05,
      "step": 92093,
      "training_step_time": 8.67164397239685
    },
    {
      "epoch": 1.725578308105469e-05,
      "model_forward_time": 1.739272117614746,
      "step": 92094
    },
    {
      "epoch": 1.725578308105469e-05,
      "step": 92094,
      "training_step_time": 8.489980697631836
    },
    {
      "epoch": 1.7256736755371093e-05,
      "model_forward_time": 2.9263415336608887,
      "step": 92095
    },
    {
      "epoch": 1.7256736755371093e-05,
      "step": 92095,
      "training_step_time": 7.843190431594849
    },
    {
      "epoch": 1.72576904296875e-05,
      "model_forward_time": 1.7452492713928223,
      "step": 92096
    },
    {
      "epoch": 1.72576904296875e-05,
      "step": 92096,
      "training_step_time": 9.551872491836548
    },
    {
      "epoch": 1.7258644104003907e-05,
      "model_forward_time": 1.7642619609832764,
      "step": 92097
    },
    {
      "epoch": 1.7258644104003907e-05,
      "step": 92097,
      "training_step_time": 8.90518307685852
    },
    {
      "epoch": 1.725959777832031e-05,
      "model_forward_time": 1.736816167831421,
      "step": 92098
    },
    {
      "epoch": 1.725959777832031e-05,
      "step": 92098,
      "training_step_time": 7.730966329574585
    },
    {
      "epoch": 1.726055145263672e-05,
      "model_forward_time": 1.736973762512207,
      "step": 92099
    },
    {
      "epoch": 1.726055145263672e-05,
      "step": 92099,
      "training_step_time": 7.637722730636597
    },
    {
      "epoch": 1.7261505126953126e-05,
      "grad_norm": 0.0816589891910553,
      "learning_rate": 1.697010625712714e-07,
      "loss": 0.0578,
      "step": 92100
    },
    {
      "epoch": 1.7261505126953126e-05,
      "model_forward_time": 1.7399260997772217,
      "step": 92100
    },
    {
      "dynamics_loss_avg": 0.04326922297477722,
      "epoch": 1.7261505126953126e-05,
      "step": 92100
    },
    {
      "action_loss_avg": 0.0013010808150283992,
      "epoch": 1.7261505126953126e-05,
      "step": 92100
    },
    {
      "epoch": 1.7261505126953126e-05,
      "step": 92100,
      "training_step_time": 7.798511266708374
    },
    {
      "epoch": 1.726245880126953e-05,
      "model_forward_time": 1.7458319664001465,
      "step": 92101
    },
    {
      "epoch": 1.726245880126953e-05,
      "step": 92101,
      "training_step_time": 7.47437858581543
    },
    {
      "epoch": 1.7263412475585938e-05,
      "model_forward_time": 1.7343192100524902,
      "step": 92102
    },
    {
      "epoch": 1.7263412475585938e-05,
      "step": 92102,
      "training_step_time": 7.913387775421143
    },
    {
      "epoch": 1.7264366149902345e-05,
      "model_forward_time": 1.7367868423461914,
      "step": 92103
    },
    {
      "epoch": 1.7264366149902345e-05,
      "step": 92103,
      "training_step_time": 7.801948547363281
    },
    {
      "epoch": 1.726531982421875e-05,
      "model_forward_time": 2.49141263961792,
      "step": 92104
    },
    {
      "epoch": 1.726531982421875e-05,
      "step": 92104,
      "training_step_time": 8.761991262435913
    },
    {
      "epoch": 1.7266273498535157e-05,
      "model_forward_time": 1.7561190128326416,
      "step": 92105
    },
    {
      "epoch": 1.7266273498535157e-05,
      "step": 92105,
      "training_step_time": 8.306924819946289
    },
    {
      "epoch": 1.7267227172851564e-05,
      "model_forward_time": 1.7698290348052979,
      "step": 92106
    },
    {
      "epoch": 1.7267227172851564e-05,
      "step": 92106,
      "training_step_time": 8.331239461898804
    },
    {
      "epoch": 1.7268180847167968e-05,
      "model_forward_time": 1.7445271015167236,
      "step": 92107
    },
    {
      "epoch": 1.7268180847167968e-05,
      "step": 92107,
      "training_step_time": 7.710787296295166
    },
    {
      "epoch": 1.7269134521484376e-05,
      "model_forward_time": 1.737441062927246,
      "step": 92108
    },
    {
      "epoch": 1.7269134521484376e-05,
      "step": 92108,
      "training_step_time": 7.61721134185791
    },
    {
      "epoch": 1.7270088195800783e-05,
      "model_forward_time": 1.7525660991668701,
      "step": 92109
    },
    {
      "epoch": 1.7270088195800783e-05,
      "step": 92109,
      "training_step_time": 7.512796878814697
    },
    {
      "epoch": 1.7271041870117187e-05,
      "grad_norm": 0.11341729760169983,
      "learning_rate": 1.6927420472646118e-07,
      "loss": 0.0556,
      "step": 92110
    },
    {
      "epoch": 1.7271041870117187e-05,
      "model_forward_time": 1.743227481842041,
      "step": 92110
    },
    {
      "dynamics_loss_avg": 0.07058265563100577,
      "epoch": 1.7271041870117187e-05,
      "step": 92110
    },
    {
      "action_loss_avg": 0.0018211080139735713,
      "epoch": 1.7271041870117187e-05,
      "step": 92110
    },
    {
      "epoch": 1.7271041870117187e-05,
      "step": 92110,
      "training_step_time": 7.887334823608398
    },
    {
      "epoch": 1.7271995544433594e-05,
      "model_forward_time": 1.7430305480957031,
      "step": 92111
    },
    {
      "epoch": 1.7271995544433594e-05,
      "step": 92111,
      "training_step_time": 8.880024909973145
    },
    {
      "epoch": 1.727294921875e-05,
      "model_forward_time": 1.7495663166046143,
      "step": 92112
    },
    {
      "epoch": 1.727294921875e-05,
      "step": 92112,
      "training_step_time": 8.337519884109497
    },
    {
      "epoch": 1.7273902893066406e-05,
      "model_forward_time": 2.8310296535491943,
      "step": 92113
    },
    {
      "epoch": 1.7273902893066406e-05,
      "step": 92113,
      "training_step_time": 8.319987058639526
    },
    {
      "epoch": 1.7274856567382813e-05,
      "model_forward_time": 1.758100986480713,
      "step": 92114
    },
    {
      "epoch": 1.7274856567382813e-05,
      "step": 92114,
      "training_step_time": 8.769927263259888
    },
    {
      "epoch": 1.7275810241699217e-05,
      "model_forward_time": 1.7674708366394043,
      "step": 92115
    },
    {
      "epoch": 1.7275810241699217e-05,
      "step": 92115,
      "training_step_time": 9.499677658081055
    },
    {
      "epoch": 1.7276763916015625e-05,
      "model_forward_time": 1.7400569915771484,
      "step": 92116
    },
    {
      "epoch": 1.7276763916015625e-05,
      "step": 92116,
      "training_step_time": 7.826653480529785
    },
    {
      "epoch": 1.7277717590332032e-05,
      "model_forward_time": 1.748143196105957,
      "step": 92117
    },
    {
      "epoch": 1.7277717590332032e-05,
      "step": 92117,
      "training_step_time": 8.57517123222351
    },
    {
      "epoch": 1.7278671264648436e-05,
      "model_forward_time": 1.7400565147399902,
      "step": 92118
    },
    {
      "epoch": 1.7278671264648436e-05,
      "step": 92118,
      "training_step_time": 8.074706792831421
    },
    {
      "epoch": 1.7279624938964844e-05,
      "model_forward_time": 1.737978458404541,
      "step": 92119
    },
    {
      "epoch": 1.7279624938964844e-05,
      "step": 92119,
      "training_step_time": 8.403941631317139
    },
    {
      "epoch": 1.728057861328125e-05,
      "grad_norm": 0.14032317698001862,
      "learning_rate": 1.6884787516256108e-07,
      "loss": 0.0543,
      "step": 92120
    },
    {
      "epoch": 1.728057861328125e-05,
      "model_forward_time": 1.736262321472168,
      "step": 92120
    },
    {
      "dynamics_loss_avg": 0.05213621705770492,
      "epoch": 1.728057861328125e-05,
      "step": 92120
    },
    {
      "action_loss_avg": 0.0013457193417707458,
      "epoch": 1.728057861328125e-05,
      "step": 92120
    },
    {
      "epoch": 1.728057861328125e-05,
      "step": 92120,
      "training_step_time": 7.796088933944702
    },
    {
      "epoch": 1.7281532287597655e-05,
      "model_forward_time": 1.7464518547058105,
      "step": 92121
    },
    {
      "epoch": 1.7281532287597655e-05,
      "step": 92121,
      "training_step_time": 8.985231876373291
    },
    {
      "epoch": 1.7282485961914062e-05,
      "model_forward_time": 2.8066282272338867,
      "step": 92122
    },
    {
      "epoch": 1.7282485961914062e-05,
      "step": 92122,
      "training_step_time": 10.454949378967285
    },
    {
      "epoch": 1.728343963623047e-05,
      "model_forward_time": 1.7522952556610107,
      "step": 92123
    },
    {
      "epoch": 1.728343963623047e-05,
      "step": 92123,
      "training_step_time": 7.76501989364624
    },
    {
      "epoch": 1.7284393310546874e-05,
      "model_forward_time": 1.766681432723999,
      "step": 92124
    },
    {
      "epoch": 1.7284393310546874e-05,
      "step": 92124,
      "training_step_time": 8.81826901435852
    },
    {
      "epoch": 1.728534698486328e-05,
      "model_forward_time": 1.7443835735321045,
      "step": 92125
    },
    {
      "epoch": 1.728534698486328e-05,
      "step": 92125,
      "training_step_time": 8.845078706741333
    },
    {
      "epoch": 1.728630065917969e-05,
      "model_forward_time": 1.7454464435577393,
      "step": 92126
    },
    {
      "epoch": 1.728630065917969e-05,
      "step": 92126,
      "training_step_time": 7.784589052200317
    },
    {
      "epoch": 1.7287254333496093e-05,
      "model_forward_time": 1.734823226928711,
      "step": 92127
    },
    {
      "epoch": 1.7287254333496093e-05,
      "step": 92127,
      "training_step_time": 7.678532838821411
    },
    {
      "epoch": 1.72882080078125e-05,
      "model_forward_time": 1.7395827770233154,
      "step": 92128
    },
    {
      "epoch": 1.72882080078125e-05,
      "step": 92128,
      "training_step_time": 7.835946798324585
    },
    {
      "epoch": 1.7289161682128908e-05,
      "model_forward_time": 1.7381300926208496,
      "step": 92129
    },
    {
      "epoch": 1.7289161682128908e-05,
      "step": 92129,
      "training_step_time": 7.745278596878052
    },
    {
      "epoch": 1.729011535644531e-05,
      "grad_norm": 0.07774195075035095,
      "learning_rate": 1.6842207392619326e-07,
      "loss": 0.0537,
      "step": 92130
    },
    {
      "epoch": 1.729011535644531e-05,
      "model_forward_time": 1.732239007949829,
      "step": 92130
    },
    {
      "dynamics_loss_avg": 0.048863948322832584,
      "epoch": 1.729011535644531e-05,
      "step": 92130
    },
    {
      "action_loss_avg": 0.0015468794154003263,
      "epoch": 1.729011535644531e-05,
      "step": 92130
    },
    {
      "epoch": 1.729011535644531e-05,
      "step": 92130,
      "training_step_time": 7.9187986850738525
    },
    {
      "epoch": 1.729106903076172e-05,
      "model_forward_time": 1.7428157329559326,
      "step": 92131
    },
    {
      "epoch": 1.729106903076172e-05,
      "step": 92131,
      "training_step_time": 10.367379903793335
    },
    {
      "epoch": 1.7292022705078126e-05,
      "model_forward_time": 1.7589290142059326,
      "step": 92132
    },
    {
      "epoch": 1.7292022705078126e-05,
      "step": 92132,
      "training_step_time": 8.326043367385864
    },
    {
      "epoch": 1.729297637939453e-05,
      "model_forward_time": 1.7293834686279297,
      "step": 92133
    },
    {
      "epoch": 1.729297637939453e-05,
      "step": 92133,
      "training_step_time": 8.400149822235107
    },
    {
      "epoch": 1.7293930053710938e-05,
      "model_forward_time": 1.7474398612976074,
      "step": 92134
    },
    {
      "epoch": 1.7293930053710938e-05,
      "step": 92134,
      "training_step_time": 7.824477672576904
    },
    {
      "epoch": 1.7294883728027345e-05,
      "model_forward_time": 1.73537015914917,
      "step": 92135
    },
    {
      "epoch": 1.7294883728027345e-05,
      "step": 92135,
      "training_step_time": 8.769390106201172
    },
    {
      "epoch": 1.729583740234375e-05,
      "model_forward_time": 1.7404000759124756,
      "step": 92136
    },
    {
      "epoch": 1.729583740234375e-05,
      "step": 92136,
      "training_step_time": 7.584250211715698
    },
    {
      "epoch": 1.7296791076660157e-05,
      "model_forward_time": 1.747128963470459,
      "step": 92137
    },
    {
      "epoch": 1.7296791076660157e-05,
      "step": 92137,
      "training_step_time": 7.438791990280151
    },
    {
      "epoch": 1.7297744750976564e-05,
      "model_forward_time": 1.7306103706359863,
      "step": 92138
    },
    {
      "epoch": 1.7297744750976564e-05,
      "step": 92138,
      "training_step_time": 7.880359888076782
    },
    {
      "epoch": 1.7298698425292968e-05,
      "model_forward_time": 1.7384514808654785,
      "step": 92139
    },
    {
      "epoch": 1.7298698425292968e-05,
      "step": 92139,
      "training_step_time": 9.609598636627197
    },
    {
      "epoch": 1.7299652099609376e-05,
      "grad_norm": 0.11612337827682495,
      "learning_rate": 1.679968010639238e-07,
      "loss": 0.0552,
      "step": 92140
    },
    {
      "epoch": 1.7299652099609376e-05,
      "model_forward_time": 1.7305667400360107,
      "step": 92140
    },
    {
      "dynamics_loss_avg": 0.048747973144054414,
      "epoch": 1.7299652099609376e-05,
      "step": 92140
    },
    {
      "action_loss_avg": 0.0014758135454030707,
      "epoch": 1.7299652099609376e-05,
      "step": 92140
    },
    {
      "epoch": 1.7299652099609376e-05,
      "step": 92140,
      "training_step_time": 7.872494220733643
    },
    {
      "epoch": 1.730060577392578e-05,
      "model_forward_time": 2.803987741470337,
      "step": 92141
    },
    {
      "epoch": 1.730060577392578e-05,
      "step": 92141,
      "training_step_time": 8.565999269485474
    },
    {
      "epoch": 1.7301559448242187e-05,
      "model_forward_time": 1.7453501224517822,
      "step": 92142
    },
    {
      "epoch": 1.7301559448242187e-05,
      "step": 92142,
      "training_step_time": 8.830828666687012
    },
    {
      "epoch": 1.7302513122558594e-05,
      "model_forward_time": 1.7425990104675293,
      "step": 92143
    },
    {
      "epoch": 1.7302513122558594e-05,
      "step": 92143,
      "training_step_time": 7.830862760543823
    },
    {
      "epoch": 1.7303466796875e-05,
      "model_forward_time": 1.7456183433532715,
      "step": 92144
    },
    {
      "epoch": 1.7303466796875e-05,
      "step": 92144,
      "training_step_time": 7.8037731647491455
    },
    {
      "epoch": 1.7304420471191406e-05,
      "model_forward_time": 1.7415781021118164,
      "step": 92145
    },
    {
      "epoch": 1.7304420471191406e-05,
      "step": 92145,
      "training_step_time": 7.836827993392944
    },
    {
      "epoch": 1.7305374145507813e-05,
      "model_forward_time": 1.738307237625122,
      "step": 92146
    },
    {
      "epoch": 1.7305374145507813e-05,
      "step": 92146,
      "training_step_time": 7.467196702957153
    },
    {
      "epoch": 1.7306327819824217e-05,
      "model_forward_time": 1.7355546951293945,
      "step": 92147
    },
    {
      "epoch": 1.7306327819824217e-05,
      "step": 92147,
      "training_step_time": 7.951087474822998
    },
    {
      "epoch": 1.7307281494140625e-05,
      "model_forward_time": 1.7526862621307373,
      "step": 92148
    },
    {
      "epoch": 1.7307281494140625e-05,
      "step": 92148,
      "training_step_time": 8.426791667938232
    },
    {
      "epoch": 1.7308235168457032e-05,
      "model_forward_time": 1.7392926216125488,
      "step": 92149
    },
    {
      "epoch": 1.7308235168457032e-05,
      "step": 92149,
      "training_step_time": 7.806161403656006
    },
    {
      "epoch": 1.7309188842773436e-05,
      "grad_norm": 0.0926462709903717,
      "learning_rate": 1.6757205662225828e-07,
      "loss": 0.0545,
      "step": 92150
    },
    {
      "epoch": 1.7309188842773436e-05,
      "model_forward_time": 2.8197543621063232,
      "step": 92150
    },
    {
      "dynamics_loss_avg": 0.055662640184164044,
      "epoch": 1.7309188842773436e-05,
      "step": 92150
    },
    {
      "action_loss_avg": 0.0013058765354799107,
      "epoch": 1.7309188842773436e-05,
      "step": 92150
    },
    {
      "epoch": 1.7309188842773436e-05,
      "step": 92150,
      "training_step_time": 7.8382861614227295
    },
    {
      "epoch": 1.7310142517089844e-05,
      "model_forward_time": 1.7482481002807617,
      "step": 92151
    },
    {
      "epoch": 1.7310142517089844e-05,
      "step": 92151,
      "training_step_time": 9.17095422744751
    },
    {
      "epoch": 1.731109619140625e-05,
      "model_forward_time": 1.7623178958892822,
      "step": 92152
    },
    {
      "epoch": 1.731109619140625e-05,
      "step": 92152,
      "training_step_time": 9.716240644454956
    },
    {
      "epoch": 1.7312049865722655e-05,
      "model_forward_time": 1.740800380706787,
      "step": 92153
    },
    {
      "epoch": 1.7312049865722655e-05,
      "step": 92153,
      "training_step_time": 7.946110963821411
    },
    {
      "epoch": 1.7313003540039062e-05,
      "model_forward_time": 1.7415051460266113,
      "step": 92154
    },
    {
      "epoch": 1.7313003540039062e-05,
      "step": 92154,
      "training_step_time": 7.801780939102173
    },
    {
      "epoch": 1.731395721435547e-05,
      "model_forward_time": 1.7514715194702148,
      "step": 92155
    },
    {
      "epoch": 1.731395721435547e-05,
      "step": 92155,
      "training_step_time": 7.446408748626709
    },
    {
      "epoch": 1.7314910888671874e-05,
      "model_forward_time": 1.7425589561462402,
      "step": 92156
    },
    {
      "epoch": 1.7314910888671874e-05,
      "step": 92156,
      "training_step_time": 7.405447006225586
    },
    {
      "epoch": 1.731586456298828e-05,
      "model_forward_time": 1.7394776344299316,
      "step": 92157
    },
    {
      "epoch": 1.731586456298828e-05,
      "step": 92157,
      "training_step_time": 8.117336511611938
    },
    {
      "epoch": 1.731681823730469e-05,
      "model_forward_time": 1.7386808395385742,
      "step": 92158
    },
    {
      "epoch": 1.731681823730469e-05,
      "step": 92158,
      "training_step_time": 7.932041168212891
    },
    {
      "epoch": 1.7317771911621093e-05,
      "model_forward_time": 2.8275256156921387,
      "step": 92159
    },
    {
      "epoch": 1.7317771911621093e-05,
      "step": 92159,
      "training_step_time": 7.70978045463562
    },
    {
      "epoch": 1.73187255859375e-05,
      "grad_norm": 0.08345726132392883,
      "learning_rate": 1.6714784064764844e-07,
      "loss": 0.0546,
      "step": 92160
    },
    {
      "epoch": 1.73187255859375e-05,
      "model_forward_time": 1.7344985008239746,
      "step": 92160
    },
    {
      "dynamics_loss_avg": 0.048245524242520335,
      "epoch": 1.73187255859375e-05,
      "step": 92160
    },
    {
      "action_loss_avg": 0.001366316166240722,
      "epoch": 1.73187255859375e-05,
      "step": 92160
    },
    {
      "epoch": 1.73187255859375e-05,
      "step": 92160,
      "training_step_time": 8.308066844940186
    },
    {
      "epoch": 1.7319679260253908e-05,
      "model_forward_time": 1.7565443515777588,
      "step": 92161
    },
    {
      "epoch": 1.7319679260253908e-05,
      "step": 92161,
      "training_step_time": 8.88190770149231
    },
    {
      "epoch": 1.732063293457031e-05,
      "model_forward_time": 1.738312005996704,
      "step": 92162
    },
    {
      "epoch": 1.732063293457031e-05,
      "step": 92162,
      "training_step_time": 7.651861190795898
    },
    {
      "epoch": 1.732158660888672e-05,
      "model_forward_time": 1.7380800247192383,
      "step": 92163
    },
    {
      "epoch": 1.732158660888672e-05,
      "step": 92163,
      "training_step_time": 7.6869800090789795
    },
    {
      "epoch": 1.7322540283203126e-05,
      "model_forward_time": 1.740553379058838,
      "step": 92164
    },
    {
      "epoch": 1.7322540283203126e-05,
      "step": 92164,
      "training_step_time": 7.430976390838623
    },
    {
      "epoch": 1.732349395751953e-05,
      "model_forward_time": 1.7476780414581299,
      "step": 92165
    },
    {
      "epoch": 1.732349395751953e-05,
      "step": 92165,
      "training_step_time": 7.932251453399658
    },
    {
      "epoch": 1.7324447631835938e-05,
      "model_forward_time": 1.743224859237671,
      "step": 92166
    },
    {
      "epoch": 1.7324447631835938e-05,
      "step": 92166,
      "training_step_time": 7.573819637298584
    },
    {
      "epoch": 1.7325401306152345e-05,
      "model_forward_time": 1.7391340732574463,
      "step": 92167
    },
    {
      "epoch": 1.7325401306152345e-05,
      "step": 92167,
      "training_step_time": 7.934503555297852
    },
    {
      "epoch": 1.732635498046875e-05,
      "model_forward_time": 2.8587517738342285,
      "step": 92168
    },
    {
      "epoch": 1.732635498046875e-05,
      "step": 92168,
      "training_step_time": 7.800412654876709
    },
    {
      "epoch": 1.7327308654785157e-05,
      "model_forward_time": 1.7407987117767334,
      "step": 92169
    },
    {
      "epoch": 1.7327308654785157e-05,
      "step": 92169,
      "training_step_time": 8.986840724945068
    },
    {
      "epoch": 1.7328262329101564e-05,
      "grad_norm": 0.15076814591884613,
      "learning_rate": 1.6672415318648384e-07,
      "loss": 0.054,
      "step": 92170
    },
    {
      "epoch": 1.7328262329101564e-05,
      "model_forward_time": 1.750340223312378,
      "step": 92170
    },
    {
      "dynamics_loss_avg": 0.05686593558639288,
      "epoch": 1.7328262329101564e-05,
      "step": 92170
    },
    {
      "action_loss_avg": 0.001490372675471008,
      "epoch": 1.7328262329101564e-05,
      "step": 92170
    },
    {
      "epoch": 1.7328262329101564e-05,
      "step": 92170,
      "training_step_time": 7.845099687576294
    },
    {
      "epoch": 1.7329216003417968e-05,
      "model_forward_time": 1.7451815605163574,
      "step": 92171
    },
    {
      "epoch": 1.7329216003417968e-05,
      "step": 92171,
      "training_step_time": 7.713369846343994
    },
    {
      "epoch": 1.7330169677734376e-05,
      "model_forward_time": 1.7383031845092773,
      "step": 92172
    },
    {
      "epoch": 1.7330169677734376e-05,
      "step": 92172,
      "training_step_time": 7.8284711837768555
    },
    {
      "epoch": 1.733112335205078e-05,
      "model_forward_time": 1.7376854419708252,
      "step": 92173
    },
    {
      "epoch": 1.733112335205078e-05,
      "step": 92173,
      "training_step_time": 7.46479606628418
    },
    {
      "epoch": 1.7332077026367187e-05,
      "model_forward_time": 1.7361018657684326,
      "step": 92174
    },
    {
      "epoch": 1.7332077026367187e-05,
      "step": 92174,
      "training_step_time": 7.298067569732666
    },
    {
      "epoch": 1.7333030700683594e-05,
      "model_forward_time": 1.7457175254821777,
      "step": 92175
    },
    {
      "epoch": 1.7333030700683594e-05,
      "step": 92175,
      "training_step_time": 9.266492128372192
    },
    {
      "epoch": 1.7333984375e-05,
      "model_forward_time": 1.735461711883545,
      "step": 92176
    },
    {
      "epoch": 1.7333984375e-05,
      "step": 92176,
      "training_step_time": 7.934398412704468
    },
    {
      "epoch": 1.7334938049316406e-05,
      "model_forward_time": 2.838869571685791,
      "step": 92177
    },
    {
      "epoch": 1.7334938049316406e-05,
      "step": 92177,
      "training_step_time": 8.619745016098022
    },
    {
      "epoch": 1.7335891723632813e-05,
      "model_forward_time": 1.7363770008087158,
      "step": 92178
    },
    {
      "epoch": 1.7335891723632813e-05,
      "step": 92178,
      "training_step_time": 9.433072805404663
    },
    {
      "epoch": 1.7336845397949217e-05,
      "model_forward_time": 1.743722915649414,
      "step": 92179
    },
    {
      "epoch": 1.7336845397949217e-05,
      "step": 92179,
      "training_step_time": 8.782709121704102
    },
    {
      "epoch": 1.7337799072265625e-05,
      "grad_norm": 0.08004933595657349,
      "learning_rate": 1.6630099428510015e-07,
      "loss": 0.0541,
      "step": 92180
    },
    {
      "epoch": 1.7337799072265625e-05,
      "model_forward_time": 1.7502148151397705,
      "step": 92180
    },
    {
      "dynamics_loss_avg": 0.06231193207204342,
      "epoch": 1.7337799072265625e-05,
      "step": 92180
    },
    {
      "action_loss_avg": 0.001471406506607309,
      "epoch": 1.7337799072265625e-05,
      "step": 92180
    },
    {
      "epoch": 1.7337799072265625e-05,
      "step": 92180,
      "training_step_time": 7.758507013320923
    },
    {
      "epoch": 1.7338752746582032e-05,
      "model_forward_time": 1.3599050045013428,
      "step": 92181
    },
    {
      "epoch": 1.7338752746582032e-05,
      "step": 92181,
      "training_step_time": 7.799379348754883
    },
    {
      "epoch": 1.7339706420898436e-05,
      "model_forward_time": 1.7471446990966797,
      "step": 92182
    },
    {
      "epoch": 1.7339706420898436e-05,
      "step": 92182,
      "training_step_time": 7.840126037597656
    },
    {
      "epoch": 1.7340660095214844e-05,
      "model_forward_time": 1.7314181327819824,
      "step": 92183
    },
    {
      "epoch": 1.7340660095214844e-05,
      "step": 92183,
      "training_step_time": 7.3533923625946045
    },
    {
      "epoch": 1.734161376953125e-05,
      "model_forward_time": 1.7529728412628174,
      "step": 92184
    },
    {
      "epoch": 1.734161376953125e-05,
      "step": 92184,
      "training_step_time": 8.722236394882202
    },
    {
      "epoch": 1.7342567443847655e-05,
      "model_forward_time": 1.7471075057983398,
      "step": 92185
    },
    {
      "epoch": 1.7342567443847655e-05,
      "step": 92185,
      "training_step_time": 8.567115068435669
    },
    {
      "epoch": 1.7343521118164062e-05,
      "model_forward_time": 1.7510101795196533,
      "step": 92186
    },
    {
      "epoch": 1.7343521118164062e-05,
      "step": 92186,
      "training_step_time": 8.617252588272095
    },
    {
      "epoch": 1.734447479248047e-05,
      "model_forward_time": 2.9137766361236572,
      "step": 92187
    },
    {
      "epoch": 1.734447479248047e-05,
      "step": 92187,
      "training_step_time": 7.861421585083008
    },
    {
      "epoch": 1.7345428466796874e-05,
      "model_forward_time": 1.7161743640899658,
      "step": 92188
    },
    {
      "epoch": 1.7345428466796874e-05,
      "step": 92188,
      "training_step_time": 9.748704433441162
    },
    {
      "epoch": 1.734638214111328e-05,
      "model_forward_time": 1.7594518661499023,
      "step": 92189
    },
    {
      "epoch": 1.734638214111328e-05,
      "step": 92189,
      "training_step_time": 8.638609647750854
    },
    {
      "epoch": 1.734733581542969e-05,
      "grad_norm": 0.10345324128866196,
      "learning_rate": 1.6587836398977152e-07,
      "loss": 0.0539,
      "step": 92190
    },
    {
      "epoch": 1.734733581542969e-05,
      "model_forward_time": 1.7323105335235596,
      "step": 92190
    },
    {
      "dynamics_loss_avg": 0.04875525906682014,
      "epoch": 1.734733581542969e-05,
      "step": 92190
    },
    {
      "action_loss_avg": 0.0009782798850210383,
      "epoch": 1.734733581542969e-05,
      "step": 92190
    },
    {
      "epoch": 1.734733581542969e-05,
      "step": 92190,
      "training_step_time": 8.2200345993042
    },
    {
      "epoch": 1.7348289489746093e-05,
      "model_forward_time": 1.7332780361175537,
      "step": 92191
    },
    {
      "epoch": 1.7348289489746093e-05,
      "step": 92191,
      "training_step_time": 7.651376008987427
    },
    {
      "epoch": 1.73492431640625e-05,
      "model_forward_time": 1.7442395687103271,
      "step": 92192
    },
    {
      "epoch": 1.73492431640625e-05,
      "step": 92192,
      "training_step_time": 7.3043882846832275
    },
    {
      "epoch": 1.7350196838378908e-05,
      "model_forward_time": 1.7338862419128418,
      "step": 92193
    },
    {
      "epoch": 1.7350196838378908e-05,
      "step": 92193,
      "training_step_time": 7.734579563140869
    },
    {
      "epoch": 1.735115051269531e-05,
      "model_forward_time": 1.7326304912567139,
      "step": 92194
    },
    {
      "epoch": 1.735115051269531e-05,
      "step": 92194,
      "training_step_time": 7.7511467933654785
    },
    {
      "epoch": 1.735210418701172e-05,
      "model_forward_time": 1.7598049640655518,
      "step": 92195
    },
    {
      "epoch": 1.735210418701172e-05,
      "step": 92195,
      "training_step_time": 7.79069709777832
    },
    {
      "epoch": 1.7353057861328126e-05,
      "model_forward_time": 2.103304386138916,
      "step": 92196
    },
    {
      "epoch": 1.7353057861328126e-05,
      "step": 92196,
      "training_step_time": 8.67306923866272
    },
    {
      "epoch": 1.735401153564453e-05,
      "model_forward_time": 1.727914810180664,
      "step": 92197
    },
    {
      "epoch": 1.735401153564453e-05,
      "step": 92197,
      "training_step_time": 8.280718803405762
    },
    {
      "epoch": 1.7354965209960938e-05,
      "model_forward_time": 1.7527744770050049,
      "step": 92198
    },
    {
      "epoch": 1.7354965209960938e-05,
      "step": 92198,
      "training_step_time": 7.774927616119385
    },
    {
      "epoch": 1.7355918884277345e-05,
      "model_forward_time": 1.3584723472595215,
      "step": 92199
    },
    {
      "epoch": 1.7355918884277345e-05,
      "step": 92199,
      "training_step_time": 8.512509822845459
    },
    {
      "epoch": 1.735687255859375e-05,
      "grad_norm": 0.07356331497430801,
      "learning_rate": 1.6545626234671763e-07,
      "loss": 0.0545,
      "step": 92200
    },
    {
      "epoch": 1.735687255859375e-05,
      "model_forward_time": 1.7296597957611084,
      "step": 92200
    },
    {
      "dynamics_loss_avg": 0.05443035569041967,
      "epoch": 1.735687255859375e-05,
      "step": 92200
    },
    {
      "action_loss_avg": 0.001106968120438978,
      "epoch": 1.735687255859375e-05,
      "step": 92200
    },
    {
      "epoch": 1.735687255859375e-05,
      "step": 92200,
      "training_step_time": 7.689570903778076
    },
    {
      "epoch": 1.7357826232910157e-05,
      "model_forward_time": 1.7373857498168945,
      "step": 92201
    },
    {
      "epoch": 1.7357826232910157e-05,
      "step": 92201,
      "training_step_time": 7.507188081741333
    },
    {
      "epoch": 1.7358779907226564e-05,
      "model_forward_time": 1.7415745258331299,
      "step": 92202
    },
    {
      "epoch": 1.7358779907226564e-05,
      "step": 92202,
      "training_step_time": 7.537912845611572
    },
    {
      "epoch": 1.7359733581542968e-05,
      "model_forward_time": 1.7361712455749512,
      "step": 92203
    },
    {
      "epoch": 1.7359733581542968e-05,
      "step": 92203,
      "training_step_time": 7.764711618423462
    },
    {
      "epoch": 1.7360687255859376e-05,
      "model_forward_time": 1.759406328201294,
      "step": 92204
    },
    {
      "epoch": 1.7360687255859376e-05,
      "step": 92204,
      "training_step_time": 8.76209545135498
    },
    {
      "epoch": 1.736164093017578e-05,
      "model_forward_time": 2.755796194076538,
      "step": 92205
    },
    {
      "epoch": 1.736164093017578e-05,
      "step": 92205,
      "training_step_time": 7.675349950790405
    },
    {
      "epoch": 1.7362594604492187e-05,
      "model_forward_time": 1.7323517799377441,
      "step": 92206
    },
    {
      "epoch": 1.7362594604492187e-05,
      "step": 92206,
      "training_step_time": 9.467876195907593
    },
    {
      "epoch": 1.7363548278808594e-05,
      "model_forward_time": 1.0317885875701904,
      "step": 92207
    },
    {
      "epoch": 1.7363548278808594e-05,
      "step": 92207,
      "training_step_time": 9.55172324180603
    },
    {
      "epoch": 1.7364501953125e-05,
      "model_forward_time": 1.7371513843536377,
      "step": 92208
    },
    {
      "epoch": 1.7364501953125e-05,
      "step": 92208,
      "training_step_time": 8.559277057647705
    },
    {
      "epoch": 1.7365455627441406e-05,
      "model_forward_time": 1.742910385131836,
      "step": 92209
    },
    {
      "epoch": 1.7365455627441406e-05,
      "step": 92209,
      "training_step_time": 7.705385208129883
    },
    {
      "epoch": 1.7366409301757813e-05,
      "grad_norm": 0.11224038898944855,
      "learning_rate": 1.650346894020971e-07,
      "loss": 0.0548,
      "step": 92210
    },
    {
      "epoch": 1.7366409301757813e-05,
      "model_forward_time": 1.731821060180664,
      "step": 92210
    },
    {
      "dynamics_loss_avg": 0.055106636695563795,
      "epoch": 1.7366409301757813e-05,
      "step": 92210
    },
    {
      "action_loss_avg": 0.0013968470477266238,
      "epoch": 1.7366409301757813e-05,
      "step": 92210
    },
    {
      "epoch": 1.7366409301757813e-05,
      "step": 92210,
      "training_step_time": 7.590003490447998
    },
    {
      "epoch": 1.7367362976074217e-05,
      "model_forward_time": 1.7415919303894043,
      "step": 92211
    },
    {
      "epoch": 1.7367362976074217e-05,
      "step": 92211,
      "training_step_time": 7.728188514709473
    },
    {
      "epoch": 1.7368316650390625e-05,
      "model_forward_time": 1.7418954372406006,
      "step": 92212
    },
    {
      "epoch": 1.7368316650390625e-05,
      "step": 92212,
      "training_step_time": 8.283380270004272
    },
    {
      "epoch": 1.7369270324707032e-05,
      "model_forward_time": 1.3842837810516357,
      "step": 92213
    },
    {
      "epoch": 1.7369270324707032e-05,
      "step": 92213,
      "training_step_time": 8.637350797653198
    },
    {
      "epoch": 1.7370223999023436e-05,
      "model_forward_time": 1.7393009662628174,
      "step": 92214
    },
    {
      "epoch": 1.7370223999023436e-05,
      "step": 92214,
      "training_step_time": 10.401689767837524
    },
    {
      "epoch": 1.7371177673339844e-05,
      "model_forward_time": 2.8205840587615967,
      "step": 92215
    },
    {
      "epoch": 1.7371177673339844e-05,
      "step": 92215,
      "training_step_time": 7.753031015396118
    },
    {
      "epoch": 1.737213134765625e-05,
      "model_forward_time": 1.7373688220977783,
      "step": 92216
    },
    {
      "epoch": 1.737213134765625e-05,
      "step": 92216,
      "training_step_time": 10.982044458389282
    },
    {
      "epoch": 1.7373085021972655e-05,
      "model_forward_time": 1.7584829330444336,
      "step": 92217
    },
    {
      "epoch": 1.7373085021972655e-05,
      "step": 92217,
      "training_step_time": 7.613745927810669
    },
    {
      "epoch": 1.7374038696289062e-05,
      "model_forward_time": 1.729248046875,
      "step": 92218
    },
    {
      "epoch": 1.7374038696289062e-05,
      "step": 92218,
      "training_step_time": 7.7553699016571045
    },
    {
      "epoch": 1.737499237060547e-05,
      "model_forward_time": 1.7368309497833252,
      "step": 92219
    },
    {
      "epoch": 1.737499237060547e-05,
      "step": 92219,
      "training_step_time": 7.55846905708313
    },
    {
      "epoch": 1.7375946044921874e-05,
      "grad_norm": 0.092470183968544,
      "learning_rate": 1.6461364520201471e-07,
      "loss": 0.0558,
      "step": 92220
    },
    {
      "epoch": 1.7375946044921874e-05,
      "model_forward_time": 1.7310504913330078,
      "step": 92220
    },
    {
      "dynamics_loss_avg": 0.051894845068454744,
      "epoch": 1.7375946044921874e-05,
      "step": 92220
    },
    {
      "action_loss_avg": 0.0015358555741840974,
      "epoch": 1.7375946044921874e-05,
      "step": 92220
    },
    {
      "epoch": 1.7375946044921874e-05,
      "step": 92220,
      "training_step_time": 8.416059732437134
    },
    {
      "epoch": 1.737689971923828e-05,
      "model_forward_time": 1.7413740158081055,
      "step": 92221
    },
    {
      "epoch": 1.737689971923828e-05,
      "step": 92221,
      "training_step_time": 7.842145681381226
    },
    {
      "epoch": 1.737785339355469e-05,
      "model_forward_time": 1.77425217628479,
      "step": 92222
    },
    {
      "epoch": 1.737785339355469e-05,
      "step": 92222,
      "training_step_time": 7.810152530670166
    },
    {
      "epoch": 1.7378807067871093e-05,
      "model_forward_time": 1.7398011684417725,
      "step": 92223
    },
    {
      "epoch": 1.7378807067871093e-05,
      "step": 92223,
      "training_step_time": 9.73683476448059
    },
    {
      "epoch": 1.73797607421875e-05,
      "model_forward_time": 2.801302433013916,
      "step": 92224
    },
    {
      "epoch": 1.73797607421875e-05,
      "step": 92224,
      "training_step_time": 9.496799230575562
    },
    {
      "epoch": 1.7380714416503908e-05,
      "model_forward_time": 1.7342793941497803,
      "step": 92225
    },
    {
      "epoch": 1.7380714416503908e-05,
      "step": 92225,
      "training_step_time": 8.363617181777954
    },
    {
      "epoch": 1.738166809082031e-05,
      "model_forward_time": 1.76104736328125,
      "step": 92226
    },
    {
      "epoch": 1.738166809082031e-05,
      "step": 92226,
      "training_step_time": 8.796664237976074
    },
    {
      "epoch": 1.738262176513672e-05,
      "model_forward_time": 1.7391877174377441,
      "step": 92227
    },
    {
      "epoch": 1.738262176513672e-05,
      "step": 92227,
      "training_step_time": 7.783324480056763
    },
    {
      "epoch": 1.7383575439453126e-05,
      "model_forward_time": 1.7347805500030518,
      "step": 92228
    },
    {
      "epoch": 1.7383575439453126e-05,
      "step": 92228,
      "training_step_time": 7.839468479156494
    },
    {
      "epoch": 1.738452911376953e-05,
      "model_forward_time": 1.746744155883789,
      "step": 92229
    },
    {
      "epoch": 1.738452911376953e-05,
      "step": 92229,
      "training_step_time": 7.838022470474243
    },
    {
      "epoch": 1.7385482788085938e-05,
      "grad_norm": 0.09150460362434387,
      "learning_rate": 1.6419312979251368e-07,
      "loss": 0.0549,
      "step": 92230
    },
    {
      "epoch": 1.7385482788085938e-05,
      "model_forward_time": 1.7277405261993408,
      "step": 92230
    },
    {
      "dynamics_loss_avg": 0.06252973079681397,
      "epoch": 1.7385482788085938e-05,
      "step": 92230
    },
    {
      "action_loss_avg": 0.0014992308104410767,
      "epoch": 1.7385482788085938e-05,
      "step": 92230
    },
    {
      "epoch": 1.7385482788085938e-05,
      "step": 92230,
      "training_step_time": 7.790290355682373
    },
    {
      "epoch": 1.7386436462402345e-05,
      "model_forward_time": 1.7575404644012451,
      "step": 92231
    },
    {
      "epoch": 1.7386436462402345e-05,
      "step": 92231,
      "training_step_time": 8.409823179244995
    },
    {
      "epoch": 1.738739013671875e-05,
      "model_forward_time": 1.7367362976074219,
      "step": 92232
    },
    {
      "epoch": 1.738739013671875e-05,
      "step": 92232,
      "training_step_time": 9.177184343338013
    },
    {
      "epoch": 1.7388343811035157e-05,
      "model_forward_time": 1.7318596839904785,
      "step": 92233
    },
    {
      "epoch": 1.7388343811035157e-05,
      "step": 92233,
      "training_step_time": 8.40360426902771
    },
    {
      "epoch": 1.7389297485351564e-05,
      "model_forward_time": 1.7436614036560059,
      "step": 92234
    },
    {
      "epoch": 1.7389297485351564e-05,
      "step": 92234,
      "training_step_time": 7.815277099609375
    },
    {
      "epoch": 1.7390251159667968e-05,
      "model_forward_time": 1.7538983821868896,
      "step": 92235
    },
    {
      "epoch": 1.7390251159667968e-05,
      "step": 92235,
      "training_step_time": 7.870319843292236
    },
    {
      "epoch": 1.7391204833984376e-05,
      "model_forward_time": 1.732637882232666,
      "step": 92236
    },
    {
      "epoch": 1.7391204833984376e-05,
      "step": 92236,
      "training_step_time": 7.804188966751099
    },
    {
      "epoch": 1.739215850830078e-05,
      "model_forward_time": 1.7417504787445068,
      "step": 92237
    },
    {
      "epoch": 1.739215850830078e-05,
      "step": 92237,
      "training_step_time": 7.8322224617004395
    },
    {
      "epoch": 1.7393112182617187e-05,
      "model_forward_time": 1.7388014793395996,
      "step": 92238
    },
    {
      "epoch": 1.7393112182617187e-05,
      "step": 92238,
      "training_step_time": 7.875861644744873
    },
    {
      "epoch": 1.7394065856933594e-05,
      "model_forward_time": 1.736422061920166,
      "step": 92239
    },
    {
      "epoch": 1.7394065856933594e-05,
      "step": 92239,
      "training_step_time": 7.945423603057861
    },
    {
      "epoch": 1.739501953125e-05,
      "grad_norm": 0.06932120770215988,
      "learning_rate": 1.6377314321958216e-07,
      "loss": 0.0553,
      "step": 92240
    },
    {
      "epoch": 1.739501953125e-05,
      "model_forward_time": 1.7426674365997314,
      "step": 92240
    },
    {
      "dynamics_loss_avg": 0.06788303107023239,
      "epoch": 1.739501953125e-05,
      "step": 92240
    },
    {
      "action_loss_avg": 0.002068843349115923,
      "epoch": 1.739501953125e-05,
      "step": 92240
    },
    {
      "epoch": 1.739501953125e-05,
      "step": 92240,
      "training_step_time": 7.759478569030762
    },
    {
      "epoch": 1.7395973205566406e-05,
      "model_forward_time": 1.7331347465515137,
      "step": 92241
    },
    {
      "epoch": 1.7395973205566406e-05,
      "step": 92241,
      "training_step_time": 8.655751943588257
    },
    {
      "epoch": 1.7396926879882813e-05,
      "model_forward_time": 1.7472515106201172,
      "step": 92242
    },
    {
      "epoch": 1.7396926879882813e-05,
      "step": 92242,
      "training_step_time": 8.252423763275146
    },
    {
      "epoch": 1.7397880554199217e-05,
      "model_forward_time": 2.910616159439087,
      "step": 92243
    },
    {
      "epoch": 1.7397880554199217e-05,
      "step": 92243,
      "training_step_time": 9.079396963119507
    },
    {
      "epoch": 1.7398834228515625e-05,
      "model_forward_time": 1.7643704414367676,
      "step": 92244
    },
    {
      "epoch": 1.7398834228515625e-05,
      "step": 92244,
      "training_step_time": 7.806781530380249
    },
    {
      "epoch": 1.7399787902832032e-05,
      "model_forward_time": 1.7431495189666748,
      "step": 92245
    },
    {
      "epoch": 1.7399787902832032e-05,
      "step": 92245,
      "training_step_time": 8.690677642822266
    },
    {
      "epoch": 1.7400741577148436e-05,
      "model_forward_time": 1.75223708152771,
      "step": 92246
    },
    {
      "epoch": 1.7400741577148436e-05,
      "step": 92246,
      "training_step_time": 8.313776731491089
    },
    {
      "epoch": 1.7401695251464844e-05,
      "model_forward_time": 1.7399609088897705,
      "step": 92247
    },
    {
      "epoch": 1.7401695251464844e-05,
      "step": 92247,
      "training_step_time": 7.870087623596191
    },
    {
      "epoch": 1.740264892578125e-05,
      "model_forward_time": 1.7495701313018799,
      "step": 92248
    },
    {
      "epoch": 1.740264892578125e-05,
      "step": 92248,
      "training_step_time": 7.749089241027832
    },
    {
      "epoch": 1.7403602600097655e-05,
      "model_forward_time": 1.7475569248199463,
      "step": 92249
    },
    {
      "epoch": 1.7403602600097655e-05,
      "step": 92249,
      "training_step_time": 7.93514084815979
    },
    {
      "epoch": 1.7404556274414062e-05,
      "grad_norm": 0.13909894227981567,
      "learning_rate": 1.6335368552914788e-07,
      "loss": 0.055,
      "step": 92250
    },
    {
      "epoch": 1.7404556274414062e-05,
      "model_forward_time": 1.7736396789550781,
      "step": 92250
    },
    {
      "dynamics_loss_avg": 0.04225100930780172,
      "epoch": 1.7404556274414062e-05,
      "step": 92250
    },
    {
      "action_loss_avg": 0.0013757964945398272,
      "epoch": 1.7404556274414062e-05,
      "step": 92250
    },
    {
      "epoch": 1.7404556274414062e-05,
      "step": 92250,
      "training_step_time": 7.9044060707092285
    },
    {
      "epoch": 1.740550994873047e-05,
      "model_forward_time": 1.7474687099456787,
      "step": 92251
    },
    {
      "epoch": 1.740550994873047e-05,
      "step": 92251,
      "training_step_time": 7.886708736419678
    },
    {
      "epoch": 1.7406463623046874e-05,
      "model_forward_time": 2.827316999435425,
      "step": 92252
    },
    {
      "epoch": 1.7406463623046874e-05,
      "step": 92252,
      "training_step_time": 8.718669414520264
    },
    {
      "epoch": 1.740741729736328e-05,
      "model_forward_time": 1.7870802879333496,
      "step": 92253
    },
    {
      "epoch": 1.740741729736328e-05,
      "step": 92253,
      "training_step_time": 9.577115535736084
    },
    {
      "epoch": 1.740837097167969e-05,
      "model_forward_time": 1.7526249885559082,
      "step": 92254
    },
    {
      "epoch": 1.740837097167969e-05,
      "step": 92254,
      "training_step_time": 9.274590015411377
    },
    {
      "epoch": 1.7409324645996093e-05,
      "model_forward_time": 1.7427234649658203,
      "step": 92255
    },
    {
      "epoch": 1.7409324645996093e-05,
      "step": 92255,
      "training_step_time": 9.918931007385254
    },
    {
      "epoch": 1.74102783203125e-05,
      "model_forward_time": 1.7443718910217285,
      "step": 92256
    },
    {
      "epoch": 1.74102783203125e-05,
      "step": 92256,
      "training_step_time": 7.736201047897339
    },
    {
      "epoch": 1.7411231994628908e-05,
      "model_forward_time": 1.740532636642456,
      "step": 92257
    },
    {
      "epoch": 1.7411231994628908e-05,
      "step": 92257,
      "training_step_time": 7.835763692855835
    },
    {
      "epoch": 1.741218566894531e-05,
      "model_forward_time": 1.7582669258117676,
      "step": 92258
    },
    {
      "epoch": 1.741218566894531e-05,
      "step": 92258,
      "training_step_time": 7.874408721923828
    },
    {
      "epoch": 1.741313934326172e-05,
      "model_forward_time": 1.7690856456756592,
      "step": 92259
    },
    {
      "epoch": 1.741313934326172e-05,
      "step": 92259,
      "training_step_time": 9.842217206954956
    },
    {
      "epoch": 1.7414093017578126e-05,
      "grad_norm": 0.07139553129673004,
      "learning_rate": 1.6293475676708358e-07,
      "loss": 0.0567,
      "step": 92260
    },
    {
      "epoch": 1.7414093017578126e-05,
      "model_forward_time": 1.7522757053375244,
      "step": 92260
    },
    {
      "dynamics_loss_avg": 0.04932588879019022,
      "epoch": 1.7414093017578126e-05,
      "step": 92260
    },
    {
      "action_loss_avg": 0.001397838181583211,
      "epoch": 1.7414093017578126e-05,
      "step": 92260
    },
    {
      "epoch": 1.7414093017578126e-05,
      "step": 92260,
      "training_step_time": 8.41357421875
    },
    {
      "epoch": 1.741504669189453e-05,
      "model_forward_time": 2.8274731636047363,
      "step": 92261
    },
    {
      "epoch": 1.741504669189453e-05,
      "step": 92261,
      "training_step_time": 8.39588737487793
    },
    {
      "epoch": 1.7416000366210938e-05,
      "model_forward_time": 1.7502961158752441,
      "step": 92262
    },
    {
      "epoch": 1.7416000366210938e-05,
      "step": 92262,
      "training_step_time": 8.807395458221436
    },
    {
      "epoch": 1.7416954040527345e-05,
      "model_forward_time": 1.7191271781921387,
      "step": 92263
    },
    {
      "epoch": 1.7416954040527345e-05,
      "step": 92263,
      "training_step_time": 7.779088973999023
    },
    {
      "epoch": 1.741790771484375e-05,
      "model_forward_time": 1.7302305698394775,
      "step": 92264
    },
    {
      "epoch": 1.741790771484375e-05,
      "step": 92264,
      "training_step_time": 7.700455665588379
    },
    {
      "epoch": 1.7418861389160157e-05,
      "model_forward_time": 1.7407050132751465,
      "step": 92265
    },
    {
      "epoch": 1.7418861389160157e-05,
      "step": 92265,
      "training_step_time": 8.69295597076416
    },
    {
      "epoch": 1.741981506347656e-05,
      "model_forward_time": 1.7368221282958984,
      "step": 92266
    },
    {
      "epoch": 1.741981506347656e-05,
      "step": 92266,
      "training_step_time": 7.7832770347595215
    },
    {
      "epoch": 1.7420768737792968e-05,
      "model_forward_time": 1.7379968166351318,
      "step": 92267
    },
    {
      "epoch": 1.7420768737792968e-05,
      "step": 92267,
      "training_step_time": 8.796342849731445
    },
    {
      "epoch": 1.7421722412109376e-05,
      "model_forward_time": 1.7581956386566162,
      "step": 92268
    },
    {
      "epoch": 1.7421722412109376e-05,
      "step": 92268,
      "training_step_time": 7.541086673736572
    },
    {
      "epoch": 1.742267608642578e-05,
      "model_forward_time": 1.7327601909637451,
      "step": 92269
    },
    {
      "epoch": 1.742267608642578e-05,
      "step": 92269,
      "training_step_time": 8.935838222503662
    },
    {
      "epoch": 1.7423629760742187e-05,
      "grad_norm": 0.07452213764190674,
      "learning_rate": 1.62516356979201e-07,
      "loss": 0.055,
      "step": 92270
    },
    {
      "epoch": 1.7423629760742187e-05,
      "model_forward_time": 2.8121490478515625,
      "step": 92270
    },
    {
      "dynamics_loss_avg": 0.04781645461916924,
      "epoch": 1.7423629760742187e-05,
      "step": 92270
    },
    {
      "action_loss_avg": 0.001015901385108009,
      "epoch": 1.7423629760742187e-05,
      "step": 92270
    },
    {
      "epoch": 1.7423629760742187e-05,
      "step": 92270,
      "training_step_time": 9.626709222793579
    },
    {
      "epoch": 1.7424583435058595e-05,
      "model_forward_time": 1.7600502967834473,
      "step": 92271
    },
    {
      "epoch": 1.7424583435058595e-05,
      "step": 92271,
      "training_step_time": 9.283353090286255
    },
    {
      "epoch": 1.7425537109375e-05,
      "model_forward_time": 1.7438979148864746,
      "step": 92272
    },
    {
      "epoch": 1.7425537109375e-05,
      "step": 92272,
      "training_step_time": 7.758060932159424
    },
    {
      "epoch": 1.7426490783691406e-05,
      "model_forward_time": 1.7337965965270996,
      "step": 92273
    },
    {
      "epoch": 1.7426490783691406e-05,
      "step": 92273,
      "training_step_time": 7.643156290054321
    },
    {
      "epoch": 1.7427444458007813e-05,
      "model_forward_time": 1.7325646877288818,
      "step": 92274
    },
    {
      "epoch": 1.7427444458007813e-05,
      "step": 92274,
      "training_step_time": 8.307005167007446
    },
    {
      "epoch": 1.7428398132324217e-05,
      "model_forward_time": 1.363046646118164,
      "step": 92275
    },
    {
      "epoch": 1.7428398132324217e-05,
      "step": 92275,
      "training_step_time": 7.707033395767212
    },
    {
      "epoch": 1.7429351806640625e-05,
      "model_forward_time": 1.7446765899658203,
      "step": 92276
    },
    {
      "epoch": 1.7429351806640625e-05,
      "step": 92276,
      "training_step_time": 8.318243026733398
    },
    {
      "epoch": 1.7430305480957032e-05,
      "model_forward_time": 1.7607531547546387,
      "step": 92277
    },
    {
      "epoch": 1.7430305480957032e-05,
      "step": 92277,
      "training_step_time": 7.5840232372283936
    },
    {
      "epoch": 1.7431259155273436e-05,
      "model_forward_time": 1.7366046905517578,
      "step": 92278
    },
    {
      "epoch": 1.7431259155273436e-05,
      "step": 92278,
      "training_step_time": 7.707532644271851
    },
    {
      "epoch": 1.7432212829589844e-05,
      "model_forward_time": 1.727952480316162,
      "step": 92279
    },
    {
      "epoch": 1.7432212829589844e-05,
      "step": 92279,
      "training_step_time": 8.776922702789307
    },
    {
      "epoch": 1.743316650390625e-05,
      "grad_norm": 0.08125229179859161,
      "learning_rate": 1.6209848621125623e-07,
      "loss": 0.0547,
      "step": 92280
    },
    {
      "epoch": 1.743316650390625e-05,
      "model_forward_time": 2.80117130279541,
      "step": 92280
    },
    {
      "dynamics_loss_avg": 0.06702518351376056,
      "epoch": 1.743316650390625e-05,
      "step": 92280
    },
    {
      "action_loss_avg": 0.0012933506921399384,
      "epoch": 1.743316650390625e-05,
      "step": 92280
    },
    {
      "epoch": 1.743316650390625e-05,
      "step": 92280,
      "training_step_time": 9.256046295166016
    },
    {
      "epoch": 1.7434120178222655e-05,
      "model_forward_time": 1.0249290466308594,
      "step": 92281
    },
    {
      "epoch": 1.7434120178222655e-05,
      "step": 92281,
      "training_step_time": 7.820069313049316
    },
    {
      "epoch": 1.7435073852539063e-05,
      "model_forward_time": 1.7329649925231934,
      "step": 92282
    },
    {
      "epoch": 1.7435073852539063e-05,
      "step": 92282,
      "training_step_time": 7.837531089782715
    },
    {
      "epoch": 1.743602752685547e-05,
      "model_forward_time": 1.7315540313720703,
      "step": 92283
    },
    {
      "epoch": 1.743602752685547e-05,
      "step": 92283,
      "training_step_time": 7.860405921936035
    },
    {
      "epoch": 1.7436981201171874e-05,
      "model_forward_time": 1.743006706237793,
      "step": 92284
    },
    {
      "epoch": 1.7436981201171874e-05,
      "step": 92284,
      "training_step_time": 7.798239469528198
    },
    {
      "epoch": 1.743793487548828e-05,
      "model_forward_time": 1.7326288223266602,
      "step": 92285
    },
    {
      "epoch": 1.743793487548828e-05,
      "step": 92285,
      "training_step_time": 7.756420612335205
    },
    {
      "epoch": 1.743888854980469e-05,
      "model_forward_time": 1.7309551239013672,
      "step": 92286
    },
    {
      "epoch": 1.743888854980469e-05,
      "step": 92286,
      "training_step_time": 7.688528537750244
    },
    {
      "epoch": 1.7439842224121093e-05,
      "model_forward_time": 1.7625513076782227,
      "step": 92287
    },
    {
      "epoch": 1.7439842224121093e-05,
      "step": 92287,
      "training_step_time": 9.620357275009155
    },
    {
      "epoch": 1.74407958984375e-05,
      "model_forward_time": 1.7270441055297852,
      "step": 92288
    },
    {
      "epoch": 1.74407958984375e-05,
      "step": 92288,
      "training_step_time": 8.945350885391235
    },
    {
      "epoch": 1.7441749572753908e-05,
      "model_forward_time": 2.826159954071045,
      "step": 92289
    },
    {
      "epoch": 1.7441749572753908e-05,
      "step": 92289,
      "training_step_time": 7.94198203086853
    },
    {
      "epoch": 1.744270324707031e-05,
      "grad_norm": 0.07398618012666702,
      "learning_rate": 1.616811445089478e-07,
      "loss": 0.0566,
      "step": 92290
    },
    {
      "epoch": 1.744270324707031e-05,
      "model_forward_time": 1.728069543838501,
      "step": 92290
    },
    {
      "dynamics_loss_avg": 0.045181343145668504,
      "epoch": 1.744270324707031e-05,
      "step": 92290
    },
    {
      "action_loss_avg": 0.0015232866018777714,
      "epoch": 1.744270324707031e-05,
      "step": 92290
    },
    {
      "epoch": 1.744270324707031e-05,
      "step": 92290,
      "training_step_time": 8.326460123062134
    },
    {
      "epoch": 1.744365692138672e-05,
      "model_forward_time": 1.725583791732788,
      "step": 92291
    },
    {
      "epoch": 1.744365692138672e-05,
      "step": 92291,
      "training_step_time": 8.695122957229614
    },
    {
      "epoch": 1.7444610595703127e-05,
      "model_forward_time": 1.730248212814331,
      "step": 92292
    },
    {
      "epoch": 1.7444610595703127e-05,
      "step": 92292,
      "training_step_time": 7.847280263900757
    },
    {
      "epoch": 1.744556427001953e-05,
      "model_forward_time": 1.728121280670166,
      "step": 92293
    },
    {
      "epoch": 1.744556427001953e-05,
      "step": 92293,
      "training_step_time": 8.328117847442627
    },
    {
      "epoch": 1.7446517944335938e-05,
      "model_forward_time": 1.7408497333526611,
      "step": 92294
    },
    {
      "epoch": 1.7446517944335938e-05,
      "step": 92294,
      "training_step_time": 7.8776795864105225
    },
    {
      "epoch": 1.7447471618652345e-05,
      "model_forward_time": 1.7376699447631836,
      "step": 92295
    },
    {
      "epoch": 1.7447471618652345e-05,
      "step": 92295,
      "training_step_time": 8.481852769851685
    },
    {
      "epoch": 1.744842529296875e-05,
      "model_forward_time": 1.7516915798187256,
      "step": 92296
    },
    {
      "epoch": 1.744842529296875e-05,
      "step": 92296,
      "training_step_time": 8.570960283279419
    },
    {
      "epoch": 1.7449378967285157e-05,
      "model_forward_time": 1.7310466766357422,
      "step": 92297
    },
    {
      "epoch": 1.7449378967285157e-05,
      "step": 92297,
      "training_step_time": 9.441602945327759
    },
    {
      "epoch": 1.745033264160156e-05,
      "model_forward_time": 2.8094820976257324,
      "step": 92298
    },
    {
      "epoch": 1.745033264160156e-05,
      "step": 92298,
      "training_step_time": 9.79100251197815
    },
    {
      "epoch": 1.7451286315917968e-05,
      "model_forward_time": 1.7370712757110596,
      "step": 92299
    },
    {
      "epoch": 1.7451286315917968e-05,
      "step": 92299,
      "training_step_time": 8.563075065612793
    },
    {
      "epoch": 1.7452239990234376e-05,
      "grad_norm": 0.061251793056726456,
      "learning_rate": 1.612643319179158e-07,
      "loss": 0.0553,
      "step": 92300
    },
    {
      "epoch": 1.7452239990234376e-05,
      "model_forward_time": 1.0161445140838623,
      "step": 92300
    },
    {
      "dynamics_loss_avg": 0.05850447807461023,
      "epoch": 1.7452239990234376e-05,
      "step": 92300
    },
    {
      "action_loss_avg": 0.0020336985209723933,
      "epoch": 1.7452239990234376e-05,
      "step": 92300
    },
    {
      "epoch": 1.7452239990234376e-05,
      "step": 92300,
      "training_step_time": 8.723395109176636
    },
    {
      "epoch": 1.745319366455078e-05,
      "model_forward_time": 1.7277820110321045,
      "step": 92301
    },
    {
      "epoch": 1.745319366455078e-05,
      "step": 92301,
      "training_step_time": 8.257676124572754
    },
    {
      "epoch": 1.7454147338867187e-05,
      "model_forward_time": 1.7331297397613525,
      "step": 92302
    },
    {
      "epoch": 1.7454147338867187e-05,
      "step": 92302,
      "training_step_time": 7.755104064941406
    },
    {
      "epoch": 1.7455101013183595e-05,
      "model_forward_time": 1.7333099842071533,
      "step": 92303
    },
    {
      "epoch": 1.7455101013183595e-05,
      "step": 92303,
      "training_step_time": 7.720342636108398
    },
    {
      "epoch": 1.74560546875e-05,
      "model_forward_time": 1.3522074222564697,
      "step": 92304
    },
    {
      "epoch": 1.74560546875e-05,
      "step": 92304,
      "training_step_time": 8.293371200561523
    },
    {
      "epoch": 1.7457008361816406e-05,
      "model_forward_time": 1.7524173259735107,
      "step": 92305
    },
    {
      "epoch": 1.7457008361816406e-05,
      "step": 92305,
      "training_step_time": 9.045592784881592
    },
    {
      "epoch": 1.7457962036132813e-05,
      "model_forward_time": 1.7369894981384277,
      "step": 92306
    },
    {
      "epoch": 1.7457962036132813e-05,
      "step": 92306,
      "training_step_time": 7.762792587280273
    },
    {
      "epoch": 1.7458915710449217e-05,
      "model_forward_time": 1.7514874935150146,
      "step": 92307
    },
    {
      "epoch": 1.7458915710449217e-05,
      "step": 92307,
      "training_step_time": 8.670781135559082
    },
    {
      "epoch": 1.7459869384765625e-05,
      "model_forward_time": 2.8717637062072754,
      "step": 92308
    },
    {
      "epoch": 1.7459869384765625e-05,
      "step": 92308,
      "training_step_time": 8.635961771011353
    },
    {
      "epoch": 1.7460823059082032e-05,
      "model_forward_time": 1.7232613563537598,
      "step": 92309
    },
    {
      "epoch": 1.7460823059082032e-05,
      "step": 92309,
      "training_step_time": 8.735633611679077
    },
    {
      "epoch": 1.7461776733398436e-05,
      "grad_norm": 0.0784311294555664,
      "learning_rate": 1.6084804848374046e-07,
      "loss": 0.0549,
      "step": 92310
    },
    {
      "epoch": 1.7461776733398436e-05,
      "model_forward_time": 1.3588593006134033,
      "step": 92310
    },
    {
      "dynamics_loss_avg": 0.058510661125183105,
      "epoch": 1.7461776733398436e-05,
      "step": 92310
    },
    {
      "action_loss_avg": 0.0013274698590976185,
      "epoch": 1.7461776733398436e-05,
      "step": 92310
    },
    {
      "epoch": 1.7461776733398436e-05,
      "step": 92310,
      "training_step_time": 7.805022716522217
    },
    {
      "epoch": 1.7462730407714844e-05,
      "model_forward_time": 1.7449777126312256,
      "step": 92311
    },
    {
      "epoch": 1.7462730407714844e-05,
      "step": 92311,
      "training_step_time": 7.798170328140259
    },
    {
      "epoch": 1.746368408203125e-05,
      "model_forward_time": 1.7332041263580322,
      "step": 92312
    },
    {
      "epoch": 1.746368408203125e-05,
      "step": 92312,
      "training_step_time": 7.793550968170166
    },
    {
      "epoch": 1.7464637756347655e-05,
      "model_forward_time": 1.7366526126861572,
      "step": 92313
    },
    {
      "epoch": 1.7464637756347655e-05,
      "step": 92313,
      "training_step_time": 8.374513387680054
    },
    {
      "epoch": 1.7465591430664063e-05,
      "model_forward_time": 1.7682318687438965,
      "step": 92314
    },
    {
      "epoch": 1.7465591430664063e-05,
      "step": 92314,
      "training_step_time": 7.568216562271118
    },
    {
      "epoch": 1.746654510498047e-05,
      "model_forward_time": 1.7299220561981201,
      "step": 92315
    },
    {
      "epoch": 1.746654510498047e-05,
      "step": 92315,
      "training_step_time": 8.462746143341064
    },
    {
      "epoch": 1.7467498779296874e-05,
      "model_forward_time": 1.7402839660644531,
      "step": 92316
    },
    {
      "epoch": 1.7467498779296874e-05,
      "step": 92316,
      "training_step_time": 7.7297632694244385
    },
    {
      "epoch": 1.746845245361328e-05,
      "model_forward_time": 2.8095436096191406,
      "step": 92317
    },
    {
      "epoch": 1.746845245361328e-05,
      "step": 92317,
      "training_step_time": 7.858144283294678
    },
    {
      "epoch": 1.746940612792969e-05,
      "model_forward_time": 1.7332594394683838,
      "step": 92318
    },
    {
      "epoch": 1.746940612792969e-05,
      "step": 92318,
      "training_step_time": 8.492550373077393
    },
    {
      "epoch": 1.7470359802246093e-05,
      "model_forward_time": 1.7421250343322754,
      "step": 92319
    },
    {
      "epoch": 1.7470359802246093e-05,
      "step": 92319,
      "training_step_time": 7.7815868854522705
    },
    {
      "epoch": 1.74713134765625e-05,
      "grad_norm": 0.0800124928355217,
      "learning_rate": 1.6043229425194817e-07,
      "loss": 0.0552,
      "step": 92320
    },
    {
      "epoch": 1.74713134765625e-05,
      "model_forward_time": 1.7476065158843994,
      "step": 92320
    },
    {
      "dynamics_loss_avg": 0.05070488126948476,
      "epoch": 1.74713134765625e-05,
      "step": 92320
    },
    {
      "action_loss_avg": 0.0008864215109497309,
      "epoch": 1.74713134765625e-05,
      "step": 92320
    },
    {
      "epoch": 1.74713134765625e-05,
      "step": 92320,
      "training_step_time": 7.83202862739563
    },
    {
      "epoch": 1.7472267150878908e-05,
      "model_forward_time": 1.7409260272979736,
      "step": 92321
    },
    {
      "epoch": 1.7472267150878908e-05,
      "step": 92321,
      "training_step_time": 8.531677484512329
    },
    {
      "epoch": 1.7473220825195312e-05,
      "model_forward_time": 1.7277801036834717,
      "step": 92322
    },
    {
      "epoch": 1.7473220825195312e-05,
      "step": 92322,
      "training_step_time": 8.207634687423706
    },
    {
      "epoch": 1.747417449951172e-05,
      "model_forward_time": 1.7614471912384033,
      "step": 92323
    },
    {
      "epoch": 1.747417449951172e-05,
      "step": 92323,
      "training_step_time": 7.696181535720825
    },
    {
      "epoch": 1.7475128173828127e-05,
      "model_forward_time": 1.7289457321166992,
      "step": 92324
    },
    {
      "epoch": 1.7475128173828127e-05,
      "step": 92324,
      "training_step_time": 8.701503276824951
    },
    {
      "epoch": 1.747608184814453e-05,
      "model_forward_time": 1.7431108951568604,
      "step": 92325
    },
    {
      "epoch": 1.747608184814453e-05,
      "step": 92325,
      "training_step_time": 9.490904331207275
    },
    {
      "epoch": 1.7477035522460938e-05,
      "model_forward_time": 1.7530159950256348,
      "step": 92326
    },
    {
      "epoch": 1.7477035522460938e-05,
      "step": 92326,
      "training_step_time": 8.342263460159302
    },
    {
      "epoch": 1.7477989196777345e-05,
      "model_forward_time": 1.7316749095916748,
      "step": 92327
    },
    {
      "epoch": 1.7477989196777345e-05,
      "step": 92327,
      "training_step_time": 7.767870903015137
    },
    {
      "epoch": 1.747894287109375e-05,
      "model_forward_time": 1.7316131591796875,
      "step": 92328
    },
    {
      "epoch": 1.747894287109375e-05,
      "step": 92328,
      "training_step_time": 7.711033582687378
    },
    {
      "epoch": 1.7479896545410157e-05,
      "model_forward_time": 1.7283647060394287,
      "step": 92329
    },
    {
      "epoch": 1.7479896545410157e-05,
      "step": 92329,
      "training_step_time": 7.791441202163696
    },
    {
      "epoch": 1.748085021972656e-05,
      "grad_norm": 0.07937008142471313,
      "learning_rate": 1.6001706926800309e-07,
      "loss": 0.0547,
      "step": 92330
    },
    {
      "epoch": 1.748085021972656e-05,
      "model_forward_time": 1.7332425117492676,
      "step": 92330
    },
    {
      "dynamics_loss_avg": 0.04985391143709421,
      "epoch": 1.748085021972656e-05,
      "step": 92330
    },
    {
      "action_loss_avg": 0.0014848719991277903,
      "epoch": 1.748085021972656e-05,
      "step": 92330
    },
    {
      "epoch": 1.748085021972656e-05,
      "step": 92330,
      "training_step_time": 8.721109867095947
    },
    {
      "epoch": 1.7481803894042968e-05,
      "model_forward_time": 1.7291603088378906,
      "step": 92331
    },
    {
      "epoch": 1.7481803894042968e-05,
      "step": 92331,
      "training_step_time": 8.197374820709229
    },
    {
      "epoch": 1.7482757568359376e-05,
      "model_forward_time": 1.7470073699951172,
      "step": 92332
    },
    {
      "epoch": 1.7482757568359376e-05,
      "step": 92332,
      "training_step_time": 7.700321197509766
    },
    {
      "epoch": 1.748371124267578e-05,
      "model_forward_time": 1.730217456817627,
      "step": 92333
    },
    {
      "epoch": 1.748371124267578e-05,
      "step": 92333,
      "training_step_time": 7.698382616043091
    },
    {
      "epoch": 1.7484664916992187e-05,
      "model_forward_time": 1.7351090908050537,
      "step": 92334
    },
    {
      "epoch": 1.7484664916992187e-05,
      "step": 92334,
      "training_step_time": 8.439159154891968
    },
    {
      "epoch": 1.7485618591308595e-05,
      "model_forward_time": 1.747525930404663,
      "step": 92335
    },
    {
      "epoch": 1.7485618591308595e-05,
      "step": 92335,
      "training_step_time": 8.53211259841919
    },
    {
      "epoch": 1.7486572265625e-05,
      "model_forward_time": 2.8309378623962402,
      "step": 92336
    },
    {
      "epoch": 1.7486572265625e-05,
      "step": 92336,
      "training_step_time": 7.814471483230591
    },
    {
      "epoch": 1.7487525939941406e-05,
      "model_forward_time": 1.7333226203918457,
      "step": 92337
    },
    {
      "epoch": 1.7487525939941406e-05,
      "step": 92337,
      "training_step_time": 9.215416431427002
    },
    {
      "epoch": 1.7488479614257813e-05,
      "model_forward_time": 1.733034610748291,
      "step": 92338
    },
    {
      "epoch": 1.7488479614257813e-05,
      "step": 92338,
      "training_step_time": 7.7241151332855225
    },
    {
      "epoch": 1.7489433288574217e-05,
      "model_forward_time": 1.725764513015747,
      "step": 92339
    },
    {
      "epoch": 1.7489433288574217e-05,
      "step": 92339,
      "training_step_time": 7.6952064037323
    },
    {
      "epoch": 1.7490386962890625e-05,
      "grad_norm": 0.08933982998132706,
      "learning_rate": 1.5960237357731445e-07,
      "loss": 0.0551,
      "step": 92340
    },
    {
      "epoch": 1.7490386962890625e-05,
      "model_forward_time": 1.7215728759765625,
      "step": 92340
    },
    {
      "dynamics_loss_avg": 0.05571927297860384,
      "epoch": 1.7490386962890625e-05,
      "step": 92340
    },
    {
      "action_loss_avg": 0.002026030415436253,
      "epoch": 1.7490386962890625e-05,
      "step": 92340
    },
    {
      "epoch": 1.7490386962890625e-05,
      "step": 92340,
      "training_step_time": 9.118959426879883
    },
    {
      "epoch": 1.7491340637207032e-05,
      "model_forward_time": 1.0256824493408203,
      "step": 92341
    },
    {
      "epoch": 1.7491340637207032e-05,
      "step": 92341,
      "training_step_time": 7.576397895812988
    },
    {
      "epoch": 1.7492294311523436e-05,
      "model_forward_time": 1.7181487083435059,
      "step": 92342
    },
    {
      "epoch": 1.7492294311523436e-05,
      "step": 92342,
      "training_step_time": 8.787434339523315
    },
    {
      "epoch": 1.7493247985839844e-05,
      "model_forward_time": 1.7241113185882568,
      "step": 92343
    },
    {
      "epoch": 1.7493247985839844e-05,
      "step": 92343,
      "training_step_time": 10.564372777938843
    },
    {
      "epoch": 1.749420166015625e-05,
      "model_forward_time": 1.722205400466919,
      "step": 92344
    },
    {
      "epoch": 1.749420166015625e-05,
      "step": 92344,
      "training_step_time": 7.833201169967651
    },
    {
      "epoch": 1.7495155334472655e-05,
      "model_forward_time": 2.8676047325134277,
      "step": 92345
    },
    {
      "epoch": 1.7495155334472655e-05,
      "step": 92345,
      "training_step_time": 8.323879957199097
    },
    {
      "epoch": 1.7496109008789063e-05,
      "model_forward_time": 1.7179265022277832,
      "step": 92346
    },
    {
      "epoch": 1.7496109008789063e-05,
      "step": 92346,
      "training_step_time": 7.807238340377808
    },
    {
      "epoch": 1.749706268310547e-05,
      "model_forward_time": 1.7090277671813965,
      "step": 92347
    },
    {
      "epoch": 1.749706268310547e-05,
      "step": 92347,
      "training_step_time": 7.779397249221802
    },
    {
      "epoch": 1.7498016357421874e-05,
      "model_forward_time": 1.7339706420898438,
      "step": 92348
    },
    {
      "epoch": 1.7498016357421874e-05,
      "step": 92348,
      "training_step_time": 8.483912229537964
    },
    {
      "epoch": 1.749897003173828e-05,
      "model_forward_time": 1.3511977195739746,
      "step": 92349
    },
    {
      "epoch": 1.749897003173828e-05,
      "step": 92349,
      "training_step_time": 7.7026872634887695
    },
    {
      "epoch": 1.749992370605469e-05,
      "grad_norm": 0.0758323222398758,
      "learning_rate": 1.5918820722523264e-07,
      "loss": 0.0568,
      "step": 92350
    },
    {
      "epoch": 1.749992370605469e-05,
      "model_forward_time": 1.7262685298919678,
      "step": 92350
    },
    {
      "dynamics_loss_avg": 0.07708415575325489,
      "epoch": 1.749992370605469e-05,
      "step": 92350
    },
    {
      "action_loss_avg": 0.0020874107372947036,
      "epoch": 1.749992370605469e-05,
      "step": 92350
    },
    {
      "epoch": 1.749992370605469e-05,
      "step": 92350,
      "training_step_time": 7.61909818649292
    },
    {
      "epoch": 1.7500877380371093e-05,
      "model_forward_time": 1.7418019771575928,
      "step": 92351
    },
    {
      "epoch": 1.7500877380371093e-05,
      "step": 92351,
      "training_step_time": 9.975420475006104
    },
    {
      "epoch": 1.75018310546875e-05,
      "model_forward_time": 1.7312345504760742,
      "step": 92352
    },
    {
      "epoch": 1.75018310546875e-05,
      "step": 92352,
      "training_step_time": 8.363430500030518
    },
    {
      "epoch": 1.7502784729003908e-05,
      "model_forward_time": 1.7313940525054932,
      "step": 92353
    },
    {
      "epoch": 1.7502784729003908e-05,
      "step": 92353,
      "training_step_time": 7.722998380661011
    },
    {
      "epoch": 1.7503738403320312e-05,
      "model_forward_time": 2.8054683208465576,
      "step": 92354
    },
    {
      "epoch": 1.7503738403320312e-05,
      "step": 92354,
      "training_step_time": 8.759827375411987
    },
    {
      "epoch": 1.750469207763672e-05,
      "model_forward_time": 1.7322666645050049,
      "step": 92355
    },
    {
      "epoch": 1.750469207763672e-05,
      "step": 92355,
      "training_step_time": 8.832186698913574
    },
    {
      "epoch": 1.7505645751953127e-05,
      "model_forward_time": 1.7318768501281738,
      "step": 92356
    },
    {
      "epoch": 1.7505645751953127e-05,
      "step": 92356,
      "training_step_time": 7.631525754928589
    },
    {
      "epoch": 1.750659942626953e-05,
      "model_forward_time": 1.7237517833709717,
      "step": 92357
    },
    {
      "epoch": 1.750659942626953e-05,
      "step": 92357,
      "training_step_time": 8.790699005126953
    },
    {
      "epoch": 1.7507553100585938e-05,
      "model_forward_time": 1.7270140647888184,
      "step": 92358
    },
    {
      "epoch": 1.7507553100585938e-05,
      "step": 92358,
      "training_step_time": 8.707225561141968
    },
    {
      "epoch": 1.7508506774902345e-05,
      "model_forward_time": 1.7298617362976074,
      "step": 92359
    },
    {
      "epoch": 1.7508506774902345e-05,
      "step": 92359,
      "training_step_time": 8.054846286773682
    },
    {
      "epoch": 1.750946044921875e-05,
      "grad_norm": 0.07386747002601624,
      "learning_rate": 1.587745702570509e-07,
      "loss": 0.056,
      "step": 92360
    },
    {
      "epoch": 1.750946044921875e-05,
      "model_forward_time": 1.7481627464294434,
      "step": 92360
    },
    {
      "dynamics_loss_avg": 0.054396902583539485,
      "epoch": 1.750946044921875e-05,
      "step": 92360
    },
    {
      "action_loss_avg": 0.0016432491538580507,
      "epoch": 1.750946044921875e-05,
      "step": 92360
    },
    {
      "epoch": 1.750946044921875e-05,
      "step": 92360,
      "training_step_time": 8.739374876022339
    },
    {
      "epoch": 1.7510414123535157e-05,
      "model_forward_time": 1.030651330947876,
      "step": 92361
    },
    {
      "epoch": 1.7510414123535157e-05,
      "step": 92361,
      "training_step_time": 8.480247497558594
    },
    {
      "epoch": 1.751136779785156e-05,
      "model_forward_time": 1.727712631225586,
      "step": 92362
    },
    {
      "epoch": 1.751136779785156e-05,
      "step": 92362,
      "training_step_time": 7.72811484336853
    },
    {
      "epoch": 1.751232147216797e-05,
      "model_forward_time": 1.74159836769104,
      "step": 92363
    },
    {
      "epoch": 1.751232147216797e-05,
      "step": 92363,
      "training_step_time": 8.778012037277222
    },
    {
      "epoch": 1.7513275146484376e-05,
      "model_forward_time": 2.831709384918213,
      "step": 92364
    },
    {
      "epoch": 1.7513275146484376e-05,
      "step": 92364,
      "training_step_time": 8.260339736938477
    },
    {
      "epoch": 1.751422882080078e-05,
      "model_forward_time": 1.724045991897583,
      "step": 92365
    },
    {
      "epoch": 1.751422882080078e-05,
      "step": 92365,
      "training_step_time": 7.74531364440918
    },
    {
      "epoch": 1.7515182495117187e-05,
      "model_forward_time": 1.7351713180541992,
      "step": 92366
    },
    {
      "epoch": 1.7515182495117187e-05,
      "step": 92366,
      "training_step_time": 7.675207614898682
    },
    {
      "epoch": 1.7516136169433595e-05,
      "model_forward_time": 1.7280969619750977,
      "step": 92367
    },
    {
      "epoch": 1.7516136169433595e-05,
      "step": 92367,
      "training_step_time": 7.635740041732788
    },
    {
      "epoch": 1.751708984375e-05,
      "model_forward_time": 1.7322173118591309,
      "step": 92368
    },
    {
      "epoch": 1.751708984375e-05,
      "step": 92368,
      "training_step_time": 7.481069564819336
    },
    {
      "epoch": 1.7518043518066406e-05,
      "model_forward_time": 1.7584447860717773,
      "step": 92369
    },
    {
      "epoch": 1.7518043518066406e-05,
      "step": 92369,
      "training_step_time": 7.576820135116577
    },
    {
      "epoch": 1.7518997192382813e-05,
      "grad_norm": 0.09357545524835587,
      "learning_rate": 1.5836146271800358e-07,
      "loss": 0.0562,
      "step": 92370
    },
    {
      "epoch": 1.7518997192382813e-05,
      "model_forward_time": 1.7381870746612549,
      "step": 92370
    },
    {
      "dynamics_loss_avg": 0.058713513612747195,
      "epoch": 1.7518997192382813e-05,
      "step": 92370
    },
    {
      "action_loss_avg": 0.0014944562135497108,
      "epoch": 1.7518997192382813e-05,
      "step": 92370
    },
    {
      "epoch": 1.7518997192382813e-05,
      "step": 92370,
      "training_step_time": 7.8253257274627686
    },
    {
      "epoch": 1.7519950866699217e-05,
      "model_forward_time": 1.7304608821868896,
      "step": 92371
    },
    {
      "epoch": 1.7519950866699217e-05,
      "step": 92371,
      "training_step_time": 8.745524644851685
    },
    {
      "epoch": 1.7520904541015625e-05,
      "model_forward_time": 1.7402880191802979,
      "step": 92372
    },
    {
      "epoch": 1.7520904541015625e-05,
      "step": 92372,
      "training_step_time": 8.258911609649658
    },
    {
      "epoch": 1.7521858215332032e-05,
      "model_forward_time": 2.823307514190674,
      "step": 92373
    },
    {
      "epoch": 1.7521858215332032e-05,
      "step": 92373,
      "training_step_time": 8.353600025177002
    },
    {
      "epoch": 1.7522811889648436e-05,
      "model_forward_time": 1.7409627437591553,
      "step": 92374
    },
    {
      "epoch": 1.7522811889648436e-05,
      "step": 92374,
      "training_step_time": 9.376014232635498
    },
    {
      "epoch": 1.7523765563964844e-05,
      "model_forward_time": 1.7303097248077393,
      "step": 92375
    },
    {
      "epoch": 1.7523765563964844e-05,
      "step": 92375,
      "training_step_time": 7.609525918960571
    },
    {
      "epoch": 1.752471923828125e-05,
      "model_forward_time": 1.7276241779327393,
      "step": 92376
    },
    {
      "epoch": 1.752471923828125e-05,
      "step": 92376,
      "training_step_time": 7.736860036849976
    },
    {
      "epoch": 1.7525672912597655e-05,
      "model_forward_time": 1.7284619808197021,
      "step": 92377
    },
    {
      "epoch": 1.7525672912597655e-05,
      "step": 92377,
      "training_step_time": 7.642111778259277
    },
    {
      "epoch": 1.7526626586914063e-05,
      "model_forward_time": 1.7544293403625488,
      "step": 92378
    },
    {
      "epoch": 1.7526626586914063e-05,
      "step": 92378,
      "training_step_time": 7.561674356460571
    },
    {
      "epoch": 1.752758026123047e-05,
      "model_forward_time": 1.7417306900024414,
      "step": 92379
    },
    {
      "epoch": 1.752758026123047e-05,
      "step": 92379,
      "training_step_time": 9.663679838180542
    },
    {
      "epoch": 1.7528533935546874e-05,
      "grad_norm": 0.09046528488397598,
      "learning_rate": 1.579488846532673e-07,
      "loss": 0.0567,
      "step": 92380
    },
    {
      "epoch": 1.7528533935546874e-05,
      "model_forward_time": 1.7313222885131836,
      "step": 92380
    },
    {
      "dynamics_loss_avg": 0.04666731003671885,
      "epoch": 1.7528533935546874e-05,
      "step": 92380
    },
    {
      "action_loss_avg": 0.0012888663535704837,
      "epoch": 1.7528533935546874e-05,
      "step": 92380
    },
    {
      "epoch": 1.7528533935546874e-05,
      "step": 92380,
      "training_step_time": 7.784139633178711
    },
    {
      "epoch": 1.752948760986328e-05,
      "model_forward_time": 1.735511064529419,
      "step": 92381
    },
    {
      "epoch": 1.752948760986328e-05,
      "step": 92381,
      "training_step_time": 8.479421615600586
    },
    {
      "epoch": 1.753044128417969e-05,
      "model_forward_time": 2.8111021518707275,
      "step": 92382
    },
    {
      "epoch": 1.753044128417969e-05,
      "step": 92382,
      "training_step_time": 7.839137077331543
    },
    {
      "epoch": 1.7531394958496093e-05,
      "model_forward_time": 1.765289068222046,
      "step": 92383
    },
    {
      "epoch": 1.7531394958496093e-05,
      "step": 92383,
      "training_step_time": 8.758952140808105
    },
    {
      "epoch": 1.75323486328125e-05,
      "model_forward_time": 1.7843642234802246,
      "step": 92384
    },
    {
      "epoch": 1.75323486328125e-05,
      "step": 92384,
      "training_step_time": 7.77551007270813
    },
    {
      "epoch": 1.7533302307128908e-05,
      "model_forward_time": 1.7734498977661133,
      "step": 92385
    },
    {
      "epoch": 1.7533302307128908e-05,
      "step": 92385,
      "training_step_time": 8.318533420562744
    },
    {
      "epoch": 1.7534255981445312e-05,
      "model_forward_time": 1.786376714706421,
      "step": 92386
    },
    {
      "epoch": 1.7534255981445312e-05,
      "step": 92386,
      "training_step_time": 7.7046661376953125
    },
    {
      "epoch": 1.753520965576172e-05,
      "model_forward_time": 1.755579948425293,
      "step": 92387
    },
    {
      "epoch": 1.753520965576172e-05,
      "step": 92387,
      "training_step_time": 8.637449502944946
    },
    {
      "epoch": 1.7536163330078127e-05,
      "model_forward_time": 1.846369743347168,
      "step": 92388
    },
    {
      "epoch": 1.7536163330078127e-05,
      "step": 92388,
      "training_step_time": 8.73621678352356
    },
    {
      "epoch": 1.753711700439453e-05,
      "model_forward_time": 1.7767717838287354,
      "step": 92389
    },
    {
      "epoch": 1.753711700439453e-05,
      "step": 92389,
      "training_step_time": 7.765251636505127
    },
    {
      "epoch": 1.7538070678710938e-05,
      "grad_norm": 0.10514574497938156,
      "learning_rate": 1.5753683610796155e-07,
      "loss": 0.0553,
      "step": 92390
    },
    {
      "epoch": 1.7538070678710938e-05,
      "model_forward_time": 1.7811644077301025,
      "step": 92390
    },
    {
      "dynamics_loss_avg": 0.04284468740224838,
      "epoch": 1.7538070678710938e-05,
      "step": 92390
    },
    {
      "action_loss_avg": 0.0009200422122376039,
      "epoch": 1.7538070678710938e-05,
      "step": 92390
    },
    {
      "epoch": 1.7538070678710938e-05,
      "step": 92390,
      "training_step_time": 9.534399271011353
    },
    {
      "epoch": 1.7539024353027342e-05,
      "model_forward_time": 2.3813605308532715,
      "step": 92391
    },
    {
      "epoch": 1.7539024353027342e-05,
      "step": 92391,
      "training_step_time": 8.421092510223389
    },
    {
      "epoch": 1.753997802734375e-05,
      "model_forward_time": 1.7821285724639893,
      "step": 92392
    },
    {
      "epoch": 1.753997802734375e-05,
      "step": 92392,
      "training_step_time": 7.795454025268555
    },
    {
      "epoch": 1.7540931701660157e-05,
      "model_forward_time": 1.7808911800384521,
      "step": 92393
    },
    {
      "epoch": 1.7540931701660157e-05,
      "step": 92393,
      "training_step_time": 7.777763366699219
    },
    {
      "epoch": 1.754188537597656e-05,
      "model_forward_time": 1.7816169261932373,
      "step": 92394
    },
    {
      "epoch": 1.754188537597656e-05,
      "step": 92394,
      "training_step_time": 7.781214714050293
    },
    {
      "epoch": 1.754283905029297e-05,
      "model_forward_time": 1.7774169445037842,
      "step": 92395
    },
    {
      "epoch": 1.754283905029297e-05,
      "step": 92395,
      "training_step_time": 7.795470237731934
    },
    {
      "epoch": 1.7543792724609376e-05,
      "model_forward_time": 1.7795374393463135,
      "step": 92396
    },
    {
      "epoch": 1.7543792724609376e-05,
      "step": 92396,
      "training_step_time": 8.198167085647583
    },
    {
      "epoch": 1.754474639892578e-05,
      "model_forward_time": 1.8446602821350098,
      "step": 92397
    },
    {
      "epoch": 1.754474639892578e-05,
      "step": 92397,
      "training_step_time": 8.624157905578613
    },
    {
      "epoch": 1.7545700073242187e-05,
      "model_forward_time": 1.7782623767852783,
      "step": 92398
    },
    {
      "epoch": 1.7545700073242187e-05,
      "step": 92398,
      "training_step_time": 7.874030828475952
    },
    {
      "epoch": 1.7546653747558595e-05,
      "model_forward_time": 1.7792577743530273,
      "step": 92399
    },
    {
      "epoch": 1.7546653747558595e-05,
      "step": 92399,
      "training_step_time": 9.803602933883667
    },
    {
      "epoch": 1.7547607421875e-05,
      "grad_norm": 0.1303839087486267,
      "learning_rate": 1.571253171271453e-07,
      "loss": 0.0571,
      "step": 92400
    },
    {
      "epoch": 1.7547607421875e-05,
      "model_forward_time": 1.778235912322998,
      "step": 92400
    },
    {
      "dynamics_loss_avg": 0.04852641504257917,
      "epoch": 1.7547607421875e-05,
      "step": 92400
    },
    {
      "action_loss_avg": 0.002071638996130787,
      "epoch": 1.7547607421875e-05,
      "step": 92400
    },
    {
      "epoch": 1.7547607421875e-05,
      "step": 92400,
      "training_step_time": 7.7719972133636475
    },
    {
      "epoch": 1.7548561096191406e-05,
      "model_forward_time": 2.388129949569702,
      "step": 92401
    },
    {
      "epoch": 1.7548561096191406e-05,
      "step": 92401,
      "training_step_time": 7.821051359176636
    },
    {
      "epoch": 1.7549514770507813e-05,
      "model_forward_time": 1.7809405326843262,
      "step": 92402
    },
    {
      "epoch": 1.7549514770507813e-05,
      "step": 92402,
      "training_step_time": 7.817837238311768
    },
    {
      "epoch": 1.7550468444824218e-05,
      "model_forward_time": 1.7818377017974854,
      "step": 92403
    },
    {
      "epoch": 1.7550468444824218e-05,
      "step": 92403,
      "training_step_time": 7.691019296646118
    },
    {
      "epoch": 1.7551422119140625e-05,
      "model_forward_time": 1.7745497226715088,
      "step": 92404
    },
    {
      "epoch": 1.7551422119140625e-05,
      "step": 92404,
      "training_step_time": 7.765605449676514
    },
    {
      "epoch": 1.7552375793457032e-05,
      "model_forward_time": 1.7835469245910645,
      "step": 92405
    },
    {
      "epoch": 1.7552375793457032e-05,
      "step": 92405,
      "training_step_time": 8.320812702178955
    },
    {
      "epoch": 1.7553329467773436e-05,
      "model_forward_time": 1.827563762664795,
      "step": 92406
    },
    {
      "epoch": 1.7553329467773436e-05,
      "step": 92406,
      "training_step_time": 8.793385744094849
    },
    {
      "epoch": 1.7554283142089844e-05,
      "model_forward_time": 1.771392583847046,
      "step": 92407
    },
    {
      "epoch": 1.7554283142089844e-05,
      "step": 92407,
      "training_step_time": 8.470507383346558
    },
    {
      "epoch": 1.755523681640625e-05,
      "model_forward_time": 1.774787187576294,
      "step": 92408
    },
    {
      "epoch": 1.755523681640625e-05,
      "step": 92408,
      "training_step_time": 8.676321983337402
    },
    {
      "epoch": 1.7556190490722655e-05,
      "model_forward_time": 1.7763557434082031,
      "step": 92409
    },
    {
      "epoch": 1.7556190490722655e-05,
      "step": 92409,
      "training_step_time": 10.273236989974976
    },
    {
      "epoch": 1.7557144165039063e-05,
      "grad_norm": 0.09855546057224274,
      "learning_rate": 1.5671432775582474e-07,
      "loss": 0.056,
      "step": 92410
    },
    {
      "epoch": 1.7557144165039063e-05,
      "model_forward_time": 2.431549549102783,
      "step": 92410
    },
    {
      "dynamics_loss_avg": 0.07286996264010667,
      "epoch": 1.7557144165039063e-05,
      "step": 92410
    },
    {
      "action_loss_avg": 0.0019221340538933874,
      "epoch": 1.7557144165039063e-05,
      "step": 92410
    },
    {
      "epoch": 1.7557144165039063e-05,
      "step": 92410,
      "training_step_time": 8.679611206054688
    },
    {
      "epoch": 1.755809783935547e-05,
      "model_forward_time": 1.7725591659545898,
      "step": 92411
    },
    {
      "epoch": 1.755809783935547e-05,
      "step": 92411,
      "training_step_time": 7.8793251514434814
    },
    {
      "epoch": 1.7559051513671874e-05,
      "model_forward_time": 1.7685215473175049,
      "step": 92412
    },
    {
      "epoch": 1.7559051513671874e-05,
      "step": 92412,
      "training_step_time": 8.512910842895508
    },
    {
      "epoch": 1.756000518798828e-05,
      "model_forward_time": 1.7849442958831787,
      "step": 92413
    },
    {
      "epoch": 1.756000518798828e-05,
      "step": 92413,
      "training_step_time": 7.830300569534302
    },
    {
      "epoch": 1.756095886230469e-05,
      "model_forward_time": 1.7776751518249512,
      "step": 92414
    },
    {
      "epoch": 1.756095886230469e-05,
      "step": 92414,
      "training_step_time": 8.553264617919922
    },
    {
      "epoch": 1.7561912536621093e-05,
      "model_forward_time": 1.7926948070526123,
      "step": 92415
    },
    {
      "epoch": 1.7561912536621093e-05,
      "step": 92415,
      "training_step_time": 8.773245096206665
    },
    {
      "epoch": 1.75628662109375e-05,
      "model_forward_time": 1.7779271602630615,
      "step": 92416
    },
    {
      "epoch": 1.75628662109375e-05,
      "step": 92416,
      "training_step_time": 7.86919903755188
    },
    {
      "epoch": 1.7563819885253908e-05,
      "model_forward_time": 1.0817031860351562,
      "step": 92417
    },
    {
      "epoch": 1.7563819885253908e-05,
      "step": 92417,
      "training_step_time": 7.75657057762146
    },
    {
      "epoch": 1.7564773559570312e-05,
      "model_forward_time": 1.7809078693389893,
      "step": 92418
    },
    {
      "epoch": 1.7564773559570312e-05,
      "step": 92418,
      "training_step_time": 7.791493892669678
    },
    {
      "epoch": 1.756572723388672e-05,
      "model_forward_time": 2.4502594470977783,
      "step": 92419
    },
    {
      "epoch": 1.756572723388672e-05,
      "step": 92419,
      "training_step_time": 9.268008708953857
    },
    {
      "epoch": 1.7566680908203127e-05,
      "grad_norm": 0.15413989126682281,
      "learning_rate": 1.563038680389428e-07,
      "loss": 0.0564,
      "step": 92420
    },
    {
      "epoch": 1.7566680908203127e-05,
      "model_forward_time": 1.7921288013458252,
      "step": 92420
    },
    {
      "dynamics_loss_avg": 0.05601628180593252,
      "epoch": 1.7566680908203127e-05,
      "step": 92420
    },
    {
      "action_loss_avg": 0.001413569157011807,
      "epoch": 1.7566680908203127e-05,
      "step": 92420
    },
    {
      "epoch": 1.7566680908203127e-05,
      "step": 92420,
      "training_step_time": 7.883671998977661
    },
    {
      "epoch": 1.756763458251953e-05,
      "model_forward_time": 1.7702877521514893,
      "step": 92421
    },
    {
      "epoch": 1.756763458251953e-05,
      "step": 92421,
      "training_step_time": 8.299689292907715
    },
    {
      "epoch": 1.7568588256835938e-05,
      "model_forward_time": 1.772456169128418,
      "step": 92422
    },
    {
      "epoch": 1.7568588256835938e-05,
      "step": 92422,
      "training_step_time": 7.793908357620239
    },
    {
      "epoch": 1.7569541931152342e-05,
      "model_forward_time": 1.7716872692108154,
      "step": 92423
    },
    {
      "epoch": 1.7569541931152342e-05,
      "step": 92423,
      "training_step_time": 7.7223896980285645
    },
    {
      "epoch": 1.757049560546875e-05,
      "model_forward_time": 1.7885010242462158,
      "step": 92424
    },
    {
      "epoch": 1.757049560546875e-05,
      "step": 92424,
      "training_step_time": 7.8655760288238525
    },
    {
      "epoch": 1.7571449279785157e-05,
      "model_forward_time": 1.834287166595459,
      "step": 92425
    },
    {
      "epoch": 1.7571449279785157e-05,
      "step": 92425,
      "training_step_time": 9.606453895568848
    },
    {
      "epoch": 1.757240295410156e-05,
      "model_forward_time": 1.7810652256011963,
      "step": 92426
    },
    {
      "epoch": 1.757240295410156e-05,
      "step": 92426,
      "training_step_time": 8.496718645095825
    },
    {
      "epoch": 1.757335662841797e-05,
      "model_forward_time": 1.7805449962615967,
      "step": 92427
    },
    {
      "epoch": 1.757335662841797e-05,
      "step": 92427,
      "training_step_time": 7.842470169067383
    },
    {
      "epoch": 1.7574310302734376e-05,
      "model_forward_time": 2.4170892238616943,
      "step": 92428
    },
    {
      "epoch": 1.7574310302734376e-05,
      "step": 92428,
      "training_step_time": 7.886614084243774
    },
    {
      "epoch": 1.757526397705078e-05,
      "model_forward_time": 1.7739310264587402,
      "step": 92429
    },
    {
      "epoch": 1.757526397705078e-05,
      "step": 92429,
      "training_step_time": 8.404322624206543
    },
    {
      "epoch": 1.7576217651367187e-05,
      "grad_norm": 0.201113760471344,
      "learning_rate": 1.5589393802138808e-07,
      "loss": 0.0558,
      "step": 92430
    },
    {
      "epoch": 1.7576217651367187e-05,
      "model_forward_time": 1.7809133529663086,
      "step": 92430
    },
    {
      "dynamics_loss_avg": 0.05272930040955544,
      "epoch": 1.7576217651367187e-05,
      "step": 92430
    },
    {
      "action_loss_avg": 0.001412351056933403,
      "epoch": 1.7576217651367187e-05,
      "step": 92430
    },
    {
      "epoch": 1.7576217651367187e-05,
      "step": 92430,
      "training_step_time": 8.747534275054932
    },
    {
      "epoch": 1.7577171325683595e-05,
      "model_forward_time": 1.7779655456542969,
      "step": 92431
    },
    {
      "epoch": 1.7577171325683595e-05,
      "step": 92431,
      "training_step_time": 7.849507570266724
    },
    {
      "epoch": 1.7578125e-05,
      "model_forward_time": 1.7830479145050049,
      "step": 92432
    },
    {
      "epoch": 1.7578125e-05,
      "step": 92432,
      "training_step_time": 7.815840721130371
    },
    {
      "epoch": 1.7579078674316406e-05,
      "model_forward_time": 1.7770729064941406,
      "step": 92433
    },
    {
      "epoch": 1.7579078674316406e-05,
      "step": 92433,
      "training_step_time": 8.272458791732788
    },
    {
      "epoch": 1.7580032348632814e-05,
      "model_forward_time": 1.842336654663086,
      "step": 92434
    },
    {
      "epoch": 1.7580032348632814e-05,
      "step": 92434,
      "training_step_time": 7.948915243148804
    },
    {
      "epoch": 1.7580986022949218e-05,
      "model_forward_time": 1.7708699703216553,
      "step": 92435
    },
    {
      "epoch": 1.7580986022949218e-05,
      "step": 92435,
      "training_step_time": 10.694536447525024
    },
    {
      "epoch": 1.7581939697265625e-05,
      "model_forward_time": 1.7738635540008545,
      "step": 92436
    },
    {
      "epoch": 1.7581939697265625e-05,
      "step": 92436,
      "training_step_time": 8.35496187210083
    },
    {
      "epoch": 1.7582893371582032e-05,
      "model_forward_time": 1.761171579360962,
      "step": 92437
    },
    {
      "epoch": 1.7582893371582032e-05,
      "step": 92437,
      "training_step_time": 8.355139017105103
    },
    {
      "epoch": 1.7583847045898436e-05,
      "model_forward_time": 1.768080234527588,
      "step": 92438
    },
    {
      "epoch": 1.7583847045898436e-05,
      "step": 92438,
      "training_step_time": 8.476289510726929
    },
    {
      "epoch": 1.7584800720214844e-05,
      "model_forward_time": 1.7908928394317627,
      "step": 92439
    },
    {
      "epoch": 1.7584800720214844e-05,
      "step": 92439,
      "training_step_time": 7.881421804428101
    },
    {
      "epoch": 1.758575439453125e-05,
      "grad_norm": 0.09205067902803421,
      "learning_rate": 1.554845377479891e-07,
      "loss": 0.0563,
      "step": 92440
    },
    {
      "epoch": 1.758575439453125e-05,
      "model_forward_time": 1.7774438858032227,
      "step": 92440
    },
    {
      "dynamics_loss_avg": 0.06252560783177614,
      "epoch": 1.758575439453125e-05,
      "step": 92440
    },
    {
      "action_loss_avg": 0.002662998941377737,
      "epoch": 1.758575439453125e-05,
      "step": 92440
    },
    {
      "epoch": 1.758575439453125e-05,
      "step": 92440,
      "training_step_time": 7.6941163539886475
    },
    {
      "epoch": 1.7586708068847655e-05,
      "model_forward_time": 1.7680587768554688,
      "step": 92441
    },
    {
      "epoch": 1.7586708068847655e-05,
      "step": 92441,
      "training_step_time": 8.735806703567505
    },
    {
      "epoch": 1.7587661743164063e-05,
      "model_forward_time": 1.7768330574035645,
      "step": 92442
    },
    {
      "epoch": 1.7587661743164063e-05,
      "step": 92442,
      "training_step_time": 7.733338117599487
    },
    {
      "epoch": 1.758861541748047e-05,
      "model_forward_time": 1.839806079864502,
      "step": 92443
    },
    {
      "epoch": 1.758861541748047e-05,
      "step": 92443,
      "training_step_time": 7.942556381225586
    },
    {
      "epoch": 1.7589569091796874e-05,
      "model_forward_time": 1.7762279510498047,
      "step": 92444
    },
    {
      "epoch": 1.7589569091796874e-05,
      "step": 92444,
      "training_step_time": 8.375462770462036
    },
    {
      "epoch": 1.759052276611328e-05,
      "model_forward_time": 1.7787694931030273,
      "step": 92445
    },
    {
      "epoch": 1.759052276611328e-05,
      "step": 92445,
      "training_step_time": 7.700035333633423
    },
    {
      "epoch": 1.759147644042969e-05,
      "model_forward_time": 1.7686851024627686,
      "step": 92446
    },
    {
      "epoch": 1.759147644042969e-05,
      "step": 92446,
      "training_step_time": 9.15619444847107
    },
    {
      "epoch": 1.7592430114746093e-05,
      "model_forward_time": 2.417645215988159,
      "step": 92447
    },
    {
      "epoch": 1.7592430114746093e-05,
      "step": 92447,
      "training_step_time": 7.738110542297363
    },
    {
      "epoch": 1.75933837890625e-05,
      "model_forward_time": 1.7783854007720947,
      "step": 92448
    },
    {
      "epoch": 1.75933837890625e-05,
      "step": 92448,
      "training_step_time": 7.780803918838501
    },
    {
      "epoch": 1.7594337463378908e-05,
      "model_forward_time": 1.7693452835083008,
      "step": 92449
    },
    {
      "epoch": 1.7594337463378908e-05,
      "step": 92449,
      "training_step_time": 8.7869873046875
    },
    {
      "epoch": 1.7595291137695312e-05,
      "grad_norm": 0.07832735031843185,
      "learning_rate": 1.550756672635173e-07,
      "loss": 0.0562,
      "step": 92450
    },
    {
      "epoch": 1.7595291137695312e-05,
      "model_forward_time": 1.7686927318572998,
      "step": 92450
    },
    {
      "dynamics_loss_avg": 0.05725373886525631,
      "epoch": 1.7595291137695312e-05,
      "step": 92450
    },
    {
      "action_loss_avg": 0.002255687021533959,
      "epoch": 1.7595291137695312e-05,
      "step": 92450
    },
    {
      "epoch": 1.7595291137695312e-05,
      "step": 92450,
      "training_step_time": 7.744817495346069
    },
    {
      "epoch": 1.759624481201172e-05,
      "model_forward_time": 1.7737038135528564,
      "step": 92451
    },
    {
      "epoch": 1.759624481201172e-05,
      "step": 92451,
      "training_step_time": 7.5607404708862305
    },
    {
      "epoch": 1.7597198486328127e-05,
      "model_forward_time": 1.8255956172943115,
      "step": 92452
    },
    {
      "epoch": 1.7597198486328127e-05,
      "step": 92452,
      "training_step_time": 7.739991903305054
    },
    {
      "epoch": 1.759815216064453e-05,
      "model_forward_time": 1.7822611331939697,
      "step": 92453
    },
    {
      "epoch": 1.759815216064453e-05,
      "step": 92453,
      "training_step_time": 8.986679553985596
    },
    {
      "epoch": 1.7599105834960938e-05,
      "model_forward_time": 1.7722516059875488,
      "step": 92454
    },
    {
      "epoch": 1.7599105834960938e-05,
      "step": 92454,
      "training_step_time": 8.739567041397095
    },
    {
      "epoch": 1.7600059509277342e-05,
      "model_forward_time": 1.7716398239135742,
      "step": 92455
    },
    {
      "epoch": 1.7600059509277342e-05,
      "step": 92455,
      "training_step_time": 9.595128774642944
    },
    {
      "epoch": 1.760101318359375e-05,
      "model_forward_time": 2.3788745403289795,
      "step": 92456
    },
    {
      "epoch": 1.760101318359375e-05,
      "step": 92456,
      "training_step_time": 8.720400333404541
    },
    {
      "epoch": 1.7601966857910157e-05,
      "model_forward_time": 1.7718837261199951,
      "step": 92457
    },
    {
      "epoch": 1.7601966857910157e-05,
      "step": 92457,
      "training_step_time": 7.696449518203735
    },
    {
      "epoch": 1.760292053222656e-05,
      "model_forward_time": 1.7707242965698242,
      "step": 92458
    },
    {
      "epoch": 1.760292053222656e-05,
      "step": 92458,
      "training_step_time": 8.308252096176147
    },
    {
      "epoch": 1.760387420654297e-05,
      "model_forward_time": 1.75831937789917,
      "step": 92459
    },
    {
      "epoch": 1.760387420654297e-05,
      "step": 92459,
      "training_step_time": 7.69866156578064
    },
    {
      "epoch": 1.7604827880859376e-05,
      "grad_norm": 0.09289142489433289,
      "learning_rate": 1.5466732661268636e-07,
      "loss": 0.0572,
      "step": 92460
    },
    {
      "epoch": 1.7604827880859376e-05,
      "model_forward_time": 1.7907261848449707,
      "step": 92460
    },
    {
      "dynamics_loss_avg": 0.049491670727729795,
      "epoch": 1.7604827880859376e-05,
      "step": 92460
    },
    {
      "action_loss_avg": 0.0012395203055348247,
      "epoch": 1.7604827880859376e-05,
      "step": 92460
    },
    {
      "epoch": 1.7604827880859376e-05,
      "step": 92460,
      "training_step_time": 7.665014982223511
    },
    {
      "epoch": 1.760578155517578e-05,
      "model_forward_time": 1.809659719467163,
      "step": 92461
    },
    {
      "epoch": 1.760578155517578e-05,
      "step": 92461,
      "training_step_time": 7.732153415679932
    },
    {
      "epoch": 1.7606735229492187e-05,
      "model_forward_time": 1.7798950672149658,
      "step": 92462
    },
    {
      "epoch": 1.7606735229492187e-05,
      "step": 92462,
      "training_step_time": 7.762084245681763
    },
    {
      "epoch": 1.7607688903808595e-05,
      "model_forward_time": 1.79400634765625,
      "step": 92463
    },
    {
      "epoch": 1.7607688903808595e-05,
      "step": 92463,
      "training_step_time": 11.286591291427612
    },
    {
      "epoch": 1.7608642578125e-05,
      "model_forward_time": 1.7814528942108154,
      "step": 92464
    },
    {
      "epoch": 1.7608642578125e-05,
      "step": 92464,
      "training_step_time": 6.744272470474243
    },
    {
      "epoch": 1.7609596252441406e-05,
      "model_forward_time": 1.785900354385376,
      "step": 92465
    },
    {
      "epoch": 1.7609596252441406e-05,
      "step": 92465,
      "training_step_time": 7.589151382446289
    },
    {
      "epoch": 1.7610549926757814e-05,
      "model_forward_time": 1.7800228595733643,
      "step": 92466
    },
    {
      "epoch": 1.7610549926757814e-05,
      "step": 92466,
      "training_step_time": 8.404893159866333
    },
    {
      "epoch": 1.7611503601074218e-05,
      "model_forward_time": 1.0866003036499023,
      "step": 92467
    },
    {
      "epoch": 1.7611503601074218e-05,
      "step": 92467,
      "training_step_time": 7.661668062210083
    },
    {
      "epoch": 1.7612457275390625e-05,
      "model_forward_time": 1.8142719268798828,
      "step": 92468
    },
    {
      "epoch": 1.7612457275390625e-05,
      "step": 92468,
      "training_step_time": 6.789571046829224
    },
    {
      "epoch": 1.7613410949707032e-05,
      "model_forward_time": 1.762134075164795,
      "step": 92469
    },
    {
      "epoch": 1.7613410949707032e-05,
      "step": 92469,
      "training_step_time": 6.779524326324463
    },
    {
      "epoch": 1.7614364624023436e-05,
      "grad_norm": 0.4334115982055664,
      "learning_rate": 1.5425951584015165e-07,
      "loss": 0.0571,
      "step": 92470
    },
    {
      "epoch": 1.7614364624023436e-05,
      "model_forward_time": 1.7761867046356201,
      "step": 92470
    },
    {
      "dynamics_loss_avg": 0.05586647773161531,
      "epoch": 1.7614364624023436e-05,
      "step": 92470
    },
    {
      "action_loss_avg": 0.0023680101032368837,
      "epoch": 1.7614364624023436e-05,
      "step": 92470
    },
    {
      "epoch": 1.7614364624023436e-05,
      "step": 92470,
      "training_step_time": 6.78741455078125
    },
    {
      "epoch": 1.7615318298339844e-05,
      "model_forward_time": 1.7792320251464844,
      "step": 92471
    },
    {
      "epoch": 1.7615318298339844e-05,
      "step": 92471,
      "training_step_time": 6.800408124923706
    },
    {
      "epoch": 1.761627197265625e-05,
      "model_forward_time": 1.7832796573638916,
      "step": 92472
    },
    {
      "epoch": 1.761627197265625e-05,
      "step": 92472,
      "training_step_time": 6.776913166046143
    },
    {
      "epoch": 1.7617225646972655e-05,
      "model_forward_time": 2.3952512741088867,
      "step": 92473
    },
    {
      "epoch": 1.7617225646972655e-05,
      "step": 92473,
      "training_step_time": 8.695371150970459
    },
    {
      "epoch": 1.7618179321289063e-05,
      "model_forward_time": 1.772592306137085,
      "step": 92474
    },
    {
      "epoch": 1.7618179321289063e-05,
      "step": 92474,
      "training_step_time": 7.506195306777954
    },
    {
      "epoch": 1.761913299560547e-05,
      "model_forward_time": 1.7729973793029785,
      "step": 92475
    },
    {
      "epoch": 1.761913299560547e-05,
      "step": 92475,
      "training_step_time": 9.293998003005981
    },
    {
      "epoch": 1.7620086669921874e-05,
      "model_forward_time": 1.7732250690460205,
      "step": 92476
    },
    {
      "epoch": 1.7620086669921874e-05,
      "step": 92476,
      "training_step_time": 7.333694219589233
    },
    {
      "epoch": 1.762104034423828e-05,
      "model_forward_time": 1.7971532344818115,
      "step": 92477
    },
    {
      "epoch": 1.762104034423828e-05,
      "step": 92477,
      "training_step_time": 6.781792402267456
    },
    {
      "epoch": 1.762199401855469e-05,
      "model_forward_time": 1.7752954959869385,
      "step": 92478
    },
    {
      "epoch": 1.762199401855469e-05,
      "step": 92478,
      "training_step_time": 6.7868170738220215
    },
    {
      "epoch": 1.7622947692871093e-05,
      "model_forward_time": 1.7736856937408447,
      "step": 92479
    },
    {
      "epoch": 1.7622947692871093e-05,
      "step": 92479,
      "training_step_time": 6.7743189334869385
    },
    {
      "epoch": 1.76239013671875e-05,
      "grad_norm": 0.0643586590886116,
      "learning_rate": 1.5385223499051027e-07,
      "loss": 0.0575,
      "step": 92480
    },
    {
      "epoch": 1.76239013671875e-05,
      "model_forward_time": 1.7761962413787842,
      "step": 92480
    },
    {
      "dynamics_loss_avg": 0.06284649055451155,
      "epoch": 1.76239013671875e-05,
      "step": 92480
    },
    {
      "action_loss_avg": 0.0011605576262809336,
      "epoch": 1.76239013671875e-05,
      "step": 92480
    },
    {
      "epoch": 1.76239013671875e-05,
      "step": 92480,
      "training_step_time": 6.794018030166626
    },
    {
      "epoch": 1.7624855041503908e-05,
      "model_forward_time": 1.7801415920257568,
      "step": 92481
    },
    {
      "epoch": 1.7624855041503908e-05,
      "step": 92481,
      "training_step_time": 6.808398962020874
    },
    {
      "epoch": 1.7625808715820312e-05,
      "model_forward_time": 2.3894999027252197,
      "step": 92482
    },
    {
      "epoch": 1.7625808715820312e-05,
      "step": 92482,
      "training_step_time": 7.820775508880615
    },
    {
      "epoch": 1.762676239013672e-05,
      "model_forward_time": 1.7570385932922363,
      "step": 92483
    },
    {
      "epoch": 1.762676239013672e-05,
      "step": 92483,
      "training_step_time": 7.532885313034058
    },
    {
      "epoch": 1.7627716064453127e-05,
      "model_forward_time": 1.7871694564819336,
      "step": 92484
    },
    {
      "epoch": 1.7627716064453127e-05,
      "step": 92484,
      "training_step_time": 7.420794248580933
    },
    {
      "epoch": 1.762866973876953e-05,
      "model_forward_time": 1.7788147926330566,
      "step": 92485
    },
    {
      "epoch": 1.762866973876953e-05,
      "step": 92485,
      "training_step_time": 7.739200830459595
    },
    {
      "epoch": 1.7629623413085938e-05,
      "model_forward_time": 1.7750513553619385,
      "step": 92486
    },
    {
      "epoch": 1.7629623413085938e-05,
      "step": 92486,
      "training_step_time": 6.8166069984436035
    },
    {
      "epoch": 1.7630577087402342e-05,
      "model_forward_time": 1.775881290435791,
      "step": 92487
    },
    {
      "epoch": 1.7630577087402342e-05,
      "step": 92487,
      "training_step_time": 6.820946931838989
    },
    {
      "epoch": 1.763153076171875e-05,
      "model_forward_time": 1.7739689350128174,
      "step": 92488
    },
    {
      "epoch": 1.763153076171875e-05,
      "step": 92488,
      "training_step_time": 6.629820108413696
    },
    {
      "epoch": 1.7632484436035157e-05,
      "model_forward_time": 1.7791471481323242,
      "step": 92489
    },
    {
      "epoch": 1.7632484436035157e-05,
      "step": 92489,
      "training_step_time": 6.758158445358276
    },
    {
      "epoch": 1.763343811035156e-05,
      "grad_norm": 0.09033401310443878,
      "learning_rate": 1.534454841083033e-07,
      "loss": 0.0594,
      "step": 92490
    },
    {
      "epoch": 1.763343811035156e-05,
      "model_forward_time": 1.7779436111450195,
      "step": 92490
    },
    {
      "dynamics_loss_avg": 0.05002512019127607,
      "epoch": 1.763343811035156e-05,
      "step": 92490
    },
    {
      "action_loss_avg": 0.0013444161624647677,
      "epoch": 1.763343811035156e-05,
      "step": 92490
    },
    {
      "epoch": 1.763343811035156e-05,
      "step": 92490,
      "training_step_time": 6.715722322463989
    },
    {
      "epoch": 1.763439178466797e-05,
      "model_forward_time": 2.4170114994049072,
      "step": 92491
    },
    {
      "epoch": 1.763439178466797e-05,
      "step": 92491,
      "training_step_time": 8.671692609786987
    },
    {
      "epoch": 1.7635345458984376e-05,
      "model_forward_time": 1.776087999343872,
      "step": 92492
    },
    {
      "epoch": 1.7635345458984376e-05,
      "step": 92492,
      "training_step_time": 7.585649251937866
    },
    {
      "epoch": 1.763629913330078e-05,
      "model_forward_time": 1.7810571193695068,
      "step": 92493
    },
    {
      "epoch": 1.763629913330078e-05,
      "step": 92493,
      "training_step_time": 7.390283823013306
    },
    {
      "epoch": 1.7637252807617187e-05,
      "model_forward_time": 1.7773659229278564,
      "step": 92494
    },
    {
      "epoch": 1.7637252807617187e-05,
      "step": 92494,
      "training_step_time": 7.681422472000122
    },
    {
      "epoch": 1.7638206481933595e-05,
      "model_forward_time": 1.7802889347076416,
      "step": 92495
    },
    {
      "epoch": 1.7638206481933595e-05,
      "step": 92495,
      "training_step_time": 6.754347562789917
    },
    {
      "epoch": 1.763916015625e-05,
      "model_forward_time": 1.7733025550842285,
      "step": 92496
    },
    {
      "epoch": 1.763916015625e-05,
      "step": 92496,
      "training_step_time": 6.807454586029053
    },
    {
      "epoch": 1.7640113830566406e-05,
      "model_forward_time": 1.774404764175415,
      "step": 92497
    },
    {
      "epoch": 1.7640113830566406e-05,
      "step": 92497,
      "training_step_time": 6.7733845710754395
    },
    {
      "epoch": 1.7641067504882814e-05,
      "model_forward_time": 1.0907540321350098,
      "step": 92498
    },
    {
      "epoch": 1.7641067504882814e-05,
      "step": 92498,
      "training_step_time": 6.783850193023682
    },
    {
      "epoch": 1.7642021179199218e-05,
      "model_forward_time": 1.7839186191558838,
      "step": 92499
    },
    {
      "epoch": 1.7642021179199218e-05,
      "step": 92499,
      "training_step_time": 6.874415159225464
    },
    {
      "epoch": 1.7642974853515625e-05,
      "grad_norm": 0.11345487833023071,
      "learning_rate": 1.5303926323801123e-07,
      "loss": 0.0569,
      "step": 92500
    },
    {
      "epoch": 1.7642974853515625e-05,
      "model_forward_time": 1.7859420776367188,
      "step": 92500
    },
    {
      "dynamics_loss_avg": 0.0397394685074687,
      "epoch": 1.7642974853515625e-05,
      "step": 92500
    },
    {
      "action_loss_avg": 0.001416995725594461,
      "epoch": 1.7642974853515625e-05,
      "step": 92500
    },
    {
      "epoch": 1.7642974853515625e-05,
      "step": 92500,
      "training_step_time": 7.649608373641968
    },
    {
      "epoch": 1.7643928527832032e-05,
      "model_forward_time": 2.456139087677002,
      "step": 92501
    },
    {
      "epoch": 1.7643928527832032e-05,
      "step": 92501,
      "training_step_time": 7.6967291831970215
    },
    {
      "epoch": 1.7644882202148436e-05,
      "model_forward_time": 1.7775447368621826,
      "step": 92502
    },
    {
      "epoch": 1.7644882202148436e-05,
      "step": 92502,
      "training_step_time": 7.646809101104736
    },
    {
      "epoch": 1.7645835876464844e-05,
      "model_forward_time": 1.7822299003601074,
      "step": 92503
    },
    {
      "epoch": 1.7645835876464844e-05,
      "step": 92503,
      "training_step_time": 7.727879762649536
    },
    {
      "epoch": 1.764678955078125e-05,
      "model_forward_time": 1.7791061401367188,
      "step": 92504
    },
    {
      "epoch": 1.764678955078125e-05,
      "step": 92504,
      "training_step_time": 6.740431070327759
    },
    {
      "epoch": 1.7647743225097655e-05,
      "model_forward_time": 1.7741162776947021,
      "step": 92505
    },
    {
      "epoch": 1.7647743225097655e-05,
      "step": 92505,
      "training_step_time": 6.751760721206665
    },
    {
      "epoch": 1.7648696899414063e-05,
      "model_forward_time": 1.7813611030578613,
      "step": 92506
    },
    {
      "epoch": 1.7648696899414063e-05,
      "step": 92506,
      "training_step_time": 6.811858415603638
    },
    {
      "epoch": 1.764965057373047e-05,
      "model_forward_time": 1.7809455394744873,
      "step": 92507
    },
    {
      "epoch": 1.764965057373047e-05,
      "step": 92507,
      "training_step_time": 6.725787162780762
    },
    {
      "epoch": 1.7650604248046874e-05,
      "model_forward_time": 1.7781193256378174,
      "step": 92508
    },
    {
      "epoch": 1.7650604248046874e-05,
      "step": 92508,
      "training_step_time": 6.72136664390564
    },
    {
      "epoch": 1.765155792236328e-05,
      "model_forward_time": 1.7786548137664795,
      "step": 92509
    },
    {
      "epoch": 1.765155792236328e-05,
      "step": 92509,
      "training_step_time": 8.656036615371704
    },
    {
      "epoch": 1.765251159667969e-05,
      "grad_norm": 0.08098574727773666,
      "learning_rate": 1.5263357242405796e-07,
      "loss": 0.0601,
      "step": 92510
    },
    {
      "epoch": 1.765251159667969e-05,
      "model_forward_time": 2.456580877304077,
      "step": 92510
    },
    {
      "dynamics_loss_avg": 0.07662970814853906,
      "epoch": 1.765251159667969e-05,
      "step": 92510
    },
    {
      "action_loss_avg": 0.001455341250402853,
      "epoch": 1.765251159667969e-05,
      "step": 92510
    },
    {
      "epoch": 1.765251159667969e-05,
      "step": 92510,
      "training_step_time": 7.828577041625977
    },
    {
      "epoch": 1.7653465270996093e-05,
      "model_forward_time": 1.7842392921447754,
      "step": 92511
    },
    {
      "epoch": 1.7653465270996093e-05,
      "step": 92511,
      "training_step_time": 7.504700660705566
    },
    {
      "epoch": 1.76544189453125e-05,
      "model_forward_time": 1.7778983116149902,
      "step": 92512
    },
    {
      "epoch": 1.76544189453125e-05,
      "step": 92512,
      "training_step_time": 7.689891815185547
    },
    {
      "epoch": 1.7655372619628908e-05,
      "model_forward_time": 1.7824971675872803,
      "step": 92513
    },
    {
      "epoch": 1.7655372619628908e-05,
      "step": 92513,
      "training_step_time": 7.694677352905273
    },
    {
      "epoch": 1.7656326293945312e-05,
      "model_forward_time": 1.78470778465271,
      "step": 92514
    },
    {
      "epoch": 1.7656326293945312e-05,
      "step": 92514,
      "training_step_time": 6.746455430984497
    },
    {
      "epoch": 1.765727996826172e-05,
      "model_forward_time": 1.7828047275543213,
      "step": 92515
    },
    {
      "epoch": 1.765727996826172e-05,
      "step": 92515,
      "training_step_time": 6.764463424682617
    },
    {
      "epoch": 1.7658233642578123e-05,
      "model_forward_time": 1.77768874168396,
      "step": 92516
    },
    {
      "epoch": 1.7658233642578123e-05,
      "step": 92516,
      "training_step_time": 6.740223407745361
    },
    {
      "epoch": 1.765918731689453e-05,
      "model_forward_time": 1.7886314392089844,
      "step": 92517
    },
    {
      "epoch": 1.765918731689453e-05,
      "step": 92517,
      "training_step_time": 6.746857166290283
    },
    {
      "epoch": 1.7660140991210938e-05,
      "model_forward_time": 1.7833607196807861,
      "step": 92518
    },
    {
      "epoch": 1.7660140991210938e-05,
      "step": 92518,
      "training_step_time": 7.788157224655151
    },
    {
      "epoch": 1.7661094665527342e-05,
      "model_forward_time": 1.8156218528747559,
      "step": 92519
    },
    {
      "epoch": 1.7661094665527342e-05,
      "step": 92519,
      "training_step_time": 9.238035202026367
    },
    {
      "epoch": 1.766204833984375e-05,
      "grad_norm": 0.08262228965759277,
      "learning_rate": 1.5222841171080915e-07,
      "loss": 0.058,
      "step": 92520
    },
    {
      "epoch": 1.766204833984375e-05,
      "model_forward_time": 1.779017686843872,
      "step": 92520
    },
    {
      "dynamics_loss_avg": 0.05242145247757435,
      "epoch": 1.766204833984375e-05,
      "step": 92520
    },
    {
      "action_loss_avg": 0.0012958454783074558,
      "epoch": 1.766204833984375e-05,
      "step": 92520
    },
    {
      "epoch": 1.766204833984375e-05,
      "step": 92520,
      "training_step_time": 7.465548753738403
    },
    {
      "epoch": 1.7663002014160157e-05,
      "model_forward_time": 1.782900094985962,
      "step": 92521
    },
    {
      "epoch": 1.7663002014160157e-05,
      "step": 92521,
      "training_step_time": 7.662144184112549
    },
    {
      "epoch": 1.766395568847656e-05,
      "model_forward_time": 1.787639856338501,
      "step": 92522
    },
    {
      "epoch": 1.766395568847656e-05,
      "step": 92522,
      "training_step_time": 7.683720588684082
    },
    {
      "epoch": 1.766490936279297e-05,
      "model_forward_time": 1.7790865898132324,
      "step": 92523
    },
    {
      "epoch": 1.766490936279297e-05,
      "step": 92523,
      "training_step_time": 6.759526014328003
    },
    {
      "epoch": 1.7665863037109376e-05,
      "model_forward_time": 1.7796080112457275,
      "step": 92524
    },
    {
      "epoch": 1.7665863037109376e-05,
      "step": 92524,
      "training_step_time": 6.730175256729126
    },
    {
      "epoch": 1.766681671142578e-05,
      "model_forward_time": 1.7827503681182861,
      "step": 92525
    },
    {
      "epoch": 1.766681671142578e-05,
      "step": 92525,
      "training_step_time": 6.798838138580322
    },
    {
      "epoch": 1.7667770385742187e-05,
      "model_forward_time": 1.778322696685791,
      "step": 92526
    },
    {
      "epoch": 1.7667770385742187e-05,
      "step": 92526,
      "training_step_time": 6.765944004058838
    },
    {
      "epoch": 1.7668724060058595e-05,
      "model_forward_time": 1.7802245616912842,
      "step": 92527
    },
    {
      "epoch": 1.7668724060058595e-05,
      "step": 92527,
      "training_step_time": 7.204799175262451
    },
    {
      "epoch": 1.7669677734375e-05,
      "model_forward_time": 1.8185639381408691,
      "step": 92528
    },
    {
      "epoch": 1.7669677734375e-05,
      "step": 92528,
      "training_step_time": 10.661442041397095
    },
    {
      "epoch": 1.7670631408691406e-05,
      "model_forward_time": 2.4132747650146484,
      "step": 92529
    },
    {
      "epoch": 1.7670631408691406e-05,
      "step": 92529,
      "training_step_time": 7.750677824020386
    },
    {
      "epoch": 1.7671585083007814e-05,
      "grad_norm": 0.07820183783769608,
      "learning_rate": 1.518237811425727e-07,
      "loss": 0.0581,
      "step": 92530
    },
    {
      "epoch": 1.7671585083007814e-05,
      "model_forward_time": 1.7844867706298828,
      "step": 92530
    },
    {
      "dynamics_loss_avg": 0.060557957738637924,
      "epoch": 1.7671585083007814e-05,
      "step": 92530
    },
    {
      "action_loss_avg": 0.0017924644227605313,
      "epoch": 1.7671585083007814e-05,
      "step": 92530
    },
    {
      "epoch": 1.7671585083007814e-05,
      "step": 92530,
      "training_step_time": 7.2973527908325195
    },
    {
      "epoch": 1.7672538757324218e-05,
      "model_forward_time": 1.7749967575073242,
      "step": 92531
    },
    {
      "epoch": 1.7672538757324218e-05,
      "step": 92531,
      "training_step_time": 7.6832380294799805
    },
    {
      "epoch": 1.7673492431640625e-05,
      "model_forward_time": 1.7812604904174805,
      "step": 92532
    },
    {
      "epoch": 1.7673492431640625e-05,
      "step": 92532,
      "training_step_time": 6.7477028369903564
    },
    {
      "epoch": 1.7674446105957032e-05,
      "model_forward_time": 1.7561655044555664,
      "step": 92533
    },
    {
      "epoch": 1.7674446105957032e-05,
      "step": 92533,
      "training_step_time": 6.793917417526245
    },
    {
      "epoch": 1.7675399780273436e-05,
      "model_forward_time": 1.7852215766906738,
      "step": 92534
    },
    {
      "epoch": 1.7675399780273436e-05,
      "step": 92534,
      "training_step_time": 6.766383171081543
    },
    {
      "epoch": 1.7676353454589844e-05,
      "model_forward_time": 1.7788755893707275,
      "step": 92535
    },
    {
      "epoch": 1.7676353454589844e-05,
      "step": 92535,
      "training_step_time": 6.774156332015991
    },
    {
      "epoch": 1.767730712890625e-05,
      "model_forward_time": 1.7796673774719238,
      "step": 92536
    },
    {
      "epoch": 1.767730712890625e-05,
      "step": 92536,
      "training_step_time": 7.201191186904907
    },
    {
      "epoch": 1.7678260803222655e-05,
      "model_forward_time": 1.8190524578094482,
      "step": 92537
    },
    {
      "epoch": 1.7678260803222655e-05,
      "step": 92537,
      "training_step_time": 9.122715950012207
    },
    {
      "epoch": 1.7679214477539063e-05,
      "model_forward_time": 2.435401678085327,
      "step": 92538
    },
    {
      "epoch": 1.7679214477539063e-05,
      "step": 92538,
      "training_step_time": 8.620435953140259
    },
    {
      "epoch": 1.768016815185547e-05,
      "model_forward_time": 1.7834868431091309,
      "step": 92539
    },
    {
      "epoch": 1.768016815185547e-05,
      "step": 92539,
      "training_step_time": 7.393726825714111
    },
    {
      "epoch": 1.7681121826171874e-05,
      "grad_norm": 0.1216074749827385,
      "learning_rate": 1.5141968076359869e-07,
      "loss": 0.0596,
      "step": 92540
    },
    {
      "epoch": 1.7681121826171874e-05,
      "model_forward_time": 1.8012168407440186,
      "step": 92540
    },
    {
      "dynamics_loss_avg": 0.06575292441993952,
      "epoch": 1.7681121826171874e-05,
      "step": 92540
    },
    {
      "action_loss_avg": 0.001470028664334677,
      "epoch": 1.7681121826171874e-05,
      "step": 92540
    },
    {
      "epoch": 1.7681121826171874e-05,
      "step": 92540,
      "training_step_time": 7.652351379394531
    },
    {
      "epoch": 1.768207550048828e-05,
      "model_forward_time": 1.780087947845459,
      "step": 92541
    },
    {
      "epoch": 1.768207550048828e-05,
      "step": 92541,
      "training_step_time": 6.785423040390015
    },
    {
      "epoch": 1.768302917480469e-05,
      "model_forward_time": 1.7919297218322754,
      "step": 92542
    },
    {
      "epoch": 1.768302917480469e-05,
      "step": 92542,
      "training_step_time": 6.736974477767944
    },
    {
      "epoch": 1.7683982849121093e-05,
      "model_forward_time": 1.7823092937469482,
      "step": 92543
    },
    {
      "epoch": 1.7683982849121093e-05,
      "step": 92543,
      "training_step_time": 6.745335817337036
    },
    {
      "epoch": 1.76849365234375e-05,
      "model_forward_time": 1.7832014560699463,
      "step": 92544
    },
    {
      "epoch": 1.76849365234375e-05,
      "step": 92544,
      "training_step_time": 6.726486682891846
    },
    {
      "epoch": 1.7685890197753908e-05,
      "model_forward_time": 1.7869462966918945,
      "step": 92545
    },
    {
      "epoch": 1.7685890197753908e-05,
      "step": 92545,
      "training_step_time": 6.724453687667847
    },
    {
      "epoch": 1.7686843872070312e-05,
      "model_forward_time": 1.7860565185546875,
      "step": 92546
    },
    {
      "epoch": 1.7686843872070312e-05,
      "step": 92546,
      "training_step_time": 10.293719291687012
    },
    {
      "epoch": 1.768779754638672e-05,
      "model_forward_time": 2.441970109939575,
      "step": 92547
    },
    {
      "epoch": 1.768779754638672e-05,
      "step": 92547,
      "training_step_time": 9.103233337402344
    },
    {
      "epoch": 1.7688751220703123e-05,
      "model_forward_time": 1.785353660583496,
      "step": 92548
    },
    {
      "epoch": 1.7688751220703123e-05,
      "step": 92548,
      "training_step_time": 7.350552558898926
    },
    {
      "epoch": 1.768970489501953e-05,
      "model_forward_time": 1.7772815227508545,
      "step": 92549
    },
    {
      "epoch": 1.768970489501953e-05,
      "step": 92549,
      "training_step_time": 8.220226287841797
    },
    {
      "epoch": 1.7690658569335938e-05,
      "grad_norm": 0.07616065442562103,
      "learning_rate": 1.5101611061807908e-07,
      "loss": 0.0594,
      "step": 92550
    },
    {
      "epoch": 1.7690658569335938e-05,
      "model_forward_time": 1.7863035202026367,
      "step": 92550
    },
    {
      "dynamics_loss_avg": 0.06806065049022436,
      "epoch": 1.7690658569335938e-05,
      "step": 92550
    },
    {
      "action_loss_avg": 0.002421562699601054,
      "epoch": 1.7690658569335938e-05,
      "step": 92550
    },
    {
      "epoch": 1.7690658569335938e-05,
      "step": 92550,
      "training_step_time": 7.270625352859497
    },
    {
      "epoch": 1.7691612243652342e-05,
      "model_forward_time": 1.7640891075134277,
      "step": 92551
    },
    {
      "epoch": 1.7691612243652342e-05,
      "step": 92551,
      "training_step_time": 6.726944446563721
    },
    {
      "epoch": 1.769256591796875e-05,
      "model_forward_time": 1.7830135822296143,
      "step": 92552
    },
    {
      "epoch": 1.769256591796875e-05,
      "step": 92552,
      "training_step_time": 6.726811647415161
    },
    {
      "epoch": 1.7693519592285157e-05,
      "model_forward_time": 1.78468918800354,
      "step": 92553
    },
    {
      "epoch": 1.7693519592285157e-05,
      "step": 92553,
      "training_step_time": 6.727069139480591
    },
    {
      "epoch": 1.769447326660156e-05,
      "model_forward_time": 1.7902960777282715,
      "step": 92554
    },
    {
      "epoch": 1.769447326660156e-05,
      "step": 92554,
      "training_step_time": 6.747491121292114
    },
    {
      "epoch": 1.769542694091797e-05,
      "model_forward_time": 1.7825844287872314,
      "step": 92555
    },
    {
      "epoch": 1.769542694091797e-05,
      "step": 92555,
      "training_step_time": 11.625038385391235
    },
    {
      "epoch": 1.7696380615234376e-05,
      "model_forward_time": 2.0750768184661865,
      "step": 92556
    },
    {
      "epoch": 1.7696380615234376e-05,
      "step": 92556,
      "training_step_time": 7.8204569816589355
    },
    {
      "epoch": 1.769733428955078e-05,
      "model_forward_time": 1.7815196514129639,
      "step": 92557
    },
    {
      "epoch": 1.769733428955078e-05,
      "step": 92557,
      "training_step_time": 7.448625326156616
    },
    {
      "epoch": 1.7698287963867187e-05,
      "model_forward_time": 1.7865509986877441,
      "step": 92558
    },
    {
      "epoch": 1.7698287963867187e-05,
      "step": 92558,
      "training_step_time": 7.689060688018799
    },
    {
      "epoch": 1.7699241638183595e-05,
      "model_forward_time": 1.7864086627960205,
      "step": 92559
    },
    {
      "epoch": 1.7699241638183595e-05,
      "step": 92559,
      "training_step_time": 7.646893501281738
    },
    {
      "epoch": 1.77001953125e-05,
      "grad_norm": 0.07591686397790909,
      "learning_rate": 1.5061307075014632e-07,
      "loss": 0.0592,
      "step": 92560
    },
    {
      "epoch": 1.77001953125e-05,
      "model_forward_time": 1.7852582931518555,
      "step": 92560
    },
    {
      "dynamics_loss_avg": 0.04902999922633171,
      "epoch": 1.77001953125e-05,
      "step": 92560
    },
    {
      "action_loss_avg": 0.0009446377516724169,
      "epoch": 1.77001953125e-05,
      "step": 92560
    },
    {
      "epoch": 1.77001953125e-05,
      "step": 92560,
      "training_step_time": 6.792147159576416
    },
    {
      "epoch": 1.7701148986816406e-05,
      "model_forward_time": 1.780771255493164,
      "step": 92561
    },
    {
      "epoch": 1.7701148986816406e-05,
      "step": 92561,
      "training_step_time": 6.810959815979004
    },
    {
      "epoch": 1.7702102661132814e-05,
      "model_forward_time": 1.7832822799682617,
      "step": 92562
    },
    {
      "epoch": 1.7702102661132814e-05,
      "step": 92562,
      "training_step_time": 6.844566583633423
    },
    {
      "epoch": 1.7703056335449218e-05,
      "model_forward_time": 1.7860357761383057,
      "step": 92563
    },
    {
      "epoch": 1.7703056335449218e-05,
      "step": 92563,
      "training_step_time": 6.78308367729187
    },
    {
      "epoch": 1.7704010009765625e-05,
      "model_forward_time": 1.7840209007263184,
      "step": 92564
    },
    {
      "epoch": 1.7704010009765625e-05,
      "step": 92564,
      "training_step_time": 7.720294237136841
    },
    {
      "epoch": 1.7704963684082032e-05,
      "model_forward_time": 1.4554972648620605,
      "step": 92565
    },
    {
      "epoch": 1.7704963684082032e-05,
      "step": 92565,
      "training_step_time": 10.119751691818237
    },
    {
      "epoch": 1.7705917358398436e-05,
      "model_forward_time": 2.4242868423461914,
      "step": 92566
    },
    {
      "epoch": 1.7705917358398436e-05,
      "step": 92566,
      "training_step_time": 7.644567251205444
    },
    {
      "epoch": 1.7706871032714844e-05,
      "model_forward_time": 1.7799806594848633,
      "step": 92567
    },
    {
      "epoch": 1.7706871032714844e-05,
      "step": 92567,
      "training_step_time": 7.663749694824219
    },
    {
      "epoch": 1.770782470703125e-05,
      "model_forward_time": 1.7806951999664307,
      "step": 92568
    },
    {
      "epoch": 1.770782470703125e-05,
      "step": 92568,
      "training_step_time": 7.7436137199401855
    },
    {
      "epoch": 1.7708778381347655e-05,
      "model_forward_time": 1.7757835388183594,
      "step": 92569
    },
    {
      "epoch": 1.7708778381347655e-05,
      "step": 92569,
      "training_step_time": 6.71148943901062
    },
    {
      "epoch": 1.7709732055664063e-05,
      "grad_norm": 0.12182220816612244,
      "learning_rate": 1.5021056120387733e-07,
      "loss": 0.0573,
      "step": 92570
    },
    {
      "epoch": 1.7709732055664063e-05,
      "model_forward_time": 1.7783317565917969,
      "step": 92570
    },
    {
      "dynamics_loss_avg": 0.05893043819814921,
      "epoch": 1.7709732055664063e-05,
      "step": 92570
    },
    {
      "action_loss_avg": 0.001976972838747315,
      "epoch": 1.7709732055664063e-05,
      "step": 92570
    },
    {
      "epoch": 1.7709732055664063e-05,
      "step": 92570,
      "training_step_time": 6.879332065582275
    },
    {
      "epoch": 1.771068572998047e-05,
      "model_forward_time": 1.789097785949707,
      "step": 92571
    },
    {
      "epoch": 1.771068572998047e-05,
      "step": 92571,
      "training_step_time": 6.788811683654785
    },
    {
      "epoch": 1.7711639404296874e-05,
      "model_forward_time": 1.7778587341308594,
      "step": 92572
    },
    {
      "epoch": 1.7711639404296874e-05,
      "step": 92572,
      "training_step_time": 6.87033486366272
    },
    {
      "epoch": 1.771259307861328e-05,
      "model_forward_time": 1.772278070449829,
      "step": 92573
    },
    {
      "epoch": 1.771259307861328e-05,
      "step": 92573,
      "training_step_time": 8.798575162887573
    },
    {
      "epoch": 1.771354675292969e-05,
      "model_forward_time": 1.816370964050293,
      "step": 92574
    },
    {
      "epoch": 1.771354675292969e-05,
      "step": 92574,
      "training_step_time": 9.51955270767212
    },
    {
      "epoch": 1.7714500427246093e-05,
      "model_forward_time": 2.4084744453430176,
      "step": 92575
    },
    {
      "epoch": 1.7714500427246093e-05,
      "step": 92575,
      "training_step_time": 8.269816398620605
    },
    {
      "epoch": 1.77154541015625e-05,
      "model_forward_time": 1.778681993484497,
      "step": 92576
    },
    {
      "epoch": 1.77154541015625e-05,
      "step": 92576,
      "training_step_time": 7.360612630844116
    },
    {
      "epoch": 1.7716407775878908e-05,
      "model_forward_time": 1.7746968269348145,
      "step": 92577
    },
    {
      "epoch": 1.7716407775878908e-05,
      "step": 92577,
      "training_step_time": 7.693545579910278
    },
    {
      "epoch": 1.7717361450195312e-05,
      "model_forward_time": 1.7744026184082031,
      "step": 92578
    },
    {
      "epoch": 1.7717361450195312e-05,
      "step": 92578,
      "training_step_time": 6.8570027351379395
    },
    {
      "epoch": 1.771831512451172e-05,
      "model_forward_time": 1.7735297679901123,
      "step": 92579
    },
    {
      "epoch": 1.771831512451172e-05,
      "step": 92579,
      "training_step_time": 6.885924816131592
    },
    {
      "epoch": 1.7719268798828123e-05,
      "grad_norm": 0.08734048157930374,
      "learning_rate": 1.4980858202329085e-07,
      "loss": 0.0583,
      "step": 92580
    },
    {
      "epoch": 1.7719268798828123e-05,
      "model_forward_time": 1.7812471389770508,
      "step": 92580
    },
    {
      "dynamics_loss_avg": 0.07698519453406334,
      "epoch": 1.7719268798828123e-05,
      "step": 92580
    },
    {
      "action_loss_avg": 0.001726821789634414,
      "epoch": 1.7719268798828123e-05,
      "step": 92580
    },
    {
      "epoch": 1.7719268798828123e-05,
      "step": 92580,
      "training_step_time": 6.911752939224243
    },
    {
      "epoch": 1.772022247314453e-05,
      "model_forward_time": 1.7763996124267578,
      "step": 92581
    },
    {
      "epoch": 1.772022247314453e-05,
      "step": 92581,
      "training_step_time": 6.899850130081177
    },
    {
      "epoch": 1.7721176147460938e-05,
      "model_forward_time": 1.777466058731079,
      "step": 92582
    },
    {
      "epoch": 1.7721176147460938e-05,
      "step": 92582,
      "training_step_time": 7.805080413818359
    },
    {
      "epoch": 1.7722129821777342e-05,
      "model_forward_time": 1.8012630939483643,
      "step": 92583
    },
    {
      "epoch": 1.7722129821777342e-05,
      "step": 92583,
      "training_step_time": 10.412676811218262
    },
    {
      "epoch": 1.772308349609375e-05,
      "model_forward_time": 2.4102797508239746,
      "step": 92584
    },
    {
      "epoch": 1.772308349609375e-05,
      "step": 92584,
      "training_step_time": 7.842388868331909
    },
    {
      "epoch": 1.7724037170410157e-05,
      "model_forward_time": 1.7542212009429932,
      "step": 92585
    },
    {
      "epoch": 1.7724037170410157e-05,
      "step": 92585,
      "training_step_time": 7.215689182281494
    },
    {
      "epoch": 1.772499084472656e-05,
      "model_forward_time": 1.7763407230377197,
      "step": 92586
    },
    {
      "epoch": 1.772499084472656e-05,
      "step": 92586,
      "training_step_time": 7.703226327896118
    },
    {
      "epoch": 1.772594451904297e-05,
      "model_forward_time": 1.7748675346374512,
      "step": 92587
    },
    {
      "epoch": 1.772594451904297e-05,
      "step": 92587,
      "training_step_time": 7.368410587310791
    },
    {
      "epoch": 1.7726898193359376e-05,
      "model_forward_time": 1.7619166374206543,
      "step": 92588
    },
    {
      "epoch": 1.7726898193359376e-05,
      "step": 92588,
      "training_step_time": 6.81058144569397
    },
    {
      "epoch": 1.772785186767578e-05,
      "model_forward_time": 1.7790963649749756,
      "step": 92589
    },
    {
      "epoch": 1.772785186767578e-05,
      "step": 92589,
      "training_step_time": 6.8711159229278564
    },
    {
      "epoch": 1.7728805541992187e-05,
      "grad_norm": 0.06774326413869858,
      "learning_rate": 1.4940713325234447e-07,
      "loss": 0.0564,
      "step": 92590
    },
    {
      "epoch": 1.7728805541992187e-05,
      "model_forward_time": 1.745025634765625,
      "step": 92590
    },
    {
      "dynamics_loss_avg": 0.05997601207345724,
      "epoch": 1.7728805541992187e-05,
      "step": 92590
    },
    {
      "action_loss_avg": 0.0017614977099583485,
      "epoch": 1.7728805541992187e-05,
      "step": 92590
    },
    {
      "epoch": 1.7728805541992187e-05,
      "step": 92590,
      "training_step_time": 6.850390672683716
    },
    {
      "epoch": 1.7729759216308595e-05,
      "model_forward_time": 1.775482177734375,
      "step": 92591
    },
    {
      "epoch": 1.7729759216308595e-05,
      "step": 92591,
      "training_step_time": 7.774565696716309
    },
    {
      "epoch": 1.7730712890625e-05,
      "model_forward_time": 1.789618968963623,
      "step": 92592
    },
    {
      "epoch": 1.7730712890625e-05,
      "step": 92592,
      "training_step_time": 8.628734827041626
    },
    {
      "epoch": 1.7731666564941406e-05,
      "model_forward_time": 2.438015937805176,
      "step": 92593
    },
    {
      "epoch": 1.7731666564941406e-05,
      "step": 92593,
      "training_step_time": 9.012357711791992
    },
    {
      "epoch": 1.7732620239257814e-05,
      "model_forward_time": 1.7766032218933105,
      "step": 92594
    },
    {
      "epoch": 1.7732620239257814e-05,
      "step": 92594,
      "training_step_time": 7.23815131187439
    },
    {
      "epoch": 1.7733573913574218e-05,
      "model_forward_time": 1.7798218727111816,
      "step": 92595
    },
    {
      "epoch": 1.7733573913574218e-05,
      "step": 92595,
      "training_step_time": 8.455162048339844
    },
    {
      "epoch": 1.7734527587890625e-05,
      "model_forward_time": 1.7749123573303223,
      "step": 92596
    },
    {
      "epoch": 1.7734527587890625e-05,
      "step": 92596,
      "training_step_time": 7.64340877532959
    },
    {
      "epoch": 1.7735481262207032e-05,
      "model_forward_time": 1.7861850261688232,
      "step": 92597
    },
    {
      "epoch": 1.7735481262207032e-05,
      "step": 92597,
      "training_step_time": 6.88190484046936
    },
    {
      "epoch": 1.7736434936523437e-05,
      "model_forward_time": 1.772463083267212,
      "step": 92598
    },
    {
      "epoch": 1.7736434936523437e-05,
      "step": 92598,
      "training_step_time": 6.858245372772217
    },
    {
      "epoch": 1.7737388610839844e-05,
      "model_forward_time": 1.7726812362670898,
      "step": 92599
    },
    {
      "epoch": 1.7737388610839844e-05,
      "step": 92599,
      "training_step_time": 6.882854223251343
    },
    {
      "epoch": 1.773834228515625e-05,
      "grad_norm": 0.1113642230629921,
      "learning_rate": 1.4900621493494137e-07,
      "loss": 0.0564,
      "step": 92600
    },
    {
      "epoch": 1.773834228515625e-05,
      "model_forward_time": 1.7740249633789062,
      "step": 92600
    },
    {
      "dynamics_loss_avg": 0.05916356761008501,
      "epoch": 1.773834228515625e-05,
      "step": 92600
    },
    {
      "action_loss_avg": 0.0015561415115371347,
      "epoch": 1.773834228515625e-05,
      "step": 92600
    },
    {
      "epoch": 1.773834228515625e-05,
      "step": 92600,
      "training_step_time": 7.73424506187439
    },
    {
      "epoch": 1.7739295959472655e-05,
      "model_forward_time": 1.7802464962005615,
      "step": 92601
    },
    {
      "epoch": 1.7739295959472655e-05,
      "step": 92601,
      "training_step_time": 11.051060199737549
    },
    {
      "epoch": 1.7740249633789063e-05,
      "model_forward_time": 2.4144721031188965,
      "step": 92602
    },
    {
      "epoch": 1.7740249633789063e-05,
      "step": 92602,
      "training_step_time": 7.840223073959351
    },
    {
      "epoch": 1.774120330810547e-05,
      "model_forward_time": 1.7743556499481201,
      "step": 92603
    },
    {
      "epoch": 1.774120330810547e-05,
      "step": 92603,
      "training_step_time": 7.524682998657227
    },
    {
      "epoch": 1.7742156982421874e-05,
      "model_forward_time": 1.7729580402374268,
      "step": 92604
    },
    {
      "epoch": 1.7742156982421874e-05,
      "step": 92604,
      "training_step_time": 8.573581218719482
    },
    {
      "epoch": 1.774311065673828e-05,
      "model_forward_time": 1.7709472179412842,
      "step": 92605
    },
    {
      "epoch": 1.774311065673828e-05,
      "step": 92605,
      "training_step_time": 7.708624839782715
    },
    {
      "epoch": 1.774406433105469e-05,
      "model_forward_time": 1.770251750946045,
      "step": 92606
    },
    {
      "epoch": 1.774406433105469e-05,
      "step": 92606,
      "training_step_time": 6.852444171905518
    },
    {
      "epoch": 1.7745018005371093e-05,
      "model_forward_time": 1.4131276607513428,
      "step": 92607
    },
    {
      "epoch": 1.7745018005371093e-05,
      "step": 92607,
      "training_step_time": 6.795933723449707
    },
    {
      "epoch": 1.77459716796875e-05,
      "model_forward_time": 1.767897605895996,
      "step": 92608
    },
    {
      "epoch": 1.77459716796875e-05,
      "step": 92608,
      "training_step_time": 6.814581394195557
    },
    {
      "epoch": 1.7746925354003908e-05,
      "model_forward_time": 1.7746140956878662,
      "step": 92609
    },
    {
      "epoch": 1.7746925354003908e-05,
      "step": 92609,
      "training_step_time": 8.897684097290039
    },
    {
      "epoch": 1.7747879028320312e-05,
      "grad_norm": 0.08170312643051147,
      "learning_rate": 1.4860582711492544e-07,
      "loss": 0.0569,
      "step": 92610
    },
    {
      "epoch": 1.7747879028320312e-05,
      "model_forward_time": 1.7702834606170654,
      "step": 92610
    },
    {
      "dynamics_loss_avg": 0.06763696298003197,
      "epoch": 1.7747879028320312e-05,
      "step": 92610
    },
    {
      "action_loss_avg": 0.0012859926471719518,
      "epoch": 1.7747879028320312e-05,
      "step": 92610
    },
    {
      "epoch": 1.7747879028320312e-05,
      "step": 92610,
      "training_step_time": 10.250215291976929
    },
    {
      "epoch": 1.774883270263672e-05,
      "model_forward_time": 2.427581310272217,
      "step": 92611
    },
    {
      "epoch": 1.774883270263672e-05,
      "step": 92611,
      "training_step_time": 10.411863088607788
    },
    {
      "epoch": 1.7749786376953123e-05,
      "model_forward_time": 1.7712182998657227,
      "step": 92612
    },
    {
      "epoch": 1.7749786376953123e-05,
      "step": 92612,
      "training_step_time": 7.662962913513184
    },
    {
      "epoch": 1.775074005126953e-05,
      "model_forward_time": 1.7731952667236328,
      "step": 92613
    },
    {
      "epoch": 1.775074005126953e-05,
      "step": 92613,
      "training_step_time": 8.215306758880615
    },
    {
      "epoch": 1.7751693725585938e-05,
      "model_forward_time": 1.7765133380889893,
      "step": 92614
    },
    {
      "epoch": 1.7751693725585938e-05,
      "step": 92614,
      "training_step_time": 7.652482748031616
    },
    {
      "epoch": 1.7752647399902342e-05,
      "model_forward_time": 1.7744266986846924,
      "step": 92615
    },
    {
      "epoch": 1.7752647399902342e-05,
      "step": 92615,
      "training_step_time": 6.791914701461792
    },
    {
      "epoch": 1.775360107421875e-05,
      "model_forward_time": 1.7799298763275146,
      "step": 92616
    },
    {
      "epoch": 1.775360107421875e-05,
      "step": 92616,
      "training_step_time": 6.8232011795043945
    },
    {
      "epoch": 1.7754554748535157e-05,
      "model_forward_time": 1.772658348083496,
      "step": 92617
    },
    {
      "epoch": 1.7754554748535157e-05,
      "step": 92617,
      "training_step_time": 6.773324728012085
    },
    {
      "epoch": 1.775550842285156e-05,
      "model_forward_time": 1.7739546298980713,
      "step": 92618
    },
    {
      "epoch": 1.775550842285156e-05,
      "step": 92618,
      "training_step_time": 7.230514287948608
    },
    {
      "epoch": 1.775646209716797e-05,
      "model_forward_time": 1.7715442180633545,
      "step": 92619
    },
    {
      "epoch": 1.775646209716797e-05,
      "step": 92619,
      "training_step_time": 9.150920391082764
    },
    {
      "epoch": 1.7757415771484376e-05,
      "grad_norm": 0.09314769506454468,
      "learning_rate": 1.4820596983608326e-07,
      "loss": 0.0555,
      "step": 92620
    },
    {
      "epoch": 1.7757415771484376e-05,
      "model_forward_time": 2.436079263687134,
      "step": 92620
    },
    {
      "dynamics_loss_avg": 0.055560064129531385,
      "epoch": 1.7757415771484376e-05,
      "step": 92620
    },
    {
      "action_loss_avg": 0.0010189536144025624,
      "epoch": 1.7757415771484376e-05,
      "step": 92620
    },
    {
      "epoch": 1.7757415771484376e-05,
      "step": 92620,
      "training_step_time": 7.792035818099976
    },
    {
      "epoch": 1.775836944580078e-05,
      "model_forward_time": 1.7683372497558594,
      "step": 92621
    },
    {
      "epoch": 1.775836944580078e-05,
      "step": 92621,
      "training_step_time": 7.742110252380371
    },
    {
      "epoch": 1.7759323120117187e-05,
      "model_forward_time": 1.7744560241699219,
      "step": 92622
    },
    {
      "epoch": 1.7759323120117187e-05,
      "step": 92622,
      "training_step_time": 7.465250730514526
    },
    {
      "epoch": 1.7760276794433595e-05,
      "model_forward_time": 1.7743451595306396,
      "step": 92623
    },
    {
      "epoch": 1.7760276794433595e-05,
      "step": 92623,
      "training_step_time": 8.373724222183228
    },
    {
      "epoch": 1.776123046875e-05,
      "model_forward_time": 1.771604299545288,
      "step": 92624
    },
    {
      "epoch": 1.776123046875e-05,
      "step": 92624,
      "training_step_time": 6.915566682815552
    },
    {
      "epoch": 1.7762184143066406e-05,
      "model_forward_time": 1.773681640625,
      "step": 92625
    },
    {
      "epoch": 1.7762184143066406e-05,
      "step": 92625,
      "training_step_time": 6.871042490005493
    },
    {
      "epoch": 1.7763137817382814e-05,
      "model_forward_time": 1.7782623767852783,
      "step": 92626
    },
    {
      "epoch": 1.7763137817382814e-05,
      "step": 92626,
      "training_step_time": 6.871202707290649
    },
    {
      "epoch": 1.7764091491699218e-05,
      "model_forward_time": 1.7699031829833984,
      "step": 92627
    },
    {
      "epoch": 1.7764091491699218e-05,
      "step": 92627,
      "training_step_time": 7.27215838432312
    },
    {
      "epoch": 1.7765045166015625e-05,
      "model_forward_time": 1.7747993469238281,
      "step": 92628
    },
    {
      "epoch": 1.7765045166015625e-05,
      "step": 92628,
      "training_step_time": 8.605183124542236
    },
    {
      "epoch": 1.7765998840332033e-05,
      "model_forward_time": 2.4388933181762695,
      "step": 92629
    },
    {
      "epoch": 1.7765998840332033e-05,
      "step": 92629,
      "training_step_time": 8.620773792266846
    },
    {
      "epoch": 1.7766952514648437e-05,
      "grad_norm": 0.08068198710680008,
      "learning_rate": 1.4780664314214044e-07,
      "loss": 0.0562,
      "step": 92630
    },
    {
      "epoch": 1.7766952514648437e-05,
      "model_forward_time": 1.7706694602966309,
      "step": 92630
    },
    {
      "dynamics_loss_avg": 0.0565240578725934,
      "epoch": 1.7766952514648437e-05,
      "step": 92630
    },
    {
      "action_loss_avg": 0.0018768644280498848,
      "epoch": 1.7766952514648437e-05,
      "step": 92630
    },
    {
      "epoch": 1.7766952514648437e-05,
      "step": 92630,
      "training_step_time": 8.822926998138428
    },
    {
      "epoch": 1.7767906188964844e-05,
      "model_forward_time": 1.7760508060455322,
      "step": 92631
    },
    {
      "epoch": 1.7767906188964844e-05,
      "step": 92631,
      "training_step_time": 7.547876358032227
    },
    {
      "epoch": 1.776885986328125e-05,
      "model_forward_time": 1.7786865234375,
      "step": 92632
    },
    {
      "epoch": 1.776885986328125e-05,
      "step": 92632,
      "training_step_time": 8.710746049880981
    },
    {
      "epoch": 1.7769813537597655e-05,
      "model_forward_time": 1.770669937133789,
      "step": 92633
    },
    {
      "epoch": 1.7769813537597655e-05,
      "step": 92633,
      "training_step_time": 7.538623332977295
    },
    {
      "epoch": 1.7770767211914063e-05,
      "model_forward_time": 1.7691864967346191,
      "step": 92634
    },
    {
      "epoch": 1.7770767211914063e-05,
      "step": 92634,
      "training_step_time": 6.7996745109558105
    },
    {
      "epoch": 1.777172088623047e-05,
      "model_forward_time": 1.7692811489105225,
      "step": 92635
    },
    {
      "epoch": 1.777172088623047e-05,
      "step": 92635,
      "training_step_time": 6.884080171585083
    },
    {
      "epoch": 1.7772674560546874e-05,
      "model_forward_time": 1.772071123123169,
      "step": 92636
    },
    {
      "epoch": 1.7772674560546874e-05,
      "step": 92636,
      "training_step_time": 7.253576278686523
    },
    {
      "epoch": 1.777362823486328e-05,
      "model_forward_time": 1.7666513919830322,
      "step": 92637
    },
    {
      "epoch": 1.777362823486328e-05,
      "step": 92637,
      "training_step_time": 8.649223804473877
    },
    {
      "epoch": 1.777458190917969e-05,
      "model_forward_time": 2.4221768379211426,
      "step": 92638
    },
    {
      "epoch": 1.777458190917969e-05,
      "step": 92638,
      "training_step_time": 7.924160718917847
    },
    {
      "epoch": 1.7775535583496093e-05,
      "model_forward_time": 1.0623457431793213,
      "step": 92639
    },
    {
      "epoch": 1.7775535583496093e-05,
      "step": 92639,
      "training_step_time": 8.321714401245117
    },
    {
      "epoch": 1.77764892578125e-05,
      "grad_norm": 0.0788898840546608,
      "learning_rate": 1.474078470767687e-07,
      "loss": 0.0551,
      "step": 92640
    },
    {
      "epoch": 1.77764892578125e-05,
      "model_forward_time": 1.7651429176330566,
      "step": 92640
    },
    {
      "dynamics_loss_avg": 0.0740905361250043,
      "epoch": 1.77764892578125e-05,
      "step": 92640
    },
    {
      "action_loss_avg": 0.002070061577251181,
      "epoch": 1.77764892578125e-05,
      "step": 92640
    },
    {
      "epoch": 1.77764892578125e-05,
      "step": 92640,
      "training_step_time": 7.575833559036255
    },
    {
      "epoch": 1.7777442932128905e-05,
      "model_forward_time": 1.7677526473999023,
      "step": 92641
    },
    {
      "epoch": 1.7777442932128905e-05,
      "step": 92641,
      "training_step_time": 7.684485197067261
    },
    {
      "epoch": 1.7778396606445312e-05,
      "model_forward_time": 1.7713992595672607,
      "step": 92642
    },
    {
      "epoch": 1.7778396606445312e-05,
      "step": 92642,
      "training_step_time": 7.5199525356292725
    },
    {
      "epoch": 1.777935028076172e-05,
      "model_forward_time": 1.7682549953460693,
      "step": 92643
    },
    {
      "epoch": 1.777935028076172e-05,
      "step": 92643,
      "training_step_time": 6.83369255065918
    },
    {
      "epoch": 1.7780303955078123e-05,
      "model_forward_time": 1.7713146209716797,
      "step": 92644
    },
    {
      "epoch": 1.7780303955078123e-05,
      "step": 92644,
      "training_step_time": 6.8368823528289795
    },
    {
      "epoch": 1.778125762939453e-05,
      "model_forward_time": 1.7706201076507568,
      "step": 92645
    },
    {
      "epoch": 1.778125762939453e-05,
      "step": 92645,
      "training_step_time": 7.243274450302124
    },
    {
      "epoch": 1.7782211303710938e-05,
      "model_forward_time": 1.765732765197754,
      "step": 92646
    },
    {
      "epoch": 1.7782211303710938e-05,
      "step": 92646,
      "training_step_time": 7.6425721645355225
    },
    {
      "epoch": 1.7783164978027342e-05,
      "model_forward_time": 1.8386774063110352,
      "step": 92647
    },
    {
      "epoch": 1.7783164978027342e-05,
      "step": 92647,
      "training_step_time": 9.53451943397522
    },
    {
      "epoch": 1.778411865234375e-05,
      "model_forward_time": 2.445434808731079,
      "step": 92648
    },
    {
      "epoch": 1.778411865234375e-05,
      "step": 92648,
      "training_step_time": 9.689567565917969
    },
    {
      "epoch": 1.7785072326660157e-05,
      "model_forward_time": 1.7652487754821777,
      "step": 92649
    },
    {
      "epoch": 1.7785072326660157e-05,
      "step": 92649,
      "training_step_time": 7.62311863899231
    },
    {
      "epoch": 1.778602600097656e-05,
      "grad_norm": 0.112821064889431,
      "learning_rate": 1.470095816835787e-07,
      "loss": 0.0542,
      "step": 92650
    },
    {
      "epoch": 1.778602600097656e-05,
      "model_forward_time": 1.770371675491333,
      "step": 92650
    },
    {
      "dynamics_loss_avg": 0.06631959564983844,
      "epoch": 1.778602600097656e-05,
      "step": 92650
    },
    {
      "action_loss_avg": 0.0013878675905289128,
      "epoch": 1.778602600097656e-05,
      "step": 92650
    },
    {
      "epoch": 1.778602600097656e-05,
      "step": 92650,
      "training_step_time": 8.102148532867432
    },
    {
      "epoch": 1.778697967529297e-05,
      "model_forward_time": 1.7689208984375,
      "step": 92651
    },
    {
      "epoch": 1.778697967529297e-05,
      "step": 92651,
      "training_step_time": 7.686548709869385
    },
    {
      "epoch": 1.7787933349609376e-05,
      "model_forward_time": 1.7650504112243652,
      "step": 92652
    },
    {
      "epoch": 1.7787933349609376e-05,
      "step": 92652,
      "training_step_time": 6.8367228507995605
    },
    {
      "epoch": 1.778888702392578e-05,
      "model_forward_time": 1.7662503719329834,
      "step": 92653
    },
    {
      "epoch": 1.778888702392578e-05,
      "step": 92653,
      "training_step_time": 6.798158645629883
    },
    {
      "epoch": 1.7789840698242187e-05,
      "model_forward_time": 1.7640280723571777,
      "step": 92654
    },
    {
      "epoch": 1.7789840698242187e-05,
      "step": 92654,
      "training_step_time": 7.275886058807373
    },
    {
      "epoch": 1.7790794372558595e-05,
      "model_forward_time": 1.777531623840332,
      "step": 92655
    },
    {
      "epoch": 1.7790794372558595e-05,
      "step": 92655,
      "training_step_time": 7.723304986953735
    },
    {
      "epoch": 1.7791748046875e-05,
      "model_forward_time": 1.8301901817321777,
      "step": 92656
    },
    {
      "epoch": 1.7791748046875e-05,
      "step": 92656,
      "training_step_time": 10.292048215866089
    },
    {
      "epoch": 1.7792701721191406e-05,
      "model_forward_time": 2.419445037841797,
      "step": 92657
    },
    {
      "epoch": 1.7792701721191406e-05,
      "step": 92657,
      "training_step_time": 9.441012620925903
    },
    {
      "epoch": 1.7793655395507814e-05,
      "model_forward_time": 1.7683846950531006,
      "step": 92658
    },
    {
      "epoch": 1.7793655395507814e-05,
      "step": 92658,
      "training_step_time": 7.717396259307861
    },
    {
      "epoch": 1.7794609069824218e-05,
      "model_forward_time": 1.7737042903900146,
      "step": 92659
    },
    {
      "epoch": 1.7794609069824218e-05,
      "step": 92659,
      "training_step_time": 7.5220043659210205
    },
    {
      "epoch": 1.7795562744140625e-05,
      "grad_norm": 0.07836533337831497,
      "learning_rate": 1.4661184700612396e-07,
      "loss": 0.055,
      "step": 92660
    },
    {
      "epoch": 1.7795562744140625e-05,
      "model_forward_time": 1.7710371017456055,
      "step": 92660
    },
    {
      "dynamics_loss_avg": 0.06536064743995666,
      "epoch": 1.7795562744140625e-05,
      "step": 92660
    },
    {
      "action_loss_avg": 0.002314720267895609,
      "epoch": 1.7795562744140625e-05,
      "step": 92660
    },
    {
      "epoch": 1.7795562744140625e-05,
      "step": 92660,
      "training_step_time": 7.7044548988342285
    },
    {
      "epoch": 1.7796516418457033e-05,
      "model_forward_time": 1.7716796398162842,
      "step": 92661
    },
    {
      "epoch": 1.7796516418457033e-05,
      "step": 92661,
      "training_step_time": 6.866574764251709
    },
    {
      "epoch": 1.7797470092773437e-05,
      "model_forward_time": 1.7680206298828125,
      "step": 92662
    },
    {
      "epoch": 1.7797470092773437e-05,
      "step": 92662,
      "training_step_time": 6.850491285324097
    },
    {
      "epoch": 1.7798423767089844e-05,
      "model_forward_time": 1.76749849319458,
      "step": 92663
    },
    {
      "epoch": 1.7798423767089844e-05,
      "step": 92663,
      "training_step_time": 7.296893119812012
    },
    {
      "epoch": 1.779937744140625e-05,
      "model_forward_time": 1.7813184261322021,
      "step": 92664
    },
    {
      "epoch": 1.779937744140625e-05,
      "step": 92664,
      "training_step_time": 7.446597099304199
    },
    {
      "epoch": 1.7800331115722655e-05,
      "model_forward_time": 1.8357031345367432,
      "step": 92665
    },
    {
      "epoch": 1.7800331115722655e-05,
      "step": 92665,
      "training_step_time": 9.131082773208618
    },
    {
      "epoch": 1.7801284790039063e-05,
      "model_forward_time": 1.7681877613067627,
      "step": 92666
    },
    {
      "epoch": 1.7801284790039063e-05,
      "step": 92666,
      "training_step_time": 10.736865043640137
    },
    {
      "epoch": 1.780223846435547e-05,
      "model_forward_time": 1.7665050029754639,
      "step": 92667
    },
    {
      "epoch": 1.780223846435547e-05,
      "step": 92667,
      "training_step_time": 9.314836502075195
    },
    {
      "epoch": 1.7803192138671874e-05,
      "model_forward_time": 1.7665340900421143,
      "step": 92668
    },
    {
      "epoch": 1.7803192138671874e-05,
      "step": 92668,
      "training_step_time": 8.307820081710815
    },
    {
      "epoch": 1.7804145812988282e-05,
      "model_forward_time": 1.7650444507598877,
      "step": 92669
    },
    {
      "epoch": 1.7804145812988282e-05,
      "step": 92669,
      "training_step_time": 7.687061786651611
    },
    {
      "epoch": 1.780509948730469e-05,
      "grad_norm": 0.08166816085577011,
      "learning_rate": 1.4621464308790134e-07,
      "loss": 0.0538,
      "step": 92670
    },
    {
      "epoch": 1.780509948730469e-05,
      "model_forward_time": 1.7681248188018799,
      "step": 92670
    },
    {
      "dynamics_loss_avg": 0.0537042886018753,
      "epoch": 1.780509948730469e-05,
      "step": 92670
    },
    {
      "action_loss_avg": 0.0017142650438472628,
      "epoch": 1.780509948730469e-05,
      "step": 92670
    },
    {
      "epoch": 1.780509948730469e-05,
      "step": 92670,
      "training_step_time": 7.119488954544067
    },
    {
      "epoch": 1.7806053161621093e-05,
      "model_forward_time": 1.7456319332122803,
      "step": 92671
    },
    {
      "epoch": 1.7806053161621093e-05,
      "step": 92671,
      "training_step_time": 6.829217910766602
    },
    {
      "epoch": 1.78070068359375e-05,
      "model_forward_time": 1.7691500186920166,
      "step": 92672
    },
    {
      "epoch": 1.78070068359375e-05,
      "step": 92672,
      "training_step_time": 7.2739667892456055
    },
    {
      "epoch": 1.7807960510253905e-05,
      "model_forward_time": 1.7636017799377441,
      "step": 92673
    },
    {
      "epoch": 1.7807960510253905e-05,
      "step": 92673,
      "training_step_time": 7.4070470333099365
    },
    {
      "epoch": 1.7808914184570312e-05,
      "model_forward_time": 1.8433096408843994,
      "step": 92674
    },
    {
      "epoch": 1.7808914184570312e-05,
      "step": 92674,
      "training_step_time": 7.757588148117065
    },
    {
      "epoch": 1.780986785888672e-05,
      "model_forward_time": 1.76524019241333,
      "step": 92675
    },
    {
      "epoch": 1.780986785888672e-05,
      "step": 92675,
      "training_step_time": 10.978665113449097
    },
    {
      "epoch": 1.7810821533203123e-05,
      "model_forward_time": 2.3766839504241943,
      "step": 92676
    },
    {
      "epoch": 1.7810821533203123e-05,
      "step": 92676,
      "training_step_time": 10.37419056892395
    },
    {
      "epoch": 1.781177520751953e-05,
      "model_forward_time": 1.7442915439605713,
      "step": 92677
    },
    {
      "epoch": 1.781177520751953e-05,
      "step": 92677,
      "training_step_time": 7.663548469543457
    },
    {
      "epoch": 1.7812728881835938e-05,
      "model_forward_time": 1.752983570098877,
      "step": 92678
    },
    {
      "epoch": 1.7812728881835938e-05,
      "step": 92678,
      "training_step_time": 7.746034145355225
    },
    {
      "epoch": 1.7813682556152342e-05,
      "model_forward_time": 1.7980542182922363,
      "step": 92679
    },
    {
      "epoch": 1.7813682556152342e-05,
      "step": 92679,
      "training_step_time": 7.5789690017700195
    },
    {
      "epoch": 1.781463623046875e-05,
      "grad_norm": 0.09143026173114777,
      "learning_rate": 1.4581796997234775e-07,
      "loss": 0.0534,
      "step": 92680
    },
    {
      "epoch": 1.781463623046875e-05,
      "model_forward_time": 1.7357444763183594,
      "step": 92680
    },
    {
      "dynamics_loss_avg": 0.05852146381512284,
      "epoch": 1.781463623046875e-05,
      "step": 92680
    },
    {
      "action_loss_avg": 0.0026522071624640376,
      "epoch": 1.781463623046875e-05,
      "step": 92680
    },
    {
      "epoch": 1.781463623046875e-05,
      "step": 92680,
      "training_step_time": 6.8079893589019775
    },
    {
      "epoch": 1.7815589904785157e-05,
      "model_forward_time": 1.86299467086792,
      "step": 92681
    },
    {
      "epoch": 1.7815589904785157e-05,
      "step": 92681,
      "training_step_time": 8.304192781448364
    },
    {
      "epoch": 1.781654357910156e-05,
      "model_forward_time": 1.8933539390563965,
      "step": 92682
    },
    {
      "epoch": 1.781654357910156e-05,
      "step": 92682,
      "training_step_time": 8.428550481796265
    },
    {
      "epoch": 1.781749725341797e-05,
      "model_forward_time": 1.901080846786499,
      "step": 92683
    },
    {
      "epoch": 1.781749725341797e-05,
      "step": 92683,
      "training_step_time": 8.507542133331299
    },
    {
      "epoch": 1.7818450927734376e-05,
      "model_forward_time": 1.9157297611236572,
      "step": 92684
    },
    {
      "epoch": 1.7818450927734376e-05,
      "step": 92684,
      "training_step_time": 8.26354718208313
    },
    {
      "epoch": 1.781940460205078e-05,
      "model_forward_time": 2.148139238357544,
      "step": 92685
    },
    {
      "epoch": 1.781940460205078e-05,
      "step": 92685,
      "training_step_time": 9.816967725753784
    },
    {
      "epoch": 1.7820358276367187e-05,
      "model_forward_time": 1.890068531036377,
      "step": 92686
    },
    {
      "epoch": 1.7820358276367187e-05,
      "step": 92686,
      "training_step_time": 7.72468113899231
    },
    {
      "epoch": 1.7821311950683595e-05,
      "model_forward_time": 1.8776283264160156,
      "step": 92687
    },
    {
      "epoch": 1.7821311950683595e-05,
      "step": 92687,
      "training_step_time": 7.554002523422241
    },
    {
      "epoch": 1.7822265625e-05,
      "model_forward_time": 1.8951196670532227,
      "step": 92688
    },
    {
      "epoch": 1.7822265625e-05,
      "step": 92688,
      "training_step_time": 7.649596214294434
    },
    {
      "epoch": 1.7823219299316406e-05,
      "model_forward_time": 1.9194886684417725,
      "step": 92689
    },
    {
      "epoch": 1.7823219299316406e-05,
      "step": 92689,
      "training_step_time": 6.841095447540283
    },
    {
      "epoch": 1.7824172973632814e-05,
      "grad_norm": 0.09558728337287903,
      "learning_rate": 1.4542182770284298e-07,
      "loss": 0.0541,
      "step": 92690
    },
    {
      "epoch": 1.7824172973632814e-05,
      "model_forward_time": 1.9225103855133057,
      "step": 92690
    },
    {
      "dynamics_loss_avg": 0.07225812412798405,
      "epoch": 1.7824172973632814e-05,
      "step": 92690
    },
    {
      "action_loss_avg": 0.0029092790209688246,
      "epoch": 1.7824172973632814e-05,
      "step": 92690
    },
    {
      "epoch": 1.7824172973632814e-05,
      "step": 92690,
      "training_step_time": 7.203286647796631
    },
    {
      "epoch": 1.7825126647949218e-05,
      "model_forward_time": 1.9128987789154053,
      "step": 92691
    },
    {
      "epoch": 1.7825126647949218e-05,
      "step": 92691,
      "training_step_time": 7.4041242599487305
    },
    {
      "epoch": 1.7826080322265625e-05,
      "model_forward_time": 1.9273734092712402,
      "step": 92692
    },
    {
      "epoch": 1.7826080322265625e-05,
      "step": 92692,
      "training_step_time": 8.303526639938354
    },
    {
      "epoch": 1.7827033996582033e-05,
      "model_forward_time": 1.9217593669891357,
      "step": 92693
    },
    {
      "epoch": 1.7827033996582033e-05,
      "step": 92693,
      "training_step_time": 13.407219886779785
    },
    {
      "epoch": 1.7827987670898437e-05,
      "model_forward_time": 2.560103178024292,
      "step": 92694
    },
    {
      "epoch": 1.7827987670898437e-05,
      "step": 92694,
      "training_step_time": 10.151248216629028
    },
    {
      "epoch": 1.7828941345214844e-05,
      "model_forward_time": 1.9275670051574707,
      "step": 92695
    },
    {
      "epoch": 1.7828941345214844e-05,
      "step": 92695,
      "training_step_time": 7.700858116149902
    },
    {
      "epoch": 1.782989501953125e-05,
      "model_forward_time": 1.931140661239624,
      "step": 92696
    },
    {
      "epoch": 1.782989501953125e-05,
      "step": 92696,
      "training_step_time": 7.575219392776489
    },
    {
      "epoch": 1.7830848693847655e-05,
      "model_forward_time": 1.9224576950073242,
      "step": 92697
    },
    {
      "epoch": 1.7830848693847655e-05,
      "step": 92697,
      "training_step_time": 7.685650587081909
    },
    {
      "epoch": 1.7831802368164063e-05,
      "model_forward_time": 1.9347281455993652,
      "step": 92698
    },
    {
      "epoch": 1.7831802368164063e-05,
      "step": 92698,
      "training_step_time": 6.823753356933594
    },
    {
      "epoch": 1.783275604248047e-05,
      "model_forward_time": 1.8874297142028809,
      "step": 92699
    },
    {
      "epoch": 1.783275604248047e-05,
      "step": 92699,
      "training_step_time": 7.305553674697876
    },
    {
      "epoch": 1.7833709716796874e-05,
      "grad_norm": 0.09828437119722366,
      "learning_rate": 1.4502621632270842e-07,
      "loss": 0.054,
      "step": 92700
    },
    {
      "epoch": 1.7833709716796874e-05,
      "model_forward_time": 1.9124655723571777,
      "step": 92700
    },
    {
      "dynamics_loss_avg": 0.05309608317911625,
      "epoch": 1.7833709716796874e-05,
      "step": 92700
    },
    {
      "action_loss_avg": 0.0015944483340717853,
      "epoch": 1.7833709716796874e-05,
      "step": 92700
    },
    {
      "epoch": 1.7833709716796874e-05,
      "step": 92700,
      "training_step_time": 7.39870285987854
    },
    {
      "epoch": 1.7834663391113282e-05,
      "model_forward_time": 1.8965060710906982,
      "step": 92701
    },
    {
      "epoch": 1.7834663391113282e-05,
      "step": 92701,
      "training_step_time": 7.712185859680176
    },
    {
      "epoch": 1.783561706542969e-05,
      "model_forward_time": 1.7734606266021729,
      "step": 92702
    },
    {
      "epoch": 1.783561706542969e-05,
      "step": 92702,
      "training_step_time": 10.660029649734497
    },
    {
      "epoch": 1.7836570739746093e-05,
      "model_forward_time": 2.4244472980499268,
      "step": 92703
    },
    {
      "epoch": 1.7836570739746093e-05,
      "step": 92703,
      "training_step_time": 8.487776517868042
    },
    {
      "epoch": 1.78375244140625e-05,
      "model_forward_time": 1.8766400814056396,
      "step": 92704
    },
    {
      "epoch": 1.78375244140625e-05,
      "step": 92704,
      "training_step_time": 7.924203395843506
    },
    {
      "epoch": 1.7838478088378905e-05,
      "model_forward_time": 1.7926006317138672,
      "step": 92705
    },
    {
      "epoch": 1.7838478088378905e-05,
      "step": 92705,
      "training_step_time": 7.507512331008911
    },
    {
      "epoch": 1.7839431762695312e-05,
      "model_forward_time": 1.089799165725708,
      "step": 92706
    },
    {
      "epoch": 1.7839431762695312e-05,
      "step": 92706,
      "training_step_time": 7.743109941482544
    },
    {
      "epoch": 1.784038543701172e-05,
      "model_forward_time": 1.7919414043426514,
      "step": 92707
    },
    {
      "epoch": 1.784038543701172e-05,
      "step": 92707,
      "training_step_time": 7.120445966720581
    },
    {
      "epoch": 1.7841339111328123e-05,
      "model_forward_time": 1.7866711616516113,
      "step": 92708
    },
    {
      "epoch": 1.7841339111328123e-05,
      "step": 92708,
      "training_step_time": 7.246455669403076
    },
    {
      "epoch": 1.784229278564453e-05,
      "model_forward_time": 1.7906227111816406,
      "step": 92709
    },
    {
      "epoch": 1.784229278564453e-05,
      "step": 92709,
      "training_step_time": 8.238028049468994
    },
    {
      "epoch": 1.784324645996094e-05,
      "grad_norm": 0.148843914270401,
      "learning_rate": 1.4463113587520673e-07,
      "loss": 0.0532,
      "step": 92710
    },
    {
      "epoch": 1.784324645996094e-05,
      "model_forward_time": 1.7783541679382324,
      "step": 92710
    },
    {
      "dynamics_loss_avg": 0.03962389267981052,
      "epoch": 1.784324645996094e-05,
      "step": 92710
    },
    {
      "action_loss_avg": 0.0011710499005857856,
      "epoch": 1.784324645996094e-05,
      "step": 92710
    },
    {
      "epoch": 1.784324645996094e-05,
      "step": 92710,
      "training_step_time": 8.735861539840698
    },
    {
      "epoch": 1.7844200134277342e-05,
      "model_forward_time": 1.7823584079742432,
      "step": 92711
    },
    {
      "epoch": 1.7844200134277342e-05,
      "step": 92711,
      "training_step_time": 11.075652360916138
    },
    {
      "epoch": 1.784515380859375e-05,
      "model_forward_time": 1.7735109329223633,
      "step": 92712
    },
    {
      "epoch": 1.784515380859375e-05,
      "step": 92712,
      "training_step_time": 7.821756601333618
    },
    {
      "epoch": 1.7846107482910157e-05,
      "model_forward_time": 2.5015523433685303,
      "step": 92713
    },
    {
      "epoch": 1.7846107482910157e-05,
      "step": 92713,
      "training_step_time": 8.812057733535767
    },
    {
      "epoch": 1.784706115722656e-05,
      "model_forward_time": 1.773085594177246,
      "step": 92714
    },
    {
      "epoch": 1.784706115722656e-05,
      "step": 92714,
      "training_step_time": 7.599441766738892
    },
    {
      "epoch": 1.784801483154297e-05,
      "model_forward_time": 1.7829396724700928,
      "step": 92715
    },
    {
      "epoch": 1.784801483154297e-05,
      "step": 92715,
      "training_step_time": 7.804443597793579
    },
    {
      "epoch": 1.7848968505859376e-05,
      "model_forward_time": 1.7822816371917725,
      "step": 92716
    },
    {
      "epoch": 1.7848968505859376e-05,
      "step": 92716,
      "training_step_time": 7.130162715911865
    },
    {
      "epoch": 1.784992218017578e-05,
      "model_forward_time": 1.7860629558563232,
      "step": 92717
    },
    {
      "epoch": 1.784992218017578e-05,
      "step": 92717,
      "training_step_time": 7.263121604919434
    },
    {
      "epoch": 1.7850875854492187e-05,
      "model_forward_time": 1.7543225288391113,
      "step": 92718
    },
    {
      "epoch": 1.7850875854492187e-05,
      "step": 92718,
      "training_step_time": 9.311323881149292
    },
    {
      "epoch": 1.7851829528808595e-05,
      "model_forward_time": 1.7755415439605713,
      "step": 92719
    },
    {
      "epoch": 1.7851829528808595e-05,
      "step": 92719,
      "training_step_time": 7.924502611160278
    },
    {
      "epoch": 1.7852783203125e-05,
      "grad_norm": 0.07526582479476929,
      "learning_rate": 1.4423658640354442e-07,
      "loss": 0.0534,
      "step": 92720
    },
    {
      "epoch": 1.7852783203125e-05,
      "model_forward_time": 1.767408847808838,
      "step": 92720
    },
    {
      "dynamics_loss_avg": 0.04289620248600841,
      "epoch": 1.7852783203125e-05,
      "step": 92720
    },
    {
      "action_loss_avg": 0.0011309737397823482,
      "epoch": 1.7852783203125e-05,
      "step": 92720
    },
    {
      "epoch": 1.7852783203125e-05,
      "step": 92720,
      "training_step_time": 7.738383769989014
    },
    {
      "epoch": 1.7853736877441406e-05,
      "model_forward_time": 1.7705028057098389,
      "step": 92721
    },
    {
      "epoch": 1.7853736877441406e-05,
      "step": 92721,
      "training_step_time": 9.690231800079346
    },
    {
      "epoch": 1.7854690551757814e-05,
      "model_forward_time": 2.4566490650177,
      "step": 92722
    },
    {
      "epoch": 1.7854690551757814e-05,
      "step": 92722,
      "training_step_time": 9.839438199996948
    },
    {
      "epoch": 1.7855644226074218e-05,
      "model_forward_time": 1.774717092514038,
      "step": 92723
    },
    {
      "epoch": 1.7855644226074218e-05,
      "step": 92723,
      "training_step_time": 8.208319664001465
    },
    {
      "epoch": 1.7856597900390625e-05,
      "model_forward_time": 1.7810089588165283,
      "step": 92724
    },
    {
      "epoch": 1.7856597900390625e-05,
      "step": 92724,
      "training_step_time": 8.542062759399414
    },
    {
      "epoch": 1.7857551574707033e-05,
      "model_forward_time": 1.7818784713745117,
      "step": 92725
    },
    {
      "epoch": 1.7857551574707033e-05,
      "step": 92725,
      "training_step_time": 7.156350374221802
    },
    {
      "epoch": 1.7858505249023437e-05,
      "model_forward_time": 1.7833044528961182,
      "step": 92726
    },
    {
      "epoch": 1.7858505249023437e-05,
      "step": 92726,
      "training_step_time": 7.234853506088257
    },
    {
      "epoch": 1.7859458923339844e-05,
      "model_forward_time": 1.7839884757995605,
      "step": 92727
    },
    {
      "epoch": 1.7859458923339844e-05,
      "step": 92727,
      "training_step_time": 7.3925628662109375
    },
    {
      "epoch": 1.786041259765625e-05,
      "model_forward_time": 1.7823500633239746,
      "step": 92728
    },
    {
      "epoch": 1.786041259765625e-05,
      "step": 92728,
      "training_step_time": 8.377912759780884
    },
    {
      "epoch": 1.7861366271972655e-05,
      "model_forward_time": 1.7852699756622314,
      "step": 92729
    },
    {
      "epoch": 1.7861366271972655e-05,
      "step": 92729,
      "training_step_time": 7.676777601242065
    },
    {
      "epoch": 1.7862319946289063e-05,
      "grad_norm": 0.0875277891755104,
      "learning_rate": 1.438425679508687e-07,
      "loss": 0.0539,
      "step": 92730
    },
    {
      "epoch": 1.7862319946289063e-05,
      "model_forward_time": 1.4111297130584717,
      "step": 92730
    },
    {
      "dynamics_loss_avg": 0.054064495116472246,
      "epoch": 1.7862319946289063e-05,
      "step": 92730
    },
    {
      "action_loss_avg": 0.0011991058505373075,
      "epoch": 1.7862319946289063e-05,
      "step": 92730
    },
    {
      "epoch": 1.7862319946289063e-05,
      "step": 92730,
      "training_step_time": 9.632891654968262
    },
    {
      "epoch": 1.786327362060547e-05,
      "model_forward_time": 2.423916816711426,
      "step": 92731
    },
    {
      "epoch": 1.786327362060547e-05,
      "step": 92731,
      "training_step_time": 8.609464406967163
    },
    {
      "epoch": 1.7864227294921874e-05,
      "model_forward_time": 1.7797489166259766,
      "step": 92732
    },
    {
      "epoch": 1.7864227294921874e-05,
      "step": 92732,
      "training_step_time": 8.999219179153442
    },
    {
      "epoch": 1.7865180969238282e-05,
      "model_forward_time": 1.4229271411895752,
      "step": 92733
    },
    {
      "epoch": 1.7865180969238282e-05,
      "step": 92733,
      "training_step_time": 7.885891675949097
    },
    {
      "epoch": 1.786613464355469e-05,
      "model_forward_time": 1.7816524505615234,
      "step": 92734
    },
    {
      "epoch": 1.786613464355469e-05,
      "step": 92734,
      "training_step_time": 8.205596685409546
    },
    {
      "epoch": 1.7867088317871093e-05,
      "model_forward_time": 1.7792353630065918,
      "step": 92735
    },
    {
      "epoch": 1.7867088317871093e-05,
      "step": 92735,
      "training_step_time": 7.210063219070435
    },
    {
      "epoch": 1.78680419921875e-05,
      "model_forward_time": 1.772792100906372,
      "step": 92736
    },
    {
      "epoch": 1.78680419921875e-05,
      "step": 92736,
      "training_step_time": 7.365410566329956
    },
    {
      "epoch": 1.7868995666503905e-05,
      "model_forward_time": 1.7868068218231201,
      "step": 92737
    },
    {
      "epoch": 1.7868995666503905e-05,
      "step": 92737,
      "training_step_time": 7.361295461654663
    },
    {
      "epoch": 1.7869949340820312e-05,
      "model_forward_time": 1.776890754699707,
      "step": 92738
    },
    {
      "epoch": 1.7869949340820312e-05,
      "step": 92738,
      "training_step_time": 9.733360528945923
    },
    {
      "epoch": 1.787090301513672e-05,
      "model_forward_time": 1.7757997512817383,
      "step": 92739
    },
    {
      "epoch": 1.787090301513672e-05,
      "step": 92739,
      "training_step_time": 9.88768720626831
    },
    {
      "epoch": 1.7871856689453124e-05,
      "grad_norm": 0.08317410200834274,
      "learning_rate": 1.4344908056026897e-07,
      "loss": 0.0544,
      "step": 92740
    },
    {
      "epoch": 1.7871856689453124e-05,
      "model_forward_time": 1.7743487358093262,
      "step": 92740
    },
    {
      "dynamics_loss_avg": 0.05159062687307596,
      "epoch": 1.7871856689453124e-05,
      "step": 92740
    },
    {
      "action_loss_avg": 0.0014080873486818745,
      "epoch": 1.7871856689453124e-05,
      "step": 92740
    },
    {
      "epoch": 1.7871856689453124e-05,
      "step": 92740,
      "training_step_time": 9.332136869430542
    },
    {
      "epoch": 1.787281036376953e-05,
      "model_forward_time": 2.456634521484375,
      "step": 92741
    },
    {
      "epoch": 1.787281036376953e-05,
      "step": 92741,
      "training_step_time": 7.865322828292847
    },
    {
      "epoch": 1.787376403808594e-05,
      "model_forward_time": 1.7861196994781494,
      "step": 92742
    },
    {
      "epoch": 1.787376403808594e-05,
      "step": 92742,
      "training_step_time": 7.47950553894043
    },
    {
      "epoch": 1.7874717712402342e-05,
      "model_forward_time": 1.7920379638671875,
      "step": 92743
    },
    {
      "epoch": 1.7874717712402342e-05,
      "step": 92743,
      "training_step_time": 7.799620866775513
    },
    {
      "epoch": 1.787567138671875e-05,
      "model_forward_time": 1.7926039695739746,
      "step": 92744
    },
    {
      "epoch": 1.787567138671875e-05,
      "step": 92744,
      "training_step_time": 7.2676591873168945
    },
    {
      "epoch": 1.7876625061035157e-05,
      "model_forward_time": 1.7907838821411133,
      "step": 92745
    },
    {
      "epoch": 1.7876625061035157e-05,
      "step": 92745,
      "training_step_time": 7.387486457824707
    },
    {
      "epoch": 1.787757873535156e-05,
      "model_forward_time": 1.8002252578735352,
      "step": 92746
    },
    {
      "epoch": 1.787757873535156e-05,
      "step": 92746,
      "training_step_time": 7.383649587631226
    },
    {
      "epoch": 1.787853240966797e-05,
      "model_forward_time": 1.1033861637115479,
      "step": 92747
    },
    {
      "epoch": 1.787853240966797e-05,
      "step": 92747,
      "training_step_time": 7.770185708999634
    },
    {
      "epoch": 1.7879486083984376e-05,
      "model_forward_time": 1.7869811058044434,
      "step": 92748
    },
    {
      "epoch": 1.7879486083984376e-05,
      "step": 92748,
      "training_step_time": 9.601294994354248
    },
    {
      "epoch": 1.788043975830078e-05,
      "model_forward_time": 1.7876296043395996,
      "step": 92749
    },
    {
      "epoch": 1.788043975830078e-05,
      "step": 92749,
      "training_step_time": 8.672425270080566
    },
    {
      "epoch": 1.7881393432617188e-05,
      "grad_norm": 0.08953101187944412,
      "learning_rate": 1.4305612427477522e-07,
      "loss": 0.0541,
      "step": 92750
    },
    {
      "epoch": 1.7881393432617188e-05,
      "model_forward_time": 2.4474329948425293,
      "step": 92750
    },
    {
      "dynamics_loss_avg": 0.04773292988538742,
      "epoch": 1.7881393432617188e-05,
      "step": 92750
    },
    {
      "action_loss_avg": 0.0011297000106424092,
      "epoch": 1.7881393432617188e-05,
      "step": 92750
    },
    {
      "epoch": 1.7881393432617188e-05,
      "step": 92750,
      "training_step_time": 9.4395432472229
    },
    {
      "epoch": 1.7882347106933595e-05,
      "model_forward_time": 1.793088674545288,
      "step": 92751
    },
    {
      "epoch": 1.7882347106933595e-05,
      "step": 92751,
      "training_step_time": 7.633118629455566
    },
    {
      "epoch": 1.788330078125e-05,
      "model_forward_time": 1.787153720855713,
      "step": 92752
    },
    {
      "epoch": 1.788330078125e-05,
      "step": 92752,
      "training_step_time": 7.706012964248657
    },
    {
      "epoch": 1.7884254455566406e-05,
      "model_forward_time": 1.8345141410827637,
      "step": 92753
    },
    {
      "epoch": 1.7884254455566406e-05,
      "step": 92753,
      "training_step_time": 7.697749137878418
    },
    {
      "epoch": 1.7885208129882814e-05,
      "model_forward_time": 1.7961018085479736,
      "step": 92754
    },
    {
      "epoch": 1.7885208129882814e-05,
      "step": 92754,
      "training_step_time": 6.794650316238403
    },
    {
      "epoch": 1.7886161804199218e-05,
      "model_forward_time": 1.4602093696594238,
      "step": 92755
    },
    {
      "epoch": 1.7886161804199218e-05,
      "step": 92755,
      "training_step_time": 7.318387508392334
    },
    {
      "epoch": 1.7887115478515625e-05,
      "model_forward_time": 1.7845704555511475,
      "step": 92756
    },
    {
      "epoch": 1.7887115478515625e-05,
      "step": 92756,
      "training_step_time": 8.42787480354309
    },
    {
      "epoch": 1.7888069152832033e-05,
      "model_forward_time": 1.8211722373962402,
      "step": 92757
    },
    {
      "epoch": 1.7888069152832033e-05,
      "step": 92757,
      "training_step_time": 9.721261501312256
    },
    {
      "epoch": 1.7889022827148437e-05,
      "model_forward_time": 1.0759944915771484,
      "step": 92758
    },
    {
      "epoch": 1.7889022827148437e-05,
      "step": 92758,
      "training_step_time": 8.806195974349976
    },
    {
      "epoch": 1.7889976501464844e-05,
      "model_forward_time": 1.4548039436340332,
      "step": 92759
    },
    {
      "epoch": 1.7889976501464844e-05,
      "step": 92759,
      "training_step_time": 9.367952108383179
    },
    {
      "epoch": 1.789093017578125e-05,
      "grad_norm": 0.07645082473754883,
      "learning_rate": 1.42663699137362e-07,
      "loss": 0.0543,
      "step": 92760
    },
    {
      "epoch": 1.789093017578125e-05,
      "model_forward_time": 2.887859582901001,
      "step": 92760
    },
    {
      "dynamics_loss_avg": 0.06647658217698335,
      "epoch": 1.789093017578125e-05,
      "step": 92760
    },
    {
      "action_loss_avg": 0.0016163106600288301,
      "epoch": 1.789093017578125e-05,
      "step": 92760
    },
    {
      "epoch": 1.789093017578125e-05,
      "step": 92760,
      "training_step_time": 7.8733813762664795
    },
    {
      "epoch": 1.7891883850097656e-05,
      "model_forward_time": 1.7365138530731201,
      "step": 92761
    },
    {
      "epoch": 1.7891883850097656e-05,
      "step": 92761,
      "training_step_time": 7.615624189376831
    },
    {
      "epoch": 1.7892837524414063e-05,
      "model_forward_time": 1.7322914600372314,
      "step": 92762
    },
    {
      "epoch": 1.7892837524414063e-05,
      "step": 92762,
      "training_step_time": 7.690895080566406
    },
    {
      "epoch": 1.789379119873047e-05,
      "model_forward_time": 1.7378361225128174,
      "step": 92763
    },
    {
      "epoch": 1.789379119873047e-05,
      "step": 92763,
      "training_step_time": 6.8019702434539795
    },
    {
      "epoch": 1.7894744873046874e-05,
      "model_forward_time": 1.7362990379333496,
      "step": 92764
    },
    {
      "epoch": 1.7894744873046874e-05,
      "step": 92764,
      "training_step_time": 7.38483190536499
    },
    {
      "epoch": 1.7895698547363282e-05,
      "model_forward_time": 1.7330303192138672,
      "step": 92765
    },
    {
      "epoch": 1.7895698547363282e-05,
      "step": 92765,
      "training_step_time": 7.676641464233398
    },
    {
      "epoch": 1.7896652221679686e-05,
      "model_forward_time": 1.730318307876587,
      "step": 92766
    },
    {
      "epoch": 1.7896652221679686e-05,
      "step": 92766,
      "training_step_time": 8.542160511016846
    },
    {
      "epoch": 1.7897605895996093e-05,
      "model_forward_time": 1.7673211097717285,
      "step": 92767
    },
    {
      "epoch": 1.7897605895996093e-05,
      "step": 92767,
      "training_step_time": 7.732797384262085
    },
    {
      "epoch": 1.78985595703125e-05,
      "model_forward_time": 1.7328965663909912,
      "step": 92768
    },
    {
      "epoch": 1.78985595703125e-05,
      "step": 92768,
      "training_step_time": 9.579257011413574
    },
    {
      "epoch": 1.7899513244628905e-05,
      "model_forward_time": 2.8281636238098145,
      "step": 92769
    },
    {
      "epoch": 1.7899513244628905e-05,
      "step": 92769,
      "training_step_time": 7.720928192138672
    },
    {
      "epoch": 1.7900466918945312e-05,
      "grad_norm": 0.08772346377372742,
      "learning_rate": 1.4227180519094386e-07,
      "loss": 0.054,
      "step": 92770
    },
    {
      "epoch": 1.7900466918945312e-05,
      "model_forward_time": 1.7299659252166748,
      "step": 92770
    },
    {
      "dynamics_loss_avg": 0.04883597735315561,
      "epoch": 1.7900466918945312e-05,
      "step": 92770
    },
    {
      "action_loss_avg": 0.001524496599449776,
      "epoch": 1.7900466918945312e-05,
      "step": 92770
    },
    {
      "epoch": 1.7900466918945312e-05,
      "step": 92770,
      "training_step_time": 7.645901441574097
    },
    {
      "epoch": 1.790142059326172e-05,
      "model_forward_time": 1.730916976928711,
      "step": 92771
    },
    {
      "epoch": 1.790142059326172e-05,
      "step": 92771,
      "training_step_time": 7.648824453353882
    },
    {
      "epoch": 1.7902374267578124e-05,
      "model_forward_time": 1.739314079284668,
      "step": 92772
    },
    {
      "epoch": 1.7902374267578124e-05,
      "step": 92772,
      "training_step_time": 6.808841705322266
    },
    {
      "epoch": 1.790332794189453e-05,
      "model_forward_time": 1.7457334995269775,
      "step": 92773
    },
    {
      "epoch": 1.790332794189453e-05,
      "step": 92773,
      "training_step_time": 7.363813400268555
    },
    {
      "epoch": 1.790428161621094e-05,
      "model_forward_time": 1.7362236976623535,
      "step": 92774
    },
    {
      "epoch": 1.790428161621094e-05,
      "step": 92774,
      "training_step_time": 7.872665882110596
    },
    {
      "epoch": 1.7905235290527342e-05,
      "model_forward_time": 1.3497066497802734,
      "step": 92775
    },
    {
      "epoch": 1.7905235290527342e-05,
      "step": 92775,
      "training_step_time": 7.6468586921691895
    },
    {
      "epoch": 1.790618896484375e-05,
      "model_forward_time": 1.7638936042785645,
      "step": 92776
    },
    {
      "epoch": 1.790618896484375e-05,
      "step": 92776,
      "training_step_time": 8.743929147720337
    },
    {
      "epoch": 1.7907142639160157e-05,
      "model_forward_time": 1.7407958507537842,
      "step": 92777
    },
    {
      "epoch": 1.7907142639160157e-05,
      "step": 92777,
      "training_step_time": 9.352657556533813
    },
    {
      "epoch": 1.790809631347656e-05,
      "model_forward_time": 1.7416012287139893,
      "step": 92778
    },
    {
      "epoch": 1.790809631347656e-05,
      "step": 92778,
      "training_step_time": 8.674142599105835
    },
    {
      "epoch": 1.790904998779297e-05,
      "model_forward_time": 2.8749337196350098,
      "step": 92779
    },
    {
      "epoch": 1.790904998779297e-05,
      "step": 92779,
      "training_step_time": 7.685619354248047
    },
    {
      "epoch": 1.7910003662109376e-05,
      "grad_norm": 0.09755494445562363,
      "learning_rate": 1.4188044247837706e-07,
      "loss": 0.0553,
      "step": 92780
    },
    {
      "epoch": 1.7910003662109376e-05,
      "model_forward_time": 1.725127935409546,
      "step": 92780
    },
    {
      "dynamics_loss_avg": 0.05426925886422396,
      "epoch": 1.7910003662109376e-05,
      "step": 92780
    },
    {
      "action_loss_avg": 0.001566794782411307,
      "epoch": 1.7910003662109376e-05,
      "step": 92780
    },
    {
      "epoch": 1.7910003662109376e-05,
      "step": 92780,
      "training_step_time": 8.19041395187378
    },
    {
      "epoch": 1.791095733642578e-05,
      "model_forward_time": 1.7438080310821533,
      "step": 92781
    },
    {
      "epoch": 1.791095733642578e-05,
      "step": 92781,
      "training_step_time": 6.8030664920806885
    },
    {
      "epoch": 1.7911911010742188e-05,
      "model_forward_time": 1.7467551231384277,
      "step": 92782
    },
    {
      "epoch": 1.7911911010742188e-05,
      "step": 92782,
      "training_step_time": 7.2338035106658936
    },
    {
      "epoch": 1.7912864685058595e-05,
      "model_forward_time": 1.7323734760284424,
      "step": 92783
    },
    {
      "epoch": 1.7912864685058595e-05,
      "step": 92783,
      "training_step_time": 7.32387375831604
    },
    {
      "epoch": 1.7913818359375e-05,
      "model_forward_time": 1.7629132270812988,
      "step": 92784
    },
    {
      "epoch": 1.7913818359375e-05,
      "step": 92784,
      "training_step_time": 7.661257743835449
    },
    {
      "epoch": 1.7914772033691406e-05,
      "model_forward_time": 1.7633287906646729,
      "step": 92785
    },
    {
      "epoch": 1.7914772033691406e-05,
      "step": 92785,
      "training_step_time": 9.701602935791016
    },
    {
      "epoch": 1.7915725708007814e-05,
      "model_forward_time": 1.7349658012390137,
      "step": 92786
    },
    {
      "epoch": 1.7915725708007814e-05,
      "step": 92786,
      "training_step_time": 8.65716290473938
    },
    {
      "epoch": 1.7916679382324218e-05,
      "model_forward_time": 1.7371065616607666,
      "step": 92787
    },
    {
      "epoch": 1.7916679382324218e-05,
      "step": 92787,
      "training_step_time": 7.692091226577759
    },
    {
      "epoch": 1.7917633056640625e-05,
      "model_forward_time": 2.8385589122772217,
      "step": 92788
    },
    {
      "epoch": 1.7917633056640625e-05,
      "step": 92788,
      "training_step_time": 7.644044637680054
    },
    {
      "epoch": 1.7918586730957033e-05,
      "model_forward_time": 1.7250525951385498,
      "step": 92789
    },
    {
      "epoch": 1.7918586730957033e-05,
      "step": 92789,
      "training_step_time": 7.693753957748413
    },
    {
      "epoch": 1.7919540405273437e-05,
      "grad_norm": 0.08440294861793518,
      "learning_rate": 1.4148961104246128e-07,
      "loss": 0.0543,
      "step": 92790
    },
    {
      "epoch": 1.7919540405273437e-05,
      "model_forward_time": 1.7315053939819336,
      "step": 92790
    },
    {
      "dynamics_loss_avg": 0.034961515106260777,
      "epoch": 1.7919540405273437e-05,
      "step": 92790
    },
    {
      "action_loss_avg": 0.001239256074768491,
      "epoch": 1.7919540405273437e-05,
      "step": 92790
    },
    {
      "epoch": 1.7919540405273437e-05,
      "step": 92790,
      "training_step_time": 7.633203506469727
    },
    {
      "epoch": 1.7920494079589844e-05,
      "model_forward_time": 1.728527307510376,
      "step": 92791
    },
    {
      "epoch": 1.7920494079589844e-05,
      "step": 92791,
      "training_step_time": 6.798480033874512
    },
    {
      "epoch": 1.792144775390625e-05,
      "model_forward_time": 1.7310035228729248,
      "step": 92792
    },
    {
      "epoch": 1.792144775390625e-05,
      "step": 92792,
      "training_step_time": 7.342058420181274
    },
    {
      "epoch": 1.7922401428222656e-05,
      "model_forward_time": 1.7500362396240234,
      "step": 92793
    },
    {
      "epoch": 1.7922401428222656e-05,
      "step": 92793,
      "training_step_time": 9.402852535247803
    },
    {
      "epoch": 1.7923355102539063e-05,
      "model_forward_time": 1.7637126445770264,
      "step": 92794
    },
    {
      "epoch": 1.7923355102539063e-05,
      "step": 92794,
      "training_step_time": 8.537239789962769
    },
    {
      "epoch": 1.792430877685547e-05,
      "model_forward_time": 1.7332713603973389,
      "step": 92795
    },
    {
      "epoch": 1.792430877685547e-05,
      "step": 92795,
      "training_step_time": 11.155421733856201
    },
    {
      "epoch": 1.7925262451171874e-05,
      "model_forward_time": 1.732377529144287,
      "step": 92796
    },
    {
      "epoch": 1.7925262451171874e-05,
      "step": 92796,
      "training_step_time": 7.809615612030029
    },
    {
      "epoch": 1.7926216125488282e-05,
      "model_forward_time": 2.8316452503204346,
      "step": 92797
    },
    {
      "epoch": 1.7926216125488282e-05,
      "step": 92797,
      "training_step_time": 7.659369945526123
    },
    {
      "epoch": 1.7927169799804686e-05,
      "model_forward_time": 1.7355554103851318,
      "step": 92798
    },
    {
      "epoch": 1.7927169799804686e-05,
      "step": 92798,
      "training_step_time": 7.701671838760376
    },
    {
      "epoch": 1.7928123474121093e-05,
      "model_forward_time": 1.737217664718628,
      "step": 92799
    },
    {
      "epoch": 1.7928123474121093e-05,
      "step": 92799,
      "training_step_time": 7.634995937347412
    },
    {
      "epoch": 1.79290771484375e-05,
      "grad_norm": 0.11828307062387466,
      "learning_rate": 1.4109931092593732e-07,
      "loss": 0.0565,
      "step": 92800
    },
    {
      "epoch": 1.79290771484375e-05,
      "model_forward_time": 1.7315833568572998,
      "step": 92800
    },
    {
      "dynamics_loss_avg": 0.061397481709718704,
      "epoch": 1.79290771484375e-05,
      "step": 92800
    },
    {
      "action_loss_avg": 0.0010793606139486655,
      "epoch": 1.79290771484375e-05,
      "step": 92800
    },
    {
      "epoch": 1.79290771484375e-05,
      "step": 92800,
      "training_step_time": 6.781853199005127
    },
    {
      "epoch": 1.7930030822753905e-05,
      "model_forward_time": 1.7284917831420898,
      "step": 92801
    },
    {
      "epoch": 1.7930030822753905e-05,
      "step": 92801,
      "training_step_time": 7.419596910476685
    },
    {
      "epoch": 1.7930984497070312e-05,
      "model_forward_time": 1.7443015575408936,
      "step": 92802
    },
    {
      "epoch": 1.7930984497070312e-05,
      "step": 92802,
      "training_step_time": 7.736230850219727
    },
    {
      "epoch": 1.793193817138672e-05,
      "model_forward_time": 1.7552783489227295,
      "step": 92803
    },
    {
      "epoch": 1.793193817138672e-05,
      "step": 92803,
      "training_step_time": 8.71127986907959
    },
    {
      "epoch": 1.7932891845703124e-05,
      "model_forward_time": 1.7288227081298828,
      "step": 92804
    },
    {
      "epoch": 1.7932891845703124e-05,
      "step": 92804,
      "training_step_time": 8.78349256515503
    },
    {
      "epoch": 1.793384552001953e-05,
      "model_forward_time": 1.7344508171081543,
      "step": 92805
    },
    {
      "epoch": 1.793384552001953e-05,
      "step": 92805,
      "training_step_time": 7.721689462661743
    },
    {
      "epoch": 1.793479919433594e-05,
      "model_forward_time": 2.8112080097198486,
      "step": 92806
    },
    {
      "epoch": 1.793479919433594e-05,
      "step": 92806,
      "training_step_time": 8.50413203239441
    },
    {
      "epoch": 1.7935752868652342e-05,
      "model_forward_time": 1.7202792167663574,
      "step": 92807
    },
    {
      "epoch": 1.7935752868652342e-05,
      "step": 92807,
      "training_step_time": 7.591929912567139
    },
    {
      "epoch": 1.793670654296875e-05,
      "model_forward_time": 1.7345750331878662,
      "step": 92808
    },
    {
      "epoch": 1.793670654296875e-05,
      "step": 92808,
      "training_step_time": 7.506459474563599
    },
    {
      "epoch": 1.7937660217285157e-05,
      "model_forward_time": 1.7260065078735352,
      "step": 92809
    },
    {
      "epoch": 1.7937660217285157e-05,
      "step": 92809,
      "training_step_time": 7.697746992111206
    },
    {
      "epoch": 1.793861389160156e-05,
      "grad_norm": 0.10157004743814468,
      "learning_rate": 1.4070954217148659e-07,
      "loss": 0.0561,
      "step": 92810
    },
    {
      "epoch": 1.793861389160156e-05,
      "model_forward_time": 1.727391004562378,
      "step": 92810
    },
    {
      "dynamics_loss_avg": 0.058091893792152405,
      "epoch": 1.793861389160156e-05,
      "step": 92810
    },
    {
      "action_loss_avg": 0.001917447050800547,
      "epoch": 1.793861389160156e-05,
      "step": 92810
    },
    {
      "epoch": 1.793861389160156e-05,
      "step": 92810,
      "training_step_time": 7.384970188140869
    },
    {
      "epoch": 1.793956756591797e-05,
      "model_forward_time": 1.7407848834991455,
      "step": 92811
    },
    {
      "epoch": 1.793956756591797e-05,
      "step": 92811,
      "training_step_time": 7.699027061462402
    },
    {
      "epoch": 1.7940521240234376e-05,
      "model_forward_time": 1.7571299076080322,
      "step": 92812
    },
    {
      "epoch": 1.7940521240234376e-05,
      "step": 92812,
      "training_step_time": 9.33828330039978
    },
    {
      "epoch": 1.794147491455078e-05,
      "model_forward_time": 1.7289788722991943,
      "step": 92813
    },
    {
      "epoch": 1.794147491455078e-05,
      "step": 92813,
      "training_step_time": 8.523881912231445
    },
    {
      "epoch": 1.7942428588867188e-05,
      "model_forward_time": 1.7411978244781494,
      "step": 92814
    },
    {
      "epoch": 1.7942428588867188e-05,
      "step": 92814,
      "training_step_time": 9.592982769012451
    },
    {
      "epoch": 1.7943382263183595e-05,
      "model_forward_time": 2.779520273208618,
      "step": 92815
    },
    {
      "epoch": 1.7943382263183595e-05,
      "step": 92815,
      "training_step_time": 9.59799861907959
    },
    {
      "epoch": 1.79443359375e-05,
      "model_forward_time": 1.7292799949645996,
      "step": 92816
    },
    {
      "epoch": 1.79443359375e-05,
      "step": 92816,
      "training_step_time": 7.615151643753052
    },
    {
      "epoch": 1.7945289611816406e-05,
      "model_forward_time": 1.7364258766174316,
      "step": 92817
    },
    {
      "epoch": 1.7945289611816406e-05,
      "step": 92817,
      "training_step_time": 7.822972774505615
    },
    {
      "epoch": 1.7946243286132814e-05,
      "model_forward_time": 1.7241520881652832,
      "step": 92818
    },
    {
      "epoch": 1.7946243286132814e-05,
      "step": 92818,
      "training_step_time": 7.688909292221069
    },
    {
      "epoch": 1.7947196960449218e-05,
      "model_forward_time": 1.7371711730957031,
      "step": 92819
    },
    {
      "epoch": 1.7947196960449218e-05,
      "step": 92819,
      "training_step_time": 8.179130792617798
    },
    {
      "epoch": 1.7948150634765625e-05,
      "grad_norm": 0.0709511786699295,
      "learning_rate": 1.4032030482173498e-07,
      "loss": 0.0564,
      "step": 92820
    },
    {
      "epoch": 1.7948150634765625e-05,
      "model_forward_time": 1.728971004486084,
      "step": 92820
    },
    {
      "dynamics_loss_avg": 0.05059434724971652,
      "epoch": 1.7948150634765625e-05,
      "step": 92820
    },
    {
      "action_loss_avg": 0.0016341997281415388,
      "epoch": 1.7948150634765625e-05,
      "step": 92820
    },
    {
      "epoch": 1.7948150634765625e-05,
      "step": 92820,
      "training_step_time": 8.662284851074219
    },
    {
      "epoch": 1.7949104309082033e-05,
      "model_forward_time": 1.759152889251709,
      "step": 92821
    },
    {
      "epoch": 1.7949104309082033e-05,
      "step": 92821,
      "training_step_time": 7.665220499038696
    },
    {
      "epoch": 1.7950057983398437e-05,
      "model_forward_time": 1.7311666011810303,
      "step": 92822
    },
    {
      "epoch": 1.7950057983398437e-05,
      "step": 92822,
      "training_step_time": 10.073258399963379
    },
    {
      "epoch": 1.7951011657714844e-05,
      "model_forward_time": 1.7343690395355225,
      "step": 92823
    },
    {
      "epoch": 1.7951011657714844e-05,
      "step": 92823,
      "training_step_time": 9.208787202835083
    },
    {
      "epoch": 1.795196533203125e-05,
      "model_forward_time": 1.7203760147094727,
      "step": 92824
    },
    {
      "epoch": 1.795196533203125e-05,
      "step": 92824,
      "training_step_time": 9.488002300262451
    },
    {
      "epoch": 1.7952919006347656e-05,
      "model_forward_time": 1.7362406253814697,
      "step": 92825
    },
    {
      "epoch": 1.7952919006347656e-05,
      "step": 92825,
      "training_step_time": 7.67989706993103
    },
    {
      "epoch": 1.7953872680664063e-05,
      "model_forward_time": 1.7340288162231445,
      "step": 92826
    },
    {
      "epoch": 1.7953872680664063e-05,
      "step": 92826,
      "training_step_time": 7.639025449752808
    },
    {
      "epoch": 1.795482635498047e-05,
      "model_forward_time": 1.7287485599517822,
      "step": 92827
    },
    {
      "epoch": 1.795482635498047e-05,
      "step": 92827,
      "training_step_time": 7.812209367752075
    },
    {
      "epoch": 1.7955780029296874e-05,
      "model_forward_time": 1.7238633632659912,
      "step": 92828
    },
    {
      "epoch": 1.7955780029296874e-05,
      "step": 92828,
      "training_step_time": 7.391883373260498
    },
    {
      "epoch": 1.7956733703613282e-05,
      "model_forward_time": 1.7464203834533691,
      "step": 92829
    },
    {
      "epoch": 1.7956733703613282e-05,
      "step": 92829,
      "training_step_time": 8.710074186325073
    },
    {
      "epoch": 1.7957687377929686e-05,
      "grad_norm": 0.11377186328172684,
      "learning_rate": 1.39931598919249e-07,
      "loss": 0.0565,
      "step": 92830
    },
    {
      "epoch": 1.7957687377929686e-05,
      "model_forward_time": 1.8252215385437012,
      "step": 92830
    },
    {
      "dynamics_loss_avg": 0.047467919811606404,
      "epoch": 1.7957687377929686e-05,
      "step": 92830
    },
    {
      "action_loss_avg": 0.001799812848912552,
      "epoch": 1.7957687377929686e-05,
      "step": 92830
    },
    {
      "epoch": 1.7957687377929686e-05,
      "step": 92830,
      "training_step_time": 7.743790149688721
    },
    {
      "epoch": 1.7958641052246093e-05,
      "model_forward_time": 1.7255375385284424,
      "step": 92831
    },
    {
      "epoch": 1.7958641052246093e-05,
      "step": 92831,
      "training_step_time": 9.449090003967285
    },
    {
      "epoch": 1.79595947265625e-05,
      "model_forward_time": 1.7100062370300293,
      "step": 92832
    },
    {
      "epoch": 1.79595947265625e-05,
      "step": 92832,
      "training_step_time": 9.138753175735474
    },
    {
      "epoch": 1.7960548400878905e-05,
      "model_forward_time": 1.7316951751708984,
      "step": 92833
    },
    {
      "epoch": 1.7960548400878905e-05,
      "step": 92833,
      "training_step_time": 9.349353075027466
    },
    {
      "epoch": 1.7961502075195312e-05,
      "model_forward_time": 2.8223447799682617,
      "step": 92834
    },
    {
      "epoch": 1.7961502075195312e-05,
      "step": 92834,
      "training_step_time": 7.709430456161499
    },
    {
      "epoch": 1.796245574951172e-05,
      "model_forward_time": 1.7251505851745605,
      "step": 92835
    },
    {
      "epoch": 1.796245574951172e-05,
      "step": 92835,
      "training_step_time": 7.5603954792022705
    },
    {
      "epoch": 1.7963409423828124e-05,
      "model_forward_time": 1.736219882965088,
      "step": 92836
    },
    {
      "epoch": 1.7963409423828124e-05,
      "step": 92836,
      "training_step_time": 7.730928182601929
    },
    {
      "epoch": 1.796436309814453e-05,
      "model_forward_time": 1.724433183670044,
      "step": 92837
    },
    {
      "epoch": 1.796436309814453e-05,
      "step": 92837,
      "training_step_time": 7.394012451171875
    },
    {
      "epoch": 1.796531677246094e-05,
      "model_forward_time": 1.7295007705688477,
      "step": 92838
    },
    {
      "epoch": 1.796531677246094e-05,
      "step": 92838,
      "training_step_time": 9.779901027679443
    },
    {
      "epoch": 1.7966270446777342e-05,
      "model_forward_time": 1.816171646118164,
      "step": 92839
    },
    {
      "epoch": 1.7966270446777342e-05,
      "step": 92839,
      "training_step_time": 7.806337594985962
    },
    {
      "epoch": 1.796722412109375e-05,
      "grad_norm": 0.09995101392269135,
      "learning_rate": 1.3954342450653523e-07,
      "loss": 0.0577,
      "step": 92840
    },
    {
      "epoch": 1.796722412109375e-05,
      "model_forward_time": 1.7232894897460938,
      "step": 92840
    },
    {
      "dynamics_loss_avg": 0.0530444635078311,
      "epoch": 1.796722412109375e-05,
      "step": 92840
    },
    {
      "action_loss_avg": 0.0016361710106139072,
      "epoch": 1.796722412109375e-05,
      "step": 92840
    },
    {
      "epoch": 1.796722412109375e-05,
      "step": 92840,
      "training_step_time": 7.762069225311279
    },
    {
      "epoch": 1.7968177795410157e-05,
      "model_forward_time": 1.7300426959991455,
      "step": 92841
    },
    {
      "epoch": 1.7968177795410157e-05,
      "step": 92841,
      "training_step_time": 9.622166156768799
    },
    {
      "epoch": 1.796913146972656e-05,
      "model_forward_time": 1.730410099029541,
      "step": 92842
    },
    {
      "epoch": 1.796913146972656e-05,
      "step": 92842,
      "training_step_time": 7.743790626525879
    },
    {
      "epoch": 1.797008514404297e-05,
      "model_forward_time": 2.8297085762023926,
      "step": 92843
    },
    {
      "epoch": 1.797008514404297e-05,
      "step": 92843,
      "training_step_time": 8.704440355300903
    },
    {
      "epoch": 1.7971038818359376e-05,
      "model_forward_time": 1.736698865890503,
      "step": 92844
    },
    {
      "epoch": 1.7971038818359376e-05,
      "step": 92844,
      "training_step_time": 8.240793228149414
    },
    {
      "epoch": 1.797199249267578e-05,
      "model_forward_time": 1.729109287261963,
      "step": 92845
    },
    {
      "epoch": 1.797199249267578e-05,
      "step": 92845,
      "training_step_time": 7.274409532546997
    },
    {
      "epoch": 1.7972946166992188e-05,
      "model_forward_time": 1.7287371158599854,
      "step": 92846
    },
    {
      "epoch": 1.7972946166992188e-05,
      "step": 92846,
      "training_step_time": 7.749783992767334
    },
    {
      "epoch": 1.7973899841308595e-05,
      "model_forward_time": 1.743605375289917,
      "step": 92847
    },
    {
      "epoch": 1.7973899841308595e-05,
      "step": 92847,
      "training_step_time": 7.7237889766693115
    },
    {
      "epoch": 1.7974853515625e-05,
      "model_forward_time": 1.8074772357940674,
      "step": 92848
    },
    {
      "epoch": 1.7974853515625e-05,
      "step": 92848,
      "training_step_time": 7.796021461486816
    },
    {
      "epoch": 1.7975807189941406e-05,
      "model_forward_time": 1.7888429164886475,
      "step": 92849
    },
    {
      "epoch": 1.7975807189941406e-05,
      "step": 92849,
      "training_step_time": 8.725307941436768
    },
    {
      "epoch": 1.7976760864257814e-05,
      "grad_norm": 0.09492985904216766,
      "learning_rate": 1.3915578162604582e-07,
      "loss": 0.0568,
      "step": 92850
    },
    {
      "epoch": 1.7976760864257814e-05,
      "model_forward_time": 1.7423038482666016,
      "step": 92850
    },
    {
      "dynamics_loss_avg": 0.038597177155315875,
      "epoch": 1.7976760864257814e-05,
      "step": 92850
    },
    {
      "action_loss_avg": 0.0013672361383214593,
      "epoch": 1.7976760864257814e-05,
      "step": 92850
    },
    {
      "epoch": 1.7976760864257814e-05,
      "step": 92850,
      "training_step_time": 7.759949684143066
    },
    {
      "epoch": 1.7977714538574218e-05,
      "model_forward_time": 1.7134339809417725,
      "step": 92851
    },
    {
      "epoch": 1.7977714538574218e-05,
      "step": 92851,
      "training_step_time": 7.784793853759766
    },
    {
      "epoch": 1.7978668212890625e-05,
      "model_forward_time": 2.866960287094116,
      "step": 92852
    },
    {
      "epoch": 1.7978668212890625e-05,
      "step": 92852,
      "training_step_time": 7.807601451873779
    },
    {
      "epoch": 1.7979621887207033e-05,
      "model_forward_time": 1.7339000701904297,
      "step": 92853
    },
    {
      "epoch": 1.7979621887207033e-05,
      "step": 92853,
      "training_step_time": 7.647279739379883
    },
    {
      "epoch": 1.7980575561523437e-05,
      "model_forward_time": 1.8059213161468506,
      "step": 92854
    },
    {
      "epoch": 1.7980575561523437e-05,
      "step": 92854,
      "training_step_time": 7.4875593185424805
    },
    {
      "epoch": 1.7981529235839844e-05,
      "model_forward_time": 1.776090383529663,
      "step": 92855
    },
    {
      "epoch": 1.7981529235839844e-05,
      "step": 92855,
      "training_step_time": 7.679522275924683
    },
    {
      "epoch": 1.798248291015625e-05,
      "model_forward_time": 1.8038930892944336,
      "step": 92856
    },
    {
      "epoch": 1.798248291015625e-05,
      "step": 92856,
      "training_step_time": 8.74901294708252
    },
    {
      "epoch": 1.7983436584472656e-05,
      "model_forward_time": 1.7728261947631836,
      "step": 92857
    },
    {
      "epoch": 1.7983436584472656e-05,
      "step": 92857,
      "training_step_time": 7.582249402999878
    },
    {
      "epoch": 1.7984390258789063e-05,
      "model_forward_time": 1.7615854740142822,
      "step": 92858
    },
    {
      "epoch": 1.7984390258789063e-05,
      "step": 92858,
      "training_step_time": 7.688017845153809
    },
    {
      "epoch": 1.798534393310547e-05,
      "model_forward_time": 1.7857038974761963,
      "step": 92859
    },
    {
      "epoch": 1.798534393310547e-05,
      "step": 92859,
      "training_step_time": 8.249280214309692
    },
    {
      "epoch": 1.7986297607421874e-05,
      "grad_norm": 0.08799058198928833,
      "learning_rate": 1.3876867032017126e-07,
      "loss": 0.058,
      "step": 92860
    },
    {
      "epoch": 1.7986297607421874e-05,
      "model_forward_time": 1.7417418956756592,
      "step": 92860
    },
    {
      "dynamics_loss_avg": 0.06074298927560449,
      "epoch": 1.7986297607421874e-05,
      "step": 92860
    },
    {
      "action_loss_avg": 0.0021627397800330073,
      "epoch": 1.7986297607421874e-05,
      "step": 92860
    },
    {
      "epoch": 1.7986297607421874e-05,
      "step": 92860,
      "training_step_time": 8.50828242301941
    },
    {
      "epoch": 1.7987251281738282e-05,
      "model_forward_time": 2.8325040340423584,
      "step": 92861
    },
    {
      "epoch": 1.7987251281738282e-05,
      "step": 92861,
      "training_step_time": 9.40099287033081
    },
    {
      "epoch": 1.7988204956054686e-05,
      "model_forward_time": 1.7474989891052246,
      "step": 92862
    },
    {
      "epoch": 1.7988204956054686e-05,
      "step": 92862,
      "training_step_time": 8.54682183265686
    },
    {
      "epoch": 1.7989158630371093e-05,
      "model_forward_time": 1.7334880828857422,
      "step": 92863
    },
    {
      "epoch": 1.7989158630371093e-05,
      "step": 92863,
      "training_step_time": 7.862083196640015
    },
    {
      "epoch": 1.79901123046875e-05,
      "model_forward_time": 1.7337801456451416,
      "step": 92864
    },
    {
      "epoch": 1.79901123046875e-05,
      "step": 92864,
      "training_step_time": 7.6980249881744385
    },
    {
      "epoch": 1.7991065979003905e-05,
      "model_forward_time": 1.7472589015960693,
      "step": 92865
    },
    {
      "epoch": 1.7991065979003905e-05,
      "step": 92865,
      "training_step_time": 7.755966901779175
    },
    {
      "epoch": 1.7992019653320312e-05,
      "model_forward_time": 1.7424588203430176,
      "step": 92866
    },
    {
      "epoch": 1.7992019653320312e-05,
      "step": 92866,
      "training_step_time": 7.638373851776123
    },
    {
      "epoch": 1.799297332763672e-05,
      "model_forward_time": 1.7576792240142822,
      "step": 92867
    },
    {
      "epoch": 1.799297332763672e-05,
      "step": 92867,
      "training_step_time": 7.7632362842559814
    },
    {
      "epoch": 1.7993927001953124e-05,
      "model_forward_time": 1.746199607849121,
      "step": 92868
    },
    {
      "epoch": 1.7993927001953124e-05,
      "step": 92868,
      "training_step_time": 7.647953987121582
    },
    {
      "epoch": 1.799488067626953e-05,
      "model_forward_time": 1.7404470443725586,
      "step": 92869
    },
    {
      "epoch": 1.799488067626953e-05,
      "step": 92869,
      "training_step_time": 8.933772325515747
    },
    {
      "epoch": 1.799583435058594e-05,
      "grad_norm": 0.12362932413816452,
      "learning_rate": 1.3838209063124718e-07,
      "loss": 0.0581,
      "step": 92870
    },
    {
      "epoch": 1.799583435058594e-05,
      "model_forward_time": 2.831352710723877,
      "step": 92870
    },
    {
      "dynamics_loss_avg": 0.060508594289422034,
      "epoch": 1.799583435058594e-05,
      "step": 92870
    },
    {
      "action_loss_avg": 0.0014329456927953288,
      "epoch": 1.799583435058594e-05,
      "step": 92870
    },
    {
      "epoch": 1.799583435058594e-05,
      "step": 92870,
      "training_step_time": 7.74672532081604
    },
    {
      "epoch": 1.7996788024902342e-05,
      "model_forward_time": 1.748321294784546,
      "step": 92871
    },
    {
      "epoch": 1.7996788024902342e-05,
      "step": 92871,
      "training_step_time": 8.520817518234253
    },
    {
      "epoch": 1.799774169921875e-05,
      "model_forward_time": 1.7346017360687256,
      "step": 92872
    },
    {
      "epoch": 1.799774169921875e-05,
      "step": 92872,
      "training_step_time": 7.543920040130615
    },
    {
      "epoch": 1.7998695373535157e-05,
      "model_forward_time": 1.739501953125,
      "step": 92873
    },
    {
      "epoch": 1.7998695373535157e-05,
      "step": 92873,
      "training_step_time": 7.6451733112335205
    },
    {
      "epoch": 1.799964904785156e-05,
      "model_forward_time": 1.751967191696167,
      "step": 92874
    },
    {
      "epoch": 1.799964904785156e-05,
      "step": 92874,
      "training_step_time": 7.688497543334961
    },
    {
      "epoch": 1.800060272216797e-05,
      "model_forward_time": 1.737454891204834,
      "step": 92875
    },
    {
      "epoch": 1.800060272216797e-05,
      "step": 92875,
      "training_step_time": 8.093361139297485
    },
    {
      "epoch": 1.8001556396484376e-05,
      "model_forward_time": 1.7538018226623535,
      "step": 92876
    },
    {
      "epoch": 1.8001556396484376e-05,
      "step": 92876,
      "training_step_time": 7.68625283241272
    },
    {
      "epoch": 1.800251007080078e-05,
      "model_forward_time": 1.7435557842254639,
      "step": 92877
    },
    {
      "epoch": 1.800251007080078e-05,
      "step": 92877,
      "training_step_time": 9.488677024841309
    },
    {
      "epoch": 1.8003463745117188e-05,
      "model_forward_time": 1.7411577701568604,
      "step": 92878
    },
    {
      "epoch": 1.8003463745117188e-05,
      "step": 92878,
      "training_step_time": 7.748739004135132
    },
    {
      "epoch": 1.8004417419433595e-05,
      "model_forward_time": 2.8498988151550293,
      "step": 92879
    },
    {
      "epoch": 1.8004417419433595e-05,
      "step": 92879,
      "training_step_time": 7.689076900482178
    },
    {
      "epoch": 1.800537109375e-05,
      "grad_norm": 0.15083158016204834,
      "learning_rate": 1.3799604260154753e-07,
      "loss": 0.0581,
      "step": 92880
    },
    {
      "epoch": 1.800537109375e-05,
      "model_forward_time": 1.7416372299194336,
      "step": 92880
    },
    {
      "dynamics_loss_avg": 0.055239968188107016,
      "epoch": 1.800537109375e-05,
      "step": 92880
    },
    {
      "action_loss_avg": 0.0012921866204123945,
      "epoch": 1.800537109375e-05,
      "step": 92880
    },
    {
      "epoch": 1.800537109375e-05,
      "step": 92880,
      "training_step_time": 7.777680397033691
    },
    {
      "epoch": 1.8006324768066406e-05,
      "model_forward_time": 1.7400565147399902,
      "step": 92881
    },
    {
      "epoch": 1.8006324768066406e-05,
      "step": 92881,
      "training_step_time": 8.120096921920776
    },
    {
      "epoch": 1.8007278442382814e-05,
      "model_forward_time": 1.7350382804870605,
      "step": 92882
    },
    {
      "epoch": 1.8007278442382814e-05,
      "step": 92882,
      "training_step_time": 7.414896488189697
    },
    {
      "epoch": 1.8008232116699218e-05,
      "model_forward_time": 1.748600959777832,
      "step": 92883
    },
    {
      "epoch": 1.8008232116699218e-05,
      "step": 92883,
      "training_step_time": 8.517234802246094
    },
    {
      "epoch": 1.8009185791015625e-05,
      "model_forward_time": 1.736360788345337,
      "step": 92884
    },
    {
      "epoch": 1.8009185791015625e-05,
      "step": 92884,
      "training_step_time": 8.177253246307373
    },
    {
      "epoch": 1.8010139465332033e-05,
      "model_forward_time": 1.8087100982666016,
      "step": 92885
    },
    {
      "epoch": 1.8010139465332033e-05,
      "step": 92885,
      "training_step_time": 9.006960391998291
    },
    {
      "epoch": 1.8011093139648437e-05,
      "model_forward_time": 1.7383677959442139,
      "step": 92886
    },
    {
      "epoch": 1.8011093139648437e-05,
      "step": 92886,
      "training_step_time": 8.708019256591797
    },
    {
      "epoch": 1.8012046813964844e-05,
      "model_forward_time": 1.734863519668579,
      "step": 92887
    },
    {
      "epoch": 1.8012046813964844e-05,
      "step": 92887,
      "training_step_time": 9.803319931030273
    },
    {
      "epoch": 1.801300048828125e-05,
      "model_forward_time": 2.8085575103759766,
      "step": 92888
    },
    {
      "epoch": 1.801300048828125e-05,
      "step": 92888,
      "training_step_time": 8.521119594573975
    },
    {
      "epoch": 1.8013954162597656e-05,
      "model_forward_time": 1.7334613800048828,
      "step": 92889
    },
    {
      "epoch": 1.8013954162597656e-05,
      "step": 92889,
      "training_step_time": 7.756307363510132
    },
    {
      "epoch": 1.8014907836914063e-05,
      "grad_norm": 0.104741171002388,
      "learning_rate": 1.376105262732913e-07,
      "loss": 0.0566,
      "step": 92890
    },
    {
      "epoch": 1.8014907836914063e-05,
      "model_forward_time": 1.7392282485961914,
      "step": 92890
    },
    {
      "dynamics_loss_avg": 0.055186356604099276,
      "epoch": 1.8014907836914063e-05,
      "step": 92890
    },
    {
      "action_loss_avg": 0.0014596947614336387,
      "epoch": 1.8014907836914063e-05,
      "step": 92890
    },
    {
      "epoch": 1.8014907836914063e-05,
      "step": 92890,
      "training_step_time": 7.676730632781982
    },
    {
      "epoch": 1.8015861511230467e-05,
      "model_forward_time": 1.736809492111206,
      "step": 92891
    },
    {
      "epoch": 1.8015861511230467e-05,
      "step": 92891,
      "training_step_time": 7.527577877044678
    },
    {
      "epoch": 1.8016815185546874e-05,
      "model_forward_time": 1.7374134063720703,
      "step": 92892
    },
    {
      "epoch": 1.8016815185546874e-05,
      "step": 92892,
      "training_step_time": 7.670613765716553
    },
    {
      "epoch": 1.8017768859863282e-05,
      "model_forward_time": 1.7340986728668213,
      "step": 92893
    },
    {
      "epoch": 1.8017768859863282e-05,
      "step": 92893,
      "training_step_time": 8.486388206481934
    },
    {
      "epoch": 1.8018722534179686e-05,
      "model_forward_time": 1.7846438884735107,
      "step": 92894
    },
    {
      "epoch": 1.8018722534179686e-05,
      "step": 92894,
      "training_step_time": 9.438607215881348
    },
    {
      "epoch": 1.8019676208496093e-05,
      "model_forward_time": 1.7444772720336914,
      "step": 92895
    },
    {
      "epoch": 1.8019676208496093e-05,
      "step": 92895,
      "training_step_time": 7.746164560317993
    },
    {
      "epoch": 1.80206298828125e-05,
      "model_forward_time": 1.354717493057251,
      "step": 92896
    },
    {
      "epoch": 1.80206298828125e-05,
      "step": 92896,
      "training_step_time": 7.761494398117065
    },
    {
      "epoch": 1.8021583557128905e-05,
      "model_forward_time": 2.815462350845337,
      "step": 92897
    },
    {
      "epoch": 1.8021583557128905e-05,
      "step": 92897,
      "training_step_time": 7.788619756698608
    },
    {
      "epoch": 1.8022537231445312e-05,
      "model_forward_time": 1.725027322769165,
      "step": 92898
    },
    {
      "epoch": 1.8022537231445312e-05,
      "step": 92898,
      "training_step_time": 7.778852939605713
    },
    {
      "epoch": 1.802349090576172e-05,
      "model_forward_time": 1.736093521118164,
      "step": 92899
    },
    {
      "epoch": 1.802349090576172e-05,
      "step": 92899,
      "training_step_time": 7.714423418045044
    },
    {
      "epoch": 1.8024444580078124e-05,
      "grad_norm": 0.09299074858427048,
      "learning_rate": 1.37225541688637e-07,
      "loss": 0.0584,
      "step": 92900
    },
    {
      "epoch": 1.8024444580078124e-05,
      "model_forward_time": 1.7312042713165283,
      "step": 92900
    },
    {
      "dynamics_loss_avg": 0.05644321590662003,
      "epoch": 1.8024444580078124e-05,
      "step": 92900
    },
    {
      "action_loss_avg": 0.0010584167728666216,
      "epoch": 1.8024444580078124e-05,
      "step": 92900
    },
    {
      "epoch": 1.8024444580078124e-05,
      "step": 92900,
      "training_step_time": 7.490039110183716
    },
    {
      "epoch": 1.802539825439453e-05,
      "model_forward_time": 1.73484206199646,
      "step": 92901
    },
    {
      "epoch": 1.802539825439453e-05,
      "step": 92901,
      "training_step_time": 8.590885639190674
    },
    {
      "epoch": 1.802635192871094e-05,
      "model_forward_time": 1.7483394145965576,
      "step": 92902
    },
    {
      "epoch": 1.802635192871094e-05,
      "step": 92902,
      "training_step_time": 7.674236059188843
    },
    {
      "epoch": 1.8027305603027343e-05,
      "model_forward_time": 1.7552821636199951,
      "step": 92903
    },
    {
      "epoch": 1.8027305603027343e-05,
      "step": 92903,
      "training_step_time": 8.494593381881714
    },
    {
      "epoch": 1.802825927734375e-05,
      "model_forward_time": 1.7526187896728516,
      "step": 92904
    },
    {
      "epoch": 1.802825927734375e-05,
      "step": 92904,
      "training_step_time": 7.751944303512573
    },
    {
      "epoch": 1.8029212951660157e-05,
      "model_forward_time": 1.73653244972229,
      "step": 92905
    },
    {
      "epoch": 1.8029212951660157e-05,
      "step": 92905,
      "training_step_time": 9.0406494140625
    },
    {
      "epoch": 1.803016662597656e-05,
      "model_forward_time": 2.8353116512298584,
      "step": 92906
    },
    {
      "epoch": 1.803016662597656e-05,
      "step": 92906,
      "training_step_time": 10.202373266220093
    },
    {
      "epoch": 1.803112030029297e-05,
      "model_forward_time": 1.3519737720489502,
      "step": 92907
    },
    {
      "epoch": 1.803112030029297e-05,
      "step": 92907,
      "training_step_time": 10.198797225952148
    },
    {
      "epoch": 1.8032073974609376e-05,
      "model_forward_time": 1.739546537399292,
      "step": 92908
    },
    {
      "epoch": 1.8032073974609376e-05,
      "step": 92908,
      "training_step_time": 7.713292837142944
    },
    {
      "epoch": 1.803302764892578e-05,
      "model_forward_time": 1.7403531074523926,
      "step": 92909
    },
    {
      "epoch": 1.803302764892578e-05,
      "step": 92909,
      "training_step_time": 7.4486212730407715
    },
    {
      "epoch": 1.8033981323242188e-05,
      "grad_norm": 0.10410533100366592,
      "learning_rate": 1.3684108888968596e-07,
      "loss": 0.0559,
      "step": 92910
    },
    {
      "epoch": 1.8033981323242188e-05,
      "model_forward_time": 1.7273163795471191,
      "step": 92910
    },
    {
      "dynamics_loss_avg": 0.05732046589255333,
      "epoch": 1.8033981323242188e-05,
      "step": 92910
    },
    {
      "action_loss_avg": 0.0011434603686211631,
      "epoch": 1.8033981323242188e-05,
      "step": 92910
    },
    {
      "epoch": 1.8033981323242188e-05,
      "step": 92910,
      "training_step_time": 7.484189987182617
    },
    {
      "epoch": 1.8034934997558595e-05,
      "model_forward_time": 1.7374403476715088,
      "step": 92911
    },
    {
      "epoch": 1.8034934997558595e-05,
      "step": 92911,
      "training_step_time": 8.60551404953003
    },
    {
      "epoch": 1.8035888671875e-05,
      "model_forward_time": 1.3662586212158203,
      "step": 92912
    },
    {
      "epoch": 1.8035888671875e-05,
      "step": 92912,
      "training_step_time": 7.703684329986572
    },
    {
      "epoch": 1.8036842346191406e-05,
      "model_forward_time": 1.7551994323730469,
      "step": 92913
    },
    {
      "epoch": 1.8036842346191406e-05,
      "step": 92913,
      "training_step_time": 7.6772074699401855
    },
    {
      "epoch": 1.8037796020507814e-05,
      "model_forward_time": 1.7311527729034424,
      "step": 92914
    },
    {
      "epoch": 1.8037796020507814e-05,
      "step": 92914,
      "training_step_time": 7.6647047996521
    },
    {
      "epoch": 1.8038749694824218e-05,
      "model_forward_time": 1.743337869644165,
      "step": 92915
    },
    {
      "epoch": 1.8038749694824218e-05,
      "step": 92915,
      "training_step_time": 7.642231225967407
    },
    {
      "epoch": 1.8039703369140625e-05,
      "model_forward_time": 1.735828161239624,
      "step": 92916
    },
    {
      "epoch": 1.8039703369140625e-05,
      "step": 92916,
      "training_step_time": 8.228200912475586
    },
    {
      "epoch": 1.8040657043457033e-05,
      "model_forward_time": 1.7406444549560547,
      "step": 92917
    },
    {
      "epoch": 1.8040657043457033e-05,
      "step": 92917,
      "training_step_time": 7.640340805053711
    },
    {
      "epoch": 1.8041610717773437e-05,
      "model_forward_time": 1.7444565296173096,
      "step": 92918
    },
    {
      "epoch": 1.8041610717773437e-05,
      "step": 92918,
      "training_step_time": 7.533283233642578
    },
    {
      "epoch": 1.8042564392089844e-05,
      "model_forward_time": 1.7396972179412842,
      "step": 92919
    },
    {
      "epoch": 1.8042564392089844e-05,
      "step": 92919,
      "training_step_time": 7.518941640853882
    },
    {
      "epoch": 1.804351806640625e-05,
      "grad_norm": 0.1371411383152008,
      "learning_rate": 1.3645716791848228e-07,
      "loss": 0.057,
      "step": 92920
    },
    {
      "epoch": 1.804351806640625e-05,
      "model_forward_time": 1.7325208187103271,
      "step": 92920
    },
    {
      "dynamics_loss_avg": 0.05112478695809841,
      "epoch": 1.804351806640625e-05,
      "step": 92920
    },
    {
      "action_loss_avg": 0.0012722097337245942,
      "epoch": 1.804351806640625e-05,
      "step": 92920
    },
    {
      "epoch": 1.804351806640625e-05,
      "step": 92920,
      "training_step_time": 7.59834623336792
    },
    {
      "epoch": 1.8044471740722656e-05,
      "model_forward_time": 1.7478079795837402,
      "step": 92921
    },
    {
      "epoch": 1.8044471740722656e-05,
      "step": 92921,
      "training_step_time": 8.646754741668701
    },
    {
      "epoch": 1.8045425415039063e-05,
      "model_forward_time": 1.7526726722717285,
      "step": 92922
    },
    {
      "epoch": 1.8045425415039063e-05,
      "step": 92922,
      "training_step_time": 7.565406084060669
    },
    {
      "epoch": 1.8046379089355467e-05,
      "model_forward_time": 1.7348318099975586,
      "step": 92923
    },
    {
      "epoch": 1.8046379089355467e-05,
      "step": 92923,
      "training_step_time": 8.321280241012573
    },
    {
      "epoch": 1.8047332763671875e-05,
      "model_forward_time": 1.3612592220306396,
      "step": 92924
    },
    {
      "epoch": 1.8047332763671875e-05,
      "step": 92924,
      "training_step_time": 8.713229656219482
    },
    {
      "epoch": 1.8048286437988282e-05,
      "model_forward_time": 2.856566905975342,
      "step": 92925
    },
    {
      "epoch": 1.8048286437988282e-05,
      "step": 92925,
      "training_step_time": 7.690399169921875
    },
    {
      "epoch": 1.8049240112304686e-05,
      "model_forward_time": 1.7368578910827637,
      "step": 92926
    },
    {
      "epoch": 1.8049240112304686e-05,
      "step": 92926,
      "training_step_time": 7.7401158809661865
    },
    {
      "epoch": 1.8050193786621093e-05,
      "model_forward_time": 1.7511887550354004,
      "step": 92927
    },
    {
      "epoch": 1.8050193786621093e-05,
      "step": 92927,
      "training_step_time": 7.448941946029663
    },
    {
      "epoch": 1.80511474609375e-05,
      "model_forward_time": 1.7435593605041504,
      "step": 92928
    },
    {
      "epoch": 1.80511474609375e-05,
      "step": 92928,
      "training_step_time": 7.4481446743011475
    },
    {
      "epoch": 1.8052101135253905e-05,
      "model_forward_time": 1.734464406967163,
      "step": 92929
    },
    {
      "epoch": 1.8052101135253905e-05,
      "step": 92929,
      "training_step_time": 7.714353561401367
    },
    {
      "epoch": 1.8053054809570312e-05,
      "grad_norm": 0.08209338784217834,
      "learning_rate": 1.3607377881701078e-07,
      "loss": 0.0574,
      "step": 92930
    },
    {
      "epoch": 1.8053054809570312e-05,
      "model_forward_time": 1.744819164276123,
      "step": 92930
    },
    {
      "dynamics_loss_avg": 0.07111006947234273,
      "epoch": 1.8053054809570312e-05,
      "step": 92930
    },
    {
      "action_loss_avg": 0.0018570452753920107,
      "epoch": 1.8053054809570312e-05,
      "step": 92930
    },
    {
      "epoch": 1.8053054809570312e-05,
      "step": 92930,
      "training_step_time": 8.310090065002441
    },
    {
      "epoch": 1.805400848388672e-05,
      "model_forward_time": 1.8187196254730225,
      "step": 92931
    },
    {
      "epoch": 1.805400848388672e-05,
      "step": 92931,
      "training_step_time": 8.62648057937622
    },
    {
      "epoch": 1.8054962158203124e-05,
      "model_forward_time": 1.7361574172973633,
      "step": 92932
    },
    {
      "epoch": 1.8054962158203124e-05,
      "step": 92932,
      "training_step_time": 7.6618123054504395
    },
    {
      "epoch": 1.805591583251953e-05,
      "model_forward_time": 1.7502167224884033,
      "step": 92933
    },
    {
      "epoch": 1.805591583251953e-05,
      "step": 92933,
      "training_step_time": 8.720155477523804
    },
    {
      "epoch": 1.805686950683594e-05,
      "model_forward_time": 2.8484466075897217,
      "step": 92934
    },
    {
      "epoch": 1.805686950683594e-05,
      "step": 92934,
      "training_step_time": 9.701746940612793
    },
    {
      "epoch": 1.8057823181152343e-05,
      "model_forward_time": 1.7450156211853027,
      "step": 92935
    },
    {
      "epoch": 1.8057823181152343e-05,
      "step": 92935,
      "training_step_time": 7.581413507461548
    },
    {
      "epoch": 1.805877685546875e-05,
      "model_forward_time": 1.7376751899719238,
      "step": 92936
    },
    {
      "epoch": 1.805877685546875e-05,
      "step": 92936,
      "training_step_time": 7.65790581703186
    },
    {
      "epoch": 1.8059730529785157e-05,
      "model_forward_time": 1.7377159595489502,
      "step": 92937
    },
    {
      "epoch": 1.8059730529785157e-05,
      "step": 92937,
      "training_step_time": 7.517510414123535
    },
    {
      "epoch": 1.806068420410156e-05,
      "model_forward_time": 1.7452282905578613,
      "step": 92938
    },
    {
      "epoch": 1.806068420410156e-05,
      "step": 92938,
      "training_step_time": 9.471917152404785
    },
    {
      "epoch": 1.806163787841797e-05,
      "model_forward_time": 1.7340764999389648,
      "step": 92939
    },
    {
      "epoch": 1.806163787841797e-05,
      "step": 92939,
      "training_step_time": 8.123447179794312
    },
    {
      "epoch": 1.8062591552734376e-05,
      "grad_norm": 0.06853113323450089,
      "learning_rate": 1.3569092162719788e-07,
      "loss": 0.0565,
      "step": 92940
    },
    {
      "epoch": 1.8062591552734376e-05,
      "model_forward_time": 1.790269136428833,
      "step": 92940
    },
    {
      "dynamics_loss_avg": 0.04566130582243204,
      "epoch": 1.8062591552734376e-05,
      "step": 92940
    },
    {
      "action_loss_avg": 0.0011957101465668528,
      "epoch": 1.8062591552734376e-05,
      "step": 92940
    },
    {
      "epoch": 1.8062591552734376e-05,
      "step": 92940,
      "training_step_time": 8.701346635818481
    },
    {
      "epoch": 1.806354522705078e-05,
      "model_forward_time": 1.7367644309997559,
      "step": 92941
    },
    {
      "epoch": 1.806354522705078e-05,
      "step": 92941,
      "training_step_time": 7.716953754425049
    },
    {
      "epoch": 1.8064498901367188e-05,
      "model_forward_time": 1.7356979846954346,
      "step": 92942
    },
    {
      "epoch": 1.8064498901367188e-05,
      "step": 92942,
      "training_step_time": 8.59659457206726
    },
    {
      "epoch": 1.8065452575683595e-05,
      "model_forward_time": 2.4331166744232178,
      "step": 92943
    },
    {
      "epoch": 1.8065452575683595e-05,
      "step": 92943,
      "training_step_time": 8.581687688827515
    },
    {
      "epoch": 1.806640625e-05,
      "model_forward_time": 1.7397878170013428,
      "step": 92944
    },
    {
      "epoch": 1.806640625e-05,
      "step": 92944,
      "training_step_time": 8.846606731414795
    },
    {
      "epoch": 1.8067359924316407e-05,
      "model_forward_time": 1.7446162700653076,
      "step": 92945
    },
    {
      "epoch": 1.8067359924316407e-05,
      "step": 92945,
      "training_step_time": 8.120595693588257
    },
    {
      "epoch": 1.8068313598632814e-05,
      "model_forward_time": 1.7328441143035889,
      "step": 92946
    },
    {
      "epoch": 1.8068313598632814e-05,
      "step": 92946,
      "training_step_time": 8.047764539718628
    },
    {
      "epoch": 1.8069267272949218e-05,
      "model_forward_time": 1.7352988719940186,
      "step": 92947
    },
    {
      "epoch": 1.8069267272949218e-05,
      "step": 92947,
      "training_step_time": 7.6051905155181885
    },
    {
      "epoch": 1.8070220947265625e-05,
      "model_forward_time": 1.744260311126709,
      "step": 92948
    },
    {
      "epoch": 1.8070220947265625e-05,
      "step": 92948,
      "training_step_time": 7.594602108001709
    },
    {
      "epoch": 1.8071174621582033e-05,
      "model_forward_time": 1.7748043537139893,
      "step": 92949
    },
    {
      "epoch": 1.8071174621582033e-05,
      "step": 92949,
      "training_step_time": 8.487416505813599
    },
    {
      "epoch": 1.8072128295898437e-05,
      "grad_norm": 0.08686663955450058,
      "learning_rate": 1.3530859639091232e-07,
      "loss": 0.0572,
      "step": 92950
    },
    {
      "epoch": 1.8072128295898437e-05,
      "model_forward_time": 1.0158681869506836,
      "step": 92950
    },
    {
      "dynamics_loss_avg": 0.058395340107381344,
      "epoch": 1.8072128295898437e-05,
      "step": 92950
    },
    {
      "action_loss_avg": 0.0015271225158357993,
      "epoch": 1.8072128295898437e-05,
      "step": 92950
    },
    {
      "epoch": 1.8072128295898437e-05,
      "step": 92950,
      "training_step_time": 7.658218145370483
    },
    {
      "epoch": 1.8073081970214844e-05,
      "model_forward_time": 1.7457306385040283,
      "step": 92951
    },
    {
      "epoch": 1.8073081970214844e-05,
      "step": 92951,
      "training_step_time": 7.752775430679321
    },
    {
      "epoch": 1.807403564453125e-05,
      "model_forward_time": 1.7275722026824951,
      "step": 92952
    },
    {
      "epoch": 1.807403564453125e-05,
      "step": 92952,
      "training_step_time": 7.704052209854126
    },
    {
      "epoch": 1.8074989318847656e-05,
      "model_forward_time": 2.844801425933838,
      "step": 92953
    },
    {
      "epoch": 1.8074989318847656e-05,
      "step": 92953,
      "training_step_time": 8.274855136871338
    },
    {
      "epoch": 1.8075942993164063e-05,
      "model_forward_time": 1.3499343395233154,
      "step": 92954
    },
    {
      "epoch": 1.8075942993164063e-05,
      "step": 92954,
      "training_step_time": 7.667953729629517
    },
    {
      "epoch": 1.8076896667480467e-05,
      "model_forward_time": 1.7398619651794434,
      "step": 92955
    },
    {
      "epoch": 1.8076896667480467e-05,
      "step": 92955,
      "training_step_time": 7.681674242019653
    },
    {
      "epoch": 1.8077850341796875e-05,
      "model_forward_time": 1.7314832210540771,
      "step": 92956
    },
    {
      "epoch": 1.8077850341796875e-05,
      "step": 92956,
      "training_step_time": 7.1956493854522705
    },
    {
      "epoch": 1.8078804016113282e-05,
      "model_forward_time": 1.7274043560028076,
      "step": 92957
    },
    {
      "epoch": 1.8078804016113282e-05,
      "step": 92957,
      "training_step_time": 7.666870594024658
    },
    {
      "epoch": 1.8079757690429686e-05,
      "model_forward_time": 1.7718515396118164,
      "step": 92958
    },
    {
      "epoch": 1.8079757690429686e-05,
      "step": 92958,
      "training_step_time": 7.64172101020813
    },
    {
      "epoch": 1.8080711364746093e-05,
      "model_forward_time": 1.7334768772125244,
      "step": 92959
    },
    {
      "epoch": 1.8080711364746093e-05,
      "step": 92959,
      "training_step_time": 7.609626770019531
    },
    {
      "epoch": 1.80816650390625e-05,
      "grad_norm": 0.09692753851413727,
      "learning_rate": 1.3492680314996454e-07,
      "loss": 0.0583,
      "step": 92960
    },
    {
      "epoch": 1.80816650390625e-05,
      "model_forward_time": 1.725149393081665,
      "step": 92960
    },
    {
      "dynamics_loss_avg": 0.058316911943256854,
      "epoch": 1.80816650390625e-05,
      "step": 92960
    },
    {
      "action_loss_avg": 0.001947409972490277,
      "epoch": 1.80816650390625e-05,
      "step": 92960
    },
    {
      "epoch": 1.80816650390625e-05,
      "step": 92960,
      "training_step_time": 8.252627849578857
    },
    {
      "epoch": 1.8082618713378905e-05,
      "model_forward_time": 1.732419729232788,
      "step": 92961
    },
    {
      "epoch": 1.8082618713378905e-05,
      "step": 92961,
      "training_step_time": 10.984154462814331
    },
    {
      "epoch": 1.8083572387695312e-05,
      "model_forward_time": 2.8082215785980225,
      "step": 92962
    },
    {
      "epoch": 1.8083572387695312e-05,
      "step": 92962,
      "training_step_time": 8.334307193756104
    },
    {
      "epoch": 1.808452606201172e-05,
      "model_forward_time": 1.0274021625518799,
      "step": 92963
    },
    {
      "epoch": 1.808452606201172e-05,
      "step": 92963,
      "training_step_time": 7.771491765975952
    },
    {
      "epoch": 1.8085479736328124e-05,
      "model_forward_time": 1.7351460456848145,
      "step": 92964
    },
    {
      "epoch": 1.8085479736328124e-05,
      "step": 92964,
      "training_step_time": 7.682920694351196
    },
    {
      "epoch": 1.808643341064453e-05,
      "model_forward_time": 1.7289164066314697,
      "step": 92965
    },
    {
      "epoch": 1.808643341064453e-05,
      "step": 92965,
      "training_step_time": 7.449903726577759
    },
    {
      "epoch": 1.808738708496094e-05,
      "model_forward_time": 1.7442190647125244,
      "step": 92966
    },
    {
      "epoch": 1.808738708496094e-05,
      "step": 92966,
      "training_step_time": 8.45613145828247
    },
    {
      "epoch": 1.8088340759277343e-05,
      "model_forward_time": 1.7863869667053223,
      "step": 92967
    },
    {
      "epoch": 1.8088340759277343e-05,
      "step": 92967,
      "training_step_time": 7.5165228843688965
    },
    {
      "epoch": 1.808929443359375e-05,
      "model_forward_time": 1.801037311553955,
      "step": 92968
    },
    {
      "epoch": 1.808929443359375e-05,
      "step": 92968,
      "training_step_time": 7.631765604019165
    },
    {
      "epoch": 1.8090248107910157e-05,
      "model_forward_time": 1.7732107639312744,
      "step": 92969
    },
    {
      "epoch": 1.8090248107910157e-05,
      "step": 92969,
      "training_step_time": 7.7361509799957275
    },
    {
      "epoch": 1.809120178222656e-05,
      "grad_norm": 0.0872158408164978,
      "learning_rate": 1.345455419461078e-07,
      "loss": 0.0588,
      "step": 92970
    },
    {
      "epoch": 1.809120178222656e-05,
      "model_forward_time": 1.789153814315796,
      "step": 92970
    },
    {
      "dynamics_loss_avg": 0.05732521247118712,
      "epoch": 1.809120178222656e-05,
      "step": 92970
    },
    {
      "action_loss_avg": 0.0018391768739093096,
      "epoch": 1.809120178222656e-05,
      "step": 92970
    },
    {
      "epoch": 1.809120178222656e-05,
      "step": 92970,
      "training_step_time": 9.431285858154297
    },
    {
      "epoch": 1.809215545654297e-05,
      "model_forward_time": 1.7905640602111816,
      "step": 92971
    },
    {
      "epoch": 1.809215545654297e-05,
      "step": 92971,
      "training_step_time": 8.624415397644043
    },
    {
      "epoch": 1.8093109130859376e-05,
      "model_forward_time": 2.1820287704467773,
      "step": 92972
    },
    {
      "epoch": 1.8093109130859376e-05,
      "step": 92972,
      "training_step_time": 8.281864643096924
    },
    {
      "epoch": 1.809406280517578e-05,
      "model_forward_time": 1.7972362041473389,
      "step": 92973
    },
    {
      "epoch": 1.809406280517578e-05,
      "step": 92973,
      "training_step_time": 7.691800832748413
    },
    {
      "epoch": 1.8095016479492188e-05,
      "model_forward_time": 1.7974953651428223,
      "step": 92974
    },
    {
      "epoch": 1.8095016479492188e-05,
      "step": 92974,
      "training_step_time": 7.451714038848877
    },
    {
      "epoch": 1.8095970153808595e-05,
      "model_forward_time": 1.743227243423462,
      "step": 92975
    },
    {
      "epoch": 1.8095970153808595e-05,
      "step": 92975,
      "training_step_time": 7.725605010986328
    },
    {
      "epoch": 1.8096923828125e-05,
      "model_forward_time": 1.7484092712402344,
      "step": 92976
    },
    {
      "epoch": 1.8096923828125e-05,
      "step": 92976,
      "training_step_time": 8.438089370727539
    },
    {
      "epoch": 1.8097877502441407e-05,
      "model_forward_time": 1.7567980289459229,
      "step": 92977
    },
    {
      "epoch": 1.8097877502441407e-05,
      "step": 92977,
      "training_step_time": 7.626171350479126
    },
    {
      "epoch": 1.8098831176757814e-05,
      "model_forward_time": 1.742002010345459,
      "step": 92978
    },
    {
      "epoch": 1.8098831176757814e-05,
      "step": 92978,
      "training_step_time": 7.71318507194519
    },
    {
      "epoch": 1.8099784851074218e-05,
      "model_forward_time": 1.718372106552124,
      "step": 92979
    },
    {
      "epoch": 1.8099784851074218e-05,
      "step": 92979,
      "training_step_time": 8.719244241714478
    },
    {
      "epoch": 1.8100738525390625e-05,
      "grad_norm": 0.11708538234233856,
      "learning_rate": 1.3416481282103432e-07,
      "loss": 0.0554,
      "step": 92980
    },
    {
      "epoch": 1.8100738525390625e-05,
      "model_forward_time": 1.736924409866333,
      "step": 92980
    },
    {
      "dynamics_loss_avg": 0.041044883988797666,
      "epoch": 1.8100738525390625e-05,
      "step": 92980
    },
    {
      "action_loss_avg": 0.0011851730232592672,
      "epoch": 1.8100738525390625e-05,
      "step": 92980
    },
    {
      "epoch": 1.8100738525390625e-05,
      "step": 92980,
      "training_step_time": 9.003066301345825
    },
    {
      "epoch": 1.8101692199707033e-05,
      "model_forward_time": 2.8246042728424072,
      "step": 92981
    },
    {
      "epoch": 1.8101692199707033e-05,
      "step": 92981,
      "training_step_time": 7.77947735786438
    },
    {
      "epoch": 1.8102645874023437e-05,
      "model_forward_time": 1.7290537357330322,
      "step": 92982
    },
    {
      "epoch": 1.8102645874023437e-05,
      "step": 92982,
      "training_step_time": 8.630659103393555
    },
    {
      "epoch": 1.8103599548339844e-05,
      "model_forward_time": 1.742220163345337,
      "step": 92983
    },
    {
      "epoch": 1.8103599548339844e-05,
      "step": 92983,
      "training_step_time": 7.441944122314453
    },
    {
      "epoch": 1.810455322265625e-05,
      "model_forward_time": 1.7432303428649902,
      "step": 92984
    },
    {
      "epoch": 1.810455322265625e-05,
      "step": 92984,
      "training_step_time": 7.664466381072998
    },
    {
      "epoch": 1.8105506896972656e-05,
      "model_forward_time": 1.7516975402832031,
      "step": 92985
    },
    {
      "epoch": 1.8105506896972656e-05,
      "step": 92985,
      "training_step_time": 8.4444420337677
    },
    {
      "epoch": 1.8106460571289063e-05,
      "model_forward_time": 1.7599871158599854,
      "step": 92986
    },
    {
      "epoch": 1.8106460571289063e-05,
      "step": 92986,
      "training_step_time": 8.343312501907349
    },
    {
      "epoch": 1.8107414245605467e-05,
      "model_forward_time": 1.7325263023376465,
      "step": 92987
    },
    {
      "epoch": 1.8107414245605467e-05,
      "step": 92987,
      "training_step_time": 7.829041242599487
    },
    {
      "epoch": 1.8108367919921875e-05,
      "model_forward_time": 1.7341251373291016,
      "step": 92988
    },
    {
      "epoch": 1.8108367919921875e-05,
      "step": 92988,
      "training_step_time": 8.764250993728638
    },
    {
      "epoch": 1.8109321594238282e-05,
      "model_forward_time": 1.365858793258667,
      "step": 92989
    },
    {
      "epoch": 1.8109321594238282e-05,
      "step": 92989,
      "training_step_time": 7.671207427978516
    },
    {
      "epoch": 1.8110275268554686e-05,
      "grad_norm": 0.07448777556419373,
      "learning_rate": 1.337846158163819e-07,
      "loss": 0.0579,
      "step": 92990
    },
    {
      "epoch": 1.8110275268554686e-05,
      "model_forward_time": 2.778688430786133,
      "step": 92990
    },
    {
      "dynamics_loss_avg": 0.05449652317911387,
      "epoch": 1.8110275268554686e-05,
      "step": 92990
    },
    {
      "action_loss_avg": 0.001828624471090734,
      "epoch": 1.8110275268554686e-05,
      "step": 92990
    },
    {
      "epoch": 1.8110275268554686e-05,
      "step": 92990,
      "training_step_time": 9.33437204360962
    },
    {
      "epoch": 1.8111228942871093e-05,
      "model_forward_time": 1.7309906482696533,
      "step": 92991
    },
    {
      "epoch": 1.8111228942871093e-05,
      "step": 92991,
      "training_step_time": 8.541998147964478
    },
    {
      "epoch": 1.81121826171875e-05,
      "model_forward_time": 1.7344231605529785,
      "step": 92992
    },
    {
      "epoch": 1.81121826171875e-05,
      "step": 92992,
      "training_step_time": 7.4543163776397705
    },
    {
      "epoch": 1.8113136291503905e-05,
      "model_forward_time": 1.737396478652954,
      "step": 92993
    },
    {
      "epoch": 1.8113136291503905e-05,
      "step": 92993,
      "training_step_time": 8.96666693687439
    },
    {
      "epoch": 1.8114089965820312e-05,
      "model_forward_time": 1.7410123348236084,
      "step": 92994
    },
    {
      "epoch": 1.8114089965820312e-05,
      "step": 92994,
      "training_step_time": 8.535367727279663
    },
    {
      "epoch": 1.811504364013672e-05,
      "model_forward_time": 1.7628834247589111,
      "step": 92995
    },
    {
      "epoch": 1.811504364013672e-05,
      "step": 92995,
      "training_step_time": 7.607706308364868
    },
    {
      "epoch": 1.8115997314453124e-05,
      "model_forward_time": 1.725820541381836,
      "step": 92996
    },
    {
      "epoch": 1.8115997314453124e-05,
      "step": 92996,
      "training_step_time": 9.41945219039917
    },
    {
      "epoch": 1.811695098876953e-05,
      "model_forward_time": 1.7362279891967773,
      "step": 92997
    },
    {
      "epoch": 1.811695098876953e-05,
      "step": 92997,
      "training_step_time": 8.291370868682861
    },
    {
      "epoch": 1.811790466308594e-05,
      "model_forward_time": 1.7738747596740723,
      "step": 92998
    },
    {
      "epoch": 1.811790466308594e-05,
      "step": 92998,
      "training_step_time": 8.556727409362793
    },
    {
      "epoch": 1.8118858337402343e-05,
      "model_forward_time": 1.732314109802246,
      "step": 92999
    },
    {
      "epoch": 1.8118858337402343e-05,
      "step": 92999,
      "training_step_time": 9.219110012054443
    },
    {
      "epoch": 1.811981201171875e-05,
      "grad_norm": 0.09685513377189636,
      "learning_rate": 1.334049509737273e-07,
      "loss": 0.0559,
      "step": 93000
    },
    {
      "epoch": 1.811981201171875e-05,
      "model_forward_time": 1.8556392192840576,
      "step": 93000
    },
    {
      "dynamics_loss_avg": 0.06019392050802708,
      "epoch": 1.811981201171875e-05,
      "step": 93000
    },
    {
      "action_loss_avg": 0.0016959736763965338,
      "epoch": 1.811981201171875e-05,
      "step": 93000
    },
    {
      "epoch": 1.811981201171875e-05,
      "step": 93000,
      "training_step_time": 7.754641056060791
    },
    {
      "epoch": 1.8120765686035157e-05,
      "model_forward_time": 1.7317447662353516,
      "step": 93001
    },
    {
      "epoch": 1.8120765686035157e-05,
      "step": 93001,
      "training_step_time": 7.462881565093994
    },
    {
      "epoch": 1.812171936035156e-05,
      "model_forward_time": 1.740556240081787,
      "step": 93002
    },
    {
      "epoch": 1.812171936035156e-05,
      "step": 93002,
      "training_step_time": 9.029164552688599
    },
    {
      "epoch": 1.812267303466797e-05,
      "model_forward_time": 1.7325921058654785,
      "step": 93003
    },
    {
      "epoch": 1.812267303466797e-05,
      "step": 93003,
      "training_step_time": 8.646175146102905
    },
    {
      "epoch": 1.8123626708984376e-05,
      "model_forward_time": 1.7411575317382812,
      "step": 93004
    },
    {
      "epoch": 1.8123626708984376e-05,
      "step": 93004,
      "training_step_time": 8.70374870300293
    },
    {
      "epoch": 1.812458038330078e-05,
      "model_forward_time": 1.724597692489624,
      "step": 93005
    },
    {
      "epoch": 1.812458038330078e-05,
      "step": 93005,
      "training_step_time": 9.399543046951294
    },
    {
      "epoch": 1.8125534057617188e-05,
      "model_forward_time": 1.7333791255950928,
      "step": 93006
    },
    {
      "epoch": 1.8125534057617188e-05,
      "step": 93006,
      "training_step_time": 10.123481273651123
    },
    {
      "epoch": 1.8126487731933595e-05,
      "model_forward_time": 1.7335059642791748,
      "step": 93007
    },
    {
      "epoch": 1.8126487731933595e-05,
      "step": 93007,
      "training_step_time": 10.470152616500854
    },
    {
      "epoch": 1.812744140625e-05,
      "model_forward_time": 3.429711103439331,
      "step": 93008
    },
    {
      "epoch": 1.812744140625e-05,
      "step": 93008,
      "training_step_time": 8.990609884262085
    },
    {
      "epoch": 1.8128395080566407e-05,
      "model_forward_time": 1.7510998249053955,
      "step": 93009
    },
    {
      "epoch": 1.8128395080566407e-05,
      "step": 93009,
      "training_step_time": 8.743560552597046
    },
    {
      "epoch": 1.8129348754882814e-05,
      "grad_norm": 0.08052034676074982,
      "learning_rate": 1.3302581833459116e-07,
      "loss": 0.0572,
      "step": 93010
    },
    {
      "epoch": 1.8129348754882814e-05,
      "model_forward_time": 1.7755126953125,
      "step": 93010
    },
    {
      "dynamics_loss_avg": 0.050799228623509406,
      "epoch": 1.8129348754882814e-05,
      "step": 93010
    },
    {
      "action_loss_avg": 0.001097474410198629,
      "epoch": 1.8129348754882814e-05,
      "step": 93010
    },
    {
      "epoch": 1.8129348754882814e-05,
      "step": 93010,
      "training_step_time": 7.7099432945251465
    },
    {
      "epoch": 1.8130302429199218e-05,
      "model_forward_time": 1.7221734523773193,
      "step": 93011
    },
    {
      "epoch": 1.8130302429199218e-05,
      "step": 93011,
      "training_step_time": 8.485039949417114
    },
    {
      "epoch": 1.8131256103515625e-05,
      "model_forward_time": 1.73411226272583,
      "step": 93012
    },
    {
      "epoch": 1.8131256103515625e-05,
      "step": 93012,
      "training_step_time": 7.739570379257202
    },
    {
      "epoch": 1.8132209777832033e-05,
      "model_forward_time": 1.7484362125396729,
      "step": 93013
    },
    {
      "epoch": 1.8132209777832033e-05,
      "step": 93013,
      "training_step_time": 7.6844470500946045
    },
    {
      "epoch": 1.8133163452148437e-05,
      "model_forward_time": 1.7366764545440674,
      "step": 93014
    },
    {
      "epoch": 1.8133163452148437e-05,
      "step": 93014,
      "training_step_time": 7.720360517501831
    },
    {
      "epoch": 1.8134117126464844e-05,
      "model_forward_time": 1.7264299392700195,
      "step": 93015
    },
    {
      "epoch": 1.8134117126464844e-05,
      "step": 93015,
      "training_step_time": 7.70263409614563
    },
    {
      "epoch": 1.813507080078125e-05,
      "model_forward_time": 1.748924970626831,
      "step": 93016
    },
    {
      "epoch": 1.813507080078125e-05,
      "step": 93016,
      "training_step_time": 8.524702548980713
    },
    {
      "epoch": 1.8136024475097656e-05,
      "model_forward_time": 2.8247156143188477,
      "step": 93017
    },
    {
      "epoch": 1.8136024475097656e-05,
      "step": 93017,
      "training_step_time": 7.7816784381866455
    },
    {
      "epoch": 1.8136978149414063e-05,
      "model_forward_time": 1.7500944137573242,
      "step": 93018
    },
    {
      "epoch": 1.8136978149414063e-05,
      "step": 93018,
      "training_step_time": 8.238534688949585
    },
    {
      "epoch": 1.8137931823730467e-05,
      "model_forward_time": 1.7392964363098145,
      "step": 93019
    },
    {
      "epoch": 1.8137931823730467e-05,
      "step": 93019,
      "training_step_time": 7.707384347915649
    },
    {
      "epoch": 1.8138885498046875e-05,
      "grad_norm": 0.15826553106307983,
      "learning_rate": 1.3264721794043366e-07,
      "loss": 0.0553,
      "step": 93020
    },
    {
      "epoch": 1.8138885498046875e-05,
      "model_forward_time": 1.7336866855621338,
      "step": 93020
    },
    {
      "dynamics_loss_avg": 0.04856403321027756,
      "epoch": 1.8138885498046875e-05,
      "step": 93020
    },
    {
      "action_loss_avg": 0.0015355677489424125,
      "epoch": 1.8138885498046875e-05,
      "step": 93020
    },
    {
      "epoch": 1.8138885498046875e-05,
      "step": 93020,
      "training_step_time": 7.65516996383667
    },
    {
      "epoch": 1.8139839172363282e-05,
      "model_forward_time": 1.7351584434509277,
      "step": 93021
    },
    {
      "epoch": 1.8139839172363282e-05,
      "step": 93021,
      "training_step_time": 7.728595733642578
    },
    {
      "epoch": 1.8140792846679686e-05,
      "model_forward_time": 1.735137701034546,
      "step": 93022
    },
    {
      "epoch": 1.8140792846679686e-05,
      "step": 93022,
      "training_step_time": 8.238130569458008
    },
    {
      "epoch": 1.8141746520996093e-05,
      "model_forward_time": 1.7335124015808105,
      "step": 93023
    },
    {
      "epoch": 1.8141746520996093e-05,
      "step": 93023,
      "training_step_time": 7.7456886768341064
    },
    {
      "epoch": 1.81427001953125e-05,
      "model_forward_time": 1.7367568016052246,
      "step": 93024
    },
    {
      "epoch": 1.81427001953125e-05,
      "step": 93024,
      "training_step_time": 7.735895156860352
    },
    {
      "epoch": 1.8143653869628905e-05,
      "model_forward_time": 1.7389216423034668,
      "step": 93025
    },
    {
      "epoch": 1.8143653869628905e-05,
      "step": 93025,
      "training_step_time": 7.698900938034058
    },
    {
      "epoch": 1.8144607543945312e-05,
      "model_forward_time": 1.7349324226379395,
      "step": 93026
    },
    {
      "epoch": 1.8144607543945312e-05,
      "step": 93026,
      "training_step_time": 9.105249404907227
    },
    {
      "epoch": 1.814556121826172e-05,
      "model_forward_time": 1.7367408275604248,
      "step": 93027
    },
    {
      "epoch": 1.814556121826172e-05,
      "step": 93027,
      "training_step_time": 7.687504768371582
    },
    {
      "epoch": 1.8146514892578124e-05,
      "model_forward_time": 1.7413520812988281,
      "step": 93028
    },
    {
      "epoch": 1.8146514892578124e-05,
      "step": 93028,
      "training_step_time": 7.7161548137664795
    },
    {
      "epoch": 1.814746856689453e-05,
      "model_forward_time": 1.7426235675811768,
      "step": 93029
    },
    {
      "epoch": 1.814746856689453e-05,
      "step": 93029,
      "training_step_time": 7.791012287139893
    },
    {
      "epoch": 1.814842224121094e-05,
      "grad_norm": 0.09517472237348557,
      "learning_rate": 1.3226914983265836e-07,
      "loss": 0.0564,
      "step": 93030
    },
    {
      "epoch": 1.814842224121094e-05,
      "model_forward_time": 1.733708381652832,
      "step": 93030
    },
    {
      "dynamics_loss_avg": 0.062130728736519814,
      "epoch": 1.814842224121094e-05,
      "step": 93030
    },
    {
      "action_loss_avg": 0.0012644993184949271,
      "epoch": 1.814842224121094e-05,
      "step": 93030
    },
    {
      "epoch": 1.814842224121094e-05,
      "step": 93030,
      "training_step_time": 8.40700888633728
    },
    {
      "epoch": 1.8149375915527343e-05,
      "model_forward_time": 1.748018741607666,
      "step": 93031
    },
    {
      "epoch": 1.8149375915527343e-05,
      "step": 93031,
      "training_step_time": 7.671737432479858
    },
    {
      "epoch": 1.815032958984375e-05,
      "model_forward_time": 1.7721569538116455,
      "step": 93032
    },
    {
      "epoch": 1.815032958984375e-05,
      "step": 93032,
      "training_step_time": 7.84267520904541
    },
    {
      "epoch": 1.8151283264160157e-05,
      "model_forward_time": 1.7351129055023193,
      "step": 93033
    },
    {
      "epoch": 1.8151283264160157e-05,
      "step": 93033,
      "training_step_time": 9.61393141746521
    },
    {
      "epoch": 1.815223693847656e-05,
      "model_forward_time": 1.7340068817138672,
      "step": 93034
    },
    {
      "epoch": 1.815223693847656e-05,
      "step": 93034,
      "training_step_time": 8.59612774848938
    },
    {
      "epoch": 1.815319061279297e-05,
      "model_forward_time": 1.733921766281128,
      "step": 93035
    },
    {
      "epoch": 1.815319061279297e-05,
      "step": 93035,
      "training_step_time": 7.670488119125366
    },
    {
      "epoch": 1.8154144287109376e-05,
      "model_forward_time": 2.8969743251800537,
      "step": 93036
    },
    {
      "epoch": 1.8154144287109376e-05,
      "step": 93036,
      "training_step_time": 8.259483098983765
    },
    {
      "epoch": 1.815509796142578e-05,
      "model_forward_time": 1.0073814392089844,
      "step": 93037
    },
    {
      "epoch": 1.815509796142578e-05,
      "step": 93037,
      "training_step_time": 8.544854402542114
    },
    {
      "epoch": 1.8156051635742188e-05,
      "model_forward_time": 1.7298600673675537,
      "step": 93038
    },
    {
      "epoch": 1.8156051635742188e-05,
      "step": 93038,
      "training_step_time": 7.669386863708496
    },
    {
      "epoch": 1.8157005310058595e-05,
      "model_forward_time": 1.746168851852417,
      "step": 93039
    },
    {
      "epoch": 1.8157005310058595e-05,
      "step": 93039,
      "training_step_time": 7.808888673782349
    },
    {
      "epoch": 1.8157958984375e-05,
      "grad_norm": 0.11923671513795853,
      "learning_rate": 1.3189161405260996e-07,
      "loss": 0.0557,
      "step": 93040
    },
    {
      "epoch": 1.8157958984375e-05,
      "model_forward_time": 1.011686086654663,
      "step": 93040
    },
    {
      "dynamics_loss_avg": 0.06277090311050415,
      "epoch": 1.8157958984375e-05,
      "step": 93040
    },
    {
      "action_loss_avg": 0.0015741940413136035,
      "epoch": 1.8157958984375e-05,
      "step": 93040
    },
    {
      "epoch": 1.8157958984375e-05,
      "step": 93040,
      "training_step_time": 7.708595514297485
    },
    {
      "epoch": 1.8158912658691407e-05,
      "model_forward_time": 1.7564327716827393,
      "step": 93041
    },
    {
      "epoch": 1.8158912658691407e-05,
      "step": 93041,
      "training_step_time": 9.45967698097229
    },
    {
      "epoch": 1.8159866333007814e-05,
      "model_forward_time": 1.7405259609222412,
      "step": 93042
    },
    {
      "epoch": 1.8159866333007814e-05,
      "step": 93042,
      "training_step_time": 8.440062046051025
    },
    {
      "epoch": 1.8160820007324218e-05,
      "model_forward_time": 1.734938383102417,
      "step": 93043
    },
    {
      "epoch": 1.8160820007324218e-05,
      "step": 93043,
      "training_step_time": 9.015011548995972
    },
    {
      "epoch": 1.8161773681640625e-05,
      "model_forward_time": 1.7435932159423828,
      "step": 93044
    },
    {
      "epoch": 1.8161773681640625e-05,
      "step": 93044,
      "training_step_time": 8.718567848205566
    },
    {
      "epoch": 1.8162727355957033e-05,
      "model_forward_time": 2.777082681655884,
      "step": 93045
    },
    {
      "epoch": 1.8162727355957033e-05,
      "step": 93045,
      "training_step_time": 7.677518129348755
    },
    {
      "epoch": 1.8163681030273437e-05,
      "model_forward_time": 1.7553188800811768,
      "step": 93046
    },
    {
      "epoch": 1.8163681030273437e-05,
      "step": 93046,
      "training_step_time": 7.7192888259887695
    },
    {
      "epoch": 1.8164634704589844e-05,
      "model_forward_time": 1.736844539642334,
      "step": 93047
    },
    {
      "epoch": 1.8164634704589844e-05,
      "step": 93047,
      "training_step_time": 8.589113712310791
    },
    {
      "epoch": 1.816558837890625e-05,
      "model_forward_time": 1.728731393814087,
      "step": 93048
    },
    {
      "epoch": 1.816558837890625e-05,
      "step": 93048,
      "training_step_time": 7.742061138153076
    },
    {
      "epoch": 1.8166542053222656e-05,
      "model_forward_time": 1.7317368984222412,
      "step": 93049
    },
    {
      "epoch": 1.8166542053222656e-05,
      "step": 93049,
      "training_step_time": 7.718805313110352
    },
    {
      "epoch": 1.8167495727539063e-05,
      "grad_norm": 0.10252118110656738,
      "learning_rate": 1.3151461064157657e-07,
      "loss": 0.056,
      "step": 93050
    },
    {
      "epoch": 1.8167495727539063e-05,
      "model_forward_time": 1.739917278289795,
      "step": 93050
    },
    {
      "dynamics_loss_avg": 0.05933162085711956,
      "epoch": 1.8167495727539063e-05,
      "step": 93050
    },
    {
      "action_loss_avg": 0.0012427444744389504,
      "epoch": 1.8167495727539063e-05,
      "step": 93050
    },
    {
      "epoch": 1.8167495727539063e-05,
      "step": 93050,
      "training_step_time": 7.533407211303711
    },
    {
      "epoch": 1.8168449401855467e-05,
      "model_forward_time": 1.7276248931884766,
      "step": 93051
    },
    {
      "epoch": 1.8168449401855467e-05,
      "step": 93051,
      "training_step_time": 8.551600456237793
    },
    {
      "epoch": 1.8169403076171875e-05,
      "model_forward_time": 1.365940809249878,
      "step": 93052
    },
    {
      "epoch": 1.8169403076171875e-05,
      "step": 93052,
      "training_step_time": 8.208799600601196
    },
    {
      "epoch": 1.8170356750488282e-05,
      "model_forward_time": 1.7446980476379395,
      "step": 93053
    },
    {
      "epoch": 1.8170356750488282e-05,
      "step": 93053,
      "training_step_time": 9.088006734848022
    },
    {
      "epoch": 1.8171310424804686e-05,
      "model_forward_time": 1.733154296875,
      "step": 93054
    },
    {
      "epoch": 1.8171310424804686e-05,
      "step": 93054,
      "training_step_time": 8.4535493850708
    },
    {
      "epoch": 1.8172264099121093e-05,
      "model_forward_time": 2.8404386043548584,
      "step": 93055
    },
    {
      "epoch": 1.8172264099121093e-05,
      "step": 93055,
      "training_step_time": 8.56718921661377
    },
    {
      "epoch": 1.81732177734375e-05,
      "model_forward_time": 1.7353687286376953,
      "step": 93056
    },
    {
      "epoch": 1.81732177734375e-05,
      "step": 93056,
      "training_step_time": 8.501107454299927
    },
    {
      "epoch": 1.8174171447753905e-05,
      "model_forward_time": 1.7449805736541748,
      "step": 93057
    },
    {
      "epoch": 1.8174171447753905e-05,
      "step": 93057,
      "training_step_time": 8.301745653152466
    },
    {
      "epoch": 1.8175125122070312e-05,
      "model_forward_time": 1.7317640781402588,
      "step": 93058
    },
    {
      "epoch": 1.8175125122070312e-05,
      "step": 93058,
      "training_step_time": 7.797247886657715
    },
    {
      "epoch": 1.817607879638672e-05,
      "model_forward_time": 1.7361161708831787,
      "step": 93059
    },
    {
      "epoch": 1.817607879638672e-05,
      "step": 93059,
      "training_step_time": 8.604363679885864
    },
    {
      "epoch": 1.8177032470703124e-05,
      "grad_norm": 0.07699280232191086,
      "learning_rate": 1.311381396407846e-07,
      "loss": 0.0565,
      "step": 93060
    },
    {
      "epoch": 1.8177032470703124e-05,
      "model_forward_time": 1.7416143417358398,
      "step": 93060
    },
    {
      "dynamics_loss_avg": 0.06161455605179071,
      "epoch": 1.8177032470703124e-05,
      "step": 93060
    },
    {
      "action_loss_avg": 0.0013302061386639252,
      "epoch": 1.8177032470703124e-05,
      "step": 93060
    },
    {
      "epoch": 1.8177032470703124e-05,
      "step": 93060,
      "training_step_time": 8.677544116973877
    },
    {
      "epoch": 1.817798614501953e-05,
      "model_forward_time": 1.7378654479980469,
      "step": 93061
    },
    {
      "epoch": 1.817798614501953e-05,
      "step": 93061,
      "training_step_time": 7.775806427001953
    },
    {
      "epoch": 1.817893981933594e-05,
      "model_forward_time": 1.7334470748901367,
      "step": 93062
    },
    {
      "epoch": 1.817893981933594e-05,
      "step": 93062,
      "training_step_time": 8.634390592575073
    },
    {
      "epoch": 1.8179893493652343e-05,
      "model_forward_time": 1.7401771545410156,
      "step": 93063
    },
    {
      "epoch": 1.8179893493652343e-05,
      "step": 93063,
      "training_step_time": 8.443689346313477
    },
    {
      "epoch": 1.818084716796875e-05,
      "model_forward_time": 2.8223626613616943,
      "step": 93064
    },
    {
      "epoch": 1.818084716796875e-05,
      "step": 93064,
      "training_step_time": 7.777679920196533
    },
    {
      "epoch": 1.8181800842285157e-05,
      "model_forward_time": 1.7424569129943848,
      "step": 93065
    },
    {
      "epoch": 1.8181800842285157e-05,
      "step": 93065,
      "training_step_time": 7.660728216171265
    },
    {
      "epoch": 1.818275451660156e-05,
      "model_forward_time": 1.7374849319458008,
      "step": 93066
    },
    {
      "epoch": 1.818275451660156e-05,
      "step": 93066,
      "training_step_time": 7.668783664703369
    },
    {
      "epoch": 1.818370819091797e-05,
      "model_forward_time": 1.7335054874420166,
      "step": 93067
    },
    {
      "epoch": 1.818370819091797e-05,
      "step": 93067,
      "training_step_time": 7.784689664840698
    },
    {
      "epoch": 1.8184661865234376e-05,
      "model_forward_time": 1.7238872051239014,
      "step": 93068
    },
    {
      "epoch": 1.8184661865234376e-05,
      "step": 93068,
      "training_step_time": 8.469878673553467
    },
    {
      "epoch": 1.818561553955078e-05,
      "model_forward_time": 1.769714117050171,
      "step": 93069
    },
    {
      "epoch": 1.818561553955078e-05,
      "step": 93069,
      "training_step_time": 8.258285999298096
    },
    {
      "epoch": 1.8186569213867188e-05,
      "grad_norm": 0.07966678589582443,
      "learning_rate": 1.3076220109140614e-07,
      "loss": 0.0566,
      "step": 93070
    },
    {
      "epoch": 1.8186569213867188e-05,
      "model_forward_time": 1.7391612529754639,
      "step": 93070
    },
    {
      "dynamics_loss_avg": 0.07253469098359347,
      "epoch": 1.8186569213867188e-05,
      "step": 93070
    },
    {
      "action_loss_avg": 0.0015441555515280925,
      "epoch": 1.8186569213867188e-05,
      "step": 93070
    },
    {
      "epoch": 1.8186569213867188e-05,
      "step": 93070,
      "training_step_time": 7.823430299758911
    },
    {
      "epoch": 1.8187522888183595e-05,
      "model_forward_time": 1.74013352394104,
      "step": 93071
    },
    {
      "epoch": 1.8187522888183595e-05,
      "step": 93071,
      "training_step_time": 7.703215837478638
    },
    {
      "epoch": 1.81884765625e-05,
      "model_forward_time": 1.7430453300476074,
      "step": 93072
    },
    {
      "epoch": 1.81884765625e-05,
      "step": 93072,
      "training_step_time": 9.306130647659302
    },
    {
      "epoch": 1.8189430236816407e-05,
      "model_forward_time": 2.879826784133911,
      "step": 93073
    },
    {
      "epoch": 1.8189430236816407e-05,
      "step": 93073,
      "training_step_time": 7.731563568115234
    },
    {
      "epoch": 1.8190383911132814e-05,
      "model_forward_time": 1.746870756149292,
      "step": 93074
    },
    {
      "epoch": 1.8190383911132814e-05,
      "step": 93074,
      "training_step_time": 8.685286521911621
    },
    {
      "epoch": 1.8191337585449218e-05,
      "model_forward_time": 1.3583974838256836,
      "step": 93075
    },
    {
      "epoch": 1.8191337585449218e-05,
      "step": 93075,
      "training_step_time": 7.429847955703735
    },
    {
      "epoch": 1.8192291259765625e-05,
      "model_forward_time": 1.7353599071502686,
      "step": 93076
    },
    {
      "epoch": 1.8192291259765625e-05,
      "step": 93076,
      "training_step_time": 7.745819807052612
    },
    {
      "epoch": 1.8193244934082033e-05,
      "model_forward_time": 1.7484357357025146,
      "step": 93077
    },
    {
      "epoch": 1.8193244934082033e-05,
      "step": 93077,
      "training_step_time": 9.421364545822144
    },
    {
      "epoch": 1.8194198608398437e-05,
      "model_forward_time": 1.760023593902588,
      "step": 93078
    },
    {
      "epoch": 1.8194198608398437e-05,
      "step": 93078,
      "training_step_time": 7.616002321243286
    },
    {
      "epoch": 1.8195152282714844e-05,
      "model_forward_time": 1.7363512516021729,
      "step": 93079
    },
    {
      "epoch": 1.8195152282714844e-05,
      "step": 93079,
      "training_step_time": 7.6672728061676025
    },
    {
      "epoch": 1.819610595703125e-05,
      "grad_norm": 0.0833301767706871,
      "learning_rate": 1.3038679503455276e-07,
      "loss": 0.0578,
      "step": 93080
    },
    {
      "epoch": 1.819610595703125e-05,
      "model_forward_time": 1.733729600906372,
      "step": 93080
    },
    {
      "dynamics_loss_avg": 0.05391584262251854,
      "epoch": 1.819610595703125e-05,
      "step": 93080
    },
    {
      "action_loss_avg": 0.001405406472622417,
      "epoch": 1.819610595703125e-05,
      "step": 93080
    },
    {
      "epoch": 1.819610595703125e-05,
      "step": 93080,
      "training_step_time": 7.650004863739014
    },
    {
      "epoch": 1.8197059631347656e-05,
      "model_forward_time": 1.725576400756836,
      "step": 93081
    },
    {
      "epoch": 1.8197059631347656e-05,
      "step": 93081,
      "training_step_time": 8.509482145309448
    },
    {
      "epoch": 1.8198013305664063e-05,
      "model_forward_time": 1.7338440418243408,
      "step": 93082
    },
    {
      "epoch": 1.8198013305664063e-05,
      "step": 93082,
      "training_step_time": 8.859411001205444
    },
    {
      "epoch": 1.8198966979980467e-05,
      "model_forward_time": 1.7553446292877197,
      "step": 93083
    },
    {
      "epoch": 1.8198966979980467e-05,
      "step": 93083,
      "training_step_time": 7.63472056388855
    },
    {
      "epoch": 1.8199920654296875e-05,
      "model_forward_time": 1.7414534091949463,
      "step": 93084
    },
    {
      "epoch": 1.8199920654296875e-05,
      "step": 93084,
      "training_step_time": 7.648238182067871
    },
    {
      "epoch": 1.8200874328613282e-05,
      "model_forward_time": 1.7499656677246094,
      "step": 93085
    },
    {
      "epoch": 1.8200874328613282e-05,
      "step": 93085,
      "training_step_time": 7.638241767883301
    },
    {
      "epoch": 1.8201828002929686e-05,
      "model_forward_time": 1.902221918106079,
      "step": 93086
    },
    {
      "epoch": 1.8201828002929686e-05,
      "step": 93086,
      "training_step_time": 7.708609342575073
    },
    {
      "epoch": 1.8202781677246094e-05,
      "model_forward_time": 1.7770249843597412,
      "step": 93087
    },
    {
      "epoch": 1.8202781677246094e-05,
      "step": 93087,
      "training_step_time": 7.598556041717529
    },
    {
      "epoch": 1.82037353515625e-05,
      "model_forward_time": 1.7742621898651123,
      "step": 93088
    },
    {
      "epoch": 1.82037353515625e-05,
      "step": 93088,
      "training_step_time": 8.58070969581604
    },
    {
      "epoch": 1.8204689025878905e-05,
      "model_forward_time": 1.792510747909546,
      "step": 93089
    },
    {
      "epoch": 1.8204689025878905e-05,
      "step": 93089,
      "training_step_time": 9.084447860717773
    },
    {
      "epoch": 1.8205642700195312e-05,
      "grad_norm": 0.07958893477916718,
      "learning_rate": 1.3001192151127773e-07,
      "loss": 0.0571,
      "step": 93090
    },
    {
      "epoch": 1.8205642700195312e-05,
      "model_forward_time": 1.4031586647033691,
      "step": 93090
    },
    {
      "dynamics_loss_avg": 0.0664544802159071,
      "epoch": 1.8205642700195312e-05,
      "step": 93090
    },
    {
      "action_loss_avg": 0.0022875200113048775,
      "epoch": 1.8205642700195312e-05,
      "step": 93090
    },
    {
      "epoch": 1.8205642700195312e-05,
      "step": 93090,
      "training_step_time": 7.643773794174194
    },
    {
      "epoch": 1.820659637451172e-05,
      "model_forward_time": 1.772733449935913,
      "step": 93091
    },
    {
      "epoch": 1.820659637451172e-05,
      "step": 93091,
      "training_step_time": 9.35273790359497
    },
    {
      "epoch": 1.8207550048828124e-05,
      "model_forward_time": 2.4245681762695312,
      "step": 93092
    },
    {
      "epoch": 1.8207550048828124e-05,
      "step": 93092,
      "training_step_time": 8.737358570098877
    },
    {
      "epoch": 1.820850372314453e-05,
      "model_forward_time": 1.7730252742767334,
      "step": 93093
    },
    {
      "epoch": 1.820850372314453e-05,
      "step": 93093,
      "training_step_time": 10.1940016746521
    },
    {
      "epoch": 1.820945739746094e-05,
      "model_forward_time": 1.7909579277038574,
      "step": 93094
    },
    {
      "epoch": 1.820945739746094e-05,
      "step": 93094,
      "training_step_time": 8.334630489349365
    },
    {
      "epoch": 1.8210411071777343e-05,
      "model_forward_time": 1.8720595836639404,
      "step": 93095
    },
    {
      "epoch": 1.8210411071777343e-05,
      "step": 93095,
      "training_step_time": 8.701151371002197
    },
    {
      "epoch": 1.821136474609375e-05,
      "model_forward_time": 1.775355577468872,
      "step": 93096
    },
    {
      "epoch": 1.821136474609375e-05,
      "step": 93096,
      "training_step_time": 7.683627367019653
    },
    {
      "epoch": 1.8212318420410157e-05,
      "model_forward_time": 1.7744925022125244,
      "step": 93097
    },
    {
      "epoch": 1.8212318420410157e-05,
      "step": 93097,
      "training_step_time": 8.69622540473938
    },
    {
      "epoch": 1.821327209472656e-05,
      "model_forward_time": 1.7807395458221436,
      "step": 93098
    },
    {
      "epoch": 1.821327209472656e-05,
      "step": 93098,
      "training_step_time": 8.789377212524414
    },
    {
      "epoch": 1.821422576904297e-05,
      "model_forward_time": 1.7734601497650146,
      "step": 93099
    },
    {
      "epoch": 1.821422576904297e-05,
      "step": 93099,
      "training_step_time": 8.526057243347168
    },
    {
      "epoch": 1.8215179443359376e-05,
      "grad_norm": 0.07144366204738617,
      "learning_rate": 1.2963758056257657e-07,
      "loss": 0.0577,
      "step": 93100
    },
    {
      "epoch": 1.8215179443359376e-05,
      "model_forward_time": 1.7690563201904297,
      "step": 93100
    },
    {
      "dynamics_loss_avg": 0.04538347534835339,
      "epoch": 1.8215179443359376e-05,
      "step": 93100
    },
    {
      "action_loss_avg": 0.0011251891031861306,
      "epoch": 1.8215179443359376e-05,
      "step": 93100
    },
    {
      "epoch": 1.8215179443359376e-05,
      "step": 93100,
      "training_step_time": 8.490761041641235
    },
    {
      "epoch": 1.821613311767578e-05,
      "model_forward_time": 2.4063875675201416,
      "step": 93101
    },
    {
      "epoch": 1.821613311767578e-05,
      "step": 93101,
      "training_step_time": 8.219554662704468
    },
    {
      "epoch": 1.8217086791992188e-05,
      "model_forward_time": 1.7731964588165283,
      "step": 93102
    },
    {
      "epoch": 1.8217086791992188e-05,
      "step": 93102,
      "training_step_time": 7.698222637176514
    },
    {
      "epoch": 1.8218040466308595e-05,
      "model_forward_time": 1.7742230892181396,
      "step": 93103
    },
    {
      "epoch": 1.8218040466308595e-05,
      "step": 93103,
      "training_step_time": 7.480016708374023
    },
    {
      "epoch": 1.8218994140625e-05,
      "model_forward_time": 1.8497023582458496,
      "step": 93104
    },
    {
      "epoch": 1.8218994140625e-05,
      "step": 93104,
      "training_step_time": 7.709597587585449
    },
    {
      "epoch": 1.8219947814941407e-05,
      "model_forward_time": 1.7759172916412354,
      "step": 93105
    },
    {
      "epoch": 1.8219947814941407e-05,
      "step": 93105,
      "training_step_time": 8.703843116760254
    },
    {
      "epoch": 1.8220901489257814e-05,
      "model_forward_time": 1.789397954940796,
      "step": 93106
    },
    {
      "epoch": 1.8220901489257814e-05,
      "step": 93106,
      "training_step_time": 7.6995460987091064
    },
    {
      "epoch": 1.8221855163574218e-05,
      "model_forward_time": 1.7768197059631348,
      "step": 93107
    },
    {
      "epoch": 1.8221855163574218e-05,
      "step": 93107,
      "training_step_time": 9.107407093048096
    },
    {
      "epoch": 1.8222808837890626e-05,
      "model_forward_time": 1.7773075103759766,
      "step": 93108
    },
    {
      "epoch": 1.8222808837890626e-05,
      "step": 93108,
      "training_step_time": 7.716363906860352
    },
    {
      "epoch": 1.8223762512207033e-05,
      "model_forward_time": 1.786039113998413,
      "step": 93109
    },
    {
      "epoch": 1.8223762512207033e-05,
      "step": 93109,
      "training_step_time": 8.50196385383606
    },
    {
      "epoch": 1.8224716186523437e-05,
      "grad_norm": 0.11166082322597504,
      "learning_rate": 1.292637722293877e-07,
      "loss": 0.0577,
      "step": 93110
    },
    {
      "epoch": 1.8224716186523437e-05,
      "model_forward_time": 2.4379334449768066,
      "step": 93110
    },
    {
      "dynamics_loss_avg": 0.03001046455465257,
      "epoch": 1.8224716186523437e-05,
      "step": 93110
    },
    {
      "action_loss_avg": 0.0008217962240451016,
      "epoch": 1.8224716186523437e-05,
      "step": 93110
    },
    {
      "epoch": 1.8224716186523437e-05,
      "step": 93110,
      "training_step_time": 8.727831602096558
    },
    {
      "epoch": 1.8225669860839844e-05,
      "model_forward_time": 1.7881603240966797,
      "step": 93111
    },
    {
      "epoch": 1.8225669860839844e-05,
      "step": 93111,
      "training_step_time": 7.59454345703125
    },
    {
      "epoch": 1.822662353515625e-05,
      "model_forward_time": 1.420158863067627,
      "step": 93112
    },
    {
      "epoch": 1.822662353515625e-05,
      "step": 93112,
      "training_step_time": 7.776023864746094
    },
    {
      "epoch": 1.8227577209472656e-05,
      "model_forward_time": 1.8235254287719727,
      "step": 93113
    },
    {
      "epoch": 1.8227577209472656e-05,
      "step": 93113,
      "training_step_time": 7.658062696456909
    },
    {
      "epoch": 1.8228530883789063e-05,
      "model_forward_time": 1.7863450050354004,
      "step": 93114
    },
    {
      "epoch": 1.8228530883789063e-05,
      "step": 93114,
      "training_step_time": 7.742276191711426
    },
    {
      "epoch": 1.8229484558105467e-05,
      "model_forward_time": 1.896907091140747,
      "step": 93115
    },
    {
      "epoch": 1.8229484558105467e-05,
      "step": 93115,
      "training_step_time": 12.129104375839233
    },
    {
      "epoch": 1.8230438232421875e-05,
      "model_forward_time": 1.9297270774841309,
      "step": 93116
    },
    {
      "epoch": 1.8230438232421875e-05,
      "step": 93116,
      "training_step_time": 6.777686595916748
    },
    {
      "epoch": 1.8231391906738282e-05,
      "model_forward_time": 1.9205143451690674,
      "step": 93117
    },
    {
      "epoch": 1.8231391906738282e-05,
      "step": 93117,
      "training_step_time": 7.671964168548584
    },
    {
      "epoch": 1.8232345581054686e-05,
      "model_forward_time": 1.9042067527770996,
      "step": 93118
    },
    {
      "epoch": 1.8232345581054686e-05,
      "step": 93118,
      "training_step_time": 8.360694885253906
    },
    {
      "epoch": 1.8233299255371094e-05,
      "model_forward_time": 1.9103477001190186,
      "step": 93119
    },
    {
      "epoch": 1.8233299255371094e-05,
      "step": 93119,
      "training_step_time": 8.138479948043823
    },
    {
      "epoch": 1.82342529296875e-05,
      "grad_norm": 0.08535696566104889,
      "learning_rate": 1.2889049655259e-07,
      "loss": 0.0575,
      "step": 93120
    },
    {
      "epoch": 1.82342529296875e-05,
      "model_forward_time": 1.9137828350067139,
      "step": 93120
    },
    {
      "dynamics_loss_avg": 0.05549236033111811,
      "epoch": 1.82342529296875e-05,
      "step": 93120
    },
    {
      "action_loss_avg": 0.0017100242985179648,
      "epoch": 1.82342529296875e-05,
      "step": 93120
    },
    {
      "epoch": 1.82342529296875e-05,
      "step": 93120,
      "training_step_time": 6.742681980133057
    },
    {
      "epoch": 1.8235206604003905e-05,
      "model_forward_time": 1.9152047634124756,
      "step": 93121
    },
    {
      "epoch": 1.8235206604003905e-05,
      "step": 93121,
      "training_step_time": 6.694447755813599
    },
    {
      "epoch": 1.8236160278320312e-05,
      "model_forward_time": 1.8864426612854004,
      "step": 93122
    },
    {
      "epoch": 1.8236160278320312e-05,
      "step": 93122,
      "training_step_time": 6.8415138721466064
    },
    {
      "epoch": 1.823711395263672e-05,
      "model_forward_time": 1.8991167545318604,
      "step": 93123
    },
    {
      "epoch": 1.823711395263672e-05,
      "step": 93123,
      "training_step_time": 6.817313194274902
    },
    {
      "epoch": 1.8238067626953124e-05,
      "model_forward_time": 1.919419765472412,
      "step": 93124
    },
    {
      "epoch": 1.8238067626953124e-05,
      "step": 93124,
      "training_step_time": 6.765472173690796
    },
    {
      "epoch": 1.823902130126953e-05,
      "model_forward_time": 2.542865753173828,
      "step": 93125
    },
    {
      "epoch": 1.823902130126953e-05,
      "step": 93125,
      "training_step_time": 7.6184916496276855
    },
    {
      "epoch": 1.823997497558594e-05,
      "model_forward_time": 1.915428876876831,
      "step": 93126
    },
    {
      "epoch": 1.823997497558594e-05,
      "step": 93126,
      "training_step_time": 6.770799875259399
    },
    {
      "epoch": 1.8240928649902343e-05,
      "model_forward_time": 1.5305235385894775,
      "step": 93127
    },
    {
      "epoch": 1.8240928649902343e-05,
      "step": 93127,
      "training_step_time": 7.332509517669678
    },
    {
      "epoch": 1.824188232421875e-05,
      "model_forward_time": 1.542849063873291,
      "step": 93128
    },
    {
      "epoch": 1.824188232421875e-05,
      "step": 93128,
      "training_step_time": 7.411821603775024
    },
    {
      "epoch": 1.8242835998535158e-05,
      "model_forward_time": 1.907606601715088,
      "step": 93129
    },
    {
      "epoch": 1.8242835998535158e-05,
      "step": 93129,
      "training_step_time": 7.688928127288818
    },
    {
      "epoch": 1.824378967285156e-05,
      "grad_norm": 0.07161741703748703,
      "learning_rate": 1.2851775357300312e-07,
      "loss": 0.0564,
      "step": 93130
    },
    {
      "epoch": 1.824378967285156e-05,
      "model_forward_time": 1.8998398780822754,
      "step": 93130
    },
    {
      "dynamics_loss_avg": 0.051407873071730135,
      "epoch": 1.824378967285156e-05,
      "step": 93130
    },
    {
      "action_loss_avg": 0.0007781633816193789,
      "epoch": 1.824378967285156e-05,
      "step": 93130
    },
    {
      "epoch": 1.824378967285156e-05,
      "step": 93130,
      "training_step_time": 6.7382426261901855
    },
    {
      "epoch": 1.824474334716797e-05,
      "model_forward_time": 1.8946139812469482,
      "step": 93131
    },
    {
      "epoch": 1.824474334716797e-05,
      "step": 93131,
      "training_step_time": 6.817983865737915
    },
    {
      "epoch": 1.8245697021484376e-05,
      "model_forward_time": 1.900184154510498,
      "step": 93132
    },
    {
      "epoch": 1.8245697021484376e-05,
      "step": 93132,
      "training_step_time": 6.8472583293914795
    },
    {
      "epoch": 1.824665069580078e-05,
      "model_forward_time": 1.9043083190917969,
      "step": 93133
    },
    {
      "epoch": 1.824665069580078e-05,
      "step": 93133,
      "training_step_time": 6.782763719558716
    },
    {
      "epoch": 1.8247604370117188e-05,
      "model_forward_time": 2.4928808212280273,
      "step": 93134
    },
    {
      "epoch": 1.8247604370117188e-05,
      "step": 93134,
      "training_step_time": 7.750394821166992
    },
    {
      "epoch": 1.8248558044433595e-05,
      "model_forward_time": 1.9083218574523926,
      "step": 93135
    },
    {
      "epoch": 1.8248558044433595e-05,
      "step": 93135,
      "training_step_time": 6.7512593269348145
    },
    {
      "epoch": 1.824951171875e-05,
      "model_forward_time": 1.8975551128387451,
      "step": 93136
    },
    {
      "epoch": 1.824951171875e-05,
      "step": 93136,
      "training_step_time": 7.412521600723267
    },
    {
      "epoch": 1.8250465393066407e-05,
      "model_forward_time": 1.8889520168304443,
      "step": 93137
    },
    {
      "epoch": 1.8250465393066407e-05,
      "step": 93137,
      "training_step_time": 7.485125303268433
    },
    {
      "epoch": 1.8251419067382814e-05,
      "model_forward_time": 1.9066014289855957,
      "step": 93138
    },
    {
      "epoch": 1.8251419067382814e-05,
      "step": 93138,
      "training_step_time": 7.581672191619873
    },
    {
      "epoch": 1.8252372741699218e-05,
      "model_forward_time": 1.882343053817749,
      "step": 93139
    },
    {
      "epoch": 1.8252372741699218e-05,
      "step": 93139,
      "training_step_time": 6.7599937915802
    },
    {
      "epoch": 1.8253326416015626e-05,
      "grad_norm": 0.43396124243736267,
      "learning_rate": 1.281455433313916e-07,
      "loss": 0.0558,
      "step": 93140
    },
    {
      "epoch": 1.8253326416015626e-05,
      "model_forward_time": 1.9191341400146484,
      "step": 93140
    },
    {
      "dynamics_loss_avg": 0.04849756807088852,
      "epoch": 1.8253326416015626e-05,
      "step": 93140
    },
    {
      "action_loss_avg": 0.0019463303993688895,
      "epoch": 1.8253326416015626e-05,
      "step": 93140
    },
    {
      "epoch": 1.8253326416015626e-05,
      "step": 93140,
      "training_step_time": 6.7447898387908936
    },
    {
      "epoch": 1.825428009033203e-05,
      "model_forward_time": 1.9026973247528076,
      "step": 93141
    },
    {
      "epoch": 1.825428009033203e-05,
      "step": 93141,
      "training_step_time": 6.771153688430786
    },
    {
      "epoch": 1.8255233764648437e-05,
      "model_forward_time": 1.9038689136505127,
      "step": 93142
    },
    {
      "epoch": 1.8255233764648437e-05,
      "step": 93142,
      "training_step_time": 6.753683805465698
    },
    {
      "epoch": 1.8256187438964844e-05,
      "model_forward_time": 2.517411947250366,
      "step": 93143
    },
    {
      "epoch": 1.8256187438964844e-05,
      "step": 93143,
      "training_step_time": 8.691279411315918
    },
    {
      "epoch": 1.825714111328125e-05,
      "model_forward_time": 1.764063835144043,
      "step": 93144
    },
    {
      "epoch": 1.825714111328125e-05,
      "step": 93144,
      "training_step_time": 7.660593271255493
    },
    {
      "epoch": 1.8258094787597656e-05,
      "model_forward_time": 1.770111083984375,
      "step": 93145
    },
    {
      "epoch": 1.8258094787597656e-05,
      "step": 93145,
      "training_step_time": 7.397969961166382
    },
    {
      "epoch": 1.8259048461914063e-05,
      "model_forward_time": 1.8874931335449219,
      "step": 93146
    },
    {
      "epoch": 1.8259048461914063e-05,
      "step": 93146,
      "training_step_time": 7.451399087905884
    },
    {
      "epoch": 1.8260002136230467e-05,
      "model_forward_time": 1.9110677242279053,
      "step": 93147
    },
    {
      "epoch": 1.8260002136230467e-05,
      "step": 93147,
      "training_step_time": 8.198468446731567
    },
    {
      "epoch": 1.8260955810546875e-05,
      "model_forward_time": 1.911806583404541,
      "step": 93148
    },
    {
      "epoch": 1.8260955810546875e-05,
      "step": 93148,
      "training_step_time": 6.751621246337891
    },
    {
      "epoch": 1.8261909484863282e-05,
      "model_forward_time": 1.9226505756378174,
      "step": 93149
    },
    {
      "epoch": 1.8261909484863282e-05,
      "step": 93149,
      "training_step_time": 6.725444078445435
    },
    {
      "epoch": 1.8262863159179686e-05,
      "grad_norm": 0.0846855565905571,
      "learning_rate": 1.2777386586845796e-07,
      "loss": 0.0575,
      "step": 93150
    },
    {
      "epoch": 1.8262863159179686e-05,
      "model_forward_time": 1.9151651859283447,
      "step": 93150
    },
    {
      "dynamics_loss_avg": 0.06537943463772536,
      "epoch": 1.8262863159179686e-05,
      "step": 93150
    },
    {
      "action_loss_avg": 0.001816930584027432,
      "epoch": 1.8262863159179686e-05,
      "step": 93150
    },
    {
      "epoch": 1.8262863159179686e-05,
      "step": 93150,
      "training_step_time": 6.757889986038208
    },
    {
      "epoch": 1.8263816833496094e-05,
      "model_forward_time": 1.9033596515655518,
      "step": 93151
    },
    {
      "epoch": 1.8263816833496094e-05,
      "step": 93151,
      "training_step_time": 6.796396732330322
    },
    {
      "epoch": 1.82647705078125e-05,
      "model_forward_time": 2.5220303535461426,
      "step": 93152
    },
    {
      "epoch": 1.82647705078125e-05,
      "step": 93152,
      "training_step_time": 13.827186584472656
    },
    {
      "epoch": 1.8265724182128905e-05,
      "model_forward_time": 1.8996813297271729,
      "step": 93153
    },
    {
      "epoch": 1.8265724182128905e-05,
      "step": 93153,
      "training_step_time": 7.7140257358551025
    },
    {
      "epoch": 1.8266677856445312e-05,
      "model_forward_time": 1.9123108386993408,
      "step": 93154
    },
    {
      "epoch": 1.8266677856445312e-05,
      "step": 93154,
      "training_step_time": 7.896068334579468
    },
    {
      "epoch": 1.826763153076172e-05,
      "model_forward_time": 1.9132404327392578,
      "step": 93155
    },
    {
      "epoch": 1.826763153076172e-05,
      "step": 93155,
      "training_step_time": 7.491985559463501
    },
    {
      "epoch": 1.8268585205078124e-05,
      "model_forward_time": 1.913451910018921,
      "step": 93156
    },
    {
      "epoch": 1.8268585205078124e-05,
      "step": 93156,
      "training_step_time": 7.2683000564575195
    },
    {
      "epoch": 1.826953887939453e-05,
      "model_forward_time": 1.9185028076171875,
      "step": 93157
    },
    {
      "epoch": 1.826953887939453e-05,
      "step": 93157,
      "training_step_time": 7.652141094207764
    },
    {
      "epoch": 1.827049255371094e-05,
      "model_forward_time": 1.8820643424987793,
      "step": 93158
    },
    {
      "epoch": 1.827049255371094e-05,
      "step": 93158,
      "training_step_time": 6.761704444885254
    },
    {
      "epoch": 1.8271446228027343e-05,
      "model_forward_time": 1.905540943145752,
      "step": 93159
    },
    {
      "epoch": 1.8271446228027343e-05,
      "step": 93159,
      "training_step_time": 6.738917112350464
    },
    {
      "epoch": 1.827239990234375e-05,
      "grad_norm": 0.1116994172334671,
      "learning_rate": 1.2740272122484965e-07,
      "loss": 0.0576,
      "step": 93160
    },
    {
      "epoch": 1.827239990234375e-05,
      "model_forward_time": 1.8974120616912842,
      "step": 93160
    },
    {
      "dynamics_loss_avg": 0.059536096081137656,
      "epoch": 1.827239990234375e-05,
      "step": 93160
    },
    {
      "action_loss_avg": 0.0014612618222599849,
      "epoch": 1.827239990234375e-05,
      "step": 93160
    },
    {
      "epoch": 1.827239990234375e-05,
      "step": 93160,
      "training_step_time": 6.740071058273315
    },
    {
      "epoch": 1.8273353576660158e-05,
      "model_forward_time": 2.4828784465789795,
      "step": 93161
    },
    {
      "epoch": 1.8273353576660158e-05,
      "step": 93161,
      "training_step_time": 9.774497747421265
    },
    {
      "epoch": 1.827430725097656e-05,
      "model_forward_time": 1.8828718662261963,
      "step": 93162
    },
    {
      "epoch": 1.827430725097656e-05,
      "step": 93162,
      "training_step_time": 7.774407148361206
    },
    {
      "epoch": 1.827526092529297e-05,
      "model_forward_time": 1.910334825515747,
      "step": 93163
    },
    {
      "epoch": 1.827526092529297e-05,
      "step": 93163,
      "training_step_time": 7.253587961196899
    },
    {
      "epoch": 1.8276214599609376e-05,
      "model_forward_time": 1.5573945045471191,
      "step": 93164
    },
    {
      "epoch": 1.8276214599609376e-05,
      "step": 93164,
      "training_step_time": 8.0493905544281
    },
    {
      "epoch": 1.827716827392578e-05,
      "model_forward_time": 1.906768560409546,
      "step": 93165
    },
    {
      "epoch": 1.827716827392578e-05,
      "step": 93165,
      "training_step_time": 6.776338815689087
    },
    {
      "epoch": 1.8278121948242188e-05,
      "model_forward_time": 1.9108283519744873,
      "step": 93166
    },
    {
      "epoch": 1.8278121948242188e-05,
      "step": 93166,
      "training_step_time": 7.592304944992065
    },
    {
      "epoch": 1.8279075622558595e-05,
      "model_forward_time": 1.9287123680114746,
      "step": 93167
    },
    {
      "epoch": 1.8279075622558595e-05,
      "step": 93167,
      "training_step_time": 6.709350824356079
    },
    {
      "epoch": 1.8280029296875e-05,
      "model_forward_time": 1.919867992401123,
      "step": 93168
    },
    {
      "epoch": 1.8280029296875e-05,
      "step": 93168,
      "training_step_time": 6.70232629776001
    },
    {
      "epoch": 1.8280982971191407e-05,
      "model_forward_time": 1.9159431457519531,
      "step": 93169
    },
    {
      "epoch": 1.8280982971191407e-05,
      "step": 93169,
      "training_step_time": 6.755047559738159
    },
    {
      "epoch": 1.8281936645507814e-05,
      "grad_norm": 0.1442103087902069,
      "learning_rate": 1.2703210944115307e-07,
      "loss": 0.0569,
      "step": 93170
    },
    {
      "epoch": 1.8281936645507814e-05,
      "model_forward_time": 1.886038064956665,
      "step": 93170
    },
    {
      "dynamics_loss_avg": 0.03975577438250184,
      "epoch": 1.8281936645507814e-05,
      "step": 93170
    },
    {
      "action_loss_avg": 0.0011932312452699989,
      "epoch": 1.8281936645507814e-05,
      "step": 93170
    },
    {
      "epoch": 1.8281936645507814e-05,
      "step": 93170,
      "training_step_time": 9.304969072341919
    },
    {
      "epoch": 1.8282890319824218e-05,
      "model_forward_time": 2.4863739013671875,
      "step": 93171
    },
    {
      "epoch": 1.8282890319824218e-05,
      "step": 93171,
      "training_step_time": 7.736602306365967
    },
    {
      "epoch": 1.8283843994140626e-05,
      "model_forward_time": 1.914057731628418,
      "step": 93172
    },
    {
      "epoch": 1.8283843994140626e-05,
      "step": 93172,
      "training_step_time": 7.40099835395813
    },
    {
      "epoch": 1.828479766845703e-05,
      "model_forward_time": 1.9063336849212646,
      "step": 93173
    },
    {
      "epoch": 1.828479766845703e-05,
      "step": 93173,
      "training_step_time": 8.199271202087402
    },
    {
      "epoch": 1.8285751342773437e-05,
      "model_forward_time": 1.901439905166626,
      "step": 93174
    },
    {
      "epoch": 1.8285751342773437e-05,
      "step": 93174,
      "training_step_time": 7.252731800079346
    },
    {
      "epoch": 1.8286705017089844e-05,
      "model_forward_time": 1.8899352550506592,
      "step": 93175
    },
    {
      "epoch": 1.8286705017089844e-05,
      "step": 93175,
      "training_step_time": 7.566403388977051
    },
    {
      "epoch": 1.828765869140625e-05,
      "model_forward_time": 1.9243342876434326,
      "step": 93176
    },
    {
      "epoch": 1.828765869140625e-05,
      "step": 93176,
      "training_step_time": 7.602864742279053
    },
    {
      "epoch": 1.8288612365722656e-05,
      "model_forward_time": 1.922328233718872,
      "step": 93177
    },
    {
      "epoch": 1.8288612365722656e-05,
      "step": 93177,
      "training_step_time": 6.79935359954834
    },
    {
      "epoch": 1.8289566040039063e-05,
      "model_forward_time": 1.8916442394256592,
      "step": 93178
    },
    {
      "epoch": 1.8289566040039063e-05,
      "step": 93178,
      "training_step_time": 6.800650119781494
    },
    {
      "epoch": 1.8290519714355467e-05,
      "model_forward_time": 1.8875148296356201,
      "step": 93179
    },
    {
      "epoch": 1.8290519714355467e-05,
      "step": 93179,
      "training_step_time": 7.639981031417847
    },
    {
      "epoch": 1.8291473388671875e-05,
      "grad_norm": 0.19547544419765472,
      "learning_rate": 1.2666203055789915e-07,
      "loss": 0.056,
      "step": 93180
    },
    {
      "epoch": 1.8291473388671875e-05,
      "model_forward_time": 2.523123264312744,
      "step": 93180
    },
    {
      "dynamics_loss_avg": 0.06287885382771492,
      "epoch": 1.8291473388671875e-05,
      "step": 93180
    },
    {
      "action_loss_avg": 0.0015143308584811166,
      "epoch": 1.8291473388671875e-05,
      "step": 93180
    },
    {
      "epoch": 1.8291473388671875e-05,
      "step": 93180,
      "training_step_time": 9.486757278442383
    },
    {
      "epoch": 1.8292427062988282e-05,
      "model_forward_time": 1.922985553741455,
      "step": 93181
    },
    {
      "epoch": 1.8292427062988282e-05,
      "step": 93181,
      "training_step_time": 7.535350561141968
    },
    {
      "epoch": 1.8293380737304686e-05,
      "model_forward_time": 1.9186551570892334,
      "step": 93182
    },
    {
      "epoch": 1.8293380737304686e-05,
      "step": 93182,
      "training_step_time": 7.463801860809326
    },
    {
      "epoch": 1.8294334411621094e-05,
      "model_forward_time": 1.929966926574707,
      "step": 93183
    },
    {
      "epoch": 1.8294334411621094e-05,
      "step": 93183,
      "training_step_time": 7.451348304748535
    },
    {
      "epoch": 1.82952880859375e-05,
      "model_forward_time": 1.89656662940979,
      "step": 93184
    },
    {
      "epoch": 1.82952880859375e-05,
      "step": 93184,
      "training_step_time": 7.311091423034668
    },
    {
      "epoch": 1.8296241760253905e-05,
      "model_forward_time": 1.9356343746185303,
      "step": 93185
    },
    {
      "epoch": 1.8296241760253905e-05,
      "step": 93185,
      "training_step_time": 7.62393593788147
    },
    {
      "epoch": 1.8297195434570312e-05,
      "model_forward_time": 1.9199535846710205,
      "step": 93186
    },
    {
      "epoch": 1.8297195434570312e-05,
      "step": 93186,
      "training_step_time": 6.79813289642334
    },
    {
      "epoch": 1.829814910888672e-05,
      "model_forward_time": 1.9210000038146973,
      "step": 93187
    },
    {
      "epoch": 1.829814910888672e-05,
      "step": 93187,
      "training_step_time": 6.777894973754883
    },
    {
      "epoch": 1.8299102783203124e-05,
      "model_forward_time": 1.9291841983795166,
      "step": 93188
    },
    {
      "epoch": 1.8299102783203124e-05,
      "step": 93188,
      "training_step_time": 7.607867479324341
    },
    {
      "epoch": 1.830005645751953e-05,
      "model_forward_time": 2.5165669918060303,
      "step": 93189
    },
    {
      "epoch": 1.830005645751953e-05,
      "step": 93189,
      "training_step_time": 13.280871629714966
    },
    {
      "epoch": 1.830101013183594e-05,
      "grad_norm": 0.09061188250780106,
      "learning_rate": 1.262924846155583e-07,
      "loss": 0.0563,
      "step": 93190
    },
    {
      "epoch": 1.830101013183594e-05,
      "model_forward_time": 1.9048972129821777,
      "step": 93190
    },
    {
      "dynamics_loss_avg": 0.056743447482585904,
      "epoch": 1.830101013183594e-05,
      "step": 93190
    },
    {
      "action_loss_avg": 0.0014826636586803942,
      "epoch": 1.830101013183594e-05,
      "step": 93190
    },
    {
      "epoch": 1.830101013183594e-05,
      "step": 93190,
      "training_step_time": 7.632738828659058
    },
    {
      "epoch": 1.8301963806152343e-05,
      "model_forward_time": 1.9184858798980713,
      "step": 93191
    },
    {
      "epoch": 1.8301963806152343e-05,
      "step": 93191,
      "training_step_time": 7.419831037521362
    },
    {
      "epoch": 1.830291748046875e-05,
      "model_forward_time": 1.916496753692627,
      "step": 93192
    },
    {
      "epoch": 1.830291748046875e-05,
      "step": 93192,
      "training_step_time": 7.473953723907471
    },
    {
      "epoch": 1.8303871154785158e-05,
      "model_forward_time": 1.9084081649780273,
      "step": 93193
    },
    {
      "epoch": 1.8303871154785158e-05,
      "step": 93193,
      "training_step_time": 6.786350965499878
    },
    {
      "epoch": 1.830482482910156e-05,
      "model_forward_time": 1.9125230312347412,
      "step": 93194
    },
    {
      "epoch": 1.830482482910156e-05,
      "step": 93194,
      "training_step_time": 7.639949560165405
    },
    {
      "epoch": 1.830577850341797e-05,
      "model_forward_time": 1.9108500480651855,
      "step": 93195
    },
    {
      "epoch": 1.830577850341797e-05,
      "step": 93195,
      "training_step_time": 6.793467044830322
    },
    {
      "epoch": 1.8306732177734376e-05,
      "model_forward_time": 1.9138157367706299,
      "step": 93196
    },
    {
      "epoch": 1.8306732177734376e-05,
      "step": 93196,
      "training_step_time": 6.780665874481201
    },
    {
      "epoch": 1.830768585205078e-05,
      "model_forward_time": 1.9090466499328613,
      "step": 93197
    },
    {
      "epoch": 1.830768585205078e-05,
      "step": 93197,
      "training_step_time": 7.093428611755371
    },
    {
      "epoch": 1.8308639526367188e-05,
      "model_forward_time": 2.511331081390381,
      "step": 93198
    },
    {
      "epoch": 1.8308639526367188e-05,
      "step": 93198,
      "training_step_time": 11.985852718353271
    },
    {
      "epoch": 1.8309593200683595e-05,
      "model_forward_time": 1.9134368896484375,
      "step": 93199
    },
    {
      "epoch": 1.8309593200683595e-05,
      "step": 93199,
      "training_step_time": 7.742027521133423
    },
    {
      "epoch": 1.8310546875e-05,
      "grad_norm": 0.10035353153944016,
      "learning_rate": 1.2592347165454434e-07,
      "loss": 0.0569,
      "step": 93200
    },
    {
      "epoch": 1.8310546875e-05,
      "model_forward_time": 1.9183294773101807,
      "step": 93200
    },
    {
      "dynamics_loss_avg": 0.05887798126786947,
      "epoch": 1.8310546875e-05,
      "step": 93200
    },
    {
      "action_loss_avg": 0.0018101702036801726,
      "epoch": 1.8310546875e-05,
      "step": 93200
    },
    {
      "epoch": 1.8310546875e-05,
      "step": 93200,
      "training_step_time": 8.248120784759521
    },
    {
      "epoch": 1.8311500549316407e-05,
      "model_forward_time": 1.9152002334594727,
      "step": 93201
    },
    {
      "epoch": 1.8311500549316407e-05,
      "step": 93201,
      "training_step_time": 7.4000959396362305
    },
    {
      "epoch": 1.8312454223632814e-05,
      "model_forward_time": 1.9077670574188232,
      "step": 93202
    },
    {
      "epoch": 1.8312454223632814e-05,
      "step": 93202,
      "training_step_time": 6.758808612823486
    },
    {
      "epoch": 1.8313407897949218e-05,
      "model_forward_time": 1.9140491485595703,
      "step": 93203
    },
    {
      "epoch": 1.8313407897949218e-05,
      "step": 93203,
      "training_step_time": 7.66814661026001
    },
    {
      "epoch": 1.8314361572265626e-05,
      "model_forward_time": 1.9264068603515625,
      "step": 93204
    },
    {
      "epoch": 1.8314361572265626e-05,
      "step": 93204,
      "training_step_time": 6.7619616985321045
    },
    {
      "epoch": 1.831531524658203e-05,
      "model_forward_time": 1.912980556488037,
      "step": 93205
    },
    {
      "epoch": 1.831531524658203e-05,
      "step": 93205,
      "training_step_time": 6.754865407943726
    },
    {
      "epoch": 1.8316268920898437e-05,
      "model_forward_time": 1.9014391899108887,
      "step": 93206
    },
    {
      "epoch": 1.8316268920898437e-05,
      "step": 93206,
      "training_step_time": 7.117995023727417
    },
    {
      "epoch": 1.8317222595214844e-05,
      "model_forward_time": 1.8725309371948242,
      "step": 93207
    },
    {
      "epoch": 1.8317222595214844e-05,
      "step": 93207,
      "training_step_time": 10.57614803314209
    },
    {
      "epoch": 1.831817626953125e-05,
      "model_forward_time": 1.920562982559204,
      "step": 93208
    },
    {
      "epoch": 1.831817626953125e-05,
      "step": 93208,
      "training_step_time": 10.17379355430603
    },
    {
      "epoch": 1.8319129943847656e-05,
      "model_forward_time": 1.923879623413086,
      "step": 93209
    },
    {
      "epoch": 1.8319129943847656e-05,
      "step": 93209,
      "training_step_time": 8.545973777770996
    },
    {
      "epoch": 1.8320083618164063e-05,
      "grad_norm": 0.08766409009695053,
      "learning_rate": 1.2555499171521102e-07,
      "loss": 0.0555,
      "step": 93210
    },
    {
      "epoch": 1.8320083618164063e-05,
      "model_forward_time": 1.9158785343170166,
      "step": 93210
    },
    {
      "dynamics_loss_avg": 0.04993226872757077,
      "epoch": 1.8320083618164063e-05,
      "step": 93210
    },
    {
      "action_loss_avg": 0.001481366829830222,
      "epoch": 1.8320083618164063e-05,
      "step": 93210
    },
    {
      "epoch": 1.8320083618164063e-05,
      "step": 93210,
      "training_step_time": 7.471046686172485
    },
    {
      "epoch": 1.8321037292480467e-05,
      "model_forward_time": 1.9088420867919922,
      "step": 93211
    },
    {
      "epoch": 1.8321037292480467e-05,
      "step": 93211,
      "training_step_time": 7.284286022186279
    },
    {
      "epoch": 1.8321990966796875e-05,
      "model_forward_time": 1.909850835800171,
      "step": 93212
    },
    {
      "epoch": 1.8321990966796875e-05,
      "step": 93212,
      "training_step_time": 8.347004175186157
    },
    {
      "epoch": 1.8322944641113282e-05,
      "model_forward_time": 1.903202772140503,
      "step": 93213
    },
    {
      "epoch": 1.8322944641113282e-05,
      "step": 93213,
      "training_step_time": 7.6298508644104
    },
    {
      "epoch": 1.8323898315429686e-05,
      "model_forward_time": 1.9261457920074463,
      "step": 93214
    },
    {
      "epoch": 1.8323898315429686e-05,
      "step": 93214,
      "training_step_time": 6.762422561645508
    },
    {
      "epoch": 1.8324851989746094e-05,
      "model_forward_time": 1.9074313640594482,
      "step": 93215
    },
    {
      "epoch": 1.8324851989746094e-05,
      "step": 93215,
      "training_step_time": 7.1053900718688965
    },
    {
      "epoch": 1.83258056640625e-05,
      "model_forward_time": 1.904881238937378,
      "step": 93216
    },
    {
      "epoch": 1.83258056640625e-05,
      "step": 93216,
      "training_step_time": 9.016072988510132
    },
    {
      "epoch": 1.8326759338378905e-05,
      "model_forward_time": 2.523627758026123,
      "step": 93217
    },
    {
      "epoch": 1.8326759338378905e-05,
      "step": 93217,
      "training_step_time": 8.468951940536499
    },
    {
      "epoch": 1.8327713012695312e-05,
      "model_forward_time": 1.9187910556793213,
      "step": 93218
    },
    {
      "epoch": 1.8327713012695312e-05,
      "step": 93218,
      "training_step_time": 7.6220383644104
    },
    {
      "epoch": 1.832866668701172e-05,
      "model_forward_time": 1.9095299243927002,
      "step": 93219
    },
    {
      "epoch": 1.832866668701172e-05,
      "step": 93219,
      "training_step_time": 7.499867916107178
    },
    {
      "epoch": 1.8329620361328124e-05,
      "grad_norm": 0.0950557067990303,
      "learning_rate": 1.251870448378556e-07,
      "loss": 0.0542,
      "step": 93220
    },
    {
      "epoch": 1.8329620361328124e-05,
      "model_forward_time": 1.9040839672088623,
      "step": 93220
    },
    {
      "dynamics_loss_avg": 0.058499073050916194,
      "epoch": 1.8329620361328124e-05,
      "step": 93220
    },
    {
      "action_loss_avg": 0.0011935365298995748,
      "epoch": 1.8329620361328124e-05,
      "step": 93220
    },
    {
      "epoch": 1.8329620361328124e-05,
      "step": 93220,
      "training_step_time": 7.292701005935669
    },
    {
      "epoch": 1.833057403564453e-05,
      "model_forward_time": 1.8944544792175293,
      "step": 93221
    },
    {
      "epoch": 1.833057403564453e-05,
      "step": 93221,
      "training_step_time": 7.6529412269592285
    },
    {
      "epoch": 1.833152770996094e-05,
      "model_forward_time": 1.9010696411132812,
      "step": 93222
    },
    {
      "epoch": 1.833152770996094e-05,
      "step": 93222,
      "training_step_time": 7.652559757232666
    },
    {
      "epoch": 1.8332481384277343e-05,
      "model_forward_time": 1.9141147136688232,
      "step": 93223
    },
    {
      "epoch": 1.8332481384277343e-05,
      "step": 93223,
      "training_step_time": 6.72674822807312
    },
    {
      "epoch": 1.833343505859375e-05,
      "model_forward_time": 1.9083826541900635,
      "step": 93224
    },
    {
      "epoch": 1.833343505859375e-05,
      "step": 93224,
      "training_step_time": 7.150163412094116
    },
    {
      "epoch": 1.8334388732910158e-05,
      "model_forward_time": 1.9030208587646484,
      "step": 93225
    },
    {
      "epoch": 1.8334388732910158e-05,
      "step": 93225,
      "training_step_time": 9.607421875
    },
    {
      "epoch": 1.833534240722656e-05,
      "model_forward_time": 2.4935827255249023,
      "step": 93226
    },
    {
      "epoch": 1.833534240722656e-05,
      "step": 93226,
      "training_step_time": 8.683694124221802
    },
    {
      "epoch": 1.833629608154297e-05,
      "model_forward_time": 1.9183497428894043,
      "step": 93227
    },
    {
      "epoch": 1.833629608154297e-05,
      "step": 93227,
      "training_step_time": 7.756393194198608
    },
    {
      "epoch": 1.8337249755859376e-05,
      "model_forward_time": 1.922084093093872,
      "step": 93228
    },
    {
      "epoch": 1.8337249755859376e-05,
      "step": 93228,
      "training_step_time": 7.501844167709351
    },
    {
      "epoch": 1.833820343017578e-05,
      "model_forward_time": 1.9041383266448975,
      "step": 93229
    },
    {
      "epoch": 1.833820343017578e-05,
      "step": 93229,
      "training_step_time": 7.568478345870972
    },
    {
      "epoch": 1.8339157104492188e-05,
      "grad_norm": 0.4517967402935028,
      "learning_rate": 1.2481963106271478e-07,
      "loss": 0.057,
      "step": 93230
    },
    {
      "epoch": 1.8339157104492188e-05,
      "model_forward_time": 1.8812270164489746,
      "step": 93230
    },
    {
      "dynamics_loss_avg": 0.05602397434413433,
      "epoch": 1.8339157104492188e-05,
      "step": 93230
    },
    {
      "action_loss_avg": 0.0013721197145059704,
      "epoch": 1.8339157104492188e-05,
      "step": 93230
    },
    {
      "epoch": 1.8339157104492188e-05,
      "step": 93230,
      "training_step_time": 7.71607518196106
    },
    {
      "epoch": 1.8340110778808595e-05,
      "model_forward_time": 1.917125940322876,
      "step": 93231
    },
    {
      "epoch": 1.8340110778808595e-05,
      "step": 93231,
      "training_step_time": 7.660502672195435
    },
    {
      "epoch": 1.8341064453125e-05,
      "model_forward_time": 1.92034912109375,
      "step": 93232
    },
    {
      "epoch": 1.8341064453125e-05,
      "step": 93232,
      "training_step_time": 6.731650352478027
    },
    {
      "epoch": 1.8342018127441407e-05,
      "model_forward_time": 1.9032015800476074,
      "step": 93233
    },
    {
      "epoch": 1.8342018127441407e-05,
      "step": 93233,
      "training_step_time": 7.134332895278931
    },
    {
      "epoch": 1.8342971801757814e-05,
      "model_forward_time": 1.7556164264678955,
      "step": 93234
    },
    {
      "epoch": 1.8342971801757814e-05,
      "step": 93234,
      "training_step_time": 7.418525457382202
    },
    {
      "epoch": 1.8343925476074218e-05,
      "model_forward_time": 2.4537227153778076,
      "step": 93235
    },
    {
      "epoch": 1.8343925476074218e-05,
      "step": 93235,
      "training_step_time": 9.135209083557129
    },
    {
      "epoch": 1.8344879150390626e-05,
      "model_forward_time": 1.7792778015136719,
      "step": 93236
    },
    {
      "epoch": 1.8344879150390626e-05,
      "step": 93236,
      "training_step_time": 7.736414194107056
    },
    {
      "epoch": 1.834583282470703e-05,
      "model_forward_time": 1.7745437622070312,
      "step": 93237
    },
    {
      "epoch": 1.834583282470703e-05,
      "step": 93237,
      "training_step_time": 7.318427801132202
    },
    {
      "epoch": 1.8346786499023437e-05,
      "model_forward_time": 1.770728588104248,
      "step": 93238
    },
    {
      "epoch": 1.8346786499023437e-05,
      "step": 93238,
      "training_step_time": 7.470104217529297
    },
    {
      "epoch": 1.8347740173339844e-05,
      "model_forward_time": 1.7879588603973389,
      "step": 93239
    },
    {
      "epoch": 1.8347740173339844e-05,
      "step": 93239,
      "training_step_time": 7.770880699157715
    },
    {
      "epoch": 1.834869384765625e-05,
      "grad_norm": 0.10142115503549576,
      "learning_rate": 1.2445275042997084e-07,
      "loss": 0.0549,
      "step": 93240
    },
    {
      "epoch": 1.834869384765625e-05,
      "model_forward_time": 1.767322301864624,
      "step": 93240
    },
    {
      "dynamics_loss_avg": 0.043471126910299064,
      "epoch": 1.834869384765625e-05,
      "step": 93240
    },
    {
      "action_loss_avg": 0.001234749666764401,
      "epoch": 1.834869384765625e-05,
      "step": 93240
    },
    {
      "epoch": 1.834869384765625e-05,
      "step": 93240,
      "training_step_time": 7.653075695037842
    },
    {
      "epoch": 1.8349647521972656e-05,
      "model_forward_time": 1.7794713973999023,
      "step": 93241
    },
    {
      "epoch": 1.8349647521972656e-05,
      "step": 93241,
      "training_step_time": 6.723443269729614
    },
    {
      "epoch": 1.8350601196289063e-05,
      "model_forward_time": 1.7878882884979248,
      "step": 93242
    },
    {
      "epoch": 1.8350601196289063e-05,
      "step": 93242,
      "training_step_time": 6.753075361251831
    },
    {
      "epoch": 1.8351554870605467e-05,
      "model_forward_time": 1.7837512493133545,
      "step": 93243
    },
    {
      "epoch": 1.8351554870605467e-05,
      "step": 93243,
      "training_step_time": 7.598698854446411
    },
    {
      "epoch": 1.8352508544921875e-05,
      "model_forward_time": 2.46860933303833,
      "step": 93244
    },
    {
      "epoch": 1.8352508544921875e-05,
      "step": 93244,
      "training_step_time": 10.469966173171997
    },
    {
      "epoch": 1.8353462219238282e-05,
      "model_forward_time": 1.8703854084014893,
      "step": 93245
    },
    {
      "epoch": 1.8353462219238282e-05,
      "step": 93245,
      "training_step_time": 7.727121591567993
    },
    {
      "epoch": 1.8354415893554686e-05,
      "model_forward_time": 1.9248106479644775,
      "step": 93246
    },
    {
      "epoch": 1.8354415893554686e-05,
      "step": 93246,
      "training_step_time": 7.639274597167969
    },
    {
      "epoch": 1.8355369567871094e-05,
      "model_forward_time": 1.9234144687652588,
      "step": 93247
    },
    {
      "epoch": 1.8355369567871094e-05,
      "step": 93247,
      "training_step_time": 7.520472526550293
    },
    {
      "epoch": 1.83563232421875e-05,
      "model_forward_time": 1.9329357147216797,
      "step": 93248
    },
    {
      "epoch": 1.83563232421875e-05,
      "step": 93248,
      "training_step_time": 7.313069820404053
    },
    {
      "epoch": 1.8357276916503905e-05,
      "model_forward_time": 1.9297990798950195,
      "step": 93249
    },
    {
      "epoch": 1.8357276916503905e-05,
      "step": 93249,
      "training_step_time": 8.355991840362549
    },
    {
      "epoch": 1.8358230590820312e-05,
      "grad_norm": 0.09186360239982605,
      "learning_rate": 1.2408640297974283e-07,
      "loss": 0.0545,
      "step": 93250
    },
    {
      "epoch": 1.8358230590820312e-05,
      "model_forward_time": 1.9163591861724854,
      "step": 93250
    },
    {
      "dynamics_loss_avg": 0.054951663967221974,
      "epoch": 1.8358230590820312e-05,
      "step": 93250
    },
    {
      "action_loss_avg": 0.0017940682708285749,
      "epoch": 1.8358230590820312e-05,
      "step": 93250
    },
    {
      "epoch": 1.8358230590820312e-05,
      "step": 93250,
      "training_step_time": 6.714565753936768
    },
    {
      "epoch": 1.835918426513672e-05,
      "model_forward_time": 1.908996343612671,
      "step": 93251
    },
    {
      "epoch": 1.835918426513672e-05,
      "step": 93251,
      "training_step_time": 6.73155403137207
    },
    {
      "epoch": 1.8360137939453124e-05,
      "model_forward_time": 1.906236171722412,
      "step": 93252
    },
    {
      "epoch": 1.8360137939453124e-05,
      "step": 93252,
      "training_step_time": 7.5116846561431885
    },
    {
      "epoch": 1.836109161376953e-05,
      "model_forward_time": 2.163301467895508,
      "step": 93253
    },
    {
      "epoch": 1.836109161376953e-05,
      "step": 93253,
      "training_step_time": 8.439005374908447
    },
    {
      "epoch": 1.836204528808594e-05,
      "model_forward_time": 1.548563003540039,
      "step": 93254
    },
    {
      "epoch": 1.836204528808594e-05,
      "step": 93254,
      "training_step_time": 7.811083078384399
    },
    {
      "epoch": 1.8362998962402343e-05,
      "model_forward_time": 1.9006874561309814,
      "step": 93255
    },
    {
      "epoch": 1.8362998962402343e-05,
      "step": 93255,
      "training_step_time": 7.617232322692871
    },
    {
      "epoch": 1.836395263671875e-05,
      "model_forward_time": 1.9012134075164795,
      "step": 93256
    },
    {
      "epoch": 1.836395263671875e-05,
      "step": 93256,
      "training_step_time": 7.48113226890564
    },
    {
      "epoch": 1.8364906311035158e-05,
      "model_forward_time": 1.9213502407073975,
      "step": 93257
    },
    {
      "epoch": 1.8364906311035158e-05,
      "step": 93257,
      "training_step_time": 7.297943592071533
    },
    {
      "epoch": 1.836585998535156e-05,
      "model_forward_time": 1.920874834060669,
      "step": 93258
    },
    {
      "epoch": 1.836585998535156e-05,
      "step": 93258,
      "training_step_time": 7.605438947677612
    },
    {
      "epoch": 1.836681365966797e-05,
      "model_forward_time": 1.911346197128296,
      "step": 93259
    },
    {
      "epoch": 1.836681365966797e-05,
      "step": 93259,
      "training_step_time": 7.211426019668579
    },
    {
      "epoch": 1.8367767333984376e-05,
      "grad_norm": 0.08824440091848373,
      "learning_rate": 1.2372058875209536e-07,
      "loss": 0.0526,
      "step": 93260
    },
    {
      "epoch": 1.8367767333984376e-05,
      "model_forward_time": 1.9213716983795166,
      "step": 93260
    },
    {
      "dynamics_loss_avg": 0.05527967996895313,
      "epoch": 1.8367767333984376e-05,
      "step": 93260
    },
    {
      "action_loss_avg": 0.0019672193040605633,
      "epoch": 1.8367767333984376e-05,
      "step": 93260
    },
    {
      "epoch": 1.8367767333984376e-05,
      "step": 93260,
      "training_step_time": 6.689271688461304
    },
    {
      "epoch": 1.836872100830078e-05,
      "model_forward_time": 1.9152438640594482,
      "step": 93261
    },
    {
      "epoch": 1.836872100830078e-05,
      "step": 93261,
      "training_step_time": 7.457859039306641
    },
    {
      "epoch": 1.8369674682617188e-05,
      "model_forward_time": 1.9097731113433838,
      "step": 93262
    },
    {
      "epoch": 1.8369674682617188e-05,
      "step": 93262,
      "training_step_time": 8.4838707447052
    },
    {
      "epoch": 1.8370628356933595e-05,
      "model_forward_time": 2.564960241317749,
      "step": 93263
    },
    {
      "epoch": 1.8370628356933595e-05,
      "step": 93263,
      "training_step_time": 9.297233819961548
    },
    {
      "epoch": 1.837158203125e-05,
      "model_forward_time": 1.9045047760009766,
      "step": 93264
    },
    {
      "epoch": 1.837158203125e-05,
      "step": 93264,
      "training_step_time": 7.682796478271484
    },
    {
      "epoch": 1.8372535705566407e-05,
      "model_forward_time": 1.9061124324798584,
      "step": 93265
    },
    {
      "epoch": 1.8372535705566407e-05,
      "step": 93265,
      "training_step_time": 7.408576011657715
    },
    {
      "epoch": 1.837348937988281e-05,
      "model_forward_time": 1.9146771430969238,
      "step": 93266
    },
    {
      "epoch": 1.837348937988281e-05,
      "step": 93266,
      "training_step_time": 7.492045879364014
    },
    {
      "epoch": 1.8374443054199218e-05,
      "model_forward_time": 1.9202492237091064,
      "step": 93267
    },
    {
      "epoch": 1.8374443054199218e-05,
      "step": 93267,
      "training_step_time": 7.560597658157349
    },
    {
      "epoch": 1.8375396728515626e-05,
      "model_forward_time": 1.5420212745666504,
      "step": 93268
    },
    {
      "epoch": 1.8375396728515626e-05,
      "step": 93268,
      "training_step_time": 7.628135681152344
    },
    {
      "epoch": 1.837635040283203e-05,
      "model_forward_time": 1.9068896770477295,
      "step": 93269
    },
    {
      "epoch": 1.837635040283203e-05,
      "step": 93269,
      "training_step_time": 6.712501764297485
    },
    {
      "epoch": 1.8377304077148437e-05,
      "grad_norm": 0.09675463289022446,
      "learning_rate": 1.2335530778703253e-07,
      "loss": 0.053,
      "step": 93270
    },
    {
      "epoch": 1.8377304077148437e-05,
      "model_forward_time": 1.9197289943695068,
      "step": 93270
    },
    {
      "dynamics_loss_avg": 0.05656870827078819,
      "epoch": 1.8377304077148437e-05,
      "step": 93270
    },
    {
      "action_loss_avg": 0.002407805123948492,
      "epoch": 1.8377304077148437e-05,
      "step": 93270
    },
    {
      "epoch": 1.8377304077148437e-05,
      "step": 93270,
      "training_step_time": 7.397342920303345
    },
    {
      "epoch": 1.8378257751464845e-05,
      "model_forward_time": 1.9101603031158447,
      "step": 93271
    },
    {
      "epoch": 1.8378257751464845e-05,
      "step": 93271,
      "training_step_time": 9.29753828048706
    },
    {
      "epoch": 1.837921142578125e-05,
      "model_forward_time": 2.5689404010772705,
      "step": 93272
    },
    {
      "epoch": 1.837921142578125e-05,
      "step": 93272,
      "training_step_time": 10.238794803619385
    },
    {
      "epoch": 1.8380165100097656e-05,
      "model_forward_time": 1.9154393672943115,
      "step": 93273
    },
    {
      "epoch": 1.8380165100097656e-05,
      "step": 93273,
      "training_step_time": 7.732832908630371
    },
    {
      "epoch": 1.8381118774414063e-05,
      "model_forward_time": 1.9008653163909912,
      "step": 93274
    },
    {
      "epoch": 1.8381118774414063e-05,
      "step": 93274,
      "training_step_time": 7.4809300899505615
    },
    {
      "epoch": 1.8382072448730467e-05,
      "model_forward_time": 1.9147593975067139,
      "step": 93275
    },
    {
      "epoch": 1.8382072448730467e-05,
      "step": 93275,
      "training_step_time": 7.388491153717041
    },
    {
      "epoch": 1.8383026123046875e-05,
      "model_forward_time": 1.9185245037078857,
      "step": 93276
    },
    {
      "epoch": 1.8383026123046875e-05,
      "step": 93276,
      "training_step_time": 7.4660115242004395
    },
    {
      "epoch": 1.8383979797363282e-05,
      "model_forward_time": 1.9241249561309814,
      "step": 93277
    },
    {
      "epoch": 1.8383979797363282e-05,
      "step": 93277,
      "training_step_time": 7.734263896942139
    },
    {
      "epoch": 1.8384933471679686e-05,
      "model_forward_time": 1.8887543678283691,
      "step": 93278
    },
    {
      "epoch": 1.8384933471679686e-05,
      "step": 93278,
      "training_step_time": 6.728421211242676
    },
    {
      "epoch": 1.8385887145996094e-05,
      "model_forward_time": 1.916520357131958,
      "step": 93279
    },
    {
      "epoch": 1.8385887145996094e-05,
      "step": 93279,
      "training_step_time": 7.179381608963013
    },
    {
      "epoch": 1.83868408203125e-05,
      "grad_norm": 0.09691166132688522,
      "learning_rate": 1.2299056012450184e-07,
      "loss": 0.0528,
      "step": 93280
    },
    {
      "epoch": 1.83868408203125e-05,
      "model_forward_time": 1.905334234237671,
      "step": 93280
    },
    {
      "dynamics_loss_avg": 0.05725667402148247,
      "epoch": 1.83868408203125e-05,
      "step": 93280
    },
    {
      "action_loss_avg": 0.0014867276331642643,
      "epoch": 1.83868408203125e-05,
      "step": 93280
    },
    {
      "epoch": 1.83868408203125e-05,
      "step": 93280,
      "training_step_time": 8.724758386611938
    },
    {
      "epoch": 1.8387794494628905e-05,
      "model_forward_time": 2.5562877655029297,
      "step": 93281
    },
    {
      "epoch": 1.8387794494628905e-05,
      "step": 93281,
      "training_step_time": 11.143943786621094
    },
    {
      "epoch": 1.8388748168945313e-05,
      "model_forward_time": 1.1949760913848877,
      "step": 93282
    },
    {
      "epoch": 1.8388748168945313e-05,
      "step": 93282,
      "training_step_time": 7.750684976577759
    },
    {
      "epoch": 1.838970184326172e-05,
      "model_forward_time": 1.8898520469665527,
      "step": 93283
    },
    {
      "epoch": 1.838970184326172e-05,
      "step": 93283,
      "training_step_time": 7.6367738246917725
    },
    {
      "epoch": 1.8390655517578124e-05,
      "model_forward_time": 1.9142684936523438,
      "step": 93284
    },
    {
      "epoch": 1.8390655517578124e-05,
      "step": 93284,
      "training_step_time": 8.160500049591064
    },
    {
      "epoch": 1.839160919189453e-05,
      "model_forward_time": 1.9112870693206787,
      "step": 93285
    },
    {
      "epoch": 1.839160919189453e-05,
      "step": 93285,
      "training_step_time": 7.478370666503906
    },
    {
      "epoch": 1.839256286621094e-05,
      "model_forward_time": 1.908651351928711,
      "step": 93286
    },
    {
      "epoch": 1.839256286621094e-05,
      "step": 93286,
      "training_step_time": 7.67496919631958
    },
    {
      "epoch": 1.8393516540527343e-05,
      "model_forward_time": 1.9144504070281982,
      "step": 93287
    },
    {
      "epoch": 1.8393516540527343e-05,
      "step": 93287,
      "training_step_time": 6.7481467723846436
    },
    {
      "epoch": 1.839447021484375e-05,
      "model_forward_time": 1.9160168170928955,
      "step": 93288
    },
    {
      "epoch": 1.839447021484375e-05,
      "step": 93288,
      "training_step_time": 6.7395405769348145
    },
    {
      "epoch": 1.8395423889160158e-05,
      "model_forward_time": 1.9187657833099365,
      "step": 93289
    },
    {
      "epoch": 1.8395423889160158e-05,
      "step": 93289,
      "training_step_time": 7.6238884925842285
    },
    {
      "epoch": 1.839637756347656e-05,
      "grad_norm": 0.09253787249326706,
      "learning_rate": 1.2262634580439025e-07,
      "loss": 0.053,
      "step": 93290
    },
    {
      "epoch": 1.839637756347656e-05,
      "model_forward_time": 1.9122183322906494,
      "step": 93290
    },
    {
      "dynamics_loss_avg": 0.06379101313650608,
      "epoch": 1.839637756347656e-05,
      "step": 93290
    },
    {
      "action_loss_avg": 0.001496061249054037,
      "epoch": 1.839637756347656e-05,
      "step": 93290
    },
    {
      "epoch": 1.839637756347656e-05,
      "step": 93290,
      "training_step_time": 9.428027153015137
    },
    {
      "epoch": 1.839733123779297e-05,
      "model_forward_time": 2.5357398986816406,
      "step": 93291
    },
    {
      "epoch": 1.839733123779297e-05,
      "step": 93291,
      "training_step_time": 7.812589406967163
    },
    {
      "epoch": 1.8398284912109377e-05,
      "model_forward_time": 1.9172616004943848,
      "step": 93292
    },
    {
      "epoch": 1.8398284912109377e-05,
      "step": 93292,
      "training_step_time": 7.57190728187561
    },
    {
      "epoch": 1.839923858642578e-05,
      "model_forward_time": 1.9159622192382812,
      "step": 93293
    },
    {
      "epoch": 1.839923858642578e-05,
      "step": 93293,
      "training_step_time": 8.034607887268066
    },
    {
      "epoch": 1.8400192260742188e-05,
      "model_forward_time": 1.9147448539733887,
      "step": 93294
    },
    {
      "epoch": 1.8400192260742188e-05,
      "step": 93294,
      "training_step_time": 7.500612497329712
    },
    {
      "epoch": 1.8401145935058595e-05,
      "model_forward_time": 1.9201991558074951,
      "step": 93295
    },
    {
      "epoch": 1.8401145935058595e-05,
      "step": 93295,
      "training_step_time": 7.214684724807739
    },
    {
      "epoch": 1.8402099609375e-05,
      "model_forward_time": 1.9230594635009766,
      "step": 93296
    },
    {
      "epoch": 1.8402099609375e-05,
      "step": 93296,
      "training_step_time": 7.706710577011108
    },
    {
      "epoch": 1.8403053283691407e-05,
      "model_forward_time": 1.9106814861297607,
      "step": 93297
    },
    {
      "epoch": 1.8403053283691407e-05,
      "step": 93297,
      "training_step_time": 6.755774021148682
    },
    {
      "epoch": 1.840400695800781e-05,
      "model_forward_time": 1.8785126209259033,
      "step": 93298
    },
    {
      "epoch": 1.840400695800781e-05,
      "step": 93298,
      "training_step_time": 7.646809816360474
    },
    {
      "epoch": 1.8404960632324218e-05,
      "model_forward_time": 1.9155888557434082,
      "step": 93299
    },
    {
      "epoch": 1.8404960632324218e-05,
      "step": 93299,
      "training_step_time": 9.811769962310791
    },
    {
      "epoch": 1.8405914306640626e-05,
      "grad_norm": 0.0951935425400734,
      "learning_rate": 1.2226266486652928e-07,
      "loss": 0.0524,
      "step": 93300
    },
    {
      "epoch": 1.8405914306640626e-05,
      "model_forward_time": 2.548475742340088,
      "step": 93300
    },
    {
      "dynamics_loss_avg": 0.048723075352609155,
      "epoch": 1.8405914306640626e-05,
      "step": 93300
    },
    {
      "action_loss_avg": 0.0014251097687520086,
      "epoch": 1.8405914306640626e-05,
      "step": 93300
    },
    {
      "epoch": 1.8405914306640626e-05,
      "step": 93300,
      "training_step_time": 8.96049976348877
    },
    {
      "epoch": 1.840686798095703e-05,
      "model_forward_time": 1.922537088394165,
      "step": 93301
    },
    {
      "epoch": 1.840686798095703e-05,
      "step": 93301,
      "training_step_time": 7.598461627960205
    },
    {
      "epoch": 1.8407821655273437e-05,
      "model_forward_time": 1.939958095550537,
      "step": 93302
    },
    {
      "epoch": 1.8407821655273437e-05,
      "step": 93302,
      "training_step_time": 7.557631731033325
    },
    {
      "epoch": 1.8408775329589845e-05,
      "model_forward_time": 1.9149909019470215,
      "step": 93303
    },
    {
      "epoch": 1.8408775329589845e-05,
      "step": 93303,
      "training_step_time": 7.4883129596710205
    },
    {
      "epoch": 1.840972900390625e-05,
      "model_forward_time": 1.9139635562896729,
      "step": 93304
    },
    {
      "epoch": 1.840972900390625e-05,
      "step": 93304,
      "training_step_time": 7.7620391845703125
    },
    {
      "epoch": 1.8410682678222656e-05,
      "model_forward_time": 1.925299882888794,
      "step": 93305
    },
    {
      "epoch": 1.8410682678222656e-05,
      "step": 93305,
      "training_step_time": 7.728316307067871
    },
    {
      "epoch": 1.8411636352539063e-05,
      "model_forward_time": 1.926090955734253,
      "step": 93306
    },
    {
      "epoch": 1.8411636352539063e-05,
      "step": 93306,
      "training_step_time": 6.685671806335449
    },
    {
      "epoch": 1.8412590026855467e-05,
      "model_forward_time": 1.9252817630767822,
      "step": 93307
    },
    {
      "epoch": 1.8412590026855467e-05,
      "step": 93307,
      "training_step_time": 7.62706184387207
    },
    {
      "epoch": 1.8413543701171875e-05,
      "model_forward_time": 1.9190263748168945,
      "step": 93308
    },
    {
      "epoch": 1.8413543701171875e-05,
      "step": 93308,
      "training_step_time": 9.596190214157104
    },
    {
      "epoch": 1.8414497375488282e-05,
      "model_forward_time": 2.5675976276397705,
      "step": 93309
    },
    {
      "epoch": 1.8414497375488282e-05,
      "step": 93309,
      "training_step_time": 11.052734375
    },
    {
      "epoch": 1.8415451049804686e-05,
      "grad_norm": 0.06838993728160858,
      "learning_rate": 1.218995173506893e-07,
      "loss": 0.0539,
      "step": 93310
    },
    {
      "epoch": 1.8415451049804686e-05,
      "model_forward_time": 1.9295876026153564,
      "step": 93310
    },
    {
      "dynamics_loss_avg": 0.06855232063680887,
      "epoch": 1.8415451049804686e-05,
      "step": 93310
    },
    {
      "action_loss_avg": 0.0015276396996341645,
      "epoch": 1.8415451049804686e-05,
      "step": 93310
    },
    {
      "epoch": 1.8415451049804686e-05,
      "step": 93310,
      "training_step_time": 7.695751428604126
    },
    {
      "epoch": 1.8416404724121094e-05,
      "model_forward_time": 1.9227302074432373,
      "step": 93311
    },
    {
      "epoch": 1.8416404724121094e-05,
      "step": 93311,
      "training_step_time": 7.55124306678772
    },
    {
      "epoch": 1.84173583984375e-05,
      "model_forward_time": 1.9311354160308838,
      "step": 93312
    },
    {
      "epoch": 1.84173583984375e-05,
      "step": 93312,
      "training_step_time": 7.565932989120483
    },
    {
      "epoch": 1.8418312072753905e-05,
      "model_forward_time": 1.9320456981658936,
      "step": 93313
    },
    {
      "epoch": 1.8418312072753905e-05,
      "step": 93313,
      "training_step_time": 7.073106527328491
    },
    {
      "epoch": 1.8419265747070313e-05,
      "model_forward_time": 1.9062976837158203,
      "step": 93314
    },
    {
      "epoch": 1.8419265747070313e-05,
      "step": 93314,
      "training_step_time": 8.232416868209839
    },
    {
      "epoch": 1.842021942138672e-05,
      "model_forward_time": 1.7786760330200195,
      "step": 93315
    },
    {
      "epoch": 1.842021942138672e-05,
      "step": 93315,
      "training_step_time": 6.772536993026733
    },
    {
      "epoch": 1.8421173095703124e-05,
      "model_forward_time": 1.1013104915618896,
      "step": 93316
    },
    {
      "epoch": 1.8421173095703124e-05,
      "step": 93316,
      "training_step_time": 7.6447062492370605
    },
    {
      "epoch": 1.842212677001953e-05,
      "model_forward_time": 1.7822699546813965,
      "step": 93317
    },
    {
      "epoch": 1.842212677001953e-05,
      "step": 93317,
      "training_step_time": 8.523485660552979
    },
    {
      "epoch": 1.842308044433594e-05,
      "model_forward_time": 1.8475971221923828,
      "step": 93318
    },
    {
      "epoch": 1.842308044433594e-05,
      "step": 93318,
      "training_step_time": 9.630680084228516
    },
    {
      "epoch": 1.8424034118652343e-05,
      "model_forward_time": 2.3889522552490234,
      "step": 93319
    },
    {
      "epoch": 1.8424034118652343e-05,
      "step": 93319,
      "training_step_time": 8.460190534591675
    },
    {
      "epoch": 1.842498779296875e-05,
      "grad_norm": 0.07547919452190399,
      "learning_rate": 1.2153690329658406e-07,
      "loss": 0.0527,
      "step": 93320
    },
    {
      "epoch": 1.842498779296875e-05,
      "model_forward_time": 1.7792365550994873,
      "step": 93320
    },
    {
      "dynamics_loss_avg": 0.05003626011312008,
      "epoch": 1.842498779296875e-05,
      "step": 93320
    },
    {
      "action_loss_avg": 0.0009807612921576947,
      "epoch": 1.842498779296875e-05,
      "step": 93320
    },
    {
      "epoch": 1.842498779296875e-05,
      "step": 93320,
      "training_step_time": 7.571112871170044
    },
    {
      "epoch": 1.8425941467285158e-05,
      "model_forward_time": 1.7829596996307373,
      "step": 93321
    },
    {
      "epoch": 1.8425941467285158e-05,
      "step": 93321,
      "training_step_time": 8.00208830833435
    },
    {
      "epoch": 1.8426895141601562e-05,
      "model_forward_time": 1.7814304828643799,
      "step": 93322
    },
    {
      "epoch": 1.8426895141601562e-05,
      "step": 93322,
      "training_step_time": 7.521230459213257
    },
    {
      "epoch": 1.842784881591797e-05,
      "model_forward_time": 1.7827472686767578,
      "step": 93323
    },
    {
      "epoch": 1.842784881591797e-05,
      "step": 93323,
      "training_step_time": 7.627998352050781
    },
    {
      "epoch": 1.8428802490234377e-05,
      "model_forward_time": 1.773496150970459,
      "step": 93324
    },
    {
      "epoch": 1.8428802490234377e-05,
      "step": 93324,
      "training_step_time": 7.649315357208252
    },
    {
      "epoch": 1.842975616455078e-05,
      "model_forward_time": 1.7784829139709473,
      "step": 93325
    },
    {
      "epoch": 1.842975616455078e-05,
      "step": 93325,
      "training_step_time": 8.62983250617981
    },
    {
      "epoch": 1.8430709838867188e-05,
      "model_forward_time": 1.7788562774658203,
      "step": 93326
    },
    {
      "epoch": 1.8430709838867188e-05,
      "step": 93326,
      "training_step_time": 9.408771991729736
    },
    {
      "epoch": 1.8431663513183595e-05,
      "model_forward_time": 1.8214654922485352,
      "step": 93327
    },
    {
      "epoch": 1.8431663513183595e-05,
      "step": 93327,
      "training_step_time": 10.00076961517334
    },
    {
      "epoch": 1.84326171875e-05,
      "model_forward_time": 2.4071929454803467,
      "step": 93328
    },
    {
      "epoch": 1.84326171875e-05,
      "step": 93328,
      "training_step_time": 7.753025531768799
    },
    {
      "epoch": 1.8433570861816407e-05,
      "model_forward_time": 1.7861862182617188,
      "step": 93329
    },
    {
      "epoch": 1.8433570861816407e-05,
      "step": 93329,
      "training_step_time": 7.499795436859131
    },
    {
      "epoch": 1.843452453613281e-05,
      "grad_norm": 0.07433469593524933,
      "learning_rate": 1.211748227438686e-07,
      "loss": 0.0537,
      "step": 93330
    },
    {
      "epoch": 1.843452453613281e-05,
      "model_forward_time": 1.7967681884765625,
      "step": 93330
    },
    {
      "dynamics_loss_avg": 0.057070108503103255,
      "epoch": 1.843452453613281e-05,
      "step": 93330
    },
    {
      "action_loss_avg": 0.00131670611735899,
      "epoch": 1.843452453613281e-05,
      "step": 93330
    },
    {
      "epoch": 1.843452453613281e-05,
      "step": 93330,
      "training_step_time": 7.406975746154785
    },
    {
      "epoch": 1.8435478210449218e-05,
      "model_forward_time": 1.7835538387298584,
      "step": 93331
    },
    {
      "epoch": 1.8435478210449218e-05,
      "step": 93331,
      "training_step_time": 7.449602842330933
    },
    {
      "epoch": 1.8436431884765626e-05,
      "model_forward_time": 1.7874608039855957,
      "step": 93332
    },
    {
      "epoch": 1.8436431884765626e-05,
      "step": 93332,
      "training_step_time": 7.619605302810669
    },
    {
      "epoch": 1.843738555908203e-05,
      "model_forward_time": 1.7798986434936523,
      "step": 93333
    },
    {
      "epoch": 1.843738555908203e-05,
      "step": 93333,
      "training_step_time": 7.630077600479126
    },
    {
      "epoch": 1.8438339233398437e-05,
      "model_forward_time": 1.7935729026794434,
      "step": 93334
    },
    {
      "epoch": 1.8438339233398437e-05,
      "step": 93334,
      "training_step_time": 7.622790575027466
    },
    {
      "epoch": 1.8439292907714845e-05,
      "model_forward_time": 1.8408713340759277,
      "step": 93335
    },
    {
      "epoch": 1.8439292907714845e-05,
      "step": 93335,
      "training_step_time": 8.533770084381104
    },
    {
      "epoch": 1.844024658203125e-05,
      "model_forward_time": 1.776435136795044,
      "step": 93336
    },
    {
      "epoch": 1.844024658203125e-05,
      "step": 93336,
      "training_step_time": 9.681153059005737
    },
    {
      "epoch": 1.8441200256347656e-05,
      "model_forward_time": 2.458019733428955,
      "step": 93337
    },
    {
      "epoch": 1.8441200256347656e-05,
      "step": 93337,
      "training_step_time": 7.7206361293792725
    },
    {
      "epoch": 1.8442153930664063e-05,
      "model_forward_time": 1.7866106033325195,
      "step": 93338
    },
    {
      "epoch": 1.8442153930664063e-05,
      "step": 93338,
      "training_step_time": 7.736349821090698
    },
    {
      "epoch": 1.8443107604980467e-05,
      "model_forward_time": 1.787017583847046,
      "step": 93339
    },
    {
      "epoch": 1.8443107604980467e-05,
      "step": 93339,
      "training_step_time": 7.442233085632324
    },
    {
      "epoch": 1.8444061279296875e-05,
      "grad_norm": 0.07758113741874695,
      "learning_rate": 1.2081327573213952e-07,
      "loss": 0.0534,
      "step": 93340
    },
    {
      "epoch": 1.8444061279296875e-05,
      "model_forward_time": 1.790173053741455,
      "step": 93340
    },
    {
      "dynamics_loss_avg": 0.04869566448032856,
      "epoch": 1.8444061279296875e-05,
      "step": 93340
    },
    {
      "action_loss_avg": 0.001649256981909275,
      "epoch": 1.8444061279296875e-05,
      "step": 93340
    },
    {
      "epoch": 1.8444061279296875e-05,
      "step": 93340,
      "training_step_time": 7.454464673995972
    },
    {
      "epoch": 1.8445014953613282e-05,
      "model_forward_time": 1.783123254776001,
      "step": 93341
    },
    {
      "epoch": 1.8445014953613282e-05,
      "step": 93341,
      "training_step_time": 7.199279546737671
    },
    {
      "epoch": 1.8445968627929686e-05,
      "model_forward_time": 1.7847449779510498,
      "step": 93342
    },
    {
      "epoch": 1.8445968627929686e-05,
      "step": 93342,
      "training_step_time": 7.650861740112305
    },
    {
      "epoch": 1.8446922302246094e-05,
      "model_forward_time": 1.8426449298858643,
      "step": 93343
    },
    {
      "epoch": 1.8446922302246094e-05,
      "step": 93343,
      "training_step_time": 7.5917439460754395
    },
    {
      "epoch": 1.84478759765625e-05,
      "model_forward_time": 1.7829070091247559,
      "step": 93344
    },
    {
      "epoch": 1.84478759765625e-05,
      "step": 93344,
      "training_step_time": 8.648939371109009
    },
    {
      "epoch": 1.8448829650878905e-05,
      "model_forward_time": 1.7748332023620605,
      "step": 93345
    },
    {
      "epoch": 1.8448829650878905e-05,
      "step": 93345,
      "training_step_time": 8.269269704818726
    },
    {
      "epoch": 1.8449783325195313e-05,
      "model_forward_time": 2.518707752227783,
      "step": 93346
    },
    {
      "epoch": 1.8449783325195313e-05,
      "step": 93346,
      "training_step_time": 8.815590858459473
    },
    {
      "epoch": 1.845073699951172e-05,
      "model_forward_time": 1.7807152271270752,
      "step": 93347
    },
    {
      "epoch": 1.845073699951172e-05,
      "step": 93347,
      "training_step_time": 7.619344711303711
    },
    {
      "epoch": 1.8451690673828124e-05,
      "model_forward_time": 1.7732059955596924,
      "step": 93348
    },
    {
      "epoch": 1.8451690673828124e-05,
      "step": 93348,
      "training_step_time": 7.442296028137207
    },
    {
      "epoch": 1.845264434814453e-05,
      "model_forward_time": 1.779221534729004,
      "step": 93349
    },
    {
      "epoch": 1.845264434814453e-05,
      "step": 93349,
      "training_step_time": 7.5636162757873535
    },
    {
      "epoch": 1.845359802246094e-05,
      "grad_norm": 0.08583754301071167,
      "learning_rate": 1.2045226230093466e-07,
      "loss": 0.0542,
      "step": 93350
    },
    {
      "epoch": 1.845359802246094e-05,
      "model_forward_time": 1.7822439670562744,
      "step": 93350
    },
    {
      "dynamics_loss_avg": 0.04365562424063683,
      "epoch": 1.845359802246094e-05,
      "step": 93350
    },
    {
      "action_loss_avg": 0.001479762519011274,
      "epoch": 1.845359802246094e-05,
      "step": 93350
    },
    {
      "epoch": 1.845359802246094e-05,
      "step": 93350,
      "training_step_time": 7.25769567489624
    },
    {
      "epoch": 1.8454551696777343e-05,
      "model_forward_time": 1.7760858535766602,
      "step": 93351
    },
    {
      "epoch": 1.8454551696777343e-05,
      "step": 93351,
      "training_step_time": 7.585303544998169
    },
    {
      "epoch": 1.845550537109375e-05,
      "model_forward_time": 1.7812268733978271,
      "step": 93352
    },
    {
      "epoch": 1.845550537109375e-05,
      "step": 93352,
      "training_step_time": 7.615272760391235
    },
    {
      "epoch": 1.8456459045410158e-05,
      "model_forward_time": 1.7781541347503662,
      "step": 93353
    },
    {
      "epoch": 1.8456459045410158e-05,
      "step": 93353,
      "training_step_time": 8.593431234359741
    },
    {
      "epoch": 1.8457412719726562e-05,
      "model_forward_time": 1.772456169128418,
      "step": 93354
    },
    {
      "epoch": 1.8457412719726562e-05,
      "step": 93354,
      "training_step_time": 9.451999425888062
    },
    {
      "epoch": 1.845836639404297e-05,
      "model_forward_time": 1.0890061855316162,
      "step": 93355
    },
    {
      "epoch": 1.845836639404297e-05,
      "step": 93355,
      "training_step_time": 9.814171552658081
    },
    {
      "epoch": 1.8459320068359377e-05,
      "model_forward_time": 2.395045518875122,
      "step": 93356
    },
    {
      "epoch": 1.8459320068359377e-05,
      "step": 93356,
      "training_step_time": 7.703503370285034
    },
    {
      "epoch": 1.846027374267578e-05,
      "model_forward_time": 1.7787792682647705,
      "step": 93357
    },
    {
      "epoch": 1.846027374267578e-05,
      "step": 93357,
      "training_step_time": 7.522491455078125
    },
    {
      "epoch": 1.8461227416992188e-05,
      "model_forward_time": 1.7788841724395752,
      "step": 93358
    },
    {
      "epoch": 1.8461227416992188e-05,
      "step": 93358,
      "training_step_time": 7.485379219055176
    },
    {
      "epoch": 1.8462181091308595e-05,
      "model_forward_time": 1.7919156551361084,
      "step": 93359
    },
    {
      "epoch": 1.8462181091308595e-05,
      "step": 93359,
      "training_step_time": 7.31109094619751
    },
    {
      "epoch": 1.8463134765625e-05,
      "grad_norm": 0.09867982566356659,
      "learning_rate": 1.2009178248973297e-07,
      "loss": 0.0546,
      "step": 93360
    },
    {
      "epoch": 1.8463134765625e-05,
      "model_forward_time": 1.7767972946166992,
      "step": 93360
    },
    {
      "dynamics_loss_avg": 0.05351088289171457,
      "epoch": 1.8463134765625e-05,
      "step": 93360
    },
    {
      "action_loss_avg": 0.0017473869782406836,
      "epoch": 1.8463134765625e-05,
      "step": 93360
    },
    {
      "epoch": 1.8463134765625e-05,
      "step": 93360,
      "training_step_time": 7.65838885307312
    },
    {
      "epoch": 1.8464088439941407e-05,
      "model_forward_time": 1.7814717292785645,
      "step": 93361
    },
    {
      "epoch": 1.8464088439941407e-05,
      "step": 93361,
      "training_step_time": 7.269911050796509
    },
    {
      "epoch": 1.846504211425781e-05,
      "model_forward_time": 1.7752552032470703,
      "step": 93362
    },
    {
      "epoch": 1.846504211425781e-05,
      "step": 93362,
      "training_step_time": 7.666447639465332
    },
    {
      "epoch": 1.846599578857422e-05,
      "model_forward_time": 1.7673957347869873,
      "step": 93363
    },
    {
      "epoch": 1.846599578857422e-05,
      "step": 93363,
      "training_step_time": 7.605324983596802
    },
    {
      "epoch": 1.8466949462890626e-05,
      "model_forward_time": 1.79325532913208,
      "step": 93364
    },
    {
      "epoch": 1.8466949462890626e-05,
      "step": 93364,
      "training_step_time": 8.820773363113403
    },
    {
      "epoch": 1.846790313720703e-05,
      "model_forward_time": 2.4203202724456787,
      "step": 93365
    },
    {
      "epoch": 1.846790313720703e-05,
      "step": 93365,
      "training_step_time": 7.670032978057861
    },
    {
      "epoch": 1.8468856811523437e-05,
      "model_forward_time": 1.7997429370880127,
      "step": 93366
    },
    {
      "epoch": 1.8468856811523437e-05,
      "step": 93366,
      "training_step_time": 7.549985647201538
    },
    {
      "epoch": 1.8469810485839845e-05,
      "model_forward_time": 1.78806471824646,
      "step": 93367
    },
    {
      "epoch": 1.8469810485839845e-05,
      "step": 93367,
      "training_step_time": 7.481724262237549
    },
    {
      "epoch": 1.847076416015625e-05,
      "model_forward_time": 1.7926506996154785,
      "step": 93368
    },
    {
      "epoch": 1.847076416015625e-05,
      "step": 93368,
      "training_step_time": 7.591591119766235
    },
    {
      "epoch": 1.8471717834472656e-05,
      "model_forward_time": 1.7919974327087402,
      "step": 93369
    },
    {
      "epoch": 1.8471717834472656e-05,
      "step": 93369,
      "training_step_time": 7.659862279891968
    },
    {
      "epoch": 1.8472671508789063e-05,
      "grad_norm": 0.0847613736987114,
      "learning_rate": 1.1973183633795849e-07,
      "loss": 0.0528,
      "step": 93370
    },
    {
      "epoch": 1.8472671508789063e-05,
      "model_forward_time": 1.7975389957427979,
      "step": 93370
    },
    {
      "dynamics_loss_avg": 0.0598759863525629,
      "epoch": 1.8472671508789063e-05,
      "step": 93370
    },
    {
      "action_loss_avg": 0.0014894767402438447,
      "epoch": 1.8472671508789063e-05,
      "step": 93370
    },
    {
      "epoch": 1.8472671508789063e-05,
      "step": 93370,
      "training_step_time": 7.293852090835571
    },
    {
      "epoch": 1.8473625183105467e-05,
      "model_forward_time": 1.7960257530212402,
      "step": 93371
    },
    {
      "epoch": 1.8473625183105467e-05,
      "step": 93371,
      "training_step_time": 8.500813961029053
    },
    {
      "epoch": 1.8474578857421875e-05,
      "model_forward_time": 1.4285047054290771,
      "step": 93372
    },
    {
      "epoch": 1.8474578857421875e-05,
      "step": 93372,
      "training_step_time": 9.240038871765137
    },
    {
      "epoch": 1.8475532531738282e-05,
      "model_forward_time": 1.7863507270812988,
      "step": 93373
    },
    {
      "epoch": 1.8475532531738282e-05,
      "step": 93373,
      "training_step_time": 9.304966688156128
    },
    {
      "epoch": 1.8476486206054686e-05,
      "model_forward_time": 2.9060866832733154,
      "step": 93374
    },
    {
      "epoch": 1.8476486206054686e-05,
      "step": 93374,
      "training_step_time": 7.748889446258545
    },
    {
      "epoch": 1.8477439880371094e-05,
      "model_forward_time": 1.7318501472473145,
      "step": 93375
    },
    {
      "epoch": 1.8477439880371094e-05,
      "step": 93375,
      "training_step_time": 8.808820486068726
    },
    {
      "epoch": 1.84783935546875e-05,
      "model_forward_time": 1.7379209995269775,
      "step": 93376
    },
    {
      "epoch": 1.84783935546875e-05,
      "step": 93376,
      "training_step_time": 7.449514627456665
    },
    {
      "epoch": 1.8479347229003905e-05,
      "model_forward_time": 1.7341644763946533,
      "step": 93377
    },
    {
      "epoch": 1.8479347229003905e-05,
      "step": 93377,
      "training_step_time": 7.596294403076172
    },
    {
      "epoch": 1.8480300903320313e-05,
      "model_forward_time": 1.7324261665344238,
      "step": 93378
    },
    {
      "epoch": 1.8480300903320313e-05,
      "step": 93378,
      "training_step_time": 7.635503768920898
    },
    {
      "epoch": 1.848125457763672e-05,
      "model_forward_time": 1.7305068969726562,
      "step": 93379
    },
    {
      "epoch": 1.848125457763672e-05,
      "step": 93379,
      "training_step_time": 7.689975261688232
    },
    {
      "epoch": 1.8482208251953124e-05,
      "grad_norm": 0.10565485805273056,
      "learning_rate": 1.193724238849725e-07,
      "loss": 0.055,
      "step": 93380
    },
    {
      "epoch": 1.8482208251953124e-05,
      "model_forward_time": 1.7355895042419434,
      "step": 93380
    },
    {
      "dynamics_loss_avg": 0.058802036382257936,
      "epoch": 1.8482208251953124e-05,
      "step": 93380
    },
    {
      "action_loss_avg": 0.0015438337111845612,
      "epoch": 1.8482208251953124e-05,
      "step": 93380
    },
    {
      "epoch": 1.8482208251953124e-05,
      "step": 93380,
      "training_step_time": 9.645856857299805
    },
    {
      "epoch": 1.848316192626953e-05,
      "model_forward_time": 1.727055311203003,
      "step": 93381
    },
    {
      "epoch": 1.848316192626953e-05,
      "step": 93381,
      "training_step_time": 7.637479305267334
    },
    {
      "epoch": 1.848411560058594e-05,
      "model_forward_time": 1.7827837467193604,
      "step": 93382
    },
    {
      "epoch": 1.848411560058594e-05,
      "step": 93382,
      "training_step_time": 8.631805658340454
    },
    {
      "epoch": 1.8485069274902343e-05,
      "model_forward_time": 2.7986464500427246,
      "step": 93383
    },
    {
      "epoch": 1.8485069274902343e-05,
      "step": 93383,
      "training_step_time": 9.966751337051392
    },
    {
      "epoch": 1.848602294921875e-05,
      "model_forward_time": 1.731109380722046,
      "step": 93384
    },
    {
      "epoch": 1.848602294921875e-05,
      "step": 93384,
      "training_step_time": 8.430031538009644
    },
    {
      "epoch": 1.8486976623535158e-05,
      "model_forward_time": 1.720332145690918,
      "step": 93385
    },
    {
      "epoch": 1.8486976623535158e-05,
      "step": 93385,
      "training_step_time": 7.407023906707764
    },
    {
      "epoch": 1.8487930297851562e-05,
      "model_forward_time": 1.725771188735962,
      "step": 93386
    },
    {
      "epoch": 1.8487930297851562e-05,
      "step": 93386,
      "training_step_time": 7.535201072692871
    },
    {
      "epoch": 1.848888397216797e-05,
      "model_forward_time": 1.7369117736816406,
      "step": 93387
    },
    {
      "epoch": 1.848888397216797e-05,
      "step": 93387,
      "training_step_time": 7.186872243881226
    },
    {
      "epoch": 1.8489837646484377e-05,
      "model_forward_time": 1.7240409851074219,
      "step": 93388
    },
    {
      "epoch": 1.8489837646484377e-05,
      "step": 93388,
      "training_step_time": 7.55047345161438
    },
    {
      "epoch": 1.849079132080078e-05,
      "model_forward_time": 1.7197997570037842,
      "step": 93389
    },
    {
      "epoch": 1.849079132080078e-05,
      "step": 93389,
      "training_step_time": 7.580318450927734
    },
    {
      "epoch": 1.8491744995117188e-05,
      "grad_norm": 0.07955215871334076,
      "learning_rate": 1.1901354517008079e-07,
      "loss": 0.0564,
      "step": 93390
    },
    {
      "epoch": 1.8491744995117188e-05,
      "model_forward_time": 1.728647232055664,
      "step": 93390
    },
    {
      "dynamics_loss_avg": 0.05863770041614771,
      "epoch": 1.8491744995117188e-05,
      "step": 93390
    },
    {
      "action_loss_avg": 0.0011859748876304366,
      "epoch": 1.8491744995117188e-05,
      "step": 93390
    },
    {
      "epoch": 1.8491744995117188e-05,
      "step": 93390,
      "training_step_time": 7.728445768356323
    },
    {
      "epoch": 1.8492698669433592e-05,
      "model_forward_time": 1.7686965465545654,
      "step": 93391
    },
    {
      "epoch": 1.8492698669433592e-05,
      "step": 93391,
      "training_step_time": 8.660361289978027
    },
    {
      "epoch": 1.849365234375e-05,
      "model_forward_time": 1.723555564880371,
      "step": 93392
    },
    {
      "epoch": 1.849365234375e-05,
      "step": 93392,
      "training_step_time": 9.931672096252441
    },
    {
      "epoch": 1.8494606018066407e-05,
      "model_forward_time": 1.716398000717163,
      "step": 93393
    },
    {
      "epoch": 1.8494606018066407e-05,
      "step": 93393,
      "training_step_time": 7.5686469078063965
    },
    {
      "epoch": 1.849555969238281e-05,
      "model_forward_time": 1.7247960567474365,
      "step": 93394
    },
    {
      "epoch": 1.849555969238281e-05,
      "step": 93394,
      "training_step_time": 8.331636905670166
    },
    {
      "epoch": 1.849651336669922e-05,
      "model_forward_time": 1.3460323810577393,
      "step": 93395
    },
    {
      "epoch": 1.849651336669922e-05,
      "step": 93395,
      "training_step_time": 7.511408805847168
    },
    {
      "epoch": 1.8497467041015626e-05,
      "model_forward_time": 1.7366042137145996,
      "step": 93396
    },
    {
      "epoch": 1.8497467041015626e-05,
      "step": 93396,
      "training_step_time": 7.880171537399292
    },
    {
      "epoch": 1.849842071533203e-05,
      "model_forward_time": 1.7256321907043457,
      "step": 93397
    },
    {
      "epoch": 1.849842071533203e-05,
      "step": 93397,
      "training_step_time": 7.621988296508789
    },
    {
      "epoch": 1.8499374389648437e-05,
      "model_forward_time": 1.7280476093292236,
      "step": 93398
    },
    {
      "epoch": 1.8499374389648437e-05,
      "step": 93398,
      "training_step_time": 7.495562314987183
    },
    {
      "epoch": 1.8500328063964845e-05,
      "model_forward_time": 1.7313449382781982,
      "step": 93399
    },
    {
      "epoch": 1.8500328063964845e-05,
      "step": 93399,
      "training_step_time": 9.33424186706543
    },
    {
      "epoch": 1.850128173828125e-05,
      "grad_norm": 0.14461494982242584,
      "learning_rate": 1.1865520023252919e-07,
      "loss": 0.0558,
      "step": 93400
    },
    {
      "epoch": 1.850128173828125e-05,
      "model_forward_time": 1.7593731880187988,
      "step": 93400
    },
    {
      "dynamics_loss_avg": 0.06423405762761832,
      "epoch": 1.850128173828125e-05,
      "step": 93400
    },
    {
      "action_loss_avg": 0.0015511674544541165,
      "epoch": 1.850128173828125e-05,
      "step": 93400
    },
    {
      "epoch": 1.850128173828125e-05,
      "step": 93400,
      "training_step_time": 8.487751722335815
    },
    {
      "epoch": 1.8502235412597656e-05,
      "model_forward_time": 1.7457199096679688,
      "step": 93401
    },
    {
      "epoch": 1.8502235412597656e-05,
      "step": 93401,
      "training_step_time": 11.233027219772339
    },
    {
      "epoch": 1.8503189086914063e-05,
      "model_forward_time": 2.8822460174560547,
      "step": 93402
    },
    {
      "epoch": 1.8503189086914063e-05,
      "step": 93402,
      "training_step_time": 7.685492038726807
    },
    {
      "epoch": 1.8504142761230468e-05,
      "model_forward_time": 1.7496941089630127,
      "step": 93403
    },
    {
      "epoch": 1.8504142761230468e-05,
      "step": 93403,
      "training_step_time": 7.649494409561157
    },
    {
      "epoch": 1.8505096435546875e-05,
      "model_forward_time": 1.738290548324585,
      "step": 93404
    },
    {
      "epoch": 1.8505096435546875e-05,
      "step": 93404,
      "training_step_time": 7.469456195831299
    },
    {
      "epoch": 1.8506050109863282e-05,
      "model_forward_time": 1.7347805500030518,
      "step": 93405
    },
    {
      "epoch": 1.8506050109863282e-05,
      "step": 93405,
      "training_step_time": 7.11018705368042
    },
    {
      "epoch": 1.8507003784179686e-05,
      "model_forward_time": 1.7265748977661133,
      "step": 93406
    },
    {
      "epoch": 1.8507003784179686e-05,
      "step": 93406,
      "training_step_time": 7.227254390716553
    },
    {
      "epoch": 1.8507957458496094e-05,
      "model_forward_time": 1.7455742359161377,
      "step": 93407
    },
    {
      "epoch": 1.8507957458496094e-05,
      "step": 93407,
      "training_step_time": 7.659343957901001
    },
    {
      "epoch": 1.85089111328125e-05,
      "model_forward_time": 1.7254772186279297,
      "step": 93408
    },
    {
      "epoch": 1.85089111328125e-05,
      "step": 93408,
      "training_step_time": 7.628059148788452
    },
    {
      "epoch": 1.8509864807128905e-05,
      "model_forward_time": 1.7448012828826904,
      "step": 93409
    },
    {
      "epoch": 1.8509864807128905e-05,
      "step": 93409,
      "training_step_time": 8.139868259429932
    },
    {
      "epoch": 1.8510818481445313e-05,
      "grad_norm": 0.15274101495742798,
      "learning_rate": 1.1829738911150578e-07,
      "loss": 0.0557,
      "step": 93410
    },
    {
      "epoch": 1.8510818481445313e-05,
      "model_forward_time": 1.3724870681762695,
      "step": 93410
    },
    {
      "dynamics_loss_avg": 0.05830953009426594,
      "epoch": 1.8510818481445313e-05,
      "step": 93410
    },
    {
      "action_loss_avg": 0.0012885005911812187,
      "epoch": 1.8510818481445313e-05,
      "step": 93410
    },
    {
      "epoch": 1.8510818481445313e-05,
      "step": 93410,
      "training_step_time": 9.576972961425781
    },
    {
      "epoch": 1.851177215576172e-05,
      "model_forward_time": 2.8398571014404297,
      "step": 93411
    },
    {
      "epoch": 1.851177215576172e-05,
      "step": 93411,
      "training_step_time": 8.564680337905884
    },
    {
      "epoch": 1.8512725830078124e-05,
      "model_forward_time": 1.753014087677002,
      "step": 93412
    },
    {
      "epoch": 1.8512725830078124e-05,
      "step": 93412,
      "training_step_time": 7.635213136672974
    },
    {
      "epoch": 1.851367950439453e-05,
      "model_forward_time": 1.7374849319458008,
      "step": 93413
    },
    {
      "epoch": 1.851367950439453e-05,
      "step": 93413,
      "training_step_time": 7.358579397201538
    },
    {
      "epoch": 1.851463317871094e-05,
      "model_forward_time": 1.740004539489746,
      "step": 93414
    },
    {
      "epoch": 1.851463317871094e-05,
      "step": 93414,
      "training_step_time": 7.524415731430054
    },
    {
      "epoch": 1.8515586853027343e-05,
      "model_forward_time": 1.7365374565124512,
      "step": 93415
    },
    {
      "epoch": 1.8515586853027343e-05,
      "step": 93415,
      "training_step_time": 7.252517461776733
    },
    {
      "epoch": 1.851654052734375e-05,
      "model_forward_time": 1.7451672554016113,
      "step": 93416
    },
    {
      "epoch": 1.851654052734375e-05,
      "step": 93416,
      "training_step_time": 7.609850883483887
    },
    {
      "epoch": 1.8517494201660158e-05,
      "model_forward_time": 1.7355880737304688,
      "step": 93417
    },
    {
      "epoch": 1.8517494201660158e-05,
      "step": 93417,
      "training_step_time": 7.651871681213379
    },
    {
      "epoch": 1.8518447875976562e-05,
      "model_forward_time": 1.7460293769836426,
      "step": 93418
    },
    {
      "epoch": 1.8518447875976562e-05,
      "step": 93418,
      "training_step_time": 9.251770496368408
    },
    {
      "epoch": 1.851940155029297e-05,
      "model_forward_time": 1.757779836654663,
      "step": 93419
    },
    {
      "epoch": 1.851940155029297e-05,
      "step": 93419,
      "training_step_time": 10.775062799453735
    },
    {
      "epoch": 1.8520355224609377e-05,
      "grad_norm": 0.10458432137966156,
      "learning_rate": 1.1794011184613985e-07,
      "loss": 0.0567,
      "step": 93420
    },
    {
      "epoch": 1.8520355224609377e-05,
      "model_forward_time": 2.845412254333496,
      "step": 93420
    },
    {
      "dynamics_loss_avg": 0.053355083987116816,
      "epoch": 1.8520355224609377e-05,
      "step": 93420
    },
    {
      "action_loss_avg": 0.0014604535477701574,
      "epoch": 1.8520355224609377e-05,
      "step": 93420
    },
    {
      "epoch": 1.8520355224609377e-05,
      "step": 93420,
      "training_step_time": 9.341919183731079
    },
    {
      "epoch": 1.852130889892578e-05,
      "model_forward_time": 1.7342004776000977,
      "step": 93421
    },
    {
      "epoch": 1.852130889892578e-05,
      "step": 93421,
      "training_step_time": 7.576937198638916
    },
    {
      "epoch": 1.8522262573242188e-05,
      "model_forward_time": 1.7342920303344727,
      "step": 93422
    },
    {
      "epoch": 1.8522262573242188e-05,
      "step": 93422,
      "training_step_time": 7.357218027114868
    },
    {
      "epoch": 1.8523216247558592e-05,
      "model_forward_time": 1.7412514686584473,
      "step": 93423
    },
    {
      "epoch": 1.8523216247558592e-05,
      "step": 93423,
      "training_step_time": 7.525616645812988
    },
    {
      "epoch": 1.8524169921875e-05,
      "model_forward_time": 1.7259509563446045,
      "step": 93424
    },
    {
      "epoch": 1.8524169921875e-05,
      "step": 93424,
      "training_step_time": 7.237350940704346
    },
    {
      "epoch": 1.8525123596191407e-05,
      "model_forward_time": 1.7450103759765625,
      "step": 93425
    },
    {
      "epoch": 1.8525123596191407e-05,
      "step": 93425,
      "training_step_time": 7.614338397979736
    },
    {
      "epoch": 1.852607727050781e-05,
      "model_forward_time": 1.7497782707214355,
      "step": 93426
    },
    {
      "epoch": 1.852607727050781e-05,
      "step": 93426,
      "training_step_time": 7.711158514022827
    },
    {
      "epoch": 1.852703094482422e-05,
      "model_forward_time": 1.7406694889068604,
      "step": 93427
    },
    {
      "epoch": 1.852703094482422e-05,
      "step": 93427,
      "training_step_time": 8.521246194839478
    },
    {
      "epoch": 1.8527984619140626e-05,
      "model_forward_time": 1.7718122005462646,
      "step": 93428
    },
    {
      "epoch": 1.8527984619140626e-05,
      "step": 93428,
      "training_step_time": 7.684367418289185
    },
    {
      "epoch": 1.852893829345703e-05,
      "model_forward_time": 1.7467257976531982,
      "step": 93429
    },
    {
      "epoch": 1.852893829345703e-05,
      "step": 93429,
      "training_step_time": 10.027647256851196
    },
    {
      "epoch": 1.8529891967773437e-05,
      "grad_norm": 0.1691088080406189,
      "learning_rate": 1.1758336847550401e-07,
      "loss": 0.0585,
      "step": 93430
    },
    {
      "epoch": 1.8529891967773437e-05,
      "model_forward_time": 1.7466192245483398,
      "step": 93430
    },
    {
      "dynamics_loss_avg": 0.052010877057909966,
      "epoch": 1.8529891967773437e-05,
      "step": 93430
    },
    {
      "action_loss_avg": 0.00128528565983288,
      "epoch": 1.8529891967773437e-05,
      "step": 93430
    },
    {
      "epoch": 1.8529891967773437e-05,
      "step": 93430,
      "training_step_time": 9.496190786361694
    },
    {
      "epoch": 1.8530845642089845e-05,
      "model_forward_time": 1.749021053314209,
      "step": 93431
    },
    {
      "epoch": 1.8530845642089845e-05,
      "step": 93431,
      "training_step_time": 7.480276584625244
    },
    {
      "epoch": 1.853179931640625e-05,
      "model_forward_time": 1.784602165222168,
      "step": 93432
    },
    {
      "epoch": 1.853179931640625e-05,
      "step": 93432,
      "training_step_time": 8.095552921295166
    },
    {
      "epoch": 1.8532752990722656e-05,
      "model_forward_time": 1.9282519817352295,
      "step": 93433
    },
    {
      "epoch": 1.8532752990722656e-05,
      "step": 93433,
      "training_step_time": 7.31919527053833
    },
    {
      "epoch": 1.8533706665039064e-05,
      "model_forward_time": 1.893547773361206,
      "step": 93434
    },
    {
      "epoch": 1.8533706665039064e-05,
      "step": 93434,
      "training_step_time": 7.296831369400024
    },
    {
      "epoch": 1.8534660339355468e-05,
      "model_forward_time": 1.5649607181549072,
      "step": 93435
    },
    {
      "epoch": 1.8534660339355468e-05,
      "step": 93435,
      "training_step_time": 8.79262113571167
    },
    {
      "epoch": 1.8535614013671875e-05,
      "model_forward_time": 1.8631491661071777,
      "step": 93436
    },
    {
      "epoch": 1.8535614013671875e-05,
      "step": 93436,
      "training_step_time": 8.661845445632935
    },
    {
      "epoch": 1.8536567687988282e-05,
      "model_forward_time": 1.8983838558197021,
      "step": 93437
    },
    {
      "epoch": 1.8536567687988282e-05,
      "step": 93437,
      "training_step_time": 9.651424407958984
    },
    {
      "epoch": 1.8537521362304686e-05,
      "model_forward_time": 1.5355587005615234,
      "step": 93438
    },
    {
      "epoch": 1.8537521362304686e-05,
      "step": 93438,
      "training_step_time": 8.560133934020996
    },
    {
      "epoch": 1.8538475036621094e-05,
      "model_forward_time": 2.464390277862549,
      "step": 93439
    },
    {
      "epoch": 1.8538475036621094e-05,
      "step": 93439,
      "training_step_time": 7.758303165435791
    },
    {
      "epoch": 1.85394287109375e-05,
      "grad_norm": 0.08460532873868942,
      "learning_rate": 1.1722715903860982e-07,
      "loss": 0.0549,
      "step": 93440
    },
    {
      "epoch": 1.85394287109375e-05,
      "model_forward_time": 1.900012493133545,
      "step": 93440
    },
    {
      "dynamics_loss_avg": 0.03989819251000881,
      "epoch": 1.85394287109375e-05,
      "step": 93440
    },
    {
      "action_loss_avg": 0.0011410874867578968,
      "epoch": 1.85394287109375e-05,
      "step": 93440
    },
    {
      "epoch": 1.85394287109375e-05,
      "step": 93440,
      "training_step_time": 7.84958815574646
    },
    {
      "epoch": 1.8540382385253905e-05,
      "model_forward_time": 1.8986554145812988,
      "step": 93441
    },
    {
      "epoch": 1.8540382385253905e-05,
      "step": 93441,
      "training_step_time": 7.73227071762085
    },
    {
      "epoch": 1.8541336059570313e-05,
      "model_forward_time": 1.9172756671905518,
      "step": 93442
    },
    {
      "epoch": 1.8541336059570313e-05,
      "step": 93442,
      "training_step_time": 7.357199668884277
    },
    {
      "epoch": 1.854228973388672e-05,
      "model_forward_time": 1.9104375839233398,
      "step": 93443
    },
    {
      "epoch": 1.854228973388672e-05,
      "step": 93443,
      "training_step_time": 7.401146411895752
    },
    {
      "epoch": 1.8543243408203124e-05,
      "model_forward_time": 1.8785159587860107,
      "step": 93444
    },
    {
      "epoch": 1.8543243408203124e-05,
      "step": 93444,
      "training_step_time": 7.795201063156128
    },
    {
      "epoch": 1.854419708251953e-05,
      "model_forward_time": 1.9220421314239502,
      "step": 93445
    },
    {
      "epoch": 1.854419708251953e-05,
      "step": 93445,
      "training_step_time": 8.670157194137573
    },
    {
      "epoch": 1.854515075683594e-05,
      "model_forward_time": 1.5571110248565674,
      "step": 93446
    },
    {
      "epoch": 1.854515075683594e-05,
      "step": 93446,
      "training_step_time": 8.64419150352478
    },
    {
      "epoch": 1.8546104431152343e-05,
      "model_forward_time": 1.944681167602539,
      "step": 93447
    },
    {
      "epoch": 1.8546104431152343e-05,
      "step": 93447,
      "training_step_time": 7.825747966766357
    },
    {
      "epoch": 1.854705810546875e-05,
      "model_forward_time": 2.54494571685791,
      "step": 93448
    },
    {
      "epoch": 1.854705810546875e-05,
      "step": 93448,
      "training_step_time": 8.428958654403687
    },
    {
      "epoch": 1.8548011779785158e-05,
      "model_forward_time": 1.9148802757263184,
      "step": 93449
    },
    {
      "epoch": 1.8548011779785158e-05,
      "step": 93449,
      "training_step_time": 7.686267375946045
    },
    {
      "epoch": 1.8548965454101562e-05,
      "grad_norm": 0.09611675143241882,
      "learning_rate": 1.168714835744128e-07,
      "loss": 0.0561,
      "step": 93450
    },
    {
      "epoch": 1.8548965454101562e-05,
      "model_forward_time": 1.92281174659729,
      "step": 93450
    },
    {
      "dynamics_loss_avg": 0.04061127379536629,
      "epoch": 1.8548965454101562e-05,
      "step": 93450
    },
    {
      "action_loss_avg": 0.0015993855573469772,
      "epoch": 1.8548965454101562e-05,
      "step": 93450
    },
    {
      "epoch": 1.8548965454101562e-05,
      "step": 93450,
      "training_step_time": 7.757968902587891
    },
    {
      "epoch": 1.854991912841797e-05,
      "model_forward_time": 1.918203353881836,
      "step": 93451
    },
    {
      "epoch": 1.854991912841797e-05,
      "step": 93451,
      "training_step_time": 7.145519971847534
    },
    {
      "epoch": 1.8550872802734377e-05,
      "model_forward_time": 1.9333200454711914,
      "step": 93452
    },
    {
      "epoch": 1.8550872802734377e-05,
      "step": 93452,
      "training_step_time": 7.271345376968384
    },
    {
      "epoch": 1.855182647705078e-05,
      "model_forward_time": 1.9145419597625732,
      "step": 93453
    },
    {
      "epoch": 1.855182647705078e-05,
      "step": 93453,
      "training_step_time": 7.6942548751831055
    },
    {
      "epoch": 1.8552780151367188e-05,
      "model_forward_time": 1.9274721145629883,
      "step": 93454
    },
    {
      "epoch": 1.8552780151367188e-05,
      "step": 93454,
      "training_step_time": 9.105882406234741
    },
    {
      "epoch": 1.8553733825683592e-05,
      "model_forward_time": 1.9188432693481445,
      "step": 93455
    },
    {
      "epoch": 1.8553733825683592e-05,
      "step": 93455,
      "training_step_time": 7.646474599838257
    },
    {
      "epoch": 1.85546875e-05,
      "model_forward_time": 1.9255566596984863,
      "step": 93456
    },
    {
      "epoch": 1.85546875e-05,
      "step": 93456,
      "training_step_time": 12.44960355758667
    },
    {
      "epoch": 1.8555641174316407e-05,
      "model_forward_time": 1.5505471229553223,
      "step": 93457
    },
    {
      "epoch": 1.8555641174316407e-05,
      "step": 93457,
      "training_step_time": 9.449913740158081
    },
    {
      "epoch": 1.855659484863281e-05,
      "model_forward_time": 2.519209384918213,
      "step": 93458
    },
    {
      "epoch": 1.855659484863281e-05,
      "step": 93458,
      "training_step_time": 7.574577331542969
    },
    {
      "epoch": 1.855754852294922e-05,
      "model_forward_time": 1.9354586601257324,
      "step": 93459
    },
    {
      "epoch": 1.855754852294922e-05,
      "step": 93459,
      "training_step_time": 7.372747182846069
    },
    {
      "epoch": 1.8558502197265626e-05,
      "grad_norm": 0.08882125467061996,
      "learning_rate": 1.1651634212180852e-07,
      "loss": 0.0559,
      "step": 93460
    },
    {
      "epoch": 1.8558502197265626e-05,
      "model_forward_time": 1.9186816215515137,
      "step": 93460
    },
    {
      "dynamics_loss_avg": 0.05357440002262592,
      "epoch": 1.8558502197265626e-05,
      "step": 93460
    },
    {
      "action_loss_avg": 0.0013473825121764094,
      "epoch": 1.8558502197265626e-05,
      "step": 93460
    },
    {
      "epoch": 1.8558502197265626e-05,
      "step": 93460,
      "training_step_time": 7.464959383010864
    },
    {
      "epoch": 1.855945587158203e-05,
      "model_forward_time": 1.7731976509094238,
      "step": 93461
    },
    {
      "epoch": 1.855945587158203e-05,
      "step": 93461,
      "training_step_time": 7.215792894363403
    },
    {
      "epoch": 1.8560409545898437e-05,
      "model_forward_time": 1.7502946853637695,
      "step": 93462
    },
    {
      "epoch": 1.8560409545898437e-05,
      "step": 93462,
      "training_step_time": 8.169365167617798
    },
    {
      "epoch": 1.8561363220214845e-05,
      "model_forward_time": 1.732908010482788,
      "step": 93463
    },
    {
      "epoch": 1.8561363220214845e-05,
      "step": 93463,
      "training_step_time": 7.5766072273254395
    },
    {
      "epoch": 1.856231689453125e-05,
      "model_forward_time": 1.7410123348236084,
      "step": 93464
    },
    {
      "epoch": 1.856231689453125e-05,
      "step": 93464,
      "training_step_time": 8.747008562088013
    },
    {
      "epoch": 1.8563270568847656e-05,
      "model_forward_time": 1.7657346725463867,
      "step": 93465
    },
    {
      "epoch": 1.8563270568847656e-05,
      "step": 93465,
      "training_step_time": 10.25757360458374
    },
    {
      "epoch": 1.8564224243164064e-05,
      "model_forward_time": 1.7479538917541504,
      "step": 93466
    },
    {
      "epoch": 1.8564224243164064e-05,
      "step": 93466,
      "training_step_time": 8.551936626434326
    },
    {
      "epoch": 1.8565177917480468e-05,
      "model_forward_time": 2.929605007171631,
      "step": 93467
    },
    {
      "epoch": 1.8565177917480468e-05,
      "step": 93467,
      "training_step_time": 7.69316029548645
    },
    {
      "epoch": 1.8566131591796875e-05,
      "model_forward_time": 1.7445268630981445,
      "step": 93468
    },
    {
      "epoch": 1.8566131591796875e-05,
      "step": 93468,
      "training_step_time": 7.34771990776062
    },
    {
      "epoch": 1.8567085266113282e-05,
      "model_forward_time": 1.7486274242401123,
      "step": 93469
    },
    {
      "epoch": 1.8567085266113282e-05,
      "step": 93469,
      "training_step_time": 7.5367677211761475
    },
    {
      "epoch": 1.8568038940429686e-05,
      "grad_norm": 0.08656308799982071,
      "learning_rate": 1.1616173471963476e-07,
      "loss": 0.0556,
      "step": 93470
    },
    {
      "epoch": 1.8568038940429686e-05,
      "model_forward_time": 1.7510154247283936,
      "step": 93470
    },
    {
      "dynamics_loss_avg": 0.04881589226424694,
      "epoch": 1.8568038940429686e-05,
      "step": 93470
    },
    {
      "action_loss_avg": 0.0012294889253098517,
      "epoch": 1.8568038940429686e-05,
      "step": 93470
    },
    {
      "epoch": 1.8568038940429686e-05,
      "step": 93470,
      "training_step_time": 7.317433834075928
    },
    {
      "epoch": 1.8568992614746094e-05,
      "model_forward_time": 1.745492935180664,
      "step": 93471
    },
    {
      "epoch": 1.8568992614746094e-05,
      "step": 93471,
      "training_step_time": 7.735195636749268
    },
    {
      "epoch": 1.85699462890625e-05,
      "model_forward_time": 1.7467608451843262,
      "step": 93472
    },
    {
      "epoch": 1.85699462890625e-05,
      "step": 93472,
      "training_step_time": 7.7293641567230225
    },
    {
      "epoch": 1.8570899963378905e-05,
      "model_forward_time": 1.7416977882385254,
      "step": 93473
    },
    {
      "epoch": 1.8570899963378905e-05,
      "step": 93473,
      "training_step_time": 8.980050086975098
    },
    {
      "epoch": 1.8571853637695313e-05,
      "model_forward_time": 1.7442257404327393,
      "step": 93474
    },
    {
      "epoch": 1.8571853637695313e-05,
      "step": 93474,
      "training_step_time": 8.24367356300354
    },
    {
      "epoch": 1.857280731201172e-05,
      "model_forward_time": 1.7881274223327637,
      "step": 93475
    },
    {
      "epoch": 1.857280731201172e-05,
      "step": 93475,
      "training_step_time": 7.72719931602478
    },
    {
      "epoch": 1.8573760986328124e-05,
      "model_forward_time": 2.887026071548462,
      "step": 93476
    },
    {
      "epoch": 1.8573760986328124e-05,
      "step": 93476,
      "training_step_time": 8.393296718597412
    },
    {
      "epoch": 1.857471466064453e-05,
      "model_forward_time": 1.739302635192871,
      "step": 93477
    },
    {
      "epoch": 1.857471466064453e-05,
      "step": 93477,
      "training_step_time": 7.577647686004639
    },
    {
      "epoch": 1.857566833496094e-05,
      "model_forward_time": 1.741628885269165,
      "step": 93478
    },
    {
      "epoch": 1.857566833496094e-05,
      "step": 93478,
      "training_step_time": 7.48138165473938
    },
    {
      "epoch": 1.8576622009277343e-05,
      "model_forward_time": 1.7378175258636475,
      "step": 93479
    },
    {
      "epoch": 1.8576622009277343e-05,
      "step": 93479,
      "training_step_time": 7.259382247924805
    },
    {
      "epoch": 1.857757568359375e-05,
      "grad_norm": 0.11395243555307388,
      "learning_rate": 1.1580766140667111e-07,
      "loss": 0.0552,
      "step": 93480
    },
    {
      "epoch": 1.857757568359375e-05,
      "model_forward_time": 1.7461133003234863,
      "step": 93480
    },
    {
      "dynamics_loss_avg": 0.06573835127055645,
      "epoch": 1.857757568359375e-05,
      "step": 93480
    },
    {
      "action_loss_avg": 0.000970702595077455,
      "epoch": 1.857757568359375e-05,
      "step": 93480
    },
    {
      "epoch": 1.857757568359375e-05,
      "step": 93480,
      "training_step_time": 7.631157636642456
    },
    {
      "epoch": 1.8578529357910158e-05,
      "model_forward_time": 1.7488207817077637,
      "step": 93481
    },
    {
      "epoch": 1.8578529357910158e-05,
      "step": 93481,
      "training_step_time": 7.610605478286743
    },
    {
      "epoch": 1.8579483032226562e-05,
      "model_forward_time": 1.7411339282989502,
      "step": 93482
    },
    {
      "epoch": 1.8579483032226562e-05,
      "step": 93482,
      "training_step_time": 8.457207679748535
    },
    {
      "epoch": 1.858043670654297e-05,
      "model_forward_time": 1.763892412185669,
      "step": 93483
    },
    {
      "epoch": 1.858043670654297e-05,
      "step": 93483,
      "training_step_time": 8.552549839019775
    },
    {
      "epoch": 1.8581390380859377e-05,
      "model_forward_time": 1.0623421669006348,
      "step": 93484
    },
    {
      "epoch": 1.8581390380859377e-05,
      "step": 93484,
      "training_step_time": 8.356651782989502
    },
    {
      "epoch": 1.858234405517578e-05,
      "model_forward_time": 1.7561981678009033,
      "step": 93485
    },
    {
      "epoch": 1.858234405517578e-05,
      "step": 93485,
      "training_step_time": 8.21541953086853
    },
    {
      "epoch": 1.8583297729492188e-05,
      "model_forward_time": 1.7531216144561768,
      "step": 93486
    },
    {
      "epoch": 1.8583297729492188e-05,
      "step": 93486,
      "training_step_time": 7.5439605712890625
    },
    {
      "epoch": 1.8584251403808592e-05,
      "model_forward_time": 1.3859422206878662,
      "step": 93487
    },
    {
      "epoch": 1.8584251403808592e-05,
      "step": 93487,
      "training_step_time": 7.481053352355957
    },
    {
      "epoch": 1.8585205078125e-05,
      "model_forward_time": 1.757673740386963,
      "step": 93488
    },
    {
      "epoch": 1.8585205078125e-05,
      "step": 93488,
      "training_step_time": 7.241711616516113
    },
    {
      "epoch": 1.8586158752441407e-05,
      "model_forward_time": 1.7566413879394531,
      "step": 93489
    },
    {
      "epoch": 1.8586158752441407e-05,
      "step": 93489,
      "training_step_time": 8.544033527374268
    },
    {
      "epoch": 1.858711242675781e-05,
      "grad_norm": 0.12388899177312851,
      "learning_rate": 1.1545412222163766e-07,
      "loss": 0.054,
      "step": 93490
    },
    {
      "epoch": 1.858711242675781e-05,
      "model_forward_time": 1.7650470733642578,
      "step": 93490
    },
    {
      "dynamics_loss_avg": 0.060182343050837514,
      "epoch": 1.858711242675781e-05,
      "step": 93490
    },
    {
      "action_loss_avg": 0.001053542416775599,
      "epoch": 1.858711242675781e-05,
      "step": 93490
    },
    {
      "epoch": 1.858711242675781e-05,
      "step": 93490,
      "training_step_time": 7.5725109577178955
    },
    {
      "epoch": 1.858806610107422e-05,
      "model_forward_time": 1.752140760421753,
      "step": 93491
    },
    {
      "epoch": 1.858806610107422e-05,
      "step": 93491,
      "training_step_time": 7.611609220504761
    },
    {
      "epoch": 1.8589019775390626e-05,
      "model_forward_time": 1.3876726627349854,
      "step": 93492
    },
    {
      "epoch": 1.8589019775390626e-05,
      "step": 93492,
      "training_step_time": 9.29856014251709
    },
    {
      "epoch": 1.858997344970703e-05,
      "model_forward_time": 1.7851898670196533,
      "step": 93493
    },
    {
      "epoch": 1.858997344970703e-05,
      "step": 93493,
      "training_step_time": 8.679976463317871
    },
    {
      "epoch": 1.8590927124023437e-05,
      "model_forward_time": 1.7590160369873047,
      "step": 93494
    },
    {
      "epoch": 1.8590927124023437e-05,
      "step": 93494,
      "training_step_time": 7.669059753417969
    },
    {
      "epoch": 1.8591880798339845e-05,
      "model_forward_time": 2.903773069381714,
      "step": 93495
    },
    {
      "epoch": 1.8591880798339845e-05,
      "step": 93495,
      "training_step_time": 7.648481369018555
    },
    {
      "epoch": 1.859283447265625e-05,
      "model_forward_time": 1.753739595413208,
      "step": 93496
    },
    {
      "epoch": 1.859283447265625e-05,
      "step": 93496,
      "training_step_time": 8.466185331344604
    },
    {
      "epoch": 1.8593788146972656e-05,
      "model_forward_time": 1.7565248012542725,
      "step": 93497
    },
    {
      "epoch": 1.8593788146972656e-05,
      "step": 93497,
      "training_step_time": 7.223869800567627
    },
    {
      "epoch": 1.8594741821289064e-05,
      "model_forward_time": 1.7578539848327637,
      "step": 93498
    },
    {
      "epoch": 1.8594741821289064e-05,
      "step": 93498,
      "training_step_time": 9.469665050506592
    },
    {
      "epoch": 1.8595695495605468e-05,
      "model_forward_time": 1.7503163814544678,
      "step": 93499
    },
    {
      "epoch": 1.8595695495605468e-05,
      "step": 93499,
      "training_step_time": 8.453237056732178
    },
    {
      "epoch": 1.8596649169921875e-05,
      "grad_norm": 0.11484602838754654,
      "learning_rate": 1.1510111720319794e-07,
      "loss": 0.0557,
      "step": 93500
    },
    {
      "epoch": 1.8596649169921875e-05,
      "model_forward_time": 1.7496654987335205,
      "step": 93500
    },
    {
      "dynamics_loss_avg": 0.06217893604189158,
      "epoch": 1.8596649169921875e-05,
      "step": 93500
    },
    {
      "action_loss_avg": 0.0013307097455253825,
      "epoch": 1.8596649169921875e-05,
      "step": 93500
    },
    {
      "epoch": 1.8596649169921875e-05,
      "step": 93500,
      "training_step_time": 8.577351570129395
    },
    {
      "epoch": 1.8597602844238282e-05,
      "model_forward_time": 1.7556631565093994,
      "step": 93501
    },
    {
      "epoch": 1.8597602844238282e-05,
      "step": 93501,
      "training_step_time": 7.759853363037109
    },
    {
      "epoch": 1.8598556518554686e-05,
      "model_forward_time": 1.8027701377868652,
      "step": 93502
    },
    {
      "epoch": 1.8598556518554686e-05,
      "step": 93502,
      "training_step_time": 8.72920846939087
    },
    {
      "epoch": 1.8599510192871094e-05,
      "model_forward_time": 1.7495722770690918,
      "step": 93503
    },
    {
      "epoch": 1.8599510192871094e-05,
      "step": 93503,
      "training_step_time": 7.7061262130737305
    },
    {
      "epoch": 1.86004638671875e-05,
      "model_forward_time": 2.881279468536377,
      "step": 93504
    },
    {
      "epoch": 1.86004638671875e-05,
      "step": 93504,
      "training_step_time": 7.609466791152954
    },
    {
      "epoch": 1.8601417541503905e-05,
      "model_forward_time": 1.7382373809814453,
      "step": 93505
    },
    {
      "epoch": 1.8601417541503905e-05,
      "step": 93505,
      "training_step_time": 7.539971351623535
    },
    {
      "epoch": 1.8602371215820313e-05,
      "model_forward_time": 1.7698817253112793,
      "step": 93506
    },
    {
      "epoch": 1.8602371215820313e-05,
      "step": 93506,
      "training_step_time": 7.499927282333374
    },
    {
      "epoch": 1.860332489013672e-05,
      "model_forward_time": 1.7536485195159912,
      "step": 93507
    },
    {
      "epoch": 1.860332489013672e-05,
      "step": 93507,
      "training_step_time": 7.54174017906189
    },
    {
      "epoch": 1.8604278564453124e-05,
      "model_forward_time": 1.762761116027832,
      "step": 93508
    },
    {
      "epoch": 1.8604278564453124e-05,
      "step": 93508,
      "training_step_time": 7.5947020053863525
    },
    {
      "epoch": 1.860523223876953e-05,
      "model_forward_time": 1.7639765739440918,
      "step": 93509
    },
    {
      "epoch": 1.860523223876953e-05,
      "step": 93509,
      "training_step_time": 9.081084966659546
    },
    {
      "epoch": 1.860618591308594e-05,
      "grad_norm": 0.08758260309696198,
      "learning_rate": 1.1474864638995665e-07,
      "loss": 0.0544,
      "step": 93510
    },
    {
      "epoch": 1.860618591308594e-05,
      "model_forward_time": 1.7535927295684814,
      "step": 93510
    },
    {
      "dynamics_loss_avg": 0.0622778968885541,
      "epoch": 1.860618591308594e-05,
      "step": 93510
    },
    {
      "action_loss_avg": 0.0014222042635083198,
      "epoch": 1.860618591308594e-05,
      "step": 93510
    },
    {
      "epoch": 1.860618591308594e-05,
      "step": 93510,
      "training_step_time": 8.765514612197876
    },
    {
      "epoch": 1.8607139587402343e-05,
      "model_forward_time": 1.7856497764587402,
      "step": 93511
    },
    {
      "epoch": 1.8607139587402343e-05,
      "step": 93511,
      "training_step_time": 8.507951498031616
    },
    {
      "epoch": 1.860809326171875e-05,
      "model_forward_time": 1.7603509426116943,
      "step": 93512
    },
    {
      "epoch": 1.860809326171875e-05,
      "step": 93512,
      "training_step_time": 10.025720357894897
    },
    {
      "epoch": 1.8609046936035158e-05,
      "model_forward_time": 2.8488478660583496,
      "step": 93513
    },
    {
      "epoch": 1.8609046936035158e-05,
      "step": 93513,
      "training_step_time": 7.572802305221558
    },
    {
      "epoch": 1.8610000610351562e-05,
      "model_forward_time": 1.7736475467681885,
      "step": 93514
    },
    {
      "epoch": 1.8610000610351562e-05,
      "step": 93514,
      "training_step_time": 8.43180775642395
    },
    {
      "epoch": 1.861095428466797e-05,
      "model_forward_time": 1.7641520500183105,
      "step": 93515
    },
    {
      "epoch": 1.861095428466797e-05,
      "step": 93515,
      "training_step_time": 7.428326606750488
    },
    {
      "epoch": 1.8611907958984373e-05,
      "model_forward_time": 1.755138874053955,
      "step": 93516
    },
    {
      "epoch": 1.8611907958984373e-05,
      "step": 93516,
      "training_step_time": 7.526731014251709
    },
    {
      "epoch": 1.861286163330078e-05,
      "model_forward_time": 1.764280080795288,
      "step": 93517
    },
    {
      "epoch": 1.861286163330078e-05,
      "step": 93517,
      "training_step_time": 7.67509913444519
    },
    {
      "epoch": 1.8613815307617188e-05,
      "model_forward_time": 1.755415439605713,
      "step": 93518
    },
    {
      "epoch": 1.8613815307617188e-05,
      "step": 93518,
      "training_step_time": 7.596001863479614
    },
    {
      "epoch": 1.8614768981933592e-05,
      "model_forward_time": 1.7681324481964111,
      "step": 93519
    },
    {
      "epoch": 1.8614768981933592e-05,
      "step": 93519,
      "training_step_time": 10.219091415405273
    },
    {
      "epoch": 1.861572265625e-05,
      "grad_norm": 0.09294112026691437,
      "learning_rate": 1.1439670982045736e-07,
      "loss": 0.053,
      "step": 93520
    },
    {
      "epoch": 1.861572265625e-05,
      "model_forward_time": 1.7938811779022217,
      "step": 93520
    },
    {
      "dynamics_loss_avg": 0.04735115263611078,
      "epoch": 1.861572265625e-05,
      "step": 93520
    },
    {
      "action_loss_avg": 0.0010008850484155118,
      "epoch": 1.861572265625e-05,
      "step": 93520
    },
    {
      "epoch": 1.861572265625e-05,
      "step": 93520,
      "training_step_time": 7.641477823257446
    },
    {
      "epoch": 1.8616676330566407e-05,
      "model_forward_time": 1.786797285079956,
      "step": 93521
    },
    {
      "epoch": 1.8616676330566407e-05,
      "step": 93521,
      "training_step_time": 9.22452163696289
    },
    {
      "epoch": 1.861763000488281e-05,
      "model_forward_time": 2.831721305847168,
      "step": 93522
    },
    {
      "epoch": 1.861763000488281e-05,
      "step": 93522,
      "training_step_time": 7.682469844818115
    },
    {
      "epoch": 1.861858367919922e-05,
      "model_forward_time": 1.7562675476074219,
      "step": 93523
    },
    {
      "epoch": 1.861858367919922e-05,
      "step": 93523,
      "training_step_time": 8.226151466369629
    },
    {
      "epoch": 1.8619537353515626e-05,
      "model_forward_time": 1.7630095481872559,
      "step": 93524
    },
    {
      "epoch": 1.8619537353515626e-05,
      "step": 93524,
      "training_step_time": 7.5039222240448
    },
    {
      "epoch": 1.862049102783203e-05,
      "model_forward_time": 1.7474513053894043,
      "step": 93525
    },
    {
      "epoch": 1.862049102783203e-05,
      "step": 93525,
      "training_step_time": 7.507683038711548
    },
    {
      "epoch": 1.8621444702148437e-05,
      "model_forward_time": 1.7541069984436035,
      "step": 93526
    },
    {
      "epoch": 1.8621444702148437e-05,
      "step": 93526,
      "training_step_time": 7.5610315799713135
    },
    {
      "epoch": 1.8622398376464845e-05,
      "model_forward_time": 1.76680588722229,
      "step": 93527
    },
    {
      "epoch": 1.8622398376464845e-05,
      "step": 93527,
      "training_step_time": 8.652748107910156
    },
    {
      "epoch": 1.862335205078125e-05,
      "model_forward_time": 1.7614877223968506,
      "step": 93528
    },
    {
      "epoch": 1.862335205078125e-05,
      "step": 93528,
      "training_step_time": 8.472974300384521
    },
    {
      "epoch": 1.8624305725097656e-05,
      "model_forward_time": 1.7621912956237793,
      "step": 93529
    },
    {
      "epoch": 1.8624305725097656e-05,
      "step": 93529,
      "training_step_time": 8.623275995254517
    },
    {
      "epoch": 1.8625259399414064e-05,
      "grad_norm": 0.08835022896528244,
      "learning_rate": 1.1404530753318877e-07,
      "loss": 0.0527,
      "step": 93530
    },
    {
      "epoch": 1.8625259399414064e-05,
      "model_forward_time": 1.7672500610351562,
      "step": 93530
    },
    {
      "dynamics_loss_avg": 0.04958755169063807,
      "epoch": 1.8625259399414064e-05,
      "step": 93530
    },
    {
      "action_loss_avg": 0.0014939673856133595,
      "epoch": 1.8625259399414064e-05,
      "step": 93530
    },
    {
      "epoch": 1.8625259399414064e-05,
      "step": 93530,
      "training_step_time": 8.657496690750122
    },
    {
      "epoch": 1.8626213073730468e-05,
      "model_forward_time": 1.7672839164733887,
      "step": 93531
    },
    {
      "epoch": 1.8626213073730468e-05,
      "step": 93531,
      "training_step_time": 7.637261152267456
    },
    {
      "epoch": 1.8627166748046875e-05,
      "model_forward_time": 2.9078454971313477,
      "step": 93532
    },
    {
      "epoch": 1.8627166748046875e-05,
      "step": 93532,
      "training_step_time": 7.646756649017334
    },
    {
      "epoch": 1.8628120422363282e-05,
      "model_forward_time": 1.756920337677002,
      "step": 93533
    },
    {
      "epoch": 1.8628120422363282e-05,
      "step": 93533,
      "training_step_time": 7.504261016845703
    },
    {
      "epoch": 1.8629074096679686e-05,
      "model_forward_time": 1.7623350620269775,
      "step": 93534
    },
    {
      "epoch": 1.8629074096679686e-05,
      "step": 93534,
      "training_step_time": 7.135700702667236
    },
    {
      "epoch": 1.8630027770996094e-05,
      "model_forward_time": 1.7583980560302734,
      "step": 93535
    },
    {
      "epoch": 1.8630027770996094e-05,
      "step": 93535,
      "training_step_time": 7.572611093521118
    },
    {
      "epoch": 1.86309814453125e-05,
      "model_forward_time": 1.7503561973571777,
      "step": 93536
    },
    {
      "epoch": 1.86309814453125e-05,
      "step": 93536,
      "training_step_time": 7.725557804107666
    },
    {
      "epoch": 1.8631935119628905e-05,
      "model_forward_time": 1.7526450157165527,
      "step": 93537
    },
    {
      "epoch": 1.8631935119628905e-05,
      "step": 93537,
      "training_step_time": 8.53280258178711
    },
    {
      "epoch": 1.8632888793945313e-05,
      "model_forward_time": 1.7559130191802979,
      "step": 93538
    },
    {
      "epoch": 1.8632888793945313e-05,
      "step": 93538,
      "training_step_time": 8.37090539932251
    },
    {
      "epoch": 1.863384246826172e-05,
      "model_forward_time": 1.7904322147369385,
      "step": 93539
    },
    {
      "epoch": 1.863384246826172e-05,
      "step": 93539,
      "training_step_time": 9.357622146606445
    },
    {
      "epoch": 1.8634796142578124e-05,
      "grad_norm": 0.07244996726512909,
      "learning_rate": 1.136944395665801e-07,
      "loss": 0.0533,
      "step": 93540
    },
    {
      "epoch": 1.8634796142578124e-05,
      "model_forward_time": 1.7521634101867676,
      "step": 93540
    },
    {
      "dynamics_loss_avg": 0.052322893775999545,
      "epoch": 1.8634796142578124e-05,
      "step": 93540
    },
    {
      "action_loss_avg": 0.0010406531946500763,
      "epoch": 1.8634796142578124e-05,
      "step": 93540
    },
    {
      "epoch": 1.8634796142578124e-05,
      "step": 93540,
      "training_step_time": 7.672641038894653
    },
    {
      "epoch": 1.863574981689453e-05,
      "model_forward_time": 2.9083712100982666,
      "step": 93541
    },
    {
      "epoch": 1.863574981689453e-05,
      "step": 93541,
      "training_step_time": 7.676668643951416
    },
    {
      "epoch": 1.863670349121094e-05,
      "model_forward_time": 1.75235915184021,
      "step": 93542
    },
    {
      "epoch": 1.863670349121094e-05,
      "step": 93542,
      "training_step_time": 8.405878782272339
    },
    {
      "epoch": 1.8637657165527343e-05,
      "model_forward_time": 1.73616623878479,
      "step": 93543
    },
    {
      "epoch": 1.8637657165527343e-05,
      "step": 93543,
      "training_step_time": 8.035882234573364
    },
    {
      "epoch": 1.863861083984375e-05,
      "model_forward_time": 1.744443416595459,
      "step": 93544
    },
    {
      "epoch": 1.863861083984375e-05,
      "step": 93544,
      "training_step_time": 7.564661502838135
    },
    {
      "epoch": 1.8639564514160158e-05,
      "model_forward_time": 1.746307611465454,
      "step": 93545
    },
    {
      "epoch": 1.8639564514160158e-05,
      "step": 93545,
      "training_step_time": 8.69753623008728
    },
    {
      "epoch": 1.8640518188476562e-05,
      "model_forward_time": 1.7558021545410156,
      "step": 93546
    },
    {
      "epoch": 1.8640518188476562e-05,
      "step": 93546,
      "training_step_time": 7.6112449169158936
    },
    {
      "epoch": 1.864147186279297e-05,
      "model_forward_time": 1.7460501194000244,
      "step": 93547
    },
    {
      "epoch": 1.864147186279297e-05,
      "step": 93547,
      "training_step_time": 7.765304327011108
    },
    {
      "epoch": 1.8642425537109373e-05,
      "model_forward_time": 1.7669932842254639,
      "step": 93548
    },
    {
      "epoch": 1.8642425537109373e-05,
      "step": 93548,
      "training_step_time": 8.717548370361328
    },
    {
      "epoch": 1.864337921142578e-05,
      "model_forward_time": 1.4266326427459717,
      "step": 93549
    },
    {
      "epoch": 1.864337921142578e-05,
      "step": 93549,
      "training_step_time": 9.911787986755371
    },
    {
      "epoch": 1.8644332885742188e-05,
      "grad_norm": 0.3667798638343811,
      "learning_rate": 1.1334410595900014e-07,
      "loss": 0.053,
      "step": 93550
    },
    {
      "epoch": 1.8644332885742188e-05,
      "model_forward_time": 2.8702707290649414,
      "step": 93550
    },
    {
      "dynamics_loss_avg": 0.07499864511191845,
      "epoch": 1.8644332885742188e-05,
      "step": 93550
    },
    {
      "action_loss_avg": 0.0018133174744434655,
      "epoch": 1.8644332885742188e-05,
      "step": 93550
    },
    {
      "epoch": 1.8644332885742188e-05,
      "step": 93550,
      "training_step_time": 7.788258075714111
    },
    {
      "epoch": 1.8645286560058592e-05,
      "model_forward_time": 1.757946252822876,
      "step": 93551
    },
    {
      "epoch": 1.8645286560058592e-05,
      "step": 93551,
      "training_step_time": 7.511080265045166
    },
    {
      "epoch": 1.8646240234375e-05,
      "model_forward_time": 1.7547671794891357,
      "step": 93552
    },
    {
      "epoch": 1.8646240234375e-05,
      "step": 93552,
      "training_step_time": 7.484563827514648
    },
    {
      "epoch": 1.8647193908691407e-05,
      "model_forward_time": 1.7581937313079834,
      "step": 93553
    },
    {
      "epoch": 1.8647193908691407e-05,
      "step": 93553,
      "training_step_time": 7.566120862960815
    },
    {
      "epoch": 1.864814758300781e-05,
      "model_forward_time": 1.755544662475586,
      "step": 93554
    },
    {
      "epoch": 1.864814758300781e-05,
      "step": 93554,
      "training_step_time": 7.7385406494140625
    },
    {
      "epoch": 1.864910125732422e-05,
      "model_forward_time": 1.759211540222168,
      "step": 93555
    },
    {
      "epoch": 1.864910125732422e-05,
      "step": 93555,
      "training_step_time": 7.688015460968018
    },
    {
      "epoch": 1.8650054931640626e-05,
      "model_forward_time": 1.749866247177124,
      "step": 93556
    },
    {
      "epoch": 1.8650054931640626e-05,
      "step": 93556,
      "training_step_time": 9.383086919784546
    },
    {
      "epoch": 1.865100860595703e-05,
      "model_forward_time": 1.777862787246704,
      "step": 93557
    },
    {
      "epoch": 1.865100860595703e-05,
      "step": 93557,
      "training_step_time": 9.27857494354248
    },
    {
      "epoch": 1.8651962280273437e-05,
      "model_forward_time": 1.7663404941558838,
      "step": 93558
    },
    {
      "epoch": 1.8651962280273437e-05,
      "step": 93558,
      "training_step_time": 8.16091775894165
    },
    {
      "epoch": 1.8652915954589845e-05,
      "model_forward_time": 2.8211541175842285,
      "step": 93559
    },
    {
      "epoch": 1.8652915954589845e-05,
      "step": 93559,
      "training_step_time": 8.40253233909607
    },
    {
      "epoch": 1.865386962890625e-05,
      "grad_norm": 0.08032175898551941,
      "learning_rate": 1.129943067487621e-07,
      "loss": 0.0519,
      "step": 93560
    },
    {
      "epoch": 1.865386962890625e-05,
      "model_forward_time": 1.7487618923187256,
      "step": 93560
    },
    {
      "dynamics_loss_avg": 0.03885695450007916,
      "epoch": 1.865386962890625e-05,
      "step": 93560
    },
    {
      "action_loss_avg": 0.0008440940277068876,
      "epoch": 1.865386962890625e-05,
      "step": 93560
    },
    {
      "epoch": 1.865386962890625e-05,
      "step": 93560,
      "training_step_time": 7.421285629272461
    },
    {
      "epoch": 1.8654823303222656e-05,
      "model_forward_time": 1.7517554759979248,
      "step": 93561
    },
    {
      "epoch": 1.8654823303222656e-05,
      "step": 93561,
      "training_step_time": 7.763616323471069
    },
    {
      "epoch": 1.8655776977539064e-05,
      "model_forward_time": 1.7500367164611816,
      "step": 93562
    },
    {
      "epoch": 1.8655776977539064e-05,
      "step": 93562,
      "training_step_time": 7.6270432472229
    },
    {
      "epoch": 1.8656730651855468e-05,
      "model_forward_time": 1.7557637691497803,
      "step": 93563
    },
    {
      "epoch": 1.8656730651855468e-05,
      "step": 93563,
      "training_step_time": 7.8055806159973145
    },
    {
      "epoch": 1.8657684326171875e-05,
      "model_forward_time": 1.7502548694610596,
      "step": 93564
    },
    {
      "epoch": 1.8657684326171875e-05,
      "step": 93564,
      "training_step_time": 8.745964765548706
    },
    {
      "epoch": 1.8658638000488282e-05,
      "model_forward_time": 1.755401611328125,
      "step": 93565
    },
    {
      "epoch": 1.8658638000488282e-05,
      "step": 93565,
      "training_step_time": 8.381022214889526
    },
    {
      "epoch": 1.8659591674804686e-05,
      "model_forward_time": 1.77919340133667,
      "step": 93566
    },
    {
      "epoch": 1.8659591674804686e-05,
      "step": 93566,
      "training_step_time": 10.066924333572388
    },
    {
      "epoch": 1.8660545349121094e-05,
      "model_forward_time": 1.3769068717956543,
      "step": 93567
    },
    {
      "epoch": 1.8660545349121094e-05,
      "step": 93567,
      "training_step_time": 8.599501132965088
    },
    {
      "epoch": 1.86614990234375e-05,
      "model_forward_time": 1.7547922134399414,
      "step": 93568
    },
    {
      "epoch": 1.86614990234375e-05,
      "step": 93568,
      "training_step_time": 8.484816312789917
    },
    {
      "epoch": 1.8662452697753905e-05,
      "model_forward_time": 2.892075777053833,
      "step": 93569
    },
    {
      "epoch": 1.8662452697753905e-05,
      "step": 93569,
      "training_step_time": 7.75187873840332
    },
    {
      "epoch": 1.8663406372070313e-05,
      "grad_norm": 0.09744422882795334,
      "learning_rate": 1.126450419741193e-07,
      "loss": 0.0542,
      "step": 93570
    },
    {
      "epoch": 1.8663406372070313e-05,
      "model_forward_time": 1.7519075870513916,
      "step": 93570
    },
    {
      "dynamics_loss_avg": 0.0647548595443368,
      "epoch": 1.8663406372070313e-05,
      "step": 93570
    },
    {
      "action_loss_avg": 0.0018711475509917363,
      "epoch": 1.8663406372070313e-05,
      "step": 93570
    },
    {
      "epoch": 1.8663406372070313e-05,
      "step": 93570,
      "training_step_time": 7.510033845901489
    },
    {
      "epoch": 1.866436004638672e-05,
      "model_forward_time": 1.7579357624053955,
      "step": 93571
    },
    {
      "epoch": 1.866436004638672e-05,
      "step": 93571,
      "training_step_time": 8.252549409866333
    },
    {
      "epoch": 1.8665313720703124e-05,
      "model_forward_time": 1.7585346698760986,
      "step": 93572
    },
    {
      "epoch": 1.8665313720703124e-05,
      "step": 93572,
      "training_step_time": 7.730201721191406
    },
    {
      "epoch": 1.866626739501953e-05,
      "model_forward_time": 1.7547166347503662,
      "step": 93573
    },
    {
      "epoch": 1.866626739501953e-05,
      "step": 93573,
      "training_step_time": 8.481413125991821
    },
    {
      "epoch": 1.866722106933594e-05,
      "model_forward_time": 1.7562596797943115,
      "step": 93574
    },
    {
      "epoch": 1.866722106933594e-05,
      "step": 93574,
      "training_step_time": 8.2496018409729
    },
    {
      "epoch": 1.8668174743652343e-05,
      "model_forward_time": 1.753870964050293,
      "step": 93575
    },
    {
      "epoch": 1.8668174743652343e-05,
      "step": 93575,
      "training_step_time": 8.847694635391235
    },
    {
      "epoch": 1.866912841796875e-05,
      "model_forward_time": 1.7615752220153809,
      "step": 93576
    },
    {
      "epoch": 1.866912841796875e-05,
      "step": 93576,
      "training_step_time": 8.62663221359253
    },
    {
      "epoch": 1.8670082092285158e-05,
      "model_forward_time": 1.7493414878845215,
      "step": 93577
    },
    {
      "epoch": 1.8670082092285158e-05,
      "step": 93577,
      "training_step_time": 7.622846603393555
    },
    {
      "epoch": 1.8671035766601562e-05,
      "model_forward_time": 2.8982503414154053,
      "step": 93578
    },
    {
      "epoch": 1.8671035766601562e-05,
      "step": 93578,
      "training_step_time": 7.6072142124176025
    },
    {
      "epoch": 1.867198944091797e-05,
      "model_forward_time": 1.7398674488067627,
      "step": 93579
    },
    {
      "epoch": 1.867198944091797e-05,
      "step": 93579,
      "training_step_time": 7.5637476444244385
    },
    {
      "epoch": 1.8672943115234373e-05,
      "grad_norm": 0.10370506346225739,
      "learning_rate": 1.1229631167326671e-07,
      "loss": 0.0535,
      "step": 93580
    },
    {
      "epoch": 1.8672943115234373e-05,
      "model_forward_time": 1.7512547969818115,
      "step": 93580
    },
    {
      "dynamics_loss_avg": 0.05116861071437597,
      "epoch": 1.8672943115234373e-05,
      "step": 93580
    },
    {
      "action_loss_avg": 0.0011821730353403837,
      "epoch": 1.8672943115234373e-05,
      "step": 93580
    },
    {
      "epoch": 1.8672943115234373e-05,
      "step": 93580,
      "training_step_time": 7.413304567337036
    },
    {
      "epoch": 1.867389678955078e-05,
      "model_forward_time": 1.7542576789855957,
      "step": 93581
    },
    {
      "epoch": 1.867389678955078e-05,
      "step": 93581,
      "training_step_time": 7.664341688156128
    },
    {
      "epoch": 1.8674850463867188e-05,
      "model_forward_time": 1.7595469951629639,
      "step": 93582
    },
    {
      "epoch": 1.8674850463867188e-05,
      "step": 93582,
      "training_step_time": 8.199901342391968
    },
    {
      "epoch": 1.8675804138183592e-05,
      "model_forward_time": 1.7586517333984375,
      "step": 93583
    },
    {
      "epoch": 1.8675804138183592e-05,
      "step": 93583,
      "training_step_time": 7.623298406600952
    },
    {
      "epoch": 1.86767578125e-05,
      "model_forward_time": 1.745401382446289,
      "step": 93584
    },
    {
      "epoch": 1.86767578125e-05,
      "step": 93584,
      "training_step_time": 7.679443836212158
    },
    {
      "epoch": 1.8677711486816407e-05,
      "model_forward_time": 1.785325050354004,
      "step": 93585
    },
    {
      "epoch": 1.8677711486816407e-05,
      "step": 93585,
      "training_step_time": 8.632453680038452
    },
    {
      "epoch": 1.867866516113281e-05,
      "model_forward_time": 1.785292387008667,
      "step": 93586
    },
    {
      "epoch": 1.867866516113281e-05,
      "step": 93586,
      "training_step_time": 7.61562705039978
    },
    {
      "epoch": 1.867961883544922e-05,
      "model_forward_time": 2.8745124340057373,
      "step": 93587
    },
    {
      "epoch": 1.867961883544922e-05,
      "step": 93587,
      "training_step_time": 7.7131664752960205
    },
    {
      "epoch": 1.8680572509765626e-05,
      "model_forward_time": 1.7580180168151855,
      "step": 93588
    },
    {
      "epoch": 1.8680572509765626e-05,
      "step": 93588,
      "training_step_time": 7.564386367797852
    },
    {
      "epoch": 1.868152618408203e-05,
      "model_forward_time": 1.74853515625,
      "step": 93589
    },
    {
      "epoch": 1.868152618408203e-05,
      "step": 93589,
      "training_step_time": 7.4446516036987305
    },
    {
      "epoch": 1.8682479858398437e-05,
      "grad_norm": 0.08450126647949219,
      "learning_rate": 1.1194811588434051e-07,
      "loss": 0.0551,
      "step": 93590
    },
    {
      "epoch": 1.8682479858398437e-05,
      "model_forward_time": 1.7551674842834473,
      "step": 93590
    },
    {
      "dynamics_loss_avg": 0.051038354821503165,
      "epoch": 1.8682479858398437e-05,
      "step": 93590
    },
    {
      "action_loss_avg": 0.0019433866604231299,
      "epoch": 1.8682479858398437e-05,
      "step": 93590
    },
    {
      "epoch": 1.8682479858398437e-05,
      "step": 93590,
      "training_step_time": 7.583380937576294
    },
    {
      "epoch": 1.8683433532714845e-05,
      "model_forward_time": 1.0502822399139404,
      "step": 93591
    },
    {
      "epoch": 1.8683433532714845e-05,
      "step": 93591,
      "training_step_time": 7.667969703674316
    },
    {
      "epoch": 1.868438720703125e-05,
      "model_forward_time": 1.7402215003967285,
      "step": 93592
    },
    {
      "epoch": 1.868438720703125e-05,
      "step": 93592,
      "training_step_time": 7.656738042831421
    },
    {
      "epoch": 1.8685340881347656e-05,
      "model_forward_time": 1.7487947940826416,
      "step": 93593
    },
    {
      "epoch": 1.8685340881347656e-05,
      "step": 93593,
      "training_step_time": 8.869230270385742
    },
    {
      "epoch": 1.8686294555664064e-05,
      "model_forward_time": 1.7721760272979736,
      "step": 93594
    },
    {
      "epoch": 1.8686294555664064e-05,
      "step": 93594,
      "training_step_time": 9.508152723312378
    },
    {
      "epoch": 1.8687248229980468e-05,
      "model_forward_time": 1.0446140766143799,
      "step": 93595
    },
    {
      "epoch": 1.8687248229980468e-05,
      "step": 93595,
      "training_step_time": 7.612221717834473
    },
    {
      "epoch": 1.8688201904296875e-05,
      "model_forward_time": 1.7491168975830078,
      "step": 93596
    },
    {
      "epoch": 1.8688201904296875e-05,
      "step": 93596,
      "training_step_time": 8.168095111846924
    },
    {
      "epoch": 1.8689155578613282e-05,
      "model_forward_time": 2.897648572921753,
      "step": 93597
    },
    {
      "epoch": 1.8689155578613282e-05,
      "step": 93597,
      "training_step_time": 9.298683404922485
    },
    {
      "epoch": 1.8690109252929687e-05,
      "model_forward_time": 1.7484931945800781,
      "step": 93598
    },
    {
      "epoch": 1.8690109252929687e-05,
      "step": 93598,
      "training_step_time": 7.465112924575806
    },
    {
      "epoch": 1.8691062927246094e-05,
      "model_forward_time": 1.7571806907653809,
      "step": 93599
    },
    {
      "epoch": 1.8691062927246094e-05,
      "step": 93599,
      "training_step_time": 8.599878311157227
    },
    {
      "epoch": 1.86920166015625e-05,
      "grad_norm": 0.11004931479692459,
      "learning_rate": 1.1160045464542025e-07,
      "loss": 0.055,
      "step": 93600
    },
    {
      "epoch": 1.86920166015625e-05,
      "model_forward_time": 1.7520751953125,
      "step": 93600
    },
    {
      "dynamics_loss_avg": 0.04133567046374083,
      "epoch": 1.86920166015625e-05,
      "step": 93600
    },
    {
      "action_loss_avg": 0.0011249398696236313,
      "epoch": 1.86920166015625e-05,
      "step": 93600
    },
    {
      "epoch": 1.86920166015625e-05,
      "step": 93600,
      "training_step_time": 8.444250583648682
    },
    {
      "epoch": 1.8692970275878905e-05,
      "model_forward_time": 1.746187686920166,
      "step": 93601
    },
    {
      "epoch": 1.8692970275878905e-05,
      "step": 93601,
      "training_step_time": 7.676120042800903
    },
    {
      "epoch": 1.8693923950195313e-05,
      "model_forward_time": 1.7488722801208496,
      "step": 93602
    },
    {
      "epoch": 1.8693923950195313e-05,
      "step": 93602,
      "training_step_time": 7.731437921524048
    },
    {
      "epoch": 1.869487762451172e-05,
      "model_forward_time": 1.7817535400390625,
      "step": 93603
    },
    {
      "epoch": 1.869487762451172e-05,
      "step": 93603,
      "training_step_time": 11.917342901229858
    },
    {
      "epoch": 1.8695831298828124e-05,
      "model_forward_time": 1.7673132419586182,
      "step": 93604
    },
    {
      "epoch": 1.8695831298828124e-05,
      "step": 93604,
      "training_step_time": 9.341991186141968
    },
    {
      "epoch": 1.869678497314453e-05,
      "model_forward_time": 1.744368553161621,
      "step": 93605
    },
    {
      "epoch": 1.869678497314453e-05,
      "step": 93605,
      "training_step_time": 8.230726480484009
    },
    {
      "epoch": 1.869773864746094e-05,
      "model_forward_time": 2.9171693325042725,
      "step": 93606
    },
    {
      "epoch": 1.869773864746094e-05,
      "step": 93606,
      "training_step_time": 8.522968530654907
    },
    {
      "epoch": 1.8698692321777343e-05,
      "model_forward_time": 1.75596022605896,
      "step": 93607
    },
    {
      "epoch": 1.8698692321777343e-05,
      "step": 93607,
      "training_step_time": 7.551063060760498
    },
    {
      "epoch": 1.869964599609375e-05,
      "model_forward_time": 1.7479534149169922,
      "step": 93608
    },
    {
      "epoch": 1.869964599609375e-05,
      "step": 93608,
      "training_step_time": 7.589704275131226
    },
    {
      "epoch": 1.8700599670410158e-05,
      "model_forward_time": 1.753483533859253,
      "step": 93609
    },
    {
      "epoch": 1.8700599670410158e-05,
      "step": 93609,
      "training_step_time": 7.555838108062744
    },
    {
      "epoch": 1.8701553344726562e-05,
      "grad_norm": 0.09798669815063477,
      "learning_rate": 1.1125332799452327e-07,
      "loss": 0.0558,
      "step": 93610
    },
    {
      "epoch": 1.8701553344726562e-05,
      "model_forward_time": 1.7477624416351318,
      "step": 93610
    },
    {
      "dynamics_loss_avg": 0.05358234839513898,
      "epoch": 1.8701553344726562e-05,
      "step": 93610
    },
    {
      "action_loss_avg": 0.0010046079754829408,
      "epoch": 1.8701553344726562e-05,
      "step": 93610
    },
    {
      "epoch": 1.8701553344726562e-05,
      "step": 93610,
      "training_step_time": 7.670550346374512
    },
    {
      "epoch": 1.870250701904297e-05,
      "model_forward_time": 1.7584216594696045,
      "step": 93611
    },
    {
      "epoch": 1.870250701904297e-05,
      "step": 93611,
      "training_step_time": 8.932499408721924
    },
    {
      "epoch": 1.8703460693359373e-05,
      "model_forward_time": 1.7759926319122314,
      "step": 93612
    },
    {
      "epoch": 1.8703460693359373e-05,
      "step": 93612,
      "training_step_time": 9.309244155883789
    },
    {
      "epoch": 1.870441436767578e-05,
      "model_forward_time": 1.7513747215270996,
      "step": 93613
    },
    {
      "epoch": 1.870441436767578e-05,
      "step": 93613,
      "training_step_time": 8.678872346878052
    },
    {
      "epoch": 1.8705368041992188e-05,
      "model_forward_time": 1.750877857208252,
      "step": 93614
    },
    {
      "epoch": 1.8705368041992188e-05,
      "step": 93614,
      "training_step_time": 8.519500732421875
    },
    {
      "epoch": 1.8706321716308592e-05,
      "model_forward_time": 2.8827011585235596,
      "step": 93615
    },
    {
      "epoch": 1.8706321716308592e-05,
      "step": 93615,
      "training_step_time": 7.627956390380859
    },
    {
      "epoch": 1.8707275390625e-05,
      "model_forward_time": 1.748595952987671,
      "step": 93616
    },
    {
      "epoch": 1.8707275390625e-05,
      "step": 93616,
      "training_step_time": 7.56902813911438
    },
    {
      "epoch": 1.8708229064941407e-05,
      "model_forward_time": 1.7591118812561035,
      "step": 93617
    },
    {
      "epoch": 1.8708229064941407e-05,
      "step": 93617,
      "training_step_time": 7.491412401199341
    },
    {
      "epoch": 1.870918273925781e-05,
      "model_forward_time": 1.750354528427124,
      "step": 93618
    },
    {
      "epoch": 1.870918273925781e-05,
      "step": 93618,
      "training_step_time": 7.560412883758545
    },
    {
      "epoch": 1.871013641357422e-05,
      "model_forward_time": 1.7522401809692383,
      "step": 93619
    },
    {
      "epoch": 1.871013641357422e-05,
      "step": 93619,
      "training_step_time": 7.64319920539856
    },
    {
      "epoch": 1.8711090087890626e-05,
      "grad_norm": 0.1660119742155075,
      "learning_rate": 1.109067359696131e-07,
      "loss": 0.0552,
      "step": 93620
    },
    {
      "epoch": 1.8711090087890626e-05,
      "model_forward_time": 1.747532844543457,
      "step": 93620
    },
    {
      "dynamics_loss_avg": 0.03749810941517353,
      "epoch": 1.8711090087890626e-05,
      "step": 93620
    },
    {
      "action_loss_avg": 0.0014289569982793183,
      "epoch": 1.8711090087890626e-05,
      "step": 93620
    },
    {
      "epoch": 1.8711090087890626e-05,
      "step": 93620,
      "training_step_time": 7.694929122924805
    },
    {
      "epoch": 1.871204376220703e-05,
      "model_forward_time": 1.3760523796081543,
      "step": 93621
    },
    {
      "epoch": 1.871204376220703e-05,
      "step": 93621,
      "training_step_time": 8.137019634246826
    },
    {
      "epoch": 1.8712997436523437e-05,
      "model_forward_time": 1.750645637512207,
      "step": 93622
    },
    {
      "epoch": 1.8712997436523437e-05,
      "step": 93622,
      "training_step_time": 7.685046434402466
    },
    {
      "epoch": 1.8713951110839845e-05,
      "model_forward_time": 1.7765517234802246,
      "step": 93623
    },
    {
      "epoch": 1.8713951110839845e-05,
      "step": 93623,
      "training_step_time": 8.667348384857178
    },
    {
      "epoch": 1.871490478515625e-05,
      "model_forward_time": 1.03690505027771,
      "step": 93624
    },
    {
      "epoch": 1.871490478515625e-05,
      "step": 93624,
      "training_step_time": 7.599555969238281
    },
    {
      "epoch": 1.8715858459472656e-05,
      "model_forward_time": 2.886129379272461,
      "step": 93625
    },
    {
      "epoch": 1.8715858459472656e-05,
      "step": 93625,
      "training_step_time": 8.378881931304932
    },
    {
      "epoch": 1.8716812133789064e-05,
      "model_forward_time": 1.7483055591583252,
      "step": 93626
    },
    {
      "epoch": 1.8716812133789064e-05,
      "step": 93626,
      "training_step_time": 7.49205756187439
    },
    {
      "epoch": 1.8717765808105468e-05,
      "model_forward_time": 1.7533938884735107,
      "step": 93627
    },
    {
      "epoch": 1.8717765808105468e-05,
      "step": 93627,
      "training_step_time": 8.69753384590149
    },
    {
      "epoch": 1.8718719482421875e-05,
      "model_forward_time": 1.7664945125579834,
      "step": 93628
    },
    {
      "epoch": 1.8718719482421875e-05,
      "step": 93628,
      "training_step_time": 7.676529407501221
    },
    {
      "epoch": 1.8719673156738283e-05,
      "model_forward_time": 1.7510795593261719,
      "step": 93629
    },
    {
      "epoch": 1.8719673156738283e-05,
      "step": 93629,
      "training_step_time": 8.630557537078857
    },
    {
      "epoch": 1.8720626831054687e-05,
      "grad_norm": 0.08032896369695663,
      "learning_rate": 1.1056067860859165e-07,
      "loss": 0.0574,
      "step": 93630
    },
    {
      "epoch": 1.8720626831054687e-05,
      "model_forward_time": 1.749988079071045,
      "step": 93630
    },
    {
      "dynamics_loss_avg": 0.06873879451304674,
      "epoch": 1.8720626831054687e-05,
      "step": 93630
    },
    {
      "action_loss_avg": 0.0020065417716978117,
      "epoch": 1.8720626831054687e-05,
      "step": 93630
    },
    {
      "epoch": 1.8720626831054687e-05,
      "step": 93630,
      "training_step_time": 9.535114526748657
    },
    {
      "epoch": 1.8721580505371094e-05,
      "model_forward_time": 1.766517162322998,
      "step": 93631
    },
    {
      "epoch": 1.8721580505371094e-05,
      "step": 93631,
      "training_step_time": 8.446900606155396
    },
    {
      "epoch": 1.87225341796875e-05,
      "model_forward_time": 1.764737606048584,
      "step": 93632
    },
    {
      "epoch": 1.87225341796875e-05,
      "step": 93632,
      "training_step_time": 9.706658601760864
    },
    {
      "epoch": 1.8723487854003905e-05,
      "model_forward_time": 1.7398085594177246,
      "step": 93633
    },
    {
      "epoch": 1.8723487854003905e-05,
      "step": 93633,
      "training_step_time": 7.728092432022095
    },
    {
      "epoch": 1.8724441528320313e-05,
      "model_forward_time": 2.892871856689453,
      "step": 93634
    },
    {
      "epoch": 1.8724441528320313e-05,
      "step": 93634,
      "training_step_time": 8.540021896362305
    },
    {
      "epoch": 1.872539520263672e-05,
      "model_forward_time": 1.754796028137207,
      "step": 93635
    },
    {
      "epoch": 1.872539520263672e-05,
      "step": 93635,
      "training_step_time": 8.55988883972168
    },
    {
      "epoch": 1.8726348876953124e-05,
      "model_forward_time": 1.7490813732147217,
      "step": 93636
    },
    {
      "epoch": 1.8726348876953124e-05,
      "step": 93636,
      "training_step_time": 7.795414447784424
    },
    {
      "epoch": 1.872730255126953e-05,
      "model_forward_time": 1.7614853382110596,
      "step": 93637
    },
    {
      "epoch": 1.872730255126953e-05,
      "step": 93637,
      "training_step_time": 7.813167572021484
    },
    {
      "epoch": 1.872825622558594e-05,
      "model_forward_time": 1.7518281936645508,
      "step": 93638
    },
    {
      "epoch": 1.872825622558594e-05,
      "step": 93638,
      "training_step_time": 8.610716819763184
    },
    {
      "epoch": 1.8729209899902343e-05,
      "model_forward_time": 1.7520065307617188,
      "step": 93639
    },
    {
      "epoch": 1.8729209899902343e-05,
      "step": 93639,
      "training_step_time": 8.942261934280396
    },
    {
      "epoch": 1.873016357421875e-05,
      "grad_norm": 0.06884491443634033,
      "learning_rate": 1.102151559493031e-07,
      "loss": 0.058,
      "step": 93640
    },
    {
      "epoch": 1.873016357421875e-05,
      "model_forward_time": 1.7774193286895752,
      "step": 93640
    },
    {
      "dynamics_loss_avg": 0.05371853355318308,
      "epoch": 1.873016357421875e-05,
      "step": 93640
    },
    {
      "action_loss_avg": 0.0010978676917147822,
      "epoch": 1.873016357421875e-05,
      "step": 93640
    },
    {
      "epoch": 1.873016357421875e-05,
      "step": 93640,
      "training_step_time": 8.719212055206299
    },
    {
      "epoch": 1.8731117248535155e-05,
      "model_forward_time": 1.7726566791534424,
      "step": 93641
    },
    {
      "epoch": 1.8731117248535155e-05,
      "step": 93641,
      "training_step_time": 9.426053524017334
    },
    {
      "epoch": 1.8732070922851562e-05,
      "model_forward_time": 1.757565975189209,
      "step": 93642
    },
    {
      "epoch": 1.8732070922851562e-05,
      "step": 93642,
      "training_step_time": 7.862285375595093
    },
    {
      "epoch": 1.873302459716797e-05,
      "model_forward_time": 2.8727457523345947,
      "step": 93643
    },
    {
      "epoch": 1.873302459716797e-05,
      "step": 93643,
      "training_step_time": 8.586376428604126
    },
    {
      "epoch": 1.8733978271484373e-05,
      "model_forward_time": 1.7498295307159424,
      "step": 93644
    },
    {
      "epoch": 1.8733978271484373e-05,
      "step": 93644,
      "training_step_time": 7.616980075836182
    },
    {
      "epoch": 1.873493194580078e-05,
      "model_forward_time": 1.7561988830566406,
      "step": 93645
    },
    {
      "epoch": 1.873493194580078e-05,
      "step": 93645,
      "training_step_time": 7.783984661102295
    },
    {
      "epoch": 1.8735885620117188e-05,
      "model_forward_time": 1.7480111122131348,
      "step": 93646
    },
    {
      "epoch": 1.8735885620117188e-05,
      "step": 93646,
      "training_step_time": 7.710561990737915
    },
    {
      "epoch": 1.8736839294433592e-05,
      "model_forward_time": 1.7510838508605957,
      "step": 93647
    },
    {
      "epoch": 1.8736839294433592e-05,
      "step": 93647,
      "training_step_time": 7.759319543838501
    },
    {
      "epoch": 1.873779296875e-05,
      "model_forward_time": 1.7544221878051758,
      "step": 93648
    },
    {
      "epoch": 1.873779296875e-05,
      "step": 93648,
      "training_step_time": 7.8163392543792725
    },
    {
      "epoch": 1.8738746643066407e-05,
      "model_forward_time": 1.7524960041046143,
      "step": 93649
    },
    {
      "epoch": 1.8738746643066407e-05,
      "step": 93649,
      "training_step_time": 8.565613746643066
    },
    {
      "epoch": 1.873970031738281e-05,
      "grad_norm": 0.10970786958932877,
      "learning_rate": 1.0987016802953332e-07,
      "loss": 0.0566,
      "step": 93650
    },
    {
      "epoch": 1.873970031738281e-05,
      "model_forward_time": 1.750556230545044,
      "step": 93650
    },
    {
      "dynamics_loss_avg": 0.04328495170921087,
      "epoch": 1.873970031738281e-05,
      "step": 93650
    },
    {
      "action_loss_avg": 0.0014665386312117334,
      "epoch": 1.873970031738281e-05,
      "step": 93650
    },
    {
      "epoch": 1.873970031738281e-05,
      "step": 93650,
      "training_step_time": 7.83269190788269
    },
    {
      "epoch": 1.874065399169922e-05,
      "model_forward_time": 1.7434606552124023,
      "step": 93651
    },
    {
      "epoch": 1.874065399169922e-05,
      "step": 93651,
      "training_step_time": 10.576521635055542
    },
    {
      "epoch": 1.8741607666015626e-05,
      "model_forward_time": 2.817108392715454,
      "step": 93652
    },
    {
      "epoch": 1.8741607666015626e-05,
      "step": 93652,
      "training_step_time": 8.682677745819092
    },
    {
      "epoch": 1.874256134033203e-05,
      "model_forward_time": 1.7403974533081055,
      "step": 93653
    },
    {
      "epoch": 1.874256134033203e-05,
      "step": 93653,
      "training_step_time": 8.219024658203125
    },
    {
      "epoch": 1.8743515014648437e-05,
      "model_forward_time": 1.7411918640136719,
      "step": 93654
    },
    {
      "epoch": 1.8743515014648437e-05,
      "step": 93654,
      "training_step_time": 8.692576885223389
    },
    {
      "epoch": 1.8744468688964845e-05,
      "model_forward_time": 1.740978479385376,
      "step": 93655
    },
    {
      "epoch": 1.8744468688964845e-05,
      "step": 93655,
      "training_step_time": 7.677116394042969
    },
    {
      "epoch": 1.874542236328125e-05,
      "model_forward_time": 1.742255449295044,
      "step": 93656
    },
    {
      "epoch": 1.874542236328125e-05,
      "step": 93656,
      "training_step_time": 8.852261304855347
    },
    {
      "epoch": 1.8746376037597656e-05,
      "model_forward_time": 1.7424147129058838,
      "step": 93657
    },
    {
      "epoch": 1.8746376037597656e-05,
      "step": 93657,
      "training_step_time": 8.70096230506897
    },
    {
      "epoch": 1.8747329711914064e-05,
      "model_forward_time": 1.7420876026153564,
      "step": 93658
    },
    {
      "epoch": 1.8747329711914064e-05,
      "step": 93658,
      "training_step_time": 8.366981267929077
    },
    {
      "epoch": 1.8748283386230468e-05,
      "model_forward_time": 1.739731788635254,
      "step": 93659
    },
    {
      "epoch": 1.8748283386230468e-05,
      "step": 93659,
      "training_step_time": 9.476139545440674
    },
    {
      "epoch": 1.8749237060546875e-05,
      "grad_norm": 0.09377835690975189,
      "learning_rate": 1.0952571488701047e-07,
      "loss": 0.0582,
      "step": 93660
    },
    {
      "epoch": 1.8749237060546875e-05,
      "model_forward_time": 1.7491893768310547,
      "step": 93660
    },
    {
      "dynamics_loss_avg": 0.04270789185538888,
      "epoch": 1.8749237060546875e-05,
      "step": 93660
    },
    {
      "action_loss_avg": 0.001189290903857909,
      "epoch": 1.8749237060546875e-05,
      "step": 93660
    },
    {
      "epoch": 1.8749237060546875e-05,
      "step": 93660,
      "training_step_time": 8.636441707611084
    },
    {
      "epoch": 1.8750190734863283e-05,
      "model_forward_time": 2.8007824420928955,
      "step": 93661
    },
    {
      "epoch": 1.8750190734863283e-05,
      "step": 93661,
      "training_step_time": 8.277381896972656
    },
    {
      "epoch": 1.8751144409179687e-05,
      "model_forward_time": 1.7430405616760254,
      "step": 93662
    },
    {
      "epoch": 1.8751144409179687e-05,
      "step": 93662,
      "training_step_time": 8.975467920303345
    },
    {
      "epoch": 1.8752098083496094e-05,
      "model_forward_time": 1.7352776527404785,
      "step": 93663
    },
    {
      "epoch": 1.8752098083496094e-05,
      "step": 93663,
      "training_step_time": 7.5650293827056885
    },
    {
      "epoch": 1.87530517578125e-05,
      "model_forward_time": 1.7411575317382812,
      "step": 93664
    },
    {
      "epoch": 1.87530517578125e-05,
      "step": 93664,
      "training_step_time": 7.603002071380615
    },
    {
      "epoch": 1.8754005432128905e-05,
      "model_forward_time": 1.7427234649658203,
      "step": 93665
    },
    {
      "epoch": 1.8754005432128905e-05,
      "step": 93665,
      "training_step_time": 7.5871522426605225
    },
    {
      "epoch": 1.8754959106445313e-05,
      "model_forward_time": 1.748567819595337,
      "step": 93666
    },
    {
      "epoch": 1.8754959106445313e-05,
      "step": 93666,
      "training_step_time": 8.70202374458313
    },
    {
      "epoch": 1.875591278076172e-05,
      "model_forward_time": 1.7541112899780273,
      "step": 93667
    },
    {
      "epoch": 1.875591278076172e-05,
      "step": 93667,
      "training_step_time": 8.62392783164978
    },
    {
      "epoch": 1.8756866455078124e-05,
      "model_forward_time": 1.7856121063232422,
      "step": 93668
    },
    {
      "epoch": 1.8756866455078124e-05,
      "step": 93668,
      "training_step_time": 8.196993112564087
    },
    {
      "epoch": 1.8757820129394532e-05,
      "model_forward_time": 1.783139944076538,
      "step": 93669
    },
    {
      "epoch": 1.8757820129394532e-05,
      "step": 93669,
      "training_step_time": 7.664154052734375
    },
    {
      "epoch": 1.875877380371094e-05,
      "grad_norm": 0.08036676794290543,
      "learning_rate": 1.0918179655940165e-07,
      "loss": 0.0574,
      "step": 93670
    },
    {
      "epoch": 1.875877380371094e-05,
      "model_forward_time": 1.7529206275939941,
      "step": 93670
    },
    {
      "dynamics_loss_avg": 0.06662739105522633,
      "epoch": 1.875877380371094e-05,
      "step": 93670
    },
    {
      "action_loss_avg": 0.0014388085633981972,
      "epoch": 1.875877380371094e-05,
      "step": 93670
    },
    {
      "epoch": 1.875877380371094e-05,
      "step": 93670,
      "training_step_time": 7.58454966545105
    },
    {
      "epoch": 1.8759727478027343e-05,
      "model_forward_time": 2.873992919921875,
      "step": 93671
    },
    {
      "epoch": 1.8759727478027343e-05,
      "step": 93671,
      "training_step_time": 9.004981517791748
    },
    {
      "epoch": 1.876068115234375e-05,
      "model_forward_time": 1.7450659275054932,
      "step": 93672
    },
    {
      "epoch": 1.876068115234375e-05,
      "step": 93672,
      "training_step_time": 7.346155643463135
    },
    {
      "epoch": 1.8761634826660155e-05,
      "model_forward_time": 1.7566843032836914,
      "step": 93673
    },
    {
      "epoch": 1.8761634826660155e-05,
      "step": 93673,
      "training_step_time": 7.671509504318237
    },
    {
      "epoch": 1.8762588500976562e-05,
      "model_forward_time": 1.735189437866211,
      "step": 93674
    },
    {
      "epoch": 1.8762588500976562e-05,
      "step": 93674,
      "training_step_time": 8.057417154312134
    },
    {
      "epoch": 1.876354217529297e-05,
      "model_forward_time": 1.034776210784912,
      "step": 93675
    },
    {
      "epoch": 1.876354217529297e-05,
      "step": 93675,
      "training_step_time": 8.513573408126831
    },
    {
      "epoch": 1.8764495849609373e-05,
      "model_forward_time": 1.7455177307128906,
      "step": 93676
    },
    {
      "epoch": 1.8764495849609373e-05,
      "step": 93676,
      "training_step_time": 7.739564895629883
    },
    {
      "epoch": 1.876544952392578e-05,
      "model_forward_time": 1.7830214500427246,
      "step": 93677
    },
    {
      "epoch": 1.876544952392578e-05,
      "step": 93677,
      "training_step_time": 8.42897653579712
    },
    {
      "epoch": 1.8766403198242188e-05,
      "model_forward_time": 1.756727695465088,
      "step": 93678
    },
    {
      "epoch": 1.8766403198242188e-05,
      "step": 93678,
      "training_step_time": 9.045920133590698
    },
    {
      "epoch": 1.8767356872558592e-05,
      "model_forward_time": 1.7462949752807617,
      "step": 93679
    },
    {
      "epoch": 1.8767356872558592e-05,
      "step": 93679,
      "training_step_time": 7.563201427459717
    },
    {
      "epoch": 1.8768310546875e-05,
      "grad_norm": 0.12443732470273972,
      "learning_rate": 1.0883841308431841e-07,
      "loss": 0.0558,
      "step": 93680
    },
    {
      "epoch": 1.8768310546875e-05,
      "model_forward_time": 2.855053424835205,
      "step": 93680
    },
    {
      "dynamics_loss_avg": 0.056661184132099154,
      "epoch": 1.8768310546875e-05,
      "step": 93680
    },
    {
      "action_loss_avg": 0.0015485475189052523,
      "epoch": 1.8768310546875e-05,
      "step": 93680
    },
    {
      "epoch": 1.8768310546875e-05,
      "step": 93680,
      "training_step_time": 8.27838397026062
    },
    {
      "epoch": 1.8769264221191407e-05,
      "model_forward_time": 1.7505011558532715,
      "step": 93681
    },
    {
      "epoch": 1.8769264221191407e-05,
      "step": 93681,
      "training_step_time": 7.50764536857605
    },
    {
      "epoch": 1.877021789550781e-05,
      "model_forward_time": 1.7560701370239258,
      "step": 93682
    },
    {
      "epoch": 1.877021789550781e-05,
      "step": 93682,
      "training_step_time": 7.653588056564331
    },
    {
      "epoch": 1.877117156982422e-05,
      "model_forward_time": 1.7531194686889648,
      "step": 93683
    },
    {
      "epoch": 1.877117156982422e-05,
      "step": 93683,
      "training_step_time": 7.6017255783081055
    },
    {
      "epoch": 1.8772125244140626e-05,
      "model_forward_time": 1.753624439239502,
      "step": 93684
    },
    {
      "epoch": 1.8772125244140626e-05,
      "step": 93684,
      "training_step_time": 7.738065958023071
    },
    {
      "epoch": 1.877307891845703e-05,
      "model_forward_time": 1.7286860942840576,
      "step": 93685
    },
    {
      "epoch": 1.877307891845703e-05,
      "step": 93685,
      "training_step_time": 7.737058162689209
    },
    {
      "epoch": 1.8774032592773437e-05,
      "model_forward_time": 1.7761776447296143,
      "step": 93686
    },
    {
      "epoch": 1.8774032592773437e-05,
      "step": 93686,
      "training_step_time": 10.680163860321045
    },
    {
      "epoch": 1.8774986267089845e-05,
      "model_forward_time": 1.7527813911437988,
      "step": 93687
    },
    {
      "epoch": 1.8774986267089845e-05,
      "step": 93687,
      "training_step_time": 7.5993547439575195
    },
    {
      "epoch": 1.877593994140625e-05,
      "model_forward_time": 1.7467153072357178,
      "step": 93688
    },
    {
      "epoch": 1.877593994140625e-05,
      "step": 93688,
      "training_step_time": 8.540367603302002
    },
    {
      "epoch": 1.8776893615722656e-05,
      "model_forward_time": 2.823927640914917,
      "step": 93689
    },
    {
      "epoch": 1.8776893615722656e-05,
      "step": 93689,
      "training_step_time": 7.659940481185913
    },
    {
      "epoch": 1.8777847290039064e-05,
      "grad_norm": 0.14695627987384796,
      "learning_rate": 1.0849556449931298e-07,
      "loss": 0.0567,
      "step": 93690
    },
    {
      "epoch": 1.8777847290039064e-05,
      "model_forward_time": 1.7500603199005127,
      "step": 93690
    },
    {
      "dynamics_loss_avg": 0.04862688593566418,
      "epoch": 1.8777847290039064e-05,
      "step": 93690
    },
    {
      "action_loss_avg": 0.0013890664966311305,
      "epoch": 1.8777847290039064e-05,
      "step": 93690
    },
    {
      "epoch": 1.8777847290039064e-05,
      "step": 93690,
      "training_step_time": 7.429889440536499
    },
    {
      "epoch": 1.8778800964355468e-05,
      "model_forward_time": 1.747558832168579,
      "step": 93691
    },
    {
      "epoch": 1.8778800964355468e-05,
      "step": 93691,
      "training_step_time": 7.610012531280518
    },
    {
      "epoch": 1.8779754638671875e-05,
      "model_forward_time": 1.7521474361419678,
      "step": 93692
    },
    {
      "epoch": 1.8779754638671875e-05,
      "step": 93692,
      "training_step_time": 7.660983085632324
    },
    {
      "epoch": 1.8780708312988283e-05,
      "model_forward_time": 1.750664472579956,
      "step": 93693
    },
    {
      "epoch": 1.8780708312988283e-05,
      "step": 93693,
      "training_step_time": 8.878881216049194
    },
    {
      "epoch": 1.8781661987304687e-05,
      "model_forward_time": 1.0294368267059326,
      "step": 93694
    },
    {
      "epoch": 1.8781661987304687e-05,
      "step": 93694,
      "training_step_time": 8.623836517333984
    },
    {
      "epoch": 1.8782615661621094e-05,
      "model_forward_time": 1.7506394386291504,
      "step": 93695
    },
    {
      "epoch": 1.8782615661621094e-05,
      "step": 93695,
      "training_step_time": 8.706080675125122
    },
    {
      "epoch": 1.87835693359375e-05,
      "model_forward_time": 1.767467737197876,
      "step": 93696
    },
    {
      "epoch": 1.87835693359375e-05,
      "step": 93696,
      "training_step_time": 7.657699108123779
    },
    {
      "epoch": 1.8784523010253905e-05,
      "model_forward_time": 1.7590961456298828,
      "step": 93697
    },
    {
      "epoch": 1.8784523010253905e-05,
      "step": 93697,
      "training_step_time": 8.539402723312378
    },
    {
      "epoch": 1.8785476684570313e-05,
      "model_forward_time": 1.7493879795074463,
      "step": 93698
    },
    {
      "epoch": 1.8785476684570313e-05,
      "step": 93698,
      "training_step_time": 8.340192317962646
    },
    {
      "epoch": 1.878643035888672e-05,
      "model_forward_time": 2.8588874340057373,
      "step": 93699
    },
    {
      "epoch": 1.878643035888672e-05,
      "step": 93699,
      "training_step_time": 8.702095031738281
    },
    {
      "epoch": 1.8787384033203124e-05,
      "grad_norm": 0.08152344077825546,
      "learning_rate": 1.0815325084187867e-07,
      "loss": 0.0565,
      "step": 93700
    },
    {
      "epoch": 1.8787384033203124e-05,
      "model_forward_time": 1.7485847473144531,
      "step": 93700
    },
    {
      "dynamics_loss_avg": 0.05710589671507478,
      "epoch": 1.8787384033203124e-05,
      "step": 93700
    },
    {
      "action_loss_avg": 0.0021684668521629647,
      "epoch": 1.8787384033203124e-05,
      "step": 93700
    },
    {
      "epoch": 1.8787384033203124e-05,
      "step": 93700,
      "training_step_time": 7.588866710662842
    },
    {
      "epoch": 1.8788337707519532e-05,
      "model_forward_time": 1.747833251953125,
      "step": 93701
    },
    {
      "epoch": 1.8788337707519532e-05,
      "step": 93701,
      "training_step_time": 8.610293626785278
    },
    {
      "epoch": 1.878929138183594e-05,
      "model_forward_time": 1.739427089691162,
      "step": 93702
    },
    {
      "epoch": 1.878929138183594e-05,
      "step": 93702,
      "training_step_time": 8.14274263381958
    },
    {
      "epoch": 1.8790245056152343e-05,
      "model_forward_time": 1.380854606628418,
      "step": 93703
    },
    {
      "epoch": 1.8790245056152343e-05,
      "step": 93703,
      "training_step_time": 8.282875537872314
    },
    {
      "epoch": 1.879119873046875e-05,
      "model_forward_time": 1.7507193088531494,
      "step": 93704
    },
    {
      "epoch": 1.879119873046875e-05,
      "step": 93704,
      "training_step_time": 8.242470502853394
    },
    {
      "epoch": 1.8792152404785155e-05,
      "model_forward_time": 1.7756145000457764,
      "step": 93705
    },
    {
      "epoch": 1.8792152404785155e-05,
      "step": 93705,
      "training_step_time": 7.652250289916992
    },
    {
      "epoch": 1.8793106079101562e-05,
      "model_forward_time": 1.786956787109375,
      "step": 93706
    },
    {
      "epoch": 1.8793106079101562e-05,
      "step": 93706,
      "training_step_time": 9.563257455825806
    },
    {
      "epoch": 1.879405975341797e-05,
      "model_forward_time": 1.7543420791625977,
      "step": 93707
    },
    {
      "epoch": 1.879405975341797e-05,
      "step": 93707,
      "training_step_time": 7.644476413726807
    },
    {
      "epoch": 1.8795013427734373e-05,
      "model_forward_time": 2.877138376235962,
      "step": 93708
    },
    {
      "epoch": 1.8795013427734373e-05,
      "step": 93708,
      "training_step_time": 7.687134265899658
    },
    {
      "epoch": 1.879596710205078e-05,
      "model_forward_time": 1.7613468170166016,
      "step": 93709
    },
    {
      "epoch": 1.879596710205078e-05,
      "step": 93709,
      "training_step_time": 8.069472551345825
    },
    {
      "epoch": 1.879692077636719e-05,
      "grad_norm": 0.13937562704086304,
      "learning_rate": 1.0781147214944942e-07,
      "loss": 0.0567,
      "step": 93710
    },
    {
      "epoch": 1.879692077636719e-05,
      "model_forward_time": 1.7508561611175537,
      "step": 93710
    },
    {
      "dynamics_loss_avg": 0.05761935990303755,
      "epoch": 1.879692077636719e-05,
      "step": 93710
    },
    {
      "action_loss_avg": 0.0015816549537703394,
      "epoch": 1.879692077636719e-05,
      "step": 93710
    },
    {
      "epoch": 1.879692077636719e-05,
      "step": 93710,
      "training_step_time": 7.634829044342041
    },
    {
      "epoch": 1.8797874450683592e-05,
      "model_forward_time": 1.7597153186798096,
      "step": 93711
    },
    {
      "epoch": 1.8797874450683592e-05,
      "step": 93711,
      "training_step_time": 8.631970405578613
    },
    {
      "epoch": 1.8798828125e-05,
      "model_forward_time": 1.7523353099822998,
      "step": 93712
    },
    {
      "epoch": 1.8798828125e-05,
      "step": 93712,
      "training_step_time": 7.638880729675293
    },
    {
      "epoch": 1.8799781799316407e-05,
      "model_forward_time": 1.7481019496917725,
      "step": 93713
    },
    {
      "epoch": 1.8799781799316407e-05,
      "step": 93713,
      "training_step_time": 9.214831590652466
    },
    {
      "epoch": 1.880073547363281e-05,
      "model_forward_time": 1.7647204399108887,
      "step": 93714
    },
    {
      "epoch": 1.880073547363281e-05,
      "step": 93714,
      "training_step_time": 7.6622490882873535
    },
    {
      "epoch": 1.880168914794922e-05,
      "model_forward_time": 1.764845848083496,
      "step": 93715
    },
    {
      "epoch": 1.880168914794922e-05,
      "step": 93715,
      "training_step_time": 8.558218717575073
    },
    {
      "epoch": 1.8802642822265626e-05,
      "model_forward_time": 1.742689609527588,
      "step": 93716
    },
    {
      "epoch": 1.8802642822265626e-05,
      "step": 93716,
      "training_step_time": 8.618757963180542
    },
    {
      "epoch": 1.880359649658203e-05,
      "model_forward_time": 2.8143322467803955,
      "step": 93717
    },
    {
      "epoch": 1.880359649658203e-05,
      "step": 93717,
      "training_step_time": 7.62301778793335
    },
    {
      "epoch": 1.8804550170898437e-05,
      "model_forward_time": 1.742011547088623,
      "step": 93718
    },
    {
      "epoch": 1.8804550170898437e-05,
      "step": 93718,
      "training_step_time": 8.246252059936523
    },
    {
      "epoch": 1.8805503845214845e-05,
      "model_forward_time": 1.7485511302947998,
      "step": 93719
    },
    {
      "epoch": 1.8805503845214845e-05,
      "step": 93719,
      "training_step_time": 7.654731750488281
    },
    {
      "epoch": 1.880645751953125e-05,
      "grad_norm": 0.09967547655105591,
      "learning_rate": 1.0747022845940313e-07,
      "loss": 0.0577,
      "step": 93720
    },
    {
      "epoch": 1.880645751953125e-05,
      "model_forward_time": 1.7389392852783203,
      "step": 93720
    },
    {
      "dynamics_loss_avg": 0.06598262642510236,
      "epoch": 1.880645751953125e-05,
      "step": 93720
    },
    {
      "action_loss_avg": 0.0014152325413306244,
      "epoch": 1.880645751953125e-05,
      "step": 93720
    },
    {
      "epoch": 1.880645751953125e-05,
      "step": 93720,
      "training_step_time": 7.647597789764404
    },
    {
      "epoch": 1.8807411193847656e-05,
      "model_forward_time": 1.7465651035308838,
      "step": 93721
    },
    {
      "epoch": 1.8807411193847656e-05,
      "step": 93721,
      "training_step_time": 9.153968572616577
    },
    {
      "epoch": 1.8808364868164064e-05,
      "model_forward_time": 1.7340819835662842,
      "step": 93722
    },
    {
      "epoch": 1.8808364868164064e-05,
      "step": 93722,
      "training_step_time": 10.154840469360352
    },
    {
      "epoch": 1.8809318542480468e-05,
      "model_forward_time": 1.7612202167510986,
      "step": 93723
    },
    {
      "epoch": 1.8809318542480468e-05,
      "step": 93723,
      "training_step_time": 8.528767108917236
    },
    {
      "epoch": 1.8810272216796875e-05,
      "model_forward_time": 1.7542033195495605,
      "step": 93724
    },
    {
      "epoch": 1.8810272216796875e-05,
      "step": 93724,
      "training_step_time": 8.375568866729736
    },
    {
      "epoch": 1.8811225891113283e-05,
      "model_forward_time": 1.7420578002929688,
      "step": 93725
    },
    {
      "epoch": 1.8811225891113283e-05,
      "step": 93725,
      "training_step_time": 8.51720905303955
    },
    {
      "epoch": 1.8812179565429687e-05,
      "model_forward_time": 1.7420177459716797,
      "step": 93726
    },
    {
      "epoch": 1.8812179565429687e-05,
      "step": 93726,
      "training_step_time": 8.749919652938843
    },
    {
      "epoch": 1.8813133239746094e-05,
      "model_forward_time": 1.742955207824707,
      "step": 93727
    },
    {
      "epoch": 1.8813133239746094e-05,
      "step": 93727,
      "training_step_time": 7.410799503326416
    },
    {
      "epoch": 1.88140869140625e-05,
      "model_forward_time": 1.7429065704345703,
      "step": 93728
    },
    {
      "epoch": 1.88140869140625e-05,
      "step": 93728,
      "training_step_time": 7.6385557651519775
    },
    {
      "epoch": 1.8815040588378905e-05,
      "model_forward_time": 1.7454485893249512,
      "step": 93729
    },
    {
      "epoch": 1.8815040588378905e-05,
      "step": 93729,
      "training_step_time": 8.389740228652954
    },
    {
      "epoch": 1.8815994262695313e-05,
      "grad_norm": 0.07109495997428894,
      "learning_rate": 1.0712951980905606e-07,
      "loss": 0.0554,
      "step": 93730
    },
    {
      "epoch": 1.8815994262695313e-05,
      "model_forward_time": 1.747206687927246,
      "step": 93730
    },
    {
      "dynamics_loss_avg": 0.06303042043000459,
      "epoch": 1.8815994262695313e-05,
      "step": 93730
    },
    {
      "action_loss_avg": 0.0015789169949130156,
      "epoch": 1.8815994262695313e-05,
      "step": 93730
    },
    {
      "epoch": 1.8815994262695313e-05,
      "step": 93730,
      "training_step_time": 7.625164031982422
    },
    {
      "epoch": 1.881694793701172e-05,
      "model_forward_time": 1.743598222732544,
      "step": 93731
    },
    {
      "epoch": 1.881694793701172e-05,
      "step": 93731,
      "training_step_time": 7.692205429077148
    },
    {
      "epoch": 1.8817901611328124e-05,
      "model_forward_time": 1.7613253593444824,
      "step": 93732
    },
    {
      "epoch": 1.8817901611328124e-05,
      "step": 93732,
      "training_step_time": 10.501989841461182
    },
    {
      "epoch": 1.8818855285644532e-05,
      "model_forward_time": 1.7558422088623047,
      "step": 93733
    },
    {
      "epoch": 1.8818855285644532e-05,
      "step": 93733,
      "training_step_time": 8.554537057876587
    },
    {
      "epoch": 1.881980895996094e-05,
      "model_forward_time": 1.7478394508361816,
      "step": 93734
    },
    {
      "epoch": 1.881980895996094e-05,
      "step": 93734,
      "training_step_time": 7.692210912704468
    },
    {
      "epoch": 1.8820762634277343e-05,
      "model_forward_time": 1.741642951965332,
      "step": 93735
    },
    {
      "epoch": 1.8820762634277343e-05,
      "step": 93735,
      "training_step_time": 8.440275192260742
    },
    {
      "epoch": 1.882171630859375e-05,
      "model_forward_time": 2.446894407272339,
      "step": 93736
    },
    {
      "epoch": 1.882171630859375e-05,
      "step": 93736,
      "training_step_time": 7.342779874801636
    },
    {
      "epoch": 1.8822669982910155e-05,
      "model_forward_time": 1.7446446418762207,
      "step": 93737
    },
    {
      "epoch": 1.8822669982910155e-05,
      "step": 93737,
      "training_step_time": 8.416086673736572
    },
    {
      "epoch": 1.8823623657226562e-05,
      "model_forward_time": 1.7374281883239746,
      "step": 93738
    },
    {
      "epoch": 1.8823623657226562e-05,
      "step": 93738,
      "training_step_time": 7.551463603973389
    },
    {
      "epoch": 1.882457733154297e-05,
      "model_forward_time": 1.7441446781158447,
      "step": 93739
    },
    {
      "epoch": 1.882457733154297e-05,
      "step": 93739,
      "training_step_time": 7.554908990859985
    },
    {
      "epoch": 1.8825531005859374e-05,
      "grad_norm": 0.08587177097797394,
      "learning_rate": 1.0678934623566839e-07,
      "loss": 0.0574,
      "step": 93740
    },
    {
      "epoch": 1.8825531005859374e-05,
      "model_forward_time": 1.0359177589416504,
      "step": 93740
    },
    {
      "dynamics_loss_avg": 0.06693864166736603,
      "epoch": 1.8825531005859374e-05,
      "step": 93740
    },
    {
      "action_loss_avg": 0.001206385626574047,
      "epoch": 1.8825531005859374e-05,
      "step": 93740
    },
    {
      "epoch": 1.8825531005859374e-05,
      "step": 93740,
      "training_step_time": 8.633330345153809
    },
    {
      "epoch": 1.882648468017578e-05,
      "model_forward_time": 1.743515968322754,
      "step": 93741
    },
    {
      "epoch": 1.882648468017578e-05,
      "step": 93741,
      "training_step_time": 7.65923810005188
    },
    {
      "epoch": 1.882743835449219e-05,
      "model_forward_time": 1.7479197978973389,
      "step": 93742
    },
    {
      "epoch": 1.882743835449219e-05,
      "step": 93742,
      "training_step_time": 7.630439281463623
    },
    {
      "epoch": 1.8828392028808592e-05,
      "model_forward_time": 1.741955041885376,
      "step": 93743
    },
    {
      "epoch": 1.8828392028808592e-05,
      "step": 93743,
      "training_step_time": 9.657299757003784
    },
    {
      "epoch": 1.8829345703125e-05,
      "model_forward_time": 1.7434873580932617,
      "step": 93744
    },
    {
      "epoch": 1.8829345703125e-05,
      "step": 93744,
      "training_step_time": 7.735889196395874
    },
    {
      "epoch": 1.8830299377441407e-05,
      "model_forward_time": 2.805572271347046,
      "step": 93745
    },
    {
      "epoch": 1.8830299377441407e-05,
      "step": 93745,
      "training_step_time": 7.671457767486572
    },
    {
      "epoch": 1.883125305175781e-05,
      "model_forward_time": 1.744729995727539,
      "step": 93746
    },
    {
      "epoch": 1.883125305175781e-05,
      "step": 93746,
      "training_step_time": 7.694252967834473
    },
    {
      "epoch": 1.883220672607422e-05,
      "model_forward_time": 1.7401437759399414,
      "step": 93747
    },
    {
      "epoch": 1.883220672607422e-05,
      "step": 93747,
      "training_step_time": 7.693922281265259
    },
    {
      "epoch": 1.8833160400390626e-05,
      "model_forward_time": 1.7484335899353027,
      "step": 93748
    },
    {
      "epoch": 1.8833160400390626e-05,
      "step": 93748,
      "training_step_time": 9.817491054534912
    },
    {
      "epoch": 1.883411407470703e-05,
      "model_forward_time": 1.7362637519836426,
      "step": 93749
    },
    {
      "epoch": 1.883411407470703e-05,
      "step": 93749,
      "training_step_time": 8.663435697555542
    },
    {
      "epoch": 1.8835067749023438e-05,
      "grad_norm": 0.07423637807369232,
      "learning_rate": 1.0644970777644093e-07,
      "loss": 0.0546,
      "step": 93750
    },
    {
      "epoch": 1.8835067749023438e-05,
      "model_forward_time": 1.7399828433990479,
      "step": 93750
    },
    {
      "dynamics_loss_avg": 0.0525733795017004,
      "epoch": 1.8835067749023438e-05,
      "step": 93750
    },
    {
      "action_loss_avg": 0.0012417994847055524,
      "epoch": 1.8835067749023438e-05,
      "step": 93750
    },
    {
      "epoch": 1.8835067749023438e-05,
      "step": 93750,
      "training_step_time": 8.737504720687866
    },
    {
      "epoch": 1.8836021423339845e-05,
      "model_forward_time": 1.7753260135650635,
      "step": 93751
    },
    {
      "epoch": 1.8836021423339845e-05,
      "step": 93751,
      "training_step_time": 9.013290405273438
    },
    {
      "epoch": 1.883697509765625e-05,
      "model_forward_time": 1.745654821395874,
      "step": 93752
    },
    {
      "epoch": 1.883697509765625e-05,
      "step": 93752,
      "training_step_time": 7.708837509155273
    },
    {
      "epoch": 1.8837928771972656e-05,
      "model_forward_time": 1.757843017578125,
      "step": 93753
    },
    {
      "epoch": 1.8837928771972656e-05,
      "step": 93753,
      "training_step_time": 8.24098825454712
    },
    {
      "epoch": 1.8838882446289064e-05,
      "model_forward_time": 1.7532927989959717,
      "step": 93754
    },
    {
      "epoch": 1.8838882446289064e-05,
      "step": 93754,
      "training_step_time": 8.279675483703613
    },
    {
      "epoch": 1.8839836120605468e-05,
      "model_forward_time": 1.7438867092132568,
      "step": 93755
    },
    {
      "epoch": 1.8839836120605468e-05,
      "step": 93755,
      "training_step_time": 8.58886456489563
    },
    {
      "epoch": 1.8840789794921875e-05,
      "model_forward_time": 1.7575232982635498,
      "step": 93756
    },
    {
      "epoch": 1.8840789794921875e-05,
      "step": 93756,
      "training_step_time": 7.646551609039307
    },
    {
      "epoch": 1.8841743469238283e-05,
      "model_forward_time": 1.752807378768921,
      "step": 93757
    },
    {
      "epoch": 1.8841743469238283e-05,
      "step": 93757,
      "training_step_time": 8.532208442687988
    },
    {
      "epoch": 1.8842697143554687e-05,
      "model_forward_time": 1.7564780712127686,
      "step": 93758
    },
    {
      "epoch": 1.8842697143554687e-05,
      "step": 93758,
      "training_step_time": 7.631516456604004
    },
    {
      "epoch": 1.8843650817871094e-05,
      "model_forward_time": 1.7545807361602783,
      "step": 93759
    },
    {
      "epoch": 1.8843650817871094e-05,
      "step": 93759,
      "training_step_time": 11.18375015258789
    },
    {
      "epoch": 1.88446044921875e-05,
      "grad_norm": 0.08070579916238785,
      "learning_rate": 1.0611060446851674e-07,
      "loss": 0.0555,
      "step": 93760
    },
    {
      "epoch": 1.88446044921875e-05,
      "model_forward_time": 1.774040937423706,
      "step": 93760
    },
    {
      "dynamics_loss_avg": 0.06475728899240493,
      "epoch": 1.88446044921875e-05,
      "step": 93760
    },
    {
      "action_loss_avg": 0.0014705736859468744,
      "epoch": 1.88446044921875e-05,
      "step": 93760
    },
    {
      "epoch": 1.88446044921875e-05,
      "step": 93760,
      "training_step_time": 9.672498941421509
    },
    {
      "epoch": 1.8845558166503906e-05,
      "model_forward_time": 1.7805466651916504,
      "step": 93761
    },
    {
      "epoch": 1.8845558166503906e-05,
      "step": 93761,
      "training_step_time": 7.665232181549072
    },
    {
      "epoch": 1.8846511840820313e-05,
      "model_forward_time": 1.7530772686004639,
      "step": 93762
    },
    {
      "epoch": 1.8846511840820313e-05,
      "step": 93762,
      "training_step_time": 7.656378746032715
    },
    {
      "epoch": 1.884746551513672e-05,
      "model_forward_time": 1.7565357685089111,
      "step": 93763
    },
    {
      "epoch": 1.884746551513672e-05,
      "step": 93763,
      "training_step_time": 8.250513315200806
    },
    {
      "epoch": 1.8848419189453124e-05,
      "model_forward_time": 2.905524492263794,
      "step": 93764
    },
    {
      "epoch": 1.8848419189453124e-05,
      "step": 93764,
      "training_step_time": 7.75216269493103
    },
    {
      "epoch": 1.8849372863769532e-05,
      "model_forward_time": 1.749631404876709,
      "step": 93765
    },
    {
      "epoch": 1.8849372863769532e-05,
      "step": 93765,
      "training_step_time": 8.471585035324097
    },
    {
      "epoch": 1.8850326538085936e-05,
      "model_forward_time": 1.0570893287658691,
      "step": 93766
    },
    {
      "epoch": 1.8850326538085936e-05,
      "step": 93766,
      "training_step_time": 8.499532461166382
    },
    {
      "epoch": 1.8851280212402343e-05,
      "model_forward_time": 1.741776466369629,
      "step": 93767
    },
    {
      "epoch": 1.8851280212402343e-05,
      "step": 93767,
      "training_step_time": 8.403570413589478
    },
    {
      "epoch": 1.885223388671875e-05,
      "model_forward_time": 1.7539911270141602,
      "step": 93768
    },
    {
      "epoch": 1.885223388671875e-05,
      "step": 93768,
      "training_step_time": 9.655985355377197
    },
    {
      "epoch": 1.8853187561035155e-05,
      "model_forward_time": 1.7616469860076904,
      "step": 93769
    },
    {
      "epoch": 1.8853187561035155e-05,
      "step": 93769,
      "training_step_time": 9.511601209640503
    },
    {
      "epoch": 1.8854141235351562e-05,
      "grad_norm": 0.13189204037189484,
      "learning_rate": 1.0577203634897837e-07,
      "loss": 0.056,
      "step": 93770
    },
    {
      "epoch": 1.8854141235351562e-05,
      "model_forward_time": 1.773772954940796,
      "step": 93770
    },
    {
      "dynamics_loss_avg": 0.05373639864847064,
      "epoch": 1.8854141235351562e-05,
      "step": 93770
    },
    {
      "action_loss_avg": 0.0010012345504947008,
      "epoch": 1.8854141235351562e-05,
      "step": 93770
    },
    {
      "epoch": 1.8854141235351562e-05,
      "step": 93770,
      "training_step_time": 8.69115138053894
    },
    {
      "epoch": 1.885509490966797e-05,
      "model_forward_time": 1.7562150955200195,
      "step": 93771
    },
    {
      "epoch": 1.885509490966797e-05,
      "step": 93771,
      "training_step_time": 7.797640562057495
    },
    {
      "epoch": 1.8856048583984374e-05,
      "model_forward_time": 1.75071382522583,
      "step": 93772
    },
    {
      "epoch": 1.8856048583984374e-05,
      "step": 93772,
      "training_step_time": 7.745802402496338
    },
    {
      "epoch": 1.885700225830078e-05,
      "model_forward_time": 2.889308452606201,
      "step": 93773
    },
    {
      "epoch": 1.885700225830078e-05,
      "step": 93773,
      "training_step_time": 8.287461519241333
    },
    {
      "epoch": 1.885795593261719e-05,
      "model_forward_time": 1.7564365863800049,
      "step": 93774
    },
    {
      "epoch": 1.885795593261719e-05,
      "step": 93774,
      "training_step_time": 7.71950626373291
    },
    {
      "epoch": 1.8858909606933592e-05,
      "model_forward_time": 1.757521152496338,
      "step": 93775
    },
    {
      "epoch": 1.8858909606933592e-05,
      "step": 93775,
      "training_step_time": 7.748539924621582
    },
    {
      "epoch": 1.885986328125e-05,
      "model_forward_time": 1.7612383365631104,
      "step": 93776
    },
    {
      "epoch": 1.885986328125e-05,
      "step": 93776,
      "training_step_time": 7.760607481002808
    },
    {
      "epoch": 1.8860816955566407e-05,
      "model_forward_time": 1.749727487564087,
      "step": 93777
    },
    {
      "epoch": 1.8860816955566407e-05,
      "step": 93777,
      "training_step_time": 9.602574825286865
    },
    {
      "epoch": 1.886177062988281e-05,
      "model_forward_time": 1.780473232269287,
      "step": 93778
    },
    {
      "epoch": 1.886177062988281e-05,
      "step": 93778,
      "training_step_time": 8.812705278396606
    },
    {
      "epoch": 1.886272430419922e-05,
      "model_forward_time": 1.0498707294464111,
      "step": 93779
    },
    {
      "epoch": 1.886272430419922e-05,
      "step": 93779,
      "training_step_time": 7.700283527374268
    },
    {
      "epoch": 1.8863677978515626e-05,
      "grad_norm": 0.12607210874557495,
      "learning_rate": 1.0543400345485177e-07,
      "loss": 0.0545,
      "step": 93780
    },
    {
      "epoch": 1.8863677978515626e-05,
      "model_forward_time": 1.7507543563842773,
      "step": 93780
    },
    {
      "dynamics_loss_avg": 0.06972198896110057,
      "epoch": 1.8863677978515626e-05,
      "step": 93780
    },
    {
      "action_loss_avg": 0.001350429150625132,
      "epoch": 1.8863677978515626e-05,
      "step": 93780
    },
    {
      "epoch": 1.8863677978515626e-05,
      "step": 93780,
      "training_step_time": 9.46052861213684
    },
    {
      "epoch": 1.886463165283203e-05,
      "model_forward_time": 1.7542047500610352,
      "step": 93781
    },
    {
      "epoch": 1.886463165283203e-05,
      "step": 93781,
      "training_step_time": 8.770251989364624
    },
    {
      "epoch": 1.8865585327148438e-05,
      "model_forward_time": 1.7521467208862305,
      "step": 93782
    },
    {
      "epoch": 1.8865585327148438e-05,
      "step": 93782,
      "training_step_time": 7.502293825149536
    },
    {
      "epoch": 1.8866539001464845e-05,
      "model_forward_time": 2.887368679046631,
      "step": 93783
    },
    {
      "epoch": 1.8866539001464845e-05,
      "step": 93783,
      "training_step_time": 7.7487499713897705
    },
    {
      "epoch": 1.886749267578125e-05,
      "model_forward_time": 1.7464795112609863,
      "step": 93784
    },
    {
      "epoch": 1.886749267578125e-05,
      "step": 93784,
      "training_step_time": 7.689987421035767
    },
    {
      "epoch": 1.8868446350097656e-05,
      "model_forward_time": 2.848496913909912,
      "step": 93785
    },
    {
      "epoch": 1.8868446350097656e-05,
      "step": 93785,
      "training_step_time": 9.4409921169281
    },
    {
      "epoch": 1.8869400024414064e-05,
      "model_forward_time": 1.7412610054016113,
      "step": 93786
    },
    {
      "epoch": 1.8869400024414064e-05,
      "step": 93786,
      "training_step_time": 8.448874473571777
    },
    {
      "epoch": 1.8870353698730468e-05,
      "model_forward_time": 1.744006633758545,
      "step": 93787
    },
    {
      "epoch": 1.8870353698730468e-05,
      "step": 93787,
      "training_step_time": 6.795774698257446
    },
    {
      "epoch": 1.8871307373046875e-05,
      "model_forward_time": 1.7396924495697021,
      "step": 93788
    },
    {
      "epoch": 1.8871307373046875e-05,
      "step": 93788,
      "training_step_time": 7.618561744689941
    },
    {
      "epoch": 1.8872261047363283e-05,
      "model_forward_time": 1.7377443313598633,
      "step": 93789
    },
    {
      "epoch": 1.8872261047363283e-05,
      "step": 93789,
      "training_step_time": 7.511077642440796
    },
    {
      "epoch": 1.8873214721679687e-05,
      "grad_norm": 0.08326476067304611,
      "learning_rate": 1.0509650582310405e-07,
      "loss": 0.0549,
      "step": 93790
    },
    {
      "epoch": 1.8873214721679687e-05,
      "model_forward_time": 1.7444300651550293,
      "step": 93790
    },
    {
      "dynamics_loss_avg": 0.058619816228747366,
      "epoch": 1.8873214721679687e-05,
      "step": 93790
    },
    {
      "action_loss_avg": 0.002445580982021056,
      "epoch": 1.8873214721679687e-05,
      "step": 93790
    },
    {
      "epoch": 1.8873214721679687e-05,
      "step": 93790,
      "training_step_time": 6.799760580062866
    },
    {
      "epoch": 1.8874168395996094e-05,
      "model_forward_time": 1.75408935546875,
      "step": 93791
    },
    {
      "epoch": 1.8874168395996094e-05,
      "step": 93791,
      "training_step_time": 6.769997596740723
    },
    {
      "epoch": 1.88751220703125e-05,
      "model_forward_time": 1.7480497360229492,
      "step": 93792
    },
    {
      "epoch": 1.88751220703125e-05,
      "step": 93792,
      "training_step_time": 6.812167644500732
    },
    {
      "epoch": 1.8876075744628906e-05,
      "model_forward_time": 1.7537848949432373,
      "step": 93793
    },
    {
      "epoch": 1.8876075744628906e-05,
      "step": 93793,
      "training_step_time": 6.788046360015869
    },
    {
      "epoch": 1.8877029418945313e-05,
      "model_forward_time": 2.8419313430786133,
      "step": 93794
    },
    {
      "epoch": 1.8877029418945313e-05,
      "step": 93794,
      "training_step_time": 9.57685136795044
    },
    {
      "epoch": 1.887798309326172e-05,
      "model_forward_time": 1.7500171661376953,
      "step": 93795
    },
    {
      "epoch": 1.887798309326172e-05,
      "step": 93795,
      "training_step_time": 7.781774282455444
    },
    {
      "epoch": 1.8878936767578124e-05,
      "model_forward_time": 1.3860700130462646,
      "step": 93796
    },
    {
      "epoch": 1.8878936767578124e-05,
      "step": 93796,
      "training_step_time": 8.10224461555481
    },
    {
      "epoch": 1.8879890441894532e-05,
      "model_forward_time": 1.751741886138916,
      "step": 93797
    },
    {
      "epoch": 1.8879890441894532e-05,
      "step": 93797,
      "training_step_time": 8.17639422416687
    },
    {
      "epoch": 1.8880844116210936e-05,
      "model_forward_time": 1.7526249885559082,
      "step": 93798
    },
    {
      "epoch": 1.8880844116210936e-05,
      "step": 93798,
      "training_step_time": 7.653690576553345
    },
    {
      "epoch": 1.8881797790527343e-05,
      "model_forward_time": 1.764878511428833,
      "step": 93799
    },
    {
      "epoch": 1.8881797790527343e-05,
      "step": 93799,
      "training_step_time": 6.757627248764038
    },
    {
      "epoch": 1.888275146484375e-05,
      "grad_norm": 0.07777877897024155,
      "learning_rate": 1.0475954349064288e-07,
      "loss": 0.0546,
      "step": 93800
    },
    {
      "epoch": 1.888275146484375e-05,
      "model_forward_time": 1.7666747570037842,
      "step": 93800
    },
    {
      "dynamics_loss_avg": 0.04418116100132465,
      "epoch": 1.888275146484375e-05,
      "step": 93800
    },
    {
      "action_loss_avg": 0.0013273214572109282,
      "epoch": 1.888275146484375e-05,
      "step": 93800
    },
    {
      "epoch": 1.888275146484375e-05,
      "step": 93800,
      "training_step_time": 6.767232894897461
    },
    {
      "epoch": 1.8883705139160155e-05,
      "model_forward_time": 1.3964927196502686,
      "step": 93801
    },
    {
      "epoch": 1.8883705139160155e-05,
      "step": 93801,
      "training_step_time": 6.766674041748047
    },
    {
      "epoch": 1.8884658813476562e-05,
      "model_forward_time": 1.7600877285003662,
      "step": 93802
    },
    {
      "epoch": 1.8884658813476562e-05,
      "step": 93802,
      "training_step_time": 6.821296215057373
    },
    {
      "epoch": 1.888561248779297e-05,
      "model_forward_time": 2.9016127586364746,
      "step": 93803
    },
    {
      "epoch": 1.888561248779297e-05,
      "step": 93803,
      "training_step_time": 9.308104515075684
    },
    {
      "epoch": 1.8886566162109374e-05,
      "model_forward_time": 1.4071896076202393,
      "step": 93804
    },
    {
      "epoch": 1.8886566162109374e-05,
      "step": 93804,
      "training_step_time": 8.454062938690186
    },
    {
      "epoch": 1.888751983642578e-05,
      "model_forward_time": 1.7646119594573975,
      "step": 93805
    },
    {
      "epoch": 1.888751983642578e-05,
      "step": 93805,
      "training_step_time": 7.685236215591431
    },
    {
      "epoch": 1.888847351074219e-05,
      "model_forward_time": 1.7630276679992676,
      "step": 93806
    },
    {
      "epoch": 1.888847351074219e-05,
      "step": 93806,
      "training_step_time": 7.555969476699829
    },
    {
      "epoch": 1.8889427185058592e-05,
      "model_forward_time": 1.3946161270141602,
      "step": 93807
    },
    {
      "epoch": 1.8889427185058592e-05,
      "step": 93807,
      "training_step_time": 7.674066781997681
    },
    {
      "epoch": 1.8890380859375e-05,
      "model_forward_time": 1.7617778778076172,
      "step": 93808
    },
    {
      "epoch": 1.8890380859375e-05,
      "step": 93808,
      "training_step_time": 6.816954135894775
    },
    {
      "epoch": 1.8891334533691407e-05,
      "model_forward_time": 1.7696912288665771,
      "step": 93809
    },
    {
      "epoch": 1.8891334533691407e-05,
      "step": 93809,
      "training_step_time": 6.745862245559692
    },
    {
      "epoch": 1.889228820800781e-05,
      "grad_norm": 0.08255483210086823,
      "learning_rate": 1.0442311649431824e-07,
      "loss": 0.0538,
      "step": 93810
    },
    {
      "epoch": 1.889228820800781e-05,
      "model_forward_time": 1.771087884902954,
      "step": 93810
    },
    {
      "dynamics_loss_avg": 0.05053950138390064,
      "epoch": 1.889228820800781e-05,
      "step": 93810
    },
    {
      "action_loss_avg": 0.0012492907495470717,
      "epoch": 1.889228820800781e-05,
      "step": 93810
    },
    {
      "epoch": 1.889228820800781e-05,
      "step": 93810,
      "training_step_time": 6.742301940917969
    },
    {
      "epoch": 1.889324188232422e-05,
      "model_forward_time": 1.7659730911254883,
      "step": 93811
    },
    {
      "epoch": 1.889324188232422e-05,
      "step": 93811,
      "training_step_time": 6.751401424407959
    },
    {
      "epoch": 1.8894195556640626e-05,
      "model_forward_time": 1.8266849517822266,
      "step": 93812
    },
    {
      "epoch": 1.8894195556640626e-05,
      "step": 93812,
      "training_step_time": 7.788059949874878
    },
    {
      "epoch": 1.889514923095703e-05,
      "model_forward_time": 2.85191011428833,
      "step": 93813
    },
    {
      "epoch": 1.889514923095703e-05,
      "step": 93813,
      "training_step_time": 9.820530414581299
    },
    {
      "epoch": 1.8896102905273438e-05,
      "model_forward_time": 1.7663609981536865,
      "step": 93814
    },
    {
      "epoch": 1.8896102905273438e-05,
      "step": 93814,
      "training_step_time": 7.674314737319946
    },
    {
      "epoch": 1.8897056579589845e-05,
      "model_forward_time": 1.7684211730957031,
      "step": 93815
    },
    {
      "epoch": 1.8897056579589845e-05,
      "step": 93815,
      "training_step_time": 7.4512269496917725
    },
    {
      "epoch": 1.889801025390625e-05,
      "model_forward_time": 1.7707271575927734,
      "step": 93816
    },
    {
      "epoch": 1.889801025390625e-05,
      "step": 93816,
      "training_step_time": 7.634101152420044
    },
    {
      "epoch": 1.8898963928222656e-05,
      "model_forward_time": 1.7670717239379883,
      "step": 93817
    },
    {
      "epoch": 1.8898963928222656e-05,
      "step": 93817,
      "training_step_time": 6.749572038650513
    },
    {
      "epoch": 1.8899917602539064e-05,
      "model_forward_time": 1.7685739994049072,
      "step": 93818
    },
    {
      "epoch": 1.8899917602539064e-05,
      "step": 93818,
      "training_step_time": 6.724822521209717
    },
    {
      "epoch": 1.8900871276855468e-05,
      "model_forward_time": 1.3949365615844727,
      "step": 93819
    },
    {
      "epoch": 1.8900871276855468e-05,
      "step": 93819,
      "training_step_time": 6.6716649532318115
    },
    {
      "epoch": 1.8901824951171875e-05,
      "grad_norm": 0.09020344913005829,
      "learning_rate": 1.0408722487092071e-07,
      "loss": 0.0543,
      "step": 93820
    },
    {
      "epoch": 1.8901824951171875e-05,
      "model_forward_time": 1.7638332843780518,
      "step": 93820
    },
    {
      "dynamics_loss_avg": 0.054277567006647584,
      "epoch": 1.8901824951171875e-05,
      "step": 93820
    },
    {
      "action_loss_avg": 0.0007854345720261335,
      "epoch": 1.8901824951171875e-05,
      "step": 93820
    },
    {
      "epoch": 1.8901824951171875e-05,
      "step": 93820,
      "training_step_time": 6.717398405075073
    },
    {
      "epoch": 1.8902778625488283e-05,
      "model_forward_time": 1.8301992416381836,
      "step": 93821
    },
    {
      "epoch": 1.8902778625488283e-05,
      "step": 93821,
      "training_step_time": 8.486600160598755
    },
    {
      "epoch": 1.8903732299804687e-05,
      "model_forward_time": 2.8115339279174805,
      "step": 93822
    },
    {
      "epoch": 1.8903732299804687e-05,
      "step": 93822,
      "training_step_time": 9.733392715454102
    },
    {
      "epoch": 1.8904685974121094e-05,
      "model_forward_time": 1.7633826732635498,
      "step": 93823
    },
    {
      "epoch": 1.8904685974121094e-05,
      "step": 93823,
      "training_step_time": 7.765292644500732
    },
    {
      "epoch": 1.89056396484375e-05,
      "model_forward_time": 1.7622230052947998,
      "step": 93824
    },
    {
      "epoch": 1.89056396484375e-05,
      "step": 93824,
      "training_step_time": 7.506831169128418
    },
    {
      "epoch": 1.8906593322753906e-05,
      "model_forward_time": 1.763535737991333,
      "step": 93825
    },
    {
      "epoch": 1.8906593322753906e-05,
      "step": 93825,
      "training_step_time": 7.653052806854248
    },
    {
      "epoch": 1.8907546997070313e-05,
      "model_forward_time": 1.7621567249298096,
      "step": 93826
    },
    {
      "epoch": 1.8907546997070313e-05,
      "step": 93826,
      "training_step_time": 7.650820970535278
    },
    {
      "epoch": 1.890850067138672e-05,
      "model_forward_time": 1.7592658996582031,
      "step": 93827
    },
    {
      "epoch": 1.890850067138672e-05,
      "step": 93827,
      "training_step_time": 6.78469705581665
    },
    {
      "epoch": 1.8909454345703124e-05,
      "model_forward_time": 1.760833501815796,
      "step": 93828
    },
    {
      "epoch": 1.8909454345703124e-05,
      "step": 93828,
      "training_step_time": 6.763001203536987
    },
    {
      "epoch": 1.8910408020019532e-05,
      "model_forward_time": 1.7604191303253174,
      "step": 93829
    },
    {
      "epoch": 1.8910408020019532e-05,
      "step": 93829,
      "training_step_time": 6.742257118225098
    },
    {
      "epoch": 1.8911361694335936e-05,
      "grad_norm": 0.17423653602600098,
      "learning_rate": 1.0375186865718424e-07,
      "loss": 0.0531,
      "step": 93830
    },
    {
      "epoch": 1.8911361694335936e-05,
      "model_forward_time": 1.8296399116516113,
      "step": 93830
    },
    {
      "dynamics_loss_avg": 0.06288904240354895,
      "epoch": 1.8911361694335936e-05,
      "step": 93830
    },
    {
      "action_loss_avg": 0.00208451728685759,
      "epoch": 1.8911361694335936e-05,
      "step": 93830
    },
    {
      "epoch": 1.8911361694335936e-05,
      "step": 93830,
      "training_step_time": 7.783797264099121
    },
    {
      "epoch": 1.8912315368652343e-05,
      "model_forward_time": 2.8664357662200928,
      "step": 93831
    },
    {
      "epoch": 1.8912315368652343e-05,
      "step": 93831,
      "training_step_time": 8.682708740234375
    },
    {
      "epoch": 1.891326904296875e-05,
      "model_forward_time": 1.7598998546600342,
      "step": 93832
    },
    {
      "epoch": 1.891326904296875e-05,
      "step": 93832,
      "training_step_time": 8.21694040298462
    },
    {
      "epoch": 1.8914222717285155e-05,
      "model_forward_time": 1.7658674716949463,
      "step": 93833
    },
    {
      "epoch": 1.8914222717285155e-05,
      "step": 93833,
      "training_step_time": 7.614980936050415
    },
    {
      "epoch": 1.8915176391601562e-05,
      "model_forward_time": 1.7628779411315918,
      "step": 93834
    },
    {
      "epoch": 1.8915176391601562e-05,
      "step": 93834,
      "training_step_time": 7.628623962402344
    },
    {
      "epoch": 1.891613006591797e-05,
      "model_forward_time": 1.7583799362182617,
      "step": 93835
    },
    {
      "epoch": 1.891613006591797e-05,
      "step": 93835,
      "training_step_time": 7.671138763427734
    },
    {
      "epoch": 1.8917083740234374e-05,
      "model_forward_time": 1.7624561786651611,
      "step": 93836
    },
    {
      "epoch": 1.8917083740234374e-05,
      "step": 93836,
      "training_step_time": 6.840695142745972
    },
    {
      "epoch": 1.891803741455078e-05,
      "model_forward_time": 1.7779335975646973,
      "step": 93837
    },
    {
      "epoch": 1.891803741455078e-05,
      "step": 93837,
      "training_step_time": 6.83649468421936
    },
    {
      "epoch": 1.891899108886719e-05,
      "model_forward_time": 1.761162281036377,
      "step": 93838
    },
    {
      "epoch": 1.891899108886719e-05,
      "step": 93838,
      "training_step_time": 6.863518714904785
    },
    {
      "epoch": 1.8919944763183592e-05,
      "model_forward_time": 1.829101800918579,
      "step": 93839
    },
    {
      "epoch": 1.8919944763183592e-05,
      "step": 93839,
      "training_step_time": 7.7549965381622314
    },
    {
      "epoch": 1.89208984375e-05,
      "grad_norm": 0.08426415920257568,
      "learning_rate": 1.0341704788978113e-07,
      "loss": 0.0525,
      "step": 93840
    },
    {
      "epoch": 1.89208984375e-05,
      "model_forward_time": 2.8416059017181396,
      "step": 93840
    },
    {
      "dynamics_loss_avg": 0.04503132794052363,
      "epoch": 1.89208984375e-05,
      "step": 93840
    },
    {
      "action_loss_avg": 0.0012392839940730481,
      "epoch": 1.89208984375e-05,
      "step": 93840
    },
    {
      "epoch": 1.89208984375e-05,
      "step": 93840,
      "training_step_time": 8.294994831085205
    },
    {
      "epoch": 1.8921852111816407e-05,
      "model_forward_time": 1.7654590606689453,
      "step": 93841
    },
    {
      "epoch": 1.8921852111816407e-05,
      "step": 93841,
      "training_step_time": 7.81389307975769
    },
    {
      "epoch": 1.892280578613281e-05,
      "model_forward_time": 1.7658884525299072,
      "step": 93842
    },
    {
      "epoch": 1.892280578613281e-05,
      "step": 93842,
      "training_step_time": 7.674565315246582
    },
    {
      "epoch": 1.892375946044922e-05,
      "model_forward_time": 1.764190912246704,
      "step": 93843
    },
    {
      "epoch": 1.892375946044922e-05,
      "step": 93843,
      "training_step_time": 8.150755405426025
    },
    {
      "epoch": 1.8924713134765626e-05,
      "model_forward_time": 1.7692909240722656,
      "step": 93844
    },
    {
      "epoch": 1.8924713134765626e-05,
      "step": 93844,
      "training_step_time": 7.642789840698242
    },
    {
      "epoch": 1.892566680908203e-05,
      "model_forward_time": 1.7635607719421387,
      "step": 93845
    },
    {
      "epoch": 1.892566680908203e-05,
      "step": 93845,
      "training_step_time": 6.812697172164917
    },
    {
      "epoch": 1.8926620483398438e-05,
      "model_forward_time": 1.7840681076049805,
      "step": 93846
    },
    {
      "epoch": 1.8926620483398438e-05,
      "step": 93846,
      "training_step_time": 6.779224157333374
    },
    {
      "epoch": 1.8927574157714845e-05,
      "model_forward_time": 1.7288124561309814,
      "step": 93847
    },
    {
      "epoch": 1.8927574157714845e-05,
      "step": 93847,
      "training_step_time": 6.763883352279663
    },
    {
      "epoch": 1.892852783203125e-05,
      "model_forward_time": 1.7606496810913086,
      "step": 93848
    },
    {
      "epoch": 1.892852783203125e-05,
      "step": 93848,
      "training_step_time": 7.728341579437256
    },
    {
      "epoch": 1.8929481506347656e-05,
      "model_forward_time": 2.3742339611053467,
      "step": 93849
    },
    {
      "epoch": 1.8929481506347656e-05,
      "step": 93849,
      "training_step_time": 7.693967580795288
    },
    {
      "epoch": 1.8930435180664064e-05,
      "grad_norm": 0.13258035480976105,
      "learning_rate": 1.030827626053288e-07,
      "loss": 0.0531,
      "step": 93850
    },
    {
      "epoch": 1.8930435180664064e-05,
      "model_forward_time": 1.7460837364196777,
      "step": 93850
    },
    {
      "dynamics_loss_avg": 0.05568304359912872,
      "epoch": 1.8930435180664064e-05,
      "step": 93850
    },
    {
      "action_loss_avg": 0.002187074307585135,
      "epoch": 1.8930435180664064e-05,
      "step": 93850
    },
    {
      "epoch": 1.8930435180664064e-05,
      "step": 93850,
      "training_step_time": 9.432213544845581
    },
    {
      "epoch": 1.8931388854980468e-05,
      "model_forward_time": 1.7504239082336426,
      "step": 93851
    },
    {
      "epoch": 1.8931388854980468e-05,
      "step": 93851,
      "training_step_time": 8.348822832107544
    },
    {
      "epoch": 1.8932342529296875e-05,
      "model_forward_time": 1.7536253929138184,
      "step": 93852
    },
    {
      "epoch": 1.8932342529296875e-05,
      "step": 93852,
      "training_step_time": 7.425658702850342
    },
    {
      "epoch": 1.8933296203613283e-05,
      "model_forward_time": 1.7453527450561523,
      "step": 93853
    },
    {
      "epoch": 1.8933296203613283e-05,
      "step": 93853,
      "training_step_time": 7.81876015663147
    },
    {
      "epoch": 1.8934249877929687e-05,
      "model_forward_time": 1.7457516193389893,
      "step": 93854
    },
    {
      "epoch": 1.8934249877929687e-05,
      "step": 93854,
      "training_step_time": 6.7634007930755615
    },
    {
      "epoch": 1.8935203552246094e-05,
      "model_forward_time": 1.7476041316986084,
      "step": 93855
    },
    {
      "epoch": 1.8935203552246094e-05,
      "step": 93855,
      "training_step_time": 6.783070087432861
    },
    {
      "epoch": 1.89361572265625e-05,
      "model_forward_time": 1.753258466720581,
      "step": 93856
    },
    {
      "epoch": 1.89361572265625e-05,
      "step": 93856,
      "training_step_time": 6.7571306228637695
    },
    {
      "epoch": 1.8937110900878906e-05,
      "model_forward_time": 1.7538893222808838,
      "step": 93857
    },
    {
      "epoch": 1.8937110900878906e-05,
      "step": 93857,
      "training_step_time": 7.1981682777404785
    },
    {
      "epoch": 1.8938064575195313e-05,
      "model_forward_time": 2.4265730381011963,
      "step": 93858
    },
    {
      "epoch": 1.8938064575195313e-05,
      "step": 93858,
      "training_step_time": 7.669837713241577
    },
    {
      "epoch": 1.893901824951172e-05,
      "model_forward_time": 1.7540805339813232,
      "step": 93859
    },
    {
      "epoch": 1.893901824951172e-05,
      "step": 93859,
      "training_step_time": 9.633500576019287
    },
    {
      "epoch": 1.8939971923828124e-05,
      "grad_norm": 0.18668806552886963,
      "learning_rate": 1.0274901284038241e-07,
      "loss": 0.0521,
      "step": 93860
    },
    {
      "epoch": 1.8939971923828124e-05,
      "model_forward_time": 1.7590906620025635,
      "step": 93860
    },
    {
      "dynamics_loss_avg": 0.04090683255344629,
      "epoch": 1.8939971923828124e-05,
      "step": 93860
    },
    {
      "action_loss_avg": 0.0017686062259599567,
      "epoch": 1.8939971923828124e-05,
      "step": 93860
    },
    {
      "epoch": 1.8939971923828124e-05,
      "step": 93860,
      "training_step_time": 7.6451637744903564
    },
    {
      "epoch": 1.8940925598144532e-05,
      "model_forward_time": 1.7657124996185303,
      "step": 93861
    },
    {
      "epoch": 1.8940925598144532e-05,
      "step": 93861,
      "training_step_time": 7.5211875438690186
    },
    {
      "epoch": 1.8941879272460936e-05,
      "model_forward_time": 1.7347562313079834,
      "step": 93862
    },
    {
      "epoch": 1.8941879272460936e-05,
      "step": 93862,
      "training_step_time": 7.698408365249634
    },
    {
      "epoch": 1.8942832946777343e-05,
      "model_forward_time": 1.7527110576629639,
      "step": 93863
    },
    {
      "epoch": 1.8942832946777343e-05,
      "step": 93863,
      "training_step_time": 7.669565439224243
    },
    {
      "epoch": 1.894378662109375e-05,
      "model_forward_time": 1.7616164684295654,
      "step": 93864
    },
    {
      "epoch": 1.894378662109375e-05,
      "step": 93864,
      "training_step_time": 6.79439377784729
    },
    {
      "epoch": 1.8944740295410155e-05,
      "model_forward_time": 1.7544844150543213,
      "step": 93865
    },
    {
      "epoch": 1.8944740295410155e-05,
      "step": 93865,
      "training_step_time": 6.784227132797241
    },
    {
      "epoch": 1.8945693969726562e-05,
      "model_forward_time": 1.7473576068878174,
      "step": 93866
    },
    {
      "epoch": 1.8945693969726562e-05,
      "step": 93866,
      "training_step_time": 7.234029531478882
    },
    {
      "epoch": 1.894664764404297e-05,
      "model_forward_time": 1.8204710483551025,
      "step": 93867
    },
    {
      "epoch": 1.894664764404297e-05,
      "step": 93867,
      "training_step_time": 8.777526378631592
    },
    {
      "epoch": 1.8947601318359374e-05,
      "model_forward_time": 1.752270221710205,
      "step": 93868
    },
    {
      "epoch": 1.8947601318359374e-05,
      "step": 93868,
      "training_step_time": 9.642537117004395
    },
    {
      "epoch": 1.894855499267578e-05,
      "model_forward_time": 1.7583754062652588,
      "step": 93869
    },
    {
      "epoch": 1.894855499267578e-05,
      "step": 93869,
      "training_step_time": 7.752200603485107
    },
    {
      "epoch": 1.894950866699219e-05,
      "grad_norm": 0.08988570421934128,
      "learning_rate": 1.0241579863144114e-07,
      "loss": 0.0537,
      "step": 93870
    },
    {
      "epoch": 1.894950866699219e-05,
      "model_forward_time": 1.7444353103637695,
      "step": 93870
    },
    {
      "dynamics_loss_avg": 0.04924692520871758,
      "epoch": 1.894950866699219e-05,
      "step": 93870
    },
    {
      "action_loss_avg": 0.0013717743000597692,
      "epoch": 1.894950866699219e-05,
      "step": 93870
    },
    {
      "epoch": 1.894950866699219e-05,
      "step": 93870,
      "training_step_time": 8.003082752227783
    },
    {
      "epoch": 1.8950462341308592e-05,
      "model_forward_time": 1.7584397792816162,
      "step": 93871
    },
    {
      "epoch": 1.8950462341308592e-05,
      "step": 93871,
      "training_step_time": 7.305186748504639
    },
    {
      "epoch": 1.8951416015625e-05,
      "model_forward_time": 1.7618458271026611,
      "step": 93872
    },
    {
      "epoch": 1.8951416015625e-05,
      "step": 93872,
      "training_step_time": 7.54799222946167
    },
    {
      "epoch": 1.8952369689941407e-05,
      "model_forward_time": 1.7632110118865967,
      "step": 93873
    },
    {
      "epoch": 1.8952369689941407e-05,
      "step": 93873,
      "training_step_time": 6.640660524368286
    },
    {
      "epoch": 1.895332336425781e-05,
      "model_forward_time": 1.764374017715454,
      "step": 93874
    },
    {
      "epoch": 1.895332336425781e-05,
      "step": 93874,
      "training_step_time": 6.714075565338135
    },
    {
      "epoch": 1.895427703857422e-05,
      "model_forward_time": 1.7675080299377441,
      "step": 93875
    },
    {
      "epoch": 1.895427703857422e-05,
      "step": 93875,
      "training_step_time": 7.006587982177734
    },
    {
      "epoch": 1.8955230712890626e-05,
      "model_forward_time": 1.831498146057129,
      "step": 93876
    },
    {
      "epoch": 1.8955230712890626e-05,
      "step": 93876,
      "training_step_time": 7.145467519760132
    },
    {
      "epoch": 1.895618438720703e-05,
      "model_forward_time": 2.3997225761413574,
      "step": 93877
    },
    {
      "epoch": 1.895618438720703e-05,
      "step": 93877,
      "training_step_time": 8.653508424758911
    },
    {
      "epoch": 1.8957138061523438e-05,
      "model_forward_time": 1.3994128704071045,
      "step": 93878
    },
    {
      "epoch": 1.8957138061523438e-05,
      "step": 93878,
      "training_step_time": 10.474700927734375
    },
    {
      "epoch": 1.8958091735839845e-05,
      "model_forward_time": 1.7797811031341553,
      "step": 93879
    },
    {
      "epoch": 1.8958091735839845e-05,
      "step": 93879,
      "training_step_time": 7.555704355239868
    },
    {
      "epoch": 1.895904541015625e-05,
      "grad_norm": 0.13470900058746338,
      "learning_rate": 1.0208312001494414e-07,
      "loss": 0.0538,
      "step": 93880
    },
    {
      "epoch": 1.895904541015625e-05,
      "model_forward_time": 1.7564666271209717,
      "step": 93880
    },
    {
      "dynamics_loss_avg": 0.04729311317205429,
      "epoch": 1.895904541015625e-05,
      "step": 93880
    },
    {
      "action_loss_avg": 0.0015212707541650162,
      "epoch": 1.895904541015625e-05,
      "step": 93880
    },
    {
      "epoch": 1.895904541015625e-05,
      "step": 93880,
      "training_step_time": 7.366424083709717
    },
    {
      "epoch": 1.8959999084472656e-05,
      "model_forward_time": 1.756429672241211,
      "step": 93881
    },
    {
      "epoch": 1.8959999084472656e-05,
      "step": 93881,
      "training_step_time": 7.671926259994507
    },
    {
      "epoch": 1.8960952758789064e-05,
      "model_forward_time": 1.769939661026001,
      "step": 93882
    },
    {
      "epoch": 1.8960952758789064e-05,
      "step": 93882,
      "training_step_time": 6.72105598449707
    },
    {
      "epoch": 1.8961906433105468e-05,
      "model_forward_time": 1.7640736103057861,
      "step": 93883
    },
    {
      "epoch": 1.8961906433105468e-05,
      "step": 93883,
      "training_step_time": 6.736188173294067
    },
    {
      "epoch": 1.8962860107421875e-05,
      "model_forward_time": 1.7672841548919678,
      "step": 93884
    },
    {
      "epoch": 1.8962860107421875e-05,
      "step": 93884,
      "training_step_time": 6.713802099227905
    },
    {
      "epoch": 1.8963813781738283e-05,
      "model_forward_time": 1.7812786102294922,
      "step": 93885
    },
    {
      "epoch": 1.8963813781738283e-05,
      "step": 93885,
      "training_step_time": 7.287254571914673
    },
    {
      "epoch": 1.8964767456054687e-05,
      "model_forward_time": 2.388731002807617,
      "step": 93886
    },
    {
      "epoch": 1.8964767456054687e-05,
      "step": 93886,
      "training_step_time": 9.147377967834473
    },
    {
      "epoch": 1.8965721130371094e-05,
      "model_forward_time": 1.0785036087036133,
      "step": 93887
    },
    {
      "epoch": 1.8965721130371094e-05,
      "step": 93887,
      "training_step_time": 9.71065378189087
    },
    {
      "epoch": 1.89666748046875e-05,
      "model_forward_time": 1.3988451957702637,
      "step": 93888
    },
    {
      "epoch": 1.89666748046875e-05,
      "step": 93888,
      "training_step_time": 7.693106412887573
    },
    {
      "epoch": 1.8967628479003906e-05,
      "model_forward_time": 1.770341396331787,
      "step": 93889
    },
    {
      "epoch": 1.8967628479003906e-05,
      "step": 93889,
      "training_step_time": 7.466897010803223
    },
    {
      "epoch": 1.8968582153320313e-05,
      "grad_norm": 0.08333063125610352,
      "learning_rate": 1.0175097702727455e-07,
      "loss": 0.0531,
      "step": 93890
    },
    {
      "epoch": 1.8968582153320313e-05,
      "model_forward_time": 1.4083714485168457,
      "step": 93890
    },
    {
      "dynamics_loss_avg": 0.044784157630056144,
      "epoch": 1.8968582153320313e-05,
      "step": 93890
    },
    {
      "action_loss_avg": 0.001081376284128055,
      "epoch": 1.8968582153320313e-05,
      "step": 93890
    },
    {
      "epoch": 1.8968582153320313e-05,
      "step": 93890,
      "training_step_time": 7.920715093612671
    },
    {
      "epoch": 1.8969535827636717e-05,
      "model_forward_time": 1.7676639556884766,
      "step": 93891
    },
    {
      "epoch": 1.8969535827636717e-05,
      "step": 93891,
      "training_step_time": 7.70020866394043
    },
    {
      "epoch": 1.8970489501953124e-05,
      "model_forward_time": 1.7619833946228027,
      "step": 93892
    },
    {
      "epoch": 1.8970489501953124e-05,
      "step": 93892,
      "training_step_time": 6.7704527378082275
    },
    {
      "epoch": 1.8971443176269532e-05,
      "model_forward_time": 1.7622363567352295,
      "step": 93893
    },
    {
      "epoch": 1.8971443176269532e-05,
      "step": 93893,
      "training_step_time": 6.773469686508179
    },
    {
      "epoch": 1.8972396850585936e-05,
      "model_forward_time": 1.8284459114074707,
      "step": 93894
    },
    {
      "epoch": 1.8972396850585936e-05,
      "step": 93894,
      "training_step_time": 7.358814001083374
    },
    {
      "epoch": 1.8973350524902343e-05,
      "model_forward_time": 2.347309112548828,
      "step": 93895
    },
    {
      "epoch": 1.8973350524902343e-05,
      "step": 93895,
      "training_step_time": 8.575676202774048
    },
    {
      "epoch": 1.897430419921875e-05,
      "model_forward_time": 1.7545173168182373,
      "step": 93896
    },
    {
      "epoch": 1.897430419921875e-05,
      "step": 93896,
      "training_step_time": 9.63341236114502
    },
    {
      "epoch": 1.8975257873535155e-05,
      "model_forward_time": 1.7512354850769043,
      "step": 93897
    },
    {
      "epoch": 1.8975257873535155e-05,
      "step": 93897,
      "training_step_time": 8.463682174682617
    },
    {
      "epoch": 1.8976211547851562e-05,
      "model_forward_time": 1.7538232803344727,
      "step": 93898
    },
    {
      "epoch": 1.8976211547851562e-05,
      "step": 93898,
      "training_step_time": 7.367265701293945
    },
    {
      "epoch": 1.897716522216797e-05,
      "model_forward_time": 1.7636034488677979,
      "step": 93899
    },
    {
      "epoch": 1.897716522216797e-05,
      "step": 93899,
      "training_step_time": 7.2720417976379395
    },
    {
      "epoch": 1.8978118896484374e-05,
      "grad_norm": 0.10937527567148209,
      "learning_rate": 1.0141936970475275e-07,
      "loss": 0.0542,
      "step": 93900
    },
    {
      "epoch": 1.8978118896484374e-05,
      "model_forward_time": 1.7530837059020996,
      "step": 93900
    },
    {
      "dynamics_loss_avg": 0.05779868140816689,
      "epoch": 1.8978118896484374e-05,
      "step": 93900
    },
    {
      "action_loss_avg": 0.0014740629761945457,
      "epoch": 1.8978118896484374e-05,
      "step": 93900
    },
    {
      "epoch": 1.8978118896484374e-05,
      "step": 93900,
      "training_step_time": 7.722974061965942
    },
    {
      "epoch": 1.897907257080078e-05,
      "model_forward_time": 1.759580135345459,
      "step": 93901
    },
    {
      "epoch": 1.897907257080078e-05,
      "step": 93901,
      "training_step_time": 6.744428634643555
    },
    {
      "epoch": 1.898002624511719e-05,
      "model_forward_time": 1.7479169368743896,
      "step": 93902
    },
    {
      "epoch": 1.898002624511719e-05,
      "step": 93902,
      "training_step_time": 6.7889564037323
    },
    {
      "epoch": 1.8980979919433593e-05,
      "model_forward_time": 1.8085033893585205,
      "step": 93903
    },
    {
      "epoch": 1.8980979919433593e-05,
      "step": 93903,
      "training_step_time": 7.309536457061768
    },
    {
      "epoch": 1.898193359375e-05,
      "model_forward_time": 1.7471396923065186,
      "step": 93904
    },
    {
      "epoch": 1.898193359375e-05,
      "step": 93904,
      "training_step_time": 9.27808690071106
    },
    {
      "epoch": 1.8982887268066407e-05,
      "model_forward_time": 2.374664545059204,
      "step": 93905
    },
    {
      "epoch": 1.8982887268066407e-05,
      "step": 93905,
      "training_step_time": 9.675945043563843
    },
    {
      "epoch": 1.898384094238281e-05,
      "model_forward_time": 1.7604413032531738,
      "step": 93906
    },
    {
      "epoch": 1.898384094238281e-05,
      "step": 93906,
      "training_step_time": 8.805618047714233
    },
    {
      "epoch": 1.898479461669922e-05,
      "model_forward_time": 1.7564342021942139,
      "step": 93907
    },
    {
      "epoch": 1.898479461669922e-05,
      "step": 93907,
      "training_step_time": 7.6675450801849365
    },
    {
      "epoch": 1.8985748291015626e-05,
      "model_forward_time": 1.7451634407043457,
      "step": 93908
    },
    {
      "epoch": 1.8985748291015626e-05,
      "step": 93908,
      "training_step_time": 7.710065126419067
    },
    {
      "epoch": 1.898670196533203e-05,
      "model_forward_time": 1.7622411251068115,
      "step": 93909
    },
    {
      "epoch": 1.898670196533203e-05,
      "step": 93909,
      "training_step_time": 7.322774171829224
    },
    {
      "epoch": 1.8987655639648438e-05,
      "grad_norm": 0.0839003399014473,
      "learning_rate": 1.0108829808364418e-07,
      "loss": 0.0547,
      "step": 93910
    },
    {
      "epoch": 1.8987655639648438e-05,
      "model_forward_time": 1.7769391536712646,
      "step": 93910
    },
    {
      "dynamics_loss_avg": 0.04860619939863682,
      "epoch": 1.8987655639648438e-05,
      "step": 93910
    },
    {
      "action_loss_avg": 0.0014596477180020885,
      "epoch": 1.8987655639648438e-05,
      "step": 93910
    },
    {
      "epoch": 1.8987655639648438e-05,
      "step": 93910,
      "training_step_time": 7.6505889892578125
    },
    {
      "epoch": 1.8988609313964845e-05,
      "model_forward_time": 1.7342967987060547,
      "step": 93911
    },
    {
      "epoch": 1.8988609313964845e-05,
      "step": 93911,
      "training_step_time": 6.735921621322632
    },
    {
      "epoch": 1.898956298828125e-05,
      "model_forward_time": 1.8056228160858154,
      "step": 93912
    },
    {
      "epoch": 1.898956298828125e-05,
      "step": 93912,
      "training_step_time": 7.16947865486145
    },
    {
      "epoch": 1.8990516662597656e-05,
      "model_forward_time": 1.7399265766143799,
      "step": 93913
    },
    {
      "epoch": 1.8990516662597656e-05,
      "step": 93913,
      "training_step_time": 7.5607569217681885
    },
    {
      "epoch": 1.8991470336914064e-05,
      "model_forward_time": 2.3709144592285156,
      "step": 93914
    },
    {
      "epoch": 1.8991470336914064e-05,
      "step": 93914,
      "training_step_time": 7.623433589935303
    },
    {
      "epoch": 1.8992424011230468e-05,
      "model_forward_time": 1.7307324409484863,
      "step": 93915
    },
    {
      "epoch": 1.8992424011230468e-05,
      "step": 93915,
      "training_step_time": 8.558409452438354
    },
    {
      "epoch": 1.8993377685546875e-05,
      "model_forward_time": 1.7315185070037842,
      "step": 93916
    },
    {
      "epoch": 1.8993377685546875e-05,
      "step": 93916,
      "training_step_time": 7.495574951171875
    },
    {
      "epoch": 1.8994331359863283e-05,
      "model_forward_time": 1.7490248680114746,
      "step": 93917
    },
    {
      "epoch": 1.8994331359863283e-05,
      "step": 93917,
      "training_step_time": 7.2924463748931885
    },
    {
      "epoch": 1.8995285034179687e-05,
      "model_forward_time": 1.735889196395874,
      "step": 93918
    },
    {
      "epoch": 1.8995285034179687e-05,
      "step": 93918,
      "training_step_time": 7.694125652313232
    },
    {
      "epoch": 1.8996238708496094e-05,
      "model_forward_time": 1.7310447692871094,
      "step": 93919
    },
    {
      "epoch": 1.8996238708496094e-05,
      "step": 93919,
      "training_step_time": 7.565379619598389
    },
    {
      "epoch": 1.89971923828125e-05,
      "grad_norm": 0.1101059839129448,
      "learning_rate": 1.0075776220015376e-07,
      "loss": 0.0546,
      "step": 93920
    },
    {
      "epoch": 1.89971923828125e-05,
      "model_forward_time": 1.7326397895812988,
      "step": 93920
    },
    {
      "dynamics_loss_avg": 0.062425268441438676,
      "epoch": 1.89971923828125e-05,
      "step": 93920
    },
    {
      "action_loss_avg": 0.0009680689399829134,
      "epoch": 1.89971923828125e-05,
      "step": 93920
    },
    {
      "epoch": 1.89971923828125e-05,
      "step": 93920,
      "training_step_time": 6.695233106613159
    },
    {
      "epoch": 1.8998146057128906e-05,
      "model_forward_time": 1.4211082458496094,
      "step": 93921
    },
    {
      "epoch": 1.8998146057128906e-05,
      "step": 93921,
      "training_step_time": 7.283599853515625
    },
    {
      "epoch": 1.8999099731445313e-05,
      "model_forward_time": 1.744422197341919,
      "step": 93922
    },
    {
      "epoch": 1.8999099731445313e-05,
      "step": 93922,
      "training_step_time": 7.528175354003906
    },
    {
      "epoch": 1.9000053405761717e-05,
      "model_forward_time": 2.3916473388671875,
      "step": 93923
    },
    {
      "epoch": 1.9000053405761717e-05,
      "step": 93923,
      "training_step_time": 9.722408294677734
    },
    {
      "epoch": 1.9001007080078125e-05,
      "model_forward_time": 1.733417272567749,
      "step": 93924
    },
    {
      "epoch": 1.9001007080078125e-05,
      "step": 93924,
      "training_step_time": 7.636588096618652
    },
    {
      "epoch": 1.9001960754394532e-05,
      "model_forward_time": 1.7351691722869873,
      "step": 93925
    },
    {
      "epoch": 1.9001960754394532e-05,
      "step": 93925,
      "training_step_time": 7.679724454879761
    },
    {
      "epoch": 1.9002914428710936e-05,
      "model_forward_time": 1.731597661972046,
      "step": 93926
    },
    {
      "epoch": 1.9002914428710936e-05,
      "step": 93926,
      "training_step_time": 7.375985622406006
    },
    {
      "epoch": 1.9003868103027343e-05,
      "model_forward_time": 1.737241506576538,
      "step": 93927
    },
    {
      "epoch": 1.9003868103027343e-05,
      "step": 93927,
      "training_step_time": 7.31406044960022
    },
    {
      "epoch": 1.900482177734375e-05,
      "model_forward_time": 1.7339849472045898,
      "step": 93928
    },
    {
      "epoch": 1.900482177734375e-05,
      "step": 93928,
      "training_step_time": 7.735147953033447
    },
    {
      "epoch": 1.9005775451660155e-05,
      "model_forward_time": 1.7442693710327148,
      "step": 93929
    },
    {
      "epoch": 1.9005775451660155e-05,
      "step": 93929,
      "training_step_time": 6.832810878753662
    },
    {
      "epoch": 1.9006729125976562e-05,
      "grad_norm": 0.06892479956150055,
      "learning_rate": 1.0042776209042815e-07,
      "loss": 0.053,
      "step": 93930
    },
    {
      "epoch": 1.9006729125976562e-05,
      "model_forward_time": 1.8067450523376465,
      "step": 93930
    },
    {
      "dynamics_loss_avg": 0.05737422928214073,
      "epoch": 1.9006729125976562e-05,
      "step": 93930
    },
    {
      "action_loss_avg": 0.001694836578099057,
      "epoch": 1.9006729125976562e-05,
      "step": 93930
    },
    {
      "epoch": 1.9006729125976562e-05,
      "step": 93930,
      "training_step_time": 7.399489879608154
    },
    {
      "epoch": 1.900768280029297e-05,
      "model_forward_time": 1.7489805221557617,
      "step": 93931
    },
    {
      "epoch": 1.900768280029297e-05,
      "step": 93931,
      "training_step_time": 7.622882843017578
    },
    {
      "epoch": 1.9008636474609374e-05,
      "model_forward_time": 2.3799703121185303,
      "step": 93932
    },
    {
      "epoch": 1.9008636474609374e-05,
      "step": 93932,
      "training_step_time": 10.592715740203857
    },
    {
      "epoch": 1.900959014892578e-05,
      "model_forward_time": 1.7481286525726318,
      "step": 93933
    },
    {
      "epoch": 1.900959014892578e-05,
      "step": 93933,
      "training_step_time": 8.459931373596191
    },
    {
      "epoch": 1.901054382324219e-05,
      "model_forward_time": 1.050863265991211,
      "step": 93934
    },
    {
      "epoch": 1.901054382324219e-05,
      "step": 93934,
      "training_step_time": 7.789433240890503
    },
    {
      "epoch": 1.9011497497558593e-05,
      "model_forward_time": 1.7556798458099365,
      "step": 93935
    },
    {
      "epoch": 1.9011497497558593e-05,
      "step": 93935,
      "training_step_time": 7.461063623428345
    },
    {
      "epoch": 1.9012451171875e-05,
      "model_forward_time": 1.7454569339752197,
      "step": 93936
    },
    {
      "epoch": 1.9012451171875e-05,
      "step": 93936,
      "training_step_time": 7.296694993972778
    },
    {
      "epoch": 1.9013404846191407e-05,
      "model_forward_time": 1.7596368789672852,
      "step": 93937
    },
    {
      "epoch": 1.9013404846191407e-05,
      "step": 93937,
      "training_step_time": 7.716846227645874
    },
    {
      "epoch": 1.901435852050781e-05,
      "model_forward_time": 1.7644155025482178,
      "step": 93938
    },
    {
      "epoch": 1.901435852050781e-05,
      "step": 93938,
      "training_step_time": 6.729106903076172
    },
    {
      "epoch": 1.901531219482422e-05,
      "model_forward_time": 1.7924728393554688,
      "step": 93939
    },
    {
      "epoch": 1.901531219482422e-05,
      "step": 93939,
      "training_step_time": 7.172692060470581
    },
    {
      "epoch": 1.9016265869140626e-05,
      "grad_norm": 0.08447936177253723,
      "learning_rate": 1.0009829779055679e-07,
      "loss": 0.055,
      "step": 93940
    },
    {
      "epoch": 1.9016265869140626e-05,
      "model_forward_time": 1.7521679401397705,
      "step": 93940
    },
    {
      "dynamics_loss_avg": 0.058055310323834416,
      "epoch": 1.9016265869140626e-05,
      "step": 93940
    },
    {
      "action_loss_avg": 0.0010434254014398902,
      "epoch": 1.9016265869140626e-05,
      "step": 93940
    },
    {
      "epoch": 1.9016265869140626e-05,
      "step": 93940,
      "training_step_time": 7.6287007331848145
    },
    {
      "epoch": 1.901721954345703e-05,
      "model_forward_time": 1.736520528793335,
      "step": 93941
    },
    {
      "epoch": 1.901721954345703e-05,
      "step": 93941,
      "training_step_time": 7.6723856925964355
    },
    {
      "epoch": 1.9018173217773438e-05,
      "model_forward_time": 2.3714466094970703,
      "step": 93942
    },
    {
      "epoch": 1.9018173217773438e-05,
      "step": 93942,
      "training_step_time": 9.44959545135498
    },
    {
      "epoch": 1.9019126892089845e-05,
      "model_forward_time": 1.7362632751464844,
      "step": 93943
    },
    {
      "epoch": 1.9019126892089845e-05,
      "step": 93943,
      "training_step_time": 7.617023468017578
    },
    {
      "epoch": 1.902008056640625e-05,
      "model_forward_time": 1.7348299026489258,
      "step": 93944
    },
    {
      "epoch": 1.902008056640625e-05,
      "step": 93944,
      "training_step_time": 8.269498109817505
    },
    {
      "epoch": 1.9021034240722657e-05,
      "model_forward_time": 1.7392234802246094,
      "step": 93945
    },
    {
      "epoch": 1.9021034240722657e-05,
      "step": 93945,
      "training_step_time": 7.007601022720337
    },
    {
      "epoch": 1.9021987915039064e-05,
      "model_forward_time": 1.7509489059448242,
      "step": 93946
    },
    {
      "epoch": 1.9021987915039064e-05,
      "step": 93946,
      "training_step_time": 7.658613920211792
    },
    {
      "epoch": 1.9022941589355468e-05,
      "model_forward_time": 1.7470612525939941,
      "step": 93947
    },
    {
      "epoch": 1.9022941589355468e-05,
      "step": 93947,
      "training_step_time": 6.622463941574097
    },
    {
      "epoch": 1.9023895263671875e-05,
      "model_forward_time": 1.7544751167297363,
      "step": 93948
    },
    {
      "epoch": 1.9023895263671875e-05,
      "step": 93948,
      "training_step_time": 7.776712656021118
    },
    {
      "epoch": 1.9024848937988283e-05,
      "model_forward_time": 1.0566785335540771,
      "step": 93949
    },
    {
      "epoch": 1.9024848937988283e-05,
      "step": 93949,
      "training_step_time": 7.524991989135742
    },
    {
      "epoch": 1.9025802612304687e-05,
      "grad_norm": 0.08879806101322174,
      "learning_rate": 9.976936933656866e-08,
      "loss": 0.0547,
      "step": 93950
    },
    {
      "epoch": 1.9025802612304687e-05,
      "model_forward_time": 1.7333250045776367,
      "step": 93950
    },
    {
      "dynamics_loss_avg": 0.03018281850963831,
      "epoch": 1.9025802612304687e-05,
      "step": 93950
    },
    {
      "action_loss_avg": 0.0010416347198770382,
      "epoch": 1.9025802612304687e-05,
      "step": 93950
    },
    {
      "epoch": 1.9025802612304687e-05,
      "step": 93950,
      "training_step_time": 8.962539911270142
    },
    {
      "epoch": 1.9026756286621094e-05,
      "model_forward_time": 2.3505337238311768,
      "step": 93951
    },
    {
      "epoch": 1.9026756286621094e-05,
      "step": 93951,
      "training_step_time": 9.424541711807251
    },
    {
      "epoch": 1.90277099609375e-05,
      "model_forward_time": 1.7387182712554932,
      "step": 93952
    },
    {
      "epoch": 1.90277099609375e-05,
      "step": 93952,
      "training_step_time": 8.557851552963257
    },
    {
      "epoch": 1.9028663635253906e-05,
      "model_forward_time": 1.7386722564697266,
      "step": 93953
    },
    {
      "epoch": 1.9028663635253906e-05,
      "step": 93953,
      "training_step_time": 7.558178663253784
    },
    {
      "epoch": 1.9029617309570313e-05,
      "model_forward_time": 1.7329933643341064,
      "step": 93954
    },
    {
      "epoch": 1.9029617309570313e-05,
      "step": 93954,
      "training_step_time": 7.369658708572388
    },
    {
      "epoch": 1.9030570983886717e-05,
      "model_forward_time": 1.7388863563537598,
      "step": 93955
    },
    {
      "epoch": 1.9030570983886717e-05,
      "step": 93955,
      "training_step_time": 7.51751184463501
    },
    {
      "epoch": 1.9031524658203125e-05,
      "model_forward_time": 1.753617525100708,
      "step": 93956
    },
    {
      "epoch": 1.9031524658203125e-05,
      "step": 93956,
      "training_step_time": 7.608561754226685
    },
    {
      "epoch": 1.9032478332519532e-05,
      "model_forward_time": 1.7383441925048828,
      "step": 93957
    },
    {
      "epoch": 1.9032478332519532e-05,
      "step": 93957,
      "training_step_time": 6.754908323287964
    },
    {
      "epoch": 1.9033432006835936e-05,
      "model_forward_time": 1.811452865600586,
      "step": 93958
    },
    {
      "epoch": 1.9033432006835936e-05,
      "step": 93958,
      "training_step_time": 7.493247985839844
    },
    {
      "epoch": 1.9034385681152343e-05,
      "model_forward_time": 1.736344337463379,
      "step": 93959
    },
    {
      "epoch": 1.9034385681152343e-05,
      "step": 93959,
      "training_step_time": 9.420398712158203
    },
    {
      "epoch": 1.903533935546875e-05,
      "grad_norm": 0.09038536995649338,
      "learning_rate": 9.944097676443443e-08,
      "loss": 0.0551,
      "step": 93960
    },
    {
      "epoch": 1.903533935546875e-05,
      "model_forward_time": 1.7316522598266602,
      "step": 93960
    },
    {
      "dynamics_loss_avg": 0.05363739961758256,
      "epoch": 1.903533935546875e-05,
      "step": 93960
    },
    {
      "action_loss_avg": 0.001499968284042552,
      "epoch": 1.903533935546875e-05,
      "step": 93960
    },
    {
      "epoch": 1.903533935546875e-05,
      "step": 93960,
      "training_step_time": 11.393164873123169
    },
    {
      "epoch": 1.9036293029785155e-05,
      "model_forward_time": 2.3577756881713867,
      "step": 93961
    },
    {
      "epoch": 1.9036293029785155e-05,
      "step": 93961,
      "training_step_time": 8.749861717224121
    },
    {
      "epoch": 1.9037246704101562e-05,
      "model_forward_time": 1.7313823699951172,
      "step": 93962
    },
    {
      "epoch": 1.9037246704101562e-05,
      "step": 93962,
      "training_step_time": 7.605712175369263
    },
    {
      "epoch": 1.903820037841797e-05,
      "model_forward_time": 1.7424767017364502,
      "step": 93963
    },
    {
      "epoch": 1.903820037841797e-05,
      "step": 93963,
      "training_step_time": 7.3756959438323975
    },
    {
      "epoch": 1.9039154052734374e-05,
      "model_forward_time": 1.7351667881011963,
      "step": 93964
    },
    {
      "epoch": 1.9039154052734374e-05,
      "step": 93964,
      "training_step_time": 7.277068853378296
    },
    {
      "epoch": 1.904010772705078e-05,
      "model_forward_time": 1.7336502075195312,
      "step": 93965
    },
    {
      "epoch": 1.904010772705078e-05,
      "step": 93965,
      "training_step_time": 7.637195348739624
    },
    {
      "epoch": 1.904106140136719e-05,
      "model_forward_time": 1.7349963188171387,
      "step": 93966
    },
    {
      "epoch": 1.904106140136719e-05,
      "step": 93966,
      "training_step_time": 6.782781600952148
    },
    {
      "epoch": 1.9042015075683593e-05,
      "model_forward_time": 1.7956762313842773,
      "step": 93967
    },
    {
      "epoch": 1.9042015075683593e-05,
      "step": 93967,
      "training_step_time": 7.399848937988281
    },
    {
      "epoch": 1.904296875e-05,
      "model_forward_time": 1.7436671257019043,
      "step": 93968
    },
    {
      "epoch": 1.904296875e-05,
      "step": 93968,
      "training_step_time": 7.588012218475342
    },
    {
      "epoch": 1.9043922424316407e-05,
      "model_forward_time": 1.738476037979126,
      "step": 93969
    },
    {
      "epoch": 1.9043922424316407e-05,
      "step": 93969,
      "training_step_time": 10.70168423652649
    },
    {
      "epoch": 1.904487609863281e-05,
      "grad_norm": 0.07026326656341553,
      "learning_rate": 9.911312011006757e-08,
      "loss": 0.0556,
      "step": 93970
    },
    {
      "epoch": 1.904487609863281e-05,
      "model_forward_time": 2.36102294921875,
      "step": 93970
    },
    {
      "dynamics_loss_avg": 0.057284427992999556,
      "epoch": 1.904487609863281e-05,
      "step": 93970
    },
    {
      "action_loss_avg": 0.0023382041894365103,
      "epoch": 1.904487609863281e-05,
      "step": 93970
    },
    {
      "epoch": 1.904487609863281e-05,
      "step": 93970,
      "training_step_time": 7.678410530090332
    },
    {
      "epoch": 1.904582977294922e-05,
      "model_forward_time": 1.7316155433654785,
      "step": 93971
    },
    {
      "epoch": 1.904582977294922e-05,
      "step": 93971,
      "training_step_time": 8.429663896560669
    },
    {
      "epoch": 1.9046783447265626e-05,
      "model_forward_time": 1.7272629737854004,
      "step": 93972
    },
    {
      "epoch": 1.9046783447265626e-05,
      "step": 93972,
      "training_step_time": 7.439667224884033
    },
    {
      "epoch": 1.904773712158203e-05,
      "model_forward_time": 1.7445309162139893,
      "step": 93973
    },
    {
      "epoch": 1.904773712158203e-05,
      "step": 93973,
      "training_step_time": 7.307645320892334
    },
    {
      "epoch": 1.9048690795898438e-05,
      "model_forward_time": 1.7374298572540283,
      "step": 93974
    },
    {
      "epoch": 1.9048690795898438e-05,
      "step": 93974,
      "training_step_time": 7.599706649780273
    },
    {
      "epoch": 1.9049644470214845e-05,
      "model_forward_time": 1.7358605861663818,
      "step": 93975
    },
    {
      "epoch": 1.9049644470214845e-05,
      "step": 93975,
      "training_step_time": 6.695398569107056
    },
    {
      "epoch": 1.905059814453125e-05,
      "model_forward_time": 1.8031246662139893,
      "step": 93976
    },
    {
      "epoch": 1.905059814453125e-05,
      "step": 93976,
      "training_step_time": 8.083222389221191
    },
    {
      "epoch": 1.9051551818847657e-05,
      "model_forward_time": 1.729776382446289,
      "step": 93977
    },
    {
      "epoch": 1.9051551818847657e-05,
      "step": 93977,
      "training_step_time": 7.58178973197937
    },
    {
      "epoch": 1.9052505493164064e-05,
      "model_forward_time": 1.738715648651123,
      "step": 93978
    },
    {
      "epoch": 1.9052505493164064e-05,
      "step": 93978,
      "training_step_time": 8.600747346878052
    },
    {
      "epoch": 1.9053459167480468e-05,
      "model_forward_time": 1.978865623474121,
      "step": 93979
    },
    {
      "epoch": 1.9053459167480468e-05,
      "step": 93979,
      "training_step_time": 8.38328504562378
    },
    {
      "epoch": 1.9054412841796875e-05,
      "grad_norm": 0.12910661101341248,
      "learning_rate": 9.87857994093211e-08,
      "loss": 0.0552,
      "step": 93980
    },
    {
      "epoch": 1.9054412841796875e-05,
      "model_forward_time": 1.7357871532440186,
      "step": 93980
    },
    {
      "dynamics_loss_avg": 0.0461823146790266,
      "epoch": 1.9054412841796875e-05,
      "step": 93980
    },
    {
      "action_loss_avg": 0.0012066469818819315,
      "epoch": 1.9054412841796875e-05,
      "step": 93980
    },
    {
      "epoch": 1.9054412841796875e-05,
      "step": 93980,
      "training_step_time": 8.269622564315796
    },
    {
      "epoch": 1.9055366516113283e-05,
      "model_forward_time": 1.7313613891601562,
      "step": 93981
    },
    {
      "epoch": 1.9055366516113283e-05,
      "step": 93981,
      "training_step_time": 8.242521047592163
    },
    {
      "epoch": 1.9056320190429687e-05,
      "model_forward_time": 1.733210563659668,
      "step": 93982
    },
    {
      "epoch": 1.9056320190429687e-05,
      "step": 93982,
      "training_step_time": 7.201553583145142
    },
    {
      "epoch": 1.9057273864746094e-05,
      "model_forward_time": 1.7455456256866455,
      "step": 93983
    },
    {
      "epoch": 1.9057273864746094e-05,
      "step": 93983,
      "training_step_time": 7.5626380443573
    },
    {
      "epoch": 1.90582275390625e-05,
      "model_forward_time": 1.7367479801177979,
      "step": 93984
    },
    {
      "epoch": 1.90582275390625e-05,
      "step": 93984,
      "training_step_time": 6.774379253387451
    },
    {
      "epoch": 1.9059181213378906e-05,
      "model_forward_time": 1.7733526229858398,
      "step": 93985
    },
    {
      "epoch": 1.9059181213378906e-05,
      "step": 93985,
      "training_step_time": 7.1041412353515625
    },
    {
      "epoch": 1.9060134887695313e-05,
      "model_forward_time": 1.7285950183868408,
      "step": 93986
    },
    {
      "epoch": 1.9060134887695313e-05,
      "step": 93986,
      "training_step_time": 7.5966832637786865
    },
    {
      "epoch": 1.9061088562011717e-05,
      "model_forward_time": 1.7342016696929932,
      "step": 93987
    },
    {
      "epoch": 1.9061088562011717e-05,
      "step": 93987,
      "training_step_time": 8.683805704116821
    },
    {
      "epoch": 1.9062042236328125e-05,
      "model_forward_time": 2.3992435932159424,
      "step": 93988
    },
    {
      "epoch": 1.9062042236328125e-05,
      "step": 93988,
      "training_step_time": 10.368716478347778
    },
    {
      "epoch": 1.9062995910644532e-05,
      "model_forward_time": 1.7341196537017822,
      "step": 93989
    },
    {
      "epoch": 1.9062995910644532e-05,
      "step": 93989,
      "training_step_time": 9.779260158538818
    },
    {
      "epoch": 1.9063949584960936e-05,
      "grad_norm": 0.0779525488615036,
      "learning_rate": 9.84590146979908e-08,
      "loss": 0.0569,
      "step": 93990
    },
    {
      "epoch": 1.9063949584960936e-05,
      "model_forward_time": 1.7353579998016357,
      "step": 93990
    },
    {
      "dynamics_loss_avg": 0.04434881377965212,
      "epoch": 1.9063949584960936e-05,
      "step": 93990
    },
    {
      "action_loss_avg": 0.001876847846142482,
      "epoch": 1.9063949584960936e-05,
      "step": 93990
    },
    {
      "epoch": 1.9063949584960936e-05,
      "step": 93990,
      "training_step_time": 7.519381523132324
    },
    {
      "epoch": 1.9064903259277343e-05,
      "model_forward_time": 1.758223056793213,
      "step": 93991
    },
    {
      "epoch": 1.9064903259277343e-05,
      "step": 93991,
      "training_step_time": 7.330257892608643
    },
    {
      "epoch": 1.906585693359375e-05,
      "model_forward_time": 1.7415597438812256,
      "step": 93992
    },
    {
      "epoch": 1.906585693359375e-05,
      "step": 93992,
      "training_step_time": 7.754345417022705
    },
    {
      "epoch": 1.9066810607910155e-05,
      "model_forward_time": 1.7459759712219238,
      "step": 93993
    },
    {
      "epoch": 1.9066810607910155e-05,
      "step": 93993,
      "training_step_time": 7.5680928230285645
    },
    {
      "epoch": 1.9067764282226562e-05,
      "model_forward_time": 1.7407350540161133,
      "step": 93994
    },
    {
      "epoch": 1.9067764282226562e-05,
      "step": 93994,
      "training_step_time": 6.8103346824646
    },
    {
      "epoch": 1.906871795654297e-05,
      "model_forward_time": 1.7915542125701904,
      "step": 93995
    },
    {
      "epoch": 1.906871795654297e-05,
      "step": 93995,
      "training_step_time": 7.520200252532959
    },
    {
      "epoch": 1.9069671630859374e-05,
      "model_forward_time": 1.7404921054840088,
      "step": 93996
    },
    {
      "epoch": 1.9069671630859374e-05,
      "step": 93996,
      "training_step_time": 8.108558416366577
    },
    {
      "epoch": 1.907062530517578e-05,
      "model_forward_time": 2.335848093032837,
      "step": 93997
    },
    {
      "epoch": 1.907062530517578e-05,
      "step": 93997,
      "training_step_time": 8.962516784667969
    },
    {
      "epoch": 1.907157897949219e-05,
      "model_forward_time": 1.7631826400756836,
      "step": 93998
    },
    {
      "epoch": 1.907157897949219e-05,
      "step": 93998,
      "training_step_time": 9.978979110717773
    },
    {
      "epoch": 1.9072532653808593e-05,
      "model_forward_time": 1.7357714176177979,
      "step": 93999
    },
    {
      "epoch": 1.9072532653808593e-05,
      "step": 93999,
      "training_step_time": 7.786613941192627
    },
    {
      "epoch": 1.9073486328125e-05,
      "grad_norm": 0.10362374037504196,
      "learning_rate": 9.813276601181366e-08,
      "loss": 0.0566,
      "step": 94000
    },
    {
      "epoch": 1.9073486328125e-05,
      "model_forward_time": 1.8250861167907715,
      "step": 94000
    },
    {
      "dynamics_loss_avg": 0.046207332797348497,
      "epoch": 1.9073486328125e-05,
      "step": 94000
    },
    {
      "action_loss_avg": 0.0012859846785431728,
      "epoch": 1.9073486328125e-05,
      "step": 94000
    },
    {
      "epoch": 1.9073486328125e-05,
      "step": 94000,
      "training_step_time": 7.4160895347595215
    },
    {
      "epoch": 1.9074440002441407e-05,
      "model_forward_time": 1.7503430843353271,
      "step": 94001
    },
    {
      "epoch": 1.9074440002441407e-05,
      "step": 94001,
      "training_step_time": 7.977609634399414
    },
    {
      "epoch": 1.907539367675781e-05,
      "model_forward_time": 1.7529971599578857,
      "step": 94002
    },
    {
      "epoch": 1.907539367675781e-05,
      "step": 94002,
      "training_step_time": 7.697431802749634
    },
    {
      "epoch": 1.907634735107422e-05,
      "model_forward_time": 1.7321507930755615,
      "step": 94003
    },
    {
      "epoch": 1.907634735107422e-05,
      "step": 94003,
      "training_step_time": 7.871314764022827
    },
    {
      "epoch": 1.9077301025390626e-05,
      "model_forward_time": 1.7428834438323975,
      "step": 94004
    },
    {
      "epoch": 1.9077301025390626e-05,
      "step": 94004,
      "training_step_time": 8.496348142623901
    },
    {
      "epoch": 1.907825469970703e-05,
      "model_forward_time": 1.7403473854064941,
      "step": 94005
    },
    {
      "epoch": 1.907825469970703e-05,
      "step": 94005,
      "training_step_time": 8.805095195770264
    },
    {
      "epoch": 1.9079208374023438e-05,
      "model_forward_time": 3.001265287399292,
      "step": 94006
    },
    {
      "epoch": 1.9079208374023438e-05,
      "step": 94006,
      "training_step_time": 11.26335334777832
    },
    {
      "epoch": 1.9080162048339845e-05,
      "model_forward_time": 1.7378549575805664,
      "step": 94007
    },
    {
      "epoch": 1.9080162048339845e-05,
      "step": 94007,
      "training_step_time": 11.235625743865967
    },
    {
      "epoch": 1.908111572265625e-05,
      "model_forward_time": 1.7432663440704346,
      "step": 94008
    },
    {
      "epoch": 1.908111572265625e-05,
      "step": 94008,
      "training_step_time": 8.53184723854065
    },
    {
      "epoch": 1.9082069396972657e-05,
      "model_forward_time": 1.7626032829284668,
      "step": 94009
    },
    {
      "epoch": 1.9082069396972657e-05,
      "step": 94009,
      "training_step_time": 7.588392019271851
    },
    {
      "epoch": 1.9083023071289064e-05,
      "grad_norm": 0.1048763170838356,
      "learning_rate": 9.780705338646723e-08,
      "loss": 0.057,
      "step": 94010
    },
    {
      "epoch": 1.9083023071289064e-05,
      "model_forward_time": 1.7659506797790527,
      "step": 94010
    },
    {
      "dynamics_loss_avg": 0.05321250874549151,
      "epoch": 1.9083023071289064e-05,
      "step": 94010
    },
    {
      "action_loss_avg": 0.002721934375585988,
      "epoch": 1.9083023071289064e-05,
      "step": 94010
    },
    {
      "epoch": 1.9083023071289064e-05,
      "step": 94010,
      "training_step_time": 7.330338954925537
    },
    {
      "epoch": 1.9083976745605468e-05,
      "model_forward_time": 1.7479922771453857,
      "step": 94011
    },
    {
      "epoch": 1.9083976745605468e-05,
      "step": 94011,
      "training_step_time": 7.777949333190918
    },
    {
      "epoch": 1.9084930419921875e-05,
      "model_forward_time": 1.757228136062622,
      "step": 94012
    },
    {
      "epoch": 1.9084930419921875e-05,
      "step": 94012,
      "training_step_time": 7.7065510749816895
    },
    {
      "epoch": 1.9085884094238283e-05,
      "model_forward_time": 1.3997917175292969,
      "step": 94013
    },
    {
      "epoch": 1.9085884094238283e-05,
      "step": 94013,
      "training_step_time": 7.096712112426758
    },
    {
      "epoch": 1.9086837768554687e-05,
      "model_forward_time": 1.7510299682617188,
      "step": 94014
    },
    {
      "epoch": 1.9086837768554687e-05,
      "step": 94014,
      "training_step_time": 7.568965911865234
    },
    {
      "epoch": 1.9087791442871094e-05,
      "model_forward_time": 2.3611128330230713,
      "step": 94015
    },
    {
      "epoch": 1.9087791442871094e-05,
      "step": 94015,
      "training_step_time": 10.288262128829956
    },
    {
      "epoch": 1.90887451171875e-05,
      "model_forward_time": 1.748431921005249,
      "step": 94016
    },
    {
      "epoch": 1.90887451171875e-05,
      "step": 94016,
      "training_step_time": 8.813932418823242
    },
    {
      "epoch": 1.9089698791503906e-05,
      "model_forward_time": 1.7537918090820312,
      "step": 94017
    },
    {
      "epoch": 1.9089698791503906e-05,
      "step": 94017,
      "training_step_time": 7.7993245124816895
    },
    {
      "epoch": 1.9090652465820313e-05,
      "model_forward_time": 1.753692865371704,
      "step": 94018
    },
    {
      "epoch": 1.9090652465820313e-05,
      "step": 94018,
      "training_step_time": 7.6089465618133545
    },
    {
      "epoch": 1.9091606140136717e-05,
      "model_forward_time": 1.7497727870941162,
      "step": 94019
    },
    {
      "epoch": 1.9091606140136717e-05,
      "step": 94019,
      "training_step_time": 7.281900405883789
    },
    {
      "epoch": 1.9092559814453125e-05,
      "grad_norm": 0.15525799989700317,
      "learning_rate": 9.748187685757082e-08,
      "loss": 0.0569,
      "step": 94020
    },
    {
      "epoch": 1.9092559814453125e-05,
      "model_forward_time": 1.772650957107544,
      "step": 94020
    },
    {
      "dynamics_loss_avg": 0.0563597247004509,
      "epoch": 1.9092559814453125e-05,
      "step": 94020
    },
    {
      "action_loss_avg": 0.0017954625975107775,
      "epoch": 1.9092559814453125e-05,
      "step": 94020
    },
    {
      "epoch": 1.9092559814453125e-05,
      "step": 94020,
      "training_step_time": 7.643061876296997
    },
    {
      "epoch": 1.9093513488769532e-05,
      "model_forward_time": 1.7451701164245605,
      "step": 94021
    },
    {
      "epoch": 1.9093513488769532e-05,
      "step": 94021,
      "training_step_time": 6.793942928314209
    },
    {
      "epoch": 1.9094467163085936e-05,
      "model_forward_time": 1.8178462982177734,
      "step": 94022
    },
    {
      "epoch": 1.9094467163085936e-05,
      "step": 94022,
      "training_step_time": 7.152493000030518
    },
    {
      "epoch": 1.9095420837402343e-05,
      "model_forward_time": 1.7551929950714111,
      "step": 94023
    },
    {
      "epoch": 1.9095420837402343e-05,
      "step": 94023,
      "training_step_time": 7.5063982009887695
    },
    {
      "epoch": 1.909637451171875e-05,
      "model_forward_time": 1.7522227764129639,
      "step": 94024
    },
    {
      "epoch": 1.909637451171875e-05,
      "step": 94024,
      "training_step_time": 8.495839834213257
    },
    {
      "epoch": 1.9097328186035155e-05,
      "model_forward_time": 2.386957883834839,
      "step": 94025
    },
    {
      "epoch": 1.9097328186035155e-05,
      "step": 94025,
      "training_step_time": 9.652044773101807
    },
    {
      "epoch": 1.9098281860351562e-05,
      "model_forward_time": 1.7412395477294922,
      "step": 94026
    },
    {
      "epoch": 1.9098281860351562e-05,
      "step": 94026,
      "training_step_time": 7.773438215255737
    },
    {
      "epoch": 1.909923553466797e-05,
      "model_forward_time": 1.7434210777282715,
      "step": 94027
    },
    {
      "epoch": 1.909923553466797e-05,
      "step": 94027,
      "training_step_time": 8.288198947906494
    },
    {
      "epoch": 1.9100189208984374e-05,
      "model_forward_time": 1.8438715934753418,
      "step": 94028
    },
    {
      "epoch": 1.9100189208984374e-05,
      "step": 94028,
      "training_step_time": 7.164585113525391
    },
    {
      "epoch": 1.910114288330078e-05,
      "model_forward_time": 1.7535958290100098,
      "step": 94029
    },
    {
      "epoch": 1.910114288330078e-05,
      "step": 94029,
      "training_step_time": 7.553012847900391
    },
    {
      "epoch": 1.910209655761719e-05,
      "grad_norm": 0.10156624764204025,
      "learning_rate": 9.715723646068543e-08,
      "loss": 0.0595,
      "step": 94030
    },
    {
      "epoch": 1.910209655761719e-05,
      "model_forward_time": 1.73701810836792,
      "step": 94030
    },
    {
      "dynamics_loss_avg": 0.048936234414577486,
      "epoch": 1.910209655761719e-05,
      "step": 94030
    },
    {
      "action_loss_avg": 0.0011060286691645161,
      "epoch": 1.910209655761719e-05,
      "step": 94030
    },
    {
      "epoch": 1.910209655761719e-05,
      "step": 94030,
      "training_step_time": 6.753010988235474
    },
    {
      "epoch": 1.9103050231933593e-05,
      "model_forward_time": 1.8135371208190918,
      "step": 94031
    },
    {
      "epoch": 1.9103050231933593e-05,
      "step": 94031,
      "training_step_time": 7.09944748878479
    },
    {
      "epoch": 1.910400390625e-05,
      "model_forward_time": 1.0372066497802734,
      "step": 94032
    },
    {
      "epoch": 1.910400390625e-05,
      "step": 94032,
      "training_step_time": 7.512868642807007
    },
    {
      "epoch": 1.9104957580566407e-05,
      "model_forward_time": 1.7514281272888184,
      "step": 94033
    },
    {
      "epoch": 1.9104957580566407e-05,
      "step": 94033,
      "training_step_time": 8.517553329467773
    },
    {
      "epoch": 1.910591125488281e-05,
      "model_forward_time": 2.4185588359832764,
      "step": 94034
    },
    {
      "epoch": 1.910591125488281e-05,
      "step": 94034,
      "training_step_time": 9.405547142028809
    },
    {
      "epoch": 1.910686492919922e-05,
      "model_forward_time": 1.7445039749145508,
      "step": 94035
    },
    {
      "epoch": 1.910686492919922e-05,
      "step": 94035,
      "training_step_time": 8.479867935180664
    },
    {
      "epoch": 1.9107818603515626e-05,
      "model_forward_time": 1.7449760437011719,
      "step": 94036
    },
    {
      "epoch": 1.9107818603515626e-05,
      "step": 94036,
      "training_step_time": 8.410776853561401
    },
    {
      "epoch": 1.910877227783203e-05,
      "model_forward_time": 1.7448315620422363,
      "step": 94037
    },
    {
      "epoch": 1.910877227783203e-05,
      "step": 94037,
      "training_step_time": 7.182362794876099
    },
    {
      "epoch": 1.9109725952148438e-05,
      "model_forward_time": 1.7513749599456787,
      "step": 94038
    },
    {
      "epoch": 1.9109725952148438e-05,
      "step": 94038,
      "training_step_time": 7.671857595443726
    },
    {
      "epoch": 1.9110679626464845e-05,
      "model_forward_time": 1.3906097412109375,
      "step": 94039
    },
    {
      "epoch": 1.9110679626464845e-05,
      "step": 94039,
      "training_step_time": 6.761188983917236
    },
    {
      "epoch": 1.911163330078125e-05,
      "grad_norm": 0.08875241130590439,
      "learning_rate": 9.68331322313143e-08,
      "loss": 0.0582,
      "step": 94040
    },
    {
      "epoch": 1.911163330078125e-05,
      "model_forward_time": 1.7282860279083252,
      "step": 94040
    },
    {
      "dynamics_loss_avg": 0.05524270050227642,
      "epoch": 1.911163330078125e-05,
      "step": 94040
    },
    {
      "action_loss_avg": 0.0010246248100884258,
      "epoch": 1.911163330078125e-05,
      "step": 94040
    },
    {
      "epoch": 1.911163330078125e-05,
      "step": 94040,
      "training_step_time": 7.768190145492554
    },
    {
      "epoch": 1.9112586975097657e-05,
      "model_forward_time": 1.7344744205474854,
      "step": 94041
    },
    {
      "epoch": 1.9112586975097657e-05,
      "step": 94041,
      "training_step_time": 8.287557125091553
    },
    {
      "epoch": 1.9113540649414064e-05,
      "model_forward_time": 1.7467327117919922,
      "step": 94042
    },
    {
      "epoch": 1.9113540649414064e-05,
      "step": 94042,
      "training_step_time": 7.6868085861206055
    },
    {
      "epoch": 1.9114494323730468e-05,
      "model_forward_time": 2.334310531616211,
      "step": 94043
    },
    {
      "epoch": 1.9114494323730468e-05,
      "step": 94043,
      "training_step_time": 10.00960922241211
    },
    {
      "epoch": 1.9115447998046875e-05,
      "model_forward_time": 1.7419095039367676,
      "step": 94044
    },
    {
      "epoch": 1.9115447998046875e-05,
      "step": 94044,
      "training_step_time": 8.800415515899658
    },
    {
      "epoch": 1.9116401672363283e-05,
      "model_forward_time": 1.7388195991516113,
      "step": 94045
    },
    {
      "epoch": 1.9116401672363283e-05,
      "step": 94045,
      "training_step_time": 8.416990756988525
    },
    {
      "epoch": 1.9117355346679687e-05,
      "model_forward_time": 1.7354927062988281,
      "step": 94046
    },
    {
      "epoch": 1.9117355346679687e-05,
      "step": 94046,
      "training_step_time": 7.563850402832031
    },
    {
      "epoch": 1.9118309020996094e-05,
      "model_forward_time": 1.7413756847381592,
      "step": 94047
    },
    {
      "epoch": 1.9118309020996094e-05,
      "step": 94047,
      "training_step_time": 7.238632917404175
    },
    {
      "epoch": 1.91192626953125e-05,
      "model_forward_time": 1.7577316761016846,
      "step": 94048
    },
    {
      "epoch": 1.91192626953125e-05,
      "step": 94048,
      "training_step_time": 7.648222208023071
    },
    {
      "epoch": 1.9120216369628906e-05,
      "model_forward_time": 1.7335071563720703,
      "step": 94049
    },
    {
      "epoch": 1.9120216369628906e-05,
      "step": 94049,
      "training_step_time": 6.693672180175781
    },
    {
      "epoch": 1.9121170043945313e-05,
      "grad_norm": 0.08788805454969406,
      "learning_rate": 9.65095642048991e-08,
      "loss": 0.0578,
      "step": 94050
    },
    {
      "epoch": 1.9121170043945313e-05,
      "model_forward_time": 1.7981204986572266,
      "step": 94050
    },
    {
      "dynamics_loss_avg": 0.061169274151325226,
      "epoch": 1.9121170043945313e-05,
      "step": 94050
    },
    {
      "action_loss_avg": 0.001514693268109113,
      "epoch": 1.9121170043945313e-05,
      "step": 94050
    },
    {
      "epoch": 1.9121170043945313e-05,
      "step": 94050,
      "training_step_time": 7.314299821853638
    },
    {
      "epoch": 1.9122123718261717e-05,
      "model_forward_time": 1.7451062202453613,
      "step": 94051
    },
    {
      "epoch": 1.9122123718261717e-05,
      "step": 94051,
      "training_step_time": 7.67138671875
    },
    {
      "epoch": 1.9123077392578125e-05,
      "model_forward_time": 1.7403864860534668,
      "step": 94052
    },
    {
      "epoch": 1.9123077392578125e-05,
      "step": 94052,
      "training_step_time": 8.584645986557007
    },
    {
      "epoch": 1.9124031066894532e-05,
      "model_forward_time": 2.35998272895813,
      "step": 94053
    },
    {
      "epoch": 1.9124031066894532e-05,
      "step": 94053,
      "training_step_time": 8.407617807388306
    },
    {
      "epoch": 1.9124984741210936e-05,
      "model_forward_time": 1.7434022426605225,
      "step": 94054
    },
    {
      "epoch": 1.9124984741210936e-05,
      "step": 94054,
      "training_step_time": 7.695819616317749
    },
    {
      "epoch": 1.9125938415527343e-05,
      "model_forward_time": 1.7754831314086914,
      "step": 94055
    },
    {
      "epoch": 1.9125938415527343e-05,
      "step": 94055,
      "training_step_time": 7.582569360733032
    },
    {
      "epoch": 1.912689208984375e-05,
      "model_forward_time": 1.7360124588012695,
      "step": 94056
    },
    {
      "epoch": 1.912689208984375e-05,
      "step": 94056,
      "training_step_time": 7.276762247085571
    },
    {
      "epoch": 1.9127845764160155e-05,
      "model_forward_time": 1.7359862327575684,
      "step": 94057
    },
    {
      "epoch": 1.9127845764160155e-05,
      "step": 94057,
      "training_step_time": 7.6768622398376465
    },
    {
      "epoch": 1.9128799438476562e-05,
      "model_forward_time": 1.744283676147461,
      "step": 94058
    },
    {
      "epoch": 1.9128799438476562e-05,
      "step": 94058,
      "training_step_time": 6.67270827293396
    },
    {
      "epoch": 1.912975311279297e-05,
      "model_forward_time": 1.8055658340454102,
      "step": 94059
    },
    {
      "epoch": 1.912975311279297e-05,
      "step": 94059,
      "training_step_time": 7.849159002304077
    },
    {
      "epoch": 1.9130706787109374e-05,
      "grad_norm": 0.07318124175071716,
      "learning_rate": 9.618653241682597e-08,
      "loss": 0.0573,
      "step": 94060
    },
    {
      "epoch": 1.9130706787109374e-05,
      "model_forward_time": 1.7398390769958496,
      "step": 94060
    },
    {
      "dynamics_loss_avg": 0.046579635515809056,
      "epoch": 1.9130706787109374e-05,
      "step": 94060
    },
    {
      "action_loss_avg": 0.0017875300356536173,
      "epoch": 1.9130706787109374e-05,
      "step": 94060
    },
    {
      "epoch": 1.9130706787109374e-05,
      "step": 94060,
      "training_step_time": 7.6066248416900635
    },
    {
      "epoch": 1.913166046142578e-05,
      "model_forward_time": 1.7354521751403809,
      "step": 94061
    },
    {
      "epoch": 1.913166046142578e-05,
      "step": 94061,
      "training_step_time": 10.564665794372559
    },
    {
      "epoch": 1.913261413574219e-05,
      "model_forward_time": 2.3877203464508057,
      "step": 94062
    },
    {
      "epoch": 1.913261413574219e-05,
      "step": 94062,
      "training_step_time": 8.36824107170105
    },
    {
      "epoch": 1.9133567810058593e-05,
      "model_forward_time": 1.7442259788513184,
      "step": 94063
    },
    {
      "epoch": 1.9133567810058593e-05,
      "step": 94063,
      "training_step_time": 9.196661233901978
    },
    {
      "epoch": 1.9134521484375e-05,
      "model_forward_time": 1.7447714805603027,
      "step": 94064
    },
    {
      "epoch": 1.9134521484375e-05,
      "step": 94064,
      "training_step_time": 7.616417646408081
    },
    {
      "epoch": 1.9135475158691407e-05,
      "model_forward_time": 1.7414524555206299,
      "step": 94065
    },
    {
      "epoch": 1.9135475158691407e-05,
      "step": 94065,
      "training_step_time": 7.237229824066162
    },
    {
      "epoch": 1.913642883300781e-05,
      "model_forward_time": 1.744217872619629,
      "step": 94066
    },
    {
      "epoch": 1.913642883300781e-05,
      "step": 94066,
      "training_step_time": 7.6105687618255615
    },
    {
      "epoch": 1.913738250732422e-05,
      "model_forward_time": 1.745980978012085,
      "step": 94067
    },
    {
      "epoch": 1.913738250732422e-05,
      "step": 94067,
      "training_step_time": 7.581208229064941
    },
    {
      "epoch": 1.9138336181640626e-05,
      "model_forward_time": 1.804145097732544,
      "step": 94068
    },
    {
      "epoch": 1.9138336181640626e-05,
      "step": 94068,
      "training_step_time": 7.084479331970215
    },
    {
      "epoch": 1.913928985595703e-05,
      "model_forward_time": 1.7459523677825928,
      "step": 94069
    },
    {
      "epoch": 1.913928985595703e-05,
      "step": 94069,
      "training_step_time": 8.500706672668457
    },
    {
      "epoch": 1.9140243530273438e-05,
      "grad_norm": 0.08678288757801056,
      "learning_rate": 9.586403690242053e-08,
      "loss": 0.0571,
      "step": 94070
    },
    {
      "epoch": 1.9140243530273438e-05,
      "model_forward_time": 1.7434988021850586,
      "step": 94070
    },
    {
      "dynamics_loss_avg": 0.05135749354958534,
      "epoch": 1.9140243530273438e-05,
      "step": 94070
    },
    {
      "action_loss_avg": 0.0011328286491334438,
      "epoch": 1.9140243530273438e-05,
      "step": 94070
    },
    {
      "epoch": 1.9140243530273438e-05,
      "step": 94070,
      "training_step_time": 9.142009019851685
    },
    {
      "epoch": 1.9141197204589845e-05,
      "model_forward_time": 2.3734190464019775,
      "step": 94071
    },
    {
      "epoch": 1.9141197204589845e-05,
      "step": 94071,
      "training_step_time": 9.716708421707153
    },
    {
      "epoch": 1.914215087890625e-05,
      "model_forward_time": 1.7528114318847656,
      "step": 94072
    },
    {
      "epoch": 1.914215087890625e-05,
      "step": 94072,
      "training_step_time": 9.201965093612671
    },
    {
      "epoch": 1.9143104553222657e-05,
      "model_forward_time": 1.7468218803405762,
      "step": 94073
    },
    {
      "epoch": 1.9143104553222657e-05,
      "step": 94073,
      "training_step_time": 7.545050621032715
    },
    {
      "epoch": 1.9144058227539064e-05,
      "model_forward_time": 1.7423462867736816,
      "step": 94074
    },
    {
      "epoch": 1.9144058227539064e-05,
      "step": 94074,
      "training_step_time": 7.1832475662231445
    },
    {
      "epoch": 1.9145011901855468e-05,
      "model_forward_time": 1.758716344833374,
      "step": 94075
    },
    {
      "epoch": 1.9145011901855468e-05,
      "step": 94075,
      "training_step_time": 7.241564512252808
    },
    {
      "epoch": 1.9145965576171875e-05,
      "model_forward_time": 1.7630672454833984,
      "step": 94076
    },
    {
      "epoch": 1.9145965576171875e-05,
      "step": 94076,
      "training_step_time": 7.561458587646484
    },
    {
      "epoch": 1.9146919250488283e-05,
      "model_forward_time": 1.7908928394317627,
      "step": 94077
    },
    {
      "epoch": 1.9146919250488283e-05,
      "step": 94077,
      "training_step_time": 7.069864988327026
    },
    {
      "epoch": 1.9147872924804687e-05,
      "model_forward_time": 1.7550969123840332,
      "step": 94078
    },
    {
      "epoch": 1.9147872924804687e-05,
      "step": 94078,
      "training_step_time": 8.709136009216309
    },
    {
      "epoch": 1.9148826599121094e-05,
      "model_forward_time": 1.76066255569458,
      "step": 94079
    },
    {
      "epoch": 1.9148826599121094e-05,
      "step": 94079,
      "training_step_time": 8.617697477340698
    },
    {
      "epoch": 1.91497802734375e-05,
      "grad_norm": 0.07326013594865799,
      "learning_rate": 9.554207769695123e-08,
      "loss": 0.0579,
      "step": 94080
    },
    {
      "epoch": 1.91497802734375e-05,
      "model_forward_time": 2.4223194122314453,
      "step": 94080
    },
    {
      "dynamics_loss_avg": 0.04369599362835288,
      "epoch": 1.91497802734375e-05,
      "step": 94080
    },
    {
      "action_loss_avg": 0.0009429309880943038,
      "epoch": 1.91497802734375e-05,
      "step": 94080
    },
    {
      "epoch": 1.91497802734375e-05,
      "step": 94080,
      "training_step_time": 9.928267478942871
    },
    {
      "epoch": 1.9150733947753906e-05,
      "model_forward_time": 1.7594926357269287,
      "step": 94081
    },
    {
      "epoch": 1.9150733947753906e-05,
      "step": 94081,
      "training_step_time": 8.243584156036377
    },
    {
      "epoch": 1.9151687622070313e-05,
      "model_forward_time": 1.7578890323638916,
      "step": 94082
    },
    {
      "epoch": 1.9151687622070313e-05,
      "step": 94082,
      "training_step_time": 7.6498754024505615
    },
    {
      "epoch": 1.9152641296386717e-05,
      "model_forward_time": 1.7611854076385498,
      "step": 94083
    },
    {
      "epoch": 1.9152641296386717e-05,
      "step": 94083,
      "training_step_time": 7.389096975326538
    },
    {
      "epoch": 1.9153594970703125e-05,
      "model_forward_time": 1.7712903022766113,
      "step": 94084
    },
    {
      "epoch": 1.9153594970703125e-05,
      "step": 94084,
      "training_step_time": 7.641845226287842
    },
    {
      "epoch": 1.9154548645019532e-05,
      "model_forward_time": 1.7645044326782227,
      "step": 94085
    },
    {
      "epoch": 1.9154548645019532e-05,
      "step": 94085,
      "training_step_time": 7.685092926025391
    },
    {
      "epoch": 1.9155502319335936e-05,
      "model_forward_time": 1.7646489143371582,
      "step": 94086
    },
    {
      "epoch": 1.9155502319335936e-05,
      "step": 94086,
      "training_step_time": 7.133964538574219
    },
    {
      "epoch": 1.9156455993652344e-05,
      "model_forward_time": 1.7541637420654297,
      "step": 94087
    },
    {
      "epoch": 1.9156455993652344e-05,
      "step": 94087,
      "training_step_time": 8.500471591949463
    },
    {
      "epoch": 1.915740966796875e-05,
      "model_forward_time": 1.7581543922424316,
      "step": 94088
    },
    {
      "epoch": 1.915740966796875e-05,
      "step": 94088,
      "training_step_time": 9.040971279144287
    },
    {
      "epoch": 1.9158363342285155e-05,
      "model_forward_time": 2.374793529510498,
      "step": 94089
    },
    {
      "epoch": 1.9158363342285155e-05,
      "step": 94089,
      "training_step_time": 9.353387832641602
    },
    {
      "epoch": 1.9159317016601562e-05,
      "grad_norm": 0.08676742017269135,
      "learning_rate": 9.52206548356266e-08,
      "loss": 0.0563,
      "step": 94090
    },
    {
      "epoch": 1.9159317016601562e-05,
      "model_forward_time": 1.7407026290893555,
      "step": 94090
    },
    {
      "dynamics_loss_avg": 0.05187285132706165,
      "epoch": 1.9159317016601562e-05,
      "step": 94090
    },
    {
      "action_loss_avg": 0.001124517689459026,
      "epoch": 1.9159317016601562e-05,
      "step": 94090
    },
    {
      "epoch": 1.9159317016601562e-05,
      "step": 94090,
      "training_step_time": 7.757868766784668
    },
    {
      "epoch": 1.916027069091797e-05,
      "model_forward_time": 1.741255521774292,
      "step": 94091
    },
    {
      "epoch": 1.916027069091797e-05,
      "step": 94091,
      "training_step_time": 8.931541919708252
    },
    {
      "epoch": 1.9161224365234374e-05,
      "model_forward_time": 1.742906093597412,
      "step": 94092
    },
    {
      "epoch": 1.9161224365234374e-05,
      "step": 94092,
      "training_step_time": 7.396839141845703
    },
    {
      "epoch": 1.916217803955078e-05,
      "model_forward_time": 1.7436184883117676,
      "step": 94093
    },
    {
      "epoch": 1.916217803955078e-05,
      "step": 94093,
      "training_step_time": 7.685552358627319
    },
    {
      "epoch": 1.916313171386719e-05,
      "model_forward_time": 1.7447903156280518,
      "step": 94094
    },
    {
      "epoch": 1.916313171386719e-05,
      "step": 94094,
      "training_step_time": 7.691014289855957
    },
    {
      "epoch": 1.9164085388183593e-05,
      "model_forward_time": 1.7436981201171875,
      "step": 94095
    },
    {
      "epoch": 1.9164085388183593e-05,
      "step": 94095,
      "training_step_time": 6.728572845458984
    },
    {
      "epoch": 1.91650390625e-05,
      "model_forward_time": 1.8185994625091553,
      "step": 94096
    },
    {
      "epoch": 1.91650390625e-05,
      "step": 94096,
      "training_step_time": 7.763524293899536
    },
    {
      "epoch": 1.9165992736816407e-05,
      "model_forward_time": 1.7422163486480713,
      "step": 94097
    },
    {
      "epoch": 1.9165992736816407e-05,
      "step": 94097,
      "training_step_time": 8.367806911468506
    },
    {
      "epoch": 1.916694641113281e-05,
      "model_forward_time": 1.7470662593841553,
      "step": 94098
    },
    {
      "epoch": 1.916694641113281e-05,
      "step": 94098,
      "training_step_time": 9.61081075668335
    },
    {
      "epoch": 1.916790008544922e-05,
      "model_forward_time": 2.383291482925415,
      "step": 94099
    },
    {
      "epoch": 1.916790008544922e-05,
      "step": 94099,
      "training_step_time": 7.776176929473877
    },
    {
      "epoch": 1.9168853759765626e-05,
      "grad_norm": 0.08185254782438278,
      "learning_rate": 9.489976835359682e-08,
      "loss": 0.0564,
      "step": 94100
    },
    {
      "epoch": 1.9168853759765626e-05,
      "model_forward_time": 1.7402675151824951,
      "step": 94100
    },
    {
      "dynamics_loss_avg": 0.04985351897776127,
      "epoch": 1.9168853759765626e-05,
      "step": 94100
    },
    {
      "action_loss_avg": 0.0010252686333842576,
      "epoch": 1.9168853759765626e-05,
      "step": 94100
    },
    {
      "epoch": 1.9168853759765626e-05,
      "step": 94100,
      "training_step_time": 8.619181394577026
    },
    {
      "epoch": 1.916980743408203e-05,
      "model_forward_time": 1.7512805461883545,
      "step": 94101
    },
    {
      "epoch": 1.916980743408203e-05,
      "step": 94101,
      "training_step_time": 8.366857767105103
    },
    {
      "epoch": 1.9170761108398438e-05,
      "model_forward_time": 1.7422542572021484,
      "step": 94102
    },
    {
      "epoch": 1.9170761108398438e-05,
      "step": 94102,
      "training_step_time": 6.74042010307312
    },
    {
      "epoch": 1.9171714782714845e-05,
      "model_forward_time": 1.7519934177398682,
      "step": 94103
    },
    {
      "epoch": 1.9171714782714845e-05,
      "step": 94103,
      "training_step_time": 7.688976049423218
    },
    {
      "epoch": 1.917266845703125e-05,
      "model_forward_time": 1.7498626708984375,
      "step": 94104
    },
    {
      "epoch": 1.917266845703125e-05,
      "step": 94104,
      "training_step_time": 7.646569013595581
    },
    {
      "epoch": 1.9173622131347657e-05,
      "model_forward_time": 1.8124234676361084,
      "step": 94105
    },
    {
      "epoch": 1.9173622131347657e-05,
      "step": 94105,
      "training_step_time": 7.169149398803711
    },
    {
      "epoch": 1.9174575805664064e-05,
      "model_forward_time": 1.7385835647583008,
      "step": 94106
    },
    {
      "epoch": 1.9174575805664064e-05,
      "step": 94106,
      "training_step_time": 7.688204526901245
    },
    {
      "epoch": 1.9175529479980468e-05,
      "model_forward_time": 1.747441291809082,
      "step": 94107
    },
    {
      "epoch": 1.9175529479980468e-05,
      "step": 94107,
      "training_step_time": 9.003870964050293
    },
    {
      "epoch": 1.9176483154296876e-05,
      "model_forward_time": 2.3832790851593018,
      "step": 94108
    },
    {
      "epoch": 1.9176483154296876e-05,
      "step": 94108,
      "training_step_time": 11.547271251678467
    },
    {
      "epoch": 1.9177436828613283e-05,
      "model_forward_time": 1.7445030212402344,
      "step": 94109
    },
    {
      "epoch": 1.9177436828613283e-05,
      "step": 94109,
      "training_step_time": 8.622285604476929
    },
    {
      "epoch": 1.9178390502929687e-05,
      "grad_norm": 0.07238222658634186,
      "learning_rate": 9.457941828595329e-08,
      "loss": 0.0579,
      "step": 94110
    },
    {
      "epoch": 1.9178390502929687e-05,
      "model_forward_time": 1.743349313735962,
      "step": 94110
    },
    {
      "dynamics_loss_avg": 0.05334252342581749,
      "epoch": 1.9178390502929687e-05,
      "step": 94110
    },
    {
      "action_loss_avg": 0.001375017606187612,
      "epoch": 1.9178390502929687e-05,
      "step": 94110
    },
    {
      "epoch": 1.9178390502929687e-05,
      "step": 94110,
      "training_step_time": 8.491145849227905
    },
    {
      "epoch": 1.9179344177246094e-05,
      "model_forward_time": 1.7476322650909424,
      "step": 94111
    },
    {
      "epoch": 1.9179344177246094e-05,
      "step": 94111,
      "training_step_time": 7.316982984542847
    },
    {
      "epoch": 1.91802978515625e-05,
      "model_forward_time": 1.7403759956359863,
      "step": 94112
    },
    {
      "epoch": 1.91802978515625e-05,
      "step": 94112,
      "training_step_time": 7.765167474746704
    },
    {
      "epoch": 1.9181251525878906e-05,
      "model_forward_time": 1.746946096420288,
      "step": 94113
    },
    {
      "epoch": 1.9181251525878906e-05,
      "step": 94113,
      "training_step_time": 7.699932813644409
    },
    {
      "epoch": 1.9182205200195313e-05,
      "model_forward_time": 1.8267664909362793,
      "step": 94114
    },
    {
      "epoch": 1.9182205200195313e-05,
      "step": 94114,
      "training_step_time": 6.8000876903533936
    },
    {
      "epoch": 1.9183158874511717e-05,
      "model_forward_time": 1.7429523468017578,
      "step": 94115
    },
    {
      "epoch": 1.9183158874511717e-05,
      "step": 94115,
      "training_step_time": 8.429728984832764
    },
    {
      "epoch": 1.9184112548828125e-05,
      "model_forward_time": 1.7574090957641602,
      "step": 94116
    },
    {
      "epoch": 1.9184112548828125e-05,
      "step": 94116,
      "training_step_time": 7.6812825202941895
    },
    {
      "epoch": 1.9185066223144532e-05,
      "model_forward_time": 2.453500747680664,
      "step": 94117
    },
    {
      "epoch": 1.9185066223144532e-05,
      "step": 94117,
      "training_step_time": 8.508481979370117
    },
    {
      "epoch": 1.9186019897460936e-05,
      "model_forward_time": 1.7617871761322021,
      "step": 94118
    },
    {
      "epoch": 1.9186019897460936e-05,
      "step": 94118,
      "training_step_time": 8.41320514678955
    },
    {
      "epoch": 1.9186973571777344e-05,
      "model_forward_time": 1.7593042850494385,
      "step": 94119
    },
    {
      "epoch": 1.9186973571777344e-05,
      "step": 94119,
      "training_step_time": 7.624930381774902
    },
    {
      "epoch": 1.918792724609375e-05,
      "grad_norm": 0.09120824187994003,
      "learning_rate": 9.425960466772966e-08,
      "loss": 0.0547,
      "step": 94120
    },
    {
      "epoch": 1.918792724609375e-05,
      "model_forward_time": 1.7632572650909424,
      "step": 94120
    },
    {
      "dynamics_loss_avg": 0.05770758986473083,
      "epoch": 1.918792724609375e-05,
      "step": 94120
    },
    {
      "action_loss_avg": 0.0011363153345882893,
      "epoch": 1.918792724609375e-05,
      "step": 94120
    },
    {
      "epoch": 1.918792724609375e-05,
      "step": 94120,
      "training_step_time": 7.184275388717651
    },
    {
      "epoch": 1.9188880920410155e-05,
      "model_forward_time": 1.7586898803710938,
      "step": 94121
    },
    {
      "epoch": 1.9188880920410155e-05,
      "step": 94121,
      "training_step_time": 7.5466132164001465
    },
    {
      "epoch": 1.9189834594726562e-05,
      "model_forward_time": 1.7537181377410889,
      "step": 94122
    },
    {
      "epoch": 1.9189834594726562e-05,
      "step": 94122,
      "training_step_time": 7.581730127334595
    },
    {
      "epoch": 1.919078826904297e-05,
      "model_forward_time": 1.760802984237671,
      "step": 94123
    },
    {
      "epoch": 1.919078826904297e-05,
      "step": 94123,
      "training_step_time": 7.052441835403442
    },
    {
      "epoch": 1.9191741943359374e-05,
      "model_forward_time": 1.7486255168914795,
      "step": 94124
    },
    {
      "epoch": 1.9191741943359374e-05,
      "step": 94124,
      "training_step_time": 7.64827036857605
    },
    {
      "epoch": 1.919269561767578e-05,
      "model_forward_time": 1.7549545764923096,
      "step": 94125
    },
    {
      "epoch": 1.919269561767578e-05,
      "step": 94125,
      "training_step_time": 9.34678864479065
    },
    {
      "epoch": 1.919364929199219e-05,
      "model_forward_time": 2.428455114364624,
      "step": 94126
    },
    {
      "epoch": 1.919364929199219e-05,
      "step": 94126,
      "training_step_time": 7.670356035232544
    },
    {
      "epoch": 1.9194602966308593e-05,
      "model_forward_time": 1.7543878555297852,
      "step": 94127
    },
    {
      "epoch": 1.9194602966308593e-05,
      "step": 94127,
      "training_step_time": 9.771969318389893
    },
    {
      "epoch": 1.9195556640625e-05,
      "model_forward_time": 1.7697019577026367,
      "step": 94128
    },
    {
      "epoch": 1.9195556640625e-05,
      "step": 94128,
      "training_step_time": 9.099826097488403
    },
    {
      "epoch": 1.9196510314941408e-05,
      "model_forward_time": 1.7561438083648682,
      "step": 94129
    },
    {
      "epoch": 1.9196510314941408e-05,
      "step": 94129,
      "training_step_time": 7.2760889530181885
    },
    {
      "epoch": 1.919746398925781e-05,
      "grad_norm": 0.07373262941837311,
      "learning_rate": 9.394032753390014e-08,
      "loss": 0.0546,
      "step": 94130
    },
    {
      "epoch": 1.919746398925781e-05,
      "model_forward_time": 1.7577812671661377,
      "step": 94130
    },
    {
      "dynamics_loss_avg": 0.05124050080776214,
      "epoch": 1.919746398925781e-05,
      "step": 94130
    },
    {
      "action_loss_avg": 0.0017827232572017238,
      "epoch": 1.919746398925781e-05,
      "step": 94130
    },
    {
      "epoch": 1.919746398925781e-05,
      "step": 94130,
      "training_step_time": 7.734471797943115
    },
    {
      "epoch": 1.919841766357422e-05,
      "model_forward_time": 1.781130075454712,
      "step": 94131
    },
    {
      "epoch": 1.919841766357422e-05,
      "step": 94131,
      "training_step_time": 7.5987162590026855
    },
    {
      "epoch": 1.9199371337890626e-05,
      "model_forward_time": 1.7574050426483154,
      "step": 94132
    },
    {
      "epoch": 1.9199371337890626e-05,
      "step": 94132,
      "training_step_time": 6.690676212310791
    },
    {
      "epoch": 1.920032501220703e-05,
      "model_forward_time": 1.8187754154205322,
      "step": 94133
    },
    {
      "epoch": 1.920032501220703e-05,
      "step": 94133,
      "training_step_time": 7.692446708679199
    },
    {
      "epoch": 1.9201278686523438e-05,
      "model_forward_time": 1.7657344341278076,
      "step": 94134
    },
    {
      "epoch": 1.9201278686523438e-05,
      "step": 94134,
      "training_step_time": 8.593675374984741
    },
    {
      "epoch": 1.9202232360839845e-05,
      "model_forward_time": 2.404283046722412,
      "step": 94135
    },
    {
      "epoch": 1.9202232360839845e-05,
      "step": 94135,
      "training_step_time": 8.509396076202393
    },
    {
      "epoch": 1.920318603515625e-05,
      "model_forward_time": 1.7600791454315186,
      "step": 94136
    },
    {
      "epoch": 1.920318603515625e-05,
      "step": 94136,
      "training_step_time": 7.709232330322266
    },
    {
      "epoch": 1.9204139709472657e-05,
      "model_forward_time": 1.7527847290039062,
      "step": 94137
    },
    {
      "epoch": 1.9204139709472657e-05,
      "step": 94137,
      "training_step_time": 9.807900428771973
    },
    {
      "epoch": 1.9205093383789064e-05,
      "model_forward_time": 1.7615432739257812,
      "step": 94138
    },
    {
      "epoch": 1.9205093383789064e-05,
      "step": 94138,
      "training_step_time": 8.379337310791016
    },
    {
      "epoch": 1.9206047058105468e-05,
      "model_forward_time": 1.384016752243042,
      "step": 94139
    },
    {
      "epoch": 1.9206047058105468e-05,
      "step": 94139,
      "training_step_time": 7.328370571136475
    },
    {
      "epoch": 1.9207000732421876e-05,
      "grad_norm": 0.15841485559940338,
      "learning_rate": 9.362158691938017e-08,
      "loss": 0.0561,
      "step": 94140
    },
    {
      "epoch": 1.9207000732421876e-05,
      "model_forward_time": 1.764561653137207,
      "step": 94140
    },
    {
      "dynamics_loss_avg": 0.060924735572189095,
      "epoch": 1.9207000732421876e-05,
      "step": 94140
    },
    {
      "action_loss_avg": 0.0016661328030750156,
      "epoch": 1.9207000732421876e-05,
      "step": 94140
    },
    {
      "epoch": 1.9207000732421876e-05,
      "step": 94140,
      "training_step_time": 7.752239942550659
    },
    {
      "epoch": 1.920795440673828e-05,
      "model_forward_time": 1.7600789070129395,
      "step": 94141
    },
    {
      "epoch": 1.920795440673828e-05,
      "step": 94141,
      "training_step_time": 6.780330181121826
    },
    {
      "epoch": 1.9208908081054687e-05,
      "model_forward_time": 1.818981647491455,
      "step": 94142
    },
    {
      "epoch": 1.9208908081054687e-05,
      "step": 94142,
      "training_step_time": 7.8183770179748535
    },
    {
      "epoch": 1.9209861755371094e-05,
      "model_forward_time": 1.7564692497253418,
      "step": 94143
    },
    {
      "epoch": 1.9209861755371094e-05,
      "step": 94143,
      "training_step_time": 8.53202509880066
    },
    {
      "epoch": 1.92108154296875e-05,
      "model_forward_time": 1.7548489570617676,
      "step": 94144
    },
    {
      "epoch": 1.92108154296875e-05,
      "step": 94144,
      "training_step_time": 8.885762691497803
    },
    {
      "epoch": 1.9211769104003906e-05,
      "model_forward_time": 2.3920133113861084,
      "step": 94145
    },
    {
      "epoch": 1.9211769104003906e-05,
      "step": 94145,
      "training_step_time": 8.750405073165894
    },
    {
      "epoch": 1.9212722778320313e-05,
      "model_forward_time": 1.75639009475708,
      "step": 94146
    },
    {
      "epoch": 1.9212722778320313e-05,
      "step": 94146,
      "training_step_time": 8.809314489364624
    },
    {
      "epoch": 1.9213676452636717e-05,
      "model_forward_time": 1.7592475414276123,
      "step": 94147
    },
    {
      "epoch": 1.9213676452636717e-05,
      "step": 94147,
      "training_step_time": 7.790410995483398
    },
    {
      "epoch": 1.9214630126953125e-05,
      "model_forward_time": 1.7500298023223877,
      "step": 94148
    },
    {
      "epoch": 1.9214630126953125e-05,
      "step": 94148,
      "training_step_time": 7.312082290649414
    },
    {
      "epoch": 1.9215583801269532e-05,
      "model_forward_time": 1.7667810916900635,
      "step": 94149
    },
    {
      "epoch": 1.9215583801269532e-05,
      "step": 94149,
      "training_step_time": 8.370797634124756
    },
    {
      "epoch": 1.9216537475585936e-05,
      "grad_norm": 0.07399563491344452,
      "learning_rate": 9.330338285902685e-08,
      "loss": 0.055,
      "step": 94150
    },
    {
      "epoch": 1.9216537475585936e-05,
      "model_forward_time": 1.7661044597625732,
      "step": 94150
    },
    {
      "dynamics_loss_avg": 0.07686720192432403,
      "epoch": 1.9216537475585936e-05,
      "step": 94150
    },
    {
      "action_loss_avg": 0.0014087398070842029,
      "epoch": 1.9216537475585936e-05,
      "step": 94150
    },
    {
      "epoch": 1.9216537475585936e-05,
      "step": 94150,
      "training_step_time": 6.74622654914856
    },
    {
      "epoch": 1.9217491149902344e-05,
      "model_forward_time": 1.8331727981567383,
      "step": 94151
    },
    {
      "epoch": 1.9217491149902344e-05,
      "step": 94151,
      "training_step_time": 7.869279623031616
    },
    {
      "epoch": 1.921844482421875e-05,
      "model_forward_time": 1.755669355392456,
      "step": 94152
    },
    {
      "epoch": 1.921844482421875e-05,
      "step": 94152,
      "training_step_time": 8.932285070419312
    },
    {
      "epoch": 1.9219398498535155e-05,
      "model_forward_time": 1.7580695152282715,
      "step": 94153
    },
    {
      "epoch": 1.9219398498535155e-05,
      "step": 94153,
      "training_step_time": 10.296988487243652
    },
    {
      "epoch": 1.9220352172851562e-05,
      "model_forward_time": 2.402125358581543,
      "step": 94154
    },
    {
      "epoch": 1.9220352172851562e-05,
      "step": 94154,
      "training_step_time": 9.65956997871399
    },
    {
      "epoch": 1.922130584716797e-05,
      "model_forward_time": 1.756685495376587,
      "step": 94155
    },
    {
      "epoch": 1.922130584716797e-05,
      "step": 94155,
      "training_step_time": 10.110579013824463
    },
    {
      "epoch": 1.9222259521484374e-05,
      "model_forward_time": 1.756242036819458,
      "step": 94156
    },
    {
      "epoch": 1.9222259521484374e-05,
      "step": 94156,
      "training_step_time": 7.850241661071777
    },
    {
      "epoch": 1.922321319580078e-05,
      "model_forward_time": 1.7578043937683105,
      "step": 94157
    },
    {
      "epoch": 1.922321319580078e-05,
      "step": 94157,
      "training_step_time": 7.403090715408325
    },
    {
      "epoch": 1.922416687011719e-05,
      "model_forward_time": 1.7569694519042969,
      "step": 94158
    },
    {
      "epoch": 1.922416687011719e-05,
      "step": 94158,
      "training_step_time": 7.750030040740967
    },
    {
      "epoch": 1.9225120544433593e-05,
      "model_forward_time": 1.731825828552246,
      "step": 94159
    },
    {
      "epoch": 1.9225120544433593e-05,
      "step": 94159,
      "training_step_time": 7.781316518783569
    },
    {
      "epoch": 1.922607421875e-05,
      "grad_norm": 0.11419907957315445,
      "learning_rate": 9.298571538763846e-08,
      "loss": 0.0544,
      "step": 94160
    },
    {
      "epoch": 1.922607421875e-05,
      "model_forward_time": 1.8119323253631592,
      "step": 94160
    },
    {
      "dynamics_loss_avg": 0.05063760746270418,
      "epoch": 1.922607421875e-05,
      "step": 94160
    },
    {
      "action_loss_avg": 0.001150312731624581,
      "epoch": 1.922607421875e-05,
      "step": 94160
    },
    {
      "epoch": 1.922607421875e-05,
      "step": 94160,
      "training_step_time": 7.747343301773071
    },
    {
      "epoch": 1.9227027893066408e-05,
      "model_forward_time": 1.7244701385498047,
      "step": 94161
    },
    {
      "epoch": 1.9227027893066408e-05,
      "step": 94161,
      "training_step_time": 8.5701162815094
    },
    {
      "epoch": 1.922798156738281e-05,
      "model_forward_time": 1.782057523727417,
      "step": 94162
    },
    {
      "epoch": 1.922798156738281e-05,
      "step": 94162,
      "training_step_time": 7.923832178115845
    },
    {
      "epoch": 1.922893524169922e-05,
      "model_forward_time": 2.391312837600708,
      "step": 94163
    },
    {
      "epoch": 1.922893524169922e-05,
      "step": 94163,
      "training_step_time": 8.753106594085693
    },
    {
      "epoch": 1.9229888916015626e-05,
      "model_forward_time": 1.7385749816894531,
      "step": 94164
    },
    {
      "epoch": 1.9229888916015626e-05,
      "step": 94164,
      "training_step_time": 7.851133584976196
    },
    {
      "epoch": 1.923084259033203e-05,
      "model_forward_time": 1.7493860721588135,
      "step": 94165
    },
    {
      "epoch": 1.923084259033203e-05,
      "step": 94165,
      "training_step_time": 8.497780561447144
    },
    {
      "epoch": 1.9231796264648438e-05,
      "model_forward_time": 1.7289958000183105,
      "step": 94166
    },
    {
      "epoch": 1.9231796264648438e-05,
      "step": 94166,
      "training_step_time": 7.678117036819458
    },
    {
      "epoch": 1.9232749938964845e-05,
      "model_forward_time": 1.7398037910461426,
      "step": 94167
    },
    {
      "epoch": 1.9232749938964845e-05,
      "step": 94167,
      "training_step_time": 7.807621240615845
    },
    {
      "epoch": 1.923370361328125e-05,
      "model_forward_time": 1.7481269836425781,
      "step": 94168
    },
    {
      "epoch": 1.923370361328125e-05,
      "step": 94168,
      "training_step_time": 7.780710697174072
    },
    {
      "epoch": 1.9234657287597657e-05,
      "model_forward_time": 1.8045904636383057,
      "step": 94169
    },
    {
      "epoch": 1.9234657287597657e-05,
      "step": 94169,
      "training_step_time": 8.543052911758423
    },
    {
      "epoch": 1.9235610961914064e-05,
      "grad_norm": 0.08941560983657837,
      "learning_rate": 9.266858453995442e-08,
      "loss": 0.0536,
      "step": 94170
    },
    {
      "epoch": 1.9235610961914064e-05,
      "model_forward_time": 1.7363226413726807,
      "step": 94170
    },
    {
      "dynamics_loss_avg": 0.04839045479893685,
      "epoch": 1.9235610961914064e-05,
      "step": 94170
    },
    {
      "action_loss_avg": 0.0015127278456930071,
      "epoch": 1.9235610961914064e-05,
      "step": 94170
    },
    {
      "epoch": 1.9235610961914064e-05,
      "step": 94170,
      "training_step_time": 7.7131102085113525
    },
    {
      "epoch": 1.9236564636230468e-05,
      "model_forward_time": 1.7428910732269287,
      "step": 94171
    },
    {
      "epoch": 1.9236564636230468e-05,
      "step": 94171,
      "training_step_time": 8.857490062713623
    },
    {
      "epoch": 1.9237518310546876e-05,
      "model_forward_time": 2.398030996322632,
      "step": 94172
    },
    {
      "epoch": 1.9237518310546876e-05,
      "step": 94172,
      "training_step_time": 8.675655364990234
    },
    {
      "epoch": 1.923847198486328e-05,
      "model_forward_time": 1.744138240814209,
      "step": 94173
    },
    {
      "epoch": 1.923847198486328e-05,
      "step": 94173,
      "training_step_time": 9.554492712020874
    },
    {
      "epoch": 1.9239425659179687e-05,
      "model_forward_time": 1.7414109706878662,
      "step": 94174
    },
    {
      "epoch": 1.9239425659179687e-05,
      "step": 94174,
      "training_step_time": 7.821706056594849
    },
    {
      "epoch": 1.9240379333496094e-05,
      "model_forward_time": 1.7406988143920898,
      "step": 94175
    },
    {
      "epoch": 1.9240379333496094e-05,
      "step": 94175,
      "training_step_time": 8.337160110473633
    },
    {
      "epoch": 1.92413330078125e-05,
      "model_forward_time": 1.750608205795288,
      "step": 94176
    },
    {
      "epoch": 1.92413330078125e-05,
      "step": 94176,
      "training_step_time": 7.776739835739136
    },
    {
      "epoch": 1.9242286682128906e-05,
      "model_forward_time": 1.7403972148895264,
      "step": 94177
    },
    {
      "epoch": 1.9242286682128906e-05,
      "step": 94177,
      "training_step_time": 7.710863351821899
    },
    {
      "epoch": 1.9243240356445313e-05,
      "model_forward_time": 1.7760727405548096,
      "step": 94178
    },
    {
      "epoch": 1.9243240356445313e-05,
      "step": 94178,
      "training_step_time": 7.549148082733154
    },
    {
      "epoch": 1.9244194030761717e-05,
      "model_forward_time": 1.7398340702056885,
      "step": 94179
    },
    {
      "epoch": 1.9244194030761717e-05,
      "step": 94179,
      "training_step_time": 8.649829626083374
    },
    {
      "epoch": 1.9245147705078125e-05,
      "grad_norm": 0.08790480345487595,
      "learning_rate": 9.235199035065646e-08,
      "loss": 0.0535,
      "step": 94180
    },
    {
      "epoch": 1.9245147705078125e-05,
      "model_forward_time": 1.7404451370239258,
      "step": 94180
    },
    {
      "dynamics_loss_avg": 0.045157173089683054,
      "epoch": 1.9245147705078125e-05,
      "step": 94180
    },
    {
      "action_loss_avg": 0.0014499704848276451,
      "epoch": 1.9245147705078125e-05,
      "step": 94180
    },
    {
      "epoch": 1.9245147705078125e-05,
      "step": 94180,
      "training_step_time": 10.215568780899048
    },
    {
      "epoch": 1.9246101379394532e-05,
      "model_forward_time": 2.372422218322754,
      "step": 94181
    },
    {
      "epoch": 1.9246101379394532e-05,
      "step": 94181,
      "training_step_time": 8.488762855529785
    },
    {
      "epoch": 1.9247055053710936e-05,
      "model_forward_time": 1.7464582920074463,
      "step": 94182
    },
    {
      "epoch": 1.9247055053710936e-05,
      "step": 94182,
      "training_step_time": 7.794565439224243
    },
    {
      "epoch": 1.9248008728027344e-05,
      "model_forward_time": 1.7428481578826904,
      "step": 94183
    },
    {
      "epoch": 1.9248008728027344e-05,
      "step": 94183,
      "training_step_time": 8.44092869758606
    },
    {
      "epoch": 1.924896240234375e-05,
      "model_forward_time": 1.7464191913604736,
      "step": 94184
    },
    {
      "epoch": 1.924896240234375e-05,
      "step": 94184,
      "training_step_time": 7.672189235687256
    },
    {
      "epoch": 1.9249916076660155e-05,
      "model_forward_time": 1.7483007907867432,
      "step": 94185
    },
    {
      "epoch": 1.9249916076660155e-05,
      "step": 94185,
      "training_step_time": 7.719022989273071
    },
    {
      "epoch": 1.9250869750976562e-05,
      "model_forward_time": 1.7366445064544678,
      "step": 94186
    },
    {
      "epoch": 1.9250869750976562e-05,
      "step": 94186,
      "training_step_time": 7.266592741012573
    },
    {
      "epoch": 1.925182342529297e-05,
      "model_forward_time": 1.7552852630615234,
      "step": 94187
    },
    {
      "epoch": 1.925182342529297e-05,
      "step": 94187,
      "training_step_time": 8.58463716506958
    },
    {
      "epoch": 1.9252777099609374e-05,
      "model_forward_time": 1.7439143657684326,
      "step": 94188
    },
    {
      "epoch": 1.9252777099609374e-05,
      "step": 94188,
      "training_step_time": 9.219551086425781
    },
    {
      "epoch": 1.925373077392578e-05,
      "model_forward_time": 1.7338457107543945,
      "step": 94189
    },
    {
      "epoch": 1.925373077392578e-05,
      "step": 94189,
      "training_step_time": 7.860281705856323
    },
    {
      "epoch": 1.925468444824219e-05,
      "grad_norm": 0.10113741457462311,
      "learning_rate": 9.203593285436573e-08,
      "loss": 0.0532,
      "step": 94190
    },
    {
      "epoch": 1.925468444824219e-05,
      "model_forward_time": 1.7437751293182373,
      "step": 94190
    },
    {
      "dynamics_loss_avg": 0.05722670964896679,
      "epoch": 1.925468444824219e-05,
      "step": 94190
    },
    {
      "action_loss_avg": 0.001350752516009379,
      "epoch": 1.925468444824219e-05,
      "step": 94190
    },
    {
      "epoch": 1.925468444824219e-05,
      "step": 94190,
      "training_step_time": 9.749647378921509
    },
    {
      "epoch": 1.9255638122558593e-05,
      "model_forward_time": 2.4733638763427734,
      "step": 94191
    },
    {
      "epoch": 1.9255638122558593e-05,
      "step": 94191,
      "training_step_time": 8.834330081939697
    },
    {
      "epoch": 1.9256591796875e-05,
      "model_forward_time": 1.7415852546691895,
      "step": 94192
    },
    {
      "epoch": 1.9256591796875e-05,
      "step": 94192,
      "training_step_time": 7.885905027389526
    },
    {
      "epoch": 1.9257545471191408e-05,
      "model_forward_time": 1.7413091659545898,
      "step": 94193
    },
    {
      "epoch": 1.9257545471191408e-05,
      "step": 94193,
      "training_step_time": 7.788577556610107
    },
    {
      "epoch": 1.925849914550781e-05,
      "model_forward_time": 1.7461531162261963,
      "step": 94194
    },
    {
      "epoch": 1.925849914550781e-05,
      "step": 94194,
      "training_step_time": 7.68095588684082
    },
    {
      "epoch": 1.925945281982422e-05,
      "model_forward_time": 1.743248462677002,
      "step": 94195
    },
    {
      "epoch": 1.925945281982422e-05,
      "step": 94195,
      "training_step_time": 7.798934459686279
    },
    {
      "epoch": 1.9260406494140626e-05,
      "model_forward_time": 1.7374567985534668,
      "step": 94196
    },
    {
      "epoch": 1.9260406494140626e-05,
      "step": 94196,
      "training_step_time": 7.64478325843811
    },
    {
      "epoch": 1.926136016845703e-05,
      "model_forward_time": 1.7969400882720947,
      "step": 94197
    },
    {
      "epoch": 1.926136016845703e-05,
      "step": 94197,
      "training_step_time": 7.800491094589233
    },
    {
      "epoch": 1.9262313842773438e-05,
      "model_forward_time": 1.7451634407043457,
      "step": 94198
    },
    {
      "epoch": 1.9262313842773438e-05,
      "step": 94198,
      "training_step_time": 8.85196042060852
    },
    {
      "epoch": 1.9263267517089845e-05,
      "model_forward_time": 1.3805475234985352,
      "step": 94199
    },
    {
      "epoch": 1.9263267517089845e-05,
      "step": 94199,
      "training_step_time": 10.795356273651123
    },
    {
      "epoch": 1.926422119140625e-05,
      "grad_norm": 0.067111536860466,
      "learning_rate": 9.172041208564687e-08,
      "loss": 0.0532,
      "step": 94200
    },
    {
      "epoch": 1.926422119140625e-05,
      "model_forward_time": 2.3622002601623535,
      "step": 94200
    },
    {
      "dynamics_loss_avg": 0.058201163448393343,
      "epoch": 1.926422119140625e-05,
      "step": 94200
    },
    {
      "action_loss_avg": 0.001506166806211695,
      "epoch": 1.926422119140625e-05,
      "step": 94200
    },
    {
      "epoch": 1.926422119140625e-05,
      "step": 94200,
      "training_step_time": 8.885841131210327
    },
    {
      "epoch": 1.9265174865722657e-05,
      "model_forward_time": 1.7434160709381104,
      "step": 94201
    },
    {
      "epoch": 1.9265174865722657e-05,
      "step": 94201,
      "training_step_time": 7.997397422790527
    },
    {
      "epoch": 1.9266128540039064e-05,
      "model_forward_time": 1.7458419799804688,
      "step": 94202
    },
    {
      "epoch": 1.9266128540039064e-05,
      "step": 94202,
      "training_step_time": 7.879379034042358
    },
    {
      "epoch": 1.9267082214355468e-05,
      "model_forward_time": 1.7382924556732178,
      "step": 94203
    },
    {
      "epoch": 1.9267082214355468e-05,
      "step": 94203,
      "training_step_time": 7.82301926612854
    },
    {
      "epoch": 1.9268035888671876e-05,
      "model_forward_time": 1.7425100803375244,
      "step": 94204
    },
    {
      "epoch": 1.9268035888671876e-05,
      "step": 94204,
      "training_step_time": 7.857357501983643
    },
    {
      "epoch": 1.926898956298828e-05,
      "model_forward_time": 1.7533936500549316,
      "step": 94205
    },
    {
      "epoch": 1.926898956298828e-05,
      "step": 94205,
      "training_step_time": 7.611565351486206
    },
    {
      "epoch": 1.9269943237304687e-05,
      "model_forward_time": 1.7780852317810059,
      "step": 94206
    },
    {
      "epoch": 1.9269943237304687e-05,
      "step": 94206,
      "training_step_time": 7.755992412567139
    },
    {
      "epoch": 1.9270896911621094e-05,
      "model_forward_time": 1.7312769889831543,
      "step": 94207
    },
    {
      "epoch": 1.9270896911621094e-05,
      "step": 94207,
      "training_step_time": 9.320720911026001
    },
    {
      "epoch": 1.92718505859375e-05,
      "model_forward_time": 1.7457807064056396,
      "step": 94208
    },
    {
      "epoch": 1.92718505859375e-05,
      "step": 94208,
      "training_step_time": 9.341290712356567
    },
    {
      "epoch": 1.9272804260253906e-05,
      "model_forward_time": 2.37685227394104,
      "step": 94209
    },
    {
      "epoch": 1.9272804260253906e-05,
      "step": 94209,
      "training_step_time": 9.465746641159058
    },
    {
      "epoch": 1.9273757934570313e-05,
      "grad_norm": 0.08875154703855515,
      "learning_rate": 9.140542807900333e-08,
      "loss": 0.0547,
      "step": 94210
    },
    {
      "epoch": 1.9273757934570313e-05,
      "model_forward_time": 1.7385532855987549,
      "step": 94210
    },
    {
      "dynamics_loss_avg": 0.046122510079294445,
      "epoch": 1.9273757934570313e-05,
      "step": 94210
    },
    {
      "action_loss_avg": 0.001107403483183589,
      "epoch": 1.9273757934570313e-05,
      "step": 94210
    },
    {
      "epoch": 1.9273757934570313e-05,
      "step": 94210,
      "training_step_time": 7.897257089614868
    },
    {
      "epoch": 1.9274711608886717e-05,
      "model_forward_time": 1.7405848503112793,
      "step": 94211
    },
    {
      "epoch": 1.9274711608886717e-05,
      "step": 94211,
      "training_step_time": 7.809962034225464
    },
    {
      "epoch": 1.9275665283203125e-05,
      "model_forward_time": 1.7290027141571045,
      "step": 94212
    },
    {
      "epoch": 1.9275665283203125e-05,
      "step": 94212,
      "training_step_time": 7.78087854385376
    },
    {
      "epoch": 1.9276618957519532e-05,
      "model_forward_time": 1.7433481216430664,
      "step": 94213
    },
    {
      "epoch": 1.9276618957519532e-05,
      "step": 94213,
      "training_step_time": 7.743972301483154
    },
    {
      "epoch": 1.9277572631835936e-05,
      "model_forward_time": 1.7772190570831299,
      "step": 94214
    },
    {
      "epoch": 1.9277572631835936e-05,
      "step": 94214,
      "training_step_time": 7.594009160995483
    },
    {
      "epoch": 1.9278526306152344e-05,
      "model_forward_time": 1.738011360168457,
      "step": 94215
    },
    {
      "epoch": 1.9278526306152344e-05,
      "step": 94215,
      "training_step_time": 9.46866250038147
    },
    {
      "epoch": 1.927947998046875e-05,
      "model_forward_time": 1.7572791576385498,
      "step": 94216
    },
    {
      "epoch": 1.927947998046875e-05,
      "step": 94216,
      "training_step_time": 7.792011022567749
    },
    {
      "epoch": 1.9280433654785155e-05,
      "model_forward_time": 1.7378957271575928,
      "step": 94217
    },
    {
      "epoch": 1.9280433654785155e-05,
      "step": 94217,
      "training_step_time": 8.554767370223999
    },
    {
      "epoch": 1.9281387329101562e-05,
      "model_forward_time": 2.4215755462646484,
      "step": 94218
    },
    {
      "epoch": 1.9281387329101562e-05,
      "step": 94218,
      "training_step_time": 8.549082040786743
    },
    {
      "epoch": 1.928234100341797e-05,
      "model_forward_time": 1.7393977642059326,
      "step": 94219
    },
    {
      "epoch": 1.928234100341797e-05,
      "step": 94219,
      "training_step_time": 8.761666536331177
    },
    {
      "epoch": 1.9283294677734374e-05,
      "grad_norm": 0.13626952469348907,
      "learning_rate": 9.109098086888313e-08,
      "loss": 0.0542,
      "step": 94220
    },
    {
      "epoch": 1.9283294677734374e-05,
      "model_forward_time": 1.7422809600830078,
      "step": 94220
    },
    {
      "dynamics_loss_avg": 0.04370494270697236,
      "epoch": 1.9283294677734374e-05,
      "step": 94220
    },
    {
      "action_loss_avg": 0.0010686916168197059,
      "epoch": 1.9283294677734374e-05,
      "step": 94220
    },
    {
      "epoch": 1.9283294677734374e-05,
      "step": 94220,
      "training_step_time": 7.903741836547852
    },
    {
      "epoch": 1.928424835205078e-05,
      "model_forward_time": 1.7397150993347168,
      "step": 94221
    },
    {
      "epoch": 1.928424835205078e-05,
      "step": 94221,
      "training_step_time": 7.8539416790008545
    },
    {
      "epoch": 1.928520202636719e-05,
      "model_forward_time": 1.7446224689483643,
      "step": 94222
    },
    {
      "epoch": 1.928520202636719e-05,
      "step": 94222,
      "training_step_time": 7.4312169551849365
    },
    {
      "epoch": 1.9286155700683593e-05,
      "model_forward_time": 1.7405872344970703,
      "step": 94223
    },
    {
      "epoch": 1.9286155700683593e-05,
      "step": 94223,
      "training_step_time": 7.768602609634399
    },
    {
      "epoch": 1.9287109375e-05,
      "model_forward_time": 1.7408835887908936,
      "step": 94224
    },
    {
      "epoch": 1.9287109375e-05,
      "step": 94224,
      "training_step_time": 8.43944787979126
    },
    {
      "epoch": 1.9288063049316408e-05,
      "model_forward_time": 1.735539436340332,
      "step": 94225
    },
    {
      "epoch": 1.9288063049316408e-05,
      "step": 94225,
      "training_step_time": 9.558786630630493
    },
    {
      "epoch": 1.928901672363281e-05,
      "model_forward_time": 1.7458407878875732,
      "step": 94226
    },
    {
      "epoch": 1.928901672363281e-05,
      "step": 94226,
      "training_step_time": 7.814957857131958
    },
    {
      "epoch": 1.928997039794922e-05,
      "model_forward_time": 2.392693042755127,
      "step": 94227
    },
    {
      "epoch": 1.928997039794922e-05,
      "step": 94227,
      "training_step_time": 9.745731592178345
    },
    {
      "epoch": 1.9290924072265626e-05,
      "model_forward_time": 1.741950273513794,
      "step": 94228
    },
    {
      "epoch": 1.9290924072265626e-05,
      "step": 94228,
      "training_step_time": 7.885638236999512
    },
    {
      "epoch": 1.929187774658203e-05,
      "model_forward_time": 1.7353150844573975,
      "step": 94229
    },
    {
      "epoch": 1.929187774658203e-05,
      "step": 94229,
      "training_step_time": 8.704859733581543
    },
    {
      "epoch": 1.9292831420898438e-05,
      "grad_norm": 0.13763827085494995,
      "learning_rate": 9.077707048967265e-08,
      "loss": 0.0547,
      "step": 94230
    },
    {
      "epoch": 1.9292831420898438e-05,
      "model_forward_time": 1.7340419292449951,
      "step": 94230
    },
    {
      "dynamics_loss_avg": 0.049832146242260934,
      "epoch": 1.9292831420898438e-05,
      "step": 94230
    },
    {
      "action_loss_avg": 0.0014509601052850485,
      "epoch": 1.9292831420898438e-05,
      "step": 94230
    },
    {
      "epoch": 1.9292831420898438e-05,
      "step": 94230,
      "training_step_time": 8.505850553512573
    },
    {
      "epoch": 1.9293785095214845e-05,
      "model_forward_time": 1.7473971843719482,
      "step": 94231
    },
    {
      "epoch": 1.9293785095214845e-05,
      "step": 94231,
      "training_step_time": 7.397372007369995
    },
    {
      "epoch": 1.929473876953125e-05,
      "model_forward_time": 1.7396879196166992,
      "step": 94232
    },
    {
      "epoch": 1.929473876953125e-05,
      "step": 94232,
      "training_step_time": 7.822892189025879
    },
    {
      "epoch": 1.9295692443847657e-05,
      "model_forward_time": 1.7384743690490723,
      "step": 94233
    },
    {
      "epoch": 1.9295692443847657e-05,
      "step": 94233,
      "training_step_time": 7.525167226791382
    },
    {
      "epoch": 1.9296646118164064e-05,
      "model_forward_time": 1.8083913326263428,
      "step": 94234
    },
    {
      "epoch": 1.9296646118164064e-05,
      "step": 94234,
      "training_step_time": 7.864740371704102
    },
    {
      "epoch": 1.9297599792480468e-05,
      "model_forward_time": 1.744462490081787,
      "step": 94235
    },
    {
      "epoch": 1.9297599792480468e-05,
      "step": 94235,
      "training_step_time": 7.865494251251221
    },
    {
      "epoch": 1.9298553466796876e-05,
      "model_forward_time": 2.3223705291748047,
      "step": 94236
    },
    {
      "epoch": 1.9298553466796876e-05,
      "step": 94236,
      "training_step_time": 9.00266432762146
    },
    {
      "epoch": 1.929950714111328e-05,
      "model_forward_time": 1.7533721923828125,
      "step": 94237
    },
    {
      "epoch": 1.929950714111328e-05,
      "step": 94237,
      "training_step_time": 7.984302997589111
    },
    {
      "epoch": 1.9300460815429687e-05,
      "model_forward_time": 1.7777364253997803,
      "step": 94238
    },
    {
      "epoch": 1.9300460815429687e-05,
      "step": 94238,
      "training_step_time": 8.03239631652832
    },
    {
      "epoch": 1.9301414489746094e-05,
      "model_forward_time": 1.7318096160888672,
      "step": 94239
    },
    {
      "epoch": 1.9301414489746094e-05,
      "step": 94239,
      "training_step_time": 7.9534080028533936
    },
    {
      "epoch": 1.93023681640625e-05,
      "grad_norm": 0.08617176860570908,
      "learning_rate": 9.046369697570056e-08,
      "loss": 0.0557,
      "step": 94240
    },
    {
      "epoch": 1.93023681640625e-05,
      "model_forward_time": 1.7405891418457031,
      "step": 94240
    },
    {
      "dynamics_loss_avg": 0.07375513687729836,
      "epoch": 1.93023681640625e-05,
      "step": 94240
    },
    {
      "action_loss_avg": 0.001329483211156912,
      "epoch": 1.93023681640625e-05,
      "step": 94240
    },
    {
      "epoch": 1.93023681640625e-05,
      "step": 94240,
      "training_step_time": 7.767421245574951
    },
    {
      "epoch": 1.9303321838378906e-05,
      "model_forward_time": 1.3719961643218994,
      "step": 94241
    },
    {
      "epoch": 1.9303321838378906e-05,
      "step": 94241,
      "training_step_time": 7.970170497894287
    },
    {
      "epoch": 1.9304275512695313e-05,
      "model_forward_time": 1.7429165840148926,
      "step": 94242
    },
    {
      "epoch": 1.9304275512695313e-05,
      "step": 94242,
      "training_step_time": 7.859869003295898
    },
    {
      "epoch": 1.9305229187011717e-05,
      "model_forward_time": 1.8132383823394775,
      "step": 94243
    },
    {
      "epoch": 1.9305229187011717e-05,
      "step": 94243,
      "training_step_time": 8.762970209121704
    },
    {
      "epoch": 1.9306182861328125e-05,
      "model_forward_time": 1.7359330654144287,
      "step": 94244
    },
    {
      "epoch": 1.9306182861328125e-05,
      "step": 94244,
      "training_step_time": 8.917059898376465
    },
    {
      "epoch": 1.9307136535644532e-05,
      "model_forward_time": 1.7378811836242676,
      "step": 94245
    },
    {
      "epoch": 1.9307136535644532e-05,
      "step": 94245,
      "training_step_time": 8.747390270233154
    },
    {
      "epoch": 1.9308090209960936e-05,
      "model_forward_time": 2.375060796737671,
      "step": 94246
    },
    {
      "epoch": 1.9308090209960936e-05,
      "step": 94246,
      "training_step_time": 8.552944660186768
    },
    {
      "epoch": 1.9309043884277344e-05,
      "model_forward_time": 1.7375664710998535,
      "step": 94247
    },
    {
      "epoch": 1.9309043884277344e-05,
      "step": 94247,
      "training_step_time": 8.039950609207153
    },
    {
      "epoch": 1.930999755859375e-05,
      "model_forward_time": 1.748533010482788,
      "step": 94248
    },
    {
      "epoch": 1.930999755859375e-05,
      "step": 94248,
      "training_step_time": 7.835506439208984
    },
    {
      "epoch": 1.9310951232910155e-05,
      "model_forward_time": 1.7329013347625732,
      "step": 94249
    },
    {
      "epoch": 1.9310951232910155e-05,
      "step": 94249,
      "training_step_time": 7.799229383468628
    },
    {
      "epoch": 1.9311904907226562e-05,
      "grad_norm": 0.09180108457803726,
      "learning_rate": 9.015086036123722e-08,
      "loss": 0.054,
      "step": 94250
    },
    {
      "epoch": 1.9311904907226562e-05,
      "model_forward_time": 1.7352797985076904,
      "step": 94250
    },
    {
      "dynamics_loss_avg": 0.04035856891423464,
      "epoch": 1.9311904907226562e-05,
      "step": 94250
    },
    {
      "action_loss_avg": 0.0007742686488199979,
      "epoch": 1.9311904907226562e-05,
      "step": 94250
    },
    {
      "epoch": 1.9311904907226562e-05,
      "step": 94250,
      "training_step_time": 8.80203914642334
    },
    {
      "epoch": 1.931285858154297e-05,
      "model_forward_time": 1.7350475788116455,
      "step": 94251
    },
    {
      "epoch": 1.931285858154297e-05,
      "step": 94251,
      "training_step_time": 7.5375001430511475
    },
    {
      "epoch": 1.9313812255859374e-05,
      "model_forward_time": 1.808751106262207,
      "step": 94252
    },
    {
      "epoch": 1.9313812255859374e-05,
      "step": 94252,
      "training_step_time": 8.873214721679688
    },
    {
      "epoch": 1.931476593017578e-05,
      "model_forward_time": 1.7415859699249268,
      "step": 94253
    },
    {
      "epoch": 1.931476593017578e-05,
      "step": 94253,
      "training_step_time": 8.571410655975342
    },
    {
      "epoch": 1.931571960449219e-05,
      "model_forward_time": 1.7378687858581543,
      "step": 94254
    },
    {
      "epoch": 1.931571960449219e-05,
      "step": 94254,
      "training_step_time": 9.717768669128418
    },
    {
      "epoch": 1.9316673278808593e-05,
      "model_forward_time": 2.3833324909210205,
      "step": 94255
    },
    {
      "epoch": 1.9316673278808593e-05,
      "step": 94255,
      "training_step_time": 9.638912916183472
    },
    {
      "epoch": 1.9317626953125e-05,
      "model_forward_time": 1.741811990737915,
      "step": 94256
    },
    {
      "epoch": 1.9317626953125e-05,
      "step": 94256,
      "training_step_time": 9.012691020965576
    },
    {
      "epoch": 1.9318580627441408e-05,
      "model_forward_time": 1.7499287128448486,
      "step": 94257
    },
    {
      "epoch": 1.9318580627441408e-05,
      "step": 94257,
      "training_step_time": 8.593509197235107
    },
    {
      "epoch": 1.931953430175781e-05,
      "model_forward_time": 1.7383105754852295,
      "step": 94258
    },
    {
      "epoch": 1.931953430175781e-05,
      "step": 94258,
      "training_step_time": 7.789939880371094
    },
    {
      "epoch": 1.932048797607422e-05,
      "model_forward_time": 1.7430024147033691,
      "step": 94259
    },
    {
      "epoch": 1.932048797607422e-05,
      "step": 94259,
      "training_step_time": 8.668893337249756
    },
    {
      "epoch": 1.9321441650390626e-05,
      "grad_norm": 0.10376814752817154,
      "learning_rate": 8.983856068049413e-08,
      "loss": 0.0538,
      "step": 94260
    },
    {
      "epoch": 1.9321441650390626e-05,
      "model_forward_time": 1.736734390258789,
      "step": 94260
    },
    {
      "dynamics_loss_avg": 0.05191781213507056,
      "epoch": 1.9321441650390626e-05,
      "step": 94260
    },
    {
      "action_loss_avg": 0.0010939724219497294,
      "epoch": 1.9321441650390626e-05,
      "step": 94260
    },
    {
      "epoch": 1.9321441650390626e-05,
      "step": 94260,
      "training_step_time": 7.406096696853638
    },
    {
      "epoch": 1.932239532470703e-05,
      "model_forward_time": 1.794600248336792,
      "step": 94261
    },
    {
      "epoch": 1.932239532470703e-05,
      "step": 94261,
      "training_step_time": 7.726278781890869
    },
    {
      "epoch": 1.9323348999023438e-05,
      "model_forward_time": 1.742009162902832,
      "step": 94262
    },
    {
      "epoch": 1.9323348999023438e-05,
      "step": 94262,
      "training_step_time": 8.497623205184937
    },
    {
      "epoch": 1.9324302673339845e-05,
      "model_forward_time": 1.7408170700073242,
      "step": 94263
    },
    {
      "epoch": 1.9324302673339845e-05,
      "step": 94263,
      "training_step_time": 10.61635422706604
    },
    {
      "epoch": 1.932525634765625e-05,
      "model_forward_time": 2.424877405166626,
      "step": 94264
    },
    {
      "epoch": 1.932525634765625e-05,
      "step": 94264,
      "training_step_time": 9.513510465621948
    },
    {
      "epoch": 1.9326210021972657e-05,
      "model_forward_time": 1.7394306659698486,
      "step": 94265
    },
    {
      "epoch": 1.9326210021972657e-05,
      "step": 94265,
      "training_step_time": 7.906290292739868
    },
    {
      "epoch": 1.932716369628906e-05,
      "model_forward_time": 1.7457756996154785,
      "step": 94266
    },
    {
      "epoch": 1.932716369628906e-05,
      "step": 94266,
      "training_step_time": 7.86348295211792
    },
    {
      "epoch": 1.9328117370605468e-05,
      "model_forward_time": 1.7468149662017822,
      "step": 94267
    },
    {
      "epoch": 1.9328117370605468e-05,
      "step": 94267,
      "training_step_time": 7.8658154010772705
    },
    {
      "epoch": 1.9329071044921876e-05,
      "model_forward_time": 1.7389049530029297,
      "step": 94268
    },
    {
      "epoch": 1.9329071044921876e-05,
      "step": 94268,
      "training_step_time": 7.981103420257568
    },
    {
      "epoch": 1.933002471923828e-05,
      "model_forward_time": 1.7432291507720947,
      "step": 94269
    },
    {
      "epoch": 1.933002471923828e-05,
      "step": 94269,
      "training_step_time": 7.469050884246826
    },
    {
      "epoch": 1.9330978393554687e-05,
      "grad_norm": 0.09864122420549393,
      "learning_rate": 8.952679796762398e-08,
      "loss": 0.0568,
      "step": 94270
    },
    {
      "epoch": 1.9330978393554687e-05,
      "model_forward_time": 1.7741749286651611,
      "step": 94270
    },
    {
      "dynamics_loss_avg": 0.040346785355359314,
      "epoch": 1.9330978393554687e-05,
      "step": 94270
    },
    {
      "action_loss_avg": 0.0007631365297129378,
      "epoch": 1.9330978393554687e-05,
      "step": 94270
    },
    {
      "epoch": 1.9330978393554687e-05,
      "step": 94270,
      "training_step_time": 7.512836217880249
    },
    {
      "epoch": 1.9331932067871095e-05,
      "model_forward_time": 1.7439136505126953,
      "step": 94271
    },
    {
      "epoch": 1.9331932067871095e-05,
      "step": 94271,
      "training_step_time": 8.805424213409424
    },
    {
      "epoch": 1.93328857421875e-05,
      "model_forward_time": 1.7345776557922363,
      "step": 94272
    },
    {
      "epoch": 1.93328857421875e-05,
      "step": 94272,
      "training_step_time": 8.852086305618286
    },
    {
      "epoch": 1.9333839416503906e-05,
      "model_forward_time": 2.3829283714294434,
      "step": 94273
    },
    {
      "epoch": 1.9333839416503906e-05,
      "step": 94273,
      "training_step_time": 9.500444412231445
    },
    {
      "epoch": 1.9334793090820313e-05,
      "model_forward_time": 1.7439451217651367,
      "step": 94274
    },
    {
      "epoch": 1.9334793090820313e-05,
      "step": 94274,
      "training_step_time": 7.92035436630249
    },
    {
      "epoch": 1.9335746765136717e-05,
      "model_forward_time": 1.7523086071014404,
      "step": 94275
    },
    {
      "epoch": 1.9335746765136717e-05,
      "step": 94275,
      "training_step_time": 7.8734705448150635
    },
    {
      "epoch": 1.9336700439453125e-05,
      "model_forward_time": 1.737175703048706,
      "step": 94276
    },
    {
      "epoch": 1.9336700439453125e-05,
      "step": 94276,
      "training_step_time": 7.792243957519531
    },
    {
      "epoch": 1.9337654113769532e-05,
      "model_forward_time": 1.7438700199127197,
      "step": 94277
    },
    {
      "epoch": 1.9337654113769532e-05,
      "step": 94277,
      "training_step_time": 7.365888357162476
    },
    {
      "epoch": 1.9338607788085936e-05,
      "model_forward_time": 1.746819019317627,
      "step": 94278
    },
    {
      "epoch": 1.9338607788085936e-05,
      "step": 94278,
      "training_step_time": 7.799469947814941
    },
    {
      "epoch": 1.9339561462402344e-05,
      "model_forward_time": 1.7487218379974365,
      "step": 94279
    },
    {
      "epoch": 1.9339561462402344e-05,
      "step": 94279,
      "training_step_time": 7.436055421829224
    },
    {
      "epoch": 1.934051513671875e-05,
      "grad_norm": 0.09732051938772202,
      "learning_rate": 8.92155722567195e-08,
      "loss": 0.0548,
      "step": 94280
    },
    {
      "epoch": 1.934051513671875e-05,
      "model_forward_time": 1.735642433166504,
      "step": 94280
    },
    {
      "dynamics_loss_avg": 0.03896585013717413,
      "epoch": 1.934051513671875e-05,
      "step": 94280
    },
    {
      "action_loss_avg": 0.001488562734448351,
      "epoch": 1.934051513671875e-05,
      "step": 94280
    },
    {
      "epoch": 1.934051513671875e-05,
      "step": 94280,
      "training_step_time": 7.771281480789185
    },
    {
      "epoch": 1.9341468811035155e-05,
      "model_forward_time": 1.7448770999908447,
      "step": 94281
    },
    {
      "epoch": 1.9341468811035155e-05,
      "step": 94281,
      "training_step_time": 8.56201434135437
    },
    {
      "epoch": 1.9342422485351563e-05,
      "model_forward_time": 1.7425329685211182,
      "step": 94282
    },
    {
      "epoch": 1.9342422485351563e-05,
      "step": 94282,
      "training_step_time": 9.25428318977356
    },
    {
      "epoch": 1.934337615966797e-05,
      "model_forward_time": 1.7407987117767334,
      "step": 94283
    },
    {
      "epoch": 1.934337615966797e-05,
      "step": 94283,
      "training_step_time": 10.145640850067139
    },
    {
      "epoch": 1.9344329833984374e-05,
      "model_forward_time": 1.7401974201202393,
      "step": 94284
    },
    {
      "epoch": 1.9344329833984374e-05,
      "step": 94284,
      "training_step_time": 8.591550350189209
    },
    {
      "epoch": 1.934528350830078e-05,
      "model_forward_time": 1.7427034378051758,
      "step": 94285
    },
    {
      "epoch": 1.934528350830078e-05,
      "step": 94285,
      "training_step_time": 7.843743085861206
    },
    {
      "epoch": 1.934623718261719e-05,
      "model_forward_time": 1.7503957748413086,
      "step": 94286
    },
    {
      "epoch": 1.934623718261719e-05,
      "step": 94286,
      "training_step_time": 8.631271123886108
    },
    {
      "epoch": 1.9347190856933593e-05,
      "model_forward_time": 1.7401299476623535,
      "step": 94287
    },
    {
      "epoch": 1.9347190856933593e-05,
      "step": 94287,
      "training_step_time": 8.500386714935303
    },
    {
      "epoch": 1.934814453125e-05,
      "model_forward_time": 1.7413654327392578,
      "step": 94288
    },
    {
      "epoch": 1.934814453125e-05,
      "step": 94288,
      "training_step_time": 8.160581588745117
    },
    {
      "epoch": 1.9349098205566408e-05,
      "model_forward_time": 1.817995309829712,
      "step": 94289
    },
    {
      "epoch": 1.9349098205566408e-05,
      "step": 94289,
      "training_step_time": 7.766268014907837
    },
    {
      "epoch": 1.935005187988281e-05,
      "grad_norm": 0.12110162526369095,
      "learning_rate": 8.890488358181792e-08,
      "loss": 0.0549,
      "step": 94290
    },
    {
      "epoch": 1.935005187988281e-05,
      "model_forward_time": 1.7358884811401367,
      "step": 94290
    },
    {
      "dynamics_loss_avg": 0.06070146095007658,
      "epoch": 1.935005187988281e-05,
      "step": 94290
    },
    {
      "action_loss_avg": 0.0011011899274308234,
      "epoch": 1.935005187988281e-05,
      "step": 94290
    },
    {
      "epoch": 1.935005187988281e-05,
      "step": 94290,
      "training_step_time": 7.693578004837036
    },
    {
      "epoch": 1.935100555419922e-05,
      "model_forward_time": 1.7415664196014404,
      "step": 94291
    },
    {
      "epoch": 1.935100555419922e-05,
      "step": 94291,
      "training_step_time": 8.456460952758789
    },
    {
      "epoch": 1.9351959228515627e-05,
      "model_forward_time": 2.361708879470825,
      "step": 94292
    },
    {
      "epoch": 1.9351959228515627e-05,
      "step": 94292,
      "training_step_time": 8.778615713119507
    },
    {
      "epoch": 1.935291290283203e-05,
      "model_forward_time": 1.7424952983856201,
      "step": 94293
    },
    {
      "epoch": 1.935291290283203e-05,
      "step": 94293,
      "training_step_time": 8.61339282989502
    },
    {
      "epoch": 1.9353866577148438e-05,
      "model_forward_time": 1.7446870803833008,
      "step": 94294
    },
    {
      "epoch": 1.9353866577148438e-05,
      "step": 94294,
      "training_step_time": 8.592343807220459
    },
    {
      "epoch": 1.9354820251464845e-05,
      "model_forward_time": 1.743028163909912,
      "step": 94295
    },
    {
      "epoch": 1.9354820251464845e-05,
      "step": 94295,
      "training_step_time": 7.7869555950164795
    },
    {
      "epoch": 1.935577392578125e-05,
      "model_forward_time": 1.7435426712036133,
      "step": 94296
    },
    {
      "epoch": 1.935577392578125e-05,
      "step": 94296,
      "training_step_time": 7.730579614639282
    },
    {
      "epoch": 1.9356727600097657e-05,
      "model_forward_time": 1.737715721130371,
      "step": 94297
    },
    {
      "epoch": 1.9356727600097657e-05,
      "step": 94297,
      "training_step_time": 7.395860433578491
    },
    {
      "epoch": 1.935768127441406e-05,
      "model_forward_time": 1.8035860061645508,
      "step": 94298
    },
    {
      "epoch": 1.935768127441406e-05,
      "step": 94298,
      "training_step_time": 7.770408391952515
    },
    {
      "epoch": 1.9358634948730468e-05,
      "model_forward_time": 1.739853858947754,
      "step": 94299
    },
    {
      "epoch": 1.9358634948730468e-05,
      "step": 94299,
      "training_step_time": 8.454920291900635
    },
    {
      "epoch": 1.9359588623046876e-05,
      "grad_norm": 0.08125614374876022,
      "learning_rate": 8.859473197689372e-08,
      "loss": 0.0562,
      "step": 94300
    },
    {
      "epoch": 1.9359588623046876e-05,
      "model_forward_time": 1.7400007247924805,
      "step": 94300
    },
    {
      "dynamics_loss_avg": 0.05905780252069235,
      "epoch": 1.9359588623046876e-05,
      "step": 94300
    },
    {
      "action_loss_avg": 0.0010501039563678206,
      "epoch": 1.9359588623046876e-05,
      "step": 94300
    },
    {
      "epoch": 1.9359588623046876e-05,
      "step": 94300,
      "training_step_time": 9.600805759429932
    },
    {
      "epoch": 1.936054229736328e-05,
      "model_forward_time": 2.3665409088134766,
      "step": 94301
    },
    {
      "epoch": 1.936054229736328e-05,
      "step": 94301,
      "training_step_time": 7.943569660186768
    },
    {
      "epoch": 1.9361495971679687e-05,
      "model_forward_time": 1.7507858276367188,
      "step": 94302
    },
    {
      "epoch": 1.9361495971679687e-05,
      "step": 94302,
      "training_step_time": 7.85856556892395
    },
    {
      "epoch": 1.9362449645996095e-05,
      "model_forward_time": 1.7396519184112549,
      "step": 94303
    },
    {
      "epoch": 1.9362449645996095e-05,
      "step": 94303,
      "training_step_time": 8.366759777069092
    },
    {
      "epoch": 1.93634033203125e-05,
      "model_forward_time": 1.7469391822814941,
      "step": 94304
    },
    {
      "epoch": 1.93634033203125e-05,
      "step": 94304,
      "training_step_time": 7.751535892486572
    },
    {
      "epoch": 1.9364356994628906e-05,
      "model_forward_time": 1.7418313026428223,
      "step": 94305
    },
    {
      "epoch": 1.9364356994628906e-05,
      "step": 94305,
      "training_step_time": 8.148996114730835
    },
    {
      "epoch": 1.9365310668945313e-05,
      "model_forward_time": 1.7386865615844727,
      "step": 94306
    },
    {
      "epoch": 1.9365310668945313e-05,
      "step": 94306,
      "training_step_time": 7.3512773513793945
    },
    {
      "epoch": 1.9366264343261717e-05,
      "model_forward_time": 1.8027544021606445,
      "step": 94307
    },
    {
      "epoch": 1.9366264343261717e-05,
      "step": 94307,
      "training_step_time": 8.76428771018982
    },
    {
      "epoch": 1.9367218017578125e-05,
      "model_forward_time": 1.7399547100067139,
      "step": 94308
    },
    {
      "epoch": 1.9367218017578125e-05,
      "step": 94308,
      "training_step_time": 7.7417683601379395
    },
    {
      "epoch": 1.9368171691894532e-05,
      "model_forward_time": 1.7243576049804688,
      "step": 94309
    },
    {
      "epoch": 1.9368171691894532e-05,
      "step": 94309,
      "training_step_time": 7.736640453338623
    },
    {
      "epoch": 1.9369125366210936e-05,
      "grad_norm": 0.06827647984027863,
      "learning_rate": 8.828511747586588e-08,
      "loss": 0.0555,
      "step": 94310
    },
    {
      "epoch": 1.9369125366210936e-05,
      "model_forward_time": 2.371643543243408,
      "step": 94310
    },
    {
      "dynamics_loss_avg": 0.06213632952421903,
      "epoch": 1.9369125366210936e-05,
      "step": 94310
    },
    {
      "action_loss_avg": 0.0011551628762390465,
      "epoch": 1.9369125366210936e-05,
      "step": 94310
    },
    {
      "epoch": 1.9369125366210936e-05,
      "step": 94310,
      "training_step_time": 7.904627799987793
    },
    {
      "epoch": 1.9370079040527344e-05,
      "model_forward_time": 1.7896451950073242,
      "step": 94311
    },
    {
      "epoch": 1.9370079040527344e-05,
      "step": 94311,
      "training_step_time": 9.197909116744995
    },
    {
      "epoch": 1.937103271484375e-05,
      "model_forward_time": 1.7276701927185059,
      "step": 94312
    },
    {
      "epoch": 1.937103271484375e-05,
      "step": 94312,
      "training_step_time": 7.836235761642456
    },
    {
      "epoch": 1.9371986389160155e-05,
      "model_forward_time": 1.7389826774597168,
      "step": 94313
    },
    {
      "epoch": 1.9371986389160155e-05,
      "step": 94313,
      "training_step_time": 7.878590106964111
    },
    {
      "epoch": 1.9372940063476563e-05,
      "model_forward_time": 1.7829339504241943,
      "step": 94314
    },
    {
      "epoch": 1.9372940063476563e-05,
      "step": 94314,
      "training_step_time": 7.3287880420684814
    },
    {
      "epoch": 1.937389373779297e-05,
      "model_forward_time": 1.7352888584136963,
      "step": 94315
    },
    {
      "epoch": 1.937389373779297e-05,
      "step": 94315,
      "training_step_time": 7.7960124015808105
    },
    {
      "epoch": 1.9374847412109374e-05,
      "model_forward_time": 1.733525276184082,
      "step": 94316
    },
    {
      "epoch": 1.9374847412109374e-05,
      "step": 94316,
      "training_step_time": 9.409362316131592
    },
    {
      "epoch": 1.937580108642578e-05,
      "model_forward_time": 1.740168571472168,
      "step": 94317
    },
    {
      "epoch": 1.937580108642578e-05,
      "step": 94317,
      "training_step_time": 9.232547044754028
    },
    {
      "epoch": 1.937675476074219e-05,
      "model_forward_time": 1.740368127822876,
      "step": 94318
    },
    {
      "epoch": 1.937675476074219e-05,
      "step": 94318,
      "training_step_time": 7.8227317333221436
    },
    {
      "epoch": 1.9377708435058593e-05,
      "model_forward_time": 2.4305830001831055,
      "step": 94319
    },
    {
      "epoch": 1.9377708435058593e-05,
      "step": 94319,
      "training_step_time": 8.411832809448242
    },
    {
      "epoch": 1.9378662109375e-05,
      "grad_norm": 0.0840953066945076,
      "learning_rate": 8.797604011259287e-08,
      "loss": 0.0563,
      "step": 94320
    },
    {
      "epoch": 1.9378662109375e-05,
      "model_forward_time": 1.7365765571594238,
      "step": 94320
    },
    {
      "dynamics_loss_avg": 0.055831399746239185,
      "epoch": 1.9378662109375e-05,
      "step": 94320
    },
    {
      "action_loss_avg": 0.0014740486338268965,
      "epoch": 1.9378662109375e-05,
      "step": 94320
    },
    {
      "epoch": 1.9378662109375e-05,
      "step": 94320,
      "training_step_time": 8.792918920516968
    },
    {
      "epoch": 1.9379615783691408e-05,
      "model_forward_time": 1.7432270050048828,
      "step": 94321
    },
    {
      "epoch": 1.9379615783691408e-05,
      "step": 94321,
      "training_step_time": 7.790684461593628
    },
    {
      "epoch": 1.9380569458007812e-05,
      "model_forward_time": 1.7419872283935547,
      "step": 94322
    },
    {
      "epoch": 1.9380569458007812e-05,
      "step": 94322,
      "training_step_time": 7.749434232711792
    },
    {
      "epoch": 1.938152313232422e-05,
      "model_forward_time": 1.7346298694610596,
      "step": 94323
    },
    {
      "epoch": 1.938152313232422e-05,
      "step": 94323,
      "training_step_time": 7.792489528656006
    },
    {
      "epoch": 1.9382476806640627e-05,
      "model_forward_time": 1.7373135089874268,
      "step": 94324
    },
    {
      "epoch": 1.9382476806640627e-05,
      "step": 94324,
      "training_step_time": 7.697681188583374
    },
    {
      "epoch": 1.938343048095703e-05,
      "model_forward_time": 1.752605676651001,
      "step": 94325
    },
    {
      "epoch": 1.938343048095703e-05,
      "step": 94325,
      "training_step_time": 7.455043077468872
    },
    {
      "epoch": 1.9384384155273438e-05,
      "model_forward_time": 1.8079848289489746,
      "step": 94326
    },
    {
      "epoch": 1.9384384155273438e-05,
      "step": 94326,
      "training_step_time": 7.852067470550537
    },
    {
      "epoch": 1.9385337829589845e-05,
      "model_forward_time": 1.7400507926940918,
      "step": 94327
    },
    {
      "epoch": 1.9385337829589845e-05,
      "step": 94327,
      "training_step_time": 7.784840106964111
    },
    {
      "epoch": 1.938629150390625e-05,
      "model_forward_time": 2.4003496170043945,
      "step": 94328
    },
    {
      "epoch": 1.938629150390625e-05,
      "step": 94328,
      "training_step_time": 8.766151905059814
    },
    {
      "epoch": 1.9387245178222657e-05,
      "model_forward_time": 1.7409896850585938,
      "step": 94329
    },
    {
      "epoch": 1.9387245178222657e-05,
      "step": 94329,
      "training_step_time": 10.15844202041626
    },
    {
      "epoch": 1.938819885253906e-05,
      "grad_norm": 0.09026075899600983,
      "learning_rate": 8.766749992087487e-08,
      "loss": 0.057,
      "step": 94330
    },
    {
      "epoch": 1.938819885253906e-05,
      "model_forward_time": 1.7388172149658203,
      "step": 94330
    },
    {
      "dynamics_loss_avg": 0.06554902102798224,
      "epoch": 1.938819885253906e-05,
      "step": 94330
    },
    {
      "action_loss_avg": 0.0009162713948171585,
      "epoch": 1.938819885253906e-05,
      "step": 94330
    },
    {
      "epoch": 1.938819885253906e-05,
      "step": 94330,
      "training_step_time": 7.71743369102478
    },
    {
      "epoch": 1.9389152526855468e-05,
      "model_forward_time": 1.7834067344665527,
      "step": 94331
    },
    {
      "epoch": 1.9389152526855468e-05,
      "step": 94331,
      "training_step_time": 7.830640077590942
    },
    {
      "epoch": 1.9390106201171876e-05,
      "model_forward_time": 1.7368743419647217,
      "step": 94332
    },
    {
      "epoch": 1.9390106201171876e-05,
      "step": 94332,
      "training_step_time": 7.818613052368164
    },
    {
      "epoch": 1.939105987548828e-05,
      "model_forward_time": 1.7379841804504395,
      "step": 94333
    },
    {
      "epoch": 1.939105987548828e-05,
      "step": 94333,
      "training_step_time": 7.873031854629517
    },
    {
      "epoch": 1.9392013549804687e-05,
      "model_forward_time": 1.7431142330169678,
      "step": 94334
    },
    {
      "epoch": 1.9392013549804687e-05,
      "step": 94334,
      "training_step_time": 7.437805891036987
    },
    {
      "epoch": 1.9392967224121095e-05,
      "model_forward_time": 1.8054423332214355,
      "step": 94335
    },
    {
      "epoch": 1.9392967224121095e-05,
      "step": 94335,
      "training_step_time": 7.747042655944824
    },
    {
      "epoch": 1.93939208984375e-05,
      "model_forward_time": 1.7501630783081055,
      "step": 94336
    },
    {
      "epoch": 1.93939208984375e-05,
      "step": 94336,
      "training_step_time": 7.7431886196136475
    },
    {
      "epoch": 1.9394874572753906e-05,
      "model_forward_time": 1.3745007514953613,
      "step": 94337
    },
    {
      "epoch": 1.9394874572753906e-05,
      "step": 94337,
      "training_step_time": 12.263347387313843
    },
    {
      "epoch": 1.9395828247070313e-05,
      "model_forward_time": 2.399425983428955,
      "step": 94338
    },
    {
      "epoch": 1.9395828247070313e-05,
      "step": 94338,
      "training_step_time": 7.886914014816284
    },
    {
      "epoch": 1.9396781921386717e-05,
      "model_forward_time": 1.7429652214050293,
      "step": 94339
    },
    {
      "epoch": 1.9396781921386717e-05,
      "step": 94339,
      "training_step_time": 9.895074605941772
    },
    {
      "epoch": 1.9397735595703125e-05,
      "grad_norm": 0.10853604227304459,
      "learning_rate": 8.735949693445378e-08,
      "loss": 0.0551,
      "step": 94340
    },
    {
      "epoch": 1.9397735595703125e-05,
      "model_forward_time": 1.7419240474700928,
      "step": 94340
    },
    {
      "dynamics_loss_avg": 0.04175150394439697,
      "epoch": 1.9397735595703125e-05,
      "step": 94340
    },
    {
      "action_loss_avg": 0.0007162215682910755,
      "epoch": 1.9397735595703125e-05,
      "step": 94340
    },
    {
      "epoch": 1.9397735595703125e-05,
      "step": 94340,
      "training_step_time": 7.804305076599121
    },
    {
      "epoch": 1.9398689270019532e-05,
      "model_forward_time": 1.7619855403900146,
      "step": 94341
    },
    {
      "epoch": 1.9398689270019532e-05,
      "step": 94341,
      "training_step_time": 7.764935255050659
    },
    {
      "epoch": 1.9399642944335936e-05,
      "model_forward_time": 1.7343342304229736,
      "step": 94342
    },
    {
      "epoch": 1.9399642944335936e-05,
      "step": 94342,
      "training_step_time": 7.680829763412476
    },
    {
      "epoch": 1.9400596618652344e-05,
      "model_forward_time": 1.7241082191467285,
      "step": 94343
    },
    {
      "epoch": 1.9400596618652344e-05,
      "step": 94343,
      "training_step_time": 7.351206064224243
    },
    {
      "epoch": 1.940155029296875e-05,
      "model_forward_time": 1.8018581867218018,
      "step": 94344
    },
    {
      "epoch": 1.940155029296875e-05,
      "step": 94344,
      "training_step_time": 7.730905532836914
    },
    {
      "epoch": 1.9402503967285155e-05,
      "model_forward_time": 1.8087236881256104,
      "step": 94345
    },
    {
      "epoch": 1.9402503967285155e-05,
      "step": 94345,
      "training_step_time": 7.764047861099243
    },
    {
      "epoch": 1.9403457641601563e-05,
      "model_forward_time": 1.7389163970947266,
      "step": 94346
    },
    {
      "epoch": 1.9403457641601563e-05,
      "step": 94346,
      "training_step_time": 8.772483825683594
    },
    {
      "epoch": 1.940441131591797e-05,
      "model_forward_time": 1.991426706314087,
      "step": 94347
    },
    {
      "epoch": 1.940441131591797e-05,
      "step": 94347,
      "training_step_time": 8.749966621398926
    },
    {
      "epoch": 1.9405364990234374e-05,
      "model_forward_time": 1.7318744659423828,
      "step": 94348
    },
    {
      "epoch": 1.9405364990234374e-05,
      "step": 94348,
      "training_step_time": 7.818689823150635
    },
    {
      "epoch": 1.940631866455078e-05,
      "model_forward_time": 1.7329869270324707,
      "step": 94349
    },
    {
      "epoch": 1.940631866455078e-05,
      "step": 94349,
      "training_step_time": 7.829648733139038
    },
    {
      "epoch": 1.940727233886719e-05,
      "grad_norm": 0.08018562942743301,
      "learning_rate": 8.705203118701211e-08,
      "loss": 0.0583,
      "step": 94350
    },
    {
      "epoch": 1.940727233886719e-05,
      "model_forward_time": 1.7338261604309082,
      "step": 94350
    },
    {
      "dynamics_loss_avg": 0.05366086661815643,
      "epoch": 1.940727233886719e-05,
      "step": 94350
    },
    {
      "action_loss_avg": 0.0006801688927225768,
      "epoch": 1.940727233886719e-05,
      "step": 94350
    },
    {
      "epoch": 1.940727233886719e-05,
      "step": 94350,
      "training_step_time": 7.730485439300537
    },
    {
      "epoch": 1.9408226013183593e-05,
      "model_forward_time": 1.7343320846557617,
      "step": 94351
    },
    {
      "epoch": 1.9408226013183593e-05,
      "step": 94351,
      "training_step_time": 7.535237789154053
    },
    {
      "epoch": 1.94091796875e-05,
      "model_forward_time": 1.738124132156372,
      "step": 94352
    },
    {
      "epoch": 1.94091796875e-05,
      "step": 94352,
      "training_step_time": 7.706500291824341
    },
    {
      "epoch": 1.9410133361816408e-05,
      "model_forward_time": 1.8013427257537842,
      "step": 94353
    },
    {
      "epoch": 1.9410133361816408e-05,
      "step": 94353,
      "training_step_time": 8.746110439300537
    },
    {
      "epoch": 1.9411087036132812e-05,
      "model_forward_time": 1.7416713237762451,
      "step": 94354
    },
    {
      "epoch": 1.9411087036132812e-05,
      "step": 94354,
      "training_step_time": 9.645491123199463
    },
    {
      "epoch": 1.941204071044922e-05,
      "model_forward_time": 1.750199317932129,
      "step": 94355
    },
    {
      "epoch": 1.941204071044922e-05,
      "step": 94355,
      "training_step_time": 8.48590087890625
    },
    {
      "epoch": 1.9412994384765627e-05,
      "model_forward_time": 2.402292013168335,
      "step": 94356
    },
    {
      "epoch": 1.9412994384765627e-05,
      "step": 94356,
      "training_step_time": 7.728675603866577
    },
    {
      "epoch": 1.941394805908203e-05,
      "model_forward_time": 1.8378732204437256,
      "step": 94357
    },
    {
      "epoch": 1.941394805908203e-05,
      "step": 94357,
      "training_step_time": 8.669563055038452
    },
    {
      "epoch": 1.9414901733398438e-05,
      "model_forward_time": 1.4067480564117432,
      "step": 94358
    },
    {
      "epoch": 1.9414901733398438e-05,
      "step": 94358,
      "training_step_time": 7.777706146240234
    },
    {
      "epoch": 1.9415855407714845e-05,
      "model_forward_time": 1.759958267211914,
      "step": 94359
    },
    {
      "epoch": 1.9415855407714845e-05,
      "step": 94359,
      "training_step_time": 8.275876522064209
    },
    {
      "epoch": 1.941680908203125e-05,
      "grad_norm": 0.11861263960599899,
      "learning_rate": 8.674510271217295e-08,
      "loss": 0.0572,
      "step": 94360
    },
    {
      "epoch": 1.941680908203125e-05,
      "model_forward_time": 1.7584152221679688,
      "step": 94360
    },
    {
      "dynamics_loss_avg": 0.0369692713022232,
      "epoch": 1.941680908203125e-05,
      "step": 94360
    },
    {
      "action_loss_avg": 0.0007677750661969184,
      "epoch": 1.941680908203125e-05,
      "step": 94360
    },
    {
      "epoch": 1.941680908203125e-05,
      "step": 94360,
      "training_step_time": 7.476882219314575
    },
    {
      "epoch": 1.9417762756347657e-05,
      "model_forward_time": 1.7659893035888672,
      "step": 94361
    },
    {
      "epoch": 1.9417762756347657e-05,
      "step": 94361,
      "training_step_time": 7.868514060974121
    },
    {
      "epoch": 1.941871643066406e-05,
      "model_forward_time": 1.7602629661560059,
      "step": 94362
    },
    {
      "epoch": 1.941871643066406e-05,
      "step": 94362,
      "training_step_time": 7.653756618499756
    },
    {
      "epoch": 1.941967010498047e-05,
      "model_forward_time": 1.826181173324585,
      "step": 94363
    },
    {
      "epoch": 1.941967010498047e-05,
      "step": 94363,
      "training_step_time": 7.81601357460022
    },
    {
      "epoch": 1.9420623779296876e-05,
      "model_forward_time": 1.7584681510925293,
      "step": 94364
    },
    {
      "epoch": 1.9420623779296876e-05,
      "step": 94364,
      "training_step_time": 7.86844277381897
    },
    {
      "epoch": 1.942157745361328e-05,
      "model_forward_time": 2.371558666229248,
      "step": 94365
    },
    {
      "epoch": 1.942157745361328e-05,
      "step": 94365,
      "training_step_time": 10.264941215515137
    },
    {
      "epoch": 1.9422531127929687e-05,
      "model_forward_time": 1.3995933532714844,
      "step": 94366
    },
    {
      "epoch": 1.9422531127929687e-05,
      "step": 94366,
      "training_step_time": 8.832542181015015
    },
    {
      "epoch": 1.9423484802246095e-05,
      "model_forward_time": 1.761878252029419,
      "step": 94367
    },
    {
      "epoch": 1.9423484802246095e-05,
      "step": 94367,
      "training_step_time": 8.649470567703247
    },
    {
      "epoch": 1.94244384765625e-05,
      "model_forward_time": 1.7495105266571045,
      "step": 94368
    },
    {
      "epoch": 1.94244384765625e-05,
      "step": 94368,
      "training_step_time": 7.793671131134033
    },
    {
      "epoch": 1.9425392150878906e-05,
      "model_forward_time": 1.7698109149932861,
      "step": 94369
    },
    {
      "epoch": 1.9425392150878906e-05,
      "step": 94369,
      "training_step_time": 7.825890064239502
    },
    {
      "epoch": 1.9426345825195313e-05,
      "grad_norm": 0.08664564788341522,
      "learning_rate": 8.643871154350336e-08,
      "loss": 0.0583,
      "step": 94370
    },
    {
      "epoch": 1.9426345825195313e-05,
      "model_forward_time": 1.7621750831604004,
      "step": 94370
    },
    {
      "dynamics_loss_avg": 0.05741772335022688,
      "epoch": 1.9426345825195313e-05,
      "step": 94370
    },
    {
      "action_loss_avg": 0.001076831590034999,
      "epoch": 1.9426345825195313e-05,
      "step": 94370
    },
    {
      "epoch": 1.9426345825195313e-05,
      "step": 94370,
      "training_step_time": 7.574530601501465
    },
    {
      "epoch": 1.9427299499511717e-05,
      "model_forward_time": 1.7608990669250488,
      "step": 94371
    },
    {
      "epoch": 1.9427299499511717e-05,
      "step": 94371,
      "training_step_time": 8.460835218429565
    },
    {
      "epoch": 1.9428253173828125e-05,
      "model_forward_time": 1.8274390697479248,
      "step": 94372
    },
    {
      "epoch": 1.9428253173828125e-05,
      "step": 94372,
      "training_step_time": 7.722567319869995
    },
    {
      "epoch": 1.9429206848144532e-05,
      "model_forward_time": 1.7604217529296875,
      "step": 94373
    },
    {
      "epoch": 1.9429206848144532e-05,
      "step": 94373,
      "training_step_time": 8.744837045669556
    },
    {
      "epoch": 1.9430160522460936e-05,
      "model_forward_time": 1.7593448162078857,
      "step": 94374
    },
    {
      "epoch": 1.9430160522460936e-05,
      "step": 94374,
      "training_step_time": 7.963494539260864
    },
    {
      "epoch": 1.9431114196777344e-05,
      "model_forward_time": 2.4171464443206787,
      "step": 94375
    },
    {
      "epoch": 1.9431114196777344e-05,
      "step": 94375,
      "training_step_time": 9.737254858016968
    },
    {
      "epoch": 1.943206787109375e-05,
      "model_forward_time": 1.7593622207641602,
      "step": 94376
    },
    {
      "epoch": 1.943206787109375e-05,
      "step": 94376,
      "training_step_time": 8.763969421386719
    },
    {
      "epoch": 1.9433021545410155e-05,
      "model_forward_time": 1.7521085739135742,
      "step": 94377
    },
    {
      "epoch": 1.9433021545410155e-05,
      "step": 94377,
      "training_step_time": 8.472658634185791
    },
    {
      "epoch": 1.9433975219726563e-05,
      "model_forward_time": 1.755591869354248,
      "step": 94378
    },
    {
      "epoch": 1.9433975219726563e-05,
      "step": 94378,
      "training_step_time": 7.802356004714966
    },
    {
      "epoch": 1.943492889404297e-05,
      "model_forward_time": 1.7595314979553223,
      "step": 94379
    },
    {
      "epoch": 1.943492889404297e-05,
      "step": 94379,
      "training_step_time": 7.609631776809692
    },
    {
      "epoch": 1.9435882568359374e-05,
      "grad_norm": 0.13072673976421356,
      "learning_rate": 8.613285771450819e-08,
      "loss": 0.0574,
      "step": 94380
    },
    {
      "epoch": 1.9435882568359374e-05,
      "model_forward_time": 1.7589025497436523,
      "step": 94380
    },
    {
      "dynamics_loss_avg": 0.051022802107036115,
      "epoch": 1.9435882568359374e-05,
      "step": 94380
    },
    {
      "action_loss_avg": 0.001190028370183427,
      "epoch": 1.9435882568359374e-05,
      "step": 94380
    },
    {
      "epoch": 1.9435882568359374e-05,
      "step": 94380,
      "training_step_time": 7.472391843795776
    },
    {
      "epoch": 1.943683624267578e-05,
      "model_forward_time": 1.8295354843139648,
      "step": 94381
    },
    {
      "epoch": 1.943683624267578e-05,
      "step": 94381,
      "training_step_time": 7.787086725234985
    },
    {
      "epoch": 1.943778991699219e-05,
      "model_forward_time": 1.398866891860962,
      "step": 94382
    },
    {
      "epoch": 1.943778991699219e-05,
      "step": 94382,
      "training_step_time": 8.744878768920898
    },
    {
      "epoch": 1.9438743591308593e-05,
      "model_forward_time": 1.763211965560913,
      "step": 94383
    },
    {
      "epoch": 1.9438743591308593e-05,
      "step": 94383,
      "training_step_time": 9.918824911117554
    },
    {
      "epoch": 1.9439697265625e-05,
      "model_forward_time": 2.447108507156372,
      "step": 94384
    },
    {
      "epoch": 1.9439697265625e-05,
      "step": 94384,
      "training_step_time": 8.533729076385498
    },
    {
      "epoch": 1.9440650939941408e-05,
      "model_forward_time": 1.9131371974945068,
      "step": 94385
    },
    {
      "epoch": 1.9440650939941408e-05,
      "step": 94385,
      "training_step_time": 7.8488523960113525
    },
    {
      "epoch": 1.9441604614257812e-05,
      "model_forward_time": 1.915804386138916,
      "step": 94386
    },
    {
      "epoch": 1.9441604614257812e-05,
      "step": 94386,
      "training_step_time": 8.954151630401611
    },
    {
      "epoch": 1.944255828857422e-05,
      "model_forward_time": 1.925851583480835,
      "step": 94387
    },
    {
      "epoch": 1.944255828857422e-05,
      "step": 94387,
      "training_step_time": 8.848567724227905
    },
    {
      "epoch": 1.9443511962890627e-05,
      "model_forward_time": 1.9188292026519775,
      "step": 94388
    },
    {
      "epoch": 1.9443511962890627e-05,
      "step": 94388,
      "training_step_time": 8.249877214431763
    },
    {
      "epoch": 1.944446563720703e-05,
      "model_forward_time": 1.9193885326385498,
      "step": 94389
    },
    {
      "epoch": 1.944446563720703e-05,
      "step": 94389,
      "training_step_time": 7.626183032989502
    },
    {
      "epoch": 1.9445419311523438e-05,
      "grad_norm": 0.07208285480737686,
      "learning_rate": 8.582754125863624e-08,
      "loss": 0.0545,
      "step": 94390
    },
    {
      "epoch": 1.9445419311523438e-05,
      "model_forward_time": 1.9088704586029053,
      "step": 94390
    },
    {
      "dynamics_loss_avg": 0.04813446402549744,
      "epoch": 1.9445419311523438e-05,
      "step": 94390
    },
    {
      "action_loss_avg": 0.0015876699966611342,
      "epoch": 1.9445419311523438e-05,
      "step": 94390
    },
    {
      "epoch": 1.9445419311523438e-05,
      "step": 94390,
      "training_step_time": 7.832057476043701
    },
    {
      "epoch": 1.9446372985839842e-05,
      "model_forward_time": 1.9171621799468994,
      "step": 94391
    },
    {
      "epoch": 1.9446372985839842e-05,
      "step": 94391,
      "training_step_time": 8.388201475143433
    },
    {
      "epoch": 1.944732666015625e-05,
      "model_forward_time": 1.9085063934326172,
      "step": 94392
    },
    {
      "epoch": 1.944732666015625e-05,
      "step": 94392,
      "training_step_time": 8.692233800888062
    },
    {
      "epoch": 1.9448280334472657e-05,
      "model_forward_time": 2.55123233795166,
      "step": 94393
    },
    {
      "epoch": 1.9448280334472657e-05,
      "step": 94393,
      "training_step_time": 7.8658530712127686
    },
    {
      "epoch": 1.944923400878906e-05,
      "model_forward_time": 1.918473720550537,
      "step": 94394
    },
    {
      "epoch": 1.944923400878906e-05,
      "step": 94394,
      "training_step_time": 7.88341760635376
    },
    {
      "epoch": 1.945018768310547e-05,
      "model_forward_time": 1.908463716506958,
      "step": 94395
    },
    {
      "epoch": 1.945018768310547e-05,
      "step": 94395,
      "training_step_time": 8.788622856140137
    },
    {
      "epoch": 1.9451141357421876e-05,
      "model_forward_time": 1.9037067890167236,
      "step": 94396
    },
    {
      "epoch": 1.9451141357421876e-05,
      "step": 94396,
      "training_step_time": 7.824982166290283
    },
    {
      "epoch": 1.945209503173828e-05,
      "model_forward_time": 1.9030015468597412,
      "step": 94397
    },
    {
      "epoch": 1.945209503173828e-05,
      "step": 94397,
      "training_step_time": 7.804526090621948
    },
    {
      "epoch": 1.9453048706054687e-05,
      "model_forward_time": 1.9110918045043945,
      "step": 94398
    },
    {
      "epoch": 1.9453048706054687e-05,
      "step": 94398,
      "training_step_time": 7.610283374786377
    },
    {
      "epoch": 1.9454002380371095e-05,
      "model_forward_time": 1.910179615020752,
      "step": 94399
    },
    {
      "epoch": 1.9454002380371095e-05,
      "step": 94399,
      "training_step_time": 7.837048053741455
    },
    {
      "epoch": 1.94549560546875e-05,
      "grad_norm": 0.07300978899002075,
      "learning_rate": 8.55227622092758e-08,
      "loss": 0.0581,
      "step": 94400
    },
    {
      "epoch": 1.94549560546875e-05,
      "model_forward_time": 1.9125571250915527,
      "step": 94400
    },
    {
      "dynamics_loss_avg": 0.04201588034629822,
      "epoch": 1.94549560546875e-05,
      "step": 94400
    },
    {
      "action_loss_avg": 0.0009428918128833175,
      "epoch": 1.94549560546875e-05,
      "step": 94400
    },
    {
      "epoch": 1.94549560546875e-05,
      "step": 94400,
      "training_step_time": 8.709626913070679
    },
    {
      "epoch": 1.9455909729003906e-05,
      "model_forward_time": 1.913726806640625,
      "step": 94401
    },
    {
      "epoch": 1.9455909729003906e-05,
      "step": 94401,
      "training_step_time": 7.856385707855225
    },
    {
      "epoch": 1.9456863403320313e-05,
      "model_forward_time": 1.9040114879608154,
      "step": 94402
    },
    {
      "epoch": 1.9456863403320313e-05,
      "step": 94402,
      "training_step_time": 8.483352422714233
    },
    {
      "epoch": 1.9457817077636718e-05,
      "model_forward_time": 1.9049725532531738,
      "step": 94403
    },
    {
      "epoch": 1.9457817077636718e-05,
      "step": 94403,
      "training_step_time": 7.879997968673706
    },
    {
      "epoch": 1.9458770751953125e-05,
      "model_forward_time": 1.922823429107666,
      "step": 94404
    },
    {
      "epoch": 1.9458770751953125e-05,
      "step": 94404,
      "training_step_time": 7.904197692871094
    },
    {
      "epoch": 1.9459724426269532e-05,
      "model_forward_time": 1.9110567569732666,
      "step": 94405
    },
    {
      "epoch": 1.9459724426269532e-05,
      "step": 94405,
      "training_step_time": 8.553104162216187
    },
    {
      "epoch": 1.9460678100585936e-05,
      "model_forward_time": 1.905693531036377,
      "step": 94406
    },
    {
      "epoch": 1.9460678100585936e-05,
      "step": 94406,
      "training_step_time": 7.815234184265137
    },
    {
      "epoch": 1.9461631774902344e-05,
      "model_forward_time": 1.888242483139038,
      "step": 94407
    },
    {
      "epoch": 1.9461631774902344e-05,
      "step": 94407,
      "training_step_time": 7.5717573165893555
    },
    {
      "epoch": 1.946258544921875e-05,
      "model_forward_time": 1.9015731811523438,
      "step": 94408
    },
    {
      "epoch": 1.946258544921875e-05,
      "step": 94408,
      "training_step_time": 7.823199033737183
    },
    {
      "epoch": 1.9463539123535155e-05,
      "model_forward_time": 1.9290854930877686,
      "step": 94409
    },
    {
      "epoch": 1.9463539123535155e-05,
      "step": 94409,
      "training_step_time": 7.832647800445557
    },
    {
      "epoch": 1.9464492797851563e-05,
      "grad_norm": 0.08563121408224106,
      "learning_rate": 8.521852059975744e-08,
      "loss": 0.0578,
      "step": 94410
    },
    {
      "epoch": 1.9464492797851563e-05,
      "model_forward_time": 1.9054539203643799,
      "step": 94410
    },
    {
      "dynamics_loss_avg": 0.06525422763079405,
      "epoch": 1.9464492797851563e-05,
      "step": 94410
    },
    {
      "action_loss_avg": 0.0015279150480637326,
      "epoch": 1.9464492797851563e-05,
      "step": 94410
    },
    {
      "epoch": 1.9464492797851563e-05,
      "step": 94410,
      "training_step_time": 7.782520532608032
    },
    {
      "epoch": 1.946544647216797e-05,
      "model_forward_time": 1.9114866256713867,
      "step": 94411
    },
    {
      "epoch": 1.946544647216797e-05,
      "step": 94411,
      "training_step_time": 9.213194370269775
    },
    {
      "epoch": 1.9466400146484374e-05,
      "model_forward_time": 2.5061428546905518,
      "step": 94412
    },
    {
      "epoch": 1.9466400146484374e-05,
      "step": 94412,
      "training_step_time": 8.890730142593384
    },
    {
      "epoch": 1.946735382080078e-05,
      "model_forward_time": 1.8940925598144531,
      "step": 94413
    },
    {
      "epoch": 1.946735382080078e-05,
      "step": 94413,
      "training_step_time": 7.967966556549072
    },
    {
      "epoch": 1.946830749511719e-05,
      "model_forward_time": 1.8946051597595215,
      "step": 94414
    },
    {
      "epoch": 1.946830749511719e-05,
      "step": 94414,
      "training_step_time": 8.701577425003052
    },
    {
      "epoch": 1.9469261169433593e-05,
      "model_forward_time": 1.8818304538726807,
      "step": 94415
    },
    {
      "epoch": 1.9469261169433593e-05,
      "step": 94415,
      "training_step_time": 7.778948545455933
    },
    {
      "epoch": 1.947021484375e-05,
      "model_forward_time": 1.9160616397857666,
      "step": 94416
    },
    {
      "epoch": 1.947021484375e-05,
      "step": 94416,
      "training_step_time": 7.6418068408966064
    },
    {
      "epoch": 1.9471168518066408e-05,
      "model_forward_time": 1.9063441753387451,
      "step": 94417
    },
    {
      "epoch": 1.9471168518066408e-05,
      "step": 94417,
      "training_step_time": 7.876632213592529
    },
    {
      "epoch": 1.9472122192382812e-05,
      "model_forward_time": 1.9187889099121094,
      "step": 94418
    },
    {
      "epoch": 1.9472122192382812e-05,
      "step": 94418,
      "training_step_time": 8.897504091262817
    },
    {
      "epoch": 1.947307586669922e-05,
      "model_forward_time": 1.9049324989318848,
      "step": 94419
    },
    {
      "epoch": 1.947307586669922e-05,
      "step": 94419,
      "training_step_time": 8.753209352493286
    },
    {
      "epoch": 1.9474029541015627e-05,
      "grad_norm": 0.23676042258739471,
      "learning_rate": 8.491481646335231e-08,
      "loss": 0.055,
      "step": 94420
    },
    {
      "epoch": 1.9474029541015627e-05,
      "model_forward_time": 1.9108152389526367,
      "step": 94420
    },
    {
      "dynamics_loss_avg": 0.06351563464850188,
      "epoch": 1.9474029541015627e-05,
      "step": 94420
    },
    {
      "action_loss_avg": 0.0010585869080387057,
      "epoch": 1.9474029541015627e-05,
      "step": 94420
    },
    {
      "epoch": 1.9474029541015627e-05,
      "step": 94420,
      "training_step_time": 9.38124418258667
    },
    {
      "epoch": 1.947498321533203e-05,
      "model_forward_time": 2.534728765487671,
      "step": 94421
    },
    {
      "epoch": 1.947498321533203e-05,
      "step": 94421,
      "training_step_time": 7.934332609176636
    },
    {
      "epoch": 1.9475936889648438e-05,
      "model_forward_time": 1.9179198741912842,
      "step": 94422
    },
    {
      "epoch": 1.9475936889648438e-05,
      "step": 94422,
      "training_step_time": 9.574987888336182
    },
    {
      "epoch": 1.9476890563964842e-05,
      "model_forward_time": 1.865722417831421,
      "step": 94423
    },
    {
      "epoch": 1.9476890563964842e-05,
      "step": 94423,
      "training_step_time": 7.902597427368164
    },
    {
      "epoch": 1.947784423828125e-05,
      "model_forward_time": 1.8824098110198975,
      "step": 94424
    },
    {
      "epoch": 1.947784423828125e-05,
      "step": 94424,
      "training_step_time": 7.82334303855896
    },
    {
      "epoch": 1.9478797912597657e-05,
      "model_forward_time": 1.895580530166626,
      "step": 94425
    },
    {
      "epoch": 1.9478797912597657e-05,
      "step": 94425,
      "training_step_time": 7.709686756134033
    },
    {
      "epoch": 1.947975158691406e-05,
      "model_forward_time": 1.903395175933838,
      "step": 94426
    },
    {
      "epoch": 1.947975158691406e-05,
      "step": 94426,
      "training_step_time": 8.881701707839966
    },
    {
      "epoch": 1.948070526123047e-05,
      "model_forward_time": 1.8898992538452148,
      "step": 94427
    },
    {
      "epoch": 1.948070526123047e-05,
      "step": 94427,
      "training_step_time": 7.896664142608643
    },
    {
      "epoch": 1.9481658935546876e-05,
      "model_forward_time": 1.8932123184204102,
      "step": 94428
    },
    {
      "epoch": 1.9481658935546876e-05,
      "step": 94428,
      "training_step_time": 7.814142942428589
    },
    {
      "epoch": 1.948261260986328e-05,
      "model_forward_time": 1.152395486831665,
      "step": 94429
    },
    {
      "epoch": 1.948261260986328e-05,
      "step": 94429,
      "training_step_time": 7.940107107162476
    },
    {
      "epoch": 1.9483566284179687e-05,
      "grad_norm": 0.09518405795097351,
      "learning_rate": 8.461164983327331e-08,
      "loss": 0.056,
      "step": 94430
    },
    {
      "epoch": 1.9483566284179687e-05,
      "model_forward_time": 2.5121283531188965,
      "step": 94430
    },
    {
      "dynamics_loss_avg": 0.05928237363696098,
      "epoch": 1.9483566284179687e-05,
      "step": 94430
    },
    {
      "action_loss_avg": 0.000895177060738206,
      "epoch": 1.9483566284179687e-05,
      "step": 94430
    },
    {
      "epoch": 1.9483566284179687e-05,
      "step": 94430,
      "training_step_time": 8.755725860595703
    },
    {
      "epoch": 1.9484519958496095e-05,
      "model_forward_time": 1.8970353603363037,
      "step": 94431
    },
    {
      "epoch": 1.9484519958496095e-05,
      "step": 94431,
      "training_step_time": 9.832152366638184
    },
    {
      "epoch": 1.94854736328125e-05,
      "model_forward_time": 1.8917765617370605,
      "step": 94432
    },
    {
      "epoch": 1.94854736328125e-05,
      "step": 94432,
      "training_step_time": 7.945537805557251
    },
    {
      "epoch": 1.9486427307128906e-05,
      "model_forward_time": 1.892592191696167,
      "step": 94433
    },
    {
      "epoch": 1.9486427307128906e-05,
      "step": 94433,
      "training_step_time": 8.548477172851562
    },
    {
      "epoch": 1.9487380981445314e-05,
      "model_forward_time": 1.8993988037109375,
      "step": 94434
    },
    {
      "epoch": 1.9487380981445314e-05,
      "step": 94434,
      "training_step_time": 11.047368049621582
    },
    {
      "epoch": 1.9488334655761718e-05,
      "model_forward_time": 1.8992919921875,
      "step": 94435
    },
    {
      "epoch": 1.9488334655761718e-05,
      "step": 94435,
      "training_step_time": 6.8702476024627686
    },
    {
      "epoch": 1.9489288330078125e-05,
      "model_forward_time": 1.907935619354248,
      "step": 94436
    },
    {
      "epoch": 1.9489288330078125e-05,
      "step": 94436,
      "training_step_time": 7.916823625564575
    },
    {
      "epoch": 1.9490242004394532e-05,
      "model_forward_time": 1.8973643779754639,
      "step": 94437
    },
    {
      "epoch": 1.9490242004394532e-05,
      "step": 94437,
      "training_step_time": 7.7521562576293945
    },
    {
      "epoch": 1.9491195678710936e-05,
      "model_forward_time": 1.9108467102050781,
      "step": 94438
    },
    {
      "epoch": 1.9491195678710936e-05,
      "step": 94438,
      "training_step_time": 7.846742153167725
    },
    {
      "epoch": 1.9492149353027344e-05,
      "model_forward_time": 1.9077668190002441,
      "step": 94439
    },
    {
      "epoch": 1.9492149353027344e-05,
      "step": 94439,
      "training_step_time": 6.885149002075195
    },
    {
      "epoch": 1.949310302734375e-05,
      "grad_norm": 0.07832522690296173,
      "learning_rate": 8.430902074267334e-08,
      "loss": 0.0535,
      "step": 94440
    },
    {
      "epoch": 1.949310302734375e-05,
      "model_forward_time": 1.9004497528076172,
      "step": 94440
    },
    {
      "dynamics_loss_avg": 0.04478857424110174,
      "epoch": 1.949310302734375e-05,
      "step": 94440
    },
    {
      "action_loss_avg": 0.0011198806780157611,
      "epoch": 1.949310302734375e-05,
      "step": 94440
    },
    {
      "epoch": 1.949310302734375e-05,
      "step": 94440,
      "training_step_time": 6.889211416244507
    },
    {
      "epoch": 1.9494056701660155e-05,
      "model_forward_time": 1.8919572830200195,
      "step": 94441
    },
    {
      "epoch": 1.9494056701660155e-05,
      "step": 94441,
      "training_step_time": 6.9019246101379395
    },
    {
      "epoch": 1.9495010375976563e-05,
      "model_forward_time": 1.907644510269165,
      "step": 94442
    },
    {
      "epoch": 1.9495010375976563e-05,
      "step": 94442,
      "training_step_time": 6.895315408706665
    },
    {
      "epoch": 1.949596405029297e-05,
      "model_forward_time": 1.888932466506958,
      "step": 94443
    },
    {
      "epoch": 1.949596405029297e-05,
      "step": 94443,
      "training_step_time": 6.911469221115112
    },
    {
      "epoch": 1.9496917724609374e-05,
      "model_forward_time": 2.5337233543395996,
      "step": 94444
    },
    {
      "epoch": 1.9496917724609374e-05,
      "step": 94444,
      "training_step_time": 7.926874399185181
    },
    {
      "epoch": 1.949787139892578e-05,
      "model_forward_time": 1.888343334197998,
      "step": 94445
    },
    {
      "epoch": 1.949787139892578e-05,
      "step": 94445,
      "training_step_time": 6.939532995223999
    },
    {
      "epoch": 1.949882507324219e-05,
      "model_forward_time": 1.8881499767303467,
      "step": 94446
    },
    {
      "epoch": 1.949882507324219e-05,
      "step": 94446,
      "training_step_time": 7.739186763763428
    },
    {
      "epoch": 1.9499778747558593e-05,
      "model_forward_time": 1.9011142253875732,
      "step": 94447
    },
    {
      "epoch": 1.9499778747558593e-05,
      "step": 94447,
      "training_step_time": 7.6756017208099365
    },
    {
      "epoch": 1.9500732421875e-05,
      "model_forward_time": 1.8938329219818115,
      "step": 94448
    },
    {
      "epoch": 1.9500732421875e-05,
      "step": 94448,
      "training_step_time": 7.541423082351685
    },
    {
      "epoch": 1.9501686096191408e-05,
      "model_forward_time": 1.8970723152160645,
      "step": 94449
    },
    {
      "epoch": 1.9501686096191408e-05,
      "step": 94449,
      "training_step_time": 6.922980546951294
    },
    {
      "epoch": 1.9502639770507812e-05,
      "grad_norm": 0.09404158592224121,
      "learning_rate": 8.400692922464926e-08,
      "loss": 0.0541,
      "step": 94450
    },
    {
      "epoch": 1.9502639770507812e-05,
      "model_forward_time": 1.892275333404541,
      "step": 94450
    },
    {
      "dynamics_loss_avg": 0.05133874416351318,
      "epoch": 1.9502639770507812e-05,
      "step": 94450
    },
    {
      "action_loss_avg": 0.001226028290693648,
      "epoch": 1.9502639770507812e-05,
      "step": 94450
    },
    {
      "epoch": 1.9502639770507812e-05,
      "step": 94450,
      "training_step_time": 6.8789222240448
    },
    {
      "epoch": 1.950359344482422e-05,
      "model_forward_time": 1.883795976638794,
      "step": 94451
    },
    {
      "epoch": 1.950359344482422e-05,
      "step": 94451,
      "training_step_time": 6.887782335281372
    },
    {
      "epoch": 1.9504547119140627e-05,
      "model_forward_time": 1.9053328037261963,
      "step": 94452
    },
    {
      "epoch": 1.9504547119140627e-05,
      "step": 94452,
      "training_step_time": 6.913994550704956
    },
    {
      "epoch": 1.950550079345703e-05,
      "model_forward_time": 2.525254249572754,
      "step": 94453
    },
    {
      "epoch": 1.950550079345703e-05,
      "step": 94453,
      "training_step_time": 7.951363563537598
    },
    {
      "epoch": 1.9506454467773438e-05,
      "model_forward_time": 1.9130806922912598,
      "step": 94454
    },
    {
      "epoch": 1.9506454467773438e-05,
      "step": 94454,
      "training_step_time": 7.675384283065796
    },
    {
      "epoch": 1.9507408142089842e-05,
      "model_forward_time": 1.8962092399597168,
      "step": 94455
    },
    {
      "epoch": 1.9507408142089842e-05,
      "step": 94455,
      "training_step_time": 7.816020727157593
    },
    {
      "epoch": 1.950836181640625e-05,
      "model_forward_time": 1.8896052837371826,
      "step": 94456
    },
    {
      "epoch": 1.950836181640625e-05,
      "step": 94456,
      "training_step_time": 7.712189435958862
    },
    {
      "epoch": 1.9509315490722657e-05,
      "model_forward_time": 1.8933494091033936,
      "step": 94457
    },
    {
      "epoch": 1.9509315490722657e-05,
      "step": 94457,
      "training_step_time": 7.652273178100586
    },
    {
      "epoch": 1.951026916503906e-05,
      "model_forward_time": 1.9006807804107666,
      "step": 94458
    },
    {
      "epoch": 1.951026916503906e-05,
      "step": 94458,
      "training_step_time": 6.982248544692993
    },
    {
      "epoch": 1.951122283935547e-05,
      "model_forward_time": 1.8919875621795654,
      "step": 94459
    },
    {
      "epoch": 1.951122283935547e-05,
      "step": 94459,
      "training_step_time": 6.956274747848511
    },
    {
      "epoch": 1.9512176513671876e-05,
      "grad_norm": 0.0746900886297226,
      "learning_rate": 8.370537531223576e-08,
      "loss": 0.054,
      "step": 94460
    },
    {
      "epoch": 1.9512176513671876e-05,
      "model_forward_time": 1.8871395587921143,
      "step": 94460
    },
    {
      "dynamics_loss_avg": 0.05990312360227108,
      "epoch": 1.9512176513671876e-05,
      "step": 94460
    },
    {
      "action_loss_avg": 0.0008404979511396959,
      "epoch": 1.9512176513671876e-05,
      "step": 94460
    },
    {
      "epoch": 1.9512176513671876e-05,
      "step": 94460,
      "training_step_time": 6.993002891540527
    },
    {
      "epoch": 1.951313018798828e-05,
      "model_forward_time": 1.8956000804901123,
      "step": 94461
    },
    {
      "epoch": 1.951313018798828e-05,
      "step": 94461,
      "training_step_time": 6.9806928634643555
    },
    {
      "epoch": 1.9514083862304687e-05,
      "model_forward_time": 2.506535530090332,
      "step": 94462
    },
    {
      "epoch": 1.9514083862304687e-05,
      "step": 94462,
      "training_step_time": 8.814500570297241
    },
    {
      "epoch": 1.9515037536621095e-05,
      "model_forward_time": 1.8940043449401855,
      "step": 94463
    },
    {
      "epoch": 1.9515037536621095e-05,
      "step": 94463,
      "training_step_time": 7.743264198303223
    },
    {
      "epoch": 1.95159912109375e-05,
      "model_forward_time": 1.8930554389953613,
      "step": 94464
    },
    {
      "epoch": 1.95159912109375e-05,
      "step": 94464,
      "training_step_time": 7.887953996658325
    },
    {
      "epoch": 1.9516944885253906e-05,
      "model_forward_time": 1.8950822353363037,
      "step": 94465
    },
    {
      "epoch": 1.9516944885253906e-05,
      "step": 94465,
      "training_step_time": 7.74485182762146
    },
    {
      "epoch": 1.9517898559570314e-05,
      "model_forward_time": 1.8952281475067139,
      "step": 94466
    },
    {
      "epoch": 1.9517898559570314e-05,
      "step": 94466,
      "training_step_time": 7.667571783065796
    },
    {
      "epoch": 1.9518852233886718e-05,
      "model_forward_time": 1.9032385349273682,
      "step": 94467
    },
    {
      "epoch": 1.9518852233886718e-05,
      "step": 94467,
      "training_step_time": 6.98897647857666
    },
    {
      "epoch": 1.9519805908203125e-05,
      "model_forward_time": 1.889164924621582,
      "step": 94468
    },
    {
      "epoch": 1.9519805908203125e-05,
      "step": 94468,
      "training_step_time": 6.9775121212005615
    },
    {
      "epoch": 1.9520759582519532e-05,
      "model_forward_time": 1.8869946002960205,
      "step": 94469
    },
    {
      "epoch": 1.9520759582519532e-05,
      "step": 94469,
      "training_step_time": 6.9547059535980225
    },
    {
      "epoch": 1.9521713256835936e-05,
      "grad_norm": 0.10238315165042877,
      "learning_rate": 8.340435903841093e-08,
      "loss": 0.0562,
      "step": 94470
    },
    {
      "epoch": 1.9521713256835936e-05,
      "model_forward_time": 1.895824909210205,
      "step": 94470
    },
    {
      "dynamics_loss_avg": 0.05285810809582472,
      "epoch": 1.9521713256835936e-05,
      "step": 94470
    },
    {
      "action_loss_avg": 0.001071670895908028,
      "epoch": 1.9521713256835936e-05,
      "step": 94470
    },
    {
      "epoch": 1.9521713256835936e-05,
      "step": 94470,
      "training_step_time": 6.972671031951904
    },
    {
      "epoch": 1.9522666931152344e-05,
      "model_forward_time": 2.5246996879577637,
      "step": 94471
    },
    {
      "epoch": 1.9522666931152344e-05,
      "step": 94471,
      "training_step_time": 10.612051725387573
    },
    {
      "epoch": 1.952362060546875e-05,
      "model_forward_time": 1.8939526081085205,
      "step": 94472
    },
    {
      "epoch": 1.952362060546875e-05,
      "step": 94472,
      "training_step_time": 7.808733701705933
    },
    {
      "epoch": 1.9524574279785155e-05,
      "model_forward_time": 1.9088149070739746,
      "step": 94473
    },
    {
      "epoch": 1.9524574279785155e-05,
      "step": 94473,
      "training_step_time": 7.77734375
    },
    {
      "epoch": 1.9525527954101563e-05,
      "model_forward_time": 1.8960719108581543,
      "step": 94474
    },
    {
      "epoch": 1.9525527954101563e-05,
      "step": 94474,
      "training_step_time": 7.715685129165649
    },
    {
      "epoch": 1.952648162841797e-05,
      "model_forward_time": 1.8990554809570312,
      "step": 94475
    },
    {
      "epoch": 1.952648162841797e-05,
      "step": 94475,
      "training_step_time": 7.549465656280518
    },
    {
      "epoch": 1.9527435302734374e-05,
      "model_forward_time": 1.895998239517212,
      "step": 94476
    },
    {
      "epoch": 1.9527435302734374e-05,
      "step": 94476,
      "training_step_time": 7.207650184631348
    },
    {
      "epoch": 1.952838897705078e-05,
      "model_forward_time": 1.9109747409820557,
      "step": 94477
    },
    {
      "epoch": 1.952838897705078e-05,
      "step": 94477,
      "training_step_time": 6.878206729888916
    },
    {
      "epoch": 1.952934265136719e-05,
      "model_forward_time": 1.925701379776001,
      "step": 94478
    },
    {
      "epoch": 1.952934265136719e-05,
      "step": 94478,
      "training_step_time": 6.880255222320557
    },
    {
      "epoch": 1.9530296325683593e-05,
      "model_forward_time": 1.9034624099731445,
      "step": 94479
    },
    {
      "epoch": 1.9530296325683593e-05,
      "step": 94479,
      "training_step_time": 6.849005937576294
    },
    {
      "epoch": 1.953125e-05,
      "grad_norm": 0.11785299330949783,
      "learning_rate": 8.31038804360934e-08,
      "loss": 0.0544,
      "step": 94480
    },
    {
      "epoch": 1.953125e-05,
      "model_forward_time": 2.506988763809204,
      "step": 94480
    },
    {
      "dynamics_loss_avg": 0.05137816686183214,
      "epoch": 1.953125e-05,
      "step": 94480
    },
    {
      "action_loss_avg": 0.0011151917889947072,
      "epoch": 1.953125e-05,
      "step": 94480
    },
    {
      "epoch": 1.953125e-05,
      "step": 94480,
      "training_step_time": 11.046631097793579
    },
    {
      "epoch": 1.9532203674316408e-05,
      "model_forward_time": 1.888958215713501,
      "step": 94481
    },
    {
      "epoch": 1.9532203674316408e-05,
      "step": 94481,
      "training_step_time": 7.806111097335815
    },
    {
      "epoch": 1.9533157348632812e-05,
      "model_forward_time": 1.8967576026916504,
      "step": 94482
    },
    {
      "epoch": 1.9533157348632812e-05,
      "step": 94482,
      "training_step_time": 7.729497671127319
    },
    {
      "epoch": 1.953411102294922e-05,
      "model_forward_time": 1.8840663433074951,
      "step": 94483
    },
    {
      "epoch": 1.953411102294922e-05,
      "step": 94483,
      "training_step_time": 7.666751861572266
    },
    {
      "epoch": 1.9535064697265627e-05,
      "model_forward_time": 1.8820843696594238,
      "step": 94484
    },
    {
      "epoch": 1.9535064697265627e-05,
      "step": 94484,
      "training_step_time": 7.546919584274292
    },
    {
      "epoch": 1.953601837158203e-05,
      "model_forward_time": 1.900801181793213,
      "step": 94485
    },
    {
      "epoch": 1.953601837158203e-05,
      "step": 94485,
      "training_step_time": 7.554493188858032
    },
    {
      "epoch": 1.9536972045898438e-05,
      "model_forward_time": 1.9037010669708252,
      "step": 94486
    },
    {
      "epoch": 1.9536972045898438e-05,
      "step": 94486,
      "training_step_time": 6.899118423461914
    },
    {
      "epoch": 1.9537925720214842e-05,
      "model_forward_time": 1.887441873550415,
      "step": 94487
    },
    {
      "epoch": 1.9537925720214842e-05,
      "step": 94487,
      "training_step_time": 6.878647089004517
    },
    {
      "epoch": 1.953887939453125e-05,
      "model_forward_time": 1.884279727935791,
      "step": 94488
    },
    {
      "epoch": 1.953887939453125e-05,
      "step": 94488,
      "training_step_time": 6.840951204299927
    },
    {
      "epoch": 1.9539833068847657e-05,
      "model_forward_time": 1.9132258892059326,
      "step": 94489
    },
    {
      "epoch": 1.9539833068847657e-05,
      "step": 94489,
      "training_step_time": 9.405928611755371
    },
    {
      "epoch": 1.954078674316406e-05,
      "grad_norm": 0.07329902052879333,
      "learning_rate": 8.280393953814303e-08,
      "loss": 0.0539,
      "step": 94490
    },
    {
      "epoch": 1.954078674316406e-05,
      "model_forward_time": 1.8777289390563965,
      "step": 94490
    },
    {
      "dynamics_loss_avg": 0.05011970121413469,
      "epoch": 1.954078674316406e-05,
      "step": 94490
    },
    {
      "action_loss_avg": 0.0007739158842014149,
      "epoch": 1.954078674316406e-05,
      "step": 94490
    },
    {
      "epoch": 1.954078674316406e-05,
      "step": 94490,
      "training_step_time": 9.552383422851562
    },
    {
      "epoch": 1.954174041748047e-05,
      "model_forward_time": 1.9076721668243408,
      "step": 94491
    },
    {
      "epoch": 1.954174041748047e-05,
      "step": 94491,
      "training_step_time": 6.891549587249756
    },
    {
      "epoch": 1.9542694091796876e-05,
      "model_forward_time": 1.899271011352539,
      "step": 94492
    },
    {
      "epoch": 1.9542694091796876e-05,
      "step": 94492,
      "training_step_time": 7.806975364685059
    },
    {
      "epoch": 1.954364776611328e-05,
      "model_forward_time": 1.8896384239196777,
      "step": 94493
    },
    {
      "epoch": 1.954364776611328e-05,
      "step": 94493,
      "training_step_time": 7.639191627502441
    },
    {
      "epoch": 1.9544601440429687e-05,
      "model_forward_time": 1.9119722843170166,
      "step": 94494
    },
    {
      "epoch": 1.9544601440429687e-05,
      "step": 94494,
      "training_step_time": 7.36650824546814
    },
    {
      "epoch": 1.9545555114746095e-05,
      "model_forward_time": 1.9003572463989258,
      "step": 94495
    },
    {
      "epoch": 1.9545555114746095e-05,
      "step": 94495,
      "training_step_time": 6.9192891120910645
    },
    {
      "epoch": 1.95465087890625e-05,
      "model_forward_time": 1.8895227909088135,
      "step": 94496
    },
    {
      "epoch": 1.95465087890625e-05,
      "step": 94496,
      "training_step_time": 6.931262969970703
    },
    {
      "epoch": 1.9547462463378906e-05,
      "model_forward_time": 1.8982934951782227,
      "step": 94497
    },
    {
      "epoch": 1.9547462463378906e-05,
      "step": 94497,
      "training_step_time": 6.916069746017456
    },
    {
      "epoch": 1.9548416137695314e-05,
      "model_forward_time": 1.8878920078277588,
      "step": 94498
    },
    {
      "epoch": 1.9548416137695314e-05,
      "step": 94498,
      "training_step_time": 9.445463180541992
    },
    {
      "epoch": 1.9549369812011718e-05,
      "model_forward_time": 2.5223512649536133,
      "step": 94499
    },
    {
      "epoch": 1.9549369812011718e-05,
      "step": 94499,
      "training_step_time": 9.86318826675415
    },
    {
      "epoch": 1.9550323486328125e-05,
      "grad_norm": 0.08913417160511017,
      "learning_rate": 8.250453637736078e-08,
      "loss": 0.056,
      "step": 94500
    },
    {
      "epoch": 1.9550323486328125e-05,
      "model_forward_time": 1.8849725723266602,
      "step": 94500
    },
    {
      "dynamics_loss_avg": 0.06248975042253733,
      "epoch": 1.9550323486328125e-05,
      "step": 94500
    },
    {
      "action_loss_avg": 0.000958374174661003,
      "epoch": 1.9550323486328125e-05,
      "step": 94500
    },
    {
      "epoch": 1.9550323486328125e-05,
      "step": 94500,
      "training_step_time": 7.696446895599365
    },
    {
      "epoch": 1.9551277160644532e-05,
      "model_forward_time": 1.9083530902862549,
      "step": 94501
    },
    {
      "epoch": 1.9551277160644532e-05,
      "step": 94501,
      "training_step_time": 7.857513189315796
    },
    {
      "epoch": 1.9552230834960936e-05,
      "model_forward_time": 1.887976884841919,
      "step": 94502
    },
    {
      "epoch": 1.9552230834960936e-05,
      "step": 94502,
      "training_step_time": 7.6822144985198975
    },
    {
      "epoch": 1.9553184509277344e-05,
      "model_forward_time": 1.892557144165039,
      "step": 94503
    },
    {
      "epoch": 1.9553184509277344e-05,
      "step": 94503,
      "training_step_time": 7.5418009757995605
    },
    {
      "epoch": 1.955413818359375e-05,
      "model_forward_time": 1.8879854679107666,
      "step": 94504
    },
    {
      "epoch": 1.955413818359375e-05,
      "step": 94504,
      "training_step_time": 6.920521974563599
    },
    {
      "epoch": 1.9555091857910155e-05,
      "model_forward_time": 1.8992292881011963,
      "step": 94505
    },
    {
      "epoch": 1.9555091857910155e-05,
      "step": 94505,
      "training_step_time": 6.915683746337891
    },
    {
      "epoch": 1.9556045532226563e-05,
      "model_forward_time": 1.8878302574157715,
      "step": 94506
    },
    {
      "epoch": 1.9556045532226563e-05,
      "step": 94506,
      "training_step_time": 6.8585145473480225
    },
    {
      "epoch": 1.955699920654297e-05,
      "model_forward_time": 1.8963570594787598,
      "step": 94507
    },
    {
      "epoch": 1.955699920654297e-05,
      "step": 94507,
      "training_step_time": 7.5790486335754395
    },
    {
      "epoch": 1.9557952880859374e-05,
      "model_forward_time": 2.514970541000366,
      "step": 94508
    },
    {
      "epoch": 1.9557952880859374e-05,
      "step": 94508,
      "training_step_time": 9.338090181350708
    },
    {
      "epoch": 1.955890655517578e-05,
      "model_forward_time": 1.890587568283081,
      "step": 94509
    },
    {
      "epoch": 1.955890655517578e-05,
      "step": 94509,
      "training_step_time": 7.855733633041382
    },
    {
      "epoch": 1.955986022949219e-05,
      "grad_norm": 0.0818832665681839,
      "learning_rate": 8.22056709864899e-08,
      "loss": 0.0539,
      "step": 94510
    },
    {
      "epoch": 1.955986022949219e-05,
      "model_forward_time": 1.8839638233184814,
      "step": 94510
    },
    {
      "dynamics_loss_avg": 0.0537650465965271,
      "epoch": 1.955986022949219e-05,
      "step": 94510
    },
    {
      "action_loss_avg": 0.001181671701488085,
      "epoch": 1.955986022949219e-05,
      "step": 94510
    },
    {
      "epoch": 1.955986022949219e-05,
      "step": 94510,
      "training_step_time": 7.8058021068573
    },
    {
      "epoch": 1.9560813903808593e-05,
      "model_forward_time": 1.888542652130127,
      "step": 94511
    },
    {
      "epoch": 1.9560813903808593e-05,
      "step": 94511,
      "training_step_time": 7.6549341678619385
    },
    {
      "epoch": 1.9561767578125e-05,
      "model_forward_time": 1.8919053077697754,
      "step": 94512
    },
    {
      "epoch": 1.9561767578125e-05,
      "step": 94512,
      "training_step_time": 8.14927077293396
    },
    {
      "epoch": 1.9562721252441408e-05,
      "model_forward_time": 1.5152792930603027,
      "step": 94513
    },
    {
      "epoch": 1.9562721252441408e-05,
      "step": 94513,
      "training_step_time": 6.910808086395264
    },
    {
      "epoch": 1.9563674926757812e-05,
      "model_forward_time": 1.8729209899902344,
      "step": 94514
    },
    {
      "epoch": 1.9563674926757812e-05,
      "step": 94514,
      "training_step_time": 6.860581874847412
    },
    {
      "epoch": 1.956462860107422e-05,
      "model_forward_time": 1.8957743644714355,
      "step": 94515
    },
    {
      "epoch": 1.956462860107422e-05,
      "step": 94515,
      "training_step_time": 6.876497507095337
    },
    {
      "epoch": 1.9565582275390623e-05,
      "model_forward_time": 1.893449306488037,
      "step": 94516
    },
    {
      "epoch": 1.9565582275390623e-05,
      "step": 94516,
      "training_step_time": 7.523198843002319
    },
    {
      "epoch": 1.956653594970703e-05,
      "model_forward_time": 2.4892237186431885,
      "step": 94517
    },
    {
      "epoch": 1.956653594970703e-05,
      "step": 94517,
      "training_step_time": 11.01611065864563
    },
    {
      "epoch": 1.9567489624023438e-05,
      "model_forward_time": 1.8913207054138184,
      "step": 94518
    },
    {
      "epoch": 1.9567489624023438e-05,
      "step": 94518,
      "training_step_time": 7.794793605804443
    },
    {
      "epoch": 1.9568443298339842e-05,
      "model_forward_time": 1.8577289581298828,
      "step": 94519
    },
    {
      "epoch": 1.9568443298339842e-05,
      "step": 94519,
      "training_step_time": 7.707806348800659
    },
    {
      "epoch": 1.956939697265625e-05,
      "grad_norm": 0.09382237493991852,
      "learning_rate": 8.190734339821205e-08,
      "loss": 0.0559,
      "step": 94520
    },
    {
      "epoch": 1.956939697265625e-05,
      "model_forward_time": 1.8905236721038818,
      "step": 94520
    },
    {
      "dynamics_loss_avg": 0.05248228460550308,
      "epoch": 1.956939697265625e-05,
      "step": 94520
    },
    {
      "action_loss_avg": 0.0008675499411765486,
      "epoch": 1.956939697265625e-05,
      "step": 94520
    },
    {
      "epoch": 1.956939697265625e-05,
      "step": 94520,
      "training_step_time": 7.675091505050659
    },
    {
      "epoch": 1.9570350646972657e-05,
      "model_forward_time": 1.8914453983306885,
      "step": 94521
    },
    {
      "epoch": 1.9570350646972657e-05,
      "step": 94521,
      "training_step_time": 8.265513896942139
    },
    {
      "epoch": 1.957130432128906e-05,
      "model_forward_time": 1.894282341003418,
      "step": 94522
    },
    {
      "epoch": 1.957130432128906e-05,
      "step": 94522,
      "training_step_time": 7.2247045040130615
    },
    {
      "epoch": 1.957225799560547e-05,
      "model_forward_time": 1.8809497356414795,
      "step": 94523
    },
    {
      "epoch": 1.957225799560547e-05,
      "step": 94523,
      "training_step_time": 6.869614601135254
    },
    {
      "epoch": 1.9573211669921876e-05,
      "model_forward_time": 1.8902075290679932,
      "step": 94524
    },
    {
      "epoch": 1.9573211669921876e-05,
      "step": 94524,
      "training_step_time": 6.866908311843872
    },
    {
      "epoch": 1.957416534423828e-05,
      "model_forward_time": 1.903960943222046,
      "step": 94525
    },
    {
      "epoch": 1.957416534423828e-05,
      "step": 94525,
      "training_step_time": 7.540121555328369
    },
    {
      "epoch": 1.9575119018554687e-05,
      "model_forward_time": 2.5098090171813965,
      "step": 94526
    },
    {
      "epoch": 1.9575119018554687e-05,
      "step": 94526,
      "training_step_time": 10.133695125579834
    },
    {
      "epoch": 1.9576072692871095e-05,
      "model_forward_time": 1.889906883239746,
      "step": 94527
    },
    {
      "epoch": 1.9576072692871095e-05,
      "step": 94527,
      "training_step_time": 7.925930738449097
    },
    {
      "epoch": 1.95770263671875e-05,
      "model_forward_time": 1.5324063301086426,
      "step": 94528
    },
    {
      "epoch": 1.95770263671875e-05,
      "step": 94528,
      "training_step_time": 8.606989622116089
    },
    {
      "epoch": 1.9577980041503906e-05,
      "model_forward_time": 1.8661894798278809,
      "step": 94529
    },
    {
      "epoch": 1.9577980041503906e-05,
      "step": 94529,
      "training_step_time": 7.688797473907471
    },
    {
      "epoch": 1.9578933715820314e-05,
      "grad_norm": 0.1718752533197403,
      "learning_rate": 8.160955364515333e-08,
      "loss": 0.0544,
      "step": 94530
    },
    {
      "epoch": 1.9578933715820314e-05,
      "model_forward_time": 1.887643814086914,
      "step": 94530
    },
    {
      "dynamics_loss_avg": 0.04496383499354124,
      "epoch": 1.9578933715820314e-05,
      "step": 94530
    },
    {
      "action_loss_avg": 0.0012133773998357356,
      "epoch": 1.9578933715820314e-05,
      "step": 94530
    },
    {
      "epoch": 1.9578933715820314e-05,
      "step": 94530,
      "training_step_time": 7.545727968215942
    },
    {
      "epoch": 1.9579887390136718e-05,
      "model_forward_time": 1.8844666481018066,
      "step": 94531
    },
    {
      "epoch": 1.9579887390136718e-05,
      "step": 94531,
      "training_step_time": 7.5639002323150635
    },
    {
      "epoch": 1.9580841064453125e-05,
      "model_forward_time": 1.907703161239624,
      "step": 94532
    },
    {
      "epoch": 1.9580841064453125e-05,
      "step": 94532,
      "training_step_time": 6.958383560180664
    },
    {
      "epoch": 1.9581794738769532e-05,
      "model_forward_time": 1.8779711723327637,
      "step": 94533
    },
    {
      "epoch": 1.9581794738769532e-05,
      "step": 94533,
      "training_step_time": 6.9284443855285645
    },
    {
      "epoch": 1.9582748413085936e-05,
      "model_forward_time": 1.8678960800170898,
      "step": 94534
    },
    {
      "epoch": 1.9582748413085936e-05,
      "step": 94534,
      "training_step_time": 7.566938638687134
    },
    {
      "epoch": 1.9583702087402344e-05,
      "model_forward_time": 2.5296905040740967,
      "step": 94535
    },
    {
      "epoch": 1.9583702087402344e-05,
      "step": 94535,
      "training_step_time": 9.848641872406006
    },
    {
      "epoch": 1.958465576171875e-05,
      "model_forward_time": 1.9039011001586914,
      "step": 94536
    },
    {
      "epoch": 1.958465576171875e-05,
      "step": 94536,
      "training_step_time": 8.736304759979248
    },
    {
      "epoch": 1.9585609436035155e-05,
      "model_forward_time": 1.8935298919677734,
      "step": 94537
    },
    {
      "epoch": 1.9585609436035155e-05,
      "step": 94537,
      "training_step_time": 8.23091173171997
    },
    {
      "epoch": 1.9586563110351563e-05,
      "model_forward_time": 1.9002680778503418,
      "step": 94538
    },
    {
      "epoch": 1.9586563110351563e-05,
      "step": 94538,
      "training_step_time": 7.8041746616363525
    },
    {
      "epoch": 1.958751678466797e-05,
      "model_forward_time": 1.9074592590332031,
      "step": 94539
    },
    {
      "epoch": 1.958751678466797e-05,
      "step": 94539,
      "training_step_time": 7.658977746963501
    },
    {
      "epoch": 1.9588470458984374e-05,
      "grad_norm": 0.08034341782331467,
      "learning_rate": 8.131230175987936e-08,
      "loss": 0.0554,
      "step": 94540
    },
    {
      "epoch": 1.9588470458984374e-05,
      "model_forward_time": 1.8827691078186035,
      "step": 94540
    },
    {
      "dynamics_loss_avg": 0.051469696871936324,
      "epoch": 1.9588470458984374e-05,
      "step": 94540
    },
    {
      "action_loss_avg": 0.0015506943862419575,
      "epoch": 1.9588470458984374e-05,
      "step": 94540
    },
    {
      "epoch": 1.9588470458984374e-05,
      "step": 94540,
      "training_step_time": 7.537185430526733
    },
    {
      "epoch": 1.958942413330078e-05,
      "model_forward_time": 1.9105432033538818,
      "step": 94541
    },
    {
      "epoch": 1.958942413330078e-05,
      "step": 94541,
      "training_step_time": 6.917353391647339
    },
    {
      "epoch": 1.959037780761719e-05,
      "model_forward_time": 1.9042563438415527,
      "step": 94542
    },
    {
      "epoch": 1.959037780761719e-05,
      "step": 94542,
      "training_step_time": 6.9195380210876465
    },
    {
      "epoch": 1.9591331481933593e-05,
      "model_forward_time": 1.9110987186431885,
      "step": 94543
    },
    {
      "epoch": 1.9591331481933593e-05,
      "step": 94543,
      "training_step_time": 7.548860311508179
    },
    {
      "epoch": 1.959228515625e-05,
      "model_forward_time": 1.9052598476409912,
      "step": 94544
    },
    {
      "epoch": 1.959228515625e-05,
      "step": 94544,
      "training_step_time": 10.146424293518066
    },
    {
      "epoch": 1.9593238830566408e-05,
      "model_forward_time": 2.514669179916382,
      "step": 94545
    },
    {
      "epoch": 1.9593238830566408e-05,
      "step": 94545,
      "training_step_time": 11.075362920761108
    },
    {
      "epoch": 1.9594192504882812e-05,
      "model_forward_time": 1.8964579105377197,
      "step": 94546
    },
    {
      "epoch": 1.9594192504882812e-05,
      "step": 94546,
      "training_step_time": 7.796624660491943
    },
    {
      "epoch": 1.959514617919922e-05,
      "model_forward_time": 1.8549401760101318,
      "step": 94547
    },
    {
      "epoch": 1.959514617919922e-05,
      "step": 94547,
      "training_step_time": 7.9345948696136475
    },
    {
      "epoch": 1.9596099853515623e-05,
      "model_forward_time": 1.8938710689544678,
      "step": 94548
    },
    {
      "epoch": 1.9596099853515623e-05,
      "step": 94548,
      "training_step_time": 8.643209457397461
    },
    {
      "epoch": 1.959705352783203e-05,
      "model_forward_time": 1.9025547504425049,
      "step": 94549
    },
    {
      "epoch": 1.959705352783203e-05,
      "step": 94549,
      "training_step_time": 7.658941745758057
    },
    {
      "epoch": 1.9598007202148438e-05,
      "grad_norm": 0.06610037386417389,
      "learning_rate": 8.101558777489693e-08,
      "loss": 0.0552,
      "step": 94550
    },
    {
      "epoch": 1.9598007202148438e-05,
      "model_forward_time": 1.887293815612793,
      "step": 94550
    },
    {
      "dynamics_loss_avg": 0.042708216980099675,
      "epoch": 1.9598007202148438e-05,
      "step": 94550
    },
    {
      "action_loss_avg": 0.0009944355260813609,
      "epoch": 1.9598007202148438e-05,
      "step": 94550
    },
    {
      "epoch": 1.9598007202148438e-05,
      "step": 94550,
      "training_step_time": 6.879419803619385
    },
    {
      "epoch": 1.9598960876464842e-05,
      "model_forward_time": 1.8943016529083252,
      "step": 94551
    },
    {
      "epoch": 1.9598960876464842e-05,
      "step": 94551,
      "training_step_time": 6.894749879837036
    },
    {
      "epoch": 1.959991455078125e-05,
      "model_forward_time": 1.8917813301086426,
      "step": 94552
    },
    {
      "epoch": 1.959991455078125e-05,
      "step": 94552,
      "training_step_time": 8.103265523910522
    },
    {
      "epoch": 1.9600868225097657e-05,
      "model_forward_time": 1.8910958766937256,
      "step": 94553
    },
    {
      "epoch": 1.9600868225097657e-05,
      "step": 94553,
      "training_step_time": 8.878966331481934
    },
    {
      "epoch": 1.960182189941406e-05,
      "model_forward_time": 2.522170066833496,
      "step": 94554
    },
    {
      "epoch": 1.960182189941406e-05,
      "step": 94554,
      "training_step_time": 9.475565433502197
    },
    {
      "epoch": 1.960277557373047e-05,
      "model_forward_time": 1.9021224975585938,
      "step": 94555
    },
    {
      "epoch": 1.960277557373047e-05,
      "step": 94555,
      "training_step_time": 7.839115619659424
    },
    {
      "epoch": 1.9603729248046876e-05,
      "model_forward_time": 1.5042247772216797,
      "step": 94556
    },
    {
      "epoch": 1.9603729248046876e-05,
      "step": 94556,
      "training_step_time": 7.756953001022339
    },
    {
      "epoch": 1.960468292236328e-05,
      "model_forward_time": 1.896820068359375,
      "step": 94557
    },
    {
      "epoch": 1.960468292236328e-05,
      "step": 94557,
      "training_step_time": 7.684358835220337
    },
    {
      "epoch": 1.9605636596679687e-05,
      "model_forward_time": 1.9044349193572998,
      "step": 94558
    },
    {
      "epoch": 1.9605636596679687e-05,
      "step": 94558,
      "training_step_time": 8.12921929359436
    },
    {
      "epoch": 1.9606590270996095e-05,
      "model_forward_time": 1.8872196674346924,
      "step": 94559
    },
    {
      "epoch": 1.9606590270996095e-05,
      "step": 94559,
      "training_step_time": 7.624927759170532
    },
    {
      "epoch": 1.96075439453125e-05,
      "grad_norm": 0.0825091153383255,
      "learning_rate": 8.071941172265396e-08,
      "loss": 0.0558,
      "step": 94560
    },
    {
      "epoch": 1.96075439453125e-05,
      "model_forward_time": 1.9042065143585205,
      "step": 94560
    },
    {
      "dynamics_loss_avg": 0.0432748575694859,
      "epoch": 1.96075439453125e-05,
      "step": 94560
    },
    {
      "action_loss_avg": 0.0008816986868623644,
      "epoch": 1.96075439453125e-05,
      "step": 94560
    },
    {
      "epoch": 1.96075439453125e-05,
      "step": 94560,
      "training_step_time": 6.916375398635864
    },
    {
      "epoch": 1.9608497619628906e-05,
      "model_forward_time": 1.913273811340332,
      "step": 94561
    },
    {
      "epoch": 1.9608497619628906e-05,
      "step": 94561,
      "training_step_time": 7.265358209609985
    },
    {
      "epoch": 1.9609451293945314e-05,
      "model_forward_time": 1.9186482429504395,
      "step": 94562
    },
    {
      "epoch": 1.9609451293945314e-05,
      "step": 94562,
      "training_step_time": 7.814590215682983
    },
    {
      "epoch": 1.9610404968261718e-05,
      "model_forward_time": 2.553959846496582,
      "step": 94563
    },
    {
      "epoch": 1.9610404968261718e-05,
      "step": 94563,
      "training_step_time": 11.953562498092651
    },
    {
      "epoch": 1.9611358642578125e-05,
      "model_forward_time": 1.8941328525543213,
      "step": 94564
    },
    {
      "epoch": 1.9611358642578125e-05,
      "step": 94564,
      "training_step_time": 7.825526237487793
    },
    {
      "epoch": 1.9612312316894532e-05,
      "model_forward_time": 1.9042115211486816,
      "step": 94565
    },
    {
      "epoch": 1.9612312316894532e-05,
      "step": 94565,
      "training_step_time": 7.858649492263794
    },
    {
      "epoch": 1.9613265991210936e-05,
      "model_forward_time": 1.8966548442840576,
      "step": 94566
    },
    {
      "epoch": 1.9613265991210936e-05,
      "step": 94566,
      "training_step_time": 7.676880359649658
    },
    {
      "epoch": 1.9614219665527344e-05,
      "model_forward_time": 1.9171462059020996,
      "step": 94567
    },
    {
      "epoch": 1.9614219665527344e-05,
      "step": 94567,
      "training_step_time": 7.516931533813477
    },
    {
      "epoch": 1.961517333984375e-05,
      "model_forward_time": 1.9171836376190186,
      "step": 94568
    },
    {
      "epoch": 1.961517333984375e-05,
      "step": 94568,
      "training_step_time": 7.641038656234741
    },
    {
      "epoch": 1.9616127014160155e-05,
      "model_forward_time": 1.9058427810668945,
      "step": 94569
    },
    {
      "epoch": 1.9616127014160155e-05,
      "step": 94569,
      "training_step_time": 6.919045448303223
    },
    {
      "epoch": 1.9617080688476563e-05,
      "grad_norm": 0.07724490016698837,
      "learning_rate": 8.042377363553954e-08,
      "loss": 0.057,
      "step": 94570
    },
    {
      "epoch": 1.9617080688476563e-05,
      "model_forward_time": 1.9009196758270264,
      "step": 94570
    },
    {
      "dynamics_loss_avg": 0.05173623207956553,
      "epoch": 1.9617080688476563e-05,
      "step": 94570
    },
    {
      "action_loss_avg": 0.0007671019702684134,
      "epoch": 1.9617080688476563e-05,
      "step": 94570
    },
    {
      "epoch": 1.9617080688476563e-05,
      "step": 94570,
      "training_step_time": 7.206151962280273
    },
    {
      "epoch": 1.961803436279297e-05,
      "model_forward_time": 1.9131321907043457,
      "step": 94571
    },
    {
      "epoch": 1.961803436279297e-05,
      "step": 94571,
      "training_step_time": 7.800143480300903
    },
    {
      "epoch": 1.9618988037109374e-05,
      "model_forward_time": 2.544390916824341,
      "step": 94572
    },
    {
      "epoch": 1.9618988037109374e-05,
      "step": 94572,
      "training_step_time": 10.643014669418335
    },
    {
      "epoch": 1.961994171142578e-05,
      "model_forward_time": 1.8968045711517334,
      "step": 94573
    },
    {
      "epoch": 1.961994171142578e-05,
      "step": 94573,
      "training_step_time": 7.83811354637146
    },
    {
      "epoch": 1.962089538574219e-05,
      "model_forward_time": 1.9130921363830566,
      "step": 94574
    },
    {
      "epoch": 1.962089538574219e-05,
      "step": 94574,
      "training_step_time": 8.832133054733276
    },
    {
      "epoch": 1.9621849060058593e-05,
      "model_forward_time": 1.9045546054840088,
      "step": 94575
    },
    {
      "epoch": 1.9621849060058593e-05,
      "step": 94575,
      "training_step_time": 7.440957546234131
    },
    {
      "epoch": 1.9622802734375e-05,
      "model_forward_time": 1.9089550971984863,
      "step": 94576
    },
    {
      "epoch": 1.9622802734375e-05,
      "step": 94576,
      "training_step_time": 7.683522462844849
    },
    {
      "epoch": 1.9623756408691408e-05,
      "model_forward_time": 1.9170341491699219,
      "step": 94577
    },
    {
      "epoch": 1.9623756408691408e-05,
      "step": 94577,
      "training_step_time": 7.632906675338745
    },
    {
      "epoch": 1.9624710083007812e-05,
      "model_forward_time": 1.9049310684204102,
      "step": 94578
    },
    {
      "epoch": 1.9624710083007812e-05,
      "step": 94578,
      "training_step_time": 6.924113035202026
    },
    {
      "epoch": 1.962566375732422e-05,
      "model_forward_time": 1.9161202907562256,
      "step": 94579
    },
    {
      "epoch": 1.962566375732422e-05,
      "step": 94579,
      "training_step_time": 7.226698637008667
    },
    {
      "epoch": 1.9626617431640623e-05,
      "grad_norm": 0.1252792924642563,
      "learning_rate": 8.012867354588561e-08,
      "loss": 0.0558,
      "step": 94580
    },
    {
      "epoch": 1.9626617431640623e-05,
      "model_forward_time": 1.9045138359069824,
      "step": 94580
    },
    {
      "dynamics_loss_avg": 0.05459756925702095,
      "epoch": 1.9626617431640623e-05,
      "step": 94580
    },
    {
      "action_loss_avg": 0.001380587334278971,
      "epoch": 1.9626617431640623e-05,
      "step": 94580
    },
    {
      "epoch": 1.9626617431640623e-05,
      "step": 94580,
      "training_step_time": 7.809262990951538
    },
    {
      "epoch": 1.962757110595703e-05,
      "model_forward_time": 2.5266518592834473,
      "step": 94581
    },
    {
      "epoch": 1.962757110595703e-05,
      "step": 94581,
      "training_step_time": 11.008105039596558
    },
    {
      "epoch": 1.9628524780273438e-05,
      "model_forward_time": 1.9179234504699707,
      "step": 94582
    },
    {
      "epoch": 1.9628524780273438e-05,
      "step": 94582,
      "training_step_time": 8.685036182403564
    },
    {
      "epoch": 1.9629478454589842e-05,
      "model_forward_time": 1.8784029483795166,
      "step": 94583
    },
    {
      "epoch": 1.9629478454589842e-05,
      "step": 94583,
      "training_step_time": 9.496283769607544
    },
    {
      "epoch": 1.963043212890625e-05,
      "model_forward_time": 1.903886318206787,
      "step": 94584
    },
    {
      "epoch": 1.963043212890625e-05,
      "step": 94584,
      "training_step_time": 7.763343811035156
    },
    {
      "epoch": 1.9631385803222657e-05,
      "model_forward_time": 1.9049062728881836,
      "step": 94585
    },
    {
      "epoch": 1.9631385803222657e-05,
      "step": 94585,
      "training_step_time": 7.732718229293823
    },
    {
      "epoch": 1.963233947753906e-05,
      "model_forward_time": 1.9030089378356934,
      "step": 94586
    },
    {
      "epoch": 1.963233947753906e-05,
      "step": 94586,
      "training_step_time": 7.628399848937988
    },
    {
      "epoch": 1.963329315185547e-05,
      "model_forward_time": 1.8757414817810059,
      "step": 94587
    },
    {
      "epoch": 1.963329315185547e-05,
      "step": 94587,
      "training_step_time": 7.307338237762451
    },
    {
      "epoch": 1.9634246826171876e-05,
      "model_forward_time": 1.909209966659546,
      "step": 94588
    },
    {
      "epoch": 1.9634246826171876e-05,
      "step": 94588,
      "training_step_time": 7.243605613708496
    },
    {
      "epoch": 1.963520050048828e-05,
      "model_forward_time": 1.9060349464416504,
      "step": 94589
    },
    {
      "epoch": 1.963520050048828e-05,
      "step": 94589,
      "training_step_time": 8.64310622215271
    },
    {
      "epoch": 1.9636154174804687e-05,
      "grad_norm": 0.1361830234527588,
      "learning_rate": 7.983411148596299e-08,
      "loss": 0.0572,
      "step": 94590
    },
    {
      "epoch": 1.9636154174804687e-05,
      "model_forward_time": 2.489406108856201,
      "step": 94590
    },
    {
      "dynamics_loss_avg": 0.04519093399867415,
      "epoch": 1.9636154174804687e-05,
      "step": 94590
    },
    {
      "action_loss_avg": 0.0018643751041963696,
      "epoch": 1.9636154174804687e-05,
      "step": 94590
    },
    {
      "epoch": 1.9636154174804687e-05,
      "step": 94590,
      "training_step_time": 9.892536401748657
    },
    {
      "epoch": 1.9637107849121095e-05,
      "model_forward_time": 1.9059045314788818,
      "step": 94591
    },
    {
      "epoch": 1.9637107849121095e-05,
      "step": 94591,
      "training_step_time": 8.625406265258789
    },
    {
      "epoch": 1.96380615234375e-05,
      "model_forward_time": 1.8922154903411865,
      "step": 94592
    },
    {
      "epoch": 1.96380615234375e-05,
      "step": 94592,
      "training_step_time": 7.865487575531006
    },
    {
      "epoch": 1.9639015197753906e-05,
      "model_forward_time": 1.8910491466522217,
      "step": 94593
    },
    {
      "epoch": 1.9639015197753906e-05,
      "step": 94593,
      "training_step_time": 7.7765586376190186
    },
    {
      "epoch": 1.9639968872070314e-05,
      "model_forward_time": 1.8861684799194336,
      "step": 94594
    },
    {
      "epoch": 1.9639968872070314e-05,
      "step": 94594,
      "training_step_time": 7.655496597290039
    },
    {
      "epoch": 1.9640922546386718e-05,
      "model_forward_time": 1.9011783599853516,
      "step": 94595
    },
    {
      "epoch": 1.9640922546386718e-05,
      "step": 94595,
      "training_step_time": 7.668236970901489
    },
    {
      "epoch": 1.9641876220703125e-05,
      "model_forward_time": 1.8840422630310059,
      "step": 94596
    },
    {
      "epoch": 1.9641876220703125e-05,
      "step": 94596,
      "training_step_time": 6.955068826675415
    },
    {
      "epoch": 1.9642829895019532e-05,
      "model_forward_time": 1.8889310359954834,
      "step": 94597
    },
    {
      "epoch": 1.9642829895019532e-05,
      "step": 94597,
      "training_step_time": 7.267798900604248
    },
    {
      "epoch": 1.9643783569335937e-05,
      "model_forward_time": 1.903801441192627,
      "step": 94598
    },
    {
      "epoch": 1.9643783569335937e-05,
      "step": 94598,
      "training_step_time": 7.797080993652344
    },
    {
      "epoch": 1.9644737243652344e-05,
      "model_forward_time": 1.8994672298431396,
      "step": 94599
    },
    {
      "epoch": 1.9644737243652344e-05,
      "step": 94599,
      "training_step_time": 10.59640121459961
    },
    {
      "epoch": 1.964569091796875e-05,
      "grad_norm": 0.09174597263336182,
      "learning_rate": 7.954008748798425e-08,
      "loss": 0.0554,
      "step": 94600
    },
    {
      "epoch": 1.964569091796875e-05,
      "model_forward_time": 2.491029977798462,
      "step": 94600
    },
    {
      "dynamics_loss_avg": 0.05241119619458914,
      "epoch": 1.964569091796875e-05,
      "step": 94600
    },
    {
      "action_loss_avg": 0.0011575266544241458,
      "epoch": 1.964569091796875e-05,
      "step": 94600
    },
    {
      "epoch": 1.964569091796875e-05,
      "step": 94600,
      "training_step_time": 9.962204217910767
    },
    {
      "epoch": 1.9646644592285155e-05,
      "model_forward_time": 1.89048171043396,
      "step": 94601
    },
    {
      "epoch": 1.9646644592285155e-05,
      "step": 94601,
      "training_step_time": 8.827252864837646
    },
    {
      "epoch": 1.9647598266601563e-05,
      "model_forward_time": 1.9042651653289795,
      "step": 94602
    },
    {
      "epoch": 1.9647598266601563e-05,
      "step": 94602,
      "training_step_time": 7.881202697753906
    },
    {
      "epoch": 1.964855194091797e-05,
      "model_forward_time": 1.8744521141052246,
      "step": 94603
    },
    {
      "epoch": 1.964855194091797e-05,
      "step": 94603,
      "training_step_time": 7.731459856033325
    },
    {
      "epoch": 1.9649505615234374e-05,
      "model_forward_time": 1.8915231227874756,
      "step": 94604
    },
    {
      "epoch": 1.9649505615234374e-05,
      "step": 94604,
      "training_step_time": 7.647921323776245
    },
    {
      "epoch": 1.965045928955078e-05,
      "model_forward_time": 1.9036617279052734,
      "step": 94605
    },
    {
      "epoch": 1.965045928955078e-05,
      "step": 94605,
      "training_step_time": 7.709960460662842
    },
    {
      "epoch": 1.965141296386719e-05,
      "model_forward_time": 1.8945763111114502,
      "step": 94606
    },
    {
      "epoch": 1.965141296386719e-05,
      "step": 94606,
      "training_step_time": 7.98298716545105
    },
    {
      "epoch": 1.9652366638183593e-05,
      "model_forward_time": 1.8897743225097656,
      "step": 94607
    },
    {
      "epoch": 1.9652366638183593e-05,
      "step": 94607,
      "training_step_time": 7.786839008331299
    },
    {
      "epoch": 1.96533203125e-05,
      "model_forward_time": 1.9055893421173096,
      "step": 94608
    },
    {
      "epoch": 1.96533203125e-05,
      "step": 94608,
      "training_step_time": 7.716354608535767
    },
    {
      "epoch": 1.9654273986816408e-05,
      "model_forward_time": 2.5099470615386963,
      "step": 94609
    },
    {
      "epoch": 1.9654273986816408e-05,
      "step": 94609,
      "training_step_time": 9.682719230651855
    },
    {
      "epoch": 1.9655227661132812e-05,
      "grad_norm": 0.07333847880363464,
      "learning_rate": 7.924660158410369e-08,
      "loss": 0.0538,
      "step": 94610
    },
    {
      "epoch": 1.9655227661132812e-05,
      "model_forward_time": 1.8778228759765625,
      "step": 94610
    },
    {
      "dynamics_loss_avg": 0.04504274819046259,
      "epoch": 1.9655227661132812e-05,
      "step": 94610
    },
    {
      "action_loss_avg": 0.0010695666685933247,
      "epoch": 1.9655227661132812e-05,
      "step": 94610
    },
    {
      "epoch": 1.9655227661132812e-05,
      "step": 94610,
      "training_step_time": 7.937907695770264
    },
    {
      "epoch": 1.965618133544922e-05,
      "model_forward_time": 1.8890328407287598,
      "step": 94611
    },
    {
      "epoch": 1.965618133544922e-05,
      "step": 94611,
      "training_step_time": 9.599372148513794
    },
    {
      "epoch": 1.9657135009765623e-05,
      "model_forward_time": 1.8962805271148682,
      "step": 94612
    },
    {
      "epoch": 1.9657135009765623e-05,
      "step": 94612,
      "training_step_time": 7.230914115905762
    },
    {
      "epoch": 1.965808868408203e-05,
      "model_forward_time": 1.9011871814727783,
      "step": 94613
    },
    {
      "epoch": 1.965808868408203e-05,
      "step": 94613,
      "training_step_time": 7.583672285079956
    },
    {
      "epoch": 1.9659042358398438e-05,
      "model_forward_time": 1.9053621292114258,
      "step": 94614
    },
    {
      "epoch": 1.9659042358398438e-05,
      "step": 94614,
      "training_step_time": 7.266728639602661
    },
    {
      "epoch": 1.9659996032714842e-05,
      "model_forward_time": 1.8980088233947754,
      "step": 94615
    },
    {
      "epoch": 1.9659996032714842e-05,
      "step": 94615,
      "training_step_time": 7.683422803878784
    },
    {
      "epoch": 1.966094970703125e-05,
      "model_forward_time": 1.8992469310760498,
      "step": 94616
    },
    {
      "epoch": 1.966094970703125e-05,
      "step": 94616,
      "training_step_time": 7.831974029541016
    },
    {
      "epoch": 1.9661903381347657e-05,
      "model_forward_time": 1.9104664325714111,
      "step": 94617
    },
    {
      "epoch": 1.9661903381347657e-05,
      "step": 94617,
      "training_step_time": 7.63327169418335
    },
    {
      "epoch": 1.966285705566406e-05,
      "model_forward_time": 2.5165627002716064,
      "step": 94618
    },
    {
      "epoch": 1.966285705566406e-05,
      "step": 94618,
      "training_step_time": 8.349739789962769
    },
    {
      "epoch": 1.966381072998047e-05,
      "model_forward_time": 1.9042048454284668,
      "step": 94619
    },
    {
      "epoch": 1.966381072998047e-05,
      "step": 94619,
      "training_step_time": 9.968677043914795
    },
    {
      "epoch": 1.9664764404296876e-05,
      "grad_norm": 0.13379736244678497,
      "learning_rate": 7.895365380641729e-08,
      "loss": 0.0559,
      "step": 94620
    },
    {
      "epoch": 1.9664764404296876e-05,
      "model_forward_time": 1.8889830112457275,
      "step": 94620
    },
    {
      "dynamics_loss_avg": 0.05974818058311939,
      "epoch": 1.9664764404296876e-05,
      "step": 94620
    },
    {
      "action_loss_avg": 0.0019130110944388435,
      "epoch": 1.9664764404296876e-05,
      "step": 94620
    },
    {
      "epoch": 1.9664764404296876e-05,
      "step": 94620,
      "training_step_time": 7.8207948207855225
    },
    {
      "epoch": 1.966571807861328e-05,
      "model_forward_time": 1.8930003643035889,
      "step": 94621
    },
    {
      "epoch": 1.966571807861328e-05,
      "step": 94621,
      "training_step_time": 7.834818124771118
    },
    {
      "epoch": 1.9666671752929687e-05,
      "model_forward_time": 1.895169973373413,
      "step": 94622
    },
    {
      "epoch": 1.9666671752929687e-05,
      "step": 94622,
      "training_step_time": 8.484283924102783
    },
    {
      "epoch": 1.9667625427246095e-05,
      "model_forward_time": 1.8960397243499756,
      "step": 94623
    },
    {
      "epoch": 1.9667625427246095e-05,
      "step": 94623,
      "training_step_time": 7.270352840423584
    },
    {
      "epoch": 1.96685791015625e-05,
      "model_forward_time": 1.902416706085205,
      "step": 94624
    },
    {
      "epoch": 1.96685791015625e-05,
      "step": 94624,
      "training_step_time": 7.773014068603516
    },
    {
      "epoch": 1.9669532775878906e-05,
      "model_forward_time": 1.9022748470306396,
      "step": 94625
    },
    {
      "epoch": 1.9669532775878906e-05,
      "step": 94625,
      "training_step_time": 7.772187232971191
    },
    {
      "epoch": 1.9670486450195314e-05,
      "model_forward_time": 1.8910911083221436,
      "step": 94626
    },
    {
      "epoch": 1.9670486450195314e-05,
      "step": 94626,
      "training_step_time": 8.69145154953003
    },
    {
      "epoch": 1.9671440124511718e-05,
      "model_forward_time": 2.536985158920288,
      "step": 94627
    },
    {
      "epoch": 1.9671440124511718e-05,
      "step": 94627,
      "training_step_time": 8.548467636108398
    },
    {
      "epoch": 1.9672393798828125e-05,
      "model_forward_time": 1.913072109222412,
      "step": 94628
    },
    {
      "epoch": 1.9672393798828125e-05,
      "step": 94628,
      "training_step_time": 10.778002738952637
    },
    {
      "epoch": 1.9673347473144533e-05,
      "model_forward_time": 1.8904120922088623,
      "step": 94629
    },
    {
      "epoch": 1.9673347473144533e-05,
      "step": 94629,
      "training_step_time": 7.872995615005493
    },
    {
      "epoch": 1.9674301147460937e-05,
      "grad_norm": 0.07813603430986404,
      "learning_rate": 7.866124418695942e-08,
      "loss": 0.0544,
      "step": 94630
    },
    {
      "epoch": 1.9674301147460937e-05,
      "model_forward_time": 1.895801067352295,
      "step": 94630
    },
    {
      "dynamics_loss_avg": 0.04513002512976527,
      "epoch": 1.9674301147460937e-05,
      "step": 94630
    },
    {
      "action_loss_avg": 0.0009406317753018811,
      "epoch": 1.9674301147460937e-05,
      "step": 94630
    },
    {
      "epoch": 1.9674301147460937e-05,
      "step": 94630,
      "training_step_time": 7.833880662918091
    },
    {
      "epoch": 1.9675254821777344e-05,
      "model_forward_time": 1.8909912109375,
      "step": 94631
    },
    {
      "epoch": 1.9675254821777344e-05,
      "step": 94631,
      "training_step_time": 8.557307958602905
    },
    {
      "epoch": 1.967620849609375e-05,
      "model_forward_time": 1.8998968601226807,
      "step": 94632
    },
    {
      "epoch": 1.967620849609375e-05,
      "step": 94632,
      "training_step_time": 7.373455047607422
    },
    {
      "epoch": 1.9677162170410155e-05,
      "model_forward_time": 1.904606819152832,
      "step": 94633
    },
    {
      "epoch": 1.9677162170410155e-05,
      "step": 94633,
      "training_step_time": 7.613250017166138
    },
    {
      "epoch": 1.9678115844726563e-05,
      "model_forward_time": 1.8898472785949707,
      "step": 94634
    },
    {
      "epoch": 1.9678115844726563e-05,
      "step": 94634,
      "training_step_time": 7.806371450424194
    },
    {
      "epoch": 1.967906951904297e-05,
      "model_forward_time": 1.8930776119232178,
      "step": 94635
    },
    {
      "epoch": 1.967906951904297e-05,
      "step": 94635,
      "training_step_time": 8.345648765563965
    },
    {
      "epoch": 1.9680023193359374e-05,
      "model_forward_time": 2.514254331588745,
      "step": 94636
    },
    {
      "epoch": 1.9680023193359374e-05,
      "step": 94636,
      "training_step_time": 8.87959909439087
    },
    {
      "epoch": 1.968097686767578e-05,
      "model_forward_time": 1.9059200286865234,
      "step": 94637
    },
    {
      "epoch": 1.968097686767578e-05,
      "step": 94637,
      "training_step_time": 8.731078386306763
    },
    {
      "epoch": 1.968193054199219e-05,
      "model_forward_time": 1.9033422470092773,
      "step": 94638
    },
    {
      "epoch": 1.968193054199219e-05,
      "step": 94638,
      "training_step_time": 8.38977575302124
    },
    {
      "epoch": 1.9682884216308593e-05,
      "model_forward_time": 1.8926606178283691,
      "step": 94639
    },
    {
      "epoch": 1.9682884216308593e-05,
      "step": 94639,
      "training_step_time": 7.793720960617065
    },
    {
      "epoch": 1.9683837890625e-05,
      "grad_norm": 0.08220502734184265,
      "learning_rate": 7.83693727577095e-08,
      "loss": 0.0551,
      "step": 94640
    },
    {
      "epoch": 1.9683837890625e-05,
      "model_forward_time": 1.8905363082885742,
      "step": 94640
    },
    {
      "dynamics_loss_avg": 0.05457985531538725,
      "epoch": 1.9683837890625e-05,
      "step": 94640
    },
    {
      "action_loss_avg": 0.00103476625517942,
      "epoch": 1.9683837890625e-05,
      "step": 94640
    },
    {
      "epoch": 1.9683837890625e-05,
      "step": 94640,
      "training_step_time": 7.71410870552063
    },
    {
      "epoch": 1.9684791564941405e-05,
      "model_forward_time": 1.897540807723999,
      "step": 94641
    },
    {
      "epoch": 1.9684791564941405e-05,
      "step": 94641,
      "training_step_time": 7.515326738357544
    },
    {
      "epoch": 1.9685745239257812e-05,
      "model_forward_time": 1.9001545906066895,
      "step": 94642
    },
    {
      "epoch": 1.9685745239257812e-05,
      "step": 94642,
      "training_step_time": 7.571013689041138
    },
    {
      "epoch": 1.968669891357422e-05,
      "model_forward_time": 1.905259609222412,
      "step": 94643
    },
    {
      "epoch": 1.968669891357422e-05,
      "step": 94643,
      "training_step_time": 7.662657260894775
    },
    {
      "epoch": 1.9687652587890623e-05,
      "model_forward_time": 1.8924784660339355,
      "step": 94644
    },
    {
      "epoch": 1.9687652587890623e-05,
      "step": 94644,
      "training_step_time": 8.306405305862427
    },
    {
      "epoch": 1.968860626220703e-05,
      "model_forward_time": 1.9057118892669678,
      "step": 94645
    },
    {
      "epoch": 1.968860626220703e-05,
      "step": 94645,
      "training_step_time": 8.780109405517578
    },
    {
      "epoch": 1.9689559936523438e-05,
      "model_forward_time": 2.5141971111297607,
      "step": 94646
    },
    {
      "epoch": 1.9689559936523438e-05,
      "step": 94646,
      "training_step_time": 7.8605263233184814
    },
    {
      "epoch": 1.9690513610839842e-05,
      "model_forward_time": 1.8873560428619385,
      "step": 94647
    },
    {
      "epoch": 1.9690513610839842e-05,
      "step": 94647,
      "training_step_time": 8.576678037643433
    },
    {
      "epoch": 1.969146728515625e-05,
      "model_forward_time": 1.8981273174285889,
      "step": 94648
    },
    {
      "epoch": 1.969146728515625e-05,
      "step": 94648,
      "training_step_time": 7.7575109004974365
    },
    {
      "epoch": 1.9692420959472657e-05,
      "model_forward_time": 1.8966939449310303,
      "step": 94649
    },
    {
      "epoch": 1.9692420959472657e-05,
      "step": 94649,
      "training_step_time": 7.609573125839233
    },
    {
      "epoch": 1.969337463378906e-05,
      "grad_norm": 0.07802902907133102,
      "learning_rate": 7.807803955058535e-08,
      "loss": 0.0536,
      "step": 94650
    },
    {
      "epoch": 1.969337463378906e-05,
      "model_forward_time": 1.89729905128479,
      "step": 94650
    },
    {
      "dynamics_loss_avg": 0.06269892631098628,
      "epoch": 1.969337463378906e-05,
      "step": 94650
    },
    {
      "action_loss_avg": 0.00129951617855113,
      "epoch": 1.969337463378906e-05,
      "step": 94650
    },
    {
      "epoch": 1.969337463378906e-05,
      "step": 94650,
      "training_step_time": 7.393890619277954
    },
    {
      "epoch": 1.969432830810547e-05,
      "model_forward_time": 1.9038517475128174,
      "step": 94651
    },
    {
      "epoch": 1.969432830810547e-05,
      "step": 94651,
      "training_step_time": 7.141900300979614
    },
    {
      "epoch": 1.9695281982421876e-05,
      "model_forward_time": 1.904651165008545,
      "step": 94652
    },
    {
      "epoch": 1.9695281982421876e-05,
      "step": 94652,
      "training_step_time": 7.757443428039551
    },
    {
      "epoch": 1.969623565673828e-05,
      "model_forward_time": 1.898050308227539,
      "step": 94653
    },
    {
      "epoch": 1.969623565673828e-05,
      "step": 94653,
      "training_step_time": 7.488256454467773
    },
    {
      "epoch": 1.9697189331054687e-05,
      "model_forward_time": 1.9104280471801758,
      "step": 94654
    },
    {
      "epoch": 1.9697189331054687e-05,
      "step": 94654,
      "training_step_time": 7.633872985839844
    },
    {
      "epoch": 1.9698143005371095e-05,
      "model_forward_time": 2.5452356338500977,
      "step": 94655
    },
    {
      "epoch": 1.9698143005371095e-05,
      "step": 94655,
      "training_step_time": 12.224046230316162
    },
    {
      "epoch": 1.96990966796875e-05,
      "model_forward_time": 1.9121859073638916,
      "step": 94656
    },
    {
      "epoch": 1.96990966796875e-05,
      "step": 94656,
      "training_step_time": 8.546698093414307
    },
    {
      "epoch": 1.9700050354003906e-05,
      "model_forward_time": 1.8891575336456299,
      "step": 94657
    },
    {
      "epoch": 1.9700050354003906e-05,
      "step": 94657,
      "training_step_time": 7.702723026275635
    },
    {
      "epoch": 1.9701004028320314e-05,
      "model_forward_time": 1.906679391860962,
      "step": 94658
    },
    {
      "epoch": 1.9701004028320314e-05,
      "step": 94658,
      "training_step_time": 7.55729866027832
    },
    {
      "epoch": 1.9701957702636718e-05,
      "model_forward_time": 1.907926321029663,
      "step": 94659
    },
    {
      "epoch": 1.9701957702636718e-05,
      "step": 94659,
      "training_step_time": 7.616432428359985
    },
    {
      "epoch": 1.9702911376953125e-05,
      "grad_norm": 0.08472573012113571,
      "learning_rate": 7.778724459744702e-08,
      "loss": 0.0555,
      "step": 94660
    },
    {
      "epoch": 1.9702911376953125e-05,
      "model_forward_time": 1.893528938293457,
      "step": 94660
    },
    {
      "dynamics_loss_avg": 0.039418551698327066,
      "epoch": 1.9702911376953125e-05,
      "step": 94660
    },
    {
      "action_loss_avg": 0.0008403786006965675,
      "epoch": 1.9702911376953125e-05,
      "step": 94660
    },
    {
      "epoch": 1.9702911376953125e-05,
      "step": 94660,
      "training_step_time": 7.1520819664001465
    },
    {
      "epoch": 1.9703865051269533e-05,
      "model_forward_time": 1.9002909660339355,
      "step": 94661
    },
    {
      "epoch": 1.9703865051269533e-05,
      "step": 94661,
      "training_step_time": 7.672491073608398
    },
    {
      "epoch": 1.9704818725585937e-05,
      "model_forward_time": 1.8874735832214355,
      "step": 94662
    },
    {
      "epoch": 1.9704818725585937e-05,
      "step": 94662,
      "training_step_time": 7.446110486984253
    },
    {
      "epoch": 1.9705772399902344e-05,
      "model_forward_time": 1.9086639881134033,
      "step": 94663
    },
    {
      "epoch": 1.9705772399902344e-05,
      "step": 94663,
      "training_step_time": 7.7127556800842285
    },
    {
      "epoch": 1.970672607421875e-05,
      "model_forward_time": 2.4834048748016357,
      "step": 94664
    },
    {
      "epoch": 1.970672607421875e-05,
      "step": 94664,
      "training_step_time": 9.484103441238403
    },
    {
      "epoch": 1.9707679748535155e-05,
      "model_forward_time": 1.902190923690796,
      "step": 94665
    },
    {
      "epoch": 1.9707679748535155e-05,
      "step": 94665,
      "training_step_time": 7.798614740371704
    },
    {
      "epoch": 1.9708633422851563e-05,
      "model_forward_time": 1.8988659381866455,
      "step": 94666
    },
    {
      "epoch": 1.9708633422851563e-05,
      "step": 94666,
      "training_step_time": 7.783698081970215
    },
    {
      "epoch": 1.970958709716797e-05,
      "model_forward_time": 1.8911621570587158,
      "step": 94667
    },
    {
      "epoch": 1.970958709716797e-05,
      "step": 94667,
      "training_step_time": 8.619572877883911
    },
    {
      "epoch": 1.9710540771484374e-05,
      "model_forward_time": 1.898261308670044,
      "step": 94668
    },
    {
      "epoch": 1.9710540771484374e-05,
      "step": 94668,
      "training_step_time": 7.611177921295166
    },
    {
      "epoch": 1.9711494445800782e-05,
      "model_forward_time": 1.90413236618042,
      "step": 94669
    },
    {
      "epoch": 1.9711494445800782e-05,
      "step": 94669,
      "training_step_time": 7.171438455581665
    },
    {
      "epoch": 1.971244812011719e-05,
      "grad_norm": 0.15251632034778595,
      "learning_rate": 7.749698793009464e-08,
      "loss": 0.0549,
      "step": 94670
    },
    {
      "epoch": 1.971244812011719e-05,
      "model_forward_time": 1.8989698886871338,
      "step": 94670
    },
    {
      "dynamics_loss_avg": 0.06913973335176707,
      "epoch": 1.971244812011719e-05,
      "step": 94670
    },
    {
      "action_loss_avg": 0.0022759082727134227,
      "epoch": 1.971244812011719e-05,
      "step": 94670
    },
    {
      "epoch": 1.971244812011719e-05,
      "step": 94670,
      "training_step_time": 7.725223541259766
    },
    {
      "epoch": 1.9713401794433593e-05,
      "model_forward_time": 1.898188829421997,
      "step": 94671
    },
    {
      "epoch": 1.9713401794433593e-05,
      "step": 94671,
      "training_step_time": 8.015953302383423
    },
    {
      "epoch": 1.971435546875e-05,
      "model_forward_time": 1.89483642578125,
      "step": 94672
    },
    {
      "epoch": 1.971435546875e-05,
      "step": 94672,
      "training_step_time": 7.447347640991211
    },
    {
      "epoch": 1.9715309143066405e-05,
      "model_forward_time": 2.5518033504486084,
      "step": 94673
    },
    {
      "epoch": 1.9715309143066405e-05,
      "step": 94673,
      "training_step_time": 9.527368783950806
    },
    {
      "epoch": 1.9716262817382812e-05,
      "model_forward_time": 1.899306297302246,
      "step": 94674
    },
    {
      "epoch": 1.9716262817382812e-05,
      "step": 94674,
      "training_step_time": 8.817087888717651
    },
    {
      "epoch": 1.971721649169922e-05,
      "model_forward_time": 1.899254560470581,
      "step": 94675
    },
    {
      "epoch": 1.971721649169922e-05,
      "step": 94675,
      "training_step_time": 8.539021968841553
    },
    {
      "epoch": 1.9718170166015623e-05,
      "model_forward_time": 1.8980984687805176,
      "step": 94676
    },
    {
      "epoch": 1.9718170166015623e-05,
      "step": 94676,
      "training_step_time": 7.649953603744507
    },
    {
      "epoch": 1.971912384033203e-05,
      "model_forward_time": 1.9028010368347168,
      "step": 94677
    },
    {
      "epoch": 1.971912384033203e-05,
      "step": 94677,
      "training_step_time": 7.651411533355713
    },
    {
      "epoch": 1.9720077514648438e-05,
      "model_forward_time": 1.8982696533203125,
      "step": 94678
    },
    {
      "epoch": 1.9720077514648438e-05,
      "step": 94678,
      "training_step_time": 7.28618597984314
    },
    {
      "epoch": 1.9721031188964842e-05,
      "model_forward_time": 1.8984432220458984,
      "step": 94679
    },
    {
      "epoch": 1.9721031188964842e-05,
      "step": 94679,
      "training_step_time": 7.623347759246826
    },
    {
      "epoch": 1.972198486328125e-05,
      "grad_norm": 0.08845508098602295,
      "learning_rate": 7.72072695802717e-08,
      "loss": 0.0557,
      "step": 94680
    },
    {
      "epoch": 1.972198486328125e-05,
      "model_forward_time": 1.8942115306854248,
      "step": 94680
    },
    {
      "dynamics_loss_avg": 0.05636013615876436,
      "epoch": 1.972198486328125e-05,
      "step": 94680
    },
    {
      "action_loss_avg": 0.0011700092407409101,
      "epoch": 1.972198486328125e-05,
      "step": 94680
    },
    {
      "epoch": 1.972198486328125e-05,
      "step": 94680,
      "training_step_time": 7.617918252944946
    },
    {
      "epoch": 1.9722938537597657e-05,
      "model_forward_time": 1.9035353660583496,
      "step": 94681
    },
    {
      "epoch": 1.9722938537597657e-05,
      "step": 94681,
      "training_step_time": 7.619031667709351
    },
    {
      "epoch": 1.972389221191406e-05,
      "model_forward_time": 2.5421316623687744,
      "step": 94682
    },
    {
      "epoch": 1.972389221191406e-05,
      "step": 94682,
      "training_step_time": 8.547412633895874
    },
    {
      "epoch": 1.972484588623047e-05,
      "model_forward_time": 1.9048268795013428,
      "step": 94683
    },
    {
      "epoch": 1.972484588623047e-05,
      "step": 94683,
      "training_step_time": 11.41269302368164
    },
    {
      "epoch": 1.9725799560546876e-05,
      "model_forward_time": 1.887437105178833,
      "step": 94684
    },
    {
      "epoch": 1.9725799560546876e-05,
      "step": 94684,
      "training_step_time": 8.660094022750854
    },
    {
      "epoch": 1.972675323486328e-05,
      "model_forward_time": 1.8956000804901123,
      "step": 94685
    },
    {
      "epoch": 1.972675323486328e-05,
      "step": 94685,
      "training_step_time": 7.602444887161255
    },
    {
      "epoch": 1.9727706909179687e-05,
      "model_forward_time": 1.9018354415893555,
      "step": 94686
    },
    {
      "epoch": 1.9727706909179687e-05,
      "step": 94686,
      "training_step_time": 7.690284252166748
    },
    {
      "epoch": 1.9728660583496095e-05,
      "model_forward_time": 1.9080493450164795,
      "step": 94687
    },
    {
      "epoch": 1.9728660583496095e-05,
      "step": 94687,
      "training_step_time": 7.301877498626709
    },
    {
      "epoch": 1.97296142578125e-05,
      "model_forward_time": 1.8999443054199219,
      "step": 94688
    },
    {
      "epoch": 1.97296142578125e-05,
      "step": 94688,
      "training_step_time": 7.629894018173218
    },
    {
      "epoch": 1.9730567932128906e-05,
      "model_forward_time": 1.9105455875396729,
      "step": 94689
    },
    {
      "epoch": 1.9730567932128906e-05,
      "step": 94689,
      "training_step_time": 7.677940368652344
    },
    {
      "epoch": 1.9731521606445314e-05,
      "grad_norm": 0.0747687965631485,
      "learning_rate": 7.691808957965951e-08,
      "loss": 0.0549,
      "step": 94690
    },
    {
      "epoch": 1.9731521606445314e-05,
      "model_forward_time": 1.8960583209991455,
      "step": 94690
    },
    {
      "dynamics_loss_avg": 0.045130039751529696,
      "epoch": 1.9731521606445314e-05,
      "step": 94690
    },
    {
      "action_loss_avg": 0.0008856243221089244,
      "epoch": 1.9731521606445314e-05,
      "step": 94690
    },
    {
      "epoch": 1.9731521606445314e-05,
      "step": 94690,
      "training_step_time": 7.594462871551514
    },
    {
      "epoch": 1.9732475280761718e-05,
      "model_forward_time": 1.9023854732513428,
      "step": 94691
    },
    {
      "epoch": 1.9732475280761718e-05,
      "step": 94691,
      "training_step_time": 9.708287954330444
    },
    {
      "epoch": 1.9733428955078125e-05,
      "model_forward_time": 1.9079275131225586,
      "step": 94692
    },
    {
      "epoch": 1.9733428955078125e-05,
      "step": 94692,
      "training_step_time": 9.014296770095825
    },
    {
      "epoch": 1.9734382629394533e-05,
      "model_forward_time": 1.5267937183380127,
      "step": 94693
    },
    {
      "epoch": 1.9734382629394533e-05,
      "step": 94693,
      "training_step_time": 7.864617347717285
    },
    {
      "epoch": 1.9735336303710937e-05,
      "model_forward_time": 1.8951208591461182,
      "step": 94694
    },
    {
      "epoch": 1.9735336303710937e-05,
      "step": 94694,
      "training_step_time": 7.714921474456787
    },
    {
      "epoch": 1.9736289978027344e-05,
      "model_forward_time": 1.8947720527648926,
      "step": 94695
    },
    {
      "epoch": 1.9736289978027344e-05,
      "step": 94695,
      "training_step_time": 7.711873292922974
    },
    {
      "epoch": 1.973724365234375e-05,
      "model_forward_time": 1.8944706916809082,
      "step": 94696
    },
    {
      "epoch": 1.973724365234375e-05,
      "step": 94696,
      "training_step_time": 7.295361280441284
    },
    {
      "epoch": 1.9738197326660155e-05,
      "model_forward_time": 1.902228832244873,
      "step": 94697
    },
    {
      "epoch": 1.9738197326660155e-05,
      "step": 94697,
      "training_step_time": 7.670357942581177
    },
    {
      "epoch": 1.9739151000976563e-05,
      "model_forward_time": 1.903071641921997,
      "step": 94698
    },
    {
      "epoch": 1.9739151000976563e-05,
      "step": 94698,
      "training_step_time": 7.526068449020386
    },
    {
      "epoch": 1.974010467529297e-05,
      "model_forward_time": 1.9062533378601074,
      "step": 94699
    },
    {
      "epoch": 1.974010467529297e-05,
      "step": 94699,
      "training_step_time": 8.038532972335815
    },
    {
      "epoch": 1.9741058349609374e-05,
      "grad_norm": 0.08798207342624664,
      "learning_rate": 7.662944795988337e-08,
      "loss": 0.0544,
      "step": 94700
    },
    {
      "epoch": 1.9741058349609374e-05,
      "model_forward_time": 1.8954493999481201,
      "step": 94700
    },
    {
      "dynamics_loss_avg": 0.04370514359325171,
      "epoch": 1.9741058349609374e-05,
      "step": 94700
    },
    {
      "action_loss_avg": 0.0016910363541683183,
      "epoch": 1.9741058349609374e-05,
      "step": 94700
    },
    {
      "epoch": 1.9741058349609374e-05,
      "step": 94700,
      "training_step_time": 7.677821636199951
    },
    {
      "epoch": 1.9742012023925782e-05,
      "model_forward_time": 2.570255994796753,
      "step": 94701
    },
    {
      "epoch": 1.9742012023925782e-05,
      "step": 94701,
      "training_step_time": 9.425833463668823
    },
    {
      "epoch": 1.974296569824219e-05,
      "model_forward_time": 1.9149739742279053,
      "step": 94702
    },
    {
      "epoch": 1.974296569824219e-05,
      "step": 94702,
      "training_step_time": 9.96453595161438
    },
    {
      "epoch": 1.9743919372558593e-05,
      "model_forward_time": 1.8941349983215332,
      "step": 94703
    },
    {
      "epoch": 1.9743919372558593e-05,
      "step": 94703,
      "training_step_time": 8.22111177444458
    },
    {
      "epoch": 1.9744873046875e-05,
      "model_forward_time": 1.8979644775390625,
      "step": 94704
    },
    {
      "epoch": 1.9744873046875e-05,
      "step": 94704,
      "training_step_time": 7.781986236572266
    },
    {
      "epoch": 1.9745826721191405e-05,
      "model_forward_time": 1.8975577354431152,
      "step": 94705
    },
    {
      "epoch": 1.9745826721191405e-05,
      "step": 94705,
      "training_step_time": 7.5566322803497314
    },
    {
      "epoch": 1.9746780395507812e-05,
      "model_forward_time": 1.8970162868499756,
      "step": 94706
    },
    {
      "epoch": 1.9746780395507812e-05,
      "step": 94706,
      "training_step_time": 7.626722097396851
    },
    {
      "epoch": 1.974773406982422e-05,
      "model_forward_time": 1.9042749404907227,
      "step": 94707
    },
    {
      "epoch": 1.974773406982422e-05,
      "step": 94707,
      "training_step_time": 7.52720046043396
    },
    {
      "epoch": 1.9748687744140623e-05,
      "model_forward_time": 1.8990020751953125,
      "step": 94708
    },
    {
      "epoch": 1.9748687744140623e-05,
      "step": 94708,
      "training_step_time": 7.51029372215271
    },
    {
      "epoch": 1.974964141845703e-05,
      "model_forward_time": 1.905665397644043,
      "step": 94709
    },
    {
      "epoch": 1.974964141845703e-05,
      "step": 94709,
      "training_step_time": 8.736610651016235
    },
    {
      "epoch": 1.975059509277344e-05,
      "grad_norm": 0.08132237941026688,
      "learning_rate": 7.634134475250854e-08,
      "loss": 0.0555,
      "step": 94710
    },
    {
      "epoch": 1.975059509277344e-05,
      "model_forward_time": 2.546438694000244,
      "step": 94710
    },
    {
      "dynamics_loss_avg": 0.04579276209697127,
      "epoch": 1.975059509277344e-05,
      "step": 94710
    },
    {
      "action_loss_avg": 0.001569416665006429,
      "epoch": 1.975059509277344e-05,
      "step": 94710
    },
    {
      "epoch": 1.975059509277344e-05,
      "step": 94710,
      "training_step_time": 9.501275777816772
    },
    {
      "epoch": 1.9751548767089842e-05,
      "model_forward_time": 1.90496826171875,
      "step": 94711
    },
    {
      "epoch": 1.9751548767089842e-05,
      "step": 94711,
      "training_step_time": 8.52698016166687
    },
    {
      "epoch": 1.975250244140625e-05,
      "model_forward_time": 1.9019722938537598,
      "step": 94712
    },
    {
      "epoch": 1.975250244140625e-05,
      "step": 94712,
      "training_step_time": 7.753071069717407
    },
    {
      "epoch": 1.9753456115722657e-05,
      "model_forward_time": 1.900773048400879,
      "step": 94713
    },
    {
      "epoch": 1.9753456115722657e-05,
      "step": 94713,
      "training_step_time": 7.656920909881592
    },
    {
      "epoch": 1.975440979003906e-05,
      "model_forward_time": 1.9056990146636963,
      "step": 94714
    },
    {
      "epoch": 1.975440979003906e-05,
      "step": 94714,
      "training_step_time": 7.592841863632202
    },
    {
      "epoch": 1.975536346435547e-05,
      "model_forward_time": 1.8998148441314697,
      "step": 94715
    },
    {
      "epoch": 1.975536346435547e-05,
      "step": 94715,
      "training_step_time": 7.592708587646484
    },
    {
      "epoch": 1.9756317138671876e-05,
      "model_forward_time": 1.9050788879394531,
      "step": 94716
    },
    {
      "epoch": 1.9756317138671876e-05,
      "step": 94716,
      "training_step_time": 8.078784465789795
    },
    {
      "epoch": 1.975727081298828e-05,
      "model_forward_time": 1.8993995189666748,
      "step": 94717
    },
    {
      "epoch": 1.975727081298828e-05,
      "step": 94717,
      "training_step_time": 7.414613246917725
    },
    {
      "epoch": 1.9758224487304687e-05,
      "model_forward_time": 1.899852991104126,
      "step": 94718
    },
    {
      "epoch": 1.9758224487304687e-05,
      "step": 94718,
      "training_step_time": 8.769517660140991
    },
    {
      "epoch": 1.9759178161621095e-05,
      "model_forward_time": 2.544511318206787,
      "step": 94719
    },
    {
      "epoch": 1.9759178161621095e-05,
      "step": 94719,
      "training_step_time": 11.682180166244507
    },
    {
      "epoch": 1.97601318359375e-05,
      "grad_norm": 0.06918511539697647,
      "learning_rate": 7.605377998904262e-08,
      "loss": 0.0551,
      "step": 94720
    },
    {
      "epoch": 1.97601318359375e-05,
      "model_forward_time": 1.913055419921875,
      "step": 94720
    },
    {
      "dynamics_loss_avg": 0.04441573992371559,
      "epoch": 1.97601318359375e-05,
      "step": 94720
    },
    {
      "action_loss_avg": 0.001098156679654494,
      "epoch": 1.97601318359375e-05,
      "step": 94720
    },
    {
      "epoch": 1.97601318359375e-05,
      "step": 94720,
      "training_step_time": 9.488391876220703
    },
    {
      "epoch": 1.9761085510253906e-05,
      "model_forward_time": 1.896148920059204,
      "step": 94721
    },
    {
      "epoch": 1.9761085510253906e-05,
      "step": 94721,
      "training_step_time": 9.081568002700806
    },
    {
      "epoch": 1.9762039184570314e-05,
      "model_forward_time": 1.9001257419586182,
      "step": 94722
    },
    {
      "epoch": 1.9762039184570314e-05,
      "step": 94722,
      "training_step_time": 7.656709432601929
    },
    {
      "epoch": 1.9762992858886718e-05,
      "model_forward_time": 1.8961734771728516,
      "step": 94723
    },
    {
      "epoch": 1.9762992858886718e-05,
      "step": 94723,
      "training_step_time": 7.496118545532227
    },
    {
      "epoch": 1.9763946533203125e-05,
      "model_forward_time": 1.8899645805358887,
      "step": 94724
    },
    {
      "epoch": 1.9763946533203125e-05,
      "step": 94724,
      "training_step_time": 7.650051593780518
    },
    {
      "epoch": 1.9764900207519533e-05,
      "model_forward_time": 1.9022698402404785,
      "step": 94725
    },
    {
      "epoch": 1.9764900207519533e-05,
      "step": 94725,
      "training_step_time": 6.759133577346802
    },
    {
      "epoch": 1.9765853881835937e-05,
      "model_forward_time": 1.9072906970977783,
      "step": 94726
    },
    {
      "epoch": 1.9765853881835937e-05,
      "step": 94726,
      "training_step_time": 7.5516462326049805
    },
    {
      "epoch": 1.9766807556152344e-05,
      "model_forward_time": 1.894899606704712,
      "step": 94727
    },
    {
      "epoch": 1.9766807556152344e-05,
      "step": 94727,
      "training_step_time": 7.52145528793335
    },
    {
      "epoch": 1.976776123046875e-05,
      "model_forward_time": 2.512918472290039,
      "step": 94728
    },
    {
      "epoch": 1.976776123046875e-05,
      "step": 94728,
      "training_step_time": 8.30775237083435
    },
    {
      "epoch": 1.9768714904785155e-05,
      "model_forward_time": 1.901397705078125,
      "step": 94729
    },
    {
      "epoch": 1.9768714904785155e-05,
      "step": 94729,
      "training_step_time": 9.448515892028809
    },
    {
      "epoch": 1.9769668579101563e-05,
      "grad_norm": 0.09263937175273895,
      "learning_rate": 7.576675370093101e-08,
      "loss": 0.0532,
      "step": 94730
    },
    {
      "epoch": 1.9769668579101563e-05,
      "model_forward_time": 1.9004974365234375,
      "step": 94730
    },
    {
      "dynamics_loss_avg": 0.03239085804671049,
      "epoch": 1.9769668579101563e-05,
      "step": 94730
    },
    {
      "action_loss_avg": 0.0009147551842033863,
      "epoch": 1.9769668579101563e-05,
      "step": 94730
    },
    {
      "epoch": 1.9769668579101563e-05,
      "step": 94730,
      "training_step_time": 9.660184144973755
    },
    {
      "epoch": 1.977062225341797e-05,
      "model_forward_time": 1.913780689239502,
      "step": 94731
    },
    {
      "epoch": 1.977062225341797e-05,
      "step": 94731,
      "training_step_time": 7.584195852279663
    },
    {
      "epoch": 1.9771575927734374e-05,
      "model_forward_time": 1.898294448852539,
      "step": 94732
    },
    {
      "epoch": 1.9771575927734374e-05,
      "step": 94732,
      "training_step_time": 7.588457822799683
    },
    {
      "epoch": 1.9772529602050782e-05,
      "model_forward_time": 1.887448787689209,
      "step": 94733
    },
    {
      "epoch": 1.9772529602050782e-05,
      "step": 94733,
      "training_step_time": 8.268716812133789
    },
    {
      "epoch": 1.977348327636719e-05,
      "model_forward_time": 1.8973422050476074,
      "step": 94734
    },
    {
      "epoch": 1.977348327636719e-05,
      "step": 94734,
      "training_step_time": 7.0669779777526855
    },
    {
      "epoch": 1.9774436950683593e-05,
      "model_forward_time": 1.9082415103912354,
      "step": 94735
    },
    {
      "epoch": 1.9774436950683593e-05,
      "step": 94735,
      "training_step_time": 7.549914360046387
    },
    {
      "epoch": 1.9775390625e-05,
      "model_forward_time": 1.8937983512878418,
      "step": 94736
    },
    {
      "epoch": 1.9775390625e-05,
      "step": 94736,
      "training_step_time": 7.626626014709473
    },
    {
      "epoch": 1.9776344299316405e-05,
      "model_forward_time": 1.8928356170654297,
      "step": 94737
    },
    {
      "epoch": 1.9776344299316405e-05,
      "step": 94737,
      "training_step_time": 9.79074764251709
    },
    {
      "epoch": 1.9777297973632812e-05,
      "model_forward_time": 2.5282187461853027,
      "step": 94738
    },
    {
      "epoch": 1.9777297973632812e-05,
      "step": 94738,
      "training_step_time": 9.677876710891724
    },
    {
      "epoch": 1.977825164794922e-05,
      "model_forward_time": 1.9159266948699951,
      "step": 94739
    },
    {
      "epoch": 1.977825164794922e-05,
      "step": 94739,
      "training_step_time": 8.617710590362549
    },
    {
      "epoch": 1.9779205322265624e-05,
      "grad_norm": 0.07816722989082336,
      "learning_rate": 7.54802659195647e-08,
      "loss": 0.055,
      "step": 94740
    },
    {
      "epoch": 1.9779205322265624e-05,
      "model_forward_time": 1.8991482257843018,
      "step": 94740
    },
    {
      "dynamics_loss_avg": 0.05119930682703853,
      "epoch": 1.9779205322265624e-05,
      "step": 94740
    },
    {
      "action_loss_avg": 0.0011316535470541566,
      "epoch": 1.9779205322265624e-05,
      "step": 94740
    },
    {
      "epoch": 1.9779205322265624e-05,
      "step": 94740,
      "training_step_time": 8.808345794677734
    },
    {
      "epoch": 1.978015899658203e-05,
      "model_forward_time": 1.5394375324249268,
      "step": 94741
    },
    {
      "epoch": 1.978015899658203e-05,
      "step": 94741,
      "training_step_time": 7.613079071044922
    },
    {
      "epoch": 1.978111267089844e-05,
      "model_forward_time": 1.9136972427368164,
      "step": 94742
    },
    {
      "epoch": 1.978111267089844e-05,
      "step": 94742,
      "training_step_time": 7.663647174835205
    },
    {
      "epoch": 1.9782066345214842e-05,
      "model_forward_time": 1.9143767356872559,
      "step": 94743
    },
    {
      "epoch": 1.9782066345214842e-05,
      "step": 94743,
      "training_step_time": 7.060038805007935
    },
    {
      "epoch": 1.978302001953125e-05,
      "model_forward_time": 1.9101438522338867,
      "step": 94744
    },
    {
      "epoch": 1.978302001953125e-05,
      "step": 94744,
      "training_step_time": 7.177777528762817
    },
    {
      "epoch": 1.9783973693847657e-05,
      "model_forward_time": 1.9144129753112793,
      "step": 94745
    },
    {
      "epoch": 1.9783973693847657e-05,
      "step": 94745,
      "training_step_time": 7.5419347286224365
    },
    {
      "epoch": 1.978492736816406e-05,
      "model_forward_time": 1.9251093864440918,
      "step": 94746
    },
    {
      "epoch": 1.978492736816406e-05,
      "step": 94746,
      "training_step_time": 7.721562623977661
    },
    {
      "epoch": 1.978588104248047e-05,
      "model_forward_time": 2.523789644241333,
      "step": 94747
    },
    {
      "epoch": 1.978588104248047e-05,
      "step": 94747,
      "training_step_time": 9.412875652313232
    },
    {
      "epoch": 1.9786834716796876e-05,
      "model_forward_time": 1.9053585529327393,
      "step": 94748
    },
    {
      "epoch": 1.9786834716796876e-05,
      "step": 94748,
      "training_step_time": 8.885934829711914
    },
    {
      "epoch": 1.978778839111328e-05,
      "model_forward_time": 1.871873140335083,
      "step": 94749
    },
    {
      "epoch": 1.978778839111328e-05,
      "step": 94749,
      "training_step_time": 8.538265705108643
    },
    {
      "epoch": 1.9788742065429688e-05,
      "grad_norm": 0.08043579757213593,
      "learning_rate": 7.519431667627252e-08,
      "loss": 0.055,
      "step": 94750
    },
    {
      "epoch": 1.9788742065429688e-05,
      "model_forward_time": 1.8946871757507324,
      "step": 94750
    },
    {
      "dynamics_loss_avg": 0.051273906603455544,
      "epoch": 1.9788742065429688e-05,
      "step": 94750
    },
    {
      "action_loss_avg": 0.0008520470699295401,
      "epoch": 1.9788742065429688e-05,
      "step": 94750
    },
    {
      "epoch": 1.9788742065429688e-05,
      "step": 94750,
      "training_step_time": 7.455227851867676
    },
    {
      "epoch": 1.9789695739746095e-05,
      "model_forward_time": 1.9164009094238281,
      "step": 94751
    },
    {
      "epoch": 1.9789695739746095e-05,
      "step": 94751,
      "training_step_time": 7.627984046936035
    },
    {
      "epoch": 1.97906494140625e-05,
      "model_forward_time": 1.9138412475585938,
      "step": 94752
    },
    {
      "epoch": 1.97906494140625e-05,
      "step": 94752,
      "training_step_time": 7.089498996734619
    },
    {
      "epoch": 1.9791603088378906e-05,
      "model_forward_time": 1.910994052886963,
      "step": 94753
    },
    {
      "epoch": 1.9791603088378906e-05,
      "step": 94753,
      "training_step_time": 7.765762567520142
    },
    {
      "epoch": 1.9792556762695314e-05,
      "model_forward_time": 1.9123809337615967,
      "step": 94754
    },
    {
      "epoch": 1.9792556762695314e-05,
      "step": 94754,
      "training_step_time": 7.494947671890259
    },
    {
      "epoch": 1.9793510437011718e-05,
      "model_forward_time": 1.904306173324585,
      "step": 94755
    },
    {
      "epoch": 1.9793510437011718e-05,
      "step": 94755,
      "training_step_time": 7.517091512680054
    },
    {
      "epoch": 1.9794464111328125e-05,
      "model_forward_time": 2.5746212005615234,
      "step": 94756
    },
    {
      "epoch": 1.9794464111328125e-05,
      "step": 94756,
      "training_step_time": 7.7697882652282715
    },
    {
      "epoch": 1.9795417785644533e-05,
      "model_forward_time": 1.9342992305755615,
      "step": 94757
    },
    {
      "epoch": 1.9795417785644533e-05,
      "step": 94757,
      "training_step_time": 7.589840650558472
    },
    {
      "epoch": 1.9796371459960937e-05,
      "model_forward_time": 1.8760278224945068,
      "step": 94758
    },
    {
      "epoch": 1.9796371459960937e-05,
      "step": 94758,
      "training_step_time": 8.58415937423706
    },
    {
      "epoch": 1.9797325134277344e-05,
      "model_forward_time": 1.9182956218719482,
      "step": 94759
    },
    {
      "epoch": 1.9797325134277344e-05,
      "step": 94759,
      "training_step_time": 8.278687238693237
    },
    {
      "epoch": 1.979827880859375e-05,
      "grad_norm": 0.079510897397995,
      "learning_rate": 7.490890600232503e-08,
      "loss": 0.0544,
      "step": 94760
    },
    {
      "epoch": 1.979827880859375e-05,
      "model_forward_time": 1.8986473083496094,
      "step": 94760
    },
    {
      "dynamics_loss_avg": 0.05329826232045889,
      "epoch": 1.979827880859375e-05,
      "step": 94760
    },
    {
      "action_loss_avg": 0.001015893477597274,
      "epoch": 1.979827880859375e-05,
      "step": 94760
    },
    {
      "epoch": 1.979827880859375e-05,
      "step": 94760,
      "training_step_time": 9.020097494125366
    },
    {
      "epoch": 1.9799232482910156e-05,
      "model_forward_time": 1.9024674892425537,
      "step": 94761
    },
    {
      "epoch": 1.9799232482910156e-05,
      "step": 94761,
      "training_step_time": 7.0428690910339355
    },
    {
      "epoch": 1.9800186157226563e-05,
      "model_forward_time": 1.9124233722686768,
      "step": 94762
    },
    {
      "epoch": 1.9800186157226563e-05,
      "step": 94762,
      "training_step_time": 7.080598592758179
    },
    {
      "epoch": 1.980113983154297e-05,
      "model_forward_time": 1.8683571815490723,
      "step": 94763
    },
    {
      "epoch": 1.980113983154297e-05,
      "step": 94763,
      "training_step_time": 7.506565809249878
    },
    {
      "epoch": 1.9802093505859374e-05,
      "model_forward_time": 1.8988454341888428,
      "step": 94764
    },
    {
      "epoch": 1.9802093505859374e-05,
      "step": 94764,
      "training_step_time": 7.607147216796875
    },
    {
      "epoch": 1.9803047180175782e-05,
      "model_forward_time": 2.5368399620056152,
      "step": 94765
    },
    {
      "epoch": 1.9803047180175782e-05,
      "step": 94765,
      "training_step_time": 9.371778964996338
    },
    {
      "epoch": 1.9804000854492186e-05,
      "model_forward_time": 1.8982536792755127,
      "step": 94766
    },
    {
      "epoch": 1.9804000854492186e-05,
      "step": 94766,
      "training_step_time": 9.629940748214722
    },
    {
      "epoch": 1.9804954528808593e-05,
      "model_forward_time": 1.8981609344482422,
      "step": 94767
    },
    {
      "epoch": 1.9804954528808593e-05,
      "step": 94767,
      "training_step_time": 9.614251613616943
    },
    {
      "epoch": 1.9805908203125e-05,
      "model_forward_time": 1.902263879776001,
      "step": 94768
    },
    {
      "epoch": 1.9805908203125e-05,
      "step": 94768,
      "training_step_time": 8.37903642654419
    },
    {
      "epoch": 1.9806861877441405e-05,
      "model_forward_time": 1.895660638809204,
      "step": 94769
    },
    {
      "epoch": 1.9806861877441405e-05,
      "step": 94769,
      "training_step_time": 7.470347166061401
    },
    {
      "epoch": 1.9807815551757812e-05,
      "grad_norm": 0.06172758340835571,
      "learning_rate": 7.462403392893502e-08,
      "loss": 0.0533,
      "step": 94770
    },
    {
      "epoch": 1.9807815551757812e-05,
      "model_forward_time": 1.9016258716583252,
      "step": 94770
    },
    {
      "dynamics_loss_avg": 0.047160374373197554,
      "epoch": 1.9807815551757812e-05,
      "step": 94770
    },
    {
      "action_loss_avg": 0.0015135405148612336,
      "epoch": 1.9807815551757812e-05,
      "step": 94770
    },
    {
      "epoch": 1.9807815551757812e-05,
      "step": 94770,
      "training_step_time": 7.634631156921387
    },
    {
      "epoch": 1.980876922607422e-05,
      "model_forward_time": 1.9095137119293213,
      "step": 94771
    },
    {
      "epoch": 1.980876922607422e-05,
      "step": 94771,
      "training_step_time": 7.063161611557007
    },
    {
      "epoch": 1.9809722900390624e-05,
      "model_forward_time": 1.8954439163208008,
      "step": 94772
    },
    {
      "epoch": 1.9809722900390624e-05,
      "step": 94772,
      "training_step_time": 7.667185068130493
    },
    {
      "epoch": 1.981067657470703e-05,
      "model_forward_time": 1.9003541469573975,
      "step": 94773
    },
    {
      "epoch": 1.981067657470703e-05,
      "step": 94773,
      "training_step_time": 7.625965595245361
    },
    {
      "epoch": 1.981163024902344e-05,
      "model_forward_time": 1.537498950958252,
      "step": 94774
    },
    {
      "epoch": 1.981163024902344e-05,
      "step": 94774,
      "training_step_time": 7.734416246414185
    },
    {
      "epoch": 1.9812583923339842e-05,
      "model_forward_time": 2.541682004928589,
      "step": 94775
    },
    {
      "epoch": 1.9812583923339842e-05,
      "step": 94775,
      "training_step_time": 8.684478759765625
    },
    {
      "epoch": 1.981353759765625e-05,
      "model_forward_time": 1.164276123046875,
      "step": 94776
    },
    {
      "epoch": 1.981353759765625e-05,
      "step": 94776,
      "training_step_time": 8.214405059814453
    },
    {
      "epoch": 1.9814491271972657e-05,
      "model_forward_time": 1.9109470844268799,
      "step": 94777
    },
    {
      "epoch": 1.9814491271972657e-05,
      "step": 94777,
      "training_step_time": 7.692775726318359
    },
    {
      "epoch": 1.981544494628906e-05,
      "model_forward_time": 1.9138000011444092,
      "step": 94778
    },
    {
      "epoch": 1.981544494628906e-05,
      "step": 94778,
      "training_step_time": 7.662304639816284
    },
    {
      "epoch": 1.981639862060547e-05,
      "model_forward_time": 1.88639497756958,
      "step": 94779
    },
    {
      "epoch": 1.981639862060547e-05,
      "step": 94779,
      "training_step_time": 7.592402696609497
    },
    {
      "epoch": 1.9817352294921876e-05,
      "grad_norm": 0.09171013534069061,
      "learning_rate": 7.433970048725536e-08,
      "loss": 0.0547,
      "step": 94780
    },
    {
      "epoch": 1.9817352294921876e-05,
      "model_forward_time": 1.9143857955932617,
      "step": 94780
    },
    {
      "dynamics_loss_avg": 0.055174493417143824,
      "epoch": 1.9817352294921876e-05,
      "step": 94780
    },
    {
      "action_loss_avg": 0.0010539686161791905,
      "epoch": 1.9817352294921876e-05,
      "step": 94780
    },
    {
      "epoch": 1.9817352294921876e-05,
      "step": 94780,
      "training_step_time": 6.756457567214966
    },
    {
      "epoch": 1.981830596923828e-05,
      "model_forward_time": 1.906646966934204,
      "step": 94781
    },
    {
      "epoch": 1.981830596923828e-05,
      "step": 94781,
      "training_step_time": 8.415372371673584
    },
    {
      "epoch": 1.9819259643554688e-05,
      "model_forward_time": 1.8992199897766113,
      "step": 94782
    },
    {
      "epoch": 1.9819259643554688e-05,
      "step": 94782,
      "training_step_time": 7.586170196533203
    },
    {
      "epoch": 1.9820213317871095e-05,
      "model_forward_time": 1.9005413055419922,
      "step": 94783
    },
    {
      "epoch": 1.9820213317871095e-05,
      "step": 94783,
      "training_step_time": 7.855287551879883
    },
    {
      "epoch": 1.98211669921875e-05,
      "model_forward_time": 2.5653727054595947,
      "step": 94784
    },
    {
      "epoch": 1.98211669921875e-05,
      "step": 94784,
      "training_step_time": 8.255643367767334
    },
    {
      "epoch": 1.9822120666503906e-05,
      "model_forward_time": 1.9047744274139404,
      "step": 94785
    },
    {
      "epoch": 1.9822120666503906e-05,
      "step": 94785,
      "training_step_time": 9.344913482666016
    },
    {
      "epoch": 1.9823074340820314e-05,
      "model_forward_time": 1.9062576293945312,
      "step": 94786
    },
    {
      "epoch": 1.9823074340820314e-05,
      "step": 94786,
      "training_step_time": 8.606947422027588
    },
    {
      "epoch": 1.9824028015136718e-05,
      "model_forward_time": 1.9098749160766602,
      "step": 94787
    },
    {
      "epoch": 1.9824028015136718e-05,
      "step": 94787,
      "training_step_time": 9.074895143508911
    },
    {
      "epoch": 1.9824981689453125e-05,
      "model_forward_time": 1.9212119579315186,
      "step": 94788
    },
    {
      "epoch": 1.9824981689453125e-05,
      "step": 94788,
      "training_step_time": 7.554912328720093
    },
    {
      "epoch": 1.9825935363769533e-05,
      "model_forward_time": 1.9038734436035156,
      "step": 94789
    },
    {
      "epoch": 1.9825935363769533e-05,
      "step": 94789,
      "training_step_time": 7.603912830352783
    },
    {
      "epoch": 1.9826889038085937e-05,
      "grad_norm": 0.07602238655090332,
      "learning_rate": 7.405590570838061e-08,
      "loss": 0.0567,
      "step": 94790
    },
    {
      "epoch": 1.9826889038085937e-05,
      "model_forward_time": 1.9001944065093994,
      "step": 94790
    },
    {
      "dynamics_loss_avg": 0.07138108238577842,
      "epoch": 1.9826889038085937e-05,
      "step": 94790
    },
    {
      "action_loss_avg": 0.0012939789914526046,
      "epoch": 1.9826889038085937e-05,
      "step": 94790
    },
    {
      "epoch": 1.9826889038085937e-05,
      "step": 94790,
      "training_step_time": 7.6031787395477295
    },
    {
      "epoch": 1.9827842712402344e-05,
      "model_forward_time": 1.9018964767456055,
      "step": 94791
    },
    {
      "epoch": 1.9827842712402344e-05,
      "step": 94791,
      "training_step_time": 7.776971101760864
    },
    {
      "epoch": 1.982879638671875e-05,
      "model_forward_time": 1.9191105365753174,
      "step": 94792
    },
    {
      "epoch": 1.982879638671875e-05,
      "step": 94792,
      "training_step_time": 7.903562545776367
    },
    {
      "epoch": 1.9829750061035156e-05,
      "model_forward_time": 2.5410025119781494,
      "step": 94793
    },
    {
      "epoch": 1.9829750061035156e-05,
      "step": 94793,
      "training_step_time": 9.812278032302856
    },
    {
      "epoch": 1.9830703735351563e-05,
      "model_forward_time": 1.9209630489349365,
      "step": 94794
    },
    {
      "epoch": 1.9830703735351563e-05,
      "step": 94794,
      "training_step_time": 8.870680332183838
    },
    {
      "epoch": 1.983165740966797e-05,
      "model_forward_time": 1.9143822193145752,
      "step": 94795
    },
    {
      "epoch": 1.983165740966797e-05,
      "step": 94795,
      "training_step_time": 7.896974563598633
    },
    {
      "epoch": 1.9832611083984374e-05,
      "model_forward_time": 1.9147934913635254,
      "step": 94796
    },
    {
      "epoch": 1.9832611083984374e-05,
      "step": 94796,
      "training_step_time": 8.9205482006073
    },
    {
      "epoch": 1.9833564758300782e-05,
      "model_forward_time": 1.904482364654541,
      "step": 94797
    },
    {
      "epoch": 1.9833564758300782e-05,
      "step": 94797,
      "training_step_time": 8.370673179626465
    },
    {
      "epoch": 1.9834518432617186e-05,
      "model_forward_time": 1.9045369625091553,
      "step": 94798
    },
    {
      "epoch": 1.9834518432617186e-05,
      "step": 94798,
      "training_step_time": 7.855733633041382
    },
    {
      "epoch": 1.9835472106933593e-05,
      "model_forward_time": 1.9209959506988525,
      "step": 94799
    },
    {
      "epoch": 1.9835472106933593e-05,
      "step": 94799,
      "training_step_time": 7.520016431808472
    },
    {
      "epoch": 1.983642578125e-05,
      "grad_norm": 0.15597553551197052,
      "learning_rate": 7.377264962334596e-08,
      "loss": 0.0543,
      "step": 94800
    },
    {
      "epoch": 1.983642578125e-05,
      "model_forward_time": 1.8853039741516113,
      "step": 94800
    },
    {
      "dynamics_loss_avg": 0.042502226680517195,
      "epoch": 1.983642578125e-05,
      "step": 94800
    },
    {
      "action_loss_avg": 0.001679228083230555,
      "epoch": 1.983642578125e-05,
      "step": 94800
    },
    {
      "epoch": 1.983642578125e-05,
      "step": 94800,
      "training_step_time": 9.388001680374146
    },
    {
      "epoch": 1.9837379455566405e-05,
      "model_forward_time": 1.886408805847168,
      "step": 94801
    },
    {
      "epoch": 1.9837379455566405e-05,
      "step": 94801,
      "training_step_time": 7.83215856552124
    },
    {
      "epoch": 1.9838333129882812e-05,
      "model_forward_time": 1.903228998184204,
      "step": 94802
    },
    {
      "epoch": 1.9838333129882812e-05,
      "step": 94802,
      "training_step_time": 9.545207977294922
    },
    {
      "epoch": 1.983928680419922e-05,
      "model_forward_time": 1.9137470722198486,
      "step": 94803
    },
    {
      "epoch": 1.983928680419922e-05,
      "step": 94803,
      "training_step_time": 10.64755630493164
    },
    {
      "epoch": 1.9840240478515624e-05,
      "model_forward_time": 1.8986711502075195,
      "step": 94804
    },
    {
      "epoch": 1.9840240478515624e-05,
      "step": 94804,
      "training_step_time": 8.700690984725952
    },
    {
      "epoch": 1.984119415283203e-05,
      "model_forward_time": 1.9039111137390137,
      "step": 94805
    },
    {
      "epoch": 1.984119415283203e-05,
      "step": 94805,
      "training_step_time": 8.309592485427856
    },
    {
      "epoch": 1.984214782714844e-05,
      "model_forward_time": 1.9123263359069824,
      "step": 94806
    },
    {
      "epoch": 1.984214782714844e-05,
      "step": 94806,
      "training_step_time": 7.749095916748047
    },
    {
      "epoch": 1.9843101501464842e-05,
      "model_forward_time": 1.9125971794128418,
      "step": 94807
    },
    {
      "epoch": 1.9843101501464842e-05,
      "step": 94807,
      "training_step_time": 7.811357498168945
    },
    {
      "epoch": 1.984405517578125e-05,
      "model_forward_time": 1.9139583110809326,
      "step": 94808
    },
    {
      "epoch": 1.984405517578125e-05,
      "step": 94808,
      "training_step_time": 7.345276355743408
    },
    {
      "epoch": 1.9845008850097657e-05,
      "model_forward_time": 1.9028420448303223,
      "step": 94809
    },
    {
      "epoch": 1.9845008850097657e-05,
      "step": 94809,
      "training_step_time": 7.724520921707153
    },
    {
      "epoch": 1.984596252441406e-05,
      "grad_norm": 0.07414484024047852,
      "learning_rate": 7.348993226312772e-08,
      "loss": 0.0545,
      "step": 94810
    },
    {
      "epoch": 1.984596252441406e-05,
      "model_forward_time": 1.9019873142242432,
      "step": 94810
    },
    {
      "dynamics_loss_avg": 0.04546324890106916,
      "epoch": 1.984596252441406e-05,
      "step": 94810
    },
    {
      "action_loss_avg": 0.0010015643609222024,
      "epoch": 1.984596252441406e-05,
      "step": 94810
    },
    {
      "epoch": 1.984596252441406e-05,
      "step": 94810,
      "training_step_time": 7.800182104110718
    },
    {
      "epoch": 1.984691619873047e-05,
      "model_forward_time": 1.9150938987731934,
      "step": 94811
    },
    {
      "epoch": 1.984691619873047e-05,
      "step": 94811,
      "training_step_time": 9.4635329246521
    },
    {
      "epoch": 1.9847869873046876e-05,
      "model_forward_time": 2.545961380004883,
      "step": 94812
    },
    {
      "epoch": 1.9847869873046876e-05,
      "step": 94812,
      "training_step_time": 10.140576362609863
    },
    {
      "epoch": 1.984882354736328e-05,
      "model_forward_time": 1.9187240600585938,
      "step": 94813
    },
    {
      "epoch": 1.984882354736328e-05,
      "step": 94813,
      "training_step_time": 7.800524711608887
    },
    {
      "epoch": 1.9849777221679688e-05,
      "model_forward_time": 1.9098551273345947,
      "step": 94814
    },
    {
      "epoch": 1.9849777221679688e-05,
      "step": 94814,
      "training_step_time": 8.787366151809692
    },
    {
      "epoch": 1.9850730895996095e-05,
      "model_forward_time": 1.9038853645324707,
      "step": 94815
    },
    {
      "epoch": 1.9850730895996095e-05,
      "step": 94815,
      "training_step_time": 7.785854339599609
    },
    {
      "epoch": 1.98516845703125e-05,
      "model_forward_time": 1.908238410949707,
      "step": 94816
    },
    {
      "epoch": 1.98516845703125e-05,
      "step": 94816,
      "training_step_time": 8.313615798950195
    },
    {
      "epoch": 1.9852638244628906e-05,
      "model_forward_time": 1.9262776374816895,
      "step": 94817
    },
    {
      "epoch": 1.9852638244628906e-05,
      "step": 94817,
      "training_step_time": 8.144885301589966
    },
    {
      "epoch": 1.9853591918945314e-05,
      "model_forward_time": 1.880605936050415,
      "step": 94818
    },
    {
      "epoch": 1.9853591918945314e-05,
      "step": 94818,
      "training_step_time": 7.6433000564575195
    },
    {
      "epoch": 1.9854545593261718e-05,
      "model_forward_time": 1.9079110622406006,
      "step": 94819
    },
    {
      "epoch": 1.9854545593261718e-05,
      "step": 94819,
      "training_step_time": 7.703024864196777
    },
    {
      "epoch": 1.9855499267578125e-05,
      "grad_norm": 0.2068442553281784,
      "learning_rate": 7.320775365864342e-08,
      "loss": 0.0543,
      "step": 94820
    },
    {
      "epoch": 1.9855499267578125e-05,
      "model_forward_time": 1.9029300212860107,
      "step": 94820
    },
    {
      "dynamics_loss_avg": 0.0444702373817563,
      "epoch": 1.9855499267578125e-05,
      "step": 94820
    },
    {
      "action_loss_avg": 0.0010190052154939622,
      "epoch": 1.9855499267578125e-05,
      "step": 94820
    },
    {
      "epoch": 1.9855499267578125e-05,
      "step": 94820,
      "training_step_time": 7.742330312728882
    },
    {
      "epoch": 1.9856452941894533e-05,
      "model_forward_time": 2.510392665863037,
      "step": 94821
    },
    {
      "epoch": 1.9856452941894533e-05,
      "step": 94821,
      "training_step_time": 7.817576169967651
    },
    {
      "epoch": 1.9857406616210937e-05,
      "model_forward_time": 1.9111382961273193,
      "step": 94822
    },
    {
      "epoch": 1.9857406616210937e-05,
      "step": 94822,
      "training_step_time": 8.503086566925049
    },
    {
      "epoch": 1.9858360290527344e-05,
      "model_forward_time": 1.8989226818084717,
      "step": 94823
    },
    {
      "epoch": 1.9858360290527344e-05,
      "step": 94823,
      "training_step_time": 7.738276958465576
    },
    {
      "epoch": 1.985931396484375e-05,
      "model_forward_time": 1.9057130813598633,
      "step": 94824
    },
    {
      "epoch": 1.985931396484375e-05,
      "step": 94824,
      "training_step_time": 7.691951513290405
    },
    {
      "epoch": 1.9860267639160156e-05,
      "model_forward_time": 1.906181812286377,
      "step": 94825
    },
    {
      "epoch": 1.9860267639160156e-05,
      "step": 94825,
      "training_step_time": 7.709463596343994
    },
    {
      "epoch": 1.9861221313476563e-05,
      "model_forward_time": 1.9072473049163818,
      "step": 94826
    },
    {
      "epoch": 1.9861221313476563e-05,
      "step": 94826,
      "training_step_time": 7.211203575134277
    },
    {
      "epoch": 1.986217498779297e-05,
      "model_forward_time": 1.905040979385376,
      "step": 94827
    },
    {
      "epoch": 1.986217498779297e-05,
      "step": 94827,
      "training_step_time": 8.365699291229248
    },
    {
      "epoch": 1.9863128662109374e-05,
      "model_forward_time": 1.909968614578247,
      "step": 94828
    },
    {
      "epoch": 1.9863128662109374e-05,
      "step": 94828,
      "training_step_time": 7.645718097686768
    },
    {
      "epoch": 1.9864082336425782e-05,
      "model_forward_time": 1.903212070465088,
      "step": 94829
    },
    {
      "epoch": 1.9864082336425782e-05,
      "step": 94829,
      "training_step_time": 7.810502052307129
    },
    {
      "epoch": 1.9865036010742186e-05,
      "grad_norm": 0.09648188203573227,
      "learning_rate": 7.292611384075221e-08,
      "loss": 0.0537,
      "step": 94830
    },
    {
      "epoch": 1.9865036010742186e-05,
      "model_forward_time": 2.5567944049835205,
      "step": 94830
    },
    {
      "dynamics_loss_avg": 0.043939793668687346,
      "epoch": 1.9865036010742186e-05,
      "step": 94830
    },
    {
      "action_loss_avg": 0.0013017313263844698,
      "epoch": 1.9865036010742186e-05,
      "step": 94830
    },
    {
      "epoch": 1.9865036010742186e-05,
      "step": 94830,
      "training_step_time": 9.463824987411499
    },
    {
      "epoch": 1.9865989685058593e-05,
      "model_forward_time": 1.9065217971801758,
      "step": 94831
    },
    {
      "epoch": 1.9865989685058593e-05,
      "step": 94831,
      "training_step_time": 7.867013692855835
    },
    {
      "epoch": 1.9866943359375e-05,
      "model_forward_time": 1.9261536598205566,
      "step": 94832
    },
    {
      "epoch": 1.9866943359375e-05,
      "step": 94832,
      "training_step_time": 7.825343370437622
    },
    {
      "epoch": 1.9867897033691405e-05,
      "model_forward_time": 1.9177918434143066,
      "step": 94833
    },
    {
      "epoch": 1.9867897033691405e-05,
      "step": 94833,
      "training_step_time": 8.155207872390747
    },
    {
      "epoch": 1.9868850708007812e-05,
      "model_forward_time": 1.9129786491394043,
      "step": 94834
    },
    {
      "epoch": 1.9868850708007812e-05,
      "step": 94834,
      "training_step_time": 7.750152587890625
    },
    {
      "epoch": 1.986980438232422e-05,
      "model_forward_time": 1.8993167877197266,
      "step": 94835
    },
    {
      "epoch": 1.986980438232422e-05,
      "step": 94835,
      "training_step_time": 7.796287536621094
    },
    {
      "epoch": 1.9870758056640624e-05,
      "model_forward_time": 1.9173333644866943,
      "step": 94836
    },
    {
      "epoch": 1.9870758056640624e-05,
      "step": 94836,
      "training_step_time": 8.276521682739258
    },
    {
      "epoch": 1.987171173095703e-05,
      "model_forward_time": 1.9077517986297607,
      "step": 94837
    },
    {
      "epoch": 1.987171173095703e-05,
      "step": 94837,
      "training_step_time": 7.746143579483032
    },
    {
      "epoch": 1.987266540527344e-05,
      "model_forward_time": 1.9080123901367188,
      "step": 94838
    },
    {
      "epoch": 1.987266540527344e-05,
      "step": 94838,
      "training_step_time": 8.845335721969604
    },
    {
      "epoch": 1.9873619079589842e-05,
      "model_forward_time": 2.534482955932617,
      "step": 94839
    },
    {
      "epoch": 1.9873619079589842e-05,
      "step": 94839,
      "training_step_time": 9.600464344024658
    },
    {
      "epoch": 1.987457275390625e-05,
      "grad_norm": 0.08100692927837372,
      "learning_rate": 7.264501284025338e-08,
      "loss": 0.0567,
      "step": 94840
    },
    {
      "epoch": 1.987457275390625e-05,
      "model_forward_time": 1.9090056419372559,
      "step": 94840
    },
    {
      "dynamics_loss_avg": 0.04469938725233078,
      "epoch": 1.987457275390625e-05,
      "step": 94840
    },
    {
      "action_loss_avg": 0.0021135046146810053,
      "epoch": 1.987457275390625e-05,
      "step": 94840
    },
    {
      "epoch": 1.987457275390625e-05,
      "step": 94840,
      "training_step_time": 7.769475936889648
    },
    {
      "epoch": 1.9875526428222657e-05,
      "model_forward_time": 1.9044606685638428,
      "step": 94841
    },
    {
      "epoch": 1.9875526428222657e-05,
      "step": 94841,
      "training_step_time": 9.575191736221313
    },
    {
      "epoch": 1.987648010253906e-05,
      "model_forward_time": 1.8929157257080078,
      "step": 94842
    },
    {
      "epoch": 1.987648010253906e-05,
      "step": 94842,
      "training_step_time": 8.591771364212036
    },
    {
      "epoch": 1.987743377685547e-05,
      "model_forward_time": 1.9022088050842285,
      "step": 94843
    },
    {
      "epoch": 1.987743377685547e-05,
      "step": 94843,
      "training_step_time": 7.803780794143677
    },
    {
      "epoch": 1.9878387451171876e-05,
      "model_forward_time": 1.8948183059692383,
      "step": 94844
    },
    {
      "epoch": 1.9878387451171876e-05,
      "step": 94844,
      "training_step_time": 7.849301099777222
    },
    {
      "epoch": 1.987934112548828e-05,
      "model_forward_time": 1.9107270240783691,
      "step": 94845
    },
    {
      "epoch": 1.987934112548828e-05,
      "step": 94845,
      "training_step_time": 7.35856294631958
    },
    {
      "epoch": 1.9880294799804688e-05,
      "model_forward_time": 1.9184691905975342,
      "step": 94846
    },
    {
      "epoch": 1.9880294799804688e-05,
      "step": 94846,
      "training_step_time": 8.394730567932129
    },
    {
      "epoch": 1.9881248474121095e-05,
      "model_forward_time": 1.9006462097167969,
      "step": 94847
    },
    {
      "epoch": 1.9881248474121095e-05,
      "step": 94847,
      "training_step_time": 8.668776035308838
    },
    {
      "epoch": 1.98822021484375e-05,
      "model_forward_time": 2.5295960903167725,
      "step": 94848
    },
    {
      "epoch": 1.98822021484375e-05,
      "step": 94848,
      "training_step_time": 9.63730764389038
    },
    {
      "epoch": 1.9883155822753906e-05,
      "model_forward_time": 1.906435489654541,
      "step": 94849
    },
    {
      "epoch": 1.9883155822753906e-05,
      "step": 94849,
      "training_step_time": 8.803744554519653
    },
    {
      "epoch": 1.9884109497070314e-05,
      "grad_norm": 0.08040043711662292,
      "learning_rate": 7.236445068788789e-08,
      "loss": 0.0545,
      "step": 94850
    },
    {
      "epoch": 1.9884109497070314e-05,
      "model_forward_time": 1.9043734073638916,
      "step": 94850
    },
    {
      "dynamics_loss_avg": 0.04623127337545156,
      "epoch": 1.9884109497070314e-05,
      "step": 94850
    },
    {
      "action_loss_avg": 0.0014104819158092142,
      "epoch": 1.9884109497070314e-05,
      "step": 94850
    },
    {
      "epoch": 1.9884109497070314e-05,
      "step": 94850,
      "training_step_time": 8.663853645324707
    },
    {
      "epoch": 1.9885063171386718e-05,
      "model_forward_time": 1.8994884490966797,
      "step": 94851
    },
    {
      "epoch": 1.9885063171386718e-05,
      "step": 94851,
      "training_step_time": 7.88067364692688
    },
    {
      "epoch": 1.9886016845703125e-05,
      "model_forward_time": 1.9085075855255127,
      "step": 94852
    },
    {
      "epoch": 1.9886016845703125e-05,
      "step": 94852,
      "training_step_time": 7.696977138519287
    },
    {
      "epoch": 1.9886970520019533e-05,
      "model_forward_time": 1.917754888534546,
      "step": 94853
    },
    {
      "epoch": 1.9886970520019533e-05,
      "step": 94853,
      "training_step_time": 7.780015468597412
    },
    {
      "epoch": 1.9887924194335937e-05,
      "model_forward_time": 1.8982834815979004,
      "step": 94854
    },
    {
      "epoch": 1.9887924194335937e-05,
      "step": 94854,
      "training_step_time": 7.278112411499023
    },
    {
      "epoch": 1.9888877868652344e-05,
      "model_forward_time": 1.896183967590332,
      "step": 94855
    },
    {
      "epoch": 1.9888877868652344e-05,
      "step": 94855,
      "training_step_time": 8.21502423286438
    },
    {
      "epoch": 1.988983154296875e-05,
      "model_forward_time": 1.900221586227417,
      "step": 94856
    },
    {
      "epoch": 1.988983154296875e-05,
      "step": 94856,
      "training_step_time": 7.924801349639893
    },
    {
      "epoch": 1.9890785217285156e-05,
      "model_forward_time": 1.9144635200500488,
      "step": 94857
    },
    {
      "epoch": 1.9890785217285156e-05,
      "step": 94857,
      "training_step_time": 8.431716442108154
    },
    {
      "epoch": 1.9891738891601563e-05,
      "model_forward_time": 2.519385576248169,
      "step": 94858
    },
    {
      "epoch": 1.9891738891601563e-05,
      "step": 94858,
      "training_step_time": 8.626364469528198
    },
    {
      "epoch": 1.989269256591797e-05,
      "model_forward_time": 1.9335193634033203,
      "step": 94859
    },
    {
      "epoch": 1.989269256591797e-05,
      "step": 94859,
      "training_step_time": 9.07909870147705
    },
    {
      "epoch": 1.9893646240234374e-05,
      "grad_norm": 0.12046225368976593,
      "learning_rate": 7.20844274143373e-08,
      "loss": 0.056,
      "step": 94860
    },
    {
      "epoch": 1.9893646240234374e-05,
      "model_forward_time": 1.8854141235351562,
      "step": 94860
    },
    {
      "dynamics_loss_avg": 0.05565580371767283,
      "epoch": 1.9893646240234374e-05,
      "step": 94860
    },
    {
      "action_loss_avg": 0.0015255418402375653,
      "epoch": 1.9893646240234374e-05,
      "step": 94860
    },
    {
      "epoch": 1.9893646240234374e-05,
      "step": 94860,
      "training_step_time": 7.944704055786133
    },
    {
      "epoch": 1.9894599914550782e-05,
      "model_forward_time": 1.897831678390503,
      "step": 94861
    },
    {
      "epoch": 1.9894599914550782e-05,
      "step": 94861,
      "training_step_time": 7.840838670730591
    },
    {
      "epoch": 1.9895553588867186e-05,
      "model_forward_time": 1.8950912952423096,
      "step": 94862
    },
    {
      "epoch": 1.9895553588867186e-05,
      "step": 94862,
      "training_step_time": 7.807279109954834
    },
    {
      "epoch": 1.9896507263183593e-05,
      "model_forward_time": 1.8903696537017822,
      "step": 94863
    },
    {
      "epoch": 1.9896507263183593e-05,
      "step": 94863,
      "training_step_time": 7.2328643798828125
    },
    {
      "epoch": 1.98974609375e-05,
      "model_forward_time": 1.9019834995269775,
      "step": 94864
    },
    {
      "epoch": 1.98974609375e-05,
      "step": 94864,
      "training_step_time": 7.755425214767456
    },
    {
      "epoch": 1.9898414611816405e-05,
      "model_forward_time": 1.898573398590088,
      "step": 94865
    },
    {
      "epoch": 1.9898414611816405e-05,
      "step": 94865,
      "training_step_time": 7.621407747268677
    },
    {
      "epoch": 1.9899368286132812e-05,
      "model_forward_time": 1.90798020362854,
      "step": 94866
    },
    {
      "epoch": 1.9899368286132812e-05,
      "step": 94866,
      "training_step_time": 8.903850793838501
    },
    {
      "epoch": 1.990032196044922e-05,
      "model_forward_time": 2.522189140319824,
      "step": 94867
    },
    {
      "epoch": 1.990032196044922e-05,
      "step": 94867,
      "training_step_time": 8.859182596206665
    },
    {
      "epoch": 1.9901275634765624e-05,
      "model_forward_time": 1.8993632793426514,
      "step": 94868
    },
    {
      "epoch": 1.9901275634765624e-05,
      "step": 94868,
      "training_step_time": 8.905164957046509
    },
    {
      "epoch": 1.990222930908203e-05,
      "model_forward_time": 1.8955543041229248,
      "step": 94869
    },
    {
      "epoch": 1.990222930908203e-05,
      "step": 94869,
      "training_step_time": 7.859631776809692
    },
    {
      "epoch": 1.990318298339844e-05,
      "grad_norm": 0.10777851939201355,
      "learning_rate": 7.180494305022489e-08,
      "loss": 0.056,
      "step": 94870
    },
    {
      "epoch": 1.990318298339844e-05,
      "model_forward_time": 1.8996896743774414,
      "step": 94870
    },
    {
      "dynamics_loss_avg": 0.059336280636489394,
      "epoch": 1.990318298339844e-05,
      "step": 94870
    },
    {
      "action_loss_avg": 0.0009778409119462594,
      "epoch": 1.990318298339844e-05,
      "step": 94870
    },
    {
      "epoch": 1.990318298339844e-05,
      "step": 94870,
      "training_step_time": 8.734272003173828
    },
    {
      "epoch": 1.9904136657714842e-05,
      "model_forward_time": 1.8964412212371826,
      "step": 94871
    },
    {
      "epoch": 1.9904136657714842e-05,
      "step": 94871,
      "training_step_time": 7.743346691131592
    },
    {
      "epoch": 1.990509033203125e-05,
      "model_forward_time": 1.9056835174560547,
      "step": 94872
    },
    {
      "epoch": 1.990509033203125e-05,
      "step": 94872,
      "training_step_time": 7.772984504699707
    },
    {
      "epoch": 1.9906044006347657e-05,
      "model_forward_time": 1.8963704109191895,
      "step": 94873
    },
    {
      "epoch": 1.9906044006347657e-05,
      "step": 94873,
      "training_step_time": 7.875804424285889
    },
    {
      "epoch": 1.990699768066406e-05,
      "model_forward_time": 1.9064958095550537,
      "step": 94874
    },
    {
      "epoch": 1.990699768066406e-05,
      "step": 94874,
      "training_step_time": 8.895012617111206
    },
    {
      "epoch": 1.990795135498047e-05,
      "model_forward_time": 1.9026222229003906,
      "step": 94875
    },
    {
      "epoch": 1.990795135498047e-05,
      "step": 94875,
      "training_step_time": 7.833660840988159
    },
    {
      "epoch": 1.9908905029296876e-05,
      "model_forward_time": 2.518852472305298,
      "step": 94876
    },
    {
      "epoch": 1.9908905029296876e-05,
      "step": 94876,
      "training_step_time": 8.605566263198853
    },
    {
      "epoch": 1.990985870361328e-05,
      "model_forward_time": 1.9027659893035889,
      "step": 94877
    },
    {
      "epoch": 1.990985870361328e-05,
      "step": 94877,
      "training_step_time": 7.957059383392334
    },
    {
      "epoch": 1.9910812377929688e-05,
      "model_forward_time": 1.529719352722168,
      "step": 94878
    },
    {
      "epoch": 1.9910812377929688e-05,
      "step": 94878,
      "training_step_time": 8.761653661727905
    },
    {
      "epoch": 1.9911766052246095e-05,
      "model_forward_time": 1.9052016735076904,
      "step": 94879
    },
    {
      "epoch": 1.9911766052246095e-05,
      "step": 94879,
      "training_step_time": 7.793771982192993
    },
    {
      "epoch": 1.99127197265625e-05,
      "grad_norm": 0.07459616661071777,
      "learning_rate": 7.152599762611457e-08,
      "loss": 0.0542,
      "step": 94880
    },
    {
      "epoch": 1.99127197265625e-05,
      "model_forward_time": 1.9046144485473633,
      "step": 94880
    },
    {
      "dynamics_loss_avg": 0.06002612765878439,
      "epoch": 1.99127197265625e-05,
      "step": 94880
    },
    {
      "action_loss_avg": 0.0017090945679228754,
      "epoch": 1.99127197265625e-05,
      "step": 94880
    },
    {
      "epoch": 1.99127197265625e-05,
      "step": 94880,
      "training_step_time": 8.810047149658203
    },
    {
      "epoch": 1.9913673400878906e-05,
      "model_forward_time": 1.913487195968628,
      "step": 94881
    },
    {
      "epoch": 1.9913673400878906e-05,
      "step": 94881,
      "training_step_time": 7.582571029663086
    },
    {
      "epoch": 1.9914627075195314e-05,
      "model_forward_time": 1.9000139236450195,
      "step": 94882
    },
    {
      "epoch": 1.9914627075195314e-05,
      "step": 94882,
      "training_step_time": 7.475754261016846
    },
    {
      "epoch": 1.9915580749511718e-05,
      "model_forward_time": 1.9031143188476562,
      "step": 94883
    },
    {
      "epoch": 1.9915580749511718e-05,
      "step": 94883,
      "training_step_time": 7.658091306686401
    },
    {
      "epoch": 1.9916534423828125e-05,
      "model_forward_time": 1.9036784172058105,
      "step": 94884
    },
    {
      "epoch": 1.9916534423828125e-05,
      "step": 94884,
      "training_step_time": 7.625420808792114
    },
    {
      "epoch": 1.9917488098144533e-05,
      "model_forward_time": 2.541961431503296,
      "step": 94885
    },
    {
      "epoch": 1.9917488098144533e-05,
      "step": 94885,
      "training_step_time": 8.355076313018799
    },
    {
      "epoch": 1.9918441772460937e-05,
      "model_forward_time": 1.9037435054779053,
      "step": 94886
    },
    {
      "epoch": 1.9918441772460937e-05,
      "step": 94886,
      "training_step_time": 10.319801807403564
    },
    {
      "epoch": 1.9919395446777344e-05,
      "model_forward_time": 1.897878885269165,
      "step": 94887
    },
    {
      "epoch": 1.9919395446777344e-05,
      "step": 94887,
      "training_step_time": 8.56327199935913
    },
    {
      "epoch": 1.992034912109375e-05,
      "model_forward_time": 1.9131712913513184,
      "step": 94888
    },
    {
      "epoch": 1.992034912109375e-05,
      "step": 94888,
      "training_step_time": 8.488107681274414
    },
    {
      "epoch": 1.9921302795410156e-05,
      "model_forward_time": 1.8927347660064697,
      "step": 94889
    },
    {
      "epoch": 1.9921302795410156e-05,
      "step": 94889,
      "training_step_time": 8.31951355934143
    },
    {
      "epoch": 1.9922256469726563e-05,
      "grad_norm": 0.0795309916138649,
      "learning_rate": 7.124759117251078e-08,
      "loss": 0.0548,
      "step": 94890
    },
    {
      "epoch": 1.9922256469726563e-05,
      "model_forward_time": 1.90565824508667,
      "step": 94890
    },
    {
      "dynamics_loss_avg": 0.04690229594707489,
      "epoch": 1.9922256469726563e-05,
      "step": 94890
    },
    {
      "action_loss_avg": 0.0010270422382745892,
      "epoch": 1.9922256469726563e-05,
      "step": 94890
    },
    {
      "epoch": 1.9922256469726563e-05,
      "step": 94890,
      "training_step_time": 7.559769153594971
    },
    {
      "epoch": 1.9923210144042967e-05,
      "model_forward_time": 1.8954460620880127,
      "step": 94891
    },
    {
      "epoch": 1.9923210144042967e-05,
      "step": 94891,
      "training_step_time": 7.2042131423950195
    },
    {
      "epoch": 1.9924163818359374e-05,
      "model_forward_time": 1.9040992259979248,
      "step": 94892
    },
    {
      "epoch": 1.9924163818359374e-05,
      "step": 94892,
      "training_step_time": 7.5258708000183105
    },
    {
      "epoch": 1.9925117492675782e-05,
      "model_forward_time": 1.9058423042297363,
      "step": 94893
    },
    {
      "epoch": 1.9925117492675782e-05,
      "step": 94893,
      "training_step_time": 8.399712324142456
    },
    {
      "epoch": 1.9926071166992186e-05,
      "model_forward_time": 2.5136189460754395,
      "step": 94894
    },
    {
      "epoch": 1.9926071166992186e-05,
      "step": 94894,
      "training_step_time": 10.964961528778076
    },
    {
      "epoch": 1.9927024841308593e-05,
      "model_forward_time": 1.9054243564605713,
      "step": 94895
    },
    {
      "epoch": 1.9927024841308593e-05,
      "step": 94895,
      "training_step_time": 8.621877670288086
    },
    {
      "epoch": 1.9927978515625e-05,
      "model_forward_time": 1.8985211849212646,
      "step": 94896
    },
    {
      "epoch": 1.9927978515625e-05,
      "step": 94896,
      "training_step_time": 7.737250804901123
    },
    {
      "epoch": 1.9928932189941405e-05,
      "model_forward_time": 1.9142475128173828,
      "step": 94897
    },
    {
      "epoch": 1.9928932189941405e-05,
      "step": 94897,
      "training_step_time": 7.508125305175781
    },
    {
      "epoch": 1.9929885864257812e-05,
      "model_forward_time": 1.9090509414672852,
      "step": 94898
    },
    {
      "epoch": 1.9929885864257812e-05,
      "step": 94898,
      "training_step_time": 8.68160629272461
    },
    {
      "epoch": 1.993083953857422e-05,
      "model_forward_time": 1.9130735397338867,
      "step": 94899
    },
    {
      "epoch": 1.993083953857422e-05,
      "step": 94899,
      "training_step_time": 7.644418954849243
    },
    {
      "epoch": 1.9931793212890624e-05,
      "grad_norm": 0.10185987502336502,
      "learning_rate": 7.096972371986033e-08,
      "loss": 0.055,
      "step": 94900
    },
    {
      "epoch": 1.9931793212890624e-05,
      "model_forward_time": 1.9080426692962646,
      "step": 94900
    },
    {
      "dynamics_loss_avg": 0.06957940496504307,
      "epoch": 1.9931793212890624e-05,
      "step": 94900
    },
    {
      "action_loss_avg": 0.0012997090787393972,
      "epoch": 1.9931793212890624e-05,
      "step": 94900
    },
    {
      "epoch": 1.9931793212890624e-05,
      "step": 94900,
      "training_step_time": 7.244174957275391
    },
    {
      "epoch": 1.993274688720703e-05,
      "model_forward_time": 1.9126343727111816,
      "step": 94901
    },
    {
      "epoch": 1.993274688720703e-05,
      "step": 94901,
      "training_step_time": 7.610662221908569
    },
    {
      "epoch": 1.993370056152344e-05,
      "model_forward_time": 1.9091277122497559,
      "step": 94902
    },
    {
      "epoch": 1.993370056152344e-05,
      "step": 94902,
      "training_step_time": 8.39669919013977
    },
    {
      "epoch": 1.9934654235839843e-05,
      "model_forward_time": 1.8880486488342285,
      "step": 94903
    },
    {
      "epoch": 1.9934654235839843e-05,
      "step": 94903,
      "training_step_time": 7.74537467956543
    },
    {
      "epoch": 1.993560791015625e-05,
      "model_forward_time": 2.5501575469970703,
      "step": 94904
    },
    {
      "epoch": 1.993560791015625e-05,
      "step": 94904,
      "training_step_time": 9.32271933555603
    },
    {
      "epoch": 1.9936561584472657e-05,
      "model_forward_time": 1.917149305343628,
      "step": 94905
    },
    {
      "epoch": 1.9936561584472657e-05,
      "step": 94905,
      "training_step_time": 9.599496364593506
    },
    {
      "epoch": 1.993751525878906e-05,
      "model_forward_time": 1.916612148284912,
      "step": 94906
    },
    {
      "epoch": 1.993751525878906e-05,
      "step": 94906,
      "training_step_time": 8.58890700340271
    },
    {
      "epoch": 1.993846893310547e-05,
      "model_forward_time": 1.9069454669952393,
      "step": 94907
    },
    {
      "epoch": 1.993846893310547e-05,
      "step": 94907,
      "training_step_time": 8.671674966812134
    },
    {
      "epoch": 1.9939422607421876e-05,
      "model_forward_time": 1.922912359237671,
      "step": 94908
    },
    {
      "epoch": 1.9939422607421876e-05,
      "step": 94908,
      "training_step_time": 7.597275972366333
    },
    {
      "epoch": 1.994037628173828e-05,
      "model_forward_time": 1.5432343482971191,
      "step": 94909
    },
    {
      "epoch": 1.994037628173828e-05,
      "step": 94909,
      "training_step_time": 7.207960844039917
    },
    {
      "epoch": 1.9941329956054688e-05,
      "grad_norm": 0.09410370141267776,
      "learning_rate": 7.069239529855054e-08,
      "loss": 0.0546,
      "step": 94910
    },
    {
      "epoch": 1.9941329956054688e-05,
      "model_forward_time": 1.917097568511963,
      "step": 94910
    },
    {
      "dynamics_loss_avg": 0.044250118918716906,
      "epoch": 1.9941329956054688e-05,
      "step": 94910
    },
    {
      "action_loss_avg": 0.0008797291346127167,
      "epoch": 1.9941329956054688e-05,
      "step": 94910
    },
    {
      "epoch": 1.9941329956054688e-05,
      "step": 94910,
      "training_step_time": 7.5210137367248535
    },
    {
      "epoch": 1.9942283630371095e-05,
      "model_forward_time": 1.8768465518951416,
      "step": 94911
    },
    {
      "epoch": 1.9942283630371095e-05,
      "step": 94911,
      "training_step_time": 7.608968496322632
    },
    {
      "epoch": 1.99432373046875e-05,
      "model_forward_time": 1.916508436203003,
      "step": 94912
    },
    {
      "epoch": 1.99432373046875e-05,
      "step": 94912,
      "training_step_time": 7.550562858581543
    },
    {
      "epoch": 1.9944190979003906e-05,
      "model_forward_time": 2.5093750953674316,
      "step": 94913
    },
    {
      "epoch": 1.9944190979003906e-05,
      "step": 94913,
      "training_step_time": 8.38916802406311
    },
    {
      "epoch": 1.9945144653320314e-05,
      "model_forward_time": 1.8953502178192139,
      "step": 94914
    },
    {
      "epoch": 1.9945144653320314e-05,
      "step": 94914,
      "training_step_time": 8.22888731956482
    },
    {
      "epoch": 1.9946098327636718e-05,
      "model_forward_time": 1.9063200950622559,
      "step": 94915
    },
    {
      "epoch": 1.9946098327636718e-05,
      "step": 94915,
      "training_step_time": 8.413277626037598
    },
    {
      "epoch": 1.9947052001953125e-05,
      "model_forward_time": 1.5313434600830078,
      "step": 94916
    },
    {
      "epoch": 1.9947052001953125e-05,
      "step": 94916,
      "training_step_time": 7.583754301071167
    },
    {
      "epoch": 1.9948005676269533e-05,
      "model_forward_time": 1.8981995582580566,
      "step": 94917
    },
    {
      "epoch": 1.9948005676269533e-05,
      "step": 94917,
      "training_step_time": 8.190202951431274
    },
    {
      "epoch": 1.9948959350585937e-05,
      "model_forward_time": 1.9026834964752197,
      "step": 94918
    },
    {
      "epoch": 1.9948959350585937e-05,
      "step": 94918,
      "training_step_time": 7.086438894271851
    },
    {
      "epoch": 1.9949913024902344e-05,
      "model_forward_time": 1.7587230205535889,
      "step": 94919
    },
    {
      "epoch": 1.9949913024902344e-05,
      "step": 94919,
      "training_step_time": 8.448707580566406
    },
    {
      "epoch": 1.995086669921875e-05,
      "grad_norm": 0.09493910521268845,
      "learning_rate": 7.041560593890939e-08,
      "loss": 0.0547,
      "step": 94920
    },
    {
      "epoch": 1.995086669921875e-05,
      "model_forward_time": 1.749852180480957,
      "step": 94920
    },
    {
      "dynamics_loss_avg": 0.05750631857663393,
      "epoch": 1.995086669921875e-05,
      "step": 94920
    },
    {
      "action_loss_avg": 0.0015326303837355226,
      "epoch": 1.995086669921875e-05,
      "step": 94920
    },
    {
      "epoch": 1.995086669921875e-05,
      "step": 94920,
      "training_step_time": 8.536938428878784
    },
    {
      "epoch": 1.9951820373535156e-05,
      "model_forward_time": 1.7526230812072754,
      "step": 94921
    },
    {
      "epoch": 1.9951820373535156e-05,
      "step": 94921,
      "training_step_time": 9.483905792236328
    },
    {
      "epoch": 1.9952774047851563e-05,
      "model_forward_time": 2.411987543106079,
      "step": 94922
    },
    {
      "epoch": 1.9952774047851563e-05,
      "step": 94922,
      "training_step_time": 8.548994541168213
    },
    {
      "epoch": 1.9953727722167967e-05,
      "model_forward_time": 1.7575554847717285,
      "step": 94923
    },
    {
      "epoch": 1.9953727722167967e-05,
      "step": 94923,
      "training_step_time": 9.211364030838013
    },
    {
      "epoch": 1.9954681396484375e-05,
      "model_forward_time": 1.7516040802001953,
      "step": 94924
    },
    {
      "epoch": 1.9954681396484375e-05,
      "step": 94924,
      "training_step_time": 7.679598093032837
    },
    {
      "epoch": 1.9955635070800782e-05,
      "model_forward_time": 1.783888816833496,
      "step": 94925
    },
    {
      "epoch": 1.9955635070800782e-05,
      "step": 94925,
      "training_step_time": 7.532114028930664
    },
    {
      "epoch": 1.9956588745117186e-05,
      "model_forward_time": 1.7582170963287354,
      "step": 94926
    },
    {
      "epoch": 1.9956588745117186e-05,
      "step": 94926,
      "training_step_time": 7.697709798812866
    },
    {
      "epoch": 1.9957542419433593e-05,
      "model_forward_time": 1.743504524230957,
      "step": 94927
    },
    {
      "epoch": 1.9957542419433593e-05,
      "step": 94927,
      "training_step_time": 7.6905293464660645
    },
    {
      "epoch": 1.995849609375e-05,
      "model_forward_time": 1.7528252601623535,
      "step": 94928
    },
    {
      "epoch": 1.995849609375e-05,
      "step": 94928,
      "training_step_time": 7.319324254989624
    },
    {
      "epoch": 1.9959449768066405e-05,
      "model_forward_time": 1.759972095489502,
      "step": 94929
    },
    {
      "epoch": 1.9959449768066405e-05,
      "step": 94929,
      "training_step_time": 7.634856939315796
    },
    {
      "epoch": 1.9960403442382812e-05,
      "grad_norm": 0.09960814565420151,
      "learning_rate": 7.0139355671206e-08,
      "loss": 0.0539,
      "step": 94930
    },
    {
      "epoch": 1.9960403442382812e-05,
      "model_forward_time": 1.7481451034545898,
      "step": 94930
    },
    {
      "dynamics_loss_avg": 0.05900391060858965,
      "epoch": 1.9960403442382812e-05,
      "step": 94930
    },
    {
      "action_loss_avg": 0.0015935159259242936,
      "epoch": 1.9960403442382812e-05,
      "step": 94930
    },
    {
      "epoch": 1.9960403442382812e-05,
      "step": 94930,
      "training_step_time": 8.748555660247803
    },
    {
      "epoch": 1.996135711669922e-05,
      "model_forward_time": 1.751840353012085,
      "step": 94931
    },
    {
      "epoch": 1.996135711669922e-05,
      "step": 94931,
      "training_step_time": 9.051488876342773
    },
    {
      "epoch": 1.9962310791015624e-05,
      "model_forward_time": 1.3748202323913574,
      "step": 94932
    },
    {
      "epoch": 1.9962310791015624e-05,
      "step": 94932,
      "training_step_time": 9.718116998672485
    },
    {
      "epoch": 1.996326446533203e-05,
      "model_forward_time": 1.4650242328643799,
      "step": 94933
    },
    {
      "epoch": 1.996326446533203e-05,
      "step": 94933,
      "training_step_time": 8.960783243179321
    },
    {
      "epoch": 1.996421813964844e-05,
      "model_forward_time": 1.7516968250274658,
      "step": 94934
    },
    {
      "epoch": 1.996421813964844e-05,
      "step": 94934,
      "training_step_time": 7.677347421646118
    },
    {
      "epoch": 1.9965171813964843e-05,
      "model_forward_time": 1.7493782043457031,
      "step": 94935
    },
    {
      "epoch": 1.9965171813964843e-05,
      "step": 94935,
      "training_step_time": 9.304008960723877
    },
    {
      "epoch": 1.996612548828125e-05,
      "model_forward_time": 1.7542304992675781,
      "step": 94936
    },
    {
      "epoch": 1.996612548828125e-05,
      "step": 94936,
      "training_step_time": 7.792084455490112
    },
    {
      "epoch": 1.9967079162597657e-05,
      "model_forward_time": 1.7536859512329102,
      "step": 94937
    },
    {
      "epoch": 1.9967079162597657e-05,
      "step": 94937,
      "training_step_time": 8.397361993789673
    },
    {
      "epoch": 1.996803283691406e-05,
      "model_forward_time": 1.7548484802246094,
      "step": 94938
    },
    {
      "epoch": 1.996803283691406e-05,
      "step": 94938,
      "training_step_time": 7.781484603881836
    },
    {
      "epoch": 1.996898651123047e-05,
      "model_forward_time": 1.7591776847839355,
      "step": 94939
    },
    {
      "epoch": 1.996898651123047e-05,
      "step": 94939,
      "training_step_time": 7.857102394104004
    },
    {
      "epoch": 1.9969940185546876e-05,
      "grad_norm": 0.07548905909061432,
      "learning_rate": 6.98636445256501e-08,
      "loss": 0.0545,
      "step": 94940
    },
    {
      "epoch": 1.9969940185546876e-05,
      "model_forward_time": 1.745204210281372,
      "step": 94940
    },
    {
      "dynamics_loss_avg": 0.06789323464035987,
      "epoch": 1.9969940185546876e-05,
      "step": 94940
    },
    {
      "action_loss_avg": 0.00141156415338628,
      "epoch": 1.9969940185546876e-05,
      "step": 94940
    },
    {
      "epoch": 1.9969940185546876e-05,
      "step": 94940,
      "training_step_time": 7.859970331192017
    },
    {
      "epoch": 1.997089385986328e-05,
      "model_forward_time": 2.411405086517334,
      "step": 94941
    },
    {
      "epoch": 1.997089385986328e-05,
      "step": 94941,
      "training_step_time": 8.57461929321289
    },
    {
      "epoch": 1.9971847534179688e-05,
      "model_forward_time": 1.82047700881958,
      "step": 94942
    },
    {
      "epoch": 1.9971847534179688e-05,
      "step": 94942,
      "training_step_time": 7.838983535766602
    },
    {
      "epoch": 1.9972801208496095e-05,
      "model_forward_time": 1.760878086090088,
      "step": 94943
    },
    {
      "epoch": 1.9972801208496095e-05,
      "step": 94943,
      "training_step_time": 7.649335145950317
    },
    {
      "epoch": 1.99737548828125e-05,
      "model_forward_time": 1.7557437419891357,
      "step": 94944
    },
    {
      "epoch": 1.99737548828125e-05,
      "step": 94944,
      "training_step_time": 9.426132678985596
    },
    {
      "epoch": 1.9974708557128907e-05,
      "model_forward_time": 1.7531890869140625,
      "step": 94945
    },
    {
      "epoch": 1.9974708557128907e-05,
      "step": 94945,
      "training_step_time": 7.718531608581543
    },
    {
      "epoch": 1.9975662231445314e-05,
      "model_forward_time": 1.7546000480651855,
      "step": 94946
    },
    {
      "epoch": 1.9975662231445314e-05,
      "step": 94946,
      "training_step_time": 7.3705315589904785
    },
    {
      "epoch": 1.9976615905761718e-05,
      "model_forward_time": 1.7504427433013916,
      "step": 94947
    },
    {
      "epoch": 1.9976615905761718e-05,
      "step": 94947,
      "training_step_time": 8.603187561035156
    },
    {
      "epoch": 1.9977569580078125e-05,
      "model_forward_time": 1.7588071823120117,
      "step": 94948
    },
    {
      "epoch": 1.9977569580078125e-05,
      "step": 94948,
      "training_step_time": 8.50214409828186
    },
    {
      "epoch": 1.9978523254394533e-05,
      "model_forward_time": 1.7539653778076172,
      "step": 94949
    },
    {
      "epoch": 1.9978523254394533e-05,
      "step": 94949,
      "training_step_time": 10.191534042358398
    },
    {
      "epoch": 1.9979476928710937e-05,
      "grad_norm": 0.12228753417730331,
      "learning_rate": 6.95884725323942e-08,
      "loss": 0.0551,
      "step": 94950
    },
    {
      "epoch": 1.9979476928710937e-05,
      "model_forward_time": 2.4032976627349854,
      "step": 94950
    },
    {
      "dynamics_loss_avg": 0.04601569026708603,
      "epoch": 1.9979476928710937e-05,
      "step": 94950
    },
    {
      "action_loss_avg": 0.00130768638337031,
      "epoch": 1.9979476928710937e-05,
      "step": 94950
    },
    {
      "epoch": 1.9979476928710937e-05,
      "step": 94950,
      "training_step_time": 8.679158210754395
    },
    {
      "epoch": 1.9980430603027344e-05,
      "model_forward_time": 1.8209614753723145,
      "step": 94951
    },
    {
      "epoch": 1.9980430603027344e-05,
      "step": 94951,
      "training_step_time": 7.8413591384887695
    },
    {
      "epoch": 1.998138427734375e-05,
      "model_forward_time": 1.760582447052002,
      "step": 94952
    },
    {
      "epoch": 1.998138427734375e-05,
      "step": 94952,
      "training_step_time": 7.75629997253418
    },
    {
      "epoch": 1.9982337951660156e-05,
      "model_forward_time": 1.7627201080322266,
      "step": 94953
    },
    {
      "epoch": 1.9982337951660156e-05,
      "step": 94953,
      "training_step_time": 7.7793073654174805
    },
    {
      "epoch": 1.9983291625976563e-05,
      "model_forward_time": 1.759305715560913,
      "step": 94954
    },
    {
      "epoch": 1.9983291625976563e-05,
      "step": 94954,
      "training_step_time": 7.729913949966431
    },
    {
      "epoch": 1.9984245300292967e-05,
      "model_forward_time": 1.7570838928222656,
      "step": 94955
    },
    {
      "epoch": 1.9984245300292967e-05,
      "step": 94955,
      "training_step_time": 8.349636554718018
    },
    {
      "epoch": 1.9985198974609375e-05,
      "model_forward_time": 1.7594895362854004,
      "step": 94956
    },
    {
      "epoch": 1.9985198974609375e-05,
      "step": 94956,
      "training_step_time": 7.6930882930755615
    },
    {
      "epoch": 1.9986152648925782e-05,
      "model_forward_time": 1.7575478553771973,
      "step": 94957
    },
    {
      "epoch": 1.9986152648925782e-05,
      "step": 94957,
      "training_step_time": 7.83364725112915
    },
    {
      "epoch": 1.9987106323242186e-05,
      "model_forward_time": 1.7741365432739258,
      "step": 94958
    },
    {
      "epoch": 1.9987106323242186e-05,
      "step": 94958,
      "training_step_time": 7.93256688117981
    },
    {
      "epoch": 1.9988059997558593e-05,
      "model_forward_time": 2.41097354888916,
      "step": 94959
    },
    {
      "epoch": 1.9988059997558593e-05,
      "step": 94959,
      "training_step_time": 9.570889234542847
    },
    {
      "epoch": 1.9989013671875e-05,
      "grad_norm": 0.06553605943918228,
      "learning_rate": 6.931383972152983e-08,
      "loss": 0.0549,
      "step": 94960
    },
    {
      "epoch": 1.9989013671875e-05,
      "model_forward_time": 1.7949135303497314,
      "step": 94960
    },
    {
      "dynamics_loss_avg": 0.04392481409013271,
      "epoch": 1.9989013671875e-05,
      "step": 94960
    },
    {
      "action_loss_avg": 0.0015968978725140915,
      "epoch": 1.9989013671875e-05,
      "step": 94960
    },
    {
      "epoch": 1.9989013671875e-05,
      "step": 94960,
      "training_step_time": 8.66025447845459
    },
    {
      "epoch": 1.9989967346191405e-05,
      "model_forward_time": 1.7534003257751465,
      "step": 94961
    },
    {
      "epoch": 1.9989967346191405e-05,
      "step": 94961,
      "training_step_time": 8.71827483177185
    },
    {
      "epoch": 1.9990921020507812e-05,
      "model_forward_time": 1.7575409412384033,
      "step": 94962
    },
    {
      "epoch": 1.9990921020507812e-05,
      "step": 94962,
      "training_step_time": 7.780190467834473
    },
    {
      "epoch": 1.999187469482422e-05,
      "model_forward_time": 1.7569997310638428,
      "step": 94963
    },
    {
      "epoch": 1.999187469482422e-05,
      "step": 94963,
      "training_step_time": 8.753838539123535
    },
    {
      "epoch": 1.9992828369140624e-05,
      "model_forward_time": 1.7595648765563965,
      "step": 94964
    },
    {
      "epoch": 1.9992828369140624e-05,
      "step": 94964,
      "training_step_time": 7.385162830352783
    },
    {
      "epoch": 1.999378204345703e-05,
      "model_forward_time": 1.7516305446624756,
      "step": 94965
    },
    {
      "epoch": 1.999378204345703e-05,
      "step": 94965,
      "training_step_time": 8.563161849975586
    },
    {
      "epoch": 1.999473571777344e-05,
      "model_forward_time": 1.7621996402740479,
      "step": 94966
    },
    {
      "epoch": 1.999473571777344e-05,
      "step": 94966,
      "training_step_time": 8.950899839401245
    },
    {
      "epoch": 1.9995689392089843e-05,
      "model_forward_time": 1.7579913139343262,
      "step": 94967
    },
    {
      "epoch": 1.9995689392089843e-05,
      "step": 94967,
      "training_step_time": 8.758462905883789
    },
    {
      "epoch": 1.999664306640625e-05,
      "model_forward_time": 2.385338544845581,
      "step": 94968
    },
    {
      "epoch": 1.999664306640625e-05,
      "step": 94968,
      "training_step_time": 8.734224319458008
    },
    {
      "epoch": 1.9997596740722657e-05,
      "model_forward_time": 1.7620553970336914,
      "step": 94969
    },
    {
      "epoch": 1.9997596740722657e-05,
      "step": 94969,
      "training_step_time": 7.784202575683594
    },
    {
      "epoch": 1.999855041503906e-05,
      "grad_norm": 0.09346827119588852,
      "learning_rate": 6.903974612309128e-08,
      "loss": 0.0546,
      "step": 94970
    },
    {
      "epoch": 1.999855041503906e-05,
      "model_forward_time": 1.7654762268066406,
      "step": 94970
    },
    {
      "dynamics_loss_avg": 0.05561367617920041,
      "epoch": 1.999855041503906e-05,
      "step": 94970
    },
    {
      "action_loss_avg": 0.0011139959970023483,
      "epoch": 1.999855041503906e-05,
      "step": 94970
    },
    {
      "epoch": 1.999855041503906e-05,
      "step": 94970,
      "training_step_time": 8.681565046310425
    },
    {
      "epoch": 1.999950408935547e-05,
      "model_forward_time": 1.7518622875213623,
      "step": 94971
    },
    {
      "epoch": 1.999950408935547e-05,
      "step": 94971,
      "training_step_time": 7.737469434738159
    },
    {
      "epoch": 2.0000457763671876e-05,
      "model_forward_time": 1.7424263954162598,
      "step": 94972
    },
    {
      "epoch": 2.0000457763671876e-05,
      "step": 94972,
      "training_step_time": 7.767829895019531
    },
    {
      "epoch": 2.000141143798828e-05,
      "model_forward_time": 1.7626631259918213,
      "step": 94973
    },
    {
      "epoch": 2.000141143798828e-05,
      "step": 94973,
      "training_step_time": 7.474618911743164
    },
    {
      "epoch": 2.0002365112304688e-05,
      "model_forward_time": 1.7582201957702637,
      "step": 94974
    },
    {
      "epoch": 2.0002365112304688e-05,
      "step": 94974,
      "training_step_time": 8.183960676193237
    },
    {
      "epoch": 2.0003318786621095e-05,
      "model_forward_time": 1.7588622570037842,
      "step": 94975
    },
    {
      "epoch": 2.0003318786621095e-05,
      "step": 94975,
      "training_step_time": 7.656553745269775
    },
    {
      "epoch": 2.00042724609375e-05,
      "model_forward_time": 1.7611546516418457,
      "step": 94976
    },
    {
      "epoch": 2.00042724609375e-05,
      "step": 94976,
      "training_step_time": 8.597479104995728
    },
    {
      "epoch": 2.0005226135253907e-05,
      "model_forward_time": 1.7594401836395264,
      "step": 94977
    },
    {
      "epoch": 2.0005226135253907e-05,
      "step": 94977,
      "training_step_time": 9.081608533859253
    },
    {
      "epoch": 2.0006179809570314e-05,
      "model_forward_time": 2.3944451808929443,
      "step": 94978
    },
    {
      "epoch": 2.0006179809570314e-05,
      "step": 94978,
      "training_step_time": 9.261403560638428
    },
    {
      "epoch": 2.0007133483886718e-05,
      "model_forward_time": 1.7709202766418457,
      "step": 94979
    },
    {
      "epoch": 2.0007133483886718e-05,
      "step": 94979,
      "training_step_time": 7.8288984298706055
    },
    {
      "epoch": 2.0008087158203125e-05,
      "grad_norm": 0.12664780020713806,
      "learning_rate": 6.87661917670518e-08,
      "loss": 0.0546,
      "step": 94980
    },
    {
      "epoch": 2.0008087158203125e-05,
      "model_forward_time": 1.7580695152282715,
      "step": 94980
    },
    {
      "dynamics_loss_avg": 0.04527794988825917,
      "epoch": 2.0008087158203125e-05,
      "step": 94980
    },
    {
      "action_loss_avg": 0.001148080817074515,
      "epoch": 2.0008087158203125e-05,
      "step": 94980
    },
    {
      "epoch": 2.0008087158203125e-05,
      "step": 94980,
      "training_step_time": 7.692647695541382
    },
    {
      "epoch": 2.0009040832519533e-05,
      "model_forward_time": 1.764791488647461,
      "step": 94981
    },
    {
      "epoch": 2.0009040832519533e-05,
      "step": 94981,
      "training_step_time": 8.784160852432251
    },
    {
      "epoch": 2.0009994506835937e-05,
      "model_forward_time": 1.7562119960784912,
      "step": 94982
    },
    {
      "epoch": 2.0009994506835937e-05,
      "step": 94982,
      "training_step_time": 7.760494232177734
    },
    {
      "epoch": 2.0010948181152344e-05,
      "model_forward_time": 1.7613823413848877,
      "step": 94983
    },
    {
      "epoch": 2.0010948181152344e-05,
      "step": 94983,
      "training_step_time": 7.561277151107788
    },
    {
      "epoch": 2.001190185546875e-05,
      "model_forward_time": 1.769820213317871,
      "step": 94984
    },
    {
      "epoch": 2.001190185546875e-05,
      "step": 94984,
      "training_step_time": 7.774243354797363
    },
    {
      "epoch": 2.0012855529785156e-05,
      "model_forward_time": 1.7587172985076904,
      "step": 94985
    },
    {
      "epoch": 2.0012855529785156e-05,
      "step": 94985,
      "training_step_time": 8.751222133636475
    },
    {
      "epoch": 2.0013809204101563e-05,
      "model_forward_time": 1.7579631805419922,
      "step": 94986
    },
    {
      "epoch": 2.0013809204101563e-05,
      "step": 94986,
      "training_step_time": 8.526592493057251
    },
    {
      "epoch": 2.0014762878417967e-05,
      "model_forward_time": 2.4036026000976562,
      "step": 94987
    },
    {
      "epoch": 2.0014762878417967e-05,
      "step": 94987,
      "training_step_time": 9.276358127593994
    },
    {
      "epoch": 2.0015716552734375e-05,
      "model_forward_time": 1.7731919288635254,
      "step": 94988
    },
    {
      "epoch": 2.0015716552734375e-05,
      "step": 94988,
      "training_step_time": 9.046446323394775
    },
    {
      "epoch": 2.0016670227050782e-05,
      "model_forward_time": 1.7559382915496826,
      "step": 94989
    },
    {
      "epoch": 2.0016670227050782e-05,
      "step": 94989,
      "training_step_time": 7.779074668884277
    },
    {
      "epoch": 2.0017623901367186e-05,
      "grad_norm": 0.09646427631378174,
      "learning_rate": 6.8493176683328e-08,
      "loss": 0.0534,
      "step": 94990
    },
    {
      "epoch": 2.0017623901367186e-05,
      "model_forward_time": 1.7607758045196533,
      "step": 94990
    },
    {
      "dynamics_loss_avg": 0.04808995518833399,
      "epoch": 2.0017623901367186e-05,
      "step": 94990
    },
    {
      "action_loss_avg": 0.0009184174647089094,
      "epoch": 2.0017623901367186e-05,
      "step": 94990
    },
    {
      "epoch": 2.0017623901367186e-05,
      "step": 94990,
      "training_step_time": 8.760425090789795
    },
    {
      "epoch": 2.0018577575683593e-05,
      "model_forward_time": 1.742715835571289,
      "step": 94991
    },
    {
      "epoch": 2.0018577575683593e-05,
      "step": 94991,
      "training_step_time": 8.325292110443115
    },
    {
      "epoch": 2.001953125e-05,
      "model_forward_time": 1.752164602279663,
      "step": 94992
    },
    {
      "epoch": 2.001953125e-05,
      "step": 94992,
      "training_step_time": 8.537428617477417
    },
    {
      "epoch": 2.0020484924316405e-05,
      "model_forward_time": 1.7530202865600586,
      "step": 94993
    },
    {
      "epoch": 2.0020484924316405e-05,
      "step": 94993,
      "training_step_time": 8.544912338256836
    },
    {
      "epoch": 2.0021438598632812e-05,
      "model_forward_time": 1.7473280429840088,
      "step": 94994
    },
    {
      "epoch": 2.0021438598632812e-05,
      "step": 94994,
      "training_step_time": 7.73333215713501
    },
    {
      "epoch": 2.002239227294922e-05,
      "model_forward_time": 1.754106044769287,
      "step": 94995
    },
    {
      "epoch": 2.002239227294922e-05,
      "step": 94995,
      "training_step_time": 8.228476285934448
    },
    {
      "epoch": 2.0023345947265624e-05,
      "model_forward_time": 2.3652312755584717,
      "step": 94996
    },
    {
      "epoch": 2.0023345947265624e-05,
      "step": 94996,
      "training_step_time": 7.504881143569946
    },
    {
      "epoch": 2.002429962158203e-05,
      "model_forward_time": 1.7649896144866943,
      "step": 94997
    },
    {
      "epoch": 2.002429962158203e-05,
      "step": 94997,
      "training_step_time": 8.554791927337646
    },
    {
      "epoch": 2.002525329589844e-05,
      "model_forward_time": 1.7755482196807861,
      "step": 94998
    },
    {
      "epoch": 2.002525329589844e-05,
      "step": 94998,
      "training_step_time": 9.161339282989502
    },
    {
      "epoch": 2.0026206970214843e-05,
      "model_forward_time": 1.756169080734253,
      "step": 94999
    },
    {
      "epoch": 2.0026206970214843e-05,
      "step": 94999,
      "training_step_time": 7.723764896392822
    },
    {
      "epoch": 2.002716064453125e-05,
      "grad_norm": 0.06948179006576538,
      "learning_rate": 6.822070090177658e-08,
      "loss": 0.054,
      "step": 95000
    },
    {
      "epoch": 2.002716064453125e-05,
      "model_forward_time": 1.4913146495819092,
      "step": 95000
    },
    {
      "dynamics_loss_avg": 0.06345428545027972,
      "epoch": 2.002716064453125e-05,
      "step": 95000
    },
    {
      "action_loss_avg": 0.0014995118544902653,
      "epoch": 2.002716064453125e-05,
      "step": 95000
    },
    {
      "epoch": 2.002716064453125e-05,
      "step": 95000,
      "training_step_time": 7.3034093379974365
    },
    {
      "epoch": 2.0028114318847657e-05,
      "model_forward_time": 1.7632412910461426,
      "step": 95001
    },
    {
      "epoch": 2.0028114318847657e-05,
      "step": 95001,
      "training_step_time": 7.625559568405151
    },
    {
      "epoch": 2.002906799316406e-05,
      "model_forward_time": 1.7489936351776123,
      "step": 95002
    },
    {
      "epoch": 2.002906799316406e-05,
      "step": 95002,
      "training_step_time": 9.692209720611572
    },
    {
      "epoch": 2.003002166748047e-05,
      "model_forward_time": 1.7631869316101074,
      "step": 95003
    },
    {
      "epoch": 2.003002166748047e-05,
      "step": 95003,
      "training_step_time": 8.921936988830566
    },
    {
      "epoch": 2.0030975341796876e-05,
      "model_forward_time": 1.8101298809051514,
      "step": 95004
    },
    {
      "epoch": 2.0030975341796876e-05,
      "step": 95004,
      "training_step_time": 9.31206202507019
    },
    {
      "epoch": 2.003192901611328e-05,
      "model_forward_time": 3.0543408393859863,
      "step": 95005
    },
    {
      "epoch": 2.003192901611328e-05,
      "step": 95005,
      "training_step_time": 8.634733200073242
    },
    {
      "epoch": 2.0032882690429688e-05,
      "model_forward_time": 1.8896846771240234,
      "step": 95006
    },
    {
      "epoch": 2.0032882690429688e-05,
      "step": 95006,
      "training_step_time": 7.607928276062012
    },
    {
      "epoch": 2.0033836364746095e-05,
      "model_forward_time": 1.7567684650421143,
      "step": 95007
    },
    {
      "epoch": 2.0033836364746095e-05,
      "step": 95007,
      "training_step_time": 7.826865196228027
    },
    {
      "epoch": 2.00347900390625e-05,
      "model_forward_time": 1.754014253616333,
      "step": 95008
    },
    {
      "epoch": 2.00347900390625e-05,
      "step": 95008,
      "training_step_time": 7.5573835372924805
    },
    {
      "epoch": 2.0035743713378907e-05,
      "model_forward_time": 1.7465474605560303,
      "step": 95009
    },
    {
      "epoch": 2.0035743713378907e-05,
      "step": 95009,
      "training_step_time": 7.733073949813843
    },
    {
      "epoch": 2.0036697387695314e-05,
      "grad_norm": 0.09546492248773575,
      "learning_rate": 6.794876445219367e-08,
      "loss": 0.0549,
      "step": 95010
    },
    {
      "epoch": 2.0036697387695314e-05,
      "model_forward_time": 1.7420644760131836,
      "step": 95010
    },
    {
      "dynamics_loss_avg": 0.0514245480298996,
      "epoch": 2.0036697387695314e-05,
      "step": 95010
    },
    {
      "action_loss_avg": 0.0013412486470770091,
      "epoch": 2.0036697387695314e-05,
      "step": 95010
    },
    {
      "epoch": 2.0036697387695314e-05,
      "step": 95010,
      "training_step_time": 7.855971574783325
    },
    {
      "epoch": 2.0037651062011718e-05,
      "model_forward_time": 1.7554495334625244,
      "step": 95011
    },
    {
      "epoch": 2.0037651062011718e-05,
      "step": 95011,
      "training_step_time": 7.476435899734497
    },
    {
      "epoch": 2.0038604736328125e-05,
      "model_forward_time": 1.760798454284668,
      "step": 95012
    },
    {
      "epoch": 2.0038604736328125e-05,
      "step": 95012,
      "training_step_time": 8.628583431243896
    },
    {
      "epoch": 2.0039558410644533e-05,
      "model_forward_time": 1.7562110424041748,
      "step": 95013
    },
    {
      "epoch": 2.0039558410644533e-05,
      "step": 95013,
      "training_step_time": 8.438364028930664
    },
    {
      "epoch": 2.0040512084960937e-05,
      "model_forward_time": 2.4230051040649414,
      "step": 95014
    },
    {
      "epoch": 2.0040512084960937e-05,
      "step": 95014,
      "training_step_time": 8.680325984954834
    },
    {
      "epoch": 2.0041465759277344e-05,
      "model_forward_time": 1.7536485195159912,
      "step": 95015
    },
    {
      "epoch": 2.0041465759277344e-05,
      "step": 95015,
      "training_step_time": 9.613756895065308
    },
    {
      "epoch": 2.004241943359375e-05,
      "model_forward_time": 1.8277688026428223,
      "step": 95016
    },
    {
      "epoch": 2.004241943359375e-05,
      "step": 95016,
      "training_step_time": 8.911547899246216
    },
    {
      "epoch": 2.0043373107910156e-05,
      "model_forward_time": 1.7524354457855225,
      "step": 95017
    },
    {
      "epoch": 2.0043373107910156e-05,
      "step": 95017,
      "training_step_time": 7.737771034240723
    },
    {
      "epoch": 2.0044326782226563e-05,
      "model_forward_time": 1.7575652599334717,
      "step": 95018
    },
    {
      "epoch": 2.0044326782226563e-05,
      "step": 95018,
      "training_step_time": 7.715312480926514
    },
    {
      "epoch": 2.0045280456542967e-05,
      "model_forward_time": 1.7584726810455322,
      "step": 95019
    },
    {
      "epoch": 2.0045280456542967e-05,
      "step": 95019,
      "training_step_time": 7.783870458602905
    },
    {
      "epoch": 2.0046234130859375e-05,
      "grad_norm": 0.09023183584213257,
      "learning_rate": 6.767736736431885e-08,
      "loss": 0.0552,
      "step": 95020
    },
    {
      "epoch": 2.0046234130859375e-05,
      "model_forward_time": 1.7621161937713623,
      "step": 95020
    },
    {
      "dynamics_loss_avg": 0.046088970825076105,
      "epoch": 2.0046234130859375e-05,
      "step": 95020
    },
    {
      "action_loss_avg": 0.0009076486938283778,
      "epoch": 2.0046234130859375e-05,
      "step": 95020
    },
    {
      "epoch": 2.0046234130859375e-05,
      "step": 95020,
      "training_step_time": 7.737352609634399
    },
    {
      "epoch": 2.0047187805175782e-05,
      "model_forward_time": 1.782139539718628,
      "step": 95021
    },
    {
      "epoch": 2.0047187805175782e-05,
      "step": 95021,
      "training_step_time": 7.7041916847229
    },
    {
      "epoch": 2.0048141479492186e-05,
      "model_forward_time": 1.7546749114990234,
      "step": 95022
    },
    {
      "epoch": 2.0048141479492186e-05,
      "step": 95022,
      "training_step_time": 9.320981979370117
    },
    {
      "epoch": 2.0049095153808593e-05,
      "model_forward_time": 2.4306492805480957,
      "step": 95023
    },
    {
      "epoch": 2.0049095153808593e-05,
      "step": 95023,
      "training_step_time": 8.402440547943115
    },
    {
      "epoch": 2.0050048828125e-05,
      "model_forward_time": 1.7532880306243896,
      "step": 95024
    },
    {
      "epoch": 2.0050048828125e-05,
      "step": 95024,
      "training_step_time": 9.18924880027771
    },
    {
      "epoch": 2.0051002502441405e-05,
      "model_forward_time": 1.8141167163848877,
      "step": 95025
    },
    {
      "epoch": 2.0051002502441405e-05,
      "step": 95025,
      "training_step_time": 9.305695533752441
    },
    {
      "epoch": 2.0051956176757812e-05,
      "model_forward_time": 1.7654838562011719,
      "step": 95026
    },
    {
      "epoch": 2.0051956176757812e-05,
      "step": 95026,
      "training_step_time": 7.720639228820801
    },
    {
      "epoch": 2.005290985107422e-05,
      "model_forward_time": 1.7543058395385742,
      "step": 95027
    },
    {
      "epoch": 2.005290985107422e-05,
      "step": 95027,
      "training_step_time": 7.789247751235962
    },
    {
      "epoch": 2.0053863525390624e-05,
      "model_forward_time": 1.7522754669189453,
      "step": 95028
    },
    {
      "epoch": 2.0053863525390624e-05,
      "step": 95028,
      "training_step_time": 7.8117969036102295
    },
    {
      "epoch": 2.005481719970703e-05,
      "model_forward_time": 1.7493431568145752,
      "step": 95029
    },
    {
      "epoch": 2.005481719970703e-05,
      "step": 95029,
      "training_step_time": 7.55372953414917
    },
    {
      "epoch": 2.005577087402344e-05,
      "grad_norm": 0.07633180171251297,
      "learning_rate": 6.740650966783113e-08,
      "loss": 0.0545,
      "step": 95030
    },
    {
      "epoch": 2.005577087402344e-05,
      "model_forward_time": 1.7597262859344482,
      "step": 95030
    },
    {
      "dynamics_loss_avg": 0.055465895868837836,
      "epoch": 2.005577087402344e-05,
      "step": 95030
    },
    {
      "action_loss_avg": 0.0011615241615800186,
      "epoch": 2.005577087402344e-05,
      "step": 95030
    },
    {
      "epoch": 2.005577087402344e-05,
      "step": 95030,
      "training_step_time": 9.008119106292725
    },
    {
      "epoch": 2.0056724548339843e-05,
      "model_forward_time": 1.750363826751709,
      "step": 95031
    },
    {
      "epoch": 2.0056724548339843e-05,
      "step": 95031,
      "training_step_time": 8.49285626411438
    },
    {
      "epoch": 2.005767822265625e-05,
      "model_forward_time": 2.4452028274536133,
      "step": 95032
    },
    {
      "epoch": 2.005767822265625e-05,
      "step": 95032,
      "training_step_time": 9.359144926071167
    },
    {
      "epoch": 2.0058631896972657e-05,
      "model_forward_time": 1.8621735572814941,
      "step": 95033
    },
    {
      "epoch": 2.0058631896972657e-05,
      "step": 95033,
      "training_step_time": 7.7957422733306885
    },
    {
      "epoch": 2.005958557128906e-05,
      "model_forward_time": 1.7546963691711426,
      "step": 95034
    },
    {
      "epoch": 2.005958557128906e-05,
      "step": 95034,
      "training_step_time": 9.657592058181763
    },
    {
      "epoch": 2.006053924560547e-05,
      "model_forward_time": 1.753345251083374,
      "step": 95035
    },
    {
      "epoch": 2.006053924560547e-05,
      "step": 95035,
      "training_step_time": 7.8596031665802
    },
    {
      "epoch": 2.0061492919921876e-05,
      "model_forward_time": 1.7592697143554688,
      "step": 95036
    },
    {
      "epoch": 2.0061492919921876e-05,
      "step": 95036,
      "training_step_time": 8.849465370178223
    },
    {
      "epoch": 2.006244659423828e-05,
      "model_forward_time": 1.7544260025024414,
      "step": 95037
    },
    {
      "epoch": 2.006244659423828e-05,
      "step": 95037,
      "training_step_time": 7.776644468307495
    },
    {
      "epoch": 2.0063400268554688e-05,
      "model_forward_time": 1.7650485038757324,
      "step": 95038
    },
    {
      "epoch": 2.0063400268554688e-05,
      "step": 95038,
      "training_step_time": 7.5191075801849365
    },
    {
      "epoch": 2.0064353942871095e-05,
      "model_forward_time": 1.7585957050323486,
      "step": 95039
    },
    {
      "epoch": 2.0064353942871095e-05,
      "step": 95039,
      "training_step_time": 8.909459590911865
    },
    {
      "epoch": 2.00653076171875e-05,
      "grad_norm": 0.08508438616991043,
      "learning_rate": 6.713619139235239e-08,
      "loss": 0.0541,
      "step": 95040
    },
    {
      "epoch": 2.00653076171875e-05,
      "model_forward_time": 1.7560336589813232,
      "step": 95040
    },
    {
      "dynamics_loss_avg": 0.04698052350431681,
      "epoch": 2.00653076171875e-05,
      "step": 95040
    },
    {
      "action_loss_avg": 0.0009191399527480826,
      "epoch": 2.00653076171875e-05,
      "step": 95040
    },
    {
      "epoch": 2.00653076171875e-05,
      "step": 95040,
      "training_step_time": 9.415830373764038
    },
    {
      "epoch": 2.0066261291503907e-05,
      "model_forward_time": 2.3746559619903564,
      "step": 95041
    },
    {
      "epoch": 2.0066261291503907e-05,
      "step": 95041,
      "training_step_time": 8.780011177062988
    },
    {
      "epoch": 2.0067214965820314e-05,
      "model_forward_time": 1.7461626529693604,
      "step": 95042
    },
    {
      "epoch": 2.0067214965820314e-05,
      "step": 95042,
      "training_step_time": 9.744924068450928
    },
    {
      "epoch": 2.0068168640136718e-05,
      "model_forward_time": 1.7565782070159912,
      "step": 95043
    },
    {
      "epoch": 2.0068168640136718e-05,
      "step": 95043,
      "training_step_time": 8.179917812347412
    },
    {
      "epoch": 2.0069122314453125e-05,
      "model_forward_time": 1.8125078678131104,
      "step": 95044
    },
    {
      "epoch": 2.0069122314453125e-05,
      "step": 95044,
      "training_step_time": 7.952829360961914
    },
    {
      "epoch": 2.0070075988769533e-05,
      "model_forward_time": 1.7639615535736084,
      "step": 95045
    },
    {
      "epoch": 2.0070075988769533e-05,
      "step": 95045,
      "training_step_time": 8.604122161865234
    },
    {
      "epoch": 2.0071029663085937e-05,
      "model_forward_time": 1.760713815689087,
      "step": 95046
    },
    {
      "epoch": 2.0071029663085937e-05,
      "step": 95046,
      "training_step_time": 7.6859166622161865
    },
    {
      "epoch": 2.0071983337402344e-05,
      "model_forward_time": 1.7499279975891113,
      "step": 95047
    },
    {
      "epoch": 2.0071983337402344e-05,
      "step": 95047,
      "training_step_time": 7.7493205070495605
    },
    {
      "epoch": 2.007293701171875e-05,
      "model_forward_time": 1.7563896179199219,
      "step": 95048
    },
    {
      "epoch": 2.007293701171875e-05,
      "step": 95048,
      "training_step_time": 7.660872936248779
    },
    {
      "epoch": 2.0073890686035156e-05,
      "model_forward_time": 1.7445869445800781,
      "step": 95049
    },
    {
      "epoch": 2.0073890686035156e-05,
      "step": 95049,
      "training_step_time": 8.56669569015503
    },
    {
      "epoch": 2.0074844360351563e-05,
      "grad_norm": 0.1559213399887085,
      "learning_rate": 6.686641256744287e-08,
      "loss": 0.056,
      "step": 95050
    },
    {
      "epoch": 2.0074844360351563e-05,
      "model_forward_time": 1.7568767070770264,
      "step": 95050
    },
    {
      "dynamics_loss_avg": 0.06079163709655404,
      "epoch": 2.0074844360351563e-05,
      "step": 95050
    },
    {
      "action_loss_avg": 0.0013469972647726536,
      "epoch": 2.0074844360351563e-05,
      "step": 95050
    },
    {
      "epoch": 2.0074844360351563e-05,
      "step": 95050,
      "training_step_time": 7.610039710998535
    },
    {
      "epoch": 2.0075798034667967e-05,
      "model_forward_time": 2.471846342086792,
      "step": 95051
    },
    {
      "epoch": 2.0075798034667967e-05,
      "step": 95051,
      "training_step_time": 9.583247900009155
    },
    {
      "epoch": 2.0076751708984375e-05,
      "model_forward_time": 1.7492241859436035,
      "step": 95052
    },
    {
      "epoch": 2.0076751708984375e-05,
      "step": 95052,
      "training_step_time": 7.6485373973846436
    },
    {
      "epoch": 2.0077705383300782e-05,
      "model_forward_time": 1.8210325241088867,
      "step": 95053
    },
    {
      "epoch": 2.0077705383300782e-05,
      "step": 95053,
      "training_step_time": 8.221076250076294
    },
    {
      "epoch": 2.0078659057617186e-05,
      "model_forward_time": 1.7490670680999756,
      "step": 95054
    },
    {
      "epoch": 2.0078659057617186e-05,
      "step": 95054,
      "training_step_time": 7.845738410949707
    },
    {
      "epoch": 2.0079612731933593e-05,
      "model_forward_time": 1.3788354396820068,
      "step": 95055
    },
    {
      "epoch": 2.0079612731933593e-05,
      "step": 95055,
      "training_step_time": 7.880813121795654
    },
    {
      "epoch": 2.008056640625e-05,
      "model_forward_time": 1.755089521408081,
      "step": 95056
    },
    {
      "epoch": 2.008056640625e-05,
      "step": 95056,
      "training_step_time": 7.730157136917114
    },
    {
      "epoch": 2.0081520080566405e-05,
      "model_forward_time": 1.7487311363220215,
      "step": 95057
    },
    {
      "epoch": 2.0081520080566405e-05,
      "step": 95057,
      "training_step_time": 7.734876394271851
    },
    {
      "epoch": 2.0082473754882812e-05,
      "model_forward_time": 1.7384724617004395,
      "step": 95058
    },
    {
      "epoch": 2.0082473754882812e-05,
      "step": 95058,
      "training_step_time": 8.699268341064453
    },
    {
      "epoch": 2.008342742919922e-05,
      "model_forward_time": 1.781017780303955,
      "step": 95059
    },
    {
      "epoch": 2.008342742919922e-05,
      "step": 95059,
      "training_step_time": 8.87598443031311
    },
    {
      "epoch": 2.0084381103515624e-05,
      "grad_norm": 0.07949109375476837,
      "learning_rate": 6.659717322260562e-08,
      "loss": 0.0534,
      "step": 95060
    },
    {
      "epoch": 2.0084381103515624e-05,
      "model_forward_time": 2.4075353145599365,
      "step": 95060
    },
    {
      "dynamics_loss_avg": 0.03647380229085684,
      "epoch": 2.0084381103515624e-05,
      "step": 95060
    },
    {
      "action_loss_avg": 0.0006183538353070617,
      "epoch": 2.0084381103515624e-05,
      "step": 95060
    },
    {
      "epoch": 2.0084381103515624e-05,
      "step": 95060,
      "training_step_time": 8.584526062011719
    },
    {
      "epoch": 2.008533477783203e-05,
      "model_forward_time": 1.753849983215332,
      "step": 95061
    },
    {
      "epoch": 2.008533477783203e-05,
      "step": 95061,
      "training_step_time": 8.322971820831299
    },
    {
      "epoch": 2.008628845214844e-05,
      "model_forward_time": 1.806595802307129,
      "step": 95062
    },
    {
      "epoch": 2.008628845214844e-05,
      "step": 95062,
      "training_step_time": 7.89739465713501
    },
    {
      "epoch": 2.0087242126464843e-05,
      "model_forward_time": 1.741039514541626,
      "step": 95063
    },
    {
      "epoch": 2.0087242126464843e-05,
      "step": 95063,
      "training_step_time": 7.784736156463623
    },
    {
      "epoch": 2.008819580078125e-05,
      "model_forward_time": 1.7501966953277588,
      "step": 95064
    },
    {
      "epoch": 2.008819580078125e-05,
      "step": 95064,
      "training_step_time": 7.755495548248291
    },
    {
      "epoch": 2.0089149475097657e-05,
      "model_forward_time": 1.8172607421875,
      "step": 95065
    },
    {
      "epoch": 2.0089149475097657e-05,
      "step": 95065,
      "training_step_time": 7.866117000579834
    },
    {
      "epoch": 2.009010314941406e-05,
      "model_forward_time": 1.752039909362793,
      "step": 95066
    },
    {
      "epoch": 2.009010314941406e-05,
      "step": 95066,
      "training_step_time": 8.533192873001099
    },
    {
      "epoch": 2.009105682373047e-05,
      "model_forward_time": 1.7624239921569824,
      "step": 95067
    },
    {
      "epoch": 2.009105682373047e-05,
      "step": 95067,
      "training_step_time": 8.852518796920776
    },
    {
      "epoch": 2.0092010498046876e-05,
      "model_forward_time": 1.7682850360870361,
      "step": 95068
    },
    {
      "epoch": 2.0092010498046876e-05,
      "step": 95068,
      "training_step_time": 7.772926330566406
    },
    {
      "epoch": 2.009296417236328e-05,
      "model_forward_time": 2.418689250946045,
      "step": 95069
    },
    {
      "epoch": 2.009296417236328e-05,
      "step": 95069,
      "training_step_time": 9.511099100112915
    },
    {
      "epoch": 2.0093917846679688e-05,
      "grad_norm": 0.1373911201953888,
      "learning_rate": 6.632847338728377e-08,
      "loss": 0.0549,
      "step": 95070
    },
    {
      "epoch": 2.0093917846679688e-05,
      "model_forward_time": 1.7490172386169434,
      "step": 95070
    },
    {
      "dynamics_loss_avg": 0.051875578425824645,
      "epoch": 2.0093917846679688e-05,
      "step": 95070
    },
    {
      "action_loss_avg": 0.0012580453578266316,
      "epoch": 2.0093917846679688e-05,
      "step": 95070
    },
    {
      "epoch": 2.0093917846679688e-05,
      "step": 95070,
      "training_step_time": 7.7842326164245605
    },
    {
      "epoch": 2.0094871520996095e-05,
      "model_forward_time": 1.7471754550933838,
      "step": 95071
    },
    {
      "epoch": 2.0094871520996095e-05,
      "step": 95071,
      "training_step_time": 9.200649499893188
    },
    {
      "epoch": 2.00958251953125e-05,
      "model_forward_time": 1.8149676322937012,
      "step": 95072
    },
    {
      "epoch": 2.00958251953125e-05,
      "step": 95072,
      "training_step_time": 7.7532734870910645
    },
    {
      "epoch": 2.0096778869628907e-05,
      "model_forward_time": 1.7729039192199707,
      "step": 95073
    },
    {
      "epoch": 2.0096778869628907e-05,
      "step": 95073,
      "training_step_time": 8.71165919303894
    },
    {
      "epoch": 2.0097732543945314e-05,
      "model_forward_time": 1.756871223449707,
      "step": 95074
    },
    {
      "epoch": 2.0097732543945314e-05,
      "step": 95074,
      "training_step_time": 7.822038173675537
    },
    {
      "epoch": 2.0098686218261718e-05,
      "model_forward_time": 1.7465004920959473,
      "step": 95075
    },
    {
      "epoch": 2.0098686218261718e-05,
      "step": 95075,
      "training_step_time": 7.699318885803223
    },
    {
      "epoch": 2.0099639892578125e-05,
      "model_forward_time": 1.7441928386688232,
      "step": 95076
    },
    {
      "epoch": 2.0099639892578125e-05,
      "step": 95076,
      "training_step_time": 7.798357725143433
    },
    {
      "epoch": 2.0100593566894533e-05,
      "model_forward_time": 1.7461483478546143,
      "step": 95077
    },
    {
      "epoch": 2.0100593566894533e-05,
      "step": 95077,
      "training_step_time": 8.48481798171997
    },
    {
      "epoch": 2.0101547241210937e-05,
      "model_forward_time": 2.426023006439209,
      "step": 95078
    },
    {
      "epoch": 2.0101547241210937e-05,
      "step": 95078,
      "training_step_time": 8.655208587646484
    },
    {
      "epoch": 2.0102500915527344e-05,
      "model_forward_time": 1.7416777610778809,
      "step": 95079
    },
    {
      "epoch": 2.0102500915527344e-05,
      "step": 95079,
      "training_step_time": 11.59838581085205
    },
    {
      "epoch": 2.010345458984375e-05,
      "grad_norm": 0.08326517790555954,
      "learning_rate": 6.606031309086269e-08,
      "loss": 0.0549,
      "step": 95080
    },
    {
      "epoch": 2.010345458984375e-05,
      "model_forward_time": 1.7409098148345947,
      "step": 95080
    },
    {
      "dynamics_loss_avg": 0.06772338654845952,
      "epoch": 2.010345458984375e-05,
      "step": 95080
    },
    {
      "action_loss_avg": 0.001260630731121637,
      "epoch": 2.010345458984375e-05,
      "step": 95080
    },
    {
      "epoch": 2.010345458984375e-05,
      "step": 95080,
      "training_step_time": 6.875657081604004
    },
    {
      "epoch": 2.0104408264160156e-05,
      "model_forward_time": 1.7535862922668457,
      "step": 95081
    },
    {
      "epoch": 2.0104408264160156e-05,
      "step": 95081,
      "training_step_time": 7.743671417236328
    },
    {
      "epoch": 2.0105361938476563e-05,
      "model_forward_time": 1.747901439666748,
      "step": 95082
    },
    {
      "epoch": 2.0105361938476563e-05,
      "step": 95082,
      "training_step_time": 9.02735948562622
    },
    {
      "epoch": 2.0106315612792967e-05,
      "model_forward_time": 1.7420830726623535,
      "step": 95083
    },
    {
      "epoch": 2.0106315612792967e-05,
      "step": 95083,
      "training_step_time": 7.432391881942749
    },
    {
      "epoch": 2.0107269287109375e-05,
      "model_forward_time": 1.7468910217285156,
      "step": 95084
    },
    {
      "epoch": 2.0107269287109375e-05,
      "step": 95084,
      "training_step_time": 6.888241767883301
    },
    {
      "epoch": 2.0108222961425782e-05,
      "model_forward_time": 1.7412049770355225,
      "step": 95085
    },
    {
      "epoch": 2.0108222961425782e-05,
      "step": 95085,
      "training_step_time": 6.878465890884399
    },
    {
      "epoch": 2.0109176635742186e-05,
      "model_forward_time": 1.742692232131958,
      "step": 95086
    },
    {
      "epoch": 2.0109176635742186e-05,
      "step": 95086,
      "training_step_time": 6.866687297821045
    },
    {
      "epoch": 2.0110130310058594e-05,
      "model_forward_time": 1.733001947402954,
      "step": 95087
    },
    {
      "epoch": 2.0110130310058594e-05,
      "step": 95087,
      "training_step_time": 6.858525276184082
    },
    {
      "epoch": 2.0111083984375e-05,
      "model_forward_time": 1.7443504333496094,
      "step": 95088
    },
    {
      "epoch": 2.0111083984375e-05,
      "step": 95088,
      "training_step_time": 6.869580268859863
    },
    {
      "epoch": 2.0112037658691405e-05,
      "model_forward_time": 2.377986192703247,
      "step": 95089
    },
    {
      "epoch": 2.0112037658691405e-05,
      "step": 95089,
      "training_step_time": 7.913369655609131
    },
    {
      "epoch": 2.0112991333007812e-05,
      "grad_norm": 0.16718211770057678,
      "learning_rate": 6.579269236266783e-08,
      "loss": 0.0523,
      "step": 95090
    },
    {
      "epoch": 2.0112991333007812e-05,
      "model_forward_time": 1.7410337924957275,
      "step": 95090
    },
    {
      "dynamics_loss_avg": 0.06441039852797985,
      "epoch": 2.0112991333007812e-05,
      "step": 95090
    },
    {
      "action_loss_avg": 0.0014219327771570534,
      "epoch": 2.0112991333007812e-05,
      "step": 95090
    },
    {
      "epoch": 2.0112991333007812e-05,
      "step": 95090,
      "training_step_time": 7.773529767990112
    },
    {
      "epoch": 2.011394500732422e-05,
      "model_forward_time": 1.7440834045410156,
      "step": 95091
    },
    {
      "epoch": 2.011394500732422e-05,
      "step": 95091,
      "training_step_time": 9.292332887649536
    },
    {
      "epoch": 2.0114898681640624e-05,
      "model_forward_time": 1.7495691776275635,
      "step": 95092
    },
    {
      "epoch": 2.0114898681640624e-05,
      "step": 95092,
      "training_step_time": 7.492564916610718
    },
    {
      "epoch": 2.011585235595703e-05,
      "model_forward_time": 1.7415454387664795,
      "step": 95093
    },
    {
      "epoch": 2.011585235595703e-05,
      "step": 95093,
      "training_step_time": 6.865862846374512
    },
    {
      "epoch": 2.011680603027344e-05,
      "model_forward_time": 1.7502648830413818,
      "step": 95094
    },
    {
      "epoch": 2.011680603027344e-05,
      "step": 95094,
      "training_step_time": 6.8616719245910645
    },
    {
      "epoch": 2.0117759704589843e-05,
      "model_forward_time": 1.737433671951294,
      "step": 95095
    },
    {
      "epoch": 2.0117759704589843e-05,
      "step": 95095,
      "training_step_time": 6.856031656265259
    },
    {
      "epoch": 2.011871337890625e-05,
      "model_forward_time": 1.747798204421997,
      "step": 95096
    },
    {
      "epoch": 2.011871337890625e-05,
      "step": 95096,
      "training_step_time": 6.886234521865845
    },
    {
      "epoch": 2.0119667053222657e-05,
      "model_forward_time": 1.7443783283233643,
      "step": 95097
    },
    {
      "epoch": 2.0119667053222657e-05,
      "step": 95097,
      "training_step_time": 6.8875892162323
    },
    {
      "epoch": 2.012062072753906e-05,
      "model_forward_time": 2.37919282913208,
      "step": 95098
    },
    {
      "epoch": 2.012062072753906e-05,
      "step": 95098,
      "training_step_time": 8.746953248977661
    },
    {
      "epoch": 2.012157440185547e-05,
      "model_forward_time": 1.3915607929229736,
      "step": 95099
    },
    {
      "epoch": 2.012157440185547e-05,
      "step": 95099,
      "training_step_time": 7.822265386581421
    },
    {
      "epoch": 2.0122528076171876e-05,
      "grad_norm": 0.08387381583452225,
      "learning_rate": 6.552561123196632e-08,
      "loss": 0.0535,
      "step": 95100
    },
    {
      "epoch": 2.0122528076171876e-05,
      "model_forward_time": 1.741889476776123,
      "step": 95100
    },
    {
      "dynamics_loss_avg": 0.05723887793719769,
      "epoch": 2.0122528076171876e-05,
      "step": 95100
    },
    {
      "action_loss_avg": 0.000974802440032363,
      "epoch": 2.0122528076171876e-05,
      "step": 95100
    },
    {
      "epoch": 2.0122528076171876e-05,
      "step": 95100,
      "training_step_time": 6.851559638977051
    },
    {
      "epoch": 2.012348175048828e-05,
      "model_forward_time": 1.7569189071655273,
      "step": 95101
    },
    {
      "epoch": 2.012348175048828e-05,
      "step": 95101,
      "training_step_time": 7.79083251953125
    },
    {
      "epoch": 2.0124435424804688e-05,
      "model_forward_time": 1.750633716583252,
      "step": 95102
    },
    {
      "epoch": 2.0124435424804688e-05,
      "step": 95102,
      "training_step_time": 6.874803066253662
    },
    {
      "epoch": 2.0125389099121095e-05,
      "model_forward_time": 1.7516138553619385,
      "step": 95103
    },
    {
      "epoch": 2.0125389099121095e-05,
      "step": 95103,
      "training_step_time": 6.877439022064209
    },
    {
      "epoch": 2.01263427734375e-05,
      "model_forward_time": 1.745124340057373,
      "step": 95104
    },
    {
      "epoch": 2.01263427734375e-05,
      "step": 95104,
      "training_step_time": 6.973772048950195
    },
    {
      "epoch": 2.0127296447753907e-05,
      "model_forward_time": 1.7414331436157227,
      "step": 95105
    },
    {
      "epoch": 2.0127296447753907e-05,
      "step": 95105,
      "training_step_time": 6.826543807983398
    },
    {
      "epoch": 2.0128250122070314e-05,
      "model_forward_time": 1.7424216270446777,
      "step": 95106
    },
    {
      "epoch": 2.0128250122070314e-05,
      "step": 95106,
      "training_step_time": 6.884137153625488
    },
    {
      "epoch": 2.0129203796386718e-05,
      "model_forward_time": 2.4067845344543457,
      "step": 95107
    },
    {
      "epoch": 2.0129203796386718e-05,
      "step": 95107,
      "training_step_time": 7.9147117137908936
    },
    {
      "epoch": 2.0130157470703126e-05,
      "model_forward_time": 1.7425510883331299,
      "step": 95108
    },
    {
      "epoch": 2.0130157470703126e-05,
      "step": 95108,
      "training_step_time": 7.823235273361206
    },
    {
      "epoch": 2.0131111145019533e-05,
      "model_forward_time": 1.7413020133972168,
      "step": 95109
    },
    {
      "epoch": 2.0131111145019533e-05,
      "step": 95109,
      "training_step_time": 6.8770904541015625
    },
    {
      "epoch": 2.0132064819335937e-05,
      "grad_norm": 0.10667439550161362,
      "learning_rate": 6.525906972796426e-08,
      "loss": 0.0545,
      "step": 95110
    },
    {
      "epoch": 2.0132064819335937e-05,
      "model_forward_time": 1.3786776065826416,
      "step": 95110
    },
    {
      "dynamics_loss_avg": 0.03863226026296616,
      "epoch": 2.0132064819335937e-05,
      "step": 95110
    },
    {
      "action_loss_avg": 0.0010477153671672567,
      "epoch": 2.0132064819335937e-05,
      "step": 95110
    },
    {
      "epoch": 2.0132064819335937e-05,
      "step": 95110,
      "training_step_time": 8.653337240219116
    },
    {
      "epoch": 2.0133018493652344e-05,
      "model_forward_time": 1.7465882301330566,
      "step": 95111
    },
    {
      "epoch": 2.0133018493652344e-05,
      "step": 95111,
      "training_step_time": 6.854181289672852
    },
    {
      "epoch": 2.013397216796875e-05,
      "model_forward_time": 1.7475435733795166,
      "step": 95112
    },
    {
      "epoch": 2.013397216796875e-05,
      "step": 95112,
      "training_step_time": 6.913547992706299
    },
    {
      "epoch": 2.0134925842285156e-05,
      "model_forward_time": 1.7867052555084229,
      "step": 95113
    },
    {
      "epoch": 2.0134925842285156e-05,
      "step": 95113,
      "training_step_time": 6.855182886123657
    },
    {
      "epoch": 2.0135879516601563e-05,
      "model_forward_time": 1.7540037631988525,
      "step": 95114
    },
    {
      "epoch": 2.0135879516601563e-05,
      "step": 95114,
      "training_step_time": 6.89611029624939
    },
    {
      "epoch": 2.0136833190917967e-05,
      "model_forward_time": 1.7407963275909424,
      "step": 95115
    },
    {
      "epoch": 2.0136833190917967e-05,
      "step": 95115,
      "training_step_time": 6.8866565227508545
    },
    {
      "epoch": 2.0137786865234375e-05,
      "model_forward_time": 2.4025630950927734,
      "step": 95116
    },
    {
      "epoch": 2.0137786865234375e-05,
      "step": 95116,
      "training_step_time": 10.789979219436646
    },
    {
      "epoch": 2.0138740539550782e-05,
      "model_forward_time": 1.789780616760254,
      "step": 95117
    },
    {
      "epoch": 2.0138740539550782e-05,
      "step": 95117,
      "training_step_time": 7.906229734420776
    },
    {
      "epoch": 2.0139694213867186e-05,
      "model_forward_time": 1.7437348365783691,
      "step": 95118
    },
    {
      "epoch": 2.0139694213867186e-05,
      "step": 95118,
      "training_step_time": 6.862499475479126
    },
    {
      "epoch": 2.0140647888183594e-05,
      "model_forward_time": 1.7423162460327148,
      "step": 95119
    },
    {
      "epoch": 2.0140647888183594e-05,
      "step": 95119,
      "training_step_time": 7.762726068496704
    },
    {
      "epoch": 2.01416015625e-05,
      "grad_norm": 0.07900838553905487,
      "learning_rate": 6.499306787981108e-08,
      "loss": 0.0534,
      "step": 95120
    },
    {
      "epoch": 2.01416015625e-05,
      "model_forward_time": 1.7494816780090332,
      "step": 95120
    },
    {
      "dynamics_loss_avg": 0.060805808380246164,
      "epoch": 2.01416015625e-05,
      "step": 95120
    },
    {
      "action_loss_avg": 0.0017977290495764463,
      "epoch": 2.01416015625e-05,
      "step": 95120
    },
    {
      "epoch": 2.01416015625e-05,
      "step": 95120,
      "training_step_time": 7.389150619506836
    },
    {
      "epoch": 2.0142555236816405e-05,
      "model_forward_time": 1.751499891281128,
      "step": 95121
    },
    {
      "epoch": 2.0142555236816405e-05,
      "step": 95121,
      "training_step_time": 6.867535591125488
    },
    {
      "epoch": 2.0143508911132812e-05,
      "model_forward_time": 1.74393630027771,
      "step": 95122
    },
    {
      "epoch": 2.0143508911132812e-05,
      "step": 95122,
      "training_step_time": 6.828589677810669
    },
    {
      "epoch": 2.014446258544922e-05,
      "model_forward_time": 1.743544340133667,
      "step": 95123
    },
    {
      "epoch": 2.014446258544922e-05,
      "step": 95123,
      "training_step_time": 6.860044717788696
    },
    {
      "epoch": 2.0145416259765624e-05,
      "model_forward_time": 1.7450478076934814,
      "step": 95124
    },
    {
      "epoch": 2.0145416259765624e-05,
      "step": 95124,
      "training_step_time": 6.856597423553467
    },
    {
      "epoch": 2.014636993408203e-05,
      "model_forward_time": 1.790877103805542,
      "step": 95125
    },
    {
      "epoch": 2.014636993408203e-05,
      "step": 95125,
      "training_step_time": 10.109098672866821
    },
    {
      "epoch": 2.014732360839844e-05,
      "model_forward_time": 2.3852245807647705,
      "step": 95126
    },
    {
      "epoch": 2.014732360839844e-05,
      "step": 95126,
      "training_step_time": 7.929114818572998
    },
    {
      "epoch": 2.0148277282714843e-05,
      "model_forward_time": 1.7532622814178467,
      "step": 95127
    },
    {
      "epoch": 2.0148277282714843e-05,
      "step": 95127,
      "training_step_time": 7.6606128215789795
    },
    {
      "epoch": 2.014923095703125e-05,
      "model_forward_time": 1.7563362121582031,
      "step": 95128
    },
    {
      "epoch": 2.014923095703125e-05,
      "step": 95128,
      "training_step_time": 8.47984766960144
    },
    {
      "epoch": 2.0150184631347658e-05,
      "model_forward_time": 1.7442009449005127,
      "step": 95129
    },
    {
      "epoch": 2.0150184631347658e-05,
      "step": 95129,
      "training_step_time": 7.83627724647522
    },
    {
      "epoch": 2.015113830566406e-05,
      "grad_norm": 0.08216343075037003,
      "learning_rate": 6.472760571659631e-08,
      "loss": 0.053,
      "step": 95130
    },
    {
      "epoch": 2.015113830566406e-05,
      "model_forward_time": 1.758143424987793,
      "step": 95130
    },
    {
      "dynamics_loss_avg": 0.049826293252408504,
      "epoch": 2.015113830566406e-05,
      "step": 95130
    },
    {
      "action_loss_avg": 0.0012700989784207195,
      "epoch": 2.015113830566406e-05,
      "step": 95130
    },
    {
      "epoch": 2.015113830566406e-05,
      "step": 95130,
      "training_step_time": 6.868940353393555
    },
    {
      "epoch": 2.015209197998047e-05,
      "model_forward_time": 1.7569477558135986,
      "step": 95131
    },
    {
      "epoch": 2.015209197998047e-05,
      "step": 95131,
      "training_step_time": 6.85808539390564
    },
    {
      "epoch": 2.0153045654296876e-05,
      "model_forward_time": 1.745269536972046,
      "step": 95132
    },
    {
      "epoch": 2.0153045654296876e-05,
      "step": 95132,
      "training_step_time": 6.6744771003723145
    },
    {
      "epoch": 2.015399932861328e-05,
      "model_forward_time": 1.7393817901611328,
      "step": 95133
    },
    {
      "epoch": 2.015399932861328e-05,
      "step": 95133,
      "training_step_time": 6.873963356018066
    },
    {
      "epoch": 2.0154953002929688e-05,
      "model_forward_time": 1.7515268325805664,
      "step": 95134
    },
    {
      "epoch": 2.0154953002929688e-05,
      "step": 95134,
      "training_step_time": 8.801846981048584
    },
    {
      "epoch": 2.0155906677246095e-05,
      "model_forward_time": 2.0104897022247314,
      "step": 95135
    },
    {
      "epoch": 2.0155906677246095e-05,
      "step": 95135,
      "training_step_time": 8.865014553070068
    },
    {
      "epoch": 2.01568603515625e-05,
      "model_forward_time": 1.740837812423706,
      "step": 95136
    },
    {
      "epoch": 2.01568603515625e-05,
      "step": 95136,
      "training_step_time": 7.649868965148926
    },
    {
      "epoch": 2.0157814025878907e-05,
      "model_forward_time": 1.7493565082550049,
      "step": 95137
    },
    {
      "epoch": 2.0157814025878907e-05,
      "step": 95137,
      "training_step_time": 7.565024137496948
    },
    {
      "epoch": 2.0158767700195314e-05,
      "model_forward_time": 1.773423671722412,
      "step": 95138
    },
    {
      "epoch": 2.0158767700195314e-05,
      "step": 95138,
      "training_step_time": 7.767486333847046
    },
    {
      "epoch": 2.0159721374511718e-05,
      "model_forward_time": 1.754082202911377,
      "step": 95139
    },
    {
      "epoch": 2.0159721374511718e-05,
      "step": 95139,
      "training_step_time": 7.1999311447143555
    },
    {
      "epoch": 2.0160675048828126e-05,
      "grad_norm": 0.08580312877893448,
      "learning_rate": 6.446268326735061e-08,
      "loss": 0.0544,
      "step": 95140
    },
    {
      "epoch": 2.0160675048828126e-05,
      "model_forward_time": 1.75559401512146,
      "step": 95140
    },
    {
      "dynamics_loss_avg": 0.06663832794874906,
      "epoch": 2.0160675048828126e-05,
      "step": 95140
    },
    {
      "action_loss_avg": 0.0016123705863719806,
      "epoch": 2.0160675048828126e-05,
      "step": 95140
    },
    {
      "epoch": 2.0160675048828126e-05,
      "step": 95140,
      "training_step_time": 6.881099700927734
    },
    {
      "epoch": 2.016162872314453e-05,
      "model_forward_time": 1.7593615055084229,
      "step": 95141
    },
    {
      "epoch": 2.016162872314453e-05,
      "step": 95141,
      "training_step_time": 6.909067153930664
    },
    {
      "epoch": 2.0162582397460937e-05,
      "model_forward_time": 1.7619121074676514,
      "step": 95142
    },
    {
      "epoch": 2.0162582397460937e-05,
      "step": 95142,
      "training_step_time": 6.873647212982178
    },
    {
      "epoch": 2.0163536071777344e-05,
      "model_forward_time": 1.7704229354858398,
      "step": 95143
    },
    {
      "epoch": 2.0163536071777344e-05,
      "step": 95143,
      "training_step_time": 8.365766286849976
    },
    {
      "epoch": 2.016448974609375e-05,
      "model_forward_time": 2.4200751781463623,
      "step": 95144
    },
    {
      "epoch": 2.016448974609375e-05,
      "step": 95144,
      "training_step_time": 8.474559307098389
    },
    {
      "epoch": 2.0165443420410156e-05,
      "model_forward_time": 1.76161789894104,
      "step": 95145
    },
    {
      "epoch": 2.0165443420410156e-05,
      "step": 95145,
      "training_step_time": 7.872933864593506
    },
    {
      "epoch": 2.0166397094726563e-05,
      "model_forward_time": 1.7547705173492432,
      "step": 95146
    },
    {
      "epoch": 2.0166397094726563e-05,
      "step": 95146,
      "training_step_time": 7.878731727600098
    },
    {
      "epoch": 2.0167350769042967e-05,
      "model_forward_time": 1.7617387771606445,
      "step": 95147
    },
    {
      "epoch": 2.0167350769042967e-05,
      "step": 95147,
      "training_step_time": 7.791021347045898
    },
    {
      "epoch": 2.0168304443359375e-05,
      "model_forward_time": 1.7545843124389648,
      "step": 95148
    },
    {
      "epoch": 2.0168304443359375e-05,
      "step": 95148,
      "training_step_time": 7.381130218505859
    },
    {
      "epoch": 2.0169258117675782e-05,
      "model_forward_time": 1.7487106323242188,
      "step": 95149
    },
    {
      "epoch": 2.0169258117675782e-05,
      "step": 95149,
      "training_step_time": 6.722807168960571
    },
    {
      "epoch": 2.0170211791992186e-05,
      "grad_norm": 0.1568227857351303,
      "learning_rate": 6.419830056104582e-08,
      "loss": 0.0536,
      "step": 95150
    },
    {
      "epoch": 2.0170211791992186e-05,
      "model_forward_time": 1.7683360576629639,
      "step": 95150
    },
    {
      "dynamics_loss_avg": 0.0567289162427187,
      "epoch": 2.0170211791992186e-05,
      "step": 95150
    },
    {
      "action_loss_avg": 0.0011594187788432464,
      "epoch": 2.0170211791992186e-05,
      "step": 95150
    },
    {
      "epoch": 2.0170211791992186e-05,
      "step": 95150,
      "training_step_time": 6.817621231079102
    },
    {
      "epoch": 2.0171165466308594e-05,
      "model_forward_time": 1.7533106803894043,
      "step": 95151
    },
    {
      "epoch": 2.0171165466308594e-05,
      "step": 95151,
      "training_step_time": 6.838748931884766
    },
    {
      "epoch": 2.0172119140625e-05,
      "model_forward_time": 1.7445402145385742,
      "step": 95152
    },
    {
      "epoch": 2.0172119140625e-05,
      "step": 95152,
      "training_step_time": 7.80476713180542
    },
    {
      "epoch": 2.0173072814941405e-05,
      "model_forward_time": 2.409794807434082,
      "step": 95153
    },
    {
      "epoch": 2.0173072814941405e-05,
      "step": 95153,
      "training_step_time": 9.641260862350464
    },
    {
      "epoch": 2.0174026489257812e-05,
      "model_forward_time": 1.7549140453338623,
      "step": 95154
    },
    {
      "epoch": 2.0174026489257812e-05,
      "step": 95154,
      "training_step_time": 7.828613042831421
    },
    {
      "epoch": 2.017498016357422e-05,
      "model_forward_time": 1.7699222564697266,
      "step": 95155
    },
    {
      "epoch": 2.017498016357422e-05,
      "step": 95155,
      "training_step_time": 6.80277419090271
    },
    {
      "epoch": 2.0175933837890624e-05,
      "model_forward_time": 1.7705905437469482,
      "step": 95156
    },
    {
      "epoch": 2.0175933837890624e-05,
      "step": 95156,
      "training_step_time": 7.595024824142456
    },
    {
      "epoch": 2.017688751220703e-05,
      "model_forward_time": 1.7728757858276367,
      "step": 95157
    },
    {
      "epoch": 2.017688751220703e-05,
      "step": 95157,
      "training_step_time": 7.58890438079834
    },
    {
      "epoch": 2.017784118652344e-05,
      "model_forward_time": 1.7611844539642334,
      "step": 95158
    },
    {
      "epoch": 2.017784118652344e-05,
      "step": 95158,
      "training_step_time": 6.94222092628479
    },
    {
      "epoch": 2.0178794860839843e-05,
      "model_forward_time": 1.7660157680511475,
      "step": 95159
    },
    {
      "epoch": 2.0178794860839843e-05,
      "step": 95159,
      "training_step_time": 6.84261417388916
    },
    {
      "epoch": 2.017974853515625e-05,
      "grad_norm": 0.09156142920255661,
      "learning_rate": 6.393445762659379e-08,
      "loss": 0.0519,
      "step": 95160
    },
    {
      "epoch": 2.017974853515625e-05,
      "model_forward_time": 1.8508210182189941,
      "step": 95160
    },
    {
      "dynamics_loss_avg": 0.06911996081471443,
      "epoch": 2.017974853515625e-05,
      "step": 95160
    },
    {
      "action_loss_avg": 0.003050333505962044,
      "epoch": 2.017974853515625e-05,
      "step": 95160
    },
    {
      "epoch": 2.017974853515625e-05,
      "step": 95160,
      "training_step_time": 6.844933986663818
    },
    {
      "epoch": 2.0180702209472658e-05,
      "model_forward_time": 1.8063185214996338,
      "step": 95161
    },
    {
      "epoch": 2.0180702209472658e-05,
      "step": 95161,
      "training_step_time": 8.864222526550293
    },
    {
      "epoch": 2.018165588378906e-05,
      "model_forward_time": 1.7775304317474365,
      "step": 95162
    },
    {
      "epoch": 2.018165588378906e-05,
      "step": 95162,
      "training_step_time": 10.989692687988281
    },
    {
      "epoch": 2.018260955810547e-05,
      "model_forward_time": 2.411691904067993,
      "step": 95163
    },
    {
      "epoch": 2.018260955810547e-05,
      "step": 95163,
      "training_step_time": 8.458808183670044
    },
    {
      "epoch": 2.0183563232421876e-05,
      "model_forward_time": 1.7662854194641113,
      "step": 95164
    },
    {
      "epoch": 2.0183563232421876e-05,
      "step": 95164,
      "training_step_time": 7.653778076171875
    },
    {
      "epoch": 2.018451690673828e-05,
      "model_forward_time": 1.7704780101776123,
      "step": 95165
    },
    {
      "epoch": 2.018451690673828e-05,
      "step": 95165,
      "training_step_time": 7.602013349533081
    },
    {
      "epoch": 2.0185470581054688e-05,
      "model_forward_time": 1.8023650646209717,
      "step": 95166
    },
    {
      "epoch": 2.0185470581054688e-05,
      "step": 95166,
      "training_step_time": 7.785643100738525
    },
    {
      "epoch": 2.0186424255371095e-05,
      "model_forward_time": 1.7667555809020996,
      "step": 95167
    },
    {
      "epoch": 2.0186424255371095e-05,
      "step": 95167,
      "training_step_time": 7.340528726577759
    },
    {
      "epoch": 2.01873779296875e-05,
      "model_forward_time": 1.776533603668213,
      "step": 95168
    },
    {
      "epoch": 2.01873779296875e-05,
      "step": 95168,
      "training_step_time": 6.891059637069702
    },
    {
      "epoch": 2.0188331604003907e-05,
      "model_forward_time": 1.76997709274292,
      "step": 95169
    },
    {
      "epoch": 2.0188331604003907e-05,
      "step": 95169,
      "training_step_time": 6.85368275642395
    },
    {
      "epoch": 2.0189285278320314e-05,
      "grad_norm": 0.0902087464928627,
      "learning_rate": 6.367115449284811e-08,
      "loss": 0.0535,
      "step": 95170
    },
    {
      "epoch": 2.0189285278320314e-05,
      "model_forward_time": 1.7746779918670654,
      "step": 95170
    },
    {
      "dynamics_loss_avg": 0.06788248531520366,
      "epoch": 2.0189285278320314e-05,
      "step": 95170
    },
    {
      "action_loss_avg": 0.0012990957882720977,
      "epoch": 2.0189285278320314e-05,
      "step": 95170
    },
    {
      "epoch": 2.0189285278320314e-05,
      "step": 95170,
      "training_step_time": 8.678713321685791
    },
    {
      "epoch": 2.0190238952636718e-05,
      "model_forward_time": 1.7729969024658203,
      "step": 95171
    },
    {
      "epoch": 2.0190238952636718e-05,
      "step": 95171,
      "training_step_time": 10.530729055404663
    },
    {
      "epoch": 2.0191192626953126e-05,
      "model_forward_time": 2.3984386920928955,
      "step": 95172
    },
    {
      "epoch": 2.0191192626953126e-05,
      "step": 95172,
      "training_step_time": 11.384277820587158
    },
    {
      "epoch": 2.019214630126953e-05,
      "model_forward_time": 1.7741868495941162,
      "step": 95173
    },
    {
      "epoch": 2.019214630126953e-05,
      "step": 95173,
      "training_step_time": 7.568983316421509
    },
    {
      "epoch": 2.0193099975585937e-05,
      "model_forward_time": 1.7608540058135986,
      "step": 95174
    },
    {
      "epoch": 2.0193099975585937e-05,
      "step": 95174,
      "training_step_time": 8.01213026046753
    },
    {
      "epoch": 2.0194053649902344e-05,
      "model_forward_time": 1.773427963256836,
      "step": 95175
    },
    {
      "epoch": 2.0194053649902344e-05,
      "step": 95175,
      "training_step_time": 7.712579727172852
    },
    {
      "epoch": 2.019500732421875e-05,
      "model_forward_time": 1.8536186218261719,
      "step": 95176
    },
    {
      "epoch": 2.019500732421875e-05,
      "step": 95176,
      "training_step_time": 7.369312524795532
    },
    {
      "epoch": 2.0195960998535156e-05,
      "model_forward_time": 1.7598483562469482,
      "step": 95177
    },
    {
      "epoch": 2.0195960998535156e-05,
      "step": 95177,
      "training_step_time": 6.815596580505371
    },
    {
      "epoch": 2.0196914672851563e-05,
      "model_forward_time": 1.7533938884735107,
      "step": 95178
    },
    {
      "epoch": 2.0196914672851563e-05,
      "step": 95178,
      "training_step_time": 6.847691297531128
    },
    {
      "epoch": 2.0197868347167967e-05,
      "model_forward_time": 1.759340524673462,
      "step": 95179
    },
    {
      "epoch": 2.0197868347167967e-05,
      "step": 95179,
      "training_step_time": 7.7024827003479
    },
    {
      "epoch": 2.0198822021484375e-05,
      "grad_norm": 0.0876903086900711,
      "learning_rate": 6.340839118860299e-08,
      "loss": 0.0525,
      "step": 95180
    },
    {
      "epoch": 2.0198822021484375e-05,
      "model_forward_time": 1.7556588649749756,
      "step": 95180
    },
    {
      "dynamics_loss_avg": 0.0663249060511589,
      "epoch": 2.0198822021484375e-05,
      "step": 95180
    },
    {
      "action_loss_avg": 0.0011720066686393693,
      "epoch": 2.0198822021484375e-05,
      "step": 95180
    },
    {
      "epoch": 2.0198822021484375e-05,
      "step": 95180,
      "training_step_time": 8.587822198867798
    },
    {
      "epoch": 2.0199775695800782e-05,
      "model_forward_time": 2.4396800994873047,
      "step": 95181
    },
    {
      "epoch": 2.0199775695800782e-05,
      "step": 95181,
      "training_step_time": 9.487133502960205
    },
    {
      "epoch": 2.0200729370117186e-05,
      "model_forward_time": 1.751016616821289,
      "step": 95182
    },
    {
      "epoch": 2.0200729370117186e-05,
      "step": 95182,
      "training_step_time": 7.665457487106323
    },
    {
      "epoch": 2.0201683044433594e-05,
      "model_forward_time": 1.7513985633850098,
      "step": 95183
    },
    {
      "epoch": 2.0201683044433594e-05,
      "step": 95183,
      "training_step_time": 7.376923561096191
    },
    {
      "epoch": 2.020263671875e-05,
      "model_forward_time": 1.396376132965088,
      "step": 95184
    },
    {
      "epoch": 2.020263671875e-05,
      "step": 95184,
      "training_step_time": 7.571290016174316
    },
    {
      "epoch": 2.0203590393066405e-05,
      "model_forward_time": 1.75079345703125,
      "step": 95185
    },
    {
      "epoch": 2.0203590393066405e-05,
      "step": 95185,
      "training_step_time": 7.233405828475952
    },
    {
      "epoch": 2.0204544067382812e-05,
      "model_forward_time": 1.7533783912658691,
      "step": 95186
    },
    {
      "epoch": 2.0204544067382812e-05,
      "step": 95186,
      "training_step_time": 6.674331426620483
    },
    {
      "epoch": 2.020549774169922e-05,
      "model_forward_time": 1.7455377578735352,
      "step": 95187
    },
    {
      "epoch": 2.020549774169922e-05,
      "step": 95187,
      "training_step_time": 6.730935096740723
    },
    {
      "epoch": 2.0206451416015624e-05,
      "model_forward_time": 1.7491424083709717,
      "step": 95188
    },
    {
      "epoch": 2.0206451416015624e-05,
      "step": 95188,
      "training_step_time": 7.7536585330963135
    },
    {
      "epoch": 2.020740509033203e-05,
      "model_forward_time": 1.7521936893463135,
      "step": 95189
    },
    {
      "epoch": 2.020740509033203e-05,
      "step": 95189,
      "training_step_time": 7.746086597442627
    },
    {
      "epoch": 2.020835876464844e-05,
      "grad_norm": 0.08812319487333298,
      "learning_rate": 6.314616774259486e-08,
      "loss": 0.0546,
      "step": 95190
    },
    {
      "epoch": 2.020835876464844e-05,
      "model_forward_time": 2.409651756286621,
      "step": 95190
    },
    {
      "dynamics_loss_avg": 0.06006758296862245,
      "epoch": 2.020835876464844e-05,
      "step": 95190
    },
    {
      "action_loss_avg": 0.0012272677675355225,
      "epoch": 2.020835876464844e-05,
      "step": 95190
    },
    {
      "epoch": 2.020835876464844e-05,
      "step": 95190,
      "training_step_time": 9.80092716217041
    },
    {
      "epoch": 2.0209312438964843e-05,
      "model_forward_time": 1.7620172500610352,
      "step": 95191
    },
    {
      "epoch": 2.0209312438964843e-05,
      "step": 95191,
      "training_step_time": 7.72318959236145
    },
    {
      "epoch": 2.021026611328125e-05,
      "model_forward_time": 1.7867846488952637,
      "step": 95192
    },
    {
      "epoch": 2.021026611328125e-05,
      "step": 95192,
      "training_step_time": 7.8929359912872314
    },
    {
      "epoch": 2.0211219787597658e-05,
      "model_forward_time": 1.7661066055297852,
      "step": 95193
    },
    {
      "epoch": 2.0211219787597658e-05,
      "step": 95193,
      "training_step_time": 7.481173515319824
    },
    {
      "epoch": 2.021217346191406e-05,
      "model_forward_time": 1.7554826736450195,
      "step": 95194
    },
    {
      "epoch": 2.021217346191406e-05,
      "step": 95194,
      "training_step_time": 7.555514097213745
    },
    {
      "epoch": 2.021312713623047e-05,
      "model_forward_time": 1.7628638744354248,
      "step": 95195
    },
    {
      "epoch": 2.021312713623047e-05,
      "step": 95195,
      "training_step_time": 6.741485595703125
    },
    {
      "epoch": 2.0214080810546876e-05,
      "model_forward_time": 1.7889094352722168,
      "step": 95196
    },
    {
      "epoch": 2.0214080810546876e-05,
      "step": 95196,
      "training_step_time": 6.73521876335144
    },
    {
      "epoch": 2.021503448486328e-05,
      "model_forward_time": 1.74993896484375,
      "step": 95197
    },
    {
      "epoch": 2.021503448486328e-05,
      "step": 95197,
      "training_step_time": 7.258844375610352
    },
    {
      "epoch": 2.0215988159179688e-05,
      "model_forward_time": 1.7468841075897217,
      "step": 95198
    },
    {
      "epoch": 2.0215988159179688e-05,
      "step": 95198,
      "training_step_time": 7.728780746459961
    },
    {
      "epoch": 2.0216941833496095e-05,
      "model_forward_time": 1.3872008323669434,
      "step": 95199
    },
    {
      "epoch": 2.0216941833496095e-05,
      "step": 95199,
      "training_step_time": 9.79918646812439
    },
    {
      "epoch": 2.02178955078125e-05,
      "grad_norm": 0.07344813644886017,
      "learning_rate": 6.288448418349858e-08,
      "loss": 0.0531,
      "step": 95200
    },
    {
      "epoch": 2.02178955078125e-05,
      "model_forward_time": 2.4504714012145996,
      "step": 95200
    },
    {
      "dynamics_loss_avg": 0.05797392576932907,
      "epoch": 2.02178955078125e-05,
      "step": 95200
    },
    {
      "action_loss_avg": 0.001662039029179141,
      "epoch": 2.02178955078125e-05,
      "step": 95200
    },
    {
      "epoch": 2.02178955078125e-05,
      "step": 95200,
      "training_step_time": 7.685102462768555
    },
    {
      "epoch": 2.0218849182128907e-05,
      "model_forward_time": 1.7501182556152344,
      "step": 95201
    },
    {
      "epoch": 2.0218849182128907e-05,
      "step": 95201,
      "training_step_time": 7.354982614517212
    },
    {
      "epoch": 2.0219802856445314e-05,
      "model_forward_time": 1.7507851123809814,
      "step": 95202
    },
    {
      "epoch": 2.0219802856445314e-05,
      "step": 95202,
      "training_step_time": 6.987748146057129
    },
    {
      "epoch": 2.0220756530761718e-05,
      "model_forward_time": 1.763169527053833,
      "step": 95203
    },
    {
      "epoch": 2.0220756530761718e-05,
      "step": 95203,
      "training_step_time": 7.523353099822998
    },
    {
      "epoch": 2.0221710205078126e-05,
      "model_forward_time": 1.7473554611206055,
      "step": 95204
    },
    {
      "epoch": 2.0221710205078126e-05,
      "step": 95204,
      "training_step_time": 6.73073935508728
    },
    {
      "epoch": 2.022266387939453e-05,
      "model_forward_time": 1.7585368156433105,
      "step": 95205
    },
    {
      "epoch": 2.022266387939453e-05,
      "step": 95205,
      "training_step_time": 6.706167936325073
    },
    {
      "epoch": 2.0223617553710937e-05,
      "model_forward_time": 1.754429578781128,
      "step": 95206
    },
    {
      "epoch": 2.0223617553710937e-05,
      "step": 95206,
      "training_step_time": 6.708934783935547
    },
    {
      "epoch": 2.0224571228027344e-05,
      "model_forward_time": 1.7460265159606934,
      "step": 95207
    },
    {
      "epoch": 2.0224571228027344e-05,
      "step": 95207,
      "training_step_time": 9.015175104141235
    },
    {
      "epoch": 2.022552490234375e-05,
      "model_forward_time": 1.7535085678100586,
      "step": 95208
    },
    {
      "epoch": 2.022552490234375e-05,
      "step": 95208,
      "training_step_time": 11.61466360092163
    },
    {
      "epoch": 2.0226478576660156e-05,
      "model_forward_time": 2.4505889415740967,
      "step": 95209
    },
    {
      "epoch": 2.0226478576660156e-05,
      "step": 95209,
      "training_step_time": 9.388733863830566
    },
    {
      "epoch": 2.0227432250976563e-05,
      "grad_norm": 0.07991564273834229,
      "learning_rate": 6.262334053993236e-08,
      "loss": 0.054,
      "step": 95210
    },
    {
      "epoch": 2.0227432250976563e-05,
      "model_forward_time": 1.7523784637451172,
      "step": 95210
    },
    {
      "dynamics_loss_avg": 0.04461768064647913,
      "epoch": 2.0227432250976563e-05,
      "step": 95210
    },
    {
      "action_loss_avg": 0.00145695600513136,
      "epoch": 2.0227432250976563e-05,
      "step": 95210
    },
    {
      "epoch": 2.0227432250976563e-05,
      "step": 95210,
      "training_step_time": 7.415419340133667
    },
    {
      "epoch": 2.0228385925292967e-05,
      "model_forward_time": 1.7515459060668945,
      "step": 95211
    },
    {
      "epoch": 2.0228385925292967e-05,
      "step": 95211,
      "training_step_time": 7.092983722686768
    },
    {
      "epoch": 2.0229339599609375e-05,
      "model_forward_time": 1.7449629306793213,
      "step": 95212
    },
    {
      "epoch": 2.0229339599609375e-05,
      "step": 95212,
      "training_step_time": 7.554366827011108
    },
    {
      "epoch": 2.0230293273925782e-05,
      "model_forward_time": 1.7594449520111084,
      "step": 95213
    },
    {
      "epoch": 2.0230293273925782e-05,
      "step": 95213,
      "training_step_time": 6.696079969406128
    },
    {
      "epoch": 2.0231246948242186e-05,
      "model_forward_time": 1.3879005908966064,
      "step": 95214
    },
    {
      "epoch": 2.0231246948242186e-05,
      "step": 95214,
      "training_step_time": 6.723050117492676
    },
    {
      "epoch": 2.0232200622558594e-05,
      "model_forward_time": 1.762127161026001,
      "step": 95215
    },
    {
      "epoch": 2.0232200622558594e-05,
      "step": 95215,
      "training_step_time": 6.729190111160278
    },
    {
      "epoch": 2.0233154296875e-05,
      "model_forward_time": 1.7586910724639893,
      "step": 95216
    },
    {
      "epoch": 2.0233154296875e-05,
      "step": 95216,
      "training_step_time": 8.557677507400513
    },
    {
      "epoch": 2.0234107971191405e-05,
      "model_forward_time": 1.7510952949523926,
      "step": 95217
    },
    {
      "epoch": 2.0234107971191405e-05,
      "step": 95217,
      "training_step_time": 9.626714944839478
    },
    {
      "epoch": 2.0235061645507812e-05,
      "model_forward_time": 2.43713116645813,
      "step": 95218
    },
    {
      "epoch": 2.0235061645507812e-05,
      "step": 95218,
      "training_step_time": 7.75797438621521
    },
    {
      "epoch": 2.023601531982422e-05,
      "model_forward_time": 1.7557792663574219,
      "step": 95219
    },
    {
      "epoch": 2.023601531982422e-05,
      "step": 95219,
      "training_step_time": 7.574926853179932
    },
    {
      "epoch": 2.0236968994140624e-05,
      "grad_norm": 0.10312546789646149,
      "learning_rate": 6.236273684045502e-08,
      "loss": 0.055,
      "step": 95220
    },
    {
      "epoch": 2.0236968994140624e-05,
      "model_forward_time": 1.7632834911346436,
      "step": 95220
    },
    {
      "dynamics_loss_avg": 0.06179278921335936,
      "epoch": 2.0236968994140624e-05,
      "step": 95220
    },
    {
      "action_loss_avg": 0.0014275366818765178,
      "epoch": 2.0236968994140624e-05,
      "step": 95220
    },
    {
      "epoch": 2.0236968994140624e-05,
      "step": 95220,
      "training_step_time": 7.376513481140137
    },
    {
      "epoch": 2.023792266845703e-05,
      "model_forward_time": 1.7444453239440918,
      "step": 95221
    },
    {
      "epoch": 2.023792266845703e-05,
      "step": 95221,
      "training_step_time": 7.589428424835205
    },
    {
      "epoch": 2.023887634277344e-05,
      "model_forward_time": 1.7526109218597412,
      "step": 95222
    },
    {
      "epoch": 2.023887634277344e-05,
      "step": 95222,
      "training_step_time": 7.206172704696655
    },
    {
      "epoch": 2.0239830017089843e-05,
      "model_forward_time": 1.7399744987487793,
      "step": 95223
    },
    {
      "epoch": 2.0239830017089843e-05,
      "step": 95223,
      "training_step_time": 6.71032977104187
    },
    {
      "epoch": 2.024078369140625e-05,
      "model_forward_time": 1.7393803596496582,
      "step": 95224
    },
    {
      "epoch": 2.024078369140625e-05,
      "step": 95224,
      "training_step_time": 6.721675872802734
    },
    {
      "epoch": 2.0241737365722658e-05,
      "model_forward_time": 1.7440180778503418,
      "step": 95225
    },
    {
      "epoch": 2.0241737365722658e-05,
      "step": 95225,
      "training_step_time": 8.358914852142334
    },
    {
      "epoch": 2.024269104003906e-05,
      "model_forward_time": 1.7486486434936523,
      "step": 95226
    },
    {
      "epoch": 2.024269104003906e-05,
      "step": 95226,
      "training_step_time": 9.583628177642822
    },
    {
      "epoch": 2.024364471435547e-05,
      "model_forward_time": 2.4173877239227295,
      "step": 95227
    },
    {
      "epoch": 2.024364471435547e-05,
      "step": 95227,
      "training_step_time": 9.611339330673218
    },
    {
      "epoch": 2.0244598388671876e-05,
      "model_forward_time": 1.7433722019195557,
      "step": 95228
    },
    {
      "epoch": 2.0244598388671876e-05,
      "step": 95228,
      "training_step_time": 7.596076011657715
    },
    {
      "epoch": 2.024555206298828e-05,
      "model_forward_time": 1.7490921020507812,
      "step": 95229
    },
    {
      "epoch": 2.024555206298828e-05,
      "step": 95229,
      "training_step_time": 7.416037559509277
    },
    {
      "epoch": 2.0246505737304688e-05,
      "grad_norm": 0.08023906499147415,
      "learning_rate": 6.210267311356489e-08,
      "loss": 0.054,
      "step": 95230
    },
    {
      "epoch": 2.0246505737304688e-05,
      "model_forward_time": 1.7418763637542725,
      "step": 95230
    },
    {
      "dynamics_loss_avg": 0.0444095415994525,
      "epoch": 2.0246505737304688e-05,
      "step": 95230
    },
    {
      "action_loss_avg": 0.0010388733586296438,
      "epoch": 2.0246505737304688e-05,
      "step": 95230
    },
    {
      "epoch": 2.0246505737304688e-05,
      "step": 95230,
      "training_step_time": 7.62063193321228
    },
    {
      "epoch": 2.0247459411621095e-05,
      "model_forward_time": 1.7490389347076416,
      "step": 95231
    },
    {
      "epoch": 2.0247459411621095e-05,
      "step": 95231,
      "training_step_time": 7.44997763633728
    },
    {
      "epoch": 2.02484130859375e-05,
      "model_forward_time": 1.7489442825317383,
      "step": 95232
    },
    {
      "epoch": 2.02484130859375e-05,
      "step": 95232,
      "training_step_time": 6.693262577056885
    },
    {
      "epoch": 2.0249366760253907e-05,
      "model_forward_time": 1.7463843822479248,
      "step": 95233
    },
    {
      "epoch": 2.0249366760253907e-05,
      "step": 95233,
      "training_step_time": 6.66006875038147
    },
    {
      "epoch": 2.0250320434570314e-05,
      "model_forward_time": 1.7552144527435303,
      "step": 95234
    },
    {
      "epoch": 2.0250320434570314e-05,
      "step": 95234,
      "training_step_time": 8.087684869766235
    },
    {
      "epoch": 2.0251274108886718e-05,
      "model_forward_time": 1.7508649826049805,
      "step": 95235
    },
    {
      "epoch": 2.0251274108886718e-05,
      "step": 95235,
      "training_step_time": 8.650426626205444
    },
    {
      "epoch": 2.0252227783203126e-05,
      "model_forward_time": 2.4381611347198486,
      "step": 95236
    },
    {
      "epoch": 2.0252227783203126e-05,
      "step": 95236,
      "training_step_time": 11.14795970916748
    },
    {
      "epoch": 2.025318145751953e-05,
      "model_forward_time": 1.8166701793670654,
      "step": 95237
    },
    {
      "epoch": 2.025318145751953e-05,
      "step": 95237,
      "training_step_time": 7.70824408531189
    },
    {
      "epoch": 2.0254135131835937e-05,
      "model_forward_time": 1.743802547454834,
      "step": 95238
    },
    {
      "epoch": 2.0254135131835937e-05,
      "step": 95238,
      "training_step_time": 7.512152194976807
    },
    {
      "epoch": 2.0255088806152344e-05,
      "model_forward_time": 1.7492942810058594,
      "step": 95239
    },
    {
      "epoch": 2.0255088806152344e-05,
      "step": 95239,
      "training_step_time": 7.759526014328003
    },
    {
      "epoch": 2.025604248046875e-05,
      "grad_norm": 0.1150670275092125,
      "learning_rate": 6.184314938770197e-08,
      "loss": 0.0523,
      "step": 95240
    },
    {
      "epoch": 2.025604248046875e-05,
      "model_forward_time": 1.7415196895599365,
      "step": 95240
    },
    {
      "dynamics_loss_avg": 0.06526636648923159,
      "epoch": 2.025604248046875e-05,
      "step": 95240
    },
    {
      "action_loss_avg": 0.0012743993749609217,
      "epoch": 2.025604248046875e-05,
      "step": 95240
    },
    {
      "epoch": 2.025604248046875e-05,
      "step": 95240,
      "training_step_time": 7.620743036270142
    },
    {
      "epoch": 2.0256996154785156e-05,
      "model_forward_time": 1.75982666015625,
      "step": 95241
    },
    {
      "epoch": 2.0256996154785156e-05,
      "step": 95241,
      "training_step_time": 6.822309732437134
    },
    {
      "epoch": 2.0257949829101563e-05,
      "model_forward_time": 1.758173942565918,
      "step": 95242
    },
    {
      "epoch": 2.0257949829101563e-05,
      "step": 95242,
      "training_step_time": 6.895206928253174
    },
    {
      "epoch": 2.0258903503417967e-05,
      "model_forward_time": 1.7581629753112793,
      "step": 95243
    },
    {
      "epoch": 2.0258903503417967e-05,
      "step": 95243,
      "training_step_time": 7.624559164047241
    },
    {
      "epoch": 2.0259857177734375e-05,
      "model_forward_time": 1.755310297012329,
      "step": 95244
    },
    {
      "epoch": 2.0259857177734375e-05,
      "step": 95244,
      "training_step_time": 7.774606704711914
    },
    {
      "epoch": 2.0260810852050782e-05,
      "model_forward_time": 2.4230282306671143,
      "step": 95245
    },
    {
      "epoch": 2.0260810852050782e-05,
      "step": 95245,
      "training_step_time": 8.806538105010986
    },
    {
      "epoch": 2.0261764526367186e-05,
      "model_forward_time": 1.8222851753234863,
      "step": 95246
    },
    {
      "epoch": 2.0261764526367186e-05,
      "step": 95246,
      "training_step_time": 8.566136598587036
    },
    {
      "epoch": 2.0262718200683594e-05,
      "model_forward_time": 1.3921997547149658,
      "step": 95247
    },
    {
      "epoch": 2.0262718200683594e-05,
      "step": 95247,
      "training_step_time": 8.372440338134766
    },
    {
      "epoch": 2.0263671875e-05,
      "model_forward_time": 1.7508678436279297,
      "step": 95248
    },
    {
      "epoch": 2.0263671875e-05,
      "step": 95248,
      "training_step_time": 7.18877100944519
    },
    {
      "epoch": 2.0264625549316405e-05,
      "model_forward_time": 1.7523102760314941,
      "step": 95249
    },
    {
      "epoch": 2.0264625549316405e-05,
      "step": 95249,
      "training_step_time": 7.688331842422485
    },
    {
      "epoch": 2.0265579223632812e-05,
      "grad_norm": 0.09627746790647507,
      "learning_rate": 6.158416569124859e-08,
      "loss": 0.0553,
      "step": 95250
    },
    {
      "epoch": 2.0265579223632812e-05,
      "model_forward_time": 1.7894206047058105,
      "step": 95250
    },
    {
      "dynamics_loss_avg": 0.05660057887434959,
      "epoch": 2.0265579223632812e-05,
      "step": 95250
    },
    {
      "action_loss_avg": 0.001117762306239456,
      "epoch": 2.0265579223632812e-05,
      "step": 95250
    },
    {
      "epoch": 2.0265579223632812e-05,
      "step": 95250,
      "training_step_time": 6.82369065284729
    },
    {
      "epoch": 2.026653289794922e-05,
      "model_forward_time": 1.7522344589233398,
      "step": 95251
    },
    {
      "epoch": 2.026653289794922e-05,
      "step": 95251,
      "training_step_time": 6.8318541049957275
    },
    {
      "epoch": 2.0267486572265624e-05,
      "model_forward_time": 1.7603576183319092,
      "step": 95252
    },
    {
      "epoch": 2.0267486572265624e-05,
      "step": 95252,
      "training_step_time": 7.486461877822876
    },
    {
      "epoch": 2.026844024658203e-05,
      "model_forward_time": 1.7554006576538086,
      "step": 95253
    },
    {
      "epoch": 2.026844024658203e-05,
      "step": 95253,
      "training_step_time": 10.056162357330322
    },
    {
      "epoch": 2.026939392089844e-05,
      "model_forward_time": 1.7399697303771973,
      "step": 95254
    },
    {
      "epoch": 2.026939392089844e-05,
      "step": 95254,
      "training_step_time": 8.870596647262573
    },
    {
      "epoch": 2.0270347595214843e-05,
      "model_forward_time": 2.4678127765655518,
      "step": 95255
    },
    {
      "epoch": 2.0270347595214843e-05,
      "step": 95255,
      "training_step_time": 8.387755155563354
    },
    {
      "epoch": 2.027130126953125e-05,
      "model_forward_time": 1.751645565032959,
      "step": 95256
    },
    {
      "epoch": 2.027130126953125e-05,
      "step": 95256,
      "training_step_time": 8.497893810272217
    },
    {
      "epoch": 2.0272254943847658e-05,
      "model_forward_time": 1.7577095031738281,
      "step": 95257
    },
    {
      "epoch": 2.0272254943847658e-05,
      "step": 95257,
      "training_step_time": 7.4371256828308105
    },
    {
      "epoch": 2.027320861816406e-05,
      "model_forward_time": 1.7541685104370117,
      "step": 95258
    },
    {
      "epoch": 2.027320861816406e-05,
      "step": 95258,
      "training_step_time": 7.496644973754883
    },
    {
      "epoch": 2.027416229248047e-05,
      "model_forward_time": 1.7437360286712646,
      "step": 95259
    },
    {
      "epoch": 2.027416229248047e-05,
      "step": 95259,
      "training_step_time": 7.562226295471191
    },
    {
      "epoch": 2.0275115966796876e-05,
      "grad_norm": 0.15477535128593445,
      "learning_rate": 6.13257220525254e-08,
      "loss": 0.0539,
      "step": 95260
    },
    {
      "epoch": 2.0275115966796876e-05,
      "model_forward_time": 1.7486300468444824,
      "step": 95260
    },
    {
      "dynamics_loss_avg": 0.04934129379689693,
      "epoch": 2.0275115966796876e-05,
      "step": 95260
    },
    {
      "action_loss_avg": 0.0008054391699261032,
      "epoch": 2.0275115966796876e-05,
      "step": 95260
    },
    {
      "epoch": 2.0275115966796876e-05,
      "step": 95260,
      "training_step_time": 6.760746479034424
    },
    {
      "epoch": 2.027606964111328e-05,
      "model_forward_time": 1.7501695156097412,
      "step": 95261
    },
    {
      "epoch": 2.027606964111328e-05,
      "step": 95261,
      "training_step_time": 6.8474273681640625
    },
    {
      "epoch": 2.0277023315429688e-05,
      "model_forward_time": 1.7552547454833984,
      "step": 95262
    },
    {
      "epoch": 2.0277023315429688e-05,
      "step": 95262,
      "training_step_time": 8.320010423660278
    },
    {
      "epoch": 2.0277976989746095e-05,
      "model_forward_time": 1.7395813465118408,
      "step": 95263
    },
    {
      "epoch": 2.0277976989746095e-05,
      "step": 95263,
      "training_step_time": 9.909149646759033
    },
    {
      "epoch": 2.02789306640625e-05,
      "model_forward_time": 2.4405369758605957,
      "step": 95264
    },
    {
      "epoch": 2.02789306640625e-05,
      "step": 95264,
      "training_step_time": 10.46391487121582
    },
    {
      "epoch": 2.0279884338378907e-05,
      "model_forward_time": 1.7477760314941406,
      "step": 95265
    },
    {
      "epoch": 2.0279884338378907e-05,
      "step": 95265,
      "training_step_time": 7.7238078117370605
    },
    {
      "epoch": 2.028083801269531e-05,
      "model_forward_time": 1.7477986812591553,
      "step": 95266
    },
    {
      "epoch": 2.028083801269531e-05,
      "step": 95266,
      "training_step_time": 7.487307786941528
    },
    {
      "epoch": 2.0281791687011718e-05,
      "model_forward_time": 1.749509572982788,
      "step": 95267
    },
    {
      "epoch": 2.0281791687011718e-05,
      "step": 95267,
      "training_step_time": 7.235381841659546
    },
    {
      "epoch": 2.0282745361328126e-05,
      "model_forward_time": 1.7426085472106934,
      "step": 95268
    },
    {
      "epoch": 2.0282745361328126e-05,
      "step": 95268,
      "training_step_time": 7.73991847038269
    },
    {
      "epoch": 2.028369903564453e-05,
      "model_forward_time": 1.7384915351867676,
      "step": 95269
    },
    {
      "epoch": 2.028369903564453e-05,
      "step": 95269,
      "training_step_time": 7.30007791519165
    },
    {
      "epoch": 2.0284652709960937e-05,
      "grad_norm": 0.09844716638326645,
      "learning_rate": 6.106781849979648e-08,
      "loss": 0.0548,
      "step": 95270
    },
    {
      "epoch": 2.0284652709960937e-05,
      "model_forward_time": 1.391829013824463,
      "step": 95270
    },
    {
      "dynamics_loss_avg": 0.05731036327779293,
      "epoch": 2.0284652709960937e-05,
      "step": 95270
    },
    {
      "action_loss_avg": 0.0016790351161034777,
      "epoch": 2.0284652709960937e-05,
      "step": 95270
    },
    {
      "epoch": 2.0284652709960937e-05,
      "step": 95270,
      "training_step_time": 6.707339763641357
    },
    {
      "epoch": 2.0285606384277345e-05,
      "model_forward_time": 1.7514336109161377,
      "step": 95271
    },
    {
      "epoch": 2.0285606384277345e-05,
      "step": 95271,
      "training_step_time": 7.791840076446533
    },
    {
      "epoch": 2.028656005859375e-05,
      "model_forward_time": 1.7391843795776367,
      "step": 95272
    },
    {
      "epoch": 2.028656005859375e-05,
      "step": 95272,
      "training_step_time": 9.485591173171997
    },
    {
      "epoch": 2.0287513732910156e-05,
      "model_forward_time": 2.4411442279815674,
      "step": 95273
    },
    {
      "epoch": 2.0287513732910156e-05,
      "step": 95273,
      "training_step_time": 9.928373098373413
    },
    {
      "epoch": 2.0288467407226563e-05,
      "model_forward_time": 1.7495601177215576,
      "step": 95274
    },
    {
      "epoch": 2.0288467407226563e-05,
      "step": 95274,
      "training_step_time": 7.736364841461182
    },
    {
      "epoch": 2.0289421081542967e-05,
      "model_forward_time": 1.7481689453125,
      "step": 95275
    },
    {
      "epoch": 2.0289421081542967e-05,
      "step": 95275,
      "training_step_time": 8.157757997512817
    },
    {
      "epoch": 2.0290374755859375e-05,
      "model_forward_time": 1.7439947128295898,
      "step": 95276
    },
    {
      "epoch": 2.0290374755859375e-05,
      "step": 95276,
      "training_step_time": 7.553984642028809
    },
    {
      "epoch": 2.0291328430175782e-05,
      "model_forward_time": 1.7421166896820068,
      "step": 95277
    },
    {
      "epoch": 2.0291328430175782e-05,
      "step": 95277,
      "training_step_time": 7.804739952087402
    },
    {
      "epoch": 2.0292282104492186e-05,
      "model_forward_time": 1.7529242038726807,
      "step": 95278
    },
    {
      "epoch": 2.0292282104492186e-05,
      "step": 95278,
      "training_step_time": 7.368358135223389
    },
    {
      "epoch": 2.0293235778808594e-05,
      "model_forward_time": 1.7515838146209717,
      "step": 95279
    },
    {
      "epoch": 2.0293235778808594e-05,
      "step": 95279,
      "training_step_time": 6.83350682258606
    },
    {
      "epoch": 2.0294189453125e-05,
      "grad_norm": 0.07924848049879074,
      "learning_rate": 6.081045506126482e-08,
      "loss": 0.0547,
      "step": 95280
    },
    {
      "epoch": 2.0294189453125e-05,
      "model_forward_time": 1.7525804042816162,
      "step": 95280
    },
    {
      "dynamics_loss_avg": 0.06134310457855463,
      "epoch": 2.0294189453125e-05,
      "step": 95280
    },
    {
      "action_loss_avg": 0.0013608203065814451,
      "epoch": 2.0294189453125e-05,
      "step": 95280
    },
    {
      "epoch": 2.0294189453125e-05,
      "step": 95280,
      "training_step_time": 7.777880430221558
    },
    {
      "epoch": 2.0295143127441405e-05,
      "model_forward_time": 1.7367992401123047,
      "step": 95281
    },
    {
      "epoch": 2.0295143127441405e-05,
      "step": 95281,
      "training_step_time": 7.8413519859313965
    },
    {
      "epoch": 2.0296096801757813e-05,
      "model_forward_time": 2.3998115062713623,
      "step": 95282
    },
    {
      "epoch": 2.0296096801757813e-05,
      "step": 95282,
      "training_step_time": 10.419399976730347
    },
    {
      "epoch": 2.029705047607422e-05,
      "model_forward_time": 1.808650255203247,
      "step": 95283
    },
    {
      "epoch": 2.029705047607422e-05,
      "step": 95283,
      "training_step_time": 8.956969499588013
    },
    {
      "epoch": 2.0298004150390624e-05,
      "model_forward_time": 1.7464790344238281,
      "step": 95284
    },
    {
      "epoch": 2.0298004150390624e-05,
      "step": 95284,
      "training_step_time": 7.721743822097778
    },
    {
      "epoch": 2.029895782470703e-05,
      "model_forward_time": 1.7460954189300537,
      "step": 95285
    },
    {
      "epoch": 2.029895782470703e-05,
      "step": 95285,
      "training_step_time": 7.972324848175049
    },
    {
      "epoch": 2.029991149902344e-05,
      "model_forward_time": 1.7426962852478027,
      "step": 95286
    },
    {
      "epoch": 2.029991149902344e-05,
      "step": 95286,
      "training_step_time": 7.3904290199279785
    },
    {
      "epoch": 2.0300865173339843e-05,
      "model_forward_time": 1.7830631732940674,
      "step": 95287
    },
    {
      "epoch": 2.0300865173339843e-05,
      "step": 95287,
      "training_step_time": 7.824848890304565
    },
    {
      "epoch": 2.030181884765625e-05,
      "model_forward_time": 1.7410898208618164,
      "step": 95288
    },
    {
      "epoch": 2.030181884765625e-05,
      "step": 95288,
      "training_step_time": 6.802311897277832
    },
    {
      "epoch": 2.0302772521972658e-05,
      "model_forward_time": 1.7544786930084229,
      "step": 95289
    },
    {
      "epoch": 2.0302772521972658e-05,
      "step": 95289,
      "training_step_time": 7.760737895965576
    },
    {
      "epoch": 2.030372619628906e-05,
      "grad_norm": 0.07280463725328445,
      "learning_rate": 6.055363176507678e-08,
      "loss": 0.0527,
      "step": 95290
    },
    {
      "epoch": 2.030372619628906e-05,
      "model_forward_time": 1.7512569427490234,
      "step": 95290
    },
    {
      "dynamics_loss_avg": 0.03968159407377243,
      "epoch": 2.030372619628906e-05,
      "step": 95290
    },
    {
      "action_loss_avg": 0.0014124265959253534,
      "epoch": 2.030372619628906e-05,
      "step": 95290
    },
    {
      "epoch": 2.030372619628906e-05,
      "step": 95290,
      "training_step_time": 8.59089732170105
    },
    {
      "epoch": 2.030467987060547e-05,
      "model_forward_time": 1.7572433948516846,
      "step": 95291
    },
    {
      "epoch": 2.030467987060547e-05,
      "step": 95291,
      "training_step_time": 10.297999620437622
    },
    {
      "epoch": 2.0305633544921877e-05,
      "model_forward_time": 1.8081271648406982,
      "step": 95292
    },
    {
      "epoch": 2.0305633544921877e-05,
      "step": 95292,
      "training_step_time": 7.83225154876709
    },
    {
      "epoch": 2.030658721923828e-05,
      "model_forward_time": 1.7493693828582764,
      "step": 95293
    },
    {
      "epoch": 2.030658721923828e-05,
      "step": 95293,
      "training_step_time": 7.7480292320251465
    },
    {
      "epoch": 2.0307540893554688e-05,
      "model_forward_time": 1.7475378513336182,
      "step": 95294
    },
    {
      "epoch": 2.0307540893554688e-05,
      "step": 95294,
      "training_step_time": 8.021095037460327
    },
    {
      "epoch": 2.0308494567871095e-05,
      "model_forward_time": 1.7559502124786377,
      "step": 95295
    },
    {
      "epoch": 2.0308494567871095e-05,
      "step": 95295,
      "training_step_time": 7.683401823043823
    },
    {
      "epoch": 2.03094482421875e-05,
      "model_forward_time": 1.7564904689788818,
      "step": 95296
    },
    {
      "epoch": 2.03094482421875e-05,
      "step": 95296,
      "training_step_time": 7.711691856384277
    },
    {
      "epoch": 2.0310401916503907e-05,
      "model_forward_time": 1.7529923915863037,
      "step": 95297
    },
    {
      "epoch": 2.0310401916503907e-05,
      "step": 95297,
      "training_step_time": 6.878939867019653
    },
    {
      "epoch": 2.031135559082031e-05,
      "model_forward_time": 1.7520298957824707,
      "step": 95298
    },
    {
      "epoch": 2.031135559082031e-05,
      "step": 95298,
      "training_step_time": 7.7642176151275635
    },
    {
      "epoch": 2.0312309265136718e-05,
      "model_forward_time": 1.7478713989257812,
      "step": 95299
    },
    {
      "epoch": 2.0312309265136718e-05,
      "step": 95299,
      "training_step_time": 7.918591499328613
    },
    {
      "epoch": 2.0313262939453126e-05,
      "grad_norm": 0.10335706919431686,
      "learning_rate": 6.029734863931602e-08,
      "loss": 0.0542,
      "step": 95300
    },
    {
      "epoch": 2.0313262939453126e-05,
      "model_forward_time": 1.7528595924377441,
      "step": 95300
    },
    {
      "dynamics_loss_avg": 0.0623175673186779,
      "epoch": 2.0313262939453126e-05,
      "step": 95300
    },
    {
      "action_loss_avg": 0.0011328027321724221,
      "epoch": 2.0313262939453126e-05,
      "step": 95300
    },
    {
      "epoch": 2.0313262939453126e-05,
      "step": 95300,
      "training_step_time": 13.021148920059204
    },
    {
      "epoch": 2.031421661376953e-05,
      "model_forward_time": 2.4261233806610107,
      "step": 95301
    },
    {
      "epoch": 2.031421661376953e-05,
      "step": 95301,
      "training_step_time": 8.926002979278564
    },
    {
      "epoch": 2.0315170288085937e-05,
      "model_forward_time": 1.7497799396514893,
      "step": 95302
    },
    {
      "epoch": 2.0315170288085937e-05,
      "step": 95302,
      "training_step_time": 7.802784442901611
    },
    {
      "epoch": 2.0316123962402345e-05,
      "model_forward_time": 1.7542145252227783,
      "step": 95303
    },
    {
      "epoch": 2.0316123962402345e-05,
      "step": 95303,
      "training_step_time": 7.5673668384552
    },
    {
      "epoch": 2.031707763671875e-05,
      "model_forward_time": 1.748138189315796,
      "step": 95304
    },
    {
      "epoch": 2.031707763671875e-05,
      "step": 95304,
      "training_step_time": 7.609283208847046
    },
    {
      "epoch": 2.0318031311035156e-05,
      "model_forward_time": 1.748931646347046,
      "step": 95305
    },
    {
      "epoch": 2.0318031311035156e-05,
      "step": 95305,
      "training_step_time": 7.752947807312012
    },
    {
      "epoch": 2.0318984985351563e-05,
      "model_forward_time": 1.7521538734436035,
      "step": 95306
    },
    {
      "epoch": 2.0318984985351563e-05,
      "step": 95306,
      "training_step_time": 7.370781898498535
    },
    {
      "epoch": 2.0319938659667967e-05,
      "model_forward_time": 1.7561538219451904,
      "step": 95307
    },
    {
      "epoch": 2.0319938659667967e-05,
      "step": 95307,
      "training_step_time": 7.775507926940918
    },
    {
      "epoch": 2.0320892333984375e-05,
      "model_forward_time": 1.7500073909759521,
      "step": 95308
    },
    {
      "epoch": 2.0320892333984375e-05,
      "step": 95308,
      "training_step_time": 9.681299686431885
    },
    {
      "epoch": 2.0321846008300782e-05,
      "model_forward_time": 1.751885175704956,
      "step": 95309
    },
    {
      "epoch": 2.0321846008300782e-05,
      "step": 95309,
      "training_step_time": 9.44925832748413
    },
    {
      "epoch": 2.0322799682617186e-05,
      "grad_norm": 0.08284275978803635,
      "learning_rate": 6.004160571201179e-08,
      "loss": 0.0534,
      "step": 95310
    },
    {
      "epoch": 2.0322799682617186e-05,
      "model_forward_time": 2.395566463470459,
      "step": 95310
    },
    {
      "dynamics_loss_avg": 0.04961089156568051,
      "epoch": 2.0322799682617186e-05,
      "step": 95310
    },
    {
      "action_loss_avg": 0.001074454069021158,
      "epoch": 2.0322799682617186e-05,
      "step": 95310
    },
    {
      "epoch": 2.0322799682617186e-05,
      "step": 95310,
      "training_step_time": 11.591809749603271
    },
    {
      "epoch": 2.0323753356933594e-05,
      "model_forward_time": 1.7503221035003662,
      "step": 95311
    },
    {
      "epoch": 2.0323753356933594e-05,
      "step": 95311,
      "training_step_time": 8.652455806732178
    },
    {
      "epoch": 2.032470703125e-05,
      "model_forward_time": 1.7901811599731445,
      "step": 95312
    },
    {
      "epoch": 2.032470703125e-05,
      "step": 95312,
      "training_step_time": 7.6876561641693115
    },
    {
      "epoch": 2.0325660705566405e-05,
      "model_forward_time": 1.7492153644561768,
      "step": 95313
    },
    {
      "epoch": 2.0325660705566405e-05,
      "step": 95313,
      "training_step_time": 7.641299724578857
    },
    {
      "epoch": 2.0326614379882813e-05,
      "model_forward_time": 1.7537257671356201,
      "step": 95314
    },
    {
      "epoch": 2.0326614379882813e-05,
      "step": 95314,
      "training_step_time": 7.79787278175354
    },
    {
      "epoch": 2.032756805419922e-05,
      "model_forward_time": 1.748274803161621,
      "step": 95315
    },
    {
      "epoch": 2.032756805419922e-05,
      "step": 95315,
      "training_step_time": 7.359063625335693
    },
    {
      "epoch": 2.0328521728515624e-05,
      "model_forward_time": 1.791116714477539,
      "step": 95316
    },
    {
      "epoch": 2.0328521728515624e-05,
      "step": 95316,
      "training_step_time": 7.741441249847412
    },
    {
      "epoch": 2.032947540283203e-05,
      "model_forward_time": 1.7542338371276855,
      "step": 95317
    },
    {
      "epoch": 2.032947540283203e-05,
      "step": 95317,
      "training_step_time": 7.7980287075042725
    },
    {
      "epoch": 2.033042907714844e-05,
      "model_forward_time": 1.7494642734527588,
      "step": 95318
    },
    {
      "epoch": 2.033042907714844e-05,
      "step": 95318,
      "training_step_time": 8.57515287399292
    },
    {
      "epoch": 2.0331382751464843e-05,
      "model_forward_time": 2.440762519836426,
      "step": 95319
    },
    {
      "epoch": 2.0331382751464843e-05,
      "step": 95319,
      "training_step_time": 9.797050714492798
    },
    {
      "epoch": 2.033233642578125e-05,
      "grad_norm": 0.0848965048789978,
      "learning_rate": 5.978640301112948e-08,
      "loss": 0.0529,
      "step": 95320
    },
    {
      "epoch": 2.033233642578125e-05,
      "model_forward_time": 1.8262157440185547,
      "step": 95320
    },
    {
      "dynamics_loss_avg": 0.03540470865555108,
      "epoch": 2.033233642578125e-05,
      "step": 95320
    },
    {
      "action_loss_avg": 0.002155088243307546,
      "epoch": 2.033233642578125e-05,
      "step": 95320
    },
    {
      "epoch": 2.033233642578125e-05,
      "step": 95320,
      "training_step_time": 7.789379358291626
    },
    {
      "epoch": 2.0333290100097658e-05,
      "model_forward_time": 1.7557194232940674,
      "step": 95321
    },
    {
      "epoch": 2.0333290100097658e-05,
      "step": 95321,
      "training_step_time": 7.632972478866577
    },
    {
      "epoch": 2.0334243774414062e-05,
      "model_forward_time": 1.7534480094909668,
      "step": 95322
    },
    {
      "epoch": 2.0334243774414062e-05,
      "step": 95322,
      "training_step_time": 7.552545070648193
    },
    {
      "epoch": 2.033519744873047e-05,
      "model_forward_time": 1.7468812465667725,
      "step": 95323
    },
    {
      "epoch": 2.033519744873047e-05,
      "step": 95323,
      "training_step_time": 7.7531538009643555
    },
    {
      "epoch": 2.0336151123046877e-05,
      "model_forward_time": 1.7538256645202637,
      "step": 95324
    },
    {
      "epoch": 2.0336151123046877e-05,
      "step": 95324,
      "training_step_time": 7.292426109313965
    },
    {
      "epoch": 2.033710479736328e-05,
      "model_forward_time": 1.753159761428833,
      "step": 95325
    },
    {
      "epoch": 2.033710479736328e-05,
      "step": 95325,
      "training_step_time": 7.729214668273926
    },
    {
      "epoch": 2.0338058471679688e-05,
      "model_forward_time": 1.7526497840881348,
      "step": 95326
    },
    {
      "epoch": 2.0338058471679688e-05,
      "step": 95326,
      "training_step_time": 8.482866525650024
    },
    {
      "epoch": 2.0339012145996095e-05,
      "model_forward_time": 1.757396936416626,
      "step": 95327
    },
    {
      "epoch": 2.0339012145996095e-05,
      "step": 95327,
      "training_step_time": 7.822816371917725
    },
    {
      "epoch": 2.03399658203125e-05,
      "model_forward_time": 2.416984796524048,
      "step": 95328
    },
    {
      "epoch": 2.03399658203125e-05,
      "step": 95328,
      "training_step_time": 14.38088870048523
    },
    {
      "epoch": 2.0340919494628907e-05,
      "model_forward_time": 1.418114185333252,
      "step": 95329
    },
    {
      "epoch": 2.0340919494628907e-05,
      "step": 95329,
      "training_step_time": 8.164042949676514
    },
    {
      "epoch": 2.034187316894531e-05,
      "grad_norm": 0.09649362415075302,
      "learning_rate": 5.9531740564579e-08,
      "loss": 0.0539,
      "step": 95330
    },
    {
      "epoch": 2.034187316894531e-05,
      "model_forward_time": 1.7588329315185547,
      "step": 95330
    },
    {
      "dynamics_loss_avg": 0.05452530272305012,
      "epoch": 2.034187316894531e-05,
      "step": 95330
    },
    {
      "action_loss_avg": 0.0015530754491919652,
      "epoch": 2.034187316894531e-05,
      "step": 95330
    },
    {
      "epoch": 2.034187316894531e-05,
      "step": 95330,
      "training_step_time": 7.661839962005615
    },
    {
      "epoch": 2.0342826843261718e-05,
      "model_forward_time": 1.76027512550354,
      "step": 95331
    },
    {
      "epoch": 2.0342826843261718e-05,
      "step": 95331,
      "training_step_time": 7.372663736343384
    },
    {
      "epoch": 2.0343780517578126e-05,
      "model_forward_time": 1.7821834087371826,
      "step": 95332
    },
    {
      "epoch": 2.0343780517578126e-05,
      "step": 95332,
      "training_step_time": 7.580821752548218
    },
    {
      "epoch": 2.034473419189453e-05,
      "model_forward_time": 1.763333797454834,
      "step": 95333
    },
    {
      "epoch": 2.034473419189453e-05,
      "step": 95333,
      "training_step_time": 7.482667446136475
    },
    {
      "epoch": 2.0345687866210937e-05,
      "model_forward_time": 1.7601654529571533,
      "step": 95334
    },
    {
      "epoch": 2.0345687866210937e-05,
      "step": 95334,
      "training_step_time": 6.7338478565216064
    },
    {
      "epoch": 2.0346641540527345e-05,
      "model_forward_time": 1.7632579803466797,
      "step": 95335
    },
    {
      "epoch": 2.0346641540527345e-05,
      "step": 95335,
      "training_step_time": 7.621929168701172
    },
    {
      "epoch": 2.034759521484375e-05,
      "model_forward_time": 1.7594728469848633,
      "step": 95336
    },
    {
      "epoch": 2.034759521484375e-05,
      "step": 95336,
      "training_step_time": 8.403250455856323
    },
    {
      "epoch": 2.0348548889160156e-05,
      "model_forward_time": 1.7584929466247559,
      "step": 95337
    },
    {
      "epoch": 2.0348548889160156e-05,
      "step": 95337,
      "training_step_time": 9.816298723220825
    },
    {
      "epoch": 2.0349502563476563e-05,
      "model_forward_time": 2.4349253177642822,
      "step": 95338
    },
    {
      "epoch": 2.0349502563476563e-05,
      "step": 95338,
      "training_step_time": 7.70872163772583
    },
    {
      "epoch": 2.0350456237792967e-05,
      "model_forward_time": 1.7722105979919434,
      "step": 95339
    },
    {
      "epoch": 2.0350456237792967e-05,
      "step": 95339,
      "training_step_time": 7.848872900009155
    },
    {
      "epoch": 2.0351409912109375e-05,
      "grad_norm": 0.07835632562637329,
      "learning_rate": 5.927761840020918e-08,
      "loss": 0.0549,
      "step": 95340
    },
    {
      "epoch": 2.0351409912109375e-05,
      "model_forward_time": 1.7632498741149902,
      "step": 95340
    },
    {
      "dynamics_loss_avg": 0.07223016489297152,
      "epoch": 2.0351409912109375e-05,
      "step": 95340
    },
    {
      "action_loss_avg": 0.002190874106599949,
      "epoch": 2.0351409912109375e-05,
      "step": 95340
    },
    {
      "epoch": 2.0351409912109375e-05,
      "step": 95340,
      "training_step_time": 7.442970275878906
    },
    {
      "epoch": 2.0352363586425782e-05,
      "model_forward_time": 1.7601194381713867,
      "step": 95341
    },
    {
      "epoch": 2.0352363586425782e-05,
      "step": 95341,
      "training_step_time": 7.65817403793335
    },
    {
      "epoch": 2.0353317260742186e-05,
      "model_forward_time": 1.761800765991211,
      "step": 95342
    },
    {
      "epoch": 2.0353317260742186e-05,
      "step": 95342,
      "training_step_time": 7.510488271713257
    },
    {
      "epoch": 2.0354270935058594e-05,
      "model_forward_time": 1.7582871913909912,
      "step": 95343
    },
    {
      "epoch": 2.0354270935058594e-05,
      "step": 95343,
      "training_step_time": 6.784074306488037
    },
    {
      "epoch": 2.0355224609375e-05,
      "model_forward_time": 1.7587180137634277,
      "step": 95344
    },
    {
      "epoch": 2.0355224609375e-05,
      "step": 95344,
      "training_step_time": 7.691529750823975
    },
    {
      "epoch": 2.0356178283691405e-05,
      "model_forward_time": 1.7498762607574463,
      "step": 95345
    },
    {
      "epoch": 2.0356178283691405e-05,
      "step": 95345,
      "training_step_time": 8.26697325706482
    },
    {
      "epoch": 2.0357131958007813e-05,
      "model_forward_time": 1.7650108337402344,
      "step": 95346
    },
    {
      "epoch": 2.0357131958007813e-05,
      "step": 95346,
      "training_step_time": 10.706868648529053
    },
    {
      "epoch": 2.035808563232422e-05,
      "model_forward_time": 2.40765118598938,
      "step": 95347
    },
    {
      "epoch": 2.035808563232422e-05,
      "step": 95347,
      "training_step_time": 11.349646091461182
    },
    {
      "epoch": 2.0359039306640624e-05,
      "model_forward_time": 1.7381248474121094,
      "step": 95348
    },
    {
      "epoch": 2.0359039306640624e-05,
      "step": 95348,
      "training_step_time": 7.9450764656066895
    },
    {
      "epoch": 2.035999298095703e-05,
      "model_forward_time": 1.7618684768676758,
      "step": 95349
    },
    {
      "epoch": 2.035999298095703e-05,
      "step": 95349,
      "training_step_time": 7.561471939086914
    },
    {
      "epoch": 2.036094665527344e-05,
      "grad_norm": 0.10893280059099197,
      "learning_rate": 5.902403654581113e-08,
      "loss": 0.0538,
      "step": 95350
    },
    {
      "epoch": 2.036094665527344e-05,
      "model_forward_time": 1.7627315521240234,
      "step": 95350
    },
    {
      "dynamics_loss_avg": 0.06335001774132251,
      "epoch": 2.036094665527344e-05,
      "step": 95350
    },
    {
      "action_loss_avg": 0.0012507423904025926,
      "epoch": 2.036094665527344e-05,
      "step": 95350
    },
    {
      "epoch": 2.036094665527344e-05,
      "step": 95350,
      "training_step_time": 7.5529961585998535
    },
    {
      "epoch": 2.0361900329589843e-05,
      "model_forward_time": 1.7436106204986572,
      "step": 95351
    },
    {
      "epoch": 2.0361900329589843e-05,
      "step": 95351,
      "training_step_time": 7.790636301040649
    },
    {
      "epoch": 2.036285400390625e-05,
      "model_forward_time": 1.7540099620819092,
      "step": 95352
    },
    {
      "epoch": 2.036285400390625e-05,
      "step": 95352,
      "training_step_time": 7.370553970336914
    },
    {
      "epoch": 2.0363807678222658e-05,
      "model_forward_time": 1.8091559410095215,
      "step": 95353
    },
    {
      "epoch": 2.0363807678222658e-05,
      "step": 95353,
      "training_step_time": 8.674953699111938
    },
    {
      "epoch": 2.0364761352539062e-05,
      "model_forward_time": 1.758732557296753,
      "step": 95354
    },
    {
      "epoch": 2.0364761352539062e-05,
      "step": 95354,
      "training_step_time": 7.768007516860962
    },
    {
      "epoch": 2.036571502685547e-05,
      "model_forward_time": 1.7586634159088135,
      "step": 95355
    },
    {
      "epoch": 2.036571502685547e-05,
      "step": 95355,
      "training_step_time": 8.295818090438843
    },
    {
      "epoch": 2.0366668701171877e-05,
      "model_forward_time": 2.422576665878296,
      "step": 95356
    },
    {
      "epoch": 2.0366668701171877e-05,
      "step": 95356,
      "training_step_time": 9.789022207260132
    },
    {
      "epoch": 2.036762237548828e-05,
      "model_forward_time": 1.8290960788726807,
      "step": 95357
    },
    {
      "epoch": 2.036762237548828e-05,
      "step": 95357,
      "training_step_time": 8.590153932571411
    },
    {
      "epoch": 2.0368576049804688e-05,
      "model_forward_time": 1.7666444778442383,
      "step": 95358
    },
    {
      "epoch": 2.0368576049804688e-05,
      "step": 95358,
      "training_step_time": 7.635427951812744
    },
    {
      "epoch": 2.0369529724121095e-05,
      "model_forward_time": 1.7684035301208496,
      "step": 95359
    },
    {
      "epoch": 2.0369529724121095e-05,
      "step": 95359,
      "training_step_time": 7.455098628997803
    },
    {
      "epoch": 2.03704833984375e-05,
      "grad_norm": 0.2687320411205292,
      "learning_rate": 5.8770995029115427e-08,
      "loss": 0.0549,
      "step": 95360
    },
    {
      "epoch": 2.03704833984375e-05,
      "model_forward_time": 1.7703139781951904,
      "step": 95360
    },
    {
      "dynamics_loss_avg": 0.05297416187822819,
      "epoch": 2.03704833984375e-05,
      "step": 95360
    },
    {
      "action_loss_avg": 0.001309392834082246,
      "epoch": 2.03704833984375e-05,
      "step": 95360
    },
    {
      "epoch": 2.03704833984375e-05,
      "step": 95360,
      "training_step_time": 7.205759525299072
    },
    {
      "epoch": 2.0371437072753907e-05,
      "model_forward_time": 1.7643487453460693,
      "step": 95361
    },
    {
      "epoch": 2.0371437072753907e-05,
      "step": 95361,
      "training_step_time": 7.5164220333099365
    },
    {
      "epoch": 2.037239074707031e-05,
      "model_forward_time": 1.7679846286773682,
      "step": 95362
    },
    {
      "epoch": 2.037239074707031e-05,
      "step": 95362,
      "training_step_time": 7.670735120773315
    },
    {
      "epoch": 2.037334442138672e-05,
      "model_forward_time": 1.7642343044281006,
      "step": 95363
    },
    {
      "epoch": 2.037334442138672e-05,
      "step": 95363,
      "training_step_time": 8.162366151809692
    },
    {
      "epoch": 2.0374298095703126e-05,
      "model_forward_time": 1.7643511295318604,
      "step": 95364
    },
    {
      "epoch": 2.0374298095703126e-05,
      "step": 95364,
      "training_step_time": 8.542995929718018
    },
    {
      "epoch": 2.037525177001953e-05,
      "model_forward_time": 2.4395346641540527,
      "step": 95365
    },
    {
      "epoch": 2.037525177001953e-05,
      "step": 95365,
      "training_step_time": 9.559959888458252
    },
    {
      "epoch": 2.0376205444335937e-05,
      "model_forward_time": 1.8205108642578125,
      "step": 95366
    },
    {
      "epoch": 2.0376205444335937e-05,
      "step": 95366,
      "training_step_time": 7.721900939941406
    },
    {
      "epoch": 2.0377159118652345e-05,
      "model_forward_time": 1.773550271987915,
      "step": 95367
    },
    {
      "epoch": 2.0377159118652345e-05,
      "step": 95367,
      "training_step_time": 8.41878890991211
    },
    {
      "epoch": 2.037811279296875e-05,
      "model_forward_time": 1.760533094406128,
      "step": 95368
    },
    {
      "epoch": 2.037811279296875e-05,
      "step": 95368,
      "training_step_time": 7.4347991943359375
    },
    {
      "epoch": 2.0379066467285156e-05,
      "model_forward_time": 1.7725138664245605,
      "step": 95369
    },
    {
      "epoch": 2.0379066467285156e-05,
      "step": 95369,
      "training_step_time": 7.6883485317230225
    },
    {
      "epoch": 2.0380020141601563e-05,
      "grad_norm": 0.08865227550268173,
      "learning_rate": 5.851849387779496e-08,
      "loss": 0.053,
      "step": 95370
    },
    {
      "epoch": 2.0380020141601563e-05,
      "model_forward_time": 1.7790107727050781,
      "step": 95370
    },
    {
      "dynamics_loss_avg": 0.05164931435137987,
      "epoch": 2.0380020141601563e-05,
      "step": 95370
    },
    {
      "action_loss_avg": 0.0012983007822185756,
      "epoch": 2.0380020141601563e-05,
      "step": 95370
    },
    {
      "epoch": 2.0380020141601563e-05,
      "step": 95370,
      "training_step_time": 7.5247962474823
    },
    {
      "epoch": 2.0380973815917967e-05,
      "model_forward_time": 1.7657616138458252,
      "step": 95371
    },
    {
      "epoch": 2.0380973815917967e-05,
      "step": 95371,
      "training_step_time": 7.682831048965454
    },
    {
      "epoch": 2.0381927490234375e-05,
      "model_forward_time": 1.763674259185791,
      "step": 95372
    },
    {
      "epoch": 2.0381927490234375e-05,
      "step": 95372,
      "training_step_time": 7.649264812469482
    },
    {
      "epoch": 2.0382881164550782e-05,
      "model_forward_time": 1.760491132736206,
      "step": 95373
    },
    {
      "epoch": 2.0382881164550782e-05,
      "step": 95373,
      "training_step_time": 10.036456108093262
    },
    {
      "epoch": 2.0383834838867186e-05,
      "model_forward_time": 2.4240689277648926,
      "step": 95374
    },
    {
      "epoch": 2.0383834838867186e-05,
      "step": 95374,
      "training_step_time": 9.589714527130127
    },
    {
      "epoch": 2.0384788513183594e-05,
      "model_forward_time": 1.4111919403076172,
      "step": 95375
    },
    {
      "epoch": 2.0384788513183594e-05,
      "step": 95375,
      "training_step_time": 8.636965274810791
    },
    {
      "epoch": 2.03857421875e-05,
      "model_forward_time": 1.7674267292022705,
      "step": 95376
    },
    {
      "epoch": 2.03857421875e-05,
      "step": 95376,
      "training_step_time": 7.716489791870117
    },
    {
      "epoch": 2.0386695861816405e-05,
      "model_forward_time": 1.7622261047363281,
      "step": 95377
    },
    {
      "epoch": 2.0386695861816405e-05,
      "step": 95377,
      "training_step_time": 7.477655649185181
    },
    {
      "epoch": 2.0387649536132813e-05,
      "model_forward_time": 1.775862693786621,
      "step": 95378
    },
    {
      "epoch": 2.0387649536132813e-05,
      "step": 95378,
      "training_step_time": 7.6821043491363525
    },
    {
      "epoch": 2.038860321044922e-05,
      "model_forward_time": 1.7643625736236572,
      "step": 95379
    },
    {
      "epoch": 2.038860321044922e-05,
      "step": 95379,
      "training_step_time": 7.508054971694946
    },
    {
      "epoch": 2.0389556884765624e-05,
      "grad_norm": 0.1083076223731041,
      "learning_rate": 5.8266533119462623e-08,
      "loss": 0.0543,
      "step": 95380
    },
    {
      "epoch": 2.0389556884765624e-05,
      "model_forward_time": 1.7700576782226562,
      "step": 95380
    },
    {
      "dynamics_loss_avg": 0.053988639079034326,
      "epoch": 2.0389556884765624e-05,
      "step": 95380
    },
    {
      "action_loss_avg": 0.0013547729788115249,
      "epoch": 2.0389556884765624e-05,
      "step": 95380
    },
    {
      "epoch": 2.0389556884765624e-05,
      "step": 95380,
      "training_step_time": 7.197979927062988
    },
    {
      "epoch": 2.039051055908203e-05,
      "model_forward_time": 1.762770652770996,
      "step": 95381
    },
    {
      "epoch": 2.039051055908203e-05,
      "step": 95381,
      "training_step_time": 7.6526288986206055
    },
    {
      "epoch": 2.039146423339844e-05,
      "model_forward_time": 1.7733335494995117,
      "step": 95382
    },
    {
      "epoch": 2.039146423339844e-05,
      "step": 95382,
      "training_step_time": 9.685638427734375
    },
    {
      "epoch": 2.0392417907714843e-05,
      "model_forward_time": 1.7540242671966553,
      "step": 95383
    },
    {
      "epoch": 2.0392417907714843e-05,
      "step": 95383,
      "training_step_time": 9.510123014450073
    },
    {
      "epoch": 2.039337158203125e-05,
      "model_forward_time": 2.4365644454956055,
      "step": 95384
    },
    {
      "epoch": 2.039337158203125e-05,
      "step": 95384,
      "training_step_time": 8.5149085521698
    },
    {
      "epoch": 2.0394325256347658e-05,
      "model_forward_time": 1.7577776908874512,
      "step": 95385
    },
    {
      "epoch": 2.0394325256347658e-05,
      "step": 95385,
      "training_step_time": 8.66347622871399
    },
    {
      "epoch": 2.0395278930664062e-05,
      "model_forward_time": 1.7655177116394043,
      "step": 95386
    },
    {
      "epoch": 2.0395278930664062e-05,
      "step": 95386,
      "training_step_time": 7.458994626998901
    },
    {
      "epoch": 2.039623260498047e-05,
      "model_forward_time": 1.7720348834991455,
      "step": 95387
    },
    {
      "epoch": 2.039623260498047e-05,
      "step": 95387,
      "training_step_time": 7.591003179550171
    },
    {
      "epoch": 2.0397186279296877e-05,
      "model_forward_time": 1.7618074417114258,
      "step": 95388
    },
    {
      "epoch": 2.0397186279296877e-05,
      "step": 95388,
      "training_step_time": 7.26427698135376
    },
    {
      "epoch": 2.039813995361328e-05,
      "model_forward_time": 1.766542911529541,
      "step": 95389
    },
    {
      "epoch": 2.039813995361328e-05,
      "step": 95389,
      "training_step_time": 7.192508220672607
    },
    {
      "epoch": 2.0399093627929688e-05,
      "grad_norm": 0.10062999278306961,
      "learning_rate": 5.8015112781672487e-08,
      "loss": 0.0557,
      "step": 95390
    },
    {
      "epoch": 2.0399093627929688e-05,
      "model_forward_time": 1.7490239143371582,
      "step": 95390
    },
    {
      "dynamics_loss_avg": 0.06018808800727129,
      "epoch": 2.0399093627929688e-05,
      "step": 95390
    },
    {
      "action_loss_avg": 0.0016287232021568343,
      "epoch": 2.0399093627929688e-05,
      "step": 95390
    },
    {
      "epoch": 2.0399093627929688e-05,
      "step": 95390,
      "training_step_time": 8.545768022537231
    },
    {
      "epoch": 2.0400047302246092e-05,
      "model_forward_time": 1.7522978782653809,
      "step": 95391
    },
    {
      "epoch": 2.0400047302246092e-05,
      "step": 95391,
      "training_step_time": 10.060181379318237
    },
    {
      "epoch": 2.04010009765625e-05,
      "model_forward_time": 1.762782335281372,
      "step": 95392
    },
    {
      "epoch": 2.04010009765625e-05,
      "step": 95392,
      "training_step_time": 10.646404504776001
    },
    {
      "epoch": 2.0401954650878907e-05,
      "model_forward_time": 2.4177162647247314,
      "step": 95393
    },
    {
      "epoch": 2.0401954650878907e-05,
      "step": 95393,
      "training_step_time": 9.331212043762207
    },
    {
      "epoch": 2.040290832519531e-05,
      "model_forward_time": 1.8292760848999023,
      "step": 95394
    },
    {
      "epoch": 2.040290832519531e-05,
      "step": 95394,
      "training_step_time": 7.779709339141846
    },
    {
      "epoch": 2.040386199951172e-05,
      "model_forward_time": 1.7581753730773926,
      "step": 95395
    },
    {
      "epoch": 2.040386199951172e-05,
      "step": 95395,
      "training_step_time": 7.5048534870147705
    },
    {
      "epoch": 2.0404815673828126e-05,
      "model_forward_time": 1.7522857189178467,
      "step": 95396
    },
    {
      "epoch": 2.0404815673828126e-05,
      "step": 95396,
      "training_step_time": 7.457493543624878
    },
    {
      "epoch": 2.040576934814453e-05,
      "model_forward_time": 1.762411117553711,
      "step": 95397
    },
    {
      "epoch": 2.040576934814453e-05,
      "step": 95397,
      "training_step_time": 7.653071880340576
    },
    {
      "epoch": 2.0406723022460937e-05,
      "model_forward_time": 1.756486177444458,
      "step": 95398
    },
    {
      "epoch": 2.0406723022460937e-05,
      "step": 95398,
      "training_step_time": 7.515980243682861
    },
    {
      "epoch": 2.0407676696777345e-05,
      "model_forward_time": 1.7616071701049805,
      "step": 95399
    },
    {
      "epoch": 2.0407676696777345e-05,
      "step": 95399,
      "training_step_time": 7.627004384994507
    },
    {
      "epoch": 2.040863037109375e-05,
      "grad_norm": 0.08730897307395935,
      "learning_rate": 5.776423289191868e-08,
      "loss": 0.0547,
      "step": 95400
    },
    {
      "epoch": 2.040863037109375e-05,
      "model_forward_time": 1.748856782913208,
      "step": 95400
    },
    {
      "dynamics_loss_avg": 0.051265861932188274,
      "epoch": 2.040863037109375e-05,
      "step": 95400
    },
    {
      "action_loss_avg": 0.0012462798389606179,
      "epoch": 2.040863037109375e-05,
      "step": 95400
    },
    {
      "epoch": 2.040863037109375e-05,
      "step": 95400,
      "training_step_time": 8.634832859039307
    },
    {
      "epoch": 2.0409584045410156e-05,
      "model_forward_time": 1.7497475147247314,
      "step": 95401
    },
    {
      "epoch": 2.0409584045410156e-05,
      "step": 95401,
      "training_step_time": 8.775726079940796
    },
    {
      "epoch": 2.0410537719726563e-05,
      "model_forward_time": 2.42515230178833,
      "step": 95402
    },
    {
      "epoch": 2.0410537719726563e-05,
      "step": 95402,
      "training_step_time": 8.555256366729736
    },
    {
      "epoch": 2.0411491394042968e-05,
      "model_forward_time": 1.8185367584228516,
      "step": 95403
    },
    {
      "epoch": 2.0411491394042968e-05,
      "step": 95403,
      "training_step_time": 8.775025844573975
    },
    {
      "epoch": 2.0412445068359375e-05,
      "model_forward_time": 1.755324363708496,
      "step": 95404
    },
    {
      "epoch": 2.0412445068359375e-05,
      "step": 95404,
      "training_step_time": 8.337243556976318
    },
    {
      "epoch": 2.0413398742675782e-05,
      "model_forward_time": 1.7557740211486816,
      "step": 95405
    },
    {
      "epoch": 2.0413398742675782e-05,
      "step": 95405,
      "training_step_time": 7.4589409828186035
    },
    {
      "epoch": 2.0414352416992186e-05,
      "model_forward_time": 1.7563209533691406,
      "step": 95406
    },
    {
      "epoch": 2.0414352416992186e-05,
      "step": 95406,
      "training_step_time": 7.596397399902344
    },
    {
      "epoch": 2.0415306091308594e-05,
      "model_forward_time": 1.7520956993103027,
      "step": 95407
    },
    {
      "epoch": 2.0415306091308594e-05,
      "step": 95407,
      "training_step_time": 7.253743886947632
    },
    {
      "epoch": 2.0416259765625e-05,
      "model_forward_time": 1.7534825801849365,
      "step": 95408
    },
    {
      "epoch": 2.0416259765625e-05,
      "step": 95408,
      "training_step_time": 7.871501445770264
    },
    {
      "epoch": 2.0417213439941405e-05,
      "model_forward_time": 1.7592813968658447,
      "step": 95409
    },
    {
      "epoch": 2.0417213439941405e-05,
      "step": 95409,
      "training_step_time": 9.41971755027771
    },
    {
      "epoch": 2.0418167114257813e-05,
      "grad_norm": 0.10039497166872025,
      "learning_rate": 5.751389347763869e-08,
      "loss": 0.0555,
      "step": 95410
    },
    {
      "epoch": 2.0418167114257813e-05,
      "model_forward_time": 1.746166706085205,
      "step": 95410
    },
    {
      "dynamics_loss_avg": 0.054381300788372756,
      "epoch": 2.0418167114257813e-05,
      "step": 95410
    },
    {
      "action_loss_avg": 0.0012799980118870734,
      "epoch": 2.0418167114257813e-05,
      "step": 95410
    },
    {
      "epoch": 2.0418167114257813e-05,
      "step": 95410,
      "training_step_time": 8.713358402252197
    },
    {
      "epoch": 2.041912078857422e-05,
      "model_forward_time": 2.4411840438842773,
      "step": 95411
    },
    {
      "epoch": 2.041912078857422e-05,
      "step": 95411,
      "training_step_time": 7.8816118240356445
    },
    {
      "epoch": 2.0420074462890624e-05,
      "model_forward_time": 1.8191359043121338,
      "step": 95412
    },
    {
      "epoch": 2.0420074462890624e-05,
      "step": 95412,
      "training_step_time": 7.894685506820679
    },
    {
      "epoch": 2.042102813720703e-05,
      "model_forward_time": 1.7447459697723389,
      "step": 95413
    },
    {
      "epoch": 2.042102813720703e-05,
      "step": 95413,
      "training_step_time": 9.236918449401855
    },
    {
      "epoch": 2.042198181152344e-05,
      "model_forward_time": 1.7463555335998535,
      "step": 95414
    },
    {
      "epoch": 2.042198181152344e-05,
      "step": 95414,
      "training_step_time": 7.500705003738403
    },
    {
      "epoch": 2.0422935485839843e-05,
      "model_forward_time": 1.3978297710418701,
      "step": 95415
    },
    {
      "epoch": 2.0422935485839843e-05,
      "step": 95415,
      "training_step_time": 7.476529121398926
    },
    {
      "epoch": 2.042388916015625e-05,
      "model_forward_time": 1.7498996257781982,
      "step": 95416
    },
    {
      "epoch": 2.042388916015625e-05,
      "step": 95416,
      "training_step_time": 7.67383599281311
    },
    {
      "epoch": 2.0424842834472658e-05,
      "model_forward_time": 1.754669427871704,
      "step": 95417
    },
    {
      "epoch": 2.0424842834472658e-05,
      "step": 95417,
      "training_step_time": 6.895737886428833
    },
    {
      "epoch": 2.0425796508789062e-05,
      "model_forward_time": 1.7451982498168945,
      "step": 95418
    },
    {
      "epoch": 2.0425796508789062e-05,
      "step": 95418,
      "training_step_time": 7.820540904998779
    },
    {
      "epoch": 2.042675018310547e-05,
      "model_forward_time": 1.7459831237792969,
      "step": 95419
    },
    {
      "epoch": 2.042675018310547e-05,
      "step": 95419,
      "training_step_time": 9.307340383529663
    },
    {
      "epoch": 2.0427703857421877e-05,
      "grad_norm": 0.08300663530826569,
      "learning_rate": 5.7264094566208404e-08,
      "loss": 0.0554,
      "step": 95420
    },
    {
      "epoch": 2.0427703857421877e-05,
      "model_forward_time": 1.747854232788086,
      "step": 95420
    },
    {
      "dynamics_loss_avg": 0.06193571835756302,
      "epoch": 2.0427703857421877e-05,
      "step": 95420
    },
    {
      "action_loss_avg": 0.0013205849769292398,
      "epoch": 2.0427703857421877e-05,
      "step": 95420
    },
    {
      "epoch": 2.0427703857421877e-05,
      "step": 95420,
      "training_step_time": 10.322785377502441
    },
    {
      "epoch": 2.042865753173828e-05,
      "model_forward_time": 1.8104641437530518,
      "step": 95421
    },
    {
      "epoch": 2.042865753173828e-05,
      "step": 95421,
      "training_step_time": 10.586689472198486
    },
    {
      "epoch": 2.0429611206054688e-05,
      "model_forward_time": 1.7596659660339355,
      "step": 95422
    },
    {
      "epoch": 2.0429611206054688e-05,
      "step": 95422,
      "training_step_time": 8.870341539382935
    },
    {
      "epoch": 2.0430564880371092e-05,
      "model_forward_time": 1.748004674911499,
      "step": 95423
    },
    {
      "epoch": 2.0430564880371092e-05,
      "step": 95423,
      "training_step_time": 8.345519781112671
    },
    {
      "epoch": 2.04315185546875e-05,
      "model_forward_time": 1.7534034252166748,
      "step": 95424
    },
    {
      "epoch": 2.04315185546875e-05,
      "step": 95424,
      "training_step_time": 7.33094334602356
    },
    {
      "epoch": 2.0432472229003907e-05,
      "model_forward_time": 1.8061201572418213,
      "step": 95425
    },
    {
      "epoch": 2.0432472229003907e-05,
      "step": 95425,
      "training_step_time": 7.682138681411743
    },
    {
      "epoch": 2.043342590332031e-05,
      "model_forward_time": 1.7512116432189941,
      "step": 95426
    },
    {
      "epoch": 2.043342590332031e-05,
      "step": 95426,
      "training_step_time": 6.831218242645264
    },
    {
      "epoch": 2.043437957763672e-05,
      "model_forward_time": 1.7541170120239258,
      "step": 95427
    },
    {
      "epoch": 2.043437957763672e-05,
      "step": 95427,
      "training_step_time": 7.734517574310303
    },
    {
      "epoch": 2.0435333251953126e-05,
      "model_forward_time": 1.738668441772461,
      "step": 95428
    },
    {
      "epoch": 2.0435333251953126e-05,
      "step": 95428,
      "training_step_time": 7.822611093521118
    },
    {
      "epoch": 2.043628692626953e-05,
      "model_forward_time": 1.7439641952514648,
      "step": 95429
    },
    {
      "epoch": 2.043628692626953e-05,
      "step": 95429,
      "training_step_time": 8.63419771194458
    },
    {
      "epoch": 2.0437240600585937e-05,
      "grad_norm": 0.2345348298549652,
      "learning_rate": 5.701483618494541e-08,
      "loss": 0.055,
      "step": 95430
    },
    {
      "epoch": 2.0437240600585937e-05,
      "model_forward_time": 2.5261693000793457,
      "step": 95430
    },
    {
      "dynamics_loss_avg": 0.057912396267056465,
      "epoch": 2.0437240600585937e-05,
      "step": 95430
    },
    {
      "action_loss_avg": 0.0015031414513941854,
      "epoch": 2.0437240600585937e-05,
      "step": 95430
    },
    {
      "epoch": 2.0437240600585937e-05,
      "step": 95430,
      "training_step_time": 9.5952627658844
    },
    {
      "epoch": 2.0438194274902345e-05,
      "model_forward_time": 1.8207080364227295,
      "step": 95431
    },
    {
      "epoch": 2.0438194274902345e-05,
      "step": 95431,
      "training_step_time": 7.758708238601685
    },
    {
      "epoch": 2.043914794921875e-05,
      "model_forward_time": 1.7443444728851318,
      "step": 95432
    },
    {
      "epoch": 2.043914794921875e-05,
      "step": 95432,
      "training_step_time": 7.71522068977356
    },
    {
      "epoch": 2.0440101623535156e-05,
      "model_forward_time": 1.7447621822357178,
      "step": 95433
    },
    {
      "epoch": 2.0440101623535156e-05,
      "step": 95433,
      "training_step_time": 7.6780664920806885
    },
    {
      "epoch": 2.0441055297851564e-05,
      "model_forward_time": 1.742098331451416,
      "step": 95434
    },
    {
      "epoch": 2.0441055297851564e-05,
      "step": 95434,
      "training_step_time": 7.8172290325164795
    },
    {
      "epoch": 2.0442008972167968e-05,
      "model_forward_time": 1.746159315109253,
      "step": 95435
    },
    {
      "epoch": 2.0442008972167968e-05,
      "step": 95435,
      "training_step_time": 7.122570991516113
    },
    {
      "epoch": 2.0442962646484375e-05,
      "model_forward_time": 1.7472078800201416,
      "step": 95436
    },
    {
      "epoch": 2.0442962646484375e-05,
      "step": 95436,
      "training_step_time": 7.569084882736206
    },
    {
      "epoch": 2.0443916320800782e-05,
      "model_forward_time": 1.7484593391418457,
      "step": 95437
    },
    {
      "epoch": 2.0443916320800782e-05,
      "step": 95437,
      "training_step_time": 10.463699579238892
    },
    {
      "epoch": 2.0444869995117186e-05,
      "model_forward_time": 1.7491390705108643,
      "step": 95438
    },
    {
      "epoch": 2.0444869995117186e-05,
      "step": 95438,
      "training_step_time": 9.004760980606079
    },
    {
      "epoch": 2.0445823669433594e-05,
      "model_forward_time": 2.398918867111206,
      "step": 95439
    },
    {
      "epoch": 2.0445823669433594e-05,
      "step": 95439,
      "training_step_time": 9.61098313331604
    },
    {
      "epoch": 2.044677734375e-05,
      "grad_norm": 0.08750832080841064,
      "learning_rate": 5.6766118361107905e-08,
      "loss": 0.0537,
      "step": 95440
    },
    {
      "epoch": 2.044677734375e-05,
      "model_forward_time": 1.8120181560516357,
      "step": 95440
    },
    {
      "dynamics_loss_avg": 0.062214143387973306,
      "epoch": 2.044677734375e-05,
      "step": 95440
    },
    {
      "action_loss_avg": 0.001500096617382951,
      "epoch": 2.044677734375e-05,
      "step": 95440
    },
    {
      "epoch": 2.044677734375e-05,
      "step": 95440,
      "training_step_time": 7.865393161773682
    },
    {
      "epoch": 2.0447731018066405e-05,
      "model_forward_time": 1.750645637512207,
      "step": 95441
    },
    {
      "epoch": 2.0447731018066405e-05,
      "step": 95441,
      "training_step_time": 8.548927545547485
    },
    {
      "epoch": 2.0448684692382813e-05,
      "model_forward_time": 1.742748737335205,
      "step": 95442
    },
    {
      "epoch": 2.0448684692382813e-05,
      "step": 95442,
      "training_step_time": 7.875612258911133
    },
    {
      "epoch": 2.044963836669922e-05,
      "model_forward_time": 1.75764799118042,
      "step": 95443
    },
    {
      "epoch": 2.044963836669922e-05,
      "step": 95443,
      "training_step_time": 7.77044415473938
    },
    {
      "epoch": 2.0450592041015624e-05,
      "model_forward_time": 1.7552332878112793,
      "step": 95444
    },
    {
      "epoch": 2.0450592041015624e-05,
      "step": 95444,
      "training_step_time": 7.3457605838775635
    },
    {
      "epoch": 2.045154571533203e-05,
      "model_forward_time": 1.753061294555664,
      "step": 95445
    },
    {
      "epoch": 2.045154571533203e-05,
      "step": 95445,
      "training_step_time": 7.479622840881348
    },
    {
      "epoch": 2.045249938964844e-05,
      "model_forward_time": 1.7499713897705078,
      "step": 95446
    },
    {
      "epoch": 2.045249938964844e-05,
      "step": 95446,
      "training_step_time": 8.961805582046509
    },
    {
      "epoch": 2.0453453063964843e-05,
      "model_forward_time": 1.7576274871826172,
      "step": 95447
    },
    {
      "epoch": 2.0453453063964843e-05,
      "step": 95447,
      "training_step_time": 8.946765422821045
    },
    {
      "epoch": 2.045440673828125e-05,
      "model_forward_time": 2.419732093811035,
      "step": 95448
    },
    {
      "epoch": 2.045440673828125e-05,
      "step": 95448,
      "training_step_time": 8.9349684715271
    },
    {
      "epoch": 2.0455360412597658e-05,
      "model_forward_time": 1.79213547706604,
      "step": 95449
    },
    {
      "epoch": 2.0455360412597658e-05,
      "step": 95449,
      "training_step_time": 8.649499654769897
    },
    {
      "epoch": 2.0456314086914062e-05,
      "grad_norm": 0.08406062424182892,
      "learning_rate": 5.651794112189579e-08,
      "loss": 0.0562,
      "step": 95450
    },
    {
      "epoch": 2.0456314086914062e-05,
      "model_forward_time": 1.764413833618164,
      "step": 95450
    },
    {
      "dynamics_loss_avg": 0.06509052384644747,
      "epoch": 2.0456314086914062e-05,
      "step": 95450
    },
    {
      "action_loss_avg": 0.0030294791329652073,
      "epoch": 2.0456314086914062e-05,
      "step": 95450
    },
    {
      "epoch": 2.0456314086914062e-05,
      "step": 95450,
      "training_step_time": 8.51148009300232
    },
    {
      "epoch": 2.045726776123047e-05,
      "model_forward_time": 1.7704625129699707,
      "step": 95451
    },
    {
      "epoch": 2.045726776123047e-05,
      "step": 95451,
      "training_step_time": 7.514085531234741
    },
    {
      "epoch": 2.0458221435546877e-05,
      "model_forward_time": 1.7604079246520996,
      "step": 95452
    },
    {
      "epoch": 2.0458221435546877e-05,
      "step": 95452,
      "training_step_time": 7.308485269546509
    },
    {
      "epoch": 2.045917510986328e-05,
      "model_forward_time": 1.7562568187713623,
      "step": 95453
    },
    {
      "epoch": 2.045917510986328e-05,
      "step": 95453,
      "training_step_time": 7.247511386871338
    },
    {
      "epoch": 2.0460128784179688e-05,
      "model_forward_time": 1.7503972053527832,
      "step": 95454
    },
    {
      "epoch": 2.0460128784179688e-05,
      "step": 95454,
      "training_step_time": 7.306220769882202
    },
    {
      "epoch": 2.0461082458496092e-05,
      "model_forward_time": 1.7605533599853516,
      "step": 95455
    },
    {
      "epoch": 2.0461082458496092e-05,
      "step": 95455,
      "training_step_time": 7.676203012466431
    },
    {
      "epoch": 2.04620361328125e-05,
      "model_forward_time": 1.7521278858184814,
      "step": 95456
    },
    {
      "epoch": 2.04620361328125e-05,
      "step": 95456,
      "training_step_time": 9.224917650222778
    },
    {
      "epoch": 2.0462989807128907e-05,
      "model_forward_time": 2.4702131748199463,
      "step": 95457
    },
    {
      "epoch": 2.0462989807128907e-05,
      "step": 95457,
      "training_step_time": 9.775516986846924
    },
    {
      "epoch": 2.046394348144531e-05,
      "model_forward_time": 1.7695400714874268,
      "step": 95458
    },
    {
      "epoch": 2.046394348144531e-05,
      "step": 95458,
      "training_step_time": 8.943567037582397
    },
    {
      "epoch": 2.046489715576172e-05,
      "model_forward_time": 1.7643556594848633,
      "step": 95459
    },
    {
      "epoch": 2.046489715576172e-05,
      "step": 95459,
      "training_step_time": 8.612807512283325
    },
    {
      "epoch": 2.0465850830078126e-05,
      "grad_norm": 0.07351870834827423,
      "learning_rate": 5.6270304494449035e-08,
      "loss": 0.0533,
      "step": 95460
    },
    {
      "epoch": 2.0465850830078126e-05,
      "model_forward_time": 1.7713396549224854,
      "step": 95460
    },
    {
      "dynamics_loss_avg": 0.06026643654331565,
      "epoch": 2.0465850830078126e-05,
      "step": 95460
    },
    {
      "action_loss_avg": 0.0016153801319887862,
      "epoch": 2.0465850830078126e-05,
      "step": 95460
    },
    {
      "epoch": 2.0465850830078126e-05,
      "step": 95460,
      "training_step_time": 7.521383285522461
    },
    {
      "epoch": 2.046680450439453e-05,
      "model_forward_time": 1.7569043636322021,
      "step": 95461
    },
    {
      "epoch": 2.046680450439453e-05,
      "step": 95461,
      "training_step_time": 7.479125022888184
    },
    {
      "epoch": 2.0467758178710937e-05,
      "model_forward_time": 1.7536756992340088,
      "step": 95462
    },
    {
      "epoch": 2.0467758178710937e-05,
      "step": 95462,
      "training_step_time": 7.767715215682983
    },
    {
      "epoch": 2.0468711853027345e-05,
      "model_forward_time": 1.7695848941802979,
      "step": 95463
    },
    {
      "epoch": 2.0468711853027345e-05,
      "step": 95463,
      "training_step_time": 7.392959833145142
    },
    {
      "epoch": 2.046966552734375e-05,
      "model_forward_time": 1.7506299018859863,
      "step": 95464
    },
    {
      "epoch": 2.046966552734375e-05,
      "step": 95464,
      "training_step_time": 7.637998342514038
    },
    {
      "epoch": 2.0470619201660156e-05,
      "model_forward_time": 1.7610459327697754,
      "step": 95465
    },
    {
      "epoch": 2.0470619201660156e-05,
      "step": 95465,
      "training_step_time": 8.535368204116821
    },
    {
      "epoch": 2.0471572875976564e-05,
      "model_forward_time": 2.4427316188812256,
      "step": 95466
    },
    {
      "epoch": 2.0471572875976564e-05,
      "step": 95466,
      "training_step_time": 8.669475793838501
    },
    {
      "epoch": 2.0472526550292968e-05,
      "model_forward_time": 1.759645700454712,
      "step": 95467
    },
    {
      "epoch": 2.0472526550292968e-05,
      "step": 95467,
      "training_step_time": 7.958450555801392
    },
    {
      "epoch": 2.0473480224609375e-05,
      "model_forward_time": 1.8184597492218018,
      "step": 95468
    },
    {
      "epoch": 2.0473480224609375e-05,
      "step": 95468,
      "training_step_time": 7.703943252563477
    },
    {
      "epoch": 2.0474433898925782e-05,
      "model_forward_time": 1.7628324031829834,
      "step": 95469
    },
    {
      "epoch": 2.0474433898925782e-05,
      "step": 95469,
      "training_step_time": 8.61193585395813
    },
    {
      "epoch": 2.0475387573242186e-05,
      "grad_norm": 0.098088338971138,
      "learning_rate": 5.6023208505849305e-08,
      "loss": 0.0547,
      "step": 95470
    },
    {
      "epoch": 2.0475387573242186e-05,
      "model_forward_time": 1.7429475784301758,
      "step": 95470
    },
    {
      "dynamics_loss_avg": 0.06065287236124277,
      "epoch": 2.0475387573242186e-05,
      "step": 95470
    },
    {
      "action_loss_avg": 0.0015332262497395276,
      "epoch": 2.0475387573242186e-05,
      "step": 95470
    },
    {
      "epoch": 2.0475387573242186e-05,
      "step": 95470,
      "training_step_time": 8.149893045425415
    },
    {
      "epoch": 2.0476341247558594e-05,
      "model_forward_time": 1.74751615524292,
      "step": 95471
    },
    {
      "epoch": 2.0476341247558594e-05,
      "step": 95471,
      "training_step_time": 7.720878601074219
    },
    {
      "epoch": 2.0477294921875e-05,
      "model_forward_time": 1.762373685836792,
      "step": 95472
    },
    {
      "epoch": 2.0477294921875e-05,
      "step": 95472,
      "training_step_time": 7.360915184020996
    },
    {
      "epoch": 2.0478248596191405e-05,
      "model_forward_time": 1.745431661605835,
      "step": 95473
    },
    {
      "epoch": 2.0478248596191405e-05,
      "step": 95473,
      "training_step_time": 8.675550699234009
    },
    {
      "epoch": 2.0479202270507813e-05,
      "model_forward_time": 1.7475640773773193,
      "step": 95474
    },
    {
      "epoch": 2.0479202270507813e-05,
      "step": 95474,
      "training_step_time": 8.893588781356812
    },
    {
      "epoch": 2.048015594482422e-05,
      "model_forward_time": 2.3809471130371094,
      "step": 95475
    },
    {
      "epoch": 2.048015594482422e-05,
      "step": 95475,
      "training_step_time": 11.236404657363892
    },
    {
      "epoch": 2.0481109619140624e-05,
      "model_forward_time": 1.7545249462127686,
      "step": 95476
    },
    {
      "epoch": 2.0481109619140624e-05,
      "step": 95476,
      "training_step_time": 7.758767127990723
    },
    {
      "epoch": 2.048206329345703e-05,
      "model_forward_time": 1.8056752681732178,
      "step": 95477
    },
    {
      "epoch": 2.048206329345703e-05,
      "step": 95477,
      "training_step_time": 7.689729690551758
    },
    {
      "epoch": 2.048301696777344e-05,
      "model_forward_time": 1.7466576099395752,
      "step": 95478
    },
    {
      "epoch": 2.048301696777344e-05,
      "step": 95478,
      "training_step_time": 8.276174306869507
    },
    {
      "epoch": 2.0483970642089843e-05,
      "model_forward_time": 1.7509796619415283,
      "step": 95479
    },
    {
      "epoch": 2.0483970642089843e-05,
      "step": 95479,
      "training_step_time": 7.522815704345703
    },
    {
      "epoch": 2.048492431640625e-05,
      "grad_norm": 0.14415186643600464,
      "learning_rate": 5.577665318311776e-08,
      "loss": 0.055,
      "step": 95480
    },
    {
      "epoch": 2.048492431640625e-05,
      "model_forward_time": 1.7468140125274658,
      "step": 95480
    },
    {
      "dynamics_loss_avg": 0.07186278607696295,
      "epoch": 2.048492431640625e-05,
      "step": 95480
    },
    {
      "action_loss_avg": 0.0013488141674315556,
      "epoch": 2.048492431640625e-05,
      "step": 95480
    },
    {
      "epoch": 2.048492431640625e-05,
      "step": 95480,
      "training_step_time": 7.608328819274902
    },
    {
      "epoch": 2.0485877990722658e-05,
      "model_forward_time": 1.7507989406585693,
      "step": 95481
    },
    {
      "epoch": 2.0485877990722658e-05,
      "step": 95481,
      "training_step_time": 6.876672983169556
    },
    {
      "epoch": 2.0486831665039062e-05,
      "model_forward_time": 1.743438720703125,
      "step": 95482
    },
    {
      "epoch": 2.0486831665039062e-05,
      "step": 95482,
      "training_step_time": 7.694785118103027
    },
    {
      "epoch": 2.048778533935547e-05,
      "model_forward_time": 1.7462713718414307,
      "step": 95483
    },
    {
      "epoch": 2.048778533935547e-05,
      "step": 95483,
      "training_step_time": 8.688443183898926
    },
    {
      "epoch": 2.0488739013671877e-05,
      "model_forward_time": 1.7482550144195557,
      "step": 95484
    },
    {
      "epoch": 2.0488739013671877e-05,
      "step": 95484,
      "training_step_time": 8.588829755783081
    },
    {
      "epoch": 2.048969268798828e-05,
      "model_forward_time": 2.408022880554199,
      "step": 95485
    },
    {
      "epoch": 2.048969268798828e-05,
      "step": 95485,
      "training_step_time": 9.660376787185669
    },
    {
      "epoch": 2.0490646362304688e-05,
      "model_forward_time": 1.818603515625,
      "step": 95486
    },
    {
      "epoch": 2.0490646362304688e-05,
      "step": 95486,
      "training_step_time": 9.43690824508667
    },
    {
      "epoch": 2.0491600036621092e-05,
      "model_forward_time": 1.7550172805786133,
      "step": 95487
    },
    {
      "epoch": 2.0491600036621092e-05,
      "step": 95487,
      "training_step_time": 8.241548538208008
    },
    {
      "epoch": 2.04925537109375e-05,
      "model_forward_time": 1.7545464038848877,
      "step": 95488
    },
    {
      "epoch": 2.04925537109375e-05,
      "step": 95488,
      "training_step_time": 7.566693544387817
    },
    {
      "epoch": 2.0493507385253907e-05,
      "model_forward_time": 1.7564001083374023,
      "step": 95489
    },
    {
      "epoch": 2.0493507385253907e-05,
      "step": 95489,
      "training_step_time": 7.922189712524414
    },
    {
      "epoch": 2.049446105957031e-05,
      "grad_norm": 0.07111452519893646,
      "learning_rate": 5.553063855321839e-08,
      "loss": 0.0549,
      "step": 95490
    },
    {
      "epoch": 2.049446105957031e-05,
      "model_forward_time": 1.7580363750457764,
      "step": 95490
    },
    {
      "dynamics_loss_avg": 0.05418539866805076,
      "epoch": 2.049446105957031e-05,
      "step": 95490
    },
    {
      "action_loss_avg": 0.0015871166076976805,
      "epoch": 2.049446105957031e-05,
      "step": 95490
    },
    {
      "epoch": 2.049446105957031e-05,
      "step": 95490,
      "training_step_time": 7.2629101276397705
    },
    {
      "epoch": 2.049541473388672e-05,
      "model_forward_time": 1.7549762725830078,
      "step": 95491
    },
    {
      "epoch": 2.049541473388672e-05,
      "step": 95491,
      "training_step_time": 7.73419189453125
    },
    {
      "epoch": 2.0496368408203126e-05,
      "model_forward_time": 1.7733845710754395,
      "step": 95492
    },
    {
      "epoch": 2.0496368408203126e-05,
      "step": 95492,
      "training_step_time": 9.527999877929688
    },
    {
      "epoch": 2.049732208251953e-05,
      "model_forward_time": 1.7552757263183594,
      "step": 95493
    },
    {
      "epoch": 2.049732208251953e-05,
      "step": 95493,
      "training_step_time": 10.180484533309937
    },
    {
      "epoch": 2.0498275756835937e-05,
      "model_forward_time": 2.415989398956299,
      "step": 95494
    },
    {
      "epoch": 2.0498275756835937e-05,
      "step": 95494,
      "training_step_time": 8.88953161239624
    },
    {
      "epoch": 2.0499229431152345e-05,
      "model_forward_time": 1.7975835800170898,
      "step": 95495
    },
    {
      "epoch": 2.0499229431152345e-05,
      "step": 95495,
      "training_step_time": 7.7953126430511475
    },
    {
      "epoch": 2.050018310546875e-05,
      "model_forward_time": 1.7617015838623047,
      "step": 95496
    },
    {
      "epoch": 2.050018310546875e-05,
      "step": 95496,
      "training_step_time": 7.6794068813323975
    },
    {
      "epoch": 2.0501136779785156e-05,
      "model_forward_time": 1.3930811882019043,
      "step": 95497
    },
    {
      "epoch": 2.0501136779785156e-05,
      "step": 95497,
      "training_step_time": 7.611120700836182
    },
    {
      "epoch": 2.0502090454101564e-05,
      "model_forward_time": 1.751185655593872,
      "step": 95498
    },
    {
      "epoch": 2.0502090454101564e-05,
      "step": 95498,
      "training_step_time": 7.361131429672241
    },
    {
      "epoch": 2.0503044128417968e-05,
      "model_forward_time": 1.753502607345581,
      "step": 95499
    },
    {
      "epoch": 2.0503044128417968e-05,
      "step": 95499,
      "training_step_time": 7.716728687286377
    },
    {
      "epoch": 2.0503997802734375e-05,
      "grad_norm": 0.08510180562734604,
      "learning_rate": 5.528516464305467e-08,
      "loss": 0.0545,
      "step": 95500
    },
    {
      "epoch": 2.0503997802734375e-05,
      "model_forward_time": 1.754044771194458,
      "step": 95500
    },
    {
      "dynamics_loss_avg": 0.05518216015771031,
      "epoch": 2.0503997802734375e-05,
      "step": 95500
    },
    {
      "action_loss_avg": 0.0010403090913314373,
      "epoch": 2.0503997802734375e-05,
      "step": 95500
    },
    {
      "epoch": 2.0503997802734375e-05,
      "step": 95500,
      "training_step_time": 7.662802219390869
    },
    {
      "epoch": 2.0504951477050782e-05,
      "model_forward_time": 1.3870649337768555,
      "step": 95501
    },
    {
      "epoch": 2.0504951477050782e-05,
      "step": 95501,
      "training_step_time": 7.864759922027588
    },
    {
      "epoch": 2.0505905151367186e-05,
      "model_forward_time": 1.052161455154419,
      "step": 95502
    },
    {
      "epoch": 2.0505905151367186e-05,
      "step": 95502,
      "training_step_time": 8.514134883880615
    },
    {
      "epoch": 2.0506858825683594e-05,
      "model_forward_time": 2.3684091567993164,
      "step": 95503
    },
    {
      "epoch": 2.0506858825683594e-05,
      "step": 95503,
      "training_step_time": 7.938229560852051
    },
    {
      "epoch": 2.05078125e-05,
      "model_forward_time": 1.7740230560302734,
      "step": 95504
    },
    {
      "epoch": 2.05078125e-05,
      "step": 95504,
      "training_step_time": 7.862378835678101
    },
    {
      "epoch": 2.0508766174316405e-05,
      "model_forward_time": 1.7638185024261475,
      "step": 95505
    },
    {
      "epoch": 2.0508766174316405e-05,
      "step": 95505,
      "training_step_time": 8.167752265930176
    },
    {
      "epoch": 2.0509719848632813e-05,
      "model_forward_time": 1.8436968326568604,
      "step": 95506
    },
    {
      "epoch": 2.0509719848632813e-05,
      "step": 95506,
      "training_step_time": 7.617350101470947
    },
    {
      "epoch": 2.051067352294922e-05,
      "model_forward_time": 1.7610993385314941,
      "step": 95507
    },
    {
      "epoch": 2.051067352294922e-05,
      "step": 95507,
      "training_step_time": 7.480792760848999
    },
    {
      "epoch": 2.0511627197265624e-05,
      "model_forward_time": 1.7548754215240479,
      "step": 95508
    },
    {
      "epoch": 2.0511627197265624e-05,
      "step": 95508,
      "training_step_time": 7.817796468734741
    },
    {
      "epoch": 2.051258087158203e-05,
      "model_forward_time": 1.7591400146484375,
      "step": 95509
    },
    {
      "epoch": 2.051258087158203e-05,
      "step": 95509,
      "training_step_time": 7.420224189758301
    },
    {
      "epoch": 2.051353454589844e-05,
      "grad_norm": 0.07363852113485336,
      "learning_rate": 5.504023147947068e-08,
      "loss": 0.054,
      "step": 95510
    },
    {
      "epoch": 2.051353454589844e-05,
      "model_forward_time": 1.763270616531372,
      "step": 95510
    },
    {
      "dynamics_loss_avg": 0.058797317743301394,
      "epoch": 2.051353454589844e-05,
      "step": 95510
    },
    {
      "action_loss_avg": 0.0013388884312007577,
      "epoch": 2.051353454589844e-05,
      "step": 95510
    },
    {
      "epoch": 2.051353454589844e-05,
      "step": 95510,
      "training_step_time": 7.809494495391846
    },
    {
      "epoch": 2.0514488220214843e-05,
      "model_forward_time": 1.7583844661712646,
      "step": 95511
    },
    {
      "epoch": 2.0514488220214843e-05,
      "step": 95511,
      "training_step_time": 8.362442016601562
    },
    {
      "epoch": 2.051544189453125e-05,
      "model_forward_time": 1.7591168880462646,
      "step": 95512
    },
    {
      "epoch": 2.051544189453125e-05,
      "step": 95512,
      "training_step_time": 9.763773679733276
    },
    {
      "epoch": 2.0516395568847658e-05,
      "model_forward_time": 2.403433322906494,
      "step": 95513
    },
    {
      "epoch": 2.0516395568847658e-05,
      "step": 95513,
      "training_step_time": 8.48584532737732
    },
    {
      "epoch": 2.0517349243164062e-05,
      "model_forward_time": 1.8283658027648926,
      "step": 95514
    },
    {
      "epoch": 2.0517349243164062e-05,
      "step": 95514,
      "training_step_time": 8.550981521606445
    },
    {
      "epoch": 2.051830291748047e-05,
      "model_forward_time": 1.7611584663391113,
      "step": 95515
    },
    {
      "epoch": 2.051830291748047e-05,
      "step": 95515,
      "training_step_time": 7.613141775131226
    },
    {
      "epoch": 2.0519256591796873e-05,
      "model_forward_time": 1.738403558731079,
      "step": 95516
    },
    {
      "epoch": 2.0519256591796873e-05,
      "step": 95516,
      "training_step_time": 7.616455316543579
    },
    {
      "epoch": 2.052021026611328e-05,
      "model_forward_time": 1.7583508491516113,
      "step": 95517
    },
    {
      "epoch": 2.052021026611328e-05,
      "step": 95517,
      "training_step_time": 7.735687255859375
    },
    {
      "epoch": 2.0521163940429688e-05,
      "model_forward_time": 1.753955602645874,
      "step": 95518
    },
    {
      "epoch": 2.0521163940429688e-05,
      "step": 95518,
      "training_step_time": 7.383058547973633
    },
    {
      "epoch": 2.0522117614746092e-05,
      "model_forward_time": 1.7479572296142578,
      "step": 95519
    },
    {
      "epoch": 2.0522117614746092e-05,
      "step": 95519,
      "training_step_time": 9.680808544158936
    },
    {
      "epoch": 2.05230712890625e-05,
      "grad_norm": 0.10878051072359085,
      "learning_rate": 5.479583908925279e-08,
      "loss": 0.0556,
      "step": 95520
    },
    {
      "epoch": 2.05230712890625e-05,
      "model_forward_time": 1.7656738758087158,
      "step": 95520
    },
    {
      "dynamics_loss_avg": 0.05004229135811329,
      "epoch": 2.05230712890625e-05,
      "step": 95520
    },
    {
      "action_loss_avg": 0.0008910557953640819,
      "epoch": 2.05230712890625e-05,
      "step": 95520
    },
    {
      "epoch": 2.05230712890625e-05,
      "step": 95520,
      "training_step_time": 9.176142930984497
    },
    {
      "epoch": 2.0524024963378907e-05,
      "model_forward_time": 1.7573423385620117,
      "step": 95521
    },
    {
      "epoch": 2.0524024963378907e-05,
      "step": 95521,
      "training_step_time": 8.929875373840332
    },
    {
      "epoch": 2.052497863769531e-05,
      "model_forward_time": 2.4220879077911377,
      "step": 95522
    },
    {
      "epoch": 2.052497863769531e-05,
      "step": 95522,
      "training_step_time": 8.894700288772583
    },
    {
      "epoch": 2.052593231201172e-05,
      "model_forward_time": 1.8226523399353027,
      "step": 95523
    },
    {
      "epoch": 2.052593231201172e-05,
      "step": 95523,
      "training_step_time": 8.535062313079834
    },
    {
      "epoch": 2.0526885986328126e-05,
      "model_forward_time": 1.765878438949585,
      "step": 95524
    },
    {
      "epoch": 2.0526885986328126e-05,
      "step": 95524,
      "training_step_time": 8.99972677230835
    },
    {
      "epoch": 2.052783966064453e-05,
      "model_forward_time": 1.758301019668579,
      "step": 95525
    },
    {
      "epoch": 2.052783966064453e-05,
      "step": 95525,
      "training_step_time": 7.633549928665161
    },
    {
      "epoch": 2.0528793334960937e-05,
      "model_forward_time": 1.7537143230438232,
      "step": 95526
    },
    {
      "epoch": 2.0528793334960937e-05,
      "step": 95526,
      "training_step_time": 7.039821147918701
    },
    {
      "epoch": 2.0529747009277345e-05,
      "model_forward_time": 1.7663953304290771,
      "step": 95527
    },
    {
      "epoch": 2.0529747009277345e-05,
      "step": 95527,
      "training_step_time": 7.711901426315308
    },
    {
      "epoch": 2.053070068359375e-05,
      "model_forward_time": 1.7594268321990967,
      "step": 95528
    },
    {
      "epoch": 2.053070068359375e-05,
      "step": 95528,
      "training_step_time": 7.751145124435425
    },
    {
      "epoch": 2.0531654357910156e-05,
      "model_forward_time": 1.7503089904785156,
      "step": 95529
    },
    {
      "epoch": 2.0531654357910156e-05,
      "step": 95529,
      "training_step_time": 8.561980724334717
    },
    {
      "epoch": 2.0532608032226564e-05,
      "grad_norm": 0.068865105509758,
      "learning_rate": 5.455198749912627e-08,
      "loss": 0.0543,
      "step": 95530
    },
    {
      "epoch": 2.0532608032226564e-05,
      "model_forward_time": 1.7719471454620361,
      "step": 95530
    },
    {
      "dynamics_loss_avg": 0.04915342982858419,
      "epoch": 2.0532608032226564e-05,
      "step": 95530
    },
    {
      "action_loss_avg": 0.00294968668022193,
      "epoch": 2.0532608032226564e-05,
      "step": 95530
    },
    {
      "epoch": 2.0532608032226564e-05,
      "step": 95530,
      "training_step_time": 7.7679619789123535
    },
    {
      "epoch": 2.0533561706542968e-05,
      "model_forward_time": 2.396601915359497,
      "step": 95531
    },
    {
      "epoch": 2.0533561706542968e-05,
      "step": 95531,
      "training_step_time": 7.777926683425903
    },
    {
      "epoch": 2.0534515380859375e-05,
      "model_forward_time": 1.7855539321899414,
      "step": 95532
    },
    {
      "epoch": 2.0534515380859375e-05,
      "step": 95532,
      "training_step_time": 7.847179651260376
    },
    {
      "epoch": 2.0535469055175782e-05,
      "model_forward_time": 1.7913308143615723,
      "step": 95533
    },
    {
      "epoch": 2.0535469055175782e-05,
      "step": 95533,
      "training_step_time": 7.5810041427612305
    },
    {
      "epoch": 2.0536422729492186e-05,
      "model_forward_time": 1.7518343925476074,
      "step": 95534
    },
    {
      "epoch": 2.0536422729492186e-05,
      "step": 95534,
      "training_step_time": 7.554697513580322
    },
    {
      "epoch": 2.0537376403808594e-05,
      "model_forward_time": 1.7518305778503418,
      "step": 95535
    },
    {
      "epoch": 2.0537376403808594e-05,
      "step": 95535,
      "training_step_time": 6.955974340438843
    },
    {
      "epoch": 2.0538330078125e-05,
      "model_forward_time": 1.7516601085662842,
      "step": 95536
    },
    {
      "epoch": 2.0538330078125e-05,
      "step": 95536,
      "training_step_time": 7.764454126358032
    },
    {
      "epoch": 2.0539283752441405e-05,
      "model_forward_time": 1.7519254684448242,
      "step": 95537
    },
    {
      "epoch": 2.0539283752441405e-05,
      "step": 95537,
      "training_step_time": 8.41560411453247
    },
    {
      "epoch": 2.0540237426757813e-05,
      "model_forward_time": 1.7592411041259766,
      "step": 95538
    },
    {
      "epoch": 2.0540237426757813e-05,
      "step": 95538,
      "training_step_time": 8.826982736587524
    },
    {
      "epoch": 2.054119110107422e-05,
      "model_forward_time": 1.7514982223510742,
      "step": 95539
    },
    {
      "epoch": 2.054119110107422e-05,
      "step": 95539,
      "training_step_time": 9.196390390396118
    },
    {
      "epoch": 2.0542144775390624e-05,
      "grad_norm": 0.07550530135631561,
      "learning_rate": 5.4308676735759793e-08,
      "loss": 0.0541,
      "step": 95540
    },
    {
      "epoch": 2.0542144775390624e-05,
      "model_forward_time": 2.4391071796417236,
      "step": 95540
    },
    {
      "dynamics_loss_avg": 0.05710010062903166,
      "epoch": 2.0542144775390624e-05,
      "step": 95540
    },
    {
      "action_loss_avg": 0.0013441215211059897,
      "epoch": 2.0542144775390624e-05,
      "step": 95540
    },
    {
      "epoch": 2.0542144775390624e-05,
      "step": 95540,
      "training_step_time": 8.72778606414795
    },
    {
      "epoch": 2.054309844970703e-05,
      "model_forward_time": 1.74851393699646,
      "step": 95541
    },
    {
      "epoch": 2.054309844970703e-05,
      "step": 95541,
      "training_step_time": 7.968181133270264
    },
    {
      "epoch": 2.054405212402344e-05,
      "model_forward_time": 1.8178389072418213,
      "step": 95542
    },
    {
      "epoch": 2.054405212402344e-05,
      "step": 95542,
      "training_step_time": 7.677429437637329
    },
    {
      "epoch": 2.0545005798339843e-05,
      "model_forward_time": 1.7504827976226807,
      "step": 95543
    },
    {
      "epoch": 2.0545005798339843e-05,
      "step": 95543,
      "training_step_time": 8.610220432281494
    },
    {
      "epoch": 2.054595947265625e-05,
      "model_forward_time": 1.7584128379821777,
      "step": 95544
    },
    {
      "epoch": 2.054595947265625e-05,
      "step": 95544,
      "training_step_time": 7.528960943222046
    },
    {
      "epoch": 2.0546913146972658e-05,
      "model_forward_time": 1.7479679584503174,
      "step": 95545
    },
    {
      "epoch": 2.0546913146972658e-05,
      "step": 95545,
      "training_step_time": 7.3594651222229
    },
    {
      "epoch": 2.0547866821289062e-05,
      "model_forward_time": 1.7346115112304688,
      "step": 95546
    },
    {
      "epoch": 2.0547866821289062e-05,
      "step": 95546,
      "training_step_time": 7.2754645347595215
    },
    {
      "epoch": 2.054882049560547e-05,
      "model_forward_time": 1.740030288696289,
      "step": 95547
    },
    {
      "epoch": 2.054882049560547e-05,
      "step": 95547,
      "training_step_time": 9.596001625061035
    },
    {
      "epoch": 2.0549774169921873e-05,
      "model_forward_time": 1.7491059303283691,
      "step": 95548
    },
    {
      "epoch": 2.0549774169921873e-05,
      "step": 95548,
      "training_step_time": 8.342466592788696
    },
    {
      "epoch": 2.055072784423828e-05,
      "model_forward_time": 2.387474536895752,
      "step": 95549
    },
    {
      "epoch": 2.055072784423828e-05,
      "step": 95549,
      "training_step_time": 10.39093565940857
    },
    {
      "epoch": 2.0551681518554688e-05,
      "grad_norm": 0.10860888659954071,
      "learning_rate": 5.406590682575985e-08,
      "loss": 0.0556,
      "step": 95550
    },
    {
      "epoch": 2.0551681518554688e-05,
      "model_forward_time": 1.745389461517334,
      "step": 95550
    },
    {
      "dynamics_loss_avg": 0.05742431376129389,
      "epoch": 2.0551681518554688e-05,
      "step": 95550
    },
    {
      "action_loss_avg": 0.001271577220177278,
      "epoch": 2.0551681518554688e-05,
      "step": 95550
    },
    {
      "epoch": 2.0551681518554688e-05,
      "step": 95550,
      "training_step_time": 8.746544361114502
    },
    {
      "epoch": 2.0552635192871092e-05,
      "model_forward_time": 1.8133063316345215,
      "step": 95551
    },
    {
      "epoch": 2.0552635192871092e-05,
      "step": 95551,
      "training_step_time": 7.704026460647583
    },
    {
      "epoch": 2.05535888671875e-05,
      "model_forward_time": 1.7476515769958496,
      "step": 95552
    },
    {
      "epoch": 2.05535888671875e-05,
      "step": 95552,
      "training_step_time": 7.563777446746826
    },
    {
      "epoch": 2.0554542541503907e-05,
      "model_forward_time": 1.745854377746582,
      "step": 95553
    },
    {
      "epoch": 2.0554542541503907e-05,
      "step": 95553,
      "training_step_time": 7.590863227844238
    },
    {
      "epoch": 2.055549621582031e-05,
      "model_forward_time": 1.3857839107513428,
      "step": 95554
    },
    {
      "epoch": 2.055549621582031e-05,
      "step": 95554,
      "training_step_time": 7.86315131187439
    },
    {
      "epoch": 2.055644989013672e-05,
      "model_forward_time": 1.748633861541748,
      "step": 95555
    },
    {
      "epoch": 2.055644989013672e-05,
      "step": 95555,
      "training_step_time": 7.67040491104126
    },
    {
      "epoch": 2.0557403564453126e-05,
      "model_forward_time": 1.7409780025482178,
      "step": 95556
    },
    {
      "epoch": 2.0557403564453126e-05,
      "step": 95556,
      "training_step_time": 8.692145347595215
    },
    {
      "epoch": 2.055835723876953e-05,
      "model_forward_time": 1.736722707748413,
      "step": 95557
    },
    {
      "epoch": 2.055835723876953e-05,
      "step": 95557,
      "training_step_time": 8.513928413391113
    },
    {
      "epoch": 2.0559310913085937e-05,
      "model_forward_time": 1.3873026371002197,
      "step": 95558
    },
    {
      "epoch": 2.0559310913085937e-05,
      "step": 95558,
      "training_step_time": 7.918241262435913
    },
    {
      "epoch": 2.0560264587402345e-05,
      "model_forward_time": 2.469660520553589,
      "step": 95559
    },
    {
      "epoch": 2.0560264587402345e-05,
      "step": 95559,
      "training_step_time": 7.948538780212402
    },
    {
      "epoch": 2.056121826171875e-05,
      "grad_norm": 0.10297121852636337,
      "learning_rate": 5.3823677795676875e-08,
      "loss": 0.0559,
      "step": 95560
    },
    {
      "epoch": 2.056121826171875e-05,
      "model_forward_time": 1.7972183227539062,
      "step": 95560
    },
    {
      "dynamics_loss_avg": 0.04135993495583534,
      "epoch": 2.056121826171875e-05,
      "step": 95560
    },
    {
      "action_loss_avg": 0.0007914624409750104,
      "epoch": 2.056121826171875e-05,
      "step": 95560
    },
    {
      "epoch": 2.056121826171875e-05,
      "step": 95560,
      "training_step_time": 7.848644971847534
    },
    {
      "epoch": 2.0562171936035156e-05,
      "model_forward_time": 1.7494935989379883,
      "step": 95561
    },
    {
      "epoch": 2.0562171936035156e-05,
      "step": 95561,
      "training_step_time": 7.614126920700073
    },
    {
      "epoch": 2.0563125610351564e-05,
      "model_forward_time": 1.7503468990325928,
      "step": 95562
    },
    {
      "epoch": 2.0563125610351564e-05,
      "step": 95562,
      "training_step_time": 7.541951656341553
    },
    {
      "epoch": 2.0564079284667968e-05,
      "model_forward_time": 1.7538764476776123,
      "step": 95563
    },
    {
      "epoch": 2.0564079284667968e-05,
      "step": 95563,
      "training_step_time": 7.020816802978516
    },
    {
      "epoch": 2.0565032958984375e-05,
      "model_forward_time": 1.7448229789733887,
      "step": 95564
    },
    {
      "epoch": 2.0565032958984375e-05,
      "step": 95564,
      "training_step_time": 7.739814519882202
    },
    {
      "epoch": 2.0565986633300782e-05,
      "model_forward_time": 1.7610316276550293,
      "step": 95565
    },
    {
      "epoch": 2.0565986633300782e-05,
      "step": 95565,
      "training_step_time": 11.287636995315552
    },
    {
      "epoch": 2.0566940307617186e-05,
      "model_forward_time": 1.7578330039978027,
      "step": 95566
    },
    {
      "epoch": 2.0566940307617186e-05,
      "step": 95566,
      "training_step_time": 7.834892988204956
    },
    {
      "epoch": 2.0567893981933594e-05,
      "model_forward_time": 1.8434598445892334,
      "step": 95567
    },
    {
      "epoch": 2.0567893981933594e-05,
      "step": 95567,
      "training_step_time": 9.772031784057617
    },
    {
      "epoch": 2.056884765625e-05,
      "model_forward_time": 2.469663619995117,
      "step": 95568
    },
    {
      "epoch": 2.056884765625e-05,
      "step": 95568,
      "training_step_time": 8.857791662216187
    },
    {
      "epoch": 2.0569801330566405e-05,
      "model_forward_time": 1.7581501007080078,
      "step": 95569
    },
    {
      "epoch": 2.0569801330566405e-05,
      "step": 95569,
      "training_step_time": 7.922816038131714
    },
    {
      "epoch": 2.0570755004882813e-05,
      "grad_norm": 0.08367603272199631,
      "learning_rate": 5.3581989672000235e-08,
      "loss": 0.056,
      "step": 95570
    },
    {
      "epoch": 2.0570755004882813e-05,
      "model_forward_time": 1.8234426975250244,
      "step": 95570
    },
    {
      "dynamics_loss_avg": 0.06659402921795846,
      "epoch": 2.0570755004882813e-05,
      "step": 95570
    },
    {
      "action_loss_avg": 0.0022386269993148744,
      "epoch": 2.0570755004882813e-05,
      "step": 95570
    },
    {
      "epoch": 2.0570755004882813e-05,
      "step": 95570,
      "training_step_time": 8.71948790550232
    },
    {
      "epoch": 2.057170867919922e-05,
      "model_forward_time": 1.7478313446044922,
      "step": 95571
    },
    {
      "epoch": 2.057170867919922e-05,
      "step": 95571,
      "training_step_time": 7.555622816085815
    },
    {
      "epoch": 2.0572662353515624e-05,
      "model_forward_time": 1.7474851608276367,
      "step": 95572
    },
    {
      "epoch": 2.0572662353515624e-05,
      "step": 95572,
      "training_step_time": 6.9980247020721436
    },
    {
      "epoch": 2.057361602783203e-05,
      "model_forward_time": 1.7525608539581299,
      "step": 95573
    },
    {
      "epoch": 2.057361602783203e-05,
      "step": 95573,
      "training_step_time": 7.685061454772949
    },
    {
      "epoch": 2.057456970214844e-05,
      "model_forward_time": 1.8207545280456543,
      "step": 95574
    },
    {
      "epoch": 2.057456970214844e-05,
      "step": 95574,
      "training_step_time": 7.6887335777282715
    },
    {
      "epoch": 2.0575523376464843e-05,
      "model_forward_time": 1.7557454109191895,
      "step": 95575
    },
    {
      "epoch": 2.0575523376464843e-05,
      "step": 95575,
      "training_step_time": 8.54361367225647
    },
    {
      "epoch": 2.057647705078125e-05,
      "model_forward_time": 1.748154878616333,
      "step": 95576
    },
    {
      "epoch": 2.057647705078125e-05,
      "step": 95576,
      "training_step_time": 8.659842491149902
    },
    {
      "epoch": 2.0577430725097658e-05,
      "model_forward_time": 2.4208757877349854,
      "step": 95577
    },
    {
      "epoch": 2.0577430725097658e-05,
      "step": 95577,
      "training_step_time": 8.501310348510742
    },
    {
      "epoch": 2.0578384399414062e-05,
      "model_forward_time": 1.7692296504974365,
      "step": 95578
    },
    {
      "epoch": 2.0578384399414062e-05,
      "step": 95578,
      "training_step_time": 7.872588396072388
    },
    {
      "epoch": 2.057933807373047e-05,
      "model_forward_time": 1.7860443592071533,
      "step": 95579
    },
    {
      "epoch": 2.057933807373047e-05,
      "step": 95579,
      "training_step_time": 8.676513195037842
    },
    {
      "epoch": 2.0580291748046873e-05,
      "grad_norm": 0.13813012838363647,
      "learning_rate": 5.3340842481159895e-08,
      "loss": 0.0565,
      "step": 95580
    },
    {
      "epoch": 2.0580291748046873e-05,
      "model_forward_time": 1.7443230152130127,
      "step": 95580
    },
    {
      "dynamics_loss_avg": 0.06142481938004494,
      "epoch": 2.0580291748046873e-05,
      "step": 95580
    },
    {
      "action_loss_avg": 0.0009782418434042483,
      "epoch": 2.0580291748046873e-05,
      "step": 95580
    },
    {
      "epoch": 2.0580291748046873e-05,
      "step": 95580,
      "training_step_time": 7.576515436172485
    },
    {
      "epoch": 2.058124542236328e-05,
      "model_forward_time": 1.7537055015563965,
      "step": 95581
    },
    {
      "epoch": 2.058124542236328e-05,
      "step": 95581,
      "training_step_time": 7.443448543548584
    },
    {
      "epoch": 2.0582199096679688e-05,
      "model_forward_time": 1.7552340030670166,
      "step": 95582
    },
    {
      "epoch": 2.0582199096679688e-05,
      "step": 95582,
      "training_step_time": 7.673010349273682
    },
    {
      "epoch": 2.0583152770996092e-05,
      "model_forward_time": 1.7554152011871338,
      "step": 95583
    },
    {
      "epoch": 2.0583152770996092e-05,
      "step": 95583,
      "training_step_time": 7.634854555130005
    },
    {
      "epoch": 2.05841064453125e-05,
      "model_forward_time": 1.7489447593688965,
      "step": 95584
    },
    {
      "epoch": 2.05841064453125e-05,
      "step": 95584,
      "training_step_time": 7.785099267959595
    },
    {
      "epoch": 2.0585060119628907e-05,
      "model_forward_time": 1.747981071472168,
      "step": 95585
    },
    {
      "epoch": 2.0585060119628907e-05,
      "step": 95585,
      "training_step_time": 9.793049335479736
    },
    {
      "epoch": 2.058601379394531e-05,
      "model_forward_time": 2.378209114074707,
      "step": 95586
    },
    {
      "epoch": 2.058601379394531e-05,
      "step": 95586,
      "training_step_time": 9.617923736572266
    },
    {
      "epoch": 2.058696746826172e-05,
      "model_forward_time": 1.7884187698364258,
      "step": 95587
    },
    {
      "epoch": 2.058696746826172e-05,
      "step": 95587,
      "training_step_time": 7.984116315841675
    },
    {
      "epoch": 2.0587921142578126e-05,
      "model_forward_time": 1.7509360313415527,
      "step": 95588
    },
    {
      "epoch": 2.0587921142578126e-05,
      "step": 95588,
      "training_step_time": 8.727383136749268
    },
    {
      "epoch": 2.058887481689453e-05,
      "model_forward_time": 1.7574653625488281,
      "step": 95589
    },
    {
      "epoch": 2.058887481689453e-05,
      "step": 95589,
      "training_step_time": 7.763209342956543
    },
    {
      "epoch": 2.0589828491210937e-05,
      "grad_norm": 0.08780889958143234,
      "learning_rate": 5.3100236249527537e-08,
      "loss": 0.055,
      "step": 95590
    },
    {
      "epoch": 2.0589828491210937e-05,
      "model_forward_time": 1.7514111995697021,
      "step": 95590
    },
    {
      "dynamics_loss_avg": 0.060150198824703696,
      "epoch": 2.0589828491210937e-05,
      "step": 95590
    },
    {
      "action_loss_avg": 0.0016284266050206498,
      "epoch": 2.0589828491210937e-05,
      "step": 95590
    },
    {
      "epoch": 2.0589828491210937e-05,
      "step": 95590,
      "training_step_time": 7.487186431884766
    },
    {
      "epoch": 2.0590782165527345e-05,
      "model_forward_time": 1.7868947982788086,
      "step": 95591
    },
    {
      "epoch": 2.0590782165527345e-05,
      "step": 95591,
      "training_step_time": 7.739478826522827
    },
    {
      "epoch": 2.059173583984375e-05,
      "model_forward_time": 1.7490177154541016,
      "step": 95592
    },
    {
      "epoch": 2.059173583984375e-05,
      "step": 95592,
      "training_step_time": 7.6173694133758545
    },
    {
      "epoch": 2.0592689514160156e-05,
      "model_forward_time": 1.7556352615356445,
      "step": 95593
    },
    {
      "epoch": 2.0592689514160156e-05,
      "step": 95593,
      "training_step_time": 8.656087160110474
    },
    {
      "epoch": 2.0593643188476564e-05,
      "model_forward_time": 1.7557542324066162,
      "step": 95594
    },
    {
      "epoch": 2.0593643188476564e-05,
      "step": 95594,
      "training_step_time": 8.50909423828125
    },
    {
      "epoch": 2.0594596862792968e-05,
      "model_forward_time": 1.762268304824829,
      "step": 95595
    },
    {
      "epoch": 2.0594596862792968e-05,
      "step": 95595,
      "training_step_time": 9.607706308364868
    },
    {
      "epoch": 2.0595550537109375e-05,
      "model_forward_time": 2.4153499603271484,
      "step": 95596
    },
    {
      "epoch": 2.0595550537109375e-05,
      "step": 95596,
      "training_step_time": 7.881278991699219
    },
    {
      "epoch": 2.0596504211425782e-05,
      "model_forward_time": 1.7558705806732178,
      "step": 95597
    },
    {
      "epoch": 2.0596504211425782e-05,
      "step": 95597,
      "training_step_time": 7.712397575378418
    },
    {
      "epoch": 2.0597457885742187e-05,
      "model_forward_time": 1.825216293334961,
      "step": 95598
    },
    {
      "epoch": 2.0597457885742187e-05,
      "step": 95598,
      "training_step_time": 7.831996917724609
    },
    {
      "epoch": 2.0598411560058594e-05,
      "model_forward_time": 1.7376248836517334,
      "step": 95599
    },
    {
      "epoch": 2.0598411560058594e-05,
      "step": 95599,
      "training_step_time": 7.5057477951049805
    },
    {
      "epoch": 2.0599365234375e-05,
      "grad_norm": 0.08699726313352585,
      "learning_rate": 5.2860171003416563e-08,
      "loss": 0.0566,
      "step": 95600
    },
    {
      "epoch": 2.0599365234375e-05,
      "model_forward_time": 1.7606732845306396,
      "step": 95600
    },
    {
      "dynamics_loss_avg": 0.06500694155693054,
      "epoch": 2.0599365234375e-05,
      "step": 95600
    },
    {
      "action_loss_avg": 0.0009565094514982775,
      "epoch": 2.0599365234375e-05,
      "step": 95600
    },
    {
      "epoch": 2.0599365234375e-05,
      "step": 95600,
      "training_step_time": 7.672842502593994
    },
    {
      "epoch": 2.0600318908691405e-05,
      "model_forward_time": 1.8363258838653564,
      "step": 95601
    },
    {
      "epoch": 2.0600318908691405e-05,
      "step": 95601,
      "training_step_time": 7.677009582519531
    },
    {
      "epoch": 2.0601272583007813e-05,
      "model_forward_time": 1.759734869003296,
      "step": 95602
    },
    {
      "epoch": 2.0601272583007813e-05,
      "step": 95602,
      "training_step_time": 8.28130030632019
    },
    {
      "epoch": 2.060222625732422e-05,
      "model_forward_time": 1.7469184398651123,
      "step": 95603
    },
    {
      "epoch": 2.060222625732422e-05,
      "step": 95603,
      "training_step_time": 8.645346641540527
    },
    {
      "epoch": 2.0603179931640624e-05,
      "model_forward_time": 1.7528817653656006,
      "step": 95604
    },
    {
      "epoch": 2.0603179931640624e-05,
      "step": 95604,
      "training_step_time": 10.705354690551758
    },
    {
      "epoch": 2.060413360595703e-05,
      "model_forward_time": 2.4115679264068604,
      "step": 95605
    },
    {
      "epoch": 2.060413360595703e-05,
      "step": 95605,
      "training_step_time": 8.863327026367188
    },
    {
      "epoch": 2.060508728027344e-05,
      "model_forward_time": 1.7507212162017822,
      "step": 95606
    },
    {
      "epoch": 2.060508728027344e-05,
      "step": 95606,
      "training_step_time": 7.67644190788269
    },
    {
      "epoch": 2.0606040954589843e-05,
      "model_forward_time": 1.7393786907196045,
      "step": 95607
    },
    {
      "epoch": 2.0606040954589843e-05,
      "step": 95607,
      "training_step_time": 7.692866802215576
    },
    {
      "epoch": 2.060699462890625e-05,
      "model_forward_time": 1.7615580558776855,
      "step": 95608
    },
    {
      "epoch": 2.060699462890625e-05,
      "step": 95608,
      "training_step_time": 7.469308614730835
    },
    {
      "epoch": 2.0607948303222658e-05,
      "model_forward_time": 1.7885892391204834,
      "step": 95609
    },
    {
      "epoch": 2.0607948303222658e-05,
      "step": 95609,
      "training_step_time": 7.798743963241577
    },
    {
      "epoch": 2.0608901977539062e-05,
      "grad_norm": 0.07898207753896713,
      "learning_rate": 5.262064676907874e-08,
      "loss": 0.0558,
      "step": 95610
    },
    {
      "epoch": 2.0608901977539062e-05,
      "model_forward_time": 1.788604497909546,
      "step": 95610
    },
    {
      "dynamics_loss_avg": 0.06556898355484009,
      "epoch": 2.0608901977539062e-05,
      "step": 95610
    },
    {
      "action_loss_avg": 0.0014924924762453884,
      "epoch": 2.0608901977539062e-05,
      "step": 95610
    },
    {
      "epoch": 2.0608901977539062e-05,
      "step": 95610,
      "training_step_time": 7.636835813522339
    },
    {
      "epoch": 2.060985565185547e-05,
      "model_forward_time": 1.77046799659729,
      "step": 95611
    },
    {
      "epoch": 2.060985565185547e-05,
      "step": 95611,
      "training_step_time": 7.866394758224487
    },
    {
      "epoch": 2.0610809326171873e-05,
      "model_forward_time": 1.7670116424560547,
      "step": 95612
    },
    {
      "epoch": 2.0610809326171873e-05,
      "step": 95612,
      "training_step_time": 9.381828308105469
    },
    {
      "epoch": 2.061176300048828e-05,
      "model_forward_time": 1.821763277053833,
      "step": 95613
    },
    {
      "epoch": 2.061176300048828e-05,
      "step": 95613,
      "training_step_time": 8.836822986602783
    },
    {
      "epoch": 2.0612716674804688e-05,
      "model_forward_time": 2.4418747425079346,
      "step": 95614
    },
    {
      "epoch": 2.0612716674804688e-05,
      "step": 95614,
      "training_step_time": 9.808619499206543
    },
    {
      "epoch": 2.0613670349121092e-05,
      "model_forward_time": 1.7737510204315186,
      "step": 95615
    },
    {
      "epoch": 2.0613670349121092e-05,
      "step": 95615,
      "training_step_time": 8.479257345199585
    },
    {
      "epoch": 2.06146240234375e-05,
      "model_forward_time": 1.4139180183410645,
      "step": 95616
    },
    {
      "epoch": 2.06146240234375e-05,
      "step": 95616,
      "training_step_time": 7.734681606292725
    },
    {
      "epoch": 2.0615577697753907e-05,
      "model_forward_time": 1.7451834678649902,
      "step": 95617
    },
    {
      "epoch": 2.0615577697753907e-05,
      "step": 95617,
      "training_step_time": 7.726570129394531
    },
    {
      "epoch": 2.061653137207031e-05,
      "model_forward_time": 1.7845182418823242,
      "step": 95618
    },
    {
      "epoch": 2.061653137207031e-05,
      "step": 95618,
      "training_step_time": 7.705007553100586
    },
    {
      "epoch": 2.061748504638672e-05,
      "model_forward_time": 1.7540652751922607,
      "step": 95619
    },
    {
      "epoch": 2.061748504638672e-05,
      "step": 95619,
      "training_step_time": 7.672669172286987
    },
    {
      "epoch": 2.0618438720703126e-05,
      "grad_norm": 0.10576458275318146,
      "learning_rate": 5.238166357270924e-08,
      "loss": 0.0551,
      "step": 95620
    },
    {
      "epoch": 2.0618438720703126e-05,
      "model_forward_time": 1.7968261241912842,
      "step": 95620
    },
    {
      "dynamics_loss_avg": 0.08124256543815137,
      "epoch": 2.0618438720703126e-05,
      "step": 95620
    },
    {
      "action_loss_avg": 0.001980804730555974,
      "epoch": 2.0618438720703126e-05,
      "step": 95620
    },
    {
      "epoch": 2.0618438720703126e-05,
      "step": 95620,
      "training_step_time": 7.733781576156616
    },
    {
      "epoch": 2.061939239501953e-05,
      "model_forward_time": 1.7659873962402344,
      "step": 95621
    },
    {
      "epoch": 2.061939239501953e-05,
      "step": 95621,
      "training_step_time": 8.547058582305908
    },
    {
      "epoch": 2.0620346069335937e-05,
      "model_forward_time": 1.7676010131835938,
      "step": 95622
    },
    {
      "epoch": 2.0620346069335937e-05,
      "step": 95622,
      "training_step_time": 8.851500272750854
    },
    {
      "epoch": 2.0621299743652345e-05,
      "model_forward_time": 2.4372458457946777,
      "step": 95623
    },
    {
      "epoch": 2.0621299743652345e-05,
      "step": 95623,
      "training_step_time": 8.463330745697021
    },
    {
      "epoch": 2.062225341796875e-05,
      "model_forward_time": 1.7559080123901367,
      "step": 95624
    },
    {
      "epoch": 2.062225341796875e-05,
      "step": 95624,
      "training_step_time": 8.656349182128906
    },
    {
      "epoch": 2.0623207092285156e-05,
      "model_forward_time": 1.7425217628479004,
      "step": 95625
    },
    {
      "epoch": 2.0623207092285156e-05,
      "step": 95625,
      "training_step_time": 7.688931703567505
    },
    {
      "epoch": 2.0624160766601564e-05,
      "model_forward_time": 1.7548425197601318,
      "step": 95626
    },
    {
      "epoch": 2.0624160766601564e-05,
      "step": 95626,
      "training_step_time": 7.799253702163696
    },
    {
      "epoch": 2.0625114440917968e-05,
      "model_forward_time": 1.7726576328277588,
      "step": 95627
    },
    {
      "epoch": 2.0625114440917968e-05,
      "step": 95627,
      "training_step_time": 7.74715256690979
    },
    {
      "epoch": 2.0626068115234375e-05,
      "model_forward_time": 1.7822964191436768,
      "step": 95628
    },
    {
      "epoch": 2.0626068115234375e-05,
      "step": 95628,
      "training_step_time": 7.709479570388794
    },
    {
      "epoch": 2.0627021789550783e-05,
      "model_forward_time": 1.7800612449645996,
      "step": 95629
    },
    {
      "epoch": 2.0627021789550783e-05,
      "step": 95629,
      "training_step_time": 7.841066122055054
    },
    {
      "epoch": 2.0627975463867187e-05,
      "grad_norm": 0.07422544062137604,
      "learning_rate": 5.214322144044215e-08,
      "loss": 0.0551,
      "step": 95630
    },
    {
      "epoch": 2.0627975463867187e-05,
      "model_forward_time": 1.7589178085327148,
      "step": 95630
    },
    {
      "dynamics_loss_avg": 0.04921682644635439,
      "epoch": 2.0627975463867187e-05,
      "step": 95630
    },
    {
      "action_loss_avg": 0.0019902789732441305,
      "epoch": 2.0627975463867187e-05,
      "step": 95630
    },
    {
      "epoch": 2.0627975463867187e-05,
      "step": 95630,
      "training_step_time": 10.730586767196655
    },
    {
      "epoch": 2.0628929138183594e-05,
      "model_forward_time": 1.7581727504730225,
      "step": 95631
    },
    {
      "epoch": 2.0628929138183594e-05,
      "step": 95631,
      "training_step_time": 9.215608835220337
    },
    {
      "epoch": 2.06298828125e-05,
      "model_forward_time": 2.482104539871216,
      "step": 95632
    },
    {
      "epoch": 2.06298828125e-05,
      "step": 95632,
      "training_step_time": 8.930299520492554
    },
    {
      "epoch": 2.0630836486816405e-05,
      "model_forward_time": 1.7685699462890625,
      "step": 95633
    },
    {
      "epoch": 2.0630836486816405e-05,
      "step": 95633,
      "training_step_time": 8.733487606048584
    },
    {
      "epoch": 2.0631790161132813e-05,
      "model_forward_time": 1.7731285095214844,
      "step": 95634
    },
    {
      "epoch": 2.0631790161132813e-05,
      "step": 95634,
      "training_step_time": 8.18907642364502
    },
    {
      "epoch": 2.063274383544922e-05,
      "model_forward_time": 1.7656142711639404,
      "step": 95635
    },
    {
      "epoch": 2.063274383544922e-05,
      "step": 95635,
      "training_step_time": 7.723689317703247
    },
    {
      "epoch": 2.0633697509765624e-05,
      "model_forward_time": 1.3939018249511719,
      "step": 95636
    },
    {
      "epoch": 2.0633697509765624e-05,
      "step": 95636,
      "training_step_time": 7.665061950683594
    },
    {
      "epoch": 2.063465118408203e-05,
      "model_forward_time": 1.7633423805236816,
      "step": 95637
    },
    {
      "epoch": 2.063465118408203e-05,
      "step": 95637,
      "training_step_time": 8.533774375915527
    },
    {
      "epoch": 2.063560485839844e-05,
      "model_forward_time": 1.7679588794708252,
      "step": 95638
    },
    {
      "epoch": 2.063560485839844e-05,
      "step": 95638,
      "training_step_time": 7.765967845916748
    },
    {
      "epoch": 2.0636558532714843e-05,
      "model_forward_time": 1.7696113586425781,
      "step": 95639
    },
    {
      "epoch": 2.0636558532714843e-05,
      "step": 95639,
      "training_step_time": 9.550872325897217
    },
    {
      "epoch": 2.063751220703125e-05,
      "grad_norm": 0.07517237961292267,
      "learning_rate": 5.190532039835328e-08,
      "loss": 0.0537,
      "step": 95640
    },
    {
      "epoch": 2.063751220703125e-05,
      "model_forward_time": 1.7583339214324951,
      "step": 95640
    },
    {
      "dynamics_loss_avg": 0.061451094970107076,
      "epoch": 2.063751220703125e-05,
      "step": 95640
    },
    {
      "action_loss_avg": 0.001695437030866742,
      "epoch": 2.063751220703125e-05,
      "step": 95640
    },
    {
      "epoch": 2.063751220703125e-05,
      "step": 95640,
      "training_step_time": 8.533042907714844
    },
    {
      "epoch": 2.0638465881347655e-05,
      "model_forward_time": 1.4572339057922363,
      "step": 95641
    },
    {
      "epoch": 2.0638465881347655e-05,
      "step": 95641,
      "training_step_time": 8.85151219367981
    },
    {
      "epoch": 2.0639419555664062e-05,
      "model_forward_time": 2.459393262863159,
      "step": 95642
    },
    {
      "epoch": 2.0639419555664062e-05,
      "step": 95642,
      "training_step_time": 9.357058048248291
    },
    {
      "epoch": 2.064037322998047e-05,
      "model_forward_time": 1.7696821689605713,
      "step": 95643
    },
    {
      "epoch": 2.064037322998047e-05,
      "step": 95643,
      "training_step_time": 7.693875312805176
    },
    {
      "epoch": 2.0641326904296873e-05,
      "model_forward_time": 1.77170729637146,
      "step": 95644
    },
    {
      "epoch": 2.0641326904296873e-05,
      "step": 95644,
      "training_step_time": 7.706555366516113
    },
    {
      "epoch": 2.064228057861328e-05,
      "model_forward_time": 1.7837882041931152,
      "step": 95645
    },
    {
      "epoch": 2.064228057861328e-05,
      "step": 95645,
      "training_step_time": 7.501342535018921
    },
    {
      "epoch": 2.0643234252929688e-05,
      "model_forward_time": 1.800041675567627,
      "step": 95646
    },
    {
      "epoch": 2.0643234252929688e-05,
      "step": 95646,
      "training_step_time": 7.701160907745361
    },
    {
      "epoch": 2.0644187927246092e-05,
      "model_forward_time": 1.7672514915466309,
      "step": 95647
    },
    {
      "epoch": 2.0644187927246092e-05,
      "step": 95647,
      "training_step_time": 7.721953630447388
    },
    {
      "epoch": 2.06451416015625e-05,
      "model_forward_time": 1.75523042678833,
      "step": 95648
    },
    {
      "epoch": 2.06451416015625e-05,
      "step": 95648,
      "training_step_time": 7.679942846298218
    },
    {
      "epoch": 2.0646095275878907e-05,
      "model_forward_time": 1.7562768459320068,
      "step": 95649
    },
    {
      "epoch": 2.0646095275878907e-05,
      "step": 95649,
      "training_step_time": 8.493733882904053
    },
    {
      "epoch": 2.064704895019531e-05,
      "grad_norm": 0.09017372876405716,
      "learning_rate": 5.1667960472459034e-08,
      "loss": 0.0549,
      "step": 95650
    },
    {
      "epoch": 2.064704895019531e-05,
      "model_forward_time": 1.8099772930145264,
      "step": 95650
    },
    {
      "dynamics_loss_avg": 0.06658222936093808,
      "epoch": 2.064704895019531e-05,
      "step": 95650
    },
    {
      "action_loss_avg": 0.0009559909100062214,
      "epoch": 2.064704895019531e-05,
      "step": 95650
    },
    {
      "epoch": 2.064704895019531e-05,
      "step": 95650,
      "training_step_time": 7.858471870422363
    },
    {
      "epoch": 2.064800262451172e-05,
      "model_forward_time": 2.4847354888916016,
      "step": 95651
    },
    {
      "epoch": 2.064800262451172e-05,
      "step": 95651,
      "training_step_time": 9.707390308380127
    },
    {
      "epoch": 2.0648956298828126e-05,
      "model_forward_time": 1.7445695400238037,
      "step": 95652
    },
    {
      "epoch": 2.0648956298828126e-05,
      "step": 95652,
      "training_step_time": 7.764196157455444
    },
    {
      "epoch": 2.064990997314453e-05,
      "model_forward_time": 1.784848690032959,
      "step": 95653
    },
    {
      "epoch": 2.064990997314453e-05,
      "step": 95653,
      "training_step_time": 8.958469152450562
    },
    {
      "epoch": 2.0650863647460937e-05,
      "model_forward_time": 1.3886425495147705,
      "step": 95654
    },
    {
      "epoch": 2.0650863647460937e-05,
      "step": 95654,
      "training_step_time": 7.642394065856934
    },
    {
      "epoch": 2.0651817321777345e-05,
      "model_forward_time": 1.7628731727600098,
      "step": 95655
    },
    {
      "epoch": 2.0651817321777345e-05,
      "step": 95655,
      "training_step_time": 7.701813220977783
    },
    {
      "epoch": 2.065277099609375e-05,
      "model_forward_time": 1.7646400928497314,
      "step": 95656
    },
    {
      "epoch": 2.065277099609375e-05,
      "step": 95656,
      "training_step_time": 7.881525993347168
    },
    {
      "epoch": 2.0653724670410156e-05,
      "model_forward_time": 1.7827317714691162,
      "step": 95657
    },
    {
      "epoch": 2.0653724670410156e-05,
      "step": 95657,
      "training_step_time": 9.13984727859497
    },
    {
      "epoch": 2.0654678344726564e-05,
      "model_forward_time": 1.7658450603485107,
      "step": 95658
    },
    {
      "epoch": 2.0654678344726564e-05,
      "step": 95658,
      "training_step_time": 8.488863229751587
    },
    {
      "epoch": 2.0655632019042968e-05,
      "model_forward_time": 1.7706215381622314,
      "step": 95659
    },
    {
      "epoch": 2.0655632019042968e-05,
      "step": 95659,
      "training_step_time": 8.464261770248413
    },
    {
      "epoch": 2.0656585693359375e-05,
      "grad_norm": 0.07161279767751694,
      "learning_rate": 5.143114168871699e-08,
      "loss": 0.0548,
      "step": 95660
    },
    {
      "epoch": 2.0656585693359375e-05,
      "model_forward_time": 2.368983030319214,
      "step": 95660
    },
    {
      "dynamics_loss_avg": 0.054553838819265364,
      "epoch": 2.0656585693359375e-05,
      "step": 95660
    },
    {
      "action_loss_avg": 0.0016308281279634683,
      "epoch": 2.0656585693359375e-05,
      "step": 95660
    },
    {
      "epoch": 2.0656585693359375e-05,
      "step": 95660,
      "training_step_time": 9.2588369846344
    },
    {
      "epoch": 2.0657539367675783e-05,
      "model_forward_time": 1.7666962146759033,
      "step": 95661
    },
    {
      "epoch": 2.0657539367675783e-05,
      "step": 95661,
      "training_step_time": 7.718522787094116
    },
    {
      "epoch": 2.0658493041992187e-05,
      "model_forward_time": 1.7465465068817139,
      "step": 95662
    },
    {
      "epoch": 2.0658493041992187e-05,
      "step": 95662,
      "training_step_time": 7.688307762145996
    },
    {
      "epoch": 2.0659446716308594e-05,
      "model_forward_time": 1.7575526237487793,
      "step": 95663
    },
    {
      "epoch": 2.0659446716308594e-05,
      "step": 95663,
      "training_step_time": 8.233883619308472
    },
    {
      "epoch": 2.0660400390625e-05,
      "model_forward_time": 1.747589349746704,
      "step": 95664
    },
    {
      "epoch": 2.0660400390625e-05,
      "step": 95664,
      "training_step_time": 7.601650714874268
    },
    {
      "epoch": 2.0661354064941405e-05,
      "model_forward_time": 1.7561829090118408,
      "step": 95665
    },
    {
      "epoch": 2.0661354064941405e-05,
      "step": 95665,
      "training_step_time": 8.534366846084595
    },
    {
      "epoch": 2.0662307739257813e-05,
      "model_forward_time": 1.7546067237854004,
      "step": 95666
    },
    {
      "epoch": 2.0662307739257813e-05,
      "step": 95666,
      "training_step_time": 7.600776433944702
    },
    {
      "epoch": 2.066326141357422e-05,
      "model_forward_time": 1.7456588745117188,
      "step": 95667
    },
    {
      "epoch": 2.066326141357422e-05,
      "step": 95667,
      "training_step_time": 7.701223611831665
    },
    {
      "epoch": 2.0664215087890624e-05,
      "model_forward_time": 1.7495207786560059,
      "step": 95668
    },
    {
      "epoch": 2.0664215087890624e-05,
      "step": 95668,
      "training_step_time": 9.210419654846191
    },
    {
      "epoch": 2.0665168762207032e-05,
      "model_forward_time": 1.785494327545166,
      "step": 95669
    },
    {
      "epoch": 2.0665168762207032e-05,
      "step": 95669,
      "training_step_time": 9.70211672782898
    },
    {
      "epoch": 2.066612243652344e-05,
      "grad_norm": 0.08159718662500381,
      "learning_rate": 5.119486407302532e-08,
      "loss": 0.0548,
      "step": 95670
    },
    {
      "epoch": 2.066612243652344e-05,
      "model_forward_time": 2.41776704788208,
      "step": 95670
    },
    {
      "dynamics_loss_avg": 0.051552937179803845,
      "epoch": 2.066612243652344e-05,
      "step": 95670
    },
    {
      "action_loss_avg": 0.0013181440532207488,
      "epoch": 2.066612243652344e-05,
      "step": 95670
    },
    {
      "epoch": 2.066612243652344e-05,
      "step": 95670,
      "training_step_time": 7.808294773101807
    },
    {
      "epoch": 2.0667076110839843e-05,
      "model_forward_time": 1.753051519393921,
      "step": 95671
    },
    {
      "epoch": 2.0667076110839843e-05,
      "step": 95671,
      "training_step_time": 7.738062143325806
    },
    {
      "epoch": 2.066802978515625e-05,
      "model_forward_time": 1.7755370140075684,
      "step": 95672
    },
    {
      "epoch": 2.066802978515625e-05,
      "step": 95672,
      "training_step_time": 7.539428472518921
    },
    {
      "epoch": 2.0668983459472655e-05,
      "model_forward_time": 1.7576494216918945,
      "step": 95673
    },
    {
      "epoch": 2.0668983459472655e-05,
      "step": 95673,
      "training_step_time": 7.585228681564331
    },
    {
      "epoch": 2.0669937133789062e-05,
      "model_forward_time": 1.760688066482544,
      "step": 95674
    },
    {
      "epoch": 2.0669937133789062e-05,
      "step": 95674,
      "training_step_time": 8.602537155151367
    },
    {
      "epoch": 2.067089080810547e-05,
      "model_forward_time": 1.7530419826507568,
      "step": 95675
    },
    {
      "epoch": 2.067089080810547e-05,
      "step": 95675,
      "training_step_time": 7.7674901485443115
    },
    {
      "epoch": 2.0671844482421873e-05,
      "model_forward_time": 1.7408781051635742,
      "step": 95676
    },
    {
      "epoch": 2.0671844482421873e-05,
      "step": 95676,
      "training_step_time": 8.761388301849365
    },
    {
      "epoch": 2.067279815673828e-05,
      "model_forward_time": 1.7397124767303467,
      "step": 95677
    },
    {
      "epoch": 2.067279815673828e-05,
      "step": 95677,
      "training_step_time": 9.812114715576172
    },
    {
      "epoch": 2.0673751831054688e-05,
      "model_forward_time": 1.7689428329467773,
      "step": 95678
    },
    {
      "epoch": 2.0673751831054688e-05,
      "step": 95678,
      "training_step_time": 8.727084159851074
    },
    {
      "epoch": 2.0674705505371092e-05,
      "model_forward_time": 2.4394242763519287,
      "step": 95679
    },
    {
      "epoch": 2.0674705505371092e-05,
      "step": 95679,
      "training_step_time": 7.830265522003174
    },
    {
      "epoch": 2.06756591796875e-05,
      "grad_norm": 0.07984470576047897,
      "learning_rate": 5.095912765122335e-08,
      "loss": 0.054,
      "step": 95680
    },
    {
      "epoch": 2.06756591796875e-05,
      "model_forward_time": 1.739549160003662,
      "step": 95680
    },
    {
      "dynamics_loss_avg": 0.06330211777240038,
      "epoch": 2.06756591796875e-05,
      "step": 95680
    },
    {
      "action_loss_avg": 0.0012133385811466725,
      "epoch": 2.06756591796875e-05,
      "step": 95680
    },
    {
      "epoch": 2.06756591796875e-05,
      "step": 95680,
      "training_step_time": 9.124343633651733
    },
    {
      "epoch": 2.0676612854003907e-05,
      "model_forward_time": 1.7550606727600098,
      "step": 95681
    },
    {
      "epoch": 2.0676612854003907e-05,
      "step": 95681,
      "training_step_time": 8.155447483062744
    },
    {
      "epoch": 2.067756652832031e-05,
      "model_forward_time": 1.748962640762329,
      "step": 95682
    },
    {
      "epoch": 2.067756652832031e-05,
      "step": 95682,
      "training_step_time": 7.625974893569946
    },
    {
      "epoch": 2.067852020263672e-05,
      "model_forward_time": 1.7502896785736084,
      "step": 95683
    },
    {
      "epoch": 2.067852020263672e-05,
      "step": 95683,
      "training_step_time": 8.342975854873657
    },
    {
      "epoch": 2.0679473876953126e-05,
      "model_forward_time": 1.7495262622833252,
      "step": 95684
    },
    {
      "epoch": 2.0679473876953126e-05,
      "step": 95684,
      "training_step_time": 7.6250901222229
    },
    {
      "epoch": 2.068042755126953e-05,
      "model_forward_time": 1.7471537590026855,
      "step": 95685
    },
    {
      "epoch": 2.068042755126953e-05,
      "step": 95685,
      "training_step_time": 9.724477291107178
    },
    {
      "epoch": 2.0681381225585937e-05,
      "model_forward_time": 1.741757869720459,
      "step": 95686
    },
    {
      "epoch": 2.0681381225585937e-05,
      "step": 95686,
      "training_step_time": 9.628166913986206
    },
    {
      "epoch": 2.0682334899902345e-05,
      "model_forward_time": 1.7667317390441895,
      "step": 95687
    },
    {
      "epoch": 2.0682334899902345e-05,
      "step": 95687,
      "training_step_time": 9.784394025802612
    },
    {
      "epoch": 2.068328857421875e-05,
      "model_forward_time": 2.461301565170288,
      "step": 95688
    },
    {
      "epoch": 2.068328857421875e-05,
      "step": 95688,
      "training_step_time": 9.256767749786377
    },
    {
      "epoch": 2.0684242248535156e-05,
      "model_forward_time": 1.7537517547607422,
      "step": 95689
    },
    {
      "epoch": 2.0684242248535156e-05,
      "step": 95689,
      "training_step_time": 8.731381177902222
    },
    {
      "epoch": 2.0685195922851564e-05,
      "grad_norm": 0.09833322465419769,
      "learning_rate": 5.072393244908991e-08,
      "loss": 0.056,
      "step": 95690
    },
    {
      "epoch": 2.0685195922851564e-05,
      "model_forward_time": 1.752969741821289,
      "step": 95690
    },
    {
      "dynamics_loss_avg": 0.06916518397629261,
      "epoch": 2.0685195922851564e-05,
      "step": 95690
    },
    {
      "action_loss_avg": 0.0016754319600295275,
      "epoch": 2.0685195922851564e-05,
      "step": 95690
    },
    {
      "epoch": 2.0685195922851564e-05,
      "step": 95690,
      "training_step_time": 7.64213752746582
    },
    {
      "epoch": 2.0686149597167968e-05,
      "model_forward_time": 1.7516157627105713,
      "step": 95691
    },
    {
      "epoch": 2.0686149597167968e-05,
      "step": 95691,
      "training_step_time": 7.503455638885498
    },
    {
      "epoch": 2.0687103271484375e-05,
      "model_forward_time": 1.757538080215454,
      "step": 95692
    },
    {
      "epoch": 2.0687103271484375e-05,
      "step": 95692,
      "training_step_time": 7.750811338424683
    },
    {
      "epoch": 2.0688056945800783e-05,
      "model_forward_time": 1.7622215747833252,
      "step": 95693
    },
    {
      "epoch": 2.0688056945800783e-05,
      "step": 95693,
      "training_step_time": 7.588992595672607
    },
    {
      "epoch": 2.0689010620117187e-05,
      "model_forward_time": 1.745415210723877,
      "step": 95694
    },
    {
      "epoch": 2.0689010620117187e-05,
      "step": 95694,
      "training_step_time": 8.741163730621338
    },
    {
      "epoch": 2.0689964294433594e-05,
      "model_forward_time": 1.7513635158538818,
      "step": 95695
    },
    {
      "epoch": 2.0689964294433594e-05,
      "step": 95695,
      "training_step_time": 7.832976579666138
    },
    {
      "epoch": 2.069091796875e-05,
      "model_forward_time": 1.7673859596252441,
      "step": 95696
    },
    {
      "epoch": 2.069091796875e-05,
      "step": 95696,
      "training_step_time": 9.896368980407715
    },
    {
      "epoch": 2.0691871643066405e-05,
      "model_forward_time": 2.442736864089966,
      "step": 95697
    },
    {
      "epoch": 2.0691871643066405e-05,
      "step": 95697,
      "training_step_time": 9.355800867080688
    },
    {
      "epoch": 2.0692825317382813e-05,
      "model_forward_time": 1.7511377334594727,
      "step": 95698
    },
    {
      "epoch": 2.0692825317382813e-05,
      "step": 95698,
      "training_step_time": 7.730520963668823
    },
    {
      "epoch": 2.069377899169922e-05,
      "model_forward_time": 1.7482678890228271,
      "step": 95699
    },
    {
      "epoch": 2.069377899169922e-05,
      "step": 95699,
      "training_step_time": 10.635534048080444
    },
    {
      "epoch": 2.0694732666015624e-05,
      "grad_norm": 0.07787953317165375,
      "learning_rate": 5.0489278492346085e-08,
      "loss": 0.0548,
      "step": 95700
    },
    {
      "epoch": 2.0694732666015624e-05,
      "model_forward_time": 1.762296438217163,
      "step": 95700
    },
    {
      "dynamics_loss_avg": 0.05226447824388743,
      "epoch": 2.0694732666015624e-05,
      "step": 95700
    },
    {
      "action_loss_avg": 0.0010893298051087186,
      "epoch": 2.0694732666015624e-05,
      "step": 95700
    },
    {
      "epoch": 2.0694732666015624e-05,
      "step": 95700,
      "training_step_time": 7.508570194244385
    },
    {
      "epoch": 2.0695686340332032e-05,
      "model_forward_time": 1.7453949451446533,
      "step": 95701
    },
    {
      "epoch": 2.0695686340332032e-05,
      "step": 95701,
      "training_step_time": 7.633354187011719
    },
    {
      "epoch": 2.069664001464844e-05,
      "model_forward_time": 1.7563846111297607,
      "step": 95702
    },
    {
      "epoch": 2.069664001464844e-05,
      "step": 95702,
      "training_step_time": 7.586904287338257
    },
    {
      "epoch": 2.0697593688964843e-05,
      "model_forward_time": 1.7444522380828857,
      "step": 95703
    },
    {
      "epoch": 2.0697593688964843e-05,
      "step": 95703,
      "training_step_time": 8.770516157150269
    },
    {
      "epoch": 2.069854736328125e-05,
      "model_forward_time": 1.744633674621582,
      "step": 95704
    },
    {
      "epoch": 2.069854736328125e-05,
      "step": 95704,
      "training_step_time": 9.202372312545776
    },
    {
      "epoch": 2.0699501037597655e-05,
      "model_forward_time": 1.738619327545166,
      "step": 95705
    },
    {
      "epoch": 2.0699501037597655e-05,
      "step": 95705,
      "training_step_time": 8.851667404174805
    },
    {
      "epoch": 2.0700454711914062e-05,
      "model_forward_time": 2.3773529529571533,
      "step": 95706
    },
    {
      "epoch": 2.0700454711914062e-05,
      "step": 95706,
      "training_step_time": 7.715819597244263
    },
    {
      "epoch": 2.070140838623047e-05,
      "model_forward_time": 1.7568843364715576,
      "step": 95707
    },
    {
      "epoch": 2.070140838623047e-05,
      "step": 95707,
      "training_step_time": 9.547766208648682
    },
    {
      "epoch": 2.0702362060546873e-05,
      "model_forward_time": 1.7469260692596436,
      "step": 95708
    },
    {
      "epoch": 2.0702362060546873e-05,
      "step": 95708,
      "training_step_time": 7.635214567184448
    },
    {
      "epoch": 2.070331573486328e-05,
      "model_forward_time": 1.7693076133728027,
      "step": 95709
    },
    {
      "epoch": 2.070331573486328e-05,
      "step": 95709,
      "training_step_time": 7.471177339553833
    },
    {
      "epoch": 2.070426940917969e-05,
      "grad_norm": 0.08105919510126114,
      "learning_rate": 5.0255165806653575e-08,
      "loss": 0.0559,
      "step": 95710
    },
    {
      "epoch": 2.070426940917969e-05,
      "model_forward_time": 1.7454628944396973,
      "step": 95710
    },
    {
      "dynamics_loss_avg": 0.055975332856178284,
      "epoch": 2.070426940917969e-05,
      "step": 95710
    },
    {
      "action_loss_avg": 0.0010289535333868117,
      "epoch": 2.070426940917969e-05,
      "step": 95710
    },
    {
      "epoch": 2.070426940917969e-05,
      "step": 95710,
      "training_step_time": 7.658373832702637
    },
    {
      "epoch": 2.0705223083496092e-05,
      "model_forward_time": 1.7563138008117676,
      "step": 95711
    },
    {
      "epoch": 2.0705223083496092e-05,
      "step": 95711,
      "training_step_time": 7.576169013977051
    },
    {
      "epoch": 2.07061767578125e-05,
      "model_forward_time": 1.7500278949737549,
      "step": 95712
    },
    {
      "epoch": 2.07061767578125e-05,
      "step": 95712,
      "training_step_time": 7.730823040008545
    },
    {
      "epoch": 2.0707130432128907e-05,
      "model_forward_time": 1.7561626434326172,
      "step": 95713
    },
    {
      "epoch": 2.0707130432128907e-05,
      "step": 95713,
      "training_step_time": 8.238848447799683
    },
    {
      "epoch": 2.070808410644531e-05,
      "model_forward_time": 1.7473840713500977,
      "step": 95714
    },
    {
      "epoch": 2.070808410644531e-05,
      "step": 95714,
      "training_step_time": 8.684544086456299
    },
    {
      "epoch": 2.070903778076172e-05,
      "model_forward_time": 1.775695562362671,
      "step": 95715
    },
    {
      "epoch": 2.070903778076172e-05,
      "step": 95715,
      "training_step_time": 7.842115163803101
    },
    {
      "epoch": 2.0709991455078126e-05,
      "model_forward_time": 2.4364075660705566,
      "step": 95716
    },
    {
      "epoch": 2.0709991455078126e-05,
      "step": 95716,
      "training_step_time": 7.688016653060913
    },
    {
      "epoch": 2.071094512939453e-05,
      "model_forward_time": 1.7453548908233643,
      "step": 95717
    },
    {
      "epoch": 2.071094512939453e-05,
      "step": 95717,
      "training_step_time": 7.705022573471069
    },
    {
      "epoch": 2.0711898803710937e-05,
      "model_forward_time": 1.7691335678100586,
      "step": 95718
    },
    {
      "epoch": 2.0711898803710937e-05,
      "step": 95718,
      "training_step_time": 7.645817518234253
    },
    {
      "epoch": 2.0712852478027345e-05,
      "model_forward_time": 1.7481598854064941,
      "step": 95719
    },
    {
      "epoch": 2.0712852478027345e-05,
      "step": 95719,
      "training_step_time": 7.663789749145508
    },
    {
      "epoch": 2.071380615234375e-05,
      "grad_norm": 0.09451128542423248,
      "learning_rate": 5.0021594417614116e-08,
      "loss": 0.0544,
      "step": 95720
    },
    {
      "epoch": 2.071380615234375e-05,
      "model_forward_time": 1.7457501888275146,
      "step": 95720
    },
    {
      "dynamics_loss_avg": 0.04677852988243103,
      "epoch": 2.071380615234375e-05,
      "step": 95720
    },
    {
      "action_loss_avg": 0.0010754561109934003,
      "epoch": 2.071380615234375e-05,
      "step": 95720
    },
    {
      "epoch": 2.071380615234375e-05,
      "step": 95720,
      "training_step_time": 8.676060438156128
    },
    {
      "epoch": 2.0714759826660156e-05,
      "model_forward_time": 1.7503907680511475,
      "step": 95721
    },
    {
      "epoch": 2.0714759826660156e-05,
      "step": 95721,
      "training_step_time": 7.78814435005188
    },
    {
      "epoch": 2.0715713500976564e-05,
      "model_forward_time": 1.7561233043670654,
      "step": 95722
    },
    {
      "epoch": 2.0715713500976564e-05,
      "step": 95722,
      "training_step_time": 7.811471223831177
    },
    {
      "epoch": 2.0716667175292968e-05,
      "model_forward_time": 1.746281385421753,
      "step": 95723
    },
    {
      "epoch": 2.0716667175292968e-05,
      "step": 95723,
      "training_step_time": 8.504705667495728
    },
    {
      "epoch": 2.0717620849609375e-05,
      "model_forward_time": 1.754990816116333,
      "step": 95724
    },
    {
      "epoch": 2.0717620849609375e-05,
      "step": 95724,
      "training_step_time": 11.561511993408203
    },
    {
      "epoch": 2.0718574523925783e-05,
      "model_forward_time": 2.403991460800171,
      "step": 95725
    },
    {
      "epoch": 2.0718574523925783e-05,
      "step": 95725,
      "training_step_time": 8.389323472976685
    },
    {
      "epoch": 2.0719528198242187e-05,
      "model_forward_time": 1.7314260005950928,
      "step": 95726
    },
    {
      "epoch": 2.0719528198242187e-05,
      "step": 95726,
      "training_step_time": 7.9326331615448
    },
    {
      "epoch": 2.0720481872558594e-05,
      "model_forward_time": 1.7688109874725342,
      "step": 95727
    },
    {
      "epoch": 2.0720481872558594e-05,
      "step": 95727,
      "training_step_time": 9.589172601699829
    },
    {
      "epoch": 2.0721435546875e-05,
      "model_forward_time": 1.7433326244354248,
      "step": 95728
    },
    {
      "epoch": 2.0721435546875e-05,
      "step": 95728,
      "training_step_time": 7.613801717758179
    },
    {
      "epoch": 2.0722389221191405e-05,
      "model_forward_time": 1.760474681854248,
      "step": 95729
    },
    {
      "epoch": 2.0722389221191405e-05,
      "step": 95729,
      "training_step_time": 7.587921380996704
    },
    {
      "epoch": 2.0723342895507813e-05,
      "grad_norm": 0.06894990056753159,
      "learning_rate": 4.978856435077062e-08,
      "loss": 0.0544,
      "step": 95730
    },
    {
      "epoch": 2.0723342895507813e-05,
      "model_forward_time": 1.7462780475616455,
      "step": 95730
    },
    {
      "dynamics_loss_avg": 0.05010595126077533,
      "epoch": 2.0723342895507813e-05,
      "step": 95730
    },
    {
      "action_loss_avg": 0.0017305679444689303,
      "epoch": 2.0723342895507813e-05,
      "step": 95730
    },
    {
      "epoch": 2.0723342895507813e-05,
      "step": 95730,
      "training_step_time": 8.228997945785522
    },
    {
      "epoch": 2.072429656982422e-05,
      "model_forward_time": 1.7378685474395752,
      "step": 95731
    },
    {
      "epoch": 2.072429656982422e-05,
      "step": 95731,
      "training_step_time": 7.8258044719696045
    },
    {
      "epoch": 2.0725250244140624e-05,
      "model_forward_time": 1.7516002655029297,
      "step": 95732
    },
    {
      "epoch": 2.0725250244140624e-05,
      "step": 95732,
      "training_step_time": 9.336868524551392
    },
    {
      "epoch": 2.0726203918457032e-05,
      "model_forward_time": 1.7575347423553467,
      "step": 95733
    },
    {
      "epoch": 2.0726203918457032e-05,
      "step": 95733,
      "training_step_time": 7.958487272262573
    },
    {
      "epoch": 2.072715759277344e-05,
      "model_forward_time": 2.409886360168457,
      "step": 95734
    },
    {
      "epoch": 2.072715759277344e-05,
      "step": 95734,
      "training_step_time": 7.79318380355835
    },
    {
      "epoch": 2.0728111267089843e-05,
      "model_forward_time": 1.7315723896026611,
      "step": 95735
    },
    {
      "epoch": 2.0728111267089843e-05,
      "step": 95735,
      "training_step_time": 7.816774606704712
    },
    {
      "epoch": 2.072906494140625e-05,
      "model_forward_time": 1.7440407276153564,
      "step": 95736
    },
    {
      "epoch": 2.072906494140625e-05,
      "step": 95736,
      "training_step_time": 9.42660665512085
    },
    {
      "epoch": 2.0730018615722655e-05,
      "model_forward_time": 1.746476411819458,
      "step": 95737
    },
    {
      "epoch": 2.0730018615722655e-05,
      "step": 95737,
      "training_step_time": 7.725067853927612
    },
    {
      "epoch": 2.0730972290039062e-05,
      "model_forward_time": 1.7443287372589111,
      "step": 95738
    },
    {
      "epoch": 2.0730972290039062e-05,
      "step": 95738,
      "training_step_time": 8.340511083602905
    },
    {
      "epoch": 2.073192596435547e-05,
      "model_forward_time": 1.7475848197937012,
      "step": 95739
    },
    {
      "epoch": 2.073192596435547e-05,
      "step": 95739,
      "training_step_time": 7.64755916595459
    },
    {
      "epoch": 2.0732879638671874e-05,
      "grad_norm": 0.06883737444877625,
      "learning_rate": 4.955607563160769e-08,
      "loss": 0.0564,
      "step": 95740
    },
    {
      "epoch": 2.0732879638671874e-05,
      "model_forward_time": 1.7517297267913818,
      "step": 95740
    },
    {
      "dynamics_loss_avg": 0.05140008926391602,
      "epoch": 2.0732879638671874e-05,
      "step": 95740
    },
    {
      "action_loss_avg": 0.0008613183279521763,
      "epoch": 2.0732879638671874e-05,
      "step": 95740
    },
    {
      "epoch": 2.0732879638671874e-05,
      "step": 95740,
      "training_step_time": 7.6559975147247314
    },
    {
      "epoch": 2.073383331298828e-05,
      "model_forward_time": 1.7474255561828613,
      "step": 95741
    },
    {
      "epoch": 2.073383331298828e-05,
      "step": 95741,
      "training_step_time": 7.692265748977661
    },
    {
      "epoch": 2.073478698730469e-05,
      "model_forward_time": 1.7475523948669434,
      "step": 95742
    },
    {
      "epoch": 2.073478698730469e-05,
      "step": 95742,
      "training_step_time": 8.610252618789673
    },
    {
      "epoch": 2.0735740661621092e-05,
      "model_forward_time": 2.4404947757720947,
      "step": 95743
    },
    {
      "epoch": 2.0735740661621092e-05,
      "step": 95743,
      "training_step_time": 8.824175119400024
    },
    {
      "epoch": 2.07366943359375e-05,
      "model_forward_time": 1.7518844604492188,
      "step": 95744
    },
    {
      "epoch": 2.07366943359375e-05,
      "step": 95744,
      "training_step_time": 11.543373584747314
    },
    {
      "epoch": 2.0737648010253907e-05,
      "model_forward_time": 1.7474448680877686,
      "step": 95745
    },
    {
      "epoch": 2.0737648010253907e-05,
      "step": 95745,
      "training_step_time": 8.437836647033691
    },
    {
      "epoch": 2.073860168457031e-05,
      "model_forward_time": 1.74111008644104,
      "step": 95746
    },
    {
      "epoch": 2.073860168457031e-05,
      "step": 95746,
      "training_step_time": 8.3059823513031
    },
    {
      "epoch": 2.073955535888672e-05,
      "model_forward_time": 1.750786542892456,
      "step": 95747
    },
    {
      "epoch": 2.073955535888672e-05,
      "step": 95747,
      "training_step_time": 7.741266965866089
    },
    {
      "epoch": 2.0740509033203126e-05,
      "model_forward_time": 1.7448816299438477,
      "step": 95748
    },
    {
      "epoch": 2.0740509033203126e-05,
      "step": 95748,
      "training_step_time": 6.853246450424194
    },
    {
      "epoch": 2.074146270751953e-05,
      "model_forward_time": 1.3630092144012451,
      "step": 95749
    },
    {
      "epoch": 2.074146270751953e-05,
      "step": 95749,
      "training_step_time": 6.834641695022583
    },
    {
      "epoch": 2.0742416381835938e-05,
      "grad_norm": 0.09243210405111313,
      "learning_rate": 4.9324128285549444e-08,
      "loss": 0.0558,
      "step": 95750
    },
    {
      "epoch": 2.0742416381835938e-05,
      "model_forward_time": 1.7353339195251465,
      "step": 95750
    },
    {
      "dynamics_loss_avg": 0.04991304222494364,
      "epoch": 2.0742416381835938e-05,
      "step": 95750
    },
    {
      "action_loss_avg": 0.0008839107933454216,
      "epoch": 2.0742416381835938e-05,
      "step": 95750
    },
    {
      "epoch": 2.0742416381835938e-05,
      "step": 95750,
      "training_step_time": 6.860342025756836
    },
    {
      "epoch": 2.0743370056152345e-05,
      "model_forward_time": 1.7508409023284912,
      "step": 95751
    },
    {
      "epoch": 2.0743370056152345e-05,
      "step": 95751,
      "training_step_time": 6.863147974014282
    },
    {
      "epoch": 2.074432373046875e-05,
      "model_forward_time": 1.0369658470153809,
      "step": 95752
    },
    {
      "epoch": 2.074432373046875e-05,
      "step": 95752,
      "training_step_time": 6.833717107772827
    },
    {
      "epoch": 2.0745277404785156e-05,
      "model_forward_time": 1.7498564720153809,
      "step": 95753
    },
    {
      "epoch": 2.0745277404785156e-05,
      "step": 95753,
      "training_step_time": 6.86266565322876
    },
    {
      "epoch": 2.0746231079101564e-05,
      "model_forward_time": 2.3882405757904053,
      "step": 95754
    },
    {
      "epoch": 2.0746231079101564e-05,
      "step": 95754,
      "training_step_time": 7.812760829925537
    },
    {
      "epoch": 2.0747184753417968e-05,
      "model_forward_time": 1.7441930770874023,
      "step": 95755
    },
    {
      "epoch": 2.0747184753417968e-05,
      "step": 95755,
      "training_step_time": 8.377922296524048
    },
    {
      "epoch": 2.0748138427734375e-05,
      "model_forward_time": 1.7514450550079346,
      "step": 95756
    },
    {
      "epoch": 2.0748138427734375e-05,
      "step": 95756,
      "training_step_time": 8.154027462005615
    },
    {
      "epoch": 2.0749092102050783e-05,
      "model_forward_time": 1.7513434886932373,
      "step": 95757
    },
    {
      "epoch": 2.0749092102050783e-05,
      "step": 95757,
      "training_step_time": 7.280904531478882
    },
    {
      "epoch": 2.0750045776367187e-05,
      "model_forward_time": 1.7523128986358643,
      "step": 95758
    },
    {
      "epoch": 2.0750045776367187e-05,
      "step": 95758,
      "training_step_time": 6.870875358581543
    },
    {
      "epoch": 2.0750999450683594e-05,
      "model_forward_time": 1.7457644939422607,
      "step": 95759
    },
    {
      "epoch": 2.0750999450683594e-05,
      "step": 95759,
      "training_step_time": 6.842896223068237
    },
    {
      "epoch": 2.0751953125e-05,
      "grad_norm": 0.12830908596515656,
      "learning_rate": 4.909272233796114e-08,
      "loss": 0.0561,
      "step": 95760
    },
    {
      "epoch": 2.0751953125e-05,
      "model_forward_time": 1.743595838546753,
      "step": 95760
    },
    {
      "dynamics_loss_avg": 0.05592262875288725,
      "epoch": 2.0751953125e-05,
      "step": 95760
    },
    {
      "action_loss_avg": 0.0012360419961623848,
      "epoch": 2.0751953125e-05,
      "step": 95760
    },
    {
      "epoch": 2.0751953125e-05,
      "step": 95760,
      "training_step_time": 6.846881628036499
    },
    {
      "epoch": 2.0752906799316406e-05,
      "model_forward_time": 1.3716866970062256,
      "step": 95761
    },
    {
      "epoch": 2.0752906799316406e-05,
      "step": 95761,
      "training_step_time": 6.838767766952515
    },
    {
      "epoch": 2.0753860473632813e-05,
      "model_forward_time": 1.7572624683380127,
      "step": 95762
    },
    {
      "epoch": 2.0753860473632813e-05,
      "step": 95762,
      "training_step_time": 6.822944402694702
    },
    {
      "epoch": 2.075481414794922e-05,
      "model_forward_time": 1.762892484664917,
      "step": 95763
    },
    {
      "epoch": 2.075481414794922e-05,
      "step": 95763,
      "training_step_time": 7.815318822860718
    },
    {
      "epoch": 2.0755767822265624e-05,
      "model_forward_time": 2.3680121898651123,
      "step": 95764
    },
    {
      "epoch": 2.0755767822265624e-05,
      "step": 95764,
      "training_step_time": 7.5616068840026855
    },
    {
      "epoch": 2.0756721496582032e-05,
      "model_forward_time": 1.743614912033081,
      "step": 95765
    },
    {
      "epoch": 2.0756721496582032e-05,
      "step": 95765,
      "training_step_time": 7.5810182094573975
    },
    {
      "epoch": 2.0757675170898436e-05,
      "model_forward_time": 1.752777338027954,
      "step": 95766
    },
    {
      "epoch": 2.0757675170898436e-05,
      "step": 95766,
      "training_step_time": 7.272565126419067
    },
    {
      "epoch": 2.0758628845214843e-05,
      "model_forward_time": 1.7498939037322998,
      "step": 95767
    },
    {
      "epoch": 2.0758628845214843e-05,
      "step": 95767,
      "training_step_time": 6.846406698226929
    },
    {
      "epoch": 2.075958251953125e-05,
      "model_forward_time": 1.7437317371368408,
      "step": 95768
    },
    {
      "epoch": 2.075958251953125e-05,
      "step": 95768,
      "training_step_time": 6.86548113822937
    },
    {
      "epoch": 2.0760536193847655e-05,
      "model_forward_time": 1.7343406677246094,
      "step": 95769
    },
    {
      "epoch": 2.0760536193847655e-05,
      "step": 95769,
      "training_step_time": 6.843850374221802
    },
    {
      "epoch": 2.0761489868164062e-05,
      "grad_norm": 0.2215902954339981,
      "learning_rate": 4.8861857814149204e-08,
      "loss": 0.0564,
      "step": 95770
    },
    {
      "epoch": 2.0761489868164062e-05,
      "model_forward_time": 1.7436034679412842,
      "step": 95770
    },
    {
      "dynamics_loss_avg": 0.03952164202928543,
      "epoch": 2.0761489868164062e-05,
      "step": 95770
    },
    {
      "action_loss_avg": 0.0007745662631350569,
      "epoch": 2.0761489868164062e-05,
      "step": 95770
    },
    {
      "epoch": 2.0761489868164062e-05,
      "step": 95770,
      "training_step_time": 6.830370903015137
    },
    {
      "epoch": 2.076244354248047e-05,
      "model_forward_time": 1.74869704246521,
      "step": 95771
    },
    {
      "epoch": 2.076244354248047e-05,
      "step": 95771,
      "training_step_time": 6.889472484588623
    },
    {
      "epoch": 2.0763397216796874e-05,
      "model_forward_time": 1.76118803024292,
      "step": 95772
    },
    {
      "epoch": 2.0763397216796874e-05,
      "step": 95772,
      "training_step_time": 9.444075107574463
    },
    {
      "epoch": 2.076435089111328e-05,
      "model_forward_time": 2.367697238922119,
      "step": 95773
    },
    {
      "epoch": 2.076435089111328e-05,
      "step": 95773,
      "training_step_time": 7.672569036483765
    },
    {
      "epoch": 2.076530456542969e-05,
      "model_forward_time": 1.7483792304992676,
      "step": 95774
    },
    {
      "epoch": 2.076530456542969e-05,
      "step": 95774,
      "training_step_time": 7.5561254024505615
    },
    {
      "epoch": 2.0766258239746092e-05,
      "model_forward_time": 1.3730566501617432,
      "step": 95775
    },
    {
      "epoch": 2.0766258239746092e-05,
      "step": 95775,
      "training_step_time": 7.262877702713013
    },
    {
      "epoch": 2.07672119140625e-05,
      "model_forward_time": 1.7511889934539795,
      "step": 95776
    },
    {
      "epoch": 2.07672119140625e-05,
      "step": 95776,
      "training_step_time": 6.834680795669556
    },
    {
      "epoch": 2.0768165588378907e-05,
      "model_forward_time": 1.7450966835021973,
      "step": 95777
    },
    {
      "epoch": 2.0768165588378907e-05,
      "step": 95777,
      "training_step_time": 6.937007904052734
    },
    {
      "epoch": 2.076911926269531e-05,
      "model_forward_time": 1.7479193210601807,
      "step": 95778
    },
    {
      "epoch": 2.076911926269531e-05,
      "step": 95778,
      "training_step_time": 6.8400702476501465
    },
    {
      "epoch": 2.077007293701172e-05,
      "model_forward_time": 1.3731787204742432,
      "step": 95779
    },
    {
      "epoch": 2.077007293701172e-05,
      "step": 95779,
      "training_step_time": 6.822430372238159
    },
    {
      "epoch": 2.0771026611328126e-05,
      "grad_norm": 0.45971915125846863,
      "learning_rate": 4.863153473936011e-08,
      "loss": 0.0571,
      "step": 95780
    },
    {
      "epoch": 2.0771026611328126e-05,
      "model_forward_time": 1.7395598888397217,
      "step": 95780
    },
    {
      "dynamics_loss_avg": 0.061802247725427154,
      "epoch": 2.0771026611328126e-05,
      "step": 95780
    },
    {
      "action_loss_avg": 0.0017462534393416717,
      "epoch": 2.0771026611328126e-05,
      "step": 95780
    },
    {
      "epoch": 2.0771026611328126e-05,
      "step": 95780,
      "training_step_time": 6.815855026245117
    },
    {
      "epoch": 2.077198028564453e-05,
      "model_forward_time": 1.7553980350494385,
      "step": 95781
    },
    {
      "epoch": 2.077198028564453e-05,
      "step": 95781,
      "training_step_time": 9.474879026412964
    },
    {
      "epoch": 2.0772933959960938e-05,
      "model_forward_time": 2.350081205368042,
      "step": 95782
    },
    {
      "epoch": 2.0772933959960938e-05,
      "step": 95782,
      "training_step_time": 7.669900178909302
    },
    {
      "epoch": 2.0773887634277345e-05,
      "model_forward_time": 1.7486317157745361,
      "step": 95783
    },
    {
      "epoch": 2.0773887634277345e-05,
      "step": 95783,
      "training_step_time": 7.5620691776275635
    },
    {
      "epoch": 2.077484130859375e-05,
      "model_forward_time": 1.7532408237457275,
      "step": 95784
    },
    {
      "epoch": 2.077484130859375e-05,
      "step": 95784,
      "training_step_time": 7.234718084335327
    },
    {
      "epoch": 2.0775794982910156e-05,
      "model_forward_time": 1.7577061653137207,
      "step": 95785
    },
    {
      "epoch": 2.0775794982910156e-05,
      "step": 95785,
      "training_step_time": 6.827560663223267
    },
    {
      "epoch": 2.0776748657226564e-05,
      "model_forward_time": 1.7476541996002197,
      "step": 95786
    },
    {
      "epoch": 2.0776748657226564e-05,
      "step": 95786,
      "training_step_time": 6.819114446640015
    },
    {
      "epoch": 2.0777702331542968e-05,
      "model_forward_time": 1.741575002670288,
      "step": 95787
    },
    {
      "epoch": 2.0777702331542968e-05,
      "step": 95787,
      "training_step_time": 6.824072360992432
    },
    {
      "epoch": 2.0778656005859375e-05,
      "model_forward_time": 1.7293448448181152,
      "step": 95788
    },
    {
      "epoch": 2.0778656005859375e-05,
      "step": 95788,
      "training_step_time": 6.8219358921051025
    },
    {
      "epoch": 2.0779609680175783e-05,
      "model_forward_time": 1.7248382568359375,
      "step": 95789
    },
    {
      "epoch": 2.0779609680175783e-05,
      "step": 95789,
      "training_step_time": 6.801525115966797
    },
    {
      "epoch": 2.0780563354492187e-05,
      "grad_norm": 0.08488360792398453,
      "learning_rate": 4.840175313878259e-08,
      "loss": 0.0544,
      "step": 95790
    },
    {
      "epoch": 2.0780563354492187e-05,
      "model_forward_time": 1.72603178024292,
      "step": 95790
    },
    {
      "dynamics_loss_avg": 0.05214936025440693,
      "epoch": 2.0780563354492187e-05,
      "step": 95790
    },
    {
      "action_loss_avg": 0.001129132800269872,
      "epoch": 2.0780563354492187e-05,
      "step": 95790
    },
    {
      "epoch": 2.0780563354492187e-05,
      "step": 95790,
      "training_step_time": 8.816209077835083
    },
    {
      "epoch": 2.0781517028808594e-05,
      "model_forward_time": 1.7313075065612793,
      "step": 95791
    },
    {
      "epoch": 2.0781517028808594e-05,
      "step": 95791,
      "training_step_time": 8.375391483306885
    },
    {
      "epoch": 2.0782470703125e-05,
      "model_forward_time": 1.7282178401947021,
      "step": 95792
    },
    {
      "epoch": 2.0782470703125e-05,
      "step": 95792,
      "training_step_time": 7.441698789596558
    },
    {
      "epoch": 2.0783424377441406e-05,
      "model_forward_time": 1.7296218872070312,
      "step": 95793
    },
    {
      "epoch": 2.0783424377441406e-05,
      "step": 95793,
      "training_step_time": 7.304133415222168
    },
    {
      "epoch": 2.0784378051757813e-05,
      "model_forward_time": 1.7283282279968262,
      "step": 95794
    },
    {
      "epoch": 2.0784378051757813e-05,
      "step": 95794,
      "training_step_time": 7.404172658920288
    },
    {
      "epoch": 2.078533172607422e-05,
      "model_forward_time": 1.7278554439544678,
      "step": 95795
    },
    {
      "epoch": 2.078533172607422e-05,
      "step": 95795,
      "training_step_time": 6.925396203994751
    },
    {
      "epoch": 2.0786285400390624e-05,
      "model_forward_time": 1.7208049297332764,
      "step": 95796
    },
    {
      "epoch": 2.0786285400390624e-05,
      "step": 95796,
      "training_step_time": 6.806793928146362
    },
    {
      "epoch": 2.0787239074707032e-05,
      "model_forward_time": 1.721247673034668,
      "step": 95797
    },
    {
      "epoch": 2.0787239074707032e-05,
      "step": 95797,
      "training_step_time": 6.868581295013428
    },
    {
      "epoch": 2.0788192749023436e-05,
      "model_forward_time": 1.7372455596923828,
      "step": 95798
    },
    {
      "epoch": 2.0788192749023436e-05,
      "step": 95798,
      "training_step_time": 6.8663671016693115
    },
    {
      "epoch": 2.0789146423339843e-05,
      "model_forward_time": 1.7335445880889893,
      "step": 95799
    },
    {
      "epoch": 2.0789146423339843e-05,
      "step": 95799,
      "training_step_time": 8.716192245483398
    },
    {
      "epoch": 2.079010009765625e-05,
      "grad_norm": 0.07496093213558197,
      "learning_rate": 4.8172513037544336e-08,
      "loss": 0.0577,
      "step": 95800
    },
    {
      "epoch": 2.079010009765625e-05,
      "model_forward_time": 1.7476584911346436,
      "step": 95800
    },
    {
      "dynamics_loss_avg": 0.06451995596289635,
      "epoch": 2.079010009765625e-05,
      "step": 95800
    },
    {
      "action_loss_avg": 0.0010942555702058599,
      "epoch": 2.079010009765625e-05,
      "step": 95800
    },
    {
      "epoch": 2.079010009765625e-05,
      "step": 95800,
      "training_step_time": 8.358696699142456
    },
    {
      "epoch": 2.0791053771972655e-05,
      "model_forward_time": 2.412828207015991,
      "step": 95801
    },
    {
      "epoch": 2.0791053771972655e-05,
      "step": 95801,
      "training_step_time": 7.681683301925659
    },
    {
      "epoch": 2.0792007446289062e-05,
      "model_forward_time": 1.7623977661132812,
      "step": 95802
    },
    {
      "epoch": 2.0792007446289062e-05,
      "step": 95802,
      "training_step_time": 7.710476636886597
    },
    {
      "epoch": 2.079296112060547e-05,
      "model_forward_time": 1.7950630187988281,
      "step": 95803
    },
    {
      "epoch": 2.079296112060547e-05,
      "step": 95803,
      "training_step_time": 7.372645616531372
    },
    {
      "epoch": 2.0793914794921874e-05,
      "model_forward_time": 1.4177708625793457,
      "step": 95804
    },
    {
      "epoch": 2.0793914794921874e-05,
      "step": 95804,
      "training_step_time": 6.909556150436401
    },
    {
      "epoch": 2.079486846923828e-05,
      "model_forward_time": 1.7852296829223633,
      "step": 95805
    },
    {
      "epoch": 2.079486846923828e-05,
      "step": 95805,
      "training_step_time": 6.863399267196655
    },
    {
      "epoch": 2.079582214355469e-05,
      "model_forward_time": 1.7729361057281494,
      "step": 95806
    },
    {
      "epoch": 2.079582214355469e-05,
      "step": 95806,
      "training_step_time": 6.854959726333618
    },
    {
      "epoch": 2.0796775817871092e-05,
      "model_forward_time": 1.773841381072998,
      "step": 95807
    },
    {
      "epoch": 2.0796775817871092e-05,
      "step": 95807,
      "training_step_time": 6.816728353500366
    },
    {
      "epoch": 2.07977294921875e-05,
      "model_forward_time": 1.7765741348266602,
      "step": 95808
    },
    {
      "epoch": 2.07977294921875e-05,
      "step": 95808,
      "training_step_time": 8.14981722831726
    },
    {
      "epoch": 2.0798683166503907e-05,
      "model_forward_time": 1.8404533863067627,
      "step": 95809
    },
    {
      "epoch": 2.0798683166503907e-05,
      "step": 95809,
      "training_step_time": 9.573298215866089
    },
    {
      "epoch": 2.079963684082031e-05,
      "grad_norm": 0.11826851963996887,
      "learning_rate": 4.7943814460715834e-08,
      "loss": 0.055,
      "step": 95810
    },
    {
      "epoch": 2.079963684082031e-05,
      "model_forward_time": 2.8512327671051025,
      "step": 95810
    },
    {
      "dynamics_loss_avg": 0.06365743950009346,
      "epoch": 2.079963684082031e-05,
      "step": 95810
    },
    {
      "action_loss_avg": 0.0015741386043373496,
      "epoch": 2.079963684082031e-05,
      "step": 95810
    },
    {
      "epoch": 2.079963684082031e-05,
      "step": 95810,
      "training_step_time": 8.794702053070068
    },
    {
      "epoch": 2.080059051513672e-05,
      "model_forward_time": 1.7813363075256348,
      "step": 95811
    },
    {
      "epoch": 2.080059051513672e-05,
      "step": 95811,
      "training_step_time": 7.346121788024902
    },
    {
      "epoch": 2.0801544189453126e-05,
      "model_forward_time": 1.7831075191497803,
      "step": 95812
    },
    {
      "epoch": 2.0801544189453126e-05,
      "step": 95812,
      "training_step_time": 7.643904685974121
    },
    {
      "epoch": 2.080249786376953e-05,
      "model_forward_time": 1.7799723148345947,
      "step": 95813
    },
    {
      "epoch": 2.080249786376953e-05,
      "step": 95813,
      "training_step_time": 6.860226392745972
    },
    {
      "epoch": 2.0803451538085938e-05,
      "model_forward_time": 1.778548240661621,
      "step": 95814
    },
    {
      "epoch": 2.0803451538085938e-05,
      "step": 95814,
      "training_step_time": 6.834642648696899
    },
    {
      "epoch": 2.0804405212402345e-05,
      "model_forward_time": 1.7846577167510986,
      "step": 95815
    },
    {
      "epoch": 2.0804405212402345e-05,
      "step": 95815,
      "training_step_time": 6.84685492515564
    },
    {
      "epoch": 2.080535888671875e-05,
      "model_forward_time": 1.7783045768737793,
      "step": 95816
    },
    {
      "epoch": 2.080535888671875e-05,
      "step": 95816,
      "training_step_time": 6.81092381477356
    },
    {
      "epoch": 2.0806312561035156e-05,
      "model_forward_time": 1.7852144241333008,
      "step": 95817
    },
    {
      "epoch": 2.0806312561035156e-05,
      "step": 95817,
      "training_step_time": 8.416664838790894
    },
    {
      "epoch": 2.0807266235351564e-05,
      "model_forward_time": 1.807267189025879,
      "step": 95818
    },
    {
      "epoch": 2.0807266235351564e-05,
      "step": 95818,
      "training_step_time": 11.919293403625488
    },
    {
      "epoch": 2.0808219909667968e-05,
      "model_forward_time": 2.866438865661621,
      "step": 95819
    },
    {
      "epoch": 2.0808219909667968e-05,
      "step": 95819,
      "training_step_time": 7.689070701599121
    },
    {
      "epoch": 2.0809173583984375e-05,
      "grad_norm": 0.1517741084098816,
      "learning_rate": 4.771565743330597e-08,
      "loss": 0.0534,
      "step": 95820
    },
    {
      "epoch": 2.0809173583984375e-05,
      "model_forward_time": 1.764803171157837,
      "step": 95820
    },
    {
      "dynamics_loss_avg": 0.050783812440931796,
      "epoch": 2.0809173583984375e-05,
      "step": 95820
    },
    {
      "action_loss_avg": 0.0012818987132050097,
      "epoch": 2.0809173583984375e-05,
      "step": 95820
    },
    {
      "epoch": 2.0809173583984375e-05,
      "step": 95820,
      "training_step_time": 7.282665252685547
    },
    {
      "epoch": 2.0810127258300783e-05,
      "model_forward_time": 1.7782809734344482,
      "step": 95821
    },
    {
      "epoch": 2.0810127258300783e-05,
      "step": 95821,
      "training_step_time": 7.645450115203857
    },
    {
      "epoch": 2.0811080932617187e-05,
      "model_forward_time": 1.781393051147461,
      "step": 95822
    },
    {
      "epoch": 2.0811080932617187e-05,
      "step": 95822,
      "training_step_time": 7.263648748397827
    },
    {
      "epoch": 2.0812034606933594e-05,
      "model_forward_time": 1.7926738262176514,
      "step": 95823
    },
    {
      "epoch": 2.0812034606933594e-05,
      "step": 95823,
      "training_step_time": 6.818378448486328
    },
    {
      "epoch": 2.081298828125e-05,
      "model_forward_time": 1.7756221294403076,
      "step": 95824
    },
    {
      "epoch": 2.081298828125e-05,
      "step": 95824,
      "training_step_time": 6.810020923614502
    },
    {
      "epoch": 2.0813941955566406e-05,
      "model_forward_time": 1.778047800064087,
      "step": 95825
    },
    {
      "epoch": 2.0813941955566406e-05,
      "step": 95825,
      "training_step_time": 6.860188007354736
    },
    {
      "epoch": 2.0814895629882813e-05,
      "model_forward_time": 1.785935640335083,
      "step": 95826
    },
    {
      "epoch": 2.0814895629882813e-05,
      "step": 95826,
      "training_step_time": 7.467207908630371
    },
    {
      "epoch": 2.081584930419922e-05,
      "model_forward_time": 1.780592679977417,
      "step": 95827
    },
    {
      "epoch": 2.081584930419922e-05,
      "step": 95827,
      "training_step_time": 10.51650619506836
    },
    {
      "epoch": 2.0816802978515624e-05,
      "model_forward_time": 2.8319742679595947,
      "step": 95828
    },
    {
      "epoch": 2.0816802978515624e-05,
      "step": 95828,
      "training_step_time": 7.8033647537231445
    },
    {
      "epoch": 2.0817756652832032e-05,
      "model_forward_time": 1.7698674201965332,
      "step": 95829
    },
    {
      "epoch": 2.0817756652832032e-05,
      "step": 95829,
      "training_step_time": 7.876630067825317
    },
    {
      "epoch": 2.0818710327148436e-05,
      "grad_norm": 0.08475348353385925,
      "learning_rate": 4.748804198026646e-08,
      "loss": 0.0574,
      "step": 95830
    },
    {
      "epoch": 2.0818710327148436e-05,
      "model_forward_time": 1.7786593437194824,
      "step": 95830
    },
    {
      "dynamics_loss_avg": 0.05263149365782738,
      "epoch": 2.0818710327148436e-05,
      "step": 95830
    },
    {
      "action_loss_avg": 0.0008245538629125804,
      "epoch": 2.0818710327148436e-05,
      "step": 95830
    },
    {
      "epoch": 2.0818710327148436e-05,
      "step": 95830,
      "training_step_time": 7.635726451873779
    },
    {
      "epoch": 2.0819664001464843e-05,
      "model_forward_time": 1.4203777313232422,
      "step": 95831
    },
    {
      "epoch": 2.0819664001464843e-05,
      "step": 95831,
      "training_step_time": 7.25586724281311
    },
    {
      "epoch": 2.082061767578125e-05,
      "model_forward_time": 1.775695562362671,
      "step": 95832
    },
    {
      "epoch": 2.082061767578125e-05,
      "step": 95832,
      "training_step_time": 6.894419193267822
    },
    {
      "epoch": 2.0821571350097655e-05,
      "model_forward_time": 1.796464443206787,
      "step": 95833
    },
    {
      "epoch": 2.0821571350097655e-05,
      "step": 95833,
      "training_step_time": 6.855377912521362
    },
    {
      "epoch": 2.0822525024414062e-05,
      "model_forward_time": 1.7907273769378662,
      "step": 95834
    },
    {
      "epoch": 2.0822525024414062e-05,
      "step": 95834,
      "training_step_time": 6.8577635288238525
    },
    {
      "epoch": 2.082347869873047e-05,
      "model_forward_time": 1.7756927013397217,
      "step": 95835
    },
    {
      "epoch": 2.082347869873047e-05,
      "step": 95835,
      "training_step_time": 7.24070143699646
    },
    {
      "epoch": 2.0824432373046874e-05,
      "model_forward_time": 1.7808876037597656,
      "step": 95836
    },
    {
      "epoch": 2.0824432373046874e-05,
      "step": 95836,
      "training_step_time": 9.435887575149536
    },
    {
      "epoch": 2.082538604736328e-05,
      "model_forward_time": 2.886611223220825,
      "step": 95837
    },
    {
      "epoch": 2.082538604736328e-05,
      "step": 95837,
      "training_step_time": 7.817003011703491
    },
    {
      "epoch": 2.082633972167969e-05,
      "model_forward_time": 1.779388427734375,
      "step": 95838
    },
    {
      "epoch": 2.082633972167969e-05,
      "step": 95838,
      "training_step_time": 8.772745847702026
    },
    {
      "epoch": 2.0827293395996092e-05,
      "model_forward_time": 1.7735528945922852,
      "step": 95839
    },
    {
      "epoch": 2.0827293395996092e-05,
      "step": 95839,
      "training_step_time": 7.5879881381988525
    },
    {
      "epoch": 2.08282470703125e-05,
      "grad_norm": 0.08390021324157715,
      "learning_rate": 4.726096812648795e-08,
      "loss": 0.0553,
      "step": 95840
    },
    {
      "epoch": 2.08282470703125e-05,
      "model_forward_time": 1.7853026390075684,
      "step": 95840
    },
    {
      "dynamics_loss_avg": 0.05707149300724268,
      "epoch": 2.08282470703125e-05,
      "step": 95840
    },
    {
      "action_loss_avg": 0.0009652227628976107,
      "epoch": 2.08282470703125e-05,
      "step": 95840
    },
    {
      "epoch": 2.08282470703125e-05,
      "step": 95840,
      "training_step_time": 7.341252565383911
    },
    {
      "epoch": 2.0829200744628907e-05,
      "model_forward_time": 1.7727365493774414,
      "step": 95841
    },
    {
      "epoch": 2.0829200744628907e-05,
      "step": 95841,
      "training_step_time": 6.833763599395752
    },
    {
      "epoch": 2.083015441894531e-05,
      "model_forward_time": 1.7793915271759033,
      "step": 95842
    },
    {
      "epoch": 2.083015441894531e-05,
      "step": 95842,
      "training_step_time": 6.864870309829712
    },
    {
      "epoch": 2.083110809326172e-05,
      "model_forward_time": 1.7792365550994873,
      "step": 95843
    },
    {
      "epoch": 2.083110809326172e-05,
      "step": 95843,
      "training_step_time": 6.887311697006226
    },
    {
      "epoch": 2.0832061767578126e-05,
      "model_forward_time": 1.7820141315460205,
      "step": 95844
    },
    {
      "epoch": 2.0832061767578126e-05,
      "step": 95844,
      "training_step_time": 7.2953941822052
    },
    {
      "epoch": 2.083301544189453e-05,
      "model_forward_time": 1.7882158756256104,
      "step": 95845
    },
    {
      "epoch": 2.083301544189453e-05,
      "step": 95845,
      "training_step_time": 7.701159954071045
    },
    {
      "epoch": 2.0833969116210938e-05,
      "model_forward_time": 1.8343193531036377,
      "step": 95846
    },
    {
      "epoch": 2.0833969116210938e-05,
      "step": 95846,
      "training_step_time": 9.64072036743164
    },
    {
      "epoch": 2.0834922790527345e-05,
      "model_forward_time": 2.8373639583587646,
      "step": 95847
    },
    {
      "epoch": 2.0834922790527345e-05,
      "step": 95847,
      "training_step_time": 8.718265295028687
    },
    {
      "epoch": 2.083587646484375e-05,
      "model_forward_time": 1.7757482528686523,
      "step": 95848
    },
    {
      "epoch": 2.083587646484375e-05,
      "step": 95848,
      "training_step_time": 7.1806488037109375
    },
    {
      "epoch": 2.0836830139160156e-05,
      "model_forward_time": 1.7827353477478027,
      "step": 95849
    },
    {
      "epoch": 2.0836830139160156e-05,
      "step": 95849,
      "training_step_time": 7.615041971206665
    },
    {
      "epoch": 2.0837783813476564e-05,
      "grad_norm": 0.06765563786029816,
      "learning_rate": 4.703443589680445e-08,
      "loss": 0.0541,
      "step": 95850
    },
    {
      "epoch": 2.0837783813476564e-05,
      "model_forward_time": 1.7802610397338867,
      "step": 95850
    },
    {
      "dynamics_loss_avg": 0.04836553558707237,
      "epoch": 2.0837783813476564e-05,
      "step": 95850
    },
    {
      "action_loss_avg": 0.0010661175838322379,
      "epoch": 2.0837783813476564e-05,
      "step": 95850
    },
    {
      "epoch": 2.0837783813476564e-05,
      "step": 95850,
      "training_step_time": 6.856560468673706
    },
    {
      "epoch": 2.0838737487792968e-05,
      "model_forward_time": 1.7790014743804932,
      "step": 95851
    },
    {
      "epoch": 2.0838737487792968e-05,
      "step": 95851,
      "training_step_time": 6.8723304271698
    },
    {
      "epoch": 2.0839691162109375e-05,
      "model_forward_time": 1.783452033996582,
      "step": 95852
    },
    {
      "epoch": 2.0839691162109375e-05,
      "step": 95852,
      "training_step_time": 6.873572587966919
    },
    {
      "epoch": 2.0840644836425783e-05,
      "model_forward_time": 1.7815523147583008,
      "step": 95853
    },
    {
      "epoch": 2.0840644836425783e-05,
      "step": 95853,
      "training_step_time": 7.347645282745361
    },
    {
      "epoch": 2.0841598510742187e-05,
      "model_forward_time": 1.7834386825561523,
      "step": 95854
    },
    {
      "epoch": 2.0841598510742187e-05,
      "step": 95854,
      "training_step_time": 8.738786935806274
    },
    {
      "epoch": 2.0842552185058594e-05,
      "model_forward_time": 1.804591178894043,
      "step": 95855
    },
    {
      "epoch": 2.0842552185058594e-05,
      "step": 95855,
      "training_step_time": 10.250158309936523
    },
    {
      "epoch": 2.0843505859375e-05,
      "model_forward_time": 2.830223798751831,
      "step": 95856
    },
    {
      "epoch": 2.0843505859375e-05,
      "step": 95856,
      "training_step_time": 7.684888124465942
    },
    {
      "epoch": 2.0844459533691406e-05,
      "model_forward_time": 1.7808918952941895,
      "step": 95857
    },
    {
      "epoch": 2.0844459533691406e-05,
      "step": 95857,
      "training_step_time": 7.287784099578857
    },
    {
      "epoch": 2.0845413208007813e-05,
      "model_forward_time": 1.7772800922393799,
      "step": 95858
    },
    {
      "epoch": 2.0845413208007813e-05,
      "step": 95858,
      "training_step_time": 7.358324766159058
    },
    {
      "epoch": 2.084636688232422e-05,
      "model_forward_time": 1.7886109352111816,
      "step": 95859
    },
    {
      "epoch": 2.084636688232422e-05,
      "step": 95859,
      "training_step_time": 6.8679358959198
    },
    {
      "epoch": 2.0847320556640624e-05,
      "grad_norm": 0.0884208083152771,
      "learning_rate": 4.680844531598783e-08,
      "loss": 0.0568,
      "step": 95860
    },
    {
      "epoch": 2.0847320556640624e-05,
      "model_forward_time": 1.7811198234558105,
      "step": 95860
    },
    {
      "dynamics_loss_avg": 0.051237198151648046,
      "epoch": 2.0847320556640624e-05,
      "step": 95860
    },
    {
      "action_loss_avg": 0.0009720453497720882,
      "epoch": 2.0847320556640624e-05,
      "step": 95860
    },
    {
      "epoch": 2.0847320556640624e-05,
      "step": 95860,
      "training_step_time": 7.0038933753967285
    },
    {
      "epoch": 2.0848274230957032e-05,
      "model_forward_time": 1.782912254333496,
      "step": 95861
    },
    {
      "epoch": 2.0848274230957032e-05,
      "step": 95861,
      "training_step_time": 6.86515736579895
    },
    {
      "epoch": 2.0849227905273436e-05,
      "model_forward_time": 1.7872207164764404,
      "step": 95862
    },
    {
      "epoch": 2.0849227905273436e-05,
      "step": 95862,
      "training_step_time": 7.3362767696380615
    },
    {
      "epoch": 2.0850181579589843e-05,
      "model_forward_time": 1.7716944217681885,
      "step": 95863
    },
    {
      "epoch": 2.0850181579589843e-05,
      "step": 95863,
      "training_step_time": 7.605495452880859
    },
    {
      "epoch": 2.085113525390625e-05,
      "model_forward_time": 1.7780869007110596,
      "step": 95864
    },
    {
      "epoch": 2.085113525390625e-05,
      "step": 95864,
      "training_step_time": 10.177875280380249
    },
    {
      "epoch": 2.0852088928222655e-05,
      "model_forward_time": 2.8340630531311035,
      "step": 95865
    },
    {
      "epoch": 2.0852088928222655e-05,
      "step": 95865,
      "training_step_time": 7.757538318634033
    },
    {
      "epoch": 2.0853042602539062e-05,
      "model_forward_time": 1.7822046279907227,
      "step": 95866
    },
    {
      "epoch": 2.0853042602539062e-05,
      "step": 95866,
      "training_step_time": 7.667217254638672
    },
    {
      "epoch": 2.085399627685547e-05,
      "model_forward_time": 1.7786436080932617,
      "step": 95867
    },
    {
      "epoch": 2.085399627685547e-05,
      "step": 95867,
      "training_step_time": 7.633005142211914
    },
    {
      "epoch": 2.0854949951171874e-05,
      "model_forward_time": 1.7819628715515137,
      "step": 95868
    },
    {
      "epoch": 2.0854949951171874e-05,
      "step": 95868,
      "training_step_time": 6.833160161972046
    },
    {
      "epoch": 2.085590362548828e-05,
      "model_forward_time": 1.7770075798034668,
      "step": 95869
    },
    {
      "epoch": 2.085590362548828e-05,
      "step": 95869,
      "training_step_time": 6.836386442184448
    },
    {
      "epoch": 2.085685729980469e-05,
      "grad_norm": 0.08837052434682846,
      "learning_rate": 4.658299640875275e-08,
      "loss": 0.0547,
      "step": 95870
    },
    {
      "epoch": 2.085685729980469e-05,
      "model_forward_time": 1.777465581893921,
      "step": 95870
    },
    {
      "dynamics_loss_avg": 0.05328638404607773,
      "epoch": 2.085685729980469e-05,
      "step": 95870
    },
    {
      "action_loss_avg": 0.0014881812618114053,
      "epoch": 2.085685729980469e-05,
      "step": 95870
    },
    {
      "epoch": 2.085685729980469e-05,
      "step": 95870,
      "training_step_time": 6.815478563308716
    },
    {
      "epoch": 2.0857810974121092e-05,
      "model_forward_time": 1.780883550643921,
      "step": 95871
    },
    {
      "epoch": 2.0857810974121092e-05,
      "step": 95871,
      "training_step_time": 6.8474812507629395
    },
    {
      "epoch": 2.08587646484375e-05,
      "model_forward_time": 1.775813102722168,
      "step": 95872
    },
    {
      "epoch": 2.08587646484375e-05,
      "step": 95872,
      "training_step_time": 7.5992255210876465
    },
    {
      "epoch": 2.0859718322753907e-05,
      "model_forward_time": 1.7706952095031738,
      "step": 95873
    },
    {
      "epoch": 2.0859718322753907e-05,
      "step": 95873,
      "training_step_time": 11.152728796005249
    },
    {
      "epoch": 2.086067199707031e-05,
      "model_forward_time": 2.864847421646118,
      "step": 95874
    },
    {
      "epoch": 2.086067199707031e-05,
      "step": 95874,
      "training_step_time": 10.014750242233276
    },
    {
      "epoch": 2.086162567138672e-05,
      "model_forward_time": 1.769169807434082,
      "step": 95875
    },
    {
      "epoch": 2.086162567138672e-05,
      "step": 95875,
      "training_step_time": 8.309154033660889
    },
    {
      "epoch": 2.0862579345703126e-05,
      "model_forward_time": 1.7853703498840332,
      "step": 95876
    },
    {
      "epoch": 2.0862579345703126e-05,
      "step": 95876,
      "training_step_time": 7.278038740158081
    },
    {
      "epoch": 2.086353302001953e-05,
      "model_forward_time": 1.783452033996582,
      "step": 95877
    },
    {
      "epoch": 2.086353302001953e-05,
      "step": 95877,
      "training_step_time": 7.661153793334961
    },
    {
      "epoch": 2.0864486694335938e-05,
      "model_forward_time": 1.7862446308135986,
      "step": 95878
    },
    {
      "epoch": 2.0864486694335938e-05,
      "step": 95878,
      "training_step_time": 6.820214033126831
    },
    {
      "epoch": 2.0865440368652345e-05,
      "model_forward_time": 1.783825397491455,
      "step": 95879
    },
    {
      "epoch": 2.0865440368652345e-05,
      "step": 95879,
      "training_step_time": 6.8391711711883545
    },
    {
      "epoch": 2.086639404296875e-05,
      "grad_norm": 0.15270133316516876,
      "learning_rate": 4.635808919975393e-08,
      "loss": 0.0571,
      "step": 95880
    },
    {
      "epoch": 2.086639404296875e-05,
      "model_forward_time": 1.783188819885254,
      "step": 95880
    },
    {
      "dynamics_loss_avg": 0.05393769228830934,
      "epoch": 2.086639404296875e-05,
      "step": 95880
    },
    {
      "action_loss_avg": 0.0013091487169731408,
      "epoch": 2.086639404296875e-05,
      "step": 95880
    },
    {
      "epoch": 2.086639404296875e-05,
      "step": 95880,
      "training_step_time": 6.823371887207031
    },
    {
      "epoch": 2.0867347717285156e-05,
      "model_forward_time": 1.7811000347137451,
      "step": 95881
    },
    {
      "epoch": 2.0867347717285156e-05,
      "step": 95881,
      "training_step_time": 8.469704866409302
    },
    {
      "epoch": 2.0868301391601564e-05,
      "model_forward_time": 1.789768934249878,
      "step": 95882
    },
    {
      "epoch": 2.0868301391601564e-05,
      "step": 95882,
      "training_step_time": 8.601415872573853
    },
    {
      "epoch": 2.0869255065917968e-05,
      "model_forward_time": 2.8894295692443848,
      "step": 95883
    },
    {
      "epoch": 2.0869255065917968e-05,
      "step": 95883,
      "training_step_time": 7.801386833190918
    },
    {
      "epoch": 2.0870208740234375e-05,
      "model_forward_time": 1.7737116813659668,
      "step": 95884
    },
    {
      "epoch": 2.0870208740234375e-05,
      "step": 95884,
      "training_step_time": 7.645299434661865
    },
    {
      "epoch": 2.0871162414550783e-05,
      "model_forward_time": 1.7822589874267578,
      "step": 95885
    },
    {
      "epoch": 2.0871162414550783e-05,
      "step": 95885,
      "training_step_time": 7.980902671813965
    },
    {
      "epoch": 2.0872116088867187e-05,
      "model_forward_time": 1.7839782238006592,
      "step": 95886
    },
    {
      "epoch": 2.0872116088867187e-05,
      "step": 95886,
      "training_step_time": 7.230447769165039
    },
    {
      "epoch": 2.0873069763183594e-05,
      "model_forward_time": 1.7803034782409668,
      "step": 95887
    },
    {
      "epoch": 2.0873069763183594e-05,
      "step": 95887,
      "training_step_time": 6.8405609130859375
    },
    {
      "epoch": 2.08740234375e-05,
      "model_forward_time": 1.790823221206665,
      "step": 95888
    },
    {
      "epoch": 2.08740234375e-05,
      "step": 95888,
      "training_step_time": 6.8504345417022705
    },
    {
      "epoch": 2.0874977111816406e-05,
      "model_forward_time": 1.782900094985962,
      "step": 95889
    },
    {
      "epoch": 2.0874977111816406e-05,
      "step": 95889,
      "training_step_time": 6.854388236999512
    },
    {
      "epoch": 2.0875930786132813e-05,
      "grad_norm": 0.08728288859128952,
      "learning_rate": 4.613372371358671e-08,
      "loss": 0.0569,
      "step": 95890
    },
    {
      "epoch": 2.0875930786132813e-05,
      "model_forward_time": 1.7774908542633057,
      "step": 95890
    },
    {
      "dynamics_loss_avg": 0.07215221598744392,
      "epoch": 2.0875930786132813e-05,
      "step": 95890
    },
    {
      "action_loss_avg": 0.00187965901568532,
      "epoch": 2.0875930786132813e-05,
      "step": 95890
    },
    {
      "epoch": 2.0875930786132813e-05,
      "step": 95890,
      "training_step_time": 7.535284519195557
    },
    {
      "epoch": 2.0876884460449217e-05,
      "model_forward_time": 1.7775280475616455,
      "step": 95891
    },
    {
      "epoch": 2.0876884460449217e-05,
      "step": 95891,
      "training_step_time": 8.795458793640137
    },
    {
      "epoch": 2.0877838134765624e-05,
      "model_forward_time": 2.943164110183716,
      "step": 95892
    },
    {
      "epoch": 2.0877838134765624e-05,
      "step": 95892,
      "training_step_time": 9.52209997177124
    },
    {
      "epoch": 2.0878791809082032e-05,
      "model_forward_time": 1.7855010032653809,
      "step": 95893
    },
    {
      "epoch": 2.0878791809082032e-05,
      "step": 95893,
      "training_step_time": 7.751270532608032
    },
    {
      "epoch": 2.0879745483398436e-05,
      "model_forward_time": 1.7785568237304688,
      "step": 95894
    },
    {
      "epoch": 2.0879745483398436e-05,
      "step": 95894,
      "training_step_time": 7.672903060913086
    },
    {
      "epoch": 2.0880699157714843e-05,
      "model_forward_time": 1.78639554977417,
      "step": 95895
    },
    {
      "epoch": 2.0880699157714843e-05,
      "step": 95895,
      "training_step_time": 7.377282619476318
    },
    {
      "epoch": 2.088165283203125e-05,
      "model_forward_time": 1.7821040153503418,
      "step": 95896
    },
    {
      "epoch": 2.088165283203125e-05,
      "step": 95896,
      "training_step_time": 7.590334892272949
    },
    {
      "epoch": 2.0882606506347655e-05,
      "model_forward_time": 1.7734639644622803,
      "step": 95897
    },
    {
      "epoch": 2.0882606506347655e-05,
      "step": 95897,
      "training_step_time": 6.850268840789795
    },
    {
      "epoch": 2.0883560180664062e-05,
      "model_forward_time": 1.7547643184661865,
      "step": 95898
    },
    {
      "epoch": 2.0883560180664062e-05,
      "step": 95898,
      "training_step_time": 6.808438062667847
    },
    {
      "epoch": 2.088451385498047e-05,
      "model_forward_time": 1.7760426998138428,
      "step": 95899
    },
    {
      "epoch": 2.088451385498047e-05,
      "step": 95899,
      "training_step_time": 7.459674596786499
    },
    {
      "epoch": 2.0885467529296874e-05,
      "grad_norm": 0.14422477781772614,
      "learning_rate": 4.590989997478756e-08,
      "loss": 0.0543,
      "step": 95900
    },
    {
      "epoch": 2.0885467529296874e-05,
      "model_forward_time": 1.7612547874450684,
      "step": 95900
    },
    {
      "dynamics_loss_avg": 0.04387253113090992,
      "epoch": 2.0885467529296874e-05,
      "step": 95900
    },
    {
      "action_loss_avg": 0.0011543702246854081,
      "epoch": 2.0885467529296874e-05,
      "step": 95900
    },
    {
      "epoch": 2.0885467529296874e-05,
      "step": 95900,
      "training_step_time": 7.699946165084839
    },
    {
      "epoch": 2.088642120361328e-05,
      "model_forward_time": 2.8946800231933594,
      "step": 95901
    },
    {
      "epoch": 2.088642120361328e-05,
      "step": 95901,
      "training_step_time": 10.3581223487854
    },
    {
      "epoch": 2.088737487792969e-05,
      "model_forward_time": 1.768176555633545,
      "step": 95902
    },
    {
      "epoch": 2.088737487792969e-05,
      "step": 95902,
      "training_step_time": 8.76011323928833
    },
    {
      "epoch": 2.0888328552246093e-05,
      "model_forward_time": 1.7671709060668945,
      "step": 95903
    },
    {
      "epoch": 2.0888328552246093e-05,
      "step": 95903,
      "training_step_time": 8.52792501449585
    },
    {
      "epoch": 2.08892822265625e-05,
      "model_forward_time": 1.7793104648590088,
      "step": 95904
    },
    {
      "epoch": 2.08892822265625e-05,
      "step": 95904,
      "training_step_time": 7.371350049972534
    },
    {
      "epoch": 2.0890235900878907e-05,
      "model_forward_time": 1.7754392623901367,
      "step": 95905
    },
    {
      "epoch": 2.0890235900878907e-05,
      "step": 95905,
      "training_step_time": 7.586807727813721
    },
    {
      "epoch": 2.089118957519531e-05,
      "model_forward_time": 1.7704765796661377,
      "step": 95906
    },
    {
      "epoch": 2.089118957519531e-05,
      "step": 95906,
      "training_step_time": 6.848356485366821
    },
    {
      "epoch": 2.089214324951172e-05,
      "model_forward_time": 1.7662370204925537,
      "step": 95907
    },
    {
      "epoch": 2.089214324951172e-05,
      "step": 95907,
      "training_step_time": 6.813068151473999
    },
    {
      "epoch": 2.0893096923828126e-05,
      "model_forward_time": 1.7750155925750732,
      "step": 95908
    },
    {
      "epoch": 2.0893096923828126e-05,
      "step": 95908,
      "training_step_time": 7.526588201522827
    },
    {
      "epoch": 2.089405059814453e-05,
      "model_forward_time": 1.780031442642212,
      "step": 95909
    },
    {
      "epoch": 2.089405059814453e-05,
      "step": 95909,
      "training_step_time": 9.260497093200684
    },
    {
      "epoch": 2.0895004272460938e-05,
      "grad_norm": 0.07354319095611572,
      "learning_rate": 4.568661800783303e-08,
      "loss": 0.0567,
      "step": 95910
    },
    {
      "epoch": 2.0895004272460938e-05,
      "model_forward_time": 1.8409404754638672,
      "step": 95910
    },
    {
      "dynamics_loss_avg": 0.05933994334191084,
      "epoch": 2.0895004272460938e-05,
      "step": 95910
    },
    {
      "action_loss_avg": 0.0014993189921369775,
      "epoch": 2.0895004272460938e-05,
      "step": 95910
    },
    {
      "epoch": 2.0895004272460938e-05,
      "step": 95910,
      "training_step_time": 10.389865636825562
    },
    {
      "epoch": 2.0895957946777345e-05,
      "model_forward_time": 1.7644286155700684,
      "step": 95911
    },
    {
      "epoch": 2.0895957946777345e-05,
      "step": 95911,
      "training_step_time": 7.82456374168396
    },
    {
      "epoch": 2.089691162109375e-05,
      "model_forward_time": 1.764587163925171,
      "step": 95912
    },
    {
      "epoch": 2.089691162109375e-05,
      "step": 95912,
      "training_step_time": 8.309964179992676
    },
    {
      "epoch": 2.0897865295410156e-05,
      "model_forward_time": 1.7700259685516357,
      "step": 95913
    },
    {
      "epoch": 2.0897865295410156e-05,
      "step": 95913,
      "training_step_time": 7.277417898178101
    },
    {
      "epoch": 2.0898818969726564e-05,
      "model_forward_time": 1.7755861282348633,
      "step": 95914
    },
    {
      "epoch": 2.0898818969726564e-05,
      "step": 95914,
      "training_step_time": 7.6499924659729
    },
    {
      "epoch": 2.0899772644042968e-05,
      "model_forward_time": 1.7903897762298584,
      "step": 95915
    },
    {
      "epoch": 2.0899772644042968e-05,
      "step": 95915,
      "training_step_time": 6.875849485397339
    },
    {
      "epoch": 2.0900726318359375e-05,
      "model_forward_time": 1.778726577758789,
      "step": 95916
    },
    {
      "epoch": 2.0900726318359375e-05,
      "step": 95916,
      "training_step_time": 6.863163232803345
    },
    {
      "epoch": 2.0901679992675783e-05,
      "model_forward_time": 1.7803609371185303,
      "step": 95917
    },
    {
      "epoch": 2.0901679992675783e-05,
      "step": 95917,
      "training_step_time": 7.9873011112213135
    },
    {
      "epoch": 2.0902633666992187e-05,
      "model_forward_time": 1.7792716026306152,
      "step": 95918
    },
    {
      "epoch": 2.0902633666992187e-05,
      "step": 95918,
      "training_step_time": 7.841975450515747
    },
    {
      "epoch": 2.0903587341308594e-05,
      "model_forward_time": 1.8102591037750244,
      "step": 95919
    },
    {
      "epoch": 2.0903587341308594e-05,
      "step": 95919,
      "training_step_time": 10.94587254524231
    },
    {
      "epoch": 2.0904541015625e-05,
      "grad_norm": 0.08899835497140884,
      "learning_rate": 4.54638778371419e-08,
      "loss": 0.0575,
      "step": 95920
    },
    {
      "epoch": 2.0904541015625e-05,
      "model_forward_time": 2.8289477825164795,
      "step": 95920
    },
    {
      "dynamics_loss_avg": 0.06059556007385254,
      "epoch": 2.0904541015625e-05,
      "step": 95920
    },
    {
      "action_loss_avg": 0.001247288633021526,
      "epoch": 2.0904541015625e-05,
      "step": 95920
    },
    {
      "epoch": 2.0904541015625e-05,
      "step": 95920,
      "training_step_time": 7.846588134765625
    },
    {
      "epoch": 2.0905494689941406e-05,
      "model_forward_time": 1.395655632019043,
      "step": 95921
    },
    {
      "epoch": 2.0905494689941406e-05,
      "step": 95921,
      "training_step_time": 8.263950824737549
    },
    {
      "epoch": 2.0906448364257813e-05,
      "model_forward_time": 1.7705564498901367,
      "step": 95922
    },
    {
      "epoch": 2.0906448364257813e-05,
      "step": 95922,
      "training_step_time": 7.3201775550842285
    },
    {
      "epoch": 2.0907402038574217e-05,
      "model_forward_time": 1.7735509872436523,
      "step": 95923
    },
    {
      "epoch": 2.0907402038574217e-05,
      "step": 95923,
      "training_step_time": 7.611049175262451
    },
    {
      "epoch": 2.0908355712890625e-05,
      "model_forward_time": 1.7624895572662354,
      "step": 95924
    },
    {
      "epoch": 2.0908355712890625e-05,
      "step": 95924,
      "training_step_time": 6.839221000671387
    },
    {
      "epoch": 2.0909309387207032e-05,
      "model_forward_time": 1.7644283771514893,
      "step": 95925
    },
    {
      "epoch": 2.0909309387207032e-05,
      "step": 95925,
      "training_step_time": 6.841188669204712
    },
    {
      "epoch": 2.0910263061523436e-05,
      "model_forward_time": 1.4011499881744385,
      "step": 95926
    },
    {
      "epoch": 2.0910263061523436e-05,
      "step": 95926,
      "training_step_time": 7.31610631942749
    },
    {
      "epoch": 2.0911216735839843e-05,
      "model_forward_time": 1.7577602863311768,
      "step": 95927
    },
    {
      "epoch": 2.0911216735839843e-05,
      "step": 95927,
      "training_step_time": 7.63205099105835
    },
    {
      "epoch": 2.091217041015625e-05,
      "model_forward_time": 1.8409624099731445,
      "step": 95928
    },
    {
      "epoch": 2.091217041015625e-05,
      "step": 95928,
      "training_step_time": 9.464386463165283
    },
    {
      "epoch": 2.0913124084472655e-05,
      "model_forward_time": 2.8423585891723633,
      "step": 95929
    },
    {
      "epoch": 2.0913124084472655e-05,
      "step": 95929,
      "training_step_time": 8.601879119873047
    },
    {
      "epoch": 2.0914077758789062e-05,
      "grad_norm": 0.0850997343659401,
      "learning_rate": 4.524167948707192e-08,
      "loss": 0.0573,
      "step": 95930
    },
    {
      "epoch": 2.0914077758789062e-05,
      "model_forward_time": 1.7709262371063232,
      "step": 95930
    },
    {
      "dynamics_loss_avg": 0.04740859046578407,
      "epoch": 2.0914077758789062e-05,
      "step": 95930
    },
    {
      "action_loss_avg": 0.00068243331334088,
      "epoch": 2.0914077758789062e-05,
      "step": 95930
    },
    {
      "epoch": 2.0914077758789062e-05,
      "step": 95930,
      "training_step_time": 7.705482244491577
    },
    {
      "epoch": 2.091503143310547e-05,
      "model_forward_time": 1.7646334171295166,
      "step": 95931
    },
    {
      "epoch": 2.091503143310547e-05,
      "step": 95931,
      "training_step_time": 7.695712327957153
    },
    {
      "epoch": 2.0915985107421874e-05,
      "model_forward_time": 1.7695083618164062,
      "step": 95932
    },
    {
      "epoch": 2.0915985107421874e-05,
      "step": 95932,
      "training_step_time": 7.592263698577881
    },
    {
      "epoch": 2.091693878173828e-05,
      "model_forward_time": 1.7858898639678955,
      "step": 95933
    },
    {
      "epoch": 2.091693878173828e-05,
      "step": 95933,
      "training_step_time": 6.8491315841674805
    },
    {
      "epoch": 2.091789245605469e-05,
      "model_forward_time": 1.76176118850708,
      "step": 95934
    },
    {
      "epoch": 2.091789245605469e-05,
      "step": 95934,
      "training_step_time": 6.816899299621582
    },
    {
      "epoch": 2.0918846130371093e-05,
      "model_forward_time": 1.7662327289581299,
      "step": 95935
    },
    {
      "epoch": 2.0918846130371093e-05,
      "step": 95935,
      "training_step_time": 7.32779598236084
    },
    {
      "epoch": 2.09197998046875e-05,
      "model_forward_time": 1.7648770809173584,
      "step": 95936
    },
    {
      "epoch": 2.09197998046875e-05,
      "step": 95936,
      "training_step_time": 7.588979005813599
    },
    {
      "epoch": 2.0920753479003907e-05,
      "model_forward_time": 1.8378965854644775,
      "step": 95937
    },
    {
      "epoch": 2.0920753479003907e-05,
      "step": 95937,
      "training_step_time": 8.815047025680542
    },
    {
      "epoch": 2.092170715332031e-05,
      "model_forward_time": 2.8290719985961914,
      "step": 95938
    },
    {
      "epoch": 2.092170715332031e-05,
      "step": 95938,
      "training_step_time": 11.325461149215698
    },
    {
      "epoch": 2.092266082763672e-05,
      "model_forward_time": 1.7616322040557861,
      "step": 95939
    },
    {
      "epoch": 2.092266082763672e-05,
      "step": 95939,
      "training_step_time": 7.68998384475708
    },
    {
      "epoch": 2.0923614501953126e-05,
      "grad_norm": 0.09179560095071793,
      "learning_rate": 4.502002298192199e-08,
      "loss": 0.0558,
      "step": 95940
    },
    {
      "epoch": 2.0923614501953126e-05,
      "model_forward_time": 1.765826940536499,
      "step": 95940
    },
    {
      "dynamics_loss_avg": 0.054239429719746116,
      "epoch": 2.0923614501953126e-05,
      "step": 95940
    },
    {
      "action_loss_avg": 0.001783269818406552,
      "epoch": 2.0923614501953126e-05,
      "step": 95940
    },
    {
      "epoch": 2.0923614501953126e-05,
      "step": 95940,
      "training_step_time": 7.6625001430511475
    },
    {
      "epoch": 2.092456817626953e-05,
      "model_forward_time": 1.7686893939971924,
      "step": 95941
    },
    {
      "epoch": 2.092456817626953e-05,
      "step": 95941,
      "training_step_time": 7.576252460479736
    },
    {
      "epoch": 2.0925521850585938e-05,
      "model_forward_time": 1.7759106159210205,
      "step": 95942
    },
    {
      "epoch": 2.0925521850585938e-05,
      "step": 95942,
      "training_step_time": 6.822566747665405
    },
    {
      "epoch": 2.0926475524902345e-05,
      "model_forward_time": 1.774308204650879,
      "step": 95943
    },
    {
      "epoch": 2.0926475524902345e-05,
      "step": 95943,
      "training_step_time": 6.832361936569214
    },
    {
      "epoch": 2.092742919921875e-05,
      "model_forward_time": 1.7676970958709717,
      "step": 95944
    },
    {
      "epoch": 2.092742919921875e-05,
      "step": 95944,
      "training_step_time": 7.247281551361084
    },
    {
      "epoch": 2.0928382873535157e-05,
      "model_forward_time": 1.7639083862304688,
      "step": 95945
    },
    {
      "epoch": 2.0928382873535157e-05,
      "step": 95945,
      "training_step_time": 7.587115049362183
    },
    {
      "epoch": 2.0929336547851564e-05,
      "model_forward_time": 1.8234872817993164,
      "step": 95946
    },
    {
      "epoch": 2.0929336547851564e-05,
      "step": 95946,
      "training_step_time": 7.852842569351196
    },
    {
      "epoch": 2.0930290222167968e-05,
      "model_forward_time": 2.7926409244537354,
      "step": 95947
    },
    {
      "epoch": 2.0930290222167968e-05,
      "step": 95947,
      "training_step_time": 9.416969776153564
    },
    {
      "epoch": 2.0931243896484375e-05,
      "model_forward_time": 1.747121810913086,
      "step": 95948
    },
    {
      "epoch": 2.0931243896484375e-05,
      "step": 95948,
      "training_step_time": 7.874032974243164
    },
    {
      "epoch": 2.0932197570800783e-05,
      "model_forward_time": 1.7861158847808838,
      "step": 95949
    },
    {
      "epoch": 2.0932197570800783e-05,
      "step": 95949,
      "training_step_time": 7.745265483856201
    },
    {
      "epoch": 2.0933151245117187e-05,
      "grad_norm": 0.14406168460845947,
      "learning_rate": 4.47989083459327e-08,
      "loss": 0.0569,
      "step": 95950
    },
    {
      "epoch": 2.0933151245117187e-05,
      "model_forward_time": 1.7672548294067383,
      "step": 95950
    },
    {
      "dynamics_loss_avg": 0.053399757109582426,
      "epoch": 2.0933151245117187e-05,
      "step": 95950
    },
    {
      "action_loss_avg": 0.0017150423896964639,
      "epoch": 2.0933151245117187e-05,
      "step": 95950
    },
    {
      "epoch": 2.0933151245117187e-05,
      "step": 95950,
      "training_step_time": 7.697498083114624
    },
    {
      "epoch": 2.0934104919433594e-05,
      "model_forward_time": 1.7629029750823975,
      "step": 95951
    },
    {
      "epoch": 2.0934104919433594e-05,
      "step": 95951,
      "training_step_time": 6.860305070877075
    },
    {
      "epoch": 2.093505859375e-05,
      "model_forward_time": 1.767500877380371,
      "step": 95952
    },
    {
      "epoch": 2.093505859375e-05,
      "step": 95952,
      "training_step_time": 6.838465690612793
    },
    {
      "epoch": 2.0936012268066406e-05,
      "model_forward_time": 1.7580797672271729,
      "step": 95953
    },
    {
      "epoch": 2.0936012268066406e-05,
      "step": 95953,
      "training_step_time": 6.857308626174927
    },
    {
      "epoch": 2.0936965942382813e-05,
      "model_forward_time": 1.7795495986938477,
      "step": 95954
    },
    {
      "epoch": 2.0936965942382813e-05,
      "step": 95954,
      "training_step_time": 7.530285835266113
    },
    {
      "epoch": 2.0937919616699217e-05,
      "model_forward_time": 1.4938223361968994,
      "step": 95955
    },
    {
      "epoch": 2.0937919616699217e-05,
      "step": 95955,
      "training_step_time": 8.716297626495361
    },
    {
      "epoch": 2.0938873291015625e-05,
      "model_forward_time": 1.7932119369506836,
      "step": 95956
    },
    {
      "epoch": 2.0938873291015625e-05,
      "step": 95956,
      "training_step_time": 11.046224117279053
    },
    {
      "epoch": 2.0939826965332032e-05,
      "model_forward_time": 2.882727861404419,
      "step": 95957
    },
    {
      "epoch": 2.0939826965332032e-05,
      "step": 95957,
      "training_step_time": 9.419469356536865
    },
    {
      "epoch": 2.0940780639648436e-05,
      "model_forward_time": 1.7783904075622559,
      "step": 95958
    },
    {
      "epoch": 2.0940780639648436e-05,
      "step": 95958,
      "training_step_time": 8.598633289337158
    },
    {
      "epoch": 2.0941734313964843e-05,
      "model_forward_time": 1.7696590423583984,
      "step": 95959
    },
    {
      "epoch": 2.0941734313964843e-05,
      "step": 95959,
      "training_step_time": 7.622495174407959
    },
    {
      "epoch": 2.094268798828125e-05,
      "grad_norm": 0.2198021411895752,
      "learning_rate": 4.457833560328528e-08,
      "loss": 0.0568,
      "step": 95960
    },
    {
      "epoch": 2.094268798828125e-05,
      "model_forward_time": 1.7863209247589111,
      "step": 95960
    },
    {
      "dynamics_loss_avg": 0.05639359932392836,
      "epoch": 2.094268798828125e-05,
      "step": 95960
    },
    {
      "action_loss_avg": 0.002146633606753312,
      "epoch": 2.094268798828125e-05,
      "step": 95960
    },
    {
      "epoch": 2.094268798828125e-05,
      "step": 95960,
      "training_step_time": 7.343266248703003
    },
    {
      "epoch": 2.0943641662597655e-05,
      "model_forward_time": 1.784492015838623,
      "step": 95961
    },
    {
      "epoch": 2.0943641662597655e-05,
      "step": 95961,
      "training_step_time": 6.832446575164795
    },
    {
      "epoch": 2.0944595336914062e-05,
      "model_forward_time": 1.783663272857666,
      "step": 95962
    },
    {
      "epoch": 2.0944595336914062e-05,
      "step": 95962,
      "training_step_time": 6.830197811126709
    },
    {
      "epoch": 2.094554901123047e-05,
      "model_forward_time": 1.7611687183380127,
      "step": 95963
    },
    {
      "epoch": 2.094554901123047e-05,
      "step": 95963,
      "training_step_time": 7.623468637466431
    },
    {
      "epoch": 2.0946502685546874e-05,
      "model_forward_time": 1.825307846069336,
      "step": 95964
    },
    {
      "epoch": 2.0946502685546874e-05,
      "step": 95964,
      "training_step_time": 7.846675157546997
    },
    {
      "epoch": 2.094745635986328e-05,
      "model_forward_time": 1.7843799591064453,
      "step": 95965
    },
    {
      "epoch": 2.094745635986328e-05,
      "step": 95965,
      "training_step_time": 10.980641603469849
    },
    {
      "epoch": 2.094841003417969e-05,
      "model_forward_time": 2.8551740646362305,
      "step": 95966
    },
    {
      "epoch": 2.094841003417969e-05,
      "step": 95966,
      "training_step_time": 9.16977596282959
    },
    {
      "epoch": 2.0949363708496093e-05,
      "model_forward_time": 1.7864820957183838,
      "step": 95967
    },
    {
      "epoch": 2.0949363708496093e-05,
      "step": 95967,
      "training_step_time": 8.929499626159668
    },
    {
      "epoch": 2.09503173828125e-05,
      "model_forward_time": 1.7750155925750732,
      "step": 95968
    },
    {
      "epoch": 2.09503173828125e-05,
      "step": 95968,
      "training_step_time": 7.66008734703064
    },
    {
      "epoch": 2.0951271057128907e-05,
      "model_forward_time": 1.7887225151062012,
      "step": 95969
    },
    {
      "epoch": 2.0951271057128907e-05,
      "step": 95969,
      "training_step_time": 7.637275218963623
    },
    {
      "epoch": 2.095222473144531e-05,
      "grad_norm": 0.07630638033151627,
      "learning_rate": 4.435830477810044e-08,
      "loss": 0.0553,
      "step": 95970
    },
    {
      "epoch": 2.095222473144531e-05,
      "model_forward_time": 1.7799406051635742,
      "step": 95970
    },
    {
      "dynamics_loss_avg": 0.0644820373505354,
      "epoch": 2.095222473144531e-05,
      "step": 95970
    },
    {
      "action_loss_avg": 0.0015084368060342968,
      "epoch": 2.095222473144531e-05,
      "step": 95970
    },
    {
      "epoch": 2.095222473144531e-05,
      "step": 95970,
      "training_step_time": 6.86793065071106
    },
    {
      "epoch": 2.095317840576172e-05,
      "model_forward_time": 1.780259370803833,
      "step": 95971
    },
    {
      "epoch": 2.095317840576172e-05,
      "step": 95971,
      "training_step_time": 6.857718467712402
    },
    {
      "epoch": 2.0954132080078126e-05,
      "model_forward_time": 1.410257339477539,
      "step": 95972
    },
    {
      "epoch": 2.0954132080078126e-05,
      "step": 95972,
      "training_step_time": 7.578888654708862
    },
    {
      "epoch": 2.095508575439453e-05,
      "model_forward_time": 1.7953624725341797,
      "step": 95973
    },
    {
      "epoch": 2.095508575439453e-05,
      "step": 95973,
      "training_step_time": 9.575976133346558
    },
    {
      "epoch": 2.0956039428710938e-05,
      "model_forward_time": 1.7611589431762695,
      "step": 95974
    },
    {
      "epoch": 2.0956039428710938e-05,
      "step": 95974,
      "training_step_time": 8.708658933639526
    },
    {
      "epoch": 2.0956993103027345e-05,
      "model_forward_time": 2.8494768142700195,
      "step": 95975
    },
    {
      "epoch": 2.0956993103027345e-05,
      "step": 95975,
      "training_step_time": 9.905246496200562
    },
    {
      "epoch": 2.095794677734375e-05,
      "model_forward_time": 1.7774064540863037,
      "step": 95976
    },
    {
      "epoch": 2.095794677734375e-05,
      "step": 95976,
      "training_step_time": 7.86644721031189
    },
    {
      "epoch": 2.0958900451660157e-05,
      "model_forward_time": 1.7933483123779297,
      "step": 95977
    },
    {
      "epoch": 2.0958900451660157e-05,
      "step": 95977,
      "training_step_time": 7.67445969581604
    },
    {
      "epoch": 2.0959854125976564e-05,
      "model_forward_time": 1.7692945003509521,
      "step": 95978
    },
    {
      "epoch": 2.0959854125976564e-05,
      "step": 95978,
      "training_step_time": 7.643572807312012
    },
    {
      "epoch": 2.0960807800292968e-05,
      "model_forward_time": 1.8020198345184326,
      "step": 95979
    },
    {
      "epoch": 2.0960807800292968e-05,
      "step": 95979,
      "training_step_time": 6.876170873641968
    },
    {
      "epoch": 2.0961761474609375e-05,
      "grad_norm": 0.07524979114532471,
      "learning_rate": 4.413881589444058e-08,
      "loss": 0.0566,
      "step": 95980
    },
    {
      "epoch": 2.0961761474609375e-05,
      "model_forward_time": 1.7763824462890625,
      "step": 95980
    },
    {
      "dynamics_loss_avg": 0.06550679318606853,
      "epoch": 2.0961761474609375e-05,
      "step": 95980
    },
    {
      "action_loss_avg": 0.0013146066397894175,
      "epoch": 2.0961761474609375e-05,
      "step": 95980
    },
    {
      "epoch": 2.0961761474609375e-05,
      "step": 95980,
      "training_step_time": 6.895845174789429
    },
    {
      "epoch": 2.0962715148925783e-05,
      "model_forward_time": 1.7770137786865234,
      "step": 95981
    },
    {
      "epoch": 2.0962715148925783e-05,
      "step": 95981,
      "training_step_time": 7.311269998550415
    },
    {
      "epoch": 2.0963668823242187e-05,
      "model_forward_time": 1.7861618995666504,
      "step": 95982
    },
    {
      "epoch": 2.0963668823242187e-05,
      "step": 95982,
      "training_step_time": 8.927560329437256
    },
    {
      "epoch": 2.0964622497558594e-05,
      "model_forward_time": 1.8286240100860596,
      "step": 95983
    },
    {
      "epoch": 2.0964622497558594e-05,
      "step": 95983,
      "training_step_time": 7.801676034927368
    },
    {
      "epoch": 2.0965576171875e-05,
      "model_forward_time": 2.8164045810699463,
      "step": 95984
    },
    {
      "epoch": 2.0965576171875e-05,
      "step": 95984,
      "training_step_time": 10.66002345085144
    },
    {
      "epoch": 2.0966529846191406e-05,
      "model_forward_time": 1.7733888626098633,
      "step": 95985
    },
    {
      "epoch": 2.0966529846191406e-05,
      "step": 95985,
      "training_step_time": 7.8736090660095215
    },
    {
      "epoch": 2.0967483520507813e-05,
      "model_forward_time": 1.7702174186706543,
      "step": 95986
    },
    {
      "epoch": 2.0967483520507813e-05,
      "step": 95986,
      "training_step_time": 9.124078512191772
    },
    {
      "epoch": 2.0968437194824217e-05,
      "model_forward_time": 1.7713913917541504,
      "step": 95987
    },
    {
      "epoch": 2.0968437194824217e-05,
      "step": 95987,
      "training_step_time": 8.345866680145264
    },
    {
      "epoch": 2.0969390869140625e-05,
      "model_forward_time": 1.7713074684143066,
      "step": 95988
    },
    {
      "epoch": 2.0969390869140625e-05,
      "step": 95988,
      "training_step_time": 6.885535478591919
    },
    {
      "epoch": 2.0970344543457032e-05,
      "model_forward_time": 1.764045238494873,
      "step": 95989
    },
    {
      "epoch": 2.0970344543457032e-05,
      "step": 95989,
      "training_step_time": 6.888518810272217
    },
    {
      "epoch": 2.0971298217773436e-05,
      "grad_norm": 0.10826306790113449,
      "learning_rate": 4.3919868976308733e-08,
      "loss": 0.0554,
      "step": 95990
    },
    {
      "epoch": 2.0971298217773436e-05,
      "model_forward_time": 1.7649056911468506,
      "step": 95990
    },
    {
      "dynamics_loss_avg": 0.049668948724865915,
      "epoch": 2.0971298217773436e-05,
      "step": 95990
    },
    {
      "action_loss_avg": 0.0012873775500338524,
      "epoch": 2.0971298217773436e-05,
      "step": 95990
    },
    {
      "epoch": 2.0971298217773436e-05,
      "step": 95990,
      "training_step_time": 7.506333589553833
    },
    {
      "epoch": 2.0972251892089843e-05,
      "model_forward_time": 1.777482509613037,
      "step": 95991
    },
    {
      "epoch": 2.0972251892089843e-05,
      "step": 95991,
      "training_step_time": 9.278218269348145
    },
    {
      "epoch": 2.097320556640625e-05,
      "model_forward_time": 1.8407618999481201,
      "step": 95992
    },
    {
      "epoch": 2.097320556640625e-05,
      "step": 95992,
      "training_step_time": 7.865858793258667
    },
    {
      "epoch": 2.0974159240722655e-05,
      "model_forward_time": 1.7666592597961426,
      "step": 95993
    },
    {
      "epoch": 2.0974159240722655e-05,
      "step": 95993,
      "training_step_time": 11.546815872192383
    },
    {
      "epoch": 2.0975112915039062e-05,
      "model_forward_time": 2.844306707382202,
      "step": 95994
    },
    {
      "epoch": 2.0975112915039062e-05,
      "step": 95994,
      "training_step_time": 8.664186716079712
    },
    {
      "epoch": 2.097606658935547e-05,
      "model_forward_time": 1.7609138488769531,
      "step": 95995
    },
    {
      "epoch": 2.097606658935547e-05,
      "step": 95995,
      "training_step_time": 7.719431638717651
    },
    {
      "epoch": 2.0977020263671874e-05,
      "model_forward_time": 1.3942842483520508,
      "step": 95996
    },
    {
      "epoch": 2.0977020263671874e-05,
      "step": 95996,
      "training_step_time": 7.560454845428467
    },
    {
      "epoch": 2.097797393798828e-05,
      "model_forward_time": 1.7710354328155518,
      "step": 95997
    },
    {
      "epoch": 2.097797393798828e-05,
      "step": 95997,
      "training_step_time": 7.320411205291748
    },
    {
      "epoch": 2.097892761230469e-05,
      "model_forward_time": 1.7713501453399658,
      "step": 95998
    },
    {
      "epoch": 2.097892761230469e-05,
      "step": 95998,
      "training_step_time": 7.044018030166626
    },
    {
      "epoch": 2.0979881286621093e-05,
      "model_forward_time": 1.808450698852539,
      "step": 95999
    },
    {
      "epoch": 2.0979881286621093e-05,
      "step": 95999,
      "training_step_time": 6.949949026107788
    },
    {
      "epoch": 2.09808349609375e-05,
      "grad_norm": 0.1369156837463379,
      "learning_rate": 4.3701464047649075e-08,
      "loss": 0.0546,
      "step": 96000
    },
    {
      "epoch": 2.09808349609375e-05,
      "model_forward_time": 1.8420789241790771,
      "step": 96000
    },
    {
      "dynamics_loss_avg": 0.07252906300127507,
      "epoch": 2.09808349609375e-05,
      "step": 96000
    },
    {
      "action_loss_avg": 0.0019823059032205493,
      "epoch": 2.09808349609375e-05,
      "step": 96000
    },
    {
      "epoch": 2.09808349609375e-05,
      "step": 96000,
      "training_step_time": 7.655145883560181
    },
    {
      "epoch": 2.0981788635253907e-05,
      "model_forward_time": 1.7435569763183594,
      "step": 96001
    },
    {
      "epoch": 2.0981788635253907e-05,
      "step": 96001,
      "training_step_time": 10.34910249710083
    },
    {
      "epoch": 2.098274230957031e-05,
      "model_forward_time": 3.0170915126800537,
      "step": 96002
    },
    {
      "epoch": 2.098274230957031e-05,
      "step": 96002,
      "training_step_time": 10.179724931716919
    },
    {
      "epoch": 2.098369598388672e-05,
      "model_forward_time": 1.7527880668640137,
      "step": 96003
    },
    {
      "epoch": 2.098369598388672e-05,
      "step": 96003,
      "training_step_time": 8.766493320465088
    },
    {
      "epoch": 2.0984649658203126e-05,
      "model_forward_time": 1.758134126663208,
      "step": 96004
    },
    {
      "epoch": 2.0984649658203126e-05,
      "step": 96004,
      "training_step_time": 8.424163818359375
    },
    {
      "epoch": 2.098560333251953e-05,
      "model_forward_time": 1.755659818649292,
      "step": 96005
    },
    {
      "epoch": 2.098560333251953e-05,
      "step": 96005,
      "training_step_time": 8.397929668426514
    },
    {
      "epoch": 2.0986557006835938e-05,
      "model_forward_time": 1.7605109214782715,
      "step": 96006
    },
    {
      "epoch": 2.0986557006835938e-05,
      "step": 96006,
      "training_step_time": 7.256150960922241
    },
    {
      "epoch": 2.0987510681152345e-05,
      "model_forward_time": 1.751826524734497,
      "step": 96007
    },
    {
      "epoch": 2.0987510681152345e-05,
      "step": 96007,
      "training_step_time": 6.729516267776489
    },
    {
      "epoch": 2.098846435546875e-05,
      "model_forward_time": 1.784550428390503,
      "step": 96008
    },
    {
      "epoch": 2.098846435546875e-05,
      "step": 96008,
      "training_step_time": 7.389123439788818
    },
    {
      "epoch": 2.0989418029785157e-05,
      "model_forward_time": 1.7614474296569824,
      "step": 96009
    },
    {
      "epoch": 2.0989418029785157e-05,
      "step": 96009,
      "training_step_time": 8.7209312915802
    },
    {
      "epoch": 2.0990371704101564e-05,
      "grad_norm": 0.10370945930480957,
      "learning_rate": 4.348360113234529e-08,
      "loss": 0.0533,
      "step": 96010
    },
    {
      "epoch": 2.0990371704101564e-05,
      "model_forward_time": 1.7482264041900635,
      "step": 96010
    },
    {
      "dynamics_loss_avg": 0.05808541513979435,
      "epoch": 2.0990371704101564e-05,
      "step": 96010
    },
    {
      "action_loss_avg": 0.0023275944229681047,
      "epoch": 2.0990371704101564e-05,
      "step": 96010
    },
    {
      "epoch": 2.0990371704101564e-05,
      "step": 96010,
      "training_step_time": 9.685961723327637
    },
    {
      "epoch": 2.0991325378417968e-05,
      "model_forward_time": 2.3687400817871094,
      "step": 96011
    },
    {
      "epoch": 2.0991325378417968e-05,
      "step": 96011,
      "training_step_time": 10.99082899093628
    },
    {
      "epoch": 2.0992279052734375e-05,
      "model_forward_time": 1.7374382019042969,
      "step": 96012
    },
    {
      "epoch": 2.0992279052734375e-05,
      "step": 96012,
      "training_step_time": 7.774522304534912
    },
    {
      "epoch": 2.0993232727050783e-05,
      "model_forward_time": 1.7618534564971924,
      "step": 96013
    },
    {
      "epoch": 2.0993232727050783e-05,
      "step": 96013,
      "training_step_time": 8.254560470581055
    },
    {
      "epoch": 2.0994186401367187e-05,
      "model_forward_time": 1.7714941501617432,
      "step": 96014
    },
    {
      "epoch": 2.0994186401367187e-05,
      "step": 96014,
      "training_step_time": 7.592684030532837
    },
    {
      "epoch": 2.0995140075683594e-05,
      "model_forward_time": 1.7565007209777832,
      "step": 96015
    },
    {
      "epoch": 2.0995140075683594e-05,
      "step": 96015,
      "training_step_time": 7.613804817199707
    },
    {
      "epoch": 2.099609375e-05,
      "model_forward_time": 1.7481589317321777,
      "step": 96016
    },
    {
      "epoch": 2.099609375e-05,
      "step": 96016,
      "training_step_time": 6.73986554145813
    },
    {
      "epoch": 2.0997047424316406e-05,
      "model_forward_time": 1.7547094821929932,
      "step": 96017
    },
    {
      "epoch": 2.0997047424316406e-05,
      "step": 96017,
      "training_step_time": 6.7905049324035645
    },
    {
      "epoch": 2.0998001098632813e-05,
      "model_forward_time": 1.7592692375183105,
      "step": 96018
    },
    {
      "epoch": 2.0998001098632813e-05,
      "step": 96018,
      "training_step_time": 7.687342405319214
    },
    {
      "epoch": 2.0998954772949217e-05,
      "model_forward_time": 1.771510124206543,
      "step": 96019
    },
    {
      "epoch": 2.0998954772949217e-05,
      "step": 96019,
      "training_step_time": 7.810509920120239
    },
    {
      "epoch": 2.0999908447265625e-05,
      "grad_norm": 0.09430850297212601,
      "learning_rate": 4.32662802542233e-08,
      "loss": 0.0529,
      "step": 96020
    },
    {
      "epoch": 2.0999908447265625e-05,
      "model_forward_time": 1.7431330680847168,
      "step": 96020
    },
    {
      "dynamics_loss_avg": 0.05751276463270187,
      "epoch": 2.0999908447265625e-05,
      "step": 96020
    },
    {
      "action_loss_avg": 0.0013536115642637015,
      "epoch": 2.0999908447265625e-05,
      "step": 96020
    },
    {
      "epoch": 2.0999908447265625e-05,
      "step": 96020,
      "training_step_time": 8.199748992919922
    },
    {
      "epoch": 2.1000862121582032e-05,
      "model_forward_time": 2.440303087234497,
      "step": 96021
    },
    {
      "epoch": 2.1000862121582032e-05,
      "step": 96021,
      "training_step_time": 9.386700630187988
    },
    {
      "epoch": 2.1001815795898436e-05,
      "model_forward_time": 1.7500102519989014,
      "step": 96022
    },
    {
      "epoch": 2.1001815795898436e-05,
      "step": 96022,
      "training_step_time": 8.873432397842407
    },
    {
      "epoch": 2.1002769470214843e-05,
      "model_forward_time": 1.7551648616790771,
      "step": 96023
    },
    {
      "epoch": 2.1002769470214843e-05,
      "step": 96023,
      "training_step_time": 8.620131492614746
    },
    {
      "epoch": 2.100372314453125e-05,
      "model_forward_time": 1.3730394840240479,
      "step": 96024
    },
    {
      "epoch": 2.100372314453125e-05,
      "step": 96024,
      "training_step_time": 7.645062685012817
    },
    {
      "epoch": 2.1004676818847655e-05,
      "model_forward_time": 1.7579867839813232,
      "step": 96025
    },
    {
      "epoch": 2.1004676818847655e-05,
      "step": 96025,
      "training_step_time": 6.822215557098389
    },
    {
      "epoch": 2.1005630493164062e-05,
      "model_forward_time": 1.7598795890808105,
      "step": 96026
    },
    {
      "epoch": 2.1005630493164062e-05,
      "step": 96026,
      "training_step_time": 6.842833757400513
    },
    {
      "epoch": 2.100658416748047e-05,
      "model_forward_time": 1.758420467376709,
      "step": 96027
    },
    {
      "epoch": 2.100658416748047e-05,
      "step": 96027,
      "training_step_time": 7.708985805511475
    },
    {
      "epoch": 2.1007537841796874e-05,
      "model_forward_time": 1.3923156261444092,
      "step": 96028
    },
    {
      "epoch": 2.1007537841796874e-05,
      "step": 96028,
      "training_step_time": 7.698378801345825
    },
    {
      "epoch": 2.100849151611328e-05,
      "model_forward_time": 1.760385513305664,
      "step": 96029
    },
    {
      "epoch": 2.100849151611328e-05,
      "step": 96029,
      "training_step_time": 8.730980157852173
    },
    {
      "epoch": 2.100944519042969e-05,
      "grad_norm": 0.08914683759212494,
      "learning_rate": 4.304950143704745e-08,
      "loss": 0.0548,
      "step": 96030
    },
    {
      "epoch": 2.100944519042969e-05,
      "model_forward_time": 2.430013418197632,
      "step": 96030
    },
    {
      "dynamics_loss_avg": 0.058975369296967985,
      "epoch": 2.100944519042969e-05,
      "step": 96030
    },
    {
      "action_loss_avg": 0.0010791656735818834,
      "epoch": 2.100944519042969e-05,
      "step": 96030
    },
    {
      "epoch": 2.100944519042969e-05,
      "step": 96030,
      "training_step_time": 12.125801801681519
    },
    {
      "epoch": 2.1010398864746093e-05,
      "model_forward_time": 1.7609202861785889,
      "step": 96031
    },
    {
      "epoch": 2.1010398864746093e-05,
      "step": 96031,
      "training_step_time": 7.801157236099243
    },
    {
      "epoch": 2.10113525390625e-05,
      "model_forward_time": 1.780322551727295,
      "step": 96032
    },
    {
      "epoch": 2.10113525390625e-05,
      "step": 96032,
      "training_step_time": 7.664866209030151
    },
    {
      "epoch": 2.1012306213378907e-05,
      "model_forward_time": 1.761117935180664,
      "step": 96033
    },
    {
      "epoch": 2.1012306213378907e-05,
      "step": 96033,
      "training_step_time": 7.577162981033325
    },
    {
      "epoch": 2.101325988769531e-05,
      "model_forward_time": 1.7587132453918457,
      "step": 96034
    },
    {
      "epoch": 2.101325988769531e-05,
      "step": 96034,
      "training_step_time": 6.807812452316284
    },
    {
      "epoch": 2.101421356201172e-05,
      "model_forward_time": 1.7572574615478516,
      "step": 96035
    },
    {
      "epoch": 2.101421356201172e-05,
      "step": 96035,
      "training_step_time": 6.813400745391846
    },
    {
      "epoch": 2.1015167236328126e-05,
      "model_forward_time": 1.7667582035064697,
      "step": 96036
    },
    {
      "epoch": 2.1015167236328126e-05,
      "step": 96036,
      "training_step_time": 8.434032440185547
    },
    {
      "epoch": 2.101612091064453e-05,
      "model_forward_time": 1.7536840438842773,
      "step": 96037
    },
    {
      "epoch": 2.101612091064453e-05,
      "step": 96037,
      "training_step_time": 9.30855679512024
    },
    {
      "epoch": 2.1017074584960938e-05,
      "model_forward_time": 1.737654447555542,
      "step": 96038
    },
    {
      "epoch": 2.1017074584960938e-05,
      "step": 96038,
      "training_step_time": 7.692015886306763
    },
    {
      "epoch": 2.1018028259277345e-05,
      "model_forward_time": 2.4404890537261963,
      "step": 96039
    },
    {
      "epoch": 2.1018028259277345e-05,
      "step": 96039,
      "training_step_time": 9.40923261642456
    },
    {
      "epoch": 2.101898193359375e-05,
      "grad_norm": 0.08197308331727982,
      "learning_rate": 4.283326470452653e-08,
      "loss": 0.0537,
      "step": 96040
    },
    {
      "epoch": 2.101898193359375e-05,
      "model_forward_time": 1.7483985424041748,
      "step": 96040
    },
    {
      "dynamics_loss_avg": 0.05491198841482401,
      "epoch": 2.101898193359375e-05,
      "step": 96040
    },
    {
      "action_loss_avg": 0.0010891150828683748,
      "epoch": 2.101898193359375e-05,
      "step": 96040
    },
    {
      "epoch": 2.101898193359375e-05,
      "step": 96040,
      "training_step_time": 8.36788010597229
    },
    {
      "epoch": 2.1019935607910157e-05,
      "model_forward_time": 1.77974271774292,
      "step": 96041
    },
    {
      "epoch": 2.1019935607910157e-05,
      "step": 96041,
      "training_step_time": 7.662115573883057
    },
    {
      "epoch": 2.1020889282226564e-05,
      "model_forward_time": 1.7602317333221436,
      "step": 96042
    },
    {
      "epoch": 2.1020889282226564e-05,
      "step": 96042,
      "training_step_time": 7.5701000690460205
    },
    {
      "epoch": 2.1021842956542968e-05,
      "model_forward_time": 1.766615867614746,
      "step": 96043
    },
    {
      "epoch": 2.1021842956542968e-05,
      "step": 96043,
      "training_step_time": 6.7986438274383545
    },
    {
      "epoch": 2.1022796630859375e-05,
      "model_forward_time": 1.7638695240020752,
      "step": 96044
    },
    {
      "epoch": 2.1022796630859375e-05,
      "step": 96044,
      "training_step_time": 6.758389711380005
    },
    {
      "epoch": 2.1023750305175783e-05,
      "model_forward_time": 1.7643282413482666,
      "step": 96045
    },
    {
      "epoch": 2.1023750305175783e-05,
      "step": 96045,
      "training_step_time": 7.346076965332031
    },
    {
      "epoch": 2.1024703979492187e-05,
      "model_forward_time": 1.7778468132019043,
      "step": 96046
    },
    {
      "epoch": 2.1024703979492187e-05,
      "step": 96046,
      "training_step_time": 7.658857583999634
    },
    {
      "epoch": 2.1025657653808594e-05,
      "model_forward_time": 1.7541780471801758,
      "step": 96047
    },
    {
      "epoch": 2.1025657653808594e-05,
      "step": 96047,
      "training_step_time": 8.489457130432129
    },
    {
      "epoch": 2.1026611328125e-05,
      "model_forward_time": 2.442492723464966,
      "step": 96048
    },
    {
      "epoch": 2.1026611328125e-05,
      "step": 96048,
      "training_step_time": 10.566246032714844
    },
    {
      "epoch": 2.1027565002441406e-05,
      "model_forward_time": 1.7657873630523682,
      "step": 96049
    },
    {
      "epoch": 2.1027565002441406e-05,
      "step": 96049,
      "training_step_time": 8.187796115875244
    },
    {
      "epoch": 2.1028518676757813e-05,
      "grad_norm": 0.1190403401851654,
      "learning_rate": 4.2617570080306645e-08,
      "loss": 0.0533,
      "step": 96050
    },
    {
      "epoch": 2.1028518676757813e-05,
      "model_forward_time": 1.7474350929260254,
      "step": 96050
    },
    {
      "dynamics_loss_avg": 0.044589667674154045,
      "epoch": 2.1028518676757813e-05,
      "step": 96050
    },
    {
      "action_loss_avg": 0.0009130343663855456,
      "epoch": 2.1028518676757813e-05,
      "step": 96050
    },
    {
      "epoch": 2.1028518676757813e-05,
      "step": 96050,
      "training_step_time": 7.646541118621826
    },
    {
      "epoch": 2.1029472351074217e-05,
      "model_forward_time": 1.9262688159942627,
      "step": 96051
    },
    {
      "epoch": 2.1029472351074217e-05,
      "step": 96051,
      "training_step_time": 8.309162855148315
    },
    {
      "epoch": 2.1030426025390625e-05,
      "model_forward_time": 1.896211862564087,
      "step": 96052
    },
    {
      "epoch": 2.1030426025390625e-05,
      "step": 96052,
      "training_step_time": 6.7864463329315186
    },
    {
      "epoch": 2.1031379699707032e-05,
      "model_forward_time": 1.9093022346496582,
      "step": 96053
    },
    {
      "epoch": 2.1031379699707032e-05,
      "step": 96053,
      "training_step_time": 6.771808385848999
    },
    {
      "epoch": 2.1032333374023436e-05,
      "model_forward_time": 1.557182788848877,
      "step": 96054
    },
    {
      "epoch": 2.1032333374023436e-05,
      "step": 96054,
      "training_step_time": 7.341413736343384
    },
    {
      "epoch": 2.1033287048339843e-05,
      "model_forward_time": 1.9048333168029785,
      "step": 96055
    },
    {
      "epoch": 2.1033287048339843e-05,
      "step": 96055,
      "training_step_time": 7.6424291133880615
    },
    {
      "epoch": 2.103424072265625e-05,
      "model_forward_time": 1.90385103225708,
      "step": 96056
    },
    {
      "epoch": 2.103424072265625e-05,
      "step": 96056,
      "training_step_time": 8.607738256454468
    },
    {
      "epoch": 2.1035194396972655e-05,
      "model_forward_time": 1.908475637435913,
      "step": 96057
    },
    {
      "epoch": 2.1035194396972655e-05,
      "step": 96057,
      "training_step_time": 9.269289255142212
    },
    {
      "epoch": 2.1036148071289062e-05,
      "model_forward_time": 2.5194694995880127,
      "step": 96058
    },
    {
      "epoch": 2.1036148071289062e-05,
      "step": 96058,
      "training_step_time": 9.514668941497803
    },
    {
      "epoch": 2.103710174560547e-05,
      "model_forward_time": 1.9096715450286865,
      "step": 96059
    },
    {
      "epoch": 2.103710174560547e-05,
      "step": 96059,
      "training_step_time": 8.514857053756714
    },
    {
      "epoch": 2.1038055419921874e-05,
      "grad_norm": 0.09850993752479553,
      "learning_rate": 4.2402417587975585e-08,
      "loss": 0.0526,
      "step": 96060
    },
    {
      "epoch": 2.1038055419921874e-05,
      "model_forward_time": 1.9214441776275635,
      "step": 96060
    },
    {
      "dynamics_loss_avg": 0.05457443976774812,
      "epoch": 2.1038055419921874e-05,
      "step": 96060
    },
    {
      "action_loss_avg": 0.0009872735681710765,
      "epoch": 2.1038055419921874e-05,
      "step": 96060
    },
    {
      "epoch": 2.1038055419921874e-05,
      "step": 96060,
      "training_step_time": 7.563255786895752
    },
    {
      "epoch": 2.103900909423828e-05,
      "model_forward_time": 1.9158942699432373,
      "step": 96061
    },
    {
      "epoch": 2.103900909423828e-05,
      "step": 96061,
      "training_step_time": 7.5681798458099365
    },
    {
      "epoch": 2.103996276855469e-05,
      "model_forward_time": 1.914820671081543,
      "step": 96062
    },
    {
      "epoch": 2.103996276855469e-05,
      "step": 96062,
      "training_step_time": 6.816089391708374
    },
    {
      "epoch": 2.1040916442871093e-05,
      "model_forward_time": 1.9134278297424316,
      "step": 96063
    },
    {
      "epoch": 2.1040916442871093e-05,
      "step": 96063,
      "training_step_time": 6.778776168823242
    },
    {
      "epoch": 2.10418701171875e-05,
      "model_forward_time": 1.920900583267212,
      "step": 96064
    },
    {
      "epoch": 2.10418701171875e-05,
      "step": 96064,
      "training_step_time": 8.237659931182861
    },
    {
      "epoch": 2.1042823791503907e-05,
      "model_forward_time": 1.9129881858825684,
      "step": 96065
    },
    {
      "epoch": 2.1042823791503907e-05,
      "step": 96065,
      "training_step_time": 8.698712348937988
    },
    {
      "epoch": 2.104377746582031e-05,
      "model_forward_time": 1.9074137210845947,
      "step": 96066
    },
    {
      "epoch": 2.104377746582031e-05,
      "step": 96066,
      "training_step_time": 8.632106065750122
    },
    {
      "epoch": 2.104473114013672e-05,
      "model_forward_time": 2.503371238708496,
      "step": 96067
    },
    {
      "epoch": 2.104473114013672e-05,
      "step": 96067,
      "training_step_time": 8.3843252658844
    },
    {
      "epoch": 2.1045684814453126e-05,
      "model_forward_time": 1.545177936553955,
      "step": 96068
    },
    {
      "epoch": 2.1045684814453126e-05,
      "step": 96068,
      "training_step_time": 8.240199565887451
    },
    {
      "epoch": 2.104663848876953e-05,
      "model_forward_time": 1.9105119705200195,
      "step": 96069
    },
    {
      "epoch": 2.104663848876953e-05,
      "step": 96069,
      "training_step_time": 8.188275575637817
    },
    {
      "epoch": 2.1047592163085938e-05,
      "grad_norm": 0.07472945749759674,
      "learning_rate": 4.2187807251062863e-08,
      "loss": 0.053,
      "step": 96070
    },
    {
      "epoch": 2.1047592163085938e-05,
      "model_forward_time": 1.9085519313812256,
      "step": 96070
    },
    {
      "dynamics_loss_avg": 0.04869256187230349,
      "epoch": 2.1047592163085938e-05,
      "step": 96070
    },
    {
      "action_loss_avg": 0.0010580397094599903,
      "epoch": 2.1047592163085938e-05,
      "step": 96070
    },
    {
      "epoch": 2.1047592163085938e-05,
      "step": 96070,
      "training_step_time": 7.64844822883606
    },
    {
      "epoch": 2.1048545837402345e-05,
      "model_forward_time": 1.9128811359405518,
      "step": 96071
    },
    {
      "epoch": 2.1048545837402345e-05,
      "step": 96071,
      "training_step_time": 6.741898775100708
    },
    {
      "epoch": 2.104949951171875e-05,
      "model_forward_time": 1.915982723236084,
      "step": 96072
    },
    {
      "epoch": 2.104949951171875e-05,
      "step": 96072,
      "training_step_time": 6.755181789398193
    },
    {
      "epoch": 2.1050453186035157e-05,
      "model_forward_time": 1.5514435768127441,
      "step": 96073
    },
    {
      "epoch": 2.1050453186035157e-05,
      "step": 96073,
      "training_step_time": 7.549461126327515
    },
    {
      "epoch": 2.1051406860351564e-05,
      "model_forward_time": 1.928191900253296,
      "step": 96074
    },
    {
      "epoch": 2.1051406860351564e-05,
      "step": 96074,
      "training_step_time": 8.556192398071289
    },
    {
      "epoch": 2.1052360534667968e-05,
      "model_forward_time": 1.9256787300109863,
      "step": 96075
    },
    {
      "epoch": 2.1052360534667968e-05,
      "step": 96075,
      "training_step_time": 7.719389200210571
    },
    {
      "epoch": 2.1053314208984375e-05,
      "model_forward_time": 2.5454647541046143,
      "step": 96076
    },
    {
      "epoch": 2.1053314208984375e-05,
      "step": 96076,
      "training_step_time": 8.95479154586792
    },
    {
      "epoch": 2.1054267883300783e-05,
      "model_forward_time": 1.9147331714630127,
      "step": 96077
    },
    {
      "epoch": 2.1054267883300783e-05,
      "step": 96077,
      "training_step_time": 7.795003890991211
    },
    {
      "epoch": 2.1055221557617187e-05,
      "model_forward_time": 1.9182298183441162,
      "step": 96078
    },
    {
      "epoch": 2.1055221557617187e-05,
      "step": 96078,
      "training_step_time": 8.49494481086731
    },
    {
      "epoch": 2.1056175231933594e-05,
      "model_forward_time": 1.9138391017913818,
      "step": 96079
    },
    {
      "epoch": 2.1056175231933594e-05,
      "step": 96079,
      "training_step_time": 7.6045732498168945
    },
    {
      "epoch": 2.105712890625e-05,
      "grad_norm": 0.18750594556331635,
      "learning_rate": 4.197373909303748e-08,
      "loss": 0.0531,
      "step": 96080
    },
    {
      "epoch": 2.105712890625e-05,
      "model_forward_time": 1.929999828338623,
      "step": 96080
    },
    {
      "dynamics_loss_avg": 0.04784134067595005,
      "epoch": 2.105712890625e-05,
      "step": 96080
    },
    {
      "action_loss_avg": 0.0011644718353636563,
      "epoch": 2.105712890625e-05,
      "step": 96080
    },
    {
      "epoch": 2.105712890625e-05,
      "step": 96080,
      "training_step_time": 6.735278129577637
    },
    {
      "epoch": 2.1058082580566406e-05,
      "model_forward_time": 1.552004337310791,
      "step": 96081
    },
    {
      "epoch": 2.1058082580566406e-05,
      "step": 96081,
      "training_step_time": 6.769362926483154
    },
    {
      "epoch": 2.1059036254882813e-05,
      "model_forward_time": 1.917433261871338,
      "step": 96082
    },
    {
      "epoch": 2.1059036254882813e-05,
      "step": 96082,
      "training_step_time": 7.449590444564819
    },
    {
      "epoch": 2.1059989929199217e-05,
      "model_forward_time": 1.9063091278076172,
      "step": 96083
    },
    {
      "epoch": 2.1059989929199217e-05,
      "step": 96083,
      "training_step_time": 9.249453067779541
    },
    {
      "epoch": 2.1060943603515625e-05,
      "model_forward_time": 1.9224956035614014,
      "step": 96084
    },
    {
      "epoch": 2.1060943603515625e-05,
      "step": 96084,
      "training_step_time": 7.718027114868164
    },
    {
      "epoch": 2.1061897277832032e-05,
      "model_forward_time": 1.9112420082092285,
      "step": 96085
    },
    {
      "epoch": 2.1061897277832032e-05,
      "step": 96085,
      "training_step_time": 9.543967485427856
    },
    {
      "epoch": 2.1062850952148436e-05,
      "model_forward_time": 2.5366878509521484,
      "step": 96086
    },
    {
      "epoch": 2.1062850952148436e-05,
      "step": 96086,
      "training_step_time": 9.845141172409058
    },
    {
      "epoch": 2.1063804626464844e-05,
      "model_forward_time": 1.904655933380127,
      "step": 96087
    },
    {
      "epoch": 2.1063804626464844e-05,
      "step": 96087,
      "training_step_time": 7.647854804992676
    },
    {
      "epoch": 2.106475830078125e-05,
      "model_forward_time": 1.531195878982544,
      "step": 96088
    },
    {
      "epoch": 2.106475830078125e-05,
      "step": 96088,
      "training_step_time": 7.563989877700806
    },
    {
      "epoch": 2.1065711975097655e-05,
      "model_forward_time": 1.1781747341156006,
      "step": 96089
    },
    {
      "epoch": 2.1065711975097655e-05,
      "step": 96089,
      "training_step_time": 7.399935483932495
    },
    {
      "epoch": 2.1066665649414062e-05,
      "grad_norm": 0.09538989514112473,
      "learning_rate": 4.17602131373096e-08,
      "loss": 0.0523,
      "step": 96090
    },
    {
      "epoch": 2.1066665649414062e-05,
      "model_forward_time": 1.5448293685913086,
      "step": 96090
    },
    {
      "dynamics_loss_avg": 0.047029143664985895,
      "epoch": 2.1066665649414062e-05,
      "step": 96090
    },
    {
      "action_loss_avg": 0.0010795469803269952,
      "epoch": 2.1066665649414062e-05,
      "step": 96090
    },
    {
      "epoch": 2.1066665649414062e-05,
      "step": 96090,
      "training_step_time": 6.751283645629883
    },
    {
      "epoch": 2.106761932373047e-05,
      "model_forward_time": 1.912332534790039,
      "step": 96091
    },
    {
      "epoch": 2.106761932373047e-05,
      "step": 96091,
      "training_step_time": 7.520313739776611
    },
    {
      "epoch": 2.1068572998046874e-05,
      "model_forward_time": 1.9174160957336426,
      "step": 96092
    },
    {
      "epoch": 2.1068572998046874e-05,
      "step": 96092,
      "training_step_time": 7.63204288482666
    },
    {
      "epoch": 2.106952667236328e-05,
      "model_forward_time": 1.888653039932251,
      "step": 96093
    },
    {
      "epoch": 2.106952667236328e-05,
      "step": 96093,
      "training_step_time": 10.124425888061523
    },
    {
      "epoch": 2.107048034667969e-05,
      "model_forward_time": 1.872596263885498,
      "step": 96094
    },
    {
      "epoch": 2.107048034667969e-05,
      "step": 96094,
      "training_step_time": 8.586034297943115
    },
    {
      "epoch": 2.1071434020996093e-05,
      "model_forward_time": 2.4983444213867188,
      "step": 96095
    },
    {
      "epoch": 2.1071434020996093e-05,
      "step": 96095,
      "training_step_time": 10.3167724609375
    },
    {
      "epoch": 2.10723876953125e-05,
      "model_forward_time": 1.9128904342651367,
      "step": 96096
    },
    {
      "epoch": 2.10723876953125e-05,
      "step": 96096,
      "training_step_time": 8.554595708847046
    },
    {
      "epoch": 2.1073341369628907e-05,
      "model_forward_time": 1.5364527702331543,
      "step": 96097
    },
    {
      "epoch": 2.1073341369628907e-05,
      "step": 96097,
      "training_step_time": 8.391721248626709
    },
    {
      "epoch": 2.107429504394531e-05,
      "model_forward_time": 1.9029278755187988,
      "step": 96098
    },
    {
      "epoch": 2.107429504394531e-05,
      "step": 96098,
      "training_step_time": 7.579194784164429
    },
    {
      "epoch": 2.107524871826172e-05,
      "model_forward_time": 1.9092838764190674,
      "step": 96099
    },
    {
      "epoch": 2.107524871826172e-05,
      "step": 96099,
      "training_step_time": 6.853179216384888
    },
    {
      "epoch": 2.1076202392578126e-05,
      "grad_norm": 0.09134212881326675,
      "learning_rate": 4.154722940723055e-08,
      "loss": 0.054,
      "step": 96100
    },
    {
      "epoch": 2.1076202392578126e-05,
      "model_forward_time": 1.9176318645477295,
      "step": 96100
    },
    {
      "dynamics_loss_avg": 0.052847592253237964,
      "epoch": 2.1076202392578126e-05,
      "step": 96100
    },
    {
      "action_loss_avg": 0.0011441836482845246,
      "epoch": 2.1076202392578126e-05,
      "step": 96100
    },
    {
      "epoch": 2.1076202392578126e-05,
      "step": 96100,
      "training_step_time": 7.257104873657227
    },
    {
      "epoch": 2.107715606689453e-05,
      "model_forward_time": 1.904916763305664,
      "step": 96101
    },
    {
      "epoch": 2.107715606689453e-05,
      "step": 96101,
      "training_step_time": 8.512500286102295
    },
    {
      "epoch": 2.1078109741210938e-05,
      "model_forward_time": 1.9129209518432617,
      "step": 96102
    },
    {
      "epoch": 2.1078109741210938e-05,
      "step": 96102,
      "training_step_time": 7.680256128311157
    },
    {
      "epoch": 2.1079063415527345e-05,
      "model_forward_time": 1.9099230766296387,
      "step": 96103
    },
    {
      "epoch": 2.1079063415527345e-05,
      "step": 96103,
      "training_step_time": 8.712034463882446
    },
    {
      "epoch": 2.108001708984375e-05,
      "model_forward_time": 1.9197795391082764,
      "step": 96104
    },
    {
      "epoch": 2.108001708984375e-05,
      "step": 96104,
      "training_step_time": 9.154014348983765
    },
    {
      "epoch": 2.1080970764160157e-05,
      "model_forward_time": 2.5326077938079834,
      "step": 96105
    },
    {
      "epoch": 2.1080970764160157e-05,
      "step": 96105,
      "training_step_time": 7.701615810394287
    },
    {
      "epoch": 2.1081924438476564e-05,
      "model_forward_time": 1.921858787536621,
      "step": 96106
    },
    {
      "epoch": 2.1081924438476564e-05,
      "step": 96106,
      "training_step_time": 7.7658302783966064
    },
    {
      "epoch": 2.1082878112792968e-05,
      "model_forward_time": 1.9159305095672607,
      "step": 96107
    },
    {
      "epoch": 2.1082878112792968e-05,
      "step": 96107,
      "training_step_time": 7.734618663787842
    },
    {
      "epoch": 2.1083831787109376e-05,
      "model_forward_time": 1.9084844589233398,
      "step": 96108
    },
    {
      "epoch": 2.1083831787109376e-05,
      "step": 96108,
      "training_step_time": 6.971142530441284
    },
    {
      "epoch": 2.1084785461425783e-05,
      "model_forward_time": 1.9156887531280518,
      "step": 96109
    },
    {
      "epoch": 2.1084785461425783e-05,
      "step": 96109,
      "training_step_time": 6.888689041137695
    },
    {
      "epoch": 2.1085739135742187e-05,
      "grad_norm": 0.08015744388103485,
      "learning_rate": 4.133478792609114e-08,
      "loss": 0.0518,
      "step": 96110
    },
    {
      "epoch": 2.1085739135742187e-05,
      "model_forward_time": 1.9079627990722656,
      "step": 96110
    },
    {
      "dynamics_loss_avg": 0.05143958795815706,
      "epoch": 2.1085739135742187e-05,
      "step": 96110
    },
    {
      "action_loss_avg": 0.000845359984668903,
      "epoch": 2.1085739135742187e-05,
      "step": 96110
    },
    {
      "epoch": 2.1085739135742187e-05,
      "step": 96110,
      "training_step_time": 7.7204084396362305
    },
    {
      "epoch": 2.1086692810058594e-05,
      "model_forward_time": 1.9100704193115234,
      "step": 96111
    },
    {
      "epoch": 2.1086692810058594e-05,
      "step": 96111,
      "training_step_time": 7.622708082199097
    },
    {
      "epoch": 2.1087646484375e-05,
      "model_forward_time": 1.9130940437316895,
      "step": 96112
    },
    {
      "epoch": 2.1087646484375e-05,
      "step": 96112,
      "training_step_time": 7.7905426025390625
    },
    {
      "epoch": 2.1088600158691406e-05,
      "model_forward_time": 1.9021615982055664,
      "step": 96113
    },
    {
      "epoch": 2.1088600158691406e-05,
      "step": 96113,
      "training_step_time": 8.781880140304565
    },
    {
      "epoch": 2.1089553833007813e-05,
      "model_forward_time": 2.5150341987609863,
      "step": 96114
    },
    {
      "epoch": 2.1089553833007813e-05,
      "step": 96114,
      "training_step_time": 7.832442760467529
    },
    {
      "epoch": 2.1090507507324217e-05,
      "model_forward_time": 1.9095556735992432,
      "step": 96115
    },
    {
      "epoch": 2.1090507507324217e-05,
      "step": 96115,
      "training_step_time": 7.611316204071045
    },
    {
      "epoch": 2.1091461181640625e-05,
      "model_forward_time": 1.9082815647125244,
      "step": 96116
    },
    {
      "epoch": 2.1091461181640625e-05,
      "step": 96116,
      "training_step_time": 7.579763412475586
    },
    {
      "epoch": 2.1092414855957032e-05,
      "model_forward_time": 1.9059386253356934,
      "step": 96117
    },
    {
      "epoch": 2.1092414855957032e-05,
      "step": 96117,
      "training_step_time": 7.283924102783203
    },
    {
      "epoch": 2.1093368530273436e-05,
      "model_forward_time": 1.9006404876708984,
      "step": 96118
    },
    {
      "epoch": 2.1093368530273436e-05,
      "step": 96118,
      "training_step_time": 6.850831508636475
    },
    {
      "epoch": 2.1094322204589844e-05,
      "model_forward_time": 1.9074604511260986,
      "step": 96119
    },
    {
      "epoch": 2.1094322204589844e-05,
      "step": 96119,
      "training_step_time": 8.537338733673096
    },
    {
      "epoch": 2.109527587890625e-05,
      "grad_norm": 0.09803502261638641,
      "learning_rate": 4.1122888717124445e-08,
      "loss": 0.0532,
      "step": 96120
    },
    {
      "epoch": 2.109527587890625e-05,
      "model_forward_time": 1.9195971488952637,
      "step": 96120
    },
    {
      "dynamics_loss_avg": 0.0639512412250042,
      "epoch": 2.109527587890625e-05,
      "step": 96120
    },
    {
      "action_loss_avg": 0.002204918535426259,
      "epoch": 2.109527587890625e-05,
      "step": 96120
    },
    {
      "epoch": 2.109527587890625e-05,
      "step": 96120,
      "training_step_time": 7.601588249206543
    },
    {
      "epoch": 2.1096229553222655e-05,
      "model_forward_time": 1.9171030521392822,
      "step": 96121
    },
    {
      "epoch": 2.1096229553222655e-05,
      "step": 96121,
      "training_step_time": 8.606529951095581
    },
    {
      "epoch": 2.1097183227539062e-05,
      "model_forward_time": 1.902925968170166,
      "step": 96122
    },
    {
      "epoch": 2.1097183227539062e-05,
      "step": 96122,
      "training_step_time": 7.728731155395508
    },
    {
      "epoch": 2.109813690185547e-05,
      "model_forward_time": 2.552494764328003,
      "step": 96123
    },
    {
      "epoch": 2.109813690185547e-05,
      "step": 96123,
      "training_step_time": 8.77851676940918
    },
    {
      "epoch": 2.1099090576171874e-05,
      "model_forward_time": 1.9099705219268799,
      "step": 96124
    },
    {
      "epoch": 2.1099090576171874e-05,
      "step": 96124,
      "training_step_time": 7.659803867340088
    },
    {
      "epoch": 2.110004425048828e-05,
      "model_forward_time": 1.919490098953247,
      "step": 96125
    },
    {
      "epoch": 2.110004425048828e-05,
      "step": 96125,
      "training_step_time": 7.6886491775512695
    },
    {
      "epoch": 2.110099792480469e-05,
      "model_forward_time": 1.9258034229278564,
      "step": 96126
    },
    {
      "epoch": 2.110099792480469e-05,
      "step": 96126,
      "training_step_time": 7.484800100326538
    },
    {
      "epoch": 2.1101951599121093e-05,
      "model_forward_time": 1.9009044170379639,
      "step": 96127
    },
    {
      "epoch": 2.1101951599121093e-05,
      "step": 96127,
      "training_step_time": 6.795391321182251
    },
    {
      "epoch": 2.11029052734375e-05,
      "model_forward_time": 1.9080734252929688,
      "step": 96128
    },
    {
      "epoch": 2.11029052734375e-05,
      "step": 96128,
      "training_step_time": 7.72674298286438
    },
    {
      "epoch": 2.1103858947753908e-05,
      "model_forward_time": 1.5453684329986572,
      "step": 96129
    },
    {
      "epoch": 2.1103858947753908e-05,
      "step": 96129,
      "training_step_time": 7.679150342941284
    },
    {
      "epoch": 2.110481262207031e-05,
      "grad_norm": 0.08947516977787018,
      "learning_rate": 4.091153180350249e-08,
      "loss": 0.0544,
      "step": 96130
    },
    {
      "epoch": 2.110481262207031e-05,
      "model_forward_time": 1.9163970947265625,
      "step": 96130
    },
    {
      "dynamics_loss_avg": 0.05635940246284008,
      "epoch": 2.110481262207031e-05,
      "step": 96130
    },
    {
      "action_loss_avg": 0.0012228459148900584,
      "epoch": 2.110481262207031e-05,
      "step": 96130
    },
    {
      "epoch": 2.110481262207031e-05,
      "step": 96130,
      "training_step_time": 9.351829290390015
    },
    {
      "epoch": 2.110576629638672e-05,
      "model_forward_time": 1.9076833724975586,
      "step": 96131
    },
    {
      "epoch": 2.110576629638672e-05,
      "step": 96131,
      "training_step_time": 10.363102912902832
    },
    {
      "epoch": 2.1106719970703126e-05,
      "model_forward_time": 2.5125818252563477,
      "step": 96132
    },
    {
      "epoch": 2.1106719970703126e-05,
      "step": 96132,
      "training_step_time": 7.9301917552948
    },
    {
      "epoch": 2.110767364501953e-05,
      "model_forward_time": 1.9059665203094482,
      "step": 96133
    },
    {
      "epoch": 2.110767364501953e-05,
      "step": 96133,
      "training_step_time": 7.766354322433472
    },
    {
      "epoch": 2.1108627319335938e-05,
      "model_forward_time": 1.9227969646453857,
      "step": 96134
    },
    {
      "epoch": 2.1108627319335938e-05,
      "step": 96134,
      "training_step_time": 7.723233938217163
    },
    {
      "epoch": 2.1109580993652345e-05,
      "model_forward_time": 1.9255826473236084,
      "step": 96135
    },
    {
      "epoch": 2.1109580993652345e-05,
      "step": 96135,
      "training_step_time": 7.495000123977661
    },
    {
      "epoch": 2.111053466796875e-05,
      "model_forward_time": 1.9188997745513916,
      "step": 96136
    },
    {
      "epoch": 2.111053466796875e-05,
      "step": 96136,
      "training_step_time": 6.916136980056763
    },
    {
      "epoch": 2.1111488342285157e-05,
      "model_forward_time": 1.9093122482299805,
      "step": 96137
    },
    {
      "epoch": 2.1111488342285157e-05,
      "step": 96137,
      "training_step_time": 7.659271240234375
    },
    {
      "epoch": 2.1112442016601564e-05,
      "model_forward_time": 1.9057512283325195,
      "step": 96138
    },
    {
      "epoch": 2.1112442016601564e-05,
      "step": 96138,
      "training_step_time": 7.782754421234131
    },
    {
      "epoch": 2.1113395690917968e-05,
      "model_forward_time": 1.919041395187378,
      "step": 96139
    },
    {
      "epoch": 2.1113395690917968e-05,
      "step": 96139,
      "training_step_time": 8.552184820175171
    },
    {
      "epoch": 2.1114349365234376e-05,
      "grad_norm": 0.09918457269668579,
      "learning_rate": 4.070071720834012e-08,
      "loss": 0.0522,
      "step": 96140
    },
    {
      "epoch": 2.1114349365234376e-05,
      "model_forward_time": 1.9133820533752441,
      "step": 96140
    },
    {
      "dynamics_loss_avg": 0.0481804565526545,
      "epoch": 2.1114349365234376e-05,
      "step": 96140
    },
    {
      "action_loss_avg": 0.0011637603864073753,
      "epoch": 2.1114349365234376e-05,
      "step": 96140
    },
    {
      "epoch": 2.1114349365234376e-05,
      "step": 96140,
      "training_step_time": 9.276950597763062
    },
    {
      "epoch": 2.111530303955078e-05,
      "model_forward_time": 1.922178030014038,
      "step": 96141
    },
    {
      "epoch": 2.111530303955078e-05,
      "step": 96141,
      "training_step_time": 9.713571310043335
    },
    {
      "epoch": 2.1116256713867187e-05,
      "model_forward_time": 2.521980047225952,
      "step": 96142
    },
    {
      "epoch": 2.1116256713867187e-05,
      "step": 96142,
      "training_step_time": 7.663591384887695
    },
    {
      "epoch": 2.1117210388183594e-05,
      "model_forward_time": 1.928889274597168,
      "step": 96143
    },
    {
      "epoch": 2.1117210388183594e-05,
      "step": 96143,
      "training_step_time": 7.763782501220703
    },
    {
      "epoch": 2.11181640625e-05,
      "model_forward_time": 1.9062371253967285,
      "step": 96144
    },
    {
      "epoch": 2.11181640625e-05,
      "step": 96144,
      "training_step_time": 7.595499277114868
    },
    {
      "epoch": 2.1119117736816406e-05,
      "model_forward_time": 1.9233834743499756,
      "step": 96145
    },
    {
      "epoch": 2.1119117736816406e-05,
      "step": 96145,
      "training_step_time": 6.808035135269165
    },
    {
      "epoch": 2.1120071411132813e-05,
      "model_forward_time": 1.899874210357666,
      "step": 96146
    },
    {
      "epoch": 2.1120071411132813e-05,
      "step": 96146,
      "training_step_time": 7.558977127075195
    },
    {
      "epoch": 2.1121025085449217e-05,
      "model_forward_time": 1.9250178337097168,
      "step": 96147
    },
    {
      "epoch": 2.1121025085449217e-05,
      "step": 96147,
      "training_step_time": 7.6995155811309814
    },
    {
      "epoch": 2.1121978759765625e-05,
      "model_forward_time": 1.9218504428863525,
      "step": 96148
    },
    {
      "epoch": 2.1121978759765625e-05,
      "step": 96148,
      "training_step_time": 7.701809406280518
    },
    {
      "epoch": 2.1122932434082032e-05,
      "model_forward_time": 1.91300368309021,
      "step": 96149
    },
    {
      "epoch": 2.1122932434082032e-05,
      "step": 96149,
      "training_step_time": 9.252081871032715
    },
    {
      "epoch": 2.1123886108398436e-05,
      "grad_norm": 0.08723840117454529,
      "learning_rate": 4.049044495469112e-08,
      "loss": 0.0541,
      "step": 96150
    },
    {
      "epoch": 2.1123886108398436e-05,
      "model_forward_time": 1.9121267795562744,
      "step": 96150
    },
    {
      "dynamics_loss_avg": 0.05356130851432681,
      "epoch": 2.1123886108398436e-05,
      "step": 96150
    },
    {
      "action_loss_avg": 0.0018839955475414172,
      "epoch": 2.1123886108398436e-05,
      "step": 96150
    },
    {
      "epoch": 2.1123886108398436e-05,
      "step": 96150,
      "training_step_time": 8.778876066207886
    },
    {
      "epoch": 2.1124839782714844e-05,
      "model_forward_time": 2.5179102420806885,
      "step": 96151
    },
    {
      "epoch": 2.1124839782714844e-05,
      "step": 96151,
      "training_step_time": 7.822223424911499
    },
    {
      "epoch": 2.112579345703125e-05,
      "model_forward_time": 1.9254841804504395,
      "step": 96152
    },
    {
      "epoch": 2.112579345703125e-05,
      "step": 96152,
      "training_step_time": 7.646691560745239
    },
    {
      "epoch": 2.1126747131347655e-05,
      "model_forward_time": 1.924473524093628,
      "step": 96153
    },
    {
      "epoch": 2.1126747131347655e-05,
      "step": 96153,
      "training_step_time": 7.727164030075073
    },
    {
      "epoch": 2.1127700805664062e-05,
      "model_forward_time": 1.9120898246765137,
      "step": 96154
    },
    {
      "epoch": 2.1127700805664062e-05,
      "step": 96154,
      "training_step_time": 6.830641746520996
    },
    {
      "epoch": 2.112865447998047e-05,
      "model_forward_time": 1.9107747077941895,
      "step": 96155
    },
    {
      "epoch": 2.112865447998047e-05,
      "step": 96155,
      "training_step_time": 7.549012660980225
    },
    {
      "epoch": 2.1129608154296874e-05,
      "model_forward_time": 1.911557912826538,
      "step": 96156
    },
    {
      "epoch": 2.1129608154296874e-05,
      "step": 96156,
      "training_step_time": 8.51012659072876
    },
    {
      "epoch": 2.113056182861328e-05,
      "model_forward_time": 1.9204351902008057,
      "step": 96157
    },
    {
      "epoch": 2.113056182861328e-05,
      "step": 96157,
      "training_step_time": 7.669990539550781
    },
    {
      "epoch": 2.113151550292969e-05,
      "model_forward_time": 1.9289836883544922,
      "step": 96158
    },
    {
      "epoch": 2.113151550292969e-05,
      "step": 96158,
      "training_step_time": 9.412290573120117
    },
    {
      "epoch": 2.1132469177246093e-05,
      "model_forward_time": 1.9024522304534912,
      "step": 96159
    },
    {
      "epoch": 2.1132469177246093e-05,
      "step": 96159,
      "training_step_time": 9.697485208511353
    },
    {
      "epoch": 2.11334228515625e-05,
      "grad_norm": 0.11335782706737518,
      "learning_rate": 4.028071506554987e-08,
      "loss": 0.0543,
      "step": 96160
    },
    {
      "epoch": 2.11334228515625e-05,
      "model_forward_time": 2.5524508953094482,
      "step": 96160
    },
    {
      "dynamics_loss_avg": 0.060704157873988154,
      "epoch": 2.11334228515625e-05,
      "step": 96160
    },
    {
      "action_loss_avg": 0.0013071826309897006,
      "epoch": 2.11334228515625e-05,
      "step": 96160
    },
    {
      "epoch": 2.11334228515625e-05,
      "step": 96160,
      "training_step_time": 7.937562704086304
    },
    {
      "epoch": 2.1134376525878908e-05,
      "model_forward_time": 1.9194889068603516,
      "step": 96161
    },
    {
      "epoch": 2.1134376525878908e-05,
      "step": 96161,
      "training_step_time": 8.331360101699829
    },
    {
      "epoch": 2.113533020019531e-05,
      "model_forward_time": 1.9181411266326904,
      "step": 96162
    },
    {
      "epoch": 2.113533020019531e-05,
      "step": 96162,
      "training_step_time": 8.62630558013916
    },
    {
      "epoch": 2.113628387451172e-05,
      "model_forward_time": 1.9206976890563965,
      "step": 96163
    },
    {
      "epoch": 2.113628387451172e-05,
      "step": 96163,
      "training_step_time": 7.578134775161743
    },
    {
      "epoch": 2.1137237548828126e-05,
      "model_forward_time": 1.9119091033935547,
      "step": 96164
    },
    {
      "epoch": 2.1137237548828126e-05,
      "step": 96164,
      "training_step_time": 8.674825429916382
    },
    {
      "epoch": 2.113819122314453e-05,
      "model_forward_time": 1.9089317321777344,
      "step": 96165
    },
    {
      "epoch": 2.113819122314453e-05,
      "step": 96165,
      "training_step_time": 8.598849773406982
    },
    {
      "epoch": 2.1139144897460938e-05,
      "model_forward_time": 1.9126334190368652,
      "step": 96166
    },
    {
      "epoch": 2.1139144897460938e-05,
      "step": 96166,
      "training_step_time": 8.750585079193115
    },
    {
      "epoch": 2.1140098571777345e-05,
      "model_forward_time": 1.9138078689575195,
      "step": 96167
    },
    {
      "epoch": 2.1140098571777345e-05,
      "step": 96167,
      "training_step_time": 8.534897089004517
    },
    {
      "epoch": 2.114105224609375e-05,
      "model_forward_time": 1.9112327098846436,
      "step": 96168
    },
    {
      "epoch": 2.114105224609375e-05,
      "step": 96168,
      "training_step_time": 8.577244281768799
    },
    {
      "epoch": 2.1142005920410157e-05,
      "model_forward_time": 2.570500373840332,
      "step": 96169
    },
    {
      "epoch": 2.1142005920410157e-05,
      "step": 96169,
      "training_step_time": 8.33668041229248
    },
    {
      "epoch": 2.1142959594726564e-05,
      "grad_norm": 0.09642736613750458,
      "learning_rate": 4.0071527563853575e-08,
      "loss": 0.0542,
      "step": 96170
    },
    {
      "epoch": 2.1142959594726564e-05,
      "model_forward_time": 1.906949758529663,
      "step": 96170
    },
    {
      "dynamics_loss_avg": 0.04241200312972069,
      "epoch": 2.1142959594726564e-05,
      "step": 96170
    },
    {
      "action_loss_avg": 0.0014341978123411537,
      "epoch": 2.1142959594726564e-05,
      "step": 96170
    },
    {
      "epoch": 2.1142959594726564e-05,
      "step": 96170,
      "training_step_time": 9.392439603805542
    },
    {
      "epoch": 2.1143913269042968e-05,
      "model_forward_time": 1.9123613834381104,
      "step": 96171
    },
    {
      "epoch": 2.1143913269042968e-05,
      "step": 96171,
      "training_step_time": 8.269479751586914
    },
    {
      "epoch": 2.1144866943359376e-05,
      "model_forward_time": 1.9117505550384521,
      "step": 96172
    },
    {
      "epoch": 2.1144866943359376e-05,
      "step": 96172,
      "training_step_time": 7.441683292388916
    },
    {
      "epoch": 2.114582061767578e-05,
      "model_forward_time": 1.910377025604248,
      "step": 96173
    },
    {
      "epoch": 2.114582061767578e-05,
      "step": 96173,
      "training_step_time": 7.478470087051392
    },
    {
      "epoch": 2.1146774291992187e-05,
      "model_forward_time": 1.9194982051849365,
      "step": 96174
    },
    {
      "epoch": 2.1146774291992187e-05,
      "step": 96174,
      "training_step_time": 7.585416555404663
    },
    {
      "epoch": 2.1147727966308594e-05,
      "model_forward_time": 1.9219303131103516,
      "step": 96175
    },
    {
      "epoch": 2.1147727966308594e-05,
      "step": 96175,
      "training_step_time": 8.417486190795898
    },
    {
      "epoch": 2.1148681640625e-05,
      "model_forward_time": 1.9162673950195312,
      "step": 96176
    },
    {
      "epoch": 2.1148681640625e-05,
      "step": 96176,
      "training_step_time": 8.368069410324097
    },
    {
      "epoch": 2.1149635314941406e-05,
      "model_forward_time": 1.9147117137908936,
      "step": 96177
    },
    {
      "epoch": 2.1149635314941406e-05,
      "step": 96177,
      "training_step_time": 7.670994520187378
    },
    {
      "epoch": 2.1150588989257813e-05,
      "model_forward_time": 2.5272748470306396,
      "step": 96178
    },
    {
      "epoch": 2.1150588989257813e-05,
      "step": 96178,
      "training_step_time": 9.64512848854065
    },
    {
      "epoch": 2.1151542663574217e-05,
      "model_forward_time": 1.9166738986968994,
      "step": 96179
    },
    {
      "epoch": 2.1151542663574217e-05,
      "step": 96179,
      "training_step_time": 7.611945629119873
    },
    {
      "epoch": 2.1152496337890625e-05,
      "grad_norm": 0.07766815274953842,
      "learning_rate": 3.986288247247782e-08,
      "loss": 0.0545,
      "step": 96180
    },
    {
      "epoch": 2.1152496337890625e-05,
      "model_forward_time": 1.5496928691864014,
      "step": 96180
    },
    {
      "dynamics_loss_avg": 0.051687544025480744,
      "epoch": 2.1152496337890625e-05,
      "step": 96180
    },
    {
      "action_loss_avg": 0.0015633665432687848,
      "epoch": 2.1152496337890625e-05,
      "step": 96180
    },
    {
      "epoch": 2.1152496337890625e-05,
      "step": 96180,
      "training_step_time": 7.772002458572388
    },
    {
      "epoch": 2.1153450012207032e-05,
      "model_forward_time": 1.9113152027130127,
      "step": 96181
    },
    {
      "epoch": 2.1153450012207032e-05,
      "step": 96181,
      "training_step_time": 8.315311908721924
    },
    {
      "epoch": 2.1154403686523436e-05,
      "model_forward_time": 1.903367042541504,
      "step": 96182
    },
    {
      "epoch": 2.1154403686523436e-05,
      "step": 96182,
      "training_step_time": 7.494341850280762
    },
    {
      "epoch": 2.1155357360839844e-05,
      "model_forward_time": 1.9110631942749023,
      "step": 96183
    },
    {
      "epoch": 2.1155357360839844e-05,
      "step": 96183,
      "training_step_time": 7.560944318771362
    },
    {
      "epoch": 2.115631103515625e-05,
      "model_forward_time": 1.9156599044799805,
      "step": 96184
    },
    {
      "epoch": 2.115631103515625e-05,
      "step": 96184,
      "training_step_time": 7.656586170196533
    },
    {
      "epoch": 2.1157264709472655e-05,
      "model_forward_time": 1.9056525230407715,
      "step": 96185
    },
    {
      "epoch": 2.1157264709472655e-05,
      "step": 96185,
      "training_step_time": 8.377641201019287
    },
    {
      "epoch": 2.1158218383789062e-05,
      "model_forward_time": 1.9173986911773682,
      "step": 96186
    },
    {
      "epoch": 2.1158218383789062e-05,
      "step": 96186,
      "training_step_time": 8.690813064575195
    },
    {
      "epoch": 2.115917205810547e-05,
      "model_forward_time": 1.903374433517456,
      "step": 96187
    },
    {
      "epoch": 2.115917205810547e-05,
      "step": 96187,
      "training_step_time": 8.761591911315918
    },
    {
      "epoch": 2.1160125732421874e-05,
      "model_forward_time": 2.517704486846924,
      "step": 96188
    },
    {
      "epoch": 2.1160125732421874e-05,
      "step": 96188,
      "training_step_time": 8.66031527519226
    },
    {
      "epoch": 2.116107940673828e-05,
      "model_forward_time": 1.9118318557739258,
      "step": 96189
    },
    {
      "epoch": 2.116107940673828e-05,
      "step": 96189,
      "training_step_time": 7.819906949996948
    },
    {
      "epoch": 2.116203308105469e-05,
      "grad_norm": 0.12289305031299591,
      "learning_rate": 3.965477981423938e-08,
      "loss": 0.0534,
      "step": 96190
    },
    {
      "epoch": 2.116203308105469e-05,
      "model_forward_time": 1.907012939453125,
      "step": 96190
    },
    {
      "dynamics_loss_avg": 0.06447203475981951,
      "epoch": 2.116203308105469e-05,
      "step": 96190
    },
    {
      "action_loss_avg": 0.0021158078277949246,
      "epoch": 2.116203308105469e-05,
      "step": 96190
    },
    {
      "epoch": 2.116203308105469e-05,
      "step": 96190,
      "training_step_time": 7.53707218170166
    },
    {
      "epoch": 2.1162986755371093e-05,
      "model_forward_time": 1.9054081439971924,
      "step": 96191
    },
    {
      "epoch": 2.1162986755371093e-05,
      "step": 96191,
      "training_step_time": 7.460792303085327
    },
    {
      "epoch": 2.11639404296875e-05,
      "model_forward_time": 1.916550636291504,
      "step": 96192
    },
    {
      "epoch": 2.11639404296875e-05,
      "step": 96192,
      "training_step_time": 7.526521921157837
    },
    {
      "epoch": 2.1164894104003908e-05,
      "model_forward_time": 1.9082942008972168,
      "step": 96193
    },
    {
      "epoch": 2.1164894104003908e-05,
      "step": 96193,
      "training_step_time": 7.660748720169067
    },
    {
      "epoch": 2.116584777832031e-05,
      "model_forward_time": 1.9118564128875732,
      "step": 96194
    },
    {
      "epoch": 2.116584777832031e-05,
      "step": 96194,
      "training_step_time": 8.53964877128601
    },
    {
      "epoch": 2.116680145263672e-05,
      "model_forward_time": 1.921156883239746,
      "step": 96195
    },
    {
      "epoch": 2.116680145263672e-05,
      "step": 96195,
      "training_step_time": 9.593544244766235
    },
    {
      "epoch": 2.1167755126953126e-05,
      "model_forward_time": 1.5319373607635498,
      "step": 96196
    },
    {
      "epoch": 2.1167755126953126e-05,
      "step": 96196,
      "training_step_time": 8.825567483901978
    },
    {
      "epoch": 2.116870880126953e-05,
      "model_forward_time": 2.5328786373138428,
      "step": 96197
    },
    {
      "epoch": 2.116870880126953e-05,
      "step": 96197,
      "training_step_time": 7.746721982955933
    },
    {
      "epoch": 2.1169662475585938e-05,
      "model_forward_time": 1.9117152690887451,
      "step": 96198
    },
    {
      "epoch": 2.1169662475585938e-05,
      "step": 96198,
      "training_step_time": 7.8975419998168945
    },
    {
      "epoch": 2.1170616149902345e-05,
      "model_forward_time": 1.528597354888916,
      "step": 96199
    },
    {
      "epoch": 2.1170616149902345e-05,
      "step": 96199,
      "training_step_time": 7.913424253463745
    },
    {
      "epoch": 2.117156982421875e-05,
      "grad_norm": 0.12266682833433151,
      "learning_rate": 3.944721961189668e-08,
      "loss": 0.0545,
      "step": 96200
    },
    {
      "epoch": 2.117156982421875e-05,
      "model_forward_time": 1.9103899002075195,
      "step": 96200
    },
    {
      "dynamics_loss_avg": 0.0497936524450779,
      "epoch": 2.117156982421875e-05,
      "step": 96200
    },
    {
      "action_loss_avg": 0.0012792712426744402,
      "epoch": 2.117156982421875e-05,
      "step": 96200
    },
    {
      "epoch": 2.117156982421875e-05,
      "step": 96200,
      "training_step_time": 7.513319253921509
    },
    {
      "epoch": 2.1172523498535157e-05,
      "model_forward_time": 1.428727388381958,
      "step": 96201
    },
    {
      "epoch": 2.1172523498535157e-05,
      "step": 96201,
      "training_step_time": 7.3016510009765625
    },
    {
      "epoch": 2.1173477172851564e-05,
      "model_forward_time": 1.7459628582000732,
      "step": 96202
    },
    {
      "epoch": 2.1173477172851564e-05,
      "step": 96202,
      "training_step_time": 7.749890327453613
    },
    {
      "epoch": 2.1174430847167968e-05,
      "model_forward_time": 1.7443406581878662,
      "step": 96203
    },
    {
      "epoch": 2.1174430847167968e-05,
      "step": 96203,
      "training_step_time": 8.514780759811401
    },
    {
      "epoch": 2.1175384521484376e-05,
      "model_forward_time": 1.7607555389404297,
      "step": 96204
    },
    {
      "epoch": 2.1175384521484376e-05,
      "step": 96204,
      "training_step_time": 9.506045579910278
    },
    {
      "epoch": 2.117633819580078e-05,
      "model_forward_time": 1.7536900043487549,
      "step": 96205
    },
    {
      "epoch": 2.117633819580078e-05,
      "step": 96205,
      "training_step_time": 8.63735818862915
    },
    {
      "epoch": 2.1177291870117187e-05,
      "model_forward_time": 2.3739402294158936,
      "step": 96206
    },
    {
      "epoch": 2.1177291870117187e-05,
      "step": 96206,
      "training_step_time": 9.040947914123535
    },
    {
      "epoch": 2.1178245544433594e-05,
      "model_forward_time": 1.7533349990844727,
      "step": 96207
    },
    {
      "epoch": 2.1178245544433594e-05,
      "step": 96207,
      "training_step_time": 8.653835773468018
    },
    {
      "epoch": 2.117919921875e-05,
      "model_forward_time": 1.7659211158752441,
      "step": 96208
    },
    {
      "epoch": 2.117919921875e-05,
      "step": 96208,
      "training_step_time": 7.815086603164673
    },
    {
      "epoch": 2.1180152893066406e-05,
      "model_forward_time": 1.8210599422454834,
      "step": 96209
    },
    {
      "epoch": 2.1180152893066406e-05,
      "step": 96209,
      "training_step_time": 7.690767526626587
    },
    {
      "epoch": 2.1181106567382813e-05,
      "grad_norm": 0.11291371285915375,
      "learning_rate": 3.92402018881477e-08,
      "loss": 0.0554,
      "step": 96210
    },
    {
      "epoch": 2.1181106567382813e-05,
      "model_forward_time": 1.9012579917907715,
      "step": 96210
    },
    {
      "dynamics_loss_avg": 0.05637287702411413,
      "epoch": 2.1181106567382813e-05,
      "step": 96210
    },
    {
      "action_loss_avg": 0.0012916994164697826,
      "epoch": 2.1181106567382813e-05,
      "step": 96210
    },
    {
      "epoch": 2.1181106567382813e-05,
      "step": 96210,
      "training_step_time": 7.000426292419434
    },
    {
      "epoch": 2.1182060241699217e-05,
      "model_forward_time": 1.9138572216033936,
      "step": 96211
    },
    {
      "epoch": 2.1182060241699217e-05,
      "step": 96211,
      "training_step_time": 7.654815435409546
    },
    {
      "epoch": 2.1183013916015625e-05,
      "model_forward_time": 1.5395264625549316,
      "step": 96212
    },
    {
      "epoch": 2.1183013916015625e-05,
      "step": 96212,
      "training_step_time": 7.600201368331909
    },
    {
      "epoch": 2.1183967590332032e-05,
      "model_forward_time": 1.907973051071167,
      "step": 96213
    },
    {
      "epoch": 2.1183967590332032e-05,
      "step": 96213,
      "training_step_time": 9.369276762008667
    },
    {
      "epoch": 2.1184921264648436e-05,
      "model_forward_time": 1.9094667434692383,
      "step": 96214
    },
    {
      "epoch": 2.1184921264648436e-05,
      "step": 96214,
      "training_step_time": 7.862807989120483
    },
    {
      "epoch": 2.1185874938964844e-05,
      "model_forward_time": 1.9128613471984863,
      "step": 96215
    },
    {
      "epoch": 2.1185874938964844e-05,
      "step": 96215,
      "training_step_time": 8.548598527908325
    },
    {
      "epoch": 2.118682861328125e-05,
      "model_forward_time": 2.494131565093994,
      "step": 96216
    },
    {
      "epoch": 2.118682861328125e-05,
      "step": 96216,
      "training_step_time": 7.8173298835754395
    },
    {
      "epoch": 2.1187782287597655e-05,
      "model_forward_time": 1.8015351295471191,
      "step": 96217
    },
    {
      "epoch": 2.1187782287597655e-05,
      "step": 96217,
      "training_step_time": 8.900325536727905
    },
    {
      "epoch": 2.1188735961914062e-05,
      "model_forward_time": 1.8104963302612305,
      "step": 96218
    },
    {
      "epoch": 2.1188735961914062e-05,
      "step": 96218,
      "training_step_time": 8.32553505897522
    },
    {
      "epoch": 2.118968963623047e-05,
      "model_forward_time": 1.753598690032959,
      "step": 96219
    },
    {
      "epoch": 2.118968963623047e-05,
      "step": 96219,
      "training_step_time": 7.023279905319214
    },
    {
      "epoch": 2.1190643310546874e-05,
      "grad_norm": 0.11070408672094345,
      "learning_rate": 3.9033726665632096e-08,
      "loss": 0.0533,
      "step": 96220
    },
    {
      "epoch": 2.1190643310546874e-05,
      "model_forward_time": 1.7539763450622559,
      "step": 96220
    },
    {
      "dynamics_loss_avg": 0.056561459042131904,
      "epoch": 2.1190643310546874e-05,
      "step": 96220
    },
    {
      "action_loss_avg": 0.0013640699500683696,
      "epoch": 2.1190643310546874e-05,
      "step": 96220
    },
    {
      "epoch": 2.1190643310546874e-05,
      "step": 96220,
      "training_step_time": 7.603261470794678
    },
    {
      "epoch": 2.119159698486328e-05,
      "model_forward_time": 1.749556303024292,
      "step": 96221
    },
    {
      "epoch": 2.119159698486328e-05,
      "step": 96221,
      "training_step_time": 8.57204294204712
    },
    {
      "epoch": 2.119255065917969e-05,
      "model_forward_time": 1.754002571105957,
      "step": 96222
    },
    {
      "epoch": 2.119255065917969e-05,
      "step": 96222,
      "training_step_time": 9.25921893119812
    },
    {
      "epoch": 2.1193504333496093e-05,
      "model_forward_time": 1.7537741661071777,
      "step": 96223
    },
    {
      "epoch": 2.1193504333496093e-05,
      "step": 96223,
      "training_step_time": 8.978448390960693
    },
    {
      "epoch": 2.11944580078125e-05,
      "model_forward_time": 1.7486963272094727,
      "step": 96224
    },
    {
      "epoch": 2.11944580078125e-05,
      "step": 96224,
      "training_step_time": 9.0890793800354
    },
    {
      "epoch": 2.1195411682128908e-05,
      "model_forward_time": 2.382581949234009,
      "step": 96225
    },
    {
      "epoch": 2.1195411682128908e-05,
      "step": 96225,
      "training_step_time": 7.604217529296875
    },
    {
      "epoch": 2.119636535644531e-05,
      "model_forward_time": 1.7563743591308594,
      "step": 96226
    },
    {
      "epoch": 2.119636535644531e-05,
      "step": 96226,
      "training_step_time": 7.79620099067688
    },
    {
      "epoch": 2.119731903076172e-05,
      "model_forward_time": 1.776179313659668,
      "step": 96227
    },
    {
      "epoch": 2.119731903076172e-05,
      "step": 96227,
      "training_step_time": 7.524244785308838
    },
    {
      "epoch": 2.1198272705078126e-05,
      "model_forward_time": 1.9204649925231934,
      "step": 96228
    },
    {
      "epoch": 2.1198272705078126e-05,
      "step": 96228,
      "training_step_time": 7.371379375457764
    },
    {
      "epoch": 2.119922637939453e-05,
      "model_forward_time": 1.9023611545562744,
      "step": 96229
    },
    {
      "epoch": 2.119922637939453e-05,
      "step": 96229,
      "training_step_time": 7.615810394287109
    },
    {
      "epoch": 2.1200180053710938e-05,
      "grad_norm": 0.09769663214683533,
      "learning_rate": 3.882779396692959e-08,
      "loss": 0.054,
      "step": 96230
    },
    {
      "epoch": 2.1200180053710938e-05,
      "model_forward_time": 1.7949955463409424,
      "step": 96230
    },
    {
      "dynamics_loss_avg": 0.06299130488187074,
      "epoch": 2.1200180053710938e-05,
      "step": 96230
    },
    {
      "action_loss_avg": 0.0010081523738335817,
      "epoch": 2.1200180053710938e-05,
      "step": 96230
    },
    {
      "epoch": 2.1200180053710938e-05,
      "step": 96230,
      "training_step_time": 7.661969423294067
    },
    {
      "epoch": 2.1201133728027345e-05,
      "model_forward_time": 1.7525224685668945,
      "step": 96231
    },
    {
      "epoch": 2.1201133728027345e-05,
      "step": 96231,
      "training_step_time": 8.308001279830933
    },
    {
      "epoch": 2.120208740234375e-05,
      "model_forward_time": 1.7425487041473389,
      "step": 96232
    },
    {
      "epoch": 2.120208740234375e-05,
      "step": 96232,
      "training_step_time": 8.49311113357544
    },
    {
      "epoch": 2.1203041076660157e-05,
      "model_forward_time": 1.745854377746582,
      "step": 96233
    },
    {
      "epoch": 2.1203041076660157e-05,
      "step": 96233,
      "training_step_time": 9.58782958984375
    },
    {
      "epoch": 2.1203994750976564e-05,
      "model_forward_time": 2.3873205184936523,
      "step": 96234
    },
    {
      "epoch": 2.1203994750976564e-05,
      "step": 96234,
      "training_step_time": 8.56999683380127
    },
    {
      "epoch": 2.1204948425292968e-05,
      "model_forward_time": 1.7414100170135498,
      "step": 96235
    },
    {
      "epoch": 2.1204948425292968e-05,
      "step": 96235,
      "training_step_time": 8.32074499130249
    },
    {
      "epoch": 2.1205902099609376e-05,
      "model_forward_time": 1.3684518337249756,
      "step": 96236
    },
    {
      "epoch": 2.1205902099609376e-05,
      "step": 96236,
      "training_step_time": 8.04359769821167
    },
    {
      "epoch": 2.120685577392578e-05,
      "model_forward_time": 1.745880365371704,
      "step": 96237
    },
    {
      "epoch": 2.120685577392578e-05,
      "step": 96237,
      "training_step_time": 8.56934380531311
    },
    {
      "epoch": 2.1207809448242187e-05,
      "model_forward_time": 1.7520129680633545,
      "step": 96238
    },
    {
      "epoch": 2.1207809448242187e-05,
      "step": 96238,
      "training_step_time": 7.566556215286255
    },
    {
      "epoch": 2.1208763122558594e-05,
      "model_forward_time": 1.758650779724121,
      "step": 96239
    },
    {
      "epoch": 2.1208763122558594e-05,
      "step": 96239,
      "training_step_time": 8.709033012390137
    },
    {
      "epoch": 2.1209716796875e-05,
      "grad_norm": 0.09119070321321487,
      "learning_rate": 3.862240381455995e-08,
      "loss": 0.0545,
      "step": 96240
    },
    {
      "epoch": 2.1209716796875e-05,
      "model_forward_time": 1.74806809425354,
      "step": 96240
    },
    {
      "dynamics_loss_avg": 0.058784475363790986,
      "epoch": 2.1209716796875e-05,
      "step": 96240
    },
    {
      "action_loss_avg": 0.001247194898314774,
      "epoch": 2.1209716796875e-05,
      "step": 96240
    },
    {
      "epoch": 2.1209716796875e-05,
      "step": 96240,
      "training_step_time": 8.510126113891602
    },
    {
      "epoch": 2.1210670471191406e-05,
      "model_forward_time": 1.7503290176391602,
      "step": 96241
    },
    {
      "epoch": 2.1210670471191406e-05,
      "step": 96241,
      "training_step_time": 7.806476354598999
    },
    {
      "epoch": 2.1211624145507813e-05,
      "model_forward_time": 1.7545726299285889,
      "step": 96242
    },
    {
      "epoch": 2.1211624145507813e-05,
      "step": 96242,
      "training_step_time": 8.807172060012817
    },
    {
      "epoch": 2.1212577819824217e-05,
      "model_forward_time": 2.355471134185791,
      "step": 96243
    },
    {
      "epoch": 2.1212577819824217e-05,
      "step": 96243,
      "training_step_time": 7.746608734130859
    },
    {
      "epoch": 2.1213531494140625e-05,
      "model_forward_time": 1.746849775314331,
      "step": 96244
    },
    {
      "epoch": 2.1213531494140625e-05,
      "step": 96244,
      "training_step_time": 7.876305341720581
    },
    {
      "epoch": 2.1214485168457032e-05,
      "model_forward_time": 1.7266454696655273,
      "step": 96245
    },
    {
      "epoch": 2.1214485168457032e-05,
      "step": 96245,
      "training_step_time": 8.113878011703491
    },
    {
      "epoch": 2.1215438842773436e-05,
      "model_forward_time": 1.8567769527435303,
      "step": 96246
    },
    {
      "epoch": 2.1215438842773436e-05,
      "step": 96246,
      "training_step_time": 7.533349990844727
    },
    {
      "epoch": 2.1216392517089844e-05,
      "model_forward_time": 1.7637145519256592,
      "step": 96247
    },
    {
      "epoch": 2.1216392517089844e-05,
      "step": 96247,
      "training_step_time": 7.520903587341309
    },
    {
      "epoch": 2.121734619140625e-05,
      "model_forward_time": 1.7581603527069092,
      "step": 96248
    },
    {
      "epoch": 2.121734619140625e-05,
      "step": 96248,
      "training_step_time": 7.650104522705078
    },
    {
      "epoch": 2.1218299865722655e-05,
      "model_forward_time": 1.7518293857574463,
      "step": 96249
    },
    {
      "epoch": 2.1218299865722655e-05,
      "step": 96249,
      "training_step_time": 9.714969158172607
    },
    {
      "epoch": 2.1219253540039062e-05,
      "grad_norm": 0.0895163044333458,
      "learning_rate": 3.841755623098575e-08,
      "loss": 0.0526,
      "step": 96250
    },
    {
      "epoch": 2.1219253540039062e-05,
      "model_forward_time": 1.7315707206726074,
      "step": 96250
    },
    {
      "dynamics_loss_avg": 0.0569450156763196,
      "epoch": 2.1219253540039062e-05,
      "step": 96250
    },
    {
      "action_loss_avg": 0.001351881882874295,
      "epoch": 2.1219253540039062e-05,
      "step": 96250
    },
    {
      "epoch": 2.1219253540039062e-05,
      "step": 96250,
      "training_step_time": 8.754744291305542
    },
    {
      "epoch": 2.122020721435547e-05,
      "model_forward_time": 1.7467281818389893,
      "step": 96251
    },
    {
      "epoch": 2.122020721435547e-05,
      "step": 96251,
      "training_step_time": 8.52782917022705
    },
    {
      "epoch": 2.1221160888671874e-05,
      "model_forward_time": 1.7434113025665283,
      "step": 96252
    },
    {
      "epoch": 2.1221160888671874e-05,
      "step": 96252,
      "training_step_time": 7.712698459625244
    },
    {
      "epoch": 2.122211456298828e-05,
      "model_forward_time": 2.4091756343841553,
      "step": 96253
    },
    {
      "epoch": 2.122211456298828e-05,
      "step": 96253,
      "training_step_time": 7.67043399810791
    },
    {
      "epoch": 2.122306823730469e-05,
      "model_forward_time": 1.0250575542449951,
      "step": 96254
    },
    {
      "epoch": 2.122306823730469e-05,
      "step": 96254,
      "training_step_time": 7.802910804748535
    },
    {
      "epoch": 2.1224021911621093e-05,
      "model_forward_time": 1.817781686782837,
      "step": 96255
    },
    {
      "epoch": 2.1224021911621093e-05,
      "step": 96255,
      "training_step_time": 7.677547931671143
    },
    {
      "epoch": 2.12249755859375e-05,
      "model_forward_time": 1.7506568431854248,
      "step": 96256
    },
    {
      "epoch": 2.12249755859375e-05,
      "step": 96256,
      "training_step_time": 7.6188883781433105
    },
    {
      "epoch": 2.1225929260253908e-05,
      "model_forward_time": 1.7292344570159912,
      "step": 96257
    },
    {
      "epoch": 2.1225929260253908e-05,
      "step": 96257,
      "training_step_time": 8.511990785598755
    },
    {
      "epoch": 2.122688293457031e-05,
      "model_forward_time": 1.7454328536987305,
      "step": 96258
    },
    {
      "epoch": 2.122688293457031e-05,
      "step": 96258,
      "training_step_time": 8.65114951133728
    },
    {
      "epoch": 2.122783660888672e-05,
      "model_forward_time": 1.7544934749603271,
      "step": 96259
    },
    {
      "epoch": 2.122783660888672e-05,
      "step": 96259,
      "training_step_time": 7.700759649276733
    },
    {
      "epoch": 2.1228790283203126e-05,
      "grad_norm": 0.09856759011745453,
      "learning_rate": 3.821325123860742e-08,
      "loss": 0.0536,
      "step": 96260
    },
    {
      "epoch": 2.1228790283203126e-05,
      "model_forward_time": 1.7339234352111816,
      "step": 96260
    },
    {
      "dynamics_loss_avg": 0.05320426691323519,
      "epoch": 2.1228790283203126e-05,
      "step": 96260
    },
    {
      "action_loss_avg": 0.001314619753975421,
      "epoch": 2.1228790283203126e-05,
      "step": 96260
    },
    {
      "epoch": 2.1228790283203126e-05,
      "step": 96260,
      "training_step_time": 8.995559453964233
    },
    {
      "epoch": 2.122974395751953e-05,
      "model_forward_time": 1.743394374847412,
      "step": 96261
    },
    {
      "epoch": 2.122974395751953e-05,
      "step": 96261,
      "training_step_time": 8.584141731262207
    },
    {
      "epoch": 2.1230697631835938e-05,
      "model_forward_time": 1.9816703796386719,
      "step": 96262
    },
    {
      "epoch": 2.1230697631835938e-05,
      "step": 96262,
      "training_step_time": 7.716794490814209
    },
    {
      "epoch": 2.1231651306152345e-05,
      "model_forward_time": 1.7505931854248047,
      "step": 96263
    },
    {
      "epoch": 2.1231651306152345e-05,
      "step": 96263,
      "training_step_time": 7.869839668273926
    },
    {
      "epoch": 2.123260498046875e-05,
      "model_forward_time": 1.3670706748962402,
      "step": 96264
    },
    {
      "epoch": 2.123260498046875e-05,
      "step": 96264,
      "training_step_time": 7.67564845085144
    },
    {
      "epoch": 2.1233558654785157e-05,
      "model_forward_time": 1.9204316139221191,
      "step": 96265
    },
    {
      "epoch": 2.1233558654785157e-05,
      "step": 96265,
      "training_step_time": 7.567411661148071
    },
    {
      "epoch": 2.123451232910156e-05,
      "model_forward_time": 1.915769338607788,
      "step": 96266
    },
    {
      "epoch": 2.123451232910156e-05,
      "step": 96266,
      "training_step_time": 7.662860155105591
    },
    {
      "epoch": 2.1235466003417968e-05,
      "model_forward_time": 1.9240047931671143,
      "step": 96267
    },
    {
      "epoch": 2.1235466003417968e-05,
      "step": 96267,
      "training_step_time": 9.309619903564453
    },
    {
      "epoch": 2.1236419677734376e-05,
      "model_forward_time": 1.5213677883148193,
      "step": 96268
    },
    {
      "epoch": 2.1236419677734376e-05,
      "step": 96268,
      "training_step_time": 8.495145082473755
    },
    {
      "epoch": 2.123737335205078e-05,
      "model_forward_time": 1.511690378189087,
      "step": 96269
    },
    {
      "epoch": 2.123737335205078e-05,
      "step": 96269,
      "training_step_time": 9.068891525268555
    },
    {
      "epoch": 2.1238327026367187e-05,
      "grad_norm": 0.08781027793884277,
      "learning_rate": 3.8009488859768183e-08,
      "loss": 0.0541,
      "step": 96270
    },
    {
      "epoch": 2.1238327026367187e-05,
      "model_forward_time": 1.924755334854126,
      "step": 96270
    },
    {
      "dynamics_loss_avg": 0.05099863559007645,
      "epoch": 2.1238327026367187e-05,
      "step": 96270
    },
    {
      "action_loss_avg": 0.001958543760702014,
      "epoch": 2.1238327026367187e-05,
      "step": 96270
    },
    {
      "epoch": 2.1238327026367187e-05,
      "step": 96270,
      "training_step_time": 8.608474969863892
    },
    {
      "epoch": 2.1239280700683595e-05,
      "model_forward_time": 1.796788215637207,
      "step": 96271
    },
    {
      "epoch": 2.1239280700683595e-05,
      "step": 96271,
      "training_step_time": 7.7576305866241455
    },
    {
      "epoch": 2.1240234375e-05,
      "model_forward_time": 2.3987009525299072,
      "step": 96272
    },
    {
      "epoch": 2.1240234375e-05,
      "step": 96272,
      "training_step_time": 7.822772026062012
    },
    {
      "epoch": 2.1241188049316406e-05,
      "model_forward_time": 1.770486831665039,
      "step": 96273
    },
    {
      "epoch": 2.1241188049316406e-05,
      "step": 96273,
      "training_step_time": 8.349310159683228
    },
    {
      "epoch": 2.1242141723632813e-05,
      "model_forward_time": 1.7545125484466553,
      "step": 96274
    },
    {
      "epoch": 2.1242141723632813e-05,
      "step": 96274,
      "training_step_time": 8.623964548110962
    },
    {
      "epoch": 2.1243095397949217e-05,
      "model_forward_time": 1.7625186443328857,
      "step": 96275
    },
    {
      "epoch": 2.1243095397949217e-05,
      "step": 96275,
      "training_step_time": 7.636331081390381
    },
    {
      "epoch": 2.1244049072265625e-05,
      "model_forward_time": 1.7540433406829834,
      "step": 96276
    },
    {
      "epoch": 2.1244049072265625e-05,
      "step": 96276,
      "training_step_time": 7.677196741104126
    },
    {
      "epoch": 2.1245002746582032e-05,
      "model_forward_time": 1.750863790512085,
      "step": 96277
    },
    {
      "epoch": 2.1245002746582032e-05,
      "step": 96277,
      "training_step_time": 8.59330415725708
    },
    {
      "epoch": 2.1245956420898436e-05,
      "model_forward_time": 1.7575182914733887,
      "step": 96278
    },
    {
      "epoch": 2.1245956420898436e-05,
      "step": 96278,
      "training_step_time": 8.675883769989014
    },
    {
      "epoch": 2.1246910095214844e-05,
      "model_forward_time": 1.7527353763580322,
      "step": 96279
    },
    {
      "epoch": 2.1246910095214844e-05,
      "step": 96279,
      "training_step_time": 8.728740215301514
    },
    {
      "epoch": 2.124786376953125e-05,
      "grad_norm": 0.09625215828418732,
      "learning_rate": 3.7806269116750784e-08,
      "loss": 0.0531,
      "step": 96280
    },
    {
      "epoch": 2.124786376953125e-05,
      "model_forward_time": 1.3766510486602783,
      "step": 96280
    },
    {
      "dynamics_loss_avg": 0.06369236819446086,
      "epoch": 2.124786376953125e-05,
      "step": 96280
    },
    {
      "action_loss_avg": 0.0010865398216992617,
      "epoch": 2.124786376953125e-05,
      "step": 96280
    },
    {
      "epoch": 2.124786376953125e-05,
      "step": 96280,
      "training_step_time": 7.779144525527954
    },
    {
      "epoch": 2.1248817443847655e-05,
      "model_forward_time": 2.43448543548584,
      "step": 96281
    },
    {
      "epoch": 2.1248817443847655e-05,
      "step": 96281,
      "training_step_time": 7.925767660140991
    },
    {
      "epoch": 2.1249771118164063e-05,
      "model_forward_time": 1.755415916442871,
      "step": 96282
    },
    {
      "epoch": 2.1249771118164063e-05,
      "step": 96282,
      "training_step_time": 7.588690757751465
    },
    {
      "epoch": 2.125072479248047e-05,
      "model_forward_time": 1.864800214767456,
      "step": 96283
    },
    {
      "epoch": 2.125072479248047e-05,
      "step": 96283,
      "training_step_time": 7.732065677642822
    },
    {
      "epoch": 2.1251678466796874e-05,
      "model_forward_time": 1.7511775493621826,
      "step": 96284
    },
    {
      "epoch": 2.1251678466796874e-05,
      "step": 96284,
      "training_step_time": 8.647823095321655
    },
    {
      "epoch": 2.125263214111328e-05,
      "model_forward_time": 1.755856990814209,
      "step": 96285
    },
    {
      "epoch": 2.125263214111328e-05,
      "step": 96285,
      "training_step_time": 8.685774564743042
    },
    {
      "epoch": 2.125358581542969e-05,
      "model_forward_time": 1.75703763961792,
      "step": 96286
    },
    {
      "epoch": 2.125358581542969e-05,
      "step": 96286,
      "training_step_time": 7.697585344314575
    },
    {
      "epoch": 2.1254539489746093e-05,
      "model_forward_time": 1.7574443817138672,
      "step": 96287
    },
    {
      "epoch": 2.1254539489746093e-05,
      "step": 96287,
      "training_step_time": 8.63073468208313
    },
    {
      "epoch": 2.12554931640625e-05,
      "model_forward_time": 1.7535068988800049,
      "step": 96288
    },
    {
      "epoch": 2.12554931640625e-05,
      "step": 96288,
      "training_step_time": 8.877726316452026
    },
    {
      "epoch": 2.1256446838378908e-05,
      "model_forward_time": 1.7672643661499023,
      "step": 96289
    },
    {
      "epoch": 2.1256446838378908e-05,
      "step": 96289,
      "training_step_time": 8.639535665512085
    },
    {
      "epoch": 2.125740051269531e-05,
      "grad_norm": 0.09357135742902756,
      "learning_rate": 3.76035920317791e-08,
      "loss": 0.0536,
      "step": 96290
    },
    {
      "epoch": 2.125740051269531e-05,
      "model_forward_time": 2.399501085281372,
      "step": 96290
    },
    {
      "dynamics_loss_avg": 0.05682127438485622,
      "epoch": 2.125740051269531e-05,
      "step": 96290
    },
    {
      "action_loss_avg": 0.0015318091260269283,
      "epoch": 2.125740051269531e-05,
      "step": 96290
    },
    {
      "epoch": 2.125740051269531e-05,
      "step": 96290,
      "training_step_time": 7.857343912124634
    },
    {
      "epoch": 2.125835418701172e-05,
      "model_forward_time": 1.7907521724700928,
      "step": 96291
    },
    {
      "epoch": 2.125835418701172e-05,
      "step": 96291,
      "training_step_time": 8.13078784942627
    },
    {
      "epoch": 2.1259307861328127e-05,
      "model_forward_time": 1.8856053352355957,
      "step": 96292
    },
    {
      "epoch": 2.1259307861328127e-05,
      "step": 96292,
      "training_step_time": 8.33332633972168
    },
    {
      "epoch": 2.126026153564453e-05,
      "model_forward_time": 1.7541699409484863,
      "step": 96293
    },
    {
      "epoch": 2.126026153564453e-05,
      "step": 96293,
      "training_step_time": 7.622951030731201
    },
    {
      "epoch": 2.1261215209960938e-05,
      "model_forward_time": 1.7513189315795898,
      "step": 96294
    },
    {
      "epoch": 2.1261215209960938e-05,
      "step": 96294,
      "training_step_time": 7.659144878387451
    },
    {
      "epoch": 2.1262168884277345e-05,
      "model_forward_time": 1.7653799057006836,
      "step": 96295
    },
    {
      "epoch": 2.1262168884277345e-05,
      "step": 96295,
      "training_step_time": 8.392458200454712
    },
    {
      "epoch": 2.126312255859375e-05,
      "model_forward_time": 1.7515428066253662,
      "step": 96296
    },
    {
      "epoch": 2.126312255859375e-05,
      "step": 96296,
      "training_step_time": 7.724263906478882
    },
    {
      "epoch": 2.1264076232910157e-05,
      "model_forward_time": 1.7562298774719238,
      "step": 96297
    },
    {
      "epoch": 2.1264076232910157e-05,
      "step": 96297,
      "training_step_time": 10.244861364364624
    },
    {
      "epoch": 2.126502990722656e-05,
      "model_forward_time": 1.7553157806396484,
      "step": 96298
    },
    {
      "epoch": 2.126502990722656e-05,
      "step": 96298,
      "training_step_time": 7.832814693450928
    },
    {
      "epoch": 2.1265983581542968e-05,
      "model_forward_time": 2.3632426261901855,
      "step": 96299
    },
    {
      "epoch": 2.1265983581542968e-05,
      "step": 96299,
      "training_step_time": 8.728909730911255
    },
    {
      "epoch": 2.1266937255859376e-05,
      "grad_norm": 0.0821862667798996,
      "learning_rate": 3.740145762701819e-08,
      "loss": 0.0541,
      "step": 96300
    },
    {
      "epoch": 2.1266937255859376e-05,
      "model_forward_time": 1.7651689052581787,
      "step": 96300
    },
    {
      "dynamics_loss_avg": 0.06633225884288549,
      "epoch": 2.1266937255859376e-05,
      "step": 96300
    },
    {
      "action_loss_avg": 0.0016575619578361512,
      "epoch": 2.1266937255859376e-05,
      "step": 96300
    },
    {
      "epoch": 2.1266937255859376e-05,
      "step": 96300,
      "training_step_time": 7.620337247848511
    },
    {
      "epoch": 2.126789093017578e-05,
      "model_forward_time": 1.8382811546325684,
      "step": 96301
    },
    {
      "epoch": 2.126789093017578e-05,
      "step": 96301,
      "training_step_time": 7.746318340301514
    },
    {
      "epoch": 2.1268844604492187e-05,
      "model_forward_time": 1.7529468536376953,
      "step": 96302
    },
    {
      "epoch": 2.1268844604492187e-05,
      "step": 96302,
      "training_step_time": 8.62604808807373
    },
    {
      "epoch": 2.1269798278808595e-05,
      "model_forward_time": 1.739332914352417,
      "step": 96303
    },
    {
      "epoch": 2.1269798278808595e-05,
      "step": 96303,
      "training_step_time": 7.677978992462158
    },
    {
      "epoch": 2.1270751953125e-05,
      "model_forward_time": 1.7507774829864502,
      "step": 96304
    },
    {
      "epoch": 2.1270751953125e-05,
      "step": 96304,
      "training_step_time": 7.701578617095947
    },
    {
      "epoch": 2.1271705627441406e-05,
      "model_forward_time": 1.7480602264404297,
      "step": 96305
    },
    {
      "epoch": 2.1271705627441406e-05,
      "step": 96305,
      "training_step_time": 9.138522624969482
    },
    {
      "epoch": 2.1272659301757813e-05,
      "model_forward_time": 1.0316994190216064,
      "step": 96306
    },
    {
      "epoch": 2.1272659301757813e-05,
      "step": 96306,
      "training_step_time": 7.760365962982178
    },
    {
      "epoch": 2.1273612976074217e-05,
      "model_forward_time": 1.7517690658569336,
      "step": 96307
    },
    {
      "epoch": 2.1273612976074217e-05,
      "step": 96307,
      "training_step_time": 8.717522621154785
    },
    {
      "epoch": 2.1274566650390625e-05,
      "model_forward_time": 1.7563157081604004,
      "step": 96308
    },
    {
      "epoch": 2.1274566650390625e-05,
      "step": 96308,
      "training_step_time": 8.649800062179565
    },
    {
      "epoch": 2.1275520324707032e-05,
      "model_forward_time": 1.992002010345459,
      "step": 96309
    },
    {
      "epoch": 2.1275520324707032e-05,
      "step": 96309,
      "training_step_time": 8.71886920928955
    },
    {
      "epoch": 2.1276473999023436e-05,
      "grad_norm": 0.0766741931438446,
      "learning_rate": 3.719986592457259e-08,
      "loss": 0.0538,
      "step": 96310
    },
    {
      "epoch": 2.1276473999023436e-05,
      "model_forward_time": 1.8020100593566895,
      "step": 96310
    },
    {
      "dynamics_loss_avg": 0.0551057904958725,
      "epoch": 2.1276473999023436e-05,
      "step": 96310
    },
    {
      "action_loss_avg": 0.0012997848185477777,
      "epoch": 2.1276473999023436e-05,
      "step": 96310
    },
    {
      "epoch": 2.1276473999023436e-05,
      "step": 96310,
      "training_step_time": 7.737047910690308
    },
    {
      "epoch": 2.1277427673339844e-05,
      "model_forward_time": 1.7476842403411865,
      "step": 96311
    },
    {
      "epoch": 2.1277427673339844e-05,
      "step": 96311,
      "training_step_time": 7.665582895278931
    },
    {
      "epoch": 2.127838134765625e-05,
      "model_forward_time": 1.7459404468536377,
      "step": 96312
    },
    {
      "epoch": 2.127838134765625e-05,
      "step": 96312,
      "training_step_time": 8.704052209854126
    },
    {
      "epoch": 2.1279335021972655e-05,
      "model_forward_time": 1.7510387897491455,
      "step": 96313
    },
    {
      "epoch": 2.1279335021972655e-05,
      "step": 96313,
      "training_step_time": 8.153172016143799
    },
    {
      "epoch": 2.1280288696289063e-05,
      "model_forward_time": 1.7760791778564453,
      "step": 96314
    },
    {
      "epoch": 2.1280288696289063e-05,
      "step": 96314,
      "training_step_time": 8.445798397064209
    },
    {
      "epoch": 2.128124237060547e-05,
      "model_forward_time": 1.7491250038146973,
      "step": 96315
    },
    {
      "epoch": 2.128124237060547e-05,
      "step": 96315,
      "training_step_time": 9.13300633430481
    },
    {
      "epoch": 2.1282196044921874e-05,
      "model_forward_time": 1.7567763328552246,
      "step": 96316
    },
    {
      "epoch": 2.1282196044921874e-05,
      "step": 96316,
      "training_step_time": 10.54208493232727
    },
    {
      "epoch": 2.128314971923828e-05,
      "model_forward_time": 1.7502148151397705,
      "step": 96317
    },
    {
      "epoch": 2.128314971923828e-05,
      "step": 96317,
      "training_step_time": 8.525897979736328
    },
    {
      "epoch": 2.128410339355469e-05,
      "model_forward_time": 2.3731331825256348,
      "step": 96318
    },
    {
      "epoch": 2.128410339355469e-05,
      "step": 96318,
      "training_step_time": 7.690871238708496
    },
    {
      "epoch": 2.1285057067871093e-05,
      "model_forward_time": 1.777148962020874,
      "step": 96319
    },
    {
      "epoch": 2.1285057067871093e-05,
      "step": 96319,
      "training_step_time": 7.6498777866363525
    },
    {
      "epoch": 2.12860107421875e-05,
      "grad_norm": 0.08008698374032974,
      "learning_rate": 3.699881694648799e-08,
      "loss": 0.053,
      "step": 96320
    },
    {
      "epoch": 2.12860107421875e-05,
      "model_forward_time": 1.7395963668823242,
      "step": 96320
    },
    {
      "dynamics_loss_avg": 0.05106864403933287,
      "epoch": 2.12860107421875e-05,
      "step": 96320
    },
    {
      "action_loss_avg": 0.0007348309707595035,
      "epoch": 2.12860107421875e-05,
      "step": 96320
    },
    {
      "epoch": 2.12860107421875e-05,
      "step": 96320,
      "training_step_time": 7.57439112663269
    },
    {
      "epoch": 2.1286964416503908e-05,
      "model_forward_time": 1.7636330127716064,
      "step": 96321
    },
    {
      "epoch": 2.1286964416503908e-05,
      "step": 96321,
      "training_step_time": 8.15179991722107
    },
    {
      "epoch": 2.1287918090820312e-05,
      "model_forward_time": 1.7527246475219727,
      "step": 96322
    },
    {
      "epoch": 2.1287918090820312e-05,
      "step": 96322,
      "training_step_time": 9.297982692718506
    },
    {
      "epoch": 2.128887176513672e-05,
      "model_forward_time": 1.7706925868988037,
      "step": 96323
    },
    {
      "epoch": 2.128887176513672e-05,
      "step": 96323,
      "training_step_time": 8.52271819114685
    },
    {
      "epoch": 2.1289825439453127e-05,
      "model_forward_time": 1.753288745880127,
      "step": 96324
    },
    {
      "epoch": 2.1289825439453127e-05,
      "step": 96324,
      "training_step_time": 8.374686002731323
    },
    {
      "epoch": 2.129077911376953e-05,
      "model_forward_time": 1.7499167919158936,
      "step": 96325
    },
    {
      "epoch": 2.129077911376953e-05,
      "step": 96325,
      "training_step_time": 8.74102234840393
    },
    {
      "epoch": 2.1291732788085938e-05,
      "model_forward_time": 1.733741283416748,
      "step": 96326
    },
    {
      "epoch": 2.1291732788085938e-05,
      "step": 96326,
      "training_step_time": 7.7706522941589355
    },
    {
      "epoch": 2.1292686462402345e-05,
      "model_forward_time": 1.745115041732788,
      "step": 96327
    },
    {
      "epoch": 2.1292686462402345e-05,
      "step": 96327,
      "training_step_time": 8.590914964675903
    },
    {
      "epoch": 2.129364013671875e-05,
      "model_forward_time": 2.422229528427124,
      "step": 96328
    },
    {
      "epoch": 2.129364013671875e-05,
      "step": 96328,
      "training_step_time": 7.68091607093811
    },
    {
      "epoch": 2.1294593811035157e-05,
      "model_forward_time": 1.7523760795593262,
      "step": 96329
    },
    {
      "epoch": 2.1294593811035157e-05,
      "step": 96329,
      "training_step_time": 7.585371971130371
    },
    {
      "epoch": 2.129554748535156e-05,
      "grad_norm": 0.07794994860887527,
      "learning_rate": 3.6798310714751265e-08,
      "loss": 0.0543,
      "step": 96330
    },
    {
      "epoch": 2.129554748535156e-05,
      "model_forward_time": 1.728858470916748,
      "step": 96330
    },
    {
      "dynamics_loss_avg": 0.060204360634088516,
      "epoch": 2.129554748535156e-05,
      "step": 96330
    },
    {
      "action_loss_avg": 0.0019604475935921075,
      "epoch": 2.129554748535156e-05,
      "step": 96330
    },
    {
      "epoch": 2.129554748535156e-05,
      "step": 96330,
      "training_step_time": 7.688482284545898
    },
    {
      "epoch": 2.1296501159667968e-05,
      "model_forward_time": 1.7472448348999023,
      "step": 96331
    },
    {
      "epoch": 2.1296501159667968e-05,
      "step": 96331,
      "training_step_time": 7.8145270347595215
    },
    {
      "epoch": 2.1297454833984376e-05,
      "model_forward_time": 1.7540857791900635,
      "step": 96332
    },
    {
      "epoch": 2.1297454833984376e-05,
      "step": 96332,
      "training_step_time": 8.569727897644043
    },
    {
      "epoch": 2.129840850830078e-05,
      "model_forward_time": 1.7510905265808105,
      "step": 96333
    },
    {
      "epoch": 2.129840850830078e-05,
      "step": 96333,
      "training_step_time": 8.806970834732056
    },
    {
      "epoch": 2.1299362182617187e-05,
      "model_forward_time": 1.7462472915649414,
      "step": 96334
    },
    {
      "epoch": 2.1299362182617187e-05,
      "step": 96334,
      "training_step_time": 8.565239906311035
    },
    {
      "epoch": 2.1300315856933595e-05,
      "model_forward_time": 1.7545013427734375,
      "step": 96335
    },
    {
      "epoch": 2.1300315856933595e-05,
      "step": 96335,
      "training_step_time": 7.862223148345947
    },
    {
      "epoch": 2.130126953125e-05,
      "model_forward_time": 1.7584233283996582,
      "step": 96336
    },
    {
      "epoch": 2.130126953125e-05,
      "step": 96336,
      "training_step_time": 7.79058837890625
    },
    {
      "epoch": 2.1302223205566406e-05,
      "model_forward_time": 2.4235098361968994,
      "step": 96337
    },
    {
      "epoch": 2.1302223205566406e-05,
      "step": 96337,
      "training_step_time": 7.699826717376709
    },
    {
      "epoch": 2.1303176879882813e-05,
      "model_forward_time": 1.8709301948547363,
      "step": 96338
    },
    {
      "epoch": 2.1303176879882813e-05,
      "step": 96338,
      "training_step_time": 7.80936861038208
    },
    {
      "epoch": 2.1304130554199217e-05,
      "model_forward_time": 1.3734581470489502,
      "step": 96339
    },
    {
      "epoch": 2.1304130554199217e-05,
      "step": 96339,
      "training_step_time": 8.367917537689209
    },
    {
      "epoch": 2.1305084228515625e-05,
      "grad_norm": 0.08176490664482117,
      "learning_rate": 3.65983472512893e-08,
      "loss": 0.0539,
      "step": 96340
    },
    {
      "epoch": 2.1305084228515625e-05,
      "model_forward_time": 1.7504692077636719,
      "step": 96340
    },
    {
      "dynamics_loss_avg": 0.03549478203058243,
      "epoch": 2.1305084228515625e-05,
      "step": 96340
    },
    {
      "action_loss_avg": 0.00135455277049914,
      "epoch": 2.1305084228515625e-05,
      "step": 96340
    },
    {
      "epoch": 2.1305084228515625e-05,
      "step": 96340,
      "training_step_time": 7.665734767913818
    },
    {
      "epoch": 2.1306037902832032e-05,
      "model_forward_time": 1.7515013217926025,
      "step": 96341
    },
    {
      "epoch": 2.1306037902832032e-05,
      "step": 96341,
      "training_step_time": 8.667735815048218
    },
    {
      "epoch": 2.1306991577148436e-05,
      "model_forward_time": 1.366816759109497,
      "step": 96342
    },
    {
      "epoch": 2.1306991577148436e-05,
      "step": 96342,
      "training_step_time": 8.221237659454346
    },
    {
      "epoch": 2.1307945251464844e-05,
      "model_forward_time": 1.7630207538604736,
      "step": 96343
    },
    {
      "epoch": 2.1307945251464844e-05,
      "step": 96343,
      "training_step_time": 7.754927635192871
    },
    {
      "epoch": 2.130889892578125e-05,
      "model_forward_time": 1.7494699954986572,
      "step": 96344
    },
    {
      "epoch": 2.130889892578125e-05,
      "step": 96344,
      "training_step_time": 7.835520029067993
    },
    {
      "epoch": 2.1309852600097655e-05,
      "model_forward_time": 1.751502513885498,
      "step": 96345
    },
    {
      "epoch": 2.1309852600097655e-05,
      "step": 96345,
      "training_step_time": 7.864223480224609
    },
    {
      "epoch": 2.1310806274414063e-05,
      "model_forward_time": 2.4313995838165283,
      "step": 96346
    },
    {
      "epoch": 2.1310806274414063e-05,
      "step": 96346,
      "training_step_time": 7.781129598617554
    },
    {
      "epoch": 2.131175994873047e-05,
      "model_forward_time": 1.4386365413665771,
      "step": 96347
    },
    {
      "epoch": 2.131175994873047e-05,
      "step": 96347,
      "training_step_time": 7.80734395980835
    },
    {
      "epoch": 2.1312713623046874e-05,
      "model_forward_time": 1.7585415840148926,
      "step": 96348
    },
    {
      "epoch": 2.1312713623046874e-05,
      "step": 96348,
      "training_step_time": 7.644832611083984
    },
    {
      "epoch": 2.131366729736328e-05,
      "model_forward_time": 1.7354674339294434,
      "step": 96349
    },
    {
      "epoch": 2.131366729736328e-05,
      "step": 96349,
      "training_step_time": 7.726960897445679
    },
    {
      "epoch": 2.131462097167969e-05,
      "grad_norm": 0.12142124027013779,
      "learning_rate": 3.639892657796962e-08,
      "loss": 0.0533,
      "step": 96350
    },
    {
      "epoch": 2.131462097167969e-05,
      "model_forward_time": 1.753758430480957,
      "step": 96350
    },
    {
      "dynamics_loss_avg": 0.05301419924944639,
      "epoch": 2.131462097167969e-05,
      "step": 96350
    },
    {
      "action_loss_avg": 0.0010894441540585831,
      "epoch": 2.131462097167969e-05,
      "step": 96350
    },
    {
      "epoch": 2.131462097167969e-05,
      "step": 96350,
      "training_step_time": 8.446089506149292
    },
    {
      "epoch": 2.1315574645996093e-05,
      "model_forward_time": 1.7419767379760742,
      "step": 96351
    },
    {
      "epoch": 2.1315574645996093e-05,
      "step": 96351,
      "training_step_time": 10.890133380889893
    },
    {
      "epoch": 2.13165283203125e-05,
      "model_forward_time": 1.7472069263458252,
      "step": 96352
    },
    {
      "epoch": 2.13165283203125e-05,
      "step": 96352,
      "training_step_time": 8.365610122680664
    },
    {
      "epoch": 2.1317481994628908e-05,
      "model_forward_time": 1.7533698081970215,
      "step": 96353
    },
    {
      "epoch": 2.1317481994628908e-05,
      "step": 96353,
      "training_step_time": 9.670385837554932
    },
    {
      "epoch": 2.1318435668945312e-05,
      "model_forward_time": 1.7461323738098145,
      "step": 96354
    },
    {
      "epoch": 2.1318435668945312e-05,
      "step": 96354,
      "training_step_time": 7.887058973312378
    },
    {
      "epoch": 2.131938934326172e-05,
      "model_forward_time": 1.753957986831665,
      "step": 96355
    },
    {
      "epoch": 2.131938934326172e-05,
      "step": 96355,
      "training_step_time": 8.567715883255005
    },
    {
      "epoch": 2.1320343017578127e-05,
      "model_forward_time": 2.5031421184539795,
      "step": 96356
    },
    {
      "epoch": 2.1320343017578127e-05,
      "step": 96356,
      "training_step_time": 7.76420521736145
    },
    {
      "epoch": 2.132129669189453e-05,
      "model_forward_time": 1.747549295425415,
      "step": 96357
    },
    {
      "epoch": 2.132129669189453e-05,
      "step": 96357,
      "training_step_time": 7.682295799255371
    },
    {
      "epoch": 2.1322250366210938e-05,
      "model_forward_time": 1.751131296157837,
      "step": 96358
    },
    {
      "epoch": 2.1322250366210938e-05,
      "step": 96358,
      "training_step_time": 7.690957307815552
    },
    {
      "epoch": 2.1323204040527345e-05,
      "model_forward_time": 1.75205659866333,
      "step": 96359
    },
    {
      "epoch": 2.1323204040527345e-05,
      "step": 96359,
      "training_step_time": 7.66453218460083
    },
    {
      "epoch": 2.132415771484375e-05,
      "grad_norm": 0.10969321429729462,
      "learning_rate": 3.6200048716600324e-08,
      "loss": 0.0532,
      "step": 96360
    },
    {
      "epoch": 2.132415771484375e-05,
      "model_forward_time": 1.7517552375793457,
      "step": 96360
    },
    {
      "dynamics_loss_avg": 0.04467255976051092,
      "epoch": 2.132415771484375e-05,
      "step": 96360
    },
    {
      "action_loss_avg": 0.000924636991112493,
      "epoch": 2.132415771484375e-05,
      "step": 96360
    },
    {
      "epoch": 2.132415771484375e-05,
      "step": 96360,
      "training_step_time": 8.442895412445068
    },
    {
      "epoch": 2.1325111389160157e-05,
      "model_forward_time": 1.7477245330810547,
      "step": 96361
    },
    {
      "epoch": 2.1325111389160157e-05,
      "step": 96361,
      "training_step_time": 8.880504369735718
    },
    {
      "epoch": 2.132606506347656e-05,
      "model_forward_time": 1.7579212188720703,
      "step": 96362
    },
    {
      "epoch": 2.132606506347656e-05,
      "step": 96362,
      "training_step_time": 7.803713798522949
    },
    {
      "epoch": 2.132701873779297e-05,
      "model_forward_time": 1.748594045639038,
      "step": 96363
    },
    {
      "epoch": 2.132701873779297e-05,
      "step": 96363,
      "training_step_time": 8.716539859771729
    },
    {
      "epoch": 2.1327972412109376e-05,
      "model_forward_time": 1.742037057876587,
      "step": 96364
    },
    {
      "epoch": 2.1327972412109376e-05,
      "step": 96364,
      "training_step_time": 8.343369722366333
    },
    {
      "epoch": 2.132892608642578e-05,
      "model_forward_time": 2.4800643920898438,
      "step": 96365
    },
    {
      "epoch": 2.132892608642578e-05,
      "step": 96365,
      "training_step_time": 7.760220289230347
    },
    {
      "epoch": 2.1329879760742187e-05,
      "model_forward_time": 1.3637936115264893,
      "step": 96366
    },
    {
      "epoch": 2.1329879760742187e-05,
      "step": 96366,
      "training_step_time": 7.702587366104126
    },
    {
      "epoch": 2.1330833435058595e-05,
      "model_forward_time": 1.7548918724060059,
      "step": 96367
    },
    {
      "epoch": 2.1330833435058595e-05,
      "step": 96367,
      "training_step_time": 8.376060962677002
    },
    {
      "epoch": 2.1331787109375e-05,
      "model_forward_time": 1.3853554725646973,
      "step": 96368
    },
    {
      "epoch": 2.1331787109375e-05,
      "step": 96368,
      "training_step_time": 7.727495431900024
    },
    {
      "epoch": 2.1332740783691406e-05,
      "model_forward_time": 1.7494025230407715,
      "step": 96369
    },
    {
      "epoch": 2.1332740783691406e-05,
      "step": 96369,
      "training_step_time": 8.665353298187256
    },
    {
      "epoch": 2.1333694458007813e-05,
      "grad_norm": 0.10141034424304962,
      "learning_rate": 3.600171368893124e-08,
      "loss": 0.0535,
      "step": 96370
    },
    {
      "epoch": 2.1333694458007813e-05,
      "model_forward_time": 1.7476277351379395,
      "step": 96370
    },
    {
      "dynamics_loss_avg": 0.046366506814956666,
      "epoch": 2.1333694458007813e-05,
      "step": 96370
    },
    {
      "action_loss_avg": 0.0011067641229601578,
      "epoch": 2.1333694458007813e-05,
      "step": 96370
    },
    {
      "epoch": 2.1333694458007813e-05,
      "step": 96370,
      "training_step_time": 7.914249897003174
    },
    {
      "epoch": 2.1334648132324217e-05,
      "model_forward_time": 1.7620034217834473,
      "step": 96371
    },
    {
      "epoch": 2.1334648132324217e-05,
      "step": 96371,
      "training_step_time": 7.799126386642456
    },
    {
      "epoch": 2.1335601806640625e-05,
      "model_forward_time": 1.7479982376098633,
      "step": 96372
    },
    {
      "epoch": 2.1335601806640625e-05,
      "step": 96372,
      "training_step_time": 7.6677539348602295
    },
    {
      "epoch": 2.1336555480957032e-05,
      "model_forward_time": 1.368943214416504,
      "step": 96373
    },
    {
      "epoch": 2.1336555480957032e-05,
      "step": 96373,
      "training_step_time": 7.7774693965911865
    },
    {
      "epoch": 2.1337509155273436e-05,
      "model_forward_time": 2.4575893878936768,
      "step": 96374
    },
    {
      "epoch": 2.1337509155273436e-05,
      "step": 96374,
      "training_step_time": 7.78683876991272
    },
    {
      "epoch": 2.1338462829589844e-05,
      "model_forward_time": 1.7450387477874756,
      "step": 96375
    },
    {
      "epoch": 2.1338462829589844e-05,
      "step": 96375,
      "training_step_time": 7.689051151275635
    },
    {
      "epoch": 2.133941650390625e-05,
      "model_forward_time": 1.7530286312103271,
      "step": 96376
    },
    {
      "epoch": 2.133941650390625e-05,
      "step": 96376,
      "training_step_time": 7.753337621688843
    },
    {
      "epoch": 2.1340370178222655e-05,
      "model_forward_time": 1.7628910541534424,
      "step": 96377
    },
    {
      "epoch": 2.1340370178222655e-05,
      "step": 96377,
      "training_step_time": 7.7277069091796875
    },
    {
      "epoch": 2.1341323852539063e-05,
      "model_forward_time": 1.7442891597747803,
      "step": 96378
    },
    {
      "epoch": 2.1341323852539063e-05,
      "step": 96378,
      "training_step_time": 8.475430965423584
    },
    {
      "epoch": 2.134227752685547e-05,
      "model_forward_time": 1.7349274158477783,
      "step": 96379
    },
    {
      "epoch": 2.134227752685547e-05,
      "step": 96379,
      "training_step_time": 8.042422533035278
    },
    {
      "epoch": 2.1343231201171874e-05,
      "grad_norm": 0.07945117354393005,
      "learning_rate": 3.580392151665113e-08,
      "loss": 0.0511,
      "step": 96380
    },
    {
      "epoch": 2.1343231201171874e-05,
      "model_forward_time": 1.7486977577209473,
      "step": 96380
    },
    {
      "dynamics_loss_avg": 0.047932271659374234,
      "epoch": 2.1343231201171874e-05,
      "step": 96380
    },
    {
      "action_loss_avg": 0.0011123648262582719,
      "epoch": 2.1343231201171874e-05,
      "step": 96380
    },
    {
      "epoch": 2.1343231201171874e-05,
      "step": 96380,
      "training_step_time": 8.838049173355103
    },
    {
      "epoch": 2.134418487548828e-05,
      "model_forward_time": 1.7511608600616455,
      "step": 96381
    },
    {
      "epoch": 2.134418487548828e-05,
      "step": 96381,
      "training_step_time": 9.790290355682373
    },
    {
      "epoch": 2.134513854980469e-05,
      "model_forward_time": 1.7535388469696045,
      "step": 96382
    },
    {
      "epoch": 2.134513854980469e-05,
      "step": 96382,
      "training_step_time": 8.674755811691284
    },
    {
      "epoch": 2.1346092224121093e-05,
      "model_forward_time": 1.7558934688568115,
      "step": 96383
    },
    {
      "epoch": 2.1346092224121093e-05,
      "step": 96383,
      "training_step_time": 8.468147277832031
    },
    {
      "epoch": 2.13470458984375e-05,
      "model_forward_time": 1.8766679763793945,
      "step": 96384
    },
    {
      "epoch": 2.13470458984375e-05,
      "step": 96384,
      "training_step_time": 7.877923965454102
    },
    {
      "epoch": 2.1347999572753908e-05,
      "model_forward_time": 1.756058692932129,
      "step": 96385
    },
    {
      "epoch": 2.1347999572753908e-05,
      "step": 96385,
      "training_step_time": 8.618198156356812
    },
    {
      "epoch": 2.1348953247070312e-05,
      "model_forward_time": 1.754295825958252,
      "step": 96386
    },
    {
      "epoch": 2.1348953247070312e-05,
      "step": 96386,
      "training_step_time": 8.59760856628418
    },
    {
      "epoch": 2.134990692138672e-05,
      "model_forward_time": 1.7503502368927002,
      "step": 96387
    },
    {
      "epoch": 2.134990692138672e-05,
      "step": 96387,
      "training_step_time": 8.358696699142456
    },
    {
      "epoch": 2.1350860595703127e-05,
      "model_forward_time": 1.755831003189087,
      "step": 96388
    },
    {
      "epoch": 2.1350860595703127e-05,
      "step": 96388,
      "training_step_time": 7.757357597351074
    },
    {
      "epoch": 2.135181427001953e-05,
      "model_forward_time": 1.7437143325805664,
      "step": 96389
    },
    {
      "epoch": 2.135181427001953e-05,
      "step": 96389,
      "training_step_time": 7.871673583984375
    },
    {
      "epoch": 2.1352767944335938e-05,
      "grad_norm": 0.12977664172649384,
      "learning_rate": 3.5606672221391024e-08,
      "loss": 0.0548,
      "step": 96390
    },
    {
      "epoch": 2.1352767944335938e-05,
      "model_forward_time": 1.7479004859924316,
      "step": 96390
    },
    {
      "dynamics_loss_avg": 0.048628483526408674,
      "epoch": 2.1352767944335938e-05,
      "step": 96390
    },
    {
      "action_loss_avg": 0.0011119527887785807,
      "epoch": 2.1352767944335938e-05,
      "step": 96390
    },
    {
      "epoch": 2.1352767944335938e-05,
      "step": 96390,
      "training_step_time": 7.765477418899536
    },
    {
      "epoch": 2.1353721618652342e-05,
      "model_forward_time": 1.3687655925750732,
      "step": 96391
    },
    {
      "epoch": 2.1353721618652342e-05,
      "step": 96391,
      "training_step_time": 7.777709484100342
    },
    {
      "epoch": 2.135467529296875e-05,
      "model_forward_time": 1.7462916374206543,
      "step": 96392
    },
    {
      "epoch": 2.135467529296875e-05,
      "step": 96392,
      "training_step_time": 8.63536024093628
    },
    {
      "epoch": 2.1355628967285157e-05,
      "model_forward_time": 2.483525514602661,
      "step": 96393
    },
    {
      "epoch": 2.1355628967285157e-05,
      "step": 96393,
      "training_step_time": 8.428560733795166
    },
    {
      "epoch": 2.135658264160156e-05,
      "model_forward_time": 1.7506649494171143,
      "step": 96394
    },
    {
      "epoch": 2.135658264160156e-05,
      "step": 96394,
      "training_step_time": 9.331552743911743
    },
    {
      "epoch": 2.135753631591797e-05,
      "model_forward_time": 1.7603404521942139,
      "step": 96395
    },
    {
      "epoch": 2.135753631591797e-05,
      "step": 96395,
      "training_step_time": 7.55422830581665
    },
    {
      "epoch": 2.1358489990234376e-05,
      "model_forward_time": 1.7452642917633057,
      "step": 96396
    },
    {
      "epoch": 2.1358489990234376e-05,
      "step": 96396,
      "training_step_time": 7.691198348999023
    },
    {
      "epoch": 2.135944366455078e-05,
      "model_forward_time": 1.745668888092041,
      "step": 96397
    },
    {
      "epoch": 2.135944366455078e-05,
      "step": 96397,
      "training_step_time": 9.542249202728271
    },
    {
      "epoch": 2.1360397338867187e-05,
      "model_forward_time": 2.824018955230713,
      "step": 96398
    },
    {
      "epoch": 2.1360397338867187e-05,
      "step": 96398,
      "training_step_time": 9.353118658065796
    },
    {
      "epoch": 2.1361351013183595e-05,
      "model_forward_time": 1.7422969341278076,
      "step": 96399
    },
    {
      "epoch": 2.1361351013183595e-05,
      "step": 96399,
      "training_step_time": 7.921030759811401
    },
    {
      "epoch": 2.13623046875e-05,
      "grad_norm": 0.07341650128364563,
      "learning_rate": 3.540996582472089e-08,
      "loss": 0.0541,
      "step": 96400
    },
    {
      "epoch": 2.13623046875e-05,
      "model_forward_time": 1.7551805973052979,
      "step": 96400
    },
    {
      "dynamics_loss_avg": 0.07121273241937161,
      "epoch": 2.13623046875e-05,
      "step": 96400
    },
    {
      "action_loss_avg": 0.0016113294870592654,
      "epoch": 2.13623046875e-05,
      "step": 96400
    },
    {
      "epoch": 2.13623046875e-05,
      "step": 96400,
      "training_step_time": 6.9518444538116455
    },
    {
      "epoch": 2.1363258361816406e-05,
      "model_forward_time": 1.7503106594085693,
      "step": 96401
    },
    {
      "epoch": 2.1363258361816406e-05,
      "step": 96401,
      "training_step_time": 8.055809736251831
    },
    {
      "epoch": 2.1364212036132813e-05,
      "model_forward_time": 1.7426567077636719,
      "step": 96402
    },
    {
      "epoch": 2.1364212036132813e-05,
      "step": 96402,
      "training_step_time": 7.558484077453613
    },
    {
      "epoch": 2.1365165710449218e-05,
      "model_forward_time": 1.7441248893737793,
      "step": 96403
    },
    {
      "epoch": 2.1365165710449218e-05,
      "step": 96403,
      "training_step_time": 6.951058864593506
    },
    {
      "epoch": 2.1366119384765625e-05,
      "model_forward_time": 1.7515654563903809,
      "step": 96404
    },
    {
      "epoch": 2.1366119384765625e-05,
      "step": 96404,
      "training_step_time": 6.9764087200164795
    },
    {
      "epoch": 2.1367073059082032e-05,
      "model_forward_time": 1.3793354034423828,
      "step": 96405
    },
    {
      "epoch": 2.1367073059082032e-05,
      "step": 96405,
      "training_step_time": 6.950218200683594
    },
    {
      "epoch": 2.1368026733398436e-05,
      "model_forward_time": 1.7477304935455322,
      "step": 96406
    },
    {
      "epoch": 2.1368026733398436e-05,
      "step": 96406,
      "training_step_time": 8.025266408920288
    },
    {
      "epoch": 2.1368980407714844e-05,
      "model_forward_time": 2.383920669555664,
      "step": 96407
    },
    {
      "epoch": 2.1368980407714844e-05,
      "step": 96407,
      "training_step_time": 11.016374111175537
    },
    {
      "epoch": 2.136993408203125e-05,
      "model_forward_time": 1.7479972839355469,
      "step": 96408
    },
    {
      "epoch": 2.136993408203125e-05,
      "step": 96408,
      "training_step_time": 8.70932412147522
    },
    {
      "epoch": 2.1370887756347655e-05,
      "model_forward_time": 1.7458021640777588,
      "step": 96409
    },
    {
      "epoch": 2.1370887756347655e-05,
      "step": 96409,
      "training_step_time": 8.580942153930664
    },
    {
      "epoch": 2.1371841430664063e-05,
      "grad_norm": 0.0875651016831398,
      "learning_rate": 3.521380234815297e-08,
      "loss": 0.0535,
      "step": 96410
    },
    {
      "epoch": 2.1371841430664063e-05,
      "model_forward_time": 1.743764877319336,
      "step": 96410
    },
    {
      "dynamics_loss_avg": 0.04257939918898046,
      "epoch": 2.1371841430664063e-05,
      "step": 96410
    },
    {
      "action_loss_avg": 0.0010160110163269565,
      "epoch": 2.1371841430664063e-05,
      "step": 96410
    },
    {
      "epoch": 2.1371841430664063e-05,
      "step": 96410,
      "training_step_time": 8.627386569976807
    },
    {
      "epoch": 2.137279510498047e-05,
      "model_forward_time": 1.744584321975708,
      "step": 96411
    },
    {
      "epoch": 2.137279510498047e-05,
      "step": 96411,
      "training_step_time": 7.766900062561035
    },
    {
      "epoch": 2.1373748779296874e-05,
      "model_forward_time": 1.7506463527679443,
      "step": 96412
    },
    {
      "epoch": 2.1373748779296874e-05,
      "step": 96412,
      "training_step_time": 6.944706201553345
    },
    {
      "epoch": 2.137470245361328e-05,
      "model_forward_time": 1.010815143585205,
      "step": 96413
    },
    {
      "epoch": 2.137470245361328e-05,
      "step": 96413,
      "training_step_time": 6.941466331481934
    },
    {
      "epoch": 2.137565612792969e-05,
      "model_forward_time": 1.740323781967163,
      "step": 96414
    },
    {
      "epoch": 2.137565612792969e-05,
      "step": 96414,
      "training_step_time": 6.982082366943359
    },
    {
      "epoch": 2.1376609802246093e-05,
      "model_forward_time": 1.7453362941741943,
      "step": 96415
    },
    {
      "epoch": 2.1376609802246093e-05,
      "step": 96415,
      "training_step_time": 6.871845722198486
    },
    {
      "epoch": 2.13775634765625e-05,
      "model_forward_time": 2.3647243976593018,
      "step": 96416
    },
    {
      "epoch": 2.13775634765625e-05,
      "step": 96416,
      "training_step_time": 9.010899543762207
    },
    {
      "epoch": 2.1378517150878908e-05,
      "model_forward_time": 1.7479918003082275,
      "step": 96417
    },
    {
      "epoch": 2.1378517150878908e-05,
      "step": 96417,
      "training_step_time": 7.920284748077393
    },
    {
      "epoch": 2.1379470825195312e-05,
      "model_forward_time": 1.7443020343780518,
      "step": 96418
    },
    {
      "epoch": 2.1379470825195312e-05,
      "step": 96418,
      "training_step_time": 7.7671027183532715
    },
    {
      "epoch": 2.138042449951172e-05,
      "model_forward_time": 1.7366273403167725,
      "step": 96419
    },
    {
      "epoch": 2.138042449951172e-05,
      "step": 96419,
      "training_step_time": 7.686058521270752
    },
    {
      "epoch": 2.1381378173828127e-05,
      "grad_norm": 0.09097430109977722,
      "learning_rate": 3.501818181313899e-08,
      "loss": 0.0541,
      "step": 96420
    },
    {
      "epoch": 2.1381378173828127e-05,
      "model_forward_time": 1.7223739624023438,
      "step": 96420
    },
    {
      "dynamics_loss_avg": 0.05223751664161682,
      "epoch": 2.1381378173828127e-05,
      "step": 96420
    },
    {
      "action_loss_avg": 0.001693358685588464,
      "epoch": 2.1381378173828127e-05,
      "step": 96420
    },
    {
      "epoch": 2.1381378173828127e-05,
      "step": 96420,
      "training_step_time": 7.718211889266968
    },
    {
      "epoch": 2.138233184814453e-05,
      "model_forward_time": 1.7517833709716797,
      "step": 96421
    },
    {
      "epoch": 2.138233184814453e-05,
      "step": 96421,
      "training_step_time": 6.94342827796936
    },
    {
      "epoch": 2.1383285522460938e-05,
      "model_forward_time": 1.0221445560455322,
      "step": 96422
    },
    {
      "epoch": 2.1383285522460938e-05,
      "step": 96422,
      "training_step_time": 6.947728872299194
    },
    {
      "epoch": 2.1384239196777342e-05,
      "model_forward_time": 1.7479403018951416,
      "step": 96423
    },
    {
      "epoch": 2.1384239196777342e-05,
      "step": 96423,
      "training_step_time": 6.966362237930298
    },
    {
      "epoch": 2.138519287109375e-05,
      "model_forward_time": 1.7346155643463135,
      "step": 96424
    },
    {
      "epoch": 2.138519287109375e-05,
      "step": 96424,
      "training_step_time": 6.934974908828735
    },
    {
      "epoch": 2.1386146545410157e-05,
      "model_forward_time": 1.7804961204528809,
      "step": 96425
    },
    {
      "epoch": 2.1386146545410157e-05,
      "step": 96425,
      "training_step_time": 8.505740404129028
    },
    {
      "epoch": 2.138710021972656e-05,
      "model_forward_time": 1.7437055110931396,
      "step": 96426
    },
    {
      "epoch": 2.138710021972656e-05,
      "step": 96426,
      "training_step_time": 9.488276243209839
    },
    {
      "epoch": 2.138805389404297e-05,
      "model_forward_time": 1.7530550956726074,
      "step": 96427
    },
    {
      "epoch": 2.138805389404297e-05,
      "step": 96427,
      "training_step_time": 7.643856763839722
    },
    {
      "epoch": 2.1389007568359376e-05,
      "model_forward_time": 1.7557954788208008,
      "step": 96428
    },
    {
      "epoch": 2.1389007568359376e-05,
      "step": 96428,
      "training_step_time": 7.70460844039917
    },
    {
      "epoch": 2.138996124267578e-05,
      "model_forward_time": 1.7434616088867188,
      "step": 96429
    },
    {
      "epoch": 2.138996124267578e-05,
      "step": 96429,
      "training_step_time": 7.718106746673584
    },
    {
      "epoch": 2.1390914916992187e-05,
      "grad_norm": 0.09302890300750732,
      "learning_rate": 3.482310424107238e-08,
      "loss": 0.0535,
      "step": 96430
    },
    {
      "epoch": 2.1390914916992187e-05,
      "model_forward_time": 1.751162052154541,
      "step": 96430
    },
    {
      "dynamics_loss_avg": 0.048643107805401084,
      "epoch": 2.1390914916992187e-05,
      "step": 96430
    },
    {
      "action_loss_avg": 0.0011265456705586985,
      "epoch": 2.1390914916992187e-05,
      "step": 96430
    },
    {
      "epoch": 2.1390914916992187e-05,
      "step": 96430,
      "training_step_time": 6.953496932983398
    },
    {
      "epoch": 2.1391868591308595e-05,
      "model_forward_time": 1.7557811737060547,
      "step": 96431
    },
    {
      "epoch": 2.1391868591308595e-05,
      "step": 96431,
      "training_step_time": 6.907510042190552
    },
    {
      "epoch": 2.1392822265625e-05,
      "model_forward_time": 1.7615816593170166,
      "step": 96432
    },
    {
      "epoch": 2.1392822265625e-05,
      "step": 96432,
      "training_step_time": 6.889394044876099
    },
    {
      "epoch": 2.1393775939941406e-05,
      "model_forward_time": 1.757652759552002,
      "step": 96433
    },
    {
      "epoch": 2.1393775939941406e-05,
      "step": 96433,
      "training_step_time": 6.926467418670654
    },
    {
      "epoch": 2.1394729614257814e-05,
      "model_forward_time": 1.760180950164795,
      "step": 96434
    },
    {
      "epoch": 2.1394729614257814e-05,
      "step": 96434,
      "training_step_time": 8.290043830871582
    },
    {
      "epoch": 2.1395683288574218e-05,
      "model_forward_time": 2.406123399734497,
      "step": 96435
    },
    {
      "epoch": 2.1395683288574218e-05,
      "step": 96435,
      "training_step_time": 9.441825151443481
    },
    {
      "epoch": 2.1396636962890625e-05,
      "model_forward_time": 1.761331558227539,
      "step": 96436
    },
    {
      "epoch": 2.1396636962890625e-05,
      "step": 96436,
      "training_step_time": 7.806185245513916
    },
    {
      "epoch": 2.1397590637207032e-05,
      "model_forward_time": 1.7499535083770752,
      "step": 96437
    },
    {
      "epoch": 2.1397590637207032e-05,
      "step": 96437,
      "training_step_time": 7.685796022415161
    },
    {
      "epoch": 2.1398544311523436e-05,
      "model_forward_time": 1.7473676204681396,
      "step": 96438
    },
    {
      "epoch": 2.1398544311523436e-05,
      "step": 96438,
      "training_step_time": 7.702292203903198
    },
    {
      "epoch": 2.1399497985839844e-05,
      "model_forward_time": 1.7575392723083496,
      "step": 96439
    },
    {
      "epoch": 2.1399497985839844e-05,
      "step": 96439,
      "training_step_time": 6.946364402770996
    },
    {
      "epoch": 2.140045166015625e-05,
      "grad_norm": 0.08166999369859695,
      "learning_rate": 3.462856965328554e-08,
      "loss": 0.0563,
      "step": 96440
    },
    {
      "epoch": 2.140045166015625e-05,
      "model_forward_time": 1.745873212814331,
      "step": 96440
    },
    {
      "dynamics_loss_avg": 0.048400606215000155,
      "epoch": 2.140045166015625e-05,
      "step": 96440
    },
    {
      "action_loss_avg": 0.0010424721040180884,
      "epoch": 2.140045166015625e-05,
      "step": 96440
    },
    {
      "epoch": 2.140045166015625e-05,
      "step": 96440,
      "training_step_time": 6.936733961105347
    },
    {
      "epoch": 2.1401405334472655e-05,
      "model_forward_time": 1.7570345401763916,
      "step": 96441
    },
    {
      "epoch": 2.1401405334472655e-05,
      "step": 96441,
      "training_step_time": 6.907817602157593
    },
    {
      "epoch": 2.1402359008789063e-05,
      "model_forward_time": 1.7479357719421387,
      "step": 96442
    },
    {
      "epoch": 2.1402359008789063e-05,
      "step": 96442,
      "training_step_time": 6.923303842544556
    },
    {
      "epoch": 2.140331268310547e-05,
      "model_forward_time": 1.7595210075378418,
      "step": 96443
    },
    {
      "epoch": 2.140331268310547e-05,
      "step": 96443,
      "training_step_time": 7.717995643615723
    },
    {
      "epoch": 2.1404266357421874e-05,
      "model_forward_time": 2.4163196086883545,
      "step": 96444
    },
    {
      "epoch": 2.1404266357421874e-05,
      "step": 96444,
      "training_step_time": 9.875853300094604
    },
    {
      "epoch": 2.140522003173828e-05,
      "model_forward_time": 1.7609670162200928,
      "step": 96445
    },
    {
      "epoch": 2.140522003173828e-05,
      "step": 96445,
      "training_step_time": 7.873291492462158
    },
    {
      "epoch": 2.140617370605469e-05,
      "model_forward_time": 1.769007921218872,
      "step": 96446
    },
    {
      "epoch": 2.140617370605469e-05,
      "step": 96446,
      "training_step_time": 7.7311131954193115
    },
    {
      "epoch": 2.1407127380371093e-05,
      "model_forward_time": 1.77024245262146,
      "step": 96447
    },
    {
      "epoch": 2.1407127380371093e-05,
      "step": 96447,
      "training_step_time": 7.8025572299957275
    },
    {
      "epoch": 2.14080810546875e-05,
      "model_forward_time": 1.749053716659546,
      "step": 96448
    },
    {
      "epoch": 2.14080810546875e-05,
      "step": 96448,
      "training_step_time": 6.891234397888184
    },
    {
      "epoch": 2.1409034729003908e-05,
      "model_forward_time": 1.7493534088134766,
      "step": 96449
    },
    {
      "epoch": 2.1409034729003908e-05,
      "step": 96449,
      "training_step_time": 6.932481288909912
    },
    {
      "epoch": 2.1409988403320312e-05,
      "grad_norm": 0.07642076909542084,
      "learning_rate": 3.44345780710531e-08,
      "loss": 0.0545,
      "step": 96450
    },
    {
      "epoch": 2.1409988403320312e-05,
      "model_forward_time": 1.7480218410491943,
      "step": 96450
    },
    {
      "dynamics_loss_avg": 0.05606314847245812,
      "epoch": 2.1409988403320312e-05,
      "step": 96450
    },
    {
      "action_loss_avg": 0.0009919608244672417,
      "epoch": 2.1409988403320312e-05,
      "step": 96450
    },
    {
      "epoch": 2.1409988403320312e-05,
      "step": 96450,
      "training_step_time": 6.9316346645355225
    },
    {
      "epoch": 2.141094207763672e-05,
      "model_forward_time": 1.7539970874786377,
      "step": 96451
    },
    {
      "epoch": 2.141094207763672e-05,
      "step": 96451,
      "training_step_time": 6.959601879119873
    },
    {
      "epoch": 2.1411895751953127e-05,
      "model_forward_time": 1.756472110748291,
      "step": 96452
    },
    {
      "epoch": 2.1411895751953127e-05,
      "step": 96452,
      "training_step_time": 8.294012308120728
    },
    {
      "epoch": 2.141284942626953e-05,
      "model_forward_time": 1.9956088066101074,
      "step": 96453
    },
    {
      "epoch": 2.141284942626953e-05,
      "step": 96453,
      "training_step_time": 10.685657739639282
    },
    {
      "epoch": 2.1413803100585938e-05,
      "model_forward_time": 1.7527508735656738,
      "step": 96454
    },
    {
      "epoch": 2.1413803100585938e-05,
      "step": 96454,
      "training_step_time": 7.843602657318115
    },
    {
      "epoch": 2.1414756774902342e-05,
      "model_forward_time": 1.7505240440368652,
      "step": 96455
    },
    {
      "epoch": 2.1414756774902342e-05,
      "step": 96455,
      "training_step_time": 8.326741218566895
    },
    {
      "epoch": 2.141571044921875e-05,
      "model_forward_time": 1.7601845264434814,
      "step": 96456
    },
    {
      "epoch": 2.141571044921875e-05,
      "step": 96456,
      "training_step_time": 8.316140174865723
    },
    {
      "epoch": 2.1416664123535157e-05,
      "model_forward_time": 1.7513127326965332,
      "step": 96457
    },
    {
      "epoch": 2.1416664123535157e-05,
      "step": 96457,
      "training_step_time": 7.6497886180877686
    },
    {
      "epoch": 2.141761779785156e-05,
      "model_forward_time": 1.7467496395111084,
      "step": 96458
    },
    {
      "epoch": 2.141761779785156e-05,
      "step": 96458,
      "training_step_time": 6.958459138870239
    },
    {
      "epoch": 2.141857147216797e-05,
      "model_forward_time": 1.7246580123901367,
      "step": 96459
    },
    {
      "epoch": 2.141857147216797e-05,
      "step": 96459,
      "training_step_time": 6.9351866245269775
    },
    {
      "epoch": 2.1419525146484376e-05,
      "grad_norm": 0.06569419801235199,
      "learning_rate": 3.4241129515589776e-08,
      "loss": 0.0569,
      "step": 96460
    },
    {
      "epoch": 2.1419525146484376e-05,
      "model_forward_time": 1.3526344299316406,
      "step": 96460
    },
    {
      "dynamics_loss_avg": 0.0498358903452754,
      "epoch": 2.1419525146484376e-05,
      "step": 96460
    },
    {
      "action_loss_avg": 0.0016597588954027742,
      "epoch": 2.1419525146484376e-05,
      "step": 96460
    },
    {
      "epoch": 2.1419525146484376e-05,
      "step": 96460,
      "training_step_time": 6.940068483352661
    },
    {
      "epoch": 2.142047882080078e-05,
      "model_forward_time": 1.73069167137146,
      "step": 96461
    },
    {
      "epoch": 2.142047882080078e-05,
      "step": 96461,
      "training_step_time": 7.504185438156128
    },
    {
      "epoch": 2.1421432495117187e-05,
      "model_forward_time": 2.383185625076294,
      "step": 96462
    },
    {
      "epoch": 2.1421432495117187e-05,
      "step": 96462,
      "training_step_time": 9.660520315170288
    },
    {
      "epoch": 2.1422386169433595e-05,
      "model_forward_time": 1.7350006103515625,
      "step": 96463
    },
    {
      "epoch": 2.1422386169433595e-05,
      "step": 96463,
      "training_step_time": 7.821119070053101
    },
    {
      "epoch": 2.142333984375e-05,
      "model_forward_time": 1.7319741249084473,
      "step": 96464
    },
    {
      "epoch": 2.142333984375e-05,
      "step": 96464,
      "training_step_time": 8.600131273269653
    },
    {
      "epoch": 2.1424293518066406e-05,
      "model_forward_time": 1.7283241748809814,
      "step": 96465
    },
    {
      "epoch": 2.1424293518066406e-05,
      "step": 96465,
      "training_step_time": 7.690777540206909
    },
    {
      "epoch": 2.1425247192382814e-05,
      "model_forward_time": 1.7338082790374756,
      "step": 96466
    },
    {
      "epoch": 2.1425247192382814e-05,
      "step": 96466,
      "training_step_time": 7.6781907081604
    },
    {
      "epoch": 2.1426200866699218e-05,
      "model_forward_time": 1.7297179698944092,
      "step": 96467
    },
    {
      "epoch": 2.1426200866699218e-05,
      "step": 96467,
      "training_step_time": 6.961063623428345
    },
    {
      "epoch": 2.1427154541015625e-05,
      "model_forward_time": 1.7306137084960938,
      "step": 96468
    },
    {
      "epoch": 2.1427154541015625e-05,
      "step": 96468,
      "training_step_time": 7.01868748664856
    },
    {
      "epoch": 2.1428108215332032e-05,
      "model_forward_time": 1.730273723602295,
      "step": 96469
    },
    {
      "epoch": 2.1428108215332032e-05,
      "step": 96469,
      "training_step_time": 6.9865663051605225
    },
    {
      "epoch": 2.1429061889648436e-05,
      "grad_norm": 0.242936909198761,
      "learning_rate": 3.40482240080503e-08,
      "loss": 0.0552,
      "step": 96470
    },
    {
      "epoch": 2.1429061889648436e-05,
      "model_forward_time": 1.722846508026123,
      "step": 96470
    },
    {
      "dynamics_loss_avg": 0.044799096137285235,
      "epoch": 2.1429061889648436e-05,
      "step": 96470
    },
    {
      "action_loss_avg": 0.001192119054030627,
      "epoch": 2.1429061889648436e-05,
      "step": 96470
    },
    {
      "epoch": 2.1429061889648436e-05,
      "step": 96470,
      "training_step_time": 6.94503378868103
    },
    {
      "epoch": 2.1430015563964844e-05,
      "model_forward_time": 1.7520055770874023,
      "step": 96471
    },
    {
      "epoch": 2.1430015563964844e-05,
      "step": 96471,
      "training_step_time": 8.739667415618896
    },
    {
      "epoch": 2.143096923828125e-05,
      "model_forward_time": 2.386753797531128,
      "step": 96472
    },
    {
      "epoch": 2.143096923828125e-05,
      "step": 96472,
      "training_step_time": 8.9386465549469
    },
    {
      "epoch": 2.1431922912597655e-05,
      "model_forward_time": 1.3460915088653564,
      "step": 96473
    },
    {
      "epoch": 2.1431922912597655e-05,
      "step": 96473,
      "training_step_time": 7.724492788314819
    },
    {
      "epoch": 2.1432876586914063e-05,
      "model_forward_time": 1.73860764503479,
      "step": 96474
    },
    {
      "epoch": 2.1432876586914063e-05,
      "step": 96474,
      "training_step_time": 7.7333409786224365
    },
    {
      "epoch": 2.143383026123047e-05,
      "model_forward_time": 1.349642038345337,
      "step": 96475
    },
    {
      "epoch": 2.143383026123047e-05,
      "step": 96475,
      "training_step_time": 7.666635274887085
    },
    {
      "epoch": 2.1434783935546874e-05,
      "model_forward_time": 1.7250480651855469,
      "step": 96476
    },
    {
      "epoch": 2.1434783935546874e-05,
      "step": 96476,
      "training_step_time": 7.570188999176025
    },
    {
      "epoch": 2.143573760986328e-05,
      "model_forward_time": 1.72733473777771,
      "step": 96477
    },
    {
      "epoch": 2.143573760986328e-05,
      "step": 96477,
      "training_step_time": 6.968287944793701
    },
    {
      "epoch": 2.143669128417969e-05,
      "model_forward_time": 1.7362232208251953,
      "step": 96478
    },
    {
      "epoch": 2.143669128417969e-05,
      "step": 96478,
      "training_step_time": 6.961878299713135
    },
    {
      "epoch": 2.1437644958496093e-05,
      "model_forward_time": 1.7298240661621094,
      "step": 96479
    },
    {
      "epoch": 2.1437644958496093e-05,
      "step": 96479,
      "training_step_time": 6.973533868789673
    },
    {
      "epoch": 2.14385986328125e-05,
      "grad_norm": 0.12686064839363098,
      "learning_rate": 3.385586156953114e-08,
      "loss": 0.0554,
      "step": 96480
    },
    {
      "epoch": 2.14385986328125e-05,
      "model_forward_time": 1.3643238544464111,
      "step": 96480
    },
    {
      "dynamics_loss_avg": 0.04872905593365431,
      "epoch": 2.14385986328125e-05,
      "step": 96480
    },
    {
      "action_loss_avg": 0.0015831663797143847,
      "epoch": 2.14385986328125e-05,
      "step": 96480
    },
    {
      "epoch": 2.14385986328125e-05,
      "step": 96480,
      "training_step_time": 9.678677797317505
    },
    {
      "epoch": 2.1439552307128908e-05,
      "model_forward_time": 2.41155743598938,
      "step": 96481
    },
    {
      "epoch": 2.1439552307128908e-05,
      "step": 96481,
      "training_step_time": 11.722517251968384
    },
    {
      "epoch": 2.1440505981445312e-05,
      "model_forward_time": 1.7193427085876465,
      "step": 96482
    },
    {
      "epoch": 2.1440505981445312e-05,
      "step": 96482,
      "training_step_time": 7.859656810760498
    },
    {
      "epoch": 2.144145965576172e-05,
      "model_forward_time": 1.7276906967163086,
      "step": 96483
    },
    {
      "epoch": 2.144145965576172e-05,
      "step": 96483,
      "training_step_time": 8.317969560623169
    },
    {
      "epoch": 2.1442413330078127e-05,
      "model_forward_time": 1.7241978645324707,
      "step": 96484
    },
    {
      "epoch": 2.1442413330078127e-05,
      "step": 96484,
      "training_step_time": 7.726026773452759
    },
    {
      "epoch": 2.144336700439453e-05,
      "model_forward_time": 1.728806972503662,
      "step": 96485
    },
    {
      "epoch": 2.144336700439453e-05,
      "step": 96485,
      "training_step_time": 7.586342811584473
    },
    {
      "epoch": 2.1444320678710938e-05,
      "model_forward_time": 1.7346696853637695,
      "step": 96486
    },
    {
      "epoch": 2.1444320678710938e-05,
      "step": 96486,
      "training_step_time": 7.014413356781006
    },
    {
      "epoch": 2.1445274353027342e-05,
      "model_forward_time": 1.7286884784698486,
      "step": 96487
    },
    {
      "epoch": 2.1445274353027342e-05,
      "step": 96487,
      "training_step_time": 6.933879375457764
    },
    {
      "epoch": 2.144622802734375e-05,
      "model_forward_time": 1.7329418659210205,
      "step": 96488
    },
    {
      "epoch": 2.144622802734375e-05,
      "step": 96488,
      "training_step_time": 6.95125675201416
    },
    {
      "epoch": 2.1447181701660157e-05,
      "model_forward_time": 1.7556419372558594,
      "step": 96489
    },
    {
      "epoch": 2.1447181701660157e-05,
      "step": 96489,
      "training_step_time": 7.771056175231934
    },
    {
      "epoch": 2.144813537597656e-05,
      "grad_norm": 0.3050386309623718,
      "learning_rate": 3.366404222106823e-08,
      "loss": 0.0553,
      "step": 96490
    },
    {
      "epoch": 2.144813537597656e-05,
      "model_forward_time": 2.364243745803833,
      "step": 96490
    },
    {
      "dynamics_loss_avg": 0.05902612451463938,
      "epoch": 2.144813537597656e-05,
      "step": 96490
    },
    {
      "action_loss_avg": 0.0010563152958638967,
      "epoch": 2.144813537597656e-05,
      "step": 96490
    },
    {
      "epoch": 2.144813537597656e-05,
      "step": 96490,
      "training_step_time": 9.553448677062988
    },
    {
      "epoch": 2.144908905029297e-05,
      "model_forward_time": 1.7303218841552734,
      "step": 96491
    },
    {
      "epoch": 2.144908905029297e-05,
      "step": 96491,
      "training_step_time": 8.844415664672852
    },
    {
      "epoch": 2.1450042724609376e-05,
      "model_forward_time": 1.7334954738616943,
      "step": 96492
    },
    {
      "epoch": 2.1450042724609376e-05,
      "step": 96492,
      "training_step_time": 8.27066421508789
    },
    {
      "epoch": 2.145099639892578e-05,
      "model_forward_time": 1.7374775409698486,
      "step": 96493
    },
    {
      "epoch": 2.145099639892578e-05,
      "step": 96493,
      "training_step_time": 7.506184101104736
    },
    {
      "epoch": 2.1451950073242187e-05,
      "model_forward_time": 1.740088701248169,
      "step": 96494
    },
    {
      "epoch": 2.1451950073242187e-05,
      "step": 96494,
      "training_step_time": 7.547596216201782
    },
    {
      "epoch": 2.1452903747558595e-05,
      "model_forward_time": 1.7389070987701416,
      "step": 96495
    },
    {
      "epoch": 2.1452903747558595e-05,
      "step": 96495,
      "training_step_time": 6.921052932739258
    },
    {
      "epoch": 2.1453857421875e-05,
      "model_forward_time": 1.7398285865783691,
      "step": 96496
    },
    {
      "epoch": 2.1453857421875e-05,
      "step": 96496,
      "training_step_time": 6.932446718215942
    },
    {
      "epoch": 2.1454811096191406e-05,
      "model_forward_time": 1.7388465404510498,
      "step": 96497
    },
    {
      "epoch": 2.1454811096191406e-05,
      "step": 96497,
      "training_step_time": 6.971380949020386
    },
    {
      "epoch": 2.1455764770507814e-05,
      "model_forward_time": 1.8095595836639404,
      "step": 96498
    },
    {
      "epoch": 2.1455764770507814e-05,
      "step": 96498,
      "training_step_time": 7.220574378967285
    },
    {
      "epoch": 2.1456718444824218e-05,
      "model_forward_time": 1.7341229915618896,
      "step": 96499
    },
    {
      "epoch": 2.1456718444824218e-05,
      "step": 96499,
      "training_step_time": 9.724190950393677
    },
    {
      "epoch": 2.1457672119140625e-05,
      "grad_norm": 0.07316946983337402,
      "learning_rate": 3.3472765983639244e-08,
      "loss": 0.0539,
      "step": 96500
    },
    {
      "epoch": 2.1457672119140625e-05,
      "model_forward_time": 2.4029340744018555,
      "step": 96500
    },
    {
      "dynamics_loss_avg": 0.05821022428572178,
      "epoch": 2.1457672119140625e-05,
      "step": 96500
    },
    {
      "action_loss_avg": 0.0015340372425271197,
      "epoch": 2.1457672119140625e-05,
      "step": 96500
    },
    {
      "epoch": 2.1457672119140625e-05,
      "step": 96500,
      "training_step_time": 9.719175100326538
    },
    {
      "epoch": 2.1458625793457032e-05,
      "model_forward_time": 1.7328338623046875,
      "step": 96501
    },
    {
      "epoch": 2.1458625793457032e-05,
      "step": 96501,
      "training_step_time": 7.784173250198364
    },
    {
      "epoch": 2.1459579467773436e-05,
      "model_forward_time": 1.7376959323883057,
      "step": 96502
    },
    {
      "epoch": 2.1459579467773436e-05,
      "step": 96502,
      "training_step_time": 7.732825994491577
    },
    {
      "epoch": 2.1460533142089844e-05,
      "model_forward_time": 1.7313203811645508,
      "step": 96503
    },
    {
      "epoch": 2.1460533142089844e-05,
      "step": 96503,
      "training_step_time": 7.585543870925903
    },
    {
      "epoch": 2.146148681640625e-05,
      "model_forward_time": 1.7320210933685303,
      "step": 96504
    },
    {
      "epoch": 2.146148681640625e-05,
      "step": 96504,
      "training_step_time": 6.911829948425293
    },
    {
      "epoch": 2.1462440490722655e-05,
      "model_forward_time": 1.7333714962005615,
      "step": 96505
    },
    {
      "epoch": 2.1462440490722655e-05,
      "step": 96505,
      "training_step_time": 6.920565843582153
    },
    {
      "epoch": 2.1463394165039063e-05,
      "model_forward_time": 1.3645637035369873,
      "step": 96506
    },
    {
      "epoch": 2.1463394165039063e-05,
      "step": 96506,
      "training_step_time": 6.968657493591309
    },
    {
      "epoch": 2.146434783935547e-05,
      "model_forward_time": 1.9198718070983887,
      "step": 96507
    },
    {
      "epoch": 2.146434783935547e-05,
      "step": 96507,
      "training_step_time": 7.737573862075806
    },
    {
      "epoch": 2.1465301513671874e-05,
      "model_forward_time": 1.9104080200195312,
      "step": 96508
    },
    {
      "epoch": 2.1465301513671874e-05,
      "step": 96508,
      "training_step_time": 8.456341981887817
    },
    {
      "epoch": 2.146625518798828e-05,
      "model_forward_time": 2.4429314136505127,
      "step": 96509
    },
    {
      "epoch": 2.146625518798828e-05,
      "step": 96509,
      "training_step_time": 13.532206058502197
    },
    {
      "epoch": 2.146720886230469e-05,
      "grad_norm": 0.07142885774374008,
      "learning_rate": 3.328203287816134e-08,
      "loss": 0.055,
      "step": 96510
    },
    {
      "epoch": 2.146720886230469e-05,
      "model_forward_time": 1.7477247714996338,
      "step": 96510
    },
    {
      "dynamics_loss_avg": 0.04767258018255234,
      "epoch": 2.146720886230469e-05,
      "step": 96510
    },
    {
      "action_loss_avg": 0.0014568801372661256,
      "epoch": 2.146720886230469e-05,
      "step": 96510
    },
    {
      "epoch": 2.146720886230469e-05,
      "step": 96510,
      "training_step_time": 8.361796379089355
    },
    {
      "epoch": 2.1468162536621093e-05,
      "model_forward_time": 1.7574975490570068,
      "step": 96511
    },
    {
      "epoch": 2.1468162536621093e-05,
      "step": 96511,
      "training_step_time": 7.8149497509002686
    },
    {
      "epoch": 2.14691162109375e-05,
      "model_forward_time": 1.7552523612976074,
      "step": 96512
    },
    {
      "epoch": 2.14691162109375e-05,
      "step": 96512,
      "training_step_time": 7.653876781463623
    },
    {
      "epoch": 2.1470069885253908e-05,
      "model_forward_time": 1.7591490745544434,
      "step": 96513
    },
    {
      "epoch": 2.1470069885253908e-05,
      "step": 96513,
      "training_step_time": 7.576135635375977
    },
    {
      "epoch": 2.1471023559570312e-05,
      "model_forward_time": 1.366969347000122,
      "step": 96514
    },
    {
      "epoch": 2.1471023559570312e-05,
      "step": 96514,
      "training_step_time": 6.855474233627319
    },
    {
      "epoch": 2.147197723388672e-05,
      "model_forward_time": 1.7542688846588135,
      "step": 96515
    },
    {
      "epoch": 2.147197723388672e-05,
      "step": 96515,
      "training_step_time": 6.934181213378906
    },
    {
      "epoch": 2.1472930908203123e-05,
      "model_forward_time": 1.7563178539276123,
      "step": 96516
    },
    {
      "epoch": 2.1472930908203123e-05,
      "step": 96516,
      "training_step_time": 7.702586889266968
    },
    {
      "epoch": 2.147388458251953e-05,
      "model_forward_time": 1.8807063102722168,
      "step": 96517
    },
    {
      "epoch": 2.147388458251953e-05,
      "step": 96517,
      "training_step_time": 8.631440877914429
    },
    {
      "epoch": 2.1474838256835938e-05,
      "model_forward_time": 2.4638168811798096,
      "step": 96518
    },
    {
      "epoch": 2.1474838256835938e-05,
      "step": 96518,
      "training_step_time": 10.671758651733398
    },
    {
      "epoch": 2.1475791931152342e-05,
      "model_forward_time": 1.7645864486694336,
      "step": 96519
    },
    {
      "epoch": 2.1475791931152342e-05,
      "step": 96519,
      "training_step_time": 9.338416814804077
    },
    {
      "epoch": 2.147674560546875e-05,
      "grad_norm": 0.10592837631702423,
      "learning_rate": 3.309184292549339e-08,
      "loss": 0.0553,
      "step": 96520
    },
    {
      "epoch": 2.147674560546875e-05,
      "model_forward_time": 1.7542481422424316,
      "step": 96520
    },
    {
      "dynamics_loss_avg": 0.044822045043110846,
      "epoch": 2.147674560546875e-05,
      "step": 96520
    },
    {
      "action_loss_avg": 0.0009274193696910515,
      "epoch": 2.147674560546875e-05,
      "step": 96520
    },
    {
      "epoch": 2.147674560546875e-05,
      "step": 96520,
      "training_step_time": 8.254657983779907
    },
    {
      "epoch": 2.1477699279785157e-05,
      "model_forward_time": 1.7659616470336914,
      "step": 96521
    },
    {
      "epoch": 2.1477699279785157e-05,
      "step": 96521,
      "training_step_time": 8.621612787246704
    },
    {
      "epoch": 2.147865295410156e-05,
      "model_forward_time": 1.75895094871521,
      "step": 96522
    },
    {
      "epoch": 2.147865295410156e-05,
      "step": 96522,
      "training_step_time": 7.546551942825317
    },
    {
      "epoch": 2.147960662841797e-05,
      "model_forward_time": 1.764747142791748,
      "step": 96523
    },
    {
      "epoch": 2.147960662841797e-05,
      "step": 96523,
      "training_step_time": 6.948182106018066
    },
    {
      "epoch": 2.1480560302734376e-05,
      "model_forward_time": 1.7532687187194824,
      "step": 96524
    },
    {
      "epoch": 2.1480560302734376e-05,
      "step": 96524,
      "training_step_time": 6.915911912918091
    },
    {
      "epoch": 2.148151397705078e-05,
      "model_forward_time": 1.7491168975830078,
      "step": 96525
    },
    {
      "epoch": 2.148151397705078e-05,
      "step": 96525,
      "training_step_time": 7.7234368324279785
    },
    {
      "epoch": 2.1482467651367187e-05,
      "model_forward_time": 1.743253231048584,
      "step": 96526
    },
    {
      "epoch": 2.1482467651367187e-05,
      "step": 96526,
      "training_step_time": 7.7498459815979
    },
    {
      "epoch": 2.1483421325683595e-05,
      "model_forward_time": 1.7257168292999268,
      "step": 96527
    },
    {
      "epoch": 2.1483421325683595e-05,
      "step": 96527,
      "training_step_time": 13.023506879806519
    },
    {
      "epoch": 2.1484375e-05,
      "model_forward_time": 1.729668378829956,
      "step": 96528
    },
    {
      "epoch": 2.1484375e-05,
      "step": 96528,
      "training_step_time": 9.929796934127808
    },
    {
      "epoch": 2.1485328674316406e-05,
      "model_forward_time": 1.7291367053985596,
      "step": 96529
    },
    {
      "epoch": 2.1485328674316406e-05,
      "step": 96529,
      "training_step_time": 7.810556888580322
    },
    {
      "epoch": 2.1486282348632814e-05,
      "grad_norm": 0.10169749706983566,
      "learning_rate": 3.29021961464332e-08,
      "loss": 0.0547,
      "step": 96530
    },
    {
      "epoch": 2.1486282348632814e-05,
      "model_forward_time": 1.724999189376831,
      "step": 96530
    },
    {
      "dynamics_loss_avg": 0.04799553351476789,
      "epoch": 2.1486282348632814e-05,
      "step": 96530
    },
    {
      "action_loss_avg": 0.000921392918098718,
      "epoch": 2.1486282348632814e-05,
      "step": 96530
    },
    {
      "epoch": 2.1486282348632814e-05,
      "step": 96530,
      "training_step_time": 7.671725511550903
    },
    {
      "epoch": 2.1487236022949218e-05,
      "model_forward_time": 1.727846384048462,
      "step": 96531
    },
    {
      "epoch": 2.1487236022949218e-05,
      "step": 96531,
      "training_step_time": 7.579574823379517
    },
    {
      "epoch": 2.1488189697265625e-05,
      "model_forward_time": 1.7343652248382568,
      "step": 96532
    },
    {
      "epoch": 2.1488189697265625e-05,
      "step": 96532,
      "training_step_time": 6.917571306228638
    },
    {
      "epoch": 2.1489143371582032e-05,
      "model_forward_time": 1.7320339679718018,
      "step": 96533
    },
    {
      "epoch": 2.1489143371582032e-05,
      "step": 96533,
      "training_step_time": 6.919127702713013
    },
    {
      "epoch": 2.1490097045898436e-05,
      "model_forward_time": 1.737424373626709,
      "step": 96534
    },
    {
      "epoch": 2.1490097045898436e-05,
      "step": 96534,
      "training_step_time": 7.3518455028533936
    },
    {
      "epoch": 2.1491050720214844e-05,
      "model_forward_time": 1.7469580173492432,
      "step": 96535
    },
    {
      "epoch": 2.1491050720214844e-05,
      "step": 96535,
      "training_step_time": 7.880047798156738
    },
    {
      "epoch": 2.149200439453125e-05,
      "model_forward_time": 1.0141897201538086,
      "step": 96536
    },
    {
      "epoch": 2.149200439453125e-05,
      "step": 96536,
      "training_step_time": 9.946636438369751
    },
    {
      "epoch": 2.1492958068847655e-05,
      "model_forward_time": 2.4157943725585938,
      "step": 96537
    },
    {
      "epoch": 2.1492958068847655e-05,
      "step": 96537,
      "training_step_time": 9.605802774429321
    },
    {
      "epoch": 2.1493911743164063e-05,
      "model_forward_time": 1.72212815284729,
      "step": 96538
    },
    {
      "epoch": 2.1493911743164063e-05,
      "step": 96538,
      "training_step_time": 8.349989414215088
    },
    {
      "epoch": 2.149486541748047e-05,
      "model_forward_time": 1.72190260887146,
      "step": 96539
    },
    {
      "epoch": 2.149486541748047e-05,
      "step": 96539,
      "training_step_time": 7.848831653594971
    },
    {
      "epoch": 2.1495819091796874e-05,
      "grad_norm": 0.16718332469463348,
      "learning_rate": 3.271309256172084e-08,
      "loss": 0.0535,
      "step": 96540
    },
    {
      "epoch": 2.1495819091796874e-05,
      "model_forward_time": 1.352735996246338,
      "step": 96540
    },
    {
      "dynamics_loss_avg": 0.04281885791569948,
      "epoch": 2.1495819091796874e-05,
      "step": 96540
    },
    {
      "action_loss_avg": 0.0009487016097409651,
      "epoch": 2.1495819091796874e-05,
      "step": 96540
    },
    {
      "epoch": 2.1495819091796874e-05,
      "step": 96540,
      "training_step_time": 7.521006345748901
    },
    {
      "epoch": 2.149677276611328e-05,
      "model_forward_time": 1.3455297946929932,
      "step": 96541
    },
    {
      "epoch": 2.149677276611328e-05,
      "step": 96541,
      "training_step_time": 6.801823616027832
    },
    {
      "epoch": 2.149772644042969e-05,
      "model_forward_time": 1.725625991821289,
      "step": 96542
    },
    {
      "epoch": 2.149772644042969e-05,
      "step": 96542,
      "training_step_time": 6.938252925872803
    },
    {
      "epoch": 2.1498680114746093e-05,
      "model_forward_time": 1.74375581741333,
      "step": 96543
    },
    {
      "epoch": 2.1498680114746093e-05,
      "step": 96543,
      "training_step_time": 7.356260538101196
    },
    {
      "epoch": 2.14996337890625e-05,
      "model_forward_time": 1.7926483154296875,
      "step": 96544
    },
    {
      "epoch": 2.14996337890625e-05,
      "step": 96544,
      "training_step_time": 7.84506893157959
    },
    {
      "epoch": 2.1500587463378908e-05,
      "model_forward_time": 1.7302193641662598,
      "step": 96545
    },
    {
      "epoch": 2.1500587463378908e-05,
      "step": 96545,
      "training_step_time": 10.164180278778076
    },
    {
      "epoch": 2.1501541137695312e-05,
      "model_forward_time": 1.7490224838256836,
      "step": 96546
    },
    {
      "epoch": 2.1501541137695312e-05,
      "step": 96546,
      "training_step_time": 9.50905704498291
    },
    {
      "epoch": 2.150249481201172e-05,
      "model_forward_time": 2.3967952728271484,
      "step": 96547
    },
    {
      "epoch": 2.150249481201172e-05,
      "step": 96547,
      "training_step_time": 8.401282787322998
    },
    {
      "epoch": 2.1503448486328123e-05,
      "model_forward_time": 1.7423741817474365,
      "step": 96548
    },
    {
      "epoch": 2.1503448486328123e-05,
      "step": 96548,
      "training_step_time": 7.781939744949341
    },
    {
      "epoch": 2.150440216064453e-05,
      "model_forward_time": 1.748551368713379,
      "step": 96549
    },
    {
      "epoch": 2.150440216064453e-05,
      "step": 96549,
      "training_step_time": 8.659693717956543
    },
    {
      "epoch": 2.1505355834960938e-05,
      "grad_norm": 0.10444676876068115,
      "learning_rate": 3.252453219203699e-08,
      "loss": 0.054,
      "step": 96550
    },
    {
      "epoch": 2.1505355834960938e-05,
      "model_forward_time": 1.7396948337554932,
      "step": 96550
    },
    {
      "dynamics_loss_avg": 0.057282411679625514,
      "epoch": 2.1505355834960938e-05,
      "step": 96550
    },
    {
      "action_loss_avg": 0.001966290723066777,
      "epoch": 2.1505355834960938e-05,
      "step": 96550
    },
    {
      "epoch": 2.1505355834960938e-05,
      "step": 96550,
      "training_step_time": 6.891188144683838
    },
    {
      "epoch": 2.1506309509277342e-05,
      "model_forward_time": 1.743483066558838,
      "step": 96551
    },
    {
      "epoch": 2.1506309509277342e-05,
      "step": 96551,
      "training_step_time": 6.929501533508301
    },
    {
      "epoch": 2.150726318359375e-05,
      "model_forward_time": 1.7428524494171143,
      "step": 96552
    },
    {
      "epoch": 2.150726318359375e-05,
      "step": 96552,
      "training_step_time": 7.719408988952637
    },
    {
      "epoch": 2.1508216857910157e-05,
      "model_forward_time": 1.8771133422851562,
      "step": 96553
    },
    {
      "epoch": 2.1508216857910157e-05,
      "step": 96553,
      "training_step_time": 8.873006105422974
    },
    {
      "epoch": 2.150917053222656e-05,
      "model_forward_time": 1.7450308799743652,
      "step": 96554
    },
    {
      "epoch": 2.150917053222656e-05,
      "step": 96554,
      "training_step_time": 9.284809350967407
    },
    {
      "epoch": 2.151012420654297e-05,
      "model_forward_time": 1.7539622783660889,
      "step": 96555
    },
    {
      "epoch": 2.151012420654297e-05,
      "step": 96555,
      "training_step_time": 10.03992509841919
    },
    {
      "epoch": 2.1511077880859376e-05,
      "model_forward_time": 2.3891537189483643,
      "step": 96556
    },
    {
      "epoch": 2.1511077880859376e-05,
      "step": 96556,
      "training_step_time": 8.669299125671387
    },
    {
      "epoch": 2.151203155517578e-05,
      "model_forward_time": 1.7445447444915771,
      "step": 96557
    },
    {
      "epoch": 2.151203155517578e-05,
      "step": 96557,
      "training_step_time": 7.807705402374268
    },
    {
      "epoch": 2.1512985229492187e-05,
      "model_forward_time": 1.743910312652588,
      "step": 96558
    },
    {
      "epoch": 2.1512985229492187e-05,
      "step": 96558,
      "training_step_time": 7.690753221511841
    },
    {
      "epoch": 2.1513938903808595e-05,
      "model_forward_time": 1.7502453327178955,
      "step": 96559
    },
    {
      "epoch": 2.1513938903808595e-05,
      "step": 96559,
      "training_step_time": 7.5637006759643555
    },
    {
      "epoch": 2.1514892578125e-05,
      "grad_norm": 0.1230047270655632,
      "learning_rate": 3.233651505800184e-08,
      "loss": 0.0521,
      "step": 96560
    },
    {
      "epoch": 2.1514892578125e-05,
      "model_forward_time": 1.7567648887634277,
      "step": 96560
    },
    {
      "dynamics_loss_avg": 0.044510243088006975,
      "epoch": 2.1514892578125e-05,
      "step": 96560
    },
    {
      "action_loss_avg": 0.0010821227042470127,
      "epoch": 2.1514892578125e-05,
      "step": 96560
    },
    {
      "epoch": 2.1514892578125e-05,
      "step": 96560,
      "training_step_time": 6.901419639587402
    },
    {
      "epoch": 2.1515846252441406e-05,
      "model_forward_time": 1.7501580715179443,
      "step": 96561
    },
    {
      "epoch": 2.1515846252441406e-05,
      "step": 96561,
      "training_step_time": 7.716213941574097
    },
    {
      "epoch": 2.1516799926757814e-05,
      "model_forward_time": 1.8656702041625977,
      "step": 96562
    },
    {
      "epoch": 2.1516799926757814e-05,
      "step": 96562,
      "training_step_time": 7.764301061630249
    },
    {
      "epoch": 2.1517753601074218e-05,
      "model_forward_time": 1.741501808166504,
      "step": 96563
    },
    {
      "epoch": 2.1517753601074218e-05,
      "step": 96563,
      "training_step_time": 9.49660062789917
    },
    {
      "epoch": 2.1518707275390625e-05,
      "model_forward_time": 1.7429637908935547,
      "step": 96564
    },
    {
      "epoch": 2.1518707275390625e-05,
      "step": 96564,
      "training_step_time": 9.06591272354126
    },
    {
      "epoch": 2.1519660949707032e-05,
      "model_forward_time": 2.4116194248199463,
      "step": 96565
    },
    {
      "epoch": 2.1519660949707032e-05,
      "step": 96565,
      "training_step_time": 7.814015865325928
    },
    {
      "epoch": 2.1520614624023436e-05,
      "model_forward_time": 1.7491419315338135,
      "step": 96566
    },
    {
      "epoch": 2.1520614624023436e-05,
      "step": 96566,
      "training_step_time": 8.360867261886597
    },
    {
      "epoch": 2.1521568298339844e-05,
      "model_forward_time": 1.7437942028045654,
      "step": 96567
    },
    {
      "epoch": 2.1521568298339844e-05,
      "step": 96567,
      "training_step_time": 7.661521911621094
    },
    {
      "epoch": 2.152252197265625e-05,
      "model_forward_time": 1.3621339797973633,
      "step": 96568
    },
    {
      "epoch": 2.152252197265625e-05,
      "step": 96568,
      "training_step_time": 7.572291612625122
    },
    {
      "epoch": 2.1523475646972655e-05,
      "model_forward_time": 1.761301040649414,
      "step": 96569
    },
    {
      "epoch": 2.1523475646972655e-05,
      "step": 96569,
      "training_step_time": 6.9170496463775635
    },
    {
      "epoch": 2.1524429321289063e-05,
      "grad_norm": 0.23752936720848083,
      "learning_rate": 3.2149041180176146e-08,
      "loss": 0.0527,
      "step": 96570
    },
    {
      "epoch": 2.1524429321289063e-05,
      "model_forward_time": 1.3697333335876465,
      "step": 96570
    },
    {
      "dynamics_loss_avg": 0.0456405945122242,
      "epoch": 2.1524429321289063e-05,
      "step": 96570
    },
    {
      "action_loss_avg": 0.0019593222823459655,
      "epoch": 2.1524429321289063e-05,
      "step": 96570
    },
    {
      "epoch": 2.1524429321289063e-05,
      "step": 96570,
      "training_step_time": 7.7051331996917725
    },
    {
      "epoch": 2.152538299560547e-05,
      "model_forward_time": 1.8689756393432617,
      "step": 96571
    },
    {
      "epoch": 2.152538299560547e-05,
      "step": 96571,
      "training_step_time": 7.789538383483887
    },
    {
      "epoch": 2.1526336669921874e-05,
      "model_forward_time": 1.7458305358886719,
      "step": 96572
    },
    {
      "epoch": 2.1526336669921874e-05,
      "step": 96572,
      "training_step_time": 7.721895456314087
    },
    {
      "epoch": 2.152729034423828e-05,
      "model_forward_time": 1.7594959735870361,
      "step": 96573
    },
    {
      "epoch": 2.152729034423828e-05,
      "step": 96573,
      "training_step_time": 9.80008840560913
    },
    {
      "epoch": 2.152824401855469e-05,
      "model_forward_time": 2.362947702407837,
      "step": 96574
    },
    {
      "epoch": 2.152824401855469e-05,
      "step": 96574,
      "training_step_time": 8.853279113769531
    },
    {
      "epoch": 2.1529197692871093e-05,
      "model_forward_time": 1.7545793056488037,
      "step": 96575
    },
    {
      "epoch": 2.1529197692871093e-05,
      "step": 96575,
      "training_step_time": 7.864765167236328
    },
    {
      "epoch": 2.15301513671875e-05,
      "model_forward_time": 1.7482788562774658,
      "step": 96576
    },
    {
      "epoch": 2.15301513671875e-05,
      "step": 96576,
      "training_step_time": 7.699545383453369
    },
    {
      "epoch": 2.1531105041503908e-05,
      "model_forward_time": 1.744173526763916,
      "step": 96577
    },
    {
      "epoch": 2.1531105041503908e-05,
      "step": 96577,
      "training_step_time": 7.594959020614624
    },
    {
      "epoch": 2.1532058715820312e-05,
      "model_forward_time": 1.755765676498413,
      "step": 96578
    },
    {
      "epoch": 2.1532058715820312e-05,
      "step": 96578,
      "training_step_time": 6.965149879455566
    },
    {
      "epoch": 2.153301239013672e-05,
      "model_forward_time": 1.7507143020629883,
      "step": 96579
    },
    {
      "epoch": 2.153301239013672e-05,
      "step": 96579,
      "training_step_time": 8.271911382675171
    },
    {
      "epoch": 2.1533966064453123e-05,
      "grad_norm": 0.0695541724562645,
      "learning_rate": 3.196211057906296e-08,
      "loss": 0.0539,
      "step": 96580
    },
    {
      "epoch": 2.1533966064453123e-05,
      "model_forward_time": 1.85689377784729,
      "step": 96580
    },
    {
      "dynamics_loss_avg": 0.05734245181083679,
      "epoch": 2.1533966064453123e-05,
      "step": 96580
    },
    {
      "action_loss_avg": 0.0014710943039972335,
      "epoch": 2.1533966064453123e-05,
      "step": 96580
    },
    {
      "epoch": 2.1533966064453123e-05,
      "step": 96580,
      "training_step_time": 7.749292850494385
    },
    {
      "epoch": 2.153491973876953e-05,
      "model_forward_time": 1.7482428550720215,
      "step": 96581
    },
    {
      "epoch": 2.153491973876953e-05,
      "step": 96581,
      "training_step_time": 7.742475509643555
    },
    {
      "epoch": 2.1535873413085938e-05,
      "model_forward_time": 1.7482240200042725,
      "step": 96582
    },
    {
      "epoch": 2.1535873413085938e-05,
      "step": 96582,
      "training_step_time": 8.971992254257202
    },
    {
      "epoch": 2.1536827087402342e-05,
      "model_forward_time": 1.7415423393249512,
      "step": 96583
    },
    {
      "epoch": 2.1536827087402342e-05,
      "step": 96583,
      "training_step_time": 8.862788200378418
    },
    {
      "epoch": 2.153778076171875e-05,
      "model_forward_time": 2.3988494873046875,
      "step": 96584
    },
    {
      "epoch": 2.153778076171875e-05,
      "step": 96584,
      "training_step_time": 9.637915849685669
    },
    {
      "epoch": 2.1538734436035157e-05,
      "model_forward_time": 1.3561639785766602,
      "step": 96585
    },
    {
      "epoch": 2.1538734436035157e-05,
      "step": 96585,
      "training_step_time": 8.42564868927002
    },
    {
      "epoch": 2.153968811035156e-05,
      "model_forward_time": 1.3782024383544922,
      "step": 96586
    },
    {
      "epoch": 2.153968811035156e-05,
      "step": 96586,
      "training_step_time": 7.728216648101807
    },
    {
      "epoch": 2.154064178466797e-05,
      "model_forward_time": 1.742396354675293,
      "step": 96587
    },
    {
      "epoch": 2.154064178466797e-05,
      "step": 96587,
      "training_step_time": 7.621644020080566
    },
    {
      "epoch": 2.1541595458984376e-05,
      "model_forward_time": 1.7449750900268555,
      "step": 96588
    },
    {
      "epoch": 2.1541595458984376e-05,
      "step": 96588,
      "training_step_time": 7.219954013824463
    },
    {
      "epoch": 2.154254913330078e-05,
      "model_forward_time": 1.8099665641784668,
      "step": 96589
    },
    {
      "epoch": 2.154254913330078e-05,
      "step": 96589,
      "training_step_time": 8.405303716659546
    },
    {
      "epoch": 2.1543502807617187e-05,
      "grad_norm": 0.11355069279670715,
      "learning_rate": 3.1775723275103146e-08,
      "loss": 0.0515,
      "step": 96590
    },
    {
      "epoch": 2.1543502807617187e-05,
      "model_forward_time": 1.7477116584777832,
      "step": 96590
    },
    {
      "dynamics_loss_avg": 0.05493214288726449,
      "epoch": 2.1543502807617187e-05,
      "step": 96590
    },
    {
      "action_loss_avg": 0.0013896382559323683,
      "epoch": 2.1543502807617187e-05,
      "step": 96590
    },
    {
      "epoch": 2.1543502807617187e-05,
      "step": 96590,
      "training_step_time": 7.8609113693237305
    },
    {
      "epoch": 2.1544456481933595e-05,
      "model_forward_time": 1.7501583099365234,
      "step": 96591
    },
    {
      "epoch": 2.1544456481933595e-05,
      "step": 96591,
      "training_step_time": 8.918410778045654
    },
    {
      "epoch": 2.154541015625e-05,
      "model_forward_time": 1.7538042068481445,
      "step": 96592
    },
    {
      "epoch": 2.154541015625e-05,
      "step": 96592,
      "training_step_time": 9.818290710449219
    },
    {
      "epoch": 2.1546363830566406e-05,
      "model_forward_time": 2.40468692779541,
      "step": 96593
    },
    {
      "epoch": 2.1546363830566406e-05,
      "step": 96593,
      "training_step_time": 8.59340238571167
    },
    {
      "epoch": 2.1547317504882814e-05,
      "model_forward_time": 1.7557744979858398,
      "step": 96594
    },
    {
      "epoch": 2.1547317504882814e-05,
      "step": 96594,
      "training_step_time": 7.902558088302612
    },
    {
      "epoch": 2.1548271179199218e-05,
      "model_forward_time": 1.7450158596038818,
      "step": 96595
    },
    {
      "epoch": 2.1548271179199218e-05,
      "step": 96595,
      "training_step_time": 7.773804187774658
    },
    {
      "epoch": 2.1549224853515625e-05,
      "model_forward_time": 1.7482187747955322,
      "step": 96596
    },
    {
      "epoch": 2.1549224853515625e-05,
      "step": 96596,
      "training_step_time": 7.618936538696289
    },
    {
      "epoch": 2.1550178527832032e-05,
      "model_forward_time": 1.7510063648223877,
      "step": 96597
    },
    {
      "epoch": 2.1550178527832032e-05,
      "step": 96597,
      "training_step_time": 7.169872760772705
    },
    {
      "epoch": 2.1551132202148437e-05,
      "model_forward_time": 1.830138921737671,
      "step": 96598
    },
    {
      "epoch": 2.1551132202148437e-05,
      "step": 96598,
      "training_step_time": 7.823440313339233
    },
    {
      "epoch": 2.1552085876464844e-05,
      "model_forward_time": 1.7591168880462646,
      "step": 96599
    },
    {
      "epoch": 2.1552085876464844e-05,
      "step": 96599,
      "training_step_time": 7.847438812255859
    },
    {
      "epoch": 2.155303955078125e-05,
      "grad_norm": 0.11569850891828537,
      "learning_rate": 3.158987928868151e-08,
      "loss": 0.0516,
      "step": 96600
    },
    {
      "epoch": 2.155303955078125e-05,
      "model_forward_time": 1.7412569522857666,
      "step": 96600
    },
    {
      "dynamics_loss_avg": 0.06981486082077026,
      "epoch": 2.155303955078125e-05,
      "step": 96600
    },
    {
      "action_loss_avg": 0.0014458863384788855,
      "epoch": 2.155303955078125e-05,
      "step": 96600
    },
    {
      "epoch": 2.155303955078125e-05,
      "step": 96600,
      "training_step_time": 8.872303485870361
    },
    {
      "epoch": 2.1553993225097655e-05,
      "model_forward_time": 1.7222254276275635,
      "step": 96601
    },
    {
      "epoch": 2.1553993225097655e-05,
      "step": 96601,
      "training_step_time": 8.660193920135498
    },
    {
      "epoch": 2.1554946899414063e-05,
      "model_forward_time": 2.4393322467803955,
      "step": 96602
    },
    {
      "epoch": 2.1554946899414063e-05,
      "step": 96602,
      "training_step_time": 9.4354248046875
    },
    {
      "epoch": 2.155590057373047e-05,
      "model_forward_time": 1.7504560947418213,
      "step": 96603
    },
    {
      "epoch": 2.155590057373047e-05,
      "step": 96603,
      "training_step_time": 7.795693874359131
    },
    {
      "epoch": 2.1556854248046874e-05,
      "model_forward_time": 1.3617420196533203,
      "step": 96604
    },
    {
      "epoch": 2.1556854248046874e-05,
      "step": 96604,
      "training_step_time": 7.675482749938965
    },
    {
      "epoch": 2.155780792236328e-05,
      "model_forward_time": 1.7387573719024658,
      "step": 96605
    },
    {
      "epoch": 2.155780792236328e-05,
      "step": 96605,
      "training_step_time": 7.539380788803101
    },
    {
      "epoch": 2.155876159667969e-05,
      "model_forward_time": 1.753068208694458,
      "step": 96606
    },
    {
      "epoch": 2.155876159667969e-05,
      "step": 96606,
      "training_step_time": 7.137529373168945
    },
    {
      "epoch": 2.1559715270996093e-05,
      "model_forward_time": 1.0835275650024414,
      "step": 96607
    },
    {
      "epoch": 2.1559715270996093e-05,
      "step": 96607,
      "training_step_time": 7.799007177352905
    },
    {
      "epoch": 2.15606689453125e-05,
      "model_forward_time": 1.7472047805786133,
      "step": 96608
    },
    {
      "epoch": 2.15606689453125e-05,
      "step": 96608,
      "training_step_time": 7.778510570526123
    },
    {
      "epoch": 2.1561622619628908e-05,
      "model_forward_time": 1.7311561107635498,
      "step": 96609
    },
    {
      "epoch": 2.1561622619628908e-05,
      "step": 96609,
      "training_step_time": 7.784785032272339
    },
    {
      "epoch": 2.1562576293945312e-05,
      "grad_norm": 0.3235270380973816,
      "learning_rate": 3.140457864012014e-08,
      "loss": 0.0515,
      "step": 96610
    },
    {
      "epoch": 2.1562576293945312e-05,
      "model_forward_time": 1.7505052089691162,
      "step": 96610
    },
    {
      "dynamics_loss_avg": 0.06692332550883293,
      "epoch": 2.1562576293945312e-05,
      "step": 96610
    },
    {
      "action_loss_avg": 0.001260440150508657,
      "epoch": 2.1562576293945312e-05,
      "step": 96610
    },
    {
      "epoch": 2.1562576293945312e-05,
      "step": 96610,
      "training_step_time": 9.681092262268066
    },
    {
      "epoch": 2.156352996826172e-05,
      "model_forward_time": 1.7482235431671143,
      "step": 96611
    },
    {
      "epoch": 2.156352996826172e-05,
      "step": 96611,
      "training_step_time": 7.858111143112183
    },
    {
      "epoch": 2.1564483642578123e-05,
      "model_forward_time": 2.4238882064819336,
      "step": 96612
    },
    {
      "epoch": 2.1564483642578123e-05,
      "step": 96612,
      "training_step_time": 8.801772356033325
    },
    {
      "epoch": 2.156543731689453e-05,
      "model_forward_time": 1.7434945106506348,
      "step": 96613
    },
    {
      "epoch": 2.156543731689453e-05,
      "step": 96613,
      "training_step_time": 7.717465877532959
    },
    {
      "epoch": 2.1566390991210938e-05,
      "model_forward_time": 1.7448444366455078,
      "step": 96614
    },
    {
      "epoch": 2.1566390991210938e-05,
      "step": 96614,
      "training_step_time": 7.701590299606323
    },
    {
      "epoch": 2.1567344665527342e-05,
      "model_forward_time": 1.753612756729126,
      "step": 96615
    },
    {
      "epoch": 2.1567344665527342e-05,
      "step": 96615,
      "training_step_time": 7.1669089794158936
    },
    {
      "epoch": 2.156829833984375e-05,
      "model_forward_time": 1.8639743328094482,
      "step": 96616
    },
    {
      "epoch": 2.156829833984375e-05,
      "step": 96616,
      "training_step_time": 7.787037372589111
    },
    {
      "epoch": 2.1569252014160157e-05,
      "model_forward_time": 1.7436680793762207,
      "step": 96617
    },
    {
      "epoch": 2.1569252014160157e-05,
      "step": 96617,
      "training_step_time": 7.760029554367065
    },
    {
      "epoch": 2.157020568847656e-05,
      "model_forward_time": 1.7444164752960205,
      "step": 96618
    },
    {
      "epoch": 2.157020568847656e-05,
      "step": 96618,
      "training_step_time": 7.962372541427612
    },
    {
      "epoch": 2.157115936279297e-05,
      "model_forward_time": 1.7530183792114258,
      "step": 96619
    },
    {
      "epoch": 2.157115936279297e-05,
      "step": 96619,
      "training_step_time": 9.45766305923462
    },
    {
      "epoch": 2.1572113037109376e-05,
      "grad_norm": 0.1236141100525856,
      "learning_rate": 3.121982134968393e-08,
      "loss": 0.0521,
      "step": 96620
    },
    {
      "epoch": 2.1572113037109376e-05,
      "model_forward_time": 1.7457554340362549,
      "step": 96620
    },
    {
      "dynamics_loss_avg": 0.045595341548323634,
      "epoch": 2.1572113037109376e-05,
      "step": 96620
    },
    {
      "action_loss_avg": 0.000963833843707107,
      "epoch": 2.1572113037109376e-05,
      "step": 96620
    },
    {
      "epoch": 2.1572113037109376e-05,
      "step": 96620,
      "training_step_time": 9.585276365280151
    },
    {
      "epoch": 2.157306671142578e-05,
      "model_forward_time": 2.391416072845459,
      "step": 96621
    },
    {
      "epoch": 2.157306671142578e-05,
      "step": 96621,
      "training_step_time": 7.980688810348511
    },
    {
      "epoch": 2.1574020385742187e-05,
      "model_forward_time": 1.7391142845153809,
      "step": 96622
    },
    {
      "epoch": 2.1574020385742187e-05,
      "step": 96622,
      "training_step_time": 7.853070497512817
    },
    {
      "epoch": 2.1574974060058595e-05,
      "model_forward_time": 1.7432422637939453,
      "step": 96623
    },
    {
      "epoch": 2.1574974060058595e-05,
      "step": 96623,
      "training_step_time": 7.761996746063232
    },
    {
      "epoch": 2.1575927734375e-05,
      "model_forward_time": 1.755054235458374,
      "step": 96624
    },
    {
      "epoch": 2.1575927734375e-05,
      "step": 96624,
      "training_step_time": 7.3076581954956055
    },
    {
      "epoch": 2.1576881408691406e-05,
      "model_forward_time": 1.8711187839508057,
      "step": 96625
    },
    {
      "epoch": 2.1576881408691406e-05,
      "step": 96625,
      "training_step_time": 7.902867794036865
    },
    {
      "epoch": 2.1577835083007814e-05,
      "model_forward_time": 1.7449960708618164,
      "step": 96626
    },
    {
      "epoch": 2.1577835083007814e-05,
      "step": 96626,
      "training_step_time": 7.880077123641968
    },
    {
      "epoch": 2.1578788757324218e-05,
      "model_forward_time": 1.751152515411377,
      "step": 96627
    },
    {
      "epoch": 2.1578788757324218e-05,
      "step": 96627,
      "training_step_time": 9.535739421844482
    },
    {
      "epoch": 2.1579742431640625e-05,
      "model_forward_time": 1.7422854900360107,
      "step": 96628
    },
    {
      "epoch": 2.1579742431640625e-05,
      "step": 96628,
      "training_step_time": 9.294974088668823
    },
    {
      "epoch": 2.1580696105957033e-05,
      "model_forward_time": 1.7512400150299072,
      "step": 96629
    },
    {
      "epoch": 2.1580696105957033e-05,
      "step": 96629,
      "training_step_time": 9.785193920135498
    },
    {
      "epoch": 2.1581649780273437e-05,
      "grad_norm": 0.08089368045330048,
      "learning_rate": 3.103560743757783e-08,
      "loss": 0.0523,
      "step": 96630
    },
    {
      "epoch": 2.1581649780273437e-05,
      "model_forward_time": 2.4181792736053467,
      "step": 96630
    },
    {
      "dynamics_loss_avg": 0.044152993336319925,
      "epoch": 2.1581649780273437e-05,
      "step": 96630
    },
    {
      "action_loss_avg": 0.0012846329555031843,
      "epoch": 2.1581649780273437e-05,
      "step": 96630
    },
    {
      "epoch": 2.1581649780273437e-05,
      "step": 96630,
      "training_step_time": 8.374660968780518
    },
    {
      "epoch": 2.1582603454589844e-05,
      "model_forward_time": 1.7562644481658936,
      "step": 96631
    },
    {
      "epoch": 2.1582603454589844e-05,
      "step": 96631,
      "training_step_time": 7.801472187042236
    },
    {
      "epoch": 2.158355712890625e-05,
      "model_forward_time": 1.7489593029022217,
      "step": 96632
    },
    {
      "epoch": 2.158355712890625e-05,
      "step": 96632,
      "training_step_time": 8.777726888656616
    },
    {
      "epoch": 2.1584510803222655e-05,
      "model_forward_time": 1.7470152378082275,
      "step": 96633
    },
    {
      "epoch": 2.1584510803222655e-05,
      "step": 96633,
      "training_step_time": 7.2320263385772705
    },
    {
      "epoch": 2.1585464477539063e-05,
      "model_forward_time": 1.8821964263916016,
      "step": 96634
    },
    {
      "epoch": 2.1585464477539063e-05,
      "step": 96634,
      "training_step_time": 7.845280885696411
    },
    {
      "epoch": 2.158641815185547e-05,
      "model_forward_time": 1.7485568523406982,
      "step": 96635
    },
    {
      "epoch": 2.158641815185547e-05,
      "step": 96635,
      "training_step_time": 7.78679895401001
    },
    {
      "epoch": 2.1587371826171874e-05,
      "model_forward_time": 1.7480263710021973,
      "step": 96636
    },
    {
      "epoch": 2.1587371826171874e-05,
      "step": 96636,
      "training_step_time": 7.737537860870361
    },
    {
      "epoch": 2.158832550048828e-05,
      "model_forward_time": 1.7467093467712402,
      "step": 96637
    },
    {
      "epoch": 2.158832550048828e-05,
      "step": 96637,
      "training_step_time": 8.417115688323975
    },
    {
      "epoch": 2.158927917480469e-05,
      "model_forward_time": 1.7512426376342773,
      "step": 96638
    },
    {
      "epoch": 2.158927917480469e-05,
      "step": 96638,
      "training_step_time": 9.576003789901733
    },
    {
      "epoch": 2.1590232849121093e-05,
      "model_forward_time": 2.435741424560547,
      "step": 96639
    },
    {
      "epoch": 2.1590232849121093e-05,
      "step": 96639,
      "training_step_time": 7.899104356765747
    },
    {
      "epoch": 2.15911865234375e-05,
      "grad_norm": 0.07244856655597687,
      "learning_rate": 3.085193692394684e-08,
      "loss": 0.051,
      "step": 96640
    },
    {
      "epoch": 2.15911865234375e-05,
      "model_forward_time": 1.7518572807312012,
      "step": 96640
    },
    {
      "dynamics_loss_avg": 0.05674531627446413,
      "epoch": 2.15911865234375e-05,
      "step": 96640
    },
    {
      "action_loss_avg": 0.0009322778438217938,
      "epoch": 2.15911865234375e-05,
      "step": 96640
    },
    {
      "epoch": 2.15911865234375e-05,
      "step": 96640,
      "training_step_time": 8.567065238952637
    },
    {
      "epoch": 2.1592140197753905e-05,
      "model_forward_time": 1.3591454029083252,
      "step": 96641
    },
    {
      "epoch": 2.1592140197753905e-05,
      "step": 96641,
      "training_step_time": 7.800528049468994
    },
    {
      "epoch": 2.1593093872070312e-05,
      "model_forward_time": 1.7482376098632812,
      "step": 96642
    },
    {
      "epoch": 2.1593093872070312e-05,
      "step": 96642,
      "training_step_time": 7.601743936538696
    },
    {
      "epoch": 2.159404754638672e-05,
      "model_forward_time": 1.4814040660858154,
      "step": 96643
    },
    {
      "epoch": 2.159404754638672e-05,
      "step": 96643,
      "training_step_time": 7.886007308959961
    },
    {
      "epoch": 2.1595001220703123e-05,
      "model_forward_time": 1.7564525604248047,
      "step": 96644
    },
    {
      "epoch": 2.1595001220703123e-05,
      "step": 96644,
      "training_step_time": 8.64978837966919
    },
    {
      "epoch": 2.159595489501953e-05,
      "model_forward_time": 1.737549066543579,
      "step": 96645
    },
    {
      "epoch": 2.159595489501953e-05,
      "step": 96645,
      "training_step_time": 8.690239191055298
    },
    {
      "epoch": 2.1596908569335938e-05,
      "model_forward_time": 1.7340190410614014,
      "step": 96646
    },
    {
      "epoch": 2.1596908569335938e-05,
      "step": 96646,
      "training_step_time": 8.633772611618042
    },
    {
      "epoch": 2.1597862243652342e-05,
      "model_forward_time": 1.7364976406097412,
      "step": 96647
    },
    {
      "epoch": 2.1597862243652342e-05,
      "step": 96647,
      "training_step_time": 8.490224123001099
    },
    {
      "epoch": 2.159881591796875e-05,
      "model_forward_time": 1.3537240028381348,
      "step": 96648
    },
    {
      "epoch": 2.159881591796875e-05,
      "step": 96648,
      "training_step_time": 8.698222637176514
    },
    {
      "epoch": 2.1599769592285157e-05,
      "model_forward_time": 2.3957860469818115,
      "step": 96649
    },
    {
      "epoch": 2.1599769592285157e-05,
      "step": 96649,
      "training_step_time": 8.462677717208862
    },
    {
      "epoch": 2.160072326660156e-05,
      "grad_norm": 0.11241437494754791,
      "learning_rate": 3.06688098288771e-08,
      "loss": 0.0506,
      "step": 96650
    },
    {
      "epoch": 2.160072326660156e-05,
      "model_forward_time": 1.7367546558380127,
      "step": 96650
    },
    {
      "dynamics_loss_avg": 0.04361236430704594,
      "epoch": 2.160072326660156e-05,
      "step": 96650
    },
    {
      "action_loss_avg": 0.001214717497350648,
      "epoch": 2.160072326660156e-05,
      "step": 96650
    },
    {
      "epoch": 2.160072326660156e-05,
      "step": 96650,
      "training_step_time": 7.766416788101196
    },
    {
      "epoch": 2.160167694091797e-05,
      "model_forward_time": 1.735743522644043,
      "step": 96651
    },
    {
      "epoch": 2.160167694091797e-05,
      "step": 96651,
      "training_step_time": 7.634238243103027
    },
    {
      "epoch": 2.1602630615234376e-05,
      "model_forward_time": 1.3833446502685547,
      "step": 96652
    },
    {
      "epoch": 2.1602630615234376e-05,
      "step": 96652,
      "training_step_time": 7.841901063919067
    },
    {
      "epoch": 2.160358428955078e-05,
      "model_forward_time": 1.7453725337982178,
      "step": 96653
    },
    {
      "epoch": 2.160358428955078e-05,
      "step": 96653,
      "training_step_time": 7.6682703495025635
    },
    {
      "epoch": 2.1604537963867187e-05,
      "model_forward_time": 1.7424907684326172,
      "step": 96654
    },
    {
      "epoch": 2.1604537963867187e-05,
      "step": 96654,
      "training_step_time": 7.733415603637695
    },
    {
      "epoch": 2.1605491638183595e-05,
      "model_forward_time": 1.7393221855163574,
      "step": 96655
    },
    {
      "epoch": 2.1605491638183595e-05,
      "step": 96655,
      "training_step_time": 7.72470235824585
    },
    {
      "epoch": 2.16064453125e-05,
      "model_forward_time": 1.7364909648895264,
      "step": 96656
    },
    {
      "epoch": 2.16064453125e-05,
      "step": 96656,
      "training_step_time": 10.026110410690308
    },
    {
      "epoch": 2.1607398986816406e-05,
      "model_forward_time": 1.7325341701507568,
      "step": 96657
    },
    {
      "epoch": 2.1607398986816406e-05,
      "step": 96657,
      "training_step_time": 9.458931922912598
    },
    {
      "epoch": 2.1608352661132814e-05,
      "model_forward_time": 2.450681686401367,
      "step": 96658
    },
    {
      "epoch": 2.1608352661132814e-05,
      "step": 96658,
      "training_step_time": 7.906486988067627
    },
    {
      "epoch": 2.1609306335449218e-05,
      "model_forward_time": 1.73909592628479,
      "step": 96659
    },
    {
      "epoch": 2.1609306335449218e-05,
      "step": 96659,
      "training_step_time": 7.6014275550842285
    },
    {
      "epoch": 2.1610260009765625e-05,
      "grad_norm": 0.08724561333656311,
      "learning_rate": 3.048622617239483e-08,
      "loss": 0.0527,
      "step": 96660
    },
    {
      "epoch": 2.1610260009765625e-05,
      "model_forward_time": 1.7339937686920166,
      "step": 96660
    },
    {
      "dynamics_loss_avg": 0.04394341576844454,
      "epoch": 2.1610260009765625e-05,
      "step": 96660
    },
    {
      "action_loss_avg": 0.000928279699292034,
      "epoch": 2.1610260009765625e-05,
      "step": 96660
    },
    {
      "epoch": 2.1610260009765625e-05,
      "step": 96660,
      "training_step_time": 7.787623167037964
    },
    {
      "epoch": 2.1611213684082033e-05,
      "model_forward_time": 1.7607288360595703,
      "step": 96661
    },
    {
      "epoch": 2.1611213684082033e-05,
      "step": 96661,
      "training_step_time": 7.868946552276611
    },
    {
      "epoch": 2.1612167358398437e-05,
      "model_forward_time": 1.7417113780975342,
      "step": 96662
    },
    {
      "epoch": 2.1612167358398437e-05,
      "step": 96662,
      "training_step_time": 7.676968097686768
    },
    {
      "epoch": 2.1613121032714844e-05,
      "model_forward_time": 1.7541167736053467,
      "step": 96663
    },
    {
      "epoch": 2.1613121032714844e-05,
      "step": 96663,
      "training_step_time": 7.798508405685425
    },
    {
      "epoch": 2.161407470703125e-05,
      "model_forward_time": 1.747901201248169,
      "step": 96664
    },
    {
      "epoch": 2.161407470703125e-05,
      "step": 96664,
      "training_step_time": 7.717892408370972
    },
    {
      "epoch": 2.1615028381347655e-05,
      "model_forward_time": 1.7722604274749756,
      "step": 96665
    },
    {
      "epoch": 2.1615028381347655e-05,
      "step": 96665,
      "training_step_time": 10.52946138381958
    },
    {
      "epoch": 2.1615982055664063e-05,
      "model_forward_time": 1.7603089809417725,
      "step": 96666
    },
    {
      "epoch": 2.1615982055664063e-05,
      "step": 96666,
      "training_step_time": 9.464497089385986
    },
    {
      "epoch": 2.161693572998047e-05,
      "model_forward_time": 2.4295969009399414,
      "step": 96667
    },
    {
      "epoch": 2.161693572998047e-05,
      "step": 96667,
      "training_step_time": 7.887091875076294
    },
    {
      "epoch": 2.1617889404296874e-05,
      "model_forward_time": 1.7538111209869385,
      "step": 96668
    },
    {
      "epoch": 2.1617889404296874e-05,
      "step": 96668,
      "training_step_time": 7.7511303424835205
    },
    {
      "epoch": 2.1618843078613282e-05,
      "model_forward_time": 1.3766965866088867,
      "step": 96669
    },
    {
      "epoch": 2.1618843078613282e-05,
      "step": 96669,
      "training_step_time": 7.807150840759277
    },
    {
      "epoch": 2.161979675292969e-05,
      "grad_norm": 0.06538941711187363,
      "learning_rate": 3.030418597446738e-08,
      "loss": 0.053,
      "step": 96670
    },
    {
      "epoch": 2.161979675292969e-05,
      "model_forward_time": 1.8766663074493408,
      "step": 96670
    },
    {
      "dynamics_loss_avg": 0.04863246139138937,
      "epoch": 2.161979675292969e-05,
      "step": 96670
    },
    {
      "action_loss_avg": 0.0014729442045791075,
      "epoch": 2.161979675292969e-05,
      "step": 96670
    },
    {
      "epoch": 2.161979675292969e-05,
      "step": 96670,
      "training_step_time": 7.81094765663147
    },
    {
      "epoch": 2.1620750427246093e-05,
      "model_forward_time": 1.7531089782714844,
      "step": 96671
    },
    {
      "epoch": 2.1620750427246093e-05,
      "step": 96671,
      "training_step_time": 7.627222776412964
    },
    {
      "epoch": 2.16217041015625e-05,
      "model_forward_time": 1.7529442310333252,
      "step": 96672
    },
    {
      "epoch": 2.16217041015625e-05,
      "step": 96672,
      "training_step_time": 7.826733350753784
    },
    {
      "epoch": 2.1622657775878905e-05,
      "model_forward_time": 1.7596752643585205,
      "step": 96673
    },
    {
      "epoch": 2.1622657775878905e-05,
      "step": 96673,
      "training_step_time": 8.27104115486145
    },
    {
      "epoch": 2.1623611450195312e-05,
      "model_forward_time": 1.749253273010254,
      "step": 96674
    },
    {
      "epoch": 2.1623611450195312e-05,
      "step": 96674,
      "training_step_time": 8.799780130386353
    },
    {
      "epoch": 2.162456512451172e-05,
      "model_forward_time": 1.7601652145385742,
      "step": 96675
    },
    {
      "epoch": 2.162456512451172e-05,
      "step": 96675,
      "training_step_time": 7.784013986587524
    },
    {
      "epoch": 2.1625518798828123e-05,
      "model_forward_time": 1.394228219985962,
      "step": 96676
    },
    {
      "epoch": 2.1625518798828123e-05,
      "step": 96676,
      "training_step_time": 7.7845458984375
    },
    {
      "epoch": 2.162647247314453e-05,
      "model_forward_time": 1.9964449405670166,
      "step": 96677
    },
    {
      "epoch": 2.162647247314453e-05,
      "step": 96677,
      "training_step_time": 7.8708860874176025
    },
    {
      "epoch": 2.1627426147460938e-05,
      "model_forward_time": 1.7582647800445557,
      "step": 96678
    },
    {
      "epoch": 2.1627426147460938e-05,
      "step": 96678,
      "training_step_time": 7.774152040481567
    },
    {
      "epoch": 2.1628379821777342e-05,
      "model_forward_time": 1.8783388137817383,
      "step": 96679
    },
    {
      "epoch": 2.1628379821777342e-05,
      "step": 96679,
      "training_step_time": 7.881422519683838
    },
    {
      "epoch": 2.162933349609375e-05,
      "grad_norm": 0.07138645648956299,
      "learning_rate": 3.0122689255002724e-08,
      "loss": 0.0512,
      "step": 96680
    },
    {
      "epoch": 2.162933349609375e-05,
      "model_forward_time": 1.7533280849456787,
      "step": 96680
    },
    {
      "dynamics_loss_avg": 0.04758038781583309,
      "epoch": 2.162933349609375e-05,
      "step": 96680
    },
    {
      "action_loss_avg": 0.0008189125132048502,
      "epoch": 2.162933349609375e-05,
      "step": 96680
    },
    {
      "epoch": 2.162933349609375e-05,
      "step": 96680,
      "training_step_time": 7.663345098495483
    },
    {
      "epoch": 2.1630287170410157e-05,
      "model_forward_time": 1.758641242980957,
      "step": 96681
    },
    {
      "epoch": 2.1630287170410157e-05,
      "step": 96681,
      "training_step_time": 8.365703344345093
    },
    {
      "epoch": 2.163124084472656e-05,
      "model_forward_time": 1.7597990036010742,
      "step": 96682
    },
    {
      "epoch": 2.163124084472656e-05,
      "step": 96682,
      "training_step_time": 8.655434608459473
    },
    {
      "epoch": 2.163219451904297e-05,
      "model_forward_time": 1.760073184967041,
      "step": 96683
    },
    {
      "epoch": 2.163219451904297e-05,
      "step": 96683,
      "training_step_time": 9.615246772766113
    },
    {
      "epoch": 2.1633148193359376e-05,
      "model_forward_time": 1.7624239921569824,
      "step": 96684
    },
    {
      "epoch": 2.1633148193359376e-05,
      "step": 96684,
      "training_step_time": 9.533262491226196
    },
    {
      "epoch": 2.163410186767578e-05,
      "model_forward_time": 1.7545192241668701,
      "step": 96685
    },
    {
      "epoch": 2.163410186767578e-05,
      "step": 96685,
      "training_step_time": 8.54417109489441
    },
    {
      "epoch": 2.1635055541992187e-05,
      "model_forward_time": 2.468869209289551,
      "step": 96686
    },
    {
      "epoch": 2.1635055541992187e-05,
      "step": 96686,
      "training_step_time": 8.936800241470337
    },
    {
      "epoch": 2.1636009216308595e-05,
      "model_forward_time": 1.7558915615081787,
      "step": 96687
    },
    {
      "epoch": 2.1636009216308595e-05,
      "step": 96687,
      "training_step_time": 7.748763561248779
    },
    {
      "epoch": 2.1636962890625e-05,
      "model_forward_time": 1.8761720657348633,
      "step": 96688
    },
    {
      "epoch": 2.1636962890625e-05,
      "step": 96688,
      "training_step_time": 7.806344985961914
    },
    {
      "epoch": 2.1637916564941406e-05,
      "model_forward_time": 1.7669835090637207,
      "step": 96689
    },
    {
      "epoch": 2.1637916564941406e-05,
      "step": 96689,
      "training_step_time": 7.246686697006226
    },
    {
      "epoch": 2.1638870239257814e-05,
      "grad_norm": 0.10038478672504425,
      "learning_rate": 2.994173603384831e-08,
      "loss": 0.0527,
      "step": 96690
    },
    {
      "epoch": 2.1638870239257814e-05,
      "model_forward_time": 1.751920223236084,
      "step": 96690
    },
    {
      "dynamics_loss_avg": 0.041006629914045335,
      "epoch": 2.1638870239257814e-05,
      "step": 96690
    },
    {
      "action_loss_avg": 0.0009933263150742277,
      "epoch": 2.1638870239257814e-05,
      "step": 96690
    },
    {
      "epoch": 2.1638870239257814e-05,
      "step": 96690,
      "training_step_time": 7.724856376647949
    },
    {
      "epoch": 2.1639823913574218e-05,
      "model_forward_time": 1.7564265727996826,
      "step": 96691
    },
    {
      "epoch": 2.1639823913574218e-05,
      "step": 96691,
      "training_step_time": 7.695183515548706
    },
    {
      "epoch": 2.1640777587890625e-05,
      "model_forward_time": 1.759253740310669,
      "step": 96692
    },
    {
      "epoch": 2.1640777587890625e-05,
      "step": 96692,
      "training_step_time": 8.998683214187622
    },
    {
      "epoch": 2.1641731262207033e-05,
      "model_forward_time": 1.7590551376342773,
      "step": 96693
    },
    {
      "epoch": 2.1641731262207033e-05,
      "step": 96693,
      "training_step_time": 10.04261827468872
    },
    {
      "epoch": 2.1642684936523437e-05,
      "model_forward_time": 1.3914058208465576,
      "step": 96694
    },
    {
      "epoch": 2.1642684936523437e-05,
      "step": 96694,
      "training_step_time": 8.708193063735962
    },
    {
      "epoch": 2.1643638610839844e-05,
      "model_forward_time": 2.3752636909484863,
      "step": 96695
    },
    {
      "epoch": 2.1643638610839844e-05,
      "step": 96695,
      "training_step_time": 7.848971843719482
    },
    {
      "epoch": 2.164459228515625e-05,
      "model_forward_time": 1.761927604675293,
      "step": 96696
    },
    {
      "epoch": 2.164459228515625e-05,
      "step": 96696,
      "training_step_time": 7.75395941734314
    },
    {
      "epoch": 2.1645545959472655e-05,
      "model_forward_time": 1.8773455619812012,
      "step": 96697
    },
    {
      "epoch": 2.1645545959472655e-05,
      "step": 96697,
      "training_step_time": 7.7792510986328125
    },
    {
      "epoch": 2.1646499633789063e-05,
      "model_forward_time": 1.7605257034301758,
      "step": 96698
    },
    {
      "epoch": 2.1646499633789063e-05,
      "step": 96698,
      "training_step_time": 7.173002243041992
    },
    {
      "epoch": 2.164745330810547e-05,
      "model_forward_time": 1.7594530582427979,
      "step": 96699
    },
    {
      "epoch": 2.164745330810547e-05,
      "step": 96699,
      "training_step_time": 7.7268054485321045
    },
    {
      "epoch": 2.1648406982421874e-05,
      "grad_norm": 0.08233194053173065,
      "learning_rate": 2.9761326330793318e-08,
      "loss": 0.0537,
      "step": 96700
    },
    {
      "epoch": 2.1648406982421874e-05,
      "model_forward_time": 1.753999948501587,
      "step": 96700
    },
    {
      "dynamics_loss_avg": 0.04777684267610312,
      "epoch": 2.1648406982421874e-05,
      "step": 96700
    },
    {
      "action_loss_avg": 0.000737248471705243,
      "epoch": 2.1648406982421874e-05,
      "step": 96700
    },
    {
      "epoch": 2.1648406982421874e-05,
      "step": 96700,
      "training_step_time": 8.748944759368896
    },
    {
      "epoch": 2.1649360656738282e-05,
      "model_forward_time": 1.3757493495941162,
      "step": 96701
    },
    {
      "epoch": 2.1649360656738282e-05,
      "step": 96701,
      "training_step_time": 7.620190382003784
    },
    {
      "epoch": 2.165031433105469e-05,
      "model_forward_time": 1.7603754997253418,
      "step": 96702
    },
    {
      "epoch": 2.165031433105469e-05,
      "step": 96702,
      "training_step_time": 9.045941591262817
    },
    {
      "epoch": 2.1651268005371093e-05,
      "model_forward_time": 1.760636329650879,
      "step": 96703
    },
    {
      "epoch": 2.1651268005371093e-05,
      "step": 96703,
      "training_step_time": 9.364394664764404
    },
    {
      "epoch": 2.16522216796875e-05,
      "model_forward_time": 1.7516374588012695,
      "step": 96704
    },
    {
      "epoch": 2.16522216796875e-05,
      "step": 96704,
      "training_step_time": 7.942229509353638
    },
    {
      "epoch": 2.1653175354003905e-05,
      "model_forward_time": 2.433666944503784,
      "step": 96705
    },
    {
      "epoch": 2.1653175354003905e-05,
      "step": 96705,
      "training_step_time": 7.777676105499268
    },
    {
      "epoch": 2.1654129028320312e-05,
      "model_forward_time": 1.4791665077209473,
      "step": 96706
    },
    {
      "epoch": 2.1654129028320312e-05,
      "step": 96706,
      "training_step_time": 7.859416246414185
    },
    {
      "epoch": 2.165508270263672e-05,
      "model_forward_time": 1.7549331188201904,
      "step": 96707
    },
    {
      "epoch": 2.165508270263672e-05,
      "step": 96707,
      "training_step_time": 7.1965742111206055
    },
    {
      "epoch": 2.1656036376953123e-05,
      "model_forward_time": 1.7505838871002197,
      "step": 96708
    },
    {
      "epoch": 2.1656036376953123e-05,
      "step": 96708,
      "training_step_time": 8.667145490646362
    },
    {
      "epoch": 2.165699005126953e-05,
      "model_forward_time": 1.7496728897094727,
      "step": 96709
    },
    {
      "epoch": 2.165699005126953e-05,
      "step": 96709,
      "training_step_time": 7.821449518203735
    },
    {
      "epoch": 2.165794372558594e-05,
      "grad_norm": 0.08262939006090164,
      "learning_rate": 2.958146016556751e-08,
      "loss": 0.053,
      "step": 96710
    },
    {
      "epoch": 2.165794372558594e-05,
      "model_forward_time": 1.3775098323822021,
      "step": 96710
    },
    {
      "dynamics_loss_avg": 0.05659672487527132,
      "epoch": 2.165794372558594e-05,
      "step": 96710
    },
    {
      "action_loss_avg": 0.002214330378046725,
      "epoch": 2.165794372558594e-05,
      "step": 96710
    },
    {
      "epoch": 2.165794372558594e-05,
      "step": 96710,
      "training_step_time": 8.766656637191772
    },
    {
      "epoch": 2.1658897399902342e-05,
      "model_forward_time": 1.3769102096557617,
      "step": 96711
    },
    {
      "epoch": 2.1658897399902342e-05,
      "step": 96711,
      "training_step_time": 10.823237419128418
    },
    {
      "epoch": 2.165985107421875e-05,
      "model_forward_time": 1.7593271732330322,
      "step": 96712
    },
    {
      "epoch": 2.165985107421875e-05,
      "step": 96712,
      "training_step_time": 9.242711305618286
    },
    {
      "epoch": 2.1660804748535157e-05,
      "model_forward_time": 1.7613253593444824,
      "step": 96713
    },
    {
      "epoch": 2.1660804748535157e-05,
      "step": 96713,
      "training_step_time": 8.699524402618408
    },
    {
      "epoch": 2.166175842285156e-05,
      "model_forward_time": 2.4352643489837646,
      "step": 96714
    },
    {
      "epoch": 2.166175842285156e-05,
      "step": 96714,
      "training_step_time": 7.894559860229492
    },
    {
      "epoch": 2.166271209716797e-05,
      "model_forward_time": 1.8768773078918457,
      "step": 96715
    },
    {
      "epoch": 2.166271209716797e-05,
      "step": 96715,
      "training_step_time": 7.886259317398071
    },
    {
      "epoch": 2.1663665771484376e-05,
      "model_forward_time": 1.7492122650146484,
      "step": 96716
    },
    {
      "epoch": 2.1663665771484376e-05,
      "step": 96716,
      "training_step_time": 7.375993251800537
    },
    {
      "epoch": 2.166461944580078e-05,
      "model_forward_time": 1.7581145763397217,
      "step": 96717
    },
    {
      "epoch": 2.166461944580078e-05,
      "step": 96717,
      "training_step_time": 7.607661008834839
    },
    {
      "epoch": 2.1665573120117187e-05,
      "model_forward_time": 1.3747353553771973,
      "step": 96718
    },
    {
      "epoch": 2.1665573120117187e-05,
      "step": 96718,
      "training_step_time": 8.317520380020142
    },
    {
      "epoch": 2.1666526794433595e-05,
      "model_forward_time": 1.7432708740234375,
      "step": 96719
    },
    {
      "epoch": 2.1666526794433595e-05,
      "step": 96719,
      "training_step_time": 8.426203489303589
    },
    {
      "epoch": 2.166748046875e-05,
      "grad_norm": 0.06321189552545547,
      "learning_rate": 2.9402137557839605e-08,
      "loss": 0.053,
      "step": 96720
    },
    {
      "epoch": 2.166748046875e-05,
      "model_forward_time": 1.733006238937378,
      "step": 96720
    },
    {
      "dynamics_loss_avg": 0.03815793879330158,
      "epoch": 2.166748046875e-05,
      "step": 96720
    },
    {
      "action_loss_avg": 0.0014338949986267835,
      "epoch": 2.166748046875e-05,
      "step": 96720
    },
    {
      "epoch": 2.166748046875e-05,
      "step": 96720,
      "training_step_time": 8.576138973236084
    },
    {
      "epoch": 2.1668434143066406e-05,
      "model_forward_time": 1.3590624332427979,
      "step": 96721
    },
    {
      "epoch": 2.1668434143066406e-05,
      "step": 96721,
      "training_step_time": 7.798727750778198
    },
    {
      "epoch": 2.1669387817382814e-05,
      "model_forward_time": 1.7410774230957031,
      "step": 96722
    },
    {
      "epoch": 2.1669387817382814e-05,
      "step": 96722,
      "training_step_time": 8.589620113372803
    },
    {
      "epoch": 2.1670341491699218e-05,
      "model_forward_time": 1.7402749061584473,
      "step": 96723
    },
    {
      "epoch": 2.1670341491699218e-05,
      "step": 96723,
      "training_step_time": 8.62886667251587
    },
    {
      "epoch": 2.1671295166015625e-05,
      "model_forward_time": 2.3960649967193604,
      "step": 96724
    },
    {
      "epoch": 2.1671295166015625e-05,
      "step": 96724,
      "training_step_time": 9.400010347366333
    },
    {
      "epoch": 2.1672248840332033e-05,
      "model_forward_time": 1.7567312717437744,
      "step": 96725
    },
    {
      "epoch": 2.1672248840332033e-05,
      "step": 96725,
      "training_step_time": 7.7899603843688965
    },
    {
      "epoch": 2.1673202514648437e-05,
      "model_forward_time": 1.7552828788757324,
      "step": 96726
    },
    {
      "epoch": 2.1673202514648437e-05,
      "step": 96726,
      "training_step_time": 7.68966269493103
    },
    {
      "epoch": 2.1674156188964844e-05,
      "model_forward_time": 1.7571711540222168,
      "step": 96727
    },
    {
      "epoch": 2.1674156188964844e-05,
      "step": 96727,
      "training_step_time": 7.846240043640137
    },
    {
      "epoch": 2.167510986328125e-05,
      "model_forward_time": 1.7339427471160889,
      "step": 96728
    },
    {
      "epoch": 2.167510986328125e-05,
      "step": 96728,
      "training_step_time": 7.781724214553833
    },
    {
      "epoch": 2.1676063537597655e-05,
      "model_forward_time": 1.7357699871063232,
      "step": 96729
    },
    {
      "epoch": 2.1676063537597655e-05,
      "step": 96729,
      "training_step_time": 8.798548698425293
    },
    {
      "epoch": 2.1677017211914063e-05,
      "grad_norm": 0.07926425337791443,
      "learning_rate": 2.9223358527221134e-08,
      "loss": 0.0549,
      "step": 96730
    },
    {
      "epoch": 2.1677017211914063e-05,
      "model_forward_time": 1.73699951171875,
      "step": 96730
    },
    {
      "dynamics_loss_avg": 0.03716139737516642,
      "epoch": 2.1677017211914063e-05,
      "step": 96730
    },
    {
      "action_loss_avg": 0.0013928220490925015,
      "epoch": 2.1677017211914063e-05,
      "step": 96730
    },
    {
      "epoch": 2.1677017211914063e-05,
      "step": 96730,
      "training_step_time": 7.846000909805298
    },
    {
      "epoch": 2.167797088623047e-05,
      "model_forward_time": 1.73769211769104,
      "step": 96731
    },
    {
      "epoch": 2.167797088623047e-05,
      "step": 96731,
      "training_step_time": 7.9945068359375
    },
    {
      "epoch": 2.1678924560546874e-05,
      "model_forward_time": 1.73750901222229,
      "step": 96732
    },
    {
      "epoch": 2.1678924560546874e-05,
      "step": 96732,
      "training_step_time": 7.839477300643921
    },
    {
      "epoch": 2.1679878234863282e-05,
      "model_forward_time": 2.3774194717407227,
      "step": 96733
    },
    {
      "epoch": 2.1679878234863282e-05,
      "step": 96733,
      "training_step_time": 8.611996173858643
    },
    {
      "epoch": 2.168083190917969e-05,
      "model_forward_time": 1.734966516494751,
      "step": 96734
    },
    {
      "epoch": 2.168083190917969e-05,
      "step": 96734,
      "training_step_time": 7.760072231292725
    },
    {
      "epoch": 2.1681785583496093e-05,
      "model_forward_time": 1.7410356998443604,
      "step": 96735
    },
    {
      "epoch": 2.1681785583496093e-05,
      "step": 96735,
      "training_step_time": 7.618479013442993
    },
    {
      "epoch": 2.16827392578125e-05,
      "model_forward_time": 1.7524545192718506,
      "step": 96736
    },
    {
      "epoch": 2.16827392578125e-05,
      "step": 96736,
      "training_step_time": 7.766258239746094
    },
    {
      "epoch": 2.1683692932128905e-05,
      "model_forward_time": 1.7333204746246338,
      "step": 96737
    },
    {
      "epoch": 2.1683692932128905e-05,
      "step": 96737,
      "training_step_time": 8.417863130569458
    },
    {
      "epoch": 2.1684646606445312e-05,
      "model_forward_time": 1.73622465133667,
      "step": 96738
    },
    {
      "epoch": 2.1684646606445312e-05,
      "step": 96738,
      "training_step_time": 9.709014415740967
    },
    {
      "epoch": 2.168560028076172e-05,
      "model_forward_time": 1.7356829643249512,
      "step": 96739
    },
    {
      "epoch": 2.168560028076172e-05,
      "step": 96739,
      "training_step_time": 9.338361501693726
    },
    {
      "epoch": 2.1686553955078124e-05,
      "grad_norm": 0.08477809280157089,
      "learning_rate": 2.9045123093262572e-08,
      "loss": 0.0554,
      "step": 96740
    },
    {
      "epoch": 2.1686553955078124e-05,
      "model_forward_time": 1.7362771034240723,
      "step": 96740
    },
    {
      "dynamics_loss_avg": 0.058058428950607775,
      "epoch": 2.1686553955078124e-05,
      "step": 96740
    },
    {
      "action_loss_avg": 0.001272304807207547,
      "epoch": 2.1686553955078124e-05,
      "step": 96740
    },
    {
      "epoch": 2.1686553955078124e-05,
      "step": 96740,
      "training_step_time": 7.836517572402954
    },
    {
      "epoch": 2.168750762939453e-05,
      "model_forward_time": 1.7268126010894775,
      "step": 96741
    },
    {
      "epoch": 2.168750762939453e-05,
      "step": 96741,
      "training_step_time": 7.790960788726807
    },
    {
      "epoch": 2.168846130371094e-05,
      "model_forward_time": 2.38022780418396,
      "step": 96742
    },
    {
      "epoch": 2.168846130371094e-05,
      "step": 96742,
      "training_step_time": 8.369782209396362
    },
    {
      "epoch": 2.1689414978027342e-05,
      "model_forward_time": 1.0158674716949463,
      "step": 96743
    },
    {
      "epoch": 2.1689414978027342e-05,
      "step": 96743,
      "training_step_time": 7.7275002002716064
    },
    {
      "epoch": 2.169036865234375e-05,
      "model_forward_time": 1.7351155281066895,
      "step": 96744
    },
    {
      "epoch": 2.169036865234375e-05,
      "step": 96744,
      "training_step_time": 7.727344751358032
    },
    {
      "epoch": 2.1691322326660157e-05,
      "model_forward_time": 1.7536098957061768,
      "step": 96745
    },
    {
      "epoch": 2.1691322326660157e-05,
      "step": 96745,
      "training_step_time": 7.772943019866943
    },
    {
      "epoch": 2.169227600097656e-05,
      "model_forward_time": 1.733450174331665,
      "step": 96746
    },
    {
      "epoch": 2.169227600097656e-05,
      "step": 96746,
      "training_step_time": 7.697371959686279
    },
    {
      "epoch": 2.169322967529297e-05,
      "model_forward_time": 1.7411062717437744,
      "step": 96747
    },
    {
      "epoch": 2.169322967529297e-05,
      "step": 96747,
      "training_step_time": 7.708127737045288
    },
    {
      "epoch": 2.1694183349609376e-05,
      "model_forward_time": 1.3573880195617676,
      "step": 96748
    },
    {
      "epoch": 2.1694183349609376e-05,
      "step": 96748,
      "training_step_time": 7.856878280639648
    },
    {
      "epoch": 2.169513702392578e-05,
      "model_forward_time": 1.7410449981689453,
      "step": 96749
    },
    {
      "epoch": 2.169513702392578e-05,
      "step": 96749,
      "training_step_time": 8.330614566802979
    },
    {
      "epoch": 2.1696090698242188e-05,
      "grad_norm": 0.08851297944784164,
      "learning_rate": 2.886743127545555e-08,
      "loss": 0.057,
      "step": 96750
    },
    {
      "epoch": 2.1696090698242188e-05,
      "model_forward_time": 1.7386250495910645,
      "step": 96750
    },
    {
      "dynamics_loss_avg": 0.05567867150530219,
      "epoch": 2.1696090698242188e-05,
      "step": 96750
    },
    {
      "action_loss_avg": 0.001958242987166159,
      "epoch": 2.1696090698242188e-05,
      "step": 96750
    },
    {
      "epoch": 2.1696090698242188e-05,
      "step": 96750,
      "training_step_time": 7.878624677658081
    },
    {
      "epoch": 2.1697044372558595e-05,
      "model_forward_time": 1.7601089477539062,
      "step": 96751
    },
    {
      "epoch": 2.1697044372558595e-05,
      "step": 96751,
      "training_step_time": 8.39368724822998
    },
    {
      "epoch": 2.1697998046875e-05,
      "model_forward_time": 2.4079885482788086,
      "step": 96752
    },
    {
      "epoch": 2.1697998046875e-05,
      "step": 96752,
      "training_step_time": 7.821307897567749
    },
    {
      "epoch": 2.1698951721191406e-05,
      "model_forward_time": 1.3636672496795654,
      "step": 96753
    },
    {
      "epoch": 2.1698951721191406e-05,
      "step": 96753,
      "training_step_time": 7.6866350173950195
    },
    {
      "epoch": 2.1699905395507814e-05,
      "model_forward_time": 1.8169677257537842,
      "step": 96754
    },
    {
      "epoch": 2.1699905395507814e-05,
      "step": 96754,
      "training_step_time": 7.817911863327026
    },
    {
      "epoch": 2.1700859069824218e-05,
      "model_forward_time": 1.7437019348144531,
      "step": 96755
    },
    {
      "epoch": 2.1700859069824218e-05,
      "step": 96755,
      "training_step_time": 7.787997245788574
    },
    {
      "epoch": 2.1701812744140625e-05,
      "model_forward_time": 1.7549281120300293,
      "step": 96756
    },
    {
      "epoch": 2.1701812744140625e-05,
      "step": 96756,
      "training_step_time": 8.641682624816895
    },
    {
      "epoch": 2.1702766418457033e-05,
      "model_forward_time": 1.739262342453003,
      "step": 96757
    },
    {
      "epoch": 2.1702766418457033e-05,
      "step": 96757,
      "training_step_time": 8.811338424682617
    },
    {
      "epoch": 2.1703720092773437e-05,
      "model_forward_time": 1.7471907138824463,
      "step": 96758
    },
    {
      "epoch": 2.1703720092773437e-05,
      "step": 96758,
      "training_step_time": 7.801028728485107
    },
    {
      "epoch": 2.1704673767089844e-05,
      "model_forward_time": 1.7475652694702148,
      "step": 96759
    },
    {
      "epoch": 2.1704673767089844e-05,
      "step": 96759,
      "training_step_time": 7.935485601425171
    },
    {
      "epoch": 2.170562744140625e-05,
      "grad_norm": 0.10019117593765259,
      "learning_rate": 2.8690283093232296e-08,
      "loss": 0.056,
      "step": 96760
    },
    {
      "epoch": 2.170562744140625e-05,
      "model_forward_time": 1.7409172058105469,
      "step": 96760
    },
    {
      "dynamics_loss_avg": 0.05785161731764674,
      "epoch": 2.170562744140625e-05,
      "step": 96760
    },
    {
      "action_loss_avg": 0.0012136714169173502,
      "epoch": 2.170562744140625e-05,
      "step": 96760
    },
    {
      "epoch": 2.170562744140625e-05,
      "step": 96760,
      "training_step_time": 8.338761568069458
    },
    {
      "epoch": 2.1706581115722656e-05,
      "model_forward_time": 2.467910051345825,
      "step": 96761
    },
    {
      "epoch": 2.1706581115722656e-05,
      "step": 96761,
      "training_step_time": 7.867126703262329
    },
    {
      "epoch": 2.1707534790039063e-05,
      "model_forward_time": 1.743154525756836,
      "step": 96762
    },
    {
      "epoch": 2.1707534790039063e-05,
      "step": 96762,
      "training_step_time": 8.468742609024048
    },
    {
      "epoch": 2.170848846435547e-05,
      "model_forward_time": 1.7877521514892578,
      "step": 96763
    },
    {
      "epoch": 2.170848846435547e-05,
      "step": 96763,
      "training_step_time": 7.793794393539429
    },
    {
      "epoch": 2.1709442138671874e-05,
      "model_forward_time": 1.7427735328674316,
      "step": 96764
    },
    {
      "epoch": 2.1709442138671874e-05,
      "step": 96764,
      "training_step_time": 7.741401195526123
    },
    {
      "epoch": 2.1710395812988282e-05,
      "model_forward_time": 1.7474758625030518,
      "step": 96765
    },
    {
      "epoch": 2.1710395812988282e-05,
      "step": 96765,
      "training_step_time": 10.77777361869812
    },
    {
      "epoch": 2.1711349487304686e-05,
      "model_forward_time": 1.3842341899871826,
      "step": 96766
    },
    {
      "epoch": 2.1711349487304686e-05,
      "step": 96766,
      "training_step_time": 8.750990152359009
    },
    {
      "epoch": 2.1712303161621093e-05,
      "model_forward_time": 1.7418522834777832,
      "step": 96767
    },
    {
      "epoch": 2.1712303161621093e-05,
      "step": 96767,
      "training_step_time": 9.048365354537964
    },
    {
      "epoch": 2.17132568359375e-05,
      "model_forward_time": 1.7558362483978271,
      "step": 96768
    },
    {
      "epoch": 2.17132568359375e-05,
      "step": 96768,
      "training_step_time": 7.868404865264893
    },
    {
      "epoch": 2.1714210510253905e-05,
      "model_forward_time": 1.7709457874298096,
      "step": 96769
    },
    {
      "epoch": 2.1714210510253905e-05,
      "step": 96769,
      "training_step_time": 7.735528230667114
    },
    {
      "epoch": 2.1715164184570312e-05,
      "grad_norm": 0.08747794479131699,
      "learning_rate": 2.8513678565965654e-08,
      "loss": 0.0561,
      "step": 96770
    },
    {
      "epoch": 2.1715164184570312e-05,
      "model_forward_time": 2.374103307723999,
      "step": 96770
    },
    {
      "dynamics_loss_avg": 0.051544314436614516,
      "epoch": 2.1715164184570312e-05,
      "step": 96770
    },
    {
      "action_loss_avg": 0.00153864809544757,
      "epoch": 2.1715164184570312e-05,
      "step": 96770
    },
    {
      "epoch": 2.1715164184570312e-05,
      "step": 96770,
      "training_step_time": 7.816332817077637
    },
    {
      "epoch": 2.171611785888672e-05,
      "model_forward_time": 1.7480497360229492,
      "step": 96771
    },
    {
      "epoch": 2.171611785888672e-05,
      "step": 96771,
      "training_step_time": 7.769705057144165
    },
    {
      "epoch": 2.1717071533203124e-05,
      "model_forward_time": 1.7669777870178223,
      "step": 96772
    },
    {
      "epoch": 2.1717071533203124e-05,
      "step": 96772,
      "training_step_time": 7.817900896072388
    },
    {
      "epoch": 2.171802520751953e-05,
      "model_forward_time": 1.7521662712097168,
      "step": 96773
    },
    {
      "epoch": 2.171802520751953e-05,
      "step": 96773,
      "training_step_time": 7.667121887207031
    },
    {
      "epoch": 2.171897888183594e-05,
      "model_forward_time": 1.7505953311920166,
      "step": 96774
    },
    {
      "epoch": 2.171897888183594e-05,
      "step": 96774,
      "training_step_time": 7.278554677963257
    },
    {
      "epoch": 2.1719932556152342e-05,
      "model_forward_time": 1.7526919841766357,
      "step": 96775
    },
    {
      "epoch": 2.1719932556152342e-05,
      "step": 96775,
      "training_step_time": 7.718717575073242
    },
    {
      "epoch": 2.172088623046875e-05,
      "model_forward_time": 1.752129316329956,
      "step": 96776
    },
    {
      "epoch": 2.172088623046875e-05,
      "step": 96776,
      "training_step_time": 8.835083723068237
    },
    {
      "epoch": 2.1721839904785157e-05,
      "model_forward_time": 1.7833623886108398,
      "step": 96777
    },
    {
      "epoch": 2.1721839904785157e-05,
      "step": 96777,
      "training_step_time": 7.749330997467041
    },
    {
      "epoch": 2.172279357910156e-05,
      "model_forward_time": 1.7509145736694336,
      "step": 96778
    },
    {
      "epoch": 2.172279357910156e-05,
      "step": 96778,
      "training_step_time": 7.8344526290893555
    },
    {
      "epoch": 2.172374725341797e-05,
      "model_forward_time": 1.7533888816833496,
      "step": 96779
    },
    {
      "epoch": 2.172374725341797e-05,
      "step": 96779,
      "training_step_time": 8.320334196090698
    },
    {
      "epoch": 2.1724700927734376e-05,
      "grad_norm": 0.11323383450508118,
      "learning_rate": 2.8337617712968502e-08,
      "loss": 0.0563,
      "step": 96780
    },
    {
      "epoch": 2.1724700927734376e-05,
      "model_forward_time": 2.4041380882263184,
      "step": 96780
    },
    {
      "dynamics_loss_avg": 0.05291498340666294,
      "epoch": 2.1724700927734376e-05,
      "step": 96780
    },
    {
      "action_loss_avg": 0.0016481085942359642,
      "epoch": 2.1724700927734376e-05,
      "step": 96780
    },
    {
      "epoch": 2.1724700927734376e-05,
      "step": 96780,
      "training_step_time": 7.768627643585205
    },
    {
      "epoch": 2.172565460205078e-05,
      "model_forward_time": 1.746612787246704,
      "step": 96781
    },
    {
      "epoch": 2.172565460205078e-05,
      "step": 96781,
      "training_step_time": 7.7862303256988525
    },
    {
      "epoch": 2.1726608276367188e-05,
      "model_forward_time": 1.824038028717041,
      "step": 96782
    },
    {
      "epoch": 2.1726608276367188e-05,
      "step": 96782,
      "training_step_time": 8.306025505065918
    },
    {
      "epoch": 2.1727561950683595e-05,
      "model_forward_time": 1.3834426403045654,
      "step": 96783
    },
    {
      "epoch": 2.1727561950683595e-05,
      "step": 96783,
      "training_step_time": 7.349851608276367
    },
    {
      "epoch": 2.1728515625e-05,
      "model_forward_time": 1.754030466079712,
      "step": 96784
    },
    {
      "epoch": 2.1728515625e-05,
      "step": 96784,
      "training_step_time": 8.464953422546387
    },
    {
      "epoch": 2.1729469299316406e-05,
      "model_forward_time": 1.726073980331421,
      "step": 96785
    },
    {
      "epoch": 2.1729469299316406e-05,
      "step": 96785,
      "training_step_time": 9.868928909301758
    },
    {
      "epoch": 2.1730422973632814e-05,
      "model_forward_time": 1.745995283126831,
      "step": 96786
    },
    {
      "epoch": 2.1730422973632814e-05,
      "step": 96786,
      "training_step_time": 8.541275262832642
    },
    {
      "epoch": 2.1731376647949218e-05,
      "model_forward_time": 1.742694616317749,
      "step": 96787
    },
    {
      "epoch": 2.1731376647949218e-05,
      "step": 96787,
      "training_step_time": 8.720707178115845
    },
    {
      "epoch": 2.1732330322265625e-05,
      "model_forward_time": 1.746077060699463,
      "step": 96788
    },
    {
      "epoch": 2.1732330322265625e-05,
      "step": 96788,
      "training_step_time": 8.86212158203125
    },
    {
      "epoch": 2.1733283996582033e-05,
      "model_forward_time": 2.410517454147339,
      "step": 96789
    },
    {
      "epoch": 2.1733283996582033e-05,
      "step": 96789,
      "training_step_time": 7.728168725967407
    },
    {
      "epoch": 2.1734237670898437e-05,
      "grad_norm": 0.08474445343017578,
      "learning_rate": 2.8162100553494887e-08,
      "loss": 0.0565,
      "step": 96790
    },
    {
      "epoch": 2.1734237670898437e-05,
      "model_forward_time": 1.738574504852295,
      "step": 96790
    },
    {
      "dynamics_loss_avg": 0.05584546979516745,
      "epoch": 2.1734237670898437e-05,
      "step": 96790
    },
    {
      "action_loss_avg": 0.001490534606273286,
      "epoch": 2.1734237670898437e-05,
      "step": 96790
    },
    {
      "epoch": 2.1734237670898437e-05,
      "step": 96790,
      "training_step_time": 7.739676475524902
    },
    {
      "epoch": 2.1735191345214844e-05,
      "model_forward_time": 1.8147988319396973,
      "step": 96791
    },
    {
      "epoch": 2.1735191345214844e-05,
      "step": 96791,
      "training_step_time": 8.27399492263794
    },
    {
      "epoch": 2.173614501953125e-05,
      "model_forward_time": 1.7506868839263916,
      "step": 96792
    },
    {
      "epoch": 2.173614501953125e-05,
      "step": 96792,
      "training_step_time": 7.783480882644653
    },
    {
      "epoch": 2.1737098693847656e-05,
      "model_forward_time": 1.7459487915039062,
      "step": 96793
    },
    {
      "epoch": 2.1737098693847656e-05,
      "step": 96793,
      "training_step_time": 7.803307056427002
    },
    {
      "epoch": 2.1738052368164063e-05,
      "model_forward_time": 1.743525505065918,
      "step": 96794
    },
    {
      "epoch": 2.1738052368164063e-05,
      "step": 96794,
      "training_step_time": 10.03085470199585
    },
    {
      "epoch": 2.173900604248047e-05,
      "model_forward_time": 1.743542194366455,
      "step": 96795
    },
    {
      "epoch": 2.173900604248047e-05,
      "step": 96795,
      "training_step_time": 10.163916110992432
    },
    {
      "epoch": 2.1739959716796874e-05,
      "model_forward_time": 1.7468857765197754,
      "step": 96796
    },
    {
      "epoch": 2.1739959716796874e-05,
      "step": 96796,
      "training_step_time": 7.789779424667358
    },
    {
      "epoch": 2.1740913391113282e-05,
      "model_forward_time": 1.7428467273712158,
      "step": 96797
    },
    {
      "epoch": 2.1740913391113282e-05,
      "step": 96797,
      "training_step_time": 8.79759931564331
    },
    {
      "epoch": 2.1741867065429686e-05,
      "model_forward_time": 2.4518256187438965,
      "step": 96798
    },
    {
      "epoch": 2.1741867065429686e-05,
      "step": 96798,
      "training_step_time": 7.824065685272217
    },
    {
      "epoch": 2.1742820739746093e-05,
      "model_forward_time": 1.7534687519073486,
      "step": 96799
    },
    {
      "epoch": 2.1742820739746093e-05,
      "step": 96799,
      "training_step_time": 7.861568212509155
    },
    {
      "epoch": 2.17437744140625e-05,
      "grad_norm": 0.09230368584394455,
      "learning_rate": 2.7987127106738343e-08,
      "loss": 0.0571,
      "step": 96800
    },
    {
      "epoch": 2.17437744140625e-05,
      "model_forward_time": 1.7951993942260742,
      "step": 96800
    },
    {
      "dynamics_loss_avg": 0.04349277392029762,
      "epoch": 2.17437744140625e-05,
      "step": 96800
    },
    {
      "action_loss_avg": 0.0010613584076054395,
      "epoch": 2.17437744140625e-05,
      "step": 96800
    },
    {
      "epoch": 2.17437744140625e-05,
      "step": 96800,
      "training_step_time": 8.261871099472046
    },
    {
      "epoch": 2.1744728088378905e-05,
      "model_forward_time": 1.7477529048919678,
      "step": 96801
    },
    {
      "epoch": 2.1744728088378905e-05,
      "step": 96801,
      "training_step_time": 8.254642963409424
    },
    {
      "epoch": 2.1745681762695312e-05,
      "model_forward_time": 1.3824243545532227,
      "step": 96802
    },
    {
      "epoch": 2.1745681762695312e-05,
      "step": 96802,
      "training_step_time": 7.617717504501343
    },
    {
      "epoch": 2.174663543701172e-05,
      "model_forward_time": 1.7488605976104736,
      "step": 96803
    },
    {
      "epoch": 2.174663543701172e-05,
      "step": 96803,
      "training_step_time": 9.580845832824707
    },
    {
      "epoch": 2.1747589111328124e-05,
      "model_forward_time": 1.755258560180664,
      "step": 96804
    },
    {
      "epoch": 2.1747589111328124e-05,
      "step": 96804,
      "training_step_time": 9.583010911941528
    },
    {
      "epoch": 2.174854278564453e-05,
      "model_forward_time": 1.7404427528381348,
      "step": 96805
    },
    {
      "epoch": 2.174854278564453e-05,
      "step": 96805,
      "training_step_time": 9.226140975952148
    },
    {
      "epoch": 2.174949645996094e-05,
      "model_forward_time": 1.7468690872192383,
      "step": 96806
    },
    {
      "epoch": 2.174949645996094e-05,
      "step": 96806,
      "training_step_time": 8.59021520614624
    },
    {
      "epoch": 2.1750450134277342e-05,
      "model_forward_time": 2.361475944519043,
      "step": 96807
    },
    {
      "epoch": 2.1750450134277342e-05,
      "step": 96807,
      "training_step_time": 7.76813006401062
    },
    {
      "epoch": 2.175140380859375e-05,
      "model_forward_time": 1.7456893920898438,
      "step": 96808
    },
    {
      "epoch": 2.175140380859375e-05,
      "step": 96808,
      "training_step_time": 7.705522298812866
    },
    {
      "epoch": 2.1752357482910157e-05,
      "model_forward_time": 1.7792065143585205,
      "step": 96809
    },
    {
      "epoch": 2.1752357482910157e-05,
      "step": 96809,
      "training_step_time": 8.663692951202393
    },
    {
      "epoch": 2.175331115722656e-05,
      "grad_norm": 0.11181948333978653,
      "learning_rate": 2.7812697391834677e-08,
      "loss": 0.0554,
      "step": 96810
    },
    {
      "epoch": 2.175331115722656e-05,
      "model_forward_time": 1.736356496810913,
      "step": 96810
    },
    {
      "dynamics_loss_avg": 0.047555308882147075,
      "epoch": 2.175331115722656e-05,
      "step": 96810
    },
    {
      "action_loss_avg": 0.0012449534784536808,
      "epoch": 2.175331115722656e-05,
      "step": 96810
    },
    {
      "epoch": 2.175331115722656e-05,
      "step": 96810,
      "training_step_time": 7.702023029327393
    },
    {
      "epoch": 2.175426483154297e-05,
      "model_forward_time": 1.3720378875732422,
      "step": 96811
    },
    {
      "epoch": 2.175426483154297e-05,
      "step": 96811,
      "training_step_time": 8.70063328742981
    },
    {
      "epoch": 2.1755218505859376e-05,
      "model_forward_time": 1.7463326454162598,
      "step": 96812
    },
    {
      "epoch": 2.1755218505859376e-05,
      "step": 96812,
      "training_step_time": 9.587284564971924
    },
    {
      "epoch": 2.175617218017578e-05,
      "model_forward_time": 1.7504184246063232,
      "step": 96813
    },
    {
      "epoch": 2.175617218017578e-05,
      "step": 96813,
      "training_step_time": 8.869416236877441
    },
    {
      "epoch": 2.1757125854492188e-05,
      "model_forward_time": 1.7492988109588623,
      "step": 96814
    },
    {
      "epoch": 2.1757125854492188e-05,
      "step": 96814,
      "training_step_time": 8.751038789749146
    },
    {
      "epoch": 2.1758079528808595e-05,
      "model_forward_time": 1.7407615184783936,
      "step": 96815
    },
    {
      "epoch": 2.1758079528808595e-05,
      "step": 96815,
      "training_step_time": 7.732530355453491
    },
    {
      "epoch": 2.1759033203125e-05,
      "model_forward_time": 1.756239891052246,
      "step": 96816
    },
    {
      "epoch": 2.1759033203125e-05,
      "step": 96816,
      "training_step_time": 7.850122690200806
    },
    {
      "epoch": 2.1759986877441406e-05,
      "model_forward_time": 2.41463303565979,
      "step": 96817
    },
    {
      "epoch": 2.1759986877441406e-05,
      "step": 96817,
      "training_step_time": 7.731415033340454
    },
    {
      "epoch": 2.1760940551757814e-05,
      "model_forward_time": 1.7480781078338623,
      "step": 96818
    },
    {
      "epoch": 2.1760940551757814e-05,
      "step": 96818,
      "training_step_time": 8.831149816513062
    },
    {
      "epoch": 2.1761894226074218e-05,
      "model_forward_time": 1.7467832565307617,
      "step": 96819
    },
    {
      "epoch": 2.1761894226074218e-05,
      "step": 96819,
      "training_step_time": 7.606473445892334
    },
    {
      "epoch": 2.1762847900390625e-05,
      "grad_norm": 0.07330545783042908,
      "learning_rate": 2.7638811427858625e-08,
      "loss": 0.0557,
      "step": 96820
    },
    {
      "epoch": 2.1762847900390625e-05,
      "model_forward_time": 1.7364342212677002,
      "step": 96820
    },
    {
      "dynamics_loss_avg": 0.05947330612689257,
      "epoch": 2.1762847900390625e-05,
      "step": 96820
    },
    {
      "action_loss_avg": 0.0014075967017561197,
      "epoch": 2.1762847900390625e-05,
      "step": 96820
    },
    {
      "epoch": 2.1762847900390625e-05,
      "step": 96820,
      "training_step_time": 7.756728172302246
    },
    {
      "epoch": 2.1763801574707033e-05,
      "model_forward_time": 1.7467331886291504,
      "step": 96821
    },
    {
      "epoch": 2.1763801574707033e-05,
      "step": 96821,
      "training_step_time": 10.34042763710022
    },
    {
      "epoch": 2.1764755249023437e-05,
      "model_forward_time": 1.3753690719604492,
      "step": 96822
    },
    {
      "epoch": 2.1764755249023437e-05,
      "step": 96822,
      "training_step_time": 10.098690032958984
    },
    {
      "epoch": 2.1765708923339844e-05,
      "model_forward_time": 1.7553939819335938,
      "step": 96823
    },
    {
      "epoch": 2.1765708923339844e-05,
      "step": 96823,
      "training_step_time": 8.570336103439331
    },
    {
      "epoch": 2.176666259765625e-05,
      "model_forward_time": 1.7843647003173828,
      "step": 96824
    },
    {
      "epoch": 2.176666259765625e-05,
      "step": 96824,
      "training_step_time": 9.724954605102539
    },
    {
      "epoch": 2.1767616271972656e-05,
      "model_forward_time": 1.7444803714752197,
      "step": 96825
    },
    {
      "epoch": 2.1767616271972656e-05,
      "step": 96825,
      "training_step_time": 9.145445823669434
    },
    {
      "epoch": 2.1768569946289063e-05,
      "model_forward_time": 2.4466748237609863,
      "step": 96826
    },
    {
      "epoch": 2.1768569946289063e-05,
      "step": 96826,
      "training_step_time": 7.7197816371917725
    },
    {
      "epoch": 2.176952362060547e-05,
      "model_forward_time": 1.7526233196258545,
      "step": 96827
    },
    {
      "epoch": 2.176952362060547e-05,
      "step": 96827,
      "training_step_time": 7.755676746368408
    },
    {
      "epoch": 2.1770477294921874e-05,
      "model_forward_time": 1.8251898288726807,
      "step": 96828
    },
    {
      "epoch": 2.1770477294921874e-05,
      "step": 96828,
      "training_step_time": 7.594555377960205
    },
    {
      "epoch": 2.1771430969238282e-05,
      "model_forward_time": 1.743546962738037,
      "step": 96829
    },
    {
      "epoch": 2.1771430969238282e-05,
      "step": 96829,
      "training_step_time": 7.658102512359619
    },
    {
      "epoch": 2.1772384643554686e-05,
      "grad_norm": 0.0948953628540039,
      "learning_rate": 2.7465469233826647e-08,
      "loss": 0.055,
      "step": 96830
    },
    {
      "epoch": 2.1772384643554686e-05,
      "model_forward_time": 1.7420783042907715,
      "step": 96830
    },
    {
      "dynamics_loss_avg": 0.044014444947242735,
      "epoch": 2.1772384643554686e-05,
      "step": 96830
    },
    {
      "action_loss_avg": 0.001011511473916471,
      "epoch": 2.1772384643554686e-05,
      "step": 96830
    },
    {
      "epoch": 2.1772384643554686e-05,
      "step": 96830,
      "training_step_time": 8.216824769973755
    },
    {
      "epoch": 2.1773338317871093e-05,
      "model_forward_time": 1.757821798324585,
      "step": 96831
    },
    {
      "epoch": 2.1773338317871093e-05,
      "step": 96831,
      "training_step_time": 9.336647272109985
    },
    {
      "epoch": 2.17742919921875e-05,
      "model_forward_time": 1.7476410865783691,
      "step": 96832
    },
    {
      "epoch": 2.17742919921875e-05,
      "step": 96832,
      "training_step_time": 7.698964357376099
    },
    {
      "epoch": 2.1775245666503905e-05,
      "model_forward_time": 1.782364845275879,
      "step": 96833
    },
    {
      "epoch": 2.1775245666503905e-05,
      "step": 96833,
      "training_step_time": 7.749444484710693
    },
    {
      "epoch": 2.1776199340820312e-05,
      "model_forward_time": 1.744901418685913,
      "step": 96834
    },
    {
      "epoch": 2.1776199340820312e-05,
      "step": 96834,
      "training_step_time": 8.351699829101562
    },
    {
      "epoch": 2.177715301513672e-05,
      "model_forward_time": 2.4043967723846436,
      "step": 96835
    },
    {
      "epoch": 2.177715301513672e-05,
      "step": 96835,
      "training_step_time": 8.826056241989136
    },
    {
      "epoch": 2.1778106689453124e-05,
      "model_forward_time": 1.7550990581512451,
      "step": 96836
    },
    {
      "epoch": 2.1778106689453124e-05,
      "step": 96836,
      "training_step_time": 7.768971920013428
    },
    {
      "epoch": 2.177906036376953e-05,
      "model_forward_time": 1.8003427982330322,
      "step": 96837
    },
    {
      "epoch": 2.177906036376953e-05,
      "step": 96837,
      "training_step_time": 8.179822206497192
    },
    {
      "epoch": 2.178001403808594e-05,
      "model_forward_time": 1.7428357601165771,
      "step": 96838
    },
    {
      "epoch": 2.178001403808594e-05,
      "step": 96838,
      "training_step_time": 7.731487989425659
    },
    {
      "epoch": 2.1780967712402342e-05,
      "model_forward_time": 1.756521224975586,
      "step": 96839
    },
    {
      "epoch": 2.1780967712402342e-05,
      "step": 96839,
      "training_step_time": 9.66503357887268
    },
    {
      "epoch": 2.178192138671875e-05,
      "grad_norm": 0.09548493474721909,
      "learning_rate": 2.729267082869469e-08,
      "loss": 0.0548,
      "step": 96840
    },
    {
      "epoch": 2.178192138671875e-05,
      "model_forward_time": 1.743783712387085,
      "step": 96840
    },
    {
      "dynamics_loss_avg": 0.05509252026677132,
      "epoch": 2.178192138671875e-05,
      "step": 96840
    },
    {
      "action_loss_avg": 0.0011907538180821575,
      "epoch": 2.178192138671875e-05,
      "step": 96840
    },
    {
      "epoch": 2.178192138671875e-05,
      "step": 96840,
      "training_step_time": 8.67176604270935
    },
    {
      "epoch": 2.1782875061035157e-05,
      "model_forward_time": 1.7538678646087646,
      "step": 96841
    },
    {
      "epoch": 2.1782875061035157e-05,
      "step": 96841,
      "training_step_time": 7.878089666366577
    },
    {
      "epoch": 2.178382873535156e-05,
      "model_forward_time": 1.7449102401733398,
      "step": 96842
    },
    {
      "epoch": 2.178382873535156e-05,
      "step": 96842,
      "training_step_time": 7.818479537963867
    },
    {
      "epoch": 2.178478240966797e-05,
      "model_forward_time": 1.754786729812622,
      "step": 96843
    },
    {
      "epoch": 2.178478240966797e-05,
      "step": 96843,
      "training_step_time": 8.404695510864258
    },
    {
      "epoch": 2.1785736083984376e-05,
      "model_forward_time": 1.756117343902588,
      "step": 96844
    },
    {
      "epoch": 2.1785736083984376e-05,
      "step": 96844,
      "training_step_time": 8.40509033203125
    },
    {
      "epoch": 2.178668975830078e-05,
      "model_forward_time": 1.745265007019043,
      "step": 96845
    },
    {
      "epoch": 2.178668975830078e-05,
      "step": 96845,
      "training_step_time": 7.800585985183716
    },
    {
      "epoch": 2.1787643432617188e-05,
      "model_forward_time": 1.768017053604126,
      "step": 96846
    },
    {
      "epoch": 2.1787643432617188e-05,
      "step": 96846,
      "training_step_time": 7.684073209762573
    },
    {
      "epoch": 2.1788597106933595e-05,
      "model_forward_time": 1.7516605854034424,
      "step": 96847
    },
    {
      "epoch": 2.1788597106933595e-05,
      "step": 96847,
      "training_step_time": 7.7400758266448975
    },
    {
      "epoch": 2.178955078125e-05,
      "model_forward_time": 1.7487943172454834,
      "step": 96848
    },
    {
      "epoch": 2.178955078125e-05,
      "step": 96848,
      "training_step_time": 8.82962155342102
    },
    {
      "epoch": 2.1790504455566406e-05,
      "model_forward_time": 1.7503859996795654,
      "step": 96849
    },
    {
      "epoch": 2.1790504455566406e-05,
      "step": 96849,
      "training_step_time": 8.60321044921875
    },
    {
      "epoch": 2.1791458129882814e-05,
      "grad_norm": 0.08174645155668259,
      "learning_rate": 2.7120416231359857e-08,
      "loss": 0.0554,
      "step": 96850
    },
    {
      "epoch": 2.1791458129882814e-05,
      "model_forward_time": 1.7418360710144043,
      "step": 96850
    },
    {
      "dynamics_loss_avg": 0.05702066496014595,
      "epoch": 2.1791458129882814e-05,
      "step": 96850
    },
    {
      "action_loss_avg": 0.001227307073713746,
      "epoch": 2.1791458129882814e-05,
      "step": 96850
    },
    {
      "epoch": 2.1791458129882814e-05,
      "step": 96850,
      "training_step_time": 7.880041122436523
    },
    {
      "epoch": 2.1792411804199218e-05,
      "model_forward_time": 1.7339451313018799,
      "step": 96851
    },
    {
      "epoch": 2.1792411804199218e-05,
      "step": 96851,
      "training_step_time": 7.853259086608887
    },
    {
      "epoch": 2.1793365478515625e-05,
      "model_forward_time": 1.7428088188171387,
      "step": 96852
    },
    {
      "epoch": 2.1793365478515625e-05,
      "step": 96852,
      "training_step_time": 7.812690019607544
    },
    {
      "epoch": 2.1794319152832033e-05,
      "model_forward_time": 1.380197286605835,
      "step": 96853
    },
    {
      "epoch": 2.1794319152832033e-05,
      "step": 96853,
      "training_step_time": 7.841280937194824
    },
    {
      "epoch": 2.1795272827148437e-05,
      "model_forward_time": 2.4173474311828613,
      "step": 96854
    },
    {
      "epoch": 2.1795272827148437e-05,
      "step": 96854,
      "training_step_time": 7.726568937301636
    },
    {
      "epoch": 2.1796226501464844e-05,
      "model_forward_time": 1.7463338375091553,
      "step": 96855
    },
    {
      "epoch": 2.1796226501464844e-05,
      "step": 96855,
      "training_step_time": 8.16972279548645
    },
    {
      "epoch": 2.179718017578125e-05,
      "model_forward_time": 1.8249437808990479,
      "step": 96856
    },
    {
      "epoch": 2.179718017578125e-05,
      "step": 96856,
      "training_step_time": 7.627626180648804
    },
    {
      "epoch": 2.1798133850097656e-05,
      "model_forward_time": 1.7497034072875977,
      "step": 96857
    },
    {
      "epoch": 2.1798133850097656e-05,
      "step": 96857,
      "training_step_time": 7.774150848388672
    },
    {
      "epoch": 2.1799087524414063e-05,
      "model_forward_time": 1.7479088306427002,
      "step": 96858
    },
    {
      "epoch": 2.1799087524414063e-05,
      "step": 96858,
      "training_step_time": 8.759346008300781
    },
    {
      "epoch": 2.180004119873047e-05,
      "model_forward_time": 1.7975950241088867,
      "step": 96859
    },
    {
      "epoch": 2.180004119873047e-05,
      "step": 96859,
      "training_step_time": 7.945608139038086
    },
    {
      "epoch": 2.1800994873046874e-05,
      "grad_norm": 0.1231103166937828,
      "learning_rate": 2.694870546065931e-08,
      "loss": 0.0547,
      "step": 96860
    },
    {
      "epoch": 2.1800994873046874e-05,
      "model_forward_time": 1.7403178215026855,
      "step": 96860
    },
    {
      "dynamics_loss_avg": 0.04152870513498783,
      "epoch": 2.1800994873046874e-05,
      "step": 96860
    },
    {
      "action_loss_avg": 0.0008219678886234761,
      "epoch": 2.1800994873046874e-05,
      "step": 96860
    },
    {
      "epoch": 2.1800994873046874e-05,
      "step": 96860,
      "training_step_time": 7.815829277038574
    },
    {
      "epoch": 2.1801948547363282e-05,
      "model_forward_time": 1.745375394821167,
      "step": 96861
    },
    {
      "epoch": 2.1801948547363282e-05,
      "step": 96861,
      "training_step_time": 8.525995969772339
    },
    {
      "epoch": 2.1802902221679686e-05,
      "model_forward_time": 1.7519736289978027,
      "step": 96862
    },
    {
      "epoch": 2.1802902221679686e-05,
      "step": 96862,
      "training_step_time": 8.876096487045288
    },
    {
      "epoch": 2.1803855895996093e-05,
      "model_forward_time": 2.410682201385498,
      "step": 96863
    },
    {
      "epoch": 2.1803855895996093e-05,
      "step": 96863,
      "training_step_time": 7.792693138122559
    },
    {
      "epoch": 2.18048095703125e-05,
      "model_forward_time": 1.7510597705841064,
      "step": 96864
    },
    {
      "epoch": 2.18048095703125e-05,
      "step": 96864,
      "training_step_time": 7.729733228683472
    },
    {
      "epoch": 2.1805763244628905e-05,
      "model_forward_time": 1.8217353820800781,
      "step": 96865
    },
    {
      "epoch": 2.1805763244628905e-05,
      "step": 96865,
      "training_step_time": 7.777299642562866
    },
    {
      "epoch": 2.1806716918945312e-05,
      "model_forward_time": 1.7510144710540771,
      "step": 96866
    },
    {
      "epoch": 2.1806716918945312e-05,
      "step": 96866,
      "training_step_time": 7.700573682785034
    },
    {
      "epoch": 2.180767059326172e-05,
      "model_forward_time": 1.7416512966156006,
      "step": 96867
    },
    {
      "epoch": 2.180767059326172e-05,
      "step": 96867,
      "training_step_time": 9.392903804779053
    },
    {
      "epoch": 2.1808624267578124e-05,
      "model_forward_time": 1.754331350326538,
      "step": 96868
    },
    {
      "epoch": 2.1808624267578124e-05,
      "step": 96868,
      "training_step_time": 7.77217698097229
    },
    {
      "epoch": 2.180957794189453e-05,
      "model_forward_time": 1.743100881576538,
      "step": 96869
    },
    {
      "epoch": 2.180957794189453e-05,
      "step": 96869,
      "training_step_time": 8.221716403961182
    },
    {
      "epoch": 2.181053161621094e-05,
      "grad_norm": 0.06931637972593307,
      "learning_rate": 2.677753853537246e-08,
      "loss": 0.0554,
      "step": 96870
    },
    {
      "epoch": 2.181053161621094e-05,
      "model_forward_time": 1.7447481155395508,
      "step": 96870
    },
    {
      "dynamics_loss_avg": 0.04380248198285699,
      "epoch": 2.181053161621094e-05,
      "step": 96870
    },
    {
      "action_loss_avg": 0.0018189606693340466,
      "epoch": 2.181053161621094e-05,
      "step": 96870
    },
    {
      "epoch": 2.181053161621094e-05,
      "step": 96870,
      "training_step_time": 8.460790634155273
    },
    {
      "epoch": 2.1811485290527342e-05,
      "model_forward_time": 1.7445156574249268,
      "step": 96871
    },
    {
      "epoch": 2.1811485290527342e-05,
      "step": 96871,
      "training_step_time": 7.8299360275268555
    },
    {
      "epoch": 2.181243896484375e-05,
      "model_forward_time": 2.4404373168945312,
      "step": 96872
    },
    {
      "epoch": 2.181243896484375e-05,
      "step": 96872,
      "training_step_time": 7.8020243644714355
    },
    {
      "epoch": 2.1813392639160157e-05,
      "model_forward_time": 1.7545335292816162,
      "step": 96873
    },
    {
      "epoch": 2.1813392639160157e-05,
      "step": 96873,
      "training_step_time": 7.7923314571380615
    },
    {
      "epoch": 2.181434631347656e-05,
      "model_forward_time": 1.8215768337249756,
      "step": 96874
    },
    {
      "epoch": 2.181434631347656e-05,
      "step": 96874,
      "training_step_time": 7.925198793411255
    },
    {
      "epoch": 2.181529998779297e-05,
      "model_forward_time": 1.7456023693084717,
      "step": 96875
    },
    {
      "epoch": 2.181529998779297e-05,
      "step": 96875,
      "training_step_time": 8.244765520095825
    },
    {
      "epoch": 2.1816253662109376e-05,
      "model_forward_time": 1.754622220993042,
      "step": 96876
    },
    {
      "epoch": 2.1816253662109376e-05,
      "step": 96876,
      "training_step_time": 8.608097076416016
    },
    {
      "epoch": 2.181720733642578e-05,
      "model_forward_time": 1.7488064765930176,
      "step": 96877
    },
    {
      "epoch": 2.181720733642578e-05,
      "step": 96877,
      "training_step_time": 9.444609642028809
    },
    {
      "epoch": 2.1818161010742188e-05,
      "model_forward_time": 1.7517683506011963,
      "step": 96878
    },
    {
      "epoch": 2.1818161010742188e-05,
      "step": 96878,
      "training_step_time": 9.435250282287598
    },
    {
      "epoch": 2.1819114685058595e-05,
      "model_forward_time": 1.743617057800293,
      "step": 96879
    },
    {
      "epoch": 2.1819114685058595e-05,
      "step": 96879,
      "training_step_time": 8.673060655593872
    },
    {
      "epoch": 2.1820068359375e-05,
      "grad_norm": 0.07111372798681259,
      "learning_rate": 2.6606915474216012e-08,
      "loss": 0.0551,
      "step": 96880
    },
    {
      "epoch": 2.1820068359375e-05,
      "model_forward_time": 1.7533280849456787,
      "step": 96880
    },
    {
      "dynamics_loss_avg": 0.06516530457884073,
      "epoch": 2.1820068359375e-05,
      "step": 96880
    },
    {
      "action_loss_avg": 0.0010642886758432724,
      "epoch": 2.1820068359375e-05,
      "step": 96880
    },
    {
      "epoch": 2.1820068359375e-05,
      "step": 96880,
      "training_step_time": 7.736995220184326
    },
    {
      "epoch": 2.1821022033691406e-05,
      "model_forward_time": 2.4255590438842773,
      "step": 96881
    },
    {
      "epoch": 2.1821022033691406e-05,
      "step": 96881,
      "training_step_time": 7.861505746841431
    },
    {
      "epoch": 2.1821975708007814e-05,
      "model_forward_time": 1.746206521987915,
      "step": 96882
    },
    {
      "epoch": 2.1821975708007814e-05,
      "step": 96882,
      "training_step_time": 8.238882303237915
    },
    {
      "epoch": 2.1822929382324218e-05,
      "model_forward_time": 1.8127820491790771,
      "step": 96883
    },
    {
      "epoch": 2.1822929382324218e-05,
      "step": 96883,
      "training_step_time": 7.894244194030762
    },
    {
      "epoch": 2.1823883056640625e-05,
      "model_forward_time": 1.7439353466033936,
      "step": 96884
    },
    {
      "epoch": 2.1823883056640625e-05,
      "step": 96884,
      "training_step_time": 8.703843355178833
    },
    {
      "epoch": 2.1824836730957033e-05,
      "model_forward_time": 1.7442967891693115,
      "step": 96885
    },
    {
      "epoch": 2.1824836730957033e-05,
      "step": 96885,
      "training_step_time": 9.851967334747314
    },
    {
      "epoch": 2.1825790405273437e-05,
      "model_forward_time": 1.7466907501220703,
      "step": 96886
    },
    {
      "epoch": 2.1825790405273437e-05,
      "step": 96886,
      "training_step_time": 9.407999992370605
    },
    {
      "epoch": 2.1826744079589844e-05,
      "model_forward_time": 1.7464187145233154,
      "step": 96887
    },
    {
      "epoch": 2.1826744079589844e-05,
      "step": 96887,
      "training_step_time": 8.742950677871704
    },
    {
      "epoch": 2.182769775390625e-05,
      "model_forward_time": 1.7439336776733398,
      "step": 96888
    },
    {
      "epoch": 2.182769775390625e-05,
      "step": 96888,
      "training_step_time": 9.454131126403809
    },
    {
      "epoch": 2.1828651428222656e-05,
      "model_forward_time": 1.7514407634735107,
      "step": 96889
    },
    {
      "epoch": 2.1828651428222656e-05,
      "step": 96889,
      "training_step_time": 7.808105707168579
    },
    {
      "epoch": 2.1829605102539063e-05,
      "grad_norm": 0.10769046097993851,
      "learning_rate": 2.643683629585003e-08,
      "loss": 0.0538,
      "step": 96890
    },
    {
      "epoch": 2.1829605102539063e-05,
      "model_forward_time": 2.3909072875976562,
      "step": 96890
    },
    {
      "dynamics_loss_avg": 0.05949979554861784,
      "epoch": 2.1829605102539063e-05,
      "step": 96890
    },
    {
      "action_loss_avg": 0.0012329749850323423,
      "epoch": 2.1829605102539063e-05,
      "step": 96890
    },
    {
      "epoch": 2.1829605102539063e-05,
      "step": 96890,
      "training_step_time": 7.723445177078247
    },
    {
      "epoch": 2.1830558776855467e-05,
      "model_forward_time": 1.7423076629638672,
      "step": 96891
    },
    {
      "epoch": 2.1830558776855467e-05,
      "step": 96891,
      "training_step_time": 7.908051490783691
    },
    {
      "epoch": 2.1831512451171874e-05,
      "model_forward_time": 1.7797935009002686,
      "step": 96892
    },
    {
      "epoch": 2.1831512451171874e-05,
      "step": 96892,
      "training_step_time": 8.605048418045044
    },
    {
      "epoch": 2.1832466125488282e-05,
      "model_forward_time": 1.7731289863586426,
      "step": 96893
    },
    {
      "epoch": 2.1832466125488282e-05,
      "step": 96893,
      "training_step_time": 7.724445104598999
    },
    {
      "epoch": 2.1833419799804686e-05,
      "model_forward_time": 1.7464892864227295,
      "step": 96894
    },
    {
      "epoch": 2.1833419799804686e-05,
      "step": 96894,
      "training_step_time": 8.43672800064087
    },
    {
      "epoch": 2.1834373474121093e-05,
      "model_forward_time": 1.75199294090271,
      "step": 96895
    },
    {
      "epoch": 2.1834373474121093e-05,
      "step": 96895,
      "training_step_time": 8.648249864578247
    },
    {
      "epoch": 2.18353271484375e-05,
      "model_forward_time": 1.7411012649536133,
      "step": 96896
    },
    {
      "epoch": 2.18353271484375e-05,
      "step": 96896,
      "training_step_time": 8.895888328552246
    },
    {
      "epoch": 2.1836280822753905e-05,
      "model_forward_time": 1.745725154876709,
      "step": 96897
    },
    {
      "epoch": 2.1836280822753905e-05,
      "step": 96897,
      "training_step_time": 9.497778415679932
    },
    {
      "epoch": 2.1837234497070312e-05,
      "model_forward_time": 1.7383594512939453,
      "step": 96898
    },
    {
      "epoch": 2.1837234497070312e-05,
      "step": 96898,
      "training_step_time": 7.873036623001099
    },
    {
      "epoch": 2.183818817138672e-05,
      "model_forward_time": 1.751084566116333,
      "step": 96899
    },
    {
      "epoch": 2.183818817138672e-05,
      "step": 96899,
      "training_step_time": 7.789824724197388
    },
    {
      "epoch": 2.1839141845703124e-05,
      "grad_norm": 0.0970548689365387,
      "learning_rate": 2.626730101887409e-08,
      "loss": 0.0539,
      "step": 96900
    },
    {
      "epoch": 2.1839141845703124e-05,
      "model_forward_time": 2.4392974376678467,
      "step": 96900
    },
    {
      "dynamics_loss_avg": 0.05894763898104429,
      "epoch": 2.1839141845703124e-05,
      "step": 96900
    },
    {
      "action_loss_avg": 0.0010985107088345102,
      "epoch": 2.1839141845703124e-05,
      "step": 96900
    },
    {
      "epoch": 2.1839141845703124e-05,
      "step": 96900,
      "training_step_time": 8.832939147949219
    },
    {
      "epoch": 2.184009552001953e-05,
      "model_forward_time": 1.7495899200439453,
      "step": 96901
    },
    {
      "epoch": 2.184009552001953e-05,
      "step": 96901,
      "training_step_time": 7.807389497756958
    },
    {
      "epoch": 2.184104919433594e-05,
      "model_forward_time": 1.374035358428955,
      "step": 96902
    },
    {
      "epoch": 2.184104919433594e-05,
      "step": 96902,
      "training_step_time": 7.7144458293914795
    },
    {
      "epoch": 2.1842002868652343e-05,
      "model_forward_time": 1.3911774158477783,
      "step": 96903
    },
    {
      "epoch": 2.1842002868652343e-05,
      "step": 96903,
      "training_step_time": 7.774795293807983
    },
    {
      "epoch": 2.184295654296875e-05,
      "model_forward_time": 1.7445921897888184,
      "step": 96904
    },
    {
      "epoch": 2.184295654296875e-05,
      "step": 96904,
      "training_step_time": 7.807135343551636
    },
    {
      "epoch": 2.1843910217285157e-05,
      "model_forward_time": 1.7424087524414062,
      "step": 96905
    },
    {
      "epoch": 2.1843910217285157e-05,
      "step": 96905,
      "training_step_time": 9.865410089492798
    },
    {
      "epoch": 2.184486389160156e-05,
      "model_forward_time": 1.7450578212738037,
      "step": 96906
    },
    {
      "epoch": 2.184486389160156e-05,
      "step": 96906,
      "training_step_time": 8.836493015289307
    },
    {
      "epoch": 2.184581756591797e-05,
      "model_forward_time": 1.7453677654266357,
      "step": 96907
    },
    {
      "epoch": 2.184581756591797e-05,
      "step": 96907,
      "training_step_time": 7.796457290649414
    },
    {
      "epoch": 2.1846771240234376e-05,
      "model_forward_time": 1.7537755966186523,
      "step": 96908
    },
    {
      "epoch": 2.1846771240234376e-05,
      "step": 96908,
      "training_step_time": 7.906991004943848
    },
    {
      "epoch": 2.184772491455078e-05,
      "model_forward_time": 2.410099506378174,
      "step": 96909
    },
    {
      "epoch": 2.184772491455078e-05,
      "step": 96909,
      "training_step_time": 7.85013222694397
    },
    {
      "epoch": 2.1848678588867188e-05,
      "grad_norm": 0.07528884708881378,
      "learning_rate": 2.609830966182836e-08,
      "loss": 0.0542,
      "step": 96910
    },
    {
      "epoch": 2.1848678588867188e-05,
      "model_forward_time": 1.3641679286956787,
      "step": 96910
    },
    {
      "dynamics_loss_avg": 0.05322728343307972,
      "epoch": 2.1848678588867188e-05,
      "step": 96910
    },
    {
      "action_loss_avg": 0.0012840860290452838,
      "epoch": 2.1848678588867188e-05,
      "step": 96910
    },
    {
      "epoch": 2.1848678588867188e-05,
      "step": 96910,
      "training_step_time": 7.788457632064819
    },
    {
      "epoch": 2.1849632263183595e-05,
      "model_forward_time": 1.8204123973846436,
      "step": 96911
    },
    {
      "epoch": 2.1849632263183595e-05,
      "step": 96911,
      "training_step_time": 7.772938966751099
    },
    {
      "epoch": 2.18505859375e-05,
      "model_forward_time": 1.3789939880371094,
      "step": 96912
    },
    {
      "epoch": 2.18505859375e-05,
      "step": 96912,
      "training_step_time": 7.747560501098633
    },
    {
      "epoch": 2.1851539611816406e-05,
      "model_forward_time": 1.7521939277648926,
      "step": 96913
    },
    {
      "epoch": 2.1851539611816406e-05,
      "step": 96913,
      "training_step_time": 7.758300065994263
    },
    {
      "epoch": 2.1852493286132814e-05,
      "model_forward_time": 1.7459092140197754,
      "step": 96914
    },
    {
      "epoch": 2.1852493286132814e-05,
      "step": 96914,
      "training_step_time": 8.879715204238892
    },
    {
      "epoch": 2.1853446960449218e-05,
      "model_forward_time": 1.7331008911132812,
      "step": 96915
    },
    {
      "epoch": 2.1853446960449218e-05,
      "step": 96915,
      "training_step_time": 8.889076709747314
    },
    {
      "epoch": 2.1854400634765625e-05,
      "model_forward_time": 1.7415082454681396,
      "step": 96916
    },
    {
      "epoch": 2.1854400634765625e-05,
      "step": 96916,
      "training_step_time": 8.758687019348145
    },
    {
      "epoch": 2.1855354309082033e-05,
      "model_forward_time": 1.7459664344787598,
      "step": 96917
    },
    {
      "epoch": 2.1855354309082033e-05,
      "step": 96917,
      "training_step_time": 7.86685585975647
    },
    {
      "epoch": 2.1856307983398437e-05,
      "model_forward_time": 2.3924102783203125,
      "step": 96918
    },
    {
      "epoch": 2.1856307983398437e-05,
      "step": 96918,
      "training_step_time": 7.885210037231445
    },
    {
      "epoch": 2.1857261657714844e-05,
      "model_forward_time": 1.751542568206787,
      "step": 96919
    },
    {
      "epoch": 2.1857261657714844e-05,
      "step": 96919,
      "training_step_time": 8.799252033233643
    },
    {
      "epoch": 2.185821533203125e-05,
      "grad_norm": 0.11591681838035583,
      "learning_rate": 2.592986224319305e-08,
      "loss": 0.0554,
      "step": 96920
    },
    {
      "epoch": 2.185821533203125e-05,
      "model_forward_time": 1.7983207702636719,
      "step": 96920
    },
    {
      "dynamics_loss_avg": 0.04822775516659021,
      "epoch": 2.185821533203125e-05,
      "step": 96920
    },
    {
      "action_loss_avg": 0.001426122128032148,
      "epoch": 2.185821533203125e-05,
      "step": 96920
    },
    {
      "epoch": 2.185821533203125e-05,
      "step": 96920,
      "training_step_time": 7.737047910690308
    },
    {
      "epoch": 2.1859169006347656e-05,
      "model_forward_time": 1.744947910308838,
      "step": 96921
    },
    {
      "epoch": 2.1859169006347656e-05,
      "step": 96921,
      "training_step_time": 7.791225910186768
    },
    {
      "epoch": 2.1860122680664063e-05,
      "model_forward_time": 1.7484714984893799,
      "step": 96922
    },
    {
      "epoch": 2.1860122680664063e-05,
      "step": 96922,
      "training_step_time": 9.676170825958252
    },
    {
      "epoch": 2.1861076354980467e-05,
      "model_forward_time": 1.7370810508728027,
      "step": 96923
    },
    {
      "epoch": 2.1861076354980467e-05,
      "step": 96923,
      "training_step_time": 7.8708343505859375
    },
    {
      "epoch": 2.1862030029296875e-05,
      "model_forward_time": 1.7420127391815186,
      "step": 96924
    },
    {
      "epoch": 2.1862030029296875e-05,
      "step": 96924,
      "training_step_time": 9.526721477508545
    },
    {
      "epoch": 2.1862983703613282e-05,
      "model_forward_time": 1.3739349842071533,
      "step": 96925
    },
    {
      "epoch": 2.1862983703613282e-05,
      "step": 96925,
      "training_step_time": 8.746004581451416
    },
    {
      "epoch": 2.1863937377929686e-05,
      "model_forward_time": 1.7407000064849854,
      "step": 96926
    },
    {
      "epoch": 2.1863937377929686e-05,
      "step": 96926,
      "training_step_time": 7.817360877990723
    },
    {
      "epoch": 2.1864891052246093e-05,
      "model_forward_time": 1.751922607421875,
      "step": 96927
    },
    {
      "epoch": 2.1864891052246093e-05,
      "step": 96927,
      "training_step_time": 8.37111520767212
    },
    {
      "epoch": 2.18658447265625e-05,
      "model_forward_time": 1.9952785968780518,
      "step": 96928
    },
    {
      "epoch": 2.18658447265625e-05,
      "step": 96928,
      "training_step_time": 7.779373407363892
    },
    {
      "epoch": 2.1866798400878905e-05,
      "model_forward_time": 1.8189163208007812,
      "step": 96929
    },
    {
      "epoch": 2.1866798400878905e-05,
      "step": 96929,
      "training_step_time": 7.7432496547698975
    },
    {
      "epoch": 2.1867752075195312e-05,
      "grad_norm": 0.08682259917259216,
      "learning_rate": 2.5761958781390097e-08,
      "loss": 0.0555,
      "step": 96930
    },
    {
      "epoch": 2.1867752075195312e-05,
      "model_forward_time": 1.75056791305542,
      "step": 96930
    },
    {
      "dynamics_loss_avg": 0.0581937512382865,
      "epoch": 2.1867752075195312e-05,
      "step": 96930
    },
    {
      "action_loss_avg": 0.001382994549931027,
      "epoch": 2.1867752075195312e-05,
      "step": 96930
    },
    {
      "epoch": 2.1867752075195312e-05,
      "step": 96930,
      "training_step_time": 7.824785470962524
    },
    {
      "epoch": 2.186870574951172e-05,
      "model_forward_time": 1.7555365562438965,
      "step": 96931
    },
    {
      "epoch": 2.186870574951172e-05,
      "step": 96931,
      "training_step_time": 8.759912490844727
    },
    {
      "epoch": 2.1869659423828124e-05,
      "model_forward_time": 1.7461292743682861,
      "step": 96932
    },
    {
      "epoch": 2.1869659423828124e-05,
      "step": 96932,
      "training_step_time": 8.828564882278442
    },
    {
      "epoch": 2.187061309814453e-05,
      "model_forward_time": 1.7549493312835693,
      "step": 96933
    },
    {
      "epoch": 2.187061309814453e-05,
      "step": 96933,
      "training_step_time": 8.59998106956482
    },
    {
      "epoch": 2.187156677246094e-05,
      "model_forward_time": 1.3754525184631348,
      "step": 96934
    },
    {
      "epoch": 2.187156677246094e-05,
      "step": 96934,
      "training_step_time": 7.6464622020721436
    },
    {
      "epoch": 2.1872520446777343e-05,
      "model_forward_time": 1.7661468982696533,
      "step": 96935
    },
    {
      "epoch": 2.1872520446777343e-05,
      "step": 96935,
      "training_step_time": 7.810290098190308
    },
    {
      "epoch": 2.187347412109375e-05,
      "model_forward_time": 1.3821356296539307,
      "step": 96936
    },
    {
      "epoch": 2.187347412109375e-05,
      "step": 96936,
      "training_step_time": 7.779391527175903
    },
    {
      "epoch": 2.1874427795410157e-05,
      "model_forward_time": 2.4272189140319824,
      "step": 96937
    },
    {
      "epoch": 2.1874427795410157e-05,
      "step": 96937,
      "training_step_time": 7.7651238441467285
    },
    {
      "epoch": 2.187538146972656e-05,
      "model_forward_time": 1.7945921421051025,
      "step": 96938
    },
    {
      "epoch": 2.187538146972656e-05,
      "step": 96938,
      "training_step_time": 7.811901330947876
    },
    {
      "epoch": 2.187633514404297e-05,
      "model_forward_time": 1.7627544403076172,
      "step": 96939
    },
    {
      "epoch": 2.187633514404297e-05,
      "step": 96939,
      "training_step_time": 7.7070863246917725
    },
    {
      "epoch": 2.1877288818359376e-05,
      "grad_norm": 0.09813407063484192,
      "learning_rate": 2.5594599294780366e-08,
      "loss": 0.0554,
      "step": 96940
    },
    {
      "epoch": 2.1877288818359376e-05,
      "model_forward_time": 1.751462697982788,
      "step": 96940
    },
    {
      "dynamics_loss_avg": 0.04500573668628931,
      "epoch": 2.1877288818359376e-05,
      "step": 96940
    },
    {
      "action_loss_avg": 0.0011867305904161185,
      "epoch": 2.1877288818359376e-05,
      "step": 96940
    },
    {
      "epoch": 2.1877288818359376e-05,
      "step": 96940,
      "training_step_time": 8.235260486602783
    },
    {
      "epoch": 2.187824249267578e-05,
      "model_forward_time": 1.7492008209228516,
      "step": 96941
    },
    {
      "epoch": 2.187824249267578e-05,
      "step": 96941,
      "training_step_time": 8.673212766647339
    },
    {
      "epoch": 2.1879196166992188e-05,
      "model_forward_time": 1.3999693393707275,
      "step": 96942
    },
    {
      "epoch": 2.1879196166992188e-05,
      "step": 96942,
      "training_step_time": 7.751219272613525
    },
    {
      "epoch": 2.1880149841308595e-05,
      "model_forward_time": 1.7561650276184082,
      "step": 96943
    },
    {
      "epoch": 2.1880149841308595e-05,
      "step": 96943,
      "training_step_time": 9.120654106140137
    },
    {
      "epoch": 2.1881103515625e-05,
      "model_forward_time": 1.7575442790985107,
      "step": 96944
    },
    {
      "epoch": 2.1881103515625e-05,
      "step": 96944,
      "training_step_time": 7.772662162780762
    },
    {
      "epoch": 2.1882057189941407e-05,
      "model_forward_time": 1.760343313217163,
      "step": 96945
    },
    {
      "epoch": 2.1882057189941407e-05,
      "step": 96945,
      "training_step_time": 7.7618091106414795
    },
    {
      "epoch": 2.1883010864257814e-05,
      "model_forward_time": 1.759160041809082,
      "step": 96946
    },
    {
      "epoch": 2.1883010864257814e-05,
      "step": 96946,
      "training_step_time": 7.8754754066467285
    },
    {
      "epoch": 2.1883964538574218e-05,
      "model_forward_time": 2.4323952198028564,
      "step": 96947
    },
    {
      "epoch": 2.1883964538574218e-05,
      "step": 96947,
      "training_step_time": 7.791211128234863
    },
    {
      "epoch": 2.1884918212890625e-05,
      "model_forward_time": 1.7763710021972656,
      "step": 96948
    },
    {
      "epoch": 2.1884918212890625e-05,
      "step": 96948,
      "training_step_time": 7.69071626663208
    },
    {
      "epoch": 2.1885871887207033e-05,
      "model_forward_time": 1.7581892013549805,
      "step": 96949
    },
    {
      "epoch": 2.1885871887207033e-05,
      "step": 96949,
      "training_step_time": 8.754552841186523
    },
    {
      "epoch": 2.1886825561523437e-05,
      "grad_norm": 0.07602962851524353,
      "learning_rate": 2.542778380166644e-08,
      "loss": 0.0539,
      "step": 96950
    },
    {
      "epoch": 2.1886825561523437e-05,
      "model_forward_time": 1.751854419708252,
      "step": 96950
    },
    {
      "dynamics_loss_avg": 0.050925619527697565,
      "epoch": 2.1886825561523437e-05,
      "step": 96950
    },
    {
      "action_loss_avg": 0.001456144527764991,
      "epoch": 2.1886825561523437e-05,
      "step": 96950
    },
    {
      "epoch": 2.1886825561523437e-05,
      "step": 96950,
      "training_step_time": 8.444672107696533
    },
    {
      "epoch": 2.1887779235839844e-05,
      "model_forward_time": 1.7547063827514648,
      "step": 96951
    },
    {
      "epoch": 2.1887779235839844e-05,
      "step": 96951,
      "training_step_time": 9.615376949310303
    },
    {
      "epoch": 2.188873291015625e-05,
      "model_forward_time": 1.767153263092041,
      "step": 96952
    },
    {
      "epoch": 2.188873291015625e-05,
      "step": 96952,
      "training_step_time": 8.531466722488403
    },
    {
      "epoch": 2.1889686584472656e-05,
      "model_forward_time": 1.7573902606964111,
      "step": 96953
    },
    {
      "epoch": 2.1889686584472656e-05,
      "step": 96953,
      "training_step_time": 8.569846153259277
    },
    {
      "epoch": 2.1890640258789063e-05,
      "model_forward_time": 1.7602169513702393,
      "step": 96954
    },
    {
      "epoch": 2.1890640258789063e-05,
      "step": 96954,
      "training_step_time": 7.876977205276489
    },
    {
      "epoch": 2.1891593933105467e-05,
      "model_forward_time": 1.7713520526885986,
      "step": 96955
    },
    {
      "epoch": 2.1891593933105467e-05,
      "step": 96955,
      "training_step_time": 7.883641242980957
    },
    {
      "epoch": 2.1892547607421875e-05,
      "model_forward_time": 2.4178411960601807,
      "step": 96956
    },
    {
      "epoch": 2.1892547607421875e-05,
      "step": 96956,
      "training_step_time": 7.818969249725342
    },
    {
      "epoch": 2.1893501281738282e-05,
      "model_forward_time": 1.83443021774292,
      "step": 96957
    },
    {
      "epoch": 2.1893501281738282e-05,
      "step": 96957,
      "training_step_time": 7.772860050201416
    },
    {
      "epoch": 2.1894454956054686e-05,
      "model_forward_time": 1.7589731216430664,
      "step": 96958
    },
    {
      "epoch": 2.1894454956054686e-05,
      "step": 96958,
      "training_step_time": 7.688107490539551
    },
    {
      "epoch": 2.1895408630371093e-05,
      "model_forward_time": 1.7552378177642822,
      "step": 96959
    },
    {
      "epoch": 2.1895408630371093e-05,
      "step": 96959,
      "training_step_time": 8.29581618309021
    },
    {
      "epoch": 2.18963623046875e-05,
      "grad_norm": 0.07637365162372589,
      "learning_rate": 2.526151232029095e-08,
      "loss": 0.0547,
      "step": 96960
    },
    {
      "epoch": 2.18963623046875e-05,
      "model_forward_time": 1.3913588523864746,
      "step": 96960
    },
    {
      "dynamics_loss_avg": 0.06209494024515152,
      "epoch": 2.18963623046875e-05,
      "step": 96960
    },
    {
      "action_loss_avg": 0.0013297523837536573,
      "epoch": 2.18963623046875e-05,
      "step": 96960
    },
    {
      "epoch": 2.18963623046875e-05,
      "step": 96960,
      "training_step_time": 7.811486482620239
    },
    {
      "epoch": 2.1897315979003905e-05,
      "model_forward_time": 1.764192819595337,
      "step": 96961
    },
    {
      "epoch": 2.1897315979003905e-05,
      "step": 96961,
      "training_step_time": 7.931347846984863
    },
    {
      "epoch": 2.1898269653320312e-05,
      "model_forward_time": 1.770686149597168,
      "step": 96962
    },
    {
      "epoch": 2.1898269653320312e-05,
      "step": 96962,
      "training_step_time": 7.785495042800903
    },
    {
      "epoch": 2.189922332763672e-05,
      "model_forward_time": 1.7622065544128418,
      "step": 96963
    },
    {
      "epoch": 2.189922332763672e-05,
      "step": 96963,
      "training_step_time": 8.6902015209198
    },
    {
      "epoch": 2.1900177001953124e-05,
      "model_forward_time": 1.7791306972503662,
      "step": 96964
    },
    {
      "epoch": 2.1900177001953124e-05,
      "step": 96964,
      "training_step_time": 8.9308500289917
    },
    {
      "epoch": 2.190113067626953e-05,
      "model_forward_time": 2.4472756385803223,
      "step": 96965
    },
    {
      "epoch": 2.190113067626953e-05,
      "step": 96965,
      "training_step_time": 7.763597011566162
    },
    {
      "epoch": 2.190208435058594e-05,
      "model_forward_time": 1.8301122188568115,
      "step": 96966
    },
    {
      "epoch": 2.190208435058594e-05,
      "step": 96966,
      "training_step_time": 8.817480564117432
    },
    {
      "epoch": 2.1903038024902343e-05,
      "model_forward_time": 1.755584716796875,
      "step": 96967
    },
    {
      "epoch": 2.1903038024902343e-05,
      "step": 96967,
      "training_step_time": 7.834581136703491
    },
    {
      "epoch": 2.190399169921875e-05,
      "model_forward_time": 1.7579154968261719,
      "step": 96968
    },
    {
      "epoch": 2.190399169921875e-05,
      "step": 96968,
      "training_step_time": 7.722682237625122
    },
    {
      "epoch": 2.1904945373535157e-05,
      "model_forward_time": 1.3866901397705078,
      "step": 96969
    },
    {
      "epoch": 2.1904945373535157e-05,
      "step": 96969,
      "training_step_time": 7.8079540729522705
    },
    {
      "epoch": 2.190589904785156e-05,
      "grad_norm": 0.09191125631332397,
      "learning_rate": 2.509578486883657e-08,
      "loss": 0.0547,
      "step": 96970
    },
    {
      "epoch": 2.190589904785156e-05,
      "model_forward_time": 1.0550127029418945,
      "step": 96970
    },
    {
      "dynamics_loss_avg": 0.04799196198582649,
      "epoch": 2.190589904785156e-05,
      "step": 96970
    },
    {
      "action_loss_avg": 0.0011775149585446342,
      "epoch": 2.190589904785156e-05,
      "step": 96970
    },
    {
      "epoch": 2.190589904785156e-05,
      "step": 96970,
      "training_step_time": 7.7356038093566895
    },
    {
      "epoch": 2.190685272216797e-05,
      "model_forward_time": 1.7518107891082764,
      "step": 96971
    },
    {
      "epoch": 2.190685272216797e-05,
      "step": 96971,
      "training_step_time": 9.107197046279907
    },
    {
      "epoch": 2.1907806396484376e-05,
      "model_forward_time": 1.7563226222991943,
      "step": 96972
    },
    {
      "epoch": 2.1907806396484376e-05,
      "step": 96972,
      "training_step_time": 7.803989410400391
    },
    {
      "epoch": 2.190876007080078e-05,
      "model_forward_time": 1.737285852432251,
      "step": 96973
    },
    {
      "epoch": 2.190876007080078e-05,
      "step": 96973,
      "training_step_time": 7.724398374557495
    },
    {
      "epoch": 2.1909713745117188e-05,
      "model_forward_time": 1.7487640380859375,
      "step": 96974
    },
    {
      "epoch": 2.1909713745117188e-05,
      "step": 96974,
      "training_step_time": 7.928175449371338
    },
    {
      "epoch": 2.1910667419433595e-05,
      "model_forward_time": 2.4717869758605957,
      "step": 96975
    },
    {
      "epoch": 2.1910667419433595e-05,
      "step": 96975,
      "training_step_time": 7.7090842723846436
    },
    {
      "epoch": 2.191162109375e-05,
      "model_forward_time": 1.7518091201782227,
      "step": 96976
    },
    {
      "epoch": 2.191162109375e-05,
      "step": 96976,
      "training_step_time": 7.800613164901733
    },
    {
      "epoch": 2.1912574768066407e-05,
      "model_forward_time": 1.7450897693634033,
      "step": 96977
    },
    {
      "epoch": 2.1912574768066407e-05,
      "step": 96977,
      "training_step_time": 7.777415037155151
    },
    {
      "epoch": 2.1913528442382814e-05,
      "model_forward_time": 1.746192216873169,
      "step": 96978
    },
    {
      "epoch": 2.1913528442382814e-05,
      "step": 96978,
      "training_step_time": 7.807071685791016
    },
    {
      "epoch": 2.1914482116699218e-05,
      "model_forward_time": 1.7496390342712402,
      "step": 96979
    },
    {
      "epoch": 2.1914482116699218e-05,
      "step": 96979,
      "training_step_time": 8.377523422241211
    },
    {
      "epoch": 2.1915435791015625e-05,
      "grad_norm": 0.08145584911108017,
      "learning_rate": 2.493060146542825e-08,
      "loss": 0.0557,
      "step": 96980
    },
    {
      "epoch": 2.1915435791015625e-05,
      "model_forward_time": 1.7426750659942627,
      "step": 96980
    },
    {
      "dynamics_loss_avg": 0.05624454505741596,
      "epoch": 2.1915435791015625e-05,
      "step": 96980
    },
    {
      "action_loss_avg": 0.0009682758915005252,
      "epoch": 2.1915435791015625e-05,
      "step": 96980
    },
    {
      "epoch": 2.1915435791015625e-05,
      "step": 96980,
      "training_step_time": 8.776960611343384
    },
    {
      "epoch": 2.1916389465332033e-05,
      "model_forward_time": 1.3779494762420654,
      "step": 96981
    },
    {
      "epoch": 2.1916389465332033e-05,
      "step": 96981,
      "training_step_time": 7.6359686851501465
    },
    {
      "epoch": 2.1917343139648437e-05,
      "model_forward_time": 1.76328706741333,
      "step": 96982
    },
    {
      "epoch": 2.1917343139648437e-05,
      "step": 96982,
      "training_step_time": 8.638111591339111
    },
    {
      "epoch": 2.1918296813964844e-05,
      "model_forward_time": 1.7497971057891846,
      "step": 96983
    },
    {
      "epoch": 2.1918296813964844e-05,
      "step": 96983,
      "training_step_time": 7.795121669769287
    },
    {
      "epoch": 2.191925048828125e-05,
      "model_forward_time": 2.4496498107910156,
      "step": 96984
    },
    {
      "epoch": 2.191925048828125e-05,
      "step": 96984,
      "training_step_time": 8.3768892288208
    },
    {
      "epoch": 2.1920204162597656e-05,
      "model_forward_time": 1.75203275680542,
      "step": 96985
    },
    {
      "epoch": 2.1920204162597656e-05,
      "step": 96985,
      "training_step_time": 8.49305248260498
    },
    {
      "epoch": 2.1921157836914063e-05,
      "model_forward_time": 1.747868537902832,
      "step": 96986
    },
    {
      "epoch": 2.1921157836914063e-05,
      "step": 96986,
      "training_step_time": 9.35306429862976
    },
    {
      "epoch": 2.1922111511230467e-05,
      "model_forward_time": 1.7477002143859863,
      "step": 96987
    },
    {
      "epoch": 2.1922111511230467e-05,
      "step": 96987,
      "training_step_time": 8.55324411392212
    },
    {
      "epoch": 2.1923065185546875e-05,
      "model_forward_time": 1.747140645980835,
      "step": 96988
    },
    {
      "epoch": 2.1923065185546875e-05,
      "step": 96988,
      "training_step_time": 8.641732215881348
    },
    {
      "epoch": 2.1924018859863282e-05,
      "model_forward_time": 1.7650678157806396,
      "step": 96989
    },
    {
      "epoch": 2.1924018859863282e-05,
      "step": 96989,
      "training_step_time": 7.674222707748413
    },
    {
      "epoch": 2.1924972534179686e-05,
      "grad_norm": 0.07545378059148788,
      "learning_rate": 2.476596212812876e-08,
      "loss": 0.0563,
      "step": 96990
    },
    {
      "epoch": 2.1924972534179686e-05,
      "model_forward_time": 1.7455353736877441,
      "step": 96990
    },
    {
      "dynamics_loss_avg": 0.061197999864816666,
      "epoch": 2.1924972534179686e-05,
      "step": 96990
    },
    {
      "action_loss_avg": 0.0013630964240292087,
      "epoch": 2.1924972534179686e-05,
      "step": 96990
    },
    {
      "epoch": 2.1924972534179686e-05,
      "step": 96990,
      "training_step_time": 7.639123439788818
    },
    {
      "epoch": 2.1925926208496093e-05,
      "model_forward_time": 1.754509687423706,
      "step": 96991
    },
    {
      "epoch": 2.1925926208496093e-05,
      "step": 96991,
      "training_step_time": 7.6172637939453125
    },
    {
      "epoch": 2.19268798828125e-05,
      "model_forward_time": 1.7504851818084717,
      "step": 96992
    },
    {
      "epoch": 2.19268798828125e-05,
      "step": 96992,
      "training_step_time": 8.485692977905273
    },
    {
      "epoch": 2.1927833557128905e-05,
      "model_forward_time": 2.487434148788452,
      "step": 96993
    },
    {
      "epoch": 2.1927833557128905e-05,
      "step": 96993,
      "training_step_time": 7.712039232254028
    },
    {
      "epoch": 2.1928787231445312e-05,
      "model_forward_time": 1.7506742477416992,
      "step": 96994
    },
    {
      "epoch": 2.1928787231445312e-05,
      "step": 96994,
      "training_step_time": 8.179418802261353
    },
    {
      "epoch": 2.192974090576172e-05,
      "model_forward_time": 1.7433481216430664,
      "step": 96995
    },
    {
      "epoch": 2.192974090576172e-05,
      "step": 96995,
      "training_step_time": 7.641305923461914
    },
    {
      "epoch": 2.1930694580078124e-05,
      "model_forward_time": 1.7485053539276123,
      "step": 96996
    },
    {
      "epoch": 2.1930694580078124e-05,
      "step": 96996,
      "training_step_time": 9.477972030639648
    },
    {
      "epoch": 2.193164825439453e-05,
      "model_forward_time": 1.7471647262573242,
      "step": 96997
    },
    {
      "epoch": 2.193164825439453e-05,
      "step": 96997,
      "training_step_time": 8.833655595779419
    },
    {
      "epoch": 2.193260192871094e-05,
      "model_forward_time": 1.791703701019287,
      "step": 96998
    },
    {
      "epoch": 2.193260192871094e-05,
      "step": 96998,
      "training_step_time": 8.463018655776978
    },
    {
      "epoch": 2.1933555603027343e-05,
      "model_forward_time": 1.752873182296753,
      "step": 96999
    },
    {
      "epoch": 2.1933555603027343e-05,
      "step": 96999,
      "training_step_time": 8.808769702911377
    },
    {
      "epoch": 2.193450927734375e-05,
      "grad_norm": 0.10651379823684692,
      "learning_rate": 2.4601866874943703e-08,
      "loss": 0.0573,
      "step": 97000
    },
    {
      "epoch": 2.193450927734375e-05,
      "model_forward_time": 1.758173942565918,
      "step": 97000
    },
    {
      "dynamics_loss_avg": 0.05927358195185661,
      "epoch": 2.193450927734375e-05,
      "step": 97000
    },
    {
      "action_loss_avg": 0.0012645518698263913,
      "epoch": 2.193450927734375e-05,
      "step": 97000
    },
    {
      "epoch": 2.193450927734375e-05,
      "step": 97000,
      "training_step_time": 6.655458688735962
    },
    {
      "epoch": 2.1935462951660157e-05,
      "model_forward_time": 1.7827141284942627,
      "step": 97001
    },
    {
      "epoch": 2.1935462951660157e-05,
      "step": 97001,
      "training_step_time": 7.907615661621094
    },
    {
      "epoch": 2.193641662597656e-05,
      "model_forward_time": 3.440264940261841,
      "step": 97002
    },
    {
      "epoch": 2.193641662597656e-05,
      "step": 97002,
      "training_step_time": 8.343560934066772
    },
    {
      "epoch": 2.193737030029297e-05,
      "model_forward_time": 1.7523839473724365,
      "step": 97003
    },
    {
      "epoch": 2.193737030029297e-05,
      "step": 97003,
      "training_step_time": 7.821991443634033
    },
    {
      "epoch": 2.1938323974609376e-05,
      "model_forward_time": 1.7786757946014404,
      "step": 97004
    },
    {
      "epoch": 2.1938323974609376e-05,
      "step": 97004,
      "training_step_time": 8.606255054473877
    },
    {
      "epoch": 2.193927764892578e-05,
      "model_forward_time": 1.7711565494537354,
      "step": 97005
    },
    {
      "epoch": 2.193927764892578e-05,
      "step": 97005,
      "training_step_time": 10.254700183868408
    },
    {
      "epoch": 2.1940231323242188e-05,
      "model_forward_time": 1.7704925537109375,
      "step": 97006
    },
    {
      "epoch": 2.1940231323242188e-05,
      "step": 97006,
      "training_step_time": 11.23877215385437
    },
    {
      "epoch": 2.1941184997558595e-05,
      "model_forward_time": 1.7697608470916748,
      "step": 97007
    },
    {
      "epoch": 2.1941184997558595e-05,
      "step": 97007,
      "training_step_time": 8.479294538497925
    },
    {
      "epoch": 2.1942138671875e-05,
      "model_forward_time": 1.7934417724609375,
      "step": 97008
    },
    {
      "epoch": 2.1942138671875e-05,
      "step": 97008,
      "training_step_time": 10.477963924407959
    },
    {
      "epoch": 2.1943092346191407e-05,
      "model_forward_time": 1.7723579406738281,
      "step": 97009
    },
    {
      "epoch": 2.1943092346191407e-05,
      "step": 97009,
      "training_step_time": 7.7406017780303955
    },
    {
      "epoch": 2.1944046020507814e-05,
      "grad_norm": 0.08351552486419678,
      "learning_rate": 2.4438315723818163e-08,
      "loss": 0.0574,
      "step": 97010
    },
    {
      "epoch": 2.1944046020507814e-05,
      "model_forward_time": 1.7514445781707764,
      "step": 97010
    },
    {
      "dynamics_loss_avg": 0.040574580896645784,
      "epoch": 2.1944046020507814e-05,
      "step": 97010
    },
    {
      "action_loss_avg": 0.0013754014653386548,
      "epoch": 2.1944046020507814e-05,
      "step": 97010
    },
    {
      "epoch": 2.1944046020507814e-05,
      "step": 97010,
      "training_step_time": 7.916475772857666
    },
    {
      "epoch": 2.1944999694824218e-05,
      "model_forward_time": 2.857980251312256,
      "step": 97011
    },
    {
      "epoch": 2.1944999694824218e-05,
      "step": 97011,
      "training_step_time": 8.784027338027954
    },
    {
      "epoch": 2.1945953369140625e-05,
      "model_forward_time": 1.7576220035552979,
      "step": 97012
    },
    {
      "epoch": 2.1945953369140625e-05,
      "step": 97012,
      "training_step_time": 8.853617668151855
    },
    {
      "epoch": 2.1946907043457033e-05,
      "model_forward_time": 1.7730872631072998,
      "step": 97013
    },
    {
      "epoch": 2.1946907043457033e-05,
      "step": 97013,
      "training_step_time": 7.5652666091918945
    },
    {
      "epoch": 2.1947860717773437e-05,
      "model_forward_time": 1.761930227279663,
      "step": 97014
    },
    {
      "epoch": 2.1947860717773437e-05,
      "step": 97014,
      "training_step_time": 8.78098726272583
    },
    {
      "epoch": 2.1948814392089844e-05,
      "model_forward_time": 1.7705638408660889,
      "step": 97015
    },
    {
      "epoch": 2.1948814392089844e-05,
      "step": 97015,
      "training_step_time": 9.8683443069458
    },
    {
      "epoch": 2.194976806640625e-05,
      "model_forward_time": 1.7543394565582275,
      "step": 97016
    },
    {
      "epoch": 2.194976806640625e-05,
      "step": 97016,
      "training_step_time": 8.487240552902222
    },
    {
      "epoch": 2.1950721740722656e-05,
      "model_forward_time": 1.7643845081329346,
      "step": 97017
    },
    {
      "epoch": 2.1950721740722656e-05,
      "step": 97017,
      "training_step_time": 9.197674036026001
    },
    {
      "epoch": 2.1951675415039063e-05,
      "model_forward_time": 1.77508544921875,
      "step": 97018
    },
    {
      "epoch": 2.1951675415039063e-05,
      "step": 97018,
      "training_step_time": 7.764845371246338
    },
    {
      "epoch": 2.1952629089355467e-05,
      "model_forward_time": 1.7700533866882324,
      "step": 97019
    },
    {
      "epoch": 2.1952629089355467e-05,
      "step": 97019,
      "training_step_time": 8.406332969665527
    },
    {
      "epoch": 2.1953582763671875e-05,
      "grad_norm": 1.1650581359863281,
      "learning_rate": 2.4275308692637833e-08,
      "loss": 0.0576,
      "step": 97020
    },
    {
      "epoch": 2.1953582763671875e-05,
      "model_forward_time": 2.837118148803711,
      "step": 97020
    },
    {
      "dynamics_loss_avg": 0.04525819607079029,
      "epoch": 2.1953582763671875e-05,
      "step": 97020
    },
    {
      "action_loss_avg": 0.0010513775021536275,
      "epoch": 2.1953582763671875e-05,
      "step": 97020
    },
    {
      "epoch": 2.1953582763671875e-05,
      "step": 97020,
      "training_step_time": 7.799928665161133
    },
    {
      "epoch": 2.1954536437988282e-05,
      "model_forward_time": 1.7621757984161377,
      "step": 97021
    },
    {
      "epoch": 2.1954536437988282e-05,
      "step": 97021,
      "training_step_time": 7.7862067222595215
    },
    {
      "epoch": 2.1955490112304686e-05,
      "model_forward_time": 1.7897751331329346,
      "step": 97022
    },
    {
      "epoch": 2.1955490112304686e-05,
      "step": 97022,
      "training_step_time": 7.738397598266602
    },
    {
      "epoch": 2.1956443786621093e-05,
      "model_forward_time": 1.7642929553985596,
      "step": 97023
    },
    {
      "epoch": 2.1956443786621093e-05,
      "step": 97023,
      "training_step_time": 7.775304079055786
    },
    {
      "epoch": 2.19573974609375e-05,
      "model_forward_time": 1.7596931457519531,
      "step": 97024
    },
    {
      "epoch": 2.19573974609375e-05,
      "step": 97024,
      "training_step_time": 7.815805435180664
    },
    {
      "epoch": 2.1958351135253905e-05,
      "model_forward_time": 1.7662417888641357,
      "step": 97025
    },
    {
      "epoch": 2.1958351135253905e-05,
      "step": 97025,
      "training_step_time": 8.52956748008728
    },
    {
      "epoch": 2.1959304809570312e-05,
      "model_forward_time": 1.7718536853790283,
      "step": 97026
    },
    {
      "epoch": 2.1959304809570312e-05,
      "step": 97026,
      "training_step_time": 7.752537488937378
    },
    {
      "epoch": 2.196025848388672e-05,
      "model_forward_time": 1.7662770748138428,
      "step": 97027
    },
    {
      "epoch": 2.196025848388672e-05,
      "step": 97027,
      "training_step_time": 7.733166456222534
    },
    {
      "epoch": 2.1961212158203124e-05,
      "model_forward_time": 1.771705150604248,
      "step": 97028
    },
    {
      "epoch": 2.1961212158203124e-05,
      "step": 97028,
      "training_step_time": 8.233310222625732
    },
    {
      "epoch": 2.196216583251953e-05,
      "model_forward_time": 2.857628107070923,
      "step": 97029
    },
    {
      "epoch": 2.196216583251953e-05,
      "step": 97029,
      "training_step_time": 8.514311075210571
    },
    {
      "epoch": 2.196311950683594e-05,
      "grad_norm": 0.1208653673529625,
      "learning_rate": 2.411284579922901e-08,
      "loss": 0.0567,
      "step": 97030
    },
    {
      "epoch": 2.196311950683594e-05,
      "model_forward_time": 1.7902603149414062,
      "step": 97030
    },
    {
      "dynamics_loss_avg": 0.0451863806694746,
      "epoch": 2.196311950683594e-05,
      "step": 97030
    },
    {
      "action_loss_avg": 0.0008000784408068284,
      "epoch": 2.196311950683594e-05,
      "step": 97030
    },
    {
      "epoch": 2.196311950683594e-05,
      "step": 97030,
      "training_step_time": 7.762893438339233
    },
    {
      "epoch": 2.1964073181152343e-05,
      "model_forward_time": 1.8165483474731445,
      "step": 97031
    },
    {
      "epoch": 2.1964073181152343e-05,
      "step": 97031,
      "training_step_time": 8.628067255020142
    },
    {
      "epoch": 2.196502685546875e-05,
      "model_forward_time": 1.7783806324005127,
      "step": 97032
    },
    {
      "epoch": 2.196502685546875e-05,
      "step": 97032,
      "training_step_time": 7.757619380950928
    },
    {
      "epoch": 2.1965980529785157e-05,
      "model_forward_time": 1.8027527332305908,
      "step": 97033
    },
    {
      "epoch": 2.1965980529785157e-05,
      "step": 97033,
      "training_step_time": 9.915943384170532
    },
    {
      "epoch": 2.196693420410156e-05,
      "model_forward_time": 1.7716927528381348,
      "step": 97034
    },
    {
      "epoch": 2.196693420410156e-05,
      "step": 97034,
      "training_step_time": 7.745759725570679
    },
    {
      "epoch": 2.196788787841797e-05,
      "model_forward_time": 1.7545621395111084,
      "step": 97035
    },
    {
      "epoch": 2.196788787841797e-05,
      "step": 97035,
      "training_step_time": 8.3784339427948
    },
    {
      "epoch": 2.1968841552734376e-05,
      "model_forward_time": 1.731860637664795,
      "step": 97036
    },
    {
      "epoch": 2.1968841552734376e-05,
      "step": 97036,
      "training_step_time": 7.691959619522095
    },
    {
      "epoch": 2.196979522705078e-05,
      "model_forward_time": 1.7526886463165283,
      "step": 97037
    },
    {
      "epoch": 2.196979522705078e-05,
      "step": 97037,
      "training_step_time": 10.700241327285767
    },
    {
      "epoch": 2.1970748901367188e-05,
      "model_forward_time": 1.7506189346313477,
      "step": 97038
    },
    {
      "epoch": 2.1970748901367188e-05,
      "step": 97038,
      "training_step_time": 7.591996192932129
    },
    {
      "epoch": 2.1971702575683595e-05,
      "model_forward_time": 1.7658374309539795,
      "step": 97039
    },
    {
      "epoch": 2.1971702575683595e-05,
      "step": 97039,
      "training_step_time": 8.515700340270996
    },
    {
      "epoch": 2.197265625e-05,
      "grad_norm": 0.08621487766504288,
      "learning_rate": 2.395092706135804e-08,
      "loss": 0.0567,
      "step": 97040
    },
    {
      "epoch": 2.197265625e-05,
      "model_forward_time": 1.7475111484527588,
      "step": 97040
    },
    {
      "dynamics_loss_avg": 0.05277139246463776,
      "epoch": 2.197265625e-05,
      "step": 97040
    },
    {
      "action_loss_avg": 0.0011682526994263754,
      "epoch": 2.197265625e-05,
      "step": 97040
    },
    {
      "epoch": 2.197265625e-05,
      "step": 97040,
      "training_step_time": 8.451779842376709
    },
    {
      "epoch": 2.1973609924316407e-05,
      "model_forward_time": 1.7630422115325928,
      "step": 97041
    },
    {
      "epoch": 2.1973609924316407e-05,
      "step": 97041,
      "training_step_time": 7.820245981216431
    },
    {
      "epoch": 2.1974563598632814e-05,
      "model_forward_time": 1.3768260478973389,
      "step": 97042
    },
    {
      "epoch": 2.1974563598632814e-05,
      "step": 97042,
      "training_step_time": 6.807264804840088
    },
    {
      "epoch": 2.1975517272949218e-05,
      "model_forward_time": 1.7439544200897217,
      "step": 97043
    },
    {
      "epoch": 2.1975517272949218e-05,
      "step": 97043,
      "training_step_time": 6.822094678878784
    },
    {
      "epoch": 2.1976470947265625e-05,
      "model_forward_time": 1.7476410865783691,
      "step": 97044
    },
    {
      "epoch": 2.1976470947265625e-05,
      "step": 97044,
      "training_step_time": 6.778059959411621
    },
    {
      "epoch": 2.1977424621582033e-05,
      "model_forward_time": 1.749370813369751,
      "step": 97045
    },
    {
      "epoch": 2.1977424621582033e-05,
      "step": 97045,
      "training_step_time": 6.8122875690460205
    },
    {
      "epoch": 2.1978378295898437e-05,
      "model_forward_time": 1.7475008964538574,
      "step": 97046
    },
    {
      "epoch": 2.1978378295898437e-05,
      "step": 97046,
      "training_step_time": 10.01309609413147
    },
    {
      "epoch": 2.1979331970214844e-05,
      "model_forward_time": 2.399566411972046,
      "step": 97047
    },
    {
      "epoch": 2.1979331970214844e-05,
      "step": 97047,
      "training_step_time": 7.706451177597046
    },
    {
      "epoch": 2.198028564453125e-05,
      "model_forward_time": 1.7287356853485107,
      "step": 97048
    },
    {
      "epoch": 2.198028564453125e-05,
      "step": 97048,
      "training_step_time": 6.8006956577301025
    },
    {
      "epoch": 2.1981239318847656e-05,
      "model_forward_time": 1.7501561641693115,
      "step": 97049
    },
    {
      "epoch": 2.1981239318847656e-05,
      "step": 97049,
      "training_step_time": 7.478971481323242
    },
    {
      "epoch": 2.1982192993164063e-05,
      "grad_norm": 0.07634967565536499,
      "learning_rate": 2.378955249673187e-08,
      "loss": 0.0548,
      "step": 97050
    },
    {
      "epoch": 2.1982192993164063e-05,
      "model_forward_time": 1.7584760189056396,
      "step": 97050
    },
    {
      "dynamics_loss_avg": 0.060207287315279245,
      "epoch": 2.1982192993164063e-05,
      "step": 97050
    },
    {
      "action_loss_avg": 0.0007817124860594049,
      "epoch": 2.1982192993164063e-05,
      "step": 97050
    },
    {
      "epoch": 2.1982192993164063e-05,
      "step": 97050,
      "training_step_time": 7.503019094467163
    },
    {
      "epoch": 2.1983146667480467e-05,
      "model_forward_time": 1.752772569656372,
      "step": 97051
    },
    {
      "epoch": 2.1983146667480467e-05,
      "step": 97051,
      "training_step_time": 7.695868253707886
    },
    {
      "epoch": 2.1984100341796875e-05,
      "model_forward_time": 1.7461378574371338,
      "step": 97052
    },
    {
      "epoch": 2.1984100341796875e-05,
      "step": 97052,
      "training_step_time": 6.8283607959747314
    },
    {
      "epoch": 2.1985054016113282e-05,
      "model_forward_time": 1.753350019454956,
      "step": 97053
    },
    {
      "epoch": 2.1985054016113282e-05,
      "step": 97053,
      "training_step_time": 6.784097671508789
    },
    {
      "epoch": 2.1986007690429686e-05,
      "model_forward_time": 1.751373529434204,
      "step": 97054
    },
    {
      "epoch": 2.1986007690429686e-05,
      "step": 97054,
      "training_step_time": 6.72050142288208
    },
    {
      "epoch": 2.1986961364746093e-05,
      "model_forward_time": 1.7600808143615723,
      "step": 97055
    },
    {
      "epoch": 2.1986961364746093e-05,
      "step": 97055,
      "training_step_time": 6.743175029754639
    },
    {
      "epoch": 2.19879150390625e-05,
      "model_forward_time": 2.3814046382904053,
      "step": 97056
    },
    {
      "epoch": 2.19879150390625e-05,
      "step": 97056,
      "training_step_time": 7.846642971038818
    },
    {
      "epoch": 2.1988868713378905e-05,
      "model_forward_time": 1.7484359741210938,
      "step": 97057
    },
    {
      "epoch": 2.1988868713378905e-05,
      "step": 97057,
      "training_step_time": 6.693671226501465
    },
    {
      "epoch": 2.1989822387695312e-05,
      "model_forward_time": 1.746227741241455,
      "step": 97058
    },
    {
      "epoch": 2.1989822387695312e-05,
      "step": 97058,
      "training_step_time": 7.43362283706665
    },
    {
      "epoch": 2.199077606201172e-05,
      "model_forward_time": 1.7484841346740723,
      "step": 97059
    },
    {
      "epoch": 2.199077606201172e-05,
      "step": 97059,
      "training_step_time": 7.497175693511963
    },
    {
      "epoch": 2.1991729736328124e-05,
      "grad_norm": 0.127218097448349,
      "learning_rate": 2.362872212299916e-08,
      "loss": 0.0572,
      "step": 97060
    },
    {
      "epoch": 2.1991729736328124e-05,
      "model_forward_time": 1.7534472942352295,
      "step": 97060
    },
    {
      "dynamics_loss_avg": 0.062054531462490556,
      "epoch": 2.1991729736328124e-05,
      "step": 97060
    },
    {
      "action_loss_avg": 0.0008949967654189095,
      "epoch": 2.1991729736328124e-05,
      "step": 97060
    },
    {
      "epoch": 2.1991729736328124e-05,
      "step": 97060,
      "training_step_time": 7.738787412643433
    },
    {
      "epoch": 2.199268341064453e-05,
      "model_forward_time": 1.785125494003296,
      "step": 97061
    },
    {
      "epoch": 2.199268341064453e-05,
      "step": 97061,
      "training_step_time": 6.826798915863037
    },
    {
      "epoch": 2.199363708496094e-05,
      "model_forward_time": 1.7559664249420166,
      "step": 97062
    },
    {
      "epoch": 2.199363708496094e-05,
      "step": 97062,
      "training_step_time": 6.8162782192230225
    },
    {
      "epoch": 2.1994590759277343e-05,
      "model_forward_time": 1.752729892730713,
      "step": 97063
    },
    {
      "epoch": 2.1994590759277343e-05,
      "step": 97063,
      "training_step_time": 6.821866989135742
    },
    {
      "epoch": 2.199554443359375e-05,
      "model_forward_time": 1.7466819286346436,
      "step": 97064
    },
    {
      "epoch": 2.199554443359375e-05,
      "step": 97064,
      "training_step_time": 7.638916015625
    },
    {
      "epoch": 2.1996498107910157e-05,
      "model_forward_time": 2.385629415512085,
      "step": 97065
    },
    {
      "epoch": 2.1996498107910157e-05,
      "step": 97065,
      "training_step_time": 7.786425352096558
    },
    {
      "epoch": 2.199745178222656e-05,
      "model_forward_time": 1.7716188430786133,
      "step": 97066
    },
    {
      "epoch": 2.199745178222656e-05,
      "step": 97066,
      "training_step_time": 7.64913535118103
    },
    {
      "epoch": 2.199840545654297e-05,
      "model_forward_time": 1.7512800693511963,
      "step": 97067
    },
    {
      "epoch": 2.199840545654297e-05,
      "step": 97067,
      "training_step_time": 7.567818641662598
    },
    {
      "epoch": 2.1999359130859376e-05,
      "model_forward_time": 1.7569127082824707,
      "step": 97068
    },
    {
      "epoch": 2.1999359130859376e-05,
      "step": 97068,
      "training_step_time": 7.562159299850464
    },
    {
      "epoch": 2.200031280517578e-05,
      "model_forward_time": 1.7618732452392578,
      "step": 97069
    },
    {
      "epoch": 2.200031280517578e-05,
      "step": 97069,
      "training_step_time": 7.741730213165283
    },
    {
      "epoch": 2.2001266479492188e-05,
      "grad_norm": 0.09321775287389755,
      "learning_rate": 2.3468435957747504e-08,
      "loss": 0.0559,
      "step": 97070
    },
    {
      "epoch": 2.2001266479492188e-05,
      "model_forward_time": 1.7448687553405762,
      "step": 97070
    },
    {
      "dynamics_loss_avg": 0.0784276019781828,
      "epoch": 2.2001266479492188e-05,
      "step": 97070
    },
    {
      "action_loss_avg": 0.0014323568204417825,
      "epoch": 2.2001266479492188e-05,
      "step": 97070
    },
    {
      "epoch": 2.2001266479492188e-05,
      "step": 97070,
      "training_step_time": 6.877061367034912
    },
    {
      "epoch": 2.2002220153808595e-05,
      "model_forward_time": 1.7448430061340332,
      "step": 97071
    },
    {
      "epoch": 2.2002220153808595e-05,
      "step": 97071,
      "training_step_time": 6.848901748657227
    },
    {
      "epoch": 2.2003173828125e-05,
      "model_forward_time": 1.7462549209594727,
      "step": 97072
    },
    {
      "epoch": 2.2003173828125e-05,
      "step": 97072,
      "training_step_time": 6.854989290237427
    },
    {
      "epoch": 2.2004127502441407e-05,
      "model_forward_time": 1.7463257312774658,
      "step": 97073
    },
    {
      "epoch": 2.2004127502441407e-05,
      "step": 97073,
      "training_step_time": 6.852855443954468
    },
    {
      "epoch": 2.2005081176757814e-05,
      "model_forward_time": 2.408604621887207,
      "step": 97074
    },
    {
      "epoch": 2.2005081176757814e-05,
      "step": 97074,
      "training_step_time": 8.631137132644653
    },
    {
      "epoch": 2.2006034851074218e-05,
      "model_forward_time": 1.7589659690856934,
      "step": 97075
    },
    {
      "epoch": 2.2006034851074218e-05,
      "step": 97075,
      "training_step_time": 7.8447582721710205
    },
    {
      "epoch": 2.2006988525390625e-05,
      "model_forward_time": 1.755033016204834,
      "step": 97076
    },
    {
      "epoch": 2.2006988525390625e-05,
      "step": 97076,
      "training_step_time": 7.598724603652954
    },
    {
      "epoch": 2.2007942199707033e-05,
      "model_forward_time": 1.745171308517456,
      "step": 97077
    },
    {
      "epoch": 2.2007942199707033e-05,
      "step": 97077,
      "training_step_time": 7.595510721206665
    },
    {
      "epoch": 2.2008895874023437e-05,
      "model_forward_time": 1.7577736377716064,
      "step": 97078
    },
    {
      "epoch": 2.2008895874023437e-05,
      "step": 97078,
      "training_step_time": 7.766911268234253
    },
    {
      "epoch": 2.2009849548339844e-05,
      "model_forward_time": 1.738050937652588,
      "step": 97079
    },
    {
      "epoch": 2.2009849548339844e-05,
      "step": 97079,
      "training_step_time": 7.688138246536255
    },
    {
      "epoch": 2.201080322265625e-05,
      "grad_norm": 0.0752311646938324,
      "learning_rate": 2.3308694018505664e-08,
      "loss": 0.0547,
      "step": 97080
    },
    {
      "epoch": 2.201080322265625e-05,
      "model_forward_time": 1.7399065494537354,
      "step": 97080
    },
    {
      "dynamics_loss_avg": 0.04896873012185097,
      "epoch": 2.201080322265625e-05,
      "step": 97080
    },
    {
      "action_loss_avg": 0.0011288096429780126,
      "epoch": 2.201080322265625e-05,
      "step": 97080
    },
    {
      "epoch": 2.201080322265625e-05,
      "step": 97080,
      "training_step_time": 6.899729013442993
    },
    {
      "epoch": 2.2011756896972656e-05,
      "model_forward_time": 1.7502930164337158,
      "step": 97081
    },
    {
      "epoch": 2.2011756896972656e-05,
      "step": 97081,
      "training_step_time": 6.81650972366333
    },
    {
      "epoch": 2.2012710571289063e-05,
      "model_forward_time": 1.746375322341919,
      "step": 97082
    },
    {
      "epoch": 2.2012710571289063e-05,
      "step": 97082,
      "training_step_time": 6.843425989151001
    },
    {
      "epoch": 2.2013664245605467e-05,
      "model_forward_time": 2.3949167728424072,
      "step": 97083
    },
    {
      "epoch": 2.2013664245605467e-05,
      "step": 97083,
      "training_step_time": 10.402891874313354
    },
    {
      "epoch": 2.2014617919921875e-05,
      "model_forward_time": 1.7451272010803223,
      "step": 97084
    },
    {
      "epoch": 2.2014617919921875e-05,
      "step": 97084,
      "training_step_time": 7.923852920532227
    },
    {
      "epoch": 2.2015571594238282e-05,
      "model_forward_time": 1.7507545948028564,
      "step": 97085
    },
    {
      "epoch": 2.2015571594238282e-05,
      "step": 97085,
      "training_step_time": 7.5910561084747314
    },
    {
      "epoch": 2.2016525268554686e-05,
      "model_forward_time": 1.7515130043029785,
      "step": 97086
    },
    {
      "epoch": 2.2016525268554686e-05,
      "step": 97086,
      "training_step_time": 7.735760927200317
    },
    {
      "epoch": 2.2017478942871094e-05,
      "model_forward_time": 1.3607807159423828,
      "step": 97087
    },
    {
      "epoch": 2.2017478942871094e-05,
      "step": 97087,
      "training_step_time": 7.693782567977905
    },
    {
      "epoch": 2.20184326171875e-05,
      "model_forward_time": 1.752049207687378,
      "step": 97088
    },
    {
      "epoch": 2.20184326171875e-05,
      "step": 97088,
      "training_step_time": 7.6236090660095215
    },
    {
      "epoch": 2.2019386291503905e-05,
      "model_forward_time": 1.7509760856628418,
      "step": 97089
    },
    {
      "epoch": 2.2019386291503905e-05,
      "step": 97089,
      "training_step_time": 6.834261894226074
    },
    {
      "epoch": 2.2020339965820312e-05,
      "grad_norm": 0.08333900570869446,
      "learning_rate": 2.3149496322742436e-08,
      "loss": 0.0583,
      "step": 97090
    },
    {
      "epoch": 2.2020339965820312e-05,
      "model_forward_time": 1.740654706954956,
      "step": 97090
    },
    {
      "dynamics_loss_avg": 0.052651332877576354,
      "epoch": 2.2020339965820312e-05,
      "step": 97090
    },
    {
      "action_loss_avg": 0.0008896846760762855,
      "epoch": 2.2020339965820312e-05,
      "step": 97090
    },
    {
      "epoch": 2.2020339965820312e-05,
      "step": 97090,
      "training_step_time": 6.8251848220825195
    },
    {
      "epoch": 2.202129364013672e-05,
      "model_forward_time": 1.7451379299163818,
      "step": 97091
    },
    {
      "epoch": 2.202129364013672e-05,
      "step": 97091,
      "training_step_time": 6.845801591873169
    },
    {
      "epoch": 2.2022247314453124e-05,
      "model_forward_time": 1.7701587677001953,
      "step": 97092
    },
    {
      "epoch": 2.2022247314453124e-05,
      "step": 97092,
      "training_step_time": 10.702565908432007
    },
    {
      "epoch": 2.202320098876953e-05,
      "model_forward_time": 2.4121899604797363,
      "step": 97093
    },
    {
      "epoch": 2.202320098876953e-05,
      "step": 97093,
      "training_step_time": 7.994029998779297
    },
    {
      "epoch": 2.202415466308594e-05,
      "model_forward_time": 1.7424111366271973,
      "step": 97094
    },
    {
      "epoch": 2.202415466308594e-05,
      "step": 97094,
      "training_step_time": 7.478405475616455
    },
    {
      "epoch": 2.2025108337402343e-05,
      "model_forward_time": 1.7485759258270264,
      "step": 97095
    },
    {
      "epoch": 2.2025108337402343e-05,
      "step": 97095,
      "training_step_time": 7.700271368026733
    },
    {
      "epoch": 2.202606201171875e-05,
      "model_forward_time": 1.7472646236419678,
      "step": 97096
    },
    {
      "epoch": 2.202606201171875e-05,
      "step": 97096,
      "training_step_time": 7.676159381866455
    },
    {
      "epoch": 2.2027015686035157e-05,
      "model_forward_time": 1.7307093143463135,
      "step": 97097
    },
    {
      "epoch": 2.2027015686035157e-05,
      "step": 97097,
      "training_step_time": 7.783083200454712
    },
    {
      "epoch": 2.202796936035156e-05,
      "model_forward_time": 1.7329137325286865,
      "step": 97098
    },
    {
      "epoch": 2.202796936035156e-05,
      "step": 97098,
      "training_step_time": 6.959015130996704
    },
    {
      "epoch": 2.202892303466797e-05,
      "model_forward_time": 1.7434844970703125,
      "step": 97099
    },
    {
      "epoch": 2.202892303466797e-05,
      "step": 97099,
      "training_step_time": 6.93623423576355
    },
    {
      "epoch": 2.2029876708984376e-05,
      "grad_norm": 0.07471925020217896,
      "learning_rate": 2.299084288786835e-08,
      "loss": 0.0547,
      "step": 97100
    },
    {
      "epoch": 2.2029876708984376e-05,
      "model_forward_time": 1.737396240234375,
      "step": 97100
    },
    {
      "dynamics_loss_avg": 0.059970653429627416,
      "epoch": 2.2029876708984376e-05,
      "step": 97100
    },
    {
      "action_loss_avg": 0.00160144834080711,
      "epoch": 2.2029876708984376e-05,
      "step": 97100
    },
    {
      "epoch": 2.2029876708984376e-05,
      "step": 97100,
      "training_step_time": 6.964025259017944
    },
    {
      "epoch": 2.203083038330078e-05,
      "model_forward_time": 1.752197027206421,
      "step": 97101
    },
    {
      "epoch": 2.203083038330078e-05,
      "step": 97101,
      "training_step_time": 11.207668542861938
    },
    {
      "epoch": 2.2031784057617188e-05,
      "model_forward_time": 2.3766207695007324,
      "step": 97102
    },
    {
      "epoch": 2.2031784057617188e-05,
      "step": 97102,
      "training_step_time": 9.910283327102661
    },
    {
      "epoch": 2.2032737731933595e-05,
      "model_forward_time": 1.732478380203247,
      "step": 97103
    },
    {
      "epoch": 2.2032737731933595e-05,
      "step": 97103,
      "training_step_time": 8.076943397521973
    },
    {
      "epoch": 2.203369140625e-05,
      "model_forward_time": 1.743100881576538,
      "step": 97104
    },
    {
      "epoch": 2.203369140625e-05,
      "step": 97104,
      "training_step_time": 7.66719913482666
    },
    {
      "epoch": 2.2034645080566407e-05,
      "model_forward_time": 1.7387466430664062,
      "step": 97105
    },
    {
      "epoch": 2.2034645080566407e-05,
      "step": 97105,
      "training_step_time": 7.6712987422943115
    },
    {
      "epoch": 2.2035598754882814e-05,
      "model_forward_time": 1.7349364757537842,
      "step": 97106
    },
    {
      "epoch": 2.2035598754882814e-05,
      "step": 97106,
      "training_step_time": 7.804171562194824
    },
    {
      "epoch": 2.2036552429199218e-05,
      "model_forward_time": 1.7463080883026123,
      "step": 97107
    },
    {
      "epoch": 2.2036552429199218e-05,
      "step": 97107,
      "training_step_time": 7.393181085586548
    },
    {
      "epoch": 2.2037506103515626e-05,
      "model_forward_time": 1.7410545349121094,
      "step": 97108
    },
    {
      "epoch": 2.2037506103515626e-05,
      "step": 97108,
      "training_step_time": 6.922471284866333
    },
    {
      "epoch": 2.2038459777832033e-05,
      "model_forward_time": 1.7393052577972412,
      "step": 97109
    },
    {
      "epoch": 2.2038459777832033e-05,
      "step": 97109,
      "training_step_time": 6.880603790283203
    },
    {
      "epoch": 2.2039413452148437e-05,
      "grad_norm": 0.11513353884220123,
      "learning_rate": 2.283273373123229e-08,
      "loss": 0.0547,
      "step": 97110
    },
    {
      "epoch": 2.2039413452148437e-05,
      "model_forward_time": 1.7546086311340332,
      "step": 97110
    },
    {
      "dynamics_loss_avg": 0.054858894646167757,
      "epoch": 2.2039413452148437e-05,
      "step": 97110
    },
    {
      "action_loss_avg": 0.001275312341749668,
      "epoch": 2.2039413452148437e-05,
      "step": 97110
    },
    {
      "epoch": 2.2039413452148437e-05,
      "step": 97110,
      "training_step_time": 9.63380217552185
    },
    {
      "epoch": 2.2040367126464844e-05,
      "model_forward_time": 2.3794963359832764,
      "step": 97111
    },
    {
      "epoch": 2.2040367126464844e-05,
      "step": 97111,
      "training_step_time": 7.8428449630737305
    },
    {
      "epoch": 2.204132080078125e-05,
      "model_forward_time": 1.740222692489624,
      "step": 97112
    },
    {
      "epoch": 2.204132080078125e-05,
      "step": 97112,
      "training_step_time": 7.97729754447937
    },
    {
      "epoch": 2.2042274475097656e-05,
      "model_forward_time": 1.7338371276855469,
      "step": 97113
    },
    {
      "epoch": 2.2042274475097656e-05,
      "step": 97113,
      "training_step_time": 7.615665435791016
    },
    {
      "epoch": 2.2043228149414063e-05,
      "model_forward_time": 1.749281883239746,
      "step": 97114
    },
    {
      "epoch": 2.2043228149414063e-05,
      "step": 97114,
      "training_step_time": 7.382548093795776
    },
    {
      "epoch": 2.2044181823730467e-05,
      "model_forward_time": 1.7402575016021729,
      "step": 97115
    },
    {
      "epoch": 2.2044181823730467e-05,
      "step": 97115,
      "training_step_time": 7.691124677658081
    },
    {
      "epoch": 2.2045135498046875e-05,
      "model_forward_time": 1.7371246814727783,
      "step": 97116
    },
    {
      "epoch": 2.2045135498046875e-05,
      "step": 97116,
      "training_step_time": 7.65875768661499
    },
    {
      "epoch": 2.2046089172363282e-05,
      "model_forward_time": 1.7387404441833496,
      "step": 97117
    },
    {
      "epoch": 2.2046089172363282e-05,
      "step": 97117,
      "training_step_time": 6.937098979949951
    },
    {
      "epoch": 2.2047042846679686e-05,
      "model_forward_time": 1.7515373229980469,
      "step": 97118
    },
    {
      "epoch": 2.2047042846679686e-05,
      "step": 97118,
      "training_step_time": 6.908499479293823
    },
    {
      "epoch": 2.2047996520996094e-05,
      "model_forward_time": 1.7710275650024414,
      "step": 97119
    },
    {
      "epoch": 2.2047996520996094e-05,
      "step": 97119,
      "training_step_time": 8.666781187057495
    },
    {
      "epoch": 2.20489501953125e-05,
      "grad_norm": 0.1168249249458313,
      "learning_rate": 2.2675168870124885e-08,
      "loss": 0.0548,
      "step": 97120
    },
    {
      "epoch": 2.20489501953125e-05,
      "model_forward_time": 2.377779483795166,
      "step": 97120
    },
    {
      "dynamics_loss_avg": 0.0591985072940588,
      "epoch": 2.20489501953125e-05,
      "step": 97120
    },
    {
      "action_loss_avg": 0.001328614991507493,
      "epoch": 2.20489501953125e-05,
      "step": 97120
    },
    {
      "epoch": 2.20489501953125e-05,
      "step": 97120,
      "training_step_time": 9.302549839019775
    },
    {
      "epoch": 2.2049903869628905e-05,
      "model_forward_time": 1.7426667213439941,
      "step": 97121
    },
    {
      "epoch": 2.2049903869628905e-05,
      "step": 97121,
      "training_step_time": 7.86702036857605
    },
    {
      "epoch": 2.2050857543945312e-05,
      "model_forward_time": 1.7380175590515137,
      "step": 97122
    },
    {
      "epoch": 2.2050857543945312e-05,
      "step": 97122,
      "training_step_time": 7.624127388000488
    },
    {
      "epoch": 2.205181121826172e-05,
      "model_forward_time": 1.7220332622528076,
      "step": 97123
    },
    {
      "epoch": 2.205181121826172e-05,
      "step": 97123,
      "training_step_time": 7.354806900024414
    },
    {
      "epoch": 2.2052764892578124e-05,
      "model_forward_time": 1.723951816558838,
      "step": 97124
    },
    {
      "epoch": 2.2052764892578124e-05,
      "step": 97124,
      "training_step_time": 7.662762880325317
    },
    {
      "epoch": 2.205371856689453e-05,
      "model_forward_time": 1.745863676071167,
      "step": 97125
    },
    {
      "epoch": 2.205371856689453e-05,
      "step": 97125,
      "training_step_time": 7.70150089263916
    },
    {
      "epoch": 2.205467224121094e-05,
      "model_forward_time": 1.732278823852539,
      "step": 97126
    },
    {
      "epoch": 2.205467224121094e-05,
      "step": 97126,
      "training_step_time": 7.381526231765747
    },
    {
      "epoch": 2.2055625915527343e-05,
      "model_forward_time": 1.735456943511963,
      "step": 97127
    },
    {
      "epoch": 2.2055625915527343e-05,
      "step": 97127,
      "training_step_time": 6.902585506439209
    },
    {
      "epoch": 2.205657958984375e-05,
      "model_forward_time": 1.7603216171264648,
      "step": 97128
    },
    {
      "epoch": 2.205657958984375e-05,
      "step": 97128,
      "training_step_time": 7.758044242858887
    },
    {
      "epoch": 2.2057533264160158e-05,
      "model_forward_time": 2.3635451793670654,
      "step": 97129
    },
    {
      "epoch": 2.2057533264160158e-05,
      "step": 97129,
      "training_step_time": 11.325552940368652
    },
    {
      "epoch": 2.205848693847656e-05,
      "grad_norm": 0.0718274861574173,
      "learning_rate": 2.2518148321778454e-08,
      "loss": 0.0537,
      "step": 97130
    },
    {
      "epoch": 2.205848693847656e-05,
      "model_forward_time": 1.7310571670532227,
      "step": 97130
    },
    {
      "dynamics_loss_avg": 0.0568908829241991,
      "epoch": 2.205848693847656e-05,
      "step": 97130
    },
    {
      "action_loss_avg": 0.0015677106159273535,
      "epoch": 2.205848693847656e-05,
      "step": 97130
    },
    {
      "epoch": 2.205848693847656e-05,
      "step": 97130,
      "training_step_time": 7.952905893325806
    },
    {
      "epoch": 2.205944061279297e-05,
      "model_forward_time": 1.742363691329956,
      "step": 97131
    },
    {
      "epoch": 2.205944061279297e-05,
      "step": 97131,
      "training_step_time": 7.680673599243164
    },
    {
      "epoch": 2.2060394287109376e-05,
      "model_forward_time": 1.7364864349365234,
      "step": 97132
    },
    {
      "epoch": 2.2060394287109376e-05,
      "step": 97132,
      "training_step_time": 8.526141166687012
    },
    {
      "epoch": 2.206134796142578e-05,
      "model_forward_time": 1.7343418598175049,
      "step": 97133
    },
    {
      "epoch": 2.206134796142578e-05,
      "step": 97133,
      "training_step_time": 7.7520976066589355
    },
    {
      "epoch": 2.2062301635742188e-05,
      "model_forward_time": 1.771315336227417,
      "step": 97134
    },
    {
      "epoch": 2.2062301635742188e-05,
      "step": 97134,
      "training_step_time": 6.9292638301849365
    },
    {
      "epoch": 2.2063255310058595e-05,
      "model_forward_time": 1.7581653594970703,
      "step": 97135
    },
    {
      "epoch": 2.2063255310058595e-05,
      "step": 97135,
      "training_step_time": 7.748624563217163
    },
    {
      "epoch": 2.2064208984375e-05,
      "model_forward_time": 1.7507233619689941,
      "step": 97136
    },
    {
      "epoch": 2.2064208984375e-05,
      "step": 97136,
      "training_step_time": 6.934762001037598
    },
    {
      "epoch": 2.2065162658691407e-05,
      "model_forward_time": 1.7763347625732422,
      "step": 97137
    },
    {
      "epoch": 2.2065162658691407e-05,
      "step": 97137,
      "training_step_time": 7.862003564834595
    },
    {
      "epoch": 2.2066116333007814e-05,
      "model_forward_time": 1.7681195735931396,
      "step": 97138
    },
    {
      "epoch": 2.2066116333007814e-05,
      "step": 97138,
      "training_step_time": 9.874483823776245
    },
    {
      "epoch": 2.2067070007324218e-05,
      "model_forward_time": 1.7319386005401611,
      "step": 97139
    },
    {
      "epoch": 2.2067070007324218e-05,
      "step": 97139,
      "training_step_time": 7.8398754596710205
    },
    {
      "epoch": 2.2068023681640626e-05,
      "grad_norm": 0.09083961695432663,
      "learning_rate": 2.236167210336371e-08,
      "loss": 0.0559,
      "step": 97140
    },
    {
      "epoch": 2.2068023681640626e-05,
      "model_forward_time": 1.7426018714904785,
      "step": 97140
    },
    {
      "dynamics_loss_avg": 0.05525065176188946,
      "epoch": 2.2068023681640626e-05,
      "step": 97140
    },
    {
      "action_loss_avg": 0.0014288544800365344,
      "epoch": 2.2068023681640626e-05,
      "step": 97140
    },
    {
      "epoch": 2.2068023681640626e-05,
      "step": 97140,
      "training_step_time": 7.705092668533325
    },
    {
      "epoch": 2.206897735595703e-05,
      "model_forward_time": 1.7291061878204346,
      "step": 97141
    },
    {
      "epoch": 2.206897735595703e-05,
      "step": 97141,
      "training_step_time": 7.644568681716919
    },
    {
      "epoch": 2.2069931030273437e-05,
      "model_forward_time": 1.7363269329071045,
      "step": 97142
    },
    {
      "epoch": 2.2069931030273437e-05,
      "step": 97142,
      "training_step_time": 7.633541584014893
    },
    {
      "epoch": 2.2070884704589844e-05,
      "model_forward_time": 1.7388627529144287,
      "step": 97143
    },
    {
      "epoch": 2.2070884704589844e-05,
      "step": 97143,
      "training_step_time": 7.734462022781372
    },
    {
      "epoch": 2.207183837890625e-05,
      "model_forward_time": 1.7425074577331543,
      "step": 97144
    },
    {
      "epoch": 2.207183837890625e-05,
      "step": 97144,
      "training_step_time": 7.719991683959961
    },
    {
      "epoch": 2.2072792053222656e-05,
      "model_forward_time": 1.7365524768829346,
      "step": 97145
    },
    {
      "epoch": 2.2072792053222656e-05,
      "step": 97145,
      "training_step_time": 6.955249309539795
    },
    {
      "epoch": 2.2073745727539063e-05,
      "model_forward_time": 1.762319564819336,
      "step": 97146
    },
    {
      "epoch": 2.2073745727539063e-05,
      "step": 97146,
      "training_step_time": 7.843915700912476
    },
    {
      "epoch": 2.2074699401855467e-05,
      "model_forward_time": 1.7616279125213623,
      "step": 97147
    },
    {
      "epoch": 2.2074699401855467e-05,
      "step": 97147,
      "training_step_time": 10.799381494522095
    },
    {
      "epoch": 2.2075653076171875e-05,
      "model_forward_time": 2.3891210556030273,
      "step": 97148
    },
    {
      "epoch": 2.2075653076171875e-05,
      "step": 97148,
      "training_step_time": 9.600229024887085
    },
    {
      "epoch": 2.2076606750488282e-05,
      "model_forward_time": 1.7382428646087646,
      "step": 97149
    },
    {
      "epoch": 2.2076606750488282e-05,
      "step": 97149,
      "training_step_time": 7.962010145187378
    },
    {
      "epoch": 2.2077560424804686e-05,
      "grad_norm": 0.09354645758867264,
      "learning_rate": 2.220574023199251e-08,
      "loss": 0.0529,
      "step": 97150
    },
    {
      "epoch": 2.2077560424804686e-05,
      "model_forward_time": 1.7356746196746826,
      "step": 97150
    },
    {
      "dynamics_loss_avg": 0.048355723079293964,
      "epoch": 2.2077560424804686e-05,
      "step": 97150
    },
    {
      "action_loss_avg": 0.0010954808079986834,
      "epoch": 2.2077560424804686e-05,
      "step": 97150
    },
    {
      "epoch": 2.2077560424804686e-05,
      "step": 97150,
      "training_step_time": 7.680160760879517
    },
    {
      "epoch": 2.2078514099121094e-05,
      "model_forward_time": 1.7330589294433594,
      "step": 97151
    },
    {
      "epoch": 2.2078514099121094e-05,
      "step": 97151,
      "training_step_time": 7.647208213806152
    },
    {
      "epoch": 2.20794677734375e-05,
      "model_forward_time": 1.7488267421722412,
      "step": 97152
    },
    {
      "epoch": 2.20794677734375e-05,
      "step": 97152,
      "training_step_time": 7.8854405879974365
    },
    {
      "epoch": 2.2080421447753905e-05,
      "model_forward_time": 1.7382595539093018,
      "step": 97153
    },
    {
      "epoch": 2.2080421447753905e-05,
      "step": 97153,
      "training_step_time": 7.687801837921143
    },
    {
      "epoch": 2.2081375122070312e-05,
      "model_forward_time": 1.7375636100769043,
      "step": 97154
    },
    {
      "epoch": 2.2081375122070312e-05,
      "step": 97154,
      "training_step_time": 6.817836284637451
    },
    {
      "epoch": 2.208232879638672e-05,
      "model_forward_time": 1.7649273872375488,
      "step": 97155
    },
    {
      "epoch": 2.208232879638672e-05,
      "step": 97155,
      "training_step_time": 7.88825535774231
    },
    {
      "epoch": 2.2083282470703124e-05,
      "model_forward_time": 1.7403955459594727,
      "step": 97156
    },
    {
      "epoch": 2.2083282470703124e-05,
      "step": 97156,
      "training_step_time": 8.611156702041626
    },
    {
      "epoch": 2.208423614501953e-05,
      "model_forward_time": 2.4028217792510986,
      "step": 97157
    },
    {
      "epoch": 2.208423614501953e-05,
      "step": 97157,
      "training_step_time": 8.739302635192871
    },
    {
      "epoch": 2.208518981933594e-05,
      "model_forward_time": 1.740159511566162,
      "step": 97158
    },
    {
      "epoch": 2.208518981933594e-05,
      "step": 97158,
      "training_step_time": 7.837496995925903
    },
    {
      "epoch": 2.2086143493652343e-05,
      "model_forward_time": 1.738487958908081,
      "step": 97159
    },
    {
      "epoch": 2.2086143493652343e-05,
      "step": 97159,
      "training_step_time": 8.00291395187378
    },
    {
      "epoch": 2.208709716796875e-05,
      "grad_norm": 0.07592847943305969,
      "learning_rate": 2.2050352724717894e-08,
      "loss": 0.0535,
      "step": 97160
    },
    {
      "epoch": 2.208709716796875e-05,
      "model_forward_time": 1.7289302349090576,
      "step": 97160
    },
    {
      "dynamics_loss_avg": 0.07358666118234396,
      "epoch": 2.208709716796875e-05,
      "step": 97160
    },
    {
      "action_loss_avg": 0.001388093241257593,
      "epoch": 2.208709716796875e-05,
      "step": 97160
    },
    {
      "epoch": 2.208709716796875e-05,
      "step": 97160,
      "training_step_time": 7.3475189208984375
    },
    {
      "epoch": 2.2088050842285158e-05,
      "model_forward_time": 1.7364153861999512,
      "step": 97161
    },
    {
      "epoch": 2.2088050842285158e-05,
      "step": 97161,
      "training_step_time": 7.812301874160767
    },
    {
      "epoch": 2.208900451660156e-05,
      "model_forward_time": 1.7385962009429932,
      "step": 97162
    },
    {
      "epoch": 2.208900451660156e-05,
      "step": 97162,
      "training_step_time": 7.636293172836304
    },
    {
      "epoch": 2.208995819091797e-05,
      "model_forward_time": 1.7348122596740723,
      "step": 97163
    },
    {
      "epoch": 2.208995819091797e-05,
      "step": 97163,
      "training_step_time": 6.90159797668457
    },
    {
      "epoch": 2.2090911865234376e-05,
      "model_forward_time": 1.7578821182250977,
      "step": 97164
    },
    {
      "epoch": 2.2090911865234376e-05,
      "step": 97164,
      "training_step_time": 7.823298454284668
    },
    {
      "epoch": 2.209186553955078e-05,
      "model_forward_time": 1.7562708854675293,
      "step": 97165
    },
    {
      "epoch": 2.209186553955078e-05,
      "step": 97165,
      "training_step_time": 8.661379337310791
    },
    {
      "epoch": 2.2092819213867188e-05,
      "model_forward_time": 2.413865089416504,
      "step": 97166
    },
    {
      "epoch": 2.2092819213867188e-05,
      "step": 97166,
      "training_step_time": 12.350996017456055
    },
    {
      "epoch": 2.2093772888183595e-05,
      "model_forward_time": 1.7358474731445312,
      "step": 97167
    },
    {
      "epoch": 2.2093772888183595e-05,
      "step": 97167,
      "training_step_time": 7.7530694007873535
    },
    {
      "epoch": 2.20947265625e-05,
      "model_forward_time": 1.7361094951629639,
      "step": 97168
    },
    {
      "epoch": 2.20947265625e-05,
      "step": 97168,
      "training_step_time": 7.817101240158081
    },
    {
      "epoch": 2.2095680236816407e-05,
      "model_forward_time": 1.746049404144287,
      "step": 97169
    },
    {
      "epoch": 2.2095680236816407e-05,
      "step": 97169,
      "training_step_time": 7.3592822551727295
    },
    {
      "epoch": 2.2096633911132814e-05,
      "grad_norm": 0.07654105126857758,
      "learning_rate": 2.1895509598532372e-08,
      "loss": 0.0562,
      "step": 97170
    },
    {
      "epoch": 2.2096633911132814e-05,
      "model_forward_time": 1.7238001823425293,
      "step": 97170
    },
    {
      "dynamics_loss_avg": 0.05688242930918932,
      "epoch": 2.2096633911132814e-05,
      "step": 97170
    },
    {
      "action_loss_avg": 0.0012507457897299901,
      "epoch": 2.2096633911132814e-05,
      "step": 97170
    },
    {
      "epoch": 2.2096633911132814e-05,
      "step": 97170,
      "training_step_time": 7.714792013168335
    },
    {
      "epoch": 2.2097587585449218e-05,
      "model_forward_time": 1.7344574928283691,
      "step": 97171
    },
    {
      "epoch": 2.2097587585449218e-05,
      "step": 97171,
      "training_step_time": 7.155632734298706
    },
    {
      "epoch": 2.2098541259765626e-05,
      "model_forward_time": 1.7384068965911865,
      "step": 97172
    },
    {
      "epoch": 2.2098541259765626e-05,
      "step": 97172,
      "training_step_time": 7.598525762557983
    },
    {
      "epoch": 2.209949493408203e-05,
      "model_forward_time": 1.7566149234771729,
      "step": 97173
    },
    {
      "epoch": 2.209949493408203e-05,
      "step": 97173,
      "training_step_time": 7.748993873596191
    },
    {
      "epoch": 2.2100448608398437e-05,
      "model_forward_time": 1.755885362625122,
      "step": 97174
    },
    {
      "epoch": 2.2100448608398437e-05,
      "step": 97174,
      "training_step_time": 7.7032670974731445
    },
    {
      "epoch": 2.2101402282714844e-05,
      "model_forward_time": 2.4401071071624756,
      "step": 97175
    },
    {
      "epoch": 2.2101402282714844e-05,
      "step": 97175,
      "training_step_time": 10.924282312393188
    },
    {
      "epoch": 2.210235595703125e-05,
      "model_forward_time": 1.3343942165374756,
      "step": 97176
    },
    {
      "epoch": 2.210235595703125e-05,
      "step": 97176,
      "training_step_time": 7.6798272132873535
    },
    {
      "epoch": 2.2103309631347656e-05,
      "model_forward_time": 1.7320220470428467,
      "step": 97177
    },
    {
      "epoch": 2.2103309631347656e-05,
      "step": 97177,
      "training_step_time": 8.619388818740845
    },
    {
      "epoch": 2.2104263305664063e-05,
      "model_forward_time": 1.7452514171600342,
      "step": 97178
    },
    {
      "epoch": 2.2104263305664063e-05,
      "step": 97178,
      "training_step_time": 7.599445343017578
    },
    {
      "epoch": 2.2105216979980467e-05,
      "model_forward_time": 1.7354710102081299,
      "step": 97179
    },
    {
      "epoch": 2.2105216979980467e-05,
      "step": 97179,
      "training_step_time": 7.550391674041748
    },
    {
      "epoch": 2.2106170654296875e-05,
      "grad_norm": 0.09796293824911118,
      "learning_rate": 2.1741210870369066e-08,
      "loss": 0.055,
      "step": 97180
    },
    {
      "epoch": 2.2106170654296875e-05,
      "model_forward_time": 1.7332756519317627,
      "step": 97180
    },
    {
      "dynamics_loss_avg": 0.052730693481862545,
      "epoch": 2.2106170654296875e-05,
      "step": 97180
    },
    {
      "action_loss_avg": 0.0013833222357789055,
      "epoch": 2.2106170654296875e-05,
      "step": 97180
    },
    {
      "epoch": 2.2106170654296875e-05,
      "step": 97180,
      "training_step_time": 7.241668939590454
    },
    {
      "epoch": 2.2107124328613282e-05,
      "model_forward_time": 1.7377781867980957,
      "step": 97181
    },
    {
      "epoch": 2.2107124328613282e-05,
      "step": 97181,
      "training_step_time": 7.655241250991821
    },
    {
      "epoch": 2.2108078002929686e-05,
      "model_forward_time": 1.7622923851013184,
      "step": 97182
    },
    {
      "epoch": 2.2108078002929686e-05,
      "step": 97182,
      "training_step_time": 7.719649791717529
    },
    {
      "epoch": 2.2109031677246094e-05,
      "model_forward_time": 1.7520115375518799,
      "step": 97183
    },
    {
      "epoch": 2.2109031677246094e-05,
      "step": 97183,
      "training_step_time": 7.76531457901001
    },
    {
      "epoch": 2.21099853515625e-05,
      "model_forward_time": 2.356221914291382,
      "step": 97184
    },
    {
      "epoch": 2.21099853515625e-05,
      "step": 97184,
      "training_step_time": 9.635334014892578
    },
    {
      "epoch": 2.2110939025878905e-05,
      "model_forward_time": 1.7480168342590332,
      "step": 97185
    },
    {
      "epoch": 2.2110939025878905e-05,
      "step": 97185,
      "training_step_time": 7.846809387207031
    },
    {
      "epoch": 2.2111892700195312e-05,
      "model_forward_time": 1.7316341400146484,
      "step": 97186
    },
    {
      "epoch": 2.2111892700195312e-05,
      "step": 97186,
      "training_step_time": 8.259240865707397
    },
    {
      "epoch": 2.211284637451172e-05,
      "model_forward_time": 1.736541986465454,
      "step": 97187
    },
    {
      "epoch": 2.211284637451172e-05,
      "step": 97187,
      "training_step_time": 7.600524663925171
    },
    {
      "epoch": 2.2113800048828124e-05,
      "model_forward_time": 1.7304892539978027,
      "step": 97188
    },
    {
      "epoch": 2.2113800048828124e-05,
      "step": 97188,
      "training_step_time": 8.318111896514893
    },
    {
      "epoch": 2.211475372314453e-05,
      "model_forward_time": 1.7341632843017578,
      "step": 97189
    },
    {
      "epoch": 2.211475372314453e-05,
      "step": 97189,
      "training_step_time": 7.241238832473755
    },
    {
      "epoch": 2.211570739746094e-05,
      "grad_norm": 0.07653141766786575,
      "learning_rate": 2.1587456557102814e-08,
      "loss": 0.0563,
      "step": 97190
    },
    {
      "epoch": 2.211570739746094e-05,
      "model_forward_time": 1.7285854816436768,
      "step": 97190
    },
    {
      "dynamics_loss_avg": 0.06162333246320486,
      "epoch": 2.211570739746094e-05,
      "step": 97190
    },
    {
      "action_loss_avg": 0.002048634205129929,
      "epoch": 2.211570739746094e-05,
      "step": 97190
    },
    {
      "epoch": 2.211570739746094e-05,
      "step": 97190,
      "training_step_time": 7.660773992538452
    },
    {
      "epoch": 2.2116661071777343e-05,
      "model_forward_time": 1.7610549926757812,
      "step": 97191
    },
    {
      "epoch": 2.2116661071777343e-05,
      "step": 97191,
      "training_step_time": 7.690202474594116
    },
    {
      "epoch": 2.211761474609375e-05,
      "model_forward_time": 1.7617499828338623,
      "step": 97192
    },
    {
      "epoch": 2.211761474609375e-05,
      "step": 97192,
      "training_step_time": 8.351943969726562
    },
    {
      "epoch": 2.2118568420410158e-05,
      "model_forward_time": 1.7372088432312012,
      "step": 97193
    },
    {
      "epoch": 2.2118568420410158e-05,
      "step": 97193,
      "training_step_time": 9.890980243682861
    },
    {
      "epoch": 2.211952209472656e-05,
      "model_forward_time": 2.406967878341675,
      "step": 97194
    },
    {
      "epoch": 2.211952209472656e-05,
      "step": 97194,
      "training_step_time": 8.818753480911255
    },
    {
      "epoch": 2.212047576904297e-05,
      "model_forward_time": 1.7295944690704346,
      "step": 97195
    },
    {
      "epoch": 2.212047576904297e-05,
      "step": 97195,
      "training_step_time": 8.782166242599487
    },
    {
      "epoch": 2.2121429443359376e-05,
      "model_forward_time": 1.7356114387512207,
      "step": 97196
    },
    {
      "epoch": 2.2121429443359376e-05,
      "step": 97196,
      "training_step_time": 7.752276182174683
    },
    {
      "epoch": 2.212238311767578e-05,
      "model_forward_time": 1.7356948852539062,
      "step": 97197
    },
    {
      "epoch": 2.212238311767578e-05,
      "step": 97197,
      "training_step_time": 7.484099864959717
    },
    {
      "epoch": 2.2123336791992188e-05,
      "model_forward_time": 1.7343199253082275,
      "step": 97198
    },
    {
      "epoch": 2.2123336791992188e-05,
      "step": 97198,
      "training_step_time": 7.871848106384277
    },
    {
      "epoch": 2.2124290466308595e-05,
      "model_forward_time": 1.7384090423583984,
      "step": 97199
    },
    {
      "epoch": 2.2124290466308595e-05,
      "step": 97199,
      "training_step_time": 7.670645236968994
    },
    {
      "epoch": 2.2125244140625e-05,
      "grad_norm": 0.09572208672761917,
      "learning_rate": 2.1434246675546833e-08,
      "loss": 0.0566,
      "step": 97200
    },
    {
      "epoch": 2.2125244140625e-05,
      "model_forward_time": 1.7314038276672363,
      "step": 97200
    },
    {
      "dynamics_loss_avg": 0.0414451589807868,
      "epoch": 2.2125244140625e-05,
      "step": 97200
    },
    {
      "action_loss_avg": 0.0009490992815699428,
      "epoch": 2.2125244140625e-05,
      "step": 97200
    },
    {
      "epoch": 2.2125244140625e-05,
      "step": 97200,
      "training_step_time": 6.8388671875
    },
    {
      "epoch": 2.2126197814941407e-05,
      "model_forward_time": 1.816819667816162,
      "step": 97201
    },
    {
      "epoch": 2.2126197814941407e-05,
      "step": 97201,
      "training_step_time": 7.787439823150635
    },
    {
      "epoch": 2.2127151489257814e-05,
      "model_forward_time": 1.7293996810913086,
      "step": 97202
    },
    {
      "epoch": 2.2127151489257814e-05,
      "step": 97202,
      "training_step_time": 10.646852016448975
    },
    {
      "epoch": 2.2128105163574218e-05,
      "model_forward_time": 2.421147346496582,
      "step": 97203
    },
    {
      "epoch": 2.2128105163574218e-05,
      "step": 97203,
      "training_step_time": 8.920049905776978
    },
    {
      "epoch": 2.2129058837890626e-05,
      "model_forward_time": 1.740574836730957,
      "step": 97204
    },
    {
      "epoch": 2.2129058837890626e-05,
      "step": 97204,
      "training_step_time": 7.7405924797058105
    },
    {
      "epoch": 2.213001251220703e-05,
      "model_forward_time": 1.7400732040405273,
      "step": 97205
    },
    {
      "epoch": 2.213001251220703e-05,
      "step": 97205,
      "training_step_time": 7.797571420669556
    },
    {
      "epoch": 2.2130966186523437e-05,
      "model_forward_time": 1.7383558750152588,
      "step": 97206
    },
    {
      "epoch": 2.2130966186523437e-05,
      "step": 97206,
      "training_step_time": 8.162095546722412
    },
    {
      "epoch": 2.2131919860839844e-05,
      "model_forward_time": 1.745450735092163,
      "step": 97207
    },
    {
      "epoch": 2.2131919860839844e-05,
      "step": 97207,
      "training_step_time": 7.626348495483398
    },
    {
      "epoch": 2.213287353515625e-05,
      "model_forward_time": 1.7348120212554932,
      "step": 97208
    },
    {
      "epoch": 2.213287353515625e-05,
      "step": 97208,
      "training_step_time": 7.204739093780518
    },
    {
      "epoch": 2.2133827209472656e-05,
      "model_forward_time": 1.7415153980255127,
      "step": 97209
    },
    {
      "epoch": 2.2133827209472656e-05,
      "step": 97209,
      "training_step_time": 7.633286952972412
    },
    {
      "epoch": 2.2134780883789063e-05,
      "grad_norm": 0.07957294583320618,
      "learning_rate": 2.128158124245716e-08,
      "loss": 0.0552,
      "step": 97210
    },
    {
      "epoch": 2.2134780883789063e-05,
      "model_forward_time": 1.8150148391723633,
      "step": 97210
    },
    {
      "dynamics_loss_avg": 0.05487433616071939,
      "epoch": 2.2134780883789063e-05,
      "step": 97210
    },
    {
      "action_loss_avg": 0.001368412934243679,
      "epoch": 2.2134780883789063e-05,
      "step": 97210
    },
    {
      "epoch": 2.2134780883789063e-05,
      "step": 97210,
      "training_step_time": 7.780835151672363
    },
    {
      "epoch": 2.2135734558105467e-05,
      "model_forward_time": 1.7427685260772705,
      "step": 97211
    },
    {
      "epoch": 2.2135734558105467e-05,
      "step": 97211,
      "training_step_time": 10.370825052261353
    },
    {
      "epoch": 2.2136688232421875e-05,
      "model_forward_time": 1.9895045757293701,
      "step": 97212
    },
    {
      "epoch": 2.2136688232421875e-05,
      "step": 97212,
      "training_step_time": 10.956639766693115
    },
    {
      "epoch": 2.2137641906738282e-05,
      "model_forward_time": 1.7387292385101318,
      "step": 97213
    },
    {
      "epoch": 2.2137641906738282e-05,
      "step": 97213,
      "training_step_time": 7.86712646484375
    },
    {
      "epoch": 2.2138595581054686e-05,
      "model_forward_time": 1.7425711154937744,
      "step": 97214
    },
    {
      "epoch": 2.2138595581054686e-05,
      "step": 97214,
      "training_step_time": 8.657720804214478
    },
    {
      "epoch": 2.2139549255371094e-05,
      "model_forward_time": 1.7360243797302246,
      "step": 97215
    },
    {
      "epoch": 2.2139549255371094e-05,
      "step": 97215,
      "training_step_time": 7.637702226638794
    },
    {
      "epoch": 2.21405029296875e-05,
      "model_forward_time": 1.7361421585083008,
      "step": 97216
    },
    {
      "epoch": 2.21405029296875e-05,
      "step": 97216,
      "training_step_time": 7.8669469356536865
    },
    {
      "epoch": 2.2141456604003905e-05,
      "model_forward_time": 1.7444489002227783,
      "step": 97217
    },
    {
      "epoch": 2.2141456604003905e-05,
      "step": 97217,
      "training_step_time": 7.262887239456177
    },
    {
      "epoch": 2.2142410278320312e-05,
      "model_forward_time": 1.7477569580078125,
      "step": 97218
    },
    {
      "epoch": 2.2142410278320312e-05,
      "step": 97218,
      "training_step_time": 7.778917074203491
    },
    {
      "epoch": 2.214336395263672e-05,
      "model_forward_time": 1.824744701385498,
      "step": 97219
    },
    {
      "epoch": 2.214336395263672e-05,
      "step": 97219,
      "training_step_time": 8.629091262817383
    },
    {
      "epoch": 2.2144317626953124e-05,
      "grad_norm": 0.11091790348291397,
      "learning_rate": 2.1129460274527667e-08,
      "loss": 0.0568,
      "step": 97220
    },
    {
      "epoch": 2.2144317626953124e-05,
      "model_forward_time": 1.729935646057129,
      "step": 97220
    },
    {
      "dynamics_loss_avg": 0.035281614400446415,
      "epoch": 2.2144317626953124e-05,
      "step": 97220
    },
    {
      "action_loss_avg": 0.0008562841219827532,
      "epoch": 2.2144317626953124e-05,
      "step": 97220
    },
    {
      "epoch": 2.2144317626953124e-05,
      "step": 97220,
      "training_step_time": 9.193207502365112
    },
    {
      "epoch": 2.214527130126953e-05,
      "model_forward_time": 2.405693531036377,
      "step": 97221
    },
    {
      "epoch": 2.214527130126953e-05,
      "step": 97221,
      "training_step_time": 9.82161283493042
    },
    {
      "epoch": 2.214622497558594e-05,
      "model_forward_time": 1.752328872680664,
      "step": 97222
    },
    {
      "epoch": 2.214622497558594e-05,
      "step": 97222,
      "training_step_time": 8.982207298278809
    },
    {
      "epoch": 2.2147178649902343e-05,
      "model_forward_time": 1.732391595840454,
      "step": 97223
    },
    {
      "epoch": 2.2147178649902343e-05,
      "step": 97223,
      "training_step_time": 7.81488037109375
    },
    {
      "epoch": 2.214813232421875e-05,
      "model_forward_time": 1.733262300491333,
      "step": 97224
    },
    {
      "epoch": 2.214813232421875e-05,
      "step": 97224,
      "training_step_time": 7.532391786575317
    },
    {
      "epoch": 2.2149085998535158e-05,
      "model_forward_time": 1.74245023727417,
      "step": 97225
    },
    {
      "epoch": 2.2149085998535158e-05,
      "step": 97225,
      "training_step_time": 7.836634397506714
    },
    {
      "epoch": 2.215003967285156e-05,
      "model_forward_time": 1.733185052871704,
      "step": 97226
    },
    {
      "epoch": 2.215003967285156e-05,
      "step": 97226,
      "training_step_time": 7.27729058265686
    },
    {
      "epoch": 2.215099334716797e-05,
      "model_forward_time": 1.7342417240142822,
      "step": 97227
    },
    {
      "epoch": 2.215099334716797e-05,
      "step": 97227,
      "training_step_time": 7.74915075302124
    },
    {
      "epoch": 2.2151947021484376e-05,
      "model_forward_time": 1.8108141422271729,
      "step": 97228
    },
    {
      "epoch": 2.2151947021484376e-05,
      "step": 97228,
      "training_step_time": 7.784119129180908
    },
    {
      "epoch": 2.215290069580078e-05,
      "model_forward_time": 1.7381281852722168,
      "step": 97229
    },
    {
      "epoch": 2.215290069580078e-05,
      "step": 97229,
      "training_step_time": 8.54798173904419
    },
    {
      "epoch": 2.2153854370117188e-05,
      "grad_norm": 0.5283867716789246,
      "learning_rate": 2.0977883788395047e-08,
      "loss": 0.057,
      "step": 97230
    },
    {
      "epoch": 2.2153854370117188e-05,
      "model_forward_time": 1.7455296516418457,
      "step": 97230
    },
    {
      "dynamics_loss_avg": 0.04940566727891564,
      "epoch": 2.2153854370117188e-05,
      "step": 97230
    },
    {
      "action_loss_avg": 0.0011014077434083446,
      "epoch": 2.2153854370117188e-05,
      "step": 97230
    },
    {
      "epoch": 2.2153854370117188e-05,
      "step": 97230,
      "training_step_time": 11.039713859558105
    },
    {
      "epoch": 2.2154808044433595e-05,
      "model_forward_time": 1.7412798404693604,
      "step": 97231
    },
    {
      "epoch": 2.2154808044433595e-05,
      "step": 97231,
      "training_step_time": 7.814797878265381
    },
    {
      "epoch": 2.215576171875e-05,
      "model_forward_time": 1.7452213764190674,
      "step": 97232
    },
    {
      "epoch": 2.215576171875e-05,
      "step": 97232,
      "training_step_time": 7.745780944824219
    },
    {
      "epoch": 2.2156715393066407e-05,
      "model_forward_time": 1.7622230052947998,
      "step": 97233
    },
    {
      "epoch": 2.2156715393066407e-05,
      "step": 97233,
      "training_step_time": 7.519654750823975
    },
    {
      "epoch": 2.2157669067382814e-05,
      "model_forward_time": 1.7495312690734863,
      "step": 97234
    },
    {
      "epoch": 2.2157669067382814e-05,
      "step": 97234,
      "training_step_time": 7.849579811096191
    },
    {
      "epoch": 2.2158622741699218e-05,
      "model_forward_time": 1.7383208274841309,
      "step": 97235
    },
    {
      "epoch": 2.2158622741699218e-05,
      "step": 97235,
      "training_step_time": 7.14330530166626
    },
    {
      "epoch": 2.2159576416015626e-05,
      "model_forward_time": 1.7355434894561768,
      "step": 97236
    },
    {
      "epoch": 2.2159576416015626e-05,
      "step": 97236,
      "training_step_time": 7.630920171737671
    },
    {
      "epoch": 2.216053009033203e-05,
      "model_forward_time": 1.823887825012207,
      "step": 97237
    },
    {
      "epoch": 2.216053009033203e-05,
      "step": 97237,
      "training_step_time": 7.7517924308776855
    },
    {
      "epoch": 2.2161483764648437e-05,
      "model_forward_time": 1.7198388576507568,
      "step": 97238
    },
    {
      "epoch": 2.2161483764648437e-05,
      "step": 97238,
      "training_step_time": 7.7427659034729
    },
    {
      "epoch": 2.2162437438964844e-05,
      "model_forward_time": 1.7322845458984375,
      "step": 97239
    },
    {
      "epoch": 2.2162437438964844e-05,
      "step": 97239,
      "training_step_time": 9.320642471313477
    },
    {
      "epoch": 2.216339111328125e-05,
      "grad_norm": 0.1479235142469406,
      "learning_rate": 2.0826851800634928e-08,
      "loss": 0.055,
      "step": 97240
    },
    {
      "epoch": 2.216339111328125e-05,
      "model_forward_time": 2.400710105895996,
      "step": 97240
    },
    {
      "dynamics_loss_avg": 0.040441636741161344,
      "epoch": 2.216339111328125e-05,
      "step": 97240
    },
    {
      "action_loss_avg": 0.0010202865494647994,
      "epoch": 2.216339111328125e-05,
      "step": 97240
    },
    {
      "epoch": 2.216339111328125e-05,
      "step": 97240,
      "training_step_time": 8.436773538589478
    },
    {
      "epoch": 2.2164344787597656e-05,
      "model_forward_time": 1.7266261577606201,
      "step": 97241
    },
    {
      "epoch": 2.2164344787597656e-05,
      "step": 97241,
      "training_step_time": 8.844202756881714
    },
    {
      "epoch": 2.2165298461914063e-05,
      "model_forward_time": 1.744452714920044,
      "step": 97242
    },
    {
      "epoch": 2.2165298461914063e-05,
      "step": 97242,
      "training_step_time": 7.804190158843994
    },
    {
      "epoch": 2.2166252136230467e-05,
      "model_forward_time": 1.7302653789520264,
      "step": 97243
    },
    {
      "epoch": 2.2166252136230467e-05,
      "step": 97243,
      "training_step_time": 7.473054647445679
    },
    {
      "epoch": 2.2167205810546875e-05,
      "model_forward_time": 1.7338290214538574,
      "step": 97244
    },
    {
      "epoch": 2.2167205810546875e-05,
      "step": 97244,
      "training_step_time": 7.769543409347534
    },
    {
      "epoch": 2.2168159484863282e-05,
      "model_forward_time": 1.7320759296417236,
      "step": 97245
    },
    {
      "epoch": 2.2168159484863282e-05,
      "step": 97245,
      "training_step_time": 6.924997091293335
    },
    {
      "epoch": 2.2169113159179686e-05,
      "model_forward_time": 1.8261888027191162,
      "step": 97246
    },
    {
      "epoch": 2.2169113159179686e-05,
      "step": 97246,
      "training_step_time": 7.803098201751709
    },
    {
      "epoch": 2.2170066833496094e-05,
      "model_forward_time": 1.733478307723999,
      "step": 97247
    },
    {
      "epoch": 2.2170066833496094e-05,
      "step": 97247,
      "training_step_time": 7.803056001663208
    },
    {
      "epoch": 2.21710205078125e-05,
      "model_forward_time": 1.7290608882904053,
      "step": 97248
    },
    {
      "epoch": 2.21710205078125e-05,
      "step": 97248,
      "training_step_time": 9.48563814163208
    },
    {
      "epoch": 2.2171974182128905e-05,
      "model_forward_time": 2.3872530460357666,
      "step": 97249
    },
    {
      "epoch": 2.2171974182128905e-05,
      "step": 97249,
      "training_step_time": 8.771095752716064
    },
    {
      "epoch": 2.2172927856445312e-05,
      "grad_norm": 0.12476086616516113,
      "learning_rate": 2.0676364327764653e-08,
      "loss": 0.0567,
      "step": 97250
    },
    {
      "epoch": 2.2172927856445312e-05,
      "model_forward_time": 1.7178306579589844,
      "step": 97250
    },
    {
      "dynamics_loss_avg": 0.05429788827896118,
      "epoch": 2.2172927856445312e-05,
      "step": 97250
    },
    {
      "action_loss_avg": 0.0010595822939649224,
      "epoch": 2.2172927856445312e-05,
      "step": 97250
    },
    {
      "epoch": 2.2172927856445312e-05,
      "step": 97250,
      "training_step_time": 7.6704301834106445
    },
    {
      "epoch": 2.217388153076172e-05,
      "model_forward_time": 1.734740972518921,
      "step": 97251
    },
    {
      "epoch": 2.217388153076172e-05,
      "step": 97251,
      "training_step_time": 7.609762191772461
    },
    {
      "epoch": 2.2174835205078124e-05,
      "model_forward_time": 1.7413253784179688,
      "step": 97252
    },
    {
      "epoch": 2.2174835205078124e-05,
      "step": 97252,
      "training_step_time": 7.3341333866119385
    },
    {
      "epoch": 2.217578887939453e-05,
      "model_forward_time": 1.7290806770324707,
      "step": 97253
    },
    {
      "epoch": 2.217578887939453e-05,
      "step": 97253,
      "training_step_time": 7.6328418254852295
    },
    {
      "epoch": 2.217674255371094e-05,
      "model_forward_time": 1.7321646213531494,
      "step": 97254
    },
    {
      "epoch": 2.217674255371094e-05,
      "step": 97254,
      "training_step_time": 6.754032611846924
    },
    {
      "epoch": 2.2177696228027343e-05,
      "model_forward_time": 1.8226027488708496,
      "step": 97255
    },
    {
      "epoch": 2.2177696228027343e-05,
      "step": 97255,
      "training_step_time": 8.284180402755737
    },
    {
      "epoch": 2.217864990234375e-05,
      "model_forward_time": 1.7419242858886719,
      "step": 97256
    },
    {
      "epoch": 2.217864990234375e-05,
      "step": 97256,
      "training_step_time": 7.5949225425720215
    },
    {
      "epoch": 2.2179603576660158e-05,
      "model_forward_time": 1.7322590351104736,
      "step": 97257
    },
    {
      "epoch": 2.2179603576660158e-05,
      "step": 97257,
      "training_step_time": 10.010782480239868
    },
    {
      "epoch": 2.218055725097656e-05,
      "model_forward_time": 2.3904924392700195,
      "step": 97258
    },
    {
      "epoch": 2.218055725097656e-05,
      "step": 97258,
      "training_step_time": 8.627832651138306
    },
    {
      "epoch": 2.218151092529297e-05,
      "model_forward_time": 1.7262144088745117,
      "step": 97259
    },
    {
      "epoch": 2.218151092529297e-05,
      "step": 97259,
      "training_step_time": 8.788934469223022
    },
    {
      "epoch": 2.2182464599609376e-05,
      "grad_norm": 0.20753803849220276,
      "learning_rate": 2.0526421386240502e-08,
      "loss": 0.0544,
      "step": 97260
    },
    {
      "epoch": 2.2182464599609376e-05,
      "model_forward_time": 1.7250087261199951,
      "step": 97260
    },
    {
      "dynamics_loss_avg": 0.05573755502700806,
      "epoch": 2.2182464599609376e-05,
      "step": 97260
    },
    {
      "action_loss_avg": 0.001589606306515634,
      "epoch": 2.2182464599609376e-05,
      "step": 97260
    },
    {
      "epoch": 2.2182464599609376e-05,
      "step": 97260,
      "training_step_time": 7.663262128829956
    },
    {
      "epoch": 2.218341827392578e-05,
      "model_forward_time": 1.7329373359680176,
      "step": 97261
    },
    {
      "epoch": 2.218341827392578e-05,
      "step": 97261,
      "training_step_time": 7.383227586746216
    },
    {
      "epoch": 2.2184371948242188e-05,
      "model_forward_time": 1.7320191860198975,
      "step": 97262
    },
    {
      "epoch": 2.2184371948242188e-05,
      "step": 97262,
      "training_step_time": 8.629764795303345
    },
    {
      "epoch": 2.2185325622558595e-05,
      "model_forward_time": 1.7312641143798828,
      "step": 97263
    },
    {
      "epoch": 2.2185325622558595e-05,
      "step": 97263,
      "training_step_time": 7.037696838378906
    },
    {
      "epoch": 2.2186279296875e-05,
      "model_forward_time": 1.818756341934204,
      "step": 97264
    },
    {
      "epoch": 2.2186279296875e-05,
      "step": 97264,
      "training_step_time": 8.444860696792603
    },
    {
      "epoch": 2.2187232971191407e-05,
      "model_forward_time": 1.717799425125122,
      "step": 97265
    },
    {
      "epoch": 2.2187232971191407e-05,
      "step": 97265,
      "training_step_time": 8.493257522583008
    },
    {
      "epoch": 2.218818664550781e-05,
      "model_forward_time": 1.724928379058838,
      "step": 97266
    },
    {
      "epoch": 2.218818664550781e-05,
      "step": 97266,
      "training_step_time": 8.915613412857056
    },
    {
      "epoch": 2.2189140319824218e-05,
      "model_forward_time": 2.381826639175415,
      "step": 97267
    },
    {
      "epoch": 2.2189140319824218e-05,
      "step": 97267,
      "training_step_time": 10.624325037002563
    },
    {
      "epoch": 2.2190093994140626e-05,
      "model_forward_time": 1.728689432144165,
      "step": 97268
    },
    {
      "epoch": 2.2190093994140626e-05,
      "step": 97268,
      "training_step_time": 7.898112773895264
    },
    {
      "epoch": 2.219104766845703e-05,
      "model_forward_time": 1.7341501712799072,
      "step": 97269
    },
    {
      "epoch": 2.219104766845703e-05,
      "step": 97269,
      "training_step_time": 7.578299045562744
    },
    {
      "epoch": 2.2192001342773437e-05,
      "grad_norm": 0.09843732416629791,
      "learning_rate": 2.0377022992461025e-08,
      "loss": 0.0559,
      "step": 97270
    },
    {
      "epoch": 2.2192001342773437e-05,
      "model_forward_time": 1.7241356372833252,
      "step": 97270
    },
    {
      "dynamics_loss_avg": 0.05884065516293049,
      "epoch": 2.2192001342773437e-05,
      "step": 97270
    },
    {
      "action_loss_avg": 0.0015062631573528051,
      "epoch": 2.2192001342773437e-05,
      "step": 97270
    },
    {
      "epoch": 2.2192001342773437e-05,
      "step": 97270,
      "training_step_time": 7.560448884963989
    },
    {
      "epoch": 2.2192955017089845e-05,
      "model_forward_time": 1.7312636375427246,
      "step": 97271
    },
    {
      "epoch": 2.2192955017089845e-05,
      "step": 97271,
      "training_step_time": 7.684265851974487
    },
    {
      "epoch": 2.219390869140625e-05,
      "model_forward_time": 1.7287352085113525,
      "step": 97272
    },
    {
      "epoch": 2.219390869140625e-05,
      "step": 97272,
      "training_step_time": 7.234123945236206
    },
    {
      "epoch": 2.2194862365722656e-05,
      "model_forward_time": 1.8087222576141357,
      "step": 97273
    },
    {
      "epoch": 2.2194862365722656e-05,
      "step": 97273,
      "training_step_time": 8.466492176055908
    },
    {
      "epoch": 2.2195816040039063e-05,
      "model_forward_time": 1.7289140224456787,
      "step": 97274
    },
    {
      "epoch": 2.2195816040039063e-05,
      "step": 97274,
      "training_step_time": 7.64874005317688
    },
    {
      "epoch": 2.2196769714355467e-05,
      "model_forward_time": 1.7284140586853027,
      "step": 97275
    },
    {
      "epoch": 2.2196769714355467e-05,
      "step": 97275,
      "training_step_time": 7.605836629867554
    },
    {
      "epoch": 2.2197723388671875e-05,
      "model_forward_time": 2.370673179626465,
      "step": 97276
    },
    {
      "epoch": 2.2197723388671875e-05,
      "step": 97276,
      "training_step_time": 9.653745412826538
    },
    {
      "epoch": 2.2198677062988282e-05,
      "model_forward_time": 1.726560115814209,
      "step": 97277
    },
    {
      "epoch": 2.2198677062988282e-05,
      "step": 97277,
      "training_step_time": 8.608956813812256
    },
    {
      "epoch": 2.2199630737304686e-05,
      "model_forward_time": 1.7284340858459473,
      "step": 97278
    },
    {
      "epoch": 2.2199630737304686e-05,
      "step": 97278,
      "training_step_time": 8.636289596557617
    },
    {
      "epoch": 2.2200584411621094e-05,
      "model_forward_time": 1.7275722026824951,
      "step": 97279
    },
    {
      "epoch": 2.2200584411621094e-05,
      "step": 97279,
      "training_step_time": 7.532572984695435
    },
    {
      "epoch": 2.22015380859375e-05,
      "grad_norm": 0.0716552883386612,
      "learning_rate": 2.0228169162763156e-08,
      "loss": 0.0576,
      "step": 97280
    },
    {
      "epoch": 2.22015380859375e-05,
      "model_forward_time": 1.7274606227874756,
      "step": 97280
    },
    {
      "dynamics_loss_avg": 0.059898373298347,
      "epoch": 2.22015380859375e-05,
      "step": 97280
    },
    {
      "action_loss_avg": 0.0017527284857351334,
      "epoch": 2.22015380859375e-05,
      "step": 97280
    },
    {
      "epoch": 2.22015380859375e-05,
      "step": 97280,
      "training_step_time": 8.27568793296814
    },
    {
      "epoch": 2.2202491760253905e-05,
      "model_forward_time": 1.7315444946289062,
      "step": 97281
    },
    {
      "epoch": 2.2202491760253905e-05,
      "step": 97281,
      "training_step_time": 7.368430852890015
    },
    {
      "epoch": 2.2203445434570313e-05,
      "model_forward_time": 1.8180451393127441,
      "step": 97282
    },
    {
      "epoch": 2.2203445434570313e-05,
      "step": 97282,
      "training_step_time": 7.690246105194092
    },
    {
      "epoch": 2.220439910888672e-05,
      "model_forward_time": 1.735107660293579,
      "step": 97283
    },
    {
      "epoch": 2.220439910888672e-05,
      "step": 97283,
      "training_step_time": 7.768859386444092
    },
    {
      "epoch": 2.2205352783203124e-05,
      "model_forward_time": 1.7355589866638184,
      "step": 97284
    },
    {
      "epoch": 2.2205352783203124e-05,
      "step": 97284,
      "training_step_time": 7.800185203552246
    },
    {
      "epoch": 2.220630645751953e-05,
      "model_forward_time": 2.3878157138824463,
      "step": 97285
    },
    {
      "epoch": 2.220630645751953e-05,
      "step": 97285,
      "training_step_time": 8.979700565338135
    },
    {
      "epoch": 2.220726013183594e-05,
      "model_forward_time": 1.7322914600372314,
      "step": 97286
    },
    {
      "epoch": 2.220726013183594e-05,
      "step": 97286,
      "training_step_time": 8.938652276992798
    },
    {
      "epoch": 2.2208213806152343e-05,
      "model_forward_time": 1.7325291633605957,
      "step": 97287
    },
    {
      "epoch": 2.2208213806152343e-05,
      "step": 97287,
      "training_step_time": 8.712883949279785
    },
    {
      "epoch": 2.220916748046875e-05,
      "model_forward_time": 1.729975700378418,
      "step": 97288
    },
    {
      "epoch": 2.220916748046875e-05,
      "step": 97288,
      "training_step_time": 7.614664077758789
    },
    {
      "epoch": 2.2210121154785158e-05,
      "model_forward_time": 1.728238821029663,
      "step": 97289
    },
    {
      "epoch": 2.2210121154785158e-05,
      "step": 97289,
      "training_step_time": 7.76781702041626
    },
    {
      "epoch": 2.221107482910156e-05,
      "grad_norm": 0.3540321886539459,
      "learning_rate": 2.0079859913425537e-08,
      "loss": 0.0536,
      "step": 97290
    },
    {
      "epoch": 2.221107482910156e-05,
      "model_forward_time": 1.7308990955352783,
      "step": 97290
    },
    {
      "dynamics_loss_avg": 0.06246085483580828,
      "epoch": 2.221107482910156e-05,
      "step": 97290
    },
    {
      "action_loss_avg": 0.0010940633539576084,
      "epoch": 2.221107482910156e-05,
      "step": 97290
    },
    {
      "epoch": 2.221107482910156e-05,
      "step": 97290,
      "training_step_time": 7.712651252746582
    },
    {
      "epoch": 2.221202850341797e-05,
      "model_forward_time": 1.8230838775634766,
      "step": 97291
    },
    {
      "epoch": 2.221202850341797e-05,
      "step": 97291,
      "training_step_time": 7.641070127487183
    },
    {
      "epoch": 2.2212982177734377e-05,
      "model_forward_time": 1.7311618328094482,
      "step": 97292
    },
    {
      "epoch": 2.2212982177734377e-05,
      "step": 97292,
      "training_step_time": 7.671295881271362
    },
    {
      "epoch": 2.221393585205078e-05,
      "model_forward_time": 1.7305560111999512,
      "step": 97293
    },
    {
      "epoch": 2.221393585205078e-05,
      "step": 97293,
      "training_step_time": 7.64337158203125
    },
    {
      "epoch": 2.2214889526367188e-05,
      "model_forward_time": 2.3774497509002686,
      "step": 97294
    },
    {
      "epoch": 2.2214889526367188e-05,
      "step": 97294,
      "training_step_time": 10.33699631690979
    },
    {
      "epoch": 2.2215843200683595e-05,
      "model_forward_time": 1.7318086624145508,
      "step": 97295
    },
    {
      "epoch": 2.2215843200683595e-05,
      "step": 97295,
      "training_step_time": 8.57358169555664
    },
    {
      "epoch": 2.2216796875e-05,
      "model_forward_time": 1.7314863204956055,
      "step": 97296
    },
    {
      "epoch": 2.2216796875e-05,
      "step": 97296,
      "training_step_time": 8.837106227874756
    },
    {
      "epoch": 2.2217750549316407e-05,
      "model_forward_time": 1.7276251316070557,
      "step": 97297
    },
    {
      "epoch": 2.2217750549316407e-05,
      "step": 97297,
      "training_step_time": 7.6999969482421875
    },
    {
      "epoch": 2.221870422363281e-05,
      "model_forward_time": 1.7322378158569336,
      "step": 97298
    },
    {
      "epoch": 2.221870422363281e-05,
      "step": 97298,
      "training_step_time": 7.705747604370117
    },
    {
      "epoch": 2.2219657897949218e-05,
      "model_forward_time": 1.7317514419555664,
      "step": 97299
    },
    {
      "epoch": 2.2219657897949218e-05,
      "step": 97299,
      "training_step_time": 7.98957085609436
    },
    {
      "epoch": 2.2220611572265626e-05,
      "grad_norm": 0.07868733257055283,
      "learning_rate": 1.9932095260667418e-08,
      "loss": 0.0547,
      "step": 97300
    },
    {
      "epoch": 2.2220611572265626e-05,
      "model_forward_time": 1.82816481590271,
      "step": 97300
    },
    {
      "dynamics_loss_avg": 0.05182715319097042,
      "epoch": 2.2220611572265626e-05,
      "step": 97300
    },
    {
      "action_loss_avg": 0.0013144900964107364,
      "epoch": 2.2220611572265626e-05,
      "step": 97300
    },
    {
      "epoch": 2.2220611572265626e-05,
      "step": 97300,
      "training_step_time": 7.834110260009766
    },
    {
      "epoch": 2.222156524658203e-05,
      "model_forward_time": 1.7314095497131348,
      "step": 97301
    },
    {
      "epoch": 2.222156524658203e-05,
      "step": 97301,
      "training_step_time": 7.786953926086426
    },
    {
      "epoch": 2.2222518920898437e-05,
      "model_forward_time": 1.7278106212615967,
      "step": 97302
    },
    {
      "epoch": 2.2222518920898437e-05,
      "step": 97302,
      "training_step_time": 7.782410621643066
    },
    {
      "epoch": 2.2223472595214845e-05,
      "model_forward_time": 2.3786213397979736,
      "step": 97303
    },
    {
      "epoch": 2.2223472595214845e-05,
      "step": 97303,
      "training_step_time": 8.86899447441101
    },
    {
      "epoch": 2.222442626953125e-05,
      "model_forward_time": 1.7328879833221436,
      "step": 97304
    },
    {
      "epoch": 2.222442626953125e-05,
      "step": 97304,
      "training_step_time": 8.655039072036743
    },
    {
      "epoch": 2.2225379943847656e-05,
      "model_forward_time": 1.7300224304199219,
      "step": 97305
    },
    {
      "epoch": 2.2225379943847656e-05,
      "step": 97305,
      "training_step_time": 8.106364727020264
    },
    {
      "epoch": 2.2226333618164063e-05,
      "model_forward_time": 1.729010820388794,
      "step": 97306
    },
    {
      "epoch": 2.2226333618164063e-05,
      "step": 97306,
      "training_step_time": 7.61322546005249
    },
    {
      "epoch": 2.2227287292480467e-05,
      "model_forward_time": 1.7328729629516602,
      "step": 97307
    },
    {
      "epoch": 2.2227287292480467e-05,
      "step": 97307,
      "training_step_time": 7.702189207077026
    },
    {
      "epoch": 2.2228240966796875e-05,
      "model_forward_time": 1.7332918643951416,
      "step": 97308
    },
    {
      "epoch": 2.2228240966796875e-05,
      "step": 97308,
      "training_step_time": 7.328503608703613
    },
    {
      "epoch": 2.2229194641113282e-05,
      "model_forward_time": 1.815422534942627,
      "step": 97309
    },
    {
      "epoch": 2.2229194641113282e-05,
      "step": 97309,
      "training_step_time": 7.806133985519409
    },
    {
      "epoch": 2.2230148315429686e-05,
      "grad_norm": 0.07938209176063538,
      "learning_rate": 1.978487522064809e-08,
      "loss": 0.0545,
      "step": 97310
    },
    {
      "epoch": 2.2230148315429686e-05,
      "model_forward_time": 1.727975845336914,
      "step": 97310
    },
    {
      "dynamics_loss_avg": 0.061181848496198656,
      "epoch": 2.2230148315429686e-05,
      "step": 97310
    },
    {
      "action_loss_avg": 0.0014254179084673523,
      "epoch": 2.2230148315429686e-05,
      "step": 97310
    },
    {
      "epoch": 2.2230148315429686e-05,
      "step": 97310,
      "training_step_time": 7.804140567779541
    },
    {
      "epoch": 2.2231101989746094e-05,
      "model_forward_time": 1.7333793640136719,
      "step": 97311
    },
    {
      "epoch": 2.2231101989746094e-05,
      "step": 97311,
      "training_step_time": 8.731916904449463
    },
    {
      "epoch": 2.22320556640625e-05,
      "model_forward_time": 2.3649518489837646,
      "step": 97312
    },
    {
      "epoch": 2.22320556640625e-05,
      "step": 97312,
      "training_step_time": 10.254834175109863
    },
    {
      "epoch": 2.2233009338378905e-05,
      "model_forward_time": 1.7310807704925537,
      "step": 97313
    },
    {
      "epoch": 2.2233009338378905e-05,
      "step": 97313,
      "training_step_time": 8.739935159683228
    },
    {
      "epoch": 2.2233963012695313e-05,
      "model_forward_time": 1.7277913093566895,
      "step": 97314
    },
    {
      "epoch": 2.2233963012695313e-05,
      "step": 97314,
      "training_step_time": 8.757867336273193
    },
    {
      "epoch": 2.223491668701172e-05,
      "model_forward_time": 1.7266035079956055,
      "step": 97315
    },
    {
      "epoch": 2.223491668701172e-05,
      "step": 97315,
      "training_step_time": 8.852085590362549
    },
    {
      "epoch": 2.2235870361328124e-05,
      "model_forward_time": 1.7301316261291504,
      "step": 97316
    },
    {
      "epoch": 2.2235870361328124e-05,
      "step": 97316,
      "training_step_time": 8.704972505569458
    },
    {
      "epoch": 2.223682403564453e-05,
      "model_forward_time": 1.7313907146453857,
      "step": 97317
    },
    {
      "epoch": 2.223682403564453e-05,
      "step": 97317,
      "training_step_time": 7.690584659576416
    },
    {
      "epoch": 2.223777770996094e-05,
      "model_forward_time": 1.8240501880645752,
      "step": 97318
    },
    {
      "epoch": 2.223777770996094e-05,
      "step": 97318,
      "training_step_time": 7.711507797241211
    },
    {
      "epoch": 2.2238731384277343e-05,
      "model_forward_time": 1.7323598861694336,
      "step": 97319
    },
    {
      "epoch": 2.2238731384277343e-05,
      "step": 97319,
      "training_step_time": 8.543695211410522
    },
    {
      "epoch": 2.223968505859375e-05,
      "grad_norm": 0.08372742682695389,
      "learning_rate": 1.9638199809466908e-08,
      "loss": 0.0542,
      "step": 97320
    },
    {
      "epoch": 2.223968505859375e-05,
      "model_forward_time": 1.723527193069458,
      "step": 97320
    },
    {
      "dynamics_loss_avg": 0.06343684196472169,
      "epoch": 2.223968505859375e-05,
      "step": 97320
    },
    {
      "action_loss_avg": 0.0012761459918692708,
      "epoch": 2.223968505859375e-05,
      "step": 97320
    },
    {
      "epoch": 2.223968505859375e-05,
      "step": 97320,
      "training_step_time": 7.719356298446655
    },
    {
      "epoch": 2.2240638732910158e-05,
      "model_forward_time": 2.376579999923706,
      "step": 97321
    },
    {
      "epoch": 2.2240638732910158e-05,
      "step": 97321,
      "training_step_time": 7.684484958648682
    },
    {
      "epoch": 2.2241592407226562e-05,
      "model_forward_time": 1.7277534008026123,
      "step": 97322
    },
    {
      "epoch": 2.2241592407226562e-05,
      "step": 97322,
      "training_step_time": 8.839685440063477
    },
    {
      "epoch": 2.224254608154297e-05,
      "model_forward_time": 1.7285618782043457,
      "step": 97323
    },
    {
      "epoch": 2.224254608154297e-05,
      "step": 97323,
      "training_step_time": 7.9905784130096436
    },
    {
      "epoch": 2.2243499755859377e-05,
      "model_forward_time": 1.7292921543121338,
      "step": 97324
    },
    {
      "epoch": 2.2243499755859377e-05,
      "step": 97324,
      "training_step_time": 7.66547966003418
    },
    {
      "epoch": 2.224445343017578e-05,
      "model_forward_time": 1.7300631999969482,
      "step": 97325
    },
    {
      "epoch": 2.224445343017578e-05,
      "step": 97325,
      "training_step_time": 8.110281944274902
    },
    {
      "epoch": 2.2245407104492188e-05,
      "model_forward_time": 1.7302567958831787,
      "step": 97326
    },
    {
      "epoch": 2.2245407104492188e-05,
      "step": 97326,
      "training_step_time": 7.1143248081207275
    },
    {
      "epoch": 2.2246360778808595e-05,
      "model_forward_time": 1.8203794956207275,
      "step": 97327
    },
    {
      "epoch": 2.2246360778808595e-05,
      "step": 97327,
      "training_step_time": 7.824860095977783
    },
    {
      "epoch": 2.2247314453125e-05,
      "model_forward_time": 1.7305564880371094,
      "step": 97328
    },
    {
      "epoch": 2.2247314453125e-05,
      "step": 97328,
      "training_step_time": 7.6158576011657715
    },
    {
      "epoch": 2.2248268127441407e-05,
      "model_forward_time": 1.737722635269165,
      "step": 97329
    },
    {
      "epoch": 2.2248268127441407e-05,
      "step": 97329,
      "training_step_time": 7.58578085899353
    },
    {
      "epoch": 2.224922180175781e-05,
      "grad_norm": 0.06367237120866776,
      "learning_rate": 1.9492069043164364e-08,
      "loss": 0.0541,
      "step": 97330
    },
    {
      "epoch": 2.224922180175781e-05,
      "model_forward_time": 2.376325845718384,
      "step": 97330
    },
    {
      "dynamics_loss_avg": 0.04590455181896687,
      "epoch": 2.224922180175781e-05,
      "step": 97330
    },
    {
      "action_loss_avg": 0.0011953489360166714,
      "epoch": 2.224922180175781e-05,
      "step": 97330
    },
    {
      "epoch": 2.224922180175781e-05,
      "step": 97330,
      "training_step_time": 7.560929536819458
    },
    {
      "epoch": 2.2250175476074218e-05,
      "model_forward_time": 1.7248435020446777,
      "step": 97331
    },
    {
      "epoch": 2.2250175476074218e-05,
      "step": 97331,
      "training_step_time": 9.330504179000854
    },
    {
      "epoch": 2.2251129150390626e-05,
      "model_forward_time": 1.7295420169830322,
      "step": 97332
    },
    {
      "epoch": 2.2251129150390626e-05,
      "step": 97332,
      "training_step_time": 7.643253326416016
    },
    {
      "epoch": 2.225208282470703e-05,
      "model_forward_time": 1.731996774673462,
      "step": 97333
    },
    {
      "epoch": 2.225208282470703e-05,
      "step": 97333,
      "training_step_time": 7.765289306640625
    },
    {
      "epoch": 2.2253036499023437e-05,
      "model_forward_time": 1.728367567062378,
      "step": 97334
    },
    {
      "epoch": 2.2253036499023437e-05,
      "step": 97334,
      "training_step_time": 7.54692816734314
    },
    {
      "epoch": 2.2253990173339845e-05,
      "model_forward_time": 1.728365421295166,
      "step": 97335
    },
    {
      "epoch": 2.2253990173339845e-05,
      "step": 97335,
      "training_step_time": 7.74219274520874
    },
    {
      "epoch": 2.225494384765625e-05,
      "model_forward_time": 1.8068039417266846,
      "step": 97336
    },
    {
      "epoch": 2.225494384765625e-05,
      "step": 97336,
      "training_step_time": 7.951552391052246
    },
    {
      "epoch": 2.2255897521972656e-05,
      "model_forward_time": 1.7280161380767822,
      "step": 97337
    },
    {
      "epoch": 2.2255897521972656e-05,
      "step": 97337,
      "training_step_time": 7.70503830909729
    },
    {
      "epoch": 2.2256851196289063e-05,
      "model_forward_time": 1.7333786487579346,
      "step": 97338
    },
    {
      "epoch": 2.2256851196289063e-05,
      "step": 97338,
      "training_step_time": 9.327011346817017
    },
    {
      "epoch": 2.2257804870605467e-05,
      "model_forward_time": 2.3853771686553955,
      "step": 97339
    },
    {
      "epoch": 2.2257804870605467e-05,
      "step": 97339,
      "training_step_time": 8.456193923950195
    },
    {
      "epoch": 2.2258758544921875e-05,
      "grad_norm": 0.12827062606811523,
      "learning_rate": 1.9346482937721567e-08,
      "loss": 0.0569,
      "step": 97340
    },
    {
      "epoch": 2.2258758544921875e-05,
      "model_forward_time": 1.7249269485473633,
      "step": 97340
    },
    {
      "dynamics_loss_avg": 0.08086231220513582,
      "epoch": 2.2258758544921875e-05,
      "step": 97340
    },
    {
      "action_loss_avg": 0.0013791800127364695,
      "epoch": 2.2258758544921875e-05,
      "step": 97340
    },
    {
      "epoch": 2.2258758544921875e-05,
      "step": 97340,
      "training_step_time": 8.494449615478516
    },
    {
      "epoch": 2.2259712219238282e-05,
      "model_forward_time": 1.3345978260040283,
      "step": 97341
    },
    {
      "epoch": 2.2259712219238282e-05,
      "step": 97341,
      "training_step_time": 8.595275163650513
    },
    {
      "epoch": 2.2260665893554686e-05,
      "model_forward_time": 1.7308733463287354,
      "step": 97342
    },
    {
      "epoch": 2.2260665893554686e-05,
      "step": 97342,
      "training_step_time": 7.801062822341919
    },
    {
      "epoch": 2.2261619567871094e-05,
      "model_forward_time": 1.7267279624938965,
      "step": 97343
    },
    {
      "epoch": 2.2261619567871094e-05,
      "step": 97343,
      "training_step_time": 7.665752410888672
    },
    {
      "epoch": 2.22625732421875e-05,
      "model_forward_time": 1.733999252319336,
      "step": 97344
    },
    {
      "epoch": 2.22625732421875e-05,
      "step": 97344,
      "training_step_time": 7.62053918838501
    },
    {
      "epoch": 2.2263526916503905e-05,
      "model_forward_time": 1.8002040386199951,
      "step": 97345
    },
    {
      "epoch": 2.2263526916503905e-05,
      "step": 97345,
      "training_step_time": 8.451456546783447
    },
    {
      "epoch": 2.2264480590820313e-05,
      "model_forward_time": 1.7329294681549072,
      "step": 97346
    },
    {
      "epoch": 2.2264480590820313e-05,
      "step": 97346,
      "training_step_time": 8.764127731323242
    },
    {
      "epoch": 2.226543426513672e-05,
      "model_forward_time": 1.717327356338501,
      "step": 97347
    },
    {
      "epoch": 2.226543426513672e-05,
      "step": 97347,
      "training_step_time": 8.450589418411255
    },
    {
      "epoch": 2.2266387939453124e-05,
      "model_forward_time": 2.388533592224121,
      "step": 97348
    },
    {
      "epoch": 2.2266387939453124e-05,
      "step": 97348,
      "training_step_time": 7.656725168228149
    },
    {
      "epoch": 2.226734161376953e-05,
      "model_forward_time": 1.7178432941436768,
      "step": 97349
    },
    {
      "epoch": 2.226734161376953e-05,
      "step": 97349,
      "training_step_time": 8.305629253387451
    },
    {
      "epoch": 2.226829528808594e-05,
      "grad_norm": 0.0784628614783287,
      "learning_rate": 1.9201441509058562e-08,
      "loss": 0.0549,
      "step": 97350
    },
    {
      "epoch": 2.226829528808594e-05,
      "model_forward_time": 1.7345919609069824,
      "step": 97350
    },
    {
      "dynamics_loss_avg": 0.0748124223202467,
      "epoch": 2.226829528808594e-05,
      "step": 97350
    },
    {
      "action_loss_avg": 0.002176568383583799,
      "epoch": 2.226829528808594e-05,
      "step": 97350
    },
    {
      "epoch": 2.226829528808594e-05,
      "step": 97350,
      "training_step_time": 8.577391624450684
    },
    {
      "epoch": 2.2269248962402343e-05,
      "model_forward_time": 1.734283685684204,
      "step": 97351
    },
    {
      "epoch": 2.2269248962402343e-05,
      "step": 97351,
      "training_step_time": 8.799094438552856
    },
    {
      "epoch": 2.227020263671875e-05,
      "model_forward_time": 1.7340335845947266,
      "step": 97352
    },
    {
      "epoch": 2.227020263671875e-05,
      "step": 97352,
      "training_step_time": 7.687470197677612
    },
    {
      "epoch": 2.2271156311035158e-05,
      "model_forward_time": 1.739992618560791,
      "step": 97353
    },
    {
      "epoch": 2.2271156311035158e-05,
      "step": 97353,
      "training_step_time": 8.598286867141724
    },
    {
      "epoch": 2.2272109985351562e-05,
      "model_forward_time": 1.7759180068969727,
      "step": 97354
    },
    {
      "epoch": 2.2272109985351562e-05,
      "step": 97354,
      "training_step_time": 7.957510948181152
    },
    {
      "epoch": 2.227306365966797e-05,
      "model_forward_time": 1.7366735935211182,
      "step": 97355
    },
    {
      "epoch": 2.227306365966797e-05,
      "step": 97355,
      "training_step_time": 7.682042121887207
    },
    {
      "epoch": 2.2274017333984377e-05,
      "model_forward_time": 1.7370691299438477,
      "step": 97356
    },
    {
      "epoch": 2.2274017333984377e-05,
      "step": 97356,
      "training_step_time": 7.6857686042785645
    },
    {
      "epoch": 2.227497100830078e-05,
      "model_forward_time": 2.404118061065674,
      "step": 97357
    },
    {
      "epoch": 2.227497100830078e-05,
      "step": 97357,
      "training_step_time": 7.5212695598602295
    },
    {
      "epoch": 2.2275924682617188e-05,
      "model_forward_time": 1.7341282367706299,
      "step": 97358
    },
    {
      "epoch": 2.2275924682617188e-05,
      "step": 97358,
      "training_step_time": 7.615345001220703
    },
    {
      "epoch": 2.2276878356933595e-05,
      "model_forward_time": 1.7375493049621582,
      "step": 97359
    },
    {
      "epoch": 2.2276878356933595e-05,
      "step": 97359,
      "training_step_time": 9.320749759674072
    },
    {
      "epoch": 2.227783203125e-05,
      "grad_norm": 0.07631443440914154,
      "learning_rate": 1.9056944773037656e-08,
      "loss": 0.0549,
      "step": 97360
    },
    {
      "epoch": 2.227783203125e-05,
      "model_forward_time": 1.7315242290496826,
      "step": 97360
    },
    {
      "dynamics_loss_avg": 0.05782642364501953,
      "epoch": 2.227783203125e-05,
      "step": 97360
    },
    {
      "action_loss_avg": 0.000862559219240211,
      "epoch": 2.227783203125e-05,
      "step": 97360
    },
    {
      "epoch": 2.227783203125e-05,
      "step": 97360,
      "training_step_time": 7.587265729904175
    },
    {
      "epoch": 2.2278785705566407e-05,
      "model_forward_time": 1.7380285263061523,
      "step": 97361
    },
    {
      "epoch": 2.2278785705566407e-05,
      "step": 97361,
      "training_step_time": 8.774755477905273
    },
    {
      "epoch": 2.227973937988281e-05,
      "model_forward_time": 1.7357501983642578,
      "step": 97362
    },
    {
      "epoch": 2.227973937988281e-05,
      "step": 97362,
      "training_step_time": 7.796997785568237
    },
    {
      "epoch": 2.228069305419922e-05,
      "model_forward_time": 1.7425971031188965,
      "step": 97363
    },
    {
      "epoch": 2.228069305419922e-05,
      "step": 97363,
      "training_step_time": 8.134871244430542
    },
    {
      "epoch": 2.2281646728515626e-05,
      "model_forward_time": 1.7375171184539795,
      "step": 97364
    },
    {
      "epoch": 2.2281646728515626e-05,
      "step": 97364,
      "training_step_time": 7.592897653579712
    },
    {
      "epoch": 2.228260040283203e-05,
      "model_forward_time": 1.7384390830993652,
      "step": 97365
    },
    {
      "epoch": 2.228260040283203e-05,
      "step": 97365,
      "training_step_time": 8.543706178665161
    },
    {
      "epoch": 2.2283554077148437e-05,
      "model_forward_time": 2.418848991394043,
      "step": 97366
    },
    {
      "epoch": 2.2283554077148437e-05,
      "step": 97366,
      "training_step_time": 7.508328437805176
    },
    {
      "epoch": 2.2284507751464845e-05,
      "model_forward_time": 1.7387075424194336,
      "step": 97367
    },
    {
      "epoch": 2.2284507751464845e-05,
      "step": 97367,
      "training_step_time": 9.227762460708618
    },
    {
      "epoch": 2.228546142578125e-05,
      "model_forward_time": 1.735816240310669,
      "step": 97368
    },
    {
      "epoch": 2.228546142578125e-05,
      "step": 97368,
      "training_step_time": 7.7701096534729
    },
    {
      "epoch": 2.2286415100097656e-05,
      "model_forward_time": 1.734924554824829,
      "step": 97369
    },
    {
      "epoch": 2.2286415100097656e-05,
      "step": 97369,
      "training_step_time": 7.643005132675171
    },
    {
      "epoch": 2.2287368774414063e-05,
      "grad_norm": 0.09919433295726776,
      "learning_rate": 1.89129927454601e-08,
      "loss": 0.0557,
      "step": 97370
    },
    {
      "epoch": 2.2287368774414063e-05,
      "model_forward_time": 1.7321879863739014,
      "step": 97370
    },
    {
      "dynamics_loss_avg": 0.059898192435503005,
      "epoch": 2.2287368774414063e-05,
      "step": 97370
    },
    {
      "action_loss_avg": 0.0013357800024095922,
      "epoch": 2.2287368774414063e-05,
      "step": 97370
    },
    {
      "epoch": 2.2287368774414063e-05,
      "step": 97370,
      "training_step_time": 8.376859903335571
    },
    {
      "epoch": 2.2288322448730467e-05,
      "model_forward_time": 1.7341058254241943,
      "step": 97371
    },
    {
      "epoch": 2.2288322448730467e-05,
      "step": 97371,
      "training_step_time": 8.474846124649048
    },
    {
      "epoch": 2.2289276123046875e-05,
      "model_forward_time": 1.7389929294586182,
      "step": 97372
    },
    {
      "epoch": 2.2289276123046875e-05,
      "step": 97372,
      "training_step_time": 7.236375331878662
    },
    {
      "epoch": 2.2290229797363282e-05,
      "model_forward_time": 1.8256657123565674,
      "step": 97373
    },
    {
      "epoch": 2.2290229797363282e-05,
      "step": 97373,
      "training_step_time": 7.780957460403442
    },
    {
      "epoch": 2.2291183471679686e-05,
      "model_forward_time": 1.7352840900421143,
      "step": 97374
    },
    {
      "epoch": 2.2291183471679686e-05,
      "step": 97374,
      "training_step_time": 7.683390378952026
    },
    {
      "epoch": 2.2292137145996094e-05,
      "model_forward_time": 2.379608392715454,
      "step": 97375
    },
    {
      "epoch": 2.2292137145996094e-05,
      "step": 97375,
      "training_step_time": 7.5704991817474365
    },
    {
      "epoch": 2.22930908203125e-05,
      "model_forward_time": 1.7411692142486572,
      "step": 97376
    },
    {
      "epoch": 2.22930908203125e-05,
      "step": 97376,
      "training_step_time": 8.553753137588501
    },
    {
      "epoch": 2.2294044494628905e-05,
      "model_forward_time": 1.736680507659912,
      "step": 97377
    },
    {
      "epoch": 2.2294044494628905e-05,
      "step": 97377,
      "training_step_time": 9.819504737854004
    },
    {
      "epoch": 2.2294998168945313e-05,
      "model_forward_time": 1.7312545776367188,
      "step": 97378
    },
    {
      "epoch": 2.2294998168945313e-05,
      "step": 97378,
      "training_step_time": 8.878936767578125
    },
    {
      "epoch": 2.229595184326172e-05,
      "model_forward_time": 1.7194528579711914,
      "step": 97379
    },
    {
      "epoch": 2.229595184326172e-05,
      "step": 97379,
      "training_step_time": 9.186109066009521
    },
    {
      "epoch": 2.2296905517578124e-05,
      "grad_norm": 0.08900734782218933,
      "learning_rate": 1.876958544206886e-08,
      "loss": 0.0544,
      "step": 97380
    },
    {
      "epoch": 2.2296905517578124e-05,
      "model_forward_time": 1.7287850379943848,
      "step": 97380
    },
    {
      "dynamics_loss_avg": 0.038979319576174024,
      "epoch": 2.2296905517578124e-05,
      "step": 97380
    },
    {
      "action_loss_avg": 0.0009864110936177895,
      "epoch": 2.2296905517578124e-05,
      "step": 97380
    },
    {
      "epoch": 2.2296905517578124e-05,
      "step": 97380,
      "training_step_time": 8.500854969024658
    },
    {
      "epoch": 2.229785919189453e-05,
      "model_forward_time": 1.731370210647583,
      "step": 97381
    },
    {
      "epoch": 2.229785919189453e-05,
      "step": 97381,
      "training_step_time": 8.593887090682983
    },
    {
      "epoch": 2.229881286621094e-05,
      "model_forward_time": 1.7930195331573486,
      "step": 97382
    },
    {
      "epoch": 2.229881286621094e-05,
      "step": 97382,
      "training_step_time": 7.91994571685791
    },
    {
      "epoch": 2.2299766540527343e-05,
      "model_forward_time": 1.731041431427002,
      "step": 97383
    },
    {
      "epoch": 2.2299766540527343e-05,
      "step": 97383,
      "training_step_time": 7.752268314361572
    },
    {
      "epoch": 2.230072021484375e-05,
      "model_forward_time": 1.73341703414917,
      "step": 97384
    },
    {
      "epoch": 2.230072021484375e-05,
      "step": 97384,
      "training_step_time": 7.56703519821167
    },
    {
      "epoch": 2.2301673889160158e-05,
      "model_forward_time": 2.3814826011657715,
      "step": 97385
    },
    {
      "epoch": 2.2301673889160158e-05,
      "step": 97385,
      "training_step_time": 7.570398330688477
    },
    {
      "epoch": 2.2302627563476562e-05,
      "model_forward_time": 1.7298765182495117,
      "step": 97386
    },
    {
      "epoch": 2.2302627563476562e-05,
      "step": 97386,
      "training_step_time": 9.428029298782349
    },
    {
      "epoch": 2.230358123779297e-05,
      "model_forward_time": 1.7280786037445068,
      "step": 97387
    },
    {
      "epoch": 2.230358123779297e-05,
      "step": 97387,
      "training_step_time": 7.667177677154541
    },
    {
      "epoch": 2.2304534912109377e-05,
      "model_forward_time": 1.7309255599975586,
      "step": 97388
    },
    {
      "epoch": 2.2304534912109377e-05,
      "step": 97388,
      "training_step_time": 8.555055379867554
    },
    {
      "epoch": 2.230548858642578e-05,
      "model_forward_time": 1.7335498332977295,
      "step": 97389
    },
    {
      "epoch": 2.230548858642578e-05,
      "step": 97389,
      "training_step_time": 8.60595703125
    },
    {
      "epoch": 2.2306442260742188e-05,
      "grad_norm": 0.09224212914705276,
      "learning_rate": 1.8626722878546943e-08,
      "loss": 0.0549,
      "step": 97390
    },
    {
      "epoch": 2.2306442260742188e-05,
      "model_forward_time": 1.728487491607666,
      "step": 97390
    },
    {
      "dynamics_loss_avg": 0.0434620700776577,
      "epoch": 2.2306442260742188e-05,
      "step": 97390
    },
    {
      "action_loss_avg": 0.0012419961421983317,
      "epoch": 2.2306442260742188e-05,
      "step": 97390
    },
    {
      "epoch": 2.2306442260742188e-05,
      "step": 97390,
      "training_step_time": 7.621025085449219
    },
    {
      "epoch": 2.2307395935058592e-05,
      "model_forward_time": 1.791123628616333,
      "step": 97391
    },
    {
      "epoch": 2.2307395935058592e-05,
      "step": 97391,
      "training_step_time": 7.8492419719696045
    },
    {
      "epoch": 2.2308349609375e-05,
      "model_forward_time": 1.7304902076721191,
      "step": 97392
    },
    {
      "epoch": 2.2308349609375e-05,
      "step": 97392,
      "training_step_time": 7.749825716018677
    },
    {
      "epoch": 2.2309303283691407e-05,
      "model_forward_time": 1.724984884262085,
      "step": 97393
    },
    {
      "epoch": 2.2309303283691407e-05,
      "step": 97393,
      "training_step_time": 7.581704139709473
    },
    {
      "epoch": 2.231025695800781e-05,
      "model_forward_time": 2.3985755443573,
      "step": 97394
    },
    {
      "epoch": 2.231025695800781e-05,
      "step": 97394,
      "training_step_time": 7.493783235549927
    },
    {
      "epoch": 2.231121063232422e-05,
      "model_forward_time": 1.7317659854888916,
      "step": 97395
    },
    {
      "epoch": 2.231121063232422e-05,
      "step": 97395,
      "training_step_time": 9.397945642471313
    },
    {
      "epoch": 2.2312164306640626e-05,
      "model_forward_time": 1.730909824371338,
      "step": 97396
    },
    {
      "epoch": 2.2312164306640626e-05,
      "step": 97396,
      "training_step_time": 7.735336542129517
    },
    {
      "epoch": 2.231311798095703e-05,
      "model_forward_time": 1.7233588695526123,
      "step": 97397
    },
    {
      "epoch": 2.231311798095703e-05,
      "step": 97397,
      "training_step_time": 7.872010946273804
    },
    {
      "epoch": 2.2314071655273437e-05,
      "model_forward_time": 1.7320208549499512,
      "step": 97398
    },
    {
      "epoch": 2.2314071655273437e-05,
      "step": 97398,
      "training_step_time": 7.721522569656372
    },
    {
      "epoch": 2.2315025329589845e-05,
      "model_forward_time": 1.731278657913208,
      "step": 97399
    },
    {
      "epoch": 2.2315025329589845e-05,
      "step": 97399,
      "training_step_time": 8.580278158187866
    },
    {
      "epoch": 2.231597900390625e-05,
      "grad_norm": 0.08369569480419159,
      "learning_rate": 1.8484405070516855e-08,
      "loss": 0.0561,
      "step": 97400
    },
    {
      "epoch": 2.231597900390625e-05,
      "model_forward_time": 1.749361276626587,
      "step": 97400
    },
    {
      "dynamics_loss_avg": 0.04162432979792356,
      "epoch": 2.231597900390625e-05,
      "step": 97400
    },
    {
      "action_loss_avg": 0.0009780689812032505,
      "epoch": 2.231597900390625e-05,
      "step": 97400
    },
    {
      "epoch": 2.231597900390625e-05,
      "step": 97400,
      "training_step_time": 7.808123588562012
    },
    {
      "epoch": 2.2316932678222656e-05,
      "model_forward_time": 1.7321538925170898,
      "step": 97401
    },
    {
      "epoch": 2.2316932678222656e-05,
      "step": 97401,
      "training_step_time": 9.401508808135986
    },
    {
      "epoch": 2.2317886352539063e-05,
      "model_forward_time": 1.7280840873718262,
      "step": 97402
    },
    {
      "epoch": 2.2317886352539063e-05,
      "step": 97402,
      "training_step_time": 7.695180416107178
    },
    {
      "epoch": 2.2318840026855468e-05,
      "model_forward_time": 2.3812155723571777,
      "step": 97403
    },
    {
      "epoch": 2.2318840026855468e-05,
      "step": 97403,
      "training_step_time": 8.348382234573364
    },
    {
      "epoch": 2.2319793701171875e-05,
      "model_forward_time": 1.731719970703125,
      "step": 97404
    },
    {
      "epoch": 2.2319793701171875e-05,
      "step": 97404,
      "training_step_time": 9.76244330406189
    },
    {
      "epoch": 2.2320747375488282e-05,
      "model_forward_time": 1.72810959815979,
      "step": 97405
    },
    {
      "epoch": 2.2320747375488282e-05,
      "step": 97405,
      "training_step_time": 8.616317987442017
    },
    {
      "epoch": 2.2321701049804686e-05,
      "model_forward_time": 1.7318496704101562,
      "step": 97406
    },
    {
      "epoch": 2.2321701049804686e-05,
      "step": 97406,
      "training_step_time": 7.915153503417969
    },
    {
      "epoch": 2.2322654724121094e-05,
      "model_forward_time": 1.7302660942077637,
      "step": 97407
    },
    {
      "epoch": 2.2322654724121094e-05,
      "step": 97407,
      "training_step_time": 7.7420830726623535
    },
    {
      "epoch": 2.23236083984375e-05,
      "model_forward_time": 1.7336211204528809,
      "step": 97408
    },
    {
      "epoch": 2.23236083984375e-05,
      "step": 97408,
      "training_step_time": 7.736856460571289
    },
    {
      "epoch": 2.2324562072753905e-05,
      "model_forward_time": 1.7294409275054932,
      "step": 97409
    },
    {
      "epoch": 2.2324562072753905e-05,
      "step": 97409,
      "training_step_time": 7.895016431808472
    },
    {
      "epoch": 2.2325515747070313e-05,
      "grad_norm": 0.10392481088638306,
      "learning_rate": 1.8342632033542805e-08,
      "loss": 0.0559,
      "step": 97410
    },
    {
      "epoch": 2.2325515747070313e-05,
      "model_forward_time": 1.8209209442138672,
      "step": 97410
    },
    {
      "dynamics_loss_avg": 0.06686576697975397,
      "epoch": 2.2325515747070313e-05,
      "step": 97410
    },
    {
      "action_loss_avg": 0.001763774739811197,
      "epoch": 2.2325515747070313e-05,
      "step": 97410
    },
    {
      "epoch": 2.2325515747070313e-05,
      "step": 97410,
      "training_step_time": 7.66108775138855
    },
    {
      "epoch": 2.232646942138672e-05,
      "model_forward_time": 1.7328979969024658,
      "step": 97411
    },
    {
      "epoch": 2.232646942138672e-05,
      "step": 97411,
      "training_step_time": 7.722956418991089
    },
    {
      "epoch": 2.2327423095703124e-05,
      "model_forward_time": 2.432042360305786,
      "step": 97412
    },
    {
      "epoch": 2.2327423095703124e-05,
      "step": 97412,
      "training_step_time": 7.5855793952941895
    },
    {
      "epoch": 2.232837677001953e-05,
      "model_forward_time": 1.7301356792449951,
      "step": 97413
    },
    {
      "epoch": 2.232837677001953e-05,
      "step": 97413,
      "training_step_time": 8.578017234802246
    },
    {
      "epoch": 2.232933044433594e-05,
      "model_forward_time": 1.7282683849334717,
      "step": 97414
    },
    {
      "epoch": 2.232933044433594e-05,
      "step": 97414,
      "training_step_time": 7.8255767822265625
    },
    {
      "epoch": 2.2330284118652343e-05,
      "model_forward_time": 1.7302892208099365,
      "step": 97415
    },
    {
      "epoch": 2.2330284118652343e-05,
      "step": 97415,
      "training_step_time": 7.895597457885742
    },
    {
      "epoch": 2.233123779296875e-05,
      "model_forward_time": 1.7257020473480225,
      "step": 97416
    },
    {
      "epoch": 2.233123779296875e-05,
      "step": 97416,
      "training_step_time": 7.690086126327515
    },
    {
      "epoch": 2.2332191467285158e-05,
      "model_forward_time": 1.7333133220672607,
      "step": 97417
    },
    {
      "epoch": 2.2332191467285158e-05,
      "step": 97417,
      "training_step_time": 7.695696592330933
    },
    {
      "epoch": 2.2333145141601562e-05,
      "model_forward_time": 1.740342378616333,
      "step": 97418
    },
    {
      "epoch": 2.2333145141601562e-05,
      "step": 97418,
      "training_step_time": 7.768323183059692
    },
    {
      "epoch": 2.233409881591797e-05,
      "model_forward_time": 1.8318822383880615,
      "step": 97419
    },
    {
      "epoch": 2.233409881591797e-05,
      "step": 97419,
      "training_step_time": 7.618191242218018
    },
    {
      "epoch": 2.2335052490234377e-05,
      "grad_norm": 0.074794702231884,
      "learning_rate": 1.8201403783129067e-08,
      "loss": 0.0559,
      "step": 97420
    },
    {
      "epoch": 2.2335052490234377e-05,
      "model_forward_time": 1.734771728515625,
      "step": 97420
    },
    {
      "dynamics_loss_avg": 0.05637110061943531,
      "epoch": 2.2335052490234377e-05,
      "step": 97420
    },
    {
      "action_loss_avg": 0.0015242060500895605,
      "epoch": 2.2335052490234377e-05,
      "step": 97420
    },
    {
      "epoch": 2.2335052490234377e-05,
      "step": 97420,
      "training_step_time": 7.683568477630615
    },
    {
      "epoch": 2.233600616455078e-05,
      "model_forward_time": 2.3964085578918457,
      "step": 97421
    },
    {
      "epoch": 2.233600616455078e-05,
      "step": 97421,
      "training_step_time": 7.495880603790283
    },
    {
      "epoch": 2.2336959838867188e-05,
      "model_forward_time": 1.7352440357208252,
      "step": 97422
    },
    {
      "epoch": 2.2336959838867188e-05,
      "step": 97422,
      "training_step_time": 7.636061429977417
    },
    {
      "epoch": 2.2337913513183592e-05,
      "model_forward_time": 1.7377214431762695,
      "step": 97423
    },
    {
      "epoch": 2.2337913513183592e-05,
      "step": 97423,
      "training_step_time": 9.810359001159668
    },
    {
      "epoch": 2.23388671875e-05,
      "model_forward_time": 1.734858751296997,
      "step": 97424
    },
    {
      "epoch": 2.23388671875e-05,
      "step": 97424,
      "training_step_time": 7.727954387664795
    },
    {
      "epoch": 2.2339820861816407e-05,
      "model_forward_time": 1.7376303672790527,
      "step": 97425
    },
    {
      "epoch": 2.2339820861816407e-05,
      "step": 97425,
      "training_step_time": 8.327141761779785
    },
    {
      "epoch": 2.234077453613281e-05,
      "model_forward_time": 1.8117296695709229,
      "step": 97426
    },
    {
      "epoch": 2.234077453613281e-05,
      "step": 97426,
      "training_step_time": 7.4206109046936035
    },
    {
      "epoch": 2.234172821044922e-05,
      "model_forward_time": 1.7299818992614746,
      "step": 97427
    },
    {
      "epoch": 2.234172821044922e-05,
      "step": 97427,
      "training_step_time": 7.736744403839111
    },
    {
      "epoch": 2.2342681884765626e-05,
      "model_forward_time": 1.8305482864379883,
      "step": 97428
    },
    {
      "epoch": 2.2342681884765626e-05,
      "step": 97428,
      "training_step_time": 7.68814492225647
    },
    {
      "epoch": 2.234363555908203e-05,
      "model_forward_time": 1.7383112907409668,
      "step": 97429
    },
    {
      "epoch": 2.234363555908203e-05,
      "step": 97429,
      "training_step_time": 7.719976902008057
    },
    {
      "epoch": 2.2344589233398437e-05,
      "grad_norm": 0.06769038736820221,
      "learning_rate": 1.806072033471884e-08,
      "loss": 0.0559,
      "step": 97430
    },
    {
      "epoch": 2.2344589233398437e-05,
      "model_forward_time": 1.7239444255828857,
      "step": 97430
    },
    {
      "dynamics_loss_avg": 0.06311918664723634,
      "epoch": 2.2344589233398437e-05,
      "step": 97430
    },
    {
      "action_loss_avg": 0.0014080741791985929,
      "epoch": 2.2344589233398437e-05,
      "step": 97430
    },
    {
      "epoch": 2.2344589233398437e-05,
      "step": 97430,
      "training_step_time": 8.378774404525757
    },
    {
      "epoch": 2.2345542907714845e-05,
      "model_forward_time": 1.7317368984222412,
      "step": 97431
    },
    {
      "epoch": 2.2345542907714845e-05,
      "step": 97431,
      "training_step_time": 9.55469012260437
    },
    {
      "epoch": 2.234649658203125e-05,
      "model_forward_time": 1.7292499542236328,
      "step": 97432
    },
    {
      "epoch": 2.234649658203125e-05,
      "step": 97432,
      "training_step_time": 8.940373420715332
    },
    {
      "epoch": 2.2347450256347656e-05,
      "model_forward_time": 1.7305655479431152,
      "step": 97433
    },
    {
      "epoch": 2.2347450256347656e-05,
      "step": 97433,
      "training_step_time": 10.357883214950562
    },
    {
      "epoch": 2.2348403930664064e-05,
      "model_forward_time": 1.7305762767791748,
      "step": 97434
    },
    {
      "epoch": 2.2348403930664064e-05,
      "step": 97434,
      "training_step_time": 8.861880779266357
    },
    {
      "epoch": 2.2349357604980468e-05,
      "model_forward_time": 1.726494550704956,
      "step": 97435
    },
    {
      "epoch": 2.2349357604980468e-05,
      "step": 97435,
      "training_step_time": 8.453633785247803
    },
    {
      "epoch": 2.2350311279296875e-05,
      "model_forward_time": 1.7286839485168457,
      "step": 97436
    },
    {
      "epoch": 2.2350311279296875e-05,
      "step": 97436,
      "training_step_time": 7.794003009796143
    },
    {
      "epoch": 2.2351264953613282e-05,
      "model_forward_time": 1.7872967720031738,
      "step": 97437
    },
    {
      "epoch": 2.2351264953613282e-05,
      "step": 97437,
      "training_step_time": 7.763180494308472
    },
    {
      "epoch": 2.2352218627929686e-05,
      "model_forward_time": 1.7288942337036133,
      "step": 97438
    },
    {
      "epoch": 2.2352218627929686e-05,
      "step": 97438,
      "training_step_time": 7.794502019882202
    },
    {
      "epoch": 2.2353172302246094e-05,
      "model_forward_time": 1.725616216659546,
      "step": 97439
    },
    {
      "epoch": 2.2353172302246094e-05,
      "step": 97439,
      "training_step_time": 8.434850692749023
    },
    {
      "epoch": 2.23541259765625e-05,
      "grad_norm": 0.08485573530197144,
      "learning_rate": 1.7920581703698704e-08,
      "loss": 0.0541,
      "step": 97440
    },
    {
      "epoch": 2.23541259765625e-05,
      "model_forward_time": 2.379160165786743,
      "step": 97440
    },
    {
      "dynamics_loss_avg": 0.04504030970856547,
      "epoch": 2.23541259765625e-05,
      "step": 97440
    },
    {
      "action_loss_avg": 0.001398248557234183,
      "epoch": 2.23541259765625e-05,
      "step": 97440
    },
    {
      "epoch": 2.23541259765625e-05,
      "step": 97440,
      "training_step_time": 8.221815586090088
    },
    {
      "epoch": 2.2355079650878905e-05,
      "model_forward_time": 1.7298853397369385,
      "step": 97441
    },
    {
      "epoch": 2.2355079650878905e-05,
      "step": 97441,
      "training_step_time": 8.528133153915405
    },
    {
      "epoch": 2.2356033325195313e-05,
      "model_forward_time": 1.7331490516662598,
      "step": 97442
    },
    {
      "epoch": 2.2356033325195313e-05,
      "step": 97442,
      "training_step_time": 9.395352125167847
    },
    {
      "epoch": 2.235698699951172e-05,
      "model_forward_time": 1.7312273979187012,
      "step": 97443
    },
    {
      "epoch": 2.235698699951172e-05,
      "step": 97443,
      "training_step_time": 8.048381328582764
    },
    {
      "epoch": 2.2357940673828124e-05,
      "model_forward_time": 1.7334511280059814,
      "step": 97444
    },
    {
      "epoch": 2.2357940673828124e-05,
      "step": 97444,
      "training_step_time": 7.903160095214844
    },
    {
      "epoch": 2.235889434814453e-05,
      "model_forward_time": 1.7291538715362549,
      "step": 97445
    },
    {
      "epoch": 2.235889434814453e-05,
      "step": 97445,
      "training_step_time": 8.59231162071228
    },
    {
      "epoch": 2.235984802246094e-05,
      "model_forward_time": 1.7557337284088135,
      "step": 97446
    },
    {
      "epoch": 2.235984802246094e-05,
      "step": 97446,
      "training_step_time": 7.881020784378052
    },
    {
      "epoch": 2.2360801696777343e-05,
      "model_forward_time": 1.7297518253326416,
      "step": 97447
    },
    {
      "epoch": 2.2360801696777343e-05,
      "step": 97447,
      "training_step_time": 8.863871574401855
    },
    {
      "epoch": 2.236175537109375e-05,
      "model_forward_time": 1.7332093715667725,
      "step": 97448
    },
    {
      "epoch": 2.236175537109375e-05,
      "step": 97448,
      "training_step_time": 8.297690868377686
    },
    {
      "epoch": 2.2362709045410158e-05,
      "model_forward_time": 2.3837952613830566,
      "step": 97449
    },
    {
      "epoch": 2.2362709045410158e-05,
      "step": 97449,
      "training_step_time": 8.32370924949646
    },
    {
      "epoch": 2.2363662719726562e-05,
      "grad_norm": 0.07086486369371414,
      "learning_rate": 1.7780987905393065e-08,
      "loss": 0.0544,
      "step": 97450
    },
    {
      "epoch": 2.2363662719726562e-05,
      "model_forward_time": 1.7196309566497803,
      "step": 97450
    },
    {
      "dynamics_loss_avg": 0.05515855550765991,
      "epoch": 2.2363662719726562e-05,
      "step": 97450
    },
    {
      "action_loss_avg": 0.0011761853093048558,
      "epoch": 2.2363662719726562e-05,
      "step": 97450
    },
    {
      "epoch": 2.2363662719726562e-05,
      "step": 97450,
      "training_step_time": 9.606215000152588
    },
    {
      "epoch": 2.236461639404297e-05,
      "model_forward_time": 1.733046054840088,
      "step": 97451
    },
    {
      "epoch": 2.236461639404297e-05,
      "step": 97451,
      "training_step_time": 10.457407474517822
    },
    {
      "epoch": 2.2365570068359377e-05,
      "model_forward_time": 1.732954740524292,
      "step": 97452
    },
    {
      "epoch": 2.2365570068359377e-05,
      "step": 97452,
      "training_step_time": 8.56730055809021
    },
    {
      "epoch": 2.236652374267578e-05,
      "model_forward_time": 1.735809564590454,
      "step": 97453
    },
    {
      "epoch": 2.236652374267578e-05,
      "step": 97453,
      "training_step_time": 7.8584678173065186
    },
    {
      "epoch": 2.2367477416992188e-05,
      "model_forward_time": 1.7337312698364258,
      "step": 97454
    },
    {
      "epoch": 2.2367477416992188e-05,
      "step": 97454,
      "training_step_time": 7.8863513469696045
    },
    {
      "epoch": 2.2368431091308592e-05,
      "model_forward_time": 1.7445673942565918,
      "step": 97455
    },
    {
      "epoch": 2.2368431091308592e-05,
      "step": 97455,
      "training_step_time": 7.845353841781616
    },
    {
      "epoch": 2.2369384765625e-05,
      "model_forward_time": 1.7611446380615234,
      "step": 97456
    },
    {
      "epoch": 2.2369384765625e-05,
      "step": 97456,
      "training_step_time": 7.7375898361206055
    },
    {
      "epoch": 2.2370338439941407e-05,
      "model_forward_time": 1.7342190742492676,
      "step": 97457
    },
    {
      "epoch": 2.2370338439941407e-05,
      "step": 97457,
      "training_step_time": 8.50775957107544
    },
    {
      "epoch": 2.237129211425781e-05,
      "model_forward_time": 2.3972556591033936,
      "step": 97458
    },
    {
      "epoch": 2.237129211425781e-05,
      "step": 97458,
      "training_step_time": 7.932060956954956
    },
    {
      "epoch": 2.237224578857422e-05,
      "model_forward_time": 1.7332956790924072,
      "step": 97459
    },
    {
      "epoch": 2.237224578857422e-05,
      "step": 97459,
      "training_step_time": 7.801910877227783
    },
    {
      "epoch": 2.2373199462890626e-05,
      "grad_norm": 0.08766872435808182,
      "learning_rate": 1.764193895506805e-08,
      "loss": 0.0544,
      "step": 97460
    },
    {
      "epoch": 2.2373199462890626e-05,
      "model_forward_time": 1.7318041324615479,
      "step": 97460
    },
    {
      "dynamics_loss_avg": 0.06472170483320952,
      "epoch": 2.2373199462890626e-05,
      "step": 97460
    },
    {
      "action_loss_avg": 0.0014607115706894546,
      "epoch": 2.2373199462890626e-05,
      "step": 97460
    },
    {
      "epoch": 2.2373199462890626e-05,
      "step": 97460,
      "training_step_time": 9.890781164169312
    },
    {
      "epoch": 2.237415313720703e-05,
      "model_forward_time": 1.741455316543579,
      "step": 97461
    },
    {
      "epoch": 2.237415313720703e-05,
      "step": 97461,
      "training_step_time": 8.359654188156128
    },
    {
      "epoch": 2.2375106811523437e-05,
      "model_forward_time": 1.7395474910736084,
      "step": 97462
    },
    {
      "epoch": 2.2375106811523437e-05,
      "step": 97462,
      "training_step_time": 7.784301519393921
    },
    {
      "epoch": 2.2376060485839845e-05,
      "model_forward_time": 1.7322826385498047,
      "step": 97463
    },
    {
      "epoch": 2.2376060485839845e-05,
      "step": 97463,
      "training_step_time": 7.896474361419678
    },
    {
      "epoch": 2.237701416015625e-05,
      "model_forward_time": 1.7337720394134521,
      "step": 97464
    },
    {
      "epoch": 2.237701416015625e-05,
      "step": 97464,
      "training_step_time": 8.661728143692017
    },
    {
      "epoch": 2.2377967834472656e-05,
      "model_forward_time": 1.8272058963775635,
      "step": 97465
    },
    {
      "epoch": 2.2377967834472656e-05,
      "step": 97465,
      "training_step_time": 9.32222580909729
    },
    {
      "epoch": 2.2378921508789064e-05,
      "model_forward_time": 1.7335302829742432,
      "step": 97466
    },
    {
      "epoch": 2.2378921508789064e-05,
      "step": 97466,
      "training_step_time": 7.47861647605896
    },
    {
      "epoch": 2.2379875183105468e-05,
      "model_forward_time": 2.3717362880706787,
      "step": 97467
    },
    {
      "epoch": 2.2379875183105468e-05,
      "step": 97467,
      "training_step_time": 7.7071263790130615
    },
    {
      "epoch": 2.2380828857421875e-05,
      "model_forward_time": 1.7332344055175781,
      "step": 97468
    },
    {
      "epoch": 2.2380828857421875e-05,
      "step": 97468,
      "training_step_time": 8.642156600952148
    },
    {
      "epoch": 2.2381782531738282e-05,
      "model_forward_time": 1.7406611442565918,
      "step": 97469
    },
    {
      "epoch": 2.2381782531738282e-05,
      "step": 97469,
      "training_step_time": 9.338682651519775
    },
    {
      "epoch": 2.2382736206054686e-05,
      "grad_norm": 0.09326186031103134,
      "learning_rate": 1.7503434867929826e-08,
      "loss": 0.0537,
      "step": 97470
    },
    {
      "epoch": 2.2382736206054686e-05,
      "model_forward_time": 1.7363102436065674,
      "step": 97470
    },
    {
      "dynamics_loss_avg": 0.0408508813008666,
      "epoch": 2.2382736206054686e-05,
      "step": 97470
    },
    {
      "action_loss_avg": 0.0013134926193743013,
      "epoch": 2.2382736206054686e-05,
      "step": 97470
    },
    {
      "epoch": 2.2382736206054686e-05,
      "step": 97470,
      "training_step_time": 8.36043381690979
    },
    {
      "epoch": 2.2383689880371094e-05,
      "model_forward_time": 1.7317771911621094,
      "step": 97471
    },
    {
      "epoch": 2.2383689880371094e-05,
      "step": 97471,
      "training_step_time": 8.894180059432983
    },
    {
      "epoch": 2.23846435546875e-05,
      "model_forward_time": 1.7374074459075928,
      "step": 97472
    },
    {
      "epoch": 2.23846435546875e-05,
      "step": 97472,
      "training_step_time": 7.9864020347595215
    },
    {
      "epoch": 2.2385597229003905e-05,
      "model_forward_time": 1.7404086589813232,
      "step": 97473
    },
    {
      "epoch": 2.2385597229003905e-05,
      "step": 97473,
      "training_step_time": 7.8337562084198
    },
    {
      "epoch": 2.2386550903320313e-05,
      "model_forward_time": 1.8222622871398926,
      "step": 97474
    },
    {
      "epoch": 2.2386550903320313e-05,
      "step": 97474,
      "training_step_time": 7.7734479904174805
    },
    {
      "epoch": 2.238750457763672e-05,
      "model_forward_time": 1.733649492263794,
      "step": 97475
    },
    {
      "epoch": 2.238750457763672e-05,
      "step": 97475,
      "training_step_time": 7.73872709274292
    },
    {
      "epoch": 2.2388458251953124e-05,
      "model_forward_time": 2.3865551948547363,
      "step": 97476
    },
    {
      "epoch": 2.2388458251953124e-05,
      "step": 97476,
      "training_step_time": 7.721973419189453
    },
    {
      "epoch": 2.238941192626953e-05,
      "model_forward_time": 1.7354381084442139,
      "step": 97477
    },
    {
      "epoch": 2.238941192626953e-05,
      "step": 97477,
      "training_step_time": 8.171331882476807
    },
    {
      "epoch": 2.239036560058594e-05,
      "model_forward_time": 1.7368831634521484,
      "step": 97478
    },
    {
      "epoch": 2.239036560058594e-05,
      "step": 97478,
      "training_step_time": 8.46052360534668
    },
    {
      "epoch": 2.2391319274902343e-05,
      "model_forward_time": 1.7347729206085205,
      "step": 97479
    },
    {
      "epoch": 2.2391319274902343e-05,
      "step": 97479,
      "training_step_time": 8.464574813842773
    },
    {
      "epoch": 2.239227294921875e-05,
      "grad_norm": 0.08760759234428406,
      "learning_rate": 1.7365475659125163e-08,
      "loss": 0.0532,
      "step": 97480
    },
    {
      "epoch": 2.239227294921875e-05,
      "model_forward_time": 1.7265675067901611,
      "step": 97480
    },
    {
      "dynamics_loss_avg": 0.05369325522333383,
      "epoch": 2.239227294921875e-05,
      "step": 97480
    },
    {
      "action_loss_avg": 0.0014966786722652615,
      "epoch": 2.239227294921875e-05,
      "step": 97480
    },
    {
      "epoch": 2.239227294921875e-05,
      "step": 97480,
      "training_step_time": 8.499766826629639
    },
    {
      "epoch": 2.2393226623535158e-05,
      "model_forward_time": 1.7272288799285889,
      "step": 97481
    },
    {
      "epoch": 2.2393226623535158e-05,
      "step": 97481,
      "training_step_time": 7.981213808059692
    },
    {
      "epoch": 2.2394180297851562e-05,
      "model_forward_time": 1.731065273284912,
      "step": 97482
    },
    {
      "epoch": 2.2394180297851562e-05,
      "step": 97482,
      "training_step_time": 7.704899787902832
    },
    {
      "epoch": 2.239513397216797e-05,
      "model_forward_time": 1.7686779499053955,
      "step": 97483
    },
    {
      "epoch": 2.239513397216797e-05,
      "step": 97483,
      "training_step_time": 7.820683240890503
    },
    {
      "epoch": 2.2396087646484377e-05,
      "model_forward_time": 1.7312767505645752,
      "step": 97484
    },
    {
      "epoch": 2.2396087646484377e-05,
      "step": 97484,
      "training_step_time": 7.734846591949463
    },
    {
      "epoch": 2.239704132080078e-05,
      "model_forward_time": 2.3857898712158203,
      "step": 97485
    },
    {
      "epoch": 2.239704132080078e-05,
      "step": 97485,
      "training_step_time": 7.59888219833374
    },
    {
      "epoch": 2.2397994995117188e-05,
      "model_forward_time": 1.7328951358795166,
      "step": 97486
    },
    {
      "epoch": 2.2397994995117188e-05,
      "step": 97486,
      "training_step_time": 7.506333112716675
    },
    {
      "epoch": 2.2398948669433592e-05,
      "model_forward_time": 1.7353382110595703,
      "step": 97487
    },
    {
      "epoch": 2.2398948669433592e-05,
      "step": 97487,
      "training_step_time": 8.469330072402954
    },
    {
      "epoch": 2.239990234375e-05,
      "model_forward_time": 1.7296009063720703,
      "step": 97488
    },
    {
      "epoch": 2.239990234375e-05,
      "step": 97488,
      "training_step_time": 8.36163878440857
    },
    {
      "epoch": 2.2400856018066407e-05,
      "model_forward_time": 1.73301362991333,
      "step": 97489
    },
    {
      "epoch": 2.2400856018066407e-05,
      "step": 97489,
      "training_step_time": 7.938472747802734
    },
    {
      "epoch": 2.240180969238281e-05,
      "grad_norm": 0.07052242755889893,
      "learning_rate": 1.7228061343740333e-08,
      "loss": 0.0542,
      "step": 97490
    },
    {
      "epoch": 2.240180969238281e-05,
      "model_forward_time": 1.7276902198791504,
      "step": 97490
    },
    {
      "dynamics_loss_avg": 0.05695767812430859,
      "epoch": 2.240180969238281e-05,
      "step": 97490
    },
    {
      "action_loss_avg": 0.001478721760213375,
      "epoch": 2.240180969238281e-05,
      "step": 97490
    },
    {
      "epoch": 2.240180969238281e-05,
      "step": 97490,
      "training_step_time": 7.927593946456909
    },
    {
      "epoch": 2.240276336669922e-05,
      "model_forward_time": 1.73463773727417,
      "step": 97491
    },
    {
      "epoch": 2.240276336669922e-05,
      "step": 97491,
      "training_step_time": 8.516266822814941
    },
    {
      "epoch": 2.2403717041015626e-05,
      "model_forward_time": 1.7414546012878418,
      "step": 97492
    },
    {
      "epoch": 2.2403717041015626e-05,
      "step": 97492,
      "training_step_time": 8.796151876449585
    },
    {
      "epoch": 2.240467071533203e-05,
      "model_forward_time": 1.7308673858642578,
      "step": 97493
    },
    {
      "epoch": 2.240467071533203e-05,
      "step": 97493,
      "training_step_time": 7.880031585693359
    },
    {
      "epoch": 2.2405624389648437e-05,
      "model_forward_time": 2.3927218914031982,
      "step": 97494
    },
    {
      "epoch": 2.2405624389648437e-05,
      "step": 97494,
      "training_step_time": 7.770471096038818
    },
    {
      "epoch": 2.2406578063964845e-05,
      "model_forward_time": 1.7389285564422607,
      "step": 97495
    },
    {
      "epoch": 2.2406578063964845e-05,
      "step": 97495,
      "training_step_time": 7.687648057937622
    },
    {
      "epoch": 2.240753173828125e-05,
      "model_forward_time": 1.73793363571167,
      "step": 97496
    },
    {
      "epoch": 2.240753173828125e-05,
      "step": 97496,
      "training_step_time": 9.144789218902588
    },
    {
      "epoch": 2.2408485412597656e-05,
      "model_forward_time": 1.733942985534668,
      "step": 97497
    },
    {
      "epoch": 2.2408485412597656e-05,
      "step": 97497,
      "training_step_time": 8.637892723083496
    },
    {
      "epoch": 2.2409439086914064e-05,
      "model_forward_time": 1.7329199314117432,
      "step": 97498
    },
    {
      "epoch": 2.2409439086914064e-05,
      "step": 97498,
      "training_step_time": 7.9156553745269775
    },
    {
      "epoch": 2.2410392761230468e-05,
      "model_forward_time": 1.7309114933013916,
      "step": 97499
    },
    {
      "epoch": 2.2410392761230468e-05,
      "step": 97499,
      "training_step_time": 7.847223997116089
    },
    {
      "epoch": 2.2411346435546875e-05,
      "grad_norm": 0.12539204955101013,
      "learning_rate": 1.7091191936803308e-08,
      "loss": 0.0533,
      "step": 97500
    },
    {
      "epoch": 2.2411346435546875e-05,
      "model_forward_time": 1.7354156970977783,
      "step": 97500
    },
    {
      "dynamics_loss_avg": 0.06426885183900595,
      "epoch": 2.2411346435546875e-05,
      "step": 97500
    },
    {
      "action_loss_avg": 0.000948873937886674,
      "epoch": 2.2411346435546875e-05,
      "step": 97500
    },
    {
      "epoch": 2.2411346435546875e-05,
      "step": 97500,
      "training_step_time": 7.891637802124023
    },
    {
      "epoch": 2.2412300109863282e-05,
      "model_forward_time": 1.7459235191345215,
      "step": 97501
    },
    {
      "epoch": 2.2412300109863282e-05,
      "step": 97501,
      "training_step_time": 7.687485218048096
    },
    {
      "epoch": 2.2413253784179686e-05,
      "model_forward_time": 1.753901481628418,
      "step": 97502
    },
    {
      "epoch": 2.2413253784179686e-05,
      "step": 97502,
      "training_step_time": 7.719109296798706
    },
    {
      "epoch": 2.2414207458496094e-05,
      "model_forward_time": 2.386953592300415,
      "step": 97503
    },
    {
      "epoch": 2.2414207458496094e-05,
      "step": 97503,
      "training_step_time": 8.358632564544678
    },
    {
      "epoch": 2.24151611328125e-05,
      "model_forward_time": 1.7410783767700195,
      "step": 97504
    },
    {
      "epoch": 2.24151611328125e-05,
      "step": 97504,
      "training_step_time": 7.544190168380737
    },
    {
      "epoch": 2.2416114807128905e-05,
      "model_forward_time": 1.7385718822479248,
      "step": 97505
    },
    {
      "epoch": 2.2416114807128905e-05,
      "step": 97505,
      "training_step_time": 8.716145515441895
    },
    {
      "epoch": 2.2417068481445313e-05,
      "model_forward_time": 1.7375178337097168,
      "step": 97506
    },
    {
      "epoch": 2.2417068481445313e-05,
      "step": 97506,
      "training_step_time": 10.118762731552124
    },
    {
      "epoch": 2.241802215576172e-05,
      "model_forward_time": 1.739410400390625,
      "step": 97507
    },
    {
      "epoch": 2.241802215576172e-05,
      "step": 97507,
      "training_step_time": 8.438201904296875
    },
    {
      "epoch": 2.2418975830078124e-05,
      "model_forward_time": 1.733720064163208,
      "step": 97508
    },
    {
      "epoch": 2.2418975830078124e-05,
      "step": 97508,
      "training_step_time": 7.779334306716919
    },
    {
      "epoch": 2.241992950439453e-05,
      "model_forward_time": 1.736793041229248,
      "step": 97509
    },
    {
      "epoch": 2.241992950439453e-05,
      "step": 97509,
      "training_step_time": 8.49170708656311
    },
    {
      "epoch": 2.242088317871094e-05,
      "grad_norm": 0.07450967282056808,
      "learning_rate": 1.6954867453281565e-08,
      "loss": 0.0554,
      "step": 97510
    },
    {
      "epoch": 2.242088317871094e-05,
      "model_forward_time": 1.7364022731781006,
      "step": 97510
    },
    {
      "dynamics_loss_avg": 0.05405746642500162,
      "epoch": 2.242088317871094e-05,
      "step": 97510
    },
    {
      "action_loss_avg": 0.0016472565563162788,
      "epoch": 2.242088317871094e-05,
      "step": 97510
    },
    {
      "epoch": 2.242088317871094e-05,
      "step": 97510,
      "training_step_time": 7.803141832351685
    },
    {
      "epoch": 2.2421836853027343e-05,
      "model_forward_time": 1.8385283946990967,
      "step": 97511
    },
    {
      "epoch": 2.2421836853027343e-05,
      "step": 97511,
      "training_step_time": 9.61184573173523
    },
    {
      "epoch": 2.242279052734375e-05,
      "model_forward_time": 2.953855037689209,
      "step": 97512
    },
    {
      "epoch": 2.242279052734375e-05,
      "step": 97512,
      "training_step_time": 9.789342403411865
    },
    {
      "epoch": 2.2423744201660158e-05,
      "model_forward_time": 1.8597021102905273,
      "step": 97513
    },
    {
      "epoch": 2.2423744201660158e-05,
      "step": 97513,
      "training_step_time": 7.673415899276733
    },
    {
      "epoch": 2.2424697875976562e-05,
      "model_forward_time": 1.852708101272583,
      "step": 97514
    },
    {
      "epoch": 2.2424697875976562e-05,
      "step": 97514,
      "training_step_time": 9.354168891906738
    },
    {
      "epoch": 2.242565155029297e-05,
      "model_forward_time": 1.8422513008117676,
      "step": 97515
    },
    {
      "epoch": 2.242565155029297e-05,
      "step": 97515,
      "training_step_time": 8.941036701202393
    },
    {
      "epoch": 2.2426605224609373e-05,
      "model_forward_time": 1.885183334350586,
      "step": 97516
    },
    {
      "epoch": 2.2426605224609373e-05,
      "step": 97516,
      "training_step_time": 8.470758199691772
    },
    {
      "epoch": 2.242755889892578e-05,
      "model_forward_time": 1.9308371543884277,
      "step": 97517
    },
    {
      "epoch": 2.242755889892578e-05,
      "step": 97517,
      "training_step_time": 9.573753118515015
    },
    {
      "epoch": 2.2428512573242188e-05,
      "model_forward_time": 1.8186850547790527,
      "step": 97518
    },
    {
      "epoch": 2.2428512573242188e-05,
      "step": 97518,
      "training_step_time": 7.8363800048828125
    },
    {
      "epoch": 2.2429466247558592e-05,
      "model_forward_time": 1.7451372146606445,
      "step": 97519
    },
    {
      "epoch": 2.2429466247558592e-05,
      "step": 97519,
      "training_step_time": 7.873382329940796
    },
    {
      "epoch": 2.2430419921875e-05,
      "grad_norm": 0.08676350116729736,
      "learning_rate": 1.6819087908084287e-08,
      "loss": 0.0544,
      "step": 97520
    },
    {
      "epoch": 2.2430419921875e-05,
      "model_forward_time": 1.8098182678222656,
      "step": 97520
    },
    {
      "dynamics_loss_avg": 0.04827956957742572,
      "epoch": 2.2430419921875e-05,
      "step": 97520
    },
    {
      "action_loss_avg": 0.0013777283296803945,
      "epoch": 2.2430419921875e-05,
      "step": 97520
    },
    {
      "epoch": 2.2430419921875e-05,
      "step": 97520,
      "training_step_time": 8.879201173782349
    },
    {
      "epoch": 2.2431373596191407e-05,
      "model_forward_time": 2.392749547958374,
      "step": 97521
    },
    {
      "epoch": 2.2431373596191407e-05,
      "step": 97521,
      "training_step_time": 8.668248891830444
    },
    {
      "epoch": 2.243232727050781e-05,
      "model_forward_time": 1.3512029647827148,
      "step": 97522
    },
    {
      "epoch": 2.243232727050781e-05,
      "step": 97522,
      "training_step_time": 8.130933284759521
    },
    {
      "epoch": 2.243328094482422e-05,
      "model_forward_time": 1.736691951751709,
      "step": 97523
    },
    {
      "epoch": 2.243328094482422e-05,
      "step": 97523,
      "training_step_time": 7.7949135303497314
    },
    {
      "epoch": 2.2434234619140626e-05,
      "model_forward_time": 1.73445725440979,
      "step": 97524
    },
    {
      "epoch": 2.2434234619140626e-05,
      "step": 97524,
      "training_step_time": 8.715138912200928
    },
    {
      "epoch": 2.243518829345703e-05,
      "model_forward_time": 1.7278687953948975,
      "step": 97525
    },
    {
      "epoch": 2.243518829345703e-05,
      "step": 97525,
      "training_step_time": 8.374181509017944
    },
    {
      "epoch": 2.2436141967773437e-05,
      "model_forward_time": 1.728879690170288,
      "step": 97526
    },
    {
      "epoch": 2.2436141967773437e-05,
      "step": 97526,
      "training_step_time": 8.598475933074951
    },
    {
      "epoch": 2.2437095642089845e-05,
      "model_forward_time": 1.3551063537597656,
      "step": 97527
    },
    {
      "epoch": 2.2437095642089845e-05,
      "step": 97527,
      "training_step_time": 7.801245212554932
    },
    {
      "epoch": 2.243804931640625e-05,
      "model_forward_time": 1.723947525024414,
      "step": 97528
    },
    {
      "epoch": 2.243804931640625e-05,
      "step": 97528,
      "training_step_time": 7.923642873764038
    },
    {
      "epoch": 2.2439002990722656e-05,
      "model_forward_time": 1.78548264503479,
      "step": 97529
    },
    {
      "epoch": 2.2439002990722656e-05,
      "step": 97529,
      "training_step_time": 7.870711088180542
    },
    {
      "epoch": 2.2439956665039064e-05,
      "grad_norm": 0.09809618443250656,
      "learning_rate": 1.6683853316058484e-08,
      "loss": 0.0544,
      "step": 97530
    },
    {
      "epoch": 2.2439956665039064e-05,
      "model_forward_time": 2.394160270690918,
      "step": 97530
    },
    {
      "dynamics_loss_avg": 0.05568181164562702,
      "epoch": 2.2439956665039064e-05,
      "step": 97530
    },
    {
      "action_loss_avg": 0.0009672895190306008,
      "epoch": 2.2439956665039064e-05,
      "step": 97530
    },
    {
      "epoch": 2.2439956665039064e-05,
      "step": 97530,
      "training_step_time": 7.762234687805176
    },
    {
      "epoch": 2.2440910339355468e-05,
      "model_forward_time": 1.7243506908416748,
      "step": 97531
    },
    {
      "epoch": 2.2440910339355468e-05,
      "step": 97531,
      "training_step_time": 8.196892499923706
    },
    {
      "epoch": 2.2441864013671875e-05,
      "model_forward_time": 1.7358677387237549,
      "step": 97532
    },
    {
      "epoch": 2.2441864013671875e-05,
      "step": 97532,
      "training_step_time": 7.768673658370972
    },
    {
      "epoch": 2.2442817687988282e-05,
      "model_forward_time": 1.7348556518554688,
      "step": 97533
    },
    {
      "epoch": 2.2442817687988282e-05,
      "step": 97533,
      "training_step_time": 9.723999261856079
    },
    {
      "epoch": 2.2443771362304686e-05,
      "model_forward_time": 1.7382848262786865,
      "step": 97534
    },
    {
      "epoch": 2.2443771362304686e-05,
      "step": 97534,
      "training_step_time": 7.917663812637329
    },
    {
      "epoch": 2.2444725036621094e-05,
      "model_forward_time": 1.7324042320251465,
      "step": 97535
    },
    {
      "epoch": 2.2444725036621094e-05,
      "step": 97535,
      "training_step_time": 9.845180749893188
    },
    {
      "epoch": 2.24456787109375e-05,
      "model_forward_time": 1.7309563159942627,
      "step": 97536
    },
    {
      "epoch": 2.24456787109375e-05,
      "step": 97536,
      "training_step_time": 7.861528396606445
    },
    {
      "epoch": 2.2446632385253905e-05,
      "model_forward_time": 1.734571933746338,
      "step": 97537
    },
    {
      "epoch": 2.2446632385253905e-05,
      "step": 97537,
      "training_step_time": 8.424934148788452
    },
    {
      "epoch": 2.2447586059570313e-05,
      "model_forward_time": 1.7680366039276123,
      "step": 97538
    },
    {
      "epoch": 2.2447586059570313e-05,
      "step": 97538,
      "training_step_time": 8.95046854019165
    },
    {
      "epoch": 2.244853973388672e-05,
      "model_forward_time": 2.416628122329712,
      "step": 97539
    },
    {
      "epoch": 2.244853973388672e-05,
      "step": 97539,
      "training_step_time": 7.8252599239349365
    },
    {
      "epoch": 2.2449493408203124e-05,
      "grad_norm": 0.23267798125743866,
      "learning_rate": 1.6549163691995108e-08,
      "loss": 0.0549,
      "step": 97540
    },
    {
      "epoch": 2.2449493408203124e-05,
      "model_forward_time": 1.7301161289215088,
      "step": 97540
    },
    {
      "dynamics_loss_avg": 0.04217761680483818,
      "epoch": 2.2449493408203124e-05,
      "step": 97540
    },
    {
      "action_loss_avg": 0.0009042465870152227,
      "epoch": 2.2449493408203124e-05,
      "step": 97540
    },
    {
      "epoch": 2.2449493408203124e-05,
      "step": 97540,
      "training_step_time": 7.726559400558472
    },
    {
      "epoch": 2.245044708251953e-05,
      "model_forward_time": 1.7318074703216553,
      "step": 97541
    },
    {
      "epoch": 2.245044708251953e-05,
      "step": 97541,
      "training_step_time": 7.743997812271118
    },
    {
      "epoch": 2.245140075683594e-05,
      "model_forward_time": 1.737825870513916,
      "step": 97542
    },
    {
      "epoch": 2.245140075683594e-05,
      "step": 97542,
      "training_step_time": 9.902539730072021
    },
    {
      "epoch": 2.2452354431152343e-05,
      "model_forward_time": 1.732193946838379,
      "step": 97543
    },
    {
      "epoch": 2.2452354431152343e-05,
      "step": 97543,
      "training_step_time": 10.484188079833984
    },
    {
      "epoch": 2.245330810546875e-05,
      "model_forward_time": 1.7319867610931396,
      "step": 97544
    },
    {
      "epoch": 2.245330810546875e-05,
      "step": 97544,
      "training_step_time": 8.817145347595215
    },
    {
      "epoch": 2.2454261779785158e-05,
      "model_forward_time": 1.7308127880096436,
      "step": 97545
    },
    {
      "epoch": 2.2454261779785158e-05,
      "step": 97545,
      "training_step_time": 8.507095098495483
    },
    {
      "epoch": 2.2455215454101562e-05,
      "model_forward_time": 1.7360467910766602,
      "step": 97546
    },
    {
      "epoch": 2.2455215454101562e-05,
      "step": 97546,
      "training_step_time": 7.721084356307983
    },
    {
      "epoch": 2.245616912841797e-05,
      "model_forward_time": 1.7619004249572754,
      "step": 97547
    },
    {
      "epoch": 2.245616912841797e-05,
      "step": 97547,
      "training_step_time": 7.858731985092163
    },
    {
      "epoch": 2.2457122802734373e-05,
      "model_forward_time": 2.3844635486602783,
      "step": 97548
    },
    {
      "epoch": 2.2457122802734373e-05,
      "step": 97548,
      "training_step_time": 8.847514867782593
    },
    {
      "epoch": 2.245807647705078e-05,
      "model_forward_time": 1.7369775772094727,
      "step": 97549
    },
    {
      "epoch": 2.245807647705078e-05,
      "step": 97549,
      "training_step_time": 7.807420969009399
    },
    {
      "epoch": 2.2459030151367188e-05,
      "grad_norm": 0.07642539590597153,
      "learning_rate": 1.6415019050622373e-08,
      "loss": 0.057,
      "step": 97550
    },
    {
      "epoch": 2.2459030151367188e-05,
      "model_forward_time": 1.7195355892181396,
      "step": 97550
    },
    {
      "dynamics_loss_avg": 0.06921819504350424,
      "epoch": 2.2459030151367188e-05,
      "step": 97550
    },
    {
      "action_loss_avg": 0.0014303004834800958,
      "epoch": 2.2459030151367188e-05,
      "step": 97550
    },
    {
      "epoch": 2.2459030151367188e-05,
      "step": 97550,
      "training_step_time": 7.796475648880005
    },
    {
      "epoch": 2.2459983825683592e-05,
      "model_forward_time": 1.7160522937774658,
      "step": 97551
    },
    {
      "epoch": 2.2459983825683592e-05,
      "step": 97551,
      "training_step_time": 8.474647521972656
    },
    {
      "epoch": 2.24609375e-05,
      "model_forward_time": 1.7349107265472412,
      "step": 97552
    },
    {
      "epoch": 2.24609375e-05,
      "step": 97552,
      "training_step_time": 8.721080303192139
    },
    {
      "epoch": 2.2461891174316407e-05,
      "model_forward_time": 1.7253429889678955,
      "step": 97553
    },
    {
      "epoch": 2.2461891174316407e-05,
      "step": 97553,
      "training_step_time": 8.830790281295776
    },
    {
      "epoch": 2.246284484863281e-05,
      "model_forward_time": 1.7371580600738525,
      "step": 97554
    },
    {
      "epoch": 2.246284484863281e-05,
      "step": 97554,
      "training_step_time": 7.886553049087524
    },
    {
      "epoch": 2.246379852294922e-05,
      "model_forward_time": 1.735731601715088,
      "step": 97555
    },
    {
      "epoch": 2.246379852294922e-05,
      "step": 97555,
      "training_step_time": 7.8076817989349365
    },
    {
      "epoch": 2.2464752197265626e-05,
      "model_forward_time": 1.7567980289459229,
      "step": 97556
    },
    {
      "epoch": 2.2464752197265626e-05,
      "step": 97556,
      "training_step_time": 7.999058246612549
    },
    {
      "epoch": 2.246570587158203e-05,
      "model_forward_time": 1.7662382125854492,
      "step": 97557
    },
    {
      "epoch": 2.246570587158203e-05,
      "step": 97557,
      "training_step_time": 8.580480813980103
    },
    {
      "epoch": 2.2466659545898437e-05,
      "model_forward_time": 2.376845598220825,
      "step": 97558
    },
    {
      "epoch": 2.2466659545898437e-05,
      "step": 97558,
      "training_step_time": 7.630015850067139
    },
    {
      "epoch": 2.2467613220214845e-05,
      "model_forward_time": 1.7311840057373047,
      "step": 97559
    },
    {
      "epoch": 2.2467613220214845e-05,
      "step": 97559,
      "training_step_time": 8.235922574996948
    },
    {
      "epoch": 2.246856689453125e-05,
      "grad_norm": 0.08901333808898926,
      "learning_rate": 1.6281419406610212e-08,
      "loss": 0.0551,
      "step": 97560
    },
    {
      "epoch": 2.246856689453125e-05,
      "model_forward_time": 1.7310569286346436,
      "step": 97560
    },
    {
      "dynamics_loss_avg": 0.050523940566927196,
      "epoch": 2.246856689453125e-05,
      "step": 97560
    },
    {
      "action_loss_avg": 0.0014411385462153703,
      "epoch": 2.246856689453125e-05,
      "step": 97560
    },
    {
      "epoch": 2.246856689453125e-05,
      "step": 97560,
      "training_step_time": 8.61620545387268
    },
    {
      "epoch": 2.2469520568847656e-05,
      "model_forward_time": 1.7352197170257568,
      "step": 97561
    },
    {
      "epoch": 2.2469520568847656e-05,
      "step": 97561,
      "training_step_time": 8.04823350906372
    },
    {
      "epoch": 2.2470474243164064e-05,
      "model_forward_time": 1.7340013980865479,
      "step": 97562
    },
    {
      "epoch": 2.2470474243164064e-05,
      "step": 97562,
      "training_step_time": 8.805450439453125
    },
    {
      "epoch": 2.2471427917480468e-05,
      "model_forward_time": 1.7333946228027344,
      "step": 97563
    },
    {
      "epoch": 2.2471427917480468e-05,
      "step": 97563,
      "training_step_time": 8.15204668045044
    },
    {
      "epoch": 2.2472381591796875e-05,
      "model_forward_time": 1.737581491470337,
      "step": 97564
    },
    {
      "epoch": 2.2472381591796875e-05,
      "step": 97564,
      "training_step_time": 7.8573384284973145
    },
    {
      "epoch": 2.2473335266113282e-05,
      "model_forward_time": 1.7654054164886475,
      "step": 97565
    },
    {
      "epoch": 2.2473335266113282e-05,
      "step": 97565,
      "training_step_time": 7.748777866363525
    },
    {
      "epoch": 2.2474288940429686e-05,
      "model_forward_time": 1.755993366241455,
      "step": 97566
    },
    {
      "epoch": 2.2474288940429686e-05,
      "step": 97566,
      "training_step_time": 7.869799852371216
    },
    {
      "epoch": 2.2475242614746094e-05,
      "model_forward_time": 2.391188621520996,
      "step": 97567
    },
    {
      "epoch": 2.2475242614746094e-05,
      "step": 97567,
      "training_step_time": 7.744971752166748
    },
    {
      "epoch": 2.24761962890625e-05,
      "model_forward_time": 1.7387852668762207,
      "step": 97568
    },
    {
      "epoch": 2.24761962890625e-05,
      "step": 97568,
      "training_step_time": 9.013214111328125
    },
    {
      "epoch": 2.2477149963378905e-05,
      "model_forward_time": 1.738631248474121,
      "step": 97569
    },
    {
      "epoch": 2.2477149963378905e-05,
      "step": 97569,
      "training_step_time": 7.877930164337158
    },
    {
      "epoch": 2.2478103637695313e-05,
      "grad_norm": 0.09321611374616623,
      "learning_rate": 1.614836477456916e-08,
      "loss": 0.0561,
      "step": 97570
    },
    {
      "epoch": 2.2478103637695313e-05,
      "model_forward_time": 1.7339167594909668,
      "step": 97570
    },
    {
      "dynamics_loss_avg": 0.04478877782821655,
      "epoch": 2.2478103637695313e-05,
      "step": 97570
    },
    {
      "action_loss_avg": 0.0014724340057000519,
      "epoch": 2.2478103637695313e-05,
      "step": 97570
    },
    {
      "epoch": 2.2478103637695313e-05,
      "step": 97570,
      "training_step_time": 8.841628313064575
    },
    {
      "epoch": 2.247905731201172e-05,
      "model_forward_time": 1.733443260192871,
      "step": 97571
    },
    {
      "epoch": 2.247905731201172e-05,
      "step": 97571,
      "training_step_time": 8.74210500717163
    },
    {
      "epoch": 2.2480010986328124e-05,
      "model_forward_time": 1.7356915473937988,
      "step": 97572
    },
    {
      "epoch": 2.2480010986328124e-05,
      "step": 97572,
      "training_step_time": 9.609222173690796
    },
    {
      "epoch": 2.248096466064453e-05,
      "model_forward_time": 1.7395143508911133,
      "step": 97573
    },
    {
      "epoch": 2.248096466064453e-05,
      "step": 97573,
      "training_step_time": 7.716685056686401
    },
    {
      "epoch": 2.248191833496094e-05,
      "model_forward_time": 1.763742208480835,
      "step": 97574
    },
    {
      "epoch": 2.248191833496094e-05,
      "step": 97574,
      "training_step_time": 7.821620464324951
    },
    {
      "epoch": 2.2482872009277343e-05,
      "model_forward_time": 1.7535314559936523,
      "step": 97575
    },
    {
      "epoch": 2.2482872009277343e-05,
      "step": 97575,
      "training_step_time": 8.858492374420166
    },
    {
      "epoch": 2.248382568359375e-05,
      "model_forward_time": 2.39766263961792,
      "step": 97576
    },
    {
      "epoch": 2.248382568359375e-05,
      "step": 97576,
      "training_step_time": 7.76907753944397
    },
    {
      "epoch": 2.2484779357910158e-05,
      "model_forward_time": 1.7391357421875,
      "step": 97577
    },
    {
      "epoch": 2.2484779357910158e-05,
      "step": 97577,
      "training_step_time": 7.577533721923828
    },
    {
      "epoch": 2.2485733032226562e-05,
      "model_forward_time": 1.720381736755371,
      "step": 97578
    },
    {
      "epoch": 2.2485733032226562e-05,
      "step": 97578,
      "training_step_time": 7.826388597488403
    },
    {
      "epoch": 2.248668670654297e-05,
      "model_forward_time": 1.3482348918914795,
      "step": 97579
    },
    {
      "epoch": 2.248668670654297e-05,
      "step": 97579,
      "training_step_time": 8.558894634246826
    },
    {
      "epoch": 2.2487640380859373e-05,
      "grad_norm": 0.10799580812454224,
      "learning_rate": 1.6015855169050355e-08,
      "loss": 0.0566,
      "step": 97580
    },
    {
      "epoch": 2.2487640380859373e-05,
      "model_forward_time": 1.729140043258667,
      "step": 97580
    },
    {
      "dynamics_loss_avg": 0.05159416478127241,
      "epoch": 2.2487640380859373e-05,
      "step": 97580
    },
    {
      "action_loss_avg": 0.0017968961532460526,
      "epoch": 2.2487640380859373e-05,
      "step": 97580
    },
    {
      "epoch": 2.2487640380859373e-05,
      "step": 97580,
      "training_step_time": 8.7083740234375
    },
    {
      "epoch": 2.248859405517578e-05,
      "model_forward_time": 1.7338225841522217,
      "step": 97581
    },
    {
      "epoch": 2.248859405517578e-05,
      "step": 97581,
      "training_step_time": 8.82363510131836
    },
    {
      "epoch": 2.2489547729492188e-05,
      "model_forward_time": 1.7330718040466309,
      "step": 97582
    },
    {
      "epoch": 2.2489547729492188e-05,
      "step": 97582,
      "training_step_time": 8.253676414489746
    },
    {
      "epoch": 2.2490501403808592e-05,
      "model_forward_time": 1.751208782196045,
      "step": 97583
    },
    {
      "epoch": 2.2490501403808592e-05,
      "step": 97583,
      "training_step_time": 8.58077621459961
    },
    {
      "epoch": 2.2491455078125e-05,
      "model_forward_time": 1.7485435009002686,
      "step": 97584
    },
    {
      "epoch": 2.2491455078125e-05,
      "step": 97584,
      "training_step_time": 7.656207084655762
    },
    {
      "epoch": 2.2492408752441407e-05,
      "model_forward_time": 2.3933019638061523,
      "step": 97585
    },
    {
      "epoch": 2.2492408752441407e-05,
      "step": 97585,
      "training_step_time": 7.806978464126587
    },
    {
      "epoch": 2.249336242675781e-05,
      "model_forward_time": 1.7404110431671143,
      "step": 97586
    },
    {
      "epoch": 2.249336242675781e-05,
      "step": 97586,
      "training_step_time": 8.327372074127197
    },
    {
      "epoch": 2.249431610107422e-05,
      "model_forward_time": 1.7383677959442139,
      "step": 97587
    },
    {
      "epoch": 2.249431610107422e-05,
      "step": 97587,
      "training_step_time": 7.688691139221191
    },
    {
      "epoch": 2.2495269775390626e-05,
      "model_forward_time": 1.7392754554748535,
      "step": 97588
    },
    {
      "epoch": 2.2495269775390626e-05,
      "step": 97588,
      "training_step_time": 7.680790901184082
    },
    {
      "epoch": 2.249622344970703e-05,
      "model_forward_time": 1.7403011322021484,
      "step": 97589
    },
    {
      "epoch": 2.249622344970703e-05,
      "step": 97589,
      "training_step_time": 8.932247400283813
    },
    {
      "epoch": 2.2497177124023437e-05,
      "grad_norm": 0.11228617280721664,
      "learning_rate": 1.5883890604543872e-08,
      "loss": 0.0565,
      "step": 97590
    },
    {
      "epoch": 2.2497177124023437e-05,
      "model_forward_time": 1.7301719188690186,
      "step": 97590
    },
    {
      "dynamics_loss_avg": 0.049313966557383536,
      "epoch": 2.2497177124023437e-05,
      "step": 97590
    },
    {
      "action_loss_avg": 0.0014773515635170043,
      "epoch": 2.2497177124023437e-05,
      "step": 97590
    },
    {
      "epoch": 2.2497177124023437e-05,
      "step": 97590,
      "training_step_time": 9.039753437042236
    },
    {
      "epoch": 2.2498130798339845e-05,
      "model_forward_time": 1.7385683059692383,
      "step": 97591
    },
    {
      "epoch": 2.2498130798339845e-05,
      "step": 97591,
      "training_step_time": 8.18037748336792
    },
    {
      "epoch": 2.249908447265625e-05,
      "model_forward_time": 1.7568457126617432,
      "step": 97592
    },
    {
      "epoch": 2.249908447265625e-05,
      "step": 97592,
      "training_step_time": 7.580859184265137
    },
    {
      "epoch": 2.2500038146972656e-05,
      "model_forward_time": 1.7534141540527344,
      "step": 97593
    },
    {
      "epoch": 2.2500038146972656e-05,
      "step": 97593,
      "training_step_time": 7.9095540046691895
    },
    {
      "epoch": 2.2500991821289064e-05,
      "model_forward_time": 1.7372736930847168,
      "step": 97594
    },
    {
      "epoch": 2.2500991821289064e-05,
      "step": 97594,
      "training_step_time": 8.276707649230957
    },
    {
      "epoch": 2.2501945495605468e-05,
      "model_forward_time": 1.7039668560028076,
      "step": 97595
    },
    {
      "epoch": 2.2501945495605468e-05,
      "step": 97595,
      "training_step_time": 7.6854798793792725
    },
    {
      "epoch": 2.2502899169921875e-05,
      "model_forward_time": 1.7135050296783447,
      "step": 97596
    },
    {
      "epoch": 2.2502899169921875e-05,
      "step": 97596,
      "training_step_time": 7.749793767929077
    },
    {
      "epoch": 2.2503852844238282e-05,
      "model_forward_time": 1.7190287113189697,
      "step": 97597
    },
    {
      "epoch": 2.2503852844238282e-05,
      "step": 97597,
      "training_step_time": 9.349697828292847
    },
    {
      "epoch": 2.2504806518554687e-05,
      "model_forward_time": 1.716261386871338,
      "step": 97598
    },
    {
      "epoch": 2.2504806518554687e-05,
      "step": 97598,
      "training_step_time": 7.947850227355957
    },
    {
      "epoch": 2.2505760192871094e-05,
      "model_forward_time": 1.7195653915405273,
      "step": 97599
    },
    {
      "epoch": 2.2505760192871094e-05,
      "step": 97599,
      "training_step_time": 9.360709190368652
    },
    {
      "epoch": 2.25067138671875e-05,
      "grad_norm": 0.09949733316898346,
      "learning_rate": 1.575247109548206e-08,
      "loss": 0.0573,
      "step": 97600
    },
    {
      "epoch": 2.25067138671875e-05,
      "model_forward_time": 1.7161102294921875,
      "step": 97600
    },
    {
      "dynamics_loss_avg": 0.06055520176887512,
      "epoch": 2.25067138671875e-05,
      "step": 97600
    },
    {
      "action_loss_avg": 0.001256589192780666,
      "epoch": 2.25067138671875e-05,
      "step": 97600
    },
    {
      "epoch": 2.25067138671875e-05,
      "step": 97600,
      "training_step_time": 7.9285666942596436
    },
    {
      "epoch": 2.2507667541503905e-05,
      "model_forward_time": 1.7293987274169922,
      "step": 97601
    },
    {
      "epoch": 2.2507667541503905e-05,
      "step": 97601,
      "training_step_time": 9.197613954544067
    },
    {
      "epoch": 2.2508621215820313e-05,
      "model_forward_time": 1.7176647186279297,
      "step": 97602
    },
    {
      "epoch": 2.2508621215820313e-05,
      "step": 97602,
      "training_step_time": 8.478616952896118
    },
    {
      "epoch": 2.250957489013672e-05,
      "model_forward_time": 1.7419061660766602,
      "step": 97603
    },
    {
      "epoch": 2.250957489013672e-05,
      "step": 97603,
      "training_step_time": 8.636929988861084
    },
    {
      "epoch": 2.2510528564453124e-05,
      "model_forward_time": 2.321904420852661,
      "step": 97604
    },
    {
      "epoch": 2.2510528564453124e-05,
      "step": 97604,
      "training_step_time": 7.868202209472656
    },
    {
      "epoch": 2.251148223876953e-05,
      "model_forward_time": 1.7157773971557617,
      "step": 97605
    },
    {
      "epoch": 2.251148223876953e-05,
      "step": 97605,
      "training_step_time": 8.567552089691162
    },
    {
      "epoch": 2.251243591308594e-05,
      "model_forward_time": 1.7205495834350586,
      "step": 97606
    },
    {
      "epoch": 2.251243591308594e-05,
      "step": 97606,
      "training_step_time": 7.835257530212402
    },
    {
      "epoch": 2.2513389587402343e-05,
      "model_forward_time": 1.7208659648895264,
      "step": 97607
    },
    {
      "epoch": 2.2513389587402343e-05,
      "step": 97607,
      "training_step_time": 9.043928623199463
    },
    {
      "epoch": 2.251434326171875e-05,
      "model_forward_time": 1.7157671451568604,
      "step": 97608
    },
    {
      "epoch": 2.251434326171875e-05,
      "step": 97608,
      "training_step_time": 7.863889932632446
    },
    {
      "epoch": 2.2515296936035158e-05,
      "model_forward_time": 1.718313217163086,
      "step": 97609
    },
    {
      "epoch": 2.2515296936035158e-05,
      "step": 97609,
      "training_step_time": 8.939175128936768
    },
    {
      "epoch": 2.2516250610351562e-05,
      "grad_norm": 0.11663883179426193,
      "learning_rate": 1.562159665623564e-08,
      "loss": 0.0558,
      "step": 97610
    },
    {
      "epoch": 2.2516250610351562e-05,
      "model_forward_time": 1.7298166751861572,
      "step": 97610
    },
    {
      "dynamics_loss_avg": 0.05867219399660826,
      "epoch": 2.2516250610351562e-05,
      "step": 97610
    },
    {
      "action_loss_avg": 0.001183354199747555,
      "epoch": 2.2516250610351562e-05,
      "step": 97610
    },
    {
      "epoch": 2.2516250610351562e-05,
      "step": 97610,
      "training_step_time": 7.7627339363098145
    },
    {
      "epoch": 2.251720428466797e-05,
      "model_forward_time": 1.719696283340454,
      "step": 97611
    },
    {
      "epoch": 2.251720428466797e-05,
      "step": 97611,
      "training_step_time": 7.915257215499878
    },
    {
      "epoch": 2.2518157958984373e-05,
      "model_forward_time": 1.7456424236297607,
      "step": 97612
    },
    {
      "epoch": 2.2518157958984373e-05,
      "step": 97612,
      "training_step_time": 8.663613080978394
    },
    {
      "epoch": 2.251911163330078e-05,
      "model_forward_time": 2.364494562149048,
      "step": 97613
    },
    {
      "epoch": 2.251911163330078e-05,
      "step": 97613,
      "training_step_time": 7.8421311378479
    },
    {
      "epoch": 2.2520065307617188e-05,
      "model_forward_time": 1.7169902324676514,
      "step": 97614
    },
    {
      "epoch": 2.2520065307617188e-05,
      "step": 97614,
      "training_step_time": 7.648249387741089
    },
    {
      "epoch": 2.2521018981933592e-05,
      "model_forward_time": 1.7169792652130127,
      "step": 97615
    },
    {
      "epoch": 2.2521018981933592e-05,
      "step": 97615,
      "training_step_time": 8.784292697906494
    },
    {
      "epoch": 2.252197265625e-05,
      "model_forward_time": 1.715935468673706,
      "step": 97616
    },
    {
      "epoch": 2.252197265625e-05,
      "step": 97616,
      "training_step_time": 7.904067516326904
    },
    {
      "epoch": 2.2522926330566407e-05,
      "model_forward_time": 1.715101957321167,
      "step": 97617
    },
    {
      "epoch": 2.2522926330566407e-05,
      "step": 97617,
      "training_step_time": 8.839258193969727
    },
    {
      "epoch": 2.252388000488281e-05,
      "model_forward_time": 1.7221834659576416,
      "step": 97618
    },
    {
      "epoch": 2.252388000488281e-05,
      "step": 97618,
      "training_step_time": 9.995067358016968
    },
    {
      "epoch": 2.252483367919922e-05,
      "model_forward_time": 1.732698917388916,
      "step": 97619
    },
    {
      "epoch": 2.252483367919922e-05,
      "step": 97619,
      "training_step_time": 7.811031341552734
    },
    {
      "epoch": 2.2525787353515626e-05,
      "grad_norm": 0.07789941132068634,
      "learning_rate": 1.5491267301117607e-08,
      "loss": 0.0577,
      "step": 97620
    },
    {
      "epoch": 2.2525787353515626e-05,
      "model_forward_time": 1.7128803730010986,
      "step": 97620
    },
    {
      "dynamics_loss_avg": 0.0501522890292108,
      "epoch": 2.2525787353515626e-05,
      "step": 97620
    },
    {
      "action_loss_avg": 0.001002255061757751,
      "epoch": 2.2525787353515626e-05,
      "step": 97620
    },
    {
      "epoch": 2.2525787353515626e-05,
      "step": 97620,
      "training_step_time": 8.470575332641602
    },
    {
      "epoch": 2.252674102783203e-05,
      "model_forward_time": 1.7269952297210693,
      "step": 97621
    },
    {
      "epoch": 2.252674102783203e-05,
      "step": 97621,
      "training_step_time": 7.780974864959717
    },
    {
      "epoch": 2.2527694702148437e-05,
      "model_forward_time": 2.3478736877441406,
      "step": 97622
    },
    {
      "epoch": 2.2527694702148437e-05,
      "step": 97622,
      "training_step_time": 9.824239730834961
    },
    {
      "epoch": 2.2528648376464845e-05,
      "model_forward_time": 1.714815378189087,
      "step": 97623
    },
    {
      "epoch": 2.2528648376464845e-05,
      "step": 97623,
      "training_step_time": 8.12906265258789
    },
    {
      "epoch": 2.252960205078125e-05,
      "model_forward_time": 1.7187433242797852,
      "step": 97624
    },
    {
      "epoch": 2.252960205078125e-05,
      "step": 97624,
      "training_step_time": 8.316917657852173
    },
    {
      "epoch": 2.2530555725097656e-05,
      "model_forward_time": 1.7155065536499023,
      "step": 97625
    },
    {
      "epoch": 2.2530555725097656e-05,
      "step": 97625,
      "training_step_time": 8.757169961929321
    },
    {
      "epoch": 2.2531509399414064e-05,
      "model_forward_time": 1.7132599353790283,
      "step": 97626
    },
    {
      "epoch": 2.2531509399414064e-05,
      "step": 97626,
      "training_step_time": 9.994986772537231
    },
    {
      "epoch": 2.2532463073730468e-05,
      "model_forward_time": 1.7187426090240479,
      "step": 97627
    },
    {
      "epoch": 2.2532463073730468e-05,
      "step": 97627,
      "training_step_time": 8.020798444747925
    },
    {
      "epoch": 2.2533416748046875e-05,
      "model_forward_time": 1.7238268852233887,
      "step": 97628
    },
    {
      "epoch": 2.2533416748046875e-05,
      "step": 97628,
      "training_step_time": 7.709554672241211
    },
    {
      "epoch": 2.2534370422363283e-05,
      "model_forward_time": 1.7232599258422852,
      "step": 97629
    },
    {
      "epoch": 2.2534370422363283e-05,
      "step": 97629,
      "training_step_time": 9.627892255783081
    },
    {
      "epoch": 2.2535324096679687e-05,
      "grad_norm": 0.08504749089479446,
      "learning_rate": 1.536148304438101e-08,
      "loss": 0.0567,
      "step": 97630
    },
    {
      "epoch": 2.2535324096679687e-05,
      "model_forward_time": 1.724593162536621,
      "step": 97630
    },
    {
      "dynamics_loss_avg": 0.041344988718628885,
      "epoch": 2.2535324096679687e-05,
      "step": 97630
    },
    {
      "action_loss_avg": 0.001105753186857328,
      "epoch": 2.2535324096679687e-05,
      "step": 97630
    },
    {
      "epoch": 2.2535324096679687e-05,
      "step": 97630,
      "training_step_time": 7.775069952011108
    },
    {
      "epoch": 2.2536277770996094e-05,
      "model_forward_time": 2.363481044769287,
      "step": 97631
    },
    {
      "epoch": 2.2536277770996094e-05,
      "step": 97631,
      "training_step_time": 7.71809458732605
    },
    {
      "epoch": 2.25372314453125e-05,
      "model_forward_time": 1.721179723739624,
      "step": 97632
    },
    {
      "epoch": 2.25372314453125e-05,
      "step": 97632,
      "training_step_time": 7.753003835678101
    },
    {
      "epoch": 2.2538185119628905e-05,
      "model_forward_time": 1.7199819087982178,
      "step": 97633
    },
    {
      "epoch": 2.2538185119628905e-05,
      "step": 97633,
      "training_step_time": 7.769949197769165
    },
    {
      "epoch": 2.2539138793945313e-05,
      "model_forward_time": 1.7150044441223145,
      "step": 97634
    },
    {
      "epoch": 2.2539138793945313e-05,
      "step": 97634,
      "training_step_time": 9.708230972290039
    },
    {
      "epoch": 2.254009246826172e-05,
      "model_forward_time": 1.7121024131774902,
      "step": 97635
    },
    {
      "epoch": 2.254009246826172e-05,
      "step": 97635,
      "training_step_time": 8.654646873474121
    },
    {
      "epoch": 2.2541046142578124e-05,
      "model_forward_time": 1.7029922008514404,
      "step": 97636
    },
    {
      "epoch": 2.2541046142578124e-05,
      "step": 97636,
      "training_step_time": 9.427503824234009
    },
    {
      "epoch": 2.254199981689453e-05,
      "model_forward_time": 1.70896315574646,
      "step": 97637
    },
    {
      "epoch": 2.254199981689453e-05,
      "step": 97637,
      "training_step_time": 9.036059379577637
    },
    {
      "epoch": 2.254295349121094e-05,
      "model_forward_time": 1.7263600826263428,
      "step": 97638
    },
    {
      "epoch": 2.254295349121094e-05,
      "step": 97638,
      "training_step_time": 7.86478853225708
    },
    {
      "epoch": 2.2543907165527343e-05,
      "model_forward_time": 1.722050428390503,
      "step": 97639
    },
    {
      "epoch": 2.2543907165527343e-05,
      "step": 97639,
      "training_step_time": 7.869160175323486
    },
    {
      "epoch": 2.254486083984375e-05,
      "grad_norm": 0.13555730879306793,
      "learning_rate": 1.5232243900217823e-08,
      "loss": 0.0567,
      "step": 97640
    },
    {
      "epoch": 2.254486083984375e-05,
      "model_forward_time": 2.35225248336792,
      "step": 97640
    },
    {
      "dynamics_loss_avg": 0.05312270242720842,
      "epoch": 2.254486083984375e-05,
      "step": 97640
    },
    {
      "action_loss_avg": 0.0012712370400549844,
      "epoch": 2.254486083984375e-05,
      "step": 97640
    },
    {
      "epoch": 2.254486083984375e-05,
      "step": 97640,
      "training_step_time": 7.76144814491272
    },
    {
      "epoch": 2.2545814514160155e-05,
      "model_forward_time": 1.709620475769043,
      "step": 97641
    },
    {
      "epoch": 2.2545814514160155e-05,
      "step": 97641,
      "training_step_time": 7.865749835968018
    },
    {
      "epoch": 2.2546768188476562e-05,
      "model_forward_time": 1.7160718441009521,
      "step": 97642
    },
    {
      "epoch": 2.2546768188476562e-05,
      "step": 97642,
      "training_step_time": 7.710427761077881
    },
    {
      "epoch": 2.254772186279297e-05,
      "model_forward_time": 1.7147464752197266,
      "step": 97643
    },
    {
      "epoch": 2.254772186279297e-05,
      "step": 97643,
      "training_step_time": 7.832162857055664
    },
    {
      "epoch": 2.2548675537109373e-05,
      "model_forward_time": 1.7167046070098877,
      "step": 97644
    },
    {
      "epoch": 2.2548675537109373e-05,
      "step": 97644,
      "training_step_time": 7.900696039199829
    },
    {
      "epoch": 2.254962921142578e-05,
      "model_forward_time": 1.7210617065429688,
      "step": 97645
    },
    {
      "epoch": 2.254962921142578e-05,
      "step": 97645,
      "training_step_time": 7.79791784286499
    },
    {
      "epoch": 2.2550582885742188e-05,
      "model_forward_time": 1.7124767303466797,
      "step": 97646
    },
    {
      "epoch": 2.2550582885742188e-05,
      "step": 97646,
      "training_step_time": 8.040197134017944
    },
    {
      "epoch": 2.2551536560058592e-05,
      "model_forward_time": 1.729095458984375,
      "step": 97647
    },
    {
      "epoch": 2.2551536560058592e-05,
      "step": 97647,
      "training_step_time": 7.755431890487671
    },
    {
      "epoch": 2.2552490234375e-05,
      "model_forward_time": 1.7167530059814453,
      "step": 97648
    },
    {
      "epoch": 2.2552490234375e-05,
      "step": 97648,
      "training_step_time": 7.846372365951538
    },
    {
      "epoch": 2.2553443908691407e-05,
      "model_forward_time": 1.7124457359313965,
      "step": 97649
    },
    {
      "epoch": 2.2553443908691407e-05,
      "step": 97649,
      "training_step_time": 8.3394033908844
    },
    {
      "epoch": 2.255439758300781e-05,
      "grad_norm": 0.16285200417041779,
      "learning_rate": 1.5103549882762302e-08,
      "loss": 0.0564,
      "step": 97650
    },
    {
      "epoch": 2.255439758300781e-05,
      "model_forward_time": 1.7048008441925049,
      "step": 97650
    },
    {
      "dynamics_loss_avg": 0.055770380422472954,
      "epoch": 2.255439758300781e-05,
      "step": 97650
    },
    {
      "action_loss_avg": 0.0014717380865477026,
      "epoch": 2.255439758300781e-05,
      "step": 97650
    },
    {
      "epoch": 2.255439758300781e-05,
      "step": 97650,
      "training_step_time": 7.938183307647705
    },
    {
      "epoch": 2.255535125732422e-05,
      "model_forward_time": 1.713547706604004,
      "step": 97651
    },
    {
      "epoch": 2.255535125732422e-05,
      "step": 97651,
      "training_step_time": 8.724623441696167
    },
    {
      "epoch": 2.2556304931640626e-05,
      "model_forward_time": 1.710533857345581,
      "step": 97652
    },
    {
      "epoch": 2.2556304931640626e-05,
      "step": 97652,
      "training_step_time": 8.555112600326538
    },
    {
      "epoch": 2.255725860595703e-05,
      "model_forward_time": 1.7141468524932861,
      "step": 97653
    },
    {
      "epoch": 2.255725860595703e-05,
      "step": 97653,
      "training_step_time": 9.076417446136475
    },
    {
      "epoch": 2.2558212280273437e-05,
      "model_forward_time": 1.7118921279907227,
      "step": 97654
    },
    {
      "epoch": 2.2558212280273437e-05,
      "step": 97654,
      "training_step_time": 8.86509084701538
    },
    {
      "epoch": 2.2559165954589845e-05,
      "model_forward_time": 1.7103009223937988,
      "step": 97655
    },
    {
      "epoch": 2.2559165954589845e-05,
      "step": 97655,
      "training_step_time": 8.103341817855835
    },
    {
      "epoch": 2.256011962890625e-05,
      "model_forward_time": 1.7333605289459229,
      "step": 97656
    },
    {
      "epoch": 2.256011962890625e-05,
      "step": 97656,
      "training_step_time": 7.822923421859741
    },
    {
      "epoch": 2.2561073303222656e-05,
      "model_forward_time": 1.7117986679077148,
      "step": 97657
    },
    {
      "epoch": 2.2561073303222656e-05,
      "step": 97657,
      "training_step_time": 7.868013620376587
    },
    {
      "epoch": 2.2562026977539064e-05,
      "model_forward_time": 1.7374181747436523,
      "step": 97658
    },
    {
      "epoch": 2.2562026977539064e-05,
      "step": 97658,
      "training_step_time": 7.78952693939209
    },
    {
      "epoch": 2.2562980651855468e-05,
      "model_forward_time": 2.345776081085205,
      "step": 97659
    },
    {
      "epoch": 2.2562980651855468e-05,
      "step": 97659,
      "training_step_time": 7.914336204528809
    },
    {
      "epoch": 2.2563934326171875e-05,
      "grad_norm": 0.07890400290489197,
      "learning_rate": 1.497540100608763e-08,
      "loss": 0.0559,
      "step": 97660
    },
    {
      "epoch": 2.2563934326171875e-05,
      "model_forward_time": 1.7146670818328857,
      "step": 97660
    },
    {
      "dynamics_loss_avg": 0.05468642245978117,
      "epoch": 2.2563934326171875e-05,
      "step": 97660
    },
    {
      "action_loss_avg": 0.0017356163880322129,
      "epoch": 2.2563934326171875e-05,
      "step": 97660
    },
    {
      "epoch": 2.2563934326171875e-05,
      "step": 97660,
      "training_step_time": 7.657434701919556
    },
    {
      "epoch": 2.2564888000488283e-05,
      "model_forward_time": 1.331989049911499,
      "step": 97661
    },
    {
      "epoch": 2.2564888000488283e-05,
      "step": 97661,
      "training_step_time": 7.951688289642334
    },
    {
      "epoch": 2.2565841674804687e-05,
      "model_forward_time": 1.7130727767944336,
      "step": 97662
    },
    {
      "epoch": 2.2565841674804687e-05,
      "step": 97662,
      "training_step_time": 7.971025466918945
    },
    {
      "epoch": 2.2566795349121094e-05,
      "model_forward_time": 1.714529275894165,
      "step": 97663
    },
    {
      "epoch": 2.2566795349121094e-05,
      "step": 97663,
      "training_step_time": 8.71797776222229
    },
    {
      "epoch": 2.25677490234375e-05,
      "model_forward_time": 1.7126448154449463,
      "step": 97664
    },
    {
      "epoch": 2.25677490234375e-05,
      "step": 97664,
      "training_step_time": 8.068931818008423
    },
    {
      "epoch": 2.2568702697753905e-05,
      "model_forward_time": 1.7243432998657227,
      "step": 97665
    },
    {
      "epoch": 2.2568702697753905e-05,
      "step": 97665,
      "training_step_time": 8.447353601455688
    },
    {
      "epoch": 2.2569656372070313e-05,
      "model_forward_time": 1.7134826183319092,
      "step": 97666
    },
    {
      "epoch": 2.2569656372070313e-05,
      "step": 97666,
      "training_step_time": 7.8147571086883545
    },
    {
      "epoch": 2.257061004638672e-05,
      "model_forward_time": 1.7276453971862793,
      "step": 97667
    },
    {
      "epoch": 2.257061004638672e-05,
      "step": 97667,
      "training_step_time": 8.313018798828125
    },
    {
      "epoch": 2.2571563720703124e-05,
      "model_forward_time": 2.3378102779388428,
      "step": 97668
    },
    {
      "epoch": 2.2571563720703124e-05,
      "step": 97668,
      "training_step_time": 7.894286632537842
    },
    {
      "epoch": 2.2572517395019532e-05,
      "model_forward_time": 1.7102000713348389,
      "step": 97669
    },
    {
      "epoch": 2.2572517395019532e-05,
      "step": 97669,
      "training_step_time": 8.153010845184326
    },
    {
      "epoch": 2.257347106933594e-05,
      "grad_norm": 0.07481789588928223,
      "learning_rate": 1.4847797284208154e-08,
      "loss": 0.0564,
      "step": 97670
    },
    {
      "epoch": 2.257347106933594e-05,
      "model_forward_time": 1.7071843147277832,
      "step": 97670
    },
    {
      "dynamics_loss_avg": 0.03587471414357424,
      "epoch": 2.257347106933594e-05,
      "step": 97670
    },
    {
      "action_loss_avg": 0.0012666345399338752,
      "epoch": 2.257347106933594e-05,
      "step": 97670
    },
    {
      "epoch": 2.257347106933594e-05,
      "step": 97670,
      "training_step_time": 8.622894048690796
    },
    {
      "epoch": 2.2574424743652343e-05,
      "model_forward_time": 1.7102701663970947,
      "step": 97671
    },
    {
      "epoch": 2.2574424743652343e-05,
      "step": 97671,
      "training_step_time": 8.612849950790405
    },
    {
      "epoch": 2.257537841796875e-05,
      "model_forward_time": 1.7104921340942383,
      "step": 97672
    },
    {
      "epoch": 2.257537841796875e-05,
      "step": 97672,
      "training_step_time": 8.688863754272461
    },
    {
      "epoch": 2.2576332092285155e-05,
      "model_forward_time": 1.7098634243011475,
      "step": 97673
    },
    {
      "epoch": 2.2576332092285155e-05,
      "step": 97673,
      "training_step_time": 8.682903528213501
    },
    {
      "epoch": 2.2577285766601562e-05,
      "model_forward_time": 1.714055061340332,
      "step": 97674
    },
    {
      "epoch": 2.2577285766601562e-05,
      "step": 97674,
      "training_step_time": 7.833480358123779
    },
    {
      "epoch": 2.257823944091797e-05,
      "model_forward_time": 1.70054292678833,
      "step": 97675
    },
    {
      "epoch": 2.257823944091797e-05,
      "step": 97675,
      "training_step_time": 7.881687879562378
    },
    {
      "epoch": 2.2579193115234373e-05,
      "model_forward_time": 1.7279915809631348,
      "step": 97676
    },
    {
      "epoch": 2.2579193115234373e-05,
      "step": 97676,
      "training_step_time": 8.313055992126465
    },
    {
      "epoch": 2.258014678955078e-05,
      "model_forward_time": 2.34694766998291,
      "step": 97677
    },
    {
      "epoch": 2.258014678955078e-05,
      "step": 97677,
      "training_step_time": 7.905812978744507
    },
    {
      "epoch": 2.2581100463867188e-05,
      "model_forward_time": 1.7124876976013184,
      "step": 97678
    },
    {
      "epoch": 2.2581100463867188e-05,
      "step": 97678,
      "training_step_time": 7.856446027755737
    },
    {
      "epoch": 2.2582054138183592e-05,
      "model_forward_time": 1.7124381065368652,
      "step": 97679
    },
    {
      "epoch": 2.2582054138183592e-05,
      "step": 97679,
      "training_step_time": 7.928178071975708
    },
    {
      "epoch": 2.25830078125e-05,
      "grad_norm": 0.22960440814495087,
      "learning_rate": 1.4720738731078821e-08,
      "loss": 0.0566,
      "step": 97680
    },
    {
      "epoch": 2.25830078125e-05,
      "model_forward_time": 1.7098968029022217,
      "step": 97680
    },
    {
      "dynamics_loss_avg": 0.056366577930748465,
      "epoch": 2.25830078125e-05,
      "step": 97680
    },
    {
      "action_loss_avg": 0.0013982753996970132,
      "epoch": 2.25830078125e-05,
      "step": 97680
    },
    {
      "epoch": 2.25830078125e-05,
      "step": 97680,
      "training_step_time": 7.8782079219818115
    },
    {
      "epoch": 2.2583961486816407e-05,
      "model_forward_time": 1.711003303527832,
      "step": 97681
    },
    {
      "epoch": 2.2583961486816407e-05,
      "step": 97681,
      "training_step_time": 7.895492792129517
    },
    {
      "epoch": 2.258491516113281e-05,
      "model_forward_time": 1.7093379497528076,
      "step": 97682
    },
    {
      "epoch": 2.258491516113281e-05,
      "step": 97682,
      "training_step_time": 8.75039029121399
    },
    {
      "epoch": 2.258586883544922e-05,
      "model_forward_time": 1.707284688949585,
      "step": 97683
    },
    {
      "epoch": 2.258586883544922e-05,
      "step": 97683,
      "training_step_time": 11.867615699768066
    },
    {
      "epoch": 2.2586822509765626e-05,
      "model_forward_time": 1.7088203430175781,
      "step": 97684
    },
    {
      "epoch": 2.2586822509765626e-05,
      "step": 97684,
      "training_step_time": 7.766981601715088
    },
    {
      "epoch": 2.258777618408203e-05,
      "model_forward_time": 1.717498540878296,
      "step": 97685
    },
    {
      "epoch": 2.258777618408203e-05,
      "step": 97685,
      "training_step_time": 7.862605810165405
    },
    {
      "epoch": 2.2588729858398437e-05,
      "model_forward_time": 2.352686882019043,
      "step": 97686
    },
    {
      "epoch": 2.2588729858398437e-05,
      "step": 97686,
      "training_step_time": 8.65068507194519
    },
    {
      "epoch": 2.2589683532714845e-05,
      "model_forward_time": 1.710038661956787,
      "step": 97687
    },
    {
      "epoch": 2.2589683532714845e-05,
      "step": 97687,
      "training_step_time": 7.873391151428223
    },
    {
      "epoch": 2.259063720703125e-05,
      "model_forward_time": 1.7115592956542969,
      "step": 97688
    },
    {
      "epoch": 2.259063720703125e-05,
      "step": 97688,
      "training_step_time": 7.862796068191528
    },
    {
      "epoch": 2.2591590881347656e-05,
      "model_forward_time": 1.7107789516448975,
      "step": 97689
    },
    {
      "epoch": 2.2591590881347656e-05,
      "step": 97689,
      "training_step_time": 9.617578983306885
    },
    {
      "epoch": 2.2592544555664064e-05,
      "grad_norm": 0.08851935714483261,
      "learning_rate": 1.4594225360594072e-08,
      "loss": 0.0553,
      "step": 97690
    },
    {
      "epoch": 2.2592544555664064e-05,
      "model_forward_time": 1.7087252140045166,
      "step": 97690
    },
    {
      "dynamics_loss_avg": 0.06027627978473902,
      "epoch": 2.2592544555664064e-05,
      "step": 97690
    },
    {
      "action_loss_avg": 0.0009523156099021435,
      "epoch": 2.2592544555664064e-05,
      "step": 97690
    },
    {
      "epoch": 2.2592544555664064e-05,
      "step": 97690,
      "training_step_time": 9.060513496398926
    },
    {
      "epoch": 2.2593498229980468e-05,
      "model_forward_time": 1.7175986766815186,
      "step": 97691
    },
    {
      "epoch": 2.2593498229980468e-05,
      "step": 97691,
      "training_step_time": 8.569699764251709
    },
    {
      "epoch": 2.2594451904296875e-05,
      "model_forward_time": 1.713860034942627,
      "step": 97692
    },
    {
      "epoch": 2.2594451904296875e-05,
      "step": 97692,
      "training_step_time": 8.966010808944702
    },
    {
      "epoch": 2.2595405578613283e-05,
      "model_forward_time": 1.7407712936401367,
      "step": 97693
    },
    {
      "epoch": 2.2595405578613283e-05,
      "step": 97693,
      "training_step_time": 7.773895263671875
    },
    {
      "epoch": 2.2596359252929687e-05,
      "model_forward_time": 1.7237136363983154,
      "step": 97694
    },
    {
      "epoch": 2.2596359252929687e-05,
      "step": 97694,
      "training_step_time": 8.520345687866211
    },
    {
      "epoch": 2.2597312927246094e-05,
      "model_forward_time": 2.3652243614196777,
      "step": 97695
    },
    {
      "epoch": 2.2597312927246094e-05,
      "step": 97695,
      "training_step_time": 7.7740278244018555
    },
    {
      "epoch": 2.25982666015625e-05,
      "model_forward_time": 1.7178010940551758,
      "step": 97696
    },
    {
      "epoch": 2.25982666015625e-05,
      "step": 97696,
      "training_step_time": 11.478180646896362
    },
    {
      "epoch": 2.2599220275878905e-05,
      "model_forward_time": 1.716585636138916,
      "step": 97697
    },
    {
      "epoch": 2.2599220275878905e-05,
      "step": 97697,
      "training_step_time": 9.449801206588745
    },
    {
      "epoch": 2.2600173950195313e-05,
      "model_forward_time": 1.717292070388794,
      "step": 97698
    },
    {
      "epoch": 2.2600173950195313e-05,
      "step": 97698,
      "training_step_time": 6.91817307472229
    },
    {
      "epoch": 2.260112762451172e-05,
      "model_forward_time": 1.717165470123291,
      "step": 97699
    },
    {
      "epoch": 2.260112762451172e-05,
      "step": 97699,
      "training_step_time": 7.409042119979858
    },
    {
      "epoch": 2.2602081298828124e-05,
      "grad_norm": 0.07944847643375397,
      "learning_rate": 1.4468257186589507e-08,
      "loss": 0.0553,
      "step": 97700
    },
    {
      "epoch": 2.2602081298828124e-05,
      "model_forward_time": 1.709909439086914,
      "step": 97700
    },
    {
      "dynamics_loss_avg": 0.05037933085113764,
      "epoch": 2.2602081298828124e-05,
      "step": 97700
    },
    {
      "action_loss_avg": 0.0012714974727714434,
      "epoch": 2.2602081298828124e-05,
      "step": 97700
    },
    {
      "epoch": 2.2602081298828124e-05,
      "step": 97700,
      "training_step_time": 7.7894909381866455
    },
    {
      "epoch": 2.2603034973144532e-05,
      "model_forward_time": 1.712341070175171,
      "step": 97701
    },
    {
      "epoch": 2.2603034973144532e-05,
      "step": 97701,
      "training_step_time": 6.894882917404175
    },
    {
      "epoch": 2.260398864746094e-05,
      "model_forward_time": 1.7122609615325928,
      "step": 97702
    },
    {
      "epoch": 2.260398864746094e-05,
      "step": 97702,
      "training_step_time": 6.845104455947876
    },
    {
      "epoch": 2.2604942321777343e-05,
      "model_forward_time": 1.7124152183532715,
      "step": 97703
    },
    {
      "epoch": 2.2604942321777343e-05,
      "step": 97703,
      "training_step_time": 6.808104991912842
    },
    {
      "epoch": 2.260589599609375e-05,
      "model_forward_time": 1.7160935401916504,
      "step": 97704
    },
    {
      "epoch": 2.260589599609375e-05,
      "step": 97704,
      "training_step_time": 6.91011905670166
    },
    {
      "epoch": 2.2606849670410155e-05,
      "model_forward_time": 1.7125799655914307,
      "step": 97705
    },
    {
      "epoch": 2.2606849670410155e-05,
      "step": 97705,
      "training_step_time": 6.891160249710083
    },
    {
      "epoch": 2.2607803344726562e-05,
      "model_forward_time": 2.3592753410339355,
      "step": 97706
    },
    {
      "epoch": 2.2607803344726562e-05,
      "step": 97706,
      "training_step_time": 8.702112674713135
    },
    {
      "epoch": 2.260875701904297e-05,
      "model_forward_time": 1.7102315425872803,
      "step": 97707
    },
    {
      "epoch": 2.260875701904297e-05,
      "step": 97707,
      "training_step_time": 7.808538198471069
    },
    {
      "epoch": 2.2609710693359373e-05,
      "model_forward_time": 1.7121212482452393,
      "step": 97708
    },
    {
      "epoch": 2.2609710693359373e-05,
      "step": 97708,
      "training_step_time": 7.988713026046753
    },
    {
      "epoch": 2.261066436767578e-05,
      "model_forward_time": 1.7176287174224854,
      "step": 97709
    },
    {
      "epoch": 2.261066436767578e-05,
      "step": 97709,
      "training_step_time": 7.599583864212036
    },
    {
      "epoch": 2.261161804199219e-05,
      "grad_norm": 0.1647944152355194,
      "learning_rate": 1.4342834222840218e-08,
      "loss": 0.055,
      "step": 97710
    },
    {
      "epoch": 2.261161804199219e-05,
      "model_forward_time": 1.710747241973877,
      "step": 97710
    },
    {
      "dynamics_loss_avg": 0.05504600014537573,
      "epoch": 2.261161804199219e-05,
      "step": 97710
    },
    {
      "action_loss_avg": 0.001129227428464219,
      "epoch": 2.261161804199219e-05,
      "step": 97710
    },
    {
      "epoch": 2.261161804199219e-05,
      "step": 97710,
      "training_step_time": 6.881868839263916
    },
    {
      "epoch": 2.2612571716308592e-05,
      "model_forward_time": 1.7157511711120605,
      "step": 97711
    },
    {
      "epoch": 2.2612571716308592e-05,
      "step": 97711,
      "training_step_time": 6.920684099197388
    },
    {
      "epoch": 2.2613525390625e-05,
      "model_forward_time": 1.7170100212097168,
      "step": 97712
    },
    {
      "epoch": 2.2613525390625e-05,
      "step": 97712,
      "training_step_time": 6.937171936035156
    },
    {
      "epoch": 2.2614479064941407e-05,
      "model_forward_time": 1.715240478515625,
      "step": 97713
    },
    {
      "epoch": 2.2614479064941407e-05,
      "step": 97713,
      "training_step_time": 6.8898656368255615
    },
    {
      "epoch": 2.261543273925781e-05,
      "model_forward_time": 1.7193238735198975,
      "step": 97714
    },
    {
      "epoch": 2.261543273925781e-05,
      "step": 97714,
      "training_step_time": 6.890079975128174
    },
    {
      "epoch": 2.261638641357422e-05,
      "model_forward_time": 2.3752124309539795,
      "step": 97715
    },
    {
      "epoch": 2.261638641357422e-05,
      "step": 97715,
      "training_step_time": 7.785107374191284
    },
    {
      "epoch": 2.2617340087890626e-05,
      "model_forward_time": 1.7145390510559082,
      "step": 97716
    },
    {
      "epoch": 2.2617340087890626e-05,
      "step": 97716,
      "training_step_time": 7.669536113739014
    },
    {
      "epoch": 2.261829376220703e-05,
      "model_forward_time": 1.7129111289978027,
      "step": 97717
    },
    {
      "epoch": 2.261829376220703e-05,
      "step": 97717,
      "training_step_time": 7.5719616413116455
    },
    {
      "epoch": 2.2619247436523437e-05,
      "model_forward_time": 1.333862543106079,
      "step": 97718
    },
    {
      "epoch": 2.2619247436523437e-05,
      "step": 97718,
      "training_step_time": 7.922363519668579
    },
    {
      "epoch": 2.2620201110839845e-05,
      "model_forward_time": 1.7198662757873535,
      "step": 97719
    },
    {
      "epoch": 2.2620201110839845e-05,
      "step": 97719,
      "training_step_time": 6.861315965652466
    },
    {
      "epoch": 2.262115478515625e-05,
      "grad_norm": 0.10239815711975098,
      "learning_rate": 1.4217956483063566e-08,
      "loss": 0.0546,
      "step": 97720
    },
    {
      "epoch": 2.262115478515625e-05,
      "model_forward_time": 1.7093169689178467,
      "step": 97720
    },
    {
      "dynamics_loss_avg": 0.048439939692616465,
      "epoch": 2.262115478515625e-05,
      "step": 97720
    },
    {
      "action_loss_avg": 0.0012941957713337616,
      "epoch": 2.262115478515625e-05,
      "step": 97720
    },
    {
      "epoch": 2.262115478515625e-05,
      "step": 97720,
      "training_step_time": 6.877024412155151
    },
    {
      "epoch": 2.2622108459472656e-05,
      "model_forward_time": 1.7212843894958496,
      "step": 97721
    },
    {
      "epoch": 2.2622108459472656e-05,
      "step": 97721,
      "training_step_time": 6.890169620513916
    },
    {
      "epoch": 2.2623062133789064e-05,
      "model_forward_time": 1.715019702911377,
      "step": 97722
    },
    {
      "epoch": 2.2623062133789064e-05,
      "step": 97722,
      "training_step_time": 6.930549383163452
    },
    {
      "epoch": 2.2624015808105468e-05,
      "model_forward_time": 1.7147150039672852,
      "step": 97723
    },
    {
      "epoch": 2.2624015808105468e-05,
      "step": 97723,
      "training_step_time": 6.930248022079468
    },
    {
      "epoch": 2.2624969482421875e-05,
      "model_forward_time": 2.385981559753418,
      "step": 97724
    },
    {
      "epoch": 2.2624969482421875e-05,
      "step": 97724,
      "training_step_time": 11.402674674987793
    },
    {
      "epoch": 2.2625923156738283e-05,
      "model_forward_time": 1.713580846786499,
      "step": 97725
    },
    {
      "epoch": 2.2625923156738283e-05,
      "step": 97725,
      "training_step_time": 7.597274303436279
    },
    {
      "epoch": 2.2626876831054687e-05,
      "model_forward_time": 1.7153303623199463,
      "step": 97726
    },
    {
      "epoch": 2.2626876831054687e-05,
      "step": 97726,
      "training_step_time": 7.5823540687561035
    },
    {
      "epoch": 2.2627830505371094e-05,
      "model_forward_time": 1.7160077095031738,
      "step": 97727
    },
    {
      "epoch": 2.2627830505371094e-05,
      "step": 97727,
      "training_step_time": 7.835810422897339
    },
    {
      "epoch": 2.26287841796875e-05,
      "model_forward_time": 1.715254783630371,
      "step": 97728
    },
    {
      "epoch": 2.26287841796875e-05,
      "step": 97728,
      "training_step_time": 6.917763710021973
    },
    {
      "epoch": 2.2629737854003905e-05,
      "model_forward_time": 1.712904453277588,
      "step": 97729
    },
    {
      "epoch": 2.2629737854003905e-05,
      "step": 97729,
      "training_step_time": 6.889405727386475
    },
    {
      "epoch": 2.2630691528320313e-05,
      "grad_norm": 0.0970730185508728,
      "learning_rate": 1.409362398091474e-08,
      "loss": 0.0551,
      "step": 97730
    },
    {
      "epoch": 2.2630691528320313e-05,
      "model_forward_time": 1.7159209251403809,
      "step": 97730
    },
    {
      "dynamics_loss_avg": 0.060068448260426524,
      "epoch": 2.2630691528320313e-05,
      "step": 97730
    },
    {
      "action_loss_avg": 0.001103129266994074,
      "epoch": 2.2630691528320313e-05,
      "step": 97730
    },
    {
      "epoch": 2.2630691528320313e-05,
      "step": 97730,
      "training_step_time": 6.8822996616363525
    },
    {
      "epoch": 2.263164520263672e-05,
      "model_forward_time": 1.7141742706298828,
      "step": 97731
    },
    {
      "epoch": 2.263164520263672e-05,
      "step": 97731,
      "training_step_time": 6.909993410110474
    },
    {
      "epoch": 2.2632598876953124e-05,
      "model_forward_time": 1.7107670307159424,
      "step": 97732
    },
    {
      "epoch": 2.2632598876953124e-05,
      "step": 97732,
      "training_step_time": 6.896232604980469
    },
    {
      "epoch": 2.2633552551269532e-05,
      "model_forward_time": 2.4023990631103516,
      "step": 97733
    },
    {
      "epoch": 2.2633552551269532e-05,
      "step": 97733,
      "training_step_time": 11.690929651260376
    },
    {
      "epoch": 2.263450622558594e-05,
      "model_forward_time": 1.71470046043396,
      "step": 97734
    },
    {
      "epoch": 2.263450622558594e-05,
      "step": 97734,
      "training_step_time": 7.810896396636963
    },
    {
      "epoch": 2.2635459899902343e-05,
      "model_forward_time": 1.7139902114868164,
      "step": 97735
    },
    {
      "epoch": 2.2635459899902343e-05,
      "step": 97735,
      "training_step_time": 7.585062503814697
    },
    {
      "epoch": 2.263641357421875e-05,
      "model_forward_time": 1.714693307876587,
      "step": 97736
    },
    {
      "epoch": 2.263641357421875e-05,
      "step": 97736,
      "training_step_time": 7.803996324539185
    },
    {
      "epoch": 2.2637367248535155e-05,
      "model_forward_time": 1.7100136280059814,
      "step": 97737
    },
    {
      "epoch": 2.2637367248535155e-05,
      "step": 97737,
      "training_step_time": 6.9011218547821045
    },
    {
      "epoch": 2.2638320922851562e-05,
      "model_forward_time": 1.7130975723266602,
      "step": 97738
    },
    {
      "epoch": 2.2638320922851562e-05,
      "step": 97738,
      "training_step_time": 6.908866882324219
    },
    {
      "epoch": 2.263927459716797e-05,
      "model_forward_time": 1.7168679237365723,
      "step": 97739
    },
    {
      "epoch": 2.263927459716797e-05,
      "step": 97739,
      "training_step_time": 6.89188289642334
    },
    {
      "epoch": 2.2640228271484374e-05,
      "grad_norm": 0.10278449952602386,
      "learning_rate": 1.3969836729990637e-08,
      "loss": 0.0557,
      "step": 97740
    },
    {
      "epoch": 2.2640228271484374e-05,
      "model_forward_time": 1.7115027904510498,
      "step": 97740
    },
    {
      "dynamics_loss_avg": 0.06851235553622245,
      "epoch": 2.2640228271484374e-05,
      "step": 97740
    },
    {
      "action_loss_avg": 0.0013012669689487666,
      "epoch": 2.2640228271484374e-05,
      "step": 97740
    },
    {
      "epoch": 2.2640228271484374e-05,
      "step": 97740,
      "training_step_time": 6.786430835723877
    },
    {
      "epoch": 2.264118194580078e-05,
      "model_forward_time": 1.713576316833496,
      "step": 97741
    },
    {
      "epoch": 2.264118194580078e-05,
      "step": 97741,
      "training_step_time": 6.916291952133179
    },
    {
      "epoch": 2.264213562011719e-05,
      "model_forward_time": 1.8001940250396729,
      "step": 97742
    },
    {
      "epoch": 2.264213562011719e-05,
      "step": 97742,
      "training_step_time": 10.645133018493652
    },
    {
      "epoch": 2.2643089294433592e-05,
      "model_forward_time": 1.7118949890136719,
      "step": 97743
    },
    {
      "epoch": 2.2643089294433592e-05,
      "step": 97743,
      "training_step_time": 7.765686750411987
    },
    {
      "epoch": 2.264404296875e-05,
      "model_forward_time": 1.7181715965270996,
      "step": 97744
    },
    {
      "epoch": 2.264404296875e-05,
      "step": 97744,
      "training_step_time": 7.473892450332642
    },
    {
      "epoch": 2.2644996643066407e-05,
      "model_forward_time": 1.71824312210083,
      "step": 97745
    },
    {
      "epoch": 2.2644996643066407e-05,
      "step": 97745,
      "training_step_time": 7.904144525527954
    },
    {
      "epoch": 2.264595031738281e-05,
      "model_forward_time": 1.7163689136505127,
      "step": 97746
    },
    {
      "epoch": 2.264595031738281e-05,
      "step": 97746,
      "training_step_time": 6.922307968139648
    },
    {
      "epoch": 2.264690399169922e-05,
      "model_forward_time": 1.719219446182251,
      "step": 97747
    },
    {
      "epoch": 2.264690399169922e-05,
      "step": 97747,
      "training_step_time": 6.9389777183532715
    },
    {
      "epoch": 2.2647857666015626e-05,
      "model_forward_time": 1.7188069820404053,
      "step": 97748
    },
    {
      "epoch": 2.2647857666015626e-05,
      "step": 97748,
      "training_step_time": 6.912156105041504
    },
    {
      "epoch": 2.264881134033203e-05,
      "model_forward_time": 1.7179985046386719,
      "step": 97749
    },
    {
      "epoch": 2.264881134033203e-05,
      "step": 97749,
      "training_step_time": 6.96716833114624
    },
    {
      "epoch": 2.2649765014648438e-05,
      "grad_norm": 0.0663982406258583,
      "learning_rate": 1.3846594743829323e-08,
      "loss": 0.0549,
      "step": 97750
    },
    {
      "epoch": 2.2649765014648438e-05,
      "model_forward_time": 1.708385705947876,
      "step": 97750
    },
    {
      "dynamics_loss_avg": 0.056072226539254186,
      "epoch": 2.2649765014648438e-05,
      "step": 97750
    },
    {
      "action_loss_avg": 0.001325215381802991,
      "epoch": 2.2649765014648438e-05,
      "step": 97750
    },
    {
      "epoch": 2.2649765014648438e-05,
      "step": 97750,
      "training_step_time": 6.968376636505127
    },
    {
      "epoch": 2.2650718688964845e-05,
      "model_forward_time": 1.8080599308013916,
      "step": 97751
    },
    {
      "epoch": 2.2650718688964845e-05,
      "step": 97751,
      "training_step_time": 8.133398056030273
    },
    {
      "epoch": 2.265167236328125e-05,
      "model_forward_time": 2.3530728816986084,
      "step": 97752
    },
    {
      "epoch": 2.265167236328125e-05,
      "step": 97752,
      "training_step_time": 9.669389963150024
    },
    {
      "epoch": 2.2652626037597656e-05,
      "model_forward_time": 1.7163207530975342,
      "step": 97753
    },
    {
      "epoch": 2.2652626037597656e-05,
      "step": 97753,
      "training_step_time": 7.1454126834869385
    },
    {
      "epoch": 2.2653579711914064e-05,
      "model_forward_time": 1.716130256652832,
      "step": 97754
    },
    {
      "epoch": 2.2653579711914064e-05,
      "step": 97754,
      "training_step_time": 8.191511869430542
    },
    {
      "epoch": 2.2654533386230468e-05,
      "model_forward_time": 1.7137601375579834,
      "step": 97755
    },
    {
      "epoch": 2.2654533386230468e-05,
      "step": 97755,
      "training_step_time": 7.678904056549072
    },
    {
      "epoch": 2.2655487060546875e-05,
      "model_forward_time": 1.7187042236328125,
      "step": 97756
    },
    {
      "epoch": 2.2655487060546875e-05,
      "step": 97756,
      "training_step_time": 6.905049800872803
    },
    {
      "epoch": 2.2656440734863283e-05,
      "model_forward_time": 1.7121689319610596,
      "step": 97757
    },
    {
      "epoch": 2.2656440734863283e-05,
      "step": 97757,
      "training_step_time": 6.853510618209839
    },
    {
      "epoch": 2.2657394409179687e-05,
      "model_forward_time": 1.7146916389465332,
      "step": 97758
    },
    {
      "epoch": 2.2657394409179687e-05,
      "step": 97758,
      "training_step_time": 6.925335168838501
    },
    {
      "epoch": 2.2658348083496094e-05,
      "model_forward_time": 1.717761516571045,
      "step": 97759
    },
    {
      "epoch": 2.2658348083496094e-05,
      "step": 97759,
      "training_step_time": 6.8729026317596436
    },
    {
      "epoch": 2.26593017578125e-05,
      "grad_norm": 0.08649394661188126,
      "learning_rate": 1.372389803590779e-08,
      "loss": 0.0558,
      "step": 97760
    },
    {
      "epoch": 2.26593017578125e-05,
      "model_forward_time": 1.7969574928283691,
      "step": 97760
    },
    {
      "dynamics_loss_avg": 0.05286723831668496,
      "epoch": 2.26593017578125e-05,
      "step": 97760
    },
    {
      "action_loss_avg": 0.0016108130104839802,
      "epoch": 2.26593017578125e-05,
      "step": 97760
    },
    {
      "epoch": 2.26593017578125e-05,
      "step": 97760,
      "training_step_time": 7.557180404663086
    },
    {
      "epoch": 2.2660255432128906e-05,
      "model_forward_time": 2.3541996479034424,
      "step": 97761
    },
    {
      "epoch": 2.2660255432128906e-05,
      "step": 97761,
      "training_step_time": 12.31618881225586
    },
    {
      "epoch": 2.2661209106445313e-05,
      "model_forward_time": 1.3340635299682617,
      "step": 97762
    },
    {
      "epoch": 2.2661209106445313e-05,
      "step": 97762,
      "training_step_time": 8.27436637878418
    },
    {
      "epoch": 2.266216278076172e-05,
      "model_forward_time": 1.7185654640197754,
      "step": 97763
    },
    {
      "epoch": 2.266216278076172e-05,
      "step": 97763,
      "training_step_time": 7.50048828125
    },
    {
      "epoch": 2.2663116455078124e-05,
      "model_forward_time": 1.718637228012085,
      "step": 97764
    },
    {
      "epoch": 2.2663116455078124e-05,
      "step": 97764,
      "training_step_time": 7.929711580276489
    },
    {
      "epoch": 2.2664070129394532e-05,
      "model_forward_time": 1.7208895683288574,
      "step": 97765
    },
    {
      "epoch": 2.2664070129394532e-05,
      "step": 97765,
      "training_step_time": 6.838029384613037
    },
    {
      "epoch": 2.2665023803710936e-05,
      "model_forward_time": 1.7190783023834229,
      "step": 97766
    },
    {
      "epoch": 2.2665023803710936e-05,
      "step": 97766,
      "training_step_time": 6.871475458145142
    },
    {
      "epoch": 2.2665977478027343e-05,
      "model_forward_time": 1.7177677154541016,
      "step": 97767
    },
    {
      "epoch": 2.2665977478027343e-05,
      "step": 97767,
      "training_step_time": 6.979564428329468
    },
    {
      "epoch": 2.266693115234375e-05,
      "model_forward_time": 1.7160933017730713,
      "step": 97768
    },
    {
      "epoch": 2.266693115234375e-05,
      "step": 97768,
      "training_step_time": 6.932331562042236
    },
    {
      "epoch": 2.2667884826660155e-05,
      "model_forward_time": 1.810859203338623,
      "step": 97769
    },
    {
      "epoch": 2.2667884826660155e-05,
      "step": 97769,
      "training_step_time": 7.662768363952637
    },
    {
      "epoch": 2.2668838500976562e-05,
      "grad_norm": 0.12325407564640045,
      "learning_rate": 1.360174661964364e-08,
      "loss": 0.0547,
      "step": 97770
    },
    {
      "epoch": 2.2668838500976562e-05,
      "model_forward_time": 2.341665029525757,
      "step": 97770
    },
    {
      "dynamics_loss_avg": 0.05187705531716347,
      "epoch": 2.2668838500976562e-05,
      "step": 97770
    },
    {
      "action_loss_avg": 0.0014511911111185327,
      "epoch": 2.2668838500976562e-05,
      "step": 97770
    },
    {
      "epoch": 2.2668838500976562e-05,
      "step": 97770,
      "training_step_time": 9.324577808380127
    },
    {
      "epoch": 2.266979217529297e-05,
      "model_forward_time": 1.721341848373413,
      "step": 97771
    },
    {
      "epoch": 2.266979217529297e-05,
      "step": 97771,
      "training_step_time": 8.326983213424683
    },
    {
      "epoch": 2.2670745849609374e-05,
      "model_forward_time": 1.7182509899139404,
      "step": 97772
    },
    {
      "epoch": 2.2670745849609374e-05,
      "step": 97772,
      "training_step_time": 7.582980155944824
    },
    {
      "epoch": 2.267169952392578e-05,
      "model_forward_time": 1.7219457626342773,
      "step": 97773
    },
    {
      "epoch": 2.267169952392578e-05,
      "step": 97773,
      "training_step_time": 7.9268434047698975
    },
    {
      "epoch": 2.267265319824219e-05,
      "model_forward_time": 1.711806058883667,
      "step": 97774
    },
    {
      "epoch": 2.267265319824219e-05,
      "step": 97774,
      "training_step_time": 6.921105861663818
    },
    {
      "epoch": 2.2673606872558592e-05,
      "model_forward_time": 1.7147388458251953,
      "step": 97775
    },
    {
      "epoch": 2.2673606872558592e-05,
      "step": 97775,
      "training_step_time": 6.925049304962158
    },
    {
      "epoch": 2.2674560546875e-05,
      "model_forward_time": 1.7128136157989502,
      "step": 97776
    },
    {
      "epoch": 2.2674560546875e-05,
      "step": 97776,
      "training_step_time": 6.891695737838745
    },
    {
      "epoch": 2.2675514221191407e-05,
      "model_forward_time": 1.7145061492919922,
      "step": 97777
    },
    {
      "epoch": 2.2675514221191407e-05,
      "step": 97777,
      "training_step_time": 6.931938648223877
    },
    {
      "epoch": 2.267646789550781e-05,
      "model_forward_time": 1.7832539081573486,
      "step": 97778
    },
    {
      "epoch": 2.267646789550781e-05,
      "step": 97778,
      "training_step_time": 7.6468048095703125
    },
    {
      "epoch": 2.267742156982422e-05,
      "model_forward_time": 2.3061163425445557,
      "step": 97779
    },
    {
      "epoch": 2.267742156982422e-05,
      "step": 97779,
      "training_step_time": 11.96556305885315
    },
    {
      "epoch": 2.2678375244140626e-05,
      "grad_norm": 0.09848619997501373,
      "learning_rate": 1.3480140508396188e-08,
      "loss": 0.0555,
      "step": 97780
    },
    {
      "epoch": 2.2678375244140626e-05,
      "model_forward_time": 1.7105064392089844,
      "step": 97780
    },
    {
      "dynamics_loss_avg": 0.042787839844822886,
      "epoch": 2.2678375244140626e-05,
      "step": 97780
    },
    {
      "action_loss_avg": 0.0008537981018889695,
      "epoch": 2.2678375244140626e-05,
      "step": 97780
    },
    {
      "epoch": 2.2678375244140626e-05,
      "step": 97780,
      "training_step_time": 7.70287299156189
    },
    {
      "epoch": 2.267932891845703e-05,
      "model_forward_time": 1.7340397834777832,
      "step": 97781
    },
    {
      "epoch": 2.267932891845703e-05,
      "step": 97781,
      "training_step_time": 7.55068564414978
    },
    {
      "epoch": 2.2680282592773438e-05,
      "model_forward_time": 1.7872068881988525,
      "step": 97782
    },
    {
      "epoch": 2.2680282592773438e-05,
      "step": 97782,
      "training_step_time": 8.004474639892578
    },
    {
      "epoch": 2.2681236267089845e-05,
      "model_forward_time": 1.759685754776001,
      "step": 97783
    },
    {
      "epoch": 2.2681236267089845e-05,
      "step": 97783,
      "training_step_time": 7.607081413269043
    },
    {
      "epoch": 2.268218994140625e-05,
      "model_forward_time": 1.7423324584960938,
      "step": 97784
    },
    {
      "epoch": 2.268218994140625e-05,
      "step": 97784,
      "training_step_time": 6.931347370147705
    },
    {
      "epoch": 2.2683143615722656e-05,
      "model_forward_time": 1.7462584972381592,
      "step": 97785
    },
    {
      "epoch": 2.2683143615722656e-05,
      "step": 97785,
      "training_step_time": 6.9119672775268555
    },
    {
      "epoch": 2.2684097290039064e-05,
      "model_forward_time": 1.766420602798462,
      "step": 97786
    },
    {
      "epoch": 2.2684097290039064e-05,
      "step": 97786,
      "training_step_time": 6.913640737533569
    },
    {
      "epoch": 2.2685050964355468e-05,
      "model_forward_time": 1.8283896446228027,
      "step": 97787
    },
    {
      "epoch": 2.2685050964355468e-05,
      "step": 97787,
      "training_step_time": 7.603857755661011
    },
    {
      "epoch": 2.2686004638671875e-05,
      "model_forward_time": 1.7315177917480469,
      "step": 97788
    },
    {
      "epoch": 2.2686004638671875e-05,
      "step": 97788,
      "training_step_time": 9.522166967391968
    },
    {
      "epoch": 2.2686958312988283e-05,
      "model_forward_time": 2.3756842613220215,
      "step": 97789
    },
    {
      "epoch": 2.2686958312988283e-05,
      "step": 97789,
      "training_step_time": 7.776423692703247
    },
    {
      "epoch": 2.2687911987304687e-05,
      "grad_norm": 0.13619165122509003,
      "learning_rate": 1.335907971546313e-08,
      "loss": 0.0561,
      "step": 97790
    },
    {
      "epoch": 2.2687911987304687e-05,
      "model_forward_time": 1.7548935413360596,
      "step": 97790
    },
    {
      "dynamics_loss_avg": 0.04487674869596958,
      "epoch": 2.2687911987304687e-05,
      "step": 97790
    },
    {
      "action_loss_avg": 0.0007935616216855123,
      "epoch": 2.2687911987304687e-05,
      "step": 97790
    },
    {
      "epoch": 2.2687911987304687e-05,
      "step": 97790,
      "training_step_time": 7.453174352645874
    },
    {
      "epoch": 2.2688865661621094e-05,
      "model_forward_time": 1.7540607452392578,
      "step": 97791
    },
    {
      "epoch": 2.2688865661621094e-05,
      "step": 97791,
      "training_step_time": 8.959038257598877
    },
    {
      "epoch": 2.26898193359375e-05,
      "model_forward_time": 1.7565710544586182,
      "step": 97792
    },
    {
      "epoch": 2.26898193359375e-05,
      "step": 97792,
      "training_step_time": 7.550121545791626
    },
    {
      "epoch": 2.2690773010253906e-05,
      "model_forward_time": 1.7552330493927002,
      "step": 97793
    },
    {
      "epoch": 2.2690773010253906e-05,
      "step": 97793,
      "training_step_time": 6.878014802932739
    },
    {
      "epoch": 2.2691726684570313e-05,
      "model_forward_time": 1.7671325206756592,
      "step": 97794
    },
    {
      "epoch": 2.2691726684570313e-05,
      "step": 97794,
      "training_step_time": 6.9390435218811035
    },
    {
      "epoch": 2.269268035888672e-05,
      "model_forward_time": 1.7544140815734863,
      "step": 97795
    },
    {
      "epoch": 2.269268035888672e-05,
      "step": 97795,
      "training_step_time": 6.9248948097229
    },
    {
      "epoch": 2.2693634033203124e-05,
      "model_forward_time": 1.7986435890197754,
      "step": 97796
    },
    {
      "epoch": 2.2693634033203124e-05,
      "step": 97796,
      "training_step_time": 7.62052321434021
    },
    {
      "epoch": 2.2694587707519532e-05,
      "model_forward_time": 1.7471768856048584,
      "step": 97797
    },
    {
      "epoch": 2.2694587707519532e-05,
      "step": 97797,
      "training_step_time": 10.503973722457886
    },
    {
      "epoch": 2.2695541381835936e-05,
      "model_forward_time": 2.414524793624878,
      "step": 97798
    },
    {
      "epoch": 2.2695541381835936e-05,
      "step": 97798,
      "training_step_time": 10.336554527282715
    },
    {
      "epoch": 2.2696495056152343e-05,
      "model_forward_time": 1.753227949142456,
      "step": 97799
    },
    {
      "epoch": 2.2696495056152343e-05,
      "step": 97799,
      "training_step_time": 7.6174421310424805
    },
    {
      "epoch": 2.269744873046875e-05,
      "grad_norm": 0.07451164722442627,
      "learning_rate": 1.3238564254083875e-08,
      "loss": 0.0562,
      "step": 97800
    },
    {
      "epoch": 2.269744873046875e-05,
      "model_forward_time": 1.7603106498718262,
      "step": 97800
    },
    {
      "dynamics_loss_avg": 0.03661435227841139,
      "epoch": 2.269744873046875e-05,
      "step": 97800
    },
    {
      "action_loss_avg": 0.0011744332878151909,
      "epoch": 2.269744873046875e-05,
      "step": 97800
    },
    {
      "epoch": 2.269744873046875e-05,
      "step": 97800,
      "training_step_time": 7.411129951477051
    },
    {
      "epoch": 2.2698402404785155e-05,
      "model_forward_time": 1.7559316158294678,
      "step": 97801
    },
    {
      "epoch": 2.2698402404785155e-05,
      "step": 97801,
      "training_step_time": 7.924019813537598
    },
    {
      "epoch": 2.2699356079101562e-05,
      "model_forward_time": 1.754960060119629,
      "step": 97802
    },
    {
      "epoch": 2.2699356079101562e-05,
      "step": 97802,
      "training_step_time": 6.891891002655029
    },
    {
      "epoch": 2.270030975341797e-05,
      "model_forward_time": 1.7548704147338867,
      "step": 97803
    },
    {
      "epoch": 2.270030975341797e-05,
      "step": 97803,
      "training_step_time": 6.928610324859619
    },
    {
      "epoch": 2.2701263427734374e-05,
      "model_forward_time": 1.7608482837677002,
      "step": 97804
    },
    {
      "epoch": 2.2701263427734374e-05,
      "step": 97804,
      "training_step_time": 6.9478981494903564
    },
    {
      "epoch": 2.270221710205078e-05,
      "model_forward_time": 1.7657065391540527,
      "step": 97805
    },
    {
      "epoch": 2.270221710205078e-05,
      "step": 97805,
      "training_step_time": 7.95490574836731
    },
    {
      "epoch": 2.270317077636719e-05,
      "model_forward_time": 1.7511589527130127,
      "step": 97806
    },
    {
      "epoch": 2.270317077636719e-05,
      "step": 97806,
      "training_step_time": 7.70638632774353
    },
    {
      "epoch": 2.2704124450683592e-05,
      "model_forward_time": 2.3909831047058105,
      "step": 97807
    },
    {
      "epoch": 2.2704124450683592e-05,
      "step": 97807,
      "training_step_time": 10.939476490020752
    },
    {
      "epoch": 2.2705078125e-05,
      "model_forward_time": 1.748795509338379,
      "step": 97808
    },
    {
      "epoch": 2.2705078125e-05,
      "step": 97808,
      "training_step_time": 7.740182399749756
    },
    {
      "epoch": 2.2706031799316407e-05,
      "model_forward_time": 1.7465627193450928,
      "step": 97809
    },
    {
      "epoch": 2.2706031799316407e-05,
      "step": 97809,
      "training_step_time": 7.5423197746276855
    },
    {
      "epoch": 2.270698547363281e-05,
      "grad_norm": 0.06682451069355011,
      "learning_rate": 1.3118594137437324e-08,
      "loss": 0.0548,
      "step": 97810
    },
    {
      "epoch": 2.270698547363281e-05,
      "model_forward_time": 1.7490394115447998,
      "step": 97810
    },
    {
      "dynamics_loss_avg": 0.06625459678471088,
      "epoch": 2.270698547363281e-05,
      "step": 97810
    },
    {
      "action_loss_avg": 0.0012756219075527042,
      "epoch": 2.270698547363281e-05,
      "step": 97810
    },
    {
      "epoch": 2.270698547363281e-05,
      "step": 97810,
      "training_step_time": 7.884168863296509
    },
    {
      "epoch": 2.270793914794922e-05,
      "model_forward_time": 1.7467138767242432,
      "step": 97811
    },
    {
      "epoch": 2.270793914794922e-05,
      "step": 97811,
      "training_step_time": 7.416811466217041
    },
    {
      "epoch": 2.2708892822265626e-05,
      "model_forward_time": 1.7468128204345703,
      "step": 97812
    },
    {
      "epoch": 2.2708892822265626e-05,
      "step": 97812,
      "training_step_time": 6.932008266448975
    },
    {
      "epoch": 2.270984649658203e-05,
      "model_forward_time": 1.750490427017212,
      "step": 97813
    },
    {
      "epoch": 2.270984649658203e-05,
      "step": 97813,
      "training_step_time": 6.90996527671814
    },
    {
      "epoch": 2.2710800170898438e-05,
      "model_forward_time": 1.7424125671386719,
      "step": 97814
    },
    {
      "epoch": 2.2710800170898438e-05,
      "step": 97814,
      "training_step_time": 6.890149354934692
    },
    {
      "epoch": 2.2711753845214845e-05,
      "model_forward_time": 1.8688671588897705,
      "step": 97815
    },
    {
      "epoch": 2.2711753845214845e-05,
      "step": 97815,
      "training_step_time": 8.265102624893188
    },
    {
      "epoch": 2.271270751953125e-05,
      "model_forward_time": 2.4307188987731934,
      "step": 97816
    },
    {
      "epoch": 2.271270751953125e-05,
      "step": 97816,
      "training_step_time": 11.81101369857788
    },
    {
      "epoch": 2.2713661193847656e-05,
      "model_forward_time": 1.746206283569336,
      "step": 97817
    },
    {
      "epoch": 2.2713661193847656e-05,
      "step": 97817,
      "training_step_time": 8.689805030822754
    },
    {
      "epoch": 2.2714614868164064e-05,
      "model_forward_time": 1.752012014389038,
      "step": 97818
    },
    {
      "epoch": 2.2714614868164064e-05,
      "step": 97818,
      "training_step_time": 7.549015283584595
    },
    {
      "epoch": 2.2715568542480468e-05,
      "model_forward_time": 1.753005027770996,
      "step": 97819
    },
    {
      "epoch": 2.2715568542480468e-05,
      "step": 97819,
      "training_step_time": 9.037848949432373
    },
    {
      "epoch": 2.2716522216796875e-05,
      "grad_norm": 0.07038429379463196,
      "learning_rate": 1.2999169378644095e-08,
      "loss": 0.0565,
      "step": 97820
    },
    {
      "epoch": 2.2716522216796875e-05,
      "model_forward_time": 1.74186372756958,
      "step": 97820
    },
    {
      "dynamics_loss_avg": 0.056148644909262656,
      "epoch": 2.2716522216796875e-05,
      "step": 97820
    },
    {
      "action_loss_avg": 0.0015111107495613396,
      "epoch": 2.2716522216796875e-05,
      "step": 97820
    },
    {
      "epoch": 2.2716522216796875e-05,
      "step": 97820,
      "training_step_time": 7.581991195678711
    },
    {
      "epoch": 2.2717475891113283e-05,
      "model_forward_time": 1.7580807209014893,
      "step": 97821
    },
    {
      "epoch": 2.2717475891113283e-05,
      "step": 97821,
      "training_step_time": 6.885143995285034
    },
    {
      "epoch": 2.2718429565429687e-05,
      "model_forward_time": 1.7497889995574951,
      "step": 97822
    },
    {
      "epoch": 2.2718429565429687e-05,
      "step": 97822,
      "training_step_time": 6.935603380203247
    },
    {
      "epoch": 2.2719383239746094e-05,
      "model_forward_time": 1.7523009777069092,
      "step": 97823
    },
    {
      "epoch": 2.2719383239746094e-05,
      "step": 97823,
      "training_step_time": 6.910487174987793
    },
    {
      "epoch": 2.27203369140625e-05,
      "model_forward_time": 1.8208293914794922,
      "step": 97824
    },
    {
      "epoch": 2.27203369140625e-05,
      "step": 97824,
      "training_step_time": 8.549546957015991
    },
    {
      "epoch": 2.2721290588378906e-05,
      "model_forward_time": 2.390610933303833,
      "step": 97825
    },
    {
      "epoch": 2.2721290588378906e-05,
      "step": 97825,
      "training_step_time": 10.34490966796875
    },
    {
      "epoch": 2.2722244262695313e-05,
      "model_forward_time": 1.7392773628234863,
      "step": 97826
    },
    {
      "epoch": 2.2722244262695313e-05,
      "step": 97826,
      "training_step_time": 8.306497812271118
    },
    {
      "epoch": 2.272319793701172e-05,
      "model_forward_time": 1.7422308921813965,
      "step": 97827
    },
    {
      "epoch": 2.272319793701172e-05,
      "step": 97827,
      "training_step_time": 8.456923007965088
    },
    {
      "epoch": 2.2724151611328124e-05,
      "model_forward_time": 1.760328769683838,
      "step": 97828
    },
    {
      "epoch": 2.2724151611328124e-05,
      "step": 97828,
      "training_step_time": 8.01093053817749
    },
    {
      "epoch": 2.2725105285644532e-05,
      "model_forward_time": 1.7524237632751465,
      "step": 97829
    },
    {
      "epoch": 2.2725105285644532e-05,
      "step": 97829,
      "training_step_time": 7.570728778839111
    },
    {
      "epoch": 2.2726058959960936e-05,
      "grad_norm": 0.9230467677116394,
      "learning_rate": 1.288028999076374e-08,
      "loss": 0.0545,
      "step": 97830
    },
    {
      "epoch": 2.2726058959960936e-05,
      "model_forward_time": 1.7401883602142334,
      "step": 97830
    },
    {
      "dynamics_loss_avg": 0.04885383620858193,
      "epoch": 2.2726058959960936e-05,
      "step": 97830
    },
    {
      "action_loss_avg": 0.0009664718701969833,
      "epoch": 2.2726058959960936e-05,
      "step": 97830
    },
    {
      "epoch": 2.2726058959960936e-05,
      "step": 97830,
      "training_step_time": 7.002071142196655
    },
    {
      "epoch": 2.2727012634277343e-05,
      "model_forward_time": 1.7737476825714111,
      "step": 97831
    },
    {
      "epoch": 2.2727012634277343e-05,
      "step": 97831,
      "training_step_time": 6.954524517059326
    },
    {
      "epoch": 2.272796630859375e-05,
      "model_forward_time": 1.7498657703399658,
      "step": 97832
    },
    {
      "epoch": 2.272796630859375e-05,
      "step": 97832,
      "training_step_time": 6.893588304519653
    },
    {
      "epoch": 2.2728919982910155e-05,
      "model_forward_time": 1.7801647186279297,
      "step": 97833
    },
    {
      "epoch": 2.2728919982910155e-05,
      "step": 97833,
      "training_step_time": 8.259517908096313
    },
    {
      "epoch": 2.2729873657226562e-05,
      "model_forward_time": 1.7405040264129639,
      "step": 97834
    },
    {
      "epoch": 2.2729873657226562e-05,
      "step": 97834,
      "training_step_time": 10.463480472564697
    },
    {
      "epoch": 2.273082733154297e-05,
      "model_forward_time": 1.7418553829193115,
      "step": 97835
    },
    {
      "epoch": 2.273082733154297e-05,
      "step": 97835,
      "training_step_time": 7.953799486160278
    },
    {
      "epoch": 2.2731781005859374e-05,
      "model_forward_time": 1.7480785846710205,
      "step": 97836
    },
    {
      "epoch": 2.2731781005859374e-05,
      "step": 97836,
      "training_step_time": 7.610555410385132
    },
    {
      "epoch": 2.273273468017578e-05,
      "model_forward_time": 1.7606403827667236,
      "step": 97837
    },
    {
      "epoch": 2.273273468017578e-05,
      "step": 97837,
      "training_step_time": 8.01572871208191
    },
    {
      "epoch": 2.273368835449219e-05,
      "model_forward_time": 1.7601404190063477,
      "step": 97838
    },
    {
      "epoch": 2.273368835449219e-05,
      "step": 97838,
      "training_step_time": 7.598826169967651
    },
    {
      "epoch": 2.2734642028808592e-05,
      "model_forward_time": 1.7534511089324951,
      "step": 97839
    },
    {
      "epoch": 2.2734642028808592e-05,
      "step": 97839,
      "training_step_time": 6.89806866645813
    },
    {
      "epoch": 2.2735595703125e-05,
      "grad_norm": 0.08893920481204987,
      "learning_rate": 1.276195598679697e-08,
      "loss": 0.0543,
      "step": 97840
    },
    {
      "epoch": 2.2735595703125e-05,
      "model_forward_time": 1.7508373260498047,
      "step": 97840
    },
    {
      "dynamics_loss_avg": 0.05573365949094296,
      "epoch": 2.2735595703125e-05,
      "step": 97840
    },
    {
      "action_loss_avg": 0.000889229541644454,
      "epoch": 2.2735595703125e-05,
      "step": 97840
    },
    {
      "epoch": 2.2735595703125e-05,
      "step": 97840,
      "training_step_time": 6.881686449050903
    },
    {
      "epoch": 2.2736549377441407e-05,
      "model_forward_time": 1.7564823627471924,
      "step": 97841
    },
    {
      "epoch": 2.2736549377441407e-05,
      "step": 97841,
      "training_step_time": 6.918670892715454
    },
    {
      "epoch": 2.273750305175781e-05,
      "model_forward_time": 1.7390806674957275,
      "step": 97842
    },
    {
      "epoch": 2.273750305175781e-05,
      "step": 97842,
      "training_step_time": 7.7386109828948975
    },
    {
      "epoch": 2.273845672607422e-05,
      "model_forward_time": 1.8605291843414307,
      "step": 97843
    },
    {
      "epoch": 2.273845672607422e-05,
      "step": 97843,
      "training_step_time": 7.835078716278076
    },
    {
      "epoch": 2.2739410400390626e-05,
      "model_forward_time": 2.3989455699920654,
      "step": 97844
    },
    {
      "epoch": 2.2739410400390626e-05,
      "step": 97844,
      "training_step_time": 9.659517049789429
    },
    {
      "epoch": 2.274036407470703e-05,
      "model_forward_time": 1.739473819732666,
      "step": 97845
    },
    {
      "epoch": 2.274036407470703e-05,
      "step": 97845,
      "training_step_time": 7.685259819030762
    },
    {
      "epoch": 2.2741317749023438e-05,
      "model_forward_time": 1.733323574066162,
      "step": 97846
    },
    {
      "epoch": 2.2741317749023438e-05,
      "step": 97846,
      "training_step_time": 7.5292649269104
    },
    {
      "epoch": 2.2742271423339845e-05,
      "model_forward_time": 1.731097936630249,
      "step": 97847
    },
    {
      "epoch": 2.2742271423339845e-05,
      "step": 97847,
      "training_step_time": 8.67036771774292
    },
    {
      "epoch": 2.274322509765625e-05,
      "model_forward_time": 1.7384274005889893,
      "step": 97848
    },
    {
      "epoch": 2.274322509765625e-05,
      "step": 97848,
      "training_step_time": 6.926368236541748
    },
    {
      "epoch": 2.2744178771972656e-05,
      "model_forward_time": 1.741741418838501,
      "step": 97849
    },
    {
      "epoch": 2.2744178771972656e-05,
      "step": 97849,
      "training_step_time": 6.880311727523804
    },
    {
      "epoch": 2.2745132446289064e-05,
      "grad_norm": 0.0721617341041565,
      "learning_rate": 1.2644167379683992e-08,
      "loss": 0.0548,
      "step": 97850
    },
    {
      "epoch": 2.2745132446289064e-05,
      "model_forward_time": 1.7269749641418457,
      "step": 97850
    },
    {
      "dynamics_loss_avg": 0.06249922811985016,
      "epoch": 2.2745132446289064e-05,
      "step": 97850
    },
    {
      "action_loss_avg": 0.0010792481712996959,
      "epoch": 2.2745132446289064e-05,
      "step": 97850
    },
    {
      "epoch": 2.2745132446289064e-05,
      "step": 97850,
      "training_step_time": 6.8594911098480225
    },
    {
      "epoch": 2.2746086120605468e-05,
      "model_forward_time": 1.733466625213623,
      "step": 97851
    },
    {
      "epoch": 2.2746086120605468e-05,
      "step": 97851,
      "training_step_time": 7.6412341594696045
    },
    {
      "epoch": 2.2747039794921875e-05,
      "model_forward_time": 1.7434983253479004,
      "step": 97852
    },
    {
      "epoch": 2.2747039794921875e-05,
      "step": 97852,
      "training_step_time": 8.673206567764282
    },
    {
      "epoch": 2.2747993469238283e-05,
      "model_forward_time": 2.4018447399139404,
      "step": 97853
    },
    {
      "epoch": 2.2747993469238283e-05,
      "step": 97853,
      "training_step_time": 10.838189601898193
    },
    {
      "epoch": 2.2748947143554687e-05,
      "model_forward_time": 1.7377545833587646,
      "step": 97854
    },
    {
      "epoch": 2.2748947143554687e-05,
      "step": 97854,
      "training_step_time": 7.980212688446045
    },
    {
      "epoch": 2.2749900817871094e-05,
      "model_forward_time": 1.7358930110931396,
      "step": 97855
    },
    {
      "epoch": 2.2749900817871094e-05,
      "step": 97855,
      "training_step_time": 7.6700663566589355
    },
    {
      "epoch": 2.27508544921875e-05,
      "model_forward_time": 1.7334120273590088,
      "step": 97856
    },
    {
      "epoch": 2.27508544921875e-05,
      "step": 97856,
      "training_step_time": 7.934462547302246
    },
    {
      "epoch": 2.2751808166503906e-05,
      "model_forward_time": 1.7428224086761475,
      "step": 97857
    },
    {
      "epoch": 2.2751808166503906e-05,
      "step": 97857,
      "training_step_time": 7.533207178115845
    },
    {
      "epoch": 2.2752761840820313e-05,
      "model_forward_time": 1.7424027919769287,
      "step": 97858
    },
    {
      "epoch": 2.2752761840820313e-05,
      "step": 97858,
      "training_step_time": 7.055846452713013
    },
    {
      "epoch": 2.275371551513672e-05,
      "model_forward_time": 1.7277700901031494,
      "step": 97859
    },
    {
      "epoch": 2.275371551513672e-05,
      "step": 97859,
      "training_step_time": 6.998376369476318
    },
    {
      "epoch": 2.2754669189453124e-05,
      "grad_norm": 0.11508446931838989,
      "learning_rate": 1.2526924182307276e-08,
      "loss": 0.0552,
      "step": 97860
    },
    {
      "epoch": 2.2754669189453124e-05,
      "model_forward_time": 1.726501703262329,
      "step": 97860
    },
    {
      "dynamics_loss_avg": 0.05553952576592565,
      "epoch": 2.2754669189453124e-05,
      "step": 97860
    },
    {
      "action_loss_avg": 0.0012100842781364919,
      "epoch": 2.2754669189453124e-05,
      "step": 97860
    },
    {
      "epoch": 2.2754669189453124e-05,
      "step": 97860,
      "training_step_time": 7.034905195236206
    },
    {
      "epoch": 2.2755622863769532e-05,
      "model_forward_time": 1.7854492664337158,
      "step": 97861
    },
    {
      "epoch": 2.2755622863769532e-05,
      "step": 97861,
      "training_step_time": 7.922396659851074
    },
    {
      "epoch": 2.2756576538085936e-05,
      "model_forward_time": 2.406550168991089,
      "step": 97862
    },
    {
      "epoch": 2.2756576538085936e-05,
      "step": 97862,
      "training_step_time": 8.92906641960144
    },
    {
      "epoch": 2.2757530212402343e-05,
      "model_forward_time": 1.7399804592132568,
      "step": 97863
    },
    {
      "epoch": 2.2757530212402343e-05,
      "step": 97863,
      "training_step_time": 9.453975677490234
    },
    {
      "epoch": 2.275848388671875e-05,
      "model_forward_time": 1.7403182983398438,
      "step": 97864
    },
    {
      "epoch": 2.275848388671875e-05,
      "step": 97864,
      "training_step_time": 7.8059446811676025
    },
    {
      "epoch": 2.2759437561035155e-05,
      "model_forward_time": 1.732125997543335,
      "step": 97865
    },
    {
      "epoch": 2.2759437561035155e-05,
      "step": 97865,
      "training_step_time": 7.972746849060059
    },
    {
      "epoch": 2.2760391235351562e-05,
      "model_forward_time": 1.7289416790008545,
      "step": 97866
    },
    {
      "epoch": 2.2760391235351562e-05,
      "step": 97866,
      "training_step_time": 7.5807130336761475
    },
    {
      "epoch": 2.276134490966797e-05,
      "model_forward_time": 1.7386658191680908,
      "step": 97867
    },
    {
      "epoch": 2.276134490966797e-05,
      "step": 97867,
      "training_step_time": 6.948212623596191
    },
    {
      "epoch": 2.2762298583984374e-05,
      "model_forward_time": 1.74033522605896,
      "step": 97868
    },
    {
      "epoch": 2.2762298583984374e-05,
      "step": 97868,
      "training_step_time": 6.8836140632629395
    },
    {
      "epoch": 2.276325225830078e-05,
      "model_forward_time": 1.737891674041748,
      "step": 97869
    },
    {
      "epoch": 2.276325225830078e-05,
      "step": 97869,
      "training_step_time": 6.8805787563323975
    },
    {
      "epoch": 2.276420593261719e-05,
      "grad_norm": 0.06433078646659851,
      "learning_rate": 1.2410226407487125e-08,
      "loss": 0.0555,
      "step": 97870
    },
    {
      "epoch": 2.276420593261719e-05,
      "model_forward_time": 1.7330200672149658,
      "step": 97870
    },
    {
      "dynamics_loss_avg": 0.05090852491557598,
      "epoch": 2.276420593261719e-05,
      "step": 97870
    },
    {
      "action_loss_avg": 0.001678651274414733,
      "epoch": 2.276420593261719e-05,
      "step": 97870
    },
    {
      "epoch": 2.276420593261719e-05,
      "step": 97870,
      "training_step_time": 7.919439315795898
    },
    {
      "epoch": 2.2765159606933592e-05,
      "model_forward_time": 2.4116806983947754,
      "step": 97871
    },
    {
      "epoch": 2.2765159606933592e-05,
      "step": 97871,
      "training_step_time": 9.35077977180481
    },
    {
      "epoch": 2.276611328125e-05,
      "model_forward_time": 1.747694730758667,
      "step": 97872
    },
    {
      "epoch": 2.276611328125e-05,
      "step": 97872,
      "training_step_time": 8.568901538848877
    },
    {
      "epoch": 2.2767066955566407e-05,
      "model_forward_time": 1.7341563701629639,
      "step": 97873
    },
    {
      "epoch": 2.2767066955566407e-05,
      "step": 97873,
      "training_step_time": 7.830057859420776
    },
    {
      "epoch": 2.276802062988281e-05,
      "model_forward_time": 1.7372088432312012,
      "step": 97874
    },
    {
      "epoch": 2.276802062988281e-05,
      "step": 97874,
      "training_step_time": 8.143846035003662
    },
    {
      "epoch": 2.276897430419922e-05,
      "model_forward_time": 1.7330234050750732,
      "step": 97875
    },
    {
      "epoch": 2.276897430419922e-05,
      "step": 97875,
      "training_step_time": 8.716825723648071
    },
    {
      "epoch": 2.2769927978515626e-05,
      "model_forward_time": 1.73295259475708,
      "step": 97876
    },
    {
      "epoch": 2.2769927978515626e-05,
      "step": 97876,
      "training_step_time": 7.042707204818726
    },
    {
      "epoch": 2.277088165283203e-05,
      "model_forward_time": 1.7285385131835938,
      "step": 97877
    },
    {
      "epoch": 2.277088165283203e-05,
      "step": 97877,
      "training_step_time": 7.103784084320068
    },
    {
      "epoch": 2.2771835327148438e-05,
      "model_forward_time": 1.7397994995117188,
      "step": 97878
    },
    {
      "epoch": 2.2771835327148438e-05,
      "step": 97878,
      "training_step_time": 7.055088520050049
    },
    {
      "epoch": 2.2772789001464845e-05,
      "model_forward_time": 1.7425460815429688,
      "step": 97879
    },
    {
      "epoch": 2.2772789001464845e-05,
      "step": 97879,
      "training_step_time": 8.762800455093384
    },
    {
      "epoch": 2.277374267578125e-05,
      "grad_norm": 0.07865986973047256,
      "learning_rate": 1.229407406798555e-08,
      "loss": 0.0549,
      "step": 97880
    },
    {
      "epoch": 2.277374267578125e-05,
      "model_forward_time": 2.387512445449829,
      "step": 97880
    },
    {
      "dynamics_loss_avg": 0.06529288552701473,
      "epoch": 2.277374267578125e-05,
      "step": 97880
    },
    {
      "action_loss_avg": 0.001211172812327277,
      "epoch": 2.277374267578125e-05,
      "step": 97880
    },
    {
      "epoch": 2.277374267578125e-05,
      "step": 97880,
      "training_step_time": 10.668308734893799
    },
    {
      "epoch": 2.2774696350097656e-05,
      "model_forward_time": 1.7417821884155273,
      "step": 97881
    },
    {
      "epoch": 2.2774696350097656e-05,
      "step": 97881,
      "training_step_time": 8.299678087234497
    },
    {
      "epoch": 2.2775650024414064e-05,
      "model_forward_time": 1.734586477279663,
      "step": 97882
    },
    {
      "epoch": 2.2775650024414064e-05,
      "step": 97882,
      "training_step_time": 7.71634578704834
    },
    {
      "epoch": 2.2776603698730468e-05,
      "model_forward_time": 1.7351536750793457,
      "step": 97883
    },
    {
      "epoch": 2.2776603698730468e-05,
      "step": 97883,
      "training_step_time": 8.054676294326782
    },
    {
      "epoch": 2.2777557373046875e-05,
      "model_forward_time": 1.7362151145935059,
      "step": 97884
    },
    {
      "epoch": 2.2777557373046875e-05,
      "step": 97884,
      "training_step_time": 7.600171327590942
    },
    {
      "epoch": 2.2778511047363283e-05,
      "model_forward_time": 1.7327961921691895,
      "step": 97885
    },
    {
      "epoch": 2.2778511047363283e-05,
      "step": 97885,
      "training_step_time": 7.171157121658325
    },
    {
      "epoch": 2.2779464721679687e-05,
      "model_forward_time": 1.7434911727905273,
      "step": 97886
    },
    {
      "epoch": 2.2779464721679687e-05,
      "step": 97886,
      "training_step_time": 6.881670236587524
    },
    {
      "epoch": 2.2780418395996094e-05,
      "model_forward_time": 1.7371904850006104,
      "step": 97887
    },
    {
      "epoch": 2.2780418395996094e-05,
      "step": 97887,
      "training_step_time": 6.868574142456055
    },
    {
      "epoch": 2.27813720703125e-05,
      "model_forward_time": 1.741206169128418,
      "step": 97888
    },
    {
      "epoch": 2.27813720703125e-05,
      "step": 97888,
      "training_step_time": 8.576486587524414
    },
    {
      "epoch": 2.2782325744628906e-05,
      "model_forward_time": 1.7316575050354004,
      "step": 97889
    },
    {
      "epoch": 2.2782325744628906e-05,
      "step": 97889,
      "training_step_time": 10.183566570281982
    },
    {
      "epoch": 2.2783279418945313e-05,
      "grad_norm": 0.09270837903022766,
      "learning_rate": 1.2178467176505726e-08,
      "loss": 0.0547,
      "step": 97890
    },
    {
      "epoch": 2.2783279418945313e-05,
      "model_forward_time": 1.7312321662902832,
      "step": 97890
    },
    {
      "dynamics_loss_avg": 0.06492204684764147,
      "epoch": 2.2783279418945313e-05,
      "step": 97890
    },
    {
      "action_loss_avg": 0.001391330169280991,
      "epoch": 2.2783279418945313e-05,
      "step": 97890
    },
    {
      "epoch": 2.2783279418945313e-05,
      "step": 97890,
      "training_step_time": 10.491960287094116
    },
    {
      "epoch": 2.2784233093261717e-05,
      "model_forward_time": 1.7418065071105957,
      "step": 97891
    },
    {
      "epoch": 2.2784233093261717e-05,
      "step": 97891,
      "training_step_time": 8.305239200592041
    },
    {
      "epoch": 2.2785186767578124e-05,
      "model_forward_time": 1.7314245700836182,
      "step": 97892
    },
    {
      "epoch": 2.2785186767578124e-05,
      "step": 97892,
      "training_step_time": 8.43541145324707
    },
    {
      "epoch": 2.2786140441894532e-05,
      "model_forward_time": 1.734177827835083,
      "step": 97893
    },
    {
      "epoch": 2.2786140441894532e-05,
      "step": 97893,
      "training_step_time": 8.446887016296387
    },
    {
      "epoch": 2.2787094116210936e-05,
      "model_forward_time": 1.7347443103790283,
      "step": 97894
    },
    {
      "epoch": 2.2787094116210936e-05,
      "step": 97894,
      "training_step_time": 7.156979560852051
    },
    {
      "epoch": 2.2788047790527343e-05,
      "model_forward_time": 1.7327303886413574,
      "step": 97895
    },
    {
      "epoch": 2.2788047790527343e-05,
      "step": 97895,
      "training_step_time": 6.844334602355957
    },
    {
      "epoch": 2.278900146484375e-05,
      "model_forward_time": 1.7317829132080078,
      "step": 97896
    },
    {
      "epoch": 2.278900146484375e-05,
      "step": 97896,
      "training_step_time": 6.850683212280273
    },
    {
      "epoch": 2.2789955139160155e-05,
      "model_forward_time": 1.729490041732788,
      "step": 97897
    },
    {
      "epoch": 2.2789955139160155e-05,
      "step": 97897,
      "training_step_time": 7.461975812911987
    },
    {
      "epoch": 2.2790908813476562e-05,
      "model_forward_time": 1.756338119506836,
      "step": 97898
    },
    {
      "epoch": 2.2790908813476562e-05,
      "step": 97898,
      "training_step_time": 8.664208889007568
    },
    {
      "epoch": 2.279186248779297e-05,
      "model_forward_time": 2.3771631717681885,
      "step": 97899
    },
    {
      "epoch": 2.279186248779297e-05,
      "step": 97899,
      "training_step_time": 8.894108057022095
    },
    {
      "epoch": 2.2792816162109374e-05,
      "grad_norm": 0.10799993574619293,
      "learning_rate": 1.2063405745689206e-08,
      "loss": 0.0546,
      "step": 97900
    },
    {
      "epoch": 2.2792816162109374e-05,
      "model_forward_time": 1.7451629638671875,
      "step": 97900
    },
    {
      "dynamics_loss_avg": 0.04432074036449194,
      "epoch": 2.2792816162109374e-05,
      "step": 97900
    },
    {
      "action_loss_avg": 0.0010399335413239896,
      "epoch": 2.2792816162109374e-05,
      "step": 97900
    },
    {
      "epoch": 2.2792816162109374e-05,
      "step": 97900,
      "training_step_time": 9.221895694732666
    },
    {
      "epoch": 2.279376983642578e-05,
      "model_forward_time": 1.7375497817993164,
      "step": 97901
    },
    {
      "epoch": 2.279376983642578e-05,
      "step": 97901,
      "training_step_time": 7.8247621059417725
    },
    {
      "epoch": 2.279472351074219e-05,
      "model_forward_time": 1.7283954620361328,
      "step": 97902
    },
    {
      "epoch": 2.279472351074219e-05,
      "step": 97902,
      "training_step_time": 8.019305944442749
    },
    {
      "epoch": 2.2795677185058593e-05,
      "model_forward_time": 1.736077070236206,
      "step": 97903
    },
    {
      "epoch": 2.2795677185058593e-05,
      "step": 97903,
      "training_step_time": 7.615700960159302
    },
    {
      "epoch": 2.2796630859375e-05,
      "model_forward_time": 1.733572244644165,
      "step": 97904
    },
    {
      "epoch": 2.2796630859375e-05,
      "step": 97904,
      "training_step_time": 7.180651903152466
    },
    {
      "epoch": 2.2797584533691407e-05,
      "model_forward_time": 1.7336103916168213,
      "step": 97905
    },
    {
      "epoch": 2.2797584533691407e-05,
      "step": 97905,
      "training_step_time": 6.854043245315552
    },
    {
      "epoch": 2.279853820800781e-05,
      "model_forward_time": 1.7314374446868896,
      "step": 97906
    },
    {
      "epoch": 2.279853820800781e-05,
      "step": 97906,
      "training_step_time": 7.911033630371094
    },
    {
      "epoch": 2.279949188232422e-05,
      "model_forward_time": 1.7506179809570312,
      "step": 97907
    },
    {
      "epoch": 2.279949188232422e-05,
      "step": 97907,
      "training_step_time": 8.453993320465088
    },
    {
      "epoch": 2.2800445556640626e-05,
      "model_forward_time": 2.384657144546509,
      "step": 97908
    },
    {
      "epoch": 2.2800445556640626e-05,
      "step": 97908,
      "training_step_time": 10.252671957015991
    },
    {
      "epoch": 2.280139923095703e-05,
      "model_forward_time": 1.7518665790557861,
      "step": 97909
    },
    {
      "epoch": 2.280139923095703e-05,
      "step": 97909,
      "training_step_time": 9.622896194458008
    },
    {
      "epoch": 2.2802352905273438e-05,
      "grad_norm": 0.10384958982467651,
      "learning_rate": 1.194888978811981e-08,
      "loss": 0.054,
      "step": 97910
    },
    {
      "epoch": 2.2802352905273438e-05,
      "model_forward_time": 1.727100133895874,
      "step": 97910
    },
    {
      "dynamics_loss_avg": 0.054471530579030515,
      "epoch": 2.2802352905273438e-05,
      "step": 97910
    },
    {
      "action_loss_avg": 0.0014184893865603954,
      "epoch": 2.2802352905273438e-05,
      "step": 97910
    },
    {
      "epoch": 2.2802352905273438e-05,
      "step": 97910,
      "training_step_time": 7.675999879837036
    },
    {
      "epoch": 2.2803306579589845e-05,
      "model_forward_time": 1.7261543273925781,
      "step": 97911
    },
    {
      "epoch": 2.2803306579589845e-05,
      "step": 97911,
      "training_step_time": 7.952419281005859
    },
    {
      "epoch": 2.280426025390625e-05,
      "model_forward_time": 1.734804630279541,
      "step": 97912
    },
    {
      "epoch": 2.280426025390625e-05,
      "step": 97912,
      "training_step_time": 7.591105222702026
    },
    {
      "epoch": 2.2805213928222656e-05,
      "model_forward_time": 1.736027717590332,
      "step": 97913
    },
    {
      "epoch": 2.2805213928222656e-05,
      "step": 97913,
      "training_step_time": 7.153333902359009
    },
    {
      "epoch": 2.2806167602539064e-05,
      "model_forward_time": 1.7284080982208252,
      "step": 97914
    },
    {
      "epoch": 2.2806167602539064e-05,
      "step": 97914,
      "training_step_time": 6.8954150676727295
    },
    {
      "epoch": 2.2807121276855468e-05,
      "model_forward_time": 1.7357213497161865,
      "step": 97915
    },
    {
      "epoch": 2.2807121276855468e-05,
      "step": 97915,
      "training_step_time": 7.368995666503906
    },
    {
      "epoch": 2.2808074951171875e-05,
      "model_forward_time": 1.769629716873169,
      "step": 97916
    },
    {
      "epoch": 2.2808074951171875e-05,
      "step": 97916,
      "training_step_time": 7.76854944229126
    },
    {
      "epoch": 2.2809028625488283e-05,
      "model_forward_time": 2.4006247520446777,
      "step": 97917
    },
    {
      "epoch": 2.2809028625488283e-05,
      "step": 97917,
      "training_step_time": 10.494319200515747
    },
    {
      "epoch": 2.2809982299804687e-05,
      "model_forward_time": 1.7527201175689697,
      "step": 97918
    },
    {
      "epoch": 2.2809982299804687e-05,
      "step": 97918,
      "training_step_time": 8.522088289260864
    },
    {
      "epoch": 2.2810935974121094e-05,
      "model_forward_time": 1.7337732315063477,
      "step": 97919
    },
    {
      "epoch": 2.2810935974121094e-05,
      "step": 97919,
      "training_step_time": 7.822100639343262
    },
    {
      "epoch": 2.28118896484375e-05,
      "grad_norm": 0.12769867479801178,
      "learning_rate": 1.1834919316320304e-08,
      "loss": 0.0553,
      "step": 97920
    },
    {
      "epoch": 2.28118896484375e-05,
      "model_forward_time": 1.7256460189819336,
      "step": 97920
    },
    {
      "dynamics_loss_avg": 0.042885865364223716,
      "epoch": 2.28118896484375e-05,
      "step": 97920
    },
    {
      "action_loss_avg": 0.0011215320351766422,
      "epoch": 2.28118896484375e-05,
      "step": 97920
    },
    {
      "epoch": 2.28118896484375e-05,
      "step": 97920,
      "training_step_time": 7.887706756591797
    },
    {
      "epoch": 2.2812843322753906e-05,
      "model_forward_time": 1.7394211292266846,
      "step": 97921
    },
    {
      "epoch": 2.2812843322753906e-05,
      "step": 97921,
      "training_step_time": 7.610645294189453
    },
    {
      "epoch": 2.2813796997070313e-05,
      "model_forward_time": 1.738708734512329,
      "step": 97922
    },
    {
      "epoch": 2.2813796997070313e-05,
      "step": 97922,
      "training_step_time": 7.137955188751221
    },
    {
      "epoch": 2.2814750671386717e-05,
      "model_forward_time": 1.7356646060943604,
      "step": 97923
    },
    {
      "epoch": 2.2814750671386717e-05,
      "step": 97923,
      "training_step_time": 6.846417188644409
    },
    {
      "epoch": 2.2815704345703125e-05,
      "model_forward_time": 1.7346513271331787,
      "step": 97924
    },
    {
      "epoch": 2.2815704345703125e-05,
      "step": 97924,
      "training_step_time": 7.419908046722412
    },
    {
      "epoch": 2.2816658020019532e-05,
      "model_forward_time": 1.7285044193267822,
      "step": 97925
    },
    {
      "epoch": 2.2816658020019532e-05,
      "step": 97925,
      "training_step_time": 7.76305890083313
    },
    {
      "epoch": 2.2817611694335936e-05,
      "model_forward_time": 2.412139654159546,
      "step": 97926
    },
    {
      "epoch": 2.2817611694335936e-05,
      "step": 97926,
      "training_step_time": 9.646606922149658
    },
    {
      "epoch": 2.2818565368652343e-05,
      "model_forward_time": 1.740192174911499,
      "step": 97927
    },
    {
      "epoch": 2.2818565368652343e-05,
      "step": 97927,
      "training_step_time": 9.724241018295288
    },
    {
      "epoch": 2.281951904296875e-05,
      "model_forward_time": 1.734590768814087,
      "step": 97928
    },
    {
      "epoch": 2.281951904296875e-05,
      "step": 97928,
      "training_step_time": 7.816017150878906
    },
    {
      "epoch": 2.2820472717285155e-05,
      "model_forward_time": 1.7339286804199219,
      "step": 97929
    },
    {
      "epoch": 2.2820472717285155e-05,
      "step": 97929,
      "training_step_time": 8.444993734359741
    },
    {
      "epoch": 2.2821426391601562e-05,
      "grad_norm": 0.07982517033815384,
      "learning_rate": 1.1721494342754048e-08,
      "loss": 0.0531,
      "step": 97930
    },
    {
      "epoch": 2.2821426391601562e-05,
      "model_forward_time": 1.732391595840454,
      "step": 97930
    },
    {
      "dynamics_loss_avg": 0.04008965622633696,
      "epoch": 2.2821426391601562e-05,
      "step": 97930
    },
    {
      "action_loss_avg": 0.0007962846109876409,
      "epoch": 2.2821426391601562e-05,
      "step": 97930
    },
    {
      "epoch": 2.2821426391601562e-05,
      "step": 97930,
      "training_step_time": 7.191664695739746
    },
    {
      "epoch": 2.282238006591797e-05,
      "model_forward_time": 1.732576608657837,
      "step": 97931
    },
    {
      "epoch": 2.282238006591797e-05,
      "step": 97931,
      "training_step_time": 7.60212779045105
    },
    {
      "epoch": 2.2823333740234374e-05,
      "model_forward_time": 1.7301733493804932,
      "step": 97932
    },
    {
      "epoch": 2.2823333740234374e-05,
      "step": 97932,
      "training_step_time": 6.896539211273193
    },
    {
      "epoch": 2.282428741455078e-05,
      "model_forward_time": 1.7374298572540283,
      "step": 97933
    },
    {
      "epoch": 2.282428741455078e-05,
      "step": 97933,
      "training_step_time": 7.382851600646973
    },
    {
      "epoch": 2.282524108886719e-05,
      "model_forward_time": 1.743100881576538,
      "step": 97934
    },
    {
      "epoch": 2.282524108886719e-05,
      "step": 97934,
      "training_step_time": 8.669313907623291
    },
    {
      "epoch": 2.2826194763183593e-05,
      "model_forward_time": 2.358342170715332,
      "step": 97935
    },
    {
      "epoch": 2.2826194763183593e-05,
      "step": 97935,
      "training_step_time": 11.112676858901978
    },
    {
      "epoch": 2.28271484375e-05,
      "model_forward_time": 1.739499568939209,
      "step": 97936
    },
    {
      "epoch": 2.28271484375e-05,
      "step": 97936,
      "training_step_time": 8.60552716255188
    },
    {
      "epoch": 2.2828102111816407e-05,
      "model_forward_time": 1.7524707317352295,
      "step": 97937
    },
    {
      "epoch": 2.2828102111816407e-05,
      "step": 97937,
      "training_step_time": 9.211886882781982
    },
    {
      "epoch": 2.282905578613281e-05,
      "model_forward_time": 1.7499644756317139,
      "step": 97938
    },
    {
      "epoch": 2.282905578613281e-05,
      "step": 97938,
      "training_step_time": 8.623502254486084
    },
    {
      "epoch": 2.283000946044922e-05,
      "model_forward_time": 1.7395014762878418,
      "step": 97939
    },
    {
      "epoch": 2.283000946044922e-05,
      "step": 97939,
      "training_step_time": 7.594214200973511
    },
    {
      "epoch": 2.2830963134765626e-05,
      "grad_norm": 0.08727636188268661,
      "learning_rate": 1.1608614879825564e-08,
      "loss": 0.0534,
      "step": 97940
    },
    {
      "epoch": 2.2830963134765626e-05,
      "model_forward_time": 1.7315137386322021,
      "step": 97940
    },
    {
      "dynamics_loss_avg": 0.052131930366158485,
      "epoch": 2.2830963134765626e-05,
      "step": 97940
    },
    {
      "action_loss_avg": 0.0010203479614574462,
      "epoch": 2.2830963134765626e-05,
      "step": 97940
    },
    {
      "epoch": 2.2830963134765626e-05,
      "step": 97940,
      "training_step_time": 7.623161315917969
    },
    {
      "epoch": 2.283191680908203e-05,
      "model_forward_time": 1.7467072010040283,
      "step": 97941
    },
    {
      "epoch": 2.283191680908203e-05,
      "step": 97941,
      "training_step_time": 6.872645854949951
    },
    {
      "epoch": 2.2832870483398438e-05,
      "model_forward_time": 1.7400872707366943,
      "step": 97942
    },
    {
      "epoch": 2.2832870483398438e-05,
      "step": 97942,
      "training_step_time": 7.316690683364868
    },
    {
      "epoch": 2.2833824157714845e-05,
      "model_forward_time": 1.732907772064209,
      "step": 97943
    },
    {
      "epoch": 2.2833824157714845e-05,
      "step": 97943,
      "training_step_time": 8.582770109176636
    },
    {
      "epoch": 2.283477783203125e-05,
      "model_forward_time": 1.7572054862976074,
      "step": 97944
    },
    {
      "epoch": 2.283477783203125e-05,
      "step": 97944,
      "training_step_time": 9.365061521530151
    },
    {
      "epoch": 2.2835731506347657e-05,
      "model_forward_time": 2.39345383644104,
      "step": 97945
    },
    {
      "epoch": 2.2835731506347657e-05,
      "step": 97945,
      "training_step_time": 7.884247541427612
    },
    {
      "epoch": 2.2836685180664064e-05,
      "model_forward_time": 1.745917797088623,
      "step": 97946
    },
    {
      "epoch": 2.2836685180664064e-05,
      "step": 97946,
      "training_step_time": 9.61362361907959
    },
    {
      "epoch": 2.2837638854980468e-05,
      "model_forward_time": 1.7370131015777588,
      "step": 97947
    },
    {
      "epoch": 2.2837638854980468e-05,
      "step": 97947,
      "training_step_time": 8.575128078460693
    },
    {
      "epoch": 2.2838592529296875e-05,
      "model_forward_time": 1.7267484664916992,
      "step": 97948
    },
    {
      "epoch": 2.2838592529296875e-05,
      "step": 97948,
      "training_step_time": 7.590923547744751
    },
    {
      "epoch": 2.2839546203613283e-05,
      "model_forward_time": 1.7378334999084473,
      "step": 97949
    },
    {
      "epoch": 2.2839546203613283e-05,
      "step": 97949,
      "training_step_time": 7.639478921890259
    },
    {
      "epoch": 2.2840499877929687e-05,
      "grad_norm": 0.10840973258018494,
      "learning_rate": 1.1496280939879422e-08,
      "loss": 0.0551,
      "step": 97950
    },
    {
      "epoch": 2.2840499877929687e-05,
      "model_forward_time": 1.7365808486938477,
      "step": 97950
    },
    {
      "dynamics_loss_avg": 0.05965480301529169,
      "epoch": 2.2840499877929687e-05,
      "step": 97950
    },
    {
      "action_loss_avg": 0.001930264208931476,
      "epoch": 2.2840499877929687e-05,
      "step": 97950
    },
    {
      "epoch": 2.2840499877929687e-05,
      "step": 97950,
      "training_step_time": 7.1531195640563965
    },
    {
      "epoch": 2.2841453552246094e-05,
      "model_forward_time": 1.7453374862670898,
      "step": 97951
    },
    {
      "epoch": 2.2841453552246094e-05,
      "step": 97951,
      "training_step_time": 7.093111515045166
    },
    {
      "epoch": 2.28424072265625e-05,
      "model_forward_time": 1.7386441230773926,
      "step": 97952
    },
    {
      "epoch": 2.28424072265625e-05,
      "step": 97952,
      "training_step_time": 7.731860160827637
    },
    {
      "epoch": 2.2843360900878906e-05,
      "model_forward_time": 1.7720215320587158,
      "step": 97953
    },
    {
      "epoch": 2.2843360900878906e-05,
      "step": 97953,
      "training_step_time": 9.361019134521484
    },
    {
      "epoch": 2.2844314575195313e-05,
      "model_forward_time": 2.411034345626831,
      "step": 97954
    },
    {
      "epoch": 2.2844314575195313e-05,
      "step": 97954,
      "training_step_time": 9.033096075057983
    },
    {
      "epoch": 2.2845268249511717e-05,
      "model_forward_time": 1.771104335784912,
      "step": 97955
    },
    {
      "epoch": 2.2845268249511717e-05,
      "step": 97955,
      "training_step_time": 9.846045017242432
    },
    {
      "epoch": 2.2846221923828125e-05,
      "model_forward_time": 1.7469840049743652,
      "step": 97956
    },
    {
      "epoch": 2.2846221923828125e-05,
      "step": 97956,
      "training_step_time": 8.613766431808472
    },
    {
      "epoch": 2.2847175598144532e-05,
      "model_forward_time": 1.7558295726776123,
      "step": 97957
    },
    {
      "epoch": 2.2847175598144532e-05,
      "step": 97957,
      "training_step_time": 7.773849010467529
    },
    {
      "epoch": 2.2848129272460936e-05,
      "model_forward_time": 1.7451112270355225,
      "step": 97958
    },
    {
      "epoch": 2.2848129272460936e-05,
      "step": 97958,
      "training_step_time": 9.171016216278076
    },
    {
      "epoch": 2.2849082946777343e-05,
      "model_forward_time": 1.7381985187530518,
      "step": 97959
    },
    {
      "epoch": 2.2849082946777343e-05,
      "step": 97959,
      "training_step_time": 7.173303842544556
    },
    {
      "epoch": 2.285003662109375e-05,
      "grad_norm": 0.14048655331134796,
      "learning_rate": 1.1384492535199688e-08,
      "loss": 0.0553,
      "step": 97960
    },
    {
      "epoch": 2.285003662109375e-05,
      "model_forward_time": 1.7348806858062744,
      "step": 97960
    },
    {
      "dynamics_loss_avg": 0.06330666951835155,
      "epoch": 2.285003662109375e-05,
      "step": 97960
    },
    {
      "action_loss_avg": 0.001112069920054637,
      "epoch": 2.285003662109375e-05,
      "step": 97960
    },
    {
      "epoch": 2.285003662109375e-05,
      "step": 97960,
      "training_step_time": 7.095712184906006
    },
    {
      "epoch": 2.2850990295410155e-05,
      "model_forward_time": 1.7621862888336182,
      "step": 97961
    },
    {
      "epoch": 2.2850990295410155e-05,
      "step": 97961,
      "training_step_time": 8.201709508895874
    },
    {
      "epoch": 2.2851943969726562e-05,
      "model_forward_time": 1.7382361888885498,
      "step": 97962
    },
    {
      "epoch": 2.2851943969726562e-05,
      "step": 97962,
      "training_step_time": 9.406362533569336
    },
    {
      "epoch": 2.285289764404297e-05,
      "model_forward_time": 2.4098451137542725,
      "step": 97963
    },
    {
      "epoch": 2.285289764404297e-05,
      "step": 97963,
      "training_step_time": 8.516228675842285
    },
    {
      "epoch": 2.2853851318359374e-05,
      "model_forward_time": 1.7455987930297852,
      "step": 97964
    },
    {
      "epoch": 2.2853851318359374e-05,
      "step": 97964,
      "training_step_time": 10.22826099395752
    },
    {
      "epoch": 2.285480499267578e-05,
      "model_forward_time": 1.751720666885376,
      "step": 97965
    },
    {
      "epoch": 2.285480499267578e-05,
      "step": 97965,
      "training_step_time": 8.367191314697266
    },
    {
      "epoch": 2.285575866699219e-05,
      "model_forward_time": 1.730802297592163,
      "step": 97966
    },
    {
      "epoch": 2.285575866699219e-05,
      "step": 97966,
      "training_step_time": 7.917552947998047
    },
    {
      "epoch": 2.2856712341308593e-05,
      "model_forward_time": 1.740915060043335,
      "step": 97967
    },
    {
      "epoch": 2.2856712341308593e-05,
      "step": 97967,
      "training_step_time": 7.269564151763916
    },
    {
      "epoch": 2.2857666015625e-05,
      "model_forward_time": 1.7378499507904053,
      "step": 97968
    },
    {
      "epoch": 2.2857666015625e-05,
      "step": 97968,
      "training_step_time": 7.672806739807129
    },
    {
      "epoch": 2.2858619689941407e-05,
      "model_forward_time": 1.7345595359802246,
      "step": 97969
    },
    {
      "epoch": 2.2858619689941407e-05,
      "step": 97969,
      "training_step_time": 7.006107807159424
    },
    {
      "epoch": 2.285957336425781e-05,
      "grad_norm": 0.15597020089626312,
      "learning_rate": 1.1273249678011577e-08,
      "loss": 0.0548,
      "step": 97970
    },
    {
      "epoch": 2.285957336425781e-05,
      "model_forward_time": 1.7189838886260986,
      "step": 97970
    },
    {
      "dynamics_loss_avg": 0.0641821863129735,
      "epoch": 2.285957336425781e-05,
      "step": 97970
    },
    {
      "action_loss_avg": 0.0012824257602915168,
      "epoch": 2.285957336425781e-05,
      "step": 97970
    },
    {
      "epoch": 2.285957336425781e-05,
      "step": 97970,
      "training_step_time": 7.58867335319519
    },
    {
      "epoch": 2.286052703857422e-05,
      "model_forward_time": 1.7294275760650635,
      "step": 97971
    },
    {
      "epoch": 2.286052703857422e-05,
      "step": 97971,
      "training_step_time": 8.818498611450195
    },
    {
      "epoch": 2.2861480712890626e-05,
      "model_forward_time": 2.360564947128296,
      "step": 97972
    },
    {
      "epoch": 2.2861480712890626e-05,
      "step": 97972,
      "training_step_time": 7.752562522888184
    },
    {
      "epoch": 2.286243438720703e-05,
      "model_forward_time": 1.7258045673370361,
      "step": 97973
    },
    {
      "epoch": 2.286243438720703e-05,
      "step": 97973,
      "training_step_time": 7.780200719833374
    },
    {
      "epoch": 2.2863388061523438e-05,
      "model_forward_time": 1.7483799457550049,
      "step": 97974
    },
    {
      "epoch": 2.2863388061523438e-05,
      "step": 97974,
      "training_step_time": 7.941140651702881
    },
    {
      "epoch": 2.2864341735839845e-05,
      "model_forward_time": 1.7329576015472412,
      "step": 97975
    },
    {
      "epoch": 2.2864341735839845e-05,
      "step": 97975,
      "training_step_time": 7.975992441177368
    },
    {
      "epoch": 2.286529541015625e-05,
      "model_forward_time": 1.733564853668213,
      "step": 97976
    },
    {
      "epoch": 2.286529541015625e-05,
      "step": 97976,
      "training_step_time": 7.212053537368774
    },
    {
      "epoch": 2.2866249084472657e-05,
      "model_forward_time": 1.7282757759094238,
      "step": 97977
    },
    {
      "epoch": 2.2866249084472657e-05,
      "step": 97977,
      "training_step_time": 7.643165349960327
    },
    {
      "epoch": 2.2867202758789064e-05,
      "model_forward_time": 1.734386920928955,
      "step": 97978
    },
    {
      "epoch": 2.2867202758789064e-05,
      "step": 97978,
      "training_step_time": 7.056577682495117
    },
    {
      "epoch": 2.2868156433105468e-05,
      "model_forward_time": 1.7338237762451172,
      "step": 97979
    },
    {
      "epoch": 2.2868156433105468e-05,
      "step": 97979,
      "training_step_time": 7.376452922821045
    },
    {
      "epoch": 2.2869110107421875e-05,
      "grad_norm": 0.08670169860124588,
      "learning_rate": 1.1162552380480362e-08,
      "loss": 0.054,
      "step": 97980
    },
    {
      "epoch": 2.2869110107421875e-05,
      "model_forward_time": 1.7215838432312012,
      "step": 97980
    },
    {
      "dynamics_loss_avg": 0.04891760051250458,
      "epoch": 2.2869110107421875e-05,
      "step": 97980
    },
    {
      "action_loss_avg": 0.0011208312585949897,
      "epoch": 2.2869110107421875e-05,
      "step": 97980
    },
    {
      "epoch": 2.2869110107421875e-05,
      "step": 97980,
      "training_step_time": 7.768434286117554
    },
    {
      "epoch": 2.2870063781738283e-05,
      "model_forward_time": 2.36389422416687,
      "step": 97981
    },
    {
      "epoch": 2.2870063781738283e-05,
      "step": 97981,
      "training_step_time": 7.7455222606658936
    },
    {
      "epoch": 2.2871017456054687e-05,
      "model_forward_time": 1.7187120914459229,
      "step": 97982
    },
    {
      "epoch": 2.2871017456054687e-05,
      "step": 97982,
      "training_step_time": 8.993344068527222
    },
    {
      "epoch": 2.2871971130371094e-05,
      "model_forward_time": 1.743811845779419,
      "step": 97983
    },
    {
      "epoch": 2.2871971130371094e-05,
      "step": 97983,
      "training_step_time": 8.425787925720215
    },
    {
      "epoch": 2.28729248046875e-05,
      "model_forward_time": 1.7263829708099365,
      "step": 97984
    },
    {
      "epoch": 2.28729248046875e-05,
      "step": 97984,
      "training_step_time": 8.100141525268555
    },
    {
      "epoch": 2.2873878479003906e-05,
      "model_forward_time": 1.7281739711761475,
      "step": 97985
    },
    {
      "epoch": 2.2873878479003906e-05,
      "step": 97985,
      "training_step_time": 7.182522535324097
    },
    {
      "epoch": 2.2874832153320313e-05,
      "model_forward_time": 1.720848798751831,
      "step": 97986
    },
    {
      "epoch": 2.2874832153320313e-05,
      "step": 97986,
      "training_step_time": 7.6502110958099365
    },
    {
      "epoch": 2.2875785827636717e-05,
      "model_forward_time": 1.725736141204834,
      "step": 97987
    },
    {
      "epoch": 2.2875785827636717e-05,
      "step": 97987,
      "training_step_time": 7.458703279495239
    },
    {
      "epoch": 2.2876739501953125e-05,
      "model_forward_time": 1.7241261005401611,
      "step": 97988
    },
    {
      "epoch": 2.2876739501953125e-05,
      "step": 97988,
      "training_step_time": 7.309969425201416
    },
    {
      "epoch": 2.2877693176269532e-05,
      "model_forward_time": 1.729475975036621,
      "step": 97989
    },
    {
      "epoch": 2.2877693176269532e-05,
      "step": 97989,
      "training_step_time": 7.736782550811768
    },
    {
      "epoch": 2.2878646850585936e-05,
      "grad_norm": 0.08001353591680527,
      "learning_rate": 1.105240065471136e-08,
      "loss": 0.0547,
      "step": 97990
    },
    {
      "epoch": 2.2878646850585936e-05,
      "model_forward_time": 1.728745937347412,
      "step": 97990
    },
    {
      "dynamics_loss_avg": 0.05371890179812908,
      "epoch": 2.2878646850585936e-05,
      "step": 97990
    },
    {
      "action_loss_avg": 0.001430538683780469,
      "epoch": 2.2878646850585936e-05,
      "step": 97990
    },
    {
      "epoch": 2.2878646850585936e-05,
      "step": 97990,
      "training_step_time": 8.379333019256592
    },
    {
      "epoch": 2.2879600524902343e-05,
      "model_forward_time": 1.7188823223114014,
      "step": 97991
    },
    {
      "epoch": 2.2879600524902343e-05,
      "step": 97991,
      "training_step_time": 8.645557403564453
    },
    {
      "epoch": 2.288055419921875e-05,
      "model_forward_time": 1.7331044673919678,
      "step": 97992
    },
    {
      "epoch": 2.288055419921875e-05,
      "step": 97992,
      "training_step_time": 7.909153461456299
    },
    {
      "epoch": 2.2881507873535155e-05,
      "model_forward_time": 1.721059799194336,
      "step": 97993
    },
    {
      "epoch": 2.2881507873535155e-05,
      "step": 97993,
      "training_step_time": 7.940483331680298
    },
    {
      "epoch": 2.2882461547851562e-05,
      "model_forward_time": 1.7203068733215332,
      "step": 97994
    },
    {
      "epoch": 2.2882461547851562e-05,
      "step": 97994,
      "training_step_time": 7.64262580871582
    },
    {
      "epoch": 2.288341522216797e-05,
      "model_forward_time": 1.7274754047393799,
      "step": 97995
    },
    {
      "epoch": 2.288341522216797e-05,
      "step": 97995,
      "training_step_time": 7.273046970367432
    },
    {
      "epoch": 2.2884368896484374e-05,
      "model_forward_time": 1.7239594459533691,
      "step": 97996
    },
    {
      "epoch": 2.2884368896484374e-05,
      "step": 97996,
      "training_step_time": 7.666869163513184
    },
    {
      "epoch": 2.288532257080078e-05,
      "model_forward_time": 1.7192163467407227,
      "step": 97997
    },
    {
      "epoch": 2.288532257080078e-05,
      "step": 97997,
      "training_step_time": 7.288516521453857
    },
    {
      "epoch": 2.288627624511719e-05,
      "model_forward_time": 1.7580621242523193,
      "step": 97998
    },
    {
      "epoch": 2.288627624511719e-05,
      "step": 97998,
      "training_step_time": 7.925912380218506
    },
    {
      "epoch": 2.2887229919433593e-05,
      "model_forward_time": 1.7756874561309814,
      "step": 97999
    },
    {
      "epoch": 2.2887229919433593e-05,
      "step": 97999,
      "training_step_time": 8.597474575042725
    },
    {
      "epoch": 2.288818359375e-05,
      "grad_norm": 0.11001840978860855,
      "learning_rate": 1.0942794512751598e-08,
      "loss": 0.0546,
      "step": 98000
    },
    {
      "epoch": 2.288818359375e-05,
      "model_forward_time": 1.8127467632293701,
      "step": 98000
    },
    {
      "dynamics_loss_avg": 0.05336298234760761,
      "epoch": 2.288818359375e-05,
      "step": 98000
    },
    {
      "action_loss_avg": 0.001034191934741102,
      "epoch": 2.288818359375e-05,
      "step": 98000
    },
    {
      "epoch": 2.288818359375e-05,
      "step": 98000,
      "training_step_time": 10.819785118103027
    },
    {
      "epoch": 2.2889137268066407e-05,
      "model_forward_time": 1.7376182079315186,
      "step": 98001
    },
    {
      "epoch": 2.2889137268066407e-05,
      "step": 98001,
      "training_step_time": 9.606062173843384
    },
    {
      "epoch": 2.289009094238281e-05,
      "model_forward_time": 1.784369707107544,
      "step": 98002
    },
    {
      "epoch": 2.289009094238281e-05,
      "step": 98002,
      "training_step_time": 8.650772094726562
    },
    {
      "epoch": 2.289104461669922e-05,
      "model_forward_time": 1.7384741306304932,
      "step": 98003
    },
    {
      "epoch": 2.289104461669922e-05,
      "step": 98003,
      "training_step_time": 8.371440172195435
    },
    {
      "epoch": 2.2891998291015626e-05,
      "model_forward_time": 1.7631173133850098,
      "step": 98004
    },
    {
      "epoch": 2.2891998291015626e-05,
      "step": 98004,
      "training_step_time": 8.255722999572754
    },
    {
      "epoch": 2.289295196533203e-05,
      "model_forward_time": 1.74949049949646,
      "step": 98005
    },
    {
      "epoch": 2.289295196533203e-05,
      "step": 98005,
      "training_step_time": 7.613887786865234
    },
    {
      "epoch": 2.2893905639648438e-05,
      "model_forward_time": 1.73698091506958,
      "step": 98006
    },
    {
      "epoch": 2.2893905639648438e-05,
      "step": 98006,
      "training_step_time": 8.00451111793518
    },
    {
      "epoch": 2.2894859313964845e-05,
      "model_forward_time": 1.7413806915283203,
      "step": 98007
    },
    {
      "epoch": 2.2894859313964845e-05,
      "step": 98007,
      "training_step_time": 9.489992380142212
    },
    {
      "epoch": 2.289581298828125e-05,
      "model_forward_time": 3.003647804260254,
      "step": 98008
    },
    {
      "epoch": 2.289581298828125e-05,
      "step": 98008,
      "training_step_time": 9.35364294052124
    },
    {
      "epoch": 2.2896766662597657e-05,
      "model_forward_time": 1.7379229068756104,
      "step": 98009
    },
    {
      "epoch": 2.2896766662597657e-05,
      "step": 98009,
      "training_step_time": 9.966748714447021
    },
    {
      "epoch": 2.2897720336914064e-05,
      "grad_norm": 0.12437039613723755,
      "learning_rate": 1.0833733966587046e-08,
      "loss": 0.0552,
      "step": 98010
    },
    {
      "epoch": 2.2897720336914064e-05,
      "model_forward_time": 1.730189323425293,
      "step": 98010
    },
    {
      "dynamics_loss_avg": 0.05406659916043281,
      "epoch": 2.2897720336914064e-05,
      "step": 98010
    },
    {
      "action_loss_avg": 0.0008801745469099842,
      "epoch": 2.2897720336914064e-05,
      "step": 98010
    },
    {
      "epoch": 2.2897720336914064e-05,
      "step": 98010,
      "training_step_time": 8.2579824924469
    },
    {
      "epoch": 2.2898674011230468e-05,
      "model_forward_time": 1.7694270610809326,
      "step": 98011
    },
    {
      "epoch": 2.2898674011230468e-05,
      "step": 98011,
      "training_step_time": 9.405850410461426
    },
    {
      "epoch": 2.2899627685546875e-05,
      "model_forward_time": 1.7444078922271729,
      "step": 98012
    },
    {
      "epoch": 2.2899627685546875e-05,
      "step": 98012,
      "training_step_time": 7.881477117538452
    },
    {
      "epoch": 2.2900581359863283e-05,
      "model_forward_time": 1.731332778930664,
      "step": 98013
    },
    {
      "epoch": 2.2900581359863283e-05,
      "step": 98013,
      "training_step_time": 7.622942209243774
    },
    {
      "epoch": 2.2901535034179687e-05,
      "model_forward_time": 1.733055830001831,
      "step": 98014
    },
    {
      "epoch": 2.2901535034179687e-05,
      "step": 98014,
      "training_step_time": 7.667142152786255
    },
    {
      "epoch": 2.2902488708496094e-05,
      "model_forward_time": 1.7150888442993164,
      "step": 98015
    },
    {
      "epoch": 2.2902488708496094e-05,
      "step": 98015,
      "training_step_time": 7.482244253158569
    },
    {
      "epoch": 2.29034423828125e-05,
      "model_forward_time": 1.7309300899505615,
      "step": 98016
    },
    {
      "epoch": 2.29034423828125e-05,
      "step": 98016,
      "training_step_time": 8.469062566757202
    },
    {
      "epoch": 2.2904396057128906e-05,
      "model_forward_time": 1.7392394542694092,
      "step": 98017
    },
    {
      "epoch": 2.2904396057128906e-05,
      "step": 98017,
      "training_step_time": 10.207153797149658
    },
    {
      "epoch": 2.2905349731445313e-05,
      "model_forward_time": 1.7579197883605957,
      "step": 98018
    },
    {
      "epoch": 2.2905349731445313e-05,
      "step": 98018,
      "training_step_time": 9.641162633895874
    },
    {
      "epoch": 2.2906303405761717e-05,
      "model_forward_time": 1.7305681705474854,
      "step": 98019
    },
    {
      "epoch": 2.2906303405761717e-05,
      "step": 98019,
      "training_step_time": 7.964615106582642
    },
    {
      "epoch": 2.2907257080078125e-05,
      "grad_norm": 0.07824139297008514,
      "learning_rate": 1.0725219028143719e-08,
      "loss": 0.0549,
      "step": 98020
    },
    {
      "epoch": 2.2907257080078125e-05,
      "model_forward_time": 1.7549099922180176,
      "step": 98020
    },
    {
      "dynamics_loss_avg": 0.05015307515859604,
      "epoch": 2.2907257080078125e-05,
      "step": 98020
    },
    {
      "action_loss_avg": 0.0010463740938575939,
      "epoch": 2.2907257080078125e-05,
      "step": 98020
    },
    {
      "epoch": 2.2907257080078125e-05,
      "step": 98020,
      "training_step_time": 9.03981065750122
    },
    {
      "epoch": 2.2908210754394532e-05,
      "model_forward_time": 1.725937843322754,
      "step": 98021
    },
    {
      "epoch": 2.2908210754394532e-05,
      "step": 98021,
      "training_step_time": 8.013327360153198
    },
    {
      "epoch": 2.2909164428710936e-05,
      "model_forward_time": 1.7246859073638916,
      "step": 98022
    },
    {
      "epoch": 2.2909164428710936e-05,
      "step": 98022,
      "training_step_time": 7.72612738609314
    },
    {
      "epoch": 2.2910118103027343e-05,
      "model_forward_time": 1.7332608699798584,
      "step": 98023
    },
    {
      "epoch": 2.2910118103027343e-05,
      "step": 98023,
      "training_step_time": 7.749238014221191
    },
    {
      "epoch": 2.291107177734375e-05,
      "model_forward_time": 1.7300653457641602,
      "step": 98024
    },
    {
      "epoch": 2.291107177734375e-05,
      "step": 98024,
      "training_step_time": 7.5102269649505615
    },
    {
      "epoch": 2.2912025451660155e-05,
      "model_forward_time": 1.7268092632293701,
      "step": 98025
    },
    {
      "epoch": 2.2912025451660155e-05,
      "step": 98025,
      "training_step_time": 7.786769866943359
    },
    {
      "epoch": 2.2912979125976562e-05,
      "model_forward_time": 1.7263026237487793,
      "step": 98026
    },
    {
      "epoch": 2.2912979125976562e-05,
      "step": 98026,
      "training_step_time": 8.701286554336548
    },
    {
      "epoch": 2.291393280029297e-05,
      "model_forward_time": 2.3752992153167725,
      "step": 98027
    },
    {
      "epoch": 2.291393280029297e-05,
      "step": 98027,
      "training_step_time": 8.407783508300781
    },
    {
      "epoch": 2.2914886474609374e-05,
      "model_forward_time": 1.7251620292663574,
      "step": 98028
    },
    {
      "epoch": 2.2914886474609374e-05,
      "step": 98028,
      "training_step_time": 8.903932809829712
    },
    {
      "epoch": 2.291584014892578e-05,
      "model_forward_time": 1.7496132850646973,
      "step": 98029
    },
    {
      "epoch": 2.291584014892578e-05,
      "step": 98029,
      "training_step_time": 9.61007833480835
    },
    {
      "epoch": 2.291679382324219e-05,
      "grad_norm": 0.5379921197891235,
      "learning_rate": 1.0617249709289346e-08,
      "loss": 0.0537,
      "step": 98030
    },
    {
      "epoch": 2.291679382324219e-05,
      "model_forward_time": 1.7210426330566406,
      "step": 98030
    },
    {
      "dynamics_loss_avg": 0.06225271159783006,
      "epoch": 2.291679382324219e-05,
      "step": 98030
    },
    {
      "action_loss_avg": 0.003265701635973528,
      "epoch": 2.291679382324219e-05,
      "step": 98030
    },
    {
      "epoch": 2.291679382324219e-05,
      "step": 98030,
      "training_step_time": 8.039325952529907
    },
    {
      "epoch": 2.2917747497558593e-05,
      "model_forward_time": 1.7407069206237793,
      "step": 98031
    },
    {
      "epoch": 2.2917747497558593e-05,
      "step": 98031,
      "training_step_time": 7.723198652267456
    },
    {
      "epoch": 2.2918701171875e-05,
      "model_forward_time": 1.7294116020202637,
      "step": 98032
    },
    {
      "epoch": 2.2918701171875e-05,
      "step": 98032,
      "training_step_time": 7.715043067932129
    },
    {
      "epoch": 2.2919654846191407e-05,
      "model_forward_time": 1.7286138534545898,
      "step": 98033
    },
    {
      "epoch": 2.2919654846191407e-05,
      "step": 98033,
      "training_step_time": 7.472999095916748
    },
    {
      "epoch": 2.292060852050781e-05,
      "model_forward_time": 1.728907585144043,
      "step": 98034
    },
    {
      "epoch": 2.292060852050781e-05,
      "step": 98034,
      "training_step_time": 7.716996192932129
    },
    {
      "epoch": 2.292156219482422e-05,
      "model_forward_time": 1.7373907566070557,
      "step": 98035
    },
    {
      "epoch": 2.292156219482422e-05,
      "step": 98035,
      "training_step_time": 7.759660482406616
    },
    {
      "epoch": 2.2922515869140626e-05,
      "model_forward_time": 2.35063099861145,
      "step": 98036
    },
    {
      "epoch": 2.2922515869140626e-05,
      "step": 98036,
      "training_step_time": 10.495934009552002
    },
    {
      "epoch": 2.292346954345703e-05,
      "model_forward_time": 1.731269359588623,
      "step": 98037
    },
    {
      "epoch": 2.292346954345703e-05,
      "step": 98037,
      "training_step_time": 9.907498598098755
    },
    {
      "epoch": 2.2924423217773438e-05,
      "model_forward_time": 1.7349071502685547,
      "step": 98038
    },
    {
      "epoch": 2.2924423217773438e-05,
      "step": 98038,
      "training_step_time": 7.764086723327637
    },
    {
      "epoch": 2.2925376892089845e-05,
      "model_forward_time": 1.7320659160614014,
      "step": 98039
    },
    {
      "epoch": 2.2925376892089845e-05,
      "step": 98039,
      "training_step_time": 7.747621774673462
    },
    {
      "epoch": 2.292633056640625e-05,
      "grad_norm": 0.09606257826089859,
      "learning_rate": 1.0509826021831704e-08,
      "loss": 0.0568,
      "step": 98040
    },
    {
      "epoch": 2.292633056640625e-05,
      "model_forward_time": 1.7164404392242432,
      "step": 98040
    },
    {
      "dynamics_loss_avg": 0.0415331494063139,
      "epoch": 2.292633056640625e-05,
      "step": 98040
    },
    {
      "action_loss_avg": 0.0011502383771585301,
      "epoch": 2.292633056640625e-05,
      "step": 98040
    },
    {
      "epoch": 2.292633056640625e-05,
      "step": 98040,
      "training_step_time": 8.388376235961914
    },
    {
      "epoch": 2.2927284240722657e-05,
      "model_forward_time": 1.7294178009033203,
      "step": 98041
    },
    {
      "epoch": 2.2927284240722657e-05,
      "step": 98041,
      "training_step_time": 8.276931047439575
    },
    {
      "epoch": 2.2928237915039064e-05,
      "model_forward_time": 1.735185146331787,
      "step": 98042
    },
    {
      "epoch": 2.2928237915039064e-05,
      "step": 98042,
      "training_step_time": 8.399338722229004
    },
    {
      "epoch": 2.2929191589355468e-05,
      "model_forward_time": 1.7329800128936768,
      "step": 98043
    },
    {
      "epoch": 2.2929191589355468e-05,
      "step": 98043,
      "training_step_time": 7.581134557723999
    },
    {
      "epoch": 2.2930145263671875e-05,
      "model_forward_time": 1.7300596237182617,
      "step": 98044
    },
    {
      "epoch": 2.2930145263671875e-05,
      "step": 98044,
      "training_step_time": 9.38133430480957
    },
    {
      "epoch": 2.2931098937988283e-05,
      "model_forward_time": 2.3746228218078613,
      "step": 98045
    },
    {
      "epoch": 2.2931098937988283e-05,
      "step": 98045,
      "training_step_time": 10.060869216918945
    },
    {
      "epoch": 2.2932052612304687e-05,
      "model_forward_time": 1.7365772724151611,
      "step": 98046
    },
    {
      "epoch": 2.2932052612304687e-05,
      "step": 98046,
      "training_step_time": 9.787558794021606
    },
    {
      "epoch": 2.2933006286621094e-05,
      "model_forward_time": 1.7280056476593018,
      "step": 98047
    },
    {
      "epoch": 2.2933006286621094e-05,
      "step": 98047,
      "training_step_time": 8.45771312713623
    },
    {
      "epoch": 2.29339599609375e-05,
      "model_forward_time": 1.734893560409546,
      "step": 98048
    },
    {
      "epoch": 2.29339599609375e-05,
      "step": 98048,
      "training_step_time": 8.757301568984985
    },
    {
      "epoch": 2.2934913635253906e-05,
      "model_forward_time": 1.7290210723876953,
      "step": 98049
    },
    {
      "epoch": 2.2934913635253906e-05,
      "step": 98049,
      "training_step_time": 7.873792409896851
    },
    {
      "epoch": 2.2935867309570313e-05,
      "grad_norm": 0.07392149418592453,
      "learning_rate": 1.0402947977517508e-08,
      "loss": 0.0565,
      "step": 98050
    },
    {
      "epoch": 2.2935867309570313e-05,
      "model_forward_time": 1.7296159267425537,
      "step": 98050
    },
    {
      "dynamics_loss_avg": 0.0625620735809207,
      "epoch": 2.2935867309570313e-05,
      "step": 98050
    },
    {
      "action_loss_avg": 0.0016617527027847246,
      "epoch": 2.2935867309570313e-05,
      "step": 98050
    },
    {
      "epoch": 2.2935867309570313e-05,
      "step": 98050,
      "training_step_time": 7.219329118728638
    },
    {
      "epoch": 2.2936820983886717e-05,
      "model_forward_time": 1.7423224449157715,
      "step": 98051
    },
    {
      "epoch": 2.2936820983886717e-05,
      "step": 98051,
      "training_step_time": 7.682844400405884
    },
    {
      "epoch": 2.2937774658203125e-05,
      "model_forward_time": 1.7267684936523438,
      "step": 98052
    },
    {
      "epoch": 2.2937774658203125e-05,
      "step": 98052,
      "training_step_time": 7.460126876831055
    },
    {
      "epoch": 2.2938728332519532e-05,
      "model_forward_time": 1.7294888496398926,
      "step": 98053
    },
    {
      "epoch": 2.2938728332519532e-05,
      "step": 98053,
      "training_step_time": 7.759315729141235
    },
    {
      "epoch": 2.2939682006835936e-05,
      "model_forward_time": 2.4070236682891846,
      "step": 98054
    },
    {
      "epoch": 2.2939682006835936e-05,
      "step": 98054,
      "training_step_time": 8.934266567230225
    },
    {
      "epoch": 2.2940635681152343e-05,
      "model_forward_time": 1.7303133010864258,
      "step": 98055
    },
    {
      "epoch": 2.2940635681152343e-05,
      "step": 98055,
      "training_step_time": 8.75506591796875
    },
    {
      "epoch": 2.294158935546875e-05,
      "model_forward_time": 1.729602336883545,
      "step": 98056
    },
    {
      "epoch": 2.294158935546875e-05,
      "step": 98056,
      "training_step_time": 9.574643850326538
    },
    {
      "epoch": 2.2942543029785155e-05,
      "model_forward_time": 1.7531852722167969,
      "step": 98057
    },
    {
      "epoch": 2.2942543029785155e-05,
      "step": 98057,
      "training_step_time": 7.875554800033569
    },
    {
      "epoch": 2.2943496704101562e-05,
      "model_forward_time": 1.7308480739593506,
      "step": 98058
    },
    {
      "epoch": 2.2943496704101562e-05,
      "step": 98058,
      "training_step_time": 8.911133050918579
    },
    {
      "epoch": 2.294445037841797e-05,
      "model_forward_time": 1.736541748046875,
      "step": 98059
    },
    {
      "epoch": 2.294445037841797e-05,
      "step": 98059,
      "training_step_time": 7.169455528259277
    },
    {
      "epoch": 2.2945404052734374e-05,
      "grad_norm": 0.07338564842939377,
      "learning_rate": 1.0296615588035186e-08,
      "loss": 0.0552,
      "step": 98060
    },
    {
      "epoch": 2.2945404052734374e-05,
      "model_forward_time": 1.7287452220916748,
      "step": 98060
    },
    {
      "dynamics_loss_avg": 0.049503292236477134,
      "epoch": 2.2945404052734374e-05,
      "step": 98060
    },
    {
      "action_loss_avg": 0.0009109004779020325,
      "epoch": 2.2945404052734374e-05,
      "step": 98060
    },
    {
      "epoch": 2.2945404052734374e-05,
      "step": 98060,
      "training_step_time": 7.739997148513794
    },
    {
      "epoch": 2.294635772705078e-05,
      "model_forward_time": 1.7249367237091064,
      "step": 98061
    },
    {
      "epoch": 2.294635772705078e-05,
      "step": 98061,
      "training_step_time": 7.5904765129089355
    },
    {
      "epoch": 2.294731140136719e-05,
      "model_forward_time": 1.737403392791748,
      "step": 98062
    },
    {
      "epoch": 2.294731140136719e-05,
      "step": 98062,
      "training_step_time": 8.398041248321533
    },
    {
      "epoch": 2.2948265075683593e-05,
      "model_forward_time": 2.3855600357055664,
      "step": 98063
    },
    {
      "epoch": 2.2948265075683593e-05,
      "step": 98063,
      "training_step_time": 8.83669662475586
    },
    {
      "epoch": 2.294921875e-05,
      "model_forward_time": 1.7488470077514648,
      "step": 98064
    },
    {
      "epoch": 2.294921875e-05,
      "step": 98064,
      "training_step_time": 8.850092887878418
    },
    {
      "epoch": 2.2950172424316407e-05,
      "model_forward_time": 1.740816593170166,
      "step": 98065
    },
    {
      "epoch": 2.2950172424316407e-05,
      "step": 98065,
      "training_step_time": 9.849950313568115
    },
    {
      "epoch": 2.295112609863281e-05,
      "model_forward_time": 1.7678804397583008,
      "step": 98066
    },
    {
      "epoch": 2.295112609863281e-05,
      "step": 98066,
      "training_step_time": 8.698607683181763
    },
    {
      "epoch": 2.295207977294922e-05,
      "model_forward_time": 1.7180936336517334,
      "step": 98067
    },
    {
      "epoch": 2.295207977294922e-05,
      "step": 98067,
      "training_step_time": 7.697733640670776
    },
    {
      "epoch": 2.2953033447265626e-05,
      "model_forward_time": 1.73521089553833,
      "step": 98068
    },
    {
      "epoch": 2.2953033447265626e-05,
      "step": 98068,
      "training_step_time": 7.919668197631836
    },
    {
      "epoch": 2.295398712158203e-05,
      "model_forward_time": 1.7347469329833984,
      "step": 98069
    },
    {
      "epoch": 2.295398712158203e-05,
      "step": 98069,
      "training_step_time": 7.65653395652771
    },
    {
      "epoch": 2.2954940795898438e-05,
      "grad_norm": 0.09524364024400711,
      "learning_rate": 1.0190828865012659e-08,
      "loss": 0.0569,
      "step": 98070
    },
    {
      "epoch": 2.2954940795898438e-05,
      "model_forward_time": 1.7332563400268555,
      "step": 98070
    },
    {
      "dynamics_loss_avg": 0.04618094088509679,
      "epoch": 2.2954940795898438e-05,
      "step": 98070
    },
    {
      "action_loss_avg": 0.0009105741308303549,
      "epoch": 2.2954940795898438e-05,
      "step": 98070
    },
    {
      "epoch": 2.2954940795898438e-05,
      "step": 98070,
      "training_step_time": 7.457624912261963
    },
    {
      "epoch": 2.2955894470214845e-05,
      "model_forward_time": 1.7304787635803223,
      "step": 98071
    },
    {
      "epoch": 2.2955894470214845e-05,
      "step": 98071,
      "training_step_time": 7.669661521911621
    },
    {
      "epoch": 2.295684814453125e-05,
      "model_forward_time": 2.3773040771484375,
      "step": 98072
    },
    {
      "epoch": 2.295684814453125e-05,
      "step": 98072,
      "training_step_time": 9.82567024230957
    },
    {
      "epoch": 2.2957801818847657e-05,
      "model_forward_time": 1.7401533126831055,
      "step": 98073
    },
    {
      "epoch": 2.2957801818847657e-05,
      "step": 98073,
      "training_step_time": 9.178891897201538
    },
    {
      "epoch": 2.2958755493164064e-05,
      "model_forward_time": 1.7285242080688477,
      "step": 98074
    },
    {
      "epoch": 2.2958755493164064e-05,
      "step": 98074,
      "training_step_time": 8.945252418518066
    },
    {
      "epoch": 2.2959709167480468e-05,
      "model_forward_time": 1.7417879104614258,
      "step": 98075
    },
    {
      "epoch": 2.2959709167480468e-05,
      "step": 98075,
      "training_step_time": 9.237237691879272
    },
    {
      "epoch": 2.2960662841796875e-05,
      "model_forward_time": 1.7362642288208008,
      "step": 98076
    },
    {
      "epoch": 2.2960662841796875e-05,
      "step": 98076,
      "training_step_time": 7.551033020019531
    },
    {
      "epoch": 2.2961616516113283e-05,
      "model_forward_time": 1.3565752506256104,
      "step": 98077
    },
    {
      "epoch": 2.2961616516113283e-05,
      "step": 98077,
      "training_step_time": 7.806222677230835
    },
    {
      "epoch": 2.2962570190429687e-05,
      "model_forward_time": 1.735372543334961,
      "step": 98078
    },
    {
      "epoch": 2.2962570190429687e-05,
      "step": 98078,
      "training_step_time": 7.7057366371154785
    },
    {
      "epoch": 2.2963523864746094e-05,
      "model_forward_time": 1.7328064441680908,
      "step": 98079
    },
    {
      "epoch": 2.2963523864746094e-05,
      "step": 98079,
      "training_step_time": 7.491811037063599
    },
    {
      "epoch": 2.29644775390625e-05,
      "grad_norm": 0.09045929461717606,
      "learning_rate": 1.008558782001956e-08,
      "loss": 0.0547,
      "step": 98080
    },
    {
      "epoch": 2.29644775390625e-05,
      "model_forward_time": 1.7392423152923584,
      "step": 98080
    },
    {
      "dynamics_loss_avg": 0.050577953457832336,
      "epoch": 2.29644775390625e-05,
      "step": 98080
    },
    {
      "action_loss_avg": 0.0012804431142285466,
      "epoch": 2.29644775390625e-05,
      "step": 98080
    },
    {
      "epoch": 2.29644775390625e-05,
      "step": 98080,
      "training_step_time": 6.855038166046143
    },
    {
      "epoch": 2.2965431213378906e-05,
      "model_forward_time": 2.391807794570923,
      "step": 98081
    },
    {
      "epoch": 2.2965431213378906e-05,
      "step": 98081,
      "training_step_time": 9.066344976425171
    },
    {
      "epoch": 2.2966384887695313e-05,
      "model_forward_time": 1.7560665607452393,
      "step": 98082
    },
    {
      "epoch": 2.2966384887695313e-05,
      "step": 98082,
      "training_step_time": 7.782648801803589
    },
    {
      "epoch": 2.2967338562011717e-05,
      "model_forward_time": 1.7344775199890137,
      "step": 98083
    },
    {
      "epoch": 2.2967338562011717e-05,
      "step": 98083,
      "training_step_time": 9.417290687561035
    },
    {
      "epoch": 2.2968292236328125e-05,
      "model_forward_time": 1.7382895946502686,
      "step": 98084
    },
    {
      "epoch": 2.2968292236328125e-05,
      "step": 98084,
      "training_step_time": 7.82534646987915
    },
    {
      "epoch": 2.2969245910644532e-05,
      "model_forward_time": 1.7486891746520996,
      "step": 98085
    },
    {
      "epoch": 2.2969245910644532e-05,
      "step": 98085,
      "training_step_time": 8.711575031280518
    },
    {
      "epoch": 2.2970199584960936e-05,
      "model_forward_time": 1.7484958171844482,
      "step": 98086
    },
    {
      "epoch": 2.2970199584960936e-05,
      "step": 98086,
      "training_step_time": 8.496726751327515
    },
    {
      "epoch": 2.2971153259277344e-05,
      "model_forward_time": 1.7266409397125244,
      "step": 98087
    },
    {
      "epoch": 2.2971153259277344e-05,
      "step": 98087,
      "training_step_time": 7.73974084854126
    },
    {
      "epoch": 2.297210693359375e-05,
      "model_forward_time": 1.7327790260314941,
      "step": 98088
    },
    {
      "epoch": 2.297210693359375e-05,
      "step": 98088,
      "training_step_time": 7.677849292755127
    },
    {
      "epoch": 2.2973060607910155e-05,
      "model_forward_time": 1.7381219863891602,
      "step": 98089
    },
    {
      "epoch": 2.2973060607910155e-05,
      "step": 98089,
      "training_step_time": 6.881720066070557
    },
    {
      "epoch": 2.2974014282226562e-05,
      "grad_norm": 0.10547872632741928,
      "learning_rate": 9.980892464563907e-09,
      "loss": 0.0553,
      "step": 98090
    },
    {
      "epoch": 2.2974014282226562e-05,
      "model_forward_time": 2.409956932067871,
      "step": 98090
    },
    {
      "dynamics_loss_avg": 0.04423593282699585,
      "epoch": 2.2974014282226562e-05,
      "step": 98090
    },
    {
      "action_loss_avg": 0.001047590866801329,
      "epoch": 2.2974014282226562e-05,
      "step": 98090
    },
    {
      "epoch": 2.2974014282226562e-05,
      "step": 98090,
      "training_step_time": 9.344396829605103
    },
    {
      "epoch": 2.297496795654297e-05,
      "model_forward_time": 1.744016170501709,
      "step": 98091
    },
    {
      "epoch": 2.297496795654297e-05,
      "step": 98091,
      "training_step_time": 9.439191341400146
    },
    {
      "epoch": 2.2975921630859374e-05,
      "model_forward_time": 1.7266004085540771,
      "step": 98092
    },
    {
      "epoch": 2.2975921630859374e-05,
      "step": 98092,
      "training_step_time": 8.534573316574097
    },
    {
      "epoch": 2.297687530517578e-05,
      "model_forward_time": 1.7300083637237549,
      "step": 98093
    },
    {
      "epoch": 2.297687530517578e-05,
      "step": 98093,
      "training_step_time": 9.118237733840942
    },
    {
      "epoch": 2.297782897949219e-05,
      "model_forward_time": 1.7554519176483154,
      "step": 98094
    },
    {
      "epoch": 2.297782897949219e-05,
      "step": 98094,
      "training_step_time": 7.794644594192505
    },
    {
      "epoch": 2.2978782653808593e-05,
      "model_forward_time": 1.7356326580047607,
      "step": 98095
    },
    {
      "epoch": 2.2978782653808593e-05,
      "step": 98095,
      "training_step_time": 7.803703308105469
    },
    {
      "epoch": 2.2979736328125e-05,
      "model_forward_time": 1.7434308528900146,
      "step": 98096
    },
    {
      "epoch": 2.2979736328125e-05,
      "step": 98096,
      "training_step_time": 8.283377408981323
    },
    {
      "epoch": 2.2980690002441407e-05,
      "model_forward_time": 1.731816053390503,
      "step": 98097
    },
    {
      "epoch": 2.2980690002441407e-05,
      "step": 98097,
      "training_step_time": 7.678396224975586
    },
    {
      "epoch": 2.298164367675781e-05,
      "model_forward_time": 1.7334704399108887,
      "step": 98098
    },
    {
      "epoch": 2.298164367675781e-05,
      "step": 98098,
      "training_step_time": 7.534447193145752
    },
    {
      "epoch": 2.298259735107422e-05,
      "model_forward_time": 2.4111721515655518,
      "step": 98099
    },
    {
      "epoch": 2.298259735107422e-05,
      "step": 98099,
      "training_step_time": 8.431464672088623
    },
    {
      "epoch": 2.2983551025390626e-05,
      "grad_norm": 0.13445141911506653,
      "learning_rate": 9.876742810095985e-09,
      "loss": 0.0538,
      "step": 98100
    },
    {
      "epoch": 2.2983551025390626e-05,
      "model_forward_time": 1.7270267009735107,
      "step": 98100
    },
    {
      "dynamics_loss_avg": 0.05895465211942792,
      "epoch": 2.2983551025390626e-05,
      "step": 98100
    },
    {
      "action_loss_avg": 0.0014128122886177153,
      "epoch": 2.2983551025390626e-05,
      "step": 98100
    },
    {
      "epoch": 2.2983551025390626e-05,
      "step": 98100,
      "training_step_time": 8.829988718032837
    },
    {
      "epoch": 2.298450469970703e-05,
      "model_forward_time": 1.725675106048584,
      "step": 98101
    },
    {
      "epoch": 2.298450469970703e-05,
      "step": 98101,
      "training_step_time": 7.804102897644043
    },
    {
      "epoch": 2.2985458374023438e-05,
      "model_forward_time": 1.7389981746673584,
      "step": 98102
    },
    {
      "epoch": 2.2985458374023438e-05,
      "step": 98102,
      "training_step_time": 9.473203420639038
    },
    {
      "epoch": 2.2986412048339845e-05,
      "model_forward_time": 1.7453815937042236,
      "step": 98103
    },
    {
      "epoch": 2.2986412048339845e-05,
      "step": 98103,
      "training_step_time": 7.789618015289307
    },
    {
      "epoch": 2.298736572265625e-05,
      "model_forward_time": 1.7421956062316895,
      "step": 98104
    },
    {
      "epoch": 2.298736572265625e-05,
      "step": 98104,
      "training_step_time": 7.802401542663574
    },
    {
      "epoch": 2.2988319396972657e-05,
      "model_forward_time": 1.7356035709381104,
      "step": 98105
    },
    {
      "epoch": 2.2988319396972657e-05,
      "step": 98105,
      "training_step_time": 7.708783864974976
    },
    {
      "epoch": 2.2989273071289064e-05,
      "model_forward_time": 1.7212908267974854,
      "step": 98106
    },
    {
      "epoch": 2.2989273071289064e-05,
      "step": 98106,
      "training_step_time": 7.647626638412476
    },
    {
      "epoch": 2.2990226745605468e-05,
      "model_forward_time": 1.730858564376831,
      "step": 98107
    },
    {
      "epoch": 2.2990226745605468e-05,
      "step": 98107,
      "training_step_time": 7.5394814014434814
    },
    {
      "epoch": 2.2991180419921876e-05,
      "model_forward_time": 1.360224723815918,
      "step": 98108
    },
    {
      "epoch": 2.2991180419921876e-05,
      "step": 98108,
      "training_step_time": 8.241828203201294
    },
    {
      "epoch": 2.2992134094238283e-05,
      "model_forward_time": 1.7332522869110107,
      "step": 98109
    },
    {
      "epoch": 2.2992134094238283e-05,
      "step": 98109,
      "training_step_time": 8.774231433868408
    },
    {
      "epoch": 2.2993087768554687e-05,
      "grad_norm": 0.07811044901609421,
      "learning_rate": 9.773138868004461e-09,
      "loss": 0.0557,
      "step": 98110
    },
    {
      "epoch": 2.2993087768554687e-05,
      "model_forward_time": 1.7426812648773193,
      "step": 98110
    },
    {
      "dynamics_loss_avg": 0.053188611194491385,
      "epoch": 2.2993087768554687e-05,
      "step": 98110
    },
    {
      "action_loss_avg": 0.0016536034061573445,
      "epoch": 2.2993087768554687e-05,
      "step": 98110
    },
    {
      "epoch": 2.2993087768554687e-05,
      "step": 98110,
      "training_step_time": 8.702347993850708
    },
    {
      "epoch": 2.2994041442871094e-05,
      "model_forward_time": 1.7224531173706055,
      "step": 98111
    },
    {
      "epoch": 2.2994041442871094e-05,
      "step": 98111,
      "training_step_time": 7.899559020996094
    },
    {
      "epoch": 2.29949951171875e-05,
      "model_forward_time": 1.3461976051330566,
      "step": 98112
    },
    {
      "epoch": 2.29949951171875e-05,
      "step": 98112,
      "training_step_time": 10.324065923690796
    },
    {
      "epoch": 2.2995948791503906e-05,
      "model_forward_time": 1.7287724018096924,
      "step": 98113
    },
    {
      "epoch": 2.2995948791503906e-05,
      "step": 98113,
      "training_step_time": 7.356390953063965
    },
    {
      "epoch": 2.2996902465820313e-05,
      "model_forward_time": 1.7255868911743164,
      "step": 98114
    },
    {
      "epoch": 2.2996902465820313e-05,
      "step": 98114,
      "training_step_time": 7.378706693649292
    },
    {
      "epoch": 2.2997856140136717e-05,
      "model_forward_time": 1.7330620288848877,
      "step": 98115
    },
    {
      "epoch": 2.2997856140136717e-05,
      "step": 98115,
      "training_step_time": 7.7623138427734375
    },
    {
      "epoch": 2.2998809814453125e-05,
      "model_forward_time": 1.7362499237060547,
      "step": 98116
    },
    {
      "epoch": 2.2998809814453125e-05,
      "step": 98116,
      "training_step_time": 7.455193519592285
    },
    {
      "epoch": 2.2999763488769532e-05,
      "model_forward_time": 1.729952096939087,
      "step": 98117
    },
    {
      "epoch": 2.2999763488769532e-05,
      "step": 98117,
      "training_step_time": 7.877482175827026
    },
    {
      "epoch": 2.3000717163085936e-05,
      "model_forward_time": 2.368212938308716,
      "step": 98118
    },
    {
      "epoch": 2.3000717163085936e-05,
      "step": 98118,
      "training_step_time": 8.277465581893921
    },
    {
      "epoch": 2.3001670837402344e-05,
      "model_forward_time": 1.7561750411987305,
      "step": 98119
    },
    {
      "epoch": 2.3001670837402344e-05,
      "step": 98119,
      "training_step_time": 7.7490222454071045
    },
    {
      "epoch": 2.300262451171875e-05,
      "grad_norm": 0.0750841423869133,
      "learning_rate": 9.670080649619717e-09,
      "loss": 0.0543,
      "step": 98120
    },
    {
      "epoch": 2.300262451171875e-05,
      "model_forward_time": 1.7268762588500977,
      "step": 98120
    },
    {
      "dynamics_loss_avg": 0.05882879607379436,
      "epoch": 2.300262451171875e-05,
      "step": 98120
    },
    {
      "action_loss_avg": 0.0015872721560299397,
      "epoch": 2.300262451171875e-05,
      "step": 98120
    },
    {
      "epoch": 2.300262451171875e-05,
      "step": 98120,
      "training_step_time": 7.814594268798828
    },
    {
      "epoch": 2.3003578186035155e-05,
      "model_forward_time": 1.7276747226715088,
      "step": 98121
    },
    {
      "epoch": 2.3003578186035155e-05,
      "step": 98121,
      "training_step_time": 7.806123495101929
    },
    {
      "epoch": 2.3004531860351562e-05,
      "model_forward_time": 1.7481184005737305,
      "step": 98122
    },
    {
      "epoch": 2.3004531860351562e-05,
      "step": 98122,
      "training_step_time": 8.022572755813599
    },
    {
      "epoch": 2.300548553466797e-05,
      "model_forward_time": 1.725712776184082,
      "step": 98123
    },
    {
      "epoch": 2.300548553466797e-05,
      "step": 98123,
      "training_step_time": 7.440621852874756
    },
    {
      "epoch": 2.3006439208984374e-05,
      "model_forward_time": 1.7236969470977783,
      "step": 98124
    },
    {
      "epoch": 2.3006439208984374e-05,
      "step": 98124,
      "training_step_time": 8.617624044418335
    },
    {
      "epoch": 2.300739288330078e-05,
      "model_forward_time": 1.7328710556030273,
      "step": 98125
    },
    {
      "epoch": 2.300739288330078e-05,
      "step": 98125,
      "training_step_time": 7.700995683670044
    },
    {
      "epoch": 2.300834655761719e-05,
      "model_forward_time": 1.7420406341552734,
      "step": 98126
    },
    {
      "epoch": 2.300834655761719e-05,
      "step": 98126,
      "training_step_time": 7.391085386276245
    },
    {
      "epoch": 2.3009300231933593e-05,
      "model_forward_time": 2.380446672439575,
      "step": 98127
    },
    {
      "epoch": 2.3009300231933593e-05,
      "step": 98127,
      "training_step_time": 8.606785297393799
    },
    {
      "epoch": 2.301025390625e-05,
      "model_forward_time": 1.8124120235443115,
      "step": 98128
    },
    {
      "epoch": 2.301025390625e-05,
      "step": 98128,
      "training_step_time": 8.876134157180786
    },
    {
      "epoch": 2.3011207580566408e-05,
      "model_forward_time": 1.7393994331359863,
      "step": 98129
    },
    {
      "epoch": 2.3011207580566408e-05,
      "step": 98129,
      "training_step_time": 12.133997440338135
    },
    {
      "epoch": 2.301216125488281e-05,
      "grad_norm": 0.23345620930194855,
      "learning_rate": 9.567568166212737e-09,
      "loss": 0.0539,
      "step": 98130
    },
    {
      "epoch": 2.301216125488281e-05,
      "model_forward_time": 1.745011806488037,
      "step": 98130
    },
    {
      "dynamics_loss_avg": 0.06489801965653896,
      "epoch": 2.301216125488281e-05,
      "step": 98130
    },
    {
      "action_loss_avg": 0.0013120203948346898,
      "epoch": 2.301216125488281e-05,
      "step": 98130
    },
    {
      "epoch": 2.301216125488281e-05,
      "step": 98130,
      "training_step_time": 9.455055475234985
    },
    {
      "epoch": 2.301311492919922e-05,
      "model_forward_time": 1.7435593605041504,
      "step": 98131
    },
    {
      "epoch": 2.301311492919922e-05,
      "step": 98131,
      "training_step_time": 7.9484052658081055
    },
    {
      "epoch": 2.3014068603515626e-05,
      "model_forward_time": 1.7226829528808594,
      "step": 98132
    },
    {
      "epoch": 2.3014068603515626e-05,
      "step": 98132,
      "training_step_time": 9.116053819656372
    },
    {
      "epoch": 2.301502227783203e-05,
      "model_forward_time": 1.7315552234649658,
      "step": 98133
    },
    {
      "epoch": 2.301502227783203e-05,
      "step": 98133,
      "training_step_time": 7.751236915588379
    },
    {
      "epoch": 2.3015975952148438e-05,
      "model_forward_time": 1.738765001296997,
      "step": 98134
    },
    {
      "epoch": 2.3015975952148438e-05,
      "step": 98134,
      "training_step_time": 7.670242547988892
    },
    {
      "epoch": 2.3016929626464845e-05,
      "model_forward_time": 1.727834701538086,
      "step": 98135
    },
    {
      "epoch": 2.3016929626464845e-05,
      "step": 98135,
      "training_step_time": 7.46227240562439
    },
    {
      "epoch": 2.301788330078125e-05,
      "model_forward_time": 2.3972630500793457,
      "step": 98136
    },
    {
      "epoch": 2.301788330078125e-05,
      "step": 98136,
      "training_step_time": 8.608995199203491
    },
    {
      "epoch": 2.3018836975097657e-05,
      "model_forward_time": 1.753150224685669,
      "step": 98137
    },
    {
      "epoch": 2.3018836975097657e-05,
      "step": 98137,
      "training_step_time": 8.616584300994873
    },
    {
      "epoch": 2.3019790649414064e-05,
      "model_forward_time": 1.7355730533599854,
      "step": 98138
    },
    {
      "epoch": 2.3019790649414064e-05,
      "step": 98138,
      "training_step_time": 8.8207848072052
    },
    {
      "epoch": 2.3020744323730468e-05,
      "model_forward_time": 1.7526891231536865,
      "step": 98139
    },
    {
      "epoch": 2.3020744323730468e-05,
      "step": 98139,
      "training_step_time": 7.862344264984131
    },
    {
      "epoch": 2.3021697998046876e-05,
      "grad_norm": 0.08718518167734146,
      "learning_rate": 9.465601428992888e-09,
      "loss": 0.0536,
      "step": 98140
    },
    {
      "epoch": 2.3021697998046876e-05,
      "model_forward_time": 1.7436413764953613,
      "step": 98140
    },
    {
      "dynamics_loss_avg": 0.05605918429791927,
      "epoch": 2.3021697998046876e-05,
      "step": 98140
    },
    {
      "action_loss_avg": 0.0014757593686226755,
      "epoch": 2.3021697998046876e-05,
      "step": 98140
    },
    {
      "epoch": 2.3021697998046876e-05,
      "step": 98140,
      "training_step_time": 8.635728359222412
    },
    {
      "epoch": 2.302265167236328e-05,
      "model_forward_time": 1.7321522235870361,
      "step": 98141
    },
    {
      "epoch": 2.302265167236328e-05,
      "step": 98141,
      "training_step_time": 7.921523332595825
    },
    {
      "epoch": 2.3023605346679687e-05,
      "model_forward_time": 1.7411394119262695,
      "step": 98142
    },
    {
      "epoch": 2.3023605346679687e-05,
      "step": 98142,
      "training_step_time": 8.36217474937439
    },
    {
      "epoch": 2.3024559020996094e-05,
      "model_forward_time": 1.7353224754333496,
      "step": 98143
    },
    {
      "epoch": 2.3024559020996094e-05,
      "step": 98143,
      "training_step_time": 7.641541004180908
    },
    {
      "epoch": 2.30255126953125e-05,
      "model_forward_time": 1.733691930770874,
      "step": 98144
    },
    {
      "epoch": 2.30255126953125e-05,
      "step": 98144,
      "training_step_time": 7.4953179359436035
    },
    {
      "epoch": 2.3026466369628906e-05,
      "model_forward_time": 2.408055067062378,
      "step": 98145
    },
    {
      "epoch": 2.3026466369628906e-05,
      "step": 98145,
      "training_step_time": 7.773678779602051
    },
    {
      "epoch": 2.3027420043945313e-05,
      "model_forward_time": 1.7759833335876465,
      "step": 98146
    },
    {
      "epoch": 2.3027420043945313e-05,
      "step": 98146,
      "training_step_time": 8.516602993011475
    },
    {
      "epoch": 2.3028373718261717e-05,
      "model_forward_time": 1.729442834854126,
      "step": 98147
    },
    {
      "epoch": 2.3028373718261717e-05,
      "step": 98147,
      "training_step_time": 8.977548122406006
    },
    {
      "epoch": 2.3029327392578125e-05,
      "model_forward_time": 1.739013433456421,
      "step": 98148
    },
    {
      "epoch": 2.3029327392578125e-05,
      "step": 98148,
      "training_step_time": 8.882289171218872
    },
    {
      "epoch": 2.3030281066894532e-05,
      "model_forward_time": 1.7374508380889893,
      "step": 98149
    },
    {
      "epoch": 2.3030281066894532e-05,
      "step": 98149,
      "training_step_time": 7.776627063751221
    },
    {
      "epoch": 2.3031234741210936e-05,
      "grad_norm": 0.07975785434246063,
      "learning_rate": 9.364180449111804e-09,
      "loss": 0.0525,
      "step": 98150
    },
    {
      "epoch": 2.3031234741210936e-05,
      "model_forward_time": 1.7277021408081055,
      "step": 98150
    },
    {
      "dynamics_loss_avg": 0.05168414693325758,
      "epoch": 2.3031234741210936e-05,
      "step": 98150
    },
    {
      "action_loss_avg": 0.0014436481462325901,
      "epoch": 2.3031234741210936e-05,
      "step": 98150
    },
    {
      "epoch": 2.3031234741210936e-05,
      "step": 98150,
      "training_step_time": 7.856855154037476
    },
    {
      "epoch": 2.3032188415527344e-05,
      "model_forward_time": 1.7300643920898438,
      "step": 98151
    },
    {
      "epoch": 2.3032188415527344e-05,
      "step": 98151,
      "training_step_time": 7.521378517150879
    },
    {
      "epoch": 2.303314208984375e-05,
      "model_forward_time": 1.734137773513794,
      "step": 98152
    },
    {
      "epoch": 2.303314208984375e-05,
      "step": 98152,
      "training_step_time": 7.663377523422241
    },
    {
      "epoch": 2.3034095764160155e-05,
      "model_forward_time": 1.7336170673370361,
      "step": 98153
    },
    {
      "epoch": 2.3034095764160155e-05,
      "step": 98153,
      "training_step_time": 7.428050994873047
    },
    {
      "epoch": 2.3035049438476562e-05,
      "model_forward_time": 2.3552141189575195,
      "step": 98154
    },
    {
      "epoch": 2.3035049438476562e-05,
      "step": 98154,
      "training_step_time": 8.740448474884033
    },
    {
      "epoch": 2.303600311279297e-05,
      "model_forward_time": 1.7354464530944824,
      "step": 98155
    },
    {
      "epoch": 2.303600311279297e-05,
      "step": 98155,
      "training_step_time": 9.0664222240448
    },
    {
      "epoch": 2.3036956787109374e-05,
      "model_forward_time": 1.7363958358764648,
      "step": 98156
    },
    {
      "epoch": 2.3036956787109374e-05,
      "step": 98156,
      "training_step_time": 7.767585039138794
    },
    {
      "epoch": 2.303791046142578e-05,
      "model_forward_time": 1.733464002609253,
      "step": 98157
    },
    {
      "epoch": 2.303791046142578e-05,
      "step": 98157,
      "training_step_time": 7.851387023925781
    },
    {
      "epoch": 2.303886413574219e-05,
      "model_forward_time": 1.3457295894622803,
      "step": 98158
    },
    {
      "epoch": 2.303886413574219e-05,
      "step": 98158,
      "training_step_time": 7.655430793762207
    },
    {
      "epoch": 2.3039817810058593e-05,
      "model_forward_time": 1.7323145866394043,
      "step": 98159
    },
    {
      "epoch": 2.3039817810058593e-05,
      "step": 98159,
      "training_step_time": 8.752902269363403
    },
    {
      "epoch": 2.3040771484375e-05,
      "grad_norm": 0.07263235002756119,
      "learning_rate": 9.263305237661169e-09,
      "loss": 0.0539,
      "step": 98160
    },
    {
      "epoch": 2.3040771484375e-05,
      "model_forward_time": 1.0052037239074707,
      "step": 98160
    },
    {
      "dynamics_loss_avg": 0.04937867559492588,
      "epoch": 2.3040771484375e-05,
      "step": 98160
    },
    {
      "action_loss_avg": 0.0008392559800995513,
      "epoch": 2.3040771484375e-05,
      "step": 98160
    },
    {
      "epoch": 2.3040771484375e-05,
      "step": 98160,
      "training_step_time": 7.592019081115723
    },
    {
      "epoch": 2.3041725158691408e-05,
      "model_forward_time": 1.7329230308532715,
      "step": 98161
    },
    {
      "epoch": 2.3041725158691408e-05,
      "step": 98161,
      "training_step_time": 7.684328317642212
    },
    {
      "epoch": 2.304267883300781e-05,
      "model_forward_time": 1.73671555519104,
      "step": 98162
    },
    {
      "epoch": 2.304267883300781e-05,
      "step": 98162,
      "training_step_time": 7.463329792022705
    },
    {
      "epoch": 2.304363250732422e-05,
      "model_forward_time": 1.739762306213379,
      "step": 98163
    },
    {
      "epoch": 2.304363250732422e-05,
      "step": 98163,
      "training_step_time": 7.703684329986572
    },
    {
      "epoch": 2.3044586181640626e-05,
      "model_forward_time": 2.3593666553497314,
      "step": 98164
    },
    {
      "epoch": 2.3044586181640626e-05,
      "step": 98164,
      "training_step_time": 8.492083549499512
    },
    {
      "epoch": 2.304553985595703e-05,
      "model_forward_time": 1.7509775161743164,
      "step": 98165
    },
    {
      "epoch": 2.304553985595703e-05,
      "step": 98165,
      "training_step_time": 8.664691686630249
    },
    {
      "epoch": 2.3046493530273438e-05,
      "model_forward_time": 1.7376596927642822,
      "step": 98166
    },
    {
      "epoch": 2.3046493530273438e-05,
      "step": 98166,
      "training_step_time": 7.915912628173828
    },
    {
      "epoch": 2.3047447204589845e-05,
      "model_forward_time": 1.7334818840026855,
      "step": 98167
    },
    {
      "epoch": 2.3047447204589845e-05,
      "step": 98167,
      "training_step_time": 7.833861827850342
    },
    {
      "epoch": 2.304840087890625e-05,
      "model_forward_time": 1.755645990371704,
      "step": 98168
    },
    {
      "epoch": 2.304840087890625e-05,
      "step": 98168,
      "training_step_time": 7.842014312744141
    },
    {
      "epoch": 2.3049354553222657e-05,
      "model_forward_time": 1.7342596054077148,
      "step": 98169
    },
    {
      "epoch": 2.3049354553222657e-05,
      "step": 98169,
      "training_step_time": 7.831551551818848
    },
    {
      "epoch": 2.3050308227539064e-05,
      "grad_norm": 0.07204354554414749,
      "learning_rate": 9.162975805671603e-09,
      "loss": 0.0518,
      "step": 98170
    },
    {
      "epoch": 2.3050308227539064e-05,
      "model_forward_time": 1.7285566329956055,
      "step": 98170
    },
    {
      "dynamics_loss_avg": 0.05102071166038513,
      "epoch": 2.3050308227539064e-05,
      "step": 98170
    },
    {
      "action_loss_avg": 0.0013835468766046689,
      "epoch": 2.3050308227539064e-05,
      "step": 98170
    },
    {
      "epoch": 2.3050308227539064e-05,
      "step": 98170,
      "training_step_time": 7.632695436477661
    },
    {
      "epoch": 2.3051261901855468e-05,
      "model_forward_time": 1.7388484477996826,
      "step": 98171
    },
    {
      "epoch": 2.3051261901855468e-05,
      "step": 98171,
      "training_step_time": 7.634414911270142
    },
    {
      "epoch": 2.3052215576171876e-05,
      "model_forward_time": 1.7406072616577148,
      "step": 98172
    },
    {
      "epoch": 2.3052215576171876e-05,
      "step": 98172,
      "training_step_time": 7.695648908615112
    },
    {
      "epoch": 2.305316925048828e-05,
      "model_forward_time": 2.3864121437072754,
      "step": 98173
    },
    {
      "epoch": 2.305316925048828e-05,
      "step": 98173,
      "training_step_time": 8.485387086868286
    },
    {
      "epoch": 2.3054122924804687e-05,
      "model_forward_time": 1.756751537322998,
      "step": 98174
    },
    {
      "epoch": 2.3054122924804687e-05,
      "step": 98174,
      "training_step_time": 7.8458411693573
    },
    {
      "epoch": 2.3055076599121094e-05,
      "model_forward_time": 1.7323095798492432,
      "step": 98175
    },
    {
      "epoch": 2.3055076599121094e-05,
      "step": 98175,
      "training_step_time": 11.007959842681885
    },
    {
      "epoch": 2.30560302734375e-05,
      "model_forward_time": 1.7351129055023193,
      "step": 98176
    },
    {
      "epoch": 2.30560302734375e-05,
      "step": 98176,
      "training_step_time": 7.738915920257568
    },
    {
      "epoch": 2.3056983947753906e-05,
      "model_forward_time": 1.7514560222625732,
      "step": 98177
    },
    {
      "epoch": 2.3056983947753906e-05,
      "step": 98177,
      "training_step_time": 7.755895614624023
    },
    {
      "epoch": 2.3057937622070313e-05,
      "model_forward_time": 1.729994773864746,
      "step": 98178
    },
    {
      "epoch": 2.3057937622070313e-05,
      "step": 98178,
      "training_step_time": 10.061704397201538
    },
    {
      "epoch": 2.3058891296386717e-05,
      "model_forward_time": 1.739210844039917,
      "step": 98179
    },
    {
      "epoch": 2.3058891296386717e-05,
      "step": 98179,
      "training_step_time": 7.425251483917236
    },
    {
      "epoch": 2.3059844970703125e-05,
      "grad_norm": 0.10716615617275238,
      "learning_rate": 9.063192164115442e-09,
      "loss": 0.0529,
      "step": 98180
    },
    {
      "epoch": 2.3059844970703125e-05,
      "model_forward_time": 1.7391817569732666,
      "step": 98180
    },
    {
      "dynamics_loss_avg": 0.047113163862377405,
      "epoch": 2.3059844970703125e-05,
      "step": 98180
    },
    {
      "action_loss_avg": 0.001092224832973443,
      "epoch": 2.3059844970703125e-05,
      "step": 98180
    },
    {
      "epoch": 2.3059844970703125e-05,
      "step": 98180,
      "training_step_time": 7.731933116912842
    },
    {
      "epoch": 2.3060798645019532e-05,
      "model_forward_time": 1.7338755130767822,
      "step": 98181
    },
    {
      "epoch": 2.3060798645019532e-05,
      "step": 98181,
      "training_step_time": 7.786651611328125
    },
    {
      "epoch": 2.3061752319335936e-05,
      "model_forward_time": 1.7303879261016846,
      "step": 98182
    },
    {
      "epoch": 2.3061752319335936e-05,
      "step": 98182,
      "training_step_time": 8.971111059188843
    },
    {
      "epoch": 2.3062705993652344e-05,
      "model_forward_time": 1.7595744132995605,
      "step": 98183
    },
    {
      "epoch": 2.3062705993652344e-05,
      "step": 98183,
      "training_step_time": 8.895134210586548
    },
    {
      "epoch": 2.306365966796875e-05,
      "model_forward_time": 1.7296538352966309,
      "step": 98184
    },
    {
      "epoch": 2.306365966796875e-05,
      "step": 98184,
      "training_step_time": 7.818809509277344
    },
    {
      "epoch": 2.3064613342285155e-05,
      "model_forward_time": 1.738218069076538,
      "step": 98185
    },
    {
      "epoch": 2.3064613342285155e-05,
      "step": 98185,
      "training_step_time": 9.655503749847412
    },
    {
      "epoch": 2.3065567016601562e-05,
      "model_forward_time": 1.7526986598968506,
      "step": 98186
    },
    {
      "epoch": 2.3065567016601562e-05,
      "step": 98186,
      "training_step_time": 9.282974243164062
    },
    {
      "epoch": 2.306652069091797e-05,
      "model_forward_time": 1.74607515335083,
      "step": 98187
    },
    {
      "epoch": 2.306652069091797e-05,
      "step": 98187,
      "training_step_time": 7.818359851837158
    },
    {
      "epoch": 2.3067474365234374e-05,
      "model_forward_time": 1.7291944026947021,
      "step": 98188
    },
    {
      "epoch": 2.3067474365234374e-05,
      "step": 98188,
      "training_step_time": 7.571897506713867
    },
    {
      "epoch": 2.306842803955078e-05,
      "model_forward_time": 1.7346186637878418,
      "step": 98189
    },
    {
      "epoch": 2.306842803955078e-05,
      "step": 98189,
      "training_step_time": 7.670379638671875
    },
    {
      "epoch": 2.306938171386719e-05,
      "grad_norm": 0.10832720249891281,
      "learning_rate": 8.963954323904511e-09,
      "loss": 0.0533,
      "step": 98190
    },
    {
      "epoch": 2.306938171386719e-05,
      "model_forward_time": 1.7350990772247314,
      "step": 98190
    },
    {
      "dynamics_loss_avg": 0.06568930502980948,
      "epoch": 2.306938171386719e-05,
      "step": 98190
    },
    {
      "action_loss_avg": 0.0015147841069847346,
      "epoch": 2.306938171386719e-05,
      "step": 98190
    },
    {
      "epoch": 2.306938171386719e-05,
      "step": 98190,
      "training_step_time": 7.722912549972534
    },
    {
      "epoch": 2.3070335388183593e-05,
      "model_forward_time": 1.7318363189697266,
      "step": 98191
    },
    {
      "epoch": 2.3070335388183593e-05,
      "step": 98191,
      "training_step_time": 7.637436389923096
    },
    {
      "epoch": 2.30712890625e-05,
      "model_forward_time": 2.384695291519165,
      "step": 98192
    },
    {
      "epoch": 2.30712890625e-05,
      "step": 98192,
      "training_step_time": 8.342297554016113
    },
    {
      "epoch": 2.3072242736816408e-05,
      "model_forward_time": 1.7555365562438965,
      "step": 98193
    },
    {
      "epoch": 2.3072242736816408e-05,
      "step": 98193,
      "training_step_time": 8.930394887924194
    },
    {
      "epoch": 2.307319641113281e-05,
      "model_forward_time": 1.7301886081695557,
      "step": 98194
    },
    {
      "epoch": 2.307319641113281e-05,
      "step": 98194,
      "training_step_time": 7.900384902954102
    },
    {
      "epoch": 2.307415008544922e-05,
      "model_forward_time": 1.744507074356079,
      "step": 98195
    },
    {
      "epoch": 2.307415008544922e-05,
      "step": 98195,
      "training_step_time": 8.393327713012695
    },
    {
      "epoch": 2.3075103759765626e-05,
      "model_forward_time": 1.749626636505127,
      "step": 98196
    },
    {
      "epoch": 2.3075103759765626e-05,
      "step": 98196,
      "training_step_time": 8.478688478469849
    },
    {
      "epoch": 2.307605743408203e-05,
      "model_forward_time": 1.7398600578308105,
      "step": 98197
    },
    {
      "epoch": 2.307605743408203e-05,
      "step": 98197,
      "training_step_time": 7.729776620864868
    },
    {
      "epoch": 2.3077011108398438e-05,
      "model_forward_time": 1.731520414352417,
      "step": 98198
    },
    {
      "epoch": 2.3077011108398438e-05,
      "step": 98198,
      "training_step_time": 7.7509400844573975
    },
    {
      "epoch": 2.3077964782714845e-05,
      "model_forward_time": 1.7281253337860107,
      "step": 98199
    },
    {
      "epoch": 2.3077964782714845e-05,
      "step": 98199,
      "training_step_time": 7.7392799854278564
    },
    {
      "epoch": 2.307891845703125e-05,
      "grad_norm": 0.09289973229169846,
      "learning_rate": 8.865262295891797e-09,
      "loss": 0.0535,
      "step": 98200
    },
    {
      "epoch": 2.307891845703125e-05,
      "model_forward_time": 1.7313430309295654,
      "step": 98200
    },
    {
      "dynamics_loss_avg": 0.06076903399080038,
      "epoch": 2.307891845703125e-05,
      "step": 98200
    },
    {
      "action_loss_avg": 0.0008013833779841661,
      "epoch": 2.307891845703125e-05,
      "step": 98200
    },
    {
      "epoch": 2.307891845703125e-05,
      "step": 98200,
      "training_step_time": 7.574197053909302
    },
    {
      "epoch": 2.3079872131347657e-05,
      "model_forward_time": 2.389833927154541,
      "step": 98201
    },
    {
      "epoch": 2.3079872131347657e-05,
      "step": 98201,
      "training_step_time": 7.823317527770996
    },
    {
      "epoch": 2.3080825805664064e-05,
      "model_forward_time": 1.753983497619629,
      "step": 98202
    },
    {
      "epoch": 2.3080825805664064e-05,
      "step": 98202,
      "training_step_time": 8.72696590423584
    },
    {
      "epoch": 2.3081779479980468e-05,
      "model_forward_time": 1.7302038669586182,
      "step": 98203
    },
    {
      "epoch": 2.3081779479980468e-05,
      "step": 98203,
      "training_step_time": 8.446603536605835
    },
    {
      "epoch": 2.3082733154296876e-05,
      "model_forward_time": 1.7361047267913818,
      "step": 98204
    },
    {
      "epoch": 2.3082733154296876e-05,
      "step": 98204,
      "training_step_time": 8.35647177696228
    },
    {
      "epoch": 2.308368682861328e-05,
      "model_forward_time": 1.7355866432189941,
      "step": 98205
    },
    {
      "epoch": 2.308368682861328e-05,
      "step": 98205,
      "training_step_time": 8.632423400878906
    },
    {
      "epoch": 2.3084640502929687e-05,
      "model_forward_time": 1.7324824333190918,
      "step": 98206
    },
    {
      "epoch": 2.3084640502929687e-05,
      "step": 98206,
      "training_step_time": 7.679530382156372
    },
    {
      "epoch": 2.3085594177246094e-05,
      "model_forward_time": 1.7478423118591309,
      "step": 98207
    },
    {
      "epoch": 2.3085594177246094e-05,
      "step": 98207,
      "training_step_time": 8.7864248752594
    },
    {
      "epoch": 2.30865478515625e-05,
      "model_forward_time": 1.7325184345245361,
      "step": 98208
    },
    {
      "epoch": 2.30865478515625e-05,
      "step": 98208,
      "training_step_time": 7.758662700653076
    },
    {
      "epoch": 2.3087501525878906e-05,
      "model_forward_time": 1.7333438396453857,
      "step": 98209
    },
    {
      "epoch": 2.3087501525878906e-05,
      "step": 98209,
      "training_step_time": 7.583956241607666
    },
    {
      "epoch": 2.3088455200195313e-05,
      "grad_norm": 0.12206518650054932,
      "learning_rate": 8.76711609087033e-09,
      "loss": 0.0524,
      "step": 98210
    },
    {
      "epoch": 2.3088455200195313e-05,
      "model_forward_time": 2.40787672996521,
      "step": 98210
    },
    {
      "dynamics_loss_avg": 0.04602297740057111,
      "epoch": 2.3088455200195313e-05,
      "step": 98210
    },
    {
      "action_loss_avg": 0.0011610153887886554,
      "epoch": 2.3088455200195313e-05,
      "step": 98210
    },
    {
      "epoch": 2.3088455200195313e-05,
      "step": 98210,
      "training_step_time": 9.837139368057251
    },
    {
      "epoch": 2.3089408874511717e-05,
      "model_forward_time": 1.748572587966919,
      "step": 98211
    },
    {
      "epoch": 2.3089408874511717e-05,
      "step": 98211,
      "training_step_time": 9.596115827560425
    },
    {
      "epoch": 2.3090362548828125e-05,
      "model_forward_time": 1.735947608947754,
      "step": 98212
    },
    {
      "epoch": 2.3090362548828125e-05,
      "step": 98212,
      "training_step_time": 8.820688724517822
    },
    {
      "epoch": 2.3091316223144532e-05,
      "model_forward_time": 1.7362761497497559,
      "step": 98213
    },
    {
      "epoch": 2.3091316223144532e-05,
      "step": 98213,
      "training_step_time": 7.76285982131958
    },
    {
      "epoch": 2.3092269897460936e-05,
      "model_forward_time": 1.7628321647644043,
      "step": 98214
    },
    {
      "epoch": 2.3092269897460936e-05,
      "step": 98214,
      "training_step_time": 7.951361894607544
    },
    {
      "epoch": 2.3093223571777344e-05,
      "model_forward_time": 1.7375049591064453,
      "step": 98215
    },
    {
      "epoch": 2.3093223571777344e-05,
      "step": 98215,
      "training_step_time": 7.557917833328247
    },
    {
      "epoch": 2.309417724609375e-05,
      "model_forward_time": 1.7390921115875244,
      "step": 98216
    },
    {
      "epoch": 2.309417724609375e-05,
      "step": 98216,
      "training_step_time": 7.538980960845947
    },
    {
      "epoch": 2.3095130920410155e-05,
      "model_forward_time": 1.7341227531433105,
      "step": 98217
    },
    {
      "epoch": 2.3095130920410155e-05,
      "step": 98217,
      "training_step_time": 7.6773622035980225
    },
    {
      "epoch": 2.3096084594726562e-05,
      "model_forward_time": 1.7395617961883545,
      "step": 98218
    },
    {
      "epoch": 2.3096084594726562e-05,
      "step": 98218,
      "training_step_time": 7.56994104385376
    },
    {
      "epoch": 2.309703826904297e-05,
      "model_forward_time": 2.4036500453948975,
      "step": 98219
    },
    {
      "epoch": 2.309703826904297e-05,
      "step": 98219,
      "training_step_time": 8.398564100265503
    },
    {
      "epoch": 2.3097991943359374e-05,
      "grad_norm": 0.09089819341897964,
      "learning_rate": 8.669515719572086e-09,
      "loss": 0.0549,
      "step": 98220
    },
    {
      "epoch": 2.3097991943359374e-05,
      "model_forward_time": 1.728205680847168,
      "step": 98220
    },
    {
      "dynamics_loss_avg": 0.05865059420466423,
      "epoch": 2.3097991943359374e-05,
      "step": 98220
    },
    {
      "action_loss_avg": 0.0014193851297022776,
      "epoch": 2.3097991943359374e-05,
      "step": 98220
    },
    {
      "epoch": 2.3097991943359374e-05,
      "step": 98220,
      "training_step_time": 9.125555515289307
    },
    {
      "epoch": 2.309894561767578e-05,
      "model_forward_time": 1.7359380722045898,
      "step": 98221
    },
    {
      "epoch": 2.309894561767578e-05,
      "step": 98221,
      "training_step_time": 8.611213684082031
    },
    {
      "epoch": 2.309989929199219e-05,
      "model_forward_time": 1.3517403602600098,
      "step": 98222
    },
    {
      "epoch": 2.309989929199219e-05,
      "step": 98222,
      "training_step_time": 8.266920328140259
    },
    {
      "epoch": 2.3100852966308593e-05,
      "model_forward_time": 1.7693700790405273,
      "step": 98223
    },
    {
      "epoch": 2.3100852966308593e-05,
      "step": 98223,
      "training_step_time": 8.482565879821777
    },
    {
      "epoch": 2.3101806640625e-05,
      "model_forward_time": 1.7324962615966797,
      "step": 98224
    },
    {
      "epoch": 2.3101806640625e-05,
      "step": 98224,
      "training_step_time": 8.252327919006348
    },
    {
      "epoch": 2.3102760314941408e-05,
      "model_forward_time": 1.7376141548156738,
      "step": 98225
    },
    {
      "epoch": 2.3102760314941408e-05,
      "step": 98225,
      "training_step_time": 7.505670547485352
    },
    {
      "epoch": 2.310371398925781e-05,
      "model_forward_time": 1.7435956001281738,
      "step": 98226
    },
    {
      "epoch": 2.310371398925781e-05,
      "step": 98226,
      "training_step_time": 8.710697412490845
    },
    {
      "epoch": 2.310466766357422e-05,
      "model_forward_time": 1.7317087650299072,
      "step": 98227
    },
    {
      "epoch": 2.310466766357422e-05,
      "step": 98227,
      "training_step_time": 7.6642773151397705
    },
    {
      "epoch": 2.3105621337890626e-05,
      "model_forward_time": 1.7387919425964355,
      "step": 98228
    },
    {
      "epoch": 2.3105621337890626e-05,
      "step": 98228,
      "training_step_time": 7.689860820770264
    },
    {
      "epoch": 2.310657501220703e-05,
      "model_forward_time": 2.383924961090088,
      "step": 98229
    },
    {
      "epoch": 2.310657501220703e-05,
      "step": 98229,
      "training_step_time": 8.837164878845215
    },
    {
      "epoch": 2.3107528686523438e-05,
      "grad_norm": 0.06840295344591141,
      "learning_rate": 8.572461192671855e-09,
      "loss": 0.0546,
      "step": 98230
    },
    {
      "epoch": 2.3107528686523438e-05,
      "model_forward_time": 1.7465887069702148,
      "step": 98230
    },
    {
      "dynamics_loss_avg": 0.03246126472949982,
      "epoch": 2.3107528686523438e-05,
      "step": 98230
    },
    {
      "action_loss_avg": 0.0007982851879205555,
      "epoch": 2.3107528686523438e-05,
      "step": 98230
    },
    {
      "epoch": 2.3107528686523438e-05,
      "step": 98230,
      "training_step_time": 7.842553377151489
    },
    {
      "epoch": 2.3108482360839845e-05,
      "model_forward_time": 1.744903564453125,
      "step": 98231
    },
    {
      "epoch": 2.3108482360839845e-05,
      "step": 98231,
      "training_step_time": 8.872437477111816
    },
    {
      "epoch": 2.310943603515625e-05,
      "model_forward_time": 1.7481398582458496,
      "step": 98232
    },
    {
      "epoch": 2.310943603515625e-05,
      "step": 98232,
      "training_step_time": 8.588406085968018
    },
    {
      "epoch": 2.3110389709472657e-05,
      "model_forward_time": 1.7320549488067627,
      "step": 98233
    },
    {
      "epoch": 2.3110389709472657e-05,
      "step": 98233,
      "training_step_time": 7.764386177062988
    },
    {
      "epoch": 2.3111343383789064e-05,
      "model_forward_time": 1.750295639038086,
      "step": 98234
    },
    {
      "epoch": 2.3111343383789064e-05,
      "step": 98234,
      "training_step_time": 7.257424831390381
    },
    {
      "epoch": 2.3112297058105468e-05,
      "model_forward_time": 1.744288444519043,
      "step": 98235
    },
    {
      "epoch": 2.3112297058105468e-05,
      "step": 98235,
      "training_step_time": 7.683221340179443
    },
    {
      "epoch": 2.3113250732421876e-05,
      "model_forward_time": 1.7373239994049072,
      "step": 98236
    },
    {
      "epoch": 2.3113250732421876e-05,
      "step": 98236,
      "training_step_time": 7.738203763961792
    },
    {
      "epoch": 2.311420440673828e-05,
      "model_forward_time": 1.735999345779419,
      "step": 98237
    },
    {
      "epoch": 2.311420440673828e-05,
      "step": 98237,
      "training_step_time": 8.367460250854492
    },
    {
      "epoch": 2.3115158081054687e-05,
      "model_forward_time": 2.3961594104766846,
      "step": 98238
    },
    {
      "epoch": 2.3115158081054687e-05,
      "step": 98238,
      "training_step_time": 7.7331321239471436
    },
    {
      "epoch": 2.3116111755371094e-05,
      "model_forward_time": 1.750713586807251,
      "step": 98239
    },
    {
      "epoch": 2.3116111755371094e-05,
      "step": 98239,
      "training_step_time": 9.188135623931885
    },
    {
      "epoch": 2.31170654296875e-05,
      "grad_norm": 0.0645529180765152,
      "learning_rate": 8.475952520782816e-09,
      "loss": 0.0549,
      "step": 98240
    },
    {
      "epoch": 2.31170654296875e-05,
      "model_forward_time": 1.7351131439208984,
      "step": 98240
    },
    {
      "dynamics_loss_avg": 0.047914120741188525,
      "epoch": 2.31170654296875e-05,
      "step": 98240
    },
    {
      "action_loss_avg": 0.0011743801209377126,
      "epoch": 2.31170654296875e-05,
      "step": 98240
    },
    {
      "epoch": 2.31170654296875e-05,
      "step": 98240,
      "training_step_time": 9.477709293365479
    },
    {
      "epoch": 2.3118019104003906e-05,
      "model_forward_time": 1.7394654750823975,
      "step": 98241
    },
    {
      "epoch": 2.3118019104003906e-05,
      "step": 98241,
      "training_step_time": 7.810710668563843
    },
    {
      "epoch": 2.3118972778320313e-05,
      "model_forward_time": 1.3501927852630615,
      "step": 98242
    },
    {
      "epoch": 2.3118972778320313e-05,
      "step": 98242,
      "training_step_time": 8.278382539749146
    },
    {
      "epoch": 2.3119926452636717e-05,
      "model_forward_time": 1.3607137203216553,
      "step": 98243
    },
    {
      "epoch": 2.3119926452636717e-05,
      "step": 98243,
      "training_step_time": 8.566266775131226
    },
    {
      "epoch": 2.3120880126953125e-05,
      "model_forward_time": 1.7373673915863037,
      "step": 98244
    },
    {
      "epoch": 2.3120880126953125e-05,
      "step": 98244,
      "training_step_time": 7.5882933139801025
    },
    {
      "epoch": 2.3121833801269532e-05,
      "model_forward_time": 1.739720344543457,
      "step": 98245
    },
    {
      "epoch": 2.3121833801269532e-05,
      "step": 98245,
      "training_step_time": 7.66394829750061
    },
    {
      "epoch": 2.3122787475585936e-05,
      "model_forward_time": 1.010084629058838,
      "step": 98246
    },
    {
      "epoch": 2.3122787475585936e-05,
      "step": 98246,
      "training_step_time": 7.399171829223633
    },
    {
      "epoch": 2.3123741149902344e-05,
      "model_forward_time": 1.738429069519043,
      "step": 98247
    },
    {
      "epoch": 2.3123741149902344e-05,
      "step": 98247,
      "training_step_time": 8.976759433746338
    },
    {
      "epoch": 2.312469482421875e-05,
      "model_forward_time": 1.7425756454467773,
      "step": 98248
    },
    {
      "epoch": 2.312469482421875e-05,
      "step": 98248,
      "training_step_time": 8.698628187179565
    },
    {
      "epoch": 2.3125648498535155e-05,
      "model_forward_time": 1.7321856021881104,
      "step": 98249
    },
    {
      "epoch": 2.3125648498535155e-05,
      "step": 98249,
      "training_step_time": 9.466802835464478
    },
    {
      "epoch": 2.3126602172851562e-05,
      "grad_norm": 0.07268743962049484,
      "learning_rate": 8.379989714458748e-09,
      "loss": 0.0549,
      "step": 98250
    },
    {
      "epoch": 2.3126602172851562e-05,
      "model_forward_time": 1.7332556247711182,
      "step": 98250
    },
    {
      "dynamics_loss_avg": 0.04443964110687375,
      "epoch": 2.3126602172851562e-05,
      "step": 98250
    },
    {
      "action_loss_avg": 0.0007670697203138843,
      "epoch": 2.3126602172851562e-05,
      "step": 98250
    },
    {
      "epoch": 2.3126602172851562e-05,
      "step": 98250,
      "training_step_time": 7.605370998382568
    },
    {
      "epoch": 2.312755584716797e-05,
      "model_forward_time": 1.7677438259124756,
      "step": 98251
    },
    {
      "epoch": 2.312755584716797e-05,
      "step": 98251,
      "training_step_time": 7.731367349624634
    },
    {
      "epoch": 2.3128509521484374e-05,
      "model_forward_time": 1.7319602966308594,
      "step": 98252
    },
    {
      "epoch": 2.3128509521484374e-05,
      "step": 98252,
      "training_step_time": 8.494709491729736
    },
    {
      "epoch": 2.312946319580078e-05,
      "model_forward_time": 1.7380266189575195,
      "step": 98253
    },
    {
      "epoch": 2.312946319580078e-05,
      "step": 98253,
      "training_step_time": 7.547152280807495
    },
    {
      "epoch": 2.313041687011719e-05,
      "model_forward_time": 0.9943497180938721,
      "step": 98254
    },
    {
      "epoch": 2.313041687011719e-05,
      "step": 98254,
      "training_step_time": 7.475482940673828
    },
    {
      "epoch": 2.3131370544433593e-05,
      "model_forward_time": 1.3451945781707764,
      "step": 98255
    },
    {
      "epoch": 2.3131370544433593e-05,
      "step": 98255,
      "training_step_time": 7.399384260177612
    },
    {
      "epoch": 2.313232421875e-05,
      "model_forward_time": 1.7388465404510498,
      "step": 98256
    },
    {
      "epoch": 2.313232421875e-05,
      "step": 98256,
      "training_step_time": 9.173696994781494
    },
    {
      "epoch": 2.3133277893066408e-05,
      "model_forward_time": 2.4030840396881104,
      "step": 98257
    },
    {
      "epoch": 2.3133277893066408e-05,
      "step": 98257,
      "training_step_time": 9.85822319984436
    },
    {
      "epoch": 2.313423156738281e-05,
      "model_forward_time": 1.751316785812378,
      "step": 98258
    },
    {
      "epoch": 2.313423156738281e-05,
      "step": 98258,
      "training_step_time": 8.205279350280762
    },
    {
      "epoch": 2.313518524169922e-05,
      "model_forward_time": 1.742248773574829,
      "step": 98259
    },
    {
      "epoch": 2.313518524169922e-05,
      "step": 98259,
      "training_step_time": 7.642915725708008
    },
    {
      "epoch": 2.3136138916015626e-05,
      "grad_norm": 0.10969442874193192,
      "learning_rate": 8.284572784194033e-09,
      "loss": 0.0563,
      "step": 98260
    },
    {
      "epoch": 2.3136138916015626e-05,
      "model_forward_time": 1.7402372360229492,
      "step": 98260
    },
    {
      "dynamics_loss_avg": 0.050954381190240386,
      "epoch": 2.3136138916015626e-05,
      "step": 98260
    },
    {
      "action_loss_avg": 0.001538249768782407,
      "epoch": 2.3136138916015626e-05,
      "step": 98260
    },
    {
      "epoch": 2.3136138916015626e-05,
      "step": 98260,
      "training_step_time": 8.991622686386108
    },
    {
      "epoch": 2.313709259033203e-05,
      "model_forward_time": 1.7416422367095947,
      "step": 98261
    },
    {
      "epoch": 2.313709259033203e-05,
      "step": 98261,
      "training_step_time": 8.220228672027588
    },
    {
      "epoch": 2.3138046264648438e-05,
      "model_forward_time": 1.7323813438415527,
      "step": 98262
    },
    {
      "epoch": 2.3138046264648438e-05,
      "step": 98262,
      "training_step_time": 7.557429552078247
    },
    {
      "epoch": 2.3138999938964845e-05,
      "model_forward_time": 1.7302405834197998,
      "step": 98263
    },
    {
      "epoch": 2.3138999938964845e-05,
      "step": 98263,
      "training_step_time": 7.664258003234863
    },
    {
      "epoch": 2.313995361328125e-05,
      "model_forward_time": 1.7329788208007812,
      "step": 98264
    },
    {
      "epoch": 2.313995361328125e-05,
      "step": 98264,
      "training_step_time": 8.484145641326904
    },
    {
      "epoch": 2.3140907287597657e-05,
      "model_forward_time": 1.7309949398040771,
      "step": 98265
    },
    {
      "epoch": 2.3140907287597657e-05,
      "step": 98265,
      "training_step_time": 7.6751549243927
    },
    {
      "epoch": 2.314186096191406e-05,
      "model_forward_time": 2.400470733642578,
      "step": 98266
    },
    {
      "epoch": 2.314186096191406e-05,
      "step": 98266,
      "training_step_time": 8.680663585662842
    },
    {
      "epoch": 2.3142814636230468e-05,
      "model_forward_time": 1.7623064517974854,
      "step": 98267
    },
    {
      "epoch": 2.3142814636230468e-05,
      "step": 98267,
      "training_step_time": 9.513211488723755
    },
    {
      "epoch": 2.3143768310546876e-05,
      "model_forward_time": 1.7326056957244873,
      "step": 98268
    },
    {
      "epoch": 2.3143768310546876e-05,
      "step": 98268,
      "training_step_time": 8.639131307601929
    },
    {
      "epoch": 2.314472198486328e-05,
      "model_forward_time": 1.7456934452056885,
      "step": 98269
    },
    {
      "epoch": 2.314472198486328e-05,
      "step": 98269,
      "training_step_time": 7.819542646408081
    },
    {
      "epoch": 2.3145675659179687e-05,
      "grad_norm": 0.07304957509040833,
      "learning_rate": 8.189701740424216e-09,
      "loss": 0.0555,
      "step": 98270
    },
    {
      "epoch": 2.3145675659179687e-05,
      "model_forward_time": 1.7288789749145508,
      "step": 98270
    },
    {
      "dynamics_loss_avg": 0.041205130983144045,
      "epoch": 2.3145675659179687e-05,
      "step": 98270
    },
    {
      "action_loss_avg": 0.0009797098609851673,
      "epoch": 2.3145675659179687e-05,
      "step": 98270
    },
    {
      "epoch": 2.3145675659179687e-05,
      "step": 98270,
      "training_step_time": 8.370198965072632
    },
    {
      "epoch": 2.3146629333496095e-05,
      "model_forward_time": 1.3408567905426025,
      "step": 98271
    },
    {
      "epoch": 2.3146629333496095e-05,
      "step": 98271,
      "training_step_time": 7.565950632095337
    },
    {
      "epoch": 2.31475830078125e-05,
      "model_forward_time": 1.7376742362976074,
      "step": 98272
    },
    {
      "epoch": 2.31475830078125e-05,
      "step": 98272,
      "training_step_time": 7.5858399868011475
    },
    {
      "epoch": 2.3148536682128906e-05,
      "model_forward_time": 1.739692211151123,
      "step": 98273
    },
    {
      "epoch": 2.3148536682128906e-05,
      "step": 98273,
      "training_step_time": 7.659780263900757
    },
    {
      "epoch": 2.3149490356445313e-05,
      "model_forward_time": 1.7309019565582275,
      "step": 98274
    },
    {
      "epoch": 2.3149490356445313e-05,
      "step": 98274,
      "training_step_time": 7.734683275222778
    },
    {
      "epoch": 2.3150444030761717e-05,
      "model_forward_time": 1.7258329391479492,
      "step": 98275
    },
    {
      "epoch": 2.3150444030761717e-05,
      "step": 98275,
      "training_step_time": 7.74302339553833
    },
    {
      "epoch": 2.3151397705078125e-05,
      "model_forward_time": 2.3778529167175293,
      "step": 98276
    },
    {
      "epoch": 2.3151397705078125e-05,
      "step": 98276,
      "training_step_time": 10.327290773391724
    },
    {
      "epoch": 2.3152351379394532e-05,
      "model_forward_time": 1.731982946395874,
      "step": 98277
    },
    {
      "epoch": 2.3152351379394532e-05,
      "step": 98277,
      "training_step_time": 9.18361783027649
    },
    {
      "epoch": 2.3153305053710936e-05,
      "model_forward_time": 1.728750228881836,
      "step": 98278
    },
    {
      "epoch": 2.3153305053710936e-05,
      "step": 98278,
      "training_step_time": 7.835902690887451
    },
    {
      "epoch": 2.3154258728027344e-05,
      "model_forward_time": 1.7291288375854492,
      "step": 98279
    },
    {
      "epoch": 2.3154258728027344e-05,
      "step": 98279,
      "training_step_time": 7.683501720428467
    },
    {
      "epoch": 2.315521240234375e-05,
      "grad_norm": 0.0853886604309082,
      "learning_rate": 8.095376593522663e-09,
      "loss": 0.0575,
      "step": 98280
    },
    {
      "epoch": 2.315521240234375e-05,
      "model_forward_time": 1.7290611267089844,
      "step": 98280
    },
    {
      "dynamics_loss_avg": 0.06228084955364466,
      "epoch": 2.315521240234375e-05,
      "step": 98280
    },
    {
      "action_loss_avg": 0.0023139907338190825,
      "epoch": 2.315521240234375e-05,
      "step": 98280
    },
    {
      "epoch": 2.315521240234375e-05,
      "step": 98280,
      "training_step_time": 7.701141357421875
    },
    {
      "epoch": 2.3156166076660155e-05,
      "model_forward_time": 1.7316317558288574,
      "step": 98281
    },
    {
      "epoch": 2.3156166076660155e-05,
      "step": 98281,
      "training_step_time": 7.6317219734191895
    },
    {
      "epoch": 2.3157119750976563e-05,
      "model_forward_time": 1.7247545719146729,
      "step": 98282
    },
    {
      "epoch": 2.3157119750976563e-05,
      "step": 98282,
      "training_step_time": 7.80220627784729
    },
    {
      "epoch": 2.315807342529297e-05,
      "model_forward_time": 1.7316815853118896,
      "step": 98283
    },
    {
      "epoch": 2.315807342529297e-05,
      "step": 98283,
      "training_step_time": 7.686892032623291
    },
    {
      "epoch": 2.3159027099609374e-05,
      "model_forward_time": 1.7345046997070312,
      "step": 98284
    },
    {
      "epoch": 2.3159027099609374e-05,
      "step": 98284,
      "training_step_time": 7.700659513473511
    },
    {
      "epoch": 2.315998077392578e-05,
      "model_forward_time": 2.3905375003814697,
      "step": 98285
    },
    {
      "epoch": 2.315998077392578e-05,
      "step": 98285,
      "training_step_time": 9.412376642227173
    },
    {
      "epoch": 2.316093444824219e-05,
      "model_forward_time": 1.7280652523040771,
      "step": 98286
    },
    {
      "epoch": 2.316093444824219e-05,
      "step": 98286,
      "training_step_time": 8.785844326019287
    },
    {
      "epoch": 2.3161888122558593e-05,
      "model_forward_time": 1.7341983318328857,
      "step": 98287
    },
    {
      "epoch": 2.3161888122558593e-05,
      "step": 98287,
      "training_step_time": 7.850308418273926
    },
    {
      "epoch": 2.3162841796875e-05,
      "model_forward_time": 1.7548496723175049,
      "step": 98288
    },
    {
      "epoch": 2.3162841796875e-05,
      "step": 98288,
      "training_step_time": 7.6076819896698
    },
    {
      "epoch": 2.3163795471191408e-05,
      "model_forward_time": 1.738640546798706,
      "step": 98289
    },
    {
      "epoch": 2.3163795471191408e-05,
      "step": 98289,
      "training_step_time": 8.302348136901855
    },
    {
      "epoch": 2.316474914550781e-05,
      "grad_norm": 0.08472897112369537,
      "learning_rate": 8.001597353806124e-09,
      "loss": 0.0552,
      "step": 98290
    },
    {
      "epoch": 2.316474914550781e-05,
      "model_forward_time": 1.7235448360443115,
      "step": 98290
    },
    {
      "dynamics_loss_avg": 0.05361471306532621,
      "epoch": 2.316474914550781e-05,
      "step": 98290
    },
    {
      "action_loss_avg": 0.001451627624919638,
      "epoch": 2.316474914550781e-05,
      "step": 98290
    },
    {
      "epoch": 2.316474914550781e-05,
      "step": 98290,
      "training_step_time": 7.4633049964904785
    },
    {
      "epoch": 2.316570281982422e-05,
      "model_forward_time": 1.728672742843628,
      "step": 98291
    },
    {
      "epoch": 2.316570281982422e-05,
      "step": 98291,
      "training_step_time": 7.699023246765137
    },
    {
      "epoch": 2.3166656494140627e-05,
      "model_forward_time": 1.7303779125213623,
      "step": 98292
    },
    {
      "epoch": 2.3166656494140627e-05,
      "step": 98292,
      "training_step_time": 8.54228925704956
    },
    {
      "epoch": 2.316761016845703e-05,
      "model_forward_time": 1.7342431545257568,
      "step": 98293
    },
    {
      "epoch": 2.316761016845703e-05,
      "step": 98293,
      "training_step_time": 7.678649187088013
    },
    {
      "epoch": 2.3168563842773438e-05,
      "model_forward_time": 2.408031940460205,
      "step": 98294
    },
    {
      "epoch": 2.3168563842773438e-05,
      "step": 98294,
      "training_step_time": 7.648106813430786
    },
    {
      "epoch": 2.3169517517089845e-05,
      "model_forward_time": 1.7254173755645752,
      "step": 98295
    },
    {
      "epoch": 2.3169517517089845e-05,
      "step": 98295,
      "training_step_time": 7.6781322956085205
    },
    {
      "epoch": 2.317047119140625e-05,
      "model_forward_time": 1.733048677444458,
      "step": 98296
    },
    {
      "epoch": 2.317047119140625e-05,
      "step": 98296,
      "training_step_time": 8.09829306602478
    },
    {
      "epoch": 2.3171424865722657e-05,
      "model_forward_time": 1.7573294639587402,
      "step": 98297
    },
    {
      "epoch": 2.3171424865722657e-05,
      "step": 98297,
      "training_step_time": 9.298667192459106
    },
    {
      "epoch": 2.317237854003906e-05,
      "model_forward_time": 1.735187292098999,
      "step": 98298
    },
    {
      "epoch": 2.317237854003906e-05,
      "step": 98298,
      "training_step_time": 7.707279682159424
    },
    {
      "epoch": 2.3173332214355468e-05,
      "model_forward_time": 1.7271323204040527,
      "step": 98299
    },
    {
      "epoch": 2.3173332214355468e-05,
      "step": 98299,
      "training_step_time": 7.3558173179626465
    },
    {
      "epoch": 2.3174285888671876e-05,
      "grad_norm": 0.07317300140857697,
      "learning_rate": 7.90836403152917e-09,
      "loss": 0.0554,
      "step": 98300
    },
    {
      "epoch": 2.3174285888671876e-05,
      "model_forward_time": 1.7260217666625977,
      "step": 98300
    },
    {
      "dynamics_loss_avg": 0.0475483606569469,
      "epoch": 2.3174285888671876e-05,
      "step": 98300
    },
    {
      "action_loss_avg": 0.0009115530556300655,
      "epoch": 2.3174285888671876e-05,
      "step": 98300
    },
    {
      "epoch": 2.3174285888671876e-05,
      "step": 98300,
      "training_step_time": 8.464730262756348
    },
    {
      "epoch": 2.317523956298828e-05,
      "model_forward_time": 1.730504035949707,
      "step": 98301
    },
    {
      "epoch": 2.317523956298828e-05,
      "step": 98301,
      "training_step_time": 7.68353009223938
    },
    {
      "epoch": 2.3176193237304687e-05,
      "model_forward_time": 1.7351055145263672,
      "step": 98302
    },
    {
      "epoch": 2.3176193237304687e-05,
      "step": 98302,
      "training_step_time": 8.302444458007812
    },
    {
      "epoch": 2.3177146911621095e-05,
      "model_forward_time": 2.3753836154937744,
      "step": 98303
    },
    {
      "epoch": 2.3177146911621095e-05,
      "step": 98303,
      "training_step_time": 8.418476581573486
    },
    {
      "epoch": 2.31781005859375e-05,
      "model_forward_time": 1.7563199996948242,
      "step": 98304
    },
    {
      "epoch": 2.31781005859375e-05,
      "step": 98304,
      "training_step_time": 8.221172571182251
    },
    {
      "epoch": 2.3179054260253906e-05,
      "model_forward_time": 1.7313947677612305,
      "step": 98305
    },
    {
      "epoch": 2.3179054260253906e-05,
      "step": 98305,
      "training_step_time": 7.638316631317139
    },
    {
      "epoch": 2.3180007934570313e-05,
      "model_forward_time": 1.7402923107147217,
      "step": 98306
    },
    {
      "epoch": 2.3180007934570313e-05,
      "step": 98306,
      "training_step_time": 7.844212293624878
    },
    {
      "epoch": 2.3180961608886717e-05,
      "model_forward_time": 1.7199580669403076,
      "step": 98307
    },
    {
      "epoch": 2.3180961608886717e-05,
      "step": 98307,
      "training_step_time": 8.598498582839966
    },
    {
      "epoch": 2.3181915283203125e-05,
      "model_forward_time": 1.7341957092285156,
      "step": 98308
    },
    {
      "epoch": 2.3181915283203125e-05,
      "step": 98308,
      "training_step_time": 7.605116605758667
    },
    {
      "epoch": 2.3182868957519532e-05,
      "model_forward_time": 1.7308745384216309,
      "step": 98309
    },
    {
      "epoch": 2.3182868957519532e-05,
      "step": 98309,
      "training_step_time": 8.642340898513794
    },
    {
      "epoch": 2.3183822631835936e-05,
      "grad_norm": 0.0897156223654747,
      "learning_rate": 7.815676636888093e-09,
      "loss": 0.0574,
      "step": 98310
    },
    {
      "epoch": 2.3183822631835936e-05,
      "model_forward_time": 1.7483086585998535,
      "step": 98310
    },
    {
      "dynamics_loss_avg": 0.04316747859120369,
      "epoch": 2.3183822631835936e-05,
      "step": 98310
    },
    {
      "action_loss_avg": 0.0010328766744351014,
      "epoch": 2.3183822631835936e-05,
      "step": 98310
    },
    {
      "epoch": 2.3183822631835936e-05,
      "step": 98310,
      "training_step_time": 7.754544258117676
    },
    {
      "epoch": 2.3184776306152344e-05,
      "model_forward_time": 1.7325479984283447,
      "step": 98311
    },
    {
      "epoch": 2.3184776306152344e-05,
      "step": 98311,
      "training_step_time": 7.738636255264282
    },
    {
      "epoch": 2.318572998046875e-05,
      "model_forward_time": 2.3840250968933105,
      "step": 98312
    },
    {
      "epoch": 2.318572998046875e-05,
      "step": 98312,
      "training_step_time": 9.464455366134644
    },
    {
      "epoch": 2.3186683654785155e-05,
      "model_forward_time": 1.751723051071167,
      "step": 98313
    },
    {
      "epoch": 2.3186683654785155e-05,
      "step": 98313,
      "training_step_time": 9.415194749832153
    },
    {
      "epoch": 2.3187637329101563e-05,
      "model_forward_time": 1.740144968032837,
      "step": 98314
    },
    {
      "epoch": 2.3187637329101563e-05,
      "step": 98314,
      "training_step_time": 7.668513059616089
    },
    {
      "epoch": 2.318859100341797e-05,
      "model_forward_time": 1.7244865894317627,
      "step": 98315
    },
    {
      "epoch": 2.318859100341797e-05,
      "step": 98315,
      "training_step_time": 8.375810861587524
    },
    {
      "epoch": 2.3189544677734374e-05,
      "model_forward_time": 1.7278859615325928,
      "step": 98316
    },
    {
      "epoch": 2.3189544677734374e-05,
      "step": 98316,
      "training_step_time": 8.688096523284912
    },
    {
      "epoch": 2.319049835205078e-05,
      "model_forward_time": 1.7281758785247803,
      "step": 98317
    },
    {
      "epoch": 2.319049835205078e-05,
      "step": 98317,
      "training_step_time": 7.719120979309082
    },
    {
      "epoch": 2.319145202636719e-05,
      "model_forward_time": 1.7348363399505615,
      "step": 98318
    },
    {
      "epoch": 2.319145202636719e-05,
      "step": 98318,
      "training_step_time": 7.872395038604736
    },
    {
      "epoch": 2.3192405700683593e-05,
      "model_forward_time": 1.731194019317627,
      "step": 98319
    },
    {
      "epoch": 2.3192405700683593e-05,
      "step": 98319,
      "training_step_time": 7.877752065658569
    },
    {
      "epoch": 2.3193359375e-05,
      "grad_norm": 0.3860050439834595,
      "learning_rate": 7.723535180019227e-09,
      "loss": 0.0564,
      "step": 98320
    },
    {
      "epoch": 2.3193359375e-05,
      "model_forward_time": 1.7291643619537354,
      "step": 98320
    },
    {
      "dynamics_loss_avg": 0.05801554545760155,
      "epoch": 2.3193359375e-05,
      "step": 98320
    },
    {
      "action_loss_avg": 0.0012831045940401963,
      "epoch": 2.3193359375e-05,
      "step": 98320
    },
    {
      "epoch": 2.3193359375e-05,
      "step": 98320,
      "training_step_time": 7.758836269378662
    },
    {
      "epoch": 2.3194313049316408e-05,
      "model_forward_time": 1.7369592189788818,
      "step": 98321
    },
    {
      "epoch": 2.3194313049316408e-05,
      "step": 98321,
      "training_step_time": 10.48650574684143
    },
    {
      "epoch": 2.3195266723632812e-05,
      "model_forward_time": 0.9984891414642334,
      "step": 98322
    },
    {
      "epoch": 2.3195266723632812e-05,
      "step": 98322,
      "training_step_time": 7.742575168609619
    },
    {
      "epoch": 2.319622039794922e-05,
      "model_forward_time": 1.7213621139526367,
      "step": 98323
    },
    {
      "epoch": 2.319622039794922e-05,
      "step": 98323,
      "training_step_time": 9.168161392211914
    },
    {
      "epoch": 2.3197174072265627e-05,
      "model_forward_time": 1.7285840511322021,
      "step": 98324
    },
    {
      "epoch": 2.3197174072265627e-05,
      "step": 98324,
      "training_step_time": 8.004739761352539
    },
    {
      "epoch": 2.319812774658203e-05,
      "model_forward_time": 1.75783109664917,
      "step": 98325
    },
    {
      "epoch": 2.319812774658203e-05,
      "step": 98325,
      "training_step_time": 8.972783327102661
    },
    {
      "epoch": 2.3199081420898438e-05,
      "model_forward_time": 1.7268011569976807,
      "step": 98326
    },
    {
      "epoch": 2.3199081420898438e-05,
      "step": 98326,
      "training_step_time": 8.559563875198364
    },
    {
      "epoch": 2.3200035095214845e-05,
      "model_forward_time": 1.7250401973724365,
      "step": 98327
    },
    {
      "epoch": 2.3200035095214845e-05,
      "step": 98327,
      "training_step_time": 7.775595664978027
    },
    {
      "epoch": 2.320098876953125e-05,
      "model_forward_time": 1.734976053237915,
      "step": 98328
    },
    {
      "epoch": 2.320098876953125e-05,
      "step": 98328,
      "training_step_time": 7.827145099639893
    },
    {
      "epoch": 2.3201942443847657e-05,
      "model_forward_time": 1.7276504039764404,
      "step": 98329
    },
    {
      "epoch": 2.3201942443847657e-05,
      "step": 98329,
      "training_step_time": 7.686407566070557
    },
    {
      "epoch": 2.320289611816406e-05,
      "grad_norm": 0.08771776407957077,
      "learning_rate": 7.6319396709984e-09,
      "loss": 0.0556,
      "step": 98330
    },
    {
      "epoch": 2.320289611816406e-05,
      "model_forward_time": 1.7242844104766846,
      "step": 98330
    },
    {
      "dynamics_loss_avg": 0.05789117459207773,
      "epoch": 2.320289611816406e-05,
      "step": 98330
    },
    {
      "action_loss_avg": 0.0016942802569246851,
      "epoch": 2.320289611816406e-05,
      "step": 98330
    },
    {
      "epoch": 2.320289611816406e-05,
      "step": 98330,
      "training_step_time": 8.757072448730469
    },
    {
      "epoch": 2.3203849792480468e-05,
      "model_forward_time": 2.4107117652893066,
      "step": 98331
    },
    {
      "epoch": 2.3203849792480468e-05,
      "step": 98331,
      "training_step_time": 8.900752305984497
    },
    {
      "epoch": 2.3204803466796876e-05,
      "model_forward_time": 1.7271099090576172,
      "step": 98332
    },
    {
      "epoch": 2.3204803466796876e-05,
      "step": 98332,
      "training_step_time": 7.9068872928619385
    },
    {
      "epoch": 2.320575714111328e-05,
      "model_forward_time": 1.7400810718536377,
      "step": 98333
    },
    {
      "epoch": 2.320575714111328e-05,
      "step": 98333,
      "training_step_time": 7.800145626068115
    },
    {
      "epoch": 2.3206710815429687e-05,
      "model_forward_time": 1.7602617740631104,
      "step": 98334
    },
    {
      "epoch": 2.3206710815429687e-05,
      "step": 98334,
      "training_step_time": 7.86011266708374
    },
    {
      "epoch": 2.3207664489746095e-05,
      "model_forward_time": 1.7265450954437256,
      "step": 98335
    },
    {
      "epoch": 2.3207664489746095e-05,
      "step": 98335,
      "training_step_time": 7.793169736862183
    },
    {
      "epoch": 2.32086181640625e-05,
      "model_forward_time": 1.7299323081970215,
      "step": 98336
    },
    {
      "epoch": 2.32086181640625e-05,
      "step": 98336,
      "training_step_time": 7.921003818511963
    },
    {
      "epoch": 2.3209571838378906e-05,
      "model_forward_time": 1.7353451251983643,
      "step": 98337
    },
    {
      "epoch": 2.3209571838378906e-05,
      "step": 98337,
      "training_step_time": 7.801499128341675
    },
    {
      "epoch": 2.3210525512695313e-05,
      "model_forward_time": 1.7301490306854248,
      "step": 98338
    },
    {
      "epoch": 2.3210525512695313e-05,
      "step": 98338,
      "training_step_time": 8.571209192276001
    },
    {
      "epoch": 2.3211479187011717e-05,
      "model_forward_time": 1.7355480194091797,
      "step": 98339
    },
    {
      "epoch": 2.3211479187011717e-05,
      "step": 98339,
      "training_step_time": 8.962001085281372
    },
    {
      "epoch": 2.3212432861328125e-05,
      "grad_norm": 0.13036872446537018,
      "learning_rate": 7.540890119843159e-09,
      "loss": 0.0555,
      "step": 98340
    },
    {
      "epoch": 2.3212432861328125e-05,
      "model_forward_time": 2.4162285327911377,
      "step": 98340
    },
    {
      "dynamics_loss_avg": 0.06287217531353236,
      "epoch": 2.3212432861328125e-05,
      "step": 98340
    },
    {
      "action_loss_avg": 0.0014795617898926139,
      "epoch": 2.3212432861328125e-05,
      "step": 98340
    },
    {
      "epoch": 2.3212432861328125e-05,
      "step": 98340,
      "training_step_time": 8.353476285934448
    },
    {
      "epoch": 2.3213386535644532e-05,
      "model_forward_time": 1.8121933937072754,
      "step": 98341
    },
    {
      "epoch": 2.3213386535644532e-05,
      "step": 98341,
      "training_step_time": 8.451112985610962
    },
    {
      "epoch": 2.3214340209960936e-05,
      "model_forward_time": 1.749114990234375,
      "step": 98342
    },
    {
      "epoch": 2.3214340209960936e-05,
      "step": 98342,
      "training_step_time": 11.534924030303955
    },
    {
      "epoch": 2.3215293884277344e-05,
      "model_forward_time": 1.7470629215240479,
      "step": 98343
    },
    {
      "epoch": 2.3215293884277344e-05,
      "step": 98343,
      "training_step_time": 7.944377183914185
    },
    {
      "epoch": 2.321624755859375e-05,
      "model_forward_time": 1.7392578125,
      "step": 98344
    },
    {
      "epoch": 2.321624755859375e-05,
      "step": 98344,
      "training_step_time": 8.125800371170044
    },
    {
      "epoch": 2.3217201232910155e-05,
      "model_forward_time": 1.7467658519744873,
      "step": 98345
    },
    {
      "epoch": 2.3217201232910155e-05,
      "step": 98345,
      "training_step_time": 7.951078653335571
    },
    {
      "epoch": 2.3218154907226563e-05,
      "model_forward_time": 1.7564332485198975,
      "step": 98346
    },
    {
      "epoch": 2.3218154907226563e-05,
      "step": 98346,
      "training_step_time": 8.326070547103882
    },
    {
      "epoch": 2.321910858154297e-05,
      "model_forward_time": 1.7395167350769043,
      "step": 98347
    },
    {
      "epoch": 2.321910858154297e-05,
      "step": 98347,
      "training_step_time": 7.003126382827759
    },
    {
      "epoch": 2.3220062255859374e-05,
      "model_forward_time": 1.7405483722686768,
      "step": 98348
    },
    {
      "epoch": 2.3220062255859374e-05,
      "step": 98348,
      "training_step_time": 7.040709495544434
    },
    {
      "epoch": 2.322101593017578e-05,
      "model_forward_time": 1.742046594619751,
      "step": 98349
    },
    {
      "epoch": 2.322101593017578e-05,
      "step": 98349,
      "training_step_time": 7.063500881195068
    },
    {
      "epoch": 2.322196960449219e-05,
      "grad_norm": 0.07782850414514542,
      "learning_rate": 7.450386536509979e-09,
      "loss": 0.0527,
      "step": 98350
    },
    {
      "epoch": 2.322196960449219e-05,
      "model_forward_time": 1.726513147354126,
      "step": 98350
    },
    {
      "dynamics_loss_avg": 0.04657704941928387,
      "epoch": 2.322196960449219e-05,
      "step": 98350
    },
    {
      "action_loss_avg": 0.001899006892926991,
      "epoch": 2.322196960449219e-05,
      "step": 98350
    },
    {
      "epoch": 2.322196960449219e-05,
      "step": 98350,
      "training_step_time": 7.009388446807861
    },
    {
      "epoch": 2.3222923278808593e-05,
      "model_forward_time": 1.751326084136963,
      "step": 98351
    },
    {
      "epoch": 2.3222923278808593e-05,
      "step": 98351,
      "training_step_time": 7.960796117782593
    },
    {
      "epoch": 2.3223876953125e-05,
      "model_forward_time": 2.3707547187805176,
      "step": 98352
    },
    {
      "epoch": 2.3223876953125e-05,
      "step": 98352,
      "training_step_time": 7.978313684463501
    },
    {
      "epoch": 2.3224830627441408e-05,
      "model_forward_time": 1.7478241920471191,
      "step": 98353
    },
    {
      "epoch": 2.3224830627441408e-05,
      "step": 98353,
      "training_step_time": 7.622221231460571
    },
    {
      "epoch": 2.3225784301757812e-05,
      "model_forward_time": 1.7497403621673584,
      "step": 98354
    },
    {
      "epoch": 2.3225784301757812e-05,
      "step": 98354,
      "training_step_time": 7.861690521240234
    },
    {
      "epoch": 2.322673797607422e-05,
      "model_forward_time": 1.7416801452636719,
      "step": 98355
    },
    {
      "epoch": 2.322673797607422e-05,
      "step": 98355,
      "training_step_time": 7.760576009750366
    },
    {
      "epoch": 2.3227691650390627e-05,
      "model_forward_time": 1.7367641925811768,
      "step": 98356
    },
    {
      "epoch": 2.3227691650390627e-05,
      "step": 98356,
      "training_step_time": 8.076256275177002
    },
    {
      "epoch": 2.322864532470703e-05,
      "model_forward_time": 1.7476568222045898,
      "step": 98357
    },
    {
      "epoch": 2.322864532470703e-05,
      "step": 98357,
      "training_step_time": 7.129283905029297
    },
    {
      "epoch": 2.3229598999023438e-05,
      "model_forward_time": 1.7476084232330322,
      "step": 98358
    },
    {
      "epoch": 2.3229598999023438e-05,
      "step": 98358,
      "training_step_time": 7.129864931106567
    },
    {
      "epoch": 2.3230552673339845e-05,
      "model_forward_time": 1.734227180480957,
      "step": 98359
    },
    {
      "epoch": 2.3230552673339845e-05,
      "step": 98359,
      "training_step_time": 7.122907400131226
    },
    {
      "epoch": 2.323150634765625e-05,
      "grad_norm": 0.07302415370941162,
      "learning_rate": 7.360428930895947e-09,
      "loss": 0.0552,
      "step": 98360
    },
    {
      "epoch": 2.323150634765625e-05,
      "model_forward_time": 1.752286672592163,
      "step": 98360
    },
    {
      "dynamics_loss_avg": 0.05410619992762804,
      "epoch": 2.323150634765625e-05,
      "step": 98360
    },
    {
      "action_loss_avg": 0.0011384985598851926,
      "epoch": 2.323150634765625e-05,
      "step": 98360
    },
    {
      "epoch": 2.323150634765625e-05,
      "step": 98360,
      "training_step_time": 7.123725175857544
    },
    {
      "epoch": 2.3232460021972657e-05,
      "model_forward_time": 2.3838791847229004,
      "step": 98361
    },
    {
      "epoch": 2.3232460021972657e-05,
      "step": 98361,
      "training_step_time": 8.013066053390503
    },
    {
      "epoch": 2.323341369628906e-05,
      "model_forward_time": 1.7563776969909668,
      "step": 98362
    },
    {
      "epoch": 2.323341369628906e-05,
      "step": 98362,
      "training_step_time": 7.074305772781372
    },
    {
      "epoch": 2.323436737060547e-05,
      "model_forward_time": 1.7465238571166992,
      "step": 98363
    },
    {
      "epoch": 2.323436737060547e-05,
      "step": 98363,
      "training_step_time": 7.878094434738159
    },
    {
      "epoch": 2.3235321044921876e-05,
      "model_forward_time": 1.7398731708526611,
      "step": 98364
    },
    {
      "epoch": 2.3235321044921876e-05,
      "step": 98364,
      "training_step_time": 7.779079914093018
    },
    {
      "epoch": 2.323627471923828e-05,
      "model_forward_time": 1.7432477474212646,
      "step": 98365
    },
    {
      "epoch": 2.323627471923828e-05,
      "step": 98365,
      "training_step_time": 8.086882829666138
    },
    {
      "epoch": 2.3237228393554687e-05,
      "model_forward_time": 1.7680881023406982,
      "step": 98366
    },
    {
      "epoch": 2.3237228393554687e-05,
      "step": 98366,
      "training_step_time": 7.092748403549194
    },
    {
      "epoch": 2.3238182067871095e-05,
      "model_forward_time": 1.737596035003662,
      "step": 98367
    },
    {
      "epoch": 2.3238182067871095e-05,
      "step": 98367,
      "training_step_time": 7.1696555614471436
    },
    {
      "epoch": 2.32391357421875e-05,
      "model_forward_time": 1.728633165359497,
      "step": 98368
    },
    {
      "epoch": 2.32391357421875e-05,
      "step": 98368,
      "training_step_time": 7.076671361923218
    },
    {
      "epoch": 2.3240089416503906e-05,
      "model_forward_time": 1.7444605827331543,
      "step": 98369
    },
    {
      "epoch": 2.3240089416503906e-05,
      "step": 98369,
      "training_step_time": 7.051172733306885
    },
    {
      "epoch": 2.3241043090820313e-05,
      "grad_norm": 0.11188986897468567,
      "learning_rate": 7.271017312839301e-09,
      "loss": 0.0558,
      "step": 98370
    },
    {
      "epoch": 2.3241043090820313e-05,
      "model_forward_time": 2.38271164894104,
      "step": 98370
    },
    {
      "dynamics_loss_avg": 0.06062702126801014,
      "epoch": 2.3241043090820313e-05,
      "step": 98370
    },
    {
      "action_loss_avg": 0.0009483281435677781,
      "epoch": 2.3241043090820313e-05,
      "step": 98370
    },
    {
      "epoch": 2.3241043090820313e-05,
      "step": 98370,
      "training_step_time": 7.901386022567749
    },
    {
      "epoch": 2.3241996765136717e-05,
      "model_forward_time": 1.7371814250946045,
      "step": 98371
    },
    {
      "epoch": 2.3241996765136717e-05,
      "step": 98371,
      "training_step_time": 7.784477472305298
    },
    {
      "epoch": 2.3242950439453125e-05,
      "model_forward_time": 1.7431728839874268,
      "step": 98372
    },
    {
      "epoch": 2.3242950439453125e-05,
      "step": 98372,
      "training_step_time": 7.559115171432495
    },
    {
      "epoch": 2.3243904113769532e-05,
      "model_forward_time": 1.7476260662078857,
      "step": 98373
    },
    {
      "epoch": 2.3243904113769532e-05,
      "step": 98373,
      "training_step_time": 7.829526662826538
    },
    {
      "epoch": 2.3244857788085936e-05,
      "model_forward_time": 1.759650707244873,
      "step": 98374
    },
    {
      "epoch": 2.3244857788085936e-05,
      "step": 98374,
      "training_step_time": 8.095743656158447
    },
    {
      "epoch": 2.3245811462402344e-05,
      "model_forward_time": 1.736236572265625,
      "step": 98375
    },
    {
      "epoch": 2.3245811462402344e-05,
      "step": 98375,
      "training_step_time": 7.517491102218628
    },
    {
      "epoch": 2.324676513671875e-05,
      "model_forward_time": 1.7372024059295654,
      "step": 98376
    },
    {
      "epoch": 2.324676513671875e-05,
      "step": 98376,
      "training_step_time": 7.051029682159424
    },
    {
      "epoch": 2.3247718811035155e-05,
      "model_forward_time": 1.7383320331573486,
      "step": 98377
    },
    {
      "epoch": 2.3247718811035155e-05,
      "step": 98377,
      "training_step_time": 7.100519895553589
    },
    {
      "epoch": 2.3248672485351563e-05,
      "model_forward_time": 1.7409303188323975,
      "step": 98378
    },
    {
      "epoch": 2.3248672485351563e-05,
      "step": 98378,
      "training_step_time": 7.130517244338989
    },
    {
      "epoch": 2.324962615966797e-05,
      "model_forward_time": 2.415159225463867,
      "step": 98379
    },
    {
      "epoch": 2.324962615966797e-05,
      "step": 98379,
      "training_step_time": 9.586800336837769
    },
    {
      "epoch": 2.3250579833984374e-05,
      "grad_norm": 0.10924385488033295,
      "learning_rate": 7.182151692118333e-09,
      "loss": 0.0532,
      "step": 98380
    },
    {
      "epoch": 2.3250579833984374e-05,
      "model_forward_time": 1.7235960960388184,
      "step": 98380
    },
    {
      "dynamics_loss_avg": 0.056299029663205144,
      "epoch": 2.3250579833984374e-05,
      "step": 98380
    },
    {
      "action_loss_avg": 0.0014339076908072457,
      "epoch": 2.3250579833984374e-05,
      "step": 98380
    },
    {
      "epoch": 2.3250579833984374e-05,
      "step": 98380,
      "training_step_time": 7.870541095733643
    },
    {
      "epoch": 2.325153350830078e-05,
      "model_forward_time": 1.7359299659729004,
      "step": 98381
    },
    {
      "epoch": 2.325153350830078e-05,
      "step": 98381,
      "training_step_time": 7.248092174530029
    },
    {
      "epoch": 2.325248718261719e-05,
      "model_forward_time": 1.7293083667755127,
      "step": 98382
    },
    {
      "epoch": 2.325248718261719e-05,
      "step": 98382,
      "training_step_time": 7.85302996635437
    },
    {
      "epoch": 2.3253440856933593e-05,
      "model_forward_time": 0.9958479404449463,
      "step": 98383
    },
    {
      "epoch": 2.3253440856933593e-05,
      "step": 98383,
      "training_step_time": 8.203731060028076
    },
    {
      "epoch": 2.325439453125e-05,
      "model_forward_time": 1.7272703647613525,
      "step": 98384
    },
    {
      "epoch": 2.325439453125e-05,
      "step": 98384,
      "training_step_time": 7.566183090209961
    },
    {
      "epoch": 2.3255348205566408e-05,
      "model_forward_time": 1.7308752536773682,
      "step": 98385
    },
    {
      "epoch": 2.3255348205566408e-05,
      "step": 98385,
      "training_step_time": 7.065322637557983
    },
    {
      "epoch": 2.3256301879882812e-05,
      "model_forward_time": 1.7283334732055664,
      "step": 98386
    },
    {
      "epoch": 2.3256301879882812e-05,
      "step": 98386,
      "training_step_time": 6.989092826843262
    },
    {
      "epoch": 2.325725555419922e-05,
      "model_forward_time": 1.7326195240020752,
      "step": 98387
    },
    {
      "epoch": 2.325725555419922e-05,
      "step": 98387,
      "training_step_time": 7.139523983001709
    },
    {
      "epoch": 2.3258209228515627e-05,
      "model_forward_time": 1.7398791313171387,
      "step": 98388
    },
    {
      "epoch": 2.3258209228515627e-05,
      "step": 98388,
      "training_step_time": 9.923099756240845
    },
    {
      "epoch": 2.325916290283203e-05,
      "model_forward_time": 2.390455961227417,
      "step": 98389
    },
    {
      "epoch": 2.325916290283203e-05,
      "step": 98389,
      "training_step_time": 7.870660305023193
    },
    {
      "epoch": 2.3260116577148438e-05,
      "grad_norm": 0.09525207430124283,
      "learning_rate": 7.093832078449714e-09,
      "loss": 0.0537,
      "step": 98390
    },
    {
      "epoch": 2.3260116577148438e-05,
      "model_forward_time": 1.7266764640808105,
      "step": 98390
    },
    {
      "dynamics_loss_avg": 0.03282849844545126,
      "epoch": 2.3260116577148438e-05,
      "step": 98390
    },
    {
      "action_loss_avg": 0.0008545234915800393,
      "epoch": 2.3260116577148438e-05,
      "step": 98390
    },
    {
      "epoch": 2.3260116577148438e-05,
      "step": 98390,
      "training_step_time": 7.209419250488281
    },
    {
      "epoch": 2.3261070251464842e-05,
      "model_forward_time": 1.7340788841247559,
      "step": 98391
    },
    {
      "epoch": 2.3261070251464842e-05,
      "step": 98391,
      "training_step_time": 8.737366914749146
    },
    {
      "epoch": 2.326202392578125e-05,
      "model_forward_time": 1.7391595840454102,
      "step": 98392
    },
    {
      "epoch": 2.326202392578125e-05,
      "step": 98392,
      "training_step_time": 7.717101097106934
    },
    {
      "epoch": 2.3262977600097657e-05,
      "model_forward_time": 1.7274906635284424,
      "step": 98393
    },
    {
      "epoch": 2.3262977600097657e-05,
      "step": 98393,
      "training_step_time": 8.086631298065186
    },
    {
      "epoch": 2.326393127441406e-05,
      "model_forward_time": 1.7464914321899414,
      "step": 98394
    },
    {
      "epoch": 2.326393127441406e-05,
      "step": 98394,
      "training_step_time": 7.09980320930481
    },
    {
      "epoch": 2.326488494873047e-05,
      "model_forward_time": 1.350135087966919,
      "step": 98395
    },
    {
      "epoch": 2.326488494873047e-05,
      "step": 98395,
      "training_step_time": 7.111231565475464
    },
    {
      "epoch": 2.3265838623046876e-05,
      "model_forward_time": 1.7368903160095215,
      "step": 98396
    },
    {
      "epoch": 2.3265838623046876e-05,
      "step": 98396,
      "training_step_time": 7.086329936981201
    },
    {
      "epoch": 2.326679229736328e-05,
      "model_forward_time": 1.73720121383667,
      "step": 98397
    },
    {
      "epoch": 2.326679229736328e-05,
      "step": 98397,
      "training_step_time": 11.35801386833191
    },
    {
      "epoch": 2.3267745971679687e-05,
      "model_forward_time": 2.3504483699798584,
      "step": 98398
    },
    {
      "epoch": 2.3267745971679687e-05,
      "step": 98398,
      "training_step_time": 8.870728015899658
    },
    {
      "epoch": 2.3268699645996095e-05,
      "model_forward_time": 1.7392752170562744,
      "step": 98399
    },
    {
      "epoch": 2.3268699645996095e-05,
      "step": 98399,
      "training_step_time": 7.767027854919434
    },
    {
      "epoch": 2.32696533203125e-05,
      "grad_norm": 0.1035711020231247,
      "learning_rate": 7.006058481493494e-09,
      "loss": 0.0535,
      "step": 98400
    },
    {
      "epoch": 2.32696533203125e-05,
      "model_forward_time": 1.3340764045715332,
      "step": 98400
    },
    {
      "dynamics_loss_avg": 0.07309964559972286,
      "epoch": 2.32696533203125e-05,
      "step": 98400
    },
    {
      "action_loss_avg": 0.0029194840375566856,
      "epoch": 2.32696533203125e-05,
      "step": 98400
    },
    {
      "epoch": 2.32696533203125e-05,
      "step": 98400,
      "training_step_time": 8.487687587738037
    },
    {
      "epoch": 2.3270606994628906e-05,
      "model_forward_time": 1.741725206375122,
      "step": 98401
    },
    {
      "epoch": 2.3270606994628906e-05,
      "step": 98401,
      "training_step_time": 7.659718751907349
    },
    {
      "epoch": 2.3271560668945313e-05,
      "model_forward_time": 1.7371549606323242,
      "step": 98402
    },
    {
      "epoch": 2.3271560668945313e-05,
      "step": 98402,
      "training_step_time": 8.024492025375366
    },
    {
      "epoch": 2.3272514343261718e-05,
      "model_forward_time": 1.7437644004821777,
      "step": 98403
    },
    {
      "epoch": 2.3272514343261718e-05,
      "step": 98403,
      "training_step_time": 7.0722808837890625
    },
    {
      "epoch": 2.3273468017578125e-05,
      "model_forward_time": 1.7401258945465088,
      "step": 98404
    },
    {
      "epoch": 2.3273468017578125e-05,
      "step": 98404,
      "training_step_time": 7.074121475219727
    },
    {
      "epoch": 2.3274421691894532e-05,
      "model_forward_time": 1.7369139194488525,
      "step": 98405
    },
    {
      "epoch": 2.3274421691894532e-05,
      "step": 98405,
      "training_step_time": 7.062718629837036
    },
    {
      "epoch": 2.3275375366210936e-05,
      "model_forward_time": 1.7396926879882812,
      "step": 98406
    },
    {
      "epoch": 2.3275375366210936e-05,
      "step": 98406,
      "training_step_time": 9.897671937942505
    },
    {
      "epoch": 2.3276329040527344e-05,
      "model_forward_time": 2.4332773685455322,
      "step": 98407
    },
    {
      "epoch": 2.3276329040527344e-05,
      "step": 98407,
      "training_step_time": 9.92627239227295
    },
    {
      "epoch": 2.327728271484375e-05,
      "model_forward_time": 1.742152214050293,
      "step": 98408
    },
    {
      "epoch": 2.327728271484375e-05,
      "step": 98408,
      "training_step_time": 7.828600883483887
    },
    {
      "epoch": 2.3278236389160155e-05,
      "model_forward_time": 1.7358965873718262,
      "step": 98409
    },
    {
      "epoch": 2.3278236389160155e-05,
      "step": 98409,
      "training_step_time": 7.623103380203247
    },
    {
      "epoch": 2.3279190063476563e-05,
      "grad_norm": 0.08346083760261536,
      "learning_rate": 6.918830910847552e-09,
      "loss": 0.0553,
      "step": 98410
    },
    {
      "epoch": 2.3279190063476563e-05,
      "model_forward_time": 1.7271616458892822,
      "step": 98410
    },
    {
      "dynamics_loss_avg": 0.04627828486263752,
      "epoch": 2.3279190063476563e-05,
      "step": 98410
    },
    {
      "action_loss_avg": 0.0014460582693573087,
      "epoch": 2.3279190063476563e-05,
      "step": 98410
    },
    {
      "epoch": 2.3279190063476563e-05,
      "step": 98410,
      "training_step_time": 8.763572692871094
    },
    {
      "epoch": 2.328014373779297e-05,
      "model_forward_time": 1.7398076057434082,
      "step": 98411
    },
    {
      "epoch": 2.328014373779297e-05,
      "step": 98411,
      "training_step_time": 8.023639917373657
    },
    {
      "epoch": 2.3281097412109374e-05,
      "model_forward_time": 1.7348575592041016,
      "step": 98412
    },
    {
      "epoch": 2.3281097412109374e-05,
      "step": 98412,
      "training_step_time": 7.508971929550171
    },
    {
      "epoch": 2.328205108642578e-05,
      "model_forward_time": 0.9993531703948975,
      "step": 98413
    },
    {
      "epoch": 2.328205108642578e-05,
      "step": 98413,
      "training_step_time": 7.099627494812012
    },
    {
      "epoch": 2.328300476074219e-05,
      "model_forward_time": 1.7396252155303955,
      "step": 98414
    },
    {
      "epoch": 2.328300476074219e-05,
      "step": 98414,
      "training_step_time": 7.055447816848755
    },
    {
      "epoch": 2.3283958435058593e-05,
      "model_forward_time": 1.7404534816741943,
      "step": 98415
    },
    {
      "epoch": 2.3283958435058593e-05,
      "step": 98415,
      "training_step_time": 9.272088050842285
    },
    {
      "epoch": 2.3284912109375e-05,
      "model_forward_time": 2.4203381538391113,
      "step": 98416
    },
    {
      "epoch": 2.3284912109375e-05,
      "step": 98416,
      "training_step_time": 8.638253927230835
    },
    {
      "epoch": 2.3285865783691408e-05,
      "model_forward_time": 1.7393300533294678,
      "step": 98417
    },
    {
      "epoch": 2.3285865783691408e-05,
      "step": 98417,
      "training_step_time": 7.8305439949035645
    },
    {
      "epoch": 2.3286819458007812e-05,
      "model_forward_time": 1.7374250888824463,
      "step": 98418
    },
    {
      "epoch": 2.3286819458007812e-05,
      "step": 98418,
      "training_step_time": 7.556659460067749
    },
    {
      "epoch": 2.328777313232422e-05,
      "model_forward_time": 1.735661268234253,
      "step": 98419
    },
    {
      "epoch": 2.328777313232422e-05,
      "step": 98419,
      "training_step_time": 7.799510478973389
    },
    {
      "epoch": 2.3288726806640627e-05,
      "grad_norm": 0.14687898755073547,
      "learning_rate": 6.832149376052033e-09,
      "loss": 0.0534,
      "step": 98420
    },
    {
      "epoch": 2.3288726806640627e-05,
      "model_forward_time": 1.7235283851623535,
      "step": 98420
    },
    {
      "dynamics_loss_avg": 0.0494460329413414,
      "epoch": 2.3288726806640627e-05,
      "step": 98420
    },
    {
      "action_loss_avg": 0.001084988568618428,
      "epoch": 2.3288726806640627e-05,
      "step": 98420
    },
    {
      "epoch": 2.3288726806640627e-05,
      "step": 98420,
      "training_step_time": 8.050116539001465
    },
    {
      "epoch": 2.328968048095703e-05,
      "model_forward_time": 1.7365305423736572,
      "step": 98421
    },
    {
      "epoch": 2.328968048095703e-05,
      "step": 98421,
      "training_step_time": 7.515206813812256
    },
    {
      "epoch": 2.3290634155273438e-05,
      "model_forward_time": 1.7365126609802246,
      "step": 98422
    },
    {
      "epoch": 2.3290634155273438e-05,
      "step": 98422,
      "training_step_time": 7.058773994445801
    },
    {
      "epoch": 2.3291587829589842e-05,
      "model_forward_time": 1.7359144687652588,
      "step": 98423
    },
    {
      "epoch": 2.3291587829589842e-05,
      "step": 98423,
      "training_step_time": 7.103955268859863
    },
    {
      "epoch": 2.329254150390625e-05,
      "model_forward_time": 1.7345881462097168,
      "step": 98424
    },
    {
      "epoch": 2.329254150390625e-05,
      "step": 98424,
      "training_step_time": 10.31721568107605
    },
    {
      "epoch": 2.3293495178222657e-05,
      "model_forward_time": 1.8239080905914307,
      "step": 98425
    },
    {
      "epoch": 2.3293495178222657e-05,
      "step": 98425,
      "training_step_time": 9.615750074386597
    },
    {
      "epoch": 2.329444885253906e-05,
      "model_forward_time": 2.408163547515869,
      "step": 98426
    },
    {
      "epoch": 2.329444885253906e-05,
      "step": 98426,
      "training_step_time": 8.705912590026855
    },
    {
      "epoch": 2.329540252685547e-05,
      "model_forward_time": 1.7329564094543457,
      "step": 98427
    },
    {
      "epoch": 2.329540252685547e-05,
      "step": 98427,
      "training_step_time": 7.760121822357178
    },
    {
      "epoch": 2.3296356201171876e-05,
      "model_forward_time": 1.7383761405944824,
      "step": 98428
    },
    {
      "epoch": 2.3296356201171876e-05,
      "step": 98428,
      "training_step_time": 7.827730178833008
    },
    {
      "epoch": 2.329730987548828e-05,
      "model_forward_time": 1.7458829879760742,
      "step": 98429
    },
    {
      "epoch": 2.329730987548828e-05,
      "step": 98429,
      "training_step_time": 7.753768444061279
    },
    {
      "epoch": 2.3298263549804687e-05,
      "grad_norm": 0.104713074862957,
      "learning_rate": 6.746013886584912e-09,
      "loss": 0.0526,
      "step": 98430
    },
    {
      "epoch": 2.3298263549804687e-05,
      "model_forward_time": 1.7347135543823242,
      "step": 98430
    },
    {
      "dynamics_loss_avg": 0.05315630435943604,
      "epoch": 2.3298263549804687e-05,
      "step": 98430
    },
    {
      "action_loss_avg": 0.001906400895677507,
      "epoch": 2.3298263549804687e-05,
      "step": 98430
    },
    {
      "epoch": 2.3298263549804687e-05,
      "step": 98430,
      "training_step_time": 8.093183279037476
    },
    {
      "epoch": 2.3299217224121095e-05,
      "model_forward_time": 1.738250732421875,
      "step": 98431
    },
    {
      "epoch": 2.3299217224121095e-05,
      "step": 98431,
      "training_step_time": 7.066592454910278
    },
    {
      "epoch": 2.33001708984375e-05,
      "model_forward_time": 1.7462191581726074,
      "step": 98432
    },
    {
      "epoch": 2.33001708984375e-05,
      "step": 98432,
      "training_step_time": 7.0434417724609375
    },
    {
      "epoch": 2.3301124572753906e-05,
      "model_forward_time": 1.7411935329437256,
      "step": 98433
    },
    {
      "epoch": 2.3301124572753906e-05,
      "step": 98433,
      "training_step_time": 8.689831256866455
    },
    {
      "epoch": 2.3302078247070314e-05,
      "model_forward_time": 1.8011150360107422,
      "step": 98434
    },
    {
      "epoch": 2.3302078247070314e-05,
      "step": 98434,
      "training_step_time": 7.903782606124878
    },
    {
      "epoch": 2.3303031921386718e-05,
      "model_forward_time": 2.412588119506836,
      "step": 98435
    },
    {
      "epoch": 2.3303031921386718e-05,
      "step": 98435,
      "training_step_time": 7.80475926399231
    },
    {
      "epoch": 2.3303985595703125e-05,
      "model_forward_time": 1.7372949123382568,
      "step": 98436
    },
    {
      "epoch": 2.3303985595703125e-05,
      "step": 98436,
      "training_step_time": 7.755244731903076
    },
    {
      "epoch": 2.3304939270019532e-05,
      "model_forward_time": 1.7321126461029053,
      "step": 98437
    },
    {
      "epoch": 2.3304939270019532e-05,
      "step": 98437,
      "training_step_time": 7.744365215301514
    },
    {
      "epoch": 2.3305892944335936e-05,
      "model_forward_time": 1.7358827590942383,
      "step": 98438
    },
    {
      "epoch": 2.3305892944335936e-05,
      "step": 98438,
      "training_step_time": 7.704423666000366
    },
    {
      "epoch": 2.3306846618652344e-05,
      "model_forward_time": 1.7354950904846191,
      "step": 98439
    },
    {
      "epoch": 2.3306846618652344e-05,
      "step": 98439,
      "training_step_time": 8.12490439414978
    },
    {
      "epoch": 2.330780029296875e-05,
      "grad_norm": 0.07984024286270142,
      "learning_rate": 6.6604244518664315e-09,
      "loss": 0.0548,
      "step": 98440
    },
    {
      "epoch": 2.330780029296875e-05,
      "model_forward_time": 1.7230103015899658,
      "step": 98440
    },
    {
      "dynamics_loss_avg": 0.043717974610626695,
      "epoch": 2.330780029296875e-05,
      "step": 98440
    },
    {
      "action_loss_avg": 0.0016628465382382274,
      "epoch": 2.330780029296875e-05,
      "step": 98440
    },
    {
      "epoch": 2.330780029296875e-05,
      "step": 98440,
      "training_step_time": 7.0399792194366455
    },
    {
      "epoch": 2.3308753967285155e-05,
      "model_forward_time": 1.7306101322174072,
      "step": 98441
    },
    {
      "epoch": 2.3308753967285155e-05,
      "step": 98441,
      "training_step_time": 7.074692487716675
    },
    {
      "epoch": 2.3309707641601563e-05,
      "model_forward_time": 1.7301008701324463,
      "step": 98442
    },
    {
      "epoch": 2.3309707641601563e-05,
      "step": 98442,
      "training_step_time": 8.647058248519897
    },
    {
      "epoch": 2.331066131591797e-05,
      "model_forward_time": 1.7366695404052734,
      "step": 98443
    },
    {
      "epoch": 2.331066131591797e-05,
      "step": 98443,
      "training_step_time": 9.805800199508667
    },
    {
      "epoch": 2.3311614990234374e-05,
      "model_forward_time": 2.378411293029785,
      "step": 98444
    },
    {
      "epoch": 2.3311614990234374e-05,
      "step": 98444,
      "training_step_time": 7.870278835296631
    },
    {
      "epoch": 2.331256866455078e-05,
      "model_forward_time": 1.7354159355163574,
      "step": 98445
    },
    {
      "epoch": 2.331256866455078e-05,
      "step": 98445,
      "training_step_time": 7.7560875415802
    },
    {
      "epoch": 2.331352233886719e-05,
      "model_forward_time": 1.7325856685638428,
      "step": 98446
    },
    {
      "epoch": 2.331352233886719e-05,
      "step": 98446,
      "training_step_time": 7.754932403564453
    },
    {
      "epoch": 2.3314476013183593e-05,
      "model_forward_time": 1.7302677631378174,
      "step": 98447
    },
    {
      "epoch": 2.3314476013183593e-05,
      "step": 98447,
      "training_step_time": 7.810803413391113
    },
    {
      "epoch": 2.33154296875e-05,
      "model_forward_time": 1.7338685989379883,
      "step": 98448
    },
    {
      "epoch": 2.33154296875e-05,
      "step": 98448,
      "training_step_time": 8.143859148025513
    },
    {
      "epoch": 2.3316383361816408e-05,
      "model_forward_time": 1.7280724048614502,
      "step": 98449
    },
    {
      "epoch": 2.3316383361816408e-05,
      "step": 98449,
      "training_step_time": 7.141034126281738
    },
    {
      "epoch": 2.3317337036132812e-05,
      "grad_norm": 0.10493145883083344,
      "learning_rate": 6.575381081256326e-09,
      "loss": 0.056,
      "step": 98450
    },
    {
      "epoch": 2.3317337036132812e-05,
      "model_forward_time": 1.7204346656799316,
      "step": 98450
    },
    {
      "dynamics_loss_avg": 0.048373892158269885,
      "epoch": 2.3317337036132812e-05,
      "step": 98450
    },
    {
      "action_loss_avg": 0.0007927321246825158,
      "epoch": 2.3317337036132812e-05,
      "step": 98450
    },
    {
      "epoch": 2.3317337036132812e-05,
      "step": 98450,
      "training_step_time": 7.019536018371582
    },
    {
      "epoch": 2.331829071044922e-05,
      "model_forward_time": 1.7295639514923096,
      "step": 98451
    },
    {
      "epoch": 2.331829071044922e-05,
      "step": 98451,
      "training_step_time": 7.81265926361084
    },
    {
      "epoch": 2.3319244384765627e-05,
      "model_forward_time": 1.73486328125,
      "step": 98452
    },
    {
      "epoch": 2.3319244384765627e-05,
      "step": 98452,
      "training_step_time": 7.88697624206543
    },
    {
      "epoch": 2.332019805908203e-05,
      "model_forward_time": 2.363373041152954,
      "step": 98453
    },
    {
      "epoch": 2.332019805908203e-05,
      "step": 98453,
      "training_step_time": 8.905548334121704
    },
    {
      "epoch": 2.3321151733398438e-05,
      "model_forward_time": 1.7274985313415527,
      "step": 98454
    },
    {
      "epoch": 2.3321151733398438e-05,
      "step": 98454,
      "training_step_time": 7.853996753692627
    },
    {
      "epoch": 2.3322105407714842e-05,
      "model_forward_time": 1.7242367267608643,
      "step": 98455
    },
    {
      "epoch": 2.3322105407714842e-05,
      "step": 98455,
      "training_step_time": 7.644356966018677
    },
    {
      "epoch": 2.332305908203125e-05,
      "model_forward_time": 1.72861909866333,
      "step": 98456
    },
    {
      "epoch": 2.332305908203125e-05,
      "step": 98456,
      "training_step_time": 7.720088243484497
    },
    {
      "epoch": 2.3324012756347657e-05,
      "model_forward_time": 1.719505786895752,
      "step": 98457
    },
    {
      "epoch": 2.3324012756347657e-05,
      "step": 98457,
      "training_step_time": 8.10443663597107
    },
    {
      "epoch": 2.332496643066406e-05,
      "model_forward_time": 1.7359049320220947,
      "step": 98458
    },
    {
      "epoch": 2.332496643066406e-05,
      "step": 98458,
      "training_step_time": 7.495717287063599
    },
    {
      "epoch": 2.332592010498047e-05,
      "model_forward_time": 1.7345051765441895,
      "step": 98459
    },
    {
      "epoch": 2.332592010498047e-05,
      "step": 98459,
      "training_step_time": 6.9865500926971436
    },
    {
      "epoch": 2.3326873779296876e-05,
      "grad_norm": 0.08323987573385239,
      "learning_rate": 6.490883784055491e-09,
      "loss": 0.0553,
      "step": 98460
    },
    {
      "epoch": 2.3326873779296876e-05,
      "model_forward_time": 1.732173204421997,
      "step": 98460
    },
    {
      "dynamics_loss_avg": 0.03991312123835087,
      "epoch": 2.3326873779296876e-05,
      "step": 98460
    },
    {
      "action_loss_avg": 0.0009153059450909496,
      "epoch": 2.3326873779296876e-05,
      "step": 98460
    },
    {
      "epoch": 2.3326873779296876e-05,
      "step": 98460,
      "training_step_time": 9.262580394744873
    },
    {
      "epoch": 2.332782745361328e-05,
      "model_forward_time": 1.7359435558319092,
      "step": 98461
    },
    {
      "epoch": 2.332782745361328e-05,
      "step": 98461,
      "training_step_time": 8.600315570831299
    },
    {
      "epoch": 2.3328781127929687e-05,
      "model_forward_time": 2.4210619926452637,
      "step": 98462
    },
    {
      "epoch": 2.3328781127929687e-05,
      "step": 98462,
      "training_step_time": 9.701275110244751
    },
    {
      "epoch": 2.3329734802246095e-05,
      "model_forward_time": 1.7361421585083008,
      "step": 98463
    },
    {
      "epoch": 2.3329734802246095e-05,
      "step": 98463,
      "training_step_time": 9.481522560119629
    },
    {
      "epoch": 2.33306884765625e-05,
      "model_forward_time": 1.732529640197754,
      "step": 98464
    },
    {
      "epoch": 2.33306884765625e-05,
      "step": 98464,
      "training_step_time": 8.175589323043823
    },
    {
      "epoch": 2.3331642150878906e-05,
      "model_forward_time": 1.7374544143676758,
      "step": 98465
    },
    {
      "epoch": 2.3331642150878906e-05,
      "step": 98465,
      "training_step_time": 8.798024415969849
    },
    {
      "epoch": 2.3332595825195314e-05,
      "model_forward_time": 1.732905626296997,
      "step": 98466
    },
    {
      "epoch": 2.3332595825195314e-05,
      "step": 98466,
      "training_step_time": 7.0628814697265625
    },
    {
      "epoch": 2.3333549499511718e-05,
      "model_forward_time": 1.7359144687652588,
      "step": 98467
    },
    {
      "epoch": 2.3333549499511718e-05,
      "step": 98467,
      "training_step_time": 8.134750604629517
    },
    {
      "epoch": 2.3334503173828125e-05,
      "model_forward_time": 1.7447583675384521,
      "step": 98468
    },
    {
      "epoch": 2.3334503173828125e-05,
      "step": 98468,
      "training_step_time": 7.02176308631897
    },
    {
      "epoch": 2.3335456848144532e-05,
      "model_forward_time": 1.736419916152954,
      "step": 98469
    },
    {
      "epoch": 2.3335456848144532e-05,
      "step": 98469,
      "training_step_time": 8.380260229110718
    },
    {
      "epoch": 2.3336410522460936e-05,
      "grad_norm": 0.0919906347990036,
      "learning_rate": 6.406932569503754e-09,
      "loss": 0.0538,
      "step": 98470
    },
    {
      "epoch": 2.3336410522460936e-05,
      "model_forward_time": 1.7230308055877686,
      "step": 98470
    },
    {
      "dynamics_loss_avg": 0.05266780164092779,
      "epoch": 2.3336410522460936e-05,
      "step": 98470
    },
    {
      "action_loss_avg": 0.0011249771283473819,
      "epoch": 2.3336410522460936e-05,
      "step": 98470
    },
    {
      "epoch": 2.3336410522460936e-05,
      "step": 98470,
      "training_step_time": 7.92705225944519
    },
    {
      "epoch": 2.3337364196777344e-05,
      "model_forward_time": 1.3740103244781494,
      "step": 98471
    },
    {
      "epoch": 2.3337364196777344e-05,
      "step": 98471,
      "training_step_time": 9.229517221450806
    },
    {
      "epoch": 2.333831787109375e-05,
      "model_forward_time": 2.376093626022339,
      "step": 98472
    },
    {
      "epoch": 2.333831787109375e-05,
      "step": 98472,
      "training_step_time": 7.855633735656738
    },
    {
      "epoch": 2.3339271545410155e-05,
      "model_forward_time": 1.7348363399505615,
      "step": 98473
    },
    {
      "epoch": 2.3339271545410155e-05,
      "step": 98473,
      "training_step_time": 7.692533254623413
    },
    {
      "epoch": 2.3340225219726563e-05,
      "model_forward_time": 1.7424850463867188,
      "step": 98474
    },
    {
      "epoch": 2.3340225219726563e-05,
      "step": 98474,
      "training_step_time": 7.7145302295684814
    },
    {
      "epoch": 2.334117889404297e-05,
      "model_forward_time": 1.7409446239471436,
      "step": 98475
    },
    {
      "epoch": 2.334117889404297e-05,
      "step": 98475,
      "training_step_time": 7.712515115737915
    },
    {
      "epoch": 2.3342132568359374e-05,
      "model_forward_time": 1.735731840133667,
      "step": 98476
    },
    {
      "epoch": 2.3342132568359374e-05,
      "step": 98476,
      "training_step_time": 7.96317195892334
    },
    {
      "epoch": 2.334308624267578e-05,
      "model_forward_time": 1.7265970706939697,
      "step": 98477
    },
    {
      "epoch": 2.334308624267578e-05,
      "step": 98477,
      "training_step_time": 6.914664030075073
    },
    {
      "epoch": 2.334403991699219e-05,
      "model_forward_time": 1.7372539043426514,
      "step": 98478
    },
    {
      "epoch": 2.334403991699219e-05,
      "step": 98478,
      "training_step_time": 7.799329042434692
    },
    {
      "epoch": 2.3344993591308593e-05,
      "model_forward_time": 1.7368159294128418,
      "step": 98479
    },
    {
      "epoch": 2.3344993591308593e-05,
      "step": 98479,
      "training_step_time": 8.584475040435791
    },
    {
      "epoch": 2.3345947265625e-05,
      "grad_norm": 0.08280787616968155,
      "learning_rate": 6.323527446782663e-09,
      "loss": 0.0541,
      "step": 98480
    },
    {
      "epoch": 2.3345947265625e-05,
      "model_forward_time": 1.7722513675689697,
      "step": 98480
    },
    {
      "dynamics_loss_avg": 0.06282532922923564,
      "epoch": 2.3345947265625e-05,
      "step": 98480
    },
    {
      "action_loss_avg": 0.0016662459558574483,
      "epoch": 2.3345947265625e-05,
      "step": 98480
    },
    {
      "epoch": 2.3345947265625e-05,
      "step": 98480,
      "training_step_time": 10.44981837272644
    },
    {
      "epoch": 2.3346900939941408e-05,
      "model_forward_time": 1.972092628479004,
      "step": 98481
    },
    {
      "epoch": 2.3346900939941408e-05,
      "step": 98481,
      "training_step_time": 8.395604610443115
    },
    {
      "epoch": 2.3347854614257812e-05,
      "model_forward_time": 1.7376704216003418,
      "step": 98482
    },
    {
      "epoch": 2.3347854614257812e-05,
      "step": 98482,
      "training_step_time": 7.6868884563446045
    },
    {
      "epoch": 2.334880828857422e-05,
      "model_forward_time": 1.731200933456421,
      "step": 98483
    },
    {
      "epoch": 2.334880828857422e-05,
      "step": 98483,
      "training_step_time": 7.7159810066223145
    },
    {
      "epoch": 2.3349761962890627e-05,
      "model_forward_time": 1.7320654392242432,
      "step": 98484
    },
    {
      "epoch": 2.3349761962890627e-05,
      "step": 98484,
      "training_step_time": 7.701308250427246
    },
    {
      "epoch": 2.335071563720703e-05,
      "model_forward_time": 1.7313711643218994,
      "step": 98485
    },
    {
      "epoch": 2.335071563720703e-05,
      "step": 98485,
      "training_step_time": 8.722095966339111
    },
    {
      "epoch": 2.3351669311523438e-05,
      "model_forward_time": 1.7324676513671875,
      "step": 98486
    },
    {
      "epoch": 2.3351669311523438e-05,
      "step": 98486,
      "training_step_time": 6.889360189437866
    },
    {
      "epoch": 2.3352622985839842e-05,
      "model_forward_time": 1.726043939590454,
      "step": 98487
    },
    {
      "epoch": 2.3352622985839842e-05,
      "step": 98487,
      "training_step_time": 7.843371868133545
    },
    {
      "epoch": 2.335357666015625e-05,
      "model_forward_time": 1.7285385131835938,
      "step": 98488
    },
    {
      "epoch": 2.335357666015625e-05,
      "step": 98488,
      "training_step_time": 7.906895160675049
    },
    {
      "epoch": 2.3354530334472657e-05,
      "model_forward_time": 1.7538032531738281,
      "step": 98489
    },
    {
      "epoch": 2.3354530334472657e-05,
      "step": 98489,
      "training_step_time": 11.992548942565918
    },
    {
      "epoch": 2.335548400878906e-05,
      "grad_norm": 0.07436054199934006,
      "learning_rate": 6.240668425012142e-09,
      "loss": 0.0536,
      "step": 98490
    },
    {
      "epoch": 2.335548400878906e-05,
      "model_forward_time": 2.414268970489502,
      "step": 98490
    },
    {
      "dynamics_loss_avg": 0.046672120504081246,
      "epoch": 2.335548400878906e-05,
      "step": 98490
    },
    {
      "action_loss_avg": 0.001436393684707582,
      "epoch": 2.335548400878906e-05,
      "step": 98490
    },
    {
      "epoch": 2.335548400878906e-05,
      "step": 98490,
      "training_step_time": 9.65471887588501
    },
    {
      "epoch": 2.335643768310547e-05,
      "model_forward_time": 1.7280912399291992,
      "step": 98491
    },
    {
      "epoch": 2.335643768310547e-05,
      "step": 98491,
      "training_step_time": 7.80293869972229
    },
    {
      "epoch": 2.3357391357421876e-05,
      "model_forward_time": 1.7352371215820312,
      "step": 98492
    },
    {
      "epoch": 2.3357391357421876e-05,
      "step": 98492,
      "training_step_time": 9.078987121582031
    },
    {
      "epoch": 2.335834503173828e-05,
      "model_forward_time": 1.731623888015747,
      "step": 98493
    },
    {
      "epoch": 2.335834503173828e-05,
      "step": 98493,
      "training_step_time": 7.7790679931640625
    },
    {
      "epoch": 2.3359298706054687e-05,
      "model_forward_time": 1.731914758682251,
      "step": 98494
    },
    {
      "epoch": 2.3359298706054687e-05,
      "step": 98494,
      "training_step_time": 7.054310321807861
    },
    {
      "epoch": 2.3360252380371095e-05,
      "model_forward_time": 1.7287440299987793,
      "step": 98495
    },
    {
      "epoch": 2.3360252380371095e-05,
      "step": 98495,
      "training_step_time": 6.986562967300415
    },
    {
      "epoch": 2.33612060546875e-05,
      "model_forward_time": 1.7423615455627441,
      "step": 98496
    },
    {
      "epoch": 2.33612060546875e-05,
      "step": 98496,
      "training_step_time": 9.459538221359253
    },
    {
      "epoch": 2.3362159729003906e-05,
      "model_forward_time": 1.7306482791900635,
      "step": 98497
    },
    {
      "epoch": 2.3362159729003906e-05,
      "step": 98497,
      "training_step_time": 7.928272724151611
    },
    {
      "epoch": 2.3363113403320314e-05,
      "model_forward_time": 1.7236871719360352,
      "step": 98498
    },
    {
      "epoch": 2.3363113403320314e-05,
      "step": 98498,
      "training_step_time": 8.71947717666626
    },
    {
      "epoch": 2.3364067077636718e-05,
      "model_forward_time": 1.8305282592773438,
      "step": 98499
    },
    {
      "epoch": 2.3364067077636718e-05,
      "step": 98499,
      "training_step_time": 9.222871541976929
    },
    {
      "epoch": 2.3365020751953125e-05,
      "grad_norm": 0.11750344187021255,
      "learning_rate": 6.1583555132543886e-09,
      "loss": 0.0556,
      "step": 98500
    },
    {
      "epoch": 2.3365020751953125e-05,
      "model_forward_time": 1.7189106941223145,
      "step": 98500
    },
    {
      "dynamics_loss_avg": 0.06601083502173424,
      "epoch": 2.3365020751953125e-05,
      "step": 98500
    },
    {
      "action_loss_avg": 0.0014551670552464201,
      "epoch": 2.3365020751953125e-05,
      "step": 98500
    },
    {
      "epoch": 2.3365020751953125e-05,
      "step": 98500,
      "training_step_time": 7.776757478713989
    },
    {
      "epoch": 2.3365974426269532e-05,
      "model_forward_time": 1.730262041091919,
      "step": 98501
    },
    {
      "epoch": 2.3365974426269532e-05,
      "step": 98501,
      "training_step_time": 7.638357639312744
    },
    {
      "epoch": 2.3366928100585936e-05,
      "model_forward_time": 1.7323429584503174,
      "step": 98502
    },
    {
      "epoch": 2.3366928100585936e-05,
      "step": 98502,
      "training_step_time": 7.7224860191345215
    },
    {
      "epoch": 2.3367881774902344e-05,
      "model_forward_time": 1.72719407081604,
      "step": 98503
    },
    {
      "epoch": 2.3367881774902344e-05,
      "step": 98503,
      "training_step_time": 6.885644435882568
    },
    {
      "epoch": 2.336883544921875e-05,
      "model_forward_time": 1.72957444190979,
      "step": 98504
    },
    {
      "epoch": 2.336883544921875e-05,
      "step": 98504,
      "training_step_time": 7.8166728019714355
    },
    {
      "epoch": 2.3369789123535155e-05,
      "model_forward_time": 1.7343335151672363,
      "step": 98505
    },
    {
      "epoch": 2.3369789123535155e-05,
      "step": 98505,
      "training_step_time": 8.521090745925903
    },
    {
      "epoch": 2.3370742797851563e-05,
      "model_forward_time": 1.734703540802002,
      "step": 98506
    },
    {
      "epoch": 2.3370742797851563e-05,
      "step": 98506,
      "training_step_time": 8.683525562286377
    },
    {
      "epoch": 2.337169647216797e-05,
      "model_forward_time": 1.7342865467071533,
      "step": 98507
    },
    {
      "epoch": 2.337169647216797e-05,
      "step": 98507,
      "training_step_time": 9.602355480194092
    },
    {
      "epoch": 2.3372650146484374e-05,
      "model_forward_time": 1.8230154514312744,
      "step": 98508
    },
    {
      "epoch": 2.3372650146484374e-05,
      "step": 98508,
      "training_step_time": 9.007497549057007
    },
    {
      "epoch": 2.337360382080078e-05,
      "model_forward_time": 2.3935608863830566,
      "step": 98509
    },
    {
      "epoch": 2.337360382080078e-05,
      "step": 98509,
      "training_step_time": 8.676731824874878
    },
    {
      "epoch": 2.337455749511719e-05,
      "grad_norm": 0.10668373852968216,
      "learning_rate": 6.076588720510535e-09,
      "loss": 0.0552,
      "step": 98510
    },
    {
      "epoch": 2.337455749511719e-05,
      "model_forward_time": 1.7240688800811768,
      "step": 98510
    },
    {
      "dynamics_loss_avg": 0.04250526502728462,
      "epoch": 2.337455749511719e-05,
      "step": 98510
    },
    {
      "action_loss_avg": 0.0019931508111767473,
      "epoch": 2.337455749511719e-05,
      "step": 98510
    },
    {
      "epoch": 2.337455749511719e-05,
      "step": 98510,
      "training_step_time": 7.667428970336914
    },
    {
      "epoch": 2.3375511169433593e-05,
      "model_forward_time": 1.349027395248413,
      "step": 98511
    },
    {
      "epoch": 2.3375511169433593e-05,
      "step": 98511,
      "training_step_time": 7.6680285930633545
    },
    {
      "epoch": 2.337646484375e-05,
      "model_forward_time": 1.7456681728363037,
      "step": 98512
    },
    {
      "epoch": 2.337646484375e-05,
      "step": 98512,
      "training_step_time": 7.7085113525390625
    },
    {
      "epoch": 2.3377418518066408e-05,
      "model_forward_time": 1.729254961013794,
      "step": 98513
    },
    {
      "epoch": 2.3377418518066408e-05,
      "step": 98513,
      "training_step_time": 7.726818084716797
    },
    {
      "epoch": 2.3378372192382812e-05,
      "model_forward_time": 1.7285830974578857,
      "step": 98514
    },
    {
      "epoch": 2.3378372192382812e-05,
      "step": 98514,
      "training_step_time": 7.603726625442505
    },
    {
      "epoch": 2.337932586669922e-05,
      "model_forward_time": 1.7394094467163086,
      "step": 98515
    },
    {
      "epoch": 2.337932586669922e-05,
      "step": 98515,
      "training_step_time": 8.470343589782715
    },
    {
      "epoch": 2.3380279541015623e-05,
      "model_forward_time": 1.7292020320892334,
      "step": 98516
    },
    {
      "epoch": 2.3380279541015623e-05,
      "step": 98516,
      "training_step_time": 8.683488607406616
    },
    {
      "epoch": 2.338123321533203e-05,
      "model_forward_time": 1.825601577758789,
      "step": 98517
    },
    {
      "epoch": 2.338123321533203e-05,
      "step": 98517,
      "training_step_time": 8.890362024307251
    },
    {
      "epoch": 2.3382186889648438e-05,
      "model_forward_time": 2.3949859142303467,
      "step": 98518
    },
    {
      "epoch": 2.3382186889648438e-05,
      "step": 98518,
      "training_step_time": 8.535179138183594
    },
    {
      "epoch": 2.3383140563964842e-05,
      "model_forward_time": 1.730057716369629,
      "step": 98519
    },
    {
      "epoch": 2.3383140563964842e-05,
      "step": 98519,
      "training_step_time": 7.79509711265564
    },
    {
      "epoch": 2.338409423828125e-05,
      "grad_norm": 0.07853110879659653,
      "learning_rate": 5.995368055722317e-09,
      "loss": 0.0544,
      "step": 98520
    },
    {
      "epoch": 2.338409423828125e-05,
      "model_forward_time": 1.7475812435150146,
      "step": 98520
    },
    {
      "dynamics_loss_avg": 0.04657671377062798,
      "epoch": 2.338409423828125e-05,
      "step": 98520
    },
    {
      "action_loss_avg": 0.0008788035629549995,
      "epoch": 2.338409423828125e-05,
      "step": 98520
    },
    {
      "epoch": 2.338409423828125e-05,
      "step": 98520,
      "training_step_time": 7.473737478256226
    },
    {
      "epoch": 2.3385047912597657e-05,
      "model_forward_time": 1.7412490844726562,
      "step": 98521
    },
    {
      "epoch": 2.3385047912597657e-05,
      "step": 98521,
      "training_step_time": 7.709003448486328
    },
    {
      "epoch": 2.338600158691406e-05,
      "model_forward_time": 1.7396037578582764,
      "step": 98522
    },
    {
      "epoch": 2.338600158691406e-05,
      "step": 98522,
      "training_step_time": 7.726861000061035
    },
    {
      "epoch": 2.338695526123047e-05,
      "model_forward_time": 1.7320623397827148,
      "step": 98523
    },
    {
      "epoch": 2.338695526123047e-05,
      "step": 98523,
      "training_step_time": 8.715384006500244
    },
    {
      "epoch": 2.3387908935546876e-05,
      "model_forward_time": 1.7373132705688477,
      "step": 98524
    },
    {
      "epoch": 2.3387908935546876e-05,
      "step": 98524,
      "training_step_time": 7.948771238327026
    },
    {
      "epoch": 2.338886260986328e-05,
      "model_forward_time": 1.7353971004486084,
      "step": 98525
    },
    {
      "epoch": 2.338886260986328e-05,
      "step": 98525,
      "training_step_time": 8.551912069320679
    },
    {
      "epoch": 2.3389816284179687e-05,
      "model_forward_time": 1.8073971271514893,
      "step": 98526
    },
    {
      "epoch": 2.3389816284179687e-05,
      "step": 98526,
      "training_step_time": 9.950003623962402
    },
    {
      "epoch": 2.3390769958496095e-05,
      "model_forward_time": 2.4312236309051514,
      "step": 98527
    },
    {
      "epoch": 2.3390769958496095e-05,
      "step": 98527,
      "training_step_time": 8.444058418273926
    },
    {
      "epoch": 2.33917236328125e-05,
      "model_forward_time": 1.7407422065734863,
      "step": 98528
    },
    {
      "epoch": 2.33917236328125e-05,
      "step": 98528,
      "training_step_time": 9.511184215545654
    },
    {
      "epoch": 2.3392677307128906e-05,
      "model_forward_time": 1.7326748371124268,
      "step": 98529
    },
    {
      "epoch": 2.3392677307128906e-05,
      "step": 98529,
      "training_step_time": 8.638920068740845
    },
    {
      "epoch": 2.3393630981445314e-05,
      "grad_norm": 0.08107957988977432,
      "learning_rate": 5.914693527773186e-09,
      "loss": 0.0575,
      "step": 98530
    },
    {
      "epoch": 2.3393630981445314e-05,
      "model_forward_time": 1.731635570526123,
      "step": 98530
    },
    {
      "dynamics_loss_avg": 0.04948188867419958,
      "epoch": 2.3393630981445314e-05,
      "step": 98530
    },
    {
      "action_loss_avg": 0.0009933335124514997,
      "epoch": 2.3393630981445314e-05,
      "step": 98530
    },
    {
      "epoch": 2.3393630981445314e-05,
      "step": 98530,
      "training_step_time": 7.701159715652466
    },
    {
      "epoch": 2.3394584655761718e-05,
      "model_forward_time": 1.7313783168792725,
      "step": 98531
    },
    {
      "epoch": 2.3394584655761718e-05,
      "step": 98531,
      "training_step_time": 7.693866968154907
    },
    {
      "epoch": 2.3395538330078125e-05,
      "model_forward_time": 1.734708309173584,
      "step": 98532
    },
    {
      "epoch": 2.3395538330078125e-05,
      "step": 98532,
      "training_step_time": 7.442867994308472
    },
    {
      "epoch": 2.3396492004394532e-05,
      "model_forward_time": 1.7380633354187012,
      "step": 98533
    },
    {
      "epoch": 2.3396492004394532e-05,
      "step": 98533,
      "training_step_time": 7.904207944869995
    },
    {
      "epoch": 2.3397445678710936e-05,
      "model_forward_time": 1.7319955825805664,
      "step": 98534
    },
    {
      "epoch": 2.3397445678710936e-05,
      "step": 98534,
      "training_step_time": 7.81768798828125
    },
    {
      "epoch": 2.3398399353027344e-05,
      "model_forward_time": 1.7467961311340332,
      "step": 98535
    },
    {
      "epoch": 2.3398399353027344e-05,
      "step": 98535,
      "training_step_time": 9.585015535354614
    },
    {
      "epoch": 2.339935302734375e-05,
      "model_forward_time": 2.3981332778930664,
      "step": 98536
    },
    {
      "epoch": 2.339935302734375e-05,
      "step": 98536,
      "training_step_time": 8.727602481842041
    },
    {
      "epoch": 2.3400306701660155e-05,
      "model_forward_time": 1.7305607795715332,
      "step": 98537
    },
    {
      "epoch": 2.3400306701660155e-05,
      "step": 98537,
      "training_step_time": 8.632313013076782
    },
    {
      "epoch": 2.3401260375976563e-05,
      "model_forward_time": 1.355248212814331,
      "step": 98538
    },
    {
      "epoch": 2.3401260375976563e-05,
      "step": 98538,
      "training_step_time": 7.490004301071167
    },
    {
      "epoch": 2.340221405029297e-05,
      "model_forward_time": 1.7427375316619873,
      "step": 98539
    },
    {
      "epoch": 2.340221405029297e-05,
      "step": 98539,
      "training_step_time": 7.69709587097168
    },
    {
      "epoch": 2.3403167724609374e-05,
      "grad_norm": 0.07619284838438034,
      "learning_rate": 5.834565145483861e-09,
      "loss": 0.0566,
      "step": 98540
    },
    {
      "epoch": 2.3403167724609374e-05,
      "model_forward_time": 1.7326037883758545,
      "step": 98540
    },
    {
      "dynamics_loss_avg": 0.04948099255561829,
      "epoch": 2.3403167724609374e-05,
      "step": 98540
    },
    {
      "action_loss_avg": 0.001391665084520355,
      "epoch": 2.3403167724609374e-05,
      "step": 98540
    },
    {
      "epoch": 2.3403167724609374e-05,
      "step": 98540,
      "training_step_time": 7.698474884033203
    },
    {
      "epoch": 2.340412139892578e-05,
      "model_forward_time": 1.7386341094970703,
      "step": 98541
    },
    {
      "epoch": 2.340412139892578e-05,
      "step": 98541,
      "training_step_time": 7.812983989715576
    },
    {
      "epoch": 2.340507507324219e-05,
      "model_forward_time": 1.7385058403015137,
      "step": 98542
    },
    {
      "epoch": 2.340507507324219e-05,
      "step": 98542,
      "training_step_time": 7.812463283538818
    },
    {
      "epoch": 2.3406028747558593e-05,
      "model_forward_time": 1.7343344688415527,
      "step": 98543
    },
    {
      "epoch": 2.3406028747558593e-05,
      "step": 98543,
      "training_step_time": 7.856689214706421
    },
    {
      "epoch": 2.3406982421875e-05,
      "model_forward_time": 1.3480730056762695,
      "step": 98544
    },
    {
      "epoch": 2.3406982421875e-05,
      "step": 98544,
      "training_step_time": 9.043837785720825
    },
    {
      "epoch": 2.3407936096191408e-05,
      "model_forward_time": 1.7669682502746582,
      "step": 98545
    },
    {
      "epoch": 2.3407936096191408e-05,
      "step": 98545,
      "training_step_time": 8.593237400054932
    },
    {
      "epoch": 2.3408889770507812e-05,
      "model_forward_time": 2.412179470062256,
      "step": 98546
    },
    {
      "epoch": 2.3408889770507812e-05,
      "step": 98546,
      "training_step_time": 7.773634672164917
    },
    {
      "epoch": 2.340984344482422e-05,
      "model_forward_time": 1.7348082065582275,
      "step": 98547
    },
    {
      "epoch": 2.340984344482422e-05,
      "step": 98547,
      "training_step_time": 7.728760719299316
    },
    {
      "epoch": 2.3410797119140623e-05,
      "model_forward_time": 1.7364857196807861,
      "step": 98548
    },
    {
      "epoch": 2.3410797119140623e-05,
      "step": 98548,
      "training_step_time": 7.704067707061768
    },
    {
      "epoch": 2.341175079345703e-05,
      "model_forward_time": 1.735067367553711,
      "step": 98549
    },
    {
      "epoch": 2.341175079345703e-05,
      "step": 98549,
      "training_step_time": 7.761877775192261
    },
    {
      "epoch": 2.3412704467773438e-05,
      "grad_norm": 0.08046050369739532,
      "learning_rate": 5.754982917618445e-09,
      "loss": 0.0566,
      "step": 98550
    },
    {
      "epoch": 2.3412704467773438e-05,
      "model_forward_time": 1.7296297550201416,
      "step": 98550
    },
    {
      "dynamics_loss_avg": 0.050352615490555766,
      "epoch": 2.3412704467773438e-05,
      "step": 98550
    },
    {
      "action_loss_avg": 0.0013913646922446786,
      "epoch": 2.3412704467773438e-05,
      "step": 98550
    },
    {
      "epoch": 2.3412704467773438e-05,
      "step": 98550,
      "training_step_time": 7.873520374298096
    },
    {
      "epoch": 2.3413658142089842e-05,
      "model_forward_time": 1.7402653694152832,
      "step": 98551
    },
    {
      "epoch": 2.3413658142089842e-05,
      "step": 98551,
      "training_step_time": 8.383303165435791
    },
    {
      "epoch": 2.341461181640625e-05,
      "model_forward_time": 1.733719825744629,
      "step": 98552
    },
    {
      "epoch": 2.341461181640625e-05,
      "step": 98552,
      "training_step_time": 9.654921054840088
    },
    {
      "epoch": 2.3415565490722657e-05,
      "model_forward_time": 1.7400753498077393,
      "step": 98553
    },
    {
      "epoch": 2.3415565490722657e-05,
      "step": 98553,
      "training_step_time": 10.21854305267334
    },
    {
      "epoch": 2.341651916503906e-05,
      "model_forward_time": 1.748831033706665,
      "step": 98554
    },
    {
      "epoch": 2.341651916503906e-05,
      "step": 98554,
      "training_step_time": 8.914108276367188
    },
    {
      "epoch": 2.341747283935547e-05,
      "model_forward_time": 2.4176998138427734,
      "step": 98555
    },
    {
      "epoch": 2.341747283935547e-05,
      "step": 98555,
      "training_step_time": 8.797632217407227
    },
    {
      "epoch": 2.3418426513671876e-05,
      "model_forward_time": 1.7369146347045898,
      "step": 98556
    },
    {
      "epoch": 2.3418426513671876e-05,
      "step": 98556,
      "training_step_time": 8.635792970657349
    },
    {
      "epoch": 2.341938018798828e-05,
      "model_forward_time": 1.7291629314422607,
      "step": 98557
    },
    {
      "epoch": 2.341938018798828e-05,
      "step": 98557,
      "training_step_time": 7.615785121917725
    },
    {
      "epoch": 2.3420333862304687e-05,
      "model_forward_time": 1.7383418083190918,
      "step": 98558
    },
    {
      "epoch": 2.3420333862304687e-05,
      "step": 98558,
      "training_step_time": 7.76065993309021
    },
    {
      "epoch": 2.3421287536621095e-05,
      "model_forward_time": 1.7344114780426025,
      "step": 98559
    },
    {
      "epoch": 2.3421287536621095e-05,
      "step": 98559,
      "training_step_time": 7.8153135776519775
    },
    {
      "epoch": 2.34222412109375e-05,
      "grad_norm": 0.09437756985425949,
      "learning_rate": 5.675946852878867e-09,
      "loss": 0.0564,
      "step": 98560
    },
    {
      "epoch": 2.34222412109375e-05,
      "model_forward_time": 1.7195837497711182,
      "step": 98560
    },
    {
      "dynamics_loss_avg": 0.05880051050335169,
      "epoch": 2.34222412109375e-05,
      "step": 98560
    },
    {
      "action_loss_avg": 0.001140665536513552,
      "epoch": 2.34222412109375e-05,
      "step": 98560
    },
    {
      "epoch": 2.34222412109375e-05,
      "step": 98560,
      "training_step_time": 8.643545389175415
    },
    {
      "epoch": 2.3423194885253906e-05,
      "model_forward_time": 1.7285637855529785,
      "step": 98561
    },
    {
      "epoch": 2.3423194885253906e-05,
      "step": 98561,
      "training_step_time": 7.811559677124023
    },
    {
      "epoch": 2.3424148559570314e-05,
      "model_forward_time": 1.7295105457305908,
      "step": 98562
    },
    {
      "epoch": 2.3424148559570314e-05,
      "step": 98562,
      "training_step_time": 8.408552408218384
    },
    {
      "epoch": 2.3425102233886718e-05,
      "model_forward_time": 1.8107316493988037,
      "step": 98563
    },
    {
      "epoch": 2.3425102233886718e-05,
      "step": 98563,
      "training_step_time": 10.735933780670166
    },
    {
      "epoch": 2.3426055908203125e-05,
      "model_forward_time": 2.414551019668579,
      "step": 98564
    },
    {
      "epoch": 2.3426055908203125e-05,
      "step": 98564,
      "training_step_time": 7.723875999450684
    },
    {
      "epoch": 2.3427009582519532e-05,
      "model_forward_time": 1.7453315258026123,
      "step": 98565
    },
    {
      "epoch": 2.3427009582519532e-05,
      "step": 98565,
      "training_step_time": 7.6154398918151855
    },
    {
      "epoch": 2.3427963256835936e-05,
      "model_forward_time": 1.742581844329834,
      "step": 98566
    },
    {
      "epoch": 2.3427963256835936e-05,
      "step": 98566,
      "training_step_time": 7.342263698577881
    },
    {
      "epoch": 2.3428916931152344e-05,
      "model_forward_time": 1.7435016632080078,
      "step": 98567
    },
    {
      "epoch": 2.3428916931152344e-05,
      "step": 98567,
      "training_step_time": 7.590336561203003
    },
    {
      "epoch": 2.342987060546875e-05,
      "model_forward_time": 1.7447645664215088,
      "step": 98568
    },
    {
      "epoch": 2.342987060546875e-05,
      "step": 98568,
      "training_step_time": 7.625908374786377
    },
    {
      "epoch": 2.3430824279785155e-05,
      "model_forward_time": 1.7428772449493408,
      "step": 98569
    },
    {
      "epoch": 2.3430824279785155e-05,
      "step": 98569,
      "training_step_time": 7.636319875717163
    },
    {
      "epoch": 2.3431777954101563e-05,
      "grad_norm": 0.07476463913917542,
      "learning_rate": 5.59745695990932e-09,
      "loss": 0.0556,
      "step": 98570
    },
    {
      "epoch": 2.3431777954101563e-05,
      "model_forward_time": 1.734621524810791,
      "step": 98570
    },
    {
      "dynamics_loss_avg": 0.06306741833686828,
      "epoch": 2.3431777954101563e-05,
      "step": 98570
    },
    {
      "action_loss_avg": 0.0014800796285271644,
      "epoch": 2.3431777954101563e-05,
      "step": 98570
    },
    {
      "epoch": 2.3431777954101563e-05,
      "step": 98570,
      "training_step_time": 8.887932777404785
    },
    {
      "epoch": 2.343273162841797e-05,
      "model_forward_time": 1.751352071762085,
      "step": 98571
    },
    {
      "epoch": 2.343273162841797e-05,
      "step": 98571,
      "training_step_time": 9.187801837921143
    },
    {
      "epoch": 2.3433685302734374e-05,
      "model_forward_time": 1.7815296649932861,
      "step": 98572
    },
    {
      "epoch": 2.3433685302734374e-05,
      "step": 98572,
      "training_step_time": 9.404623031616211
    },
    {
      "epoch": 2.343463897705078e-05,
      "model_forward_time": 2.3747899532318115,
      "step": 98573
    },
    {
      "epoch": 2.343463897705078e-05,
      "step": 98573,
      "training_step_time": 7.656665563583374
    },
    {
      "epoch": 2.343559265136719e-05,
      "model_forward_time": 1.743619680404663,
      "step": 98574
    },
    {
      "epoch": 2.343559265136719e-05,
      "step": 98574,
      "training_step_time": 7.643546104431152
    },
    {
      "epoch": 2.3436546325683593e-05,
      "model_forward_time": 1.746433973312378,
      "step": 98575
    },
    {
      "epoch": 2.3436546325683593e-05,
      "step": 98575,
      "training_step_time": 7.601269721984863
    },
    {
      "epoch": 2.34375e-05,
      "model_forward_time": 1.752986192703247,
      "step": 98576
    },
    {
      "epoch": 2.34375e-05,
      "step": 98576,
      "training_step_time": 8.656185388565063
    },
    {
      "epoch": 2.3438453674316408e-05,
      "model_forward_time": 1.748556137084961,
      "step": 98577
    },
    {
      "epoch": 2.3438453674316408e-05,
      "step": 98577,
      "training_step_time": 7.62750768661499
    },
    {
      "epoch": 2.3439407348632812e-05,
      "model_forward_time": 1.7525620460510254,
      "step": 98578
    },
    {
      "epoch": 2.3439407348632812e-05,
      "step": 98578,
      "training_step_time": 7.67462944984436
    },
    {
      "epoch": 2.344036102294922e-05,
      "model_forward_time": 1.7569689750671387,
      "step": 98579
    },
    {
      "epoch": 2.344036102294922e-05,
      "step": 98579,
      "training_step_time": 7.872971534729004
    },
    {
      "epoch": 2.3441314697265623e-05,
      "grad_norm": 0.08944351226091385,
      "learning_rate": 5.51951324729294e-09,
      "loss": 0.0557,
      "step": 98580
    },
    {
      "epoch": 2.3441314697265623e-05,
      "model_forward_time": 1.7525036334991455,
      "step": 98580
    },
    {
      "dynamics_loss_avg": 0.04694143515080214,
      "epoch": 2.3441314697265623e-05,
      "step": 98580
    },
    {
      "action_loss_avg": 0.0009516890539089217,
      "epoch": 2.3441314697265623e-05,
      "step": 98580
    },
    {
      "epoch": 2.3441314697265623e-05,
      "step": 98580,
      "training_step_time": 8.701218366622925
    },
    {
      "epoch": 2.344226837158203e-05,
      "model_forward_time": 1.7607676982879639,
      "step": 98581
    },
    {
      "epoch": 2.344226837158203e-05,
      "step": 98581,
      "training_step_time": 9.794135332107544
    },
    {
      "epoch": 2.3443222045898438e-05,
      "model_forward_time": 1.7530584335327148,
      "step": 98582
    },
    {
      "epoch": 2.3443222045898438e-05,
      "step": 98582,
      "training_step_time": 9.200538635253906
    },
    {
      "epoch": 2.3444175720214842e-05,
      "model_forward_time": 1.753394603729248,
      "step": 98583
    },
    {
      "epoch": 2.3444175720214842e-05,
      "step": 98583,
      "training_step_time": 7.659423589706421
    },
    {
      "epoch": 2.344512939453125e-05,
      "model_forward_time": 1.7600243091583252,
      "step": 98584
    },
    {
      "epoch": 2.344512939453125e-05,
      "step": 98584,
      "training_step_time": 7.680919170379639
    },
    {
      "epoch": 2.3446083068847657e-05,
      "model_forward_time": 1.8010175228118896,
      "step": 98585
    },
    {
      "epoch": 2.3446083068847657e-05,
      "step": 98585,
      "training_step_time": 8.71930480003357
    },
    {
      "epoch": 2.344703674316406e-05,
      "model_forward_time": 1.7518224716186523,
      "step": 98586
    },
    {
      "epoch": 2.344703674316406e-05,
      "step": 98586,
      "training_step_time": 8.622842788696289
    },
    {
      "epoch": 2.344799041748047e-05,
      "model_forward_time": 1.778855800628662,
      "step": 98587
    },
    {
      "epoch": 2.344799041748047e-05,
      "step": 98587,
      "training_step_time": 7.58137321472168
    },
    {
      "epoch": 2.3448944091796876e-05,
      "model_forward_time": 1.7495064735412598,
      "step": 98588
    },
    {
      "epoch": 2.3448944091796876e-05,
      "step": 98588,
      "training_step_time": 8.704394340515137
    },
    {
      "epoch": 2.344989776611328e-05,
      "model_forward_time": 1.758573293685913,
      "step": 98589
    },
    {
      "epoch": 2.344989776611328e-05,
      "step": 98589,
      "training_step_time": 8.154085159301758
    },
    {
      "epoch": 2.3450851440429687e-05,
      "grad_norm": 0.07368738204240799,
      "learning_rate": 5.44211572355291e-09,
      "loss": 0.0553,
      "step": 98590
    },
    {
      "epoch": 2.3450851440429687e-05,
      "model_forward_time": 1.7473835945129395,
      "step": 98590
    },
    {
      "dynamics_loss_avg": 0.08304407671093941,
      "epoch": 2.3450851440429687e-05,
      "step": 98590
    },
    {
      "action_loss_avg": 0.0012858685775427147,
      "epoch": 2.3450851440429687e-05,
      "step": 98590
    },
    {
      "epoch": 2.3450851440429687e-05,
      "step": 98590,
      "training_step_time": 8.393695831298828
    },
    {
      "epoch": 2.3451805114746095e-05,
      "model_forward_time": 1.8298966884613037,
      "step": 98591
    },
    {
      "epoch": 2.3451805114746095e-05,
      "step": 98591,
      "training_step_time": 9.640210628509521
    },
    {
      "epoch": 2.34527587890625e-05,
      "model_forward_time": 2.41141676902771,
      "step": 98592
    },
    {
      "epoch": 2.34527587890625e-05,
      "step": 98592,
      "training_step_time": 7.6927478313446045
    },
    {
      "epoch": 2.3453712463378906e-05,
      "model_forward_time": 1.7553815841674805,
      "step": 98593
    },
    {
      "epoch": 2.3453712463378906e-05,
      "step": 98593,
      "training_step_time": 7.620775461196899
    },
    {
      "epoch": 2.3454666137695314e-05,
      "model_forward_time": 1.7590234279632568,
      "step": 98594
    },
    {
      "epoch": 2.3454666137695314e-05,
      "step": 98594,
      "training_step_time": 7.560903310775757
    },
    {
      "epoch": 2.3455619812011718e-05,
      "model_forward_time": 1.7596712112426758,
      "step": 98595
    },
    {
      "epoch": 2.3455619812011718e-05,
      "step": 98595,
      "training_step_time": 7.632478713989258
    },
    {
      "epoch": 2.3456573486328125e-05,
      "model_forward_time": 1.7544739246368408,
      "step": 98596
    },
    {
      "epoch": 2.3456573486328125e-05,
      "step": 98596,
      "training_step_time": 7.676452398300171
    },
    {
      "epoch": 2.3457527160644532e-05,
      "model_forward_time": 1.7635865211486816,
      "step": 98597
    },
    {
      "epoch": 2.3457527160644532e-05,
      "step": 98597,
      "training_step_time": 7.652729511260986
    },
    {
      "epoch": 2.3458480834960937e-05,
      "model_forward_time": 1.7553718090057373,
      "step": 98598
    },
    {
      "epoch": 2.3458480834960937e-05,
      "step": 98598,
      "training_step_time": 8.378174781799316
    },
    {
      "epoch": 2.3459434509277344e-05,
      "model_forward_time": 1.7550508975982666,
      "step": 98599
    },
    {
      "epoch": 2.3459434509277344e-05,
      "step": 98599,
      "training_step_time": 8.256092071533203
    },
    {
      "epoch": 2.346038818359375e-05,
      "grad_norm": 0.10898778587579727,
      "learning_rate": 5.365264397154679e-09,
      "loss": 0.0544,
      "step": 98600
    },
    {
      "epoch": 2.346038818359375e-05,
      "model_forward_time": 1.8162038326263428,
      "step": 98600
    },
    {
      "dynamics_loss_avg": 0.05041095335036516,
      "epoch": 2.346038818359375e-05,
      "step": 98600
    },
    {
      "action_loss_avg": 0.0009560384001815692,
      "epoch": 2.346038818359375e-05,
      "step": 98600
    },
    {
      "epoch": 2.346038818359375e-05,
      "step": 98600,
      "training_step_time": 9.24794864654541
    },
    {
      "epoch": 2.3461341857910155e-05,
      "model_forward_time": 2.4000654220581055,
      "step": 98601
    },
    {
      "epoch": 2.3461341857910155e-05,
      "step": 98601,
      "training_step_time": 8.793903112411499
    },
    {
      "epoch": 2.3462295532226563e-05,
      "model_forward_time": 1.7463629245758057,
      "step": 98602
    },
    {
      "epoch": 2.3462295532226563e-05,
      "step": 98602,
      "training_step_time": 8.490074157714844
    },
    {
      "epoch": 2.346324920654297e-05,
      "model_forward_time": 1.796013355255127,
      "step": 98603
    },
    {
      "epoch": 2.346324920654297e-05,
      "step": 98603,
      "training_step_time": 7.57207179069519
    },
    {
      "epoch": 2.3464202880859374e-05,
      "model_forward_time": 1.7863407135009766,
      "step": 98604
    },
    {
      "epoch": 2.3464202880859374e-05,
      "step": 98604,
      "training_step_time": 7.603990316390991
    },
    {
      "epoch": 2.346515655517578e-05,
      "model_forward_time": 1.755079984664917,
      "step": 98605
    },
    {
      "epoch": 2.346515655517578e-05,
      "step": 98605,
      "training_step_time": 7.679560899734497
    },
    {
      "epoch": 2.346611022949219e-05,
      "model_forward_time": 1.7585515975952148,
      "step": 98606
    },
    {
      "epoch": 2.346611022949219e-05,
      "step": 98606,
      "training_step_time": 7.70911169052124
    },
    {
      "epoch": 2.3467063903808593e-05,
      "model_forward_time": 1.7658729553222656,
      "step": 98607
    },
    {
      "epoch": 2.3467063903808593e-05,
      "step": 98607,
      "training_step_time": 9.59088397026062
    },
    {
      "epoch": 2.3468017578125e-05,
      "model_forward_time": 1.759824514389038,
      "step": 98608
    },
    {
      "epoch": 2.3468017578125e-05,
      "step": 98608,
      "training_step_time": 7.660420656204224
    },
    {
      "epoch": 2.3468971252441408e-05,
      "model_forward_time": 1.3951048851013184,
      "step": 98609
    },
    {
      "epoch": 2.3468971252441408e-05,
      "step": 98609,
      "training_step_time": 10.265576124191284
    },
    {
      "epoch": 2.3469924926757812e-05,
      "grad_norm": 0.07081005722284317,
      "learning_rate": 5.2889592765015265e-09,
      "loss": 0.0553,
      "step": 98610
    },
    {
      "epoch": 2.3469924926757812e-05,
      "model_forward_time": 2.451961040496826,
      "step": 98610
    },
    {
      "dynamics_loss_avg": 0.04805873576551676,
      "epoch": 2.3469924926757812e-05,
      "step": 98610
    },
    {
      "action_loss_avg": 0.0009419465583050624,
      "epoch": 2.3469924926757812e-05,
      "step": 98610
    },
    {
      "epoch": 2.3469924926757812e-05,
      "step": 98610,
      "training_step_time": 9.843467950820923
    },
    {
      "epoch": 2.347087860107422e-05,
      "model_forward_time": 1.748305320739746,
      "step": 98611
    },
    {
      "epoch": 2.347087860107422e-05,
      "step": 98611,
      "training_step_time": 8.404486179351807
    },
    {
      "epoch": 2.3471832275390623e-05,
      "model_forward_time": 1.7565903663635254,
      "step": 98612
    },
    {
      "epoch": 2.3471832275390623e-05,
      "step": 98612,
      "training_step_time": 8.074812412261963
    },
    {
      "epoch": 2.347278594970703e-05,
      "model_forward_time": 1.764906883239746,
      "step": 98613
    },
    {
      "epoch": 2.347278594970703e-05,
      "step": 98613,
      "training_step_time": 7.607424020767212
    },
    {
      "epoch": 2.3473739624023438e-05,
      "model_forward_time": 1.7494006156921387,
      "step": 98614
    },
    {
      "epoch": 2.3473739624023438e-05,
      "step": 98614,
      "training_step_time": 7.672607898712158
    },
    {
      "epoch": 2.3474693298339842e-05,
      "model_forward_time": 1.7573027610778809,
      "step": 98615
    },
    {
      "epoch": 2.3474693298339842e-05,
      "step": 98615,
      "training_step_time": 7.663279056549072
    },
    {
      "epoch": 2.347564697265625e-05,
      "model_forward_time": 1.7602119445800781,
      "step": 98616
    },
    {
      "epoch": 2.347564697265625e-05,
      "step": 98616,
      "training_step_time": 8.275283813476562
    },
    {
      "epoch": 2.3476600646972657e-05,
      "model_forward_time": 1.7488598823547363,
      "step": 98617
    },
    {
      "epoch": 2.3476600646972657e-05,
      "step": 98617,
      "training_step_time": 8.305965662002563
    },
    {
      "epoch": 2.347755432128906e-05,
      "model_forward_time": 1.7558369636535645,
      "step": 98618
    },
    {
      "epoch": 2.347755432128906e-05,
      "step": 98618,
      "training_step_time": 8.654057025909424
    },
    {
      "epoch": 2.347850799560547e-05,
      "model_forward_time": 2.4308149814605713,
      "step": 98619
    },
    {
      "epoch": 2.347850799560547e-05,
      "step": 98619,
      "training_step_time": 8.951541423797607
    },
    {
      "epoch": 2.3479461669921876e-05,
      "grad_norm": 0.07625972479581833,
      "learning_rate": 5.213200369937888e-09,
      "loss": 0.0558,
      "step": 98620
    },
    {
      "epoch": 2.3479461669921876e-05,
      "model_forward_time": 1.7419707775115967,
      "step": 98620
    },
    {
      "dynamics_loss_avg": 0.04555772878229618,
      "epoch": 2.3479461669921876e-05,
      "step": 98620
    },
    {
      "action_loss_avg": 0.0010413711832370609,
      "epoch": 2.3479461669921876e-05,
      "step": 98620
    },
    {
      "epoch": 2.3479461669921876e-05,
      "step": 98620,
      "training_step_time": 8.648596048355103
    },
    {
      "epoch": 2.348041534423828e-05,
      "model_forward_time": 1.760533094406128,
      "step": 98621
    },
    {
      "epoch": 2.348041534423828e-05,
      "step": 98621,
      "training_step_time": 7.52726936340332
    },
    {
      "epoch": 2.3481369018554687e-05,
      "model_forward_time": 1.751706600189209,
      "step": 98622
    },
    {
      "epoch": 2.3481369018554687e-05,
      "step": 98622,
      "training_step_time": 7.629286527633667
    },
    {
      "epoch": 2.3482322692871095e-05,
      "model_forward_time": 1.7556791305541992,
      "step": 98623
    },
    {
      "epoch": 2.3482322692871095e-05,
      "step": 98623,
      "training_step_time": 7.6541523933410645
    },
    {
      "epoch": 2.34832763671875e-05,
      "model_forward_time": 1.7530887126922607,
      "step": 98624
    },
    {
      "epoch": 2.34832763671875e-05,
      "step": 98624,
      "training_step_time": 7.684368133544922
    },
    {
      "epoch": 2.3484230041503906e-05,
      "model_forward_time": 1.7603929042816162,
      "step": 98625
    },
    {
      "epoch": 2.3484230041503906e-05,
      "step": 98625,
      "training_step_time": 8.531616449356079
    },
    {
      "epoch": 2.3485183715820314e-05,
      "model_forward_time": 1.7562789916992188,
      "step": 98626
    },
    {
      "epoch": 2.3485183715820314e-05,
      "step": 98626,
      "training_step_time": 8.615107536315918
    },
    {
      "epoch": 2.3486137390136718e-05,
      "model_forward_time": 1.7572553157806396,
      "step": 98627
    },
    {
      "epoch": 2.3486137390136718e-05,
      "step": 98627,
      "training_step_time": 9.575465679168701
    },
    {
      "epoch": 2.3487091064453125e-05,
      "model_forward_time": 1.8194375038146973,
      "step": 98628
    },
    {
      "epoch": 2.3487091064453125e-05,
      "step": 98628,
      "training_step_time": 9.29590892791748
    },
    {
      "epoch": 2.3488044738769533e-05,
      "model_forward_time": 1.7370643615722656,
      "step": 98629
    },
    {
      "epoch": 2.3488044738769533e-05,
      "step": 98629,
      "training_step_time": 7.7159340381622314
    },
    {
      "epoch": 2.3488998413085937e-05,
      "grad_norm": 0.08482712507247925,
      "learning_rate": 5.137987685749912e-09,
      "loss": 0.0552,
      "step": 98630
    },
    {
      "epoch": 2.3488998413085937e-05,
      "model_forward_time": 1.7502493858337402,
      "step": 98630
    },
    {
      "dynamics_loss_avg": 0.058702051639556885,
      "epoch": 2.3488998413085937e-05,
      "step": 98630
    },
    {
      "action_loss_avg": 0.0011838750913739204,
      "epoch": 2.3488998413085937e-05,
      "step": 98630
    },
    {
      "epoch": 2.3488998413085937e-05,
      "step": 98630,
      "training_step_time": 7.590642213821411
    },
    {
      "epoch": 2.3489952087402344e-05,
      "model_forward_time": 1.7486097812652588,
      "step": 98631
    },
    {
      "epoch": 2.3489952087402344e-05,
      "step": 98631,
      "training_step_time": 8.106935024261475
    },
    {
      "epoch": 2.349090576171875e-05,
      "model_forward_time": 1.7595231533050537,
      "step": 98632
    },
    {
      "epoch": 2.349090576171875e-05,
      "step": 98632,
      "training_step_time": 7.569028377532959
    },
    {
      "epoch": 2.3491859436035155e-05,
      "model_forward_time": 1.7545812129974365,
      "step": 98633
    },
    {
      "epoch": 2.3491859436035155e-05,
      "step": 98633,
      "training_step_time": 7.609936714172363
    },
    {
      "epoch": 2.3492813110351563e-05,
      "model_forward_time": 1.7530252933502197,
      "step": 98634
    },
    {
      "epoch": 2.3492813110351563e-05,
      "step": 98634,
      "training_step_time": 8.340365648269653
    },
    {
      "epoch": 2.349376678466797e-05,
      "model_forward_time": 1.7524714469909668,
      "step": 98635
    },
    {
      "epoch": 2.349376678466797e-05,
      "step": 98635,
      "training_step_time": 8.461372375488281
    },
    {
      "epoch": 2.3494720458984374e-05,
      "model_forward_time": 1.7449193000793457,
      "step": 98636
    },
    {
      "epoch": 2.3494720458984374e-05,
      "step": 98636,
      "training_step_time": 7.70782732963562
    },
    {
      "epoch": 2.349567413330078e-05,
      "model_forward_time": 1.8171415328979492,
      "step": 98637
    },
    {
      "epoch": 2.349567413330078e-05,
      "step": 98637,
      "training_step_time": 8.42404580116272
    },
    {
      "epoch": 2.349662780761719e-05,
      "model_forward_time": 2.433318614959717,
      "step": 98638
    },
    {
      "epoch": 2.349662780761719e-05,
      "step": 98638,
      "training_step_time": 8.740827083587646
    },
    {
      "epoch": 2.3497581481933593e-05,
      "model_forward_time": 1.7491588592529297,
      "step": 98639
    },
    {
      "epoch": 2.3497581481933593e-05,
      "step": 98639,
      "training_step_time": 9.196231842041016
    },
    {
      "epoch": 2.349853515625e-05,
      "grad_norm": 0.06698031723499298,
      "learning_rate": 5.063321232161022e-09,
      "loss": 0.0555,
      "step": 98640
    },
    {
      "epoch": 2.349853515625e-05,
      "model_forward_time": 1.749502420425415,
      "step": 98640
    },
    {
      "dynamics_loss_avg": 0.062130107544362545,
      "epoch": 2.349853515625e-05,
      "step": 98640
    },
    {
      "action_loss_avg": 0.001215853967005387,
      "epoch": 2.349853515625e-05,
      "step": 98640
    },
    {
      "epoch": 2.349853515625e-05,
      "step": 98640,
      "training_step_time": 7.187030553817749
    },
    {
      "epoch": 2.3499488830566405e-05,
      "model_forward_time": 1.7551181316375732,
      "step": 98641
    },
    {
      "epoch": 2.3499488830566405e-05,
      "step": 98641,
      "training_step_time": 7.6000142097473145
    },
    {
      "epoch": 2.3500442504882812e-05,
      "model_forward_time": 1.7476348876953125,
      "step": 98642
    },
    {
      "epoch": 2.3500442504882812e-05,
      "step": 98642,
      "training_step_time": 7.686422109603882
    },
    {
      "epoch": 2.350139617919922e-05,
      "model_forward_time": 1.7542364597320557,
      "step": 98643
    },
    {
      "epoch": 2.350139617919922e-05,
      "step": 98643,
      "training_step_time": 7.707014083862305
    },
    {
      "epoch": 2.3502349853515623e-05,
      "model_forward_time": 1.7588272094726562,
      "step": 98644
    },
    {
      "epoch": 2.3502349853515623e-05,
      "step": 98644,
      "training_step_time": 8.852785587310791
    },
    {
      "epoch": 2.350330352783203e-05,
      "model_forward_time": 1.7491767406463623,
      "step": 98645
    },
    {
      "epoch": 2.350330352783203e-05,
      "step": 98645,
      "training_step_time": 11.298364877700806
    },
    {
      "epoch": 2.3504257202148438e-05,
      "model_forward_time": 1.8218019008636475,
      "step": 98646
    },
    {
      "epoch": 2.3504257202148438e-05,
      "step": 98646,
      "training_step_time": 8.633236408233643
    },
    {
      "epoch": 2.3505210876464842e-05,
      "model_forward_time": 2.4115095138549805,
      "step": 98647
    },
    {
      "epoch": 2.3505210876464842e-05,
      "step": 98647,
      "training_step_time": 9.847078800201416
    },
    {
      "epoch": 2.350616455078125e-05,
      "model_forward_time": 1.7525556087493896,
      "step": 98648
    },
    {
      "epoch": 2.350616455078125e-05,
      "step": 98648,
      "training_step_time": 7.70662784576416
    },
    {
      "epoch": 2.3507118225097657e-05,
      "model_forward_time": 1.7473704814910889,
      "step": 98649
    },
    {
      "epoch": 2.3507118225097657e-05,
      "step": 98649,
      "training_step_time": 7.484025955200195
    },
    {
      "epoch": 2.350807189941406e-05,
      "grad_norm": 0.08083430677652359,
      "learning_rate": 4.989201017338019e-09,
      "loss": 0.0544,
      "step": 98650
    },
    {
      "epoch": 2.350807189941406e-05,
      "model_forward_time": 1.7534167766571045,
      "step": 98650
    },
    {
      "dynamics_loss_avg": 0.05471680834889412,
      "epoch": 2.350807189941406e-05,
      "step": 98650
    },
    {
      "action_loss_avg": 0.0010204537800746038,
      "epoch": 2.350807189941406e-05,
      "step": 98650
    },
    {
      "epoch": 2.350807189941406e-05,
      "step": 98650,
      "training_step_time": 7.641443729400635
    },
    {
      "epoch": 2.350902557373047e-05,
      "model_forward_time": 1.7984728813171387,
      "step": 98651
    },
    {
      "epoch": 2.350902557373047e-05,
      "step": 98651,
      "training_step_time": 7.7653186321258545
    },
    {
      "epoch": 2.3509979248046876e-05,
      "model_forward_time": 1.7453808784484863,
      "step": 98652
    },
    {
      "epoch": 2.3509979248046876e-05,
      "step": 98652,
      "training_step_time": 7.786981582641602
    },
    {
      "epoch": 2.351093292236328e-05,
      "model_forward_time": 1.7557272911071777,
      "step": 98653
    },
    {
      "epoch": 2.351093292236328e-05,
      "step": 98653,
      "training_step_time": 8.403905153274536
    },
    {
      "epoch": 2.3511886596679687e-05,
      "model_forward_time": 1.74031400680542,
      "step": 98654
    },
    {
      "epoch": 2.3511886596679687e-05,
      "step": 98654,
      "training_step_time": 9.476319313049316
    },
    {
      "epoch": 2.3512840270996095e-05,
      "model_forward_time": 1.820802927017212,
      "step": 98655
    },
    {
      "epoch": 2.3512840270996095e-05,
      "step": 98655,
      "training_step_time": 9.253939390182495
    },
    {
      "epoch": 2.35137939453125e-05,
      "model_forward_time": 2.4148290157318115,
      "step": 98656
    },
    {
      "epoch": 2.35137939453125e-05,
      "step": 98656,
      "training_step_time": 7.755851745605469
    },
    {
      "epoch": 2.3514747619628906e-05,
      "model_forward_time": 1.7577307224273682,
      "step": 98657
    },
    {
      "epoch": 2.3514747619628906e-05,
      "step": 98657,
      "training_step_time": 7.763095140457153
    },
    {
      "epoch": 2.3515701293945314e-05,
      "model_forward_time": 1.7522435188293457,
      "step": 98658
    },
    {
      "epoch": 2.3515701293945314e-05,
      "step": 98658,
      "training_step_time": 7.525589466094971
    },
    {
      "epoch": 2.3516654968261718e-05,
      "model_forward_time": 1.7616477012634277,
      "step": 98659
    },
    {
      "epoch": 2.3516654968261718e-05,
      "step": 98659,
      "training_step_time": 7.644932270050049
    },
    {
      "epoch": 2.3517608642578125e-05,
      "grad_norm": 0.4941082298755646,
      "learning_rate": 4.915627049386085e-09,
      "loss": 0.0552,
      "step": 98660
    },
    {
      "epoch": 2.3517608642578125e-05,
      "model_forward_time": 1.7459149360656738,
      "step": 98660
    },
    {
      "dynamics_loss_avg": 0.04273215718567371,
      "epoch": 2.3517608642578125e-05,
      "step": 98660
    },
    {
      "action_loss_avg": 0.0019119955541100353,
      "epoch": 2.3517608642578125e-05,
      "step": 98660
    },
    {
      "epoch": 2.3517608642578125e-05,
      "step": 98660,
      "training_step_time": 8.619297742843628
    },
    {
      "epoch": 2.3518562316894533e-05,
      "model_forward_time": 1.7505879402160645,
      "step": 98661
    },
    {
      "epoch": 2.3518562316894533e-05,
      "step": 98661,
      "training_step_time": 7.6887366771698
    },
    {
      "epoch": 2.3519515991210937e-05,
      "model_forward_time": 1.7540614604949951,
      "step": 98662
    },
    {
      "epoch": 2.3519515991210937e-05,
      "step": 98662,
      "training_step_time": 7.328552484512329
    },
    {
      "epoch": 2.3520469665527344e-05,
      "model_forward_time": 1.7535138130187988,
      "step": 98663
    },
    {
      "epoch": 2.3520469665527344e-05,
      "step": 98663,
      "training_step_time": 8.782557964324951
    },
    {
      "epoch": 2.352142333984375e-05,
      "model_forward_time": 1.8197252750396729,
      "step": 98664
    },
    {
      "epoch": 2.352142333984375e-05,
      "step": 98664,
      "training_step_time": 9.985694646835327
    },
    {
      "epoch": 2.3522377014160155e-05,
      "model_forward_time": 2.444617748260498,
      "step": 98665
    },
    {
      "epoch": 2.3522377014160155e-05,
      "step": 98665,
      "training_step_time": 8.788339614868164
    },
    {
      "epoch": 2.3523330688476563e-05,
      "model_forward_time": 1.7433969974517822,
      "step": 98666
    },
    {
      "epoch": 2.3523330688476563e-05,
      "step": 98666,
      "training_step_time": 7.747455835342407
    },
    {
      "epoch": 2.352428436279297e-05,
      "model_forward_time": 1.7791557312011719,
      "step": 98667
    },
    {
      "epoch": 2.352428436279297e-05,
      "step": 98667,
      "training_step_time": 7.405196666717529
    },
    {
      "epoch": 2.3525238037109374e-05,
      "model_forward_time": 1.7405712604522705,
      "step": 98668
    },
    {
      "epoch": 2.3525238037109374e-05,
      "step": 98668,
      "training_step_time": 7.627298355102539
    },
    {
      "epoch": 2.3526191711425782e-05,
      "model_forward_time": 1.7502357959747314,
      "step": 98669
    },
    {
      "epoch": 2.3526191711425782e-05,
      "step": 98669,
      "training_step_time": 8.745701313018799
    },
    {
      "epoch": 2.352714538574219e-05,
      "grad_norm": 0.08688530325889587,
      "learning_rate": 4.842599336351561e-09,
      "loss": 0.0556,
      "step": 98670
    },
    {
      "epoch": 2.352714538574219e-05,
      "model_forward_time": 1.7444274425506592,
      "step": 98670
    },
    {
      "dynamics_loss_avg": 0.046178702637553214,
      "epoch": 2.352714538574219e-05,
      "step": 98670
    },
    {
      "action_loss_avg": 0.000798068754374981,
      "epoch": 2.352714538574219e-05,
      "step": 98670
    },
    {
      "epoch": 2.352714538574219e-05,
      "step": 98670,
      "training_step_time": 7.67100191116333
    },
    {
      "epoch": 2.3528099060058593e-05,
      "model_forward_time": 1.73679780960083,
      "step": 98671
    },
    {
      "epoch": 2.3528099060058593e-05,
      "step": 98671,
      "training_step_time": 7.459245204925537
    },
    {
      "epoch": 2.3529052734375e-05,
      "model_forward_time": 1.7480683326721191,
      "step": 98672
    },
    {
      "epoch": 2.3529052734375e-05,
      "step": 98672,
      "training_step_time": 8.878407716751099
    },
    {
      "epoch": 2.3530006408691405e-05,
      "model_forward_time": 1.8243908882141113,
      "step": 98673
    },
    {
      "epoch": 2.3530006408691405e-05,
      "step": 98673,
      "training_step_time": 9.943281650543213
    },
    {
      "epoch": 2.3530960083007812e-05,
      "model_forward_time": 2.4048404693603516,
      "step": 98674
    },
    {
      "epoch": 2.3530960083007812e-05,
      "step": 98674,
      "training_step_time": 7.59917950630188
    },
    {
      "epoch": 2.353191375732422e-05,
      "model_forward_time": 1.7602717876434326,
      "step": 98675
    },
    {
      "epoch": 2.353191375732422e-05,
      "step": 98675,
      "training_step_time": 8.483644008636475
    },
    {
      "epoch": 2.3532867431640623e-05,
      "model_forward_time": 1.7557976245880127,
      "step": 98676
    },
    {
      "epoch": 2.3532867431640623e-05,
      "step": 98676,
      "training_step_time": 7.639127492904663
    },
    {
      "epoch": 2.353382110595703e-05,
      "model_forward_time": 1.7521440982818604,
      "step": 98677
    },
    {
      "epoch": 2.353382110595703e-05,
      "step": 98677,
      "training_step_time": 7.659950017929077
    },
    {
      "epoch": 2.3534774780273438e-05,
      "model_forward_time": 1.760232925415039,
      "step": 98678
    },
    {
      "epoch": 2.3534774780273438e-05,
      "step": 98678,
      "training_step_time": 7.667864561080933
    },
    {
      "epoch": 2.3535728454589842e-05,
      "model_forward_time": 1.7512049674987793,
      "step": 98679
    },
    {
      "epoch": 2.3535728454589842e-05,
      "step": 98679,
      "training_step_time": 8.701000690460205
    },
    {
      "epoch": 2.353668212890625e-05,
      "grad_norm": 0.15401247143745422,
      "learning_rate": 4.770117886219727e-09,
      "loss": 0.0554,
      "step": 98680
    },
    {
      "epoch": 2.353668212890625e-05,
      "model_forward_time": 1.7442681789398193,
      "step": 98680
    },
    {
      "dynamics_loss_avg": 0.044972182158380745,
      "epoch": 2.353668212890625e-05,
      "step": 98680
    },
    {
      "action_loss_avg": 0.0011700800299877302,
      "epoch": 2.353668212890625e-05,
      "step": 98680
    },
    {
      "epoch": 2.353668212890625e-05,
      "step": 98680,
      "training_step_time": 7.648642301559448
    },
    {
      "epoch": 2.3537635803222657e-05,
      "model_forward_time": 1.7504825592041016,
      "step": 98681
    },
    {
      "epoch": 2.3537635803222657e-05,
      "step": 98681,
      "training_step_time": 8.629141330718994
    },
    {
      "epoch": 2.353858947753906e-05,
      "model_forward_time": 1.760312795639038,
      "step": 98682
    },
    {
      "epoch": 2.353858947753906e-05,
      "step": 98682,
      "training_step_time": 8.698903560638428
    },
    {
      "epoch": 2.353954315185547e-05,
      "model_forward_time": 1.757728099822998,
      "step": 98683
    },
    {
      "epoch": 2.353954315185547e-05,
      "step": 98683,
      "training_step_time": 9.757668733596802
    },
    {
      "epoch": 2.3540496826171876e-05,
      "model_forward_time": 1.750420331954956,
      "step": 98684
    },
    {
      "epoch": 2.3540496826171876e-05,
      "step": 98684,
      "training_step_time": 7.812840938568115
    },
    {
      "epoch": 2.354145050048828e-05,
      "model_forward_time": 1.7298004627227783,
      "step": 98685
    },
    {
      "epoch": 2.354145050048828e-05,
      "step": 98685,
      "training_step_time": 7.51355767250061
    },
    {
      "epoch": 2.3542404174804687e-05,
      "model_forward_time": 1.7551147937774658,
      "step": 98686
    },
    {
      "epoch": 2.3542404174804687e-05,
      "step": 98686,
      "training_step_time": 7.703676462173462
    },
    {
      "epoch": 2.3543357849121095e-05,
      "model_forward_time": 1.3757390975952148,
      "step": 98687
    },
    {
      "epoch": 2.3543357849121095e-05,
      "step": 98687,
      "training_step_time": 7.763880729675293
    },
    {
      "epoch": 2.35443115234375e-05,
      "model_forward_time": 1.750758409500122,
      "step": 98688
    },
    {
      "epoch": 2.35443115234375e-05,
      "step": 98688,
      "training_step_time": 7.826006174087524
    },
    {
      "epoch": 2.3545265197753906e-05,
      "model_forward_time": 1.7535083293914795,
      "step": 98689
    },
    {
      "epoch": 2.3545265197753906e-05,
      "step": 98689,
      "training_step_time": 7.874927520751953
    },
    {
      "epoch": 2.3546218872070314e-05,
      "grad_norm": 0.09972542524337769,
      "learning_rate": 4.6981827069181305e-09,
      "loss": 0.0551,
      "step": 98690
    },
    {
      "epoch": 2.3546218872070314e-05,
      "model_forward_time": 1.7380914688110352,
      "step": 98690
    },
    {
      "dynamics_loss_avg": 0.052606400847434995,
      "epoch": 2.3546218872070314e-05,
      "step": 98690
    },
    {
      "action_loss_avg": 0.0009030623390572145,
      "epoch": 2.3546218872070314e-05,
      "step": 98690
    },
    {
      "epoch": 2.3546218872070314e-05,
      "step": 98690,
      "training_step_time": 7.842314958572388
    },
    {
      "epoch": 2.3547172546386718e-05,
      "model_forward_time": 1.7477006912231445,
      "step": 98691
    },
    {
      "epoch": 2.3547172546386718e-05,
      "step": 98691,
      "training_step_time": 8.952317237854004
    },
    {
      "epoch": 2.3548126220703125e-05,
      "model_forward_time": 1.8174586296081543,
      "step": 98692
    },
    {
      "epoch": 2.3548126220703125e-05,
      "step": 98692,
      "training_step_time": 8.704241752624512
    },
    {
      "epoch": 2.3549079895019533e-05,
      "model_forward_time": 2.3919379711151123,
      "step": 98693
    },
    {
      "epoch": 2.3549079895019533e-05,
      "step": 98693,
      "training_step_time": 8.431199073791504
    },
    {
      "epoch": 2.3550033569335937e-05,
      "model_forward_time": 1.7379300594329834,
      "step": 98694
    },
    {
      "epoch": 2.3550033569335937e-05,
      "step": 98694,
      "training_step_time": 7.845922470092773
    },
    {
      "epoch": 2.3550987243652344e-05,
      "model_forward_time": 1.7486286163330078,
      "step": 98695
    },
    {
      "epoch": 2.3550987243652344e-05,
      "step": 98695,
      "training_step_time": 7.7949113845825195
    },
    {
      "epoch": 2.355194091796875e-05,
      "model_forward_time": 1.7447478771209717,
      "step": 98696
    },
    {
      "epoch": 2.355194091796875e-05,
      "step": 98696,
      "training_step_time": 7.744266986846924
    },
    {
      "epoch": 2.3552894592285155e-05,
      "model_forward_time": 1.7489871978759766,
      "step": 98697
    },
    {
      "epoch": 2.3552894592285155e-05,
      "step": 98697,
      "training_step_time": 7.929286003112793
    },
    {
      "epoch": 2.3553848266601563e-05,
      "model_forward_time": 1.7446935176849365,
      "step": 98698
    },
    {
      "epoch": 2.3553848266601563e-05,
      "step": 98698,
      "training_step_time": 8.719625234603882
    },
    {
      "epoch": 2.355480194091797e-05,
      "model_forward_time": 1.7487854957580566,
      "step": 98699
    },
    {
      "epoch": 2.355480194091797e-05,
      "step": 98699,
      "training_step_time": 7.831008434295654
    },
    {
      "epoch": 2.3555755615234374e-05,
      "grad_norm": 0.08509621024131775,
      "learning_rate": 4.626793806312147e-09,
      "loss": 0.0552,
      "step": 98700
    },
    {
      "epoch": 2.3555755615234374e-05,
      "model_forward_time": 1.7402327060699463,
      "step": 98700
    },
    {
      "dynamics_loss_avg": 0.055606746952980755,
      "epoch": 2.3555755615234374e-05,
      "step": 98700
    },
    {
      "action_loss_avg": 0.0010097972641233354,
      "epoch": 2.3555755615234374e-05,
      "step": 98700
    },
    {
      "epoch": 2.3555755615234374e-05,
      "step": 98700,
      "training_step_time": 7.875415563583374
    },
    {
      "epoch": 2.3556709289550782e-05,
      "model_forward_time": 1.8157873153686523,
      "step": 98701
    },
    {
      "epoch": 2.3556709289550782e-05,
      "step": 98701,
      "training_step_time": 9.200019836425781
    },
    {
      "epoch": 2.355766296386719e-05,
      "model_forward_time": 2.3986666202545166,
      "step": 98702
    },
    {
      "epoch": 2.355766296386719e-05,
      "step": 98702,
      "training_step_time": 9.688411235809326
    },
    {
      "epoch": 2.3558616638183593e-05,
      "model_forward_time": 1.7504923343658447,
      "step": 98703
    },
    {
      "epoch": 2.3558616638183593e-05,
      "step": 98703,
      "training_step_time": 7.884754419326782
    },
    {
      "epoch": 2.35595703125e-05,
      "model_forward_time": 1.7291176319122314,
      "step": 98704
    },
    {
      "epoch": 2.35595703125e-05,
      "step": 98704,
      "training_step_time": 7.793944597244263
    },
    {
      "epoch": 2.3560523986816405e-05,
      "model_forward_time": 1.759977102279663,
      "step": 98705
    },
    {
      "epoch": 2.3560523986816405e-05,
      "step": 98705,
      "training_step_time": 7.690333366394043
    },
    {
      "epoch": 2.3561477661132812e-05,
      "model_forward_time": 1.7479982376098633,
      "step": 98706
    },
    {
      "epoch": 2.3561477661132812e-05,
      "step": 98706,
      "training_step_time": 8.824187517166138
    },
    {
      "epoch": 2.356243133544922e-05,
      "model_forward_time": 1.7448034286499023,
      "step": 98707
    },
    {
      "epoch": 2.356243133544922e-05,
      "step": 98707,
      "training_step_time": 7.525597810745239
    },
    {
      "epoch": 2.3563385009765623e-05,
      "model_forward_time": 1.7541308403015137,
      "step": 98708
    },
    {
      "epoch": 2.3563385009765623e-05,
      "step": 98708,
      "training_step_time": 8.372255086898804
    },
    {
      "epoch": 2.356433868408203e-05,
      "model_forward_time": 1.7448599338531494,
      "step": 98709
    },
    {
      "epoch": 2.356433868408203e-05,
      "step": 98709,
      "training_step_time": 8.561914205551147
    },
    {
      "epoch": 2.356529235839844e-05,
      "grad_norm": 0.07770980894565582,
      "learning_rate": 4.555951192209973e-09,
      "loss": 0.0556,
      "step": 98710
    },
    {
      "epoch": 2.356529235839844e-05,
      "model_forward_time": 1.8173787593841553,
      "step": 98710
    },
    {
      "dynamics_loss_avg": 0.055986527167260644,
      "epoch": 2.356529235839844e-05,
      "step": 98710
    },
    {
      "action_loss_avg": 0.0012689110735664144,
      "epoch": 2.356529235839844e-05,
      "step": 98710
    },
    {
      "epoch": 2.356529235839844e-05,
      "step": 98710,
      "training_step_time": 8.573557376861572
    },
    {
      "epoch": 2.3566246032714842e-05,
      "model_forward_time": 2.395440101623535,
      "step": 98711
    },
    {
      "epoch": 2.3566246032714842e-05,
      "step": 98711,
      "training_step_time": 8.406376600265503
    },
    {
      "epoch": 2.356719970703125e-05,
      "model_forward_time": 1.7435986995697021,
      "step": 98712
    },
    {
      "epoch": 2.356719970703125e-05,
      "step": 98712,
      "training_step_time": 8.466864347457886
    },
    {
      "epoch": 2.3568153381347657e-05,
      "model_forward_time": 1.7386786937713623,
      "step": 98713
    },
    {
      "epoch": 2.3568153381347657e-05,
      "step": 98713,
      "training_step_time": 7.835548162460327
    },
    {
      "epoch": 2.356910705566406e-05,
      "model_forward_time": 1.744173288345337,
      "step": 98714
    },
    {
      "epoch": 2.356910705566406e-05,
      "step": 98714,
      "training_step_time": 7.735116243362427
    },
    {
      "epoch": 2.357006072998047e-05,
      "model_forward_time": 1.7435431480407715,
      "step": 98715
    },
    {
      "epoch": 2.357006072998047e-05,
      "step": 98715,
      "training_step_time": 7.770110607147217
    },
    {
      "epoch": 2.3571014404296876e-05,
      "model_forward_time": 1.7629060745239258,
      "step": 98716
    },
    {
      "epoch": 2.3571014404296876e-05,
      "step": 98716,
      "training_step_time": 8.509921789169312
    },
    {
      "epoch": 2.357196807861328e-05,
      "model_forward_time": 1.7505550384521484,
      "step": 98717
    },
    {
      "epoch": 2.357196807861328e-05,
      "step": 98717,
      "training_step_time": 10.400025367736816
    },
    {
      "epoch": 2.3572921752929687e-05,
      "model_forward_time": 1.7484910488128662,
      "step": 98718
    },
    {
      "epoch": 2.3572921752929687e-05,
      "step": 98718,
      "training_step_time": 7.793307304382324
    },
    {
      "epoch": 2.3573875427246095e-05,
      "model_forward_time": 1.7917110919952393,
      "step": 98719
    },
    {
      "epoch": 2.3573875427246095e-05,
      "step": 98719,
      "training_step_time": 9.421552658081055
    },
    {
      "epoch": 2.35748291015625e-05,
      "grad_norm": 0.07349992543458939,
      "learning_rate": 4.485654872358747e-09,
      "loss": 0.0559,
      "step": 98720
    },
    {
      "epoch": 2.35748291015625e-05,
      "model_forward_time": 2.4104676246643066,
      "step": 98720
    },
    {
      "dynamics_loss_avg": 0.050294460542500016,
      "epoch": 2.35748291015625e-05,
      "step": 98720
    },
    {
      "action_loss_avg": 0.000957303773611784,
      "epoch": 2.35748291015625e-05,
      "step": 98720
    },
    {
      "epoch": 2.35748291015625e-05,
      "step": 98720,
      "training_step_time": 7.67124342918396
    },
    {
      "epoch": 2.3575782775878906e-05,
      "model_forward_time": 1.7428624629974365,
      "step": 98721
    },
    {
      "epoch": 2.3575782775878906e-05,
      "step": 98721,
      "training_step_time": 8.586410522460938
    },
    {
      "epoch": 2.3576736450195314e-05,
      "model_forward_time": 1.739009141921997,
      "step": 98722
    },
    {
      "epoch": 2.3576736450195314e-05,
      "step": 98722,
      "training_step_time": 8.437018632888794
    },
    {
      "epoch": 2.3577690124511718e-05,
      "model_forward_time": 1.7579143047332764,
      "step": 98723
    },
    {
      "epoch": 2.3577690124511718e-05,
      "step": 98723,
      "training_step_time": 8.377009153366089
    },
    {
      "epoch": 2.3578643798828125e-05,
      "model_forward_time": 1.7379379272460938,
      "step": 98724
    },
    {
      "epoch": 2.3578643798828125e-05,
      "step": 98724,
      "training_step_time": 7.71761155128479
    },
    {
      "epoch": 2.3579597473144533e-05,
      "model_forward_time": 1.744736671447754,
      "step": 98725
    },
    {
      "epoch": 2.3579597473144533e-05,
      "step": 98725,
      "training_step_time": 7.681567430496216
    },
    {
      "epoch": 2.3580551147460937e-05,
      "model_forward_time": 1.7851102352142334,
      "step": 98726
    },
    {
      "epoch": 2.3580551147460937e-05,
      "step": 98726,
      "training_step_time": 7.750714302062988
    },
    {
      "epoch": 2.3581504821777344e-05,
      "model_forward_time": 1.74332857131958,
      "step": 98727
    },
    {
      "epoch": 2.3581504821777344e-05,
      "step": 98727,
      "training_step_time": 9.723205804824829
    },
    {
      "epoch": 2.358245849609375e-05,
      "model_forward_time": 1.7422854900360107,
      "step": 98728
    },
    {
      "epoch": 2.358245849609375e-05,
      "step": 98728,
      "training_step_time": 8.885411024093628
    },
    {
      "epoch": 2.3583412170410155e-05,
      "model_forward_time": 2.4154932498931885,
      "step": 98729
    },
    {
      "epoch": 2.3583412170410155e-05,
      "step": 98729,
      "training_step_time": 7.840895175933838
    },
    {
      "epoch": 2.3584365844726563e-05,
      "grad_norm": 0.08665114641189575,
      "learning_rate": 4.4159048544450965e-09,
      "loss": 0.0554,
      "step": 98730
    },
    {
      "epoch": 2.3584365844726563e-05,
      "model_forward_time": 1.7483258247375488,
      "step": 98730
    },
    {
      "dynamics_loss_avg": 0.04827347034588456,
      "epoch": 2.3584365844726563e-05,
      "step": 98730
    },
    {
      "action_loss_avg": 0.0012210947024868801,
      "epoch": 2.3584365844726563e-05,
      "step": 98730
    },
    {
      "epoch": 2.3584365844726563e-05,
      "step": 98730,
      "training_step_time": 7.7920081615448
    },
    {
      "epoch": 2.358531951904297e-05,
      "model_forward_time": 1.7551696300506592,
      "step": 98731
    },
    {
      "epoch": 2.358531951904297e-05,
      "step": 98731,
      "training_step_time": 7.761164665222168
    },
    {
      "epoch": 2.3586273193359374e-05,
      "model_forward_time": 1.7473671436309814,
      "step": 98732
    },
    {
      "epoch": 2.3586273193359374e-05,
      "step": 98732,
      "training_step_time": 7.377319812774658
    },
    {
      "epoch": 2.3587226867675782e-05,
      "model_forward_time": 1.7534542083740234,
      "step": 98733
    },
    {
      "epoch": 2.3587226867675782e-05,
      "step": 98733,
      "training_step_time": 8.712898969650269
    },
    {
      "epoch": 2.358818054199219e-05,
      "model_forward_time": 1.7623364925384521,
      "step": 98734
    },
    {
      "epoch": 2.358818054199219e-05,
      "step": 98734,
      "training_step_time": 7.623282432556152
    },
    {
      "epoch": 2.3589134216308593e-05,
      "model_forward_time": 1.7448375225067139,
      "step": 98735
    },
    {
      "epoch": 2.3589134216308593e-05,
      "step": 98735,
      "training_step_time": 7.681290626525879
    },
    {
      "epoch": 2.3590087890625e-05,
      "model_forward_time": 1.7510035037994385,
      "step": 98736
    },
    {
      "epoch": 2.3590087890625e-05,
      "step": 98736,
      "training_step_time": 8.285766363143921
    },
    {
      "epoch": 2.3591041564941405e-05,
      "model_forward_time": 1.7529456615447998,
      "step": 98737
    },
    {
      "epoch": 2.3591041564941405e-05,
      "step": 98737,
      "training_step_time": 8.486873626708984
    },
    {
      "epoch": 2.3591995239257812e-05,
      "model_forward_time": 2.4425878524780273,
      "step": 98738
    },
    {
      "epoch": 2.3591995239257812e-05,
      "step": 98738,
      "training_step_time": 8.567580223083496
    },
    {
      "epoch": 2.359294891357422e-05,
      "model_forward_time": 1.7540583610534668,
      "step": 98739
    },
    {
      "epoch": 2.359294891357422e-05,
      "step": 98739,
      "training_step_time": 7.683257102966309
    },
    {
      "epoch": 2.3593902587890624e-05,
      "grad_norm": 0.08280371129512787,
      "learning_rate": 4.34670114609792e-09,
      "loss": 0.0555,
      "step": 98740
    },
    {
      "epoch": 2.3593902587890624e-05,
      "model_forward_time": 1.756359577178955,
      "step": 98740
    },
    {
      "dynamics_loss_avg": 0.06376108396798372,
      "epoch": 2.3593902587890624e-05,
      "step": 98740
    },
    {
      "action_loss_avg": 0.0014488205488305539,
      "epoch": 2.3593902587890624e-05,
      "step": 98740
    },
    {
      "epoch": 2.3593902587890624e-05,
      "step": 98740,
      "training_step_time": 7.747433185577393
    },
    {
      "epoch": 2.359485626220703e-05,
      "model_forward_time": 1.7435898780822754,
      "step": 98741
    },
    {
      "epoch": 2.359485626220703e-05,
      "step": 98741,
      "training_step_time": 8.186211824417114
    },
    {
      "epoch": 2.359580993652344e-05,
      "model_forward_time": 1.7508716583251953,
      "step": 98742
    },
    {
      "epoch": 2.359580993652344e-05,
      "step": 98742,
      "training_step_time": 8.155741930007935
    },
    {
      "epoch": 2.3596763610839842e-05,
      "model_forward_time": 1.7515361309051514,
      "step": 98743
    },
    {
      "epoch": 2.3596763610839842e-05,
      "step": 98743,
      "training_step_time": 7.801920413970947
    },
    {
      "epoch": 2.359771728515625e-05,
      "model_forward_time": 1.7502448558807373,
      "step": 98744
    },
    {
      "epoch": 2.359771728515625e-05,
      "step": 98744,
      "training_step_time": 8.223504543304443
    },
    {
      "epoch": 2.3598670959472657e-05,
      "model_forward_time": 1.7471003532409668,
      "step": 98745
    },
    {
      "epoch": 2.3598670959472657e-05,
      "step": 98745,
      "training_step_time": 8.399116039276123
    },
    {
      "epoch": 2.359962463378906e-05,
      "model_forward_time": 1.7558348178863525,
      "step": 98746
    },
    {
      "epoch": 2.359962463378906e-05,
      "step": 98746,
      "training_step_time": 7.700111150741577
    },
    {
      "epoch": 2.360057830810547e-05,
      "model_forward_time": 2.4620165824890137,
      "step": 98747
    },
    {
      "epoch": 2.360057830810547e-05,
      "step": 98747,
      "training_step_time": 11.356637001037598
    },
    {
      "epoch": 2.3601531982421876e-05,
      "model_forward_time": 1.765549659729004,
      "step": 98748
    },
    {
      "epoch": 2.3601531982421876e-05,
      "step": 98748,
      "training_step_time": 7.728116035461426
    },
    {
      "epoch": 2.360248565673828e-05,
      "model_forward_time": 1.7448153495788574,
      "step": 98749
    },
    {
      "epoch": 2.360248565673828e-05,
      "step": 98749,
      "training_step_time": 9.475722312927246
    },
    {
      "epoch": 2.3603439331054688e-05,
      "grad_norm": 0.0791109949350357,
      "learning_rate": 4.278043754884498e-09,
      "loss": 0.0557,
      "step": 98750
    },
    {
      "epoch": 2.3603439331054688e-05,
      "model_forward_time": 1.7494430541992188,
      "step": 98750
    },
    {
      "dynamics_loss_avg": 0.0447438164614141,
      "epoch": 2.3603439331054688e-05,
      "step": 98750
    },
    {
      "action_loss_avg": 0.0011635785776888952,
      "epoch": 2.3603439331054688e-05,
      "step": 98750
    },
    {
      "epoch": 2.3603439331054688e-05,
      "step": 98750,
      "training_step_time": 8.5455002784729
    },
    {
      "epoch": 2.3604393005371095e-05,
      "model_forward_time": 1.741257905960083,
      "step": 98751
    },
    {
      "epoch": 2.3604393005371095e-05,
      "step": 98751,
      "training_step_time": 7.683557033538818
    },
    {
      "epoch": 2.36053466796875e-05,
      "model_forward_time": 1.7358930110931396,
      "step": 98752
    },
    {
      "epoch": 2.36053466796875e-05,
      "step": 98752,
      "training_step_time": 7.799604654312134
    },
    {
      "epoch": 2.3606300354003906e-05,
      "model_forward_time": 1.7407469749450684,
      "step": 98753
    },
    {
      "epoch": 2.3606300354003906e-05,
      "step": 98753,
      "training_step_time": 7.685251235961914
    },
    {
      "epoch": 2.3607254028320314e-05,
      "model_forward_time": 1.737318992614746,
      "step": 98754
    },
    {
      "epoch": 2.3607254028320314e-05,
      "step": 98754,
      "training_step_time": 8.777297019958496
    },
    {
      "epoch": 2.3608207702636718e-05,
      "model_forward_time": 1.733215570449829,
      "step": 98755
    },
    {
      "epoch": 2.3608207702636718e-05,
      "step": 98755,
      "training_step_time": 9.4643075466156
    },
    {
      "epoch": 2.3609161376953125e-05,
      "model_forward_time": 2.4249279499053955,
      "step": 98756
    },
    {
      "epoch": 2.3609161376953125e-05,
      "step": 98756,
      "training_step_time": 8.55159044265747
    },
    {
      "epoch": 2.3610115051269533e-05,
      "model_forward_time": 1.7381131649017334,
      "step": 98757
    },
    {
      "epoch": 2.3610115051269533e-05,
      "step": 98757,
      "training_step_time": 7.752581357955933
    },
    {
      "epoch": 2.3611068725585937e-05,
      "model_forward_time": 1.7479162216186523,
      "step": 98758
    },
    {
      "epoch": 2.3611068725585937e-05,
      "step": 98758,
      "training_step_time": 8.620854616165161
    },
    {
      "epoch": 2.3612022399902344e-05,
      "model_forward_time": 1.7428534030914307,
      "step": 98759
    },
    {
      "epoch": 2.3612022399902344e-05,
      "step": 98759,
      "training_step_time": 8.337594032287598
    },
    {
      "epoch": 2.361297607421875e-05,
      "grad_norm": 0.08114394545555115,
      "learning_rate": 4.209932688312712e-09,
      "loss": 0.0549,
      "step": 98760
    },
    {
      "epoch": 2.361297607421875e-05,
      "model_forward_time": 1.7460978031158447,
      "step": 98760
    },
    {
      "dynamics_loss_avg": 0.05977795943617821,
      "epoch": 2.361297607421875e-05,
      "step": 98760
    },
    {
      "action_loss_avg": 0.0016943545982940123,
      "epoch": 2.361297607421875e-05,
      "step": 98760
    },
    {
      "epoch": 2.361297607421875e-05,
      "step": 98760,
      "training_step_time": 7.668428897857666
    },
    {
      "epoch": 2.3613929748535156e-05,
      "model_forward_time": 1.735975742340088,
      "step": 98761
    },
    {
      "epoch": 2.3613929748535156e-05,
      "step": 98761,
      "training_step_time": 8.722288608551025
    },
    {
      "epoch": 2.3614883422851563e-05,
      "model_forward_time": 1.7335131168365479,
      "step": 98762
    },
    {
      "epoch": 2.3614883422851563e-05,
      "step": 98762,
      "training_step_time": 7.623900413513184
    },
    {
      "epoch": 2.361583709716797e-05,
      "model_forward_time": 1.7353675365447998,
      "step": 98763
    },
    {
      "epoch": 2.361583709716797e-05,
      "step": 98763,
      "training_step_time": 8.848081350326538
    },
    {
      "epoch": 2.3616790771484374e-05,
      "model_forward_time": 1.74558687210083,
      "step": 98764
    },
    {
      "epoch": 2.3616790771484374e-05,
      "step": 98764,
      "training_step_time": 8.283849000930786
    },
    {
      "epoch": 2.3617744445800782e-05,
      "model_forward_time": 2.4742820262908936,
      "step": 98765
    },
    {
      "epoch": 2.3617744445800782e-05,
      "step": 98765,
      "training_step_time": 8.618739366531372
    },
    {
      "epoch": 2.3618698120117186e-05,
      "model_forward_time": 1.7452318668365479,
      "step": 98766
    },
    {
      "epoch": 2.3618698120117186e-05,
      "step": 98766,
      "training_step_time": 7.710531949996948
    },
    {
      "epoch": 2.3619651794433593e-05,
      "model_forward_time": 1.7525029182434082,
      "step": 98767
    },
    {
      "epoch": 2.3619651794433593e-05,
      "step": 98767,
      "training_step_time": 7.795796155929565
    },
    {
      "epoch": 2.362060546875e-05,
      "model_forward_time": 1.754469871520996,
      "step": 98768
    },
    {
      "epoch": 2.362060546875e-05,
      "step": 98768,
      "training_step_time": 7.658101558685303
    },
    {
      "epoch": 2.3621559143066405e-05,
      "model_forward_time": 1.7523202896118164,
      "step": 98769
    },
    {
      "epoch": 2.3621559143066405e-05,
      "step": 98769,
      "training_step_time": 7.488238334655762
    },
    {
      "epoch": 2.3622512817382812e-05,
      "grad_norm": 0.1701866239309311,
      "learning_rate": 4.1423679538321605e-09,
      "loss": 0.0552,
      "step": 98770
    },
    {
      "epoch": 2.3622512817382812e-05,
      "model_forward_time": 1.7429533004760742,
      "step": 98770
    },
    {
      "dynamics_loss_avg": 0.04503101110458374,
      "epoch": 2.3622512817382812e-05,
      "step": 98770
    },
    {
      "action_loss_avg": 0.0014844999124761671,
      "epoch": 2.3622512817382812e-05,
      "step": 98770
    },
    {
      "epoch": 2.3622512817382812e-05,
      "step": 98770,
      "training_step_time": 7.716049432754517
    },
    {
      "epoch": 2.362346649169922e-05,
      "model_forward_time": 1.745903491973877,
      "step": 98771
    },
    {
      "epoch": 2.362346649169922e-05,
      "step": 98771,
      "training_step_time": 7.645938396453857
    },
    {
      "epoch": 2.3624420166015624e-05,
      "model_forward_time": 1.7534620761871338,
      "step": 98772
    },
    {
      "epoch": 2.3624420166015624e-05,
      "step": 98772,
      "training_step_time": 7.710453748703003
    },
    {
      "epoch": 2.362537384033203e-05,
      "model_forward_time": 1.749932050704956,
      "step": 98773
    },
    {
      "epoch": 2.362537384033203e-05,
      "step": 98773,
      "training_step_time": 7.775659561157227
    },
    {
      "epoch": 2.362632751464844e-05,
      "model_forward_time": 2.400879383087158,
      "step": 98774
    },
    {
      "epoch": 2.362632751464844e-05,
      "step": 98774,
      "training_step_time": 7.749568223953247
    },
    {
      "epoch": 2.3627281188964842e-05,
      "model_forward_time": 1.82114839553833,
      "step": 98775
    },
    {
      "epoch": 2.3627281188964842e-05,
      "step": 98775,
      "training_step_time": 7.763277292251587
    },
    {
      "epoch": 2.362823486328125e-05,
      "model_forward_time": 1.756279468536377,
      "step": 98776
    },
    {
      "epoch": 2.362823486328125e-05,
      "step": 98776,
      "training_step_time": 9.497578859329224
    },
    {
      "epoch": 2.3629188537597657e-05,
      "model_forward_time": 1.7400932312011719,
      "step": 98777
    },
    {
      "epoch": 2.3629188537597657e-05,
      "step": 98777,
      "training_step_time": 7.817208766937256
    },
    {
      "epoch": 2.363014221191406e-05,
      "model_forward_time": 1.7526633739471436,
      "step": 98778
    },
    {
      "epoch": 2.363014221191406e-05,
      "step": 98778,
      "training_step_time": 7.641024827957153
    },
    {
      "epoch": 2.363109588623047e-05,
      "model_forward_time": 1.8271400928497314,
      "step": 98779
    },
    {
      "epoch": 2.363109588623047e-05,
      "step": 98779,
      "training_step_time": 7.686582326889038
    },
    {
      "epoch": 2.3632049560546876e-05,
      "grad_norm": 0.12486238777637482,
      "learning_rate": 4.075349558830821e-09,
      "loss": 0.0557,
      "step": 98780
    },
    {
      "epoch": 2.3632049560546876e-05,
      "model_forward_time": 1.7443652153015137,
      "step": 98780
    },
    {
      "dynamics_loss_avg": 0.051833100989460944,
      "epoch": 2.3632049560546876e-05,
      "step": 98780
    },
    {
      "action_loss_avg": 0.0017282592598348856,
      "epoch": 2.3632049560546876e-05,
      "step": 98780
    },
    {
      "epoch": 2.3632049560546876e-05,
      "step": 98780,
      "training_step_time": 7.74589991569519
    },
    {
      "epoch": 2.363300323486328e-05,
      "model_forward_time": 1.7848832607269287,
      "step": 98781
    },
    {
      "epoch": 2.363300323486328e-05,
      "step": 98781,
      "training_step_time": 8.247197389602661
    },
    {
      "epoch": 2.3633956909179688e-05,
      "model_forward_time": 1.7420074939727783,
      "step": 98782
    },
    {
      "epoch": 2.3633956909179688e-05,
      "step": 98782,
      "training_step_time": 8.677050590515137
    },
    {
      "epoch": 2.3634910583496095e-05,
      "model_forward_time": 2.4055771827697754,
      "step": 98783
    },
    {
      "epoch": 2.3634910583496095e-05,
      "step": 98783,
      "training_step_time": 9.3968985080719
    },
    {
      "epoch": 2.36358642578125e-05,
      "model_forward_time": 1.7976069450378418,
      "step": 98784
    },
    {
      "epoch": 2.36358642578125e-05,
      "step": 98784,
      "training_step_time": 9.948121070861816
    },
    {
      "epoch": 2.3636817932128906e-05,
      "model_forward_time": 1.7480170726776123,
      "step": 98785
    },
    {
      "epoch": 2.3636817932128906e-05,
      "step": 98785,
      "training_step_time": 7.983278036117554
    },
    {
      "epoch": 2.3637771606445314e-05,
      "model_forward_time": 1.7486293315887451,
      "step": 98786
    },
    {
      "epoch": 2.3637771606445314e-05,
      "step": 98786,
      "training_step_time": 8.673285961151123
    },
    {
      "epoch": 2.3638725280761718e-05,
      "model_forward_time": 1.7450485229492188,
      "step": 98787
    },
    {
      "epoch": 2.3638725280761718e-05,
      "step": 98787,
      "training_step_time": 8.479490280151367
    },
    {
      "epoch": 2.3639678955078125e-05,
      "model_forward_time": 1.7562222480773926,
      "step": 98788
    },
    {
      "epoch": 2.3639678955078125e-05,
      "step": 98788,
      "training_step_time": 7.670457601547241
    },
    {
      "epoch": 2.3640632629394533e-05,
      "model_forward_time": 1.7467401027679443,
      "step": 98789
    },
    {
      "epoch": 2.3640632629394533e-05,
      "step": 98789,
      "training_step_time": 7.7181477546691895
    },
    {
      "epoch": 2.3641586303710937e-05,
      "grad_norm": 0.07276570796966553,
      "learning_rate": 4.008877510638387e-09,
      "loss": 0.0548,
      "step": 98790
    },
    {
      "epoch": 2.3641586303710937e-05,
      "model_forward_time": 1.7420921325683594,
      "step": 98790
    },
    {
      "dynamics_loss_avg": 0.0387406762689352,
      "epoch": 2.3641586303710937e-05,
      "step": 98790
    },
    {
      "action_loss_avg": 0.0013096083654090761,
      "epoch": 2.3641586303710937e-05,
      "step": 98790
    },
    {
      "epoch": 2.3641586303710937e-05,
      "step": 98790,
      "training_step_time": 7.578537702560425
    },
    {
      "epoch": 2.3642539978027344e-05,
      "model_forward_time": 1.7428529262542725,
      "step": 98791
    },
    {
      "epoch": 2.3642539978027344e-05,
      "step": 98791,
      "training_step_time": 7.873166084289551
    },
    {
      "epoch": 2.364349365234375e-05,
      "model_forward_time": 2.372117757797241,
      "step": 98792
    },
    {
      "epoch": 2.364349365234375e-05,
      "step": 98792,
      "training_step_time": 9.043396234512329
    },
    {
      "epoch": 2.3644447326660156e-05,
      "model_forward_time": 1.765681266784668,
      "step": 98793
    },
    {
      "epoch": 2.3644447326660156e-05,
      "step": 98793,
      "training_step_time": 7.752808570861816
    },
    {
      "epoch": 2.3645401000976563e-05,
      "model_forward_time": 1.0279185771942139,
      "step": 98794
    },
    {
      "epoch": 2.3645401000976563e-05,
      "step": 98794,
      "training_step_time": 8.376063823699951
    },
    {
      "epoch": 2.364635467529297e-05,
      "model_forward_time": 1.7378220558166504,
      "step": 98795
    },
    {
      "epoch": 2.364635467529297e-05,
      "step": 98795,
      "training_step_time": 9.447075366973877
    },
    {
      "epoch": 2.3647308349609374e-05,
      "model_forward_time": 1.7446479797363281,
      "step": 98796
    },
    {
      "epoch": 2.3647308349609374e-05,
      "step": 98796,
      "training_step_time": 7.720071792602539
    },
    {
      "epoch": 2.3648262023925782e-05,
      "model_forward_time": 1.7625770568847656,
      "step": 98797
    },
    {
      "epoch": 2.3648262023925782e-05,
      "step": 98797,
      "training_step_time": 7.686842679977417
    },
    {
      "epoch": 2.3649215698242186e-05,
      "model_forward_time": 1.7324397563934326,
      "step": 98798
    },
    {
      "epoch": 2.3649215698242186e-05,
      "step": 98798,
      "training_step_time": 7.801919460296631
    },
    {
      "epoch": 2.3650169372558593e-05,
      "model_forward_time": 1.7427821159362793,
      "step": 98799
    },
    {
      "epoch": 2.3650169372558593e-05,
      "step": 98799,
      "training_step_time": 7.794399976730347
    },
    {
      "epoch": 2.3651123046875e-05,
      "grad_norm": 0.09049707651138306,
      "learning_rate": 3.942951816523488e-09,
      "loss": 0.0561,
      "step": 98800
    },
    {
      "epoch": 2.3651123046875e-05,
      "model_forward_time": 1.783808946609497,
      "step": 98800
    },
    {
      "dynamics_loss_avg": 0.048700308054685594,
      "epoch": 2.3651123046875e-05,
      "step": 98800
    },
    {
      "action_loss_avg": 0.0013128392776707188,
      "epoch": 2.3651123046875e-05,
      "step": 98800
    },
    {
      "epoch": 2.3651123046875e-05,
      "step": 98800,
      "training_step_time": 8.593234777450562
    },
    {
      "epoch": 2.3652076721191405e-05,
      "model_forward_time": 2.4224700927734375,
      "step": 98801
    },
    {
      "epoch": 2.3652076721191405e-05,
      "step": 98801,
      "training_step_time": 7.816942453384399
    },
    {
      "epoch": 2.3653030395507812e-05,
      "model_forward_time": 1.3705201148986816,
      "step": 98802
    },
    {
      "epoch": 2.3653030395507812e-05,
      "step": 98802,
      "training_step_time": 8.900915384292603
    },
    {
      "epoch": 2.365398406982422e-05,
      "model_forward_time": 1.8040413856506348,
      "step": 98803
    },
    {
      "epoch": 2.365398406982422e-05,
      "step": 98803,
      "training_step_time": 9.720468521118164
    },
    {
      "epoch": 2.3654937744140624e-05,
      "model_forward_time": 1.7502899169921875,
      "step": 98804
    },
    {
      "epoch": 2.3654937744140624e-05,
      "step": 98804,
      "training_step_time": 9.867141723632812
    },
    {
      "epoch": 2.365589141845703e-05,
      "model_forward_time": 1.7425708770751953,
      "step": 98805
    },
    {
      "epoch": 2.365589141845703e-05,
      "step": 98805,
      "training_step_time": 7.712840557098389
    },
    {
      "epoch": 2.365684509277344e-05,
      "model_forward_time": 1.7353813648223877,
      "step": 98806
    },
    {
      "epoch": 2.365684509277344e-05,
      "step": 98806,
      "training_step_time": 7.889372825622559
    },
    {
      "epoch": 2.3657798767089842e-05,
      "model_forward_time": 1.7457284927368164,
      "step": 98807
    },
    {
      "epoch": 2.3657798767089842e-05,
      "step": 98807,
      "training_step_time": 7.758611440658569
    },
    {
      "epoch": 2.365875244140625e-05,
      "model_forward_time": 1.7398416996002197,
      "step": 98808
    },
    {
      "epoch": 2.365875244140625e-05,
      "step": 98808,
      "training_step_time": 7.814832448959351
    },
    {
      "epoch": 2.3659706115722657e-05,
      "model_forward_time": 1.7438249588012695,
      "step": 98809
    },
    {
      "epoch": 2.3659706115722657e-05,
      "step": 98809,
      "training_step_time": 8.600300312042236
    },
    {
      "epoch": 2.366065979003906e-05,
      "grad_norm": 0.08739815652370453,
      "learning_rate": 3.877572483695913e-09,
      "loss": 0.0556,
      "step": 98810
    },
    {
      "epoch": 2.366065979003906e-05,
      "model_forward_time": 2.366143226623535,
      "step": 98810
    },
    {
      "dynamics_loss_avg": 0.04508549999445677,
      "epoch": 2.366065979003906e-05,
      "step": 98810
    },
    {
      "action_loss_avg": 0.0009435242682229728,
      "epoch": 2.366065979003906e-05,
      "step": 98810
    },
    {
      "epoch": 2.366065979003906e-05,
      "step": 98810,
      "training_step_time": 9.5827054977417
    },
    {
      "epoch": 2.366161346435547e-05,
      "model_forward_time": 1.7495708465576172,
      "step": 98811
    },
    {
      "epoch": 2.366161346435547e-05,
      "step": 98811,
      "training_step_time": 9.541960954666138
    },
    {
      "epoch": 2.3662567138671876e-05,
      "model_forward_time": 1.8091034889221191,
      "step": 98812
    },
    {
      "epoch": 2.3662567138671876e-05,
      "step": 98812,
      "training_step_time": 8.679112672805786
    },
    {
      "epoch": 2.366352081298828e-05,
      "model_forward_time": 1.7496225833892822,
      "step": 98813
    },
    {
      "epoch": 2.366352081298828e-05,
      "step": 98813,
      "training_step_time": 8.808480501174927
    },
    {
      "epoch": 2.3664474487304688e-05,
      "model_forward_time": 1.7451577186584473,
      "step": 98814
    },
    {
      "epoch": 2.3664474487304688e-05,
      "step": 98814,
      "training_step_time": 7.874433517456055
    },
    {
      "epoch": 2.3665428161621095e-05,
      "model_forward_time": 1.7497282028198242,
      "step": 98815
    },
    {
      "epoch": 2.3665428161621095e-05,
      "step": 98815,
      "training_step_time": 8.46997594833374
    },
    {
      "epoch": 2.36663818359375e-05,
      "model_forward_time": 1.749938726425171,
      "step": 98816
    },
    {
      "epoch": 2.36663818359375e-05,
      "step": 98816,
      "training_step_time": 7.830514669418335
    },
    {
      "epoch": 2.3667335510253906e-05,
      "model_forward_time": 1.7527203559875488,
      "step": 98817
    },
    {
      "epoch": 2.3667335510253906e-05,
      "step": 98817,
      "training_step_time": 8.687063455581665
    },
    {
      "epoch": 2.3668289184570314e-05,
      "model_forward_time": 1.7410919666290283,
      "step": 98818
    },
    {
      "epoch": 2.3668289184570314e-05,
      "step": 98818,
      "training_step_time": 8.548272609710693
    },
    {
      "epoch": 2.3669242858886718e-05,
      "model_forward_time": 1.747861623764038,
      "step": 98819
    },
    {
      "epoch": 2.3669242858886718e-05,
      "step": 98819,
      "training_step_time": 7.8767876625061035
    },
    {
      "epoch": 2.3670196533203125e-05,
      "grad_norm": 0.08437386155128479,
      "learning_rate": 3.812739519305498e-09,
      "loss": 0.0536,
      "step": 98820
    },
    {
      "epoch": 2.3670196533203125e-05,
      "model_forward_time": 2.3892245292663574,
      "step": 98820
    },
    {
      "dynamics_loss_avg": 0.04870644100010395,
      "epoch": 2.3670196533203125e-05,
      "step": 98820
    },
    {
      "action_loss_avg": 0.0012239080650033429,
      "epoch": 2.3670196533203125e-05,
      "step": 98820
    },
    {
      "epoch": 2.3670196533203125e-05,
      "step": 98820,
      "training_step_time": 8.376872301101685
    },
    {
      "epoch": 2.3671150207519533e-05,
      "model_forward_time": 1.8117473125457764,
      "step": 98821
    },
    {
      "epoch": 2.3671150207519533e-05,
      "step": 98821,
      "training_step_time": 8.3502516746521
    },
    {
      "epoch": 2.3672103881835937e-05,
      "model_forward_time": 1.744722843170166,
      "step": 98822
    },
    {
      "epoch": 2.3672103881835937e-05,
      "step": 98822,
      "training_step_time": 8.706620454788208
    },
    {
      "epoch": 2.3673057556152344e-05,
      "model_forward_time": 1.74684476852417,
      "step": 98823
    },
    {
      "epoch": 2.3673057556152344e-05,
      "step": 98823,
      "training_step_time": 7.725391626358032
    },
    {
      "epoch": 2.367401123046875e-05,
      "model_forward_time": 1.7506744861602783,
      "step": 98824
    },
    {
      "epoch": 2.367401123046875e-05,
      "step": 98824,
      "training_step_time": 7.3946521282196045
    },
    {
      "epoch": 2.3674964904785156e-05,
      "model_forward_time": 1.7438240051269531,
      "step": 98825
    },
    {
      "epoch": 2.3674964904785156e-05,
      "step": 98825,
      "training_step_time": 7.719504117965698
    },
    {
      "epoch": 2.3675918579101563e-05,
      "model_forward_time": 1.7473058700561523,
      "step": 98826
    },
    {
      "epoch": 2.3675918579101563e-05,
      "step": 98826,
      "training_step_time": 8.525379419326782
    },
    {
      "epoch": 2.367687225341797e-05,
      "model_forward_time": 1.747727870941162,
      "step": 98827
    },
    {
      "epoch": 2.367687225341797e-05,
      "step": 98827,
      "training_step_time": 7.721609115600586
    },
    {
      "epoch": 2.3677825927734374e-05,
      "model_forward_time": 1.7449851036071777,
      "step": 98828
    },
    {
      "epoch": 2.3677825927734374e-05,
      "step": 98828,
      "training_step_time": 8.67239761352539
    },
    {
      "epoch": 2.3678779602050782e-05,
      "model_forward_time": 2.382230758666992,
      "step": 98829
    },
    {
      "epoch": 2.3678779602050782e-05,
      "step": 98829,
      "training_step_time": 8.374935388565063
    },
    {
      "epoch": 2.3679733276367186e-05,
      "grad_norm": 0.08241968601942062,
      "learning_rate": 3.748452930442125e-09,
      "loss": 0.0533,
      "step": 98830
    },
    {
      "epoch": 2.3679733276367186e-05,
      "model_forward_time": 1.8110816478729248,
      "step": 98830
    },
    {
      "dynamics_loss_avg": 0.05815989561378956,
      "epoch": 2.3679733276367186e-05,
      "step": 98830
    },
    {
      "action_loss_avg": 0.0010823698248714209,
      "epoch": 2.3679733276367186e-05,
      "step": 98830
    },
    {
      "epoch": 2.3679733276367186e-05,
      "step": 98830,
      "training_step_time": 9.451128244400024
    },
    {
      "epoch": 2.3680686950683593e-05,
      "model_forward_time": 1.748159408569336,
      "step": 98831
    },
    {
      "epoch": 2.3680686950683593e-05,
      "step": 98831,
      "training_step_time": 9.390787839889526
    },
    {
      "epoch": 2.3681640625e-05,
      "model_forward_time": 1.7495837211608887,
      "step": 98832
    },
    {
      "epoch": 2.3681640625e-05,
      "step": 98832,
      "training_step_time": 8.647052764892578
    },
    {
      "epoch": 2.3682594299316405e-05,
      "model_forward_time": 1.74997878074646,
      "step": 98833
    },
    {
      "epoch": 2.3682594299316405e-05,
      "step": 98833,
      "training_step_time": 7.717164754867554
    },
    {
      "epoch": 2.3683547973632812e-05,
      "model_forward_time": 1.7505178451538086,
      "step": 98834
    },
    {
      "epoch": 2.3683547973632812e-05,
      "step": 98834,
      "training_step_time": 7.7156102657318115
    },
    {
      "epoch": 2.368450164794922e-05,
      "model_forward_time": 1.7523188591003418,
      "step": 98835
    },
    {
      "epoch": 2.368450164794922e-05,
      "step": 98835,
      "training_step_time": 7.803873538970947
    },
    {
      "epoch": 2.3685455322265624e-05,
      "model_forward_time": 1.74562668800354,
      "step": 98836
    },
    {
      "epoch": 2.3685455322265624e-05,
      "step": 98836,
      "training_step_time": 7.765576362609863
    },
    {
      "epoch": 2.368640899658203e-05,
      "model_forward_time": 1.7790374755859375,
      "step": 98837
    },
    {
      "epoch": 2.368640899658203e-05,
      "step": 98837,
      "training_step_time": 8.716910600662231
    },
    {
      "epoch": 2.368736267089844e-05,
      "model_forward_time": 2.4064571857452393,
      "step": 98838
    },
    {
      "epoch": 2.368736267089844e-05,
      "step": 98838,
      "training_step_time": 8.825054168701172
    },
    {
      "epoch": 2.3688316345214842e-05,
      "model_forward_time": 1.8174562454223633,
      "step": 98839
    },
    {
      "epoch": 2.3688316345214842e-05,
      "step": 98839,
      "training_step_time": 7.800524473190308
    },
    {
      "epoch": 2.368927001953125e-05,
      "grad_norm": 0.1232617124915123,
      "learning_rate": 3.6847127241362855e-09,
      "loss": 0.0529,
      "step": 98840
    },
    {
      "epoch": 2.368927001953125e-05,
      "model_forward_time": 1.753469467163086,
      "step": 98840
    },
    {
      "dynamics_loss_avg": 0.04451968772336841,
      "epoch": 2.368927001953125e-05,
      "step": 98840
    },
    {
      "action_loss_avg": 0.001249998394632712,
      "epoch": 2.368927001953125e-05,
      "step": 98840
    },
    {
      "epoch": 2.368927001953125e-05,
      "step": 98840,
      "training_step_time": 8.599671602249146
    },
    {
      "epoch": 2.3690223693847657e-05,
      "model_forward_time": 1.7449066638946533,
      "step": 98841
    },
    {
      "epoch": 2.3690223693847657e-05,
      "step": 98841,
      "training_step_time": 7.763449668884277
    },
    {
      "epoch": 2.369117736816406e-05,
      "model_forward_time": 1.7440991401672363,
      "step": 98842
    },
    {
      "epoch": 2.369117736816406e-05,
      "step": 98842,
      "training_step_time": 7.707275867462158
    },
    {
      "epoch": 2.369213104248047e-05,
      "model_forward_time": 1.75075101852417,
      "step": 98843
    },
    {
      "epoch": 2.369213104248047e-05,
      "step": 98843,
      "training_step_time": 8.208834648132324
    },
    {
      "epoch": 2.3693084716796876e-05,
      "model_forward_time": 1.7497951984405518,
      "step": 98844
    },
    {
      "epoch": 2.3693084716796876e-05,
      "step": 98844,
      "training_step_time": 7.746776103973389
    },
    {
      "epoch": 2.369403839111328e-05,
      "model_forward_time": 1.7392699718475342,
      "step": 98845
    },
    {
      "epoch": 2.369403839111328e-05,
      "step": 98845,
      "training_step_time": 7.758609771728516
    },
    {
      "epoch": 2.3694992065429688e-05,
      "model_forward_time": 1.751455545425415,
      "step": 98846
    },
    {
      "epoch": 2.3694992065429688e-05,
      "step": 98846,
      "training_step_time": 7.778964519500732
    },
    {
      "epoch": 2.3695945739746095e-05,
      "model_forward_time": 2.428560256958008,
      "step": 98847
    },
    {
      "epoch": 2.3695945739746095e-05,
      "step": 98847,
      "training_step_time": 8.842650651931763
    },
    {
      "epoch": 2.36968994140625e-05,
      "model_forward_time": 1.8069884777069092,
      "step": 98848
    },
    {
      "epoch": 2.36968994140625e-05,
      "step": 98848,
      "training_step_time": 9.092259883880615
    },
    {
      "epoch": 2.3697853088378906e-05,
      "model_forward_time": 1.7408652305603027,
      "step": 98849
    },
    {
      "epoch": 2.3697853088378906e-05,
      "step": 98849,
      "training_step_time": 7.912278890609741
    },
    {
      "epoch": 2.3698806762695314e-05,
      "grad_norm": 0.1009400486946106,
      "learning_rate": 3.621518907358512e-09,
      "loss": 0.0539,
      "step": 98850
    },
    {
      "epoch": 2.3698806762695314e-05,
      "model_forward_time": 1.7424790859222412,
      "step": 98850
    },
    {
      "dynamics_loss_avg": 0.05500837545841932,
      "epoch": 2.3698806762695314e-05,
      "step": 98850
    },
    {
      "action_loss_avg": 0.0009851803246419878,
      "epoch": 2.3698806762695314e-05,
      "step": 98850
    },
    {
      "epoch": 2.3698806762695314e-05,
      "step": 98850,
      "training_step_time": 7.782495498657227
    },
    {
      "epoch": 2.3699760437011718e-05,
      "model_forward_time": 1.750260591506958,
      "step": 98851
    },
    {
      "epoch": 2.3699760437011718e-05,
      "step": 98851,
      "training_step_time": 7.758302450180054
    },
    {
      "epoch": 2.3700714111328125e-05,
      "model_forward_time": 1.7424736022949219,
      "step": 98852
    },
    {
      "epoch": 2.3700714111328125e-05,
      "step": 98852,
      "training_step_time": 7.6825385093688965
    },
    {
      "epoch": 2.3701667785644533e-05,
      "model_forward_time": 1.74637770652771,
      "step": 98853
    },
    {
      "epoch": 2.3701667785644533e-05,
      "step": 98853,
      "training_step_time": 7.788399934768677
    },
    {
      "epoch": 2.3702621459960937e-05,
      "model_forward_time": 1.7391035556793213,
      "step": 98854
    },
    {
      "epoch": 2.3702621459960937e-05,
      "step": 98854,
      "training_step_time": 8.663028955459595
    },
    {
      "epoch": 2.3703575134277344e-05,
      "model_forward_time": 1.7321741580963135,
      "step": 98855
    },
    {
      "epoch": 2.3703575134277344e-05,
      "step": 98855,
      "training_step_time": 8.444190263748169
    },
    {
      "epoch": 2.370452880859375e-05,
      "model_forward_time": 2.372948408126831,
      "step": 98856
    },
    {
      "epoch": 2.370452880859375e-05,
      "step": 98856,
      "training_step_time": 9.686527252197266
    },
    {
      "epoch": 2.3705482482910156e-05,
      "model_forward_time": 1.834726095199585,
      "step": 98857
    },
    {
      "epoch": 2.3705482482910156e-05,
      "step": 98857,
      "training_step_time": 9.430826902389526
    },
    {
      "epoch": 2.3706436157226563e-05,
      "model_forward_time": 1.7458903789520264,
      "step": 98858
    },
    {
      "epoch": 2.3706436157226563e-05,
      "step": 98858,
      "training_step_time": 9.80254316329956
    },
    {
      "epoch": 2.370738983154297e-05,
      "model_forward_time": 1.7510993480682373,
      "step": 98859
    },
    {
      "epoch": 2.370738983154297e-05,
      "step": 98859,
      "training_step_time": 8.840723752975464
    },
    {
      "epoch": 2.3708343505859374e-05,
      "grad_norm": 0.07195184379816055,
      "learning_rate": 3.5588714870193886e-09,
      "loss": 0.0528,
      "step": 98860
    },
    {
      "epoch": 2.3708343505859374e-05,
      "model_forward_time": 1.7342939376831055,
      "step": 98860
    },
    {
      "dynamics_loss_avg": 0.04649199098348618,
      "epoch": 2.3708343505859374e-05,
      "step": 98860
    },
    {
      "action_loss_avg": 0.0011570124072022737,
      "epoch": 2.3708343505859374e-05,
      "step": 98860
    },
    {
      "epoch": 2.3708343505859374e-05,
      "step": 98860,
      "training_step_time": 7.820581912994385
    },
    {
      "epoch": 2.3709297180175782e-05,
      "model_forward_time": 1.7395756244659424,
      "step": 98861
    },
    {
      "epoch": 2.3709297180175782e-05,
      "step": 98861,
      "training_step_time": 7.740387678146362
    },
    {
      "epoch": 2.3710250854492186e-05,
      "model_forward_time": 1.7481131553649902,
      "step": 98862
    },
    {
      "epoch": 2.3710250854492186e-05,
      "step": 98862,
      "training_step_time": 7.8895652294158936
    },
    {
      "epoch": 2.3711204528808593e-05,
      "model_forward_time": 1.7446141242980957,
      "step": 98863
    },
    {
      "epoch": 2.3711204528808593e-05,
      "step": 98863,
      "training_step_time": 7.5883166790008545
    },
    {
      "epoch": 2.3712158203125e-05,
      "model_forward_time": 1.754737138748169,
      "step": 98864
    },
    {
      "epoch": 2.3712158203125e-05,
      "step": 98864,
      "training_step_time": 7.7995054721832275
    },
    {
      "epoch": 2.3713111877441405e-05,
      "model_forward_time": 1.7438931465148926,
      "step": 98865
    },
    {
      "epoch": 2.3713111877441405e-05,
      "step": 98865,
      "training_step_time": 9.438032865524292
    },
    {
      "epoch": 2.3714065551757812e-05,
      "model_forward_time": 1.7580933570861816,
      "step": 98866
    },
    {
      "epoch": 2.3714065551757812e-05,
      "step": 98866,
      "training_step_time": 7.8074727058410645
    },
    {
      "epoch": 2.371501922607422e-05,
      "model_forward_time": 1.7493696212768555,
      "step": 98867
    },
    {
      "epoch": 2.371501922607422e-05,
      "step": 98867,
      "training_step_time": 7.816231966018677
    },
    {
      "epoch": 2.3715972900390624e-05,
      "model_forward_time": 1.7487599849700928,
      "step": 98868
    },
    {
      "epoch": 2.3715972900390624e-05,
      "step": 98868,
      "training_step_time": 9.17587947845459
    },
    {
      "epoch": 2.371692657470703e-05,
      "model_forward_time": 1.7517666816711426,
      "step": 98869
    },
    {
      "epoch": 2.371692657470703e-05,
      "step": 98869,
      "training_step_time": 8.714241981506348
    },
    {
      "epoch": 2.371788024902344e-05,
      "grad_norm": 0.10343827307224274,
      "learning_rate": 3.496770469970101e-09,
      "loss": 0.0532,
      "step": 98870
    },
    {
      "epoch": 2.371788024902344e-05,
      "model_forward_time": 1.7363054752349854,
      "step": 98870
    },
    {
      "dynamics_loss_avg": 0.05795127972960472,
      "epoch": 2.371788024902344e-05,
      "step": 98870
    },
    {
      "action_loss_avg": 0.002357531653251499,
      "epoch": 2.371788024902344e-05,
      "step": 98870
    },
    {
      "epoch": 2.371788024902344e-05,
      "step": 98870,
      "training_step_time": 8.544680118560791
    },
    {
      "epoch": 2.3718833923339842e-05,
      "model_forward_time": 1.758704423904419,
      "step": 98871
    },
    {
      "epoch": 2.3718833923339842e-05,
      "step": 98871,
      "training_step_time": 7.877520561218262
    },
    {
      "epoch": 2.371978759765625e-05,
      "model_forward_time": 1.7442212104797363,
      "step": 98872
    },
    {
      "epoch": 2.371978759765625e-05,
      "step": 98872,
      "training_step_time": 7.6458961963653564
    },
    {
      "epoch": 2.3720741271972657e-05,
      "model_forward_time": 1.7492234706878662,
      "step": 98873
    },
    {
      "epoch": 2.3720741271972657e-05,
      "step": 98873,
      "training_step_time": 8.39760422706604
    },
    {
      "epoch": 2.372169494628906e-05,
      "model_forward_time": 1.7618138790130615,
      "step": 98874
    },
    {
      "epoch": 2.372169494628906e-05,
      "step": 98874,
      "training_step_time": 7.76252818107605
    },
    {
      "epoch": 2.372264862060547e-05,
      "model_forward_time": 2.3925113677978516,
      "step": 98875
    },
    {
      "epoch": 2.372264862060547e-05,
      "step": 98875,
      "training_step_time": 8.518864631652832
    },
    {
      "epoch": 2.3723602294921876e-05,
      "model_forward_time": 1.7479541301727295,
      "step": 98876
    },
    {
      "epoch": 2.3723602294921876e-05,
      "step": 98876,
      "training_step_time": 9.442338228225708
    },
    {
      "epoch": 2.372455596923828e-05,
      "model_forward_time": 1.7522096633911133,
      "step": 98877
    },
    {
      "epoch": 2.372455596923828e-05,
      "step": 98877,
      "training_step_time": 7.8122406005859375
    },
    {
      "epoch": 2.3725509643554688e-05,
      "model_forward_time": 1.0282440185546875,
      "step": 98878
    },
    {
      "epoch": 2.3725509643554688e-05,
      "step": 98878,
      "training_step_time": 7.766913175582886
    },
    {
      "epoch": 2.3726463317871095e-05,
      "model_forward_time": 1.7449169158935547,
      "step": 98879
    },
    {
      "epoch": 2.3726463317871095e-05,
      "step": 98879,
      "training_step_time": 8.299108028411865
    },
    {
      "epoch": 2.37274169921875e-05,
      "grad_norm": 0.09094849973917007,
      "learning_rate": 3.4352158630018837e-09,
      "loss": 0.0528,
      "step": 98880
    },
    {
      "epoch": 2.37274169921875e-05,
      "model_forward_time": 1.7454862594604492,
      "step": 98880
    },
    {
      "dynamics_loss_avg": 0.04271283764392138,
      "epoch": 2.37274169921875e-05,
      "step": 98880
    },
    {
      "action_loss_avg": 0.000732001505093649,
      "epoch": 2.37274169921875e-05,
      "step": 98880
    },
    {
      "epoch": 2.37274169921875e-05,
      "step": 98880,
      "training_step_time": 7.7843708992004395
    },
    {
      "epoch": 2.3728370666503906e-05,
      "model_forward_time": 1.7438185214996338,
      "step": 98881
    },
    {
      "epoch": 2.3728370666503906e-05,
      "step": 98881,
      "training_step_time": 7.635989665985107
    },
    {
      "epoch": 2.3729324340820314e-05,
      "model_forward_time": 1.7450308799743652,
      "step": 98882
    },
    {
      "epoch": 2.3729324340820314e-05,
      "step": 98882,
      "training_step_time": 7.766970872879028
    },
    {
      "epoch": 2.3730278015136718e-05,
      "model_forward_time": 1.7453546524047852,
      "step": 98883
    },
    {
      "epoch": 2.3730278015136718e-05,
      "step": 98883,
      "training_step_time": 9.352394104003906
    },
    {
      "epoch": 2.3731231689453125e-05,
      "model_forward_time": 2.405571699142456,
      "step": 98884
    },
    {
      "epoch": 2.3731231689453125e-05,
      "step": 98884,
      "training_step_time": 7.694985389709473
    },
    {
      "epoch": 2.3732185363769533e-05,
      "model_forward_time": 1.8141164779663086,
      "step": 98885
    },
    {
      "epoch": 2.3732185363769533e-05,
      "step": 98885,
      "training_step_time": 7.793507099151611
    },
    {
      "epoch": 2.3733139038085937e-05,
      "model_forward_time": 1.797943353652954,
      "step": 98886
    },
    {
      "epoch": 2.3733139038085937e-05,
      "step": 98886,
      "training_step_time": 9.693446636199951
    },
    {
      "epoch": 2.3734092712402344e-05,
      "model_forward_time": 1.7281677722930908,
      "step": 98887
    },
    {
      "epoch": 2.3734092712402344e-05,
      "step": 98887,
      "training_step_time": 7.763243675231934
    },
    {
      "epoch": 2.373504638671875e-05,
      "model_forward_time": 1.7497711181640625,
      "step": 98888
    },
    {
      "epoch": 2.373504638671875e-05,
      "step": 98888,
      "training_step_time": 7.791473388671875
    },
    {
      "epoch": 2.3736000061035156e-05,
      "model_forward_time": 1.742156744003296,
      "step": 98889
    },
    {
      "epoch": 2.3736000061035156e-05,
      "step": 98889,
      "training_step_time": 7.757325887680054
    },
    {
      "epoch": 2.3736953735351563e-05,
      "grad_norm": 0.09543486684560776,
      "learning_rate": 3.3742076728460194e-09,
      "loss": 0.0543,
      "step": 98890
    },
    {
      "epoch": 2.3736953735351563e-05,
      "model_forward_time": 1.745842695236206,
      "step": 98890
    },
    {
      "dynamics_loss_avg": 0.06507921814918519,
      "epoch": 2.3736953735351563e-05,
      "step": 98890
    },
    {
      "action_loss_avg": 0.00160617457004264,
      "epoch": 2.3736953735351563e-05,
      "step": 98890
    },
    {
      "epoch": 2.3736953735351563e-05,
      "step": 98890,
      "training_step_time": 7.547529220581055
    },
    {
      "epoch": 2.3737907409667967e-05,
      "model_forward_time": 1.75162935256958,
      "step": 98891
    },
    {
      "epoch": 2.3737907409667967e-05,
      "step": 98891,
      "training_step_time": 7.807352066040039
    },
    {
      "epoch": 2.3738861083984374e-05,
      "model_forward_time": 1.7442739009857178,
      "step": 98892
    },
    {
      "epoch": 2.3738861083984374e-05,
      "step": 98892,
      "training_step_time": 7.772873163223267
    },
    {
      "epoch": 2.3739814758300782e-05,
      "model_forward_time": 2.372378349304199,
      "step": 98893
    },
    {
      "epoch": 2.3739814758300782e-05,
      "step": 98893,
      "training_step_time": 7.726469278335571
    },
    {
      "epoch": 2.3740768432617186e-05,
      "model_forward_time": 1.8191421031951904,
      "step": 98894
    },
    {
      "epoch": 2.3740768432617186e-05,
      "step": 98894,
      "training_step_time": 9.709238767623901
    },
    {
      "epoch": 2.3741722106933593e-05,
      "model_forward_time": 1.742246150970459,
      "step": 98895
    },
    {
      "epoch": 2.3741722106933593e-05,
      "step": 98895,
      "training_step_time": 8.692882537841797
    },
    {
      "epoch": 2.374267578125e-05,
      "model_forward_time": 1.7441563606262207,
      "step": 98896
    },
    {
      "epoch": 2.374267578125e-05,
      "step": 98896,
      "training_step_time": 7.81727409362793
    },
    {
      "epoch": 2.3743629455566405e-05,
      "model_forward_time": 1.7503149509429932,
      "step": 98897
    },
    {
      "epoch": 2.3743629455566405e-05,
      "step": 98897,
      "training_step_time": 7.720299482345581
    },
    {
      "epoch": 2.3744583129882812e-05,
      "model_forward_time": 1.738455057144165,
      "step": 98898
    },
    {
      "epoch": 2.3744583129882812e-05,
      "step": 98898,
      "training_step_time": 7.750808238983154
    },
    {
      "epoch": 2.374553680419922e-05,
      "model_forward_time": 1.744823694229126,
      "step": 98899
    },
    {
      "epoch": 2.374553680419922e-05,
      "step": 98899,
      "training_step_time": 7.8054585456848145
    },
    {
      "epoch": 2.3746490478515624e-05,
      "grad_norm": 0.07870873808860779,
      "learning_rate": 3.3137459061743925e-09,
      "loss": 0.0548,
      "step": 98900
    },
    {
      "epoch": 2.3746490478515624e-05,
      "model_forward_time": 1.7476983070373535,
      "step": 98900
    },
    {
      "dynamics_loss_avg": 0.064869474619627,
      "epoch": 2.3746490478515624e-05,
      "step": 98900
    },
    {
      "action_loss_avg": 0.001284193730680272,
      "epoch": 2.3746490478515624e-05,
      "step": 98900
    },
    {
      "epoch": 2.3746490478515624e-05,
      "step": 98900,
      "training_step_time": 7.734558582305908
    },
    {
      "epoch": 2.374744415283203e-05,
      "model_forward_time": 1.74283766746521,
      "step": 98901
    },
    {
      "epoch": 2.374744415283203e-05,
      "step": 98901,
      "training_step_time": 9.464953184127808
    },
    {
      "epoch": 2.374839782714844e-05,
      "model_forward_time": 1.73612380027771,
      "step": 98902
    },
    {
      "epoch": 2.374839782714844e-05,
      "step": 98902,
      "training_step_time": 8.296868562698364
    },
    {
      "epoch": 2.3749351501464843e-05,
      "model_forward_time": 1.7974655628204346,
      "step": 98903
    },
    {
      "epoch": 2.3749351501464843e-05,
      "step": 98903,
      "training_step_time": 8.612188339233398
    },
    {
      "epoch": 2.375030517578125e-05,
      "model_forward_time": 1.743272066116333,
      "step": 98904
    },
    {
      "epoch": 2.375030517578125e-05,
      "step": 98904,
      "training_step_time": 7.870751857757568
    },
    {
      "epoch": 2.3751258850097657e-05,
      "model_forward_time": 1.743290901184082,
      "step": 98905
    },
    {
      "epoch": 2.3751258850097657e-05,
      "step": 98905,
      "training_step_time": 8.836268424987793
    },
    {
      "epoch": 2.375221252441406e-05,
      "model_forward_time": 1.743180274963379,
      "step": 98906
    },
    {
      "epoch": 2.375221252441406e-05,
      "step": 98906,
      "training_step_time": 7.71541953086853
    },
    {
      "epoch": 2.375316619873047e-05,
      "model_forward_time": 1.7403013706207275,
      "step": 98907
    },
    {
      "epoch": 2.375316619873047e-05,
      "step": 98907,
      "training_step_time": 8.746114730834961
    },
    {
      "epoch": 2.3754119873046876e-05,
      "model_forward_time": 1.7439594268798828,
      "step": 98908
    },
    {
      "epoch": 2.3754119873046876e-05,
      "step": 98908,
      "training_step_time": 7.763105392456055
    },
    {
      "epoch": 2.375507354736328e-05,
      "model_forward_time": 1.7503325939178467,
      "step": 98909
    },
    {
      "epoch": 2.375507354736328e-05,
      "step": 98909,
      "training_step_time": 9.427675008773804
    },
    {
      "epoch": 2.3756027221679688e-05,
      "grad_norm": 0.08591888099908829,
      "learning_rate": 3.253830569599492e-09,
      "loss": 0.0533,
      "step": 98910
    },
    {
      "epoch": 2.3756027221679688e-05,
      "model_forward_time": 1.7507743835449219,
      "step": 98910
    },
    {
      "dynamics_loss_avg": 0.048592520598322156,
      "epoch": 2.3756027221679688e-05,
      "step": 98910
    },
    {
      "action_loss_avg": 0.0009659751958679408,
      "epoch": 2.3756027221679688e-05,
      "step": 98910
    },
    {
      "epoch": 2.3756027221679688e-05,
      "step": 98910,
      "training_step_time": 8.35942554473877
    },
    {
      "epoch": 2.3756980895996095e-05,
      "model_forward_time": 1.7386789321899414,
      "step": 98911
    },
    {
      "epoch": 2.3756980895996095e-05,
      "step": 98911,
      "training_step_time": 7.786267518997192
    },
    {
      "epoch": 2.37579345703125e-05,
      "model_forward_time": 2.416013240814209,
      "step": 98912
    },
    {
      "epoch": 2.37579345703125e-05,
      "step": 98912,
      "training_step_time": 8.701953887939453
    },
    {
      "epoch": 2.3758888244628906e-05,
      "model_forward_time": 1.7510936260223389,
      "step": 98913
    },
    {
      "epoch": 2.3758888244628906e-05,
      "step": 98913,
      "training_step_time": 8.989629030227661
    },
    {
      "epoch": 2.3759841918945314e-05,
      "model_forward_time": 1.7362284660339355,
      "step": 98914
    },
    {
      "epoch": 2.3759841918945314e-05,
      "step": 98914,
      "training_step_time": 8.509256362915039
    },
    {
      "epoch": 2.3760795593261718e-05,
      "model_forward_time": 1.7547712326049805,
      "step": 98915
    },
    {
      "epoch": 2.3760795593261718e-05,
      "step": 98915,
      "training_step_time": 7.773209810256958
    },
    {
      "epoch": 2.3761749267578125e-05,
      "model_forward_time": 1.7455296516418457,
      "step": 98916
    },
    {
      "epoch": 2.3761749267578125e-05,
      "step": 98916,
      "training_step_time": 8.781013488769531
    },
    {
      "epoch": 2.3762702941894533e-05,
      "model_forward_time": 1.7824110984802246,
      "step": 98917
    },
    {
      "epoch": 2.3762702941894533e-05,
      "step": 98917,
      "training_step_time": 8.669867038726807
    },
    {
      "epoch": 2.3763656616210937e-05,
      "model_forward_time": 1.7542040348052979,
      "step": 98918
    },
    {
      "epoch": 2.3763656616210937e-05,
      "step": 98918,
      "training_step_time": 7.5016162395477295
    },
    {
      "epoch": 2.3764610290527344e-05,
      "model_forward_time": 1.7501652240753174,
      "step": 98919
    },
    {
      "epoch": 2.3764610290527344e-05,
      "step": 98919,
      "training_step_time": 7.741216421127319
    },
    {
      "epoch": 2.376556396484375e-05,
      "grad_norm": 0.12857040762901306,
      "learning_rate": 3.1944616696727437e-09,
      "loss": 0.0534,
      "step": 98920
    },
    {
      "epoch": 2.376556396484375e-05,
      "model_forward_time": 1.740365982055664,
      "step": 98920
    },
    {
      "dynamics_loss_avg": 0.05212977807968855,
      "epoch": 2.376556396484375e-05,
      "step": 98920
    },
    {
      "action_loss_avg": 0.0007487401802791282,
      "epoch": 2.376556396484375e-05,
      "step": 98920
    },
    {
      "epoch": 2.376556396484375e-05,
      "step": 98920,
      "training_step_time": 8.336098909378052
    },
    {
      "epoch": 2.3766517639160156e-05,
      "model_forward_time": 2.3913044929504395,
      "step": 98921
    },
    {
      "epoch": 2.3766517639160156e-05,
      "step": 98921,
      "training_step_time": 7.815891265869141
    },
    {
      "epoch": 2.3767471313476563e-05,
      "model_forward_time": 1.8137476444244385,
      "step": 98922
    },
    {
      "epoch": 2.3767471313476563e-05,
      "step": 98922,
      "training_step_time": 9.170081377029419
    },
    {
      "epoch": 2.3768424987792967e-05,
      "model_forward_time": 1.7569007873535156,
      "step": 98923
    },
    {
      "epoch": 2.3768424987792967e-05,
      "step": 98923,
      "training_step_time": 7.884931325912476
    },
    {
      "epoch": 2.3769378662109375e-05,
      "model_forward_time": 1.7446770668029785,
      "step": 98924
    },
    {
      "epoch": 2.3769378662109375e-05,
      "step": 98924,
      "training_step_time": 7.822607755661011
    },
    {
      "epoch": 2.3770332336425782e-05,
      "model_forward_time": 1.7440967559814453,
      "step": 98925
    },
    {
      "epoch": 2.3770332336425782e-05,
      "step": 98925,
      "training_step_time": 8.783381700515747
    },
    {
      "epoch": 2.3771286010742186e-05,
      "model_forward_time": 1.7416095733642578,
      "step": 98926
    },
    {
      "epoch": 2.3771286010742186e-05,
      "step": 98926,
      "training_step_time": 8.813662767410278
    },
    {
      "epoch": 2.3772239685058593e-05,
      "model_forward_time": 1.7427396774291992,
      "step": 98927
    },
    {
      "epoch": 2.3772239685058593e-05,
      "step": 98927,
      "training_step_time": 7.778769254684448
    },
    {
      "epoch": 2.3773193359375e-05,
      "model_forward_time": 1.7408788204193115,
      "step": 98928
    },
    {
      "epoch": 2.3773193359375e-05,
      "step": 98928,
      "training_step_time": 7.806305408477783
    },
    {
      "epoch": 2.3774147033691405e-05,
      "model_forward_time": 1.7916138172149658,
      "step": 98929
    },
    {
      "epoch": 2.3774147033691405e-05,
      "step": 98929,
      "training_step_time": 7.773174524307251
    },
    {
      "epoch": 2.3775100708007812e-05,
      "grad_norm": 0.10031087696552277,
      "learning_rate": 3.135639212887287e-09,
      "loss": 0.0554,
      "step": 98930
    },
    {
      "epoch": 2.3775100708007812e-05,
      "model_forward_time": 2.4167020320892334,
      "step": 98930
    },
    {
      "dynamics_loss_avg": 0.04245754526928067,
      "epoch": 2.3775100708007812e-05,
      "step": 98930
    },
    {
      "action_loss_avg": 0.0007052561348245944,
      "epoch": 2.3775100708007812e-05,
      "step": 98930
    },
    {
      "epoch": 2.3775100708007812e-05,
      "step": 98930,
      "training_step_time": 8.792968273162842
    },
    {
      "epoch": 2.377605438232422e-05,
      "model_forward_time": 1.8146250247955322,
      "step": 98931
    },
    {
      "epoch": 2.377605438232422e-05,
      "step": 98931,
      "training_step_time": 9.237180471420288
    },
    {
      "epoch": 2.3777008056640624e-05,
      "model_forward_time": 1.7448782920837402,
      "step": 98932
    },
    {
      "epoch": 2.3777008056640624e-05,
      "step": 98932,
      "training_step_time": 8.759673118591309
    },
    {
      "epoch": 2.377796173095703e-05,
      "model_forward_time": 1.7586028575897217,
      "step": 98933
    },
    {
      "epoch": 2.377796173095703e-05,
      "step": 98933,
      "training_step_time": 8.52868914604187
    },
    {
      "epoch": 2.377891540527344e-05,
      "model_forward_time": 1.7414679527282715,
      "step": 98934
    },
    {
      "epoch": 2.377891540527344e-05,
      "step": 98934,
      "training_step_time": 7.773083448410034
    },
    {
      "epoch": 2.3779869079589843e-05,
      "model_forward_time": 1.7492866516113281,
      "step": 98935
    },
    {
      "epoch": 2.3779869079589843e-05,
      "step": 98935,
      "training_step_time": 7.7378644943237305
    },
    {
      "epoch": 2.378082275390625e-05,
      "model_forward_time": 1.8017997741699219,
      "step": 98936
    },
    {
      "epoch": 2.378082275390625e-05,
      "step": 98936,
      "training_step_time": 7.777405500411987
    },
    {
      "epoch": 2.3781776428222657e-05,
      "model_forward_time": 1.7616939544677734,
      "step": 98937
    },
    {
      "epoch": 2.3781776428222657e-05,
      "step": 98937,
      "training_step_time": 7.756483793258667
    },
    {
      "epoch": 2.378273010253906e-05,
      "model_forward_time": 1.7464799880981445,
      "step": 98938
    },
    {
      "epoch": 2.378273010253906e-05,
      "step": 98938,
      "training_step_time": 8.59238052368164
    },
    {
      "epoch": 2.378368377685547e-05,
      "model_forward_time": 2.43725848197937,
      "step": 98939
    },
    {
      "epoch": 2.378368377685547e-05,
      "step": 98939,
      "training_step_time": 8.809616327285767
    },
    {
      "epoch": 2.3784637451171876e-05,
      "grad_norm": 0.3069079518318176,
      "learning_rate": 3.077363205675754e-09,
      "loss": 0.0552,
      "step": 98940
    },
    {
      "epoch": 2.3784637451171876e-05,
      "model_forward_time": 1.7646191120147705,
      "step": 98940
    },
    {
      "dynamics_loss_avg": 0.04953795447945595,
      "epoch": 2.3784637451171876e-05,
      "step": 98940
    },
    {
      "action_loss_avg": 0.0010590981401037425,
      "epoch": 2.3784637451171876e-05,
      "step": 98940
    },
    {
      "epoch": 2.3784637451171876e-05,
      "step": 98940,
      "training_step_time": 7.763657569885254
    },
    {
      "epoch": 2.378559112548828e-05,
      "model_forward_time": 1.7436864376068115,
      "step": 98941
    },
    {
      "epoch": 2.378559112548828e-05,
      "step": 98941,
      "training_step_time": 7.831936836242676
    },
    {
      "epoch": 2.3786544799804688e-05,
      "model_forward_time": 1.7627003192901611,
      "step": 98942
    },
    {
      "epoch": 2.3786544799804688e-05,
      "step": 98942,
      "training_step_time": 9.948532342910767
    },
    {
      "epoch": 2.3787498474121095e-05,
      "model_forward_time": 1.7502410411834717,
      "step": 98943
    },
    {
      "epoch": 2.3787498474121095e-05,
      "step": 98943,
      "training_step_time": 8.326905965805054
    },
    {
      "epoch": 2.37884521484375e-05,
      "model_forward_time": 1.7566680908203125,
      "step": 98944
    },
    {
      "epoch": 2.37884521484375e-05,
      "step": 98944,
      "training_step_time": 7.772451877593994
    },
    {
      "epoch": 2.3789405822753907e-05,
      "model_forward_time": 1.7487730979919434,
      "step": 98945
    },
    {
      "epoch": 2.3789405822753907e-05,
      "step": 98945,
      "training_step_time": 7.85623025894165
    },
    {
      "epoch": 2.3790359497070314e-05,
      "model_forward_time": 1.7533671855926514,
      "step": 98946
    },
    {
      "epoch": 2.3790359497070314e-05,
      "step": 98946,
      "training_step_time": 7.739612579345703
    },
    {
      "epoch": 2.3791313171386718e-05,
      "model_forward_time": 1.3772318363189697,
      "step": 98947
    },
    {
      "epoch": 2.3791313171386718e-05,
      "step": 98947,
      "training_step_time": 8.698668718338013
    },
    {
      "epoch": 2.3792266845703125e-05,
      "model_forward_time": 2.361593723297119,
      "step": 98948
    },
    {
      "epoch": 2.3792266845703125e-05,
      "step": 98948,
      "training_step_time": 8.362849473953247
    },
    {
      "epoch": 2.3793220520019533e-05,
      "model_forward_time": 1.7412519454956055,
      "step": 98949
    },
    {
      "epoch": 2.3793220520019533e-05,
      "step": 98949,
      "training_step_time": 9.04762077331543
    },
    {
      "epoch": 2.3794174194335937e-05,
      "grad_norm": 0.09587816148996353,
      "learning_rate": 3.0196336544113804e-09,
      "loss": 0.0547,
      "step": 98950
    },
    {
      "epoch": 2.3794174194335937e-05,
      "model_forward_time": 1.797795295715332,
      "step": 98950
    },
    {
      "dynamics_loss_avg": 0.04136154595762491,
      "epoch": 2.3794174194335937e-05,
      "step": 98950
    },
    {
      "action_loss_avg": 0.0010139760619495065,
      "epoch": 2.3794174194335937e-05,
      "step": 98950
    },
    {
      "epoch": 2.3794174194335937e-05,
      "step": 98950,
      "training_step_time": 10.085338354110718
    },
    {
      "epoch": 2.3795127868652344e-05,
      "model_forward_time": 1.7898602485656738,
      "step": 98951
    },
    {
      "epoch": 2.3795127868652344e-05,
      "step": 98951,
      "training_step_time": 8.828274011611938
    },
    {
      "epoch": 2.379608154296875e-05,
      "model_forward_time": 1.7425346374511719,
      "step": 98952
    },
    {
      "epoch": 2.379608154296875e-05,
      "step": 98952,
      "training_step_time": 7.837194204330444
    },
    {
      "epoch": 2.3797035217285156e-05,
      "model_forward_time": 1.7406604290008545,
      "step": 98953
    },
    {
      "epoch": 2.3797035217285156e-05,
      "step": 98953,
      "training_step_time": 7.986250162124634
    },
    {
      "epoch": 2.3797988891601563e-05,
      "model_forward_time": 1.7351059913635254,
      "step": 98954
    },
    {
      "epoch": 2.3797988891601563e-05,
      "step": 98954,
      "training_step_time": 7.853005886077881
    },
    {
      "epoch": 2.3798942565917967e-05,
      "model_forward_time": 1.7319777011871338,
      "step": 98955
    },
    {
      "epoch": 2.3798942565917967e-05,
      "step": 98955,
      "training_step_time": 9.220656156539917
    },
    {
      "epoch": 2.3799896240234375e-05,
      "model_forward_time": 1.7908694744110107,
      "step": 98956
    },
    {
      "epoch": 2.3799896240234375e-05,
      "step": 98956,
      "training_step_time": 8.396174192428589
    },
    {
      "epoch": 2.3800849914550782e-05,
      "model_forward_time": 1.7418665885925293,
      "step": 98957
    },
    {
      "epoch": 2.3800849914550782e-05,
      "step": 98957,
      "training_step_time": 8.668927907943726
    },
    {
      "epoch": 2.3801803588867186e-05,
      "model_forward_time": 2.3888330459594727,
      "step": 98958
    },
    {
      "epoch": 2.3801803588867186e-05,
      "step": 98958,
      "training_step_time": 9.045025825500488
    },
    {
      "epoch": 2.3802757263183593e-05,
      "model_forward_time": 1.8101558685302734,
      "step": 98959
    },
    {
      "epoch": 2.3802757263183593e-05,
      "step": 98959,
      "training_step_time": 8.792052745819092
    },
    {
      "epoch": 2.38037109375e-05,
      "grad_norm": 0.10702111572027206,
      "learning_rate": 2.9624505654063386e-09,
      "loss": 0.0561,
      "step": 98960
    },
    {
      "epoch": 2.38037109375e-05,
      "model_forward_time": 1.7387340068817139,
      "step": 98960
    },
    {
      "dynamics_loss_avg": 0.056060316041111946,
      "epoch": 2.38037109375e-05,
      "step": 98960
    },
    {
      "action_loss_avg": 0.0013176073203794657,
      "epoch": 2.38037109375e-05,
      "step": 98960
    },
    {
      "epoch": 2.38037109375e-05,
      "step": 98960,
      "training_step_time": 8.479360103607178
    },
    {
      "epoch": 2.3804664611816405e-05,
      "model_forward_time": 1.7446973323822021,
      "step": 98961
    },
    {
      "epoch": 2.3804664611816405e-05,
      "step": 98961,
      "training_step_time": 7.84655499458313
    },
    {
      "epoch": 2.3805618286132812e-05,
      "model_forward_time": 1.7471110820770264,
      "step": 98962
    },
    {
      "epoch": 2.3805618286132812e-05,
      "step": 98962,
      "training_step_time": 7.896191835403442
    },
    {
      "epoch": 2.380657196044922e-05,
      "model_forward_time": 1.7388086318969727,
      "step": 98963
    },
    {
      "epoch": 2.380657196044922e-05,
      "step": 98963,
      "training_step_time": 8.814453840255737
    },
    {
      "epoch": 2.3807525634765624e-05,
      "model_forward_time": 1.740940809249878,
      "step": 98964
    },
    {
      "epoch": 2.3807525634765624e-05,
      "step": 98964,
      "training_step_time": 7.813278675079346
    },
    {
      "epoch": 2.380847930908203e-05,
      "model_forward_time": 1.741227388381958,
      "step": 98965
    },
    {
      "epoch": 2.380847930908203e-05,
      "step": 98965,
      "training_step_time": 8.489779233932495
    },
    {
      "epoch": 2.380943298339844e-05,
      "model_forward_time": 1.7464017868041992,
      "step": 98966
    },
    {
      "epoch": 2.380943298339844e-05,
      "step": 98966,
      "training_step_time": 8.346009731292725
    },
    {
      "epoch": 2.3810386657714843e-05,
      "model_forward_time": 2.3960793018341064,
      "step": 98967
    },
    {
      "epoch": 2.3810386657714843e-05,
      "step": 98967,
      "training_step_time": 8.62300992012024
    },
    {
      "epoch": 2.381134033203125e-05,
      "model_forward_time": 1.7961153984069824,
      "step": 98968
    },
    {
      "epoch": 2.381134033203125e-05,
      "step": 98968,
      "training_step_time": 8.831599473953247
    },
    {
      "epoch": 2.3812294006347657e-05,
      "model_forward_time": 1.7451725006103516,
      "step": 98969
    },
    {
      "epoch": 2.3812294006347657e-05,
      "step": 98969,
      "training_step_time": 8.568877458572388
    },
    {
      "epoch": 2.381324768066406e-05,
      "grad_norm": 0.08775224536657333,
      "learning_rate": 2.90581394491507e-09,
      "loss": 0.0551,
      "step": 98970
    },
    {
      "epoch": 2.381324768066406e-05,
      "model_forward_time": 1.74361252784729,
      "step": 98970
    },
    {
      "dynamics_loss_avg": 0.050431196205317975,
      "epoch": 2.381324768066406e-05,
      "step": 98970
    },
    {
      "action_loss_avg": 0.0012192246067570522,
      "epoch": 2.381324768066406e-05,
      "step": 98970
    },
    {
      "epoch": 2.381324768066406e-05,
      "step": 98970,
      "training_step_time": 7.812022686004639
    },
    {
      "epoch": 2.381420135498047e-05,
      "model_forward_time": 1.7540297508239746,
      "step": 98971
    },
    {
      "epoch": 2.381420135498047e-05,
      "step": 98971,
      "training_step_time": 7.790044784545898
    },
    {
      "epoch": 2.3815155029296876e-05,
      "model_forward_time": 1.7649617195129395,
      "step": 98972
    },
    {
      "epoch": 2.3815155029296876e-05,
      "step": 98972,
      "training_step_time": 8.614015579223633
    },
    {
      "epoch": 2.381610870361328e-05,
      "model_forward_time": 1.7389044761657715,
      "step": 98973
    },
    {
      "epoch": 2.381610870361328e-05,
      "step": 98973,
      "training_step_time": 7.7736053466796875
    },
    {
      "epoch": 2.3817062377929688e-05,
      "model_forward_time": 1.7529046535491943,
      "step": 98974
    },
    {
      "epoch": 2.3817062377929688e-05,
      "step": 98974,
      "training_step_time": 7.566431760787964
    },
    {
      "epoch": 2.3818016052246095e-05,
      "model_forward_time": 1.7368783950805664,
      "step": 98975
    },
    {
      "epoch": 2.3818016052246095e-05,
      "step": 98975,
      "training_step_time": 7.7628350257873535
    },
    {
      "epoch": 2.38189697265625e-05,
      "model_forward_time": 2.414548397064209,
      "step": 98976
    },
    {
      "epoch": 2.38189697265625e-05,
      "step": 98976,
      "training_step_time": 8.47479248046875
    },
    {
      "epoch": 2.3819923400878907e-05,
      "model_forward_time": 1.769871711730957,
      "step": 98977
    },
    {
      "epoch": 2.3819923400878907e-05,
      "step": 98977,
      "training_step_time": 7.825477600097656
    },
    {
      "epoch": 2.3820877075195314e-05,
      "model_forward_time": 1.7312119007110596,
      "step": 98978
    },
    {
      "epoch": 2.3820877075195314e-05,
      "step": 98978,
      "training_step_time": 7.895026922225952
    },
    {
      "epoch": 2.3821830749511718e-05,
      "model_forward_time": 1.7322006225585938,
      "step": 98979
    },
    {
      "epoch": 2.3821830749511718e-05,
      "step": 98979,
      "training_step_time": 9.341323614120483
    },
    {
      "epoch": 2.3822784423828125e-05,
      "grad_norm": 0.10249069333076477,
      "learning_rate": 2.8497237991309545e-09,
      "loss": 0.0559,
      "step": 98980
    },
    {
      "epoch": 2.3822784423828125e-05,
      "model_forward_time": 1.7579116821289062,
      "step": 98980
    },
    {
      "dynamics_loss_avg": 0.05979087185114622,
      "epoch": 2.3822784423828125e-05,
      "step": 98980
    },
    {
      "action_loss_avg": 0.0009731257683597505,
      "epoch": 2.3822784423828125e-05,
      "step": 98980
    },
    {
      "epoch": 2.3822784423828125e-05,
      "step": 98980,
      "training_step_time": 7.894930124282837
    },
    {
      "epoch": 2.3823738098144533e-05,
      "model_forward_time": 1.0280158519744873,
      "step": 98981
    },
    {
      "epoch": 2.3823738098144533e-05,
      "step": 98981,
      "training_step_time": 7.830329179763794
    },
    {
      "epoch": 2.3824691772460937e-05,
      "model_forward_time": 1.734999179840088,
      "step": 98982
    },
    {
      "epoch": 2.3824691772460937e-05,
      "step": 98982,
      "training_step_time": 7.7986531257629395
    },
    {
      "epoch": 2.3825645446777344e-05,
      "model_forward_time": 1.7426223754882812,
      "step": 98983
    },
    {
      "epoch": 2.3825645446777344e-05,
      "step": 98983,
      "training_step_time": 7.770676851272583
    },
    {
      "epoch": 2.382659912109375e-05,
      "model_forward_time": 1.7358794212341309,
      "step": 98984
    },
    {
      "epoch": 2.382659912109375e-05,
      "step": 98984,
      "training_step_time": 8.642752885818481
    },
    {
      "epoch": 2.3827552795410156e-05,
      "model_forward_time": 1.73512864112854,
      "step": 98985
    },
    {
      "epoch": 2.3827552795410156e-05,
      "step": 98985,
      "training_step_time": 8.957650661468506
    },
    {
      "epoch": 2.3828506469726563e-05,
      "model_forward_time": 1.7447888851165771,
      "step": 98986
    },
    {
      "epoch": 2.3828506469726563e-05,
      "step": 98986,
      "training_step_time": 7.819722890853882
    },
    {
      "epoch": 2.3829460144042967e-05,
      "model_forward_time": 1.7354762554168701,
      "step": 98987
    },
    {
      "epoch": 2.3829460144042967e-05,
      "step": 98987,
      "training_step_time": 7.866628885269165
    },
    {
      "epoch": 2.3830413818359375e-05,
      "model_forward_time": 1.7401349544525146,
      "step": 98988
    },
    {
      "epoch": 2.3830413818359375e-05,
      "step": 98988,
      "training_step_time": 7.82806921005249
    },
    {
      "epoch": 2.3831367492675782e-05,
      "model_forward_time": 1.7367303371429443,
      "step": 98989
    },
    {
      "epoch": 2.3831367492675782e-05,
      "step": 98989,
      "training_step_time": 7.803071975708008
    },
    {
      "epoch": 2.3832321166992186e-05,
      "grad_norm": 0.09104285389184952,
      "learning_rate": 2.7941801341879736e-09,
      "loss": 0.0552,
      "step": 98990
    },
    {
      "epoch": 2.3832321166992186e-05,
      "model_forward_time": 1.7382452487945557,
      "step": 98990
    },
    {
      "dynamics_loss_avg": 0.06190420016646385,
      "epoch": 2.3832321166992186e-05,
      "step": 98990
    },
    {
      "action_loss_avg": 0.0019523132301401346,
      "epoch": 2.3832321166992186e-05,
      "step": 98990
    },
    {
      "epoch": 2.3832321166992186e-05,
      "step": 98990,
      "training_step_time": 7.76424765586853
    },
    {
      "epoch": 2.3833274841308593e-05,
      "model_forward_time": 1.7766075134277344,
      "step": 98991
    },
    {
      "epoch": 2.3833274841308593e-05,
      "step": 98991,
      "training_step_time": 9.814805507659912
    },
    {
      "epoch": 2.3834228515625e-05,
      "model_forward_time": 1.7303619384765625,
      "step": 98992
    },
    {
      "epoch": 2.3834228515625e-05,
      "step": 98992,
      "training_step_time": 7.769285678863525
    },
    {
      "epoch": 2.3835182189941405e-05,
      "model_forward_time": 1.750246286392212,
      "step": 98993
    },
    {
      "epoch": 2.3835182189941405e-05,
      "step": 98993,
      "training_step_time": 7.7993323802948
    },
    {
      "epoch": 2.3836135864257812e-05,
      "model_forward_time": 1.8111765384674072,
      "step": 98994
    },
    {
      "epoch": 2.3836135864257812e-05,
      "step": 98994,
      "training_step_time": 7.7922632694244385
    },
    {
      "epoch": 2.383708953857422e-05,
      "model_forward_time": 2.4077541828155518,
      "step": 98995
    },
    {
      "epoch": 2.383708953857422e-05,
      "step": 98995,
      "training_step_time": 8.499658107757568
    },
    {
      "epoch": 2.3838043212890624e-05,
      "model_forward_time": 1.8171956539154053,
      "step": 98996
    },
    {
      "epoch": 2.3838043212890624e-05,
      "step": 98996,
      "training_step_time": 8.933146715164185
    },
    {
      "epoch": 2.383899688720703e-05,
      "model_forward_time": 1.7621920108795166,
      "step": 98997
    },
    {
      "epoch": 2.383899688720703e-05,
      "step": 98997,
      "training_step_time": 8.66913366317749
    },
    {
      "epoch": 2.383995056152344e-05,
      "model_forward_time": 1.7913610935211182,
      "step": 98998
    },
    {
      "epoch": 2.383995056152344e-05,
      "step": 98998,
      "training_step_time": 9.149987697601318
    },
    {
      "epoch": 2.3840904235839843e-05,
      "model_forward_time": 1.7562580108642578,
      "step": 98999
    },
    {
      "epoch": 2.3840904235839843e-05,
      "step": 98999,
      "training_step_time": 8.859091520309448
    },
    {
      "epoch": 2.384185791015625e-05,
      "grad_norm": 0.07546765357255936,
      "learning_rate": 2.739182956160158e-09,
      "loss": 0.0541,
      "step": 99000
    },
    {
      "epoch": 2.384185791015625e-05,
      "model_forward_time": 1.438871145248413,
      "step": 99000
    },
    {
      "dynamics_loss_avg": 0.04800521582365036,
      "epoch": 2.384185791015625e-05,
      "step": 99000
    },
    {
      "action_loss_avg": 0.0009989388519898057,
      "epoch": 2.384185791015625e-05,
      "step": 99000
    },
    {
      "epoch": 2.384185791015625e-05,
      "step": 99000,
      "training_step_time": 6.363667726516724
    },
    {
      "epoch": 2.3842811584472657e-05,
      "model_forward_time": 1.744990587234497,
      "step": 99001
    },
    {
      "epoch": 2.3842811584472657e-05,
      "step": 99001,
      "training_step_time": 6.999390602111816
    },
    {
      "epoch": 2.384376525878906e-05,
      "model_forward_time": 1.7364728450775146,
      "step": 99002
    },
    {
      "epoch": 2.384376525878906e-05,
      "step": 99002,
      "training_step_time": 6.970529079437256
    },
    {
      "epoch": 2.384471893310547e-05,
      "model_forward_time": 1.7435188293457031,
      "step": 99003
    },
    {
      "epoch": 2.384471893310547e-05,
      "step": 99003,
      "training_step_time": 6.9161460399627686
    },
    {
      "epoch": 2.3845672607421876e-05,
      "model_forward_time": 1.77278470993042,
      "step": 99004
    },
    {
      "epoch": 2.3845672607421876e-05,
      "step": 99004,
      "training_step_time": 6.955902338027954
    },
    {
      "epoch": 2.384662628173828e-05,
      "model_forward_time": 1.741309404373169,
      "step": 99005
    },
    {
      "epoch": 2.384662628173828e-05,
      "step": 99005,
      "training_step_time": 6.917083024978638
    },
    {
      "epoch": 2.3847579956054688e-05,
      "model_forward_time": 1.7509098052978516,
      "step": 99006
    },
    {
      "epoch": 2.3847579956054688e-05,
      "step": 99006,
      "training_step_time": 6.904126167297363
    },
    {
      "epoch": 2.3848533630371095e-05,
      "model_forward_time": 2.948058605194092,
      "step": 99007
    },
    {
      "epoch": 2.3848533630371095e-05,
      "step": 99007,
      "training_step_time": 10.570179462432861
    },
    {
      "epoch": 2.38494873046875e-05,
      "model_forward_time": 1.8668701648712158,
      "step": 99008
    },
    {
      "epoch": 2.38494873046875e-05,
      "step": 99008,
      "training_step_time": 13.772939920425415
    },
    {
      "epoch": 2.3850440979003907e-05,
      "model_forward_time": 1.7432126998901367,
      "step": 99009
    },
    {
      "epoch": 2.3850440979003907e-05,
      "step": 99009,
      "training_step_time": 7.781838417053223
    },
    {
      "epoch": 2.3851394653320314e-05,
      "grad_norm": 0.08178593963384628,
      "learning_rate": 2.6847322710621405e-09,
      "loss": 0.0537,
      "step": 99010
    },
    {
      "epoch": 2.3851394653320314e-05,
      "model_forward_time": 1.7411303520202637,
      "step": 99010
    },
    {
      "dynamics_loss_avg": 0.05982349831610918,
      "epoch": 2.3851394653320314e-05,
      "step": 99010
    },
    {
      "action_loss_avg": 0.002309339347993955,
      "epoch": 2.3851394653320314e-05,
      "step": 99010
    },
    {
      "epoch": 2.3851394653320314e-05,
      "step": 99010,
      "training_step_time": 6.8919837474823
    },
    {
      "epoch": 2.3852348327636718e-05,
      "model_forward_time": 1.7350139617919922,
      "step": 99011
    },
    {
      "epoch": 2.3852348327636718e-05,
      "step": 99011,
      "training_step_time": 6.925431728363037
    },
    {
      "epoch": 2.3853302001953125e-05,
      "model_forward_time": 1.7305634021759033,
      "step": 99012
    },
    {
      "epoch": 2.3853302001953125e-05,
      "step": 99012,
      "training_step_time": 6.930186033248901
    },
    {
      "epoch": 2.3854255676269533e-05,
      "model_forward_time": 1.7438347339630127,
      "step": 99013
    },
    {
      "epoch": 2.3854255676269533e-05,
      "step": 99013,
      "training_step_time": 6.9499452114105225
    },
    {
      "epoch": 2.3855209350585937e-05,
      "model_forward_time": 1.7322885990142822,
      "step": 99014
    },
    {
      "epoch": 2.3855209350585937e-05,
      "step": 99014,
      "training_step_time": 6.892282962799072
    },
    {
      "epoch": 2.3856163024902344e-05,
      "model_forward_time": 1.7362890243530273,
      "step": 99015
    },
    {
      "epoch": 2.3856163024902344e-05,
      "step": 99015,
      "training_step_time": 6.8166184425354
    },
    {
      "epoch": 2.385711669921875e-05,
      "model_forward_time": 2.39877986907959,
      "step": 99016
    },
    {
      "epoch": 2.385711669921875e-05,
      "step": 99016,
      "training_step_time": 7.245982646942139
    },
    {
      "epoch": 2.3858070373535156e-05,
      "model_forward_time": 1.763265609741211,
      "step": 99017
    },
    {
      "epoch": 2.3858070373535156e-05,
      "step": 99017,
      "training_step_time": 12.23248839378357
    },
    {
      "epoch": 2.3859024047851563e-05,
      "model_forward_time": 1.36478853225708,
      "step": 99018
    },
    {
      "epoch": 2.3859024047851563e-05,
      "step": 99018,
      "training_step_time": 7.776612043380737
    },
    {
      "epoch": 2.3859977722167967e-05,
      "model_forward_time": 1.7335712909698486,
      "step": 99019
    },
    {
      "epoch": 2.3859977722167967e-05,
      "step": 99019,
      "training_step_time": 6.864943981170654
    },
    {
      "epoch": 2.3860931396484375e-05,
      "grad_norm": 0.1070176437497139,
      "learning_rate": 2.630828084848602e-09,
      "loss": 0.0542,
      "step": 99020
    },
    {
      "epoch": 2.3860931396484375e-05,
      "model_forward_time": 1.7381465435028076,
      "step": 99020
    },
    {
      "dynamics_loss_avg": 0.05109862890094519,
      "epoch": 2.3860931396484375e-05,
      "step": 99020
    },
    {
      "action_loss_avg": 0.0024479906482156367,
      "epoch": 2.3860931396484375e-05,
      "step": 99020
    },
    {
      "epoch": 2.3860931396484375e-05,
      "step": 99020,
      "training_step_time": 6.91509485244751
    },
    {
      "epoch": 2.3861885070800782e-05,
      "model_forward_time": 1.7361657619476318,
      "step": 99021
    },
    {
      "epoch": 2.3861885070800782e-05,
      "step": 99021,
      "training_step_time": 6.943779230117798
    },
    {
      "epoch": 2.3862838745117186e-05,
      "model_forward_time": 1.7468163967132568,
      "step": 99022
    },
    {
      "epoch": 2.3862838745117186e-05,
      "step": 99022,
      "training_step_time": 6.9009411334991455
    },
    {
      "epoch": 2.3863792419433593e-05,
      "model_forward_time": 1.785815715789795,
      "step": 99023
    },
    {
      "epoch": 2.3863792419433593e-05,
      "step": 99023,
      "training_step_time": 6.939261436462402
    },
    {
      "epoch": 2.386474609375e-05,
      "model_forward_time": 1.740206241607666,
      "step": 99024
    },
    {
      "epoch": 2.386474609375e-05,
      "step": 99024,
      "training_step_time": 6.918451547622681
    },
    {
      "epoch": 2.3865699768066405e-05,
      "model_forward_time": 1.7380294799804688,
      "step": 99025
    },
    {
      "epoch": 2.3865699768066405e-05,
      "step": 99025,
      "training_step_time": 6.896952867507935
    },
    {
      "epoch": 2.3866653442382812e-05,
      "model_forward_time": 2.3750619888305664,
      "step": 99026
    },
    {
      "epoch": 2.3866653442382812e-05,
      "step": 99026,
      "training_step_time": 10.607293605804443
    },
    {
      "epoch": 2.386760711669922e-05,
      "model_forward_time": 1.741797685623169,
      "step": 99027
    },
    {
      "epoch": 2.386760711669922e-05,
      "step": 99027,
      "training_step_time": 7.82161283493042
    },
    {
      "epoch": 2.3868560791015624e-05,
      "model_forward_time": 1.7410011291503906,
      "step": 99028
    },
    {
      "epoch": 2.3868560791015624e-05,
      "step": 99028,
      "training_step_time": 7.113049745559692
    },
    {
      "epoch": 2.386951446533203e-05,
      "model_forward_time": 1.7375807762145996,
      "step": 99029
    },
    {
      "epoch": 2.386951446533203e-05,
      "step": 99029,
      "training_step_time": 6.947972297668457
    },
    {
      "epoch": 2.387046813964844e-05,
      "grad_norm": 0.09974334388971329,
      "learning_rate": 2.5774704034137176e-09,
      "loss": 0.0551,
      "step": 99030
    },
    {
      "epoch": 2.387046813964844e-05,
      "model_forward_time": 1.7369804382324219,
      "step": 99030
    },
    {
      "dynamics_loss_avg": 0.06524972449988127,
      "epoch": 2.387046813964844e-05,
      "step": 99030
    },
    {
      "action_loss_avg": 0.0026574767311103643,
      "epoch": 2.387046813964844e-05,
      "step": 99030
    },
    {
      "epoch": 2.387046813964844e-05,
      "step": 99030,
      "training_step_time": 6.934991359710693
    },
    {
      "epoch": 2.3871421813964843e-05,
      "model_forward_time": 1.7394866943359375,
      "step": 99031
    },
    {
      "epoch": 2.3871421813964843e-05,
      "step": 99031,
      "training_step_time": 6.907045602798462
    },
    {
      "epoch": 2.387237548828125e-05,
      "model_forward_time": 1.740222692489624,
      "step": 99032
    },
    {
      "epoch": 2.387237548828125e-05,
      "step": 99032,
      "training_step_time": 6.94509220123291
    },
    {
      "epoch": 2.3873329162597657e-05,
      "model_forward_time": 1.7398724555969238,
      "step": 99033
    },
    {
      "epoch": 2.3873329162597657e-05,
      "step": 99033,
      "training_step_time": 6.8935463428497314
    },
    {
      "epoch": 2.387428283691406e-05,
      "model_forward_time": 1.7404398918151855,
      "step": 99034
    },
    {
      "epoch": 2.387428283691406e-05,
      "step": 99034,
      "training_step_time": 6.934168815612793
    },
    {
      "epoch": 2.387523651123047e-05,
      "model_forward_time": 2.3719899654388428,
      "step": 99035
    },
    {
      "epoch": 2.387523651123047e-05,
      "step": 99035,
      "training_step_time": 9.621202945709229
    },
    {
      "epoch": 2.3876190185546876e-05,
      "model_forward_time": 1.736030101776123,
      "step": 99036
    },
    {
      "epoch": 2.3876190185546876e-05,
      "step": 99036,
      "training_step_time": 9.934953451156616
    },
    {
      "epoch": 2.387714385986328e-05,
      "model_forward_time": 1.7454922199249268,
      "step": 99037
    },
    {
      "epoch": 2.387714385986328e-05,
      "step": 99037,
      "training_step_time": 7.789978265762329
    },
    {
      "epoch": 2.3878097534179688e-05,
      "model_forward_time": 1.7380425930023193,
      "step": 99038
    },
    {
      "epoch": 2.3878097534179688e-05,
      "step": 99038,
      "training_step_time": 6.956551551818848
    },
    {
      "epoch": 2.3879051208496095e-05,
      "model_forward_time": 1.7454402446746826,
      "step": 99039
    },
    {
      "epoch": 2.3879051208496095e-05,
      "step": 99039,
      "training_step_time": 6.909140110015869
    },
    {
      "epoch": 2.38800048828125e-05,
      "grad_norm": 0.08081384003162384,
      "learning_rate": 2.524659232593929e-09,
      "loss": 0.0547,
      "step": 99040
    },
    {
      "epoch": 2.38800048828125e-05,
      "model_forward_time": 1.7363674640655518,
      "step": 99040
    },
    {
      "dynamics_loss_avg": 0.04870801959186792,
      "epoch": 2.38800048828125e-05,
      "step": 99040
    },
    {
      "action_loss_avg": 0.0011655664449790493,
      "epoch": 2.38800048828125e-05,
      "step": 99040
    },
    {
      "epoch": 2.38800048828125e-05,
      "step": 99040,
      "training_step_time": 6.937634706497192
    },
    {
      "epoch": 2.3880958557128907e-05,
      "model_forward_time": 1.735842227935791,
      "step": 99041
    },
    {
      "epoch": 2.3880958557128907e-05,
      "step": 99041,
      "training_step_time": 6.971489429473877
    },
    {
      "epoch": 2.3881912231445314e-05,
      "model_forward_time": 1.7389204502105713,
      "step": 99042
    },
    {
      "epoch": 2.3881912231445314e-05,
      "step": 99042,
      "training_step_time": 6.7484331130981445
    },
    {
      "epoch": 2.3882865905761718e-05,
      "model_forward_time": 1.739680290222168,
      "step": 99043
    },
    {
      "epoch": 2.3882865905761718e-05,
      "step": 99043,
      "training_step_time": 6.954335451126099
    },
    {
      "epoch": 2.3883819580078125e-05,
      "model_forward_time": 2.384352445602417,
      "step": 99044
    },
    {
      "epoch": 2.3883819580078125e-05,
      "step": 99044,
      "training_step_time": 8.812793970108032
    },
    {
      "epoch": 2.3884773254394533e-05,
      "model_forward_time": 1.7368521690368652,
      "step": 99045
    },
    {
      "epoch": 2.3884773254394533e-05,
      "step": 99045,
      "training_step_time": 11.618024826049805
    },
    {
      "epoch": 2.3885726928710937e-05,
      "model_forward_time": 1.7385284900665283,
      "step": 99046
    },
    {
      "epoch": 2.3885726928710937e-05,
      "step": 99046,
      "training_step_time": 7.813517808914185
    },
    {
      "epoch": 2.3886680603027344e-05,
      "model_forward_time": 1.7555389404296875,
      "step": 99047
    },
    {
      "epoch": 2.3886680603027344e-05,
      "step": 99047,
      "training_step_time": 7.1463093757629395
    },
    {
      "epoch": 2.388763427734375e-05,
      "model_forward_time": 1.7397880554199219,
      "step": 99048
    },
    {
      "epoch": 2.388763427734375e-05,
      "step": 99048,
      "training_step_time": 6.927714824676514
    },
    {
      "epoch": 2.3888587951660156e-05,
      "model_forward_time": 1.7428138256072998,
      "step": 99049
    },
    {
      "epoch": 2.3888587951660156e-05,
      "step": 99049,
      "training_step_time": 6.789599180221558
    },
    {
      "epoch": 2.3889541625976563e-05,
      "grad_norm": 0.10128574818372726,
      "learning_rate": 2.472394578163506e-09,
      "loss": 0.0549,
      "step": 99050
    },
    {
      "epoch": 2.3889541625976563e-05,
      "model_forward_time": 1.7361133098602295,
      "step": 99050
    },
    {
      "dynamics_loss_avg": 0.0528947101905942,
      "epoch": 2.3889541625976563e-05,
      "step": 99050
    },
    {
      "action_loss_avg": 0.002029132060124539,
      "epoch": 2.3889541625976563e-05,
      "step": 99050
    },
    {
      "epoch": 2.3889541625976563e-05,
      "step": 99050,
      "training_step_time": 7.00074315071106
    },
    {
      "epoch": 2.3890495300292967e-05,
      "model_forward_time": 1.7407286167144775,
      "step": 99051
    },
    {
      "epoch": 2.3890495300292967e-05,
      "step": 99051,
      "training_step_time": 6.933787822723389
    },
    {
      "epoch": 2.3891448974609375e-05,
      "model_forward_time": 1.737497091293335,
      "step": 99052
    },
    {
      "epoch": 2.3891448974609375e-05,
      "step": 99052,
      "training_step_time": 6.949041366577148
    },
    {
      "epoch": 2.3892402648925782e-05,
      "model_forward_time": 2.385979652404785,
      "step": 99053
    },
    {
      "epoch": 2.3892402648925782e-05,
      "step": 99053,
      "training_step_time": 8.570462703704834
    },
    {
      "epoch": 2.3893356323242186e-05,
      "model_forward_time": 1.7393443584442139,
      "step": 99054
    },
    {
      "epoch": 2.3893356323242186e-05,
      "step": 99054,
      "training_step_time": 10.945078134536743
    },
    {
      "epoch": 2.3894309997558593e-05,
      "model_forward_time": 1.7423412799835205,
      "step": 99055
    },
    {
      "epoch": 2.3894309997558593e-05,
      "step": 99055,
      "training_step_time": 7.889114856719971
    },
    {
      "epoch": 2.3895263671875e-05,
      "model_forward_time": 1.7401800155639648,
      "step": 99056
    },
    {
      "epoch": 2.3895263671875e-05,
      "step": 99056,
      "training_step_time": 7.220642566680908
    },
    {
      "epoch": 2.3896217346191405e-05,
      "model_forward_time": 1.758164405822754,
      "step": 99057
    },
    {
      "epoch": 2.3896217346191405e-05,
      "step": 99057,
      "training_step_time": 6.97461724281311
    },
    {
      "epoch": 2.3897171020507812e-05,
      "model_forward_time": 1.741222858428955,
      "step": 99058
    },
    {
      "epoch": 2.3897171020507812e-05,
      "step": 99058,
      "training_step_time": 6.971635103225708
    },
    {
      "epoch": 2.389812469482422e-05,
      "model_forward_time": 1.7437443733215332,
      "step": 99059
    },
    {
      "epoch": 2.389812469482422e-05,
      "step": 99059,
      "training_step_time": 6.9550087451934814
    },
    {
      "epoch": 2.3899078369140624e-05,
      "grad_norm": 0.09469178318977356,
      "learning_rate": 2.4206764458378774e-09,
      "loss": 0.0549,
      "step": 99060
    },
    {
      "epoch": 2.3899078369140624e-05,
      "model_forward_time": 1.7410173416137695,
      "step": 99060
    },
    {
      "dynamics_loss_avg": 0.05471252128481865,
      "epoch": 2.3899078369140624e-05,
      "step": 99060
    },
    {
      "action_loss_avg": 0.0017473284096922725,
      "epoch": 2.3899078369140624e-05,
      "step": 99060
    },
    {
      "epoch": 2.3899078369140624e-05,
      "step": 99060,
      "training_step_time": 6.899518728256226
    },
    {
      "epoch": 2.390003204345703e-05,
      "model_forward_time": 1.7429673671722412,
      "step": 99061
    },
    {
      "epoch": 2.390003204345703e-05,
      "step": 99061,
      "training_step_time": 6.957904815673828
    },
    {
      "epoch": 2.390098571777344e-05,
      "model_forward_time": 2.35229754447937,
      "step": 99062
    },
    {
      "epoch": 2.390098571777344e-05,
      "step": 99062,
      "training_step_time": 9.672840595245361
    },
    {
      "epoch": 2.3901939392089843e-05,
      "model_forward_time": 1.7363202571868896,
      "step": 99063
    },
    {
      "epoch": 2.3901939392089843e-05,
      "step": 99063,
      "training_step_time": 11.49818730354309
    },
    {
      "epoch": 2.390289306640625e-05,
      "model_forward_time": 1.371945858001709,
      "step": 99064
    },
    {
      "epoch": 2.390289306640625e-05,
      "step": 99064,
      "training_step_time": 8.023802757263184
    },
    {
      "epoch": 2.3903846740722657e-05,
      "model_forward_time": 1.7383744716644287,
      "step": 99065
    },
    {
      "epoch": 2.3903846740722657e-05,
      "step": 99065,
      "training_step_time": 7.711250305175781
    },
    {
      "epoch": 2.390480041503906e-05,
      "model_forward_time": 1.7439699172973633,
      "step": 99066
    },
    {
      "epoch": 2.390480041503906e-05,
      "step": 99066,
      "training_step_time": 6.92052960395813
    },
    {
      "epoch": 2.390575408935547e-05,
      "model_forward_time": 1.738386869430542,
      "step": 99067
    },
    {
      "epoch": 2.390575408935547e-05,
      "step": 99067,
      "training_step_time": 6.958219051361084
    },
    {
      "epoch": 2.3906707763671876e-05,
      "model_forward_time": 1.7417628765106201,
      "step": 99068
    },
    {
      "epoch": 2.3906707763671876e-05,
      "step": 99068,
      "training_step_time": 6.953006982803345
    },
    {
      "epoch": 2.390766143798828e-05,
      "model_forward_time": 1.7433087825775146,
      "step": 99069
    },
    {
      "epoch": 2.390766143798828e-05,
      "step": 99069,
      "training_step_time": 6.9672932624816895
    },
    {
      "epoch": 2.3908615112304688e-05,
      "grad_norm": 0.11025458574295044,
      "learning_rate": 2.369504841273629e-09,
      "loss": 0.0551,
      "step": 99070
    },
    {
      "epoch": 2.3908615112304688e-05,
      "model_forward_time": 1.7427349090576172,
      "step": 99070
    },
    {
      "dynamics_loss_avg": 0.05969812506809831,
      "epoch": 2.3908615112304688e-05,
      "step": 99070
    },
    {
      "action_loss_avg": 0.0014785859995754436,
      "epoch": 2.3908615112304688e-05,
      "step": 99070
    },
    {
      "epoch": 2.3908615112304688e-05,
      "step": 99070,
      "training_step_time": 6.9707865715026855
    },
    {
      "epoch": 2.3909568786621095e-05,
      "model_forward_time": 1.744239330291748,
      "step": 99071
    },
    {
      "epoch": 2.3909568786621095e-05,
      "step": 99071,
      "training_step_time": 7.80080509185791
    },
    {
      "epoch": 2.39105224609375e-05,
      "model_forward_time": 2.364132881164551,
      "step": 99072
    },
    {
      "epoch": 2.39105224609375e-05,
      "step": 99072,
      "training_step_time": 10.55667495727539
    },
    {
      "epoch": 2.3911476135253907e-05,
      "model_forward_time": 1.7404232025146484,
      "step": 99073
    },
    {
      "epoch": 2.3911476135253907e-05,
      "step": 99073,
      "training_step_time": 9.103678464889526
    },
    {
      "epoch": 2.3912429809570314e-05,
      "model_forward_time": 1.738152265548706,
      "step": 99074
    },
    {
      "epoch": 2.3912429809570314e-05,
      "step": 99074,
      "training_step_time": 7.752137660980225
    },
    {
      "epoch": 2.3913383483886718e-05,
      "model_forward_time": 1.7402598857879639,
      "step": 99075
    },
    {
      "epoch": 2.3913383483886718e-05,
      "step": 99075,
      "training_step_time": 6.810511827468872
    },
    {
      "epoch": 2.3914337158203125e-05,
      "model_forward_time": 1.7438600063323975,
      "step": 99076
    },
    {
      "epoch": 2.3914337158203125e-05,
      "step": 99076,
      "training_step_time": 6.8840601444244385
    },
    {
      "epoch": 2.3915290832519533e-05,
      "model_forward_time": 1.752490758895874,
      "step": 99077
    },
    {
      "epoch": 2.3915290832519533e-05,
      "step": 99077,
      "training_step_time": 6.903134346008301
    },
    {
      "epoch": 2.3916244506835937e-05,
      "model_forward_time": 1.7413911819458008,
      "step": 99078
    },
    {
      "epoch": 2.3916244506835937e-05,
      "step": 99078,
      "training_step_time": 6.746997356414795
    },
    {
      "epoch": 2.3917198181152344e-05,
      "model_forward_time": 1.749513864517212,
      "step": 99079
    },
    {
      "epoch": 2.3917198181152344e-05,
      "step": 99079,
      "training_step_time": 6.75914454460144
    },
    {
      "epoch": 2.391815185546875e-05,
      "grad_norm": 0.0774834007024765,
      "learning_rate": 2.3188797700668397e-09,
      "loss": 0.0546,
      "step": 99080
    },
    {
      "epoch": 2.391815185546875e-05,
      "model_forward_time": 1.7425806522369385,
      "step": 99080
    },
    {
      "dynamics_loss_avg": 0.05596708636730909,
      "epoch": 2.391815185546875e-05,
      "step": 99080
    },
    {
      "action_loss_avg": 0.001508399358135648,
      "epoch": 2.391815185546875e-05,
      "step": 99080
    },
    {
      "epoch": 2.391815185546875e-05,
      "step": 99080,
      "training_step_time": 6.723850965499878
    },
    {
      "epoch": 2.3919105529785156e-05,
      "model_forward_time": 2.397585868835449,
      "step": 99081
    },
    {
      "epoch": 2.3919105529785156e-05,
      "step": 99081,
      "training_step_time": 9.998760461807251
    },
    {
      "epoch": 2.3920059204101563e-05,
      "model_forward_time": 1.7430040836334229,
      "step": 99082
    },
    {
      "epoch": 2.3920059204101563e-05,
      "step": 99082,
      "training_step_time": 9.460142850875854
    },
    {
      "epoch": 2.3921012878417967e-05,
      "model_forward_time": 1.7378191947937012,
      "step": 99083
    },
    {
      "epoch": 2.3921012878417967e-05,
      "step": 99083,
      "training_step_time": 8.17781400680542
    },
    {
      "epoch": 2.3921966552734375e-05,
      "model_forward_time": 1.7376863956451416,
      "step": 99084
    },
    {
      "epoch": 2.3921966552734375e-05,
      "step": 99084,
      "training_step_time": 6.725178480148315
    },
    {
      "epoch": 2.3922920227050782e-05,
      "model_forward_time": 1.7390086650848389,
      "step": 99085
    },
    {
      "epoch": 2.3922920227050782e-05,
      "step": 99085,
      "training_step_time": 6.778583526611328
    },
    {
      "epoch": 2.3923873901367186e-05,
      "model_forward_time": 1.7386465072631836,
      "step": 99086
    },
    {
      "epoch": 2.3923873901367186e-05,
      "step": 99086,
      "training_step_time": 6.761655330657959
    },
    {
      "epoch": 2.3924827575683594e-05,
      "model_forward_time": 1.7368957996368408,
      "step": 99087
    },
    {
      "epoch": 2.3924827575683594e-05,
      "step": 99087,
      "training_step_time": 6.76372766494751
    },
    {
      "epoch": 2.392578125e-05,
      "model_forward_time": 1.7420377731323242,
      "step": 99088
    },
    {
      "epoch": 2.392578125e-05,
      "step": 99088,
      "training_step_time": 6.764744997024536
    },
    {
      "epoch": 2.3926734924316405e-05,
      "model_forward_time": 1.7389159202575684,
      "step": 99089
    },
    {
      "epoch": 2.3926734924316405e-05,
      "step": 99089,
      "training_step_time": 6.76717209815979
    },
    {
      "epoch": 2.3927688598632812e-05,
      "grad_norm": 0.0844387486577034,
      "learning_rate": 2.268801237753082e-09,
      "loss": 0.0541,
      "step": 99090
    },
    {
      "epoch": 2.3927688598632812e-05,
      "model_forward_time": 2.3935439586639404,
      "step": 99090
    },
    {
      "dynamics_loss_avg": 0.05144101632758975,
      "epoch": 2.3927688598632812e-05,
      "step": 99090
    },
    {
      "action_loss_avg": 0.00136807196540758,
      "epoch": 2.3927688598632812e-05,
      "step": 99090
    },
    {
      "epoch": 2.3927688598632812e-05,
      "step": 99090,
      "training_step_time": 7.692663669586182
    },
    {
      "epoch": 2.392864227294922e-05,
      "model_forward_time": 1.7413702011108398,
      "step": 99091
    },
    {
      "epoch": 2.392864227294922e-05,
      "step": 99091,
      "training_step_time": 9.974551439285278
    },
    {
      "epoch": 2.3929595947265624e-05,
      "model_forward_time": 1.7419054508209229,
      "step": 99092
    },
    {
      "epoch": 2.3929595947265624e-05,
      "step": 99092,
      "training_step_time": 7.710247278213501
    },
    {
      "epoch": 2.393054962158203e-05,
      "model_forward_time": 1.7342925071716309,
      "step": 99093
    },
    {
      "epoch": 2.393054962158203e-05,
      "step": 99093,
      "training_step_time": 7.375507116317749
    },
    {
      "epoch": 2.393150329589844e-05,
      "model_forward_time": 1.7404303550720215,
      "step": 99094
    },
    {
      "epoch": 2.393150329589844e-05,
      "step": 99094,
      "training_step_time": 6.801802158355713
    },
    {
      "epoch": 2.3932456970214843e-05,
      "model_forward_time": 1.7772259712219238,
      "step": 99095
    },
    {
      "epoch": 2.3932456970214843e-05,
      "step": 99095,
      "training_step_time": 6.816929578781128
    },
    {
      "epoch": 2.393341064453125e-05,
      "model_forward_time": 1.738049030303955,
      "step": 99096
    },
    {
      "epoch": 2.393341064453125e-05,
      "step": 99096,
      "training_step_time": 6.77996826171875
    },
    {
      "epoch": 2.3934364318847657e-05,
      "model_forward_time": 1.7407901287078857,
      "step": 99097
    },
    {
      "epoch": 2.3934364318847657e-05,
      "step": 99097,
      "training_step_time": 6.799809455871582
    },
    {
      "epoch": 2.393531799316406e-05,
      "model_forward_time": 1.7772603034973145,
      "step": 99098
    },
    {
      "epoch": 2.393531799316406e-05,
      "step": 99098,
      "training_step_time": 6.776306390762329
    },
    {
      "epoch": 2.393627166748047e-05,
      "model_forward_time": 2.413321018218994,
      "step": 99099
    },
    {
      "epoch": 2.393627166748047e-05,
      "step": 99099,
      "training_step_time": 8.225759506225586
    },
    {
      "epoch": 2.3937225341796876e-05,
      "grad_norm": 0.07516393065452576,
      "learning_rate": 2.2192692498090864e-09,
      "loss": 0.0562,
      "step": 99100
    },
    {
      "epoch": 2.3937225341796876e-05,
      "model_forward_time": 1.8080708980560303,
      "step": 99100
    },
    {
      "dynamics_loss_avg": 0.03853352721780538,
      "epoch": 2.3937225341796876e-05,
      "step": 99100
    },
    {
      "action_loss_avg": 0.0017824347130954265,
      "epoch": 2.3937225341796876e-05,
      "step": 99100
    },
    {
      "epoch": 2.3937225341796876e-05,
      "step": 99100,
      "training_step_time": 10.435434341430664
    },
    {
      "epoch": 2.393817901611328e-05,
      "model_forward_time": 1.7321641445159912,
      "step": 99101
    },
    {
      "epoch": 2.393817901611328e-05,
      "step": 99101,
      "training_step_time": 7.632999658584595
    },
    {
      "epoch": 2.3939132690429688e-05,
      "model_forward_time": 1.7395906448364258,
      "step": 99102
    },
    {
      "epoch": 2.3939132690429688e-05,
      "step": 99102,
      "training_step_time": 7.421020030975342
    },
    {
      "epoch": 2.3940086364746095e-05,
      "model_forward_time": 1.7430884838104248,
      "step": 99103
    },
    {
      "epoch": 2.3940086364746095e-05,
      "step": 99103,
      "training_step_time": 6.783045530319214
    },
    {
      "epoch": 2.39410400390625e-05,
      "model_forward_time": 1.7411353588104248,
      "step": 99104
    },
    {
      "epoch": 2.39410400390625e-05,
      "step": 99104,
      "training_step_time": 6.7817137241363525
    },
    {
      "epoch": 2.3941993713378907e-05,
      "model_forward_time": 1.7370257377624512,
      "step": 99105
    },
    {
      "epoch": 2.3941993713378907e-05,
      "step": 99105,
      "training_step_time": 6.809270858764648
    },
    {
      "epoch": 2.3942947387695314e-05,
      "model_forward_time": 1.7532846927642822,
      "step": 99106
    },
    {
      "epoch": 2.3942947387695314e-05,
      "step": 99106,
      "training_step_time": 6.872254133224487
    },
    {
      "epoch": 2.3943901062011718e-05,
      "model_forward_time": 1.7328648567199707,
      "step": 99107
    },
    {
      "epoch": 2.3943901062011718e-05,
      "step": 99107,
      "training_step_time": 6.859824180603027
    },
    {
      "epoch": 2.3944854736328126e-05,
      "model_forward_time": 1.7389390468597412,
      "step": 99108
    },
    {
      "epoch": 2.3944854736328126e-05,
      "step": 99108,
      "training_step_time": 9.047622680664062
    },
    {
      "epoch": 2.3945808410644533e-05,
      "model_forward_time": 1.7912249565124512,
      "step": 99109
    },
    {
      "epoch": 2.3945808410644533e-05,
      "step": 99109,
      "training_step_time": 10.199102401733398
    },
    {
      "epoch": 2.3946762084960937e-05,
      "grad_norm": 0.10495822131633759,
      "learning_rate": 2.1702838116521852e-09,
      "loss": 0.0556,
      "step": 99110
    },
    {
      "epoch": 2.3946762084960937e-05,
      "model_forward_time": 1.7376062870025635,
      "step": 99110
    },
    {
      "dynamics_loss_avg": 0.06049170847982168,
      "epoch": 2.3946762084960937e-05,
      "step": 99110
    },
    {
      "action_loss_avg": 0.0032173046929528937,
      "epoch": 2.3946762084960937e-05,
      "step": 99110
    },
    {
      "epoch": 2.3946762084960937e-05,
      "step": 99110,
      "training_step_time": 8.019004583358765
    },
    {
      "epoch": 2.3947715759277344e-05,
      "model_forward_time": 1.7425267696380615,
      "step": 99111
    },
    {
      "epoch": 2.3947715759277344e-05,
      "step": 99111,
      "training_step_time": 7.761329412460327
    },
    {
      "epoch": 2.394866943359375e-05,
      "model_forward_time": 1.7497107982635498,
      "step": 99112
    },
    {
      "epoch": 2.394866943359375e-05,
      "step": 99112,
      "training_step_time": 7.036070108413696
    },
    {
      "epoch": 2.3949623107910156e-05,
      "model_forward_time": 1.7449076175689697,
      "step": 99113
    },
    {
      "epoch": 2.3949623107910156e-05,
      "step": 99113,
      "training_step_time": 7.01980185508728
    },
    {
      "epoch": 2.3950576782226563e-05,
      "model_forward_time": 1.7555596828460693,
      "step": 99114
    },
    {
      "epoch": 2.3950576782226563e-05,
      "step": 99114,
      "training_step_time": 7.021880626678467
    },
    {
      "epoch": 2.3951530456542967e-05,
      "model_forward_time": 1.7518134117126465,
      "step": 99115
    },
    {
      "epoch": 2.3951530456542967e-05,
      "step": 99115,
      "training_step_time": 7.044349193572998
    },
    {
      "epoch": 2.3952484130859375e-05,
      "model_forward_time": 1.7610507011413574,
      "step": 99116
    },
    {
      "epoch": 2.3952484130859375e-05,
      "step": 99116,
      "training_step_time": 6.803315877914429
    },
    {
      "epoch": 2.3953437805175782e-05,
      "model_forward_time": 1.7515459060668945,
      "step": 99117
    },
    {
      "epoch": 2.3953437805175782e-05,
      "step": 99117,
      "training_step_time": 7.825234651565552
    },
    {
      "epoch": 2.3954391479492186e-05,
      "model_forward_time": 2.45276141166687,
      "step": 99118
    },
    {
      "epoch": 2.3954391479492186e-05,
      "step": 99118,
      "training_step_time": 8.51215410232544
    },
    {
      "epoch": 2.3955345153808594e-05,
      "model_forward_time": 1.753371238708496,
      "step": 99119
    },
    {
      "epoch": 2.3955345153808594e-05,
      "step": 99119,
      "training_step_time": 10.834187269210815
    },
    {
      "epoch": 2.3956298828125e-05,
      "grad_norm": 0.09304345399141312,
      "learning_rate": 2.12184492863865e-09,
      "loss": 0.0552,
      "step": 99120
    },
    {
      "epoch": 2.3956298828125e-05,
      "model_forward_time": 1.7409696578979492,
      "step": 99120
    },
    {
      "dynamics_loss_avg": 0.0628182414919138,
      "epoch": 2.3956298828125e-05,
      "step": 99120
    },
    {
      "action_loss_avg": 0.002447611029492691,
      "epoch": 2.3956298828125e-05,
      "step": 99120
    },
    {
      "epoch": 2.3956298828125e-05,
      "step": 99120,
      "training_step_time": 7.737691640853882
    },
    {
      "epoch": 2.3957252502441405e-05,
      "model_forward_time": 1.744053840637207,
      "step": 99121
    },
    {
      "epoch": 2.3957252502441405e-05,
      "step": 99121,
      "training_step_time": 6.9597649574279785
    },
    {
      "epoch": 2.3958206176757812e-05,
      "model_forward_time": 1.7788794040679932,
      "step": 99122
    },
    {
      "epoch": 2.3958206176757812e-05,
      "step": 99122,
      "training_step_time": 6.955729246139526
    },
    {
      "epoch": 2.395915985107422e-05,
      "model_forward_time": 1.7532289028167725,
      "step": 99123
    },
    {
      "epoch": 2.395915985107422e-05,
      "step": 99123,
      "training_step_time": 6.952997922897339
    },
    {
      "epoch": 2.3960113525390624e-05,
      "model_forward_time": 1.7494239807128906,
      "step": 99124
    },
    {
      "epoch": 2.3960113525390624e-05,
      "step": 99124,
      "training_step_time": 6.996918678283691
    },
    {
      "epoch": 2.396106719970703e-05,
      "model_forward_time": 1.7729556560516357,
      "step": 99125
    },
    {
      "epoch": 2.396106719970703e-05,
      "step": 99125,
      "training_step_time": 6.962002754211426
    },
    {
      "epoch": 2.396202087402344e-05,
      "model_forward_time": 1.773193120956421,
      "step": 99126
    },
    {
      "epoch": 2.396202087402344e-05,
      "step": 99126,
      "training_step_time": 7.7751784324646
    },
    {
      "epoch": 2.3962974548339843e-05,
      "model_forward_time": 2.4640440940856934,
      "step": 99127
    },
    {
      "epoch": 2.3962974548339843e-05,
      "step": 99127,
      "training_step_time": 8.71414566040039
    },
    {
      "epoch": 2.396392822265625e-05,
      "model_forward_time": 1.7694883346557617,
      "step": 99128
    },
    {
      "epoch": 2.396392822265625e-05,
      "step": 99128,
      "training_step_time": 10.356678247451782
    },
    {
      "epoch": 2.3964881896972658e-05,
      "model_forward_time": 1.7658450603485107,
      "step": 99129
    },
    {
      "epoch": 2.3964881896972658e-05,
      "step": 99129,
      "training_step_time": 7.786420583724976
    },
    {
      "epoch": 2.396583557128906e-05,
      "grad_norm": 0.11646158993244171,
      "learning_rate": 2.073952606066465e-09,
      "loss": 0.057,
      "step": 99130
    },
    {
      "epoch": 2.396583557128906e-05,
      "model_forward_time": 1.775446891784668,
      "step": 99130
    },
    {
      "dynamics_loss_avg": 0.0501647837460041,
      "epoch": 2.396583557128906e-05,
      "step": 99130
    },
    {
      "action_loss_avg": 0.0025169939297484236,
      "epoch": 2.396583557128906e-05,
      "step": 99130
    },
    {
      "epoch": 2.396583557128906e-05,
      "step": 99130,
      "training_step_time": 7.5280468463897705
    },
    {
      "epoch": 2.396678924560547e-05,
      "model_forward_time": 1.765345811843872,
      "step": 99131
    },
    {
      "epoch": 2.396678924560547e-05,
      "step": 99131,
      "training_step_time": 6.785488128662109
    },
    {
      "epoch": 2.3967742919921876e-05,
      "model_forward_time": 1.7663533687591553,
      "step": 99132
    },
    {
      "epoch": 2.3967742919921876e-05,
      "step": 99132,
      "training_step_time": 6.921378135681152
    },
    {
      "epoch": 2.396869659423828e-05,
      "model_forward_time": 1.758998155593872,
      "step": 99133
    },
    {
      "epoch": 2.396869659423828e-05,
      "step": 99133,
      "training_step_time": 6.977783679962158
    },
    {
      "epoch": 2.3969650268554688e-05,
      "model_forward_time": 1.7672028541564941,
      "step": 99134
    },
    {
      "epoch": 2.3969650268554688e-05,
      "step": 99134,
      "training_step_time": 6.952834367752075
    },
    {
      "epoch": 2.3970603942871095e-05,
      "model_forward_time": 1.7650642395019531,
      "step": 99135
    },
    {
      "epoch": 2.3970603942871095e-05,
      "step": 99135,
      "training_step_time": 8.675943851470947
    },
    {
      "epoch": 2.39715576171875e-05,
      "model_forward_time": 2.4086625576019287,
      "step": 99136
    },
    {
      "epoch": 2.39715576171875e-05,
      "step": 99136,
      "training_step_time": 7.9843385219573975
    },
    {
      "epoch": 2.3972511291503907e-05,
      "model_forward_time": 1.8264095783233643,
      "step": 99137
    },
    {
      "epoch": 2.3972511291503907e-05,
      "step": 99137,
      "training_step_time": 11.472632646560669
    },
    {
      "epoch": 2.3973464965820314e-05,
      "model_forward_time": 1.7630882263183594,
      "step": 99138
    },
    {
      "epoch": 2.3973464965820314e-05,
      "step": 99138,
      "training_step_time": 8.854973316192627
    },
    {
      "epoch": 2.3974418640136718e-05,
      "model_forward_time": 1.7594850063323975,
      "step": 99139
    },
    {
      "epoch": 2.3974418640136718e-05,
      "step": 99139,
      "training_step_time": 7.542034864425659
    },
    {
      "epoch": 2.3975372314453126e-05,
      "grad_norm": 0.07831183075904846,
      "learning_rate": 2.026606849171997e-09,
      "loss": 0.0558,
      "step": 99140
    },
    {
      "epoch": 2.3975372314453126e-05,
      "model_forward_time": 1.7657561302185059,
      "step": 99140
    },
    {
      "dynamics_loss_avg": 0.06147650238126516,
      "epoch": 2.3975372314453126e-05,
      "step": 99140
    },
    {
      "action_loss_avg": 0.002578964689746499,
      "epoch": 2.3975372314453126e-05,
      "step": 99140
    },
    {
      "epoch": 2.3975372314453126e-05,
      "step": 99140,
      "training_step_time": 6.957960844039917
    },
    {
      "epoch": 2.397632598876953e-05,
      "model_forward_time": 1.7668359279632568,
      "step": 99141
    },
    {
      "epoch": 2.397632598876953e-05,
      "step": 99141,
      "training_step_time": 7.014057397842407
    },
    {
      "epoch": 2.3977279663085937e-05,
      "model_forward_time": 1.7604758739471436,
      "step": 99142
    },
    {
      "epoch": 2.3977279663085937e-05,
      "step": 99142,
      "training_step_time": 6.950956344604492
    },
    {
      "epoch": 2.3978233337402344e-05,
      "model_forward_time": 1.7712581157684326,
      "step": 99143
    },
    {
      "epoch": 2.3978233337402344e-05,
      "step": 99143,
      "training_step_time": 6.9324421882629395
    },
    {
      "epoch": 2.397918701171875e-05,
      "model_forward_time": 1.769296407699585,
      "step": 99144
    },
    {
      "epoch": 2.397918701171875e-05,
      "step": 99144,
      "training_step_time": 7.820342063903809
    },
    {
      "epoch": 2.3980140686035156e-05,
      "model_forward_time": 2.407461404800415,
      "step": 99145
    },
    {
      "epoch": 2.3980140686035156e-05,
      "step": 99145,
      "training_step_time": 8.69041895866394
    },
    {
      "epoch": 2.3981094360351563e-05,
      "model_forward_time": 1.838475227355957,
      "step": 99146
    },
    {
      "epoch": 2.3981094360351563e-05,
      "step": 99146,
      "training_step_time": 9.4253671169281
    },
    {
      "epoch": 2.3982048034667967e-05,
      "model_forward_time": 1.7658286094665527,
      "step": 99147
    },
    {
      "epoch": 2.3982048034667967e-05,
      "step": 99147,
      "training_step_time": 8.950184106826782
    },
    {
      "epoch": 2.3983001708984375e-05,
      "model_forward_time": 1.764535665512085,
      "step": 99148
    },
    {
      "epoch": 2.3983001708984375e-05,
      "step": 99148,
      "training_step_time": 7.824163913726807
    },
    {
      "epoch": 2.3983955383300782e-05,
      "model_forward_time": 1.7690463066101074,
      "step": 99149
    },
    {
      "epoch": 2.3983955383300782e-05,
      "step": 99149,
      "training_step_time": 6.9705729484558105
    },
    {
      "epoch": 2.3984909057617186e-05,
      "grad_norm": 0.08050751686096191,
      "learning_rate": 1.979807663133326e-09,
      "loss": 0.0551,
      "step": 99150
    },
    {
      "epoch": 2.3984909057617186e-05,
      "model_forward_time": 1.7720131874084473,
      "step": 99150
    },
    {
      "dynamics_loss_avg": 0.05468033738434315,
      "epoch": 2.3984909057617186e-05,
      "step": 99150
    },
    {
      "action_loss_avg": 0.0015977182658389211,
      "epoch": 2.3984909057617186e-05,
      "step": 99150
    },
    {
      "epoch": 2.3984909057617186e-05,
      "step": 99150,
      "training_step_time": 6.921393156051636
    },
    {
      "epoch": 2.3985862731933594e-05,
      "model_forward_time": 1.7691807746887207,
      "step": 99151
    },
    {
      "epoch": 2.3985862731933594e-05,
      "step": 99151,
      "training_step_time": 6.91966700553894
    },
    {
      "epoch": 2.398681640625e-05,
      "model_forward_time": 1.3971974849700928,
      "step": 99152
    },
    {
      "epoch": 2.398681640625e-05,
      "step": 99152,
      "training_step_time": 6.781509637832642
    },
    {
      "epoch": 2.3987770080566405e-05,
      "model_forward_time": 1.7650039196014404,
      "step": 99153
    },
    {
      "epoch": 2.3987770080566405e-05,
      "step": 99153,
      "training_step_time": 7.7768003940582275
    },
    {
      "epoch": 2.3988723754882812e-05,
      "model_forward_time": 1.7648766040802002,
      "step": 99154
    },
    {
      "epoch": 2.3988723754882812e-05,
      "step": 99154,
      "training_step_time": 8.68600058555603
    },
    {
      "epoch": 2.398967742919922e-05,
      "model_forward_time": 1.8285813331604004,
      "step": 99155
    },
    {
      "epoch": 2.398967742919922e-05,
      "step": 99155,
      "training_step_time": 11.672648668289185
    },
    {
      "epoch": 2.3990631103515624e-05,
      "model_forward_time": 1.7636539936065674,
      "step": 99156
    },
    {
      "epoch": 2.3990631103515624e-05,
      "step": 99156,
      "training_step_time": 9.747655391693115
    },
    {
      "epoch": 2.399158477783203e-05,
      "model_forward_time": 1.76710844039917,
      "step": 99157
    },
    {
      "epoch": 2.399158477783203e-05,
      "step": 99157,
      "training_step_time": 7.773204326629639
    },
    {
      "epoch": 2.399253845214844e-05,
      "model_forward_time": 1.7629437446594238,
      "step": 99158
    },
    {
      "epoch": 2.399253845214844e-05,
      "step": 99158,
      "training_step_time": 7.378821849822998
    },
    {
      "epoch": 2.3993492126464843e-05,
      "model_forward_time": 1.7618446350097656,
      "step": 99159
    },
    {
      "epoch": 2.3993492126464843e-05,
      "step": 99159,
      "training_step_time": 6.925556898117065
    },
    {
      "epoch": 2.399444580078125e-05,
      "grad_norm": 0.08096791058778763,
      "learning_rate": 1.933555053069136e-09,
      "loss": 0.0545,
      "step": 99160
    },
    {
      "epoch": 2.399444580078125e-05,
      "model_forward_time": 1.762660264968872,
      "step": 99160
    },
    {
      "dynamics_loss_avg": 0.060417532548308374,
      "epoch": 2.399444580078125e-05,
      "step": 99160
    },
    {
      "action_loss_avg": 0.00209112954325974,
      "epoch": 2.399444580078125e-05,
      "step": 99160
    },
    {
      "epoch": 2.399444580078125e-05,
      "step": 99160,
      "training_step_time": 6.888589382171631
    },
    {
      "epoch": 2.3995399475097658e-05,
      "model_forward_time": 1.77815842628479,
      "step": 99161
    },
    {
      "epoch": 2.3995399475097658e-05,
      "step": 99161,
      "training_step_time": 6.6961376667022705
    },
    {
      "epoch": 2.399635314941406e-05,
      "model_forward_time": 1.7679829597473145,
      "step": 99162
    },
    {
      "epoch": 2.399635314941406e-05,
      "step": 99162,
      "training_step_time": 7.731719732284546
    },
    {
      "epoch": 2.399730682373047e-05,
      "model_forward_time": 1.7620313167572021,
      "step": 99163
    },
    {
      "epoch": 2.399730682373047e-05,
      "step": 99163,
      "training_step_time": 7.712080001831055
    },
    {
      "epoch": 2.3998260498046876e-05,
      "model_forward_time": 2.491541624069214,
      "step": 99164
    },
    {
      "epoch": 2.3998260498046876e-05,
      "step": 99164,
      "training_step_time": 7.981058359146118
    },
    {
      "epoch": 2.399921417236328e-05,
      "model_forward_time": 1.7644679546356201,
      "step": 99165
    },
    {
      "epoch": 2.399921417236328e-05,
      "step": 99165,
      "training_step_time": 7.858347177505493
    },
    {
      "epoch": 2.4000167846679688e-05,
      "model_forward_time": 1.744443655014038,
      "step": 99166
    },
    {
      "epoch": 2.4000167846679688e-05,
      "step": 99166,
      "training_step_time": 7.83783483505249
    },
    {
      "epoch": 2.4001121520996095e-05,
      "model_forward_time": 1.772184133529663,
      "step": 99167
    },
    {
      "epoch": 2.4001121520996095e-05,
      "step": 99167,
      "training_step_time": 7.4584503173828125
    },
    {
      "epoch": 2.40020751953125e-05,
      "model_forward_time": 1.7659692764282227,
      "step": 99168
    },
    {
      "epoch": 2.40020751953125e-05,
      "step": 99168,
      "training_step_time": 6.881607532501221
    },
    {
      "epoch": 2.4003028869628907e-05,
      "model_forward_time": 1.7687568664550781,
      "step": 99169
    },
    {
      "epoch": 2.4003028869628907e-05,
      "step": 99169,
      "training_step_time": 6.91419792175293
    },
    {
      "epoch": 2.4003982543945314e-05,
      "grad_norm": 0.09606558084487915,
      "learning_rate": 1.887849024036492e-09,
      "loss": 0.0554,
      "step": 99170
    },
    {
      "epoch": 2.4003982543945314e-05,
      "model_forward_time": 1.7632153034210205,
      "step": 99170
    },
    {
      "dynamics_loss_avg": 0.05053402502089739,
      "epoch": 2.4003982543945314e-05,
      "step": 99170
    },
    {
      "action_loss_avg": 0.0019377023272681981,
      "epoch": 2.4003982543945314e-05,
      "step": 99170
    },
    {
      "epoch": 2.4003982543945314e-05,
      "step": 99170,
      "training_step_time": 6.915103912353516
    },
    {
      "epoch": 2.4004936218261718e-05,
      "model_forward_time": 1.770401954650879,
      "step": 99171
    },
    {
      "epoch": 2.4004936218261718e-05,
      "step": 99171,
      "training_step_time": 7.753103971481323
    },
    {
      "epoch": 2.4005889892578126e-05,
      "model_forward_time": 1.7648241519927979,
      "step": 99172
    },
    {
      "epoch": 2.4005889892578126e-05,
      "step": 99172,
      "training_step_time": 7.745467662811279
    },
    {
      "epoch": 2.400684356689453e-05,
      "model_forward_time": 2.467829942703247,
      "step": 99173
    },
    {
      "epoch": 2.400684356689453e-05,
      "step": 99173,
      "training_step_time": 9.561645984649658
    },
    {
      "epoch": 2.4007797241210937e-05,
      "model_forward_time": 1.7649710178375244,
      "step": 99174
    },
    {
      "epoch": 2.4007797241210937e-05,
      "step": 99174,
      "training_step_time": 10.540710210800171
    },
    {
      "epoch": 2.4008750915527344e-05,
      "model_forward_time": 1.7656688690185547,
      "step": 99175
    },
    {
      "epoch": 2.4008750915527344e-05,
      "step": 99175,
      "training_step_time": 8.68823528289795
    },
    {
      "epoch": 2.400970458984375e-05,
      "model_forward_time": 1.77378511428833,
      "step": 99176
    },
    {
      "epoch": 2.400970458984375e-05,
      "step": 99176,
      "training_step_time": 7.4978344440460205
    },
    {
      "epoch": 2.4010658264160156e-05,
      "model_forward_time": 1.7652125358581543,
      "step": 99177
    },
    {
      "epoch": 2.4010658264160156e-05,
      "step": 99177,
      "training_step_time": 6.97778582572937
    },
    {
      "epoch": 2.4011611938476563e-05,
      "model_forward_time": 1.767651081085205,
      "step": 99178
    },
    {
      "epoch": 2.4011611938476563e-05,
      "step": 99178,
      "training_step_time": 6.965183973312378
    },
    {
      "epoch": 2.4012565612792967e-05,
      "model_forward_time": 1.7634434700012207,
      "step": 99179
    },
    {
      "epoch": 2.4012565612792967e-05,
      "step": 99179,
      "training_step_time": 6.944085359573364
    },
    {
      "epoch": 2.4013519287109375e-05,
      "grad_norm": 0.14807242155075073,
      "learning_rate": 1.8426895810341738e-09,
      "loss": 0.0564,
      "step": 99180
    },
    {
      "epoch": 2.4013519287109375e-05,
      "model_forward_time": 1.7700636386871338,
      "step": 99180
    },
    {
      "dynamics_loss_avg": 0.050660118274390695,
      "epoch": 2.4013519287109375e-05,
      "step": 99180
    },
    {
      "action_loss_avg": 0.0018891574698500335,
      "epoch": 2.4013519287109375e-05,
      "step": 99180
    },
    {
      "epoch": 2.4013519287109375e-05,
      "step": 99180,
      "training_step_time": 7.721347093582153
    },
    {
      "epoch": 2.4014472961425782e-05,
      "model_forward_time": 1.7640750408172607,
      "step": 99181
    },
    {
      "epoch": 2.4014472961425782e-05,
      "step": 99181,
      "training_step_time": 7.755882740020752
    },
    {
      "epoch": 2.4015426635742186e-05,
      "model_forward_time": 2.426265239715576,
      "step": 99182
    },
    {
      "epoch": 2.4015426635742186e-05,
      "step": 99182,
      "training_step_time": 11.408553123474121
    },
    {
      "epoch": 2.4016380310058594e-05,
      "model_forward_time": 1.757084608078003,
      "step": 99183
    },
    {
      "epoch": 2.4016380310058594e-05,
      "step": 99183,
      "training_step_time": 8.982115030288696
    },
    {
      "epoch": 2.4017333984375e-05,
      "model_forward_time": 1.7640321254730225,
      "step": 99184
    },
    {
      "epoch": 2.4017333984375e-05,
      "step": 99184,
      "training_step_time": 8.875916719436646
    },
    {
      "epoch": 2.4018287658691405e-05,
      "model_forward_time": 1.8078980445861816,
      "step": 99185
    },
    {
      "epoch": 2.4018287658691405e-05,
      "step": 99185,
      "training_step_time": 7.605786085128784
    },
    {
      "epoch": 2.4019241333007812e-05,
      "model_forward_time": 1.7664544582366943,
      "step": 99186
    },
    {
      "epoch": 2.4019241333007812e-05,
      "step": 99186,
      "training_step_time": 7.512813568115234
    },
    {
      "epoch": 2.402019500732422e-05,
      "model_forward_time": 1.7654697895050049,
      "step": 99187
    },
    {
      "epoch": 2.402019500732422e-05,
      "step": 99187,
      "training_step_time": 6.938328981399536
    },
    {
      "epoch": 2.4021148681640624e-05,
      "model_forward_time": 1.7670199871063232,
      "step": 99188
    },
    {
      "epoch": 2.4021148681640624e-05,
      "step": 99188,
      "training_step_time": 6.958442211151123
    },
    {
      "epoch": 2.402210235595703e-05,
      "model_forward_time": 1.760066270828247,
      "step": 99189
    },
    {
      "epoch": 2.402210235595703e-05,
      "step": 99189,
      "training_step_time": 7.745851278305054
    },
    {
      "epoch": 2.402305603027344e-05,
      "grad_norm": 0.17173095047473907,
      "learning_rate": 1.7980767290004531e-09,
      "loss": 0.0549,
      "step": 99190
    },
    {
      "epoch": 2.402305603027344e-05,
      "model_forward_time": 1.7928354740142822,
      "step": 99190
    },
    {
      "dynamics_loss_avg": 0.06759223267436028,
      "epoch": 2.402305603027344e-05,
      "step": 99190
    },
    {
      "action_loss_avg": 0.0024215002718847244,
      "epoch": 2.402305603027344e-05,
      "step": 99190
    },
    {
      "epoch": 2.402305603027344e-05,
      "step": 99190,
      "training_step_time": 7.7933690547943115
    },
    {
      "epoch": 2.4024009704589843e-05,
      "model_forward_time": 2.4195802211761475,
      "step": 99191
    },
    {
      "epoch": 2.4024009704589843e-05,
      "step": 99191,
      "training_step_time": 8.02655291557312
    },
    {
      "epoch": 2.402496337890625e-05,
      "model_forward_time": 1.7831289768218994,
      "step": 99192
    },
    {
      "epoch": 2.402496337890625e-05,
      "step": 99192,
      "training_step_time": 10.67324423789978
    },
    {
      "epoch": 2.4025917053222658e-05,
      "model_forward_time": 1.3994827270507812,
      "step": 99193
    },
    {
      "epoch": 2.4025917053222658e-05,
      "step": 99193,
      "training_step_time": 7.987000226974487
    },
    {
      "epoch": 2.402687072753906e-05,
      "model_forward_time": 1.7725214958190918,
      "step": 99194
    },
    {
      "epoch": 2.402687072753906e-05,
      "step": 99194,
      "training_step_time": 8.876615285873413
    },
    {
      "epoch": 2.402782440185547e-05,
      "model_forward_time": 1.76646089553833,
      "step": 99195
    },
    {
      "epoch": 2.402782440185547e-05,
      "step": 99195,
      "training_step_time": 7.475786447525024
    },
    {
      "epoch": 2.4028778076171876e-05,
      "model_forward_time": 1.7560696601867676,
      "step": 99196
    },
    {
      "epoch": 2.4028778076171876e-05,
      "step": 99196,
      "training_step_time": 6.927294969558716
    },
    {
      "epoch": 2.402973175048828e-05,
      "model_forward_time": 1.759160041809082,
      "step": 99197
    },
    {
      "epoch": 2.402973175048828e-05,
      "step": 99197,
      "training_step_time": 7.005162477493286
    },
    {
      "epoch": 2.4030685424804688e-05,
      "model_forward_time": 1.7707958221435547,
      "step": 99198
    },
    {
      "epoch": 2.4030685424804688e-05,
      "step": 99198,
      "training_step_time": 6.9493491649627686
    },
    {
      "epoch": 2.4031639099121095e-05,
      "model_forward_time": 1.7894392013549805,
      "step": 99199
    },
    {
      "epoch": 2.4031639099121095e-05,
      "step": 99199,
      "training_step_time": 8.500641584396362
    },
    {
      "epoch": 2.40325927734375e-05,
      "grad_norm": 0.11748107522726059,
      "learning_rate": 1.75401047281476e-09,
      "loss": 0.0532,
      "step": 99200
    },
    {
      "epoch": 2.40325927734375e-05,
      "model_forward_time": 1.764941930770874,
      "step": 99200
    },
    {
      "dynamics_loss_avg": 0.040899186953902246,
      "epoch": 2.40325927734375e-05,
      "step": 99200
    },
    {
      "action_loss_avg": 0.0021832326776348053,
      "epoch": 2.40325927734375e-05,
      "step": 99200
    },
    {
      "epoch": 2.40325927734375e-05,
      "step": 99200,
      "training_step_time": 9.456407308578491
    },
    {
      "epoch": 2.4033546447753907e-05,
      "model_forward_time": 2.4151265621185303,
      "step": 99201
    },
    {
      "epoch": 2.4033546447753907e-05,
      "step": 99201,
      "training_step_time": 11.223649263381958
    },
    {
      "epoch": 2.4034500122070314e-05,
      "model_forward_time": 1.7704942226409912,
      "step": 99202
    },
    {
      "epoch": 2.4034500122070314e-05,
      "step": 99202,
      "training_step_time": 9.704413890838623
    },
    {
      "epoch": 2.4035453796386718e-05,
      "model_forward_time": 1.7620015144348145,
      "step": 99203
    },
    {
      "epoch": 2.4035453796386718e-05,
      "step": 99203,
      "training_step_time": 8.537740707397461
    },
    {
      "epoch": 2.4036407470703126e-05,
      "model_forward_time": 1.7752361297607422,
      "step": 99204
    },
    {
      "epoch": 2.4036407470703126e-05,
      "step": 99204,
      "training_step_time": 7.46488881111145
    },
    {
      "epoch": 2.403736114501953e-05,
      "model_forward_time": 1.7629146575927734,
      "step": 99205
    },
    {
      "epoch": 2.403736114501953e-05,
      "step": 99205,
      "training_step_time": 6.965740203857422
    },
    {
      "epoch": 2.4038314819335937e-05,
      "model_forward_time": 1.763899803161621,
      "step": 99206
    },
    {
      "epoch": 2.4038314819335937e-05,
      "step": 99206,
      "training_step_time": 6.954970598220825
    },
    {
      "epoch": 2.4039268493652344e-05,
      "model_forward_time": 1.770909070968628,
      "step": 99207
    },
    {
      "epoch": 2.4039268493652344e-05,
      "step": 99207,
      "training_step_time": 6.89591908454895
    },
    {
      "epoch": 2.404022216796875e-05,
      "model_forward_time": 1.765209436416626,
      "step": 99208
    },
    {
      "epoch": 2.404022216796875e-05,
      "step": 99208,
      "training_step_time": 7.787107944488525
    },
    {
      "epoch": 2.4041175842285156e-05,
      "model_forward_time": 1.727860927581787,
      "step": 99209
    },
    {
      "epoch": 2.4041175842285156e-05,
      "step": 99209,
      "training_step_time": 9.09049105644226
    },
    {
      "epoch": 2.4042129516601563e-05,
      "grad_norm": 0.0888296365737915,
      "learning_rate": 1.7104908172954626e-09,
      "loss": 0.0562,
      "step": 99210
    },
    {
      "epoch": 2.4042129516601563e-05,
      "model_forward_time": 2.351501703262329,
      "step": 99210
    },
    {
      "dynamics_loss_avg": 0.044911821000278,
      "epoch": 2.4042129516601563e-05,
      "step": 99210
    },
    {
      "action_loss_avg": 0.0018658393586520106,
      "epoch": 2.4042129516601563e-05,
      "step": 99210
    },
    {
      "epoch": 2.4042129516601563e-05,
      "step": 99210,
      "training_step_time": 8.986191272735596
    },
    {
      "epoch": 2.4043083190917967e-05,
      "model_forward_time": 1.7410173416137695,
      "step": 99211
    },
    {
      "epoch": 2.4043083190917967e-05,
      "step": 99211,
      "training_step_time": 9.937474250793457
    },
    {
      "epoch": 2.4044036865234375e-05,
      "model_forward_time": 1.7472646236419678,
      "step": 99212
    },
    {
      "epoch": 2.4044036865234375e-05,
      "step": 99212,
      "training_step_time": 8.314026355743408
    },
    {
      "epoch": 2.4044990539550782e-05,
      "model_forward_time": 1.7633306980133057,
      "step": 99213
    },
    {
      "epoch": 2.4044990539550782e-05,
      "step": 99213,
      "training_step_time": 7.205132722854614
    },
    {
      "epoch": 2.4045944213867186e-05,
      "model_forward_time": 1.748887062072754,
      "step": 99214
    },
    {
      "epoch": 2.4045944213867186e-05,
      "step": 99214,
      "training_step_time": 7.260764837265015
    },
    {
      "epoch": 2.4046897888183594e-05,
      "model_forward_time": 1.7454612255096436,
      "step": 99215
    },
    {
      "epoch": 2.4046897888183594e-05,
      "step": 99215,
      "training_step_time": 6.722043991088867
    },
    {
      "epoch": 2.40478515625e-05,
      "model_forward_time": 1.7463366985321045,
      "step": 99216
    },
    {
      "epoch": 2.40478515625e-05,
      "step": 99216,
      "training_step_time": 6.833326578140259
    },
    {
      "epoch": 2.4048805236816405e-05,
      "model_forward_time": 1.7471013069152832,
      "step": 99217
    },
    {
      "epoch": 2.4048805236816405e-05,
      "step": 99217,
      "training_step_time": 7.649846315383911
    },
    {
      "epoch": 2.4049758911132812e-05,
      "model_forward_time": 1.7471590042114258,
      "step": 99218
    },
    {
      "epoch": 2.4049758911132812e-05,
      "step": 99218,
      "training_step_time": 9.635828971862793
    },
    {
      "epoch": 2.405071258544922e-05,
      "model_forward_time": 2.383418560028076,
      "step": 99219
    },
    {
      "epoch": 2.405071258544922e-05,
      "step": 99219,
      "training_step_time": 8.813062906265259
    },
    {
      "epoch": 2.4051666259765624e-05,
      "grad_norm": 0.08052282780408859,
      "learning_rate": 1.667517767202087e-09,
      "loss": 0.0549,
      "step": 99220
    },
    {
      "epoch": 2.4051666259765624e-05,
      "model_forward_time": 1.744541883468628,
      "step": 99220
    },
    {
      "dynamics_loss_avg": 0.06171093229204416,
      "epoch": 2.4051666259765624e-05,
      "step": 99220
    },
    {
      "action_loss_avg": 0.0024811170762404798,
      "epoch": 2.4051666259765624e-05,
      "step": 99220
    },
    {
      "epoch": 2.4051666259765624e-05,
      "step": 99220,
      "training_step_time": 8.950269222259521
    },
    {
      "epoch": 2.405261993408203e-05,
      "model_forward_time": 1.7550597190856934,
      "step": 99221
    },
    {
      "epoch": 2.405261993408203e-05,
      "step": 99221,
      "training_step_time": 8.456826448440552
    },
    {
      "epoch": 2.405357360839844e-05,
      "model_forward_time": 1.7548303604125977,
      "step": 99222
    },
    {
      "epoch": 2.405357360839844e-05,
      "step": 99222,
      "training_step_time": 7.799959182739258
    },
    {
      "epoch": 2.4054527282714843e-05,
      "model_forward_time": 1.7490918636322021,
      "step": 99223
    },
    {
      "epoch": 2.4054527282714843e-05,
      "step": 99223,
      "training_step_time": 7.539410829544067
    },
    {
      "epoch": 2.405548095703125e-05,
      "model_forward_time": 1.7522075176239014,
      "step": 99224
    },
    {
      "epoch": 2.405548095703125e-05,
      "step": 99224,
      "training_step_time": 6.999628782272339
    },
    {
      "epoch": 2.4056434631347658e-05,
      "model_forward_time": 1.7477798461914062,
      "step": 99225
    },
    {
      "epoch": 2.4056434631347658e-05,
      "step": 99225,
      "training_step_time": 6.971936225891113
    },
    {
      "epoch": 2.405738830566406e-05,
      "model_forward_time": 1.7463757991790771,
      "step": 99226
    },
    {
      "epoch": 2.405738830566406e-05,
      "step": 99226,
      "training_step_time": 7.849448919296265
    },
    {
      "epoch": 2.405834197998047e-05,
      "model_forward_time": 1.7536275386810303,
      "step": 99227
    },
    {
      "epoch": 2.405834197998047e-05,
      "step": 99227,
      "training_step_time": 8.88762640953064
    },
    {
      "epoch": 2.4059295654296876e-05,
      "model_forward_time": 2.425179958343506,
      "step": 99228
    },
    {
      "epoch": 2.4059295654296876e-05,
      "step": 99228,
      "training_step_time": 9.513692855834961
    },
    {
      "epoch": 2.406024932861328e-05,
      "model_forward_time": 1.7998175621032715,
      "step": 99229
    },
    {
      "epoch": 2.406024932861328e-05,
      "step": 99229,
      "training_step_time": 8.797532558441162
    },
    {
      "epoch": 2.4061203002929688e-05,
      "grad_norm": 0.06788980960845947,
      "learning_rate": 1.6250913272342072e-09,
      "loss": 0.0578,
      "step": 99230
    },
    {
      "epoch": 2.4061203002929688e-05,
      "model_forward_time": 1.743962287902832,
      "step": 99230
    },
    {
      "dynamics_loss_avg": 0.04475747048854828,
      "epoch": 2.4061203002929688e-05,
      "step": 99230
    },
    {
      "action_loss_avg": 0.0012987094465643167,
      "epoch": 2.4061203002929688e-05,
      "step": 99230
    },
    {
      "epoch": 2.4061203002929688e-05,
      "step": 99230,
      "training_step_time": 8.863282680511475
    },
    {
      "epoch": 2.4062156677246095e-05,
      "model_forward_time": 1.748053789138794,
      "step": 99231
    },
    {
      "epoch": 2.4062156677246095e-05,
      "step": 99231,
      "training_step_time": 7.915364742279053
    },
    {
      "epoch": 2.40631103515625e-05,
      "model_forward_time": 1.7553610801696777,
      "step": 99232
    },
    {
      "epoch": 2.40631103515625e-05,
      "step": 99232,
      "training_step_time": 7.552511692047119
    },
    {
      "epoch": 2.4064064025878907e-05,
      "model_forward_time": 1.7505638599395752,
      "step": 99233
    },
    {
      "epoch": 2.4064064025878907e-05,
      "step": 99233,
      "training_step_time": 7.034334897994995
    },
    {
      "epoch": 2.4065017700195314e-05,
      "model_forward_time": 1.7529969215393066,
      "step": 99234
    },
    {
      "epoch": 2.4065017700195314e-05,
      "step": 99234,
      "training_step_time": 7.0099780559539795
    },
    {
      "epoch": 2.4065971374511718e-05,
      "model_forward_time": 1.7467379570007324,
      "step": 99235
    },
    {
      "epoch": 2.4065971374511718e-05,
      "step": 99235,
      "training_step_time": 7.837626695632935
    },
    {
      "epoch": 2.4066925048828126e-05,
      "model_forward_time": 1.7469613552093506,
      "step": 99236
    },
    {
      "epoch": 2.4066925048828126e-05,
      "step": 99236,
      "training_step_time": 7.70503568649292
    },
    {
      "epoch": 2.406787872314453e-05,
      "model_forward_time": 2.370213747024536,
      "step": 99237
    },
    {
      "epoch": 2.406787872314453e-05,
      "step": 99237,
      "training_step_time": 8.078792810440063
    },
    {
      "epoch": 2.4068832397460937e-05,
      "model_forward_time": 1.8098039627075195,
      "step": 99238
    },
    {
      "epoch": 2.4068832397460937e-05,
      "step": 99238,
      "training_step_time": 11.695180892944336
    },
    {
      "epoch": 2.4069786071777344e-05,
      "model_forward_time": 1.745591163635254,
      "step": 99239
    },
    {
      "epoch": 2.4069786071777344e-05,
      "step": 99239,
      "training_step_time": 7.97919487953186
    },
    {
      "epoch": 2.407073974609375e-05,
      "grad_norm": 0.07304096966981888,
      "learning_rate": 1.5832115020314454e-09,
      "loss": 0.0551,
      "step": 99240
    },
    {
      "epoch": 2.407073974609375e-05,
      "model_forward_time": 1.7472634315490723,
      "step": 99240
    },
    {
      "dynamics_loss_avg": 0.06324307881295681,
      "epoch": 2.407073974609375e-05,
      "step": 99240
    },
    {
      "action_loss_avg": 0.0014062478207051755,
      "epoch": 2.407073974609375e-05,
      "step": 99240
    },
    {
      "epoch": 2.407073974609375e-05,
      "step": 99240,
      "training_step_time": 7.8483335971832275
    },
    {
      "epoch": 2.4071693420410156e-05,
      "model_forward_time": 1.748337984085083,
      "step": 99241
    },
    {
      "epoch": 2.4071693420410156e-05,
      "step": 99241,
      "training_step_time": 7.586209774017334
    },
    {
      "epoch": 2.4072647094726563e-05,
      "model_forward_time": 1.7473955154418945,
      "step": 99242
    },
    {
      "epoch": 2.4072647094726563e-05,
      "step": 99242,
      "training_step_time": 6.975135564804077
    },
    {
      "epoch": 2.4073600769042967e-05,
      "model_forward_time": 1.743518352508545,
      "step": 99243
    },
    {
      "epoch": 2.4073600769042967e-05,
      "step": 99243,
      "training_step_time": 6.912325620651245
    },
    {
      "epoch": 2.4074554443359375e-05,
      "model_forward_time": 1.7453558444976807,
      "step": 99244
    },
    {
      "epoch": 2.4074554443359375e-05,
      "step": 99244,
      "training_step_time": 6.778954267501831
    },
    {
      "epoch": 2.4075508117675782e-05,
      "model_forward_time": 1.7462279796600342,
      "step": 99245
    },
    {
      "epoch": 2.4075508117675782e-05,
      "step": 99245,
      "training_step_time": 7.645789861679077
    },
    {
      "epoch": 2.4076461791992186e-05,
      "model_forward_time": 2.4061994552612305,
      "step": 99246
    },
    {
      "epoch": 2.4076461791992186e-05,
      "step": 99246,
      "training_step_time": 9.580857038497925
    },
    {
      "epoch": 2.4077415466308594e-05,
      "model_forward_time": 1.8075041770935059,
      "step": 99247
    },
    {
      "epoch": 2.4077415466308594e-05,
      "step": 99247,
      "training_step_time": 8.743971586227417
    },
    {
      "epoch": 2.4078369140625e-05,
      "model_forward_time": 1.7399277687072754,
      "step": 99248
    },
    {
      "epoch": 2.4078369140625e-05,
      "step": 99248,
      "training_step_time": 8.434038877487183
    },
    {
      "epoch": 2.4079322814941405e-05,
      "model_forward_time": 1.7483634948730469,
      "step": 99249
    },
    {
      "epoch": 2.4079322814941405e-05,
      "step": 99249,
      "training_step_time": 7.630449295043945
    },
    {
      "epoch": 2.4080276489257812e-05,
      "grad_norm": 0.0668441504240036,
      "learning_rate": 1.5418782961734713e-09,
      "loss": 0.0555,
      "step": 99250
    },
    {
      "epoch": 2.4080276489257812e-05,
      "model_forward_time": 1.7741355895996094,
      "step": 99250
    },
    {
      "dynamics_loss_avg": 0.05351390317082405,
      "epoch": 2.4080276489257812e-05,
      "step": 99250
    },
    {
      "action_loss_avg": 0.002119338547345251,
      "epoch": 2.4080276489257812e-05,
      "step": 99250
    },
    {
      "epoch": 2.4080276489257812e-05,
      "step": 99250,
      "training_step_time": 7.577293157577515
    },
    {
      "epoch": 2.408123016357422e-05,
      "model_forward_time": 1.7698805332183838,
      "step": 99251
    },
    {
      "epoch": 2.408123016357422e-05,
      "step": 99251,
      "training_step_time": 7.289986848831177
    },
    {
      "epoch": 2.4082183837890624e-05,
      "model_forward_time": 1.7682812213897705,
      "step": 99252
    },
    {
      "epoch": 2.4082183837890624e-05,
      "step": 99252,
      "training_step_time": 6.8215649127960205
    },
    {
      "epoch": 2.408313751220703e-05,
      "model_forward_time": 1.7742466926574707,
      "step": 99253
    },
    {
      "epoch": 2.408313751220703e-05,
      "step": 99253,
      "training_step_time": 6.812037706375122
    },
    {
      "epoch": 2.408409118652344e-05,
      "model_forward_time": 1.7630116939544678,
      "step": 99254
    },
    {
      "epoch": 2.408409118652344e-05,
      "step": 99254,
      "training_step_time": 7.797360897064209
    },
    {
      "epoch": 2.4085044860839843e-05,
      "model_forward_time": 2.424825429916382,
      "step": 99255
    },
    {
      "epoch": 2.4085044860839843e-05,
      "step": 99255,
      "training_step_time": 10.131252527236938
    },
    {
      "epoch": 2.408599853515625e-05,
      "model_forward_time": 1.8050074577331543,
      "step": 99256
    },
    {
      "epoch": 2.408599853515625e-05,
      "step": 99256,
      "training_step_time": 8.827400922775269
    },
    {
      "epoch": 2.4086952209472658e-05,
      "model_forward_time": 1.764632225036621,
      "step": 99257
    },
    {
      "epoch": 2.4086952209472658e-05,
      "step": 99257,
      "training_step_time": 11.552639484405518
    },
    {
      "epoch": 2.408790588378906e-05,
      "model_forward_time": 1.771456003189087,
      "step": 99258
    },
    {
      "epoch": 2.408790588378906e-05,
      "step": 99258,
      "training_step_time": 8.822957277297974
    },
    {
      "epoch": 2.408885955810547e-05,
      "model_forward_time": 1.7698845863342285,
      "step": 99259
    },
    {
      "epoch": 2.408885955810547e-05,
      "step": 99259,
      "training_step_time": 7.54280161857605
    },
    {
      "epoch": 2.4089813232421876e-05,
      "grad_norm": 0.07018409669399261,
      "learning_rate": 1.5010917141811132e-09,
      "loss": 0.0566,
      "step": 99260
    },
    {
      "epoch": 2.4089813232421876e-05,
      "model_forward_time": 1.7761311531066895,
      "step": 99260
    },
    {
      "dynamics_loss_avg": 0.05736774317920208,
      "epoch": 2.4089813232421876e-05,
      "step": 99260
    },
    {
      "action_loss_avg": 0.0019596369296777994,
      "epoch": 2.4089813232421876e-05,
      "step": 99260
    },
    {
      "epoch": 2.4089813232421876e-05,
      "step": 99260,
      "training_step_time": 7.260406732559204
    },
    {
      "epoch": 2.409076690673828e-05,
      "model_forward_time": 1.7394070625305176,
      "step": 99261
    },
    {
      "epoch": 2.409076690673828e-05,
      "step": 99261,
      "training_step_time": 6.809853553771973
    },
    {
      "epoch": 2.4091720581054688e-05,
      "model_forward_time": 1.7458252906799316,
      "step": 99262
    },
    {
      "epoch": 2.4091720581054688e-05,
      "step": 99262,
      "training_step_time": 6.787142276763916
    },
    {
      "epoch": 2.4092674255371095e-05,
      "model_forward_time": 1.7429301738739014,
      "step": 99263
    },
    {
      "epoch": 2.4092674255371095e-05,
      "step": 99263,
      "training_step_time": 8.658284425735474
    },
    {
      "epoch": 2.40936279296875e-05,
      "model_forward_time": 1.7637395858764648,
      "step": 99264
    },
    {
      "epoch": 2.40936279296875e-05,
      "step": 99264,
      "training_step_time": 8.281282663345337
    },
    {
      "epoch": 2.4094581604003907e-05,
      "model_forward_time": 1.7491772174835205,
      "step": 99265
    },
    {
      "epoch": 2.4094581604003907e-05,
      "step": 99265,
      "training_step_time": 9.293592929840088
    },
    {
      "epoch": 2.409553527832031e-05,
      "model_forward_time": 1.7425270080566406,
      "step": 99266
    },
    {
      "epoch": 2.409553527832031e-05,
      "step": 99266,
      "training_step_time": 9.449867486953735
    },
    {
      "epoch": 2.4096488952636718e-05,
      "model_forward_time": 1.752277135848999,
      "step": 99267
    },
    {
      "epoch": 2.4096488952636718e-05,
      "step": 99267,
      "training_step_time": 7.663518190383911
    },
    {
      "epoch": 2.4097442626953126e-05,
      "model_forward_time": 1.7417128086090088,
      "step": 99268
    },
    {
      "epoch": 2.4097442626953126e-05,
      "step": 99268,
      "training_step_time": 7.590185165405273
    },
    {
      "epoch": 2.409839630126953e-05,
      "model_forward_time": 1.7554595470428467,
      "step": 99269
    },
    {
      "epoch": 2.409839630126953e-05,
      "step": 99269,
      "training_step_time": 7.451855659484863
    },
    {
      "epoch": 2.4099349975585937e-05,
      "grad_norm": 0.11981453746557236,
      "learning_rate": 1.460851760513582e-09,
      "loss": 0.0542,
      "step": 99270
    },
    {
      "epoch": 2.4099349975585937e-05,
      "model_forward_time": 1.736311912536621,
      "step": 99270
    },
    {
      "dynamics_loss_avg": 0.06610780246555806,
      "epoch": 2.4099349975585937e-05,
      "step": 99270
    },
    {
      "action_loss_avg": 0.002431784535292536,
      "epoch": 2.4099349975585937e-05,
      "step": 99270
    },
    {
      "epoch": 2.4099349975585937e-05,
      "step": 99270,
      "training_step_time": 7.238403081893921
    },
    {
      "epoch": 2.4100303649902345e-05,
      "model_forward_time": 1.7478232383728027,
      "step": 99271
    },
    {
      "epoch": 2.4100303649902345e-05,
      "step": 99271,
      "training_step_time": 6.806115388870239
    },
    {
      "epoch": 2.410125732421875e-05,
      "model_forward_time": 1.0405938625335693,
      "step": 99272
    },
    {
      "epoch": 2.410125732421875e-05,
      "step": 99272,
      "training_step_time": 7.579513072967529
    },
    {
      "epoch": 2.4102210998535156e-05,
      "model_forward_time": 1.7531447410583496,
      "step": 99273
    },
    {
      "epoch": 2.4102210998535156e-05,
      "step": 99273,
      "training_step_time": 7.65241813659668
    },
    {
      "epoch": 2.4103164672851563e-05,
      "model_forward_time": 2.396104097366333,
      "step": 99274
    },
    {
      "epoch": 2.4103164672851563e-05,
      "step": 99274,
      "training_step_time": 9.249860525131226
    },
    {
      "epoch": 2.4104118347167967e-05,
      "model_forward_time": 1.8087000846862793,
      "step": 99275
    },
    {
      "epoch": 2.4104118347167967e-05,
      "step": 99275,
      "training_step_time": 10.305450439453125
    },
    {
      "epoch": 2.4105072021484375e-05,
      "model_forward_time": 1.742356538772583,
      "step": 99276
    },
    {
      "epoch": 2.4105072021484375e-05,
      "step": 99276,
      "training_step_time": 8.462197542190552
    },
    {
      "epoch": 2.4106025695800782e-05,
      "model_forward_time": 1.7459228038787842,
      "step": 99277
    },
    {
      "epoch": 2.4106025695800782e-05,
      "step": 99277,
      "training_step_time": 9.043991565704346
    },
    {
      "epoch": 2.4106979370117186e-05,
      "model_forward_time": 1.7460401058197021,
      "step": 99278
    },
    {
      "epoch": 2.4106979370117186e-05,
      "step": 99278,
      "training_step_time": 7.568741798400879
    },
    {
      "epoch": 2.4107933044433594e-05,
      "model_forward_time": 1.7485792636871338,
      "step": 99279
    },
    {
      "epoch": 2.4107933044433594e-05,
      "step": 99279,
      "training_step_time": 7.259767293930054
    },
    {
      "epoch": 2.410888671875e-05,
      "grad_norm": 0.09392767399549484,
      "learning_rate": 1.4211584395723565e-09,
      "loss": 0.0541,
      "step": 99280
    },
    {
      "epoch": 2.410888671875e-05,
      "model_forward_time": 1.742112398147583,
      "step": 99280
    },
    {
      "dynamics_loss_avg": 0.05233171321451664,
      "epoch": 2.410888671875e-05,
      "step": 99280
    },
    {
      "action_loss_avg": 0.0016206153610255568,
      "epoch": 2.410888671875e-05,
      "step": 99280
    },
    {
      "epoch": 2.410888671875e-05,
      "step": 99280,
      "training_step_time": 6.7947998046875
    },
    {
      "epoch": 2.4109840393066405e-05,
      "model_forward_time": 1.7418854236602783,
      "step": 99281
    },
    {
      "epoch": 2.4109840393066405e-05,
      "step": 99281,
      "training_step_time": 7.227626323699951
    },
    {
      "epoch": 2.4110794067382813e-05,
      "model_forward_time": 1.7446694374084473,
      "step": 99282
    },
    {
      "epoch": 2.4110794067382813e-05,
      "step": 99282,
      "training_step_time": 7.6390700340271
    },
    {
      "epoch": 2.411174774169922e-05,
      "model_forward_time": 2.413602113723755,
      "step": 99283
    },
    {
      "epoch": 2.411174774169922e-05,
      "step": 99283,
      "training_step_time": 7.774501800537109
    },
    {
      "epoch": 2.4112701416015624e-05,
      "model_forward_time": 1.8142664432525635,
      "step": 99284
    },
    {
      "epoch": 2.4112701416015624e-05,
      "step": 99284,
      "training_step_time": 8.403091430664062
    },
    {
      "epoch": 2.411365509033203e-05,
      "model_forward_time": 1.7436113357543945,
      "step": 99285
    },
    {
      "epoch": 2.411365509033203e-05,
      "step": 99285,
      "training_step_time": 8.39682126045227
    },
    {
      "epoch": 2.411460876464844e-05,
      "model_forward_time": 1.782738447189331,
      "step": 99286
    },
    {
      "epoch": 2.411460876464844e-05,
      "step": 99286,
      "training_step_time": 8.297241687774658
    },
    {
      "epoch": 2.4115562438964843e-05,
      "model_forward_time": 1.7403590679168701,
      "step": 99287
    },
    {
      "epoch": 2.4115562438964843e-05,
      "step": 99287,
      "training_step_time": 7.4875428676605225
    },
    {
      "epoch": 2.411651611328125e-05,
      "model_forward_time": 1.746668815612793,
      "step": 99288
    },
    {
      "epoch": 2.411651611328125e-05,
      "step": 99288,
      "training_step_time": 7.267781972885132
    },
    {
      "epoch": 2.4117469787597658e-05,
      "model_forward_time": 1.7411601543426514,
      "step": 99289
    },
    {
      "epoch": 2.4117469787597658e-05,
      "step": 99289,
      "training_step_time": 6.758301019668579
    },
    {
      "epoch": 2.411842346191406e-05,
      "grad_norm": 0.07940223813056946,
      "learning_rate": 1.3820117556978542e-09,
      "loss": 0.0552,
      "step": 99290
    },
    {
      "epoch": 2.411842346191406e-05,
      "model_forward_time": 1.7403178215026855,
      "step": 99290
    },
    {
      "dynamics_loss_avg": 0.04207948045805097,
      "epoch": 2.411842346191406e-05,
      "step": 99290
    },
    {
      "action_loss_avg": 0.00228907119599171,
      "epoch": 2.411842346191406e-05,
      "step": 99290
    },
    {
      "epoch": 2.411842346191406e-05,
      "step": 99290,
      "training_step_time": 7.608654022216797
    },
    {
      "epoch": 2.411937713623047e-05,
      "model_forward_time": 1.7433738708496094,
      "step": 99291
    },
    {
      "epoch": 2.411937713623047e-05,
      "step": 99291,
      "training_step_time": 8.66295337677002
    },
    {
      "epoch": 2.4120330810546877e-05,
      "model_forward_time": 2.358145236968994,
      "step": 99292
    },
    {
      "epoch": 2.4120330810546877e-05,
      "step": 99292,
      "training_step_time": 8.62435245513916
    },
    {
      "epoch": 2.412128448486328e-05,
      "model_forward_time": 1.8118813037872314,
      "step": 99293
    },
    {
      "epoch": 2.412128448486328e-05,
      "step": 99293,
      "training_step_time": 10.365487575531006
    },
    {
      "epoch": 2.4122238159179688e-05,
      "model_forward_time": 1.735246181488037,
      "step": 99294
    },
    {
      "epoch": 2.4122238159179688e-05,
      "step": 99294,
      "training_step_time": 8.179533004760742
    },
    {
      "epoch": 2.4123191833496095e-05,
      "model_forward_time": 1.7431573867797852,
      "step": 99295
    },
    {
      "epoch": 2.4123191833496095e-05,
      "step": 99295,
      "training_step_time": 7.611886739730835
    },
    {
      "epoch": 2.41241455078125e-05,
      "model_forward_time": 1.735581398010254,
      "step": 99296
    },
    {
      "epoch": 2.41241455078125e-05,
      "step": 99296,
      "training_step_time": 8.472042560577393
    },
    {
      "epoch": 2.4125099182128907e-05,
      "model_forward_time": 1.7472763061523438,
      "step": 99297
    },
    {
      "epoch": 2.4125099182128907e-05,
      "step": 99297,
      "training_step_time": 7.247538089752197
    },
    {
      "epoch": 2.412605285644531e-05,
      "model_forward_time": 1.742330551147461,
      "step": 99298
    },
    {
      "epoch": 2.412605285644531e-05,
      "step": 99298,
      "training_step_time": 6.768595933914185
    },
    {
      "epoch": 2.4127006530761718e-05,
      "model_forward_time": 1.748955488204956,
      "step": 99299
    },
    {
      "epoch": 2.4127006530761718e-05,
      "step": 99299,
      "training_step_time": 6.72630500793457
    },
    {
      "epoch": 2.4127960205078126e-05,
      "grad_norm": 0.08642648905515671,
      "learning_rate": 1.3434117131716495e-09,
      "loss": 0.0558,
      "step": 99300
    },
    {
      "epoch": 2.4127960205078126e-05,
      "model_forward_time": 1.7367043495178223,
      "step": 99300
    },
    {
      "dynamics_loss_avg": 0.04975008983165026,
      "epoch": 2.4127960205078126e-05,
      "step": 99300
    },
    {
      "action_loss_avg": 0.0010693083750084043,
      "epoch": 2.4127960205078126e-05,
      "step": 99300
    },
    {
      "epoch": 2.4127960205078126e-05,
      "step": 99300,
      "training_step_time": 7.5158562660217285
    },
    {
      "epoch": 2.412891387939453e-05,
      "model_forward_time": 1.7379398345947266,
      "step": 99301
    },
    {
      "epoch": 2.412891387939453e-05,
      "step": 99301,
      "training_step_time": 7.821018695831299
    },
    {
      "epoch": 2.4129867553710937e-05,
      "model_forward_time": 2.4127817153930664,
      "step": 99302
    },
    {
      "epoch": 2.4129867553710937e-05,
      "step": 99302,
      "training_step_time": 8.755484580993652
    },
    {
      "epoch": 2.4130821228027345e-05,
      "model_forward_time": 1.7432408332824707,
      "step": 99303
    },
    {
      "epoch": 2.4130821228027345e-05,
      "step": 99303,
      "training_step_time": 10.288475513458252
    },
    {
      "epoch": 2.413177490234375e-05,
      "model_forward_time": 1.738206386566162,
      "step": 99304
    },
    {
      "epoch": 2.413177490234375e-05,
      "step": 99304,
      "training_step_time": 7.7216408252716064
    },
    {
      "epoch": 2.4132728576660156e-05,
      "model_forward_time": 1.7414686679840088,
      "step": 99305
    },
    {
      "epoch": 2.4132728576660156e-05,
      "step": 99305,
      "training_step_time": 7.664744138717651
    },
    {
      "epoch": 2.4133682250976563e-05,
      "model_forward_time": 1.7426822185516357,
      "step": 99306
    },
    {
      "epoch": 2.4133682250976563e-05,
      "step": 99306,
      "training_step_time": 7.6187708377838135
    },
    {
      "epoch": 2.4134635925292967e-05,
      "model_forward_time": 1.7359294891357422,
      "step": 99307
    },
    {
      "epoch": 2.4134635925292967e-05,
      "step": 99307,
      "training_step_time": 7.290663003921509
    },
    {
      "epoch": 2.4135589599609375e-05,
      "model_forward_time": 1.7489104270935059,
      "step": 99308
    },
    {
      "epoch": 2.4135589599609375e-05,
      "step": 99308,
      "training_step_time": 6.761634826660156
    },
    {
      "epoch": 2.4136543273925782e-05,
      "model_forward_time": 1.7427809238433838,
      "step": 99309
    },
    {
      "epoch": 2.4136543273925782e-05,
      "step": 99309,
      "training_step_time": 7.7125983238220215
    },
    {
      "epoch": 2.4137496948242186e-05,
      "grad_norm": 0.11868829280138016,
      "learning_rate": 1.3053583162142558e-09,
      "loss": 0.0567,
      "step": 99310
    },
    {
      "epoch": 2.4137496948242186e-05,
      "model_forward_time": 1.7374238967895508,
      "step": 99310
    },
    {
      "dynamics_loss_avg": 0.04291280377656222,
      "epoch": 2.4137496948242186e-05,
      "step": 99310
    },
    {
      "action_loss_avg": 0.0017137243878096343,
      "epoch": 2.4137496948242186e-05,
      "step": 99310
    },
    {
      "epoch": 2.4137496948242186e-05,
      "step": 99310,
      "training_step_time": 7.830890417098999
    },
    {
      "epoch": 2.4138450622558594e-05,
      "model_forward_time": 2.411957025527954,
      "step": 99311
    },
    {
      "epoch": 2.4138450622558594e-05,
      "step": 99311,
      "training_step_time": 9.420926332473755
    },
    {
      "epoch": 2.4139404296875e-05,
      "model_forward_time": 1.7379281520843506,
      "step": 99312
    },
    {
      "epoch": 2.4139404296875e-05,
      "step": 99312,
      "training_step_time": 10.625885486602783
    },
    {
      "epoch": 2.4140357971191405e-05,
      "model_forward_time": 1.7371954917907715,
      "step": 99313
    },
    {
      "epoch": 2.4140357971191405e-05,
      "step": 99313,
      "training_step_time": 7.69542121887207
    },
    {
      "epoch": 2.4141311645507813e-05,
      "model_forward_time": 1.7373652458190918,
      "step": 99314
    },
    {
      "epoch": 2.4141311645507813e-05,
      "step": 99314,
      "training_step_time": 7.568339586257935
    },
    {
      "epoch": 2.414226531982422e-05,
      "model_forward_time": 1.7448031902313232,
      "step": 99315
    },
    {
      "epoch": 2.414226531982422e-05,
      "step": 99315,
      "training_step_time": 7.574551820755005
    },
    {
      "epoch": 2.4143218994140624e-05,
      "model_forward_time": 1.742807388305664,
      "step": 99316
    },
    {
      "epoch": 2.4143218994140624e-05,
      "step": 99316,
      "training_step_time": 7.331007957458496
    },
    {
      "epoch": 2.414417266845703e-05,
      "model_forward_time": 1.7414698600769043,
      "step": 99317
    },
    {
      "epoch": 2.414417266845703e-05,
      "step": 99317,
      "training_step_time": 6.788992643356323
    },
    {
      "epoch": 2.414512634277344e-05,
      "model_forward_time": 1.7377867698669434,
      "step": 99318
    },
    {
      "epoch": 2.414512634277344e-05,
      "step": 99318,
      "training_step_time": 7.6503400802612305
    },
    {
      "epoch": 2.4146080017089843e-05,
      "model_forward_time": 1.7482104301452637,
      "step": 99319
    },
    {
      "epoch": 2.4146080017089843e-05,
      "step": 99319,
      "training_step_time": 8.812323093414307
    },
    {
      "epoch": 2.414703369140625e-05,
      "grad_norm": 0.08522826433181763,
      "learning_rate": 1.2678515689873439e-09,
      "loss": 0.0544,
      "step": 99320
    },
    {
      "epoch": 2.414703369140625e-05,
      "model_forward_time": 2.3599605560302734,
      "step": 99320
    },
    {
      "dynamics_loss_avg": 0.05131405536085367,
      "epoch": 2.414703369140625e-05,
      "step": 99320
    },
    {
      "action_loss_avg": 0.0012687089969404043,
      "epoch": 2.414703369140625e-05,
      "step": 99320
    },
    {
      "epoch": 2.414703369140625e-05,
      "step": 99320,
      "training_step_time": 9.378025770187378
    },
    {
      "epoch": 2.4147987365722658e-05,
      "model_forward_time": 1.739811658859253,
      "step": 99321
    },
    {
      "epoch": 2.4147987365722658e-05,
      "step": 99321,
      "training_step_time": 9.798954725265503
    },
    {
      "epoch": 2.4148941040039062e-05,
      "model_forward_time": 1.73968505859375,
      "step": 99322
    },
    {
      "epoch": 2.4148941040039062e-05,
      "step": 99322,
      "training_step_time": 7.714308738708496
    },
    {
      "epoch": 2.414989471435547e-05,
      "model_forward_time": 1.7408559322357178,
      "step": 99323
    },
    {
      "epoch": 2.414989471435547e-05,
      "step": 99323,
      "training_step_time": 8.28725266456604
    },
    {
      "epoch": 2.4150848388671877e-05,
      "model_forward_time": 1.7738709449768066,
      "step": 99324
    },
    {
      "epoch": 2.4150848388671877e-05,
      "step": 99324,
      "training_step_time": 7.423311948776245
    },
    {
      "epoch": 2.415180206298828e-05,
      "model_forward_time": 1.745988368988037,
      "step": 99325
    },
    {
      "epoch": 2.415180206298828e-05,
      "step": 99325,
      "training_step_time": 7.323998689651489
    },
    {
      "epoch": 2.4152755737304688e-05,
      "model_forward_time": 1.7447612285614014,
      "step": 99326
    },
    {
      "epoch": 2.4152755737304688e-05,
      "step": 99326,
      "training_step_time": 6.835065841674805
    },
    {
      "epoch": 2.4153709411621095e-05,
      "model_forward_time": 1.746263027191162,
      "step": 99327
    },
    {
      "epoch": 2.4153709411621095e-05,
      "step": 99327,
      "training_step_time": 7.602014064788818
    },
    {
      "epoch": 2.41546630859375e-05,
      "model_forward_time": 1.7388718128204346,
      "step": 99328
    },
    {
      "epoch": 2.41546630859375e-05,
      "step": 99328,
      "training_step_time": 7.656225681304932
    },
    {
      "epoch": 2.4155616760253907e-05,
      "model_forward_time": 2.3756816387176514,
      "step": 99329
    },
    {
      "epoch": 2.4155616760253907e-05,
      "step": 99329,
      "training_step_time": 8.387201309204102
    },
    {
      "epoch": 2.415657043457031e-05,
      "grad_norm": 0.10961877554655075,
      "learning_rate": 1.2308914755931877e-09,
      "loss": 0.0563,
      "step": 99330
    },
    {
      "epoch": 2.415657043457031e-05,
      "model_forward_time": 1.8090739250183105,
      "step": 99330
    },
    {
      "dynamics_loss_avg": 0.0566438952460885,
      "epoch": 2.415657043457031e-05,
      "step": 99330
    },
    {
      "action_loss_avg": 0.0018202999286586418,
      "epoch": 2.415657043457031e-05,
      "step": 99330
    },
    {
      "epoch": 2.415657043457031e-05,
      "step": 99330,
      "training_step_time": 8.35268759727478
    },
    {
      "epoch": 2.4157524108886718e-05,
      "model_forward_time": 1.3715794086456299,
      "step": 99331
    },
    {
      "epoch": 2.4157524108886718e-05,
      "step": 99331,
      "training_step_time": 9.538940668106079
    },
    {
      "epoch": 2.4158477783203126e-05,
      "model_forward_time": 1.7472546100616455,
      "step": 99332
    },
    {
      "epoch": 2.4158477783203126e-05,
      "step": 99332,
      "training_step_time": 7.621528387069702
    },
    {
      "epoch": 2.415943145751953e-05,
      "model_forward_time": 1.7404441833496094,
      "step": 99333
    },
    {
      "epoch": 2.415943145751953e-05,
      "step": 99333,
      "training_step_time": 9.643400430679321
    },
    {
      "epoch": 2.4160385131835937e-05,
      "model_forward_time": 1.7381837368011475,
      "step": 99334
    },
    {
      "epoch": 2.4160385131835937e-05,
      "step": 99334,
      "training_step_time": 7.287134885787964
    },
    {
      "epoch": 2.4161338806152345e-05,
      "model_forward_time": 1.737325668334961,
      "step": 99335
    },
    {
      "epoch": 2.4161338806152345e-05,
      "step": 99335,
      "training_step_time": 7.119974136352539
    },
    {
      "epoch": 2.416229248046875e-05,
      "model_forward_time": 1.73897123336792,
      "step": 99336
    },
    {
      "epoch": 2.416229248046875e-05,
      "step": 99336,
      "training_step_time": 7.563082695007324
    },
    {
      "epoch": 2.4163246154785156e-05,
      "model_forward_time": 1.7442693710327148,
      "step": 99337
    },
    {
      "epoch": 2.4163246154785156e-05,
      "step": 99337,
      "training_step_time": 7.64050817489624
    },
    {
      "epoch": 2.4164199829101563e-05,
      "model_forward_time": 2.345237970352173,
      "step": 99338
    },
    {
      "epoch": 2.4164199829101563e-05,
      "step": 99338,
      "training_step_time": 7.815141201019287
    },
    {
      "epoch": 2.4165153503417967e-05,
      "model_forward_time": 1.8086943626403809,
      "step": 99339
    },
    {
      "epoch": 2.4165153503417967e-05,
      "step": 99339,
      "training_step_time": 10.521420955657959
    },
    {
      "epoch": 2.4166107177734375e-05,
      "grad_norm": 0.07937514036893845,
      "learning_rate": 1.1944780400729995e-09,
      "loss": 0.0565,
      "step": 99340
    },
    {
      "epoch": 2.4166107177734375e-05,
      "model_forward_time": 1.7372124195098877,
      "step": 99340
    },
    {
      "dynamics_loss_avg": 0.06292342618107796,
      "epoch": 2.4166107177734375e-05,
      "step": 99340
    },
    {
      "action_loss_avg": 0.0019896991085261106,
      "epoch": 2.4166107177734375e-05,
      "step": 99340
    },
    {
      "epoch": 2.4166107177734375e-05,
      "step": 99340,
      "training_step_time": 8.492426633834839
    },
    {
      "epoch": 2.4167060852050782e-05,
      "model_forward_time": 1.7377874851226807,
      "step": 99341
    },
    {
      "epoch": 2.4167060852050782e-05,
      "step": 99341,
      "training_step_time": 8.857899188995361
    },
    {
      "epoch": 2.4168014526367186e-05,
      "model_forward_time": 1.74476957321167,
      "step": 99342
    },
    {
      "epoch": 2.4168014526367186e-05,
      "step": 99342,
      "training_step_time": 8.04057788848877
    },
    {
      "epoch": 2.4168968200683594e-05,
      "model_forward_time": 1.738058090209961,
      "step": 99343
    },
    {
      "epoch": 2.4168968200683594e-05,
      "step": 99343,
      "training_step_time": 7.584445238113403
    },
    {
      "epoch": 2.4169921875e-05,
      "model_forward_time": 1.742504596710205,
      "step": 99344
    },
    {
      "epoch": 2.4169921875e-05,
      "step": 99344,
      "training_step_time": 7.243755340576172
    },
    {
      "epoch": 2.4170875549316405e-05,
      "model_forward_time": 1.7418944835662842,
      "step": 99345
    },
    {
      "epoch": 2.4170875549316405e-05,
      "step": 99345,
      "training_step_time": 8.451515197753906
    },
    {
      "epoch": 2.4171829223632813e-05,
      "model_forward_time": 1.3574707508087158,
      "step": 99346
    },
    {
      "epoch": 2.4171829223632813e-05,
      "step": 99346,
      "training_step_time": 7.552248477935791
    },
    {
      "epoch": 2.417278289794922e-05,
      "model_forward_time": 1.746917486190796,
      "step": 99347
    },
    {
      "epoch": 2.417278289794922e-05,
      "step": 99347,
      "training_step_time": 9.005781173706055
    },
    {
      "epoch": 2.4173736572265624e-05,
      "model_forward_time": 2.4292409420013428,
      "step": 99348
    },
    {
      "epoch": 2.4173736572265624e-05,
      "step": 99348,
      "training_step_time": 7.7694714069366455
    },
    {
      "epoch": 2.417469024658203e-05,
      "model_forward_time": 1.739645004272461,
      "step": 99349
    },
    {
      "epoch": 2.417469024658203e-05,
      "step": 99349,
      "training_step_time": 7.9530510902404785
    },
    {
      "epoch": 2.417564392089844e-05,
      "grad_norm": 0.07987593859434128,
      "learning_rate": 1.1586112664085936e-09,
      "loss": 0.0547,
      "step": 99350
    },
    {
      "epoch": 2.417564392089844e-05,
      "model_forward_time": 1.7467741966247559,
      "step": 99350
    },
    {
      "dynamics_loss_avg": 0.061558060348033905,
      "epoch": 2.417564392089844e-05,
      "step": 99350
    },
    {
      "action_loss_avg": 0.00106933055940317,
      "epoch": 2.417564392089844e-05,
      "step": 99350
    },
    {
      "epoch": 2.417564392089844e-05,
      "step": 99350,
      "training_step_time": 7.646706581115723
    },
    {
      "epoch": 2.4176597595214843e-05,
      "model_forward_time": 1.7369024753570557,
      "step": 99351
    },
    {
      "epoch": 2.4176597595214843e-05,
      "step": 99351,
      "training_step_time": 7.806913137435913
    },
    {
      "epoch": 2.417755126953125e-05,
      "model_forward_time": 1.7399919033050537,
      "step": 99352
    },
    {
      "epoch": 2.417755126953125e-05,
      "step": 99352,
      "training_step_time": 7.518738508224487
    },
    {
      "epoch": 2.4178504943847658e-05,
      "model_forward_time": 1.7410171031951904,
      "step": 99353
    },
    {
      "epoch": 2.4178504943847658e-05,
      "step": 99353,
      "training_step_time": 7.2424585819244385
    },
    {
      "epoch": 2.4179458618164062e-05,
      "model_forward_time": 1.7466492652893066,
      "step": 99354
    },
    {
      "epoch": 2.4179458618164062e-05,
      "step": 99354,
      "training_step_time": 7.567248821258545
    },
    {
      "epoch": 2.418041229248047e-05,
      "model_forward_time": 1.7412490844726562,
      "step": 99355
    },
    {
      "epoch": 2.418041229248047e-05,
      "step": 99355,
      "training_step_time": 9.323755741119385
    },
    {
      "epoch": 2.4181365966796877e-05,
      "model_forward_time": 1.7406282424926758,
      "step": 99356
    },
    {
      "epoch": 2.4181365966796877e-05,
      "step": 99356,
      "training_step_time": 8.559551477432251
    },
    {
      "epoch": 2.418231964111328e-05,
      "model_forward_time": 2.4041945934295654,
      "step": 99357
    },
    {
      "epoch": 2.418231964111328e-05,
      "step": 99357,
      "training_step_time": 8.722386121749878
    },
    {
      "epoch": 2.4183273315429688e-05,
      "model_forward_time": 1.737309217453003,
      "step": 99358
    },
    {
      "epoch": 2.4183273315429688e-05,
      "step": 99358,
      "training_step_time": 8.439605474472046
    },
    {
      "epoch": 2.4184226989746095e-05,
      "model_forward_time": 1.7387361526489258,
      "step": 99359
    },
    {
      "epoch": 2.4184226989746095e-05,
      "step": 99359,
      "training_step_time": 9.2277512550354
    },
    {
      "epoch": 2.41851806640625e-05,
      "grad_norm": 0.08086448907852173,
      "learning_rate": 1.1232911585234985e-09,
      "loss": 0.0557,
      "step": 99360
    },
    {
      "epoch": 2.41851806640625e-05,
      "model_forward_time": 1.7399108409881592,
      "step": 99360
    },
    {
      "dynamics_loss_avg": 0.04572404120117426,
      "epoch": 2.41851806640625e-05,
      "step": 99360
    },
    {
      "action_loss_avg": 0.0013448013283777981,
      "epoch": 2.41851806640625e-05,
      "step": 99360
    },
    {
      "epoch": 2.41851806640625e-05,
      "step": 99360,
      "training_step_time": 8.431502342224121
    },
    {
      "epoch": 2.4186134338378907e-05,
      "model_forward_time": 1.7485542297363281,
      "step": 99361
    },
    {
      "epoch": 2.4186134338378907e-05,
      "step": 99361,
      "training_step_time": 7.678818941116333
    },
    {
      "epoch": 2.418708801269531e-05,
      "model_forward_time": 1.750974416732788,
      "step": 99362
    },
    {
      "epoch": 2.418708801269531e-05,
      "step": 99362,
      "training_step_time": 7.317589044570923
    },
    {
      "epoch": 2.418804168701172e-05,
      "model_forward_time": 1.7507495880126953,
      "step": 99363
    },
    {
      "epoch": 2.418804168701172e-05,
      "step": 99363,
      "training_step_time": 7.147332191467285
    },
    {
      "epoch": 2.4188995361328126e-05,
      "model_forward_time": 1.7442471981048584,
      "step": 99364
    },
    {
      "epoch": 2.4188995361328126e-05,
      "step": 99364,
      "training_step_time": 7.294766426086426
    },
    {
      "epoch": 2.418994903564453e-05,
      "model_forward_time": 1.7466070652008057,
      "step": 99365
    },
    {
      "epoch": 2.418994903564453e-05,
      "step": 99365,
      "training_step_time": 8.616482257843018
    },
    {
      "epoch": 2.4190902709960937e-05,
      "model_forward_time": 2.422823190689087,
      "step": 99366
    },
    {
      "epoch": 2.4190902709960937e-05,
      "step": 99366,
      "training_step_time": 8.28187870979309
    },
    {
      "epoch": 2.4191856384277345e-05,
      "model_forward_time": 1.744201898574829,
      "step": 99367
    },
    {
      "epoch": 2.4191856384277345e-05,
      "step": 99367,
      "training_step_time": 8.922227621078491
    },
    {
      "epoch": 2.419281005859375e-05,
      "model_forward_time": 1.7499282360076904,
      "step": 99368
    },
    {
      "epoch": 2.419281005859375e-05,
      "step": 99368,
      "training_step_time": 8.404453992843628
    },
    {
      "epoch": 2.4193763732910156e-05,
      "model_forward_time": 1.7400450706481934,
      "step": 99369
    },
    {
      "epoch": 2.4193763732910156e-05,
      "step": 99369,
      "training_step_time": 7.7394044399261475
    },
    {
      "epoch": 2.4194717407226563e-05,
      "grad_norm": 0.09265689551830292,
      "learning_rate": 1.0885177202790697e-09,
      "loss": 0.055,
      "step": 99370
    },
    {
      "epoch": 2.4194717407226563e-05,
      "model_forward_time": 1.7418816089630127,
      "step": 99370
    },
    {
      "dynamics_loss_avg": 0.04180393796414137,
      "epoch": 2.4194717407226563e-05,
      "step": 99370
    },
    {
      "action_loss_avg": 0.002532676715054549,
      "epoch": 2.4194717407226563e-05,
      "step": 99370
    },
    {
      "epoch": 2.4194717407226563e-05,
      "step": 99370,
      "training_step_time": 7.7558958530426025
    },
    {
      "epoch": 2.4195671081542967e-05,
      "model_forward_time": 1.742626667022705,
      "step": 99371
    },
    {
      "epoch": 2.4195671081542967e-05,
      "step": 99371,
      "training_step_time": 7.6293768882751465
    },
    {
      "epoch": 2.4196624755859375e-05,
      "model_forward_time": 1.7495687007904053,
      "step": 99372
    },
    {
      "epoch": 2.4196624755859375e-05,
      "step": 99372,
      "training_step_time": 7.356259822845459
    },
    {
      "epoch": 2.4197578430175782e-05,
      "model_forward_time": 1.7447175979614258,
      "step": 99373
    },
    {
      "epoch": 2.4197578430175782e-05,
      "step": 99373,
      "training_step_time": 7.526801109313965
    },
    {
      "epoch": 2.4198532104492186e-05,
      "model_forward_time": 1.7472710609436035,
      "step": 99374
    },
    {
      "epoch": 2.4198532104492186e-05,
      "step": 99374,
      "training_step_time": 9.923074722290039
    },
    {
      "epoch": 2.4199485778808594e-05,
      "model_forward_time": 2.383155107498169,
      "step": 99375
    },
    {
      "epoch": 2.4199485778808594e-05,
      "step": 99375,
      "training_step_time": 10.91264033317566
    },
    {
      "epoch": 2.4200439453125e-05,
      "model_forward_time": 1.8137383460998535,
      "step": 99376
    },
    {
      "epoch": 2.4200439453125e-05,
      "step": 99376,
      "training_step_time": 9.868643999099731
    },
    {
      "epoch": 2.4201393127441405e-05,
      "model_forward_time": 1.7478508949279785,
      "step": 99377
    },
    {
      "epoch": 2.4201393127441405e-05,
      "step": 99377,
      "training_step_time": 7.704698324203491
    },
    {
      "epoch": 2.4202346801757813e-05,
      "model_forward_time": 1.7517356872558594,
      "step": 99378
    },
    {
      "epoch": 2.4202346801757813e-05,
      "step": 99378,
      "training_step_time": 8.422080278396606
    },
    {
      "epoch": 2.420330047607422e-05,
      "model_forward_time": 1.757023811340332,
      "step": 99379
    },
    {
      "epoch": 2.420330047607422e-05,
      "step": 99379,
      "training_step_time": 7.631852865219116
    },
    {
      "epoch": 2.4204254150390624e-05,
      "grad_norm": 0.08602079749107361,
      "learning_rate": 1.0542909554789315e-09,
      "loss": 0.0565,
      "step": 99380
    },
    {
      "epoch": 2.4204254150390624e-05,
      "model_forward_time": 1.779585599899292,
      "step": 99380
    },
    {
      "dynamics_loss_avg": 0.05954632461071015,
      "epoch": 2.4204254150390624e-05,
      "step": 99380
    },
    {
      "action_loss_avg": 0.001264879445079714,
      "epoch": 2.4204254150390624e-05,
      "step": 99380
    },
    {
      "epoch": 2.4204254150390624e-05,
      "step": 99380,
      "training_step_time": 7.700038909912109
    },
    {
      "epoch": 2.420520782470703e-05,
      "model_forward_time": 1.7649023532867432,
      "step": 99381
    },
    {
      "epoch": 2.420520782470703e-05,
      "step": 99381,
      "training_step_time": 8.570430278778076
    },
    {
      "epoch": 2.420616149902344e-05,
      "model_forward_time": 1.7705039978027344,
      "step": 99382
    },
    {
      "epoch": 2.420616149902344e-05,
      "step": 99382,
      "training_step_time": 7.410834074020386
    },
    {
      "epoch": 2.4207115173339843e-05,
      "model_forward_time": 1.7689757347106934,
      "step": 99383
    },
    {
      "epoch": 2.4207115173339843e-05,
      "step": 99383,
      "training_step_time": 7.813455581665039
    },
    {
      "epoch": 2.420806884765625e-05,
      "model_forward_time": 1.7643895149230957,
      "step": 99384
    },
    {
      "epoch": 2.420806884765625e-05,
      "step": 99384,
      "training_step_time": 9.041954278945923
    },
    {
      "epoch": 2.4209022521972658e-05,
      "model_forward_time": 2.4652247428894043,
      "step": 99385
    },
    {
      "epoch": 2.4209022521972658e-05,
      "step": 99385,
      "training_step_time": 7.811175107955933
    },
    {
      "epoch": 2.4209976196289062e-05,
      "model_forward_time": 1.7581536769866943,
      "step": 99386
    },
    {
      "epoch": 2.4209976196289062e-05,
      "step": 99386,
      "training_step_time": 8.93447470664978
    },
    {
      "epoch": 2.421092987060547e-05,
      "model_forward_time": 1.7568824291229248,
      "step": 99387
    },
    {
      "epoch": 2.421092987060547e-05,
      "step": 99387,
      "training_step_time": 7.825915336608887
    },
    {
      "epoch": 2.4211883544921877e-05,
      "model_forward_time": 1.0563762187957764,
      "step": 99388
    },
    {
      "epoch": 2.4211883544921877e-05,
      "step": 99388,
      "training_step_time": 7.676249742507935
    },
    {
      "epoch": 2.421283721923828e-05,
      "model_forward_time": 1.768183946609497,
      "step": 99389
    },
    {
      "epoch": 2.421283721923828e-05,
      "step": 99389,
      "training_step_time": 7.72331428527832
    },
    {
      "epoch": 2.4213790893554688e-05,
      "grad_norm": 0.06865804642438889,
      "learning_rate": 1.0206108678656457e-09,
      "loss": 0.0538,
      "step": 99390
    },
    {
      "epoch": 2.4213790893554688e-05,
      "model_forward_time": 1.4079337120056152,
      "step": 99390
    },
    {
      "dynamics_loss_avg": 0.05649937503039837,
      "epoch": 2.4213790893554688e-05,
      "step": 99390
    },
    {
      "action_loss_avg": 0.0012789468455594033,
      "epoch": 2.4213790893554688e-05,
      "step": 99390
    },
    {
      "epoch": 2.4213790893554688e-05,
      "step": 99390,
      "training_step_time": 7.477494478225708
    },
    {
      "epoch": 2.4214744567871092e-05,
      "model_forward_time": 1.796658992767334,
      "step": 99391
    },
    {
      "epoch": 2.4214744567871092e-05,
      "step": 99391,
      "training_step_time": 7.800955772399902
    },
    {
      "epoch": 2.42156982421875e-05,
      "model_forward_time": 1.7663161754608154,
      "step": 99392
    },
    {
      "epoch": 2.42156982421875e-05,
      "step": 99392,
      "training_step_time": 8.9187753200531
    },
    {
      "epoch": 2.4216651916503907e-05,
      "model_forward_time": 1.7754032611846924,
      "step": 99393
    },
    {
      "epoch": 2.4216651916503907e-05,
      "step": 99393,
      "training_step_time": 8.061235427856445
    },
    {
      "epoch": 2.421760559082031e-05,
      "model_forward_time": 2.4019429683685303,
      "step": 99394
    },
    {
      "epoch": 2.421760559082031e-05,
      "step": 99394,
      "training_step_time": 7.904470443725586
    },
    {
      "epoch": 2.421855926513672e-05,
      "model_forward_time": 1.7617406845092773,
      "step": 99395
    },
    {
      "epoch": 2.421855926513672e-05,
      "step": 99395,
      "training_step_time": 8.578500270843506
    },
    {
      "epoch": 2.4219512939453126e-05,
      "model_forward_time": 1.7774503231048584,
      "step": 99396
    },
    {
      "epoch": 2.4219512939453126e-05,
      "step": 99396,
      "training_step_time": 8.053253173828125
    },
    {
      "epoch": 2.422046661376953e-05,
      "model_forward_time": 1.7692375183105469,
      "step": 99397
    },
    {
      "epoch": 2.422046661376953e-05,
      "step": 99397,
      "training_step_time": 7.772573471069336
    },
    {
      "epoch": 2.4221420288085937e-05,
      "model_forward_time": 1.771392583847046,
      "step": 99398
    },
    {
      "epoch": 2.4221420288085937e-05,
      "step": 99398,
      "training_step_time": 7.615120887756348
    },
    {
      "epoch": 2.4222373962402345e-05,
      "model_forward_time": 1.770944595336914,
      "step": 99399
    },
    {
      "epoch": 2.4222373962402345e-05,
      "step": 99399,
      "training_step_time": 7.738718748092651
    },
    {
      "epoch": 2.422332763671875e-05,
      "grad_norm": 0.11867707967758179,
      "learning_rate": 9.87477461122377e-10,
      "loss": 0.0545,
      "step": 99400
    },
    {
      "epoch": 2.422332763671875e-05,
      "model_forward_time": 1.770583152770996,
      "step": 99400
    },
    {
      "dynamics_loss_avg": 0.05190849099308252,
      "epoch": 2.422332763671875e-05,
      "step": 99400
    },
    {
      "action_loss_avg": 0.0023814931337255985,
      "epoch": 2.422332763671875e-05,
      "step": 99400
    },
    {
      "epoch": 2.422332763671875e-05,
      "step": 99400,
      "training_step_time": 7.725000381469727
    },
    {
      "epoch": 2.4224281311035156e-05,
      "model_forward_time": 1.7602651119232178,
      "step": 99401
    },
    {
      "epoch": 2.4224281311035156e-05,
      "step": 99401,
      "training_step_time": 8.981766939163208
    },
    {
      "epoch": 2.4225234985351563e-05,
      "model_forward_time": 1.7699189186096191,
      "step": 99402
    },
    {
      "epoch": 2.4225234985351563e-05,
      "step": 99402,
      "training_step_time": 8.009893417358398
    },
    {
      "epoch": 2.4226188659667968e-05,
      "model_forward_time": 2.397383451461792,
      "step": 99403
    },
    {
      "epoch": 2.4226188659667968e-05,
      "step": 99403,
      "training_step_time": 7.858897924423218
    },
    {
      "epoch": 2.4227142333984375e-05,
      "model_forward_time": 1.7702522277832031,
      "step": 99404
    },
    {
      "epoch": 2.4227142333984375e-05,
      "step": 99404,
      "training_step_time": 9.586371183395386
    },
    {
      "epoch": 2.4228096008300782e-05,
      "model_forward_time": 1.776759147644043,
      "step": 99405
    },
    {
      "epoch": 2.4228096008300782e-05,
      "step": 99405,
      "training_step_time": 8.785592794418335
    },
    {
      "epoch": 2.4229049682617186e-05,
      "model_forward_time": 1.7663931846618652,
      "step": 99406
    },
    {
      "epoch": 2.4229049682617186e-05,
      "step": 99406,
      "training_step_time": 8.684669256210327
    },
    {
      "epoch": 2.4230003356933594e-05,
      "model_forward_time": 1.7731940746307373,
      "step": 99407
    },
    {
      "epoch": 2.4230003356933594e-05,
      "step": 99407,
      "training_step_time": 7.638438701629639
    },
    {
      "epoch": 2.423095703125e-05,
      "model_forward_time": 1.7591323852539062,
      "step": 99408
    },
    {
      "epoch": 2.423095703125e-05,
      "step": 99408,
      "training_step_time": 8.423196792602539
    },
    {
      "epoch": 2.4231910705566405e-05,
      "model_forward_time": 1.7712342739105225,
      "step": 99409
    },
    {
      "epoch": 2.4231910705566405e-05,
      "step": 99409,
      "training_step_time": 7.72131609916687
    },
    {
      "epoch": 2.4232864379882813e-05,
      "grad_norm": 0.0865638330578804,
      "learning_rate": 9.548907388728934e-10,
      "loss": 0.0565,
      "step": 99410
    },
    {
      "epoch": 2.4232864379882813e-05,
      "model_forward_time": 1.779876947402954,
      "step": 99410
    },
    {
      "dynamics_loss_avg": 0.04756890069693327,
      "epoch": 2.4232864379882813e-05,
      "step": 99410
    },
    {
      "action_loss_avg": 0.0013035522191785277,
      "epoch": 2.4232864379882813e-05,
      "step": 99410
    },
    {
      "epoch": 2.4232864379882813e-05,
      "step": 99410,
      "training_step_time": 8.725983142852783
    },
    {
      "epoch": 2.423381805419922e-05,
      "model_forward_time": 1.7697017192840576,
      "step": 99411
    },
    {
      "epoch": 2.423381805419922e-05,
      "step": 99411,
      "training_step_time": 7.907377243041992
    },
    {
      "epoch": 2.4234771728515624e-05,
      "model_forward_time": 1.76619553565979,
      "step": 99412
    },
    {
      "epoch": 2.4234771728515624e-05,
      "step": 99412,
      "training_step_time": 9.022174596786499
    },
    {
      "epoch": 2.423572540283203e-05,
      "model_forward_time": 2.423940896987915,
      "step": 99413
    },
    {
      "epoch": 2.423572540283203e-05,
      "step": 99413,
      "training_step_time": 10.104098320007324
    },
    {
      "epoch": 2.423667907714844e-05,
      "model_forward_time": 1.7744042873382568,
      "step": 99414
    },
    {
      "epoch": 2.423667907714844e-05,
      "step": 99414,
      "training_step_time": 8.92489242553711
    },
    {
      "epoch": 2.4237632751464843e-05,
      "model_forward_time": 1.769479513168335,
      "step": 99415
    },
    {
      "epoch": 2.4237632751464843e-05,
      "step": 99415,
      "training_step_time": 8.782610177993774
    },
    {
      "epoch": 2.423858642578125e-05,
      "model_forward_time": 1.7707970142364502,
      "step": 99416
    },
    {
      "epoch": 2.423858642578125e-05,
      "step": 99416,
      "training_step_time": 7.608904838562012
    },
    {
      "epoch": 2.4239540100097658e-05,
      "model_forward_time": 1.771416187286377,
      "step": 99417
    },
    {
      "epoch": 2.4239540100097658e-05,
      "step": 99417,
      "training_step_time": 8.600990533828735
    },
    {
      "epoch": 2.4240493774414062e-05,
      "model_forward_time": 1.7712574005126953,
      "step": 99418
    },
    {
      "epoch": 2.4240493774414062e-05,
      "step": 99418,
      "training_step_time": 8.77703857421875
    },
    {
      "epoch": 2.424144744873047e-05,
      "model_forward_time": 1.77079439163208,
      "step": 99419
    },
    {
      "epoch": 2.424144744873047e-05,
      "step": 99419,
      "training_step_time": 8.182649612426758
    },
    {
      "epoch": 2.4242401123046877e-05,
      "grad_norm": 0.07836408168077469,
      "learning_rate": 9.228507046804558e-10,
      "loss": 0.0552,
      "step": 99420
    },
    {
      "epoch": 2.4242401123046877e-05,
      "model_forward_time": 1.7687902450561523,
      "step": 99420
    },
    {
      "dynamics_loss_avg": 0.06755577605217696,
      "epoch": 2.4242401123046877e-05,
      "step": 99420
    },
    {
      "action_loss_avg": 0.002704746287781745,
      "epoch": 2.4242401123046877e-05,
      "step": 99420
    },
    {
      "epoch": 2.4242401123046877e-05,
      "step": 99420,
      "training_step_time": 7.828824758529663
    },
    {
      "epoch": 2.424335479736328e-05,
      "model_forward_time": 1.7692081928253174,
      "step": 99421
    },
    {
      "epoch": 2.424335479736328e-05,
      "step": 99421,
      "training_step_time": 7.851381063461304
    },
    {
      "epoch": 2.4244308471679688e-05,
      "model_forward_time": 2.4256463050842285,
      "step": 99422
    },
    {
      "epoch": 2.4244308471679688e-05,
      "step": 99422,
      "training_step_time": 8.404074668884277
    },
    {
      "epoch": 2.4245262145996092e-05,
      "model_forward_time": 1.7704577445983887,
      "step": 99423
    },
    {
      "epoch": 2.4245262145996092e-05,
      "step": 99423,
      "training_step_time": 10.327160358428955
    },
    {
      "epoch": 2.42462158203125e-05,
      "model_forward_time": 1.7641892433166504,
      "step": 99424
    },
    {
      "epoch": 2.42462158203125e-05,
      "step": 99424,
      "training_step_time": 8.768294334411621
    },
    {
      "epoch": 2.4247169494628907e-05,
      "model_forward_time": 1.7611961364746094,
      "step": 99425
    },
    {
      "epoch": 2.4247169494628907e-05,
      "step": 99425,
      "training_step_time": 7.493865728378296
    },
    {
      "epoch": 2.424812316894531e-05,
      "model_forward_time": 1.7640135288238525,
      "step": 99426
    },
    {
      "epoch": 2.424812316894531e-05,
      "step": 99426,
      "training_step_time": 7.4314515590667725
    },
    {
      "epoch": 2.424907684326172e-05,
      "model_forward_time": 1.7670326232910156,
      "step": 99427
    },
    {
      "epoch": 2.424907684326172e-05,
      "step": 99427,
      "training_step_time": 7.601204872131348
    },
    {
      "epoch": 2.4250030517578126e-05,
      "model_forward_time": 1.7661094665527344,
      "step": 99428
    },
    {
      "epoch": 2.4250030517578126e-05,
      "step": 99428,
      "training_step_time": 7.601879835128784
    },
    {
      "epoch": 2.425098419189453e-05,
      "model_forward_time": 1.7675707340240479,
      "step": 99429
    },
    {
      "epoch": 2.425098419189453e-05,
      "step": 99429,
      "training_step_time": 7.599069833755493
    },
    {
      "epoch": 2.4251937866210937e-05,
      "grad_norm": 0.10011361539363861,
      "learning_rate": 8.91357362048928e-10,
      "loss": 0.0541,
      "step": 99430
    },
    {
      "epoch": 2.4251937866210937e-05,
      "model_forward_time": 1.7614328861236572,
      "step": 99430
    },
    {
      "dynamics_loss_avg": 0.0404187262058258,
      "epoch": 2.4251937866210937e-05,
      "step": 99430
    },
    {
      "action_loss_avg": 0.0008206787810195237,
      "epoch": 2.4251937866210937e-05,
      "step": 99430
    },
    {
      "epoch": 2.4251937866210937e-05,
      "step": 99430,
      "training_step_time": 9.950109720230103
    },
    {
      "epoch": 2.4252891540527345e-05,
      "model_forward_time": 2.403883695602417,
      "step": 99431
    },
    {
      "epoch": 2.4252891540527345e-05,
      "step": 99431,
      "training_step_time": 10.316052436828613
    },
    {
      "epoch": 2.425384521484375e-05,
      "model_forward_time": 1.7759466171264648,
      "step": 99432
    },
    {
      "epoch": 2.425384521484375e-05,
      "step": 99432,
      "training_step_time": 9.156436920166016
    },
    {
      "epoch": 2.4254798889160156e-05,
      "model_forward_time": 1.7771966457366943,
      "step": 99433
    },
    {
      "epoch": 2.4254798889160156e-05,
      "step": 99433,
      "training_step_time": 7.880841016769409
    },
    {
      "epoch": 2.4255752563476564e-05,
      "model_forward_time": 1.7742152214050293,
      "step": 99434
    },
    {
      "epoch": 2.4255752563476564e-05,
      "step": 99434,
      "training_step_time": 7.6327595710754395
    },
    {
      "epoch": 2.4256706237792968e-05,
      "model_forward_time": 1.7774450778961182,
      "step": 99435
    },
    {
      "epoch": 2.4256706237792968e-05,
      "step": 99435,
      "training_step_time": 7.454764127731323
    },
    {
      "epoch": 2.4257659912109375e-05,
      "model_forward_time": 1.7719166278839111,
      "step": 99436
    },
    {
      "epoch": 2.4257659912109375e-05,
      "step": 99436,
      "training_step_time": 7.597018718719482
    },
    {
      "epoch": 2.4258613586425782e-05,
      "model_forward_time": 1.7723169326782227,
      "step": 99437
    },
    {
      "epoch": 2.4258613586425782e-05,
      "step": 99437,
      "training_step_time": 7.568204164505005
    },
    {
      "epoch": 2.4259567260742186e-05,
      "model_forward_time": 1.775266170501709,
      "step": 99438
    },
    {
      "epoch": 2.4259567260742186e-05,
      "step": 99438,
      "training_step_time": 9.006448030471802
    },
    {
      "epoch": 2.4260520935058594e-05,
      "model_forward_time": 1.773444652557373,
      "step": 99439
    },
    {
      "epoch": 2.4260520935058594e-05,
      "step": 99439,
      "training_step_time": 7.813809633255005
    },
    {
      "epoch": 2.4261474609375e-05,
      "grad_norm": 0.07550844550132751,
      "learning_rate": 8.604107144227769e-10,
      "loss": 0.0532,
      "step": 99440
    },
    {
      "epoch": 2.4261474609375e-05,
      "model_forward_time": 2.427196979522705,
      "step": 99440
    },
    {
      "dynamics_loss_avg": 0.060619466751813886,
      "epoch": 2.4261474609375e-05,
      "step": 99440
    },
    {
      "action_loss_avg": 0.0024440257402602583,
      "epoch": 2.4261474609375e-05,
      "step": 99440
    },
    {
      "epoch": 2.4261474609375e-05,
      "step": 99440,
      "training_step_time": 8.554523229598999
    },
    {
      "epoch": 2.4262428283691405e-05,
      "model_forward_time": 1.774923324584961,
      "step": 99441
    },
    {
      "epoch": 2.4262428283691405e-05,
      "step": 99441,
      "training_step_time": 10.38182020187378
    },
    {
      "epoch": 2.4263381958007813e-05,
      "model_forward_time": 1.7716407775878906,
      "step": 99442
    },
    {
      "epoch": 2.4263381958007813e-05,
      "step": 99442,
      "training_step_time": 7.906733989715576
    },
    {
      "epoch": 2.426433563232422e-05,
      "model_forward_time": 1.7790706157684326,
      "step": 99443
    },
    {
      "epoch": 2.426433563232422e-05,
      "step": 99443,
      "training_step_time": 7.714492321014404
    },
    {
      "epoch": 2.4265289306640624e-05,
      "model_forward_time": 1.7715508937835693,
      "step": 99444
    },
    {
      "epoch": 2.4265289306640624e-05,
      "step": 99444,
      "training_step_time": 7.6457579135894775
    },
    {
      "epoch": 2.426624298095703e-05,
      "model_forward_time": 1.7711937427520752,
      "step": 99445
    },
    {
      "epoch": 2.426624298095703e-05,
      "step": 99445,
      "training_step_time": 7.860816478729248
    },
    {
      "epoch": 2.426719665527344e-05,
      "model_forward_time": 1.7726941108703613,
      "step": 99446
    },
    {
      "epoch": 2.426719665527344e-05,
      "step": 99446,
      "training_step_time": 7.80408787727356
    },
    {
      "epoch": 2.4268150329589843e-05,
      "model_forward_time": 1.7746148109436035,
      "step": 99447
    },
    {
      "epoch": 2.4268150329589843e-05,
      "step": 99447,
      "training_step_time": 7.802913665771484
    },
    {
      "epoch": 2.426910400390625e-05,
      "model_forward_time": 1.3977913856506348,
      "step": 99448
    },
    {
      "epoch": 2.426910400390625e-05,
      "step": 99448,
      "training_step_time": 8.948869228363037
    },
    {
      "epoch": 2.4270057678222658e-05,
      "model_forward_time": 2.432914972305298,
      "step": 99449
    },
    {
      "epoch": 2.4270057678222658e-05,
      "step": 99449,
      "training_step_time": 9.059870481491089
    },
    {
      "epoch": 2.4271011352539062e-05,
      "grad_norm": 0.07473722845315933,
      "learning_rate": 8.300107651859623e-10,
      "loss": 0.0527,
      "step": 99450
    },
    {
      "epoch": 2.4271011352539062e-05,
      "model_forward_time": 1.813338041305542,
      "step": 99450
    },
    {
      "dynamics_loss_avg": 0.05423871520906687,
      "epoch": 2.4271011352539062e-05,
      "step": 99450
    },
    {
      "action_loss_avg": 0.002186521899420768,
      "epoch": 2.4271011352539062e-05,
      "step": 99450
    },
    {
      "epoch": 2.4271011352539062e-05,
      "step": 99450,
      "training_step_time": 9.47535514831543
    },
    {
      "epoch": 2.427196502685547e-05,
      "model_forward_time": 1.7678346633911133,
      "step": 99451
    },
    {
      "epoch": 2.427196502685547e-05,
      "step": 99451,
      "training_step_time": 9.782784700393677
    },
    {
      "epoch": 2.4272918701171877e-05,
      "model_forward_time": 1.7642052173614502,
      "step": 99452
    },
    {
      "epoch": 2.4272918701171877e-05,
      "step": 99452,
      "training_step_time": 7.827834606170654
    },
    {
      "epoch": 2.427387237548828e-05,
      "model_forward_time": 1.7658348083496094,
      "step": 99453
    },
    {
      "epoch": 2.427387237548828e-05,
      "step": 99453,
      "training_step_time": 7.686783790588379
    },
    {
      "epoch": 2.4274826049804688e-05,
      "model_forward_time": 1.7654712200164795,
      "step": 99454
    },
    {
      "epoch": 2.4274826049804688e-05,
      "step": 99454,
      "training_step_time": 7.788793087005615
    },
    {
      "epoch": 2.4275779724121092e-05,
      "model_forward_time": 1.7681770324707031,
      "step": 99455
    },
    {
      "epoch": 2.4275779724121092e-05,
      "step": 99455,
      "training_step_time": 7.746325969696045
    },
    {
      "epoch": 2.42767333984375e-05,
      "model_forward_time": 1.7741856575012207,
      "step": 99456
    },
    {
      "epoch": 2.42767333984375e-05,
      "step": 99456,
      "training_step_time": 8.648823976516724
    },
    {
      "epoch": 2.4277687072753907e-05,
      "model_forward_time": 1.7616045475006104,
      "step": 99457
    },
    {
      "epoch": 2.4277687072753907e-05,
      "step": 99457,
      "training_step_time": 8.064660549163818
    },
    {
      "epoch": 2.427864074707031e-05,
      "model_forward_time": 1.7670695781707764,
      "step": 99458
    },
    {
      "epoch": 2.427864074707031e-05,
      "step": 99458,
      "training_step_time": 9.271229982376099
    },
    {
      "epoch": 2.427959442138672e-05,
      "model_forward_time": 2.4240734577178955,
      "step": 99459
    },
    {
      "epoch": 2.427959442138672e-05,
      "step": 99459,
      "training_step_time": 9.475640296936035
    },
    {
      "epoch": 2.4280548095703126e-05,
      "grad_norm": 0.07424420863389969,
      "learning_rate": 8.001575176630471e-10,
      "loss": 0.053,
      "step": 99460
    },
    {
      "epoch": 2.4280548095703126e-05,
      "model_forward_time": 1.7579526901245117,
      "step": 99460
    },
    {
      "dynamics_loss_avg": 0.049477407895028594,
      "epoch": 2.4280548095703126e-05,
      "step": 99460
    },
    {
      "action_loss_avg": 0.0017292048258241266,
      "epoch": 2.4280548095703126e-05,
      "step": 99460
    },
    {
      "epoch": 2.4280548095703126e-05,
      "step": 99460,
      "training_step_time": 8.728512048721313
    },
    {
      "epoch": 2.428150177001953e-05,
      "model_forward_time": 1.7672247886657715,
      "step": 99461
    },
    {
      "epoch": 2.428150177001953e-05,
      "step": 99461,
      "training_step_time": 7.826288223266602
    },
    {
      "epoch": 2.4282455444335937e-05,
      "model_forward_time": 1.7643606662750244,
      "step": 99462
    },
    {
      "epoch": 2.4282455444335937e-05,
      "step": 99462,
      "training_step_time": 7.817201375961304
    },
    {
      "epoch": 2.4283409118652345e-05,
      "model_forward_time": 1.7673900127410889,
      "step": 99463
    },
    {
      "epoch": 2.4283409118652345e-05,
      "step": 99463,
      "training_step_time": 7.60882830619812
    },
    {
      "epoch": 2.428436279296875e-05,
      "model_forward_time": 1.7794511318206787,
      "step": 99464
    },
    {
      "epoch": 2.428436279296875e-05,
      "step": 99464,
      "training_step_time": 7.861310005187988
    },
    {
      "epoch": 2.4285316467285156e-05,
      "model_forward_time": 1.7720861434936523,
      "step": 99465
    },
    {
      "epoch": 2.4285316467285156e-05,
      "step": 99465,
      "training_step_time": 7.548194169998169
    },
    {
      "epoch": 2.4286270141601564e-05,
      "model_forward_time": 1.7598254680633545,
      "step": 99466
    },
    {
      "epoch": 2.4286270141601564e-05,
      "step": 99466,
      "training_step_time": 8.085034608840942
    },
    {
      "epoch": 2.4287223815917968e-05,
      "model_forward_time": 1.7633519172668457,
      "step": 99467
    },
    {
      "epoch": 2.4287223815917968e-05,
      "step": 99467,
      "training_step_time": 10.266022682189941
    },
    {
      "epoch": 2.4288177490234375e-05,
      "model_forward_time": 2.038912773132324,
      "step": 99468
    },
    {
      "epoch": 2.4288177490234375e-05,
      "step": 99468,
      "training_step_time": 10.668374300003052
    },
    {
      "epoch": 2.4289131164550782e-05,
      "model_forward_time": 1.7690074443817139,
      "step": 99469
    },
    {
      "epoch": 2.4289131164550782e-05,
      "step": 99469,
      "training_step_time": 9.948562860488892
    },
    {
      "epoch": 2.4290084838867186e-05,
      "grad_norm": 0.13773296773433685,
      "learning_rate": 7.708509751186422e-10,
      "loss": 0.0525,
      "step": 99470
    },
    {
      "epoch": 2.4290084838867186e-05,
      "model_forward_time": 1.762355089187622,
      "step": 99470
    },
    {
      "dynamics_loss_avg": 0.06619811877608299,
      "epoch": 2.4290084838867186e-05,
      "step": 99470
    },
    {
      "action_loss_avg": 0.00162740612286143,
      "epoch": 2.4290084838867186e-05,
      "step": 99470
    },
    {
      "epoch": 2.4290084838867186e-05,
      "step": 99470,
      "training_step_time": 8.466574430465698
    },
    {
      "epoch": 2.4291038513183594e-05,
      "model_forward_time": 1.7620253562927246,
      "step": 99471
    },
    {
      "epoch": 2.4291038513183594e-05,
      "step": 99471,
      "training_step_time": 7.771652460098267
    },
    {
      "epoch": 2.42919921875e-05,
      "model_forward_time": 1.7696218490600586,
      "step": 99472
    },
    {
      "epoch": 2.42919921875e-05,
      "step": 99472,
      "training_step_time": 7.6488988399505615
    },
    {
      "epoch": 2.4292945861816405e-05,
      "model_forward_time": 1.7655627727508545,
      "step": 99473
    },
    {
      "epoch": 2.4292945861816405e-05,
      "step": 99473,
      "training_step_time": 7.771908283233643
    },
    {
      "epoch": 2.4293899536132813e-05,
      "model_forward_time": 1.7766234874725342,
      "step": 99474
    },
    {
      "epoch": 2.4293899536132813e-05,
      "step": 99474,
      "training_step_time": 7.403861045837402
    },
    {
      "epoch": 2.429485321044922e-05,
      "model_forward_time": 1.7712361812591553,
      "step": 99475
    },
    {
      "epoch": 2.429485321044922e-05,
      "step": 99475,
      "training_step_time": 9.351812601089478
    },
    {
      "epoch": 2.4295806884765624e-05,
      "model_forward_time": 1.7585539817810059,
      "step": 99476
    },
    {
      "epoch": 2.4295806884765624e-05,
      "step": 99476,
      "training_step_time": 7.9330103397369385
    },
    {
      "epoch": 2.429676055908203e-05,
      "model_forward_time": 2.405118942260742,
      "step": 99477
    },
    {
      "epoch": 2.429676055908203e-05,
      "step": 99477,
      "training_step_time": 9.35303521156311
    },
    {
      "epoch": 2.429771423339844e-05,
      "model_forward_time": 1.7599380016326904,
      "step": 99478
    },
    {
      "epoch": 2.429771423339844e-05,
      "step": 99478,
      "training_step_time": 9.70037579536438
    },
    {
      "epoch": 2.4298667907714843e-05,
      "model_forward_time": 1.7615773677825928,
      "step": 99479
    },
    {
      "epoch": 2.4298667907714843e-05,
      "step": 99479,
      "training_step_time": 7.828280448913574
    },
    {
      "epoch": 2.429962158203125e-05,
      "grad_norm": 0.09183045476675034,
      "learning_rate": 7.420911407579612e-10,
      "loss": 0.0526,
      "step": 99480
    },
    {
      "epoch": 2.429962158203125e-05,
      "model_forward_time": 1.8019704818725586,
      "step": 99480
    },
    {
      "dynamics_loss_avg": 0.05107800234109163,
      "epoch": 2.429962158203125e-05,
      "step": 99480
    },
    {
      "action_loss_avg": 0.0015029682457679884,
      "epoch": 2.429962158203125e-05,
      "step": 99480
    },
    {
      "epoch": 2.429962158203125e-05,
      "step": 99480,
      "training_step_time": 8.852973461151123
    },
    {
      "epoch": 2.4300575256347658e-05,
      "model_forward_time": 1.7630088329315186,
      "step": 99481
    },
    {
      "epoch": 2.4300575256347658e-05,
      "step": 99481,
      "training_step_time": 8.258605718612671
    },
    {
      "epoch": 2.4301528930664062e-05,
      "model_forward_time": 1.7648344039916992,
      "step": 99482
    },
    {
      "epoch": 2.4301528930664062e-05,
      "step": 99482,
      "training_step_time": 7.864374160766602
    },
    {
      "epoch": 2.430248260498047e-05,
      "model_forward_time": 1.7660431861877441,
      "step": 99483
    },
    {
      "epoch": 2.430248260498047e-05,
      "step": 99483,
      "training_step_time": 7.793810129165649
    },
    {
      "epoch": 2.4303436279296877e-05,
      "model_forward_time": 1.769395112991333,
      "step": 99484
    },
    {
      "epoch": 2.4303436279296877e-05,
      "step": 99484,
      "training_step_time": 9.92972731590271
    },
    {
      "epoch": 2.430438995361328e-05,
      "model_forward_time": 1.7610673904418945,
      "step": 99485
    },
    {
      "epoch": 2.430438995361328e-05,
      "step": 99485,
      "training_step_time": 8.72784948348999
    },
    {
      "epoch": 2.4305343627929688e-05,
      "model_forward_time": 2.4110007286071777,
      "step": 99486
    },
    {
      "epoch": 2.4305343627929688e-05,
      "step": 99486,
      "training_step_time": 8.997107982635498
    },
    {
      "epoch": 2.4306297302246092e-05,
      "model_forward_time": 1.7670619487762451,
      "step": 99487
    },
    {
      "epoch": 2.4306297302246092e-05,
      "step": 99487,
      "training_step_time": 7.830493927001953
    },
    {
      "epoch": 2.43072509765625e-05,
      "model_forward_time": 1.7716498374938965,
      "step": 99488
    },
    {
      "epoch": 2.43072509765625e-05,
      "step": 99488,
      "training_step_time": 8.725973129272461
    },
    {
      "epoch": 2.4308204650878907e-05,
      "model_forward_time": 1.7810239791870117,
      "step": 99489
    },
    {
      "epoch": 2.4308204650878907e-05,
      "step": 99489,
      "training_step_time": 7.79725193977356
    },
    {
      "epoch": 2.430915832519531e-05,
      "grad_norm": 0.07686490565538406,
      "learning_rate": 7.138780177257109e-10,
      "loss": 0.0521,
      "step": 99490
    },
    {
      "epoch": 2.430915832519531e-05,
      "model_forward_time": 1.7799475193023682,
      "step": 99490
    },
    {
      "dynamics_loss_avg": 0.08052201047539712,
      "epoch": 2.430915832519531e-05,
      "step": 99490
    },
    {
      "action_loss_avg": 0.0032449153950437903,
      "epoch": 2.430915832519531e-05,
      "step": 99490
    },
    {
      "epoch": 2.430915832519531e-05,
      "step": 99490,
      "training_step_time": 7.607195615768433
    },
    {
      "epoch": 2.431011199951172e-05,
      "model_forward_time": 1.7753567695617676,
      "step": 99491
    },
    {
      "epoch": 2.431011199951172e-05,
      "step": 99491,
      "training_step_time": 8.899869680404663
    },
    {
      "epoch": 2.4311065673828126e-05,
      "model_forward_time": 1.7699825763702393,
      "step": 99492
    },
    {
      "epoch": 2.4311065673828126e-05,
      "step": 99492,
      "training_step_time": 7.753453493118286
    },
    {
      "epoch": 2.431201934814453e-05,
      "model_forward_time": 1.776888370513916,
      "step": 99493
    },
    {
      "epoch": 2.431201934814453e-05,
      "step": 99493,
      "training_step_time": 10.25096607208252
    },
    {
      "epoch": 2.4312973022460937e-05,
      "model_forward_time": 1.7543702125549316,
      "step": 99494
    },
    {
      "epoch": 2.4312973022460937e-05,
      "step": 99494,
      "training_step_time": 8.020360946655273
    },
    {
      "epoch": 2.4313926696777345e-05,
      "model_forward_time": 1.7840008735656738,
      "step": 99495
    },
    {
      "epoch": 2.4313926696777345e-05,
      "step": 99495,
      "training_step_time": 9.104853868484497
    },
    {
      "epoch": 2.431488037109375e-05,
      "model_forward_time": 2.396580934524536,
      "step": 99496
    },
    {
      "epoch": 2.431488037109375e-05,
      "step": 99496,
      "training_step_time": 8.023622989654541
    },
    {
      "epoch": 2.4315834045410156e-05,
      "model_forward_time": 1.757319450378418,
      "step": 99497
    },
    {
      "epoch": 2.4315834045410156e-05,
      "step": 99497,
      "training_step_time": 8.245900630950928
    },
    {
      "epoch": 2.4316787719726564e-05,
      "model_forward_time": 1.762925148010254,
      "step": 99498
    },
    {
      "epoch": 2.4316787719726564e-05,
      "step": 99498,
      "training_step_time": 7.7275731563568115
    },
    {
      "epoch": 2.4317741394042968e-05,
      "model_forward_time": 1.7654376029968262,
      "step": 99499
    },
    {
      "epoch": 2.4317741394042968e-05,
      "step": 99499,
      "training_step_time": 7.823162078857422
    },
    {
      "epoch": 2.4318695068359375e-05,
      "grad_norm": 0.07124792784452438,
      "learning_rate": 6.862116091072013e-10,
      "loss": 0.0521,
      "step": 99500
    },
    {
      "epoch": 2.4318695068359375e-05,
      "model_forward_time": 1.7554612159729004,
      "step": 99500
    },
    {
      "dynamics_loss_avg": 0.05294565642252565,
      "epoch": 2.4318695068359375e-05,
      "step": 99500
    },
    {
      "action_loss_avg": 0.002140243782196194,
      "epoch": 2.4318695068359375e-05,
      "step": 99500
    },
    {
      "epoch": 2.4318695068359375e-05,
      "step": 99500,
      "training_step_time": 7.534756898880005
    },
    {
      "epoch": 2.4319648742675782e-05,
      "model_forward_time": 1.7593514919281006,
      "step": 99501
    },
    {
      "epoch": 2.4319648742675782e-05,
      "step": 99501,
      "training_step_time": 7.752913475036621
    },
    {
      "epoch": 2.4320602416992186e-05,
      "model_forward_time": 1.750579595565796,
      "step": 99502
    },
    {
      "epoch": 2.4320602416992186e-05,
      "step": 99502,
      "training_step_time": 8.784517765045166
    },
    {
      "epoch": 2.4321556091308594e-05,
      "model_forward_time": 1.7563996315002441,
      "step": 99503
    },
    {
      "epoch": 2.4321556091308594e-05,
      "step": 99503,
      "training_step_time": 7.988574028015137
    },
    {
      "epoch": 2.4322509765625e-05,
      "model_forward_time": 1.7627949714660645,
      "step": 99504
    },
    {
      "epoch": 2.4322509765625e-05,
      "step": 99504,
      "training_step_time": 9.4206383228302
    },
    {
      "epoch": 2.4323463439941405e-05,
      "model_forward_time": 2.4091649055480957,
      "step": 99505
    },
    {
      "epoch": 2.4323463439941405e-05,
      "step": 99505,
      "training_step_time": 7.947705507278442
    },
    {
      "epoch": 2.4324417114257813e-05,
      "model_forward_time": 1.7535858154296875,
      "step": 99506
    },
    {
      "epoch": 2.4324417114257813e-05,
      "step": 99506,
      "training_step_time": 8.101341724395752
    },
    {
      "epoch": 2.432537078857422e-05,
      "model_forward_time": 1.7558388710021973,
      "step": 99507
    },
    {
      "epoch": 2.432537078857422e-05,
      "step": 99507,
      "training_step_time": 8.593770503997803
    },
    {
      "epoch": 2.4326324462890624e-05,
      "model_forward_time": 1.7487096786499023,
      "step": 99508
    },
    {
      "epoch": 2.4326324462890624e-05,
      "step": 99508,
      "training_step_time": 8.703510046005249
    },
    {
      "epoch": 2.432727813720703e-05,
      "model_forward_time": 1.751310110092163,
      "step": 99509
    },
    {
      "epoch": 2.432727813720703e-05,
      "step": 99509,
      "training_step_time": 7.724238872528076
    },
    {
      "epoch": 2.432823181152344e-05,
      "grad_norm": 0.07887484133243561,
      "learning_rate": 6.59091917928345e-10,
      "loss": 0.0522,
      "step": 99510
    },
    {
      "epoch": 2.432823181152344e-05,
      "model_forward_time": 1.7447021007537842,
      "step": 99510
    },
    {
      "dynamics_loss_avg": 0.06637422051280736,
      "epoch": 2.432823181152344e-05,
      "step": 99510
    },
    {
      "action_loss_avg": 0.0018932991952169687,
      "epoch": 2.432823181152344e-05,
      "step": 99510
    },
    {
      "epoch": 2.432823181152344e-05,
      "step": 99510,
      "training_step_time": 7.74319052696228
    },
    {
      "epoch": 2.4329185485839843e-05,
      "model_forward_time": 1.7497038841247559,
      "step": 99511
    },
    {
      "epoch": 2.4329185485839843e-05,
      "step": 99511,
      "training_step_time": 7.728621482849121
    },
    {
      "epoch": 2.433013916015625e-05,
      "model_forward_time": 1.7501261234283447,
      "step": 99512
    },
    {
      "epoch": 2.433013916015625e-05,
      "step": 99512,
      "training_step_time": 8.052765846252441
    },
    {
      "epoch": 2.4331092834472658e-05,
      "model_forward_time": 1.7444994449615479,
      "step": 99513
    },
    {
      "epoch": 2.4331092834472658e-05,
      "step": 99513,
      "training_step_time": 9.675167083740234
    },
    {
      "epoch": 2.4332046508789062e-05,
      "model_forward_time": 2.369145154953003,
      "step": 99514
    },
    {
      "epoch": 2.4332046508789062e-05,
      "step": 99514,
      "training_step_time": 8.04396939277649
    },
    {
      "epoch": 2.433300018310547e-05,
      "model_forward_time": 1.7859675884246826,
      "step": 99515
    },
    {
      "epoch": 2.433300018310547e-05,
      "step": 99515,
      "training_step_time": 9.140716075897217
    },
    {
      "epoch": 2.4333953857421873e-05,
      "model_forward_time": 1.7480719089508057,
      "step": 99516
    },
    {
      "epoch": 2.4333953857421873e-05,
      "step": 99516,
      "training_step_time": 7.8605921268463135
    },
    {
      "epoch": 2.433490753173828e-05,
      "model_forward_time": 1.75449800491333,
      "step": 99517
    },
    {
      "epoch": 2.433490753173828e-05,
      "step": 99517,
      "training_step_time": 8.492160320281982
    },
    {
      "epoch": 2.4335861206054688e-05,
      "model_forward_time": 1.75260591506958,
      "step": 99518
    },
    {
      "epoch": 2.4335861206054688e-05,
      "step": 99518,
      "training_step_time": 7.560181379318237
    },
    {
      "epoch": 2.4336814880371092e-05,
      "model_forward_time": 1.746370553970337,
      "step": 99519
    },
    {
      "epoch": 2.4336814880371092e-05,
      "step": 99519,
      "training_step_time": 7.9128334522247314
    },
    {
      "epoch": 2.43377685546875e-05,
      "grad_norm": 0.07832921296358109,
      "learning_rate": 6.32518947155103e-10,
      "loss": 0.0526,
      "step": 99520
    },
    {
      "epoch": 2.43377685546875e-05,
      "model_forward_time": 1.7473573684692383,
      "step": 99520
    },
    {
      "dynamics_loss_avg": 0.048938949406147,
      "epoch": 2.43377685546875e-05,
      "step": 99520
    },
    {
      "action_loss_avg": 0.0013249526025902014,
      "epoch": 2.43377685546875e-05,
      "step": 99520
    },
    {
      "epoch": 2.43377685546875e-05,
      "step": 99520,
      "training_step_time": 7.571058034896851
    },
    {
      "epoch": 2.4338722229003907e-05,
      "model_forward_time": 1.7454733848571777,
      "step": 99521
    },
    {
      "epoch": 2.4338722229003907e-05,
      "step": 99521,
      "training_step_time": 7.9325852394104
    },
    {
      "epoch": 2.433967590332031e-05,
      "model_forward_time": 1.7463595867156982,
      "step": 99522
    },
    {
      "epoch": 2.433967590332031e-05,
      "step": 99522,
      "training_step_time": 7.7690184116363525
    },
    {
      "epoch": 2.434062957763672e-05,
      "model_forward_time": 2.374997615814209,
      "step": 99523
    },
    {
      "epoch": 2.434062957763672e-05,
      "step": 99523,
      "training_step_time": 10.044374227523804
    },
    {
      "epoch": 2.4341583251953126e-05,
      "model_forward_time": 1.7615711688995361,
      "step": 99524
    },
    {
      "epoch": 2.4341583251953126e-05,
      "step": 99524,
      "training_step_time": 9.176194429397583
    },
    {
      "epoch": 2.434253692626953e-05,
      "model_forward_time": 1.749621868133545,
      "step": 99525
    },
    {
      "epoch": 2.434253692626953e-05,
      "step": 99525,
      "training_step_time": 8.043270111083984
    },
    {
      "epoch": 2.4343490600585937e-05,
      "model_forward_time": 1.7435173988342285,
      "step": 99526
    },
    {
      "epoch": 2.4343490600585937e-05,
      "step": 99526,
      "training_step_time": 9.898120880126953
    },
    {
      "epoch": 2.4344444274902345e-05,
      "model_forward_time": 1.750065803527832,
      "step": 99527
    },
    {
      "epoch": 2.4344444274902345e-05,
      "step": 99527,
      "training_step_time": 7.69677472114563
    },
    {
      "epoch": 2.434539794921875e-05,
      "model_forward_time": 1.7472124099731445,
      "step": 99528
    },
    {
      "epoch": 2.434539794921875e-05,
      "step": 99528,
      "training_step_time": 7.818012475967407
    },
    {
      "epoch": 2.4346351623535156e-05,
      "model_forward_time": 1.7492475509643555,
      "step": 99529
    },
    {
      "epoch": 2.4346351623535156e-05,
      "step": 99529,
      "training_step_time": 7.80818510055542
    },
    {
      "epoch": 2.4347305297851564e-05,
      "grad_norm": 0.0774156004190445,
      "learning_rate": 6.064926996929288e-10,
      "loss": 0.053,
      "step": 99530
    },
    {
      "epoch": 2.4347305297851564e-05,
      "model_forward_time": 1.744027853012085,
      "step": 99530
    },
    {
      "dynamics_loss_avg": 0.055598579719662665,
      "epoch": 2.4347305297851564e-05,
      "step": 99530
    },
    {
      "action_loss_avg": 0.002081418572925031,
      "epoch": 2.4347305297851564e-05,
      "step": 99530
    },
    {
      "epoch": 2.4347305297851564e-05,
      "step": 99530,
      "training_step_time": 7.990668535232544
    },
    {
      "epoch": 2.4348258972167968e-05,
      "model_forward_time": 1.7620303630828857,
      "step": 99531
    },
    {
      "epoch": 2.4348258972167968e-05,
      "step": 99531,
      "training_step_time": 8.798077821731567
    },
    {
      "epoch": 2.4349212646484375e-05,
      "model_forward_time": 2.380671501159668,
      "step": 99532
    },
    {
      "epoch": 2.4349212646484375e-05,
      "step": 99532,
      "training_step_time": 8.934748888015747
    },
    {
      "epoch": 2.4350166320800782e-05,
      "model_forward_time": 1.763282060623169,
      "step": 99533
    },
    {
      "epoch": 2.4350166320800782e-05,
      "step": 99533,
      "training_step_time": 9.22804307937622
    },
    {
      "epoch": 2.4351119995117186e-05,
      "model_forward_time": 1.746959924697876,
      "step": 99534
    },
    {
      "epoch": 2.4351119995117186e-05,
      "step": 99534,
      "training_step_time": 8.985606670379639
    },
    {
      "epoch": 2.4352073669433594e-05,
      "model_forward_time": 1.7568697929382324,
      "step": 99535
    },
    {
      "epoch": 2.4352073669433594e-05,
      "step": 99535,
      "training_step_time": 7.767298698425293
    },
    {
      "epoch": 2.435302734375e-05,
      "model_forward_time": 1.7511134147644043,
      "step": 99536
    },
    {
      "epoch": 2.435302734375e-05,
      "step": 99536,
      "training_step_time": 7.795464038848877
    },
    {
      "epoch": 2.4353981018066405e-05,
      "model_forward_time": 1.7525417804718018,
      "step": 99537
    },
    {
      "epoch": 2.4353981018066405e-05,
      "step": 99537,
      "training_step_time": 8.55862021446228
    },
    {
      "epoch": 2.4354934692382813e-05,
      "model_forward_time": 1.7515010833740234,
      "step": 99538
    },
    {
      "epoch": 2.4354934692382813e-05,
      "step": 99538,
      "training_step_time": 7.771550416946411
    },
    {
      "epoch": 2.435588836669922e-05,
      "model_forward_time": 1.7555677890777588,
      "step": 99539
    },
    {
      "epoch": 2.435588836669922e-05,
      "step": 99539,
      "training_step_time": 8.719288349151611
    },
    {
      "epoch": 2.4356842041015624e-05,
      "grad_norm": 0.0702003613114357,
      "learning_rate": 5.810131783884342e-10,
      "loss": 0.054,
      "step": 99540
    },
    {
      "epoch": 2.4356842041015624e-05,
      "model_forward_time": 1.7623226642608643,
      "step": 99540
    },
    {
      "dynamics_loss_avg": 0.06125400234013796,
      "epoch": 2.4356842041015624e-05,
      "step": 99540
    },
    {
      "action_loss_avg": 0.0016151522693689912,
      "epoch": 2.4356842041015624e-05,
      "step": 99540
    },
    {
      "epoch": 2.4356842041015624e-05,
      "step": 99540,
      "training_step_time": 8.6361825466156
    },
    {
      "epoch": 2.435779571533203e-05,
      "model_forward_time": 1.7804090976715088,
      "step": 99541
    },
    {
      "epoch": 2.435779571533203e-05,
      "step": 99541,
      "training_step_time": 8.597047328948975
    },
    {
      "epoch": 2.435874938964844e-05,
      "model_forward_time": 1.815241813659668,
      "step": 99542
    },
    {
      "epoch": 2.435874938964844e-05,
      "step": 99542,
      "training_step_time": 7.896482944488525
    },
    {
      "epoch": 2.4359703063964843e-05,
      "model_forward_time": 1.7758495807647705,
      "step": 99543
    },
    {
      "epoch": 2.4359703063964843e-05,
      "step": 99543,
      "training_step_time": 8.807949304580688
    },
    {
      "epoch": 2.436065673828125e-05,
      "model_forward_time": 1.7742128372192383,
      "step": 99544
    },
    {
      "epoch": 2.436065673828125e-05,
      "step": 99544,
      "training_step_time": 7.71557354927063
    },
    {
      "epoch": 2.4361610412597658e-05,
      "model_forward_time": 1.7687499523162842,
      "step": 99545
    },
    {
      "epoch": 2.4361610412597658e-05,
      "step": 99545,
      "training_step_time": 7.812220335006714
    },
    {
      "epoch": 2.4362564086914062e-05,
      "model_forward_time": 1.7734405994415283,
      "step": 99546
    },
    {
      "epoch": 2.4362564086914062e-05,
      "step": 99546,
      "training_step_time": 7.753037929534912
    },
    {
      "epoch": 2.436351776123047e-05,
      "model_forward_time": 1.7775318622589111,
      "step": 99547
    },
    {
      "epoch": 2.436351776123047e-05,
      "step": 99547,
      "training_step_time": 7.802816152572632
    },
    {
      "epoch": 2.4364471435546873e-05,
      "model_forward_time": 1.7768189907073975,
      "step": 99548
    },
    {
      "epoch": 2.4364471435546873e-05,
      "step": 99548,
      "training_step_time": 7.865402936935425
    },
    {
      "epoch": 2.436542510986328e-05,
      "model_forward_time": 1.777848720550537,
      "step": 99549
    },
    {
      "epoch": 2.436542510986328e-05,
      "step": 99549,
      "training_step_time": 8.099671602249146
    },
    {
      "epoch": 2.4366378784179688e-05,
      "grad_norm": 0.0740995779633522,
      "learning_rate": 5.560803860277242e-10,
      "loss": 0.0542,
      "step": 99550
    },
    {
      "epoch": 2.4366378784179688e-05,
      "model_forward_time": 1.7868223190307617,
      "step": 99550
    },
    {
      "dynamics_loss_avg": 0.044239438697695734,
      "epoch": 2.4366378784179688e-05,
      "step": 99550
    },
    {
      "action_loss_avg": 0.0013342731312150137,
      "epoch": 2.4366378784179688e-05,
      "step": 99550
    },
    {
      "epoch": 2.4366378784179688e-05,
      "step": 99550,
      "training_step_time": 9.777147054672241
    },
    {
      "epoch": 2.4367332458496092e-05,
      "model_forward_time": 2.4099202156066895,
      "step": 99551
    },
    {
      "epoch": 2.4367332458496092e-05,
      "step": 99551,
      "training_step_time": 8.352826833724976
    },
    {
      "epoch": 2.43682861328125e-05,
      "model_forward_time": 1.885373592376709,
      "step": 99552
    },
    {
      "epoch": 2.43682861328125e-05,
      "step": 99552,
      "training_step_time": 7.828651428222656
    },
    {
      "epoch": 2.4369239807128907e-05,
      "model_forward_time": 1.7646002769470215,
      "step": 99553
    },
    {
      "epoch": 2.4369239807128907e-05,
      "step": 99553,
      "training_step_time": 7.846286058425903
    },
    {
      "epoch": 2.437019348144531e-05,
      "model_forward_time": 1.7779686450958252,
      "step": 99554
    },
    {
      "epoch": 2.437019348144531e-05,
      "step": 99554,
      "training_step_time": 8.68393325805664
    },
    {
      "epoch": 2.437114715576172e-05,
      "model_forward_time": 1.7598397731781006,
      "step": 99555
    },
    {
      "epoch": 2.437114715576172e-05,
      "step": 99555,
      "training_step_time": 8.432942152023315
    },
    {
      "epoch": 2.4372100830078126e-05,
      "model_forward_time": 1.7742371559143066,
      "step": 99556
    },
    {
      "epoch": 2.4372100830078126e-05,
      "step": 99556,
      "training_step_time": 8.374633550643921
    },
    {
      "epoch": 2.437305450439453e-05,
      "model_forward_time": 1.7657268047332764,
      "step": 99557
    },
    {
      "epoch": 2.437305450439453e-05,
      "step": 99557,
      "training_step_time": 8.726150751113892
    },
    {
      "epoch": 2.4374008178710937e-05,
      "model_forward_time": 1.7753782272338867,
      "step": 99558
    },
    {
      "epoch": 2.4374008178710937e-05,
      "step": 99558,
      "training_step_time": 8.06334638595581
    },
    {
      "epoch": 2.4374961853027345e-05,
      "model_forward_time": 1.7715723514556885,
      "step": 99559
    },
    {
      "epoch": 2.4374961853027345e-05,
      "step": 99559,
      "training_step_time": 7.948122978210449
    },
    {
      "epoch": 2.437591552734375e-05,
      "grad_norm": 0.06778571009635925,
      "learning_rate": 5.316943253375062e-10,
      "loss": 0.0538,
      "step": 99560
    },
    {
      "epoch": 2.437591552734375e-05,
      "model_forward_time": 2.4118576049804688,
      "step": 99560
    },
    {
      "dynamics_loss_avg": 0.04796733446419239,
      "epoch": 2.437591552734375e-05,
      "step": 99560
    },
    {
      "action_loss_avg": 0.0022049436665838584,
      "epoch": 2.437591552734375e-05,
      "step": 99560
    },
    {
      "epoch": 2.437591552734375e-05,
      "step": 99560,
      "training_step_time": 8.644940853118896
    },
    {
      "epoch": 2.4376869201660156e-05,
      "model_forward_time": 1.876974105834961,
      "step": 99561
    },
    {
      "epoch": 2.4376869201660156e-05,
      "step": 99561,
      "training_step_time": 9.146994352340698
    },
    {
      "epoch": 2.4377822875976564e-05,
      "model_forward_time": 1.7547054290771484,
      "step": 99562
    },
    {
      "epoch": 2.4377822875976564e-05,
      "step": 99562,
      "training_step_time": 8.829498052597046
    },
    {
      "epoch": 2.4378776550292968e-05,
      "model_forward_time": 1.7669892311096191,
      "step": 99563
    },
    {
      "epoch": 2.4378776550292968e-05,
      "step": 99563,
      "training_step_time": 7.673957109451294
    },
    {
      "epoch": 2.4379730224609375e-05,
      "model_forward_time": 1.7579751014709473,
      "step": 99564
    },
    {
      "epoch": 2.4379730224609375e-05,
      "step": 99564,
      "training_step_time": 7.577331781387329
    },
    {
      "epoch": 2.4380683898925782e-05,
      "model_forward_time": 1.7737915515899658,
      "step": 99565
    },
    {
      "epoch": 2.4380683898925782e-05,
      "step": 99565,
      "training_step_time": 7.766943454742432
    },
    {
      "epoch": 2.4381637573242186e-05,
      "model_forward_time": 1.7791640758514404,
      "step": 99566
    },
    {
      "epoch": 2.4381637573242186e-05,
      "step": 99566,
      "training_step_time": 7.874987602233887
    },
    {
      "epoch": 2.4382591247558594e-05,
      "model_forward_time": 1.7696807384490967,
      "step": 99567
    },
    {
      "epoch": 2.4382591247558594e-05,
      "step": 99567,
      "training_step_time": 8.524825811386108
    },
    {
      "epoch": 2.4383544921875e-05,
      "model_forward_time": 1.7517991065979004,
      "step": 99568
    },
    {
      "epoch": 2.4383544921875e-05,
      "step": 99568,
      "training_step_time": 8.88682770729065
    },
    {
      "epoch": 2.4384498596191405e-05,
      "model_forward_time": 2.406747579574585,
      "step": 99569
    },
    {
      "epoch": 2.4384498596191405e-05,
      "step": 99569,
      "training_step_time": 7.995940208435059
    },
    {
      "epoch": 2.4385452270507813e-05,
      "grad_norm": 0.09432259947061539,
      "learning_rate": 5.07854998984536e-10,
      "loss": 0.0541,
      "step": 99570
    },
    {
      "epoch": 2.4385452270507813e-05,
      "model_forward_time": 1.8224573135375977,
      "step": 99570
    },
    {
      "dynamics_loss_avg": 0.04938645800575614,
      "epoch": 2.4385452270507813e-05,
      "step": 99570
    },
    {
      "action_loss_avg": 0.001364497942267917,
      "epoch": 2.4385452270507813e-05,
      "step": 99570
    },
    {
      "epoch": 2.4385452270507813e-05,
      "step": 99570,
      "training_step_time": 7.933436393737793
    },
    {
      "epoch": 2.438640594482422e-05,
      "model_forward_time": 1.7699165344238281,
      "step": 99571
    },
    {
      "epoch": 2.438640594482422e-05,
      "step": 99571,
      "training_step_time": 8.01791524887085
    },
    {
      "epoch": 2.4387359619140624e-05,
      "model_forward_time": 1.7678401470184326,
      "step": 99572
    },
    {
      "epoch": 2.4387359619140624e-05,
      "step": 99572,
      "training_step_time": 8.430313348770142
    },
    {
      "epoch": 2.438831329345703e-05,
      "model_forward_time": 1.7577857971191406,
      "step": 99573
    },
    {
      "epoch": 2.438831329345703e-05,
      "step": 99573,
      "training_step_time": 8.411529064178467
    },
    {
      "epoch": 2.438926696777344e-05,
      "model_forward_time": 1.769759178161621,
      "step": 99574
    },
    {
      "epoch": 2.438926696777344e-05,
      "step": 99574,
      "training_step_time": 7.89612889289856
    },
    {
      "epoch": 2.4390220642089843e-05,
      "model_forward_time": 1.778693437576294,
      "step": 99575
    },
    {
      "epoch": 2.4390220642089843e-05,
      "step": 99575,
      "training_step_time": 7.987293004989624
    },
    {
      "epoch": 2.439117431640625e-05,
      "model_forward_time": 1.7750351428985596,
      "step": 99576
    },
    {
      "epoch": 2.439117431640625e-05,
      "step": 99576,
      "training_step_time": 8.946960926055908
    },
    {
      "epoch": 2.4392127990722658e-05,
      "model_forward_time": 1.7773852348327637,
      "step": 99577
    },
    {
      "epoch": 2.4392127990722658e-05,
      "step": 99577,
      "training_step_time": 8.01242733001709
    },
    {
      "epoch": 2.4393081665039062e-05,
      "model_forward_time": 1.9852571487426758,
      "step": 99578
    },
    {
      "epoch": 2.4393081665039062e-05,
      "step": 99578,
      "training_step_time": 11.698372602462769
    },
    {
      "epoch": 2.439403533935547e-05,
      "model_forward_time": 1.8000502586364746,
      "step": 99579
    },
    {
      "epoch": 2.439403533935547e-05,
      "step": 99579,
      "training_step_time": 9.814673900604248
    },
    {
      "epoch": 2.4394989013671873e-05,
      "grad_norm": 0.0848173126578331,
      "learning_rate": 4.845624095756174e-10,
      "loss": 0.0557,
      "step": 99580
    },
    {
      "epoch": 2.4394989013671873e-05,
      "model_forward_time": 1.7614531517028809,
      "step": 99580
    },
    {
      "dynamics_loss_avg": 0.05209831744432449,
      "epoch": 2.4394989013671873e-05,
      "step": 99580
    },
    {
      "action_loss_avg": 0.0017117201583459972,
      "epoch": 2.4394989013671873e-05,
      "step": 99580
    },
    {
      "epoch": 2.4394989013671873e-05,
      "step": 99580,
      "training_step_time": 8.642200231552124
    },
    {
      "epoch": 2.439594268798828e-05,
      "model_forward_time": 1.7622225284576416,
      "step": 99581
    },
    {
      "epoch": 2.439594268798828e-05,
      "step": 99581,
      "training_step_time": 7.68828272819519
    },
    {
      "epoch": 2.4396896362304688e-05,
      "model_forward_time": 1.7819058895111084,
      "step": 99582
    },
    {
      "epoch": 2.4396896362304688e-05,
      "step": 99582,
      "training_step_time": 7.85039496421814
    },
    {
      "epoch": 2.4397850036621092e-05,
      "model_forward_time": 1.7686388492584229,
      "step": 99583
    },
    {
      "epoch": 2.4397850036621092e-05,
      "step": 99583,
      "training_step_time": 7.800292253494263
    },
    {
      "epoch": 2.43988037109375e-05,
      "model_forward_time": 1.7585585117340088,
      "step": 99584
    },
    {
      "epoch": 2.43988037109375e-05,
      "step": 99584,
      "training_step_time": 8.723435401916504
    },
    {
      "epoch": 2.4399757385253907e-05,
      "model_forward_time": 1.7735357284545898,
      "step": 99585
    },
    {
      "epoch": 2.4399757385253907e-05,
      "step": 99585,
      "training_step_time": 8.905770540237427
    },
    {
      "epoch": 2.440071105957031e-05,
      "model_forward_time": 1.784550428390503,
      "step": 99586
    },
    {
      "epoch": 2.440071105957031e-05,
      "step": 99586,
      "training_step_time": 8.492149829864502
    },
    {
      "epoch": 2.440166473388672e-05,
      "model_forward_time": 1.7734689712524414,
      "step": 99587
    },
    {
      "epoch": 2.440166473388672e-05,
      "step": 99587,
      "training_step_time": 10.615246772766113
    },
    {
      "epoch": 2.4402618408203126e-05,
      "model_forward_time": 1.752631425857544,
      "step": 99588
    },
    {
      "epoch": 2.4402618408203126e-05,
      "step": 99588,
      "training_step_time": 8.921325445175171
    },
    {
      "epoch": 2.440357208251953e-05,
      "model_forward_time": 1.3998873233795166,
      "step": 99589
    },
    {
      "epoch": 2.440357208251953e-05,
      "step": 99589,
      "training_step_time": 8.0612633228302
    },
    {
      "epoch": 2.4404525756835937e-05,
      "grad_norm": 0.06461191922426224,
      "learning_rate": 4.618165596587121e-10,
      "loss": 0.0567,
      "step": 99590
    },
    {
      "epoch": 2.4404525756835937e-05,
      "model_forward_time": 1.7633585929870605,
      "step": 99590
    },
    {
      "dynamics_loss_avg": 0.053500395454466346,
      "epoch": 2.4404525756835937e-05,
      "step": 99590
    },
    {
      "action_loss_avg": 0.0023686883534537627,
      "epoch": 2.4404525756835937e-05,
      "step": 99590
    },
    {
      "epoch": 2.4404525756835937e-05,
      "step": 99590,
      "training_step_time": 8.598894596099854
    },
    {
      "epoch": 2.4405479431152345e-05,
      "model_forward_time": 1.7689878940582275,
      "step": 99591
    },
    {
      "epoch": 2.4405479431152345e-05,
      "step": 99591,
      "training_step_time": 7.90162205696106
    },
    {
      "epoch": 2.440643310546875e-05,
      "model_forward_time": 1.765892744064331,
      "step": 99592
    },
    {
      "epoch": 2.440643310546875e-05,
      "step": 99592,
      "training_step_time": 8.782850503921509
    },
    {
      "epoch": 2.4407386779785156e-05,
      "model_forward_time": 1.7615270614624023,
      "step": 99593
    },
    {
      "epoch": 2.4407386779785156e-05,
      "step": 99593,
      "training_step_time": 7.907882213592529
    },
    {
      "epoch": 2.4408340454101564e-05,
      "model_forward_time": 1.771810531616211,
      "step": 99594
    },
    {
      "epoch": 2.4408340454101564e-05,
      "step": 99594,
      "training_step_time": 7.979049444198608
    },
    {
      "epoch": 2.4409294128417968e-05,
      "model_forward_time": 1.7746999263763428,
      "step": 99595
    },
    {
      "epoch": 2.4409294128417968e-05,
      "step": 99595,
      "training_step_time": 7.929990291595459
    },
    {
      "epoch": 2.4410247802734375e-05,
      "model_forward_time": 1.7590651512145996,
      "step": 99596
    },
    {
      "epoch": 2.4410247802734375e-05,
      "step": 99596,
      "training_step_time": 7.922295808792114
    },
    {
      "epoch": 2.4411201477050782e-05,
      "model_forward_time": 2.4210221767425537,
      "step": 99597
    },
    {
      "epoch": 2.4411201477050782e-05,
      "step": 99597,
      "training_step_time": 8.523375272750854
    },
    {
      "epoch": 2.4412155151367187e-05,
      "model_forward_time": 1.88570237159729,
      "step": 99598
    },
    {
      "epoch": 2.4412155151367187e-05,
      "step": 99598,
      "training_step_time": 7.856462001800537
    },
    {
      "epoch": 2.4413108825683594e-05,
      "model_forward_time": 1.7784676551818848,
      "step": 99599
    },
    {
      "epoch": 2.4413108825683594e-05,
      "step": 99599,
      "training_step_time": 7.946611166000366
    },
    {
      "epoch": 2.44140625e-05,
      "grad_norm": 0.0958845242857933,
      "learning_rate": 4.3961745172071966e-10,
      "loss": 0.0573,
      "step": 99600
    },
    {
      "epoch": 2.44140625e-05,
      "model_forward_time": 1.750943899154663,
      "step": 99600
    },
    {
      "dynamics_loss_avg": 0.06212636567652226,
      "epoch": 2.44140625e-05,
      "step": 99600
    },
    {
      "action_loss_avg": 0.0015155147528275847,
      "epoch": 2.44140625e-05,
      "step": 99600
    },
    {
      "epoch": 2.44140625e-05,
      "step": 99600,
      "training_step_time": 8.415128469467163
    },
    {
      "epoch": 2.4415016174316405e-05,
      "model_forward_time": 1.7736949920654297,
      "step": 99601
    },
    {
      "epoch": 2.4415016174316405e-05,
      "step": 99601,
      "training_step_time": 7.325754404067993
    },
    {
      "epoch": 2.4415969848632813e-05,
      "model_forward_time": 1.7707464694976807,
      "step": 99602
    },
    {
      "epoch": 2.4415969848632813e-05,
      "step": 99602,
      "training_step_time": 8.65089726448059
    },
    {
      "epoch": 2.441692352294922e-05,
      "model_forward_time": 1.7749876976013184,
      "step": 99603
    },
    {
      "epoch": 2.441692352294922e-05,
      "step": 99603,
      "training_step_time": 8.834553241729736
    },
    {
      "epoch": 2.4417877197265624e-05,
      "model_forward_time": 1.7770135402679443,
      "step": 99604
    },
    {
      "epoch": 2.4417877197265624e-05,
      "step": 99604,
      "training_step_time": 7.964595556259155
    },
    {
      "epoch": 2.441883087158203e-05,
      "model_forward_time": 1.7666552066802979,
      "step": 99605
    },
    {
      "epoch": 2.441883087158203e-05,
      "step": 99605,
      "training_step_time": 7.858537435531616
    },
    {
      "epoch": 2.441978454589844e-05,
      "model_forward_time": 2.420366048812866,
      "step": 99606
    },
    {
      "epoch": 2.441978454589844e-05,
      "step": 99606,
      "training_step_time": 8.823745250701904
    },
    {
      "epoch": 2.4420738220214843e-05,
      "model_forward_time": 1.8838253021240234,
      "step": 99607
    },
    {
      "epoch": 2.4420738220214843e-05,
      "step": 99607,
      "training_step_time": 7.936479568481445
    },
    {
      "epoch": 2.442169189453125e-05,
      "model_forward_time": 1.7759861946105957,
      "step": 99608
    },
    {
      "epoch": 2.442169189453125e-05,
      "step": 99608,
      "training_step_time": 7.846410751342773
    },
    {
      "epoch": 2.4422645568847658e-05,
      "model_forward_time": 1.7804219722747803,
      "step": 99609
    },
    {
      "epoch": 2.4422645568847658e-05,
      "step": 99609,
      "training_step_time": 7.733332872390747
    },
    {
      "epoch": 2.4423599243164062e-05,
      "grad_norm": 0.10387396067380905,
      "learning_rate": 4.1796508818969796e-10,
      "loss": 0.0564,
      "step": 99610
    },
    {
      "epoch": 2.4423599243164062e-05,
      "model_forward_time": 1.767012357711792,
      "step": 99610
    },
    {
      "dynamics_loss_avg": 0.05193321760743856,
      "epoch": 2.4423599243164062e-05,
      "step": 99610
    },
    {
      "action_loss_avg": 0.0023583730042446406,
      "epoch": 2.4423599243164062e-05,
      "step": 99610
    },
    {
      "epoch": 2.4423599243164062e-05,
      "step": 99610,
      "training_step_time": 7.532988786697388
    },
    {
      "epoch": 2.442455291748047e-05,
      "model_forward_time": 1.7891535758972168,
      "step": 99611
    },
    {
      "epoch": 2.442455291748047e-05,
      "step": 99611,
      "training_step_time": 7.77673602104187
    },
    {
      "epoch": 2.4425506591796873e-05,
      "model_forward_time": 1.7702248096466064,
      "step": 99612
    },
    {
      "epoch": 2.4425506591796873e-05,
      "step": 99612,
      "training_step_time": 7.742519378662109
    },
    {
      "epoch": 2.442646026611328e-05,
      "model_forward_time": 1.774824857711792,
      "step": 99613
    },
    {
      "epoch": 2.442646026611328e-05,
      "step": 99613,
      "training_step_time": 8.877768993377686
    },
    {
      "epoch": 2.4427413940429688e-05,
      "model_forward_time": 1.7778136730194092,
      "step": 99614
    },
    {
      "epoch": 2.4427413940429688e-05,
      "step": 99614,
      "training_step_time": 7.718544006347656
    },
    {
      "epoch": 2.4428367614746092e-05,
      "model_forward_time": 2.3981611728668213,
      "step": 99615
    },
    {
      "epoch": 2.4428367614746092e-05,
      "step": 99615,
      "training_step_time": 7.865651607513428
    },
    {
      "epoch": 2.44293212890625e-05,
      "model_forward_time": 1.8639822006225586,
      "step": 99616
    },
    {
      "epoch": 2.44293212890625e-05,
      "step": 99616,
      "training_step_time": 8.860785245895386
    },
    {
      "epoch": 2.4430274963378907e-05,
      "model_forward_time": 1.7678797245025635,
      "step": 99617
    },
    {
      "epoch": 2.4430274963378907e-05,
      "step": 99617,
      "training_step_time": 7.985130548477173
    },
    {
      "epoch": 2.443122863769531e-05,
      "model_forward_time": 1.7805655002593994,
      "step": 99618
    },
    {
      "epoch": 2.443122863769531e-05,
      "step": 99618,
      "training_step_time": 7.489580869674683
    },
    {
      "epoch": 2.443218231201172e-05,
      "model_forward_time": 1.4073195457458496,
      "step": 99619
    },
    {
      "epoch": 2.443218231201172e-05,
      "step": 99619,
      "training_step_time": 7.75138521194458
    },
    {
      "epoch": 2.4433135986328126e-05,
      "grad_norm": 0.09192115068435669,
      "learning_rate": 3.9685947143319746e-10,
      "loss": 0.0566,
      "step": 99620
    },
    {
      "epoch": 2.4433135986328126e-05,
      "model_forward_time": 1.7689788341522217,
      "step": 99620
    },
    {
      "dynamics_loss_avg": 0.06374805346131325,
      "epoch": 2.4433135986328126e-05,
      "step": 99620
    },
    {
      "action_loss_avg": 0.001603717208490707,
      "epoch": 2.4433135986328126e-05,
      "step": 99620
    },
    {
      "epoch": 2.4433135986328126e-05,
      "step": 99620,
      "training_step_time": 7.780886650085449
    },
    {
      "epoch": 2.443408966064453e-05,
      "model_forward_time": 1.7719764709472656,
      "step": 99621
    },
    {
      "epoch": 2.443408966064453e-05,
      "step": 99621,
      "training_step_time": 7.663792610168457
    },
    {
      "epoch": 2.4435043334960937e-05,
      "model_forward_time": 1.765942096710205,
      "step": 99622
    },
    {
      "epoch": 2.4435043334960937e-05,
      "step": 99622,
      "training_step_time": 9.876545429229736
    },
    {
      "epoch": 2.4435997009277345e-05,
      "model_forward_time": 1.7762744426727295,
      "step": 99623
    },
    {
      "epoch": 2.4435997009277345e-05,
      "step": 99623,
      "training_step_time": 7.879106521606445
    },
    {
      "epoch": 2.443695068359375e-05,
      "model_forward_time": 2.3935043811798096,
      "step": 99624
    },
    {
      "epoch": 2.443695068359375e-05,
      "step": 99624,
      "training_step_time": 7.961593866348267
    },
    {
      "epoch": 2.4437904357910156e-05,
      "model_forward_time": 1.8385889530181885,
      "step": 99625
    },
    {
      "epoch": 2.4437904357910156e-05,
      "step": 99625,
      "training_step_time": 7.859544038772583
    },
    {
      "epoch": 2.4438858032226564e-05,
      "model_forward_time": 1.7716379165649414,
      "step": 99626
    },
    {
      "epoch": 2.4438858032226564e-05,
      "step": 99626,
      "training_step_time": 7.916780710220337
    },
    {
      "epoch": 2.4439811706542968e-05,
      "model_forward_time": 1.7653770446777344,
      "step": 99627
    },
    {
      "epoch": 2.4439811706542968e-05,
      "step": 99627,
      "training_step_time": 7.486976146697998
    },
    {
      "epoch": 2.4440765380859375e-05,
      "model_forward_time": 1.7693073749542236,
      "step": 99628
    },
    {
      "epoch": 2.4440765380859375e-05,
      "step": 99628,
      "training_step_time": 8.567364931106567
    },
    {
      "epoch": 2.4441719055175783e-05,
      "model_forward_time": 1.778296947479248,
      "step": 99629
    },
    {
      "epoch": 2.4441719055175783e-05,
      "step": 99629,
      "training_step_time": 7.673360586166382
    },
    {
      "epoch": 2.4442672729492187e-05,
      "grad_norm": 0.15281738340854645,
      "learning_rate": 3.7630060375937197e-10,
      "loss": 0.0565,
      "step": 99630
    },
    {
      "epoch": 2.4442672729492187e-05,
      "model_forward_time": 1.762437105178833,
      "step": 99630
    },
    {
      "dynamics_loss_avg": 0.0602681489661336,
      "epoch": 2.4442672729492187e-05,
      "step": 99630
    },
    {
      "action_loss_avg": 0.0018051732215099037,
      "epoch": 2.4442672729492187e-05,
      "step": 99630
    },
    {
      "epoch": 2.4442672729492187e-05,
      "step": 99630,
      "training_step_time": 7.653427839279175
    },
    {
      "epoch": 2.4443626403808594e-05,
      "model_forward_time": 1.7681849002838135,
      "step": 99631
    },
    {
      "epoch": 2.4443626403808594e-05,
      "step": 99631,
      "training_step_time": 7.621371269226074
    },
    {
      "epoch": 2.4444580078125e-05,
      "model_forward_time": 1.7717790603637695,
      "step": 99632
    },
    {
      "epoch": 2.4444580078125e-05,
      "step": 99632,
      "training_step_time": 7.682037830352783
    },
    {
      "epoch": 2.4445533752441405e-05,
      "model_forward_time": 2.370797872543335,
      "step": 99633
    },
    {
      "epoch": 2.4445533752441405e-05,
      "step": 99633,
      "training_step_time": 9.303922176361084
    },
    {
      "epoch": 2.4446487426757813e-05,
      "model_forward_time": 1.808741569519043,
      "step": 99634
    },
    {
      "epoch": 2.4446487426757813e-05,
      "step": 99634,
      "training_step_time": 7.752806186676025
    },
    {
      "epoch": 2.444744110107422e-05,
      "model_forward_time": 1.768592119216919,
      "step": 99635
    },
    {
      "epoch": 2.444744110107422e-05,
      "step": 99635,
      "training_step_time": 7.8526952266693115
    },
    {
      "epoch": 2.4448394775390624e-05,
      "model_forward_time": 1.764324426651001,
      "step": 99636
    },
    {
      "epoch": 2.4448394775390624e-05,
      "step": 99636,
      "training_step_time": 7.677091598510742
    },
    {
      "epoch": 2.444934844970703e-05,
      "model_forward_time": 1.7744081020355225,
      "step": 99637
    },
    {
      "epoch": 2.444934844970703e-05,
      "step": 99637,
      "training_step_time": 7.757537126541138
    },
    {
      "epoch": 2.445030212402344e-05,
      "model_forward_time": 1.7531704902648926,
      "step": 99638
    },
    {
      "epoch": 2.445030212402344e-05,
      "step": 99638,
      "training_step_time": 7.546937942504883
    },
    {
      "epoch": 2.4451255798339843e-05,
      "model_forward_time": 1.7706077098846436,
      "step": 99639
    },
    {
      "epoch": 2.4451255798339843e-05,
      "step": 99639,
      "training_step_time": 8.44368577003479
    },
    {
      "epoch": 2.445220947265625e-05,
      "grad_norm": 0.1767507791519165,
      "learning_rate": 3.562884874158679e-10,
      "loss": 0.0564,
      "step": 99640
    },
    {
      "epoch": 2.445220947265625e-05,
      "model_forward_time": 1.3949077129364014,
      "step": 99640
    },
    {
      "dynamics_loss_avg": 0.044890542048960924,
      "epoch": 2.445220947265625e-05,
      "step": 99640
    },
    {
      "action_loss_avg": 0.0017195378051837907,
      "epoch": 2.445220947265625e-05,
      "step": 99640
    },
    {
      "epoch": 2.445220947265625e-05,
      "step": 99640,
      "training_step_time": 7.702503204345703
    },
    {
      "epoch": 2.4453163146972655e-05,
      "model_forward_time": 1.7705886363983154,
      "step": 99641
    },
    {
      "epoch": 2.4453163146972655e-05,
      "step": 99641,
      "training_step_time": 7.8502421379089355
    },
    {
      "epoch": 2.4454116821289062e-05,
      "model_forward_time": 1.765963077545166,
      "step": 99642
    },
    {
      "epoch": 2.4454116821289062e-05,
      "step": 99642,
      "training_step_time": 8.612442255020142
    },
    {
      "epoch": 2.445507049560547e-05,
      "model_forward_time": 2.4727227687835693,
      "step": 99643
    },
    {
      "epoch": 2.445507049560547e-05,
      "step": 99643,
      "training_step_time": 8.94324254989624
    },
    {
      "epoch": 2.4456024169921873e-05,
      "model_forward_time": 1.7654168605804443,
      "step": 99644
    },
    {
      "epoch": 2.4456024169921873e-05,
      "step": 99644,
      "training_step_time": 9.980807781219482
    },
    {
      "epoch": 2.445697784423828e-05,
      "model_forward_time": 1.7865307331085205,
      "step": 99645
    },
    {
      "epoch": 2.445697784423828e-05,
      "step": 99645,
      "training_step_time": 11.613352537155151
    },
    {
      "epoch": 2.4457931518554688e-05,
      "model_forward_time": 1.7874016761779785,
      "step": 99646
    },
    {
      "epoch": 2.4457931518554688e-05,
      "step": 99646,
      "training_step_time": 6.895582914352417
    },
    {
      "epoch": 2.4458885192871092e-05,
      "model_forward_time": 1.7832434177398682,
      "step": 99647
    },
    {
      "epoch": 2.4458885192871092e-05,
      "step": 99647,
      "training_step_time": 7.601344108581543
    },
    {
      "epoch": 2.44598388671875e-05,
      "model_forward_time": 1.7812275886535645,
      "step": 99648
    },
    {
      "epoch": 2.44598388671875e-05,
      "step": 99648,
      "training_step_time": 7.754733562469482
    },
    {
      "epoch": 2.4460792541503907e-05,
      "model_forward_time": 1.7884089946746826,
      "step": 99649
    },
    {
      "epoch": 2.4460792541503907e-05,
      "step": 99649,
      "training_step_time": 7.6452085971832275
    },
    {
      "epoch": 2.446174621582031e-05,
      "grad_norm": 0.07987727224826813,
      "learning_rate": 3.3682312459260015e-10,
      "loss": 0.055,
      "step": 99650
    },
    {
      "epoch": 2.446174621582031e-05,
      "model_forward_time": 1.7669224739074707,
      "step": 99650
    },
    {
      "dynamics_loss_avg": 0.06738362070173025,
      "epoch": 2.446174621582031e-05,
      "step": 99650
    },
    {
      "action_loss_avg": 0.0024899412062950434,
      "epoch": 2.446174621582031e-05,
      "step": 99650
    },
    {
      "epoch": 2.446174621582031e-05,
      "step": 99650,
      "training_step_time": 6.909033298492432
    },
    {
      "epoch": 2.446269989013672e-05,
      "model_forward_time": 1.7867710590362549,
      "step": 99651
    },
    {
      "epoch": 2.446269989013672e-05,
      "step": 99651,
      "training_step_time": 6.971322536468506
    },
    {
      "epoch": 2.4463653564453126e-05,
      "model_forward_time": 1.7773699760437012,
      "step": 99652
    },
    {
      "epoch": 2.4463653564453126e-05,
      "step": 99652,
      "training_step_time": 6.964739799499512
    },
    {
      "epoch": 2.446460723876953e-05,
      "model_forward_time": 1.7776477336883545,
      "step": 99653
    },
    {
      "epoch": 2.446460723876953e-05,
      "step": 99653,
      "training_step_time": 6.942729711532593
    },
    {
      "epoch": 2.4465560913085937e-05,
      "model_forward_time": 1.7808563709259033,
      "step": 99654
    },
    {
      "epoch": 2.4465560913085937e-05,
      "step": 99654,
      "training_step_time": 6.9261369705200195
    },
    {
      "epoch": 2.4466514587402345e-05,
      "model_forward_time": 2.4959030151367188,
      "step": 99655
    },
    {
      "epoch": 2.4466514587402345e-05,
      "step": 99655,
      "training_step_time": 9.06169605255127
    },
    {
      "epoch": 2.446746826171875e-05,
      "model_forward_time": 1.7773830890655518,
      "step": 99656
    },
    {
      "epoch": 2.446746826171875e-05,
      "step": 99656,
      "training_step_time": 6.936147212982178
    },
    {
      "epoch": 2.4468421936035156e-05,
      "model_forward_time": 1.7876255512237549,
      "step": 99657
    },
    {
      "epoch": 2.4468421936035156e-05,
      "step": 99657,
      "training_step_time": 7.84590220451355
    },
    {
      "epoch": 2.4469375610351564e-05,
      "model_forward_time": 1.7828259468078613,
      "step": 99658
    },
    {
      "epoch": 2.4469375610351564e-05,
      "step": 99658,
      "training_step_time": 7.71045446395874
    },
    {
      "epoch": 2.4470329284667968e-05,
      "model_forward_time": 1.782768964767456,
      "step": 99659
    },
    {
      "epoch": 2.4470329284667968e-05,
      "step": 99659,
      "training_step_time": 6.9748125076293945
    },
    {
      "epoch": 2.4471282958984375e-05,
      "grad_norm": 0.1293783038854599,
      "learning_rate": 3.179045174167561e-10,
      "loss": 0.0551,
      "step": 99660
    },
    {
      "epoch": 2.4471282958984375e-05,
      "model_forward_time": 1.7662127017974854,
      "step": 99660
    },
    {
      "dynamics_loss_avg": 0.052056449465453625,
      "epoch": 2.4471282958984375e-05,
      "step": 99660
    },
    {
      "action_loss_avg": 0.0015552502532955258,
      "epoch": 2.4471282958984375e-05,
      "step": 99660
    },
    {
      "epoch": 2.4471282958984375e-05,
      "step": 99660,
      "training_step_time": 6.910064220428467
    },
    {
      "epoch": 2.4472236633300783e-05,
      "model_forward_time": 1.7819406986236572,
      "step": 99661
    },
    {
      "epoch": 2.4472236633300783e-05,
      "step": 99661,
      "training_step_time": 6.932367563247681
    },
    {
      "epoch": 2.4473190307617187e-05,
      "model_forward_time": 1.772942066192627,
      "step": 99662
    },
    {
      "epoch": 2.4473190307617187e-05,
      "step": 99662,
      "training_step_time": 6.949835300445557
    },
    {
      "epoch": 2.4474143981933594e-05,
      "model_forward_time": 1.7734875679016113,
      "step": 99663
    },
    {
      "epoch": 2.4474143981933594e-05,
      "step": 99663,
      "training_step_time": 6.962634086608887
    },
    {
      "epoch": 2.447509765625e-05,
      "model_forward_time": 2.4623494148254395,
      "step": 99664
    },
    {
      "epoch": 2.447509765625e-05,
      "step": 99664,
      "training_step_time": 8.527849674224854
    },
    {
      "epoch": 2.4476051330566405e-05,
      "model_forward_time": 1.777991533279419,
      "step": 99665
    },
    {
      "epoch": 2.4476051330566405e-05,
      "step": 99665,
      "training_step_time": 7.791555404663086
    },
    {
      "epoch": 2.4477005004882813e-05,
      "model_forward_time": 1.785613775253296,
      "step": 99666
    },
    {
      "epoch": 2.4477005004882813e-05,
      "step": 99666,
      "training_step_time": 7.5439133644104
    },
    {
      "epoch": 2.447795867919922e-05,
      "model_forward_time": 1.7883327007293701,
      "step": 99667
    },
    {
      "epoch": 2.447795867919922e-05,
      "step": 99667,
      "training_step_time": 9.752748966217041
    },
    {
      "epoch": 2.4478912353515624e-05,
      "model_forward_time": 1.7898132801055908,
      "step": 99668
    },
    {
      "epoch": 2.4478912353515624e-05,
      "step": 99668,
      "training_step_time": 6.9483184814453125
    },
    {
      "epoch": 2.4479866027832032e-05,
      "model_forward_time": 1.7813372611999512,
      "step": 99669
    },
    {
      "epoch": 2.4479866027832032e-05,
      "step": 99669,
      "training_step_time": 6.9245383739471436
    },
    {
      "epoch": 2.448081970214844e-05,
      "grad_norm": 0.07175211608409882,
      "learning_rate": 2.9953266795834656e-10,
      "loss": 0.0553,
      "step": 99670
    },
    {
      "epoch": 2.448081970214844e-05,
      "model_forward_time": 1.7716927528381348,
      "step": 99670
    },
    {
      "dynamics_loss_avg": 0.057834592461585996,
      "epoch": 2.448081970214844e-05,
      "step": 99670
    },
    {
      "action_loss_avg": 0.0018854780922993087,
      "epoch": 2.448081970214844e-05,
      "step": 99670
    },
    {
      "epoch": 2.448081970214844e-05,
      "step": 99670,
      "training_step_time": 6.923975706100464
    },
    {
      "epoch": 2.4481773376464843e-05,
      "model_forward_time": 1.7768495082855225,
      "step": 99671
    },
    {
      "epoch": 2.4481773376464843e-05,
      "step": 99671,
      "training_step_time": 6.96103572845459
    },
    {
      "epoch": 2.448272705078125e-05,
      "model_forward_time": 1.7826406955718994,
      "step": 99672
    },
    {
      "epoch": 2.448272705078125e-05,
      "step": 99672,
      "training_step_time": 6.917430877685547
    },
    {
      "epoch": 2.4483680725097655e-05,
      "model_forward_time": 2.437809705734253,
      "step": 99673
    },
    {
      "epoch": 2.4483680725097655e-05,
      "step": 99673,
      "training_step_time": 9.758396863937378
    },
    {
      "epoch": 2.4484634399414062e-05,
      "model_forward_time": 1.7821061611175537,
      "step": 99674
    },
    {
      "epoch": 2.4484634399414062e-05,
      "step": 99674,
      "training_step_time": 7.747280836105347
    },
    {
      "epoch": 2.448558807373047e-05,
      "model_forward_time": 1.7609176635742188,
      "step": 99675
    },
    {
      "epoch": 2.448558807373047e-05,
      "step": 99675,
      "training_step_time": 6.812088966369629
    },
    {
      "epoch": 2.4486541748046873e-05,
      "model_forward_time": 1.7842459678649902,
      "step": 99676
    },
    {
      "epoch": 2.4486541748046873e-05,
      "step": 99676,
      "training_step_time": 7.85511326789856
    },
    {
      "epoch": 2.448749542236328e-05,
      "model_forward_time": 1.7789654731750488,
      "step": 99677
    },
    {
      "epoch": 2.448749542236328e-05,
      "step": 99677,
      "training_step_time": 7.6787354946136475
    },
    {
      "epoch": 2.4488449096679688e-05,
      "model_forward_time": 1.778956413269043,
      "step": 99678
    },
    {
      "epoch": 2.4488449096679688e-05,
      "step": 99678,
      "training_step_time": 6.925533056259155
    },
    {
      "epoch": 2.4489402770996092e-05,
      "model_forward_time": 1.7785227298736572,
      "step": 99679
    },
    {
      "epoch": 2.4489402770996092e-05,
      "step": 99679,
      "training_step_time": 6.897232294082642
    },
    {
      "epoch": 2.44903564453125e-05,
      "grad_norm": 0.07973916083574295,
      "learning_rate": 2.817075782263201e-10,
      "loss": 0.0533,
      "step": 99680
    },
    {
      "epoch": 2.44903564453125e-05,
      "model_forward_time": 1.782247543334961,
      "step": 99680
    },
    {
      "dynamics_loss_avg": 0.055809400603175165,
      "epoch": 2.44903564453125e-05,
      "step": 99680
    },
    {
      "action_loss_avg": 0.002321259425661992,
      "epoch": 2.44903564453125e-05,
      "step": 99680
    },
    {
      "epoch": 2.44903564453125e-05,
      "step": 99680,
      "training_step_time": 6.904545545578003
    },
    {
      "epoch": 2.4491310119628907e-05,
      "model_forward_time": 1.7763760089874268,
      "step": 99681
    },
    {
      "epoch": 2.4491310119628907e-05,
      "step": 99681,
      "training_step_time": 6.859511137008667
    },
    {
      "epoch": 2.449226379394531e-05,
      "model_forward_time": 2.4051945209503174,
      "step": 99682
    },
    {
      "epoch": 2.449226379394531e-05,
      "step": 99682,
      "training_step_time": 7.955425500869751
    },
    {
      "epoch": 2.449321746826172e-05,
      "model_forward_time": 1.775348424911499,
      "step": 99683
    },
    {
      "epoch": 2.449321746826172e-05,
      "step": 99683,
      "training_step_time": 7.8348517417907715
    },
    {
      "epoch": 2.4494171142578126e-05,
      "model_forward_time": 1.7772533893585205,
      "step": 99684
    },
    {
      "epoch": 2.4494171142578126e-05,
      "step": 99684,
      "training_step_time": 6.891007900238037
    },
    {
      "epoch": 2.449512481689453e-05,
      "model_forward_time": 1.7825267314910889,
      "step": 99685
    },
    {
      "epoch": 2.449512481689453e-05,
      "step": 99685,
      "training_step_time": 7.792643308639526
    },
    {
      "epoch": 2.4496078491210937e-05,
      "model_forward_time": 1.782250165939331,
      "step": 99686
    },
    {
      "epoch": 2.4496078491210937e-05,
      "step": 99686,
      "training_step_time": 7.683096408843994
    },
    {
      "epoch": 2.4497032165527345e-05,
      "model_forward_time": 1.7824883460998535,
      "step": 99687
    },
    {
      "epoch": 2.4497032165527345e-05,
      "step": 99687,
      "training_step_time": 6.929309368133545
    },
    {
      "epoch": 2.449798583984375e-05,
      "model_forward_time": 1.7762186527252197,
      "step": 99688
    },
    {
      "epoch": 2.449798583984375e-05,
      "step": 99688,
      "training_step_time": 6.9403369426727295
    },
    {
      "epoch": 2.4498939514160156e-05,
      "model_forward_time": 1.786390781402588,
      "step": 99689
    },
    {
      "epoch": 2.4498939514160156e-05,
      "step": 99689,
      "training_step_time": 6.869911193847656
    },
    {
      "epoch": 2.4499893188476564e-05,
      "grad_norm": 0.08144957572221756,
      "learning_rate": 2.6442925016967326e-10,
      "loss": 0.053,
      "step": 99690
    },
    {
      "epoch": 2.4499893188476564e-05,
      "model_forward_time": 1.764354944229126,
      "step": 99690
    },
    {
      "dynamics_loss_avg": 0.06255252119153738,
      "epoch": 2.4499893188476564e-05,
      "step": 99690
    },
    {
      "action_loss_avg": 0.0021980965961120092,
      "epoch": 2.4499893188476564e-05,
      "step": 99690
    },
    {
      "epoch": 2.4499893188476564e-05,
      "step": 99690,
      "training_step_time": 6.90386962890625
    },
    {
      "epoch": 2.4500846862792968e-05,
      "model_forward_time": 2.402402639389038,
      "step": 99691
    },
    {
      "epoch": 2.4500846862792968e-05,
      "step": 99691,
      "training_step_time": 8.943419456481934
    },
    {
      "epoch": 2.4501800537109375e-05,
      "model_forward_time": 1.8927853107452393,
      "step": 99692
    },
    {
      "epoch": 2.4501800537109375e-05,
      "step": 99692,
      "training_step_time": 8.283271789550781
    },
    {
      "epoch": 2.4502754211425783e-05,
      "model_forward_time": 1.7792155742645264,
      "step": 99693
    },
    {
      "epoch": 2.4502754211425783e-05,
      "step": 99693,
      "training_step_time": 6.85360312461853
    },
    {
      "epoch": 2.4503707885742187e-05,
      "model_forward_time": 1.7663509845733643,
      "step": 99694
    },
    {
      "epoch": 2.4503707885742187e-05,
      "step": 99694,
      "training_step_time": 7.849955320358276
    },
    {
      "epoch": 2.4504661560058594e-05,
      "model_forward_time": 1.7834222316741943,
      "step": 99695
    },
    {
      "epoch": 2.4504661560058594e-05,
      "step": 99695,
      "training_step_time": 7.7684454917907715
    },
    {
      "epoch": 2.4505615234375e-05,
      "model_forward_time": 1.7835619449615479,
      "step": 99696
    },
    {
      "epoch": 2.4505615234375e-05,
      "step": 99696,
      "training_step_time": 6.864746332168579
    },
    {
      "epoch": 2.4506568908691405e-05,
      "model_forward_time": 1.7784128189086914,
      "step": 99697
    },
    {
      "epoch": 2.4506568908691405e-05,
      "step": 99697,
      "training_step_time": 6.837536573410034
    },
    {
      "epoch": 2.4507522583007813e-05,
      "model_forward_time": 1.7782995700836182,
      "step": 99698
    },
    {
      "epoch": 2.4507522583007813e-05,
      "step": 99698,
      "training_step_time": 6.883864164352417
    },
    {
      "epoch": 2.450847625732422e-05,
      "model_forward_time": 1.775865077972412,
      "step": 99699
    },
    {
      "epoch": 2.450847625732422e-05,
      "step": 99699,
      "training_step_time": 6.845202684402466
    },
    {
      "epoch": 2.4509429931640624e-05,
      "grad_norm": 0.0995519831776619,
      "learning_rate": 2.476976856780056e-10,
      "loss": 0.0538,
      "step": 99700
    },
    {
      "epoch": 2.4509429931640624e-05,
      "model_forward_time": 2.4179937839508057,
      "step": 99700
    },
    {
      "dynamics_loss_avg": 0.0792288864031434,
      "epoch": 2.4509429931640624e-05,
      "step": 99700
    },
    {
      "action_loss_avg": 0.0022678068722598254,
      "epoch": 2.4509429931640624e-05,
      "step": 99700
    },
    {
      "epoch": 2.4509429931640624e-05,
      "step": 99700,
      "training_step_time": 8.770931005477905
    },
    {
      "epoch": 2.4510383605957032e-05,
      "model_forward_time": 1.841144323348999,
      "step": 99701
    },
    {
      "epoch": 2.4510383605957032e-05,
      "step": 99701,
      "training_step_time": 9.733854055404663
    },
    {
      "epoch": 2.451133728027344e-05,
      "model_forward_time": 1.7742719650268555,
      "step": 99702
    },
    {
      "epoch": 2.451133728027344e-05,
      "step": 99702,
      "training_step_time": 7.789602041244507
    },
    {
      "epoch": 2.4512290954589843e-05,
      "model_forward_time": 1.777259111404419,
      "step": 99703
    },
    {
      "epoch": 2.4512290954589843e-05,
      "step": 99703,
      "training_step_time": 7.742580890655518
    },
    {
      "epoch": 2.451324462890625e-05,
      "model_forward_time": 1.7767813205718994,
      "step": 99704
    },
    {
      "epoch": 2.451324462890625e-05,
      "step": 99704,
      "training_step_time": 7.637443542480469
    },
    {
      "epoch": 2.4514198303222655e-05,
      "model_forward_time": 1.7760396003723145,
      "step": 99705
    },
    {
      "epoch": 2.4514198303222655e-05,
      "step": 99705,
      "training_step_time": 7.156185150146484
    },
    {
      "epoch": 2.4515151977539062e-05,
      "model_forward_time": 1.7872028350830078,
      "step": 99706
    },
    {
      "epoch": 2.4515151977539062e-05,
      "step": 99706,
      "training_step_time": 6.865330219268799
    },
    {
      "epoch": 2.451610565185547e-05,
      "model_forward_time": 1.7748830318450928,
      "step": 99707
    },
    {
      "epoch": 2.451610565185547e-05,
      "step": 99707,
      "training_step_time": 6.878751754760742
    },
    {
      "epoch": 2.4517059326171873e-05,
      "model_forward_time": 1.7847683429718018,
      "step": 99708
    },
    {
      "epoch": 2.4517059326171873e-05,
      "step": 99708,
      "training_step_time": 6.885619878768921
    },
    {
      "epoch": 2.451801300048828e-05,
      "model_forward_time": 1.7967090606689453,
      "step": 99709
    },
    {
      "epoch": 2.451801300048828e-05,
      "step": 99709,
      "training_step_time": 8.300865888595581
    },
    {
      "epoch": 2.451896667480469e-05,
      "grad_norm": 0.4776473343372345,
      "learning_rate": 2.315128865809646e-10,
      "loss": 0.0516,
      "step": 99710
    },
    {
      "epoch": 2.451896667480469e-05,
      "model_forward_time": 1.8083758354187012,
      "step": 99710
    },
    {
      "dynamics_loss_avg": 0.04735316280275583,
      "epoch": 2.451896667480469e-05,
      "step": 99710
    },
    {
      "action_loss_avg": 0.002642504154937342,
      "epoch": 2.451896667480469e-05,
      "step": 99710
    },
    {
      "epoch": 2.451896667480469e-05,
      "step": 99710,
      "training_step_time": 9.45514726638794
    },
    {
      "epoch": 2.4519920349121092e-05,
      "model_forward_time": 1.7616190910339355,
      "step": 99711
    },
    {
      "epoch": 2.4519920349121092e-05,
      "step": 99711,
      "training_step_time": 7.6238484382629395
    },
    {
      "epoch": 2.45208740234375e-05,
      "model_forward_time": 1.7723593711853027,
      "step": 99712
    },
    {
      "epoch": 2.45208740234375e-05,
      "step": 99712,
      "training_step_time": 7.785365104675293
    },
    {
      "epoch": 2.4521827697753907e-05,
      "model_forward_time": 1.7691171169281006,
      "step": 99713
    },
    {
      "epoch": 2.4521827697753907e-05,
      "step": 99713,
      "training_step_time": 7.646460771560669
    },
    {
      "epoch": 2.452278137207031e-05,
      "model_forward_time": 1.767009973526001,
      "step": 99714
    },
    {
      "epoch": 2.452278137207031e-05,
      "step": 99714,
      "training_step_time": 7.1829540729522705
    },
    {
      "epoch": 2.452373504638672e-05,
      "model_forward_time": 1.749561071395874,
      "step": 99715
    },
    {
      "epoch": 2.452373504638672e-05,
      "step": 99715,
      "training_step_time": 6.785725116729736
    },
    {
      "epoch": 2.4524688720703126e-05,
      "model_forward_time": 1.77872896194458,
      "step": 99716
    },
    {
      "epoch": 2.4524688720703126e-05,
      "step": 99716,
      "training_step_time": 6.9386444091796875
    },
    {
      "epoch": 2.452564239501953e-05,
      "model_forward_time": 1.7739624977111816,
      "step": 99717
    },
    {
      "epoch": 2.452564239501953e-05,
      "step": 99717,
      "training_step_time": 6.914052486419678
    },
    {
      "epoch": 2.4526596069335937e-05,
      "model_forward_time": 1.767991304397583,
      "step": 99718
    },
    {
      "epoch": 2.4526596069335937e-05,
      "step": 99718,
      "training_step_time": 8.035487174987793
    },
    {
      "epoch": 2.4527549743652345e-05,
      "model_forward_time": 2.4525952339172363,
      "step": 99719
    },
    {
      "epoch": 2.4527549743652345e-05,
      "step": 99719,
      "training_step_time": 10.358842611312866
    },
    {
      "epoch": 2.452850341796875e-05,
      "grad_norm": 0.08316826075315475,
      "learning_rate": 2.1587485464880097e-10,
      "loss": 0.0529,
      "step": 99720
    },
    {
      "epoch": 2.452850341796875e-05,
      "model_forward_time": 1.7640163898468018,
      "step": 99720
    },
    {
      "dynamics_loss_avg": 0.052920847292989494,
      "epoch": 2.452850341796875e-05,
      "step": 99720
    },
    {
      "action_loss_avg": 0.002053700252145063,
      "epoch": 2.452850341796875e-05,
      "step": 99720
    },
    {
      "epoch": 2.452850341796875e-05,
      "step": 99720,
      "training_step_time": 7.580684423446655
    },
    {
      "epoch": 2.4529457092285156e-05,
      "model_forward_time": 1.7802894115447998,
      "step": 99721
    },
    {
      "epoch": 2.4529457092285156e-05,
      "step": 99721,
      "training_step_time": 7.6732378005981445
    },
    {
      "epoch": 2.4530410766601564e-05,
      "model_forward_time": 1.7689716815948486,
      "step": 99722
    },
    {
      "epoch": 2.4530410766601564e-05,
      "step": 99722,
      "training_step_time": 8.58591079711914
    },
    {
      "epoch": 2.4531364440917968e-05,
      "model_forward_time": 1.7773382663726807,
      "step": 99723
    },
    {
      "epoch": 2.4531364440917968e-05,
      "step": 99723,
      "training_step_time": 7.118450880050659
    },
    {
      "epoch": 2.4532318115234375e-05,
      "model_forward_time": 1.7767236232757568,
      "step": 99724
    },
    {
      "epoch": 2.4532318115234375e-05,
      "step": 99724,
      "training_step_time": 6.808307886123657
    },
    {
      "epoch": 2.4533271789550783e-05,
      "model_forward_time": 1.7537574768066406,
      "step": 99725
    },
    {
      "epoch": 2.4533271789550783e-05,
      "step": 99725,
      "training_step_time": 6.750299692153931
    },
    {
      "epoch": 2.4534225463867187e-05,
      "model_forward_time": 1.7793903350830078,
      "step": 99726
    },
    {
      "epoch": 2.4534225463867187e-05,
      "step": 99726,
      "training_step_time": 6.805819272994995
    },
    {
      "epoch": 2.4535179138183594e-05,
      "model_forward_time": 1.7652978897094727,
      "step": 99727
    },
    {
      "epoch": 2.4535179138183594e-05,
      "step": 99727,
      "training_step_time": 7.769726276397705
    },
    {
      "epoch": 2.45361328125e-05,
      "model_forward_time": 2.446044683456421,
      "step": 99728
    },
    {
      "epoch": 2.45361328125e-05,
      "step": 99728,
      "training_step_time": 10.855830669403076
    },
    {
      "epoch": 2.4537086486816405e-05,
      "model_forward_time": 1.771306037902832,
      "step": 99729
    },
    {
      "epoch": 2.4537086486816405e-05,
      "step": 99729,
      "training_step_time": 8.19916820526123
    },
    {
      "epoch": 2.4538040161132813e-05,
      "grad_norm": 0.09165260195732117,
      "learning_rate": 2.0078359159125814e-10,
      "loss": 0.0528,
      "step": 99730
    },
    {
      "epoch": 2.4538040161132813e-05,
      "model_forward_time": 1.756795883178711,
      "step": 99730
    },
    {
      "dynamics_loss_avg": 0.055071518011391166,
      "epoch": 2.4538040161132813e-05,
      "step": 99730
    },
    {
      "action_loss_avg": 0.0025303784175775947,
      "epoch": 2.4538040161132813e-05,
      "step": 99730
    },
    {
      "epoch": 2.4538040161132813e-05,
      "step": 99730,
      "training_step_time": 7.725148677825928
    },
    {
      "epoch": 2.453899383544922e-05,
      "model_forward_time": 1.7750446796417236,
      "step": 99731
    },
    {
      "epoch": 2.453899383544922e-05,
      "step": 99731,
      "training_step_time": 7.773640394210815
    },
    {
      "epoch": 2.4539947509765624e-05,
      "model_forward_time": 1.7481377124786377,
      "step": 99732
    },
    {
      "epoch": 2.4539947509765624e-05,
      "step": 99732,
      "training_step_time": 7.865679979324341
    },
    {
      "epoch": 2.4540901184082032e-05,
      "model_forward_time": 1.7654767036437988,
      "step": 99733
    },
    {
      "epoch": 2.4540901184082032e-05,
      "step": 99733,
      "training_step_time": 6.961426496505737
    },
    {
      "epoch": 2.454185485839844e-05,
      "model_forward_time": 1.774768352508545,
      "step": 99734
    },
    {
      "epoch": 2.454185485839844e-05,
      "step": 99734,
      "training_step_time": 6.942198038101196
    },
    {
      "epoch": 2.4542808532714843e-05,
      "model_forward_time": 1.7756669521331787,
      "step": 99735
    },
    {
      "epoch": 2.4542808532714843e-05,
      "step": 99735,
      "training_step_time": 6.902317762374878
    },
    {
      "epoch": 2.454376220703125e-05,
      "model_forward_time": 1.7659633159637451,
      "step": 99736
    },
    {
      "epoch": 2.454376220703125e-05,
      "step": 99736,
      "training_step_time": 7.8537187576293945
    },
    {
      "epoch": 2.4544715881347655e-05,
      "model_forward_time": 2.476933240890503,
      "step": 99737
    },
    {
      "epoch": 2.4544715881347655e-05,
      "step": 99737,
      "training_step_time": 9.680400133132935
    },
    {
      "epoch": 2.4545669555664062e-05,
      "model_forward_time": 1.777968168258667,
      "step": 99738
    },
    {
      "epoch": 2.4545669555664062e-05,
      "step": 99738,
      "training_step_time": 8.520863056182861
    },
    {
      "epoch": 2.454662322998047e-05,
      "model_forward_time": 1.770831823348999,
      "step": 99739
    },
    {
      "epoch": 2.454662322998047e-05,
      "step": 99739,
      "training_step_time": 7.82702898979187
    },
    {
      "epoch": 2.4547576904296874e-05,
      "grad_norm": 0.08395548909902573,
      "learning_rate": 1.8623909905923776e-10,
      "loss": 0.0525,
      "step": 99740
    },
    {
      "epoch": 2.4547576904296874e-05,
      "model_forward_time": 1.7465262413024902,
      "step": 99740
    },
    {
      "dynamics_loss_avg": 0.04353675134479999,
      "epoch": 2.4547576904296874e-05,
      "step": 99740
    },
    {
      "action_loss_avg": 0.001695155369816348,
      "epoch": 2.4547576904296874e-05,
      "step": 99740
    },
    {
      "epoch": 2.4547576904296874e-05,
      "step": 99740,
      "training_step_time": 7.692800521850586
    },
    {
      "epoch": 2.454853057861328e-05,
      "model_forward_time": 1.3773353099822998,
      "step": 99741
    },
    {
      "epoch": 2.454853057861328e-05,
      "step": 99741,
      "training_step_time": 7.6597490310668945
    },
    {
      "epoch": 2.454948425292969e-05,
      "model_forward_time": 1.7583723068237305,
      "step": 99742
    },
    {
      "epoch": 2.454948425292969e-05,
      "step": 99742,
      "training_step_time": 7.210348606109619
    },
    {
      "epoch": 2.4550437927246092e-05,
      "model_forward_time": 1.7623863220214844,
      "step": 99743
    },
    {
      "epoch": 2.4550437927246092e-05,
      "step": 99743,
      "training_step_time": 6.967419862747192
    },
    {
      "epoch": 2.45513916015625e-05,
      "model_forward_time": 1.7695448398590088,
      "step": 99744
    },
    {
      "epoch": 2.45513916015625e-05,
      "step": 99744,
      "training_step_time": 6.954437971115112
    },
    {
      "epoch": 2.4552345275878907e-05,
      "model_forward_time": 1.7680327892303467,
      "step": 99745
    },
    {
      "epoch": 2.4552345275878907e-05,
      "step": 99745,
      "training_step_time": 7.751862525939941
    },
    {
      "epoch": 2.455329895019531e-05,
      "model_forward_time": 2.412950277328491,
      "step": 99746
    },
    {
      "epoch": 2.455329895019531e-05,
      "step": 99746,
      "training_step_time": 9.688852787017822
    },
    {
      "epoch": 2.455425262451172e-05,
      "model_forward_time": 1.479767084121704,
      "step": 99747
    },
    {
      "epoch": 2.455425262451172e-05,
      "step": 99747,
      "training_step_time": 7.904343605041504
    },
    {
      "epoch": 2.4555206298828126e-05,
      "model_forward_time": 1.3995416164398193,
      "step": 99748
    },
    {
      "epoch": 2.4555206298828126e-05,
      "step": 99748,
      "training_step_time": 7.803342342376709
    },
    {
      "epoch": 2.455615997314453e-05,
      "model_forward_time": 1.7727384567260742,
      "step": 99749
    },
    {
      "epoch": 2.455615997314453e-05,
      "step": 99749,
      "training_step_time": 8.627412796020508
    },
    {
      "epoch": 2.4557113647460938e-05,
      "grad_norm": 0.06562893837690353,
      "learning_rate": 1.7224137864257918e-10,
      "loss": 0.0537,
      "step": 99750
    },
    {
      "epoch": 2.4557113647460938e-05,
      "model_forward_time": 1.7747948169708252,
      "step": 99750
    },
    {
      "dynamics_loss_avg": 0.04744663015007973,
      "epoch": 2.4557113647460938e-05,
      "step": 99750
    },
    {
      "action_loss_avg": 0.0015558120881905778,
      "epoch": 2.4557113647460938e-05,
      "step": 99750
    },
    {
      "epoch": 2.4557113647460938e-05,
      "step": 99750,
      "training_step_time": 7.745667457580566
    },
    {
      "epoch": 2.4558067321777345e-05,
      "model_forward_time": 1.7685248851776123,
      "step": 99751
    },
    {
      "epoch": 2.4558067321777345e-05,
      "step": 99751,
      "training_step_time": 7.304280996322632
    },
    {
      "epoch": 2.455902099609375e-05,
      "model_forward_time": 1.779784917831421,
      "step": 99752
    },
    {
      "epoch": 2.455902099609375e-05,
      "step": 99752,
      "training_step_time": 6.990095376968384
    },
    {
      "epoch": 2.4559974670410156e-05,
      "model_forward_time": 1.4112029075622559,
      "step": 99753
    },
    {
      "epoch": 2.4559974670410156e-05,
      "step": 99753,
      "training_step_time": 6.95356011390686
    },
    {
      "epoch": 2.4560928344726564e-05,
      "model_forward_time": 1.7713654041290283,
      "step": 99754
    },
    {
      "epoch": 2.4560928344726564e-05,
      "step": 99754,
      "training_step_time": 7.770333528518677
    },
    {
      "epoch": 2.4561882019042968e-05,
      "model_forward_time": 2.373077869415283,
      "step": 99755
    },
    {
      "epoch": 2.4561882019042968e-05,
      "step": 99755,
      "training_step_time": 10.129161357879639
    },
    {
      "epoch": 2.4562835693359375e-05,
      "model_forward_time": 1.8277289867401123,
      "step": 99756
    },
    {
      "epoch": 2.4562835693359375e-05,
      "step": 99756,
      "training_step_time": 9.243759155273438
    },
    {
      "epoch": 2.4563789367675783e-05,
      "model_forward_time": 1.767533779144287,
      "step": 99757
    },
    {
      "epoch": 2.4563789367675783e-05,
      "step": 99757,
      "training_step_time": 7.867189407348633
    },
    {
      "epoch": 2.4564743041992187e-05,
      "model_forward_time": 1.761749029159546,
      "step": 99758
    },
    {
      "epoch": 2.4564743041992187e-05,
      "step": 99758,
      "training_step_time": 7.861337184906006
    },
    {
      "epoch": 2.4565696716308594e-05,
      "model_forward_time": 1.771979808807373,
      "step": 99759
    },
    {
      "epoch": 2.4565696716308594e-05,
      "step": 99759,
      "training_step_time": 7.791265249252319
    },
    {
      "epoch": 2.4566650390625e-05,
      "grad_norm": 0.11760248988866806,
      "learning_rate": 1.5879043187283506e-10,
      "loss": 0.0525,
      "step": 99760
    },
    {
      "epoch": 2.4566650390625e-05,
      "model_forward_time": 1.7653381824493408,
      "step": 99760
    },
    {
      "dynamics_loss_avg": 0.05126930847764015,
      "epoch": 2.4566650390625e-05,
      "step": 99760
    },
    {
      "action_loss_avg": 0.001608883129665628,
      "epoch": 2.4566650390625e-05,
      "step": 99760
    },
    {
      "epoch": 2.4566650390625e-05,
      "step": 99760,
      "training_step_time": 7.322254419326782
    },
    {
      "epoch": 2.4567604064941406e-05,
      "model_forward_time": 1.7638051509857178,
      "step": 99761
    },
    {
      "epoch": 2.4567604064941406e-05,
      "step": 99761,
      "training_step_time": 7.005982398986816
    },
    {
      "epoch": 2.4568557739257813e-05,
      "model_forward_time": 1.7766730785369873,
      "step": 99762
    },
    {
      "epoch": 2.4568557739257813e-05,
      "step": 99762,
      "training_step_time": 6.970430135726929
    },
    {
      "epoch": 2.456951141357422e-05,
      "model_forward_time": 1.7734408378601074,
      "step": 99763
    },
    {
      "epoch": 2.456951141357422e-05,
      "step": 99763,
      "training_step_time": 7.778897285461426
    },
    {
      "epoch": 2.4570465087890624e-05,
      "model_forward_time": 1.7720491886138916,
      "step": 99764
    },
    {
      "epoch": 2.4570465087890624e-05,
      "step": 99764,
      "training_step_time": 8.337883472442627
    },
    {
      "epoch": 2.4571418762207032e-05,
      "model_forward_time": 2.494608163833618,
      "step": 99765
    },
    {
      "epoch": 2.4571418762207032e-05,
      "step": 99765,
      "training_step_time": 11.39504623413086
    },
    {
      "epoch": 2.4572372436523436e-05,
      "model_forward_time": 1.7765429019927979,
      "step": 99766
    },
    {
      "epoch": 2.4572372436523436e-05,
      "step": 99766,
      "training_step_time": 8.61901044845581
    },
    {
      "epoch": 2.4573326110839843e-05,
      "model_forward_time": 1.7709746360778809,
      "step": 99767
    },
    {
      "epoch": 2.4573326110839843e-05,
      "step": 99767,
      "training_step_time": 8.694707155227661
    },
    {
      "epoch": 2.457427978515625e-05,
      "model_forward_time": 1.7802207469940186,
      "step": 99768
    },
    {
      "epoch": 2.457427978515625e-05,
      "step": 99768,
      "training_step_time": 7.844254970550537
    },
    {
      "epoch": 2.4575233459472655e-05,
      "model_forward_time": 1.7817723751068115,
      "step": 99769
    },
    {
      "epoch": 2.4575233459472655e-05,
      "step": 99769,
      "training_step_time": 7.097869157791138
    },
    {
      "epoch": 2.4576187133789062e-05,
      "grad_norm": 0.07125723361968994,
      "learning_rate": 1.4588626022049578e-10,
      "loss": 0.0535,
      "step": 99770
    },
    {
      "epoch": 2.4576187133789062e-05,
      "model_forward_time": 1.7800006866455078,
      "step": 99770
    },
    {
      "dynamics_loss_avg": 0.05953057873994112,
      "epoch": 2.4576187133789062e-05,
      "step": 99770
    },
    {
      "action_loss_avg": 0.0021328884235117585,
      "epoch": 2.4576187133789062e-05,
      "step": 99770
    },
    {
      "epoch": 2.4576187133789062e-05,
      "step": 99770,
      "training_step_time": 6.919454574584961
    },
    {
      "epoch": 2.457714080810547e-05,
      "model_forward_time": 1.776569128036499,
      "step": 99771
    },
    {
      "epoch": 2.457714080810547e-05,
      "step": 99771,
      "training_step_time": 6.925874710083008
    },
    {
      "epoch": 2.4578094482421874e-05,
      "model_forward_time": 1.7723674774169922,
      "step": 99772
    },
    {
      "epoch": 2.4578094482421874e-05,
      "step": 99772,
      "training_step_time": 7.763803958892822
    },
    {
      "epoch": 2.457904815673828e-05,
      "model_forward_time": 1.7740685939788818,
      "step": 99773
    },
    {
      "epoch": 2.457904815673828e-05,
      "step": 99773,
      "training_step_time": 9.304977416992188
    },
    {
      "epoch": 2.458000183105469e-05,
      "model_forward_time": 2.465399980545044,
      "step": 99774
    },
    {
      "epoch": 2.458000183105469e-05,
      "step": 99774,
      "training_step_time": 11.293460845947266
    },
    {
      "epoch": 2.4580955505371092e-05,
      "model_forward_time": 1.7710397243499756,
      "step": 99775
    },
    {
      "epoch": 2.4580955505371092e-05,
      "step": 99775,
      "training_step_time": 7.920503377914429
    },
    {
      "epoch": 2.45819091796875e-05,
      "model_forward_time": 1.7762384414672852,
      "step": 99776
    },
    {
      "epoch": 2.45819091796875e-05,
      "step": 99776,
      "training_step_time": 7.846836805343628
    },
    {
      "epoch": 2.4582862854003907e-05,
      "model_forward_time": 1.7811179161071777,
      "step": 99777
    },
    {
      "epoch": 2.4582862854003907e-05,
      "step": 99777,
      "training_step_time": 7.775322914123535
    },
    {
      "epoch": 2.458381652832031e-05,
      "model_forward_time": 1.7739031314849854,
      "step": 99778
    },
    {
      "epoch": 2.458381652832031e-05,
      "step": 99778,
      "training_step_time": 7.758758544921875
    },
    {
      "epoch": 2.458477020263672e-05,
      "model_forward_time": 1.7753548622131348,
      "step": 99779
    },
    {
      "epoch": 2.458477020263672e-05,
      "step": 99779,
      "training_step_time": 7.3379316329956055
    },
    {
      "epoch": 2.4585723876953126e-05,
      "grad_norm": 0.099278025329113,
      "learning_rate": 1.3352886509720997e-10,
      "loss": 0.0528,
      "step": 99780
    },
    {
      "epoch": 2.4585723876953126e-05,
      "model_forward_time": 1.794743299484253,
      "step": 99780
    },
    {
      "dynamics_loss_avg": 0.04862010944634676,
      "epoch": 2.4585723876953126e-05,
      "step": 99780
    },
    {
      "action_loss_avg": 0.002730255777714774,
      "epoch": 2.4585723876953126e-05,
      "step": 99780
    },
    {
      "epoch": 2.4585723876953126e-05,
      "step": 99780,
      "training_step_time": 6.982760190963745
    },
    {
      "epoch": 2.458667755126953e-05,
      "model_forward_time": 1.769366979598999,
      "step": 99781
    },
    {
      "epoch": 2.458667755126953e-05,
      "step": 99781,
      "training_step_time": 7.69320273399353
    },
    {
      "epoch": 2.4587631225585938e-05,
      "model_forward_time": 1.7785699367523193,
      "step": 99782
    },
    {
      "epoch": 2.4587631225585938e-05,
      "step": 99782,
      "training_step_time": 7.847370386123657
    },
    {
      "epoch": 2.4588584899902345e-05,
      "model_forward_time": 2.4755184650421143,
      "step": 99783
    },
    {
      "epoch": 2.4588584899902345e-05,
      "step": 99783,
      "training_step_time": 10.721267700195312
    },
    {
      "epoch": 2.458953857421875e-05,
      "model_forward_time": 1.770916223526001,
      "step": 99784
    },
    {
      "epoch": 2.458953857421875e-05,
      "step": 99784,
      "training_step_time": 7.9203455448150635
    },
    {
      "epoch": 2.4590492248535156e-05,
      "model_forward_time": 1.7755858898162842,
      "step": 99785
    },
    {
      "epoch": 2.4590492248535156e-05,
      "step": 99785,
      "training_step_time": 7.887303352355957
    },
    {
      "epoch": 2.4591445922851564e-05,
      "model_forward_time": 1.7596588134765625,
      "step": 99786
    },
    {
      "epoch": 2.4591445922851564e-05,
      "step": 99786,
      "training_step_time": 9.668839454650879
    },
    {
      "epoch": 2.4592399597167968e-05,
      "model_forward_time": 1.7685956954956055,
      "step": 99787
    },
    {
      "epoch": 2.4592399597167968e-05,
      "step": 99787,
      "training_step_time": 7.9327473640441895
    },
    {
      "epoch": 2.4593353271484375e-05,
      "model_forward_time": 1.7631189823150635,
      "step": 99788
    },
    {
      "epoch": 2.4593353271484375e-05,
      "step": 99788,
      "training_step_time": 7.409065008163452
    },
    {
      "epoch": 2.4594306945800783e-05,
      "model_forward_time": 1.763681411743164,
      "step": 99789
    },
    {
      "epoch": 2.4594306945800783e-05,
      "step": 99789,
      "training_step_time": 7.098166227340698
    },
    {
      "epoch": 2.4595260620117187e-05,
      "grad_norm": 0.13952094316482544,
      "learning_rate": 1.217182478535639e-10,
      "loss": 0.0537,
      "step": 99790
    },
    {
      "epoch": 2.4595260620117187e-05,
      "model_forward_time": 1.7620534896850586,
      "step": 99790
    },
    {
      "dynamics_loss_avg": 0.050108818151056764,
      "epoch": 2.4595260620117187e-05,
      "step": 99790
    },
    {
      "action_loss_avg": 0.0017132034990936518,
      "epoch": 2.4595260620117187e-05,
      "step": 99790
    },
    {
      "epoch": 2.4595260620117187e-05,
      "step": 99790,
      "training_step_time": 7.826570749282837
    },
    {
      "epoch": 2.4596214294433594e-05,
      "model_forward_time": 1.7747406959533691,
      "step": 99791
    },
    {
      "epoch": 2.4596214294433594e-05,
      "step": 99791,
      "training_step_time": 8.744810581207275
    },
    {
      "epoch": 2.459716796875e-05,
      "model_forward_time": 2.432015895843506,
      "step": 99792
    },
    {
      "epoch": 2.459716796875e-05,
      "step": 99792,
      "training_step_time": 10.039323806762695
    },
    {
      "epoch": 2.4598121643066406e-05,
      "model_forward_time": 1.8304810523986816,
      "step": 99793
    },
    {
      "epoch": 2.4598121643066406e-05,
      "step": 99793,
      "training_step_time": 9.125697374343872
    },
    {
      "epoch": 2.4599075317382813e-05,
      "model_forward_time": 1.7727363109588623,
      "step": 99794
    },
    {
      "epoch": 2.4599075317382813e-05,
      "step": 99794,
      "training_step_time": 7.812853574752808
    },
    {
      "epoch": 2.460002899169922e-05,
      "model_forward_time": 1.7702205181121826,
      "step": 99795
    },
    {
      "epoch": 2.460002899169922e-05,
      "step": 99795,
      "training_step_time": 8.77156138420105
    },
    {
      "epoch": 2.4600982666015624e-05,
      "model_forward_time": 1.770381212234497,
      "step": 99796
    },
    {
      "epoch": 2.4600982666015624e-05,
      "step": 99796,
      "training_step_time": 7.834414720535278
    },
    {
      "epoch": 2.4601936340332032e-05,
      "model_forward_time": 1.7609264850616455,
      "step": 99797
    },
    {
      "epoch": 2.4601936340332032e-05,
      "step": 99797,
      "training_step_time": 7.310391664505005
    },
    {
      "epoch": 2.4602890014648436e-05,
      "model_forward_time": 1.8059065341949463,
      "step": 99798
    },
    {
      "epoch": 2.4602890014648436e-05,
      "step": 99798,
      "training_step_time": 7.0964014530181885
    },
    {
      "epoch": 2.4603843688964843e-05,
      "model_forward_time": 1.7677521705627441,
      "step": 99799
    },
    {
      "epoch": 2.4603843688964843e-05,
      "step": 99799,
      "training_step_time": 7.807634592056274
    },
    {
      "epoch": 2.460479736328125e-05,
      "grad_norm": 0.06845298409461975,
      "learning_rate": 1.1045440978185717e-10,
      "loss": 0.0539,
      "step": 99800
    },
    {
      "epoch": 2.460479736328125e-05,
      "model_forward_time": 1.769829273223877,
      "step": 99800
    },
    {
      "dynamics_loss_avg": 0.05566457975655794,
      "epoch": 2.460479736328125e-05,
      "step": 99800
    },
    {
      "action_loss_avg": 0.0016190795402508229,
      "epoch": 2.460479736328125e-05,
      "step": 99800
    },
    {
      "epoch": 2.460479736328125e-05,
      "step": 99800,
      "training_step_time": 7.884487152099609
    },
    {
      "epoch": 2.4605751037597655e-05,
      "model_forward_time": 2.3846986293792725,
      "step": 99801
    },
    {
      "epoch": 2.4605751037597655e-05,
      "step": 99801,
      "training_step_time": 10.723470687866211
    },
    {
      "epoch": 2.4606704711914062e-05,
      "model_forward_time": 1.813931941986084,
      "step": 99802
    },
    {
      "epoch": 2.4606704711914062e-05,
      "step": 99802,
      "training_step_time": 10.683576107025146
    },
    {
      "epoch": 2.460765838623047e-05,
      "model_forward_time": 1.774446725845337,
      "step": 99803
    },
    {
      "epoch": 2.460765838623047e-05,
      "step": 99803,
      "training_step_time": 8.86069369316101
    },
    {
      "epoch": 2.4608612060546874e-05,
      "model_forward_time": 1.8049523830413818,
      "step": 99804
    },
    {
      "epoch": 2.4608612060546874e-05,
      "step": 99804,
      "training_step_time": 8.74552869796753
    },
    {
      "epoch": 2.460956573486328e-05,
      "model_forward_time": 1.7759747505187988,
      "step": 99805
    },
    {
      "epoch": 2.460956573486328e-05,
      "step": 99805,
      "training_step_time": 7.8963987827301025
    },
    {
      "epoch": 2.461051940917969e-05,
      "model_forward_time": 1.7774946689605713,
      "step": 99806
    },
    {
      "epoch": 2.461051940917969e-05,
      "step": 99806,
      "training_step_time": 7.310963869094849
    },
    {
      "epoch": 2.4611473083496092e-05,
      "model_forward_time": 1.7711057662963867,
      "step": 99807
    },
    {
      "epoch": 2.4611473083496092e-05,
      "step": 99807,
      "training_step_time": 7.01494574546814
    },
    {
      "epoch": 2.46124267578125e-05,
      "model_forward_time": 1.7654047012329102,
      "step": 99808
    },
    {
      "epoch": 2.46124267578125e-05,
      "step": 99808,
      "training_step_time": 7.0103981494903564
    },
    {
      "epoch": 2.4613380432128907e-05,
      "model_forward_time": 1.767829418182373,
      "step": 99809
    },
    {
      "epoch": 2.4613380432128907e-05,
      "step": 99809,
      "training_step_time": 7.862824440002441
    },
    {
      "epoch": 2.461433410644531e-05,
      "grad_norm": 0.07870034128427505,
      "learning_rate": 9.973735211332714e-11,
      "loss": 0.0546,
      "step": 99810
    },
    {
      "epoch": 2.461433410644531e-05,
      "model_forward_time": 1.8116908073425293,
      "step": 99810
    },
    {
      "dynamics_loss_avg": 0.06901596505194903,
      "epoch": 2.461433410644531e-05,
      "step": 99810
    },
    {
      "action_loss_avg": 0.0020632517902413384,
      "epoch": 2.461433410644531e-05,
      "step": 99810
    },
    {
      "epoch": 2.461433410644531e-05,
      "step": 99810,
      "training_step_time": 8.061946392059326
    },
    {
      "epoch": 2.461528778076172e-05,
      "model_forward_time": 2.389594793319702,
      "step": 99811
    },
    {
      "epoch": 2.461528778076172e-05,
      "step": 99811,
      "training_step_time": 9.008016109466553
    },
    {
      "epoch": 2.4616241455078126e-05,
      "model_forward_time": 1.7643024921417236,
      "step": 99812
    },
    {
      "epoch": 2.4616241455078126e-05,
      "step": 99812,
      "training_step_time": 8.660597085952759
    },
    {
      "epoch": 2.461719512939453e-05,
      "model_forward_time": 1.7715833187103271,
      "step": 99813
    },
    {
      "epoch": 2.461719512939453e-05,
      "step": 99813,
      "training_step_time": 8.690409660339355
    },
    {
      "epoch": 2.4618148803710938e-05,
      "model_forward_time": 1.762207269668579,
      "step": 99814
    },
    {
      "epoch": 2.4618148803710938e-05,
      "step": 99814,
      "training_step_time": 7.8268468379974365
    },
    {
      "epoch": 2.4619102478027345e-05,
      "model_forward_time": 1.7620067596435547,
      "step": 99815
    },
    {
      "epoch": 2.4619102478027345e-05,
      "step": 99815,
      "training_step_time": 7.32043981552124
    },
    {
      "epoch": 2.462005615234375e-05,
      "model_forward_time": 1.7734417915344238,
      "step": 99816
    },
    {
      "epoch": 2.462005615234375e-05,
      "step": 99816,
      "training_step_time": 6.970227479934692
    },
    {
      "epoch": 2.4621009826660156e-05,
      "model_forward_time": 1.7820556163787842,
      "step": 99817
    },
    {
      "epoch": 2.4621009826660156e-05,
      "step": 99817,
      "training_step_time": 6.952266216278076
    },
    {
      "epoch": 2.4621963500976564e-05,
      "model_forward_time": 1.7622041702270508,
      "step": 99818
    },
    {
      "epoch": 2.4621963500976564e-05,
      "step": 99818,
      "training_step_time": 7.851989507675171
    },
    {
      "epoch": 2.4622917175292968e-05,
      "model_forward_time": 1.7719616889953613,
      "step": 99819
    },
    {
      "epoch": 2.4622917175292968e-05,
      "step": 99819,
      "training_step_time": 9.819349765777588
    },
    {
      "epoch": 2.4623870849609375e-05,
      "grad_norm": 0.09723364561796188,
      "learning_rate": 8.95670760203693e-11,
      "loss": 0.0566,
      "step": 99820
    },
    {
      "epoch": 2.4623870849609375e-05,
      "model_forward_time": 2.389394998550415,
      "step": 99820
    },
    {
      "dynamics_loss_avg": 0.05788255576044321,
      "epoch": 2.4623870849609375e-05,
      "step": 99820
    },
    {
      "action_loss_avg": 0.0027867341646924614,
      "epoch": 2.4623870849609375e-05,
      "step": 99820
    },
    {
      "epoch": 2.4623870849609375e-05,
      "step": 99820,
      "training_step_time": 9.207957983016968
    },
    {
      "epoch": 2.4624824523925783e-05,
      "model_forward_time": 1.822068214416504,
      "step": 99821
    },
    {
      "epoch": 2.4624824523925783e-05,
      "step": 99821,
      "training_step_time": 8.970570802688599
    },
    {
      "epoch": 2.4625778198242187e-05,
      "model_forward_time": 1.7650527954101562,
      "step": 99822
    },
    {
      "epoch": 2.4625778198242187e-05,
      "step": 99822,
      "training_step_time": 9.219889402389526
    },
    {
      "epoch": 2.4626731872558594e-05,
      "model_forward_time": 1.7680325508117676,
      "step": 99823
    },
    {
      "epoch": 2.4626731872558594e-05,
      "step": 99823,
      "training_step_time": 7.840718030929565
    },
    {
      "epoch": 2.4627685546875e-05,
      "model_forward_time": 1.8249306678771973,
      "step": 99824
    },
    {
      "epoch": 2.4627685546875e-05,
      "step": 99824,
      "training_step_time": 8.373034477233887
    },
    {
      "epoch": 2.4628639221191406e-05,
      "model_forward_time": 1.8127708435058594,
      "step": 99825
    },
    {
      "epoch": 2.4628639221191406e-05,
      "step": 99825,
      "training_step_time": 6.99556303024292
    },
    {
      "epoch": 2.4629592895507813e-05,
      "model_forward_time": 1.8185784816741943,
      "step": 99826
    },
    {
      "epoch": 2.4629592895507813e-05,
      "step": 99826,
      "training_step_time": 6.99543833732605
    },
    {
      "epoch": 2.463054656982422e-05,
      "model_forward_time": 1.8145806789398193,
      "step": 99827
    },
    {
      "epoch": 2.463054656982422e-05,
      "step": 99827,
      "training_step_time": 8.420453071594238
    },
    {
      "epoch": 2.4631500244140624e-05,
      "model_forward_time": 1.8078486919403076,
      "step": 99828
    },
    {
      "epoch": 2.4631500244140624e-05,
      "step": 99828,
      "training_step_time": 7.702038049697876
    },
    {
      "epoch": 2.4632453918457032e-05,
      "model_forward_time": 2.4750423431396484,
      "step": 99829
    },
    {
      "epoch": 2.4632453918457032e-05,
      "step": 99829,
      "training_step_time": 8.91091275215149
    },
    {
      "epoch": 2.4633407592773436e-05,
      "grad_norm": 0.08504943549633026,
      "learning_rate": 7.994358261542712e-11,
      "loss": 0.0544,
      "step": 99830
    },
    {
      "epoch": 2.4633407592773436e-05,
      "model_forward_time": 1.832975149154663,
      "step": 99830
    },
    {
      "dynamics_loss_avg": 0.049336409009993074,
      "epoch": 2.4633407592773436e-05,
      "step": 99830
    },
    {
      "action_loss_avg": 0.0012957967323018239,
      "epoch": 2.4633407592773436e-05,
      "step": 99830
    },
    {
      "epoch": 2.4633407592773436e-05,
      "step": 99830,
      "training_step_time": 10.549160718917847
    },
    {
      "epoch": 2.4634361267089843e-05,
      "model_forward_time": 1.8208320140838623,
      "step": 99831
    },
    {
      "epoch": 2.4634361267089843e-05,
      "step": 99831,
      "training_step_time": 7.754695653915405
    },
    {
      "epoch": 2.463531494140625e-05,
      "model_forward_time": 1.8155663013458252,
      "step": 99832
    },
    {
      "epoch": 2.463531494140625e-05,
      "step": 99832,
      "training_step_time": 7.792403221130371
    },
    {
      "epoch": 2.4636268615722655e-05,
      "model_forward_time": 1.811321496963501,
      "step": 99833
    },
    {
      "epoch": 2.4636268615722655e-05,
      "step": 99833,
      "training_step_time": 7.35849142074585
    },
    {
      "epoch": 2.4637222290039062e-05,
      "model_forward_time": 1.7930433750152588,
      "step": 99834
    },
    {
      "epoch": 2.4637222290039062e-05,
      "step": 99834,
      "training_step_time": 7.164548873901367
    },
    {
      "epoch": 2.463817596435547e-05,
      "model_forward_time": 1.8121695518493652,
      "step": 99835
    },
    {
      "epoch": 2.463817596435547e-05,
      "step": 99835,
      "training_step_time": 6.993429183959961
    },
    {
      "epoch": 2.4639129638671874e-05,
      "model_forward_time": 1.8100991249084473,
      "step": 99836
    },
    {
      "epoch": 2.4639129638671874e-05,
      "step": 99836,
      "training_step_time": 7.794045925140381
    },
    {
      "epoch": 2.464008331298828e-05,
      "model_forward_time": 1.0975279808044434,
      "step": 99837
    },
    {
      "epoch": 2.464008331298828e-05,
      "step": 99837,
      "training_step_time": 7.866705417633057
    },
    {
      "epoch": 2.464103698730469e-05,
      "model_forward_time": 2.439389944076538,
      "step": 99838
    },
    {
      "epoch": 2.464103698730469e-05,
      "step": 99838,
      "training_step_time": 8.09541630744934
    },
    {
      "epoch": 2.4641990661621092e-05,
      "model_forward_time": 1.8588130474090576,
      "step": 99839
    },
    {
      "epoch": 2.4641990661621092e-05,
      "step": 99839,
      "training_step_time": 8.614984512329102
    },
    {
      "epoch": 2.46429443359375e-05,
      "grad_norm": 0.08021017909049988,
      "learning_rate": 7.086687295043692e-11,
      "loss": 0.054,
      "step": 99840
    },
    {
      "epoch": 2.46429443359375e-05,
      "model_forward_time": 1.8125250339508057,
      "step": 99840
    },
    {
      "dynamics_loss_avg": 0.04636855442076922,
      "epoch": 2.46429443359375e-05,
      "step": 99840
    },
    {
      "action_loss_avg": 0.0013017581251915545,
      "epoch": 2.46429443359375e-05,
      "step": 99840
    },
    {
      "epoch": 2.46429443359375e-05,
      "step": 99840,
      "training_step_time": 8.70517635345459
    },
    {
      "epoch": 2.4643898010253907e-05,
      "model_forward_time": 1.811225414276123,
      "step": 99841
    },
    {
      "epoch": 2.4643898010253907e-05,
      "step": 99841,
      "training_step_time": 7.813477516174316
    },
    {
      "epoch": 2.464485168457031e-05,
      "model_forward_time": 1.8225195407867432,
      "step": 99842
    },
    {
      "epoch": 2.464485168457031e-05,
      "step": 99842,
      "training_step_time": 7.73074746131897
    },
    {
      "epoch": 2.464580535888672e-05,
      "model_forward_time": 1.8180749416351318,
      "step": 99843
    },
    {
      "epoch": 2.464580535888672e-05,
      "step": 99843,
      "training_step_time": 7.280890703201294
    },
    {
      "epoch": 2.4646759033203126e-05,
      "model_forward_time": 1.8123610019683838,
      "step": 99844
    },
    {
      "epoch": 2.4646759033203126e-05,
      "step": 99844,
      "training_step_time": 7.007057428359985
    },
    {
      "epoch": 2.464771270751953e-05,
      "model_forward_time": 1.8097093105316162,
      "step": 99845
    },
    {
      "epoch": 2.464771270751953e-05,
      "step": 99845,
      "training_step_time": 7.826398849487305
    },
    {
      "epoch": 2.4648666381835938e-05,
      "model_forward_time": 1.8105590343475342,
      "step": 99846
    },
    {
      "epoch": 2.4648666381835938e-05,
      "step": 99846,
      "training_step_time": 7.922952651977539
    },
    {
      "epoch": 2.4649620056152345e-05,
      "model_forward_time": 1.810567855834961,
      "step": 99847
    },
    {
      "epoch": 2.4649620056152345e-05,
      "step": 99847,
      "training_step_time": 10.400680303573608
    },
    {
      "epoch": 2.465057373046875e-05,
      "model_forward_time": 2.5306031703948975,
      "step": 99848
    },
    {
      "epoch": 2.465057373046875e-05,
      "step": 99848,
      "training_step_time": 10.34789514541626
    },
    {
      "epoch": 2.4651527404785156e-05,
      "model_forward_time": 1.8119943141937256,
      "step": 99849
    },
    {
      "epoch": 2.4651527404785156e-05,
      "step": 99849,
      "training_step_time": 7.845303058624268
    },
    {
      "epoch": 2.4652481079101564e-05,
      "grad_norm": 0.09677205234766006,
      "learning_rate": 6.233694801793811e-11,
      "loss": 0.0548,
      "step": 99850
    },
    {
      "epoch": 2.4652481079101564e-05,
      "model_forward_time": 1.8151216506958008,
      "step": 99850
    },
    {
      "dynamics_loss_avg": 0.05979039240628481,
      "epoch": 2.4652481079101564e-05,
      "step": 99850
    },
    {
      "action_loss_avg": 0.003183265755069442,
      "epoch": 2.4652481079101564e-05,
      "step": 99850
    },
    {
      "epoch": 2.4652481079101564e-05,
      "step": 99850,
      "training_step_time": 7.878578424453735
    },
    {
      "epoch": 2.4653434753417968e-05,
      "model_forward_time": 1.8186569213867188,
      "step": 99851
    },
    {
      "epoch": 2.4653434753417968e-05,
      "step": 99851,
      "training_step_time": 7.883550643920898
    },
    {
      "epoch": 2.4654388427734375e-05,
      "model_forward_time": 1.8060016632080078,
      "step": 99852
    },
    {
      "epoch": 2.4654388427734375e-05,
      "step": 99852,
      "training_step_time": 7.351900577545166
    },
    {
      "epoch": 2.4655342102050783e-05,
      "model_forward_time": 1.8067762851715088,
      "step": 99853
    },
    {
      "epoch": 2.4655342102050783e-05,
      "step": 99853,
      "training_step_time": 7.130308151245117
    },
    {
      "epoch": 2.4656295776367187e-05,
      "model_forward_time": 1.8111908435821533,
      "step": 99854
    },
    {
      "epoch": 2.4656295776367187e-05,
      "step": 99854,
      "training_step_time": 7.8230531215667725
    },
    {
      "epoch": 2.4657249450683594e-05,
      "model_forward_time": 1.1073942184448242,
      "step": 99855
    },
    {
      "epoch": 2.4657249450683594e-05,
      "step": 99855,
      "training_step_time": 8.7743399143219
    },
    {
      "epoch": 2.4658203125e-05,
      "model_forward_time": 1.804619550704956,
      "step": 99856
    },
    {
      "epoch": 2.4658203125e-05,
      "step": 99856,
      "training_step_time": 9.199116945266724
    },
    {
      "epoch": 2.4659156799316406e-05,
      "model_forward_time": 2.5002472400665283,
      "step": 99857
    },
    {
      "epoch": 2.4659156799316406e-05,
      "step": 99857,
      "training_step_time": 10.888349294662476
    },
    {
      "epoch": 2.4660110473632813e-05,
      "model_forward_time": 1.8039908409118652,
      "step": 99858
    },
    {
      "epoch": 2.4660110473632813e-05,
      "step": 99858,
      "training_step_time": 8.308035612106323
    },
    {
      "epoch": 2.466106414794922e-05,
      "model_forward_time": 1.8201987743377686,
      "step": 99859
    },
    {
      "epoch": 2.466106414794922e-05,
      "step": 99859,
      "training_step_time": 8.767507076263428
    },
    {
      "epoch": 2.4662017822265624e-05,
      "grad_norm": 0.07307860255241394,
      "learning_rate": 5.43538087510731e-11,
      "loss": 0.0534,
      "step": 99860
    },
    {
      "epoch": 2.4662017822265624e-05,
      "model_forward_time": 1.8090317249298096,
      "step": 99860
    },
    {
      "dynamics_loss_avg": 0.05947103686630726,
      "epoch": 2.4662017822265624e-05,
      "step": 99860
    },
    {
      "action_loss_avg": 0.0030082793498877438,
      "epoch": 2.4662017822265624e-05,
      "step": 99860
    },
    {
      "epoch": 2.4662017822265624e-05,
      "step": 99860,
      "training_step_time": 7.86763334274292
    },
    {
      "epoch": 2.4662971496582032e-05,
      "model_forward_time": 1.8097143173217773,
      "step": 99861
    },
    {
      "epoch": 2.4662971496582032e-05,
      "step": 99861,
      "training_step_time": 7.7743377685546875
    },
    {
      "epoch": 2.4663925170898436e-05,
      "model_forward_time": 1.8091974258422852,
      "step": 99862
    },
    {
      "epoch": 2.4663925170898436e-05,
      "step": 99862,
      "training_step_time": 7.303742408752441
    },
    {
      "epoch": 2.4664878845214843e-05,
      "model_forward_time": 1.803325891494751,
      "step": 99863
    },
    {
      "epoch": 2.4664878845214843e-05,
      "step": 99863,
      "training_step_time": 7.794448375701904
    },
    {
      "epoch": 2.466583251953125e-05,
      "model_forward_time": 1.8121507167816162,
      "step": 99864
    },
    {
      "epoch": 2.466583251953125e-05,
      "step": 99864,
      "training_step_time": 7.859950304031372
    },
    {
      "epoch": 2.4666786193847655e-05,
      "model_forward_time": 1.8206839561462402,
      "step": 99865
    },
    {
      "epoch": 2.4666786193847655e-05,
      "step": 99865,
      "training_step_time": 8.876574039459229
    },
    {
      "epoch": 2.4667739868164062e-05,
      "model_forward_time": 2.4594504833221436,
      "step": 99866
    },
    {
      "epoch": 2.4667739868164062e-05,
      "step": 99866,
      "training_step_time": 10.751464366912842
    },
    {
      "epoch": 2.466869354248047e-05,
      "model_forward_time": 1.8127455711364746,
      "step": 99867
    },
    {
      "epoch": 2.466869354248047e-05,
      "step": 99867,
      "training_step_time": 9.283815145492554
    },
    {
      "epoch": 2.4669647216796874e-05,
      "model_forward_time": 1.7698757648468018,
      "step": 99868
    },
    {
      "epoch": 2.4669647216796874e-05,
      "step": 99868,
      "training_step_time": 9.334070920944214
    },
    {
      "epoch": 2.467060089111328e-05,
      "model_forward_time": 1.7698869705200195,
      "step": 99869
    },
    {
      "epoch": 2.467060089111328e-05,
      "step": 99869,
      "training_step_time": 7.797592401504517
    },
    {
      "epoch": 2.467155456542969e-05,
      "grad_norm": 0.074683278799057,
      "learning_rate": 4.6917456023032324e-11,
      "loss": 0.0546,
      "step": 99870
    },
    {
      "epoch": 2.467155456542969e-05,
      "model_forward_time": 1.7602403163909912,
      "step": 99870
    },
    {
      "dynamics_loss_avg": 0.04882657928392291,
      "epoch": 2.467155456542969e-05,
      "step": 99870
    },
    {
      "action_loss_avg": 0.0018111991259502247,
      "epoch": 2.467155456542969e-05,
      "step": 99870
    },
    {
      "epoch": 2.467155456542969e-05,
      "step": 99870,
      "training_step_time": 7.702666521072388
    },
    {
      "epoch": 2.4672508239746092e-05,
      "model_forward_time": 1.7612273693084717,
      "step": 99871
    },
    {
      "epoch": 2.4672508239746092e-05,
      "step": 99871,
      "training_step_time": 7.279443979263306
    },
    {
      "epoch": 2.46734619140625e-05,
      "model_forward_time": 1.7713377475738525,
      "step": 99872
    },
    {
      "epoch": 2.46734619140625e-05,
      "step": 99872,
      "training_step_time": 7.809614181518555
    },
    {
      "epoch": 2.4674415588378907e-05,
      "model_forward_time": 1.0711588859558105,
      "step": 99873
    },
    {
      "epoch": 2.4674415588378907e-05,
      "step": 99873,
      "training_step_time": 8.495054721832275
    },
    {
      "epoch": 2.467536926269531e-05,
      "model_forward_time": 1.4024763107299805,
      "step": 99874
    },
    {
      "epoch": 2.467536926269531e-05,
      "step": 99874,
      "training_step_time": 7.843329906463623
    },
    {
      "epoch": 2.467632293701172e-05,
      "model_forward_time": 1.7650766372680664,
      "step": 99875
    },
    {
      "epoch": 2.467632293701172e-05,
      "step": 99875,
      "training_step_time": 9.975510597229004
    },
    {
      "epoch": 2.4677276611328126e-05,
      "model_forward_time": 2.4190893173217773,
      "step": 99876
    },
    {
      "epoch": 2.4677276611328126e-05,
      "step": 99876,
      "training_step_time": 8.416800498962402
    },
    {
      "epoch": 2.467823028564453e-05,
      "model_forward_time": 1.7671723365783691,
      "step": 99877
    },
    {
      "epoch": 2.467823028564453e-05,
      "step": 99877,
      "training_step_time": 8.860888957977295
    },
    {
      "epoch": 2.4679183959960938e-05,
      "model_forward_time": 1.7739717960357666,
      "step": 99878
    },
    {
      "epoch": 2.4679183959960938e-05,
      "step": 99878,
      "training_step_time": 7.75835919380188
    },
    {
      "epoch": 2.4680137634277345e-05,
      "model_forward_time": 1.7678508758544922,
      "step": 99879
    },
    {
      "epoch": 2.4680137634277345e-05,
      "step": 99879,
      "training_step_time": 7.705821752548218
    },
    {
      "epoch": 2.468109130859375e-05,
      "grad_norm": 0.07975910604000092,
      "learning_rate": 4.0027890646499034e-11,
      "loss": 0.0535,
      "step": 99880
    },
    {
      "epoch": 2.468109130859375e-05,
      "model_forward_time": 1.7665421962738037,
      "step": 99880
    },
    {
      "dynamics_loss_avg": 0.03704053778201342,
      "epoch": 2.468109130859375e-05,
      "step": 99880
    },
    {
      "action_loss_avg": 0.001149985495430883,
      "epoch": 2.468109130859375e-05,
      "step": 99880
    },
    {
      "epoch": 2.468109130859375e-05,
      "step": 99880,
      "training_step_time": 7.172030925750732
    },
    {
      "epoch": 2.4682044982910156e-05,
      "model_forward_time": 1.7641432285308838,
      "step": 99881
    },
    {
      "epoch": 2.4682044982910156e-05,
      "step": 99881,
      "training_step_time": 7.771841049194336
    },
    {
      "epoch": 2.4682998657226564e-05,
      "model_forward_time": 1.7679128646850586,
      "step": 99882
    },
    {
      "epoch": 2.4682998657226564e-05,
      "step": 99882,
      "training_step_time": 7.752179145812988
    },
    {
      "epoch": 2.4683952331542968e-05,
      "model_forward_time": 1.7954299449920654,
      "step": 99883
    },
    {
      "epoch": 2.4683952331542968e-05,
      "step": 99883,
      "training_step_time": 7.755488157272339
    },
    {
      "epoch": 2.4684906005859375e-05,
      "model_forward_time": 1.771033525466919,
      "step": 99884
    },
    {
      "epoch": 2.4684906005859375e-05,
      "step": 99884,
      "training_step_time": 9.18714427947998
    },
    {
      "epoch": 2.4685859680175783e-05,
      "model_forward_time": 2.423103094100952,
      "step": 99885
    },
    {
      "epoch": 2.4685859680175783e-05,
      "step": 99885,
      "training_step_time": 10.47931456565857
    },
    {
      "epoch": 2.4686813354492187e-05,
      "model_forward_time": 1.7734375,
      "step": 99886
    },
    {
      "epoch": 2.4686813354492187e-05,
      "step": 99886,
      "training_step_time": 8.809068441390991
    },
    {
      "epoch": 2.4687767028808594e-05,
      "model_forward_time": 1.7710635662078857,
      "step": 99887
    },
    {
      "epoch": 2.4687767028808594e-05,
      "step": 99887,
      "training_step_time": 7.731017351150513
    },
    {
      "epoch": 2.4688720703125e-05,
      "model_forward_time": 1.764275074005127,
      "step": 99888
    },
    {
      "epoch": 2.4688720703125e-05,
      "step": 99888,
      "training_step_time": 8.642455339431763
    },
    {
      "epoch": 2.4689674377441406e-05,
      "model_forward_time": 1.7714259624481201,
      "step": 99889
    },
    {
      "epoch": 2.4689674377441406e-05,
      "step": 99889,
      "training_step_time": 8.003567457199097
    },
    {
      "epoch": 2.4690628051757813e-05,
      "grad_norm": 0.12239577621221542,
      "learning_rate": 3.368511337531466e-11,
      "loss": 0.0538,
      "step": 99890
    },
    {
      "epoch": 2.4690628051757813e-05,
      "model_forward_time": 1.7649850845336914,
      "step": 99890
    },
    {
      "dynamics_loss_avg": 0.04858245113864541,
      "epoch": 2.4690628051757813e-05,
      "step": 99890
    },
    {
      "action_loss_avg": 0.0023152108187787235,
      "epoch": 2.4690628051757813e-05,
      "step": 99890
    },
    {
      "epoch": 2.4690628051757813e-05,
      "step": 99890,
      "training_step_time": 7.813293218612671
    },
    {
      "epoch": 2.4691581726074217e-05,
      "model_forward_time": 1.768394947052002,
      "step": 99891
    },
    {
      "epoch": 2.4691581726074217e-05,
      "step": 99891,
      "training_step_time": 7.789398670196533
    },
    {
      "epoch": 2.4692535400390624e-05,
      "model_forward_time": 1.7679197788238525,
      "step": 99892
    },
    {
      "epoch": 2.4692535400390624e-05,
      "step": 99892,
      "training_step_time": 7.701988458633423
    },
    {
      "epoch": 2.4693489074707032e-05,
      "model_forward_time": 1.7669193744659424,
      "step": 99893
    },
    {
      "epoch": 2.4693489074707032e-05,
      "step": 99893,
      "training_step_time": 7.953776597976685
    },
    {
      "epoch": 2.4694442749023436e-05,
      "model_forward_time": 1.8201467990875244,
      "step": 99894
    },
    {
      "epoch": 2.4694442749023436e-05,
      "step": 99894,
      "training_step_time": 8.060622930526733
    },
    {
      "epoch": 2.4695396423339843e-05,
      "model_forward_time": 2.4194698333740234,
      "step": 99895
    },
    {
      "epoch": 2.4695396423339843e-05,
      "step": 99895,
      "training_step_time": 8.419268608093262
    },
    {
      "epoch": 2.469635009765625e-05,
      "model_forward_time": 1.76106595993042,
      "step": 99896
    },
    {
      "epoch": 2.469635009765625e-05,
      "step": 99896,
      "training_step_time": 8.3943350315094
    },
    {
      "epoch": 2.4697303771972655e-05,
      "model_forward_time": 1.7726666927337646,
      "step": 99897
    },
    {
      "epoch": 2.4697303771972655e-05,
      "step": 99897,
      "training_step_time": 7.723520278930664
    },
    {
      "epoch": 2.4698257446289062e-05,
      "model_forward_time": 1.7775609493255615,
      "step": 99898
    },
    {
      "epoch": 2.4698257446289062e-05,
      "step": 99898,
      "training_step_time": 7.598556041717529
    },
    {
      "epoch": 2.469921112060547e-05,
      "model_forward_time": 1.7756688594818115,
      "step": 99899
    },
    {
      "epoch": 2.469921112060547e-05,
      "step": 99899,
      "training_step_time": 7.677655220031738
    },
    {
      "epoch": 2.4700164794921874e-05,
      "grad_norm": 0.09102148562669754,
      "learning_rate": 2.788912490281348e-11,
      "loss": 0.0539,
      "step": 99900
    },
    {
      "epoch": 2.4700164794921874e-05,
      "model_forward_time": 1.772212028503418,
      "step": 99900
    },
    {
      "dynamics_loss_avg": 0.05728423930704594,
      "epoch": 2.4700164794921874e-05,
      "step": 99900
    },
    {
      "action_loss_avg": 0.0027103235624963418,
      "epoch": 2.4700164794921874e-05,
      "step": 99900
    },
    {
      "epoch": 2.4700164794921874e-05,
      "step": 99900,
      "training_step_time": 7.755274295806885
    },
    {
      "epoch": 2.470111846923828e-05,
      "model_forward_time": 1.7802619934082031,
      "step": 99901
    },
    {
      "epoch": 2.470111846923828e-05,
      "step": 99901,
      "training_step_time": 7.603415250778198
    },
    {
      "epoch": 2.470207214355469e-05,
      "model_forward_time": 1.7667298316955566,
      "step": 99902
    },
    {
      "epoch": 2.470207214355469e-05,
      "step": 99902,
      "training_step_time": 8.989173173904419
    },
    {
      "epoch": 2.4703025817871093e-05,
      "model_forward_time": 1.8321213722229004,
      "step": 99903
    },
    {
      "epoch": 2.4703025817871093e-05,
      "step": 99903,
      "training_step_time": 10.634342908859253
    },
    {
      "epoch": 2.47039794921875e-05,
      "model_forward_time": 2.4260854721069336,
      "step": 99904
    },
    {
      "epoch": 2.47039794921875e-05,
      "step": 99904,
      "training_step_time": 8.697011709213257
    },
    {
      "epoch": 2.4704933166503907e-05,
      "model_forward_time": 1.7676959037780762,
      "step": 99905
    },
    {
      "epoch": 2.4704933166503907e-05,
      "step": 99905,
      "training_step_time": 7.706480264663696
    },
    {
      "epoch": 2.470588684082031e-05,
      "model_forward_time": 1.771573543548584,
      "step": 99906
    },
    {
      "epoch": 2.470588684082031e-05,
      "step": 99906,
      "training_step_time": 7.731236457824707
    },
    {
      "epoch": 2.470684051513672e-05,
      "model_forward_time": 1.773177146911621,
      "step": 99907
    },
    {
      "epoch": 2.470684051513672e-05,
      "step": 99907,
      "training_step_time": 7.704773664474487
    },
    {
      "epoch": 2.4707794189453126e-05,
      "model_forward_time": 1.7711827754974365,
      "step": 99908
    },
    {
      "epoch": 2.4707794189453126e-05,
      "step": 99908,
      "training_step_time": 7.73343825340271
    },
    {
      "epoch": 2.470874786376953e-05,
      "model_forward_time": 1.7706587314605713,
      "step": 99909
    },
    {
      "epoch": 2.470874786376953e-05,
      "step": 99909,
      "training_step_time": 7.751916408538818
    },
    {
      "epoch": 2.4709701538085938e-05,
      "grad_norm": 0.10320210456848145,
      "learning_rate": 2.2639925863487955e-11,
      "loss": 0.0532,
      "step": 99910
    },
    {
      "epoch": 2.4709701538085938e-05,
      "model_forward_time": 1.767185926437378,
      "step": 99910
    },
    {
      "dynamics_loss_avg": 0.06845715343952179,
      "epoch": 2.4709701538085938e-05,
      "step": 99910
    },
    {
      "action_loss_avg": 0.0016913084880798123,
      "epoch": 2.4709701538085938e-05,
      "step": 99910
    },
    {
      "epoch": 2.4709701538085938e-05,
      "step": 99910,
      "training_step_time": 7.590633869171143
    },
    {
      "epoch": 2.4710655212402345e-05,
      "model_forward_time": 1.7651042938232422,
      "step": 99911
    },
    {
      "epoch": 2.4710655212402345e-05,
      "step": 99911,
      "training_step_time": 9.745460271835327
    },
    {
      "epoch": 2.471160888671875e-05,
      "model_forward_time": 1.824338674545288,
      "step": 99912
    },
    {
      "epoch": 2.471160888671875e-05,
      "step": 99912,
      "training_step_time": 10.535337209701538
    },
    {
      "epoch": 2.4712562561035156e-05,
      "model_forward_time": 2.418894052505493,
      "step": 99913
    },
    {
      "epoch": 2.4712562561035156e-05,
      "step": 99913,
      "training_step_time": 10.47391414642334
    },
    {
      "epoch": 2.4713516235351564e-05,
      "model_forward_time": 1.7954037189483643,
      "step": 99914
    },
    {
      "epoch": 2.4713516235351564e-05,
      "step": 99914,
      "training_step_time": 7.79568886756897
    },
    {
      "epoch": 2.4714469909667968e-05,
      "model_forward_time": 1.7665867805480957,
      "step": 99915
    },
    {
      "epoch": 2.4714469909667968e-05,
      "step": 99915,
      "training_step_time": 7.752733469009399
    },
    {
      "epoch": 2.4715423583984375e-05,
      "model_forward_time": 1.7756803035736084,
      "step": 99916
    },
    {
      "epoch": 2.4715423583984375e-05,
      "step": 99916,
      "training_step_time": 7.683691024780273
    },
    {
      "epoch": 2.4716377258300783e-05,
      "model_forward_time": 1.398951768875122,
      "step": 99917
    },
    {
      "epoch": 2.4716377258300783e-05,
      "step": 99917,
      "training_step_time": 7.788280248641968
    },
    {
      "epoch": 2.4717330932617187e-05,
      "model_forward_time": 1.7779152393341064,
      "step": 99918
    },
    {
      "epoch": 2.4717330932617187e-05,
      "step": 99918,
      "training_step_time": 8.182483911514282
    },
    {
      "epoch": 2.4718284606933594e-05,
      "model_forward_time": 1.7653567790985107,
      "step": 99919
    },
    {
      "epoch": 2.4718284606933594e-05,
      "step": 99919,
      "training_step_time": 7.598198890686035
    },
    {
      "epoch": 2.471923828125e-05,
      "grad_norm": 0.07824067771434784,
      "learning_rate": 1.7937516830213166e-11,
      "loss": 0.0536,
      "step": 99920
    },
    {
      "epoch": 2.471923828125e-05,
      "model_forward_time": 1.7665927410125732,
      "step": 99920
    },
    {
      "dynamics_loss_avg": 0.04770663883537054,
      "epoch": 2.471923828125e-05,
      "step": 99920
    },
    {
      "action_loss_avg": 0.001318387733772397,
      "epoch": 2.471923828125e-05,
      "step": 99920
    },
    {
      "epoch": 2.471923828125e-05,
      "step": 99920,
      "training_step_time": 7.6834845542907715
    },
    {
      "epoch": 2.4720191955566406e-05,
      "model_forward_time": 1.8048474788665771,
      "step": 99921
    },
    {
      "epoch": 2.4720191955566406e-05,
      "step": 99921,
      "training_step_time": 10.690625190734863
    },
    {
      "epoch": 2.4721145629882813e-05,
      "model_forward_time": 2.373408317565918,
      "step": 99922
    },
    {
      "epoch": 2.4721145629882813e-05,
      "step": 99922,
      "training_step_time": 8.681142568588257
    },
    {
      "epoch": 2.4722099304199217e-05,
      "model_forward_time": 1.7525947093963623,
      "step": 99923
    },
    {
      "epoch": 2.4722099304199217e-05,
      "step": 99923,
      "training_step_time": 8.699973583221436
    },
    {
      "epoch": 2.4723052978515625e-05,
      "model_forward_time": 1.7569851875305176,
      "step": 99924
    },
    {
      "epoch": 2.4723052978515625e-05,
      "step": 99924,
      "training_step_time": 8.400490999221802
    },
    {
      "epoch": 2.4724006652832032e-05,
      "model_forward_time": 1.7484519481658936,
      "step": 99925
    },
    {
      "epoch": 2.4724006652832032e-05,
      "step": 99925,
      "training_step_time": 7.833473443984985
    },
    {
      "epoch": 2.4724960327148436e-05,
      "model_forward_time": 1.7511825561523438,
      "step": 99926
    },
    {
      "epoch": 2.4724960327148436e-05,
      "step": 99926,
      "training_step_time": 7.794105052947998
    },
    {
      "epoch": 2.4725914001464843e-05,
      "model_forward_time": 1.7338743209838867,
      "step": 99927
    },
    {
      "epoch": 2.4725914001464843e-05,
      "step": 99927,
      "training_step_time": 7.541029691696167
    },
    {
      "epoch": 2.472686767578125e-05,
      "model_forward_time": 1.7334415912628174,
      "step": 99928
    },
    {
      "epoch": 2.472686767578125e-05,
      "step": 99928,
      "training_step_time": 7.822896242141724
    },
    {
      "epoch": 2.4727821350097655e-05,
      "model_forward_time": 1.7385280132293701,
      "step": 99929
    },
    {
      "epoch": 2.4727821350097655e-05,
      "step": 99929,
      "training_step_time": 7.49921989440918
    },
    {
      "epoch": 2.4728775024414062e-05,
      "grad_norm": 0.08925170451402664,
      "learning_rate": 1.3781898318687703e-11,
      "loss": 0.0533,
      "step": 99930
    },
    {
      "epoch": 2.4728775024414062e-05,
      "model_forward_time": 1.7544314861297607,
      "step": 99930
    },
    {
      "dynamics_loss_avg": 0.06532651782035828,
      "epoch": 2.4728775024414062e-05,
      "step": 99930
    },
    {
      "action_loss_avg": 0.0018392852391116322,
      "epoch": 2.4728775024414062e-05,
      "step": 99930
    },
    {
      "epoch": 2.4728775024414062e-05,
      "step": 99930,
      "training_step_time": 8.032946586608887
    },
    {
      "epoch": 2.472972869873047e-05,
      "model_forward_time": 2.4067208766937256,
      "step": 99931
    },
    {
      "epoch": 2.472972869873047e-05,
      "step": 99931,
      "training_step_time": 8.535740613937378
    },
    {
      "epoch": 2.4730682373046874e-05,
      "model_forward_time": 1.7574961185455322,
      "step": 99932
    },
    {
      "epoch": 2.4730682373046874e-05,
      "step": 99932,
      "training_step_time": 8.435080289840698
    },
    {
      "epoch": 2.473163604736328e-05,
      "model_forward_time": 1.735520362854004,
      "step": 99933
    },
    {
      "epoch": 2.473163604736328e-05,
      "step": 99933,
      "training_step_time": 8.517886877059937
    },
    {
      "epoch": 2.473258972167969e-05,
      "model_forward_time": 1.7367753982543945,
      "step": 99934
    },
    {
      "epoch": 2.473258972167969e-05,
      "step": 99934,
      "training_step_time": 7.784206390380859
    },
    {
      "epoch": 2.4733543395996093e-05,
      "model_forward_time": 1.7355730533599854,
      "step": 99935
    },
    {
      "epoch": 2.4733543395996093e-05,
      "step": 99935,
      "training_step_time": 7.845656394958496
    },
    {
      "epoch": 2.47344970703125e-05,
      "model_forward_time": 1.7383027076721191,
      "step": 99936
    },
    {
      "epoch": 2.47344970703125e-05,
      "step": 99936,
      "training_step_time": 7.941795110702515
    },
    {
      "epoch": 2.4735450744628907e-05,
      "model_forward_time": 1.7343435287475586,
      "step": 99937
    },
    {
      "epoch": 2.4735450744628907e-05,
      "step": 99937,
      "training_step_time": 7.820186376571655
    },
    {
      "epoch": 2.473640441894531e-05,
      "model_forward_time": 1.7439959049224854,
      "step": 99938
    },
    {
      "epoch": 2.473640441894531e-05,
      "step": 99938,
      "training_step_time": 7.619186162948608
    },
    {
      "epoch": 2.473735809326172e-05,
      "model_forward_time": 1.7544326782226562,
      "step": 99939
    },
    {
      "epoch": 2.473735809326172e-05,
      "step": 99939,
      "training_step_time": 8.386680841445923
    },
    {
      "epoch": 2.4738311767578126e-05,
      "grad_norm": 0.13356229662895203,
      "learning_rate": 1.0173070781882566e-11,
      "loss": 0.0534,
      "step": 99940
    },
    {
      "epoch": 2.4738311767578126e-05,
      "model_forward_time": 2.3975636959075928,
      "step": 99940
    },
    {
      "dynamics_loss_avg": 0.05654505416750908,
      "epoch": 2.4738311767578126e-05,
      "step": 99940
    },
    {
      "action_loss_avg": 0.0027807591832242906,
      "epoch": 2.4738311767578126e-05,
      "step": 99940
    },
    {
      "epoch": 2.4738311767578126e-05,
      "step": 99940,
      "training_step_time": 8.082849740982056
    },
    {
      "epoch": 2.473926544189453e-05,
      "model_forward_time": 1.7681365013122559,
      "step": 99941
    },
    {
      "epoch": 2.473926544189453e-05,
      "step": 99941,
      "training_step_time": 8.77835726737976
    },
    {
      "epoch": 2.4740219116210938e-05,
      "model_forward_time": 1.7321114540100098,
      "step": 99942
    },
    {
      "epoch": 2.4740219116210938e-05,
      "step": 99942,
      "training_step_time": 7.787448406219482
    },
    {
      "epoch": 2.4741172790527345e-05,
      "model_forward_time": 1.7396798133850098,
      "step": 99943
    },
    {
      "epoch": 2.4741172790527345e-05,
      "step": 99943,
      "training_step_time": 8.605783224105835
    },
    {
      "epoch": 2.474212646484375e-05,
      "model_forward_time": 1.737708330154419,
      "step": 99944
    },
    {
      "epoch": 2.474212646484375e-05,
      "step": 99944,
      "training_step_time": 7.840076684951782
    },
    {
      "epoch": 2.4743080139160157e-05,
      "model_forward_time": 1.3601973056793213,
      "step": 99945
    },
    {
      "epoch": 2.4743080139160157e-05,
      "step": 99945,
      "training_step_time": 7.590275526046753
    },
    {
      "epoch": 2.4744033813476564e-05,
      "model_forward_time": 1.7344141006469727,
      "step": 99946
    },
    {
      "epoch": 2.4744033813476564e-05,
      "step": 99946,
      "training_step_time": 7.8972694873809814
    },
    {
      "epoch": 2.4744987487792968e-05,
      "model_forward_time": 1.736126184463501,
      "step": 99947
    },
    {
      "epoch": 2.4744987487792968e-05,
      "step": 99947,
      "training_step_time": 8.389369487762451
    },
    {
      "epoch": 2.4745941162109375e-05,
      "model_forward_time": 1.7472307682037354,
      "step": 99948
    },
    {
      "epoch": 2.4745941162109375e-05,
      "step": 99948,
      "training_step_time": 8.811752080917358
    },
    {
      "epoch": 2.4746894836425783e-05,
      "model_forward_time": 2.3294010162353516,
      "step": 99949
    },
    {
      "epoch": 2.4746894836425783e-05,
      "step": 99949,
      "training_step_time": 9.98477029800415
    },
    {
      "epoch": 2.4747848510742187e-05,
      "grad_norm": 0.08823344856500626,
      "learning_rate": 7.1110346155922605e-12,
      "loss": 0.0545,
      "step": 99950
    },
    {
      "epoch": 2.4747848510742187e-05,
      "model_forward_time": 1.7567594051361084,
      "step": 99950
    },
    {
      "dynamics_loss_avg": 0.04519810676574707,
      "epoch": 2.4747848510742187e-05,
      "step": 99950
    },
    {
      "action_loss_avg": 0.0010635583370458336,
      "epoch": 2.4747848510742187e-05,
      "step": 99950
    },
    {
      "epoch": 2.4747848510742187e-05,
      "step": 99950,
      "training_step_time": 10.35695743560791
    },
    {
      "epoch": 2.4748802185058594e-05,
      "model_forward_time": 1.7364399433135986,
      "step": 99951
    },
    {
      "epoch": 2.4748802185058594e-05,
      "step": 99951,
      "training_step_time": 7.631813287734985
    },
    {
      "epoch": 2.4749755859375e-05,
      "model_forward_time": 1.7333297729492188,
      "step": 99952
    },
    {
      "epoch": 2.4749755859375e-05,
      "step": 99952,
      "training_step_time": 7.704869270324707
    },
    {
      "epoch": 2.4750709533691406e-05,
      "model_forward_time": 1.734663486480713,
      "step": 99953
    },
    {
      "epoch": 2.4750709533691406e-05,
      "step": 99953,
      "training_step_time": 7.679132699966431
    },
    {
      "epoch": 2.4751663208007813e-05,
      "model_forward_time": 1.7398159503936768,
      "step": 99954
    },
    {
      "epoch": 2.4751663208007813e-05,
      "step": 99954,
      "training_step_time": 7.453268766403198
    },
    {
      "epoch": 2.4752616882324217e-05,
      "model_forward_time": 1.7413973808288574,
      "step": 99955
    },
    {
      "epoch": 2.4752616882324217e-05,
      "step": 99955,
      "training_step_time": 7.6941773891448975
    },
    {
      "epoch": 2.4753570556640625e-05,
      "model_forward_time": 1.737471103668213,
      "step": 99956
    },
    {
      "epoch": 2.4753570556640625e-05,
      "step": 99956,
      "training_step_time": 7.6864094734191895
    },
    {
      "epoch": 2.4754524230957032e-05,
      "model_forward_time": 1.7351267337799072,
      "step": 99957
    },
    {
      "epoch": 2.4754524230957032e-05,
      "step": 99957,
      "training_step_time": 8.797537326812744
    },
    {
      "epoch": 2.4755477905273436e-05,
      "model_forward_time": 1.75197434425354,
      "step": 99958
    },
    {
      "epoch": 2.4755477905273436e-05,
      "step": 99958,
      "training_step_time": 8.467084407806396
    },
    {
      "epoch": 2.4756431579589843e-05,
      "model_forward_time": 2.420614004135132,
      "step": 99959
    },
    {
      "epoch": 2.4756431579589843e-05,
      "step": 99959,
      "training_step_time": 9.107101678848267
    },
    {
      "epoch": 2.475738525390625e-05,
      "grad_norm": 0.13810868561267853,
      "learning_rate": 4.5957901539939174e-12,
      "loss": 0.0537,
      "step": 99960
    },
    {
      "epoch": 2.475738525390625e-05,
      "model_forward_time": 1.7371642589569092,
      "step": 99960
    },
    {
      "dynamics_loss_avg": 0.05318062584847212,
      "epoch": 2.475738525390625e-05,
      "step": 99960
    },
    {
      "action_loss_avg": 0.0018081606336636468,
      "epoch": 2.475738525390625e-05,
      "step": 99960
    },
    {
      "epoch": 2.475738525390625e-05,
      "step": 99960,
      "training_step_time": 7.731476783752441
    },
    {
      "epoch": 2.4758338928222655e-05,
      "model_forward_time": 1.7405595779418945,
      "step": 99961
    },
    {
      "epoch": 2.4758338928222655e-05,
      "step": 99961,
      "training_step_time": 7.71690821647644
    },
    {
      "epoch": 2.4759292602539062e-05,
      "model_forward_time": 1.737546682357788,
      "step": 99962
    },
    {
      "epoch": 2.4759292602539062e-05,
      "step": 99962,
      "training_step_time": 7.735071897506714
    },
    {
      "epoch": 2.476024627685547e-05,
      "model_forward_time": 1.7383155822753906,
      "step": 99963
    },
    {
      "epoch": 2.476024627685547e-05,
      "step": 99963,
      "training_step_time": 7.395493745803833
    },
    {
      "epoch": 2.4761199951171874e-05,
      "model_forward_time": 1.7313077449798584,
      "step": 99964
    },
    {
      "epoch": 2.4761199951171874e-05,
      "step": 99964,
      "training_step_time": 7.700064182281494
    },
    {
      "epoch": 2.476215362548828e-05,
      "model_forward_time": 1.73716139793396,
      "step": 99965
    },
    {
      "epoch": 2.476215362548828e-05,
      "step": 99965,
      "training_step_time": 8.233951807022095
    },
    {
      "epoch": 2.476310729980469e-05,
      "model_forward_time": 1.7362616062164307,
      "step": 99966
    },
    {
      "epoch": 2.476310729980469e-05,
      "step": 99966,
      "training_step_time": 7.912065029144287
    },
    {
      "epoch": 2.4764060974121093e-05,
      "model_forward_time": 1.74552321434021,
      "step": 99967
    },
    {
      "epoch": 2.4764060974121093e-05,
      "step": 99967,
      "training_step_time": 8.554308414459229
    },
    {
      "epoch": 2.47650146484375e-05,
      "model_forward_time": 2.4029009342193604,
      "step": 99968
    },
    {
      "epoch": 2.47650146484375e-05,
      "step": 99968,
      "training_step_time": 8.446923971176147
    },
    {
      "epoch": 2.4765968322753907e-05,
      "model_forward_time": 1.740884780883789,
      "step": 99969
    },
    {
      "epoch": 2.4765968322753907e-05,
      "step": 99969,
      "training_step_time": 8.849449157714844
    },
    {
      "epoch": 2.476692199707031e-05,
      "grad_norm": 0.09055984765291214,
      "learning_rate": 2.627337672977959e-12,
      "loss": 0.0556,
      "step": 99970
    },
    {
      "epoch": 2.476692199707031e-05,
      "model_forward_time": 1.7394776344299316,
      "step": 99970
    },
    {
      "dynamics_loss_avg": 0.058686851896345614,
      "epoch": 2.476692199707031e-05,
      "step": 99970
    },
    {
      "action_loss_avg": 0.001969146789633669,
      "epoch": 2.476692199707031e-05,
      "step": 99970
    },
    {
      "epoch": 2.476692199707031e-05,
      "step": 99970,
      "training_step_time": 7.811147689819336
    },
    {
      "epoch": 2.476787567138672e-05,
      "model_forward_time": 1.718461513519287,
      "step": 99971
    },
    {
      "epoch": 2.476787567138672e-05,
      "step": 99971,
      "training_step_time": 8.901798486709595
    },
    {
      "epoch": 2.4768829345703126e-05,
      "model_forward_time": 1.7402334213256836,
      "step": 99972
    },
    {
      "epoch": 2.4768829345703126e-05,
      "step": 99972,
      "training_step_time": 7.50753116607666
    },
    {
      "epoch": 2.476978302001953e-05,
      "model_forward_time": 1.740508794784546,
      "step": 99973
    },
    {
      "epoch": 2.476978302001953e-05,
      "step": 99973,
      "training_step_time": 7.423017978668213
    },
    {
      "epoch": 2.4770736694335938e-05,
      "model_forward_time": 1.7321102619171143,
      "step": 99974
    },
    {
      "epoch": 2.4770736694335938e-05,
      "step": 99974,
      "training_step_time": 7.811441421508789
    },
    {
      "epoch": 2.4771690368652345e-05,
      "model_forward_time": 1.7332344055175781,
      "step": 99975
    },
    {
      "epoch": 2.4771690368652345e-05,
      "step": 99975,
      "training_step_time": 7.912260055541992
    },
    {
      "epoch": 2.477264404296875e-05,
      "model_forward_time": 1.7424383163452148,
      "step": 99976
    },
    {
      "epoch": 2.477264404296875e-05,
      "step": 99976,
      "training_step_time": 9.023049354553223
    },
    {
      "epoch": 2.4773597717285157e-05,
      "model_forward_time": 2.383230447769165,
      "step": 99977
    },
    {
      "epoch": 2.4773597717285157e-05,
      "step": 99977,
      "training_step_time": 9.821012020111084
    },
    {
      "epoch": 2.4774551391601564e-05,
      "model_forward_time": 1.7548952102661133,
      "step": 99978
    },
    {
      "epoch": 2.4774551391601564e-05,
      "step": 99978,
      "training_step_time": 8.652595043182373
    },
    {
      "epoch": 2.4775505065917968e-05,
      "model_forward_time": 1.7215607166290283,
      "step": 99979
    },
    {
      "epoch": 2.4775505065917968e-05,
      "step": 99979,
      "training_step_time": 7.686795711517334
    },
    {
      "epoch": 2.4776458740234375e-05,
      "grad_norm": 0.07317827641963959,
      "learning_rate": 1.2056773868174276e-12,
      "loss": 0.0536,
      "step": 99980
    },
    {
      "epoch": 2.4776458740234375e-05,
      "model_forward_time": 1.70719313621521,
      "step": 99980
    },
    {
      "dynamics_loss_avg": 0.053355440869927405,
      "epoch": 2.4776458740234375e-05,
      "step": 99980
    },
    {
      "action_loss_avg": 0.00193147849640809,
      "epoch": 2.4776458740234375e-05,
      "step": 99980
    },
    {
      "epoch": 2.4776458740234375e-05,
      "step": 99980,
      "training_step_time": 7.811326265335083
    },
    {
      "epoch": 2.4777412414550783e-05,
      "model_forward_time": 1.73274827003479,
      "step": 99981
    },
    {
      "epoch": 2.4777412414550783e-05,
      "step": 99981,
      "training_step_time": 7.791191101074219
    },
    {
      "epoch": 2.4778366088867187e-05,
      "model_forward_time": 0.7021629810333252,
      "step": 99982
    },
    {
      "epoch": 2.4778366088867187e-05,
      "step": 99982,
      "training_step_time": 7.3495471477508545
    },
    {
      "epoch": 2.4779319763183594e-05,
      "model_forward_time": 1.73050856590271,
      "step": 99983
    },
    {
      "epoch": 2.4779319763183594e-05,
      "step": 99983,
      "training_step_time": 7.921630382537842
    },
    {
      "epoch": 2.47802734375e-05,
      "model_forward_time": 1.722621202468872,
      "step": 99984
    },
    {
      "epoch": 2.47802734375e-05,
      "step": 99984,
      "training_step_time": 7.83951210975647
    },
    {
      "epoch": 2.4781227111816406e-05,
      "model_forward_time": 1.7254118919372559,
      "step": 99985
    },
    {
      "epoch": 2.4781227111816406e-05,
      "step": 99985,
      "training_step_time": 9.998012781143188
    },
    {
      "epoch": 2.4782180786132813e-05,
      "model_forward_time": 1.747664213180542,
      "step": 99986
    },
    {
      "epoch": 2.4782180786132813e-05,
      "step": 99986,
      "training_step_time": 9.639629602432251
    },
    {
      "epoch": 2.4783134460449217e-05,
      "model_forward_time": 2.396000385284424,
      "step": 99987
    },
    {
      "epoch": 2.4783134460449217e-05,
      "step": 99987,
      "training_step_time": 9.571690320968628
    },
    {
      "epoch": 2.4784088134765625e-05,
      "model_forward_time": 1.7248358726501465,
      "step": 99988
    },
    {
      "epoch": 2.4784088134765625e-05,
      "step": 99988,
      "training_step_time": 8.25198745727539
    },
    {
      "epoch": 2.4785041809082032e-05,
      "model_forward_time": 1.7234926223754883,
      "step": 99989
    },
    {
      "epoch": 2.4785041809082032e-05,
      "step": 99989,
      "training_step_time": 7.722560167312622
    },
    {
      "epoch": 2.4785995483398436e-05,
      "grad_norm": 0.09689880162477493,
      "learning_rate": 3.308094514986593e-13,
      "loss": 0.0536,
      "step": 99990
    },
    {
      "epoch": 2.4785995483398436e-05,
      "model_forward_time": 1.719973087310791,
      "step": 99990
    },
    {
      "dynamics_loss_avg": 0.04731303472071886,
      "epoch": 2.4785995483398436e-05,
      "step": 99990
    },
    {
      "action_loss_avg": 0.0013079453288810327,
      "epoch": 2.4785995483398436e-05,
      "step": 99990
    },
    {
      "epoch": 2.4785995483398436e-05,
      "step": 99990,
      "training_step_time": 8.55443024635315
    },
    {
      "epoch": 2.4786949157714843e-05,
      "model_forward_time": 1.729431390762329,
      "step": 99991
    },
    {
      "epoch": 2.4786949157714843e-05,
      "step": 99991,
      "training_step_time": 7.269216299057007
    },
    {
      "epoch": 2.478790283203125e-05,
      "model_forward_time": 1.727708101272583,
      "step": 99992
    },
    {
      "epoch": 2.478790283203125e-05,
      "step": 99992,
      "training_step_time": 7.752438545227051
    },
    {
      "epoch": 2.4788856506347655e-05,
      "model_forward_time": 1.7234408855438232,
      "step": 99993
    },
    {
      "epoch": 2.4788856506347655e-05,
      "step": 99993,
      "training_step_time": 8.506673336029053
    },
    {
      "epoch": 2.4789810180664062e-05,
      "model_forward_time": 1.724066972732544,
      "step": 99994
    },
    {
      "epoch": 2.4789810180664062e-05,
      "step": 99994,
      "training_step_time": 9.21257734298706
    },
    {
      "epoch": 2.479076385498047e-05,
      "model_forward_time": 1.7458398342132568,
      "step": 99995
    },
    {
      "epoch": 2.479076385498047e-05,
      "step": 99995,
      "training_step_time": 10.155122756958008
    },
    {
      "epoch": 2.4791717529296874e-05,
      "model_forward_time": 2.429248094558716,
      "step": 99996
    },
    {
      "epoch": 2.4791717529296874e-05,
      "step": 99996,
      "training_step_time": 10.004191637039185
    },
    {
      "epoch": 2.479267120361328e-05,
      "model_forward_time": 1.726564884185791,
      "step": 99997
    },
    {
      "epoch": 2.479267120361328e-05,
      "step": 99997,
      "training_step_time": 8.828680515289307
    },
    {
      "epoch": 2.479362487792969e-05,
      "model_forward_time": 1.7603912353515625,
      "step": 99998
    },
    {
      "epoch": 2.479362487792969e-05,
      "step": 99998,
      "training_step_time": 7.839264154434204
    },
    {
      "epoch": 2.4794578552246093e-05,
      "model_forward_time": 1.7248995304107666,
      "step": 99999
    },
    {
      "epoch": 2.4794578552246093e-05,
      "step": 99999,
      "training_step_time": 7.692425012588501
    },
    {
      "epoch": 2.47955322265625e-05,
      "grad_norm": 0.07403672486543655,
      "learning_rate": 2.733962500833798e-15,
      "loss": 0.0535,
      "step": 100000
    }
  ],
  "logging_steps": 10,
  "max_steps": 100000,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 1,
  "save_steps": 1000,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": true
      },
      "attributes": {}
    }
  },
  "total_flos": 0.0,
  "train_batch_size": 1,
  "trial_name": null,
  "trial_params": null
}